Regular Testing for Non-Shuffled Data¶
In [1]:
"""
This file tests the model's performance on the testing dataset.
For CLS, this script returns the testing accuracy.
For Grasp, this script returns the testing accuracy and visualizes the
grasp prediction.
Comment or uncomment certain lines of code for swapping between
training CLS model and Grasping model.
E.g. Uncomment the lines with NO SPACE between '#' and the codes:
# Get test acc for CLS model
#accuracy, loss = get_test_acc(model)
# Get test acc for Grasp model
accuracy, loss = get_grasp_acc(model)
----->
# Get test acc for CLS model
accuracy, loss = get_test_acc(model)
# Get test acc for Grasp model
#accuracy, loss = get_grasp_acc(model)
"""
import torch
import os
import time
import sys
from utils.parameters import Params
from multi_task_models.grcn_multi_alex import Multi_AlexnetMap_v3
from training.single_task.evaluation import get_cls_acc, get_grasp_acc, visualize_grasp, visualize_cls
params = Params()
model_name = params.MODEL_NAME
weights_dir = params.MODEL_PATH
weights_path = os.path.join(weights_dir, model_name, model_name + '_final.pth')
# AlexNet with 1st, 2nd layer pretrained on Imagenet
model = Multi_AlexnetMap_v3().to('cuda')
model.load_state_dict(torch.load(weights_path))
model.eval()
path = params.TEST_PATH
# if len(sys.argv) > 1:
# path = params.TEST_PATH_SHUFFLE
# Visualize CLS predictions one by one
#visualize_cls(model)
# Visualize grasp predictions one by one
#visualize_grasp(model)
/home/billy/anaconda3/envs/gr_cls_multitask/lib/python3.11/site-packages/torchvision/models/_utils.py:208: UserWarning: The parameter 'pretrained' is deprecated since 0.13 and may be removed in the future, please use 'weights' instead. warnings.warn( /home/billy/anaconda3/envs/gr_cls_multitask/lib/python3.11/site-packages/torchvision/models/_utils.py:223: UserWarning: Arguments other than a weight enum or `None` for 'weights' are deprecated since 0.13 and may be removed in the future. The current behavior is equivalent to passing `weights=AlexNet_Weights.IMAGENET1K_V1`. You can also use `weights=AlexNet_Weights.DEFAULT` to get the most up-to-date weights. warnings.warn(msg) /tmp/ipykernel_37298/1692255469.py:41: FutureWarning: You are using `torch.load` with `weights_only=False` (the current default value), which uses the default pickle module implicitly. It is possible to construct malicious pickle data which will execute arbitrary code during unpickling (See https://github.com/pytorch/pytorch/blob/main/SECURITY.md#untrusted-models for more details). In a future release, the default value for `weights_only` will be flipped to `True`. This limits the functions that could be executed during unpickling. Arbitrary objects will no longer be allowed to be loaded via this mode unless they are explicitly allowlisted by the user via `torch.serialization.add_safe_globals`. We recommend you start setting `weights_only=True` for any use case where you don't have full control of the loaded file. Please open an issue on GitHub for any issues related to this experimental feature. model.load_state_dict(torch.load(weights_path))
In [2]:
def register_custom_hooks(model):
"""
Registers hooks on specific layers of the model.
Returns a dictionary to store activations and a list of hooks.
"""
layers_to_track = {
'rgb_features': 2,
'd_features': 2,
'features': 12,
'grasp': 4,
'cls': 4
}
activations = {layer_name: [] for layer_name in layers_to_track}
def hook_fn(layer_name):
def hook(module, input, output):
activations[layer_name].append(output.clone().detach())
return hook
hooks = []
for layer_name, layer_idx in layers_to_track.items():
if layer_name in ['rgb_features', 'd_features']:
block = getattr(model, layer_name)
else:
block = getattr(model, layer_name.split('_')[0])
hooks.append(block[layer_idx].register_forward_hook(hook_fn(layer_name)))
return hooks, activations
In [3]:
# Register hooks
hooks, activations = register_custom_hooks(model)
In [4]:
# Get test acc for CLS model
c_accuracy, c_loss = get_cls_acc(model, include_depth=True, seed=None, dataset=path, truncation=None)
Dataset size: 400 Training steps: 360 -- Val steps: 40
In [6]:
activations
Out[6]:
{0: [tensor([[[[ 2.5037e-01, 5.2440e-01, 6.2645e-01, ..., -3.2223e-01,
-5.7417e-02, 4.9824e-02],
[-2.1219e+00, -1.3909e+00, -3.0508e+00, ..., -3.2132e+00,
-1.2632e+00, -3.8997e-01],
[-1.1038e-01, -9.0209e-01, -2.5217e-02, ..., -2.0783e-01,
-3.3751e-01, 1.9400e-01],
...,
[-1.0035e+00, -2.1310e+00, -1.9921e+00, ..., -1.9065e+00,
-1.9084e+00, -1.2866e+00],
[ 6.8005e-01, -2.1758e-01, -1.9644e-01, ..., 3.6869e-02,
2.6509e-01, 4.3772e-01],
[ 1.1443e-01, -6.5849e-01, -4.6432e-01, ..., 2.3581e-01,
4.6474e-01, 4.8898e-01]],
[[-1.7578e+00, -1.2954e+00, -4.2917e+00, ..., -6.8321e+00,
-6.0321e+00, -4.5300e+00],
[ 6.2969e-01, 1.8516e+00, -2.3204e+00, ..., -5.4585e+00,
-5.5297e+00, -4.2750e+00],
[ 1.4777e+00, 1.6402e+00, -4.9915e+00, ..., -7.8593e+00,
-7.8068e+00, -6.5895e+00],
...,
[ 3.2641e-01, 2.2916e-01, -4.1839e+00, ..., -7.0922e+00,
-6.6609e+00, -5.1738e+00],
[ 2.3751e+00, 2.2639e+00, -1.6156e-01, ..., -1.9327e+00,
-2.5581e+00, -2.0407e+00],
[ 1.3975e+00, 1.1700e+00, -1.7592e+00, ..., -2.9166e+00,
-3.4608e+00, -3.0252e+00]],
[[-1.7271e-01, -1.3080e+00, -1.5413e+00, ..., -1.3798e+00,
-1.0638e+00, -1.3568e+00],
[-6.1406e-01, -9.7800e-01, -9.9041e-01, ..., -9.5055e-01,
-6.1399e-01, -9.0428e-01],
[-4.4269e-01, -3.2213e-01, -4.7946e-01, ..., -1.0939e+00,
-5.7410e-01, -4.6736e-01],
...,
[-1.2336e-02, -5.1309e-01, -5.7507e-01, ..., -1.0383e+00,
-8.7914e-01, -1.2459e+00],
[-6.6577e-01, -1.6918e-01, -5.3715e-01, ..., -5.5014e-01,
-2.4417e-01, -4.2072e-01],
[-8.1542e-01, -2.3761e-01, -4.7011e-01, ..., -4.8333e-01,
-4.2576e-01, -5.9602e-01]],
...,
[[-3.0393e+00, -2.5344e+00, -2.6139e+00, ..., -3.0877e+00,
-2.5162e+00, -1.8523e+00],
[-2.1963e+00, -7.4293e-01, -1.1047e+00, ..., -2.4422e+00,
-2.4751e+00, -1.9532e+00],
[-1.1889e+00, 1.0268e+00, 9.0316e-01, ..., -1.1677e-01,
-1.7732e-02, 2.8452e-01],
...,
[-2.1957e+00, -3.2357e-01, -1.9443e-01, ..., -9.1556e-01,
-2.4093e-01, 3.0395e-01],
[-1.1804e+00, 7.6027e-01, 9.6734e-01, ..., 4.2595e-01,
4.7996e-01, 8.1975e-01],
[-2.0748e+00, -1.4079e+00, -1.6137e+00, ..., -2.2171e+00,
-1.1821e+00, -3.7218e-01]],
[[-1.6403e+00, -1.5982e+00, -2.8728e+00, ..., -2.2138e+00,
-1.9045e+00, -1.8100e+00],
[-6.0836e-01, 1.3556e-01, -3.2426e+00, ..., -2.8360e+00,
-2.8623e+00, -2.9031e+00],
[-2.0659e+00, -8.2529e-01, -5.3776e+00, ..., -4.4367e+00,
-4.5033e+00, -3.7491e+00],
...,
[-1.3165e+00, 1.3866e-01, -3.3326e+00, ..., -2.8160e+00,
-2.7354e+00, -2.3088e+00],
[-1.6429e+00, -4.6726e-01, -2.9522e+00, ..., -2.4406e+00,
-2.2431e+00, -1.6530e+00],
[-2.2754e+00, -1.3281e+00, -2.9780e+00, ..., -2.8514e+00,
-2.4081e+00, -2.1903e+00]],
[[-1.3578e+00, -1.9260e+00, -1.7037e+00, ..., -2.5586e+00,
-2.1375e+00, -1.6706e+00],
[-1.2868e+00, -1.7761e+00, -1.2178e+00, ..., -2.3966e+00,
-1.9256e+00, -1.5850e+00],
[-1.6146e+00, -1.9520e+00, -1.4454e+00, ..., -2.8054e+00,
-2.2415e+00, -1.7406e+00],
...,
[-2.4350e+00, -3.0307e+00, -2.7825e+00, ..., -3.9968e+00,
-3.2442e+00, -2.5407e+00],
[-1.5888e+00, -2.0851e+00, -1.9924e+00, ..., -2.7599e+00,
-2.3139e+00, -1.9107e+00],
[-1.0392e+00, -1.3539e+00, -1.3793e+00, ..., -1.8255e+00,
-1.5217e+00, -1.1777e+00]]],
[[[ 4.6499e-01, 9.4721e-01, 8.7816e-01, ..., -3.4108e-01,
-7.5264e-02, 1.3208e-02],
[-1.2480e+00, -3.3948e-01, -1.4714e+00, ..., -3.2456e+00,
-1.2860e+00, -4.4411e-01],
[ 5.5780e-01, 3.9930e-01, 3.8548e-01, ..., -1.7658e-01,
-3.4090e-01, 1.7342e-01],
...,
[-1.7017e-01, -8.1827e-01, -1.5019e+00, ..., -2.0053e+00,
-1.8853e+00, -1.2856e+00],
[ 1.0855e+00, 8.2625e-01, 3.1033e-01, ..., -2.4488e-02,
2.9910e-01, 4.5979e-01],
[ 7.2127e-01, 5.3969e-01, 1.8616e-01, ..., 1.9416e-01,
4.9541e-01, 5.0270e-01]],
[[-2.2713e+00, -1.0994e+00, -4.1119e+00, ..., -6.8672e+00,
-6.0769e+00, -4.5367e+00],
[-2.5974e-01, 1.6780e+00, -2.5495e+00, ..., -5.4524e+00,
-5.5632e+00, -4.2999e+00],
[ 4.2743e-01, 1.4499e+00, -5.1002e+00, ..., -7.8624e+00,
-7.8317e+00, -6.6117e+00],
...,
[-4.5435e-01, 6.1797e-01, -4.0513e+00, ..., -7.0775e+00,
-6.6958e+00, -5.1430e+00],
[ 2.1097e+00, 2.7979e+00, 4.0307e-01, ..., -1.9011e+00,
-2.5600e+00, -2.0109e+00],
[ 1.2181e+00, 1.5315e+00, -1.4468e+00, ..., -2.9256e+00,
-3.4870e+00, -3.0316e+00]],
[[-5.8296e-01, -1.5675e+00, -1.3908e+00, ..., -1.4123e+00,
-1.0558e+00, -1.3400e+00],
[-1.2859e+00, -1.6116e+00, -9.8404e-01, ..., -9.9793e-01,
-6.1788e-01, -8.9717e-01],
[-1.1305e+00, -9.8056e-01, -3.7674e-01, ..., -1.1039e+00,
-5.8586e-01, -4.6891e-01],
...,
[-9.3326e-01, -1.3503e+00, -5.3457e-01, ..., -1.0464e+00,
-8.6365e-01, -1.2353e+00],
[-1.2818e+00, -8.7267e-01, -5.4542e-01, ..., -5.5593e-01,
-2.3483e-01, -4.0188e-01],
[-1.3624e+00, -7.5728e-01, -3.5422e-01, ..., -5.1032e-01,
-4.4566e-01, -6.0316e-01]],
...,
[[-2.1688e+00, -1.2234e+00, -2.5273e+00, ..., -3.1059e+00,
-2.5174e+00, -1.8266e+00],
[-7.0127e-01, 9.2207e-01, -9.5684e-01, ..., -2.4247e+00,
-2.5068e+00, -2.0219e+00],
[-2.4143e-01, 2.1816e+00, 4.8805e-01, ..., -9.1181e-02,
3.6888e-02, 2.9363e-01],
...,
[-1.5056e+00, 1.2478e+00, -1.9469e-01, ..., -8.9869e-01,
-2.9014e-01, 2.9168e-01],
[-5.0468e-01, 1.5845e+00, 8.2083e-01, ..., 4.4458e-01,
4.7199e-01, 8.1266e-01],
[-1.6245e+00, -1.0918e+00, -1.6020e+00, ..., -2.2169e+00,
-1.1926e+00, -3.6175e-01]],
[[-1.2997e+00, -1.2208e+00, -2.5986e+00, ..., -2.2227e+00,
-1.9001e+00, -1.7967e+00],
[-6.2963e-01, -1.3815e-01, -3.3277e+00, ..., -2.8642e+00,
-2.8715e+00, -2.8860e+00],
[-2.1010e+00, -1.4281e+00, -5.6795e+00, ..., -4.4815e+00,
-4.5236e+00, -3.7097e+00],
...,
[-1.6307e+00, -4.9028e-01, -4.0950e+00, ..., -2.8065e+00,
-2.7151e+00, -2.2870e+00],
[-1.6555e+00, -7.1474e-01, -3.4230e+00, ..., -2.4275e+00,
-2.2279e+00, -1.6501e+00],
[-2.1074e+00, -1.3672e+00, -2.9762e+00, ..., -2.8595e+00,
-2.4070e+00, -2.1876e+00]],
[[-1.1496e+00, -1.7031e+00, -1.5229e+00, ..., -2.5698e+00,
-2.1466e+00, -1.6683e+00],
[-1.1560e+00, -1.6263e+00, -1.2100e+00, ..., -2.3947e+00,
-1.9274e+00, -1.5808e+00],
[-1.5051e+00, -1.8072e+00, -1.4204e+00, ..., -2.8072e+00,
-2.2395e+00, -1.7381e+00],
...,
[-2.3127e+00, -2.8836e+00, -2.6677e+00, ..., -4.0100e+00,
-3.2594e+00, -2.5567e+00],
[-1.5485e+00, -1.9694e+00, -1.9025e+00, ..., -2.7734e+00,
-2.3283e+00, -1.9264e+00],
[-1.0385e+00, -1.2392e+00, -1.2971e+00, ..., -1.8325e+00,
-1.5316e+00, -1.1888e+00]]],
[[[ 2.3972e-01, 5.0429e-01, 6.2171e-01, ..., 5.5033e-01,
9.3032e-01, 7.5942e-01],
[-1.9117e+00, -1.2295e+00, -2.7033e+00, ..., -1.4206e+00,
1.2786e-02, 5.8485e-01],
[-8.1309e-02, -8.3455e-01, 2.4990e-02, ..., 4.4103e-01,
9.0357e-01, 9.1664e-01],
...,
[-9.2803e-01, -2.0840e+00, -2.0183e+00, ..., -1.3383e+00,
-9.2153e-01, -6.6778e-01],
[ 8.1450e-01, -6.0550e-02, -8.3147e-02, ..., 4.3729e-01,
7.5962e-01, 8.6836e-01],
[ 2.4006e-01, -5.0381e-01, -3.4260e-01, ..., 6.0614e-01,
8.5199e-01, 7.8314e-01]],
[[-1.6871e+00, -1.3312e+00, -4.2315e+00, ..., -5.8443e+00,
-4.9459e+00, -3.8466e+00],
[ 4.3223e-01, 1.4685e+00, -2.5563e+00, ..., -4.7656e+00,
-4.4946e+00, -3.6738e+00],
[ 1.1494e+00, 1.1920e+00, -5.0885e+00, ..., -6.8979e+00,
-6.4340e+00, -5.8412e+00],
...,
[ 3.7049e-01, 3.8189e-01, -4.1316e+00, ..., -6.1148e+00,
-5.2321e+00, -4.3007e+00],
[ 2.5375e+00, 2.5114e+00, 1.1685e-01, ..., -1.4125e+00,
-1.4678e+00, -1.5376e+00],
[ 1.5141e+00, 1.3380e+00, -1.6254e+00, ..., -2.6996e+00,
-2.8218e+00, -2.7864e+00]],
[[-1.3318e-01, -1.1727e+00, -1.3876e+00, ..., -8.0813e-01,
-5.4462e-01, -6.8455e-01],
[-5.3329e-01, -8.7545e-01, -9.1811e-01, ..., -6.9340e-01,
-3.6487e-01, -2.4423e-01],
[-3.6556e-01, -2.7257e-01, -4.7216e-01, ..., -4.1267e-01,
-1.3836e-02, 8.0646e-02],
...,
[-1.0535e-01, -6.5303e-01, -6.2135e-01, ..., -6.9267e-01,
-5.0098e-01, -5.8524e-01],
[-7.4774e-01, -2.6362e-01, -5.7806e-01, ..., -2.9793e-01,
-2.7135e-02, 6.7822e-02],
[-9.0985e-01, -2.9294e-01, -4.5934e-01, ..., -2.8950e-01,
-2.0758e-01, -1.4458e-01]],
...,
[[-2.7907e+00, -2.3349e+00, -2.3617e+00, ..., -3.1576e+00,
-2.6069e+00, -1.7426e+00],
[-2.0522e+00, -7.3244e-01, -1.0628e+00, ..., -1.6786e+00,
-1.7892e+00, -1.4748e+00],
[-1.0923e+00, 8.8183e-01, 8.3771e-01, ..., -2.2156e-01,
2.0048e-02, 1.8637e-01],
...,
[-2.2256e+00, -9.4380e-02, -1.1115e-01, ..., -1.0689e+00,
-3.5197e-01, 2.0014e-01],
[-1.1897e+00, 9.1734e-01, 1.0785e+00, ..., 2.1278e-01,
3.6632e-01, 4.7138e-01],
[-2.1259e+00, -1.4441e+00, -1.6655e+00, ..., -2.2875e+00,
-1.2611e+00, -6.2029e-01]],
[[-1.6211e+00, -1.6331e+00, -2.8215e+00, ..., -1.3709e+00,
-7.8005e-01, -5.6546e-01],
[-7.0573e-01, -1.4040e-01, -3.1753e+00, ..., -2.0354e+00,
-1.6688e+00, -1.7468e+00],
[-2.0558e+00, -1.0062e+00, -5.1128e+00, ..., -4.0121e+00,
-3.3811e+00, -2.5850e+00],
...,
[-1.3489e+00, 1.5462e-01, -3.4609e+00, ..., -2.4468e+00,
-1.8010e+00, -1.3298e+00],
[-1.6668e+00, -4.6746e-01, -3.0629e+00, ..., -2.2907e+00,
-1.4989e+00, -8.6836e-01],
[-2.2997e+00, -1.3440e+00, -3.0375e+00, ..., -2.8788e+00,
-1.9361e+00, -1.6312e+00]],
[[-1.3006e+00, -1.8301e+00, -1.6590e+00, ..., -1.8975e+00,
-1.5373e+00, -1.1253e+00],
[-1.2420e+00, -1.7056e+00, -1.2368e+00, ..., -1.6622e+00,
-1.3150e+00, -1.0055e+00],
[-1.5684e+00, -1.9112e+00, -1.5059e+00, ..., -1.9519e+00,
-1.4662e+00, -1.0372e+00],
...,
[-2.4469e+00, -3.0462e+00, -2.7716e+00, ..., -3.2164e+00,
-2.5355e+00, -1.8873e+00],
[-1.5872e+00, -2.0824e+00, -1.9779e+00, ..., -2.2603e+00,
-1.8366e+00, -1.4461e+00],
[-1.0413e+00, -1.3430e+00, -1.3590e+00, ..., -1.4039e+00,
-1.1833e+00, -8.5643e-01]]],
[[[ 4.0317e-01, 8.7384e-01, 7.0805e-01, ..., 7.0286e-01,
9.3750e-01, 7.4890e-01],
[-1.1631e+00, -2.2221e-01, -1.3308e+00, ..., -9.7420e-01,
1.2021e-01, 6.0760e-01],
[ 7.3024e-01, 5.4249e-01, 6.1885e-01, ..., 7.3102e-01,
9.8561e-01, 9.0870e-01],
...,
[-2.2445e-01, -1.2193e+00, -2.7311e+00, ..., -1.4752e+00,
-1.4504e+00, -8.8214e-01],
[ 1.4136e+00, 7.9505e-01, -1.0919e+00, ..., -3.8338e-01,
3.9221e-03, 2.9127e-01],
[ 7.9655e-01, 1.8780e-01, -8.2902e-01, ..., -3.3237e-01,
1.2991e-01, 3.2610e-01]],
[[-2.3929e+00, -1.2193e+00, -4.3205e+00, ..., -5.3372e+00,
-4.5268e+00, -3.5940e+00],
[-2.4600e-01, 1.6824e+00, -2.6660e+00, ..., -4.2989e+00,
-4.0985e+00, -3.4068e+00],
[ 5.8112e-01, 1.4821e+00, -5.2303e+00, ..., -6.1991e+00,
-5.8918e+00, -5.4203e+00],
...,
[-1.3782e+00, 6.7647e-02, -7.0029e+00, ..., -7.8862e+00,
-7.2415e+00, -5.5377e+00],
[ 1.7165e+00, 2.7699e+00, -1.2379e+00, ..., -2.7674e+00,
-3.1831e+00, -2.4248e+00],
[ 1.1733e+00, 1.8113e+00, -2.3699e+00, ..., -3.3269e+00,
-3.7578e+00, -3.1507e+00]],
[[-6.9292e-01, -1.8363e+00, -1.6749e+00, ..., -9.1270e-01,
-6.7992e-01, -7.6106e-01],
[-1.4038e+00, -1.8906e+00, -1.2504e+00, ..., -7.0392e-01,
-4.0919e-01, -2.8911e-01],
[-1.1796e+00, -1.1217e+00, -4.9165e-01, ..., -3.4402e-01,
-2.3036e-02, 3.7356e-02],
...,
[-1.4775e+00, -1.7938e+00, -7.3667e-01, ..., -8.9876e-01,
-7.7063e-01, -1.0238e+00],
[-1.7958e+00, -1.1754e+00, -7.8565e-01, ..., -5.5517e-01,
-2.7358e-01, -3.4888e-01],
[-1.9623e+00, -1.3832e+00, -9.5876e-01, ..., -1.0673e+00,
-9.2943e-01, -8.9935e-01]],
...,
[[-2.1688e+00, -1.0992e+00, -2.2736e+00, ..., -2.8215e+00,
-2.1790e+00, -1.4679e+00],
[-6.0506e-01, 1.2272e+00, -4.6384e-01, ..., -1.3614e+00,
-1.3326e+00, -1.1235e+00],
[-3.0809e-01, 2.2971e+00, 8.0198e-01, ..., -1.8553e-01,
2.2788e-01, 3.6347e-01],
...,
[-1.8780e+00, 1.1902e+00, -1.6605e-02, ..., -1.1551e+00,
-6.2163e-01, -4.5110e-02],
[-7.9962e-01, 1.9424e+00, 1.7165e+00, ..., 5.4817e-01,
4.1639e-01, 7.0140e-01],
[-2.2441e+00, -1.1416e+00, -9.2394e-01, ..., -1.7875e+00,
-9.2424e-01, -2.5757e-01]],
[[-1.3896e+00, -1.3115e+00, -2.8971e+00, ..., -1.2565e+00,
-7.3459e-01, -5.6457e-01],
[-6.9262e-01, -2.1065e-01, -3.7423e+00, ..., -1.9551e+00,
-1.6712e+00, -1.7657e+00],
[-2.1006e+00, -1.4226e+00, -6.1019e+00, ..., -3.8729e+00,
-3.3140e+00, -2.6229e+00],
...,
[-2.5393e+00, -1.6856e+00, -5.5228e+00, ..., -3.1885e+00,
-2.7866e+00, -2.1925e+00],
[-2.2613e+00, -1.5537e+00, -4.3032e+00, ..., -2.7585e+00,
-2.3766e+00, -1.6837e+00],
[-2.3986e+00, -1.8340e+00, -3.3575e+00, ..., -3.0136e+00,
-2.4925e+00, -2.1656e+00]],
[[-1.1660e+00, -1.7229e+00, -1.5062e+00, ..., -1.6409e+00,
-1.3657e+00, -1.0177e+00],
[-1.1812e+00, -1.6591e+00, -1.1860e+00, ..., -1.3963e+00,
-1.1422e+00, -8.8745e-01],
[-1.5487e+00, -1.8439e+00, -1.3849e+00, ..., -1.6016e+00,
-1.2322e+00, -8.9293e-01],
...,
[-3.2863e+00, -4.1114e+00, -3.8944e+00, ..., -4.0900e+00,
-3.2834e+00, -2.5448e+00],
[-2.1820e+00, -2.8242e+00, -2.7876e+00, ..., -2.8592e+00,
-2.3585e+00, -1.9262e+00],
[-1.5946e+00, -1.9412e+00, -1.9933e+00, ..., -1.9438e+00,
-1.5666e+00, -1.2038e+00]]],
[[[ 4.9740e-01, 9.4705e-01, 7.2040e-01, ..., 7.7350e-01,
1.0439e+00, 8.3035e-01],
[-1.0635e+00, -1.5455e-01, -1.4210e+00, ..., -1.2198e+00,
5.7599e-02, 6.3618e-01],
[ 7.4182e-01, 5.4892e-01, 2.7330e-01, ..., 5.0175e-01,
8.8804e-01, 9.0056e-01],
...,
[-8.8901e-01, -1.8074e+00, -1.8323e+00, ..., -1.7999e+00,
-1.3543e+00, -9.8619e-01],
[ 7.2062e-01, 6.8064e-03, -1.0598e-01, ..., -1.6652e-02,
6.9555e-01, 7.8158e-01],
[ 1.2780e-01, -5.6071e-01, -4.5924e-01, ..., -1.5511e-01,
4.3777e-01, 3.4474e-01]],
[[-2.6426e+00, -1.5186e+00, -4.8499e+00, ..., -5.8195e+00,
-4.9732e+00, -3.8775e+00],
[-6.8101e-01, 1.2675e+00, -3.4019e+00, ..., -4.6761e+00,
-4.4996e+00, -3.6679e+00],
[-2.2540e-02, 1.0674e+00, -6.0908e+00, ..., -6.8045e+00,
-6.5016e+00, -5.9224e+00],
...,
[ 2.7466e-01, 3.4377e-01, -4.0744e+00, ..., -4.6624e+00,
-4.2166e+00, -3.4869e+00],
[ 2.2750e+00, 2.2550e+00, -2.0492e-01, ..., -7.9049e-01,
-1.1698e+00, -1.1895e+00],
[ 1.3325e+00, 1.1553e+00, -1.7997e+00, ..., -2.2261e+00,
-2.5039e+00, -2.3970e+00]],
[[-5.5585e-01, -1.5767e+00, -1.4115e+00, ..., -7.2828e-01,
-4.9860e-01, -7.2270e-01],
[-1.2682e+00, -1.6671e+00, -1.0713e+00, ..., -6.8545e-01,
-3.7139e-01, -3.1441e-01],
[-1.1256e+00, -1.0104e+00, -4.4393e-01, ..., -4.1034e-01,
-6.6288e-02, -7.1809e-02],
...,
[-1.5011e-01, -6.1932e-01, -4.7741e-01, ..., -5.0559e-01,
-2.0977e-01, -4.5460e-01],
[-7.0318e-01, -2.0419e-01, -4.2933e-01, ..., -2.0782e-01,
2.5382e-01, 2.4392e-01],
[-8.0355e-01, -1.9204e-01, -3.3436e-01, ..., -3.9469e-01,
-1.9336e-01, -2.2683e-01]],
...,
[[-2.1235e+00, -1.2492e+00, -2.5459e+00, ..., -3.2638e+00,
-2.7022e+00, -1.8361e+00],
[-5.2717e-01, 1.0822e+00, -6.9403e-01, ..., -1.6155e+00,
-1.8008e+00, -1.5691e+00],
[-3.6967e-01, 1.9932e+00, 4.0128e-01, ..., -3.5723e-01,
-5.7236e-02, 1.0996e-01],
...,
[-2.0314e+00, -8.5042e-02, -1.8273e-01, ..., -1.3689e+00,
-3.9850e-01, 1.0466e-01],
[-1.1671e+00, 7.9782e-01, 9.3174e-01, ..., 1.2252e-02,
2.9858e-01, 4.6696e-01],
[-2.0656e+00, -1.4079e+00, -1.6365e+00, ..., -2.4660e+00,
-1.4785e+00, -9.4542e-01]],
[[-1.4470e+00, -1.4188e+00, -2.7361e+00, ..., -1.1045e+00,
-5.8006e-01, -4.7171e-01],
[-8.6499e-01, -5.0093e-01, -3.6853e+00, ..., -1.8897e+00,
-1.6308e+00, -1.8314e+00],
[-2.4521e+00, -1.9929e+00, -6.2961e+00, ..., -4.1750e+00,
-3.6195e+00, -2.8513e+00],
...,
[-1.3721e+00, -4.4744e-02, -3.4001e+00, ..., -1.4336e+00,
-1.0051e+00, -9.1983e-01],
[-1.6846e+00, -6.5651e-01, -3.0623e+00, ..., -1.7254e+00,
-1.1415e+00, -7.9570e-01],
[-2.2977e+00, -1.4991e+00, -3.1129e+00, ..., -2.9565e+00,
-2.1256e+00, -1.9796e+00]],
[[-1.2244e+00, -1.8044e+00, -1.6513e+00, ..., -1.8436e+00,
-1.5305e+00, -1.1393e+00],
[-1.2823e+00, -1.7899e+00, -1.4207e+00, ..., -1.6187e+00,
-1.3125e+00, -1.0276e+00],
[-1.6851e+00, -2.0334e+00, -1.6847e+00, ..., -1.9094e+00,
-1.4658e+00, -1.0735e+00],
...,
[-2.3941e+00, -2.9562e+00, -2.7418e+00, ..., -2.8588e+00,
-2.2884e+00, -1.7569e+00],
[-1.5527e+00, -2.0101e+00, -1.9403e+00, ..., -1.9380e+00,
-1.5973e+00, -1.2900e+00],
[-1.0128e+00, -1.3004e+00, -1.3323e+00, ..., -1.1843e+00,
-1.0098e+00, -7.6519e-01]]]], device='cuda:0'),
tensor([[[[ 6.9734e-02, 2.2009e-01, -1.4239e+00, ..., -6.5320e-01,
-1.9651e-01, -6.0725e-02],
[-2.1200e+00, -1.6358e+00, -4.5675e+00, ..., -3.8561e+00,
-1.5477e+00, -5.6867e-01],
[ 5.7658e-01, -7.3155e-02, -8.4883e-01, ..., -2.6445e-01,
-2.4443e-01, 2.3463e-01],
...,
[-8.6788e-01, -1.8846e+00, -1.8887e+00, ..., -1.7753e+00,
-1.3481e+00, -9.6734e-01],
[ 7.5108e-01, -3.5405e-02, -1.4467e-01, ..., 1.1303e-02,
6.9582e-01, 7.9845e-01],
[ 1.3591e-01, -5.9555e-01, -5.0073e-01, ..., -1.5098e-01,
4.2921e-01, 3.5482e-01]],
[[-3.3210e+00, -1.8265e+00, -6.2826e+00, ..., -7.1532e+00,
-6.2808e+00, -4.5999e+00],
[-8.0683e-01, 1.5383e+00, -4.3595e+00, ..., -5.5037e+00,
-5.5769e+00, -4.2680e+00],
[-3.6322e-01, 1.0909e+00, -7.4446e+00, ..., -8.0249e+00,
-7.8799e+00, -6.6844e+00],
...,
[ 2.9724e-01, 3.6023e-01, -4.0534e+00, ..., -4.6707e+00,
-4.1917e+00, -3.5051e+00],
[ 2.2919e+00, 2.2601e+00, -2.0013e-01, ..., -8.4556e-01,
-1.1795e+00, -1.2357e+00],
[ 1.3444e+00, 1.1588e+00, -1.8017e+00, ..., -2.2453e+00,
-2.5117e+00, -2.4153e+00]],
[[-1.3633e+00, -2.3798e+00, -2.3385e+00, ..., -1.7565e+00,
-1.2387e+00, -1.3990e+00],
[-1.7098e+00, -2.0096e+00, -1.5464e+00, ..., -1.2501e+00,
-7.1438e-01, -8.1647e-01],
[-2.1778e+00, -1.9397e+00, -1.0135e+00, ..., -1.1117e+00,
-4.5251e-01, -1.8696e-01],
...,
[-1.5084e-01, -6.0570e-01, -4.7381e-01, ..., -5.0003e-01,
-1.9718e-01, -4.4331e-01],
[-7.0743e-01, -2.0003e-01, -4.3269e-01, ..., -1.9512e-01,
2.4325e-01, 2.3654e-01],
[-7.9051e-01, -1.8448e-01, -3.3965e-01, ..., -3.8207e-01,
-1.9037e-01, -2.2766e-01]],
...,
[[-2.0460e+00, -1.1146e+00, -2.3624e+00, ..., -3.2043e+00,
-2.6835e+00, -1.8323e+00],
[-1.7591e+00, 2.7709e-02, -1.3655e+00, ..., -2.4776e+00,
-2.5876e+00, -2.0032e+00],
[-3.8186e-01, 2.3662e+00, 1.0691e+00, ..., 1.8401e-01,
2.4931e-01, 5.2214e-01],
...,
[-2.0493e+00, -3.1720e-02, -1.3400e-01, ..., -1.3604e+00,
-3.8548e-01, 9.4389e-02],
[-1.1942e+00, 8.5093e-01, 9.8766e-01, ..., -5.1236e-03,
3.2108e-01, 4.6484e-01],
[-2.0915e+00, -1.3838e+00, -1.5772e+00, ..., -2.4684e+00,
-1.4559e+00, -9.2207e-01]],
[[-2.4283e+00, -2.8314e+00, -4.2025e+00, ..., -2.3753e+00,
-1.9089e+00, -1.7437e+00],
[-1.7726e+00, -1.6360e+00, -4.8825e+00, ..., -2.9030e+00,
-2.7889e+00, -2.8055e+00],
[-2.8506e+00, -2.4835e+00, -6.6884e+00, ..., -4.4363e+00,
-4.3649e+00, -3.5099e+00],
...,
[-1.4116e+00, -3.9449e-02, -3.3695e+00, ..., -1.4638e+00,
-1.0261e+00, -9.3308e-01],
[-1.7061e+00, -6.3352e-01, -3.0094e+00, ..., -1.7343e+00,
-1.1758e+00, -8.1486e-01],
[-2.3149e+00, -1.4677e+00, -3.0796e+00, ..., -2.9540e+00,
-2.1563e+00, -1.9924e+00]],
[[-1.8124e+00, -2.5457e+00, -2.4655e+00, ..., -2.7113e+00,
-2.2192e+00, -1.7056e+00],
[-1.8047e+00, -2.5152e+00, -2.1507e+00, ..., -2.4237e+00,
-1.9069e+00, -1.5554e+00],
[-2.2564e+00, -2.8174e+00, -2.5175e+00, ..., -2.7721e+00,
-2.1638e+00, -1.6509e+00],
...,
[-2.3912e+00, -2.9657e+00, -2.7430e+00, ..., -2.8628e+00,
-2.2858e+00, -1.7529e+00],
[-1.5558e+00, -2.0276e+00, -1.9608e+00, ..., -1.9461e+00,
-1.6012e+00, -1.2917e+00],
[-1.0153e+00, -1.3164e+00, -1.3528e+00, ..., -1.1872e+00,
-1.0147e+00, -7.6858e-01]]],
[[[ 3.3556e-01, 7.9841e-01, 7.4128e-01, ..., 6.6454e-01,
9.6474e-01, 7.2141e-01],
[-1.2050e+00, -2.8574e-01, -1.3254e+00, ..., -1.3205e+00,
-3.5214e-02, 5.0940e-01],
[ 6.2087e-01, 4.2652e-01, 5.6371e-01, ..., 5.1603e-01,
9.3339e-01, 8.4545e-01],
...,
[-1.4929e-01, -1.1468e+00, -2.6274e+00, ..., -1.5324e+00,
-1.3818e+00, -8.3362e-01],
[ 1.4289e+00, 8.2834e-01, -9.4561e-01, ..., -3.0967e-01,
1.0905e-01, 3.8842e-01],
[ 8.5654e-01, 2.9138e-01, -6.5492e-01, ..., -1.8232e-01,
2.3747e-01, 3.9498e-01]],
[[-2.3348e+00, -1.3763e+00, -4.3083e+00, ..., -5.4155e+00,
-4.6569e+00, -3.6016e+00],
[-2.3788e-01, 1.4261e+00, -2.6898e+00, ..., -4.0573e+00,
-3.9899e+00, -3.2513e+00],
[ 5.4925e-01, 1.2130e+00, -5.1967e+00, ..., -5.9844e+00,
-5.8132e+00, -5.3775e+00],
...,
[-1.3451e+00, 1.6897e-02, -6.7200e+00, ..., -7.5856e+00,
-6.9060e+00, -5.4393e+00],
[ 1.7038e+00, 2.7013e+00, -1.0628e+00, ..., -2.5935e+00,
-2.9638e+00, -2.4082e+00],
[ 1.1448e+00, 1.7338e+00, -2.2856e+00, ..., -3.2718e+00,
-3.6366e+00, -3.1581e+00]],
[[-6.6282e-01, -1.7446e+00, -1.6654e+00, ..., -9.9955e-01,
-6.8974e-01, -8.2826e-01],
[-1.3020e+00, -1.7211e+00, -1.2192e+00, ..., -8.0165e-01,
-4.1115e-01, -2.8792e-01],
[-9.6469e-01, -8.7708e-01, -4.1298e-01, ..., -3.4848e-01,
5.2696e-02, 7.0076e-02],
...,
[-1.3366e+00, -1.7200e+00, -7.5157e-01, ..., -8.7512e-01,
-7.7158e-01, -1.0338e+00],
[-1.6417e+00, -1.0940e+00, -7.9713e-01, ..., -5.8002e-01,
-3.0310e-01, -3.7122e-01],
[-1.7852e+00, -1.2273e+00, -8.6566e-01, ..., -9.5673e-01,
-8.1946e-01, -7.9989e-01]],
...,
[[-2.1673e+00, -1.3199e+00, -2.2625e+00, ..., -3.1540e+00,
-2.5053e+00, -1.6307e+00],
[-6.2428e-01, 9.3652e-01, -4.1814e-01, ..., -1.4232e+00,
-1.5350e+00, -1.3474e+00],
[-4.3520e-01, 1.8135e+00, 6.8727e-01, ..., -1.1657e-01,
2.7456e-01, 3.9754e-01],
...,
[-1.8423e+00, 1.1772e+00, -2.3806e-02, ..., -1.0916e+00,
-5.5196e-01, 5.2027e-03],
[-8.3220e-01, 1.8264e+00, 1.5720e+00, ..., 5.1242e-01,
4.1555e-01, 6.7085e-01],
[-2.2965e+00, -1.2186e+00, -1.0666e+00, ..., -1.8327e+00,
-9.5459e-01, -2.9112e-01]],
[[-1.3711e+00, -1.2847e+00, -2.5999e+00, ..., -9.7960e-01,
-4.5680e-01, -3.7532e-01],
[-7.0508e-01, -2.4250e-01, -3.3873e+00, ..., -1.6365e+00,
-1.4074e+00, -1.6669e+00],
[-2.1136e+00, -1.4667e+00, -5.6957e+00, ..., -3.7172e+00,
-3.2225e+00, -2.5664e+00],
...,
[-2.4875e+00, -1.6025e+00, -5.3413e+00, ..., -3.0470e+00,
-2.7238e+00, -2.1984e+00],
[-2.2382e+00, -1.4859e+00, -4.1796e+00, ..., -2.6564e+00,
-2.3101e+00, -1.6510e+00],
[-2.3902e+00, -1.7963e+00, -3.2989e+00, ..., -2.9475e+00,
-2.4324e+00, -2.1141e+00]],
[[-1.1400e+00, -1.6866e+00, -1.5089e+00, ..., -1.7046e+00,
-1.4117e+00, -1.0620e+00],
[-1.1544e+00, -1.6110e+00, -1.1987e+00, ..., -1.3731e+00,
-1.1126e+00, -8.8547e-01],
[-1.4885e+00, -1.7717e+00, -1.3848e+00, ..., -1.5408e+00,
-1.1664e+00, -8.5224e-01],
...,
[-3.1414e+00, -3.9326e+00, -3.7630e+00, ..., -3.9570e+00,
-3.1660e+00, -2.4393e+00],
[-2.0802e+00, -2.6965e+00, -2.6838e+00, ..., -2.7779e+00,
-2.2878e+00, -1.8584e+00],
[-1.5095e+00, -1.8363e+00, -1.9182e+00, ..., -1.8742e+00,
-1.5206e+00, -1.1610e+00]]],
[[[ 8.3196e-02, 2.3184e-01, -1.4876e+00, ..., -6.8300e-01,
-2.5955e-01, -6.2331e-02],
[-2.2209e+00, -1.7154e+00, -4.8205e+00, ..., -3.7282e+00,
-1.5108e+00, -5.3903e-01],
[ 5.9050e-01, -1.8233e-01, -9.7730e-01, ..., -2.1360e-01,
-2.7882e-01, 2.6995e-01],
...,
[-8.3465e-01, -1.7651e+00, -1.7829e+00, ..., -1.7529e+00,
-1.2903e+00, -9.1844e-01],
[ 7.5906e-01, 6.3126e-02, -4.8861e-02, ..., 4.1242e-02,
7.1532e-01, 8.1310e-01],
[ 1.7096e-01, -4.7351e-01, -3.9818e-01, ..., -9.4834e-02,
4.6094e-01, 3.7873e-01]],
[[-3.3712e+00, -1.6360e+00, -6.3144e+00, ..., -7.0765e+00,
-6.2227e+00, -4.6104e+00],
[-6.2714e-01, 1.9905e+00, -4.2145e+00, ..., -5.5875e+00,
-5.6473e+00, -4.3443e+00],
[-6.9711e-02, 1.5748e+00, -7.4736e+00, ..., -8.0913e+00,
-7.9288e+00, -6.6629e+00],
...,
[ 1.6969e-01, 2.2833e-01, -4.0918e+00, ..., -4.6725e+00,
-4.2378e+00, -3.5664e+00],
[ 2.1962e+00, 2.1888e+00, -1.6894e-01, ..., -8.3443e-01,
-1.2178e+00, -1.2764e+00],
[ 1.2659e+00, 1.1242e+00, -1.7428e+00, ..., -2.2704e+00,
-2.5497e+00, -2.4318e+00]],
[[-1.4585e+00, -2.5933e+00, -2.5164e+00, ..., -1.7466e+00,
-1.2557e+00, -1.3700e+00],
[-1.9127e+00, -2.2362e+00, -1.6695e+00, ..., -1.2039e+00,
-7.2782e-01, -8.3834e-01],
[-2.4206e+00, -2.1275e+00, -1.0752e+00, ..., -1.1245e+00,
-4.9243e-01, -2.0705e-01],
...,
[-1.3198e-01, -5.9706e-01, -4.5654e-01, ..., -5.1811e-01,
-2.1984e-01, -4.4713e-01],
[-6.6496e-01, -1.9340e-01, -4.1948e-01, ..., -2.0289e-01,
2.3143e-01, 2.3579e-01],
[-7.7073e-01, -1.8829e-01, -3.3864e-01, ..., -3.8259e-01,
-1.8636e-01, -2.2323e-01]],
...,
[[-2.2516e+00, -1.1080e+00, -2.4817e+00, ..., -3.0150e+00,
-2.4802e+00, -1.7495e+00],
[-1.8915e+00, 1.8964e-01, -1.3717e+00, ..., -2.4967e+00,
-2.4897e+00, -1.8987e+00],
[-3.5195e-01, 2.7463e+00, 1.2918e+00, ..., 1.0274e-01,
1.6490e-01, 4.5268e-01],
...,
[-2.0655e+00, -1.2846e-01, -2.0050e-01, ..., -1.3060e+00,
-3.6436e-01, 1.1333e-01],
[-1.1896e+00, 7.4685e-01, 9.0302e-01, ..., 2.0162e-02,
3.2051e-01, 4.8050e-01],
[-2.0927e+00, -1.4740e+00, -1.6609e+00, ..., -2.4185e+00,
-1.4428e+00, -9.0366e-01]],
[[-2.4533e+00, -2.8577e+00, -4.4466e+00, ..., -2.5490e+00,
-2.0814e+00, -1.8710e+00],
[-1.7056e+00, -1.5006e+00, -5.1458e+00, ..., -3.0811e+00,
-2.9556e+00, -2.8729e+00],
[-2.8315e+00, -2.3239e+00, -7.0400e+00, ..., -4.4802e+00,
-4.4308e+00, -3.5483e+00],
...,
[-1.4253e+00, -1.2359e-01, -3.3609e+00, ..., -1.4548e+00,
-1.0765e+00, -9.6885e-01],
[-1.7250e+00, -7.1426e-01, -3.0328e+00, ..., -1.7454e+00,
-1.1953e+00, -8.3316e-01],
[-2.3303e+00, -1.5415e+00, -3.0906e+00, ..., -2.9557e+00,
-2.1482e+00, -1.9640e+00]],
[[-1.9046e+00, -2.6765e+00, -2.5241e+00, ..., -2.6977e+00,
-2.2144e+00, -1.7066e+00],
[-1.8923e+00, -2.6412e+00, -2.1764e+00, ..., -2.4585e+00,
-1.9411e+00, -1.5766e+00],
[-2.3762e+00, -2.9551e+00, -2.5566e+00, ..., -2.8273e+00,
-2.2135e+00, -1.6950e+00],
...,
[-2.3485e+00, -2.9119e+00, -2.7147e+00, ..., -2.8559e+00,
-2.2889e+00, -1.7476e+00],
[-1.5135e+00, -1.9729e+00, -1.9184e+00, ..., -1.9409e+00,
-1.6014e+00, -1.2885e+00],
[-9.8172e-01, -1.2693e+00, -1.3133e+00, ..., -1.1894e+00,
-1.0129e+00, -7.6701e-01]]],
[[[ 2.9374e-01, 5.6100e-01, 7.0376e-01, ..., 4.9707e-01,
1.0231e+00, 6.1754e-01],
[-2.1809e+00, -1.5191e+00, -3.2136e+00, ..., -2.5972e+00,
-8.3180e-01, -1.7278e-01],
[-1.7333e-01, -1.0242e+00, -2.3662e-01, ..., -1.3487e-01,
5.6416e-01, 6.7473e-01],
...,
[-1.6418e-01, -9.8076e-01, -2.4028e+00, ..., -1.6991e+00,
-1.6099e+00, -1.0457e+00],
[ 1.2979e+00, 8.3456e-01, -8.9782e-01, ..., -2.7160e-01,
8.4158e-02, 3.5433e-01],
[ 7.5584e-01, 2.7353e-01, -6.4390e-01, ..., -1.4599e-01,
2.3297e-01, 3.7316e-01]],
[[-1.6676e+00, -1.1199e+00, -4.1981e+00, ..., -4.8410e+00,
-4.4227e+00, -3.3887e+00],
[ 6.1155e-01, 1.9932e+00, -2.2537e+00, ..., -3.5375e+00,
-3.6447e+00, -2.9935e+00],
[ 1.3694e+00, 1.7521e+00, -4.9169e+00, ..., -5.4634e+00,
-5.2865e+00, -4.9185e+00],
...,
[-1.5185e+00, -2.6640e-01, -6.8817e+00, ..., -7.5848e+00,
-6.8508e+00, -5.3486e+00],
[ 1.3466e+00, 2.2642e+00, -1.5705e+00, ..., -2.3071e+00,
-2.6820e+00, -2.1513e+00],
[ 9.0419e-01, 1.4662e+00, -2.5095e+00, ..., -3.1448e+00,
-3.5565e+00, -3.1022e+00]],
[[-8.8079e-02, -1.1455e+00, -1.3219e+00, ..., -4.7140e-01,
-1.4382e-01, -4.5471e-01],
[-5.6775e-01, -8.6356e-01, -7.8916e-01, ..., -2.6121e-01,
3.6107e-02, -6.9147e-02],
[-5.3717e-01, -3.6728e-01, -4.1378e-01, ..., -3.5376e-01,
7.1305e-02, 7.9197e-02],
...,
[-1.1864e+00, -1.5669e+00, -7.2110e-01, ..., -9.2091e-01,
-7.9057e-01, -1.0553e+00],
[-1.5089e+00, -1.0243e+00, -7.8248e-01, ..., -5.7463e-01,
-2.7765e-01, -3.4249e-01],
[-1.6614e+00, -1.1680e+00, -8.6026e-01, ..., -9.4379e-01,
-8.0317e-01, -7.8731e-01]],
...,
[[-3.0592e+00, -2.5194e+00, -2.8324e+00, ..., -3.4401e+00,
-2.6048e+00, -1.7446e+00],
[-2.3276e+00, -8.3182e-01, -1.6136e+00, ..., -2.6567e+00,
-2.3316e+00, -1.8143e+00],
[-1.1297e+00, 1.2030e+00, 6.5218e-01, ..., -4.8967e-01,
-5.4013e-02, 9.9803e-02],
...,
[-1.7105e+00, 9.7390e-01, -1.0476e-01, ..., -1.1977e+00,
-4.9843e-01, 7.3168e-02],
[-7.7903e-01, 1.6278e+00, 1.4269e+00, ..., 4.5316e-01,
4.6117e-01, 7.3990e-01],
[-2.1212e+00, -1.1520e+00, -9.6738e-01, ..., -2.0973e+00,
-1.1042e+00, -3.7117e-01]],
[[-1.6057e+00, -1.5771e+00, -2.9079e+00, ..., -1.3245e+00,
-8.8089e-01, -8.7260e-01],
[-5.9278e-01, 1.0394e-01, -3.2297e+00, ..., -1.4061e+00,
-1.2385e+00, -1.5809e+00],
[-2.0851e+00, -8.7452e-01, -5.3717e+00, ..., -3.1835e+00,
-2.6971e+00, -2.1995e+00],
...,
[-2.5234e+00, -1.8431e+00, -5.3434e+00, ..., -2.9790e+00,
-2.5955e+00, -2.0732e+00],
[-2.2361e+00, -1.6542e+00, -4.1802e+00, ..., -2.6333e+00,
-2.1989e+00, -1.5306e+00],
[-2.3182e+00, -1.8457e+00, -3.2777e+00, ..., -3.0340e+00,
-2.4217e+00, -2.1294e+00]],
[[-1.3714e+00, -1.9342e+00, -1.7286e+00, ..., -1.7545e+00,
-1.4533e+00, -1.1065e+00],
[-1.2914e+00, -1.7838e+00, -1.2441e+00, ..., -1.3780e+00,
-1.0851e+00, -8.6152e-01],
[-1.6146e+00, -1.9665e+00, -1.4891e+00, ..., -1.6055e+00,
-1.2037e+00, -8.9155e-01],
...,
[-3.0397e+00, -3.8026e+00, -3.6784e+00, ..., -4.0139e+00,
-3.2317e+00, -2.4872e+00],
[-2.0318e+00, -2.6229e+00, -2.6477e+00, ..., -2.8030e+00,
-2.3272e+00, -1.8963e+00],
[-1.4933e+00, -1.8113e+00, -1.9048e+00, ..., -1.8697e+00,
-1.5335e+00, -1.1730e+00]]],
[[[ 3.0046e-01, 5.6711e-01, 7.2777e-01, ..., 5.0094e-01,
1.0100e+00, 6.0129e-01],
[-2.1162e+00, -1.4484e+00, -3.0447e+00, ..., -2.5674e+00,
-8.1323e-01, -1.6742e-01],
[-1.3355e-01, -9.4082e-01, -2.2369e-01, ..., -9.6589e-02,
5.7012e-01, 6.6091e-01],
...,
[ 1.3402e-01, -6.7679e-01, -1.6155e+00, ..., -8.0673e+00,
-2.8684e+00, -5.5714e-01],
[ 1.5647e+00, 1.2436e+00, 5.7333e-01, ..., -7.7681e+00,
-2.7498e+00, 9.1665e-01],
[ 1.1761e+00, 1.0101e+00, 6.8228e-01, ..., -5.4368e+00,
-1.9314e+00, 8.4182e-01]],
[[-1.7010e+00, -1.1183e+00, -4.1203e+00, ..., -4.8842e+00,
-4.4272e+00, -3.4056e+00],
[ 5.5983e-01, 1.9540e+00, -2.2539e+00, ..., -3.5280e+00,
-3.6133e+00, -2.9845e+00],
[ 1.3106e+00, 1.7161e+00, -4.8971e+00, ..., -5.4356e+00,
-5.2640e+00, -4.9480e+00],
...,
[-6.5737e-01, 5.5464e-01, -4.8257e+00, ..., -7.5764e+00,
-6.3666e+00, -4.3013e+00],
[ 2.1355e+00, 2.8834e+00, 8.3904e-02, ..., -3.0669e+00,
-2.4598e+00, -1.7366e+00],
[ 1.2612e+00, 1.5666e+00, -1.7057e+00, ..., -3.3848e+00,
-3.1535e+00, -2.7334e+00]],
[[-1.2957e-01, -1.1544e+00, -1.2999e+00, ..., -4.8108e-01,
-1.7061e-01, -4.9349e-01],
[-6.2789e-01, -8.8658e-01, -7.5768e-01, ..., -2.9806e-01,
4.4826e-03, -9.3492e-02],
[-5.9246e-01, -3.9514e-01, -4.0165e-01, ..., -3.4784e-01,
7.8427e-02, 6.9752e-02],
...,
[-1.1350e+00, -1.5837e+00, -6.5584e-01, ..., -4.9954e-01,
-3.5323e-01, -6.3765e-01],
[-1.3499e+00, -8.8094e-01, -4.6035e-01, ..., -1.1148e-01,
1.0889e-01, 6.6965e-02],
[-1.3608e+00, -6.7675e-01, -1.3781e-01, ..., 1.0329e+00,
2.4242e-01, -5.4275e-02]],
...,
[[-3.0067e+00, -2.4157e+00, -2.7524e+00, ..., -3.4101e+00,
-2.6134e+00, -1.7519e+00],
[-2.2123e+00, -7.2616e-01, -1.5480e+00, ..., -2.5479e+00,
-2.3013e+00, -1.8285e+00],
[-1.0070e+00, 1.2709e+00, 6.6340e-01, ..., -4.3810e-01,
-4.7404e-03, 1.3031e-01],
...,
[-1.5861e+00, 1.6475e+00, 3.5939e-01, ..., -5.2213e+00,
-1.1963e+00, 4.1603e-01],
[-6.3237e-01, 1.7924e+00, 1.1423e+00, ..., -1.0565e+00,
5.5805e-02, 5.1620e-01],
[-1.7552e+00, -1.0392e+00, -1.4142e+00, ..., -1.3599e+00,
-9.2273e-01, -5.3505e-01]],
[[-1.5682e+00, -1.5424e+00, -2.8985e+00, ..., -1.2694e+00,
-8.2682e-01, -8.0933e-01],
[-5.6861e-01, 9.0013e-02, -3.2749e+00, ..., -1.3636e+00,
-1.2053e+00, -1.5567e+00],
[-2.0626e+00, -8.9533e-01, -5.3748e+00, ..., -3.1819e+00,
-2.7048e+00, -2.2062e+00],
...,
[-1.8066e+00, -7.1934e-01, -4.6956e+00, ..., -1.7486e+00,
-1.8162e+00, -1.6480e+00],
[-1.7037e+00, -8.1492e-01, -3.7487e+00, ..., -3.3539e+00,
-1.7415e+00, -1.1197e+00],
[-2.0881e+00, -1.3467e+00, -3.0208e+00, ..., -4.1664e+00,
-2.1148e+00, -1.6773e+00]],
[[-1.3530e+00, -1.9061e+00, -1.6980e+00, ..., -1.7604e+00,
-1.4624e+00, -1.1107e+00],
[-1.2788e+00, -1.7702e+00, -1.2427e+00, ..., -1.3762e+00,
-1.0867e+00, -8.6562e-01],
[-1.6087e+00, -1.9629e+00, -1.4987e+00, ..., -1.6052e+00,
-1.2008e+00, -8.8310e-01],
...,
[-2.5216e+00, -3.1471e+00, -2.8798e+00, ..., -3.4451e+00,
-2.5240e+00, -1.8470e+00],
[-1.6852e+00, -2.1466e+00, -2.0446e+00, ..., -2.7913e+00,
-1.9030e+00, -1.3871e+00],
[-1.1240e+00, -1.3547e+00, -1.3887e+00, ..., -1.9299e+00,
-1.2441e+00, -8.2304e-01]]]], device='cuda:0'),
tensor([[[[ 5.8901e-02, 1.8526e-01, -1.4628e+00, ..., -6.4360e-01,
-2.1600e-01, -5.6105e-02],
[-2.0614e+00, -1.6483e+00, -4.5814e+00, ..., -3.8353e+00,
-1.5674e+00, -5.7041e-01],
[ 5.9997e-01, -1.1875e-01, -8.7755e-01, ..., -2.0826e-01,
-2.8239e-01, 2.3546e-01],
...,
[-8.9636e-01, -2.0811e+00, -2.2901e+00, ..., -1.5871e+00,
-1.2036e+00, -8.2124e-01],
[ 8.8810e-01, 1.4979e-01, 4.8310e-03, ..., -7.9369e-03,
7.1827e-01, 8.1842e-01],
[ 2.0407e-01, -4.8270e-01, -4.0906e-01, ..., -1.5864e-01,
4.2853e-01, 3.4256e-01]],
[[-3.3074e+00, -1.8133e+00, -6.2937e+00, ..., -7.1979e+00,
-6.3034e+00, -4.6532e+00],
[-8.0854e-01, 1.5060e+00, -4.4157e+00, ..., -5.5043e+00,
-5.5816e+00, -4.3084e+00],
[-3.8456e-01, 1.0422e+00, -7.4938e+00, ..., -8.0296e+00,
-7.8997e+00, -6.7259e+00],
...,
[ 4.0112e-01, 6.3775e-01, -3.8824e+00, ..., -4.9980e+00,
-4.2752e+00, -3.5673e+00],
[ 2.7123e+00, 2.7524e+00, 4.2081e-01, ..., -1.1144e+00,
-1.3192e+00, -1.3464e+00],
[ 1.6802e+00, 1.5073e+00, -1.5519e+00, ..., -2.3521e+00,
-2.5537e+00, -2.4326e+00]],
[[-1.3682e+00, -2.3692e+00, -2.3474e+00, ..., -1.7784e+00,
-1.2555e+00, -1.3927e+00],
[-1.6880e+00, -1.9871e+00, -1.5596e+00, ..., -1.2656e+00,
-7.3783e-01, -8.2672e-01],
[-2.1804e+00, -1.9195e+00, -1.0047e+00, ..., -1.1138e+00,
-4.6323e-01, -1.8519e-01],
...,
[-2.9741e-01, -8.3794e-01, -5.4104e-01, ..., -4.3350e-01,
-1.6837e-01, -4.2158e-01],
[-8.8884e-01, -3.2674e-01, -5.0123e-01, ..., -1.4803e-01,
2.4536e-01, 2.3498e-01],
[-1.0336e+00, -3.1997e-01, -3.6674e-01, ..., -3.8152e-01,
-1.9600e-01, -2.3279e-01]],
...,
[[-2.0266e+00, -1.0436e+00, -2.2982e+00, ..., -3.2428e+00,
-2.6599e+00, -1.8493e+00],
[-1.8118e+00, 3.9475e-03, -1.3571e+00, ..., -2.5565e+00,
-2.5757e+00, -1.9993e+00],
[-4.5147e-01, 2.2774e+00, 1.0563e+00, ..., 1.0170e-01,
2.4046e-01, 5.1521e-01],
...,
[-2.2163e+00, 1.8489e-01, -2.7816e-02, ..., -1.3018e+00,
-3.8478e-01, 8.9293e-02],
[-1.3123e+00, 9.3831e-01, 1.0735e+00, ..., -1.5113e-02,
2.9181e-01, 4.6103e-01],
[-2.2321e+00, -1.4617e+00, -1.6558e+00, ..., -2.3793e+00,
-1.4361e+00, -8.9388e-01]],
[[-2.4621e+00, -2.8960e+00, -4.2501e+00, ..., -2.4113e+00,
-1.9203e+00, -1.7322e+00],
[-1.8097e+00, -1.7265e+00, -4.9131e+00, ..., -2.9275e+00,
-2.8113e+00, -2.8040e+00],
[-2.8455e+00, -2.5553e+00, -6.6794e+00, ..., -4.4646e+00,
-4.4013e+00, -3.5218e+00],
...,
[-1.3728e+00, 9.6525e-02, -3.7059e+00, ..., -1.5767e+00,
-1.1030e+00, -1.0057e+00],
[-1.6380e+00, -4.5013e-01, -3.2122e+00, ..., -1.7748e+00,
-1.1937e+00, -8.7621e-01],
[-2.2835e+00, -1.3359e+00, -3.1841e+00, ..., -2.9300e+00,
-2.1178e+00, -1.9851e+00]],
[[-1.8422e+00, -2.5719e+00, -2.4974e+00, ..., -2.7300e+00,
-2.2407e+00, -1.7151e+00],
[-1.8363e+00, -2.5613e+00, -2.2137e+00, ..., -2.4286e+00,
-1.9221e+00, -1.5558e+00],
[-2.2922e+00, -2.8688e+00, -2.5957e+00, ..., -2.7703e+00,
-2.1627e+00, -1.6453e+00],
...,
[-2.5132e+00, -3.1215e+00, -2.8343e+00, ..., -2.8960e+00,
-2.2772e+00, -1.7318e+00],
[-1.6496e+00, -2.1589e+00, -2.0406e+00, ..., -1.9823e+00,
-1.6048e+00, -1.2841e+00],
[-1.0807e+00, -1.4030e+00, -1.4128e+00, ..., -1.2097e+00,
-1.0175e+00, -7.7163e-01]]],
[[[ 7.8282e-02, 3.1319e-01, -1.2136e+00, ..., 4.6715e-01,
1.0147e+00, 6.4029e-01],
[-2.0287e+00, -1.3926e+00, -4.1724e+00, ..., -2.5878e+00,
-7.6385e-01, -7.7527e-02],
[ 3.6646e-01, -3.0028e-01, -1.0528e+00, ..., -2.6737e-01,
4.9311e-01, 6.3885e-01],
...,
[-2.9530e-01, -1.4294e+00, -2.8082e+00, ..., -2.0045e+00,
-1.3916e+00, -9.1395e-01],
[ 1.3591e+00, 6.6649e-01, -9.2646e-01, ..., -2.9639e-01,
5.5306e-01, 7.2268e-01],
[ 9.1294e-01, 3.2876e-01, -4.2510e-01, ..., -3.5587e-01,
2.9719e-01, 2.8339e-01]],
[[-3.1797e+00, -1.5555e+00, -6.1139e+00, ..., -4.8643e+00,
-4.4575e+00, -3.3768e+00],
[-7.3430e-01, 1.7076e+00, -4.4150e+00, ..., -3.3742e+00,
-3.5635e+00, -2.8908e+00],
[-2.5802e-01, 1.3259e+00, -7.5028e+00, ..., -5.3619e+00,
-5.2681e+00, -4.9211e+00],
...,
[-1.0526e+00, 6.6894e-01, -6.3081e+00, ..., -4.5686e+00,
-4.1512e+00, -3.4920e+00],
[ 1.9407e+00, 3.1296e+00, -7.9770e-01, ..., -9.2829e-01,
-1.2742e+00, -1.3216e+00],
[ 1.2854e+00, 1.9449e+00, -2.1952e+00, ..., -2.2735e+00,
-2.5132e+00, -2.3866e+00]],
[[-1.3001e+00, -2.1658e+00, -2.0334e+00, ..., -6.1656e-01,
-2.0503e-01, -4.6157e-01],
[-1.8334e+00, -1.9767e+00, -1.4583e+00, ..., -3.5845e-01,
6.8065e-02, 4.3390e-02],
[-2.3564e+00, -1.8421e+00, -7.9889e-01, ..., -2.8492e-01,
2.5253e-01, 2.9483e-01],
...,
[-1.7265e+00, -1.9888e+00, -8.1559e-01, ..., -4.8248e-01,
-1.6212e-01, -3.3586e-01],
[-1.9449e+00, -1.3204e+00, -8.4099e-01, ..., -3.1810e-01,
1.3517e-01, 2.0132e-01],
[-1.9284e+00, -1.2396e+00, -6.4201e-01, ..., -5.0219e-01,
-2.7815e-01, -2.5198e-01]],
...,
[[-2.1080e+00, -1.2033e+00, -2.5357e+00, ..., -3.7023e+00,
-2.8536e+00, -1.8520e+00],
[-1.7890e+00, 7.7305e-02, -1.2867e+00, ..., -2.6578e+00,
-2.4340e+00, -1.8831e+00],
[-4.8001e-01, 2.2063e+00, 8.3799e-01, ..., -5.7798e-01,
-6.2024e-02, 1.6848e-01],
...,
[-1.9040e+00, 1.3681e+00, 2.8595e-01, ..., -1.4884e+00,
-6.7249e-01, -1.5175e-01],
[-8.9938e-01, 1.8841e+00, 1.6376e+00, ..., -1.4054e-01,
1.1954e-01, 3.1859e-01],
[-2.1767e+00, -1.0438e+00, -9.2170e-01, ..., -2.2917e+00,
-1.3920e+00, -8.8537e-01]],
[[-2.3789e+00, -2.6768e+00, -3.8801e+00, ..., -1.1341e+00,
-7.1897e-01, -7.3955e-01],
[-1.7710e+00, -1.5548e+00, -4.6050e+00, ..., -1.3572e+00,
-1.2100e+00, -1.5905e+00],
[-3.0042e+00, -2.5986e+00, -6.5745e+00, ..., -3.2984e+00,
-2.8073e+00, -2.2387e+00],
...,
[-2.4315e+00, -1.4757e+00, -5.0979e+00, ..., -1.5150e+00,
-1.0916e+00, -9.2657e-01],
[-2.1573e+00, -1.3573e+00, -3.9745e+00, ..., -1.7478e+00,
-1.1971e+00, -7.9989e-01],
[-2.3383e+00, -1.6832e+00, -3.1203e+00, ..., -2.8024e+00,
-2.0436e+00, -1.8520e+00]],
[[-1.8499e+00, -2.5402e+00, -2.3816e+00, ..., -1.8066e+00,
-1.5013e+00, -1.1436e+00],
[-1.8965e+00, -2.5747e+00, -2.1456e+00, ..., -1.3981e+00,
-1.1099e+00, -8.8315e-01],
[-2.4124e+00, -2.9606e+00, -2.5873e+00, ..., -1.5908e+00,
-1.1943e+00, -8.8198e-01],
...,
[-3.2224e+00, -4.0220e+00, -3.7214e+00, ..., -2.8700e+00,
-2.2934e+00, -1.7656e+00],
[-2.1388e+00, -2.7677e+00, -2.6762e+00, ..., -1.9995e+00,
-1.6444e+00, -1.3231e+00],
[-1.5288e+00, -1.8632e+00, -1.8740e+00, ..., -1.2493e+00,
-1.0525e+00, -8.0694e-01]]],
[[[-1.8652e-02, 9.0411e-02, -1.7515e+00, ..., -8.1378e-01,
-3.8583e-01, -1.7094e-01],
[-2.2175e+00, -1.8246e+00, -4.9955e+00, ..., -4.1416e+00,
-1.7754e+00, -7.2421e-01],
[ 4.9847e-01, -2.6130e-01, -1.1124e+00, ..., -3.2908e-01,
-4.6587e-01, 1.3044e-01],
...,
[-7.1349e-01, -1.6161e+00, -1.7055e+00, ..., -1.1559e+01,
-7.7939e+00, -8.3012e-01],
[ 8.6083e-01, 1.9428e-01, 3.7808e-02, ..., -9.8149e+00,
-3.2850e+00, 8.2961e-01],
[ 2.7048e-01, -3.5569e-01, -3.1431e-01, ..., -4.6046e+00,
6.7310e-01, 4.3271e-01]],
[[-3.4414e+00, -1.9125e+00, -6.5396e+00, ..., -7.3957e+00,
-6.4911e+00, -4.7731e+00],
[-9.1383e-01, 1.4407e+00, -4.6315e+00, ..., -5.6848e+00,
-5.7796e+00, -4.4201e+00],
[-5.5147e-01, 9.5258e-01, -7.7444e+00, ..., -8.2731e+00,
-8.1389e+00, -6.8585e+00],
...,
[ 1.5920e-01, 3.5575e-01, -4.0703e+00, ..., -6.0734e+00,
-4.6745e+00, -3.5404e+00],
[ 2.2456e+00, 2.3045e+00, -1.2679e-01, ..., -1.3534e+00,
-1.4717e+00, -1.2855e+00],
[ 1.3162e+00, 1.1964e+00, -1.7460e+00, ..., -2.5317e+00,
-2.9294e+00, -2.4342e+00]],
[[-1.5098e+00, -2.4651e+00, -2.4109e+00, ..., -1.8923e+00,
-1.3438e+00, -1.4786e+00],
[-1.7832e+00, -2.0240e+00, -1.5589e+00, ..., -1.3507e+00,
-7.9844e-01, -9.0622e-01],
[-2.3527e+00, -2.0642e+00, -1.0573e+00, ..., -1.2218e+00,
-5.4428e-01, -2.3284e-01],
...,
[-2.6633e-01, -7.3204e-01, -4.7652e-01, ..., 1.3626e+00,
2.6215e-01, -4.5560e-01],
[-7.6920e-01, -2.8048e-01, -4.1782e-01, ..., 1.9434e+00,
9.0924e-01, 2.3458e-01],
[-8.5464e-01, -2.4377e-01, -3.0574e-01, ..., 7.7522e-01,
2.6659e-01, -2.0396e-01]],
...,
[[-2.0074e+00, -1.0688e+00, -2.2945e+00, ..., -3.2105e+00,
-2.6378e+00, -1.8388e+00],
[-1.9654e+00, -1.8652e-01, -1.4944e+00, ..., -2.6589e+00,
-2.6808e+00, -2.0748e+00],
[-4.7379e-01, 2.2553e+00, 1.0614e+00, ..., 1.3581e-01,
2.6820e-01, 5.2819e-01],
...,
[-1.9542e+00, 1.4743e-01, -1.2768e-01, ..., -9.3272e-01,
9.8136e-01, 1.2585e-01],
[-1.0960e+00, 9.3608e-01, 9.5444e-01, ..., 1.4726e+00,
1.8956e+00, 4.3769e-01],
[-2.0279e+00, -1.3521e+00, -1.5919e+00, ..., -4.4025e-01,
-5.0334e-01, -8.6807e-01]],
[[-2.5736e+00, -3.0589e+00, -4.3164e+00, ..., -2.5646e+00,
-2.0789e+00, -1.8839e+00],
[-1.9442e+00, -1.9002e+00, -4.9660e+00, ..., -3.0657e+00,
-2.9585e+00, -2.9203e+00],
[-2.9719e+00, -2.7000e+00, -6.6549e+00, ..., -4.5491e+00,
-4.5325e+00, -3.6217e+00],
...,
[-1.4401e+00, -1.3265e-01, -3.5230e+00, ..., -3.3990e+00,
-1.0089e+00, -9.6179e-01],
[-1.6990e+00, -6.6181e-01, -3.1056e+00, ..., -1.9491e+00,
-6.5173e-01, -8.1863e-01],
[-2.2890e+00, -1.4608e+00, -3.0816e+00, ..., -2.3389e+00,
-1.4990e+00, -1.9265e+00]],
[[-1.9057e+00, -2.6627e+00, -2.6024e+00, ..., -2.8461e+00,
-2.3359e+00, -1.7898e+00],
[-1.9061e+00, -2.6572e+00, -2.3201e+00, ..., -2.5580e+00,
-2.0243e+00, -1.6375e+00],
[-2.3724e+00, -2.9921e+00, -2.7245e+00, ..., -2.9349e+00,
-2.2987e+00, -1.7491e+00],
...,
[-2.3650e+00, -2.9315e+00, -2.7019e+00, ..., -3.9178e+00,
-2.5857e+00, -1.7265e+00],
[-1.5415e+00, -2.0007e+00, -1.9268e+00, ..., -2.7283e+00,
-1.7769e+00, -1.2698e+00],
[-1.0115e+00, -1.2923e+00, -1.3264e+00, ..., -1.6795e+00,
-1.1446e+00, -7.5264e-01]]],
[[[ 1.4112e-01, 3.6859e-01, -1.2540e+00, ..., 4.6686e-01,
1.0149e+00, 6.5353e-01],
[-2.1400e+00, -1.4968e+00, -4.5219e+00, ..., -2.5233e+00,
-7.0568e-01, -3.5873e-02],
[ 4.0470e-01, -3.2386e-01, -1.1921e+00, ..., -2.3843e-01,
5.1187e-01, 6.4712e-01],
...,
[-2.4232e-01, -1.2566e+00, -2.5768e+00, ..., -1.9650e+00,
-1.3658e+00, -8.8578e-01],
[ 1.2477e+00, 6.6540e-01, -9.2121e-01, ..., -2.6285e-01,
5.2738e-01, 7.1011e-01],
[ 7.8755e-01, 2.8632e-01, -4.1920e-01, ..., -3.1630e-01,
2.9887e-01, 2.8137e-01]],
[[-3.2698e+00, -1.4777e+00, -6.2097e+00, ..., -4.8599e+00,
-4.4342e+00, -3.3680e+00],
[-5.7393e-01, 2.1520e+00, -4.2415e+00, ..., -3.4198e+00,
-3.5642e+00, -2.9108e+00],
[ 1.8895e-02, 1.8488e+00, -7.4650e+00, ..., -5.4131e+00,
-5.2891e+00, -4.9450e+00],
...,
[-1.2775e+00, 1.9789e-01, -6.6754e+00, ..., -4.6309e+00,
-4.1719e+00, -3.4818e+00],
[ 1.4495e+00, 2.5122e+00, -1.5472e+00, ..., -9.6098e-01,
-1.2745e+00, -1.2911e+00],
[ 9.5167e-01, 1.5747e+00, -2.5023e+00, ..., -2.2763e+00,
-2.4868e+00, -2.3662e+00]],
[[-1.4245e+00, -2.3907e+00, -2.2020e+00, ..., -6.0306e-01,
-2.1158e-01, -4.7302e-01],
[-2.0561e+00, -2.2256e+00, -1.5769e+00, ..., -3.5709e-01,
4.7610e-02, 2.3295e-02],
[-2.5928e+00, -2.0441e+00, -8.5397e-01, ..., -2.9584e-01,
2.2928e-01, 2.6595e-01],
...,
[-1.6154e+00, -1.7906e+00, -7.6848e-01, ..., -4.8879e-01,
-1.6966e-01, -3.4088e-01],
[-1.7700e+00, -1.1716e+00, -7.4143e-01, ..., -3.2693e-01,
1.3481e-01, 2.1099e-01],
[-1.7195e+00, -1.1135e+00, -5.9914e-01, ..., -4.9451e-01,
-2.7187e-01, -2.3434e-01]],
...,
[[-2.3716e+00, -1.3184e+00, -2.7675e+00, ..., -3.6664e+00,
-2.8495e+00, -1.8495e+00],
[-1.9331e+00, 1.5262e-01, -1.3635e+00, ..., -2.6316e+00,
-2.4169e+00, -1.8568e+00],
[-5.2344e-01, 2.4660e+00, 9.3400e-01, ..., -5.2989e-01,
-6.4260e-02, 1.6848e-01],
...,
[-1.7266e+00, 1.1041e+00, 2.1681e-01, ..., -1.5027e+00,
-6.6609e-01, -1.2422e-01],
[-8.5527e-01, 1.6061e+00, 1.4808e+00, ..., -1.4637e-01,
1.1491e-01, 3.3312e-01],
[-2.0840e+00, -1.0633e+00, -8.4772e-01, ..., -2.3020e+00,
-1.4235e+00, -9.0547e-01]],
[[-2.4006e+00, -2.6799e+00, -4.0516e+00, ..., -1.1031e+00,
-7.0613e-01, -7.3057e-01],
[-1.6884e+00, -1.3719e+00, -4.7960e+00, ..., -1.3544e+00,
-1.2187e+00, -1.5868e+00],
[-3.0387e+00, -2.4807e+00, -6.9145e+00, ..., -3.2900e+00,
-2.8106e+00, -2.2515e+00],
...,
[-2.5761e+00, -1.8231e+00, -4.9944e+00, ..., -1.5446e+00,
-1.0997e+00, -9.1526e-01],
[-2.2472e+00, -1.6385e+00, -3.9109e+00, ..., -1.7648e+00,
-1.1792e+00, -7.7486e-01],
[-2.3124e+00, -1.8537e+00, -3.1113e+00, ..., -2.8003e+00,
-2.0062e+00, -1.8157e+00]],
[[-1.9197e+00, -2.6557e+00, -2.4435e+00, ..., -1.7968e+00,
-1.4929e+00, -1.1359e+00],
[-1.9437e+00, -2.6705e+00, -2.1289e+00, ..., -1.3968e+00,
-1.1099e+00, -8.8400e-01],
[-2.4740e+00, -3.0443e+00, -2.5493e+00, ..., -1.6008e+00,
-1.2006e+00, -8.8806e-01],
...,
[-3.1367e+00, -3.9052e+00, -3.6994e+00, ..., -2.8617e+00,
-2.2869e+00, -1.7569e+00],
[-2.0920e+00, -2.6847e+00, -2.6601e+00, ..., -1.9922e+00,
-1.6308e+00, -1.3121e+00],
[-1.5032e+00, -1.8167e+00, -1.8681e+00, ..., -1.2427e+00,
-1.0440e+00, -7.9864e-01]]],
[[[ 2.8304e-01, 5.4086e-01, 6.7614e-01, ..., 4.6611e-01,
1.0018e+00, 5.8930e-01],
[-2.1943e+00, -1.5390e+00, -3.2230e+00, ..., -2.6984e+00,
-8.9608e-01, -2.2243e-01],
[-1.8130e-01, -1.0044e+00, -2.2519e-01, ..., -1.5597e-01,
5.3210e-01, 6.3324e-01],
...,
[-1.5541e-01, -9.8480e-01, -2.4266e+00, ..., -1.6698e+00,
-1.6096e+00, -1.0520e+00],
[ 1.2991e+00, 8.1037e-01, -9.6001e-01, ..., -2.8742e-01,
7.6421e-02, 3.5390e-01],
[ 7.6296e-01, 2.6163e-01, -6.7741e-01, ..., -1.6288e-01,
2.3220e-01, 3.7735e-01]],
[[-1.6992e+00, -1.1982e+00, -4.1896e+00, ..., -4.8834e+00,
-4.4493e+00, -3.4022e+00],
[ 5.5916e-01, 1.8904e+00, -2.2787e+00, ..., -3.5049e+00,
-3.6113e+00, -2.9671e+00],
[ 1.2862e+00, 1.6565e+00, -4.9097e+00, ..., -5.4383e+00,
-5.2594e+00, -4.9286e+00],
...,
[-1.5247e+00, -3.1775e-01, -6.8976e+00, ..., -7.5801e+00,
-6.8627e+00, -5.3742e+00],
[ 1.3039e+00, 2.2339e+00, -1.6010e+00, ..., -2.3031e+00,
-2.7137e+00, -2.1826e+00],
[ 8.8348e-01, 1.4516e+00, -2.5314e+00, ..., -3.1331e+00,
-3.5595e+00, -3.1062e+00]],
[[-1.1606e-01, -1.1239e+00, -1.2991e+00, ..., -4.9328e-01,
-1.5938e-01, -4.7865e-01],
[-5.6423e-01, -8.2254e-01, -7.6512e-01, ..., -2.8176e-01,
3.9306e-02, -6.8922e-02],
[-5.0048e-01, -3.1633e-01, -3.9640e-01, ..., -3.4218e-01,
8.6548e-02, 8.0564e-02],
...,
[-1.2299e+00, -1.5647e+00, -7.4775e-01, ..., -9.1826e-01,
-7.9613e-01, -1.0596e+00],
[-1.5164e+00, -9.9175e-01, -7.5295e-01, ..., -5.6295e-01,
-2.7639e-01, -3.4039e-01],
[-1.6221e+00, -1.1190e+00, -8.1905e-01, ..., -9.3018e-01,
-8.0505e-01, -7.8826e-01]],
...,
[[-3.0275e+00, -2.5261e+00, -2.8189e+00, ..., -3.4757e+00,
-2.6380e+00, -1.7635e+00],
[-2.3003e+00, -8.8246e-01, -1.6052e+00, ..., -2.6378e+00,
-2.3669e+00, -1.8697e+00],
[-1.1149e+00, 1.1193e+00, 6.0053e-01, ..., -4.5131e-01,
-1.5947e-02, 1.2229e-01],
...,
[-1.6251e+00, 1.0274e+00, -4.8678e-02, ..., -1.1507e+00,
-4.9578e-01, 6.6722e-02],
[-7.2438e-01, 1.6565e+00, 1.4711e+00, ..., 4.8731e-01,
4.8050e-01, 7.4153e-01],
[-2.1461e+00, -1.1663e+00, -9.8898e-01, ..., -2.0922e+00,
-1.0827e+00, -3.6159e-01]],
[[-1.6096e+00, -1.5920e+00, -2.8502e+00, ..., -1.2700e+00,
-8.3648e-01, -8.3514e-01],
[-6.2023e-01, 7.0631e-02, -3.1704e+00, ..., -1.3652e+00,
-1.2047e+00, -1.5642e+00],
[-2.1263e+00, -9.2455e-01, -5.3111e+00, ..., -3.1871e+00,
-2.6889e+00, -2.2005e+00],
...,
[-2.5565e+00, -1.8538e+00, -5.2421e+00, ..., -2.9856e+00,
-2.5978e+00, -2.0806e+00],
[-2.2665e+00, -1.6356e+00, -4.1045e+00, ..., -2.6401e+00,
-2.2158e+00, -1.5399e+00],
[-2.3399e+00, -1.8439e+00, -3.2862e+00, ..., -3.0441e+00,
-2.4331e+00, -2.1271e+00]],
[[-1.3500e+00, -1.9168e+00, -1.7225e+00, ..., -1.7713e+00,
-1.4676e+00, -1.1130e+00],
[-1.2721e+00, -1.7675e+00, -1.2431e+00, ..., -1.3727e+00,
-1.0827e+00, -8.6209e-01],
[-1.5933e+00, -1.9445e+00, -1.4913e+00, ..., -1.6026e+00,
-1.1940e+00, -8.8257e-01],
...,
[-3.0452e+00, -3.8130e+00, -3.6983e+00, ..., -4.0274e+00,
-3.2333e+00, -2.4881e+00],
[-2.0345e+00, -2.6302e+00, -2.6606e+00, ..., -2.8103e+00,
-2.3268e+00, -1.8922e+00],
[-1.4925e+00, -1.8074e+00, -1.9048e+00, ..., -1.8747e+00,
-1.5365e+00, -1.1736e+00]]]], device='cuda:0'),
tensor([[[[ 7.5313e-02, 2.2890e-01, -1.4046e+00, ..., -6.3068e-01,
-1.9841e-01, -6.2459e-02],
[-2.0845e+00, -1.6117e+00, -4.5397e+00, ..., -3.7937e+00,
-1.5337e+00, -5.7106e-01],
[ 5.8391e-01, -7.9997e-02, -8.6354e-01, ..., -2.0159e-01,
-2.5722e-01, 2.2636e-01],
...,
[-3.0081e-01, -1.1016e+00, -1.4134e+00, ..., -1.3414e+00,
-8.9221e-01, -6.7958e-01],
[ 9.7731e-01, 5.6774e-01, 3.3374e-01, ..., 3.6587e-01,
6.8985e-01, 7.9580e-01],
[ 6.7987e-01, 4.3731e-01, 3.3322e-01, ..., 5.0455e-01,
7.7454e-01, 7.0417e-01]],
[[-3.3047e+00, -1.7934e+00, -6.2532e+00, ..., -7.1697e+00,
-6.2754e+00, -4.6318e+00],
[-7.9497e-01, 1.5492e+00, -4.3247e+00, ..., -5.4846e+00,
-5.5641e+00, -4.2702e+00],
[-3.5500e-01, 1.0993e+00, -7.3945e+00, ..., -7.9907e+00,
-7.8656e+00, -6.6853e+00],
...,
[-4.5289e-01, 2.7225e-01, -4.4101e+00, ..., -5.3378e+00,
-4.6407e+00, -3.8692e+00],
[ 1.9526e+00, 2.4292e+00, -6.7151e-02, ..., -1.0722e+00,
-1.2216e+00, -1.3585e+00],
[ 1.1205e+00, 1.3018e+00, -1.6799e+00, ..., -2.4659e+00,
-2.5758e+00, -2.5666e+00]],
[[-1.3483e+00, -2.3623e+00, -2.3257e+00, ..., -1.7681e+00,
-1.2485e+00, -1.3877e+00],
[-1.7171e+00, -2.0143e+00, -1.5490e+00, ..., -1.2538e+00,
-7.2332e-01, -8.1524e-01],
[-2.1902e+00, -1.9475e+00, -1.0217e+00, ..., -1.1007e+00,
-4.5357e-01, -1.7576e-01],
...,
[-6.4705e-01, -1.0880e+00, -5.6088e-01, ..., -5.9217e-01,
-4.6926e-01, -6.3145e-01],
[-1.0433e+00, -6.8341e-01, -5.5667e-01, ..., -2.5975e-01,
-6.5044e-02, -4.4073e-02],
[-1.1969e+00, -6.5820e-01, -3.7828e-01, ..., -3.3525e-01,
-2.8795e-01, -2.5877e-01]],
...,
[[-2.0435e+00, -1.0928e+00, -2.3659e+00, ..., -3.2171e+00,
-2.6226e+00, -1.8190e+00],
[-1.7471e+00, 5.2142e-02, -1.3695e+00, ..., -2.4916e+00,
-2.5232e+00, -1.9547e+00],
[-3.5287e-01, 2.4088e+00, 1.0667e+00, ..., 1.2520e-01,
2.9042e-01, 5.5330e-01],
...,
[-1.5558e+00, 9.5550e-01, -1.3857e-01, ..., -1.1948e+00,
-4.8610e-01, 2.7947e-02],
[-5.8121e-01, 1.3680e+00, 7.5978e-01, ..., 3.9789e-03,
1.6438e-01, 2.9558e-01],
[-1.6157e+00, -1.0975e+00, -1.5281e+00, ..., -2.2856e+00,
-1.3463e+00, -7.0339e-01]],
[[-2.4229e+00, -2.8200e+00, -4.2331e+00, ..., -2.3878e+00,
-1.8962e+00, -1.7026e+00],
[-1.7673e+00, -1.6222e+00, -4.8845e+00, ..., -2.9298e+00,
-2.7739e+00, -2.7622e+00],
[-2.8386e+00, -2.4495e+00, -6.6768e+00, ..., -4.4948e+00,
-4.3610e+00, -3.4936e+00],
...,
[-1.7136e+00, -5.5211e-01, -4.0658e+00, ..., -2.0557e+00,
-1.5443e+00, -1.2150e+00],
[-1.6842e+00, -7.2800e-01, -3.3735e+00, ..., -2.0400e+00,
-1.3435e+00, -8.1822e-01],
[-2.0476e+00, -1.2722e+00, -2.8301e+00, ..., -2.6770e+00,
-1.7927e+00, -1.5394e+00]],
[[-1.8064e+00, -2.5384e+00, -2.4486e+00, ..., -2.7036e+00,
-2.2228e+00, -1.7037e+00],
[-1.7983e+00, -2.5100e+00, -2.1378e+00, ..., -2.4049e+00,
-1.9085e+00, -1.5493e+00],
[-2.2508e+00, -2.8144e+00, -2.5018e+00, ..., -2.7502e+00,
-2.1538e+00, -1.6416e+00],
...,
[-2.2713e+00, -2.8535e+00, -2.6562e+00, ..., -2.8681e+00,
-2.2896e+00, -1.7243e+00],
[-1.5361e+00, -1.9773e+00, -1.9316e+00, ..., -2.0307e+00,
-1.6669e+00, -1.3281e+00],
[-1.0431e+00, -1.2700e+00, -1.3392e+00, ..., -1.2625e+00,
-1.0743e+00, -7.9271e-01]]],
[[[ 1.5914e-01, 3.8096e-01, -1.1854e+00, ..., 4.5607e-01,
1.0083e+00, 6.4698e-01],
[-2.0616e+00, -1.4126e+00, -4.3349e+00, ..., -2.6025e+00,
-7.6569e-01, -6.7365e-02],
[ 4.4381e-01, -2.6369e-01, -1.1344e+00, ..., -2.8591e-01,
4.8288e-01, 6.4390e-01],
...,
[-2.6298e-01, -1.2750e+00, -2.5928e+00, ..., -2.0027e+00,
-1.4099e+00, -9.1575e-01],
[ 1.2542e+00, 6.5347e-01, -9.1843e-01, ..., -3.3241e-01,
4.9855e-01, 6.8500e-01],
[ 8.2803e-01, 3.1811e-01, -4.1632e-01, ..., -3.6328e-01,
2.8957e-01, 2.5844e-01]],
[[-3.2228e+00, -1.4727e+00, -6.1384e+00, ..., -4.8608e+00,
-4.4256e+00, -3.3684e+00],
[-5.7491e-01, 2.1007e+00, -4.1805e+00, ..., -3.3973e+00,
-3.5230e+00, -2.8894e+00],
[ 1.8801e-02, 1.8041e+00, -7.3789e+00, ..., -5.3688e+00,
-5.2322e+00, -4.9216e+00],
...,
[-1.2640e+00, 2.3787e-01, -6.5674e+00, ..., -4.5695e+00,
-4.1267e+00, -3.4310e+00],
[ 1.5082e+00, 2.5869e+00, -1.3647e+00, ..., -8.9591e-01,
-1.2093e+00, -1.2225e+00],
[ 9.8500e-01, 1.6178e+00, -2.4000e+00, ..., -2.2374e+00,
-2.4644e+00, -2.3397e+00]],
[[-1.3854e+00, -2.3535e+00, -2.1591e+00, ..., -6.0168e-01,
-2.0520e-01, -4.7062e-01],
[-2.0154e+00, -2.2002e+00, -1.5628e+00, ..., -3.4906e-01,
6.0781e-02, 3.1720e-02],
[-2.5400e+00, -2.0034e+00, -8.4336e-01, ..., -2.8704e-01,
2.5451e-01, 2.8287e-01],
...,
[-1.5967e+00, -1.8194e+00, -7.8308e-01, ..., -4.8019e-01,
-1.4734e-01, -3.2927e-01],
[-1.7567e+00, -1.1840e+00, -7.7141e-01, ..., -3.1188e-01,
1.6427e-01, 2.2295e-01],
[-1.6988e+00, -1.0984e+00, -5.9515e-01, ..., -5.0630e-01,
-2.6011e-01, -2.3593e-01]],
...,
[[-2.3606e+00, -1.3073e+00, -2.7040e+00, ..., -3.6827e+00,
-2.8541e+00, -1.8573e+00],
[-1.9022e+00, 1.7389e-01, -1.3221e+00, ..., -2.6454e+00,
-2.4161e+00, -1.8630e+00],
[-5.2294e-01, 2.4508e+00, 9.3928e-01, ..., -5.5125e-01,
-5.3903e-02, 1.8250e-01],
...,
[-1.7273e+00, 1.1493e+00, 1.8294e-01, ..., -1.5835e+00,
-7.0183e-01, -1.2257e-01],
[-8.5098e-01, 1.6528e+00, 1.4809e+00, ..., -1.6840e-01,
6.8575e-02, 3.3801e-01],
[-2.1047e+00, -1.0751e+00, -8.8652e-01, ..., -2.3695e+00,
-1.4848e+00, -9.2740e-01]],
[[-2.3826e+00, -2.6547e+00, -4.0277e+00, ..., -1.1175e+00,
-7.2148e-01, -7.4295e-01],
[-1.6734e+00, -1.3553e+00, -4.7459e+00, ..., -1.3443e+00,
-1.2114e+00, -1.5936e+00],
[-3.0229e+00, -2.4595e+00, -6.8695e+00, ..., -3.2821e+00,
-2.7904e+00, -2.2505e+00],
...,
[-2.5346e+00, -1.7423e+00, -4.9532e+00, ..., -1.4506e+00,
-1.0412e+00, -8.9067e-01],
[-2.2316e+00, -1.5696e+00, -3.8801e+00, ..., -1.7271e+00,
-1.1396e+00, -7.6108e-01],
[-2.3183e+00, -1.8013e+00, -3.0920e+00, ..., -2.7952e+00,
-1.9871e+00, -1.8321e+00]],
[[-1.8951e+00, -2.6237e+00, -2.4204e+00, ..., -1.8011e+00,
-1.4970e+00, -1.1397e+00],
[-1.9193e+00, -2.6265e+00, -2.1090e+00, ..., -1.4000e+00,
-1.1056e+00, -8.8137e-01],
[-2.4444e+00, -3.0020e+00, -2.5256e+00, ..., -1.6002e+00,
-1.1948e+00, -8.8359e-01],
...,
[-3.1035e+00, -3.8753e+00, -3.6506e+00, ..., -2.8921e+00,
-2.3140e+00, -1.7842e+00],
[-2.0583e+00, -2.6564e+00, -2.6227e+00, ..., -1.9907e+00,
-1.6361e+00, -1.3249e+00],
[-1.4758e+00, -1.7917e+00, -1.8399e+00, ..., -1.2386e+00,
-1.0450e+00, -8.0270e-01]]],
[[[ 1.0972e-01, 2.6599e-01, -1.3709e+00, ..., -5.7595e-01,
-1.8186e-01, -2.3093e-02],
[-2.1638e+00, -1.6625e+00, -4.6887e+00, ..., -3.5226e+00,
-1.3841e+00, -4.8063e-01],
[ 6.5590e-01, -8.9773e-02, -8.7696e-01, ..., -1.1850e-01,
-1.6995e-01, 3.1700e-01],
...,
[-8.3990e-01, -1.8154e+00, -1.8661e+00, ..., -1.7928e+00,
-1.3136e+00, -9.2653e-01],
[ 7.4660e-01, -1.0969e-02, -1.4140e-01, ..., -2.3007e-02,
7.0260e-01, 7.9588e-01],
[ 1.3005e-01, -5.8136e-01, -4.8931e-01, ..., -1.6209e-01,
4.3254e-01, 3.5438e-01]],
[[-3.3694e+00, -1.6787e+00, -6.2470e+00, ..., -6.9666e+00,
-6.1368e+00, -4.5327e+00],
[-6.4317e-01, 1.9659e+00, -4.1202e+00, ..., -5.5707e+00,
-5.5914e+00, -4.2972e+00],
[-6.2638e-02, 1.5796e+00, -7.3266e+00, ..., -8.0153e+00,
-7.8189e+00, -6.5817e+00],
...,
[ 2.7332e-01, 3.0917e-01, -4.0506e+00, ..., -4.6437e+00,
-4.2303e+00, -3.5187e+00],
[ 2.2565e+00, 2.2269e+00, -2.1994e-01, ..., -8.3415e-01,
-1.2192e+00, -1.2425e+00],
[ 1.3138e+00, 1.1428e+00, -1.8060e+00, ..., -2.2501e+00,
-2.5400e+00, -2.4115e+00]],
[[-1.4057e+00, -2.5218e+00, -2.4416e+00, ..., -1.6817e+00,
-1.2121e+00, -1.3244e+00],
[-1.8557e+00, -2.1815e+00, -1.6218e+00, ..., -1.1634e+00,
-7.0529e-01, -8.1259e-01],
[-2.3191e+00, -2.0562e+00, -1.0220e+00, ..., -1.0854e+00,
-4.6992e-01, -1.9806e-01],
...,
[-1.4252e-01, -5.8294e-01, -4.7076e-01, ..., -5.0726e-01,
-2.0218e-01, -4.3783e-01],
[-6.6937e-01, -1.4756e-01, -3.9591e-01, ..., -2.1299e-01,
2.5060e-01, 2.4582e-01],
[-7.3635e-01, -1.3804e-01, -3.1716e-01, ..., -4.0370e-01,
-1.8701e-01, -2.1958e-01]],
...,
[[-2.2659e+00, -1.1446e+00, -2.4901e+00, ..., -2.9584e+00,
-2.4481e+00, -1.7208e+00],
[-1.8486e+00, 2.0139e-01, -1.3422e+00, ..., -2.4131e+00,
-2.4331e+00, -1.8479e+00],
[-3.9813e-01, 2.6531e+00, 1.1796e+00, ..., 1.2476e-01,
1.5914e-01, 4.2623e-01],
...,
[-2.0567e+00, -1.3410e-01, -1.8601e-01, ..., -1.3595e+00,
-3.7943e-01, 9.9524e-02],
[-1.1915e+00, 7.4681e-01, 9.1425e-01, ..., 2.1798e-03,
3.1782e-01, 4.7375e-01],
[-2.1219e+00, -1.4633e+00, -1.6681e+00, ..., -2.4550e+00,
-1.4647e+00, -9.2558e-01]],
[[-2.4173e+00, -2.7800e+00, -4.3272e+00, ..., -2.5279e+00,
-2.0659e+00, -1.8405e+00],
[-1.6577e+00, -1.4061e+00, -5.0340e+00, ..., -3.0669e+00,
-2.9112e+00, -2.8205e+00],
[-2.8217e+00, -2.2763e+00, -6.9833e+00, ..., -4.4521e+00,
-4.3603e+00, -3.4850e+00],
...,
[-1.4205e+00, -6.4782e-02, -3.3046e+00, ..., -1.4434e+00,
-1.0442e+00, -9.3356e-01],
[-1.7354e+00, -6.5474e-01, -2.9873e+00, ..., -1.7292e+00,
-1.1796e+00, -8.1636e-01],
[-2.3397e+00, -1.5058e+00, -3.1117e+00, ..., -2.9673e+00,
-2.1486e+00, -1.9820e+00]],
[[-1.8584e+00, -2.6321e+00, -2.4891e+00, ..., -2.6262e+00,
-2.1563e+00, -1.6603e+00],
[-1.8393e+00, -2.5819e+00, -2.1265e+00, ..., -2.3993e+00,
-1.8966e+00, -1.5335e+00],
[-2.3034e+00, -2.8722e+00, -2.4683e+00, ..., -2.7703e+00,
-2.1649e+00, -1.6504e+00],
...,
[-2.3764e+00, -2.9485e+00, -2.7397e+00, ..., -2.8599e+00,
-2.2858e+00, -1.7567e+00],
[-1.5374e+00, -2.0037e+00, -1.9422e+00, ..., -1.9421e+00,
-1.6019e+00, -1.2909e+00],
[-9.9884e-01, -1.2892e+00, -1.3320e+00, ..., -1.1879e+00,
-1.0118e+00, -7.7075e-01]]],
[[[ 1.6550e-01, 3.9159e-01, -1.1619e+00, ..., 4.4688e-01,
1.0124e+00, 6.4389e-01],
[-2.0754e+00, -1.4309e+00, -4.3821e+00, ..., -2.6148e+00,
-7.7673e-01, -7.3441e-02],
[ 4.4922e-01, -2.6898e-01, -1.1393e+00, ..., -2.7823e-01,
4.8950e-01, 6.4242e-01],
...,
[-2.8219e-01, -1.2728e+00, -2.6150e+00, ..., -2.0655e+00,
-1.3437e+00, -9.1532e-01],
[ 1.2470e+00, 6.3531e-01, -9.4224e-01, ..., -3.9919e-01,
5.4383e-01, 6.8257e-01],
[ 8.2533e-01, 3.0236e-01, -4.3693e-01, ..., -4.1298e-01,
2.9259e-01, 2.4835e-01]],
[[-3.2483e+00, -1.4557e+00, -6.1369e+00, ..., -4.8608e+00,
-4.4422e+00, -3.3702e+00],
[-5.6831e-01, 2.1665e+00, -4.1605e+00, ..., -3.3876e+00,
-3.5473e+00, -2.8982e+00],
[ 4.7558e-02, 1.8765e+00, -7.3912e+00, ..., -5.3848e+00,
-5.2606e+00, -4.9252e+00],
...,
[-1.2696e+00, 2.4974e-01, -6.5392e+00, ..., -4.5423e+00,
-4.1418e+00, -3.3838e+00],
[ 1.5033e+00, 2.6079e+00, -1.3298e+00, ..., -8.9178e-01,
-1.2183e+00, -1.2325e+00],
[ 9.8254e-01, 1.6329e+00, -2.4074e+00, ..., -2.2575e+00,
-2.4741e+00, -2.3503e+00]],
[[-1.3826e+00, -2.3640e+00, -2.1782e+00, ..., -6.0331e-01,
-1.9609e-01, -4.6997e-01],
[-2.0249e+00, -2.2212e+00, -1.5800e+00, ..., -3.5490e-01,
7.4497e-02, 3.8468e-02],
[-2.5290e+00, -2.0073e+00, -8.4906e-01, ..., -2.8619e-01,
2.5068e-01, 2.8371e-01],
...,
[-1.5799e+00, -1.8154e+00, -7.6665e-01, ..., -4.9298e-01,
-1.4816e-01, -3.4526e-01],
[-1.7385e+00, -1.1762e+00, -7.5833e-01, ..., -3.2801e-01,
1.4786e-01, 2.0893e-01],
[-1.6916e+00, -1.1045e+00, -5.9565e-01, ..., -5.2179e-01,
-2.7781e-01, -2.6345e-01]],
...,
[[-2.3761e+00, -1.3031e+00, -2.7589e+00, ..., -3.7108e+00,
-2.8732e+00, -1.8474e+00],
[-1.8830e+00, 2.2403e-01, -1.3235e+00, ..., -2.6849e+00,
-2.4656e+00, -1.9017e+00],
[-5.0918e-01, 2.5011e+00, 9.5084e-01, ..., -5.5090e-01,
-5.8935e-02, 1.6301e-01],
...,
[-1.7130e+00, 1.1723e+00, 2.1195e-01, ..., -1.4662e+00,
-7.3737e-01, -1.5923e-01],
[-8.6675e-01, 1.6384e+00, 1.4741e+00, ..., -7.9416e-02,
8.2654e-02, 3.1274e-01],
[-2.1249e+00, -1.0957e+00, -9.0420e-01, ..., -2.2725e+00,
-1.4445e+00, -9.0950e-01]],
[[-2.3687e+00, -2.6228e+00, -4.0178e+00, ..., -1.1200e+00,
-7.2222e-01, -7.5347e-01],
[-1.6383e+00, -1.3025e+00, -4.7375e+00, ..., -1.3425e+00,
-1.2158e+00, -1.5954e+00],
[-3.0072e+00, -2.4196e+00, -6.8887e+00, ..., -3.2807e+00,
-2.8058e+00, -2.2407e+00],
...,
[-2.5165e+00, -1.7313e+00, -4.9219e+00, ..., -1.4258e+00,
-1.0081e+00, -8.8156e-01],
[-2.2154e+00, -1.5619e+00, -3.8649e+00, ..., -1.6780e+00,
-1.1202e+00, -7.6134e-01],
[-2.3108e+00, -1.7909e+00, -3.0752e+00, ..., -2.7597e+00,
-2.0025e+00, -1.8291e+00]],
[[-1.8966e+00, -2.6258e+00, -2.4044e+00, ..., -1.8057e+00,
-1.4980e+00, -1.1397e+00],
[-1.9170e+00, -2.6347e+00, -2.0900e+00, ..., -1.3980e+00,
-1.1056e+00, -8.8388e-01],
[-2.4458e+00, -2.9947e+00, -2.5032e+00, ..., -1.5937e+00,
-1.1998e+00, -8.8771e-01],
...,
[-3.1006e+00, -3.8729e+00, -3.6599e+00, ..., -2.8863e+00,
-2.3050e+00, -1.7799e+00],
[-2.0541e+00, -2.6591e+00, -2.6340e+00, ..., -1.9998e+00,
-1.6449e+00, -1.3269e+00],
[-1.4700e+00, -1.7961e+00, -1.8510e+00, ..., -1.2466e+00,
-1.0515e+00, -8.0984e-01]]],
[[[ 2.1985e-01, 4.8550e-01, 6.1647e-01, ..., -3.4502e-01,
-8.8514e-02, 7.7705e-03],
[-2.0054e+00, -1.3349e+00, -2.8433e+00, ..., -3.2000e+00,
-1.3109e+00, -4.4931e-01],
[-1.6634e-01, -8.8828e-01, 4.7802e-02, ..., -1.6521e-01,
-3.6164e-01, 1.6527e-01],
...,
[-9.7624e-01, -2.2272e+00, -2.0269e+00, ..., -1.9167e+00,
-1.8824e+00, -1.2561e+00],
[ 7.7614e-01, -1.6684e-01, -8.6734e-02, ..., 3.5354e-02,
2.8292e-01, 4.7756e-01],
[ 2.0418e-01, -6.0634e-01, -4.0673e-01, ..., 2.2549e-01,
4.7050e-01, 5.1652e-01]],
[[-1.6726e+00, -1.4521e+00, -4.3032e+00, ..., -6.8274e+00,
-6.0375e+00, -4.5221e+00],
[ 4.3512e-01, 1.3472e+00, -2.6213e+00, ..., -5.4113e+00,
-5.5195e+00, -4.2732e+00],
[ 1.1354e+00, 1.0742e+00, -5.1105e+00, ..., -7.8006e+00,
-7.7642e+00, -6.5710e+00],
...,
[ 4.2788e-01, 3.9717e-01, -4.2214e+00, ..., -7.0922e+00,
-6.6691e+00, -5.1926e+00],
[ 2.5591e+00, 2.5067e+00, 5.1295e-02, ..., -1.8895e+00,
-2.5331e+00, -2.0462e+00],
[ 1.5096e+00, 1.3096e+00, -1.6485e+00, ..., -2.9147e+00,
-3.4736e+00, -3.0499e+00]],
[[-1.2657e-01, -1.1239e+00, -1.3669e+00, ..., -1.3986e+00,
-1.0440e+00, -1.3322e+00],
[-4.4789e-01, -7.6100e-01, -8.5356e-01, ..., -9.8197e-01,
-6.1844e-01, -8.9543e-01],
[-2.5678e-01, -1.7505e-01, -4.2512e-01, ..., -1.0875e+00,
-5.8726e-01, -4.6504e-01],
...,
[-2.9845e-02, -5.7815e-01, -5.8091e-01, ..., -1.0398e+00,
-8.8311e-01, -1.2551e+00],
[-6.8679e-01, -2.1687e-01, -5.7012e-01, ..., -5.5452e-01,
-2.4390e-01, -4.1202e-01],
[-8.7697e-01, -2.7720e-01, -4.8662e-01, ..., -4.9650e-01,
-4.3962e-01, -5.9894e-01]],
...,
[[-2.8085e+00, -2.5071e+00, -2.4494e+00, ..., -3.0599e+00,
-2.4857e+00, -1.8199e+00],
[-2.1528e+00, -9.5878e-01, -1.1400e+00, ..., -2.4174e+00,
-2.4875e+00, -2.0243e+00],
[-1.2216e+00, 6.1043e-01, 6.8565e-01, ..., -1.3547e-01,
5.5217e-03, 2.6944e-01],
...,
[-2.3714e+00, -2.1007e-01, -1.6195e-01, ..., -9.3878e-01,
-2.5303e-01, 2.9721e-01],
[-1.2811e+00, 8.4028e-01, 1.0722e+00, ..., 4.2943e-01,
4.9407e-01, 8.0893e-01],
[-2.1857e+00, -1.4860e+00, -1.6784e+00, ..., -2.2429e+00,
-1.1854e+00, -3.6400e-01]],
[[-1.6625e+00, -1.6741e+00, -2.7269e+00, ..., -2.2350e+00,
-1.8927e+00, -1.7919e+00],
[-7.4503e-01, -1.5888e-01, -3.0655e+00, ..., -2.8372e+00,
-2.8442e+00, -2.8679e+00],
[-2.1057e+00, -1.0595e+00, -5.0602e+00, ..., -4.4447e+00,
-4.5039e+00, -3.7027e+00],
...,
[-1.2803e+00, 1.8352e-01, -3.4311e+00, ..., -2.8389e+00,
-2.7279e+00, -2.3024e+00],
[-1.6283e+00, -5.0547e-01, -3.0703e+00, ..., -2.4514e+00,
-2.2531e+00, -1.6427e+00],
[-2.3078e+00, -1.3758e+00, -3.0732e+00, ..., -2.8746e+00,
-2.4292e+00, -2.1822e+00]],
[[-1.2908e+00, -1.8207e+00, -1.6692e+00, ..., -2.5621e+00,
-2.1409e+00, -1.6618e+00],
[-1.2308e+00, -1.6916e+00, -1.2446e+00, ..., -2.3916e+00,
-1.9267e+00, -1.5767e+00],
[-1.5420e+00, -1.8751e+00, -1.4918e+00, ..., -2.7994e+00,
-2.2340e+00, -1.7321e+00],
...,
[-2.4901e+00, -3.0773e+00, -2.8060e+00, ..., -3.9999e+00,
-3.2366e+00, -2.5424e+00],
[-1.6093e+00, -2.0974e+00, -1.9955e+00, ..., -2.7599e+00,
-2.3096e+00, -1.9117e+00],
[-1.0504e+00, -1.3587e+00, -1.3662e+00, ..., -1.8203e+00,
-1.5177e+00, -1.1760e+00]]]], device='cuda:0'),
tensor([[[[ 2.1666e-01, 4.8057e-01, 6.2629e-01, ..., 5.8772e-01,
9.0945e-01, 7.3820e-01],
[-2.0411e+00, -1.3536e+00, -2.8554e+00, ..., -1.4938e+00,
-2.9172e-01, 3.2792e-01],
[-2.0257e-01, -9.4196e-01, -2.1238e-02, ..., -6.0272e-02,
4.2149e-01, 4.9065e-01],
...,
[-1.0580e+00, -2.2278e+00, -2.0988e+00, ..., -1.2742e+00,
-8.5434e-01, -6.5013e-01],
[ 7.7823e-01, -6.1049e-02, -6.5846e-02, ..., 5.7279e-01,
8.3305e-01, 9.1036e-01],
[ 1.6457e-01, -5.5050e-01, -4.1286e-01, ..., 6.9385e-01,
8.9374e-01, 7.9626e-01]],
[[-1.6650e+00, -1.3937e+00, -4.2089e+00, ..., -5.7551e+00,
-4.8789e+00, -3.8077e+00],
[ 4.8424e-01, 1.4583e+00, -2.4792e+00, ..., -4.6617e+00,
-4.4344e+00, -3.6145e+00],
[ 1.2087e+00, 1.1884e+00, -4.9898e+00, ..., -6.6592e+00,
-6.3113e+00, -5.6409e+00],
...,
[ 3.9559e-01, 3.9531e-01, -4.1249e+00, ..., -6.1545e+00,
-5.2141e+00, -4.2824e+00],
[ 2.5067e+00, 2.4714e+00, 3.5150e-02, ..., -1.5301e+00,
-1.5251e+00, -1.5713e+00],
[ 1.4858e+00, 1.3021e+00, -1.6889e+00, ..., -2.8246e+00,
-2.8431e+00, -2.7704e+00]],
[[-1.3362e-01, -1.1426e+00, -1.3957e+00, ..., -8.0932e-01,
-5.5397e-01, -6.7923e-01],
[-4.9069e-01, -7.8738e-01, -8.5730e-01, ..., -7.0399e-01,
-3.4973e-01, -2.2080e-01],
[-3.1901e-01, -1.9264e-01, -4.3403e-01, ..., -3.8201e-01,
1.2507e-01, 2.1099e-01],
...,
[-1.7728e-02, -5.6533e-01, -6.0137e-01, ..., -6.8097e-01,
-4.4675e-01, -5.4331e-01],
[-6.7867e-01, -2.2641e-01, -6.1111e-01, ..., -3.0196e-01,
-1.8534e-03, 8.7355e-02],
[-8.9490e-01, -2.7801e-01, -5.0674e-01, ..., -2.8370e-01,
-1.8329e-01, -1.2657e-01]],
...,
[[-2.8464e+00, -2.4958e+00, -2.4684e+00, ..., -3.1462e+00,
-2.5343e+00, -1.7018e+00],
[-2.1142e+00, -9.0376e-01, -1.1672e+00, ..., -1.6926e+00,
-1.7817e+00, -1.5298e+00],
[-1.1144e+00, 7.2978e-01, 7.3507e-01, ..., -4.2417e-01,
-2.4065e-01, 2.9548e-02],
...,
[-2.2636e+00, -1.9765e-01, -1.2402e-01, ..., -1.1214e+00,
-3.6399e-01, 1.8795e-01],
[-1.2827e+00, 8.3332e-01, 1.0878e+00, ..., 1.3983e-01,
3.3595e-01, 4.3547e-01],
[-2.1260e+00, -1.4514e+00, -1.5735e+00, ..., -2.2169e+00,
-1.2396e+00, -6.1924e-01]],
[[-1.6234e+00, -1.6314e+00, -2.8025e+00, ..., -1.3239e+00,
-7.1989e-01, -5.0414e-01],
[-6.7495e-01, -8.8077e-02, -3.1594e+00, ..., -1.9146e+00,
-1.5623e+00, -1.6228e+00],
[-2.0481e+00, -9.7534e-01, -5.1358e+00, ..., -3.8577e+00,
-3.3142e+00, -2.5423e+00],
...,
[-1.3539e+00, 1.5403e-01, -3.5179e+00, ..., -2.5198e+00,
-1.8294e+00, -1.3414e+00],
[-1.6408e+00, -4.6317e-01, -3.1135e+00, ..., -2.3552e+00,
-1.5246e+00, -8.8163e-01],
[-2.2660e+00, -1.3341e+00, -3.0618e+00, ..., -2.9079e+00,
-1.9392e+00, -1.6220e+00]],
[[-1.3018e+00, -1.8255e+00, -1.6512e+00, ..., -1.8601e+00,
-1.5104e+00, -1.1037e+00],
[-1.2412e+00, -1.7047e+00, -1.2235e+00, ..., -1.6839e+00,
-1.3466e+00, -1.0216e+00],
[-1.5584e+00, -1.9062e+00, -1.4818e+00, ..., -2.0356e+00,
-1.5669e+00, -1.1257e+00],
...,
[-2.4668e+00, -3.0807e+00, -2.7954e+00, ..., -3.1670e+00,
-2.4968e+00, -1.8488e+00],
[-1.6240e+00, -2.1300e+00, -2.0283e+00, ..., -2.2530e+00,
-1.8188e+00, -1.4236e+00],
[-1.0793e+00, -1.3972e+00, -1.4107e+00, ..., -1.3950e+00,
-1.1723e+00, -8.4558e-01]]],
[[[ 9.3422e-02, 3.0643e-01, -1.2068e+00, ..., 4.5118e-01,
9.9432e-01, 6.4738e-01],
[-2.1208e+00, -1.4772e+00, -4.3163e+00, ..., -2.4630e+00,
-6.9498e-01, -3.0610e-02],
[ 3.6872e-01, -3.8584e-01, -1.1851e+00, ..., -2.3124e-01,
5.1598e-01, 6.4634e-01],
...,
[-2.3968e-01, -1.2714e+00, -2.5545e+00, ..., -4.0444e+00,
-3.2935e+00, -2.3680e+00],
[ 1.2600e+00, 6.5815e-01, -9.0507e-01, ..., -7.3106e-01,
-2.1267e-02, 2.9712e-01],
[ 8.2094e-01, 3.0338e-01, -4.3021e-01, ..., -2.9957e-01,
3.3980e-01, 3.0948e-01]],
[[-3.2106e+00, -1.4395e+00, -6.0905e+00, ..., -4.7451e+00,
-4.3416e+00, -3.3122e+00],
[-5.7954e-01, 2.1001e+00, -4.2024e+00, ..., -3.4039e+00,
-3.5394e+00, -2.9113e+00],
[ 3.0859e-03, 1.7902e+00, -7.4312e+00, ..., -5.3195e+00,
-5.1934e+00, -4.8674e+00],
...,
[-1.2254e+00, 2.9050e-01, -6.5809e+00, ..., -4.8219e+00,
-4.5741e+00, -4.2688e+00],
[ 1.5426e+00, 2.6144e+00, -1.3759e+00, ..., -3.5619e-01,
-9.7424e-01, -1.4297e+00],
[ 1.0118e+00, 1.6442e+00, -2.4252e+00, ..., -2.1333e+00,
-2.4065e+00, -2.3483e+00]],
[[-1.3987e+00, -2.3019e+00, -2.1054e+00, ..., -5.4929e-01,
-1.7306e-01, -4.3553e-01],
[-2.0405e+00, -2.1747e+00, -1.4901e+00, ..., -3.1864e-01,
7.5658e-02, 3.6925e-02],
[-2.5882e+00, -2.0437e+00, -8.1722e-01, ..., -2.7232e-01,
2.2705e-01, 2.6280e-01],
...,
[-1.6187e+00, -1.8535e+00, -7.7553e-01, ..., -6.9146e-01,
-3.7966e-01, -4.6422e-01],
[-1.7936e+00, -1.2156e+00, -7.7383e-01, ..., -4.2658e-01,
-7.6423e-03, 4.6137e-02],
[-1.7608e+00, -1.1484e+00, -6.1957e-01, ..., -4.8447e-01,
-2.6084e-01, -2.4497e-01]],
...,
[[-2.2991e+00, -1.2143e+00, -2.6607e+00, ..., -3.5025e+00,
-2.6978e+00, -1.7529e+00],
[-1.8765e+00, 1.7992e-01, -1.4276e+00, ..., -2.5554e+00,
-2.3469e+00, -1.8188e+00],
[-4.3001e-01, 2.5797e+00, 8.9745e-01, ..., -5.1888e-01,
-5.5297e-02, 1.5268e-01],
...,
[-1.7710e+00, 1.1633e+00, 2.0498e-01, ..., -3.0316e+00,
-2.0208e+00, -6.5984e-01],
[-8.6439e-01, 1.6611e+00, 1.4826e+00, ..., -4.6808e-02,
2.2348e-01, 3.7804e-01],
[-2.0774e+00, -1.0277e+00, -8.2378e-01, ..., -2.5815e+00,
-1.5579e+00, -9.5246e-01]],
[[-2.3940e+00, -2.6392e+00, -4.0365e+00, ..., -1.1845e+00,
-7.9079e-01, -7.9329e-01],
[-1.6916e+00, -1.3243e+00, -4.7382e+00, ..., -1.4074e+00,
-1.2658e+00, -1.5851e+00],
[-3.0113e+00, -2.3719e+00, -6.8079e+00, ..., -3.2509e+00,
-2.7614e+00, -2.1954e+00],
...,
[-2.5313e+00, -1.7675e+00, -5.0304e+00, ..., -5.1171e-01,
2.9559e-02, -1.2952e-01],
[-2.2133e+00, -1.5695e+00, -3.9208e+00, ..., -1.4463e+00,
-6.4665e-01, -4.7200e-01],
[-2.2933e+00, -1.7820e+00, -3.0938e+00, ..., -2.8741e+00,
-2.0247e+00, -1.8509e+00]],
[[-1.9189e+00, -2.6468e+00, -2.4262e+00, ..., -1.7686e+00,
-1.4685e+00, -1.1221e+00],
[-1.9378e+00, -2.6701e+00, -2.1340e+00, ..., -1.4017e+00,
-1.1138e+00, -8.9099e-01],
[-2.4750e+00, -3.0607e+00, -2.5767e+00, ..., -1.6215e+00,
-1.2215e+00, -9.0928e-01],
...,
[-3.1314e+00, -3.9033e+00, -3.6651e+00, ..., -3.3899e+00,
-2.5886e+00, -1.9603e+00],
[-2.0838e+00, -2.6870e+00, -2.6422e+00, ..., -2.1430e+00,
-1.6983e+00, -1.3243e+00],
[-1.4993e+00, -1.8181e+00, -1.8619e+00, ..., -1.2279e+00,
-1.0529e+00, -8.0561e-01]]],
[[[ 2.7205e-01, 5.4531e-01, 6.8552e-01, ..., -3.7336e-01,
-1.1373e-01, -8.4993e-03],
[-2.0977e+00, -1.3540e+00, -2.9411e+00, ..., -3.2966e+00,
-1.3419e+00, -4.7383e-01],
[-6.3074e-02, -8.6278e-01, 5.7772e-02, ..., -2.0354e-01,
-3.9317e-01, 1.5573e-01],
...,
[-9.5226e-01, -2.1205e+00, -1.9789e+00, ..., -1.9792e+00,
-1.8754e+00, -1.2549e+00],
[ 7.3807e-01, -1.9282e-01, -1.6081e-01, ..., -1.1689e-01,
2.6540e-01, 4.4542e-01],
[ 1.5629e-01, -6.3540e-01, -4.3811e-01, ..., 1.2674e-01,
4.9855e-01, 4.8912e-01]],
[[-1.7409e+00, -1.2781e+00, -4.2827e+00, ..., -6.8794e+00,
-6.1065e+00, -4.5516e+00],
[ 6.6379e-01, 1.8945e+00, -2.3206e+00, ..., -5.4656e+00,
-5.5775e+00, -4.3107e+00],
[ 1.5214e+00, 1.6651e+00, -5.0326e+00, ..., -7.8714e+00,
-7.8330e+00, -6.6132e+00],
...,
[ 3.4107e-01, 2.6454e-01, -4.2194e+00, ..., -7.1283e+00,
-6.7518e+00, -5.1818e+00],
[ 2.3556e+00, 2.2625e+00, -2.0586e-01, ..., -1.9245e+00,
-2.6128e+00, -2.0287e+00],
[ 1.3831e+00, 1.1725e+00, -1.8067e+00, ..., -2.9444e+00,
-3.5217e+00, -3.0282e+00]],
[[-2.3104e-01, -1.3855e+00, -1.5968e+00, ..., -1.4315e+00,
-1.0622e+00, -1.3473e+00],
[-7.0845e-01, -1.0494e+00, -1.0019e+00, ..., -1.0172e+00,
-6.3448e-01, -9.0769e-01],
[-5.2280e-01, -3.6185e-01, -4.6531e-01, ..., -1.1245e+00,
-6.0855e-01, -4.7083e-01],
...,
[-3.9879e-02, -5.5073e-01, -5.7208e-01, ..., -1.0164e+00,
-8.5874e-01, -1.2443e+00],
[-6.6624e-01, -2.0841e-01, -5.6381e-01, ..., -5.6509e-01,
-2.2342e-01, -4.2279e-01],
[-8.1108e-01, -2.4868e-01, -4.7787e-01, ..., -5.3589e-01,
-4.4889e-01, -6.1478e-01]],
...,
[[-3.1066e+00, -2.5507e+00, -2.6418e+00, ..., -3.0855e+00,
-2.5242e+00, -1.8331e+00],
[-2.1703e+00, -6.7500e-01, -1.1058e+00, ..., -2.4389e+00,
-2.5329e+00, -2.0551e+00],
[-1.1124e+00, 1.0799e+00, 8.6551e-01, ..., -8.9820e-02,
1.2010e-02, 2.6475e-01],
...,
[-2.1442e+00, -1.7292e-01, -1.3397e-01, ..., -8.8671e-01,
-3.0442e-01, 2.8959e-01],
[-1.1719e+00, 8.6165e-01, 1.0594e+00, ..., 5.0576e-01,
4.5053e-01, 8.5467e-01],
[-2.0532e+00, -1.3453e+00, -1.5173e+00, ..., -2.1598e+00,
-1.1882e+00, -3.4603e-01]],
[[-1.6172e+00, -1.5962e+00, -2.9453e+00, ..., -2.2482e+00,
-1.9291e+00, -1.8257e+00],
[-5.6777e-01, 1.1595e-01, -3.3310e+00, ..., -2.8618e+00,
-2.8743e+00, -2.8982e+00],
[-2.0507e+00, -8.4458e-01, -5.4744e+00, ..., -4.4644e+00,
-4.5276e+00, -3.7229e+00],
...,
[-1.3646e+00, 9.1828e-02, -3.3510e+00, ..., -2.8038e+00,
-2.7264e+00, -2.3430e+00],
[-1.6621e+00, -5.1454e-01, -2.9728e+00, ..., -2.4182e+00,
-2.2338e+00, -1.6878e+00],
[-2.2606e+00, -1.3404e+00, -2.9820e+00, ..., -2.8372e+00,
-2.3845e+00, -2.1973e+00]],
[[-1.3521e+00, -1.9099e+00, -1.6759e+00, ..., -2.5899e+00,
-2.1585e+00, -1.6776e+00],
[-1.2790e+00, -1.7613e+00, -1.1873e+00, ..., -2.4176e+00,
-1.9425e+00, -1.5926e+00],
[-1.6059e+00, -1.9318e+00, -1.4149e+00, ..., -2.8353e+00,
-2.2556e+00, -1.7518e+00],
...,
[-2.4116e+00, -3.0061e+00, -2.7542e+00, ..., -4.0391e+00,
-3.2760e+00, -2.5823e+00],
[-1.5740e+00, -2.0672e+00, -1.9844e+00, ..., -2.8002e+00,
-2.3407e+00, -1.9412e+00],
[-1.0346e+00, -1.3495e+00, -1.3771e+00, ..., -1.8613e+00,
-1.5438e+00, -1.2038e+00]]],
[[[ 1.4563e-01, 3.8823e-01, -1.1808e+00, ..., 6.5098e-01,
9.4563e-01, 6.9327e-01],
[-2.0930e+00, -1.4248e+00, -4.3747e+00, ..., -1.5167e+00,
-1.7349e-01, 3.9389e-01],
[ 4.0938e-01, -2.9205e-01, -1.1371e+00, ..., 5.2796e-01,
9.4669e-01, 8.3393e-01],
...,
[-2.6882e-01, -1.2677e+00, -2.6085e+00, ..., -1.1656e+00,
-8.2476e-01, -6.4500e-01],
[ 1.2614e+00, 6.7303e-01, -9.0451e-01, ..., 5.1722e-01,
7.4555e-01, 8.0023e-01],
[ 8.3478e-01, 3.3009e-01, -4.0525e-01, ..., 5.3275e-01,
7.7357e-01, 6.8597e-01]],
[[-3.2602e+00, -1.5033e+00, -6.1476e+00, ..., -5.4729e+00,
-4.6817e+00, -3.6449e+00],
[-6.1099e-01, 2.0768e+00, -4.1926e+00, ..., -4.2105e+00,
-4.0883e+00, -3.3550e+00],
[-1.8249e-02, 1.7797e+00, -7.4015e+00, ..., -6.1638e+00,
-5.9056e+00, -5.4866e+00],
...,
[-1.2485e+00, 2.7360e-01, -6.5231e+00, ..., -5.3974e+00,
-4.7226e+00, -3.9570e+00],
[ 1.5498e+00, 2.6361e+00, -1.2986e+00, ..., -1.3015e+00,
-1.4222e+00, -1.5091e+00],
[ 1.0198e+00, 1.6505e+00, -2.3794e+00, ..., -2.4957e+00,
-2.6066e+00, -2.5703e+00]],
[[-1.3773e+00, -2.3339e+00, -2.1610e+00, ..., -8.4217e-01,
-6.2924e-01, -8.1687e-01],
[-1.9885e+00, -2.1620e+00, -1.5492e+00, ..., -7.0291e-01,
-4.2341e-01, -3.5212e-01],
[-2.5010e+00, -1.9753e+00, -8.2264e-01, ..., -4.2074e-01,
-1.0980e-01, -9.4713e-02],
...,
[-1.5887e+00, -1.8422e+00, -7.9617e-01, ..., -6.2808e-01,
-5.4394e-01, -7.0030e-01],
[-1.7603e+00, -1.1946e+00, -7.8868e-01, ..., -2.1294e-01,
-6.0014e-02, -5.4434e-02],
[-1.7270e+00, -1.1161e+00, -6.0044e-01, ..., -3.1381e-01,
-2.7790e-01, -2.6018e-01]],
...,
[[-2.3405e+00, -1.3165e+00, -2.7496e+00, ..., -3.1201e+00,
-2.4720e+00, -1.6465e+00],
[-1.8740e+00, 1.8864e-01, -1.3235e+00, ..., -1.5359e+00,
-1.5904e+00, -1.3688e+00],
[-5.3540e-01, 2.4330e+00, 9.2468e-01, ..., -8.5538e-02,
2.3753e-01, 3.2766e-01],
...,
[-1.7346e+00, 1.1747e+00, 2.0836e-01, ..., -1.0084e+00,
-2.8105e-01, 1.7904e-01],
[-8.5602e-01, 1.6663e+00, 1.5086e+00, ..., 1.4357e-01,
3.1997e-01, 4.1826e-01],
[-2.1001e+00, -1.0650e+00, -8.6471e-01, ..., -2.1652e+00,
-1.2414e+00, -6.3723e-01]],
[[-2.3730e+00, -2.6236e+00, -3.9500e+00, ..., -1.1168e+00,
-6.1792e-01, -4.9794e-01],
[-1.6497e+00, -1.3177e+00, -4.6719e+00, ..., -1.6971e+00,
-1.4990e+00, -1.7342e+00],
[-3.0018e+00, -2.4294e+00, -6.8329e+00, ..., -3.7601e+00,
-3.2672e+00, -2.6342e+00],
...,
[-2.5154e+00, -1.7342e+00, -4.9867e+00, ..., -2.2045e+00,
-1.7530e+00, -1.4100e+00],
[-2.2157e+00, -1.5563e+00, -3.8935e+00, ..., -2.1293e+00,
-1.4989e+00, -9.8340e-01],
[-2.3176e+00, -1.7823e+00, -3.0844e+00, ..., -2.8078e+00,
-1.9633e+00, -1.6878e+00]],
[[-1.8945e+00, -2.6227e+00, -2.4116e+00, ..., -1.6954e+00,
-1.4064e+00, -1.0532e+00],
[-1.9135e+00, -2.6361e+00, -2.1046e+00, ..., -1.3679e+00,
-1.1092e+00, -8.7704e-01],
[-2.4340e+00, -2.9894e+00, -2.5097e+00, ..., -1.5714e+00,
-1.2005e+00, -8.6792e-01],
...,
[-3.0987e+00, -3.8670e+00, -3.6405e+00, ..., -2.7887e+00,
-2.2339e+00, -1.6780e+00],
[-2.0564e+00, -2.6574e+00, -2.6189e+00, ..., -1.9567e+00,
-1.6084e+00, -1.2759e+00],
[-1.4781e+00, -1.7939e+00, -1.8388e+00, ..., -1.2223e+00,
-1.0429e+00, -7.6610e-01]]],
[[[ 2.2340e-01, 4.9848e-01, 6.3713e-01, ..., 6.3369e-01,
9.8992e-01, 7.9553e-01],
[-2.0037e+00, -1.3105e+00, -2.8292e+00, ..., -1.4309e+00,
-5.3474e-03, 6.0703e-01],
[-1.1912e-01, -8.7194e-01, 2.1723e-02, ..., -2.5353e+00,
-2.2152e+00, -8.9609e-01],
...,
[-9.6913e-01, -2.1317e+00, -2.0597e+00, ..., -1.1614e+00,
-7.3477e-01, -4.9855e-01],
[ 8.0207e-01, -9.5817e-02, -1.3979e-01, ..., 4.3867e-01,
7.9210e-01, 8.8754e-01],
[ 2.0808e-01, -5.6701e-01, -4.2096e-01, ..., 5.9634e-01,
8.6602e-01, 7.8238e-01]],
[[-1.6918e+00, -1.3621e+00, -4.2718e+00, ..., -5.9772e+00,
-5.0342e+00, -3.9229e+00],
[ 4.7983e-01, 1.5025e+00, -2.5352e+00, ..., -4.7382e+00,
-4.4631e+00, -3.6673e+00],
[ 1.2171e+00, 1.2287e+00, -5.0642e+00, ..., -6.6086e+00,
-6.1644e+00, -6.8652e+00],
...,
[ 3.7734e-01, 3.6675e-01, -4.1330e+00, ..., -6.0331e+00,
-5.2149e+00, -4.2852e+00],
[ 2.5077e+00, 2.4963e+00, 9.1007e-02, ..., -1.6494e+00,
-1.7022e+00, -1.6965e+00],
[ 1.4897e+00, 1.3296e+00, -1.6482e+00, ..., -2.8000e+00,
-2.8596e+00, -2.7591e+00]],
[[-1.4103e-01, -1.1862e+00, -1.4195e+00, ..., -8.7142e-01,
-5.8810e-01, -7.2023e-01],
[-5.2483e-01, -8.5477e-01, -8.9985e-01, ..., -7.2342e-01,
-3.8931e-01, -2.4535e-01],
[-3.4493e-01, -2.4181e-01, -4.4517e-01, ..., -2.9311e-01,
-1.3732e-01, -5.2723e-01],
...,
[-6.8039e-02, -5.9718e-01, -6.1054e-01, ..., -6.2553e-01,
-4.4505e-01, -5.2519e-01],
[-7.0479e-01, -2.1366e-01, -5.6173e-01, ..., -2.8120e-01,
-1.7809e-02, 6.2236e-02],
[-8.5997e-01, -2.4953e-01, -4.5733e-01, ..., -2.8984e-01,
-2.0139e-01, -1.4004e-01]],
...,
[[-2.8507e+00, -2.4392e+00, -2.4583e+00, ..., -3.4415e+00,
-2.7914e+00, -1.8723e+00],
[-2.0976e+00, -8.0685e-01, -1.1150e+00, ..., -1.8284e+00,
-1.8722e+00, -1.5716e+00],
[-1.1153e+00, 8.3929e-01, 7.8456e-01, ..., 7.7670e-02,
-8.0708e-01, -2.8949e+00],
...,
[-2.2447e+00, -1.5389e-01, -1.2724e-01, ..., -1.0304e+00,
-3.8064e-01, 1.3654e-01],
[-1.2329e+00, 8.6899e-01, 1.0822e+00, ..., 1.6162e-01,
2.7582e-01, 3.9214e-01],
[-2.1734e+00, -1.5147e+00, -1.7110e+00, ..., -2.0657e+00,
-1.1752e+00, -5.8831e-01]],
[[-1.6336e+00, -1.6248e+00, -2.8000e+00, ..., -1.2313e+00,
-6.3676e-01, -4.5124e-01],
[-6.9136e-01, -9.3026e-02, -3.1674e+00, ..., -1.9117e+00,
-1.5952e+00, -1.7311e+00],
[-2.0627e+00, -9.8453e-01, -5.1527e+00, ..., -3.2586e+00,
-2.3938e+00, -1.4441e+00],
...,
[-1.3534e+00, 1.8303e-01, -3.3780e+00, ..., -2.5250e+00,
-1.8958e+00, -1.4009e+00],
[-1.6762e+00, -4.5325e-01, -3.0180e+00, ..., -2.3415e+00,
-1.5734e+00, -9.4363e-01],
[-2.3104e+00, -1.3488e+00, -3.0500e+00, ..., -2.8344e+00,
-1.9150e+00, -1.5931e+00]],
[[-1.3069e+00, -1.8368e+00, -1.6519e+00, ..., -1.9337e+00,
-1.5647e+00, -1.1339e+00],
[-1.2430e+00, -1.7091e+00, -1.2201e+00, ..., -1.6394e+00,
-1.3017e+00, -9.8544e-01],
[-1.5598e+00, -1.9016e+00, -1.4703e+00, ..., -2.5434e+00,
-1.9495e+00, -1.3674e+00],
...,
[-2.4552e+00, -3.0666e+00, -2.7924e+00, ..., -3.0802e+00,
-2.4227e+00, -1.8036e+00],
[-1.5953e+00, -2.0933e+00, -1.9960e+00, ..., -2.1825e+00,
-1.7662e+00, -1.3856e+00],
[-1.0450e+00, -1.3535e+00, -1.3679e+00, ..., -1.3701e+00,
-1.1411e+00, -8.3110e-01]]]], device='cuda:0'),
tensor([[[[ 5.0590e-01, 9.6231e-01, -2.1596e+00, ..., 7.8972e-01,
1.0492e+00, 8.4098e-01],
[-1.1210e+00, -1.5057e-01, -4.5737e+00, ..., -1.0793e+00,
1.3621e-01, 6.8003e-01],
[ 8.1614e-01, 5.2715e-01, -4.1103e+00, ..., 5.2564e-01,
8.9823e-01, 9.1645e-01],
...,
[-8.5818e-01, -1.8396e+00, -1.8610e+00, ..., -1.8257e+00,
-1.3544e+00, -9.6024e-01],
[ 7.6266e-01, 2.4936e-02, -1.0535e-01, ..., -1.6425e-02,
7.1452e-01, 8.1157e-01],
[ 1.5360e-01, -5.4133e-01, -4.6435e-01, ..., -1.7796e-01,
4.3331e-01, 3.5835e-01]],
[[-2.6716e+00, -1.2928e+00, -3.9058e+00, ..., -5.7161e+00,
-4.8898e+00, -3.8181e+00],
[-4.9752e-01, 1.7665e+00, -2.0367e+00, ..., -4.7296e+00,
-4.5186e+00, -3.6814e+00],
[ 3.2007e-01, 1.6202e+00, -4.9656e+00, ..., -6.8331e+00,
-6.4888e+00, -5.8680e+00],
...,
[ 2.8308e-01, 3.7086e-01, -4.0205e+00, ..., -4.6785e+00,
-4.2066e+00, -3.4898e+00],
[ 2.3119e+00, 2.3051e+00, -1.0703e-01, ..., -8.4392e-01,
-1.1804e+00, -1.2319e+00],
[ 1.3584e+00, 1.1929e+00, -1.7414e+00, ..., -2.2648e+00,
-2.5315e+00, -2.4347e+00]],
[[-6.0654e-01, -1.7348e+00, -1.2996e+00, ..., -6.9489e-01,
-4.8184e-01, -6.9325e-01],
[-1.4589e+00, -1.8750e+00, -1.0977e+00, ..., -6.3105e-01,
-3.4160e-01, -3.0424e-01],
[-1.3142e+00, -1.1941e+00, -4.1346e-01, ..., -4.0487e-01,
-8.4192e-02, -9.3958e-02],
...,
[-1.5616e-01, -6.2166e-01, -4.7201e-01, ..., -4.9047e-01,
-1.8189e-01, -4.4046e-01],
[-7.0952e-01, -2.0716e-01, -4.3459e-01, ..., -1.9511e-01,
2.5775e-01, 2.2453e-01],
[-8.0725e-01, -1.8608e-01, -3.3264e-01, ..., -4.0917e-01,
-1.9158e-01, -2.2674e-01]],
...,
[[-2.2878e+00, -1.1808e+00, -3.3397e+00, ..., -3.1102e+00,
-2.5780e+00, -1.7596e+00],
[-5.6176e-01, 1.3280e+00, -2.0131e+00, ..., -1.5570e+00,
-1.7353e+00, -1.5035e+00],
[-2.1240e-01, 2.4942e+00, -1.6802e+00, ..., -3.3757e-01,
-9.3281e-02, 7.7121e-02],
...,
[-2.0741e+00, -7.3086e-02, -1.6895e-01, ..., -1.3616e+00,
-3.9774e-01, 1.0334e-01],
[-1.2071e+00, 8.1180e-01, 9.5653e-01, ..., -2.1822e-02,
3.0254e-01, 4.9450e-01],
[-2.1109e+00, -1.4379e+00, -1.6482e+00, ..., -2.4440e+00,
-1.4586e+00, -8.9227e-01]],
[[-1.4203e+00, -1.3524e+00, -2.5850e+00, ..., -1.1627e+00,
-6.7199e-01, -5.3591e-01],
[-7.4027e-01, -2.6712e-01, -3.8076e+00, ..., -1.9781e+00,
-1.7161e+00, -1.8474e+00],
[-2.3947e+00, -1.7285e+00, -6.4238e+00, ..., -4.1731e+00,
-3.6150e+00, -2.8439e+00],
...,
[-1.4131e+00, -3.9087e-02, -3.3771e+00, ..., -1.4506e+00,
-1.0532e+00, -9.5736e-01],
[-1.7112e+00, -6.3156e-01, -3.0519e+00, ..., -1.6972e+00,
-1.1674e+00, -8.4898e-01],
[-2.3233e+00, -1.4799e+00, -3.1182e+00, ..., -2.9237e+00,
-2.1419e+00, -2.0147e+00]],
[[-1.2750e+00, -1.8848e+00, -1.9749e+00, ..., -1.7938e+00,
-1.4871e+00, -1.1080e+00],
[-1.3198e+00, -1.8572e+00, -1.6814e+00, ..., -1.6034e+00,
-1.2996e+00, -1.0176e+00],
[-1.7549e+00, -2.1053e+00, -2.0663e+00, ..., -1.9119e+00,
-1.4749e+00, -1.0809e+00],
...,
[-2.3884e+00, -2.9689e+00, -2.7391e+00, ..., -2.8646e+00,
-2.2920e+00, -1.7571e+00],
[-1.5522e+00, -2.0235e+00, -1.9445e+00, ..., -1.9562e+00,
-1.6163e+00, -1.3015e+00],
[-1.0129e+00, -1.3046e+00, -1.3386e+00, ..., -1.1991e+00,
-1.0224e+00, -7.7711e-01]]],
[[[ 7.6180e-02, 2.2400e-01, -1.4151e+00, ..., -6.2101e-01,
-1.9530e-01, -4.7972e-02],
[-2.1024e+00, -1.6190e+00, -4.5621e+00, ..., -3.7510e+00,
-1.5043e+00, -5.4036e-01],
[ 5.7254e-01, -7.4796e-02, -8.5188e-01, ..., -1.7989e-01,
-2.2718e-01, 2.6595e-01],
...,
[-8.8328e-01, -1.8675e+00, -1.8666e+00, ..., -1.8347e+00,
-1.3474e+00, -9.6356e-01],
[ 7.3778e-01, -2.5451e-02, -1.4948e-01, ..., -3.9428e-02,
6.9471e-01, 8.0065e-01],
[ 1.3324e-01, -5.8406e-01, -4.9820e-01, ..., -1.5507e-01,
4.3854e-01, 3.4704e-01]],
[[-3.3390e+00, -1.8432e+00, -6.2835e+00, ..., -7.1346e+00,
-6.2528e+00, -4.6334e+00],
[-8.4237e-01, 1.4825e+00, -4.3681e+00, ..., -5.4714e+00,
-5.5536e+00, -4.3043e+00],
[-4.1441e-01, 1.0092e+00, -7.4281e+00, ..., -7.9745e+00,
-7.8577e+00, -6.7142e+00],
...,
[ 2.9644e-01, 3.5947e-01, -4.0043e+00, ..., -4.6527e+00,
-4.2080e+00, -3.4857e+00],
[ 2.3205e+00, 2.2929e+00, -1.1268e-01, ..., -8.0981e-01,
-1.1656e+00, -1.2028e+00],
[ 1.3634e+00, 1.1868e+00, -1.7557e+00, ..., -2.2444e+00,
-2.5101e+00, -2.3991e+00]],
[[-1.3495e+00, -2.3561e+00, -2.3296e+00, ..., -1.7537e+00,
-1.2499e+00, -1.3837e+00],
[-1.6876e+00, -1.9796e+00, -1.5369e+00, ..., -1.2327e+00,
-7.2958e-01, -8.1854e-01],
[-2.1339e+00, -1.8799e+00, -9.9872e-01, ..., -1.0921e+00,
-4.7102e-01, -2.0228e-01],
...,
[-1.4939e-01, -6.2479e-01, -4.7758e-01, ..., -4.8752e-01,
-1.7614e-01, -4.3859e-01],
[-7.0874e-01, -1.9947e-01, -4.3320e-01, ..., -1.9283e-01,
2.5972e-01, 2.4078e-01],
[-7.9671e-01, -1.8216e-01, -3.4329e-01, ..., -3.9467e-01,
-1.8338e-01, -2.2725e-01]],
...,
[[-2.0417e+00, -1.1248e+00, -2.3705e+00, ..., -3.2112e+00,
-2.6243e+00, -1.8255e+00],
[-1.7189e+00, 3.5091e-02, -1.3566e+00, ..., -2.5136e+00,
-2.5382e+00, -1.9687e+00],
[-3.7697e-01, 2.3286e+00, 1.0777e+00, ..., 1.2157e-01,
2.5543e-01, 5.1486e-01],
...,
[-2.0823e+00, -1.0272e-01, -2.0803e-01, ..., -1.3615e+00,
-3.9239e-01, 9.3510e-02],
[-1.1876e+00, 8.2664e-01, 9.5421e-01, ..., -2.9045e-02,
3.1784e-01, 4.9250e-01],
[-2.1030e+00, -1.4206e+00, -1.6308e+00, ..., -2.4978e+00,
-1.4982e+00, -9.2071e-01]],
[[-2.4201e+00, -2.8240e+00, -4.1569e+00, ..., -2.3890e+00,
-1.9107e+00, -1.7225e+00],
[-1.7707e+00, -1.6583e+00, -4.8228e+00, ..., -2.9178e+00,
-2.7915e+00, -2.7809e+00],
[-2.8475e+00, -2.5061e+00, -6.6168e+00, ..., -4.4588e+00,
-4.3858e+00, -3.5121e+00],
...,
[-1.3803e+00, -2.7029e-02, -3.3584e+00, ..., -1.4168e+00,
-1.0213e+00, -9.2801e-01],
[-1.7005e+00, -6.2303e-01, -3.0281e+00, ..., -1.7186e+00,
-1.1626e+00, -8.2342e-01],
[-2.3256e+00, -1.4831e+00, -3.1081e+00, ..., -2.9568e+00,
-2.1361e+00, -1.9978e+00]],
[[-1.8046e+00, -2.5369e+00, -2.4636e+00, ..., -2.6951e+00,
-2.2099e+00, -1.6943e+00],
[-1.7943e+00, -2.5077e+00, -2.1575e+00, ..., -2.4040e+00,
-1.9046e+00, -1.5451e+00],
[-2.2401e+00, -2.8076e+00, -2.5286e+00, ..., -2.7505e+00,
-2.1443e+00, -1.6344e+00],
...,
[-2.3866e+00, -2.9618e+00, -2.7314e+00, ..., -2.8754e+00,
-2.3036e+00, -1.7603e+00],
[-1.5461e+00, -2.0119e+00, -1.9369e+00, ..., -1.9403e+00,
-1.6086e+00, -1.2943e+00],
[-1.0039e+00, -1.2969e+00, -1.3287e+00, ..., -1.1855e+00,
-1.0171e+00, -7.7427e-01]]],
[[[ 2.8494e-01, 5.6876e-01, 6.9960e-01, ..., 5.1285e-01,
1.0664e+00, 6.4127e-01],
[-2.2044e+00, -1.5034e+00, -3.1902e+00, ..., -2.6000e+00,
-7.9494e-01, -1.3356e-01],
[-1.9055e-01, -1.0090e+00, -2.5078e-01, ..., -2.0278e-01,
5.5643e-01, 6.9292e-01],
...,
[-1.9226e-01, -1.1668e+00, -2.6340e+00, ..., -1.5009e+00,
-1.4196e+00, -8.8972e-01],
[ 1.4193e+00, 8.3798e-01, -9.1328e-01, ..., -2.5878e-01,
1.0382e-01, 3.7124e-01],
[ 8.3755e-01, 2.8738e-01, -6.7001e-01, ..., -1.6033e-01,
2.3066e-01, 3.7915e-01]],
[[-1.6818e+00, -1.1278e+00, -4.1140e+00, ..., -4.8332e+00,
-4.4758e+00, -3.4127e+00],
[ 5.9591e-01, 1.9870e+00, -2.1967e+00, ..., -3.5275e+00,
-3.6990e+00, -2.9936e+00],
[ 1.3533e+00, 1.7541e+00, -4.8864e+00, ..., -5.4963e+00,
-5.3604e+00, -4.9355e+00],
...,
[-1.2641e+00, 1.7822e-01, -6.6999e+00, ..., -7.6110e+00,
-6.9030e+00, -5.4030e+00],
[ 1.8188e+00, 2.8413e+00, -1.0378e+00, ..., -2.6149e+00,
-2.9634e+00, -2.3730e+00],
[ 1.2258e+00, 1.8241e+00, -2.2888e+00, ..., -3.2741e+00,
-3.6164e+00, -3.1273e+00]],
[[-1.0335e-01, -1.1403e+00, -1.3316e+00, ..., -4.7319e-01,
-1.6420e-01, -4.7389e-01],
[-5.9806e-01, -8.4734e-01, -7.7291e-01, ..., -2.5751e-01,
3.3185e-02, -7.6596e-02],
[-5.6718e-01, -3.4968e-01, -4.0244e-01, ..., -3.7186e-01,
8.0448e-02, 8.0954e-02],
...,
[-1.3739e+00, -1.7630e+00, -7.7939e-01, ..., -8.8365e-01,
-7.5942e-01, -1.0081e+00],
[-1.7334e+00, -1.1635e+00, -8.3518e-01, ..., -5.6073e-01,
-2.9569e-01, -3.6302e-01],
[-1.8877e+00, -1.2882e+00, -8.9473e-01, ..., -9.4243e-01,
-8.1467e-01, -7.9183e-01]],
...,
[[-3.0653e+00, -2.5448e+00, -2.8185e+00, ..., -3.4241e+00,
-2.6073e+00, -1.7515e+00],
[-2.3024e+00, -8.6910e-01, -1.6129e+00, ..., -2.6723e+00,
-2.3232e+00, -1.7949e+00],
[-1.0448e+00, 1.1877e+00, 6.0536e-01, ..., -4.8575e-01,
-4.0785e-02, 1.2405e-01],
...,
[-1.8307e+00, 1.2868e+00, -8.6395e-03, ..., -1.1133e+00,
-5.3963e-01, -1.0826e-02],
[-7.8625e-01, 1.9319e+00, 1.6024e+00, ..., 4.3584e-01,
4.0276e-01, 6.7913e-01],
[-2.2028e+00, -1.1366e+00, -1.0061e+00, ..., -1.8802e+00,
-9.8360e-01, -3.0520e-01]],
[[-1.5793e+00, -1.5723e+00, -2.9258e+00, ..., -1.3145e+00,
-8.9512e-01, -8.6629e-01],
[-5.6719e-01, 1.1924e-01, -3.2540e+00, ..., -1.3928e+00,
-1.2331e+00, -1.5657e+00],
[-2.0669e+00, -8.5556e-01, -5.3729e+00, ..., -3.1730e+00,
-2.6991e+00, -2.1985e+00],
...,
[-2.4369e+00, -1.5479e+00, -5.5144e+00, ..., -3.0953e+00,
-2.7204e+00, -2.1614e+00],
[-2.1859e+00, -1.4346e+00, -4.2852e+00, ..., -2.6933e+00,
-2.3083e+00, -1.6360e+00],
[-2.3667e+00, -1.7403e+00, -3.3239e+00, ..., -2.9764e+00,
-2.4235e+00, -2.1225e+00]],
[[-1.3605e+00, -1.9234e+00, -1.7140e+00, ..., -1.7683e+00,
-1.4673e+00, -1.1214e+00],
[-1.2869e+00, -1.7732e+00, -1.2350e+00, ..., -1.3940e+00,
-1.1049e+00, -8.8087e-01],
[-1.6100e+00, -1.9625e+00, -1.4808e+00, ..., -1.6169e+00,
-1.2139e+00, -8.9824e-01],
...,
[-3.1846e+00, -3.9734e+00, -3.7532e+00, ..., -3.9471e+00,
-3.1700e+00, -2.4410e+00],
[-2.1230e+00, -2.7372e+00, -2.6972e+00, ..., -2.7672e+00,
-2.2887e+00, -1.8598e+00],
[-1.5444e+00, -1.8748e+00, -1.9287e+00, ..., -1.8665e+00,
-1.5189e+00, -1.1615e+00]]],
[[[ 2.1899e-01, 4.6988e-01, 6.2173e-01, ..., -3.3021e-01,
-5.4819e-02, 2.0454e-02],
[-1.9813e+00, -1.3703e+00, -2.8366e+00, ..., -3.2249e+00,
-1.2604e+00, -4.3333e-01],
[-1.9453e-01, -9.6611e-01, -1.6152e-03, ..., -1.9566e-01,
-3.5308e-01, 1.6980e-01],
...,
[-1.0025e+00, -2.1519e+00, -2.0844e+00, ..., -1.8225e+00,
-1.8722e+00, -1.2605e+00],
[ 7.6251e-01, -1.2327e-01, -1.4372e-01, ..., 6.3544e-02,
2.5979e-01, 4.2833e-01],
[ 1.7434e-01, -5.8299e-01, -4.1310e-01, ..., 2.3754e-01,
4.3743e-01, 4.8568e-01]],
[[-1.6539e+00, -1.4567e+00, -4.2658e+00, ..., -6.8516e+00,
-6.0696e+00, -4.5190e+00],
[ 4.0943e-01, 1.3067e+00, -2.5785e+00, ..., -5.4452e+00,
-5.5548e+00, -4.2886e+00],
[ 1.1022e+00, 1.0686e+00, -5.0205e+00, ..., -7.8664e+00,
-7.8179e+00, -6.6168e+00],
...,
[ 4.1130e-01, 3.6821e-01, -4.0827e+00, ..., -7.1600e+00,
-6.6865e+00, -5.1771e+00],
[ 2.5981e+00, 2.5391e+00, 1.8715e-01, ..., -1.9865e+00,
-2.5559e+00, -2.0255e+00],
[ 1.5444e+00, 1.3502e+00, -1.5980e+00, ..., -2.9451e+00,
-3.4496e+00, -3.0353e+00]],
[[-1.0856e-01, -1.0651e+00, -1.3308e+00, ..., -1.3993e+00,
-1.0460e+00, -1.3418e+00],
[-4.3471e-01, -7.2657e-01, -8.3132e-01, ..., -9.8095e-01,
-6.0557e-01, -9.0162e-01],
[-2.6344e-01, -1.7959e-01, -4.2436e-01, ..., -1.0977e+00,
-5.8126e-01, -4.7038e-01],
...,
[-4.4202e-02, -6.0157e-01, -6.0129e-01, ..., -1.0287e+00,
-8.5349e-01, -1.2270e+00],
[-7.1619e-01, -2.2251e-01, -5.7315e-01, ..., -5.3909e-01,
-2.3417e-01, -4.1305e-01],
[-8.9870e-01, -2.7629e-01, -4.8087e-01, ..., -4.8148e-01,
-4.2439e-01, -6.0159e-01]],
...,
[[-2.7411e+00, -2.4339e+00, -2.4050e+00, ..., -3.0874e+00,
-2.5232e+00, -1.8284e+00],
[-2.1021e+00, -9.1976e-01, -1.1327e+00, ..., -2.4322e+00,
-2.5159e+00, -2.0105e+00],
[-1.1915e+00, 6.8151e-01, 7.1452e-01, ..., -1.0093e-01,
1.6597e-02, 2.7656e-01],
...,
[-2.3155e+00, -2.2902e-01, -1.6018e-01, ..., -9.9914e-01,
-2.5366e-01, 3.1901e-01],
[-1.2608e+00, 8.1697e-01, 1.0572e+00, ..., 3.3982e-01,
4.5505e-01, 8.1332e-01],
[-2.1882e+00, -1.5351e+00, -1.7275e+00, ..., -2.2616e+00,
-1.2038e+00, -3.6181e-01]],
[[-1.6589e+00, -1.6717e+00, -2.7238e+00, ..., -2.2166e+00,
-1.9013e+00, -1.7973e+00],
[-7.5541e-01, -1.4566e-01, -3.0228e+00, ..., -2.8541e+00,
-2.8554e+00, -2.8869e+00],
[-2.0823e+00, -1.0089e+00, -5.0086e+00, ..., -4.4566e+00,
-4.5036e+00, -3.7203e+00],
...,
[-1.3314e+00, 2.1602e-01, -3.3689e+00, ..., -2.8814e+00,
-2.7482e+00, -2.2856e+00],
[-1.6727e+00, -4.3142e-01, -3.0091e+00, ..., -2.4656e+00,
-2.2286e+00, -1.6387e+00],
[-2.3322e+00, -1.3446e+00, -3.0327e+00, ..., -2.8546e+00,
-2.4061e+00, -2.1967e+00]],
[[-1.2950e+00, -1.8111e+00, -1.6694e+00, ..., -2.5678e+00,
-2.1474e+00, -1.6702e+00],
[-1.2339e+00, -1.6966e+00, -1.2517e+00, ..., -2.3961e+00,
-1.9284e+00, -1.5811e+00],
[-1.5502e+00, -1.8996e+00, -1.4961e+00, ..., -2.8023e+00,
-2.2377e+00, -1.7362e+00],
...,
[-2.4586e+00, -3.0728e+00, -2.8035e+00, ..., -4.0063e+00,
-3.2573e+00, -2.5511e+00],
[-1.5954e+00, -2.0997e+00, -1.9964e+00, ..., -2.7719e+00,
-2.3221e+00, -1.9222e+00],
[-1.0429e+00, -1.3527e+00, -1.3664e+00, ..., -1.8306e+00,
-1.5270e+00, -1.1823e+00]]],
[[[ 2.8214e-01, 4.7772e-01, 6.2872e-01, ..., 4.5259e-01,
1.0326e+00, 6.3382e-01],
[-2.1609e+00, -1.6145e+00, -3.2159e+00, ..., -2.8253e+00,
-9.0338e-01, -1.8796e-01],
[-2.4492e-01, -1.1054e+00, -2.8454e-01, ..., -1.9507e-01,
5.4901e-01, 6.2649e-01],
...,
[-1.9666e-01, -1.1472e+00, -2.6610e+00, ..., -1.5310e+00,
-1.4088e+00, -8.6373e-01],
[ 1.4345e+00, 8.5002e-01, -9.6997e-01, ..., -2.6850e-01,
1.3721e-01, 3.8822e-01],
[ 8.6875e-01, 3.2882e-01, -6.7262e-01, ..., -1.7181e-01,
2.3558e-01, 3.8092e-01]],
[[-1.6520e+00, -1.1732e+00, -4.1549e+00, ..., -4.9936e+00,
-4.5744e+00, -3.5007e+00],
[ 5.3465e-01, 1.7644e+00, -2.3452e+00, ..., -3.4159e+00,
-3.6518e+00, -3.0122e+00],
[ 1.2183e+00, 1.4563e+00, -4.9504e+00, ..., -5.4263e+00,
-5.3918e+00, -5.0693e+00],
...,
[-1.3118e+00, 1.2196e-01, -6.6749e+00, ..., -7.5939e+00,
-6.9059e+00, -5.4118e+00],
[ 1.7861e+00, 2.8193e+00, -9.9116e-01, ..., -2.6302e+00,
-2.9970e+00, -2.3957e+00],
[ 1.2071e+00, 1.8111e+00, -2.2519e+00, ..., -3.2798e+00,
-3.6304e+00, -3.1197e+00]],
[[-8.7640e-02, -1.0453e+00, -1.1978e+00, ..., -5.3230e-01,
-1.9384e-01, -5.2499e-01],
[-5.0953e-01, -7.3853e-01, -6.7525e-01, ..., -3.1948e-01,
1.4142e-02, -7.9561e-02],
[-4.6290e-01, -2.8518e-01, -3.9355e-01, ..., -3.7589e-01,
8.5798e-02, 9.5694e-02],
...,
[-1.3606e+00, -1.7481e+00, -7.7107e-01, ..., -8.6436e-01,
-7.5994e-01, -1.0295e+00],
[-1.7108e+00, -1.1639e+00, -8.6499e-01, ..., -5.6402e-01,
-3.0151e-01, -3.7970e-01],
[-1.8619e+00, -1.2689e+00, -9.0203e-01, ..., -9.3813e-01,
-8.0823e-01, -7.9949e-01]],
...,
[[-2.9486e+00, -2.4821e+00, -2.7737e+00, ..., -3.6755e+00,
-2.7767e+00, -1.8519e+00],
[-2.2662e+00, -9.4769e-01, -1.6606e+00, ..., -2.7523e+00,
-2.4480e+00, -1.9074e+00],
[-1.0716e+00, 1.0140e+00, 5.2990e-01, ..., -4.5677e-01,
2.6480e-02, 1.9151e-01],
...,
[-1.8520e+00, 1.2092e+00, -5.8822e-02, ..., -1.0313e+00,
-5.1201e-01, 1.6962e-02],
[-8.1064e-01, 1.8928e+00, 1.6111e+00, ..., 4.9477e-01,
4.2932e-01, 6.8750e-01],
[-2.2372e+00, -1.1831e+00, -1.0067e+00, ..., -1.8207e+00,
-9.5282e-01, -2.9509e-01]],
[[-1.6060e+00, -1.6043e+00, -2.8052e+00, ..., -1.1900e+00,
-7.3915e-01, -7.8004e-01],
[-6.5622e-01, 1.2772e-03, -3.0827e+00, ..., -1.2641e+00,
-1.1479e+00, -1.5685e+00],
[-2.1268e+00, -9.5299e-01, -5.1024e+00, ..., -3.1972e+00,
-2.7274e+00, -2.2154e+00],
...,
[-2.4524e+00, -1.5747e+00, -5.4432e+00, ..., -3.1090e+00,
-2.7287e+00, -2.1815e+00],
[-2.2099e+00, -1.4703e+00, -4.2492e+00, ..., -2.6855e+00,
-2.3065e+00, -1.6376e+00],
[-2.3994e+00, -1.7821e+00, -3.3184e+00, ..., -2.9421e+00,
-2.4319e+00, -2.0988e+00]],
[[-1.3292e+00, -1.8854e+00, -1.7163e+00, ..., -1.8516e+00,
-1.5263e+00, -1.1693e+00],
[-1.2569e+00, -1.7509e+00, -1.2806e+00, ..., -1.4115e+00,
-1.1166e+00, -8.9837e-01],
[-1.5760e+00, -1.9463e+00, -1.5465e+00, ..., -1.6180e+00,
-1.2073e+00, -8.9671e-01],
...,
[-3.1637e+00, -3.9566e+00, -3.7545e+00, ..., -3.9393e+00,
-3.1585e+00, -2.4332e+00],
[-2.0998e+00, -2.7175e+00, -2.6908e+00, ..., -2.7833e+00,
-2.2938e+00, -1.8651e+00],
[-1.5259e+00, -1.8514e+00, -1.9150e+00, ..., -1.8871e+00,
-1.5298e+00, -1.1743e+00]]]], device='cuda:0'),
tensor([[[[ 3.7218e-01, 8.1193e-01, 4.9780e-01, ..., 4.5260e-01,
1.0038e+00, 6.3787e-01],
[-1.2263e+00, -3.7093e-01, -1.7879e+00, ..., -2.6091e+00,
-7.8504e-01, -9.1537e-02],
[ 7.8129e-01, 6.1121e-01, 3.8379e-01, ..., -2.8052e-01,
4.6445e-01, 6.2657e-01],
...,
[-2.0416e-02, -6.8893e-01, -1.5500e+00, ..., -2.0342e+00,
-1.4160e+00, -9.4384e-01],
[ 1.4415e+00, 1.2323e+00, 6.0788e-01, ..., -3.4107e-01,
5.1173e-01, 6.9715e-01],
[ 1.0815e+00, 9.7582e-01, 6.4002e-01, ..., -3.7364e-01,
2.9146e-01, 2.7511e-01]],
[[-2.7243e+00, -1.6643e+00, -4.7539e+00, ..., -4.8457e+00,
-4.4249e+00, -3.3736e+00],
[-6.9324e-01, 1.2062e+00, -3.1276e+00, ..., -3.3753e+00,
-3.5389e+00, -2.9097e+00],
[-4.5547e-03, 9.7986e-01, -5.7585e+00, ..., -5.3496e+00,
-5.2626e+00, -4.9327e+00],
...,
[-7.6782e-01, 3.1143e-01, -4.6365e+00, ..., -4.5446e+00,
-4.1026e+00, -3.4324e+00],
[ 2.0904e+00, 2.7830e+00, 2.8279e-01, ..., -8.4375e-01,
-1.1798e+00, -1.2319e+00],
[ 1.2527e+00, 1.5591e+00, -1.5553e+00, ..., -2.2219e+00,
-2.4676e+00, -2.3615e+00]],
[[-6.1633e-01, -1.6912e+00, -1.6017e+00, ..., -5.9844e-01,
-1.9650e-01, -4.6126e-01],
[-1.2709e+00, -1.7029e+00, -1.1394e+00, ..., -3.4539e-01,
6.9536e-02, 3.4882e-02],
[-1.1674e+00, -1.1967e+00, -6.7842e-01, ..., -2.8016e-01,
2.4094e-01, 2.8003e-01],
...,
[-1.0096e+00, -1.5907e+00, -7.4539e-01, ..., -5.0057e-01,
-1.6926e-01, -3.5607e-01],
[-1.3273e+00, -9.8141e-01, -6.4250e-01, ..., -3.1931e-01,
1.5259e-01, 2.1196e-01],
[-1.4105e+00, -8.0953e-01, -3.2845e-01, ..., -5.0804e-01,
-2.7869e-01, -2.5432e-01]],
...,
[[-2.0370e+00, -1.1110e+00, -2.3369e+00, ..., -3.6709e+00,
-2.8369e+00, -1.8406e+00],
[-5.5334e-01, 1.0548e+00, -7.3454e-01, ..., -2.6744e+00,
-2.4363e+00, -1.8852e+00],
[-2.4799e-01, 2.2122e+00, 6.4341e-01, ..., -5.5757e-01,
-4.8776e-02, 1.6883e-01],
...,
[-1.5263e+00, 1.4057e+00, 1.2313e-02, ..., -1.5161e+00,
-6.9818e-01, -1.3920e-01],
[-5.6516e-01, 1.6440e+00, 9.1568e-01, ..., -1.1334e-01,
1.1263e-01, 3.2934e-01],
[-1.7594e+00, -1.1960e+00, -1.6510e+00, ..., -2.3422e+00,
-1.4455e+00, -9.1187e-01]],
[[-1.3948e+00, -1.4249e+00, -2.8676e+00, ..., -1.1321e+00,
-7.2043e-01, -7.4736e-01],
[-7.5779e-01, -4.1079e-01, -3.7622e+00, ..., -1.3498e+00,
-1.2114e+00, -1.5801e+00],
[-2.1640e+00, -1.6831e+00, -6.1715e+00, ..., -3.2699e+00,
-2.7921e+00, -2.2336e+00],
...,
[-1.7246e+00, -7.1719e-01, -4.7432e+00, ..., -1.4237e+00,
-1.0184e+00, -8.8559e-01],
[-1.6781e+00, -8.3593e-01, -3.8386e+00, ..., -1.7018e+00,
-1.1428e+00, -7.6368e-01],
[-2.1047e+00, -1.3968e+00, -3.1170e+00, ..., -2.8001e+00,
-2.0234e+00, -1.8456e+00]],
[[-1.2033e+00, -1.8208e+00, -1.7092e+00, ..., -1.8054e+00,
-1.4978e+00, -1.1389e+00],
[-1.2154e+00, -1.7507e+00, -1.4159e+00, ..., -1.3999e+00,
-1.1094e+00, -8.8500e-01],
[-1.5677e+00, -1.9496e+00, -1.6499e+00, ..., -1.6012e+00,
-1.1997e+00, -8.8833e-01],
...,
[-2.4300e+00, -3.0759e+00, -2.8974e+00, ..., -2.8729e+00,
-2.2975e+00, -1.7667e+00],
[-1.6238e+00, -2.0968e+00, -2.0481e+00, ..., -1.9818e+00,
-1.6308e+00, -1.3141e+00],
[-1.0969e+00, -1.3403e+00, -1.4181e+00, ..., -1.2281e+00,
-1.0372e+00, -7.9682e-01]]],
[[[ 3.0501e-01, 6.4776e-01, 6.6541e-01, ..., -1.0529e+00,
-7.9915e-01, -4.2538e-01],
[-1.4968e+00, -7.0297e-01, -1.9952e+00, ..., -4.5286e+00,
-2.1751e+00, -1.0164e+00],
[ 3.3972e-01, -1.0607e-01, 3.4092e-01, ..., -7.1404e-01,
-1.1054e+00, -1.5254e-01],
...,
[-5.2405e-01, -1.4574e+00, -2.5294e+00, ..., -2.0378e+00,
-2.0650e+00, -1.2017e+00],
[ 1.0894e+00, 4.8247e-01, -9.4652e-01, ..., -4.9903e-01,
-2.0399e-01, 1.9512e-01],
[ 6.1411e-01, 1.5884e-01, -9.2397e-01, ..., -3.4193e-01,
5.2553e-02, 2.8584e-01]],
[[-2.0703e+00, -1.4416e+00, -4.3906e+00, ..., -7.7279e+00,
-6.9031e+00, -5.0405e+00],
[-2.5557e-02, 1.3405e+00, -2.7817e+00, ..., -6.2085e+00,
-6.4372e+00, -4.8428e+00],
[ 7.1351e-01, 1.1359e+00, -5.2557e+00, ..., -8.8952e+00,
-8.9425e+00, -7.2444e+00],
...,
[-1.2372e-01, 3.3487e-01, -3.9466e+00, ..., -8.3006e+00,
-7.9828e+00, -6.0412e+00],
[ 2.3255e+00, 2.6225e+00, 7.3970e-01, ..., -2.6207e+00,
-3.4498e+00, -2.5249e+00],
[ 1.3713e+00, 1.4122e+00, -1.1876e+00, ..., -3.2778e+00,
-4.0281e+00, -3.3248e+00]],
[[-3.5050e-01, -1.4061e+00, -1.4686e+00, ..., -1.8372e+00,
-1.3956e+00, -1.6685e+00],
[-8.8342e-01, -1.2924e+00, -1.0555e+00, ..., -1.2691e+00,
-8.6113e-01, -1.2430e+00],
[-6.6988e-01, -6.3346e-01, -4.5858e-01, ..., -1.4754e+00,
-8.6106e-01, -6.1563e-01],
...,
[-4.9246e-01, -1.0374e+00, -8.1159e-01, ..., -1.0610e+00,
-9.4248e-01, -1.3684e+00],
[-9.6605e-01, -5.4676e-01, -7.4557e-01, ..., -5.3123e-01,
-2.2330e-01, -4.2978e-01],
[-1.0902e+00, -4.9635e-01, -4.8553e-01, ..., -9.8784e-01,
-9.5425e-01, -1.0926e+00]],
...,
[[-2.4528e+00, -1.7943e+00, -2.3064e+00, ..., -3.0362e+00,
-2.4774e+00, -1.8632e+00],
[-1.2875e+00, 1.9732e-01, -6.5577e-01, ..., -3.0282e+00,
-3.0159e+00, -2.3409e+00],
[-6.8832e-01, 1.4581e+00, 7.3947e-01, ..., -8.7188e-02,
-8.9902e-02, 2.2934e-01],
...,
[-1.8371e+00, 6.0838e-01, -7.0033e-02, ..., -1.0167e+00,
-3.8137e-01, 1.6013e-01],
[-8.3666e-01, 1.3033e+00, 9.1137e-01, ..., 7.7883e-01,
6.6090e-01, 1.0008e+00],
[-1.8766e+00, -1.2514e+00, -1.9434e+00, ..., -1.8847e+00,
-8.6362e-01, -8.3748e-02]],
[[-1.4822e+00, -1.4555e+00, -2.7258e+00, ..., -2.8747e+00,
-2.5758e+00, -2.4413e+00],
[-7.1289e-01, -2.1779e-01, -3.3563e+00, ..., -3.4559e+00,
-3.5261e+00, -3.4475e+00],
[-2.0946e+00, -1.2925e+00, -5.5572e+00, ..., -4.7806e+00,
-5.0842e+00, -4.2006e+00],
...,
[-1.4850e+00, -2.1880e-01, -3.6954e+00, ..., -3.0592e+00,
-3.1307e+00, -2.6001e+00],
[-1.6340e+00, -6.0507e-01, -3.0327e+00, ..., -2.6385e+00,
-2.6666e+00, -2.0036e+00],
[-2.1853e+00, -1.3394e+00, -2.7288e+00, ..., -2.9924e+00,
-2.7466e+00, -2.4996e+00]],
[[-1.2032e+00, -1.7398e+00, -1.5668e+00, ..., -3.0956e+00,
-2.5782e+00, -2.0087e+00],
[-1.1796e+00, -1.6512e+00, -1.2177e+00, ..., -2.9822e+00,
-2.3796e+00, -1.9536e+00],
[-1.5142e+00, -1.8494e+00, -1.4523e+00, ..., -3.5177e+00,
-2.8145e+00, -2.2036e+00],
...,
[-2.3940e+00, -2.9805e+00, -2.8704e+00, ..., -4.6458e+00,
-3.7683e+00, -2.9702e+00],
[-1.5794e+00, -2.0490e+00, -2.1649e+00, ..., -3.1740e+00,
-2.6490e+00, -2.2041e+00],
[-1.0443e+00, -1.3136e+00, -1.5088e+00, ..., -2.1383e+00,
-1.7397e+00, -1.3652e+00]]],
[[[ 2.1351e-01, 4.9076e-01, 6.4294e-01, ..., -3.2828e-01,
-3.0697e-02, 2.0268e-02],
[-2.0475e+00, -1.3644e+00, -2.8406e+00, ..., -3.4374e+00,
-1.3501e+00, -4.6455e-01],
[-1.9047e-01, -9.1833e-01, 1.5986e-02, ..., -1.8781e-01,
-3.1724e-01, 1.7878e-01],
...,
[-9.9579e-01, -2.1988e+00, -2.1221e+00, ..., -1.7032e+00,
-1.6851e+00, -1.0775e+00],
[ 8.4102e-01, -1.0505e-01, -1.3857e-01, ..., 4.6253e-02,
2.7189e-01, 4.6936e-01],
[ 2.3695e-01, -5.7775e-01, -4.2693e-01, ..., 2.0710e-01,
4.2497e-01, 4.8644e-01]],
[[-1.6606e+00, -1.4345e+00, -4.2788e+00, ..., -7.0436e+00,
-6.2402e+00, -4.6279e+00],
[ 4.5061e-01, 1.3667e+00, -2.5810e+00, ..., -5.4129e+00,
-5.5646e+00, -4.2887e+00],
[ 1.1454e+00, 1.0824e+00, -5.0858e+00, ..., -7.9099e+00,
-7.9007e+00, -6.7259e+00],
...,
[ 3.9378e-01, 3.9744e-01, -4.0876e+00, ..., -7.0980e+00,
-6.7045e+00, -5.2554e+00],
[ 2.5697e+00, 2.5532e+00, 1.8860e-01, ..., -2.2640e+00,
-2.8539e+00, -2.2947e+00],
[ 1.5350e+00, 1.3561e+00, -1.6043e+00, ..., -3.0095e+00,
-3.5135e+00, -3.0446e+00]],
[[-1.2596e-01, -1.1263e+00, -1.3649e+00, ..., -1.4884e+00,
-1.1020e+00, -1.4048e+00],
[-4.7541e-01, -7.8103e-01, -8.4508e-01, ..., -1.0589e+00,
-6.3969e-01, -9.0335e-01],
[-2.8928e-01, -1.7527e-01, -3.8955e-01, ..., -1.0928e+00,
-5.4917e-01, -4.3436e-01],
...,
[-6.2070e-02, -6.0970e-01, -6.1160e-01, ..., -9.9544e-01,
-8.4982e-01, -1.2025e+00],
[-7.1026e-01, -2.3390e-01, -5.8793e-01, ..., -5.5750e-01,
-2.7006e-01, -4.3143e-01],
[-8.8179e-01, -2.6497e-01, -4.7866e-01, ..., -4.8473e-01,
-4.3481e-01, -5.8755e-01]],
...,
[[-2.8193e+00, -2.4928e+00, -2.4703e+00, ..., -3.4055e+00,
-2.7729e+00, -1.9744e+00],
[-2.1004e+00, -8.9918e-01, -1.1555e+00, ..., -2.5348e+00,
-2.6365e+00, -2.1456e+00],
[-1.1567e+00, 6.8885e-01, 6.9536e-01, ..., -8.6787e-02,
5.2981e-02, 3.1954e-01],
...,
[-2.3329e+00, -1.7091e-01, -1.1643e-01, ..., -8.2461e-01,
-2.3480e-01, 2.3913e-01],
[-1.3106e+00, 8.8338e-01, 1.1233e+00, ..., 3.7455e-01,
4.4038e-01, 7.5400e-01],
[-2.2234e+00, -1.5215e+00, -1.6824e+00, ..., -2.0202e+00,
-1.0691e+00, -3.1729e-01]],
[[-1.6472e+00, -1.6397e+00, -2.7393e+00, ..., -2.0969e+00,
-1.7703e+00, -1.6959e+00],
[-7.2094e-01, -1.2342e-01, -3.0702e+00, ..., -2.7416e+00,
-2.7857e+00, -2.8917e+00],
[-2.0803e+00, -9.9623e-01, -5.0196e+00, ..., -4.4982e+00,
-4.5650e+00, -3.7814e+00],
...,
[-1.3473e+00, 1.9902e-01, -3.3893e+00, ..., -2.9520e+00,
-2.8713e+00, -2.4003e+00],
[-1.6699e+00, -4.5307e-01, -3.0389e+00, ..., -2.4917e+00,
-2.3471e+00, -1.7507e+00],
[-2.3307e+00, -1.3535e+00, -3.0607e+00, ..., -2.7842e+00,
-2.4038e+00, -2.1569e+00]],
[[-1.3024e+00, -1.8269e+00, -1.6703e+00, ..., -2.6322e+00,
-2.1985e+00, -1.7065e+00],
[-1.2379e+00, -1.7033e+00, -1.2434e+00, ..., -2.3917e+00,
-1.9269e+00, -1.5837e+00],
[-1.5455e+00, -1.8928e+00, -1.4976e+00, ..., -2.7747e+00,
-2.2112e+00, -1.7175e+00],
...,
[-2.4634e+00, -3.0798e+00, -2.7906e+00, ..., -3.9047e+00,
-3.1675e+00, -2.4890e+00],
[-1.5948e+00, -2.1043e+00, -1.9947e+00, ..., -2.7149e+00,
-2.2684e+00, -1.8716e+00],
[-1.0441e+00, -1.3607e+00, -1.3729e+00, ..., -1.8191e+00,
-1.5064e+00, -1.1686e+00]]],
[[[ 4.7523e-01, 9.4179e-01, 8.6021e-01, ..., -3.6902e-01,
-9.1881e-02, -6.0322e-03],
[-1.2031e+00, -3.2046e-01, -1.4453e+00, ..., -3.4812e+00,
-1.4051e+00, -4.9318e-01],
[ 6.1633e-01, 4.7186e-01, 1.0201e+00, ..., -2.2027e-01,
-3.6383e-01, 1.6316e-01],
...,
[-1.5898e-01, -8.5534e-01, -1.5001e+00, ..., -1.7110e+00,
-1.7261e+00, -1.1164e+00],
[ 1.1058e+00, 8.1101e-01, 2.9879e-01, ..., 4.4085e-02,
2.3649e-01, 4.4100e-01],
[ 7.4496e-01, 5.5620e-01, 2.1833e-01, ..., 1.9517e-01,
4.0899e-01, 4.9025e-01]],
[[-2.3089e+00, -1.1261e+00, -4.1212e+00, ..., -7.0917e+00,
-6.2712e+00, -4.6617e+00],
[-3.0028e-01, 1.6645e+00, -2.5336e+00, ..., -5.4664e+00,
-5.6360e+00, -4.3563e+00],
[ 4.0182e-01, 1.4436e+00, -4.8224e+00, ..., -7.9744e+00,
-7.9975e+00, -6.8087e+00],
...,
[-5.2087e-01, 5.7391e-01, -4.0988e+00, ..., -7.2195e+00,
-6.7882e+00, -5.2861e+00],
[ 2.0812e+00, 2.7804e+00, 4.1536e-01, ..., -2.3313e+00,
-2.8860e+00, -2.3087e+00],
[ 1.2180e+00, 1.5274e+00, -1.4511e+00, ..., -3.0382e+00,
-3.5383e+00, -3.0597e+00]],
[[-5.7691e-01, -1.5679e+00, -1.4065e+00, ..., -1.5063e+00,
-1.1247e+00, -1.4274e+00],
[-1.2852e+00, -1.6373e+00, -1.0266e+00, ..., -1.0555e+00,
-6.5411e-01, -9.3460e-01],
[-1.1353e+00, -9.7622e-01, -4.4936e-01, ..., -1.1212e+00,
-6.0181e-01, -4.7337e-01],
...,
[-8.8944e-01, -1.3412e+00, -5.2786e-01, ..., -9.8195e-01,
-8.4119e-01, -1.2048e+00],
[-1.2575e+00, -8.6904e-01, -5.5178e-01, ..., -5.4883e-01,
-2.7535e-01, -4.4814e-01],
[-1.3659e+00, -7.7040e-01, -3.5130e-01, ..., -4.9784e-01,
-4.6601e-01, -6.3040e-01]],
...,
[[-2.1800e+00, -1.2324e+00, -2.5328e+00, ..., -3.3949e+00,
-2.7505e+00, -1.9655e+00],
[-6.7300e-01, 9.6278e-01, -9.0860e-01, ..., -2.6103e+00,
-2.6941e+00, -2.1716e+00],
[-2.4883e-01, 2.1828e+00, 7.0573e-01, ..., -9.0875e-02,
3.5649e-02, 3.1410e-01],
...,
[-1.5939e+00, 1.2161e+00, -1.7116e-01, ..., -8.6132e-01,
-2.5336e-01, 2.2805e-01],
[-5.7282e-01, 1.5564e+00, 8.1812e-01, ..., 3.8332e-01,
4.5135e-01, 7.6312e-01],
[-1.6605e+00, -1.0965e+00, -1.5871e+00, ..., -2.0046e+00,
-1.0305e+00, -2.7536e-01]],
[[-1.2942e+00, -1.2151e+00, -2.6101e+00, ..., -2.1466e+00,
-1.8379e+00, -1.7730e+00],
[-6.3484e-01, -1.5302e-01, -3.3399e+00, ..., -2.7906e+00,
-2.8603e+00, -2.9612e+00],
[-2.1040e+00, -1.4480e+00, -5.5102e+00, ..., -4.5274e+00,
-4.6329e+00, -3.8362e+00],
...,
[-1.6486e+00, -5.1641e-01, -4.1068e+00, ..., -3.0062e+00,
-2.8921e+00, -2.4120e+00],
[-1.6646e+00, -7.2211e-01, -3.4409e+00, ..., -2.5339e+00,
-2.3759e+00, -1.7837e+00],
[-2.1157e+00, -1.3635e+00, -2.9665e+00, ..., -2.8084e+00,
-2.4418e+00, -2.1980e+00]],
[[-1.1505e+00, -1.7004e+00, -1.5270e+00, ..., -2.6632e+00,
-2.2229e+00, -1.7199e+00],
[-1.1581e+00, -1.6220e+00, -1.2107e+00, ..., -2.4352e+00,
-1.9615e+00, -1.6111e+00],
[-1.5024e+00, -1.8052e+00, -1.6459e+00, ..., -2.8362e+00,
-2.2594e+00, -1.7556e+00],
...,
[-2.3072e+00, -2.8846e+00, -2.6741e+00, ..., -3.9452e+00,
-3.2017e+00, -2.5130e+00],
[-1.5453e+00, -1.9691e+00, -1.9035e+00, ..., -2.7401e+00,
-2.2911e+00, -1.8886e+00],
[-1.0317e+00, -1.2378e+00, -1.2961e+00, ..., -1.8394e+00,
-1.5178e+00, -1.1791e+00]]],
[[[ 6.3892e-02, 2.1849e-01, -1.3923e+00, ..., -6.3972e-01,
-2.0047e-01, -5.8997e-02],
[-2.0773e+00, -1.6064e+00, -4.4682e+00, ..., -3.8367e+00,
-1.5583e+00, -5.6808e-01],
[ 5.7528e-01, -7.9086e-02, -8.3053e-01, ..., -2.1998e-01,
-2.6185e-01, 2.3644e-01],
...,
[-3.4705e-01, -1.2378e+00, -1.6453e+00, ..., -1.2624e+00,
-7.9571e-01, -5.8681e-01],
[ 1.0776e+00, 6.1155e-01, 3.0898e-01, ..., 3.8117e-01,
7.0317e-01, 8.0925e-01],
[ 7.5741e-01, 4.8351e-01, 3.2275e-01, ..., 5.0440e-01,
7.5592e-01, 7.1098e-01]],
[[-3.2908e+00, -1.7835e+00, -6.2374e+00, ..., -7.1464e+00,
-6.2766e+00, -4.6275e+00],
[-8.0877e-01, 1.5384e+00, -4.3428e+00, ..., -5.4567e+00,
-5.5498e+00, -4.2750e+00],
[-3.7990e-01, 1.1071e+00, -7.3903e+00, ..., -7.9623e+00,
-7.8529e+00, -6.6929e+00],
...,
[-2.1774e-01, 6.8344e-01, -4.1750e+00, ..., -5.3570e+00,
-4.6588e+00, -3.9046e+00],
[ 2.4338e+00, 3.0052e+00, 5.2650e-01, ..., -1.2205e+00,
-1.3287e+00, -1.4617e+00],
[ 1.4445e+00, 1.6647e+00, -1.4461e+00, ..., -2.5046e+00,
-2.5963e+00, -2.5849e+00]],
[[-1.3261e+00, -2.3309e+00, -2.3018e+00, ..., -1.7687e+00,
-1.2443e+00, -1.3838e+00],
[-1.6804e+00, -1.9893e+00, -1.5327e+00, ..., -1.2664e+00,
-7.3718e-01, -8.1335e-01],
[-2.1686e+00, -1.9518e+00, -1.0312e+00, ..., -1.1179e+00,
-4.7168e-01, -1.8712e-01],
...,
[-8.0191e-01, -1.3038e+00, -6.2852e-01, ..., -5.8339e-01,
-4.5145e-01, -6.1565e-01],
[-1.2534e+00, -8.3269e-01, -6.2750e-01, ..., -2.6454e-01,
-6.9512e-02, -4.5627e-02],
[-1.4318e+00, -7.9508e-01, -4.2593e-01, ..., -3.3514e-01,
-2.8781e-01, -2.4474e-01]],
...,
[[-2.0273e+00, -1.0777e+00, -2.3230e+00, ..., -3.2109e+00,
-2.6568e+00, -1.8362e+00],
[-1.7721e+00, 5.3618e-02, -1.3381e+00, ..., -2.4769e+00,
-2.5268e+00, -1.9717e+00],
[-3.9460e-01, 2.4106e+00, 1.1140e+00, ..., 1.3661e-01,
2.9959e-01, 5.4691e-01],
...,
[-1.7126e+00, 1.1678e+00, -4.7832e-02, ..., -1.1496e+00,
-4.6935e-01, 2.6141e-02],
[-5.7979e-01, 1.6196e+00, 9.2803e-01, ..., -1.8246e-02,
1.7016e-01, 2.7355e-01],
[-1.6652e+00, -1.0924e+00, -1.5793e+00, ..., -2.2342e+00,
-1.2974e+00, -6.7575e-01]],
[[-2.4228e+00, -2.8274e+00, -4.2333e+00, ..., -2.3793e+00,
-1.8830e+00, -1.7003e+00],
[-1.7822e+00, -1.6557e+00, -4.9093e+00, ..., -2.8978e+00,
-2.7580e+00, -2.7657e+00],
[-2.8400e+00, -2.4702e+00, -6.6873e+00, ..., -4.4616e+00,
-4.3690e+00, -3.4841e+00],
...,
[-1.5844e+00, -2.6248e-01, -4.2318e+00, ..., -2.1172e+00,
-1.5970e+00, -1.2470e+00],
[-1.6137e+00, -5.2114e-01, -3.4749e+00, ..., -2.0598e+00,
-1.3809e+00, -8.5190e-01],
[-2.0866e+00, -1.1852e+00, -2.8653e+00, ..., -2.6475e+00,
-1.7911e+00, -1.5304e+00]],
[[-1.8034e+00, -2.5311e+00, -2.4431e+00, ..., -2.7005e+00,
-2.2154e+00, -1.6986e+00],
[-1.8029e+00, -2.5090e+00, -2.1399e+00, ..., -2.4032e+00,
-1.8997e+00, -1.5404e+00],
[-2.2594e+00, -2.8258e+00, -2.5119e+00, ..., -2.7478e+00,
-2.1419e+00, -1.6293e+00],
...,
[-2.4024e+00, -2.9979e+00, -2.7208e+00, ..., -2.8377e+00,
-2.2624e+00, -1.7088e+00],
[-1.6142e+00, -2.0684e+00, -1.9633e+00, ..., -2.0172e+00,
-1.6567e+00, -1.3165e+00],
[-1.0854e+00, -1.3176e+00, -1.3547e+00, ..., -1.2605e+00,
-1.0736e+00, -7.9245e-01]]]], device='cuda:0'),
tensor([[[[ 2.6513e-01, 5.5901e-01, 6.8429e-01, ..., -3.5780e-01,
-1.0467e-01, -1.2633e-02],
[-2.5098e+00, -2.0309e+00, -3.6215e+00, ..., -3.2412e+00,
-1.3215e+00, -4.7113e-01],
[-1.8321e+00, -3.6114e+00, -2.7323e+00, ..., -1.5008e-01,
-3.8554e-01, 1.5221e-01],
...,
[-9.4609e-01, -2.0787e+00, -1.9538e+00, ..., -1.9560e+00,
-1.9585e+00, -1.3028e+00],
[ 7.4830e-01, -1.2996e-01, -1.2763e-01, ..., 6.1273e-02,
2.5055e-01, 4.3592e-01],
[ 1.7124e-01, -5.8217e-01, -4.0989e-01, ..., 2.0605e-01,
4.1695e-01, 4.6745e-01]],
[[-1.7458e+00, -1.2573e+00, -4.2359e+00, ..., -6.8993e+00,
-6.1209e+00, -4.5506e+00],
[ 5.5659e-01, 1.8787e+00, -2.3434e+00, ..., -5.5201e+00,
-5.6099e+00, -4.3128e+00],
[ 1.1664e+00, 1.4663e+00, -5.2075e+00, ..., -7.9294e+00,
-7.8458e+00, -6.6122e+00],
...,
[ 3.1405e-01, 2.1812e-01, -4.2324e+00, ..., -7.1773e+00,
-6.7217e+00, -5.2682e+00],
[ 2.3481e+00, 2.2507e+00, -1.9768e-01, ..., -2.0305e+00,
-2.6106e+00, -2.1270e+00],
[ 1.3726e+00, 1.1735e+00, -1.7698e+00, ..., -2.9406e+00,
-3.4751e+00, -3.0659e+00]],
[[-2.3068e-01, -1.3840e+00, -1.5944e+00, ..., -1.4397e+00,
-1.0652e+00, -1.3489e+00],
[-7.7279e-01, -1.0808e+00, -1.0530e+00, ..., -1.0084e+00,
-6.3055e-01, -9.1157e-01],
[-4.7373e-01, -3.2715e-01, -7.3431e-01, ..., -1.1207e+00,
-6.0715e-01, -4.7358e-01],
...,
[-3.3869e-02, -5.3743e-01, -5.9548e-01, ..., -1.0428e+00,
-8.9749e-01, -1.2520e+00],
[-6.5855e-01, -1.9083e-01, -5.5004e-01, ..., -5.3955e-01,
-2.8255e-01, -4.5264e-01],
[-8.0003e-01, -2.2717e-01, -4.6690e-01, ..., -4.8100e-01,
-4.7126e-01, -6.4183e-01]],
...,
[[-3.1176e+00, -2.5300e+00, -2.6136e+00, ..., -3.0667e+00,
-2.4680e+00, -1.7974e+00],
[-2.1410e+00, -6.0348e-01, -9.6902e-01, ..., -2.4652e+00,
-2.4895e+00, -2.0098e+00],
[-1.9120e+00, 2.3650e-01, 6.0584e-01, ..., -1.5675e-01,
2.0269e-02, 2.7814e-01],
...,
[-2.1473e+00, -2.0155e-01, -9.0796e-02, ..., -8.1015e-01,
-2.1811e-01, 2.7752e-01],
[-1.1847e+00, 8.1124e-01, 1.0440e+00, ..., 4.9072e-01,
5.3911e-01, 8.4967e-01],
[-2.0892e+00, -1.4346e+00, -1.5938e+00, ..., -2.1226e+00,
-1.1140e+00, -3.1197e-01]],
[[-1.6152e+00, -1.5726e+00, -2.9376e+00, ..., -2.2989e+00,
-1.9549e+00, -1.8257e+00],
[-6.2237e-01, 7.4147e-02, -3.3675e+00, ..., -2.9241e+00,
-2.8962e+00, -2.8885e+00],
[-2.1775e+00, -8.4521e-01, -5.3034e+00, ..., -4.5136e+00,
-4.5398e+00, -3.7128e+00],
...,
[-1.3653e+00, 8.2366e-02, -3.3174e+00, ..., -2.8688e+00,
-2.7723e+00, -2.3604e+00],
[-1.6808e+00, -5.2998e-01, -2.9665e+00, ..., -2.4454e+00,
-2.2789e+00, -1.7102e+00],
[-2.2869e+00, -1.3774e+00, -2.9983e+00, ..., -2.8358e+00,
-2.4520e+00, -2.2279e+00]],
[[-1.3632e+00, -1.9273e+00, -1.6884e+00, ..., -2.5877e+00,
-2.1664e+00, -1.6842e+00],
[-1.3413e+00, -1.8663e+00, -1.2612e+00, ..., -2.4246e+00,
-1.9587e+00, -1.6001e+00],
[-1.9372e+00, -2.4083e+00, -1.8070e+00, ..., -2.8443e+00,
-2.2763e+00, -1.7619e+00],
...,
[-2.4069e+00, -3.0051e+00, -2.7536e+00, ..., -4.0484e+00,
-3.2812e+00, -2.5634e+00],
[-1.5665e+00, -2.0545e+00, -1.9750e+00, ..., -2.8228e+00,
-2.3586e+00, -1.9378e+00],
[-1.0270e+00, -1.3342e+00, -1.3614e+00, ..., -1.8782e+00,
-1.5636e+00, -1.2056e+00]]],
[[[ 2.3340e+00, -3.8409e+00, -1.0335e+01, ..., -9.8360e-01,
-5.9038e-01, -2.9423e-01],
[-7.1300e+00, -1.1634e+01, -1.4192e+01, ..., -4.4367e+00,
-1.9932e+00, -8.7541e-01],
[-7.3683e+00, -1.0635e+01, -1.0877e+01, ..., -4.0237e-01,
-6.2034e-01, 8.4215e-02],
...,
[-1.8434e+01, -2.4774e+01, -2.8173e+01, ..., -3.6883e+00,
-3.0471e+00, -1.9853e+00],
[-1.6308e+01, -2.7144e+01, -3.0318e+01, ..., -1.1596e+00,
-5.6285e-01, -2.9340e-01],
[-9.4294e+00, -1.9336e+01, -2.4257e+01, ..., -1.2807e+00,
-8.5115e-01, -8.1160e-01]],
[[-3.9408e+00, -4.3028e+00, -8.8661e+00, ..., -7.5889e+00,
-6.6843e+00, -4.9078e+00],
[-7.7207e-01, -4.1774e+00, -1.2006e+01, ..., -5.8928e+00,
-6.0161e+00, -4.5958e+00],
[ 1.2450e+00, -5.1468e+00, -1.6455e+01, ..., -8.5142e+00,
-8.4016e+00, -7.0104e+00],
...,
[-2.3140e+00, -8.7617e+00, -1.9277e+01, ..., -5.8234e+00,
-4.9862e+00, -3.7864e+00],
[ 1.0915e+00, -2.7859e+00, -1.4275e+01, ..., -1.7564e+00,
-1.5566e+00, -1.3495e+00],
[ 1.5632e+00, 8.4426e-01, -8.1879e+00, ..., -2.7781e+00,
-2.5380e+00, -2.3741e+00]],
[[-1.9433e+00, -2.9962e+00, -3.2661e+00, ..., -1.9960e+00,
-1.4306e+00, -1.5532e+00],
[-1.5345e+00, -4.1977e-01, -1.5348e+00, ..., -1.4186e+00,
-8.7564e-01, -1.0057e+00],
[-2.9542e+00, -1.7957e+00, -3.7724e+00, ..., -1.3145e+00,
-6.2050e-01, -2.7168e-01],
...,
[-2.1747e+00, 8.3027e-01, -1.4421e-01, ..., -5.2898e-01,
-3.9553e-01, -5.8523e-01],
[-1.0215e+00, 1.5363e+00, -1.8302e+00, ..., -4.5811e-02,
3.5084e-01, 3.4728e-01],
[-6.5777e-01, 3.0495e-01, -1.7538e+00, ..., -1.5203e-01,
-2.8157e-02, -1.6596e-01]],
...,
[[-2.5248e+00, -1.6902e+00, -2.0496e+00, ..., -3.1336e+00,
-2.5828e+00, -1.8243e+00],
[-4.9817e+00, -7.2630e+00, -2.8300e+00, ..., -2.7884e+00,
-2.7774e+00, -2.1456e+00],
[-3.2575e+00, -3.2203e+00, 1.8319e+00, ..., 1.1520e-01,
2.2761e-01, 5.1110e-01],
...,
[-7.3554e+00, -1.4543e+01, -2.0837e+01, ..., -1.1589e+00,
1.0246e-01, 5.6685e-01],
[-6.4449e+00, -1.7031e+01, -2.5672e+01, ..., -2.4868e-01,
1.1995e-01, 5.3181e-01],
[-1.2050e+00, -7.2841e+00, -1.3757e+01, ..., -2.7880e+00,
-1.7585e+00, -8.3162e-01]],
[[-2.8468e+00, -1.9531e+00, -8.2519e-01, ..., -2.7871e+00,
-2.3164e+00, -2.0929e+00],
[-1.2652e+00, -1.7320e-01, -1.9509e+00, ..., -3.2863e+00,
-3.1763e+00, -3.0884e+00],
[-3.0594e+00, -3.7823e+00, -1.2955e+01, ..., -4.6555e+00,
-4.6947e+00, -3.7536e+00],
...,
[-3.8275e-01, 1.6420e+00, -4.4944e+00, ..., -2.0737e+00,
-1.3824e+00, -1.0555e+00],
[-1.1366e+00, 1.0041e+00, -3.1595e+00, ..., -2.2158e+00,
-1.3358e+00, -9.5594e-01],
[-9.2893e-01, 1.6277e-01, -1.7475e+00, ..., -3.0425e+00,
-1.8790e+00, -1.8445e+00]],
[[-3.2834e+00, -5.1071e+00, -5.9063e+00, ..., -2.9717e+00,
-2.4429e+00, -1.8750e+00],
[-3.4836e+00, -6.0659e+00, -7.6001e+00, ..., -2.7134e+00,
-2.1471e+00, -1.7425e+00],
[-3.4159e+00, -5.4412e+00, -7.6259e+00, ..., -3.1290e+00,
-2.4596e+00, -1.8848e+00],
...,
[-3.7699e+00, -6.9178e+00, -8.6871e+00, ..., -3.0490e+00,
-2.3957e+00, -1.7956e+00],
[-3.4748e+00, -6.4797e+00, -8.6594e+00, ..., -2.2103e+00,
-1.7838e+00, -1.3973e+00],
[-2.5701e+00, -4.7836e+00, -6.8386e+00, ..., -1.4266e+00,
-1.1799e+00, -9.1440e-01]]],
[[[ 2.8065e-01, 5.2742e-01, 6.5755e-01, ..., 4.8838e-01,
1.0879e+00, 6.6207e-01],
[-2.2190e+00, -1.5608e+00, -3.2188e+00, ..., -2.5947e+00,
-7.4459e-01, -1.1205e-01],
[-2.0897e-01, -1.0291e+00, -2.3411e-01, ..., -2.0620e-01,
5.7551e-01, 7.1427e-01],
...,
[-2.5197e-01, -1.1700e+00, -2.5917e+00, ..., -1.5309e+00,
-1.4225e+00, -8.6848e-01],
[ 1.3053e+00, 7.7975e-01, -8.9126e-01, ..., -2.7797e-01,
1.0165e-01, 3.8176e-01],
[ 7.6526e-01, 2.5049e-01, -6.5126e-01, ..., -1.6478e-01,
2.2481e-01, 3.8642e-01]],
[[-1.6598e+00, -1.1284e+00, -4.1566e+00, ..., -4.7363e+00,
-4.4883e+00, -3.4285e+00],
[ 6.4505e-01, 1.9841e+00, -2.2220e+00, ..., -3.4766e+00,
-3.7542e+00, -3.0343e+00],
[ 1.3895e+00, 1.7411e+00, -4.8907e+00, ..., -5.4913e+00,
-5.4144e+00, -4.9505e+00],
...,
[-1.2555e+00, 1.3547e-01, -6.6737e+00, ..., -7.5937e+00,
-6.8897e+00, -5.4101e+00],
[ 1.8016e+00, 2.8040e+00, -1.0315e+00, ..., -2.5867e+00,
-2.9493e+00, -2.3765e+00],
[ 1.2009e+00, 1.7720e+00, -2.2801e+00, ..., -3.2547e+00,
-3.6207e+00, -3.1302e+00]],
[[-1.2050e-01, -1.1591e+00, -1.3222e+00, ..., -4.8995e-01,
-1.7429e-01, -4.6040e-01],
[-6.0209e-01, -8.5588e-01, -7.7239e-01, ..., -2.4339e-01,
5.7866e-02, -5.1583e-02],
[-5.6018e-01, -3.4345e-01, -3.8596e-01, ..., -3.4952e-01,
8.7807e-02, 1.0626e-01],
...,
[-1.3311e+00, -1.7019e+00, -7.3715e-01, ..., -8.9487e-01,
-7.7430e-01, -1.0221e+00],
[-1.6923e+00, -1.1105e+00, -8.0247e-01, ..., -5.6700e-01,
-2.9657e-01, -3.6309e-01],
[-1.8902e+00, -1.2884e+00, -8.8226e-01, ..., -9.4914e-01,
-8.1570e-01, -7.8832e-01]],
...,
[[-3.0516e+00, -2.5051e+00, -2.8153e+00, ..., -3.3710e+00,
-2.5822e+00, -1.7048e+00],
[-2.2708e+00, -8.3132e-01, -1.5952e+00, ..., -2.6106e+00,
-2.2772e+00, -1.7802e+00],
[-1.0440e+00, 1.2241e+00, 6.6628e-01, ..., -5.0437e-01,
-2.8221e-02, 1.3088e-01],
...,
[-1.7792e+00, 1.2658e+00, -3.1828e-02, ..., -1.0714e+00,
-5.1827e-01, 1.7238e-02],
[-7.5276e-01, 1.9021e+00, 1.5957e+00, ..., 4.7738e-01,
4.2126e-01, 6.8335e-01],
[-2.1957e+00, -1.1783e+00, -1.0750e+00, ..., -1.8670e+00,
-9.7028e-01, -3.0793e-01]],
[[-1.5997e+00, -1.5844e+00, -2.9287e+00, ..., -1.3387e+00,
-9.2973e-01, -9.1447e-01],
[-6.0380e-01, 1.0917e-01, -3.2283e+00, ..., -1.4636e+00,
-1.3059e+00, -1.6099e+00],
[-2.1081e+00, -8.7361e-01, -5.3290e+00, ..., -3.2156e+00,
-2.7557e+00, -2.1993e+00],
...,
[-2.4327e+00, -1.5424e+00, -5.4422e+00, ..., -3.0798e+00,
-2.7148e+00, -2.1705e+00],
[-2.1822e+00, -1.4113e+00, -4.2383e+00, ..., -2.6734e+00,
-2.2973e+00, -1.6335e+00],
[-2.3440e+00, -1.6948e+00, -3.2724e+00, ..., -2.9623e+00,
-2.4268e+00, -2.1076e+00]],
[[-1.3627e+00, -1.9236e+00, -1.7200e+00, ..., -1.7754e+00,
-1.4682e+00, -1.1273e+00],
[-1.2876e+00, -1.7799e+00, -1.2513e+00, ..., -1.4084e+00,
-1.1177e+00, -8.9049e-01],
[-1.6149e+00, -1.9756e+00, -1.5007e+00, ..., -1.6399e+00,
-1.2379e+00, -9.2502e-01],
...,
[-3.1778e+00, -3.9707e+00, -3.7813e+00, ..., -3.9330e+00,
-3.1524e+00, -2.4327e+00],
[-2.1238e+00, -2.7512e+00, -2.7195e+00, ..., -2.7609e+00,
-2.2791e+00, -1.8551e+00],
[-1.5627e+00, -1.8988e+00, -1.9551e+00, ..., -1.8640e+00,
-1.5168e+00, -1.1621e+00]]],
[[[-5.7563e+00, -3.5398e+00, -4.2049e+00, ..., 4.6354e-01,
7.5836e-01, 6.1669e-01],
[-9.2589e+00, -5.2062e+00, -8.9932e+00, ..., -1.8997e+00,
-4.5877e-01, 3.1248e-01],
[-1.1179e+01, -1.2552e+01, -1.7342e+01, ..., 8.7949e-02,
2.8628e-01, 6.3318e-01],
...,
[-4.5724e-01, -1.2341e+00, -1.4468e+00, ..., -1.4402e+00,
-1.0793e+00, -8.2360e-01],
[ 9.9629e-01, 4.6957e-01, 2.7613e-01, ..., 3.3156e-01,
8.3628e-01, 8.6830e-01],
[ 4.5903e-01, -1.3848e-02, -2.0032e-02, ..., 2.5199e-01,
6.7261e-01, 5.4488e-01]],
[[-2.5813e-01, -1.0662e+00, -3.9731e+00, ..., -6.3563e+00,
-5.4760e+00, -4.2296e+00],
[ 3.2888e+00, 2.2962e+00, -3.1685e+00, ..., -5.1766e+00,
-5.1191e+00, -4.1587e+00],
[ 2.6184e+00, 2.4666e-01, -8.4130e+00, ..., -7.4128e+00,
-7.4016e+00, -6.6909e+00],
...,
[-1.2157e-01, 2.5464e-01, -4.2274e+00, ..., -5.0008e+00,
-4.4362e+00, -3.6296e+00],
[ 2.0765e+00, 2.2659e+00, -1.6292e-01, ..., -9.6792e-01,
-1.2324e+00, -1.2556e+00],
[ 1.1872e+00, 1.1679e+00, -1.7498e+00, ..., -2.3179e+00,
-2.5336e+00, -2.4669e+00]],
[[-1.9829e+00, -1.6616e+00, 6.4664e-01, ..., -8.7678e-01,
-6.0677e-01, -8.4617e-01],
[-2.4831e+00, -1.4522e-01, 2.7352e+00, ..., -7.3376e-01,
-4.3242e-01, -4.7385e-01],
[-5.1401e+00, -1.9804e+00, 3.6615e+00, ..., -5.6748e-01,
-2.5833e-01, -2.5415e-01],
...,
[-4.5647e-01, -9.2497e-01, -4.9150e-01, ..., -4.9690e-01,
-2.7738e-01, -4.8048e-01],
[-8.5948e-01, -4.1908e-01, -4.0426e-01, ..., -1.4166e-01,
2.1018e-01, 2.0040e-01],
[-9.3044e-01, -3.5575e-01, -2.8031e-01, ..., -2.5899e-01,
-1.2951e-01, -1.5738e-01]],
...,
[[-5.6711e+00, -7.3037e+00, -7.0097e+00, ..., -3.2880e+00,
-2.7059e+00, -1.8897e+00],
[-5.8703e+00, -7.3197e+00, -9.1912e+00, ..., -2.0051e+00,
-2.0981e+00, -1.6634e+00],
[-1.6729e+00, -3.6824e+00, -1.0205e+01, ..., -3.2032e-01,
-2.6629e-01, 8.1224e-02],
...,
[-1.7098e+00, 5.0036e-01, -7.5826e-02, ..., -1.1675e+00,
-2.8108e-01, 2.3405e-01],
[-8.5394e-01, 1.1268e+00, 9.2734e-01, ..., 5.6828e-02,
3.4870e-01, 5.1601e-01],
[-1.8520e+00, -1.2052e+00, -1.5210e+00, ..., -2.3826e+00,
-1.3676e+00, -7.9674e-01]],
[[-9.4005e-01, -1.9572e+00, -7.2057e+00, ..., -1.5999e+00,
-1.1114e+00, -9.3844e-01],
[-8.0352e-01, -1.8674e+00, -9.6539e+00, ..., -2.3899e+00,
-2.1547e+00, -2.3234e+00],
[-2.2704e+00, -3.2398e+00, -1.0672e+01, ..., -4.5417e+00,
-4.1398e+00, -3.3812e+00],
...,
[-1.4922e+00, -3.4609e-01, -3.7148e+00, ..., -1.8111e+00,
-1.3023e+00, -1.0642e+00],
[-1.6603e+00, -7.4172e-01, -3.1816e+00, ..., -1.9150e+00,
-1.2623e+00, -8.2352e-01],
[-2.1839e+00, -1.4568e+00, -3.0154e+00, ..., -2.9009e+00,
-2.0419e+00, -1.8399e+00]],
[[-4.2617e+00, -5.8359e+00, -5.3604e+00, ..., -2.1564e+00,
-1.7919e+00, -1.3408e+00],
[-4.8866e+00, -6.2386e+00, -5.5167e+00, ..., -2.0399e+00,
-1.6350e+00, -1.2984e+00],
[-6.2588e+00, -7.6144e+00, -7.2813e+00, ..., -2.5411e+00,
-1.9366e+00, -1.4519e+00],
...,
[-2.3250e+00, -2.8714e+00, -2.6809e+00, ..., -2.8402e+00,
-2.2735e+00, -1.7323e+00],
[-1.5270e+00, -1.9635e+00, -1.9103e+00, ..., -1.9480e+00,
-1.6130e+00, -1.2889e+00],
[-1.0072e+00, -1.2636e+00, -1.3156e+00, ..., -1.1936e+00,
-1.0266e+00, -7.6649e-01]]],
[[[ 4.1736e-01, 8.3920e-01, 5.7508e-01, ..., 4.7670e-01,
1.0401e+00, 6.4471e-01],
[-1.2807e+00, -3.9929e-01, -1.8178e+00, ..., -2.4706e+00,
-7.0041e-01, -8.0799e-02],
[ 8.5424e-01, 6.2706e-01, 4.5573e-01, ..., -2.9549e-01,
4.4554e-01, 5.7650e-01],
...,
[-1.6359e-01, -9.4975e-01, -1.9628e+00, ..., -2.0066e+00,
-1.4257e+00, -9.4717e-01],
[ 1.1353e+00, 8.3515e-01, 3.2823e-01, ..., -3.3161e-01,
4.9553e-01, 6.9137e-01],
[ 9.5085e-01, 8.2381e-01, 5.6783e-01, ..., -3.7089e-01,
2.7475e-01, 2.5780e-01]],
[[-2.7249e+00, -1.4800e+00, -4.7833e+00, ..., -4.8200e+00,
-4.4104e+00, -3.3317e+00],
[-4.7870e-01, 1.6571e+00, -2.9465e+00, ..., -3.4133e+00,
-3.5921e+00, -2.8547e+00],
[ 3.2752e-01, 1.4221e+00, -5.6952e+00, ..., -5.3841e+00,
-5.2308e+00, -4.8324e+00],
...,
[-7.9652e-01, 9.9663e-02, -4.9864e+00, ..., -4.6057e+00,
-4.0754e+00, -3.3924e+00],
[ 1.7687e+00, 2.3139e+00, -4.0267e-01, ..., -9.0463e-01,
-1.1631e+00, -1.2103e+00],
[ 9.9363e-01, 1.2514e+00, -1.8313e+00, ..., -2.2459e+00,
-2.4391e+00, -2.3352e+00]],
[[-6.8222e-01, -1.8960e+00, -1.7629e+00, ..., -5.6712e-01,
-1.4896e-01, -4.3482e-01],
[-1.4696e+00, -1.9582e+00, -1.2531e+00, ..., -3.1089e-01,
1.2659e-01, 4.6873e-02],
[-1.3592e+00, -1.3605e+00, -7.3873e-01, ..., -2.8067e-01,
2.4501e-01, 2.8561e-01],
...,
[-8.1643e-01, -1.2483e+00, -5.9785e-01, ..., -4.7920e-01,
-1.4164e-01, -3.5031e-01],
[-1.2306e+00, -8.4753e-01, -5.4759e-01, ..., -3.0459e-01,
1.7626e-01, 2.1970e-01],
[-1.2577e+00, -7.0455e-01, -2.9759e-01, ..., -5.0320e-01,
-2.6387e-01, -2.4969e-01]],
...,
[[-2.2778e+00, -1.1342e+00, -2.5412e+00, ..., -3.5575e+00,
-2.7111e+00, -1.7662e+00],
[-6.1483e-01, 1.2676e+00, -8.0264e-01, ..., -2.6485e+00,
-2.4285e+00, -1.8441e+00],
[-1.5876e-01, 2.5810e+00, 7.2413e-01, ..., -6.1244e-01,
-9.6256e-02, 1.5638e-01],
...,
[-1.7086e+00, 1.0439e+00, 1.7145e-01, ..., -1.5882e+00,
-7.0559e-01, -1.2665e-01],
[-5.7510e-01, 1.3240e+00, 8.6034e-01, ..., -1.6401e-01,
9.1807e-02, 3.4061e-01],
[-1.6202e+00, -1.0981e+00, -1.5243e+00, ..., -2.3664e+00,
-1.4679e+00, -9.1158e-01]],
[[-1.3897e+00, -1.3934e+00, -3.1322e+00, ..., -1.2102e+00,
-7.7283e-01, -7.8616e-01],
[-6.4161e-01, -2.3742e-01, -4.0204e+00, ..., -1.4071e+00,
-1.1854e+00, -1.5442e+00],
[-2.1361e+00, -1.5272e+00, -6.5063e+00, ..., -3.2597e+00,
-2.7122e+00, -2.1733e+00],
...,
[-1.9484e+00, -9.4556e-01, -4.5450e+00, ..., -1.4513e+00,
-1.0022e+00, -8.8093e-01],
[-1.7317e+00, -9.6744e-01, -3.6647e+00, ..., -1.7144e+00,
-1.1225e+00, -7.6362e-01],
[-2.0271e+00, -1.4412e+00, -3.0444e+00, ..., -2.7933e+00,
-1.9988e+00, -1.8418e+00]],
[[-1.2573e+00, -1.8978e+00, -1.7260e+00, ..., -1.8137e+00,
-1.5042e+00, -1.1521e+00],
[-1.2564e+00, -1.8223e+00, -1.3828e+00, ..., -1.4309e+00,
-1.1388e+00, -9.1507e-01],
[-1.6352e+00, -2.0141e+00, -1.6231e+00, ..., -1.6458e+00,
-1.2329e+00, -9.2201e-01],
...,
[-2.3705e+00, -3.0762e+00, -2.9748e+00, ..., -2.8737e+00,
-2.2970e+00, -1.7673e+00],
[-1.6059e+00, -2.0895e+00, -2.1149e+00, ..., -1.9751e+00,
-1.6244e+00, -1.3115e+00],
[-1.0911e+00, -1.3306e+00, -1.4315e+00, ..., -1.2273e+00,
-1.0345e+00, -7.9409e-01]]]], device='cuda:0'),
tensor([[[[ 2.9642e-01, 6.1272e-01, 6.2191e-01, ..., -9.0328e-01,
-6.3862e-01, -3.2866e-01],
[-1.6136e+00, -8.5253e-01, -2.2302e+00, ..., -4.2535e+00,
-1.9895e+00, -9.0505e-01],
[ 2.1755e-01, -3.2580e-01, 2.1433e-01, ..., -5.9565e-01,
-9.4936e-01, -1.0374e-01],
...,
[-6.2434e-01, -1.6108e+00, -1.7905e+00, ..., -1.8613e+00,
-1.9621e+00, -1.1506e+00],
[ 9.9646e-01, 3.2156e-01, 1.7149e-01, ..., -2.1577e-01,
-5.6565e-02, 2.7314e-01],
[ 5.0897e-01, -2.3610e-02, 3.8871e-02, ..., -9.6601e-02,
1.7452e-01, 3.4678e-01]],
[[-2.0071e+00, -1.4360e+00, -4.2969e+00, ..., -7.5333e+00,
-6.7275e+00, -4.9434e+00],
[ 6.1385e-02, 1.3513e+00, -2.6486e+00, ..., -6.0479e+00,
-6.2465e+00, -4.7379e+00],
[ 7.8407e-01, 1.1340e+00, -5.1382e+00, ..., -8.6489e+00,
-8.6901e+00, -7.1170e+00],
...,
[-3.0468e-02, 2.9718e-01, -4.2583e+00, ..., -8.0683e+00,
-7.7076e+00, -5.8743e+00],
[ 2.3409e+00, 2.5433e+00, 1.6815e-01, ..., -2.6242e+00,
-3.3954e+00, -2.5056e+00],
[ 1.3817e+00, 1.3664e+00, -1.5885e+00, ..., -3.2613e+00,
-3.9417e+00, -3.2572e+00]],
[[-3.1796e-01, -1.3395e+00, -1.4470e+00, ..., -1.7473e+00,
-1.3335e+00, -1.6006e+00],
[-8.2815e-01, -1.1772e+00, -1.0165e+00, ..., -1.2335e+00,
-8.3318e-01, -1.1686e+00],
[-6.1117e-01, -5.2700e-01, -4.7116e-01, ..., -1.4096e+00,
-8.2366e-01, -5.8172e-01],
...,
[-3.9267e-01, -9.4614e-01, -6.2752e-01, ..., -1.0010e+00,
-8.8423e-01, -1.3146e+00],
[-8.9229e-01, -4.7008e-01, -5.5388e-01, ..., -5.2096e-01,
-2.1959e-01, -4.1420e-01],
[-1.0270e+00, -4.4816e-01, -4.0257e-01, ..., -8.3813e-01,
-8.1985e-01, -9.5313e-01]],
...,
[[-2.5235e+00, -1.9398e+00, -2.3316e+00, ..., -3.0663e+00,
-2.5005e+00, -1.8631e+00],
[-1.4359e+00, -9.7663e-03, -7.3432e-01, ..., -2.8856e+00,
-2.8954e+00, -2.2842e+00],
[-7.1641e-01, 1.3738e+00, 8.1750e-01, ..., -5.2444e-02,
-4.1395e-02, 2.2802e-01],
...,
[-1.8944e+00, 4.3446e-01, -1.5628e-02, ..., -1.0367e+00,
-3.1461e-01, 2.2558e-01],
[-9.0679e-01, 1.1972e+00, 1.0552e+00, ..., 5.8640e-01,
6.2865e-01, 9.4041e-01],
[-1.9339e+00, -1.3049e+00, -1.5955e+00, ..., -1.9910e+00,
-8.9740e-01, -1.1893e-01]],
[[-1.4931e+00, -1.4609e+00, -2.6584e+00, ..., -2.7370e+00,
-2.4329e+00, -2.3035e+00],
[-6.6962e-01, -1.3607e-01, -3.2248e+00, ..., -3.3226e+00,
-3.3738e+00, -3.3205e+00],
[-2.0542e+00, -1.1655e+00, -5.3598e+00, ..., -4.7183e+00,
-4.9472e+00, -4.0864e+00],
...,
[-1.4310e+00, -1.2787e-01, -3.7918e+00, ..., -3.1644e+00,
-3.1072e+00, -2.5716e+00],
[-1.6353e+00, -5.7734e-01, -3.2388e+00, ..., -2.7043e+00,
-2.6442e+00, -1.9785e+00],
[-2.2130e+00, -1.3512e+00, -2.9937e+00, ..., -2.9945e+00,
-2.6747e+00, -2.4275e+00]],
[[-1.2178e+00, -1.7617e+00, -1.5833e+00, ..., -2.9716e+00,
-2.4782e+00, -1.9279e+00],
[-1.1821e+00, -1.6538e+00, -1.2079e+00, ..., -2.8421e+00,
-2.2696e+00, -1.8649e+00],
[-1.5082e+00, -1.8467e+00, -1.4477e+00, ..., -3.3510e+00,
-2.6793e+00, -2.0962e+00],
...,
[-2.3845e+00, -2.9797e+00, -2.7278e+00, ..., -4.5048e+00,
-3.6717e+00, -2.8779e+00],
[-1.5630e+00, -2.0385e+00, -1.9501e+00, ..., -3.1076e+00,
-2.6019e+00, -2.1509e+00],
[-1.0306e+00, -1.3061e+00, -1.3408e+00, ..., -2.0947e+00,
-1.7234e+00, -1.3414e+00]]],
[[[ 4.3021e-01, 8.5191e-01, 5.0758e-01, ..., 4.3133e-01,
1.0123e+00, 6.3663e-01],
[-1.3163e+00, -4.6054e-01, -2.0009e+00, ..., -2.6321e+00,
-7.6614e-01, -8.2023e-02],
[ 8.9029e-01, 6.4708e-01, 3.5464e-01, ..., -2.9263e-01,
4.9298e-01, 6.1981e-01],
...,
[ 6.9258e-02, -5.2538e-01, -1.1948e+00, ..., -1.9472e+00,
-1.3802e+00, -9.2457e-01],
[ 1.3680e+00, 1.1634e+00, 6.9543e-01, ..., -2.8446e-01,
5.1759e-01, 6.9810e-01],
[ 9.9166e-01, 8.6919e-01, 6.4570e-01, ..., -3.1409e-01,
3.0521e-01, 2.9451e-01]],
[[-2.8217e+00, -1.5730e+00, -4.8293e+00, ..., -4.8572e+00,
-4.4451e+00, -3.3583e+00],
[-5.3657e-01, 1.6041e+00, -2.8830e+00, ..., -3.3803e+00,
-3.5507e+00, -2.8985e+00],
[ 2.6415e-01, 1.3938e+00, -5.6503e+00, ..., -5.3644e+00,
-5.2567e+00, -4.9379e+00],
...,
[-8.6126e-01, -3.2136e-03, -4.9455e+00, ..., -4.5894e+00,
-4.1261e+00, -3.4391e+00],
[ 1.6998e+00, 2.2087e+00, -4.5274e-01, ..., -8.9169e-01,
-1.1968e+00, -1.2431e+00],
[ 9.7339e-01, 1.1902e+00, -1.9170e+00, ..., -2.2560e+00,
-2.4679e+00, -2.3552e+00]],
[[-7.1496e-01, -1.9105e+00, -1.7713e+00, ..., -5.9846e-01,
-1.8722e-01, -4.7070e-01],
[-1.4877e+00, -1.9542e+00, -1.2949e+00, ..., -3.6061e-01,
6.3213e-02, 3.2221e-02],
[-1.3795e+00, -1.3541e+00, -7.4826e-01, ..., -2.8323e-01,
2.4251e-01, 2.7974e-01],
...,
[-9.2716e-01, -1.4582e+00, -6.7952e-01, ..., -4.8267e-01,
-1.5100e-01, -3.4114e-01],
[-1.1992e+00, -8.8957e-01, -5.5075e-01, ..., -3.0262e-01,
1.5364e-01, 2.1818e-01],
[-1.2434e+00, -7.2434e-01, -2.7588e-01, ..., -4.8167e-01,
-2.6360e-01, -2.4380e-01]],
...,
[[-2.2860e+00, -1.1576e+00, -2.5649e+00, ..., -3.6761e+00,
-2.8691e+00, -1.8359e+00],
[-5.9372e-01, 1.2686e+00, -7.4959e-01, ..., -2.6546e+00,
-2.4579e+00, -1.8986e+00],
[-1.8403e-01, 2.5825e+00, 8.1088e-01, ..., -5.4702e-01,
-5.6345e-02, 1.6903e-01],
...,
[-1.3255e+00, 1.3520e+00, 2.2951e-02, ..., -1.5598e+00,
-6.9360e-01, -1.4065e-01],
[-4.9934e-01, 1.5148e+00, 7.7470e-01, ..., -1.7582e-01,
1.0937e-01, 3.2576e-01],
[-1.5978e+00, -1.0249e+00, -1.4865e+00, ..., -2.3684e+00,
-1.4497e+00, -9.0856e-01]],
[[-1.4106e+00, -1.3932e+00, -3.0443e+00, ..., -1.1011e+00,
-7.0094e-01, -7.4186e-01],
[-6.5713e-01, -2.0949e-01, -3.8874e+00, ..., -1.3443e+00,
-1.1920e+00, -1.5791e+00],
[-2.1859e+00, -1.5433e+00, -6.3910e+00, ..., -3.2822e+00,
-2.7880e+00, -2.2283e+00],
...,
[-1.8110e+00, -1.0503e+00, -4.7390e+00, ..., -1.5179e+00,
-1.0560e+00, -8.6826e-01],
[-1.6864e+00, -1.0399e+00, -3.7840e+00, ..., -1.7697e+00,
-1.1592e+00, -7.5812e-01],
[-2.0066e+00, -1.4605e+00, -3.0427e+00, ..., -2.8237e+00,
-2.0175e+00, -1.8299e+00]],
[[-1.2615e+00, -1.9182e+00, -1.7522e+00, ..., -1.8096e+00,
-1.5013e+00, -1.1435e+00],
[-1.2558e+00, -1.8232e+00, -1.3992e+00, ..., -1.3983e+00,
-1.1079e+00, -8.9089e-01],
[-1.6234e+00, -2.0042e+00, -1.6267e+00, ..., -1.6006e+00,
-1.2062e+00, -8.9586e-01],
...,
[-2.3610e+00, -2.9611e+00, -2.8578e+00, ..., -2.8619e+00,
-2.2894e+00, -1.7595e+00],
[-1.6085e+00, -2.0472e+00, -2.0522e+00, ..., -1.9701e+00,
-1.6182e+00, -1.3068e+00],
[-1.1034e+00, -1.3397e+00, -1.4530e+00, ..., -1.2196e+00,
-1.0306e+00, -7.9105e-01]]],
[[[ 1.6311e-01, 3.9606e-01, -1.1694e+00, ..., 6.3435e-01,
8.9588e-01, 6.7611e-01],
[-2.0855e+00, -1.4309e+00, -4.3806e+00, ..., -1.4054e+00,
-1.4458e-01, 4.0331e-01],
[ 4.4097e-01, -2.8468e-01, -1.1168e+00, ..., 5.4570e-01,
9.3744e-01, 8.4983e-01],
...,
[-2.6211e-01, -1.2339e+00, -2.5657e+00, ..., -1.4059e+00,
-1.0275e+00, -8.3908e-01],
[ 1.2571e+00, 6.8476e-01, -8.9505e-01, ..., 4.8279e-01,
7.2648e-01, 7.7958e-01],
[ 8.4039e-01, 3.2744e-01, -4.2757e-01, ..., 5.6509e-01,
7.9788e-01, 6.9732e-01]],
[[-3.2421e+00, -1.4534e+00, -6.1170e+00, ..., -5.3277e+00,
-4.5927e+00, -3.6040e+00],
[-5.5253e-01, 2.1722e+00, -4.1437e+00, ..., -4.1574e+00,
-4.0622e+00, -3.3658e+00],
[ 5.9848e-02, 1.8698e+00, -7.3750e+00, ..., -6.0624e+00,
-5.8354e+00, -5.4103e+00],
...,
[-1.2720e+00, 2.4377e-01, -6.5122e+00, ..., -5.3346e+00,
-4.6565e+00, -3.8812e+00],
[ 1.4990e+00, 2.5938e+00, -1.3419e+00, ..., -8.9719e-01,
-1.0932e+00, -1.2582e+00],
[ 9.8475e-01, 1.6321e+00, -2.4022e+00, ..., -2.3597e+00,
-2.5250e+00, -2.5509e+00]],
[[-1.3839e+00, -2.3539e+00, -2.1714e+00, ..., -8.0517e-01,
-6.1944e-01, -7.8576e-01],
[-2.0309e+00, -2.1999e+00, -1.5566e+00, ..., -6.5816e-01,
-4.2740e-01, -3.6293e-01],
[-2.5468e+00, -2.0040e+00, -8.2915e-01, ..., -3.9680e-01,
-1.2679e-01, -1.2163e-01],
...,
[-1.5638e+00, -1.7944e+00, -7.7250e-01, ..., -6.8165e-01,
-5.8012e-01, -7.6843e-01],
[-1.7402e+00, -1.1615e+00, -7.4411e-01, ..., -2.3379e-01,
-3.6940e-02, -3.3192e-02],
[-1.6996e+00, -1.0885e+00, -5.6760e-01, ..., -3.2040e-01,
-2.8738e-01, -2.6776e-01]],
...,
[[-2.3829e+00, -1.3197e+00, -2.7589e+00, ..., -2.9185e+00,
-2.3119e+00, -1.5654e+00],
[-1.8783e+00, 2.1241e-01, -1.3311e+00, ..., -1.4334e+00,
-1.4849e+00, -1.2882e+00],
[-4.8166e-01, 2.5011e+00, 9.1964e-01, ..., -5.1741e-02,
2.3414e-01, 3.0902e-01],
...,
[-1.7505e+00, 1.1447e+00, 2.3127e-01, ..., -1.0785e+00,
-2.8372e-01, 2.3404e-01],
[-8.9508e-01, 1.5864e+00, 1.4739e+00, ..., 1.8116e-01,
3.7317e-01, 4.6148e-01],
[-2.1393e+00, -1.1507e+00, -9.5724e-01, ..., -2.4141e+00,
-1.3738e+00, -7.1027e-01]],
[[-2.3622e+00, -2.6207e+00, -4.0184e+00, ..., -1.1715e+00,
-6.7888e-01, -5.4714e-01],
[-1.6360e+00, -1.2974e+00, -4.7522e+00, ..., -1.7265e+00,
-1.5244e+00, -1.7243e+00],
[-2.9993e+00, -2.4122e+00, -6.8990e+00, ..., -3.7092e+00,
-3.2399e+00, -2.5977e+00],
...,
[-2.5276e+00, -1.7391e+00, -4.9376e+00, ..., -2.0311e+00,
-1.5879e+00, -1.2834e+00],
[-2.2396e+00, -1.5605e+00, -3.8625e+00, ..., -2.0593e+00,
-1.3875e+00, -8.5200e-01],
[-2.3253e+00, -1.7711e+00, -3.0633e+00, ..., -2.8908e+00,
-1.9653e+00, -1.6904e+00]],
[[-1.8933e+00, -2.6221e+00, -2.4011e+00, ..., -1.6660e+00,
-1.3838e+00, -1.0364e+00],
[-1.9146e+00, -2.6295e+00, -2.0890e+00, ..., -1.3740e+00,
-1.1206e+00, -8.8519e-01],
[-2.4393e+00, -2.9921e+00, -2.4981e+00, ..., -1.5979e+00,
-1.2238e+00, -8.9270e-01],
...,
[-3.0985e+00, -3.8806e+00, -3.6630e+00, ..., -2.8688e+00,
-2.2908e+00, -1.7325e+00],
[-2.0555e+00, -2.6622e+00, -2.6386e+00, ..., -1.9827e+00,
-1.6353e+00, -1.3104e+00],
[-1.4803e+00, -1.8035e+00, -1.8564e+00, ..., -1.2067e+00,
-1.0455e+00, -7.6921e-01]]],
[[[ 1.5758e-01, 3.8524e-01, -1.1842e+00, ..., 4.8929e-01,
1.0188e+00, 6.7590e-01],
[-2.1004e+00, -1.4506e+00, -4.4007e+00, ..., -2.4933e+00,
-7.2048e-01, -2.9021e-02],
[ 4.4530e-01, -2.6893e-01, -1.1130e+00, ..., -2.4612e-01,
4.8814e-01, 6.5373e-01],
...,
[-2.9776e-01, -1.2949e+00, -2.6128e+00, ..., -1.9918e+00,
-1.3967e+00, -9.2398e-01],
[ 1.2255e+00, 6.3457e-01, -9.3453e-01, ..., -3.0736e-01,
5.1628e-01, 7.0156e-01],
[ 8.0402e-01, 2.8395e-01, -4.5966e-01, ..., -3.4681e-01,
2.8940e-01, 2.7518e-01]],
[[-3.2626e+00, -1.4751e+00, -6.1482e+00, ..., -4.8131e+00,
-4.3848e+00, -3.3728e+00],
[-5.6234e-01, 2.1596e+00, -4.1572e+00, ..., -3.4036e+00,
-3.5579e+00, -2.9387e+00],
[ 5.9642e-02, 1.8687e+00, -7.3707e+00, ..., -5.4046e+00,
-5.2754e+00, -4.9402e+00],
...,
[-1.2633e+00, 2.6912e-01, -6.5255e+00, ..., -4.5588e+00,
-4.1326e+00, -3.4579e+00],
[ 1.5288e+00, 2.6239e+00, -1.3180e+00, ..., -8.8528e-01,
-1.2270e+00, -1.2607e+00],
[ 1.0049e+00, 1.6428e+00, -2.3941e+00, ..., -2.2436e+00,
-2.4753e+00, -2.3643e+00]],
[[-1.3887e+00, -2.3600e+00, -2.1816e+00, ..., -5.7032e-01,
-1.9297e-01, -4.5436e-01],
[-2.0343e+00, -2.2156e+00, -1.5764e+00, ..., -3.2681e-01,
7.2941e-02, 3.6405e-02],
[-2.5427e+00, -2.0055e+00, -8.2914e-01, ..., -2.6930e-01,
2.3470e-01, 2.6753e-01],
...,
[-1.5980e+00, -1.8440e+00, -7.9352e-01, ..., -4.9262e-01,
-1.7107e-01, -3.4582e-01],
[-1.7778e+00, -1.1964e+00, -7.8083e-01, ..., -3.2171e-01,
1.4834e-01, 2.1383e-01],
[-1.7464e+00, -1.1393e+00, -6.1622e-01, ..., -4.9944e-01,
-2.7585e-01, -2.4596e-01]],
...,
[[-2.3913e+00, -1.3382e+00, -2.7776e+00, ..., -3.6312e+00,
-2.7924e+00, -1.8154e+00],
[-1.8886e+00, 2.0480e-01, -1.3338e+00, ..., -2.6798e+00,
-2.4190e+00, -1.8502e+00],
[-5.0416e-01, 2.5001e+00, 9.3828e-01, ..., -5.7401e-01,
-7.3097e-02, 1.4785e-01],
...,
[-1.7088e+00, 1.1900e+00, 2.3607e-01, ..., -1.5242e+00,
-6.8041e-01, -1.3801e-01],
[-8.3385e-01, 1.6666e+00, 1.5017e+00, ..., -1.4444e-01,
1.0745e-01, 3.2352e-01],
[-2.0714e+00, -1.0355e+00, -8.3721e-01, ..., -2.3380e+00,
-1.4331e+00, -9.1161e-01]],
[[-2.3718e+00, -2.6242e+00, -4.0172e+00, ..., -1.1474e+00,
-7.6409e-01, -7.8413e-01],
[-1.6448e+00, -1.3084e+00, -4.7568e+00, ..., -1.3805e+00,
-1.2555e+00, -1.6043e+00],
[-3.0083e+00, -2.4292e+00, -6.9237e+00, ..., -3.2694e+00,
-2.8044e+00, -2.2388e+00],
...,
[-2.4751e+00, -1.7011e+00, -4.9889e+00, ..., -1.4907e+00,
-1.0577e+00, -8.9148e-01],
[-2.1748e+00, -1.5268e+00, -3.8850e+00, ..., -1.7413e+00,
-1.1626e+00, -7.6496e-01],
[-2.2792e+00, -1.7456e+00, -3.0722e+00, ..., -2.8075e+00,
-2.0189e+00, -1.8350e+00]],
[[-1.8950e+00, -2.6269e+00, -2.4065e+00, ..., -1.7961e+00,
-1.4884e+00, -1.1320e+00],
[-1.9157e+00, -2.6319e+00, -2.0874e+00, ..., -1.4034e+00,
-1.1130e+00, -8.8884e-01],
[-2.4373e+00, -2.9923e+00, -2.4990e+00, ..., -1.6122e+00,
-1.2113e+00, -8.9450e-01],
...,
[-3.1180e+00, -3.8894e+00, -3.6596e+00, ..., -2.8638e+00,
-2.2890e+00, -1.7629e+00],
[-2.0743e+00, -2.6783e+00, -2.6344e+00, ..., -1.9800e+00,
-1.6260e+00, -1.3135e+00],
[-1.4886e+00, -1.8141e+00, -1.8549e+00, ..., -1.2288e+00,
-1.0395e+00, -7.9639e-01]]],
[[[ 2.3713e-01, 5.3923e-01, 6.5892e-01, ..., 4.5522e-01,
8.3331e-01, 6.7284e-01],
[-1.8425e+00, -1.0882e+00, -2.4873e+00, ..., -1.5703e+00,
-1.0936e-01, 5.0549e-01],
[-5.9208e-03, -6.5063e-01, 1.3453e-01, ..., 4.0758e-01,
7.9518e-01, 8.9541e-01],
...,
[-8.2815e-01, -1.9107e+00, -1.9123e+00, ..., -1.4629e+00,
-1.0421e+00, -7.1380e-01],
[ 8.8648e-01, 8.4913e-02, 4.1519e-03, ..., 3.1869e-01,
6.9912e-01, 8.4241e-01],
[ 3.3927e-01, -3.2908e-01, -2.3124e-01, ..., 5.3671e-01,
8.2982e-01, 7.6528e-01]],
[[-1.7781e+00, -1.3574e+00, -4.2420e+00, ..., -6.0130e+00,
-5.0917e+00, -3.9686e+00],
[ 3.1974e-01, 1.4330e+00, -2.6223e+00, ..., -4.9237e+00,
-4.6352e+00, -3.7750e+00],
[ 1.0407e+00, 1.1653e+00, -5.1532e+00, ..., -7.0900e+00,
-6.6367e+00, -5.9584e+00],
...,
[ 2.1406e-01, 3.5995e-01, -4.1429e+00, ..., -6.2788e+00,
-5.4479e+00, -4.4522e+00],
[ 2.4467e+00, 2.5184e+00, 1.5721e-01, ..., -1.4709e+00,
-1.6089e+00, -1.6234e+00],
[ 1.4488e+00, 1.3503e+00, -1.5954e+00, ..., -2.8115e+00,
-2.9496e+00, -2.8498e+00]],
[[-2.0099e-01, -1.2297e+00, -1.4229e+00, ..., -8.7780e-01,
-5.9384e-01, -7.1051e-01],
[-6.4973e-01, -9.8097e-01, -9.3321e-01, ..., -7.0426e-01,
-3.7878e-01, -2.6259e-01],
[-4.6245e-01, -3.7201e-01, -4.5545e-01, ..., -4.5157e-01,
-4.0085e-02, 8.3178e-02],
...,
[-1.9066e-01, -7.3870e-01, -6.1848e-01, ..., -7.6988e-01,
-5.5620e-01, -6.4969e-01],
[-7.8285e-01, -3.1761e-01, -5.6548e-01, ..., -3.6135e-01,
-6.2681e-02, 1.2842e-02],
[-9.3861e-01, -3.4049e-01, -4.4370e-01, ..., -3.4931e-01,
-2.4093e-01, -1.8902e-01]],
...,
[[-2.6971e+00, -2.2265e+00, -2.3733e+00, ..., -3.1714e+00,
-2.5941e+00, -1.7596e+00],
[-1.8224e+00, -5.1715e-01, -9.9080e-01, ..., -1.8159e+00,
-1.8216e+00, -1.4934e+00],
[-9.1665e-01, 1.0394e+00, 7.4817e-01, ..., -2.8351e-01,
2.4981e-02, 1.9371e-01],
...,
[-2.1061e+00, 8.9342e-02, -7.0733e-02, ..., -1.0252e+00,
-3.3048e-01, 2.1938e-01],
[-1.1015e+00, 1.0132e+00, 1.0780e+00, ..., 2.9231e-01,
4.0682e-01, 5.4494e-01],
[-2.0883e+00, -1.4270e+00, -1.6489e+00, ..., -2.2355e+00,
-1.2304e+00, -5.7082e-01]],
[[-1.5683e+00, -1.5626e+00, -2.7713e+00, ..., -1.5749e+00,
-9.4321e-01, -6.7783e-01],
[-6.8328e-01, -1.3620e-01, -3.2203e+00, ..., -2.2271e+00,
-1.8085e+00, -1.8310e+00],
[-2.0395e+00, -1.0615e+00, -5.2138e+00, ..., -4.1237e+00,
-3.4962e+00, -2.6760e+00],
...,
[-1.3796e+00, 8.6122e-02, -3.5216e+00, ..., -2.4976e+00,
-1.9211e+00, -1.4187e+00],
[-1.6527e+00, -4.8773e-01, -3.0926e+00, ..., -2.3243e+00,
-1.5879e+00, -9.2229e-01],
[-2.2716e+00, -1.3330e+00, -3.0065e+00, ..., -2.9227e+00,
-1.9954e+00, -1.6710e+00]],
[[-1.2709e+00, -1.7920e+00, -1.6147e+00, ..., -1.9992e+00,
-1.6276e+00, -1.1841e+00],
[-1.2228e+00, -1.6830e+00, -1.2124e+00, ..., -1.7765e+00,
-1.4189e+00, -1.0764e+00],
[-1.5460e+00, -1.8840e+00, -1.4696e+00, ..., -2.0912e+00,
-1.5823e+00, -1.1223e+00],
...,
[-2.4213e+00, -3.0239e+00, -2.7519e+00, ..., -3.3383e+00,
-2.6237e+00, -1.9566e+00],
[-1.5758e+00, -2.0647e+00, -1.9659e+00, ..., -2.3405e+00,
-1.9032e+00, -1.5010e+00],
[-1.0319e+00, -1.3316e+00, -1.3499e+00, ..., -1.4637e+00,
-1.2221e+00, -8.9129e-01]]]], device='cuda:0'),
tensor([[[[ 1.1024e-01, 3.2850e-01, -1.2446e+00, ..., 6.5403e-01,
9.3724e-01, 6.9829e-01],
[-2.0041e+00, -1.4010e+00, -4.2549e+00, ..., -1.4806e+00,
-1.5799e-01, 4.2786e-01],
[ 3.4166e-01, -3.1079e-01, -1.1319e+00, ..., 5.2398e-01,
9.3327e-01, 8.7361e-01],
...,
[-3.0880e-01, -1.3644e+00, -2.8317e+00, ..., -3.0636e-01,
-7.7763e-01, -6.1195e-01],
[ 1.3667e+00, 7.4815e-01, -9.3902e-01, ..., 5.2501e-01,
7.7457e-01, 8.0458e-01],
[ 8.9680e-01, 3.7703e-01, -4.2944e-01, ..., 5.6915e-01,
8.0181e-01, 7.0441e-01]],
[[-3.1610e+00, -1.5263e+00, -6.1374e+00, ..., -5.4617e+00,
-4.6781e+00, -3.6623e+00],
[-7.2762e-01, 1.7564e+00, -4.4539e+00, ..., -4.2044e+00,
-4.0724e+00, -3.3777e+00],
[-2.6121e-01, 1.4204e+00, -7.5225e+00, ..., -6.1584e+00,
-5.9005e+00, -5.5017e+00],
...,
[-1.1807e+00, 4.6867e-01, -6.4157e+00, ..., -5.8740e+00,
-4.7535e+00, -3.9581e+00],
[ 1.8582e+00, 3.0470e+00, -7.9848e-01, ..., -1.3038e+00,
-1.4405e+00, -1.5144e+00],
[ 1.2351e+00, 1.9291e+00, -2.1354e+00, ..., -2.5171e+00,
-2.6246e+00, -2.5543e+00]],
[[-1.3173e+00, -2.1681e+00, -2.0117e+00, ..., -8.4217e-01,
-6.3819e-01, -8.2396e-01],
[-1.8569e+00, -2.0122e+00, -1.4567e+00, ..., -6.8535e-01,
-4.2374e-01, -3.7318e-01],
[-2.4159e+00, -1.9309e+00, -8.3922e-01, ..., -3.8518e-01,
-9.8795e-02, -1.0887e-01],
...,
[-1.6880e+00, -1.9743e+00, -8.2980e-01, ..., -4.7439e-01,
-5.3935e-01, -6.9615e-01],
[-1.8949e+00, -1.2904e+00, -8.3176e-01, ..., -2.2829e-01,
-4.8947e-02, -4.6422e-02],
[-1.8984e+00, -1.2249e+00, -6.5168e-01, ..., -3.2079e-01,
-2.7924e-01, -2.5621e-01]],
...,
[[-2.1250e+00, -1.2124e+00, -2.5557e+00, ..., -3.0984e+00,
-2.4702e+00, -1.6543e+00],
[-1.8788e+00, 1.2523e-02, -1.3627e+00, ..., -1.5082e+00,
-1.5679e+00, -1.3516e+00],
[-5.3250e-01, 2.2283e+00, 8.2351e-01, ..., -1.2234e-01,
2.1427e-01, 3.2217e-01],
...,
[-1.9610e+00, 1.1715e+00, 1.7916e-01, ..., -6.4403e-01,
-2.6500e-01, 1.7693e-01],
[-9.5530e-01, 1.7492e+00, 1.5948e+00, ..., 1.5958e-01,
3.0753e-01, 3.9462e-01],
[-2.2577e+00, -1.2113e+00, -1.0136e+00, ..., -2.1699e+00,
-1.2590e+00, -6.3773e-01]],
[[-2.3976e+00, -2.7182e+00, -3.9699e+00, ..., -1.1137e+00,
-6.0866e-01, -4.9473e-01],
[-1.8092e+00, -1.6026e+00, -4.7158e+00, ..., -1.6972e+00,
-1.5086e+00, -1.7489e+00],
[-3.0373e+00, -2.6158e+00, -6.6905e+00, ..., -3.7402e+00,
-3.3025e+00, -2.6727e+00],
...,
[-2.4976e+00, -1.5884e+00, -5.0610e+00, ..., -2.0401e+00,
-1.7538e+00, -1.4195e+00],
[-2.2314e+00, -1.4724e+00, -3.9631e+00, ..., -2.1416e+00,
-1.4989e+00, -9.8277e-01],
[-2.3949e+00, -1.7836e+00, -3.1457e+00, ..., -2.8222e+00,
-1.9388e+00, -1.6543e+00]],
[[-1.8624e+00, -2.5592e+00, -2.3948e+00, ..., -1.6995e+00,
-1.4084e+00, -1.0553e+00],
[-1.9079e+00, -2.6034e+00, -2.1497e+00, ..., -1.3765e+00,
-1.1175e+00, -8.8714e-01],
[-2.4370e+00, -3.0085e+00, -2.5974e+00, ..., -1.5846e+00,
-1.2105e+00, -8.8306e-01],
...,
[-3.2050e+00, -4.0079e+00, -3.7384e+00, ..., -2.9297e+00,
-2.2291e+00, -1.6824e+00],
[-2.1111e+00, -2.7365e+00, -2.6603e+00, ..., -1.9575e+00,
-1.6114e+00, -1.2809e+00],
[-1.5063e+00, -1.8317e+00, -1.8519e+00, ..., -1.2226e+00,
-1.0390e+00, -7.7136e-01]]],
[[[ 2.6574e-01, 5.5599e-01, 7.1113e-01, ..., 4.6843e-01,
1.0118e+00, 5.9756e-01],
[-2.1538e+00, -1.5016e+00, -3.1504e+00, ..., -2.6775e+00,
-8.7728e-01, -2.0341e-01],
[-1.8629e-01, -1.0150e+00, -2.6825e-01, ..., -1.4752e-01,
5.3960e-01, 6.3728e-01],
...,
[-1.8384e-01, -1.1529e+00, -2.6646e+00, ..., -1.5170e+00,
-1.4184e+00, -8.5709e-01],
[ 1.4375e+00, 8.5973e-01, -9.2879e-01, ..., -2.7668e-01,
1.0675e-01, 3.7973e-01],
[ 8.4914e-01, 3.0126e-01, -6.7567e-01, ..., -1.6255e-01,
2.2280e-01, 3.7943e-01]],
[[-1.6295e+00, -1.1111e+00, -4.1482e+00, ..., -4.8852e+00,
-4.4632e+00, -3.4029e+00],
[ 5.7116e-01, 1.9069e+00, -2.2914e+00, ..., -3.5206e+00,
-3.6272e+00, -2.9754e+00],
[ 1.2944e+00, 1.6595e+00, -4.9044e+00, ..., -5.4801e+00,
-5.2865e+00, -4.9428e+00],
...,
[-1.3264e+00, 6.2261e-02, -6.7481e+00, ..., -7.5999e+00,
-6.8982e+00, -5.4069e+00],
[ 1.7231e+00, 2.7460e+00, -1.1073e+00, ..., -2.5927e+00,
-2.9533e+00, -2.3732e+00],
[ 1.1642e+00, 1.7580e+00, -2.2957e+00, ..., -3.2586e+00,
-3.6254e+00, -3.1336e+00]],
[[-9.2858e-02, -1.1169e+00, -1.2793e+00, ..., -5.0027e-01,
-1.6247e-01, -4.8184e-01],
[-5.5861e-01, -8.2565e-01, -7.3728e-01, ..., -2.9215e-01,
3.5015e-02, -7.1564e-02],
[-5.2699e-01, -3.6143e-01, -4.0659e-01, ..., -3.5296e-01,
8.1301e-02, 7.7131e-02],
...,
[-1.3801e+00, -1.7261e+00, -7.7993e-01, ..., -8.8691e-01,
-7.7130e-01, -1.0234e+00],
[-1.6939e+00, -1.1229e+00, -8.0940e-01, ..., -5.6496e-01,
-2.9254e-01, -3.5656e-01],
[-1.8109e+00, -1.2233e+00, -8.6434e-01, ..., -9.5990e-01,
-8.2142e-01, -7.9213e-01]],
...,
[[-2.9652e+00, -2.4399e+00, -2.7215e+00, ..., -3.4865e+00,
-2.6589e+00, -1.7701e+00],
[-2.2869e+00, -8.3899e-01, -1.5665e+00, ..., -2.6371e+00,
-2.3718e+00, -1.8744e+00],
[-1.0607e+00, 1.1581e+00, 5.8560e-01, ..., -4.3102e-01,
-1.5353e-02, 1.3122e-01],
...,
[-1.8106e+00, 1.2449e+00, -2.9492e-03, ..., -1.1166e+00,
-5.3584e-01, 1.8023e-02],
[-8.1460e-01, 1.8831e+00, 1.6102e+00, ..., 4.5968e-01,
4.0319e-01, 6.7765e-01],
[-2.2523e+00, -1.2139e+00, -1.0353e+00, ..., -1.8692e+00,
-9.7330e-01, -3.1038e-01]],
[[-1.6061e+00, -1.6061e+00, -2.9370e+00, ..., -1.2633e+00,
-8.3743e-01, -8.4161e-01],
[-6.4099e-01, 2.5295e-02, -3.2556e+00, ..., -1.3639e+00,
-1.2154e+00, -1.5759e+00],
[-2.0961e+00, -9.2050e-01, -5.3427e+00, ..., -3.1917e+00,
-2.7035e+00, -2.2111e+00],
...,
[-2.5075e+00, -1.6189e+00, -5.4031e+00, ..., -3.0802e+00,
-2.7116e+00, -2.1707e+00],
[-2.2328e+00, -1.4960e+00, -4.1954e+00, ..., -2.6834e+00,
-2.2913e+00, -1.6342e+00],
[-2.3913e+00, -1.7936e+00, -3.3151e+00, ..., -2.9665e+00,
-2.4208e+00, -2.1086e+00]],
[[-1.3449e+00, -1.8980e+00, -1.7005e+00, ..., -1.7859e+00,
-1.4759e+00, -1.1149e+00],
[-1.2812e+00, -1.7619e+00, -1.2471e+00, ..., -1.3877e+00,
-1.0895e+00, -8.6492e-01],
[-1.6116e+00, -1.9702e+00, -1.5067e+00, ..., -1.6156e+00,
-1.2024e+00, -8.8824e-01],
...,
[-3.1658e+00, -3.9625e+00, -3.7686e+00, ..., -3.9373e+00,
-3.1578e+00, -2.4382e+00],
[-2.1106e+00, -2.7251e+00, -2.7056e+00, ..., -2.7619e+00,
-2.2817e+00, -1.8584e+00],
[-1.5348e+00, -1.8613e+00, -1.9302e+00, ..., -1.8662e+00,
-1.5180e+00, -1.1628e+00]]],
[[[-1.9619e-02, 1.3558e-01, -1.9517e+00, ..., 6.2681e-01,
1.0654e+00, 7.3876e-01],
[-2.4228e+00, -1.9324e+00, -5.5135e+00, ..., -1.9202e+00,
-3.5130e-01, 2.4100e-01],
[ 4.7029e+00, -2.2152e+00, -1.8659e+01, ..., 1.1979e-01,
7.2814e-01, 7.6508e-01],
...,
[-2.4023e-01, -1.3129e+00, -2.8151e+00, ..., -1.6059e+00,
-1.1286e+00, -7.8197e-01],
[ 1.2832e+00, 5.9332e-01, -1.4032e+00, ..., 3.1783e-02,
6.7797e-01, 8.1045e-01],
[ 7.1498e-01, 5.5696e-02, -8.7281e-01, ..., 3.7466e-03,
5.1007e-01, 4.6140e-01]],
[[-3.5263e+00, -1.6388e+00, -6.7093e+00, ..., -4.9844e+00,
-4.4368e+00, -3.3964e+00],
[-7.5399e-01, 2.1055e+00, -4.6725e+00, ..., -3.6585e+00,
-3.6855e+00, -3.0007e+00],
[-1.9603e-01, -1.2293e+00, -1.1091e+01, ..., -5.6403e+00,
-5.4336e+00, -5.0527e+00],
...,
[-1.6090e+00, -2.2045e-02, -7.3381e+00, ..., -4.8113e+00,
-4.2610e+00, -3.5922e+00],
[ 1.2620e+00, 2.4563e+00, -1.8629e+00, ..., -9.5859e-01,
-1.2099e+00, -1.3046e+00],
[ 8.9117e-01, 1.6600e+00, -2.6255e+00, ..., -2.2965e+00,
-2.4948e+00, -2.4360e+00]],
[[-1.7458e+00, -2.6292e+00, -2.4038e+00, ..., -6.5508e-01,
-2.9536e-01, -5.1489e-01],
[-2.3030e+00, -2.3850e+00, -1.6688e+00, ..., -4.5245e-01,
-6.1746e-02, -3.4631e-02],
[-2.6592e+00, -4.6686e+00, -5.0023e+00, ..., -2.7323e-01,
1.9707e-01, 2.4359e-01],
...,
[-1.8098e+00, -1.9189e+00, -6.5168e-01, ..., -4.8351e-01,
-2.3153e-01, -3.9733e-01],
[-1.9139e+00, -1.2435e+00, -6.3245e-01, ..., -2.3447e-01,
1.3440e-01, 1.8695e-01],
[-1.9288e+00, -1.3885e+00, -8.0712e-01, ..., -3.6894e-01,
-2.1678e-01, -1.9578e-01]],
...,
[[-2.3580e+00, -1.2886e+00, -2.7173e+00, ..., -3.5408e+00,
-2.7757e+00, -1.8105e+00],
[-2.3622e+00, -2.7256e-01, -1.6921e+00, ..., -2.3321e+00,
-2.1951e+00, -1.6875e+00],
[ 1.1846e+00, 5.2434e+00, 3.1526e+00, ..., -5.2901e-01,
-9.0812e-02, 1.6237e-01],
...,
[-2.0516e+00, 8.1593e-01, 1.2379e-01, ..., -1.3705e+00,
-5.7515e-01, -2.9135e-02],
[-1.0347e+00, 1.5666e+00, 1.6823e+00, ..., -8.1053e-02,
1.6613e-01, 3.5168e-01],
[-2.2912e+00, -1.1867e+00, -6.9400e-01, ..., -2.3088e+00,
-1.3513e+00, -7.9158e-01]],
[[-2.6835e+00, -3.0959e+00, -4.3893e+00, ..., -1.0261e+00,
-5.9843e-01, -5.9149e-01],
[-1.9973e+00, -1.8042e+00, -5.0906e+00, ..., -1.4460e+00,
-1.2612e+00, -1.5804e+00],
[-2.7601e+00, -2.3896e+00, -3.3277e+00, ..., -3.4513e+00,
-2.9199e+00, -2.3172e+00],
...,
[-2.8120e+00, -2.1622e+00, -5.0691e+00, ..., -1.7257e+00,
-1.2452e+00, -9.9089e-01],
[-2.4220e+00, -1.9059e+00, -3.9978e+00, ..., -1.8532e+00,
-1.2356e+00, -7.7658e-01],
[-2.4179e+00, -2.0468e+00, -3.1569e+00, ..., -2.7347e+00,
-1.9373e+00, -1.7275e+00]],
[[-2.1276e+00, -2.9154e+00, -2.7289e+00, ..., -1.7520e+00,
-1.4658e+00, -1.1059e+00],
[-2.1631e+00, -2.9695e+00, -2.4423e+00, ..., -1.3937e+00,
-1.1215e+00, -8.8465e-01],
[-4.1598e+00, -6.8262e+00, -7.3107e+00, ..., -1.5953e+00,
-1.2147e+00, -8.9208e-01],
...,
[-3.3770e+00, -4.2505e+00, -4.0510e+00, ..., -2.8601e+00,
-2.2931e+00, -1.7452e+00],
[-2.2188e+00, -2.9023e+00, -2.8839e+00, ..., -1.9955e+00,
-1.6478e+00, -1.3189e+00],
[-1.6205e+00, -1.9835e+00, -2.0359e+00, ..., -1.2443e+00,
-1.0591e+00, -7.9524e-01]]],
[[[ 4.1843e-01, 8.4873e-01, 5.0986e-01, ..., 4.4750e-01,
1.0007e+00, 6.4528e-01],
[-1.3093e+00, -4.4780e-01, -1.9839e+00, ..., -2.5683e+00,
-7.5621e-01, -6.2721e-02],
[ 8.5960e-01, 6.4513e-01, 3.5246e-01, ..., -2.2929e-01,
5.2153e-01, 6.5464e-01],
...,
[ 3.5700e-02, -5.9957e-01, -1.3488e+00, ..., -1.9498e+00,
-1.3993e+00, -9.2345e-01],
[ 1.3253e+00, 1.1264e+00, 5.7829e-01, ..., -2.7934e-01,
5.0108e-01, 6.9167e-01],
[ 9.8962e-01, 8.8704e-01, 5.9720e-01, ..., -3.1756e-01,
2.8716e-01, 2.7829e-01]],
[[-2.8252e+00, -1.6300e+00, -4.8416e+00, ..., -4.8528e+00,
-4.4156e+00, -3.3559e+00],
[-5.7016e-01, 1.5465e+00, -2.9196e+00, ..., -3.3924e+00,
-3.5330e+00, -2.8863e+00],
[ 2.1823e-01, 1.3730e+00, -5.6358e+00, ..., -5.3717e+00,
-5.2475e+00, -4.8940e+00],
...,
[-9.2813e-01, -2.8791e-03, -4.8646e+00, ..., -4.6016e+00,
-4.1059e+00, -3.4195e+00],
[ 1.6842e+00, 2.2858e+00, -3.0580e-01, ..., -9.1933e-01,
-1.1962e+00, -1.2357e+00],
[ 9.6466e-01, 1.2380e+00, -1.8052e+00, ..., -2.2564e+00,
-2.4575e+00, -2.3531e+00]],
[[-6.9558e-01, -1.8635e+00, -1.7379e+00, ..., -5.9176e-01,
-1.9822e-01, -4.7098e-01],
[-1.4363e+00, -1.9055e+00, -1.2568e+00, ..., -3.3739e-01,
7.4537e-02, 2.9437e-02],
[-1.3417e+00, -1.3492e+00, -7.4448e-01, ..., -2.5719e-01,
2.7596e-01, 2.9228e-01],
...,
[-9.0779e-01, -1.4094e+00, -6.6035e-01, ..., -4.8385e-01,
-1.5850e-01, -3.3386e-01],
[-1.1805e+00, -8.7957e-01, -5.7561e-01, ..., -3.0787e-01,
1.5475e-01, 2.2030e-01],
[-1.2267e+00, -7.1871e-01, -3.2085e-01, ..., -4.9184e-01,
-2.7030e-01, -2.4041e-01]],
...,
[[-2.2544e+00, -1.1801e+00, -2.5415e+00, ..., -3.6786e+00,
-2.8381e+00, -1.8303e+00],
[-6.2759e-01, 1.1934e+00, -7.5993e-01, ..., -2.6561e+00,
-2.4247e+00, -1.8549e+00],
[-2.3633e-01, 2.4766e+00, 7.4288e-01, ..., -5.7199e-01,
-9.1867e-02, 1.5481e-01],
...,
[-1.3267e+00, 1.3260e+00, 9.7272e-04, ..., -1.5764e+00,
-6.9920e-01, -1.3675e-01],
[-4.9172e-01, 1.5047e+00, 7.8924e-01, ..., -1.7974e-01,
8.6479e-02, 3.1623e-01],
[-1.6245e+00, -1.0843e+00, -1.5116e+00, ..., -2.3517e+00,
-1.4421e+00, -9.1180e-01]],
[[-1.4143e+00, -1.3977e+00, -3.0080e+00, ..., -1.1205e+00,
-7.1811e-01, -7.4594e-01],
[-6.6700e-01, -2.2254e-01, -3.8809e+00, ..., -1.3582e+00,
-1.2241e+00, -1.6038e+00],
[-2.1896e+00, -1.5727e+00, -6.4162e+00, ..., -3.2773e+00,
-2.8044e+00, -2.2524e+00],
...,
[-1.8052e+00, -9.4404e-01, -4.6338e+00, ..., -1.5088e+00,
-1.0358e+00, -8.7920e-01],
[-1.7195e+00, -9.9427e-01, -3.7312e+00, ..., -1.7488e+00,
-1.1423e+00, -7.6263e-01],
[-2.0505e+00, -1.4403e+00, -3.0276e+00, ..., -2.8004e+00,
-2.0058e+00, -1.8334e+00]],
[[-1.2521e+00, -1.9042e+00, -1.7508e+00, ..., -1.8021e+00,
-1.5007e+00, -1.1415e+00],
[-1.2463e+00, -1.8104e+00, -1.4001e+00, ..., -1.4009e+00,
-1.1138e+00, -8.9020e-01],
[-1.6130e+00, -1.9927e+00, -1.6147e+00, ..., -1.5997e+00,
-1.2110e+00, -8.9717e-01],
...,
[-2.3410e+00, -2.9684e+00, -2.8534e+00, ..., -2.8615e+00,
-2.2839e+00, -1.7620e+00],
[-1.5783e+00, -2.0339e+00, -2.0380e+00, ..., -1.9747e+00,
-1.6189e+00, -1.3102e+00],
[-1.0756e+00, -1.3228e+00, -1.4405e+00, ..., -1.2245e+00,
-1.0335e+00, -7.9190e-01]]],
[[[ 6.9879e-02, 2.2448e-01, -1.4140e+00, ..., -5.8300e-01,
-1.7598e-01, -4.7939e-02],
[-2.0975e+00, -1.6198e+00, -4.5684e+00, ..., -3.6980e+00,
-1.4886e+00, -5.6576e-01],
[ 5.6798e-01, -9.3321e-02, -8.4745e-01, ..., -2.2098e-01,
-2.6980e-01, 2.3183e-01],
...,
[-3.0907e-01, -1.1269e+00, -1.4272e+00, ..., -1.3324e+00,
-9.1226e-01, -6.9579e-01],
[ 9.8643e-01, 5.5823e-01, 3.4342e-01, ..., 3.9267e-01,
6.8288e-01, 7.7996e-01],
[ 6.8486e-01, 4.1110e-01, 3.4221e-01, ..., 5.1397e-01,
7.6424e-01, 6.9102e-01]],
[[-3.3150e+00, -1.8344e+00, -6.2716e+00, ..., -7.1280e+00,
-6.2497e+00, -4.6135e+00],
[-8.0597e-01, 1.5027e+00, -4.3466e+00, ..., -5.4924e+00,
-5.5915e+00, -4.2811e+00],
[-3.7463e-01, 1.0330e+00, -7.4175e+00, ..., -7.9965e+00,
-7.8657e+00, -6.6507e+00],
...,
[-4.8895e-01, 2.2711e-01, -4.4520e+00, ..., -5.3672e+00,
-4.6403e+00, -3.8596e+00],
[ 1.9356e+00, 2.4174e+00, -8.2329e-02, ..., -1.1195e+00,
-1.2291e+00, -1.3472e+00],
[ 1.1104e+00, 1.3004e+00, -1.7023e+00, ..., -2.4472e+00,
-2.5620e+00, -2.5401e+00]],
[[-1.3596e+00, -2.3657e+00, -2.3291e+00, ..., -1.7467e+00,
-1.2191e+00, -1.3488e+00],
[-1.7113e+00, -1.9873e+00, -1.5349e+00, ..., -1.2303e+00,
-7.0641e-01, -8.0603e-01],
[-2.1717e+00, -1.9067e+00, -1.0110e+00, ..., -1.1032e+00,
-4.6168e-01, -1.8309e-01],
...,
[-6.3738e-01, -1.0913e+00, -5.6833e-01, ..., -6.1035e-01,
-4.7338e-01, -6.2234e-01],
[-1.0252e+00, -6.8212e-01, -5.5377e-01, ..., -2.5740e-01,
-5.7042e-02, -3.2530e-02],
[-1.1708e+00, -6.6001e-01, -3.8167e-01, ..., -3.2400e-01,
-2.6621e-01, -2.3680e-01]],
...,
[[-2.0390e+00, -1.1111e+00, -2.3456e+00, ..., -3.1284e+00,
-2.5595e+00, -1.7854e+00],
[-1.7387e+00, 3.3668e-02, -1.3574e+00, ..., -2.4719e+00,
-2.5147e+00, -1.9516e+00],
[-3.7007e-01, 2.3496e+00, 1.0699e+00, ..., 1.2224e-01,
2.5592e-01, 5.1106e-01],
...,
[-1.5569e+00, 8.9703e-01, -1.8741e-01, ..., -1.1700e+00,
-4.4264e-01, 6.6256e-02],
[-5.6736e-01, 1.3492e+00, 7.0842e-01, ..., -1.2586e-03,
1.8741e-01, 3.0535e-01],
[-1.6282e+00, -1.1010e+00, -1.5564e+00, ..., -2.3105e+00,
-1.3432e+00, -7.1417e-01]],
[[-2.4283e+00, -2.8256e+00, -4.1842e+00, ..., -2.4396e+00,
-1.9352e+00, -1.7301e+00],
[-1.7727e+00, -1.6358e+00, -4.8415e+00, ..., -2.9664e+00,
-2.7826e+00, -2.7522e+00],
[-2.8501e+00, -2.4903e+00, -6.6179e+00, ..., -4.4813e+00,
-4.3475e+00, -3.4497e+00],
...,
[-1.7206e+00, -5.9025e-01, -4.0233e+00, ..., -2.0848e+00,
-1.5393e+00, -1.2023e+00],
[-1.7050e+00, -7.6035e-01, -3.3301e+00, ..., -2.0546e+00,
-1.3338e+00, -8.0021e-01],
[-2.0657e+00, -1.2905e+00, -2.8208e+00, ..., -2.6829e+00,
-1.7856e+00, -1.5272e+00]],
[[-1.8063e+00, -2.5380e+00, -2.4608e+00, ..., -2.6946e+00,
-2.2132e+00, -1.6934e+00],
[-1.7992e+00, -2.5094e+00, -2.1554e+00, ..., -2.4133e+00,
-1.9130e+00, -1.5496e+00],
[-2.2473e+00, -2.8125e+00, -2.5224e+00, ..., -2.7677e+00,
-2.1625e+00, -1.6385e+00],
...,
[-2.2636e+00, -2.8487e+00, -2.6625e+00, ..., -2.8713e+00,
-2.2971e+00, -1.7305e+00],
[-1.5279e+00, -1.9748e+00, -1.9351e+00, ..., -2.0347e+00,
-1.6733e+00, -1.3291e+00],
[-1.0351e+00, -1.2736e+00, -1.3415e+00, ..., -1.2651e+00,
-1.0841e+00, -8.0091e-01]]]], device='cuda:0'),
tensor([[[[ 3.7874e-01, 7.7458e-01, 2.5922e-01, ..., 3.9957e-01,
7.3890e-01, 5.7903e-01],
[-1.3150e+00, -4.4733e-01, -2.0485e+00, ..., -1.5843e+00,
-1.7892e-01, 3.8649e-01],
[ 8.4847e-01, 5.6074e-01, 3.8291e-01, ..., 5.7309e-01,
7.8057e-01, 7.9845e-01],
...,
[-7.1734e-02, -9.4405e-01, -2.6206e+01, ..., -1.6324e+00,
-1.5730e+00, -8.3744e-01],
[ 1.4494e+00, 4.6867e-01, -1.8052e+01, ..., -7.4318e-01,
-2.6535e-01, 1.5259e-01],
[ 6.8624e-01, -6.7248e-01, -1.2060e+01, ..., -6.8737e-01,
-1.1013e-01, 2.1468e-01]],
[[-2.6748e+00, -1.3584e+00, -4.7508e+00, ..., -5.6999e+00,
-4.9131e+00, -3.8018e+00],
[-4.1928e-01, 1.7272e+00, -2.9787e+00, ..., -4.5214e+00,
-4.3874e+00, -3.5648e+00],
[ 2.9070e-01, 1.4570e+00, -5.7054e+00, ..., -6.5813e+00,
-6.2874e+00, -5.6799e+00],
...,
[-1.7037e+00, -4.7518e-01, -3.2382e+00, ..., -8.4984e+00,
-7.8914e+00, -6.0260e+00],
[ 1.4732e+00, 2.7067e+00, 4.4204e+00, ..., -3.1381e+00,
-3.6548e+00, -2.7281e+00],
[ 1.0855e+00, 2.0286e+00, 1.9998e+00, ..., -3.5388e+00,
-4.0539e+00, -3.3453e+00]],
[[-8.2497e-01, -1.9467e+00, -1.8339e+00, ..., -1.0799e+00,
-7.5044e-01, -8.5434e-01],
[-1.4686e+00, -1.8722e+00, -1.2600e+00, ..., -7.6352e-01,
-4.1419e-01, -3.2936e-01],
[-1.4219e+00, -1.3033e+00, -5.6127e-01, ..., -4.6847e-01,
-8.5974e-02, 2.6244e-02],
...,
[-1.7419e+00, -3.9628e+00, -8.0388e+00, ..., -8.6796e-01,
-7.8473e-01, -1.1118e+00],
[-1.9413e+00, -1.3464e+00, -4.1332e+00, ..., -5.1939e-01,
-2.6725e-01, -4.0062e-01],
[-2.1543e+00, -1.6500e+00, -2.4262e+00, ..., -1.3532e+00,
-1.2291e+00, -1.2003e+00]],
...,
[[-2.1624e+00, -1.0319e+00, -2.3188e+00, ..., -2.8877e+00,
-2.3276e+00, -1.5697e+00],
[-7.9330e-01, 1.1728e+00, -5.8999e-01, ..., -1.5555e+00,
-1.6236e+00, -1.3714e+00],
[-2.2175e-01, 2.5822e+00, 9.1515e-01, ..., -8.6391e-02,
2.1294e-01, 3.6785e-01],
...,
[-1.9158e+00, 4.3954e+00, 3.8864e+00, ..., -1.1325e+00,
-6.5827e-01, -1.0254e-01],
[-8.8717e-01, 1.9872e+00, -1.1896e+00, ..., 7.8473e-01,
5.4001e-01, 8.3814e-01],
[-2.3227e+00, -1.1452e+00, -1.3111e-01, ..., -1.5537e+00,
-7.0441e-01, -5.4296e-02]],
[[-1.6098e+00, -1.6321e+00, -3.1570e+00, ..., -1.4704e+00,
-9.6616e-01, -7.8964e-01],
[-8.5128e-01, -4.0411e-01, -3.9062e+00, ..., -2.0892e+00,
-1.8369e+00, -1.9001e+00],
[-2.1718e+00, -1.5416e+00, -6.1866e+00, ..., -3.9303e+00,
-3.4734e+00, -2.7080e+00],
...,
[-2.7021e+00, -2.0911e+00, -6.9449e+00, ..., -3.2613e+00,
-3.0437e+00, -2.4325e+00],
[-2.4188e+00, -1.7340e+00, -3.0213e+00, ..., -2.8320e+00,
-2.6355e+00, -1.9509e+00],
[-2.4924e+00, -1.9001e+00, -1.5541e+00, ..., -3.0743e+00,
-2.6877e+00, -2.3767e+00]],
[[-1.2868e+00, -1.8887e+00, -1.6962e+00, ..., -1.8360e+00,
-1.5197e+00, -1.1455e+00],
[-1.2655e+00, -1.7952e+00, -1.3414e+00, ..., -1.5504e+00,
-1.2527e+00, -9.9281e-01],
[-1.6409e+00, -1.9776e+00, -1.5397e+00, ..., -1.7724e+00,
-1.3614e+00, -9.9554e-01],
...,
[-3.5650e+00, -5.3985e+00, -1.0324e+01, ..., -4.4388e+00,
-3.5688e+00, -2.7727e+00],
[-2.3205e+00, -3.0925e+00, -6.1447e+00, ..., -3.0890e+00,
-2.5512e+00, -2.0912e+00],
[-1.7162e+00, -2.1971e+00, -4.8878e+00, ..., -2.1187e+00,
-1.6927e+00, -1.3138e+00]]],
[[[ 3.6051e-01, 8.3073e-01, 7.6342e-01, ..., 7.0442e-01,
9.4677e-01, 7.3475e-01],
[-1.2521e+00, -3.1605e-01, -1.3970e+00, ..., -1.0633e+00,
6.8282e-02, 5.5446e-01],
[ 6.4193e-01, 4.1405e-01, 5.1776e-01, ..., 5.9026e-01,
9.4448e-01, 8.6803e-01],
...,
[-1.7602e-01, -1.0196e+00, -2.4534e+00, ..., -1.7035e+00,
-1.6232e+00, -1.0555e+00],
[ 1.3143e+00, 7.9665e-01, -9.3801e-01, ..., -2.9624e-01,
8.9095e-02, 3.6312e-01],
[ 7.7515e-01, 2.6952e-01, -6.6058e-01, ..., -1.6626e-01,
2.4305e-01, 3.9066e-01]],
[[-2.3339e+00, -1.2161e+00, -4.2405e+00, ..., -5.2448e+00,
-4.4906e+00, -3.5281e+00],
[-1.0968e-01, 1.7635e+00, -2.5380e+00, ..., -4.1062e+00,
-3.9863e+00, -3.2845e+00],
[ 7.8058e-01, 1.6124e+00, -5.1162e+00, ..., -5.9603e+00,
-5.7450e+00, -5.2954e+00],
...,
[-1.5413e+00, -2.8219e-01, -6.8722e+00, ..., -7.5648e+00,
-6.8590e+00, -5.3595e+00],
[ 1.3668e+00, 2.3080e+00, -1.4942e+00, ..., -2.2772e+00,
-2.6963e+00, -2.1638e+00],
[ 9.0904e-01, 1.4913e+00, -2.4544e+00, ..., -3.1459e+00,
-3.5679e+00, -3.1138e+00]],
[[-6.7781e-01, -1.8448e+00, -1.7287e+00, ..., -9.2464e-01,
-6.6727e-01, -7.6741e-01],
[-1.4020e+00, -1.8692e+00, -1.2866e+00, ..., -7.2740e-01,
-3.9120e-01, -2.7124e-01],
[-1.0807e+00, -1.0089e+00, -4.6335e-01, ..., -3.2797e-01,
2.7574e-02, 4.4041e-02],
...,
[-1.2002e+00, -1.5655e+00, -7.3310e-01, ..., -9.2405e-01,
-7.9026e-01, -1.0521e+00],
[-1.5094e+00, -1.0192e+00, -7.7425e-01, ..., -5.7293e-01,
-2.7867e-01, -3.4752e-01],
[-1.6499e+00, -1.1624e+00, -8.6682e-01, ..., -9.4475e-01,
-8.0930e-01, -7.8711e-01]],
...,
[[-2.2720e+00, -1.2920e+00, -2.3301e+00, ..., -2.8886e+00,
-2.2584e+00, -1.4954e+00],
[-6.3735e-01, 1.1089e+00, -3.9349e-01, ..., -1.3324e+00,
-1.3887e+00, -1.2045e+00],
[-3.5173e-01, 2.1579e+00, 8.7859e-01, ..., -1.3266e-01,
2.2992e-01, 3.5309e-01],
...,
[-1.7207e+00, 9.9947e-01, -7.8564e-02, ..., -1.1750e+00,
-5.2816e-01, 4.6660e-02],
[-8.0039e-01, 1.6632e+00, 1.4644e+00, ..., 4.7822e-01,
4.5521e-01, 7.3644e-01],
[-2.1801e+00, -1.1909e+00, -1.0065e+00, ..., -2.0836e+00,
-1.0883e+00, -3.6558e-01]],
[[-1.3389e+00, -1.2356e+00, -2.7348e+00, ..., -1.1011e+00,
-5.9908e-01, -4.6292e-01],
[-6.1263e-01, -7.3684e-02, -3.5678e+00, ..., -1.7718e+00,
-1.5129e+00, -1.6863e+00],
[-2.0503e+00, -1.2788e+00, -5.9524e+00, ..., -3.7100e+00,
-3.1848e+00, -2.5457e+00],
...,
[-2.5361e+00, -1.8169e+00, -5.2541e+00, ..., -2.9719e+00,
-2.5934e+00, -2.0840e+00],
[-2.2578e+00, -1.6361e+00, -4.1410e+00, ..., -2.6272e+00,
-2.2083e+00, -1.5428e+00],
[-2.3608e+00, -1.8682e+00, -3.2743e+00, ..., -3.0318e+00,
-2.4285e+00, -2.1368e+00]],
[[-1.1714e+00, -1.7313e+00, -1.4961e+00, ..., -1.6298e+00,
-1.3626e+00, -1.0186e+00],
[-1.1767e+00, -1.6500e+00, -1.1542e+00, ..., -1.3628e+00,
-1.1153e+00, -8.8056e-01],
[-1.5332e+00, -1.8182e+00, -1.3325e+00, ..., -1.5600e+00,
-1.1907e+00, -8.6919e-01],
...,
[-3.0376e+00, -3.8146e+00, -3.6870e+00, ..., -4.0242e+00,
-3.2305e+00, -2.4937e+00],
[-2.0220e+00, -2.6187e+00, -2.6413e+00, ..., -2.8121e+00,
-2.3279e+00, -1.8976e+00],
[-1.4766e+00, -1.7984e+00, -1.8955e+00, ..., -1.8747e+00,
-1.5337e+00, -1.1744e+00]]],
[[[ 8.1552e-02, 2.2302e-01, -1.3982e+00, ..., -6.1956e-01,
-1.9700e-01, -5.4500e-02],
[-2.0903e+00, -1.6231e+00, -4.5218e+00, ..., -3.7699e+00,
-1.5427e+00, -5.6884e-01],
[ 5.7987e-01, -9.9730e-02, -8.4834e-01, ..., -1.8340e-01,
-2.6827e-01, 2.3402e-01],
...,
[-1.1218e+00, -4.8003e+00, -7.3718e+00, ..., -1.2780e+00,
-8.1908e-01, -5.9911e-01],
[ 6.2775e-02, -2.6599e+00, -5.2380e+00, ..., 3.6836e-01,
6.8880e-01, 8.0889e-01],
[-1.0501e-01, -2.8923e+00, -5.0699e+00, ..., 4.9395e-01,
7.5594e-01, 7.1432e-01]],
[[-3.3067e+00, -1.7914e+00, -6.2423e+00, ..., -7.1438e+00,
-6.2421e+00, -4.6176e+00],
[-7.8318e-01, 1.5559e+00, -4.3190e+00, ..., -5.4717e+00,
-5.5229e+00, -4.2649e+00],
[-3.3876e-01, 1.0887e+00, -7.4027e+00, ..., -7.9730e+00,
-7.7954e+00, -6.6700e+00],
...,
[-2.7195e-01, 8.1963e-01, -4.8745e+00, ..., -5.3272e+00,
-4.6535e+00, -3.9035e+00],
[ 2.3778e+00, 3.0367e+00, -2.7024e-01, ..., -1.1318e+00,
-1.2882e+00, -1.4298e+00],
[ 1.4552e+00, 1.9901e+00, -1.4893e+00, ..., -2.4686e+00,
-2.6013e+00, -2.5803e+00]],
[[-1.3461e+00, -2.3587e+00, -2.3159e+00, ..., -1.7589e+00,
-1.2438e+00, -1.3785e+00],
[-1.7081e+00, -2.0031e+00, -1.5266e+00, ..., -1.2573e+00,
-7.3220e-01, -8.1623e-01],
[-2.1814e+00, -1.9364e+00, -1.0089e+00, ..., -1.0953e+00,
-4.4579e-01, -1.8858e-01],
...,
[-6.9940e-01, -1.4445e+00, -6.0373e-01, ..., -6.1002e-01,
-4.8890e-01, -6.3542e-01],
[-1.2548e+00, -7.8160e-01, -1.2404e-01, ..., -2.6450e-01,
-7.4733e-02, -4.0476e-02],
[-1.5700e+00, -9.0711e-01, -5.0543e-01, ..., -3.2679e-01,
-2.8869e-01, -2.4725e-01]],
...,
[[-2.0670e+00, -1.1055e+00, -2.3776e+00, ..., -3.1845e+00,
-2.6096e+00, -1.8168e+00],
[-1.7574e+00, 5.7231e-02, -1.3537e+00, ..., -2.4690e+00,
-2.4852e+00, -1.9269e+00],
[-3.8058e-01, 2.3736e+00, 1.0716e+00, ..., 9.1912e-02,
2.8691e-01, 5.5387e-01],
...,
[-1.6614e+00, 7.2612e-01, -1.1703e+00, ..., -1.1050e+00,
-4.5386e-01, 2.9944e-02],
[-1.2242e+00, 4.6237e-01, -1.3421e+00, ..., 4.9100e-02,
1.9064e-01, 2.7453e-01],
[-2.3513e+00, -2.4849e+00, -4.4412e+00, ..., -2.2363e+00,
-1.2992e+00, -6.9163e-01]],
[[-2.4176e+00, -2.8170e+00, -4.2062e+00, ..., -2.3914e+00,
-1.8909e+00, -1.6915e+00],
[-1.7529e+00, -1.6130e+00, -4.8704e+00, ..., -2.9154e+00,
-2.7629e+00, -2.7521e+00],
[-2.8332e+00, -2.4501e+00, -6.6695e+00, ..., -4.4725e+00,
-4.3542e+00, -3.4771e+00],
...,
[-1.5516e+00, 4.1467e-01, -3.4423e+00, ..., -2.0776e+00,
-1.5679e+00, -1.2252e+00],
[-1.4486e+00, 2.5247e-01, -2.2480e+00, ..., -2.0307e+00,
-1.3762e+00, -8.2287e-01],
[-1.8462e+00, -3.6474e-01, -1.0935e+00, ..., -2.6471e+00,
-1.8156e+00, -1.5165e+00]],
[[-1.8038e+00, -2.5388e+00, -2.4501e+00, ..., -2.6910e+00,
-2.2113e+00, -1.6932e+00],
[-1.7943e+00, -2.5112e+00, -2.1402e+00, ..., -2.4003e+00,
-1.8990e+00, -1.5370e+00],
[-2.2444e+00, -2.8077e+00, -2.5013e+00, ..., -2.7426e+00,
-2.1416e+00, -1.6239e+00],
...,
[-2.5279e+00, -3.3123e+00, -3.1960e+00, ..., -2.8550e+00,
-2.2649e+00, -1.7096e+00],
[-1.7425e+00, -2.4386e+00, -2.4601e+00, ..., -2.0226e+00,
-1.6557e+00, -1.3151e+00],
[-1.1927e+00, -1.7368e+00, -1.8678e+00, ..., -1.2623e+00,
-1.0691e+00, -7.9056e-01]]],
[[[ 2.1496e-01, 4.8385e-01, 6.3534e-01, ..., -3.1934e-01,
-6.5377e-02, 1.6345e-02],
[-2.0282e+00, -1.3517e+00, -2.8608e+00, ..., -3.1910e+00,
-1.2633e+00, -4.3728e-01],
[-1.5723e-01, -9.2167e-01, -9.0090e-03, ..., -1.6088e-01,
-3.2376e-01, 1.7733e-01],
...,
[-9.8117e-01, -2.1758e+00, -2.0840e+00, ..., -1.9239e+00,
-1.8558e+00, -1.2604e+00],
[ 7.7245e-01, -1.4815e-01, -1.4494e-01, ..., 3.1583e-02,
2.8503e-01, 4.5978e-01],
[ 1.8253e-01, -5.9415e-01, -4.0794e-01, ..., 2.4135e-01,
4.8713e-01, 5.0568e-01]],
[[-1.6737e+00, -1.3692e+00, -4.2512e+00, ..., -6.8473e+00,
-6.0692e+00, -4.5323e+00],
[ 4.9733e-01, 1.4892e+00, -2.5100e+00, ..., -5.4587e+00,
-5.5607e+00, -4.2914e+00],
[ 1.2341e+00, 1.2239e+00, -5.0541e+00, ..., -7.8748e+00,
-7.8303e+00, -6.6090e+00],
...,
[ 3.5660e-01, 2.7897e-01, -4.1305e+00, ..., -7.0776e+00,
-6.6666e+00, -5.1643e+00],
[ 2.4737e+00, 2.4232e+00, 7.8187e-02, ..., -1.9130e+00,
-2.5498e+00, -2.0404e+00],
[ 1.4598e+00, 1.2866e+00, -1.6296e+00, ..., -2.9265e+00,
-3.4734e+00, -3.0431e+00]],
[[-1.4373e-01, -1.1773e+00, -1.4133e+00, ..., -1.4088e+00,
-1.0632e+00, -1.3425e+00],
[-5.2192e-01, -8.3702e-01, -8.8766e-01, ..., -9.8707e-01,
-6.2294e-01, -8.9894e-01],
[-3.3941e-01, -2.2549e-01, -4.3652e-01, ..., -1.0933e+00,
-5.8822e-01, -4.7008e-01],
...,
[-4.0615e-02, -5.4770e-01, -6.0291e-01, ..., -1.0423e+00,
-8.7353e-01, -1.2399e+00],
[-6.6694e-01, -1.6646e-01, -5.4990e-01, ..., -5.6672e-01,
-2.5138e-01, -4.1060e-01],
[-8.1338e-01, -2.1279e-01, -4.5581e-01, ..., -5.0082e-01,
-4.4291e-01, -5.9650e-01]],
...,
[[-2.8602e+00, -2.4632e+00, -2.4555e+00, ..., -3.1002e+00,
-2.5211e+00, -1.8311e+00],
[-2.1243e+00, -8.5804e-01, -1.1377e+00, ..., -2.4220e+00,
-2.4919e+00, -2.0064e+00],
[-1.1110e+00, 8.1630e-01, 7.7052e-01, ..., -1.0049e-01,
3.1831e-02, 2.9722e-01],
...,
[-2.2465e+00, -2.4253e-01, -1.6501e-01, ..., -9.4761e-01,
-2.6231e-01, 2.8631e-01],
[-1.2418e+00, 8.0270e-01, 1.0448e+00, ..., 4.0051e-01,
4.6365e-01, 8.0028e-01],
[-2.2102e+00, -1.5556e+00, -1.7421e+00, ..., -2.2261e+00,
-1.1839e+00, -3.7184e-01]],
[[-1.6432e+00, -1.6419e+00, -2.8158e+00, ..., -2.2316e+00,
-1.9120e+00, -1.8017e+00],
[-7.0019e-01, -8.3519e-02, -3.1470e+00, ..., -2.8724e+00,
-2.8797e+00, -2.8934e+00],
[-2.0731e+00, -9.6952e-01, -5.1246e+00, ..., -4.4869e+00,
-4.5314e+00, -3.7322e+00],
...,
[-1.3495e+00, 1.8009e-01, -3.2831e+00, ..., -2.8273e+00,
-2.7207e+00, -2.2939e+00],
[-1.6878e+00, -4.7009e-01, -2.9542e+00, ..., -2.4560e+00,
-2.2410e+00, -1.6487e+00],
[-2.3311e+00, -1.3746e+00, -3.0365e+00, ..., -2.8765e+00,
-2.4114e+00, -2.1823e+00]],
[[-1.3079e+00, -1.8375e+00, -1.6543e+00, ..., -2.5669e+00,
-2.1491e+00, -1.6642e+00],
[-1.2456e+00, -1.7127e+00, -1.2231e+00, ..., -2.3972e+00,
-1.9331e+00, -1.5764e+00],
[-1.5636e+00, -1.9106e+00, -1.4813e+00, ..., -2.8101e+00,
-2.2450e+00, -1.7370e+00],
...,
[-2.4427e+00, -3.0503e+00, -2.7991e+00, ..., -3.9918e+00,
-3.2424e+00, -2.5462e+00],
[-1.5755e+00, -2.0744e+00, -1.9868e+00, ..., -2.7552e+00,
-2.3104e+00, -1.9136e+00],
[-1.0216e+00, -1.3322e+00, -1.3576e+00, ..., -1.8190e+00,
-1.5197e+00, -1.1777e+00]]],
[[[-1.4527e-01, -8.3429e-02, -2.3833e+00, ..., -2.5950e+01,
-4.1696e+00, -3.8905e-01],
[-2.5176e+00, -2.2430e+00, -5.9616e+00, ..., -3.7015e+01,
-5.6708e+00, -9.9958e-01],
[ 3.2242e-01, -5.2262e-01, -1.6349e+00, ..., -3.4896e+01,
-4.2937e+00, -3.0195e-01],
...,
[-1.2802e-01, -7.2759e-01, -1.4209e+00, ..., -1.2043e+00,
-9.3664e-01, -6.6864e-01],
[ 1.2003e+00, 9.6165e-01, 2.7407e-01, ..., 4.4926e-01,
7.5433e-01, 8.4253e-01],
[ 8.6528e-01, 6.8212e-01, 2.8366e-01, ..., 5.4418e-01,
7.9873e-01, 7.2237e-01]],
[[-3.6555e+00, -2.0359e+00, -6.9636e+00, ..., -1.2938e+01,
-7.1110e+00, -5.0202e+00],
[-1.0379e+00, 1.4640e+00, -4.9431e+00, ..., -9.2064e+00,
-5.8474e+00, -4.7235e+00],
[-7.1556e-01, 9.9393e-01, -8.1231e+00, ..., -1.7559e+01,
-1.1360e+01, -7.4946e+00],
...,
[-9.2621e-01, 5.4168e-02, -5.0616e+00, ..., -5.9107e+00,
-5.1531e+00, -4.2527e+00],
[ 1.6347e+00, 2.3014e+00, -5.0961e-01, ..., -1.4608e+00,
-1.6285e+00, -1.6210e+00],
[ 9.4984e-01, 1.2736e+00, -1.9695e+00, ..., -2.6726e+00,
-2.8478e+00, -2.7355e+00]],
[[-1.7517e+00, -2.6564e+00, -2.5865e+00, ..., -4.3952e+00,
-1.8563e+00, -1.6279e+00],
[-1.9862e+00, -2.1787e+00, -1.6650e+00, ..., -2.2460e+00,
-7.6746e-01, -1.0874e+00],
[-2.7681e+00, -2.3890e+00, -1.1712e+00, ..., -2.1555e+00,
-5.3338e-01, -2.0002e-01],
...,
[-8.8797e-01, -1.3640e+00, -6.0693e-01, ..., -6.0505e-01,
-4.8953e-01, -6.7156e-01],
[-1.2554e+00, -9.3191e-01, -6.7581e-01, ..., -3.2663e-01,
-9.8003e-02, -1.0191e-01],
[-1.3370e+00, -8.4350e-01, -4.9274e-01, ..., -4.1571e-01,
-3.3272e-01, -3.0860e-01]],
...,
[[-1.9934e+00, -1.0910e+00, -2.3340e+00, ..., -4.3814e+00,
-2.1459e+00, -1.8271e+00],
[-2.3763e+00, -5.9016e-01, -1.7874e+00, ..., -1.6233e+00,
-1.9671e+00, -2.2142e+00],
[-6.2906e-01, 2.1604e+00, 1.0546e+00, ..., 3.3056e+00,
-6.1934e-01, 1.5284e-02],
...,
[-1.3096e+00, 1.1683e+00, -2.1659e-01, ..., -1.2083e+00,
-4.6354e-01, 5.2473e-02],
[-4.6582e-01, 1.5371e+00, 8.1345e-01, ..., -5.6693e-03,
1.8771e-01, 3.7780e-01],
[-1.6656e+00, -1.0394e+00, -1.4349e+00, ..., -2.3704e+00,
-1.3422e+00, -6.6390e-01]],
[[-2.8117e+00, -3.4103e+00, -4.6310e+00, ..., -5.8233e+00,
-2.5116e+00, -2.2402e+00],
[-2.1982e+00, -2.2398e+00, -5.2691e+00, ..., -5.8056e+00,
-3.3958e+00, -3.2164e+00],
[-3.1312e+00, -2.9325e+00, -6.8305e+00, ..., 9.0125e-01,
-2.8163e+00, -3.8194e+00],
...,
[-1.9338e+00, -1.0461e+00, -4.5910e+00, ..., -2.5277e+00,
-1.9320e+00, -1.5000e+00],
[-1.8462e+00, -1.1163e+00, -3.7359e+00, ..., -2.3641e+00,
-1.6543e+00, -1.0548e+00],
[-2.1341e+00, -1.5276e+00, -3.0663e+00, ..., -2.8641e+00,
-1.9967e+00, -1.7156e+00]],
[[-2.0974e+00, -2.9129e+00, -2.8673e+00, ..., -6.5737e+00,
-2.7996e+00, -1.9558e+00],
[-2.1130e+00, -2.9394e+00, -2.6014e+00, ..., -6.2998e+00,
-2.4109e+00, -1.8314e+00],
[-2.6219e+00, -3.3349e+00, -3.0599e+00, ..., -8.0195e+00,
-3.4493e+00, -2.0404e+00],
...,
[-2.3910e+00, -2.9771e+00, -2.8080e+00, ..., -3.0857e+00,
-2.4664e+00, -1.8695e+00],
[-1.6208e+00, -2.0666e+00, -2.0419e+00, ..., -2.1811e+00,
-1.7992e+00, -1.4454e+00],
[-1.1338e+00, -1.3695e+00, -1.4438e+00, ..., -1.3983e+00,
-1.1842e+00, -8.8271e-01]]]], device='cuda:0'),
tensor([[[[ 4.6234e-01, 9.5220e-01, 8.7098e-01, ..., -3.3527e-01,
-8.4537e-02, 9.5737e-03],
[-1.2491e+00, -3.4292e-01, -1.4768e+00, ..., -3.2283e+00,
-1.2931e+00, -4.5778e-01],
[ 5.6741e-01, 4.0021e-01, 3.3172e-01, ..., -1.4644e-01,
-3.6945e-01, 1.5335e-01],
...,
[-1.8223e-01, -8.1779e-01, -1.4915e+00, ..., -1.9607e+00,
-1.8857e+00, -1.2656e+00],
[ 1.0896e+00, 8.2666e-01, 3.1111e-01, ..., -2.3313e-02,
3.0442e-01, 4.7149e-01],
[ 7.1480e-01, 5.1738e-01, 1.8489e-01, ..., 1.8087e-01,
4.8557e-01, 5.0220e-01]],
[[-2.2778e+00, -1.0693e+00, -4.0915e+00, ..., -6.8531e+00,
-6.0537e+00, -4.5240e+00],
[-2.9523e-01, 1.7052e+00, -2.5155e+00, ..., -5.4535e+00,
-5.5332e+00, -4.2839e+00],
[ 3.9766e-01, 1.4952e+00, -5.0262e+00, ..., -7.8403e+00,
-7.7654e+00, -6.5840e+00],
...,
[-4.8649e-01, 6.0685e-01, -4.0519e+00, ..., -7.1018e+00,
-6.7101e+00, -5.1904e+00],
[ 2.1012e+00, 2.8037e+00, 4.2398e-01, ..., -1.9268e+00,
-2.5891e+00, -2.0621e+00],
[ 1.2296e+00, 1.5346e+00, -1.4387e+00, ..., -2.9398e+00,
-3.4996e+00, -3.0456e+00]],
[[-5.5908e-01, -1.5530e+00, -1.3769e+00, ..., -1.4110e+00,
-1.0471e+00, -1.3310e+00],
[-1.2666e+00, -1.6208e+00, -9.8823e-01, ..., -1.0224e+00,
-6.3005e-01, -8.8895e-01],
[-1.1212e+00, -9.9275e-01, -3.7867e-01, ..., -1.0963e+00,
-5.9493e-01, -4.6428e-01],
...,
[-9.1626e-01, -1.3408e+00, -5.4578e-01, ..., -1.0292e+00,
-8.7047e-01, -1.2547e+00],
[-1.2795e+00, -8.7131e-01, -5.4727e-01, ..., -5.5832e-01,
-2.4731e-01, -4.2469e-01],
[-1.3758e+00, -7.7079e-01, -3.5061e-01, ..., -5.0739e-01,
-4.4228e-01, -6.1360e-01]],
...,
[[-2.1615e+00, -1.2251e+00, -2.5383e+00, ..., -3.0641e+00,
-2.4950e+00, -1.8155e+00],
[-6.9805e-01, 9.8066e-01, -9.2228e-01, ..., -2.3865e+00,
-2.4679e+00, -2.0198e+00],
[-2.3388e-01, 2.2575e+00, 5.5965e-01, ..., -1.0195e-01,
1.3575e-02, 2.7054e-01],
...,
[-1.5251e+00, 1.2310e+00, -1.9972e-01, ..., -8.9895e-01,
-2.9442e-01, 2.8607e-01],
[-5.2113e-01, 1.5743e+00, 8.2014e-01, ..., 4.8754e-01,
4.7732e-01, 8.3400e-01],
[-1.6468e+00, -1.1040e+00, -1.6130e+00, ..., -2.1656e+00,
-1.1734e+00, -3.5262e-01]],
[[-1.2936e+00, -1.2013e+00, -2.5968e+00, ..., -2.2499e+00,
-1.9067e+00, -1.7934e+00],
[-6.2650e-01, -1.2460e-01, -3.3256e+00, ..., -2.8532e+00,
-2.8482e+00, -2.8690e+00],
[-2.0750e+00, -1.3958e+00, -5.6890e+00, ..., -4.4604e+00,
-4.4911e+00, -3.7000e+00],
...,
[-1.6364e+00, -5.0418e-01, -4.0811e+00, ..., -2.8317e+00,
-2.7508e+00, -2.3298e+00],
[-1.6525e+00, -7.1363e-01, -3.4113e+00, ..., -2.4527e+00,
-2.2518e+00, -1.6709e+00],
[-2.1074e+00, -1.3634e+00, -2.9726e+00, ..., -2.8670e+00,
-2.4228e+00, -2.1893e+00]],
[[-1.1449e+00, -1.7009e+00, -1.5105e+00, ..., -2.5789e+00,
-2.1488e+00, -1.6691e+00],
[-1.1579e+00, -1.6252e+00, -1.2065e+00, ..., -2.4070e+00,
-1.9387e+00, -1.5825e+00],
[-1.5067e+00, -1.8104e+00, -1.4171e+00, ..., -2.8269e+00,
-2.2503e+00, -1.7406e+00],
...,
[-2.3064e+00, -2.8783e+00, -2.6650e+00, ..., -4.0104e+00,
-3.2428e+00, -2.5464e+00],
[-1.5460e+00, -1.9718e+00, -1.9023e+00, ..., -2.7778e+00,
-2.3205e+00, -1.9181e+00],
[-1.0390e+00, -1.2398e+00, -1.2946e+00, ..., -1.8417e+00,
-1.5273e+00, -1.1861e+00]]],
[[[ 1.6141e-02, 1.5002e-01, -1.6404e+00, ..., -7.2388e-01,
-3.1522e-01, -1.2960e-01],
[-2.2058e+00, -1.7423e+00, -4.8411e+00, ..., -3.9077e+00,
-1.6502e+00, -6.5288e-01],
[ 4.9879e-01, -2.0163e-01, -1.0063e+00, ..., -2.4239e-01,
-4.0307e-01, 1.4211e-01],
...,
[-2.4928e-01, -1.0692e+00, -1.5267e+00, ..., -1.1939e+00,
-7.8117e-01, -5.7162e-01],
[ 1.1591e+00, 7.8047e-01, 3.9040e-01, ..., 4.2123e-01,
7.3795e-01, 8.2373e-01],
[ 8.2921e-01, 6.0605e-01, 3.6405e-01, ..., 5.6123e-01,
8.0240e-01, 7.3208e-01]],
[[-3.3444e+00, -1.7807e+00, -6.3841e+00, ..., -7.3029e+00,
-6.4282e+00, -4.7207e+00],
[-8.2720e-01, 1.5642e+00, -4.5300e+00, ..., -5.6719e+00,
-5.7558e+00, -4.3792e+00],
[-4.2078e-01, 1.0854e+00, -7.6562e+00, ..., -8.1949e+00,
-8.0273e+00, -6.7500e+00],
...,
[-4.2694e-01, 5.0983e-01, -4.3533e+00, ..., -5.4626e+00,
-4.7515e+00, -3.9636e+00],
[ 2.2190e+00, 2.8265e+00, 3.7243e-01, ..., -1.2701e+00,
-1.3971e+00, -1.4961e+00],
[ 1.3008e+00, 1.5528e+00, -1.4995e+00, ..., -2.5573e+00,
-2.6443e+00, -2.6042e+00]],
[[-1.4345e+00, -2.4444e+00, -2.3871e+00, ..., -1.8468e+00,
-1.2965e+00, -1.4256e+00],
[-1.7679e+00, -2.0397e+00, -1.5542e+00, ..., -1.3076e+00,
-7.6996e-01, -8.6389e-01],
[-2.3188e+00, -2.0419e+00, -1.0488e+00, ..., -1.2095e+00,
-5.4116e-01, -2.1275e-01],
...,
[-7.8975e-01, -1.3151e+00, -6.2022e-01, ..., -5.8390e-01,
-4.7056e-01, -6.3017e-01],
[-1.2298e+00, -8.6416e-01, -6.3010e-01, ..., -2.5927e-01,
-7.0203e-02, -4.3670e-02],
[-1.4148e+00, -8.1794e-01, -4.2806e-01, ..., -3.2552e-01,
-2.8348e-01, -2.4950e-01]],
...,
[[-2.0407e+00, -1.0785e+00, -2.3235e+00, ..., -3.1312e+00,
-2.5366e+00, -1.7756e+00],
[-1.9000e+00, -8.5307e-02, -1.4502e+00, ..., -2.6010e+00,
-2.5913e+00, -1.9977e+00],
[-3.8319e-01, 2.3728e+00, 1.0931e+00, ..., 1.0442e-01,
2.4696e-01, 4.9281e-01],
...,
[-1.6828e+00, 1.1584e+00, -1.2616e-01, ..., -1.1357e+00,
-4.8159e-01, 2.9038e-02],
[-5.9090e-01, 1.5879e+00, 8.6462e-01, ..., 2.0011e-02,
1.6351e-01, 2.6971e-01],
[-1.6964e+00, -1.1530e+00, -1.5874e+00, ..., -2.2270e+00,
-1.3058e+00, -6.9021e-01]],
[[-2.4944e+00, -2.9717e+00, -4.3558e+00, ..., -2.6008e+00,
-2.0888e+00, -1.8661e+00],
[-1.8561e+00, -1.8106e+00, -5.0690e+00, ..., -3.1044e+00,
-2.9180e+00, -2.8518e+00],
[-2.8764e+00, -2.5620e+00, -6.7649e+00, ..., -4.5508e+00,
-4.4597e+00, -3.5297e+00],
...,
[-1.6626e+00, -4.4615e-01, -4.3013e+00, ..., -2.1814e+00,
-1.6501e+00, -1.2691e+00],
[-1.6590e+00, -6.7851e-01, -3.5417e+00, ..., -2.1425e+00,
-1.4270e+00, -8.6363e-01],
[-2.1012e+00, -1.2951e+00, -2.9184e+00, ..., -2.7239e+00,
-1.8307e+00, -1.5422e+00]],
[[-1.8768e+00, -2.6317e+00, -2.5417e+00, ..., -2.7972e+00,
-2.3017e+00, -1.7666e+00],
[-1.8812e+00, -2.6235e+00, -2.2606e+00, ..., -2.5310e+00,
-2.0124e+00, -1.6265e+00],
[-2.3581e+00, -2.9565e+00, -2.6621e+00, ..., -2.9097e+00,
-2.2824e+00, -1.7271e+00],
...,
[-2.3602e+00, -2.9521e+00, -2.7099e+00, ..., -2.8411e+00,
-2.2602e+00, -1.7016e+00],
[-1.5857e+00, -2.0354e+00, -1.9543e+00, ..., -2.0114e+00,
-1.6478e+00, -1.3070e+00],
[-1.0773e+00, -1.3001e+00, -1.3503e+00, ..., -1.2583e+00,
-1.0685e+00, -7.8652e-01]]],
[[[ 4.8143e-01, 9.4372e-01, 7.2042e-01, ..., 7.7962e-01,
1.0583e+00, 8.5075e-01],
[-1.0320e+00, -1.0709e-01, -1.4568e+00, ..., -1.1895e+00,
8.3679e-02, 6.5700e-01],
[ 7.6104e-01, 5.5662e-01, 2.3119e-01, ..., 4.9478e-01,
8.7646e-01, 8.9978e-01],
...,
[-9.0136e-01, -1.8726e+00, -1.8565e+00, ..., -1.7558e+00,
-1.3182e+00, -9.3283e-01],
[ 7.2981e-01, -2.0523e-02, -1.0201e-01, ..., 2.0539e-02,
6.9699e-01, 7.9777e-01],
[ 1.2762e-01, -5.7901e-01, -4.5675e-01, ..., -1.3557e-01,
4.3712e-01, 3.5093e-01]],
[[-2.6205e+00, -1.4493e+00, -4.7834e+00, ..., -5.8088e+00,
-4.9618e+00, -3.8717e+00],
[-6.6385e-01, 1.3430e+00, -3.3694e+00, ..., -4.7203e+00,
-4.5135e+00, -3.6859e+00],
[ 6.7196e-03, 1.1415e+00, -6.0674e+00, ..., -6.8684e+00,
-6.5067e+00, -5.9292e+00],
...,
[ 2.6068e-01, 3.4458e-01, -4.0903e+00, ..., -4.6635e+00,
-4.2112e+00, -3.5302e+00],
[ 2.2764e+00, 2.2619e+00, -1.8983e-01, ..., -8.2116e-01,
-1.1921e+00, -1.2505e+00],
[ 1.3315e+00, 1.1573e+00, -1.7833e+00, ..., -2.2511e+00,
-2.5241e+00, -2.4248e+00]],
[[-5.4371e-01, -1.5669e+00, -1.3942e+00, ..., -7.2640e-01,
-5.0222e-01, -7.2787e-01],
[-1.2950e+00, -1.6950e+00, -1.0997e+00, ..., -6.6626e-01,
-3.5758e-01, -3.1826e-01],
[-1.1614e+00, -1.0767e+00, -5.1002e-01, ..., -4.1882e-01,
-7.8338e-02, -8.1592e-02],
...,
[-1.4180e-01, -6.1417e-01, -4.7807e-01, ..., -5.0548e-01,
-1.9949e-01, -4.4753e-01],
[-6.8630e-01, -1.9922e-01, -4.3381e-01, ..., -2.0401e-01,
2.4551e-01, 2.4156e-01],
[-7.9122e-01, -1.9890e-01, -3.4723e-01, ..., -3.9656e-01,
-1.9262e-01, -2.3289e-01]],
...,
[[-2.0685e+00, -1.1742e+00, -2.4659e+00, ..., -3.2389e+00,
-2.6771e+00, -1.8131e+00],
[-5.1065e-01, 1.1150e+00, -6.6700e-01, ..., -1.6213e+00,
-1.7914e+00, -1.5427e+00],
[-2.4148e-01, 2.1305e+00, 4.6998e-01, ..., -3.3133e-01,
-6.1893e-02, 1.1177e-01],
...,
[-2.0300e+00, -7.5756e-02, -1.7663e-01, ..., -1.3850e+00,
-3.9408e-01, 1.0949e-01],
[-1.1724e+00, 8.2316e-01, 9.5131e-01, ..., -3.1028e-03,
3.0814e-01, 4.7045e-01],
[-2.0909e+00, -1.4101e+00, -1.6312e+00, ..., -2.4503e+00,
-1.4559e+00, -9.1355e-01]],
[[-1.4392e+00, -1.4112e+00, -2.8005e+00, ..., -1.0987e+00,
-6.0743e-01, -4.9695e-01],
[-8.7271e-01, -4.9667e-01, -3.7584e+00, ..., -1.8979e+00,
-1.6527e+00, -1.8422e+00],
[-2.4310e+00, -1.9506e+00, -6.3474e+00, ..., -4.1624e+00,
-3.6116e+00, -2.8552e+00],
...,
[-1.3757e+00, -5.9110e-02, -3.3603e+00, ..., -1.4332e+00,
-1.0537e+00, -9.4888e-01],
[-1.6849e+00, -6.4965e-01, -3.0313e+00, ..., -1.7266e+00,
-1.1829e+00, -8.2429e-01],
[-2.3090e+00, -1.4922e+00, -3.1034e+00, ..., -2.9654e+00,
-2.1548e+00, -1.9904e+00]],
[[-1.2189e+00, -1.7958e+00, -1.6201e+00, ..., -1.8355e+00,
-1.5192e+00, -1.1288e+00],
[-1.2771e+00, -1.7833e+00, -1.3857e+00, ..., -1.6169e+00,
-1.3125e+00, -1.0273e+00],
[-1.6941e+00, -2.0471e+00, -1.6734e+00, ..., -1.9168e+00,
-1.4744e+00, -1.0779e+00],
...,
[-2.3882e+00, -2.9552e+00, -2.7357e+00, ..., -2.8598e+00,
-2.2907e+00, -1.7471e+00],
[-1.5458e+00, -2.0157e+00, -1.9428e+00, ..., -1.9454e+00,
-1.6033e+00, -1.2871e+00],
[-1.0066e+00, -1.3008e+00, -1.3354e+00, ..., -1.1849e+00,
-1.0153e+00, -7.6318e-01]]],
[[[ 2.2117e-01, 4.9442e-01, 6.2107e-01, ..., -3.5138e-01,
-1.0519e-01, -1.3473e-02],
[-1.9561e+00, -1.2719e+00, -2.7502e+00, ..., -3.2283e+00,
-1.2990e+00, -4.8934e-01],
[-1.1229e-01, -8.5686e-01, 2.5284e-02, ..., -1.8974e-01,
-4.0404e-01, 1.4505e-01],
...,
[-9.6248e-01, -2.1356e+00, -2.0489e+00, ..., -1.9647e+00,
-1.9021e+00, -1.2748e+00],
[ 8.0190e-01, -8.6399e-02, -1.0225e-01, ..., -4.3375e-02,
2.2413e-01, 4.2487e-01],
[ 2.2537e-01, -5.2578e-01, -3.7445e-01, ..., 1.7063e-01,
4.4827e-01, 4.9093e-01]],
[[-1.6744e+00, -1.3498e+00, -4.2387e+00, ..., -6.8974e+00,
-6.0916e+00, -4.5460e+00],
[ 4.4526e-01, 1.4480e+00, -2.5766e+00, ..., -5.5614e+00,
-5.6532e+00, -4.3706e+00],
[ 1.1586e+00, 1.1678e+00, -5.0972e+00, ..., -7.9861e+00,
-7.8954e+00, -6.6513e+00],
...,
[ 3.4313e-01, 3.4378e-01, -4.1272e+00, ..., -7.1852e+00,
-6.7944e+00, -5.2583e+00],
[ 2.5360e+00, 2.5075e+00, 1.7166e-01, ..., -1.9630e+00,
-2.6261e+00, -2.0917e+00],
[ 1.5066e+00, 1.3378e+00, -1.5824e+00, ..., -2.9435e+00,
-3.5277e+00, -3.0683e+00]],
[[-1.4023e-01, -1.1732e+00, -1.3908e+00, ..., -1.4213e+00,
-1.0584e+00, -1.3372e+00],
[-5.2202e-01, -8.5871e-01, -8.9723e-01, ..., -9.9477e-01,
-6.1354e-01, -9.1035e-01],
[-3.6117e-01, -2.7251e-01, -4.5450e-01, ..., -1.1006e+00,
-6.1476e-01, -4.9003e-01],
...,
[-9.0672e-02, -6.4640e-01, -6.1474e-01, ..., -1.0436e+00,
-8.8092e-01, -1.2566e+00],
[-7.2612e-01, -2.4149e-01, -5.7251e-01, ..., -5.6435e-01,
-2.5248e-01, -4.1960e-01],
[-8.9506e-01, -2.9654e-01, -4.7973e-01, ..., -5.2830e-01,
-4.8047e-01, -6.4102e-01]],
...,
[[-2.7822e+00, -2.3664e+00, -2.3885e+00, ..., -3.0389e+00,
-2.4526e+00, -1.7771e+00],
[-2.0544e+00, -7.6187e-01, -1.0746e+00, ..., -2.4694e+00,
-2.5328e+00, -2.0413e+00],
[-1.0764e+00, 8.7394e-01, 7.9999e-01, ..., -9.9905e-02,
4.4095e-03, 2.5582e-01],
...,
[-2.2401e+00, -1.2547e-01, -1.4858e-01, ..., -9.2061e-01,
-2.5675e-01, 2.8994e-01],
[-1.2141e+00, 8.8162e-01, 1.0472e+00, ..., 4.7163e-01,
4.9586e-01, 8.3126e-01],
[-2.1593e+00, -1.4868e+00, -1.7039e+00, ..., -2.1783e+00,
-1.1376e+00, -3.3928e-01]],
[[-1.6289e+00, -1.6264e+00, -2.8000e+00, ..., -2.3223e+00,
-2.0181e+00, -1.8961e+00],
[-7.1862e-01, -1.2501e-01, -3.1714e+00, ..., -2.9805e+00,
-2.9677e+00, -2.9531e+00],
[-2.0714e+00, -1.0075e+00, -5.1222e+00, ..., -4.5104e+00,
-4.5397e+00, -3.7424e+00],
...,
[-1.3369e+00, 1.8528e-01, -3.3995e+00, ..., -2.8471e+00,
-2.7725e+00, -2.3324e+00],
[-1.6629e+00, -4.5698e-01, -3.0241e+00, ..., -2.4577e+00,
-2.2840e+00, -1.6915e+00],
[-2.3178e+00, -1.3464e+00, -3.0188e+00, ..., -2.8822e+00,
-2.4569e+00, -2.2197e+00]],
[[-1.2952e+00, -1.8167e+00, -1.6406e+00, ..., -2.5901e+00,
-2.1704e+00, -1.6780e+00],
[-1.2385e+00, -1.6988e+00, -1.2221e+00, ..., -2.4392e+00,
-1.9702e+00, -1.6140e+00],
[-1.5586e+00, -1.8987e+00, -1.4855e+00, ..., -2.8792e+00,
-2.3039e+00, -1.7929e+00],
...,
[-2.4426e+00, -3.0518e+00, -2.7816e+00, ..., -4.0487e+00,
-3.2868e+00, -2.5747e+00],
[-1.5776e+00, -2.0793e+00, -1.9756e+00, ..., -2.7943e+00,
-2.3406e+00, -1.9336e+00],
[-1.0271e+00, -1.3409e+00, -1.3552e+00, ..., -1.8500e+00,
-1.5360e+00, -1.1902e+00]]],
[[[ 4.5108e-01, 9.3842e-01, 8.6086e-01, ..., -3.1623e-01,
-4.8945e-02, 2.9522e-02],
[-1.2684e+00, -3.5790e-01, -1.5006e+00, ..., -3.3920e+00,
-1.3663e+00, -4.6438e-01],
[ 5.7462e-01, 4.2726e-01, 3.7130e-01, ..., -2.0097e-01,
-3.6410e-01, 1.3860e-01],
...,
[-1.7281e-01, -8.4791e-01, -1.5127e+00, ..., -1.7337e+00,
-1.7223e+00, -1.1128e+00],
[ 1.0870e+00, 8.0457e-01, 2.9619e-01, ..., 6.2635e-02,
2.7464e-01, 4.6228e-01],
[ 7.1812e-01, 5.1823e-01, 1.8667e-01, ..., 2.2591e-01,
4.4116e-01, 4.9731e-01]],
[[-2.2950e+00, -1.1252e+00, -4.1060e+00, ..., -7.0325e+00,
-6.2052e+00, -4.6170e+00],
[-2.8464e-01, 1.7120e+00, -2.4947e+00, ..., -5.4317e+00,
-5.5606e+00, -4.3195e+00],
[ 4.0748e-01, 1.5220e+00, -5.0294e+00, ..., -7.9207e+00,
-7.9080e+00, -6.7441e+00],
...,
[-4.5859e-01, 6.5988e-01, -4.0181e+00, ..., -7.1086e+00,
-6.6989e+00, -5.2330e+00],
[ 2.1353e+00, 2.8461e+00, 4.5401e-01, ..., -2.2688e+00,
-2.8419e+00, -2.2764e+00],
[ 1.2458e+00, 1.5615e+00, -1.4390e+00, ..., -3.0108e+00,
-3.5167e+00, -3.0384e+00]],
[[-5.7903e-01, -1.5574e+00, -1.3785e+00, ..., -1.4759e+00,
-1.0921e+00, -1.3944e+00],
[-1.2724e+00, -1.6162e+00, -9.9438e-01, ..., -1.0521e+00,
-6.3291e-01, -8.9525e-01],
[-1.1120e+00, -9.6748e-01, -3.7372e-01, ..., -1.0920e+00,
-5.7314e-01, -4.5168e-01],
...,
[-9.2870e-01, -1.3785e+00, -5.4191e-01, ..., -9.8741e-01,
-8.4387e-01, -1.1902e+00],
[-1.2906e+00, -8.8883e-01, -5.5033e-01, ..., -5.5435e-01,
-2.7098e-01, -4.3757e-01],
[-1.3905e+00, -7.9395e-01, -3.6475e-01, ..., -4.8046e-01,
-4.3295e-01, -5.9012e-01]],
...,
[[-2.1823e+00, -1.2580e+00, -2.5468e+00, ..., -3.3660e+00,
-2.7345e+00, -1.9457e+00],
[-6.8961e-01, 9.4233e-01, -9.7842e-01, ..., -2.5488e+00,
-2.6575e+00, -2.1619e+00],
[-2.5134e-01, 2.2313e+00, 4.9825e-01, ..., -6.4194e-02,
5.6861e-02, 3.1311e-01],
...,
[-1.5366e+00, 1.2801e+00, -1.8685e-01, ..., -8.5270e-01,
-2.2477e-01, 2.5223e-01],
[-5.1922e-01, 1.6206e+00, 8.2506e-01, ..., 3.6041e-01,
4.3432e-01, 7.6001e-01],
[-1.6210e+00, -1.0576e+00, -1.5877e+00, ..., -2.0154e+00,
-1.0546e+00, -3.0546e-01]],
[[-1.2963e+00, -1.1982e+00, -2.5714e+00, ..., -2.1020e+00,
-1.7798e+00, -1.7287e+00],
[-6.1467e-01, -9.8498e-02, -3.2852e+00, ..., -2.7405e+00,
-2.7985e+00, -2.8960e+00],
[-2.0842e+00, -1.3933e+00, -5.6584e+00, ..., -4.4849e+00,
-4.5560e+00, -3.7543e+00],
...,
[-1.6147e+00, -4.5321e-01, -4.0929e+00, ..., -2.9866e+00,
-2.8629e+00, -2.3913e+00],
[-1.6406e+00, -6.8083e-01, -3.4335e+00, ..., -2.5172e+00,
-2.3421e+00, -1.7520e+00],
[-2.1032e+00, -1.3465e+00, -2.9800e+00, ..., -2.8061e+00,
-2.4125e+00, -2.1640e+00]],
[[-1.1511e+00, -1.7077e+00, -1.5319e+00, ..., -2.6427e+00,
-2.1923e+00, -1.6965e+00],
[-1.1571e+00, -1.6201e+00, -1.2080e+00, ..., -2.4026e+00,
-1.9295e+00, -1.5846e+00],
[-1.5009e+00, -1.8049e+00, -1.4145e+00, ..., -2.7912e+00,
-2.2182e+00, -1.7250e+00],
...,
[-2.3087e+00, -2.8858e+00, -2.6603e+00, ..., -3.8966e+00,
-3.1625e+00, -2.4882e+00],
[-1.5466e+00, -1.9689e+00, -1.8924e+00, ..., -2.7141e+00,
-2.2708e+00, -1.8773e+00],
[-1.0336e+00, -1.2396e+00, -1.2899e+00, ..., -1.8214e+00,
-1.5067e+00, -1.1713e+00]]]], device='cuda:0'),
tensor([[[[ 3.2901e-01, 6.1278e-01, 7.3090e-01, ..., 5.4089e-01,
1.0514e+00, 6.4620e-01],
[-2.0151e+00, -1.3157e+00, -2.9266e+00, ..., -2.3938e+00,
-6.8052e-01, -6.1639e-02],
[-1.6862e-02, -7.5718e-01, -9.2804e-02, ..., -3.6708e-02,
6.1758e-01, 7.0289e-01],
...,
[ 1.0703e-01, -7.1912e-01, -1.7308e+00, ..., -1.1957e+00,
-8.5721e-01, -6.1294e-01],
[ 1.5614e+00, 1.2422e+00, 4.8264e-01, ..., 6.2481e-01,
8.6750e-01, 8.9275e-01],
[ 1.1867e+00, 1.0115e+00, 6.4716e-01, ..., 8.4306e-01,
9.7167e-01, 7.9536e-01]],
[[-1.7811e+00, -1.1635e+00, -4.1734e+00, ..., -4.8958e+00,
-4.4463e+00, -3.4065e+00],
[ 4.5540e-01, 1.8982e+00, -2.3201e+00, ..., -3.6199e+00,
-3.7014e+00, -3.0353e+00],
[ 1.1862e+00, 1.6424e+00, -4.9880e+00, ..., -5.5610e+00,
-5.3657e+00, -5.0022e+00],
...,
[-7.3239e-01, 4.8498e-01, -5.0010e+00, ..., -6.0540e+00,
-5.4022e+00, -4.2997e+00],
[ 2.0972e+00, 2.8595e+00, -1.2483e-02, ..., -1.6495e+00,
-1.8294e+00, -1.7287e+00],
[ 1.2472e+00, 1.5765e+00, -1.7512e+00, ..., -2.7906e+00,
-2.9236e+00, -2.7844e+00]],
[[-1.8901e-01, -1.2178e+00, -1.3445e+00, ..., -5.0104e-01,
-1.8857e-01, -5.0052e-01],
[-7.1205e-01, -9.8212e-01, -8.2549e-01, ..., -3.1701e-01,
-1.7049e-02, -1.0966e-01],
[-6.3262e-01, -4.4371e-01, -4.1601e-01, ..., -3.4988e-01,
5.2169e-02, 5.1333e-02],
...,
[-1.1385e+00, -1.6010e+00, -7.1710e-01, ..., -6.7593e-01,
-5.0437e-01, -6.9486e-01],
[-1.3736e+00, -9.1803e-01, -5.2715e-01, ..., -1.9942e-01,
4.3442e-02, -5.2592e-04],
[-1.3572e+00, -6.7633e-01, -1.6155e-01, ..., -1.2569e-01,
-6.3354e-02, -1.0435e-01]],
...,
[[-2.9469e+00, -2.3456e+00, -2.7824e+00, ..., -3.3598e+00,
-2.6074e+00, -1.7357e+00],
[-2.0470e+00, -5.6367e-01, -1.4869e+00, ..., -2.4623e+00,
-2.2475e+00, -1.7774e+00],
[-9.0484e-01, 1.3612e+00, 6.0676e-01, ..., -4.1672e-01,
-3.8017e-03, 1.4590e-01],
...,
[-1.5800e+00, 1.6303e+00, 4.2540e-01, ..., -7.3779e-01,
-1.5741e-01, 4.4184e-01],
[-6.3687e-01, 1.7927e+00, 1.2207e+00, ..., 2.5542e-01,
3.1731e-01, 5.6507e-01],
[-1.7907e+00, -1.0543e+00, -1.3688e+00, ..., -2.0022e+00,
-1.1698e+00, -4.8878e-01]],
[[-1.5450e+00, -1.5052e+00, -2.8307e+00, ..., -1.2183e+00,
-8.0018e-01, -7.9257e-01],
[-5.7990e-01, 8.6965e-02, -3.1997e+00, ..., -1.4184e+00,
-1.2460e+00, -1.5645e+00],
[-2.0924e+00, -9.5716e-01, -5.3452e+00, ..., -3.2325e+00,
-2.7555e+00, -2.2267e+00],
...,
[-1.8551e+00, -8.0747e-01, -4.7549e+00, ..., -2.4598e+00,
-2.1245e+00, -1.7464e+00],
[-1.7348e+00, -8.5760e-01, -3.7519e+00, ..., -2.1907e+00,
-1.6871e+00, -1.1710e+00],
[-2.0919e+00, -1.3622e+00, -3.0005e+00, ..., -2.6565e+00,
-1.9454e+00, -1.7183e+00]],
[[-1.3229e+00, -1.8845e+00, -1.6773e+00, ..., -1.7551e+00,
-1.4489e+00, -1.1011e+00],
[-1.2561e+00, -1.7453e+00, -1.2259e+00, ..., -1.3836e+00,
-1.0941e+00, -8.6982e-01],
[-1.5812e+00, -1.9227e+00, -1.4737e+00, ..., -1.6162e+00,
-1.2167e+00, -8.9442e-01],
...,
[-2.5695e+00, -3.2123e+00, -2.9595e+00, ..., -3.1559e+00,
-2.5197e+00, -1.9340e+00],
[-1.7151e+00, -2.1816e+00, -2.0935e+00, ..., -2.2150e+00,
-1.8138e+00, -1.4577e+00],
[-1.1449e+00, -1.3810e+00, -1.4221e+00, ..., -1.3794e+00,
-1.1683e+00, -8.7136e-01]]],
[[[ 2.4045e-01, 5.2134e-01, 6.3268e-01, ..., 4.4497e-01,
1.0143e+00, 5.7348e-01],
[-2.1946e+00, -1.4907e+00, -3.1937e+00, ..., -2.9362e+00,
-9.9400e-01, -2.8055e-01],
[-2.2202e-01, -9.5922e-01, -2.2477e-01, ..., -2.1525e-01,
5.3730e-01, 6.0856e-01],
...,
[ 1.0657e-01, -7.3630e-01, -1.5810e+00, ..., -1.0260e+00,
-6.7845e-01, -4.8411e-01],
[ 1.5179e+00, 1.1555e+00, 5.7781e-01, ..., 6.5237e-01,
1.0086e+00, 9.8618e-01],
[ 1.1402e+00, 9.3415e-01, 6.6235e-01, ..., 7.9416e-01,
1.0548e+00, 8.8613e-01]],
[[-1.6822e+00, -1.2738e+00, -4.2026e+00, ..., -5.0618e+00,
-4.6080e+00, -3.4857e+00],
[ 4.9941e-01, 1.7046e+00, -2.3621e+00, ..., -3.4819e+00,
-3.6240e+00, -2.9581e+00],
[ 1.1702e+00, 1.4005e+00, -4.9887e+00, ..., -5.4926e+00,
-5.3382e+00, -5.0188e+00],
...,
[-5.9562e-01, 5.9473e-01, -4.6871e+00, ..., -5.8382e+00,
-5.2751e+00, -4.3170e+00],
[ 2.1844e+00, 2.8980e+00, 1.5973e-01, ..., -1.5829e+00,
-1.8481e+00, -1.7995e+00],
[ 1.2901e+00, 1.5754e+00, -1.6600e+00, ..., -2.7273e+00,
-2.9284e+00, -2.7647e+00]],
[[-9.9771e-02, -1.0735e+00, -1.2665e+00, ..., -5.7710e-01,
-2.0766e-01, -5.3264e-01],
[-5.0853e-01, -7.3168e-01, -7.2825e-01, ..., -3.5384e-01,
8.8947e-04, -8.3239e-02],
[-4.4633e-01, -2.3358e-01, -3.8392e-01, ..., -3.6418e-01,
1.0729e-01, 9.0079e-02],
...,
[-1.1233e+00, -1.5821e+00, -6.2666e-01, ..., -6.0529e-01,
-4.9443e-01, -6.0718e-01],
[-1.3473e+00, -8.8874e-01, -4.3704e-01, ..., -9.4675e-02,
8.6410e-02, 8.0189e-02],
[-1.3913e+00, -7.0718e-01, -1.5015e-01, ..., -4.8224e-02,
-2.4104e-02, -3.5316e-02]],
...,
[[-2.9637e+00, -2.5625e+00, -2.7382e+00, ..., -3.7262e+00,
-2.8516e+00, -1.9005e+00],
[-2.2705e+00, -1.0193e+00, -1.6154e+00, ..., -2.7135e+00,
-2.4521e+00, -1.9651e+00],
[-1.0837e+00, 9.0553e-01, 5.1192e-01, ..., -4.1792e-01,
4.1787e-02, 2.0098e-01],
...,
[-1.5468e+00, 1.7014e+00, 3.4258e-01, ..., -8.2226e-01,
-1.2180e-01, 3.6210e-01],
[-5.7281e-01, 1.8492e+00, 1.1100e+00, ..., 1.6903e-01,
3.2888e-01, 4.7476e-01],
[-1.6984e+00, -9.6944e-01, -1.4084e+00, ..., -1.9925e+00,
-1.0987e+00, -5.2092e-01]],
[[-1.6242e+00, -1.6281e+00, -2.7833e+00, ..., -1.1557e+00,
-7.1237e-01, -7.4500e-01],
[-6.6467e-01, -2.1841e-02, -3.0455e+00, ..., -1.2433e+00,
-1.1155e+00, -1.5583e+00],
[-2.1440e+00, -1.0316e+00, -5.0681e+00, ..., -3.2114e+00,
-2.7249e+00, -2.2354e+00],
...,
[-1.7422e+00, -6.1258e-01, -4.6332e+00, ..., -2.5023e+00,
-2.1328e+00, -1.6547e+00],
[-1.6745e+00, -7.2005e-01, -3.6814e+00, ..., -2.3343e+00,
-1.7690e+00, -1.1365e+00],
[-2.0778e+00, -1.2670e+00, -2.9307e+00, ..., -2.7919e+00,
-1.9890e+00, -1.6823e+00]],
[[-1.3372e+00, -1.8908e+00, -1.7304e+00, ..., -1.8338e+00,
-1.5245e+00, -1.1507e+00],
[-1.2679e+00, -1.7448e+00, -1.2862e+00, ..., -1.3854e+00,
-1.1009e+00, -8.7221e-01],
[-1.5772e+00, -1.9255e+00, -1.5410e+00, ..., -1.5946e+00,
-1.1927e+00, -8.7271e-01],
...,
[-2.4875e+00, -3.1184e+00, -2.8462e+00, ..., -2.9765e+00,
-2.3700e+00, -1.8153e+00],
[-1.6694e+00, -2.1391e+00, -2.0372e+00, ..., -2.0985e+00,
-1.7174e+00, -1.3687e+00],
[-1.1150e+00, -1.3518e+00, -1.3994e+00, ..., -1.3153e+00,
-1.1039e+00, -8.1522e-01]]],
[[[ 1.3644e-01, 3.7769e-01, -1.1945e+00, ..., 6.7090e-01,
9.5679e-01, 6.9965e-01],
[-2.1037e+00, -1.4530e+00, -4.4175e+00, ..., -1.4012e+00,
-1.1098e-01, 4.2246e-01],
[ 4.4083e-01, -2.6958e-01, -1.1495e+00, ..., 5.5738e-01,
9.7045e-01, 8.9133e-01],
...,
[-3.0001e-01, -1.2566e+00, -2.5550e+00, ..., -1.1301e+00,
-8.0222e-01, -6.4286e-01],
[ 1.2064e+00, 6.2031e-01, -9.4706e-01, ..., 5.6417e-01,
7.7161e-01, 7.8716e-01],
[ 7.8985e-01, 2.7695e-01, -4.4375e-01, ..., 5.6418e-01,
7.9850e-01, 6.9566e-01]],
[[-3.2605e+00, -1.5079e+00, -6.1854e+00, ..., -5.4385e+00,
-4.6372e+00, -3.6411e+00],
[-5.9232e-01, 2.0901e+00, -4.2142e+00, ..., -4.2469e+00,
-4.0937e+00, -3.3844e+00],
[ 8.2161e-04, 1.7896e+00, -7.4051e+00, ..., -6.2246e+00,
-5.9090e+00, -5.4890e+00],
...,
[-1.2461e+00, 2.8823e-01, -6.5500e+00, ..., -5.3984e+00,
-4.7473e+00, -3.9280e+00],
[ 1.5125e+00, 2.5931e+00, -1.4057e+00, ..., -1.3139e+00,
-1.4655e+00, -1.4978e+00],
[ 9.8255e-01, 1.6151e+00, -2.4441e+00, ..., -2.5030e+00,
-2.6308e+00, -2.5503e+00]],
[[-1.4003e+00, -2.3538e+00, -2.1642e+00, ..., -8.3975e-01,
-6.3673e-01, -8.0696e-01],
[-2.0338e+00, -2.2088e+00, -1.5663e+00, ..., -6.5491e-01,
-4.0443e-01, -3.6136e-01],
[-2.5551e+00, -2.0142e+00, -8.3232e-01, ..., -3.7642e-01,
-1.0249e-01, -1.0232e-01],
...,
[-1.6059e+00, -1.8282e+00, -7.6541e-01, ..., -6.3521e-01,
-5.5559e-01, -6.8981e-01],
[-1.7744e+00, -1.2103e+00, -7.8413e-01, ..., -2.3066e-01,
-5.6148e-02, -4.3844e-02],
[-1.7399e+00, -1.1409e+00, -6.2357e-01, ..., -3.2345e-01,
-2.8186e-01, -2.4841e-01]],
...,
[[-2.3757e+00, -1.3251e+00, -2.7425e+00, ..., -3.0445e+00,
-2.4035e+00, -1.6070e+00],
[-1.9218e+00, 1.5243e-01, -1.3331e+00, ..., -1.5001e+00,
-1.5218e+00, -1.3106e+00],
[-5.3791e-01, 2.4190e+00, 9.0342e-01, ..., -1.1008e-01,
2.3109e-01, 3.1063e-01],
...,
[-1.7179e+00, 1.1229e+00, 1.1941e-01, ..., -1.0329e+00,
-3.0634e-01, 1.7379e-01],
[-8.3339e-01, 1.6655e+00, 1.4495e+00, ..., 1.3747e-01,
2.9179e-01, 3.7696e-01],
[-2.0674e+00, -1.0183e+00, -8.1661e-01, ..., -2.1704e+00,
-1.2420e+00, -6.5164e-01]],
[[-2.3986e+00, -2.6561e+00, -3.9970e+00, ..., -1.1460e+00,
-6.4452e-01, -5.1489e-01],
[-1.6831e+00, -1.3502e+00, -4.7145e+00, ..., -1.7608e+00,
-1.5537e+00, -1.7576e+00],
[-3.0472e+00, -2.4627e+00, -6.8510e+00, ..., -3.7707e+00,
-3.3011e+00, -2.6631e+00],
...,
[-2.4936e+00, -1.7180e+00, -4.9724e+00, ..., -2.2306e+00,
-1.7446e+00, -1.4002e+00],
[-2.1911e+00, -1.5406e+00, -3.8899e+00, ..., -2.1629e+00,
-1.5019e+00, -9.6305e-01],
[-2.2799e+00, -1.7554e+00, -3.0661e+00, ..., -2.8366e+00,
-1.9501e+00, -1.6573e+00]],
[[-1.9045e+00, -2.6408e+00, -2.4305e+00, ..., -1.6784e+00,
-1.3950e+00, -1.0447e+00],
[-1.9230e+00, -2.6426e+00, -2.1191e+00, ..., -1.3758e+00,
-1.1129e+00, -8.8227e-01],
[-2.4464e+00, -3.0080e+00, -2.5280e+00, ..., -1.5729e+00,
-1.2136e+00, -8.8472e-01],
...,
[-3.1267e+00, -3.9088e+00, -3.6816e+00, ..., -2.7753e+00,
-2.2243e+00, -1.6839e+00],
[-2.0791e+00, -2.6904e+00, -2.6509e+00, ..., -1.9523e+00,
-1.6075e+00, -1.2769e+00],
[-1.4955e+00, -1.8176e+00, -1.8635e+00, ..., -1.2136e+00,
-1.0358e+00, -7.6459e-01]]],
[[[ 4.1379e-01, 7.5617e-01, 8.0319e-01, ..., 7.7291e-01,
1.0093e+00, 8.0250e-01],
[-1.6155e+00, -8.5793e-01, -2.2841e+00, ..., -1.8631e+00,
-3.6078e-01, 4.0048e-01],
[ 2.9718e-01, -2.3819e-01, 2.2609e-01, ..., 1.8509e-01,
4.3216e-01, 9.0849e-01],
...,
[ 4.0150e-02, -9.7138e-01, -2.1715e+00, ..., -2.1720e+01,
-1.8568e+01, -1.9709e+01],
[ 1.6175e+00, 1.1775e+00, 1.3384e-01, ..., -6.5061e+00,
-4.1500e+00, -9.6600e+00],
[ 1.2283e+00, 9.2816e-01, 4.0805e-01, ..., -9.7815e+00,
-8.9097e+00, -1.0036e+01]],
[[-1.9915e+00, -1.2493e+00, -4.2441e+00, ..., -5.2208e+00,
-3.9727e+00, -3.8900e+00],
[ 8.7872e-02, 1.6579e+00, -2.5210e+00, ..., -3.6059e+00,
-2.9035e+00, -3.5813e+00],
[ 7.7856e-01, 1.4457e+00, -5.0955e+00, ..., -5.3635e+00,
-4.9605e+00, -6.1144e+00],
...,
[-8.6755e-01, 5.3449e-01, -5.5607e+00, ..., -1.7247e+01,
-1.6817e+01, -1.4204e+01],
[ 2.0523e+00, 2.9299e+00, -3.5321e-01, ..., -8.6537e+00,
-8.3476e+00, -7.5889e+00],
[ 1.2561e+00, 1.6533e+00, -1.9853e+00, ..., -4.4116e+00,
-3.4930e+00, -3.3844e+00]],
[[-2.7344e-01, -1.2718e+00, -1.3294e+00, ..., -3.1043e-01,
-4.9674e-01, -9.0573e-01],
[-8.5410e-01, -1.1604e+00, -9.0133e-01, ..., -1.2018e-01,
-3.0512e-01, -3.8648e-01],
[-7.4099e-01, -5.9012e-01, -4.0011e-01, ..., -7.7516e-02,
-2.9709e-02, -1.3829e-01],
...,
[-1.2220e+00, -1.7377e+00, -8.2510e-01, ..., -1.6556e+00,
3.1545e+00, -2.3705e-01],
[-1.4323e+00, -1.0006e+00, -6.6401e-01, ..., -1.2926e+00,
2.4788e+00, -4.1643e-01],
[-1.4146e+00, -7.4168e-01, -2.4166e-01, ..., 8.3155e-02,
3.5657e+00, 1.6717e+00]],
...,
[[-2.6232e+00, -1.9305e+00, -2.6267e+00, ..., -3.3392e+00,
-2.5560e+00, -1.9395e+00],
[-1.5546e+00, 1.0714e-02, -1.1846e+00, ..., -2.2023e+00,
-1.7610e+00, -1.6535e+00],
[-7.0838e-01, 1.6380e+00, 5.7032e-01, ..., -5.9536e-01,
3.1879e-01, 4.1955e-01],
...,
[-1.7032e+00, 1.6189e+00, 5.2014e-01, ..., -1.4933e+01,
-6.1371e+00, -9.8717e+00],
[-8.1186e-01, 1.8387e+00, 1.4404e+00, ..., -1.8683e+01,
-1.2190e+01, -4.3978e+00],
[-2.0004e+00, -1.0572e+00, -1.1787e+00, ..., -1.0370e+01,
-1.0810e+01, -3.9232e+00]],
[[-1.4643e+00, -1.3983e+00, -2.6793e+00, ..., -9.1629e-01,
-5.4866e-01, -6.8648e-01],
[-6.2354e-01, -3.1622e-02, -3.2091e+00, ..., -8.8088e-01,
-1.0609e+00, -1.9300e+00],
[-2.0993e+00, -1.1558e+00, -5.4574e+00, ..., -2.8808e+00,
-2.9121e+00, -2.9564e+00],
...,
[-2.0560e+00, -1.0630e+00, -5.0129e+00, ..., -6.3828e+00,
-8.7677e+00, -1.0780e+01],
[-1.8318e+00, -9.9084e-01, -3.9116e+00, ..., -5.7907e+00,
-9.8023e+00, -8.3787e+00],
[-2.1042e+00, -1.4035e+00, -3.0547e+00, ..., -7.8044e+00,
-9.2666e+00, -4.8293e+00]],
[[-1.2510e+00, -1.8016e+00, -1.6218e+00, ..., -2.0342e+00,
-1.7294e+00, -1.2511e+00],
[-1.2055e+00, -1.6894e+00, -1.2233e+00, ..., -1.7534e+00,
-1.4153e+00, -1.0818e+00],
[-1.5410e+00, -1.8856e+00, -1.4787e+00, ..., -2.1103e+00,
-1.5319e+00, -1.0912e+00],
...,
[-2.8101e+00, -3.5255e+00, -3.2950e+00, ..., -1.1572e+01,
-9.7248e+00, -7.4441e+00],
[-1.8763e+00, -2.4028e+00, -2.3347e+00, ..., -8.4357e+00,
-6.7000e+00, -5.9488e+00],
[-1.2859e+00, -1.5585e+00, -1.6107e+00, ..., -4.7090e+00,
-4.2008e+00, -4.2919e+00]]],
[[[ 2.6445e-01, 5.4502e-01, 6.4943e-01, ..., 6.1931e-01,
9.7703e-01, 7.9851e-01],
[-2.1333e+00, -1.3834e+00, -3.0633e+00, ..., -1.2112e+00,
1.1196e-01, 6.3836e-01],
[-7.4979e-02, -8.9145e-01, -2.7428e-02, ..., 5.3677e-01,
9.4756e-01, 9.4353e-01],
...,
[-9.8587e-01, -2.2536e+00, -2.0222e+00, ..., -1.3196e+00,
-8.4630e-01, -5.9391e-01],
[ 7.2839e-01, -2.5383e-01, -1.7797e-01, ..., 4.2280e-01,
8.1024e-01, 9.1431e-01],
[ 1.5130e-01, -6.5986e-01, -4.6308e-01, ..., 6.1459e-01,
9.0377e-01, 8.1820e-01]],
[[-1.7546e+00, -1.3008e+00, -4.2714e+00, ..., -5.7667e+00,
-4.8745e+00, -3.8305e+00],
[ 6.5427e-01, 1.8648e+00, -2.2738e+00, ..., -4.7295e+00,
-4.4758e+00, -3.6862e+00],
[ 1.5085e+00, 1.6058e+00, -4.9867e+00, ..., -6.8494e+00,
-6.4107e+00, -5.8272e+00],
...,
[ 3.6354e-01, 2.8235e-01, -4.2384e+00, ..., -5.9846e+00,
-5.1706e+00, -4.2591e+00],
[ 2.3811e+00, 2.2654e+00, -2.2296e-01, ..., -1.3564e+00,
-1.4801e+00, -1.5642e+00],
[ 1.3976e+00, 1.1726e+00, -1.8036e+00, ..., -2.7407e+00,
-2.8515e+00, -2.7960e+00]],
[[-2.3235e-01, -1.3799e+00, -1.5872e+00, ..., -7.8559e-01,
-5.3890e-01, -6.5397e-01],
[-7.2138e-01, -1.0418e+00, -1.0185e+00, ..., -6.6969e-01,
-3.5974e-01, -2.2326e-01],
[-5.1742e-01, -3.2080e-01, -4.6079e-01, ..., -3.7730e-01,
-1.6255e-02, 8.4531e-02],
...,
[-1.1276e-02, -5.5565e-01, -6.1123e-01, ..., -6.5573e-01,
-4.4976e-01, -5.5443e-01],
[-6.5864e-01, -2.3221e-01, -6.0301e-01, ..., -2.7972e-01,
1.8989e-03, 8.0259e-02],
[-8.1464e-01, -2.6888e-01, -5.1896e-01, ..., -3.0518e-01,
-2.0531e-01, -1.3782e-01]],
...,
[[-3.0959e+00, -2.5590e+00, -2.6440e+00, ..., -3.1386e+00,
-2.5368e+00, -1.7064e+00],
[-2.1476e+00, -6.9030e-01, -1.1092e+00, ..., -1.7034e+00,
-1.7407e+00, -1.4431e+00],
[-1.0860e+00, 1.0642e+00, 9.2065e-01, ..., -2.7600e-01,
1.7325e-02, 1.5786e-01],
...,
[-2.1868e+00, -1.6844e-01, -1.3904e-01, ..., -1.1069e+00,
-4.2354e-01, 1.5607e-01],
[-1.1858e+00, 9.0943e-01, 1.0799e+00, ..., 1.8361e-01,
3.1037e-01, 4.3392e-01],
[-2.0684e+00, -1.3211e+00, -1.4991e+00, ..., -2.2275e+00,
-1.2464e+00, -6.0384e-01]],
[[-1.6261e+00, -1.5850e+00, -2.9026e+00, ..., -1.3393e+00,
-7.5033e-01, -5.2653e-01],
[-5.6691e-01, 1.3869e-01, -3.2267e+00, ..., -2.0396e+00,
-1.6621e+00, -1.7034e+00],
[-2.0539e+00, -8.1480e-01, -5.3296e+00, ..., -4.0297e+00,
-3.3652e+00, -2.5567e+00],
...,
[-1.3692e+00, 1.2487e-01, -3.2867e+00, ..., -2.3845e+00,
-1.8079e+00, -1.3510e+00],
[-1.6782e+00, -4.8604e-01, -2.9109e+00, ..., -2.2714e+00,
-1.5183e+00, -9.0898e-01],
[-2.2788e+00, -1.3295e+00, -2.9524e+00, ..., -2.8739e+00,
-1.9302e+00, -1.6314e+00]],
[[-1.3592e+00, -1.9259e+00, -1.6892e+00, ..., -1.8510e+00,
-1.5058e+00, -1.0925e+00],
[-1.2772e+00, -1.7687e+00, -1.1970e+00, ..., -1.6204e+00,
-1.2898e+00, -9.7664e-01],
[-1.5988e+00, -1.9310e+00, -1.4262e+00, ..., -1.9008e+00,
-1.4280e+00, -1.0045e+00],
...,
[-2.4153e+00, -3.0348e+00, -2.7911e+00, ..., -3.1540e+00,
-2.4831e+00, -1.8446e+00],
[-1.5766e+00, -2.0876e+00, -2.0196e+00, ..., -2.2240e+00,
-1.8054e+00, -1.4151e+00],
[-1.0385e+00, -1.3638e+00, -1.4005e+00, ..., -1.3768e+00,
-1.1573e+00, -8.3843e-01]]]], device='cuda:0'),
tensor([[[[ 4.6457e-01, 9.3338e-01, 7.1979e-01, ..., 8.2513e-01,
1.0679e+00, 8.5461e-01],
[-1.0790e+00, -1.4576e-01, -1.4233e+00, ..., -1.1387e+00,
9.2286e-02, 6.6678e-01],
[ 7.4095e-01, 5.6243e-01, 2.7923e-01, ..., 5.4534e-01,
8.8737e-01, 9.0329e-01],
...,
[-8.9282e-01, -1.9629e+00, -2.0538e+00, ..., -1.7477e+00,
-1.2535e+00, -8.4490e-01],
[ 8.3956e-01, 4.0671e-02, -1.2476e-01, ..., -5.9508e-02,
6.8787e-01, 8.1970e-01],
[ 1.8587e-01, -5.4705e-01, -4.9107e-01, ..., -1.9585e-01,
4.0190e-01, 3.4644e-01]],
[[-2.6395e+00, -1.4532e+00, -4.7916e+00, ..., -5.8185e+00,
-4.9631e+00, -3.8962e+00],
[-6.7322e-01, 1.3373e+00, -3.3487e+00, ..., -4.7065e+00,
-4.5130e+00, -3.7071e+00],
[ 1.7607e-03, 1.1515e+00, -6.0197e+00, ..., -6.8656e+00,
-6.5225e+00, -5.9351e+00],
...,
[ 5.3878e-01, 7.5739e-01, -3.8200e+00, ..., -4.6326e+00,
-4.1923e+00, -3.5422e+00],
[ 2.7356e+00, 2.7704e+00, 3.4781e-01, ..., -8.9376e-01,
-1.2423e+00, -1.3178e+00],
[ 1.6524e+00, 1.4839e+00, -1.5702e+00, ..., -2.2641e+00,
-2.5232e+00, -2.4437e+00]],
[[-5.1789e-01, -1.5320e+00, -1.3710e+00, ..., -7.4856e-01,
-5.2031e-01, -7.2551e-01],
[-1.2597e+00, -1.6700e+00, -1.0805e+00, ..., -6.5227e-01,
-3.6476e-01, -3.0806e-01],
[-1.1350e+00, -1.0592e+00, -4.9038e-01, ..., -3.9858e-01,
-7.6079e-02, -7.7269e-02],
...,
[-2.9445e-01, -7.8554e-01, -5.0019e-01, ..., -4.7340e-01,
-1.9328e-01, -4.4213e-01],
[-8.9742e-01, -3.1443e-01, -4.8368e-01, ..., -1.8804e-01,
2.4019e-01, 2.2554e-01],
[-1.0150e+00, -2.9993e-01, -3.6496e-01, ..., -4.0604e-01,
-1.9991e-01, -2.4258e-01]],
...,
[[-2.0676e+00, -1.1897e+00, -2.5165e+00, ..., -3.2810e+00,
-2.6760e+00, -1.8313e+00],
[-5.0226e-01, 1.1259e+00, -7.1532e-01, ..., -1.6487e+00,
-1.7899e+00, -1.5661e+00],
[-2.6288e-01, 2.1442e+00, 4.3652e-01, ..., -3.5510e-01,
-4.8111e-02, 1.1373e-01],
...,
[-2.1906e+00, 1.5964e-01, -7.2004e-02, ..., -1.3701e+00,
-4.0478e-01, 5.8979e-02],
[-1.2280e+00, 1.0180e+00, 1.0855e+00, ..., 2.3004e-03,
2.9568e-01, 4.6734e-01],
[-2.1418e+00, -1.4047e+00, -1.6619e+00, ..., -2.3779e+00,
-1.4251e+00, -8.9086e-01]],
[[-1.4239e+00, -1.3696e+00, -2.7649e+00, ..., -1.0853e+00,
-5.9450e-01, -4.9073e-01],
[-8.4989e-01, -4.7437e-01, -3.7574e+00, ..., -1.9112e+00,
-1.6722e+00, -1.8421e+00],
[-2.4103e+00, -1.9321e+00, -6.3433e+00, ..., -4.1902e+00,
-3.6446e+00, -2.8575e+00],
...,
[-1.3245e+00, 2.1881e-01, -3.5553e+00, ..., -1.4341e+00,
-1.0768e+00, -9.8027e-01],
[-1.6503e+00, -4.3015e-01, -3.1389e+00, ..., -1.7034e+00,
-1.1999e+00, -8.6729e-01],
[-2.3392e+00, -1.3712e+00, -3.1276e+00, ..., -2.9049e+00,
-2.1484e+00, -2.0057e+00]],
[[-1.2135e+00, -1.7880e+00, -1.6174e+00, ..., -1.8339e+00,
-1.5163e+00, -1.1243e+00],
[-1.2714e+00, -1.7710e+00, -1.3770e+00, ..., -1.5991e+00,
-1.3036e+00, -1.0172e+00],
[-1.6758e+00, -2.0323e+00, -1.6578e+00, ..., -1.8962e+00,
-1.4599e+00, -1.0682e+00],
...,
[-2.5049e+00, -3.0929e+00, -2.7886e+00, ..., -2.8261e+00,
-2.2578e+00, -1.7335e+00],
[-1.6303e+00, -2.1144e+00, -1.9863e+00, ..., -1.9242e+00,
-1.5821e+00, -1.2734e+00],
[-1.0622e+00, -1.3636e+00, -1.3633e+00, ..., -1.1835e+00,
-1.0028e+00, -7.6283e-01]]],
[[[ 7.9688e-02, 2.9133e-01, -1.4935e+00, ..., 5.3866e-01,
1.0248e+00, 6.7349e-01],
[-2.2632e+00, -1.6374e+00, -4.8517e+00, ..., -5.1607e+00,
-2.7179e+00, -4.7696e-01],
[ 3.6013e-01, -4.1992e-01, -1.3816e+00, ..., -4.3449e+00,
-5.6057e+00, -4.6627e+00],
...,
[-2.5312e-01, -1.3047e+00, -2.6952e+00, ..., -1.8657e+00,
-1.2914e+00, -8.6530e-01],
[ 1.2498e+00, 5.9673e-01, -1.0794e+00, ..., -1.9216e-01,
5.6217e-01, 7.3392e-01],
[ 7.5199e-01, 1.6830e-01, -5.7082e-01, ..., -2.0355e-01,
3.7837e-01, 3.5277e-01]],
[[-3.3660e+00, -1.5071e+00, -6.3475e+00, ..., -4.8711e+00,
-4.4119e+00, -3.3644e+00],
[-6.3390e-01, 2.1922e+00, -4.3459e+00, ..., -3.8273e+00,
-4.7541e+00, -3.9445e+00],
[-4.8327e-02, 1.8804e+00, -7.6489e+00, ..., -6.1020e+00,
-7.4728e+00, -7.2897e+00],
...,
[-1.3982e+00, 1.6601e-01, -6.8768e+00, ..., -4.6683e+00,
-4.1787e+00, -3.4888e+00],
[ 1.3784e+00, 2.5028e+00, -1.6594e+00, ..., -8.9599e-01,
-1.1943e+00, -1.2492e+00],
[ 9.3452e-01, 1.6181e+00, -2.5480e+00, ..., -2.2597e+00,
-2.4739e+00, -2.3817e+00]],
[[-1.5301e+00, -2.4731e+00, -2.3001e+00, ..., -6.2635e-01,
-2.5301e-01, -4.8213e-01],
[-2.1568e+00, -2.2866e+00, -1.6198e+00, ..., -1.0511e+00,
-8.2632e-01, -3.9889e-01],
[-2.7777e+00, -2.1801e+00, -8.9555e-01, ..., -8.9125e-01,
-1.3231e+00, -6.8782e-01],
...,
[-1.6892e+00, -1.8461e+00, -7.3418e-01, ..., -4.9141e-01,
-1.9474e-01, -3.6765e-01],
[-1.8192e+00, -1.1924e+00, -7.2047e-01, ..., -2.9572e-01,
1.4553e-01, 2.0836e-01],
[-1.8154e+00, -1.2335e+00, -6.9199e-01, ..., -4.5686e-01,
-2.5154e-01, -2.2407e-01]],
...,
[[-2.3823e+00, -1.3208e+00, -2.7603e+00, ..., -3.5301e+00,
-2.7210e+00, -1.7845e+00],
[-2.0748e+00, 9.3905e-03, -1.4944e+00, ..., -3.0727e+00,
-4.3512e+00, -2.8781e+00],
[-5.1744e-01, 2.4949e+00, 9.0835e-01, ..., -1.8534e+00,
-4.5201e+00, -4.0773e+00],
...,
[-1.7975e+00, 1.0878e+00, 2.0216e-01, ..., -1.4621e+00,
-6.5571e-01, -9.5095e-02],
[-8.9142e-01, 1.6439e+00, 1.5411e+00, ..., -1.0182e-01,
1.2882e-01, 3.3405e-01],
[-2.1144e+00, -1.0002e+00, -7.1282e-01, ..., -2.3293e+00,
-1.4210e+00, -8.7084e-01]],
[[-2.4798e+00, -2.8079e+00, -4.1964e+00, ..., -1.1375e+00,
-7.1798e-01, -6.8951e-01],
[-1.7588e+00, -1.4932e+00, -4.9438e+00, ..., -3.8361e-01,
7.7853e-02, -9.8057e-01],
[-3.0758e+00, -2.5490e+00, -7.0201e+00, ..., -2.7241e+00,
-1.6933e+00, -1.3875e+00],
...,
[-2.6571e+00, -1.9386e+00, -5.0573e+00, ..., -1.5705e+00,
-1.1114e+00, -9.1593e-01],
[-2.2906e+00, -1.6861e+00, -3.9524e+00, ..., -1.7757e+00,
-1.1775e+00, -7.5055e-01],
[-2.3180e+00, -1.8558e+00, -3.1056e+00, ..., -2.7797e+00,
-1.9818e+00, -1.7758e+00]],
[[-1.9993e+00, -2.7512e+00, -2.5360e+00, ..., -1.7697e+00,
-1.4783e+00, -1.1186e+00],
[-2.0270e+00, -2.7778e+00, -2.2296e+00, ..., -1.8657e+00,
-1.4623e+00, -1.0520e+00],
[-2.5715e+00, -3.1768e+00, -2.6748e+00, ..., -2.9574e+00,
-2.2003e+00, -1.4097e+00],
...,
[-3.2133e+00, -4.0175e+00, -3.8023e+00, ..., -2.8576e+00,
-2.2819e+00, -1.7539e+00],
[-2.1368e+00, -2.7724e+00, -2.7367e+00, ..., -1.9771e+00,
-1.6240e+00, -1.3096e+00],
[-1.5510e+00, -1.8944e+00, -1.9391e+00, ..., -1.2216e+00,
-1.0351e+00, -7.8912e-01]]],
[[[ 9.3659e-02, 3.1337e-01, -1.2739e+00, ..., 4.7732e-01,
1.0189e+00, 6.6457e-01],
[-2.0120e+00, -1.4125e+00, -4.2823e+00, ..., -2.4760e+00,
-6.8431e-01, -9.6081e-03],
[ 3.2731e-01, -3.5703e-01, -1.1968e+00, ..., -1.9444e-01,
5.5765e-01, 6.7623e-01],
...,
[-3.1280e-01, -1.4225e+00, -2.8817e+00, ..., -1.8996e+00,
-1.2969e+00, -8.4873e-01],
[ 1.3761e+00, 6.9807e-01, -1.0248e+00, ..., -2.3496e-01,
5.4896e-01, 7.1411e-01],
[ 8.8833e-01, 3.0989e-01, -5.0240e-01, ..., -2.7801e-01,
3.2359e-01, 2.8598e-01]],
[[-3.1439e+00, -1.5153e+00, -6.1460e+00, ..., -4.8428e+00,
-4.4155e+00, -3.3578e+00],
[-7.3586e-01, 1.7497e+00, -4.4808e+00, ..., -3.4012e+00,
-3.5622e+00, -2.9027e+00],
[-2.7845e-01, 1.4052e+00, -7.5518e+00, ..., -5.3752e+00,
-5.2803e+00, -4.9366e+00],
...,
[-1.1750e+00, 5.4959e-01, -6.4250e+00, ..., -4.6561e+00,
-4.1733e+00, -3.4879e+00],
[ 1.8886e+00, 3.1188e+00, -7.9883e-01, ..., -1.0185e+00,
-1.2890e+00, -1.3080e+00],
[ 1.2611e+00, 1.9742e+00, -2.1834e+00, ..., -2.2978e+00,
-2.4901e+00, -2.3620e+00]],
[[-1.3451e+00, -2.1893e+00, -2.0189e+00, ..., -5.8780e-01,
-2.1102e-01, -4.7160e-01],
[-1.8829e+00, -2.0242e+00, -1.4394e+00, ..., -3.6689e-01,
4.9872e-02, 2.0167e-02],
[-2.4724e+00, -1.9689e+00, -8.2103e-01, ..., -2.7384e-01,
2.5049e-01, 2.7677e-01],
...,
[-1.7245e+00, -2.0061e+00, -8.0136e-01, ..., -4.7266e-01,
-1.6696e-01, -3.4539e-01],
[-1.9198e+00, -1.2940e+00, -8.0861e-01, ..., -3.1300e-01,
1.5146e-01, 2.1375e-01],
[-1.9396e+00, -1.2700e+00, -6.8181e-01, ..., -4.8579e-01,
-2.5562e-01, -2.2438e-01]],
...,
[[-2.0944e+00, -1.1547e+00, -2.4944e+00, ..., -3.6447e+00,
-2.8361e+00, -1.8447e+00],
[-1.9128e+00, 5.7093e-03, -1.3660e+00, ..., -2.5857e+00,
-2.3937e+00, -1.8275e+00],
[-5.2296e-01, 2.2493e+00, 8.3492e-01, ..., -5.3725e-01,
-8.1284e-02, 1.6598e-01],
...,
[-1.9518e+00, 1.2850e+00, 2.3819e-01, ..., -1.5015e+00,
-6.3939e-01, -1.1000e-01],
[-9.7414e-01, 1.8299e+00, 1.6849e+00, ..., -1.5005e-01,
1.1502e-01, 3.3464e-01],
[-2.2968e+00, -1.1835e+00, -9.3007e-01, ..., -2.2967e+00,
-1.4116e+00, -8.8365e-01]],
[[-2.4294e+00, -2.7643e+00, -4.0128e+00, ..., -1.1155e+00,
-7.0803e-01, -7.2045e-01],
[-1.8382e+00, -1.6432e+00, -4.7435e+00, ..., -1.3681e+00,
-1.2347e+00, -1.5990e+00],
[-3.0416e+00, -2.6133e+00, -6.6601e+00, ..., -3.2984e+00,
-2.8201e+00, -2.2617e+00],
...,
[-2.5066e+00, -1.5801e+00, -5.0798e+00, ..., -1.5905e+00,
-1.1177e+00, -9.1700e-01],
[-2.2157e+00, -1.4378e+00, -3.9786e+00, ..., -1.7944e+00,
-1.1897e+00, -7.7154e-01],
[-2.3729e+00, -1.7601e+00, -3.1389e+00, ..., -2.8001e+00,
-1.9894e+00, -1.8017e+00]],
[[-1.8861e+00, -2.5800e+00, -2.4164e+00, ..., -1.7937e+00,
-1.4888e+00, -1.1369e+00],
[-1.9304e+00, -2.6354e+00, -2.1767e+00, ..., -1.4021e+00,
-1.1053e+00, -8.8521e-01],
[-2.4675e+00, -3.0466e+00, -2.6403e+00, ..., -1.5983e+00,
-1.1957e+00, -8.8745e-01],
...,
[-3.2463e+00, -4.0742e+00, -3.7877e+00, ..., -2.8480e+00,
-2.2801e+00, -1.7569e+00],
[-2.1485e+00, -2.7943e+00, -2.7153e+00, ..., -1.9819e+00,
-1.6282e+00, -1.3113e+00],
[-1.5373e+00, -1.8778e+00, -1.9012e+00, ..., -1.2345e+00,
-1.0428e+00, -7.9764e-01]]],
[[[ 2.6417e-01, 5.4566e-01, 6.6546e-01, ..., -3.6574e-01,
-1.0054e-01, 2.0571e-03],
[-2.0791e+00, -1.3656e+00, -2.9913e+00, ..., -3.2896e+00,
-1.3272e+00, -4.6217e-01],
[-4.4416e-02, -8.7241e-01, -5.1613e-03, ..., -2.1541e-01,
-3.8312e-01, 1.5605e-01],
...,
[-9.1851e-01, -2.0478e+00, -1.9182e+00, ..., -1.9417e+00,
-1.8641e+00, -1.2499e+00],
[ 7.3127e-01, -1.5106e-01, -9.9905e-02, ..., -1.2539e-02,
2.5363e-01, 4.4111e-01],
[ 1.5051e-01, -6.0336e-01, -3.7839e-01, ..., 2.0296e-01,
4.5895e-01, 4.9736e-01]],
[[-1.7605e+00, -1.2944e+00, -4.2864e+00, ..., -6.8958e+00,
-6.0969e+00, -4.5410e+00],
[ 6.3489e-01, 1.8835e+00, -2.3134e+00, ..., -5.5116e+00,
-5.5894e+00, -4.3106e+00],
[ 1.5017e+00, 1.6691e+00, -4.9853e+00, ..., -7.9172e+00,
-7.8570e+00, -6.6273e+00],
...,
[ 3.0707e-01, 1.9844e-01, -4.2542e+00, ..., -7.1643e+00,
-6.7303e+00, -5.2232e+00],
[ 2.2846e+00, 2.1699e+00, -3.1580e-01, ..., -1.9743e+00,
-2.5985e+00, -2.0772e+00],
[ 1.3360e+00, 1.1175e+00, -1.8278e+00, ..., -2.9606e+00,
-3.5012e+00, -3.0564e+00]],
[[-2.2414e-01, -1.3620e+00, -1.5689e+00, ..., -1.4195e+00,
-1.0664e+00, -1.3573e+00],
[-7.0450e-01, -1.0476e+00, -1.0037e+00, ..., -1.0021e+00,
-6.2871e-01, -9.2249e-01],
[-5.1276e-01, -3.6811e-01, -4.6486e-01, ..., -1.1091e+00,
-6.0118e-01, -4.9224e-01],
...,
[-1.1041e-02, -4.9738e-01, -5.5719e-01, ..., -1.0610e+00,
-8.8577e-01, -1.2619e+00],
[-6.2508e-01, -1.6276e-01, -5.3125e-01, ..., -5.7337e-01,
-2.4675e-01, -4.1823e-01],
[-7.7809e-01, -2.1875e-01, -4.5550e-01, ..., -5.2912e-01,
-4.6865e-01, -6.2668e-01]],
...,
[[-3.0542e+00, -2.5041e+00, -2.6239e+00, ..., -3.0829e+00,
-2.5285e+00, -1.8408e+00],
[-2.1427e+00, -6.3958e-01, -1.0979e+00, ..., -2.4445e+00,
-2.5240e+00, -2.0407e+00],
[-1.1219e+00, 1.1105e+00, 9.0181e-01, ..., -7.4592e-02,
2.0078e-02, 2.9073e-01],
...,
[-2.1360e+00, -2.2391e-01, -1.2760e-01, ..., -8.9487e-01,
-2.6722e-01, 2.8599e-01],
[-1.1937e+00, 7.6659e-01, 9.7550e-01, ..., 4.8017e-01,
4.8823e-01, 8.1275e-01],
[-2.0563e+00, -1.3765e+00, -1.5548e+00, ..., -2.1563e+00,
-1.1528e+00, -3.5285e-01]],
[[-1.6279e+00, -1.5817e+00, -2.9044e+00, ..., -2.2517e+00,
-1.9440e+00, -1.8396e+00],
[-5.8134e-01, 1.3105e-01, -3.2679e+00, ..., -2.8777e+00,
-2.9100e+00, -2.9150e+00],
[-2.0507e+00, -8.2978e-01, -5.4210e+00, ..., -4.4536e+00,
-4.5375e+00, -3.7427e+00],
...,
[-1.4008e+00, 3.2536e-02, -3.3703e+00, ..., -2.8267e+00,
-2.7427e+00, -2.3267e+00],
[-1.6902e+00, -5.5604e-01, -2.9928e+00, ..., -2.4411e+00,
-2.2615e+00, -1.6746e+00],
[-2.2695e+00, -1.3726e+00, -2.9816e+00, ..., -2.8607e+00,
-2.4342e+00, -2.1946e+00]],
[[-1.3568e+00, -1.9132e+00, -1.6831e+00, ..., -2.5863e+00,
-2.1571e+00, -1.6800e+00],
[-1.2770e+00, -1.7641e+00, -1.1906e+00, ..., -2.4179e+00,
-1.9439e+00, -1.5968e+00],
[-1.5994e+00, -1.9343e+00, -1.4145e+00, ..., -2.8393e+00,
-2.2685e+00, -1.7673e+00],
...,
[-2.3920e+00, -2.9776e+00, -2.7393e+00, ..., -4.0240e+00,
-3.2598e+00, -2.5587e+00],
[-1.5627e+00, -2.0505e+00, -1.9766e+00, ..., -2.7823e+00,
-2.3221e+00, -1.9210e+00],
[-1.0328e+00, -1.3397e+00, -1.3721e+00, ..., -1.8382e+00,
-1.5264e+00, -1.1846e+00]]],
[[[ 2.2054e-01, 4.8803e-01, 6.4233e-01, ..., -3.3773e-01,
-9.0997e-02, 1.4894e-02],
[-2.0201e+00, -1.3088e+00, -2.7715e+00, ..., -3.1867e+00,
-1.2888e+00, -4.2221e-01],
[-1.5480e-01, -9.0642e-01, 2.7215e-02, ..., -1.9045e-01,
-3.8552e-01, 1.7582e-01],
...,
[-9.7293e-01, -2.1615e+00, -2.0746e+00, ..., -1.9460e+00,
-1.9076e+00, -1.3033e+00],
[ 8.0237e-01, -9.8974e-02, -1.0559e-01, ..., 1.3488e-02,
2.5825e-01, 4.4012e-01],
[ 2.1057e-01, -5.4578e-01, -3.8388e-01, ..., 2.0202e-01,
4.6319e-01, 5.0112e-01]],
[[-1.6565e+00, -1.3463e+00, -4.2358e+00, ..., -6.8468e+00,
-6.0636e+00, -4.5213e+00],
[ 4.8323e-01, 1.4701e+00, -2.5434e+00, ..., -5.5239e+00,
-5.6150e+00, -4.3213e+00],
[ 1.1991e+00, 1.1726e+00, -5.0945e+00, ..., -7.9551e+00,
-7.8642e+00, -6.5838e+00],
...,
[ 4.0151e-01, 3.7020e-01, -4.1362e+00, ..., -7.1635e+00,
-6.7346e+00, -5.2016e+00],
[ 2.5581e+00, 2.5100e+00, 1.2065e-01, ..., -1.9575e+00,
-2.5870e+00, -2.0482e+00],
[ 1.5146e+00, 1.3334e+00, -1.6093e+00, ..., -2.9413e+00,
-3.4878e+00, -3.0442e+00]],
[[-1.4065e-01, -1.1763e+00, -1.3984e+00, ..., -1.4102e+00,
-1.0508e+00, -1.3261e+00],
[-5.2780e-01, -8.4301e-01, -8.7997e-01, ..., -9.6213e-01,
-6.0046e-01, -9.0973e-01],
[-3.5928e-01, -2.3983e-01, -4.3766e-01, ..., -1.1142e+00,
-6.0905e-01, -4.8011e-01],
...,
[-8.0243e-02, -6.0684e-01, -6.2430e-01, ..., -1.0495e+00,
-8.8507e-01, -1.2563e+00],
[-7.2649e-01, -2.3510e-01, -5.8392e-01, ..., -5.6932e-01,
-2.5182e-01, -4.2571e-01],
[-8.8638e-01, -2.6330e-01, -4.7985e-01, ..., -5.1356e-01,
-4.5886e-01, -6.2408e-01]],
...,
[[-2.8369e+00, -2.4299e+00, -2.4176e+00, ..., -3.0141e+00,
-2.4458e+00, -1.7930e+00],
[-2.0846e+00, -8.0930e-01, -1.1134e+00, ..., -2.4303e+00,
-2.4755e+00, -1.9780e+00],
[-1.0867e+00, 8.4835e-01, 8.0282e-01, ..., -9.4404e-02,
1.3379e-02, 2.8199e-01],
...,
[-2.2613e+00, -1.6355e-01, -1.2219e-01, ..., -9.1735e-01,
-2.6108e-01, 2.8221e-01],
[-1.2219e+00, 8.8859e-01, 1.1071e+00, ..., 4.4749e-01,
4.8696e-01, 8.1323e-01],
[-2.1618e+00, -1.5023e+00, -1.6814e+00, ..., -2.1950e+00,
-1.1660e+00, -3.5925e-01]],
[[-1.6246e+00, -1.6272e+00, -2.8129e+00, ..., -2.2854e+00,
-1.9802e+00, -1.8679e+00],
[-6.8869e-01, -1.1338e-01, -3.1789e+00, ..., -2.9316e+00,
-2.9352e+00, -2.9217e+00],
[-2.0515e+00, -9.8771e-01, -5.1194e+00, ..., -4.4758e+00,
-4.5293e+00, -3.7167e+00],
...,
[-1.3261e+00, 1.9060e-01, -3.3966e+00, ..., -2.8341e+00,
-2.7447e+00, -2.3105e+00],
[-1.6696e+00, -4.7355e-01, -3.0339e+00, ..., -2.4653e+00,
-2.2629e+00, -1.6698e+00],
[-2.3228e+00, -1.3719e+00, -3.0508e+00, ..., -2.8812e+00,
-2.4282e+00, -2.2055e+00]],
[[-1.3017e+00, -1.8261e+00, -1.6433e+00, ..., -2.5634e+00,
-2.1405e+00, -1.6623e+00],
[-1.2409e+00, -1.7027e+00, -1.2203e+00, ..., -2.4131e+00,
-1.9465e+00, -1.5933e+00],
[-1.5595e+00, -1.9026e+00, -1.4806e+00, ..., -2.8384e+00,
-2.2688e+00, -1.7627e+00],
...,
[-2.4598e+00, -3.0643e+00, -2.7849e+00, ..., -4.0156e+00,
-3.2603e+00, -2.5540e+00],
[-1.5939e+00, -2.0859e+00, -1.9814e+00, ..., -2.7748e+00,
-2.3203e+00, -1.9172e+00],
[-1.0407e+00, -1.3448e+00, -1.3569e+00, ..., -1.8355e+00,
-1.5261e+00, -1.1802e+00]]]], device='cuda:0'),
tensor([[[[ 0.4360, 0.8691, 0.5154, ..., 0.4584, 1.0096, 0.6648],
[-1.3163, -0.4276, -1.9905, ..., -2.5066, -0.7171, -0.0347],
[ 0.8781, 0.6416, 0.3760, ..., -0.2688, 0.4975, 0.6607],
...,
[ 0.0113, -0.5857, -1.2953, ..., -2.2226, -1.5816, -1.0813],
[ 1.2954, 1.1314, 0.6163, ..., -0.3410, 0.5021, 0.7044],
[ 0.9573, 0.8714, 0.6018, ..., -0.3025, 0.3433, 0.3028]],
[[-2.8020, -1.5428, -4.8112, ..., -4.7320, -4.3401, -3.3146],
[-0.4982, 1.6672, -2.8907, ..., -3.3577, -3.5183, -2.8934],
[ 0.3276, 1.4831, -5.6750, ..., -5.2843, -5.1943, -4.8596],
...,
[-0.9002, 0.0348, -4.8272, ..., -4.5492, -4.0900, -3.4253],
[ 1.7137, 2.2764, -0.3256, ..., -0.5761, -0.9679, -1.0762],
[ 0.9885, 1.2358, -1.8372, ..., -2.1584, -2.4472, -2.3579]],
[[-0.6979, -1.9048, -1.7656, ..., -0.5358, -0.1571, -0.4257],
[-1.4691, -1.9444, -1.2742, ..., -0.3076, 0.0783, 0.0434],
[-1.3495, -1.3330, -0.7100, ..., -0.2647, 0.2556, 0.2921],
...,
[-0.8964, -1.4298, -0.6870, ..., -0.5382, -0.1672, -0.3963],
[-1.1942, -0.8827, -0.5860, ..., -0.3412, 0.1799, 0.2434],
[-1.2652, -0.7461, -0.3307, ..., -0.5082, -0.2600, -0.2365]],
...,
[[-2.3037, -1.1819, -2.5539, ..., -3.5428, -2.7410, -1.7886],
[-0.6196, 1.2566, -0.7457, ..., -2.6161, -2.3658, -1.8168],
[-0.2289, 2.5464, 0.7706, ..., -0.6130, -0.1219, 0.1449],
...,
[-1.3289, 1.3327, 0.0231, ..., -1.6493, -0.7022, -0.0618],
[-0.4617, 1.5060, 0.7773, ..., -0.1249, 0.1638, 0.3884],
[-1.5608, -1.0031, -1.4389, ..., -2.5694, -1.5422, -0.9541]],
[[-1.3965, -1.3737, -3.0374, ..., -1.1855, -0.7753, -0.7912],
[-0.6357, -0.1847, -3.9278, ..., -1.3871, -1.2409, -1.5932],
[-2.1564, -1.5111, -6.4709, ..., -3.2381, -2.7487, -2.2192],
...,
[-1.7720, -0.9409, -4.6958, ..., -1.3477, -0.9422, -0.8148],
[-1.6847, -0.9627, -3.7429, ..., -1.6916, -1.0942, -0.6856],
[-2.0123, -1.3872, -3.0100, ..., -2.8910, -2.0336, -1.8518]],
[[-1.2608, -1.9164, -1.7462, ..., -1.7814, -1.4822, -1.1376],
[-1.2560, -1.8193, -1.3926, ..., -1.4044, -1.1155, -0.8969],
[-1.6210, -1.9990, -1.6062, ..., -1.6182, -1.2166, -0.9122],
...,
[-2.3347, -2.9522, -2.8237, ..., -2.9609, -2.3720, -1.8191],
[-1.5835, -2.0370, -2.0312, ..., -2.0243, -1.6688, -1.3496],
[-1.0874, -1.3357, -1.4406, ..., -1.2282, -1.0569, -0.8048]]],
[[[ 0.3016, 0.5549, 0.6923, ..., 0.4685, 1.0240, 0.6111],
[-2.1869, -1.5311, -3.2205, ..., -2.7105, -0.8536, -0.1886],
[-0.1674, -1.0031, -0.2338, ..., -0.1959, 0.5951, 0.6763],
...,
[-0.2550, -1.0857, -2.4023, ..., -1.6900, -1.5991, -1.0370],
[ 1.2005, 0.7303, -0.8523, ..., -0.2961, 0.0708, 0.3390],
[ 0.6768, 0.2134, -0.6053, ..., -0.1551, 0.2521, 0.3785]],
[[-1.6780, -1.1627, -4.1605, ..., -4.8502, -4.4435, -3.3690],
[ 0.6147, 1.9555, -2.2315, ..., -3.5204, -3.6506, -2.9662],
[ 1.3614, 1.7281, -4.9045, ..., -5.4716, -5.3142, -4.9332],
...,
[-1.4881, -0.2416, -6.8678, ..., -7.5977, -6.8981, -5.3681],
[ 1.3702, 2.2799, -1.5642, ..., -2.2714, -2.7066, -2.1481],
[ 0.9043, 1.4516, -2.4968, ..., -3.1614, -3.5778, -3.1075]],
[[-0.1329, -1.1588, -1.3288, ..., -0.4659, -0.1415, -0.4751],
[-0.6048, -0.8597, -0.7785, ..., -0.2677, 0.0510, -0.0687],
[-0.5460, -0.3437, -0.3978, ..., -0.3595, 0.0855, 0.0824],
...,
[-1.1876, -1.5088, -0.6802, ..., -0.9242, -0.7839, -1.0499],
[-1.5286, -1.0207, -0.7753, ..., -0.5608, -0.2667, -0.3354],
[-1.6980, -1.1990, -0.8764, ..., -0.9517, -0.8080, -0.7918]],
...,
[[-3.0736, -2.5242, -2.8117, ..., -3.4695, -2.6853, -1.7679],
[-2.3014, -0.8566, -1.5969, ..., -2.6374, -2.4063, -1.8599],
[-1.0777, 1.1723, 0.6324, ..., -0.4222, -0.0543, 0.1091],
...,
[-1.6482, 1.0321, -0.1495, ..., -1.1864, -0.5059, 0.0690],
[-0.7415, 1.6958, 1.4416, ..., 0.4648, 0.4488, 0.7520],
[-2.0719, -1.1097, -0.9687, ..., -2.0961, -1.1141, -0.3732]],
[[-1.5999, -1.5865, -2.8974, ..., -1.2597, -0.8513, -0.8738],
[-0.5970, 0.0994, -3.2114, ..., -1.3676, -1.2239, -1.6065],
[-2.1075, -0.8952, -5.3428, ..., -3.1651, -2.6940, -2.2212],
...,
[-2.5273, -1.8112, -5.3036, ..., -2.9405, -2.6012, -2.0721],
[-2.2225, -1.6170, -4.1431, ..., -2.6305, -2.2082, -1.5429],
[-2.3040, -1.8071, -3.2333, ..., -3.0603, -2.4250, -2.1383]],
[[-1.3581, -1.9180, -1.7131, ..., -1.7615, -1.4558, -1.1147],
[-1.2795, -1.7705, -1.2376, ..., -1.3776, -1.0789, -0.8694],
[-1.6052, -1.9518, -1.4832, ..., -1.6064, -1.1995, -0.8933],
...,
[-3.0575, -3.8367, -3.7102, ..., -4.0578, -3.2655, -2.5035],
[-2.0619, -2.6671, -2.6856, ..., -2.8323, -2.3434, -1.8996],
[-1.5237, -1.8565, -1.9345, ..., -1.8784, -1.5382, -1.1726]]],
[[[ 0.0680, 0.2245, -1.4249, ..., -0.6225, -0.2151, -0.0621],
[-2.1161, -1.6248, -4.5681, ..., -3.7733, -1.5526, -0.5706],
[ 0.5745, -0.0866, -0.8715, ..., -0.1989, -0.2868, 0.2249],
...,
[-0.9137, -1.8676, -1.8442, ..., -1.8062, -1.3435, -0.9619],
[ 0.7148, -0.0172, -0.1023, ..., -0.0443, 0.6487, 0.7700],
[ 0.0950, -0.6131, -0.4881, ..., -0.1869, 0.3925, 0.3280]],
[[-3.3172, -1.7952, -6.2537, ..., -7.1451, -6.2676, -4.6307],
[-0.8041, 1.5611, -4.3344, ..., -5.4841, -5.5568, -4.2693],
[-0.3627, 1.0953, -7.4266, ..., -7.9585, -7.8492, -6.6600],
...,
[ 0.2877, 0.3709, -4.0791, ..., -4.6661, -4.1695, -3.4562],
[ 2.2779, 2.2480, -0.2367, ..., -0.8183, -1.1202, -1.1829],
[ 1.3322, 1.1457, -1.8310, ..., -2.2143, -2.4764, -2.3913]],
[[-1.3474, -2.3690, -2.3324, ..., -1.7611, -1.2446, -1.3795],
[-1.7078, -2.0112, -1.5456, ..., -1.2690, -0.7412, -0.8173],
[-2.1960, -1.9544, -1.0252, ..., -1.1235, -0.4676, -0.1819],
...,
[-0.1124, -0.5964, -0.4673, ..., -0.5062, -0.1823, -0.4511],
[-0.6762, -0.1960, -0.4418, ..., -0.1965, 0.2710, 0.2509],
[-0.7983, -0.2002, -0.3523, ..., -0.3963, -0.1846, -0.2292]],
...,
[[-2.0489, -1.0996, -2.3663, ..., -3.1765, -2.6199, -1.8298],
[-1.7698, 0.0583, -1.3459, ..., -2.4704, -2.5141, -1.9680],
[-0.3812, 2.4067, 1.1091, ..., 0.1180, 0.2696, 0.5255],
...,
[-2.0494, -0.0824, -0.1907, ..., -1.3669, -0.3825, 0.1178],
[-1.2072, 0.8184, 0.9538, ..., -0.0122, 0.3106, 0.4836],
[-2.0877, -1.3819, -1.5960, ..., -2.5060, -1.5011, -0.9315]],
[[-2.4263, -2.8237, -4.2154, ..., -2.4199, -1.9134, -1.7171],
[-1.7673, -1.6207, -4.8941, ..., -2.9049, -2.7677, -2.7567],
[-2.8375, -2.4446, -6.6859, ..., -4.4367, -4.3608, -3.4756],
...,
[-1.3941, -0.0551, -3.3996, ..., -1.3899, -0.9908, -0.9076],
[-1.6784, -0.6294, -3.0477, ..., -1.6750, -1.1324, -0.8049],
[-2.2703, -1.4675, -3.0895, ..., -2.9363, -2.1287, -1.9931]],
[[-1.8126, -2.5540, -2.4605, ..., -2.7008, -2.2180, -1.7023],
[-1.8071, -2.5258, -2.1497, ..., -2.4112, -1.9080, -1.5441],
[-2.2598, -2.8270, -2.5172, ..., -2.7555, -2.1483, -1.6316],
...,
[-2.4007, -2.9747, -2.7561, ..., -2.8742, -2.3007, -1.7613],
[-1.5678, -2.0397, -1.9735, ..., -1.9435, -1.6069, -1.2913],
[-1.0339, -1.3306, -1.3668, ..., -1.1823, -1.0129, -0.7644]]],
[[[ 0.0824, 0.2289, -1.3934, ..., -0.6376, -0.2235, -0.1007],
[-2.0691, -1.6035, -4.5191, ..., -3.7836, -1.5412, -0.6226],
[ 0.5808, -0.0870, -0.8599, ..., -0.1599, -0.2609, 0.2101],
...,
[-0.8198, -1.7926, -1.8097, ..., -1.8082, -1.3607, -0.9790],
[ 0.7782, 0.0294, -0.0923, ..., -0.0391, 0.6610, 0.7715],
[ 0.1706, -0.5423, -0.4751, ..., -0.2093, 0.3871, 0.3204]],
[[-3.3001, -1.7812, -6.2539, ..., -7.1492, -6.2161, -4.6241],
[-0.7974, 1.5532, -4.3347, ..., -5.4507, -5.4615, -4.2407],
[-0.3552, 1.1088, -7.3970, ..., -7.9343, -7.7786, -6.6783],
...,
[ 0.2735, 0.3171, -4.1325, ..., -4.6342, -4.1790, -3.4874],
[ 2.2077, 2.1747, -0.3597, ..., -0.7878, -1.1438, -1.2123],
[ 1.2925, 1.1029, -1.8870, ..., -2.2096, -2.4990, -2.4086]],
[[-1.3417, -2.3699, -2.3260, ..., -1.7471, -1.2786, -1.4032],
[-1.6928, -2.0011, -1.5426, ..., -1.2131, -0.7334, -0.8371],
[-2.1772, -1.9476, -1.0228, ..., -1.0480, -0.4136, -0.1728],
...,
[-0.1088, -0.5568, -0.4470, ..., -0.4888, -0.1904, -0.4504],
[-0.6348, -0.1349, -0.3587, ..., -0.1867, 0.2629, 0.2505],
[-0.7114, -0.1207, -0.2811, ..., -0.3949, -0.1859, -0.2308]],
...,
[[-2.0266, -1.0640, -2.3427, ..., -3.0734, -2.5379, -1.7867],
[-1.7524, 0.0717, -1.3289, ..., -2.4118, -2.4295, -1.9123],
[-0.3759, 2.4046, 1.0986, ..., 0.0955, 0.2997, 0.5554],
...,
[-2.0577, -0.1503, -0.1573, ..., -1.3717, -0.3768, 0.1222],
[-1.2620, 0.6977, 0.9216, ..., -0.0232, 0.3083, 0.4821],
[-2.1909, -1.5129, -1.6561, ..., -2.5031, -1.5018, -0.9452]],
[[-2.4186, -2.8286, -4.2250, ..., -2.3913, -1.8555, -1.6752],
[-1.7643, -1.6250, -4.8870, ..., -2.8907, -2.7268, -2.7613],
[-2.8296, -2.4437, -6.6919, ..., -4.4283, -4.3523, -3.4981],
...,
[-1.4774, -0.1721, -3.3548, ..., -1.4018, -1.0084, -0.9277],
[-1.7618, -0.7213, -3.0284, ..., -1.6873, -1.1446, -0.8202],
[-2.3439, -1.5266, -3.1167, ..., -2.9488, -2.1493, -2.0039]],
[[-1.8045, -2.5359, -2.4436, ..., -2.7375, -2.2628, -1.7225],
[-1.7987, -2.5105, -2.1378, ..., -2.4549, -1.9449, -1.5655],
[-2.2546, -2.8158, -2.4983, ..., -2.7904, -2.1779, -1.6488],
...,
[-2.3761, -2.9567, -2.7545, ..., -2.8788, -2.3062, -1.7649],
[-1.5470, -2.0295, -1.9738, ..., -1.9564, -1.6132, -1.2982],
[-1.0149, -1.3153, -1.3650, ..., -1.1905, -1.0159, -0.7677]]],
[[[ 0.2243, 0.4971, 0.6466, ..., 0.6113, 0.9675, 0.7864],
[-1.9867, -1.3005, -2.7788, ..., -1.2032, 0.1190, 0.6525],
[-0.1645, -0.8701, 0.0526, ..., 0.5593, 0.9696, 0.9667],
...,
[-0.9702, -2.1677, -2.1038, ..., -1.4135, -0.8874, -0.6419],
[ 0.7842, -0.1335, -0.1512, ..., 0.4156, 0.8163, 0.9018],
[ 0.2001, -0.5677, -0.4036, ..., 0.6195, 0.9096, 0.8079]],
[[-1.6424, -1.3706, -4.2542, ..., -5.7778, -4.8852, -3.8213],
[ 0.4826, 1.4557, -2.5643, ..., -4.7937, -4.4840, -3.6599],
[ 1.1984, 1.1891, -5.0683, ..., -6.8987, -6.3821, -5.7888],
...,
[ 0.3903, 0.3851, -4.1150, ..., -6.0429, -5.1999, -4.2999],
[ 2.5593, 2.5289, 0.1534, ..., -1.3536, -1.4719, -1.5591],
[ 1.5188, 1.3445, -1.6146, ..., -2.7502, -2.8528, -2.8029]],
[[-0.1228, -1.1535, -1.3864, ..., -0.7819, -0.5310, -0.6466],
[-0.4882, -0.8212, -0.8696, ..., -0.6504, -0.3566, -0.2244],
[-0.3225, -0.2281, -0.4345, ..., -0.3927, -0.0190, 0.1109],
...,
[-0.0810, -0.6225, -0.6100, ..., -0.6896, -0.4785, -0.5833],
[-0.7273, -0.2339, -0.5824, ..., -0.3076, -0.0168, 0.0657],
[-0.8808, -0.2658, -0.4768, ..., -0.3076, -0.2097, -0.1471]],
...,
[[-2.8109, -2.4440, -2.4682, ..., -3.1116, -2.5013, -1.6863],
[-2.0981, -0.8603, -1.1624, ..., -1.7249, -1.6811, -1.3537],
[-1.1243, 0.7506, 0.7039, ..., -0.4094, -0.0104, 0.1785],
...,
[-2.2771, -0.1620, -0.1446, ..., -1.0307, -0.3896, 0.1829],
[-1.2223, 0.8883, 1.0938, ..., 0.2584, 0.3485, 0.4603],
[-2.1455, -1.4819, -1.6743, ..., -2.2111, -1.2450, -0.6152]],
[[-1.6265, -1.6412, -2.8127, ..., -1.4333, -0.7886, -0.5556],
[-0.7058, -0.1239, -3.1585, ..., -2.1185, -1.6619, -1.7129],
[-2.0553, -0.9960, -5.1235, ..., -4.0785, -3.3522, -2.5657],
...,
[-1.3238, 0.2023, -3.3814, ..., -2.3678, -1.7960, -1.3620],
[-1.6564, -0.4470, -3.0122, ..., -2.2569, -1.5071, -0.8943],
[-2.3181, -1.3421, -3.0374, ..., -2.8770, -1.9388, -1.6259]],
[[-1.3022, -1.8253, -1.6518, ..., -1.8596, -1.5186, -1.1126],
[-1.2424, -1.7057, -1.2314, ..., -1.6474, -1.3226, -0.9986],
[-1.5640, -1.9023, -1.4886, ..., -1.9304, -1.4601, -1.0304],
...,
[-2.4581, -3.0665, -2.7911, ..., -3.1915, -2.5098, -1.8619],
[-1.5933, -2.0896, -1.9822, ..., -2.2492, -1.8233, -1.4321],
[-1.0369, -1.3489, -1.3613, ..., -1.3934, -1.1712, -0.8511]]]],
device='cuda:0'),
tensor([[[[ 2.2766e-01, 5.2169e-01, 6.4386e-01, ..., -4.3112e-01,
-1.5144e-01, -1.7471e-02],
[-1.9489e+00, -1.2196e+00, -2.7135e+00, ..., -3.3744e+00,
-1.3567e+00, -4.8311e-01],
[-1.3162e-01, -8.5639e-01, 4.2590e-02, ..., -2.8356e-01,
-4.3153e-01, 1.4725e-01],
...,
[-9.4932e-01, -2.0989e+00, -2.0095e+00, ..., -1.9497e+00,
-1.9194e+00, -1.3093e+00],
[ 8.1089e-01, -5.7547e-02, -1.1019e-01, ..., -4.6271e-02,
1.8656e-01, 3.7575e-01],
[ 2.4476e-01, -5.0565e-01, -3.7209e-01, ..., 1.5597e-01,
4.2288e-01, 4.6900e-01]],
[[-1.7034e+00, -1.3782e+00, -4.2661e+00, ..., -6.9075e+00,
-6.1546e+00, -4.5767e+00],
[ 4.1893e-01, 1.4119e+00, -2.5748e+00, ..., -5.5365e+00,
-5.6664e+00, -4.3605e+00],
[ 1.1278e+00, 1.0913e+00, -5.0899e+00, ..., -7.9958e+00,
-7.9572e+00, -6.6712e+00],
...,
[ 3.1343e-01, 3.2916e-01, -4.1476e+00, ..., -7.2635e+00,
-6.8334e+00, -5.2307e+00],
[ 2.5159e+00, 2.5054e+00, 1.5889e-01, ..., -2.0266e+00,
-2.6491e+00, -2.0581e+00],
[ 1.5026e+00, 1.3370e+00, -1.5975e+00, ..., -2.9651e+00,
-3.5175e+00, -3.0439e+00]],
[[-1.5964e-01, -1.2024e+00, -1.4273e+00, ..., -1.4473e+00,
-1.0962e+00, -1.3705e+00],
[-5.4661e-01, -8.6437e-01, -9.1097e-01, ..., -1.0045e+00,
-6.3866e-01, -9.3724e-01],
[-3.7630e-01, -2.4807e-01, -4.5342e-01, ..., -1.1464e+00,
-6.2643e-01, -4.9768e-01],
...,
[-1.1232e-01, -6.7245e-01, -6.3167e-01, ..., -1.0507e+00,
-8.7612e-01, -1.2440e+00],
[-7.5528e-01, -2.7557e-01, -5.8669e-01, ..., -5.6101e-01,
-2.4415e-01, -4.1691e-01],
[-9.2086e-01, -3.1098e-01, -4.8016e-01, ..., -5.3442e-01,
-4.9787e-01, -6.6069e-01]],
...,
[[-2.7601e+00, -2.3577e+00, -2.4229e+00, ..., -3.0508e+00,
-2.5116e+00, -1.8359e+00],
[-1.9976e+00, -7.2526e-01, -1.0422e+00, ..., -2.4838e+00,
-2.5652e+00, -2.0455e+00],
[-1.0677e+00, 8.4674e-01, 8.0804e-01, ..., -6.1344e-02,
-2.1001e-02, 2.6253e-01],
...,
[-2.2270e+00, -1.1339e-01, -1.2927e-01, ..., -9.5677e-01,
-3.0905e-01, 2.5889e-01],
[-1.1720e+00, 9.1310e-01, 1.0673e+00, ..., 4.6312e-01,
4.7756e-01, 8.2351e-01],
[-2.1225e+00, -1.4650e+00, -1.6780e+00, ..., -2.1896e+00,
-1.1561e+00, -3.4109e-01]],
[[-1.6168e+00, -1.6288e+00, -2.7770e+00, ..., -2.3081e+00,
-2.0032e+00, -1.9011e+00],
[-7.0290e-01, -1.3064e-01, -3.1533e+00, ..., -2.9352e+00,
-2.9734e+00, -2.9734e+00],
[-2.0733e+00, -1.0531e+00, -5.1403e+00, ..., -4.4830e+00,
-4.5875e+00, -3.7858e+00],
...,
[-1.3292e+00, 1.5927e-01, -3.4318e+00, ..., -2.8621e+00,
-2.7443e+00, -2.3067e+00],
[-1.6463e+00, -4.5793e-01, -3.0395e+00, ..., -2.4666e+00,
-2.2719e+00, -1.6819e+00],
[-2.2979e+00, -1.3466e+00, -3.0168e+00, ..., -2.8814e+00,
-2.4453e+00, -2.2247e+00]],
[[-1.2929e+00, -1.8205e+00, -1.6462e+00, ..., -2.6212e+00,
-2.1868e+00, -1.7056e+00],
[-1.2387e+00, -1.7046e+00, -1.2301e+00, ..., -2.4626e+00,
-1.9793e+00, -1.6267e+00],
[-1.5624e+00, -1.8936e+00, -1.4906e+00, ..., -2.8937e+00,
-2.3087e+00, -1.7983e+00],
...,
[-2.4447e+00, -3.0498e+00, -2.7806e+00, ..., -4.0853e+00,
-3.3194e+00, -2.6083e+00],
[-1.5857e+00, -2.0826e+00, -1.9776e+00, ..., -2.8139e+00,
-2.3554e+00, -1.9496e+00],
[-1.0369e+00, -1.3421e+00, -1.3576e+00, ..., -1.8588e+00,
-1.5431e+00, -1.2000e+00]]],
[[[ 1.4675e-01, 3.7382e-01, -1.1827e+00, ..., 6.7345e-01,
9.3833e-01, 6.8809e-01],
[-2.1005e+00, -1.4415e+00, -4.3645e+00, ..., -1.4810e+00,
-1.8738e-01, 3.7692e-01],
[ 4.3643e-01, -2.9476e-01, -1.1461e+00, ..., 5.2525e-01,
9.0583e-01, 8.2682e-01],
...,
[-2.5702e-01, -1.2739e+00, -2.6310e+00, ..., -1.2053e+00,
-8.0531e-01, -6.1591e-01],
[ 1.2486e+00, 6.4688e-01, -9.3481e-01, ..., 5.2293e-01,
7.7418e-01, 8.1603e-01],
[ 8.2438e-01, 3.0912e-01, -4.3694e-01, ..., 5.4576e-01,
7.8945e-01, 6.8505e-01]],
[[-3.2404e+00, -1.4171e+00, -6.0726e+00, ..., -5.4495e+00,
-4.6682e+00, -3.6480e+00],
[-5.5410e-01, 2.1992e+00, -4.1519e+00, ..., -4.2181e+00,
-4.0997e+00, -3.3876e+00],
[ 6.3396e-02, 1.8902e+00, -7.4067e+00, ..., -6.1650e+00,
-5.9192e+00, -5.4829e+00],
...,
[-1.2405e+00, 2.7683e-01, -6.5020e+00, ..., -5.3722e+00,
-4.7351e+00, -3.9479e+00],
[ 1.5424e+00, 2.6296e+00, -1.3140e+00, ..., -1.2905e+00,
-1.4426e+00, -1.5125e+00],
[ 1.0068e+00, 1.6518e+00, -2.3894e+00, ..., -2.5130e+00,
-2.6290e+00, -2.5578e+00]],
[[-1.3778e+00, -2.3436e+00, -2.1543e+00, ..., -8.4868e-01,
-6.3087e-01, -8.0764e-01],
[-2.0281e+00, -2.1986e+00, -1.5455e+00, ..., -6.9363e-01,
-4.1712e-01, -3.5620e-01],
[-2.5410e+00, -2.0073e+00, -8.3262e-01, ..., -4.0790e-01,
-1.1987e-01, -1.0661e-01],
...,
[-1.6085e+00, -1.8311e+00, -7.9681e-01, ..., -6.4000e-01,
-5.3434e-01, -7.0090e-01],
[-1.7707e+00, -1.1892e+00, -7.7749e-01, ..., -2.3525e-01,
-5.5389e-02, -4.3802e-02],
[-1.7124e+00, -1.0986e+00, -5.9517e-01, ..., -3.3879e-01,
-2.8884e-01, -2.5623e-01]],
...,
[[-2.3759e+00, -1.3031e+00, -2.7177e+00, ..., -3.0677e+00,
-2.4297e+00, -1.6153e+00],
[-1.8783e+00, 2.1573e-01, -1.3108e+00, ..., -1.5120e+00,
-1.5743e+00, -1.3631e+00],
[-4.7900e-01, 2.5377e+00, 9.8851e-01, ..., -7.0132e-02,
2.5881e-01, 3.3819e-01],
...,
[-1.7246e+00, 1.2008e+00, 2.4311e-01, ..., -9.7876e-01,
-2.8845e-01, 1.9206e-01],
[-8.4169e-01, 1.6772e+00, 1.5229e+00, ..., 1.6515e-01,
3.0853e-01, 4.1270e-01],
[-2.1048e+00, -1.0566e+00, -8.5329e-01, ..., -2.1551e+00,
-1.2613e+00, -6.4908e-01]],
[[-2.3551e+00, -2.6176e+00, -4.0082e+00, ..., -1.1305e+00,
-6.3612e-01, -5.1959e-01],
[-1.6305e+00, -1.3039e+00, -4.7641e+00, ..., -1.7159e+00,
-1.5143e+00, -1.7290e+00],
[-2.9829e+00, -2.4006e+00, -6.9024e+00, ..., -3.7391e+00,
-3.2653e+00, -2.6163e+00],
...,
[-2.4960e+00, -1.6982e+00, -4.9501e+00, ..., -2.1849e+00,
-1.7521e+00, -1.4201e+00],
[-2.2066e+00, -1.5228e+00, -3.8584e+00, ..., -2.1358e+00,
-1.5005e+00, -9.7415e-01],
[-2.3055e+00, -1.7504e+00, -3.0798e+00, ..., -2.8261e+00,
-1.9536e+00, -1.6562e+00]],
[[-1.8968e+00, -2.6280e+00, -2.4089e+00, ..., -1.6881e+00,
-1.3954e+00, -1.0396e+00],
[-1.9213e+00, -2.6371e+00, -2.0962e+00, ..., -1.3615e+00,
-1.1064e+00, -8.7614e-01],
[-2.4462e+00, -3.0056e+00, -2.5172e+00, ..., -1.5732e+00,
-1.2018e+00, -8.7754e-01],
...,
[-3.1129e+00, -3.8860e+00, -3.6532e+00, ..., -2.7782e+00,
-2.2164e+00, -1.6767e+00],
[-2.0662e+00, -2.6681e+00, -2.6293e+00, ..., -1.9505e+00,
-1.6043e+00, -1.2780e+00],
[-1.4796e+00, -1.7990e+00, -1.8471e+00, ..., -1.2194e+00,
-1.0404e+00, -7.6899e-01]]],
[[[-1.0566e-02, 1.1359e-01, -1.7731e+00, ..., -8.2724e-01,
-4.0170e-01, -1.7743e-01],
[-2.2547e+00, -1.8331e+00, -5.0435e+00, ..., -4.1438e+00,
-1.7809e+00, -7.3187e-01],
[ 4.9752e-01, -2.2168e-01, -1.1139e+00, ..., -3.2059e-01,
-4.8713e-01, 1.1216e-01],
...,
[-7.2194e-01, -1.6877e+00, -1.8601e+00, ..., -3.1082e+00,
-1.1086e+00, -7.9472e-01],
[ 9.6305e-01, 2.5435e-01, 4.7835e-02, ..., -1.2795e+00,
7.4911e-01, 8.2792e-01],
[ 3.4162e-01, -3.0927e-01, -3.0760e-01, ..., -3.9407e-01,
5.1854e-01, 4.2397e-01]],
[[-3.4302e+00, -1.8667e+00, -6.5001e+00, ..., -7.4200e+00,
-6.4978e+00, -4.7713e+00],
[-8.6616e-01, 1.5446e+00, -4.5752e+00, ..., -5.7087e+00,
-5.7916e+00, -4.4353e+00],
[-4.6875e-01, 1.0846e+00, -7.7319e+00, ..., -8.2990e+00,
-8.1494e+00, -6.8792e+00],
...,
[ 3.6965e-01, 7.3301e-01, -3.8705e+00, ..., -4.7543e+00,
-4.2696e+00, -3.5569e+00],
[ 2.6730e+00, 2.8102e+00, 3.9112e-01, ..., -7.9961e-01,
-1.2943e+00, -1.3241e+00],
[ 1.6080e+00, 1.5132e+00, -1.5317e+00, ..., -2.0977e+00,
-2.5123e+00, -2.4214e+00]],
[[-1.5045e+00, -2.4888e+00, -2.4386e+00, ..., -1.8913e+00,
-1.3382e+00, -1.4786e+00],
[-1.8159e+00, -2.0718e+00, -1.5822e+00, ..., -1.3596e+00,
-8.0480e-01, -9.1794e-01],
[-2.4044e+00, -2.1108e+00, -1.0637e+00, ..., -1.2387e+00,
-5.6073e-01, -2.4329e-01],
...,
[-3.9779e-01, -8.9314e-01, -5.0022e-01, ..., -4.2843e-01,
-2.0231e-01, -4.4191e-01],
[-9.5055e-01, -3.9581e-01, -4.6114e-01, ..., 1.3971e-02,
2.3445e-01, 2.2294e-01],
[-1.0726e+00, -3.7006e-01, -3.4428e-01, ..., -4.9513e-02,
-1.7247e-01, -2.0431e-01]],
...,
[[-2.0561e+00, -1.1240e+00, -2.3484e+00, ..., -3.1902e+00,
-2.6151e+00, -1.8319e+00],
[-1.9870e+00, -1.8329e-01, -1.5155e+00, ..., -2.6767e+00,
-2.6944e+00, -2.0761e+00],
[-4.7322e-01, 2.3202e+00, 1.0615e+00, ..., 1.6497e-01,
2.7309e-01, 5.2323e-01],
...,
[-2.1154e+00, 3.4623e-01, -3.6971e-02, ..., -1.1098e+00,
-3.6791e-01, 1.1999e-01],
[-1.1474e+00, 1.1220e+00, 1.0879e+00, ..., 3.7747e-01,
2.9868e-01, 4.3749e-01],
[-2.1148e+00, -1.3964e+00, -1.6617e+00, ..., -2.0403e+00,
-1.3901e+00, -8.6667e-01]],
[[-2.5581e+00, -3.0368e+00, -4.3894e+00, ..., -2.5675e+00,
-2.0921e+00, -1.8910e+00],
[-1.9144e+00, -1.8595e+00, -5.0818e+00, ..., -3.0769e+00,
-2.9625e+00, -2.9194e+00],
[-2.9425e+00, -2.6472e+00, -6.7993e+00, ..., -4.5643e+00,
-4.5293e+00, -3.6096e+00],
...,
[-1.3468e+00, 1.0413e-01, -3.6761e+00, ..., -1.7318e+00,
-1.1376e+00, -1.0011e+00],
[-1.6351e+00, -4.8296e-01, -3.2110e+00, ..., -1.7842e+00,
-1.2121e+00, -8.3820e-01],
[-2.3175e+00, -1.3809e+00, -3.0964e+00, ..., -2.5734e+00,
-2.0798e+00, -1.8994e+00]],
[[-1.9161e+00, -2.6740e+00, -2.6056e+00, ..., -2.8626e+00,
-2.3431e+00, -1.7955e+00],
[-1.9194e+00, -2.6652e+00, -2.3070e+00, ..., -2.5742e+00,
-2.0318e+00, -1.6451e+00],
[-2.3807e+00, -3.0004e+00, -2.7006e+00, ..., -2.9512e+00,
-2.3053e+00, -1.7566e+00],
...,
[-2.4797e+00, -3.0608e+00, -2.7599e+00, ..., -2.8819e+00,
-2.2380e+00, -1.7174e+00],
[-1.6134e+00, -2.0873e+00, -1.9626e+00, ..., -1.9365e+00,
-1.5690e+00, -1.2640e+00],
[-1.0542e+00, -1.3406e+00, -1.3461e+00, ..., -1.1982e+00,
-1.0033e+00, -7.5725e-01]]],
[[[ 3.5469e-01, 8.2145e-01, 7.5643e-01, ..., 7.1612e-01,
9.5884e-01, 7.4531e-01],
[-1.1774e+00, -2.7239e-01, -1.3201e+00, ..., -1.0473e+00,
6.0437e-02, 5.6884e-01],
[ 6.6860e-01, 4.8352e-01, 5.4472e-01, ..., 6.0240e-01,
9.6059e-01, 8.9800e-01],
...,
[-1.4989e-01, -1.1065e+00, -2.6067e+00, ..., -1.4926e+00,
-1.3956e+00, -8.4316e-01],
[ 1.4533e+00, 8.7036e-01, -9.2120e-01, ..., -2.5946e-01,
1.0789e-01, 3.9517e-01],
[ 8.4904e-01, 2.9083e-01, -6.7350e-01, ..., -1.5942e-01,
1.9011e-01, 3.9120e-01]],
[[-2.3300e+00, -1.2965e+00, -4.2840e+00, ..., -5.2682e+00,
-4.5161e+00, -3.5411e+00],
[-2.0976e-01, 1.5874e+00, -2.6060e+00, ..., -4.1192e+00,
-4.0072e+00, -3.2960e+00],
[ 6.2080e-01, 1.4441e+00, -5.0837e+00, ..., -5.9912e+00,
-5.7948e+00, -5.3342e+00],
...,
[-1.2779e+00, 1.4331e-01, -6.6958e+00, ..., -7.6706e+00,
-6.9589e+00, -5.4683e+00],
[ 1.7721e+00, 2.7876e+00, -1.0764e+00, ..., -2.6336e+00,
-3.0015e+00, -2.4187e+00],
[ 1.1984e+00, 1.7941e+00, -2.2974e+00, ..., -3.2903e+00,
-3.6317e+00, -3.1542e+00]],
[[-6.7125e-01, -1.7904e+00, -1.6688e+00, ..., -9.3188e-01,
-6.6786e-01, -7.7816e-01],
[-1.3450e+00, -1.8066e+00, -1.2361e+00, ..., -7.2594e-01,
-4.0154e-01, -2.8325e-01],
[-1.0520e+00, -9.9887e-01, -4.5898e-01, ..., -3.1865e-01,
3.4523e-02, 5.3366e-02],
...,
[-1.4000e+00, -1.7486e+00, -7.6114e-01, ..., -8.8842e-01,
-7.7931e-01, -1.0337e+00],
[-1.7329e+00, -1.1472e+00, -8.2654e-01, ..., -5.5984e-01,
-2.9837e-01, -3.6555e-01],
[-1.8762e+00, -1.2633e+00, -8.7329e-01, ..., -9.5030e-01,
-8.3389e-01, -8.1255e-01]],
...,
[[-2.2085e+00, -1.2576e+00, -2.2500e+00, ..., -2.9121e+00,
-2.2830e+00, -1.5019e+00],
[-6.3425e-01, 1.0447e+00, -3.9514e-01, ..., -1.3306e+00,
-1.3884e+00, -1.2034e+00],
[-3.9691e-01, 2.0012e+00, 7.4017e-01, ..., -1.5116e-01,
2.3686e-01, 3.5059e-01],
...,
[-1.8183e+00, 1.2618e+00, -1.3103e-03, ..., -1.1003e+00,
-5.3210e-01, -1.5958e-02],
[-8.2361e-01, 1.9301e+00, 1.6514e+00, ..., 4.8776e-01,
4.5985e-01, 6.8644e-01],
[-2.2397e+00, -1.1534e+00, -9.7979e-01, ..., -1.8771e+00,
-9.5029e-01, -2.8237e-01]],
[[-1.3660e+00, -1.2828e+00, -2.7362e+00, ..., -1.0982e+00,
-5.8473e-01, -4.6311e-01],
[-6.9746e-01, -2.0151e-01, -3.5551e+00, ..., -1.7887e+00,
-1.5362e+00, -1.6965e+00],
[-2.1162e+00, -1.4056e+00, -5.9125e+00, ..., -3.7414e+00,
-3.2438e+00, -2.5648e+00],
...,
[-2.4876e+00, -1.5961e+00, -5.4805e+00, ..., -3.1240e+00,
-2.7566e+00, -2.1757e+00],
[-2.2139e+00, -1.4825e+00, -4.2741e+00, ..., -2.7146e+00,
-2.3356e+00, -1.6450e+00],
[-2.3630e+00, -1.7739e+00, -3.3357e+00, ..., -2.9852e+00,
-2.4499e+00, -2.1252e+00]],
[[-1.1402e+00, -1.6891e+00, -1.4922e+00, ..., -1.6455e+00,
-1.3663e+00, -1.0279e+00],
[-1.1545e+00, -1.6126e+00, -1.1670e+00, ..., -1.3730e+00,
-1.1118e+00, -8.8153e-01],
[-1.5059e+00, -1.7964e+00, -1.3451e+00, ..., -1.5597e+00,
-1.1892e+00, -8.6769e-01],
...,
[-3.1886e+00, -3.9849e+00, -3.7692e+00, ..., -3.9616e+00,
-3.1720e+00, -2.4431e+00],
[-2.1283e+00, -2.7411e+00, -2.7042e+00, ..., -2.7772e+00,
-2.2896e+00, -1.8611e+00],
[-1.5528e+00, -1.8764e+00, -1.9344e+00, ..., -1.8740e+00,
-1.5231e+00, -1.1632e+00]]],
[[[ 5.7372e-02, 2.1924e-01, -1.4222e+00, ..., -6.3979e-01,
-2.0503e-01, -5.6529e-02],
[-2.1214e+00, -1.6056e+00, -4.5352e+00, ..., -3.8150e+00,
-1.5511e+00, -5.6765e-01],
[ 5.2988e-01, -9.3728e-02, -8.4825e-01, ..., -1.9626e-01,
-2.7478e-01, 2.3103e-01],
...,
[-8.7434e-01, -1.8448e+00, -1.8647e+00, ..., -1.8011e+00,
-1.3092e+00, -9.3267e-01],
[ 7.4916e-01, -5.6331e-03, -1.2506e-01, ..., -6.7573e-02,
6.8647e-01, 7.8643e-01],
[ 1.4122e-01, -5.6513e-01, -4.7896e-01, ..., -1.7596e-01,
4.3825e-01, 3.5115e-01]],
[[-3.2840e+00, -1.7827e+00, -6.2377e+00, ..., -7.1718e+00,
-6.2688e+00, -4.6329e+00],
[-7.6954e-01, 1.5748e+00, -4.3484e+00, ..., -5.4892e+00,
-5.5432e+00, -4.2824e+00],
[-3.3260e-01, 1.1187e+00, -7.4585e+00, ..., -8.0128e+00,
-7.8329e+00, -6.6961e+00],
...,
[ 2.6665e-01, 3.2223e-01, -4.0395e+00, ..., -4.6088e+00,
-4.1929e+00, -3.4799e+00],
[ 2.2816e+00, 2.2639e+00, -1.4614e-01, ..., -8.1257e-01,
-1.1867e+00, -1.2191e+00],
[ 1.3347e+00, 1.1706e+00, -1.7505e+00, ..., -2.2415e+00,
-2.5083e+00, -2.4031e+00]],
[[-1.3444e+00, -2.3645e+00, -2.3260e+00, ..., -1.7672e+00,
-1.2436e+00, -1.3875e+00],
[-1.6902e+00, -1.9856e+00, -1.5190e+00, ..., -1.2617e+00,
-7.2941e-01, -8.1798e-01],
[-2.1664e+00, -1.9269e+00, -9.9678e-01, ..., -1.1090e+00,
-4.5919e-01, -1.8655e-01],
...,
[-1.5076e-01, -6.1263e-01, -4.7343e-01, ..., -4.8550e-01,
-1.8645e-01, -4.3903e-01],
[-6.8871e-01, -1.8423e-01, -4.2475e-01, ..., -2.0486e-01,
2.5150e-01, 2.4209e-01],
[-7.7320e-01, -1.6733e-01, -3.3427e-01, ..., -3.9352e-01,
-1.8971e-01, -2.3301e-01]],
...,
[[-2.0304e+00, -1.1143e+00, -2.3684e+00, ..., -3.2192e+00,
-2.6436e+00, -1.8299e+00],
[-1.7465e+00, 1.3454e-02, -1.3904e+00, ..., -2.4960e+00,
-2.5370e+00, -1.9768e+00],
[-3.5030e-01, 2.3752e+00, 1.0627e+00, ..., 1.3374e-01,
2.8580e-01, 5.3412e-01],
...,
[-2.0604e+00, -1.0437e-01, -2.0406e-01, ..., -1.3635e+00,
-4.3708e-01, 8.8117e-02],
[-1.1984e+00, 7.9340e-01, 9.2855e-01, ..., 3.2125e-02,
2.9300e-01, 4.6551e-01],
[-2.1168e+00, -1.4501e+00, -1.6597e+00, ..., -2.4406e+00,
-1.4737e+00, -9.1802e-01]],
[[-2.4134e+00, -2.8239e+00, -4.2159e+00, ..., -2.3832e+00,
-1.8964e+00, -1.7103e+00],
[-1.7526e+00, -1.6251e+00, -4.8900e+00, ..., -2.9097e+00,
-2.7725e+00, -2.7778e+00],
[-2.8150e+00, -2.4490e+00, -6.6818e+00, ..., -4.4599e+00,
-4.3626e+00, -3.4915e+00],
...,
[-1.4134e+00, -3.3364e-02, -3.3209e+00, ..., -1.4079e+00,
-9.9844e-01, -9.2756e-01],
[-1.7234e+00, -6.5049e-01, -3.0114e+00, ..., -1.7131e+00,
-1.1536e+00, -8.1499e-01],
[-2.3434e+00, -1.5110e+00, -3.1150e+00, ..., -2.9440e+00,
-2.1412e+00, -1.9799e+00]],
[[-1.8120e+00, -2.5444e+00, -2.4585e+00, ..., -2.7085e+00,
-2.2240e+00, -1.7011e+00],
[-1.8114e+00, -2.5240e+00, -2.1591e+00, ..., -2.4108e+00,
-1.9056e+00, -1.5432e+00],
[-2.2682e+00, -2.8280e+00, -2.5240e+00, ..., -2.7564e+00,
-2.1470e+00, -1.6312e+00],
...,
[-2.3784e+00, -2.9545e+00, -2.7380e+00, ..., -2.8715e+00,
-2.2917e+00, -1.7641e+00],
[-1.5362e+00, -2.0035e+00, -1.9343e+00, ..., -1.9475e+00,
-1.6015e+00, -1.2939e+00],
[-9.9508e-01, -1.2879e+00, -1.3238e+00, ..., -1.1877e+00,
-1.0125e+00, -7.7157e-01]]]], device='cuda:0'),
tensor([[[[ 5.4996e-02, 2.0341e-01, -1.4090e+00, ..., -6.3745e-01,
-2.0227e-01, -5.3609e-02],
[-2.0888e+00, -1.6476e+00, -4.5244e+00, ..., -3.8149e+00,
-1.5512e+00, -5.6457e-01],
[ 5.8971e-01, -1.4264e-01, -9.0936e-01, ..., -2.0763e-01,
-2.7778e-01, 2.1702e-01],
...,
[-8.5967e-01, -1.8146e+00, -1.8512e+00, ..., -1.8423e+00,
-1.3773e+00, -9.7443e-01],
[ 7.3920e-01, -1.0845e-02, -1.1704e-01, ..., -3.2744e-02,
6.8800e-01, 7.9989e-01],
[ 1.3549e-01, -5.8003e-01, -4.8503e-01, ..., -1.6854e-01,
4.2392e-01, 3.4139e-01]],
[[-3.3008e+00, -1.7692e+00, -6.2096e+00, ..., -7.1408e+00,
-6.2809e+00, -4.6193e+00],
[-8.1426e-01, 1.5594e+00, -4.3112e+00, ..., -5.4465e+00,
-5.5534e+00, -4.2664e+00],
[-3.8094e-01, 1.1013e+00, -7.3852e+00, ..., -7.9506e+00,
-7.8314e+00, -6.6635e+00],
...,
[ 2.8468e-01, 2.9891e-01, -4.0639e+00, ..., -4.6379e+00,
-4.1888e+00, -3.4911e+00],
[ 2.2685e+00, 2.2226e+00, -1.9166e-01, ..., -7.7545e-01,
-1.1400e+00, -1.1951e+00],
[ 1.3257e+00, 1.1471e+00, -1.7753e+00, ..., -2.2205e+00,
-2.5006e+00, -2.4048e+00]],
[[-1.3301e+00, -2.3305e+00, -2.2925e+00, ..., -1.7730e+00,
-1.2338e+00, -1.3753e+00],
[-1.6822e+00, -1.9659e+00, -1.4817e+00, ..., -1.2713e+00,
-7.1966e-01, -7.9922e-01],
[-2.1592e+00, -1.9201e+00, -9.4990e-01, ..., -1.1121e+00,
-4.6178e-01, -1.7843e-01],
...,
[-1.2122e-01, -5.7197e-01, -4.6794e-01, ..., -5.0083e-01,
-1.9489e-01, -4.4921e-01],
[-6.8394e-01, -1.6407e-01, -4.1619e-01, ..., -2.0522e-01,
2.5417e-01, 2.4656e-01],
[-7.7748e-01, -1.6860e-01, -3.2633e-01, ..., -4.0027e-01,
-1.9654e-01, -2.3156e-01]],
...,
[[-2.0164e+00, -1.0565e+00, -2.3059e+00, ..., -3.1957e+00,
-2.6373e+00, -1.8165e+00],
[-1.7431e+00, 9.8673e-02, -1.3016e+00, ..., -2.4586e+00,
-2.5429e+00, -1.9943e+00],
[-3.8716e-01, 2.4347e+00, 1.1293e+00, ..., 1.5790e-01,
2.8972e-01, 5.3099e-01],
...,
[-2.0917e+00, -1.7182e-01, -1.8225e-01, ..., -1.4125e+00,
-3.9799e-01, 1.0143e-01],
[-1.2242e+00, 7.2170e-01, 9.0429e-01, ..., -9.7687e-03,
3.1521e-01, 4.8156e-01],
[-2.1269e+00, -1.4726e+00, -1.6809e+00, ..., -2.4962e+00,
-1.4888e+00, -9.3368e-01]],
[[-2.4196e+00, -2.8102e+00, -4.1815e+00, ..., -2.3728e+00,
-1.8861e+00, -1.7023e+00],
[-1.7677e+00, -1.6316e+00, -4.8556e+00, ..., -2.8894e+00,
-2.7596e+00, -2.7510e+00],
[-2.8232e+00, -2.4433e+00, -6.6310e+00, ..., -4.4434e+00,
-4.3508e+00, -3.4558e+00],
...,
[-1.4173e+00, -7.7125e-02, -3.3499e+00, ..., -1.4106e+00,
-1.0093e+00, -9.2297e-01],
[-1.7305e+00, -6.8192e-01, -3.0210e+00, ..., -1.7068e+00,
-1.1481e+00, -8.0174e-01],
[-2.3368e+00, -1.5178e+00, -3.1085e+00, ..., -2.9681e+00,
-2.1440e+00, -1.9961e+00]],
[[-1.8071e+00, -2.5370e+00, -2.4558e+00, ..., -2.7084e+00,
-2.2172e+00, -1.6988e+00],
[-1.8051e+00, -2.5175e+00, -2.1536e+00, ..., -2.4058e+00,
-1.9054e+00, -1.5415e+00],
[-2.2502e+00, -2.8277e+00, -2.5351e+00, ..., -2.7513e+00,
-2.1466e+00, -1.6281e+00],
...,
[-2.3760e+00, -2.9407e+00, -2.7371e+00, ..., -2.8632e+00,
-2.2905e+00, -1.7578e+00],
[-1.5340e+00, -1.9976e+00, -1.9392e+00, ..., -1.9375e+00,
-1.6014e+00, -1.2929e+00],
[-9.9501e-01, -1.2899e+00, -1.3255e+00, ..., -1.1803e+00,
-1.0107e+00, -7.6804e-01]]],
[[[ 3.7207e-01, 8.5100e-01, 7.6016e-01, ..., 6.9693e-01,
9.6512e-01, 7.4647e-01],
[-1.2284e+00, -2.5277e-01, -1.3674e+00, ..., -1.0781e+00,
7.8482e-02, 5.7903e-01],
[ 6.6482e-01, 4.6484e-01, 5.3941e-01, ..., 5.5617e-01,
9.3124e-01, 8.7087e-01],
...,
[-1.8205e-01, -1.1355e+00, -2.6262e+00, ..., -1.5532e+00,
-1.4047e+00, -8.6865e-01],
[ 1.4132e+00, 8.3456e-01, -9.6563e-01, ..., -3.4133e-01,
1.2300e-01, 3.6692e-01],
[ 8.2921e-01, 2.6586e-01, -7.2243e-01, ..., -2.4691e-01,
2.2918e-01, 3.7287e-01]],
[[-2.3327e+00, -1.2117e+00, -4.2479e+00, ..., -5.2518e+00,
-4.5087e+00, -3.5325e+00],
[-1.1257e-01, 1.7535e+00, -2.5421e+00, ..., -4.1368e+00,
-4.0000e+00, -3.3007e+00],
[ 7.6699e-01, 1.5875e+00, -5.1414e+00, ..., -6.0181e+00,
-5.7443e+00, -5.3154e+00],
...,
[-1.3299e+00, 5.5876e-02, -6.7509e+00, ..., -7.6124e+00,
-6.9441e+00, -5.4121e+00],
[ 1.7465e+00, 2.7630e+00, -1.0403e+00, ..., -2.6259e+00,
-3.0090e+00, -2.3888e+00],
[ 1.1834e+00, 1.7756e+00, -2.2836e+00, ..., -3.2722e+00,
-3.6474e+00, -3.1356e+00]],
[[-6.8198e-01, -1.8561e+00, -1.7279e+00, ..., -9.2758e-01,
-6.5974e-01, -7.7100e-01],
[-1.3993e+00, -1.8693e+00, -1.2839e+00, ..., -7.2743e-01,
-3.9717e-01, -2.8207e-01],
[-1.0827e+00, -1.0024e+00, -4.6556e-01, ..., -3.3838e-01,
1.5945e-02, 3.6471e-02],
...,
[-1.3311e+00, -1.7293e+00, -7.5558e-01, ..., -8.7159e-01,
-7.5456e-01, -1.0054e+00],
[-1.6916e+00, -1.1308e+00, -8.1436e-01, ..., -5.6893e-01,
-3.0116e-01, -3.8082e-01],
[-1.8653e+00, -1.2818e+00, -8.9311e-01, ..., -9.7036e-01,
-8.3485e-01, -8.1764e-01]],
...,
[[-2.2739e+00, -1.2850e+00, -2.3226e+00, ..., -2.8722e+00,
-2.2882e+00, -1.5103e+00],
[-6.3418e-01, 1.1109e+00, -4.0309e-01, ..., -1.3201e+00,
-1.4000e+00, -1.2162e+00],
[-3.6055e-01, 2.1328e+00, 8.3321e-01, ..., -1.2757e-01,
2.2990e-01, 3.4363e-01],
...,
[-1.9096e+00, 1.1436e+00, -4.8713e-02, ..., -1.0553e+00,
-5.7054e-01, -3.6747e-02],
[-8.6376e-01, 1.8128e+00, 1.5858e+00, ..., 5.1813e-01,
3.9490e-01, 6.7236e-01],
[-2.2453e+00, -1.2173e+00, -1.0377e+00, ..., -1.8110e+00,
-9.6440e-01, -2.8500e-01]],
[[-1.3470e+00, -1.2454e+00, -2.7574e+00, ..., -1.1042e+00,
-6.0794e-01, -4.8747e-01],
[-6.2669e-01, -1.0043e-01, -3.5996e+00, ..., -1.7800e+00,
-1.5165e+00, -1.6934e+00],
[-2.0641e+00, -1.3107e+00, -5.9893e+00, ..., -3.7153e+00,
-3.1931e+00, -2.5460e+00],
...,
[-2.4760e+00, -1.6513e+00, -5.4655e+00, ..., -3.0791e+00,
-2.7241e+00, -2.1981e+00],
[-2.2186e+00, -1.4980e+00, -4.2417e+00, ..., -2.6667e+00,
-2.3052e+00, -1.6623e+00],
[-2.3771e+00, -1.7660e+00, -3.3125e+00, ..., -2.9416e+00,
-2.4154e+00, -2.1305e+00]],
[[-1.1661e+00, -1.7273e+00, -1.4997e+00, ..., -1.6329e+00,
-1.3527e+00, -1.0182e+00],
[-1.1781e+00, -1.6449e+00, -1.1623e+00, ..., -1.3692e+00,
-1.1065e+00, -8.7526e-01],
[-1.5347e+00, -1.8118e+00, -1.3407e+00, ..., -1.5601e+00,
-1.1868e+00, -8.6735e-01],
...,
[-3.1762e+00, -3.9716e+00, -3.7806e+00, ..., -3.9571e+00,
-3.1754e+00, -2.4517e+00],
[-2.1161e+00, -2.7339e+00, -2.7091e+00, ..., -2.7887e+00,
-2.3024e+00, -1.8756e+00],
[-1.5467e+00, -1.8756e+00, -1.9379e+00, ..., -1.8939e+00,
-1.5338e+00, -1.1758e+00]]],
[[[ 2.0922e-01, 4.7267e-01, 6.2171e-01, ..., -3.4895e-01,
-8.0679e-02, 1.9684e-02],
[-2.0420e+00, -1.3658e+00, -2.8708e+00, ..., -3.2158e+00,
-1.2828e+00, -4.3015e-01],
[-1.4850e-01, -9.3156e-01, 2.3843e-02, ..., -1.9786e-01,
-3.7968e-01, 1.5530e-01],
...,
[-9.5717e-01, -2.2019e+00, -2.0598e+00, ..., -1.9357e+00,
-1.8890e+00, -1.2753e+00],
[ 7.5811e-01, -2.0150e-01, -1.4490e-01, ..., 2.0132e-02,
3.0396e-01, 4.6874e-01],
[ 1.8449e-01, -6.0538e-01, -4.1719e-01, ..., 2.1363e-01,
4.8337e-01, 5.0765e-01]],
[[-1.6752e+00, -1.3467e+00, -4.2683e+00, ..., -6.8332e+00,
-6.0468e+00, -4.5280e+00],
[ 4.9664e-01, 1.5141e+00, -2.5422e+00, ..., -5.4489e+00,
-5.5397e+00, -4.3020e+00],
[ 1.2385e+00, 1.2297e+00, -5.0983e+00, ..., -7.8607e+00,
-7.8026e+00, -6.6168e+00],
...,
[ 2.9030e-01, 2.2096e-01, -4.1033e+00, ..., -7.1241e+00,
-6.7189e+00, -5.1858e+00],
[ 2.4144e+00, 2.3896e+00, 1.2748e-01, ..., -1.9771e+00,
-2.6116e+00, -2.0688e+00],
[ 1.4261e+00, 1.2615e+00, -1.5765e+00, ..., -2.9523e+00,
-3.5228e+00, -3.0478e+00]],
[[-1.3754e-01, -1.1833e+00, -1.4173e+00, ..., -1.4029e+00,
-1.0565e+00, -1.3408e+00],
[-5.1222e-01, -8.4017e-01, -8.8231e-01, ..., -9.7742e-01,
-6.1890e-01, -9.0012e-01],
[-3.3310e-01, -2.3347e-01, -4.3178e-01, ..., -1.0951e+00,
-6.0510e-01, -4.7939e-01],
...,
[-3.6357e-03, -4.7504e-01, -5.4531e-01, ..., -1.0501e+00,
-8.7511e-01, -1.2422e+00],
[-6.6520e-01, -1.4965e-01, -5.4169e-01, ..., -5.6888e-01,
-2.4614e-01, -4.2963e-01],
[-8.1760e-01, -2.2788e-01, -4.6653e-01, ..., -5.1320e-01,
-4.3571e-01, -5.9526e-01]],
...,
[[-2.8717e+00, -2.4571e+00, -2.4543e+00, ..., -3.0577e+00,
-2.4887e+00, -1.8207e+00],
[-2.1342e+00, -8.1234e-01, -1.1026e+00, ..., -2.4216e+00,
-2.4754e+00, -2.0063e+00],
[-1.1322e+00, 8.5302e-01, 8.0003e-01, ..., -1.0525e-01,
4.0819e-02, 2.9088e-01],
...,
[-2.2859e+00, -3.0277e-01, -1.8841e-01, ..., -8.7211e-01,
-2.4519e-01, 3.0482e-01],
[-1.3124e+00, 7.1977e-01, 1.0155e+00, ..., 4.5779e-01,
4.7024e-01, 8.1910e-01],
[-2.2480e+00, -1.6077e+00, -1.7534e+00, ..., -2.1639e+00,
-1.1379e+00, -3.4305e-01]],
[[-1.6471e+00, -1.6301e+00, -2.8073e+00, ..., -2.2219e+00,
-1.9115e+00, -1.8090e+00],
[-7.0981e-01, -9.2325e-02, -3.1781e+00, ..., -2.8610e+00,
-2.8767e+00, -2.8912e+00],
[-2.0825e+00, -9.7508e-01, -5.1399e+00, ..., -4.4599e+00,
-4.5158e+00, -3.7206e+00],
...,
[-1.3570e+00, 1.8448e-01, -3.2334e+00, ..., -2.8540e+00,
-2.7590e+00, -2.3428e+00],
[-1.6837e+00, -4.9601e-01, -2.9099e+00, ..., -2.4433e+00,
-2.2579e+00, -1.6819e+00],
[-2.3277e+00, -1.3806e+00, -2.9782e+00, ..., -2.8349e+00,
-2.4172e+00, -2.2049e+00]],
[[-1.3086e+00, -1.8412e+00, -1.6526e+00, ..., -2.5704e+00,
-2.1457e+00, -1.6673e+00],
[-1.2514e+00, -1.7211e+00, -1.2314e+00, ..., -2.4020e+00,
-1.9366e+00, -1.5820e+00],
[-1.5707e+00, -1.9182e+00, -1.4948e+00, ..., -2.8115e+00,
-2.2501e+00, -1.7455e+00],
...,
[-2.4400e+00, -3.0460e+00, -2.8160e+00, ..., -4.0101e+00,
-3.2489e+00, -2.5550e+00],
[-1.5622e+00, -2.0706e+00, -1.9953e+00, ..., -2.7854e+00,
-2.3360e+00, -1.9304e+00],
[-1.0081e+00, -1.3403e+00, -1.3643e+00, ..., -1.8530e+00,
-1.5404e+00, -1.1938e+00]]],
[[[ 8.7893e-02, 3.0147e-01, -1.2428e+00, ..., 4.7466e-01,
1.0122e+00, 6.6956e-01],
[-2.0762e+00, -1.4664e+00, -4.2367e+00, ..., -2.5418e+00,
-7.3230e-01, -3.5719e-02],
[ 2.7235e-01, -3.7386e-01, -1.0919e+00, ..., -2.4739e-01,
4.8365e-01, 6.4188e-01],
...,
[-2.9549e-01, -1.3886e+00, -2.7438e+00, ..., -1.9610e+00,
-1.3477e+00, -8.7857e-01],
[ 1.3513e+00, 6.9212e-01, -8.9737e-01, ..., -3.2624e-01,
5.3316e-01, 7.1384e-01],
[ 8.8590e-01, 3.4809e-01, -3.9865e-01, ..., -3.5100e-01,
3.1466e-01, 2.9428e-01]],
[[-3.1356e+00, -1.4955e+00, -6.0802e+00, ..., -4.8568e+00,
-4.4233e+00, -3.3771e+00],
[-7.0454e-01, 1.7750e+00, -4.4111e+00, ..., -3.4029e+00,
-3.5373e+00, -2.9125e+00],
[-2.5897e-01, 1.4155e+00, -7.5101e+00, ..., -5.3591e+00,
-5.2496e+00, -4.9407e+00],
...,
[-1.1004e+00, 5.5892e-01, -6.4442e+00, ..., -4.5839e+00,
-4.1451e+00, -3.4472e+00],
[ 1.8892e+00, 3.0733e+00, -8.4836e-01, ..., -9.1210e-01,
-1.2460e+00, -1.2667e+00],
[ 1.2370e+00, 1.9358e+00, -2.1900e+00, ..., -2.2656e+00,
-2.4907e+00, -2.3722e+00]],
[[-1.3048e+00, -2.1454e+00, -2.0000e+00, ..., -5.9101e-01,
-2.0206e-01, -4.7472e-01],
[-1.8590e+00, -1.9830e+00, -1.4279e+00, ..., -3.5707e-01,
5.8179e-02, 3.3306e-02],
[-2.4184e+00, -1.8975e+00, -7.9671e-01, ..., -2.9270e-01,
2.4280e-01, 2.7877e-01],
...,
[-1.7042e+00, -1.9860e+00, -8.1252e-01, ..., -4.7438e-01,
-1.6392e-01, -3.3996e-01],
[-1.9006e+00, -1.2900e+00, -8.1291e-01, ..., -3.1416e-01,
1.4905e-01, 2.0633e-01],
[-1.9006e+00, -1.2290e+00, -6.4190e-01, ..., -5.0618e-01,
-2.7578e-01, -2.4957e-01]],
...,
[[-2.1076e+00, -1.2308e+00, -2.6176e+00, ..., -3.6770e+00,
-2.8616e+00, -1.8545e+00],
[-1.8338e+00, 3.4063e-03, -1.4026e+00, ..., -2.6466e+00,
-2.4401e+00, -1.8828e+00],
[-4.6108e-01, 2.2210e+00, 7.7075e-01, ..., -5.4482e-01,
-1.0074e-01, 1.4614e-01],
...,
[-1.9658e+00, 1.2516e+00, 1.7192e-01, ..., -1.5106e+00,
-7.1248e-01, -1.6273e-01],
[-9.4203e-01, 1.7937e+00, 1.5840e+00, ..., -1.2172e-01,
8.6215e-02, 3.1957e-01],
[-2.2420e+00, -1.1623e+00, -9.9729e-01, ..., -2.2898e+00,
-1.4321e+00, -8.9675e-01]],
[[-2.3723e+00, -2.6956e+00, -3.9324e+00, ..., -1.1143e+00,
-7.2927e-01, -7.5410e-01],
[-1.7589e+00, -1.5512e+00, -4.6666e+00, ..., -1.3253e+00,
-1.2115e+00, -1.6040e+00],
[-2.9803e+00, -2.5434e+00, -6.6137e+00, ..., -3.2505e+00,
-2.7934e+00, -2.2593e+00],
...,
[-2.4314e+00, -1.5647e+00, -5.1075e+00, ..., -1.4880e+00,
-1.0709e+00, -9.1709e-01],
[-2.1691e+00, -1.4345e+00, -4.0216e+00, ..., -1.7305e+00,
-1.1677e+00, -7.9123e-01],
[-2.3443e+00, -1.7574e+00, -3.1860e+00, ..., -2.7723e+00,
-1.9979e+00, -1.8324e+00]],
[[-1.8628e+00, -2.5714e+00, -2.4047e+00, ..., -1.8067e+00,
-1.4970e+00, -1.1379e+00],
[-1.9138e+00, -2.6184e+00, -2.1654e+00, ..., -1.4052e+00,
-1.1124e+00, -8.8695e-01],
[-2.4410e+00, -3.0175e+00, -2.6112e+00, ..., -1.6177e+00,
-1.2054e+00, -8.8866e-01],
...,
[-3.2421e+00, -4.0312e+00, -3.7481e+00, ..., -2.8597e+00,
-2.2837e+00, -1.7568e+00],
[-2.1380e+00, -2.7576e+00, -2.6662e+00, ..., -1.9782e+00,
-1.6213e+00, -1.3080e+00],
[-1.5247e+00, -1.8441e+00, -1.8591e+00, ..., -1.2343e+00,
-1.0358e+00, -7.9571e-01]]],
[[[ 2.7957e-01, 5.4187e-01, 6.8751e-01, ..., 4.7713e-01,
1.0080e+00, 6.0958e-01],
[-2.1950e+00, -1.5327e+00, -3.2148e+00, ..., -2.6419e+00,
-8.6084e-01, -1.7227e-01],
[-1.7642e-01, -1.0005e+00, -2.4047e-01, ..., -9.9210e-02,
5.5397e-01, 6.7311e-01],
...,
[ 1.0306e-01, -6.4919e-01, -1.5539e+00, ..., -1.0161e+00,
-7.7887e-01, -5.6920e-01],
[ 1.5465e+00, 1.2896e+00, 6.5717e-01, ..., 7.5583e-01,
9.1403e-01, 9.4358e-01],
[ 1.1487e+00, 1.0227e+00, 7.2949e-01, ..., 9.1279e-01,
1.0198e+00, 8.7520e-01]],
[[-1.6772e+00, -1.1648e+00, -4.1637e+00, ..., -4.8748e+00,
-4.4513e+00, -3.4203e+00],
[ 5.9904e-01, 1.9397e+00, -2.2353e+00, ..., -3.5516e+00,
-3.6465e+00, -3.0077e+00],
[ 1.3377e+00, 1.7120e+00, -4.8986e+00, ..., -5.5245e+00,
-5.3254e+00, -4.9677e+00],
...,
[-6.1367e-01, 5.6646e-01, -4.7666e+00, ..., -5.8634e+00,
-5.2008e+00, -4.2218e+00],
[ 2.1314e+00, 2.8177e+00, 4.8683e-02, ..., -1.4815e+00,
-1.6734e+00, -1.6634e+00],
[ 1.2683e+00, 1.5250e+00, -1.7244e+00, ..., -2.6963e+00,
-2.8391e+00, -2.7319e+00]],
[[-1.1317e-01, -1.1410e+00, -1.3049e+00, ..., -4.9987e-01,
-1.7144e-01, -4.8519e-01],
[-5.9035e-01, -8.4765e-01, -7.6421e-01, ..., -2.9330e-01,
2.4889e-02, -7.8600e-02],
[-5.3737e-01, -3.4047e-01, -3.8792e-01, ..., -3.7159e-01,
6.6430e-02, 7.7219e-02],
...,
[-1.1246e+00, -1.5716e+00, -6.5224e-01, ..., -6.1715e-01,
-4.6649e-01, -6.2464e-01],
[-1.3534e+00, -8.8443e-01, -4.6813e-01, ..., -9.6118e-02,
1.0303e-01, 7.9295e-02],
[-1.3881e+00, -6.8109e-01, -1.2436e-01, ..., -5.6851e-02,
-3.2872e-02, -6.1002e-02]],
...,
[[-3.0632e+00, -2.5507e+00, -2.8054e+00, ..., -3.4885e+00,
-2.6330e+00, -1.7439e+00],
[-2.3071e+00, -8.7522e-01, -1.5936e+00, ..., -2.6500e+00,
-2.3327e+00, -1.8191e+00],
[-1.0989e+00, 1.1355e+00, 6.2051e-01, ..., -4.6326e-01,
2.4351e-02, 1.5143e-01],
...,
[-1.5182e+00, 1.6850e+00, 3.6917e-01, ..., -8.5615e-01,
-1.6712e-01, 3.5836e-01],
[-5.9594e-01, 1.7933e+00, 1.1110e+00, ..., 1.4130e-01,
3.0851e-01, 4.8422e-01],
[-1.6657e+00, -9.4736e-01, -1.3740e+00, ..., -2.1107e+00,
-1.1842e+00, -5.3887e-01]],
[[-1.6010e+00, -1.5882e+00, -2.9012e+00, ..., -1.2946e+00,
-8.6228e-01, -8.5852e-01],
[-5.9921e-01, 7.1281e-02, -3.2287e+00, ..., -1.4247e+00,
-1.2594e+00, -1.5788e+00],
[-2.1036e+00, -9.1805e-01, -5.3403e+00, ..., -3.2450e+00,
-2.7496e+00, -2.2046e+00],
...,
[-1.7819e+00, -7.1987e-01, -4.7792e+00, ..., -2.4195e+00,
-2.0270e+00, -1.5898e+00],
[-1.6589e+00, -7.9361e-01, -3.8085e+00, ..., -2.2537e+00,
-1.6756e+00, -1.0970e+00],
[-2.0319e+00, -1.3164e+00, -3.0309e+00, ..., -2.7610e+00,
-1.9625e+00, -1.6898e+00]],
[[-1.3558e+00, -1.9163e+00, -1.7212e+00, ..., -1.7746e+00,
-1.4610e+00, -1.1114e+00],
[-1.2786e+00, -1.7648e+00, -1.2436e+00, ..., -1.3840e+00,
-1.0851e+00, -8.6099e-01],
[-1.5973e+00, -1.9508e+00, -1.4906e+00, ..., -1.6148e+00,
-1.2069e+00, -8.8525e-01],
...,
[-2.4938e+00, -3.0969e+00, -2.8225e+00, ..., -2.9844e+00,
-2.3816e+00, -1.8183e+00],
[-1.6915e+00, -2.1349e+00, -2.0259e+00, ..., -2.0832e+00,
-1.7024e+00, -1.3611e+00],
[-1.1360e+00, -1.3583e+00, -1.3846e+00, ..., -1.2787e+00,
-1.0851e+00, -8.0170e-01]]]], device='cuda:0'),
tensor([[[[ 1.1167e-01, 3.4763e-01, -1.1874e+00, ..., 6.5205e-01,
9.1963e-01, 6.7299e-01],
[-2.0088e+00, -1.3622e+00, -4.1651e+00, ..., -1.4758e+00,
-1.8459e-01, 3.8973e-01],
[ 3.5362e-01, -2.8767e-01, -1.1213e+00, ..., 4.9823e-01,
9.2265e-01, 8.4845e-01],
...,
[-3.5202e-01, -1.3468e+00, -2.8124e+00, ..., -1.1705e+00,
-8.2265e-01, -6.6050e-01],
[ 1.3267e+00, 7.4245e-01, -9.6545e-01, ..., 5.1486e-01,
7.5049e-01, 7.8153e-01],
[ 8.8974e-01, 3.8460e-01, -4.8331e-01, ..., 5.4122e-01,
7.7799e-01, 6.8325e-01]],
[[-3.1664e+00, -1.5185e+00, -6.0895e+00, ..., -5.4110e+00,
-4.6400e+00, -3.6298e+00],
[-7.2508e-01, 1.7778e+00, -4.3931e+00, ..., -4.1936e+00,
-4.0684e+00, -3.3576e+00],
[-2.4436e-01, 1.4312e+00, -7.4670e+00, ..., -6.1186e+00,
-5.8614e+00, -5.4320e+00],
...,
[-1.0991e+00, 6.3583e-01, -6.2692e+00, ..., -5.3832e+00,
-4.7056e+00, -3.8988e+00],
[ 1.9128e+00, 3.1519e+00, -7.5226e-01, ..., -1.2761e+00,
-1.4035e+00, -1.4602e+00],
[ 1.2609e+00, 1.9895e+00, -2.1542e+00, ..., -2.4886e+00,
-2.5846e+00, -2.5381e+00]],
[[-1.2974e+00, -2.1679e+00, -2.0114e+00, ..., -8.1059e-01,
-6.0594e-01, -8.0159e-01],
[-1.8476e+00, -2.0025e+00, -1.4377e+00, ..., -6.7722e-01,
-4.2025e-01, -3.6483e-01],
[-2.3941e+00, -1.9014e+00, -8.1053e-01, ..., -4.0099e-01,
-1.2377e-01, -1.1426e-01],
...,
[-1.7122e+00, -1.9741e+00, -8.3966e-01, ..., -6.2608e-01,
-5.4507e-01, -7.0688e-01],
[-1.9672e+00, -1.3003e+00, -8.3791e-01, ..., -2.2090e-01,
-5.1165e-02, -5.2399e-02],
[-1.9563e+00, -1.2339e+00, -6.5305e-01, ..., -3.1484e-01,
-2.8013e-01, -2.6508e-01]],
...,
[[-2.1551e+00, -1.2287e+00, -2.5491e+00, ..., -3.0199e+00,
-2.4246e+00, -1.6130e+00],
[-1.8394e+00, 4.3236e-02, -1.3191e+00, ..., -1.4920e+00,
-1.5730e+00, -1.3699e+00],
[-5.0444e-01, 2.2220e+00, 8.2790e-01, ..., -5.9420e-02,
2.5233e-01, 3.1457e-01],
...,
[-1.8969e+00, 1.2677e+00, 2.5844e-01, ..., -9.9331e-01,
-2.9288e-01, 1.7323e-01],
[-8.8527e-01, 1.7658e+00, 1.5916e+00, ..., 1.6095e-01,
3.0359e-01, 3.9827e-01],
[-2.1669e+00, -1.1690e+00, -9.8281e-01, ..., -2.1754e+00,
-1.2591e+00, -6.5052e-01]],
[[-2.3695e+00, -2.6701e+00, -3.9228e+00, ..., -1.1522e+00,
-6.4914e-01, -5.3314e-01],
[-1.7532e+00, -1.5400e+00, -4.6651e+00, ..., -1.6982e+00,
-1.4827e+00, -1.7204e+00],
[-3.0018e+00, -2.5659e+00, -6.6266e+00, ..., -3.6964e+00,
-3.2372e+00, -2.6035e+00],
...,
[-2.3641e+00, -1.4429e+00, -5.1773e+00, ..., -2.1731e+00,
-1.7172e+00, -1.3795e+00],
[-2.0937e+00, -1.3186e+00, -4.0084e+00, ..., -2.1235e+00,
-1.4730e+00, -9.5493e-01],
[-2.2979e+00, -1.6746e+00, -3.1286e+00, ..., -2.8180e+00,
-1.9417e+00, -1.6649e+00]],
[[-1.8498e+00, -2.5453e+00, -2.3766e+00, ..., -1.6826e+00,
-1.3964e+00, -1.0427e+00],
[-1.8906e+00, -2.5828e+00, -2.1298e+00, ..., -1.3701e+00,
-1.1139e+00, -8.8678e-01],
[-2.4113e+00, -2.9759e+00, -2.5617e+00, ..., -1.5901e+00,
-1.2148e+00, -8.8682e-01],
...,
[-3.2580e+00, -4.0546e+00, -3.7527e+00, ..., -2.7788e+00,
-2.2201e+00, -1.6796e+00],
[-2.1721e+00, -2.7859e+00, -2.7004e+00, ..., -1.9497e+00,
-1.5986e+00, -1.2751e+00],
[-1.5407e+00, -1.8710e+00, -1.8850e+00, ..., -1.2115e+00,
-1.0347e+00, -7.6287e-01]]],
[[[ 1.4297e-01, 3.7279e-01, -1.1785e+00, ..., 4.5141e-01,
1.0169e+00, 6.4922e-01],
[-2.0587e+00, -1.4125e+00, -4.3345e+00, ..., -2.5957e+00,
-7.4429e-01, -5.8592e-02],
[ 4.2795e-01, -2.8258e-01, -1.1310e+00, ..., -2.6508e-01,
5.1498e-01, 6.4113e-01],
...,
[-2.6079e-01, -1.2890e+00, -2.6124e+00, ..., -2.0259e+00,
-1.3876e+00, -9.1282e-01],
[ 1.2709e+00, 6.5247e-01, -9.1439e-01, ..., -3.2735e-01,
5.3577e-01, 7.2309e-01],
[ 8.3058e-01, 2.9905e-01, -4.2474e-01, ..., -3.6574e-01,
2.9772e-01, 2.7459e-01]],
[[-3.2198e+00, -1.4244e+00, -6.1021e+00, ..., -4.8495e+00,
-4.4496e+00, -3.3647e+00],
[-5.8193e-01, 2.1324e+00, -4.1746e+00, ..., -3.3970e+00,
-3.5665e+00, -2.8989e+00],
[ 1.2374e-02, 1.8254e+00, -7.3774e+00, ..., -5.3908e+00,
-5.2894e+00, -4.9300e+00],
...,
[-1.2354e+00, 3.0515e-01, -6.5336e+00, ..., -4.5620e+00,
-4.1327e+00, -3.4402e+00],
[ 1.5508e+00, 2.6417e+00, -1.3418e+00, ..., -8.6757e-01,
-1.2054e+00, -1.2345e+00],
[ 1.0231e+00, 1.6566e+00, -2.4109e+00, ..., -2.2388e+00,
-2.4711e+00, -2.3489e+00]],
[[-1.3663e+00, -2.3182e+00, -2.1447e+00, ..., -6.0383e-01,
-1.9987e-01, -4.6958e-01],
[-2.0106e+00, -2.1860e+00, -1.5418e+00, ..., -3.5650e-01,
6.4460e-02, 4.1518e-02],
[-2.5392e+00, -2.0157e+00, -8.4690e-01, ..., -2.9103e-01,
2.4052e-01, 2.8256e-01],
...,
[-1.6132e+00, -1.8555e+00, -7.9234e-01, ..., -4.9281e-01,
-1.5897e-01, -3.4773e-01],
[-1.7838e+00, -1.2180e+00, -7.8813e-01, ..., -3.1107e-01,
1.6528e-01, 2.2238e-01],
[-1.7411e+00, -1.1298e+00, -6.0423e-01, ..., -5.0775e-01,
-2.6940e-01, -2.4169e-01]],
...,
[[-2.3227e+00, -1.2501e+00, -2.6959e+00, ..., -3.7038e+00,
-2.8741e+00, -1.8459e+00],
[-1.8849e+00, 2.0808e-01, -1.3300e+00, ..., -2.6657e+00,
-2.4482e+00, -1.8800e+00],
[-4.5661e-01, 2.5406e+00, 9.3587e-01, ..., -5.5267e-01,
-7.3966e-02, 1.5547e-01],
...,
[-1.7360e+00, 1.2088e+00, 2.3346e-01, ..., -1.5107e+00,
-6.7792e-01, -1.3230e-01],
[-8.5701e-01, 1.7028e+00, 1.5242e+00, ..., -1.3354e-01,
1.1665e-01, 3.4099e-01],
[-2.0892e+00, -1.0217e+00, -8.2088e-01, ..., -2.3611e+00,
-1.4741e+00, -9.2054e-01]],
[[-2.3774e+00, -2.6420e+00, -4.0577e+00, ..., -1.1255e+00,
-7.1571e-01, -7.4940e-01],
[-1.6794e+00, -1.3519e+00, -4.7869e+00, ..., -1.3579e+00,
-1.2165e+00, -1.5895e+00],
[-3.0082e+00, -2.4205e+00, -6.9070e+00, ..., -3.2819e+00,
-2.8086e+00, -2.2428e+00],
...,
[-2.5199e+00, -1.7230e+00, -4.9979e+00, ..., -1.4662e+00,
-1.0462e+00, -8.9612e-01],
[-2.2149e+00, -1.5441e+00, -3.8914e+00, ..., -1.7176e+00,
-1.1495e+00, -7.6137e-01],
[-2.3043e+00, -1.7665e+00, -3.0791e+00, ..., -2.7962e+00,
-2.0015e+00, -1.8245e+00]],
[[-1.8893e+00, -2.6163e+00, -2.3963e+00, ..., -1.8040e+00,
-1.4947e+00, -1.1416e+00],
[-1.9186e+00, -2.6335e+00, -2.0964e+00, ..., -1.3951e+00,
-1.1059e+00, -8.8520e-01],
[-2.4542e+00, -3.0150e+00, -2.5227e+00, ..., -1.5950e+00,
-1.2013e+00, -8.8677e-01],
...,
[-3.1145e+00, -3.8910e+00, -3.6534e+00, ..., -2.8715e+00,
-2.2928e+00, -1.7692e+00],
[-2.0700e+00, -2.6771e+00, -2.6319e+00, ..., -1.9831e+00,
-1.6310e+00, -1.3193e+00],
[-1.4886e+00, -1.8088e+00, -1.8536e+00, ..., -1.2314e+00,
-1.0420e+00, -8.0525e-01]]],
[[[ 3.9898e-02, 2.0580e-01, -1.4601e+00, ..., -6.7247e-01,
-2.3473e-01, -8.3191e-02],
[-2.1164e+00, -1.6338e+00, -4.5729e+00, ..., -3.8393e+00,
-1.5671e+00, -5.7332e-01],
[ 5.2852e-01, -9.2794e-02, -8.7235e-01, ..., -1.8599e-01,
-2.5924e-01, 2.6751e-01],
...,
[-3.0420e-01, -1.1067e+00, -1.4562e+00, ..., -1.3158e+00,
-9.1453e-01, -6.9464e-01],
[ 9.8023e-01, 5.7962e-01, 3.1865e-01, ..., 4.1855e-01,
6.9089e-01, 7.8233e-01],
[ 6.8639e-01, 4.5842e-01, 3.3305e-01, ..., 5.5094e-01,
7.7879e-01, 7.0176e-01]],
[[-3.3334e+00, -1.8349e+00, -6.2892e+00, ..., -7.1722e+00,
-6.2795e+00, -4.6460e+00],
[-8.6188e-01, 1.4787e+00, -4.3927e+00, ..., -5.4765e+00,
-5.5573e+00, -4.2975e+00],
[-4.4395e-01, 1.0448e+00, -7.4254e+00, ..., -8.0095e+00,
-7.8734e+00, -6.7478e+00],
...,
[-4.7130e-01, 2.4781e-01, -4.4313e+00, ..., -5.3785e+00,
-4.6660e+00, -3.8841e+00],
[ 1.9293e+00, 2.4084e+00, -9.1934e-02, ..., -1.0870e+00,
-1.2264e+00, -1.3466e+00],
[ 1.0996e+00, 1.2967e+00, -1.7032e+00, ..., -2.4565e+00,
-2.5781e+00, -2.5518e+00]],
[[-1.3223e+00, -2.3141e+00, -2.2851e+00, ..., -1.7693e+00,
-1.2628e+00, -1.4096e+00],
[-1.6493e+00, -1.9660e+00, -1.5294e+00, ..., -1.2401e+00,
-7.3280e-01, -8.4688e-01],
[-2.1412e+00, -1.9421e+00, -1.0353e+00, ..., -1.0995e+00,
-4.6552e-01, -1.9173e-01],
...,
[-6.7324e-01, -1.1175e+00, -5.8246e-01, ..., -6.1359e-01,
-4.8390e-01, -6.3981e-01],
[-1.0468e+00, -6.9262e-01, -5.5114e-01, ..., -2.7646e-01,
-6.0448e-02, -3.2661e-02],
[-1.1770e+00, -6.5252e-01, -3.6936e-01, ..., -3.3049e-01,
-2.7829e-01, -2.4021e-01]],
...,
[[-1.9774e+00, -1.0917e+00, -2.3491e+00, ..., -3.1877e+00,
-2.6153e+00, -1.8026e+00],
[-1.7620e+00, 1.9583e-03, -1.3772e+00, ..., -2.5275e+00,
-2.5272e+00, -1.9369e+00],
[-3.8798e-01, 2.3430e+00, 1.0611e+00, ..., 9.2775e-02,
3.0407e-01, 5.6672e-01],
...,
[-1.5235e+00, 9.8315e-01, -1.2565e-01, ..., -1.2256e+00,
-4.5844e-01, 5.7066e-02],
[-5.5063e-01, 1.3865e+00, 7.5120e-01, ..., -5.0266e-03,
1.8127e-01, 2.9886e-01],
[-1.6052e+00, -1.0825e+00, -1.5389e+00, ..., -2.3265e+00,
-1.3469e+00, -7.2171e-01]],
[[-2.4349e+00, -2.8313e+00, -4.1823e+00, ..., -2.4013e+00,
-1.9060e+00, -1.7284e+00],
[-1.7942e+00, -1.6619e+00, -4.8683e+00, ..., -2.9397e+00,
-2.7850e+00, -2.7937e+00],
[-2.8427e+00, -2.4882e+00, -6.6533e+00, ..., -4.4854e+00,
-4.4030e+00, -3.5252e+00],
...,
[-1.7173e+00, -5.5098e-01, -4.0507e+00, ..., -2.0932e+00,
-1.5435e+00, -1.1908e+00],
[-1.6903e+00, -7.2212e-01, -3.3521e+00, ..., -2.0751e+00,
-1.3390e+00, -7.8165e-01],
[-2.0533e+00, -1.2690e+00, -2.8397e+00, ..., -2.7169e+00,
-1.7884e+00, -1.5259e+00]],
[[-1.8037e+00, -2.5424e+00, -2.4655e+00, ..., -2.7310e+00,
-2.2402e+00, -1.7164e+00],
[-1.8034e+00, -2.5151e+00, -2.1624e+00, ..., -2.4401e+00,
-1.9311e+00, -1.5686e+00],
[-2.2540e+00, -2.8248e+00, -2.5281e+00, ..., -2.7897e+00,
-2.1785e+00, -1.6602e+00],
...,
[-2.2751e+00, -2.8588e+00, -2.6678e+00, ..., -2.8691e+00,
-2.2878e+00, -1.7282e+00],
[-1.5360e+00, -1.9789e+00, -1.9363e+00, ..., -2.0242e+00,
-1.6641e+00, -1.3265e+00],
[-1.0376e+00, -1.2684e+00, -1.3439e+00, ..., -1.2560e+00,
-1.0736e+00, -7.9002e-01]]],
[[[ 1.4056e-01, 3.4763e-01, -1.2951e+00, ..., 6.6436e-01,
9.3086e-01, 6.9650e-01],
[-2.1324e+00, -1.5215e+00, -4.5579e+00, ..., -1.3215e+00,
-6.2169e-02, 4.6003e-01],
[ 4.4071e-01, -3.1075e-01, -1.2005e+00, ..., 5.4428e-01,
9.5105e-01, 8.5711e-01],
...,
[-2.9457e-01, -1.2610e+00, -2.6178e+00, ..., -4.8524e+00,
-3.9865e+00, -1.2749e+00],
[ 1.2243e+00, 6.4603e-01, -9.8932e-01, ..., -6.9337e-01,
3.1878e-01, 9.7791e-01],
[ 7.9939e-01, 2.7937e-01, -4.7215e-01, ..., 5.9328e-01,
8.2819e-01, 7.3300e-01]],
[[-3.2960e+00, -1.5090e+00, -6.2586e+00, ..., -5.3456e+00,
-4.5884e+00, -3.5986e+00],
[-6.0739e-01, 2.1216e+00, -4.2790e+00, ..., -4.2232e+00,
-4.0966e+00, -3.3799e+00],
[-2.1852e-02, 1.8117e+00, -7.4945e+00, ..., -6.1282e+00,
-5.8777e+00, -5.4345e+00],
...,
[-1.3555e+00, 1.9209e-01, -6.6117e+00, ..., -5.2755e+00,
-5.3518e+00, -4.3692e+00],
[ 1.4591e+00, 2.5729e+00, -1.3675e+00, ..., -1.1981e+00,
-1.7548e+00, -1.5409e+00],
[ 9.6872e-01, 1.6426e+00, -2.4032e+00, ..., -2.3877e+00,
-2.5843e+00, -2.5816e+00]],
[[-1.4503e+00, -2.4060e+00, -2.2159e+00, ..., -7.8049e-01,
-6.0268e-01, -7.8174e-01],
[-2.0675e+00, -2.2361e+00, -1.5963e+00, ..., -6.5704e-01,
-4.2665e-01, -3.6942e-01],
[-2.6032e+00, -2.0630e+00, -8.6317e-01, ..., -3.9235e-01,
-1.2717e-01, -1.1936e-01],
...,
[-1.5667e+00, -1.8107e+00, -7.5822e-01, ..., -8.5641e-01,
-5.2496e-01, -4.7738e-01],
[-1.7589e+00, -1.1902e+00, -7.8498e-01, ..., -4.2817e-01,
-1.8089e-01, 1.2737e-02],
[-1.7621e+00, -1.1742e+00, -6.5528e-01, ..., -3.2660e-01,
-2.8818e-01, -2.5669e-01]],
...,
[[-2.3727e+00, -1.3227e+00, -2.7693e+00, ..., -2.9032e+00,
-2.3348e+00, -1.5880e+00],
[-1.9472e+00, 1.4148e-01, -1.3813e+00, ..., -1.4002e+00,
-1.4853e+00, -1.2861e+00],
[-5.5225e-01, 2.4713e+00, 9.4525e-01, ..., -8.8400e-02,
1.9614e-01, 2.9165e-01],
...,
[-1.8225e+00, 1.0358e+00, 1.4319e-01, ..., -1.0617e+00,
1.0766e+00, 7.8096e-01],
[-9.1510e-01, 1.5876e+00, 1.4537e+00, ..., 2.8092e-01,
8.0505e-01, 6.5298e-01],
[-2.1462e+00, -1.0771e+00, -8.3800e-01, ..., -2.3633e+00,
-1.3477e+00, -6.9108e-01]],
[[-2.4232e+00, -2.7035e+00, -4.0527e+00, ..., -1.1821e+00,
-6.7425e-01, -5.5884e-01],
[-1.7202e+00, -1.4147e+00, -4.7787e+00, ..., -1.7668e+00,
-1.5396e+00, -1.7629e+00],
[-3.0682e+00, -2.5191e+00, -6.9129e+00, ..., -3.7168e+00,
-3.2471e+00, -2.6452e+00],
...,
[-2.5611e+00, -1.7798e+00, -4.9801e+00, ..., -1.5412e+00,
-1.1182e+00, -1.0013e+00],
[-2.2435e+00, -1.5992e+00, -3.8951e+00, ..., -1.7963e+00,
-1.1468e+00, -7.7224e-01],
[-2.3125e+00, -1.8097e+00, -3.0608e+00, ..., -2.8796e+00,
-1.9657e+00, -1.6878e+00]],
[[-1.9296e+00, -2.6644e+00, -2.4570e+00, ..., -1.6737e+00,
-1.3966e+00, -1.0449e+00],
[-1.9512e+00, -2.6799e+00, -2.1406e+00, ..., -1.3952e+00,
-1.1414e+00, -9.0039e-01],
[-2.4769e+00, -3.0520e+00, -2.5650e+00, ..., -1.6205e+00,
-1.2422e+00, -9.1067e-01],
...,
[-3.1353e+00, -3.9264e+00, -3.7173e+00, ..., -3.9629e+00,
-3.0197e+00, -1.8841e+00],
[-2.0782e+00, -2.6965e+00, -2.6705e+00, ..., -2.5302e+00,
-2.0438e+00, -1.3992e+00],
[-1.4942e+00, -1.8305e+00, -1.8771e+00, ..., -1.2275e+00,
-1.0551e+00, -7.7513e-01]]],
[[[ 5.5748e-02, 2.0443e-01, -1.4657e+00, ..., -6.6863e-01,
-2.1992e-01, -7.1487e-02],
[-2.1371e+00, -1.6490e+00, -4.5993e+00, ..., -3.8736e+00,
-1.5766e+00, -5.9632e-01],
[ 5.4871e-01, -9.4193e-02, -8.8231e-01, ..., -2.4405e-01,
-3.0312e-01, 2.0729e-01],
...,
[-1.4743e+01, -1.8181e+01, -2.1269e+01, ..., -1.6980e+00,
-1.2374e+00, -8.7363e-01],
[-1.3659e+01, -1.8123e+01, -2.2077e+01, ..., 1.0106e-02,
7.2674e-01, 7.9628e-01],
[-1.0898e+01, -1.7345e+01, -2.2624e+01, ..., -1.4533e-01,
4.5144e-01, 3.6252e-01]],
[[-3.3164e+00, -1.7978e+00, -6.2911e+00, ..., -7.1899e+00,
-6.3115e+00, -4.6518e+00],
[-8.0471e-01, 1.5689e+00, -4.3718e+00, ..., -5.4945e+00,
-5.6024e+00, -4.3068e+00],
[-3.5646e-01, 1.1259e+00, -7.4639e+00, ..., -8.0286e+00,
-7.9022e+00, -6.7381e+00],
...,
[ 3.9446e-01, -1.4082e-01, -8.1404e+00, ..., -4.6713e+00,
-4.2468e+00, -3.5277e+00],
[ 2.2114e+00, 2.0663e+00, -4.4945e+00, ..., -9.5168e-01,
-1.3006e+00, -1.3022e+00],
[ 1.7609e+00, 1.2384e+00, -3.5188e+00, ..., -2.2916e+00,
-2.5453e+00, -2.4194e+00]],
[[-1.3663e+00, -2.3887e+00, -2.3440e+00, ..., -1.7796e+00,
-1.2568e+00, -1.4005e+00],
[-1.7136e+00, -2.0261e+00, -1.5520e+00, ..., -1.2800e+00,
-7.4926e-01, -8.3861e-01],
[-2.2145e+00, -1.9754e+00, -1.0250e+00, ..., -1.1316e+00,
-4.8150e-01, -1.9629e-01],
...,
[-7.7998e-01, -7.1547e-01, -2.9713e-01, ..., -5.0232e-01,
-2.0352e-01, -4.2801e-01],
[-8.5182e-01, -5.5300e-01, -2.1030e+00, ..., -2.1212e-01,
2.3565e-01, 2.3055e-01],
[-7.0385e-01, -1.7598e-01, -1.5818e+00, ..., -3.9739e-01,
-1.9874e-01, -2.2708e-01]],
...,
[[-2.0487e+00, -1.1217e+00, -2.3773e+00, ..., -3.2084e+00,
-2.6555e+00, -1.8409e+00],
[-1.8025e+00, -4.2173e-04, -1.4028e+00, ..., -2.5165e+00,
-2.5693e+00, -1.9952e+00],
[-3.7707e-01, 2.3832e+00, 1.0589e+00, ..., 1.5951e-01,
2.8474e-01, 5.2205e-01],
...,
[-3.2120e+00, 8.7255e-01, -1.3951e-01, ..., -1.3285e+00,
-4.0863e-01, 7.3538e-02],
[-9.1708e+00, -8.6219e+00, -7.0126e+00, ..., 2.8357e-02,
2.8492e-01, 4.3384e-01],
[-4.3203e+00, -1.0433e+01, -1.4558e+01, ..., -2.3482e+00,
-1.4052e+00, -9.0655e-01]],
[[-2.4393e+00, -2.8581e+00, -4.2613e+00, ..., -2.4030e+00,
-1.9274e+00, -1.7425e+00],
[-1.7814e+00, -1.6572e+00, -4.9402e+00, ..., -2.9214e+00,
-2.8027e+00, -2.7983e+00],
[-2.8386e+00, -2.4759e+00, -6.7227e+00, ..., -4.4556e+00,
-4.3954e+00, -3.5093e+00],
...,
[ 1.8236e-02, 1.1954e+00, -1.6901e+00, ..., -1.4855e+00,
-1.0842e+00, -9.7195e-01],
[ 1.5147e+00, 3.5080e+00, 6.4013e-01, ..., -1.7560e+00,
-1.1932e+00, -8.4277e-01],
[ 2.2413e-01, 2.5478e+00, 1.5988e+00, ..., -2.9375e+00,
-2.1266e+00, -1.9673e+00]],
[[-1.8255e+00, -2.5626e+00, -2.4771e+00, ..., -2.7307e+00,
-2.2433e+00, -1.7179e+00],
[-1.8232e+00, -2.5379e+00, -2.1687e+00, ..., -2.4342e+00,
-1.9254e+00, -1.5572e+00],
[-2.2815e+00, -2.8490e+00, -2.5427e+00, ..., -2.7789e+00,
-2.1732e+00, -1.6481e+00],
...,
[-3.0681e+00, -4.6151e+00, -4.8163e+00, ..., -2.8113e+00,
-2.2463e+00, -1.7286e+00],
[-2.5455e+00, -3.5759e+00, -3.5897e+00, ..., -1.9158e+00,
-1.5761e+00, -1.2695e+00],
[-1.8731e+00, -2.6323e+00, -2.7851e+00, ..., -1.1775e+00,
-1.0012e+00, -7.5859e-01]]]], device='cuda:0'),
tensor([[[[ 2.8462e-01, 5.6693e-01, 7.1128e-01, ..., 4.9336e-01,
1.0110e+00, 6.0995e-01],
[-2.1553e+00, -1.4639e+00, -3.0568e+00, ..., -2.5911e+00,
-8.2774e-01, -1.6246e-01],
[-1.6666e-01, -9.5392e-01, -1.7712e-01, ..., -8.8977e-02,
5.6546e-01, 6.5148e-01],
...,
[ 1.1289e-01, -7.1048e-01, -1.6605e+00, ..., -1.1039e+00,
-7.8071e-01, -5.5105e-01],
[ 1.5677e+00, 1.2549e+00, 5.6039e-01, ..., 7.2211e-01,
9.3370e-01, 9.4533e-01],
[ 1.1760e+00, 1.0165e+00, 6.7558e-01, ..., 8.8595e-01,
9.9989e-01, 8.4737e-01]],
[[-1.6838e+00, -1.1205e+00, -4.1129e+00, ..., -4.9158e+00,
-4.4574e+00, -3.4136e+00],
[ 5.5553e-01, 1.9366e+00, -2.2381e+00, ..., -3.5566e+00,
-3.6484e+00, -3.0013e+00],
[ 1.2905e+00, 1.6924e+00, -4.9242e+00, ..., -5.5054e+00,
-5.3220e+00, -4.9689e+00],
...,
[-6.6527e-01, 5.4626e-01, -4.8356e+00, ..., -5.9332e+00,
-5.2920e+00, -4.3140e+00],
[ 2.1344e+00, 2.8784e+00, 9.1582e-02, ..., -1.6046e+00,
-1.8049e+00, -1.7767e+00],
[ 1.2664e+00, 1.5731e+00, -1.7049e+00, ..., -2.7690e+00,
-2.9026e+00, -2.7869e+00]],
[[-1.3100e-01, -1.1494e+00, -1.3073e+00, ..., -5.1264e-01,
-1.8078e-01, -4.9094e-01],
[-6.3585e-01, -8.8750e-01, -7.8155e-01, ..., -3.1001e-01,
1.3254e-02, -7.9341e-02],
[-5.9203e-01, -3.8836e-01, -3.9629e-01, ..., -3.6054e-01,
6.5065e-02, 6.8152e-02],
...,
[-1.1391e+00, -1.5963e+00, -6.6729e-01, ..., -6.2985e-01,
-4.9234e-01, -6.4524e-01],
[-1.3569e+00, -8.9454e-01, -4.8603e-01, ..., -1.3694e-01,
5.3744e-02, 3.9418e-02],
[-1.3717e+00, -6.8061e-01, -1.4661e-01, ..., -8.9964e-02,
-6.1006e-02, -7.8785e-02]],
...,
[[-2.9936e+00, -2.4581e+00, -2.7962e+00, ..., -3.4684e+00,
-2.6407e+00, -1.7593e+00],
[-2.1736e+00, -7.4554e-01, -1.5771e+00, ..., -2.5939e+00,
-2.3299e+00, -1.8400e+00],
[-9.5793e-01, 1.2847e+00, 6.0743e-01, ..., -4.4473e-01,
-9.2417e-03, 1.2941e-01],
...,
[-1.5620e+00, 1.6997e+00, 4.0023e-01, ..., -7.7867e-01,
-1.6165e-01, 3.8569e-01],
[-6.2506e-01, 1.8078e+00, 1.1731e+00, ..., 2.1239e-01,
3.2093e-01, 5.1162e-01],
[-1.7403e+00, -1.0215e+00, -1.4053e+00, ..., -1.9967e+00,
-1.1233e+00, -5.0731e-01]],
[[-1.5570e+00, -1.5381e+00, -2.8791e+00, ..., -1.2425e+00,
-8.1229e-01, -8.0823e-01],
[-5.6557e-01, 1.0225e-01, -3.2294e+00, ..., -1.3756e+00,
-1.2219e+00, -1.5652e+00],
[-2.0558e+00, -8.8525e-01, -5.3414e+00, ..., -3.2193e+00,
-2.7331e+00, -2.2091e+00],
...,
[-1.8132e+00, -7.2417e-01, -4.7297e+00, ..., -2.4753e+00,
-2.1000e+00, -1.7075e+00],
[-1.7011e+00, -7.9317e-01, -3.7640e+00, ..., -2.2636e+00,
-1.6994e+00, -1.1650e+00],
[-2.0774e+00, -1.3286e+00, -3.0238e+00, ..., -2.7108e+00,
-1.9668e+00, -1.7073e+00]],
[[-1.3441e+00, -1.9028e+00, -1.6950e+00, ..., -1.7614e+00,
-1.4567e+00, -1.1024e+00],
[-1.2770e+00, -1.7652e+00, -1.2363e+00, ..., -1.3652e+00,
-1.0800e+00, -8.5606e-01],
[-1.6045e+00, -1.9566e+00, -1.4927e+00, ..., -1.5920e+00,
-1.1923e+00, -8.7629e-01],
...,
[-2.5233e+00, -3.1543e+00, -2.8838e+00, ..., -3.0470e+00,
-2.4282e+00, -1.8557e+00],
[-1.6939e+00, -2.1569e+00, -2.0561e+00, ..., -2.1446e+00,
-1.7532e+00, -1.4018e+00],
[-1.1332e+00, -1.3627e+00, -1.3978e+00, ..., -1.3401e+00,
-1.1243e+00, -8.3515e-01]]],
[[[ 4.4192e-01, 9.1252e-01, 6.9364e-01, ..., 7.8439e-01,
1.0537e+00, 8.2770e-01],
[-1.0815e+00, -1.3252e-01, -1.4897e+00, ..., -1.2476e+00,
5.2717e-02, 6.2646e-01],
[ 7.2866e-01, 5.0595e-01, 1.2434e-01, ..., 4.8608e-01,
8.8033e-01, 8.8252e-01],
...,
[-8.6024e-01, -1.9460e+00, -2.0581e+00, ..., -1.7211e+00,
-1.2392e+00, -8.6831e-01],
[ 8.8792e-01, 6.0576e-02, -1.1979e-01, ..., -1.9464e-02,
7.0397e-01, 7.8848e-01],
[ 2.4856e-01, -4.9049e-01, -4.4994e-01, ..., -1.4824e-01,
4.4043e-01, 3.6385e-01]],
[[-2.6173e+00, -1.4095e+00, -4.7366e+00, ..., -5.8739e+00,
-5.0196e+00, -3.9043e+00],
[-6.6516e-01, 1.3702e+00, -3.3367e+00, ..., -4.7025e+00,
-4.5143e+00, -3.6730e+00],
[ 6.6827e-03, 1.1443e+00, -6.0767e+00, ..., -6.8503e+00,
-6.5269e+00, -5.9440e+00],
...,
[ 4.6484e-01, 6.8046e-01, -3.8396e+00, ..., -4.6666e+00,
-4.2377e+00, -3.5451e+00],
[ 2.6924e+00, 2.7595e+00, 3.8324e-01, ..., -9.2266e-01,
-1.2814e+00, -1.3160e+00],
[ 1.6090e+00, 1.4767e+00, -1.5294e+00, ..., -2.2787e+00,
-2.5359e+00, -2.4274e+00]],
[[-5.4141e-01, -1.5546e+00, -1.3693e+00, ..., -7.6090e-01,
-5.2190e-01, -7.4200e-01],
[-1.2975e+00, -1.6868e+00, -1.0621e+00, ..., -7.1060e-01,
-3.8858e-01, -3.2016e-01],
[-1.1708e+00, -1.0793e+00, -4.8502e-01, ..., -4.2832e-01,
-6.8790e-02, -7.6605e-02],
...,
[-3.1283e-01, -7.9426e-01, -5.0209e-01, ..., -4.9992e-01,
-1.9372e-01, -4.3808e-01],
[-8.7598e-01, -3.0359e-01, -4.5878e-01, ..., -2.1846e-01,
2.2945e-01, 2.2612e-01],
[-9.7433e-01, -2.8433e-01, -3.5306e-01, ..., -3.9896e-01,
-2.0191e-01, -2.3068e-01]],
...,
[[-2.0423e+00, -1.1354e+00, -2.4228e+00, ..., -3.3187e+00,
-2.7447e+00, -1.8699e+00],
[-4.8596e-01, 1.1675e+00, -6.2672e-01, ..., -1.6194e+00,
-1.8004e+00, -1.5812e+00],
[-2.0679e-01, 2.2180e+00, 5.5703e-01, ..., -3.4542e-01,
-4.6065e-02, 1.3639e-01],
...,
[-2.2259e+00, 1.0389e-01, -1.1089e-01, ..., -1.3111e+00,
-4.1443e-01, 8.3599e-02],
[-1.2394e+00, 1.0101e+00, 1.0664e+00, ..., 3.3128e-02,
2.8693e-01, 4.3704e-01],
[-2.2127e+00, -1.4841e+00, -1.7217e+00, ..., -2.3598e+00,
-1.4112e+00, -9.0007e-01]],
[[-1.4486e+00, -1.3951e+00, -2.7975e+00, ..., -1.0881e+00,
-5.6482e-01, -4.5150e-01],
[-8.7427e-01, -4.8517e-01, -3.7686e+00, ..., -1.8654e+00,
-1.6194e+00, -1.8296e+00],
[-2.4176e+00, -1.9175e+00, -6.3271e+00, ..., -4.1874e+00,
-3.6352e+00, -2.8704e+00],
...,
[-1.3328e+00, 2.0925e-01, -3.4842e+00, ..., -1.4565e+00,
-1.0707e+00, -9.7311e-01],
[-1.6823e+00, -4.7747e-01, -3.1093e+00, ..., -1.7386e+00,
-1.1919e+00, -8.4322e-01],
[-2.3826e+00, -1.4389e+00, -3.1302e+00, ..., -2.9323e+00,
-2.1368e+00, -1.9691e+00]],
[[-1.2284e+00, -1.8074e+00, -1.6263e+00, ..., -1.8592e+00,
-1.5415e+00, -1.1474e+00],
[-1.2925e+00, -1.8054e+00, -1.4041e+00, ..., -1.6255e+00,
-1.3180e+00, -1.0278e+00],
[-1.7086e+00, -2.0730e+00, -1.7099e+00, ..., -1.9049e+00,
-1.4684e+00, -1.0695e+00],
...,
[-2.4943e+00, -3.0850e+00, -2.7975e+00, ..., -2.8218e+00,
-2.2538e+00, -1.7308e+00],
[-1.6042e+00, -2.0836e+00, -1.9662e+00, ..., -1.9189e+00,
-1.5802e+00, -1.2714e+00],
[-1.0294e+00, -1.3277e+00, -1.3391e+00, ..., -1.1771e+00,
-1.0002e+00, -7.5778e-01]]],
[[[ 1.0785e-01, 2.6175e-01, -1.3648e+00, ..., -6.1027e-01,
-1.9395e-01, -3.1623e-02],
[-2.1769e+00, -1.6563e+00, -4.6489e+00, ..., -3.6029e+00,
-1.4217e+00, -4.9783e-01],
[ 6.5725e-01, -7.2268e-02, -8.4967e-01, ..., -1.5743e-01,
-2.0802e-01, 2.8447e-01],
...,
[-3.4732e-01, -1.1167e+00, -1.4318e+00, ..., -1.3634e+00,
-9.3414e-01, -7.0606e-01],
[ 9.4595e-01, 5.5997e-01, 3.6794e-01, ..., 3.7226e-01,
6.7434e-01, 7.8768e-01],
[ 6.6639e-01, 4.2986e-01, 3.5954e-01, ..., 5.2272e-01,
7.6621e-01, 7.0292e-01]],
[[-3.3622e+00, -1.6498e+00, -6.2373e+00, ..., -6.9903e+00,
-6.1346e+00, -4.5562e+00],
[-6.2064e-01, 2.0090e+00, -4.0965e+00, ..., -5.4970e+00,
-5.5400e+00, -4.2908e+00],
[-3.0683e-02, 1.6100e+00, -7.3152e+00, ..., -7.9597e+00,
-7.7936e+00, -6.6159e+00],
...,
[-4.4564e-01, 2.7194e-01, -4.4392e+00, ..., -5.3541e+00,
-4.6544e+00, -3.8725e+00],
[ 1.9668e+00, 2.4429e+00, -7.8263e-02, ..., -1.0893e+00,
-1.2119e+00, -1.3411e+00],
[ 1.1264e+00, 1.3131e+00, -1.6907e+00, ..., -2.4397e+00,
-2.5596e+00, -2.5511e+00]],
[[-1.3905e+00, -2.5317e+00, -2.4662e+00, ..., -1.6949e+00,
-1.2202e+00, -1.3370e+00],
[-1.8543e+00, -2.2091e+00, -1.6502e+00, ..., -1.1803e+00,
-7.0197e-01, -8.0015e-01],
[-2.3363e+00, -2.0881e+00, -1.0578e+00, ..., -1.0764e+00,
-4.7013e-01, -2.0284e-01],
...,
[-6.4333e-01, -1.1042e+00, -5.9183e-01, ..., -6.0558e-01,
-4.6705e-01, -6.3668e-01],
[-1.0477e+00, -6.9895e-01, -5.7409e-01, ..., -2.6980e-01,
-6.1858e-02, -4.2949e-02],
[-1.1976e+00, -6.6059e-01, -3.7369e-01, ..., -3.2599e-01,
-2.7361e-01, -2.4433e-01]],
...,
[[-2.2722e+00, -1.1374e+00, -2.4989e+00, ..., -3.0068e+00,
-2.4922e+00, -1.7480e+00],
[-1.8207e+00, 2.5400e-01, -1.3181e+00, ..., -2.4079e+00,
-2.4523e+00, -1.8826e+00],
[-3.5991e-01, 2.7660e+00, 1.2697e+00, ..., 1.4000e-01,
2.0794e-01, 4.6623e-01],
...,
[-1.5468e+00, 9.3739e-01, -1.4366e-01, ..., -1.1993e+00,
-4.4426e-01, 6.5398e-02],
[-5.5535e-01, 1.3670e+00, 7.4108e-01, ..., -6.4928e-03,
1.8960e-01, 3.1088e-01],
[-1.5943e+00, -1.0821e+00, -1.5510e+00, ..., -2.3197e+00,
-1.3422e+00, -7.0633e-01]],
[[-2.4046e+00, -2.7565e+00, -4.3655e+00, ..., -2.4568e+00,
-1.9943e+00, -1.7897e+00],
[-1.6439e+00, -1.3893e+00, -5.0896e+00, ..., -3.0006e+00,
-2.8590e+00, -2.7995e+00],
[-2.8047e+00, -2.2675e+00, -7.0362e+00, ..., -4.4278e+00,
-4.3508e+00, -3.4863e+00],
...,
[-1.6873e+00, -5.4311e-01, -4.0506e+00, ..., -2.0777e+00,
-1.5563e+00, -1.1907e+00],
[-1.6704e+00, -7.2979e-01, -3.3533e+00, ..., -2.0446e+00,
-1.3436e+00, -7.9037e-01],
[-2.0386e+00, -1.2716e+00, -2.8275e+00, ..., -2.6815e+00,
-1.7928e+00, -1.5325e+00]],
[[-1.8561e+00, -2.6198e+00, -2.4717e+00, ..., -2.6458e+00,
-2.1687e+00, -1.6662e+00],
[-1.8395e+00, -2.5716e+00, -2.1014e+00, ..., -2.3957e+00,
-1.8927e+00, -1.5343e+00],
[-2.3070e+00, -2.8632e+00, -2.4514e+00, ..., -2.7552e+00,
-2.1531e+00, -1.6391e+00],
...,
[-2.2819e+00, -2.8623e+00, -2.6724e+00, ..., -2.8698e+00,
-2.2911e+00, -1.7256e+00],
[-1.5432e+00, -1.9852e+00, -1.9397e+00, ..., -2.0294e+00,
-1.6682e+00, -1.3323e+00],
[-1.0466e+00, -1.2779e+00, -1.3392e+00, ..., -1.2591e+00,
-1.0802e+00, -7.9612e-01]]],
[[[ 3.0009e-01, 5.6404e-01, 6.9047e-01, ..., 4.8171e-01,
1.0161e+00, 6.0264e-01],
[-2.1818e+00, -1.5271e+00, -3.2409e+00, ..., -2.6741e+00,
-8.7121e-01, -2.0430e-01],
[-1.6981e-01, -1.0146e+00, -2.6223e-01, ..., -1.8000e-01,
5.2594e-01, 6.3709e-01],
...,
[-1.6516e-01, -1.1439e+00, -2.6525e+00, ..., -1.5344e+00,
-1.4151e+00, -8.4475e-01],
[ 1.4554e+00, 8.3969e-01, -9.5028e-01, ..., -2.6511e-01,
1.4496e-01, 4.1968e-01],
[ 8.6208e-01, 2.9084e-01, -6.8504e-01, ..., -1.7092e-01,
2.1975e-01, 3.8955e-01]],
[[-1.6854e+00, -1.1300e+00, -4.1786e+00, ..., -4.8528e+00,
-4.4413e+00, -3.3886e+00],
[ 5.9555e-01, 1.9773e+00, -2.2493e+00, ..., -3.5167e+00,
-3.6297e+00, -2.9728e+00],
[ 1.3486e+00, 1.7293e+00, -4.9306e+00, ..., -5.4592e+00,
-5.2871e+00, -4.9249e+00],
...,
[-1.2670e+00, 1.8463e-01, -6.6557e+00, ..., -7.5611e+00,
-6.8683e+00, -5.4090e+00],
[ 1.8082e+00, 2.8430e+00, -1.0224e+00, ..., -2.5639e+00,
-2.9323e+00, -2.3832e+00],
[ 1.2182e+00, 1.8253e+00, -2.2806e+00, ..., -3.2451e+00,
-3.6297e+00, -3.1388e+00]],
[[-1.0978e-01, -1.1582e+00, -1.3393e+00, ..., -4.7574e-01,
-1.4700e-01, -4.7163e-01],
[-5.9504e-01, -8.6298e-01, -7.9154e-01, ..., -2.6296e-01,
4.9113e-02, -6.9628e-02],
[-5.2705e-01, -3.3944e-01, -4.0196e-01, ..., -3.6088e-01,
7.4370e-02, 7.3106e-02],
...,
[-1.3759e+00, -1.7599e+00, -7.7583e-01, ..., -8.7409e-01,
-7.4599e-01, -1.0141e+00],
[-1.7232e+00, -1.1603e+00, -8.4180e-01, ..., -5.5296e-01,
-2.8648e-01, -3.5744e-01],
[-1.8754e+00, -1.2769e+00, -8.9108e-01, ..., -9.5419e-01,
-8.0900e-01, -7.8457e-01]],
...,
[[-3.0629e+00, -2.5331e+00, -2.8458e+00, ..., -3.4372e+00,
-2.6235e+00, -1.7486e+00],
[-2.2956e+00, -8.5057e-01, -1.6214e+00, ..., -2.6533e+00,
-2.3724e+00, -1.8629e+00],
[-1.0602e+00, 1.1963e+00, 6.3309e-01, ..., -4.4448e-01,
-2.2820e-02, 1.2017e-01],
...,
[-1.8203e+00, 1.3084e+00, 5.6822e-02, ..., -1.0748e+00,
-5.2946e-01, 5.1309e-03],
[-8.0070e-01, 1.9496e+00, 1.6522e+00, ..., 4.4302e-01,
4.1632e-01, 6.9092e-01],
[-2.2144e+00, -1.1310e+00, -9.6398e-01, ..., -1.9099e+00,
-1.0072e+00, -2.9865e-01]],
[[-1.5983e+00, -1.5803e+00, -2.8942e+00, ..., -1.2892e+00,
-8.6342e-01, -8.6362e-01],
[-5.8813e-01, 1.0127e-01, -3.1811e+00, ..., -1.3679e+00,
-1.2149e+00, -1.5687e+00],
[-2.0928e+00, -8.6740e-01, -5.3249e+00, ..., -3.1611e+00,
-2.6865e+00, -2.1907e+00],
...,
[-2.4558e+00, -1.5556e+00, -5.4703e+00, ..., -3.0603e+00,
-2.7163e+00, -2.1935e+00],
[-2.1980e+00, -1.4392e+00, -4.2526e+00, ..., -2.6525e+00,
-2.2995e+00, -1.6522e+00],
[-2.3741e+00, -1.7468e+00, -3.3017e+00, ..., -2.9379e+00,
-2.4187e+00, -2.1173e+00]],
[[-1.3654e+00, -1.9310e+00, -1.7224e+00, ..., -1.7680e+00,
-1.4611e+00, -1.1105e+00],
[-1.2843e+00, -1.7783e+00, -1.2426e+00, ..., -1.3743e+00,
-1.0866e+00, -8.6556e-01],
[-1.6049e+00, -1.9617e+00, -1.4874e+00, ..., -1.6085e+00,
-1.2014e+00, -8.8846e-01],
...,
[-3.1682e+00, -3.9583e+00, -3.7412e+00, ..., -3.9484e+00,
-3.1671e+00, -2.4460e+00],
[-2.1135e+00, -2.7289e+00, -2.6914e+00, ..., -2.7765e+00,
-2.3003e+00, -1.8736e+00],
[-1.5377e+00, -1.8669e+00, -1.9288e+00, ..., -1.8826e+00,
-1.5312e+00, -1.1762e+00]]],
[[[ 1.2844e-01, 3.6430e-01, -1.2028e+00, ..., 6.5409e-01,
9.3156e-01, 6.8694e-01],
[-2.0900e+00, -1.4018e+00, -4.3619e+00, ..., -1.5209e+00,
-2.0376e-01, 3.7804e-01],
[ 4.0879e-01, -3.0235e-01, -1.1529e+00, ..., 5.5276e-01,
9.4222e-01, 8.4984e-01],
...,
[-2.7249e-01, -1.2681e+00, -2.5811e+00, ..., -1.1767e+00,
-8.2631e-01, -6.3863e-01],
[ 1.2526e+00, 6.4795e-01, -9.2948e-01, ..., 4.9915e-01,
7.2667e-01, 7.6584e-01],
[ 8.2795e-01, 3.0811e-01, -4.4986e-01, ..., 5.2426e-01,
7.7218e-01, 6.7354e-01]],
[[-3.2727e+00, -1.5095e+00, -6.1452e+00, ..., -5.4698e+00,
-4.6710e+00, -3.6551e+00],
[-6.1898e-01, 2.0707e+00, -4.2268e+00, ..., -4.1760e+00,
-4.0659e+00, -3.3585e+00],
[-2.1357e-02, 1.7835e+00, -7.4418e+00, ..., -6.1313e+00,
-5.8927e+00, -5.4739e+00],
...,
[-1.2260e+00, 3.2217e-01, -6.4711e+00, ..., -5.3466e+00,
-4.7017e+00, -3.9321e+00],
[ 1.5461e+00, 2.6264e+00, -1.3359e+00, ..., -1.3000e+00,
-1.4382e+00, -1.5058e+00],
[ 1.0151e+00, 1.6422e+00, -2.4106e+00, ..., -2.4782e+00,
-2.6101e+00, -2.5461e+00]],
[[-1.3667e+00, -2.3160e+00, -2.1580e+00, ..., -8.4014e-01,
-6.2610e-01, -8.1926e-01],
[-1.9661e+00, -2.1322e+00, -1.5243e+00, ..., -6.9263e-01,
-4.1983e-01, -3.6149e-01],
[-2.4748e+00, -1.9528e+00, -8.1381e-01, ..., -3.8419e-01,
-9.0720e-02, -9.3006e-02],
...,
[-1.6089e+00, -1.8325e+00, -7.6938e-01, ..., -6.3396e-01,
-5.3787e-01, -6.8395e-01],
[-1.7797e+00, -1.2056e+00, -7.6875e-01, ..., -2.1335e-01,
-5.3698e-02, -4.8166e-02],
[-1.7337e+00, -1.1221e+00, -6.0533e-01, ..., -3.1390e-01,
-2.7931e-01, -2.5418e-01]],
...,
[[-2.3243e+00, -1.2936e+00, -2.6912e+00, ..., -3.1120e+00,
-2.4715e+00, -1.6423e+00],
[-1.8668e+00, 1.6075e-01, -1.3238e+00, ..., -1.5363e+00,
-1.5933e+00, -1.3793e+00],
[-5.2983e-01, 2.4300e+00, 9.0486e-01, ..., -1.1420e-01,
2.3176e-01, 3.1523e-01],
...,
[-1.7103e+00, 1.1922e+00, 1.9393e-01, ..., -9.8540e-01,
-2.8737e-01, 1.7972e-01],
[-8.3977e-01, 1.6797e+00, 1.4567e+00, ..., 1.3255e-01,
2.8876e-01, 3.8391e-01],
[-2.0814e+00, -1.0097e+00, -8.2370e-01, ..., -2.1764e+00,
-1.2554e+00, -6.4978e-01]],
[[-2.3699e+00, -2.6315e+00, -3.9416e+00, ..., -1.1075e+00,
-6.0802e-01, -4.9718e-01],
[-1.6575e+00, -1.3310e+00, -4.7033e+00, ..., -1.6726e+00,
-1.4874e+00, -1.7406e+00],
[-3.0154e+00, -2.4537e+00, -6.8644e+00, ..., -3.7190e+00,
-3.2750e+00, -2.6344e+00],
...,
[-2.5056e+00, -1.7091e+00, -4.9936e+00, ..., -2.2080e+00,
-1.7239e+00, -1.4006e+00],
[-2.1983e+00, -1.5268e+00, -3.8899e+00, ..., -2.1399e+00,
-1.4904e+00, -9.6365e-01],
[-2.2949e+00, -1.7381e+00, -3.0571e+00, ..., -2.8116e+00,
-1.9497e+00, -1.6600e+00]],
[[-1.8934e+00, -2.6176e+00, -2.4194e+00, ..., -1.6986e+00,
-1.4154e+00, -1.0528e+00],
[-1.9120e+00, -2.6256e+00, -2.1147e+00, ..., -1.3640e+00,
-1.1119e+00, -8.8013e-01],
[-2.4353e+00, -2.9861e+00, -2.5235e+00, ..., -1.5665e+00,
-1.2039e+00, -8.7279e-01],
...,
[-3.1134e+00, -3.8866e+00, -3.6528e+00, ..., -2.7954e+00,
-2.2318e+00, -1.6877e+00],
[-2.0711e+00, -2.6814e+00, -2.6436e+00, ..., -1.9581e+00,
-1.6087e+00, -1.2802e+00],
[-1.4877e+00, -1.8139e+00, -1.8722e+00, ..., -1.2131e+00,
-1.0411e+00, -7.6860e-01]]]], device='cuda:0'),
tensor([[[[ 1.5723e-01, 3.8983e-01, -1.1588e+00, ..., 4.7506e-01,
1.0254e+00, 6.5528e-01],
[-2.0466e+00, -1.4206e+00, -4.3318e+00, ..., -2.6031e+00,
-7.4384e-01, -3.6219e-02],
[ 4.2948e-01, -3.0300e-01, -1.1085e+00, ..., -3.5381e-01,
5.4464e-01, 7.2262e-01],
...,
[-2.6878e-01, -1.2777e+00, -2.6142e+00, ..., -2.0103e+00,
-1.4248e+00, -9.2909e-01],
[ 1.2664e+00, 6.5062e-01, -9.3222e-01, ..., -3.3589e-01,
4.6894e-01, 6.8853e-01],
[ 8.3479e-01, 3.1433e-01, -4.4053e-01, ..., -3.6840e-01,
2.6019e-01, 2.5800e-01]],
[[-3.2632e+00, -1.4707e+00, -6.1477e+00, ..., -4.8033e+00,
-4.3976e+00, -3.2720e+00],
[-6.2899e-01, 2.0928e+00, -4.1895e+00, ..., -3.3688e+00,
-3.5126e+00, -2.7850e+00],
[-2.3290e-02, 1.8080e+00, -7.3572e+00, ..., -5.3678e+00,
-5.2412e+00, -4.8481e+00],
...,
[-1.2374e+00, 2.7111e-01, -6.5001e+00, ..., -4.5451e+00,
-4.1010e+00, -3.4540e+00],
[ 1.5291e+00, 2.6021e+00, -1.3251e+00, ..., -8.3779e-01,
-1.1754e+00, -1.2397e+00],
[ 9.9536e-01, 1.6230e+00, -2.3918e+00, ..., -2.2115e+00,
-2.4514e+00, -2.3564e+00]],
[[-1.3394e+00, -2.2998e+00, -2.1432e+00, ..., -5.3990e-01,
-1.5547e-01, -4.6083e-01],
[-1.9742e+00, -2.1599e+00, -1.5571e+00, ..., -2.6571e-01,
1.1881e-01, 1.5003e-02],
[-2.5036e+00, -1.9888e+00, -8.4290e-01, ..., -2.1419e-01,
3.0884e-01, 2.5042e-01],
...,
[-1.6143e+00, -1.8237e+00, -7.6690e-01, ..., -4.9168e-01,
-1.6641e-01, -3.4347e-01],
[-1.7623e+00, -1.1792e+00, -7.4614e-01, ..., -3.1028e-01,
1.5868e-01, 2.3139e-01],
[-1.7190e+00, -1.1067e+00, -5.9636e-01, ..., -5.0279e-01,
-2.7892e-01, -2.4538e-01]],
...,
[[-2.3186e+00, -1.2843e+00, -2.7373e+00, ..., -3.6601e+00,
-2.9141e+00, -1.9122e+00],
[-1.9159e+00, 1.8509e-01, -1.3149e+00, ..., -2.6727e+00,
-2.5120e+00, -1.8823e+00],
[-5.2633e-01, 2.4306e+00, 8.9429e-01, ..., -5.2786e-01,
-1.4143e-01, 1.8811e-01],
...,
[-1.7085e+00, 1.1863e+00, 2.4155e-01, ..., -1.5594e+00,
-6.7458e-01, -1.3329e-01],
[-8.6213e-01, 1.6670e+00, 1.5238e+00, ..., -1.8840e-01,
9.2318e-02, 3.2527e-01],
[-2.1121e+00, -1.0687e+00, -8.4004e-01, ..., -2.3962e+00,
-1.4745e+00, -9.2630e-01]],
[[-2.3745e+00, -2.6449e+00, -4.0025e+00, ..., -1.0794e+00,
-7.2396e-01, -7.6200e-01],
[-1.6529e+00, -1.3398e+00, -4.7336e+00, ..., -1.2878e+00,
-1.2143e+00, -1.6404e+00],
[-2.9921e+00, -2.4249e+00, -6.8973e+00, ..., -3.1804e+00,
-2.7722e+00, -2.3237e+00],
...,
[-2.5169e+00, -1.7242e+00, -4.9423e+00, ..., -1.4326e+00,
-1.0305e+00, -8.6954e-01],
[-2.2190e+00, -1.5408e+00, -3.8676e+00, ..., -1.7048e+00,
-1.1476e+00, -7.5181e-01],
[-2.3115e+00, -1.7698e+00, -3.0571e+00, ..., -2.8085e+00,
-2.0147e+00, -1.8406e+00]],
[[-1.8845e+00, -2.6193e+00, -2.4066e+00, ..., -1.8070e+00,
-1.5101e+00, -1.1484e+00],
[-1.9112e+00, -2.6271e+00, -2.1001e+00, ..., -1.4229e+00,
-1.1270e+00, -9.0266e-01],
[-2.4392e+00, -2.9939e+00, -2.5039e+00, ..., -1.6356e+00,
-1.2263e+00, -9.1868e-01],
...,
[-3.1106e+00, -3.8830e+00, -3.6527e+00, ..., -2.8882e+00,
-2.3095e+00, -1.7809e+00],
[-2.0632e+00, -2.6696e+00, -2.6309e+00, ..., -1.9910e+00,
-1.6337e+00, -1.3213e+00],
[-1.4812e+00, -1.8022e+00, -1.8594e+00, ..., -1.2267e+00,
-1.0363e+00, -7.9456e-01]]],
[[[ 2.0573e-01, 4.8681e-01, 6.4215e-01, ..., 6.0409e-01,
9.8487e-01, 7.7615e-01],
[-2.0308e+00, -1.2990e+00, -2.7321e+00, ..., -1.4798e+00,
-3.1994e-02, 5.7580e-01],
[-1.6831e-01, -8.6516e-01, 1.6505e-01, ..., 3.8265e-01,
9.1441e-01, 9.2917e-01],
...,
[-9.4382e-01, -2.0959e+00, -2.0091e+00, ..., -1.2003e+00,
-7.2053e-01, -5.0154e-01],
[ 7.5368e-01, -1.5633e-01, -1.2314e-01, ..., 4.2202e-01,
8.0667e-01, 8.9167e-01],
[ 1.7581e-01, -6.1437e-01, -4.3049e-01, ..., 5.8413e-01,
8.6588e-01, 7.8049e-01]],
[[-1.6392e+00, -1.3709e+00, -4.3041e+00, ..., -5.9502e+00,
-5.0366e+00, -3.9084e+00],
[ 5.0055e-01, 1.4627e+00, -2.5944e+00, ..., -4.6547e+00,
-4.4316e+00, -3.6263e+00],
[ 1.2056e+00, 1.1457e+00, -5.1265e+00, ..., -6.8847e+00,
-6.4297e+00, -5.8952e+00],
...,
[ 3.8820e-01, 2.9008e-01, -4.1975e+00, ..., -5.9983e+00,
-5.1970e+00, -4.2606e+00],
[ 2.4505e+00, 2.3858e+00, -9.0592e-03, ..., -1.6368e+00,
-1.7007e+00, -1.6885e+00],
[ 1.4393e+00, 1.2493e+00, -1.6671e+00, ..., -2.7954e+00,
-2.8481e+00, -2.7468e+00]],
[[-9.7013e-02, -1.1385e+00, -1.4198e+00, ..., -8.7793e-01,
-5.8539e-01, -7.2538e-01],
[-4.4963e-01, -7.7282e-01, -8.6626e-01, ..., -7.3910e-01,
-3.9421e-01, -2.4055e-01],
[-2.8912e-01, -1.6841e-01, -4.1341e-01, ..., -4.0562e-01,
1.8159e-02, 1.2316e-01],
...,
[-2.0716e-03, -5.2028e-01, -5.4580e-01, ..., -6.3074e-01,
-4.3567e-01, -5.0482e-01],
[-6.6340e-01, -1.7306e-01, -5.3693e-01, ..., -2.8337e-01,
-1.1555e-02, 7.1136e-02],
[-8.2025e-01, -2.3390e-01, -4.6386e-01, ..., -2.9344e-01,
-1.9378e-01, -1.2756e-01]],
...,
[[-2.8340e+00, -2.5233e+00, -2.4828e+00, ..., -3.4307e+00,
-2.8009e+00, -1.8686e+00],
[-2.1142e+00, -9.7091e-01, -1.1989e+00, ..., -1.7798e+00,
-1.8641e+00, -1.5690e+00],
[-1.1647e+00, 6.1640e-01, 5.7758e-01, ..., -2.7849e-01,
4.3571e-02, 2.0108e-01],
...,
[-2.3606e+00, -4.1609e-01, -2.8922e-01, ..., -1.0162e+00,
-3.9048e-01, 1.1524e-01],
[-1.3114e+00, 6.7997e-01, 9.2144e-01, ..., 1.6546e-01,
2.6854e-01, 3.7448e-01],
[-2.2023e+00, -1.5719e+00, -1.7347e+00, ..., -2.0469e+00,
-1.1720e+00, -6.0047e-01]],
[[-1.6353e+00, -1.6836e+00, -2.8412e+00, ..., -1.1851e+00,
-5.5913e-01, -4.0264e-01],
[-6.9690e-01, -1.6765e-01, -3.1692e+00, ..., -1.8592e+00,
-1.5110e+00, -1.6853e+00],
[-2.0386e+00, -1.0745e+00, -5.1051e+00, ..., -4.0138e+00,
-3.3965e+00, -2.6061e+00],
...,
[-1.3884e+00, 1.3561e-01, -3.3225e+00, ..., -2.5189e+00,
-1.8851e+00, -1.3939e+00],
[-1.6848e+00, -5.2857e-01, -2.9915e+00, ..., -2.3316e+00,
-1.5667e+00, -9.3661e-01],
[-2.3020e+00, -1.3847e+00, -3.0386e+00, ..., -2.8297e+00,
-1.9102e+00, -1.5862e+00]],
[[-1.3159e+00, -1.8338e+00, -1.7001e+00, ..., -1.9256e+00,
-1.5582e+00, -1.1372e+00],
[-1.2656e+00, -1.7142e+00, -1.2697e+00, ..., -1.6263e+00,
-1.2829e+00, -9.7895e-01],
[-1.5753e+00, -1.9075e+00, -1.5177e+00, ..., -1.8778e+00,
-1.3879e+00, -9.7672e-01],
...,
[-2.4559e+00, -3.0615e+00, -2.8191e+00, ..., -3.0587e+00,
-2.4082e+00, -1.7910e+00],
[-1.5948e+00, -2.0897e+00, -1.9996e+00, ..., -2.1755e+00,
-1.7579e+00, -1.3772e+00],
[-1.0366e+00, -1.3519e+00, -1.3709e+00, ..., -1.3613e+00,
-1.1359e+00, -8.2745e-01]]],
[[[ 2.3001e-01, 4.9753e-01, 6.4860e-01, ..., 5.9550e-01,
9.6594e-01, 7.8091e-01],
[-1.9995e+00, -1.3215e+00, -2.7915e+00, ..., -1.2867e+00,
7.5947e-02, 6.2101e-01],
[-1.6166e-01, -9.1940e-01, 4.4936e-02, ..., 4.8620e-01,
9.3842e-01, 9.3992e-01],
...,
[-9.4016e-01, -2.1343e+00, -2.0337e+00, ..., -1.3926e+00,
-8.6393e-01, -6.1274e-01],
[ 8.0148e-01, -1.2325e-01, -1.2962e-01, ..., 4.1191e-01,
8.3821e-01, 9.2684e-01],
[ 2.0512e-01, -5.8761e-01, -3.9894e-01, ..., 5.9709e-01,
9.0567e-01, 8.1284e-01]],
[[-1.6461e+00, -1.3539e+00, -4.2434e+00, ..., -5.7795e+00,
-4.9011e+00, -3.8176e+00],
[ 4.8971e-01, 1.4779e+00, -2.5478e+00, ..., -4.7324e+00,
-4.4593e+00, -3.6528e+00],
[ 1.2139e+00, 1.2023e+00, -5.0661e+00, ..., -6.8543e+00,
-6.3757e+00, -5.8023e+00],
...,
[ 3.8463e-01, 3.6618e-01, -4.1002e+00, ..., -6.0286e+00,
-5.1926e+00, -4.2740e+00],
[ 2.5490e+00, 2.5098e+00, 1.3617e-01, ..., -1.3751e+00,
-1.4768e+00, -1.5593e+00],
[ 1.5119e+00, 1.3374e+00, -1.6191e+00, ..., -2.7685e+00,
-2.8571e+00, -2.8109e+00]],
[[-1.2784e-01, -1.1597e+00, -1.3986e+00, ..., -8.0617e-01,
-5.4065e-01, -6.6909e-01],
[-4.9829e-01, -8.2142e-01, -8.7118e-01, ..., -6.7612e-01,
-3.6161e-01, -2.2446e-01],
[-3.2620e-01, -2.2506e-01, -4.2749e-01, ..., -3.8450e-01,
-1.0704e-02, 9.4777e-02],
...,
[-6.2435e-02, -6.0919e-01, -6.0697e-01, ..., -6.7283e-01,
-4.7691e-01, -5.6904e-01],
[-7.0319e-01, -2.2042e-01, -5.6753e-01, ..., -2.8444e-01,
-9.5887e-03, 6.1658e-02],
[-8.7741e-01, -2.7118e-01, -4.7879e-01, ..., -3.1348e-01,
-2.1152e-01, -1.5066e-01]],
...,
[[-2.8168e+00, -2.4369e+00, -2.4676e+00, ..., -3.1318e+00,
-2.5747e+00, -1.7257e+00],
[-2.0921e+00, -8.3146e-01, -1.1461e+00, ..., -1.6639e+00,
-1.7530e+00, -1.4670e+00],
[-1.1089e+00, 7.9788e-01, 7.1783e-01, ..., -2.7526e-01,
-7.7215e-03, 1.5119e-01],
...,
[-2.2848e+00, -1.8830e-01, -1.4615e-01, ..., -1.0605e+00,
-4.0110e-01, 1.6200e-01],
[-1.2476e+00, 8.6101e-01, 1.0747e+00, ..., 2.3922e-01,
3.3500e-01, 4.5138e-01],
[-2.1832e+00, -1.5108e+00, -1.6932e+00, ..., -2.1889e+00,
-1.2349e+00, -5.9623e-01]],
[[-1.6260e+00, -1.6342e+00, -2.8129e+00, ..., -1.3333e+00,
-7.4073e-01, -5.3477e-01],
[-7.0128e-01, -9.8399e-02, -3.1722e+00, ..., -2.0107e+00,
-1.6384e+00, -1.7153e+00],
[-2.0542e+00, -9.7396e-01, -5.1474e+00, ..., -4.0090e+00,
-3.3633e+00, -2.5711e+00],
...,
[-1.3253e+00, 1.9950e-01, -3.3353e+00, ..., -2.4013e+00,
-1.8121e+00, -1.3667e+00],
[-1.6683e+00, -4.5653e-01, -3.0069e+00, ..., -2.2692e+00,
-1.5138e+00, -9.0866e-01],
[-2.3235e+00, -1.3598e+00, -3.0383e+00, ..., -2.8637e+00,
-1.9350e+00, -1.6362e+00]],
[[-1.2998e+00, -1.8242e+00, -1.6452e+00, ..., -1.8725e+00,
-1.5121e+00, -1.1065e+00],
[-1.2439e+00, -1.7047e+00, -1.2243e+00, ..., -1.6416e+00,
-1.2966e+00, -9.8425e-01],
[-1.5668e+00, -1.9049e+00, -1.4820e+00, ..., -1.9112e+00,
-1.4377e+00, -1.0143e+00],
...,
[-2.4483e+00, -3.0609e+00, -2.7901e+00, ..., -3.1716e+00,
-2.4910e+00, -1.8539e+00],
[-1.5833e+00, -2.0850e+00, -1.9819e+00, ..., -2.2437e+00,
-1.8161e+00, -1.4287e+00],
[-1.0352e+00, -1.3396e+00, -1.3540e+00, ..., -1.3950e+00,
-1.1693e+00, -8.5146e-01]]],
[[[ 1.4302e-01, -2.3778e-01, -4.2279e+00, ..., 4.8652e-01,
1.0209e+00, 6.6581e-01],
[-2.1336e+00, -3.7859e+00, -8.3878e+00, ..., -2.5294e+00,
-7.2187e-01, -2.4639e-02],
[ 4.2623e-01, -3.2014e+00, -6.1206e+00, ..., -2.2355e-01,
5.2820e-01, 6.8241e-01],
...,
[-2.9357e-01, -1.2703e+00, -2.6023e+00, ..., -1.9543e+00,
-1.3596e+00, -9.1968e-01],
[ 1.2445e+00, 6.5497e-01, -9.3002e-01, ..., -3.6470e-01,
4.7399e-01, 6.9021e-01],
[ 8.0369e-01, 2.9931e-01, -4.3972e-01, ..., -3.2049e-01,
2.9988e-01, 2.8454e-01]],
[[-3.2660e+00, -1.1929e+00, -6.4834e+00, ..., -4.8526e+00,
-4.4003e+00, -3.3639e+00],
[-5.7622e-01, 2.4302e+00, -4.1976e+00, ..., -3.4124e+00,
-3.5349e+00, -2.9128e+00],
[ 3.6431e-02, 2.1265e+00, -6.3549e+00, ..., -5.3804e+00,
-5.2572e+00, -4.9429e+00],
...,
[-1.2817e+00, 2.4687e-01, -6.5447e+00, ..., -4.5630e+00,
-4.0571e+00, -3.3894e+00],
[ 1.5205e+00, 2.6123e+00, -1.3347e+00, ..., -8.3498e-01,
-1.1347e+00, -1.2312e+00],
[ 9.9457e-01, 1.6407e+00, -2.3949e+00, ..., -2.2230e+00,
-2.4302e+00, -2.3622e+00]],
[[-1.4079e+00, -2.5575e+00, -2.0666e+00, ..., -5.7858e-01,
-1.9772e-01, -4.7023e-01],
[-2.0350e+00, -2.4461e+00, -1.5316e+00, ..., -3.3403e-01,
5.4995e-02, 1.8367e-02],
[-2.5357e+00, -2.1068e+00, -1.1525e+00, ..., -2.6619e-01,
2.4989e-01, 2.6381e-01],
...,
[-1.6014e+00, -1.8353e+00, -7.7614e-01, ..., -4.6628e-01,
-1.4234e-01, -3.4131e-01],
[-1.7628e+00, -1.1880e+00, -7.7099e-01, ..., -2.9337e-01,
1.6575e-01, 2.2323e-01],
[-1.7399e+00, -1.1433e+00, -6.3002e-01, ..., -4.7618e-01,
-2.6172e-01, -2.4288e-01]],
...,
[[-2.3728e+00, -1.3503e+00, -3.4918e+00, ..., -3.6828e+00,
-2.8649e+00, -1.8686e+00],
[-1.8921e+00, -4.0701e-01, -2.8409e+00, ..., -2.6731e+00,
-2.4338e+00, -1.8508e+00],
[-4.9885e-01, 1.4923e+00, -6.7654e-01, ..., -5.9790e-01,
-1.1467e-01, 1.4974e-01],
...,
[-1.7663e+00, 1.1361e+00, 1.8005e-01, ..., -1.6026e+00,
-7.1937e-01, -1.4397e-01],
[-8.7134e-01, 1.6500e+00, 1.4865e+00, ..., -2.0674e-01,
9.4112e-02, 3.1669e-01],
[-2.1143e+00, -1.0766e+00, -8.4868e-01, ..., -2.3884e+00,
-1.4628e+00, -9.0663e-01]],
[[-2.3813e+00, -2.4478e+00, -3.2662e+00, ..., -1.1173e+00,
-7.3058e-01, -7.5424e-01],
[-1.6638e+00, -1.1269e+00, -4.5193e+00, ..., -1.3526e+00,
-1.2327e+00, -1.6215e+00],
[-3.0151e+00, -2.1759e+00, -6.7142e+00, ..., -3.2741e+00,
-2.8232e+00, -2.2884e+00],
...,
[-2.5341e+00, -1.7474e+00, -4.9483e+00, ..., -1.4663e+00,
-9.9980e-01, -8.7245e-01],
[-2.2193e+00, -1.5581e+00, -3.8766e+00, ..., -1.7088e+00,
-1.1354e+00, -7.7228e-01],
[-2.3128e+00, -1.7863e+00, -3.0718e+00, ..., -2.7866e+00,
-2.0012e+00, -1.8336e+00]],
[[-1.9126e+00, -2.8915e+00, -3.2279e+00, ..., -1.7905e+00,
-1.4909e+00, -1.1350e+00],
[-1.9396e+00, -2.8295e+00, -3.2594e+00, ..., -1.3993e+00,
-1.1055e+00, -8.8253e-01],
[-2.4684e+00, -3.2994e+00, -3.9162e+00, ..., -1.6002e+00,
-1.2000e+00, -8.9015e-01],
...,
[-3.1208e+00, -3.8986e+00, -3.6717e+00, ..., -2.8838e+00,
-2.2990e+00, -1.7593e+00],
[-2.0711e+00, -2.6822e+00, -2.6450e+00, ..., -1.9714e+00,
-1.6172e+00, -1.3068e+00],
[-1.4912e+00, -1.8135e+00, -1.8579e+00, ..., -1.2189e+00,
-1.0322e+00, -7.9092e-01]]],
[[[ 9.9605e-02, 2.5892e-01, -1.3501e+00, ..., -5.8910e-01,
-1.8250e-01, -4.5071e-02],
[-2.1703e+00, -1.6368e+00, -4.6113e+00, ..., -3.5493e+00,
-1.4033e+00, -5.0491e-01],
[ 6.6538e-01, -6.1905e-02, -8.4125e-01, ..., -1.1204e-01,
-1.9136e-01, 2.8405e-01],
...,
[-8.4846e-01, -1.8325e+00, -1.8105e+00, ..., -1.8598e+00,
-1.3592e+00, -9.6733e-01],
[ 7.5067e-01, -4.1305e-03, -6.7137e-02, ..., -5.6813e-02,
7.1002e-01, 8.0630e-01],
[ 1.3652e-01, -5.7933e-01, -4.5213e-01, ..., -1.8478e-01,
4.3790e-01, 3.4135e-01]],
[[-3.3277e+00, -1.6441e+00, -6.2092e+00, ..., -6.9705e+00,
-6.1312e+00, -4.5433e+00],
[-5.9274e-01, 2.0077e+00, -4.0995e+00, ..., -5.5342e+00,
-5.5614e+00, -4.2641e+00],
[-1.2549e-02, 1.6056e+00, -7.3208e+00, ..., -7.9767e+00,
-7.7755e+00, -6.5466e+00],
...,
[ 2.8448e-01, 3.0286e-01, -4.0933e+00, ..., -4.6153e+00,
-4.2013e+00, -3.4569e+00],
[ 2.2441e+00, 2.1723e+00, -2.7922e-01, ..., -7.6578e-01,
-1.1605e+00, -1.1747e+00],
[ 1.3192e+00, 1.1119e+00, -1.8176e+00, ..., -2.2256e+00,
-2.5144e+00, -2.3816e+00]],
[[-1.4057e+00, -2.5422e+00, -2.4675e+00, ..., -1.6908e+00,
-1.2054e+00, -1.3243e+00],
[-1.8740e+00, -2.2139e+00, -1.6456e+00, ..., -1.1677e+00,
-7.0108e-01, -8.0161e-01],
[-2.3513e+00, -2.0832e+00, -1.0397e+00, ..., -1.0733e+00,
-4.6441e-01, -1.9941e-01],
...,
[-1.2267e-01, -5.7697e-01, -4.6134e-01, ..., -4.9595e-01,
-1.8097e-01, -4.3496e-01],
[-6.7875e-01, -1.7810e-01, -4.1854e-01, ..., -1.9598e-01,
2.6813e-01, 2.5472e-01],
[-7.6338e-01, -1.6290e-01, -3.2270e-01, ..., -4.0202e-01,
-1.8669e-01, -2.1851e-01]],
...,
[[-2.2630e+00, -1.1290e+00, -2.4611e+00, ..., -2.9519e+00,
-2.4202e+00, -1.7067e+00],
[-1.7988e+00, 2.6062e-01, -1.3080e+00, ..., -2.3887e+00,
-2.3916e+00, -1.8617e+00],
[-3.2730e-01, 2.7837e+00, 1.2797e+00, ..., 8.4496e-02,
2.1573e-01, 4.6675e-01],
...,
[-2.0700e+00, -1.5420e-01, -2.0383e-01, ..., -1.3528e+00,
-4.0216e-01, 1.0073e-01],
[-1.2141e+00, 7.4840e-01, 8.7205e-01, ..., 7.6197e-03,
3.1377e-01, 4.9632e-01],
[-2.0977e+00, -1.4040e+00, -1.6360e+00, ..., -2.5156e+00,
-1.5143e+00, -9.3781e-01]],
[[-2.4045e+00, -2.7568e+00, -4.3752e+00, ..., -2.5237e+00,
-2.0233e+00, -1.7975e+00],
[-1.6437e+00, -1.3905e+00, -5.0849e+00, ..., -3.0625e+00,
-2.8673e+00, -2.7661e+00],
[-2.7997e+00, -2.2695e+00, -7.0117e+00, ..., -4.4711e+00,
-4.3502e+00, -3.4423e+00],
...,
[-1.4446e+00, -1.1263e-01, -3.3983e+00, ..., -1.4169e+00,
-9.9052e-01, -9.1554e-01],
[-1.7335e+00, -6.8833e-01, -3.0351e+00, ..., -1.7127e+00,
-1.1330e+00, -8.0491e-01],
[-2.3201e+00, -1.5118e+00, -3.0987e+00, ..., -2.9630e+00,
-2.1214e+00, -1.9836e+00]],
[[-1.8616e+00, -2.6145e+00, -2.4729e+00, ..., -2.6313e+00,
-2.1619e+00, -1.6683e+00],
[-1.8459e+00, -2.5709e+00, -2.1029e+00, ..., -2.3938e+00,
-1.8965e+00, -1.5393e+00],
[-2.3101e+00, -2.8660e+00, -2.4545e+00, ..., -2.7630e+00,
-2.1632e+00, -1.6500e+00],
...,
[-2.3697e+00, -2.9342e+00, -2.7235e+00, ..., -2.8683e+00,
-2.2968e+00, -1.7702e+00],
[-1.5409e+00, -2.0074e+00, -1.9482e+00, ..., -1.9493e+00,
-1.6134e+00, -1.3060e+00],
[-1.0073e+00, -1.3036e+00, -1.3430e+00, ..., -1.1933e+00,
-1.0224e+00, -7.8230e-01]]]], device='cuda:0'),
tensor([[[[ 4.8140e-01, 9.5260e-01, 7.6985e-01, ..., 7.7292e-01,
1.0546e+00, 8.0856e-01],
[-1.0047e+00, -9.9333e-02, -1.3557e+00, ..., -1.2082e+00,
5.2885e-02, 6.0665e-01],
[ 7.1911e-01, 5.3799e-01, 2.7683e-01, ..., 5.4187e-01,
8.8460e-01, 8.6803e-01],
...,
[-8.7282e-01, -1.8288e+00, -1.8216e+00, ..., -1.7772e+00,
-1.3441e+00, -9.5842e-01],
[ 7.3312e-01, -1.4051e-02, -1.1269e-01, ..., -1.2975e-02,
6.9513e-01, 7.9211e-01],
[ 1.3017e-01, -5.8185e-01, -4.7887e-01, ..., -1.6037e-01,
4.2850e-01, 3.5126e-01]],
[[-2.5250e+00, -1.4124e+00, -4.7297e+00, ..., -5.8402e+00,
-4.9985e+00, -3.8540e+00],
[-6.4949e-01, 1.3019e+00, -3.3894e+00, ..., -4.6917e+00,
-4.4829e+00, -3.6017e+00],
[-1.7650e-03, 1.1161e+00, -6.0305e+00, ..., -6.8139e+00,
-6.4578e+00, -5.8558e+00],
...,
[ 2.7822e-01, 3.2023e-01, -4.0509e+00, ..., -4.6432e+00,
-4.1668e+00, -3.4657e+00],
[ 2.2968e+00, 2.2519e+00, -1.5886e-01, ..., -8.0827e-01,
-1.1534e+00, -1.2004e+00],
[ 1.3529e+00, 1.1629e+00, -1.7588e+00, ..., -2.2286e+00,
-2.5002e+00, -2.3993e+00]],
[[-5.1822e-01, -1.5311e+00, -1.3486e+00, ..., -7.6032e-01,
-5.0401e-01, -7.2371e-01],
[-1.2583e+00, -1.6712e+00, -1.0678e+00, ..., -6.7140e-01,
-3.5685e-01, -3.0433e-01],
[-1.1284e+00, -1.0595e+00, -4.7316e-01, ..., -3.9991e-01,
-4.6132e-02, -4.9009e-02],
...,
[-9.4276e-02, -5.8523e-01, -4.6708e-01, ..., -5.0292e-01,
-1.9216e-01, -4.5087e-01],
[-6.8893e-01, -1.9686e-01, -4.3763e-01, ..., -1.9753e-01,
2.5738e-01, 2.3941e-01],
[-7.9691e-01, -1.8400e-01, -3.3658e-01, ..., -3.9843e-01,
-1.8857e-01, -2.2816e-01]],
...,
[[-1.9719e+00, -1.1157e+00, -2.4441e+00, ..., -3.2447e+00,
-2.6598e+00, -1.8238e+00],
[-4.7779e-01, 1.1475e+00, -6.8433e-01, ..., -1.6171e+00,
-1.7724e+00, -1.5612e+00],
[-2.2889e-01, 2.1489e+00, 4.5179e-01, ..., -4.0459e-01,
-6.3922e-02, 1.0971e-01],
...,
[-2.1306e+00, -1.8114e-01, -2.1898e-01, ..., -1.3750e+00,
-4.0460e-01, 9.4847e-02],
[-1.2218e+00, 7.6661e-01, 9.2529e-01, ..., 3.1090e-04,
3.0294e-01, 4.7283e-01],
[-2.1232e+00, -1.4334e+00, -1.6455e+00, ..., -2.4675e+00,
-1.4725e+00, -9.3069e-01]],
[[-1.4277e+00, -1.4157e+00, -2.8133e+00, ..., -1.1012e+00,
-5.5243e-01, -4.2383e-01],
[-8.9654e-01, -5.1872e-01, -3.7302e+00, ..., -1.9019e+00,
-1.5896e+00, -1.7724e+00],
[-2.4054e+00, -1.9234e+00, -6.3104e+00, ..., -4.2116e+00,
-3.5903e+00, -2.8182e+00],
...,
[-1.3998e+00, -8.3275e-02, -3.4040e+00, ..., -1.4150e+00,
-1.0083e+00, -9.2135e-01],
[-1.7078e+00, -6.7983e-01, -3.0455e+00, ..., -1.7224e+00,
-1.1556e+00, -8.1236e-01],
[-2.3178e+00, -1.5120e+00, -3.1073e+00, ..., -2.9522e+00,
-2.1390e+00, -1.9917e+00]],
[[-1.2083e+00, -1.7748e+00, -1.6044e+00, ..., -1.8402e+00,
-1.5312e+00, -1.1420e+00],
[-1.2780e+00, -1.7756e+00, -1.3852e+00, ..., -1.6051e+00,
-1.3105e+00, -1.0206e+00],
[-1.6996e+00, -2.0522e+00, -1.6811e+00, ..., -1.8881e+00,
-1.4548e+00, -1.0618e+00],
...,
[-2.3832e+00, -2.9415e+00, -2.7313e+00, ..., -2.8506e+00,
-2.2780e+00, -1.7503e+00],
[-1.5461e+00, -2.0056e+00, -1.9367e+00, ..., -1.9334e+00,
-1.5911e+00, -1.2850e+00],
[-1.0066e+00, -1.2969e+00, -1.3313e+00, ..., -1.1810e+00,
-1.0084e+00, -7.6414e-01]]],
[[[ 1.2151e-01, 3.5758e-01, -1.1707e+00, ..., 4.7437e-01,
1.0182e+00, 6.4473e-01],
[-2.0761e+00, -1.4193e+00, -4.2934e+00, ..., -2.5706e+00,
-7.4415e-01, -7.2564e-02],
[ 4.6003e-01, -2.5444e-01, -1.0817e+00, ..., -2.6386e-01,
4.9256e-01, 6.2701e-01],
...,
[-2.8472e-01, -1.2906e+00, -2.6428e+00, ..., -2.0211e+00,
-1.3969e+00, -9.3629e-01],
[ 1.2483e+00, 6.7229e-01, -9.0818e-01, ..., -3.4955e-01,
4.9507e-01, 6.8689e-01],
[ 8.3419e-01, 3.4705e-01, -3.9870e-01, ..., -3.7248e-01,
2.7856e-01, 2.7170e-01]],
[[-3.2463e+00, -1.5627e+00, -6.1603e+00, ..., -4.8501e+00,
-4.4282e+00, -3.3663e+00],
[-6.2762e-01, 1.9792e+00, -4.2291e+00, ..., -3.3974e+00,
-3.5430e+00, -2.8827e+00],
[-6.1072e-02, 1.6866e+00, -7.3714e+00, ..., -5.3772e+00,
-5.2587e+00, -4.9097e+00],
...,
[-1.2489e+00, 2.5126e-01, -6.5089e+00, ..., -4.5459e+00,
-4.1062e+00, -3.4183e+00],
[ 1.5254e+00, 2.5886e+00, -1.3285e+00, ..., -8.4745e-01,
-1.1806e+00, -1.2080e+00],
[ 9.9691e-01, 1.6250e+00, -2.3868e+00, ..., -2.2128e+00,
-2.4482e+00, -2.3443e+00]],
[[-1.3978e+00, -2.3138e+00, -2.1400e+00, ..., -5.9475e-01,
-2.0187e-01, -4.6622e-01],
[-1.9896e+00, -2.1281e+00, -1.5166e+00, ..., -3.2676e-01,
8.0423e-02, 4.8663e-02],
[-2.4562e+00, -1.9095e+00, -7.6480e-01, ..., -2.8227e-01,
2.5243e-01, 2.8528e-01],
...,
[-1.5878e+00, -1.8328e+00, -8.0908e-01, ..., -5.0036e-01,
-1.7038e-01, -3.4722e-01],
[-1.7615e+00, -1.1989e+00, -8.0373e-01, ..., -3.2498e-01,
1.5726e-01, 2.2532e-01],
[-1.7191e+00, -1.0989e+00, -5.9417e-01, ..., -5.0100e-01,
-2.6972e-01, -2.4397e-01]],
...,
[[-2.3225e+00, -1.3026e+00, -2.6812e+00, ..., -3.6826e+00,
-2.8350e+00, -1.8384e+00],
[-1.8407e+00, 1.8586e-01, -1.2955e+00, ..., -2.6911e+00,
-2.4385e+00, -1.8763e+00],
[-5.5081e-01, 2.3926e+00, 8.8694e-01, ..., -5.9644e-01,
-9.2819e-02, 1.5658e-01],
...,
[-1.7249e+00, 1.1827e+00, 2.4975e-01, ..., -1.5189e+00,
-7.0239e-01, -1.4656e-01],
[-8.5674e-01, 1.6593e+00, 1.5407e+00, ..., -1.1765e-01,
1.0772e-01, 3.2592e-01],
[-2.0969e+00, -1.0723e+00, -8.4671e-01, ..., -2.3691e+00,
-1.4719e+00, -9.2993e-01]],
[[-2.3909e+00, -2.6422e+00, -3.9315e+00, ..., -1.1493e+00,
-7.4809e-01, -7.5987e-01],
[-1.7097e+00, -1.3916e+00, -4.6321e+00, ..., -1.3623e+00,
-1.2196e+00, -1.6026e+00],
[-3.0500e+00, -2.5056e+00, -6.7822e+00, ..., -3.2812e+00,
-2.7990e+00, -2.2578e+00],
...,
[-2.4908e+00, -1.7169e+00, -4.9680e+00, ..., -1.4518e+00,
-1.0127e+00, -8.5572e-01],
[-2.2002e+00, -1.5368e+00, -3.8823e+00, ..., -1.7225e+00,
-1.1343e+00, -7.3992e-01],
[-2.3009e+00, -1.7625e+00, -3.0841e+00, ..., -2.8060e+00,
-2.0160e+00, -1.8331e+00]],
[[-1.8812e+00, -2.6026e+00, -2.4116e+00, ..., -1.8032e+00,
-1.5057e+00, -1.1413e+00],
[-1.9056e+00, -2.6149e+00, -2.1132e+00, ..., -1.4013e+00,
-1.1171e+00, -8.8869e-01],
[-2.4162e+00, -2.9779e+00, -2.5311e+00, ..., -1.6036e+00,
-1.2063e+00, -8.9018e-01],
...,
[-3.1017e+00, -3.8743e+00, -3.6548e+00, ..., -2.8683e+00,
-2.2898e+00, -1.7662e+00],
[-2.0622e+00, -2.6629e+00, -2.6283e+00, ..., -1.9745e+00,
-1.6179e+00, -1.3105e+00],
[-1.4805e+00, -1.7952e+00, -1.8440e+00, ..., -1.2205e+00,
-1.0302e+00, -7.9182e-01]]],
[[[ 3.4636e-01, 8.3255e-01, 7.6588e-01, ..., 6.9501e-01,
9.2862e-01, 7.2740e-01],
[-1.1967e+00, -2.4493e-01, -1.3121e+00, ..., -1.0216e+00,
5.7492e-02, 5.5146e-01],
[ 6.6995e-01, 4.5019e-01, 5.4273e-01, ..., 4.1016e-01,
9.4762e-01, 8.6267e-01],
...,
[-1.9755e-01, -1.0799e+00, -2.4702e+00, ..., -1.6585e+00,
-1.6153e+00, -1.0467e+00],
[ 1.3076e+00, 7.0831e-01, -1.0392e+00, ..., -3.1029e-01,
2.8335e-02, 3.1311e-01],
[ 7.7513e-01, 1.9015e-01, -7.5038e-01, ..., -1.9325e-01,
1.8168e-01, 3.5588e-01]],
[[-2.3233e+00, -1.1848e+00, -4.2266e+00, ..., -5.2868e+00,
-4.4836e+00, -3.5301e+00],
[-1.4563e-01, 1.7263e+00, -2.5886e+00, ..., -4.1343e+00,
-3.9716e+00, -3.2815e+00],
[ 7.0596e-01, 1.5473e+00, -5.1546e+00, ..., -6.3813e+00,
-5.7402e+00, -5.3225e+00],
...,
[-1.5523e+00, -2.5329e-01, -6.9717e+00, ..., -7.6491e+00,
-6.9181e+00, -5.4050e+00],
[ 1.3348e+00, 2.3101e+00, -1.5901e+00, ..., -2.3890e+00,
-2.7741e+00, -2.2050e+00],
[ 8.9536e-01, 1.4965e+00, -2.5184e+00, ..., -3.1532e+00,
-3.5668e+00, -3.1093e+00]],
[[-6.6233e-01, -1.8156e+00, -1.6730e+00, ..., -9.1734e-01,
-6.6334e-01, -7.8164e-01],
[-1.3863e+00, -1.8583e+00, -1.2489e+00, ..., -7.2730e-01,
-3.9590e-01, -2.8091e-01],
[-1.0928e+00, -1.0398e+00, -4.6559e-01, ..., -5.1184e-01,
2.7875e-02, 5.2905e-02],
...,
[-1.2131e+00, -1.5614e+00, -7.0235e-01, ..., -9.0431e-01,
-7.8953e-01, -1.0683e+00],
[-1.5313e+00, -1.0309e+00, -7.5802e-01, ..., -5.5619e-01,
-2.7687e-01, -3.4978e-01],
[-1.6680e+00, -1.2008e+00, -8.9447e-01, ..., -9.4697e-01,
-8.2473e-01, -8.1528e-01]],
...,
[[-2.2072e+00, -1.1853e+00, -2.2591e+00, ..., -2.8539e+00,
-2.2419e+00, -1.4814e+00],
[-6.1773e-01, 1.1666e+00, -4.0889e-01, ..., -1.3187e+00,
-1.3723e+00, -1.1775e+00],
[-3.1941e-01, 2.2191e+00, 8.4589e-01, ..., -4.4856e-01,
2.2542e-01, 3.3238e-01],
...,
[-1.7340e+00, 1.0235e+00, -7.8454e-02, ..., -1.1723e+00,
-5.0813e-01, 6.0399e-02],
[-7.7116e-01, 1.7443e+00, 1.5240e+00, ..., 4.7554e-01,
4.7259e-01, 7.4744e-01],
[-2.1843e+00, -1.1296e+00, -9.0638e-01, ..., -2.0722e+00,
-1.0649e+00, -3.5010e-01]],
[[-1.3585e+00, -1.2500e+00, -2.7766e+00, ..., -1.1315e+00,
-6.0296e-01, -4.6195e-01],
[-6.5393e-01, -1.3415e-01, -3.6377e+00, ..., -1.7766e+00,
-1.5213e+00, -1.6930e+00],
[-2.0658e+00, -1.3215e+00, -5.9638e+00, ..., -3.3425e+00,
-3.2088e+00, -2.5633e+00],
...,
[-2.5395e+00, -1.8294e+00, -5.2636e+00, ..., -3.0148e+00,
-2.6280e+00, -2.0812e+00],
[-2.2583e+00, -1.6726e+00, -4.1308e+00, ..., -2.6648e+00,
-2.2470e+00, -1.5541e+00],
[-2.3481e+00, -1.8737e+00, -3.2459e+00, ..., -3.0308e+00,
-2.4551e+00, -2.1498e+00]],
[[-1.1632e+00, -1.7193e+00, -1.4920e+00, ..., -1.6378e+00,
-1.3724e+00, -1.0266e+00],
[-1.1794e+00, -1.6503e+00, -1.1662e+00, ..., -1.3832e+00,
-1.1213e+00, -8.7978e-01],
[-1.5365e+00, -1.8269e+00, -1.3576e+00, ..., -1.6753e+00,
-1.1961e+00, -8.6505e-01],
...,
[-3.0815e+00, -3.8621e+00, -3.7404e+00, ..., -4.0624e+00,
-3.2664e+00, -2.5174e+00],
[-2.0529e+00, -2.6590e+00, -2.6901e+00, ..., -2.8320e+00,
-2.3452e+00, -1.9129e+00],
[-1.4981e+00, -1.8396e+00, -1.9394e+00, ..., -1.8958e+00,
-1.5467e+00, -1.1829e+00]]],
[[[ 1.5548e-01, 3.9085e-01, -1.1665e+00, ..., 4.6876e-01,
1.0066e+00, 6.5089e-01],
[-2.0899e+00, -1.4216e+00, -4.3635e+00, ..., -2.5456e+00,
-7.4480e-01, -8.4349e-02],
[ 4.5697e-01, -2.5795e-01, -1.1129e+00, ..., -2.2716e-01,
4.5910e-01, 6.3341e-01],
...,
[-2.8558e-01, -1.2762e+00, -2.5965e+00, ..., -1.9832e+00,
-1.3631e+00, -8.8784e-01],
[ 1.2345e+00, 6.4162e-01, -8.9683e-01, ..., -3.2255e-01,
5.3044e-01, 7.2505e-01],
[ 8.0672e-01, 2.9855e-01, -4.3672e-01, ..., -3.8088e-01,
2.9312e-01, 2.8254e-01]],
[[-3.2557e+00, -1.4758e+00, -6.1423e+00, ..., -4.8493e+00,
-4.4216e+00, -3.3653e+00],
[-5.5336e-01, 2.1519e+00, -4.1572e+00, ..., -3.4019e+00,
-3.5434e+00, -2.9267e+00],
[ 6.0438e-02, 1.8475e+00, -7.3865e+00, ..., -5.3888e+00,
-5.2713e+00, -4.9525e+00],
...,
[-1.2352e+00, 2.8604e-01, -6.4834e+00, ..., -4.5446e+00,
-4.1508e+00, -3.4750e+00],
[ 1.5417e+00, 2.6103e+00, -1.3289e+00, ..., -9.0136e-01,
-1.2569e+00, -1.2944e+00],
[ 1.0040e+00, 1.6261e+00, -2.4011e+00, ..., -2.2608e+00,
-2.5013e+00, -2.3948e+00]],
[[-1.3903e+00, -2.3674e+00, -2.1864e+00, ..., -5.9978e-01,
-1.9650e-01, -4.6440e-01],
[-2.0266e+00, -2.2041e+00, -1.5614e+00, ..., -3.3074e-01,
8.2070e-02, 4.6064e-02],
[-2.5228e+00, -1.9927e+00, -8.2101e-01, ..., -2.6952e-01,
2.6417e-01, 2.9083e-01],
...,
[-1.5793e+00, -1.8228e+00, -7.6730e-01, ..., -4.8711e-01,
-1.5773e-01, -3.4424e-01],
[-1.7570e+00, -1.1906e+00, -7.6993e-01, ..., -3.3366e-01,
1.4639e-01, 2.1562e-01],
[-1.7336e+00, -1.1311e+00, -6.0960e-01, ..., -5.2387e-01,
-2.9722e-01, -2.5816e-01]],
...,
[[-2.4048e+00, -1.3465e+00, -2.7634e+00, ..., -3.6710e+00,
-2.7736e+00, -1.7740e+00],
[-1.8896e+00, 2.0283e-01, -1.3110e+00, ..., -2.7209e+00,
-2.3946e+00, -1.8082e+00],
[-5.2520e-01, 2.4531e+00, 9.2902e-01, ..., -6.6134e-01,
-5.2713e-02, 2.1282e-01],
...,
[-1.7278e+00, 1.1787e+00, 2.3058e-01, ..., -1.5737e+00,
-7.4460e-01, -1.6094e-01],
[-8.5688e-01, 1.6687e+00, 1.4873e+00, ..., -1.4420e-01,
8.4548e-02, 2.9796e-01],
[-2.0717e+00, -1.0221e+00, -8.0490e-01, ..., -2.3077e+00,
-1.4184e+00, -8.9495e-01]],
[[-2.3684e+00, -2.6225e+00, -4.0030e+00, ..., -1.1642e+00,
-7.4028e-01, -7.4745e-01],
[-1.6393e+00, -1.3095e+00, -4.7513e+00, ..., -1.4199e+00,
-1.2359e+00, -1.5867e+00],
[-3.0109e+00, -2.4436e+00, -6.9304e+00, ..., -3.3506e+00,
-2.8023e+00, -2.2272e+00],
...,
[-2.5123e+00, -1.7124e+00, -4.9925e+00, ..., -1.4838e+00,
-1.0673e+00, -9.1643e-01],
[-2.1995e+00, -1.5340e+00, -3.8901e+00, ..., -1.7378e+00,
-1.1801e+00, -7.8675e-01],
[-2.2916e+00, -1.7482e+00, -3.0438e+00, ..., -2.7946e+00,
-2.0295e+00, -1.8418e+00]],
[[-1.8883e+00, -2.6217e+00, -2.3990e+00, ..., -1.8189e+00,
-1.5132e+00, -1.1510e+00],
[-1.9091e+00, -2.6229e+00, -2.0840e+00, ..., -1.4143e+00,
-1.1336e+00, -9.0486e-01],
[-2.4287e+00, -2.9796e+00, -2.4934e+00, ..., -1.6213e+00,
-1.2269e+00, -9.1058e-01],
...,
[-3.0956e+00, -3.8623e+00, -3.6320e+00, ..., -2.8659e+00,
-2.2877e+00, -1.7614e+00],
[-2.0586e+00, -2.6641e+00, -2.6257e+00, ..., -1.9867e+00,
-1.6262e+00, -1.3136e+00],
[-1.4823e+00, -1.8055e+00, -1.8585e+00, ..., -1.2347e+00,
-1.0434e+00, -7.9406e-01]]],
[[[ 6.2953e-02, 1.9631e-01, -1.4334e+00, ..., -6.3925e-01,
-2.1518e-01, -7.4047e-02],
[-2.1144e+00, -1.6396e+00, -4.5301e+00, ..., -3.8311e+00,
-1.5654e+00, -5.9105e-01],
[ 5.5653e-01, -9.9553e-02, -8.4732e-01, ..., -2.2358e-01,
-2.8788e-01, 2.0273e-01],
...,
[-9.1295e-01, -1.9936e+00, -2.1154e+00, ..., -1.6256e+00,
-1.1948e+00, -8.6065e-01],
[ 8.5896e-01, 2.5967e-02, -1.8420e-01, ..., 2.2099e-02,
6.9938e-01, 7.9097e-01],
[ 2.1982e-01, -5.3419e-01, -5.1482e-01, ..., -1.3389e-01,
4.2764e-01, 3.4247e-01]],
[[-3.3062e+00, -1.7977e+00, -6.2370e+00, ..., -7.1491e+00,
-6.2674e+00, -4.6119e+00],
[-7.9242e-01, 1.5357e+00, -4.3514e+00, ..., -5.4663e+00,
-5.5421e+00, -4.2568e+00],
[-3.5801e-01, 1.0729e+00, -7.4544e+00, ..., -7.9589e+00,
-7.8254e+00, -6.6564e+00],
...,
[ 5.2020e-01, 7.6601e-01, -3.8136e+00, ..., -4.7049e+00,
-4.2348e+00, -3.5059e+00],
[ 2.7733e+00, 2.8407e+00, 4.3480e-01, ..., -1.0495e+00,
-1.3434e+00, -1.3393e+00],
[ 1.6679e+00, 1.5296e+00, -1.5228e+00, ..., -2.3187e+00,
-2.5354e+00, -2.3944e+00]],
[[-1.3483e+00, -2.3609e+00, -2.3258e+00, ..., -1.7696e+00,
-1.2452e+00, -1.3823e+00],
[-1.6818e+00, -1.9711e+00, -1.5058e+00, ..., -1.2659e+00,
-7.2513e-01, -8.1524e-01],
[-2.1522e+00, -1.9180e+00, -1.0035e+00, ..., -1.1046e+00,
-4.5091e-01, -1.8954e-01],
...,
[-3.0625e-01, -8.1236e-01, -5.1630e-01, ..., -4.7511e-01,
-1.7160e-01, -4.3205e-01],
[-8.9817e-01, -3.3348e-01, -4.9502e-01, ..., -1.9364e-01,
2.5228e-01, 2.3856e-01],
[-1.0066e+00, -3.0713e-01, -3.8681e-01, ..., -3.9845e-01,
-1.8119e-01, -2.2261e-01]],
...,
[[-2.0479e+00, -1.1062e+00, -2.3569e+00, ..., -3.1849e+00,
-2.6067e+00, -1.8078e+00],
[-1.7567e+00, 3.7452e-02, -1.3593e+00, ..., -2.4423e+00,
-2.5094e+00, -1.9583e+00],
[-3.8515e-01, 2.3735e+00, 1.0940e+00, ..., 1.5833e-01,
3.0080e-01, 5.5840e-01],
...,
[-2.2088e+00, 1.3991e-01, -1.1521e-01, ..., -1.3436e+00,
-4.0091e-01, 8.0486e-02],
[-1.2174e+00, 1.0542e+00, 1.1212e+00, ..., -1.3833e-02,
2.6786e-01, 4.3708e-01],
[-2.2005e+00, -1.4522e+00, -1.6913e+00, ..., -2.3504e+00,
-1.4058e+00, -9.0496e-01]],
[[-2.4162e+00, -2.8219e+00, -4.1839e+00, ..., -2.3793e+00,
-1.8901e+00, -1.6969e+00],
[-1.7579e+00, -1.6274e+00, -4.8736e+00, ..., -2.9065e+00,
-2.7635e+00, -2.7539e+00],
[-2.8217e+00, -2.4539e+00, -6.6692e+00, ..., -4.4588e+00,
-4.3478e+00, -3.4704e+00],
...,
[-1.3032e+00, 2.5818e-01, -3.4736e+00, ..., -1.5075e+00,
-1.0889e+00, -9.6716e-01],
[-1.6614e+00, -4.2285e-01, -3.1076e+00, ..., -1.7847e+00,
-1.2157e+00, -8.6391e-01],
[-2.3831e+00, -1.4069e+00, -3.1349e+00, ..., -2.9381e+00,
-2.1296e+00, -1.9745e+00]],
[[-1.8113e+00, -2.5466e+00, -2.4596e+00, ..., -2.7061e+00,
-2.2245e+00, -1.6989e+00],
[-1.8066e+00, -2.5233e+00, -2.1626e+00, ..., -2.4118e+00,
-1.9147e+00, -1.5453e+00],
[-2.2575e+00, -2.8232e+00, -2.5286e+00, ..., -2.7668e+00,
-2.1622e+00, -1.6342e+00],
...,
[-2.5115e+00, -3.1135e+00, -2.8119e+00, ..., -2.8245e+00,
-2.2592e+00, -1.7283e+00],
[-1.6213e+00, -2.1115e+00, -1.9870e+00, ..., -1.9226e+00,
-1.5785e+00, -1.2662e+00],
[-1.0445e+00, -1.3451e+00, -1.3524e+00, ..., -1.1815e+00,
-1.0028e+00, -7.5467e-01]]]], device='cuda:0'),
tensor([[[[ 5.0207e-01, 9.4559e-01, 5.8923e-01, ..., 7.3369e-01,
9.8247e-01, 7.9415e-01],
[-1.2051e+00, -2.5784e-01, -1.8238e+00, ..., -1.1783e+00,
6.3896e-02, 6.1913e-01],
[ 7.9397e-01, 4.6320e-01, 5.3103e-02, ..., 4.4974e-01,
7.9154e-01, 8.5487e-01],
...,
[-7.1517e-01, -1.6433e+00, -1.7377e+00, ..., -1.6539e+00,
-1.2044e+00, -8.7639e-01],
[ 8.3969e-01, 1.5738e-01, 2.3224e-03, ..., 1.1494e-01,
7.6811e-01, 8.3436e-01],
[ 2.4997e-01, -3.8100e-01, -3.2092e-01, ..., -5.6886e-03,
5.1336e-01, 4.1924e-01]],
[[-2.7357e+00, -1.3135e+00, -4.9266e+00, ..., -5.8635e+00,
-5.0171e+00, -3.9190e+00],
[-4.7087e-01, 1.8592e+00, -3.2771e+00, ..., -4.8600e+00,
-4.6506e+00, -3.7848e+00],
[ 3.4189e-01, 1.7151e+00, -6.2332e+00, ..., -7.0002e+00,
-6.6718e+00, -6.0066e+00],
...,
[ 1.7078e-01, 3.5292e-01, -4.0652e+00, ..., -4.7375e+00,
-4.2827e+00, -3.5556e+00],
[ 2.2590e+00, 2.3159e+00, -1.1117e-01, ..., -8.6509e-01,
-1.2076e+00, -1.2564e+00],
[ 1.3240e+00, 1.2051e+00, -1.7415e+00, ..., -2.2678e+00,
-2.5335e+00, -2.4323e+00]],
[[-6.4150e-01, -1.7903e+00, -1.5916e+00, ..., -7.3563e-01,
-5.1495e-01, -7.3005e-01],
[-1.5091e+00, -1.9356e+00, -1.2530e+00, ..., -6.8276e-01,
-3.8285e-01, -3.4783e-01],
[-1.4032e+00, -1.2729e+00, -5.9270e-01, ..., -4.8015e-01,
-1.4163e-01, -1.4291e-01],
...,
[-2.5483e-01, -7.2648e-01, -4.8977e-01, ..., -5.0476e-01,
-2.2995e-01, -4.5769e-01],
[-7.6344e-01, -2.6934e-01, -4.2198e-01, ..., -1.8468e-01,
2.3640e-01, 2.4718e-01],
[-8.4678e-01, -2.3973e-01, -3.1985e-01, ..., -3.5533e-01,
-1.7766e-01, -1.9780e-01]],
...,
[[-2.3251e+00, -1.2684e+00, -2.6727e+00, ..., -3.1330e+00,
-2.5923e+00, -1.7786e+00],
[-6.4725e-01, 1.2521e+00, -7.1666e-01, ..., -1.6263e+00,
-1.7876e+00, -1.5365e+00],
[-2.4328e-01, 2.5251e+00, 6.8957e-01, ..., -3.5328e-01,
-1.1358e-01, 8.5595e-02],
...,
[-1.9688e+00, 1.2234e-01, -1.1575e-01, ..., -1.3180e+00,
-3.4273e-01, 1.4807e-01],
[-1.0915e+00, 9.2294e-01, 9.5075e-01, ..., 1.7978e-02,
3.3231e-01, 4.6393e-01],
[-2.0251e+00, -1.3568e+00, -1.6071e+00, ..., -2.4428e+00,
-1.4226e+00, -8.9779e-01]],
[[-1.5207e+00, -1.4815e+00, -3.1227e+00, ..., -1.3075e+00,
-7.9785e-01, -6.4477e-01],
[-8.2704e-01, -3.5886e-01, -4.1227e+00, ..., -2.0833e+00,
-1.8398e+00, -1.9577e+00],
[-2.4681e+00, -1.8272e+00, -6.8233e+00, ..., -4.2456e+00,
-3.7337e+00, -2.9510e+00],
...,
[-1.4253e+00, -1.1893e-01, -3.4814e+00, ..., -1.5499e+00,
-1.1079e+00, -9.6623e-01],
[-1.6942e+00, -6.6091e-01, -3.0903e+00, ..., -1.7909e+00,
-1.1897e+00, -8.0235e-01],
[-2.2879e+00, -1.4793e+00, -3.0859e+00, ..., -2.9520e+00,
-2.1080e+00, -1.9278e+00]],
[[-1.3423e+00, -1.9699e+00, -1.7311e+00, ..., -1.8891e+00,
-1.5634e+00, -1.1679e+00],
[-1.4003e+00, -1.9457e+00, -1.4460e+00, ..., -1.7130e+00,
-1.3839e+00, -1.0840e+00],
[-1.8500e+00, -2.2189e+00, -1.7449e+00, ..., -2.0413e+00,
-1.5724e+00, -1.1609e+00],
...,
[-2.3637e+00, -2.9271e+00, -2.7062e+00, ..., -2.8460e+00,
-2.2762e+00, -1.7473e+00],
[-1.5381e+00, -1.9917e+00, -1.9209e+00, ..., -1.9353e+00,
-1.6024e+00, -1.2886e+00],
[-1.0042e+00, -1.2839e+00, -1.3202e+00, ..., -1.1815e+00,
-1.0122e+00, -7.6562e-01]]],
[[[-6.4128e+00, -8.1064e+00, -1.7061e+01, ..., 4.0576e-01,
7.3586e-01, 5.9062e-01],
[-5.0569e+00, -3.0950e+00, -2.0213e+01, ..., -2.0262e+00,
-4.3153e-01, 3.0770e-01],
[-7.1207e+00, -1.0248e+01, -2.4384e+01, ..., 9.6851e-02,
3.8013e-01, 6.1303e-01],
...,
[-6.5422e-01, -1.7585e+00, -1.9697e+00, ..., -1.2910e+00,
-9.4677e-01, -6.5623e-01],
[ 8.8862e-01, 2.8687e-01, 5.6543e-01, ..., 3.8506e-01,
8.4885e-01, 9.1405e-01],
[ 3.9278e-01, -1.4057e-01, 8.8613e-02, ..., 3.2286e-01,
6.8983e-01, 5.7179e-01]],
[[-2.8916e+00, -4.7186e+00, -1.2625e+01, ..., -6.4692e+00,
-5.5941e+00, -4.2789e+00],
[-1.6037e+00, -3.8249e+00, -1.4405e+01, ..., -5.2239e+00,
-5.1065e+00, -4.0471e+00],
[-1.9864e+00, -5.0115e+00, -1.9601e+01, ..., -7.5674e+00,
-7.3261e+00, -6.4480e+00],
...,
[ 3.0830e-01, 4.7730e-01, -3.8596e+00, ..., -5.0256e+00,
-4.4401e+00, -3.7248e+00],
[ 2.8148e+00, 2.7388e+00, 6.8592e-01, ..., -1.0610e+00,
-1.3040e+00, -1.3888e+00],
[ 1.6212e+00, 1.3461e+00, -1.2903e+00, ..., -2.3986e+00,
-2.5815e+00, -2.5077e+00]],
[[-6.6546e-01, 1.0748e-01, -1.4657e+00, ..., -9.0628e-01,
-6.0861e-01, -8.8291e-01],
[ 6.3699e-01, 3.6609e+00, 3.1941e+00, ..., -7.8915e-01,
-4.2479e-01, -4.6504e-01],
[-3.6184e-01, 5.2642e+00, 4.6721e+00, ..., -6.8573e-01,
-2.3671e-01, -1.9689e-01],
...,
[-5.7893e-01, -8.6225e-01, -3.3832e-01, ..., -4.8456e-01,
-2.6165e-01, -4.7523e-01],
[-1.0484e+00, -2.8714e-01, -1.2407e-01, ..., -1.3205e-01,
1.9603e-01, 1.9348e-01],
[-1.2767e+00, -4.7613e-01, -8.9491e-02, ..., -2.5983e-01,
-1.3154e-01, -1.6420e-01]],
...,
[[-7.7478e+00, -1.0581e+01, -1.5105e+01, ..., -3.4388e+00,
-2.8571e+00, -1.9851e+00],
[-9.8110e+00, -1.5063e+01, -1.4644e+01, ..., -2.0345e+00,
-2.1589e+00, -1.7975e+00],
[-5.5594e+00, -7.0828e+00, -7.5959e+00, ..., -3.5892e-01,
-1.1689e-01, 1.6132e-01],
...,
[-2.0751e+00, 7.7097e-01, -2.2206e-01, ..., -1.1645e+00,
-2.8079e-01, 2.0516e-01],
[-1.1528e+00, 1.2691e+00, 9.0893e-01, ..., 3.2959e-02,
3.2561e-01, 4.9141e-01],
[-2.4589e+00, -1.4900e+00, -1.8396e+00, ..., -2.3067e+00,
-1.3328e+00, -7.5070e-01]],
[[-2.1314e+00, -5.7994e+00, -7.5498e+00, ..., -1.5854e+00,
-1.0738e+00, -9.3174e-01],
[-3.5806e+00, -1.0063e+01, -1.0292e+01, ..., -2.3106e+00,
-2.1171e+00, -2.2530e+00],
[-8.5388e+00, -1.4307e+01, -1.2704e+01, ..., -4.4338e+00,
-4.0466e+00, -3.2390e+00],
...,
[-1.7724e+00, 1.5762e-01, -3.5224e+00, ..., -1.8235e+00,
-1.4002e+00, -1.1369e+00],
[-1.8281e+00, -1.0567e-01, -2.7407e+00, ..., -1.9049e+00,
-1.3281e+00, -8.8375e-01],
[-2.2730e+00, -8.1891e-01, -2.5145e+00, ..., -2.8446e+00,
-2.0312e+00, -1.8271e+00]],
[[-6.2233e+00, -6.6784e+00, -6.3787e+00, ..., -2.2237e+00,
-1.8344e+00, -1.3860e+00],
[-6.8716e+00, -7.4364e+00, -8.0213e+00, ..., -2.0239e+00,
-1.6324e+00, -1.2958e+00],
[-6.8057e+00, -9.3728e+00, -1.0735e+01, ..., -2.4074e+00,
-1.8664e+00, -1.3932e+00],
...,
[-2.6288e+00, -3.3447e+00, -3.2338e+00, ..., -2.8093e+00,
-2.2504e+00, -1.7150e+00],
[-1.7932e+00, -2.4022e+00, -2.3534e+00, ..., -1.9357e+00,
-1.5965e+00, -1.2800e+00],
[-1.2335e+00, -1.6147e+00, -1.6555e+00, ..., -1.1908e+00,
-1.0227e+00, -7.6341e-01]]],
[[[ 3.6559e-01, 8.2748e-01, 7.5663e-01, ..., 6.8719e-01,
9.6025e-01, 7.4463e-01],
[-1.2330e+00, -2.9617e-01, -1.3953e+00, ..., -1.0725e+00,
7.8960e-02, 5.7927e-01],
[ 6.7098e-01, 4.2240e-01, 4.8207e-01, ..., 5.8128e-01,
9.4563e-01, 8.7757e-01],
...,
[-1.5558e-01, -1.1049e+00, -2.6555e+00, ..., -1.5295e+00,
-1.4287e+00, -8.6155e-01],
[ 1.4070e+00, 8.5678e-01, -9.6568e-01, ..., -2.8890e-01,
1.0571e-01, 3.7435e-01],
[ 8.4314e-01, 3.1367e-01, -6.7390e-01, ..., -1.9037e-01,
2.1323e-01, 3.8100e-01]],
[[-2.3413e+00, -1.1974e+00, -4.2509e+00, ..., -5.2393e+00,
-4.5052e+00, -3.5259e+00],
[-1.0011e-01, 1.7946e+00, -2.5390e+00, ..., -4.1108e+00,
-4.0047e+00, -3.2923e+00],
[ 7.8875e-01, 1.6227e+00, -5.1198e+00, ..., -5.9901e+00,
-5.7604e+00, -5.3146e+00],
...,
[-1.3772e+00, 1.0280e-02, -6.7148e+00, ..., -7.5953e+00,
-6.9112e+00, -5.4135e+00],
[ 1.6834e+00, 2.7133e+00, -1.0514e+00, ..., -2.6138e+00,
-2.9773e+00, -2.3837e+00],
[ 1.1371e+00, 1.7548e+00, -2.2633e+00, ..., -3.2487e+00,
-3.6304e+00, -3.1268e+00]],
[[-6.8361e-01, -1.8540e+00, -1.7163e+00, ..., -9.1524e-01,
-6.5071e-01, -7.6748e-01],
[-1.4003e+00, -1.8790e+00, -1.2713e+00, ..., -7.2607e-01,
-3.8851e-01, -2.8326e-01],
[-1.0784e+00, -1.0117e+00, -4.6085e-01, ..., -3.2568e-01,
2.7344e-02, 3.6900e-02],
...,
[-1.3241e+00, -1.6986e+00, -7.5685e-01, ..., -8.8066e-01,
-7.6469e-01, -1.0080e+00],
[-1.6447e+00, -1.0796e+00, -7.9386e-01, ..., -5.6401e-01,
-2.9876e-01, -3.7372e-01],
[-1.8065e+00, -1.2487e+00, -8.9647e-01, ..., -9.5351e-01,
-8.1990e-01, -7.9901e-01]],
...,
[[-2.2736e+00, -1.2451e+00, -2.3084e+00, ..., -2.8870e+00,
-2.2839e+00, -1.5005e+00],
[-6.2469e-01, 1.1501e+00, -4.0371e-01, ..., -1.3497e+00,
-1.4157e+00, -1.2167e+00],
[-3.3801e-01, 2.2044e+00, 8.7749e-01, ..., -1.5248e-01,
2.3051e-01, 3.5114e-01],
...,
[-1.8424e+00, 1.1500e+00, -2.1688e-02, ..., -1.0713e+00,
-5.2235e-01, -3.8619e-03],
[-8.3230e-01, 1.7852e+00, 1.5448e+00, ..., 4.6835e-01,
4.0592e-01, 6.7554e-01],
[-2.2876e+00, -1.2682e+00, -1.0768e+00, ..., -1.8509e+00,
-9.6775e-01, -3.0127e-01]],
[[-1.3479e+00, -1.2284e+00, -2.7416e+00, ..., -1.0975e+00,
-5.9508e-01, -4.7584e-01],
[-6.1972e-01, -7.8833e-02, -3.5898e+00, ..., -1.7835e+00,
-1.5227e+00, -1.6835e+00],
[-2.0652e+00, -1.2976e+00, -5.9665e+00, ..., -3.7178e+00,
-3.1990e+00, -2.5381e+00],
...,
[-2.4684e+00, -1.5879e+00, -5.3615e+00, ..., -3.0834e+00,
-2.7314e+00, -2.1864e+00],
[-2.2256e+00, -1.4847e+00, -4.2106e+00, ..., -2.6699e+00,
-2.3010e+00, -1.6478e+00],
[-2.3971e+00, -1.7941e+00, -3.3013e+00, ..., -2.9458e+00,
-2.4214e+00, -2.1194e+00]],
[[-1.1752e+00, -1.7345e+00, -1.5015e+00, ..., -1.6377e+00,
-1.3589e+00, -1.0241e+00],
[-1.1792e+00, -1.6524e+00, -1.1649e+00, ..., -1.3675e+00,
-1.1094e+00, -8.8062e-01],
[-1.5357e+00, -1.8241e+00, -1.3494e+00, ..., -1.5601e+00,
-1.1891e+00, -8.7191e-01],
...,
[-3.1451e+00, -3.9345e+00, -3.7650e+00, ..., -3.9445e+00,
-3.1624e+00, -2.4442e+00],
[-2.0768e+00, -2.6938e+00, -2.6872e+00, ..., -2.7744e+00,
-2.2918e+00, -1.8659e+00],
[-1.5082e+00, -1.8385e+00, -1.9145e+00, ..., -1.8790e+00,
-1.5245e+00, -1.1722e+00]]],
[[[ 9.3715e-02, 2.4907e-01, -1.3861e+00, ..., -7.0640e-01,
-2.6271e-01, -9.5627e-02],
[-2.1850e+00, -1.6665e+00, -4.6567e+00, ..., -3.6953e+00,
-1.4795e+00, -5.5022e-01],
[ 6.3983e-01, -1.1285e-01, -8.3394e-01, ..., -1.2549e-01,
-2.3159e-01, 2.7549e-01],
...,
[-8.2802e-01, -1.7899e+00, -1.7781e+00, ..., -1.8274e+00,
-1.3455e+00, -9.4754e-01],
[ 7.5769e-01, 7.4594e-03, -6.6790e-02, ..., 1.8047e-03,
7.0287e-01, 8.0592e-01],
[ 1.5774e-01, -5.5091e-01, -4.3728e-01, ..., -1.2632e-01,
4.4396e-01, 3.6091e-01]],
[[-3.3230e+00, -1.6337e+00, -6.3039e+00, ..., -7.0588e+00,
-6.2237e+00, -4.5902e+00],
[-6.1296e-01, 1.9796e+00, -4.1998e+00, ..., -5.5798e+00,
-5.6323e+00, -4.3035e+00],
[-5.0575e-02, 1.5601e+00, -7.4374e+00, ..., -8.0476e+00,
-7.8482e+00, -6.6012e+00],
...,
[ 2.6886e-01, 3.1321e-01, -4.1035e+00, ..., -4.6788e+00,
-4.2102e+00, -3.5338e+00],
[ 2.2264e+00, 2.1988e+00, -2.7250e-01, ..., -8.4412e-01,
-1.1892e+00, -1.2547e+00],
[ 1.3035e+00, 1.1201e+00, -1.8270e+00, ..., -2.2497e+00,
-2.5116e+00, -2.4197e+00]],
[[-1.3946e+00, -2.5369e+00, -2.4735e+00, ..., -1.7543e+00,
-1.2497e+00, -1.3654e+00],
[-1.8391e+00, -2.1896e+00, -1.6308e+00, ..., -1.2248e+00,
-7.1191e-01, -8.3500e-01],
[-2.3574e+00, -2.0906e+00, -1.0274e+00, ..., -1.1102e+00,
-4.6744e-01, -1.9543e-01],
...,
[-1.6909e-01, -5.9280e-01, -4.6887e-01, ..., -5.0246e-01,
-1.8792e-01, -4.4201e-01],
[-7.1541e-01, -2.0066e-01, -4.1341e-01, ..., -2.1235e-01,
2.3876e-01, 2.3845e-01],
[-7.6530e-01, -1.5786e-01, -3.0433e-01, ..., -3.9177e-01,
-1.9105e-01, -2.2347e-01]],
...,
[[-2.2635e+00, -1.1224e+00, -2.5420e+00, ..., -2.9390e+00,
-2.3937e+00, -1.6591e+00],
[-1.9203e+00, 1.8550e-01, -1.4237e+00, ..., -2.3927e+00,
-2.3878e+00, -1.8065e+00],
[-4.3312e-01, 2.6622e+00, 1.1017e+00, ..., 9.6324e-02,
2.6954e-01, 5.2487e-01],
...,
[-2.0169e+00, -6.3707e-02, -1.9194e-01, ..., -1.3777e+00,
-4.0069e-01, 1.1206e-01],
[-1.1693e+00, 8.0497e-01, 9.1350e-01, ..., 9.9265e-03,
3.2332e-01, 4.7242e-01],
[-2.0624e+00, -1.3919e+00, -1.6369e+00, ..., -2.4566e+00,
-1.4599e+00, -9.1777e-01]],
[[-2.4433e+00, -2.8220e+00, -4.4321e+00, ..., -2.5709e+00,
-2.0563e+00, -1.8221e+00],
[-1.6897e+00, -1.4619e+00, -5.1505e+00, ..., -3.1197e+00,
-2.8977e+00, -2.7809e+00],
[-2.8066e+00, -2.2976e+00, -7.0538e+00, ..., -4.5276e+00,
-4.3700e+00, -3.4620e+00],
...,
[-1.4393e+00, -1.1612e-01, -3.3904e+00, ..., -1.4427e+00,
-1.0491e+00, -9.5954e-01],
[-1.7234e+00, -6.9299e-01, -3.0314e+00, ..., -1.7356e+00,
-1.1812e+00, -8.2089e-01],
[-2.3090e+00, -1.5089e+00, -3.1251e+00, ..., -2.9596e+00,
-2.1436e+00, -1.9776e+00]],
[[-1.8883e+00, -2.6547e+00, -2.5103e+00, ..., -2.6918e+00,
-2.2057e+00, -1.7036e+00],
[-1.8794e+00, -2.6211e+00, -2.1666e+00, ..., -2.4683e+00,
-1.9524e+00, -1.5871e+00],
[-2.3553e+00, -2.9166e+00, -2.5258e+00, ..., -2.8460e+00,
-2.2311e+00, -1.7010e+00],
...,
[-2.3786e+00, -2.9453e+00, -2.7210e+00, ..., -2.8681e+00,
-2.2923e+00, -1.7561e+00],
[-1.5500e+00, -2.0083e+00, -1.9447e+00, ..., -1.9501e+00,
-1.6054e+00, -1.2933e+00],
[-1.0121e+00, -1.2996e+00, -1.3313e+00, ..., -1.1868e+00,
-1.0188e+00, -7.6878e-01]]],
[[[ 3.3781e-01, 8.1143e-01, 7.3815e-01, ..., 6.8887e-01,
9.5154e-01, 7.4478e-01],
[-1.2341e+00, -3.0016e-01, -1.3795e+00, ..., -1.0926e+00,
6.9021e-02, 5.7662e-01],
[ 6.7159e-01, 4.5659e-01, 5.3319e-01, ..., 6.0494e-01,
9.5801e-01, 8.8832e-01],
...,
[-1.5898e-01, -1.0184e+00, -2.4490e+00, ..., -1.6336e+00,
-1.6068e+00, -1.0283e+00],
[ 1.3203e+00, 7.8829e-01, -9.4300e-01, ..., -2.8984e-01,
3.2725e-02, 3.5268e-01],
[ 7.7061e-01, 2.4364e-01, -6.8620e-01, ..., -1.5372e-01,
2.2004e-01, 3.8698e-01]],
[[-2.3749e+00, -1.3174e+00, -4.2866e+00, ..., -5.2476e+00,
-4.4863e+00, -3.5343e+00],
[-1.8152e-01, 1.6299e+00, -2.5798e+00, ..., -4.0985e+00,
-3.9748e+00, -3.3080e+00],
[ 6.6613e-01, 1.4692e+00, -5.1239e+00, ..., -5.9673e+00,
-5.7347e+00, -5.3371e+00],
...,
[-1.5196e+00, -2.3907e-01, -6.8899e+00, ..., -7.5836e+00,
-6.8309e+00, -5.3550e+00],
[ 1.3798e+00, 2.3299e+00, -1.5211e+00, ..., -2.2866e+00,
-2.6976e+00, -2.1710e+00],
[ 9.2653e-01, 1.5080e+00, -2.4756e+00, ..., -3.1487e+00,
-3.5675e+00, -3.1117e+00]],
[[-6.8766e-01, -1.8119e+00, -1.7076e+00, ..., -9.1341e-01,
-6.5033e-01, -7.6816e-01],
[-1.3624e+00, -1.8075e+00, -1.2648e+00, ..., -7.2624e-01,
-3.8566e-01, -2.7696e-01],
[-1.0372e+00, -9.4794e-01, -4.4803e-01, ..., -3.1376e-01,
2.8048e-02, 3.9989e-02],
...,
[-1.2131e+00, -1.5785e+00, -7.3019e-01, ..., -9.0119e-01,
-7.6883e-01, -1.0247e+00],
[-1.5320e+00, -1.0309e+00, -7.7952e-01, ..., -5.5825e-01,
-2.7067e-01, -3.1635e-01],
[-1.6827e+00, -1.1831e+00, -8.7512e-01, ..., -9.4691e-01,
-8.2010e-01, -7.9701e-01]],
...,
[[-2.2440e+00, -1.3044e+00, -2.2955e+00, ..., -2.9093e+00,
-2.3033e+00, -1.5160e+00],
[-6.3504e-01, 1.0649e+00, -4.0138e-01, ..., -1.3588e+00,
-1.4405e+00, -1.2555e+00],
[-4.2776e-01, 2.0283e+00, 8.2427e-01, ..., -1.6147e-01,
2.1252e-01, 3.3387e-01],
...,
[-1.7252e+00, 1.0356e+00, -6.5711e-02, ..., -1.3400e+00,
-5.7813e-01, 3.7638e-02],
[-7.9279e-01, 1.6897e+00, 1.4812e+00, ..., 3.7907e-01,
4.1647e-01, 7.0980e-01],
[-2.1531e+00, -1.1496e+00, -9.5699e-01, ..., -2.1184e+00,
-1.1115e+00, -3.6724e-01]],
[[-1.3621e+00, -1.2547e+00, -2.6621e+00, ..., -1.0851e+00,
-6.0058e-01, -4.8485e-01],
[-6.4485e-01, -1.3910e-01, -3.4930e+00, ..., -1.7589e+00,
-1.5307e+00, -1.7099e+00],
[-2.0858e+00, -1.3853e+00, -5.8478e+00, ..., -3.6984e+00,
-3.2076e+00, -2.5624e+00],
...,
[-2.5357e+00, -1.8261e+00, -5.3230e+00, ..., -2.9629e+00,
-2.5597e+00, -2.0770e+00],
[-2.2473e+00, -1.6410e+00, -4.1684e+00, ..., -2.6355e+00,
-2.2230e+00, -1.5372e+00],
[-2.3440e+00, -1.8498e+00, -3.2682e+00, ..., -3.0407e+00,
-2.4521e+00, -2.1195e+00]],
[[-1.1639e+00, -1.7179e+00, -1.5114e+00, ..., -1.6376e+00,
-1.3590e+00, -1.0177e+00],
[-1.1696e+00, -1.6351e+00, -1.1744e+00, ..., -1.3617e+00,
-1.1069e+00, -8.7374e-01],
[-1.5072e+00, -1.7960e+00, -1.3469e+00, ..., -1.5555e+00,
-1.1848e+00, -8.6502e-01],
...,
[-3.0527e+00, -3.8242e+00, -3.6913e+00, ..., -4.0461e+00,
-3.2527e+00, -2.5057e+00],
[-2.0353e+00, -2.6345e+00, -2.6540e+00, ..., -2.8026e+00,
-2.3211e+00, -1.8934e+00],
[-1.4917e+00, -1.8164e+00, -1.9073e+00, ..., -1.8588e+00,
-1.5216e+00, -1.1654e+00]]]], device='cuda:0'),
tensor([[[[ 3.3841e-01, 6.2771e-01, 7.5049e-01, ..., 5.3450e-01,
1.0276e+00, 6.2992e-01],
[-2.0011e+00, -1.2961e+00, -2.9120e+00, ..., -2.4581e+00,
-7.3838e-01, -9.2008e-02],
[-1.4730e-02, -7.7161e-01, -9.7524e-02, ..., -3.8400e-02,
6.1143e-01, 6.9269e-01],
...,
[ 9.5427e-02, -6.2470e-01, -1.5554e+00, ..., -1.2640e+00,
-1.0774e+00, -8.5433e-01],
[ 1.4459e+00, 1.1928e+00, 4.9880e-01, ..., 7.0131e-01,
8.4962e-01, 8.8229e-01],
[ 1.1039e+00, 9.6445e-01, 6.4108e-01, ..., 9.1880e-01,
1.0069e+00, 8.6036e-01]],
[[-1.7540e+00, -1.1190e+00, -4.2009e+00, ..., -4.9429e+00,
-4.4717e+00, -3.4199e+00],
[ 4.9213e-01, 1.9422e+00, -2.3198e+00, ..., -3.6180e+00,
-3.6838e+00, -3.0269e+00],
[ 1.2477e+00, 1.7105e+00, -4.9755e+00, ..., -5.5711e+00,
-5.3668e+00, -5.0115e+00],
...,
[-9.1456e-01, 1.6973e-01, -5.1517e+00, ..., -6.0108e+00,
-5.3596e+00, -4.2712e+00],
[ 1.7466e+00, 2.4259e+00, -4.4186e-01, ..., -1.3094e+00,
-1.5694e+00, -1.5202e+00],
[ 1.0070e+00, 1.3024e+00, -1.9276e+00, ..., -2.6451e+00,
-2.8606e+00, -2.7487e+00]],
[[-1.7436e-01, -1.2388e+00, -1.3525e+00, ..., -5.2536e-01,
-2.0255e-01, -5.0319e-01],
[-7.0088e-01, -9.9142e-01, -8.2479e-01, ..., -3.3534e-01,
-2.1804e-02, -1.0185e-01],
[-6.3671e-01, -4.5498e-01, -4.0319e-01, ..., -3.6020e-01,
5.6365e-02, 5.9659e-02],
...,
[-9.9102e-01, -1.4516e+00, -6.6840e-01, ..., -7.2466e-01,
-5.4307e-01, -7.2233e-01],
[-1.2036e+00, -8.1681e-01, -4.8716e-01, ..., -1.7989e-01,
6.6152e-02, 4.0631e-02],
[-1.2060e+00, -6.0391e-01, -1.5455e-01, ..., -7.1905e-02,
-4.2409e-02, -8.2832e-02]],
...,
[[-2.9677e+00, -2.3476e+00, -2.7607e+00, ..., -3.4499e+00,
-2.6444e+00, -1.7599e+00],
[-2.0870e+00, -5.6664e-01, -1.4800e+00, ..., -2.5124e+00,
-2.2743e+00, -1.7898e+00],
[-9.4993e-01, 1.3557e+00, 6.0095e-01, ..., -4.2998e-01,
-7.1192e-03, 1.4020e-01],
...,
[-1.4801e+00, 1.4507e+00, 3.5104e-01, ..., -8.9257e-01,
-1.2991e-01, 4.5754e-01],
[-6.1813e-01, 1.6041e+00, 1.0818e+00, ..., 2.1044e-01,
4.1327e-01, 5.9619e-01],
[-1.7022e+00, -1.0042e+00, -1.3057e+00, ..., -2.3112e+00,
-1.2506e+00, -5.5225e-01]],
[[-1.5519e+00, -1.5362e+00, -2.8872e+00, ..., -1.2148e+00,
-7.7654e-01, -7.7080e-01],
[-5.8072e-01, 6.8936e-02, -3.2688e+00, ..., -1.3967e+00,
-1.2260e+00, -1.5644e+00],
[-2.0778e+00, -9.5670e-01, -5.4368e+00, ..., -3.2574e+00,
-2.7631e+00, -2.2325e+00],
...,
[-1.9207e+00, -1.0261e+00, -4.6560e+00, ..., -2.4272e+00,
-2.0091e+00, -1.5839e+00],
[-1.7608e+00, -1.0013e+00, -3.6936e+00, ..., -2.2571e+00,
-1.6484e+00, -1.0391e+00],
[-2.0512e+00, -1.4165e+00, -2.9567e+00, ..., -2.8233e+00,
-1.9925e+00, -1.7271e+00]],
[[-1.3270e+00, -1.8889e+00, -1.6767e+00, ..., -1.7519e+00,
-1.4488e+00, -1.0966e+00],
[-1.2592e+00, -1.7460e+00, -1.2223e+00, ..., -1.3636e+00,
-1.0786e+00, -8.5554e-01],
[-1.5870e+00, -1.9300e+00, -1.4634e+00, ..., -1.5915e+00,
-1.1924e+00, -8.7543e-01],
...,
[-2.4593e+00, -3.0812e+00, -2.8956e+00, ..., -3.2066e+00,
-2.5681e+00, -1.9659e+00],
[-1.6492e+00, -2.1058e+00, -2.0607e+00, ..., -2.2060e+00,
-1.8182e+00, -1.4622e+00],
[-1.1122e+00, -1.3465e+00, -1.4143e+00, ..., -1.3455e+00,
-1.1519e+00, -8.5497e-01]]],
[[[-2.4575e+00, -3.5587e+00, -2.5447e+00, ..., 5.1016e-01,
1.0005e+00, 5.7528e-01],
[-4.6583e+00, -6.3514e+00, -8.0552e+00, ..., -2.6336e+00,
-8.8374e-01, -2.0155e-01],
[-2.1727e+00, -7.5352e+00, -5.8251e+00, ..., 4.9108e-02,
6.0948e-01, 6.7038e-01],
...,
[-3.0185e-01, -1.2689e+00, -2.6836e+00, ..., -1.5401e+00,
-1.4523e+00, -8.7128e-01],
[ 1.3434e+00, 7.3208e-01, -1.0381e+00, ..., -3.4104e-01,
2.6327e-02, 3.3182e-01],
[ 7.7121e-01, 1.7555e-01, -7.7450e-01, ..., -2.4459e-01,
1.6415e-01, 3.5716e-01]],
[[-1.8787e+00, -2.9105e+00, -7.8243e+00, ..., -5.0867e+00,
-4.5940e+00, -3.4957e+00],
[ 2.5099e-01, 7.6618e-03, -7.1283e+00, ..., -3.5607e+00,
-3.6421e+00, -2.9937e+00],
[ 1.0173e+00, 9.0975e-01, -8.9780e+00, ..., -5.5052e+00,
-5.3546e+00, -5.0526e+00],
...,
[-1.3437e+00, 9.4227e-02, -6.8370e+00, ..., -7.7405e+00,
-7.0398e+00, -5.4940e+00],
[ 1.7416e+00, 2.7750e+00, -1.1144e+00, ..., -2.6444e+00,
-3.0224e+00, -2.4035e+00],
[ 1.1792e+00, 1.7938e+00, -2.3072e+00, ..., -3.2834e+00,
-3.6728e+00, -3.1581e+00]],
[[ 6.0070e-02, -5.2067e-01, -8.2720e-01, ..., -5.9161e-01,
-2.3690e-01, -5.5977e-01],
[-6.0082e-01, -5.0591e-01, 4.3178e-01, ..., -3.5870e-01,
-2.0312e-02, -8.9449e-02],
[-6.3731e-01, 1.4026e-01, 8.3706e-01, ..., -3.2056e-01,
1.2900e-01, 1.4117e-01],
...,
[-1.4334e+00, -1.7345e+00, -7.0239e-01, ..., -8.8431e-01,
-7.7810e-01, -1.0411e+00],
[-1.7845e+00, -1.1916e+00, -8.1439e-01, ..., -5.5812e-01,
-2.8931e-01, -3.5944e-01],
[-1.9493e+00, -1.3847e+00, -9.5698e-01, ..., -1.0133e+00,
-8.8526e-01, -8.5602e-01]],
...,
[[-2.8755e+00, -5.1242e+00, -5.4144e+00, ..., -3.6350e+00,
-2.6919e+00, -1.7486e+00],
[-2.1999e+00, -3.3702e+00, -7.1084e+00, ..., -2.6608e+00,
-2.2700e+00, -1.7891e+00],
[-9.6018e-01, -1.7881e+00, -7.5998e+00, ..., -5.8398e-01,
7.6249e-02, 2.1821e-01],
...,
[-1.8065e+00, 1.2251e+00, -4.1973e-02, ..., -1.1102e+00,
-5.4307e-01, -3.9905e-03],
[-7.6533e-01, 1.9551e+00, 1.6171e+00, ..., 5.1292e-01,
4.5263e-01, 6.9958e-01],
[-2.1896e+00, -1.0876e+00, -9.7679e-01, ..., -1.8534e+00,
-9.4821e-01, -2.7884e-01]],
[[-1.1674e+00, -4.8691e-01, -1.8811e+00, ..., -1.2451e+00,
-7.2356e-01, -6.9873e-01],
[-1.8895e-01, 1.5060e+00, -2.0863e+00, ..., -1.3762e+00,
-1.1514e+00, -1.5392e+00],
[-1.6015e+00, 7.7197e-02, -3.3991e+00, ..., -3.3075e+00,
-2.7635e+00, -2.2427e+00],
...,
[-2.4681e+00, -1.5662e+00, -5.4389e+00, ..., -3.0970e+00,
-2.7485e+00, -2.1914e+00],
[-2.1971e+00, -1.4230e+00, -4.2067e+00, ..., -2.6996e+00,
-2.3451e+00, -1.6669e+00],
[-2.3637e+00, -1.7358e+00, -3.2719e+00, ..., -3.0015e+00,
-2.4766e+00, -2.1497e+00]],
[[-1.5191e+00, -2.5450e+00, -2.9347e+00, ..., -1.8474e+00,
-1.5446e+00, -1.1568e+00],
[-1.5891e+00, -2.6828e+00, -2.7001e+00, ..., -1.4167e+00,
-1.1427e+00, -8.9544e-01],
[-1.9219e+00, -3.0800e+00, -3.2491e+00, ..., -1.6206e+00,
-1.2380e+00, -8.9766e-01],
...,
[-3.2520e+00, -4.0759e+00, -3.8612e+00, ..., -4.0190e+00,
-3.2254e+00, -2.4880e+00],
[-2.1696e+00, -2.8254e+00, -2.7805e+00, ..., -2.8109e+00,
-2.3222e+00, -1.8887e+00],
[-1.5805e+00, -1.9377e+00, -1.9950e+00, ..., -1.8960e+00,
-1.5384e+00, -1.1806e+00]]],
[[[ 1.0637e-01, 3.4708e-01, -1.1744e+00, ..., 6.3283e-01,
9.1868e-01, 6.7082e-01],
[-1.9796e+00, -1.3473e+00, -4.1154e+00, ..., -1.5343e+00,
-1.9652e-01, 3.8207e-01],
[ 3.2795e-01, -2.8926e-01, -1.0877e+00, ..., 5.2644e-01,
9.3599e-01, 8.5831e-01],
...,
[-2.7810e-01, -1.3458e+00, -2.7701e+00, ..., -1.3346e+00,
-8.1364e-01, -6.4429e-01],
[ 1.3753e+00, 7.4102e-01, -8.9981e-01, ..., 3.7744e-01,
7.9714e-01, 7.9926e-01],
[ 9.3317e-01, 3.9584e-01, -4.0460e-01, ..., 4.3869e-01,
8.0457e-01, 6.6701e-01]],
[[-3.1400e+00, -1.5298e+00, -6.0887e+00, ..., -5.4575e+00,
-4.6696e+00, -3.6575e+00],
[-7.4253e-01, 1.6787e+00, -4.3921e+00, ..., -4.1771e+00,
-4.0476e+00, -3.3564e+00],
[-2.5984e-01, 1.3043e+00, -7.4386e+00, ..., -6.1312e+00,
-5.8726e+00, -5.4811e+00],
...,
[-1.1005e+00, 5.7239e-01, -6.2698e+00, ..., -5.3294e+00,
-4.7198e+00, -3.8786e+00],
[ 1.9134e+00, 3.1078e+00, -7.3627e-01, ..., -1.2630e+00,
-1.4245e+00, -1.4757e+00],
[ 1.2614e+00, 1.9474e+00, -2.1337e+00, ..., -2.5178e+00,
-2.6267e+00, -2.5605e+00]],
[[-1.2770e+00, -2.1519e+00, -2.0020e+00, ..., -8.4345e-01,
-6.4106e-01, -8.2952e-01],
[-1.8243e+00, -1.9879e+00, -1.4572e+00, ..., -6.8275e-01,
-4.3071e-01, -3.7939e-01],
[-2.3703e+00, -1.8590e+00, -8.1197e-01, ..., -3.9040e-01,
-1.0466e-01, -1.0699e-01],
...,
[-1.6830e+00, -1.9656e+00, -8.1492e-01, ..., -6.1236e-01,
-4.9485e-01, -6.9352e-01],
[-1.8918e+00, -1.2624e+00, -7.8822e-01, ..., -2.3065e-01,
-4.2995e-02, -5.8166e-02],
[-1.8747e+00, -1.1840e+00, -5.9246e-01, ..., -3.6243e-01,
-3.0099e-01, -2.6800e-01]],
...,
[[-2.0904e+00, -1.1851e+00, -2.5294e+00, ..., -3.0942e+00,
-2.4559e+00, -1.6396e+00],
[-1.7937e+00, 8.6692e-02, -1.2718e+00, ..., -1.5178e+00,
-1.5556e+00, -1.3305e+00],
[-4.6378e-01, 2.2298e+00, 8.6941e-01, ..., -7.1889e-02,
2.8617e-01, 3.6757e-01],
...,
[-1.9303e+00, 1.3147e+00, 2.8256e-01, ..., -9.0325e-01,
-3.4425e-01, 1.8937e-01],
[-9.6746e-01, 1.7677e+00, 1.6038e+00, ..., 2.1697e-01,
2.7932e-01, 4.1565e-01],
[-2.2813e+00, -1.2490e+00, -1.0807e+00, ..., -2.0988e+00,
-1.2689e+00, -6.4642e-01]],
[[-2.3720e+00, -2.6820e+00, -3.9045e+00, ..., -1.1151e+00,
-5.9885e-01, -4.9067e-01],
[-1.7837e+00, -1.5652e+00, -4.6023e+00, ..., -1.6859e+00,
-1.4732e+00, -1.7239e+00],
[-3.0233e+00, -2.5775e+00, -6.5312e+00, ..., -3.7239e+00,
-3.2625e+00, -2.6431e+00],
...,
[-2.4588e+00, -1.4912e+00, -5.0199e+00, ..., -2.1240e+00,
-1.6923e+00, -1.4451e+00],
[-2.2068e+00, -1.3844e+00, -3.9464e+00, ..., -2.0546e+00,
-1.4434e+00, -9.9871e-01],
[-2.3837e+00, -1.7158e+00, -3.1235e+00, ..., -2.7561e+00,
-1.9225e+00, -1.6612e+00]],
[[-1.8369e+00, -2.5237e+00, -2.3615e+00, ..., -1.7084e+00,
-1.4169e+00, -1.0591e+00],
[-1.8831e+00, -2.5612e+00, -2.1313e+00, ..., -1.3821e+00,
-1.1251e+00, -8.9104e-01],
[-2.4105e+00, -2.9511e+00, -2.5839e+00, ..., -1.5862e+00,
-1.2180e+00, -8.8493e-01],
...,
[-3.1892e+00, -3.9992e+00, -3.7179e+00, ..., -2.8127e+00,
-2.2405e+00, -1.6962e+00],
[-2.1051e+00, -2.7337e+00, -2.6589e+00, ..., -1.9850e+00,
-1.6307e+00, -1.3028e+00],
[-1.5013e+00, -1.8303e+00, -1.8559e+00, ..., -1.2467e+00,
-1.0635e+00, -7.9139e-01]]],
[[[ 1.6104e-01, 3.9529e-01, -1.1653e+00, ..., 4.7190e-01,
1.0017e+00, 6.4105e-01],
[-2.0619e+00, -1.3989e+00, -4.3485e+00, ..., -2.4572e+00,
-7.1117e-01, -5.3122e-02],
[ 4.6861e-01, -2.3125e-01, -1.0780e+00, ..., -2.1913e-01,
4.8941e-01, 6.4070e-01],
...,
[-2.7052e-01, -1.2696e+00, -2.6173e+00, ..., -2.1565e+00,
-1.5091e+00, -1.0244e+00],
[ 1.2445e+00, 6.6999e-01, -9.1627e-01, ..., -3.0381e-01,
5.4495e-01, 7.2480e-01],
[ 8.1181e-01, 3.1601e-01, -4.2843e-01, ..., -3.2024e-01,
3.3671e-01, 2.9008e-01]],
[[-3.2384e+00, -1.4831e+00, -6.1607e+00, ..., -4.7518e+00,
-4.3386e+00, -3.3090e+00],
[-5.8596e-01, 2.1098e+00, -4.2179e+00, ..., -3.4228e+00,
-3.5253e+00, -2.8786e+00],
[ 1.6768e-02, 1.8199e+00, -7.3960e+00, ..., -5.3381e+00,
-5.1702e+00, -4.8304e+00],
...,
[-1.2451e+00, 2.9836e-01, -6.5293e+00, ..., -4.5328e+00,
-4.0804e+00, -3.3797e+00],
[ 1.5194e+00, 2.6315e+00, -1.3510e+00, ..., -6.0597e-01,
-9.8837e-01, -1.0689e+00],
[ 9.9343e-01, 1.6419e+00, -2.4209e+00, ..., -2.2012e+00,
-2.4668e+00, -2.3801e+00]],
[[-1.3785e+00, -2.3462e+00, -2.1656e+00, ..., -5.5190e-01,
-1.6604e-01, -4.2863e-01],
[-1.9990e+00, -2.1966e+00, -1.5673e+00, ..., -3.0733e-01,
9.5666e-02, 5.8266e-02],
[-2.4986e+00, -1.9962e+00, -8.4459e-01, ..., -2.6673e-01,
2.4792e-01, 2.8757e-01],
...,
[-1.5870e+00, -1.8393e+00, -8.0481e-01, ..., -5.2076e-01,
-1.8709e-01, -3.8562e-01],
[-1.7473e+00, -1.1934e+00, -7.8810e-01, ..., -2.9634e-01,
1.9071e-01, 2.4996e-01],
[-1.6875e+00, -1.0871e+00, -5.8866e-01, ..., -5.2536e-01,
-2.7684e-01, -2.3305e-01]],
...,
[[-2.3791e+00, -1.3323e+00, -2.7407e+00, ..., -3.5177e+00,
-2.6959e+00, -1.7557e+00],
[-1.8928e+00, 1.8670e-01, -1.3209e+00, ..., -2.5950e+00,
-2.3504e+00, -1.8188e+00],
[-5.3862e-01, 2.4342e+00, 9.0417e-01, ..., -5.9470e-01,
-1.0447e-01, 1.2512e-01],
...,
[-1.6972e+00, 1.1991e+00, 2.1901e-01, ..., -1.6306e+00,
-6.8944e-01, -6.2599e-02],
[-8.3315e-01, 1.6942e+00, 1.5290e+00, ..., -1.3513e-01,
1.2635e-01, 3.8915e-01],
[-2.0830e+00, -1.0548e+00, -8.5080e-01, ..., -2.5255e+00,
-1.5369e+00, -9.5001e-01]],
[[-2.3730e+00, -2.6385e+00, -4.0149e+00, ..., -1.2218e+00,
-8.1804e-01, -7.9476e-01],
[-1.6712e+00, -1.3676e+00, -4.7687e+00, ..., -1.4296e+00,
-1.2540e+00, -1.5834e+00],
[-3.0220e+00, -2.4883e+00, -6.9392e+00, ..., -3.2685e+00,
-2.7463e+00, -2.2186e+00],
...,
[-2.5112e+00, -1.7112e+00, -4.9460e+00, ..., -1.3747e+00,
-9.4450e-01, -8.2960e-01],
[-2.2056e+00, -1.5449e+00, -3.8605e+00, ..., -1.6820e+00,
-1.0739e+00, -7.1417e-01],
[-2.2944e+00, -1.7629e+00, -3.1054e+00, ..., -2.8579e+00,
-2.0156e+00, -1.8603e+00]],
[[-1.8834e+00, -2.6078e+00, -2.3976e+00, ..., -1.7623e+00,
-1.4700e+00, -1.1229e+00],
[-1.9070e+00, -2.6161e+00, -2.0868e+00, ..., -1.3975e+00,
-1.1123e+00, -8.8732e-01],
[-2.4309e+00, -2.9789e+00, -2.4979e+00, ..., -1.6169e+00,
-1.2151e+00, -9.0240e-01],
...,
[-3.1043e+00, -3.8809e+00, -3.6495e+00, ..., -2.9474e+00,
-2.3562e+00, -1.8116e+00],
[-2.0678e+00, -2.6660e+00, -2.6271e+00, ..., -2.0217e+00,
-1.6684e+00, -1.3543e+00],
[-1.4854e+00, -1.8028e+00, -1.8451e+00, ..., -1.2408e+00,
-1.0570e+00, -8.1367e-01]]],
[[[ 1.9588e-01, 4.5820e-01, 6.0062e-01, ..., -3.5680e-01,
-6.6479e-02, 6.1444e-03],
[-2.0362e+00, -1.3735e+00, -2.8637e+00, ..., -3.4021e+00,
-1.3767e+00, -4.8959e-01],
[-1.5940e-01, -8.8935e-01, 2.3161e-02, ..., -2.4974e-01,
-3.9594e-01, 1.2790e-01],
...,
[-9.9149e-01, -2.1916e+00, -2.1075e+00, ..., -1.7187e+00,
-1.7192e+00, -1.1080e+00],
[ 7.9614e-01, -1.2801e-01, -1.3922e-01, ..., 4.3974e-02,
2.7580e-01, 4.5359e-01],
[ 1.9988e-01, -5.8848e-01, -4.2975e-01, ..., 2.0120e-01,
4.3993e-01, 4.9097e-01]],
[[-1.6982e+00, -1.4455e+00, -4.2759e+00, ..., -7.0149e+00,
-6.1813e+00, -4.6039e+00],
[ 4.2510e-01, 1.3822e+00, -2.5415e+00, ..., -5.3925e+00,
-5.5263e+00, -4.2779e+00],
[ 1.1269e+00, 1.1352e+00, -5.0102e+00, ..., -7.8428e+00,
-7.8622e+00, -6.7005e+00],
...,
[ 4.1535e-01, 3.7086e-01, -4.1288e+00, ..., -7.1001e+00,
-6.7185e+00, -5.2355e+00],
[ 2.5627e+00, 2.5047e+00, 1.1624e-01, ..., -2.2805e+00,
-2.8633e+00, -2.2789e+00],
[ 1.5325e+00, 1.3308e+00, -1.6208e+00, ..., -3.0221e+00,
-3.5294e+00, -3.0444e+00]],
[[-1.2739e-01, -1.1195e+00, -1.3735e+00, ..., -1.4743e+00,
-1.0895e+00, -1.3915e+00],
[-4.6920e-01, -7.8021e-01, -8.6944e-01, ..., -1.0295e+00,
-6.2058e-01, -9.0314e-01],
[-2.8721e-01, -1.9652e-01, -4.4098e-01, ..., -1.0957e+00,
-5.6643e-01, -4.4576e-01],
...,
[-8.4308e-02, -6.1805e-01, -6.1585e-01, ..., -9.9607e-01,
-8.4504e-01, -1.1780e+00],
[-7.2787e-01, -2.3131e-01, -5.8182e-01, ..., -5.6706e-01,
-2.8056e-01, -4.4161e-01],
[-8.8602e-01, -2.6702e-01, -4.7360e-01, ..., -4.9256e-01,
-4.4392e-01, -6.0031e-01]],
...,
[[-2.8151e+00, -2.4751e+00, -2.4424e+00, ..., -3.3004e+00,
-2.6653e+00, -1.9252e+00],
[-2.1026e+00, -8.9948e-01, -1.1308e+00, ..., -2.5147e+00,
-2.5973e+00, -2.0931e+00],
[-1.1566e+00, 7.2089e-01, 7.4843e-01, ..., -1.3067e-01,
4.7286e-02, 3.3481e-01],
...,
[-2.2498e+00, -1.5390e-01, -1.3426e-01, ..., -8.8283e-01,
-2.6918e-01, 2.2790e-01],
[-1.2393e+00, 8.8598e-01, 1.0876e+00, ..., 3.6395e-01,
4.0895e-01, 7.3804e-01],
[-2.1672e+00, -1.4748e+00, -1.6653e+00, ..., -1.9898e+00,
-1.0575e+00, -3.0599e-01]],
[[-1.6469e+00, -1.6391e+00, -2.7623e+00, ..., -2.1272e+00,
-1.7610e+00, -1.7103e+00],
[-7.2241e-01, -1.3704e-01, -3.1035e+00, ..., -2.7301e+00,
-2.7435e+00, -2.8718e+00],
[-2.0933e+00, -1.0280e+00, -5.0669e+00, ..., -4.4543e+00,
-4.5205e+00, -3.7588e+00],
...,
[-1.3516e+00, 1.9026e-01, -3.3985e+00, ..., -2.9717e+00,
-2.8633e+00, -2.4103e+00],
[-1.6769e+00, -4.5504e-01, -3.0138e+00, ..., -2.5195e+00,
-2.3380e+00, -1.7575e+00],
[-2.3250e+00, -1.3406e+00, -3.0313e+00, ..., -2.8062e+00,
-2.4068e+00, -2.1650e+00]],
[[-1.2948e+00, -1.8194e+00, -1.6582e+00, ..., -2.6483e+00,
-2.2043e+00, -1.7061e+00],
[-1.2331e+00, -1.6927e+00, -1.2297e+00, ..., -2.4241e+00,
-1.9523e+00, -1.5975e+00],
[-1.5408e+00, -1.8869e+00, -1.4777e+00, ..., -2.8095e+00,
-2.2357e+00, -1.7329e+00],
...,
[-2.4567e+00, -3.0667e+00, -2.7882e+00, ..., -3.9008e+00,
-3.1630e+00, -2.4853e+00],
[-1.5930e+00, -2.0982e+00, -1.9947e+00, ..., -2.7166e+00,
-2.2704e+00, -1.8743e+00],
[-1.0428e+00, -1.3583e+00, -1.3703e+00, ..., -1.8231e+00,
-1.5052e+00, -1.1686e+00]]]], device='cuda:0'),
tensor([[[[ 3.9789e-01, 8.1600e-01, 3.6992e-01, ..., 5.0908e-01,
1.0074e+00, 6.6439e-01],
[-1.4069e+00, -5.5573e-01, -2.2340e+00, ..., -2.3434e+00,
-6.2280e-01, 3.7271e-02],
[ 8.6482e-01, 5.5023e-01, 2.1479e-01, ..., -1.2152e-01,
5.6250e-01, 7.0148e-01],
...,
[ 1.5756e-02, -6.6385e-01, -1.4564e+00, ..., -1.8502e+00,
-1.2763e+00, -8.2251e-01],
[ 1.2797e+00, 1.0693e+00, 5.0569e-01, ..., -1.9524e-01,
5.5975e-01, 7.3659e-01],
[ 9.3951e-01, 8.2607e-01, 5.4688e-01, ..., -2.2397e-01,
3.5595e-01, 3.3782e-01]],
[[-2.8676e+00, -1.5691e+00, -4.9765e+00, ..., -4.8662e+00,
-4.4039e+00, -3.3866e+00],
[-5.4830e-01, 1.6652e+00, -3.0718e+00, ..., -3.4667e+00,
-3.5478e+00, -2.9486e+00],
[ 2.3166e-01, 1.4426e+00, -5.9040e+00, ..., -5.4121e+00,
-5.2515e+00, -4.9735e+00],
...,
[-9.5348e-01, -3.3772e-03, -5.0748e+00, ..., -4.6482e+00,
-4.1868e+00, -3.5140e+00],
[ 1.6383e+00, 2.2351e+00, -5.3845e-01, ..., -9.2106e-01,
-1.2384e+00, -1.3009e+00],
[ 9.5240e-01, 1.2251e+00, -1.9550e+00, ..., -2.2538e+00,
-2.4906e+00, -2.3984e+00]],
[[-7.2210e-01, -1.9179e+00, -1.7918e+00, ..., -6.0493e-01,
-2.3610e-01, -4.8495e-01],
[-1.4786e+00, -1.9329e+00, -1.2643e+00, ..., -3.9915e-01,
-5.4280e-03, 2.1995e-03],
[-1.4676e+00, -1.4392e+00, -7.9041e-01, ..., -2.8323e-01,
2.1428e-01, 2.5866e-01],
...,
[-9.2063e-01, -1.4384e+00, -7.1362e-01, ..., -4.7153e-01,
-1.8431e-01, -3.4938e-01],
[-1.2123e+00, -9.0147e-01, -6.3831e-01, ..., -3.0253e-01,
1.3900e-01, 2.0972e-01],
[-1.2591e+00, -7.4111e-01, -3.4370e-01, ..., -4.7028e-01,
-2.6165e-01, -2.1995e-01]],
...,
[[-2.2936e+00, -1.1792e+00, -2.5634e+00, ..., -3.5867e+00,
-2.8137e+00, -1.8243e+00],
[-7.1939e-01, 1.1240e+00, -8.7664e-01, ..., -2.5152e+00,
-2.3163e+00, -1.8155e+00],
[-1.8429e-01, 2.6194e+00, 7.9473e-01, ..., -5.6814e-01,
-6.8151e-02, 1.5311e-01],
...,
[-1.2884e+00, 1.3830e+00, 7.7724e-02, ..., -1.5139e+00,
-6.5608e-01, -7.5990e-02],
[-4.8420e-01, 1.5395e+00, 8.7060e-01, ..., -1.6226e-01,
9.0155e-02, 3.2356e-01],
[-1.5977e+00, -1.0229e+00, -1.4065e+00, ..., -2.3119e+00,
-1.4147e+00, -8.7406e-01]],
[[-1.4988e+00, -1.5470e+00, -3.2732e+00, ..., -1.0961e+00,
-6.7473e-01, -6.8071e-01],
[-7.5360e-01, -3.5109e-01, -4.1174e+00, ..., -1.3562e+00,
-1.2036e+00, -1.5698e+00],
[-2.2103e+00, -1.6140e+00, -6.5487e+00, ..., -3.3003e+00,
-2.8318e+00, -2.2646e+00],
...,
[-1.8601e+00, -1.0381e+00, -4.7615e+00, ..., -1.5805e+00,
-1.1408e+00, -9.5491e-01],
[-1.7325e+00, -1.0122e+00, -3.7531e+00, ..., -1.7760e+00,
-1.1833e+00, -7.8037e-01],
[-2.0082e+00, -1.3831e+00, -2.9881e+00, ..., -2.7641e+00,
-1.9746e+00, -1.7801e+00]],
[[-1.3340e+00, -1.9971e+00, -1.8398e+00, ..., -1.7749e+00,
-1.4799e+00, -1.1221e+00],
[-1.3358e+00, -1.9263e+00, -1.5041e+00, ..., -1.3967e+00,
-1.1106e+00, -8.8389e-01],
[-1.7277e+00, -2.1414e+00, -1.7510e+00, ..., -1.5987e+00,
-1.1982e+00, -8.8514e-01],
...,
[-2.4280e+00, -3.0863e+00, -2.9798e+00, ..., -2.8567e+00,
-2.2866e+00, -1.7607e+00],
[-1.6507e+00, -2.1297e+00, -2.1524e+00, ..., -1.9867e+00,
-1.6348e+00, -1.3202e+00],
[-1.1537e+00, -1.4193e+00, -1.5317e+00, ..., -1.2395e+00,
-1.0443e+00, -7.9706e-01]]],
[[[ 2.4835e-01, 5.4474e-01, 6.3795e-01, ..., -4.9982e-01,
-2.6030e-01, -1.0461e-01],
[-1.8858e+00, -1.1527e+00, -2.6148e+00, ..., -3.5039e+00,
-1.5083e+00, -6.1990e-01],
[-1.9339e-02, -6.9683e-01, 9.2813e-02, ..., -3.2955e-01,
-5.6434e-01, 8.3165e-02],
...,
[-6.4427e+00, -8.4199e+00, -5.3988e+00, ..., -1.9601e+00,
-1.9496e+00, -1.2907e+00],
[ 9.8297e-01, -1.7159e+00, -2.4870e+00, ..., -1.1733e-01,
1.3627e-01, 3.6481e-01],
[ 2.7745e-01, -4.0255e-01, -2.3440e-01, ..., 7.4172e-02,
3.5700e-01, 4.4207e-01]],
[[-1.7892e+00, -1.3861e+00, -4.2658e+00, ..., -7.0554e+00,
-6.2687e+00, -4.6546e+00],
[ 3.4762e-01, 1.4658e+00, -2.5618e+00, ..., -5.6823e+00,
-5.8116e+00, -4.4414e+00],
[ 1.0858e+00, 1.2108e+00, -5.0851e+00, ..., -8.1213e+00,
-8.1006e+00, -6.7277e+00],
...,
[-6.3875e-02, 1.0619e+00, -3.6252e+00, ..., -7.4543e+00,
-7.0149e+00, -5.3630e+00],
[ 1.9001e+00, 2.0104e+00, 3.2521e-01, ..., -2.1291e+00,
-2.7751e+00, -2.1281e+00],
[ 1.4011e+00, 1.2458e+00, -1.6892e+00, ..., -3.0273e+00,
-3.5958e+00, -3.0819e+00]],
[[-2.0428e-01, -1.2409e+00, -1.4341e+00, ..., -1.5093e+00,
-1.1402e+00, -1.4094e+00],
[-6.2857e-01, -9.6540e-01, -9.3777e-01, ..., -1.0602e+00,
-6.8652e-01, -9.9789e-01],
[-4.3941e-01, -3.4099e-01, -4.4693e-01, ..., -1.2046e+00,
-6.6823e-01, -5.1456e-01],
...,
[-2.3120e+00, -3.2310e+00, -1.6269e+00, ..., -1.0454e+00,
-8.9074e-01, -1.2894e+00],
[-1.7590e+00, -2.2465e+00, -1.8076e+00, ..., -5.4581e-01,
-2.2984e-01, -4.2985e-01],
[-9.0265e-01, -3.3727e-01, -4.6185e-01, ..., -6.1251e-01,
-5.6793e-01, -7.4636e-01]],
...,
[[-2.7485e+00, -2.2815e+00, -2.4249e+00, ..., -3.0030e+00,
-2.4444e+00, -1.8147e+00],
[-1.8797e+00, -5.4254e-01, -9.9978e-01, ..., -2.5672e+00,
-2.6178e+00, -2.0800e+00],
[-9.8536e-01, 1.0131e+00, 7.8597e-01, ..., -9.0143e-02,
-3.0713e-02, 2.4657e-01],
...,
[-1.6954e+00, -2.1529e+00, -3.8812e+00, ..., -9.2967e-01,
-2.8184e-01, 2.4775e-01],
[-8.3452e-02, 1.5906e+00, 1.6828e+00, ..., 5.3488e-01,
5.2938e-01, 8.6995e-01],
[-2.0500e+00, -1.4167e+00, -1.6633e+00, ..., -2.1204e+00,
-1.0923e+00, -2.8278e-01]],
[[-1.5853e+00, -1.5625e+00, -2.7490e+00, ..., -2.4705e+00,
-2.1421e+00, -2.0074e+00],
[-6.8021e-01, -9.0526e-02, -3.1878e+00, ..., -3.0728e+00,
-3.0648e+00, -3.0409e+00],
[-2.0614e+00, -1.0281e+00, -5.2267e+00, ..., -4.5452e+00,
-4.6477e+00, -3.8362e+00],
...,
[-2.3685e+00, -1.6138e+00, -3.8187e+00, ..., -2.8926e+00,
-2.8280e+00, -2.3666e+00],
[-2.1614e+00, -1.5987e+00, -3.8920e+00, ..., -2.4984e+00,
-2.3555e+00, -1.7460e+00],
[-2.2562e+00, -1.3824e+00, -2.9956e+00, ..., -2.9052e+00,
-2.5056e+00, -2.2789e+00]],
[[-1.2766e+00, -1.8093e+00, -1.6239e+00, ..., -2.6938e+00,
-2.2530e+00, -1.7495e+00],
[-1.2218e+00, -1.6880e+00, -1.2096e+00, ..., -2.5563e+00,
-2.0566e+00, -1.6871e+00],
[-1.5445e+00, -1.8795e+00, -1.4572e+00, ..., -3.0128e+00,
-2.4075e+00, -1.8730e+00],
...,
[-2.9573e+00, -3.8033e+00, -3.6360e+00, ..., -4.1694e+00,
-3.3882e+00, -2.6591e+00],
[-1.8240e+00, -2.4832e+00, -2.3324e+00, ..., -2.8709e+00,
-2.4009e+00, -1.9889e+00],
[-1.0271e+00, -1.3285e+00, -1.3585e+00, ..., -1.9051e+00,
-1.5758e+00, -1.2272e+00]]],
[[[ 1.3049e-01, 3.4048e-01, -1.3224e+00, ..., 6.5252e-01,
9.3797e-01, 6.9946e-01],
[-2.1597e+00, -1.5514e+00, -4.6038e+00, ..., -1.4126e+00,
-1.0800e-01, 4.3827e-01],
[ 4.1213e-01, -3.5399e-01, -1.2288e+00, ..., 5.1525e-01,
9.2909e-01, 8.4020e-01],
...,
[-2.8385e-01, -1.3057e+00, -2.6225e+00, ..., -1.1663e+00,
-7.9044e-01, -6.2924e-01],
[ 1.2662e+00, 6.3138e-01, -9.7450e-01, ..., 5.2877e-01,
7.9081e-01, 8.1682e-01],
[ 8.0117e-01, 2.4573e-01, -4.7771e-01, ..., 5.7122e-01,
8.1946e-01, 7.1861e-01]],
[[-3.3018e+00, -1.4814e+00, -6.2327e+00, ..., -5.4599e+00,
-4.6840e+00, -3.6531e+00],
[-5.8668e-01, 2.1655e+00, -4.2399e+00, ..., -4.2449e+00,
-4.1374e+00, -3.4090e+00],
[ 7.4287e-03, 1.8549e+00, -7.5085e+00, ..., -6.2016e+00,
-5.9900e+00, -5.5386e+00],
...,
[-1.3059e+00, 2.6282e-01, -6.6638e+00, ..., -5.4469e+00,
-4.7742e+00, -3.9704e+00],
[ 1.4916e+00, 2.6123e+00, -1.4272e+00, ..., -1.3400e+00,
-1.4631e+00, -1.5314e+00],
[ 9.9101e-01, 1.6611e+00, -2.4466e+00, ..., -2.5155e+00,
-2.6285e+00, -2.5811e+00]],
[[-1.4642e+00, -2.4159e+00, -2.2230e+00, ..., -8.3826e-01,
-6.3856e-01, -8.2278e-01],
[-2.0903e+00, -2.2413e+00, -1.5787e+00, ..., -6.9393e-01,
-4.3189e-01, -3.7121e-01],
[-2.6373e+00, -2.0727e+00, -8.4453e-01, ..., -4.0877e-01,
-1.2944e-01, -1.1856e-01],
...,
[-1.6532e+00, -1.8670e+00, -7.7693e-01, ..., -6.2916e-01,
-5.3926e-01, -7.0022e-01],
[-1.8066e+00, -1.2183e+00, -7.8057e-01, ..., -2.2210e-01,
-5.5178e-02, -6.1873e-02],
[-1.7805e+00, -1.1821e+00, -6.5199e-01, ..., -3.1769e-01,
-2.7703e-01, -2.4814e-01]],
...,
[[-2.3872e+00, -1.3124e+00, -2.7762e+00, ..., -3.0469e+00,
-2.4408e+00, -1.6290e+00],
[-1.9623e+00, 1.3621e-01, -1.4123e+00, ..., -1.4780e+00,
-1.5634e+00, -1.3555e+00],
[-5.1370e-01, 2.4937e+00, 9.0904e-01, ..., -5.8210e-02,
2.1824e-01, 2.9374e-01],
...,
[-1.7270e+00, 1.1750e+00, 1.8635e-01, ..., -9.5798e-01,
-3.0408e-01, 1.6462e-01],
[-8.4837e-01, 1.7165e+00, 1.5155e+00, ..., 1.5291e-01,
2.8315e-01, 3.7973e-01],
[-2.0978e+00, -1.0087e+00, -7.8865e-01, ..., -2.1241e+00,
-1.2371e+00, -6.3287e-01]],
[[-2.4213e+00, -2.7041e+00, -4.0788e+00, ..., -1.1109e+00,
-6.1585e-01, -5.0625e-01],
[-1.6956e+00, -1.3911e+00, -4.8029e+00, ..., -1.7251e+00,
-1.5311e+00, -1.7505e+00],
[-3.0441e+00, -2.4802e+00, -6.9198e+00, ..., -3.7538e+00,
-3.3044e+00, -2.6675e+00],
...,
[-2.5349e+00, -1.7896e+00, -5.0399e+00, ..., -2.2674e+00,
-1.7925e+00, -1.4688e+00],
[-2.2175e+00, -1.5868e+00, -3.9226e+00, ..., -2.1616e+00,
-1.5210e+00, -1.0183e+00],
[-2.2984e+00, -1.7908e+00, -3.1004e+00, ..., -2.7857e+00,
-1.9353e+00, -1.6673e+00]],
[[-1.9390e+00, -2.6776e+00, -2.4638e+00, ..., -1.6953e+00,
-1.4057e+00, -1.0515e+00],
[-1.9612e+00, -2.6952e+00, -2.1537e+00, ..., -1.3821e+00,
-1.1245e+00, -8.9309e-01],
[-2.4903e+00, -3.0693e+00, -2.5790e+00, ..., -1.5956e+00,
-1.2204e+00, -8.9251e-01],
...,
[-3.1666e+00, -3.9539e+00, -3.7126e+00, ..., -2.7865e+00,
-2.2222e+00, -1.6827e+00],
[-2.1036e+00, -2.7243e+00, -2.6759e+00, ..., -1.9666e+00,
-1.6118e+00, -1.2853e+00],
[-1.5159e+00, -1.8490e+00, -1.8865e+00, ..., -1.2333e+00,
-1.0482e+00, -7.7482e-01]]],
[[[ 3.2740e-01, 8.0447e-01, -3.6228e+00, ..., 5.3280e-01,
1.0243e+00, 6.3700e-01],
[-1.9983e+00, -3.5254e+00, -8.9044e+00, ..., -2.4217e+00,
-7.1373e-01, -6.3533e-02],
[-4.9047e-02, -3.5298e+00, -4.7750e+00, ..., -2.3808e-02,
6.1395e-01, 7.0222e-01],
...,
[-2.1828e-01, -1.2155e+00, -2.7807e+00, ..., -1.5573e+00,
-1.4513e+00, -7.9972e-01],
[ 1.4542e+00, 7.9521e-01, -1.1715e+00, ..., -4.1310e-01,
-2.1858e-02, 3.0988e-01],
[ 8.2069e-01, 1.7227e-01, -8.8321e-01, ..., -3.3850e-01,
8.7192e-02, 3.0445e-01]],
[[-1.7220e+00, -4.7319e-01, -3.0146e+00, ..., -4.9355e+00,
-4.4551e+00, -3.4349e+00],
[ 4.4049e-01, 3.2985e+00, 1.2223e-01, ..., -3.6118e+00,
-3.6804e+00, -3.0464e+00],
[ 1.1515e+00, 3.3454e+00, -2.8910e+00, ..., -5.5698e+00,
-5.3602e+00, -5.0186e+00],
...,
[-1.4280e+00, 5.2957e-02, -7.0254e+00, ..., -7.9197e+00,
-7.2786e+00, -5.6808e+00],
[ 1.7104e+00, 2.8067e+00, -1.1887e+00, ..., -2.8141e+00,
-3.2370e+00, -2.5467e+00],
[ 1.1894e+00, 1.8614e+00, -2.3321e+00, ..., -3.3730e+00,
-3.7944e+00, -3.2274e+00]],
[[-1.6540e-01, -9.3697e-01, -5.4801e-01, ..., -5.2397e-01,
-2.0637e-01, -5.0924e-01],
[-6.5661e-01, -7.3518e-01, 3.4700e-01, ..., -3.3400e-01,
-2.6012e-02, -1.0390e-01],
[-5.8898e-01, -6.6907e-01, 6.3166e-01, ..., -3.5646e-01,
4.9932e-02, 5.4876e-02],
...,
[-1.5028e+00, -1.8114e+00, -7.3895e-01, ..., -8.9091e-01,
-7.8079e-01, -1.0532e+00],
[-1.8210e+00, -1.2013e+00, -8.0632e-01, ..., -5.7180e-01,
-3.0166e-01, -3.6354e-01],
[-1.9799e+00, -1.3969e+00, -9.7917e-01, ..., -1.1134e+00,
-9.8032e-01, -9.3807e-01]],
...,
[[-2.8864e+00, -1.7902e+00, -3.3846e+00, ..., -3.4211e+00,
-2.6285e+00, -1.7525e+00],
[-2.0461e+00, -5.5754e-01, -5.2834e+00, ..., -2.4740e+00,
-2.2488e+00, -1.7950e+00],
[-9.5800e-01, 4.2662e-01, -4.1554e+00, ..., -4.1539e-01,
1.9733e-02, 1.5088e-01],
...,
[-1.9365e+00, 1.1554e+00, -4.3897e-02, ..., -1.1074e+00,
-5.8634e-01, -1.6466e-02],
[-8.1431e-01, 1.9725e+00, 1.7575e+00, ..., 5.9367e-01,
4.5598e-01, 7.2486e-01],
[-2.2645e+00, -1.1415e+00, -9.0373e-01, ..., -1.7093e+00,
-8.6718e-01, -2.1832e-01]],
[[-1.5414e+00, -1.2561e+00, -2.7770e+00, ..., -1.2136e+00,
-7.7744e-01, -7.7257e-01],
[-6.0462e-01, 6.5016e-01, -2.5723e+00, ..., -1.3910e+00,
-1.2356e+00, -1.5652e+00],
[-2.0654e+00, -6.3432e-01, -4.9841e+00, ..., -3.2450e+00,
-2.7709e+00, -2.2266e+00],
...,
[-2.5687e+00, -1.7082e+00, -5.5235e+00, ..., -3.1341e+00,
-2.8599e+00, -2.2981e+00],
[-2.2897e+00, -1.5760e+00, -4.3040e+00, ..., -2.7204e+00,
-2.4335e+00, -1.7503e+00],
[-2.4374e+00, -1.8608e+00, -3.3577e+00, ..., -2.9975e+00,
-2.5423e+00, -2.1974e+00]],
[[-1.3066e+00, -2.1785e+00, -2.6547e+00, ..., -1.7533e+00,
-1.4463e+00, -1.0980e+00],
[-1.2519e+00, -2.2250e+00, -2.7226e+00, ..., -1.3713e+00,
-1.0837e+00, -8.6140e-01],
[-1.5769e+00, -2.3825e+00, -3.0116e+00, ..., -1.5964e+00,
-1.2010e+00, -8.8278e-01],
...,
[-3.2971e+00, -4.1299e+00, -3.9117e+00, ..., -4.1277e+00,
-3.3103e+00, -2.5563e+00],
[-2.1799e+00, -2.8321e+00, -2.7927e+00, ..., -2.8983e+00,
-2.3925e+00, -1.9442e+00],
[-1.5949e+00, -1.9448e+00, -1.9954e+00, ..., -1.9672e+00,
-1.5879e+00, -1.2186e+00]]],
[[[ 1.3582e-01, 3.9089e-01, -1.1925e+00, ..., 4.6399e-01,
1.0189e+00, 6.5886e-01],
[-2.1115e+00, -1.4213e+00, -4.4051e+00, ..., -2.6001e+00,
-7.4675e-01, -5.2434e-02],
[ 4.3206e-01, -2.5601e-01, -1.1355e+00, ..., -2.6043e-01,
5.3084e-01, 6.6373e-01],
...,
[-2.5602e-01, -1.2516e+00, -2.5294e+00, ..., -1.9735e+00,
-1.3728e+00, -9.1102e-01],
[ 1.2425e+00, 6.5826e-01, -8.4841e-01, ..., -3.2358e-01,
5.0300e-01, 6.8989e-01],
[ 8.1637e-01, 2.9580e-01, -3.9394e-01, ..., -3.4368e-01,
2.9223e-01, 2.7482e-01]],
[[-3.2674e+00, -1.5069e+00, -6.1388e+00, ..., -4.8732e+00,
-4.4293e+00, -3.3763e+00],
[-6.0120e-01, 2.0944e+00, -4.1850e+00, ..., -3.4017e+00,
-3.5510e+00, -2.9275e+00],
[-1.2029e-02, 1.7931e+00, -7.4012e+00, ..., -5.3956e+00,
-5.2977e+00, -4.9755e+00],
...,
[-1.3393e+00, 1.7068e-01, -6.6010e+00, ..., -4.5566e+00,
-4.1271e+00, -3.4427e+00],
[ 1.4410e+00, 2.5250e+00, -1.4020e+00, ..., -8.5206e-01,
-1.1878e+00, -1.2277e+00],
[ 9.2874e-01, 1.5842e+00, -2.4102e+00, ..., -2.2198e+00,
-2.4594e+00, -2.3491e+00]],
[[-1.3936e+00, -2.3346e+00, -2.1645e+00, ..., -6.0088e-01,
-2.0931e-01, -4.7259e-01],
[-2.0268e+00, -2.1742e+00, -1.5524e+00, ..., -3.5040e-01,
5.3646e-02, 3.0957e-02],
[-2.5462e+00, -1.9836e+00, -8.2333e-01, ..., -2.6880e-01,
2.4348e-01, 2.7050e-01],
...,
[-1.5223e+00, -1.7714e+00, -7.6215e-01, ..., -4.9675e-01,
-1.5889e-01, -3.4806e-01],
[-1.6913e+00, -1.1512e+00, -7.5913e-01, ..., -3.2330e-01,
1.5979e-01, 2.3056e-01],
[-1.6922e+00, -1.1225e+00, -6.1758e-01, ..., -5.0255e-01,
-2.7350e-01, -2.3845e-01]],
...,
[[-2.3755e+00, -1.3256e+00, -2.7107e+00, ..., -3.7300e+00,
-2.8900e+00, -1.8670e+00],
[-1.9112e+00, 1.4737e-01, -1.3129e+00, ..., -2.6686e+00,
-2.4331e+00, -1.8875e+00],
[-5.3519e-01, 2.3885e+00, 9.0559e-01, ..., -5.4031e-01,
-5.9133e-02, 1.7704e-01],
...,
[-1.7801e+00, 1.0621e+00, 1.4546e-01, ..., -1.5293e+00,
-6.8368e-01, -1.2628e-01],
[-9.2986e-01, 1.5685e+00, 1.4247e+00, ..., -1.5534e-01,
9.4557e-02, 3.1433e-01],
[-2.1593e+00, -1.1026e+00, -8.7197e-01, ..., -2.3812e+00,
-1.4689e+00, -9.2755e-01]],
[[-2.3844e+00, -2.6465e+00, -3.9996e+00, ..., -1.1001e+00,
-7.1249e-01, -7.4608e-01],
[-1.6662e+00, -1.3535e+00, -4.7547e+00, ..., -1.3513e+00,
-1.2408e+00, -1.6212e+00],
[-3.0279e+00, -2.4581e+00, -6.8669e+00, ..., -3.2921e+00,
-2.8335e+00, -2.2789e+00],
...,
[-2.5375e+00, -1.8006e+00, -4.9494e+00, ..., -1.4444e+00,
-1.0288e+00, -8.7422e-01],
[-2.2224e+00, -1.6161e+00, -3.8912e+00, ..., -1.7166e+00,
-1.1416e+00, -7.3891e-01],
[-2.2935e+00, -1.8142e+00, -3.0903e+00, ..., -2.8053e+00,
-2.0051e+00, -1.8199e+00]],
[[-1.9003e+00, -2.6357e+00, -2.4270e+00, ..., -1.8059e+00,
-1.4984e+00, -1.1378e+00],
[-1.9224e+00, -2.6403e+00, -2.1160e+00, ..., -1.3985e+00,
-1.1072e+00, -8.8699e-01],
[-2.4443e+00, -3.0064e+00, -2.5260e+00, ..., -1.6022e+00,
-1.2036e+00, -8.9470e-01],
...,
[-3.0967e+00, -3.8738e+00, -3.6626e+00, ..., -2.8769e+00,
-2.2995e+00, -1.7725e+00],
[-2.0562e+00, -2.6632e+00, -2.6361e+00, ..., -1.9756e+00,
-1.6275e+00, -1.3143e+00],
[-1.4768e+00, -1.8051e+00, -1.8561e+00, ..., -1.2190e+00,
-1.0330e+00, -7.9268e-01]]]], device='cuda:0'),
tensor([[[[ 2.5652e-01, 5.6197e-01, 6.3144e-01, ..., -3.6986e-01,
-1.1624e-01, -2.2871e-02],
[-2.1210e+00, -1.3137e+00, -3.0395e+00, ..., -3.2470e+00,
-1.3432e+00, -5.0258e-01],
[-5.9596e-02, -8.6799e-01, -1.0719e-01, ..., -1.8554e-01,
-3.8620e-01, 1.3260e-01],
...,
[-9.6033e-01, -2.1876e+00, -2.1056e+00, ..., -1.9278e+00,
-1.9050e+00, -1.2756e+00],
[ 6.8022e-01, -1.8866e-01, -1.9313e-01, ..., 2.4361e-02,
2.6102e-01, 4.4224e-01],
[ 1.2630e-01, -6.1847e-01, -4.2496e-01, ..., 2.1205e-01,
4.5385e-01, 4.9425e-01]],
[[-1.7969e+00, -1.2267e+00, -4.2750e+00, ..., -6.8951e+00,
-6.1002e+00, -4.5123e+00],
[ 5.8038e-01, 1.9311e+00, -2.3123e+00, ..., -5.5218e+00,
-5.5813e+00, -4.2406e+00],
[ 1.4476e+00, 1.7053e+00, -5.0619e+00, ..., -7.8737e+00,
-7.7853e+00, -6.5325e+00],
...,
[ 3.4925e-01, 2.6805e-01, -4.2684e+00, ..., -7.1728e+00,
-6.7488e+00, -5.2431e+00],
[ 2.3559e+00, 2.2524e+00, -2.7159e-01, ..., -1.9992e+00,
-2.6302e+00, -2.0904e+00],
[ 1.3719e+00, 1.1504e+00, -1.8387e+00, ..., -2.9420e+00,
-3.5114e+00, -3.0543e+00]],
[[-2.5029e-01, -1.4476e+00, -1.6329e+00, ..., -1.4185e+00,
-1.0534e+00, -1.3362e+00],
[-7.1798e-01, -1.0730e+00, -1.0285e+00, ..., -9.9681e-01,
-6.2820e-01, -9.1989e-01],
[-4.1476e-01, -2.6113e-01, -4.1660e-01, ..., -1.0867e+00,
-5.7426e-01, -4.5602e-01],
...,
[ 9.0312e-03, -5.2958e-01, -6.2247e-01, ..., -1.0329e+00,
-8.8264e-01, -1.2477e+00],
[-6.2633e-01, -1.9963e-01, -5.9584e-01, ..., -5.6293e-01,
-2.5315e-01, -4.1857e-01],
[-8.0288e-01, -2.5778e-01, -5.0992e-01, ..., -5.0925e-01,
-4.6449e-01, -6.2357e-01]],
...,
[[-3.0121e+00, -2.4161e+00, -2.6496e+00, ..., -2.9629e+00,
-2.4270e+00, -1.8011e+00],
[-2.0003e+00, -5.3047e-01, -1.0923e+00, ..., -2.3512e+00,
-2.4391e+00, -1.9740e+00],
[-1.0130e+00, 1.2044e+00, 9.2266e-01, ..., -1.4618e-01,
-4.8681e-03, 2.8732e-01],
...,
[-2.2208e+00, -3.6338e-01, -2.2929e-01, ..., -9.2399e-01,
-2.5005e-01, 2.9339e-01],
[-1.1613e+00, 8.4239e-01, 1.0979e+00, ..., 4.4349e-01,
4.9004e-01, 8.2301e-01],
[-2.0463e+00, -1.3111e+00, -1.4536e+00, ..., -2.1878e+00,
-1.1459e+00, -3.3873e-01]],
[[-1.6069e+00, -1.5424e+00, -2.8589e+00, ..., -2.3213e+00,
-1.9344e+00, -1.8166e+00],
[-5.5068e-01, 1.8776e-01, -3.1346e+00, ..., -2.9141e+00,
-2.8505e+00, -2.8756e+00],
[-2.0203e+00, -7.0763e-01, -5.3583e+00, ..., -4.4703e+00,
-4.4657e+00, -3.7004e+00],
...,
[-1.3860e+00, 9.5332e-02, -3.1989e+00, ..., -2.8817e+00,
-2.7690e+00, -2.3306e+00],
[-1.6457e+00, -5.1096e-01, -2.9345e+00, ..., -2.4866e+00,
-2.2793e+00, -1.6806e+00],
[-2.2373e+00, -1.3275e+00, -2.9606e+00, ..., -2.8847e+00,
-2.4412e+00, -2.2070e+00]],
[[-1.3752e+00, -1.9564e+00, -1.7038e+00, ..., -2.6033e+00,
-2.1805e+00, -1.7019e+00],
[-1.2979e+00, -1.8153e+00, -1.2348e+00, ..., -2.4573e+00,
-1.9783e+00, -1.6189e+00],
[-1.6340e+00, -1.9796e+00, -1.4634e+00, ..., -2.8766e+00,
-2.2967e+00, -1.7811e+00],
...,
[-2.4245e+00, -3.0236e+00, -2.8358e+00, ..., -4.0266e+00,
-3.2670e+00, -2.5665e+00],
[-1.5810e+00, -2.0761e+00, -2.0089e+00, ..., -2.7862e+00,
-2.3338e+00, -1.9337e+00],
[-1.0458e+00, -1.3558e+00, -1.3951e+00, ..., -1.8491e+00,
-1.5371e+00, -1.1921e+00]]],
[[[ 4.6936e-01, 9.4631e-01, 8.5071e-01, ..., -4.1876e-01,
-1.7793e-01, -5.4087e-02],
[-1.1718e+00, -2.8008e-01, -1.4335e+00, ..., -3.3604e+00,
-1.4127e+00, -5.3641e-01],
[ 6.0303e-01, 4.3921e-01, 3.8669e-01, ..., -2.1621e-01,
-4.6267e-01, 9.9387e-02],
...,
[-1.5761e-01, -8.0350e-01, -1.5284e+00, ..., -1.9541e+00,
-1.9217e+00, -1.2776e+00],
[ 1.0996e+00, 8.3998e-01, 3.1841e-01, ..., -3.6960e-02,
1.9543e-01, 4.1435e-01],
[ 7.4539e-01, 5.7603e-01, 2.4638e-01, ..., 1.6293e-01,
4.1993e-01, 4.7779e-01]],
[[-2.3293e+00, -1.1443e+00, -4.1646e+00, ..., -6.9668e+00,
-6.1773e+00, -4.5799e+00],
[-3.3296e-01, 1.6455e+00, -2.5905e+00, ..., -5.5678e+00,
-5.6696e+00, -4.3448e+00],
[ 3.7198e-01, 1.4431e+00, -5.1106e+00, ..., -7.9930e+00,
-7.9318e+00, -6.6535e+00],
...,
[-5.0720e-01, 5.5801e-01, -4.1448e+00, ..., -7.2490e+00,
-6.8090e+00, -5.2791e+00],
[ 2.0646e+00, 2.7573e+00, 3.1675e-01, ..., -1.9963e+00,
-2.6331e+00, -2.0944e+00],
[ 1.2021e+00, 1.5013e+00, -1.5103e+00, ..., -2.9712e+00,
-3.5291e+00, -3.0735e+00]],
[[-5.7231e-01, -1.5524e+00, -1.3788e+00, ..., -1.4762e+00,
-1.0898e+00, -1.3701e+00],
[-1.2755e+00, -1.6236e+00, -1.0072e+00, ..., -1.0466e+00,
-6.5167e-01, -9.3537e-01],
[-1.1165e+00, -9.9017e-01, -4.0892e-01, ..., -1.1533e+00,
-6.4004e-01, -4.9026e-01],
...,
[-9.4693e-01, -1.3383e+00, -5.3421e-01, ..., -1.0501e+00,
-8.9472e-01, -1.2739e+00],
[-1.2756e+00, -8.4602e-01, -5.3790e-01, ..., -5.6410e-01,
-2.4859e-01, -4.2381e-01],
[-1.3439e+00, -7.2065e-01, -3.1370e-01, ..., -5.3967e-01,
-5.0743e-01, -6.7055e-01]],
...,
[[-2.1354e+00, -1.1908e+00, -2.5086e+00, ..., -3.0510e+00,
-2.4913e+00, -1.8119e+00],
[-6.6447e-01, 1.0049e+00, -9.0893e-01, ..., -2.4583e+00,
-2.5404e+00, -2.0578e+00],
[-2.4696e-01, 2.2252e+00, 5.1251e-01, ..., -7.8606e-02,
3.9326e-02, 2.9639e-01],
...,
[-1.4892e+00, 1.2351e+00, -1.7375e-01, ..., -9.1349e-01,
-2.6816e-01, 2.7620e-01],
[-5.0138e-01, 1.5625e+00, 8.2323e-01, ..., 4.9660e-01,
5.1682e-01, 8.3540e-01],
[-1.6148e+00, -1.0805e+00, -1.5888e+00, ..., -2.1637e+00,
-1.1302e+00, -3.2420e-01]],
[[-1.3092e+00, -1.2037e+00, -2.5740e+00, ..., -2.3395e+00,
-1.9946e+00, -1.8816e+00],
[-6.4779e-01, -1.4719e-01, -3.3526e+00, ..., -2.9467e+00,
-2.9398e+00, -2.9342e+00],
[-2.1138e+00, -1.4338e+00, -5.7488e+00, ..., -4.5115e+00,
-4.5642e+00, -3.7462e+00],
...,
[-1.6896e+00, -5.5467e-01, -4.1310e+00, ..., -2.8545e+00,
-2.7717e+00, -2.3441e+00],
[-1.6802e+00, -7.4960e-01, -3.4403e+00, ..., -2.4603e+00,
-2.3018e+00, -1.7019e+00],
[-2.0999e+00, -1.3587e+00, -2.9698e+00, ..., -2.8846e+00,
-2.4720e+00, -2.2309e+00]],
[[-1.1459e+00, -1.7038e+00, -1.5216e+00, ..., -2.6358e+00,
-2.2012e+00, -1.7061e+00],
[-1.1520e+00, -1.6242e+00, -1.2108e+00, ..., -2.4732e+00,
-1.9929e+00, -1.6271e+00],
[-1.5066e+00, -1.8042e+00, -1.4200e+00, ..., -2.9073e+00,
-2.3209e+00, -1.8015e+00],
...,
[-2.3199e+00, -2.8922e+00, -2.6900e+00, ..., -4.0715e+00,
-3.3008e+00, -2.5865e+00],
[-1.5622e+00, -1.9814e+00, -1.9234e+00, ..., -2.8096e+00,
-2.3472e+00, -1.9395e+00],
[-1.0472e+00, -1.2519e+00, -1.3054e+00, ..., -1.8581e+00,
-1.5426e+00, -1.1958e+00]]],
[[[ 3.7295e-02, 1.8751e-01, -1.5300e+00, ..., -7.1279e-01,
-2.9337e-01, -1.0411e-01],
[-2.0973e+00, -1.6632e+00, -4.6870e+00, ..., -3.9252e+00,
-1.6497e+00, -6.3676e-01],
[ 5.3069e-01, -1.6083e-01, -9.6228e-01, ..., -2.5701e-01,
-3.6938e-01, 1.7500e-01],
...,
[-8.0732e-01, -1.7542e+00, -1.7988e+00, ..., -1.7372e+00,
-1.2826e+00, -9.4162e-01],
[ 7.6673e-01, 4.9207e-02, -8.6810e-02, ..., 1.4262e-02,
6.9568e-01, 8.0143e-01],
[ 1.6428e-01, -5.1082e-01, -4.2904e-01, ..., -8.3938e-02,
4.6419e-01, 3.8226e-01]],
[[-3.3074e+00, -1.7962e+00, -6.3654e+00, ..., -7.2726e+00,
-6.3643e+00, -4.7076e+00],
[-8.3657e-01, 1.5131e+00, -4.4927e+00, ..., -5.5764e+00,
-5.6449e+00, -4.3568e+00],
[-4.2721e-01, 1.0493e+00, -7.5356e+00, ..., -8.1234e+00,
-7.9761e+00, -6.7731e+00],
...,
[ 2.1779e-01, 3.3984e-01, -4.0428e+00, ..., -4.6652e+00,
-4.1799e+00, -3.4899e+00],
[ 2.2603e+00, 2.2556e+00, -1.5429e-01, ..., -8.1645e-01,
-1.1371e+00, -1.2212e+00],
[ 1.3154e+00, 1.1480e+00, -1.7818e+00, ..., -2.2152e+00,
-2.4773e+00, -2.4076e+00]],
[[-1.3796e+00, -2.3912e+00, -2.3449e+00, ..., -1.8311e+00,
-1.2979e+00, -1.4277e+00],
[-1.7027e+00, -2.0110e+00, -1.5442e+00, ..., -1.3019e+00,
-7.7437e-01, -8.6011e-01],
[-2.2534e+00, -2.0156e+00, -1.0380e+00, ..., -1.1713e+00,
-5.1216e-01, -2.0832e-01],
...,
[-1.5523e-01, -6.1892e-01, -4.3125e-01, ..., -4.8405e-01,
-1.9308e-01, -4.5384e-01],
[-6.9133e-01, -2.0040e-01, -4.1001e-01, ..., -1.7488e-01,
2.5585e-01, 2.4158e-01],
[-8.0199e-01, -2.1377e-01, -3.4776e-01, ..., -3.6193e-01,
-1.7559e-01, -2.1243e-01]],
...,
[[-1.9796e+00, -1.0445e+00, -2.2994e+00, ..., -3.1942e+00,
-2.6162e+00, -1.8246e+00],
[-1.8608e+00, -5.1832e-02, -1.4117e+00, ..., -2.5780e+00,
-2.5785e+00, -1.9977e+00],
[-3.9887e-01, 2.3426e+00, 1.0509e+00, ..., 1.2700e-01,
2.8507e-01, 5.3175e-01],
...,
[-2.0383e+00, -3.9984e-02, -1.8384e-01, ..., -1.3509e+00,
-3.6840e-01, 1.1746e-01],
[-1.1731e+00, 8.1613e-01, 9.2220e-01, ..., -2.8297e-02,
3.1790e-01, 4.7340e-01],
[-2.0904e+00, -1.3985e+00, -1.5894e+00, ..., -2.5001e+00,
-1.4636e+00, -9.0897e-01]],
[[-2.4830e+00, -2.9449e+00, -4.3284e+00, ..., -2.4719e+00,
-1.9770e+00, -1.7891e+00],
[-1.8615e+00, -1.7747e+00, -4.9886e+00, ..., -2.9823e+00,
-2.8470e+00, -2.8251e+00],
[-2.8739e+00, -2.5538e+00, -6.7199e+00, ..., -4.5103e+00,
-4.4483e+00, -3.5315e+00],
...,
[-1.4253e+00, -8.6731e-02, -3.4039e+00, ..., -1.4791e+00,
-1.0311e+00, -9.2432e-01],
[-1.7044e+00, -6.4997e-01, -3.0519e+00, ..., -1.7464e+00,
-1.1648e+00, -8.0784e-01],
[-2.2940e+00, -1.4758e+00, -3.0702e+00, ..., -2.9360e+00,
-2.1165e+00, -1.9643e+00]],
[[-1.8431e+00, -2.5782e+00, -2.5008e+00, ..., -2.7773e+00,
-2.2712e+00, -1.7392e+00],
[-1.8473e+00, -2.5699e+00, -2.2128e+00, ..., -2.4831e+00,
-1.9616e+00, -1.5865e+00],
[-2.3174e+00, -2.9030e+00, -2.6004e+00, ..., -2.8387e+00,
-2.2156e+00, -1.6809e+00],
...,
[-2.3725e+00, -2.9426e+00, -2.7319e+00, ..., -2.8478e+00,
-2.2823e+00, -1.7530e+00],
[-1.5459e+00, -2.0126e+00, -1.9501e+00, ..., -1.9236e+00,
-1.5938e+00, -1.2877e+00],
[-1.0127e+00, -1.3071e+00, -1.3456e+00, ..., -1.1701e+00,
-1.0095e+00, -7.6378e-01]]],
[[[-2.0048e-01, -1.3064e-01, -2.5585e+00, ..., -1.3464e+00,
-8.9966e-01, -4.7901e-01],
[-2.6294e+00, -2.3302e+00, -6.2736e+00, ..., -5.0780e+00,
-2.3730e+00, -1.1016e+00],
[ 2.8192e-01, -6.3095e-01, -1.8248e+00, ..., -6.7735e-01,
-9.4673e-01, -5.1329e-02],
...,
[-3.1191e-01, -1.0120e+00, -1.3226e+00, ..., -2.7315e+00,
-2.6677e+00, -1.8584e+00],
[ 1.0602e+00, 6.1961e-01, 3.6300e-01, ..., 3.9175e-01,
8.6413e-01, 9.1816e-01],
[ 5.5415e-01, 1.6260e-01, 1.2548e-01, ..., 3.6524e-01,
7.5676e-01, 6.1669e-01]],
[[-3.6873e+00, -1.9915e+00, -7.1370e+00, ..., -7.9874e+00,
-7.0717e+00, -5.1443e+00],
[-1.0178e+00, 1.5603e+00, -5.1163e+00, ..., -6.2019e+00,
-6.3728e+00, -4.8139e+00],
[-7.1814e-01, 1.0357e+00, -8.4089e+00, ..., -8.9674e+00,
-8.8974e+00, -7.3066e+00],
...,
[-2.4659e-01, 2.5659e-01, -4.2492e+00, ..., -5.3066e+00,
-4.5253e+00, -4.2213e+00],
[ 2.0054e+00, 2.2901e+00, -1.7877e-01, ..., -9.5796e-01,
-1.2345e+00, -1.3130e+00],
[ 1.1508e+00, 1.1879e+00, -1.7749e+00, ..., -2.3361e+00,
-2.5832e+00, -2.5062e+00]],
[[-1.8774e+00, -2.8107e+00, -2.7196e+00, ..., -2.1842e+00,
-1.5977e+00, -1.7312e+00],
[-2.1422e+00, -2.3048e+00, -1.7315e+00, ..., -1.5475e+00,
-9.8907e-01, -1.1770e+00],
[-3.0128e+00, -2.5669e+00, -1.2162e+00, ..., -1.5014e+00,
-7.3850e-01, -3.3093e-01],
...,
[-5.3622e-01, -9.9029e-01, -4.9332e-01, ..., -2.1003e+00,
-9.7598e-01, -4.2235e-01],
[-9.1846e-01, -4.8333e-01, -3.9770e-01, ..., -1.0900e-01,
2.0495e-01, 1.8992e-01],
[-9.7475e-01, -3.9087e-01, -2.4040e-01, ..., -2.2457e-01,
-1.1421e-01, -1.4895e-01]],
...,
[[-2.0492e+00, -1.0613e+00, -2.3142e+00, ..., -3.1266e+00,
-2.5693e+00, -1.8447e+00],
[-2.5428e+00, -6.8794e-01, -1.8932e+00, ..., -3.0510e+00,
-2.9972e+00, -2.2662e+00],
[-6.0169e-01, 2.2308e+00, 1.0605e+00, ..., 1.4330e-01,
2.2096e-01, 5.0895e-01],
...,
[-1.6092e+00, 6.9318e-01, -1.7634e-02, ..., -4.1346e-01,
3.5114e-01, 2.7382e-01],
[-7.5886e-01, 1.2447e+00, 9.6964e-01, ..., 7.1532e-02,
3.4584e-01, 5.1419e-01],
[-1.7712e+00, -1.1373e+00, -1.4740e+00, ..., -2.3542e+00,
-1.3526e+00, -7.5887e-01]],
[[-2.9046e+00, -3.5650e+00, -4.8649e+00, ..., -3.0339e+00,
-2.5596e+00, -2.3468e+00],
[-2.2891e+00, -2.3896e+00, -5.5106e+00, ..., -3.4921e+00,
-3.4216e+00, -3.3062e+00],
[-3.1900e+00, -3.0229e+00, -6.9853e+00, ..., -4.7664e+00,
-4.9254e+00, -3.9571e+00],
...,
[-1.5424e+00, -4.4029e-01, -3.8582e+00, ..., -2.5770e+00,
-1.9204e+00, -7.5554e-01],
[-1.6589e+00, -7.7212e-01, -3.2783e+00, ..., -1.9673e+00,
-1.3303e+00, -8.6411e-01],
[-2.1518e+00, -1.4453e+00, -3.0353e+00, ..., -2.8841e+00,
-2.0319e+00, -1.8116e+00]],
[[-2.1876e+00, -3.0171e+00, -2.9600e+00, ..., -3.2367e+00,
-2.6559e+00, -2.0522e+00],
[-2.2086e+00, -3.0587e+00, -2.6977e+00, ..., -2.9922e+00,
-2.3617e+00, -1.9210e+00],
[-2.7392e+00, -3.4846e+00, -3.1886e+00, ..., -3.4498e+00,
-2.7253e+00, -2.0982e+00],
...,
[-2.3036e+00, -2.8462e+00, -2.6404e+00, ..., -3.2034e+00,
-2.6343e+00, -2.2102e+00],
[-1.5247e+00, -1.9474e+00, -1.8865e+00, ..., -1.9469e+00,
-1.6105e+00, -1.2918e+00],
[-1.0119e+00, -1.2514e+00, -1.2978e+00, ..., -1.1921e+00,
-1.0268e+00, -7.6538e-01]]],
[[[ 3.8117e-01, 7.2482e-01, 8.1624e-01, ..., 5.9552e-01,
1.0225e+00, 6.5998e-01],
[-1.7572e+00, -9.3671e-01, -2.3815e+00, ..., -2.0889e+00,
-5.1647e-01, 9.0617e-02],
[ 2.1754e-01, -4.0185e-01, 1.2931e-01, ..., 1.7231e-01,
7.3707e-01, 7.7839e-01],
...,
[-1.8399e-01, -1.2819e+00, -1.9536e+01, ..., -1.4562e+00,
-1.4775e+00, -7.9664e-01],
[ 1.4647e+00, 6.6802e-01, -1.1464e+01, ..., -5.6390e-01,
-2.0572e-01, 1.8474e-01],
[ 7.4188e-01, -1.7396e-02, -8.8009e-01, ..., -5.3992e-01,
-8.3440e-02, 2.3575e-01]],
[[-1.8780e+00, -1.1131e+00, -4.1544e+00, ..., -5.0162e+00,
-4.4633e+00, -3.4452e+00],
[ 2.7283e-01, 1.8774e+00, -2.4433e+00, ..., -3.7738e+00,
-3.7627e+00, -3.1025e+00],
[ 1.0036e+00, 1.6548e+00, -5.0787e+00, ..., -5.7234e+00,
-5.4627e+00, -5.1061e+00],
...,
[-1.6093e+00, -8.5539e-02, -1.4716e+00, ..., -8.3846e+00,
-7.7095e+00, -5.9281e+00],
[ 1.5614e+00, 2.7192e+00, 1.0131e-01, ..., -3.1299e+00,
-3.5794e+00, -2.6977e+00],
[ 1.1245e+00, 1.8762e+00, -2.8578e+00, ..., -3.5330e+00,
-3.9656e+00, -3.2900e+00]],
[[-2.3917e-01, -1.2759e+00, -1.3469e+00, ..., -5.5461e-01,
-2.6411e-01, -5.4592e-01],
[-8.2892e-01, -1.1334e+00, -8.5756e-01, ..., -3.9979e-01,
-1.0749e-01, -1.5990e-01],
[-7.4661e-01, -5.8251e-01, -4.0540e-01, ..., -3.4860e-01,
3.1499e-02, 2.9622e-02],
...,
[-1.6599e+00, -1.8452e+00, -2.5008e+00, ..., -8.5516e-01,
-7.7854e-01, -1.0822e+00],
[-1.9362e+00, -1.2220e+00, -1.4927e+00, ..., -5.1537e-01,
-2.5498e-01, -3.6252e-01],
[-2.1043e+00, -1.5454e+00, -3.4140e+00, ..., -1.2645e+00,
-1.1434e+00, -1.1016e+00]],
...,
[[-2.7924e+00, -2.0522e+00, -2.6116e+00, ..., -3.3242e+00,
-2.5804e+00, -1.7207e+00],
[-1.7533e+00, -1.6188e-01, -1.2832e+00, ..., -2.2549e+00,
-2.0935e+00, -1.6594e+00],
[-7.8345e-01, 1.6217e+00, 5.9079e-01, ..., -3.8374e-01,
3.0274e-02, 1.6933e-01],
...,
[-2.0771e+00, 1.0232e+00, -7.4424e+00, ..., -1.1209e+00,
-6.0546e-01, -7.3931e-02],
[-8.7483e-01, 1.9654e+00, -2.3426e+00, ..., 6.5595e-01,
5.0556e-01, 7.7792e-01],
[-2.3360e+00, -1.1343e+00, 1.2698e+00, ..., -1.6389e+00,
-7.4650e-01, -1.0786e-01]],
[[-1.4664e+00, -1.4375e+00, -2.8339e+00, ..., -1.1594e+00,
-7.0039e-01, -6.7727e-01],
[-5.7431e-01, 8.2932e-03, -3.3610e+00, ..., -1.4512e+00,
-1.2511e+00, -1.5497e+00],
[-2.0438e+00, -1.0669e+00, -5.5686e+00, ..., -3.3578e+00,
-2.8473e+00, -2.2792e+00],
...,
[-2.7237e+00, -1.9017e+00, -4.1324e+00, ..., -3.2860e+00,
-3.0052e+00, -2.3811e+00],
[-2.3950e+00, -1.7313e+00, -4.4541e+00, ..., -2.8615e+00,
-2.5936e+00, -1.8855e+00],
[-2.4886e+00, -1.9793e+00, -3.3682e+00, ..., -3.0781e+00,
-2.6487e+00, -2.3119e+00]],
[[-1.2879e+00, -1.8367e+00, -1.6350e+00, ..., -1.7219e+00,
-1.4239e+00, -1.0784e+00],
[-1.2336e+00, -1.7152e+00, -1.2153e+00, ..., -1.3727e+00,
-1.0917e+00, -8.6454e-01],
[-1.5766e+00, -1.9187e+00, -1.4709e+00, ..., -1.6114e+00,
-1.2188e+00, -8.9280e-01],
...,
[-3.4653e+00, -4.3578e+00, -7.2038e+00, ..., -4.3555e+00,
-3.4989e+00, -2.7134e+00],
[-2.2855e+00, -2.9779e+00, -5.0113e+00, ..., -3.0363e+00,
-2.5043e+00, -2.0408e+00],
[-1.6736e+00, -2.0539e+00, -2.7409e+00, ..., -2.0694e+00,
-1.6565e+00, -1.2769e+00]]]], device='cuda:0'),
tensor([[[[ 4.0937e-02, 1.9274e-01, -1.5397e+00, ..., -7.1026e-01,
-2.9068e-01, -1.1823e-01],
[-2.1699e+00, -1.6949e+00, -4.7648e+00, ..., -3.9405e+00,
-1.6610e+00, -6.5081e-01],
[ 5.2620e-01, -1.5471e-01, -9.6803e-01, ..., -2.5241e-01,
-3.6559e-01, 1.8275e-01],
...,
[-2.8034e-01, -1.0308e+00, -1.4232e+00, ..., -1.3188e+00,
-8.7837e-01, -6.5684e-01],
[ 1.0097e+00, 6.6375e-01, 3.3925e-01, ..., 4.0415e-01,
7.2722e-01, 8.0991e-01],
[ 7.1613e-01, 5.1778e-01, 3.5326e-01, ..., 5.1716e-01,
7.8681e-01, 7.0620e-01]],
[[-3.3585e+00, -1.8291e+00, -6.3519e+00, ..., -7.2587e+00,
-6.3546e+00, -4.6871e+00],
[-8.1466e-01, 1.5441e+00, -4.4454e+00, ..., -5.5772e+00,
-5.6385e+00, -4.3344e+00],
[-3.8220e-01, 1.0703e+00, -7.5634e+00, ..., -8.1037e+00,
-7.9492e+00, -6.7544e+00],
...,
[-5.6748e-01, 2.0385e-01, -4.4556e+00, ..., -5.4082e+00,
-4.7448e+00, -3.9160e+00],
[ 1.8587e+00, 2.3962e+00, -8.0187e-02, ..., -1.1571e+00,
-1.3086e+00, -1.3942e+00],
[ 1.0510e+00, 1.2893e+00, -1.7104e+00, ..., -2.5057e+00,
-2.6257e+00, -2.5804e+00]],
[[-1.4166e+00, -2.4167e+00, -2.3700e+00, ..., -1.8201e+00,
-1.2857e+00, -1.4197e+00],
[-1.7527e+00, -2.0200e+00, -1.5393e+00, ..., -1.2983e+00,
-7.6949e-01, -8.6415e-01],
[-2.2571e+00, -1.9746e+00, -1.0086e+00, ..., -1.1548e+00,
-5.0226e-01, -2.0867e-01],
...,
[-6.7822e-01, -1.1368e+00, -5.7231e-01, ..., -5.9609e-01,
-4.6226e-01, -6.1942e-01],
[-1.0547e+00, -7.1717e-01, -5.7078e-01, ..., -2.6632e-01,
-6.2353e-02, -4.0466e-02],
[-1.1925e+00, -6.6829e-01, -3.7248e-01, ..., -3.3665e-01,
-2.8042e-01, -2.4378e-01]],
...,
[[-2.0539e+00, -1.1168e+00, -2.3536e+00, ..., -3.1821e+00,
-2.6151e+00, -1.8295e+00],
[-1.8398e+00, -6.2082e-02, -1.4219e+00, ..., -2.5481e+00,
-2.5650e+00, -2.0005e+00],
[-3.8113e-01, 2.3488e+00, 1.0687e+00, ..., 1.4728e-01,
2.9129e-01, 5.3810e-01],
...,
[-1.4825e+00, 1.0151e+00, -1.2621e-01, ..., -1.1690e+00,
-4.6117e-01, 6.4732e-02],
[-5.2267e-01, 1.4089e+00, 7.6719e-01, ..., -1.4678e-02,
1.5120e-01, 3.0340e-01],
[-1.5990e+00, -1.0985e+00, -1.5436e+00, ..., -2.2883e+00,
-1.3485e+00, -7.1224e-01]],
[[-2.4767e+00, -2.9073e+00, -4.2489e+00, ..., -2.4725e+00,
-1.9817e+00, -1.7823e+00],
[-1.8224e+00, -1.7061e+00, -4.9306e+00, ..., -2.9870e+00,
-2.8441e+00, -2.8205e+00],
[-2.8779e+00, -2.5215e+00, -6.6783e+00, ..., -4.5090e+00,
-4.4223e+00, -3.5369e+00],
...,
[-1.7344e+00, -6.0666e-01, -4.0855e+00, ..., -2.1435e+00,
-1.6019e+00, -1.2397e+00],
[-1.6900e+00, -7.6632e-01, -3.4017e+00, ..., -2.0952e+00,
-1.3669e+00, -8.2263e-01],
[-2.0478e+00, -1.3156e+00, -2.8818e+00, ..., -2.7060e+00,
-1.8076e+00, -1.5412e+00]],
[[-1.8552e+00, -2.5998e+00, -2.5180e+00, ..., -2.7599e+00,
-2.2673e+00, -1.7346e+00],
[-1.8500e+00, -2.5800e+00, -2.2202e+00, ..., -2.4711e+00,
-1.9551e+00, -1.5818e+00],
[-2.3118e+00, -2.8942e+00, -2.6035e+00, ..., -2.8316e+00,
-2.2094e+00, -1.6823e+00],
...,
[-2.2644e+00, -2.8512e+00, -2.6666e+00, ..., -2.8894e+00,
-2.3057e+00, -1.7443e+00],
[-1.5361e+00, -1.9722e+00, -1.9363e+00, ..., -2.0538e+00,
-1.6866e+00, -1.3464e+00],
[-1.0428e+00, -1.2690e+00, -1.3457e+00, ..., -1.2843e+00,
-1.0926e+00, -8.0824e-01]]],
[[[ 4.2322e-01, 8.5810e-01, 5.0132e-01, ..., 7.1068e-01,
9.9301e-01, 7.8378e-01],
[-1.1390e+00, -2.7283e-01, -1.7343e+00, ..., -1.3831e+00,
-2.4816e-02, 5.7715e-01],
[ 7.1287e-01, 4.2673e-01, 3.7001e-02, ..., 4.2141e-01,
7.7628e-01, 8.3127e-01],
...,
[-7.1310e-01, -1.6200e+00, -1.6850e+00, ..., -1.6659e+00,
-1.2167e+00, -8.8488e-01],
[ 8.2504e-01, 1.7801e-01, 2.6617e-02, ..., 8.4003e-02,
7.5542e-01, 8.2480e-01],
[ 2.4348e-01, -3.5181e-01, -3.0474e-01, ..., -1.4576e-02,
5.2073e-01, 4.1259e-01]],
[[-2.7263e+00, -1.5468e+00, -4.9475e+00, ..., -6.0174e+00,
-5.1668e+00, -3.9901e+00],
[-7.3130e-01, 1.2546e+00, -3.5276e+00, ..., -4.8644e+00,
-4.6808e+00, -3.7848e+00],
[-1.0409e-01, 1.0296e+00, -6.3211e+00, ..., -7.0709e+00,
-6.7500e+00, -6.0897e+00],
...,
[ 1.2057e-01, 2.9252e-01, -4.0907e+00, ..., -4.7494e+00,
-4.2704e+00, -3.5505e+00],
[ 2.1733e+00, 2.2226e+00, -1.9472e-01, ..., -8.5689e-01,
-1.1781e+00, -1.2415e+00],
[ 1.2618e+00, 1.1401e+00, -1.7724e+00, ..., -2.2744e+00,
-2.5129e+00, -2.4296e+00]],
[[-5.8169e-01, -1.5605e+00, -1.4411e+00, ..., -8.0493e-01,
-5.4500e-01, -7.7884e-01],
[-1.3156e+00, -1.6637e+00, -1.1496e+00, ..., -7.4124e-01,
-3.9640e-01, -3.5537e-01],
[-1.2222e+00, -1.0697e+00, -5.4568e-01, ..., -5.0232e-01,
-1.2953e-01, -1.2496e-01],
...,
[-2.2910e-01, -6.8105e-01, -4.5037e-01, ..., -5.1500e-01,
-2.2342e-01, -4.7151e-01],
[-7.2973e-01, -2.5098e-01, -4.0910e-01, ..., -1.9207e-01,
2.2807e-01, 2.2150e-01],
[-8.1177e-01, -2.1663e-01, -3.0525e-01, ..., -3.5364e-01,
-1.7237e-01, -2.1823e-01]],
...,
[[-2.0290e+00, -1.1797e+00, -2.4679e+00, ..., -3.3519e+00,
-2.7726e+00, -1.8754e+00],
[-5.4210e-01, 1.0388e+00, -6.7521e-01, ..., -1.7120e+00,
-1.8988e+00, -1.6410e+00],
[-2.3917e-01, 2.0896e+00, 5.3909e-01, ..., -3.4388e-01,
-7.1624e-02, 1.2618e-01],
...,
[-1.9342e+00, 8.3314e-02, -1.4664e-01, ..., -1.2732e+00,
-3.7012e-01, 1.3561e-01],
[-1.1010e+00, 8.6134e-01, 9.1928e-01, ..., 6.5977e-02,
3.1787e-01, 4.6973e-01],
[-2.0293e+00, -1.3873e+00, -1.6127e+00, ..., -2.4033e+00,
-1.4324e+00, -8.8302e-01]],
[[-1.5574e+00, -1.5572e+00, -2.9000e+00, ..., -1.2267e+00,
-7.1371e-01, -5.9451e-01],
[-9.8759e-01, -6.3458e-01, -3.8116e+00, ..., -2.0170e+00,
-1.7853e+00, -1.9663e+00],
[-2.5238e+00, -2.0620e+00, -6.3008e+00, ..., -4.2756e+00,
-3.7663e+00, -2.9788e+00],
...,
[-1.4525e+00, -1.7852e-01, -3.4686e+00, ..., -1.5083e+00,
-1.1020e+00, -9.6755e-01],
[-1.7028e+00, -6.8850e-01, -3.0828e+00, ..., -1.7521e+00,
-1.1851e+00, -7.9798e-01],
[-2.2768e+00, -1.4791e+00, -3.0788e+00, ..., -2.9148e+00,
-2.1047e+00, -1.9301e+00]],
[[-1.2963e+00, -1.8841e+00, -1.7309e+00, ..., -1.9460e+00,
-1.6087e+00, -1.2014e+00],
[-1.3685e+00, -1.8873e+00, -1.5273e+00, ..., -1.7198e+00,
-1.3938e+00, -1.0938e+00],
[-1.7939e+00, -2.1733e+00, -1.8436e+00, ..., -2.0348e+00,
-1.5671e+00, -1.1575e+00],
...,
[-2.3476e+00, -2.9175e+00, -2.7075e+00, ..., -2.8491e+00,
-2.2733e+00, -1.7413e+00],
[-1.5343e+00, -1.9948e+00, -1.9312e+00, ..., -1.9362e+00,
-1.5968e+00, -1.2871e+00],
[-1.0047e+00, -1.2858e+00, -1.3333e+00, ..., -1.1811e+00,
-1.0133e+00, -7.6329e-01]]],
[[[ 2.9713e-01, 5.7203e-01, 6.8869e-01, ..., 4.7509e-01,
1.0033e+00, 5.9806e-01],
[-2.1745e+00, -1.5000e+00, -3.1901e+00, ..., -2.6401e+00,
-8.3861e-01, -1.8860e-01],
[-1.6445e-01, -9.9633e-01, -2.1141e-01, ..., -1.5260e-01,
5.6693e-01, 6.5083e-01],
...,
[ 1.3743e-01, -6.4582e-01, -1.5648e+00, ..., -9.7767e-01,
-7.1341e-01, -4.8899e-01],
[ 1.5503e+00, 1.2452e+00, 5.7859e-01, ..., 7.6242e-01,
9.4002e-01, 9.7148e-01],
[ 1.1656e+00, 9.9188e-01, 6.6482e-01, ..., 9.4364e-01,
1.0307e+00, 8.8678e-01]],
[[-1.6931e+00, -1.1272e+00, -4.1769e+00, ..., -4.8866e+00,
-4.4489e+00, -3.4075e+00],
[ 5.9387e-01, 1.9884e+00, -2.2617e+00, ..., -3.5292e+00,
-3.6425e+00, -2.9922e+00],
[ 1.3568e+00, 1.7262e+00, -4.9393e+00, ..., -5.4844e+00,
-5.3176e+00, -4.9480e+00],
...,
[-6.2484e-01, 5.5156e-01, -4.7284e+00, ..., -5.8410e+00,
-5.2061e+00, -4.2864e+00],
[ 2.1403e+00, 2.8458e+00, 1.1185e-01, ..., -1.4915e+00,
-1.7308e+00, -1.7324e+00],
[ 1.2736e+00, 1.5515e+00, -1.6901e+00, ..., -2.6964e+00,
-2.8742e+00, -2.7714e+00]],
[[-1.2325e-01, -1.1688e+00, -1.3373e+00, ..., -5.0622e-01,
-1.8255e-01, -4.8144e-01],
[-6.1535e-01, -8.8781e-01, -7.9778e-01, ..., -2.7396e-01,
3.0758e-02, -7.0193e-02],
[-5.6273e-01, -3.7442e-01, -4.2938e-01, ..., -3.5834e-01,
8.0469e-02, 8.3455e-02],
...,
[-1.1249e+00, -1.5673e+00, -6.4615e-01, ..., -5.9695e-01,
-4.8297e-01, -6.4880e-01],
[-1.3544e+00, -8.7115e-01, -4.4944e-01, ..., -9.0789e-02,
1.1116e-01, 9.9354e-02],
[-1.3756e+00, -6.6699e-01, -1.2340e-01, ..., -6.4161e-02,
-4.8422e-02, -5.3547e-02]],
...,
[[-3.0445e+00, -2.5092e+00, -2.8385e+00, ..., -3.4477e+00,
-2.5981e+00, -1.7353e+00],
[-2.2733e+00, -8.1139e-01, -1.6042e+00, ..., -2.6008e+00,
-2.2987e+00, -1.7975e+00],
[-1.0278e+00, 1.2269e+00, 6.1931e-01, ..., -4.1906e-01,
1.1895e-02, 1.5190e-01],
...,
[-1.5603e+00, 1.7014e+00, 3.9147e-01, ..., -8.7082e-01,
-1.4642e-01, 3.9686e-01],
[-6.1388e-01, 1.7949e+00, 1.1509e+00, ..., 1.3627e-01,
3.1140e-01, 4.7768e-01],
[-1.7239e+00, -9.9777e-01, -1.3917e+00, ..., -2.1219e+00,
-1.1507e+00, -5.2647e-01]],
[[-1.5906e+00, -1.5648e+00, -2.9044e+00, ..., -1.2853e+00,
-8.3494e-01, -8.3121e-01],
[-5.7907e-01, 1.2075e-01, -3.2261e+00, ..., -1.4106e+00,
-1.2247e+00, -1.5757e+00],
[-2.0835e+00, -8.8663e-01, -5.3808e+00, ..., -3.2433e+00,
-2.7087e+00, -2.2023e+00],
...,
[-1.7864e+00, -6.9860e-01, -4.7047e+00, ..., -2.4346e+00,
-2.0442e+00, -1.6240e+00],
[-1.6801e+00, -7.6730e-01, -3.7323e+00, ..., -2.2674e+00,
-1.6903e+00, -1.1168e+00],
[-2.0572e+00, -1.2895e+00, -2.9855e+00, ..., -2.7518e+00,
-1.9617e+00, -1.6744e+00]],
[[-1.3536e+00, -1.9202e+00, -1.7129e+00, ..., -1.7644e+00,
-1.4608e+00, -1.1131e+00],
[-1.2786e+00, -1.7662e+00, -1.2348e+00, ..., -1.3782e+00,
-1.0870e+00, -8.7134e-01],
[-1.6070e+00, -1.9487e+00, -1.4805e+00, ..., -1.6081e+00,
-1.2053e+00, -8.9600e-01],
...,
[-2.4986e+00, -3.1157e+00, -2.8476e+00, ..., -3.0104e+00,
-2.3998e+00, -1.8357e+00],
[-1.6828e+00, -2.1398e+00, -2.0388e+00, ..., -2.0947e+00,
-1.7137e+00, -1.3774e+00],
[-1.1290e+00, -1.3556e+00, -1.3949e+00, ..., -1.2920e+00,
-1.0940e+00, -8.1091e-01]]],
[[[ 4.3507e-01, 8.5957e-01, 5.4253e-01, ..., 4.3896e-01,
1.0040e+00, 6.4239e-01],
[-1.2849e+00, -4.1702e-01, -1.9414e+00, ..., -2.5901e+00,
-7.6637e-01, -8.5592e-02],
[ 8.9925e-01, 6.3836e-01, 3.9268e-01, ..., -2.5836e-01,
4.7970e-01, 6.2767e-01],
...,
[ 5.9099e-02, -5.7387e-01, -1.3021e+00, ..., -1.9813e+00,
-1.4043e+00, -9.2700e-01],
[ 1.3349e+00, 1.1191e+00, 6.2249e-01, ..., -3.2684e-01,
4.8344e-01, 6.9287e-01],
[ 1.0109e+00, 8.7361e-01, 6.4672e-01, ..., -3.4936e-01,
2.7626e-01, 2.7851e-01]],
[[-2.7828e+00, -1.5342e+00, -4.7972e+00, ..., -4.8498e+00,
-4.4323e+00, -3.3705e+00],
[-4.7858e-01, 1.6257e+00, -2.8927e+00, ..., -3.3992e+00,
-3.5417e+00, -2.9025e+00],
[ 3.3849e-01, 1.3956e+00, -5.7002e+00, ..., -5.3747e+00,
-5.2437e+00, -4.9053e+00],
...,
[-9.8111e-01, -8.9072e-02, -4.9306e+00, ..., -4.5495e+00,
-4.0994e+00, -3.4331e+00],
[ 1.6060e+00, 2.1867e+00, -4.1178e-01, ..., -8.7326e-01,
-1.1877e+00, -1.2387e+00],
[ 9.1357e-01, 1.1795e+00, -1.8637e+00, ..., -2.2070e+00,
-2.4339e+00, -2.3389e+00]],
[[-7.1436e-01, -1.9312e+00, -1.8016e+00, ..., -6.1331e-01,
-2.0398e-01, -4.6291e-01],
[-1.5038e+00, -1.9660e+00, -1.3068e+00, ..., -3.5135e-01,
7.4106e-02, 4.6485e-02],
[-1.3815e+00, -1.3349e+00, -7.2814e-01, ..., -2.7797e-01,
2.5155e-01, 2.9071e-01],
...,
[-8.5320e-01, -1.3750e+00, -6.7233e-01, ..., -4.8036e-01,
-1.5064e-01, -3.4576e-01],
[-1.1155e+00, -8.1956e-01, -5.4598e-01, ..., -3.1172e-01,
1.4895e-01, 2.2078e-01],
[-1.1553e+00, -6.8043e-01, -2.8896e-01, ..., -4.8135e-01,
-2.6919e-01, -2.4443e-01]],
...,
[[-2.3100e+00, -1.1634e+00, -2.5126e+00, ..., -3.6677e+00,
-2.8307e+00, -1.8218e+00],
[-6.1393e-01, 1.2460e+00, -6.9906e-01, ..., -2.6535e+00,
-2.4166e+00, -1.8728e+00],
[-2.0770e-01, 2.5270e+00, 8.2912e-01, ..., -5.3753e-01,
-4.9011e-02, 1.6079e-01],
...,
[-1.3506e+00, 1.2619e+00, 2.2221e-03, ..., -1.5306e+00,
-6.8866e-01, -1.4181e-01],
[-5.3676e-01, 1.4285e+00, 7.3034e-01, ..., -1.5575e-01,
1.2343e-01, 3.2381e-01],
[-1.6889e+00, -1.1277e+00, -1.5494e+00, ..., -2.3853e+00,
-1.4596e+00, -9.1963e-01]],
[[-1.3998e+00, -1.3858e+00, -3.0556e+00, ..., -1.1435e+00,
-7.2795e-01, -7.5192e-01],
[-6.4071e-01, -2.0156e-01, -3.9080e+00, ..., -1.3723e+00,
-1.2223e+00, -1.5796e+00],
[-2.1682e+00, -1.5426e+00, -6.4235e+00, ..., -3.2787e+00,
-2.7906e+00, -2.2229e+00],
...,
[-1.8325e+00, -9.9210e-01, -4.5561e+00, ..., -1.4765e+00,
-1.0198e+00, -8.6205e-01],
[-1.7368e+00, -1.0353e+00, -3.6974e+00, ..., -1.7388e+00,
-1.1520e+00, -7.4852e-01],
[-2.0477e+00, -1.4703e+00, -3.0295e+00, ..., -2.8117e+00,
-2.0209e+00, -1.8262e+00]],
[[-1.2620e+00, -1.9126e+00, -1.7390e+00, ..., -1.8043e+00,
-1.4938e+00, -1.1394e+00],
[-1.2611e+00, -1.8160e+00, -1.3932e+00, ..., -1.3993e+00,
-1.1096e+00, -8.8579e-01],
[-1.6277e+00, -1.9907e+00, -1.6174e+00, ..., -1.5981e+00,
-1.2070e+00, -8.9279e-01],
...,
[-2.3132e+00, -2.9413e+00, -2.8524e+00, ..., -2.8655e+00,
-2.2871e+00, -1.7608e+00],
[-1.5638e+00, -2.0137e+00, -2.0370e+00, ..., -1.9703e+00,
-1.6173e+00, -1.3062e+00],
[-1.0612e+00, -1.3138e+00, -1.4278e+00, ..., -1.2200e+00,
-1.0321e+00, -7.9240e-01]]],
[[[ 4.0685e-01, 8.2059e-01, 4.5225e-01, ..., -9.1046e-01,
-6.4889e-01, -3.0276e-01],
[-1.1865e+00, -2.6558e-01, -1.6618e+00, ..., -4.2155e+00,
-1.9754e+00, -8.6037e-01],
[ 6.6521e-01, 4.1335e-01, 2.0206e-01, ..., -5.8415e-01,
-9.6170e-01, -8.3583e-02],
...,
[-1.7367e-01, -8.7427e-01, -1.7165e+00, ..., -2.8891e+01,
-1.7935e+01, -1.2200e+01],
[ 1.2467e+00, 9.6085e-01, 1.7034e-01, ..., 9.5315e-01,
-1.9987e+00, -1.2052e+00],
[ 9.1117e-01, 7.3272e-01, 2.6879e-01, ..., 7.7986e-01,
1.3822e-01, 3.3238e-01]],
[[-2.5346e+00, -1.2708e+00, -4.6946e+00, ..., -7.4933e+00,
-6.6815e+00, -4.9342e+00],
[-4.6110e-01, 1.5744e+00, -3.2003e+00, ..., -6.0397e+00,
-6.2379e+00, -4.7708e+00],
[ 1.9687e-01, 1.3413e+00, -5.8438e+00, ..., -8.6555e+00,
-8.6765e+00, -7.1230e+00],
...,
[-6.9854e-01, 5.0431e-01, -4.8167e+00, ..., -1.4958e+01,
-1.3390e+01, -1.1002e+01],
[ 1.9877e+00, 2.7823e+00, -6.5481e-02, ..., -6.2822e+00,
-3.2773e+00, -2.5461e+00],
[ 1.1603e+00, 1.5418e+00, -1.7631e+00, ..., -4.0626e+00,
-3.9048e+00, -3.2421e+00]],
[[-6.8104e-01, -1.6184e+00, -1.4789e+00, ..., -1.7159e+00,
-1.3147e+00, -1.5950e+00],
[-1.3625e+00, -1.6645e+00, -1.1181e+00, ..., -1.1883e+00,
-8.1162e-01, -1.1776e+00],
[-1.3117e+00, -1.0905e+00, -4.2783e-01, ..., -1.3797e+00,
-8.1910e-01, -5.8870e-01],
...,
[-1.0559e+00, -1.4569e+00, -6.3855e-01, ..., -3.0408e+00,
-4.7768e+00, -3.0635e+00],
[-1.3456e+00, -8.9528e-01, -5.9946e-01, ..., -1.6166e-02,
-1.6736e+00, -9.0006e-01],
[-1.4005e+00, -7.6749e-01, -3.7168e-01, ..., -3.3918e-01,
-8.4045e-01, -9.9848e-01]],
...,
[[-2.0777e+00, -1.1593e+00, -2.4560e+00, ..., -3.0022e+00,
-2.4073e+00, -1.8032e+00],
[-7.1240e-01, 1.0406e+00, -6.6896e-01, ..., -2.9038e+00,
-2.8637e+00, -2.2164e+00],
[-2.2054e-01, 2.3105e+00, 6.6227e-01, ..., -9.6892e-02,
-3.7624e-02, 2.4087e-01],
...,
[-1.4545e+00, 1.3578e+00, 4.1707e-02, ..., 2.7689e+00,
-6.2376e+00, -4.4974e+00],
[-5.4197e-01, 1.6793e+00, 1.0537e+00, ..., 3.1068e+00,
1.1150e+00, 1.2705e+00],
[-1.7167e+00, -1.0403e+00, -1.3564e+00, ..., -2.1612e+00,
-8.9279e-01, -1.2039e-01]],
[[-1.5857e+00, -1.5125e+00, -2.7817e+00, ..., -2.7579e+00,
-2.4655e+00, -2.3571e+00],
[-9.5964e-01, -4.7104e-01, -3.5870e+00, ..., -3.3599e+00,
-3.4045e+00, -3.3410e+00],
[-2.3891e+00, -1.7550e+00, -5.9748e+00, ..., -4.6980e+00,
-4.9313e+00, -4.0873e+00],
...,
[-1.9099e+00, -8.6013e-01, -4.4717e+00, ..., -5.5335e-01,
-2.3100e+00, -1.8000e+00],
[-1.8251e+00, -9.3436e-01, -3.6537e+00, ..., -1.7806e+00,
-3.1700e+00, -1.9165e+00],
[-2.1785e+00, -1.4591e+00, -3.0222e+00, ..., -3.0767e+00,
-2.6782e+00, -2.4352e+00]],
[[-1.2789e+00, -1.8340e+00, -1.6393e+00, ..., -2.9918e+00,
-2.4926e+00, -1.9452e+00],
[-1.2977e+00, -1.7938e+00, -1.3635e+00, ..., -2.8742e+00,
-2.3090e+00, -1.9018e+00],
[-1.6874e+00, -2.0147e+00, -1.6121e+00, ..., -3.4038e+00,
-2.7298e+00, -2.1439e+00],
...,
[-2.4976e+00, -3.0955e+00, -2.8436e+00, ..., -8.9799e+00,
-6.3111e+00, -5.1335e+00],
[-1.6827e+00, -2.1364e+00, -2.0514e+00, ..., -4.1701e+00,
-2.9407e+00, -2.4986e+00],
[-1.1435e+00, -1.3649e+00, -1.4072e+00, ..., -2.2099e+00,
-1.6872e+00, -1.3193e+00]]]], device='cuda:0'),
tensor([[[[-4.6825e-02, 1.0484e-01, -1.5049e+00, ..., 5.3187e-01,
8.4449e-01, 6.2066e-01],
[-2.8371e+00, -2.1631e+00, -4.1812e+00, ..., -1.5534e+00,
-1.3460e-01, 4.3322e-01],
[-1.5697e+01, -2.1960e+01, -3.1252e+01, ..., 4.8971e-01,
7.3371e-01, 7.4490e-01],
...,
[-2.6818e-01, -1.4418e+00, -2.9696e+00, ..., -1.0948e+00,
-8.9780e-01, -6.4261e-01],
[ 1.3272e+00, 4.6362e-01, -1.5499e+00, ..., 6.4322e-01,
8.2985e-01, 8.8074e-01],
[ 7.1089e-01, -1.0832e-01, -9.8738e-01, ..., 7.8181e-01,
9.1831e-01, 8.0473e-01]],
[[-3.5803e+00, -1.6527e+00, -6.9451e+00, ..., -5.7643e+00,
-4.9950e+00, -3.8156e+00],
[-9.6708e-01, 1.8520e+00, -5.7014e+00, ..., -4.5114e+00,
-4.4085e+00, -3.5299e+00],
[-4.5209e+00, -5.5257e+00, -1.9305e+01, ..., -6.5620e+00,
-6.3186e+00, -5.7105e+00],
...,
[-1.6939e+00, 1.2069e-03, -7.5111e+00, ..., -5.7654e+00,
-5.0876e+00, -4.2142e+00],
[ 1.2151e+00, 2.4991e+00, -1.9767e+00, ..., -1.4714e+00,
-1.6734e+00, -1.6765e+00],
[ 8.8422e-01, 1.7054e+00, -2.6995e+00, ..., -2.5962e+00,
-2.7786e+00, -2.6934e+00]],
[[-1.8029e+00, -2.6844e+00, -2.2700e+00, ..., -9.8233e-01,
-6.7709e-01, -9.2033e-01],
[-2.4127e+00, -2.4909e+00, -1.5162e+00, ..., -7.6014e-01,
-3.9726e-01, -4.3145e-01],
[-4.6686e+00, -5.5597e+00, -4.1793e+00, ..., -5.7023e-01,
-2.0979e-01, -1.9962e-01],
...,
[-1.9194e+00, -1.9653e+00, -6.7562e-01, ..., -6.4222e-01,
-5.4537e-01, -8.1115e-01],
[-1.9980e+00, -1.2792e+00, -6.4734e-01, ..., -2.3810e-01,
-5.3694e-02, -1.5379e-01],
[-1.9843e+00, -1.4362e+00, -8.5058e-01, ..., -2.2426e-01,
-1.8755e-01, -2.5802e-01]],
...,
[[-2.3334e+00, -1.2671e+00, -2.5105e+00, ..., -3.0747e+00,
-2.4720e+00, -1.6878e+00],
[-2.3817e+00, -4.9556e-01, 3.2640e-02, ..., -1.5680e+00,
-1.7225e+00, -1.5082e+00],
[ 2.4800e-01, 9.8893e-01, 2.6389e-01, ..., -1.4306e-01,
1.5534e-01, 3.2168e-01],
...,
[-1.9848e+00, 9.9015e-01, 1.9364e-01, ..., -9.7193e-01,
-2.7302e-01, 2.0286e-01],
[-9.5366e-01, 1.8006e+00, 1.8512e+00, ..., 3.2515e-02,
2.1741e-01, 4.1573e-01],
[-2.2687e+00, -9.8787e-01, -4.9557e-01, ..., -2.1883e+00,
-1.2590e+00, -5.8121e-01]],
[[-2.7350e+00, -3.2019e+00, -4.5672e+00, ..., -1.2090e+00,
-8.0119e-01, -7.3221e-01],
[-2.0417e+00, -1.7623e+00, -3.7642e+00, ..., -1.9318e+00,
-1.7938e+00, -1.9678e+00],
[-1.0004e+00, 6.4444e-01, -1.4804e+00, ..., -4.0005e+00,
-3.6205e+00, -2.8926e+00],
...,
[-2.8540e+00, -2.1662e+00, -5.0837e+00, ..., -2.5161e+00,
-2.0987e+00, -1.7151e+00],
[-2.4376e+00, -1.8866e+00, -3.9953e+00, ..., -2.3013e+00,
-1.7598e+00, -1.2048e+00],
[-2.4069e+00, -2.0230e+00, -3.2020e+00, ..., -2.7453e+00,
-2.0217e+00, -1.7627e+00]],
[[-2.1807e+00, -2.9832e+00, -2.8821e+00, ..., -1.8544e+00,
-1.5432e+00, -1.1895e+00],
[-2.3026e+00, -3.1209e+00, -3.6040e+00, ..., -1.5832e+00,
-1.2980e+00, -1.0579e+00],
[-6.0397e+00, -6.4837e+00, -7.2335e+00, ..., -1.8242e+00,
-1.4307e+00, -1.0841e+00],
...,
[-3.4721e+00, -4.3734e+00, -4.1322e+00, ..., -2.9673e+00,
-2.3873e+00, -1.8424e+00],
[-2.2978e+00, -2.9984e+00, -2.9621e+00, ..., -2.0762e+00,
-1.7231e+00, -1.4013e+00],
[-1.6796e+00, -2.0618e+00, -2.0874e+00, ..., -1.3202e+00,
-1.1273e+00, -8.5277e-01]]],
[[[ 3.5791e-01, 8.1018e-01, 7.4487e-01, ..., 6.9901e-01,
9.4687e-01, 7.3437e-01],
[-1.2008e+00, -2.9411e-01, -1.3794e+00, ..., -1.0587e+00,
7.4607e-02, 5.6209e-01],
[ 6.4900e-01, 4.3010e-01, 5.0811e-01, ..., 6.2647e-01,
9.4916e-01, 8.7544e-01],
...,
[-1.7035e-01, -1.1539e+00, -2.6653e+00, ..., -1.4699e+00,
-1.4104e+00, -8.5415e-01],
[ 1.4300e+00, 8.1946e-01, -9.4322e-01, ..., -2.3020e-01,
1.0191e-01, 3.9239e-01],
[ 8.5497e-01, 2.9781e-01, -6.6345e-01, ..., -1.4427e-01,
2.1472e-01, 3.9192e-01]],
[[-2.3092e+00, -1.2040e+00, -4.2317e+00, ..., -5.2144e+00,
-4.4741e+00, -3.4981e+00],
[-1.2583e-01, 1.7192e+00, -2.5281e+00, ..., -4.0985e+00,
-3.9649e+00, -3.2592e+00],
[ 7.3932e-01, 1.5665e+00, -5.0490e+00, ..., -5.9250e+00,
-5.6906e+00, -5.2630e+00],
...,
[-1.2679e+00, 1.6035e-01, -6.6792e+00, ..., -7.6134e+00,
-6.8779e+00, -5.4301e+00],
[ 1.8147e+00, 2.8205e+00, -1.0340e+00, ..., -2.5911e+00,
-2.9307e+00, -2.3930e+00],
[ 1.2209e+00, 1.8125e+00, -2.2695e+00, ..., -3.2591e+00,
-3.6199e+00, -3.1424e+00]],
[[-6.4997e-01, -1.7914e+00, -1.6687e+00, ..., -9.1083e-01,
-6.3584e-01, -7.5174e-01],
[-1.3664e+00, -1.8426e+00, -1.2730e+00, ..., -7.3499e-01,
-3.7768e-01, -2.5954e-01],
[-1.0583e+00, -1.0019e+00, -4.7907e-01, ..., -3.1586e-01,
3.6591e-02, 5.5981e-02],
...,
[-1.3755e+00, -1.7521e+00, -7.5865e-01, ..., -8.8311e-01,
-7.7436e-01, -1.0226e+00],
[-1.7262e+00, -1.1610e+00, -8.3652e-01, ..., -5.6080e-01,
-2.9889e-01, -3.6889e-01],
[-1.8765e+00, -1.2818e+00, -8.9826e-01, ..., -9.4630e-01,
-8.0913e-01, -7.9263e-01]],
...,
[[-2.2206e+00, -1.2211e+00, -2.2756e+00, ..., -2.8680e+00,
-2.2471e+00, -1.4737e+00],
[-6.3254e-01, 1.1042e+00, -3.8522e-01, ..., -1.3280e+00,
-1.3889e+00, -1.2182e+00],
[-3.5232e-01, 2.1359e+00, 8.7656e-01, ..., -2.3016e-01,
1.8830e-01, 3.2686e-01],
...,
[-1.8356e+00, 1.2838e+00, 2.7420e-02, ..., -1.1265e+00,
-5.0872e-01, 4.1153e-03],
[-8.0530e-01, 1.9221e+00, 1.6151e+00, ..., 4.3275e-01,
4.2159e-01, 6.8140e-01],
[-2.2193e+00, -1.1440e+00, -9.8922e-01, ..., -1.8840e+00,
-9.7930e-01, -3.0155e-01]],
[[-1.3561e+00, -1.2518e+00, -2.7674e+00, ..., -1.1278e+00,
-6.0420e-01, -4.6803e-01],
[-6.6163e-01, -1.3423e-01, -3.5654e+00, ..., -1.7754e+00,
-1.5134e+00, -1.6750e+00],
[-2.0788e+00, -1.3126e+00, -5.9041e+00, ..., -3.7006e+00,
-3.1810e+00, -2.5309e+00],
...,
[-2.4437e+00, -1.5457e+00, -5.4730e+00, ..., -3.0967e+00,
-2.7336e+00, -2.1805e+00],
[-2.1985e+00, -1.4364e+00, -4.2504e+00, ..., -2.6780e+00,
-2.3114e+00, -1.6453e+00],
[-2.3844e+00, -1.7438e+00, -3.2778e+00, ..., -2.9650e+00,
-2.4349e+00, -2.1174e+00]],
[[-1.1603e+00, -1.7156e+00, -1.4951e+00, ..., -1.6318e+00,
-1.3578e+00, -1.0232e+00],
[-1.1703e+00, -1.6429e+00, -1.1652e+00, ..., -1.3640e+00,
-1.1158e+00, -8.7853e-01],
[-1.5321e+00, -1.8258e+00, -1.3497e+00, ..., -1.5605e+00,
-1.1914e+00, -8.7126e-01],
...,
[-3.1748e+00, -3.9661e+00, -3.7551e+00, ..., -3.9328e+00,
-3.1572e+00, -2.4318e+00],
[-2.1134e+00, -2.7305e+00, -2.7008e+00, ..., -2.7621e+00,
-2.2791e+00, -1.8539e+00],
[-1.5365e+00, -1.8658e+00, -1.9372e+00, ..., -1.8657e+00,
-1.5157e+00, -1.1608e+00]]],
[[[ 1.6522e-01, 3.9021e-01, -1.1330e+00, ..., 6.4207e-01,
8.9880e-01, 6.7615e-01],
[-2.0848e+00, -1.4191e+00, -4.2711e+00, ..., -1.4155e+00,
-1.7312e-01, 3.8324e-01],
[ 4.3327e-01, -2.5502e-01, -1.0106e+00, ..., 5.1885e-01,
9.1328e-01, 8.2917e-01],
...,
[-2.5719e-01, -1.2721e+00, -2.6047e+00, ..., -1.3984e+00,
-9.9821e-01, -8.3232e-01],
[ 1.2472e+00, 6.4051e-01, -9.1773e-01, ..., 4.8028e-01,
7.5044e-01, 7.6446e-01],
[ 8.3344e-01, 3.3003e-01, -4.0175e-01, ..., 5.5825e-01,
8.1962e-01, 6.7449e-01]],
[[-3.2120e+00, -1.4502e+00, -6.1430e+00, ..., -5.3186e+00,
-4.5706e+00, -3.5850e+00],
[-5.7106e-01, 2.1216e+00, -4.2260e+00, ..., -4.1470e+00,
-4.0381e+00, -3.3219e+00],
[ 2.6952e-02, 1.8223e+00, -7.4313e+00, ..., -6.0221e+00,
-5.8172e+00, -5.3488e+00],
...,
[-1.2642e+00, 2.5163e-01, -6.4875e+00, ..., -5.3742e+00,
-4.7228e+00, -3.8743e+00],
[ 1.5319e+00, 2.6204e+00, -1.2621e+00, ..., -9.5136e-01,
-1.1568e+00, -1.2542e+00],
[ 1.0022e+00, 1.6367e+00, -2.3522e+00, ..., -2.4037e+00,
-2.5699e+00, -2.5471e+00]],
[[-1.3625e+00, -2.3303e+00, -2.1590e+00, ..., -7.9666e-01,
-6.0645e-01, -7.7802e-01],
[-1.9734e+00, -2.1520e+00, -1.5210e+00, ..., -6.4139e-01,
-4.1259e-01, -3.5314e-01],
[-2.4673e+00, -1.9581e+00, -8.0682e-01, ..., -3.7530e-01,
-1.0759e-01, -1.0217e-01],
...,
[-1.5755e+00, -1.8177e+00, -7.5917e-01, ..., -6.6532e-01,
-5.5853e-01, -7.5665e-01],
[-1.7491e+00, -1.1989e+00, -7.8908e-01, ..., -2.5167e-01,
-3.2265e-02, -2.4766e-02],
[-1.7110e+00, -1.1161e+00, -6.0918e-01, ..., -3.3496e-01,
-2.8589e-01, -2.6796e-01]],
...,
[[-2.3792e+00, -1.3717e+00, -2.7895e+00, ..., -2.9042e+00,
-2.2956e+00, -1.5663e+00],
[-1.8666e+00, 1.6166e-01, -1.3782e+00, ..., -1.4379e+00,
-1.4861e+00, -1.2917e+00],
[-5.3543e-01, 2.4033e+00, 8.1430e-01, ..., -5.9495e-02,
2.0432e-01, 3.0338e-01],
...,
[-1.7740e+00, 1.0838e+00, 1.3125e-01, ..., -1.1051e+00,
-2.9212e-01, 2.5135e-01],
[-8.8008e-01, 1.6195e+00, 1.4383e+00, ..., 1.8992e-01,
3.4962e-01, 4.6326e-01],
[-2.1206e+00, -1.0811e+00, -8.8231e-01, ..., -2.3699e+00,
-1.3781e+00, -7.2064e-01]],
[[-2.3485e+00, -2.6259e+00, -4.0007e+00, ..., -1.1702e+00,
-6.7068e-01, -5.3564e-01],
[-1.6475e+00, -1.3542e+00, -4.7646e+00, ..., -1.7134e+00,
-1.5076e+00, -1.7070e+00],
[-2.9901e+00, -2.4683e+00, -6.9278e+00, ..., -3.6861e+00,
-3.2117e+00, -2.5971e+00],
...,
[-2.5204e+00, -1.7071e+00, -4.9253e+00, ..., -2.0458e+00,
-1.6032e+00, -1.3071e+00],
[-2.2216e+00, -1.5429e+00, -3.8584e+00, ..., -2.0719e+00,
-1.3831e+00, -8.5994e-01],
[-2.3158e+00, -1.7741e+00, -3.0714e+00, ..., -2.9067e+00,
-1.9590e+00, -1.6905e+00]],
[[-1.8819e+00, -2.6021e+00, -2.4006e+00, ..., -1.6622e+00,
-1.3822e+00, -1.0350e+00],
[-1.9105e+00, -2.6171e+00, -2.0937e+00, ..., -1.3649e+00,
-1.1167e+00, -8.8652e-01],
[-2.4319e+00, -2.9850e+00, -2.5093e+00, ..., -1.5885e+00,
-1.2179e+00, -8.9923e-01],
...,
[-3.0919e+00, -3.8708e+00, -3.6506e+00, ..., -2.9037e+00,
-2.3244e+00, -1.7409e+00],
[-2.0491e+00, -2.6584e+00, -2.6175e+00, ..., -2.0179e+00,
-1.6632e+00, -1.3227e+00],
[-1.4665e+00, -1.7885e+00, -1.8406e+00, ..., -1.2297e+00,
-1.0613e+00, -7.7612e-01]]],
[[[ 4.3244e-01, 8.6379e-01, 4.8857e-01, ..., 4.7868e-01,
1.0188e+00, 6.7006e-01],
[-1.3150e+00, -4.5386e-01, -2.0448e+00, ..., -2.4807e+00,
-6.8951e-01, 3.1069e-05],
[ 9.2219e-01, 6.4512e-01, 3.5111e-01, ..., -2.1262e-01,
5.4117e-01, 6.7707e-01],
...,
[-1.8448e+00, -2.0167e+00, -4.8050e+00, ..., -1.9863e+00,
-1.3410e+00, -8.9703e-01],
[ 4.8401e-01, 5.1135e-01, 7.4106e-01, ..., -3.0162e-01,
5.3559e-01, 7.0182e-01],
[ 9.9963e-01, 9.1800e-01, 6.1646e-01, ..., -3.1814e-01,
3.2759e-01, 3.0243e-01]],
[[-2.8314e+00, -1.5632e+00, -4.9003e+00, ..., -4.8875e+00,
-4.4293e+00, -3.4009e+00],
[-5.3675e-01, 1.6485e+00, -2.9609e+00, ..., -3.4203e+00,
-3.5688e+00, -2.9151e+00],
[ 2.7332e-01, 1.4311e+00, -5.7550e+00, ..., -5.3944e+00,
-5.3093e+00, -4.9470e+00],
...,
[-8.5708e-01, 4.0881e-01, -6.1366e+00, ..., -4.5918e+00,
-4.1613e+00, -3.4872e+00],
[ 1.8673e+00, 2.3019e+00, -7.3531e-01, ..., -8.5747e-01,
-1.2148e+00, -1.2655e+00],
[ 9.5458e-01, 1.2168e+00, -1.8422e+00, ..., -2.2430e+00,
-2.4788e+00, -2.3753e+00]],
[[-7.1163e-01, -1.9287e+00, -1.7892e+00, ..., -6.0760e-01,
-2.3469e-01, -4.8926e-01],
[-1.4875e+00, -1.9694e+00, -1.2941e+00, ..., -3.4682e-01,
4.4802e-02, 2.9386e-02],
[-1.4132e+00, -1.3943e+00, -7.5373e-01, ..., -2.6610e-01,
2.5072e-01, 2.7925e-01],
...,
[-1.7893e+00, -8.1905e-01, -3.0678e-01, ..., -5.0768e-01,
-1.8175e-01, -3.6118e-01],
[-1.5597e+00, -8.2577e-01, -6.8784e-01, ..., -3.2926e-01,
1.4370e-01, 2.1918e-01],
[-1.2230e+00, -7.3693e-01, -3.3493e-01, ..., -4.9329e-01,
-2.7337e-01, -2.4115e-01]],
...,
[[-2.3126e+00, -1.1687e+00, -2.5450e+00, ..., -3.7100e+00,
-2.8545e+00, -1.8610e+00],
[-6.6209e-01, 1.2399e+00, -7.7435e-01, ..., -2.6460e+00,
-2.4000e+00, -1.8405e+00],
[-2.1298e-01, 2.5966e+00, 7.9725e-01, ..., -5.6408e-01,
-1.0765e-01, 1.5987e-01],
...,
[-1.8623e+00, 5.5159e-01, 1.0405e+00, ..., -1.4760e+00,
-6.8604e-01, -1.1691e-01],
[ 1.7584e-01, 1.4216e+00, 8.5560e-01, ..., -8.0443e-02,
1.1966e-01, 3.2197e-01],
[-1.6689e+00, -1.1308e+00, -1.5151e+00, ..., -2.3269e+00,
-1.4313e+00, -9.0732e-01]],
[[-1.4458e+00, -1.4482e+00, -3.1398e+00, ..., -1.1050e+00,
-7.0169e-01, -7.3112e-01],
[-6.9470e-01, -2.6089e-01, -3.9801e+00, ..., -1.3541e+00,
-1.2139e+00, -1.6136e+00],
[-2.2013e+00, -1.5766e+00, -6.4821e+00, ..., -3.2992e+00,
-2.8131e+00, -2.2930e+00],
...,
[-2.8053e+00, -3.2868e-01, -3.6155e+00, ..., -1.4628e+00,
-1.0638e+00, -9.0130e-01],
[-2.1961e+00, -7.9031e-01, -3.2175e+00, ..., -1.7275e+00,
-1.1570e+00, -7.5459e-01],
[-2.0519e+00, -1.4700e+00, -3.0324e+00, ..., -2.8059e+00,
-2.0053e+00, -1.8070e+00]],
[[-1.2827e+00, -1.9411e+00, -1.7794e+00, ..., -1.7986e+00,
-1.5003e+00, -1.1390e+00],
[-1.2792e+00, -1.8477e+00, -1.4283e+00, ..., -1.4057e+00,
-1.1131e+00, -8.9834e-01],
[-1.6515e+00, -2.0397e+00, -1.6543e+00, ..., -1.6095e+00,
-1.2053e+00, -9.0049e-01],
...,
[-2.6535e+00, -3.7833e+00, -3.7607e+00, ..., -2.8730e+00,
-2.2942e+00, -1.7658e+00],
[-1.6911e+00, -2.2851e+00, -2.3801e+00, ..., -1.9848e+00,
-1.6294e+00, -1.3130e+00],
[-1.0821e+00, -1.3303e+00, -1.4475e+00, ..., -1.2264e+00,
-1.0360e+00, -7.9198e-01]]],
[[[ 2.8135e-01, 5.5315e-01, 6.6605e-01, ..., 4.9707e-01,
1.0458e+00, 6.1345e-01],
[-2.0929e+00, -1.4493e+00, -3.0734e+00, ..., -2.7150e+00,
-8.8342e-01, -1.9205e-01],
[-1.7423e-01, -9.2300e-01, -1.9762e-01, ..., -1.3087e-01,
5.4944e-01, 6.4083e-01],
...,
[-1.5885e-01, -1.1452e+00, -2.6704e+00, ..., -1.5323e+00,
-1.4157e+00, -8.5287e-01],
[ 1.4475e+00, 8.5166e-01, -9.7538e-01, ..., -3.2822e-01,
5.3524e-02, 3.3971e-01],
[ 8.3281e-01, 2.6059e-01, -7.3686e-01, ..., -2.2397e-01,
1.8338e-01, 3.5613e-01]],
[[-1.7191e+00, -1.3006e+00, -4.1912e+00, ..., -5.0560e+00,
-4.5757e+00, -3.4743e+00],
[ 4.0281e-01, 1.6098e+00, -2.4158e+00, ..., -3.5578e+00,
-3.6842e+00, -2.9938e+00],
[ 1.0593e+00, 1.3397e+00, -4.9942e+00, ..., -5.5999e+00,
-5.4101e+00, -5.0494e+00],
...,
[-1.3154e+00, 1.1930e-01, -6.8135e+00, ..., -7.7193e+00,
-7.0061e+00, -5.4834e+00],
[ 1.7352e+00, 2.7520e+00, -1.1748e+00, ..., -2.6687e+00,
-3.0222e+00, -2.4090e+00],
[ 1.1834e+00, 1.7904e+00, -2.3552e+00, ..., -3.2916e+00,
-3.6671e+00, -3.1533e+00]],
[[-1.2408e-01, -1.0665e+00, -1.2528e+00, ..., -5.6095e-01,
-2.0220e-01, -5.3013e-01],
[-5.4146e-01, -7.5926e-01, -7.1890e-01, ..., -3.4364e-01,
7.8942e-03, -1.0188e-01],
[-4.6155e-01, -2.7456e-01, -3.7657e-01, ..., -3.6330e-01,
8.3666e-02, 8.3703e-02],
...,
[-1.4049e+00, -1.7582e+00, -7.4772e-01, ..., -8.9920e-01,
-7.7627e-01, -1.0327e+00],
[-1.7447e+00, -1.1616e+00, -8.1657e-01, ..., -5.6757e-01,
-2.9388e-01, -3.6618e-01],
[-1.9103e+00, -1.3077e+00, -9.1123e-01, ..., -1.0076e+00,
-8.6437e-01, -8.4092e-01]],
...,
[[-2.8734e+00, -2.4710e+00, -2.7305e+00, ..., -3.6455e+00,
-2.7642e+00, -1.8313e+00],
[-2.1596e+00, -8.9979e-01, -1.5779e+00, ..., -2.6870e+00,
-2.3981e+00, -1.8727e+00],
[-1.0465e+00, 9.8191e-01, 5.0995e-01, ..., -4.4114e-01,
6.0876e-02, 1.9788e-01],
...,
[-1.8549e+00, 1.2845e+00, 6.6505e-02, ..., -1.0766e+00,
-5.4738e-01, -1.2231e-02],
[-8.5084e-01, 1.8969e+00, 1.6808e+00, ..., 5.2704e-01,
4.2816e-01, 6.9563e-01],
[-2.2218e+00, -1.1486e+00, -9.4137e-01, ..., -1.8157e+00,
-9.4837e-01, -2.8023e-01]],
[[-1.5914e+00, -1.5919e+00, -2.7241e+00, ..., -1.1725e+00,
-7.1968e-01, -7.4058e-01],
[-6.6622e-01, -3.7755e-02, -3.0309e+00, ..., -1.3330e+00,
-1.1623e+00, -1.5517e+00],
[-2.1346e+00, -1.0414e+00, -5.1138e+00, ..., -3.2750e+00,
-2.7490e+00, -2.2205e+00],
...,
[-2.5181e+00, -1.6524e+00, -5.5391e+00, ..., -3.0887e+00,
-2.7498e+00, -2.1954e+00],
[-2.2292e+00, -1.4906e+00, -4.2919e+00, ..., -2.6875e+00,
-2.3352e+00, -1.6581e+00],
[-2.3583e+00, -1.7631e+00, -3.3243e+00, ..., -2.9727e+00,
-2.4604e+00, -2.1384e+00]],
[[-1.3103e+00, -1.8559e+00, -1.6993e+00, ..., -1.8217e+00,
-1.5062e+00, -1.1386e+00],
[-1.2382e+00, -1.7135e+00, -1.2519e+00, ..., -1.3855e+00,
-1.0937e+00, -8.7358e-01],
[-1.5429e+00, -1.8943e+00, -1.5018e+00, ..., -1.5896e+00,
-1.1930e+00, -8.7571e-01],
...,
[-3.2163e+00, -4.0341e+00, -3.8183e+00, ..., -3.9929e+00,
-3.2014e+00, -2.4718e+00],
[-2.1569e+00, -2.7868e+00, -2.7630e+00, ..., -2.7984e+00,
-2.3088e+00, -1.8799e+00],
[-1.5845e+00, -1.9176e+00, -1.9809e+00, ..., -1.8884e+00,
-1.5322e+00, -1.1747e+00]]]], device='cuda:0'),
tensor([[[[ 2.4245e-01, 5.1072e-01, 6.4178e-01, ..., 5.1657e-01,
9.0963e-01, 7.4410e-01],
[-1.9277e+00, -1.2516e+00, -2.7008e+00, ..., -1.4672e+00,
-1.4456e-02, 5.6849e-01],
[-9.1763e-02, -8.2534e-01, 4.0243e-02, ..., -2.3901e+00,
2.6546e-01, 8.8653e-01],
...,
[-9.1084e-01, -2.0175e+00, -1.9719e+00, ..., -1.3703e+00,
-9.4372e-01, -6.6540e-01],
[ 8.1147e-01, -9.3731e-03, -2.2756e-02, ..., 4.6423e-01,
8.0058e-01, 8.9399e-01],
[ 2.5714e-01, -4.4757e-01, -2.7652e-01, ..., 6.1170e-01,
8.6939e-01, 7.9048e-01]],
[[-1.7624e+00, -1.4149e+00, -4.2664e+00, ..., -5.8780e+00,
-5.0002e+00, -3.8946e+00],
[ 3.8090e-01, 1.4025e+00, -2.5503e+00, ..., -4.7960e+00,
-4.5356e+00, -3.7288e+00],
[ 1.0962e+00, 1.1338e+00, -5.0794e+00, ..., -1.0328e+01,
-7.9932e+00, -5.9697e+00],
...,
[ 2.9021e-01, 3.2465e-01, -4.1489e+00, ..., -6.1969e+00,
-5.3282e+00, -4.3833e+00],
[ 2.4786e+00, 2.4748e+00, 1.1401e-01, ..., -1.4938e+00,
-1.5955e+00, -1.6460e+00],
[ 1.4651e+00, 1.3154e+00, -1.6260e+00, ..., -2.8173e+00,
-2.9396e+00, -2.8503e+00]],
[[-1.7962e-01, -1.1853e+00, -1.4109e+00, ..., -8.3558e-01,
-5.6565e-01, -6.9135e-01],
[-5.8684e-01, -8.8129e-01, -9.1075e-01, ..., -6.9108e-01,
-3.6906e-01, -2.4205e-01],
[-4.1022e-01, -2.8274e-01, -4.4415e-01, ..., -1.3129e+00,
7.7856e-02, 6.7982e-02],
...,
[-1.0261e-01, -6.6114e-01, -6.1996e-01, ..., -7.2305e-01,
-5.1175e-01, -5.8059e-01],
[-7.2561e-01, -2.6258e-01, -5.7321e-01, ..., -3.3135e-01,
-5.4465e-02, 2.3006e-02],
[-8.9978e-01, -3.0335e-01, -4.6246e-01, ..., -3.2977e-01,
-2.2140e-01, -1.6427e-01]],
...,
[[-2.7627e+00, -2.3391e+00, -2.4374e+00, ..., -3.1636e+00,
-2.6083e+00, -1.7495e+00],
[-1.9622e+00, -6.9916e-01, -1.0617e+00, ..., -1.7273e+00,
-1.8192e+00, -1.5116e+00],
[-1.0387e+00, 8.8241e-01, 7.6931e-01, ..., -4.7802e+00,
-1.3446e+00, 9.5326e-02],
...,
[-2.1904e+00, -1.2016e-01, -1.6686e-01, ..., -1.0461e+00,
-3.2436e-01, 1.8261e-01],
[-1.1653e+00, 8.9382e-01, 1.0314e+00, ..., 2.0888e-01,
3.6259e-01, 4.8314e-01],
[-2.1214e+00, -1.4589e+00, -1.6735e+00, ..., -2.1884e+00,
-1.1998e+00, -5.6649e-01]],
[[-1.5998e+00, -1.6014e+00, -2.7375e+00, ..., -1.4210e+00,
-8.1901e-01, -6.0156e-01],
[-6.8344e-01, -1.0664e-01, -3.1039e+00, ..., -2.0804e+00,
-1.7050e+00, -1.7716e+00],
[-2.0718e+00, -1.0251e+00, -5.0865e+00, ..., -2.0546e+00,
-2.8409e+00, -2.6164e+00],
...,
[-1.3572e+00, 9.8239e-02, -3.4649e+00, ..., -2.5155e+00,
-1.9080e+00, -1.4197e+00],
[-1.6619e+00, -5.0767e-01, -3.0769e+00, ..., -2.3177e+00,
-1.5840e+00, -9.5348e-01],
[-2.2880e+00, -1.3723e+00, -3.0429e+00, ..., -2.8842e+00,
-1.9705e+00, -1.6725e+00]],
[[-1.2871e+00, -1.8125e+00, -1.6358e+00, ..., -1.9309e+00,
-1.5589e+00, -1.1423e+00],
[-1.2238e+00, -1.6901e+00, -1.2175e+00, ..., -1.6957e+00,
-1.3425e+00, -1.0251e+00],
[-1.5393e+00, -1.8788e+00, -1.4627e+00, ..., -2.5970e+00,
-1.7129e+00, -1.0670e+00],
...,
[-2.4263e+00, -3.0252e+00, -2.7642e+00, ..., -3.2542e+00,
-2.5691e+00, -1.9042e+00],
[-1.5721e+00, -2.0641e+00, -1.9675e+00, ..., -2.3138e+00,
-1.8794e+00, -1.4732e+00],
[-1.0288e+00, -1.3299e+00, -1.3502e+00, ..., -1.4509e+00,
-1.2167e+00, -8.8011e-01]]],
[[[ 3.1797e-01, 5.9243e-01, 6.6661e-01, ..., 5.1141e-01,
1.0198e+00, 6.1896e-01],
[-1.9853e+00, -1.3081e+00, -2.9173e+00, ..., -2.6414e+00,
-8.2631e-01, -1.2613e-01],
[-8.6609e-02, -7.6339e-01, -1.0081e-01, ..., -4.7618e-02,
6.2264e-01, 6.7524e-01],
...,
[-2.0243e-01, -1.1898e+00, -2.7852e+00, ..., -1.0829e+00,
-1.5016e+00, -9.0366e-01],
[ 1.4566e+00, 7.9398e-01, -1.1844e+00, ..., -4.2776e-01,
-5.2676e-02, 2.8162e-01],
[ 8.1716e-01, 1.6435e-01, -9.0180e-01, ..., -3.3778e-01,
8.2764e-02, 3.0554e-01]],
[[-1.7767e+00, -1.2632e+00, -4.2023e+00, ..., -5.1363e+00,
-4.6217e+00, -3.5091e+00],
[ 3.2323e-01, 1.6055e+00, -2.4677e+00, ..., -3.5775e+00,
-3.6918e+00, -3.0078e+00],
[ 9.7400e-01, 1.3573e+00, -5.0021e+00, ..., -5.6088e+00,
-5.4503e+00, -5.0954e+00],
...,
[-1.4286e+00, 6.3060e-02, -7.0005e+00, ..., -8.2539e+00,
-7.2374e+00, -5.6278e+00],
[ 1.7167e+00, 2.8087e+00, -1.1873e+00, ..., -2.8185e+00,
-3.1798e+00, -2.4870e+00],
[ 1.1930e+00, 1.8609e+00, -2.3435e+00, ..., -3.3386e+00,
-3.7275e+00, -3.1756e+00]],
[[-1.2429e-01, -1.0578e+00, -1.2164e+00, ..., -6.1044e-01,
-2.5440e-01, -5.7215e-01],
[-5.6908e-01, -8.2203e-01, -7.4019e-01, ..., -4.0917e-01,
-5.1276e-02, -1.2494e-01],
[-4.9365e-01, -3.4242e-01, -4.2424e-01, ..., -3.7484e-01,
7.7427e-02, 6.6978e-02],
...,
[-1.4867e+00, -1.8053e+00, -7.3494e-01, ..., -5.6858e-01,
-7.8195e-01, -1.0597e+00],
[-1.8143e+00, -1.1873e+00, -7.9313e-01, ..., -5.5373e-01,
-2.9523e-01, -3.7895e-01],
[-1.9893e+00, -1.4003e+00, -9.8045e-01, ..., -1.0696e+00,
-9.5222e-01, -9.3931e-01]],
...,
[[-2.8104e+00, -2.3552e+00, -2.7305e+00, ..., -3.6935e+00,
-2.8518e+00, -1.8892e+00],
[-2.0211e+00, -7.2086e-01, -1.5665e+00, ..., -2.5639e+00,
-2.3814e+00, -1.8850e+00],
[-9.8112e-01, 1.1054e+00, 4.8904e-01, ..., -3.8144e-01,
5.3720e-02, 2.1589e-01],
...,
[-1.9519e+00, 1.1669e+00, -7.5982e-03, ..., -9.7871e-01,
-5.7121e-01, -5.1642e-02],
[-8.4643e-01, 1.9436e+00, 1.7535e+00, ..., 5.7127e-01,
4.7960e-01, 7.4151e-01],
[-2.2798e+00, -1.1626e+00, -8.9510e-01, ..., -1.7747e+00,
-8.8789e-01, -2.2692e-01]],
[[-1.5448e+00, -1.5309e+00, -2.6968e+00, ..., -1.0855e+00,
-6.3405e-01, -6.6766e-01],
[-6.5866e-01, -7.6361e-02, -3.0784e+00, ..., -1.2680e+00,
-1.1333e+00, -1.5626e+00],
[-2.1088e+00, -1.0939e+00, -5.1520e+00, ..., -3.2910e+00,
-2.8119e+00, -2.2740e+00],
...,
[-2.5796e+00, -1.7162e+00, -5.5273e+00, ..., -2.8223e+00,
-2.8152e+00, -2.2285e+00],
[-2.2892e+00, -1.5641e+00, -4.3138e+00, ..., -2.7590e+00,
-2.4202e+00, -1.7149e+00],
[-2.4313e+00, -1.8461e+00, -3.3494e+00, ..., -3.0143e+00,
-2.5307e+00, -2.2031e+00]],
[[-1.2852e+00, -1.8343e+00, -1.6865e+00, ..., -1.8236e+00,
-1.5043e+00, -1.1375e+00],
[-1.2200e+00, -1.7001e+00, -1.2533e+00, ..., -1.3875e+00,
-1.0930e+00, -8.7100e-01],
[-1.5325e+00, -1.8979e+00, -1.5150e+00, ..., -1.5886e+00,
-1.1840e+00, -8.6758e-01],
...,
[-3.2939e+00, -4.1342e+00, -3.9138e+00, ..., -4.2262e+00,
-3.3059e+00, -2.5605e+00],
[-2.1831e+00, -2.8384e+00, -2.8015e+00, ..., -2.8796e+00,
-2.3756e+00, -1.9386e+00],
[-1.5981e+00, -1.9491e+00, -2.0067e+00, ..., -1.9531e+00,
-1.5767e+00, -1.2150e+00]]],
[[[ 4.8069e-01, 9.5956e-01, 8.6979e-01, ..., -3.5528e-01,
-8.9507e-02, 1.9586e-03],
[-1.2036e+00, -3.1185e-01, -1.4876e+00, ..., -3.2765e+00,
-1.3096e+00, -4.5963e-01],
[ 5.9862e-01, 4.4017e-01, 3.6176e-01, ..., -1.9800e-01,
-3.6157e-01, 1.4889e-01],
...,
[-1.5415e-01, -8.3741e-01, -1.4958e+00, ..., -1.9399e+00,
-1.9138e+00, -1.2920e+00],
[ 1.0931e+00, 7.9857e-01, 3.1158e-01, ..., 9.4411e-03,
2.3810e-01, 4.3996e-01],
[ 7.2934e-01, 5.2949e-01, 2.1323e-01, ..., 2.0399e-01,
4.4596e-01, 4.9674e-01]],
[[-2.3004e+00, -1.1156e+00, -4.1359e+00, ..., -6.8721e+00,
-6.0674e+00, -4.5183e+00],
[-3.0171e-01, 1.6658e+00, -2.5541e+00, ..., -5.4595e+00,
-5.5470e+00, -4.2851e+00],
[ 3.8316e-01, 1.4425e+00, -5.0660e+00, ..., -7.8608e+00,
-7.7934e+00, -6.5947e+00],
...,
[-4.3031e-01, 6.3858e-01, -4.1034e+00, ..., -7.0966e+00,
-6.6747e+00, -5.1667e+00],
[ 2.0849e+00, 2.7787e+00, 3.1102e-01, ..., -1.9185e+00,
-2.5454e+00, -2.0293e+00],
[ 1.2110e+00, 1.5125e+00, -1.4982e+00, ..., -2.9043e+00,
-3.4525e+00, -3.0256e+00]],
[[-5.7497e-01, -1.5662e+00, -1.3893e+00, ..., -1.4035e+00,
-1.0431e+00, -1.3390e+00],
[-1.2637e+00, -1.6180e+00, -1.0045e+00, ..., -1.0021e+00,
-6.1365e-01, -8.9411e-01],
[-1.1124e+00, -9.8426e-01, -4.0198e-01, ..., -1.1072e+00,
-5.9130e-01, -4.7285e-01],
...,
[-9.5784e-01, -1.3515e+00, -5.3479e-01, ..., -1.0305e+00,
-8.7775e-01, -1.2516e+00],
[-1.3075e+00, -8.8518e-01, -5.4699e-01, ..., -5.6098e-01,
-2.4847e-01, -4.2242e-01],
[-1.3619e+00, -7.4812e-01, -3.2593e-01, ..., -4.8960e-01,
-4.3969e-01, -6.1035e-01]],
...,
[[-2.1668e+00, -1.2255e+00, -2.5314e+00, ..., -3.0894e+00,
-2.5309e+00, -1.8247e+00],
[-7.0289e-01, 9.4221e-01, -9.4998e-01, ..., -2.4212e+00,
-2.5305e+00, -2.0407e+00],
[-2.5945e-01, 2.1939e+00, 4.9953e-01, ..., -6.9942e-02,
2.0546e-02, 2.7235e-01],
...,
[-1.4764e+00, 1.2811e+00, -1.7310e-01, ..., -9.2817e-01,
-2.3266e-01, 3.0038e-01],
[-4.8095e-01, 1.6083e+00, 8.1538e-01, ..., 4.1843e-01,
4.9668e-01, 8.1850e-01],
[-1.5832e+00, -1.0466e+00, -1.5992e+00, ..., -2.2286e+00,
-1.1808e+00, -3.5401e-01]],
[[-1.3015e+00, -1.2207e+00, -2.5811e+00, ..., -2.2169e+00,
-1.8974e+00, -1.8067e+00],
[-6.3982e-01, -1.4681e-01, -3.3142e+00, ..., -2.8361e+00,
-2.8447e+00, -2.8842e+00],
[-2.1069e+00, -1.4416e+00, -5.6910e+00, ..., -4.4393e+00,
-4.4921e+00, -3.7006e+00],
...,
[-1.6465e+00, -4.9012e-01, -4.1213e+00, ..., -2.8277e+00,
-2.7266e+00, -2.2889e+00],
[-1.6648e+00, -7.2607e-01, -3.4294e+00, ..., -2.4366e+00,
-2.2456e+00, -1.6491e+00],
[-2.1033e+00, -1.3748e+00, -2.9845e+00, ..., -2.8658e+00,
-2.4252e+00, -2.1955e+00]],
[[-1.1428e+00, -1.6975e+00, -1.5181e+00, ..., -2.5668e+00,
-2.1460e+00, -1.6684e+00],
[-1.1479e+00, -1.6119e+00, -1.1984e+00, ..., -2.3949e+00,
-1.9289e+00, -1.5814e+00],
[-1.4909e+00, -1.7904e+00, -1.4051e+00, ..., -2.8100e+00,
-2.2435e+00, -1.7412e+00],
...,
[-2.3197e+00, -2.8878e+00, -2.6692e+00, ..., -4.0021e+00,
-3.2465e+00, -2.5426e+00],
[-1.5586e+00, -1.9708e+00, -1.9061e+00, ..., -2.7664e+00,
-2.3182e+00, -1.9136e+00],
[-1.0421e+00, -1.2426e+00, -1.2936e+00, ..., -1.8306e+00,
-1.5214e+00, -1.1781e+00]]],
[[[ 4.2057e-01, 8.6195e-01, 5.0302e-01, ..., 4.7493e-01,
1.0131e+00, 6.3700e-01],
[-1.3388e+00, -4.3515e-01, -2.0173e+00, ..., -2.5739e+00,
-7.6713e-01, -8.6086e-02],
[ 8.5400e-01, 6.2025e-01, 3.3538e-01, ..., -2.4587e-01,
5.0140e-01, 6.2733e-01],
...,
[ 3.6242e-02, -5.7904e-01, -1.3221e+00, ..., -1.9538e+00,
-1.3648e+00, -9.2821e-01],
[ 1.3392e+00, 1.1467e+00, 6.0164e-01, ..., -3.3079e-01,
4.9297e-01, 6.6973e-01],
[ 1.0089e+00, 9.0122e-01, 6.0917e-01, ..., -3.5620e-01,
2.8009e-01, 2.5983e-01]],
[[-2.8352e+00, -1.6090e+00, -4.8314e+00, ..., -4.8682e+00,
-4.4393e+00, -3.3622e+00],
[-5.4905e-01, 1.5858e+00, -2.9155e+00, ..., -3.4087e+00,
-3.5646e+00, -2.9047e+00],
[ 2.5797e-01, 1.3813e+00, -5.7014e+00, ..., -5.3912e+00,
-5.2723e+00, -4.9220e+00],
...,
[-8.9131e-01, 5.1247e-02, -4.8498e+00, ..., -4.5428e+00,
-4.0989e+00, -3.4087e+00],
[ 1.7161e+00, 2.3109e+00, -3.1488e-01, ..., -8.8300e-01,
-1.1983e+00, -1.2258e+00],
[ 9.9891e-01, 1.2623e+00, -1.8236e+00, ..., -2.2167e+00,
-2.4403e+00, -2.3269e+00]],
[[-7.1790e-01, -1.9096e+00, -1.7780e+00, ..., -6.1317e-01,
-1.9884e-01, -4.6227e-01],
[-1.4749e+00, -1.9243e+00, -1.2635e+00, ..., -3.5113e-01,
7.7088e-02, 5.3335e-02],
[-1.3407e+00, -1.3086e+00, -7.1340e-01, ..., -2.7157e-01,
2.4954e-01, 2.9295e-01],
...,
[-9.1741e-01, -1.4497e+00, -7.0642e-01, ..., -4.7733e-01,
-1.4854e-01, -3.3542e-01],
[-1.2108e+00, -9.0302e-01, -5.9583e-01, ..., -3.0750e-01,
1.6574e-01, 2.3634e-01],
[-1.2476e+00, -7.1893e-01, -3.0124e-01, ..., -4.9781e-01,
-2.7390e-01, -2.4362e-01]],
...,
[[-2.2906e+00, -1.2049e+00, -2.5552e+00, ..., -3.6775e+00,
-2.8401e+00, -1.8208e+00],
[-5.8571e-01, 1.2222e+00, -7.4593e-01, ..., -2.6565e+00,
-2.4384e+00, -1.8957e+00],
[-2.1171e-01, 2.4889e+00, 7.5883e-01, ..., -5.6664e-01,
-6.3730e-02, 1.5549e-01],
...,
[-1.3320e+00, 1.3670e+00, 4.2429e-02, ..., -1.5528e+00,
-6.9905e-01, -1.4744e-01],
[-4.7315e-01, 1.5414e+00, 8.2905e-01, ..., -1.8256e-01,
7.6121e-02, 2.9091e-01],
[-1.5909e+00, -1.0477e+00, -1.4938e+00, ..., -2.3645e+00,
-1.4664e+00, -9.3535e-01]],
[[-1.3988e+00, -1.3731e+00, -2.9648e+00, ..., -1.1451e+00,
-7.3827e-01, -7.5819e-01],
[-6.3945e-01, -1.7541e-01, -3.8286e+00, ..., -1.3750e+00,
-1.2322e+00, -1.5968e+00],
[-2.1816e+00, -1.5339e+00, -6.3753e+00, ..., -3.2997e+00,
-2.8015e+00, -2.2384e+00],
...,
[-1.7846e+00, -9.2025e-01, -4.6735e+00, ..., -1.4734e+00,
-1.0159e+00, -8.6361e-01],
[-1.7018e+00, -9.7572e-01, -3.7519e+00, ..., -1.7503e+00,
-1.1302e+00, -7.4478e-01],
[-2.0402e+00, -1.4134e+00, -3.0438e+00, ..., -2.8148e+00,
-1.9977e+00, -1.8244e+00]],
[[-1.2524e+00, -1.9108e+00, -1.7509e+00, ..., -1.8000e+00,
-1.4972e+00, -1.1366e+00],
[-1.2424e+00, -1.8091e+00, -1.3937e+00, ..., -1.3944e+00,
-1.1095e+00, -8.8667e-01],
[-1.6022e+00, -1.9782e+00, -1.5990e+00, ..., -1.5970e+00,
-1.2078e+00, -8.9528e-01],
...,
[-2.3397e+00, -2.9635e+00, -2.8334e+00, ..., -2.8527e+00,
-2.2830e+00, -1.7548e+00],
[-1.5831e+00, -2.0331e+00, -2.0269e+00, ..., -1.9579e+00,
-1.6109e+00, -1.3004e+00],
[-1.0824e+00, -1.3248e+00, -1.4251e+00, ..., -1.2106e+00,
-1.0229e+00, -7.8268e-01]]],
[[[ 9.8852e-02, 3.5043e-01, -1.2750e+00, ..., 4.6797e-01,
1.0173e+00, 6.5943e-01],
[-2.1113e+00, -1.4337e+00, -4.4435e+00, ..., -2.5108e+00,
-6.9170e-01, -1.2288e-02],
[ 3.9406e-01, -3.7767e-01, -1.2373e+00, ..., -2.0163e-01,
5.4071e-01, 6.6829e-01],
...,
[-2.7131e-01, -1.2729e+00, -2.6523e+00, ..., -1.8061e+00,
-1.2736e+00, -8.3860e-01],
[ 1.2551e+00, 6.3552e-01, -1.0342e+00, ..., -1.7546e-01,
5.1815e-01, 7.1258e-01],
[ 8.0092e-01, 2.4674e-01, -5.2956e-01, ..., -2.1723e-01,
3.0521e-01, 3.0885e-01]],
[[-3.2447e+00, -1.3597e+00, -6.1495e+00, ..., -4.8868e+00,
-4.4482e+00, -3.3751e+00],
[-6.0580e-01, 2.1979e+00, -4.2818e+00, ..., -3.4342e+00,
-3.5661e+00, -2.9172e+00],
[-2.0225e-02, 1.8588e+00, -7.5736e+00, ..., -5.4212e+00,
-5.2911e+00, -4.9586e+00],
...,
[-1.2901e+00, 2.6244e-01, -6.6460e+00, ..., -4.7110e+00,
-4.1434e+00, -3.4716e+00],
[ 1.4905e+00, 2.6118e+00, -1.4121e+00, ..., -1.0540e+00,
-1.2388e+00, -1.2811e+00],
[ 9.9205e-01, 1.6580e+00, -2.4456e+00, ..., -2.3651e+00,
-2.4624e+00, -2.3666e+00]],
[[-1.3782e+00, -2.3620e+00, -2.1984e+00, ..., -6.0748e-01,
-2.0943e-01, -4.8044e-01],
[-2.0190e+00, -2.1900e+00, -1.5432e+00, ..., -3.7101e-01,
5.1871e-02, 2.9367e-02],
[-2.6332e+00, -2.0782e+00, -8.4070e-01, ..., -2.8585e-01,
2.3792e-01, 2.8150e-01],
...,
[-1.6650e+00, -1.8539e+00, -7.6249e-01, ..., -5.0450e-01,
-1.6399e-01, -3.4769e-01],
[-1.8273e+00, -1.2147e+00, -7.6158e-01, ..., -3.0195e-01,
1.6145e-01, 2.1860e-01],
[-1.7811e+00, -1.1743e+00, -6.3790e-01, ..., -4.6467e-01,
-2.3976e-01, -2.2283e-01]],
...,
[[-2.3419e+00, -1.2086e+00, -2.6355e+00, ..., -3.6860e+00,
-2.8627e+00, -1.8325e+00],
[-2.0335e+00, 1.4182e-01, -1.3318e+00, ..., -2.6258e+00,
-2.4172e+00, -1.8614e+00],
[-5.4234e-01, 2.4741e+00, 9.0183e-01, ..., -5.5059e-01,
-5.9340e-02, 1.6673e-01],
...,
[-1.7603e+00, 1.1576e+00, 2.4269e-01, ..., -1.5394e+00,
-6.7041e-01, -1.1429e-01],
[-8.6956e-01, 1.6623e+00, 1.5353e+00, ..., -1.5695e-01,
9.9727e-02, 3.3257e-01],
[-2.1064e+00, -1.0354e+00, -7.9811e-01, ..., -2.2946e+00,
-1.4272e+00, -8.8899e-01]],
[[-2.4150e+00, -2.7357e+00, -4.1881e+00, ..., -1.1044e+00,
-6.8817e-01, -7.1428e-01],
[-1.7118e+00, -1.4386e+00, -4.9632e+00, ..., -1.3555e+00,
-1.2051e+00, -1.5855e+00],
[-3.0099e+00, -2.4573e+00, -6.9964e+00, ..., -3.2997e+00,
-2.8079e+00, -2.2511e+00],
...,
[-2.5583e+00, -1.7724e+00, -5.0033e+00, ..., -1.6043e+00,
-1.0960e+00, -8.9863e-01],
[-2.2334e+00, -1.5762e+00, -3.8766e+00, ..., -1.8036e+00,
-1.1749e+00, -7.6537e-01],
[-2.3018e+00, -1.7787e+00, -3.0786e+00, ..., -2.7918e+00,
-2.0007e+00, -1.8064e+00]],
[[-1.9476e+00, -2.6892e+00, -2.4620e+00, ..., -1.8020e+00,
-1.4955e+00, -1.1405e+00],
[-1.9908e+00, -2.7281e+00, -2.1876e+00, ..., -1.3990e+00,
-1.1117e+00, -8.9017e-01],
[-2.5415e+00, -3.1216e+00, -2.6309e+00, ..., -1.5993e+00,
-1.2025e+00, -8.9361e-01],
...,
[-3.1729e+00, -3.9672e+00, -3.7312e+00, ..., -2.8831e+00,
-2.2848e+00, -1.7625e+00],
[-2.1121e+00, -2.7326e+00, -2.6876e+00, ..., -2.0011e+00,
-1.6154e+00, -1.3064e+00],
[-1.5206e+00, -1.8551e+00, -1.8926e+00, ..., -1.2396e+00,
-1.0315e+00, -7.9149e-01]]]], device='cuda:0'),
tensor([[[[ 3.5923e-01, 8.1595e-01, 7.8209e-01, ..., 6.7319e-01,
9.4429e-01, 7.3698e-01],
[-1.2095e+00, -2.5770e-01, -1.2840e+00, ..., -1.1391e+00,
4.2139e-02, 5.5205e-01],
[ 6.5075e-01, 4.4780e-01, 5.9822e-01, ..., 5.4572e-01,
9.3744e-01, 8.6554e-01],
...,
[-1.7128e-01, -1.1390e+00, -2.6585e+00, ..., -1.5222e+00,
-1.3779e+00, -8.6440e-01],
[ 1.4517e+00, 8.5165e-01, -9.3631e-01, ..., -2.8756e-01,
1.1945e-01, 3.8725e-01],
[ 8.7061e-01, 3.1204e-01, -6.6167e-01, ..., -1.8660e-01,
2.1462e-01, 3.8663e-01]],
[[-2.2957e+00, -1.2392e+00, -4.2492e+00, ..., -5.2377e+00,
-4.5063e+00, -3.5256e+00],
[-1.2911e-01, 1.6378e+00, -2.5994e+00, ..., -4.0638e+00,
-3.9744e+00, -3.2727e+00],
[ 7.1054e-01, 1.4534e+00, -5.1598e+00, ..., -5.9234e+00,
-5.7358e+00, -5.3062e+00],
...,
[-1.2962e+00, 1.3779e-01, -6.6891e+00, ..., -7.5941e+00,
-6.9168e+00, -5.4314e+00],
[ 1.7766e+00, 2.8121e+00, -1.0505e+00, ..., -2.6059e+00,
-2.9682e+00, -2.3837e+00],
[ 1.1958e+00, 1.7975e+00, -2.2704e+00, ..., -3.2515e+00,
-3.6205e+00, -3.1432e+00]],
[[-6.8961e-01, -1.8255e+00, -1.7218e+00, ..., -9.2710e-01,
-6.5650e-01, -7.6814e-01],
[-1.3608e+00, -1.7910e+00, -1.2419e+00, ..., -7.3247e-01,
-3.8840e-01, -2.7313e-01],
[-1.0225e+00, -9.3571e-01, -4.2194e-01, ..., -3.3512e-01,
3.1813e-02, 5.6094e-02],
...,
[-1.3798e+00, -1.7352e+00, -7.7357e-01, ..., -9.0408e-01,
-7.8699e-01, -1.0326e+00],
[-1.7118e+00, -1.1487e+00, -8.3006e-01, ..., -5.7482e-01,
-2.9635e-01, -3.6385e-01],
[-1.8433e+00, -1.2481e+00, -8.7939e-01, ..., -9.5362e-01,
-8.0757e-01, -7.8352e-01]],
...,
[[-2.2478e+00, -1.3133e+00, -2.2833e+00, ..., -2.8997e+00,
-2.3044e+00, -1.5207e+00],
[-6.3055e-01, 1.0350e+00, -4.1736e-01, ..., -1.3393e+00,
-1.4238e+00, -1.2293e+00],
[-4.1554e-01, 1.9825e+00, 7.5786e-01, ..., -1.5281e-01,
2.1219e-01, 3.4375e-01],
...,
[-1.8113e+00, 1.2684e+00, 2.4307e-02, ..., -1.0600e+00,
-5.2785e-01, -6.6140e-03],
[-8.0377e-01, 1.9160e+00, 1.6336e+00, ..., 4.8750e-01,
4.2951e-01, 6.7632e-01],
[-2.2366e+00, -1.1840e+00, -1.0348e+00, ..., -1.8782e+00,
-9.7599e-01, -3.1228e-01]],
[[-1.3488e+00, -1.2835e+00, -2.7359e+00, ..., -1.0732e+00,
-5.7548e-01, -4.5933e-01],
[-6.4803e-01, -2.0694e-01, -3.5843e+00, ..., -1.7330e+00,
-1.4970e+00, -1.6831e+00],
[-2.0582e+00, -1.3970e+00, -5.9604e+00, ..., -3.6916e+00,
-3.1903e+00, -2.5451e+00],
...,
[-2.4752e+00, -1.5598e+00, -5.4265e+00, ..., -3.0676e+00,
-2.7210e+00, -2.1623e+00],
[-2.2298e+00, -1.4704e+00, -4.2419e+00, ..., -2.6681e+00,
-2.3048e+00, -1.6234e+00],
[-2.3973e+00, -1.7848e+00, -3.3191e+00, ..., -2.9508e+00,
-2.4260e+00, -2.1100e+00]],
[[-1.1597e+00, -1.7064e+00, -1.4918e+00, ..., -1.6400e+00,
-1.3626e+00, -1.0277e+00],
[-1.1732e+00, -1.6419e+00, -1.1746e+00, ..., -1.3635e+00,
-1.1088e+00, -8.7818e-01],
[-1.5196e+00, -1.8115e+00, -1.3604e+00, ..., -1.5526e+00,
-1.1803e+00, -8.6337e-01],
...,
[-3.1604e+00, -3.9564e+00, -3.7431e+00, ..., -3.9522e+00,
-3.1736e+00, -2.4470e+00],
[-2.1012e+00, -2.7154e+00, -2.6844e+00, ..., -2.7737e+00,
-2.2880e+00, -1.8606e+00],
[-1.5295e+00, -1.8534e+00, -1.9167e+00, ..., -1.8660e+00,
-1.5203e+00, -1.1620e+00]]],
[[[ 2.2277e-01, 4.7905e-01, 6.3299e-01, ..., -3.2682e-01,
-6.6152e-02, 1.9954e-02],
[-2.0520e+00, -1.3689e+00, -2.8850e+00, ..., -3.2144e+00,
-1.2803e+00, -4.3914e-01],
[-1.6309e-01, -9.1043e-01, -2.6254e-03, ..., -1.8025e-01,
-3.6704e-01, 1.6316e-01],
...,
[-9.9535e-01, -2.1617e+00, -2.0419e+00, ..., -1.8990e+00,
-1.8143e+00, -1.1854e+00],
[ 7.7363e-01, -1.2486e-01, -9.5327e-02, ..., 4.9649e-02,
3.0044e-01, 4.7415e-01],
[ 1.8152e-01, -5.9348e-01, -3.8597e-01, ..., 2.2705e-01,
4.5865e-01, 4.9123e-01]],
[[-1.6689e+00, -1.3697e+00, -4.2408e+00, ..., -6.8640e+00,
-6.0454e+00, -4.5210e+00],
[ 4.9731e-01, 1.4916e+00, -2.5077e+00, ..., -5.4875e+00,
-5.5441e+00, -4.3023e+00],
[ 1.2310e+00, 1.2169e+00, -5.0604e+00, ..., -7.8689e+00,
-7.7783e+00, -6.5978e+00],
...,
[ 3.9763e-01, 3.6571e-01, -4.1378e+00, ..., -7.1173e+00,
-6.6769e+00, -5.2196e+00],
[ 2.5510e+00, 2.4999e+00, 1.1110e-01, ..., -1.9898e+00,
-2.6004e+00, -2.1167e+00],
[ 1.5145e+00, 1.3220e+00, -1.6298e+00, ..., -2.9730e+00,
-3.4920e+00, -3.0887e+00]],
[[-1.4143e-01, -1.1739e+00, -1.4114e+00, ..., -1.3920e+00,
-1.0407e+00, -1.3362e+00],
[-5.1839e-01, -8.3723e-01, -8.8677e-01, ..., -9.9140e-01,
-6.1158e-01, -8.9838e-01],
[-3.4957e-01, -2.3682e-01, -4.4794e-01, ..., -1.1068e+00,
-5.9213e-01, -4.7372e-01],
...,
[-2.8096e-02, -5.8809e-01, -6.0445e-01, ..., -1.0171e+00,
-8.7337e-01, -1.2320e+00],
[-6.8518e-01, -2.0914e-01, -5.7485e-01, ..., -5.5575e-01,
-2.6003e-01, -4.1501e-01],
[-8.7003e-01, -2.6809e-01, -4.7858e-01, ..., -5.1112e-01,
-4.6536e-01, -6.1191e-01]],
...,
[[-2.8765e+00, -2.4931e+00, -2.4916e+00, ..., -3.0789e+00,
-2.4825e+00, -1.8000e+00],
[-2.1246e+00, -8.6324e-01, -1.1384e+00, ..., -2.4102e+00,
-2.4746e+00, -1.9959e+00],
[-1.1112e+00, 8.0477e-01, 7.7691e-01, ..., -7.3647e-02,
5.8622e-02, 2.9686e-01],
...,
[-2.2872e+00, -2.4149e-01, -2.0207e-01, ..., -9.3254e-01,
-2.6115e-01, 2.7595e-01],
[-1.2540e+00, 8.3489e-01, 1.0335e+00, ..., 3.9838e-01,
4.5502e-01, 7.8914e-01],
[-2.1893e+00, -1.5184e+00, -1.7225e+00, ..., -2.1758e+00,
-1.1657e+00, -3.4004e-01]],
[[-1.6309e+00, -1.6323e+00, -2.8043e+00, ..., -2.2293e+00,
-1.9134e+00, -1.8000e+00],
[-6.8154e-01, -7.2630e-02, -3.1511e+00, ..., -2.8625e+00,
-2.8597e+00, -2.8732e+00],
[-2.0597e+00, -9.5622e-01, -5.1147e+00, ..., -4.4487e+00,
-4.4865e+00, -3.6939e+00],
...,
[-1.3282e+00, 1.9015e-01, -3.3841e+00, ..., -2.8195e+00,
-2.7650e+00, -2.3477e+00],
[-1.6673e+00, -4.7673e-01, -3.0320e+00, ..., -2.4265e+00,
-2.2646e+00, -1.6954e+00],
[-2.3230e+00, -1.3704e+00, -3.0468e+00, ..., -2.8403e+00,
-2.4138e+00, -2.2052e+00]],
[[-1.3054e+00, -1.8332e+00, -1.6493e+00, ..., -2.5594e+00,
-2.1370e+00, -1.6628e+00],
[-1.2423e+00, -1.7088e+00, -1.2152e+00, ..., -2.3932e+00,
-1.9284e+00, -1.5817e+00],
[-1.5574e+00, -1.8962e+00, -1.4689e+00, ..., -2.8159e+00,
-2.2438e+00, -1.7413e+00],
...,
[-2.4493e+00, -3.0522e+00, -2.7845e+00, ..., -4.0138e+00,
-3.2556e+00, -2.5474e+00],
[-1.5806e+00, -2.0827e+00, -1.9813e+00, ..., -2.7857e+00,
-2.3343e+00, -1.9242e+00],
[-1.0327e+00, -1.3455e+00, -1.3563e+00, ..., -1.8447e+00,
-1.5318e+00, -1.1843e+00]]],
[[[ 1.5891e-01, 3.7741e-01, -1.2148e+00, ..., 4.5420e-01,
9.9176e-01, 6.3317e-01],
[-2.0964e+00, -1.4471e+00, -4.4240e+00, ..., -2.5510e+00,
-7.6440e-01, -8.4568e-02],
[ 4.5241e-01, -2.7042e-01, -1.1320e+00, ..., -2.0137e-01,
5.2758e-01, 6.4780e-01],
...,
[-2.7222e-01, -1.2673e+00, -2.6264e+00, ..., -1.9695e+00,
-1.3427e+00, -8.8707e-01],
[ 1.2653e+00, 6.6199e-01, -9.2711e-01, ..., -3.3611e-01,
5.4095e-01, 7.3614e-01],
[ 8.2381e-01, 3.0308e-01, -4.4194e-01, ..., -3.9535e-01,
2.9050e-01, 2.8223e-01]],
[[-3.2572e+00, -1.4866e+00, -6.1401e+00, ..., -4.8685e+00,
-4.4366e+00, -3.3654e+00],
[-5.7523e-01, 2.1178e+00, -4.1864e+00, ..., -3.4190e+00,
-3.5408e+00, -2.8813e+00],
[ 2.3508e-02, 1.7908e+00, -7.4108e+00, ..., -5.3873e+00,
-5.2617e+00, -4.9075e+00],
...,
[-1.2444e+00, 2.8497e-01, -6.5017e+00, ..., -4.5978e+00,
-4.1363e+00, -3.4822e+00],
[ 1.5565e+00, 2.6419e+00, -1.3066e+00, ..., -9.0331e-01,
-1.2341e+00, -1.2788e+00],
[ 1.0207e+00, 1.6568e+00, -2.3920e+00, ..., -2.2297e+00,
-2.4818e+00, -2.3819e+00]],
[[-1.4090e+00, -2.3657e+00, -2.1783e+00, ..., -6.1265e-01,
-2.1066e-01, -4.7403e-01],
[-2.0451e+00, -2.2064e+00, -1.5686e+00, ..., -3.4583e-01,
6.6550e-02, 2.8190e-02],
[-2.5375e+00, -1.9865e+00, -8.3737e-01, ..., -2.7762e-01,
2.5972e-01, 2.9552e-01],
...,
[-1.6098e+00, -1.8406e+00, -7.8127e-01, ..., -4.5340e-01,
-1.4678e-01, -3.4773e-01],
[-1.7788e+00, -1.2006e+00, -7.7700e-01, ..., -3.0086e-01,
1.6301e-01, 2.2051e-01],
[-1.7379e+00, -1.1208e+00, -6.0228e-01, ..., -4.9211e-01,
-2.5737e-01, -2.3745e-01]],
...,
[[-2.3771e+00, -1.3198e+00, -2.7584e+00, ..., -3.6712e+00,
-2.8307e+00, -1.8308e+00],
[-1.8624e+00, 2.0201e-01, -1.3511e+00, ..., -2.6662e+00,
-2.4071e+00, -1.8443e+00],
[-4.8450e-01, 2.4833e+00, 9.3381e-01, ..., -5.9727e-01,
-7.8758e-02, 1.6230e-01],
...,
[-1.7339e+00, 1.1934e+00, 2.3688e-01, ..., -1.5040e+00,
-6.6483e-01, -1.2285e-01],
[-8.6676e-01, 1.6748e+00, 1.5281e+00, ..., -1.3881e-01,
1.2284e-01, 3.5842e-01],
[-2.1018e+00, -1.0585e+00, -8.3916e-01, ..., -2.3688e+00,
-1.4467e+00, -8.9634e-01]],
[[-2.3772e+00, -2.6311e+00, -3.9964e+00, ..., -1.1433e+00,
-7.2860e-01, -7.3827e-01],
[-1.6607e+00, -1.3260e+00, -4.7049e+00, ..., -1.3702e+00,
-1.2231e+00, -1.5885e+00],
[-3.0202e+00, -2.4394e+00, -6.8295e+00, ..., -3.3095e+00,
-2.8167e+00, -2.2526e+00],
...,
[-2.5283e+00, -1.7265e+00, -4.9792e+00, ..., -1.5026e+00,
-1.0890e+00, -9.0629e-01],
[-2.2184e+00, -1.5388e+00, -3.8764e+00, ..., -1.7274e+00,
-1.1815e+00, -7.8381e-01],
[-2.3117e+00, -1.7652e+00, -3.0610e+00, ..., -2.7856e+00,
-2.0202e+00, -1.8513e+00]],
[[-1.8975e+00, -2.6275e+00, -2.4202e+00, ..., -1.8007e+00,
-1.5001e+00, -1.1406e+00],
[-1.9153e+00, -2.6359e+00, -2.1046e+00, ..., -1.4026e+00,
-1.1107e+00, -8.8406e-01],
[-2.4358e+00, -2.9952e+00, -2.5221e+00, ..., -1.5935e+00,
-1.2023e+00, -8.8793e-01],
...,
[-3.1111e+00, -3.8888e+00, -3.6516e+00, ..., -2.8937e+00,
-2.3123e+00, -1.7835e+00],
[-2.0696e+00, -2.6753e+00, -2.6372e+00, ..., -2.0095e+00,
-1.6469e+00, -1.3344e+00],
[-1.4880e+00, -1.8097e+00, -1.8565e+00, ..., -1.2515e+00,
-1.0594e+00, -8.1056e-01]]],
[[[ 2.7433e-01, 5.3410e-01, 6.8040e-01, ..., -3.6826e-01,
-6.9226e-02, 3.2754e-03],
[-2.0361e+00, -1.3668e+00, -2.9847e+00, ..., -3.2747e+00,
-1.2801e+00, -4.5530e-01],
[-8.7695e-02, -8.8302e-01, 2.8690e-02, ..., -2.2562e-01,
-3.6562e-01, 1.3031e-01],
...,
[-9.8686e-01, -2.1188e+00, -1.9366e+00, ..., -1.8950e+00,
-1.8508e+00, -1.2294e+00],
[ 7.2713e-01, -1.3321e-01, -8.1302e-02, ..., 5.7604e-02,
2.9790e-01, 4.7133e-01],
[ 1.4322e-01, -6.0799e-01, -3.9698e-01, ..., 2.4438e-01,
4.8317e-01, 5.1581e-01]],
[[-1.6914e+00, -1.3180e+00, -4.3258e+00, ..., -6.8356e+00,
-6.0989e+00, -4.5022e+00],
[ 6.3381e-01, 1.7294e+00, -2.3825e+00, ..., -5.4396e+00,
-5.5771e+00, -4.2730e+00],
[ 1.4281e+00, 1.5058e+00, -4.9742e+00, ..., -7.8389e+00,
-7.8083e+00, -6.5665e+00],
...,
[ 3.2131e-01, 1.9841e-01, -4.2308e+00, ..., -7.1104e+00,
-6.7055e+00, -5.2160e+00],
[ 2.3512e+00, 2.2240e+00, -1.9843e-01, ..., -1.9533e+00,
-2.6104e+00, -2.0953e+00],
[ 1.3773e+00, 1.1520e+00, -1.7767e+00, ..., -2.9527e+00,
-3.5093e+00, -3.0587e+00]],
[[-2.0438e-01, -1.3289e+00, -1.5329e+00, ..., -1.4117e+00,
-1.0260e+00, -1.3231e+00],
[-6.1186e-01, -9.5442e-01, -9.6110e-01, ..., -9.8953e-01,
-5.8704e-01, -8.7296e-01],
[-4.5107e-01, -3.1037e-01, -4.3901e-01, ..., -1.1069e+00,
-5.8709e-01, -4.5443e-01],
...,
[ 2.5094e-02, -5.1907e-01, -5.9879e-01, ..., -1.0349e+00,
-8.6934e-01, -1.2398e+00],
[-6.1643e-01, -1.6548e-01, -5.6507e-01, ..., -5.6861e-01,
-2.5431e-01, -4.1605e-01],
[-7.9109e-01, -2.2266e-01, -4.7264e-01, ..., -5.0475e-01,
-4.4819e-01, -6.0263e-01]],
...,
[[-3.0365e+00, -2.5504e+00, -2.6126e+00, ..., -3.0500e+00,
-2.5122e+00, -1.7976e+00],
[-2.1888e+00, -7.6490e-01, -1.1020e+00, ..., -2.4042e+00,
-2.5459e+00, -2.0658e+00],
[-1.1966e+00, 8.8872e-01, 7.7760e-01, ..., -6.3669e-02,
3.4816e-03, 2.5740e-01],
...,
[-2.1970e+00, -2.8796e-01, -1.9627e-01, ..., -9.6519e-01,
-2.6505e-01, 2.8737e-01],
[-1.2241e+00, 7.7074e-01, 1.0096e+00, ..., 4.1398e-01,
4.7638e-01, 7.9866e-01],
[-2.1168e+00, -1.4438e+00, -1.5848e+00, ..., -2.1994e+00,
-1.1602e+00, -3.5435e-01]],
[[-1.6476e+00, -1.6389e+00, -2.9200e+00, ..., -2.2414e+00,
-1.9105e+00, -1.8182e+00],
[-6.4533e-01, 4.9626e-02, -3.2415e+00, ..., -2.8667e+00,
-2.8459e+00, -2.8676e+00],
[-2.0810e+00, -9.0311e-01, -5.3292e+00, ..., -4.4468e+00,
-4.4785e+00, -3.6729e+00],
...,
[-1.3745e+00, 6.1192e-02, -3.3489e+00, ..., -2.8682e+00,
-2.7533e+00, -2.3302e+00],
[-1.6790e+00, -5.4890e-01, -2.9888e+00, ..., -2.4812e+00,
-2.2712e+00, -1.6692e+00],
[-2.2862e+00, -1.3891e+00, -3.0134e+00, ..., -2.8861e+00,
-2.4289e+00, -2.1769e+00]],
[[-1.3357e+00, -1.8876e+00, -1.6844e+00, ..., -2.5730e+00,
-2.1449e+00, -1.6711e+00],
[-1.2788e+00, -1.7501e+00, -1.2165e+00, ..., -2.3931e+00,
-1.9323e+00, -1.5879e+00],
[-1.5981e+00, -1.9270e+00, -1.4402e+00, ..., -2.8166e+00,
-2.2483e+00, -1.7521e+00],
...,
[-2.4031e+00, -2.9924e+00, -2.7519e+00, ..., -3.9958e+00,
-3.2383e+00, -2.5404e+00],
[-1.5594e+00, -2.0529e+00, -1.9789e+00, ..., -2.7633e+00,
-2.3132e+00, -1.9114e+00],
[-1.0271e+00, -1.3339e+00, -1.3635e+00, ..., -1.8277e+00,
-1.5227e+00, -1.1781e+00]]],
[[[ 1.5703e-01, 3.6773e-01, -1.1615e+00, ..., 6.3699e-01,
9.0318e-01, 6.7373e-01],
[-2.0874e+00, -1.4606e+00, -4.3503e+00, ..., -1.4466e+00,
-1.9314e-01, 3.6930e-01],
[ 4.2671e-01, -3.2122e-01, -1.1182e+00, ..., 5.2917e-01,
8.9483e-01, 8.4027e-01],
...,
[-2.2160e-01, -1.2593e+00, -2.5748e+00, ..., -4.1143e+00,
-4.2170e+00, -4.7786e+00],
[ 1.2938e+00, 6.5653e-01, -8.9675e-01, ..., -1.0925e+00,
-1.3193e+00, -1.8995e+00],
[ 8.4997e-01, 3.0249e-01, -4.2195e-01, ..., 1.7252e-01,
1.9403e-01, 2.2615e-01]],
[[-3.2246e+00, -1.4286e+00, -6.0999e+00, ..., -5.3386e+00,
-4.5861e+00, -3.5924e+00],
[-5.4397e-01, 2.1570e+00, -4.1518e+00, ..., -4.1657e+00,
-4.0403e+00, -3.3409e+00],
[ 6.4080e-02, 1.8292e+00, -7.3734e+00, ..., -6.0448e+00,
-5.7994e+00, -5.3820e+00],
...,
[-1.2202e+00, 3.2818e-01, -6.5035e+00, ..., -6.7299e+00,
-6.6985e+00, -7.1227e+00],
[ 1.5632e+00, 2.6501e+00, -1.3199e+00, ..., -1.4549e+00,
-1.8471e+00, -2.7404e+00],
[ 1.0284e+00, 1.6591e+00, -2.3980e+00, ..., -2.2793e+00,
-2.4742e+00, -2.6196e+00]],
[[-1.3762e+00, -2.3387e+00, -2.1485e+00, ..., -7.9941e-01,
-5.9189e-01, -7.7645e-01],
[-2.0146e+00, -2.1887e+00, -1.5370e+00, ..., -6.8045e-01,
-4.2002e-01, -3.5848e-01],
[-2.5264e+00, -1.9972e+00, -8.3486e-01, ..., -4.0529e-01,
-1.1393e-01, -1.0608e-01],
...,
[-1.5915e+00, -1.8411e+00, -7.9139e-01, ..., -1.3460e-01,
5.9996e-02, -1.2312e+00],
[-1.7698e+00, -1.2008e+00, -7.8189e-01, ..., 1.4342e-01,
4.0018e-01, -2.4723e-01],
[-1.7304e+00, -1.1127e+00, -5.8865e-01, ..., -2.0854e-01,
-9.0981e-02, -1.2164e-01]],
...,
[[-2.3701e+00, -1.2930e+00, -2.7591e+00, ..., -2.9334e+00,
-2.3150e+00, -1.5433e+00],
[-1.8486e+00, 2.3787e-01, -1.3428e+00, ..., -1.4575e+00,
-1.4752e+00, -1.2799e+00],
[-4.6396e-01, 2.5317e+00, 9.3776e-01, ..., -1.2552e-01,
2.5302e-01, 3.4343e-01],
...,
[-1.7577e+00, 1.2174e+00, 2.5288e-01, ..., -1.9139e+00,
-1.1160e+00, -2.4704e-01],
[-8.8323e-01, 1.6937e+00, 1.5277e+00, ..., -4.0808e-02,
1.9737e-01, 7.0967e-01],
[-2.1243e+00, -1.0439e+00, -8.2899e-01, ..., -2.8729e+00,
-1.7211e+00, -6.8917e-01]],
[[-2.3529e+00, -2.6014e+00, -3.9976e+00, ..., -1.1900e+00,
-6.6929e-01, -5.2721e-01],
[-1.6278e+00, -1.2937e+00, -4.7398e+00, ..., -1.7275e+00,
-1.4969e+00, -1.6933e+00],
[-2.9853e+00, -2.4028e+00, -6.8647e+00, ..., -3.7056e+00,
-3.2135e+00, -2.5650e+00],
...,
[-2.4910e+00, -1.7134e+00, -5.0132e+00, ..., -2.3061e-01,
-1.9882e-01, -7.7880e-02],
[-2.1919e+00, -1.5269e+00, -3.8979e+00, ..., -1.0978e+00,
-6.6041e-01, 1.3674e-01],
[-2.2872e+00, -1.7399e+00, -3.0724e+00, ..., -2.7770e+00,
-1.9401e+00, -1.4966e+00]],
[[-1.8898e+00, -2.6131e+00, -2.3943e+00, ..., -1.6627e+00,
-1.3853e+00, -1.0362e+00],
[-1.9131e+00, -2.6270e+00, -2.0910e+00, ..., -1.3633e+00,
-1.1123e+00, -8.8049e-01],
[-2.4381e+00, -2.9922e+00, -2.5078e+00, ..., -1.5837e+00,
-1.2167e+00, -8.8994e-01],
...,
[-3.1158e+00, -3.8843e+00, -3.6440e+00, ..., -3.4371e+00,
-3.0310e+00, -2.6397e+00],
[-2.0733e+00, -2.6764e+00, -2.6302e+00, ..., -2.3393e+00,
-2.0268e+00, -1.7868e+00],
[-1.4905e+00, -1.8099e+00, -1.8522e+00, ..., -1.3391e+00,
-1.1896e+00, -9.2907e-01]]]], device='cuda:0'),
tensor([[[[ 1.4136e-01, 3.8243e-01, -1.1982e+00, ..., 6.4211e-01,
9.0336e-01, 6.7476e-01],
[-2.0809e+00, -1.4234e+00, -4.3977e+00, ..., -1.4056e+00,
-1.6252e-01, 3.9045e-01],
[ 4.5190e-01, -2.5645e-01, -1.1293e+00, ..., 5.5270e-01,
9.1014e-01, 8.2828e-01],
...,
[-2.6759e-01, -1.2565e+00, -2.5293e+00, ..., -1.3929e+00,
-1.0372e+00, -8.3299e-01],
[ 1.2064e+00, 6.3363e-01, -8.6940e-01, ..., 4.9183e-01,
7.2500e-01, 7.9285e-01],
[ 7.6686e-01, 2.7052e-01, -4.1461e-01, ..., 5.6443e-01,
7.9793e-01, 7.0143e-01]],
[[-3.2639e+00, -1.4522e+00, -6.1438e+00, ..., -5.3599e+00,
-4.5935e+00, -3.6061e+00],
[-5.8694e-01, 2.1564e+00, -4.1861e+00, ..., -4.1918e+00,
-4.0598e+00, -3.3506e+00],
[ 1.6539e-02, 1.8574e+00, -7.4041e+00, ..., -6.0994e+00,
-5.8362e+00, -5.3944e+00],
...,
[-1.1974e+00, 3.2157e-01, -6.5501e+00, ..., -5.3551e+00,
-4.6790e+00, -3.8954e+00],
[ 1.5112e+00, 2.5627e+00, -1.4779e+00, ..., -9.2712e-01,
-1.1200e+00, -1.2714e+00],
[ 9.8059e-01, 1.5920e+00, -2.4923e+00, ..., -2.3681e+00,
-2.5371e+00, -2.5513e+00]],
[[-1.3791e+00, -2.3567e+00, -2.1838e+00, ..., -8.1817e-01,
-6.0889e-01, -7.7929e-01],
[-2.0169e+00, -2.2085e+00, -1.5830e+00, ..., -6.7584e-01,
-4.2449e-01, -3.5396e-01],
[-2.5342e+00, -2.0149e+00, -8.5324e-01, ..., -4.1071e-01,
-1.3380e-01, -1.1473e-01],
...,
[-1.5894e+00, -1.7852e+00, -7.2208e-01, ..., -6.8034e-01,
-5.7770e-01, -7.6097e-01],
[-1.7702e+00, -1.1786e+00, -7.3994e-01, ..., -2.3200e-01,
-3.9564e-02, -3.2257e-02],
[-1.7364e+00, -1.1285e+00, -5.9551e-01, ..., -3.1938e-01,
-2.8718e-01, -2.6086e-01]],
...,
[[-2.3558e+00, -1.2907e+00, -2.7274e+00, ..., -2.9451e+00,
-2.3160e+00, -1.5635e+00],
[-1.8832e+00, 2.1215e-01, -1.3204e+00, ..., -1.4518e+00,
-1.4665e+00, -1.2942e+00],
[-4.7556e-01, 2.5245e+00, 9.4775e-01, ..., -1.1175e-01,
2.5037e-01, 3.2895e-01],
...,
[-1.7105e+00, 1.1571e+00, 2.2126e-01, ..., -1.0736e+00,
-2.7680e-01, 2.3680e-01],
[-8.5748e-01, 1.6098e+00, 1.4409e+00, ..., 1.8133e-01,
3.7781e-01, 4.6993e-01],
[-2.0425e+00, -1.0039e+00, -8.3267e-01, ..., -2.4023e+00,
-1.3672e+00, -7.0334e-01]],
[[-2.3773e+00, -2.6284e+00, -4.0259e+00, ..., -1.1838e+00,
-6.6233e-01, -5.2550e-01],
[-1.6583e+00, -1.3162e+00, -4.7562e+00, ..., -1.7400e+00,
-1.5158e+00, -1.7010e+00],
[-3.0085e+00, -2.4339e+00, -6.8875e+00, ..., -3.7334e+00,
-3.2459e+00, -2.5785e+00],
...,
[-2.5238e+00, -1.7586e+00, -5.0263e+00, ..., -2.0584e+00,
-1.6063e+00, -1.3007e+00],
[-2.2043e+00, -1.5581e+00, -3.8885e+00, ..., -2.0681e+00,
-1.3986e+00, -8.6200e-01],
[-2.2506e+00, -1.7370e+00, -3.0363e+00, ..., -2.8883e+00,
-1.9657e+00, -1.6850e+00]],
[[-1.8981e+00, -2.6284e+00, -2.4043e+00, ..., -1.6645e+00,
-1.3844e+00, -1.0342e+00],
[-1.9252e+00, -2.6401e+00, -2.0989e+00, ..., -1.3645e+00,
-1.1158e+00, -8.7868e-01],
[-2.4502e+00, -3.0077e+00, -2.5098e+00, ..., -1.5795e+00,
-1.2157e+00, -8.8170e-01],
...,
[-3.1295e+00, -3.9028e+00, -3.6746e+00, ..., -2.8701e+00,
-2.2895e+00, -1.7305e+00],
[-2.1000e+00, -2.7017e+00, -2.6670e+00, ..., -1.9885e+00,
-1.6375e+00, -1.3108e+00],
[-1.5191e+00, -1.8498e+00, -1.8928e+00, ..., -1.2122e+00,
-1.0499e+00, -7.7292e-01]]],
[[[ 2.5601e-01, 5.2058e-01, 6.8694e-01, ..., 5.2499e-01,
1.0422e+00, 6.0999e-01],
[-2.1449e+00, -1.4985e+00, -3.1189e+00, ..., -2.7710e+00,
-9.0979e-01, -1.9800e-01],
[-2.0758e-01, -1.0178e+00, -2.2948e-01, ..., -1.5157e-01,
5.5347e-01, 6.5537e-01],
...,
[-1.4876e-01, -1.1358e+00, -2.6223e+00, ..., -1.5413e+00,
-1.4543e+00, -8.9662e-01],
[ 1.4317e+00, 8.2993e-01, -9.2881e-01, ..., -2.9352e-01,
1.0200e-01, 3.8423e-01],
[ 8.4727e-01, 2.8760e-01, -6.5525e-01, ..., -1.8937e-01,
2.1761e-01, 3.8002e-01]],
[[-1.6425e+00, -1.2419e+00, -4.2000e+00, ..., -5.0290e+00,
-4.5517e+00, -3.4807e+00],
[ 5.0055e-01, 1.6705e+00, -2.3996e+00, ..., -3.5266e+00,
-3.6306e+00, -2.9859e+00],
[ 1.1610e+00, 1.4021e+00, -4.9856e+00, ..., -5.5194e+00,
-5.3620e+00, -5.0436e+00],
...,
[-1.2727e+00, 1.1776e-01, -6.6966e+00, ..., -7.6328e+00,
-6.9130e+00, -5.4264e+00],
[ 1.7682e+00, 2.7617e+00, -1.0739e+00, ..., -2.5931e+00,
-2.9484e+00, -2.3781e+00],
[ 1.1849e+00, 1.7634e+00, -2.2939e+00, ..., -3.2499e+00,
-3.6239e+00, -3.1327e+00]],
[[-1.0485e-01, -1.0638e+00, -1.2464e+00, ..., -5.3966e-01,
-2.1090e-01, -5.2738e-01],
[-5.1880e-01, -7.4137e-01, -7.0193e-01, ..., -3.1839e-01,
4.4051e-03, -8.5949e-02],
[-4.6109e-01, -2.6238e-01, -3.5602e-01, ..., -3.7513e-01,
9.7109e-02, 1.0491e-01],
...,
[-1.3841e+00, -1.7411e+00, -7.4753e-01, ..., -8.9265e-01,
-7.6455e-01, -1.0198e+00],
[-1.7155e+00, -1.1393e+00, -8.1347e-01, ..., -5.8300e-01,
-3.0597e-01, -3.7482e-01],
[-1.8569e+00, -1.2629e+00, -8.7349e-01, ..., -9.5493e-01,
-8.1649e-01, -7.9924e-01]],
...,
[[-2.9544e+00, -2.5001e+00, -2.7181e+00, ..., -3.6880e+00,
-2.7893e+00, -1.8694e+00],
[-2.2714e+00, -9.6916e-01, -1.6371e+00, ..., -2.7347e+00,
-2.3781e+00, -1.8447e+00],
[-1.0898e+00, 9.6141e-01, 4.8136e-01, ..., -5.2342e-01,
2.2838e-02, 2.0774e-01],
...,
[-1.8118e+00, 1.2517e+00, -6.1294e-03, ..., -1.0883e+00,
-5.3351e-01, -6.3925e-03],
[-8.0352e-01, 1.8984e+00, 1.6214e+00, ..., 4.7910e-01,
4.1677e-01, 6.8568e-01],
[-2.2140e+00, -1.1739e+00, -1.0385e+00, ..., -1.8645e+00,
-9.7609e-01, -3.0681e-01]],
[[-1.6182e+00, -1.6174e+00, -2.8171e+00, ..., -1.2296e+00,
-7.7990e-01, -7.7618e-01],
[-6.6900e-01, -4.3918e-02, -3.0926e+00, ..., -1.3205e+00,
-1.1761e+00, -1.5953e+00],
[-2.1273e+00, -1.0092e+00, -5.1282e+00, ..., -3.2355e+00,
-2.7525e+00, -2.2713e+00],
...,
[-2.4790e+00, -1.5819e+00, -5.4520e+00, ..., -3.0834e+00,
-2.7234e+00, -2.1778e+00],
[-2.2195e+00, -1.4668e+00, -4.2544e+00, ..., -2.6647e+00,
-2.3047e+00, -1.6342e+00],
[-2.3830e+00, -1.7783e+00, -3.3159e+00, ..., -2.9568e+00,
-2.4311e+00, -2.1228e+00]],
[[-1.3337e+00, -1.8806e+00, -1.7070e+00, ..., -1.8120e+00,
-1.5158e+00, -1.1438e+00],
[-1.2559e+00, -1.7362e+00, -1.2657e+00, ..., -1.3876e+00,
-1.1044e+00, -8.7170e-01],
[-1.5666e+00, -1.9268e+00, -1.5265e+00, ..., -1.5929e+00,
-1.1937e+00, -8.6966e-01],
...,
[-3.1561e+00, -3.9497e+00, -3.7501e+00, ..., -3.9474e+00,
-3.1638e+00, -2.4424e+00],
[-2.1014e+00, -2.7147e+00, -2.6882e+00, ..., -2.7742e+00,
-2.2901e+00, -1.8626e+00],
[-1.5293e+00, -1.8543e+00, -1.9197e+00, ..., -1.8723e+00,
-1.5227e+00, -1.1659e+00]]],
[[[ 4.4721e-01, 8.9269e-01, 6.7328e-01, ..., -6.6794e-01,
-3.9403e-01, -1.7808e-01],
[-1.0972e+00, -1.7867e-01, -1.4760e+00, ..., -3.8535e+00,
-1.6986e+00, -7.1093e-01],
[ 6.4829e-01, 4.8470e-01, 3.6421e-01, ..., -4.7044e-01,
-7.0560e-01, 3.3346e-03],
...,
[-1.5292e-01, -8.3814e-01, -1.5423e+00, ..., -1.9438e+00,
-1.9572e+00, -1.2029e+00],
[ 1.2243e+00, 9.1861e-01, 3.1504e-01, ..., -1.8341e-01,
4.8598e-02, 3.4490e-01],
[ 8.7868e-01, 6.8567e-01, 3.3362e-01, ..., 7.5083e-03,
2.7643e-01, 4.0934e-01]],
[[-2.4200e+00, -1.2097e+00, -4.4165e+00, ..., -7.2388e+00,
-6.4561e+00, -4.7603e+00],
[-4.1327e-01, 1.5592e+00, -2.9038e+00, ..., -5.7800e+00,
-5.9605e+00, -4.5395e+00],
[ 2.7568e-01, 1.3178e+00, -5.5112e+00, ..., -8.3161e+00,
-8.3295e+00, -6.8848e+00],
...,
[-5.7403e-01, 5.6085e-01, -4.4291e+00, ..., -7.7009e+00,
-7.2976e+00, -5.6141e+00],
[ 2.0898e+00, 2.8284e+00, 2.4209e-01, ..., -2.2988e+00,
-2.9781e+00, -2.2949e+00],
[ 1.2309e+00, 1.5612e+00, -1.5792e+00, ..., -3.1223e+00,
-3.7321e+00, -3.1872e+00]],
[[-6.2593e-01, -1.5969e+00, -1.4476e+00, ..., -1.6082e+00,
-1.2099e+00, -1.4887e+00],
[-1.3427e+00, -1.6527e+00, -1.0743e+00, ..., -1.1356e+00,
-7.4014e-01, -1.0622e+00],
[-1.1769e+00, -9.5933e-01, -3.8541e-01, ..., -1.2898e+00,
-7.3164e-01, -5.5081e-01],
...,
[-9.9295e-01, -1.4129e+00, -5.6413e-01, ..., -1.0465e+00,
-9.0508e-01, -1.3066e+00],
[-1.3080e+00, -8.6122e-01, -5.3550e-01, ..., -5.5006e-01,
-2.4355e-01, -4.2798e-01],
[-1.3935e+00, -7.5209e-01, -3.0883e-01, ..., -7.1181e-01,
-6.7597e-01, -8.2785e-01]],
...,
[[-2.1010e+00, -1.1835e+00, -2.4700e+00, ..., -3.0657e+00,
-2.5252e+00, -1.8593e+00],
[-5.8001e-01, 1.0729e+00, -7.3374e-01, ..., -2.6671e+00,
-2.7379e+00, -2.1819e+00],
[-2.2757e-01, 2.1916e+00, 5.5031e-01, ..., -1.7397e-02,
1.1769e-02, 2.6985e-01],
...,
[-1.5065e+00, 1.3471e+00, -1.1466e-01, ..., -9.9367e-01,
-3.0058e-01, 2.3084e-01],
[-5.4846e-01, 1.6528e+00, 8.8733e-01, ..., 5.7921e-01,
5.5876e-01, 8.9860e-01],
[-1.6708e+00, -1.0654e+00, -1.5209e+00, ..., -2.0426e+00,
-1.0115e+00, -2.2001e-01]],
[[-1.4124e+00, -1.3189e+00, -2.6119e+00, ..., -2.5027e+00,
-2.2040e+00, -2.0964e+00],
[-7.8103e-01, -3.1195e-01, -3.3963e+00, ..., -3.1011e+00,
-3.1571e+00, -3.1359e+00],
[-2.2451e+00, -1.6139e+00, -5.7754e+00, ..., -4.5814e+00,
-4.7666e+00, -3.9180e+00],
...,
[-1.8123e+00, -6.8332e-01, -4.3593e+00, ..., -2.9648e+00,
-2.9566e+00, -2.4686e+00],
[-1.7703e+00, -8.3517e-01, -3.5820e+00, ..., -2.5468e+00,
-2.4629e+00, -1.8318e+00],
[-2.1599e+00, -1.4031e+00, -2.9862e+00, ..., -2.9289e+00,
-2.5818e+00, -2.3403e+00]],
[[-1.1859e+00, -1.7415e+00, -1.5642e+00, ..., -2.8043e+00,
-2.3382e+00, -1.8196e+00],
[-1.2014e+00, -1.6855e+00, -1.2845e+00, ..., -2.6555e+00,
-2.1279e+00, -1.7482e+00],
[-1.5635e+00, -1.8738e+00, -1.5063e+00, ..., -3.1245e+00,
-2.5012e+00, -1.9498e+00],
...,
[-2.3716e+00, -2.9539e+00, -2.7164e+00, ..., -4.3040e+00,
-3.4845e+00, -2.7333e+00],
[-1.5952e+00, -2.0289e+00, -1.9446e+00, ..., -2.9547e+00,
-2.4683e+00, -2.0420e+00],
[-1.0724e+00, -1.2844e+00, -1.3291e+00, ..., -1.9672e+00,
-1.6177e+00, -1.2593e+00]]],
[[[ 2.6567e-01, 5.2394e-01, 6.5080e-01, ..., 4.6600e-01,
1.0164e+00, 5.7221e-01],
[-2.1752e+00, -1.5588e+00, -3.2186e+00, ..., -2.8804e+00,
-9.8011e-01, -2.6446e-01],
[-2.0941e-01, -1.0183e+00, -2.5764e-01, ..., -1.8602e-01,
5.3290e-01, 6.2320e-01],
...,
[ 1.4156e-01, -6.2437e-01, -1.5007e+00, ..., -1.0115e+00,
-7.4130e-01, -5.4583e-01],
[ 1.5570e+00, 1.2632e+00, 6.5146e-01, ..., 7.4284e-01,
9.3763e-01, 9.3532e-01],
[ 1.1863e+00, 1.0114e+00, 7.3810e-01, ..., 8.9827e-01,
1.0323e+00, 8.5935e-01]],
[[-1.6835e+00, -1.2819e+00, -4.2035e+00, ..., -5.0376e+00,
-4.5774e+00, -3.4788e+00],
[ 4.9758e-01, 1.6806e+00, -2.3821e+00, ..., -3.4841e+00,
-3.6029e+00, -2.9439e+00],
[ 1.1886e+00, 1.4261e+00, -4.9800e+00, ..., -5.4452e+00,
-5.2896e+00, -5.0029e+00],
...,
[-6.4265e-01, 4.9829e-01, -4.7601e+00, ..., -5.8302e+00,
-5.2097e+00, -4.2050e+00],
[ 2.0831e+00, 2.7907e+00, 6.7117e-02, ..., -1.5083e+00,
-1.7200e+00, -1.6645e+00],
[ 1.2292e+00, 1.4987e+00, -1.6945e+00, ..., -2.7055e+00,
-2.8594e+00, -2.7071e+00]],
[[-9.9827e-02, -1.0612e+00, -1.2611e+00, ..., -5.3786e-01,
-1.8628e-01, -5.3522e-01],
[-5.1495e-01, -7.4664e-01, -7.2773e-01, ..., -3.4857e-01,
-7.7824e-03, -1.0121e-01],
[-4.3809e-01, -2.5426e-01, -3.7499e-01, ..., -3.6909e-01,
9.7502e-02, 9.8342e-02],
...,
[-1.1244e+00, -1.5472e+00, -6.4636e-01, ..., -6.0524e-01,
-4.6457e-01, -6.1985e-01],
[-1.3299e+00, -8.5066e-01, -4.2542e-01, ..., -7.2989e-02,
1.1683e-01, 9.1710e-02],
[-1.3378e+00, -6.4102e-01, -9.4362e-02, ..., -5.3822e-02,
-2.4604e-02, -5.1536e-02]],
...,
[[-2.9528e+00, -2.5316e+00, -2.7548e+00, ..., -3.6950e+00,
-2.8203e+00, -1.8809e+00],
[-2.2778e+00, -9.8172e-01, -1.5985e+00, ..., -2.7208e+00,
-2.4343e+00, -1.9503e+00],
[-1.1302e+00, 9.3036e-01, 5.1808e-01, ..., -4.5817e-01,
4.6055e-02, 1.9156e-01],
...,
[-1.5510e+00, 1.6005e+00, 3.1573e-01, ..., -7.8584e-01,
-1.5132e-01, 3.7984e-01],
[-5.9689e-01, 1.7550e+00, 1.0776e+00, ..., 1.6745e-01,
3.0060e-01, 4.8798e-01],
[-1.7170e+00, -1.0319e+00, -1.4668e+00, ..., -2.0788e+00,
-1.1905e+00, -5.5048e-01]],
[[-1.6187e+00, -1.6056e+00, -2.7733e+00, ..., -1.1975e+00,
-7.2823e-01, -7.4688e-01],
[-6.7364e-01, -1.6025e-02, -3.0820e+00, ..., -1.2526e+00,
-1.0975e+00, -1.5418e+00],
[-2.1421e+00, -1.0169e+00, -5.1730e+00, ..., -3.1945e+00,
-2.7014e+00, -2.2184e+00],
...,
[-1.8026e+00, -7.2945e-01, -4.6466e+00, ..., -2.4414e+00,
-2.0232e+00, -1.6010e+00],
[-1.7085e+00, -8.3233e-01, -3.7229e+00, ..., -2.2738e+00,
-1.6644e+00, -1.0987e+00],
[-2.0786e+00, -1.3561e+00, -3.0204e+00, ..., -2.7432e+00,
-1.9452e+00, -1.6672e+00]],
[[-1.3313e+00, -1.8821e+00, -1.7163e+00, ..., -1.8289e+00,
-1.5170e+00, -1.1502e+00],
[-1.2592e+00, -1.7384e+00, -1.2686e+00, ..., -1.3818e+00,
-1.0931e+00, -8.7217e-01],
[-1.5682e+00, -1.9257e+00, -1.5237e+00, ..., -1.5890e+00,
-1.1823e+00, -8.6573e-01],
...,
[-2.4685e+00, -3.0823e+00, -2.8247e+00, ..., -2.9834e+00,
-2.3808e+00, -1.8184e+00],
[-1.6617e+00, -2.1058e+00, -2.0049e+00, ..., -2.0846e+00,
-1.7070e+00, -1.3648e+00],
[-1.1097e+00, -1.3295e+00, -1.3575e+00, ..., -1.2917e+00,
-1.0895e+00, -8.1275e-01]]],
[[[ 1.9970e-01, 4.7031e-01, 6.3393e-01, ..., 6.4419e-01,
1.0093e+00, 8.3486e-01],
[-2.0323e+00, -1.3424e+00, -2.8400e+00, ..., -1.1608e+00,
1.6435e-01, 7.0969e-01],
[-1.7234e-01, -9.3478e-01, -3.2896e-02, ..., 5.2092e-01,
9.7919e-01, 9.9304e-01],
...,
[-1.0045e+00, -2.2058e+00, -2.0715e+00, ..., -1.3549e+00,
-8.7516e-01, -6.3357e-01],
[ 7.8282e-01, -1.4968e-01, -1.4144e-01, ..., 4.4328e-01,
8.1233e-01, 9.2744e-01],
[ 1.9618e-01, -5.8901e-01, -4.1328e-01, ..., 6.2810e-01,
8.9537e-01, 8.0641e-01]],
[[-1.6496e+00, -1.3455e+00, -4.2287e+00, ..., -5.7273e+00,
-4.8822e+00, -3.8046e+00],
[ 4.9593e-01, 1.4901e+00, -2.5440e+00, ..., -4.7356e+00,
-4.4721e+00, -3.6501e+00],
[ 1.2211e+00, 1.1967e+00, -5.0873e+00, ..., -6.8431e+00,
-6.4090e+00, -5.7765e+00],
...,
[ 4.0974e-01, 3.6933e-01, -4.1299e+00, ..., -6.0521e+00,
-5.1934e+00, -4.2759e+00],
[ 2.5605e+00, 2.4990e+00, 1.0761e-01, ..., -1.3625e+00,
-1.4478e+00, -1.5335e+00],
[ 1.5174e+00, 1.3231e+00, -1.6366e+00, ..., -2.7267e+00,
-2.8269e+00, -2.7766e+00]],
[[-1.2450e-01, -1.1552e+00, -1.3769e+00, ..., -7.7079e-01,
-5.3730e-01, -6.5949e-01],
[-4.9274e-01, -8.2391e-01, -8.6674e-01, ..., -6.2920e-01,
-3.5554e-01, -2.2087e-01],
[-3.3469e-01, -2.3652e-01, -4.4129e-01, ..., -3.7614e-01,
7.3958e-03, 1.1055e-01],
...,
[-8.4995e-02, -6.2171e-01, -6.1660e-01, ..., -6.7562e-01,
-4.6754e-01, -5.7346e-01],
[-7.2999e-01, -2.4509e-01, -5.8058e-01, ..., -2.8721e-01,
6.7085e-03, 8.6224e-02],
[-8.7998e-01, -2.7605e-01, -4.8635e-01, ..., -3.0262e-01,
-1.9168e-01, -1.3059e-01]],
...,
[[-2.8474e+00, -2.4546e+00, -2.4525e+00, ..., -3.1437e+00,
-2.5392e+00, -1.7144e+00],
[-2.1214e+00, -8.5824e-01, -1.1436e+00, ..., -1.7193e+00,
-1.7077e+00, -1.3795e+00],
[-1.1114e+00, 8.2257e-01, 8.0194e-01, ..., -3.3302e-01,
-7.3246e-02, 1.3981e-01],
...,
[-2.2573e+00, -1.5487e-01, -1.3104e-01, ..., -1.1030e+00,
-3.8725e-01, 1.6856e-01],
[-1.2171e+00, 9.1094e-01, 1.0876e+00, ..., 2.0356e-01,
3.4178e-01, 4.5970e-01],
[-2.1505e+00, -1.4589e+00, -1.6625e+00, ..., -2.2578e+00,
-1.2672e+00, -6.2952e-01]],
[[-1.6234e+00, -1.6293e+00, -2.8039e+00, ..., -1.3529e+00,
-7.4852e-01, -5.3672e-01],
[-6.9517e-01, -1.0252e-01, -3.1625e+00, ..., -2.0373e+00,
-1.6546e+00, -1.7117e+00],
[-2.0593e+00, -9.6555e-01, -5.0918e+00, ..., -4.0042e+00,
-3.3547e+00, -2.5823e+00],
...,
[-1.3132e+00, 2.2546e-01, -3.3628e+00, ..., -2.4052e+00,
-1.8002e+00, -1.3338e+00],
[-1.6585e+00, -4.4304e-01, -3.0017e+00, ..., -2.2824e+00,
-1.5058e+00, -8.7752e-01],
[-2.3229e+00, -1.3612e+00, -3.0346e+00, ..., -2.8862e+00,
-1.9230e+00, -1.6254e+00]],
[[-1.3037e+00, -1.8282e+00, -1.6444e+00, ..., -1.8596e+00,
-1.5090e+00, -1.1140e+00],
[-1.2408e+00, -1.7091e+00, -1.2197e+00, ..., -1.6356e+00,
-1.3018e+00, -9.9743e-01],
[-1.5620e+00, -1.9059e+00, -1.4836e+00, ..., -1.9083e+00,
-1.4348e+00, -1.0256e+00],
...,
[-2.4584e+00, -3.0669e+00, -2.7860e+00, ..., -3.1632e+00,
-2.4897e+00, -1.8513e+00],
[-1.5913e+00, -2.0899e+00, -1.9831e+00, ..., -2.2232e+00,
-1.8034e+00, -1.4167e+00],
[-1.0343e+00, -1.3464e+00, -1.3590e+00, ..., -1.3742e+00,
-1.1586e+00, -8.3835e-01]]]], device='cuda:0'),
tensor([[[[ 6.0445e-02, 2.2389e-01, -1.4127e+00, ..., -6.2515e-01,
-1.6677e-01, -3.8756e-02],
[-2.1001e+00, -1.6123e+00, -4.5134e+00, ..., -3.8324e+00,
-1.4943e+00, -5.4556e-01],
[ 5.6931e-01, -7.3516e-02, -8.5985e-01, ..., -2.2464e-01,
-2.1407e-01, 2.8083e-01],
...,
[-3.0476e-01, -1.1099e+00, -1.4454e+00, ..., -1.3617e+00,
-9.4295e-01, -7.0562e-01],
[ 9.8720e-01, 5.7731e-01, 3.3543e-01, ..., 3.7457e-01,
6.7127e-01, 7.7890e-01],
[ 6.7860e-01, 4.3634e-01, 3.2566e-01, ..., 5.2106e-01,
7.6008e-01, 6.8943e-01]],
[[-3.2953e+00, -1.7975e+00, -6.2388e+00, ..., -7.1567e+00,
-6.2829e+00, -4.6393e+00],
[-8.2814e-01, 1.5188e+00, -4.3626e+00, ..., -5.4812e+00,
-5.5522e+00, -4.2821e+00],
[-3.9323e-01, 1.0874e+00, -7.4020e+00, ..., -8.0175e+00,
-7.8447e+00, -6.7076e+00],
...,
[-4.8789e-01, 2.4748e-01, -4.4214e+00, ..., -5.3342e+00,
-4.6564e+00, -3.8836e+00],
[ 1.9382e+00, 2.4361e+00, -5.5130e-02, ..., -1.0564e+00,
-1.2274e+00, -1.3412e+00],
[ 1.1102e+00, 1.3159e+00, -1.6731e+00, ..., -2.4370e+00,
-2.5699e+00, -2.5449e+00]],
[[-1.3173e+00, -2.3251e+00, -2.2956e+00, ..., -1.7618e+00,
-1.2545e+00, -1.3968e+00],
[-1.6780e+00, -1.9951e+00, -1.5378e+00, ..., -1.2407e+00,
-7.1166e-01, -8.1427e-01],
[-2.1630e+00, -1.9431e+00, -1.0224e+00, ..., -1.0942e+00,
-4.4010e-01, -1.8754e-01],
...,
[-6.4032e-01, -1.0987e+00, -5.7849e-01, ..., -6.0721e-01,
-4.7701e-01, -6.3445e-01],
[-1.0369e+00, -6.9048e-01, -5.6778e-01, ..., -2.7393e-01,
-6.6709e-02, -3.4917e-02],
[-1.1927e+00, -6.6091e-01, -3.8432e-01, ..., -3.2168e-01,
-2.7830e-01, -2.4374e-01]],
...,
[[-2.0136e+00, -1.0890e+00, -2.3386e+00, ..., -3.2411e+00,
-2.6925e+00, -1.8542e+00],
[-1.7593e+00, 3.4778e-02, -1.3521e+00, ..., -2.4858e+00,
-2.5646e+00, -1.9976e+00],
[-3.6312e-01, 2.4021e+00, 1.0820e+00, ..., 1.1818e-01,
2.3623e-01, 5.1469e-01],
...,
[-1.5717e+00, 9.3578e-01, -1.3795e-01, ..., -1.1935e+00,
-4.5333e-01, 6.6479e-02],
[-5.9706e-01, 1.3606e+00, 7.6117e-01, ..., -3.8114e-03,
1.8188e-01, 3.0288e-01],
[-1.6179e+00, -1.1191e+00, -1.5420e+00, ..., -2.3306e+00,
-1.3626e+00, -7.3046e-01]],
[[-2.4239e+00, -2.8268e+00, -4.2287e+00, ..., -2.3880e+00,
-1.9242e+00, -1.7335e+00],
[-1.7833e+00, -1.6597e+00, -4.9001e+00, ..., -2.8947e+00,
-2.7942e+00, -2.8247e+00],
[-2.8307e+00, -2.4662e+00, -6.6775e+00, ..., -4.4393e+00,
-4.3885e+00, -3.5403e+00],
...,
[-1.7166e+00, -5.6892e-01, -4.0611e+00, ..., -2.0714e+00,
-1.5386e+00, -1.2023e+00],
[-1.6778e+00, -7.3668e-01, -3.3728e+00, ..., -2.0612e+00,
-1.3381e+00, -7.8190e-01],
[-2.0445e+00, -1.2806e+00, -2.8401e+00, ..., -2.7052e+00,
-1.7979e+00, -1.5197e+00]],
[[-1.8063e+00, -2.5327e+00, -2.4532e+00, ..., -2.7015e+00,
-2.2259e+00, -1.7044e+00],
[-1.8029e+00, -2.5102e+00, -2.1529e+00, ..., -2.4137e+00,
-1.9092e+00, -1.5526e+00],
[-2.2591e+00, -2.8228e+00, -2.5216e+00, ..., -2.7638e+00,
-2.1534e+00, -1.6387e+00],
...,
[-2.2713e+00, -2.8594e+00, -2.6638e+00, ..., -2.8830e+00,
-2.2986e+00, -1.7303e+00],
[-1.5402e+00, -1.9819e+00, -1.9346e+00, ..., -2.0373e+00,
-1.6715e+00, -1.3327e+00],
[-1.0453e+00, -1.2744e+00, -1.3404e+00, ..., -1.2641e+00,
-1.0788e+00, -7.9586e-01]]],
[[[ 9.0732e-02, 2.4352e-01, -1.3941e+00, ..., -5.9335e-01,
-2.1941e-01, -9.0303e-02],
[-2.2363e+00, -1.6876e+00, -4.6736e+00, ..., -3.5334e+00,
-1.4320e+00, -5.7575e-01],
[ 6.2101e-01, -9.5338e-02, -8.2298e-01, ..., -1.3104e-01,
-2.5478e-01, 2.2375e-01],
...,
[-8.4983e-01, -1.8088e+00, -1.8756e+00, ..., -1.8259e+00,
-1.3601e+00, -9.6923e-01],
[ 7.3061e-01, -8.3310e-03, -1.4579e-01, ..., -3.3592e-02,
6.8546e-01, 7.9448e-01],
[ 1.2436e-01, -5.7463e-01, -4.8562e-01, ..., -1.6123e-01,
4.2255e-01, 3.4922e-01]],
[[-3.3636e+00, -1.6888e+00, -6.2762e+00, ..., -6.9652e+00,
-6.0772e+00, -4.5103e+00],
[-6.1062e-01, 1.9742e+00, -4.1344e+00, ..., -5.5143e+00,
-5.4833e+00, -4.2466e+00],
[-2.6305e-02, 1.5385e+00, -7.3648e+00, ..., -7.9379e+00,
-7.6806e+00, -6.5539e+00],
...,
[ 2.5609e-01, 2.9876e-01, -4.0473e+00, ..., -4.6227e+00,
-4.1800e+00, -3.4864e+00],
[ 2.2420e+00, 2.2199e+00, -2.1321e-01, ..., -7.6937e-01,
-1.1384e+00, -1.2076e+00],
[ 1.3018e+00, 1.1452e+00, -1.7954e+00, ..., -2.2193e+00,
-2.5018e+00, -2.4128e+00]],
[[-1.4129e+00, -2.5454e+00, -2.4886e+00, ..., -1.6655e+00,
-1.1938e+00, -1.3234e+00],
[-1.8590e+00, -2.1828e+00, -1.6316e+00, ..., -1.1350e+00,
-6.9949e-01, -8.2293e-01],
[-2.3303e+00, -2.0386e+00, -1.0253e+00, ..., -1.0298e+00,
-4.5991e-01, -2.1702e-01],
...,
[-1.1550e-01, -5.5434e-01, -4.5388e-01, ..., -5.0225e-01,
-2.0042e-01, -4.5261e-01],
[-6.5335e-01, -1.4356e-01, -4.0439e-01, ..., -2.0120e-01,
2.5146e-01, 2.4176e-01],
[-7.3292e-01, -1.3943e-01, -3.2270e-01, ..., -3.9622e-01,
-1.8971e-01, -2.3130e-01]],
...,
[[-2.2742e+00, -1.1884e+00, -2.5342e+00, ..., -2.8372e+00,
-2.3708e+00, -1.6997e+00],
[-1.8000e+00, 1.9833e-01, -1.3651e+00, ..., -2.3255e+00,
-2.3966e+00, -1.8746e+00],
[-3.8213e-01, 2.6448e+00, 1.1897e+00, ..., 1.2068e-01,
2.5161e-01, 4.6665e-01],
...,
[-2.0489e+00, -1.5738e-01, -1.9786e-01, ..., -1.3964e+00,
-4.2536e-01, 9.0502e-02],
[-1.2106e+00, 7.3144e-01, 9.0671e-01, ..., 1.2099e-02,
3.1801e-01, 4.7832e-01],
[-2.1482e+00, -1.4791e+00, -1.6724e+00, ..., -2.4836e+00,
-1.4777e+00, -9.2299e-01]],
[[-2.4087e+00, -2.7810e+00, -4.3453e+00, ..., -2.5021e+00,
-2.0251e+00, -1.7962e+00],
[-1.6485e+00, -1.4239e+00, -5.0554e+00, ..., -3.0401e+00,
-2.8347e+00, -2.7743e+00],
[-2.8129e+00, -2.3283e+00, -6.9883e+00, ..., -4.4132e+00,
-4.2965e+00, -3.4662e+00],
...,
[-1.4150e+00, -5.7409e-02, -3.3018e+00, ..., -1.4102e+00,
-1.0009e+00, -9.1794e-01],
[-1.7294e+00, -6.5575e-01, -3.0069e+00, ..., -1.7113e+00,
-1.1520e+00, -8.0611e-01],
[-2.3366e+00, -1.5088e+00, -3.1186e+00, ..., -2.9586e+00,
-2.1519e+00, -1.9933e+00]],
[[-1.8673e+00, -2.6282e+00, -2.5008e+00, ..., -2.6537e+00,
-2.1962e+00, -1.6695e+00],
[-1.8478e+00, -2.5842e+00, -2.1367e+00, ..., -2.4270e+00,
-1.9356e+00, -1.5466e+00],
[-2.3086e+00, -2.8739e+00, -2.4904e+00, ..., -2.8021e+00,
-2.2000e+00, -1.6565e+00],
...,
[-2.3767e+00, -2.9500e+00, -2.7426e+00, ..., -2.8611e+00,
-2.2882e+00, -1.7590e+00],
[-1.5359e+00, -2.0034e+00, -1.9439e+00, ..., -1.9346e+00,
-1.5971e+00, -1.2881e+00],
[-9.9362e-01, -1.2888e+00, -1.3322e+00, ..., -1.1772e+00,
-1.0100e+00, -7.6533e-01]]],
[[[ 5.1185e-01, 9.9615e-01, 8.8228e-01, ..., -3.3429e-01,
-6.6611e-02, 2.8669e-02],
[-1.3608e+00, -3.9860e-01, -1.6927e+00, ..., -3.2330e+00,
-1.2943e+00, -4.3445e-01],
[ 6.6292e-01, 4.3725e-01, 3.5017e-01, ..., -1.5138e-01,
-3.2721e-01, 1.9191e-01],
...,
[-1.9382e-01, -8.0478e-01, -1.4021e+00, ..., -1.8085e+00,
-1.8411e+00, -1.2346e+00],
[ 1.0040e+00, 7.3765e-01, 2.7620e-01, ..., 1.2424e-01,
2.8975e-01, 4.6316e-01],
[ 6.6027e-01, 4.6966e-01, 1.6845e-01, ..., 3.0845e-01,
4.9712e-01, 5.1667e-01]],
[[-2.3691e+00, -9.9151e-01, -4.1166e+00, ..., -6.8478e+00,
-6.0703e+00, -4.5283e+00],
[-7.5060e-02, 2.1772e+00, -2.2630e+00, ..., -5.4333e+00,
-5.5537e+00, -4.3015e+00],
[ 7.6971e-01, 1.9998e+00, -4.9895e+00, ..., -7.8240e+00,
-7.8128e+00, -6.6207e+00],
...,
[-5.3543e-01, 5.2471e-01, -4.1486e+00, ..., -7.1135e+00,
-6.6604e+00, -5.1621e+00],
[ 1.9072e+00, 2.5639e+00, 7.8488e-02, ..., -1.9789e+00,
-2.5784e+00, -2.0635e+00],
[ 1.0841e+00, 1.3766e+00, -1.6153e+00, ..., -2.9655e+00,
-3.4874e+00, -3.0362e+00]],
[[-6.7305e-01, -1.7909e+00, -1.5741e+00, ..., -1.4097e+00,
-1.0484e+00, -1.3334e+00],
[-1.4838e+00, -1.8451e+00, -1.1221e+00, ..., -9.8852e-01,
-6.2532e-01, -9.0010e-01],
[-1.2901e+00, -1.0841e+00, -4.0053e-01, ..., -1.0951e+00,
-5.9931e-01, -4.7303e-01],
...,
[-8.5023e-01, -1.2549e+00, -4.9168e-01, ..., -1.0352e+00,
-8.7196e-01, -1.2368e+00],
[-1.2173e+00, -8.2903e-01, -5.2193e-01, ..., -5.4258e-01,
-2.4923e-01, -4.0671e-01],
[-1.2963e+00, -7.4395e-01, -3.4847e-01, ..., -4.8626e-01,
-4.3276e-01, -5.9526e-01]],
...,
[[-2.4527e+00, -1.3358e+00, -2.7378e+00, ..., -3.0999e+00,
-2.5368e+00, -1.8415e+00],
[-7.4663e-01, 1.0983e+00, -9.7975e-01, ..., -2.4361e+00,
-2.5061e+00, -2.0224e+00],
[-2.2734e-01, 2.4933e+00, 5.9711e-01, ..., -9.4909e-02,
1.0748e-02, 2.6753e-01],
...,
[-1.4416e+00, 1.1796e+00, -1.8214e-01, ..., -9.8597e-01,
-2.6793e-01, 2.7964e-01],
[-4.7902e-01, 1.5191e+00, 7.5684e-01, ..., 3.7466e-01,
4.6040e-01, 7.9012e-01],
[-1.5321e+00, -9.7110e-01, -1.4791e+00, ..., -2.2084e+00,
-1.1733e+00, -3.5531e-01]],
[[-1.2869e+00, -1.1653e+00, -2.7238e+00, ..., -2.2125e+00,
-1.8880e+00, -1.7957e+00],
[-4.8264e-01, 1.2164e-01, -3.4655e+00, ..., -2.8477e+00,
-2.8600e+00, -2.8921e+00],
[-2.0747e+00, -1.2543e+00, -6.0062e+00, ..., -4.4561e+00,
-4.5261e+00, -3.7250e+00],
...,
[-1.6385e+00, -5.4908e-01, -4.0568e+00, ..., -2.8557e+00,
-2.7444e+00, -2.2864e+00],
[-1.6457e+00, -7.5688e-01, -3.3797e+00, ..., -2.4981e+00,
-2.2744e+00, -1.6566e+00],
[-2.0579e+00, -1.3637e+00, -2.9151e+00, ..., -2.9012e+00,
-2.4226e+00, -2.1814e+00]],
[[-1.2078e+00, -1.8030e+00, -1.5709e+00, ..., -2.5642e+00,
-2.1395e+00, -1.6645e+00],
[-1.1967e+00, -1.6934e+00, -1.1871e+00, ..., -2.3959e+00,
-1.9255e+00, -1.5785e+00],
[-1.5507e+00, -1.8475e+00, -1.3736e+00, ..., -2.8091e+00,
-2.2397e+00, -1.7411e+00],
...,
[-2.2739e+00, -2.8332e+00, -2.6422e+00, ..., -3.9978e+00,
-3.2412e+00, -2.5354e+00],
[-1.5341e+00, -1.9455e+00, -1.8943e+00, ..., -2.7602e+00,
-2.3063e+00, -1.9009e+00],
[-1.0276e+00, -1.2410e+00, -1.3066e+00, ..., -1.8181e+00,
-1.5163e+00, -1.1699e+00]]],
[[[ 3.1807e-01, 6.4477e-01, -1.7533e-01, ..., 6.6366e-01,
1.0445e+00, 7.3695e-01],
[-1.6587e+00, -9.3777e-01, -3.1514e+00, ..., -1.6937e+00,
-2.4469e-01, 3.0488e-01],
[ 8.0004e-01, 3.0667e-01, -3.4172e-01, ..., 2.0428e-01,
7.3677e-01, 7.7385e-01],
...,
[-1.0557e+01, -1.0202e+01, -2.1995e+01, ..., -1.9427e+00,
-1.1356e+00, -8.1074e-01],
[-3.8297e+00, 3.2712e-01, -1.3859e+01, ..., -2.6059e-01,
5.1282e-01, 6.1318e-01],
[-3.4834e+00, -2.0763e+00, -1.3126e+01, ..., -1.2295e-01,
4.0810e-01, 2.9170e-01]],
[[-3.0938e+00, -1.6807e+00, -5.4689e+00, ..., -4.9418e+00,
-4.3812e+00, -3.3901e+00],
[-6.2859e-01, 1.7070e+00, -3.4790e+00, ..., -3.7119e+00,
-3.7002e+00, -3.0373e+00],
[ 8.1257e-02, 1.4205e+00, -6.5109e+00, ..., -5.6538e+00,
-5.4016e+00, -5.0221e+00],
...,
[-3.1049e+00, -1.1110e-01, -8.6039e+00, ..., -5.2241e+00,
-4.5349e+00, -3.7983e+00],
[-4.8066e-01, 2.0425e+00, -2.6755e+00, ..., -7.4609e-01,
-1.2295e+00, -1.4101e+00],
[-1.6457e+00, -1.3635e-01, -2.2008e+00, ..., -2.2330e+00,
-2.6233e+00, -2.7310e+00]],
[[-7.8991e-01, -2.0180e+00, -1.9315e+00, ..., -6.6368e-01,
-3.2115e-01, -5.0626e-01],
[-1.4490e+00, -1.8976e+00, -1.3272e+00, ..., -4.8232e-01,
-1.1892e-01, -6.3230e-02],
[-1.6394e+00, -1.6045e+00, -9.4308e-01, ..., -2.7518e-01,
1.5906e-01, 2.0428e-01],
...,
[-4.7935e+00, -2.8017e+00, 5.3747e+00, ..., -3.7971e-01,
-1.6168e-01, -5.7651e-01],
[-5.3879e+00, -2.2051e+00, 4.7182e+00, ..., -2.7792e-01,
1.8463e-01, 1.4809e-01],
[-5.7759e+00, -3.1383e+00, 1.8845e+00, ..., -2.6965e-01,
-1.3279e-01, -1.7545e-01]],
...,
[[-2.3097e+00, -1.2337e+00, -2.6440e+00, ..., -3.2976e+00,
-2.6247e+00, -1.7415e+00],
[-1.0580e+00, 7.8631e-01, -1.1166e+00, ..., -2.0841e+00,
-2.0259e+00, -1.6134e+00],
[-3.3027e-01, 2.5733e+00, 9.2643e-01, ..., -4.6373e-01,
-3.3157e-02, 1.8984e-01],
...,
[-2.1612e+00, -8.4200e+00, -1.4451e+01, ..., -1.4632e+00,
-5.6680e-01, 1.1503e-01],
[-5.3210e+00, -6.5250e+00, -7.9041e+00, ..., -1.7772e-01,
3.2668e-02, 4.3980e-01],
[-1.0261e+01, -7.1301e+00, -5.8435e+00, ..., -3.1059e+00,
-1.8588e+00, -7.8365e-01]],
[[-1.7688e+00, -1.9433e+00, -3.6893e+00, ..., -1.0424e+00,
-5.9569e-01, -5.6856e-01],
[-1.0343e+00, -7.0372e-01, -4.5104e+00, ..., -1.4851e+00,
-1.2760e+00, -1.5263e+00],
[-2.4172e+00, -1.8597e+00, -6.8469e+00, ..., -3.4498e+00,
-2.9062e+00, -2.2633e+00],
...,
[-4.5938e+00, -7.5303e+00, -1.1918e+01, ..., -1.4421e+00,
-1.1378e+00, -9.4307e-01],
[-6.5703e+00, -1.1153e+01, -1.7735e+01, ..., -1.3856e+00,
-8.2983e-01, -4.8001e-01],
[-4.2596e+00, -6.4668e+00, -1.1790e+01, ..., -2.4465e+00,
-1.5706e+00, -1.5147e+00]],
[[-1.5242e+00, -2.2436e+00, -2.1148e+00, ..., -1.7284e+00,
-1.4370e+00, -1.0807e+00],
[-1.5426e+00, -2.2041e+00, -1.8002e+00, ..., -1.4071e+00,
-1.1302e+00, -8.8332e-01],
[-1.9789e+00, -2.4600e+00, -2.1115e+00, ..., -1.6218e+00,
-1.2324e+00, -8.9688e-01],
...,
[-7.7330e+00, -1.1368e+01, -1.2270e+01, ..., -3.4078e+00,
-2.6877e+00, -2.0549e+00],
[-5.3869e+00, -8.2204e+00, -9.3239e+00, ..., -2.3163e+00,
-1.9270e+00, -1.5211e+00],
[-3.5274e+00, -4.5818e+00, -5.4355e+00, ..., -1.5123e+00,
-1.2929e+00, -9.5807e-01]]],
[[[ 2.1652e-01, 4.9340e-01, 6.2940e-01, ..., 6.3795e-01,
1.0186e+00, 7.9605e-01],
[-2.0454e+00, -1.3632e+00, -2.8875e+00, ..., -1.4582e+00,
1.9806e-02, 5.9728e-01],
[-1.6746e-01, -9.1641e-01, -3.2515e-03, ..., 4.7124e-01,
9.8115e-01, 9.5230e-01],
...,
[-9.9948e-01, -2.2011e+00, -2.0951e+00, ..., -1.1675e+00,
-6.8642e-01, -4.4785e-01],
[ 7.8502e-01, -1.4300e-01, -1.3958e-01, ..., 4.2334e-01,
8.1202e-01, 9.2690e-01],
[ 1.8371e-01, -6.0141e-01, -4.2361e-01, ..., 5.7741e-01,
8.6378e-01, 7.8738e-01]],
[[-1.6796e+00, -1.3976e+00, -4.2615e+00, ..., -5.9391e+00,
-5.0592e+00, -3.9187e+00],
[ 4.8024e-01, 1.4617e+00, -2.5172e+00, ..., -4.6949e+00,
-4.4916e+00, -3.6516e+00],
[ 1.2059e+00, 1.1871e+00, -5.0465e+00, ..., -6.8893e+00,
-6.4779e+00, -5.8992e+00],
...,
[ 4.3423e-01, 4.0283e-01, -4.1166e+00, ..., -6.0031e+00,
-5.1888e+00, -4.2806e+00],
[ 2.5847e+00, 2.5359e+00, 1.4011e-01, ..., -1.6462e+00,
-1.7006e+00, -1.7174e+00],
[ 1.5429e+00, 1.3454e+00, -1.6111e+00, ..., -2.8217e+00,
-2.8625e+00, -2.7569e+00]],
[[-1.3816e-01, -1.1612e+00, -1.4008e+00, ..., -8.7387e-01,
-5.8612e-01, -7.2179e-01],
[-5.1046e-01, -8.2226e-01, -8.8141e-01, ..., -7.4520e-01,
-3.9302e-01, -2.3057e-01],
[-3.3161e-01, -2.1236e-01, -4.3321e-01, ..., -3.8831e-01,
1.5470e-02, 1.2248e-01],
...,
[-7.8995e-02, -6.1662e-01, -6.0264e-01, ..., -6.0973e-01,
-4.2343e-01, -5.1223e-01],
[-7.3019e-01, -2.3830e-01, -5.8284e-01, ..., -2.6603e-01,
-3.5465e-03, 6.8744e-02],
[-9.0030e-01, -2.8403e-01, -4.8695e-01, ..., -2.9596e-01,
-1.9033e-01, -1.3824e-01]],
...,
[[-2.8498e+00, -2.4884e+00, -2.4774e+00, ..., -3.4533e+00,
-2.8250e+00, -1.8751e+00],
[-2.1175e+00, -8.7006e-01, -1.1355e+00, ..., -1.7661e+00,
-1.8671e+00, -1.5806e+00],
[-1.1297e+00, 7.7265e-01, 7.6164e-01, ..., -2.6251e-01,
4.6335e-02, 1.9891e-01],
...,
[-2.2813e+00, -1.7129e-01, -1.5519e-01, ..., -1.0191e+00,
-4.0500e-01, 1.0889e-01],
[-1.2486e+00, 8.9102e-01, 1.0789e+00, ..., 1.8263e-01,
2.7054e-01, 3.9536e-01],
[-2.1639e+00, -1.4851e+00, -1.6756e+00, ..., -2.0253e+00,
-1.1666e+00, -5.6474e-01]],
[[-1.6408e+00, -1.6360e+00, -2.7753e+00, ..., -1.2079e+00,
-6.0463e-01, -4.2860e-01],
[-6.9644e-01, -8.0711e-02, -3.1112e+00, ..., -1.9117e+00,
-1.5624e+00, -1.7066e+00],
[-2.0700e+00, -9.7126e-01, -5.0828e+00, ..., -4.0515e+00,
-3.4164e+00, -2.6125e+00],
...,
[-1.3343e+00, 2.2517e-01, -3.3906e+00, ..., -2.5067e+00,
-1.9078e+00, -1.4163e+00],
[-1.6618e+00, -4.3819e-01, -3.0205e+00, ..., -2.3191e+00,
-1.5819e+00, -9.6144e-01],
[-2.3202e+00, -1.3470e+00, -3.0305e+00, ..., -2.8072e+00,
-1.9140e+00, -1.5966e+00]],
[[-1.3043e+00, -1.8357e+00, -1.6593e+00, ..., -1.9095e+00,
-1.5469e+00, -1.1362e+00],
[-1.2419e+00, -1.7056e+00, -1.2228e+00, ..., -1.6150e+00,
-1.2830e+00, -9.8190e-01],
[-1.5528e+00, -1.8953e+00, -1.4711e+00, ..., -1.8672e+00,
-1.3989e+00, -9.8336e-01],
...,
[-2.4615e+00, -3.0736e+00, -2.7916e+00, ..., -3.0590e+00,
-2.4055e+00, -1.7930e+00],
[-1.5958e+00, -2.0995e+00, -1.9908e+00, ..., -2.1793e+00,
-1.7595e+00, -1.3789e+00],
[-1.0432e+00, -1.3538e+00, -1.3646e+00, ..., -1.3727e+00,
-1.1390e+00, -8.2973e-01]]]], device='cuda:0'),
tensor([[[[ 5.9554e-02, 2.0851e-01, -1.4638e+00, ..., -6.6124e-01,
-2.3852e-01, -9.1452e-02],
[-2.0989e+00, -1.6437e+00, -4.5974e+00, ..., -7.0726e+00,
-4.3277e+00, -2.7554e+00],
[ 5.6230e-01, -1.3769e-01, -8.9269e-01, ..., -6.7267e+00,
-5.9193e+00, -5.0873e+00],
...,
[-9.0659e-01, -1.8608e+00, -1.8856e+00, ..., -1.7750e+00,
-1.3021e+00, -9.3030e-01],
[ 6.8534e-01, -6.6848e-02, -1.9535e-01, ..., 1.2266e-02,
6.9659e-01, 7.8804e-01],
[ 1.0995e-01, -5.9233e-01, -5.0943e-01, ..., -1.4364e-01,
4.2722e-01, 3.4096e-01]],
[[-3.3086e+00, -1.7792e+00, -6.2770e+00, ..., -7.1783e+00,
-6.2840e+00, -4.6507e+00],
[-8.0490e-01, 1.5553e+00, -4.3856e+00, ..., -5.2138e+00,
-5.7755e+00, -4.9241e+00],
[-3.5848e-01, 1.0876e+00, -7.4896e+00, ..., -8.9259e+00,
-9.4874e+00, -8.5013e+00],
...,
[ 2.9206e-01, 2.9003e-01, -4.0195e+00, ..., -4.7111e+00,
-4.2223e+00, -3.5225e+00],
[ 2.2685e+00, 2.2032e+00, -1.8176e-01, ..., -9.3839e-01,
-1.2406e+00, -1.2690e+00],
[ 1.3251e+00, 1.1238e+00, -1.7905e+00, ..., -2.2657e+00,
-2.5143e+00, -2.3987e+00]],
[[-1.3584e+00, -2.3849e+00, -2.3566e+00, ..., -1.7535e+00,
-1.2574e+00, -1.3989e+00],
[-1.7146e+00, -2.0124e+00, -1.5501e+00, ..., -1.2875e+00,
-1.1980e+00, -1.2357e+00],
[-2.2273e+00, -1.9754e+00, -1.0235e+00, ..., -1.0543e+00,
-1.1025e+00, -6.7389e-01],
...,
[-1.7416e-01, -6.2336e-01, -4.4650e-01, ..., -5.0252e-01,
-1.9557e-01, -4.3676e-01],
[-7.2223e-01, -2.2266e-01, -4.2753e-01, ..., -2.0941e-01,
2.3989e-01, 2.3686e-01],
[-8.1165e-01, -2.1478e-01, -3.5139e-01, ..., -3.8626e-01,
-1.7726e-01, -2.1323e-01]],
...,
[[-2.0260e+00, -1.0616e+00, -2.3226e+00, ..., -3.1650e+00,
-2.6023e+00, -1.8289e+00],
[-1.8038e+00, 3.4445e-02, -1.3484e+00, ..., -2.1894e+00,
-4.0788e+00, -4.1254e+00],
[-3.7485e-01, 2.3985e+00, 1.0913e+00, ..., -2.8337e+00,
-4.3358e+00, -3.2356e+00],
...,
[-2.0078e+00, -4.6941e-02, -1.5595e-01, ..., -1.3301e+00,
-3.6977e-01, 1.0620e-01],
[-1.1582e+00, 8.4703e-01, 9.8163e-01, ..., 6.6520e-03,
3.1283e-01, 4.7284e-01],
[-2.0543e+00, -1.3560e+00, -1.5600e+00, ..., -2.4147e+00,
-1.4357e+00, -9.2166e-01]],
[[-2.4421e+00, -2.8666e+00, -4.2702e+00, ..., -2.4220e+00,
-1.9314e+00, -1.7329e+00],
[-1.7832e+00, -1.6590e+00, -4.9357e+00, ..., -2.2882e+00,
-1.8854e+00, -2.0683e+00],
[-2.8374e+00, -2.4708e+00, -6.6907e+00, ..., -3.1072e+00,
-3.7422e+00, -3.1106e+00],
...,
[-1.3931e+00, -4.5081e-02, -3.2695e+00, ..., -1.5070e+00,
-1.0532e+00, -9.4197e-01],
[-1.6803e+00, -6.0803e-01, -2.9508e+00, ..., -1.7609e+00,
-1.1800e+00, -8.1271e-01],
[-2.2899e+00, -1.4427e+00, -3.0341e+00, ..., -2.9496e+00,
-2.1433e+00, -1.9740e+00]],
[[-1.8285e+00, -2.5580e+00, -2.4762e+00, ..., -2.7288e+00,
-2.2416e+00, -1.7146e+00],
[-1.8270e+00, -2.5396e+00, -2.1732e+00, ..., -2.9296e+00,
-2.3471e+00, -1.7859e+00],
[-2.2831e+00, -2.8486e+00, -2.5454e+00, ..., -4.2419e+00,
-3.2882e+00, -2.4006e+00],
...,
[-2.3984e+00, -2.9735e+00, -2.7742e+00, ..., -2.8533e+00,
-2.2818e+00, -1.7534e+00],
[-1.5640e+00, -2.0337e+00, -1.9772e+00, ..., -1.9522e+00,
-1.6039e+00, -1.2907e+00],
[-1.0188e+00, -1.3148e+00, -1.3607e+00, ..., -1.1965e+00,
-1.0161e+00, -7.6925e-01]]],
[[[ 1.1589e-01, 4.0602e-01, 3.9285e-01, ..., -1.7411e+01,
-1.2800e+01, -1.3156e+01],
[-1.7932e+00, -9.9967e-01, -2.6855e+00, ..., -2.4326e+01,
-1.6629e+01, -1.8676e+01],
[ 4.7292e-01, -3.3802e-02, 1.0451e-02, ..., -2.9877e+01,
-2.2147e+01, -1.4667e+01],
...,
[-5.7526e-01, -1.5178e+00, -1.6258e+00, ..., -1.7073e+00,
-1.2901e+00, -8.6880e-01],
[ 9.8355e-01, 3.6323e-01, 2.4982e-01, ..., 1.0370e-01,
5.4539e-01, 7.2143e-01],
[ 5.0216e-01, -2.6556e-03, 6.5447e-02, ..., 3.8491e-01,
7.0391e-01, 6.7732e-01]],
[[-2.4305e+00, -1.4661e+00, -4.8063e+00, ..., -1.1706e+01,
-1.4079e+01, -1.1320e+01],
[ 9.8320e-04, 1.7065e+00, -2.7772e+00, ..., -1.3228e+01,
-1.4265e+01, -9.2505e+00],
[ 1.1195e+00, 1.6527e+00, -5.1929e+00, ..., -1.5059e+01,
-1.5353e+01, -1.0985e+01],
...,
[-6.3098e-02, 2.4019e-01, -4.3368e+00, ..., -6.7231e+00,
-5.9129e+00, -4.7540e+00],
[ 2.1850e+00, 2.3569e+00, -8.8829e-02, ..., -1.7966e+00,
-1.9887e+00, -1.8522e+00],
[ 1.2754e+00, 1.2481e+00, -1.7101e+00, ..., -2.9874e+00,
-3.1613e+00, -2.9847e+00]],
[[-3.6526e-01, -1.4500e+00, -1.5639e+00, ..., 6.4597e-02,
-3.6034e+00, -3.5987e+00],
[-1.0129e+00, -1.4083e+00, -1.3692e+00, ..., 1.4120e+00,
-2.4493e+00, -2.0994e+00],
[-8.5861e-01, -6.6715e-01, -6.9653e-01, ..., 1.4875e+00,
-3.1700e+00, -1.9095e+00],
...,
[-3.7533e-01, -8.9625e-01, -5.9809e-01, ..., -8.8810e-01,
-6.4057e-01, -7.5413e-01],
[-8.6230e-01, -4.5149e-01, -5.1773e-01, ..., -4.8565e-01,
-1.4506e-01, -7.8398e-02],
[-9.8660e-01, -4.3336e-01, -3.6965e-01, ..., -4.3478e-01,
-2.8627e-01, -2.2710e-01]],
...,
[[-2.6359e+00, -1.8145e+00, -2.5290e+00, ..., -1.5538e+01,
-1.0809e+01, -7.5936e+00],
[-1.3588e+00, 5.2495e-01, -3.8013e-01, ..., -2.3922e+01,
-1.0248e+01, -4.3503e+00],
[-9.2190e-01, 1.5556e+00, 1.0599e+00, ..., -2.6383e+01,
-9.4419e+00, -2.8362e+00],
...,
[-1.7978e+00, 4.8790e-01, 2.2900e-02, ..., -9.5336e-01,
-2.9333e-01, 2.8564e-01],
[-8.7343e-01, 1.1785e+00, 9.9759e-01, ..., 3.5162e-01,
4.8426e-01, 6.5852e-01],
[-1.8543e+00, -1.2079e+00, -1.5266e+00, ..., -2.1991e+00,
-1.1735e+00, -4.8984e-01]],
[[-1.7342e+00, -1.6845e+00, -3.4155e+00, ..., -9.5552e+00,
-7.2139e+00, -5.6390e+00],
[-6.3117e-01, -1.8492e-01, -3.7914e+00, ..., -1.3676e+01,
-1.1963e+01, -1.0782e+01],
[-1.9109e+00, -1.0063e+00, -5.6823e+00, ..., -1.3418e+01,
-7.7327e+00, -6.8532e+00],
...,
[-1.4999e+00, -2.3322e-01, -3.8482e+00, ..., -2.6906e+00,
-2.1595e+00, -1.6341e+00],
[-1.6523e+00, -6.3756e-01, -3.2431e+00, ..., -2.4156e+00,
-1.7589e+00, -1.0882e+00],
[-2.1588e+00, -1.3264e+00, -2.9608e+00, ..., -2.9345e+00,
-2.0877e+00, -1.7642e+00]],
[[-1.4630e+00, -2.0861e+00, -1.8188e+00, ..., -8.6920e+00,
-7.8495e+00, -4.8963e+00],
[-1.5334e+00, -2.0651e+00, -1.5446e+00, ..., -9.8410e+00,
-8.6893e+00, -5.1955e+00],
[-1.8100e+00, -2.2169e+00, -1.7915e+00, ..., -1.1674e+01,
-9.1704e+00, -5.2714e+00],
...,
[-2.3559e+00, -2.9389e+00, -2.7007e+00, ..., -3.6268e+00,
-2.8595e+00, -2.1531e+00],
[-1.5608e+00, -2.0296e+00, -1.9549e+00, ..., -2.5490e+00,
-2.0758e+00, -1.6499e+00],
[-1.0425e+00, -1.3224e+00, -1.3640e+00, ..., -1.6248e+00,
-1.3552e+00, -1.0007e+00]]],
[[[ 2.4705e-01, 5.4157e-01, 6.6825e-01, ..., 5.9212e-01,
9.4743e-01, 7.8092e-01],
[-2.0856e+00, -1.3308e+00, -2.9435e+00, ..., -1.3173e+00,
4.9051e-02, 6.2281e-01],
[-6.1321e-02, -8.5850e-01, -2.5959e-03, ..., 4.7936e-01,
9.0368e-01, 9.3437e-01],
...,
[-9.4936e-01, -2.0865e+00, -1.9779e+00, ..., -1.4009e+00,
-8.8938e-01, -6.4678e-01],
[ 7.7991e-01, -1.0009e-01, -9.3169e-02, ..., 4.0838e-01,
7.7950e-01, 8.7627e-01],
[ 2.0405e-01, -5.5837e-01, -3.8045e-01, ..., 6.1396e-01,
8.8930e-01, 7.7811e-01]],
[[-1.7741e+00, -1.3004e+00, -4.2615e+00, ..., -5.8282e+00,
-4.9234e+00, -3.8594e+00],
[ 6.2954e-01, 1.8453e+00, -2.2868e+00, ..., -4.7351e+00,
-4.4745e+00, -3.6943e+00],
[ 1.4627e+00, 1.5949e+00, -4.9686e+00, ..., -6.8666e+00,
-6.4270e+00, -5.8558e+00],
...,
[ 2.9151e-01, 2.4250e-01, -4.2310e+00, ..., -6.0837e+00,
-5.2120e+00, -4.2797e+00],
[ 2.3550e+00, 2.2855e+00, -1.6296e-01, ..., -1.3859e+00,
-1.4624e+00, -1.5402e+00],
[ 1.3798e+00, 1.1852e+00, -1.7537e+00, ..., -2.7690e+00,
-2.8469e+00, -2.7959e+00]],
[[-2.4384e-01, -1.3716e+00, -1.5797e+00, ..., -7.9977e-01,
-5.4250e-01, -6.7226e-01],
[-7.2314e-01, -1.0324e+00, -1.0089e+00, ..., -6.7933e-01,
-3.6807e-01, -2.3933e-01],
[-5.2455e-01, -3.2973e-01, -4.4808e-01, ..., -3.9286e-01,
-1.8763e-02, 8.0414e-02],
...,
[-4.7616e-02, -5.6698e-01, -6.0062e-01, ..., -7.0963e-01,
-4.8461e-01, -5.9131e-01],
[-6.5823e-01, -2.0784e-01, -5.6500e-01, ..., -3.1087e-01,
-1.5670e-02, 6.6358e-02],
[-8.1961e-01, -2.4697e-01, -4.7169e-01, ..., -3.1777e-01,
-2.0934e-01, -1.5104e-01]],
...,
[[-3.0398e+00, -2.4976e+00, -2.6312e+00, ..., -3.1802e+00,
-2.5739e+00, -1.7361e+00],
[-2.0896e+00, -6.5039e-01, -1.0955e+00, ..., -1.7223e+00,
-1.7764e+00, -1.4738e+00],
[-1.0483e+00, 1.0906e+00, 8.8615e-01, ..., -2.7344e-01,
2.3318e-02, 1.7972e-01],
...,
[-2.1527e+00, -1.5531e-01, -9.0523e-02, ..., -1.0395e+00,
-3.7605e-01, 2.0404e-01],
[-1.1809e+00, 8.6053e-01, 1.0504e+00, ..., 2.3924e-01,
3.4265e-01, 4.7980e-01],
[-2.0730e+00, -1.3708e+00, -1.5428e+00, ..., -2.2108e+00,
-1.2616e+00, -6.1734e-01]],
[[-1.6106e+00, -1.5771e+00, -2.8841e+00, ..., -1.3602e+00,
-7.6310e-01, -5.4535e-01],
[-5.6369e-01, 1.1986e-01, -3.2366e+00, ..., -2.0269e+00,
-1.6661e+00, -1.7282e+00],
[-2.0399e+00, -8.4415e-01, -5.3540e+00, ..., -4.0197e+00,
-3.3910e+00, -2.5757e+00],
...,
[-1.3961e+00, 6.2270e-02, -3.3891e+00, ..., -2.3640e+00,
-1.8109e+00, -1.3486e+00],
[-1.6806e+00, -5.4107e-01, -3.0174e+00, ..., -2.2482e+00,
-1.5111e+00, -8.8145e-01],
[-2.2821e+00, -1.3821e+00, -3.0039e+00, ..., -2.8815e+00,
-1.9381e+00, -1.6301e+00]],
[[-1.3527e+00, -1.9115e+00, -1.6819e+00, ..., -1.8949e+00,
-1.5355e+00, -1.1188e+00],
[-1.2742e+00, -1.7667e+00, -1.1946e+00, ..., -1.6505e+00,
-1.3122e+00, -9.9807e-01],
[-1.5982e+00, -1.9357e+00, -1.4310e+00, ..., -1.9330e+00,
-1.4521e+00, -1.0280e+00],
...,
[-2.4049e+00, -2.9962e+00, -2.7447e+00, ..., -3.2037e+00,
-2.5226e+00, -1.8722e+00],
[-1.5708e+00, -2.0597e+00, -1.9760e+00, ..., -2.2516e+00,
-1.8293e+00, -1.4341e+00],
[-1.0341e+00, -1.3418e+00, -1.3685e+00, ..., -1.3895e+00,
-1.1717e+00, -8.4829e-01]]],
[[[ 2.9348e-01, 5.8033e-01, 6.7340e-01, ..., -5.4477e-01,
-2.7615e-01, -1.1462e-01],
[-1.9741e+00, -1.2156e+00, -2.7968e+00, ..., -3.6159e+00,
-1.5323e+00, -6.1017e-01],
[ 6.9571e-02, -7.1295e-01, 4.2146e-02, ..., -3.2654e-01,
-5.5970e-01, 6.4060e-02],
...,
[-8.8321e-01, -1.9595e+00, -1.8856e+00, ..., -1.9907e+00,
-1.9594e+00, -1.2836e+00],
[ 8.5202e-01, 3.9169e-02, 5.9022e-02, ..., -1.2596e-01,
1.3167e-01, 3.6950e-01],
[ 2.9494e-01, -4.2536e-01, -2.3734e-01, ..., 7.0171e-02,
3.5564e-01, 4.4604e-01]],
[[-1.8395e+00, -1.2294e+00, -4.2458e+00, ..., -7.1008e+00,
-6.3107e+00, -4.6578e+00],
[ 5.5531e-01, 1.9424e+00, -2.3053e+00, ..., -5.6557e+00,
-5.7988e+00, -4.4307e+00],
[ 1.4383e+00, 1.7303e+00, -5.0162e+00, ..., -8.5352e+00,
-8.1107e+00, -6.7726e+00],
...,
[ 1.8713e-01, 2.2500e-01, -4.3909e+00, ..., -7.4441e+00,
-7.0329e+00, -5.4139e+00],
[ 2.2206e+00, 2.2034e+00, -3.2619e-01, ..., -2.1137e+00,
-2.7832e+00, -2.1655e+00],
[ 1.3054e+00, 1.1499e+00, -1.8261e+00, ..., -3.0185e+00,
-3.6140e+00, -3.1055e+00]],
[[-2.8331e-01, -1.4488e+00, -1.5940e+00, ..., -1.5287e+00,
-1.1465e+00, -1.4346e+00],
[-8.2153e-01, -1.1835e+00, -1.0505e+00, ..., -1.0673e+00,
-6.7900e-01, -9.9639e-01],
[-6.2721e-01, -4.8566e-01, -4.7627e-01, ..., -1.2321e+00,
-6.6143e-01, -5.2004e-01],
...,
[-5.8548e-02, -5.9097e-01, -6.3308e-01, ..., -1.0521e+00,
-8.8679e-01, -1.2916e+00],
[-6.7728e-01, -2.5346e-01, -5.3785e-01, ..., -5.6536e-01,
-2.4927e-01, -4.2828e-01],
[-8.2459e-01, -2.6337e-01, -4.1849e-01, ..., -6.1990e-01,
-5.8403e-01, -7.4889e-01]],
...,
[[-3.0115e+00, -2.3350e+00, -2.5860e+00, ..., -3.0743e+00,
-2.5109e+00, -1.8416e+00],
[-1.9461e+00, -3.5330e-01, -9.6749e-01, ..., -2.5821e+00,
-2.6714e+00, -2.1271e+00],
[-9.6416e-01, 1.3438e+00, 9.5368e-01, ..., -4.7457e-01,
-2.3714e-02, 2.5580e-01],
...,
[-2.2159e+00, -1.6599e-01, -1.3021e-01, ..., -9.2888e-01,
-3.0023e-01, 2.5621e-01],
[-1.1521e+00, 8.1702e-01, 9.3835e-01, ..., 5.5923e-01,
5.4222e-01, 8.6488e-01],
[-2.0308e+00, -1.3643e+00, -1.5787e+00, ..., -2.1122e+00,
-1.0809e+00, -2.7849e-01]],
[[-1.5613e+00, -1.5124e+00, -2.9040e+00, ..., -2.3966e+00,
-2.0880e+00, -1.9829e+00],
[-5.3847e-01, 1.4483e-01, -3.3613e+00, ..., -3.0084e+00,
-3.0394e+00, -3.0519e+00],
[-2.0214e+00, -8.4464e-01, -5.5397e+00, ..., -4.2668e+00,
-4.6583e+00, -3.8606e+00],
...,
[-1.3859e+00, -1.2304e-01, -3.5356e+00, ..., -2.8846e+00,
-2.8320e+00, -2.3893e+00],
[-1.6673e+00, -6.6719e-01, -3.1305e+00, ..., -2.5031e+00,
-2.3669e+00, -1.7504e+00],
[-2.2210e+00, -1.4122e+00, -3.0412e+00, ..., -2.9144e+00,
-2.5252e+00, -2.2764e+00]],
[[-1.3363e+00, -1.9006e+00, -1.6505e+00, ..., -2.7200e+00,
-2.2712e+00, -1.7699e+00],
[-1.2663e+00, -1.7625e+00, -1.1826e+00, ..., -2.5627e+00,
-2.0594e+00, -1.6936e+00],
[-1.6057e+00, -1.9350e+00, -1.4151e+00, ..., -3.0771e+00,
-2.4090e+00, -1.8805e+00],
...,
[-2.3908e+00, -2.9564e+00, -2.7298e+00, ..., -4.1830e+00,
-3.3933e+00, -2.6622e+00],
[-1.5707e+00, -2.0351e+00, -1.9630e+00, ..., -2.8811e+00,
-2.4082e+00, -1.9947e+00],
[-1.0419e+00, -1.3280e+00, -1.3591e+00, ..., -1.9158e+00,
-1.5817e+00, -1.2287e+00]]],
[[[ 4.2092e-01, 8.5519e-01, 4.5274e-01, ..., 4.7854e-01,
1.0148e+00, 6.5823e-01],
[-1.3411e+00, -4.7542e-01, -2.0941e+00, ..., -2.4678e+00,
-7.0122e-01, -2.7540e-02],
[ 8.7063e-01, 6.2178e-01, 2.8974e-01, ..., -2.0531e-01,
5.1683e-01, 6.5498e-01],
...,
[ 5.3573e-02, -5.6734e-01, -1.3430e+00, ..., -1.9473e+00,
-1.3712e+00, -9.1783e-01],
[ 1.3268e+00, 1.1575e+00, 5.7328e-01, ..., -2.8243e-01,
5.2491e-01, 7.1191e-01],
[ 9.8881e-01, 9.0742e-01, 6.0406e-01, ..., -3.0409e-01,
3.2545e-01, 3.0596e-01]],
[[-2.8357e+00, -1.5681e+00, -4.8639e+00, ..., -4.8708e+00,
-4.4257e+00, -3.3646e+00],
[-5.3543e-01, 1.6481e+00, -2.9265e+00, ..., -3.4314e+00,
-3.5516e+00, -2.9004e+00],
[ 2.7165e-01, 1.4530e+00, -5.7125e+00, ..., -5.4071e+00,
-5.2680e+00, -4.9286e+00],
...,
[-9.5010e-01, -2.6465e-02, -4.9608e+00, ..., -4.5977e+00,
-4.1367e+00, -3.4525e+00],
[ 1.6281e+00, 2.2019e+00, -4.8121e-01, ..., -8.6444e-01,
-1.1870e+00, -1.2317e+00],
[ 9.4888e-01, 1.1941e+00, -1.9033e+00, ..., -2.2276e+00,
-2.4579e+00, -2.3571e+00]],
[[-7.0868e-01, -1.9144e+00, -1.7753e+00, ..., -6.1196e-01,
-2.1414e-01, -4.7689e-01],
[-1.4745e+00, -1.9498e+00, -1.2830e+00, ..., -3.6272e-01,
5.5664e-02, 2.6000e-02],
[-1.4012e+00, -1.3872e+00, -7.6754e-01, ..., -2.7490e-01,
2.5130e-01, 2.8237e-01],
...,
[-9.0103e-01, -1.4075e+00, -7.0457e-01, ..., -4.9523e-01,
-1.7469e-01, -3.5147e-01],
[-1.1894e+00, -8.8288e-01, -6.0513e-01, ..., -3.0916e-01,
1.5454e-01, 2.1883e-01],
[-1.2244e+00, -7.0601e-01, -3.1366e-01, ..., -4.8176e-01,
-2.6710e-01, -2.4137e-01]],
...,
[[-2.2659e+00, -1.1458e+00, -2.5567e+00, ..., -3.6427e+00,
-2.8241e+00, -1.8272e+00],
[-6.2918e-01, 1.2400e+00, -7.8238e-01, ..., -2.6035e+00,
-2.3815e+00, -1.8308e+00],
[-1.7306e-01, 2.6136e+00, 8.1590e-01, ..., -5.8714e-01,
-7.9067e-02, 1.6667e-01],
...,
[-1.3064e+00, 1.3487e+00, 7.5417e-02, ..., -1.5285e+00,
-6.8906e-01, -1.2905e-01],
[-4.9142e-01, 1.4713e+00, 8.2157e-01, ..., -1.3520e-01,
1.0810e-01, 3.2429e-01],
[-1.5932e+00, -1.0447e+00, -1.4577e+00, ..., -2.3533e+00,
-1.4480e+00, -9.0287e-01]],
[[-1.4385e+00, -1.4373e+00, -3.1142e+00, ..., -1.1225e+00,
-7.0243e-01, -7.1476e-01],
[-6.8507e-01, -2.2761e-01, -3.9518e+00, ..., -1.3646e+00,
-1.2168e+00, -1.5819e+00],
[-2.1918e+00, -1.5402e+00, -6.4591e+00, ..., -3.3007e+00,
-2.8221e+00, -2.2590e+00],
...,
[-1.8340e+00, -1.0198e+00, -4.7057e+00, ..., -1.5027e+00,
-1.0569e+00, -8.8369e-01],
[-1.7266e+00, -1.0200e+00, -3.7539e+00, ..., -1.7478e+00,
-1.1556e+00, -7.4927e-01],
[-2.0318e+00, -1.4310e+00, -3.0306e+00, ..., -2.8057e+00,
-1.9990e+00, -1.8108e+00]],
[[-1.2824e+00, -1.9417e+00, -1.7806e+00, ..., -1.7970e+00,
-1.4927e+00, -1.1372e+00],
[-1.2797e+00, -1.8541e+00, -1.4344e+00, ..., -1.4052e+00,
-1.1133e+00, -8.8524e-01],
[-1.6576e+00, -2.0487e+00, -1.6595e+00, ..., -1.5989e+00,
-1.2071e+00, -8.9223e-01],
...,
[-2.3619e+00, -2.9889e+00, -2.8850e+00, ..., -2.8579e+00,
-2.2855e+00, -1.7579e+00],
[-1.5983e+00, -2.0565e+00, -2.0724e+00, ..., -1.9702e+00,
-1.6182e+00, -1.3084e+00],
[-1.1029e+00, -1.3469e+00, -1.4637e+00, ..., -1.2189e+00,
-1.0316e+00, -7.8985e-01]]]], device='cuda:0'),
tensor([[[[ 4.6853e-01, 9.1044e-01, 6.0793e-01, ..., 7.1911e-01,
1.0317e+00, 8.0714e-01],
[-1.0756e+00, -1.5812e-01, -2.3714e-01, ..., -1.3632e+00,
3.8002e-02, 6.1809e-01],
[ 7.2920e-01, 4.8531e-01, -6.8647e-01, ..., 4.0791e-01,
8.4067e-01, 8.4917e-01],
...,
[-7.6617e-01, -1.6995e+00, -1.7465e+00, ..., -1.6577e+00,
-1.2635e+00, -8.8452e-01],
[ 8.0795e-01, 1.1431e-01, -1.9744e-02, ..., 9.1608e-02,
7.2064e-01, 8.3421e-01],
[ 2.0584e-01, -4.5222e-01, -3.6092e-01, ..., -3.6409e-02,
4.9202e-01, 4.1111e-01]],
[[-2.6675e+00, -1.4391e+00, -4.8644e+00, ..., -5.9453e+00,
-5.1328e+00, -3.9379e+00],
[-6.7248e-01, 1.3840e+00, -3.3550e+00, ..., -4.7888e+00,
-4.6199e+00, -3.7097e+00],
[ 1.0104e-02, 1.1814e+00, -4.7682e+00, ..., -6.9477e+00,
-6.6738e+00, -6.0051e+00],
...,
[ 2.1682e-01, 3.7084e-01, -4.0624e+00, ..., -4.7300e+00,
-4.2205e+00, -3.5570e+00],
[ 2.2603e+00, 2.2896e+00, -1.8474e-01, ..., -8.8685e-01,
-1.1999e+00, -1.2820e+00],
[ 1.3250e+00, 1.1780e+00, -1.7876e+00, ..., -2.2912e+00,
-2.5286e+00, -2.4350e+00]],
[[-5.5225e-01, -1.5740e+00, -1.3948e+00, ..., -7.9852e-01,
-5.4645e-01, -7.8085e-01],
[-1.3196e+00, -1.7210e+00, -8.3217e-01, ..., -7.3989e-01,
-3.8711e-01, -3.2666e-01],
[-1.1983e+00, -1.1090e+00, 4.0092e-02, ..., -4.8471e-01,
-9.4600e-02, -8.8350e-02],
...,
[-2.1067e-01, -6.7118e-01, -4.8147e-01, ..., -4.9574e-01,
-2.1319e-01, -4.5922e-01],
[-7.2981e-01, -2.3259e-01, -4.2285e-01, ..., -1.8286e-01,
2.2952e-01, 2.2356e-01],
[-8.2886e-01, -2.2584e-01, -3.2831e-01, ..., -3.5111e-01,
-1.7504e-01, -2.1940e-01]],
...,
[[-2.0663e+00, -1.1770e+00, -2.4926e+00, ..., -3.3385e+00,
-2.7938e+00, -1.8562e+00],
[-5.4073e-01, 1.1080e+00, -9.7252e-02, ..., -1.6181e+00,
-1.8470e+00, -1.5896e+00],
[-2.4313e-01, 2.1814e+00, 1.8855e+00, ..., -3.0480e-01,
-7.3993e-02, 1.3224e-01],
...,
[-1.9930e+00, 5.5912e-02, -1.1945e-01, ..., -1.3269e+00,
-3.7196e-01, 1.1269e-01],
[-1.1143e+00, 8.7150e-01, 9.2501e-01, ..., 1.4786e-02,
3.3675e-01, 4.7141e-01],
[-2.0217e+00, -1.3271e+00, -1.5832e+00, ..., -2.4291e+00,
-1.4257e+00, -8.8117e-01]],
[[-1.5004e+00, -1.4739e+00, -2.8888e+00, ..., -1.1664e+00,
-6.2309e-01, -5.2705e-01],
[-9.2237e-01, -5.4420e-01, -3.8443e+00, ..., -1.9486e+00,
-1.6903e+00, -1.9062e+00],
[-2.4765e+00, -1.9749e+00, -6.0204e+00, ..., -4.2276e+00,
-3.6919e+00, -2.9424e+00],
...,
[-1.4266e+00, -1.0228e-01, -3.4705e+00, ..., -1.5176e+00,
-1.0943e+00, -9.6769e-01],
[-1.6962e+00, -6.5296e-01, -3.0900e+00, ..., -1.7753e+00,
-1.2230e+00, -8.2854e-01],
[-2.2910e+00, -1.4802e+00, -3.0934e+00, ..., -2.9480e+00,
-2.1442e+00, -1.9541e+00]],
[[-1.2577e+00, -1.8522e+00, -1.6749e+00, ..., -1.9117e+00,
-1.5786e+00, -1.1894e+00],
[-1.3235e+00, -1.8432e+00, -1.8315e+00, ..., -1.6820e+00,
-1.3598e+00, -1.0759e+00],
[-1.7499e+00, -2.1191e+00, -2.6518e+00, ..., -1.9839e+00,
-1.5288e+00, -1.1342e+00],
...,
[-2.3753e+00, -2.9318e+00, -2.7106e+00, ..., -2.8569e+00,
-2.2754e+00, -1.7381e+00],
[-1.5507e+00, -2.0077e+00, -1.9326e+00, ..., -1.9435e+00,
-1.5934e+00, -1.2812e+00],
[-1.0166e+00, -1.2975e+00, -1.3317e+00, ..., -1.1848e+00,
-1.0114e+00, -7.6359e-01]]],
[[[ 9.1366e-02, 2.4783e-01, -1.3636e+00, ..., -6.0567e-01,
-1.9445e-01, -3.4558e-02],
[-2.1493e+00, -1.6359e+00, -4.6064e+00, ..., -3.5435e+00,
-1.4006e+00, -4.8968e-01],
[ 6.8223e-01, -4.9127e-02, -8.1900e-01, ..., -1.2212e-01,
-1.7969e-01, 3.0683e-01],
...,
[-3.3669e-01, -1.1552e+00, -1.4690e+00, ..., -1.3692e+00,
-9.0657e-01, -6.8494e-01],
[ 9.7510e-01, 5.6572e-01, 3.3566e-01, ..., 3.8177e-01,
7.0634e-01, 7.9859e-01],
[ 6.5989e-01, 4.0141e-01, 3.2186e-01, ..., 5.1219e-01,
7.6829e-01, 7.0666e-01]],
[[-3.3317e+00, -1.6355e+00, -6.2143e+00, ..., -6.9459e+00,
-6.1130e+00, -4.5244e+00],
[-6.0724e-01, 2.0018e+00, -4.1086e+00, ..., -5.4798e+00,
-5.5276e+00, -4.2702e+00],
[-2.8494e-02, 1.6084e+00, -7.3028e+00, ..., -7.8939e+00,
-7.7571e+00, -6.5944e+00],
...,
[-4.7767e-01, 2.5379e-01, -4.4477e+00, ..., -5.3244e+00,
-4.6682e+00, -3.8802e+00],
[ 1.8778e+00, 2.3732e+00, -1.7480e-01, ..., -1.0689e+00,
-1.2409e+00, -1.3519e+00],
[ 1.0915e+00, 1.2699e+00, -1.7734e+00, ..., -2.4533e+00,
-2.5834e+00, -2.5609e+00]],
[[-1.3932e+00, -2.5169e+00, -2.4488e+00, ..., -1.6598e+00,
-1.1933e+00, -1.3222e+00],
[-1.8495e+00, -2.1951e+00, -1.6405e+00, ..., -1.1715e+00,
-7.0962e-01, -8.0729e-01],
[-2.3184e+00, -2.0789e+00, -1.0533e+00, ..., -1.0631e+00,
-4.6540e-01, -1.9705e-01],
...,
[-6.4801e-01, -1.0945e+00, -5.6059e-01, ..., -6.0188e-01,
-4.7032e-01, -6.3506e-01],
[-1.0278e+00, -6.7913e-01, -5.7098e-01, ..., -2.6511e-01,
-5.3659e-02, -3.4125e-02],
[-1.1729e+00, -6.3025e-01, -3.5210e-01, ..., -3.2738e-01,
-2.6957e-01, -2.3063e-01]],
...,
[[-2.2509e+00, -1.1106e+00, -2.4534e+00, ..., -2.9605e+00,
-2.4568e+00, -1.7226e+00],
[-1.8181e+00, 2.6510e-01, -1.3136e+00, ..., -2.3878e+00,
-2.4196e+00, -1.8596e+00],
[-3.5257e-01, 2.7909e+00, 1.2843e+00, ..., 1.0682e-01,
1.9967e-01, 4.5233e-01],
...,
[-1.4655e+00, 1.0133e+00, -1.0173e-01, ..., -1.1914e+00,
-4.4567e-01, 7.3924e-02],
[-5.7111e-01, 1.3756e+00, 7.4847e-01, ..., 2.5430e-03,
1.8711e-01, 3.1467e-01],
[-1.5741e+00, -1.0212e+00, -1.4719e+00, ..., -2.3384e+00,
-1.3282e+00, -7.0655e-01]],
[[-2.4116e+00, -2.7616e+00, -4.3824e+00, ..., -2.5135e+00,
-2.0114e+00, -1.7853e+00],
[-1.6586e+00, -1.4175e+00, -5.0980e+00, ..., -3.0206e+00,
-2.8619e+00, -2.7843e+00],
[-2.8044e+00, -2.2829e+00, -7.0256e+00, ..., -4.4222e+00,
-4.3382e+00, -3.4698e+00],
...,
[-1.7509e+00, -6.0483e-01, -4.1319e+00, ..., -2.0756e+00,
-1.5620e+00, -1.1933e+00],
[-1.6674e+00, -7.2476e-01, -3.4134e+00, ..., -2.0537e+00,
-1.3431e+00, -7.9208e-01],
[-1.9819e+00, -1.2438e+00, -2.8609e+00, ..., -2.6823e+00,
-1.7881e+00, -1.5340e+00]],
[[-1.8569e+00, -2.6122e+00, -2.4649e+00, ..., -2.6441e+00,
-2.1728e+00, -1.6712e+00],
[-1.8454e+00, -2.5716e+00, -2.1051e+00, ..., -2.4083e+00,
-1.9039e+00, -1.5393e+00],
[-2.3133e+00, -2.8716e+00, -2.4632e+00, ..., -2.7717e+00,
-2.1651e+00, -1.6436e+00],
...,
[-2.2849e+00, -2.8713e+00, -2.6741e+00, ..., -2.8832e+00,
-2.3028e+00, -1.7465e+00],
[-1.5675e+00, -2.0126e+00, -1.9717e+00, ..., -2.0454e+00,
-1.6842e+00, -1.3460e+00],
[-1.0752e+00, -1.3074e+00, -1.3786e+00, ..., -1.2742e+00,
-1.0903e+00, -8.0405e-01]]],
[[[ 2.2812e-01, 4.8379e-01, 6.3268e-01, ..., -3.0498e-01,
-8.8007e-02, 1.3079e-02],
[-2.0355e+00, -1.3554e+00, -2.8637e+00, ..., -3.1427e+00,
-1.2910e+00, -4.4929e-01],
[-1.6494e-01, -9.1899e-01, 2.1541e-02, ..., -1.4340e-01,
-3.7020e-01, 1.6209e-01],
...,
[-9.9191e-01, -2.1956e+00, -2.0679e+00, ..., -1.9061e+00,
-1.8730e+00, -1.2606e+00],
[ 7.7087e-01, -1.8512e-01, -1.7871e-01, ..., 2.0001e-02,
2.5790e-01, 4.4159e-01],
[ 1.9522e-01, -6.3027e-01, -4.4670e-01, ..., 2.1543e-01,
4.5656e-01, 4.8946e-01]],
[[-1.6551e+00, -1.3553e+00, -4.2396e+00, ..., -6.8319e+00,
-6.0243e+00, -4.5103e+00],
[ 5.1463e-01, 1.5047e+00, -2.5122e+00, ..., -5.4670e+00,
-5.5469e+00, -4.3025e+00],
[ 1.2518e+00, 1.2295e+00, -5.0600e+00, ..., -7.8581e+00,
-7.7853e+00, -6.5783e+00],
...,
[ 4.1573e-01, 3.9785e-01, -4.0903e+00, ..., -7.0973e+00,
-6.6886e+00, -5.1789e+00],
[ 2.5711e+00, 2.5320e+00, 1.3205e-01, ..., -1.9766e+00,
-2.6035e+00, -2.0627e+00],
[ 1.5274e+00, 1.3375e+00, -1.6376e+00, ..., -2.9267e+00,
-3.4650e+00, -3.0148e+00]],
[[-1.3631e-01, -1.1772e+00, -1.4119e+00, ..., -1.3910e+00,
-1.0256e+00, -1.3132e+00],
[-5.1548e-01, -8.3787e-01, -8.9170e-01, ..., -9.9319e-01,
-6.2412e-01, -8.9563e-01],
[-3.4671e-01, -2.3329e-01, -4.4583e-01, ..., -1.0924e+00,
-6.0927e-01, -4.7216e-01],
...,
[-5.9657e-02, -6.0764e-01, -5.8530e-01, ..., -1.0320e+00,
-8.6932e-01, -1.2388e+00],
[-7.0156e-01, -2.1411e-01, -5.5806e-01, ..., -5.6398e-01,
-2.5196e-01, -4.1924e-01],
[-8.7533e-01, -2.7444e-01, -4.7984e-01, ..., -4.8757e-01,
-4.3349e-01, -6.0290e-01]],
...,
[[-2.8584e+00, -2.4702e+00, -2.4825e+00, ..., -3.0138e+00,
-2.4587e+00, -1.7954e+00],
[-2.1134e+00, -8.4563e-01, -1.1439e+00, ..., -2.3867e+00,
-2.4785e+00, -2.0190e+00],
[-1.1049e+00, 8.2159e-01, 7.7596e-01, ..., -9.0068e-02,
2.5885e-02, 2.6069e-01],
...,
[-2.2687e+00, -1.4049e-01, -1.3955e-01, ..., -9.0099e-01,
-2.4412e-01, 2.9092e-01],
[-1.2255e+00, 9.1817e-01, 1.0865e+00, ..., 4.2077e-01,
4.7135e-01, 8.0574e-01],
[-2.1837e+00, -1.4656e+00, -1.6536e+00, ..., -2.1924e+00,
-1.1766e+00, -3.6418e-01]],
[[-1.6297e+00, -1.6289e+00, -2.8151e+00, ..., -2.2778e+00,
-1.9208e+00, -1.8238e+00],
[-6.8047e-01, -6.9910e-02, -3.1669e+00, ..., -2.8804e+00,
-2.8612e+00, -2.8745e+00],
[-2.0499e+00, -9.4991e-01, -5.1419e+00, ..., -4.4566e+00,
-4.4926e+00, -3.6788e+00],
...,
[-1.3192e+00, 2.3519e-01, -3.3475e+00, ..., -2.8466e+00,
-2.7422e+00, -2.2999e+00],
[-1.6629e+00, -4.2966e-01, -2.9920e+00, ..., -2.4575e+00,
-2.2553e+00, -1.6523e+00],
[-2.3167e+00, -1.3327e+00, -3.0025e+00, ..., -2.8551e+00,
-2.4117e+00, -2.1768e+00]],
[[-1.3079e+00, -1.8344e+00, -1.6513e+00, ..., -2.5776e+00,
-2.1416e+00, -1.6625e+00],
[-1.2458e+00, -1.7111e+00, -1.2192e+00, ..., -2.4200e+00,
-1.9391e+00, -1.5839e+00],
[-1.5657e+00, -1.8999e+00, -1.4723e+00, ..., -2.8404e+00,
-2.2548e+00, -1.7478e+00],
...,
[-2.4585e+00, -3.0680e+00, -2.7939e+00, ..., -3.9942e+00,
-3.2409e+00, -2.5421e+00],
[-1.5904e+00, -2.0969e+00, -2.0008e+00, ..., -2.7646e+00,
-2.3128e+00, -1.9127e+00],
[-1.0399e+00, -1.3582e+00, -1.3793e+00, ..., -1.8343e+00,
-1.5234e+00, -1.1803e+00]]],
[[[ 1.0830e-02, 1.4600e-01, -1.6508e+00, ..., -7.6572e-01,
-3.5083e-01, -1.4750e-01],
[-2.2117e+00, -1.7821e+00, -4.9163e+00, ..., -4.0420e+00,
-1.7167e+00, -6.8465e-01],
[ 4.9660e-01, -2.3691e-01, -1.0576e+00, ..., -3.1015e-01,
-4.2767e-01, 1.5248e-01],
...,
[-7.4726e-01, -1.6826e+00, -1.7260e+00, ..., -1.5687e+00,
-1.1802e+00, -8.7758e-01],
[ 8.5254e-01, 1.4572e-01, 1.6759e-02, ..., 1.1289e-01,
7.5831e-01, 8.1005e-01],
[ 2.4993e-01, -4.1172e-01, -3.3817e-01, ..., -5.1818e-02,
4.6566e-01, 3.9774e-01]],
[[-3.3705e+00, -1.8253e+00, -6.4379e+00, ..., -7.3162e+00,
-6.4375e+00, -4.7389e+00],
[-8.2909e-01, 1.5735e+00, -4.5157e+00, ..., -5.6451e+00,
-5.7392e+00, -4.3876e+00],
[-4.1329e-01, 1.1050e+00, -7.6411e+00, ..., -8.1922e+00,
-8.0678e+00, -6.7875e+00],
...,
[ 1.4558e-01, 2.8579e-01, -4.1180e+00, ..., -4.7889e+00,
-4.2506e+00, -3.5238e+00],
[ 2.1855e+00, 2.2219e+00, -2.0709e-01, ..., -9.6877e-01,
-1.2366e+00, -1.2598e+00],
[ 1.2888e+00, 1.1382e+00, -1.7963e+00, ..., -2.3165e+00,
-2.5012e+00, -2.4295e+00]],
[[-1.4553e+00, -2.4567e+00, -2.4039e+00, ..., -1.8620e+00,
-1.3192e+00, -1.4431e+00],
[-1.7947e+00, -2.0696e+00, -1.5782e+00, ..., -1.3156e+00,
-7.8607e-01, -8.8447e-01],
[-2.3589e+00, -2.0897e+00, -1.0658e+00, ..., -1.2018e+00,
-5.3450e-01, -2.1541e-01],
...,
[-2.3362e-01, -6.9009e-01, -4.5803e-01, ..., -4.6652e-01,
-1.8260e-01, -4.4640e-01],
[-7.3803e-01, -2.5014e-01, -4.0279e-01, ..., -1.5522e-01,
2.7844e-01, 2.4326e-01],
[-8.1994e-01, -2.2536e-01, -2.9698e-01, ..., -3.5673e-01,
-1.6510e-01, -2.0221e-01]],
...,
[[-2.0357e+00, -1.0811e+00, -2.3538e+00, ..., -3.1532e+00,
-2.5923e+00, -1.8248e+00],
[-1.9340e+00, -1.0273e-01, -1.4816e+00, ..., -2.6037e+00,
-2.6179e+00, -2.0350e+00],
[-4.0336e-01, 2.3818e+00, 1.0955e+00, ..., 1.4791e-01,
2.6946e-01, 5.2095e-01],
...,
[-1.9670e+00, 8.0469e-02, -1.2040e-01, ..., -1.3050e+00,
-3.2718e-01, 1.4332e-01],
[-1.1491e+00, 8.7384e-01, 9.3507e-01, ..., -7.3872e-02,
2.8443e-01, 4.7248e-01],
[-2.0493e+00, -1.3681e+00, -1.5923e+00, ..., -2.4441e+00,
-1.4349e+00, -8.7826e-01]],
[[-2.5201e+00, -2.9777e+00, -4.3588e+00, ..., -2.5526e+00,
-2.0649e+00, -1.8577e+00],
[-1.8613e+00, -1.7744e+00, -5.0136e+00, ..., -3.0583e+00,
-2.9253e+00, -2.8756e+00],
[-2.8845e+00, -2.5486e+00, -6.7468e+00, ..., -4.5236e+00,
-4.4838e+00, -3.5725e+00],
...,
[-1.4817e+00, -1.7709e-01, -3.4688e+00, ..., -1.5451e+00,
-1.1233e+00, -9.6324e-01],
[-1.7269e+00, -7.0775e-01, -3.0652e+00, ..., -1.7840e+00,
-1.1892e+00, -8.3674e-01],
[-2.2969e+00, -1.5087e+00, -3.0741e+00, ..., -2.9406e+00,
-2.1002e+00, -1.9643e+00]],
[[-1.8952e+00, -2.6495e+00, -2.5640e+00, ..., -2.8093e+00,
-2.3052e+00, -1.7685e+00],
[-1.8907e+00, -2.6348e+00, -2.2636e+00, ..., -2.5275e+00,
-2.0058e+00, -1.6217e+00],
[-2.3653e+00, -2.9640e+00, -2.6574e+00, ..., -2.9013e+00,
-2.2722e+00, -1.7275e+00],
...,
[-2.3542e+00, -2.9213e+00, -2.7096e+00, ..., -2.8670e+00,
-2.3003e+00, -1.7624e+00],
[-1.5356e+00, -1.9972e+00, -1.9360e+00, ..., -1.9585e+00,
-1.6165e+00, -1.2938e+00],
[-1.0065e+00, -1.2957e+00, -1.3334e+00, ..., -1.1906e+00,
-1.0135e+00, -7.6585e-01]]],
[[[ 3.5504e-01, 8.4802e-01, 7.3733e-01, ..., 7.1184e-01,
9.5906e-01, 7.3188e-01],
[-1.2582e+00, -2.4938e-01, -1.3758e+00, ..., -1.0364e+00,
8.4080e-02, 5.6348e-01],
[ 6.6858e-01, 4.8530e-01, 5.7092e-01, ..., 6.3104e-01,
9.5596e-01, 8.7243e-01],
...,
[-2.1696e-01, -1.1817e+00, -2.6874e+00, ..., -1.5250e+00,
-1.4281e+00, -8.7184e-01],
[ 1.4107e+00, 8.0043e-01, -1.0294e+00, ..., -3.3955e-01,
7.6330e-02, 3.6525e-01],
[ 8.3873e-01, 2.6931e-01, -7.4604e-01, ..., -2.3739e-01,
1.8795e-01, 3.7339e-01]],
[[-2.3744e+00, -1.2382e+00, -4.2891e+00, ..., -5.2625e+00,
-4.5197e+00, -3.5302e+00],
[-1.3302e-01, 1.7770e+00, -2.5743e+00, ..., -4.1602e+00,
-4.0215e+00, -3.2902e+00],
[ 7.4219e-01, 1.6181e+00, -5.1838e+00, ..., -6.0338e+00,
-5.7493e+00, -5.2886e+00],
...,
[-1.3282e+00, 1.0802e-01, -6.7837e+00, ..., -7.6388e+00,
-6.9740e+00, -5.4617e+00],
[ 1.7641e+00, 2.7991e+00, -1.0809e+00, ..., -2.5762e+00,
-2.9942e+00, -2.3899e+00],
[ 1.1971e+00, 1.8124e+00, -2.2985e+00, ..., -3.2383e+00,
-3.6471e+00, -3.1406e+00]],
[[-6.8881e-01, -1.8561e+00, -1.7339e+00, ..., -9.3185e-01,
-6.5536e-01, -7.6050e-01],
[-1.4091e+00, -1.8670e+00, -1.2892e+00, ..., -7.2982e-01,
-3.8670e-01, -2.7567e-01],
[-1.0776e+00, -9.9045e-01, -4.5509e-01, ..., -3.1697e-01,
3.1164e-02, 4.4916e-02],
...,
[-1.3888e+00, -1.7668e+00, -7.6726e-01, ..., -8.9518e-01,
-7.8281e-01, -1.0337e+00],
[-1.7450e+00, -1.1838e+00, -8.4223e-01, ..., -5.6407e-01,
-2.9440e-01, -3.6263e-01],
[-1.9052e+00, -1.3148e+00, -9.2641e-01, ..., -9.7471e-01,
-8.4494e-01, -8.2440e-01]],
...,
[[-2.2606e+00, -1.2882e+00, -2.3493e+00, ..., -2.8446e+00,
-2.2335e+00, -1.4757e+00],
[-6.0157e-01, 1.0994e+00, -4.6153e-01, ..., -1.2874e+00,
-1.3646e+00, -1.2057e+00],
[-3.3660e-01, 2.1367e+00, 7.5325e-01, ..., -1.5774e-01,
2.3418e-01, 3.4754e-01],
...,
[-1.8574e+00, 1.2358e+00, 2.7354e-02, ..., -1.0562e+00,
-5.4632e-01, -1.4706e-02],
[-7.9331e-01, 1.9291e+00, 1.6743e+00, ..., 5.3502e-01,
4.4155e-01, 6.9287e-01],
[-2.2008e+00, -1.1299e+00, -9.3794e-01, ..., -1.8574e+00,
-9.6542e-01, -2.9640e-01]],
[[-1.3552e+00, -1.2618e+00, -2.7639e+00, ..., -1.1428e+00,
-6.1798e-01, -4.7773e-01],
[-6.2583e-01, -1.1572e-01, -3.5884e+00, ..., -1.8271e+00,
-1.5423e+00, -1.6777e+00],
[-2.0682e+00, -1.3509e+00, -5.9985e+00, ..., -3.7585e+00,
-3.2092e+00, -2.5288e+00],
...,
[-2.4491e+00, -1.5870e+00, -5.4639e+00, ..., -3.0659e+00,
-2.7313e+00, -2.1806e+00],
[-2.1968e+00, -1.4643e+00, -4.2670e+00, ..., -2.6796e+00,
-2.3238e+00, -1.6429e+00],
[-2.3814e+00, -1.7726e+00, -3.3088e+00, ..., -2.9768e+00,
-2.4524e+00, -2.1269e+00]],
[[-1.1745e+00, -1.7351e+00, -1.5232e+00, ..., -1.6365e+00,
-1.3600e+00, -1.0227e+00],
[-1.1833e+00, -1.6568e+00, -1.1828e+00, ..., -1.3719e+00,
-1.1163e+00, -8.8062e-01],
[-1.5383e+00, -1.8242e+00, -1.3602e+00, ..., -1.5682e+00,
-1.1946e+00, -8.7003e-01],
...,
[-3.2054e+00, -4.0048e+00, -3.7915e+00, ..., -3.9806e+00,
-3.1896e+00, -2.4633e+00],
[-2.1334e+00, -2.7539e+00, -2.7173e+00, ..., -2.7889e+00,
-2.2996e+00, -1.8716e+00],
[-1.5511e+00, -1.8833e+00, -1.9470e+00, ..., -1.8837e+00,
-1.5283e+00, -1.1704e+00]]]], device='cuda:0'),
tensor([[[[-4.3632e-02, 1.2586e-01, -1.9980e+00, ..., 6.3337e-01,
1.0492e+00, 7.3379e-01],
[-2.4088e+00, -1.9161e+00, -5.5189e+00, ..., -1.8505e+00,
-3.0176e-01, 2.8654e-01],
[ 2.7736e-01, -6.6722e-01, -1.7587e+00, ..., 1.2559e-01,
7.4363e-01, 7.8189e-01],
...,
[-2.7592e-01, -1.4070e+00, -2.9660e+00, ..., -1.5925e+00,
-1.0697e+00, -7.3578e-01],
[ 1.3136e+00, 4.9569e-01, -1.5445e+00, ..., 1.5871e-02,
6.7056e-01, 7.9760e-01],
[ 7.0364e-01, -6.1999e-02, -9.6273e-01, ..., 5.5594e-02,
5.5323e-01, 4.8868e-01]],
[[-3.4747e+00, -1.5039e+00, -6.8294e+00, ..., -5.0244e+00,
-4.4722e+00, -3.4219e+00],
[-7.4831e-01, 2.1885e+00, -4.8648e+00, ..., -3.6994e+00,
-3.7242e+00, -3.0408e+00],
[-2.8517e-01, 1.7969e+00, -8.1934e+00, ..., -5.6809e+00,
-5.4802e+00, -5.0883e+00],
...,
[-1.6549e+00, 4.2710e-02, -7.4337e+00, ..., -4.8890e+00,
-4.3127e+00, -3.5848e+00],
[ 1.2899e+00, 2.5649e+00, -1.8419e+00, ..., -1.0093e+00,
-1.2306e+00, -1.3059e+00],
[ 9.3488e-01, 1.7559e+00, -2.6233e+00, ..., -2.3291e+00,
-2.5051e+00, -2.4368e+00]],
[[-1.7494e+00, -2.6740e+00, -2.4614e+00, ..., -7.0295e-01,
-3.3970e-01, -5.3982e-01],
[-2.3151e+00, -2.4267e+00, -1.7131e+00, ..., -5.0134e-01,
-1.0922e-01, -5.5892e-02],
[-3.1620e+00, -2.5100e+00, -1.0000e+00, ..., -3.0746e-01,
1.6453e-01, 2.2223e-01],
...,
[-1.9352e+00, -2.0133e+00, -6.9230e-01, ..., -5.0140e-01,
-2.3233e-01, -3.9809e-01],
[-2.0238e+00, -1.2994e+00, -6.7228e-01, ..., -2.5910e-01,
1.2160e-01, 1.6677e-01],
[-2.0262e+00, -1.4615e+00, -8.6317e-01, ..., -3.7423e-01,
-2.1241e-01, -1.9337e-01]],
...,
[[-2.3098e+00, -1.1903e+00, -2.6566e+00, ..., -3.4738e+00,
-2.7611e+00, -1.7985e+00],
[-2.4973e+00, -2.8376e-01, -1.6895e+00, ..., -2.1638e+00,
-2.1169e+00, -1.6574e+00],
[-7.3914e-01, 2.3911e+00, 9.3530e-01, ..., -4.3115e-01,
-5.3046e-03, 2.0983e-01],
...,
[-1.9942e+00, 9.7004e-01, 1.3547e-01, ..., -1.3003e+00,
-5.8844e-01, -2.6411e-02],
[-9.1572e-01, 1.8057e+00, 1.8355e+00, ..., -4.1344e-02,
1.3936e-01, 3.5886e-01],
[-2.2313e+00, -9.9675e-01, -5.2677e-01, ..., -2.2243e+00,
-1.3381e+00, -7.6788e-01]],
[[-2.7595e+00, -3.2570e+00, -4.6130e+00, ..., -9.9710e-01,
-5.6822e-01, -5.5429e-01],
[-2.0857e+00, -2.0218e+00, -5.3065e+00, ..., -1.4461e+00,
-1.2568e+00, -1.5606e+00],
[-3.2436e+00, -2.9204e+00, -7.1828e+00, ..., -3.4734e+00,
-2.9485e+00, -2.3003e+00],
...,
[-2.8203e+00, -2.1422e+00, -5.1422e+00, ..., -1.7412e+00,
-1.2798e+00, -1.0040e+00],
[-2.4250e+00, -1.8721e+00, -4.0171e+00, ..., -1.8536e+00,
-1.2410e+00, -7.7549e-01],
[-2.4257e+00, -2.0177e+00, -3.1927e+00, ..., -2.7096e+00,
-1.9048e+00, -1.6900e+00]],
[[-2.1986e+00, -2.9856e+00, -2.8063e+00, ..., -1.7488e+00,
-1.4495e+00, -1.0951e+00],
[-2.2530e+00, -3.0774e+00, -2.5440e+00, ..., -1.4035e+00,
-1.1175e+00, -8.8082e-01],
[-2.8495e+00, -3.5514e+00, -3.0700e+00, ..., -1.6053e+00,
-1.2152e+00, -8.8514e-01],
...,
[-3.4477e+00, -4.3336e+00, -4.0892e+00, ..., -2.8413e+00,
-2.2724e+00, -1.7345e+00],
[-2.2649e+00, -2.9609e+00, -2.9122e+00, ..., -1.9805e+00,
-1.6322e+00, -1.3055e+00],
[-1.6517e+00, -2.0288e+00, -2.0530e+00, ..., -1.2316e+00,
-1.0469e+00, -7.8786e-01]]],
[[[ 5.1531e-01, 9.8734e-01, 7.6811e-01, ..., 8.4562e-01,
1.0721e+00, 8.5068e-01],
[-1.1152e+00, -1.7105e-01, -1.5877e+00, ..., -9.3028e-01,
1.9020e-01, 6.8700e-01],
[ 8.4400e-01, 5.3709e-01, 2.2450e-01, ..., 5.9467e-01,
9.1420e-01, 9.2608e-01],
...,
[-8.7146e-01, -1.8610e+00, -1.8486e+00, ..., -1.8481e+00,
-1.3596e+00, -9.6582e-01],
[ 7.5449e-01, -1.6622e-02, -1.0942e-01, ..., -6.1371e-02,
6.8972e-01, 7.8662e-01],
[ 1.4190e-01, -5.8366e-01, -4.8667e-01, ..., -1.8294e-01,
4.2846e-01, 3.3941e-01]],
[[-2.6460e+00, -1.2946e+00, -4.7281e+00, ..., -5.6724e+00,
-4.8634e+00, -3.8102e+00],
[-4.5437e-01, 1.7841e+00, -3.1004e+00, ..., -4.7303e+00,
-4.5393e+00, -3.6679e+00],
[ 3.5385e-01, 1.6428e+00, -5.9386e+00, ..., -6.8202e+00,
-6.4562e+00, -5.7859e+00],
...,
[ 3.1333e-01, 3.7699e-01, -4.0346e+00, ..., -4.6428e+00,
-4.2069e+00, -3.4779e+00],
[ 2.3090e+00, 2.2781e+00, -1.7155e-01, ..., -7.9068e-01,
-1.1622e+00, -1.1969e+00],
[ 1.3581e+00, 1.1772e+00, -1.7744e+00, ..., -2.2400e+00,
-2.5128e+00, -2.4043e+00]],
[[-6.1842e-01, -1.7443e+00, -1.5253e+00, ..., -6.9396e-01,
-4.7435e-01, -6.5801e-01],
[-1.4796e+00, -1.8909e+00, -1.1703e+00, ..., -5.9664e-01,
-3.2060e-01, -2.7980e-01],
[-1.3328e+00, -1.1987e+00, -4.9216e-01, ..., -3.7808e-01,
-6.6971e-02, -7.6007e-02],
...,
[-1.4709e-01, -5.9652e-01, -4.6395e-01, ..., -5.1002e-01,
-1.9559e-01, -4.5212e-01],
[-7.0899e-01, -1.9094e-01, -4.2444e-01, ..., -2.1389e-01,
2.5571e-01, 2.4631e-01],
[-7.9565e-01, -1.8070e-01, -3.3318e-01, ..., -4.1157e-01,
-1.9835e-01, -2.3189e-01]],
...,
[[-2.2994e+00, -1.1729e+00, -2.5776e+00, ..., -3.0326e+00,
-2.4626e+00, -1.6973e+00],
[-5.7294e-01, 1.3507e+00, -6.4704e-01, ..., -1.5639e+00,
-1.6954e+00, -1.4594e+00],
[-2.2169e-01, 2.5245e+00, 6.4298e-01, ..., -3.9944e-01,
-1.2489e-01, 4.9763e-02],
...,
[-2.0817e+00, -9.2604e-02, -1.6895e-01, ..., -1.3761e+00,
-4.1617e-01, 1.0531e-01],
[-1.2131e+00, 8.0570e-01, 9.3538e-01, ..., 2.9909e-02,
3.0523e-01, 4.7669e-01],
[-2.1080e+00, -1.4069e+00, -1.6135e+00, ..., -2.4529e+00,
-1.4804e+00, -9.4250e-01]],
[[-1.4234e+00, -1.3303e+00, -2.9329e+00, ..., -1.2197e+00,
-7.0983e-01, -5.4716e-01],
[-7.4955e-01, -2.3503e-01, -3.9231e+00, ..., -2.0455e+00,
-1.7171e+00, -1.8141e+00],
[-2.4005e+00, -1.7197e+00, -6.6251e+00, ..., -4.1925e+00,
-3.5855e+00, -2.7934e+00],
...,
[-1.4011e+00, -3.9579e-02, -3.3905e+00, ..., -1.4005e+00,
-1.0089e+00, -9.1679e-01],
[-1.7088e+00, -6.4153e-01, -3.0327e+00, ..., -1.7039e+00,
-1.1497e+00, -8.0192e-01],
[-2.3244e+00, -1.4818e+00, -3.1025e+00, ..., -2.9619e+00,
-2.1448e+00, -1.9861e+00]],
[[-1.2651e+00, -1.8715e+00, -1.6275e+00, ..., -1.7735e+00,
-1.4822e+00, -1.0963e+00],
[-1.3059e+00, -1.8357e+00, -1.3388e+00, ..., -1.5998e+00,
-1.3090e+00, -1.0180e+00],
[-1.7317e+00, -2.0845e+00, -1.6093e+00, ..., -1.9209e+00,
-1.4811e+00, -1.0824e+00],
...,
[-2.3968e+00, -2.9633e+00, -2.7281e+00, ..., -2.8661e+00,
-2.2874e+00, -1.7563e+00],
[-1.5541e+00, -2.0223e+00, -1.9456e+00, ..., -1.9434e+00,
-1.5978e+00, -1.2901e+00],
[-1.0118e+00, -1.3112e+00, -1.3397e+00, ..., -1.1852e+00,
-1.0097e+00, -7.6654e-01]]],
[[[ 3.7889e-01, 6.9906e-01, 7.7595e-01, ..., 5.7796e-01,
1.0315e+00, 6.3576e-01],
[-1.8425e+00, -1.1305e+00, -2.6456e+00, ..., -2.1803e+00,
-5.7116e-01, 1.2873e-02],
[ 1.2952e-01, -5.5655e-01, 4.1011e-02, ..., 1.0581e-01,
6.9869e-01, 7.1335e-01],
...,
[-2.0089e-01, -1.2818e+00, -2.9664e+00, ..., -1.5942e+00,
-1.5205e+00, -8.0060e-01],
[ 1.4671e+00, 6.6150e-01, -1.4835e+00, ..., -5.5133e-01,
-1.2172e-01, 2.3855e-01],
[ 7.5639e-01, -3.5900e-02, -1.1864e+00, ..., -4.9409e-01,
2.0416e-03, 2.5821e-01]],
[[-1.8537e+00, -1.1715e+00, -4.1982e+00, ..., -4.9737e+00,
-4.4575e+00, -3.4103e+00],
[ 3.4039e-01, 1.8510e+00, -2.3528e+00, ..., -3.7120e+00,
-3.7354e+00, -3.0260e+00],
[ 1.0676e+00, 1.6377e+00, -4.9843e+00, ..., -5.6265e+00,
-5.4170e+00, -5.0024e+00],
...,
[-1.5528e+00, 7.1118e-03, -7.2796e+00, ..., -8.1932e+00,
-7.5748e+00, -5.8878e+00],
[ 1.6360e+00, 2.7921e+00, -1.4001e+00, ..., -2.9286e+00,
-3.4044e+00, -2.6411e+00],
[ 1.1683e+00, 1.8945e+00, -2.4571e+00, ..., -3.4334e+00,
-3.9021e+00, -3.2732e+00]],
[[-2.3553e-01, -1.2684e+00, -1.3573e+00, ..., -5.3381e-01,
-2.3353e-01, -5.2331e-01],
[-8.1531e-01, -1.1079e+00, -8.9865e-01, ..., -3.7863e-01,
-6.9317e-02, -1.2630e-01],
[-7.2572e-01, -5.4191e-01, -4.2683e-01, ..., -3.5126e-01,
5.2623e-02, 5.5908e-02],
...,
[-1.5972e+00, -1.8431e+00, -6.4944e-01, ..., -8.6309e-01,
-7.8734e-01, -1.0961e+00],
[-1.8953e+00, -1.2127e+00, -7.2796e-01, ..., -5.3162e-01,
-2.8181e-01, -3.8205e-01],
[-2.0820e+00, -1.5306e+00, -1.0972e+00, ..., -1.2082e+00,
-1.0775e+00, -1.0698e+00]],
...,
[[-2.8124e+00, -2.1445e+00, -2.7158e+00, ..., -3.3084e+00,
-2.5421e+00, -1.7012e+00],
[-1.8388e+00, -2.7210e-01, -1.3192e+00, ..., -2.2775e+00,
-2.1108e+00, -1.6875e+00],
[-8.1309e-01, 1.5589e+00, 6.3905e-01, ..., -3.9775e-01,
1.5856e-02, 1.5031e-01],
...,
[-2.0471e+00, 1.1143e+00, 1.9625e-02, ..., -1.1670e+00,
-6.0956e-01, -4.0642e-02],
[-8.7385e-01, 1.9800e+00, 1.8579e+00, ..., 6.9259e-01,
5.1370e-01, 7.7829e-01],
[-2.3099e+00, -1.1089e+00, -7.0737e-01, ..., -1.6635e+00,
-7.9414e-01, -1.4488e-01]],
[[-1.5101e+00, -1.4662e+00, -2.8059e+00, ..., -1.1997e+00,
-7.2513e-01, -6.9370e-01],
[-5.9053e-01, 6.1466e-02, -3.2153e+00, ..., -1.4422e+00,
-1.2255e+00, -1.5193e+00],
[-2.0843e+00, -1.0209e+00, -5.4283e+00, ..., -3.3250e+00,
-2.7855e+00, -2.2233e+00],
...,
[-2.6754e+00, -1.8304e+00, -5.5421e+00, ..., -3.1813e+00,
-2.9607e+00, -2.3702e+00],
[-2.3485e+00, -1.6581e+00, -4.2837e+00, ..., -2.7864e+00,
-2.5411e+00, -1.8330e+00],
[-2.4570e+00, -1.8989e+00, -3.2755e+00, ..., -3.0403e+00,
-2.6202e+00, -2.2726e+00]],
[[-1.2963e+00, -1.8516e+00, -1.6551e+00, ..., -1.7344e+00,
-1.4417e+00, -1.0956e+00],
[-1.2377e+00, -1.7183e+00, -1.2120e+00, ..., -1.3763e+00,
-1.0991e+00, -8.7198e-01],
[-1.5733e+00, -1.9105e+00, -1.4630e+00, ..., -1.6127e+00,
-1.2217e+00, -8.9275e-01],
...,
[-3.4050e+00, -4.2853e+00, -4.0732e+00, ..., -4.2972e+00,
-3.4486e+00, -2.6678e+00],
[-2.2510e+00, -2.9488e+00, -2.9292e+00, ..., -2.9955e+00,
-2.4741e+00, -2.0186e+00],
[-1.6540e+00, -2.0450e+00, -2.1172e+00, ..., -2.0452e+00,
-1.6453e+00, -1.2648e+00]]],
[[[-5.5277e-01, -5.6701e-01, -5.9307e-01, ..., -2.3222e+01,
-1.9343e+01, -1.1648e+01],
[-2.5053e+00, -1.7153e+00, -3.5604e+00, ..., -2.1864e+01,
-1.8734e+01, -1.4962e+01],
[-1.3600e+00, -1.9186e+00, -1.8204e+00, ..., -1.9922e+01,
-1.4456e+01, -1.5837e+01],
...,
[-2.3406e-01, -1.2566e+00, -2.9024e+00, ..., -1.6626e+01,
-2.1530e+01, -2.1758e+01],
[ 1.4354e+00, 7.0102e-01, -1.3950e+00, ..., -1.1354e+01,
-1.6666e+01, -1.6928e+01],
[ 7.1983e-01, -7.5464e-03, -1.1086e+00, ..., -6.0859e+00,
-5.0039e+00, -2.5275e+00]],
[[-1.7221e+00, -1.4209e+00, -4.2542e+00, ..., -2.2319e+01,
-1.3813e+01, -7.7996e+00],
[ 3.7400e-01, 1.4874e+00, -2.8011e+00, ..., -2.4563e+01,
-1.2881e+01, -4.6394e+00],
[ 9.4913e-01, 1.3384e+00, -5.1191e+00, ..., -2.5998e+01,
-1.2554e+01, -3.7020e+00],
...,
[-1.5942e+00, -8.2706e-02, -7.4281e+00, ..., -1.6293e+01,
-1.0814e+01, -8.8880e+00],
[ 1.5763e+00, 2.6976e+00, -1.5069e+00, ..., -1.1261e+01,
-4.5208e+00, -3.8726e+00],
[ 1.1315e+00, 1.8415e+00, -2.4752e+00, ..., -6.9907e+00,
-4.8246e+00, -4.2941e+00]],
[[-2.7750e-02, -9.2799e-01, -1.0449e+00, ..., -1.4245e+00,
-1.6417e+00, -1.6823e+00],
[-6.5148e-01, -6.9046e-01, -4.0289e-01, ..., -2.4071e+00,
2.0307e-01, -6.7310e-01],
[-7.6126e-01, -5.7176e-01, -5.4643e-01, ..., -6.1330e+00,
-1.2206e+00, -1.3271e+00],
...,
[-1.5808e+00, -1.8268e+00, -6.8339e-01, ..., -7.5612e+00,
-6.4291e-01, 8.0697e-01],
[-1.8878e+00, -1.2238e+00, -7.5397e-01, ..., -5.4772e+00,
-7.3564e-01, 1.1897e+00],
[-2.0724e+00, -1.5252e+00, -1.0830e+00, ..., -3.2506e+00,
-1.0254e+00, -1.5200e+00]],
...,
[[-2.5566e+00, -2.8846e+00, -1.9684e+00, ..., -3.6009e+00,
-1.1392e+00, -1.6487e-01],
[-2.1821e+00, -1.8659e+00, -2.6866e+00, ..., -1.0338e+01,
-5.0767e+00, -8.9252e-01],
[-4.1119e-01, -2.7008e-02, -1.8612e+00, ..., -1.4650e+01,
-8.9170e+00, -4.7755e-01],
...,
[-2.0347e+00, 1.0364e+00, -7.4883e-02, ..., -1.4385e+00,
-5.6238e+00, -8.3754e-01],
[-8.6494e-01, 1.9552e+00, 1.8334e+00, ..., -4.5944e+00,
-3.2147e+00, 7.8179e-01],
[-2.2928e+00, -1.1181e+00, -7.6015e-01, ..., -3.0077e+00,
-1.0460e+00, 2.4887e-01]],
[[-1.3497e+00, -1.8065e+00, -3.1568e+00, ..., -6.4385e+00,
-6.2606e+00, -3.6481e+00],
[-2.9320e-01, 7.6229e-02, -3.3217e+00, ..., -1.3628e+01,
-1.0053e+01, -8.7707e+00],
[-1.6985e+00, -5.2976e-01, -4.3563e+00, ..., -1.6774e+01,
-1.1948e+01, -6.0472e+00],
...,
[-2.6757e+00, -1.8976e+00, -5.5869e+00, ..., -1.5877e+01,
-1.0840e+01, -5.1701e+00],
[-2.3435e+00, -1.7106e+00, -4.3561e+00, ..., -2.9265e+00,
-1.8486e+00, -6.8258e-01],
[-2.4460e+00, -1.9416e+00, -3.3804e+00, ..., -1.3280e+00,
-2.0416e-01, -3.7351e-01]],
[[-1.4719e+00, -1.9883e+00, -1.9346e+00, ..., -5.6025e+00,
-3.8942e+00, -2.1457e+00],
[-1.2940e+00, -1.8520e+00, -1.5049e+00, ..., -6.8146e+00,
-5.3040e+00, -3.2278e+00],
[-1.7864e+00, -2.1478e+00, -1.8204e+00, ..., -5.5993e+00,
-4.0973e+00, -2.8441e+00],
...,
[-3.4158e+00, -4.2909e+00, -4.0768e+00, ..., -8.8560e+00,
-6.1547e+00, -4.0580e+00],
[-2.2648e+00, -2.9491e+00, -2.9169e+00, ..., -5.9690e+00,
-4.0536e+00, -2.8503e+00],
[-1.6665e+00, -2.0409e+00, -2.0900e+00, ..., -3.7979e+00,
-2.8420e+00, -1.8048e+00]]],
[[[ 3.8823e-01, 7.9056e-01, 2.9688e-01, ..., 5.0774e-01,
1.0333e+00, 6.7912e-01],
[-1.4663e+00, -6.2676e-01, -2.3987e+00, ..., -2.3298e+00,
-5.5542e-01, 7.2722e-02],
[ 8.7260e-01, 5.2984e-01, 1.6707e-01, ..., -1.2562e-01,
6.1445e-01, 6.9867e-01],
...,
[ 2.8836e-02, -6.6115e-01, -1.4777e+00, ..., -1.7737e+00,
-1.2165e+00, -8.4655e-01],
[ 1.3309e+00, 1.0916e+00, 4.5900e-01, ..., -1.5133e-01,
5.7731e-01, 7.2733e-01],
[ 9.6887e-01, 8.1880e-01, 5.0683e-01, ..., -1.6803e-01,
3.8548e-01, 3.4910e-01]],
[[-2.9244e+00, -1.5850e+00, -5.0543e+00, ..., -4.8859e+00,
-4.4526e+00, -3.3760e+00],
[-5.4323e-01, 1.7200e+00, -3.0849e+00, ..., -3.4770e+00,
-3.6163e+00, -2.9314e+00],
[ 2.5396e-01, 1.4788e+00, -5.9838e+00, ..., -5.4651e+00,
-5.3196e+00, -4.9620e+00],
...,
[-9.9017e-01, 2.1178e-02, -5.1448e+00, ..., -4.7451e+00,
-4.2210e+00, -3.5065e+00],
[ 1.6588e+00, 2.2970e+00, -4.9592e-01, ..., -9.9210e-01,
-1.2436e+00, -1.2814e+00],
[ 9.8442e-01, 1.2666e+00, -1.9353e+00, ..., -2.2884e+00,
-2.4666e+00, -2.3773e+00]],
[[-7.3865e-01, -1.9546e+00, -1.8308e+00, ..., -6.2776e-01,
-2.4100e-01, -4.8565e-01],
[-1.4875e+00, -1.9473e+00, -1.2956e+00, ..., -4.0599e-01,
1.0895e-02, 5.2504e-03],
[-1.4934e+00, -1.4646e+00, -8.2242e-01, ..., -2.7357e-01,
2.3200e-01, 2.6361e-01],
...,
[-9.2981e-01, -1.4765e+00, -7.4572e-01, ..., -4.8295e-01,
-1.9679e-01, -3.6555e-01],
[-1.2328e+00, -9.1516e-01, -6.4506e-01, ..., -2.8200e-01,
1.4160e-01, 1.9912e-01],
[-1.2901e+00, -7.5825e-01, -3.4332e-01, ..., -4.3326e-01,
-2.3840e-01, -2.2049e-01]],
...,
[[-2.3254e+00, -1.2159e+00, -2.6291e+00, ..., -3.5796e+00,
-2.8321e+00, -1.8267e+00],
[-7.6901e-01, 1.0983e+00, -9.0573e-01, ..., -2.4252e+00,
-2.3217e+00, -1.7982e+00],
[-2.3098e-01, 2.6173e+00, 8.2146e-01, ..., -5.0866e-01,
-6.4481e-02, 1.6621e-01],
...,
[-1.3601e+00, 1.4066e+00, 1.1866e-01, ..., -1.4148e+00,
-6.1773e-01, -8.7624e-02],
[-5.6978e-01, 1.5101e+00, 9.1789e-01, ..., -1.1812e-01,
1.2730e-01, 3.3817e-01],
[-1.6790e+00, -1.0803e+00, -1.4379e+00, ..., -2.2966e+00,
-1.3916e+00, -8.5799e-01]],
[[-1.5278e+00, -1.5783e+00, -3.2992e+00, ..., -1.0620e+00,
-6.4382e-01, -6.6821e-01],
[-7.6680e-01, -3.4520e-01, -4.1551e+00, ..., -1.3840e+00,
-1.2201e+00, -1.5752e+00],
[-2.2392e+00, -1.6270e+00, -6.6252e+00, ..., -3.3417e+00,
-2.8440e+00, -2.2556e+00],
...,
[-1.8534e+00, -1.0632e+00, -4.8297e+00, ..., -1.6456e+00,
-1.1546e+00, -9.2831e-01],
[-1.7073e+00, -1.0230e+00, -3.8235e+00, ..., -1.8155e+00,
-1.1883e+00, -7.6202e-01],
[-2.0072e+00, -1.4206e+00, -3.0555e+00, ..., -2.7612e+00,
-1.9643e+00, -1.7758e+00]],
[[-1.3583e+00, -2.0430e+00, -1.8797e+00, ..., -1.7784e+00,
-1.4792e+00, -1.1302e+00],
[-1.3614e+00, -1.9630e+00, -1.5339e+00, ..., -1.3948e+00,
-1.1104e+00, -8.9014e-01],
[-1.7557e+00, -2.1695e+00, -1.7820e+00, ..., -1.6015e+00,
-1.2088e+00, -8.9384e-01],
...,
[-2.4690e+00, -3.1290e+00, -3.0287e+00, ..., -2.8480e+00,
-2.2825e+00, -1.7467e+00],
[-1.6745e+00, -2.1580e+00, -2.1702e+00, ..., -1.9800e+00,
-1.6329e+00, -1.3072e+00],
[-1.1649e+00, -1.4261e+00, -1.5316e+00, ..., -1.2292e+00,
-1.0411e+00, -7.8968e-01]]]], device='cuda:0'),
tensor([[[[ 2.1486e-01, 5.1360e-01, 6.6432e-01, ..., -4.3003e-01,
-1.2802e-01, -3.0426e-02],
[-1.9290e+00, -1.1946e+00, -2.6446e+00, ..., -3.5784e+00,
-1.4713e+00, -5.2436e-01],
[-1.1791e-01, -8.1923e-01, 5.8937e-02, ..., -3.0376e-01,
-4.3679e-01, 1.2891e-01],
...,
[-9.1020e-01, -2.0521e+00, -2.0152e+00, ..., -1.7026e+00,
-1.7179e+00, -1.0906e+00],
[ 8.1738e-01, -5.1256e-02, -9.2029e-02, ..., 4.0244e-02,
2.2168e-01, 4.1767e-01],
[ 2.3766e-01, -4.9307e-01, -3.5666e-01, ..., 1.8422e-01,
3.8468e-01, 4.6519e-01]],
[[-1.6411e+00, -1.3583e+00, -4.2008e+00, ..., -7.1026e+00,
-6.3214e+00, -4.6768e+00],
[ 4.1796e-01, 1.4040e+00, -2.5567e+00, ..., -5.4948e+00,
-5.6619e+00, -4.3251e+00],
[ 1.1217e+00, 1.1309e+00, -5.0494e+00, ..., -8.0107e+00,
-8.0172e+00, -6.7541e+00],
...,
[ 3.3787e-01, 3.5227e-01, -4.1010e+00, ..., -7.2894e+00,
-6.8484e+00, -5.3231e+00],
[ 2.5091e+00, 2.4979e+00, 1.2304e-01, ..., -2.4319e+00,
-2.9760e+00, -2.3404e+00],
[ 1.4774e+00, 1.3283e+00, -1.6225e+00, ..., -3.0759e+00,
-3.5641e+00, -3.0664e+00]],
[[-1.3920e-01, -1.1665e+00, -1.3930e+00, ..., -1.5364e+00,
-1.1461e+00, -1.4493e+00],
[-5.1828e-01, -8.4454e-01, -8.7808e-01, ..., -1.0624e+00,
-6.6299e-01, -9.5629e-01],
[-3.7878e-01, -2.6182e-01, -4.2786e-01, ..., -1.1532e+00,
-5.8686e-01, -4.5328e-01],
...,
[-9.3089e-02, -6.2110e-01, -5.8798e-01, ..., -9.7887e-01,
-8.5458e-01, -1.2071e+00],
[-7.0869e-01, -2.2775e-01, -5.4780e-01, ..., -5.5369e-01,
-2.7530e-01, -4.4830e-01],
[-8.7707e-01, -2.7714e-01, -4.6453e-01, ..., -5.1311e-01,
-4.9290e-01, -6.4861e-01]],
...,
[[-2.7455e+00, -2.3259e+00, -2.3351e+00, ..., -3.3018e+00,
-2.6972e+00, -1.9520e+00],
[-2.0216e+00, -7.4937e-01, -1.0491e+00, ..., -2.5857e+00,
-2.6499e+00, -2.1324e+00],
[-1.0580e+00, 8.3792e-01, 7.3349e-01, ..., -1.2969e-01,
3.3371e-02, 3.1688e-01],
...,
[-2.2269e+00, -1.2101e-01, -1.0986e-01, ..., -8.5533e-01,
-2.4194e-01, 2.3622e-01],
[-1.2201e+00, 8.6613e-01, 1.0523e+00, ..., 4.0249e-01,
4.6195e-01, 7.6692e-01],
[-2.1699e+00, -1.5116e+00, -1.6883e+00, ..., -1.9711e+00,
-9.8996e-01, -2.6376e-01]],
[[-1.6145e+00, -1.6338e+00, -2.8362e+00, ..., -2.1849e+00,
-1.8631e+00, -1.7885e+00],
[-7.2464e-01, -1.5321e-01, -3.2114e+00, ..., -2.8058e+00,
-2.8623e+00, -2.9419e+00],
[-2.0454e+00, -1.0064e+00, -5.1555e+00, ..., -4.5288e+00,
-4.6316e+00, -3.8181e+00],
...,
[-1.3540e+00, 1.6092e-01, -3.3982e+00, ..., -3.0604e+00,
-2.9245e+00, -2.4354e+00],
[-1.6763e+00, -4.7255e-01, -3.0414e+00, ..., -2.5719e+00,
-2.3972e+00, -1.8015e+00],
[-2.3110e+00, -1.3617e+00, -3.0318e+00, ..., -2.8161e+00,
-2.4461e+00, -2.2036e+00]],
[[-1.2854e+00, -1.8020e+00, -1.6272e+00, ..., -2.7039e+00,
-2.2519e+00, -1.7525e+00],
[-1.2391e+00, -1.6953e+00, -1.2238e+00, ..., -2.4884e+00,
-2.0001e+00, -1.6389e+00],
[-1.5668e+00, -1.9017e+00, -1.4849e+00, ..., -2.8806e+00,
-2.2985e+00, -1.7762e+00],
...,
[-2.4362e+00, -3.0441e+00, -2.7763e+00, ..., -3.9580e+00,
-3.2173e+00, -2.5253e+00],
[-1.5765e+00, -2.0752e+00, -1.9786e+00, ..., -2.7582e+00,
-2.3063e+00, -1.9012e+00],
[-1.0302e+00, -1.3363e+00, -1.3640e+00, ..., -1.8545e+00,
-1.5334e+00, -1.1891e+00]]],
[[[ 1.1383e-01, 3.3981e-01, -1.1919e+00, ..., 4.5760e-01,
1.0157e+00, 6.5451e-01],
[-1.9615e+00, -1.3528e+00, -4.1570e+00, ..., -2.5914e+00,
-7.7062e-01, -8.1367e-02],
[ 3.5663e-01, -2.7980e-01, -1.1080e+00, ..., -2.8622e-01,
4.5519e-01, 6.2259e-01],
...,
[-3.3500e-01, -1.4427e+00, -2.8545e+00, ..., -2.0215e+00,
-1.3791e+00, -9.0429e-01],
[ 1.3453e+00, 6.8348e-01, -9.4284e-01, ..., -3.6738e-01,
5.2408e-01, 7.0688e-01],
[ 8.8124e-01, 3.5685e-01, -4.3255e-01, ..., -3.9459e-01,
2.9196e-01, 2.7055e-01]],
[[-3.1470e+00, -1.5254e+00, -6.0850e+00, ..., -4.8430e+00,
-4.4118e+00, -3.3748e+00],
[-7.4273e-01, 1.7335e+00, -4.4070e+00, ..., -3.3954e+00,
-3.5428e+00, -2.9157e+00],
[-2.7042e-01, 1.3972e+00, -7.4478e+00, ..., -5.3917e+00,
-5.2725e+00, -4.9404e+00],
...,
[-1.0680e+00, 6.1045e-01, -6.3085e+00, ..., -4.5503e+00,
-4.1222e+00, -3.4100e+00],
[ 1.9430e+00, 3.1341e+00, -7.4696e-01, ..., -8.8139e-01,
-1.2080e+00, -1.2278e+00],
[ 1.2702e+00, 1.9462e+00, -2.1551e+00, ..., -2.2496e+00,
-2.4731e+00, -2.3584e+00]],
[[-1.2792e+00, -2.1366e+00, -1.9798e+00, ..., -5.8261e-01,
-1.9474e-01, -4.6434e-01],
[-1.8175e+00, -1.9906e+00, -1.4370e+00, ..., -3.4877e-01,
6.4760e-02, 3.0043e-02],
[-2.3483e+00, -1.8865e+00, -8.2205e-01, ..., -2.9810e-01,
2.3248e-01, 2.7586e-01],
...,
[-1.6812e+00, -1.9608e+00, -7.9998e-01, ..., -4.7771e-01,
-1.4998e-01, -3.3979e-01],
[-1.8947e+00, -1.2963e+00, -8.3584e-01, ..., -3.0997e-01,
1.5757e-01, 2.1409e-01],
[-1.9042e+00, -1.2273e+00, -6.4108e-01, ..., -5.1783e-01,
-2.8080e-01, -2.6136e-01]],
...,
[[-2.1062e+00, -1.2057e+00, -2.5306e+00, ..., -3.6711e+00,
-2.8327e+00, -1.8379e+00],
[-1.8125e+00, 5.7326e-02, -1.3099e+00, ..., -2.6939e+00,
-2.4368e+00, -1.8769e+00],
[-4.8968e-01, 2.2416e+00, 8.3710e-01, ..., -5.4750e-01,
-4.7675e-02, 1.6460e-01],
...,
[-1.9077e+00, 1.3220e+00, 2.6428e-01, ..., -1.5245e+00,
-7.2840e-01, -1.6705e-01],
[-9.3599e-01, 1.8618e+00, 1.6530e+00, ..., -1.2480e-01,
8.2472e-02, 3.3023e-01],
[-2.2305e+00, -1.1357e+00, -9.2385e-01, ..., -2.3198e+00,
-1.4661e+00, -9.0342e-01]],
[[-2.3720e+00, -2.6733e+00, -3.9047e+00, ..., -1.1221e+00,
-7.3009e-01, -7.5768e-01],
[-1.7808e+00, -1.5586e+00, -4.6283e+00, ..., -1.3300e+00,
-1.2095e+00, -1.5822e+00],
[-3.0225e+00, -2.5884e+00, -6.6062e+00, ..., -3.2467e+00,
-2.7837e+00, -2.2304e+00],
...,
[-2.4465e+00, -1.4706e+00, -5.0690e+00, ..., -1.4402e+00,
-1.0376e+00, -8.9058e-01],
[-2.1630e+00, -1.3703e+00, -3.9660e+00, ..., -1.7056e+00,
-1.1439e+00, -7.8336e-01],
[-2.3533e+00, -1.7114e+00, -3.1159e+00, ..., -2.7837e+00,
-2.0052e+00, -1.8526e+00]],
[[-1.8322e+00, -2.5221e+00, -2.3594e+00, ..., -1.8083e+00,
-1.4997e+00, -1.1386e+00],
[-1.8762e+00, -2.5609e+00, -2.1178e+00, ..., -1.4026e+00,
-1.1117e+00, -8.8535e-01],
[-2.3998e+00, -2.9577e+00, -2.5591e+00, ..., -1.6009e+00,
-1.2004e+00, -8.8664e-01],
...,
[-3.2180e+00, -4.0149e+00, -3.7304e+00, ..., -2.8678e+00,
-2.2939e+00, -1.7634e+00],
[-2.1352e+00, -2.7585e+00, -2.6819e+00, ..., -1.9826e+00,
-1.6302e+00, -1.3155e+00],
[-1.5206e+00, -1.8475e+00, -1.8742e+00, ..., -1.2361e+00,
-1.0373e+00, -7.9967e-01]]],
[[[ 1.7898e-02, 1.3085e-01, -1.6974e+00, ..., -7.7844e-01,
-3.5688e-01, -1.6384e-01],
[-2.2106e+00, -1.7888e+00, -4.9153e+00, ..., -4.0573e+00,
-1.7200e+00, -7.0292e-01],
[ 4.8253e-01, -2.1399e-01, -1.0117e+00, ..., -3.0424e-01,
-4.3645e-01, 1.3505e-01],
...,
[-1.7926e-01, -8.7843e-01, -1.1956e+00, ..., -1.1666e+00,
-8.6133e-01, -6.2071e-01],
[ 1.0344e+00, 6.8032e-01, 4.4770e-01, ..., 5.1080e-01,
7.1129e-01, 8.2217e-01],
[ 7.5671e-01, 5.3004e-01, 4.1869e-01, ..., 6.2104e-01,
7.9234e-01, 7.3727e-01]],
[[-3.4056e+00, -1.8100e+00, -6.4683e+00, ..., -7.3319e+00,
-6.4518e+00, -4.7606e+00],
[-8.5806e-01, 1.5859e+00, -4.5548e+00, ..., -5.6514e+00,
-5.7488e+00, -4.4182e+00],
[-4.4284e-01, 1.1525e+00, -7.7022e+00, ..., -8.2307e+00,
-8.0839e+00, -6.8359e+00],
...,
[-5.5826e-01, 1.6470e-01, -4.6283e+00, ..., -5.5914e+00,
-4.7623e+00, -3.9989e+00],
[ 1.7841e+00, 2.2570e+00, -3.5681e-01, ..., -1.3399e+00,
-1.3869e+00, -1.4850e+00],
[ 9.9463e-01, 1.1991e+00, -1.8323e+00, ..., -2.5795e+00,
-2.6265e+00, -2.6051e+00]],
[[-1.4445e+00, -2.4569e+00, -2.4219e+00, ..., -1.8738e+00,
-1.3391e+00, -1.4573e+00],
[-1.7525e+00, -2.0500e+00, -1.5835e+00, ..., -1.3180e+00,
-7.9426e-01, -8.9511e-01],
[-2.3085e+00, -2.0726e+00, -1.0889e+00, ..., -1.1952e+00,
-5.4210e-01, -2.2379e-01],
...,
[-7.4081e-01, -1.1590e+00, -5.6893e-01, ..., -5.7088e-01,
-4.7100e-01, -6.0445e-01],
[-1.1313e+00, -7.6612e-01, -5.2646e-01, ..., -2.4442e-01,
-6.9967e-02, -3.5478e-02],
[-1.2096e+00, -6.7348e-01, -3.1577e-01, ..., -2.8460e-01,
-2.7211e-01, -2.2858e-01]],
...,
[[-2.0461e+00, -1.1075e+00, -2.3711e+00, ..., -3.1398e+00,
-2.5762e+00, -1.8176e+00],
[-1.9661e+00, -1.6224e-01, -1.5107e+00, ..., -2.5885e+00,
-2.6027e+00, -2.0278e+00],
[-4.8679e-01, 2.3041e+00, 1.0308e+00, ..., 1.5023e-01,
3.0987e-01, 5.5409e-01],
...,
[-1.4052e+00, 1.0589e+00, -2.0445e-01, ..., -1.2038e+00,
-4.2906e-01, 4.6219e-02],
[-4.4783e-01, 1.4449e+00, 6.6696e-01, ..., -5.8074e-02,
1.8264e-01, 2.8510e-01],
[-1.5354e+00, -1.0270e+00, -1.5604e+00, ..., -2.3261e+00,
-1.3166e+00, -6.8607e-01]],
[[-2.5191e+00, -3.0034e+00, -4.3697e+00, ..., -2.5654e+00,
-2.0625e+00, -1.8617e+00],
[-1.8731e+00, -1.8117e+00, -5.0738e+00, ..., -3.0971e+00,
-2.9296e+00, -2.8894e+00],
[-2.9014e+00, -2.6018e+00, -6.8377e+00, ..., -4.5850e+00,
-4.4910e+00, -3.5844e+00],
...,
[-1.7555e+00, -7.4034e-01, -4.2029e+00, ..., -2.2486e+00,
-1.6403e+00, -1.2570e+00],
[-1.7098e+00, -8.7461e-01, -3.4380e+00, ..., -2.1647e+00,
-1.4277e+00, -8.5807e-01],
[-2.0406e+00, -1.3712e+00, -2.8920e+00, ..., -2.7421e+00,
-1.8357e+00, -1.5663e+00]],
[[-1.8892e+00, -2.6506e+00, -2.5686e+00, ..., -2.8245e+00,
-2.3160e+00, -1.7791e+00],
[-1.8968e+00, -2.6404e+00, -2.2772e+00, ..., -2.5441e+00,
-2.0154e+00, -1.6353e+00],
[-2.3740e+00, -2.9759e+00, -2.6696e+00, ..., -2.9272e+00,
-2.2855e+00, -1.7466e+00],
...,
[-2.2687e+00, -2.8366e+00, -2.6595e+00, ..., -2.8989e+00,
-2.3061e+00, -1.7398e+00],
[-1.5499e+00, -1.9651e+00, -1.9326e+00, ..., -2.0603e+00,
-1.6845e+00, -1.3376e+00],
[-1.0535e+00, -1.2719e+00, -1.3402e+00, ..., -1.2891e+00,
-1.0907e+00, -8.0580e-01]]],
[[[-1.1777e-02, 1.1073e-01, -1.7420e+00, ..., -8.1261e-01,
-3.8893e-01, -1.6410e-01],
[-2.2481e+00, -1.8200e+00, -4.9893e+00, ..., -4.1589e+00,
-1.7865e+00, -7.1630e-01],
[ 4.7620e-01, -2.6424e-01, -5.3113e-01, ..., -3.4080e-01,
-4.4917e-01, 1.4975e-01],
...,
[-7.3728e-01, -1.7429e+00, -1.9144e+00, ..., -1.5414e+00,
-1.1101e+00, -7.6315e-01],
[ 9.6941e-01, 2.3091e-01, 1.2376e-03, ..., 8.9961e-02,
7.2761e-01, 8.1566e-01],
[ 3.5082e-01, -3.1814e-01, -3.1771e-01, ..., -1.4014e-02,
5.1212e-01, 4.3186e-01]],
[[-3.4079e+00, -1.8169e+00, -6.4565e+00, ..., -7.3862e+00,
-6.4875e+00, -4.7766e+00],
[-8.5136e-01, 1.5738e+00, -4.5559e+00, ..., -5.6728e+00,
-5.7711e+00, -4.4125e+00],
[-4.4920e-01, 1.1056e+00, -7.5601e+00, ..., -8.2555e+00,
-8.1373e+00, -6.8479e+00],
...,
[ 3.2547e-01, 7.0712e-01, -3.8379e+00, ..., -4.7115e+00,
-4.2634e+00, -3.5713e+00],
[ 2.6709e+00, 2.8429e+00, 4.8820e-01, ..., -9.8246e-01,
-1.3217e+00, -1.3590e+00],
[ 1.6058e+00, 1.5324e+00, -1.4848e+00, ..., -2.3185e+00,
-2.5408e+00, -2.4334e+00]],
[[-1.4889e+00, -2.4778e+00, -2.4213e+00, ..., -1.8819e+00,
-1.3507e+00, -1.4877e+00],
[-1.8179e+00, -2.0784e+00, -1.5648e+00, ..., -1.3509e+00,
-8.1443e-01, -9.2183e-01],
[-2.4089e+00, -2.1216e+00, -9.3358e-01, ..., -1.2291e+00,
-5.3567e-01, -2.2726e-01],
...,
[-3.9140e-01, -8.9372e-01, -5.2284e-01, ..., -4.6119e-01,
-1.9425e-01, -4.2618e-01],
[-9.5124e-01, -3.9967e-01, -4.7440e-01, ..., -1.7496e-01,
2.2012e-01, 2.2097e-01],
[-1.0675e+00, -3.7180e-01, -3.5938e-01, ..., -3.4116e-01,
-1.7693e-01, -2.1396e-01]],
...,
[[-2.0350e+00, -1.0562e+00, -2.3091e+00, ..., -3.2096e+00,
-2.6437e+00, -1.8635e+00],
[-1.9724e+00, -1.4406e-01, -1.5141e+00, ..., -2.6503e+00,
-2.6655e+00, -2.0464e+00],
[-4.0470e-01, 2.4013e+00, 1.3549e+00, ..., 1.4420e-01,
2.4562e-01, 5.2478e-01],
...,
[-2.1447e+00, 2.8627e-01, -6.9490e-02, ..., -1.2615e+00,
-3.9068e-01, 1.0025e-01],
[-1.1375e+00, 1.1124e+00, 1.0602e+00, ..., 2.0811e-02,
2.8234e-01, 4.3874e-01],
[-2.1280e+00, -1.4231e+00, -1.7251e+00, ..., -2.3015e+00,
-1.3852e+00, -8.4472e-01]],
[[-2.5464e+00, -3.0198e+00, -4.4068e+00, ..., -2.5592e+00,
-2.0784e+00, -1.8795e+00],
[-1.8988e+00, -1.8268e+00, -5.0993e+00, ..., -3.0533e+00,
-2.9507e+00, -2.9256e+00],
[-2.9183e+00, -2.5821e+00, -6.6795e+00, ..., -4.5431e+00,
-4.5325e+00, -3.6323e+00],
...,
[-1.3319e+00, 1.3713e-01, -3.6296e+00, ..., -1.5400e+00,
-1.1584e+00, -1.0098e+00],
[-1.6494e+00, -4.6480e-01, -3.1889e+00, ..., -1.7832e+00,
-1.2416e+00, -8.6449e-01],
[-2.3361e+00, -1.4008e+00, -3.1301e+00, ..., -2.8969e+00,
-2.1042e+00, -1.9122e+00]],
[[-1.9143e+00, -2.6727e+00, -2.5840e+00, ..., -2.8452e+00,
-2.3347e+00, -1.7946e+00],
[-1.9149e+00, -2.6672e+00, -2.2950e+00, ..., -2.5582e+00,
-2.0192e+00, -1.6389e+00],
[-2.3912e+00, -3.0137e+00, -2.8822e+00, ..., -2.9293e+00,
-2.2926e+00, -1.7442e+00],
...,
[-2.4754e+00, -3.0677e+00, -2.7671e+00, ..., -2.8094e+00,
-2.2412e+00, -1.7162e+00],
[-1.6053e+00, -2.0838e+00, -1.9595e+00, ..., -1.9131e+00,
-1.5699e+00, -1.2592e+00],
[-1.0411e+00, -1.3277e+00, -1.3314e+00, ..., -1.1774e+00,
-9.9338e-01, -7.5228e-01]]],
[[[ 3.8983e-01, 8.2714e-01, 5.2747e-01, ..., 4.5624e-01,
9.6249e-01, 6.1416e-01],
[-1.3440e+00, -4.9482e-01, -2.0060e+00, ..., -2.4524e+00,
-7.4901e-01, -7.4747e-02],
[ 8.9260e-01, 6.0142e-01, 3.6166e-01, ..., -2.5106e-01,
4.7614e-01, 6.3533e-01],
...,
[ 3.9321e-02, -6.3920e-01, -1.2964e+00, ..., -2.2033e+00,
-1.6014e+00, -1.1442e+00],
[ 1.3177e+00, 1.0908e+00, 6.0031e-01, ..., -3.5619e-01,
4.8041e-01, 6.9105e-01],
[ 9.6193e-01, 8.6044e-01, 6.1518e-01, ..., -3.5911e-01,
2.9424e-01, 2.7855e-01]],
[[-2.7782e+00, -1.5766e+00, -4.8018e+00, ..., -4.6972e+00,
-4.3149e+00, -3.2983e+00],
[-4.9374e-01, 1.6189e+00, -2.8931e+00, ..., -3.3579e+00,
-3.5000e+00, -2.8708e+00],
[ 3.1612e-01, 1.4202e+00, -5.6635e+00, ..., -5.2515e+00,
-5.1553e+00, -4.8125e+00],
...,
[-9.6803e-01, -9.7217e-02, -4.9093e+00, ..., -4.5579e+00,
-4.0663e+00, -3.3846e+00],
[ 1.5866e+00, 2.1797e+00, -3.9555e-01, ..., -5.3056e-01,
-8.9923e-01, -1.0308e+00],
[ 9.0002e-01, 1.1746e+00, -1.8275e+00, ..., -2.1278e+00,
-2.4060e+00, -2.3485e+00]],
[[-7.2683e-01, -1.9085e+00, -1.7703e+00, ..., -5.2096e-01,
-1.5604e-01, -4.3103e-01],
[-1.5071e+00, -1.9696e+00, -1.2994e+00, ..., -3.0493e-01,
7.9545e-02, 4.0401e-02],
[-1.3727e+00, -1.3522e+00, -7.2681e-01, ..., -2.5394e-01,
2.3332e-01, 2.6767e-01],
...,
[-8.7938e-01, -1.3156e+00, -6.1131e-01, ..., -5.2363e-01,
-1.8057e-01, -4.0203e-01],
[-1.1302e+00, -8.2039e-01, -5.5137e-01, ..., -3.1351e-01,
1.7755e-01, 2.3552e-01],
[-1.1654e+00, -6.8071e-01, -2.9764e-01, ..., -4.8730e-01,
-2.6119e-01, -2.5055e-01]],
...,
[[-2.3077e+00, -1.1510e+00, -2.4879e+00, ..., -3.4516e+00,
-2.6527e+00, -1.7131e+00],
[-6.1543e-01, 1.3109e+00, -6.3951e-01, ..., -2.5845e+00,
-2.3276e+00, -1.8024e+00],
[-2.2337e-01, 2.6005e+00, 8.7367e-01, ..., -5.6742e-01,
-4.7102e-02, 1.4808e-01],
...,
[-1.3170e+00, 1.2500e+00, -1.5497e-02, ..., -1.6019e+00,
-6.8034e-01, -9.1428e-02],
[-5.7410e-01, 1.3964e+00, 7.5436e-01, ..., -1.3941e-01,
1.8260e-01, 3.9974e-01],
[-1.6777e+00, -1.1613e+00, -1.5458e+00, ..., -2.6075e+00,
-1.5713e+00, -9.6689e-01]],
[[-1.4310e+00, -1.3651e+00, -3.0248e+00, ..., -1.2386e+00,
-8.1548e-01, -7.9976e-01],
[-6.8917e-01, -1.8658e-01, -3.8755e+00, ..., -1.3968e+00,
-1.2239e+00, -1.5540e+00],
[-2.2123e+00, -1.5454e+00, -6.3693e+00, ..., -3.2159e+00,
-2.7398e+00, -2.1775e+00],
...,
[-1.8306e+00, -9.5498e-01, -4.5006e+00, ..., -1.3565e+00,
-9.0160e-01, -7.8220e-01],
[-1.7227e+00, -1.0350e+00, -3.6726e+00, ..., -1.6683e+00,
-1.0622e+00, -6.7545e-01],
[-2.0392e+00, -1.4808e+00, -3.0221e+00, ..., -2.8815e+00,
-2.0448e+00, -1.8773e+00]],
[[-1.2643e+00, -1.9117e+00, -1.7420e+00, ..., -1.7827e+00,
-1.4865e+00, -1.1268e+00],
[-1.2680e+00, -1.8366e+00, -1.4059e+00, ..., -1.4199e+00,
-1.1346e+00, -9.0424e-01],
[-1.6408e+00, -2.0281e+00, -1.6427e+00, ..., -1.6434e+00,
-1.2385e+00, -9.1655e-01],
...,
[-2.3214e+00, -2.9575e+00, -2.8600e+00, ..., -2.9766e+00,
-2.3850e+00, -1.8256e+00],
[-1.5613e+00, -2.0177e+00, -2.0340e+00, ..., -2.0339e+00,
-1.6802e+00, -1.3535e+00],
[-1.0612e+00, -1.3057e+00, -1.4230e+00, ..., -1.2351e+00,
-1.0561e+00, -8.0722e-01]]]], device='cuda:0'),
tensor([[[[ 1.6335e-01, 3.9661e-01, -1.1626e+00, ..., 4.2836e-01,
9.9491e-01, 6.2745e-01],
[-2.0773e+00, -1.4253e+00, -4.3628e+00, ..., -2.6244e+00,
-7.8979e-01, -9.7017e-02],
[ 4.5620e-01, -2.6166e-01, -1.1092e+00, ..., -2.8275e-01,
4.6781e-01, 6.0900e-01],
...,
[-2.5920e-01, -1.2616e+00, -2.5884e+00, ..., -2.0226e+00,
-1.4264e+00, -9.4345e-01],
[ 1.2398e+00, 6.5812e-01, -8.8078e-01, ..., -3.3737e-01,
4.9672e-01, 7.0025e-01],
[ 8.1317e-01, 3.1576e-01, -3.9504e-01, ..., -3.6486e-01,
2.8710e-01, 2.7984e-01]],
[[-3.2383e+00, -1.4469e+00, -6.1114e+00, ..., -4.8508e+00,
-4.4377e+00, -3.3638e+00],
[-5.5982e-01, 2.1693e+00, -4.1469e+00, ..., -3.3736e+00,
-3.5394e+00, -2.8765e+00],
[ 5.7519e-02, 1.8821e+00, -7.3548e+00, ..., -5.3722e+00,
-5.2547e+00, -4.8943e+00],
...,
[-1.2924e+00, 1.8222e-01, -6.5298e+00, ..., -4.5457e+00,
-4.1130e+00, -3.4430e+00],
[ 1.4510e+00, 2.5229e+00, -1.4053e+00, ..., -8.5618e-01,
-1.1922e+00, -1.2489e+00],
[ 9.4521e-01, 1.5751e+00, -2.4180e+00, ..., -2.2347e+00,
-2.4706e+00, -2.3654e+00]],
[[-1.3776e+00, -2.3466e+00, -2.1531e+00, ..., -6.0285e-01,
-2.0159e-01, -4.7003e-01],
[-2.0177e+00, -2.2064e+00, -1.5624e+00, ..., -3.4705e-01,
6.4540e-02, 3.0968e-02],
[-2.5152e+00, -1.9982e+00, -8.3276e-01, ..., -2.8572e-01,
2.4840e-01, 2.7831e-01],
...,
[-1.5749e+00, -1.7633e+00, -7.7120e-01, ..., -4.9036e-01,
-1.5692e-01, -3.4527e-01],
[-1.7216e+00, -1.1313e+00, -7.3990e-01, ..., -3.2153e-01,
1.4747e-01, 2.1103e-01],
[-1.6617e+00, -1.0544e+00, -5.6332e-01, ..., -5.0432e-01,
-2.7963e-01, -2.5567e-01]],
...,
[[-2.3696e+00, -1.3013e+00, -2.7501e+00, ..., -3.6586e+00,
-2.8172e+00, -1.8306e+00],
[-1.8629e+00, 2.3450e-01, -1.3203e+00, ..., -2.6575e+00,
-2.4122e+00, -1.8606e+00],
[-4.9732e-01, 2.5159e+00, 9.3782e-01, ..., -5.4288e-01,
-3.3864e-02, 1.8806e-01],
...,
[-1.6903e+00, 1.1183e+00, 2.2100e-01, ..., -1.5491e+00,
-7.0981e-01, -1.5605e-01],
[-8.5741e-01, 1.5797e+00, 1.4475e+00, ..., -1.4430e-01,
1.0598e-01, 3.2463e-01],
[-2.1325e+00, -1.1130e+00, -9.1386e-01, ..., -2.3468e+00,
-1.4470e+00, -9.0583e-01]],
[[-2.3602e+00, -2.6062e+00, -4.0065e+00, ..., -1.1315e+00,
-7.1280e-01, -7.4490e-01],
[-1.6446e+00, -1.3015e+00, -4.7517e+00, ..., -1.3494e+00,
-1.1853e+00, -1.5660e+00],
[-3.0045e+00, -2.4163e+00, -6.9125e+00, ..., -3.2774e+00,
-2.7742e+00, -2.2124e+00],
...,
[-2.5401e+00, -1.7562e+00, -4.9082e+00, ..., -1.4586e+00,
-1.0351e+00, -8.8676e-01],
[-2.2347e+00, -1.5653e+00, -3.8385e+00, ..., -1.7183e+00,
-1.1587e+00, -7.7175e-01],
[-2.3029e+00, -1.8011e+00, -3.0773e+00, ..., -2.8016e+00,
-2.0289e+00, -1.8474e+00]],
[[-1.8862e+00, -2.6125e+00, -2.3926e+00, ..., -1.8173e+00,
-1.5070e+00, -1.1489e+00],
[-1.9081e+00, -2.6216e+00, -2.0767e+00, ..., -1.4077e+00,
-1.1192e+00, -8.9656e-01],
[-2.4311e+00, -2.9847e+00, -2.4869e+00, ..., -1.6122e+00,
-1.2042e+00, -8.9710e-01],
...,
[-3.0864e+00, -3.8540e+00, -3.6477e+00, ..., -2.8701e+00,
-2.2925e+00, -1.7603e+00],
[-2.0491e+00, -2.6451e+00, -2.6244e+00, ..., -1.9769e+00,
-1.6238e+00, -1.3108e+00],
[-1.4658e+00, -1.7834e+00, -1.8413e+00, ..., -1.2267e+00,
-1.0354e+00, -7.9381e-01]]],
[[[ 2.7698e-01, 5.9152e-01, 6.4607e-01, ..., -7.9351e-01,
-4.9237e-01, -2.3018e-01],
[-1.6932e+00, -9.1519e-01, -2.2653e+00, ..., -3.9647e+00,
-1.7353e+00, -7.4891e-01],
[ 1.2604e-01, -4.2351e-01, 2.3165e-01, ..., -4.4695e-01,
-7.3576e-01, 4.4609e-02],
...,
[-6.6702e-01, -1.6608e+00, -1.7220e+00, ..., -1.3919e+01,
-2.6602e+00, 2.3844e+00],
[ 9.4892e-01, 2.7784e-01, 1.9633e-01, ..., -2.8958e-01,
-1.7448e-02, 2.9254e-01],
[ 4.4560e-01, -1.1667e-01, 3.5210e-03, ..., -1.3668e-01,
2.1026e-01, 3.7387e-01]],
[[-1.9108e+00, -1.3826e+00, -4.3025e+00, ..., -7.3133e+00,
-6.6253e+00, -4.9505e+00],
[ 1.8717e-01, 1.4269e+00, -2.6408e+00, ..., -5.8974e+00,
-6.2085e+00, -4.7501e+00],
[ 9.1698e-01, 1.1829e+00, -5.1521e+00, ..., -8.5213e+00,
-8.6156e+00, -7.0861e+00],
...,
[ 3.0560e-03, 2.2462e-01, -4.2922e+00, ..., -1.0633e+01,
-1.3823e+01, -1.1531e+01],
[ 2.3272e+00, 2.4484e+00, 8.4584e-02, ..., -2.3659e+00,
-3.1323e+00, -2.3670e+00],
[ 1.3686e+00, 1.3070e+00, -1.6261e+00, ..., -3.1657e+00,
-3.8438e+00, -3.2312e+00]],
[[-2.8266e-01, -1.3205e+00, -1.4380e+00, ..., -1.6874e+00,
-1.3231e+00, -1.5607e+00],
[-7.7555e-01, -1.1231e+00, -9.7443e-01, ..., -1.1752e+00,
-8.1308e-01, -1.1326e+00],
[-5.8778e-01, -4.8603e-01, -4.4203e-01, ..., -1.3304e+00,
-7.7029e-01, -5.5703e-01],
...,
[-2.5818e-01, -8.2446e-01, -6.2674e-01, ..., -1.6046e+00,
-2.9901e+00, -1.1585e+00],
[-8.1515e-01, -4.1111e-01, -5.5350e-01, ..., -5.3748e-01,
-2.3930e-01, -4.3421e-01],
[-9.7211e-01, -3.9690e-01, -4.0922e-01, ..., -8.1347e-01,
-7.8542e-01, -9.3716e-01]],
...,
[[-2.5898e+00, -2.0160e+00, -2.3645e+00, ..., -3.0456e+00,
-2.4589e+00, -1.8261e+00],
[-1.5673e+00, -1.4720e-01, -8.1830e-01, ..., -2.7878e+00,
-2.7565e+00, -2.1829e+00],
[-8.1859e-01, 1.2469e+00, 7.5245e-01, ..., -1.0735e-01,
2.8104e-03, 2.4527e-01],
...,
[-2.0294e+00, 1.6631e-01, -1.3006e-01, ..., 1.2362e+00,
-1.6734e+00, 2.7788e-01],
[-1.0042e+00, 1.0234e+00, 9.8367e-01, ..., 6.7573e-01,
5.9220e-01, 9.2524e-01],
[-1.9889e+00, -1.3626e+00, -1.6328e+00, ..., -1.9272e+00,
-9.3722e-01, -1.6582e-01]],
[[-1.5214e+00, -1.4944e+00, -2.7263e+00, ..., -2.6941e+00,
-2.4112e+00, -2.2722e+00],
[-6.6813e-01, -1.1797e-01, -3.2444e+00, ..., -3.3586e+00,
-3.4048e+00, -3.2910e+00],
[-2.0520e+00, -1.1115e+00, -5.3284e+00, ..., -4.7544e+00,
-4.9704e+00, -4.0547e+00],
...,
[-1.4421e+00, -1.6441e-01, -3.7193e+00, ..., -1.5945e+00,
1.7446e+00, -1.6048e-01],
[-1.6656e+00, -6.4025e-01, -3.1996e+00, ..., -2.5703e+00,
-2.5247e+00, -1.9050e+00],
[-2.2358e+00, -1.4152e+00, -3.0107e+00, ..., -2.9653e+00,
-2.6605e+00, -2.3931e+00]],
[[-1.2427e+00, -1.7822e+00, -1.6067e+00, ..., -2.9099e+00,
-2.4342e+00, -1.9013e+00],
[-1.2051e+00, -1.6788e+00, -1.2282e+00, ..., -2.7960e+00,
-2.2425e+00, -1.8398e+00],
[-1.5373e+00, -1.8701e+00, -1.4644e+00, ..., -3.2930e+00,
-2.6485e+00, -2.0646e+00],
...,
[-2.3777e+00, -2.9631e+00, -2.7357e+00, ..., -7.5728e+00,
-6.7304e+00, -3.7977e+00],
[-1.5590e+00, -2.0256e+00, -1.9494e+00, ..., -3.0339e+00,
-2.5320e+00, -2.0995e+00],
[-1.0262e+00, -1.3019e+00, -1.3370e+00, ..., -2.0344e+00,
-1.6603e+00, -1.3006e+00]]],
[[[-1.1914e-01, -5.9610e-02, -2.3254e+00, ..., -1.1768e+00,
-7.2951e-01, -3.5125e-01],
[-2.5315e+00, -2.2382e+00, -5.9231e+00, ..., -4.8020e+00,
-2.1920e+00, -9.5532e-01],
[ 3.5778e-01, -5.1897e-01, -1.6103e+00, ..., -5.7423e-01,
-7.6745e-01, 4.5048e-02],
...,
[-4.1757e-01, -1.1799e+00, -1.4266e+00, ..., -1.3768e+00,
-1.0601e+00, -7.7904e-01],
[ 1.0230e+00, 5.2384e-01, 2.6919e-01, ..., 3.4895e-01,
8.1601e-01, 8.7163e-01],
[ 4.8830e-01, 3.3333e-02, -8.1851e-03, ..., 2.9632e-01,
6.8364e-01, 5.5324e-01]],
[[-3.6333e+00, -1.9877e+00, -6.9514e+00, ..., -7.8052e+00,
-6.8729e+00, -5.0138e+00],
[-9.8272e-01, 1.5553e+00, -4.9114e+00, ..., -6.0369e+00,
-6.1952e+00, -4.7129e+00],
[-6.3340e-01, 1.0510e+00, -8.1440e+00, ..., -8.7484e+00,
-8.7084e+00, -7.2374e+00],
...,
[-1.3126e-01, 2.7926e-01, -4.1955e+00, ..., -4.9860e+00,
-4.3807e+00, -3.6568e+00],
[ 2.0653e+00, 2.2782e+00, -1.7023e-01, ..., -9.7057e-01,
-1.2140e+00, -1.2945e+00],
[ 1.2035e+00, 1.1900e+00, -1.7537e+00, ..., -2.3466e+00,
-2.5353e+00, -2.4798e+00]],
[[-1.7785e+00, -2.7264e+00, -2.6403e+00, ..., -2.0641e+00,
-1.5056e+00, -1.6601e+00],
[-2.0602e+00, -2.2581e+00, -1.7059e+00, ..., -1.4773e+00,
-9.3661e-01, -1.1059e+00],
[-2.8340e+00, -2.4254e+00, -1.1777e+00, ..., -1.4312e+00,
-6.8842e-01, -3.0038e-01],
...,
[-4.6376e-01, -9.3132e-01, -5.0086e-01, ..., -4.8641e-01,
-2.5941e-01, -4.7639e-01],
[-8.8145e-01, -4.3129e-01, -4.1441e-01, ..., -1.4052e-01,
2.0480e-01, 1.9792e-01],
[-9.2762e-01, -3.3123e-01, -2.4876e-01, ..., -2.5729e-01,
-1.3417e-01, -1.6822e-01]],
...,
[[-2.0667e+00, -1.1033e+00, -2.3724e+00, ..., -3.2198e+00,
-2.6868e+00, -1.8873e+00],
[-2.3523e+00, -5.0830e-01, -1.7864e+00, ..., -2.9886e+00,
-2.9637e+00, -2.1973e+00],
[-5.4208e-01, 2.2785e+00, 1.0850e+00, ..., 1.6321e-01,
1.7557e-01, 4.7602e-01],
...,
[-1.7021e+00, 5.3990e-01, -3.1439e-02, ..., -1.2393e+00,
-3.0807e-01, 2.1392e-01],
[-8.5977e-01, 1.1419e+00, 9.4296e-01, ..., 2.1190e-02,
3.2131e-01, 4.8781e-01],
[-1.8376e+00, -1.1992e+00, -1.5173e+00, ..., -2.3736e+00,
-1.3636e+00, -7.8888e-01]],
[[-2.7914e+00, -3.3901e+00, -4.6919e+00, ..., -2.8620e+00,
-2.4102e+00, -2.2330e+00],
[-2.1524e+00, -2.1984e+00, -5.3345e+00, ..., -3.3525e+00,
-3.2951e+00, -3.2621e+00],
[-3.1141e+00, -2.9037e+00, -6.9048e+00, ..., -4.6823e+00,
-4.8264e+00, -3.9258e+00],
...,
[-1.5231e+00, -3.6161e-01, -3.7790e+00, ..., -1.7832e+00,
-1.3028e+00, -1.0893e+00],
[-1.6743e+00, -7.3878e-01, -3.2174e+00, ..., -1.9041e+00,
-1.2767e+00, -8.3930e-01],
[-2.1899e+00, -1.4426e+00, -3.0398e+00, ..., -2.8872e+00,
-2.0404e+00, -1.8429e+00]],
[[-2.1011e+00, -2.9102e+00, -2.8500e+00, ..., -3.1191e+00,
-2.5592e+00, -1.9719e+00],
[-2.1092e+00, -2.9297e+00, -2.5651e+00, ..., -2.8694e+00,
-2.2482e+00, -1.8338e+00],
[-2.6171e+00, -3.3197e+00, -3.0291e+00, ..., -3.3006e+00,
-2.5921e+00, -1.9898e+00],
...,
[-2.3108e+00, -2.8619e+00, -2.6472e+00, ..., -2.8244e+00,
-2.2615e+00, -1.7242e+00],
[-1.5267e+00, -1.9594e+00, -1.8978e+00, ..., -1.9334e+00,
-1.5968e+00, -1.2801e+00],
[-1.0109e+00, -1.2644e+00, -1.3098e+00, ..., -1.1777e+00,
-1.0151e+00, -7.5725e-01]]],
[[[ 4.2962e-02, 2.3023e-01, -1.4055e+00, ..., -6.3237e-01,
-2.1482e-01, -5.7507e-02],
[-2.0771e+00, -1.5233e+00, -4.4426e+00, ..., -3.8024e+00,
-1.5487e+00, -5.6143e-01],
[ 5.4672e-01, -5.3323e-02, -8.2942e-01, ..., -2.1064e-01,
-2.7159e-01, 2.4582e-01],
...,
[-8.5685e-01, -1.8124e+00, -1.8127e+00, ..., -1.7734e+00,
-1.3205e+00, -9.5068e-01],
[ 7.5486e-01, 5.8891e-03, -9.3060e-02, ..., 3.4191e-03,
7.0685e-01, 7.9667e-01],
[ 1.3915e-01, -5.6536e-01, -4.6545e-01, ..., -1.5559e-01,
4.4184e-01, 3.6079e-01]],
[[-3.2520e+00, -1.7557e+00, -6.2270e+00, ..., -7.1683e+00,
-6.2513e+00, -4.6098e+00],
[-8.0077e-01, 1.5247e+00, -4.4279e+00, ..., -5.5093e+00,
-5.5577e+00, -4.2731e+00],
[-3.8827e-01, 1.0359e+00, -7.4967e+00, ..., -8.0149e+00,
-7.8539e+00, -6.6762e+00],
...,
[ 2.9488e-01, 3.7239e-01, -4.0309e+00, ..., -4.6436e+00,
-4.2051e+00, -3.5153e+00],
[ 2.3138e+00, 2.2881e+00, -1.6209e-01, ..., -8.3706e-01,
-1.2150e+00, -1.2515e+00],
[ 1.3570e+00, 1.1883e+00, -1.7706e+00, ..., -2.2354e+00,
-2.5298e+00, -2.4270e+00]],
[[-1.3326e+00, -2.3531e+00, -2.3228e+00, ..., -1.7506e+00,
-1.2308e+00, -1.3762e+00],
[-1.6652e+00, -1.9633e+00, -1.5022e+00, ..., -1.2452e+00,
-7.2240e-01, -8.2917e-01],
[-2.1759e+00, -1.9309e+00, -1.0079e+00, ..., -1.1094e+00,
-4.6023e-01, -1.9156e-01],
...,
[-1.6143e-01, -6.2166e-01, -4.5795e-01, ..., -4.7655e-01,
-2.0509e-01, -4.5277e-01],
[-7.2114e-01, -2.0674e-01, -4.2049e-01, ..., -1.9810e-01,
2.4845e-01, 2.3769e-01],
[-8.1427e-01, -1.9838e-01, -3.3857e-01, ..., -3.8844e-01,
-1.8422e-01, -2.1780e-01]],
...,
[[-1.9985e+00, -1.0770e+00, -2.2696e+00, ..., -3.1926e+00,
-2.6309e+00, -1.8339e+00],
[-1.7715e+00, -2.8264e-02, -1.3795e+00, ..., -2.5152e+00,
-2.5331e+00, -1.9523e+00],
[-3.6551e-01, 2.3138e+00, 1.0091e+00, ..., 1.1496e-01,
2.5433e-01, 5.5306e-01],
...,
[-2.0723e+00, -8.1862e-02, -1.9944e-01, ..., -1.3586e+00,
-3.6754e-01, 1.1587e-01],
[-1.1920e+00, 7.9557e-01, 9.0155e-01, ..., 1.0369e-03,
3.2058e-01, 4.7478e-01],
[-2.0765e+00, -1.4134e+00, -1.6434e+00, ..., -2.4627e+00,
-1.4413e+00, -9.1920e-01]],
[[-2.4255e+00, -2.8715e+00, -4.2814e+00, ..., -2.4169e+00,
-1.9281e+00, -1.7357e+00],
[-1.7946e+00, -1.7228e+00, -4.9824e+00, ..., -2.9252e+00,
-2.7994e+00, -2.8056e+00],
[-2.8198e+00, -2.5233e+00, -6.7195e+00, ..., -4.4463e+00,
-4.3653e+00, -3.5123e+00],
...,
[-1.4006e+00, -4.4229e-02, -3.4149e+00, ..., -1.4460e+00,
-1.0476e+00, -9.4941e-01],
[-1.7023e+00, -6.4255e-01, -3.0466e+00, ..., -1.7353e+00,
-1.1883e+00, -8.3260e-01],
[-2.3183e+00, -1.4853e+00, -3.0862e+00, ..., -2.9653e+00,
-2.1551e+00, -1.9923e+00]],
[[-1.8144e+00, -2.5332e+00, -2.4573e+00, ..., -2.7078e+00,
-2.2280e+00, -1.7014e+00],
[-1.8203e+00, -2.5248e+00, -2.1724e+00, ..., -2.4221e+00,
-1.9195e+00, -1.5521e+00],
[-2.2794e+00, -2.8432e+00, -2.5481e+00, ..., -2.7717e+00,
-2.1640e+00, -1.6474e+00],
...,
[-2.3877e+00, -2.9556e+00, -2.7309e+00, ..., -2.8615e+00,
-2.2915e+00, -1.7547e+00],
[-1.5515e+00, -2.0167e+00, -1.9418e+00, ..., -1.9470e+00,
-1.6049e+00, -1.2921e+00],
[-1.0080e+00, -1.3075e+00, -1.3379e+00, ..., -1.1864e+00,
-1.0147e+00, -7.6686e-01]]],
[[[ 3.4300e-01, 8.0429e-01, 7.2976e-01, ..., 6.5011e-01,
9.4025e-01, 7.0810e-01],
[-1.1886e+00, -2.5151e-01, -1.2726e+00, ..., -1.3465e+00,
-8.7650e-02, 4.8029e-01],
[ 6.0506e-01, 4.3472e-01, 5.3370e-01, ..., 5.1330e-01,
9.0336e-01, 8.3835e-01],
...,
[-1.5785e-01, -1.1098e+00, -2.5777e+00, ..., -1.5160e+00,
-1.4265e+00, -8.8440e-01],
[ 1.4397e+00, 8.6035e-01, -8.7854e-01, ..., -2.8019e-01,
9.2326e-02, 3.5569e-01],
[ 8.6731e-01, 2.9520e-01, -6.3572e-01, ..., -1.6859e-01,
2.2681e-01, 3.7367e-01]],
[[-2.3082e+00, -1.3173e+00, -4.2545e+00, ..., -5.4235e+00,
-4.6417e+00, -3.6000e+00],
[-2.0833e-01, 1.4868e+00, -2.6462e+00, ..., -4.0580e+00,
-3.9418e+00, -3.2323e+00],
[ 5.9707e-01, 1.2920e+00, -5.1437e+00, ..., -5.9407e+00,
-5.7377e+00, -5.3599e+00],
...,
[-1.2835e+00, 1.1248e-01, -6.7432e+00, ..., -7.5934e+00,
-6.8793e+00, -5.4001e+00],
[ 1.7490e+00, 2.7411e+00, -1.1233e+00, ..., -2.5914e+00,
-2.9426e+00, -2.3663e+00],
[ 1.1737e+00, 1.7474e+00, -2.3179e+00, ..., -3.2425e+00,
-3.6147e+00, -3.1197e+00]],
[[-6.4301e-01, -1.7398e+00, -1.6236e+00, ..., -9.9801e-01,
-6.9765e-01, -8.3172e-01],
[-1.2892e+00, -1.7290e+00, -1.2055e+00, ..., -8.1261e-01,
-4.3551e-01, -3.1215e-01],
[-9.7717e-01, -9.1515e-01, -4.4792e-01, ..., -3.4984e-01,
5.5652e-02, 6.8747e-02],
...,
[-1.3678e+00, -1.7358e+00, -7.7014e-01, ..., -8.8668e-01,
-7.6409e-01, -1.0167e+00],
[-1.7074e+00, -1.1507e+00, -8.2499e-01, ..., -5.6849e-01,
-2.9640e-01, -3.6235e-01],
[-1.8554e+00, -1.2762e+00, -8.8622e-01, ..., -9.4196e-01,
-8.0035e-01, -7.8271e-01]],
...,
[[-2.1483e+00, -1.2679e+00, -2.2083e+00, ..., -3.1126e+00,
-2.4571e+00, -1.6231e+00],
[-6.2220e-01, 9.6536e-01, -4.2651e-01, ..., -1.3981e+00,
-1.4637e+00, -1.2811e+00],
[-4.0680e-01, 1.8842e+00, 7.5137e-01, ..., -1.5941e-01,
3.1104e-01, 4.2948e-01],
...,
[-1.8600e+00, 1.1476e+00, -6.1124e-02, ..., -1.0920e+00,
-5.0644e-01, 2.1788e-02],
[-8.0627e-01, 1.8640e+00, 1.5484e+00, ..., 4.5823e-01,
4.1362e-01, 6.8027e-01],
[-2.2032e+00, -1.1455e+00, -1.0227e+00, ..., -1.8736e+00,
-9.8508e-01, -3.2095e-01]],
[[-1.3598e+00, -1.2797e+00, -2.6456e+00, ..., -9.8365e-01,
-4.4407e-01, -3.4658e-01],
[-6.9928e-01, -2.2118e-01, -3.4693e+00, ..., -1.6283e+00,
-1.3676e+00, -1.6378e+00],
[-2.0933e+00, -1.4089e+00, -5.7535e+00, ..., -3.7100e+00,
-3.1904e+00, -2.5540e+00],
...,
[-2.4455e+00, -1.6317e+00, -5.4957e+00, ..., -3.0691e+00,
-2.7092e+00, -2.1605e+00],
[-2.1932e+00, -1.5142e+00, -4.2754e+00, ..., -2.6641e+00,
-2.2921e+00, -1.6227e+00],
[-2.3620e+00, -1.8012e+00, -3.3248e+00, ..., -2.9643e+00,
-2.4250e+00, -2.1104e+00]],
[[-1.1383e+00, -1.6898e+00, -1.5004e+00, ..., -1.7078e+00,
-1.4186e+00, -1.0649e+00],
[-1.1502e+00, -1.6106e+00, -1.1842e+00, ..., -1.3855e+00,
-1.1246e+00, -8.8728e-01],
[-1.5001e+00, -1.7876e+00, -1.3802e+00, ..., -1.5505e+00,
-1.1754e+00, -8.4959e-01],
...,
[-3.1640e+00, -3.9395e+00, -3.7381e+00, ..., -3.9399e+00,
-3.1661e+00, -2.4345e+00],
[-2.1048e+00, -2.7096e+00, -2.6816e+00, ..., -2.7672e+00,
-2.2852e+00, -1.8543e+00],
[-1.5317e+00, -1.8548e+00, -1.9166e+00, ..., -1.8621e+00,
-1.5160e+00, -1.1580e+00]]]], device='cuda:0'),
tensor([[[[ 5.1124e-02, 1.8306e-01, -1.5611e+00, ..., -7.3128e-01,
-3.3036e-01, -1.4183e-01],
[-2.1706e+00, -1.7204e+00, -4.7651e+00, ..., -3.9459e+00,
-1.6891e+00, -6.8162e-01],
[ 5.4051e-01, -1.4187e-01, -9.5584e-01, ..., -2.5127e-01,
-3.7291e-01, 1.3623e-01],
...,
[-7.7575e-01, -1.7589e+00, -1.7907e+00, ..., -6.3798e+00,
-6.2661e+00, -2.3749e+00],
[ 9.0239e-01, 1.1787e-01, 2.2212e-02, ..., -3.2387e+00,
-3.7253e+00, -1.1626e+00],
[ 2.5833e-01, -4.8579e-01, -3.7675e-01, ..., -2.1303e+00,
-2.6773e+00, -1.4576e+00]],
[[-3.3543e+00, -1.8145e+00, -6.3670e+00, ..., -7.2555e+00,
-6.3275e+00, -4.6963e+00],
[-8.0386e-01, 1.5814e+00, -4.4139e+00, ..., -5.5496e+00,
-5.5949e+00, -4.3194e+00],
[-3.7320e-01, 1.1207e+00, -7.5169e+00, ..., -8.0617e+00,
-7.8812e+00, -6.7114e+00],
...,
[ 4.6754e-01, 8.0299e-01, -3.9108e+00, ..., -8.1117e+00,
-5.7880e+00, -3.7507e+00],
[ 2.6983e+00, 2.8105e+00, 2.6236e-01, ..., -4.9715e+00,
-3.3303e+00, -1.0825e+00],
[ 1.6262e+00, 1.4909e+00, -1.6326e+00, ..., -6.1290e+00,
-4.8764e+00, -2.5519e+00]],
[[-1.4215e+00, -2.4362e+00, -2.3816e+00, ..., -1.8273e+00,
-1.3088e+00, -1.4282e+00],
[-1.7605e+00, -2.0409e+00, -1.5551e+00, ..., -1.2881e+00,
-8.0106e-01, -8.8991e-01],
[-2.2813e+00, -2.0144e+00, -1.0297e+00, ..., -1.1201e+00,
-5.2198e-01, -2.4080e-01],
...,
[-2.7272e-01, -8.1321e-01, -5.2269e-01, ..., -9.4481e-02,
-6.5668e-02, -6.5497e-01],
[-9.0101e-01, -3.5218e-01, -4.7531e-01, ..., 5.2337e-01,
3.4630e-01, 3.6176e-03],
[-1.0352e+00, -3.0750e-01, -3.2654e-01, ..., -2.3790e-01,
-1.5412e-01, -2.9053e-01]],
...,
[[-2.0619e+00, -1.1085e+00, -2.3736e+00, ..., -3.0598e+00,
-2.5678e+00, -1.8591e+00],
[-1.8479e+00, -4.1377e-02, -1.4355e+00, ..., -2.4055e+00,
-2.4984e+00, -1.9934e+00],
[-4.0512e-01, 2.3700e+00, 1.0928e+00, ..., 2.2516e-01,
3.6277e-01, 5.8301e-01],
...,
[-2.2109e+00, 2.1486e-01, -8.5568e-02, ..., -4.2322e+00,
-1.6460e-01, 4.7237e-01],
[-1.2262e+00, 1.0443e+00, 1.1093e+00, ..., -3.6467e+00,
-8.0315e-01, 6.5732e-01],
[-2.1744e+00, -1.4355e+00, -1.6842e+00, ..., -5.6204e+00,
-2.6933e+00, -5.2397e-01]],
[[-2.4767e+00, -2.9140e+00, -4.3072e+00, ..., -2.4503e+00,
-1.9572e+00, -1.7819e+00],
[-1.8179e+00, -1.7224e+00, -4.9716e+00, ..., -2.9441e+00,
-2.8126e+00, -2.8101e+00],
[-2.8717e+00, -2.5259e+00, -6.7304e+00, ..., -4.4538e+00,
-4.3917e+00, -3.5285e+00],
...,
[-1.2778e+00, 7.6563e-02, -3.6766e+00, ..., -2.7225e+00,
-2.2734e+00, -1.3734e+00],
[-1.5798e+00, -5.2033e-01, -3.2326e+00, ..., -3.0822e+00,
-2.5847e+00, -1.0093e+00],
[-2.2648e+00, -1.3986e+00, -3.1701e+00, ..., -3.5427e+00,
-2.4452e+00, -1.6089e+00]],
[[-1.8577e+00, -2.6007e+00, -2.5187e+00, ..., -2.7922e+00,
-2.2929e+00, -1.7503e+00],
[-1.8500e+00, -2.5817e+00, -2.2129e+00, ..., -2.5213e+00,
-2.0030e+00, -1.6037e+00],
[-2.3128e+00, -2.9004e+00, -2.5925e+00, ..., -2.8828e+00,
-2.2654e+00, -1.7036e+00],
...,
[-2.5170e+00, -3.0951e+00, -2.8076e+00, ..., -4.2375e+00,
-3.3012e+00, -2.0824e+00],
[-1.6574e+00, -2.1282e+00, -1.9984e+00, ..., -3.0512e+00,
-2.4428e+00, -1.5922e+00],
[-1.0898e+00, -1.3734e+00, -1.3703e+00, ..., -1.9183e+00,
-1.5563e+00, -1.0152e+00]]],
[[[ 4.2875e-01, 8.4414e-01, 4.9009e-01, ..., 4.3704e-01,
9.7246e-01, 6.3012e-01],
[-1.3314e+00, -4.5561e-01, -2.0208e+00, ..., -2.4863e+00,
-7.3309e-01, -6.4886e-02],
[ 8.8944e-01, 6.1718e-01, 2.9217e-01, ..., -2.6302e-01,
4.8022e-01, 6.3140e-01],
...,
[ 1.6560e-02, -5.9663e-01, -1.3596e+00, ..., -2.1361e+00,
-1.5552e+00, -1.0613e+00],
[ 1.3116e+00, 1.1430e+00, 6.0948e-01, ..., -2.8696e-01,
4.8922e-01, 6.9309e-01],
[ 9.7073e-01, 8.9474e-01, 6.0518e-01, ..., -2.9292e-01,
3.2222e-01, 2.9456e-01]],
[[-2.8043e+00, -1.5113e+00, -4.7867e+00, ..., -4.7205e+00,
-4.3317e+00, -3.3085e+00],
[-4.9398e-01, 1.7090e+00, -2.8830e+00, ..., -3.3465e+00,
-3.5076e+00, -2.8761e+00],
[ 3.2213e-01, 1.5045e+00, -5.7038e+00, ..., -5.2540e+00,
-5.1645e+00, -4.8395e+00],
...,
[-9.6408e-01, -4.7029e-02, -4.9017e+00, ..., -4.5689e+00,
-4.0891e+00, -3.4316e+00],
[ 1.6703e+00, 2.2532e+00, -3.2097e-01, ..., -5.8716e-01,
-9.5244e-01, -1.0735e+00],
[ 9.7386e-01, 1.2256e+00, -1.8086e+00, ..., -2.1530e+00,
-2.4415e+00, -2.3668e+00]],
[[-7.1545e-01, -1.9305e+00, -1.7635e+00, ..., -5.5308e-01,
-1.7275e-01, -4.3401e-01],
[-1.5054e+00, -1.9800e+00, -1.2768e+00, ..., -3.3135e-01,
6.5988e-02, 3.9235e-02],
[-1.4090e+00, -1.3972e+00, -7.4628e-01, ..., -2.7122e-01,
2.3481e-01, 2.7976e-01],
...,
[-8.7635e-01, -1.4167e+00, -6.8626e-01, ..., -5.4257e-01,
-2.0621e-01, -3.9170e-01],
[-1.1765e+00, -8.8205e-01, -5.9333e-01, ..., -3.2395e-01,
1.6183e-01, 2.3955e-01],
[-1.2465e+00, -7.3203e-01, -3.2365e-01, ..., -4.8594e-01,
-2.6724e-01, -2.3808e-01]],
...,
[[-2.3103e+00, -1.1493e+00, -2.5180e+00, ..., -3.4780e+00,
-2.6985e+00, -1.7563e+00],
[-6.0724e-01, 1.2853e+00, -7.4050e-01, ..., -2.5396e+00,
-2.3341e+00, -1.8087e+00],
[-1.5411e-01, 2.6716e+00, 8.7111e-01, ..., -5.7359e-01,
-8.0797e-02, 1.4238e-01],
...,
[-1.3786e+00, 1.2866e+00, -8.8621e-03, ..., -1.5988e+00,
-6.5897e-01, -7.0857e-02],
[-5.3654e-01, 1.4483e+00, 7.8090e-01, ..., -1.2317e-01,
1.6363e-01, 3.8191e-01],
[-1.6125e+00, -1.0998e+00, -1.5248e+00, ..., -2.5435e+00,
-1.5405e+00, -9.5958e-01]],
[[-1.4049e+00, -1.3920e+00, -3.0767e+00, ..., -1.1795e+00,
-7.6739e-01, -7.7305e-01],
[-6.4764e-01, -1.8282e-01, -3.9432e+00, ..., -1.3691e+00,
-1.2209e+00, -1.5674e+00],
[-2.1645e+00, -1.5002e+00, -6.4353e+00, ..., -3.2423e+00,
-2.7489e+00, -2.1996e+00],
...,
[-1.8172e+00, -9.8945e-01, -4.6609e+00, ..., -1.3846e+00,
-9.4392e-01, -8.0928e-01],
[-1.7189e+00, -1.0058e+00, -3.7346e+00, ..., -1.6907e+00,
-1.0865e+00, -6.7736e-01],
[-2.0412e+00, -1.4305e+00, -3.0110e+00, ..., -2.8787e+00,
-2.0388e+00, -1.8486e+00]],
[[-1.2706e+00, -1.9303e+00, -1.7523e+00, ..., -1.7730e+00,
-1.4759e+00, -1.1243e+00],
[-1.2663e+00, -1.8398e+00, -1.4042e+00, ..., -1.4041e+00,
-1.1131e+00, -8.8870e-01],
[-1.6399e+00, -2.0321e+00, -1.6345e+00, ..., -1.6152e+00,
-1.2110e+00, -9.0188e-01],
...,
[-2.3462e+00, -2.9717e+00, -2.8692e+00, ..., -2.9646e+00,
-2.3647e+00, -1.8159e+00],
[-1.5829e+00, -2.0398e+00, -2.0512e+00, ..., -2.0327e+00,
-1.6710e+00, -1.3501e+00],
[-1.0899e+00, -1.3373e+00, -1.4470e+00, ..., -1.2374e+00,
-1.0520e+00, -8.0388e-01]]],
[[[ 2.9676e-01, 5.6406e-01, 6.9337e-01, ..., 4.6878e-01,
1.0004e+00, 6.0547e-01],
[-2.1600e+00, -1.4986e+00, -3.1689e+00, ..., -2.6679e+00,
-8.8022e-01, -1.9020e-01],
[-1.7117e-01, -1.0028e+00, -2.0673e-01, ..., -1.7586e-01,
5.1917e-01, 6.4760e-01],
...,
[-1.9534e-01, -1.1448e+00, -2.6648e+00, ..., -1.5418e+00,
-1.4221e+00, -8.6484e-01],
[ 1.4066e+00, 8.2545e-01, -9.7130e-01, ..., -2.9269e-01,
9.1584e-02, 3.5859e-01],
[ 8.4019e-01, 2.9381e-01, -7.0620e-01, ..., -1.8618e-01,
2.1411e-01, 3.7839e-01]],
[[-1.6583e+00, -1.1111e+00, -4.1581e+00, ..., -4.8522e+00,
-4.4096e+00, -3.3905e+00],
[ 6.0479e-01, 1.9742e+00, -2.2456e+00, ..., -3.4992e+00,
-3.6021e+00, -2.9848e+00],
[ 1.3544e+00, 1.7322e+00, -4.8920e+00, ..., -5.4252e+00,
-5.2612e+00, -4.9345e+00],
...,
[-1.3563e+00, 6.0052e-02, -6.7269e+00, ..., -7.6222e+00,
-6.9285e+00, -5.4239e+00],
[ 1.6948e+00, 2.7452e+00, -1.0714e+00, ..., -2.5794e+00,
-2.9476e+00, -2.3639e+00],
[ 1.1366e+00, 1.7696e+00, -2.2839e+00, ..., -3.2466e+00,
-3.6281e+00, -3.1374e+00]],
[[-1.0560e-01, -1.1427e+00, -1.3087e+00, ..., -4.7426e-01,
-1.4523e-01, -4.7218e-01],
[-5.9770e-01, -8.6816e-01, -7.7250e-01, ..., -2.7279e-01,
3.6515e-02, -7.6870e-02],
[-5.5745e-01, -3.6850e-01, -3.9736e-01, ..., -3.4943e-01,
7.3146e-02, 7.1271e-02],
...,
[-1.3393e+00, -1.7049e+00, -7.6292e-01, ..., -8.8819e-01,
-7.7712e-01, -1.0238e+00],
[-1.6727e+00, -1.1084e+00, -8.1760e-01, ..., -5.7401e-01,
-3.0141e-01, -3.6155e-01],
[-1.8203e+00, -1.2393e+00, -8.8354e-01, ..., -9.6634e-01,
-8.3260e-01, -8.0769e-01]],
...,
[[-3.0422e+00, -2.4949e+00, -2.8182e+00, ..., -3.4234e+00,
-2.6147e+00, -1.7460e+00],
[-2.3041e+00, -8.2696e-01, -1.6207e+00, ..., -2.6127e+00,
-2.3452e+00, -1.8495e+00],
[-1.0575e+00, 1.2147e+00, 6.0029e-01, ..., -4.4655e-01,
-1.1496e-02, 1.3364e-01],
...,
[-1.8439e+00, 1.2002e+00, -2.2776e-03, ..., -1.1125e+00,
-5.4738e-01, -2.1573e-04],
[-8.2893e-01, 1.8524e+00, 1.6275e+00, ..., 4.9408e-01,
4.1604e-01, 6.8014e-01],
[-2.2596e+00, -1.2392e+00, -1.0410e+00, ..., -1.8679e+00,
-9.8185e-01, -3.1160e-01]],
[[-1.6002e+00, -1.5768e+00, -2.9421e+00, ..., -1.2773e+00,
-8.3745e-01, -8.5399e-01],
[-6.0053e-01, 9.5554e-02, -3.2541e+00, ..., -1.3577e+00,
-1.2138e+00, -1.5665e+00],
[-2.0872e+00, -8.7289e-01, -5.3730e+00, ..., -3.1540e+00,
-2.6887e+00, -2.1877e+00],
...,
[-2.4623e+00, -1.5894e+00, -5.3949e+00, ..., -3.0581e+00,
-2.7146e+00, -2.1639e+00],
[-2.2108e+00, -1.4781e+00, -4.2140e+00, ..., -2.6722e+00,
-2.2885e+00, -1.6301e+00],
[-2.3776e+00, -1.7762e+00, -3.3126e+00, ..., -2.9743e+00,
-2.4294e+00, -2.1174e+00]],
[[-1.3548e+00, -1.9187e+00, -1.7073e+00, ..., -1.7740e+00,
-1.4610e+00, -1.1111e+00],
[-1.2792e+00, -1.7738e+00, -1.2375e+00, ..., -1.3841e+00,
-1.0857e+00, -8.6585e-01],
[-1.6075e+00, -1.9610e+00, -1.4844e+00, ..., -1.6159e+00,
-1.2028e+00, -8.9130e-01],
...,
[-3.1657e+00, -3.9655e+00, -3.7714e+00, ..., -3.9504e+00,
-3.1697e+00, -2.4451e+00],
[-2.1010e+00, -2.7170e+00, -2.7005e+00, ..., -2.7709e+00,
-2.2876e+00, -1.8616e+00],
[-1.5260e+00, -1.8585e+00, -1.9248e+00, ..., -1.8688e+00,
-1.5161e+00, -1.1611e+00]]],
[[[ 2.6248e-01, 5.3874e-01, 6.4410e-01, ..., -3.8793e-01,
-1.3645e-01, -1.7493e-02],
[-2.0691e+00, -1.3591e+00, -2.9302e+00, ..., -3.3184e+00,
-1.3590e+00, -4.7906e-01],
[-4.1344e-02, -8.2300e-01, 3.0287e-02, ..., -2.1559e-01,
-4.2323e-01, 1.6080e-01],
...,
[-9.1566e-01, -2.0660e+00, -1.9262e+00, ..., -1.9716e+00,
-1.8598e+00, -1.2423e+00],
[ 7.5872e-01, -9.6517e-02, -7.1237e-02, ..., -4.9607e-02,
2.5221e-01, 4.3142e-01],
[ 1.8368e-01, -5.4646e-01, -3.5846e-01, ..., 1.3610e-01,
4.2615e-01, 4.7513e-01]],
[[-1.7860e+00, -1.3046e+00, -4.2800e+00, ..., -6.9249e+00,
-6.1339e+00, -4.5768e+00],
[ 5.7367e-01, 1.8368e+00, -2.3177e+00, ..., -5.5407e+00,
-5.6342e+00, -4.3567e+00],
[ 1.4293e+00, 1.6454e+00, -4.9774e+00, ..., -7.9641e+00,
-7.9035e+00, -6.6645e+00],
...,
[ 2.7722e-01, 2.0550e-01, -4.2692e+00, ..., -7.1912e+00,
-6.7852e+00, -5.2608e+00],
[ 2.2984e+00, 2.2026e+00, -2.6232e-01, ..., -1.9782e+00,
-2.6286e+00, -2.1041e+00],
[ 1.3536e+00, 1.1445e+00, -1.8202e+00, ..., -2.9615e+00,
-3.5283e+00, -3.0843e+00]],
[[-2.1586e-01, -1.3441e+00, -1.5547e+00, ..., -1.4403e+00,
-1.0819e+00, -1.3685e+00],
[-6.7007e-01, -1.0293e+00, -1.0023e+00, ..., -1.0120e+00,
-6.4086e-01, -9.3776e-01],
[-4.9336e-01, -3.6805e-01, -4.5799e-01, ..., -1.1366e+00,
-6.1925e-01, -4.8531e-01],
...,
[-2.5601e-02, -5.4991e-01, -5.7285e-01, ..., -1.0328e+00,
-8.6682e-01, -1.2671e+00],
[-6.5291e-01, -2.0128e-01, -5.4787e-01, ..., -5.6187e-01,
-2.4260e-01, -4.2484e-01],
[-8.0391e-01, -2.4881e-01, -4.5469e-01, ..., -5.3656e-01,
-4.8464e-01, -6.5826e-01]],
...,
[[-3.0293e+00, -2.4815e+00, -2.6395e+00, ..., -3.0644e+00,
-2.4991e+00, -1.8229e+00],
[-2.1091e+00, -6.2181e-01, -1.0780e+00, ..., -2.4827e+00,
-2.5408e+00, -2.0441e+00],
[-1.1398e+00, 1.0775e+00, 8.4075e-01, ..., -1.0431e-01,
3.3447e-03, 2.7167e-01],
...,
[-2.1485e+00, -1.8402e-01, -1.1558e-01, ..., -8.9166e-01,
-2.7128e-01, 2.9551e-01],
[-1.1815e+00, 7.8948e-01, 9.8713e-01, ..., 4.9679e-01,
5.0626e-01, 8.4641e-01],
[-2.0468e+00, -1.3729e+00, -1.5507e+00, ..., -2.1816e+00,
-1.1543e+00, -3.2092e-01]],
[[-1.6096e+00, -1.5608e+00, -2.8772e+00, ..., -2.3098e+00,
-1.9975e+00, -1.8782e+00],
[-5.9056e-01, 1.2412e-01, -3.2673e+00, ..., -2.9224e+00,
-2.9459e+00, -2.9476e+00],
[-2.0642e+00, -8.4428e-01, -5.4231e+00, ..., -4.4794e+00,
-4.5724e+00, -3.7609e+00],
...,
[-1.4019e+00, -2.6008e-04, -3.4402e+00, ..., -2.8122e+00,
-2.7763e+00, -2.3628e+00],
[-1.6806e+00, -5.6982e-01, -3.0193e+00, ..., -2.4243e+00,
-2.2841e+00, -1.7152e+00],
[-2.2569e+00, -1.3638e+00, -2.9978e+00, ..., -2.8506e+00,
-2.4507e+00, -2.2318e+00]],
[[-1.3435e+00, -1.8991e+00, -1.6786e+00, ..., -2.6095e+00,
-2.1765e+00, -1.6935e+00],
[-1.2741e+00, -1.7586e+00, -1.1970e+00, ..., -2.4476e+00,
-1.9676e+00, -1.6128e+00],
[-1.5963e+00, -1.9305e+00, -1.4170e+00, ..., -2.8727e+00,
-2.2909e+00, -1.7812e+00],
...,
[-2.3966e+00, -2.9816e+00, -2.7406e+00, ..., -4.0770e+00,
-3.3122e+00, -2.5960e+00],
[-1.5666e+00, -2.0564e+00, -1.9837e+00, ..., -2.8214e+00,
-2.3670e+00, -1.9564e+00],
[-1.0346e+00, -1.3464e+00, -1.3808e+00, ..., -1.8745e+00,
-1.5586e+00, -1.2078e+00]]],
[[[ 2.2028e-01, 4.8628e-01, 6.2512e-01, ..., -3.2302e-01,
-6.3098e-02, 1.9219e-02],
[-2.0413e+00, -1.3457e+00, -2.8756e+00, ..., -3.3989e+00,
-1.3750e+00, -4.7189e-01],
[-1.5666e-01, -8.9677e-01, 1.6801e-02, ..., -1.5604e-01,
-3.4024e-01, 1.4459e-01],
...,
[-9.5174e-01, -2.1618e+00, -2.0706e+00, ..., -1.7143e+00,
-1.7237e+00, -1.1102e+00],
[ 8.1832e-01, -1.0634e-01, -1.1928e-01, ..., 4.6686e-02,
2.8953e-01, 4.8204e-01],
[ 2.0628e-01, -5.8392e-01, -3.9766e-01, ..., 1.9754e-01,
4.3770e-01, 5.0755e-01]],
[[-1.6833e+00, -1.3891e+00, -4.2709e+00, ..., -7.0522e+00,
-6.1944e+00, -4.6278e+00],
[ 4.9677e-01, 1.4724e+00, -2.5218e+00, ..., -5.4195e+00,
-5.5427e+00, -4.2916e+00],
[ 1.2365e+00, 1.1898e+00, -5.0651e+00, ..., -7.8988e+00,
-7.8792e+00, -6.7121e+00],
...,
[ 4.0545e-01, 3.5446e-01, -4.1435e+00, ..., -7.0972e+00,
-6.6448e+00, -5.2351e+00],
[ 2.5074e+00, 2.4466e+00, 5.6292e-02, ..., -2.2751e+00,
-2.8114e+00, -2.2928e+00],
[ 1.4968e+00, 1.2999e+00, -1.6596e+00, ..., -3.0258e+00,
-3.5134e+00, -3.0914e+00]],
[[-1.5130e-01, -1.1827e+00, -1.4102e+00, ..., -1.4711e+00,
-1.0977e+00, -1.4028e+00],
[-5.1833e-01, -8.2739e-01, -8.8260e-01, ..., -1.0426e+00,
-6.4267e-01, -9.1326e-01],
[-3.3015e-01, -2.1041e-01, -4.3379e-01, ..., -1.0892e+00,
-5.5866e-01, -4.5218e-01],
...,
[-5.0372e-02, -5.8981e-01, -5.8380e-01, ..., -9.5548e-01,
-8.4425e-01, -1.1926e+00],
[-6.9540e-01, -2.1596e-01, -5.6402e-01, ..., -5.1764e-01,
-2.7095e-01, -4.6211e-01],
[-8.6518e-01, -2.5142e-01, -4.5748e-01, ..., -4.8431e-01,
-4.4468e-01, -6.1148e-01]],
...,
[[-2.8755e+00, -2.4915e+00, -2.4697e+00, ..., -3.3802e+00,
-2.7267e+00, -1.9692e+00],
[-2.1275e+00, -8.8441e-01, -1.1416e+00, ..., -2.5474e+00,
-2.6364e+00, -2.1254e+00],
[-1.1506e+00, 7.5046e-01, 7.6882e-01, ..., -1.1027e-01,
3.7953e-02, 3.1500e-01],
...,
[-2.2718e+00, -1.8914e-01, -1.3945e-01, ..., -8.6976e-01,
-2.6149e-01, 2.2632e-01],
[-1.2886e+00, 8.2276e-01, 1.0537e+00, ..., 3.4588e-01,
4.2765e-01, 7.5871e-01],
[-2.1678e+00, -1.5079e+00, -1.6956e+00, ..., -2.0003e+00,
-1.0435e+00, -2.6134e-01]],
[[-1.6432e+00, -1.6420e+00, -2.7939e+00, ..., -2.0926e+00,
-1.7746e+00, -1.7155e+00],
[-7.0227e-01, -9.7225e-02, -3.1383e+00, ..., -2.7377e+00,
-2.7853e+00, -2.9002e+00],
[-2.0844e+00, -9.9468e-01, -5.1207e+00, ..., -4.4852e+00,
-4.5583e+00, -3.7797e+00],
...,
[-1.3891e+00, 1.4754e-01, -3.4285e+00, ..., -2.9527e+00,
-2.8692e+00, -2.4498e+00],
[-1.6907e+00, -4.8057e-01, -3.0627e+00, ..., -2.4734e+00,
-2.3440e+00, -1.8182e+00],
[-2.3095e+00, -1.3574e+00, -3.0525e+00, ..., -2.7427e+00,
-2.4146e+00, -2.2204e+00]],
[[-1.3046e+00, -1.8388e+00, -1.6566e+00, ..., -2.6342e+00,
-2.2053e+00, -1.7023e+00],
[-1.2397e+00, -1.7110e+00, -1.2268e+00, ..., -2.3983e+00,
-1.9403e+00, -1.5853e+00],
[-1.5575e+00, -1.8973e+00, -1.4803e+00, ..., -2.7882e+00,
-2.2309e+00, -1.7156e+00],
...,
[-2.4462e+00, -3.0619e+00, -2.7904e+00, ..., -3.9084e+00,
-3.1774e+00, -2.4813e+00],
[-1.5958e+00, -2.0985e+00, -1.9992e+00, ..., -2.7245e+00,
-2.2836e+00, -1.8823e+00],
[-1.0506e+00, -1.3599e+00, -1.3827e+00, ..., -1.8342e+00,
-1.5166e+00, -1.1790e+00]]]], device='cuda:0'),
tensor([[[[ 3.4528e-01, 8.1959e-01, 7.4232e-01, ..., 7.0326e-01,
9.4763e-01, 7.3937e-01],
[-1.2385e+00, -2.7892e-01, -1.3359e+00, ..., -1.0642e+00,
6.2555e-02, 5.6581e-01],
[ 6.2639e-01, 4.1033e-01, 5.2800e-01, ..., 5.9662e-01,
9.5330e-01, 8.9736e-01],
...,
[-1.8971e-01, -1.1397e+00, -2.6539e+00, ..., -1.5425e+00,
-1.4217e+00, -8.6621e-01],
[ 1.4535e+00, 8.8510e-01, -9.1352e-01, ..., -3.1838e-01,
9.3079e-02, 3.7646e-01],
[ 8.5484e-01, 3.2640e-01, -6.5091e-01, ..., -1.9387e-01,
2.2129e-01, 3.8305e-01]],
[[-2.3147e+00, -1.1868e+00, -4.2089e+00, ..., -5.2335e+00,
-4.4987e+00, -3.5367e+00],
[-1.0993e-01, 1.7614e+00, -2.5070e+00, ..., -4.0731e+00,
-3.9836e+00, -3.3098e+00],
[ 7.7087e-01, 1.6069e+00, -5.0767e+00, ..., -5.9395e+00,
-5.7572e+00, -5.3430e+00],
...,
[-1.3086e+00, 1.3746e-01, -6.7175e+00, ..., -7.6056e+00,
-6.9032e+00, -5.4388e+00],
[ 1.7297e+00, 2.7710e+00, -1.1293e+00, ..., -2.5678e+00,
-2.9372e+00, -2.3898e+00],
[ 1.1756e+00, 1.7728e+00, -2.3310e+00, ..., -3.2444e+00,
-3.6169e+00, -3.1509e+00]],
[[-6.5408e-01, -1.8126e+00, -1.7026e+00, ..., -9.2488e-01,
-6.5752e-01, -7.7066e-01],
[-1.3669e+00, -1.8301e+00, -1.2569e+00, ..., -7.2228e-01,
-3.9523e-01, -2.7725e-01],
[-1.0754e+00, -9.9546e-01, -4.5665e-01, ..., -3.2400e-01,
1.6092e-02, 4.6041e-02],
...,
[-1.3581e+00, -1.7359e+00, -7.6583e-01, ..., -8.8835e-01,
-7.7411e-01, -1.0274e+00],
[-1.7065e+00, -1.1675e+00, -8.5928e-01, ..., -5.6944e-01,
-3.0421e-01, -3.6883e-01],
[-1.8694e+00, -1.2718e+00, -8.9432e-01, ..., -9.5522e-01,
-8.2705e-01, -8.0487e-01]],
...,
[[-2.2424e+00, -1.2334e+00, -2.2702e+00, ..., -2.8909e+00,
-2.2851e+00, -1.4972e+00],
[-6.3598e-01, 1.1319e+00, -3.8476e-01, ..., -1.3517e+00,
-1.4245e+00, -1.2446e+00],
[-3.5446e-01, 2.1510e+00, 8.7954e-01, ..., -1.7167e-01,
2.1691e-01, 3.3694e-01],
...,
[-1.8114e+00, 1.2674e+00, 1.0486e-02, ..., -1.0804e+00,
-5.4304e-01, -2.6166e-03],
[-8.3359e-01, 1.8920e+00, 1.6335e+00, ..., 4.9776e-01,
4.1653e-01, 6.7668e-01],
[-2.1832e+00, -1.1312e+00, -9.6678e-01, ..., -1.8610e+00,
-9.6740e-01, -3.0071e-01]],
[[-1.3487e+00, -1.2371e+00, -2.7781e+00, ..., -1.1042e+00,
-6.0115e-01, -4.8048e-01],
[-6.1928e-01, -9.8063e-02, -3.6147e+00, ..., -1.7604e+00,
-1.5229e+00, -1.7030e+00],
[-2.0390e+00, -1.2820e+00, -5.9515e+00, ..., -3.7105e+00,
-3.2318e+00, -2.5643e+00],
...,
[-2.5017e+00, -1.6314e+00, -5.5393e+00, ..., -3.0238e+00,
-2.7181e+00, -2.1813e+00],
[-2.2062e+00, -1.5078e+00, -4.3199e+00, ..., -2.6438e+00,
-2.3027e+00, -1.6394e+00],
[-2.3544e+00, -1.7969e+00, -3.3438e+00, ..., -2.9578e+00,
-2.4349e+00, -2.1248e+00]],
[[-1.1751e+00, -1.7295e+00, -1.5025e+00, ..., -1.6445e+00,
-1.3648e+00, -1.0220e+00],
[-1.1866e+00, -1.6588e+00, -1.1743e+00, ..., -1.3731e+00,
-1.1176e+00, -8.8178e-01],
[-1.5474e+00, -1.8395e+00, -1.3613e+00, ..., -1.5685e+00,
-1.1923e+00, -8.7372e-01],
...,
[-3.1641e+00, -3.9565e+00, -3.7528e+00, ..., -3.9632e+00,
-3.1735e+00, -2.4426e+00],
[-2.1230e+00, -2.7346e+00, -2.7075e+00, ..., -2.7760e+00,
-2.2910e+00, -1.8602e+00],
[-1.5540e+00, -1.8800e+00, -1.9389e+00, ..., -1.8684e+00,
-1.5185e+00, -1.1595e+00]]],
[[[ 4.7550e-01, 9.2629e-01, 6.5834e-01, ..., 7.8057e-01,
1.0579e+00, 8.2547e-01],
[-1.0516e+00, -1.4976e-01, -1.5310e+00, ..., -1.2418e+00,
4.0390e-02, 6.1156e-01],
[ 7.4515e-01, 5.1934e-01, 1.6081e-01, ..., 5.0851e-01,
8.7252e-01, 8.8459e-01],
...,
[-8.1672e-01, -1.7568e+00, -1.7901e+00, ..., -1.8006e+00,
-1.3049e+00, -9.3301e-01],
[ 7.7525e-01, 6.4200e-02, -5.3879e-02, ..., -5.7314e-03,
7.0926e-01, 8.1599e-01],
[ 1.7928e-01, -5.0466e-01, -4.1544e-01, ..., -1.1077e-01,
4.6204e-01, 3.7670e-01]],
[[-2.6469e+00, -1.4479e+00, -4.8327e+00, ..., -5.9139e+00,
-5.0508e+00, -3.9304e+00],
[-6.5834e-01, 1.3617e+00, -3.3871e+00, ..., -4.7526e+00,
-4.5565e+00, -3.6899e+00],
[ 2.6240e-02, 1.1671e+00, -6.1033e+00, ..., -6.8962e+00,
-6.5685e+00, -5.9693e+00],
...,
[ 2.1025e-01, 3.3959e-01, -4.0563e+00, ..., -4.6789e+00,
-4.2274e+00, -3.5211e+00],
[ 2.2552e+00, 2.2686e+00, -1.6046e-01, ..., -7.9241e-01,
-1.1602e+00, -1.2276e+00],
[ 1.3186e+00, 1.1678e+00, -1.7672e+00, ..., -2.2510e+00,
-2.5035e+00, -2.4228e+00]],
[[-5.4584e-01, -1.5652e+00, -1.3933e+00, ..., -7.6127e-01,
-5.1849e-01, -7.5128e-01],
[-1.3210e+00, -1.7210e+00, -1.1277e+00, ..., -7.0219e-01,
-3.8356e-01, -3.3492e-01],
[-1.1940e+00, -1.1019e+00, -5.3072e-01, ..., -4.1646e-01,
-6.7656e-02, -7.5991e-02],
...,
[-1.4541e-01, -6.2543e-01, -4.7545e-01, ..., -5.0825e-01,
-1.9930e-01, -4.5984e-01],
[-6.8856e-01, -1.9832e-01, -4.1035e-01, ..., -2.1120e-01,
2.3549e-01, 2.3719e-01],
[-7.8958e-01, -1.8962e-01, -3.1755e-01, ..., -3.8600e-01,
-1.9510e-01, -2.3264e-01]],
...,
[[-2.0666e+00, -1.1707e+00, -2.5025e+00, ..., -3.3159e+00,
-2.7190e+00, -1.8470e+00],
[-5.1970e-01, 1.1272e+00, -6.9284e-01, ..., -1.6729e+00,
-1.8079e+00, -1.5792e+00],
[-2.4224e-01, 2.1754e+00, 5.0765e-01, ..., -4.3868e-01,
-7.2499e-02, 1.3370e-01],
...,
[-2.0770e+00, -4.7219e-02, -1.7605e-01, ..., -1.3496e+00,
-4.0342e-01, 1.0326e-01],
[-1.1959e+00, 8.1241e-01, 9.1905e-01, ..., 3.6184e-02,
3.2199e-01, 4.7786e-01],
[-2.1058e+00, -1.4229e+00, -1.6515e+00, ..., -2.4438e+00,
-1.4645e+00, -9.1428e-01]],
[[-1.4762e+00, -1.4376e+00, -2.8626e+00, ..., -1.1427e+00,
-6.0650e-01, -4.8107e-01],
[-8.9873e-01, -5.1077e-01, -3.8019e+00, ..., -1.9408e+00,
-1.6631e+00, -1.8575e+00],
[-2.4596e+00, -1.9496e+00, -6.3793e+00, ..., -4.2461e+00,
-3.6705e+00, -2.9038e+00],
...,
[-1.4147e+00, -8.5767e-02, -3.4179e+00, ..., -1.4477e+00,
-1.0512e+00, -9.4082e-01],
[-1.7091e+00, -6.4694e-01, -3.0589e+00, ..., -1.7407e+00,
-1.1769e+00, -8.0788e-01],
[-2.3149e+00, -1.4719e+00, -3.1128e+00, ..., -2.9730e+00,
-2.1400e+00, -1.9705e+00]],
[[-1.2412e+00, -1.8256e+00, -1.6457e+00, ..., -1.8849e+00,
-1.5689e+00, -1.1686e+00],
[-1.3009e+00, -1.8136e+00, -1.4120e+00, ..., -1.6530e+00,
-1.3492e+00, -1.0578e+00],
[-1.7206e+00, -2.0809e+00, -1.7066e+00, ..., -1.9465e+00,
-1.5076e+00, -1.1050e+00],
...,
[-2.3689e+00, -2.9449e+00, -2.7225e+00, ..., -2.8607e+00,
-2.2842e+00, -1.7454e+00],
[-1.5386e+00, -2.0089e+00, -1.9365e+00, ..., -1.9430e+00,
-1.5968e+00, -1.2847e+00],
[-1.0059e+00, -1.2962e+00, -1.3331e+00, ..., -1.1826e+00,
-1.0093e+00, -7.6089e-01]]],
[[[ 4.1267e-01, 8.3670e-01, 4.8316e-01, ..., 4.9170e-01,
1.0130e+00, 6.5058e-01],
[-1.3672e+00, -4.9194e-01, -2.0127e+00, ..., -2.4761e+00,
-7.0911e-01, -3.4475e-02],
[ 8.5798e-01, 5.8551e-01, 3.4726e-01, ..., -2.1335e-01,
5.2759e-01, 6.4929e-01],
...,
[ 2.3959e-02, -6.0555e-01, -1.3975e+00, ..., -1.9696e+00,
-1.3591e+00, -8.8045e-01],
[ 1.3283e+00, 1.1383e+00, 5.4126e-01, ..., -2.4143e-01,
5.5038e-01, 7.3247e-01],
[ 9.9865e-01, 8.8844e-01, 5.6600e-01, ..., -3.1230e-01,
3.0716e-01, 2.9908e-01]],
[[-2.7994e+00, -1.4996e+00, -4.8315e+00, ..., -4.8817e+00,
-4.4198e+00, -3.3689e+00],
[-4.8097e-01, 1.7052e+00, -2.9371e+00, ..., -3.4436e+00,
-3.5654e+00, -2.9072e+00],
[ 3.2050e-01, 1.4724e+00, -5.7815e+00, ..., -5.4088e+00,
-5.2740e+00, -4.9271e+00],
...,
[-9.4357e-01, 5.8162e-03, -4.8957e+00, ..., -4.6894e+00,
-4.1899e+00, -3.4893e+00],
[ 1.6862e+00, 2.3083e+00, -3.1371e-01, ..., -1.0052e+00,
-1.2773e+00, -1.2991e+00],
[ 9.7906e-01, 1.2657e+00, -1.8255e+00, ..., -2.2749e+00,
-2.4942e+00, -2.3759e+00]],
[[-7.0353e-01, -1.9222e+00, -1.7703e+00, ..., -6.0000e-01,
-2.1016e-01, -4.7139e-01],
[-1.4790e+00, -1.9503e+00, -1.2646e+00, ..., -3.6728e-01,
4.6549e-02, 2.1434e-02],
[-1.4196e+00, -1.4044e+00, -7.4900e-01, ..., -2.7650e-01,
2.4276e-01, 2.7408e-01],
...,
[-9.3188e-01, -1.4439e+00, -7.1802e-01, ..., -4.9286e-01,
-1.6552e-01, -3.4344e-01],
[-1.2121e+00, -8.9343e-01, -6.1194e-01, ..., -3.4602e-01,
1.3392e-01, 2.0890e-01],
[-1.2448e+00, -7.1425e-01, -3.2317e-01, ..., -4.9849e-01,
-2.6933e-01, -2.3538e-01]],
...,
[[-2.3020e+00, -1.1505e+00, -2.5554e+00, ..., -3.6472e+00,
-2.8279e+00, -1.8393e+00],
[-6.4201e-01, 1.2578e+00, -7.9647e-01, ..., -2.5964e+00,
-2.3923e+00, -1.8435e+00],
[-1.7058e-01, 2.6697e+00, 8.1795e-01, ..., -5.3896e-01,
-5.5944e-02, 1.7436e-01],
...,
[-1.3162e+00, 1.3850e+00, 7.2637e-02, ..., -1.5166e+00,
-6.7516e-01, -1.2177e-01],
[-4.6355e-01, 1.5407e+00, 8.7286e-01, ..., -1.2157e-01,
1.2705e-01, 3.3459e-01],
[-1.6142e+00, -1.0707e+00, -1.4876e+00, ..., -2.3077e+00,
-1.4149e+00, -8.8068e-01]],
[[-1.4259e+00, -1.4155e+00, -3.1381e+00, ..., -1.1182e+00,
-7.1244e-01, -7.2548e-01],
[-6.5525e-01, -1.9272e-01, -4.0169e+00, ..., -1.3705e+00,
-1.2197e+00, -1.5837e+00],
[-2.1532e+00, -1.4919e+00, -6.5041e+00, ..., -3.3059e+00,
-2.7996e+00, -2.2463e+00],
...,
[-1.7948e+00, -9.2652e-01, -4.6431e+00, ..., -1.5741e+00,
-1.1099e+00, -9.2011e-01],
[-1.7091e+00, -9.6011e-01, -3.7164e+00, ..., -1.7734e+00,
-1.1848e+00, -7.7887e-01],
[-2.0379e+00, -1.4040e+00, -3.0333e+00, ..., -2.7986e+00,
-2.0042e+00, -1.8160e+00]],
[[-1.2912e+00, -1.9552e+00, -1.7811e+00, ..., -1.7888e+00,
-1.4885e+00, -1.1305e+00],
[-1.2938e+00, -1.8711e+00, -1.4379e+00, ..., -1.3955e+00,
-1.1036e+00, -8.8341e-01],
[-1.6742e+00, -2.0663e+00, -1.6718e+00, ..., -1.5956e+00,
-1.1997e+00, -8.8826e-01],
...,
[-2.3785e+00, -3.0141e+00, -2.8936e+00, ..., -2.8686e+00,
-2.2908e+00, -1.7624e+00],
[-1.6014e+00, -2.0625e+00, -2.0679e+00, ..., -2.0052e+00,
-1.6402e+00, -1.3209e+00],
[-1.0984e+00, -1.3464e+00, -1.4539e+00, ..., -1.2578e+00,
-1.0543e+00, -8.0465e-01]]],
[[[ 3.1051e-01, 5.7668e-01, 7.0223e-01, ..., 5.1175e-01,
1.0389e+00, 6.2052e-01],
[-2.1286e+00, -1.4488e+00, -3.1305e+00, ..., -2.5871e+00,
-7.9338e-01, -1.4950e-01],
[-1.4454e-01, -9.5213e-01, -2.2867e-01, ..., -1.3411e-01,
5.8717e-01, 6.7647e-01],
...,
[-1.7919e-01, -1.2053e+00, -2.7268e+00, ..., -1.4928e+00,
-1.3709e+00, -7.8912e-01],
[ 1.4449e+00, 7.7001e-01, -1.0647e+00, ..., -3.0540e-01,
1.0984e-01, 3.8856e-01],
[ 8.4372e-01, 2.2035e-01, -7.8278e-01, ..., -2.1078e-01,
2.1167e-01, 3.8840e-01]],
[[-1.6923e+00, -1.1179e+00, -4.1458e+00, ..., -4.8909e+00,
-4.4545e+00, -3.3959e+00],
[ 5.7737e-01, 1.9670e+00, -2.2436e+00, ..., -3.5511e+00,
-3.6478e+00, -2.9758e+00],
[ 1.3201e+00, 1.7244e+00, -4.9030e+00, ..., -5.4943e+00,
-5.3247e+00, -4.9502e+00],
...,
[-1.3112e+00, 1.6967e-01, -6.7902e+00, ..., -7.6330e+00,
-7.0261e+00, -5.5146e+00],
[ 1.7778e+00, 2.8274e+00, -1.1127e+00, ..., -2.6408e+00,
-3.0894e+00, -2.4842e+00],
[ 1.2063e+00, 1.8183e+00, -2.3217e+00, ..., -3.3008e+00,
-3.7289e+00, -3.1961e+00]],
[[-1.3498e-01, -1.1777e+00, -1.3205e+00, ..., -4.8997e-01,
-1.7035e-01, -4.8846e-01],
[-6.2995e-01, -8.9581e-01, -7.8333e-01, ..., -2.7256e-01,
3.5344e-02, -7.8508e-02],
[-5.7698e-01, -3.7983e-01, -4.0286e-01, ..., -3.3882e-01,
8.9189e-02, 7.4642e-02],
...,
[-1.4113e+00, -1.7619e+00, -7.4619e-01, ..., -8.7411e-01,
-7.7692e-01, -1.0286e+00],
[-1.7552e+00, -1.1795e+00, -8.3189e-01, ..., -5.7849e-01,
-3.1510e-01, -3.6588e-01],
[-1.9107e+00, -1.3309e+00, -9.4612e-01, ..., -1.0138e+00,
-8.7882e-01, -8.4068e-01]],
...,
[[-3.0119e+00, -2.4545e+00, -2.7913e+00, ..., -3.4530e+00,
-2.6496e+00, -1.7659e+00],
[-2.2239e+00, -7.6084e-01, -1.5589e+00, ..., -2.6011e+00,
-2.3494e+00, -1.8328e+00],
[-1.0057e+00, 1.2652e+00, 6.5700e-01, ..., -4.5402e-01,
-5.1308e-02, 1.0787e-01],
...,
[-1.8486e+00, 1.3004e+00, 5.5176e-02, ..., -1.1320e+00,
-5.9508e-01, -2.1493e-02],
[-8.0215e-01, 1.9754e+00, 1.7162e+00, ..., 5.2703e-01,
4.0213e-01, 6.6795e-01],
[-2.2186e+00, -1.0963e+00, -9.0492e-01, ..., -1.7568e+00,
-9.0026e-01, -2.5909e-01]],
[[-1.5779e+00, -1.5672e+00, -2.8971e+00, ..., -1.2666e+00,
-8.2639e-01, -8.2493e-01],
[-5.8374e-01, 9.3241e-02, -3.2317e+00, ..., -1.3784e+00,
-1.2225e+00, -1.5760e+00],
[-2.0862e+00, -8.9830e-01, -5.3545e+00, ..., -3.1922e+00,
-2.7189e+00, -2.2300e+00],
...,
[-2.4830e+00, -1.5891e+00, -5.4594e+00, ..., -3.1019e+00,
-2.7889e+00, -2.2378e+00],
[-2.2175e+00, -1.4695e+00, -4.2466e+00, ..., -2.7096e+00,
-2.3728e+00, -1.6955e+00],
[-2.3848e+00, -1.7647e+00, -3.2879e+00, ..., -2.9792e+00,
-2.4836e+00, -2.1356e+00]],
[[-1.3437e+00, -1.9055e+00, -1.6964e+00, ..., -1.7659e+00,
-1.4598e+00, -1.1147e+00],
[-1.2719e+00, -1.7616e+00, -1.2284e+00, ..., -1.3800e+00,
-1.0841e+00, -8.6849e-01],
[-1.5983e+00, -1.9477e+00, -1.4760e+00, ..., -1.6027e+00,
-1.1996e+00, -8.8871e-01],
...,
[-3.2107e+00, -4.0190e+00, -3.8037e+00, ..., -3.9836e+00,
-3.1889e+00, -2.4634e+00],
[-2.1396e+00, -2.7712e+00, -2.7351e+00, ..., -2.8026e+00,
-2.3105e+00, -1.8811e+00],
[-1.5593e+00, -1.9006e+00, -1.9661e+00, ..., -1.9052e+00,
-1.5362e+00, -1.1750e+00]]],
[[[ 7.1537e-02, 2.1212e-01, -1.4420e+00, ..., -6.2628e-01,
-2.1282e-01, -8.5717e-02],
[-2.1050e+00, -1.6413e+00, -4.6023e+00, ..., -3.7632e+00,
-1.5466e+00, -6.0050e-01],
[ 5.5519e-01, -1.4812e-01, -9.1860e-01, ..., -2.2281e-01,
-3.4009e-01, 1.8302e-01],
...,
[-8.6402e-01, -1.9242e+00, -2.0158e+00, ..., -1.7106e+00,
-1.2552e+00, -8.3603e-01],
[ 8.6982e-01, 5.7112e-02, -1.0678e-01, ..., -3.6072e-03,
7.1936e-01, 8.1880e-01],
[ 2.2916e-01, -5.0966e-01, -4.5018e-01, ..., -1.5334e-01,
4.5412e-01, 3.6714e-01]],
[[-3.3353e+00, -1.8126e+00, -6.2832e+00, ..., -7.1871e+00,
-6.2602e+00, -4.6108e+00],
[-8.1208e-01, 1.5393e+00, -4.3538e+00, ..., -5.5697e+00,
-5.5789e+00, -4.2633e+00],
[-3.7805e-01, 1.0692e+00, -7.4494e+00, ..., -8.0595e+00,
-7.8131e+00, -6.6458e+00],
...,
[ 5.0378e-01, 7.1328e-01, -3.8522e+00, ..., -4.6649e+00,
-4.2275e+00, -3.5361e+00],
[ 2.7251e+00, 2.7769e+00, 3.8401e-01, ..., -9.2657e-01,
-1.2881e+00, -1.3147e+00],
[ 1.6269e+00, 1.4779e+00, -1.5387e+00, ..., -2.2958e+00,
-2.5596e+00, -2.4320e+00]],
[[-1.3824e+00, -2.3945e+00, -2.3546e+00, ..., -1.7399e+00,
-1.2025e+00, -1.3719e+00],
[-1.7326e+00, -2.0099e+00, -1.5404e+00, ..., -1.2358e+00,
-7.1041e-01, -8.4151e-01],
[-2.2210e+00, -1.9588e+00, -1.0246e+00, ..., -1.1125e+00,
-4.8182e-01, -2.0941e-01],
...,
[-2.9939e-01, -7.9049e-01, -5.0265e-01, ..., -4.8266e-01,
-1.9898e-01, -4.3390e-01],
[-8.8064e-01, -3.0728e-01, -4.5447e-01, ..., -1.9804e-01,
2.3591e-01, 2.2936e-01],
[-9.9395e-01, -3.0343e-01, -3.6747e-01, ..., -3.9973e-01,
-2.0074e-01, -2.3214e-01]],
...,
[[-2.0496e+00, -1.0935e+00, -2.3555e+00, ..., -3.0971e+00,
-2.5510e+00, -1.7884e+00],
[-1.7842e+00, 2.7265e-02, -1.3717e+00, ..., -2.5155e+00,
-2.5535e+00, -2.0011e+00],
[-3.6760e-01, 2.3815e+00, 1.0927e+00, ..., 1.0702e-01,
2.8049e-01, 5.2915e-01],
...,
[-2.2278e+00, 9.9453e-02, -1.2146e-01, ..., -1.3334e+00,
-4.1246e-01, 7.1454e-02],
[-1.2289e+00, 1.0116e+00, 1.0679e+00, ..., 3.7299e-02,
3.0414e-01, 4.5109e-01],
[-2.1856e+00, -1.4749e+00, -1.7320e+00, ..., -2.3454e+00,
-1.4123e+00, -8.9451e-01]],
[[-2.4352e+00, -2.8571e+00, -4.2247e+00, ..., -2.4685e+00,
-1.9664e+00, -1.7592e+00],
[-1.7686e+00, -1.6427e+00, -4.8876e+00, ..., -2.9608e+00,
-2.7738e+00, -2.7393e+00],
[-2.8501e+00, -2.4750e+00, -6.6761e+00, ..., -4.4432e+00,
-4.3224e+00, -3.4396e+00],
...,
[-1.3278e+00, 1.9144e-01, -3.5076e+00, ..., -1.4803e+00,
-1.0902e+00, -9.7896e-01],
[-1.6730e+00, -4.8576e-01, -3.1267e+00, ..., -1.7339e+00,
-1.2143e+00, -8.5563e-01],
[-2.3753e+00, -1.4459e+00, -3.1379e+00, ..., -2.9232e+00,
-2.1468e+00, -1.9703e+00]],
[[-1.8205e+00, -2.5628e+00, -2.4713e+00, ..., -2.7243e+00,
-2.2296e+00, -1.7076e+00],
[-1.8123e+00, -2.5338e+00, -2.1635e+00, ..., -2.4379e+00,
-1.9377e+00, -1.5627e+00],
[-2.2671e+00, -2.8406e+00, -2.5337e+00, ..., -2.8050e+00,
-2.1924e+00, -1.6574e+00],
...,
[-2.4901e+00, -3.0754e+00, -2.7874e+00, ..., -2.8174e+00,
-2.2413e+00, -1.7175e+00],
[-1.6059e+00, -2.0813e+00, -1.9617e+00, ..., -1.9174e+00,
-1.5732e+00, -1.2649e+00],
[-1.0359e+00, -1.3296e+00, -1.3381e+00, ..., -1.1788e+00,
-1.0004e+00, -7.5846e-01]]]], device='cuda:0'),
tensor([[[[ 2.9978e-01, 6.0707e-01, 6.5752e-01, ..., -1.8961e+01,
-1.0568e+01, -7.3771e+00],
[-1.5975e+00, -8.2719e-01, -2.1542e+00, ..., -2.4515e+01,
-1.1978e+01, -8.5301e+00],
[ 2.1567e-01, -2.8876e-01, 2.9539e-01, ..., -2.1282e+01,
-1.2339e+01, -8.8493e+00],
...,
[-6.3888e-01, -1.6268e+00, -1.7866e+00, ..., -1.7266e+00,
-1.1089e+00, -8.1723e-01],
[ 9.9854e-01, 3.2819e-01, 1.4914e-01, ..., -1.7773e-01,
5.8154e-01, 7.6502e-01],
[ 5.1669e-01, -1.9101e-02, 7.8380e-03, ..., 1.8968e-01,
7.3142e-01, 6.7672e-01]],
[[-1.9692e+00, -1.3759e+00, -4.3404e+00, ..., -1.6316e+01,
-1.4548e+01, -1.0801e+01],
[ 9.4438e-02, 1.4197e+00, -2.7152e+00, ..., -1.6957e+01,
-1.7842e+01, -1.3662e+01],
[ 8.2903e-01, 1.2330e+00, -5.2014e+00, ..., -1.7732e+01,
-1.7041e+01, -1.3298e+01],
...,
[-6.0701e-02, 2.9836e-01, -4.2234e+00, ..., -6.8927e+00,
-5.9964e+00, -4.6043e+00],
[ 2.3373e+00, 2.5705e+00, 2.2918e-01, ..., -2.1026e+00,
-2.0891e+00, -1.7653e+00],
[ 1.3871e+00, 1.3853e+00, -1.5563e+00, ..., -3.2635e+00,
-3.2142e+00, -2.9723e+00]],
[[-2.9873e-01, -1.3627e+00, -1.4678e+00, ..., -2.0675e+00,
-7.4389e-01, -1.1067e+00],
[-7.8614e-01, -1.1957e+00, -1.0143e+00, ..., -2.2246e-01,
1.6596e-01, -6.9144e-01],
[-5.7139e-01, -5.3964e-01, -4.7533e-01, ..., 2.1308e+00,
1.7792e+00, -4.9484e-02],
...,
[-3.7108e-01, -9.3360e-01, -6.4303e-01, ..., -9.0987e-01,
-6.1519e-01, -7.5560e-01],
[-8.9513e-01, -4.6511e-01, -5.5523e-01, ..., -3.9509e-01,
-5.1542e-02, -1.0165e-01],
[-1.0399e+00, -4.4587e-01, -4.1177e-01, ..., -3.9806e-01,
-2.3627e-01, -2.7338e-01]],
...,
[[-2.5705e+00, -1.9649e+00, -2.3576e+00, ..., -9.1359e+00,
-7.0731e+00, -5.3706e+00],
[-1.4995e+00, -4.2821e-02, -7.6855e-01, ..., -7.4429e+00,
-7.1898e+00, -5.8201e+00],
[-7.9914e-01, 1.3096e+00, 7.2066e-01, ..., -6.2089e+00,
-1.9796e+00, -1.2771e+00],
...,
[-1.9605e+00, 4.0645e-01, -4.4412e-03, ..., -7.1188e-01,
-2.8923e-01, 2.5193e-01],
[-9.3634e-01, 1.1597e+00, 1.0428e+00, ..., 5.3158e-01,
4.7137e-01, 7.0084e-01],
[-1.9565e+00, -1.3307e+00, -1.6161e+00, ..., -2.0379e+00,
-1.1747e+00, -4.1624e-01]],
[[-1.4963e+00, -1.4835e+00, -2.7517e+00, ..., -6.7608e-01,
7.0899e-01, 6.9732e-01],
[-6.8804e-01, -1.7949e-01, -3.3614e+00, ..., -6.8816e+00,
-6.9376e+00, -6.7397e+00],
[-2.0678e+00, -1.2291e+00, -5.5146e+00, ..., -1.0678e+01,
-1.2650e+01, -1.1707e+01],
...,
[-1.4227e+00, -1.1395e-01, -3.7873e+00, ..., -2.7050e+00,
-2.1207e+00, -1.6564e+00],
[-1.6243e+00, -5.4043e-01, -3.2379e+00, ..., -2.4229e+00,
-1.6985e+00, -1.1552e+00],
[-2.2077e+00, -1.3347e+00, -2.9990e+00, ..., -2.8915e+00,
-2.0488e+00, -1.8058e+00]],
[[-1.2144e+00, -1.7507e+00, -1.5742e+00, ..., -5.0989e+00,
-3.4284e+00, -2.7447e+00],
[-1.1946e+00, -1.6610e+00, -1.2061e+00, ..., -5.3348e+00,
-3.8997e+00, -2.7918e+00],
[-1.5287e+00, -1.8642e+00, -1.4493e+00, ..., -6.3890e+00,
-4.4705e+00, -3.1506e+00],
...,
[-2.3915e+00, -2.9905e+00, -2.7367e+00, ..., -3.7113e+00,
-2.8893e+00, -2.1287e+00],
[-1.5692e+00, -2.0510e+00, -1.9588e+00, ..., -2.6269e+00,
-2.1225e+00, -1.6468e+00],
[-1.0348e+00, -1.3125e+00, -1.3436e+00, ..., -1.6996e+00,
-1.3935e+00, -1.0053e+00]]],
[[[ 2.9067e-01, 5.3425e-01, 6.7468e-01, ..., 4.8043e-01,
1.0022e+00, 5.8940e-01],
[-2.1557e+00, -1.4925e+00, -3.1320e+00, ..., -2.6212e+00,
-8.5298e-01, -2.0665e-01],
[-2.1742e-01, -9.8914e-01, -1.5290e-01, ..., -9.4259e-02,
5.5840e-01, 6.3110e-01],
...,
[ 1.0981e-01, -5.9728e-01, -1.4159e+00, ..., -1.2322e+00,
-9.8805e-01, -7.6248e-01],
[ 1.4310e+00, 1.2022e+00, 6.1145e-01, ..., 7.3176e-01,
8.8248e-01, 9.1707e-01],
[ 1.0933e+00, 9.8104e-01, 7.0350e-01, ..., 9.4540e-01,
1.0306e+00, 8.6696e-01]],
[[-1.6207e+00, -1.1607e+00, -4.1978e+00, ..., -4.8932e+00,
-4.4327e+00, -3.3958e+00],
[ 5.6308e-01, 1.8458e+00, -2.3315e+00, ..., -3.5263e+00,
-3.6203e+00, -2.9671e+00],
[ 1.2829e+00, 1.5884e+00, -4.9161e+00, ..., -5.4665e+00,
-5.2727e+00, -4.9170e+00],
...,
[-8.7137e-01, 1.5626e-01, -4.9081e+00, ..., -5.8547e+00,
-5.1486e+00, -4.1815e+00],
[ 1.7666e+00, 2.4127e+00, -2.5073e-01, ..., -1.1648e+00,
-1.3995e+00, -1.4600e+00],
[ 1.0132e+00, 1.2818e+00, -1.8266e+00, ..., -2.5789e+00,
-2.7734e+00, -2.7122e+00]],
[[-7.4850e-02, -1.0898e+00, -1.2812e+00, ..., -4.9371e-01,
-1.7047e-01, -4.8152e-01],
[-5.1484e-01, -7.8825e-01, -7.4951e-01, ..., -2.7094e-01,
4.2851e-02, -6.8789e-02],
[-4.7971e-01, -3.0302e-01, -3.9239e-01, ..., -3.3065e-01,
9.7530e-02, 7.8778e-02],
...,
[-9.6594e-01, -1.4304e+00, -6.2939e-01, ..., -6.2129e-01,
-4.9058e-01, -6.7012e-01],
[-1.1590e+00, -7.6259e-01, -4.0438e-01, ..., -1.0323e-01,
1.2403e-01, 1.0692e-01],
[-1.1818e+00, -5.8397e-01, -1.0706e-01, ..., -4.8271e-02,
-1.9481e-02, -4.7308e-02]],
...,
[[-2.9900e+00, -2.5366e+00, -2.7799e+00, ..., -3.4426e+00,
-2.5511e+00, -1.6998e+00],
[-2.3061e+00, -9.6652e-01, -1.6390e+00, ..., -2.6487e+00,
-2.2822e+00, -1.7736e+00],
[-1.1150e+00, 9.8385e-01, 5.2377e-01, ..., -5.0522e-01,
3.4277e-03, 1.6628e-01],
...,
[-1.4846e+00, 1.4056e+00, 2.4398e-01, ..., -9.0015e-01,
-1.4138e-01, 4.4204e-01],
[-5.8781e-01, 1.5722e+00, 9.7683e-01, ..., 1.6757e-01,
3.7153e-01, 5.4899e-01],
[-1.6739e+00, -1.0289e+00, -1.4056e+00, ..., -2.3456e+00,
-1.2937e+00, -6.0216e-01]],
[[-1.6028e+00, -1.6426e+00, -2.9349e+00, ..., -1.3206e+00,
-8.6306e-01, -8.3173e-01],
[-6.4642e-01, -5.3154e-02, -3.2427e+00, ..., -1.4199e+00,
-1.2282e+00, -1.5579e+00],
[-2.1102e+00, -1.0156e+00, -5.3371e+00, ..., -3.2431e+00,
-2.6885e+00, -2.1903e+00],
...,
[-1.8209e+00, -9.0096e-01, -4.4978e+00, ..., -2.2782e+00,
-1.8951e+00, -1.5007e+00],
[-1.7257e+00, -9.5320e-01, -3.6384e+00, ..., -2.2000e+00,
-1.5756e+00, -9.9534e-01],
[-2.0534e+00, -1.4156e+00, -2.9924e+00, ..., -2.8252e+00,
-1.9676e+00, -1.7088e+00]],
[[-1.3434e+00, -1.8929e+00, -1.7243e+00, ..., -1.7759e+00,
-1.4841e+00, -1.1230e+00],
[-1.2727e+00, -1.7541e+00, -1.2666e+00, ..., -1.3905e+00,
-1.1039e+00, -8.8708e-01],
[-1.5974e+00, -1.9538e+00, -1.5185e+00, ..., -1.6249e+00,
-1.2230e+00, -9.1555e-01],
...,
[-2.3715e+00, -2.9716e+00, -2.7732e+00, ..., -3.0883e+00,
-2.4607e+00, -1.8764e+00],
[-1.5878e+00, -2.0246e+00, -1.9678e+00, ..., -2.1281e+00,
-1.7527e+00, -1.4054e+00],
[-1.0604e+00, -1.2805e+00, -1.3368e+00, ..., -1.2854e+00,
-1.1066e+00, -8.1551e-01]]],
[[[ 2.1708e-01, 4.8696e-01, 6.1926e-01, ..., -3.2927e-01,
-6.4317e-02, 2.8560e-02],
[-2.0379e+00, -1.3468e+00, -2.8868e+00, ..., -3.2074e+00,
-1.2449e+00, -4.1237e-01],
[-1.6065e-01, -9.2034e-01, 8.6655e-04, ..., -1.8761e-01,
-3.4513e-01, 1.6948e-01],
...,
[-9.6890e-01, -2.2015e+00, -2.0967e+00, ..., -1.9613e+00,
-1.9055e+00, -1.2783e+00],
[ 8.3829e-01, -9.9075e-02, -1.0691e-01, ..., 1.6418e-02,
2.6580e-01, 4.5563e-01],
[ 2.2193e-01, -5.8594e-01, -4.3611e-01, ..., 2.2345e-01,
4.7038e-01, 5.1036e-01]],
[[-1.6918e+00, -1.3952e+00, -4.2807e+00, ..., -6.8061e+00,
-6.0375e+00, -4.5229e+00],
[ 4.6958e-01, 1.4604e+00, -2.5512e+00, ..., -5.4067e+00,
-5.5527e+00, -4.3104e+00],
[ 1.1926e+00, 1.1811e+00, -5.0928e+00, ..., -7.8221e+00,
-7.7894e+00, -6.5973e+00],
...,
[ 4.0346e-01, 3.9142e-01, -4.1114e+00, ..., -7.0912e+00,
-6.6759e+00, -5.1872e+00],
[ 2.5659e+00, 2.5288e+00, 1.5162e-01, ..., -1.8960e+00,
-2.5372e+00, -2.0432e+00],
[ 1.5417e+00, 1.3445e+00, -1.6129e+00, ..., -2.9106e+00,
-3.4746e+00, -3.0478e+00]],
[[-1.3586e-01, -1.1614e+00, -1.4062e+00, ..., -1.3758e+00,
-1.0270e+00, -1.3231e+00],
[-5.0046e-01, -8.1429e-01, -8.8217e-01, ..., -9.8131e-01,
-5.9928e-01, -8.7983e-01],
[-3.1315e-01, -2.1111e-01, -4.4362e-01, ..., -1.0845e+00,
-5.8791e-01, -4.6784e-01],
...,
[-5.7071e-02, -6.1408e-01, -6.1303e-01, ..., -1.0485e+00,
-8.8732e-01, -1.2495e+00],
[-7.0394e-01, -2.1660e-01, -5.7434e-01, ..., -5.6781e-01,
-2.5278e-01, -4.1708e-01],
[-8.8363e-01, -2.6388e-01, -4.7390e-01, ..., -5.0030e-01,
-4.3872e-01, -5.9638e-01]],
...,
[[-2.8500e+00, -2.4791e+00, -2.4724e+00, ..., -3.0606e+00,
-2.5000e+00, -1.7956e+00],
[-2.1288e+00, -8.7847e-01, -1.1387e+00, ..., -2.3895e+00,
-2.5219e+00, -2.0393e+00],
[-1.1309e+00, 7.7073e-01, 7.5864e-01, ..., -8.3595e-02,
3.2571e-02, 2.6939e-01],
...,
[-2.3145e+00, -1.4576e-01, -1.1637e-01, ..., -9.2744e-01,
-2.4575e-01, 3.0676e-01],
[-1.3066e+00, 8.7520e-01, 1.0877e+00, ..., 4.3873e-01,
4.8790e-01, 8.1711e-01],
[-2.2329e+00, -1.5227e+00, -1.6639e+00, ..., -2.2234e+00,
-1.1803e+00, -3.6694e-01]],
[[-1.6446e+00, -1.6424e+00, -2.7750e+00, ..., -2.2212e+00,
-1.9186e+00, -1.8397e+00],
[-7.0191e-01, -9.9000e-02, -3.1133e+00, ..., -2.8529e+00,
-2.8635e+00, -2.8967e+00],
[-2.0782e+00, -9.8696e-01, -5.0945e+00, ..., -4.4275e+00,
-4.4903e+00, -3.6826e+00],
...,
[-1.3579e+00, 1.9290e-01, -3.4225e+00, ..., -2.8091e+00,
-2.7301e+00, -2.3045e+00],
[-1.6719e+00, -4.4776e-01, -3.0384e+00, ..., -2.4270e+00,
-2.2394e+00, -1.6543e+00],
[-2.3194e+00, -1.3304e+00, -3.0456e+00, ..., -2.8650e+00,
-2.4180e+00, -2.1905e+00]],
[[-1.3039e+00, -1.8359e+00, -1.6577e+00, ..., -2.5677e+00,
-2.1456e+00, -1.6698e+00],
[-1.2380e+00, -1.7059e+00, -1.2248e+00, ..., -2.3966e+00,
-1.9353e+00, -1.5935e+00],
[-1.5527e+00, -1.8921e+00, -1.4746e+00, ..., -2.8215e+00,
-2.2516e+00, -1.7526e+00],
...,
[-2.4635e+00, -3.0772e+00, -2.7950e+00, ..., -3.9985e+00,
-3.2421e+00, -2.5429e+00],
[-1.5973e+00, -2.1092e+00, -2.0063e+00, ..., -2.7626e+00,
-2.3143e+00, -1.9144e+00],
[-1.0499e+00, -1.3662e+00, -1.3825e+00, ..., -1.8260e+00,
-1.5204e+00, -1.1775e+00]]],
[[[ 3.4635e-01, 7.9711e-01, 7.1802e-01, ..., -3.8269e+00,
-2.5152e+00, -1.6237e-01],
[-1.2151e+00, -3.1609e-01, -1.3639e+00, ..., -4.5337e+00,
-2.7131e+00, -3.9036e-01],
[ 6.1712e-01, 4.1406e-01, 5.3350e-01, ..., -2.7097e+00,
-1.8154e+00, 6.1947e-02],
...,
[-1.9423e-01, -1.1531e+00, -2.6275e+00, ..., -1.5118e+00,
-1.4281e+00, -8.7216e-01],
[ 1.4251e+00, 8.2886e-01, -9.4372e-01, ..., -2.6903e-01,
9.4278e-02, 3.8483e-01],
[ 8.5801e-01, 3.0430e-01, -6.6825e-01, ..., -1.7444e-01,
2.1402e-01, 3.9359e-01]],
[[-2.3388e+00, -1.3190e+00, -4.2759e+00, ..., -8.6215e+00,
-7.1286e+00, -4.2525e+00],
[-2.0694e-01, 1.5277e+00, -2.6234e+00, ..., -7.1514e+00,
-5.7365e+00, -3.5653e+00],
[ 6.0586e-01, 1.3146e+00, -5.1397e+00, ..., -8.9083e+00,
-7.2317e+00, -5.4176e+00],
...,
[-1.2947e+00, 1.3802e-01, -6.6604e+00, ..., -7.5670e+00,
-6.8534e+00, -5.4123e+00],
[ 1.7968e+00, 2.8131e+00, -1.0333e+00, ..., -2.5334e+00,
-2.8985e+00, -2.3630e+00],
[ 1.2080e+00, 1.8043e+00, -2.2832e+00, ..., -3.2131e+00,
-3.6044e+00, -3.1406e+00]],
[[-6.5701e-01, -1.7543e+00, -1.6502e+00, ..., -1.4289e+00,
-1.1489e+00, -9.3187e-01],
[-1.3142e+00, -1.7539e+00, -1.2128e+00, ..., -9.7522e-01,
-7.0593e-01, -3.6116e-01],
[-9.9368e-01, -9.1826e-01, -4.3334e-01, ..., -1.0004e+00,
-5.2549e-01, 9.0652e-02],
...,
[-1.3572e+00, -1.7577e+00, -7.8021e-01, ..., -8.9616e-01,
-7.8619e-01, -1.0339e+00],
[-1.7164e+00, -1.1651e+00, -8.4531e-01, ..., -5.6204e-01,
-3.0099e-01, -3.6368e-01],
[-1.8714e+00, -1.2785e+00, -8.9918e-01, ..., -9.3005e-01,
-8.0988e-01, -7.8958e-01]],
...,
[[-2.1982e+00, -1.2982e+00, -2.3020e+00, ..., -3.0429e+00,
-1.8967e+00, -1.5542e+00],
[-6.3139e-01, 1.0054e+00, -4.3765e-01, ..., -1.8854e+00,
-1.3867e+00, -1.3332e+00],
[-4.1738e-01, 1.8993e+00, 7.2855e-01, ..., -1.2757e+00,
3.9881e-02, 2.6545e-01],
...,
[-1.8256e+00, 1.2700e+00, 6.5144e-03, ..., -1.0566e+00,
-5.1272e-01, 1.3956e-02],
[-7.8722e-01, 1.9454e+00, 1.6541e+00, ..., 5.0418e-01,
4.4243e-01, 6.8148e-01],
[-2.2064e+00, -1.1363e+00, -9.7375e-01, ..., -1.9038e+00,
-9.8480e-01, -3.1148e-01]],
[[-1.3655e+00, -1.2612e+00, -2.6281e+00, ..., -1.0542e+00,
-1.8565e-01, -2.9481e-01],
[-6.7647e-01, -1.7561e-01, -3.4312e+00, ..., -2.5380e+00,
-1.5216e+00, -1.9546e+00],
[-2.0927e+00, -1.3850e+00, -5.7341e+00, ..., -3.2940e+00,
-2.6141e+00, -2.6081e+00],
...,
[-2.4391e+00, -1.5565e+00, -5.4640e+00, ..., -3.0465e+00,
-2.6877e+00, -2.1594e+00],
[-2.1813e+00, -1.4443e+00, -4.2701e+00, ..., -2.6452e+00,
-2.2979e+00, -1.6188e+00],
[-2.3701e+00, -1.7585e+00, -3.3065e+00, ..., -2.9519e+00,
-2.4502e+00, -2.1111e+00]],
[[-1.1439e+00, -1.7019e+00, -1.5056e+00, ..., -2.4003e+00,
-1.7919e+00, -1.1627e+00],
[-1.1541e+00, -1.6264e+00, -1.1899e+00, ..., -2.0474e+00,
-1.4850e+00, -1.0709e+00],
[-1.4943e+00, -1.7882e+00, -1.3697e+00, ..., -1.9377e+00,
-1.3778e+00, -9.4576e-01],
...,
[-3.1664e+00, -3.9538e+00, -3.7411e+00, ..., -3.9555e+00,
-3.1631e+00, -2.4385e+00],
[-2.1121e+00, -2.7248e+00, -2.6874e+00, ..., -2.7729e+00,
-2.2871e+00, -1.8598e+00],
[-1.5358e+00, -1.8607e+00, -1.9275e+00, ..., -1.8684e+00,
-1.5177e+00, -1.1604e+00]]],
[[[ 1.6147e-01, 4.0713e-01, -1.1641e+00, ..., 4.5377e-01,
1.0250e+00, 6.6879e-01],
[-2.0899e+00, -1.4157e+00, -4.3737e+00, ..., -2.5760e+00,
-7.4637e-01, -2.0721e-02],
[ 4.5081e-01, -2.5879e-01, -1.1137e+00, ..., -2.6972e-01,
4.6767e-01, 6.5860e-01],
...,
[-2.7378e-01, -1.2572e+00, -2.6096e+00, ..., -2.0359e+00,
-1.4061e+00, -9.3365e-01],
[ 1.2587e+00, 7.0541e-01, -8.8040e-01, ..., -3.5659e-01,
5.0771e-01, 6.9775e-01],
[ 8.2196e-01, 3.3259e-01, -4.1808e-01, ..., -3.8870e-01,
2.8830e-01, 2.7177e-01]],
[[-3.2484e+00, -1.4625e+00, -6.1178e+00, ..., -4.8380e+00,
-4.4207e+00, -3.3809e+00],
[-5.6012e-01, 2.1578e+00, -4.1578e+00, ..., -3.3936e+00,
-3.5280e+00, -2.9072e+00],
[ 5.4457e-02, 1.8656e+00, -7.3901e+00, ..., -5.3748e+00,
-5.2356e+00, -4.9298e+00],
...,
[-1.2373e+00, 2.8008e-01, -6.5228e+00, ..., -4.5493e+00,
-4.1271e+00, -3.4490e+00],
[ 1.5167e+00, 2.6000e+00, -1.3998e+00, ..., -8.5637e-01,
-1.2039e+00, -1.2441e+00],
[ 9.8461e-01, 1.6133e+00, -2.4355e+00, ..., -2.2309e+00,
-2.4670e+00, -2.3596e+00]],
[[-1.3789e+00, -2.3558e+00, -2.1725e+00, ..., -5.7707e-01,
-1.8287e-01, -4.7465e-01],
[-2.0259e+00, -2.2088e+00, -1.5768e+00, ..., -3.2683e-01,
8.1778e-02, 3.1875e-02],
[-2.5197e+00, -1.9979e+00, -8.5091e-01, ..., -2.9065e-01,
2.4652e-01, 2.8895e-01],
...,
[-1.5938e+00, -1.7928e+00, -7.8542e-01, ..., -4.9176e-01,
-1.6315e-01, -3.4899e-01],
[-1.7426e+00, -1.1602e+00, -7.5768e-01, ..., -3.2318e-01,
1.4932e-01, 2.1792e-01],
[-1.6802e+00, -1.0675e+00, -5.7158e-01, ..., -5.0879e-01,
-2.8058e-01, -2.5317e-01]],
...,
[[-2.3884e+00, -1.3247e+00, -2.7467e+00, ..., -3.6802e+00,
-2.8359e+00, -1.8351e+00],
[-1.8852e+00, 2.1260e-01, -1.3008e+00, ..., -2.7057e+00,
-2.4277e+00, -1.8775e+00],
[-5.1390e-01, 2.4959e+00, 9.6628e-01, ..., -6.1571e-01,
-6.2537e-02, 1.7719e-01],
...,
[-1.6890e+00, 1.1960e+00, 2.5721e-01, ..., -1.5205e+00,
-7.1598e-01, -1.5535e-01],
[-8.5280e-01, 1.6434e+00, 1.5195e+00, ..., -1.1474e-01,
1.0370e-01, 3.1901e-01],
[-2.1153e+00, -1.0878e+00, -8.6643e-01, ..., -2.3420e+00,
-1.4544e+00, -9.2709e-01]],
[[-2.3659e+00, -2.6133e+00, -3.9948e+00, ..., -1.1574e+00,
-7.4789e-01, -7.7667e-01],
[-1.6478e+00, -1.2956e+00, -4.7404e+00, ..., -1.3383e+00,
-1.2114e+00, -1.5981e+00],
[-3.0179e+00, -2.4267e+00, -6.8952e+00, ..., -3.2392e+00,
-2.7814e+00, -2.2379e+00],
...,
[-2.5356e+00, -1.7312e+00, -4.9554e+00, ..., -1.4562e+00,
-1.0306e+00, -8.8286e-01],
[-2.2205e+00, -1.5406e+00, -3.8599e+00, ..., -1.7257e+00,
-1.1525e+00, -7.5603e-01],
[-2.3050e+00, -1.7615e+00, -3.0788e+00, ..., -2.8053e+00,
-2.0247e+00, -1.8356e+00]],
[[-1.8890e+00, -2.6208e+00, -2.4014e+00, ..., -1.8210e+00,
-1.5073e+00, -1.1531e+00],
[-1.9108e+00, -2.6242e+00, -2.0832e+00, ..., -1.4205e+00,
-1.1234e+00, -9.0041e-01],
[-2.4373e+00, -2.9877e+00, -2.4973e+00, ..., -1.6187e+00,
-1.2123e+00, -9.0202e-01],
...,
[-3.1094e+00, -3.8851e+00, -3.6598e+00, ..., -2.8760e+00,
-2.2932e+00, -1.7690e+00],
[-2.0790e+00, -2.6825e+00, -2.6531e+00, ..., -1.9851e+00,
-1.6246e+00, -1.3129e+00],
[-1.4968e+00, -1.8159e+00, -1.8666e+00, ..., -1.2273e+00,
-1.0351e+00, -7.9500e-01]]]], device='cuda:0'),
tensor([[[[ 4.7387e-01, 9.3828e-01, 7.2540e-01, ..., 7.8033e-01,
1.0574e+00, 8.2727e-01],
[-1.0640e+00, -1.3918e-01, -1.4438e+00, ..., -1.2434e+00,
4.2892e-02, 6.1669e-01],
[ 7.3018e-01, 5.3247e-01, 2.5471e-01, ..., 4.7711e-01,
8.7968e-01, 8.7990e-01],
...,
[-8.5818e-01, -1.9780e+00, -2.0785e+00, ..., -1.7497e+00,
-1.2642e+00, -8.5861e-01],
[ 8.5985e-01, 3.3130e-02, -1.2995e-01, ..., -2.9758e-02,
6.9044e-01, 8.1797e-01],
[ 2.1311e-01, -5.5355e-01, -4.9546e-01, ..., -1.6728e-01,
4.1144e-01, 3.4131e-01]],
[[-2.6002e+00, -1.4281e+00, -4.7347e+00, ..., -5.8402e+00,
-4.9933e+00, -3.8661e+00],
[-6.2601e-01, 1.3746e+00, -3.2978e+00, ..., -4.6562e+00,
-4.4843e+00, -3.6178e+00],
[ 4.8943e-02, 1.1702e+00, -5.9953e+00, ..., -6.7979e+00,
-6.4909e+00, -5.8751e+00],
...,
[ 5.0854e-01, 6.7667e-01, -3.8495e+00, ..., -4.6431e+00,
-4.1895e+00, -3.5250e+00],
[ 2.6703e+00, 2.7140e+00, 3.1629e-01, ..., -9.0910e-01,
-1.2479e+00, -1.3059e+00],
[ 1.6061e+00, 1.4341e+00, -1.5748e+00, ..., -2.2793e+00,
-2.5304e+00, -2.4223e+00]],
[[-5.4188e-01, -1.5583e+00, -1.3632e+00, ..., -7.5398e-01,
-5.0860e-01, -7.2938e-01],
[-1.2961e+00, -1.6939e+00, -1.0762e+00, ..., -6.8289e-01,
-3.5890e-01, -3.0536e-01],
[-1.1738e+00, -1.0884e+00, -4.9461e-01, ..., -3.9780e-01,
-4.5773e-02, -5.6173e-02],
...,
[-2.9945e-01, -7.5016e-01, -5.0795e-01, ..., -4.8826e-01,
-1.8623e-01, -4.3365e-01],
[-8.7027e-01, -2.6904e-01, -4.4073e-01, ..., -2.0891e-01,
2.4436e-01, 2.4224e-01],
[-9.5283e-01, -2.4246e-01, -3.2326e-01, ..., -4.1268e-01,
-1.9998e-01, -2.2803e-01]],
...,
[[-2.0589e+00, -1.1541e+00, -2.4891e+00, ..., -3.2998e+00,
-2.7239e+00, -1.8604e+00],
[-5.0596e-01, 1.1492e+00, -7.1322e-01, ..., -1.6273e+00,
-1.8213e+00, -1.5861e+00],
[-2.2172e-01, 2.1920e+00, 4.7456e-01, ..., -3.3565e-01,
-6.5524e-02, 1.2604e-01],
...,
[-2.2056e+00, 1.0251e-01, -8.6325e-02, ..., -1.3660e+00,
-4.0778e-01, 7.7654e-02],
[-1.2507e+00, 9.7061e-01, 1.0806e+00, ..., -6.8681e-03,
2.9314e-01, 4.6390e-01],
[-2.2136e+00, -1.4869e+00, -1.7483e+00, ..., -2.3979e+00,
-1.4503e+00, -9.1051e-01]],
[[-1.4291e+00, -1.3732e+00, -2.7862e+00, ..., -1.0579e+00,
-5.3768e-01, -4.2934e-01],
[-8.4934e-01, -4.3515e-01, -3.7458e+00, ..., -1.8472e+00,
-1.5875e+00, -1.8012e+00],
[-2.4168e+00, -1.8863e+00, -6.3288e+00, ..., -4.1681e+00,
-3.5886e+00, -2.8405e+00],
...,
[-1.3448e+00, 1.7132e-01, -3.4990e+00, ..., -1.4203e+00,
-1.0567e+00, -9.7786e-01],
[-1.6852e+00, -4.8122e-01, -3.0860e+00, ..., -1.7131e+00,
-1.1883e+00, -8.5790e-01],
[-2.3633e+00, -1.4051e+00, -3.1606e+00, ..., -2.9231e+00,
-2.1354e+00, -1.9844e+00]],
[[-1.2132e+00, -1.7945e+00, -1.6118e+00, ..., -1.8443e+00,
-1.5319e+00, -1.1347e+00],
[-1.2697e+00, -1.7790e+00, -1.3716e+00, ..., -1.6008e+00,
-1.3032e+00, -1.0188e+00],
[-1.6854e+00, -2.0381e+00, -1.6596e+00, ..., -1.8847e+00,
-1.4511e+00, -1.0600e+00],
...,
[-2.5052e+00, -3.0940e+00, -2.8124e+00, ..., -2.8357e+00,
-2.2621e+00, -1.7401e+00],
[-1.6194e+00, -2.1081e+00, -1.9982e+00, ..., -1.9262e+00,
-1.5846e+00, -1.2782e+00],
[-1.0528e+00, -1.3519e+00, -1.3635e+00, ..., -1.1860e+00,
-1.0060e+00, -7.6518e-01]]],
[[[ 2.8498e-01, 5.4565e-01, 6.6990e-01, ..., 4.7020e-01,
1.0230e+00, 5.9257e-01],
[-2.1331e+00, -1.5082e+00, -3.1219e+00, ..., -2.8453e+00,
-9.4582e-01, -2.3074e-01],
[-1.9197e-01, -9.8734e-01, -1.8447e-01, ..., -1.4418e-01,
5.5788e-01, 6.4758e-01],
...,
[-2.1314e-01, -1.1812e+00, -2.6614e+00, ..., -1.6023e+00,
-1.4763e+00, -8.9101e-01],
[ 1.4378e+00, 8.6762e-01, -9.1537e-01, ..., -3.1564e-01,
7.2761e-02, 3.6199e-01],
[ 8.1214e-01, 2.9659e-01, -6.9360e-01, ..., -1.9406e-01,
1.9954e-01, 3.7247e-01]],
[[-1.6818e+00, -1.2563e+00, -4.2280e+00, ..., -5.0722e+00,
-4.6028e+00, -3.4920e+00],
[ 4.7602e-01, 1.7003e+00, -2.3965e+00, ..., -3.5139e+00,
-3.6353e+00, -2.9827e+00],
[ 1.1596e+00, 1.4253e+00, -4.9829e+00, ..., -5.5362e+00,
-5.3685e+00, -5.0440e+00],
...,
[-1.3193e+00, 1.3207e-01, -6.7370e+00, ..., -7.6465e+00,
-6.9410e+00, -5.4483e+00],
[ 1.7052e+00, 2.7516e+00, -1.1357e+00, ..., -2.5760e+00,
-2.9444e+00, -2.3722e+00],
[ 1.1618e+00, 1.7588e+00, -2.3393e+00, ..., -3.2459e+00,
-3.6155e+00, -3.1358e+00]],
[[-9.9897e-02, -1.0725e+00, -1.2582e+00, ..., -5.9185e-01,
-2.1714e-01, -5.3935e-01],
[-5.1498e-01, -7.6448e-01, -7.3034e-01, ..., -3.4616e-01,
4.9227e-03, -8.4088e-02],
[-4.5050e-01, -2.7524e-01, -3.7386e-01, ..., -3.6965e-01,
1.0042e-01, 1.0655e-01],
...,
[-1.3843e+00, -1.7230e+00, -7.3232e-01, ..., -9.0349e-01,
-7.7195e-01, -1.0380e+00],
[-1.7231e+00, -1.1612e+00, -8.5029e-01, ..., -5.7684e-01,
-2.9668e-01, -3.6390e-01],
[-1.8928e+00, -1.2892e+00, -8.9945e-01, ..., -9.7269e-01,
-8.4227e-01, -8.1553e-01]],
...,
[[-2.9354e+00, -2.5143e+00, -2.7921e+00, ..., -3.7372e+00,
-2.8355e+00, -1.8705e+00],
[-2.2482e+00, -9.3948e-01, -1.6342e+00, ..., -2.7307e+00,
-2.4405e+00, -1.9304e+00],
[-1.1267e+00, 9.3429e-01, 4.7395e-01, ..., -4.7986e-01,
2.5575e-02, 1.7589e-01],
...,
[-1.8043e+00, 1.2784e+00, 2.0037e-02, ..., -1.0981e+00,
-5.5275e-01, 1.0164e-02],
[-8.5869e-01, 1.9007e+00, 1.6557e+00, ..., 5.1903e-01,
4.3355e-01, 6.9299e-01],
[-2.1710e+00, -1.1156e+00, -9.2621e-01, ..., -1.8731e+00,
-9.7458e-01, -3.0358e-01]],
[[-1.6048e+00, -1.6088e+00, -2.7888e+00, ..., -1.1604e+00,
-7.0492e-01, -7.4347e-01],
[-6.5900e-01, -2.9177e-02, -3.0824e+00, ..., -1.2871e+00,
-1.1454e+00, -1.5754e+00],
[-2.1249e+00, -1.0254e+00, -5.1781e+00, ..., -3.2522e+00,
-2.7695e+00, -2.2527e+00],
...,
[-2.5106e+00, -1.6597e+00, -5.5610e+00, ..., -3.0366e+00,
-2.7118e+00, -2.1732e+00],
[-2.1813e+00, -1.5103e+00, -4.3491e+00, ..., -2.6595e+00,
-2.2990e+00, -1.6291e+00],
[-2.3296e+00, -1.7677e+00, -3.3733e+00, ..., -2.9715e+00,
-2.4375e+00, -2.1214e+00]],
[[-1.3244e+00, -1.8742e+00, -1.7075e+00, ..., -1.8310e+00,
-1.5119e+00, -1.1435e+00],
[-1.2505e+00, -1.7321e+00, -1.2569e+00, ..., -1.3793e+00,
-1.0877e+00, -8.6415e-01],
[-1.5618e+00, -1.9122e+00, -1.5071e+00, ..., -1.5744e+00,
-1.1733e+00, -8.6223e-01],
...,
[-3.1962e+00, -3.9927e+00, -3.7730e+00, ..., -3.9790e+00,
-3.1861e+00, -2.4583e+00],
[-2.1507e+00, -2.7696e+00, -2.7314e+00, ..., -2.7818e+00,
-2.2952e+00, -1.8690e+00],
[-1.5792e+00, -1.9116e+00, -1.9636e+00, ..., -1.8747e+00,
-1.5243e+00, -1.1649e+00]]],
[[[ 1.5049e-01, 3.6644e-01, -1.1976e+00, ..., 4.5466e-01,
9.9570e-01, 6.3044e-01],
[-2.1037e+00, -1.4564e+00, -4.3919e+00, ..., -2.4169e+00,
-6.7118e-01, -5.4012e-02],
[ 4.4769e-01, -2.9807e-01, -1.1002e+00, ..., -1.3625e-01,
5.8719e-01, 6.6126e-01],
...,
[-2.8069e-01, -1.2341e+00, -2.5447e+00, ..., -2.1990e+00,
-1.6087e+00, -1.1292e+00],
[ 1.2476e+00, 6.7742e-01, -8.4908e-01, ..., -3.1908e-01,
4.9713e-01, 6.9086e-01],
[ 8.1855e-01, 3.1248e-01, -3.8517e-01, ..., -3.4053e-01,
3.1435e-01, 2.9068e-01]],
[[-3.2478e+00, -1.4529e+00, -6.1448e+00, ..., -4.7478e+00,
-4.3770e+00, -3.3058e+00],
[-5.4907e-01, 2.1812e+00, -4.1758e+00, ..., -3.4535e+00,
-3.6087e+00, -2.8885e+00],
[ 6.1986e-02, 1.8684e+00, -7.4107e+00, ..., -5.3917e+00,
-5.2793e+00, -4.8200e+00],
...,
[-1.2512e+00, 2.7589e-01, -6.5145e+00, ..., -4.5690e+00,
-4.0716e+00, -3.3990e+00],
[ 1.5312e+00, 2.6131e+00, -1.3423e+00, ..., -5.4405e-01,
-9.0073e-01, -1.0203e+00],
[ 1.0020e+00, 1.6350e+00, -2.4112e+00, ..., -2.1356e+00,
-2.4169e+00, -2.3579e+00]],
[[-1.3881e+00, -2.3607e+00, -2.1838e+00, ..., -5.6991e-01,
-1.8359e-01, -4.2600e-01],
[-2.0250e+00, -2.2032e+00, -1.5585e+00, ..., -3.2883e-01,
6.8106e-02, 4.2112e-02],
[-2.5340e+00, -1.9983e+00, -8.2220e-01, ..., -2.9398e-01,
2.3250e-01, 2.7648e-01],
...,
[-1.5889e+00, -1.8384e+00, -7.9179e-01, ..., -5.5227e-01,
-2.0200e-01, -3.9860e-01],
[-1.7686e+00, -1.1922e+00, -7.7254e-01, ..., -3.1746e-01,
1.7196e-01, 2.3552e-01],
[-1.7414e+00, -1.1253e+00, -5.9922e-01, ..., -4.9708e-01,
-2.6290e-01, -2.4688e-01]],
...,
[[-2.3913e+00, -1.3161e+00, -2.7683e+00, ..., -3.5161e+00,
-2.6841e+00, -1.7411e+00],
[-1.8831e+00, 2.1713e-01, -1.3656e+00, ..., -2.6080e+00,
-2.3155e+00, -1.7654e+00],
[-4.9746e-01, 2.5207e+00, 8.8633e-01, ..., -6.0855e-01,
-9.8954e-02, 1.4800e-01],
...,
[-1.7179e+00, 1.1361e+00, 1.6813e-01, ..., -1.6031e+00,
-6.8156e-01, -8.3331e-02],
[-8.3882e-01, 1.6397e+00, 1.4525e+00, ..., -9.2795e-02,
1.7452e-01, 3.8855e-01],
[-2.0714e+00, -1.0554e+00, -8.7657e-01, ..., -2.5762e+00,
-1.5662e+00, -9.7270e-01]],
[[-2.3695e+00, -2.6170e+00, -4.0287e+00, ..., -1.2489e+00,
-8.4085e-01, -8.0180e-01],
[-1.6501e+00, -1.3090e+00, -4.7816e+00, ..., -1.4893e+00,
-1.3099e+00, -1.5820e+00],
[-3.0111e+00, -2.4371e+00, -6.9399e+00, ..., -3.3412e+00,
-2.8119e+00, -2.2092e+00],
...,
[-2.5031e+00, -1.7734e+00, -5.0256e+00, ..., -1.3414e+00,
-9.1784e-01, -7.8439e-01],
[-2.1942e+00, -1.5629e+00, -3.9170e+00, ..., -1.6712e+00,
-1.0704e+00, -6.6029e-01],
[-2.2833e+00, -1.7754e+00, -3.0867e+00, ..., -2.8805e+00,
-2.0430e+00, -1.8664e+00]],
[[-1.8969e+00, -2.6253e+00, -2.4086e+00, ..., -1.7622e+00,
-1.4775e+00, -1.1210e+00],
[-1.9203e+00, -2.6352e+00, -2.1012e+00, ..., -1.4078e+00,
-1.1335e+00, -8.9102e-01],
[-2.4427e+00, -2.9993e+00, -2.5140e+00, ..., -1.6316e+00,
-1.2430e+00, -9.0922e-01],
...,
[-3.1027e+00, -3.8739e+00, -3.6376e+00, ..., -2.9707e+00,
-2.3694e+00, -1.8198e+00],
[-2.0711e+00, -2.6692e+00, -2.6268e+00, ..., -2.0307e+00,
-1.6658e+00, -1.3514e+00],
[-1.4918e+00, -1.8109e+00, -1.8488e+00, ..., -1.2288e+00,
-1.0517e+00, -8.0263e-01]]],
[[[ 2.4945e-01, 5.3193e-01, 6.4216e-01, ..., 4.8962e-01,
8.8821e-01, 7.1425e-01],
[-1.8834e+00, -1.1655e+00, -2.6412e+00, ..., -1.7191e+00,
-1.5555e-01, 5.0041e-01],
[-2.7684e-02, -7.1729e-01, 7.7174e-02, ..., 4.1275e-01,
8.5256e-01, 8.8551e-01],
...,
[-8.5008e-01, -1.9662e+00, -1.9948e+00, ..., -1.2710e+00,
-7.9963e-01, -5.3387e-01],
[ 8.6562e-01, 3.9047e-02, -3.4519e-02, ..., 3.5955e-01,
7.4510e-01, 8.4462e-01],
[ 3.0156e-01, -3.9095e-01, -2.8056e-01, ..., 5.4496e-01,
8.2345e-01, 7.6168e-01]],
[[-1.7645e+00, -1.3666e+00, -4.2550e+00, ..., -6.1646e+00,
-5.2201e+00, -4.0350e+00],
[ 3.7703e-01, 1.4744e+00, -2.5577e+00, ..., -4.8439e+00,
-4.6125e+00, -3.7572e+00],
[ 1.1153e+00, 1.2208e+00, -5.0855e+00, ..., -7.1100e+00,
-6.6928e+00, -6.0656e+00],
...,
[ 2.6780e-01, 3.4117e-01, -4.1535e+00, ..., -6.2192e+00,
-5.4072e+00, -4.4482e+00],
[ 2.5027e+00, 2.5167e+00, 1.7524e-01, ..., -1.7611e+00,
-1.8509e+00, -1.8106e+00],
[ 1.4862e+00, 1.3480e+00, -1.5863e+00, ..., -2.8589e+00,
-2.9571e+00, -2.8217e+00]],
[[-2.0367e-01, -1.2469e+00, -1.4294e+00, ..., -9.5554e-01,
-6.3892e-01, -7.6967e-01],
[-6.2614e-01, -9.6412e-01, -9.3221e-01, ..., -7.8344e-01,
-4.1231e-01, -2.6856e-01],
[-4.3539e-01, -3.3798e-01, -4.4588e-01, ..., -4.7224e-01,
-1.7461e-02, 9.9268e-02],
...,
[-1.6451e-01, -7.1959e-01, -6.3620e-01, ..., -7.1278e-01,
-5.1545e-01, -5.7121e-01],
[-7.7088e-01, -3.0960e-01, -5.7937e-01, ..., -3.5180e-01,
-7.7978e-02, 1.1024e-02],
[-9.3731e-01, -3.4050e-01, -4.7136e-01, ..., -3.3339e-01,
-2.2748e-01, -1.6816e-01]],
...,
[[-2.7555e+00, -2.2876e+00, -2.4223e+00, ..., -3.5023e+00,
-2.8374e+00, -1.8909e+00],
[-1.9063e+00, -5.6488e-01, -9.9865e-01, ..., -1.9046e+00,
-1.9511e+00, -1.6009e+00],
[-1.0038e+00, 9.9753e-01, 7.8228e-01, ..., -2.5122e-01,
5.3666e-02, 2.3909e-01],
...,
[-2.1912e+00, -1.3056e-02, -1.0791e-01, ..., -9.3395e-01,
-3.4337e-01, 1.3929e-01],
[-1.1652e+00, 9.5210e-01, 1.0700e+00, ..., 2.5146e-01,
3.4036e-01, 4.4705e-01],
[-2.1130e+00, -1.4517e+00, -1.6537e+00, ..., -2.0256e+00,
-1.1226e+00, -5.4271e-01]],
[[-1.5918e+00, -1.5747e+00, -2.7642e+00, ..., -1.3603e+00,
-7.3259e-01, -5.3772e-01],
[-6.8873e-01, -9.0671e-02, -3.1840e+00, ..., -2.0434e+00,
-1.6875e+00, -1.8074e+00],
[-2.0690e+00, -1.0239e+00, -5.2116e+00, ..., -4.1684e+00,
-3.5433e+00, -2.6959e+00],
...,
[-1.3712e+00, 8.6352e-02, -3.4946e+00, ..., -2.5938e+00,
-1.9858e+00, -1.4682e+00],
[-1.6578e+00, -4.9570e-01, -3.0771e+00, ..., -2.3734e+00,
-1.6382e+00, -9.8447e-01],
[-2.2944e+00, -1.3518e+00, -3.0174e+00, ..., -2.8505e+00,
-1.9586e+00, -1.6235e+00]],
[[-1.2788e+00, -1.8111e+00, -1.6253e+00, ..., -2.0409e+00,
-1.6489e+00, -1.2029e+00],
[-1.2282e+00, -1.6941e+00, -1.2125e+00, ..., -1.7442e+00,
-1.3770e+00, -1.0530e+00],
[-1.5490e+00, -1.8835e+00, -1.4585e+00, ..., -2.0139e+00,
-1.5095e+00, -1.0656e+00],
...,
[-2.4268e+00, -3.0289e+00, -2.7681e+00, ..., -3.1976e+00,
-2.5227e+00, -1.8845e+00],
[-1.5732e+00, -2.0704e+00, -1.9716e+00, ..., -2.2665e+00,
-1.8385e+00, -1.4451e+00],
[-1.0304e+00, -1.3309e+00, -1.3538e+00, ..., -1.4347e+00,
-1.1912e+00, -8.6832e-01]]],
[[[ 4.6706e-01, 9.4092e-01, 7.3825e-01, ..., 8.0510e-01,
1.0520e+00, 8.2298e-01],
[-1.0533e+00, -1.3123e-01, -1.4009e+00, ..., -1.2069e+00,
5.8703e-02, 6.3080e-01],
[ 7.3514e-01, 5.4462e-01, 1.2427e-01, ..., 5.5803e-01,
9.0924e-01, 9.0093e-01],
...,
[-8.5402e-01, -1.8136e+00, -1.8256e+00, ..., -1.8100e+00,
-1.3603e+00, -9.7978e-01],
[ 7.4014e-01, -5.0642e-03, -8.3712e-02, ..., -2.0215e-02,
6.8819e-01, 7.8301e-01],
[ 1.1016e-01, -5.9946e-01, -4.5928e-01, ..., -1.6008e-01,
4.2199e-01, 3.2720e-01]],
[[-2.6030e+00, -1.4577e+00, -4.7412e+00, ..., -5.8484e+00,
-4.9846e+00, -3.8894e+00],
[-6.5219e-01, 1.3266e+00, -3.3118e+00, ..., -4.6622e+00,
-4.4642e+00, -3.6419e+00],
[ 1.5495e-02, 1.1442e+00, -5.9046e+00, ..., -6.7671e+00,
-6.4654e+00, -5.9013e+00],
...,
[ 2.8434e-01, 3.4657e-01, -4.0527e+00, ..., -4.6493e+00,
-4.2042e+00, -3.4755e+00],
[ 2.2809e+00, 2.2428e+00, -2.1687e-01, ..., -8.0521e-01,
-1.1711e+00, -1.1943e+00],
[ 1.3324e+00, 1.1574e+00, -1.8034e+00, ..., -2.2286e+00,
-2.5021e+00, -2.3843e+00]],
[[-5.4172e-01, -1.5522e+00, -1.3806e+00, ..., -7.5119e-01,
-5.0968e-01, -7.3072e-01],
[-1.3009e+00, -1.6861e+00, -1.0803e+00, ..., -6.8486e-01,
-3.7765e-01, -2.9915e-01],
[-1.1764e+00, -1.0675e+00, -4.4710e-01, ..., -3.8261e-01,
-5.2153e-02, -5.1490e-02],
...,
[-8.9149e-02, -5.6067e-01, -4.4671e-01, ..., -5.0070e-01,
-1.9914e-01, -4.4505e-01],
[-6.5657e-01, -1.6112e-01, -4.1926e-01, ..., -2.1256e-01,
2.5377e-01, 2.5164e-01],
[-7.6650e-01, -1.6199e-01, -3.3959e-01, ..., -4.0136e-01,
-1.9418e-01, -2.2580e-01]],
...,
[[-2.0452e+00, -1.1601e+00, -2.4626e+00, ..., -3.3294e+00,
-2.7256e+00, -1.8584e+00],
[-4.9818e-01, 1.1214e+00, -6.7190e-01, ..., -1.6373e+00,
-1.8107e+00, -1.6211e+00],
[-2.0074e-01, 2.1790e+00, 5.4413e-01, ..., -3.8366e-01,
-7.0998e-02, 1.1115e-01],
...,
[-2.1125e+00, -1.6549e-01, -2.0221e-01, ..., -1.4119e+00,
-4.0358e-01, 1.0293e-01],
[-1.2579e+00, 7.1235e-01, 8.7741e-01, ..., -2.0059e-02,
3.0129e-01, 4.7435e-01],
[-2.1401e+00, -1.4744e+00, -1.6707e+00, ..., -2.4844e+00,
-1.4938e+00, -9.5025e-01]],
[[-1.4278e+00, -1.3868e+00, -2.7744e+00, ..., -1.0838e+00,
-5.4659e-01, -4.3494e-01],
[-8.5571e-01, -4.5981e-01, -3.6950e+00, ..., -1.8591e+00,
-1.6004e+00, -1.8045e+00],
[-2.4126e+00, -1.8856e+00, -6.2424e+00, ..., -4.1861e+00,
-3.6122e+00, -2.8561e+00],
...,
[-1.4114e+00, -7.9044e-02, -3.3773e+00, ..., -1.4312e+00,
-9.9517e-01, -9.1457e-01],
[-1.7097e+00, -6.6564e-01, -3.0454e+00, ..., -1.7349e+00,
-1.1401e+00, -7.9427e-01],
[-2.3075e+00, -1.5056e+00, -3.1118e+00, ..., -2.9760e+00,
-2.1306e+00, -1.9834e+00]],
[[-1.2091e+00, -1.7808e+00, -1.6089e+00, ..., -1.8482e+00,
-1.5352e+00, -1.1397e+00],
[-1.2669e+00, -1.7657e+00, -1.3771e+00, ..., -1.6014e+00,
-1.3093e+00, -1.0190e+00],
[-1.6786e+00, -2.0403e+00, -1.7151e+00, ..., -1.8966e+00,
-1.4594e+00, -1.0650e+00],
...,
[-2.3821e+00, -2.9555e+00, -2.7398e+00, ..., -2.8597e+00,
-2.2880e+00, -1.7582e+00],
[-1.5498e+00, -2.0181e+00, -1.9554e+00, ..., -1.9349e+00,
-1.6004e+00, -1.2918e+00],
[-1.0136e+00, -1.3087e+00, -1.3480e+00, ..., -1.1847e+00,
-1.0101e+00, -7.6806e-01]]]], device='cuda:0'),
tensor([[[[ 4.2197e-01, 8.3849e-01, 5.1386e-01, ..., 4.0862e-01,
9.6717e-01, 6.2753e-01],
[-1.2834e+00, -4.4703e-01, -1.9638e+00, ..., -2.5310e+00,
-7.7754e-01, -7.7862e-02],
[ 8.7760e-01, 6.3193e-01, 3.9787e-01, ..., -3.0516e-01,
4.0185e-01, 6.0868e-01],
...,
[ 2.4927e-02, -5.9474e-01, -1.3236e+00, ..., -2.2569e+00,
-1.5821e+00, -1.1115e+00],
[ 1.3400e+00, 1.1400e+00, 6.2318e-01, ..., -3.9554e-01,
5.0624e-01, 6.8514e-01],
[ 9.9658e-01, 8.7774e-01, 6.2010e-01, ..., -3.6711e-01,
3.2724e-01, 2.8039e-01]],
[[-2.7840e+00, -1.5953e+00, -4.8302e+00, ..., -4.7298e+00,
-4.3406e+00, -3.3107e+00],
[-5.4861e-01, 1.5510e+00, -2.9425e+00, ..., -3.3520e+00,
-3.5180e+00, -2.8794e+00],
[ 2.3677e-01, 1.3731e+00, -5.6410e+00, ..., -5.2645e+00,
-5.1433e+00, -4.8222e+00],
...,
[-9.0794e-01, -1.5558e-02, -4.9070e+00, ..., -4.5378e+00,
-4.1031e+00, -3.3974e+00],
[ 1.6804e+00, 2.2653e+00, -3.5140e-01, ..., -5.2033e-01,
-9.1916e-01, -1.0203e+00],
[ 9.7244e-01, 1.2242e+00, -1.8231e+00, ..., -2.1309e+00,
-2.4231e+00, -2.3521e+00]],
[[-6.9491e-01, -1.8625e+00, -1.7239e+00, ..., -5.7741e-01,
-1.6328e-01, -4.1634e-01],
[-1.4371e+00, -1.9118e+00, -1.2570e+00, ..., -3.3116e-01,
9.0472e-02, 5.1299e-02],
[-1.3342e+00, -1.3450e+00, -7.3538e-01, ..., -2.7426e-01,
2.2826e-01, 2.7034e-01],
...,
[-9.1025e-01, -1.4102e+00, -6.8218e-01, ..., -5.5891e-01,
-1.8880e-01, -4.0809e-01],
[-1.1941e+00, -8.8519e-01, -5.9568e-01, ..., -3.4259e-01,
1.7819e-01, 2.4102e-01],
[-1.2296e+00, -7.0381e-01, -2.9657e-01, ..., -5.1813e-01,
-2.7624e-01, -2.4850e-01]],
...,
[[-2.2552e+00, -1.1733e+00, -2.5109e+00, ..., -3.4402e+00,
-2.6790e+00, -1.7516e+00],
[-6.2922e-01, 1.2344e+00, -7.1539e-01, ..., -2.5229e+00,
-2.3447e+00, -1.8277e+00],
[-2.6543e-01, 2.4944e+00, 7.7477e-01, ..., -5.3985e-01,
-8.5281e-02, 1.3943e-01],
...,
[-1.3287e+00, 1.3257e+00, 2.5081e-02, ..., -1.5648e+00,
-6.9635e-01, -6.7428e-02],
[-4.9644e-01, 1.4997e+00, 8.1221e-01, ..., -5.8774e-02,
1.6378e-01, 3.8939e-01],
[-1.6126e+00, -1.0680e+00, -1.5099e+00, ..., -2.5525e+00,
-1.5670e+00, -9.8473e-01]],
[[-1.4155e+00, -1.3942e+00, -3.0191e+00, ..., -1.1925e+00,
-7.8147e-01, -7.8943e-01],
[-6.8934e-01, -2.5145e-01, -3.9007e+00, ..., -1.3819e+00,
-1.2235e+00, -1.5609e+00],
[-2.1929e+00, -1.5893e+00, -6.4324e+00, ..., -3.2206e+00,
-2.7280e+00, -2.1836e+00],
...,
[-1.8175e+00, -9.8232e-01, -4.6604e+00, ..., -1.2962e+00,
-9.0179e-01, -7.9536e-01],
[-1.7163e+00, -1.0322e+00, -3.7337e+00, ..., -1.6439e+00,
-1.0466e+00, -6.5810e-01],
[-2.0301e+00, -1.4606e+00, -3.0416e+00, ..., -2.8748e+00,
-2.0203e+00, -1.8490e+00]],
[[-1.2390e+00, -1.8853e+00, -1.7296e+00, ..., -1.7880e+00,
-1.4743e+00, -1.1212e+00],
[-1.2410e+00, -1.7986e+00, -1.3851e+00, ..., -1.4248e+00,
-1.1169e+00, -8.8712e-01],
[-1.6030e+00, -1.9862e+00, -1.5994e+00, ..., -1.6524e+00,
-1.2259e+00, -9.0649e-01],
...,
[-2.3460e+00, -2.9574e+00, -2.8488e+00, ..., -2.9684e+00,
-2.3713e+00, -1.8241e+00],
[-1.5884e+00, -2.0354e+00, -2.0422e+00, ..., -2.0264e+00,
-1.6709e+00, -1.3533e+00],
[-1.0895e+00, -1.3290e+00, -1.4334e+00, ..., -1.2290e+00,
-1.0538e+00, -8.0627e-01]]],
[[[ 4.1760e-02, 1.8256e-01, -1.5112e+00, ..., -7.0079e-01,
-2.6999e-01, -9.0292e-02],
[-2.1033e+00, -1.6577e+00, -4.6308e+00, ..., -3.9024e+00,
-1.6260e+00, -6.1110e-01],
[ 5.5280e-01, -9.5422e-02, -8.8366e-01, ..., -2.6461e-01,
-3.4584e-01, 2.0425e-01],
...,
[-8.4358e-01, -1.7698e+00, -1.8164e+00, ..., -1.8012e+00,
-1.3232e+00, -9.6457e-01],
[ 7.5260e-01, 6.1695e-02, -5.8647e-02, ..., -1.2638e-02,
7.0899e-01, 7.8898e-01],
[ 1.4826e-01, -5.0658e-01, -4.1014e-01, ..., -1.2926e-01,
4.5515e-01, 3.6249e-01]],
[[-3.3612e+00, -1.8990e+00, -6.3878e+00, ..., -7.1900e+00,
-6.3097e+00, -4.6665e+00],
[-8.7484e-01, 1.4122e+00, -4.5034e+00, ..., -5.4876e+00,
-5.5900e+00, -4.3090e+00],
[-4.7874e-01, 9.7542e-01, -7.5176e+00, ..., -7.9994e+00,
-7.9004e+00, -6.7174e+00],
...,
[ 1.9359e-01, 2.9963e-01, -4.0423e+00, ..., -4.6557e+00,
-4.2193e+00, -3.4845e+00],
[ 2.2345e+00, 2.2440e+00, -1.5623e-01, ..., -8.1238e-01,
-1.1782e+00, -1.2048e+00],
[ 1.3024e+00, 1.1533e+00, -1.7621e+00, ..., -2.2280e+00,
-2.5030e+00, -2.4042e+00]],
[[-1.3734e+00, -2.3502e+00, -2.3139e+00, ..., -1.7981e+00,
-1.2717e+00, -1.4006e+00],
[-1.6625e+00, -1.9529e+00, -1.5071e+00, ..., -1.2832e+00,
-7.6024e-01, -8.4709e-01],
[-2.1657e+00, -1.9479e+00, -1.0176e+00, ..., -1.1334e+00,
-4.8532e-01, -1.9719e-01],
...,
[-1.2613e-01, -5.9986e-01, -4.7152e-01, ..., -5.0624e-01,
-1.9952e-01, -4.4735e-01],
[-6.6767e-01, -1.8045e-01, -4.2385e-01, ..., -2.0790e-01,
2.4820e-01, 2.4241e-01],
[-7.8075e-01, -1.8094e-01, -3.2891e-01, ..., -3.8793e-01,
-1.8593e-01, -2.1471e-01]],
...,
[[-2.0242e+00, -1.1233e+00, -2.3200e+00, ..., -3.1541e+00,
-2.6135e+00, -1.8315e+00],
[-1.8477e+00, -1.0485e-01, -1.4085e+00, ..., -2.5016e+00,
-2.5312e+00, -1.9740e+00],
[-4.6424e-01, 2.2054e+00, 9.7747e-01, ..., 1.3969e-01,
2.8976e-01, 5.4695e-01],
...,
[-2.0650e+00, -1.1909e-01, -2.0296e-01, ..., -1.3560e+00,
-3.9059e-01, 1.1460e-01],
[-1.2072e+00, 7.6492e-01, 9.1704e-01, ..., 1.7227e-03,
3.0456e-01, 4.7144e-01],
[-2.1084e+00, -1.4509e+00, -1.6581e+00, ..., -2.4649e+00,
-1.4639e+00, -9.3579e-01]],
[[-2.4768e+00, -2.9062e+00, -4.2023e+00, ..., -2.4306e+00,
-1.9337e+00, -1.7502e+00],
[-1.8497e+00, -1.7599e+00, -4.8995e+00, ..., -2.9395e+00,
-2.8085e+00, -2.7968e+00],
[-2.9091e+00, -2.6037e+00, -6.6684e+00, ..., -4.4649e+00,
-4.4060e+00, -3.5066e+00],
...,
[-1.4211e+00, -1.0762e-01, -3.3925e+00, ..., -1.4610e+00,
-1.0358e+00, -9.2194e-01],
[-1.7086e+00, -6.7760e-01, -3.0557e+00, ..., -1.7444e+00,
-1.1598e+00, -7.9681e-01],
[-2.3037e+00, -1.5035e+00, -3.1173e+00, ..., -2.9656e+00,
-2.1296e+00, -1.9687e+00]],
[[-1.8165e+00, -2.5528e+00, -2.4964e+00, ..., -2.7479e+00,
-2.2498e+00, -1.7237e+00],
[-1.8146e+00, -2.5251e+00, -2.2032e+00, ..., -2.4553e+00,
-1.9382e+00, -1.5675e+00],
[-2.2653e+00, -2.8373e+00, -2.5636e+00, ..., -2.8021e+00,
-2.1876e+00, -1.6609e+00],
...,
[-2.3632e+00, -2.9349e+00, -2.7252e+00, ..., -2.8505e+00,
-2.2831e+00, -1.7535e+00],
[-1.5342e+00, -1.9997e+00, -1.9343e+00, ..., -1.9310e+00,
-1.5963e+00, -1.2886e+00],
[-1.0033e+00, -1.2925e+00, -1.3311e+00, ..., -1.1749e+00,
-1.0086e+00, -7.6355e-01]]],
[[[ 9.0457e-02, 2.8517e-01, -1.5067e+00, ..., 5.4015e-01,
1.0088e+00, 6.8860e-01],
[-2.2358e+00, -1.6339e+00, -4.8780e+00, ..., -2.1071e+00,
-5.1832e-01, 9.7620e-02],
[ 3.7614e-01, -4.1886e-01, -1.4216e+00, ..., -1.4115e-02,
6.0721e-01, 7.2372e-01],
...,
[-3.1361e-01, -1.3371e+00, -2.0206e+00, ..., -2.0866e+00,
-1.5219e+00, -1.0705e+00],
[ 1.2548e+00, 6.1141e-01, -1.1487e+00, ..., -2.1254e-01,
5.3386e-01, 7.2327e-01],
[ 7.8214e-01, 1.9846e-01, -6.3341e-01, ..., -1.7951e-01,
4.0341e-01, 3.5946e-01]],
[[-3.3639e+00, -1.5008e+00, -6.3519e+00, ..., -4.7785e+00,
-4.3350e+00, -3.3442e+00],
[-6.2248e-01, 2.1826e+00, -4.3625e+00, ..., -3.5345e+00,
-3.6393e+00, -2.9867e+00],
[-5.5246e-02, 1.8496e+00, -7.6532e+00, ..., -5.4413e+00,
-5.3084e+00, -4.9151e+00],
...,
[-1.4549e+00, 1.4486e-01, -7.1172e+00, ..., -4.6347e+00,
-4.1329e+00, -3.4437e+00],
[ 1.4111e+00, 2.5882e+00, -1.4581e+00, ..., -5.3970e-01,
-8.8802e-01, -1.0265e+00],
[ 9.6263e-01, 1.6846e+00, -2.4374e+00, ..., -2.1342e+00,
-2.4139e+00, -2.3686e+00]],
[[-1.5504e+00, -2.4895e+00, -2.2802e+00, ..., -5.7205e-01,
-2.0890e-01, -4.3793e-01],
[-2.1763e+00, -2.3018e+00, -1.6218e+00, ..., -3.6680e-01,
2.3109e-02, 1.8117e-02],
[-2.7899e+00, -2.1870e+00, -9.0041e-01, ..., -2.5666e-01,
2.1637e-01, 2.5262e-01],
...,
[-1.6718e+00, -1.8803e+00, -1.0938e+00, ..., -5.4626e-01,
-2.1405e-01, -4.1518e-01],
[-1.8382e+00, -1.2324e+00, -7.6995e-01, ..., -2.8796e-01,
1.6999e-01, 2.2546e-01],
[-1.8350e+00, -1.2424e+00, -7.1683e-01, ..., -4.3661e-01,
-2.3325e-01, -2.2000e-01]],
...,
[[-2.3855e+00, -1.2909e+00, -2.7228e+00, ..., -3.4243e+00,
-2.5996e+00, -1.6756e+00],
[-2.0742e+00, 2.8794e-02, -1.4664e+00, ..., -2.4664e+00,
-2.1825e+00, -1.6752e+00],
[-5.2643e-01, 2.5177e+00, 9.8975e-01, ..., -6.1868e-01,
-5.5224e-02, 1.9745e-01],
...,
[-1.8714e+00, 1.0159e+00, 7.1244e-01, ..., -1.5242e+00,
-6.0559e-01, -5.4763e-03],
[-8.9985e-01, 1.6708e+00, 1.6046e+00, ..., -9.0596e-02,
2.0257e-01, 4.2182e-01],
[-2.1556e+00, -1.0784e+00, -7.6106e-01, ..., -2.5727e+00,
-1.5252e+00, -9.1781e-01]],
[[-2.4956e+00, -2.8208e+00, -4.1896e+00, ..., -1.2218e+00,
-7.9976e-01, -7.5030e-01],
[-1.7855e+00, -1.5103e+00, -4.9120e+00, ..., -1.5406e+00,
-1.3282e+00, -1.5762e+00],
[-3.1099e+00, -2.5633e+00, -6.9618e+00, ..., -3.4055e+00,
-2.8342e+00, -2.2139e+00],
...,
[-2.6217e+00, -1.8735e+00, -4.7356e+00, ..., -1.4549e+00,
-9.9297e-01, -8.1868e-01],
[-2.2902e+00, -1.6696e+00, -3.9202e+00, ..., -1.7191e+00,
-1.0877e+00, -6.5586e-01],
[-2.3493e+00, -1.8607e+00, -3.1010e+00, ..., -2.8639e+00,
-2.0045e+00, -1.8052e+00]],
[[-2.0012e+00, -2.7576e+00, -2.5406e+00, ..., -1.7409e+00,
-1.4534e+00, -1.1027e+00],
[-2.0265e+00, -2.7843e+00, -2.2371e+00, ..., -1.4024e+00,
-1.1251e+00, -9.0082e-01],
[-2.5716e+00, -3.1833e+00, -2.6855e+00, ..., -1.6366e+00,
-1.2458e+00, -9.3523e-01],
...,
[-3.2188e+00, -4.0398e+00, -3.9236e+00, ..., -2.9595e+00,
-2.3705e+00, -1.8119e+00],
[-2.1260e+00, -2.7646e+00, -2.7309e+00, ..., -2.0326e+00,
-1.6786e+00, -1.3543e+00],
[-1.5392e+00, -1.8814e+00, -1.9241e+00, ..., -1.2307e+00,
-1.0639e+00, -8.0675e-01]]],
[[[ 3.8759e-01, 8.1465e-01, 4.8802e-01, ..., 4.8732e-01,
1.0225e+00, 6.6372e-01],
[-1.2429e+00, -4.0846e-01, -1.8229e+00, ..., -2.5301e+00,
-7.3956e-01, -3.1265e-02],
[ 7.9129e-01, 6.1310e-01, 3.7296e-01, ..., -2.4804e-01,
4.9504e-01, 6.5458e-01],
...,
[ 8.3509e-03, -6.9862e-01, -1.6241e+00, ..., -1.9596e+00,
-1.3512e+00, -9.0225e-01],
[ 1.4675e+00, 1.2305e+00, 5.3135e-01, ..., -2.7676e-01,
5.2916e-01, 7.0577e-01],
[ 1.0919e+00, 9.6673e-01, 5.6334e-01, ..., -3.0427e-01,
3.2312e-01, 2.9990e-01]],
[[-2.7376e+00, -1.6261e+00, -4.8232e+00, ..., -4.8631e+00,
-4.4396e+00, -3.3844e+00],
[-6.8423e-01, 1.2499e+00, -3.1523e+00, ..., -3.3883e+00,
-3.5503e+00, -2.9166e+00],
[-8.5244e-03, 1.0043e+00, -5.7849e+00, ..., -5.3792e+00,
-5.2760e+00, -4.9502e+00],
...,
[-7.4528e-01, 3.5704e-01, -4.6777e+00, ..., -4.6132e+00,
-4.1453e+00, -3.4590e+00],
[ 2.1220e+00, 2.8260e+00, 2.7398e-01, ..., -9.1190e-01,
-1.2164e+00, -1.2544e+00],
[ 1.2782e+00, 1.5874e+00, -1.5653e+00, ..., -2.2656e+00,
-2.4784e+00, -2.3715e+00]],
[[-6.2280e-01, -1.7242e+00, -1.6122e+00, ..., -6.1092e-01,
-2.1073e-01, -4.7300e-01],
[-1.2999e+00, -1.7509e+00, -1.1588e+00, ..., -3.6199e-01,
5.0108e-02, 3.0807e-02],
[-1.2423e+00, -1.2705e+00, -7.3349e-01, ..., -2.6411e-01,
2.4989e-01, 2.8792e-01],
...,
[-1.0566e+00, -1.6226e+00, -7.4430e-01, ..., -5.0156e-01,
-1.7206e-01, -3.4944e-01],
[-1.3799e+00, -1.0115e+00, -6.5960e-01, ..., -3.1855e-01,
1.4482e-01, 2.0814e-01],
[-1.4491e+00, -8.3087e-01, -3.5685e-01, ..., -4.9678e-01,
-2.6771e-01, -2.4289e-01]],
...,
[[-2.0474e+00, -1.0679e+00, -2.3774e+00, ..., -3.6657e+00,
-2.8576e+00, -1.8510e+00],
[-5.7057e-01, 1.0901e+00, -7.9001e-01, ..., -2.6196e+00,
-2.4120e+00, -1.8780e+00],
[-1.7832e-01, 2.3485e+00, 6.7819e-01, ..., -5.8071e-01,
-7.4903e-02, 1.5933e-01],
...,
[-1.4933e+00, 1.5547e+00, 1.4260e-01, ..., -1.4966e+00,
-6.8475e-01, -1.3008e-01],
[-5.5116e-01, 1.6989e+00, 9.8852e-01, ..., -1.2230e-01,
1.0174e-01, 3.2882e-01],
[-1.7273e+00, -1.1625e+00, -1.6175e+00, ..., -2.3000e+00,
-1.4350e+00, -8.9720e-01]],
[[-1.4234e+00, -1.4719e+00, -3.0062e+00, ..., -1.1091e+00,
-6.9525e-01, -7.3112e-01],
[-7.7876e-01, -4.4764e-01, -3.8745e+00, ..., -1.3362e+00,
-1.2118e+00, -1.5851e+00],
[-2.1777e+00, -1.6923e+00, -6.2248e+00, ..., -3.2738e+00,
-2.8257e+00, -2.2487e+00],
...,
[-1.7558e+00, -7.1982e-01, -4.8146e+00, ..., -1.5076e+00,
-1.0737e+00, -8.9697e-01],
[-1.6897e+00, -8.1619e-01, -3.8420e+00, ..., -1.7446e+00,
-1.1705e+00, -7.6265e-01],
[-2.1032e+00, -1.3714e+00, -3.0749e+00, ..., -2.7909e+00,
-2.0184e+00, -1.8199e+00]],
[[-1.2206e+00, -1.8435e+00, -1.7237e+00, ..., -1.8171e+00,
-1.5018e+00, -1.1442e+00],
[-1.2357e+00, -1.7804e+00, -1.4395e+00, ..., -1.4081e+00,
-1.1128e+00, -8.8913e-01],
[-1.6082e+00, -1.9957e+00, -1.6878e+00, ..., -1.6034e+00,
-1.1983e+00, -8.9219e-01],
...,
[-2.4707e+00, -3.1275e+00, -2.9498e+00, ..., -2.8629e+00,
-2.2849e+00, -1.7541e+00],
[-1.6579e+00, -2.1363e+00, -2.0980e+00, ..., -1.9814e+00,
-1.6226e+00, -1.3083e+00],
[-1.1293e+00, -1.3806e+00, -1.4699e+00, ..., -1.2262e+00,
-1.0337e+00, -7.9218e-01]]],
[[[ 2.8657e-01, 5.5964e-01, 6.7220e-01, ..., 4.8514e-01,
1.0225e+00, 5.8683e-01],
[-2.1011e+00, -1.4476e+00, -3.1027e+00, ..., -2.7840e+00,
-9.1488e-01, -2.1960e-01],
[-1.6760e-01, -9.4195e-01, -2.0529e-01, ..., -1.4309e-01,
5.6570e-01, 6.4504e-01],
...,
[ 1.2165e-01, -6.5295e-01, -1.5818e+00, ..., -1.0802e+00,
-8.0199e-01, -5.8746e-01],
[ 1.5369e+00, 1.2880e+00, 5.9477e-01, ..., 7.1681e-01,
9.2696e-01, 9.4856e-01],
[ 1.1474e+00, 1.0339e+00, 6.7996e-01, ..., 8.9964e-01,
1.0355e+00, 8.7948e-01]],
[[-1.7090e+00, -1.2306e+00, -4.1965e+00, ..., -5.0625e+00,
-4.5699e+00, -3.4672e+00],
[ 4.6201e-01, 1.7259e+00, -2.3763e+00, ..., -3.5246e+00,
-3.6316e+00, -2.9459e+00],
[ 1.1464e+00, 1.4599e+00, -4.9746e+00, ..., -5.4953e+00,
-5.3407e+00, -5.0148e+00],
...,
[-6.3241e-01, 5.5190e-01, -4.7850e+00, ..., -5.8771e+00,
-5.2546e+00, -4.2374e+00],
[ 2.1464e+00, 2.8725e+00, 1.0792e-01, ..., -1.5048e+00,
-1.7284e+00, -1.6683e+00],
[ 1.2708e+00, 1.5754e+00, -1.6989e+00, ..., -2.7052e+00,
-2.8721e+00, -2.7304e+00]],
[[-1.0343e-01, -1.0860e+00, -1.2633e+00, ..., -5.4738e-01,
-1.9692e-01, -5.3183e-01],
[-5.3882e-01, -7.8545e-01, -7.3584e-01, ..., -3.4142e-01,
3.7745e-03, -1.0491e-01],
[-4.8732e-01, -3.1199e-01, -3.9675e-01, ..., -3.7161e-01,
1.0775e-01, 9.6277e-02],
...,
[-1.1096e+00, -1.5782e+00, -6.5411e-01, ..., -6.2403e-01,
-4.7758e-01, -6.3397e-01],
[-1.3389e+00, -8.8405e-01, -4.7488e-01, ..., -1.0690e-01,
9.3684e-02, 6.5893e-02],
[-1.3852e+00, -6.8816e-01, -1.4810e-01, ..., -6.1803e-02,
-3.4177e-02, -6.1060e-02]],
...,
[[-2.9205e+00, -2.4624e+00, -2.7477e+00, ..., -3.6500e+00,
-2.7748e+00, -1.8536e+00],
[-2.2103e+00, -8.7608e-01, -1.5970e+00, ..., -2.6711e+00,
-2.3931e+00, -1.8733e+00],
[-1.0626e+00, 1.0424e+00, 5.2531e-01, ..., -4.8497e-01,
3.4817e-02, 2.0987e-01],
...,
[-1.5889e+00, 1.6289e+00, 3.4738e-01, ..., -7.9632e-01,
-1.6097e-01, 3.8052e-01],
[-6.3114e-01, 1.7593e+00, 1.1322e+00, ..., 1.9378e-01,
3.3597e-01, 5.1605e-01],
[-1.7291e+00, -1.0396e+00, -1.4078e+00, ..., -2.1101e+00,
-1.1758e+00, -5.1767e-01]],
[[-1.5903e+00, -1.5798e+00, -2.7905e+00, ..., -1.1942e+00,
-7.1029e-01, -7.2802e-01],
[-6.4216e-01, 7.8229e-04, -3.1247e+00, ..., -1.2952e+00,
-1.1223e+00, -1.5450e+00],
[-2.1114e+00, -9.9538e-01, -5.2033e+00, ..., -3.2530e+00,
-2.7269e+00, -2.2313e+00],
...,
[-1.7761e+00, -7.0798e-01, -4.7492e+00, ..., -2.4663e+00,
-2.0417e+00, -1.6202e+00],
[-1.6786e+00, -7.8527e-01, -3.7923e+00, ..., -2.2824e+00,
-1.6801e+00, -1.1111e+00],
[-2.0565e+00, -1.3169e+00, -3.0262e+00, ..., -2.7523e+00,
-1.9549e+00, -1.6866e+00]],
[[-1.3220e+00, -1.8741e+00, -1.6971e+00, ..., -1.8154e+00,
-1.5097e+00, -1.1518e+00],
[-1.2495e+00, -1.7298e+00, -1.2470e+00, ..., -1.3843e+00,
-1.0921e+00, -8.7749e-01],
[-1.5617e+00, -1.9110e+00, -1.4938e+00, ..., -1.5930e+00,
-1.1875e+00, -8.7680e-01],
...,
[-2.5164e+00, -3.1381e+00, -2.8649e+00, ..., -3.0211e+00,
-2.4132e+00, -1.8490e+00],
[-1.6904e+00, -2.1425e+00, -2.0399e+00, ..., -2.1106e+00,
-1.7304e+00, -1.3877e+00],
[-1.1350e+00, -1.3526e+00, -1.3893e+00, ..., -1.3071e+00,
-1.1085e+00, -8.2671e-01]]]], device='cuda:0'),
tensor([[[[ 8.3912e-02, 3.6025e-01, -1.1797e+00, ..., 6.4274e-01,
9.2536e-01, 6.7361e-01],
[-2.1965e+00, -1.4611e+00, -4.3337e+00, ..., -1.5230e+00,
-2.1005e-01, 3.7008e-01],
[ 3.2389e-01, -3.0752e-01, -1.1315e+00, ..., 5.0632e-01,
9.1029e-01, 8.3432e-01],
...,
[-2.7393e-01, -1.2592e+00, -2.6212e+00, ..., -1.2069e+00,
-7.9843e-01, -6.2484e-01],
[ 1.2651e+00, 6.6889e-01, -9.1833e-01, ..., 5.0768e-01,
7.7004e-01, 7.9875e-01],
[ 8.2616e-01, 2.9835e-01, -4.4125e-01, ..., 5.4301e-01,
7.8980e-01, 6.9017e-01]],
[[-3.2048e+00, -1.4136e+00, -6.0771e+00, ..., -5.4775e+00,
-4.6899e+00, -3.6492e+00],
[-5.4332e-01, 2.1585e+00, -4.1711e+00, ..., -4.1878e+00,
-4.0606e+00, -3.3462e+00],
[ 2.6446e-02, 1.8526e+00, -7.4217e+00, ..., -6.1231e+00,
-5.8698e+00, -5.4710e+00],
...,
[-1.2960e+00, 2.0112e-01, -6.5687e+00, ..., -5.3864e+00,
-4.7260e+00, -3.9418e+00],
[ 1.4432e+00, 2.5305e+00, -1.4241e+00, ..., -1.3019e+00,
-1.4415e+00, -1.5250e+00],
[ 9.5941e-01, 1.5837e+00, -2.4492e+00, ..., -2.5206e+00,
-2.6252e+00, -2.5856e+00]],
[[-1.3470e+00, -2.2850e+00, -2.1338e+00, ..., -8.6084e-01,
-6.3113e-01, -8.2041e-01],
[-2.0006e+00, -2.1331e+00, -1.5097e+00, ..., -7.0484e-01,
-4.3043e-01, -3.6429e-01],
[-2.5463e+00, -2.0116e+00, -8.3540e-01, ..., -4.0914e-01,
-1.0798e-01, -9.4080e-02],
...,
[-1.6149e+00, -1.8130e+00, -7.9145e-01, ..., -6.3022e-01,
-5.4605e-01, -7.0347e-01],
[-1.7637e+00, -1.1695e+00, -7.6721e-01, ..., -2.4022e-01,
-7.0097e-02, -6.4723e-02],
[-1.6961e+00, -1.0713e+00, -5.6323e-01, ..., -3.3205e-01,
-2.9608e-01, -2.6563e-01]],
...,
[[-2.3081e+00, -1.3139e+00, -2.7627e+00, ..., -3.1042e+00,
-2.4723e+00, -1.6502e+00],
[-1.8563e+00, 1.3442e-01, -1.3659e+00, ..., -1.4998e+00,
-1.5757e+00, -1.3784e+00],
[-4.4526e-01, 2.4993e+00, 8.8590e-01, ..., -8.6969e-02,
2.5156e-01, 3.2381e-01],
...,
[-1.6817e+00, 1.1861e+00, 2.4028e-01, ..., -9.7739e-01,
-3.0098e-01, 1.6891e-01],
[-8.5647e-01, 1.6354e+00, 1.5180e+00, ..., 1.6448e-01,
3.0150e-01, 3.9128e-01],
[-2.0850e+00, -1.0442e+00, -8.4602e-01, ..., -2.1207e+00,
-1.2138e+00, -6.2750e-01]],
[[-2.3538e+00, -2.6150e+00, -4.0273e+00, ..., -1.1028e+00,
-5.9195e-01, -4.7928e-01],
[-1.6413e+00, -1.3163e+00, -4.7964e+00, ..., -1.6697e+00,
-1.4722e+00, -1.7137e+00],
[-2.9747e+00, -2.3931e+00, -6.9011e+00, ..., -3.7307e+00,
-3.2653e+00, -2.6281e+00],
...,
[-2.5734e+00, -1.8079e+00, -4.9817e+00, ..., -2.1975e+00,
-1.7536e+00, -1.4254e+00],
[-2.2343e+00, -1.5824e+00, -3.8716e+00, ..., -2.1226e+00,
-1.4956e+00, -9.9278e-01],
[-2.2966e+00, -1.7726e+00, -3.0963e+00, ..., -2.7996e+00,
-1.9541e+00, -1.6788e+00]],
[[-1.9051e+00, -2.6359e+00, -2.4266e+00, ..., -1.7005e+00,
-1.4096e+00, -1.0523e+00],
[-1.9442e+00, -2.6638e+00, -2.1300e+00, ..., -1.3672e+00,
-1.1078e+00, -8.7421e-01],
[-2.4796e+00, -3.0499e+00, -2.5471e+00, ..., -1.5644e+00,
-1.1945e+00, -8.6420e-01],
...,
[-3.1075e+00, -3.8799e+00, -3.6587e+00, ..., -2.7775e+00,
-2.2180e+00, -1.6751e+00],
[-2.0735e+00, -2.6791e+00, -2.6480e+00, ..., -1.9618e+00,
-1.6096e+00, -1.2794e+00],
[-1.4960e+00, -1.8187e+00, -1.8622e+00, ..., -1.2256e+00,
-1.0451e+00, -7.6793e-01]]],
[[[ 2.8617e-01, 5.5312e-01, 6.7225e-01, ..., -1.9271e+01,
-1.4144e+01, -1.0233e+01],
[-2.1221e+00, -1.4606e+00, -3.0852e+00, ..., -2.3388e+01,
-1.8225e+01, -1.4126e+01],
[-1.9288e-01, -9.4882e-01, -1.9727e-01, ..., -2.1956e+01,
-1.6391e+01, -1.2698e+01],
...,
[-1.8666e-01, -1.1570e+00, -2.7118e+00, ..., -1.5048e+00,
-1.4441e+00, -8.8594e-01],
[ 1.4335e+00, 8.3463e-01, -1.0091e+00, ..., -2.8925e-01,
7.6775e-02, 3.4086e-01],
[ 8.4562e-01, 2.8264e-01, -7.2938e-01, ..., -2.0170e-01,
2.0170e-01, 3.6854e-01]],
[[-1.6783e+00, -1.2260e+00, -4.1870e+00, ..., -1.0532e+01,
-9.9467e+00, -7.7896e+00],
[ 4.7839e-01, 1.7261e+00, -2.3661e+00, ..., -7.5129e+00,
-8.3571e+00, -7.5287e+00],
[ 1.1652e+00, 1.4516e+00, -4.9680e+00, ..., -9.9065e+00,
-1.0157e+01, -8.7007e+00],
...,
[-1.3084e+00, 1.2049e-01, -6.7255e+00, ..., -7.6565e+00,
-6.9591e+00, -5.4438e+00],
[ 1.7826e+00, 2.8200e+00, -1.0400e+00, ..., -2.6215e+00,
-2.9870e+00, -2.3726e+00],
[ 1.2053e+00, 1.8183e+00, -2.2695e+00, ..., -3.2576e+00,
-3.6338e+00, -3.1322e+00]],
[[-9.7647e-02, -1.0825e+00, -1.2610e+00, ..., -1.7810e+00,
-1.2511e+00, -2.0013e+00],
[-5.3881e-01, -7.9122e-01, -7.3984e-01, ..., -2.0736e-01,
3.1941e-01, -7.9280e-01],
[-4.8880e-01, -3.0518e-01, -3.9601e-01, ..., -7.3075e-01,
-4.0409e-01, -2.7254e-01],
...,
[-1.4146e+00, -1.7579e+00, -7.6605e-01, ..., -8.8665e-01,
-7.7186e-01, -1.0170e+00],
[-1.7517e+00, -1.1594e+00, -8.2865e-01, ..., -5.6445e-01,
-2.9331e-01, -3.6529e-01],
[-1.8989e+00, -1.2946e+00, -9.1008e-01, ..., -9.7332e-01,
-8.4569e-01, -8.1983e-01]],
...,
[[-2.9284e+00, -2.4990e+00, -2.7612e+00, ..., -3.2655e+00,
-8.7897e-01, -9.7247e-02],
[-2.2182e+00, -9.1338e-01, -1.6145e+00, ..., -5.1161e+00,
-3.4956e+00, -1.8264e+00],
[-1.0357e+00, 1.0170e+00, 5.0335e-01, ..., -3.0316e+00,
-4.1411e-01, -2.0368e+00],
...,
[-1.8352e+00, 1.2471e+00, 3.6648e-02, ..., -1.1170e+00,
-5.4644e-01, -8.8865e-03],
[-7.9320e-01, 1.9268e+00, 1.6859e+00, ..., 4.8435e-01,
4.1518e-01, 6.7443e-01],
[-2.2223e+00, -1.1712e+00, -9.7936e-01, ..., -1.8646e+00,
-9.6509e-01, -3.0374e-01]],
[[-1.5931e+00, -1.6006e+00, -2.8224e+00, ..., 4.7472e-01,
5.3100e-01, -1.0382e-01],
[-6.4387e-01, -1.3432e-02, -3.1232e+00, ..., 7.9495e-01,
1.1756e+00, 1.2945e-01],
[-2.1080e+00, -1.0018e+00, -5.1834e+00, ..., 3.1207e-01,
2.9460e-01, -3.9429e-01],
...,
[-2.4857e+00, -1.5762e+00, -5.4445e+00, ..., -3.0871e+00,
-2.7210e+00, -2.1766e+00],
[-2.2270e+00, -1.4682e+00, -4.2425e+00, ..., -2.6925e+00,
-2.3055e+00, -1.6361e+00],
[-2.4008e+00, -1.7772e+00, -3.3085e+00, ..., -2.9773e+00,
-2.4334e+00, -2.1240e+00]],
[[-1.3210e+00, -1.8693e+00, -1.6990e+00, ..., -4.5545e+00,
-3.1317e+00, -1.8617e+00],
[-1.2508e+00, -1.7242e+00, -1.2498e+00, ..., -4.2487e+00,
-3.5407e+00, -2.3985e+00],
[-1.5657e+00, -1.9128e+00, -1.5016e+00, ..., -3.0212e+00,
-2.4236e+00, -2.0457e+00],
...,
[-3.1935e+00, -3.9982e+00, -3.7864e+00, ..., -3.9696e+00,
-3.1857e+00, -2.4637e+00],
[-2.1240e+00, -2.7472e+00, -2.7130e+00, ..., -2.7825e+00,
-2.2981e+00, -1.8729e+00],
[-1.5458e+00, -1.8778e+00, -1.9392e+00, ..., -1.8777e+00,
-1.5265e+00, -1.1702e+00]]],
[[[ 2.8226e-01, 5.6612e-01, 6.7446e-01, ..., 4.4918e-01,
1.0101e+00, 6.1410e-01],
[-2.2277e+00, -1.5352e+00, -3.2502e+00, ..., -2.7118e+00,
-8.8682e-01, -1.9057e-01],
[-1.9949e-01, -1.0336e+00, -2.4564e-01, ..., -1.7588e-01,
5.2739e-01, 6.5418e-01],
...,
[ 8.5687e-02, -6.7938e-01, -1.5174e+00, ..., -1.8924e+00,
-1.2579e+00, -6.9748e-01],
[ 1.5013e+00, 1.2222e+00, 6.1250e-01, ..., 7.0074e-03,
3.4219e-01, 7.7932e-01],
[ 1.1398e+00, 9.8835e-01, 6.9411e-01, ..., 4.8144e-01,
5.8329e-01, 7.4009e-01]],
[[-1.6898e+00, -1.2023e+00, -4.1650e+00, ..., -4.8824e+00,
-4.4444e+00, -3.4071e+00],
[ 5.5404e-01, 1.8754e+00, -2.2842e+00, ..., -3.5080e+00,
-3.6189e+00, -2.9901e+00],
[ 1.2922e+00, 1.6306e+00, -5.0008e+00, ..., -5.4707e+00,
-5.2738e+00, -4.9320e+00],
...,
[-7.2660e-01, 4.3993e-01, -4.7348e+00, ..., -6.0900e+00,
-5.3355e+00, -4.2318e+00],
[ 2.0710e+00, 2.7875e+00, 1.7273e-01, ..., -1.7513e+00,
-1.8259e+00, -1.6883e+00],
[ 1.2182e+00, 1.5164e+00, -1.6538e+00, ..., -2.8390e+00,
-2.9239e+00, -2.7333e+00]],
[[-1.0114e-01, -1.1163e+00, -1.3267e+00, ..., -5.0960e-01,
-1.6244e-01, -4.8098e-01],
[-5.5617e-01, -7.9145e-01, -7.5875e-01, ..., -2.9066e-01,
3.5499e-02, -7.1698e-02],
[-4.6320e-01, -2.5835e-01, -3.5264e-01, ..., -3.5866e-01,
8.1236e-02, 7.8856e-02],
...,
[-1.0457e+00, -1.5355e+00, -6.2992e-01, ..., -6.2981e-01,
-4.3947e-01, -6.1706e-01],
[-1.2687e+00, -8.4305e-01, -4.3451e-01, ..., 5.6909e-02,
1.5844e-01, 8.9536e-02],
[-1.3269e+00, -6.6704e-01, -1.4202e-01, ..., 2.8735e-01,
1.2025e-01, -2.1288e-02]],
...,
[[-3.0060e+00, -2.5206e+00, -2.7634e+00, ..., -3.4824e+00,
-2.6323e+00, -1.7519e+00],
[-2.2726e+00, -9.3685e-01, -1.6236e+00, ..., -2.6313e+00,
-2.3259e+00, -1.8308e+00],
[-1.1014e+00, 1.0748e+00, 5.5242e-01, ..., -4.6029e-01,
-5.5787e-03, 1.4763e-01],
...,
[-1.6758e+00, 1.4971e+00, 2.5844e-01, ..., -1.1820e+00,
-3.7541e-01, 3.1464e-01],
[-6.6648e-01, 1.6884e+00, 1.0421e+00, ..., 3.8281e-03,
2.6082e-01, 4.6081e-01],
[-1.7622e+00, -1.0910e+00, -1.4887e+00, ..., -2.1824e+00,
-1.1721e+00, -5.4442e-01]],
[[-1.6017e+00, -1.5916e+00, -2.8032e+00, ..., -1.2526e+00,
-8.4133e-01, -8.5430e-01],
[-6.0560e-01, 9.5214e-02, -3.1124e+00, ..., -1.3598e+00,
-1.2241e+00, -1.5786e+00],
[-2.0868e+00, -8.9584e-01, -5.2633e+00, ..., -3.1998e+00,
-2.7086e+00, -2.2012e+00],
...,
[-1.7434e+00, -7.1928e-01, -4.5769e+00, ..., -2.4037e+00,
-2.0068e+00, -1.5947e+00],
[-1.6796e+00, -8.2042e-01, -3.7090e+00, ..., -2.5283e+00,
-1.7610e+00, -1.1107e+00],
[-2.0778e+00, -1.3589e+00, -3.0108e+00, ..., -3.0901e+00,
-2.0560e+00, -1.6964e+00]],
[[-1.3665e+00, -1.9307e+00, -1.7450e+00, ..., -1.7824e+00,
-1.4685e+00, -1.1163e+00],
[-1.2841e+00, -1.7762e+00, -1.2757e+00, ..., -1.3895e+00,
-1.0916e+00, -8.6924e-01],
[-1.6074e+00, -1.9557e+00, -1.5198e+00, ..., -1.6177e+00,
-1.2074e+00, -8.9536e-01],
...,
[-2.4503e+00, -3.0740e+00, -2.8270e+00, ..., -3.0408e+00,
-2.4036e+00, -1.8136e+00],
[-1.6325e+00, -2.0842e+00, -1.9952e+00, ..., -2.1328e+00,
-1.7321e+00, -1.3642e+00],
[-1.0857e+00, -1.3142e+00, -1.3540e+00, ..., -1.3373e+00,
-1.1062e+00, -8.0752e-01]]],
[[[ 5.0679e-01, 9.5924e-01, 6.9079e-01, ..., 8.0051e-01,
1.0524e+00, 8.4537e-01],
[-1.1592e+00, -2.0169e-01, -1.6866e+00, ..., -1.0455e+00,
1.6649e-01, 6.8625e-01],
[ 8.1709e-01, 5.4014e-01, 1.4960e-01, ..., 5.3469e-01,
8.9974e-01, 9.3915e-01],
...,
[-8.0270e-01, -1.7674e+00, -1.7804e+00, ..., -1.7682e+00,
-1.3313e+00, -9.5091e-01],
[ 7.8766e-01, 5.2960e-02, -6.6266e-02, ..., 4.0472e-02,
7.0228e-01, 7.9041e-01],
[ 1.8130e-01, -5.1412e-01, -4.2836e-01, ..., -8.6548e-02,
4.5539e-01, 3.5982e-01]],
[[-2.7038e+00, -1.3167e+00, -4.8052e+00, ..., -5.7240e+00,
-4.9074e+00, -3.8402e+00],
[-4.7004e-01, 1.8297e+00, -3.1277e+00, ..., -4.7765e+00,
-4.5654e+00, -3.7168e+00],
[ 3.5386e-01, 1.6856e+00, -6.0012e+00, ..., -6.8885e+00,
-6.5277e+00, -5.8907e+00],
...,
[ 2.3730e-01, 3.3250e-01, -4.0794e+00, ..., -4.6960e+00,
-4.2051e+00, -3.5144e+00],
[ 2.2457e+00, 2.2504e+00, -1.8578e-01, ..., -8.1309e-01,
-1.1522e+00, -1.2236e+00],
[ 1.3113e+00, 1.1630e+00, -1.7777e+00, ..., -2.2395e+00,
-2.5179e+00, -2.4170e+00]],
[[-6.2033e-01, -1.7628e+00, -1.5491e+00, ..., -7.0074e-01,
-4.8728e-01, -6.9066e-01],
[-1.4824e+00, -1.9193e+00, -1.2209e+00, ..., -6.1818e-01,
-3.3826e-01, -3.1117e-01],
[-1.3425e+00, -1.2187e+00, -5.3466e-01, ..., -4.0044e-01,
-8.4343e-02, -9.8221e-02],
...,
[-1.6018e-01, -6.1634e-01, -4.6468e-01, ..., -5.1651e-01,
-2.0728e-01, -4.4677e-01],
[-6.9556e-01, -1.9841e-01, -4.1311e-01, ..., -2.0796e-01,
2.3784e-01, 2.4099e-01],
[-7.8614e-01, -1.8505e-01, -3.2653e-01, ..., -3.7955e-01,
-1.8334e-01, -2.1763e-01]],
...,
[[-2.3018e+00, -1.2297e+00, -2.6354e+00, ..., -3.0767e+00,
-2.5473e+00, -1.7386e+00],
[-5.7667e-01, 1.3283e+00, -6.3913e-01, ..., -1.5676e+00,
-1.7267e+00, -1.4852e+00],
[-2.5777e-01, 2.5007e+00, 7.1066e-01, ..., -3.7855e-01,
-1.2698e-01, 6.4269e-02],
...,
[-2.0545e+00, -5.7423e-02, -1.6556e-01, ..., -1.3762e+00,
-3.8993e-01, 1.1108e-01],
[-1.1860e+00, 8.0038e-01, 9.2331e-01, ..., -6.0852e-05,
3.1866e-01, 4.8500e-01],
[-2.0981e+00, -1.4286e+00, -1.6376e+00, ..., -2.4929e+00,
-1.4849e+00, -9.2305e-01]],
[[-1.4605e+00, -1.3736e+00, -2.9713e+00, ..., -1.2234e+00,
-7.2933e-01, -5.7748e-01],
[-7.7599e-01, -2.6895e-01, -3.9627e+00, ..., -2.0508e+00,
-1.7734e+00, -1.8821e+00],
[-2.4334e+00, -1.7702e+00, -6.7002e+00, ..., -4.2068e+00,
-3.6536e+00, -2.8765e+00],
...,
[-1.4246e+00, -1.1245e-01, -3.4035e+00, ..., -1.4310e+00,
-1.0245e+00, -9.3890e-01],
[-1.7138e+00, -6.8115e-01, -3.0564e+00, ..., -1.7162e+00,
-1.1533e+00, -8.1523e-01],
[-2.3115e+00, -1.5058e+00, -3.1141e+00, ..., -2.9618e+00,
-2.1258e+00, -1.9830e+00]],
[[-1.2925e+00, -1.9124e+00, -1.6708e+00, ..., -1.8069e+00,
-1.4972e+00, -1.1193e+00],
[-1.3406e+00, -1.8850e+00, -1.3791e+00, ..., -1.6310e+00,
-1.3248e+00, -1.0373e+00],
[-1.7695e+00, -2.1383e+00, -1.6562e+00, ..., -1.9480e+00,
-1.5034e+00, -1.1064e+00],
...,
[-2.3706e+00, -2.9392e+00, -2.7182e+00, ..., -2.8748e+00,
-2.3045e+00, -1.7618e+00],
[-1.5403e+00, -2.0006e+00, -1.9336e+00, ..., -1.9495e+00,
-1.6161e+00, -1.2987e+00],
[-1.0033e+00, -1.2923e+00, -1.3289e+00, ..., -1.1903e+00,
-1.0184e+00, -7.7194e-01]]],
[[[ 3.0043e-01, 5.7091e-01, 6.9691e-01, ..., 4.8869e-01,
1.0189e+00, 6.0929e-01],
[-2.1498e+00, -1.5053e+00, -3.1768e+00, ..., -2.6195e+00,
-8.3993e-01, -1.6743e-01],
[-1.4340e-01, -9.9319e-01, -2.1294e-01, ..., -1.1976e-01,
5.8299e-01, 6.8526e-01],
...,
[-1.7233e-01, -1.1546e+00, -2.6713e+00, ..., -1.4962e+00,
-1.3952e+00, -8.2750e-01],
[ 1.4262e+00, 8.2016e-01, -1.0124e+00, ..., -2.9408e-01,
9.9657e-02, 3.9178e-01],
[ 8.4410e-01, 2.8665e-01, -7.3170e-01, ..., -1.9375e-01,
2.0601e-01, 3.9074e-01]],
[[-1.6905e+00, -1.1473e+00, -4.1805e+00, ..., -4.8794e+00,
-4.4553e+00, -3.3931e+00],
[ 5.8275e-01, 1.9524e+00, -2.2383e+00, ..., -3.5218e+00,
-3.6210e+00, -2.9688e+00],
[ 1.3337e+00, 1.7323e+00, -4.8878e+00, ..., -5.4643e+00,
-5.2877e+00, -4.9382e+00],
...,
[-1.2946e+00, 1.4708e-01, -6.6936e+00, ..., -7.6190e+00,
-6.9043e+00, -5.4453e+00],
[ 1.7806e+00, 2.8138e+00, -1.0435e+00, ..., -2.6249e+00,
-2.9803e+00, -2.4263e+00],
[ 1.2020e+00, 1.8074e+00, -2.2896e+00, ..., -3.2868e+00,
-3.6409e+00, -3.1867e+00]],
[[-1.1744e-01, -1.1520e+00, -1.3226e+00, ..., -4.9900e-01,
-1.6513e-01, -4.8134e-01],
[-5.9789e-01, -8.6044e-01, -7.7411e-01, ..., -2.6266e-01,
3.4454e-02, -7.0989e-02],
[-5.5607e-01, -3.5823e-01, -3.8298e-01, ..., -3.3606e-01,
9.4392e-02, 8.8057e-02],
...,
[-1.3896e+00, -1.7534e+00, -7.6375e-01, ..., -8.6792e-01,
-7.8401e-01, -1.0282e+00],
[-1.7398e+00, -1.1561e+00, -8.3760e-01, ..., -5.5055e-01,
-3.0285e-01, -3.7600e-01],
[-1.8824e+00, -1.2784e+00, -9.0306e-01, ..., -9.7766e-01,
-8.5712e-01, -8.2620e-01]],
...,
[[-3.0561e+00, -2.4960e+00, -2.8047e+00, ..., -3.4848e+00,
-2.6561e+00, -1.7713e+00],
[-2.3039e+00, -8.2060e-01, -1.5773e+00, ..., -2.6436e+00,
-2.3551e+00, -1.8409e+00],
[-1.1000e+00, 1.1758e+00, 5.8840e-01, ..., -4.9300e-01,
-6.9588e-02, 9.0866e-02],
...,
[-1.8602e+00, 1.2484e+00, 2.2491e-02, ..., -1.1050e+00,
-5.7656e-01, -2.6282e-02],
[-8.0286e-01, 1.9229e+00, 1.6797e+00, ..., 4.9175e-01,
4.0405e-01, 6.6402e-01],
[-2.2043e+00, -1.1364e+00, -9.5285e-01, ..., -1.8007e+00,
-9.2618e-01, -2.6273e-01]],
[[-1.6006e+00, -1.5813e+00, -2.9185e+00, ..., -1.2770e+00,
-8.3210e-01, -8.3491e-01],
[-5.9996e-01, 8.8496e-02, -3.2342e+00, ..., -1.3782e+00,
-1.2212e+00, -1.5843e+00],
[-2.0982e+00, -8.9968e-01, -5.3669e+00, ..., -3.1913e+00,
-2.7258e+00, -2.2382e+00],
...,
[-2.4499e+00, -1.5760e+00, -5.4696e+00, ..., -3.0684e+00,
-2.7335e+00, -2.2107e+00],
[-2.2022e+00, -1.4627e+00, -4.2509e+00, ..., -2.6605e+00,
-2.3230e+00, -1.6861e+00],
[-2.3747e+00, -1.7588e+00, -3.3171e+00, ..., -2.9299e+00,
-2.4473e+00, -2.1439e+00]],
[[-1.3506e+00, -1.9127e+00, -1.7089e+00, ..., -1.7668e+00,
-1.4630e+00, -1.1143e+00],
[-1.2766e+00, -1.7653e+00, -1.2354e+00, ..., -1.3746e+00,
-1.0825e+00, -8.6068e-01],
[-1.5984e+00, -1.9479e+00, -1.4738e+00, ..., -1.5938e+00,
-1.1942e+00, -8.8338e-01],
...,
[-3.1905e+00, -3.9841e+00, -3.7684e+00, ..., -3.9526e+00,
-3.1643e+00, -2.4391e+00],
[-2.1236e+00, -2.7369e+00, -2.7033e+00, ..., -2.7748e+00,
-2.2863e+00, -1.8604e+00],
[-1.5461e+00, -1.8730e+00, -1.9342e+00, ..., -1.8785e+00,
-1.5213e+00, -1.1641e+00]]]], device='cuda:0'),
tensor([[[[ 5.7812e-02, 1.9280e-01, -1.4896e+00, ..., -6.7631e-01,
-2.4758e-01, -8.7195e-02],
[-2.1263e+00, -1.6514e+00, -4.6414e+00, ..., -3.8806e+00,
-1.6036e+00, -6.1667e-01],
[ 5.6380e-01, -1.1587e-01, -9.0839e-01, ..., -2.4690e-01,
-3.0975e-01, 2.0117e-01],
...,
[-3.0063e-01, -1.0626e+00, -1.3822e+00, ..., -1.3240e+00,
-8.9861e-01, -6.9219e-01],
[ 9.8742e-01, 6.2464e-01, 3.6220e-01, ..., 4.3258e-01,
7.0155e-01, 7.7493e-01],
[ 6.8486e-01, 4.8082e-01, 3.4835e-01, ..., 5.4093e-01,
7.8824e-01, 6.8850e-01]],
[[-3.3453e+00, -1.8118e+00, -6.3142e+00, ..., -7.1895e+00,
-6.3033e+00, -4.6529e+00],
[-8.1831e-01, 1.5465e+00, -4.4085e+00, ..., -5.4872e+00,
-5.5917e+00, -4.3152e+00],
[-3.8218e-01, 1.0792e+00, -7.5130e+00, ..., -7.9929e+00,
-7.9060e+00, -6.7283e+00],
...,
[-4.6546e-01, 2.9928e-01, -4.4217e+00, ..., -5.4179e+00,
-4.7109e+00, -3.8813e+00],
[ 1.9415e+00, 2.4429e+00, -9.8147e-02, ..., -1.0946e+00,
-1.2319e+00, -1.3353e+00],
[ 1.1082e+00, 1.3138e+00, -1.7165e+00, ..., -2.4853e+00,
-2.6012e+00, -2.5597e+00]],
[[-1.3762e+00, -2.3886e+00, -2.3480e+00, ..., -1.7952e+00,
-1.2693e+00, -1.3968e+00],
[-1.7192e+00, -2.0115e+00, -1.5381e+00, ..., -1.2756e+00,
-7.5619e-01, -8.3795e-01],
[-2.2132e+00, -1.9675e+00, -1.0220e+00, ..., -1.1208e+00,
-4.8095e-01, -2.0055e-01],
...,
[-6.7345e-01, -1.1294e+00, -5.8048e-01, ..., -6.0320e-01,
-4.7982e-01, -6.4035e-01],
[-1.0745e+00, -7.2335e-01, -5.7594e-01, ..., -2.6850e-01,
-5.8092e-02, -2.8257e-02],
[-1.2257e+00, -6.9059e-01, -3.8763e-01, ..., -3.3613e-01,
-2.7674e-01, -2.4785e-01]],
...,
[[-2.0570e+00, -1.1114e+00, -2.3466e+00, ..., -3.1829e+00,
-2.6063e+00, -1.8217e+00],
[-1.8083e+00, -1.8376e-02, -1.3940e+00, ..., -2.5030e+00,
-2.5205e+00, -1.9573e+00],
[-3.9163e-01, 2.3659e+00, 1.0670e+00, ..., 1.3052e-01,
3.0303e-01, 5.5941e-01],
...,
[-1.5116e+00, 1.0088e+00, -1.2725e-01, ..., -1.2150e+00,
-4.4795e-01, 8.1066e-02],
[-5.2667e-01, 1.4185e+00, 7.6101e-01, ..., -1.1944e-02,
1.7566e-01, 3.1130e-01],
[-1.5619e+00, -1.0561e+00, -1.5056e+00, ..., -2.3138e+00,
-1.3678e+00, -7.2767e-01]],
[[-2.4470e+00, -2.8661e+00, -4.2242e+00, ..., -2.4205e+00,
-1.9233e+00, -1.7339e+00],
[-1.7921e+00, -1.6669e+00, -4.9127e+00, ..., -2.9489e+00,
-2.8067e+00, -2.7936e+00],
[-2.8570e+00, -2.4973e+00, -6.6922e+00, ..., -4.4948e+00,
-4.4048e+00, -3.5079e+00],
...,
[-1.6938e+00, -5.4859e-01, -4.1232e+00, ..., -2.0895e+00,
-1.5718e+00, -1.1890e+00],
[-1.6652e+00, -7.3191e-01, -3.4048e+00, ..., -2.0677e+00,
-1.3446e+00, -7.8672e-01],
[-2.0314e+00, -1.2740e+00, -2.8487e+00, ..., -2.7264e+00,
-1.8067e+00, -1.5383e+00]],
[[-1.8278e+00, -2.5704e+00, -2.4847e+00, ..., -2.7296e+00,
-2.2386e+00, -1.7164e+00],
[-1.8200e+00, -2.5414e+00, -2.1798e+00, ..., -2.4364e+00,
-1.9298e+00, -1.5632e+00],
[-2.2725e+00, -2.8481e+00, -2.5487e+00, ..., -2.7881e+00,
-2.1811e+00, -1.6617e+00],
...,
[-2.2799e+00, -2.8579e+00, -2.6579e+00, ..., -2.9022e+00,
-2.3183e+00, -1.7386e+00],
[-1.5467e+00, -1.9834e+00, -1.9344e+00, ..., -2.0565e+00,
-1.6891e+00, -1.3379e+00],
[-1.0532e+00, -1.2791e+00, -1.3425e+00, ..., -1.2742e+00,
-1.0843e+00, -7.9505e-01]]],
[[[ 4.2462e-02, 2.2500e-01, -1.6859e+00, ..., 6.2502e-01,
9.1731e-01, 7.0617e-01],
[-2.3105e+00, -1.7733e+00, -5.1460e+00, ..., -1.4110e+00,
-7.0875e-02, 5.2262e-01],
[ 3.5562e-01, -5.0374e-01, -1.4963e+00, ..., 5.4550e-01,
8.5623e-01, 8.1332e-01],
...,
[-2.7670e-01, -1.3557e+00, -2.8599e+00, ..., -1.0539e+00,
-7.6654e-01, -5.7326e-01],
[ 1.2895e+00, 5.6010e-01, -1.3097e+00, ..., 6.0235e-01,
8.3230e-01, 8.8904e-01],
[ 7.6112e-01, 1.0782e-01, -7.4117e-01, ..., 7.0374e-01,
8.9515e-01, 8.0668e-01]],
[[-3.4369e+00, -1.6025e+00, -6.5648e+00, ..., -5.6070e+00,
-4.8302e+00, -3.7339e+00],
[-6.9596e-01, 2.0707e+00, -4.5663e+00, ..., -4.3728e+00,
-4.2668e+00, -3.4416e+00],
[-1.6618e-01, 1.7244e+00, -7.8935e+00, ..., -6.3751e+00,
-6.1596e+00, -5.5893e+00],
...,
[-1.5031e+00, 1.3852e-01, -7.0239e+00, ..., -5.5994e+00,
-4.9797e+00, -4.0953e+00],
[ 1.3750e+00, 2.5931e+00, -1.5906e+00, ..., -1.3866e+00,
-1.6076e+00, -1.6141e+00],
[ 9.5173e-01, 1.7110e+00, -2.5016e+00, ..., -2.5538e+00,
-2.7221e+00, -2.6368e+00]],
[[-1.6463e+00, -2.5631e+00, -2.3740e+00, ..., -9.1291e-01,
-6.6365e-01, -8.7149e-01],
[-2.2445e+00, -2.3405e+00, -1.6591e+00, ..., -7.4394e-01,
-4.3506e-01, -4.0236e-01],
[-2.9001e+00, -2.2673e+00, -9.1102e-01, ..., -4.7155e-01,
-1.4177e-01, -1.4679e-01],
...,
[-1.8108e+00, -1.9330e+00, -7.3256e-01, ..., -6.4319e-01,
-5.3325e-01, -7.3710e-01],
[-1.9162e+00, -1.2449e+00, -7.1658e-01, ..., -2.3777e-01,
-4.5633e-02, -7.4822e-02],
[-1.8907e+00, -1.3091e+00, -7.5655e-01, ..., -2.7286e-01,
-2.1593e-01, -2.1669e-01]],
...,
[[-2.3943e+00, -1.3233e+00, -2.7238e+00, ..., -3.0896e+00,
-2.4934e+00, -1.7076e+00],
[-2.2323e+00, -1.4473e-01, -1.5537e+00, ..., -1.4446e+00,
-1.6261e+00, -1.4404e+00],
[-6.2906e-01, 2.3648e+00, 9.0389e-01, ..., -9.0571e-02,
1.2655e-01, 2.6688e-01],
...,
[-1.8585e+00, 1.0512e+00, 1.8024e-01, ..., -9.4451e-01,
-2.9091e-01, 1.7665e-01],
[-8.9022e-01, 1.7362e+00, 1.6929e+00, ..., 1.1552e-01,
2.3402e-01, 3.5508e-01],
[-2.1824e+00, -1.0494e+00, -6.8899e-01, ..., -2.1472e+00,
-1.2547e+00, -6.2693e-01]],
[[-2.5988e+00, -2.9747e+00, -4.2764e+00, ..., -1.0823e+00,
-6.1438e-01, -5.4388e-01],
[-1.9085e+00, -1.7041e+00, -4.9803e+00, ..., -1.7759e+00,
-1.6079e+00, -1.8375e+00],
[-3.2108e+00, -2.7529e+00, -6.9879e+00, ..., -3.8672e+00,
-3.4563e+00, -2.7952e+00],
...,
[-2.6809e+00, -1.9395e+00, -5.0247e+00, ..., -2.4015e+00,
-1.9872e+00, -1.5807e+00],
[-2.3342e+00, -1.7117e+00, -3.9371e+00, ..., -2.2376e+00,
-1.6679e+00, -1.1111e+00],
[-2.3823e+00, -1.9111e+00, -3.1392e+00, ..., -2.7750e+00,
-1.9649e+00, -1.6886e+00]],
[[-2.0586e+00, -2.8278e+00, -2.6318e+00, ..., -1.7442e+00,
-1.4478e+00, -1.0967e+00],
[-2.0895e+00, -2.8656e+00, -2.3415e+00, ..., -1.4459e+00,
-1.1753e+00, -9.4215e-01],
[-2.6419e+00, -3.2760e+00, -2.8128e+00, ..., -1.6558e+00,
-1.2728e+00, -9.3853e-01],
...,
[-3.3026e+00, -4.1506e+00, -3.9103e+00, ..., -2.8428e+00,
-2.2663e+00, -1.7299e+00],
[-2.1774e+00, -2.8378e+00, -2.7906e+00, ..., -2.0051e+00,
-1.6432e+00, -1.3175e+00],
[-1.5744e+00, -1.9298e+00, -1.9621e+00, ..., -1.2588e+00,
-1.0663e+00, -7.9286e-01]]],
[[[ 1.0482e-01, 3.4234e-01, -1.1802e+00, ..., 4.7927e-01,
1.0238e+00, 6.4923e-01],
[-1.9785e+00, -1.4041e+00, -4.1484e+00, ..., -2.5284e+00,
-7.3588e-01, -4.5432e-02],
[ 3.0140e-01, -3.2248e-01, -1.0935e+00, ..., -2.6502e-01,
4.9138e-01, 6.5734e-01],
...,
[-3.2844e-01, -1.3870e+00, -2.7874e+00, ..., -2.0044e+00,
-1.3833e+00, -9.1345e-01],
[ 1.3463e+00, 7.2643e-01, -8.8712e-01, ..., -3.4189e-01,
5.0350e-01, 7.0600e-01],
[ 9.0100e-01, 3.7736e-01, -3.8660e-01, ..., -3.6836e-01,
2.8842e-01, 2.8285e-01]],
[[-3.1452e+00, -1.6032e+00, -6.1189e+00, ..., -4.8217e+00,
-4.4176e+00, -3.3527e+00],
[-7.5418e-01, 1.6360e+00, -4.4084e+00, ..., -3.4062e+00,
-3.5753e+00, -2.9050e+00],
[-2.9120e-01, 1.3368e+00, -7.3915e+00, ..., -5.3865e+00,
-5.2670e+00, -4.9045e+00],
...,
[-1.0906e+00, 5.5841e-01, -6.2948e+00, ..., -4.5604e+00,
-4.1340e+00, -3.4648e+00],
[ 1.9290e+00, 3.0990e+00, -7.3084e-01, ..., -8.5504e-01,
-1.2023e+00, -1.2560e+00],
[ 1.2738e+00, 1.9376e+00, -2.1182e+00, ..., -2.2281e+00,
-2.4736e+00, -2.3680e+00]],
[[-1.2717e+00, -2.0936e+00, -1.9624e+00, ..., -5.7557e-01,
-1.7621e-01, -4.5922e-01],
[-1.7900e+00, -1.9381e+00, -1.4114e+00, ..., -3.2391e-01,
7.3197e-02, 2.3173e-02],
[-2.3440e+00, -1.8684e+00, -7.9772e-01, ..., -2.7879e-01,
2.3759e-01, 2.7178e-01],
...,
[-1.6943e+00, -1.9731e+00, -8.3138e-01, ..., -4.8519e-01,
-1.6565e-01, -3.5735e-01],
[-1.9118e+00, -1.2936e+00, -8.2881e-01, ..., -3.1899e-01,
1.5323e-01, 2.2458e-01],
[-1.9003e+00, -1.2085e+00, -6.1884e-01, ..., -5.0437e-01,
-2.8193e-01, -2.4849e-01]],
...,
[[-2.0518e+00, -1.1958e+00, -2.5555e+00, ..., -3.6107e+00,
-2.8079e+00, -1.8170e+00],
[-1.8190e+00, 2.2794e-02, -1.3306e+00, ..., -2.6482e+00,
-2.4321e+00, -1.8651e+00],
[-5.2688e-01, 2.1498e+00, 7.6575e-01, ..., -5.4629e-01,
-5.9665e-02, 1.5668e-01],
...,
[-1.9182e+00, 1.2830e+00, 2.2006e-01, ..., -1.5362e+00,
-6.8016e-01, -1.3222e-01],
[-9.2079e-01, 1.8177e+00, 1.6091e+00, ..., -1.4006e-01,
1.2138e-01, 3.2496e-01],
[-2.2193e+00, -1.1541e+00, -1.0228e+00, ..., -2.3599e+00,
-1.4512e+00, -9.1574e-01]],
[[-2.3866e+00, -2.6815e+00, -3.8803e+00, ..., -1.1634e+00,
-7.6871e-01, -7.9106e-01],
[-1.7849e+00, -1.5525e+00, -4.5718e+00, ..., -1.3922e+00,
-1.2264e+00, -1.5905e+00],
[-3.0145e+00, -2.5814e+00, -6.5489e+00, ..., -3.2710e+00,
-2.7832e+00, -2.2222e+00],
...,
[-2.4597e+00, -1.4975e+00, -5.0424e+00, ..., -1.4696e+00,
-1.0435e+00, -8.7810e-01],
[-2.1965e+00, -1.3949e+00, -3.9522e+00, ..., -1.7248e+00,
-1.1638e+00, -7.5655e-01],
[-2.3725e+00, -1.7312e+00, -3.1300e+00, ..., -2.8140e+00,
-2.0291e+00, -1.8332e+00]],
[[-1.8377e+00, -2.5157e+00, -2.3719e+00, ..., -1.8046e+00,
-1.4920e+00, -1.1409e+00],
[-1.8770e+00, -2.5612e+00, -2.1337e+00, ..., -1.4177e+00,
-1.1154e+00, -8.9709e-01],
[-2.4014e+00, -2.9551e+00, -2.5641e+00, ..., -1.6232e+00,
-1.2160e+00, -9.0426e-01],
...,
[-3.1957e+00, -3.9904e+00, -3.7104e+00, ..., -2.8786e+00,
-2.2965e+00, -1.7651e+00],
[-2.1110e+00, -2.7320e+00, -2.6482e+00, ..., -1.9841e+00,
-1.6275e+00, -1.3144e+00],
[-1.5075e+00, -1.8275e+00, -1.8464e+00, ..., -1.2263e+00,
-1.0342e+00, -7.9393e-01]]],
[[[ 3.5741e-01, 8.1848e-01, 7.3857e-01, ..., 6.7118e-01,
9.4839e-01, 7.1573e-01],
[-1.2110e+00, -2.9302e-01, -1.3403e+00, ..., -1.2964e+00,
-5.7837e-02, 4.9224e-01],
[ 6.0318e-01, 4.2065e-01, 5.4438e-01, ..., 5.7164e-01,
9.2749e-01, 8.4049e-01],
...,
[-1.8314e-01, -1.1393e+00, -2.6914e+00, ..., -1.4988e+00,
-1.4049e+00, -8.4465e-01],
[ 1.4371e+00, 8.8425e-01, -9.4550e-01, ..., -2.9778e-01,
1.0418e-01, 3.8187e-01],
[ 8.4708e-01, 3.2593e-01, -6.9007e-01, ..., -2.0235e-01,
2.1001e-01, 3.8588e-01]],
[[-2.3176e+00, -1.2996e+00, -4.2766e+00, ..., -5.4394e+00,
-4.6613e+00, -3.6133e+00],
[-1.9568e-01, 1.5274e+00, -2.6684e+00, ..., -4.0752e+00,
-3.9847e+00, -3.2567e+00],
[ 6.1385e-01, 1.3182e+00, -5.1811e+00, ..., -5.9873e+00,
-5.8012e+00, -5.3815e+00],
...,
[-1.3008e+00, 1.1652e-01, -6.7001e+00, ..., -7.6193e+00,
-6.9278e+00, -5.4567e+00],
[ 1.7792e+00, 2.7988e+00, -1.0470e+00, ..., -2.6016e+00,
-2.9649e+00, -2.3915e+00],
[ 1.1970e+00, 1.8097e+00, -2.2837e+00, ..., -3.2724e+00,
-3.6400e+00, -3.1681e+00]],
[[-6.5450e-01, -1.7588e+00, -1.6423e+00, ..., -1.0141e+00,
-7.0152e-01, -8.2587e-01],
[-1.3133e+00, -1.7523e+00, -1.2024e+00, ..., -8.1239e-01,
-4.2791e-01, -2.8998e-01],
[-9.9446e-01, -9.1504e-01, -4.1686e-01, ..., -3.4503e-01,
5.5169e-02, 7.2951e-02],
...,
[-1.3814e+00, -1.7412e+00, -7.7184e-01, ..., -8.9222e-01,
-7.7915e-01, -1.0379e+00],
[-1.7184e+00, -1.1401e+00, -8.2617e-01, ..., -5.6225e-01,
-2.8961e-01, -3.6637e-01],
[-1.8683e+00, -1.2673e+00, -8.9363e-01, ..., -9.6353e-01,
-8.3024e-01, -8.0294e-01]],
...,
[[-2.1673e+00, -1.2737e+00, -2.2935e+00, ..., -3.1458e+00,
-2.4750e+00, -1.6248e+00],
[-5.9449e-01, 1.0255e+00, -4.3434e-01, ..., -1.4230e+00,
-1.4941e+00, -1.3100e+00],
[-3.6800e-01, 1.9564e+00, 7.3748e-01, ..., -1.6037e-01,
2.7493e-01, 4.1133e-01],
...,
[-1.8304e+00, 1.2423e+00, 5.8071e-02, ..., -1.0815e+00,
-5.5068e-01, 1.6880e-02],
[-8.1892e-01, 1.8675e+00, 1.6449e+00, ..., 5.0099e-01,
4.1915e-01, 6.8496e-01],
[-2.2382e+00, -1.2099e+00, -1.0131e+00, ..., -1.8548e+00,
-9.6306e-01, -2.9070e-01]],
[[-1.3545e+00, -1.2605e+00, -2.6366e+00, ..., -9.9218e-01,
-4.5321e-01, -3.5484e-01],
[-6.7877e-01, -1.8424e-01, -3.4597e+00, ..., -1.6654e+00,
-1.4092e+00, -1.6573e+00],
[-2.0783e+00, -1.3770e+00, -5.7564e+00, ..., -3.7653e+00,
-3.2286e+00, -2.5684e+00],
...,
[-2.4801e+00, -1.6033e+00, -5.4731e+00, ..., -3.0685e+00,
-2.7294e+00, -2.1986e+00],
[-2.2195e+00, -1.4607e+00, -4.2483e+00, ..., -2.6683e+00,
-2.3014e+00, -1.6539e+00],
[-2.3867e+00, -1.7633e+00, -3.3020e+00, ..., -2.9420e+00,
-2.4340e+00, -2.1306e+00]],
[[-1.1400e+00, -1.6889e+00, -1.4911e+00, ..., -1.7026e+00,
-1.4146e+00, -1.0597e+00],
[-1.1535e+00, -1.6172e+00, -1.1861e+00, ..., -1.3714e+00,
-1.1146e+00, -8.8070e-01],
[-1.4989e+00, -1.7852e+00, -1.3668e+00, ..., -1.5346e+00,
-1.1645e+00, -8.4359e-01],
...,
[-3.1756e+00, -3.9758e+00, -3.7675e+00, ..., -3.9769e+00,
-3.1825e+00, -2.4509e+00],
[-2.1167e+00, -2.7379e+00, -2.7125e+00, ..., -2.7951e+00,
-2.2995e+00, -1.8684e+00],
[-1.5417e+00, -1.8743e+00, -1.9427e+00, ..., -1.8897e+00,
-1.5297e+00, -1.1691e+00]]],
[[[ 2.4927e-01, 5.2592e-01, 6.4947e-01, ..., 5.9134e-01,
9.8013e-01, 7.8821e-01],
[-2.1202e+00, -1.4259e+00, -3.0217e+00, ..., -1.3067e+00,
9.0646e-02, 6.2625e-01],
[-8.5328e-02, -8.9845e-01, -1.6797e-02, ..., 4.3664e-01,
9.5477e-01, 9.5825e-01],
...,
[-9.6986e-01, -2.1727e+00, -2.0104e+00, ..., -1.3742e+00,
-9.2484e-01, -6.7065e-01],
[ 7.5270e-01, -1.8491e-01, -1.4466e-01, ..., 4.4363e-01,
8.0030e-01, 8.9712e-01],
[ 1.6057e-01, -6.4156e-01, -4.4867e-01, ..., 6.2945e-01,
8.9158e-01, 8.0311e-01]],
[[-1.7586e+00, -1.3289e+00, -4.2971e+00, ..., -5.7626e+00,
-4.8983e+00, -3.8143e+00],
[ 6.0067e-01, 1.8205e+00, -2.3208e+00, ..., -4.6972e+00,
-4.4541e+00, -3.6414e+00],
[ 1.4455e+00, 1.6245e+00, -4.9530e+00, ..., -6.8251e+00,
-6.3835e+00, -5.8001e+00],
...,
[ 3.6829e-01, 2.8610e-01, -4.2068e+00, ..., -5.9950e+00,
-5.1718e+00, -4.2612e+00],
[ 2.4023e+00, 2.2943e+00, -1.7602e-01, ..., -1.2773e+00,
-1.4118e+00, -1.4894e+00],
[ 1.4162e+00, 1.1911e+00, -1.7696e+00, ..., -2.6819e+00,
-2.8040e+00, -2.7754e+00]],
[[-1.9729e-01, -1.3168e+00, -1.5478e+00, ..., -8.0004e-01,
-5.4412e-01, -6.6554e-01],
[-6.5202e-01, -1.0064e+00, -1.0002e+00, ..., -6.8061e-01,
-3.6287e-01, -2.2467e-01],
[-4.6449e-01, -3.2916e-01, -4.7382e-01, ..., -3.7962e-01,
-3.4229e-03, 9.3210e-02],
...,
[-2.9013e-02, -5.4096e-01, -5.6470e-01, ..., -6.6461e-01,
-4.8843e-01, -5.8637e-01],
[-6.6306e-01, -1.9851e-01, -5.5515e-01, ..., -2.8979e-01,
-2.8851e-03, 8.6950e-02],
[-8.1581e-01, -2.4439e-01, -4.8348e-01, ..., -2.8997e-01,
-2.0736e-01, -1.3981e-01]],
...,
[[-3.0644e+00, -2.5607e+00, -2.6459e+00, ..., -3.1695e+00,
-2.6034e+00, -1.7424e+00],
[-2.1747e+00, -7.1272e-01, -1.1145e+00, ..., -1.6694e+00,
-1.7580e+00, -1.4547e+00],
[-1.1588e+00, 1.0244e+00, 8.8402e-01, ..., -2.5392e-01,
1.4440e-03, 1.6087e-01],
...,
[-2.1995e+00, -1.9056e-01, -1.1339e-01, ..., -1.1082e+00,
-3.8237e-01, 1.8475e-01],
[-1.2243e+00, 8.2783e-01, 1.0443e+00, ..., 2.2504e-01,
3.4453e-01, 4.5503e-01],
[-2.1007e+00, -1.4085e+00, -1.5861e+00, ..., -2.2718e+00,
-1.2725e+00, -6.3541e-01]],
[[-1.6367e+00, -1.5878e+00, -2.8936e+00, ..., -1.3073e+00,
-7.0794e-01, -5.0687e-01],
[-6.0071e-01, 1.0662e-01, -3.2264e+00, ..., -1.9961e+00,
-1.6209e+00, -1.7019e+00],
[-2.0724e+00, -8.4485e-01, -5.3509e+00, ..., -4.0050e+00,
-3.3548e+00, -2.5716e+00],
...,
[-1.3673e+00, 1.1403e-01, -3.3438e+00, ..., -2.3614e+00,
-1.7688e+00, -1.3024e+00],
[-1.6767e+00, -5.0393e-01, -2.9701e+00, ..., -2.2698e+00,
-1.4875e+00, -8.4633e-01],
[-2.2959e+00, -1.3507e+00, -2.9776e+00, ..., -2.9099e+00,
-1.9339e+00, -1.6219e+00]],
[[-1.3535e+00, -1.9089e+00, -1.6927e+00, ..., -1.8534e+00,
-1.4989e+00, -1.0983e+00],
[-1.2775e+00, -1.7590e+00, -1.2009e+00, ..., -1.6164e+00,
-1.2755e+00, -9.7481e-01],
[-1.5984e+00, -1.9404e+00, -1.4349e+00, ..., -1.8933e+00,
-1.4104e+00, -1.0028e+00],
...,
[-2.4218e+00, -3.0240e+00, -2.7687e+00, ..., -3.1674e+00,
-2.4885e+00, -1.8516e+00],
[-1.5770e+00, -2.0787e+00, -1.9956e+00, ..., -2.2249e+00,
-1.8039e+00, -1.4198e+00],
[-1.0365e+00, -1.3550e+00, -1.3876e+00, ..., -1.3736e+00,
-1.1528e+00, -8.3543e-01]]]], device='cuda:0'),
tensor([[[[ 2.6105e-01, 5.1544e-01, 6.5580e-01, ..., 6.2193e-01,
9.7723e-01, 7.8301e-01],
[-2.1114e+00, -1.4051e+00, -3.0509e+00, ..., -1.2182e+00,
1.2050e-01, 6.6497e-01],
[-8.3693e-02, -9.4185e-01, -2.4004e-02, ..., 4.8654e-01,
9.5586e-01, 9.7704e-01],
...,
[-9.7901e-01, -2.1226e+00, -1.9653e+00, ..., -1.3742e+00,
-9.0281e-01, -5.9831e-01],
[ 7.2878e-01, -1.8803e-01, -1.1664e-01, ..., 4.0525e-01,
7.4581e-01, 9.0029e-01],
[ 1.2662e-01, -6.7977e-01, -4.3415e-01, ..., 6.1691e-01,
8.4091e-01, 7.8736e-01]],
[[-1.7174e+00, -1.2350e+00, -4.2460e+00, ..., -5.7047e+00,
-4.8500e+00, -3.7687e+00],
[ 7.0233e-01, 1.9519e+00, -2.2774e+00, ..., -4.6951e+00,
-4.4554e+00, -3.5902e+00],
[ 1.5726e+00, 1.7083e+00, -4.9808e+00, ..., -6.8134e+00,
-6.3641e+00, -5.6917e+00],
...,
[ 3.6364e-01, 2.8012e-01, -4.2232e+00, ..., -5.9865e+00,
-5.1040e+00, -4.3215e+00],
[ 2.3792e+00, 2.2797e+00, -2.2831e-01, ..., -1.2812e+00,
-1.3749e+00, -1.5728e+00],
[ 1.4034e+00, 1.1820e+00, -1.8060e+00, ..., -2.6905e+00,
-2.7917e+00, -2.8239e+00]],
[[-2.1752e-01, -1.3664e+00, -1.5667e+00, ..., -7.6513e-01,
-5.0825e-01, -6.3707e-01],
[-6.9253e-01, -1.0231e+00, -9.7553e-01, ..., -6.1888e-01,
-3.0711e-01, -2.0500e-01],
[-5.1010e-01, -3.4092e-01, -4.4738e-01, ..., -3.3527e-01,
3.3031e-02, 9.0490e-02],
...,
[-8.6197e-03, -5.1275e-01, -5.5507e-01, ..., -6.3670e-01,
-4.6631e-01, -5.5823e-01],
[-6.5250e-01, -1.7657e-01, -5.3411e-01, ..., -2.7330e-01,
-4.7406e-03, 8.9680e-02],
[-8.2335e-01, -2.3955e-01, -4.6266e-01, ..., -2.8983e-01,
-2.0732e-01, -1.4248e-01]],
...,
[[-3.0918e+00, -2.5099e+00, -2.6062e+00, ..., -3.1207e+00,
-2.5799e+00, -1.7183e+00],
[-2.1815e+00, -6.7369e-01, -1.1234e+00, ..., -1.6397e+00,
-1.7695e+00, -1.4601e+00],
[-1.0953e+00, 1.1185e+00, 9.1787e-01, ..., -2.5665e-01,
-1.9956e-02, 1.5869e-01],
...,
[-2.1850e+00, -1.8432e-01, -1.0408e-01, ..., -1.1703e+00,
-3.9386e-01, 1.7690e-01],
[-1.2488e+00, 7.7531e-01, 1.0189e+00, ..., 1.2442e-01,
3.1630e-01, 4.4470e-01],
[-2.0950e+00, -1.4086e+00, -1.5712e+00, ..., -2.3056e+00,
-1.2824e+00, -6.2057e-01]],
[[-1.6303e+00, -1.5966e+00, -2.9584e+00, ..., -1.3314e+00,
-7.4230e-01, -5.4574e-01],
[-5.6523e-01, 1.3882e-01, -3.3080e+00, ..., -2.0289e+00,
-1.6562e+00, -1.7300e+00],
[-2.0331e+00, -7.8738e-01, -5.4114e+00, ..., -3.9814e+00,
-3.3488e+00, -2.5884e+00],
...,
[-1.3559e+00, 8.5928e-02, -3.3943e+00, ..., -2.3100e+00,
-1.7754e+00, -1.3488e+00],
[-1.6470e+00, -5.1003e-01, -2.9835e+00, ..., -2.1967e+00,
-1.4917e+00, -8.9856e-01],
[-2.2513e+00, -1.3397e+00, -2.9830e+00, ..., -2.8496e+00,
-1.9384e+00, -1.6540e+00]],
[[-1.3666e+00, -1.9249e+00, -1.6894e+00, ..., -1.8360e+00,
-1.4978e+00, -1.0962e+00],
[-1.2868e+00, -1.7827e+00, -1.2013e+00, ..., -1.6247e+00,
-1.2948e+00, -9.9034e-01],
[-1.6180e+00, -1.9592e+00, -1.4431e+00, ..., -1.9179e+00,
-1.4468e+00, -1.0332e+00],
...,
[-2.4330e+00, -3.0268e+00, -2.7793e+00, ..., -3.2019e+00,
-2.5209e+00, -1.8797e+00],
[-1.5958e+00, -2.0968e+00, -2.0130e+00, ..., -2.2534e+00,
-1.8298e+00, -1.4413e+00],
[-1.0543e+00, -1.3742e+00, -1.4017e+00, ..., -1.3838e+00,
-1.1647e+00, -8.4446e-01]]],
[[[ 7.5980e-02, 2.9204e-01, -1.4376e+00, ..., 4.9592e-01,
1.0517e+00, 6.9657e-01],
[-2.2432e+00, -1.5935e+00, -4.7390e+00, ..., -2.2983e+00,
-5.1925e-01, 8.6637e-02],
[ 3.5830e-01, -3.9313e-01, -1.3343e+00, ..., -5.6479e-02,
6.7693e-01, 7.4474e-01],
...,
[-3.0141e-01, -1.3194e+00, -2.7100e+00, ..., -1.9059e+00,
-1.2780e+00, -8.3918e-01],
[ 1.2508e+00, 6.0917e-01, -1.0943e+00, ..., -2.4202e-01,
5.7699e-01, 7.4361e-01],
[ 7.8547e-01, 2.0869e-01, -5.7132e-01, ..., -2.5480e-01,
3.7205e-01, 3.2977e-01]],
[[-3.3272e+00, -1.5265e+00, -6.3047e+00, ..., -4.7978e+00,
-4.4504e+00, -3.4317e+00],
[-6.1830e-01, 2.1227e+00, -4.3403e+00, ..., -3.4263e+00,
-3.6708e+00, -3.0339e+00],
[-4.8229e-02, 1.8181e+00, -7.6204e+00, ..., -5.4583e+00,
-5.4004e+00, -5.0326e+00],
...,
[-1.3947e+00, 1.9973e-01, -6.7772e+00, ..., -4.6299e+00,
-4.1880e+00, -3.4785e+00],
[ 1.4584e+00, 2.6154e+00, -1.4567e+00, ..., -9.3595e-01,
-1.2458e+00, -1.2755e+00],
[ 9.8141e-01, 1.6766e+00, -2.4576e+00, ..., -2.2714e+00,
-2.4902e+00, -2.3728e+00]],
[[-1.5154e+00, -2.4417e+00, -2.2513e+00, ..., -6.1827e-01,
-2.6617e-01, -4.9419e-01],
[-2.1236e+00, -2.2421e+00, -1.5685e+00, ..., -3.7420e-01,
6.2759e-03, 2.1762e-02],
[-2.7064e+00, -2.1184e+00, -8.6381e-01, ..., -2.6820e-01,
2.2613e-01, 2.7745e-01],
...,
[-1.6566e+00, -1.8745e+00, -7.7260e-01, ..., -4.8037e-01,
-1.7795e-01, -3.5719e-01],
[-1.8187e+00, -1.2263e+00, -7.6668e-01, ..., -3.0002e-01,
1.4948e-01, 2.1551e-01],
[-1.8015e+00, -1.2211e+00, -6.9640e-01, ..., -4.7669e-01,
-2.6163e-01, -2.2477e-01]],
...,
[[-2.3637e+00, -1.3166e+00, -2.7249e+00, ..., -3.6127e+00,
-2.7851e+00, -1.7811e+00],
[-2.0173e+00, 3.0913e-02, -1.4570e+00, ..., -2.5253e+00,
-2.3095e+00, -1.7834e+00],
[-5.1445e-01, 2.4819e+00, 9.1189e-01, ..., -5.6646e-01,
-5.4013e-02, 1.3732e-01],
...,
[-1.8187e+00, 1.0901e+00, 1.7233e-01, ..., -1.4639e+00,
-6.7263e-01, -1.0293e-01],
[-8.8442e-01, 1.6990e+00, 1.5750e+00, ..., -1.0091e-01,
1.1421e-01, 3.2849e-01],
[-2.1414e+00, -1.0419e+00, -7.7099e-01, ..., -2.2876e+00,
-1.4152e+00, -8.7892e-01]],
[[-2.4703e+00, -2.7702e+00, -4.1111e+00, ..., -1.1185e+00,
-7.4844e-01, -7.4774e-01],
[-1.7636e+00, -1.4584e+00, -4.8317e+00, ..., -1.4689e+00,
-1.3271e+00, -1.6364e+00],
[-3.0854e+00, -2.5229e+00, -6.9004e+00, ..., -3.4090e+00,
-2.9171e+00, -2.2853e+00],
...,
[-2.5910e+00, -1.8431e+00, -4.9992e+00, ..., -1.5819e+00,
-1.1138e+00, -9.2941e-01],
[-2.2612e+00, -1.6428e+00, -3.9217e+00, ..., -1.7860e+00,
-1.1784e+00, -7.6647e-01],
[-2.3372e+00, -1.8518e+00, -3.1181e+00, ..., -2.7762e+00,
-1.9753e+00, -1.7751e+00]],
[[-1.9767e+00, -2.7257e+00, -2.5206e+00, ..., -1.7836e+00,
-1.4903e+00, -1.1383e+00],
[-2.0034e+00, -2.7480e+00, -2.2171e+00, ..., -1.4148e+00,
-1.1319e+00, -9.0341e-01],
[-2.5450e+00, -3.1432e+00, -2.6635e+00, ..., -1.6228e+00,
-1.2325e+00, -9.1627e-01],
...,
[-3.1948e+00, -4.0017e+00, -3.7727e+00, ..., -2.8508e+00,
-2.2750e+00, -1.7529e+00],
[-2.1165e+00, -2.7451e+00, -2.7025e+00, ..., -1.9772e+00,
-1.6231e+00, -1.3098e+00],
[-1.5250e+00, -1.8635e+00, -1.9013e+00, ..., -1.2320e+00,
-1.0408e+00, -7.9704e-01]]],
[[[ 4.0244e-01, 7.3982e-01, 7.8731e-01, ..., 6.3737e-01,
1.0502e+00, 7.0024e-01],
[-1.6857e+00, -8.9933e-01, -2.3825e+00, ..., -1.9574e+00,
-4.2330e-01, 1.5836e-01],
[ 2.5503e-01, -3.5555e-01, 1.0140e-01, ..., 2.2879e-01,
7.3516e-01, 7.7965e-01],
...,
[ 1.7080e-01, -9.1766e-01, -1.9391e+00, ..., -1.6933e+01,
-1.9747e+01, -1.9096e+01],
[ 1.5925e+00, 1.0692e+00, 1.4492e-01, ..., -4.2875e+00,
-7.7418e+00, -1.0240e+01],
[ 1.2215e+00, 9.1323e-01, 4.4456e-01, ..., -4.1722e+00,
-6.7678e+00, -6.6261e+00]],
[[-1.9425e+00, -1.0999e+00, -4.1436e+00, ..., -5.0512e+00,
-4.4760e+00, -3.4739e+00],
[ 2.3873e-01, 1.9253e+00, -2.3738e+00, ..., -3.8226e+00,
-3.8149e+00, -3.1667e+00],
[ 9.8936e-01, 1.7396e+00, -5.0367e+00, ..., -5.7957e+00,
-5.5414e+00, -5.1895e+00],
...,
[-7.5109e-01, 5.0667e-01, -5.5407e+00, ..., -6.4666e+00,
-7.1545e+00, -7.1791e+00],
[ 2.0048e+00, 2.8007e+00, -4.6524e-01, ..., 2.0255e-01,
-1.1157e+00, -3.5444e+00],
[ 1.2265e+00, 1.5775e+00, -2.0230e+00, ..., -3.5309e-01,
-7.8389e-01, -2.8794e+00]],
[[-2.7809e-01, -1.3286e+00, -1.3534e+00, ..., -5.8158e-01,
-2.9171e-01, -5.5183e-01],
[-8.9821e-01, -1.2141e+00, -9.0426e-01, ..., -4.2378e-01,
-1.2782e-01, -1.5218e-01],
[-8.2157e-01, -6.7795e-01, -4.4661e-01, ..., -3.5543e-01,
-2.3531e-03, 2.3005e-02],
...,
[-1.2394e+00, -1.7474e+00, -7.9041e-01, ..., 2.3632e+00,
-6.7278e-01, -7.0872e-01],
[-1.4648e+00, -1.0215e+00, -6.5373e-01, ..., 3.8366e+00,
1.9988e+00, 4.8818e-01],
[-1.4115e+00, -6.9744e-01, -1.7727e-01, ..., 3.4331e+00,
1.9523e+00, 1.0114e+00]],
...,
[[-2.7327e+00, -1.9320e+00, -2.6224e+00, ..., -3.2863e+00,
-2.5627e+00, -1.7155e+00],
[-1.6397e+00, 3.3807e-02, -1.2121e+00, ..., -2.1928e+00,
-2.0507e+00, -1.6446e+00],
[-6.4378e-01, 1.8153e+00, 6.8716e-01, ..., -3.6341e-01,
5.6808e-03, 1.4418e-01],
...,
[-1.7555e+00, 1.4591e+00, 3.5476e-01, ..., -7.7550e+00,
-1.2172e+01, -2.0162e+00],
[-7.8817e-01, 1.6962e+00, 1.2792e+00, ..., -6.4482e+00,
-6.9408e+00, 9.0084e-01],
[-1.8757e+00, -9.6566e-01, -1.1823e+00, ..., -7.9535e+00,
-5.0289e+00, 1.5360e+00]],
[[-1.4491e+00, -1.3817e+00, -2.8048e+00, ..., -1.1225e+00,
-6.8418e-01, -6.5441e-01],
[-5.4480e-01, 8.4353e-02, -3.3554e+00, ..., -1.4680e+00,
-1.2721e+00, -1.5773e+00],
[-2.0388e+00, -1.0116e+00, -5.6066e+00, ..., -3.4058e+00,
-2.8868e+00, -2.3244e+00],
...,
[-2.1441e+00, -1.0427e+00, -4.8354e+00, ..., -8.8080e+00,
-5.7115e+00, -2.2548e+00],
[-1.8837e+00, -9.5718e-01, -3.7996e+00, ..., -9.8628e+00,
-5.2965e+00, -1.4334e+00],
[-2.0841e+00, -1.3547e+00, -3.0419e+00, ..., -9.5035e+00,
-4.4218e+00, -1.8739e+00]],
[[-1.2781e+00, -1.8443e+00, -1.6263e+00, ..., -1.7276e+00,
-1.4298e+00, -1.0709e+00],
[-1.2296e+00, -1.7239e+00, -1.2103e+00, ..., -1.3910e+00,
-1.1079e+00, -8.6875e-01],
[-1.5857e+00, -1.9290e+00, -1.4742e+00, ..., -1.6408e+00,
-1.2358e+00, -9.0053e-01],
...,
[-2.7866e+00, -3.5271e+00, -3.3290e+00, ..., -8.5280e+00,
-7.9519e+00, -6.2635e+00],
[-1.8728e+00, -2.4036e+00, -2.3432e+00, ..., -5.7534e+00,
-6.0549e+00, -4.7425e+00],
[-1.2832e+00, -1.5396e+00, -1.5838e+00, ..., -3.3245e+00,
-4.1835e+00, -3.0853e+00]]],
[[[ 7.4810e-02, 2.3630e-01, -1.3748e+00, ..., -5.8332e-01,
-1.9835e-01, -4.0804e-02],
[-2.0877e+00, -1.5911e+00, -4.4652e+00, ..., -3.6558e+00,
-1.5104e+00, -5.4315e-01],
[ 5.6069e-01, -1.0772e-01, -8.1447e-01, ..., -1.0298e-01,
-2.3676e-01, 2.5793e-01],
...,
[-3.5853e-01, -1.2906e+00, -1.7158e+00, ..., -1.2797e+00,
-8.1666e-01, -5.9023e-01],
[ 1.0679e+00, 5.8186e-01, 2.5482e-01, ..., 4.0540e-01,
7.1349e-01, 8.0409e-01],
[ 7.4724e-01, 4.5908e-01, 2.9236e-01, ..., 5.0426e-01,
7.6081e-01, 7.0241e-01]],
[[-3.2704e+00, -1.7556e+00, -6.2200e+00, ..., -7.1563e+00,
-6.2522e+00, -4.6378e+00],
[-7.7993e-01, 1.5618e+00, -4.3369e+00, ..., -5.5157e+00,
-5.5586e+00, -4.3060e+00],
[-3.3860e-01, 1.1029e+00, -7.4137e+00, ..., -7.9876e+00,
-7.8517e+00, -6.6833e+00],
...,
[-2.0449e-01, 7.0387e-01, -4.1609e+00, ..., -5.4103e+00,
-4.7023e+00, -3.9277e+00],
[ 2.4377e+00, 3.0062e+00, 5.2792e-01, ..., -1.2860e+00,
-1.3764e+00, -1.4815e+00],
[ 1.4459e+00, 1.6635e+00, -1.4586e+00, ..., -2.5151e+00,
-2.6143e+00, -2.5764e+00]],
[[-1.3324e+00, -2.3609e+00, -2.3228e+00, ..., -1.7771e+00,
-1.2548e+00, -1.3686e+00],
[-1.7008e+00, -1.9947e+00, -1.5301e+00, ..., -1.2419e+00,
-7.3464e-01, -7.9546e-01],
[-2.1794e+00, -1.9318e+00, -1.0024e+00, ..., -1.0869e+00,
-4.6607e-01, -1.7863e-01],
...,
[-7.9479e-01, -1.2963e+00, -6.1912e-01, ..., -5.8219e-01,
-4.6790e-01, -6.3076e-01],
[-1.2498e+00, -8.3367e-01, -6.4001e-01, ..., -2.9387e-01,
-8.5657e-02, -5.6889e-02],
[-1.4326e+00, -7.9762e-01, -4.4211e-01, ..., -3.5033e-01,
-2.8822e-01, -2.4965e-01]],
...,
[[-2.0250e+00, -1.0689e+00, -2.3374e+00, ..., -3.1980e+00,
-2.5530e+00, -1.7785e+00],
[-1.7330e+00, 7.5699e-02, -1.3384e+00, ..., -2.5146e+00,
-2.4496e+00, -1.9261e+00],
[-3.3624e-01, 2.4229e+00, 1.0873e+00, ..., 5.8351e-02,
2.9475e-01, 5.4658e-01],
...,
[-1.7335e+00, 1.1772e+00, -2.0821e-02, ..., -1.1555e+00,
-4.6196e-01, 2.8525e-02],
[-6.0755e-01, 1.6217e+00, 9.6278e-01, ..., 3.9907e-02,
1.8522e-01, 2.7793e-01],
[-1.6709e+00, -1.0852e+00, -1.5341e+00, ..., -2.1660e+00,
-1.2525e+00, -6.7624e-01]],
[[-2.4055e+00, -2.8176e+00, -4.2351e+00, ..., -2.4330e+00,
-1.9309e+00, -1.7043e+00],
[-1.7581e+00, -1.6333e+00, -4.9132e+00, ..., -2.9633e+00,
-2.8154e+00, -2.7564e+00],
[-2.8221e+00, -2.4465e+00, -6.6885e+00, ..., -4.5199e+00,
-4.3937e+00, -3.4737e+00],
...,
[-1.5770e+00, -2.2856e-01, -4.2195e+00, ..., -2.1500e+00,
-1.6217e+00, -1.2631e+00],
[-1.5973e+00, -4.8228e-01, -3.4682e+00, ..., -2.0838e+00,
-1.3908e+00, -8.5106e-01],
[-2.0621e+00, -1.1558e+00, -2.8555e+00, ..., -2.6587e+00,
-1.7955e+00, -1.5206e+00]],
[[-1.8017e+00, -2.5278e+00, -2.4382e+00, ..., -2.6907e+00,
-2.2122e+00, -1.6869e+00],
[-1.7991e+00, -2.5081e+00, -2.1338e+00, ..., -2.3986e+00,
-1.9142e+00, -1.5403e+00],
[-2.2551e+00, -2.8179e+00, -2.5025e+00, ..., -2.7450e+00,
-2.1525e+00, -1.6368e+00],
...,
[-2.4188e+00, -3.0247e+00, -2.7447e+00, ..., -2.8279e+00,
-2.2523e+00, -1.6990e+00],
[-1.6295e+00, -2.0908e+00, -1.9899e+00, ..., -2.0234e+00,
-1.6561e+00, -1.3109e+00],
[-1.0961e+00, -1.3316e+00, -1.3757e+00, ..., -1.2719e+00,
-1.0757e+00, -7.8991e-01]]],
[[[-1.0950e+01, -1.3795e+01, -1.3039e+01, ..., -3.9902e-01,
-1.3078e-01, -1.2506e-02],
[-1.9072e+01, -1.9454e+01, -2.1671e+01, ..., -3.3083e+00,
-1.3513e+00, -4.7220e-01],
[-2.3712e+01, -2.2233e+01, -2.2272e+01, ..., -2.2144e-01,
-4.1941e-01, 1.2851e-01],
...,
[-9.4789e-01, -2.0930e+00, -1.9713e+00, ..., -1.9678e+00,
-1.9188e+00, -1.2869e+00],
[ 7.5206e-01, -1.4390e-01, -1.2289e-01, ..., -1.4408e-02,
2.3903e-01, 4.3544e-01],
[ 1.7512e-01, -6.0516e-01, -4.1243e-01, ..., 1.9623e-01,
4.5221e-01, 4.9781e-01]],
[[-1.7832e+00, -2.7031e+00, -4.4853e+00, ..., -6.8940e+00,
-6.1053e+00, -4.5629e+00],
[-7.2510e-01, -1.1350e-01, -2.9681e+00, ..., -5.4688e+00,
-5.5867e+00, -4.3310e+00],
[-8.0118e-01, 9.4281e-01, -7.2549e+00, ..., -7.8780e+00,
-7.8545e+00, -6.6486e+00],
...,
[ 3.1060e-01, 2.3752e-01, -4.2462e+00, ..., -7.1578e+00,
-6.7462e+00, -5.2330e+00],
[ 2.3571e+00, 2.2699e+00, -1.9274e-01, ..., -1.9341e+00,
-2.5860e+00, -2.0622e+00],
[ 1.3844e+00, 1.1791e+00, -1.7803e+00, ..., -2.9345e+00,
-3.5059e+00, -3.0631e+00]],
[[-4.9088e-01, -1.0545e+00, -2.0677e+00, ..., -1.4282e+00,
-1.0726e+00, -1.3610e+00],
[-4.9223e-01, -9.5768e-01, -1.5238e+00, ..., -1.0074e+00,
-6.3009e-01, -9.0759e-01],
[ 2.3324e-01, -3.2862e-01, 1.2293e-01, ..., -1.1223e+00,
-6.1658e-01, -4.7125e-01],
...,
[-3.1409e-02, -5.6813e-01, -6.0569e-01, ..., -1.0543e+00,
-8.8675e-01, -1.2486e+00],
[-6.5112e-01, -1.9669e-01, -5.5479e-01, ..., -5.6607e-01,
-2.5164e-01, -4.1369e-01],
[-8.0031e-01, -2.3831e-01, -4.6703e-01, ..., -5.1983e-01,
-4.7627e-01, -6.2889e-01]],
...,
[[-6.0671e+00, -8.1155e+00, -4.7702e+00, ..., -3.0760e+00,
-2.5051e+00, -1.8102e+00],
[-8.2431e+00, -7.8979e+00, -6.9890e+00, ..., -2.4630e+00,
-2.5418e+00, -2.0513e+00],
[-1.1524e+01, -9.4450e+00, -1.0389e+01, ..., -8.0378e-02,
1.6518e-02, 2.5989e-01],
...,
[-2.1597e+00, -1.5898e-01, -9.9522e-02, ..., -9.4955e-01,
-2.7149e-01, 2.8626e-01],
[-1.1802e+00, 8.5565e-01, 1.0535e+00, ..., 4.4325e-01,
4.8410e-01, 8.1542e-01],
[-2.0739e+00, -1.3779e+00, -1.5483e+00, ..., -2.2197e+00,
-1.1549e+00, -3.4976e-01]],
[[-2.6608e-01, 1.0204e+00, -2.8965e-01, ..., -2.2590e+00,
-1.9357e+00, -1.8441e+00],
[ 2.3859e+00, 4.5230e+00, 5.3259e-01, ..., -2.8643e+00,
-2.8878e+00, -2.9131e+00],
[ 2.8650e+00, 4.6730e+00, -4.9050e-01, ..., -4.4455e+00,
-4.5280e+00, -3.7274e+00],
...,
[-1.3743e+00, 7.1618e-02, -3.3505e+00, ..., -2.8406e+00,
-2.7516e+00, -2.3208e+00],
[-1.6734e+00, -5.3419e-01, -2.9643e+00, ..., -2.4667e+00,
-2.2665e+00, -1.6734e+00],
[-2.2775e+00, -1.3600e+00, -2.9860e+00, ..., -2.8918e+00,
-2.4438e+00, -2.2151e+00]],
[[-1.7786e+00, -2.8362e+00, -2.7264e+00, ..., -2.6049e+00,
-2.1731e+00, -1.6873e+00],
[-2.1572e+00, -3.1857e+00, -2.7292e+00, ..., -2.4309e+00,
-1.9559e+00, -1.6056e+00],
[-3.1308e+00, -3.9625e+00, -3.8681e+00, ..., -2.8499e+00,
-2.2748e+00, -1.7662e+00],
...,
[-2.4037e+00, -2.9964e+00, -2.7469e+00, ..., -4.0318e+00,
-3.2718e+00, -2.5652e+00],
[-1.5662e+00, -2.0552e+00, -1.9752e+00, ..., -2.7794e+00,
-2.3303e+00, -1.9292e+00],
[-1.0300e+00, -1.3400e+00, -1.3675e+00, ..., -1.8375e+00,
-1.5306e+00, -1.1842e+00]]]], device='cuda:0'),
tensor([[[[ 2.6613e-01, 5.4671e-01, 6.9785e-01, ..., 4.3621e-01,
1.0155e+00, 5.9156e-01],
[-2.1295e+00, -1.4544e+00, -3.0468e+00, ..., -2.9238e+00,
-9.9182e-01, -2.5041e-01],
[-2.7185e-01, -1.0276e+00, -1.6475e-01, ..., -2.1564e-01,
5.1926e-01, 6.1680e-01],
...,
[-2.1409e-01, -1.1982e+00, -2.6817e+00, ..., -1.5163e+00,
-1.4193e+00, -8.7298e-01],
[ 1.4109e+00, 8.0346e-01, -9.9451e-01, ..., -2.9707e-01,
6.6100e-02, 3.6332e-01],
[ 8.3872e-01, 2.8035e-01, -6.9979e-01, ..., -2.0381e-01,
1.7729e-01, 3.6575e-01]],
[[-1.6079e+00, -1.2206e+00, -4.1930e+00, ..., -5.0381e+00,
-4.6072e+00, -3.4903e+00],
[ 5.1241e-01, 1.6512e+00, -2.4589e+00, ..., -3.4482e+00,
-3.6191e+00, -2.9533e+00],
[ 1.1410e+00, 1.3456e+00, -5.0445e+00, ..., -5.4666e+00,
-5.3571e+00, -5.0325e+00],
...,
[-1.2679e+00, 1.7262e-01, -6.6832e+00, ..., -7.6123e+00,
-6.8701e+00, -5.3857e+00],
[ 1.8043e+00, 2.8344e+00, -1.0369e+00, ..., -2.5827e+00,
-2.8947e+00, -2.3442e+00],
[ 1.2134e+00, 1.8203e+00, -2.2828e+00, ..., -3.2226e+00,
-3.5772e+00, -3.1260e+00]],
[[-7.2166e-02, -1.0430e+00, -1.2423e+00, ..., -5.7092e-01,
-2.0703e-01, -5.4138e-01],
[-4.7461e-01, -6.8408e-01, -6.6631e-01, ..., -3.6458e-01,
-7.4267e-03, -8.6967e-02],
[-4.3144e-01, -2.2939e-01, -3.4797e-01, ..., -3.8976e-01,
8.9899e-02, 1.0142e-01],
...,
[-1.3894e+00, -1.7674e+00, -7.6724e-01, ..., -8.7352e-01,
-7.5019e-01, -1.0270e+00],
[-1.7441e+00, -1.1841e+00, -8.6087e-01, ..., -5.4874e-01,
-2.7658e-01, -3.5560e-01],
[-1.8942e+00, -1.2907e+00, -9.0867e-01, ..., -9.5218e-01,
-8.2050e-01, -7.9986e-01]],
...,
[[-2.9000e+00, -2.5052e+00, -2.7112e+00, ..., -3.7372e+00,
-2.8670e+00, -1.8963e+00],
[-2.2514e+00, -1.0215e+00, -1.6513e+00, ..., -2.7318e+00,
-2.4581e+00, -1.9457e+00],
[-1.0610e+00, 8.7399e-01, 4.2508e-01, ..., -4.2598e-01,
4.9076e-02, 1.7924e-01],
...,
[-1.8397e+00, 1.2748e+00, 2.7186e-02, ..., -1.1038e+00,
-5.2359e-01, 1.8198e-02],
[-7.8110e-01, 1.9538e+00, 1.6739e+00, ..., 4.2677e-01,
4.0931e-01, 6.8695e-01],
[-2.1901e+00, -1.1354e+00, -9.6090e-01, ..., -1.9319e+00,
-1.0143e+00, -3.0760e-01]],
[[-1.5981e+00, -1.6498e+00, -2.8407e+00, ..., -1.1471e+00,
-6.8614e-01, -7.3143e-01],
[-6.5975e-01, -8.9266e-02, -3.1389e+00, ..., -1.2327e+00,
-1.1018e+00, -1.5483e+00],
[-2.0967e+00, -1.0413e+00, -5.1366e+00, ..., -3.1985e+00,
-2.7420e+00, -2.2296e+00],
...,
[-2.4222e+00, -1.5138e+00, -5.4404e+00, ..., -3.0554e+00,
-2.6888e+00, -2.1524e+00],
[-2.1775e+00, -1.4234e+00, -4.2387e+00, ..., -2.6404e+00,
-2.2689e+00, -1.6281e+00],
[-2.3673e+00, -1.7524e+00, -3.2893e+00, ..., -2.9406e+00,
-2.4163e+00, -2.1238e+00]],
[[-1.3247e+00, -1.8653e+00, -1.7095e+00, ..., -1.8382e+00,
-1.5203e+00, -1.1540e+00],
[-1.2612e+00, -1.7315e+00, -1.2865e+00, ..., -1.3842e+00,
-1.0894e+00, -8.6687e-01],
[-1.5816e+00, -1.9284e+00, -1.5391e+00, ..., -1.5818e+00,
-1.1795e+00, -8.5830e-01],
...,
[-3.1939e+00, -3.9917e+00, -3.7783e+00, ..., -3.9587e+00,
-3.1755e+00, -2.4481e+00],
[-2.1282e+00, -2.7436e+00, -2.7128e+00, ..., -2.7713e+00,
-2.2928e+00, -1.8650e+00],
[-1.5452e+00, -1.8722e+00, -1.9334e+00, ..., -1.8659e+00,
-1.5189e+00, -1.1611e+00]]],
[[[ 2.2251e-01, 4.9467e-01, 6.8637e-01, ..., 6.2998e-01,
1.0031e+00, 7.9721e-01],
[-1.9871e+00, -1.2862e+00, -2.7427e+00, ..., -1.4111e+00,
3.1374e-02, 6.1084e-01],
[-1.7866e-01, -9.8723e-01, -6.6820e-03, ..., 4.4904e-01,
9.4024e-01, 9.4401e-01],
...,
[-9.7257e-01, -2.1784e+00, -2.0757e+00, ..., -1.0885e+00,
-6.7554e-01, -5.0691e-01],
[ 7.9063e-01, -1.3686e-01, -1.3464e-01, ..., 5.0010e-01,
8.5130e-01, 8.8324e-01],
[ 2.0293e-01, -5.8285e-01, -4.2262e-01, ..., 6.0451e-01,
8.6544e-01, 7.6136e-01]],
[[-1.6303e+00, -1.2529e+00, -4.2198e+00, ..., -5.9294e+00,
-5.0027e+00, -3.9027e+00],
[ 5.0801e-01, 1.5066e+00, -2.5838e+00, ..., -4.6948e+00,
-4.4599e+00, -3.6654e+00],
[ 1.2204e+00, 1.1799e+00, -5.1675e+00, ..., -6.8798e+00,
-6.4635e+00, -5.9161e+00],
...,
[ 3.7507e-01, 3.1098e-01, -4.1288e+00, ..., -6.0753e+00,
-5.1848e+00, -4.2273e+00],
[ 2.5099e+00, 2.4571e+00, 9.3952e-02, ..., -1.7611e+00,
-1.7133e+00, -1.6596e+00],
[ 1.4965e+00, 1.3041e+00, -1.6246e+00, ..., -2.8294e+00,
-2.8113e+00, -2.7152e+00]],
[[-9.3008e-02, -1.1574e+00, -1.4148e+00, ..., -8.4813e-01,
-5.8807e-01, -7.2175e-01],
[-4.9174e-01, -7.9767e-01, -8.4347e-01, ..., -7.1525e-01,
-3.8197e-01, -2.4519e-01],
[-3.4890e-01, -2.0962e-01, -4.0823e-01, ..., -3.8062e-01,
1.5435e-02, 9.8936e-02],
...,
[-5.0532e-02, -5.7053e-01, -6.0720e-01, ..., -5.6096e-01,
-4.0581e-01, -5.2123e-01],
[-7.0210e-01, -2.0900e-01, -5.7055e-01, ..., -2.2338e-01,
3.8284e-02, 6.6858e-02],
[-8.4740e-01, -2.3555e-01, -4.6493e-01, ..., -2.5990e-01,
-1.4582e-01, -1.0895e-01]],
...,
[[-2.8562e+00, -2.3889e+00, -2.3858e+00, ..., -3.3932e+00,
-2.7646e+00, -1.8512e+00],
[-2.1468e+00, -8.5253e-01, -1.0965e+00, ..., -1.7761e+00,
-1.8590e+00, -1.5343e+00],
[-1.0818e+00, 8.0931e-01, 7.3109e-01, ..., -2.3767e-01,
3.2885e-02, 2.1035e-01],
...,
[-2.3002e+00, -2.2517e-01, -1.6507e-01, ..., -1.0031e+00,
-3.3650e-01, 1.5896e-01],
[-1.2683e+00, 8.2020e-01, 1.0556e+00, ..., 1.1473e-01,
2.6474e-01, 4.0824e-01],
[-2.2007e+00, -1.5305e+00, -1.7071e+00, ..., -2.1126e+00,
-1.1777e+00, -5.8066e-01]],
[[-1.6150e+00, -1.6726e+00, -2.8864e+00, ..., -1.2068e+00,
-6.1671e-01, -4.4069e-01],
[-6.8615e-01, -1.2497e-01, -3.2411e+00, ..., -1.9004e+00,
-1.5668e+00, -1.7174e+00],
[-2.0305e+00, -9.4258e-01, -5.1324e+00, ..., -4.0181e+00,
-3.3888e+00, -2.6276e+00],
...,
[-1.3596e+00, 1.7391e-01, -3.3505e+00, ..., -2.5739e+00,
-1.9053e+00, -1.3780e+00],
[-1.6949e+00, -4.7736e-01, -2.9922e+00, ..., -2.3783e+00,
-1.5615e+00, -9.3494e-01],
[-2.3361e+00, -1.3618e+00, -3.0405e+00, ..., -2.8059e+00,
-1.8654e+00, -1.6062e+00]],
[[-1.3107e+00, -1.8390e+00, -1.6473e+00, ..., -1.9058e+00,
-1.5463e+00, -1.1244e+00],
[-1.2599e+00, -1.7246e+00, -1.2480e+00, ..., -1.6225e+00,
-1.2859e+00, -9.8116e-01],
[-1.5921e+00, -1.9357e+00, -1.5102e+00, ..., -1.8802e+00,
-1.4045e+00, -9.8985e-01],
...,
[-2.4430e+00, -3.0558e+00, -2.7886e+00, ..., -3.0944e+00,
-2.4414e+00, -1.8263e+00],
[-1.5798e+00, -2.0834e+00, -1.9904e+00, ..., -2.2017e+00,
-1.7877e+00, -1.4049e+00],
[-1.0320e+00, -1.3445e+00, -1.3596e+00, ..., -1.3899e+00,
-1.1601e+00, -8.4448e-01]]],
[[[ 2.6897e-01, 5.5768e-01, 7.3870e-01, ..., 4.9932e-01,
1.0050e+00, 6.0063e-01],
[-2.1897e+00, -1.4976e+00, -3.0895e+00, ..., -2.5608e+00,
-8.2421e-01, -1.7803e-01],
[-1.8856e-01, -1.0056e+00, -1.2779e-01, ..., -1.3714e-01,
5.4575e-01, 6.5874e-01],
...,
[-1.8697e-01, -1.1527e+00, -2.6792e+00, ..., -1.5241e+00,
-1.4073e+00, -8.6943e-01],
[ 1.4238e+00, 8.4769e-01, -9.7349e-01, ..., -2.7056e-01,
1.1335e-01, 3.8000e-01],
[ 8.4879e-01, 3.0477e-01, -6.8879e-01, ..., -1.6674e-01,
2.2938e-01, 3.8943e-01]],
[[-1.6350e+00, -1.0969e+00, -4.1707e+00, ..., -4.8470e+00,
-4.4061e+00, -3.3602e+00],
[ 6.2100e-01, 1.9621e+00, -2.2903e+00, ..., -3.5618e+00,
-3.6277e+00, -2.9821e+00],
[ 1.3416e+00, 1.7031e+00, -4.9553e+00, ..., -5.4787e+00,
-5.2754e+00, -4.9144e+00],
...,
[-1.2539e+00, 1.8340e-01, -6.6643e+00, ..., -7.6011e+00,
-6.9225e+00, -5.4279e+00],
[ 1.8133e+00, 2.8199e+00, -1.0454e+00, ..., -2.5832e+00,
-2.9582e+00, -2.3702e+00],
[ 1.2203e+00, 1.7981e+00, -2.2879e+00, ..., -3.2452e+00,
-3.6132e+00, -3.1294e+00]],
[[-5.9557e-02, -1.0999e+00, -1.3203e+00, ..., -4.6572e-01,
-1.3677e-01, -4.6055e-01],
[-5.4922e-01, -8.0633e-01, -7.6009e-01, ..., -2.2909e-01,
6.7339e-02, -6.1922e-02],
[-5.0587e-01, -3.1658e-01, -3.7791e-01, ..., -3.3786e-01,
6.9837e-02, 6.2530e-02],
...,
[-1.3876e+00, -1.7527e+00, -7.7674e-01, ..., -8.9149e-01,
-7.8395e-01, -1.0263e+00],
[-1.7383e+00, -1.1641e+00, -8.5023e-01, ..., -5.7722e-01,
-3.0410e-01, -3.6127e-01],
[-1.8878e+00, -1.2831e+00, -8.9619e-01, ..., -9.4924e-01,
-8.1928e-01, -7.8993e-01]],
...,
[[-3.0589e+00, -2.5714e+00, -2.8621e+00, ..., -3.4130e+00,
-2.5748e+00, -1.7000e+00],
[-2.3276e+00, -9.0506e-01, -1.6933e+00, ..., -2.6621e+00,
-2.3451e+00, -1.8115e+00],
[-1.1373e+00, 1.0974e+00, 5.0484e-01, ..., -4.5996e-01,
-4.0015e-02, 1.1717e-01],
...,
[-1.8022e+00, 1.2956e+00, 8.6934e-02, ..., -1.0825e+00,
-5.3492e-01, 4.2273e-03],
[-7.8050e-01, 1.9374e+00, 1.6887e+00, ..., 5.0226e-01,
4.2196e-01, 6.6960e-01],
[-2.1932e+00, -1.1310e+00, -9.6139e-01, ..., -1.8644e+00,
-9.7169e-01, -3.1746e-01]],
[[-1.5951e+00, -1.5875e+00, -2.9586e+00, ..., -1.3352e+00,
-8.7869e-01, -8.8353e-01],
[-5.8673e-01, 4.2434e-02, -3.2778e+00, ..., -1.4259e+00,
-1.2391e+00, -1.5778e+00],
[-2.0588e+00, -9.1754e-01, -5.3751e+00, ..., -3.1807e+00,
-2.6862e+00, -2.1973e+00],
...,
[-2.4413e+00, -1.5319e+00, -5.4441e+00, ..., -3.0720e+00,
-2.7138e+00, -2.1597e+00],
[-2.1943e+00, -1.4334e+00, -4.2394e+00, ..., -2.6846e+00,
-2.2947e+00, -1.6119e+00],
[-2.3757e+00, -1.7440e+00, -3.2804e+00, ..., -2.9752e+00,
-2.4276e+00, -2.0972e+00]],
[[-1.3732e+00, -1.9256e+00, -1.7426e+00, ..., -1.7646e+00,
-1.4574e+00, -1.1081e+00],
[-1.3074e+00, -1.7871e+00, -1.2699e+00, ..., -1.3889e+00,
-1.0975e+00, -8.7891e-01],
[-1.6236e+00, -1.9840e+00, -1.5219e+00, ..., -1.6267e+00,
-1.2209e+00, -9.0861e-01],
...,
[-3.1737e+00, -3.9666e+00, -3.7567e+00, ..., -3.9376e+00,
-3.1541e+00, -2.4350e+00],
[-2.1163e+00, -2.7323e+00, -2.6990e+00, ..., -2.7634e+00,
-2.2769e+00, -1.8536e+00],
[-1.5433e+00, -1.8695e+00, -1.9302e+00, ..., -1.8611e+00,
-1.5118e+00, -1.1562e+00]]],
[[[ 1.0297e-01, 3.3331e-01, -1.1952e+00, ..., 6.8709e-01,
9.2662e-01, 6.7545e-01],
[-2.0090e+00, -1.3771e+00, -4.1842e+00, ..., -1.4022e+00,
-1.6863e-01, 3.6951e-01],
[ 3.4597e-01, -2.8686e-01, -1.0965e+00, ..., 5.7131e-01,
9.2562e-01, 8.3767e-01],
...,
[-3.0233e-01, -1.4212e+00, -2.7223e+00, ..., -1.1949e+00,
-8.0602e-01, -6.2863e-01],
[ 1.3438e+00, 6.9579e-01, -8.5395e-01, ..., 4.9496e-01,
7.5679e-01, 7.9210e-01],
[ 8.9359e-01, 3.7127e-01, -3.6120e-01, ..., 5.2609e-01,
7.7181e-01, 6.7961e-01]],
[[-3.1571e+00, -1.5267e+00, -6.0974e+00, ..., -5.4247e+00,
-4.6493e+00, -3.6320e+00],
[-7.1822e-01, 1.7603e+00, -4.4055e+00, ..., -4.2032e+00,
-4.0879e+00, -3.3459e+00],
[-2.4285e-01, 1.4123e+00, -7.4803e+00, ..., -6.1124e+00,
-5.8709e+00, -5.4169e+00],
...,
[-1.0589e+00, 5.8759e-01, -6.3950e+00, ..., -5.3838e+00,
-4.7309e+00, -3.9454e+00],
[ 1.9251e+00, 3.0807e+00, -8.3614e-01, ..., -1.3192e+00,
-1.4560e+00, -1.5163e+00],
[ 1.2648e+00, 1.9190e+00, -2.1739e+00, ..., -2.5121e+00,
-2.6183e+00, -2.5474e+00]],
[[-1.3124e+00, -2.1747e+00, -2.0131e+00, ..., -8.3112e-01,
-6.2339e-01, -7.8852e-01],
[-1.8646e+00, -2.0172e+00, -1.4517e+00, ..., -6.8838e-01,
-4.2082e-01, -3.5072e-01],
[-2.3930e+00, -1.9066e+00, -8.2157e-01, ..., -4.0090e-01,
-1.0533e-01, -9.5422e-02],
...,
[-1.7149e+00, -1.9934e+00, -8.2160e-01, ..., -6.3830e-01,
-5.3630e-01, -6.8909e-01],
[-1.9278e+00, -1.3214e+00, -8.2555e-01, ..., -2.2932e-01,
-5.3050e-02, -5.1430e-02],
[-1.9144e+00, -1.2261e+00, -6.2745e-01, ..., -3.2250e-01,
-2.8033e-01, -2.5506e-01]],
...,
[[-2.1292e+00, -1.2228e+00, -2.5514e+00, ..., -3.0182e+00,
-2.3659e+00, -1.5890e+00],
[-1.8168e+00, 5.6692e-02, -1.3281e+00, ..., -1.4885e+00,
-1.5171e+00, -1.3199e+00],
[-4.6551e-01, 2.2586e+00, 8.3363e-01, ..., -1.2477e-01,
2.1086e-01, 3.1059e-01],
...,
[-1.9053e+00, 1.2976e+00, 1.7643e-01, ..., -9.5835e-01,
-2.7084e-01, 1.7219e-01],
[-8.8721e-01, 1.8282e+00, 1.5617e+00, ..., 1.4740e-01,
3.0430e-01, 4.0293e-01],
[-2.1830e+00, -1.1162e+00, -1.0119e+00, ..., -2.1521e+00,
-1.2522e+00, -6.4495e-01]],
[[-2.3774e+00, -2.6809e+00, -3.9214e+00, ..., -1.1897e+00,
-6.5266e-01, -4.9750e-01],
[-1.7722e+00, -1.5469e+00, -4.6460e+00, ..., -1.7395e+00,
-1.5004e+00, -1.7085e+00],
[-3.0216e+00, -2.5790e+00, -6.6168e+00, ..., -3.7483e+00,
-3.2553e+00, -2.6017e+00],
...,
[-2.4137e+00, -1.5095e+00, -5.1329e+00, ..., -2.2103e+00,
-1.7494e+00, -1.4113e+00],
[-2.1620e+00, -1.4054e+00, -4.0077e+00, ..., -2.1395e+00,
-1.4957e+00, -9.7409e-01],
[-2.3386e+00, -1.7496e+00, -3.1545e+00, ..., -2.8163e+00,
-1.9455e+00, -1.6607e+00]],
[[-1.8444e+00, -2.5397e+00, -2.3703e+00, ..., -1.6731e+00,
-1.4014e+00, -1.0406e+00],
[-1.8876e+00, -2.5759e+00, -2.1225e+00, ..., -1.3624e+00,
-1.1147e+00, -8.7883e-01],
[-2.4090e+00, -2.9703e+00, -2.5671e+00, ..., -1.5791e+00,
-1.2043e+00, -8.7162e-01],
...,
[-3.2272e+00, -4.0131e+00, -3.7328e+00, ..., -2.7809e+00,
-2.2187e+00, -1.6796e+00],
[-2.1357e+00, -2.7491e+00, -2.6661e+00, ..., -1.9575e+00,
-1.6061e+00, -1.2799e+00],
[-1.5214e+00, -1.8338e+00, -1.8597e+00, ..., -1.2232e+00,
-1.0397e+00, -7.6975e-01]]],
[[[ 2.8613e-01, 5.4912e-01, 6.7188e-01, ..., 4.9302e-01,
1.0246e+00, 6.2205e-01],
[-2.2059e+00, -1.5492e+00, -3.2592e+00, ..., -2.6313e+00,
-8.5909e-01, -1.9329e-01],
[-1.9142e-01, -1.0167e+00, -2.7164e-01, ..., -1.5592e-01,
5.1469e-01, 6.2848e-01],
...,
[-1.7229e-01, -1.0164e+00, -2.4600e+00, ..., -1.7082e+00,
-1.5825e+00, -1.0617e+00],
[ 1.3076e+00, 8.0827e-01, -9.2768e-01, ..., -3.0609e-01,
8.0304e-02, 3.4812e-01],
[ 7.7299e-01, 2.7570e-01, -6.6131e-01, ..., -1.5426e-01,
2.4823e-01, 3.7400e-01]],
[[-1.7024e+00, -1.1662e+00, -4.1780e+00, ..., -4.8557e+00,
-4.4229e+00, -3.3969e+00],
[ 5.8366e-01, 1.9575e+00, -2.2451e+00, ..., -3.5402e+00,
-3.6508e+00, -3.0068e+00],
[ 1.3313e+00, 1.7279e+00, -4.9026e+00, ..., -5.5040e+00,
-5.3051e+00, -4.9434e+00],
...,
[-1.5324e+00, -2.8470e-01, -6.8880e+00, ..., -7.6443e+00,
-6.9099e+00, -5.3496e+00],
[ 1.3399e+00, 2.2795e+00, -1.5449e+00, ..., -2.3439e+00,
-2.7213e+00, -2.1685e+00],
[ 8.9498e-01, 1.4736e+00, -2.4825e+00, ..., -3.1678e+00,
-3.5802e+00, -3.1021e+00]],
[[-1.0919e-01, -1.1420e+00, -1.3204e+00, ..., -4.7821e-01,
-1.5652e-01, -4.7967e-01],
[-5.7503e-01, -8.4508e-01, -7.7826e-01, ..., -2.6540e-01,
5.8824e-02, -6.5577e-02],
[-5.2917e-01, -3.5453e-01, -4.1634e-01, ..., -3.5449e-01,
7.5425e-02, 6.7847e-02],
...,
[-1.2089e+00, -1.5675e+00, -7.4780e-01, ..., -9.0271e-01,
-7.5674e-01, -1.0468e+00],
[-1.5166e+00, -1.0166e+00, -7.8728e-01, ..., -5.6915e-01,
-2.5939e-01, -3.4085e-01],
[-1.6382e+00, -1.1408e+00, -8.5624e-01, ..., -9.4229e-01,
-7.9450e-01, -7.7630e-01]],
...,
[[-3.0517e+00, -2.5475e+00, -2.8450e+00, ..., -3.4183e+00,
-2.5961e+00, -1.7224e+00],
[-2.2974e+00, -8.7523e-01, -1.6253e+00, ..., -2.6485e+00,
-2.3669e+00, -1.8449e+00],
[-1.0728e+00, 1.1607e+00, 6.0977e-01, ..., -4.3789e-01,
-3.1233e-02, 1.1195e-01],
...,
[-1.6938e+00, 9.8809e-01, -9.9128e-02, ..., -1.2044e+00,
-5.3340e-01, 7.3451e-02],
[-7.6743e-01, 1.6497e+00, 1.4588e+00, ..., 4.4266e-01,
4.3624e-01, 7.4209e-01],
[-2.1551e+00, -1.1761e+00, -9.9095e-01, ..., -2.0896e+00,
-1.0950e+00, -3.5922e-01]],
[[-1.5953e+00, -1.5708e+00, -2.8764e+00, ..., -1.2818e+00,
-8.9772e-01, -8.8699e-01],
[-5.8598e-01, 1.1614e-01, -3.1988e+00, ..., -1.3834e+00,
-1.2504e+00, -1.5921e+00],
[-2.0990e+00, -8.8522e-01, -5.3365e+00, ..., -3.1623e+00,
-2.6959e+00, -2.1950e+00],
...,
[-2.5288e+00, -1.8191e+00, -5.2662e+00, ..., -3.0442e+00,
-2.5810e+00, -2.0957e+00],
[-2.2556e+00, -1.6523e+00, -4.1330e+00, ..., -2.6535e+00,
-2.1844e+00, -1.5473e+00],
[-2.3389e+00, -1.8568e+00, -3.2868e+00, ..., -3.0250e+00,
-2.4103e+00, -2.1285e+00]],
[[-1.3579e+00, -1.9213e+00, -1.7185e+00, ..., -1.7613e+00,
-1.4561e+00, -1.0992e+00],
[-1.2785e+00, -1.7705e+00, -1.2350e+00, ..., -1.3749e+00,
-1.0850e+00, -8.6464e-01],
[-1.5985e+00, -1.9536e+00, -1.4810e+00, ..., -1.6118e+00,
-1.2074e+00, -8.9001e-01],
...,
[-3.0476e+00, -3.8193e+00, -3.6976e+00, ..., -4.0531e+00,
-3.2563e+00, -2.5060e+00],
[-2.0296e+00, -2.6227e+00, -2.6517e+00, ..., -2.8281e+00,
-2.3435e+00, -1.9087e+00],
[-1.4842e+00, -1.8054e+00, -1.8972e+00, ..., -1.8850e+00,
-1.5452e+00, -1.1823e+00]]]], device='cuda:0'),
tensor([[[[ 2.2935e-01, 4.9673e-01, 6.4455e-01, ..., 5.6523e-01,
9.6442e-01, 7.6684e-01],
[-1.9843e+00, -1.2967e+00, -2.7656e+00, ..., -1.5314e+00,
-2.8952e-02, 5.6831e-01],
[-1.2478e-01, -8.5435e-01, 5.5294e-02, ..., 4.3324e-01,
9.1829e-01, 9.2213e-01],
...,
[-9.3192e-01, -2.1245e+00, -2.0763e+00, ..., -1.1922e+00,
-7.3316e-01, -4.8867e-01],
[ 8.3495e-01, -9.8745e-02, -1.2554e-01, ..., 4.3089e-01,
7.8800e-01, 8.7791e-01],
[ 2.5765e-01, -5.2367e-01, -3.5810e-01, ..., 5.9528e-01,
8.5494e-01, 7.7226e-01]],
[[-1.6983e+00, -1.3931e+00, -4.2685e+00, ..., -6.0111e+00,
-5.0925e+00, -3.9511e+00],
[ 4.3133e-01, 1.4318e+00, -2.5622e+00, ..., -4.7215e+00,
-4.5135e+00, -3.6874e+00],
[ 1.1467e+00, 1.1621e+00, -5.0765e+00, ..., -6.9315e+00,
-6.5408e+00, -5.9664e+00],
...,
[ 3.7965e-01, 4.1816e-01, -4.1057e+00, ..., -6.1007e+00,
-5.2649e+00, -4.3488e+00],
[ 2.5751e+00, 2.5688e+00, 1.7422e-01, ..., -1.7178e+00,
-1.7617e+00, -1.7684e+00],
[ 1.5359e+00, 1.3650e+00, -1.6075e+00, ..., -2.8463e+00,
-2.9054e+00, -2.7862e+00]],
[[-1.5250e-01, -1.1748e+00, -1.4038e+00, ..., -8.8555e-01,
-6.0135e-01, -7.4051e-01],
[-5.3388e-01, -8.5624e-01, -8.9832e-01, ..., -7.5682e-01,
-4.0448e-01, -2.5873e-01],
[-3.5582e-01, -2.5971e-01, -4.4510e-01, ..., -4.2488e-01,
-3.2365e-03, 1.0298e-01],
...,
[-1.1289e-01, -6.4959e-01, -5.9999e-01, ..., -6.5608e-01,
-4.6147e-01, -5.3089e-01],
[-7.4630e-01, -2.5787e-01, -5.5658e-01, ..., -3.0932e-01,
-4.2609e-02, 4.6697e-02],
[-9.0200e-01, -2.9702e-01, -4.6843e-01, ..., -3.0803e-01,
-2.0923e-01, -1.4468e-01]],
...,
[[-2.7978e+00, -2.4160e+00, -2.4694e+00, ..., -3.4363e+00,
-2.7933e+00, -1.8634e+00],
[-2.0373e+00, -7.8436e-01, -1.1150e+00, ..., -1.8053e+00,
-1.8879e+00, -1.5560e+00],
[-1.0840e+00, 8.2775e-01, 7.2992e-01, ..., -2.2757e-01,
6.4768e-02, 2.2327e-01],
...,
[-2.2649e+00, -9.6418e-02, -1.2458e-01, ..., -9.8924e-01,
-3.7519e-01, 1.3698e-01],
[-1.2250e+00, 9.3465e-01, 1.0853e+00, ..., 1.9193e-01,
2.9736e-01, 4.0878e-01],
[-2.1838e+00, -1.4814e+00, -1.6772e+00, ..., -2.0209e+00,
-1.1401e+00, -5.6517e-01]],
[[-1.6183e+00, -1.6270e+00, -2.7758e+00, ..., -1.2527e+00,
-6.5050e-01, -4.6254e-01],
[-7.0258e-01, -1.1541e-01, -3.1397e+00, ..., -1.9301e+00,
-1.5874e+00, -1.7342e+00],
[-2.0710e+00, -1.0089e+00, -5.1335e+00, ..., -4.0691e+00,
-3.4354e+00, -2.6347e+00],
...,
[-1.3555e+00, 1.9105e-01, -3.4196e+00, ..., -2.5565e+00,
-1.9332e+00, -1.4408e+00],
[-1.6777e+00, -4.5448e-01, -3.0330e+00, ..., -2.3522e+00,
-1.5926e+00, -9.7072e-01],
[-2.3332e+00, -1.3454e+00, -3.0226e+00, ..., -2.8332e+00,
-1.9260e+00, -1.6011e+00]],
[[-1.2891e+00, -1.8149e+00, -1.6421e+00, ..., -1.9611e+00,
-1.5888e+00, -1.1556e+00],
[-1.2301e+00, -1.6957e+00, -1.2132e+00, ..., -1.6689e+00,
-1.3190e+00, -1.0045e+00],
[-1.5480e+00, -1.8867e+00, -1.4673e+00, ..., -1.9257e+00,
-1.4382e+00, -1.0097e+00],
...,
[-2.4522e+00, -3.0644e+00, -2.7849e+00, ..., -3.1057e+00,
-2.4427e+00, -1.8181e+00],
[-1.5875e+00, -2.0958e+00, -1.9856e+00, ..., -2.2079e+00,
-1.7867e+00, -1.3987e+00],
[-1.0340e+00, -1.3513e+00, -1.3677e+00, ..., -1.3924e+00,
-1.1565e+00, -8.4033e-01]]],
[[[ 6.9063e-02, 2.3319e-01, -1.3906e+00, ..., -5.8458e-01,
-1.9828e-01, -4.0407e-02],
[-2.0983e+00, -1.6023e+00, -4.5072e+00, ..., -3.6752e+00,
-1.5089e+00, -5.3724e-01],
[ 5.6677e-01, -1.0733e-01, -8.7233e-01, ..., -1.5542e-01,
-2.6827e-01, 2.4508e-01],
...,
[-3.2144e-01, -1.1130e+00, -1.4069e+00, ..., -1.3119e+00,
-9.0917e-01, -6.8278e-01],
[ 9.4695e-01, 5.3932e-01, 3.0420e-01, ..., 3.8707e-01,
6.9074e-01, 7.9381e-01],
[ 6.5981e-01, 4.1222e-01, 3.2063e-01, ..., 5.1860e-01,
7.6400e-01, 7.0106e-01]],
[[-3.2983e+00, -1.7727e+00, -6.2278e+00, ..., -7.1538e+00,
-6.2110e+00, -4.6239e+00],
[-7.8846e-01, 1.5732e+00, -4.3154e+00, ..., -5.5364e+00,
-5.5347e+00, -4.2907e+00],
[-3.4248e-01, 1.1165e+00, -7.4056e+00, ..., -8.0131e+00,
-7.8227e+00, -6.6788e+00],
...,
[-4.6013e-01, 2.7250e-01, -4.4021e+00, ..., -5.3381e+00,
-4.6362e+00, -3.8734e+00],
[ 1.9545e+00, 2.4391e+00, -5.2564e-02, ..., -1.0789e+00,
-1.2275e+00, -1.3588e+00],
[ 1.1167e+00, 1.3131e+00, -1.6955e+00, ..., -2.4550e+00,
-2.5783e+00, -2.5526e+00]],
[[-1.3373e+00, -2.3573e+00, -2.3264e+00, ..., -1.7310e+00,
-1.2340e+00, -1.3822e+00],
[-1.7048e+00, -1.9949e+00, -1.5351e+00, ..., -1.2194e+00,
-7.3492e-01, -8.2684e-01],
[-2.1920e+00, -1.9452e+00, -1.0177e+00, ..., -1.1020e+00,
-4.7697e-01, -1.9155e-01],
...,
[-6.2553e-01, -1.0946e+00, -5.6287e-01, ..., -5.9731e-01,
-4.7867e-01, -6.3230e-01],
[-1.0307e+00, -6.9141e-01, -5.5712e-01, ..., -2.5300e-01,
-5.6921e-02, -4.0898e-02],
[-1.1983e+00, -6.8758e-01, -4.0162e-01, ..., -3.2206e-01,
-2.7329e-01, -2.4395e-01]],
...,
[[-2.0421e+00, -1.0972e+00, -2.3608e+00, ..., -3.1211e+00,
-2.5541e+00, -1.7816e+00],
[-1.7452e+00, 5.0473e-02, -1.3621e+00, ..., -2.4964e+00,
-2.4661e+00, -1.9061e+00],
[-3.3886e-01, 2.4186e+00, 1.0744e+00, ..., 9.5822e-02,
2.8861e-01, 5.2969e-01],
...,
[-1.5658e+00, 9.1132e-01, -1.6152e-01, ..., -1.1682e+00,
-4.4140e-01, 4.9937e-02],
[-5.5070e-01, 1.3753e+00, 7.3476e-01, ..., -1.6421e-03,
1.8461e-01, 2.9940e-01],
[-1.5965e+00, -1.0541e+00, -1.5105e+00, ..., -2.3076e+00,
-1.3396e+00, -7.0506e-01]],
[[-2.4044e+00, -2.8119e+00, -4.2303e+00, ..., -2.4450e+00,
-1.9301e+00, -1.7324e+00],
[-1.7462e+00, -1.6141e+00, -4.8986e+00, ..., -2.9352e+00,
-2.7711e+00, -2.7630e+00],
[-2.8121e+00, -2.4356e+00, -6.6988e+00, ..., -4.4440e+00,
-4.3434e+00, -3.4685e+00],
...,
[-1.6873e+00, -5.4597e-01, -4.0467e+00, ..., -2.0667e+00,
-1.5491e+00, -1.2152e+00],
[-1.6716e+00, -7.1325e-01, -3.3603e+00, ..., -2.0410e+00,
-1.3462e+00, -8.0809e-01],
[-2.0398e+00, -1.2610e+00, -2.8211e+00, ..., -2.6773e+00,
-1.7918e+00, -1.5317e+00]],
[[-1.8087e+00, -2.5386e+00, -2.4478e+00, ..., -2.7007e+00,
-2.2152e+00, -1.6984e+00],
[-1.8033e+00, -2.5103e+00, -2.1378e+00, ..., -2.4255e+00,
-1.9148e+00, -1.5525e+00],
[-2.2569e+00, -2.8211e+00, -2.5054e+00, ..., -2.7722e+00,
-2.1640e+00, -1.6432e+00],
...,
[-2.2661e+00, -2.8472e+00, -2.6546e+00, ..., -2.8729e+00,
-2.2897e+00, -1.7268e+00],
[-1.5321e+00, -1.9756e+00, -1.9275e+00, ..., -2.0335e+00,
-1.6711e+00, -1.3306e+00],
[-1.0385e+00, -1.2703e+00, -1.3416e+00, ..., -1.2657e+00,
-1.0793e+00, -7.9855e-01]]],
[[[ 3.2109e-01, 5.9904e-01, 7.1713e-01, ..., 5.1441e-01,
1.0146e+00, 6.2003e-01],
[-2.0509e+00, -1.3886e+00, -3.0055e+00, ..., -2.5070e+00,
-7.6292e-01, -1.2468e-01],
[-6.7385e-02, -8.7161e-01, -1.5558e-01, ..., -8.1560e-02,
5.9750e-01, 6.8185e-01],
...,
[ 1.1663e-01, -6.8129e-01, -1.6779e+00, ..., -3.9728e+00,
-8.5382e-01, -6.1911e-01],
[ 1.5299e+00, 1.2251e+00, 4.9030e-01, ..., 1.8663e+00,
8.9814e-01, 9.2106e-01],
[ 1.1681e+00, 9.9268e-01, 6.4923e-01, ..., 8.6627e-01,
9.9751e-01, 8.5493e-01]],
[[-1.7208e+00, -1.1558e+00, -4.1768e+00, ..., -4.8914e+00,
-4.4298e+00, -3.3860e+00],
[ 5.1378e-01, 1.9034e+00, -2.3008e+00, ..., -3.5583e+00,
-3.6553e+00, -2.9886e+00],
[ 1.2488e+00, 1.6704e+00, -4.9392e+00, ..., -5.5023e+00,
-5.3077e+00, -4.9542e+00],
...,
[-7.1133e-01, 4.9994e-01, -4.8592e+00, ..., -6.5937e+00,
-5.3440e+00, -4.2972e+00],
[ 2.1018e+00, 2.8705e+00, 7.8663e-02, ..., -2.4628e+00,
-1.8054e+00, -1.7209e+00],
[ 1.2335e+00, 1.5784e+00, -1.7343e+00, ..., -2.8745e+00,
-2.8919e+00, -2.7553e+00]],
[[-1.5599e-01, -1.1815e+00, -1.3133e+00, ..., -4.8816e-01,
-1.7232e-01, -4.7468e-01],
[-6.5711e-01, -9.1892e-01, -7.7944e-01, ..., -2.9681e-01,
1.6484e-02, -8.9506e-02],
[-5.9676e-01, -4.0847e-01, -3.8357e-01, ..., -3.4968e-01,
7.4358e-02, 6.1784e-02],
...,
[-1.1311e+00, -1.6117e+00, -7.1242e-01, ..., 2.0947e-01,
-4.9487e-01, -6.6110e-01],
[-1.3294e+00, -8.6989e-01, -4.7713e-01, ..., 4.7915e-01,
5.7545e-02, 3.0985e-02],
[-1.3136e+00, -6.3688e-01, -1.2039e-01, ..., -2.8144e-03,
-4.9712e-02, -8.1766e-02]],
...,
[[-2.9800e+00, -2.3809e+00, -2.7609e+00, ..., -3.4080e+00,
-2.6056e+00, -1.7441e+00],
[-2.1614e+00, -6.5996e-01, -1.5267e+00, ..., -2.5442e+00,
-2.3034e+00, -1.7744e+00],
[-1.0077e+00, 1.2718e+00, 6.0131e-01, ..., -4.6338e-01,
-5.1372e-02, 1.2155e-01],
...,
[-1.5814e+00, 1.6283e+00, 3.7846e-01, ..., 1.3555e+00,
-1.3767e-01, 3.9085e-01],
[-6.0202e-01, 1.7978e+00, 1.1921e+00, ..., 8.8631e-01,
3.4442e-01, 5.2225e-01],
[-1.8095e+00, -1.0920e+00, -1.4351e+00, ..., -2.0244e+00,
-1.1439e+00, -5.0882e-01]],
[[-1.5718e+00, -1.5502e+00, -2.8826e+00, ..., -1.2495e+00,
-8.1263e-01, -8.1044e-01],
[-6.0500e-01, 4.6206e-02, -3.2447e+00, ..., -1.3944e+00,
-1.2169e+00, -1.5811e+00],
[-2.1063e+00, -9.5267e-01, -5.3663e+00, ..., -3.2247e+00,
-2.7206e+00, -2.2193e+00],
...,
[-1.7725e+00, -7.0188e-01, -4.6184e+00, ..., -8.9335e-01,
-2.1100e+00, -1.6695e+00],
[-1.6918e+00, -7.7124e-01, -3.6671e+00, ..., -1.5526e+00,
-1.7130e+00, -1.1274e+00],
[-2.0797e+00, -1.3020e+00, -3.0156e+00, ..., -2.6838e+00,
-1.9708e+00, -1.6987e+00]],
[[-1.3346e+00, -1.8891e+00, -1.6879e+00, ..., -1.7455e+00,
-1.4539e+00, -1.1021e+00],
[-1.2582e+00, -1.7466e+00, -1.2321e+00, ..., -1.3722e+00,
-1.0880e+00, -8.6673e-01],
[-1.5837e+00, -1.9327e+00, -1.4842e+00, ..., -1.6072e+00,
-1.2002e+00, -8.8816e-01],
...,
[-2.5434e+00, -3.1904e+00, -2.9363e+00, ..., -3.4600e+00,
-2.4442e+00, -1.8736e+00],
[-1.6927e+00, -2.1619e+00, -2.0703e+00, ..., -2.3942e+00,
-1.7505e+00, -1.4054e+00],
[-1.1214e+00, -1.3536e+00, -1.3982e+00, ..., -1.3512e+00,
-1.1232e+00, -8.3453e-01]]],
[[[ 4.2305e-01, 8.5607e-01, 5.2743e-01, ..., 4.2813e-01,
9.7278e-01, 6.3784e-01],
[-1.3146e+00, -4.1711e-01, -1.9740e+00, ..., -2.5021e+00,
-7.3664e-01, -6.0957e-02],
[ 8.8710e-01, 6.3139e-01, 3.2264e-01, ..., -2.4630e-01,
4.6718e-01, 6.2941e-01],
...,
[ 4.2313e-02, -5.8846e-01, -1.4296e+00, ..., -2.2297e+00,
-1.5879e+00, -1.1038e+00],
[ 1.3369e+00, 1.1417e+00, 5.1750e-01, ..., -3.6012e-01,
4.9560e-01, 6.9387e-01],
[ 9.9534e-01, 8.9455e-01, 5.6742e-01, ..., -3.3819e-01,
3.2498e-01, 2.9373e-01]],
[[-2.8024e+00, -1.5237e+00, -4.7788e+00, ..., -4.7392e+00,
-4.3454e+00, -3.3250e+00],
[-4.8511e-01, 1.6907e+00, -2.8625e+00, ..., -3.3704e+00,
-3.5165e+00, -2.9050e+00],
[ 3.4253e-01, 1.4854e+00, -5.6547e+00, ..., -5.2891e+00,
-5.1725e+00, -4.8643e+00],
...,
[-9.3037e-01, -1.1464e-02, -4.8427e+00, ..., -4.5437e+00,
-4.0712e+00, -3.3719e+00],
[ 1.6932e+00, 2.2970e+00, -2.5857e-01, ..., -5.3031e-01,
-9.0218e-01, -1.0169e+00],
[ 9.8435e-01, 1.2538e+00, -1.7957e+00, ..., -2.1401e+00,
-2.4154e+00, -2.3530e+00]],
[[-7.1298e-01, -1.9219e+00, -1.7818e+00, ..., -5.5618e-01,
-1.7247e-01, -4.3238e-01],
[-1.5053e+00, -1.9608e+00, -1.2722e+00, ..., -3.3587e-01,
7.3024e-02, 4.3931e-02],
[-1.3945e+00, -1.3685e+00, -7.4510e-01, ..., -2.8611e-01,
2.2560e-01, 2.6935e-01],
...,
[-9.0782e-01, -1.4450e+00, -7.1343e-01, ..., -5.4031e-01,
-1.8615e-01, -3.9906e-01],
[-1.1842e+00, -8.8170e-01, -6.0010e-01, ..., -3.2705e-01,
1.7389e-01, 2.3708e-01],
[-1.2284e+00, -7.0780e-01, -3.1367e-01, ..., -5.1084e-01,
-2.7086e-01, -2.5041e-01]],
...,
[[-2.3180e+00, -1.1588e+00, -2.5260e+00, ..., -3.5137e+00,
-2.7127e+00, -1.7609e+00],
[-6.0999e-01, 1.2630e+00, -7.3580e-01, ..., -2.5895e+00,
-2.3699e+00, -1.8315e+00],
[-1.7391e-01, 2.6001e+00, 8.1893e-01, ..., -5.5036e-01,
-7.7930e-02, 1.3778e-01],
...,
[-1.3618e+00, 1.3168e+00, 7.8126e-02, ..., -1.6058e+00,
-7.1628e-01, -9.4091e-02],
[-5.0515e-01, 1.4905e+00, 8.3902e-01, ..., -8.9644e-02,
1.6108e-01, 3.9458e-01],
[-1.6288e+00, -1.1044e+00, -1.5129e+00, ..., -2.5622e+00,
-1.5685e+00, -9.6463e-01]],
[[-1.3881e+00, -1.3709e+00, -3.0693e+00, ..., -1.1869e+00,
-7.8780e-01, -7.8843e-01],
[-6.2379e-01, -1.7382e-01, -3.9438e+00, ..., -1.3798e+00,
-1.2350e+00, -1.5707e+00],
[-2.1526e+00, -1.5017e+00, -6.4520e+00, ..., -3.2422e+00,
-2.7476e+00, -2.1941e+00],
...,
[-1.7959e+00, -9.2787e-01, -4.6061e+00, ..., -1.3090e+00,
-9.0954e-01, -7.8097e-01],
[-1.7117e+00, -9.8216e-01, -3.7204e+00, ..., -1.6523e+00,
-1.0658e+00, -6.7013e-01],
[-2.0451e+00, -1.4382e+00, -3.0389e+00, ..., -2.8749e+00,
-2.0331e+00, -1.8666e+00]],
[[-1.2668e+00, -1.9179e+00, -1.7347e+00, ..., -1.7778e+00,
-1.4743e+00, -1.1240e+00],
[-1.2582e+00, -1.8277e+00, -1.3834e+00, ..., -1.4060e+00,
-1.1108e+00, -8.8818e-01],
[-1.6291e+00, -2.0106e+00, -1.6038e+00, ..., -1.6217e+00,
-1.2130e+00, -8.9866e-01],
...,
[-2.3360e+00, -2.9587e+00, -2.8607e+00, ..., -2.9593e+00,
-2.3662e+00, -1.8182e+00],
[-1.5719e+00, -2.0229e+00, -2.0325e+00, ..., -2.0196e+00,
-1.6681e+00, -1.3491e+00],
[-1.0720e+00, -1.3106e+00, -1.4271e+00, ..., -1.2261e+00,
-1.0470e+00, -8.0061e-01]]],
[[[ 5.2017e-01, 9.8695e-01, 8.6771e-01, ..., -3.9268e-01,
-1.4823e-01, -2.9732e-02],
[-1.2938e+00, -7.8759e-01, -3.3540e+00, ..., -3.3509e+00,
-1.3711e+00, -5.0167e-01],
[ 5.0091e-01, -2.1103e+00, -5.0214e+00, ..., -2.0967e-01,
-3.8533e-01, 1.7068e-01],
...,
[-1.4308e-01, -7.7716e-01, -1.3662e+00, ..., -1.9673e+00,
-1.8969e+00, -1.2440e+00],
[ 1.0234e+00, 7.7904e-01, 3.3619e-01, ..., -3.0763e-02,
2.2243e-01, 4.3640e-01],
[ 6.8683e-01, 5.2667e-01, 2.4218e-01, ..., 1.6396e-01,
4.3149e-01, 4.8930e-01]],
[[-2.4061e+00, -1.0335e+00, -4.1772e+00, ..., -6.9291e+00,
-6.1265e+00, -4.5750e+00],
[ 3.4114e-02, 2.1787e+00, -2.3752e+00, ..., -5.5153e+00,
-5.6253e+00, -4.3350e+00],
[ 1.1614e+00, 2.3457e+00, -5.3350e+00, ..., -7.9313e+00,
-7.9049e+00, -6.6506e+00],
...,
[-5.6643e-01, 4.1906e-01, -4.2501e+00, ..., -7.2364e+00,
-6.8086e+00, -5.2820e+00],
[ 1.8614e+00, 2.4835e+00, -8.1684e-03, ..., -2.0019e+00,
-2.6417e+00, -2.1062e+00],
[ 1.0520e+00, 1.3305e+00, -1.6393e+00, ..., -2.9757e+00,
-3.5336e+00, -3.0847e+00]],
[[-6.7730e-01, -1.7808e+00, -1.5722e+00, ..., -1.4292e+00,
-1.0877e+00, -1.3692e+00],
[-1.5779e+00, -2.1088e+00, -1.4890e+00, ..., -9.9958e-01,
-6.4419e-01, -9.4898e-01],
[-1.3955e+00, -1.6525e+00, -9.6105e-01, ..., -1.1047e+00,
-5.9736e-01, -4.8267e-01],
...,
[-8.5343e-01, -1.2259e+00, -4.7338e-01, ..., -1.0464e+00,
-8.8019e-01, -1.2605e+00],
[-1.1719e+00, -7.7701e-01, -4.8476e-01, ..., -5.6713e-01,
-2.5450e-01, -4.1912e-01],
[-1.2526e+00, -6.8880e-01, -3.1372e-01, ..., -5.4614e-01,
-5.0508e-01, -6.6000e-01]],
...,
[[-2.4304e+00, -1.3128e+00, -2.6878e+00, ..., -3.0766e+00,
-2.5007e+00, -1.8496e+00],
[-6.2815e-01, 1.1727e+00, -4.1566e-01, ..., -2.4918e+00,
-2.5445e+00, -2.0219e+00],
[ 9.4831e-03, 2.0431e+00, -3.6476e-01, ..., -1.2187e-01,
-4.6661e-02, 2.5127e-01],
...,
[-1.4136e+00, 1.1962e+00, -1.6805e-01, ..., -9.3415e-01,
-2.8420e-01, 2.7484e-01],
[-4.9254e-01, 1.4497e+00, 7.2337e-01, ..., 4.8742e-01,
5.0235e-01, 8.3206e-01],
[-1.5627e+00, -1.0504e+00, -1.5334e+00, ..., -2.1702e+00,
-1.1436e+00, -3.2906e-01]],
[[-1.2968e+00, -1.1547e+00, -2.6859e+00, ..., -2.3065e+00,
-1.9807e+00, -1.8681e+00],
[-3.3846e-01, 4.1473e-01, -2.9173e+00, ..., -2.9164e+00,
-2.9386e+00, -2.9684e+00],
[-1.7927e+00, -8.3189e-01, -5.4106e+00, ..., -4.4887e+00,
-4.5727e+00, -3.8081e+00],
...,
[-1.7008e+00, -6.4556e-01, -4.0882e+00, ..., -2.8473e+00,
-2.7790e+00, -2.3553e+00],
[-1.6821e+00, -8.0386e-01, -3.3936e+00, ..., -2.4609e+00,
-2.3035e+00, -1.7083e+00],
[-2.0611e+00, -1.3565e+00, -2.8893e+00, ..., -2.8833e+00,
-2.4661e+00, -2.2256e+00]],
[[-1.2032e+00, -1.8004e+00, -1.5617e+00, ..., -2.6202e+00,
-2.2003e+00, -1.7091e+00],
[-1.2887e+00, -2.0202e+00, -1.5673e+00, ..., -2.4690e+00,
-1.9910e+00, -1.6311e+00],
[-1.7488e+00, -2.6117e+00, -2.5531e+00, ..., -2.9016e+00,
-2.3165e+00, -1.8019e+00],
...,
[-2.2748e+00, -2.8328e+00, -2.6657e+00, ..., -4.0657e+00,
-3.2950e+00, -2.5824e+00],
[-1.5338e+00, -1.9518e+00, -1.9191e+00, ..., -2.8059e+00,
-2.3453e+00, -1.9397e+00],
[-1.0353e+00, -1.2449e+00, -1.3294e+00, ..., -1.8582e+00,
-1.5416e+00, -1.1956e+00]]]], device='cuda:0'),
tensor([[[[ 1.5344e-01, 3.9177e-01, -1.1672e+00, ..., 4.5401e-01,
1.0110e+00, 6.5399e-01],
[-2.0965e+00, -1.4243e+00, -4.3517e+00, ..., -2.6093e+00,
-7.7867e-01, -7.1952e-02],
[ 4.3520e-01, -2.8617e-01, -1.0990e+00, ..., -2.8272e-01,
4.6711e-01, 6.2441e-01],
...,
[-2.7041e-01, -1.2826e+00, -2.6037e+00, ..., -1.9656e+00,
-1.3566e+00, -8.7692e-01],
[ 1.2728e+00, 6.9155e-01, -8.7564e-01, ..., -3.1030e-01,
5.1543e-01, 7.0419e-01],
[ 8.3662e-01, 3.4024e-01, -3.9488e-01, ..., -3.4904e-01,
2.8375e-01, 2.6760e-01]],
[[-3.2485e+00, -1.4347e+00, -6.1207e+00, ..., -4.8636e+00,
-4.4306e+00, -3.3730e+00],
[-5.6056e-01, 2.1754e+00, -4.1568e+00, ..., -3.3799e+00,
-3.5209e+00, -2.8941e+00],
[ 6.5899e-02, 1.8830e+00, -7.3803e+00, ..., -5.3608e+00,
-5.2304e+00, -4.9249e+00],
...,
[-1.2786e+00, 2.4380e-01, -6.5310e+00, ..., -4.5498e+00,
-4.1137e+00, -3.4578e+00],
[ 1.5024e+00, 2.5938e+00, -1.3356e+00, ..., -9.0470e-01,
-1.2237e+00, -1.2828e+00],
[ 9.8752e-01, 1.6263e+00, -2.3895e+00, ..., -2.2558e+00,
-2.4835e+00, -2.3669e+00]],
[[-1.3715e+00, -2.3473e+00, -2.1777e+00, ..., -5.9544e-01,
-1.9317e-01, -4.6816e-01],
[-2.0218e+00, -2.1957e+00, -1.5559e+00, ..., -3.4338e-01,
7.3397e-02, 4.1740e-02],
[-2.5152e+00, -1.9876e+00, -8.2550e-01, ..., -2.8107e-01,
2.5235e-01, 2.8998e-01],
...,
[-1.5719e+00, -1.8265e+00, -8.1094e-01, ..., -4.7541e-01,
-1.5885e-01, -3.4856e-01],
[-1.7412e+00, -1.1867e+00, -7.9892e-01, ..., -3.0871e-01,
1.4951e-01, 2.1798e-01],
[-1.7132e+00, -1.1045e+00, -5.9955e-01, ..., -5.0147e-01,
-2.7042e-01, -2.4584e-01]],
...,
[[-2.3861e+00, -1.3231e+00, -2.7561e+00, ..., -3.7133e+00,
-2.8557e+00, -1.8460e+00],
[-1.8794e+00, 2.0709e-01, -1.3382e+00, ..., -2.6987e+00,
-2.4469e+00, -1.8860e+00],
[-4.9082e-01, 2.5205e+00, 9.3727e-01, ..., -5.7673e-01,
-8.2219e-02, 1.5617e-01],
...,
[-1.7228e+00, 1.1530e+00, 1.9511e-01, ..., -1.5258e+00,
-6.8254e-01, -1.4690e-01],
[-8.8905e-01, 1.6378e+00, 1.5098e+00, ..., -1.4746e-01,
1.0911e-01, 3.2605e-01],
[-2.1083e+00, -1.0720e+00, -8.4231e-01, ..., -2.3322e+00,
-1.4403e+00, -9.0727e-01]],
[[-2.3600e+00, -2.6071e+00, -4.0144e+00, ..., -1.1182e+00,
-7.0534e-01, -7.4505e-01],
[-1.6340e+00, -1.3014e+00, -4.7642e+00, ..., -1.3288e+00,
-1.1911e+00, -1.5853e+00],
[-2.9851e+00, -2.4053e+00, -6.9215e+00, ..., -3.2586e+00,
-2.7783e+00, -2.2372e+00],
...,
[-2.5081e+00, -1.7571e+00, -4.9880e+00, ..., -1.4729e+00,
-1.0699e+00, -9.0052e-01],
[-2.2124e+00, -1.5774e+00, -3.9069e+00, ..., -1.7381e+00,
-1.1756e+00, -7.7778e-01],
[-2.2959e+00, -1.7860e+00, -3.0977e+00, ..., -2.8032e+00,
-2.0259e+00, -1.8401e+00]],
[[-1.8928e+00, -2.6226e+00, -2.3959e+00, ..., -1.8160e+00,
-1.5064e+00, -1.1505e+00],
[-1.9139e+00, -2.6317e+00, -2.0845e+00, ..., -1.4022e+00,
-1.1094e+00, -8.9379e-01],
[-2.4407e+00, -3.0021e+00, -2.4984e+00, ..., -1.6007e+00,
-1.1974e+00, -8.9205e-01],
...,
[-3.1000e+00, -3.8747e+00, -3.6521e+00, ..., -2.8710e+00,
-2.2908e+00, -1.7585e+00],
[-2.0600e+00, -2.6607e+00, -2.6230e+00, ..., -1.9808e+00,
-1.6273e+00, -1.3086e+00],
[-1.4834e+00, -1.7998e+00, -1.8459e+00, ..., -1.2314e+00,
-1.0358e+00, -7.9325e-01]]],
[[[ 2.6983e-01, 5.4738e-01, 6.6334e-01, ..., -3.3112e-01,
-9.2861e-02, -6.9998e-03],
[-2.1164e+00, -1.3940e+00, -3.0433e+00, ..., -3.1832e+00,
-1.2916e+00, -4.7007e-01],
[-7.1373e-02, -9.2637e-01, -2.6435e-02, ..., -1.4686e-01,
-3.5447e-01, 1.4709e-01],
...,
[-1.0003e+00, -2.0793e+00, -1.8264e+00, ..., -1.9138e+00,
-1.9048e+00, -1.2598e+00],
[ 6.9485e-01, -1.5367e-01, -5.7987e-02, ..., 3.4057e-02,
2.4769e-01, 4.4937e-01],
[ 1.2112e-01, -6.3014e-01, -4.0255e-01, ..., 2.3019e-01,
4.5607e-01, 4.9920e-01]],
[[-1.7379e+00, -1.2463e+00, -4.2750e+00, ..., -6.8669e+00,
-6.0826e+00, -4.5235e+00],
[ 6.8139e-01, 1.9519e+00, -2.2874e+00, ..., -5.4862e+00,
-5.5681e+00, -4.2945e+00],
[ 1.5546e+00, 1.7084e+00, -4.9939e+00, ..., -7.8857e+00,
-7.8317e+00, -6.5815e+00],
...,
[ 2.8872e-01, 1.8227e-01, -4.3167e+00, ..., -7.1428e+00,
-6.6842e+00, -5.2053e+00],
[ 2.2936e+00, 2.1721e+00, -3.0343e-01, ..., -1.9368e+00,
-2.5505e+00, -2.0543e+00],
[ 1.3352e+00, 1.1227e+00, -1.8213e+00, ..., -2.9449e+00,
-3.4770e+00, -3.0512e+00]],
[[-2.2112e-01, -1.3814e+00, -1.5875e+00, ..., -1.4215e+00,
-1.0589e+00, -1.3377e+00],
[-7.1191e-01, -1.0546e+00, -1.0069e+00, ..., -9.8431e-01,
-6.1911e-01, -9.0367e-01],
[-5.3078e-01, -3.5914e-01, -4.5970e-01, ..., -1.1067e+00,
-5.9324e-01, -4.6784e-01],
...,
[ 4.8296e-02, -4.8702e-01, -5.6367e-01, ..., -1.0389e+00,
-8.8536e-01, -1.2485e+00],
[-6.0318e-01, -1.5716e-01, -5.5429e-01, ..., -5.5592e-01,
-2.5114e-01, -4.1901e-01],
[-7.8627e-01, -2.3949e-01, -4.9560e-01, ..., -4.9753e-01,
-4.5080e-01, -6.1173e-01]],
...,
[[-3.1058e+00, -2.5349e+00, -2.6634e+00, ..., -3.0416e+00,
-2.4659e+00, -1.7952e+00],
[-2.1792e+00, -6.4751e-01, -1.1335e+00, ..., -2.4181e+00,
-2.4790e+00, -2.0000e+00],
[-1.0983e+00, 1.1422e+00, 8.9493e-01, ..., -1.1390e-01,
2.2179e-02, 2.7326e-01],
...,
[-2.2361e+00, -4.4831e-01, -2.9447e-01, ..., -9.2627e-01,
-2.4120e-01, 2.9310e-01],
[-1.2261e+00, 6.3437e-01, 8.2942e-01, ..., 4.2249e-01,
4.8548e-01, 8.1285e-01],
[-2.0769e+00, -1.4728e+00, -1.6600e+00, ..., -2.2117e+00,
-1.1825e+00, -3.5872e-01]],
[[-1.6261e+00, -1.5770e+00, -2.9444e+00, ..., -2.2598e+00,
-1.9279e+00, -1.8084e+00],
[-5.6596e-01, 1.7204e-01, -3.3076e+00, ..., -2.8903e+00,
-2.8696e+00, -2.8771e+00],
[-2.0465e+00, -7.8421e-01, -5.4337e+00, ..., -4.4864e+00,
-4.5237e+00, -3.6982e+00],
...,
[-1.3429e+00, -3.0830e-02, -3.3881e+00, ..., -2.8385e+00,
-2.7478e+00, -2.3073e+00],
[-1.6428e+00, -6.1382e-01, -3.0139e+00, ..., -2.4500e+00,
-2.2616e+00, -1.6563e+00],
[-2.2550e+00, -1.4250e+00, -3.0077e+00, ..., -2.8800e+00,
-2.4333e+00, -2.1976e+00]],
[[-1.3619e+00, -1.9249e+00, -1.6803e+00, ..., -2.5692e+00,
-2.1509e+00, -1.6654e+00],
[-1.2836e+00, -1.7744e+00, -1.1861e+00, ..., -2.4032e+00,
-1.9424e+00, -1.5852e+00],
[-1.6092e+00, -1.9463e+00, -1.4166e+00, ..., -2.8188e+00,
-2.2583e+00, -1.7462e+00],
...,
[-2.4025e+00, -2.9877e+00, -2.7677e+00, ..., -4.0203e+00,
-3.2459e+00, -2.5430e+00],
[-1.5667e+00, -2.0551e+00, -1.9913e+00, ..., -2.7809e+00,
-2.3170e+00, -1.9140e+00],
[-1.0273e+00, -1.3425e+00, -1.3708e+00, ..., -1.8320e+00,
-1.5192e+00, -1.1758e+00]]],
[[[ 3.6094e-01, 7.6977e-01, 3.5108e-01, ..., 5.1541e-01,
1.0284e+00, 6.7370e-01],
[-1.3243e+00, -4.9695e-01, -2.0614e+00, ..., -2.3540e+00,
-6.0251e-01, 4.7397e-02],
[ 7.9167e-01, 5.5881e-01, 2.2799e-01, ..., -1.3223e-01,
5.7122e-01, 6.9345e-01],
...,
[-8.8492e-03, -7.7583e-01, -1.6835e+00, ..., -1.8791e+00,
-1.3061e+00, -8.7223e-01],
[ 1.4674e+00, 1.1688e+00, 4.5568e-01, ..., -2.0775e-01,
5.5694e-01, 7.2924e-01],
[ 1.1010e+00, 9.1890e-01, 5.2957e-01, ..., -2.0245e-01,
3.7325e-01, 3.4917e-01]],
[[-2.7936e+00, -1.6334e+00, -4.9485e+00, ..., -4.9119e+00,
-4.4335e+00, -3.3943e+00],
[-6.8627e-01, 1.3188e+00, -3.2680e+00, ..., -3.4885e+00,
-3.5727e+00, -2.9436e+00],
[-1.7973e-02, 1.0325e+00, -5.9701e+00, ..., -5.4543e+00,
-5.2814e+00, -4.9768e+00],
...,
[-8.0153e-01, 3.5873e-01, -4.8434e+00, ..., -4.6507e+00,
-4.1728e+00, -3.4954e+00],
[ 2.1107e+00, 2.8610e+00, 2.0664e-01, ..., -8.9589e-01,
-1.1904e+00, -1.2573e+00],
[ 1.2820e+00, 1.6175e+00, -1.6149e+00, ..., -2.2558e+00,
-2.4685e+00, -2.3851e+00]],
[[-6.5164e-01, -1.7644e+00, -1.6473e+00, ..., -6.2251e-01,
-2.3987e-01, -4.9008e-01],
[-1.3141e+00, -1.7681e+00, -1.1718e+00, ..., -3.9319e-01,
8.3691e-03, 1.0484e-02],
[-1.3247e+00, -1.3439e+00, -7.7846e-01, ..., -2.7777e-01,
2.3065e-01, 2.6520e-01],
...,
[-1.0576e+00, -1.6613e+00, -7.8739e-01, ..., -5.0077e-01,
-1.9271e-01, -3.6706e-01],
[-1.3976e+00, -1.0431e+00, -6.9462e-01, ..., -2.9349e-01,
1.4171e-01, 2.0639e-01],
[-1.4615e+00, -8.5544e-01, -3.7461e-01, ..., -4.5459e-01,
-2.5498e-01, -2.2520e-01]],
...,
[[-2.0806e+00, -1.0960e+00, -2.4044e+00, ..., -3.6275e+00,
-2.8236e+00, -1.8329e+00],
[-6.5567e-01, 1.0102e+00, -8.5944e-01, ..., -2.5248e+00,
-2.3255e+00, -1.8173e+00],
[-1.7047e-01, 2.3763e+00, 7.1614e-01, ..., -5.6486e-01,
-6.2179e-02, 1.6802e-01],
...,
[-1.5505e+00, 1.5367e+00, 1.1312e-01, ..., -1.4799e+00,
-6.4776e-01, -9.1143e-02],
[-5.7072e-01, 1.7403e+00, 1.0285e+00, ..., -1.2119e-01,
1.2513e-01, 3.3213e-01],
[-1.7760e+00, -1.1634e+00, -1.6069e+00, ..., -2.3313e+00,
-1.4064e+00, -8.7688e-01]],
[[-1.4978e+00, -1.5778e+00, -3.1500e+00, ..., -1.0874e+00,
-6.7762e-01, -6.7378e-01],
[-8.5265e-01, -5.3718e-01, -3.9949e+00, ..., -1.3765e+00,
-1.2165e+00, -1.5732e+00],
[-2.2289e+00, -1.7524e+00, -6.2847e+00, ..., -3.3325e+00,
-2.8384e+00, -2.2659e+00],
...,
[-1.7673e+00, -7.6774e-01, -4.8734e+00, ..., -1.5666e+00,
-1.1165e+00, -9.1379e-01],
[-1.7012e+00, -8.6068e-01, -3.8907e+00, ..., -1.7865e+00,
-1.1761e+00, -7.5630e-01],
[-2.1123e+00, -1.3945e+00, -3.1117e+00, ..., -2.7884e+00,
-1.9853e+00, -1.7911e+00]],
[[-1.2798e+00, -1.9172e+00, -1.8021e+00, ..., -1.7821e+00,
-1.4863e+00, -1.1286e+00],
[-1.2984e+00, -1.8660e+00, -1.5204e+00, ..., -1.3955e+00,
-1.1123e+00, -8.8487e-01],
[-1.6837e+00, -2.0977e+00, -1.7961e+00, ..., -1.5936e+00,
-1.2024e+00, -8.8482e-01],
...,
[-2.5322e+00, -3.2083e+00, -3.0333e+00, ..., -2.8564e+00,
-2.2831e+00, -1.7511e+00],
[-1.6914e+00, -2.1853e+00, -2.1452e+00, ..., -1.9741e+00,
-1.6238e+00, -1.3079e+00],
[-1.1567e+00, -1.4199e+00, -1.5022e+00, ..., -1.2214e+00,
-1.0359e+00, -7.8884e-01]]],
[[[ 4.7115e-01, 9.6311e-01, 8.5056e-01, ..., -3.5865e-01,
-7.4433e-02, 2.5459e-02],
[-1.2477e+00, -3.2493e-01, -1.5143e+00, ..., -3.2595e+00,
-1.3016e+00, -4.3112e-01],
[ 5.7445e-01, 4.3048e-01, 3.3806e-01, ..., -1.8697e-01,
-3.5571e-01, 1.7756e-01],
...,
[-1.8497e-01, -8.4810e-01, -1.5406e+00, ..., -1.9106e+00,
-1.8604e+00, -1.2454e+00],
[ 1.1084e+00, 8.2474e-01, 3.0119e-01, ..., 6.5633e-02,
3.0540e-01, 4.6356e-01],
[ 7.3947e-01, 5.4494e-01, 2.0004e-01, ..., 2.4934e-01,
4.7525e-01, 4.8999e-01]],
[[-2.2956e+00, -1.0940e+00, -4.0945e+00, ..., -6.8527e+00,
-6.0689e+00, -4.5136e+00],
[-2.9041e-01, 1.7141e+00, -2.5071e+00, ..., -5.4730e+00,
-5.5802e+00, -4.3067e+00],
[ 4.1600e-01, 1.5145e+00, -5.0515e+00, ..., -7.8830e+00,
-7.8386e+00, -6.6290e+00],
...,
[-4.4917e-01, 6.8252e-01, -4.0147e+00, ..., -7.1395e+00,
-6.7131e+00, -5.2020e+00],
[ 2.1526e+00, 2.8612e+00, 4.5797e-01, ..., -2.0368e+00,
-2.6574e+00, -2.1046e+00],
[ 1.2614e+00, 1.5725e+00, -1.4429e+00, ..., -2.9904e+00,
-3.5197e+00, -3.0485e+00]],
[[-5.6652e-01, -1.5700e+00, -1.3940e+00, ..., -1.3952e+00,
-1.0363e+00, -1.3463e+00],
[-1.2746e+00, -1.6385e+00, -1.0141e+00, ..., -1.0051e+00,
-6.0778e-01, -9.0245e-01],
[-1.1201e+00, -9.8939e-01, -3.9850e-01, ..., -1.1276e+00,
-6.0515e-01, -4.7350e-01],
...,
[-9.3514e-01, -1.4026e+00, -5.4826e-01, ..., -1.0246e+00,
-8.6483e-01, -1.2296e+00],
[-1.2989e+00, -9.0019e-01, -5.7152e-01, ..., -5.4767e-01,
-2.4801e-01, -4.2255e-01],
[-1.3953e+00, -7.8630e-01, -3.5986e-01, ..., -4.9313e-01,
-4.3235e-01, -5.9006e-01]],
...,
[[-2.1705e+00, -1.2337e+00, -2.5398e+00, ..., -3.0807e+00,
-2.5416e+00, -1.8152e+00],
[-6.8887e-01, 9.6718e-01, -9.3119e-01, ..., -2.4243e+00,
-2.5590e+00, -2.0415e+00],
[-2.3059e-01, 2.2430e+00, 5.3862e-01, ..., -4.1267e-02,
5.7007e-03, 2.5793e-01],
...,
[-1.5289e+00, 1.2929e+00, -2.0420e-01, ..., -8.7123e-01,
-2.1228e-01, 3.1993e-01],
[-5.3127e-01, 1.6399e+00, 8.5415e-01, ..., 4.0351e-01,
4.8992e-01, 8.3787e-01],
[-1.6410e+00, -1.0660e+00, -1.5652e+00, ..., -2.1866e+00,
-1.1396e+00, -3.2953e-01]],
[[-1.2894e+00, -1.1987e+00, -2.5885e+00, ..., -2.2256e+00,
-1.9208e+00, -1.8358e+00],
[-6.0973e-01, -1.0354e-01, -3.3279e+00, ..., -2.8470e+00,
-2.8783e+00, -2.9139e+00],
[-2.0734e+00, -1.3848e+00, -5.7168e+00, ..., -4.4268e+00,
-4.5090e+00, -3.7200e+00],
...,
[-1.6219e+00, -4.5313e-01, -4.1232e+00, ..., -2.8915e+00,
-2.7907e+00, -2.3471e+00],
[-1.6372e+00, -6.8062e-01, -3.4527e+00, ..., -2.4704e+00,
-2.2759e+00, -1.6979e+00],
[-2.0973e+00, -1.3380e+00, -2.9687e+00, ..., -2.8561e+00,
-2.4171e+00, -2.2030e+00]],
[[-1.1470e+00, -1.7058e+00, -1.5205e+00, ..., -2.5743e+00,
-2.1450e+00, -1.6685e+00],
[-1.1539e+00, -1.6188e+00, -1.2000e+00, ..., -2.4153e+00,
-1.9308e+00, -1.5850e+00],
[-1.5007e+00, -1.7987e+00, -1.4014e+00, ..., -2.8369e+00,
-2.2541e+00, -1.7504e+00],
...,
[-2.3240e+00, -2.9048e+00, -2.6785e+00, ..., -4.0060e+00,
-3.2577e+00, -2.5471e+00],
[-1.5565e+00, -1.9818e+00, -1.9126e+00, ..., -2.7859e+00,
-2.3429e+00, -1.9316e+00],
[-1.0430e+00, -1.2500e+00, -1.3034e+00, ..., -1.8520e+00,
-1.5459e+00, -1.1989e+00]]],
[[[ 3.8772e-01, 8.2103e-01, 5.0185e-01, ..., 4.3564e-01,
1.0096e+00, 6.4902e-01],
[-1.2473e+00, -3.8458e-01, -1.8035e+00, ..., -2.5842e+00,
-7.2913e-01, -4.7903e-02],
[ 8.0403e-01, 6.2294e-01, 3.9037e-01, ..., -2.6656e-01,
5.2145e-01, 6.4337e-01],
...,
[-1.1647e+00, -2.7509e+00, -3.3332e+00, ..., -2.0141e+00,
-1.3926e+00, -9.2725e-01],
[ 1.4576e+00, 1.2294e+00, 5.8877e-01, ..., -3.1952e-01,
5.2067e-01, 7.0535e-01],
[ 1.0912e+00, 9.6178e-01, 6.1685e-01, ..., -3.4812e-01,
3.0132e-01, 2.7859e-01]],
[[-2.7245e+00, -1.5935e+00, -4.7751e+00, ..., -4.8226e+00,
-4.4344e+00, -3.3455e+00],
[-6.4379e-01, 1.3257e+00, -3.0856e+00, ..., -3.3835e+00,
-3.5674e+00, -2.8955e+00],
[ 5.4941e-02, 1.0881e+00, -5.7345e+00, ..., -5.3946e+00,
-5.2911e+00, -4.9085e+00],
...,
[-1.0316e+00, 5.3537e-01, -4.6271e+00, ..., -4.5773e+00,
-4.1332e+00, -3.4407e+00],
[ 2.0429e+00, 2.7362e+00, 2.7322e-01, ..., -8.5349e-01,
-1.1860e+00, -1.2316e+00],
[ 1.2350e+00, 1.5358e+00, -1.5395e+00, ..., -2.2302e+00,
-2.4690e+00, -2.3562e+00]],
[[-6.3250e-01, -1.7443e+00, -1.6236e+00, ..., -5.9330e-01,
-1.9938e-01, -4.6411e-01],
[-1.3233e+00, -1.7783e+00, -1.1631e+00, ..., -3.3519e-01,
8.5552e-02, 3.8275e-02],
[-1.2519e+00, -1.2716e+00, -7.0826e-01, ..., -2.7144e-01,
2.5379e-01, 2.8033e-01],
...,
[-2.2822e+00, -1.9742e+00, -3.8326e-01, ..., -4.9151e-01,
-1.5842e-01, -3.4927e-01],
[-1.3268e+00, -9.8381e-01, -6.4402e-01, ..., -3.1784e-01,
1.5510e-01, 2.2255e-01],
[-1.4151e+00, -8.2547e-01, -3.4313e-01, ..., -5.0251e-01,
-2.7291e-01, -2.4206e-01]],
...,
[[-2.0907e+00, -1.0848e+00, -2.3869e+00, ..., -3.6291e+00,
-2.8268e+00, -1.8129e+00],
[-5.6153e-01, 1.1313e+00, -7.5808e-01, ..., -2.6229e+00,
-2.4396e+00, -1.8556e+00],
[-1.8250e-01, 2.3739e+00, 6.7695e-01, ..., -5.1898e-01,
-7.3658e-02, 1.5467e-01],
...,
[-1.5357e-01, 2.6355e+00, 7.0864e-01, ..., -1.5240e+00,
-6.8302e-01, -1.2452e-01],
[-5.9670e-01, 1.6303e+00, 8.9716e-01, ..., -1.4180e-01,
1.1065e-01, 3.3546e-01],
[-1.7624e+00, -1.1882e+00, -1.6616e+00, ..., -2.3641e+00,
-1.4666e+00, -9.2342e-01]],
[[-1.4018e+00, -1.4236e+00, -2.9874e+00, ..., -1.1119e+00,
-7.1499e-01, -7.6037e-01],
[-7.4467e-01, -3.9039e-01, -3.8626e+00, ..., -1.3662e+00,
-1.2216e+00, -1.5987e+00],
[-2.1653e+00, -1.6564e+00, -6.2342e+00, ..., -3.2805e+00,
-2.7863e+00, -2.2300e+00],
...,
[-2.4025e+00, -1.6488e+00, -4.6055e+00, ..., -1.4757e+00,
-1.0436e+00, -8.8387e-01],
[-1.7006e+00, -8.6344e-01, -3.8255e+00, ..., -1.7291e+00,
-1.1478e+00, -7.5238e-01],
[-2.1048e+00, -1.3984e+00, -3.0865e+00, ..., -2.8099e+00,
-2.0120e+00, -1.8252e+00]],
[[-1.2138e+00, -1.8360e+00, -1.7075e+00, ..., -1.8065e+00,
-1.4973e+00, -1.1444e+00],
[-1.2290e+00, -1.7692e+00, -1.4134e+00, ..., -1.4090e+00,
-1.1136e+00, -8.9919e-01],
[-1.5939e+00, -1.9760e+00, -1.6525e+00, ..., -1.6139e+00,
-1.2085e+00, -9.0776e-01],
...,
[-2.6758e+00, -3.3421e+00, -3.3210e+00, ..., -2.8727e+00,
-2.2962e+00, -1.7665e+00],
[-1.6215e+00, -2.0986e+00, -2.0648e+00, ..., -1.9802e+00,
-1.6294e+00, -1.3160e+00],
[-1.1027e+00, -1.3549e+00, -1.4403e+00, ..., -1.2269e+00,
-1.0368e+00, -7.9681e-01]]]], device='cuda:0'),
tensor([[[[ 3.4505e-01, 7.9981e-01, 7.3494e-01, ..., 6.8220e-01,
9.4382e-01, 7.1135e-01],
[-1.1865e+00, -2.6610e-01, -1.3100e+00, ..., -1.2913e+00,
-6.4402e-02, 4.8476e-01],
[ 6.0548e-01, 4.5152e-01, 5.4075e-01, ..., 5.6655e-01,
9.2696e-01, 8.3712e-01],
...,
[-1.7339e-01, -1.1482e+00, -2.6616e+00, ..., -1.4959e+00,
-1.4190e+00, -8.5784e-01],
[ 1.4365e+00, 8.3454e-01, -9.7123e-01, ..., -2.3586e-01,
1.0608e-01, 3.8562e-01],
[ 8.5967e-01, 2.9742e-01, -6.9070e-01, ..., -1.7017e-01,
2.0733e-01, 3.7417e-01]],
[[-2.3153e+00, -1.3212e+00, -4.2751e+00, ..., -5.4428e+00,
-4.6534e+00, -3.6078e+00],
[-2.1672e-01, 1.4876e+00, -2.6517e+00, ..., -4.0799e+00,
-3.9715e+00, -3.2454e+00],
[ 5.7945e-01, 1.2934e+00, -5.1200e+00, ..., -5.9899e+00,
-5.7805e+00, -5.3714e+00],
...,
[-1.2695e+00, 1.6630e-01, -6.6983e+00, ..., -7.6021e+00,
-6.8965e+00, -5.4109e+00],
[ 1.8163e+00, 2.8469e+00, -1.0179e+00, ..., -2.6270e+00,
-2.9766e+00, -2.3856e+00],
[ 1.2204e+00, 1.8282e+00, -2.2801e+00, ..., -3.2716e+00,
-3.6190e+00, -3.1276e+00]],
[[-6.4272e-01, -1.7367e+00, -1.6308e+00, ..., -1.0150e+00,
-7.0504e-01, -8.3074e-01],
[-1.2812e+00, -1.7363e+00, -1.2160e+00, ..., -8.0633e-01,
-4.2645e-01, -2.9436e-01],
[-9.6451e-01, -9.1340e-01, -4.4605e-01, ..., -3.3978e-01,
5.8695e-02, 7.3428e-02],
...,
[-1.3865e+00, -1.7728e+00, -7.7871e-01, ..., -8.8360e-01,
-7.6239e-01, -1.0027e+00],
[-1.7299e+00, -1.1671e+00, -8.3845e-01, ..., -5.6397e-01,
-3.0005e-01, -3.6333e-01],
[-1.8717e+00, -1.2768e+00, -8.9458e-01, ..., -9.4605e-01,
-8.1555e-01, -7.9298e-01]],
...,
[[-2.1408e+00, -1.2686e+00, -2.2551e+00, ..., -3.1429e+00,
-2.4689e+00, -1.6212e+00],
[-6.0828e-01, 9.9013e-01, -4.1766e-01, ..., -1.4216e+00,
-1.4883e+00, -1.2997e+00],
[-4.2283e-01, 1.8953e+00, 7.6252e-01, ..., -1.6848e-01,
2.8242e-01, 4.0641e-01],
...,
[-1.8299e+00, 1.2729e+00, 1.5748e-02, ..., -1.1206e+00,
-5.5100e-01, -1.6590e-02],
[-7.8924e-01, 1.9411e+00, 1.6506e+00, ..., 4.3549e-01,
4.0778e-01, 6.8187e-01],
[-2.2239e+00, -1.1531e+00, -9.8666e-01, ..., -1.8844e+00,
-9.9687e-01, -3.0125e-01]],
[[-1.3656e+00, -1.2796e+00, -2.6459e+00, ..., -9.8748e-01,
-4.5319e-01, -3.5145e-01],
[-7.0380e-01, -2.2197e-01, -3.4553e+00, ..., -1.6511e+00,
-1.4035e+00, -1.6522e+00],
[-2.0922e+00, -1.4180e+00, -5.7717e+00, ..., -3.7516e+00,
-3.2231e+00, -2.5680e+00],
...,
[-2.4399e+00, -1.5410e+00, -5.4514e+00, ..., -3.0915e+00,
-2.7322e+00, -2.1725e+00],
[-2.1976e+00, -1.4393e+00, -4.2524e+00, ..., -2.6837e+00,
-2.3139e+00, -1.6382e+00],
[-2.3829e+00, -1.7625e+00, -3.3185e+00, ..., -2.9597e+00,
-2.4237e+00, -2.1142e+00]],
[[-1.1393e+00, -1.6883e+00, -1.4963e+00, ..., -1.6991e+00,
-1.4126e+00, -1.0586e+00],
[-1.1542e+00, -1.6195e+00, -1.1875e+00, ..., -1.3699e+00,
-1.1143e+00, -8.7958e-01],
[-1.5008e+00, -1.7925e+00, -1.3741e+00, ..., -1.5305e+00,
-1.1648e+00, -8.4137e-01],
...,
[-3.1834e+00, -3.9765e+00, -3.7566e+00, ..., -3.9393e+00,
-3.1633e+00, -2.4402e+00],
[-2.1173e+00, -2.7305e+00, -2.6934e+00, ..., -2.7684e+00,
-2.2878e+00, -1.8619e+00],
[-1.5364e+00, -1.8617e+00, -1.9247e+00, ..., -1.8724e+00,
-1.5208e+00, -1.1656e+00]]],
[[[ 3.1236e-01, 5.8230e-01, 7.0861e-01, ..., 5.0207e-01,
1.0230e+00, 6.3106e-01],
[-2.1236e+00, -1.4659e+00, -3.1250e+00, ..., -2.5736e+00,
-8.1235e-01, -1.3079e-01],
[-1.0426e-01, -9.4055e-01, -1.6670e-01, ..., -1.0195e-01,
5.9554e-01, 7.0872e-01],
...,
[ 1.2993e-01, -5.5775e-01, -1.3939e+00, ..., -1.2805e+00,
-9.7505e-01, -7.5829e-01],
[ 1.4481e+00, 1.2084e+00, 6.0720e-01, ..., 6.9721e-01,
8.9496e-01, 9.3115e-01],
[ 1.0915e+00, 9.7715e-01, 6.9959e-01, ..., 8.9891e-01,
1.0229e+00, 8.7438e-01]],
[[-1.7032e+00, -1.1341e+00, -4.1724e+00, ..., -4.8777e+00,
-4.4520e+00, -3.3980e+00],
[ 5.8816e-01, 1.9725e+00, -2.2341e+00, ..., -3.5510e+00,
-3.6548e+00, -2.9927e+00],
[ 1.3455e+00, 1.7279e+00, -4.9057e+00, ..., -5.5076e+00,
-5.3322e+00, -4.9435e+00],
...,
[-9.1498e-01, 1.2631e-01, -5.0198e+00, ..., -5.8594e+00,
-5.2474e+00, -4.2181e+00],
[ 1.7012e+00, 2.3520e+00, -3.8855e-01, ..., -1.2136e+00,
-1.4819e+00, -1.4997e+00],
[ 9.7526e-01, 1.2506e+00, -1.8841e+00, ..., -2.6191e+00,
-2.8304e+00, -2.7402e+00]],
[[-1.4383e-01, -1.1855e+00, -1.3446e+00, ..., -4.9710e-01,
-1.7087e-01, -4.8498e-01],
[-6.5175e-01, -9.1306e-01, -8.0869e-01, ..., -2.8079e-01,
2.9268e-02, -8.2022e-02],
[-5.9857e-01, -3.8984e-01, -4.1617e-01, ..., -3.4861e-01,
8.6940e-02, 8.0066e-02],
...,
[-9.4649e-01, -1.4006e+00, -6.2554e-01, ..., -6.7805e-01,
-5.0553e-01, -6.7943e-01],
[-1.1510e+00, -7.6403e-01, -4.2914e-01, ..., -1.1676e-01,
1.0586e-01, 8.3854e-02],
[-1.1739e+00, -5.7786e-01, -1.1801e-01, ..., -6.2733e-02,
-2.9633e-02, -6.0378e-02]],
...,
[[-3.0393e+00, -2.4539e+00, -2.7932e+00, ..., -3.4576e+00,
-2.6361e+00, -1.7527e+00],
[-2.2274e+00, -7.2610e-01, -1.5497e+00, ..., -2.6067e+00,
-2.3243e+00, -1.8034e+00],
[-1.0339e+00, 1.2832e+00, 6.4977e-01, ..., -4.6631e-01,
-4.2549e-02, 1.2663e-01],
...,
[-1.4756e+00, 1.4008e+00, 2.6452e-01, ..., -8.6274e-01,
-1.2277e-01, 4.4391e-01],
[-6.2786e-01, 1.5378e+00, 9.8531e-01, ..., 1.8727e-01,
3.9784e-01, 5.7401e-01],
[-1.6806e+00, -1.0329e+00, -1.3820e+00, ..., -2.3155e+00,
-1.2720e+00, -5.7394e-01]],
[[-1.5888e+00, -1.5623e+00, -2.9171e+00, ..., -1.2663e+00,
-8.3226e-01, -8.2959e-01],
[-5.8917e-01, 1.0293e-01, -3.2454e+00, ..., -1.3905e+00,
-1.2444e+00, -1.5841e+00],
[-2.0973e+00, -9.1059e-01, -5.3815e+00, ..., -3.2065e+00,
-2.7415e+00, -2.2222e+00],
...,
[-1.8839e+00, -1.0086e+00, -4.5619e+00, ..., -2.3548e+00,
-1.9563e+00, -1.5358e+00],
[-1.7386e+00, -9.9972e-01, -3.6801e+00, ..., -2.2184e+00,
-1.6140e+00, -1.0251e+00],
[-2.0443e+00, -1.4245e+00, -2.9924e+00, ..., -2.8116e+00,
-1.9837e+00, -1.7171e+00]],
[[-1.3559e+00, -1.9148e+00, -1.7119e+00, ..., -1.7592e+00,
-1.4509e+00, -1.1090e+00],
[-1.2819e+00, -1.7662e+00, -1.2365e+00, ..., -1.3726e+00,
-1.0813e+00, -8.6336e-01],
[-1.6087e+00, -1.9502e+00, -1.4820e+00, ..., -1.5942e+00,
-1.1951e+00, -8.8783e-01],
...,
[-2.3832e+00, -2.9904e+00, -2.8041e+00, ..., -3.1266e+00,
-2.4995e+00, -1.9103e+00],
[-1.6064e+00, -2.0490e+00, -1.9977e+00, ..., -2.1665e+00,
-1.7840e+00, -1.4303e+00],
[-1.0795e+00, -1.3033e+00, -1.3610e+00, ..., -1.3162e+00,
-1.1279e+00, -8.3648e-01]]],
[[[ 2.7076e-01, 5.6007e-01, 6.9589e-01, ..., 4.7324e-01,
9.9440e-01, 5.9529e-01],
[-2.2028e+00, -1.5116e+00, -3.2252e+00, ..., -2.6984e+00,
-9.2073e-01, -2.0705e-01],
[-1.5136e-01, -9.7900e-01, -2.2387e-01, ..., -1.7060e-01,
5.1157e-01, 6.4244e-01],
...,
[-1.8723e-01, -1.1128e+00, -2.5991e+00, ..., -1.5194e+00,
-1.4207e+00, -8.9042e-01],
[ 1.4116e+00, 8.7650e-01, -8.7255e-01, ..., -2.4598e-01,
1.2120e-01, 3.7482e-01],
[ 8.4126e-01, 3.2111e-01, -6.2581e-01, ..., -1.5622e-01,
2.3544e-01, 3.7806e-01]],
[[-1.6707e+00, -1.1543e+00, -4.1848e+00, ..., -4.8997e+00,
-4.4545e+00, -3.3923e+00],
[ 6.1804e-01, 1.9544e+00, -2.2449e+00, ..., -3.5103e+00,
-3.6051e+00, -2.9652e+00],
[ 1.3654e+00, 1.6910e+00, -4.9235e+00, ..., -5.4789e+00,
-5.3021e+00, -4.9450e+00],
...,
[-1.2968e+00, 1.2850e-01, -6.7139e+00, ..., -7.5815e+00,
-6.9023e+00, -5.3964e+00],
[ 1.7758e+00, 2.7882e+00, -1.0722e+00, ..., -2.5788e+00,
-2.9663e+00, -2.3630e+00],
[ 1.1976e+00, 1.7893e+00, -2.2883e+00, ..., -3.2562e+00,
-3.6221e+00, -3.1146e+00]],
[[-1.2111e-01, -1.1696e+00, -1.3393e+00, ..., -4.9203e-01,
-1.5719e-01, -4.8614e-01],
[-6.0595e-01, -8.6256e-01, -7.7941e-01, ..., -2.7347e-01,
3.7330e-02, -7.9658e-02],
[-5.5109e-01, -3.4025e-01, -3.8841e-01, ..., -3.4975e-01,
8.1366e-02, 7.4555e-02],
...,
[-1.3391e+00, -1.7335e+00, -7.8241e-01, ..., -8.8706e-01,
-7.6819e-01, -1.0164e+00],
[-1.6937e+00, -1.1438e+00, -8.3898e-01, ..., -5.5978e-01,
-2.8793e-01, -3.5587e-01],
[-1.8545e+00, -1.2649e+00, -8.8378e-01, ..., -9.3870e-01,
-8.0922e-01, -7.8065e-01]],
...,
[[-3.0688e+00, -2.5408e+00, -2.7856e+00, ..., -3.5051e+00,
-2.6742e+00, -1.7809e+00],
[-2.3054e+00, -8.5249e-01, -1.5831e+00, ..., -2.6609e+00,
-2.3813e+00, -1.8647e+00],
[-1.0984e+00, 1.1678e+00, 6.3504e-01, ..., -4.2411e-01,
3.3629e-03, 1.4207e-01],
...,
[-1.8307e+00, 1.1939e+00, -6.2709e-02, ..., -1.0700e+00,
-5.1158e-01, 2.1651e-02],
[-8.0439e-01, 1.8700e+00, 1.5599e+00, ..., 4.7032e-01,
4.1471e-01, 6.7662e-01],
[-2.2304e+00, -1.1753e+00, -1.0405e+00, ..., -1.8813e+00,
-9.7410e-01, -3.1505e-01]],
[[-1.6203e+00, -1.5821e+00, -2.8929e+00, ..., -1.2484e+00,
-8.0926e-01, -8.2792e-01],
[-6.0882e-01, 9.7942e-02, -3.2119e+00, ..., -1.3493e+00,
-1.1953e+00, -1.5676e+00],
[-2.1169e+00, -9.0920e-01, -5.3270e+00, ..., -3.1745e+00,
-2.7047e+00, -2.2033e+00],
...,
[-2.4628e+00, -1.5894e+00, -5.4798e+00, ..., -3.0961e+00,
-2.7137e+00, -2.1629e+00],
[-2.2071e+00, -1.4864e+00, -4.2715e+00, ..., -2.6938e+00,
-2.2950e+00, -1.6199e+00],
[-2.3777e+00, -1.7858e+00, -3.3254e+00, ..., -2.9786e+00,
-2.4209e+00, -2.1010e+00]],
[[-1.3654e+00, -1.9290e+00, -1.7300e+00, ..., -1.7857e+00,
-1.4703e+00, -1.1156e+00],
[-1.2901e+00, -1.7750e+00, -1.2489e+00, ..., -1.3761e+00,
-1.0802e+00, -8.6285e-01],
[-1.6071e+00, -1.9565e+00, -1.4937e+00, ..., -1.5973e+00,
-1.1860e+00, -8.7789e-01],
...,
[-3.1601e+00, -3.9455e+00, -3.7493e+00, ..., -3.9371e+00,
-3.1580e+00, -2.4353e+00],
[-2.1011e+00, -2.7109e+00, -2.6839e+00, ..., -2.7639e+00,
-2.2839e+00, -1.8577e+00],
[-1.5282e+00, -1.8544e+00, -1.9133e+00, ..., -1.8663e+00,
-1.5200e+00, -1.1613e+00]]],
[[[ 1.3789e-01, 3.8796e-01, -1.1317e+00, ..., 6.5813e-01,
9.4187e-01, 6.8608e-01],
[-2.0254e+00, -1.3499e+00, -4.2384e+00, ..., -1.5300e+00,
-2.0096e-01, 3.7984e-01],
[ 4.3048e-01, -3.0775e-01, -1.1081e+00, ..., 4.9730e-01,
9.1239e-01, 8.2749e-01],
...,
[-2.4487e-01, -1.2901e+00, -2.6033e+00, ..., -1.2161e+00,
-8.2551e-01, -6.5094e-01],
[ 1.2620e+00, 6.3203e-01, -9.2170e-01, ..., 4.8058e-01,
7.5868e-01, 7.8582e-01],
[ 8.3643e-01, 3.0622e-01, -4.2558e-01, ..., 5.1043e-01,
7.8306e-01, 6.7321e-01]],
[[-3.1918e+00, -1.4313e+00, -6.1298e+00, ..., -5.4698e+00,
-4.6973e+00, -3.6476e+00],
[-5.7377e-01, 2.0865e+00, -4.2264e+00, ..., -4.1925e+00,
-4.0843e+00, -3.3382e+00],
[-2.7447e-03, 1.7760e+00, -7.4251e+00, ..., -6.1357e+00,
-5.9016e+00, -5.4668e+00],
...,
[-1.2883e+00, 2.1133e-01, -6.5478e+00, ..., -5.3679e+00,
-4.7277e+00, -3.9230e+00],
[ 1.4841e+00, 2.5711e+00, -1.3646e+00, ..., -1.2765e+00,
-1.4269e+00, -1.4888e+00],
[ 9.7199e-01, 1.6131e+00, -2.4062e+00, ..., -2.4865e+00,
-2.6119e+00, -2.5473e+00]],
[[-1.3612e+00, -2.3290e+00, -2.1640e+00, ..., -8.4414e-01,
-6.2011e-01, -8.1447e-01],
[-2.0007e+00, -2.1435e+00, -1.5172e+00, ..., -7.0540e-01,
-4.2955e-01, -3.6117e-01],
[-2.5285e+00, -1.9894e+00, -8.2575e-01, ..., -4.1874e-01,
-1.1692e-01, -9.5099e-02],
...,
[-1.5850e+00, -1.8228e+00, -8.0501e-01, ..., -6.3707e-01,
-5.4011e-01, -6.9629e-01],
[-1.7481e+00, -1.1851e+00, -7.8366e-01, ..., -2.3571e-01,
-5.5562e-02, -5.0900e-02],
[-1.6799e+00, -1.0779e+00, -5.8090e-01, ..., -3.3047e-01,
-2.8840e-01, -2.5999e-01]],
...,
[[-2.3089e+00, -1.2453e+00, -2.6497e+00, ..., -3.1046e+00,
-2.4820e+00, -1.6601e+00],
[-1.8811e+00, 2.0003e-01, -1.3149e+00, ..., -1.5178e+00,
-1.5924e+00, -1.3789e+00],
[-4.7676e-01, 2.4907e+00, 9.0866e-01, ..., -9.2594e-02,
2.3787e-01, 3.2467e-01],
...,
[-1.7297e+00, 1.1704e+00, 1.9534e-01, ..., -9.6462e-01,
-2.8592e-01, 1.8024e-01],
[-8.6891e-01, 1.6468e+00, 1.4983e+00, ..., 1.7190e-01,
2.9851e-01, 3.9826e-01],
[-2.1286e+00, -1.0925e+00, -8.8281e-01, ..., -2.1491e+00,
-1.2489e+00, -6.4854e-01]],
[[-2.3744e+00, -2.6706e+00, -4.0607e+00, ..., -1.1078e+00,
-5.9561e-01, -4.8320e-01],
[-1.6670e+00, -1.3999e+00, -4.7829e+00, ..., -1.6668e+00,
-1.4565e+00, -1.7076e+00],
[-2.9875e+00, -2.4511e+00, -6.8744e+00, ..., -3.7152e+00,
-3.2491e+00, -2.6247e+00],
...,
[-2.5360e+00, -1.7543e+00, -4.9075e+00, ..., -2.1869e+00,
-1.7257e+00, -1.4047e+00],
[-2.2301e+00, -1.5617e+00, -3.8349e+00, ..., -2.1220e+00,
-1.4743e+00, -9.6425e-01],
[-2.3147e+00, -1.7711e+00, -3.0896e+00, ..., -2.8154e+00,
-1.9390e+00, -1.6606e+00]],
[[-1.8985e+00, -2.6145e+00, -2.3982e+00, ..., -1.7055e+00,
-1.4141e+00, -1.0591e+00],
[-1.9225e+00, -2.6348e+00, -2.1103e+00, ..., -1.3723e+00,
-1.1145e+00, -8.8235e-01],
[-2.4582e+00, -3.0181e+00, -2.5251e+00, ..., -1.5778e+00,
-1.2084e+00, -8.7007e-01],
...,
[-3.0976e+00, -3.8854e+00, -3.6680e+00, ..., -2.7826e+00,
-2.2204e+00, -1.6828e+00],
[-2.0582e+00, -2.6699e+00, -2.6414e+00, ..., -1.9602e+00,
-1.6088e+00, -1.2830e+00],
[-1.4769e+00, -1.7996e+00, -1.8500e+00, ..., -1.2255e+00,
-1.0452e+00, -7.7114e-01]]],
[[[ 4.2268e-01, 7.5691e-01, 8.1526e-01, ..., 6.3955e-01,
1.0535e+00, 6.8966e-01],
[-1.6832e+00, -9.1034e-01, -2.3749e+00, ..., -1.9511e+00,
-4.1806e-01, 1.5532e-01],
[ 2.5558e-01, -3.2685e-01, 1.6213e-01, ..., 2.1973e-01,
7.5876e-01, 7.8086e-01],
...,
[-2.0665e-01, -1.2877e+00, -3.0209e+00, ..., -1.6349e+00,
-1.5992e+00, -8.7027e-01],
[ 1.4339e+00, 6.2445e-01, -1.6188e+00, ..., -7.0992e-01,
-2.8529e-01, 1.4267e-01],
[ 6.8619e-01, -1.2341e-01, -1.2491e+00, ..., -6.3478e-01,
-1.1305e-01, 2.0733e-01]],
[[-1.9454e+00, -1.1423e+00, -4.1915e+00, ..., -5.0275e+00,
-4.4831e+00, -3.4593e+00],
[ 2.2902e-01, 1.8701e+00, -2.4040e+00, ..., -3.8509e+00,
-3.8345e+00, -3.1517e+00],
[ 9.6762e-01, 1.6749e+00, -5.0230e+00, ..., -5.8051e+00,
-5.5301e+00, -5.1411e+00],
...,
[-1.7444e+00, -2.2529e-01, -7.6397e+00, ..., -8.5399e+00,
-7.8658e+00, -6.0232e+00],
[ 1.4679e+00, 2.6467e+00, -1.5879e+00, ..., -3.1281e+00,
-3.6019e+00, -2.6951e+00],
[ 1.0615e+00, 1.8448e+00, -2.4924e+00, ..., -3.5153e+00,
-4.0206e+00, -3.3264e+00]],
[[-2.8479e-01, -1.3304e+00, -1.3600e+00, ..., -5.6204e-01,
-2.7241e-01, -5.4050e-01],
[-9.0381e-01, -1.2197e+00, -9.0745e-01, ..., -4.2944e-01,
-1.2786e-01, -1.5781e-01],
[-8.0747e-01, -6.4732e-01, -4.1218e-01, ..., -3.5842e-01,
5.8648e-03, 1.8462e-02],
...,
[-1.6584e+00, -1.8354e+00, -5.9694e-01, ..., -8.8617e-01,
-8.0565e-01, -1.1307e+00],
[-1.8849e+00, -1.1744e+00, -6.6130e-01, ..., -5.3365e-01,
-2.7176e-01, -3.9072e-01],
[-2.0846e+00, -1.5775e+00, -1.1336e+00, ..., -1.3446e+00,
-1.2239e+00, -1.1942e+00]],
...,
[[-2.7286e+00, -1.9710e+00, -2.6899e+00, ..., -3.2717e+00,
-2.5622e+00, -1.7134e+00],
[-1.6171e+00, 2.4445e-02, -1.2318e+00, ..., -2.1642e+00,
-2.0649e+00, -1.6698e+00],
[-6.6130e-01, 1.7794e+00, 6.1809e-01, ..., -3.4201e-01,
1.9279e-02, 1.3622e-01],
...,
[-2.1182e+00, 8.5573e-01, -2.1225e-01, ..., -1.1395e+00,
-6.3542e-01, -8.7513e-02],
[-8.9681e-01, 1.9073e+00, 1.8596e+00, ..., 7.8270e-01,
5.6351e-01, 8.2949e-01],
[-2.4414e+00, -1.2207e+00, -7.7690e-01, ..., -1.5878e+00,
-7.2098e-01, -8.2027e-02]],
[[-1.4487e+00, -1.3827e+00, -2.7836e+00, ..., -1.1429e+00,
-7.0111e-01, -6.6909e-01],
[-5.6678e-01, 7.3837e-02, -3.3240e+00, ..., -1.4832e+00,
-1.2767e+00, -1.5542e+00],
[-2.0664e+00, -1.0427e+00, -5.6000e+00, ..., -3.4033e+00,
-2.8744e+00, -2.2826e+00],
...,
[-2.7742e+00, -1.9784e+00, -5.4482e+00, ..., -3.2553e+00,
-3.0134e+00, -2.4061e+00],
[-2.4410e+00, -1.8291e+00, -4.3081e+00, ..., -2.8420e+00,
-2.6156e+00, -1.9083e+00],
[-2.5141e+00, -2.0742e+00, -3.4015e+00, ..., -3.0913e+00,
-2.6883e+00, -2.3567e+00]],
[[-1.2643e+00, -1.8230e+00, -1.6101e+00, ..., -1.7075e+00,
-1.4096e+00, -1.0642e+00],
[-1.2173e+00, -1.7023e+00, -1.1923e+00, ..., -1.3702e+00,
-1.0915e+00, -8.6143e-01],
[-1.5618e+00, -1.9007e+00, -1.4428e+00, ..., -1.6226e+00,
-1.2252e+00, -8.9540e-01],
...,
[-3.5043e+00, -4.4211e+00, -4.2532e+00, ..., -4.4291e+00,
-3.5608e+00, -2.7567e+00],
[-2.2946e+00, -3.0096e+00, -2.9980e+00, ..., -3.0699e+00,
-2.5349e+00, -2.0746e+00],
[-1.6825e+00, -2.0698e+00, -2.1378e+00, ..., -2.0961e+00,
-1.6792e+00, -1.3002e+00]]]], device='cuda:0'),
tensor([[[[ 1.8966e-02, 1.8375e-01, -1.8461e+00, ..., -6.0425e-01,
-2.6201e-01, -9.5477e-01],
[-2.3582e+00, -1.8324e+00, -5.3323e+00, ..., -3.2526e+00,
-1.2369e+00, -1.3899e+00],
[ 3.0205e-01, -5.5699e-01, -1.6242e+00, ..., -1.8965e+00,
-9.9065e-01, -7.5723e-01],
...,
[-5.9199e+00, -7.6765e+00, -1.0568e+01, ..., -2.1384e+01,
-1.9301e+01, -2.1532e+01],
[-1.3537e+00, -5.8194e+00, -1.0452e+01, ..., -1.8281e+01,
-1.6365e+01, -1.4296e+01],
[ 2.8025e+00, -4.0382e+00, -1.0089e+01, ..., -1.6279e+01,
-1.2685e+01, -5.8216e+00]],
[[-3.4698e+00, -1.5763e+00, -6.6139e+00, ..., -5.2764e+00,
-4.8036e+00, -3.7781e+00],
[-7.3890e-01, 2.0874e+00, -4.6505e+00, ..., -3.8916e+00,
-4.2880e+00, -3.4056e+00],
[-2.3452e-01, 1.7204e+00, -7.9837e+00, ..., -5.6804e+00,
-5.7308e+00, -4.9904e+00],
...,
[ 4.9612e+00, 7.2644e-01, -6.8263e+00, ..., -3.0125e+01,
-2.0701e+01, -1.1022e+01],
[ 3.7636e+00, 2.7034e+00, -1.8706e+00, ..., -1.8169e+01,
-1.1465e+01, -9.9230e+00],
[ 5.3014e-01, 4.4884e-01, -1.9625e+00, ..., -1.0708e+01,
-7.2268e+00, -9.6977e+00]],
[[-1.6839e+00, -2.5862e+00, -2.3748e+00, ..., -6.2401e-01,
8.0955e-02, -9.9511e-02],
[-2.2497e+00, -2.3419e+00, -1.6428e+00, ..., -2.2339e-01,
-2.3622e-02, -1.7095e-02],
[-2.9650e+00, -2.3338e+00, -9.4816e-01, ..., -2.2920e-01,
1.9374e-01, 3.4681e-01],
...,
[-3.4435e+00, -6.3304e-01, -1.5295e+00, ..., -5.0796e+00,
-4.9400e+00, -2.8332e+00],
[-2.1232e+00, -1.6109e-01, -1.0552e+00, ..., -1.8346e+00,
-3.7600e-01, 7.0974e-01],
[-2.7726e+00, -3.7281e+00, -2.8344e+00, ..., -1.6591e+00,
8.7598e-03, 5.5149e-01]],
...,
[[-2.3570e+00, -1.2963e+00, -2.7079e+00, ..., -3.5656e+00,
-2.7997e+00, -1.8828e+00],
[-2.2860e+00, -2.1532e-01, -1.6306e+00, ..., -2.6130e+00,
-2.4084e+00, -1.8679e+00],
[-5.9948e-01, 2.3727e+00, 8.8632e-01, ..., -3.7348e-01,
-6.9234e-02, 4.1843e-01],
...,
[-3.0629e+00, -1.9668e+00, -6.4079e+00, ..., -1.8187e+01,
-2.3039e+01, -7.7880e+00],
[-3.3965e+00, -3.4236e+00, -6.6906e-01, ..., -1.4220e+01,
-1.0988e+01, -3.5621e+00],
[-4.6160e-02, 9.8604e-01, 1.6061e+00, ..., -6.5562e+00,
-4.0346e+00, -1.5124e+00]],
[[-2.6296e+00, -3.0315e+00, -4.3290e+00, ..., -9.1135e-01,
-4.3069e-01, -5.0648e-01],
[-1.9391e+00, -1.7639e+00, -5.0499e+00, ..., -1.4936e+00,
-1.0076e+00, -1.3871e+00],
[-3.1918e+00, -2.7664e+00, -6.9856e+00, ..., -3.6403e+00,
-2.7388e+00, -2.2096e+00],
...,
[-2.2825e+00, -5.0355e+00, -2.2024e+01, ..., -5.9289e+00,
-8.5037e+00, -3.5723e-01],
[ 4.2035e-01, -6.5708e-01, -1.0518e+01, ..., -1.4071e+00,
-2.8151e-01, 4.8681e+00],
[-1.3094e+00, -1.8936e+00, -5.0378e+00, ..., -1.0118e+00,
-4.1406e-01, 4.2539e-01]],
[[-2.0844e+00, -2.8626e+00, -2.6692e+00, ..., -1.9084e+00,
-1.5788e+00, -1.1868e+00],
[-2.1222e+00, -2.9131e+00, -2.3886e+00, ..., -1.6966e+00,
-1.4149e+00, -1.1554e+00],
[-2.6822e+00, -3.3369e+00, -2.8666e+00, ..., -1.8936e+00,
-1.5042e+00, -1.1365e+00],
...,
[-6.0449e+00, -8.2414e+00, -8.2056e+00, ..., -8.7962e+00,
-7.2812e+00, -5.2217e+00],
[-4.0637e+00, -5.4630e+00, -5.0420e+00, ..., -7.0703e+00,
-5.9229e+00, -3.7938e+00],
[-2.4637e+00, -3.2276e+00, -3.1198e+00, ..., -3.7606e+00,
-2.8756e+00, -2.0232e+00]]],
[[[ 4.6094e-01, 8.1443e-01, 8.2815e-01, ..., 6.9372e-01,
1.0534e+00, 7.0831e-01],
[-1.5679e+00, -7.7069e-01, -2.2164e+00, ..., -1.7350e+00,
-3.0755e-01, 2.4647e-01],
[-1.7804e-01, -7.8073e-01, 9.6215e-02, ..., 3.4238e-01,
8.2670e-01, 8.3571e-01],
...,
[-2.3616e-01, -1.3363e+00, -3.1438e+00, ..., -1.5863e+00,
-1.6140e+00, -8.2547e-01],
[ 1.4747e+00, 6.3618e-01, -1.8051e+00, ..., -8.0132e-01,
-3.9075e-01, 9.1605e-02],
[ 6.5878e-01, -2.0869e-01, -1.4955e+00, ..., -7.4573e-01,
-2.1477e-01, 1.4752e-01]],
[[-2.0829e+00, -1.1431e+00, -4.2300e+00, ..., -5.1298e+00,
-4.4994e+00, -3.4874e+00],
[ 1.0050e-01, 1.9015e+00, -2.4467e+00, ..., -3.9529e+00,
-3.8761e+00, -3.1890e+00],
[ 5.3850e-01, 1.2304e+00, -5.8866e+00, ..., -5.9086e+00,
-5.6077e+00, -5.2189e+00],
...,
[-1.7597e+00, -1.1206e-01, -7.8176e+00, ..., -8.7975e+00,
-8.1667e+00, -6.2268e+00],
[ 1.4951e+00, 2.7690e+00, -1.6692e+00, ..., -3.3212e+00,
-3.8597e+00, -2.8376e+00],
[ 1.1221e+00, 1.9788e+00, -2.5592e+00, ..., -3.6192e+00,
-4.1555e+00, -3.3760e+00]],
[[-3.3252e-01, -1.3980e+00, -1.3954e+00, ..., -5.9756e-01,
-3.2778e-01, -5.6995e-01],
[-1.0023e+00, -1.3448e+00, -9.6700e-01, ..., -4.3939e-01,
-1.6265e-01, -1.7496e-01],
[-7.7534e-01, -8.5302e-01, -5.5070e-01, ..., -3.1748e-01,
1.7498e-02, 7.5847e-03],
...,
[-1.7951e+00, -1.9287e+00, -6.0133e-01, ..., -8.6107e-01,
-8.2074e-01, -1.1545e+00],
[-2.0472e+00, -1.2525e+00, -6.4732e-01, ..., -5.0471e-01,
-2.6213e-01, -4.0004e-01],
[-2.2694e+00, -1.7044e+00, -1.2024e+00, ..., -1.4379e+00,
-1.3278e+00, -1.3015e+00]],
...,
[[-2.6659e+00, -1.8095e+00, -2.6385e+00, ..., -3.2112e+00,
-2.5186e+00, -1.7065e+00],
[-1.4410e+00, 2.9822e-01, -1.0727e+00, ..., -2.0327e+00,
-1.9544e+00, -1.5921e+00],
[-3.8076e-01, 1.9726e+00, 3.9961e-02, ..., -3.4522e-01,
1.2285e-02, 1.4571e-01],
...,
[-2.2145e+00, 9.4832e-01, -1.0086e-01, ..., -1.1453e+00,
-6.3264e-01, -9.9999e-02],
[-9.3097e-01, 2.0302e+00, 2.0814e+00, ..., 8.4038e-01,
6.0985e-01, 8.8376e-01],
[-2.4055e+00, -1.1601e+00, -6.0368e-01, ..., -1.5061e+00,
-6.3058e-01, -7.1808e-03]],
[[-1.4226e+00, -1.3344e+00, -2.7447e+00, ..., -1.1143e+00,
-6.5057e-01, -5.9314e-01],
[-5.5384e-01, 8.3535e-02, -3.3488e+00, ..., -1.5294e+00,
-1.3047e+00, -1.5656e+00],
[-2.2206e+00, -9.3557e-01, -5.0881e+00, ..., -3.5002e+00,
-2.9549e+00, -2.3523e+00],
...,
[-2.8435e+00, -2.0824e+00, -5.6702e+00, ..., -3.3645e+00,
-3.1434e+00, -2.4661e+00],
[-2.4634e+00, -1.8447e+00, -4.4224e+00, ..., -2.9304e+00,
-2.7419e+00, -1.9875e+00],
[-2.5132e+00, -2.0315e+00, -3.4334e+00, ..., -3.1351e+00,
-2.7604e+00, -2.4218e+00]],
[[-1.2549e+00, -1.8264e+00, -1.6047e+00, ..., -1.6992e+00,
-1.4134e+00, -1.0587e+00],
[-1.2141e+00, -1.7098e+00, -1.2007e+00, ..., -1.3843e+00,
-1.1086e+00, -8.6620e-01],
[-1.7509e+00, -2.0799e+00, -1.5832e+00, ..., -1.6283e+00,
-1.2388e+00, -8.9560e-01],
...,
[-3.6388e+00, -4.6000e+00, -4.3656e+00, ..., -4.5706e+00,
-3.6725e+00, -2.8558e+00],
[-2.3993e+00, -3.1448e+00, -3.1064e+00, ..., -3.1630e+00,
-2.6131e+00, -2.1436e+00],
[-1.7806e+00, -2.1929e+00, -2.2320e+00, ..., -2.1716e+00,
-1.7308e+00, -1.3458e+00]]],
[[[ 2.0706e-01, 4.7187e-01, 6.0612e-01, ..., -3.3414e-01,
-7.6085e-02, 2.2610e-02],
[-2.0317e+00, -1.3449e+00, -2.8598e+00, ..., -3.2556e+00,
-1.2995e+00, -4.3173e-01],
[-1.9392e-01, -9.0015e-01, -3.3699e-02, ..., -1.8943e-01,
-3.4118e-01, 1.7300e-01],
...,
[-9.7899e-01, -2.1965e+00, -2.0961e+00, ..., -1.9742e+00,
-1.9051e+00, -1.2890e+00],
[ 7.9903e-01, -1.2851e-01, -1.3845e-01, ..., -5.2319e-04,
2.8081e-01, 4.5996e-01],
[ 2.0555e-01, -5.6717e-01, -4.1788e-01, ..., 2.3084e-01,
4.7724e-01, 5.1545e-01]],
[[-1.6846e+00, -1.4345e+00, -4.2469e+00, ..., -6.8422e+00,
-6.0833e+00, -4.5290e+00],
[ 4.2472e-01, 1.3825e+00, -2.5537e+00, ..., -5.4431e+00,
-5.5634e+00, -4.2984e+00],
[ 1.1172e+00, 1.1457e+00, -5.0580e+00, ..., -7.8812e+00,
-7.8298e+00, -6.6048e+00],
...,
[ 3.9021e-01, 3.6259e-01, -4.1219e+00, ..., -7.0965e+00,
-6.6879e+00, -5.1853e+00],
[ 2.5628e+00, 2.5223e+00, 1.6625e-01, ..., -1.9106e+00,
-2.5471e+00, -2.0456e+00],
[ 1.5301e+00, 1.3494e+00, -1.5839e+00, ..., -2.9287e+00,
-3.4791e+00, -3.0543e+00]],
[[-1.3143e-01, -1.1240e+00, -1.3703e+00, ..., -1.4162e+00,
-1.0573e+00, -1.3379e+00],
[-4.6964e-01, -7.7366e-01, -8.6119e-01, ..., -9.9904e-01,
-6.2140e-01, -8.9474e-01],
[-2.7338e-01, -1.8532e-01, -4.3258e-01, ..., -1.0964e+00,
-5.9035e-01, -4.6656e-01],
...,
[-6.1732e-02, -6.1084e-01, -6.1278e-01, ..., -1.0482e+00,
-8.8134e-01, -1.2451e+00],
[-7.1555e-01, -2.4167e-01, -5.9618e-01, ..., -5.6429e-01,
-2.6052e-01, -4.2208e-01],
[-8.7866e-01, -2.7092e-01, -4.9401e-01, ..., -4.9802e-01,
-4.4988e-01, -6.0716e-01]],
...,
[[-2.7799e+00, -2.4552e+00, -2.4430e+00, ..., -3.0850e+00,
-2.5336e+00, -1.8336e+00],
[-2.0564e+00, -9.1359e-01, -1.1540e+00, ..., -2.4047e+00,
-2.5155e+00, -2.0295e+00],
[-1.1185e+00, 7.2664e-01, 7.3545e-01, ..., -6.8410e-02,
1.9790e-02, 2.6890e-01],
...,
[-2.3096e+00, -1.8699e-01, -1.6284e-01, ..., -9.2979e-01,
-2.7208e-01, 2.8225e-01],
[-1.2693e+00, 8.6637e-01, 1.0835e+00, ..., 4.3096e-01,
4.8920e-01, 8.0567e-01],
[-2.1816e+00, -1.5128e+00, -1.6826e+00, ..., -2.2159e+00,
-1.1633e+00, -3.5099e-01]],
[[-1.6254e+00, -1.6424e+00, -2.7294e+00, ..., -2.2182e+00,
-1.9034e+00, -1.8143e+00],
[-7.1186e-01, -1.2117e-01, -3.0465e+00, ..., -2.8510e+00,
-2.8730e+00, -2.8981e+00],
[-2.0833e+00, -9.9014e-01, -5.0335e+00, ..., -4.4601e+00,
-4.5311e+00, -3.7094e+00],
...,
[-1.3266e+00, 1.9865e-01, -3.3772e+00, ..., -2.8359e+00,
-2.7228e+00, -2.2999e+00],
[-1.6759e+00, -4.7238e-01, -3.0122e+00, ..., -2.4491e+00,
-2.2400e+00, -1.6507e+00],
[-2.3378e+00, -1.3629e+00, -3.0399e+00, ..., -2.8665e+00,
-2.4268e+00, -2.1927e+00]],
[[-1.2991e+00, -1.8246e+00, -1.6611e+00, ..., -2.5825e+00,
-2.1486e+00, -1.6673e+00],
[-1.2345e+00, -1.7010e+00, -1.2375e+00, ..., -2.4078e+00,
-1.9333e+00, -1.5798e+00],
[-1.5481e+00, -1.8993e+00, -1.4857e+00, ..., -2.8205e+00,
-2.2440e+00, -1.7384e+00],
...,
[-2.4495e+00, -3.0611e+00, -2.7820e+00, ..., -4.0032e+00,
-3.2432e+00, -2.5455e+00],
[-1.5794e+00, -2.0829e+00, -1.9787e+00, ..., -2.7630e+00,
-2.3157e+00, -1.9180e+00],
[-1.0313e+00, -1.3410e+00, -1.3531e+00, ..., -1.8255e+00,
-1.5228e+00, -1.1839e+00]]],
[[[ 2.7565e-01, 5.3466e-01, 6.6826e-01, ..., 4.8993e-01,
1.0298e+00, 6.0072e-01],
[-2.1545e+00, -1.5291e+00, -3.1751e+00, ..., -2.8445e+00,
-9.4806e-01, -2.3059e-01],
[-1.9913e-01, -1.0028e+00, -2.1421e-01, ..., -2.0188e-01,
5.2211e-01, 6.4021e-01],
...,
[ 8.4838e-02, -6.8421e-01, -1.5662e+00, ..., -1.0358e+00,
-7.8774e-01, -5.7883e-01],
[ 1.5393e+00, 1.2423e+00, 6.0206e-01, ..., 7.7853e-01,
9.2954e-01, 9.4281e-01],
[ 1.1612e+00, 1.0086e+00, 6.8731e-01, ..., 9.2848e-01,
1.0227e+00, 8.7125e-01]],
[[-1.6795e+00, -1.2344e+00, -4.1911e+00, ..., -5.0587e+00,
-4.5571e+00, -3.4919e+00],
[ 5.0733e-01, 1.7450e+00, -2.3543e+00, ..., -3.5151e+00,
-3.6022e+00, -2.9842e+00],
[ 1.1938e+00, 1.4675e+00, -4.9463e+00, ..., -5.5116e+00,
-5.3333e+00, -5.0601e+00],
...,
[-6.2537e-01, 5.7974e-01, -4.6789e+00, ..., -5.8628e+00,
-5.2010e+00, -4.2455e+00],
[ 2.1923e+00, 2.9149e+00, 2.1865e-01, ..., -1.5192e+00,
-1.7017e+00, -1.6905e+00],
[ 1.2968e+00, 1.5893e+00, -1.6281e+00, ..., -2.7069e+00,
-2.8468e+00, -2.7357e+00]],
[[-9.3617e-02, -1.0723e+00, -1.2699e+00, ..., -5.4655e-01,
-2.0187e-01, -5.4041e-01],
[-5.1075e-01, -7.6144e-01, -7.3445e-01, ..., -3.2099e-01,
1.6502e-02, -8.4425e-02],
[-4.4768e-01, -2.6118e-01, -3.8155e-01, ..., -3.5612e-01,
9.6678e-02, 9.4258e-02],
...,
[-1.1131e+00, -1.5828e+00, -6.5840e-01, ..., -6.1956e-01,
-4.7810e-01, -6.2931e-01],
[-1.3583e+00, -8.8207e-01, -4.5198e-01, ..., -9.7282e-02,
9.4802e-02, 7.4637e-02],
[-1.4047e+00, -6.9594e-01, -1.4105e-01, ..., -4.2919e-02,
-2.8287e-02, -5.1653e-02]],
...,
[[-2.9747e+00, -2.5345e+00, -2.7823e+00, ..., -3.7009e+00,
-2.8339e+00, -1.8787e+00],
[-2.2868e+00, -9.5453e-01, -1.6419e+00, ..., -2.7371e+00,
-2.4590e+00, -1.9566e+00],
[-1.1214e+00, 9.7080e-01, 5.2917e-01, ..., -4.5944e-01,
4.5163e-02, 1.8456e-01],
...,
[-1.5748e+00, 1.6712e+00, 3.5701e-01, ..., -7.9417e-01,
-1.3672e-01, 3.7551e-01],
[-5.9356e-01, 1.8315e+00, 1.1543e+00, ..., 1.6852e-01,
3.3564e-01, 4.8827e-01],
[-1.7190e+00, -1.0070e+00, -1.4077e+00, ..., -2.0859e+00,
-1.1529e+00, -5.3773e-01]],
[[-1.6123e+00, -1.6059e+00, -2.8105e+00, ..., -1.1917e+00,
-7.3818e-01, -7.7829e-01],
[-6.5063e-01, -1.1481e-02, -3.1195e+00, ..., -1.2592e+00,
-1.1369e+00, -1.5876e+00],
[-2.1238e+00, -1.0025e+00, -5.1902e+00, ..., -3.1876e+00,
-2.7296e+00, -2.2579e+00],
...,
[-1.7384e+00, -6.3620e-01, -4.6885e+00, ..., -2.4673e+00,
-2.0370e+00, -1.6029e+00],
[-1.6628e+00, -7.5411e-01, -3.7530e+00, ..., -2.2874e+00,
-1.6863e+00, -1.0994e+00],
[-2.0755e+00, -1.3180e+00, -3.0022e+00, ..., -2.7592e+00,
-1.9708e+00, -1.6850e+00]],
[[-1.3368e+00, -1.8894e+00, -1.7157e+00, ..., -1.8395e+00,
-1.5156e+00, -1.1490e+00],
[-1.2637e+00, -1.7467e+00, -1.2591e+00, ..., -1.3979e+00,
-1.0978e+00, -8.7379e-01],
[-1.5720e+00, -1.9337e+00, -1.5169e+00, ..., -1.5997e+00,
-1.1917e+00, -8.7228e-01],
...,
[-2.4927e+00, -3.1067e+00, -2.8239e+00, ..., -2.9840e+00,
-2.3749e+00, -1.8116e+00],
[-1.6737e+00, -2.1212e+00, -2.0066e+00, ..., -2.0884e+00,
-1.7051e+00, -1.3623e+00],
[-1.1117e+00, -1.3376e+00, -1.3664e+00, ..., -1.2911e+00,
-1.0942e+00, -8.0657e-01]]],
[[[ 4.6409e-01, 9.3441e-01, 8.1168e-01, ..., -5.3021e-01,
-2.5802e-01, -9.2147e-02],
[-1.1361e+00, -2.8427e-01, -1.4074e+00, ..., -3.5579e+00,
-1.5218e+00, -6.0118e-01],
[ 5.9397e-01, 4.8270e-01, 4.3131e-01, ..., -3.1889e-01,
-5.7031e-01, 5.1023e-02],
...,
[-1.7252e-01, -8.2447e-01, -1.4575e+00, ..., -1.9667e+00,
-1.9498e+00, -1.2567e+00],
[ 1.1127e+00, 8.3411e-01, 3.2540e-01, ..., -9.4662e-02,
1.5545e-01, 4.0183e-01],
[ 7.8492e-01, 6.1296e-01, 2.6976e-01, ..., 9.8885e-02,
3.7081e-01, 4.5503e-01]],
[[-2.3370e+00, -1.2870e+00, -4.2546e+00, ..., -7.0417e+00,
-6.2375e+00, -4.6501e+00],
[-4.0973e-01, 1.4556e+00, -2.6861e+00, ..., -5.6013e+00,
-5.7325e+00, -4.4301e+00],
[ 2.4070e-01, 1.2708e+00, -5.1375e+00, ..., -8.0916e+00,
-8.0389e+00, -6.7330e+00],
...,
[-5.8996e-01, 4.7898e-01, -4.2003e+00, ..., -7.3757e+00,
-6.9539e+00, -5.3609e+00],
[ 2.0327e+00, 2.7251e+00, 3.2900e-01, ..., -2.0846e+00,
-2.7424e+00, -2.1434e+00],
[ 1.1773e+00, 1.4939e+00, -1.4955e+00, ..., -3.0182e+00,
-3.5899e+00, -3.1039e+00]],
[[-6.0291e-01, -1.5302e+00, -1.3688e+00, ..., -1.5270e+00,
-1.1377e+00, -1.4177e+00],
[-1.2469e+00, -1.5660e+00, -9.8602e-01, ..., -1.0666e+00,
-6.7942e-01, -9.8767e-01],
[-1.0907e+00, -9.3078e-01, -3.6237e-01, ..., -1.1716e+00,
-6.5356e-01, -5.1002e-01],
...,
[-8.9824e-01, -1.3333e+00, -5.1513e-01, ..., -1.0238e+00,
-8.8183e-01, -1.2775e+00],
[-1.2344e+00, -8.2395e-01, -5.1423e-01, ..., -5.3883e-01,
-2.3865e-01, -4.2445e-01],
[-1.3347e+00, -7.3104e-01, -3.2476e-01, ..., -5.9297e-01,
-5.6147e-01, -7.2217e-01]],
...,
[[-2.1079e+00, -1.2497e+00, -2.5217e+00, ..., -3.0029e+00,
-2.4383e+00, -1.7977e+00],
[-6.2293e-01, 9.7279e-01, -8.4867e-01, ..., -2.5321e+00,
-2.5838e+00, -2.0704e+00],
[-3.5255e-01, 2.0121e+00, 4.0292e-01, ..., -8.0915e-02,
2.7367e-02, 2.8056e-01],
...,
[-1.6019e+00, 1.1231e+00, -2.3944e-01, ..., -9.6950e-01,
-2.7502e-01, 2.6693e-01],
[-5.1614e-01, 1.5184e+00, 7.6169e-01, ..., 4.8162e-01,
5.2338e-01, 8.7227e-01],
[-1.6491e+00, -1.1053e+00, -1.6025e+00, ..., -2.1395e+00,
-1.0979e+00, -2.8024e-01]],
[[-1.3352e+00, -1.2430e+00, -2.5177e+00, ..., -2.3675e+00,
-2.0562e+00, -1.9574e+00],
[-7.0819e-01, -2.3904e-01, -3.2940e+00, ..., -2.9992e+00,
-3.0057e+00, -2.9920e+00],
[-2.1724e+00, -1.5709e+00, -5.6925e+00, ..., -4.5469e+00,
-4.6270e+00, -3.7903e+00],
...,
[-1.6915e+00, -6.0191e-01, -4.1663e+00, ..., -2.9013e+00,
-2.8446e+00, -2.3829e+00],
[-1.7024e+00, -7.9804e-01, -3.4775e+00, ..., -2.4888e+00,
-2.3521e+00, -1.7530e+00],
[-2.1250e+00, -1.3914e+00, -2.9487e+00, ..., -2.9029e+00,
-2.5024e+00, -2.2720e+00]],
[[-1.1365e+00, -1.6833e+00, -1.5400e+00, ..., -2.7174e+00,
-2.2527e+00, -1.7481e+00],
[-1.1547e+00, -1.6204e+00, -1.2485e+00, ..., -2.5575e+00,
-2.0479e+00, -1.6763e+00],
[-1.4971e+00, -1.7937e+00, -1.4459e+00, ..., -2.9919e+00,
-2.3885e+00, -1.8585e+00],
...,
[-2.3149e+00, -2.8818e+00, -2.6938e+00, ..., -4.1339e+00,
-3.3571e+00, -2.6326e+00],
[-1.5455e+00, -1.9698e+00, -1.9133e+00, ..., -2.8487e+00,
-2.3855e+00, -1.9762e+00],
[-1.0327e+00, -1.2415e+00, -1.3086e+00, ..., -1.8907e+00,
-1.5661e+00, -1.2212e+00]]]], device='cuda:0'),
tensor([[[[ 1.1973e-01, 3.5070e-01, -1.1840e+00, ..., 4.6265e-01,
1.0208e+00, 6.7039e-01],
[-1.9842e+00, -1.3458e+00, -4.1368e+00, ..., -2.5804e+00,
-7.4473e-01, -4.0074e-02],
[ 3.5211e-01, -2.6660e-01, -1.0549e+00, ..., -2.7585e-01,
4.8932e-01, 6.4734e-01],
...,
[-2.9036e-01, -1.3721e+00, -2.8008e+00, ..., -2.0218e+00,
-1.3633e+00, -9.0616e-01],
[ 1.3681e+00, 7.5365e-01, -9.0608e-01, ..., -3.5070e-01,
5.1806e-01, 7.0545e-01],
[ 9.2333e-01, 4.1993e-01, -3.9793e-01, ..., -3.6155e-01,
2.9652e-01, 2.8673e-01]],
[[-3.1392e+00, -1.5476e+00, -6.0817e+00, ..., -4.8477e+00,
-4.4246e+00, -3.3714e+00],
[-7.2895e-01, 1.6974e+00, -4.4038e+00, ..., -3.3876e+00,
-3.5386e+00, -2.9012e+00],
[-2.6778e-01, 1.3386e+00, -7.4945e+00, ..., -5.3827e+00,
-5.2599e+00, -4.9329e+00],
...,
[-1.0693e+00, 6.0372e-01, -6.2669e+00, ..., -4.5495e+00,
-4.0926e+00, -3.4507e+00],
[ 1.9664e+00, 3.1451e+00, -6.7590e-01, ..., -8.9545e-01,
-1.2038e+00, -1.2814e+00],
[ 1.2979e+00, 1.9757e+00, -2.1051e+00, ..., -2.2376e+00,
-2.4655e+00, -2.3844e+00]],
[[-1.3005e+00, -2.1603e+00, -2.0159e+00, ..., -5.8125e-01,
-1.8779e-01, -4.6799e-01],
[-1.8429e+00, -1.9800e+00, -1.4366e+00, ..., -3.3487e-01,
7.3191e-02, 3.1516e-02],
[-2.3470e+00, -1.8405e+00, -7.7877e-01, ..., -2.7775e-01,
2.5338e-01, 2.8194e-01],
...,
[-1.6981e+00, -1.9944e+00, -8.3723e-01, ..., -4.8157e-01,
-1.6392e-01, -3.5195e-01],
[-1.9168e+00, -1.2960e+00, -8.2338e-01, ..., -3.1661e-01,
1.4208e-01, 2.0988e-01],
[-1.9015e+00, -1.2026e+00, -6.1570e-01, ..., -5.0081e-01,
-2.8271e-01, -2.5543e-01]],
...,
[[-2.1190e+00, -1.2418e+00, -2.5523e+00, ..., -3.7048e+00,
-2.8507e+00, -1.8496e+00],
[-1.7898e+00, 4.1465e-02, -1.2981e+00, ..., -2.7048e+00,
-2.4483e+00, -1.8657e+00],
[-4.8663e-01, 2.1992e+00, 8.2952e-01, ..., -5.7868e-01,
-7.2767e-02, 1.7439e-01],
...,
[-1.9216e+00, 1.3102e+00, 2.6675e-01, ..., -1.5094e+00,
-7.1295e-01, -1.6707e-01],
[-9.2064e-01, 1.8190e+00, 1.6415e+00, ..., -1.1842e-01,
1.2185e-01, 3.1443e-01],
[-2.2445e+00, -1.1905e+00, -1.0141e+00, ..., -2.3319e+00,
-1.4290e+00, -8.9435e-01]],
[[-2.3663e+00, -2.6836e+00, -3.8862e+00, ..., -1.1255e+00,
-7.2423e-01, -7.5663e-01],
[-1.7695e+00, -1.5664e+00, -4.5786e+00, ..., -1.3485e+00,
-1.2159e+00, -1.5867e+00],
[-3.0063e+00, -2.5850e+00, -6.5755e+00, ..., -3.2727e+00,
-2.7882e+00, -2.2367e+00],
...,
[-2.4411e+00, -1.4670e+00, -5.0685e+00, ..., -1.4294e+00,
-1.0271e+00, -8.9730e-01],
[-2.1965e+00, -1.3704e+00, -3.9678e+00, ..., -1.7060e+00,
-1.1624e+00, -7.9554e-01],
[-2.3871e+00, -1.7162e+00, -3.1434e+00, ..., -2.7851e+00,
-2.0211e+00, -1.8550e+00]],
[[-1.8349e+00, -2.5262e+00, -2.3669e+00, ..., -1.8172e+00,
-1.5007e+00, -1.1498e+00],
[-1.8775e+00, -2.5622e+00, -2.1259e+00, ..., -1.4055e+00,
-1.1115e+00, -8.9221e-01],
[-2.3946e+00, -2.9494e+00, -2.5601e+00, ..., -1.6004e+00,
-1.2033e+00, -8.9263e-01],
...,
[-3.1962e+00, -3.9954e+00, -3.6977e+00, ..., -2.8769e+00,
-2.3016e+00, -1.7659e+00],
[-2.1064e+00, -2.7283e+00, -2.6391e+00, ..., -1.9846e+00,
-1.6319e+00, -1.3129e+00],
[-1.4994e+00, -1.8199e+00, -1.8418e+00, ..., -1.2328e+00,
-1.0410e+00, -7.9771e-01]]],
[[[ 7.6262e-02, 2.3505e-01, -1.4035e+00, ..., -6.0393e-01,
-1.9373e-01, -4.1336e-02],
[-2.0795e+00, -1.5890e+00, -4.5276e+00, ..., -3.7190e+00,
-1.5175e+00, -5.4667e-01],
[ 5.8759e-01, -8.7487e-02, -8.3538e-01, ..., -1.8909e-01,
-2.7285e-01, 2.4419e-01],
...,
[-8.5373e-01, -1.9067e+00, -1.9035e+00, ..., -1.7278e+00,
-1.2498e+00, -8.4225e-01],
[ 8.3980e-01, 3.8089e-03, -5.3540e-02, ..., -1.7773e-02,
7.0145e-01, 8.1863e-01],
[ 1.9038e-01, -5.6794e-01, -4.3032e-01, ..., -1.5837e-01,
4.0726e-01, 3.4964e-01]],
[[-3.3315e+00, -1.7982e+00, -6.2563e+00, ..., -7.1260e+00,
-6.2375e+00, -4.6198e+00],
[-7.9849e-01, 1.5606e+00, -4.3439e+00, ..., -5.4885e+00,
-5.5503e+00, -4.2855e+00],
[-3.3534e-01, 1.0960e+00, -7.4359e+00, ..., -8.0051e+00,
-7.8532e+00, -6.6853e+00],
...,
[ 5.3568e-01, 6.5472e-01, -3.9130e+00, ..., -4.6539e+00,
-4.2160e+00, -3.5697e+00],
[ 2.6645e+00, 2.6390e+00, 1.8806e-01, ..., -9.4320e-01,
-1.3003e+00, -1.3643e+00],
[ 1.5775e+00, 1.3851e+00, -1.6058e+00, ..., -2.2925e+00,
-2.5269e+00, -2.4353e+00]],
[[-1.3303e+00, -2.3491e+00, -2.3306e+00, ..., -1.7568e+00,
-1.2490e+00, -1.3798e+00],
[-1.7115e+00, -2.0061e+00, -1.5460e+00, ..., -1.2280e+00,
-7.2478e-01, -8.1340e-01],
[-2.1747e+00, -1.9373e+00, -1.0319e+00, ..., -1.0969e+00,
-4.6485e-01, -1.8894e-01],
...,
[-3.0688e-01, -7.4221e-01, -4.9070e-01, ..., -4.6634e-01,
-1.9299e-01, -4.4101e-01],
[-8.5700e-01, -2.4342e-01, -3.9737e-01, ..., -2.0353e-01,
2.2905e-01, 2.2918e-01],
[-9.5794e-01, -2.6154e-01, -3.3346e-01, ..., -3.9255e-01,
-2.0945e-01, -2.4177e-01]],
...,
[[-2.0580e+00, -1.1117e+00, -2.3504e+00, ..., -3.1758e+00,
-2.5941e+00, -1.8089e+00],
[-1.7834e+00, 1.7366e-02, -1.3583e+00, ..., -2.4872e+00,
-2.5078e+00, -1.9373e+00],
[-3.5393e-01, 2.3945e+00, 1.0943e+00, ..., 1.1891e-01,
2.6955e-01, 5.1866e-01],
...,
[-2.1513e+00, 8.7468e-02, -1.4951e-01, ..., -1.3202e+00,
-3.9946e-01, 6.5622e-02],
[-1.2234e+00, 9.4972e-01, 9.7433e-01, ..., 3.5072e-02,
3.1280e-01, 4.3884e-01],
[-2.1800e+00, -1.4752e+00, -1.7585e+00, ..., -2.3454e+00,
-1.3965e+00, -8.9137e-01]],
[[-2.4207e+00, -2.8177e+00, -4.2111e+00, ..., -2.3879e+00,
-1.9066e+00, -1.7154e+00],
[-1.7494e+00, -1.6175e+00, -4.8869e+00, ..., -2.9305e+00,
-2.7944e+00, -2.7725e+00],
[-2.8193e+00, -2.4504e+00, -6.7055e+00, ..., -4.4598e+00,
-4.3781e+00, -3.4967e+00],
...,
[-1.3373e+00, 1.3929e-01, -3.4659e+00, ..., -1.4651e+00,
-1.0878e+00, -9.9623e-01],
[-1.6671e+00, -4.9579e-01, -3.1197e+00, ..., -1.7395e+00,
-1.2208e+00, -8.6828e-01],
[-2.3624e+00, -1.4547e+00, -3.1320e+00, ..., -2.9338e+00,
-2.1554e+00, -1.9771e+00]],
[[-1.8095e+00, -2.5456e+00, -2.4549e+00, ..., -2.6959e+00,
-2.2068e+00, -1.6906e+00],
[-1.7972e+00, -2.5067e+00, -2.1417e+00, ..., -2.4039e+00,
-1.8999e+00, -1.5365e+00],
[-2.2511e+00, -2.8067e+00, -2.4993e+00, ..., -2.7430e+00,
-2.1395e+00, -1.6215e+00],
...,
[-2.4904e+00, -3.0685e+00, -2.7933e+00, ..., -2.8308e+00,
-2.2516e+00, -1.7272e+00],
[-1.6125e+00, -2.0876e+00, -1.9695e+00, ..., -1.9277e+00,
-1.5783e+00, -1.2692e+00],
[-1.0423e+00, -1.3289e+00, -1.3386e+00, ..., -1.1862e+00,
-1.0065e+00, -7.6144e-01]]],
[[[ 2.3454e-01, 5.2255e-01, 6.0938e-01, ..., 4.9958e-01,
8.5788e-01, 6.8031e-01],
[-1.8606e+00, -1.1459e+00, -2.6074e+00, ..., -1.4878e+00,
-5.9425e-02, 5.1583e-01],
[-2.5689e-02, -6.9458e-01, 9.5767e-02, ..., 4.0838e-01,
8.1952e-01, 8.9357e-01],
...,
[-8.5255e-01, -1.9750e+00, -1.9685e+00, ..., -1.4141e+00,
-9.7310e-01, -6.7826e-01],
[ 8.4843e-01, 8.8869e-03, -3.2161e-02, ..., 3.8097e-01,
7.3811e-01, 8.5798e-01],
[ 2.8728e-01, -4.1692e-01, -2.8888e-01, ..., 5.8399e-01,
8.5048e-01, 7.7604e-01]],
[[-1.7681e+00, -1.3712e+00, -4.2528e+00, ..., -5.9263e+00,
-5.0159e+00, -3.8986e+00],
[ 3.3289e-01, 1.4294e+00, -2.6005e+00, ..., -4.8767e+00,
-4.5642e+00, -3.7296e+00],
[ 1.0487e+00, 1.1738e+00, -5.0960e+00, ..., -6.9761e+00,
-6.4951e+00, -5.9078e+00],
...,
[ 2.5791e-01, 3.4883e-01, -4.1661e+00, ..., -6.2381e+00,
-5.3795e+00, -4.4086e+00],
[ 2.4779e+00, 2.4884e+00, 9.4887e-02, ..., -1.5218e+00,
-1.6227e+00, -1.6486e+00],
[ 1.4744e+00, 1.3304e+00, -1.6520e+00, ..., -2.8228e+00,
-2.9319e+00, -2.8406e+00]],
[[-1.8732e-01, -1.2203e+00, -1.4073e+00, ..., -8.3374e-01,
-5.5442e-01, -6.8256e-01],
[-5.8483e-01, -9.4007e-01, -9.2806e-01, ..., -6.8346e-01,
-3.7070e-01, -2.6729e-01],
[-3.9881e-01, -3.1875e-01, -4.5504e-01, ..., -4.3068e-01,
-4.1576e-02, 7.0320e-02],
...,
[-1.5608e-01, -7.1712e-01, -6.1310e-01, ..., -7.5363e-01,
-5.3565e-01, -6.1670e-01],
[-7.7394e-01, -3.1328e-01, -5.6933e-01, ..., -3.5264e-01,
-6.5827e-02, 1.1018e-02],
[-9.3714e-01, -3.5039e-01, -4.6692e-01, ..., -3.3239e-01,
-2.2860e-01, -1.7735e-01]],
...,
[[-2.7108e+00, -2.2536e+00, -2.3733e+00, ..., -3.0623e+00,
-2.5427e+00, -1.7268e+00],
[-1.8839e+00, -5.5107e-01, -9.7369e-01, ..., -1.7000e+00,
-1.7765e+00, -1.4877e+00],
[-1.0022e+00, 9.9615e-01, 8.0278e-01, ..., -2.6670e-01,
2.0446e-02, 1.8300e-01],
...,
[-2.1639e+00, 3.9093e-02, -8.9047e-02, ..., -1.0132e+00,
-3.2233e-01, 1.9162e-01],
[-1.1439e+00, 9.8016e-01, 1.0578e+00, ..., 2.5078e-01,
3.9805e-01, 5.1659e-01],
[-2.0710e+00, -1.3711e+00, -1.5786e+00, ..., -2.1933e+00,
-1.2060e+00, -5.6266e-01]],
[[-1.5972e+00, -1.5791e+00, -2.7468e+00, ..., -1.5592e+00,
-9.0729e-01, -6.5985e-01],
[-7.1733e-01, -1.3248e-01, -3.1788e+00, ..., -2.1655e+00,
-1.7548e+00, -1.8017e+00],
[-2.0743e+00, -1.0695e+00, -5.1981e+00, ..., -4.0555e+00,
-3.4375e+00, -2.6356e+00],
...,
[-1.3751e+00, 7.3147e-02, -3.5434e+00, ..., -2.5014e+00,
-1.9101e+00, -1.4125e+00],
[-1.6505e+00, -4.8619e-01, -3.0859e+00, ..., -2.3145e+00,
-1.5892e+00, -9.3989e-01],
[-2.2652e+00, -1.3121e+00, -2.9818e+00, ..., -2.9008e+00,
-1.9760e+00, -1.6661e+00]],
[[-1.2703e+00, -1.7972e+00, -1.6253e+00, ..., -1.9739e+00,
-1.6006e+00, -1.1723e+00],
[-1.2263e+00, -1.6864e+00, -1.2268e+00, ..., -1.7671e+00,
-1.3971e+00, -1.0624e+00],
[-1.5493e+00, -1.8895e+00, -1.4782e+00, ..., -2.0794e+00,
-1.5622e+00, -1.1104e+00],
...,
[-2.4244e+00, -3.0242e+00, -2.7546e+00, ..., -3.2868e+00,
-2.5958e+00, -1.9352e+00],
[-1.5832e+00, -2.0765e+00, -1.9846e+00, ..., -2.3218e+00,
-1.8887e+00, -1.4853e+00],
[-1.0400e+00, -1.3512e+00, -1.3717e+00, ..., -1.4511e+00,
-1.2186e+00, -8.8944e-01]]],
[[[ 2.2675e-01, 4.8649e-01, 6.3973e-01, ..., -3.1594e-01,
-5.9507e-02, 3.5449e-02],
[-2.0288e+00, -1.3664e+00, -2.8664e+00, ..., -3.1645e+00,
-1.2395e+00, -4.0935e-01],
[-1.6665e-01, -9.1357e-01, 7.0132e-03, ..., -1.8304e-01,
-3.3696e-01, 1.8010e-01],
...,
[-1.0291e+00, -2.2199e+00, -2.1116e+00, ..., -1.9463e+00,
-1.8938e+00, -1.2667e+00],
[ 7.5799e-01, -1.5931e-01, -1.7805e-01, ..., 1.9261e-02,
2.8167e-01, 4.6022e-01],
[ 1.7159e-01, -6.1691e-01, -4.5665e-01, ..., 2.4055e-01,
4.8808e-01, 5.1548e-01]],
[[-1.6600e+00, -1.3719e+00, -4.2695e+00, ..., -6.8185e+00,
-6.0286e+00, -4.4916e+00],
[ 4.9287e-01, 1.4684e+00, -2.5322e+00, ..., -5.4662e+00,
-5.5653e+00, -4.3003e+00],
[ 1.2233e+00, 1.2035e+00, -5.0441e+00, ..., -7.8697e+00,
-7.8273e+00, -6.5972e+00],
...,
[ 3.9575e-01, 3.6850e-01, -4.1120e+00, ..., -7.0803e+00,
-6.6585e+00, -5.1773e+00],
[ 2.5589e+00, 2.5119e+00, 1.5701e-01, ..., -1.9146e+00,
-2.5500e+00, -2.0546e+00],
[ 1.5308e+00, 1.3331e+00, -1.6176e+00, ..., -2.9288e+00,
-3.4750e+00, -3.0510e+00]],
[[-1.2566e-01, -1.1547e+00, -1.3869e+00, ..., -1.3745e+00,
-1.0357e+00, -1.3249e+00],
[-5.0578e-01, -8.3561e-01, -8.9990e-01, ..., -9.6486e-01,
-5.9378e-01, -8.8927e-01],
[-3.2854e-01, -2.3549e-01, -4.5883e-01, ..., -1.0883e+00,
-5.8549e-01, -4.7613e-01],
...,
[-5.4699e-02, -6.1968e-01, -6.1509e-01, ..., -1.0285e+00,
-8.7099e-01, -1.2424e+00],
[-7.1576e-01, -2.3118e-01, -5.8803e-01, ..., -5.6503e-01,
-2.6374e-01, -4.2469e-01],
[-8.9656e-01, -2.8948e-01, -4.9711e-01, ..., -4.9056e-01,
-4.3999e-01, -6.0263e-01]],
...,
[[-2.8467e+00, -2.4711e+00, -2.4824e+00, ..., -3.0594e+00,
-2.4982e+00, -1.8119e+00],
[-2.1270e+00, -8.8017e-01, -1.1474e+00, ..., -2.4221e+00,
-2.5138e+00, -2.0031e+00],
[-1.1268e+00, 7.8197e-01, 7.7023e-01, ..., -5.7337e-02,
-1.4487e-02, 2.6746e-01],
...,
[-2.2686e+00, -1.7816e-01, -1.3309e-01, ..., -9.3073e-01,
-2.6642e-01, 2.8365e-01],
[-1.2378e+00, 8.7548e-01, 1.0907e+00, ..., 4.3053e-01,
4.8658e-01, 8.0673e-01],
[-2.1544e+00, -1.4758e+00, -1.6563e+00, ..., -2.2020e+00,
-1.1680e+00, -3.5501e-01]],
[[-1.6398e+00, -1.6413e+00, -2.8140e+00, ..., -2.2256e+00,
-1.9325e+00, -1.8274e+00],
[-7.1392e-01, -1.0616e-01, -3.1283e+00, ..., -2.8610e+00,
-2.8901e+00, -2.9141e+00],
[-2.0794e+00, -9.7339e-01, -5.0996e+00, ..., -4.4316e+00,
-4.4908e+00, -3.7238e+00],
...,
[-1.3037e+00, 2.3403e-01, -3.3755e+00, ..., -2.8238e+00,
-2.7289e+00, -2.3067e+00],
[-1.6368e+00, -4.1815e-01, -3.0133e+00, ..., -2.4466e+00,
-2.2515e+00, -1.6602e+00],
[-2.3031e+00, -1.3229e+00, -3.0251e+00, ..., -2.8707e+00,
-2.4257e+00, -2.1919e+00]],
[[-1.3019e+00, -1.8302e+00, -1.6490e+00, ..., -2.5501e+00,
-2.1320e+00, -1.6537e+00],
[-1.2401e+00, -1.7034e+00, -1.2209e+00, ..., -2.3907e+00,
-1.9249e+00, -1.5795e+00],
[-1.5608e+00, -1.9012e+00, -1.4761e+00, ..., -2.8094e+00,
-2.2403e+00, -1.7457e+00],
...,
[-2.4632e+00, -3.0739e+00, -2.7998e+00, ..., -3.9921e+00,
-3.2362e+00, -2.5370e+00],
[-1.5965e+00, -2.1043e+00, -1.9947e+00, ..., -2.7614e+00,
-2.3113e+00, -1.9110e+00],
[-1.0449e+00, -1.3605e+00, -1.3737e+00, ..., -1.8248e+00,
-1.5206e+00, -1.1790e+00]]],
[[[ 2.9990e-01, 5.6149e-01, 6.8059e-01, ..., 4.7516e-01,
1.0118e+00, 6.0568e-01],
[-2.1618e+00, -1.4984e+00, -3.1851e+00, ..., -2.6458e+00,
-8.4854e-01, -1.7745e-01],
[-1.6287e-01, -9.6277e-01, -2.1207e-01, ..., -1.4588e-01,
5.4436e-01, 6.5175e-01],
...,
[ 1.4916e-01, -6.4489e-01, -1.5469e+00, ..., -1.0531e+00,
-7.6668e-01, -5.6890e-01],
[ 1.5618e+00, 1.2517e+00, 5.8877e-01, ..., 7.5985e-01,
9.3559e-01, 9.5200e-01],
[ 1.1680e+00, 9.9598e-01, 6.7598e-01, ..., 9.3589e-01,
1.0312e+00, 8.7730e-01]],
[[-1.6748e+00, -1.1670e+00, -4.1697e+00, ..., -4.8716e+00,
-4.4694e+00, -3.3961e+00],
[ 5.9076e-01, 1.9104e+00, -2.2622e+00, ..., -3.5184e+00,
-3.6494e+00, -2.9757e+00],
[ 1.3347e+00, 1.6796e+00, -4.9063e+00, ..., -5.4806e+00,
-5.3089e+00, -4.9354e+00],
...,
[-6.4586e-01, 5.1836e-01, -4.7674e+00, ..., -5.8835e+00,
-5.2592e+00, -4.2617e+00],
[ 2.0746e+00, 2.7866e+00, 4.4219e-02, ..., -1.5385e+00,
-1.7490e+00, -1.7177e+00],
[ 1.2314e+00, 1.5075e+00, -1.7207e+00, ..., -2.7306e+00,
-2.8849e+00, -2.7450e+00]],
[[-1.2533e-01, -1.1537e+00, -1.3210e+00, ..., -5.1361e-01,
-1.6892e-01, -4.8040e-01],
[-6.1248e-01, -8.7116e-01, -7.9528e-01, ..., -2.8477e-01,
3.2762e-02, -6.5948e-02],
[-5.6309e-01, -3.5827e-01, -4.1494e-01, ..., -3.5979e-01,
7.0657e-02, 8.3729e-02],
...,
[-1.1376e+00, -1.5669e+00, -6.5200e-01, ..., -6.3455e-01,
-4.7376e-01, -6.3209e-01],
[-1.3445e+00, -8.5874e-01, -4.3998e-01, ..., -1.1468e-01,
9.1510e-02, 7.3595e-02],
[-1.3457e+00, -6.4152e-01, -1.0199e-01, ..., -6.3160e-02,
-3.1729e-02, -5.6764e-02]],
...,
[[-3.0455e+00, -2.5197e+00, -2.8108e+00, ..., -3.4583e+00,
-2.6347e+00, -1.7496e+00],
[-2.2872e+00, -8.5133e-01, -1.5792e+00, ..., -2.6114e+00,
-2.3275e+00, -1.8238e+00],
[-1.0718e+00, 1.1544e+00, 6.2429e-01, ..., -4.4398e-01,
-7.1686e-03, 1.3339e-01],
...,
[-1.5299e+00, 1.6898e+00, 3.9573e-01, ..., -8.1755e-01,
-1.5334e-01, 3.7144e-01],
[-6.0772e-01, 1.7667e+00, 1.1314e+00, ..., 1.5891e-01,
3.2459e-01, 4.9324e-01],
[-1.7027e+00, -9.9491e-01, -1.4124e+00, ..., -2.0746e+00,
-1.1429e+00, -5.2478e-01]],
[[-1.5990e+00, -1.5991e+00, -2.9189e+00, ..., -1.2692e+00,
-8.2817e-01, -8.3006e-01],
[-6.1286e-01, 5.8601e-02, -3.2175e+00, ..., -1.3839e+00,
-1.2162e+00, -1.5687e+00],
[-2.1157e+00, -9.3097e-01, -5.3391e+00, ..., -3.2023e+00,
-2.7139e+00, -2.1981e+00],
...,
[-1.8095e+00, -7.2900e-01, -4.6822e+00, ..., -2.4739e+00,
-2.0628e+00, -1.6074e+00],
[-1.6897e+00, -7.9054e-01, -3.7208e+00, ..., -2.2987e+00,
-1.7021e+00, -1.1128e+00],
[-2.0511e+00, -1.3063e+00, -3.0077e+00, ..., -2.7697e+00,
-1.9705e+00, -1.6826e+00]],
[[-1.3517e+00, -1.9087e+00, -1.7105e+00, ..., -1.7738e+00,
-1.4636e+00, -1.1149e+00],
[-1.2755e+00, -1.7615e+00, -1.2435e+00, ..., -1.3795e+00,
-1.0869e+00, -8.6582e-01],
[-1.5993e+00, -1.9443e+00, -1.4874e+00, ..., -1.6006e+00,
-1.1979e+00, -8.8569e-01],
...,
[-2.4866e+00, -3.1066e+00, -2.8447e+00, ..., -2.9982e+00,
-2.3943e+00, -1.8297e+00],
[-1.6812e+00, -2.1329e+00, -2.0340e+00, ..., -2.0957e+00,
-1.7142e+00, -1.3711e+00],
[-1.1274e+00, -1.3510e+00, -1.3808e+00, ..., -1.2953e+00,
-1.0994e+00, -8.1078e-01]]]], device='cuda:0'),
tensor([[[[ 2.2520e-01, 4.9240e-01, 6.0175e-01, ..., -3.8706e-01,
-1.1345e-01, -9.8751e-03],
[-2.0274e+00, -1.3705e+00, -2.9296e+00, ..., -3.2947e+00,
-1.3375e+00, -4.7788e-01],
[-1.7401e-01, -9.3134e-01, -3.8669e-02, ..., -1.7597e-01,
-3.7072e-01, 1.5080e-01],
...,
[-9.5103e-01, -2.1319e+00, -2.0317e+00, ..., -1.0304e+01,
-6.8214e+00, -1.6212e+00],
[ 7.9417e-01, -1.0698e-01, -9.5435e-02, ..., -5.6019e+00,
-3.6040e+00, 1.4580e-01],
[ 2.1689e-01, -5.4269e-01, -3.6764e-01, ..., -2.2755e+00,
-1.1380e+00, 3.9218e-01]],
[[-1.6688e+00, -1.3867e+00, -4.2703e+00, ..., -6.8966e+00,
-6.1017e+00, -4.5367e+00],
[ 4.8167e-01, 1.4507e+00, -2.5774e+00, ..., -5.4811e+00,
-5.5875e+00, -4.3011e+00],
[ 1.2139e+00, 1.2096e+00, -5.1138e+00, ..., -7.8881e+00,
-7.8448e+00, -6.6213e+00],
...,
[ 3.0652e-01, 2.5480e-01, -4.1790e+00, ..., -9.4397e+00,
-7.5037e+00, -5.2095e+00],
[ 2.4761e+00, 2.4111e+00, 9.1416e-02, ..., -2.9827e+00,
-2.7805e+00, -2.0436e+00],
[ 1.4681e+00, 1.2821e+00, -1.6037e+00, ..., -3.4912e+00,
-3.5548e+00, -3.0585e+00]],
[[-1.3805e-01, -1.1748e+00, -1.4009e+00, ..., -1.4174e+00,
-1.0551e+00, -1.3478e+00],
[-4.9628e-01, -8.2964e-01, -8.8863e-01, ..., -1.0118e+00,
-6.2782e-01, -9.1589e-01],
[-2.9108e-01, -2.1004e-01, -4.5311e-01, ..., -1.1131e+00,
-6.0217e-01, -4.7365e-01],
...,
[-1.4107e-02, -5.8231e-01, -5.9645e-01, ..., -8.1416e-01,
-5.8863e-01, -1.2612e+00],
[-6.7428e-01, -2.2230e-01, -5.7460e-01, ..., -4.7640e-01,
-1.8006e-01, -3.9929e-01],
[-8.5467e-01, -2.7630e-01, -4.8456e-01, ..., -7.3259e-01,
-7.8918e-01, -5.9593e-01]],
...,
[[-2.8108e+00, -2.4448e+00, -2.5008e+00, ..., -3.0869e+00,
-2.5194e+00, -1.8256e+00],
[-2.0645e+00, -8.0584e-01, -1.1042e+00, ..., -2.4554e+00,
-2.5343e+00, -2.0360e+00],
[-1.1431e+00, 7.6867e-01, 7.3794e-01, ..., -1.0537e-01,
-7.2921e-03, 2.5603e-01],
...,
[-2.3351e+00, -2.8740e-01, -2.1184e-01, ..., -5.7213e-01,
-5.7677e-01, 3.1078e-01],
[-1.2765e+00, 7.8492e-01, 9.9934e-01, ..., 8.6533e-01,
7.8076e-01, 8.8858e-01],
[-2.1889e+00, -1.5229e+00, -1.7108e+00, ..., -1.6167e+00,
-5.5300e-01, -2.8470e-01]],
[[-1.6267e+00, -1.6149e+00, -2.7668e+00, ..., -2.2638e+00,
-1.9245e+00, -1.8251e+00],
[-6.9930e-01, -1.0210e-01, -3.1311e+00, ..., -2.8743e+00,
-2.8848e+00, -2.9141e+00],
[-2.0563e+00, -1.0003e+00, -5.1371e+00, ..., -4.4663e+00,
-4.5370e+00, -3.7315e+00],
...,
[-1.3749e+00, 1.1373e-01, -3.3759e+00, ..., -4.7183e+00,
-3.3433e+00, -2.3424e+00],
[-1.6979e+00, -5.2410e-01, -3.0179e+00, ..., -3.2622e+00,
-2.4213e+00, -1.6422e+00],
[-2.3365e+00, -1.3924e+00, -3.0230e+00, ..., -2.3953e+00,
-2.0723e+00, -2.1333e+00]],
[[-1.2991e+00, -1.8231e+00, -1.6545e+00, ..., -2.5923e+00,
-2.1724e+00, -1.6849e+00],
[-1.2439e+00, -1.7087e+00, -1.2442e+00, ..., -2.4255e+00,
-1.9539e+00, -1.5995e+00],
[-1.5672e+00, -1.9040e+00, -1.5068e+00, ..., -2.8439e+00,
-2.2713e+00, -1.7592e+00],
...,
[-2.4179e+00, -3.0208e+00, -2.7692e+00, ..., -5.3632e+00,
-3.5105e+00, -2.5840e+00],
[-1.5577e+00, -2.0556e+00, -1.9662e+00, ..., -3.7360e+00,
-2.5249e+00, -1.9408e+00],
[-1.0138e+00, -1.3241e+00, -1.3493e+00, ..., -2.3783e+00,
-1.6726e+00, -1.1948e+00]]],
[[[ 6.2667e-02, 2.0919e-01, -1.4431e+00, ..., -6.2450e-01,
-1.7930e-01, -5.6755e-02],
[-2.1204e+00, -1.6505e+00, -4.5523e+00, ..., -3.8060e+00,
-1.4947e+00, -5.6231e-01],
[ 5.4140e-01, -9.3575e-02, -8.3174e-01, ..., -2.4052e-01,
-2.5615e-01, 2.3700e-01],
...,
[-3.4837e-01, -1.2504e+00, -1.6361e+00, ..., -1.2326e+00,
-8.3135e-01, -6.2078e-01],
[ 1.0902e+00, 6.2252e-01, 3.1504e-01, ..., 3.8160e-01,
6.7932e-01, 7.8895e-01],
[ 7.6894e-01, 4.9012e-01, 3.1837e-01, ..., 5.1695e-01,
7.6138e-01, 7.0521e-01]],
[[-3.3137e+00, -1.8662e+00, -6.2778e+00, ..., -7.1486e+00,
-6.2860e+00, -4.6319e+00],
[-8.3499e-01, 1.4558e+00, -4.3589e+00, ..., -5.4863e+00,
-5.5693e+00, -4.2933e+00],
[-4.1587e-01, 9.7238e-01, -7.4436e+00, ..., -8.0038e+00,
-7.8640e+00, -6.7179e+00],
...,
[-2.3540e-01, 6.8427e-01, -4.1869e+00, ..., -5.3431e+00,
-4.6293e+00, -3.8845e+00],
[ 2.4243e+00, 3.0091e+00, 5.3361e-01, ..., -1.1748e+00,
-1.2977e+00, -1.4225e+00],
[ 1.4356e+00, 1.6670e+00, -1.4418e+00, ..., -2.4598e+00,
-2.5760e+00, -2.5578e+00]],
[[-1.3549e+00, -2.3553e+00, -2.3321e+00, ..., -1.7500e+00,
-1.2412e+00, -1.3867e+00],
[-1.6821e+00, -1.9662e+00, -1.5341e+00, ..., -1.2606e+00,
-7.3166e-01, -8.2598e-01],
[-2.1626e+00, -1.8963e+00, -9.9569e-01, ..., -1.1230e+00,
-4.7167e-01, -1.9723e-01],
...,
[-7.9151e-01, -1.3111e+00, -6.2744e-01, ..., -5.8146e-01,
-4.7052e-01, -6.3279e-01],
[-1.2467e+00, -8.4286e-01, -6.3570e-01, ..., -2.5451e-01,
-6.3335e-02, -4.5387e-02],
[-1.4249e+00, -8.0409e-01, -4.3214e-01, ..., -3.2358e-01,
-2.7653e-01, -2.4319e-01]],
...,
[[-2.0073e+00, -1.1187e+00, -2.3722e+00, ..., -3.1861e+00,
-2.6564e+00, -1.8363e+00],
[-1.7653e+00, -5.9574e-03, -1.3504e+00, ..., -2.4837e+00,
-2.5583e+00, -1.9968e+00],
[-4.2088e-01, 2.2681e+00, 1.0448e+00, ..., 1.4939e-01,
2.8914e-01, 5.3251e-01],
...,
[-1.7206e+00, 1.1796e+00, -6.3900e-02, ..., -1.1344e+00,
-4.5671e-01, 3.3300e-02],
[-5.7590e-01, 1.6452e+00, 9.2751e-01, ..., 1.2587e-02,
1.7134e-01, 2.7862e-01],
[-1.6789e+00, -1.0891e+00, -1.5819e+00, ..., -2.2535e+00,
-1.3053e+00, -6.9307e-01]],
[[-2.4393e+00, -2.8538e+00, -4.1783e+00, ..., -2.3970e+00,
-1.9213e+00, -1.7227e+00],
[-1.7860e+00, -1.6610e+00, -4.8072e+00, ..., -2.9006e+00,
-2.7754e+00, -2.7754e+00],
[-2.8575e+00, -2.5097e+00, -6.6103e+00, ..., -4.4357e+00,
-4.3613e+00, -3.4972e+00],
...,
[-1.5985e+00, -2.6126e-01, -4.2304e+00, ..., -2.0901e+00,
-1.5729e+00, -1.2285e+00],
[-1.6193e+00, -5.3048e-01, -3.4768e+00, ..., -2.0558e+00,
-1.3701e+00, -8.2914e-01],
[-2.0883e+00, -1.1886e+00, -2.8800e+00, ..., -2.6578e+00,
-1.7913e+00, -1.5220e+00]],
[[-1.8106e+00, -2.5448e+00, -2.4838e+00, ..., -2.7114e+00,
-2.2283e+00, -1.7087e+00],
[-1.8089e+00, -2.5192e+00, -2.1819e+00, ..., -2.4213e+00,
-1.9182e+00, -1.5514e+00],
[-2.2586e+00, -2.8113e+00, -2.5446e+00, ..., -2.7666e+00,
-2.1572e+00, -1.6365e+00],
...,
[-2.3968e+00, -2.9971e+00, -2.7183e+00, ..., -2.8324e+00,
-2.2557e+00, -1.7009e+00],
[-1.6122e+00, -2.0701e+00, -1.9634e+00, ..., -2.0022e+00,
-1.6422e+00, -1.3057e+00],
[-1.0833e+00, -1.3189e+00, -1.3552e+00, ..., -1.2453e+00,
-1.0630e+00, -7.8261e-01]]],
[[[ 4.7900e-01, 9.5703e-01, 8.4879e-01, ..., -3.7379e-01,
-1.2366e-01, -1.9941e-02],
[-1.2191e+00, -3.2329e-01, -1.4988e+00, ..., -3.2748e+00,
-1.3390e+00, -4.8895e-01],
[ 6.1113e-01, 4.4822e-01, 3.8390e-01, ..., -1.8250e-01,
-4.0000e-01, 1.3687e-01],
...,
[-1.8028e-01, -8.5905e-01, -1.5293e+00, ..., -1.9444e+00,
-1.9130e+00, -1.2973e+00],
[ 1.0676e+00, 7.8345e-01, 2.7287e-01, ..., 2.6002e-03,
2.3903e-01, 4.1491e-01],
[ 7.2210e-01, 5.2797e-01, 2.0681e-01, ..., 1.9078e-01,
4.4774e-01, 4.8135e-01]],
[[-2.3336e+00, -1.1642e+00, -4.1549e+00, ..., -6.8941e+00,
-6.1097e+00, -4.5534e+00],
[-3.1660e-01, 1.6600e+00, -2.5596e+00, ..., -5.4984e+00,
-5.6174e+00, -4.3313e+00],
[ 3.8707e-01, 1.4579e+00, -5.1095e+00, ..., -7.9014e+00,
-7.8587e+00, -6.6298e+00],
...,
[-4.1971e-01, 6.8166e-01, -4.0670e+00, ..., -7.1715e+00,
-6.7263e+00, -5.2145e+00],
[ 2.1528e+00, 2.8541e+00, 4.1477e-01, ..., -1.9543e+00,
-2.5785e+00, -2.0582e+00],
[ 1.2538e+00, 1.5617e+00, -1.4712e+00, ..., -2.9225e+00,
-3.4895e+00, -3.0527e+00]],
[[-5.9369e-01, -1.5825e+00, -1.4201e+00, ..., -1.4284e+00,
-1.0565e+00, -1.3440e+00],
[-1.2886e+00, -1.6247e+00, -1.0118e+00, ..., -1.0134e+00,
-6.3018e-01, -9.1196e-01],
[-1.1151e+00, -9.6628e-01, -3.8412e-01, ..., -1.1205e+00,
-6.2103e-01, -4.7601e-01],
...,
[-9.5522e-01, -1.3797e+00, -5.1885e-01, ..., -1.0400e+00,
-8.8096e-01, -1.2615e+00],
[-1.3164e+00, -9.0113e-01, -5.6226e-01, ..., -5.5409e-01,
-2.4965e-01, -4.2594e-01],
[-1.4111e+00, -7.9382e-01, -3.6347e-01, ..., -5.0865e-01,
-4.6519e-01, -6.3685e-01]],
...,
[[-2.1919e+00, -1.2625e+00, -2.5424e+00, ..., -3.0553e+00,
-2.4802e+00, -1.8009e+00],
[-6.9448e-01, 9.4034e-01, -9.2899e-01, ..., -2.4429e+00,
-2.5365e+00, -2.0550e+00],
[-2.7094e-01, 2.1706e+00, 4.7428e-01, ..., -7.7294e-02,
3.2689e-02, 2.6136e-01],
...,
[-1.5151e+00, 1.3239e+00, -1.3443e-01, ..., -9.0351e-01,
-2.6024e-01, 2.9040e-01],
[-5.0556e-01, 1.6315e+00, 8.6461e-01, ..., 4.6595e-01,
4.9570e-01, 8.2909e-01],
[-1.5878e+00, -1.0359e+00, -1.5319e+00, ..., -2.2007e+00,
-1.1544e+00, -3.4805e-01]],
[[-1.3006e+00, -1.2107e+00, -2.5516e+00, ..., -2.2837e+00,
-1.9557e+00, -1.8520e+00],
[-6.2491e-01, -1.2272e-01, -3.3038e+00, ..., -2.9168e+00,
-2.9028e+00, -2.9057e+00],
[-2.1094e+00, -1.4390e+00, -5.7171e+00, ..., -4.4889e+00,
-4.5312e+00, -3.7143e+00],
...,
[-1.5990e+00, -4.4526e-01, -4.1470e+00, ..., -2.8384e+00,
-2.7385e+00, -2.3183e+00],
[-1.6074e+00, -6.7570e-01, -3.4525e+00, ..., -2.4549e+00,
-2.2611e+00, -1.6753e+00],
[-2.0663e+00, -1.3197e+00, -2.9462e+00, ..., -2.8685e+00,
-2.4445e+00, -2.2185e+00]],
[[-1.1463e+00, -1.7038e+00, -1.5284e+00, ..., -2.5982e+00,
-2.1660e+00, -1.6859e+00],
[-1.1496e+00, -1.6139e+00, -1.2068e+00, ..., -2.4317e+00,
-1.9599e+00, -1.6061e+00],
[-1.4914e+00, -1.7869e+00, -1.4016e+00, ..., -2.8592e+00,
-2.2828e+00, -1.7720e+00],
...,
[-2.3481e+00, -2.9221e+00, -2.7025e+00, ..., -4.0400e+00,
-3.2810e+00, -2.5673e+00],
[-1.5816e+00, -2.0050e+00, -1.9319e+00, ..., -2.7888e+00,
-2.3389e+00, -1.9291e+00],
[-1.0588e+00, -1.2669e+00, -1.3197e+00, ..., -1.8458e+00,
-1.5366e+00, -1.1873e+00]]],
[[[ 4.1440e-01, 8.6141e-01, 5.0725e-01, ..., 4.7243e-01,
9.9739e-01, 6.6329e-01],
[-1.3497e+00, -4.4514e-01, -2.0030e+00, ..., -2.4228e+00,
-6.8985e-01, -1.6677e-02],
[ 8.6952e-01, 6.4123e-01, 3.5519e-01, ..., -2.2073e-01,
4.7358e-01, 6.3639e-01],
...,
[ 2.8359e-02, -6.0328e-01, -1.3742e+00, ..., -2.1915e+00,
-1.5545e+00, -1.0868e+00],
[ 1.3890e+00, 1.1748e+00, 6.2040e-01, ..., -3.0289e-01,
5.0688e-01, 6.8714e-01],
[ 1.0299e+00, 8.9342e-01, 6.0566e-01, ..., -2.9018e-01,
3.4366e-01, 2.8816e-01]],
[[-2.8271e+00, -1.5830e+00, -4.8359e+00, ..., -4.7574e+00,
-4.3327e+00, -3.3275e+00],
[-5.1865e-01, 1.6052e+00, -2.8946e+00, ..., -3.4090e+00,
-3.5348e+00, -2.9146e+00],
[ 2.6228e-01, 1.3942e+00, -5.6877e+00, ..., -5.3378e+00,
-5.1946e+00, -4.8712e+00],
...,
[-9.3524e-01, 4.8811e-02, -4.9063e+00, ..., -4.5876e+00,
-4.0862e+00, -3.4131e+00],
[ 1.7059e+00, 2.3394e+00, -3.1323e-01, ..., -5.6832e-01,
-9.1215e-01, -1.0400e+00],
[ 9.9905e-01, 1.2732e+00, -1.8149e+00, ..., -2.1602e+00,
-2.4240e+00, -2.3689e+00]],
[[-7.2550e-01, -1.9101e+00, -1.7923e+00, ..., -5.5048e-01,
-1.6763e-01, -4.3545e-01],
[-1.4926e+00, -1.9252e+00, -1.2838e+00, ..., -3.2976e-01,
7.4411e-02, 3.9393e-02],
[-1.3973e+00, -1.3390e+00, -7.3642e-01, ..., -2.7567e-01,
2.2716e-01, 2.6778e-01],
...,
[-9.1241e-01, -1.4335e+00, -7.1940e-01, ..., -5.5384e-01,
-1.9279e-01, -4.1359e-01],
[-1.2084e+00, -9.0757e-01, -6.2016e-01, ..., -3.2257e-01,
1.7097e-01, 2.3096e-01],
[-1.2593e+00, -7.2532e-01, -3.2051e-01, ..., -4.9714e-01,
-2.6489e-01, -2.4860e-01]],
...,
[[-2.2794e+00, -1.1781e+00, -2.5644e+00, ..., -3.5094e+00,
-2.7063e+00, -1.7565e+00],
[-6.0167e-01, 1.2036e+00, -7.8170e-01, ..., -2.5802e+00,
-2.3505e+00, -1.8148e+00],
[-1.5223e-01, 2.5610e+00, 7.6601e-01, ..., -5.7559e-01,
-8.6221e-02, 1.3975e-01],
...,
[-1.3432e+00, 1.4004e+00, 6.1873e-02, ..., -1.5593e+00,
-6.7137e-01, -4.8426e-02],
[-5.1765e-01, 1.5664e+00, 8.6865e-01, ..., -9.2840e-02,
1.6437e-01, 4.1124e-01],
[-1.6405e+00, -1.0696e+00, -1.4855e+00, ..., -2.5296e+00,
-1.5617e+00, -9.5794e-01]],
[[-1.4168e+00, -1.4227e+00, -3.0730e+00, ..., -1.1760e+00,
-7.8616e-01, -7.9035e-01],
[-6.5600e-01, -2.2553e-01, -3.9117e+00, ..., -1.3945e+00,
-1.2464e+00, -1.5776e+00],
[-2.1673e+00, -1.5539e+00, -6.4138e+00, ..., -3.2440e+00,
-2.7599e+00, -2.1991e+00],
...,
[-1.8250e+00, -9.6759e-01, -4.7143e+00, ..., -1.3281e+00,
-9.4387e-01, -8.0656e-01],
[-1.7118e+00, -1.0075e+00, -3.7773e+00, ..., -1.6532e+00,
-1.0737e+00, -6.6989e-01],
[-2.0304e+00, -1.4357e+00, -3.0468e+00, ..., -2.8600e+00,
-2.0273e+00, -1.8485e+00]],
[[-1.2768e+00, -1.9299e+00, -1.7658e+00, ..., -1.7701e+00,
-1.4694e+00, -1.1195e+00],
[-1.2717e+00, -1.8410e+00, -1.4157e+00, ..., -1.3960e+00,
-1.1101e+00, -8.8852e-01],
[-1.6401e+00, -2.0290e+00, -1.6420e+00, ..., -1.6164e+00,
-1.2121e+00, -9.0303e-01],
...,
[-2.3728e+00, -3.0043e+00, -2.8814e+00, ..., -2.9636e+00,
-2.3732e+00, -1.8128e+00],
[-1.6097e+00, -2.0696e+00, -2.0695e+00, ..., -2.0241e+00,
-1.6734e+00, -1.3515e+00],
[-1.1119e+00, -1.3590e+00, -1.4607e+00, ..., -1.2264e+00,
-1.0533e+00, -8.0269e-01]]],
[[[ 2.2441e-01, 5.0006e-01, 6.3671e-01, ..., -3.0356e-01,
-4.9263e-02, 3.3344e-02],
[-2.0342e+00, -1.3418e+00, -2.8850e+00, ..., -3.1219e+00,
-1.2429e+00, -4.0981e-01],
[-1.3892e-01, -8.8490e-01, 1.7603e-02, ..., -1.3044e-01,
-3.3808e-01, 1.5932e-01],
...,
[-1.0114e+00, -2.1991e+00, -2.1130e+00, ..., -1.9394e+00,
-1.8950e+00, -1.2676e+00],
[ 7.7752e-01, -1.5039e-01, -1.5323e-01, ..., 5.5901e-02,
2.9837e-01, 4.7292e-01],
[ 2.0631e-01, -5.8987e-01, -4.2056e-01, ..., 2.4439e-01,
4.7676e-01, 5.0755e-01]],
[[-1.6558e+00, -1.3888e+00, -4.2553e+00, ..., -6.8148e+00,
-6.0832e+00, -4.5067e+00],
[ 5.1028e-01, 1.4652e+00, -2.5216e+00, ..., -5.4998e+00,
-5.5869e+00, -4.2570e+00],
[ 1.2403e+00, 1.1918e+00, -5.0592e+00, ..., -7.8611e+00,
-7.7675e+00, -6.5194e+00],
...,
[ 3.2432e-01, 3.1604e-01, -4.1051e+00, ..., -7.1156e+00,
-6.6784e+00, -5.1957e+00],
[ 2.5174e+00, 2.4928e+00, 1.4303e-01, ..., -1.9756e+00,
-2.6018e+00, -2.0840e+00],
[ 1.5008e+00, 1.3234e+00, -1.6071e+00, ..., -2.9575e+00,
-3.4871e+00, -3.0555e+00]],
[[-1.3634e-01, -1.1713e+00, -1.4141e+00, ..., -1.3868e+00,
-1.0198e+00, -1.3129e+00],
[-5.1044e-01, -8.2092e-01, -8.8653e-01, ..., -9.7607e-01,
-6.0966e-01, -8.8070e-01],
[-3.1971e-01, -1.9122e-01, -4.1243e-01, ..., -1.1062e+00,
-5.9511e-01, -4.5101e-01],
...,
[-2.2299e-02, -5.8506e-01, -6.0459e-01, ..., -1.0280e+00,
-8.7458e-01, -1.2350e+00],
[-6.7637e-01, -2.1894e-01, -5.7677e-01, ..., -5.5523e-01,
-2.6090e-01, -4.2984e-01],
[-8.4731e-01, -2.6884e-01, -4.9709e-01, ..., -4.8912e-01,
-4.4094e-01, -5.9884e-01]],
...,
[[-2.8713e+00, -2.4970e+00, -2.4595e+00, ..., -2.9997e+00,
-2.4124e+00, -1.7613e+00],
[-2.1327e+00, -8.8175e-01, -1.1076e+00, ..., -2.4428e+00,
-2.4374e+00, -1.9682e+00],
[-1.1785e+00, 7.0312e-01, 7.6290e-01, ..., -1.8771e-01,
-2.3403e-02, 2.4685e-01],
...,
[-2.3273e+00, -2.7722e-01, -2.1554e-01, ..., -8.9772e-01,
-2.4059e-01, 2.9971e-01],
[-1.2869e+00, 8.2503e-01, 1.0271e+00, ..., 4.1311e-01,
4.9403e-01, 8.2911e-01],
[-2.2345e+00, -1.5450e+00, -1.7072e+00, ..., -2.1892e+00,
-1.1546e+00, -3.3824e-01]],
[[-1.6426e+00, -1.6515e+00, -2.8100e+00, ..., -2.3105e+00,
-1.9534e+00, -1.8127e+00],
[-6.9699e-01, -1.1449e-01, -3.1507e+00, ..., -2.9047e+00,
-2.8371e+00, -2.8246e+00],
[-2.0670e+00, -9.9415e-01, -5.1160e+00, ..., -4.4759e+00,
-4.4550e+00, -3.6378e+00],
...,
[-1.3388e+00, 2.0406e-01, -3.2895e+00, ..., -2.8683e+00,
-2.7596e+00, -2.3391e+00],
[-1.6861e+00, -4.7347e-01, -2.9725e+00, ..., -2.4617e+00,
-2.2533e+00, -1.6940e+00],
[-2.3458e+00, -1.3784e+00, -3.0284e+00, ..., -2.8568e+00,
-2.4151e+00, -2.2015e+00]],
[[-1.3135e+00, -1.8418e+00, -1.6709e+00, ..., -2.5633e+00,
-2.1459e+00, -1.6779e+00],
[-1.2496e+00, -1.7122e+00, -1.2394e+00, ..., -2.4065e+00,
-1.9510e+00, -1.6029e+00],
[-1.5642e+00, -1.9001e+00, -1.4930e+00, ..., -2.8299e+00,
-2.2643e+00, -1.7585e+00],
...,
[-2.4351e+00, -3.0632e+00, -2.7993e+00, ..., -3.9893e+00,
-3.2399e+00, -2.5331e+00],
[-1.5677e+00, -2.0840e+00, -1.9893e+00, ..., -2.7720e+00,
-2.3236e+00, -1.9168e+00],
[-1.0170e+00, -1.3388e+00, -1.3602e+00, ..., -1.8435e+00,
-1.5340e+00, -1.1901e+00]]]], device='cuda:0'),
tensor([[[[ 2.8217e-01, 5.5548e-01, 6.9271e-01, ..., 4.8180e-01,
1.0143e+00, 6.0313e-01],
[-2.1807e+00, -1.4790e+00, -3.2086e+00, ..., -2.6842e+00,
-8.8929e-01, -2.1376e-01],
[-1.8251e-01, -9.9008e-01, -2.6003e-01, ..., -1.5208e-01,
5.4553e-01, 6.5336e-01],
...,
[-1.4547e-01, -1.0786e+00, -2.5195e+00, ..., -1.5314e+00,
-1.4405e+00, -8.6358e-01],
[ 1.4103e+00, 8.6698e-01, -8.5040e-01, ..., -2.6884e-01,
9.1308e-02, 3.6578e-01],
[ 8.4102e-01, 3.2028e-01, -6.1415e-01, ..., -1.6368e-01,
2.1889e-01, 3.7883e-01]],
[[-1.6506e+00, -1.1495e+00, -4.1564e+00, ..., -4.8974e+00,
-4.4619e+00, -3.4145e+00],
[ 6.2252e-01, 1.9353e+00, -2.2671e+00, ..., -3.5063e+00,
-3.6207e+00, -2.9621e+00],
[ 1.3508e+00, 1.6747e+00, -4.9367e+00, ..., -5.4494e+00,
-5.2850e+00, -4.9252e+00],
...,
[-1.2694e+00, 1.0092e-01, -6.7221e+00, ..., -7.5561e+00,
-6.8315e+00, -5.3995e+00],
[ 1.7046e+00, 2.6633e+00, -1.2016e+00, ..., -2.5689e+00,
-2.9056e+00, -2.3585e+00],
[ 1.1531e+00, 1.7064e+00, -2.3440e+00, ..., -3.2417e+00,
-3.5939e+00, -3.1176e+00]],
[[-1.0411e-01, -1.1387e+00, -1.3022e+00, ..., -5.0203e-01,
-1.7004e-01, -4.8538e-01],
[-5.7819e-01, -8.4178e-01, -7.6106e-01, ..., -2.8302e-01,
4.0434e-02, -6.5573e-02],
[-5.4064e-01, -3.5241e-01, -4.0877e-01, ..., -3.3996e-01,
9.9452e-02, 8.9768e-02],
...,
[-1.3471e+00, -1.7042e+00, -7.4878e-01, ..., -8.5645e-01,
-7.4748e-01, -1.0128e+00],
[-1.7013e+00, -1.1257e+00, -8.0016e-01, ..., -5.4667e-01,
-2.8188e-01, -3.5853e-01],
[-1.8424e+00, -1.2524e+00, -8.6381e-01, ..., -9.4569e-01,
-8.0699e-01, -7.8991e-01]],
...,
[[-3.0610e+00, -2.5345e+00, -2.7748e+00, ..., -3.5299e+00,
-2.6664e+00, -1.7793e+00],
[-2.3397e+00, -9.1039e-01, -1.5970e+00, ..., -2.6790e+00,
-2.3839e+00, -1.8634e+00],
[-1.0881e+00, 1.1269e+00, 6.2749e-01, ..., -4.7910e-01,
-4.4606e-02, 1.1562e-01],
...,
[-1.8079e+00, 1.2195e+00, -2.8780e-02, ..., -1.1387e+00,
-5.2199e-01, 9.0015e-03],
[-8.3054e-01, 1.7927e+00, 1.5065e+00, ..., 4.4129e-01,
3.9984e-01, 6.7480e-01],
[-2.1939e+00, -1.1634e+00, -1.0638e+00, ..., -1.8879e+00,
-1.0096e+00, -3.1599e-01]],
[[-1.6110e+00, -1.5998e+00, -2.9231e+00, ..., -1.2731e+00,
-8.3377e-01, -8.3618e-01],
[-6.0877e-01, 6.5880e-02, -3.2429e+00, ..., -1.3568e+00,
-1.2090e+00, -1.5783e+00],
[-2.1072e+00, -9.1466e-01, -5.3406e+00, ..., -3.1845e+00,
-2.7094e+00, -2.2234e+00],
...,
[-2.4790e+00, -1.6288e+00, -5.4984e+00, ..., -3.0192e+00,
-2.7040e+00, -2.1932e+00],
[-2.2046e+00, -1.4938e+00, -4.2663e+00, ..., -2.6413e+00,
-2.2821e+00, -1.6454e+00],
[-2.3436e+00, -1.7700e+00, -3.2975e+00, ..., -2.9540e+00,
-2.4090e+00, -2.1151e+00]],
[[-1.3597e+00, -1.9199e+00, -1.7175e+00, ..., -1.7721e+00,
-1.4692e+00, -1.1134e+00],
[-1.2802e+00, -1.7723e+00, -1.2452e+00, ..., -1.3673e+00,
-1.0829e+00, -8.5914e-01],
[-1.6092e+00, -1.9522e+00, -1.4955e+00, ..., -1.5905e+00,
-1.1941e+00, -8.7951e-01],
...,
[-3.1572e+00, -3.9415e+00, -3.7626e+00, ..., -3.9442e+00,
-3.1641e+00, -2.4353e+00],
[-2.1100e+00, -2.7201e+00, -2.7057e+00, ..., -2.7655e+00,
-2.2801e+00, -1.8558e+00],
[-1.5417e+00, -1.8670e+00, -1.9361e+00, ..., -1.8644e+00,
-1.5110e+00, -1.1590e+00]]],
[[[ 4.4000e-01, 8.7521e-01, 4.7338e-01, ..., 6.6574e-01,
9.4487e-01, 7.4564e-01],
[-1.1505e+00, -2.5258e-01, -1.7909e+00, ..., -1.4699e+00,
-1.1244e-01, 5.0912e-01],
[ 7.1294e-01, 4.7021e-01, 3.8129e-03, ..., 3.8850e-01,
7.0398e-01, 7.8739e-01],
...,
[-6.8982e-01, -1.5590e+00, -1.6351e+00, ..., -1.6372e+00,
-1.2299e+00, -8.8538e-01],
[ 8.3090e-01, 2.1259e-01, 5.9662e-02, ..., 1.2166e-01,
7.6286e-01, 8.5121e-01],
[ 2.6897e-01, -3.1653e-01, -2.7450e-01, ..., 1.8657e-02,
5.4405e-01, 4.4797e-01]],
[[-2.7291e+00, -1.5302e+00, -5.0431e+00, ..., -6.0890e+00,
-5.2116e+00, -4.0386e+00],
[-7.0683e-01, 1.3213e+00, -3.6128e+00, ..., -4.8908e+00,
-4.7074e+00, -3.8021e+00],
[-5.8411e-02, 1.1099e+00, -6.3938e+00, ..., -7.1009e+00,
-6.7831e+00, -6.1291e+00],
...,
[ 8.8291e-02, 3.0380e-01, -4.0923e+00, ..., -4.7857e+00,
-4.2772e+00, -3.5735e+00],
[ 2.1894e+00, 2.2737e+00, -1.3885e-01, ..., -8.6083e-01,
-1.1889e+00, -1.2651e+00],
[ 1.2785e+00, 1.1806e+00, -1.7493e+00, ..., -2.2791e+00,
-2.5362e+00, -2.4457e+00]],
[[-5.9853e-01, -1.6049e+00, -1.4373e+00, ..., -8.1961e-01,
-5.5553e-01, -7.9164e-01],
[-1.3197e+00, -1.6943e+00, -1.1297e+00, ..., -7.5774e-01,
-4.0951e-01, -3.7957e-01],
[-1.2278e+00, -1.1273e+00, -5.7343e-01, ..., -5.2794e-01,
-1.3805e-01, -1.2592e-01],
...,
[-2.2153e-01, -7.0986e-01, -4.6211e-01, ..., -4.9203e-01,
-2.1716e-01, -4.7303e-01],
[-7.4776e-01, -2.9586e-01, -4.2482e-01, ..., -1.7748e-01,
2.3367e-01, 2.1451e-01],
[-8.4681e-01, -2.6245e-01, -3.2298e-01, ..., -3.3963e-01,
-1.6191e-01, -2.0775e-01]],
...,
[[-2.0526e+00, -1.2348e+00, -2.5175e+00, ..., -3.3217e+00,
-2.7395e+00, -1.8757e+00],
[-6.0009e-01, 9.5281e-01, -7.9438e-01, ..., -1.7321e+00,
-1.8919e+00, -1.6272e+00],
[-2.9771e-01, 2.0540e+00, 4.7571e-01, ..., -3.6083e-01,
-4.3551e-02, 1.7356e-01],
...,
[-1.9908e+00, 7.9527e-02, -1.5917e-01, ..., -1.2981e+00,
-3.5844e-01, 1.6021e-01],
[-1.0808e+00, 8.8614e-01, 9.1043e-01, ..., 2.5933e-02,
3.2487e-01, 5.0196e-01],
[-1.9989e+00, -1.3463e+00, -1.5884e+00, ..., -2.4205e+00,
-1.4171e+00, -8.5790e-01]],
[[-1.5891e+00, -1.6209e+00, -2.9897e+00, ..., -1.2864e+00,
-7.4286e-01, -6.0674e-01],
[-1.0165e+00, -6.8381e-01, -3.9309e+00, ..., -2.0576e+00,
-1.7886e+00, -1.9644e+00],
[-2.5517e+00, -2.1084e+00, -6.4680e+00, ..., -4.3225e+00,
-3.7798e+00, -2.9833e+00],
...,
[-1.4302e+00, -1.7585e-01, -3.5196e+00, ..., -1.5650e+00,
-1.1374e+00, -9.9549e-01],
[-1.6867e+00, -6.9856e-01, -3.1013e+00, ..., -1.7795e+00,
-1.2103e+00, -8.3116e-01],
[-2.2675e+00, -1.4762e+00, -3.0601e+00, ..., -2.9218e+00,
-2.1102e+00, -1.9374e+00]],
[[-1.3088e+00, -1.9160e+00, -1.7583e+00, ..., -1.9865e+00,
-1.6429e+00, -1.2279e+00],
[-1.3765e+00, -1.9108e+00, -1.5319e+00, ..., -1.7635e+00,
-1.4290e+00, -1.1238e+00],
[-1.8117e+00, -2.1996e+00, -1.8554e+00, ..., -2.0899e+00,
-1.6075e+00, -1.1863e+00],
...,
[-2.3433e+00, -2.9041e+00, -2.6967e+00, ..., -2.8535e+00,
-2.2827e+00, -1.7450e+00],
[-1.5284e+00, -1.9819e+00, -1.9194e+00, ..., -1.9499e+00,
-1.6110e+00, -1.2950e+00],
[-1.0039e+00, -1.2806e+00, -1.3248e+00, ..., -1.1925e+00,
-1.0193e+00, -7.6812e-01]]],
[[[ 7.4346e-02, 2.3836e-01, -1.4436e+00, ..., -6.7448e-01,
-2.5346e-01, -4.5873e-02],
[-2.2305e+00, -1.6760e+00, -4.7086e+00, ..., -3.6971e+00,
-1.5112e+00, -5.3337e-01],
[ 6.0817e-01, -1.4135e-01, -9.1973e-01, ..., -2.0728e-01,
-2.9191e-01, 2.4276e-01],
...,
[-3.1422e-01, -1.0586e+00, -8.8430e-01, ..., -1.3079e+00,
-9.0382e-01, -6.8526e-01],
[ 9.4496e-01, 5.8036e-01, 3.7336e-01, ..., 4.3706e-01,
6.9842e-01, 8.0340e-01],
[ 6.6555e-01, 4.3684e-01, 3.3717e-01, ..., 5.4902e-01,
7.7211e-01, 6.9698e-01]],
[[-3.3384e+00, -1.6160e+00, -6.2286e+00, ..., -6.9863e+00,
-6.1617e+00, -4.5785e+00],
[-6.2863e-01, 2.0004e+00, -4.1880e+00, ..., -5.4875e+00,
-5.5742e+00, -4.3177e+00],
[-6.9441e-02, 1.5743e+00, -7.4391e+00, ..., -7.9862e+00,
-7.8599e+00, -6.6271e+00],
...,
[-4.7406e-01, 2.6972e-01, -4.6483e+00, ..., -5.4157e+00,
-4.6814e+00, -3.9281e+00],
[ 1.9427e+00, 2.4239e+00, -1.1023e-01, ..., -1.1735e+00,
-1.2826e+00, -1.4080e+00],
[ 1.1035e+00, 1.2915e+00, -1.7224e+00, ..., -2.4816e+00,
-2.5801e+00, -2.5602e+00]],
[[-1.4165e+00, -2.5397e+00, -2.4642e+00, ..., -1.7262e+00,
-1.2423e+00, -1.3459e+00],
[-1.8705e+00, -2.1892e+00, -1.6204e+00, ..., -1.2060e+00,
-7.2393e-01, -8.1685e-01],
[-2.3815e+00, -2.1017e+00, -1.0505e+00, ..., -1.0908e+00,
-4.7282e-01, -1.9714e-01],
...,
[-6.6881e-01, -1.1175e+00, -9.8071e-01, ..., -6.0086e-01,
-4.7625e-01, -6.3167e-01],
[-1.0609e+00, -7.0866e-01, -5.5920e-01, ..., -2.7058e-01,
-7.3132e-02, -3.9401e-02],
[-1.2135e+00, -6.8384e-01, -3.8412e-01, ..., -3.1744e-01,
-2.7985e-01, -2.4969e-01]],
...,
[[-2.2318e+00, -1.0987e+00, -2.4342e+00, ..., -2.9479e+00,
-2.4341e+00, -1.7237e+00],
[-1.8446e+00, 2.1463e-01, -1.3370e+00, ..., -2.4148e+00,
-2.4303e+00, -1.8636e+00],
[-3.3112e-01, 2.7707e+00, 1.2854e+00, ..., 9.5990e-02,
1.9696e-01, 4.8371e-01],
...,
[-1.4959e+00, 1.0041e+00, 5.1697e-01, ..., -1.1881e+00,
-4.4019e-01, 5.6721e-02],
[-4.9632e-01, 1.4183e+00, 7.2463e-01, ..., -1.6507e-02,
1.8018e-01, 3.0124e-01],
[-1.5659e+00, -1.0203e+00, -1.5173e+00, ..., -2.2897e+00,
-1.3284e+00, -7.0732e-01]],
[[-2.4217e+00, -2.8057e+00, -4.4151e+00, ..., -2.5015e+00,
-2.0327e+00, -1.8356e+00],
[-1.6746e+00, -1.4710e+00, -5.1614e+00, ..., -3.0399e+00,
-2.9097e+00, -2.8243e+00],
[-2.7992e+00, -2.2978e+00, -7.0230e+00, ..., -4.4659e+00,
-4.3916e+00, -3.4952e+00],
...,
[-1.6877e+00, -5.5272e-01, -4.1332e+00, ..., -2.1535e+00,
-1.5888e+00, -1.2191e+00],
[-1.6651e+00, -7.2024e-01, -3.3731e+00, ..., -2.1135e+00,
-1.3769e+00, -8.0807e-01],
[-2.0322e+00, -1.2589e+00, -2.8349e+00, ..., -2.7182e+00,
-1.8127e+00, -1.5390e+00]],
[[-1.8809e+00, -2.6390e+00, -2.4986e+00, ..., -2.6959e+00,
-2.2039e+00, -1.6937e+00],
[-1.8692e+00, -2.6092e+00, -2.1469e+00, ..., -2.4497e+00,
-1.9337e+00, -1.5645e+00],
[-2.3474e+00, -2.9184e+00, -2.5155e+00, ..., -2.8129e+00,
-2.1988e+00, -1.6808e+00],
...,
[-2.2805e+00, -2.8564e+00, -2.7533e+00, ..., -2.8664e+00,
-2.2889e+00, -1.7233e+00],
[-1.5453e+00, -1.9887e+00, -1.9426e+00, ..., -2.0370e+00,
-1.6707e+00, -1.3311e+00],
[-1.0543e+00, -1.2797e+00, -1.3499e+00, ..., -1.2728e+00,
-1.0811e+00, -7.9643e-01]]],
[[[ 2.8298e-01, 5.5201e-01, 6.8268e-01, ..., 4.6876e-01,
9.9254e-01, 5.9203e-01],
[-2.1774e+00, -1.5054e+00, -3.1959e+00, ..., -2.6874e+00,
-9.0574e-01, -2.0798e-01],
[-2.0891e-01, -1.0371e+00, -2.5407e-01, ..., -1.6831e-01,
5.0867e-01, 6.4272e-01],
...,
[-1.7347e-01, -1.1169e+00, -2.6274e+00, ..., -1.5455e+00,
-1.3641e+00, -8.6327e-01],
[ 1.4403e+00, 8.6644e-01, -9.3214e-01, ..., -3.2592e-01,
1.4197e-01, 3.7850e-01],
[ 8.7682e-01, 3.3231e-01, -6.5683e-01, ..., -2.1593e-01,
2.4270e-01, 3.7469e-01]],
[[-1.6400e+00, -1.1115e+00, -4.1461e+00, ..., -4.8885e+00,
-4.4276e+00, -3.3982e+00],
[ 6.2033e-01, 1.9491e+00, -2.2740e+00, ..., -3.5149e+00,
-3.5825e+00, -2.9767e+00],
[ 1.3589e+00, 1.7080e+00, -4.9274e+00, ..., -5.4516e+00,
-5.2459e+00, -4.9303e+00],
...,
[-1.3001e+00, 1.3089e-01, -6.6680e+00, ..., -7.5815e+00,
-6.9113e+00, -5.3950e+00],
[ 1.7827e+00, 2.8121e+00, -1.0156e+00, ..., -2.5799e+00,
-2.9453e+00, -2.3578e+00],
[ 1.1958e+00, 1.8003e+00, -2.2673e+00, ..., -3.2588e+00,
-3.6274e+00, -3.1319e+00]],
[[-1.0213e-01, -1.1381e+00, -1.2990e+00, ..., -4.9632e-01,
-1.5917e-01, -4.7795e-01],
[-5.8253e-01, -8.4091e-01, -7.5892e-01, ..., -2.6512e-01,
3.6875e-02, -7.1448e-02],
[-5.4787e-01, -3.5947e-01, -4.1552e-01, ..., -3.5121e-01,
8.4867e-02, 7.9448e-02],
...,
[-1.3439e+00, -1.7298e+00, -7.8553e-01, ..., -8.9801e-01,
-7.6896e-01, -1.0234e+00],
[-1.6991e+00, -1.1296e+00, -8.2972e-01, ..., -5.7017e-01,
-2.9514e-01, -3.6310e-01],
[-1.8446e+00, -1.2580e+00, -8.8457e-01, ..., -9.5911e-01,
-8.1254e-01, -7.8556e-01]],
...,
[[-3.0213e+00, -2.4768e+00, -2.7557e+00, ..., -3.4540e+00,
-2.6142e+00, -1.7529e+00],
[-2.3004e+00, -8.6543e-01, -1.6037e+00, ..., -2.6235e+00,
-2.3222e+00, -1.8338e+00],
[-1.0456e+00, 1.1873e+00, 6.4053e-01, ..., -4.5097e-01,
1.6662e-02, 1.5089e-01],
...,
[-1.8428e+00, 1.2398e+00, 2.4769e-02, ..., -1.0469e+00,
-5.5618e-01, -9.0018e-03],
[-8.0902e-01, 1.8926e+00, 1.6232e+00, ..., 5.1443e-01,
4.0008e-01, 6.7470e-01],
[-2.2650e+00, -1.1891e+00, -1.0211e+00, ..., -1.8427e+00,
-9.9000e-01, -3.2780e-01]],
[[-1.6020e+00, -1.5982e+00, -2.9356e+00, ..., -1.2862e+00,
-8.2909e-01, -8.4031e-01],
[-6.1363e-01, 7.2932e-02, -3.2483e+00, ..., -1.3816e+00,
-1.2013e+00, -1.5732e+00],
[-2.1001e+00, -8.8684e-01, -5.3272e+00, ..., -3.1870e+00,
-2.6922e+00, -2.1975e+00],
...,
[-2.4597e+00, -1.5454e+00, -5.4311e+00, ..., -3.0662e+00,
-2.6896e+00, -2.1675e+00],
[-2.2154e+00, -1.4523e+00, -4.2370e+00, ..., -2.6603e+00,
-2.2694e+00, -1.6187e+00],
[-2.3947e+00, -1.7737e+00, -3.3067e+00, ..., -2.9452e+00,
-2.4065e+00, -2.1004e+00]],
[[-1.3585e+00, -1.9168e+00, -1.7114e+00, ..., -1.7772e+00,
-1.4659e+00, -1.1183e+00],
[-1.2812e+00, -1.7734e+00, -1.2457e+00, ..., -1.3856e+00,
-1.0876e+00, -8.6862e-01],
[-1.6153e+00, -1.9701e+00, -1.5001e+00, ..., -1.6163e+00,
-1.2040e+00, -8.9296e-01],
...,
[-3.1634e+00, -3.9488e+00, -3.7442e+00, ..., -3.9540e+00,
-3.1712e+00, -2.4495e+00],
[-2.0990e+00, -2.7113e+00, -2.6855e+00, ..., -2.7767e+00,
-2.2893e+00, -1.8662e+00],
[-1.5238e+00, -1.8522e+00, -1.9148e+00, ..., -1.8751e+00,
-1.5231e+00, -1.1685e+00]]],
[[[ 2.0855e-02, 1.6493e-01, -1.4952e+00, ..., -6.9312e-01,
-2.6887e-01, -8.7372e-02],
[-2.1484e+00, -1.6538e+00, -4.6038e+00, ..., -3.9048e+00,
-1.6324e+00, -6.0646e-01],
[ 5.2259e-01, -1.9882e-01, -9.5089e-01, ..., -2.4849e-01,
-3.4615e-01, 2.1152e-01],
...,
[-3.2249e-01, -3.3008e+00, -6.5741e+00, ..., -1.2486e+00,
-8.0620e-01, -5.6385e-01],
[ 1.1239e+00, -4.3543e-01, -2.6522e+00, ..., 4.6106e-01,
7.5262e-01, 8.5335e-01],
[ 7.9467e-01, 5.9686e-01, -5.0694e-02, ..., 5.7730e-01,
7.9377e-01, 7.3124e-01]],
[[-3.3116e+00, -1.7861e+00, -6.3229e+00, ..., -7.2266e+00,
-6.3208e+00, -4.6777e+00],
[-7.9940e-01, 1.5417e+00, -4.4661e+00, ..., -5.5689e+00,
-5.6219e+00, -4.3232e+00],
[-4.0229e-01, 1.0567e+00, -7.5755e+00, ..., -8.0963e+00,
-7.9304e+00, -6.7296e+00],
...,
[-2.9658e-01, 1.3666e+00, -3.6514e+00, ..., -5.4420e+00,
-4.7525e+00, -3.9635e+00],
[ 2.3682e+00, 3.2668e+00, 2.2955e-01, ..., -1.3073e+00,
-1.4260e+00, -1.5094e+00],
[ 1.3907e+00, 1.5728e+00, -1.8561e+00, ..., -2.5728e+00,
-2.6636e+00, -2.6054e+00]],
[[-1.4039e+00, -2.3972e+00, -2.3354e+00, ..., -1.7939e+00,
-1.2772e+00, -1.4137e+00],
[-1.7472e+00, -1.9958e+00, -1.5026e+00, ..., -1.2762e+00,
-7.6404e-01, -8.6205e-01],
[-2.2662e+00, -1.9966e+00, -1.0205e+00, ..., -1.1536e+00,
-4.9567e-01, -2.0299e-01],
...,
[-8.2509e-01, -1.3419e+00, -2.6806e-01, ..., -5.8461e-01,
-4.5545e-01, -6.0403e-01],
[-1.2565e+00, -1.0507e+00, -9.4566e-01, ..., -2.6399e-01,
-7.1591e-02, -4.5176e-02],
[-1.4264e+00, -1.0205e+00, -1.1919e+00, ..., -3.3594e-01,
-2.8341e-01, -2.3725e-01]],
...,
[[-2.0088e+00, -1.0437e+00, -2.2814e+00, ..., -3.1882e+00,
-2.6158e+00, -1.8399e+00],
[-1.8173e+00, -3.1962e-02, -1.4465e+00, ..., -2.5650e+00,
-2.5594e+00, -1.9711e+00],
[-3.6509e-01, 2.3946e+00, 1.0488e+00, ..., 1.1366e-01,
2.5816e-01, 5.1650e-01],
...,
[-1.6618e+00, 3.5016e-01, -3.3478e+00, ..., -1.1666e+00,
-4.5811e-01, 7.3082e-02],
[-5.4198e-01, 1.1631e+00, -5.2154e-01, ..., -3.7793e-02,
1.5137e-01, 2.9146e-01],
[-1.6637e+00, -8.7520e-01, -1.0867e+00, ..., -2.2195e+00,
-1.2610e+00, -6.4388e-01]],
[[-2.4674e+00, -2.9159e+00, -4.2857e+00, ..., -2.4577e+00,
-1.9725e+00, -1.7824e+00],
[-1.8279e+00, -1.7399e+00, -4.9712e+00, ..., -2.9783e+00,
-2.8424e+00, -2.8244e+00],
[-2.8725e+00, -2.5204e+00, -6.6659e+00, ..., -4.4913e+00,
-4.4208e+00, -3.5354e+00],
...,
[-1.6074e+00, -1.6346e-02, -4.2786e+00, ..., -2.2010e+00,
-1.6600e+00, -1.2762e+00],
[-1.6216e+00, -5.0384e-01, -4.2290e+00, ..., -2.1106e+00,
-1.4208e+00, -8.6982e-01],
[-2.0821e+00, -1.2358e+00, -3.2695e+00, ..., -2.6714e+00,
-1.8259e+00, -1.5411e+00]],
[[-1.8538e+00, -2.5945e+00, -2.5113e+00, ..., -2.7449e+00,
-2.2500e+00, -1.7292e+00],
[-1.8481e+00, -2.5773e+00, -2.2161e+00, ..., -2.4591e+00,
-1.9395e+00, -1.5734e+00],
[-2.3136e+00, -2.9082e+00, -2.6130e+00, ..., -2.8120e+00,
-2.1918e+00, -1.6679e+00],
...,
[-2.3974e+00, -3.2718e+00, -3.3713e+00, ..., -2.8628e+00,
-2.2874e+00, -1.7352e+00],
[-1.6206e+00, -2.2146e+00, -2.3683e+00, ..., -2.0560e+00,
-1.6920e+00, -1.3484e+00],
[-1.0915e+00, -1.3673e+00, -1.5143e+00, ..., -1.2939e+00,
-1.1010e+00, -8.1813e-01]]]], device='cuda:0'),
tensor([[[[ 3.4269e-01, 8.1387e-01, 7.3244e-01, ..., 6.9461e-01,
9.4918e-01, 7.5342e-01],
[-1.2479e+00, -3.0296e-01, -1.3964e+00, ..., -1.0649e+00,
6.8619e-02, 5.8431e-01],
[ 6.4607e-01, 4.5568e-01, 5.3880e-01, ..., 6.0466e-01,
9.4822e-01, 8.9587e-01],
...,
[-1.9199e-01, -1.1413e+00, -2.6254e+00, ..., -1.5343e+00,
-1.4297e+00, -8.7191e-01],
[ 1.4273e+00, 8.6662e-01, -9.2733e-01, ..., -2.8051e-01,
9.8403e-02, 3.7686e-01],
[ 8.4302e-01, 3.1475e-01, -6.8394e-01, ..., -1.7370e-01,
2.0988e-01, 3.8011e-01]],
[[-2.3656e+00, -1.3024e+00, -4.2874e+00, ..., -5.2303e+00,
-4.4729e+00, -3.5255e+00],
[-1.6005e-01, 1.6662e+00, -2.5649e+00, ..., -4.0951e+00,
-3.9751e+00, -3.2808e+00],
[ 7.0016e-01, 1.5230e+00, -5.0913e+00, ..., -5.9602e+00,
-5.7399e+00, -5.3001e+00],
...,
[-1.2990e+00, 1.3047e-01, -6.7043e+00, ..., -7.6120e+00,
-6.8918e+00, -5.4201e+00],
[ 1.7945e+00, 2.8212e+00, -1.0166e+00, ..., -2.5817e+00,
-2.9318e+00, -2.3737e+00],
[ 1.2097e+00, 1.8154e+00, -2.2585e+00, ..., -3.2496e+00,
-3.6129e+00, -3.1413e+00]],
[[-6.7262e-01, -1.7985e+00, -1.6898e+00, ..., -9.0571e-01,
-6.5178e-01, -7.6842e-01],
[-1.3578e+00, -1.8201e+00, -1.2576e+00, ..., -7.1971e-01,
-3.8515e-01, -2.8423e-01],
[-1.0405e+00, -9.8950e-01, -4.6593e-01, ..., -3.2445e-01,
3.9113e-02, 5.0649e-02],
...,
[-1.3484e+00, -1.7594e+00, -7.8949e-01, ..., -8.9329e-01,
-7.7919e-01, -1.0251e+00],
[-1.7164e+00, -1.1569e+00, -8.4663e-01, ..., -5.6861e-01,
-3.0591e-01, -3.7091e-01],
[-1.8874e+00, -1.2816e+00, -8.9496e-01, ..., -9.5300e-01,
-8.2384e-01, -7.9951e-01]],
...,
[[-2.2546e+00, -1.3146e+00, -2.3271e+00, ..., -2.8717e+00,
-2.2618e+00, -1.5005e+00],
[-6.4305e-01, 1.0543e+00, -4.2607e-01, ..., -1.3413e+00,
-1.3860e+00, -1.1793e+00],
[-3.9011e-01, 2.0596e+00, 7.9223e-01, ..., -1.8000e-01,
2.1427e-01, 3.5166e-01],
...,
[-1.8689e+00, 1.2182e+00, -4.5262e-03, ..., -1.0601e+00,
-5.2051e-01, -2.0941e-03],
[-8.2489e-01, 1.8947e+00, 1.6459e+00, ..., 4.9142e-01,
4.3223e-01, 6.8343e-01],
[-2.2130e+00, -1.1833e+00, -1.0169e+00, ..., -1.8647e+00,
-9.7504e-01, -3.0226e-01]],
[[-1.3601e+00, -1.2344e+00, -2.6887e+00, ..., -1.1000e+00,
-6.0196e-01, -4.6849e-01],
[-6.4632e-01, -1.1774e-01, -3.5419e+00, ..., -1.7719e+00,
-1.5249e+00, -1.6922e+00],
[-2.0879e+00, -1.3525e+00, -5.9247e+00, ..., -3.7085e+00,
-3.1983e+00, -2.5501e+00],
...,
[-2.4406e+00, -1.5880e+00, -5.5064e+00, ..., -3.0609e+00,
-2.7167e+00, -2.1716e+00],
[-2.1903e+00, -1.4654e+00, -4.2892e+00, ..., -2.6549e+00,
-2.2989e+00, -1.6309e+00],
[-2.3716e+00, -1.7720e+00, -3.3317e+00, ..., -2.9533e+00,
-2.4336e+00, -2.1185e+00]],
[[-1.1576e+00, -1.7138e+00, -1.5021e+00, ..., -1.6305e+00,
-1.3576e+00, -1.0230e+00],
[-1.1642e+00, -1.6306e+00, -1.1630e+00, ..., -1.3646e+00,
-1.1075e+00, -8.7858e-01],
[-1.5098e+00, -1.7952e+00, -1.3325e+00, ..., -1.5529e+00,
-1.1839e+00, -8.6376e-01],
...,
[-3.1789e+00, -3.9676e+00, -3.7529e+00, ..., -3.9457e+00,
-3.1576e+00, -2.4356e+00],
[-2.1184e+00, -2.7268e+00, -2.6899e+00, ..., -2.7697e+00,
-2.2849e+00, -1.8587e+00],
[-1.5450e+00, -1.8644e+00, -1.9198e+00, ..., -1.8681e+00,
-1.5183e+00, -1.1628e+00]]],
[[[ 2.9367e-01, 5.5093e-01, 6.7634e-01, ..., 4.7210e-01,
1.0132e+00, 6.0608e-01],
[-2.1954e+00, -1.5502e+00, -3.2590e+00, ..., -2.6836e+00,
-8.8270e-01, -1.9830e-01],
[-1.7821e-01, -1.0265e+00, -2.5042e-01, ..., -1.5863e-01,
5.3825e-01, 6.4732e-01],
...,
[ 1.1275e-01, -5.6315e-01, -1.3183e+00, ..., -1.1786e+00,
-9.6323e-01, -7.2430e-01],
[ 1.4425e+00, 1.2423e+00, 7.3012e-01, ..., 7.7226e-01,
9.0790e-01, 9.4678e-01],
[ 1.0853e+00, 9.9190e-01, 7.7230e-01, ..., 9.4299e-01,
1.0286e+00, 8.8364e-01]],
[[-1.6895e+00, -1.1573e+00, -4.1859e+00, ..., -4.8614e+00,
-4.4449e+00, -3.3950e+00],
[ 6.0046e-01, 1.9572e+00, -2.2490e+00, ..., -3.5050e+00,
-3.6250e+00, -2.9720e+00],
[ 1.3474e+00, 1.7204e+00, -4.9065e+00, ..., -5.4573e+00,
-5.2802e+00, -4.9226e+00],
...,
[-8.5998e-01, 1.9241e-01, -4.9826e+00, ..., -5.8492e+00,
-5.1505e+00, -4.2181e+00],
[ 1.7054e+00, 2.3632e+00, -4.3088e-01, ..., -1.2482e+00,
-1.4690e+00, -1.5319e+00],
[ 9.7100e-01, 1.2469e+00, -1.9093e+00, ..., -2.6083e+00,
-2.7999e+00, -2.7397e+00]],
[[-1.2051e-01, -1.1550e+00, -1.3227e+00, ..., -4.8249e-01,
-1.4825e-01, -4.7178e-01],
[-5.8905e-01, -8.5089e-01, -7.8176e-01, ..., -2.7973e-01,
3.8164e-02, -7.1924e-02],
[-5.3581e-01, -3.4019e-01, -4.0008e-01, ..., -3.5253e-01,
8.0488e-02, 7.4884e-02],
...,
[-9.2464e-01, -1.3881e+00, -6.2313e-01, ..., -6.3572e-01,
-4.9799e-01, -6.7162e-01],
[-1.1389e+00, -7.8000e-01, -4.4321e-01, ..., -9.2575e-02,
1.1109e-01, 9.7346e-02],
[-1.1814e+00, -5.8589e-01, -1.1694e-01, ..., -3.9281e-02,
-2.0988e-02, -4.7822e-02]],
...,
[[-3.0593e+00, -2.5334e+00, -2.8379e+00, ..., -3.4783e+00,
-2.6474e+00, -1.7663e+00],
[-2.3120e+00, -8.7514e-01, -1.6273e+00, ..., -2.6574e+00,
-2.3649e+00, -1.8577e+00],
[-1.0960e+00, 1.1421e+00, 5.8848e-01, ..., -4.6975e-01,
-2.1346e-02, 1.3114e-01],
...,
[-1.4421e+00, 1.4103e+00, 2.2178e-01, ..., -9.1332e-01,
-1.4019e-01, 4.3103e-01],
[-5.9839e-01, 1.5508e+00, 9.3861e-01, ..., 1.6711e-01,
3.8410e-01, 5.4165e-01],
[-1.6119e+00, -9.8248e-01, -1.3912e+00, ..., -2.2963e+00,
-1.2555e+00, -5.8519e-01]],
[[-1.6034e+00, -1.5872e+00, -2.8973e+00, ..., -1.2801e+00,
-8.4422e-01, -8.4890e-01],
[-5.9989e-01, 1.0017e-01, -3.2087e+00, ..., -1.3648e+00,
-1.2157e+00, -1.5687e+00],
[-2.1110e+00, -9.0130e-01, -5.3394e+00, ..., -3.1746e+00,
-2.6995e+00, -2.1910e+00],
...,
[-1.8569e+00, -9.8129e-01, -4.5925e+00, ..., -2.3631e+00,
-1.9358e+00, -1.5282e+00],
[-1.7134e+00, -1.0254e+00, -3.7287e+00, ..., -2.2357e+00,
-1.6340e+00, -1.0210e+00],
[-2.0034e+00, -1.4574e+00, -3.0309e+00, ..., -2.8210e+00,
-1.9995e+00, -1.7020e+00]],
[[-1.3564e+00, -1.9217e+00, -1.7200e+00, ..., -1.7702e+00,
-1.4645e+00, -1.1132e+00],
[-1.2778e+00, -1.7716e+00, -1.2384e+00, ..., -1.3720e+00,
-1.0840e+00, -8.6281e-01],
[-1.5986e+00, -1.9526e+00, -1.4838e+00, ..., -1.5971e+00,
-1.1980e+00, -8.8471e-01],
...,
[-2.3669e+00, -2.9625e+00, -2.7711e+00, ..., -3.0755e+00,
-2.4445e+00, -1.8663e+00],
[-1.6103e+00, -2.0381e+00, -1.9847e+00, ..., -2.1347e+00,
-1.7441e+00, -1.4011e+00],
[-1.0882e+00, -1.3064e+00, -1.3523e+00, ..., -1.2953e+00,
-1.1076e+00, -8.1703e-01]]],
[[[ 4.7734e-01, 9.5300e-01, 8.6381e-01, ..., -3.8603e-01,
-8.1560e-02, -3.2018e-02],
[-1.2450e+00, -3.3705e-01, -1.4913e+00, ..., -3.4441e+00,
-1.3849e+00, -5.2226e-01],
[ 5.8748e-01, 4.3724e-01, 3.7510e-01, ..., -2.0348e-01,
-3.4079e-01, 1.2108e-01],
...,
[-1.4916e-01, -8.3219e-01, -1.5098e+00, ..., -1.7274e+00,
-1.7057e+00, -1.1038e+00],
[ 1.0876e+00, 7.9497e-01, 3.0283e-01, ..., 4.3890e-02,
2.7987e-01, 4.5864e-01],
[ 7.1219e-01, 5.0301e-01, 1.8952e-01, ..., 2.1887e-01,
4.3668e-01, 4.9237e-01]],
[[-2.3004e+00, -1.0905e+00, -4.0925e+00, ..., -7.0928e+00,
-6.2567e+00, -4.5869e+00],
[-2.8328e-01, 1.7307e+00, -2.5022e+00, ..., -5.5365e+00,
-5.5963e+00, -4.2479e+00],
[ 4.2890e-01, 1.5289e+00, -5.0547e+00, ..., -8.0186e+00,
-7.9004e+00, -6.6246e+00],
...,
[-4.5416e-01, 6.4154e-01, -4.0858e+00, ..., -7.1142e+00,
-6.7154e+00, -5.2104e+00],
[ 2.0665e+00, 2.7602e+00, 2.9396e-01, ..., -2.2589e+00,
-2.8276e+00, -2.2322e+00],
[ 1.2103e+00, 1.5098e+00, -1.5271e+00, ..., -2.9965e+00,
-3.4982e+00, -3.0244e+00]],
[[-5.7641e-01, -1.5732e+00, -1.4043e+00, ..., -1.5102e+00,
-1.1030e+00, -1.4070e+00],
[-1.2886e+00, -1.6375e+00, -1.0069e+00, ..., -1.0639e+00,
-6.5390e-01, -9.4646e-01],
[-1.1368e+00, -9.8915e-01, -3.8795e-01, ..., -1.1260e+00,
-5.6075e-01, -4.8148e-01],
...,
[-9.3181e-01, -1.3368e+00, -5.1568e-01, ..., -9.7734e-01,
-8.4896e-01, -1.2138e+00],
[-1.2912e+00, -8.7173e-01, -5.3593e-01, ..., -5.3963e-01,
-2.5905e-01, -4.3103e-01],
[-1.3649e+00, -7.6222e-01, -3.3759e-01, ..., -4.8643e-01,
-4.4075e-01, -6.0543e-01]],
...,
[[-2.1937e+00, -1.2430e+00, -2.5510e+00, ..., -3.2685e+00,
-2.6265e+00, -1.9108e+00],
[-6.9381e-01, 9.5583e-01, -9.5977e-01, ..., -2.4611e+00,
-2.4979e+00, -2.0420e+00],
[-2.2197e-01, 2.2345e+00, 4.8092e-01, ..., -3.0147e-02,
1.7161e-01, 4.4336e-01],
...,
[-1.4910e+00, 1.2918e+00, -1.5004e-01, ..., -8.5905e-01,
-2.3831e-01, 2.6116e-01],
[-5.1316e-01, 1.5755e+00, 8.0333e-01, ..., 3.8507e-01,
4.3931e-01, 7.6648e-01],
[-1.5820e+00, -1.0244e+00, -1.5564e+00, ..., -2.0388e+00,
-1.0709e+00, -3.0695e-01]],
[[-1.2825e+00, -1.2008e+00, -2.6067e+00, ..., -2.1586e+00,
-1.7799e+00, -1.6985e+00],
[-5.9646e-01, -9.8751e-02, -3.3552e+00, ..., -2.7976e+00,
-2.7594e+00, -2.8401e+00],
[-2.0659e+00, -1.3798e+00, -5.7305e+00, ..., -4.5480e+00,
-4.5214e+00, -3.6933e+00],
...,
[-1.6427e+00, -4.9586e-01, -4.1530e+00, ..., -2.9619e+00,
-2.8485e+00, -2.3675e+00],
[-1.6493e+00, -7.0816e-01, -3.4378e+00, ..., -2.5219e+00,
-2.3330e+00, -1.7349e+00],
[-2.0827e+00, -1.3361e+00, -2.9653e+00, ..., -2.8066e+00,
-2.4068e+00, -2.1667e+00]],
[[-1.1473e+00, -1.7050e+00, -1.5163e+00, ..., -2.6594e+00,
-2.2187e+00, -1.7280e+00],
[-1.1538e+00, -1.6203e+00, -1.1973e+00, ..., -2.4588e+00,
-1.9833e+00, -1.6273e+00],
[-1.4997e+00, -1.7992e+00, -1.4003e+00, ..., -2.8590e+00,
-2.2776e+00, -1.7792e+00],
...,
[-2.3171e+00, -2.8846e+00, -2.6632e+00, ..., -3.9103e+00,
-3.1767e+00, -2.4943e+00],
[-1.5593e+00, -1.9781e+00, -1.9150e+00, ..., -2.7138e+00,
-2.2767e+00, -1.8811e+00],
[-1.0464e+00, -1.2551e+00, -1.3110e+00, ..., -1.8235e+00,
-1.5068e+00, -1.1723e+00]]],
[[[ 2.8551e-01, 5.2266e-01, -2.7611e-01, ..., 6.3135e-01,
1.0595e+00, 7.4052e-01],
[-1.6970e+00, -1.0975e+00, -3.2584e+00, ..., -1.9082e+00,
-3.3321e-01, 2.6464e-01],
[ 7.2946e-01, 3.6116e-01, -4.3729e-01, ..., 1.0664e-01,
7.2240e-01, 7.7324e-01],
...,
[-3.9671e+00, -7.1318e+00, -1.8223e+01, ..., -1.4790e+00,
-1.0415e+00, -7.1691e-01],
[-3.3479e+00, -4.4138e+00, -1.6316e+01, ..., 1.1252e-01,
6.7627e-01, 7.9730e-01],
[-5.4415e+00, -7.0818e+00, -1.2925e+01, ..., 1.1694e-01,
5.5086e-01, 4.9005e-01]],
[[-3.1205e+00, -1.9277e+00, -5.3788e+00, ..., -5.0459e+00,
-4.4744e+00, -3.4079e+00],
[-7.1009e-01, 1.4574e+00, -3.2608e+00, ..., -3.7062e+00,
-3.7111e+00, -3.0174e+00],
[-3.7547e-02, 1.2661e+00, -6.1401e+00, ..., -5.6874e+00,
-5.4657e+00, -5.0773e+00],
...,
[-6.8634e+00, -6.5148e+00, -1.2048e+01, ..., -4.9213e+00,
-4.3712e+00, -3.5544e+00],
[-4.2182e+00, -3.4065e+00, -4.0857e+00, ..., -1.0680e+00,
-1.2959e+00, -1.2975e+00],
[-2.1817e+00, -1.9201e+00, -3.3765e+00, ..., -2.3779e+00,
-2.5368e+00, -2.4102e+00]],
[[-7.9649e-01, -1.9165e+00, -1.9217e+00, ..., -6.9462e-01,
-3.2811e-01, -5.4190e-01],
[-1.4102e+00, -1.7817e+00, -1.2492e+00, ..., -5.0820e-01,
-1.0193e-01, -6.4875e-02],
[-1.5881e+00, -1.5069e+00, -8.5767e-01, ..., -2.9064e-01,
1.8769e-01, 2.2148e-01],
...,
[-2.4067e+00, -1.3568e+00, -9.3784e-02, ..., -5.0328e-01,
-2.2343e-01, -3.5043e-01],
[-3.6740e+00, -2.8693e+00, -3.5438e+00, ..., -2.5071e-01,
1.2913e-01, 2.0667e-01],
[-4.4184e+00, -3.9568e+00, -4.6215e+00, ..., -3.7243e-01,
-2.1350e-01, -1.7219e-01]],
...,
[[-2.2326e+00, -1.3124e+00, -2.5591e+00, ..., -3.4816e+00,
-2.7908e+00, -1.8185e+00],
[-9.4002e-01, 6.8964e-01, -9.4961e-01, ..., -2.1807e+00,
-2.1410e+00, -1.6658e+00],
[-3.5341e-01, 2.2467e+00, 9.3907e-01, ..., -4.1802e-01,
-2.3275e-02, 2.2663e-01],
...,
[-4.4101e+00, -7.5122e-01, -9.2933e-01, ..., -1.4714e+00,
-6.3080e-01, -3.8056e-02],
[-1.1690e+01, -7.8769e+00, -5.9377e+00, ..., -1.6386e-01,
8.2338e-02, 3.0804e-01],
[-8.3215e+00, -7.2845e+00, -5.9277e+00, ..., -2.2732e+00,
-1.3462e+00, -7.9661e-01]],
[[-1.7345e+00, -1.9748e+00, -3.6030e+00, ..., -9.7603e-01,
-5.4957e-01, -5.4251e-01],
[-1.0118e+00, -8.2976e-01, -4.4568e+00, ..., -1.4168e+00,
-1.2312e+00, -1.5586e+00],
[-2.4304e+00, -2.0244e+00, -6.7029e+00, ..., -3.4503e+00,
-2.9205e+00, -2.3039e+00],
...,
[-5.2780e+00, -8.1059e+00, -1.2735e+01, ..., -1.8458e+00,
-1.2894e+00, -9.5957e-01],
[-1.7781e+00, -2.1316e+00, -6.4696e+00, ..., -1.9503e+00,
-1.2479e+00, -7.3649e-01],
[-3.6751e+00, -2.4640e+00, -4.2358e+00, ..., -2.7622e+00,
-1.9071e+00, -1.6623e+00]],
[[-1.4974e+00, -2.1951e+00, -2.1395e+00, ..., -1.7567e+00,
-1.4559e+00, -1.1013e+00],
[-1.5053e+00, -2.1530e+00, -1.8476e+00, ..., -1.4082e+00,
-1.1186e+00, -8.8708e-01],
[-1.9118e+00, -2.4025e+00, -2.1865e+00, ..., -1.6106e+00,
-1.2140e+00, -8.9584e-01],
...,
[-4.5051e+00, -5.9903e+00, -7.5819e+00, ..., -2.8378e+00,
-2.2723e+00, -1.7428e+00],
[-3.2420e+00, -4.2483e+00, -5.1334e+00, ..., -1.9726e+00,
-1.6275e+00, -1.3057e+00],
[-2.3603e+00, -3.1364e+00, -4.3276e+00, ..., -1.2218e+00,
-1.0345e+00, -7.8036e-01]]],
[[[ 2.2895e-01, 4.8303e-01, 6.3580e-01, ..., -3.3219e-01,
-4.7750e-02, 3.1415e-02],
[-1.9758e+00, -1.3278e+00, -2.8114e+00, ..., -3.3890e+00,
-1.3599e+00, -4.4953e-01],
[-1.4805e-01, -8.9464e-01, 1.2783e-02, ..., -1.8904e-01,
-3.6620e-01, 1.6671e-01],
...,
[-9.5161e-01, -2.1471e+00, -2.0621e+00, ..., -1.6746e+00,
-1.6235e+00, -1.0303e+00],
[ 8.0901e-01, -1.1512e-01, -1.0842e-01, ..., 6.3363e-02,
2.9939e-01, 4.8150e-01],
[ 2.1050e-01, -5.7808e-01, -4.0619e-01, ..., 1.9954e-01,
4.2321e-01, 4.9562e-01]],
[[-1.6601e+00, -1.3873e+00, -4.2715e+00, ..., -7.0378e+00,
-6.2065e+00, -4.6168e+00],
[ 4.5723e-01, 1.4036e+00, -2.5500e+00, ..., -5.4643e+00,
-5.5841e+00, -4.3163e+00],
[ 1.1714e+00, 1.1357e+00, -5.0271e+00, ..., -7.9578e+00,
-7.9106e+00, -6.7234e+00],
...,
[ 3.8930e-01, 3.5306e-01, -4.1309e+00, ..., -7.1558e+00,
-6.7525e+00, -5.2748e+00],
[ 2.5556e+00, 2.5017e+00, 1.1761e-01, ..., -2.4263e+00,
-2.9548e+00, -2.3695e+00],
[ 1.5133e+00, 1.3264e+00, -1.6190e+00, ..., -3.0902e+00,
-3.5545e+00, -3.0714e+00]],
[[-1.0985e-01, -1.1288e+00, -1.3725e+00, ..., -1.4722e+00,
-1.0811e+00, -1.3912e+00],
[-4.8428e-01, -8.0794e-01, -8.7964e-01, ..., -1.0459e+00,
-6.3574e-01, -9.1460e-01],
[-3.3056e-01, -2.3051e-01, -4.4961e-01, ..., -1.1140e+00,
-5.7622e-01, -4.5818e-01],
...,
[-4.9135e-02, -5.9097e-01, -6.0629e-01, ..., -9.7882e-01,
-8.2893e-01, -1.1731e+00],
[-6.8981e-01, -2.0289e-01, -5.5580e-01, ..., -5.4914e-01,
-2.7291e-01, -4.4546e-01],
[-8.6350e-01, -2.5672e-01, -4.7688e-01, ..., -4.9405e-01,
-4.4211e-01, -5.9412e-01]],
...,
[[-2.8093e+00, -2.4365e+00, -2.4307e+00, ..., -3.3571e+00,
-2.7176e+00, -1.9402e+00],
[-2.1355e+00, -8.8974e-01, -1.1080e+00, ..., -2.5659e+00,
-2.6197e+00, -2.1019e+00],
[-1.1478e+00, 7.3541e-01, 7.4373e-01, ..., -1.0157e-01,
6.0426e-02, 3.4387e-01],
...,
[-2.3131e+00, -2.0800e-01, -1.4470e-01, ..., -8.3896e-01,
-2.3913e-01, 2.2750e-01],
[-1.2803e+00, 8.3506e-01, 1.0652e+00, ..., 3.4862e-01,
4.3449e-01, 7.4086e-01],
[-2.2198e+00, -1.5345e+00, -1.6928e+00, ..., -1.9514e+00,
-1.0103e+00, -2.6417e-01]],
[[-1.6521e+00, -1.6619e+00, -2.8130e+00, ..., -2.1282e+00,
-1.8100e+00, -1.7420e+00],
[-7.3965e-01, -1.3483e-01, -3.1168e+00, ..., -2.7769e+00,
-2.8066e+00, -2.9018e+00],
[-2.0921e+00, -9.9525e-01, -5.0594e+00, ..., -4.4878e+00,
-4.5471e+00, -3.7685e+00],
...,
[-1.3493e+00, 1.6911e-01, -3.3758e+00, ..., -3.0504e+00,
-2.9241e+00, -2.4473e+00],
[-1.6846e+00, -4.8386e-01, -3.0220e+00, ..., -2.5327e+00,
-2.3834e+00, -1.8068e+00],
[-2.3350e+00, -1.3798e+00, -3.0489e+00, ..., -2.7568e+00,
-2.4064e+00, -2.1828e+00]],
[[-1.2953e+00, -1.8214e+00, -1.6517e+00, ..., -2.6341e+00,
-2.1987e+00, -1.6995e+00],
[-1.2373e+00, -1.6986e+00, -1.2317e+00, ..., -2.4106e+00,
-1.9426e+00, -1.5929e+00],
[-1.5566e+00, -1.8974e+00, -1.4802e+00, ..., -2.8091e+00,
-2.2373e+00, -1.7364e+00],
...,
[-2.4475e+00, -3.0566e+00, -2.7856e+00, ..., -3.9037e+00,
-3.1701e+00, -2.4858e+00],
[-1.5827e+00, -2.0845e+00, -1.9866e+00, ..., -2.7328e+00,
-2.2863e+00, -1.8806e+00],
[-1.0335e+00, -1.3432e+00, -1.3619e+00, ..., -1.8440e+00,
-1.5229e+00, -1.1814e+00]]]], device='cuda:0'),
tensor([[[[ 2.8090e-01, 5.4916e-01, 6.9579e-01, ..., 4.6283e-01,
1.0085e+00, 6.0983e-01],
[-2.2104e+00, -1.5324e+00, -3.2054e+00, ..., -2.7124e+00,
-8.8843e-01, -1.9860e-01],
[-2.2916e-01, -1.0381e+00, -2.2309e-01, ..., -1.7774e-01,
5.3501e-01, 6.3630e-01],
...,
[-1.8143e-01, -1.1795e+00, -2.6890e+00, ..., -1.5323e+00,
-1.3886e+00, -8.1671e-01],
[ 1.4363e+00, 8.2178e-01, -9.7063e-01, ..., -2.6201e-01,
1.3148e-01, 4.2056e-01],
[ 8.6376e-01, 2.9529e-01, -6.8162e-01, ..., -1.5595e-01,
2.2822e-01, 4.0216e-01]],
[[-1.6525e+00, -1.1128e+00, -4.1629e+00, ..., -4.8871e+00,
-4.4386e+00, -3.3759e+00],
[ 6.3484e-01, 1.9884e+00, -2.2538e+00, ..., -3.5256e+00,
-3.6066e+00, -2.9497e+00],
[ 1.3828e+00, 1.7468e+00, -4.9421e+00, ..., -5.4761e+00,
-5.2761e+00, -4.9280e+00],
...,
[-1.2848e+00, 1.4530e-01, -6.6785e+00, ..., -7.5759e+00,
-6.8747e+00, -5.4303e+00],
[ 1.7974e+00, 2.8160e+00, -1.0120e+00, ..., -2.6238e+00,
-2.9855e+00, -2.4386e+00],
[ 1.2132e+00, 1.8039e+00, -2.2607e+00, ..., -3.2905e+00,
-3.6410e+00, -3.1608e+00]],
[[-9.9490e-02, -1.1369e+00, -1.3053e+00, ..., -4.8011e-01,
-1.5285e-01, -4.8667e-01],
[-5.7565e-01, -8.3595e-01, -7.5717e-01, ..., -2.7409e-01,
5.1516e-02, -7.1681e-02],
[-5.3038e-01, -3.4001e-01, -3.8993e-01, ..., -3.6374e-01,
9.3975e-02, 8.4716e-02],
...,
[-1.3818e+00, -1.7512e+00, -7.7046e-01, ..., -8.6210e-01,
-7.5253e-01, -1.0148e+00],
[-1.7308e+00, -1.1587e+00, -8.3859e-01, ..., -5.6638e-01,
-3.0761e-01, -3.7755e-01],
[-1.8742e+00, -1.2751e+00, -8.8671e-01, ..., -9.4761e-01,
-8.1430e-01, -7.8972e-01]],
...,
[[-3.0478e+00, -2.5393e+00, -2.8478e+00, ..., -3.4721e+00,
-2.6491e+00, -1.7607e+00],
[-2.3077e+00, -8.9502e-01, -1.6638e+00, ..., -2.6536e+00,
-2.3758e+00, -1.8241e+00],
[-1.0624e+00, 1.1554e+00, 5.8221e-01, ..., -4.5378e-01,
-4.1113e-02, 1.2820e-01],
...,
[-1.8285e+00, 1.2742e+00, 4.0189e-02, ..., -1.0714e+00,
-5.1761e-01, 8.3477e-03],
[-8.0193e-01, 1.9283e+00, 1.6511e+00, ..., 4.6049e-01,
4.2495e-01, 6.8880e-01],
[-2.2218e+00, -1.1476e+00, -9.9889e-01, ..., -1.8416e+00,
-9.4803e-01, -2.7223e-01]],
[[-1.5951e+00, -1.5801e+00, -2.9084e+00, ..., -1.2562e+00,
-8.3311e-01, -8.4600e-01],
[-5.7907e-01, 1.0546e-01, -3.2189e+00, ..., -1.3491e+00,
-1.1955e+00, -1.5814e+00],
[-2.0813e+00, -8.7741e-01, -5.3338e+00, ..., -3.1711e+00,
-2.6797e+00, -2.2106e+00],
...,
[-2.4374e+00, -1.5410e+00, -5.4575e+00, ..., -3.0888e+00,
-2.7474e+00, -2.2108e+00],
[-2.1980e+00, -1.4212e+00, -4.2434e+00, ..., -2.6646e+00,
-2.3312e+00, -1.6803e+00],
[-2.3842e+00, -1.7433e+00, -3.2994e+00, ..., -2.9385e+00,
-2.4364e+00, -2.1212e+00]],
[[-1.3622e+00, -1.9249e+00, -1.7222e+00, ..., -1.7745e+00,
-1.4665e+00, -1.1188e+00],
[-1.2871e+00, -1.7790e+00, -1.2503e+00, ..., -1.3798e+00,
-1.0823e+00, -8.6528e-01],
[-1.6139e+00, -1.9653e+00, -1.4974e+00, ..., -1.6005e+00,
-1.1952e+00, -8.8473e-01],
...,
[-3.1755e+00, -3.9656e+00, -3.7481e+00, ..., -3.9347e+00,
-3.1522e+00, -2.4350e+00],
[-2.1096e+00, -2.7280e+00, -2.6921e+00, ..., -2.7803e+00,
-2.2935e+00, -1.8672e+00],
[-1.5328e+00, -1.8602e+00, -1.9217e+00, ..., -1.8849e+00,
-1.5308e+00, -1.1747e+00]]],
[[[ 2.6279e-01, 5.5946e-01, 6.5014e-01, ..., -4.3591e-01,
-1.7953e-01, -5.6341e-02],
[-2.1021e+00, -1.2910e+00, -2.8677e+00, ..., -3.4099e+00,
-1.4275e+00, -5.4213e-01],
[-2.9713e-02, -7.5363e-01, 9.9706e-02, ..., -2.5044e-01,
-4.7182e-01, 1.0288e-01],
...,
[-3.4368e+00, -6.7750e+00, -6.1246e+00, ..., -1.9760e+00,
-1.9203e+00, -1.2662e+00],
[-1.9797e+00, -4.3363e+00, -5.2622e+00, ..., -8.1298e-02,
1.7210e-01, 3.9867e-01],
[-1.3546e+00, -2.8843e+00, -2.3631e+00, ..., 1.5092e-01,
4.1771e-01, 4.6991e-01]],
[[-1.7735e+00, -1.3061e+00, -4.2642e+00, ..., -6.9842e+00,
-6.1829e+00, -4.5974e+00],
[ 6.0979e-01, 1.8508e+00, -2.3430e+00, ..., -5.5439e+00,
-5.6586e+00, -4.3421e+00],
[ 1.4569e+00, 1.6044e+00, -5.0464e+00, ..., -7.9700e+00,
-7.9425e+00, -6.6648e+00],
...,
[-1.6648e+00, -2.0213e+00, -7.8556e+00, ..., -7.2602e+00,
-6.8296e+00, -5.2590e+00],
[ 1.5325e+00, 5.2656e-01, -2.2880e+00, ..., -2.0118e+00,
-2.6498e+00, -2.0885e+00],
[ 1.2391e+00, 6.0074e-01, -2.4479e+00, ..., -2.9521e+00,
-3.5309e+00, -3.0718e+00]],
[[-2.2482e-01, -1.3642e+00, -1.5745e+00, ..., -1.4790e+00,
-1.1052e+00, -1.3893e+00],
[-7.1751e-01, -1.0605e+00, -1.0137e+00, ..., -1.0421e+00,
-6.5857e-01, -9.4875e-01],
[-5.2759e-01, -3.6728e-01, -4.6832e-01, ..., -1.1587e+00,
-6.2783e-01, -4.8730e-01],
...,
[ 2.4569e-01, -5.7687e-01, -8.8216e-01, ..., -1.0136e+00,
-8.6887e-01, -1.2534e+00],
[-1.2901e+00, -1.0463e+00, -8.9675e-01, ..., -5.7286e-01,
-2.5050e-01, -4.1575e-01],
[-1.2332e+00, -5.4025e-01, -5.3113e-01, ..., -5.4350e-01,
-5.0678e-01, -6.6052e-01]],
...,
[[-3.0725e+00, -2.5533e+00, -2.6675e+00, ..., -3.0858e+00,
-2.5071e+00, -1.8381e+00],
[-2.0713e+00, -6.2650e-01, -1.0987e+00, ..., -2.4910e+00,
-2.5542e+00, -2.0583e+00],
[-1.0898e+00, 1.0628e+00, 8.3605e-01, ..., -7.6539e-02,
3.0249e-02, 2.9074e-01],
...,
[-6.9837e+00, -5.1503e+00, -7.1938e+00, ..., -9.8840e-01,
-2.8805e-01, 2.9619e-01],
[-4.1788e+00, -1.6971e+00, -1.8776e+00, ..., 4.4402e-01,
4.9187e-01, 8.4278e-01],
[-2.4579e+00, -2.2621e+00, -2.4831e+00, ..., -2.1690e+00,
-1.1343e+00, -3.3069e-01]],
[[-1.5882e+00, -1.5482e+00, -2.8872e+00, ..., -2.3046e+00,
-1.9793e+00, -1.8678e+00],
[-5.6007e-01, 1.0923e-01, -3.3051e+00, ..., -2.9180e+00,
-2.9321e+00, -2.9427e+00],
[-2.0223e+00, -8.6807e-01, -5.4224e+00, ..., -4.5023e+00,
-4.5751e+00, -3.7672e+00],
...,
[-4.1755e-01, -5.3941e-01, -3.9127e+00, ..., -2.8771e+00,
-2.7611e+00, -2.3464e+00],
[-4.0553e-01, 6.0720e-01, -2.0821e+00, ..., -2.4556e+00,
-2.2813e+00, -1.7063e+00],
[-1.6500e+00, -4.3928e-01, -2.3321e+00, ..., -2.8776e+00,
-2.4630e+00, -2.2273e+00]],
[[-1.3499e+00, -1.9090e+00, -1.6849e+00, ..., -2.6368e+00,
-2.2052e+00, -1.7121e+00],
[-1.2746e+00, -1.7589e+00, -1.2015e+00, ..., -2.4690e+00,
-1.9899e+00, -1.6272e+00],
[-1.5957e+00, -1.9242e+00, -1.4339e+00, ..., -2.8959e+00,
-2.3141e+00, -1.7955e+00],
...,
[-3.6776e+00, -4.7559e+00, -4.5741e+00, ..., -4.0834e+00,
-3.3145e+00, -2.6019e+00],
[-2.6809e+00, -3.3859e+00, -3.1746e+00, ..., -2.8160e+00,
-2.3563e+00, -1.9501e+00],
[-1.4337e+00, -1.7028e+00, -1.5758e+00, ..., -1.8629e+00,
-1.5478e+00, -1.2027e+00]]],
[[[ 2.8312e-01, 5.5660e-01, 6.9490e-01, ..., 4.8784e-01,
1.0119e+00, 5.9822e-01],
[-2.1520e+00, -1.4878e+00, -3.1401e+00, ..., -2.6487e+00,
-8.6034e-01, -1.9616e-01],
[-1.7326e-01, -1.0105e+00, -2.4258e-01, ..., -9.3048e-02,
5.6885e-01, 6.5607e-01],
...,
[-1.6487e-01, -1.1385e+00, -2.6328e+00, ..., -1.4365e+00,
-1.3443e+00, -8.2277e-01],
[ 1.4449e+00, 8.3576e-01, -9.3945e-01, ..., -2.0923e-01,
1.2106e-01, 3.6441e-01],
[ 8.5057e-01, 2.7600e-01, -6.7985e-01, ..., -1.3710e-01,
2.1720e-01, 3.5205e-01]],
[[-1.6506e+00, -1.1160e+00, -4.1590e+00, ..., -4.8864e+00,
-4.4629e+00, -3.4143e+00],
[ 6.0412e-01, 1.9679e+00, -2.2559e+00, ..., -3.5212e+00,
-3.6352e+00, -2.9954e+00],
[ 1.3558e+00, 1.7266e+00, -4.8908e+00, ..., -5.4652e+00,
-5.2944e+00, -4.9589e+00],
...,
[-1.2914e+00, 1.2101e-01, -6.7384e+00, ..., -7.6705e+00,
-6.9732e+00, -5.4568e+00],
[ 1.7579e+00, 2.7700e+00, -1.1016e+00, ..., -2.7248e+00,
-3.0746e+00, -2.4384e+00],
[ 1.1856e+00, 1.7773e+00, -2.2975e+00, ..., -3.2989e+00,
-3.6551e+00, -3.1042e+00]],
[[-1.1168e-01, -1.1547e+00, -1.3021e+00, ..., -4.9717e-01,
-1.6989e-01, -4.8460e-01],
[-5.8966e-01, -8.6149e-01, -7.6192e-01, ..., -3.0853e-01,
1.3294e-02, -7.6912e-02],
[-5.6154e-01, -3.6300e-01, -3.9121e-01, ..., -3.5794e-01,
7.2727e-02, 6.9070e-02],
...,
[-1.3907e+00, -1.7373e+00, -7.5194e-01, ..., -8.6693e-01,
-7.5423e-01, -1.0008e+00],
[-1.7283e+00, -1.1428e+00, -8.1867e-01, ..., -5.6707e-01,
-2.9202e-01, -3.3927e-01],
[-1.8747e+00, -1.2767e+00, -8.8351e-01, ..., -9.5237e-01,
-8.1851e-01, -7.8449e-01]],
...,
[[-3.0253e+00, -2.4688e+00, -2.7782e+00, ..., -3.4983e+00,
-2.6556e+00, -1.7622e+00],
[-2.2790e+00, -8.0673e-01, -1.5893e+00, ..., -2.6401e+00,
-2.3501e+00, -1.8624e+00],
[-1.0376e+00, 1.2231e+00, 6.2012e-01, ..., -4.3769e-01,
1.1026e-03, 1.3392e-01],
...,
[-1.8220e+00, 1.2625e+00, 3.6701e-02, ..., -1.1300e+00,
-4.9242e-01, 1.1852e-02],
[-8.1219e-01, 1.9082e+00, 1.6180e+00, ..., 3.9760e-01,
4.1505e-01, 6.6806e-01],
[-2.2240e+00, -1.1420e+00, -9.8877e-01, ..., -1.8676e+00,
-9.8057e-01, -3.2215e-01]],
[[-1.5949e+00, -1.5924e+00, -2.9434e+00, ..., -1.2786e+00,
-8.4085e-01, -8.2907e-01],
[-6.1390e-01, 6.5095e-02, -3.2658e+00, ..., -1.3770e+00,
-1.2230e+00, -1.5643e+00],
[-2.1023e+00, -9.0250e-01, -5.3586e+00, ..., -3.2038e+00,
-2.7223e+00, -2.2018e+00],
...,
[-2.4879e+00, -1.5957e+00, -5.4922e+00, ..., -3.1917e+00,
-2.7550e+00, -2.1719e+00],
[-2.2140e+00, -1.4780e+00, -4.2630e+00, ..., -2.7668e+00,
-2.3293e+00, -1.6293e+00],
[-2.3731e+00, -1.7806e+00, -3.3036e+00, ..., -3.0111e+00,
-2.4152e+00, -2.0822e+00]],
[[-1.3453e+00, -1.9048e+00, -1.6999e+00, ..., -1.7706e+00,
-1.4688e+00, -1.1102e+00],
[-1.2728e+00, -1.7637e+00, -1.2352e+00, ..., -1.3732e+00,
-1.0871e+00, -8.6134e-01],
[-1.6069e+00, -1.9630e+00, -1.4912e+00, ..., -1.6034e+00,
-1.1981e+00, -8.8104e-01],
...,
[-3.1836e+00, -3.9700e+00, -3.7660e+00, ..., -3.9445e+00,
-3.1763e+00, -2.4518e+00],
[-2.1206e+00, -2.7356e+00, -2.7051e+00, ..., -2.7717e+00,
-2.2953e+00, -1.8680e+00],
[-1.5419e+00, -1.8722e+00, -1.9371e+00, ..., -1.8791e+00,
-1.5265e+00, -1.1715e+00]]],
[[[ 4.6026e-01, 9.0217e-01, 6.6194e-01, ..., 7.8031e-01,
1.0491e+00, 8.2578e-01],
[-1.0439e+00, -1.5771e-01, -1.4874e+00, ..., -1.2440e+00,
5.3473e-02, 6.3105e-01],
[ 7.1419e-01, 4.9031e-01, 1.6059e-01, ..., 4.3438e-01,
8.2241e-01, 8.5805e-01],
...,
[-8.2452e-01, -1.7658e+00, -1.7948e+00, ..., -1.7393e+00,
-1.3177e+00, -9.6244e-01],
[ 7.8115e-01, 5.8018e-02, -6.3130e-02, ..., 2.6333e-02,
6.8571e-01, 7.7206e-01],
[ 1.7945e-01, -4.9618e-01, -4.1461e-01, ..., -9.4594e-02,
4.6142e-01, 3.5244e-01]],
[[-2.6406e+00, -1.5186e+00, -4.8603e+00, ..., -5.8929e+00,
-5.0485e+00, -3.9117e+00],
[-7.0349e-01, 1.2140e+00, -3.4656e+00, ..., -4.7549e+00,
-4.5718e+00, -3.7072e+00],
[-7.3628e-02, 1.0035e+00, -6.1344e+00, ..., -6.9177e+00,
-6.5915e+00, -5.9859e+00],
...,
[ 2.4444e-01, 3.6327e-01, -4.0452e+00, ..., -4.6814e+00,
-4.2178e+00, -3.4858e+00],
[ 2.2933e+00, 2.2985e+00, -1.3317e-01, ..., -8.5739e-01,
-1.1641e+00, -1.1939e+00],
[ 1.3450e+00, 1.1841e+00, -1.7614e+00, ..., -2.2210e+00,
-2.4846e+00, -2.3857e+00]],
[[-5.5435e-01, -1.5287e+00, -1.3657e+00, ..., -7.5324e-01,
-5.1083e-01, -7.4263e-01],
[-1.2806e+00, -1.6509e+00, -1.0756e+00, ..., -7.0392e-01,
-3.8425e-01, -3.3538e-01],
[-1.1509e+00, -1.0455e+00, -4.9580e-01, ..., -4.5312e-01,
-1.0784e-01, -1.0825e-01],
...,
[-1.9374e-01, -6.5622e-01, -4.7888e-01, ..., -5.0427e-01,
-2.0214e-01, -4.5581e-01],
[-7.3372e-01, -2.3661e-01, -4.3618e-01, ..., -1.9672e-01,
2.4904e-01, 2.3286e-01],
[-8.2212e-01, -2.1023e-01, -3.3250e-01, ..., -3.6572e-01,
-1.7486e-01, -2.2181e-01]],
...,
[[-2.0131e+00, -1.1515e+00, -2.4453e+00, ..., -3.2750e+00,
-2.7360e+00, -1.8564e+00],
[-4.9707e-01, 1.0829e+00, -6.8673e-01, ..., -1.6348e+00,
-1.8461e+00, -1.5993e+00],
[-2.6091e-01, 2.0672e+00, 4.5705e-01, ..., -3.0846e-01,
-3.4288e-02, 1.2609e-01],
...,
[-2.0142e+00, -6.8574e-03, -1.7521e-01, ..., -1.3387e+00,
-3.7413e-01, 1.2125e-01],
[-1.1410e+00, 8.7276e-01, 9.4921e-01, ..., 4.0192e-03,
3.1389e-01, 4.8500e-01],
[-2.0592e+00, -1.3740e+00, -1.6109e+00, ..., -2.4504e+00,
-1.4708e+00, -9.1824e-01]],
[[-1.4897e+00, -1.4597e+00, -2.7942e+00, ..., -1.1324e+00,
-6.1399e-01, -5.0555e-01],
[-9.3719e-01, -5.6771e-01, -3.7285e+00, ..., -1.9290e+00,
-1.6598e+00, -1.8589e+00],
[-2.4909e+00, -2.0190e+00, -6.2624e+00, ..., -4.2108e+00,
-3.6521e+00, -2.8822e+00],
...,
[-1.4142e+00, -8.1461e-02, -3.4226e+00, ..., -1.4477e+00,
-1.0271e+00, -9.3103e-01],
[-1.7060e+00, -6.5528e-01, -3.0593e+00, ..., -1.7333e+00,
-1.1503e+00, -7.9697e-01],
[-2.3121e+00, -1.4900e+00, -3.1003e+00, ..., -2.9434e+00,
-2.1124e+00, -1.9624e+00]],
[[-1.2346e+00, -1.8097e+00, -1.6595e+00, ..., -1.8776e+00,
-1.5503e+00, -1.1585e+00],
[-1.2980e+00, -1.8028e+00, -1.4383e+00, ..., -1.6528e+00,
-1.3366e+00, -1.0489e+00],
[-1.7098e+00, -2.0759e+00, -1.7369e+00, ..., -1.9540e+00,
-1.4982e+00, -1.0989e+00],
...,
[-2.3768e+00, -2.9455e+00, -2.7174e+00, ..., -2.8604e+00,
-2.2938e+00, -1.7483e+00],
[-1.5457e+00, -2.0080e+00, -1.9323e+00, ..., -1.9320e+00,
-1.6004e+00, -1.2841e+00],
[-1.0087e+00, -1.2967e+00, -1.3268e+00, ..., -1.1763e+00,
-1.0124e+00, -7.6322e-01]]],
[[[ 2.7739e-01, 5.8069e-01, 6.6118e-01, ..., -7.1837e-01,
-4.6976e-01, -2.4042e-01],
[-1.7099e+00, -9.5791e-01, -2.3500e+00, ..., -3.8615e+00,
-1.7456e+00, -7.6875e-01],
[ 1.1316e-01, -5.1050e-01, 2.0168e-01, ..., -4.0761e-01,
-7.3829e-01, -1.4745e-02],
...,
[-6.9287e-01, -1.7433e+00, -1.8393e+00, ..., -2.0377e+00,
-1.9756e+00, -1.2838e+00],
[ 9.3187e-01, 1.7273e-01, 5.4756e-02, ..., -2.8254e-01,
3.9365e-02, 2.8291e-01],
[ 3.9257e-01, -2.2458e-01, -1.4137e-01, ..., -8.9483e-02,
2.6750e-01, 3.8647e-01]],
[[-1.8530e+00, -1.3545e+00, -4.2972e+00, ..., -7.2974e+00,
-6.4806e+00, -4.7855e+00],
[ 2.4309e-01, 1.4383e+00, -2.6740e+00, ..., -5.8572e+00,
-5.9970e+00, -4.5389e+00],
[ 9.6392e-01, 1.1634e+00, -5.1815e+00, ..., -8.3887e+00,
-8.3500e+00, -6.8743e+00],
...,
[ 1.4288e-01, 3.4492e-01, -4.1779e+00, ..., -7.7196e+00,
-7.3936e+00, -5.6074e+00],
[ 2.4128e+00, 2.5267e+00, 1.4935e-01, ..., -2.2851e+00,
-3.0461e+00, -2.2698e+00],
[ 1.4232e+00, 1.3591e+00, -1.6054e+00, ..., -3.0975e+00,
-3.7645e+00, -3.1674e+00]],
[[-2.3941e-01, -1.2907e+00, -1.4486e+00, ..., -1.6318e+00,
-1.2392e+00, -1.5139e+00],
[-7.0437e-01, -1.0690e+00, -9.8143e-01, ..., -1.1392e+00,
-7.5035e-01, -1.0868e+00],
[-5.1123e-01, -4.1851e-01, -4.5690e-01, ..., -1.2764e+00,
-7.2399e-01, -5.4045e-01],
...,
[-2.6106e-01, -8.0820e-01, -5.9531e-01, ..., -1.0750e+00,
-9.1288e-01, -1.3045e+00],
[-8.1068e-01, -3.6731e-01, -5.3322e-01, ..., -5.6762e-01,
-2.4239e-01, -4.1558e-01],
[-9.6517e-01, -3.8378e-01, -4.2328e-01, ..., -7.5174e-01,
-7.1342e-01, -8.5341e-01]],
...,
[[-2.6397e+00, -2.0981e+00, -2.3741e+00, ..., -2.9538e+00,
-2.4055e+00, -1.7998e+00],
[-1.6924e+00, -2.9116e-01, -8.7670e-01, ..., -2.6389e+00,
-2.6898e+00, -2.1420e+00],
[-8.9586e-01, 1.1494e+00, 7.6575e-01, ..., -1.0268e-01,
1.5468e-03, 2.6786e-01],
...,
[-2.0555e+00, 1.9744e-01, -7.3703e-02, ..., -9.1329e-01,
-3.4617e-01, 2.1014e-01],
[-1.0578e+00, 1.0506e+00, 1.0455e+00, ..., 6.5633e-01,
5.6231e-01, 8.7554e-01],
[-2.0478e+00, -1.3870e+00, -1.5960e+00, ..., -2.0128e+00,
-1.0062e+00, -2.4430e-01]],
[[-1.5424e+00, -1.5215e+00, -2.7349e+00, ..., -2.6165e+00,
-2.2617e+00, -2.1277e+00],
[-6.8518e-01, -1.3356e-01, -3.2321e+00, ..., -3.2090e+00,
-3.1949e+00, -3.1510e+00],
[-2.0497e+00, -1.1236e+00, -5.2899e+00, ..., -4.6617e+00,
-4.7836e+00, -3.9431e+00],
...,
[-1.4222e+00, -1.4383e-02, -3.6242e+00, ..., -2.9391e+00,
-2.9015e+00, -2.4467e+00],
[-1.6590e+00, -5.2638e-01, -3.1352e+00, ..., -2.5547e+00,
-2.4444e+00, -1.8156e+00],
[-2.2424e+00, -1.3219e+00, -2.9789e+00, ..., -2.9532e+00,
-2.5901e+00, -2.3286e+00]],
[[-1.2563e+00, -1.7843e+00, -1.6075e+00, ..., -2.8520e+00,
-2.3800e+00, -1.8577e+00],
[-1.2182e+00, -1.6762e+00, -1.2221e+00, ..., -2.7265e+00,
-2.1843e+00, -1.7901e+00],
[-1.5432e+00, -1.8720e+00, -1.4681e+00, ..., -3.2111e+00,
-2.5710e+00, -1.9991e+00],
...,
[-2.4060e+00, -3.0058e+00, -2.7473e+00, ..., -4.3425e+00,
-3.5272e+00, -2.7714e+00],
[-1.5715e+00, -2.0586e+00, -1.9659e+00, ..., -2.9758e+00,
-2.4886e+00, -2.0630e+00],
[-1.0360e+00, -1.3239e+00, -1.3599e+00, ..., -1.9821e+00,
-1.6308e+00, -1.2732e+00]]]], device='cuda:0'),
tensor([[[[ 4.7229e-01, 9.5892e-01, 8.6604e-01, ..., -3.4610e-01,
-6.9235e-02, 1.1003e-02],
[-1.2010e+00, -3.0863e-01, -1.4804e+00, ..., -3.2145e+00,
-1.2709e+00, -4.4583e-01],
[ 6.0252e-01, 4.2900e-01, 3.2881e-01, ..., -1.5486e-01,
-3.2957e-01, 1.7930e-01],
...,
[-1.8135e-01, -8.4359e-01, -1.5360e+00, ..., -1.9625e+00,
-1.8907e+00, -1.2837e+00],
[ 1.0664e+00, 7.9535e-01, 2.8080e-01, ..., 1.8578e-02,
2.6515e-01, 4.2503e-01],
[ 7.1126e-01, 5.1625e-01, 1.8721e-01, ..., 2.3301e-01,
4.6434e-01, 4.7236e-01]],
[[-2.2904e+00, -1.1357e+00, -4.1048e+00, ..., -6.8464e+00,
-6.0784e+00, -4.5054e+00],
[-3.1337e-01, 1.6253e+00, -2.5407e+00, ..., -5.4649e+00,
-5.5671e+00, -4.2702e+00],
[ 3.6046e-01, 1.3763e+00, -5.0665e+00, ..., -7.8834e+00,
-7.8175e+00, -6.5817e+00],
...,
[-4.8525e-01, 5.9833e-01, -4.0790e+00, ..., -7.1134e+00,
-6.6584e+00, -5.1586e+00],
[ 2.0630e+00, 2.7753e+00, 3.4331e-01, ..., -1.9161e+00,
-2.5290e+00, -2.0367e+00],
[ 1.1880e+00, 1.5055e+00, -1.4803e+00, ..., -2.9109e+00,
-3.4622e+00, -3.0415e+00]],
[[-5.7501e-01, -1.5486e+00, -1.3880e+00, ..., -1.4155e+00,
-1.0474e+00, -1.3284e+00],
[-1.2665e+00, -1.5975e+00, -9.8115e-01, ..., -9.8875e-01,
-6.0338e-01, -8.8340e-01],
[-1.1207e+00, -9.6524e-01, -3.8474e-01, ..., -1.0905e+00,
-5.7964e-01, -4.5977e-01],
...,
[-9.0918e-01, -1.3073e+00, -5.3316e-01, ..., -1.0530e+00,
-8.7762e-01, -1.2533e+00],
[-1.2625e+00, -8.4342e-01, -5.3110e-01, ..., -5.8890e-01,
-2.5499e-01, -4.1362e-01],
[-1.3267e+00, -7.2753e-01, -3.3885e-01, ..., -5.1420e-01,
-4.4611e-01, -6.0359e-01]],
...,
[[-2.1728e+00, -1.2282e+00, -2.4871e+00, ..., -3.0870e+00,
-2.5130e+00, -1.8135e+00],
[-7.2056e-01, 9.1817e-01, -9.3648e-01, ..., -2.4226e+00,
-2.4979e+00, -2.0098e+00],
[-2.8048e-01, 2.1209e+00, 4.9015e-01, ..., -9.0872e-02,
7.5147e-03, 2.6968e-01],
...,
[-1.5034e+00, 1.2280e+00, -1.8354e-01, ..., -9.3622e-01,
-2.6832e-01, 3.1111e-01],
[-5.0109e-01, 1.5526e+00, 8.1739e-01, ..., 4.2998e-01,
4.6278e-01, 8.0611e-01],
[-1.6361e+00, -1.1072e+00, -1.6270e+00, ..., -2.2125e+00,
-1.1956e+00, -3.8266e-01]],
[[-1.3130e+00, -1.2244e+00, -2.5698e+00, ..., -2.2313e+00,
-1.9010e+00, -1.7884e+00],
[-6.4203e-01, -1.5468e-01, -3.2807e+00, ..., -2.8840e+00,
-2.8710e+00, -2.8830e+00],
[-2.0991e+00, -1.4461e+00, -5.6120e+00, ..., -4.4845e+00,
-4.5078e+00, -3.7089e+00],
...,
[-1.6347e+00, -4.8105e-01, -4.0541e+00, ..., -2.7825e+00,
-2.6971e+00, -2.3022e+00],
[-1.6619e+00, -7.1341e-01, -3.3923e+00, ..., -2.4068e+00,
-2.2108e+00, -1.6440e+00],
[-2.1078e+00, -1.3780e+00, -2.9796e+00, ..., -2.8580e+00,
-2.4025e+00, -2.1835e+00]],
[[-1.1489e+00, -1.7019e+00, -1.5292e+00, ..., -2.5619e+00,
-2.1453e+00, -1.6644e+00],
[-1.1514e+00, -1.6182e+00, -1.2193e+00, ..., -2.3910e+00,
-1.9293e+00, -1.5782e+00],
[-1.4921e+00, -1.7968e+00, -1.4295e+00, ..., -2.8053e+00,
-2.2392e+00, -1.7373e+00],
...,
[-2.3169e+00, -2.8856e+00, -2.6785e+00, ..., -4.0079e+00,
-3.2531e+00, -2.5519e+00],
[-1.5505e+00, -1.9655e+00, -1.9090e+00, ..., -2.7670e+00,
-2.3236e+00, -1.9187e+00],
[-1.0326e+00, -1.2353e+00, -1.2961e+00, ..., -1.8280e+00,
-1.5254e+00, -1.1799e+00]]],
[[[ 6.4331e-02, 2.4198e-01, -1.3609e+00, ..., -6.1350e-01,
-1.8184e-01, -3.8762e-02],
[-2.0619e+00, -1.5720e+00, -4.4276e+00, ..., -3.7967e+00,
-1.5325e+00, -5.5194e-01],
[ 5.5421e-01, -1.0364e-01, -8.6223e-01, ..., -1.8340e-01,
-2.4908e-01, 2.4051e-01],
...,
[-8.9587e-01, -1.9399e+00, -2.0826e+00, ..., -1.7519e+00,
-1.3316e+00, -9.0119e-01],
[ 8.8684e-01, 7.5554e-02, -1.3551e-01, ..., -1.9104e-02,
6.4350e-01, 7.9308e-01],
[ 2.2811e-01, -5.5044e-01, -5.2656e-01, ..., -1.7751e-01,
3.9470e-01, 3.4453e-01]],
[[-3.2590e+00, -1.7822e+00, -6.2216e+00, ..., -7.1547e+00,
-6.2548e+00, -4.6242e+00],
[-8.0584e-01, 1.5195e+00, -4.3630e+00, ..., -5.4826e+00,
-5.5473e+00, -4.2869e+00],
[-3.8168e-01, 1.0747e+00, -7.3944e+00, ..., -7.9884e+00,
-7.8476e+00, -6.7130e+00],
...,
[ 5.0706e-01, 7.2491e-01, -3.8880e+00, ..., -4.6186e+00,
-4.1792e+00, -3.5124e+00],
[ 2.6825e+00, 2.7575e+00, 2.6559e-01, ..., -8.9640e-01,
-1.2432e+00, -1.3083e+00],
[ 1.6079e+00, 1.4538e+00, -1.6091e+00, ..., -2.2447e+00,
-2.5296e+00, -2.4232e+00]],
[[-1.3141e+00, -2.3188e+00, -2.2786e+00, ..., -1.7509e+00,
-1.2340e+00, -1.3809e+00],
[-1.6626e+00, -1.9640e+00, -1.5022e+00, ..., -1.2650e+00,
-7.3299e-01, -8.1202e-01],
[-2.1616e+00, -1.9311e+00, -1.0003e+00, ..., -1.1157e+00,
-4.7259e-01, -1.9016e-01],
...,
[-2.8377e-01, -7.3026e-01, -4.9340e-01, ..., -4.9410e-01,
-2.0589e-01, -4.3654e-01],
[-8.6679e-01, -2.9132e-01, -4.6168e-01, ..., -2.0336e-01,
2.1911e-01, 2.2808e-01],
[-9.5908e-01, -2.5474e-01, -3.4882e-01, ..., -3.9180e-01,
-2.0792e-01, -2.3502e-01]],
...,
[[-1.9836e+00, -1.0390e+00, -2.3006e+00, ..., -3.2332e+00,
-2.6570e+00, -1.8411e+00],
[-1.7485e+00, 4.6378e-02, -1.3568e+00, ..., -2.4860e+00,
-2.5400e+00, -1.9883e+00],
[-3.8590e-01, 2.3729e+00, 1.0506e+00, ..., 1.3640e-01,
2.7944e-01, 5.1972e-01],
...,
[-2.2324e+00, 7.3791e-02, -5.9937e-02, ..., -1.3016e+00,
-4.0927e-01, 6.9412e-02],
[-1.2947e+00, 9.4911e-01, 1.0786e+00, ..., 2.4788e-02,
3.0637e-01, 4.4739e-01],
[-2.2384e+00, -1.4929e+00, -1.7012e+00, ..., -2.3450e+00,
-1.4127e+00, -9.0806e-01]],
[[-2.4081e+00, -2.8321e+00, -4.2286e+00, ..., -2.3674e+00,
-1.8878e+00, -1.7013e+00],
[-1.7753e+00, -1.6613e+00, -4.8950e+00, ..., -2.8921e+00,
-2.7765e+00, -2.7691e+00],
[-2.8186e+00, -2.4497e+00, -6.6615e+00, ..., -4.4570e+00,
-4.3796e+00, -3.4908e+00],
...,
[-1.3787e+00, 1.7580e-01, -3.4996e+00, ..., -1.4362e+00,
-1.0453e+00, -9.6954e-01],
[-1.6958e+00, -4.8734e-01, -3.0953e+00, ..., -1.7029e+00,
-1.1967e+00, -8.6054e-01],
[-2.3595e+00, -1.4228e+00, -3.1530e+00, ..., -2.9139e+00,
-2.1793e+00, -1.9867e+00]],
[[-1.8055e+00, -2.5266e+00, -2.4478e+00, ..., -2.6978e+00,
-2.2112e+00, -1.6946e+00],
[-1.7972e+00, -2.5074e+00, -2.1457e+00, ..., -2.4008e+00,
-1.8997e+00, -1.5345e+00],
[-2.2571e+00, -2.8247e+00, -2.5164e+00, ..., -2.7505e+00,
-2.1465e+00, -1.6167e+00],
...,
[-2.5213e+00, -3.1255e+00, -2.8298e+00, ..., -2.8503e+00,
-2.2553e+00, -1.7256e+00],
[-1.6478e+00, -2.1384e+00, -2.0211e+00, ..., -1.9402e+00,
-1.5803e+00, -1.2668e+00],
[-1.0744e+00, -1.3807e+00, -1.3793e+00, ..., -1.1952e+00,
-1.0033e+00, -7.5843e-01]]],
[[[ 7.7775e-02, 2.3245e-01, -1.3939e+00, ..., -6.3407e-01,
-1.7609e-01, -4.0037e-02],
[-2.0853e+00, -1.6012e+00, -4.5283e+00, ..., -3.7887e+00,
-1.4990e+00, -5.3289e-01],
[ 5.7765e-01, -6.8176e-02, -8.4739e-01, ..., -2.3952e-01,
-2.7263e-01, 2.2591e-01],
...,
[-9.1399e-01, -1.8881e+00, -1.8627e+00, ..., -1.8016e+00,
-1.3014e+00, -9.1653e-01],
[ 7.3075e-01, -2.0182e-02, -9.8864e-02, ..., -2.4326e-02,
7.1291e-01, 7.9747e-01],
[ 1.1996e-01, -5.8469e-01, -4.8780e-01, ..., -1.5090e-01,
4.4929e-01, 3.3263e-01]],
[[-3.2908e+00, -1.8088e+00, -6.2582e+00, ..., -7.1160e+00,
-6.2555e+00, -4.6148e+00],
[-7.9927e-01, 1.5254e+00, -4.3483e+00, ..., -5.4825e+00,
-5.5660e+00, -4.3168e+00],
[-3.6248e-01, 1.0783e+00, -7.4051e+00, ..., -8.0013e+00,
-7.8684e+00, -6.7164e+00],
...,
[ 3.0485e-01, 3.5813e-01, -4.0263e+00, ..., -4.6292e+00,
-4.2186e+00, -3.5148e+00],
[ 2.3139e+00, 2.2783e+00, -1.3399e-01, ..., -9.0083e-01,
-1.2737e+00, -1.2857e+00],
[ 1.3558e+00, 1.1773e+00, -1.7492e+00, ..., -2.2895e+00,
-2.5368e+00, -2.3978e+00]],
[[-1.3436e+00, -2.3645e+00, -2.3250e+00, ..., -1.7394e+00,
-1.2248e+00, -1.3739e+00],
[-1.6876e+00, -1.9955e+00, -1.5369e+00, ..., -1.2328e+00,
-7.0314e-01, -7.9558e-01],
[-2.1675e+00, -1.9373e+00, -1.0168e+00, ..., -1.1061e+00,
-4.8050e-01, -1.9339e-01],
...,
[-1.4762e-01, -6.1072e-01, -4.6036e-01, ..., -4.8063e-01,
-1.9943e-01, -4.3913e-01],
[-7.1051e-01, -1.9676e-01, -4.2879e-01, ..., -2.0598e-01,
2.4098e-01, 2.2690e-01],
[-8.0825e-01, -1.9550e-01, -3.4004e-01, ..., -3.9154e-01,
-1.8962e-01, -2.2438e-01]],
...,
[[-2.0381e+00, -1.0994e+00, -2.3414e+00, ..., -3.1765e+00,
-2.6448e+00, -1.8019e+00],
[-1.7613e+00, 3.5796e-02, -1.3285e+00, ..., -2.5061e+00,
-2.5892e+00, -2.0209e+00],
[-3.8965e-01, 2.3591e+00, 1.0802e+00, ..., 1.5136e-01,
2.6783e-01, 4.9992e-01],
...,
[-2.0766e+00, -1.0127e-01, -2.0416e-01, ..., -1.3161e+00,
-3.6360e-01, 9.9205e-02],
[-1.2001e+00, 8.0305e-01, 9.1225e-01, ..., -2.1387e-03,
3.0732e-01, 4.6344e-01],
[-2.0927e+00, -1.4356e+00, -1.6812e+00, ..., -2.3820e+00,
-1.4223e+00, -9.1882e-01]],
[[-2.4214e+00, -2.8327e+00, -4.2219e+00, ..., -2.3881e+00,
-1.9246e+00, -1.7674e+00],
[-1.7775e+00, -1.6481e+00, -4.9006e+00, ..., -2.9210e+00,
-2.7887e+00, -2.7970e+00],
[-2.8407e+00, -2.4800e+00, -6.7083e+00, ..., -4.4246e+00,
-4.3555e+00, -3.4835e+00],
...,
[-1.4056e+00, -3.7463e-02, -3.3791e+00, ..., -1.4834e+00,
-1.0487e+00, -9.5957e-01],
[-1.7109e+00, -6.3609e-01, -3.0118e+00, ..., -1.7591e+00,
-1.1967e+00, -8.3672e-01],
[-2.3229e+00, -1.4825e+00, -3.0689e+00, ..., -2.9533e+00,
-2.1438e+00, -1.9761e+00]],
[[-1.8026e+00, -2.5303e+00, -2.4493e+00, ..., -2.7065e+00,
-2.2151e+00, -1.6952e+00],
[-1.8001e+00, -2.5045e+00, -2.1436e+00, ..., -2.4120e+00,
-1.9133e+00, -1.5519e+00],
[-2.2555e+00, -2.8101e+00, -2.5068e+00, ..., -2.7718e+00,
-2.1634e+00, -1.6489e+00],
...,
[-2.3872e+00, -2.9630e+00, -2.7467e+00, ..., -2.8530e+00,
-2.2801e+00, -1.7511e+00],
[-1.5494e+00, -2.0197e+00, -1.9546e+00, ..., -1.9555e+00,
-1.6095e+00, -1.2940e+00],
[-1.0093e+00, -1.3132e+00, -1.3472e+00, ..., -1.2001e+00,
-1.0231e+00, -7.7902e-01]]],
[[[ 4.2059e-01, 8.2925e-01, 4.6025e-01, ..., 5.1885e-01,
1.0327e+00, 6.6641e-01],
[-1.3265e+00, -5.1544e-01, -2.0779e+00, ..., -2.3978e+00,
-6.4376e-01, 4.2799e-03],
[ 8.9895e-01, 6.0950e-01, 3.4376e-01, ..., -1.6282e-01,
5.6851e-01, 6.7392e-01],
...,
[ 4.3875e-02, -1.8385e-01, -3.5414e+00, ..., -1.9535e+00,
-1.3532e+00, -8.9438e-01],
[ 1.3302e+00, 1.1305e+00, 5.0277e-01, ..., -2.5880e-01,
5.4466e-01, 7.3650e-01],
[ 9.9806e-01, 8.8157e-01, 5.5004e-01, ..., -2.7775e-01,
3.4928e-01, 3.3001e-01]],
[[-2.8075e+00, -1.5410e+00, -4.9135e+00, ..., -4.8868e+00,
-4.4303e+00, -3.3638e+00],
[-4.9982e-01, 1.6743e+00, -3.0063e+00, ..., -3.4840e+00,
-3.5819e+00, -2.9114e+00],
[ 2.9090e-01, 1.4664e+00, -5.7651e+00, ..., -5.4482e+00,
-5.2904e+00, -4.9377e+00],
...,
[-9.9129e-01, -7.6910e-01, -6.5776e+00, ..., -4.6130e+00,
-4.1710e+00, -3.4890e+00],
[ 1.6737e+00, 2.2966e+00, -3.5535e-01, ..., -8.7823e-01,
-1.1999e+00, -1.2573e+00],
[ 9.8488e-01, 1.2609e+00, -1.8509e+00, ..., -2.2563e+00,
-2.4914e+00, -2.3949e+00]],
[[-7.0406e-01, -1.8917e+00, -1.7565e+00, ..., -6.1084e-01,
-2.2412e-01, -4.7894e-01],
[-1.4889e+00, -1.9636e+00, -1.2928e+00, ..., -3.6733e-01,
3.9903e-02, 2.2429e-02],
[-1.4398e+00, -1.4281e+00, -7.9019e-01, ..., -2.7761e-01,
2.4360e-01, 2.7695e-01],
...,
[-9.4869e-01, -2.6138e+00, -4.4224e+00, ..., -4.9921e-01,
-1.8992e-01, -3.6575e-01],
[-1.2190e+00, -9.2132e-01, -6.3912e-01, ..., -2.9564e-01,
1.4131e-01, 1.9935e-01],
[-1.2449e+00, -7.3233e-01, -3.3806e-01, ..., -4.7531e-01,
-2.5947e-01, -2.3730e-01]],
...,
[[-2.2803e+00, -1.1697e+00, -2.5728e+00, ..., -3.6279e+00,
-2.8172e+00, -1.8311e+00],
[-6.8962e-01, 1.1731e+00, -8.4958e-01, ..., -2.5482e+00,
-2.3522e+00, -1.8193e+00],
[-1.6954e-01, 2.6113e+00, 7.6902e-01, ..., -5.6597e-01,
-7.5003e-02, 1.6192e-01],
...,
[-1.3137e+00, 2.9094e+00, 2.7869e+00, ..., -1.4448e+00,
-6.6792e-01, -1.0025e-01],
[-4.8973e-01, 1.5412e+00, 8.8112e-01, ..., -8.8132e-02,
1.2840e-01, 3.6493e-01],
[-1.6412e+00, -1.0812e+00, -1.4673e+00, ..., -2.3035e+00,
-1.4172e+00, -8.6661e-01]],
[[-1.4771e+00, -1.4927e+00, -3.2257e+00, ..., -1.1173e+00,
-7.0985e-01, -7.0811e-01],
[-7.2950e-01, -2.8909e-01, -4.0308e+00, ..., -1.3895e+00,
-1.2340e+00, -1.5893e+00],
[-2.1890e+00, -1.5742e+00, -6.5028e+00, ..., -3.3224e+00,
-2.8216e+00, -2.2645e+00],
...,
[-1.8135e+00, -1.2049e+00, -6.2438e+00, ..., -1.5166e+00,
-1.1120e+00, -9.2435e-01],
[-1.7245e+00, -9.9044e-01, -3.7168e+00, ..., -1.7390e+00,
-1.1811e+00, -7.7951e-01],
[-2.0478e+00, -1.4305e+00, -3.0298e+00, ..., -2.7711e+00,
-2.0187e+00, -1.8209e+00]],
[[-1.3036e+00, -1.9694e+00, -1.8083e+00, ..., -1.7785e+00,
-1.4828e+00, -1.1298e+00],
[-1.3059e+00, -1.8857e+00, -1.4662e+00, ..., -1.3923e+00,
-1.1097e+00, -8.8361e-01],
[-1.6956e+00, -2.0995e+00, -1.7106e+00, ..., -1.5922e+00,
-1.2043e+00, -8.8626e-01],
...,
[-2.4038e+00, -3.3171e+00, -3.5825e+00, ..., -2.8685e+00,
-2.2885e+00, -1.7516e+00],
[-1.6103e+00, -2.0818e+00, -2.0964e+00, ..., -1.9847e+00,
-1.6326e+00, -1.3137e+00],
[-1.1058e+00, -1.3605e+00, -1.4720e+00, ..., -1.2337e+00,
-1.0409e+00, -7.9663e-01]]],
[[[ 2.2180e-01, 4.9108e-01, 6.4433e-01, ..., -3.7021e-01,
-1.0172e-01, -9.4693e-03],
[-2.0106e+00, -1.3258e+00, -2.8275e+00, ..., -3.2652e+00,
-1.3125e+00, -4.7079e-01],
[-1.4927e-01, -9.1125e-01, 1.7777e-02, ..., -1.8686e-01,
-3.7555e-01, 1.4370e-01],
...,
[-9.6794e-01, -2.1631e+00, -2.0510e+00, ..., -1.9475e+00,
-1.8975e+00, -1.2733e+00],
[ 8.0803e-01, -1.1937e-01, -9.0268e-02, ..., 1.0170e-02,
2.6148e-01, 4.5790e-01],
[ 2.1327e-01, -5.8654e-01, -3.8556e-01, ..., 2.2399e-01,
4.6673e-01, 5.1225e-01]],
[[-1.6671e+00, -1.3626e+00, -4.2641e+00, ..., -6.8652e+00,
-6.0810e+00, -4.5253e+00],
[ 4.7765e-01, 1.4823e+00, -2.5382e+00, ..., -5.4599e+00,
-5.5586e+00, -4.2851e+00],
[ 1.2106e+00, 1.2193e+00, -5.0643e+00, ..., -7.8683e+00,
-7.8252e+00, -6.5862e+00],
...,
[ 3.6472e-01, 3.1125e-01, -4.1634e+00, ..., -7.1122e+00,
-6.6923e+00, -5.1911e+00],
[ 2.4925e+00, 2.4411e+00, 6.2191e-02, ..., -1.9160e+00,
-2.5496e+00, -2.0423e+00],
[ 1.4911e+00, 1.2921e+00, -1.6370e+00, ..., -2.9330e+00,
-3.4790e+00, -3.0527e+00]],
[[-1.3099e-01, -1.1658e+00, -1.4033e+00, ..., -1.4230e+00,
-1.0617e+00, -1.3440e+00],
[-5.0638e-01, -8.3482e-01, -8.8297e-01, ..., -9.9958e-01,
-6.2724e-01, -9.0774e-01],
[-3.4398e-01, -2.4637e-01, -4.4561e-01, ..., -1.1147e+00,
-5.9620e-01, -4.7839e-01],
...,
[-6.4106e-02, -5.8575e-01, -6.1838e-01, ..., -1.0377e+00,
-8.7265e-01, -1.2528e+00],
[-7.0715e-01, -2.2253e-01, -5.7246e-01, ..., -5.6162e-01,
-2.4958e-01, -4.2336e-01],
[-8.6199e-01, -2.5391e-01, -4.6008e-01, ..., -5.0264e-01,
-4.4619e-01, -6.1247e-01]],
...,
[[-2.8414e+00, -2.4490e+00, -2.4495e+00, ..., -3.0563e+00,
-2.4813e+00, -1.8124e+00],
[-2.1250e+00, -8.4588e-01, -1.1245e+00, ..., -2.4069e+00,
-2.4729e+00, -2.0038e+00],
[-1.1103e+00, 8.2194e-01, 7.5363e-01, ..., -6.6881e-02,
5.5915e-02, 3.1284e-01],
...,
[-2.2740e+00, -2.2196e-01, -1.6976e-01, ..., -9.2038e-01,
-2.4590e-01, 2.9151e-01],
[-1.2593e+00, 8.2305e-01, 1.0301e+00, ..., 4.3426e-01,
4.9761e-01, 8.2395e-01],
[-2.1678e+00, -1.4923e+00, -1.7047e+00, ..., -2.2190e+00,
-1.1697e+00, -3.4881e-01]],
[[-1.6381e+00, -1.6411e+00, -2.8103e+00, ..., -2.2338e+00,
-1.9011e+00, -1.7952e+00],
[-7.0248e-01, -9.5772e-02, -3.1578e+00, ..., -2.8653e+00,
-2.8541e+00, -2.8682e+00],
[-2.0638e+00, -9.7549e-01, -5.1314e+00, ..., -4.4672e+00,
-4.5052e+00, -3.6921e+00],
...,
[-1.3713e+00, 1.2444e-01, -3.3782e+00, ..., -2.8226e+00,
-2.7486e+00, -2.3114e+00],
[-1.6879e+00, -5.0665e-01, -3.0101e+00, ..., -2.4342e+00,
-2.2585e+00, -1.6646e+00],
[-2.3269e+00, -1.3815e+00, -3.0384e+00, ..., -2.8635e+00,
-2.4320e+00, -2.2025e+00]],
[[-1.3004e+00, -1.8289e+00, -1.6456e+00, ..., -2.5790e+00,
-2.1528e+00, -1.6731e+00],
[-1.2404e+00, -1.7028e+00, -1.2186e+00, ..., -2.4055e+00,
-1.9422e+00, -1.5905e+00],
[-1.5618e+00, -1.8988e+00, -1.4695e+00, ..., -2.8232e+00,
-2.2502e+00, -1.7506e+00],
...,
[-2.4374e+00, -3.0430e+00, -2.7862e+00, ..., -4.0035e+00,
-3.2475e+00, -2.5428e+00],
[-1.5785e+00, -2.0826e+00, -1.9840e+00, ..., -2.7643e+00,
-2.3163e+00, -1.9137e+00],
[-1.0330e+00, -1.3443e+00, -1.3590e+00, ..., -1.8247e+00,
-1.5224e+00, -1.1798e+00]]]], device='cuda:0'),
tensor([[[[ 2.2292e-01, 4.9591e-01, 6.2940e-01, ..., -3.4181e-01,
-9.0039e-02, 6.2186e-03],
[-2.0208e+00, -1.3204e+00, -2.8435e+00, ..., -3.2364e+00,
-1.3131e+00, -4.5985e-01],
[-1.3554e-01, -8.8875e-01, 1.1025e-02, ..., -1.8482e-01,
-3.8938e-01, 1.4619e-01],
...,
[-9.7595e-01, -2.1489e+00, -2.0699e+00, ..., -1.9039e+00,
-1.9011e+00, -1.2763e+00],
[ 7.9777e-01, -1.1640e-01, -1.3879e-01, ..., 5.1530e-02,
2.6831e-01, 4.4784e-01],
[ 2.0832e-01, -5.7187e-01, -4.0296e-01, ..., 2.1981e-01,
4.4758e-01, 4.9528e-01]],
[[-1.6989e+00, -1.3888e+00, -4.2700e+00, ..., -6.8586e+00,
-6.0766e+00, -4.5359e+00],
[ 4.6594e-01, 1.4617e+00, -2.5546e+00, ..., -5.4595e+00,
-5.5779e+00, -4.3087e+00],
[ 1.2005e+00, 1.1724e+00, -5.0978e+00, ..., -7.8527e+00,
-7.8245e+00, -6.6006e+00],
...,
[ 3.6952e-01, 3.4742e-01, -4.1302e+00, ..., -7.1475e+00,
-6.7242e+00, -5.2333e+00],
[ 2.5410e+00, 2.5088e+00, 1.4049e-01, ..., -2.0368e+00,
-2.6556e+00, -2.1272e+00],
[ 1.5132e+00, 1.3388e+00, -1.6228e+00, ..., -2.9806e+00,
-3.5066e+00, -3.0623e+00]],
[[-1.3951e-01, -1.1687e+00, -1.4182e+00, ..., -1.4114e+00,
-1.0479e+00, -1.3338e+00],
[-5.1101e-01, -8.2775e-01, -8.8592e-01, ..., -1.0045e+00,
-6.2138e-01, -8.9689e-01],
[-3.2903e-01, -2.1327e-01, -4.3628e-01, ..., -1.1072e+00,
-5.9965e-01, -4.6717e-01],
...,
[-6.5333e-02, -6.2125e-01, -6.2274e-01, ..., -1.0259e+00,
-8.6578e-01, -1.2363e+00],
[-7.0462e-01, -2.2693e-01, -5.7436e-01, ..., -5.7087e-01,
-2.6539e-01, -4.4627e-01],
[-8.5732e-01, -2.5844e-01, -4.6994e-01, ..., -4.9566e-01,
-4.4796e-01, -6.1116e-01]],
...,
[[-2.8538e+00, -2.4726e+00, -2.4658e+00, ..., -3.0667e+00,
-2.4977e+00, -1.8137e+00],
[-2.1019e+00, -8.5568e-01, -1.1285e+00, ..., -2.4288e+00,
-2.5142e+00, -2.0338e+00],
[-1.1221e+00, 7.7439e-01, 7.4872e-01, ..., -8.0658e-02,
1.7569e-02, 2.6237e-01],
...,
[-2.2867e+00, -2.1404e-01, -1.8227e-01, ..., -8.9765e-01,
-2.3613e-01, 2.7922e-01],
[-1.2355e+00, 8.5917e-01, 1.0536e+00, ..., 3.9534e-01,
4.8369e-01, 8.1228e-01],
[-2.1872e+00, -1.5185e+00, -1.7132e+00, ..., -2.1668e+00,
-1.1263e+00, -3.3529e-01]],
[[-1.6345e+00, -1.6273e+00, -2.7780e+00, ..., -2.2504e+00,
-1.9231e+00, -1.8163e+00],
[-6.9569e-01, -9.7127e-02, -3.1338e+00, ..., -2.8711e+00,
-2.8725e+00, -2.8813e+00],
[-2.0670e+00, -9.9552e-01, -5.1105e+00, ..., -4.4524e+00,
-4.5030e+00, -3.6984e+00],
...,
[-1.3324e+00, 1.8154e-01, -3.3465e+00, ..., -2.8958e+00,
-2.7963e+00, -2.3598e+00],
[-1.6722e+00, -4.6929e-01, -3.0020e+00, ..., -2.4765e+00,
-2.2939e+00, -1.7044e+00],
[-2.3366e+00, -1.3691e+00, -3.0517e+00, ..., -2.8552e+00,
-2.4347e+00, -2.2142e+00]],
[[-1.3052e+00, -1.8332e+00, -1.6572e+00, ..., -2.5725e+00,
-2.1471e+00, -1.6685e+00],
[-1.2413e+00, -1.7046e+00, -1.2267e+00, ..., -2.4023e+00,
-1.9327e+00, -1.5845e+00],
[-1.5541e+00, -1.8932e+00, -1.4804e+00, ..., -2.8197e+00,
-2.2504e+00, -1.7469e+00],
...,
[-2.4442e+00, -3.0529e+00, -2.7845e+00, ..., -4.0066e+00,
-3.2470e+00, -2.5455e+00],
[-1.5750e+00, -2.0765e+00, -1.9737e+00, ..., -2.7866e+00,
-2.3308e+00, -1.9231e+00],
[-1.0227e+00, -1.3336e+00, -1.3486e+00, ..., -1.8478e+00,
-1.5370e+00, -1.1877e+00]]],
[[[ 2.7348e-01, 5.2511e-01, 6.7792e-01, ..., 5.7814e-01,
1.1134e+00, 6.4011e-01],
[-2.0971e+00, -1.4869e+00, -3.1512e+00, ..., -2.7507e+00,
-8.8391e-01, -1.6796e-01],
[-1.8311e-01, -9.4951e-01, -2.4758e-01, ..., -3.3229e-01,
4.2644e-01, 6.0415e-01],
...,
[-1.9110e-01, -1.1519e+00, -2.6644e+00, ..., -1.4879e+00,
-1.3893e+00, -8.3076e-01],
[ 1.4569e+00, 8.4733e-01, -9.6189e-01, ..., -2.9070e-01,
1.1396e-01, 4.2072e-01],
[ 8.6051e-01, 2.9039e-01, -6.9913e-01, ..., -2.0017e-01,
1.6887e-01, 3.6808e-01]],
[[-1.6530e+00, -1.2592e+00, -4.2177e+00, ..., -5.0921e+00,
-4.6536e+00, -3.5130e+00],
[ 5.1140e-01, 1.6354e+00, -2.4603e+00, ..., -3.5715e+00,
-3.6666e+00, -2.9592e+00],
[ 1.1825e+00, 1.3592e+00, -5.0240e+00, ..., -5.5439e+00,
-5.3650e+00, -5.0631e+00],
...,
[-1.3551e+00, 9.7925e-02, -6.7732e+00, ..., -7.7417e+00,
-7.0068e+00, -5.4857e+00],
[ 1.7535e+00, 2.8073e+00, -1.0550e+00, ..., -2.6850e+00,
-3.0307e+00, -2.4367e+00],
[ 1.1874e+00, 1.8161e+00, -2.2699e+00, ..., -3.2907e+00,
-3.6551e+00, -3.1699e+00]],
[[-9.4286e-02, -1.0640e+00, -1.2688e+00, ..., -5.1379e-01,
-1.8757e-01, -5.4625e-01],
[-5.2001e-01, -7.7102e-01, -7.8643e-01, ..., -3.2751e-01,
-3.7246e-02, -1.3381e-01],
[-4.4231e-01, -2.6156e-01, -4.0960e-01, ..., -4.2866e-01,
3.6212e-02, 6.3224e-02],
...,
[-1.3515e+00, -1.7410e+00, -7.6980e-01, ..., -8.4295e-01,
-7.4670e-01, -1.0360e+00],
[-1.7029e+00, -1.1564e+00, -8.3700e-01, ..., -5.3213e-01,
-2.7313e-01, -3.5593e-01],
[-1.8716e+00, -1.2961e+00, -9.1658e-01, ..., -9.6793e-01,
-8.4071e-01, -8.1728e-01]],
...,
[[-2.8989e+00, -2.4674e+00, -2.7274e+00, ..., -3.6449e+00,
-2.8470e+00, -1.9533e+00],
[-2.2165e+00, -9.4603e-01, -1.5766e+00, ..., -2.8055e+00,
-2.4602e+00, -1.9616e+00],
[-1.0802e+00, 9.0301e-01, 4.8970e-01, ..., -4.5987e-01,
3.4670e-02, 1.8549e-01],
...,
[-1.8791e+00, 1.2189e+00, -7.2941e-04, ..., -1.0950e+00,
-5.1098e-01, 2.1505e-02],
[-8.4568e-01, 1.9129e+00, 1.6480e+00, ..., 4.3078e-01,
4.5257e-01, 7.0777e-01],
[-2.2711e+00, -1.1955e+00, -9.9690e-01, ..., -1.9130e+00,
-9.6902e-01, -2.7033e-01]],
[[-1.6161e+00, -1.6226e+00, -2.8218e+00, ..., -1.2212e+00,
-7.3850e-01, -7.2286e-01],
[-6.7366e-01, -8.4257e-02, -3.1872e+00, ..., -1.1960e+00,
-1.0119e+00, -1.4746e+00],
[-2.1273e+00, -1.0851e+00, -5.2780e+00, ..., -3.1238e+00,
-2.6277e+00, -2.1958e+00],
...,
[-2.4916e+00, -1.6088e+00, -5.4621e+00, ..., -3.1559e+00,
-2.7720e+00, -2.2224e+00],
[-2.2289e+00, -1.5087e+00, -4.2811e+00, ..., -2.7027e+00,
-2.3449e+00, -1.6770e+00],
[-2.3984e+00, -1.8178e+00, -3.3312e+00, ..., -2.9505e+00,
-2.4534e+00, -2.1381e+00]],
[[-1.3428e+00, -1.8739e+00, -1.7310e+00, ..., -1.8476e+00,
-1.5243e+00, -1.1650e+00],
[-1.2741e+00, -1.7427e+00, -1.3030e+00, ..., -1.4091e+00,
-1.1169e+00, -8.8664e-01],
[-1.5877e+00, -1.9314e+00, -1.5749e+00, ..., -1.6125e+00,
-1.1792e+00, -8.5684e-01],
...,
[-3.1806e+00, -3.9809e+00, -3.7748e+00, ..., -4.0289e+00,
-3.2269e+00, -2.4705e+00],
[-2.1141e+00, -2.7344e+00, -2.7027e+00, ..., -2.8267e+00,
-2.3367e+00, -1.8948e+00],
[-1.5393e+00, -1.8708e+00, -1.9320e+00, ..., -1.9113e+00,
-1.5538e+00, -1.1889e+00]]],
[[[ 1.0150e-01, 3.2729e-01, -1.3055e+00, ..., 4.8783e-01,
1.0082e+00, 6.5516e-01],
[-2.1546e+00, -1.5024e+00, -4.5466e+00, ..., -2.3669e+00,
-6.6192e-01, -1.2819e-02],
[ 4.1606e-01, -2.7456e-01, -1.2102e+00, ..., -2.1189e-01,
5.0124e-01, 6.7216e-01],
...,
[-2.3616e-01, -1.2792e+00, -2.6113e+00, ..., -2.1295e+00,
-1.5227e+00, -1.0466e+00],
[ 1.2821e+00, 6.5150e-01, -9.7356e-01, ..., -3.1893e-01,
5.1999e-01, 7.0167e-01],
[ 8.2194e-01, 2.7627e-01, -4.7069e-01, ..., -3.0650e-01,
3.4794e-01, 3.1622e-01]],
[[-3.3070e+00, -1.5243e+00, -6.2266e+00, ..., -4.7063e+00,
-4.3310e+00, -3.3223e+00],
[-6.2264e-01, 2.0899e+00, -4.2430e+00, ..., -3.4195e+00,
-3.5609e+00, -2.9175e+00],
[-5.0400e-02, 1.7750e+00, -7.4552e+00, ..., -5.3328e+00,
-5.2008e+00, -4.8434e+00],
...,
[-1.3024e+00, 2.2097e-01, -6.6345e+00, ..., -4.5174e+00,
-4.0893e+00, -3.3779e+00],
[ 1.4787e+00, 2.5717e+00, -1.4119e+00, ..., -5.3031e-01,
-9.2891e-01, -1.0184e+00],
[ 9.8090e-01, 1.6310e+00, -2.4314e+00, ..., -2.1459e+00,
-2.4305e+00, -2.3590e+00]],
[[-1.4210e+00, -2.3690e+00, -2.1946e+00, ..., -5.4906e-01,
-1.7893e-01, -4.2448e-01],
[-2.0407e+00, -2.2003e+00, -1.5786e+00, ..., -3.2479e-01,
5.7442e-02, 2.8444e-02],
[-2.5817e+00, -2.0200e+00, -8.4143e-01, ..., -2.6985e-01,
2.2191e-01, 2.6072e-01],
...,
[-1.6352e+00, -1.8437e+00, -7.7431e-01, ..., -5.1124e-01,
-1.9468e-01, -3.9034e-01],
[-1.7856e+00, -1.1963e+00, -7.6318e-01, ..., -2.8627e-01,
1.8106e-01, 2.3583e-01],
[-1.7492e+00, -1.1448e+00, -6.2956e-01, ..., -4.9044e-01,
-2.6315e-01, -2.4506e-01]],
...,
[[-2.3558e+00, -1.3163e+00, -2.7378e+00, ..., -3.4329e+00,
-2.6714e+00, -1.7473e+00],
[-1.9372e+00, 1.1860e-01, -1.3508e+00, ..., -2.5267e+00,
-2.3001e+00, -1.7864e+00],
[-5.3123e-01, 2.4150e+00, 9.5668e-01, ..., -5.7543e-01,
-8.5646e-02, 1.3557e-01],
...,
[-1.7657e+00, 1.1446e+00, 2.1744e-01, ..., -1.5883e+00,
-6.8774e-01, -6.2077e-02],
[-8.8150e-01, 1.6460e+00, 1.5160e+00, ..., -1.1140e-01,
1.3080e-01, 3.8380e-01],
[-2.1179e+00, -1.0652e+00, -8.2374e-01, ..., -2.5289e+00,
-1.5558e+00, -9.4793e-01]],
[[-2.4292e+00, -2.6979e+00, -4.0213e+00, ..., -1.2140e+00,
-8.0869e-01, -7.9632e-01],
[-1.7092e+00, -1.4049e+00, -4.7317e+00, ..., -1.4422e+00,
-1.2670e+00, -1.5724e+00],
[-3.0519e+00, -2.5007e+00, -6.8188e+00, ..., -3.2698e+00,
-2.7769e+00, -2.1967e+00],
...,
[-2.5697e+00, -1.8173e+00, -4.9953e+00, ..., -1.3561e+00,
-9.3626e-01, -8.1042e-01],
[-2.2502e+00, -1.6070e+00, -3.9052e+00, ..., -1.6663e+00,
-1.0595e+00, -6.8291e-01],
[-2.3226e+00, -1.8116e+00, -3.0940e+00, ..., -2.8292e+00,
-2.0099e+00, -1.8411e+00]],
[[-1.9308e+00, -2.6723e+00, -2.4653e+00, ..., -1.7524e+00,
-1.4558e+00, -1.1108e+00],
[-1.9524e+00, -2.6913e+00, -2.1647e+00, ..., -1.4001e+00,
-1.1111e+00, -8.8574e-01],
[-2.4793e+00, -3.0593e+00, -2.5901e+00, ..., -1.6220e+00,
-1.2170e+00, -9.0685e-01],
...,
[-3.1326e+00, -3.9183e+00, -3.6888e+00, ..., -2.9545e+00,
-2.3647e+00, -1.8185e+00],
[-2.0781e+00, -2.6917e+00, -2.6542e+00, ..., -2.0265e+00,
-1.6703e+00, -1.3562e+00],
[-1.4980e+00, -1.8213e+00, -1.8703e+00, ..., -1.2413e+00,
-1.0529e+00, -8.1222e-01]]],
[[[ 7.3680e-02, 2.1902e-01, -1.3912e+00, ..., -5.9849e-01,
-1.8132e-01, -3.4289e-02],
[-2.0665e+00, -1.5907e+00, -4.4849e+00, ..., -3.7296e+00,
-1.5105e+00, -5.3686e-01],
[ 6.0212e-01, -5.9452e-02, -8.1718e-01, ..., -1.6883e-01,
-2.6309e-01, 2.3208e-01],
...,
[-8.7596e-01, -1.8632e+00, -1.8807e+00, ..., -1.8346e+00,
-1.3624e+00, -9.4780e-01],
[ 7.3548e-01, -1.7703e-02, -1.3010e-01, ..., 2.8674e-03,
7.1021e-01, 8.1321e-01],
[ 1.2658e-01, -5.7664e-01, -4.8291e-01, ..., -1.5042e-01,
4.3843e-01, 3.4838e-01]],
[[-3.2786e+00, -1.7561e+00, -6.2431e+00, ..., -7.1557e+00,
-6.2475e+00, -4.6339e+00],
[-7.8909e-01, 1.5634e+00, -4.3675e+00, ..., -5.5137e+00,
-5.5414e+00, -4.2858e+00],
[-3.5251e-01, 1.0954e+00, -7.4283e+00, ..., -8.0211e+00,
-7.8377e+00, -6.6816e+00],
...,
[ 3.0989e-01, 3.7138e-01, -4.0235e+00, ..., -4.6564e+00,
-4.2015e+00, -3.5121e+00],
[ 2.3221e+00, 2.2925e+00, -1.3326e-01, ..., -8.7158e-01,
-1.2213e+00, -1.2754e+00],
[ 1.3618e+00, 1.1851e+00, -1.7469e+00, ..., -2.2742e+00,
-2.5403e+00, -2.4379e+00]],
[[-1.3320e+00, -2.3609e+00, -2.3150e+00, ..., -1.7494e+00,
-1.2424e+00, -1.3799e+00],
[-1.6746e+00, -1.9949e+00, -1.5343e+00, ..., -1.2507e+00,
-7.3864e-01, -8.1845e-01],
[-2.1578e+00, -1.9426e+00, -1.0251e+00, ..., -1.1204e+00,
-4.6724e-01, -1.8929e-01],
...,
[-1.3940e-01, -6.0121e-01, -4.6748e-01, ..., -4.8599e-01,
-1.7897e-01, -4.3062e-01],
[-7.0418e-01, -1.9527e-01, -4.2877e-01, ..., -2.0980e-01,
2.3201e-01, 2.2284e-01],
[-7.9996e-01, -1.8870e-01, -3.4574e-01, ..., -3.9829e-01,
-1.9620e-01, -2.3148e-01]],
...,
[[-2.0410e+00, -1.0780e+00, -2.3268e+00, ..., -3.2154e+00,
-2.6090e+00, -1.8270e+00],
[-1.7636e+00, 7.0232e-02, -1.3279e+00, ..., -2.5193e+00,
-2.4981e+00, -1.9459e+00],
[-3.8835e-01, 2.4204e+00, 1.0968e+00, ..., 1.0069e-01,
2.8366e-01, 5.4314e-01],
...,
[-2.1094e+00, -1.1786e-01, -1.8156e-01, ..., -1.3840e+00,
-3.9998e-01, 9.0675e-02],
[-1.2247e+00, 7.7507e-01, 9.2660e-01, ..., -2.9013e-02,
3.1058e-01, 4.9200e-01],
[-2.1061e+00, -1.4352e+00, -1.6611e+00, ..., -2.4281e+00,
-1.4427e+00, -9.0016e-01]],
[[-2.4188e+00, -2.8268e+00, -4.2377e+00, ..., -2.3935e+00,
-1.9105e+00, -1.7119e+00],
[-1.7795e+00, -1.6525e+00, -4.9263e+00, ..., -2.9089e+00,
-2.7789e+00, -2.7550e+00],
[-2.8289e+00, -2.4752e+00, -6.7169e+00, ..., -4.4629e+00,
-4.3601e+00, -3.4752e+00],
...,
[-1.4003e+00, -2.0707e-02, -3.3658e+00, ..., -1.4578e+00,
-1.0616e+00, -9.6697e-01],
[-1.7149e+00, -6.3141e-01, -3.0314e+00, ..., -1.7289e+00,
-1.1953e+00, -8.6991e-01],
[-2.3342e+00, -1.4864e+00, -3.1040e+00, ..., -2.9569e+00,
-2.1575e+00, -2.0209e+00]],
[[-1.8028e+00, -2.5290e+00, -2.4378e+00, ..., -2.6975e+00,
-2.2146e+00, -1.6959e+00],
[-1.8050e+00, -2.5115e+00, -2.1438e+00, ..., -2.4008e+00,
-1.9055e+00, -1.5397e+00],
[-2.2596e+00, -2.8210e+00, -2.5115e+00, ..., -2.7434e+00,
-2.1423e+00, -1.6260e+00],
...,
[-2.3909e+00, -2.9644e+00, -2.7430e+00, ..., -2.8546e+00,
-2.2883e+00, -1.7504e+00],
[-1.5480e+00, -2.0158e+00, -1.9402e+00, ..., -1.9519e+00,
-1.6134e+00, -1.2963e+00],
[-1.0052e+00, -1.3014e+00, -1.3353e+00, ..., -1.2011e+00,
-1.0276e+00, -7.7837e-01]]],
[[[ 1.0032e-01, 2.4382e-01, -1.3921e+00, ..., -6.1849e-01,
-2.1808e-01, -5.7212e-02],
[-2.1833e+00, -1.6980e+00, -4.6793e+00, ..., -3.6184e+00,
-1.4545e+00, -5.2632e-01],
[ 6.3261e-01, -1.0867e-01, -8.0745e-01, ..., -1.6425e-01,
-2.3625e-01, 2.7630e-01],
...,
[-3.2336e-01, -1.1096e+00, -1.4415e+00, ..., -1.3444e+00,
-8.7637e-01, -6.5176e-01],
[ 9.7694e-01, 5.7700e-01, 3.2998e-01, ..., 3.7100e-01,
6.8891e-01, 8.1132e-01],
[ 6.8619e-01, 4.4557e-01, 3.3086e-01, ..., 5.0438e-01,
7.6561e-01, 7.0478e-01]],
[[-3.3592e+00, -1.6686e+00, -6.2863e+00, ..., -6.9978e+00,
-6.1387e+00, -4.5495e+00],
[-6.2242e-01, 1.9942e+00, -4.1851e+00, ..., -5.5083e+00,
-5.5450e+00, -4.2831e+00],
[-4.3389e-02, 1.5967e+00, -7.3519e+00, ..., -7.9673e+00,
-7.7976e+00, -6.6005e+00],
...,
[-4.9195e-01, 2.1496e-01, -4.4482e+00, ..., -5.3577e+00,
-4.6536e+00, -3.8955e+00],
[ 1.9262e+00, 2.4095e+00, -7.2574e-02, ..., -1.0814e+00,
-1.2293e+00, -1.3759e+00],
[ 1.0976e+00, 1.2961e+00, -1.6762e+00, ..., -2.4721e+00,
-2.5828e+00, -2.5738e+00]],
[[-1.4098e+00, -2.5333e+00, -2.4768e+00, ..., -1.7002e+00,
-1.2180e+00, -1.3386e+00],
[-1.8318e+00, -2.1686e+00, -1.6113e+00, ..., -1.1893e+00,
-7.0289e-01, -8.0356e-01],
[-2.2797e+00, -2.0383e+00, -1.0274e+00, ..., -1.0847e+00,
-4.6819e-01, -2.0391e-01],
...,
[-6.3538e-01, -1.0816e+00, -5.7286e-01, ..., -6.0443e-01,
-4.7440e-01, -6.3950e-01],
[-1.0276e+00, -6.8300e-01, -5.6546e-01, ..., -2.6342e-01,
-6.1228e-02, -3.7032e-02],
[-1.1702e+00, -6.4764e-01, -3.8042e-01, ..., -3.4078e-01,
-2.8671e-01, -2.5894e-01]],
...,
[[-2.2922e+00, -1.1843e+00, -2.5344e+00, ..., -2.9850e+00,
-2.4640e+00, -1.7224e+00],
[-1.8446e+00, 1.9578e-01, -1.3958e+00, ..., -2.4121e+00,
-2.4464e+00, -1.8849e+00],
[-3.8826e-01, 2.6511e+00, 1.1520e+00, ..., 1.1413e-01,
2.1332e-01, 4.6805e-01],
...,
[-1.5604e+00, 9.1507e-01, -1.5556e-01, ..., -1.1875e+00,
-4.9439e-01, 4.1035e-02],
[-5.6970e-01, 1.3531e+00, 7.4266e-01, ..., 1.5581e-02,
1.7663e-01, 3.1164e-01],
[-1.6271e+00, -1.1133e+00, -1.5554e+00, ..., -2.2963e+00,
-1.3546e+00, -6.9210e-01]],
[[-2.4080e+00, -2.7807e+00, -4.3571e+00, ..., -2.4850e+00,
-2.0125e+00, -1.7934e+00],
[-1.6506e+00, -1.4479e+00, -5.1186e+00, ..., -3.0069e+00,
-2.8629e+00, -2.7907e+00],
[-2.8025e+00, -2.3410e+00, -7.0674e+00, ..., -4.4319e+00,
-4.3612e+00, -3.4790e+00],
...,
[-1.7330e+00, -5.7327e-01, -4.0294e+00, ..., -2.0287e+00,
-1.5576e+00, -1.2129e+00],
[-1.7067e+00, -7.6061e-01, -3.3364e+00, ..., -2.0294e+00,
-1.3506e+00, -8.1817e-01],
[-2.0701e+00, -1.3065e+00, -2.8265e+00, ..., -2.6777e+00,
-1.7970e+00, -1.5356e+00]],
[[-1.8572e+00, -2.6120e+00, -2.4768e+00, ..., -2.6471e+00,
-2.1741e+00, -1.6681e+00],
[-1.8440e+00, -2.5774e+00, -2.1210e+00, ..., -2.4025e+00,
-1.9008e+00, -1.5383e+00],
[-2.3087e+00, -2.8825e+00, -2.4818e+00, ..., -2.7643e+00,
-2.1605e+00, -1.6460e+00],
...,
[-2.2642e+00, -2.8431e+00, -2.6631e+00, ..., -2.8770e+00,
-2.2991e+00, -1.7321e+00],
[-1.5273e+00, -1.9649e+00, -1.9310e+00, ..., -2.0347e+00,
-1.6702e+00, -1.3298e+00],
[-1.0308e+00, -1.2626e+00, -1.3352e+00, ..., -1.2656e+00,
-1.0776e+00, -7.9498e-01]]]], device='cuda:0'),
tensor([[[[ 4.9798e-01, 9.8495e-01, 8.8794e-01, ..., -4.7669e-01,
-2.1603e-01, -7.2579e-02],
[-1.2277e+00, -2.5146e-01, -1.4656e+00, ..., -3.4945e+00,
-1.4835e+00, -5.6605e-01],
[ 7.0764e-01, 4.6783e-01, 4.0194e-01, ..., -2.6451e-01,
-4.7752e-01, 1.2604e-01],
...,
[-1.5698e-01, -7.9743e-01, -1.3596e+00, ..., -1.9666e+00,
-1.9006e+00, -1.2324e+00],
[ 1.0562e+00, 7.9299e-01, 3.3756e-01, ..., -1.2150e-01,
1.8203e-01, 4.2915e-01],
[ 7.3007e-01, 5.4680e-01, 2.7638e-01, ..., 7.9228e-02,
3.8047e-01, 4.6770e-01]],
[[-2.3718e+00, -9.9727e-01, -4.1955e+00, ..., -7.0286e+00,
-6.2252e+00, -4.6230e+00],
[-1.6698e-01, 2.0458e+00, -2.4794e+00, ..., -5.6185e+00,
-5.7226e+00, -4.3928e+00],
[ 6.4840e-01, 1.8386e+00, -5.1965e+00, ..., -8.0736e+00,
-8.0312e+00, -6.7132e+00],
...,
[-5.4716e-01, 4.6541e-01, -4.2851e+00, ..., -7.3226e+00,
-6.9196e+00, -5.3624e+00],
[ 1.9228e+00, 2.5528e+00, -5.4286e-04, ..., -2.0681e+00,
-2.7301e+00, -2.1582e+00],
[ 1.1083e+00, 1.3666e+00, -1.6798e+00, ..., -3.0050e+00,
-3.5804e+00, -3.1071e+00]],
[[-6.4484e-01, -1.7512e+00, -1.5571e+00, ..., -1.4933e+00,
-1.1182e+00, -1.4107e+00],
[-1.4543e+00, -1.8227e+00, -1.0950e+00, ..., -1.0504e+00,
-6.6793e-01, -9.7817e-01],
[-1.2904e+00, -1.0926e+00, -3.9572e-01, ..., -1.1581e+00,
-6.2978e-01, -4.9685e-01],
...,
[-9.2066e-01, -1.3142e+00, -5.2247e-01, ..., -1.0393e+00,
-8.9302e-01, -1.2896e+00],
[-1.2360e+00, -8.2959e-01, -5.1789e-01, ..., -5.5184e-01,
-2.3494e-01, -4.2997e-01],
[-1.3130e+00, -7.1505e-01, -3.1432e-01, ..., -5.9583e-01,
-5.4362e-01, -7.0976e-01]],
...,
[[-2.3931e+00, -1.2551e+00, -2.6385e+00, ..., -3.1001e+00,
-2.5367e+00, -1.8470e+00],
[-7.0344e-01, 1.1986e+00, -8.6934e-01, ..., -2.5558e+00,
-2.6221e+00, -2.1051e+00],
[-2.4036e-01, 2.5069e+00, 6.0158e-01, ..., -9.9857e-02,
-1.8674e-02, 2.6362e-01],
...,
[-1.3728e+00, 1.3122e+00, -1.2659e-01, ..., -9.1546e-01,
-2.8703e-01, 2.6053e-01],
[-4.3430e-01, 1.5960e+00, 8.1025e-01, ..., 5.3260e-01,
5.1392e-01, 8.6546e-01],
[-1.4959e+00, -9.3149e-01, -1.4254e+00, ..., -2.1081e+00,
-1.0955e+00, -2.8506e-01]],
[[-1.2984e+00, -1.1758e+00, -2.7488e+00, ..., -2.3496e+00,
-2.0463e+00, -1.9424e+00],
[-5.5348e-01, -1.6970e-03, -3.5908e+00, ..., -2.9792e+00,
-3.0193e+00, -3.0161e+00],
[-2.1031e+00, -1.3381e+00, -6.0738e+00, ..., -4.5235e+00,
-4.6471e+00, -3.8339e+00],
...,
[-1.6807e+00, -6.3584e-01, -4.1909e+00, ..., -2.8565e+00,
-2.8364e+00, -2.4019e+00],
[-1.6609e+00, -8.0150e-01, -3.4739e+00, ..., -2.4586e+00,
-2.3487e+00, -1.7555e+00],
[-2.0544e+00, -1.3575e+00, -2.9332e+00, ..., -2.8608e+00,
-2.4907e+00, -2.2646e+00]],
[[-1.2064e+00, -1.7877e+00, -1.5509e+00, ..., -2.6736e+00,
-2.2312e+00, -1.7313e+00],
[-1.2050e+00, -1.6982e+00, -1.2002e+00, ..., -2.5116e+00,
-2.0179e+00, -1.6521e+00],
[-1.5662e+00, -1.8686e+00, -1.4057e+00, ..., -2.9490e+00,
-2.3590e+00, -1.8350e+00],
...,
[-2.3014e+00, -2.8546e+00, -2.6544e+00, ..., -4.1231e+00,
-3.3400e+00, -2.6263e+00],
[-1.5569e+00, -1.9683e+00, -1.9115e+00, ..., -2.8418e+00,
-2.3742e+00, -1.9688e+00],
[-1.0529e+00, -1.2523e+00, -1.3165e+00, ..., -1.8904e+00,
-1.5615e+00, -1.2160e+00]]],
[[[ 6.7126e-02, 2.3234e-01, -1.3810e+00, ..., -6.2958e-01,
-1.9960e-01, -7.0017e-02],
[-2.0882e+00, -1.5875e+00, -4.4825e+00, ..., -3.7953e+00,
-1.5299e+00, -5.8675e-01],
[ 5.4534e-01, -8.5865e-02, -8.2107e-01, ..., -2.1890e-01,
-2.7056e-01, 2.0892e-01],
...,
[-8.2837e-01, -1.8170e+00, -1.8713e+00, ..., -1.8161e+00,
-1.3665e+00, -9.9636e-01],
[ 7.5864e-01, 4.1342e-02, -8.1951e-02, ..., -1.7696e-02,
6.9281e-01, 7.7020e-01],
[ 1.5091e-01, -5.5324e-01, -4.7985e-01, ..., -1.6273e-01,
4.3281e-01, 3.2998e-01]],
[[-3.2834e+00, -1.7763e+00, -6.2595e+00, ..., -7.1567e+00,
-6.2755e+00, -4.6242e+00],
[-7.9064e-01, 1.5385e+00, -4.3559e+00, ..., -5.4763e+00,
-5.5538e+00, -4.2661e+00],
[-3.5346e-01, 1.1004e+00, -7.4118e+00, ..., -7.9871e+00,
-7.8500e+00, -6.6847e+00],
...,
[ 2.6673e-01, 2.8954e-01, -4.1047e+00, ..., -4.6455e+00,
-4.1842e+00, -3.4425e+00],
[ 2.2324e+00, 2.1750e+00, -2.8223e-01, ..., -8.0443e-01,
-1.1534e+00, -1.1621e+00],
[ 1.3223e+00, 1.1223e+00, -1.8302e+00, ..., -2.2270e+00,
-2.4885e+00, -2.3721e+00]],
[[-1.3203e+00, -2.3389e+00, -2.3127e+00, ..., -1.7634e+00,
-1.2489e+00, -1.3775e+00],
[-1.6778e+00, -1.9826e+00, -1.5296e+00, ..., -1.2650e+00,
-7.3509e-01, -8.0814e-01],
[-2.1554e+00, -1.9400e+00, -1.0239e+00, ..., -1.1171e+00,
-4.6905e-01, -1.8580e-01],
...,
[-1.1002e-01, -5.7271e-01, -5.0571e-01, ..., -4.9906e-01,
-1.9508e-01, -4.5617e-01],
[-6.7795e-01, -1.6432e-01, -4.2308e-01, ..., -2.0192e-01,
2.6744e-01, 2.3814e-01],
[-7.5782e-01, -1.3985e-01, -3.1478e-01, ..., -3.9778e-01,
-1.7742e-01, -2.2915e-01]],
...,
[[-2.0133e+00, -1.0848e+00, -2.3535e+00, ..., -3.1855e+00,
-2.6290e+00, -1.8282e+00],
[-1.7523e+00, 3.3112e-02, -1.3717e+00, ..., -2.4823e+00,
-2.5242e+00, -1.9716e+00],
[-3.6589e-01, 2.3994e+00, 1.0634e+00, ..., 1.3818e-01,
2.8526e-01, 5.4332e-01],
...,
[-2.0786e+00, -1.4210e-01, -1.4789e-01, ..., -1.3711e+00,
-3.9726e-01, 1.1360e-01],
[-1.2430e+00, 6.9307e-01, 9.1839e-01, ..., 9.1697e-03,
2.9680e-01, 4.8921e-01],
[-2.1279e+00, -1.4527e+00, -1.6376e+00, ..., -2.4685e+00,
-1.4944e+00, -9.4023e-01]],
[[-2.4186e+00, -2.8327e+00, -4.2410e+00, ..., -2.3980e+00,
-1.8841e+00, -1.6869e+00],
[-1.7730e+00, -1.6537e+00, -4.9256e+00, ..., -2.9087e+00,
-2.7412e+00, -2.7475e+00],
[-2.8218e+00, -2.4652e+00, -6.7163e+00, ..., -4.4731e+00,
-4.3406e+00, -3.4747e+00],
...,
[-1.4562e+00, -1.8031e-01, -3.4258e+00, ..., -1.3983e+00,
-9.9415e-01, -9.0607e-01],
[-1.7297e+00, -7.1570e-01, -3.0401e+00, ..., -1.7085e+00,
-1.1368e+00, -7.9426e-01],
[-2.3070e+00, -1.4917e+00, -3.1291e+00, ..., -2.9605e+00,
-2.1268e+00, -1.9885e+00]],
[[-1.8051e+00, -2.5320e+00, -2.4446e+00, ..., -2.7035e+00,
-2.2220e+00, -1.7010e+00],
[-1.8025e+00, -2.5126e+00, -2.1434e+00, ..., -2.4073e+00,
-1.9088e+00, -1.5450e+00],
[-2.2646e+00, -2.8249e+00, -2.5141e+00, ..., -2.7478e+00,
-2.1430e+00, -1.6269e+00],
...,
[-2.3773e+00, -2.9459e+00, -2.7446e+00, ..., -2.8595e+00,
-2.2932e+00, -1.7621e+00],
[-1.5527e+00, -2.0232e+00, -1.9716e+00, ..., -1.9372e+00,
-1.6026e+00, -1.2923e+00],
[-1.0184e+00, -1.3153e+00, -1.3587e+00, ..., -1.1814e+00,
-1.0112e+00, -7.6907e-01]]],
[[[ 2.0266e-01, 4.7497e-01, 6.3857e-01, ..., -3.2920e-01,
-6.2689e-02, 2.6665e-02],
[-2.0148e+00, -1.3282e+00, -2.8251e+00, ..., -3.1981e+00,
-1.2763e+00, -4.2406e-01],
[-1.6237e-01, -8.9598e-01, 1.3342e-02, ..., -1.8855e-01,
-3.7545e-01, 1.6273e-01],
...,
[-9.8755e-01, -2.1962e+00, -2.1202e+00, ..., -1.9228e+00,
-1.9100e+00, -1.2828e+00],
[ 8.0894e-01, -1.1798e-01, -1.3361e-01, ..., 2.6364e-02,
2.6627e-01, 4.5597e-01],
[ 2.2211e-01, -5.6835e-01, -3.9878e-01, ..., 2.2067e-01,
4.6601e-01, 5.0570e-01]],
[[-1.6431e+00, -1.3918e+00, -4.2545e+00, ..., -6.8572e+00,
-6.0606e+00, -4.5154e+00],
[ 4.8492e-01, 1.4286e+00, -2.5412e+00, ..., -5.4761e+00,
-5.5742e+00, -4.2929e+00],
[ 1.1834e+00, 1.1573e+00, -5.0368e+00, ..., -7.8873e+00,
-7.8239e+00, -6.5932e+00],
...,
[ 4.2224e-01, 4.1425e-01, -4.0952e+00, ..., -7.0912e+00,
-6.6547e+00, -5.1774e+00],
[ 2.6013e+00, 2.5614e+00, 1.7910e-01, ..., -1.9046e+00,
-2.5446e+00, -2.0477e+00],
[ 1.5498e+00, 1.3633e+00, -1.5999e+00, ..., -2.9076e+00,
-3.4745e+00, -3.0336e+00]],
[[-1.3989e-01, -1.1496e+00, -1.3710e+00, ..., -1.3975e+00,
-1.0353e+00, -1.3295e+00],
[-4.9401e-01, -7.9611e-01, -8.5334e-01, ..., -9.8532e-01,
-5.9740e-01, -8.9042e-01],
[-3.2604e-01, -2.1243e-01, -4.1530e-01, ..., -1.1204e+00,
-6.0141e-01, -4.7239e-01],
...,
[-6.1841e-02, -6.2568e-01, -6.2774e-01, ..., -1.0359e+00,
-8.6806e-01, -1.2410e+00],
[-7.1893e-01, -2.3875e-01, -5.9440e-01, ..., -5.5890e-01,
-2.4516e-01, -4.1313e-01],
[-8.8440e-01, -2.7561e-01, -4.8876e-01, ..., -4.9240e-01,
-4.3220e-01, -5.9781e-01]],
...,
[[-2.8202e+00, -2.4561e+00, -2.4302e+00, ..., -3.0833e+00,
-2.4871e+00, -1.7999e+00],
[-2.1022e+00, -8.8041e-01, -1.1415e+00, ..., -2.4442e+00,
-2.5082e+00, -2.0087e+00],
[-1.1353e+00, 7.4437e-01, 7.1342e-01, ..., -8.0947e-02,
2.5117e-02, 2.7359e-01],
...,
[-2.3081e+00, -1.6952e-01, -1.2966e-01, ..., -9.5755e-01,
-2.5205e-01, 2.9124e-01],
[-1.2525e+00, 9.1348e-01, 1.1132e+00, ..., 3.9957e-01,
4.8522e-01, 8.2378e-01],
[-2.2041e+00, -1.4888e+00, -1.6652e+00, ..., -2.2393e+00,
-1.1891e+00, -3.6559e-01]],
[[-1.6456e+00, -1.6494e+00, -2.8117e+00, ..., -2.2400e+00,
-1.9227e+00, -1.8106e+00],
[-7.2714e-01, -1.3919e-01, -3.1471e+00, ..., -2.8746e+00,
-2.8761e+00, -2.8779e+00],
[-2.0927e+00, -1.0094e+00, -5.0945e+00, ..., -4.4627e+00,
-4.5019e+00, -3.6957e+00],
...,
[-1.3311e+00, 2.5362e-01, -3.3636e+00, ..., -2.8268e+00,
-2.7258e+00, -2.3025e+00],
[-1.6717e+00, -4.3624e-01, -3.0190e+00, ..., -2.4380e+00,
-2.2469e+00, -1.6570e+00],
[-2.3292e+00, -1.3492e+00, -3.0400e+00, ..., -2.8688e+00,
-2.4169e+00, -2.1908e+00]],
[[-1.2962e+00, -1.8168e+00, -1.6497e+00, ..., -2.5608e+00,
-2.1405e+00, -1.6643e+00],
[-1.2389e+00, -1.7000e+00, -1.2251e+00, ..., -2.3961e+00,
-1.9316e+00, -1.5851e+00],
[-1.5517e+00, -1.8979e+00, -1.4820e+00, ..., -2.8214e+00,
-2.2548e+00, -1.7480e+00],
...,
[-2.4643e+00, -3.0785e+00, -2.7995e+00, ..., -3.9915e+00,
-3.2433e+00, -2.5474e+00],
[-1.5946e+00, -2.0969e+00, -1.9876e+00, ..., -2.7557e+00,
-2.3125e+00, -1.9139e+00],
[-1.0397e+00, -1.3544e+00, -1.3655e+00, ..., -1.8225e+00,
-1.5200e+00, -1.1802e+00]]],
[[[ 8.1203e-02, 2.3763e-01, -1.3725e+00, ..., -5.8849e-01,
-1.7538e-01, -5.4670e-02],
[-2.0751e+00, -1.5744e+00, -4.4765e+00, ..., -3.7361e+00,
-1.4877e+00, -5.6200e-01],
[ 5.5561e-01, -1.0193e-01, -8.1892e-01, ..., -1.7750e-01,
-2.1829e-01, 2.3707e-01],
...,
[-8.8860e-01, -1.8988e+00, -1.8990e+00, ..., -1.8194e+00,
-1.3456e+00, -9.4131e-01],
[ 7.4577e-01, -5.6162e-02, -1.3955e-01, ..., -1.9185e-02,
7.1014e-01, 8.0300e-01],
[ 1.3171e-01, -6.2017e-01, -4.9323e-01, ..., -1.5817e-01,
4.3515e-01, 3.3820e-01]],
[[-3.2716e+00, -1.7518e+00, -6.2511e+00, ..., -7.1458e+00,
-6.2656e+00, -4.6060e+00],
[-7.6836e-01, 1.5822e+00, -4.3580e+00, ..., -5.5003e+00,
-5.5737e+00, -4.2611e+00],
[-3.2265e-01, 1.1189e+00, -7.4351e+00, ..., -8.0032e+00,
-7.8685e+00, -6.6629e+00],
...,
[ 2.9996e-01, 3.8535e-01, -3.9948e+00, ..., -4.6308e+00,
-4.1970e+00, -3.5076e+00],
[ 2.3386e+00, 2.3263e+00, -8.9368e-02, ..., -7.9183e-01,
-1.1664e+00, -1.2144e+00],
[ 1.3789e+00, 1.2039e+00, -1.7420e+00, ..., -2.2280e+00,
-2.5222e+00, -2.4052e+00]],
[[-1.3304e+00, -2.3619e+00, -2.3216e+00, ..., -1.7493e+00,
-1.2323e+00, -1.3670e+00],
[-1.6834e+00, -1.9924e+00, -1.5201e+00, ..., -1.2459e+00,
-7.1429e-01, -7.9429e-01],
[-2.1652e+00, -1.9392e+00, -1.0134e+00, ..., -1.1018e+00,
-4.5245e-01, -1.7943e-01],
...,
[-1.5993e-01, -6.0842e-01, -4.6950e-01, ..., -4.9705e-01,
-2.0159e-01, -4.5169e-01],
[-7.2599e-01, -1.9448e-01, -4.2252e-01, ..., -2.0825e-01,
2.5452e-01, 2.4822e-01],
[-8.1554e-01, -1.9005e-01, -3.4164e-01, ..., -4.0594e-01,
-1.9130e-01, -2.2372e-01]],
...,
[[-2.0292e+00, -1.0737e+00, -2.3669e+00, ..., -3.2070e+00,
-2.6288e+00, -1.8197e+00],
[-1.7471e+00, 6.2867e-02, -1.3848e+00, ..., -2.4739e+00,
-2.5409e+00, -1.9813e+00],
[-3.4445e-01, 2.4082e+00, 1.0328e+00, ..., 1.2381e-01,
2.4910e-01, 5.1418e-01],
...,
[-2.0819e+00, -7.6551e-02, -1.9042e-01, ..., -1.3970e+00,
-3.8114e-01, 1.0952e-01],
[-1.1987e+00, 8.3359e-01, 9.5919e-01, ..., -1.9789e-03,
3.1364e-01, 4.8636e-01],
[-2.1152e+00, -1.4282e+00, -1.6442e+00, ..., -2.4955e+00,
-1.4960e+00, -9.4142e-01]],
[[-2.4082e+00, -2.8216e+00, -4.2449e+00, ..., -2.4110e+00,
-1.9151e+00, -1.7125e+00],
[-1.7548e+00, -1.6219e+00, -4.9220e+00, ..., -2.9372e+00,
-2.7788e+00, -2.7699e+00],
[-2.8129e+00, -2.4398e+00, -6.7237e+00, ..., -4.4774e+00,
-4.3572e+00, -3.4926e+00],
...,
[-1.3767e+00, 2.9393e-03, -3.3603e+00, ..., -1.4049e+00,
-1.0166e+00, -9.4057e-01],
[-1.6985e+00, -6.1289e-01, -3.0190e+00, ..., -1.7218e+00,
-1.1554e+00, -8.1661e-01],
[-2.3267e+00, -1.4781e+00, -3.1035e+00, ..., -2.9733e+00,
-2.1292e+00, -1.9931e+00]],
[[-1.8046e+00, -2.5348e+00, -2.4430e+00, ..., -2.6904e+00,
-2.2147e+00, -1.6950e+00],
[-1.8006e+00, -2.5113e+00, -2.1387e+00, ..., -2.3987e+00,
-1.9091e+00, -1.5426e+00],
[-2.2604e+00, -2.8197e+00, -2.5048e+00, ..., -2.7537e+00,
-2.1527e+00, -1.6292e+00],
...,
[-2.3954e+00, -2.9756e+00, -2.7422e+00, ..., -2.8686e+00,
-2.2986e+00, -1.7637e+00],
[-1.5531e+00, -2.0238e+00, -1.9451e+00, ..., -1.9453e+00,
-1.6118e+00, -1.3014e+00],
[-1.0062e+00, -1.3082e+00, -1.3345e+00, ..., -1.1929e+00,
-1.0188e+00, -7.7498e-01]]],
[[[ 2.1691e-01, 4.9704e-01, 6.4840e-01, ..., -3.4114e-01,
-7.9128e-02, 1.7366e-02],
[-2.0453e+00, -1.3416e+00, -2.8377e+00, ..., -3.2538e+00,
-1.3014e+00, -4.4447e-01],
[-1.5634e-01, -9.3556e-01, 1.5862e-02, ..., -1.7442e-01,
-3.6729e-01, 1.5765e-01],
...,
[-9.6756e-01, -2.1837e+00, -2.1039e+00, ..., -1.9321e+00,
-1.9058e+00, -1.2512e+00],
[ 7.8497e-01, -1.4768e-01, -1.6140e-01, ..., 6.4712e-02,
2.8863e-01, 4.6427e-01],
[ 2.0598e-01, -5.7738e-01, -4.1243e-01, ..., 2.3951e-01,
4.6714e-01, 5.1645e-01]],
[[-1.6563e+00, -1.3596e+00, -4.2457e+00, ..., -6.8727e+00,
-6.0782e+00, -4.5321e+00],
[ 4.8431e-01, 1.4887e+00, -2.5413e+00, ..., -5.4798e+00,
-5.5728e+00, -4.3108e+00],
[ 1.2147e+00, 1.2127e+00, -5.0875e+00, ..., -7.9017e+00,
-7.8357e+00, -6.6162e+00],
...,
[ 4.0298e-01, 3.6076e-01, -4.1092e+00, ..., -7.1455e+00,
-6.7393e+00, -5.2447e+00],
[ 2.5602e+00, 2.4987e+00, 1.3793e-01, ..., -1.9797e+00,
-2.6386e+00, -2.1169e+00],
[ 1.5209e+00, 1.3279e+00, -1.6121e+00, ..., -2.9481e+00,
-3.5223e+00, -3.0856e+00]],
[[-1.1938e-01, -1.1636e+00, -1.4193e+00, ..., -1.4102e+00,
-1.0509e+00, -1.3384e+00],
[-5.1034e-01, -8.3232e-01, -9.0192e-01, ..., -1.0004e+00,
-6.1724e-01, -9.0018e-01],
[-3.4734e-01, -2.3186e-01, -4.4543e-01, ..., -1.1106e+00,
-5.9702e-01, -4.7353e-01],
...,
[-7.1470e-02, -6.2029e-01, -6.1514e-01, ..., -1.0331e+00,
-9.0082e-01, -1.2587e+00],
[-7.1455e-01, -2.3015e-01, -5.7893e-01, ..., -5.8311e-01,
-2.7429e-01, -4.3269e-01],
[-8.7834e-01, -2.7343e-01, -4.8764e-01, ..., -5.0910e-01,
-4.6323e-01, -6.1206e-01]],
...,
[[-2.8429e+00, -2.4516e+00, -2.4417e+00, ..., -3.0985e+00,
-2.5193e+00, -1.8273e+00],
[-2.1087e+00, -8.3650e-01, -1.0893e+00, ..., -2.4415e+00,
-2.5284e+00, -2.0300e+00],
[-1.1045e+00, 8.2961e-01, 7.9102e-01, ..., -8.1885e-02,
1.6699e-02, 2.7755e-01],
...,
[-2.2671e+00, -1.4744e-01, -1.1530e-01, ..., -9.4954e-01,
-2.5313e-01, 3.0155e-01],
[-1.2248e+00, 8.9514e-01, 1.0970e+00, ..., 4.6568e-01,
5.0589e-01, 8.2006e-01],
[-2.1743e+00, -1.4825e+00, -1.6820e+00, ..., -2.1680e+00,
-1.1146e+00, -3.3762e-01]],
[[-1.6277e+00, -1.6301e+00, -2.8101e+00, ..., -2.2315e+00,
-1.9194e+00, -1.8222e+00],
[-7.0431e-01, -1.0126e-01, -3.1722e+00, ..., -2.8749e+00,
-2.8795e+00, -2.9025e+00],
[-2.0678e+00, -9.8521e-01, -5.1371e+00, ..., -4.4711e+00,
-4.5143e+00, -3.7151e+00],
...,
[-1.3185e+00, 2.2900e-01, -3.3498e+00, ..., -2.8712e+00,
-2.7853e+00, -2.3473e+00],
[-1.6663e+00, -4.3823e-01, -3.0004e+00, ..., -2.4778e+00,
-2.2844e+00, -1.6929e+00],
[-2.3309e+00, -1.3553e+00, -3.0251e+00, ..., -2.8850e+00,
-2.4433e+00, -2.2057e+00]],
[[-1.3107e+00, -1.8349e+00, -1.6589e+00, ..., -2.5718e+00,
-2.1480e+00, -1.6675e+00],
[-1.2534e+00, -1.7111e+00, -1.2322e+00, ..., -2.3968e+00,
-1.9299e+00, -1.5825e+00],
[-1.5783e+00, -1.9128e+00, -1.4957e+00, ..., -2.8171e+00,
-2.2435e+00, -1.7462e+00],
...,
[-2.4492e+00, -3.0551e+00, -2.7895e+00, ..., -4.0107e+00,
-3.2471e+00, -2.5426e+00],
[-1.5779e+00, -2.0788e+00, -1.9777e+00, ..., -2.7817e+00,
-2.3266e+00, -1.9171e+00],
[-1.0252e+00, -1.3336e+00, -1.3562e+00, ..., -1.8490e+00,
-1.5327e+00, -1.1835e+00]]]], device='cuda:0'),
tensor([[[[ 1.0797e-01, 2.8010e-01, -1.3420e+00, ..., -6.1029e-01,
-1.8527e-01, -3.2169e-02],
[-2.1824e+00, -1.6524e+00, -4.6490e+00, ..., -3.5893e+00,
-1.4088e+00, -4.9633e-01],
[ 6.3260e-01, -1.2411e-01, -8.7949e-01, ..., -1.8281e-01,
-2.3041e-01, 2.6224e-01],
...,
[-8.7260e-01, -1.8406e+00, -1.8322e+00, ..., -1.8922e+00,
-1.3525e+00, -9.7952e-01],
[ 7.3621e-01, -7.1748e-03, -7.7857e-02, ..., -7.8432e-02,
7.0251e-01, 7.8462e-01],
[ 1.3697e-01, -5.8869e-01, -4.7329e-01, ..., -1.9717e-01,
4.3562e-01, 3.4066e-01]],
[[-3.3341e+00, -1.6404e+00, -6.2436e+00, ..., -6.9803e+00,
-6.1205e+00, -4.5240e+00],
[-6.0982e-01, 1.9921e+00, -4.1389e+00, ..., -5.5249e+00,
-5.5503e+00, -4.2785e+00],
[-1.9943e-02, 1.5733e+00, -7.3750e+00, ..., -7.9850e+00,
-7.7854e+00, -6.5979e+00],
...,
[ 2.9486e-01, 3.0767e-01, -4.0640e+00, ..., -4.6108e+00,
-4.1904e+00, -3.4553e+00],
[ 2.2415e+00, 2.1950e+00, -2.5936e-01, ..., -7.7248e-01,
-1.1508e+00, -1.1800e+00],
[ 1.3081e+00, 1.1156e+00, -1.8137e+00, ..., -2.2253e+00,
-2.5039e+00, -2.3905e+00]],
[[-1.3877e+00, -2.5400e+00, -2.4871e+00, ..., -1.6797e+00,
-1.2026e+00, -1.3305e+00],
[-1.8395e+00, -2.1882e+00, -1.6661e+00, ..., -1.1771e+00,
-6.9484e-01, -7.9700e-01],
[-2.3036e+00, -2.0456e+00, -1.0684e+00, ..., -1.0929e+00,
-4.7825e-01, -2.0979e-01],
...,
[-1.4649e-01, -5.6592e-01, -4.7846e-01, ..., -5.1401e-01,
-1.9006e-01, -4.3887e-01],
[-6.9611e-01, -1.6332e-01, -4.1094e-01, ..., -2.0328e-01,
2.5608e-01, 2.4515e-01],
[-7.6652e-01, -1.3541e-01, -2.9978e-01, ..., -4.0856e-01,
-1.9465e-01, -2.3054e-01]],
...,
[[-2.2406e+00, -1.0978e+00, -2.4746e+00, ..., -2.9696e+00,
-2.4577e+00, -1.7152e+00],
[-1.8186e+00, 2.4425e-01, -1.2872e+00, ..., -2.4034e+00,
-2.4328e+00, -1.8637e+00],
[-3.5692e-01, 2.7069e+00, 1.2482e+00, ..., 1.5596e-01,
2.0845e-01, 4.6444e-01],
...,
[-2.0360e+00, -1.2273e-01, -1.7143e-01, ..., -1.3490e+00,
-4.2110e-01, 8.1179e-02],
[-1.1783e+00, 7.5618e-01, 9.1709e-01, ..., 2.5904e-02,
2.9966e-01, 4.7052e-01],
[-2.0748e+00, -1.4064e+00, -1.6447e+00, ..., -2.4618e+00,
-1.4930e+00, -9.4107e-01]],
[[-2.4042e+00, -2.7854e+00, -4.3728e+00, ..., -2.4761e+00,
-2.0162e+00, -1.7963e+00],
[-1.6442e+00, -1.4205e+00, -5.0932e+00, ..., -3.0083e+00,
-2.8554e+00, -2.7846e+00],
[-2.7792e+00, -2.3097e+00, -7.0592e+00, ..., -4.4205e+00,
-4.3205e+00, -3.4661e+00],
...,
[-1.4462e+00, -1.0540e-01, -3.3817e+00, ..., -1.3800e+00,
-9.7802e-01, -9.0602e-01],
[-1.7383e+00, -6.8188e-01, -3.0231e+00, ..., -1.6935e+00,
-1.1328e+00, -7.9721e-01],
[-2.3223e+00, -1.4967e+00, -3.1179e+00, ..., -2.9498e+00,
-2.1332e+00, -1.9831e+00]],
[[-1.8660e+00, -2.6226e+00, -2.4826e+00, ..., -2.6338e+00,
-2.1625e+00, -1.6638e+00],
[-1.8543e+00, -2.5873e+00, -2.1325e+00, ..., -2.3960e+00,
-1.8945e+00, -1.5380e+00],
[-2.3389e+00, -2.8871e+00, -2.5007e+00, ..., -2.7641e+00,
-2.1621e+00, -1.6485e+00],
...,
[-2.3797e+00, -2.9472e+00, -2.7312e+00, ..., -2.8650e+00,
-2.2930e+00, -1.7652e+00],
[-1.5514e+00, -2.0167e+00, -1.9508e+00, ..., -1.9386e+00,
-1.6005e+00, -1.2954e+00],
[-1.0148e+00, -1.3109e+00, -1.3419e+00, ..., -1.1825e+00,
-1.0108e+00, -7.7234e-01]]],
[[[ 5.0482e-01, 9.6410e-01, 7.3595e-01, ..., 7.6516e-01,
1.0371e+00, 8.2886e-01],
[-1.1395e+00, -1.9572e-01, -1.5972e+00, ..., -1.0753e+00,
1.4447e-01, 6.8506e-01],
[ 8.1871e-01, 5.4826e-01, 1.8961e-01, ..., 5.6845e-01,
9.4557e-01, 9.5210e-01],
...,
[-8.7011e-01, -1.8592e+00, -1.8523e+00, ..., -1.8367e+00,
-1.3228e+00, -9.4972e-01],
[ 7.4207e-01, -1.0264e-02, -1.0999e-01, ..., -4.8800e-02,
7.1205e-01, 8.0109e-01],
[ 1.3568e-01, -5.5532e-01, -4.6236e-01, ..., -1.9670e-01,
4.1650e-01, 3.3613e-01]],
[[-2.6851e+00, -1.3341e+00, -4.7157e+00, ..., -5.6828e+00,
-4.8698e+00, -3.8195e+00],
[-5.0474e-01, 1.7435e+00, -3.0690e+00, ..., -4.6612e+00,
-4.4915e+00, -3.6861e+00],
[ 2.9587e-01, 1.6178e+00, -5.8873e+00, ..., -6.7968e+00,
-6.4759e+00, -5.8903e+00],
...,
[ 2.1829e-01, 2.6794e-01, -4.0682e+00, ..., -4.6455e+00,
-4.2061e+00, -3.4858e+00],
[ 2.2380e+00, 2.2144e+00, -1.5917e-01, ..., -8.4889e-01,
-1.2018e+00, -1.2342e+00],
[ 1.3012e+00, 1.1442e+00, -1.7594e+00, ..., -2.2668e+00,
-2.5182e+00, -2.4068e+00]],
[[-5.9714e-01, -1.7007e+00, -1.4941e+00, ..., -6.9712e-01,
-4.9732e-01, -6.9906e-01],
[-1.4363e+00, -1.8561e+00, -1.1787e+00, ..., -6.0332e-01,
-3.3914e-01, -3.0843e-01],
[-1.2963e+00, -1.1855e+00, -5.1039e-01, ..., -3.7601e-01,
-7.5296e-02, -8.9755e-02],
...,
[-3.4250e-02, -5.2469e-01, -4.3776e-01, ..., -4.8882e-01,
-1.8722e-01, -4.5372e-01],
[-6.1669e-01, -1.6710e-01, -4.5650e-01, ..., -2.0268e-01,
2.5140e-01, 2.1807e-01],
[-7.4867e-01, -1.8162e-01, -3.7203e-01, ..., -4.0397e-01,
-1.8076e-01, -2.4135e-01]],
...,
[[-2.2718e+00, -1.1998e+00, -2.6016e+00, ..., -3.0850e+00,
-2.5439e+00, -1.7304e+00],
[-5.5794e-01, 1.3006e+00, -6.5383e-01, ..., -1.5878e+00,
-1.7333e+00, -1.4895e+00],
[-2.3291e-01, 2.4779e+00, 6.5618e-01, ..., -3.7623e-01,
-9.1081e-02, 8.5823e-02],
...,
[-2.1868e+00, -2.8034e-01, -2.8764e-01, ..., -1.3176e+00,
-3.9800e-01, 8.0321e-02],
[-1.3024e+00, 6.7665e-01, 8.8358e-01, ..., 3.3453e-02,
3.1751e-01, 4.9235e-01],
[-2.1961e+00, -1.5077e+00, -1.6421e+00, ..., -2.4227e+00,
-1.4616e+00, -9.0277e-01]],
[[-1.4080e+00, -1.3158e+00, -2.8826e+00, ..., -1.1478e+00,
-6.5020e-01, -5.2922e-01],
[-7.4293e-01, -2.3203e-01, -3.8725e+00, ..., -1.9704e+00,
-1.7079e+00, -1.8456e+00],
[-2.3934e+00, -1.7184e+00, -6.6044e+00, ..., -4.1939e+00,
-3.6386e+00, -2.8483e+00],
...,
[-1.4175e+00, -1.0004e-01, -3.3344e+00, ..., -1.4093e+00,
-1.0358e+00, -9.4445e-01],
[-1.7233e+00, -7.2342e-01, -3.0143e+00, ..., -1.6957e+00,
-1.1689e+00, -8.3483e-01],
[-2.3288e+00, -1.5409e+00, -3.0874e+00, ..., -2.9173e+00,
-2.1432e+00, -2.0043e+00]],
[[-1.2579e+00, -1.8706e+00, -1.6300e+00, ..., -1.8018e+00,
-1.4949e+00, -1.1100e+00],
[-1.3032e+00, -1.8356e+00, -1.3439e+00, ..., -1.6130e+00,
-1.3106e+00, -1.0262e+00],
[-1.7261e+00, -2.0838e+00, -1.6092e+00, ..., -1.9241e+00,
-1.4873e+00, -1.0911e+00],
...,
[-2.3740e+00, -2.9432e+00, -2.7538e+00, ..., -2.8684e+00,
-2.2948e+00, -1.7613e+00],
[-1.5298e+00, -1.9987e+00, -1.9478e+00, ..., -1.9575e+00,
-1.6139e+00, -1.2999e+00],
[-9.8979e-01, -1.2921e+00, -1.3372e+00, ..., -1.2045e+00,
-1.0239e+00, -7.7808e-01]]],
[[[ 5.2856e-01, 1.0047e+00, 8.9367e-01, ..., -3.6077e-01,
-1.0324e-01, -1.4660e-02],
[-1.3270e+00, -3.9173e-01, -1.6652e+00, ..., -3.1804e+00,
-1.2790e+00, -4.7139e-01],
[ 6.9137e-01, 4.2870e-01, 3.2514e-01, ..., -1.5316e-01,
-3.5957e-01, 1.4614e-01],
...,
[-1.8639e-01, -7.9419e-01, -1.4001e+00, ..., -1.9753e+00,
-1.8838e+00, -1.2792e+00],
[ 1.0288e+00, 7.6253e-01, 2.8675e-01, ..., -1.3375e-02,
2.7587e-01, 4.4062e-01],
[ 6.8513e-01, 4.8998e-01, 1.8641e-01, ..., 1.9789e-01,
4.6397e-01, 4.8797e-01]],
[[-2.3750e+00, -9.7977e-01, -4.1050e+00, ..., -6.7878e+00,
-6.0522e+00, -4.5088e+00],
[-8.5601e-02, 2.1761e+00, -2.2713e+00, ..., -5.4479e+00,
-5.5618e+00, -4.2917e+00],
[ 7.6458e-01, 2.0005e+00, -5.0066e+00, ..., -7.8203e+00,
-7.7749e+00, -6.5584e+00],
...,
[-5.1588e-01, 5.2711e-01, -4.1551e+00, ..., -7.1169e+00,
-6.7186e+00, -5.1887e+00],
[ 1.9520e+00, 2.5948e+00, 1.0778e-01, ..., -1.9355e+00,
-2.5782e+00, -2.0409e+00],
[ 1.1152e+00, 1.3959e+00, -1.5963e+00, ..., -2.9359e+00,
-3.4962e+00, -3.0436e+00]],
[[-6.7202e-01, -1.7876e+00, -1.5689e+00, ..., -1.3890e+00,
-1.0410e+00, -1.3247e+00],
[-1.4921e+00, -1.8627e+00, -1.1207e+00, ..., -9.7751e-01,
-6.3027e-01, -9.1686e-01],
[-1.3168e+00, -1.1216e+00, -4.1531e-01, ..., -1.0888e+00,
-6.0292e-01, -4.8116e-01],
...,
[-8.9635e-01, -1.3082e+00, -5.1107e-01, ..., -1.0495e+00,
-8.7270e-01, -1.2379e+00],
[-1.2378e+00, -8.5577e-01, -5.3677e-01, ..., -5.7567e-01,
-2.4972e-01, -4.1420e-01],
[-1.3197e+00, -7.5696e-01, -3.5834e-01, ..., -5.1705e-01,
-4.5131e-01, -6.1158e-01]],
...,
[[-2.4662e+00, -1.3173e+00, -2.7165e+00, ..., -2.9547e+00,
-2.4090e+00, -1.7669e+00],
[-7.7007e-01, 1.1443e+00, -9.3788e-01, ..., -2.3806e+00,
-2.4347e+00, -1.9801e+00],
[-2.3718e-01, 2.5662e+00, 6.7859e-01, ..., -8.1593e-02,
4.8329e-02, 2.8827e-01],
...,
[-1.3934e+00, 1.2419e+00, -1.6562e-01, ..., -9.2188e-01,
-2.6063e-01, 2.9800e-01],
[-4.4046e-01, 1.5813e+00, 7.8518e-01, ..., 4.3263e-01,
4.5887e-01, 8.1329e-01],
[-1.5207e+00, -9.4564e-01, -1.4441e+00, ..., -2.2035e+00,
-1.1850e+00, -3.6181e-01]],
[[-1.2806e+00, -1.1414e+00, -2.7204e+00, ..., -2.2879e+00,
-1.9509e+00, -1.8257e+00],
[-4.7836e-01, 1.4787e-01, -3.4853e+00, ..., -2.9151e+00,
-2.8798e+00, -2.8661e+00],
[-2.0682e+00, -1.2146e+00, -6.0221e+00, ..., -4.4735e+00,
-4.4943e+00, -3.6763e+00],
...,
[-1.6247e+00, -5.6247e-01, -4.0862e+00, ..., -2.8234e+00,
-2.7335e+00, -2.3075e+00],
[-1.6280e+00, -7.5553e-01, -3.4099e+00, ..., -2.4340e+00,
-2.2361e+00, -1.6488e+00],
[-2.0563e+00, -1.3604e+00, -2.9182e+00, ..., -2.8610e+00,
-2.4185e+00, -2.1933e+00]],
[[-1.2058e+00, -1.8013e+00, -1.5515e+00, ..., -2.5767e+00,
-2.1610e+00, -1.6743e+00],
[-1.1923e+00, -1.6850e+00, -1.1639e+00, ..., -2.4267e+00,
-1.9640e+00, -1.6026e+00],
[-1.5460e+00, -1.8408e+00, -1.3502e+00, ..., -2.8597e+00,
-2.2849e+00, -1.7712e+00],
...,
[-2.2841e+00, -2.8362e+00, -2.6362e+00, ..., -4.0167e+00,
-3.2595e+00, -2.5622e+00],
[-1.5404e+00, -1.9526e+00, -1.8939e+00, ..., -2.7781e+00,
-2.3310e+00, -1.9350e+00],
[-1.0351e+00, -1.2410e+00, -1.3063e+00, ..., -1.8384e+00,
-1.5292e+00, -1.1893e+00]]],
[[[ 3.5249e-01, 6.4105e-01, 7.3959e-01, ..., 5.4886e-01,
1.0231e+00, 6.3715e-01],
[-1.9437e+00, -1.2502e+00, -2.8492e+00, ..., -2.3479e+00,
-6.8532e-01, -5.3907e-02],
[ 3.0647e-02, -7.3357e-01, -8.1222e-02, ..., 3.3945e-02,
6.5283e-01, 7.2405e-01],
...,
[ 7.4465e-02, -7.7786e-01, -1.7789e+00, ..., -1.1861e+00,
-8.9431e-01, -6.4200e-01],
[ 1.5705e+00, 1.2117e+00, 4.5702e-01, ..., 6.7024e-01,
8.7841e-01, 9.0595e-01],
[ 1.1837e+00, 9.8136e-01, 6.1845e-01, ..., 8.7172e-01,
9.9910e-01, 8.4280e-01]],
[[-1.7947e+00, -1.1520e+00, -4.1727e+00, ..., -4.9397e+00,
-4.4526e+00, -3.4131e+00],
[ 4.4614e-01, 1.9025e+00, -2.3253e+00, ..., -3.6372e+00,
-3.6900e+00, -3.0270e+00],
[ 1.1977e+00, 1.6643e+00, -4.9780e+00, ..., -5.5835e+00,
-5.3612e+00, -4.9929e+00],
...,
[-6.7272e-01, 6.0461e-01, -5.0303e+00, ..., -6.0955e+00,
-5.4869e+00, -4.3987e+00],
[ 2.1766e+00, 2.9631e+00, 2.0907e-03, ..., -1.6522e+00,
-1.8941e+00, -1.7920e+00],
[ 1.2955e+00, 1.6333e+00, -1.7874e+00, ..., -2.7985e+00,
-2.9800e+00, -2.8075e+00]],
[[-1.9685e-01, -1.2302e+00, -1.3462e+00, ..., -5.2995e-01,
-2.1285e-01, -5.0931e-01],
[-7.4215e-01, -1.0044e+00, -8.2276e-01, ..., -3.4152e-01,
-3.0972e-02, -1.1036e-01],
[-6.6422e-01, -4.7417e-01, -4.1981e-01, ..., -3.3716e-01,
6.5679e-02, 5.7274e-02],
...,
[-1.1628e+00, -1.6484e+00, -7.1829e-01, ..., -7.0742e-01,
-5.3208e-01, -7.0818e-01],
[-1.4128e+00, -9.5480e-01, -5.5125e-01, ..., -2.2335e-01,
6.9759e-03, -7.1147e-03],
[-1.4168e+00, -7.2415e-01, -1.7953e-01, ..., -1.1638e-01,
-7.7802e-02, -1.1035e-01]],
...,
[[-2.9073e+00, -2.2675e+00, -2.7087e+00, ..., -3.3963e+00,
-2.6125e+00, -1.7375e+00],
[-2.0110e+00, -4.8741e-01, -1.4163e+00, ..., -2.4387e+00,
-2.2302e+00, -1.7612e+00],
[-8.9589e-01, 1.4232e+00, 6.3788e-01, ..., -4.4168e-01,
-2.8676e-02, 1.2507e-01],
...,
[-1.6239e+00, 1.7023e+00, 4.1967e-01, ..., -7.5237e-01,
-1.2658e-01, 4.1407e-01],
[-6.5764e-01, 1.8674e+00, 1.2560e+00, ..., 2.7365e-01,
3.7930e-01, 5.6211e-01],
[-1.7799e+00, -9.8756e-01, -1.3282e+00, ..., -2.0023e+00,
-1.0981e+00, -4.6510e-01]],
[[-1.5346e+00, -1.5078e+00, -2.8379e+00, ..., -1.2089e+00,
-7.6436e-01, -7.5816e-01],
[-5.8171e-01, 6.9746e-02, -3.2466e+00, ..., -1.4047e+00,
-1.2376e+00, -1.5678e+00],
[-2.0817e+00, -9.6955e-01, -5.4212e+00, ..., -3.2583e+00,
-2.7862e+00, -2.2413e+00],
...,
[-1.8581e+00, -8.0100e-01, -4.8743e+00, ..., -2.5463e+00,
-2.1946e+00, -1.7485e+00],
[-1.7134e+00, -8.1991e-01, -3.8394e+00, ..., -2.3023e+00,
-1.7716e+00, -1.1772e+00],
[-2.0746e+00, -1.3233e+00, -3.0395e+00, ..., -2.7481e+00,
-2.0024e+00, -1.7245e+00]],
[[-1.3164e+00, -1.8739e+00, -1.6726e+00, ..., -1.7472e+00,
-1.4447e+00, -1.0941e+00],
[-1.2491e+00, -1.7313e+00, -1.2229e+00, ..., -1.3759e+00,
-1.0889e+00, -8.6187e-01],
[-1.5798e+00, -1.9186e+00, -1.4677e+00, ..., -1.6028e+00,
-1.2048e+00, -8.8446e-01],
...,
[-2.6297e+00, -3.2739e+00, -3.0007e+00, ..., -3.1665e+00,
-2.5285e+00, -1.9331e+00],
[-1.7602e+00, -2.2343e+00, -2.1285e+00, ..., -2.2088e+00,
-1.8093e+00, -1.4477e+00],
[-1.1819e+00, -1.4247e+00, -1.4512e+00, ..., -1.3803e+00,
-1.1639e+00, -8.6057e-01]]],
[[[ 2.1290e-01, 4.8551e-01, 6.1481e-01, ..., -3.7857e-01,
-8.5159e-02, -1.2287e-02],
[-1.9932e+00, -1.3272e+00, -2.7895e+00, ..., -3.4232e+00,
-1.3923e+00, -5.0173e-01],
[-1.4854e-01, -8.4607e-01, 5.4434e-02, ..., -1.7708e-01,
-3.6395e-01, 1.3912e-01],
...,
[-9.6289e-01, -2.1414e+00, -2.0919e+00, ..., -1.6803e+00,
-1.7459e+00, -1.1646e+00],
[ 8.1268e-01, -9.4288e-02, -1.2802e-01, ..., 5.5270e-02,
2.4169e-01, 4.1264e-01],
[ 2.3365e-01, -5.5047e-01, -3.8421e-01, ..., 1.8942e-01,
4.0963e-01, 4.6376e-01]],
[[-1.7081e+00, -1.4452e+00, -4.2855e+00, ..., -7.1001e+00,
-6.2732e+00, -4.6420e+00],
[ 3.9525e-01, 1.3633e+00, -2.5631e+00, ..., -5.5247e+00,
-5.6328e+00, -4.2973e+00],
[ 1.1088e+00, 1.1131e+00, -5.0255e+00, ..., -8.0158e+00,
-7.9531e+00, -6.7382e+00],
...,
[ 3.5135e-01, 3.5497e-01, -4.1166e+00, ..., -7.1865e+00,
-6.7501e+00, -5.2239e+00],
[ 2.5374e+00, 2.5063e+00, 1.3004e-01, ..., -2.3152e+00,
-2.8515e+00, -2.2430e+00],
[ 1.5067e+00, 1.3311e+00, -1.6183e+00, ..., -3.0013e+00,
-3.5078e+00, -3.0276e+00]],
[[-1.3306e-01, -1.1374e+00, -1.3814e+00, ..., -1.5038e+00,
-1.1205e+00, -1.4209e+00],
[-4.8378e-01, -8.1499e-01, -8.8310e-01, ..., -1.0585e+00,
-6.5530e-01, -9.2692e-01],
[-3.1027e-01, -2.1581e-01, -4.3291e-01, ..., -1.1324e+00,
-5.7193e-01, -4.2845e-01],
...,
[-7.3852e-02, -6.2702e-01, -6.1460e-01, ..., -9.9094e-01,
-8.4482e-01, -1.2036e+00],
[-7.0430e-01, -2.2826e-01, -5.7785e-01, ..., -5.3432e-01,
-2.5866e-01, -4.4493e-01],
[-8.7568e-01, -2.7553e-01, -4.8274e-01, ..., -4.8298e-01,
-4.5368e-01, -6.2674e-01]],
...,
[[-2.7806e+00, -2.4482e+00, -2.4541e+00, ..., -3.3548e+00,
-2.6759e+00, -1.9111e+00],
[-2.0562e+00, -8.5031e-01, -1.1246e+00, ..., -2.6102e+00,
-2.6000e+00, -2.0588e+00],
[-1.1271e+00, 7.3066e-01, 6.9577e-01, ..., -1.7491e-01,
2.6760e-02, 3.2820e-01],
...,
[-2.2673e+00, -1.3869e-01, -1.4889e-01, ..., -8.5563e-01,
-2.6348e-01, 2.3033e-01],
[-1.2329e+00, 8.8785e-01, 1.0732e+00, ..., 4.0877e-01,
4.3850e-01, 7.6210e-01],
[-2.1858e+00, -1.4688e+00, -1.6609e+00, ..., -2.0152e+00,
-1.0471e+00, -2.9434e-01]],
[[-1.6347e+00, -1.6336e+00, -2.7533e+00, ..., -2.2050e+00,
-1.8252e+00, -1.7361e+00],
[-7.2560e-01, -1.3661e-01, -3.1100e+00, ..., -2.8580e+00,
-2.8227e+00, -2.9050e+00],
[-2.0899e+00, -1.0406e+00, -5.1008e+00, ..., -4.5918e+00,
-4.5870e+00, -3.7867e+00],
...,
[-1.3628e+00, 1.8235e-01, -3.3842e+00, ..., -2.9818e+00,
-2.8475e+00, -2.3949e+00],
[-1.6831e+00, -4.5934e-01, -3.0098e+00, ..., -2.5255e+00,
-2.3400e+00, -1.7566e+00],
[-2.3286e+00, -1.3598e+00, -3.0186e+00, ..., -2.7802e+00,
-2.4159e+00, -2.1930e+00]],
[[-1.2824e+00, -1.8085e+00, -1.6484e+00, ..., -2.6622e+00,
-2.2316e+00, -1.7322e+00],
[-1.2266e+00, -1.6811e+00, -1.2254e+00, ..., -2.4469e+00,
-1.9721e+00, -1.6200e+00],
[-1.5354e+00, -1.8759e+00, -1.4727e+00, ..., -2.8365e+00,
-2.2720e+00, -1.7531e+00],
...,
[-2.4438e+00, -3.0539e+00, -2.7931e+00, ..., -3.9545e+00,
-3.2128e+00, -2.5127e+00],
[-1.5807e+00, -2.0851e+00, -1.9909e+00, ..., -2.7370e+00,
-2.2941e+00, -1.8882e+00],
[-1.0310e+00, -1.3457e+00, -1.3683e+00, ..., -1.8429e+00,
-1.5258e+00, -1.1757e+00]]]], device='cuda:0'),
tensor([[[[ 2.7159e-01, 5.5747e-01, 6.9325e-01, ..., 6.1121e-01,
9.6034e-01, 7.5895e-01],
[-2.0784e+00, -1.3571e+00, -2.9767e+00, ..., -1.2685e+00,
5.9156e-02, 5.9272e-01],
[-6.5835e-02, -9.2321e-01, -2.3808e-02, ..., 5.2102e-01,
9.3126e-01, 9.4452e-01],
...,
[-9.7145e-01, -2.1262e+00, -2.0003e+00, ..., -1.3443e+00,
-8.8023e-01, -6.4603e-01],
[ 7.2914e-01, -1.6267e-01, -1.2681e-01, ..., 4.1816e-01,
8.1181e-01, 9.0567e-01],
[ 1.6270e-01, -6.1799e-01, -4.0540e-01, ..., 6.0606e-01,
9.1218e-01, 8.1166e-01]],
[[-1.7115e+00, -1.2656e+00, -4.2712e+00, ..., -5.7742e+00,
-4.8975e+00, -3.8059e+00],
[ 6.4761e-01, 1.8671e+00, -2.3245e+00, ..., -4.7194e+00,
-4.4122e+00, -3.6119e+00],
[ 1.4982e+00, 1.6105e+00, -4.9979e+00, ..., -6.7799e+00,
-6.3014e+00, -5.7586e+00],
...,
[ 3.1534e-01, 2.3451e-01, -4.2400e+00, ..., -5.9998e+00,
-5.1620e+00, -4.2375e+00],
[ 2.3653e+00, 2.2809e+00, -1.7627e-01, ..., -1.3013e+00,
-1.4295e+00, -1.5065e+00],
[ 1.3885e+00, 1.1910e+00, -1.7656e+00, ..., -2.7157e+00,
-2.8361e+00, -2.7717e+00]],
[[-2.0352e-01, -1.3563e+00, -1.5691e+00, ..., -8.0573e-01,
-5.4034e-01, -6.6771e-01],
[-6.7901e-01, -1.0235e+00, -1.0048e+00, ..., -6.8129e-01,
-3.7348e-01, -2.3228e-01],
[-5.0121e-01, -3.2218e-01, -4.4583e-01, ..., -3.7719e-01,
9.6972e-04, 1.0153e-01],
...,
[ 3.8384e-02, -5.0807e-01, -5.8725e-01, ..., -6.5853e-01,
-4.6351e-01, -5.6715e-01],
[-6.0976e-01, -1.6332e-01, -5.5487e-01, ..., -2.8034e-01,
8.7250e-03, 8.5038e-02],
[-7.8366e-01, -2.3078e-01, -4.8658e-01, ..., -3.0166e-01,
-1.9597e-01, -1.3535e-01]],
...,
[[-3.0717e+00, -2.5131e+00, -2.6115e+00, ..., -3.1167e+00,
-2.5561e+00, -1.7265e+00],
[-2.1813e+00, -6.7201e-01, -1.0793e+00, ..., -1.6420e+00,
-1.7069e+00, -1.4517e+00],
[-1.1328e+00, 1.0423e+00, 9.0686e-01, ..., -3.0858e-01,
-1.7069e-03, 1.5594e-01],
...,
[-2.2253e+00, -2.5806e-01, -1.3703e-01, ..., -1.0911e+00,
-3.9769e-01, 1.6987e-01],
[-1.2450e+00, 7.8992e-01, 1.0304e+00, ..., 2.1588e-01,
3.3417e-01, 4.4289e-01],
[-2.1576e+00, -1.4583e+00, -1.5722e+00, ..., -2.2552e+00,
-1.2688e+00, -6.3180e-01]],
[[-1.6233e+00, -1.6222e+00, -2.9485e+00, ..., -1.3642e+00,
-7.3076e-01, -4.9000e-01],
[-5.7827e-01, 7.6562e-02, -3.2895e+00, ..., -1.9969e+00,
-1.6021e+00, -1.6772e+00],
[-2.0324e+00, -8.4469e-01, -5.3782e+00, ..., -3.9874e+00,
-3.3443e+00, -2.5623e+00],
...,
[-1.3555e+00, 9.0533e-02, -3.3233e+00, ..., -2.3903e+00,
-1.7735e+00, -1.3274e+00],
[-1.6678e+00, -5.3769e-01, -2.9864e+00, ..., -2.2714e+00,
-1.4960e+00, -8.7168e-01],
[-2.2802e+00, -1.3692e+00, -2.9972e+00, ..., -2.8949e+00,
-1.9311e+00, -1.6187e+00]],
[[-1.3588e+00, -1.9109e+00, -1.6857e+00, ..., -1.8549e+00,
-1.5083e+00, -1.1030e+00],
[-1.2794e+00, -1.7705e+00, -1.2022e+00, ..., -1.6239e+00,
-1.2904e+00, -9.7709e-01],
[-1.6109e+00, -1.9446e+00, -1.4412e+00, ..., -1.8988e+00,
-1.4200e+00, -1.0029e+00],
...,
[-2.4139e+00, -3.0121e+00, -2.7714e+00, ..., -3.1594e+00,
-2.4856e+00, -1.8512e+00],
[-1.5698e+00, -2.0637e+00, -1.9867e+00, ..., -2.2273e+00,
-1.8039e+00, -1.4165e+00],
[-1.0288e+00, -1.3472e+00, -1.3781e+00, ..., -1.3723e+00,
-1.1564e+00, -8.3961e-01]]],
[[[ 4.8027e-01, 9.3556e-01, 7.0530e-01, ..., 8.3281e-01,
1.0686e+00, 8.2897e-01],
[-1.0665e+00, -1.4513e-01, -1.4818e+00, ..., -1.1442e+00,
8.5906e-02, 6.1762e-01],
[ 7.7356e-01, 5.6628e-01, 2.7422e-01, ..., 5.2979e-01,
8.6650e-01, 8.8188e-01],
...,
[-8.6148e-01, -1.8606e+00, -1.7971e+00, ..., -1.8026e+00,
-1.3505e+00, -9.5592e-01],
[ 7.4807e-01, -3.5455e-02, -9.8103e-02, ..., -2.2038e-02,
6.8754e-01, 7.9001e-01],
[ 1.3916e-01, -5.9994e-01, -4.7431e-01, ..., -1.6356e-01,
4.2281e-01, 3.4471e-01]],
[[-2.6435e+00, -1.4866e+00, -4.7735e+00, ..., -5.8322e+00,
-4.9497e+00, -3.8742e+00],
[-6.5583e-01, 1.3435e+00, -3.3371e+00, ..., -4.7317e+00,
-4.4980e+00, -3.6670e+00],
[ 3.1820e-02, 1.1505e+00, -6.0570e+00, ..., -6.8310e+00,
-6.4529e+00, -5.8835e+00],
...,
[ 2.6991e-01, 2.9218e-01, -4.1032e+00, ..., -4.6309e+00,
-4.2054e+00, -3.4810e+00],
[ 2.2652e+00, 2.2101e+00, -2.0751e-01, ..., -7.9957e-01,
-1.1711e+00, -1.2123e+00],
[ 1.3276e+00, 1.1333e+00, -1.7808e+00, ..., -2.2393e+00,
-2.5091e+00, -2.4052e+00]],
[[-5.6216e-01, -1.5720e+00, -1.4021e+00, ..., -7.2460e-01,
-4.8299e-01, -7.0253e-01],
[-1.3169e+00, -1.6936e+00, -1.0893e+00, ..., -6.7164e-01,
-3.6796e-01, -3.1105e-01],
[-1.1523e+00, -1.0537e+00, -4.8319e-01, ..., -4.1527e-01,
-8.4083e-02, -7.5944e-02],
...,
[-1.2787e-01, -5.9831e-01, -4.6735e-01, ..., -4.9845e-01,
-1.9189e-01, -4.5064e-01],
[-6.8687e-01, -1.9042e-01, -4.2001e-01, ..., -2.0510e-01,
2.4961e-01, 2.4380e-01],
[-7.7516e-01, -1.7492e-01, -3.2968e-01, ..., -3.9689e-01,
-1.9588e-01, -2.3513e-01]],
...,
[[-2.0965e+00, -1.2190e+00, -2.4994e+00, ..., -3.2175e+00,
-2.6369e+00, -1.8108e+00],
[-5.0704e-01, 1.1179e+00, -7.0097e-01, ..., -1.6292e+00,
-1.7731e+00, -1.5713e+00],
[-2.5304e-01, 2.1391e+00, 4.5598e-01, ..., -3.7746e-01,
-4.3559e-02, 1.1451e-01],
...,
[-2.0919e+00, -1.4423e-01, -2.4911e-01, ..., -1.4102e+00,
-4.2012e-01, 9.7202e-02],
[-1.2272e+00, 7.7110e-01, 8.9884e-01, ..., -1.7780e-03,
2.9922e-01, 4.7059e-01],
[-2.1148e+00, -1.4535e+00, -1.6898e+00, ..., -2.4744e+00,
-1.4827e+00, -9.3421e-01]],
[[-1.4333e+00, -1.3782e+00, -2.7433e+00, ..., -1.1402e+00,
-6.0743e-01, -4.7139e-01],
[-8.5118e-01, -4.4925e-01, -3.7017e+00, ..., -1.9155e+00,
-1.6071e+00, -1.7864e+00],
[-2.4289e+00, -1.9248e+00, -6.3211e+00, ..., -4.1783e+00,
-3.5722e+00, -2.8014e+00],
...,
[-1.3969e+00, -1.1117e-01, -3.3660e+00, ..., -1.4200e+00,
-1.0221e+00, -9.1895e-01],
[-1.7122e+00, -6.8887e-01, -3.0370e+00, ..., -1.7356e+00,
-1.1641e+00, -8.1446e-01],
[-2.3297e+00, -1.5226e+00, -3.1175e+00, ..., -2.9792e+00,
-2.1457e+00, -1.9929e+00]],
[[-1.2148e+00, -1.7937e+00, -1.6188e+00, ..., -1.8226e+00,
-1.5160e+00, -1.1240e+00],
[-1.2690e+00, -1.7702e+00, -1.3766e+00, ..., -1.6016e+00,
-1.3091e+00, -1.0197e+00],
[-1.6702e+00, -2.0223e+00, -1.6508e+00, ..., -1.9050e+00,
-1.4638e+00, -1.0652e+00],
...,
[-2.3764e+00, -2.9473e+00, -2.7354e+00, ..., -2.8626e+00,
-2.2888e+00, -1.7524e+00],
[-1.5364e+00, -2.0019e+00, -1.9369e+00, ..., -1.9405e+00,
-1.5992e+00, -1.2879e+00],
[-9.9932e-01, -1.2928e+00, -1.3283e+00, ..., -1.1860e+00,
-1.0098e+00, -7.6311e-01]]],
[[[ 1.4204e-01, 3.6986e-01, -1.2329e+00, ..., 4.8561e-01,
9.9217e-01, 6.5316e-01],
[-2.1151e+00, -1.4712e+00, -4.4627e+00, ..., -2.3878e+00,
-6.8649e-01, -3.2788e-02],
[ 4.2509e-01, -2.9294e-01, -1.1702e+00, ..., -2.3216e-01,
5.0802e-01, 6.6878e-01],
...,
[-2.6298e-01, -1.2808e+00, -2.6276e+00, ..., -2.1746e+00,
-1.5611e+00, -1.0810e+00],
[ 1.2567e+00, 6.5466e-01, -9.4714e-01, ..., -3.2509e-01,
5.1494e-01, 7.1307e-01],
[ 8.1789e-01, 2.9913e-01, -4.4091e-01, ..., -3.2185e-01,
3.2642e-01, 3.0006e-01]],
[[-3.2830e+00, -1.5022e+00, -6.1889e+00, ..., -4.7157e+00,
-4.2994e+00, -3.2995e+00],
[-6.1501e-01, 2.1055e+00, -4.2138e+00, ..., -3.4060e+00,
-3.5300e+00, -2.8998e+00],
[-2.1502e-02, 1.7902e+00, -7.4418e+00, ..., -5.2993e+00,
-5.1726e+00, -4.8392e+00],
...,
[-1.2918e+00, 2.1571e-01, -6.5472e+00, ..., -4.5474e+00,
-4.0977e+00, -3.4059e+00],
[ 1.4814e+00, 2.5787e+00, -1.3457e+00, ..., -5.4122e-01,
-9.3095e-01, -1.0436e+00],
[ 9.7734e-01, 1.6275e+00, -2.4036e+00, ..., -2.1449e+00,
-2.4337e+00, -2.3679e+00]],
[[-1.4159e+00, -2.3857e+00, -2.2078e+00, ..., -5.3527e-01,
-1.6816e-01, -4.1372e-01],
[-2.0369e+00, -2.2113e+00, -1.5773e+00, ..., -3.0153e-01,
7.9733e-02, 4.4171e-02],
[-2.5597e+00, -2.0158e+00, -8.4198e-01, ..., -2.5034e-01,
2.4171e-01, 2.7156e-01],
...,
[-1.6196e+00, -1.8191e+00, -7.7208e-01, ..., -5.3138e-01,
-1.8619e-01, -3.9244e-01],
[-1.7737e+00, -1.1857e+00, -7.6773e-01, ..., -3.1685e-01,
1.7048e-01, 2.3413e-01],
[-1.7216e+00, -1.1177e+00, -6.0256e-01, ..., -4.9758e-01,
-2.6684e-01, -2.4468e-01]],
...,
[[-2.3653e+00, -1.3282e+00, -2.7798e+00, ..., -3.4696e+00,
-2.6655e+00, -1.7463e+00],
[-1.8931e+00, 1.6762e-01, -1.3566e+00, ..., -2.5554e+00,
-2.3164e+00, -1.7763e+00],
[-4.9683e-01, 2.4525e+00, 8.9972e-01, ..., -5.8954e-01,
-1.0795e-01, 1.2942e-01],
...,
[-1.7383e+00, 1.1026e+00, 1.8577e-01, ..., -1.6081e+00,
-6.8364e-01, -8.6899e-02],
[-8.6437e-01, 1.6319e+00, 1.4921e+00, ..., -1.2919e-01,
1.6472e-01, 3.8979e-01],
[-2.1154e+00, -1.0785e+00, -8.5630e-01, ..., -2.5693e+00,
-1.5588e+00, -9.5344e-01]],
[[-2.3849e+00, -2.6591e+00, -4.0230e+00, ..., -1.2233e+00,
-8.1627e-01, -8.0537e-01],
[-1.6739e+00, -1.3555e+00, -4.7331e+00, ..., -1.4341e+00,
-1.2798e+00, -1.6016e+00],
[-3.0339e+00, -2.4631e+00, -6.8744e+00, ..., -3.2648e+00,
-2.7661e+00, -2.2138e+00],
...,
[-2.5481e+00, -1.7597e+00, -4.9457e+00, ..., -1.3668e+00,
-9.4514e-01, -8.0956e-01],
[-2.2339e+00, -1.5723e+00, -3.8599e+00, ..., -1.6733e+00,
-1.0800e+00, -6.7881e-01],
[-2.3151e+00, -1.7839e+00, -3.0794e+00, ..., -2.8619e+00,
-2.0301e+00, -1.8612e+00]],
[[-1.9005e+00, -2.6319e+00, -2.4220e+00, ..., -1.7464e+00,
-1.4557e+00, -1.1097e+00],
[-1.9244e+00, -2.6448e+00, -2.1090e+00, ..., -1.3935e+00,
-1.1036e+00, -8.8201e-01],
[-2.4482e+00, -3.0074e+00, -2.5248e+00, ..., -1.6140e+00,
-1.2123e+00, -9.0588e-01],
...,
[-3.1193e+00, -3.8981e+00, -3.6712e+00, ..., -2.9608e+00,
-2.3697e+00, -1.8170e+00],
[-2.0661e+00, -2.6779e+00, -2.6381e+00, ..., -2.0280e+00,
-1.6744e+00, -1.3538e+00],
[-1.4831e+00, -1.8094e+00, -1.8569e+00, ..., -1.2332e+00,
-1.0569e+00, -8.0811e-01]]],
[[[ 3.0289e-01, 5.6695e-01, 6.9417e-01, ..., 4.8851e-01,
1.0206e+00, 6.1953e-01],
[-2.1202e+00, -1.4535e+00, -3.1496e+00, ..., -2.5759e+00,
-8.2125e-01, -1.6216e-01],
[-1.3962e-01, -9.8402e-01, -2.5479e-01, ..., -1.0307e-01,
5.6852e-01, 6.6195e-01],
...,
[-1.8889e-01, -1.0247e+00, -2.4206e+00, ..., -1.5631e+00,
-1.5861e+00, -1.0086e+00],
[ 1.3020e+00, 7.6496e-01, -9.7710e-01, ..., -2.3189e-01,
6.1254e-02, 3.5085e-01],
[ 7.3533e-01, 2.1513e-01, -7.4919e-01, ..., -1.2951e-01,
2.0236e-01, 3.8458e-01]],
[[-1.6854e+00, -1.1306e+00, -4.1691e+00, ..., -4.8512e+00,
-4.4176e+00, -3.3841e+00],
[ 5.7366e-01, 1.9382e+00, -2.2834e+00, ..., -3.5703e+00,
-3.6564e+00, -3.0066e+00],
[ 1.3141e+00, 1.6832e+00, -4.9163e+00, ..., -5.5288e+00,
-5.3084e+00, -4.9297e+00],
...,
[-1.5743e+00, -2.7435e-01, -6.9456e+00, ..., -7.6611e+00,
-6.9077e+00, -5.4227e+00],
[ 1.3389e+00, 2.2977e+00, -1.5728e+00, ..., -2.3858e+00,
-2.7748e+00, -2.2113e+00],
[ 8.9660e-01, 1.5006e+00, -2.5142e+00, ..., -3.2103e+00,
-3.5979e+00, -3.1194e+00]],
[[-1.3105e-01, -1.1668e+00, -1.3142e+00, ..., -4.7816e-01,
-1.5397e-01, -4.6492e-01],
[-6.2184e-01, -8.8542e-01, -7.7704e-01, ..., -2.7178e-01,
3.5597e-02, -6.7546e-02],
[-5.6958e-01, -3.7547e-01, -4.1793e-01, ..., -3.4888e-01,
6.8846e-02, 7.5595e-02],
...,
[-1.1950e+00, -1.5782e+00, -7.3541e-01, ..., -9.0972e-01,
-7.8394e-01, -1.0473e+00],
[-1.5362e+00, -1.0373e+00, -7.9002e-01, ..., -5.3013e-01,
-2.5323e-01, -3.3029e-01],
[-1.7136e+00, -1.2212e+00, -9.1774e-01, ..., -9.3613e-01,
-8.1142e-01, -8.0192e-01]],
...,
[[-3.0201e+00, -2.4497e+00, -2.7677e+00, ..., -3.4064e+00,
-2.5922e+00, -1.7191e+00],
[-2.2534e+00, -7.7577e-01, -1.5685e+00, ..., -2.6139e+00,
-2.3233e+00, -1.8141e+00],
[-1.0269e+00, 1.2309e+00, 6.5315e-01, ..., -4.4664e-01,
-2.5791e-02, 1.0310e-01],
...,
[-1.7830e+00, 9.4909e-01, -1.1513e-01, ..., -1.1712e+00,
-4.9631e-01, 8.2594e-02],
[-8.4477e-01, 1.6377e+00, 1.4695e+00, ..., 4.5002e-01,
4.6758e-01, 7.3181e-01],
[-2.1494e+00, -1.1846e+00, -9.5118e-01, ..., -2.1000e+00,
-1.0601e+00, -3.4373e-01]],
[[-1.5986e+00, -1.5796e+00, -2.9142e+00, ..., -1.2854e+00,
-8.6247e-01, -8.6392e-01],
[-6.1631e-01, 6.9802e-02, -3.2273e+00, ..., -1.4248e+00,
-1.2472e+00, -1.5758e+00],
[-2.1089e+00, -9.2158e-01, -5.3357e+00, ..., -3.2020e+00,
-2.6975e+00, -2.1870e+00],
...,
[-2.5072e+00, -1.8795e+00, -5.3576e+00, ..., -3.0426e+00,
-2.6309e+00, -2.1108e+00],
[-2.2036e+00, -1.6385e+00, -4.2055e+00, ..., -2.6818e+00,
-2.2467e+00, -1.5685e+00],
[-2.3057e+00, -1.8401e+00, -3.2871e+00, ..., -3.0279e+00,
-2.4578e+00, -2.1370e+00]],
[[-1.3488e+00, -1.9104e+00, -1.7018e+00, ..., -1.7524e+00,
-1.4498e+00, -1.1003e+00],
[-1.2726e+00, -1.7662e+00, -1.2386e+00, ..., -1.3736e+00,
-1.0869e+00, -8.6533e-01],
[-1.6014e+00, -1.9575e+00, -1.4951e+00, ..., -1.6047e+00,
-1.2081e+00, -8.9280e-01],
...,
[-3.0937e+00, -3.8888e+00, -3.7564e+00, ..., -4.0878e+00,
-3.2774e+00, -2.5264e+00],
[-2.0791e+00, -2.6848e+00, -2.7142e+00, ..., -2.8534e+00,
-2.3509e+00, -1.9231e+00],
[-1.5280e+00, -1.8554e+00, -1.9478e+00, ..., -1.9067e+00,
-1.5593e+00, -1.1938e+00]]],
[[[ 2.4274e-01, 5.2918e-01, 6.3673e-01, ..., -5.0583e-01,
-2.3600e-01, -9.1435e-02],
[-1.8929e+00, -1.1778e+00, -2.6323e+00, ..., -3.5635e+00,
-1.4920e+00, -5.8919e-01],
[-7.0375e-02, -7.6149e-01, 4.9140e-02, ..., -3.3110e-01,
-5.0404e-01, 8.6558e-02],
...,
[-8.7944e-01, -1.9103e+00, -1.8811e+00, ..., -1.9707e+00,
-1.9641e+00, -1.2844e+00],
[ 8.2144e-01, 7.9093e-02, -6.4152e-03, ..., -7.8442e-02,
1.5439e-01, 3.7016e-01],
[ 2.7084e-01, -3.7672e-01, -2.9389e-01, ..., 1.0247e-01,
3.7687e-01, 4.4604e-01]],
[[-1.7607e+00, -1.3961e+00, -4.2700e+00, ..., -7.0359e+00,
-6.2689e+00, -4.6322e+00],
[ 3.4526e-01, 1.4049e+00, -2.5990e+00, ..., -5.6013e+00,
-5.7564e+00, -4.3933e+00],
[ 1.0607e+00, 1.1505e+00, -5.0869e+00, ..., -8.0787e+00,
-8.0508e+00, -6.7122e+00],
...,
[ 2.4869e-01, 2.2926e-01, -4.2428e+00, ..., -7.3404e+00,
-6.9478e+00, -5.3637e+00],
[ 2.4192e+00, 2.4013e+00, 1.9981e-02, ..., -2.0627e+00,
-2.7297e+00, -2.1433e+00],
[ 1.4224e+00, 1.2770e+00, -1.6593e+00, ..., -2.9841e+00,
-3.5748e+00, -3.0993e+00]],
[[-1.8277e-01, -1.2037e+00, -1.3917e+00, ..., -1.5129e+00,
-1.1241e+00, -1.4000e+00],
[-5.5879e-01, -8.9447e-01, -8.8407e-01, ..., -1.0540e+00,
-6.6332e-01, -9.6105e-01],
[-3.8444e-01, -3.0194e-01, -4.4085e-01, ..., -1.1773e+00,
-6.3980e-01, -4.9136e-01],
...,
[-8.4863e-02, -6.4720e-01, -5.9491e-01, ..., -1.0333e+00,
-8.9211e-01, -1.2700e+00],
[-7.0538e-01, -2.5689e-01, -5.3269e-01, ..., -5.5846e-01,
-2.6031e-01, -4.3128e-01],
[-8.7105e-01, -2.9488e-01, -4.4405e-01, ..., -5.8289e-01,
-5.5822e-01, -7.2220e-01]],
...,
[[-2.7272e+00, -2.2924e+00, -2.3934e+00, ..., -3.0664e+00,
-2.5181e+00, -1.8426e+00],
[-1.9274e+00, -6.2462e-01, -1.0206e+00, ..., -2.5370e+00,
-2.6412e+00, -2.1145e+00],
[-1.0516e+00, 9.0821e-01, 7.6137e-01, ..., -7.9073e-02,
-1.9802e-02, 2.6606e-01],
...,
[-2.2474e+00, -2.0783e-01, -1.6509e-01, ..., -9.6915e-01,
-3.0375e-01, 2.5467e-01],
[-1.2114e+00, 7.6151e-01, 9.7513e-01, ..., 5.1697e-01,
5.1821e-01, 8.5443e-01],
[-2.1533e+00, -1.5925e+00, -1.7512e+00, ..., -2.1229e+00,
-1.0922e+00, -2.9134e-01]],
[[-1.5966e+00, -1.5915e+00, -2.7261e+00, ..., -2.3643e+00,
-2.0449e+00, -1.9463e+00],
[-7.0295e-01, -1.3018e-01, -3.1665e+00, ..., -2.9953e+00,
-2.9980e+00, -3.0024e+00],
[-2.0668e+00, -1.0535e+00, -5.1725e+00, ..., -4.5454e+00,
-4.6218e+00, -3.8117e+00],
...,
[-1.4054e+00, -1.9338e-02, -3.4658e+00, ..., -2.8657e+00,
-2.8196e+00, -2.3799e+00],
[-1.6776e+00, -6.0847e-01, -3.0694e+00, ..., -2.4829e+00,
-2.3318e+00, -1.7439e+00],
[-2.3011e+00, -1.4371e+00, -3.0287e+00, ..., -2.9030e+00,
-2.4956e+00, -2.2680e+00]],
[[-1.2731e+00, -1.8009e+00, -1.6261e+00, ..., -2.6846e+00,
-2.2369e+00, -1.7462e+00],
[-1.2178e+00, -1.6849e+00, -1.2175e+00, ..., -2.5181e+00,
-2.0277e+00, -1.6667e+00],
[-1.5398e+00, -1.8773e+00, -1.4710e+00, ..., -2.9627e+00,
-2.3670e+00, -1.8461e+00],
...,
[-2.4174e+00, -3.0091e+00, -2.7851e+00, ..., -4.1418e+00,
-3.3662e+00, -2.6395e+00],
[-1.5704e+00, -2.0571e+00, -1.9866e+00, ..., -2.8555e+00,
-2.3905e+00, -1.9748e+00],
[-1.0297e+00, -1.3218e+00, -1.3601e+00, ..., -1.8981e+00,
-1.5737e+00, -1.2194e+00]]]], device='cuda:0'),
tensor([[[[ 3.5065e-01, 8.1743e-01, 7.3124e-01, ..., 6.9418e-01,
9.4953e-01, 7.3189e-01],
[-1.2350e+00, -2.9214e-01, -1.3936e+00, ..., -1.0840e+00,
6.7401e-02, 5.6079e-01],
[ 6.3629e-01, 4.4616e-01, 4.9363e-01, ..., 6.0277e-01,
9.5724e-01, 8.7730e-01],
...,
[-1.7994e-01, -1.0561e+00, -2.4565e+00, ..., -1.6872e+00,
-1.6218e+00, -1.0979e+00],
[ 1.3311e+00, 7.9254e-01, -9.2870e-01, ..., -2.9583e-01,
7.5021e-02, 3.3997e-01],
[ 7.8341e-01, 2.5796e-01, -6.7460e-01, ..., -1.9213e-01,
2.2684e-01, 3.8235e-01]],
[[-2.3759e+00, -1.2975e+00, -4.2549e+00, ..., -5.2721e+00,
-4.5128e+00, -3.5340e+00],
[-1.8379e-01, 1.6224e+00, -2.5445e+00, ..., -4.1293e+00,
-3.9822e+00, -3.2806e+00],
[ 6.6854e-01, 1.4479e+00, -5.0894e+00, ..., -6.0083e+00,
-5.7495e+00, -5.3106e+00],
...,
[-1.5344e+00, -2.8180e-01, -6.8877e+00, ..., -7.5787e+00,
-6.8300e+00, -5.3143e+00],
[ 1.3851e+00, 2.3150e+00, -1.4842e+00, ..., -2.2645e+00,
-2.6574e+00, -2.1392e+00],
[ 9.2828e-01, 1.4981e+00, -2.4470e+00, ..., -3.1060e+00,
-3.5503e+00, -3.1171e+00]],
[[-6.8387e-01, -1.8042e+00, -1.6943e+00, ..., -9.3642e-01,
-6.7106e-01, -7.7636e-01],
[-1.3775e+00, -1.8047e+00, -1.2544e+00, ..., -7.3673e-01,
-3.9356e-01, -2.7273e-01],
[-1.0472e+00, -9.4949e-01, -4.5925e-01, ..., -3.2800e-01,
3.0747e-02, 4.7370e-02],
...,
[-1.2019e+00, -1.5878e+00, -7.4804e-01, ..., -9.0906e-01,
-7.8862e-01, -1.0676e+00],
[-1.5267e+00, -1.0320e+00, -7.9240e-01, ..., -5.4795e-01,
-2.7165e-01, -3.5422e-01],
[-1.6818e+00, -1.1859e+00, -8.8006e-01, ..., -9.2953e-01,
-7.9718e-01, -7.9274e-01]],
...,
[[-2.2148e+00, -1.2633e+00, -2.2897e+00, ..., -2.9132e+00,
-2.2920e+00, -1.5098e+00],
[-5.9469e-01, 1.0658e+00, -4.0094e-01, ..., -1.3336e+00,
-1.3994e+00, -1.2280e+00],
[-3.5308e-01, 2.0569e+00, 8.3902e-01, ..., -1.4326e-01,
2.3724e-01, 3.4735e-01],
...,
[-1.7451e+00, 1.0216e+00, -7.3263e-02, ..., -1.1662e+00,
-5.2739e-01, 4.6606e-02],
[-7.8936e-01, 1.6943e+00, 1.4924e+00, ..., 4.9521e-01,
4.7192e-01, 7.4898e-01],
[-2.1691e+00, -1.1521e+00, -9.3577e-01, ..., -2.1157e+00,
-1.0890e+00, -3.5646e-01]],
[[-1.3536e+00, -1.2476e+00, -2.6717e+00, ..., -1.0938e+00,
-5.8520e-01, -4.5658e-01],
[-6.4653e-01, -1.2259e-01, -3.4822e+00, ..., -1.7683e+00,
-1.5105e+00, -1.6900e+00],
[-2.0923e+00, -1.3390e+00, -5.8326e+00, ..., -3.7293e+00,
-3.2070e+00, -2.5524e+00],
...,
[-2.5446e+00, -1.8220e+00, -5.2991e+00, ..., -2.9457e+00,
-2.5668e+00, -2.0616e+00],
[-2.2534e+00, -1.6451e+00, -4.1418e+00, ..., -2.5980e+00,
-2.2031e+00, -1.5460e+00],
[-2.3562e+00, -1.8559e+00, -3.2506e+00, ..., -3.0072e+00,
-2.4430e+00, -2.1582e+00]],
[[-1.1594e+00, -1.7189e+00, -1.5076e+00, ..., -1.6386e+00,
-1.3631e+00, -1.0221e+00],
[-1.1642e+00, -1.6361e+00, -1.1768e+00, ..., -1.3655e+00,
-1.1093e+00, -8.7470e-01],
[-1.5133e+00, -1.8000e+00, -1.3542e+00, ..., -1.5523e+00,
-1.1811e+00, -8.6121e-01],
...,
[-3.0551e+00, -3.8254e+00, -3.7033e+00, ..., -4.0563e+00,
-3.2607e+00, -2.5076e+00],
[-2.0363e+00, -2.6376e+00, -2.6637e+00, ..., -2.8258e+00,
-2.3412e+00, -1.9064e+00],
[-1.4920e+00, -1.8156e+00, -1.9106e+00, ..., -1.8862e+00,
-1.5435e+00, -1.1755e+00]]],
[[[ 5.0005e-02, 1.8699e-01, -1.5128e+00, ..., -6.7214e-01,
-2.4362e-01, -9.2493e-02],
[-2.1354e+00, -1.6795e+00, -4.6807e+00, ..., -3.8452e+00,
-1.5748e+00, -5.9803e-01],
[ 5.4004e-01, -1.3622e-01, -9.2456e-01, ..., -2.7754e-01,
-3.4658e-01, 1.9975e-01],
...,
[-8.2673e-01, -1.8572e+00, -1.9555e+00, ..., -1.7213e+00,
-1.2023e+00, -8.3268e-01],
[ 8.8138e-01, 1.1635e-01, -4.8754e-02, ..., -3.3546e-02,
7.2730e-01, 8.1357e-01],
[ 2.4675e-01, -4.4703e-01, -4.1610e-01, ..., -1.5642e-01,
4.5454e-01, 3.6871e-01]],
[[-3.3405e+00, -1.8322e+00, -6.3435e+00, ..., -7.1991e+00,
-6.3167e+00, -4.6606e+00],
[-8.4098e-01, 1.5003e+00, -4.4392e+00, ..., -5.5519e+00,
-5.6162e+00, -4.3222e+00],
[-4.3010e-01, 1.0414e+00, -7.5218e+00, ..., -8.0515e+00,
-7.8966e+00, -6.7246e+00],
...,
[ 4.7092e-01, 6.8683e-01, -3.8597e+00, ..., -4.6828e+00,
-4.2647e+00, -3.5416e+00],
[ 2.6446e+00, 2.7230e+00, 3.2969e-01, ..., -9.6817e-01,
-1.3190e+00, -1.3355e+00],
[ 1.5846e+00, 1.4483e+00, -1.5509e+00, ..., -2.3129e+00,
-2.5582e+00, -2.4300e+00]],
[[-1.3993e+00, -2.4010e+00, -2.3602e+00, ..., -1.7708e+00,
-1.2570e+00, -1.4125e+00],
[-1.7277e+00, -2.0111e+00, -1.5468e+00, ..., -1.2627e+00,
-7.3967e-01, -8.5847e-01],
[-2.2354e+00, -1.9860e+00, -1.0244e+00, ..., -1.1413e+00,
-4.8444e-01, -2.0783e-01],
...,
[-3.4653e-01, -8.0584e-01, -5.0913e-01, ..., -4.9304e-01,
-2.0362e-01, -4.3947e-01],
[-9.0769e-01, -3.3259e-01, -4.7116e-01, ..., -2.1321e-01,
2.2570e-01, 2.2062e-01],
[-1.0005e+00, -2.8090e-01, -3.2873e-01, ..., -4.0546e-01,
-2.0050e-01, -2.3806e-01]],
...,
[[-2.0209e+00, -1.0995e+00, -2.3693e+00, ..., -3.0995e+00,
-2.5548e+00, -1.7871e+00],
[-1.8085e+00, -3.7962e-02, -1.4255e+00, ..., -2.4925e+00,
-2.5340e+00, -1.9623e+00],
[-3.9080e-01, 2.3386e+00, 1.0376e+00, ..., 1.1886e-01,
2.9991e-01, 5.3658e-01],
...,
[-2.1290e+00, 1.8292e-01, -1.1395e-01, ..., -1.2791e+00,
-4.2684e-01, 6.9396e-02],
[-1.1918e+00, 1.0286e+00, 1.0921e+00, ..., 7.6059e-02,
2.8867e-01, 4.4109e-01],
[-2.1208e+00, -1.4310e+00, -1.7067e+00, ..., -2.2847e+00,
-1.3930e+00, -8.8099e-01]],
[[-2.4619e+00, -2.9003e+00, -4.2542e+00, ..., -2.5001e+00,
-2.0086e+00, -1.7841e+00],
[-1.8234e+00, -1.7186e+00, -4.9192e+00, ..., -2.9806e+00,
-2.8240e+00, -2.8007e+00],
[-2.8731e+00, -2.5276e+00, -6.6910e+00, ..., -4.4622e+00,
-4.3764e+00, -3.5084e+00],
...,
[-1.3511e+00, 1.2406e-01, -3.5822e+00, ..., -1.4830e+00,
-1.1053e+00, -9.9304e-01],
[-1.6601e+00, -5.0253e-01, -3.1794e+00, ..., -1.7328e+00,
-1.2101e+00, -8.5570e-01],
[-2.3223e+00, -1.4129e+00, -3.1804e+00, ..., -2.8993e+00,
-2.1334e+00, -1.9560e+00]],
[[-1.8343e+00, -2.5727e+00, -2.4941e+00, ..., -2.7507e+00,
-2.2614e+00, -1.7365e+00],
[-1.8316e+00, -2.5569e+00, -2.1906e+00, ..., -2.4821e+00,
-1.9659e+00, -1.5930e+00],
[-2.2859e+00, -2.8651e+00, -2.5648e+00, ..., -2.8428e+00,
-2.2248e+00, -1.6836e+00],
...,
[-2.4906e+00, -3.0783e+00, -2.7769e+00, ..., -2.8135e+00,
-2.2407e+00, -1.7211e+00],
[-1.6232e+00, -2.0980e+00, -1.9724e+00, ..., -1.9208e+00,
-1.5776e+00, -1.2687e+00],
[-1.0630e+00, -1.3430e+00, -1.3449e+00, ..., -1.1845e+00,
-1.0004e+00, -7.5946e-01]]],
[[[ 9.7189e-02, 2.5424e-01, -1.3892e+00, ..., -6.2603e-01,
-1.8201e-01, -1.9383e-02],
[-2.1877e+00, -1.6437e+00, -4.6327e+00, ..., -3.5908e+00,
-1.3792e+00, -4.7264e-01],
[ 6.3321e-01, -8.3535e-02, -8.5774e-01, ..., -1.7519e-01,
-1.9754e-01, 2.9383e-01],
...,
[-8.8312e-01, -1.8646e+00, -1.8625e+00, ..., -1.8565e+00,
-1.3267e+00, -9.3277e-01],
[ 7.2982e-01, -2.5420e-02, -1.1495e-01, ..., -5.5592e-02,
7.0403e-01, 7.9340e-01],
[ 1.2061e-01, -5.9646e-01, -4.8664e-01, ..., -1.8426e-01,
4.3149e-01, 3.3630e-01]],
[[-3.3269e+00, -1.6385e+00, -6.2130e+00, ..., -6.9373e+00,
-6.1163e+00, -4.5304e+00],
[-5.9767e-01, 1.9919e+00, -4.1102e+00, ..., -5.5204e+00,
-5.5788e+00, -4.3170e+00],
[-2.2837e-02, 1.5915e+00, -7.3429e+00, ..., -8.0044e+00,
-7.8367e+00, -6.6150e+00],
...,
[ 3.2507e-01, 3.9336e-01, -4.0141e+00, ..., -4.6379e+00,
-4.2136e+00, -3.4951e+00],
[ 2.3363e+00, 2.2955e+00, -1.5471e-01, ..., -8.2275e-01,
-1.1904e+00, -1.2267e+00],
[ 1.3743e+00, 1.1803e+00, -1.7737e+00, ..., -2.2589e+00,
-2.5264e+00, -2.4214e+00]],
[[-1.3992e+00, -2.5329e+00, -2.4495e+00, ..., -1.6724e+00,
-1.2134e+00, -1.3238e+00],
[-1.8582e+00, -2.1942e+00, -1.6347e+00, ..., -1.1585e+00,
-6.9312e-01, -8.0583e-01],
[-2.3407e+00, -2.0730e+00, -1.0436e+00, ..., -1.0923e+00,
-4.7967e-01, -2.1033e-01],
...,
[-1.4937e-01, -6.0881e-01, -4.6477e-01, ..., -5.0357e-01,
-1.8405e-01, -4.4163e-01],
[-7.1448e-01, -1.9712e-01, -4.2943e-01, ..., -2.2773e-01,
2.4252e-01, 2.4191e-01],
[-8.1118e-01, -1.9025e-01, -3.4092e-01, ..., -4.1714e-01,
-2.0193e-01, -2.3134e-01]],
...,
[[-2.2505e+00, -1.1137e+00, -2.4583e+00, ..., -2.9250e+00,
-2.4294e+00, -1.6969e+00],
[-1.7988e+00, 2.4745e-01, -1.3287e+00, ..., -2.3802e+00,
-2.4015e+00, -1.8184e+00],
[-3.4480e-01, 2.7467e+00, 1.2668e+00, ..., 1.6368e-01,
2.2375e-01, 4.7132e-01],
...,
[-2.0798e+00, -8.6962e-02, -1.9324e-01, ..., -1.3597e+00,
-4.4203e-01, 9.8707e-02],
[-1.1929e+00, 8.2348e-01, 9.4425e-01, ..., 3.7334e-02,
2.9042e-01, 4.7077e-01],
[-2.0911e+00, -1.3963e+00, -1.6197e+00, ..., -2.4285e+00,
-1.4767e+00, -9.3618e-01]],
[[-2.4021e+00, -2.7665e+00, -4.3885e+00, ..., -2.4923e+00,
-2.0392e+00, -1.8318e+00],
[-1.6515e+00, -1.4257e+00, -5.0963e+00, ..., -3.0492e+00,
-2.8971e+00, -2.8168e+00],
[-2.8005e+00, -2.2747e+00, -7.0046e+00, ..., -4.4311e+00,
-4.3460e+00, -3.4635e+00],
...,
[-1.3954e+00, -1.5439e-02, -3.3807e+00, ..., -1.4009e+00,
-1.0145e+00, -9.4860e-01],
[-1.7070e+00, -6.2749e-01, -3.0360e+00, ..., -1.6980e+00,
-1.1488e+00, -8.1806e-01],
[-2.3211e+00, -1.4766e+00, -3.0991e+00, ..., -2.9363e+00,
-2.1426e+00, -1.9819e+00]],
[[-1.8627e+00, -2.6195e+00, -2.4741e+00, ..., -2.6309e+00,
-2.1605e+00, -1.6645e+00],
[-1.8445e+00, -2.5769e+00, -2.1187e+00, ..., -2.4044e+00,
-1.9016e+00, -1.5443e+00],
[-2.3137e+00, -2.8768e+00, -2.4753e+00, ..., -2.7756e+00,
-2.1706e+00, -1.6598e+00],
...,
[-2.3932e+00, -2.9642e+00, -2.7355e+00, ..., -2.8716e+00,
-2.2925e+00, -1.7648e+00],
[-1.5534e+00, -2.0205e+00, -1.9457e+00, ..., -1.9563e+00,
-1.6084e+00, -1.3009e+00],
[-1.0139e+00, -1.3093e+00, -1.3371e+00, ..., -1.1959e+00,
-1.0212e+00, -7.7523e-01]]],
[[[ 3.9564e-02, 1.7787e-01, -1.5594e+00, ..., -7.0360e-01,
-2.6557e-01, -8.4504e-02],
[-2.1897e+00, -1.7163e+00, -4.7282e+00, ..., -3.9438e+00,
-1.6214e+00, -6.0531e-01],
[ 5.1402e-01, -1.4295e-01, -9.3954e-01, ..., -2.7307e-01,
-3.2406e-01, 2.4479e-01],
...,
[-7.7461e-01, -1.7336e+00, -1.6748e+00, ..., -1.7814e+00,
-1.3054e+00, -9.4555e-01],
[ 7.9731e-01, 6.9033e-02, 1.9450e-02, ..., 2.7638e-02,
7.3798e-01, 8.1090e-01],
[ 1.8543e-01, -5.1524e-01, -3.7451e-01, ..., -1.0239e-01,
4.8346e-01, 3.8018e-01]],
[[-3.3652e+00, -1.8504e+00, -6.3692e+00, ..., -7.2331e+00,
-6.3149e+00, -4.6621e+00],
[-8.3667e-01, 1.5447e+00, -4.4234e+00, ..., -5.5441e+00,
-5.6184e+00, -4.3231e+00],
[-4.0671e-01, 1.1067e+00, -7.5254e+00, ..., -8.0956e+00,
-7.9623e+00, -6.7776e+00],
...,
[ 2.3608e-01, 2.8246e-01, -4.1609e+00, ..., -4.6793e+00,
-4.2275e+00, -3.5175e+00],
[ 2.2118e+00, 2.1787e+00, -2.6525e-01, ..., -8.0840e-01,
-1.1575e+00, -1.2055e+00],
[ 1.2801e+00, 1.1130e+00, -1.8231e+00, ..., -2.2376e+00,
-2.5171e+00, -2.4187e+00]],
[[-1.4088e+00, -2.4088e+00, -2.3794e+00, ..., -1.7729e+00,
-1.2622e+00, -1.4214e+00],
[-1.7426e+00, -2.0190e+00, -1.5537e+00, ..., -1.2505e+00,
-7.4066e-01, -8.7410e-01],
[-2.2590e+00, -2.0024e+00, -1.0324e+00, ..., -1.1273e+00,
-4.8016e-01, -2.1008e-01],
...,
[-1.6484e-01, -6.3284e-01, -4.4844e-01, ..., -5.0405e-01,
-2.0886e-01, -4.6252e-01],
[-6.8306e-01, -2.0820e-01, -3.9626e-01, ..., -1.9691e-01,
2.5172e-01, 2.2940e-01],
[-7.9197e-01, -2.1272e-01, -3.2393e-01, ..., -3.7660e-01,
-1.7128e-01, -2.1406e-01]],
...,
[[-2.0318e+00, -1.1346e+00, -2.3911e+00, ..., -3.2094e+00,
-2.6604e+00, -1.8600e+00],
[-1.8325e+00, -8.4314e-02, -1.4580e+00, ..., -2.5945e+00,
-2.6268e+00, -2.0105e+00],
[-3.9757e-01, 2.3268e+00, 1.0267e+00, ..., 1.0105e-01,
2.1812e-01, 5.3053e-01],
...,
[-2.0787e+00, -9.6952e-02, -2.3159e-01, ..., -1.3233e+00,
-3.5528e-01, 1.3775e-01],
[-1.1815e+00, 7.7538e-01, 8.3456e-01, ..., 3.5051e-02,
3.3258e-01, 4.9656e-01],
[-2.0682e+00, -1.4056e+00, -1.6508e+00, ..., -2.4827e+00,
-1.4611e+00, -9.0610e-01]],
[[-2.4687e+00, -2.9181e+00, -4.2727e+00, ..., -2.4358e+00,
-1.9644e+00, -1.7956e+00],
[-1.8173e+00, -1.7175e+00, -4.9367e+00, ..., -2.9568e+00,
-2.8659e+00, -2.8762e+00],
[-2.8715e+00, -2.5309e+00, -6.7211e+00, ..., -4.4573e+00,
-4.4454e+00, -3.6000e+00],
...,
[-1.4233e+00, -1.7342e-01, -3.4572e+00, ..., -1.4639e+00,
-1.0575e+00, -9.4070e-01],
[-1.7146e+00, -7.3896e-01, -3.0765e+00, ..., -1.7396e+00,
-1.1720e+00, -8.0097e-01],
[-2.3044e+00, -1.5411e+00, -3.1162e+00, ..., -2.9478e+00,
-2.1205e+00, -1.9715e+00]],
[[-1.8525e+00, -2.5966e+00, -2.5232e+00, ..., -2.7753e+00,
-2.2793e+00, -1.7481e+00],
[-1.8468e+00, -2.5743e+00, -2.2147e+00, ..., -2.4858e+00,
-1.9680e+00, -1.5953e+00],
[-2.3067e+00, -2.8935e+00, -2.5867e+00, ..., -2.8490e+00,
-2.2267e+00, -1.6941e+00],
...,
[-2.3637e+00, -2.9164e+00, -2.7118e+00, ..., -2.8690e+00,
-2.2974e+00, -1.7639e+00],
[-1.5319e+00, -1.9844e+00, -1.9279e+00, ..., -1.9519e+00,
-1.6148e+00, -1.3057e+00],
[-9.9973e-01, -1.2816e+00, -1.3220e+00, ..., -1.1939e+00,
-1.0267e+00, -7.7514e-01]]],
[[[ 2.2196e-01, 4.8711e-01, 6.4727e-01, ..., 6.1973e-01,
9.7593e-01, 7.9021e-01],
[-2.0363e+00, -1.3339e+00, -2.8425e+00, ..., -1.2527e+00,
8.6068e-02, 6.2933e-01],
[-1.9414e-01, -8.8975e-01, 4.1265e-02, ..., 5.1542e-01,
9.3647e-01, 9.3509e-01],
...,
[-9.8549e-01, -2.1548e+00, -2.0190e+00, ..., -1.4155e+00,
-8.6100e-01, -6.3205e-01],
[ 7.7647e-01, -1.2373e-01, -8.8837e-02, ..., 3.6942e-01,
8.2533e-01, 9.0992e-01],
[ 1.9207e-01, -5.8113e-01, -3.8777e-01, ..., 5.7456e-01,
9.1797e-01, 8.0233e-01]],
[[-1.6384e+00, -1.3521e+00, -4.2511e+00, ..., -5.7570e+00,
-4.9020e+00, -3.8244e+00],
[ 4.9490e-01, 1.4593e+00, -2.5639e+00, ..., -4.7307e+00,
-4.4632e+00, -3.6613e+00],
[ 1.2181e+00, 1.1692e+00, -5.0874e+00, ..., -6.8467e+00,
-6.3738e+00, -5.7740e+00],
...,
[ 4.1868e-01, 3.7021e-01, -4.1480e+00, ..., -5.9795e+00,
-5.1825e+00, -4.2350e+00],
[ 2.5591e+00, 2.5005e+00, 1.0470e-01, ..., -1.2969e+00,
-1.4407e+00, -1.5137e+00],
[ 1.5182e+00, 1.3232e+00, -1.6248e+00, ..., -2.7186e+00,
-2.8462e+00, -2.7769e+00]],
[[-1.2448e-01, -1.1580e+00, -1.3918e+00, ..., -8.1125e-01,
-5.4798e-01, -6.6440e-01],
[-4.7842e-01, -8.0404e-01, -8.5161e-01, ..., -6.8860e-01,
-3.6854e-01, -2.0956e-01],
[-3.0679e-01, -2.0515e-01, -4.1993e-01, ..., -3.7962e-01,
-2.2043e-03, 1.0728e-01],
...,
[-6.0423e-02, -6.0661e-01, -6.2291e-01, ..., -6.7582e-01,
-4.6543e-01, -5.7470e-01],
[-7.1478e-01, -2.2604e-01, -5.7194e-01, ..., -2.9545e-01,
9.9840e-03, 8.5857e-02],
[-8.7619e-01, -2.6827e-01, -4.7754e-01, ..., -3.2521e-01,
-2.0501e-01, -1.4141e-01]],
...,
[[-2.8384e+00, -2.5022e+00, -2.4917e+00, ..., -3.1370e+00,
-2.5531e+00, -1.6977e+00],
[-2.0983e+00, -8.9209e-01, -1.1561e+00, ..., -1.6384e+00,
-1.7109e+00, -1.4501e+00],
[-1.1058e+00, 7.4165e-01, 7.0193e-01, ..., -2.6343e-01,
8.1417e-03, 1.4067e-01],
...,
[-2.2739e+00, -2.2973e-01, -2.0372e-01, ..., -1.0627e+00,
-4.3470e-01, 1.6154e-01],
[-1.2254e+00, 8.6433e-01, 1.0363e+00, ..., 2.7287e-01,
3.1211e-01, 4.4392e-01],
[-2.1631e+00, -1.5051e+00, -1.7105e+00, ..., -2.1972e+00,
-1.2686e+00, -6.3182e-01]],
[[-1.6213e+00, -1.6375e+00, -2.7956e+00, ..., -1.3432e+00,
-7.3727e-01, -5.2446e-01],
[-6.9520e-01, -1.1760e-01, -3.1657e+00, ..., -2.0226e+00,
-1.6364e+00, -1.6882e+00],
[-2.0456e+00, -9.9536e-01, -5.1353e+00, ..., -4.0101e+00,
-3.3605e+00, -2.5347e+00],
...,
[-1.3185e+00, 1.8610e-01, -3.3902e+00, ..., -2.3405e+00,
-1.7695e+00, -1.3377e+00],
[-1.6661e+00, -4.7831e-01, -3.0197e+00, ..., -2.2358e+00,
-1.4846e+00, -8.7515e-01],
[-2.3219e+00, -1.3681e+00, -3.0506e+00, ..., -2.8712e+00,
-1.9224e+00, -1.6152e+00]],
[[-1.2937e+00, -1.8161e+00, -1.6449e+00, ..., -1.8578e+00,
-1.5005e+00, -1.1005e+00],
[-1.2414e+00, -1.6992e+00, -1.2256e+00, ..., -1.6201e+00,
-1.2840e+00, -9.7841e-01],
[-1.5549e+00, -1.8899e+00, -1.4797e+00, ..., -1.8978e+00,
-1.4245e+00, -1.0080e+00],
...,
[-2.4542e+00, -3.0601e+00, -2.7817e+00, ..., -3.1649e+00,
-2.4836e+00, -1.8560e+00],
[-1.5873e+00, -2.0846e+00, -1.9779e+00, ..., -2.2293e+00,
-1.8051e+00, -1.4202e+00],
[-1.0361e+00, -1.3430e+00, -1.3519e+00, ..., -1.3793e+00,
-1.1554e+00, -8.3993e-01]]]], device='cuda:0'),
tensor([[[[ 4.5492e-01, 1.2714e-02, -1.3414e+00, ..., 7.7654e-01,
1.0568e+00, 8.3799e-01],
[-1.0540e+00, -1.8723e+00, -7.7997e+00, ..., -1.2352e+00,
5.4313e-02, 6.3510e-01],
[ 7.2866e-01, 2.6403e-01, -1.1164e+00, ..., 4.6686e-01,
8.4221e-01, 8.7388e-01],
...,
[-8.2149e-01, -1.7450e+00, -1.7784e+00, ..., -1.7497e+00,
-1.3036e+00, -9.4255e-01],
[ 7.5447e-01, 5.7677e-02, -5.1961e-02, ..., 5.8047e-03,
7.1146e-01, 8.0568e-01],
[ 1.4963e-01, -5.0933e-01, -4.2586e-01, ..., -1.1626e-01,
4.5042e-01, 3.7442e-01]],
[[-2.5911e+00, -1.0396e+00, -4.1913e+00, ..., -5.8715e+00,
-5.0428e+00, -3.9149e+00],
[-6.5578e-01, 1.6301e+00, -2.7446e+00, ..., -4.7402e+00,
-4.5725e+00, -3.7072e+00],
[-5.7266e-04, 1.1496e+00, -6.4883e+00, ..., -6.9046e+00,
-6.5904e+00, -5.9729e+00],
...,
[ 2.6258e-01, 3.6936e-01, -4.0145e+00, ..., -4.6597e+00,
-4.1911e+00, -3.4985e+00],
[ 2.2950e+00, 2.2675e+00, -1.6742e-01, ..., -8.2757e-01,
-1.1564e+00, -1.2156e+00],
[ 1.3458e+00, 1.1655e+00, -1.7858e+00, ..., -2.2061e+00,
-2.4818e+00, -2.3978e+00]],
[[-5.4412e-01, -1.7839e+00, -1.0730e+00, ..., -7.5566e-01,
-5.1239e-01, -7.3934e-01],
[-1.2952e+00, -1.7230e+00, -2.3778e-01, ..., -7.0657e-01,
-3.8219e-01, -3.2306e-01],
[-1.1772e+00, -1.3260e+00, -1.2377e+00, ..., -4.4254e-01,
-9.4750e-02, -8.8326e-02],
...,
[-1.8525e-01, -6.3740e-01, -4.5373e-01, ..., -4.7740e-01,
-1.8560e-01, -4.5182e-01],
[-7.3474e-01, -2.1114e-01, -4.1168e-01, ..., -1.7494e-01,
2.5497e-01, 2.4110e-01],
[-8.3499e-01, -2.0864e-01, -3.2336e-01, ..., -3.6374e-01,
-1.7407e-01, -2.1850e-01]],
...,
[[-2.0355e+00, -1.4375e+00, -4.0106e+00, ..., -3.3021e+00,
-2.7318e+00, -1.8464e+00],
[-4.8785e-01, 2.1227e-01, -4.9080e+00, ..., -1.6560e+00,
-1.8307e+00, -1.6024e+00],
[-2.1742e-01, 2.0041e+00, -1.0254e+00, ..., -3.4923e-01,
-7.4755e-02, 1.0651e-01],
...,
[-2.0064e+00, 1.2148e-02, -1.3907e-01, ..., -1.3433e+00,
-3.7470e-01, 1.2467e-01],
[-1.1451e+00, 8.4601e-01, 9.5039e-01, ..., -1.8412e-02,
3.1220e-01, 4.8277e-01],
[-2.0484e+00, -1.3703e+00, -1.5801e+00, ..., -2.4852e+00,
-1.4770e+00, -9.1913e-01]],
[[-1.4726e+00, -1.2395e+00, -2.7564e+00, ..., -1.1382e+00,
-6.2087e-01, -5.0611e-01],
[-9.1840e-01, -3.3311e-01, -3.4728e+00, ..., -1.9335e+00,
-1.6823e+00, -1.8691e+00],
[-2.4416e+00, -1.7657e+00, -6.4080e+00, ..., -4.2170e+00,
-3.6746e+00, -2.8884e+00],
...,
[-1.4060e+00, -6.4109e-02, -3.4596e+00, ..., -1.4673e+00,
-1.0233e+00, -9.3634e-01],
[-1.6902e+00, -6.2179e-01, -3.0677e+00, ..., -1.7428e+00,
-1.1561e+00, -8.0530e-01],
[-2.2910e+00, -1.4434e+00, -3.0722e+00, ..., -2.9356e+00,
-2.1286e+00, -1.9611e+00]],
[[-1.2449e+00, -1.9592e+00, -2.4386e+00, ..., -1.8768e+00,
-1.5533e+00, -1.1565e+00],
[-1.3129e+00, -1.9777e+00, -2.1733e+00, ..., -1.6457e+00,
-1.3365e+00, -1.0464e+00],
[-1.7350e+00, -2.1890e+00, -2.1469e+00, ..., -1.9430e+00,
-1.4954e+00, -1.0986e+00],
...,
[-2.3811e+00, -2.9467e+00, -2.7160e+00, ..., -2.8680e+00,
-2.2971e+00, -1.7645e+00],
[-1.5537e+00, -2.0184e+00, -1.9434e+00, ..., -1.9448e+00,
-1.6078e+00, -1.2993e+00],
[-1.0201e+00, -1.3122e+00, -1.3443e+00, ..., -1.1878e+00,
-1.0160e+00, -7.7668e-01]]],
[[[ 4.4369e-01, 9.4013e-01, 7.2096e-01, ..., 7.9121e-01,
1.0167e+00, 8.1071e-01],
[-1.0242e+00, -4.7548e-02, -1.4313e+00, ..., -1.2048e+00,
-3.6586e-03, 6.0247e-01],
[ 7.4764e-01, 5.7865e-01, 2.8661e-01, ..., 5.1232e-01,
8.4409e-01, 8.8297e-01],
...,
[-9.6291e-01, -2.0020e+00, -2.0810e+00, ..., -1.7771e+00,
-1.2149e+00, -8.4976e-01],
[ 8.1523e-01, 2.9213e-02, -1.4095e-01, ..., -6.0775e-02,
7.3644e-01, 7.9519e-01],
[ 1.9787e-01, -5.2871e-01, -4.9743e-01, ..., -2.2435e-01,
4.2668e-01, 3.2276e-01]],
[[-2.5775e+00, -1.4414e+00, -4.7437e+00, ..., -5.8322e+00,
-4.9696e+00, -3.8639e+00],
[-6.7430e-01, 1.2533e+00, -3.3900e+00, ..., -4.6252e+00,
-4.4525e+00, -3.6288e+00],
[-4.0399e-02, 1.0558e+00, -6.0680e+00, ..., -6.7293e+00,
-6.4671e+00, -5.8977e+00],
...,
[ 5.0647e-01, 7.2142e-01, -3.8277e+00, ..., -4.5974e+00,
-4.2370e+00, -3.5454e+00],
[ 2.7377e+00, 2.8023e+00, 4.1199e-01, ..., -8.6510e-01,
-1.2912e+00, -1.3153e+00],
[ 1.6419e+00, 1.5049e+00, -1.5202e+00, ..., -2.2763e+00,
-2.5637e+00, -2.4321e+00]],
[[-5.1173e-01, -1.5160e+00, -1.3503e+00, ..., -7.7104e-01,
-5.1937e-01, -7.2961e-01],
[-1.2518e+00, -1.6323e+00, -1.0475e+00, ..., -7.0199e-01,
-3.7592e-01, -3.0274e-01],
[-1.1332e+00, -1.0432e+00, -4.7534e-01, ..., -4.0544e-01,
-6.8892e-02, -6.4107e-02],
...,
[-2.6611e-01, -7.7115e-01, -5.2125e-01, ..., -4.9021e-01,
-2.0188e-01, -4.3878e-01],
[-8.8181e-01, -3.1769e-01, -4.9004e-01, ..., -2.2002e-01,
2.3935e-01, 2.4669e-01],
[-1.0020e+00, -3.0405e-01, -3.8080e-01, ..., -4.2618e-01,
-2.0761e-01, -2.3814e-01]],
...,
[[-2.0156e+00, -1.1425e+00, -2.3737e+00, ..., -3.2695e+00,
-2.7112e+00, -1.8417e+00],
[-5.3566e-01, 1.0731e+00, -6.7482e-01, ..., -1.5928e+00,
-1.8054e+00, -1.5761e+00],
[-2.6503e-01, 2.0548e+00, 4.2131e-01, ..., -3.5643e-01,
-6.0270e-02, 1.2392e-01],
...,
[-2.2469e+00, 6.3250e-02, -1.2254e-01, ..., -1.3103e+00,
-4.1145e-01, 8.5769e-02],
[-1.2415e+00, 9.9097e-01, 1.0729e+00, ..., 5.9561e-02,
2.6952e-01, 4.3802e-01],
[-2.1960e+00, -1.4794e+00, -1.7419e+00, ..., -2.3501e+00,
-1.4336e+00, -9.2520e-01]],
[[-1.4526e+00, -1.4265e+00, -2.7732e+00, ..., -1.0704e+00,
-5.1268e-01, -4.2642e-01],
[-9.0726e-01, -5.5698e-01, -3.7411e+00, ..., -1.8239e+00,
-1.5707e+00, -1.8042e+00],
[-2.4314e+00, -1.9768e+00, -6.2762e+00, ..., -4.1280e+00,
-3.6135e+00, -2.8499e+00],
...,
[-1.3107e+00, 2.4236e-01, -3.4628e+00, ..., -1.3922e+00,
-1.0586e+00, -9.8141e-01],
[-1.6533e+00, -4.4281e-01, -3.1060e+00, ..., -1.7084e+00,
-1.1801e+00, -8.4687e-01],
[-2.3737e+00, -1.4211e+00, -3.1439e+00, ..., -2.9374e+00,
-2.1400e+00, -1.9775e+00]],
[[-1.1998e+00, -1.7746e+00, -1.6146e+00, ..., -1.8600e+00,
-1.5317e+00, -1.1315e+00],
[-1.2659e+00, -1.7650e+00, -1.3944e+00, ..., -1.6247e+00,
-1.3058e+00, -1.0144e+00],
[-1.6770e+00, -2.0365e+00, -1.6837e+00, ..., -1.9058e+00,
-1.4496e+00, -1.0521e+00],
...,
[-2.5124e+00, -3.1090e+00, -2.8156e+00, ..., -2.8460e+00,
-2.2777e+00, -1.7542e+00],
[-1.6229e+00, -2.1100e+00, -1.9859e+00, ..., -1.9443e+00,
-1.6015e+00, -1.2919e+00],
[-1.0446e+00, -1.3490e+00, -1.3512e+00, ..., -1.1989e+00,
-1.0160e+00, -7.7023e-01]]],
[[[ 6.5999e-02, 2.0825e-01, -1.4371e+00, ..., -6.8045e-01,
-2.1861e-01, -6.8123e-02],
[-2.0943e+00, -1.6835e+00, -4.5589e+00, ..., -3.8446e+00,
-1.5601e+00, -5.8428e-01],
[ 5.4238e-01, -1.1150e-01, -8.5589e-01, ..., -2.3573e-01,
-2.7599e-01, 2.2163e-01],
...,
[-8.6564e-01, -1.8139e+00, -1.8063e+00, ..., -1.7827e+00,
-1.3356e+00, -9.3639e-01],
[ 7.3969e-01, -5.0530e-05, -8.3732e-02, ..., -1.6645e-02,
6.9519e-01, 7.9974e-01],
[ 1.3452e-01, -5.6124e-01, -4.6064e-01, ..., -1.4702e-01,
4.4324e-01, 3.5883e-01]],
[[-3.3225e+00, -1.9028e+00, -6.3197e+00, ..., -7.1523e+00,
-6.2790e+00, -4.6239e+00],
[-8.7774e-01, 1.3901e+00, -4.4064e+00, ..., -5.4860e+00,
-5.5637e+00, -4.2674e+00],
[-4.7105e-01, 9.7924e-01, -7.3984e+00, ..., -7.9933e+00,
-7.8687e+00, -6.6852e+00],
...,
[ 2.3423e-01, 2.9222e-01, -4.0420e+00, ..., -4.6261e+00,
-4.1882e+00, -3.5009e+00],
[ 2.2619e+00, 2.2318e+00, -1.4885e-01, ..., -7.9547e-01,
-1.1652e+00, -1.2194e+00],
[ 1.3284e+00, 1.1485e+00, -1.7566e+00, ..., -2.2324e+00,
-2.5193e+00, -2.4100e+00]],
[[-1.3543e+00, -2.3219e+00, -2.3030e+00, ..., -1.7663e+00,
-1.2559e+00, -1.3955e+00],
[-1.6490e+00, -1.9375e+00, -1.5093e+00, ..., -1.2534e+00,
-7.4016e-01, -8.2954e-01],
[-2.1359e+00, -1.9076e+00, -9.8982e-01, ..., -1.1147e+00,
-4.6230e-01, -1.8342e-01],
...,
[-1.1466e-01, -6.0387e-01, -4.6321e-01, ..., -4.9636e-01,
-1.9776e-01, -4.4952e-01],
[-6.7493e-01, -1.8199e-01, -4.1160e-01, ..., -1.9641e-01,
2.5410e-01, 2.4404e-01],
[-7.7757e-01, -1.8492e-01, -3.3941e-01, ..., -3.9303e-01,
-1.8790e-01, -2.2757e-01]],
...,
[[-1.9777e+00, -1.1056e+00, -2.3526e+00, ..., -3.1856e+00,
-2.6208e+00, -1.8210e+00],
[-1.7761e+00, -2.7070e-02, -1.3631e+00, ..., -2.5091e+00,
-2.5210e+00, -1.9376e+00],
[-4.6082e-01, 2.2424e+00, 9.9180e-01, ..., 1.2023e-01,
2.6782e-01, 5.2594e-01],
...,
[-2.0886e+00, -1.4496e-01, -2.3828e-01, ..., -1.3840e+00,
-4.0158e-01, 1.0537e-01],
[-1.2105e+00, 7.6913e-01, 8.9462e-01, ..., 9.5902e-03,
3.0532e-01, 4.8089e-01],
[-2.1152e+00, -1.4463e+00, -1.6497e+00, ..., -2.4619e+00,
-1.4655e+00, -9.1791e-01]],
[[-2.4455e+00, -2.8733e+00, -4.1669e+00, ..., -2.4011e+00,
-1.9006e+00, -1.7113e+00],
[-1.8192e+00, -1.6942e+00, -4.8230e+00, ..., -2.9368e+00,
-2.7833e+00, -2.7726e+00],
[-2.8708e+00, -2.5295e+00, -6.6291e+00, ..., -4.4809e+00,
-4.3756e+00, -3.4979e+00],
...,
[-1.4177e+00, -9.0535e-02, -3.3551e+00, ..., -1.4293e+00,
-1.0311e+00, -9.3333e-01],
[-1.7194e+00, -6.8148e-01, -3.0140e+00, ..., -1.7278e+00,
-1.1639e+00, -8.1666e-01],
[-2.3289e+00, -1.5047e+00, -3.0840e+00, ..., -2.9608e+00,
-2.1428e+00, -1.9887e+00]],
[[-1.8002e+00, -2.5292e+00, -2.4716e+00, ..., -2.7100e+00,
-2.2255e+00, -1.7114e+00],
[-1.7999e+00, -2.5042e+00, -2.1696e+00, ..., -2.4180e+00,
-1.9114e+00, -1.5535e+00],
[-2.2485e+00, -2.8152e+00, -2.5303e+00, ..., -2.7561e+00,
-2.1598e+00, -1.6416e+00],
...,
[-2.3612e+00, -2.9294e+00, -2.7216e+00, ..., -2.8485e+00,
-2.2790e+00, -1.7520e+00],
[-1.5255e+00, -1.9946e+00, -1.9290e+00, ..., -1.9342e+00,
-1.5961e+00, -1.2886e+00],
[-9.9447e-01, -1.2911e+00, -1.3301e+00, ..., -1.1829e+00,
-1.0087e+00, -7.6545e-01]]],
[[[ 2.9646e-01, 5.5481e-01, 6.8703e-01, ..., 4.9479e-01,
1.0274e+00, 6.1974e-01],
[-2.1871e+00, -1.5330e+00, -3.2409e+00, ..., -2.6474e+00,
-8.3922e-01, -1.6421e-01],
[-1.6400e-01, -1.0094e+00, -2.4466e-01, ..., -1.4806e-01,
5.8268e-01, 6.9481e-01],
...,
[-2.0605e-01, -1.1835e+00, -2.6741e+00, ..., -1.5232e+00,
-1.4353e+00, -8.4093e-01],
[ 1.4298e+00, 8.3315e-01, -9.6360e-01, ..., -2.9976e-01,
7.7416e-02, 3.8353e-01],
[ 8.5731e-01, 3.0312e-01, -6.9370e-01, ..., -1.9573e-01,
1.9285e-01, 3.6800e-01]],
[[-1.6831e+00, -1.1367e+00, -4.1699e+00, ..., -4.8782e+00,
-4.4397e+00, -3.4116e+00],
[ 6.2781e-01, 1.9960e+00, -2.2269e+00, ..., -3.5260e+00,
-3.6420e+00, -3.0070e+00],
[ 1.3930e+00, 1.7584e+00, -4.8985e+00, ..., -5.4918e+00,
-5.3341e+00, -4.9711e+00],
...,
[-1.2982e+00, 1.4869e-01, -6.7149e+00, ..., -7.5560e+00,
-6.8621e+00, -5.4192e+00],
[ 1.7894e+00, 2.8237e+00, -1.0455e+00, ..., -2.5800e+00,
-2.9339e+00, -2.3724e+00],
[ 1.2093e+00, 1.8101e+00, -2.2967e+00, ..., -3.2191e+00,
-3.6056e+00, -3.1256e+00]],
[[-1.2908e-01, -1.1740e+00, -1.3394e+00, ..., -4.9036e-01,
-1.7654e-01, -4.8401e-01],
[-6.1858e-01, -8.8231e-01, -7.9024e-01, ..., -2.5855e-01,
3.7170e-02, -7.5764e-02],
[-5.6390e-01, -3.6124e-01, -4.1046e-01, ..., -3.4939e-01,
8.7302e-02, 7.2339e-02],
...,
[-1.3701e+00, -1.7698e+00, -7.9265e-01, ..., -8.7776e-01,
-7.7449e-01, -1.0260e+00],
[-1.7196e+00, -1.1814e+00, -8.6761e-01, ..., -5.4476e-01,
-2.9143e-01, -3.6204e-01],
[-1.8629e+00, -1.2824e+00, -9.0925e-01, ..., -9.3376e-01,
-8.0077e-01, -7.8233e-01]],
...,
[[-3.0818e+00, -2.5340e+00, -2.8244e+00, ..., -3.4980e+00,
-2.6461e+00, -1.7711e+00],
[-2.3121e+00, -8.4915e-01, -1.6074e+00, ..., -2.6640e+00,
-2.3440e+00, -1.8162e+00],
[-1.0720e+00, 1.2036e+00, 6.4080e-01, ..., -4.6182e-01,
-2.4936e-02, 1.2950e-01],
...,
[-1.8107e+00, 1.3030e+00, 3.1163e-02, ..., -1.0530e+00,
-4.9434e-01, 1.2483e-02],
[-7.8454e-01, 1.9650e+00, 1.6802e+00, ..., 4.6542e-01,
4.3585e-01, 7.0734e-01],
[-2.2082e+00, -1.1294e+00, -9.6389e-01, ..., -1.9174e+00,
-1.0102e+00, -3.1372e-01]],
[[-1.6018e+00, -1.5794e+00, -2.9196e+00, ..., -1.2785e+00,
-8.6265e-01, -8.5519e-01],
[-5.8715e-01, 1.1187e-01, -3.2373e+00, ..., -1.3937e+00,
-1.2613e+00, -1.6099e+00],
[-2.1032e+00, -8.9264e-01, -5.3709e+00, ..., -3.2039e+00,
-2.7453e+00, -2.2416e+00],
...,
[-2.4412e+00, -1.5469e+00, -5.4450e+00, ..., -3.0655e+00,
-2.7369e+00, -2.1741e+00],
[-2.1957e+00, -1.4482e+00, -4.2330e+00, ..., -2.6682e+00,
-2.3227e+00, -1.6310e+00],
[-2.3815e+00, -1.7622e+00, -3.3117e+00, ..., -2.9519e+00,
-2.4339e+00, -2.1156e+00]],
[[-1.3600e+00, -1.9232e+00, -1.7139e+00, ..., -1.7602e+00,
-1.4608e+00, -1.1101e+00],
[-1.2813e+00, -1.7714e+00, -1.2299e+00, ..., -1.3762e+00,
-1.0799e+00, -8.6225e-01],
[-1.6028e+00, -1.9538e+00, -1.4772e+00, ..., -1.5984e+00,
-1.1932e+00, -8.8460e-01],
...,
[-3.1717e+00, -3.9679e+00, -3.7525e+00, ..., -3.9664e+00,
-3.1722e+00, -2.4426e+00],
[-2.1106e+00, -2.7289e+00, -2.6978e+00, ..., -2.7820e+00,
-2.2934e+00, -1.8628e+00],
[-1.5363e+00, -1.8649e+00, -1.9276e+00, ..., -1.8755e+00,
-1.5232e+00, -1.1683e+00]]],
[[[ 3.6028e-01, 8.3557e-01, 7.5765e-01, ..., 6.8717e-01,
9.5471e-01, 7.5984e-01],
[-1.2327e+00, -2.6277e-01, -1.3876e+00, ..., -1.0230e+00,
8.7826e-02, 5.8959e-01],
[ 6.3906e-01, 4.4763e-01, 5.1429e-01, ..., 5.4191e-01,
9.1026e-01, 8.6316e-01],
...,
[-1.4885e-01, -9.9447e-01, -2.3969e+00, ..., -1.6577e+00,
-1.5844e+00, -1.0214e+00],
[ 1.3141e+00, 7.9920e-01, -8.9480e-01, ..., -2.5033e-01,
7.7982e-02, 3.5561e-01],
[ 7.5590e-01, 2.4918e-01, -6.4040e-01, ..., -1.5006e-01,
2.0944e-01, 3.6807e-01]],
[[-2.3125e+00, -1.2003e+00, -4.2384e+00, ..., -5.1716e+00,
-4.4517e+00, -3.4922e+00],
[-9.8697e-02, 1.7563e+00, -2.5145e+00, ..., -4.1101e+00,
-4.0194e+00, -3.2704e+00],
[ 7.8000e-01, 1.6036e+00, -5.0762e+00, ..., -5.9590e+00,
-5.7604e+00, -5.2506e+00],
...,
[-1.5232e+00, -2.5125e-01, -6.8797e+00, ..., -7.5761e+00,
-6.8452e+00, -5.3713e+00],
[ 1.3506e+00, 2.2713e+00, -1.5747e+00, ..., -2.3287e+00,
-2.7191e+00, -2.1915e+00],
[ 8.9808e-01, 1.4651e+00, -2.5158e+00, ..., -3.1601e+00,
-3.5565e+00, -3.1181e+00]],
[[-6.6111e-01, -1.8417e+00, -1.7108e+00, ..., -8.7029e-01,
-6.2093e-01, -7.4171e-01],
[-1.3770e+00, -1.8545e+00, -1.2830e+00, ..., -7.0417e-01,
-3.7772e-01, -2.8532e-01],
[-1.0686e+00, -9.9850e-01, -4.6649e-01, ..., -3.5086e-01,
6.9229e-03, 3.9686e-02],
...,
[-1.1704e+00, -1.5388e+00, -6.9682e-01, ..., -9.0979e-01,
-7.7984e-01, -1.0410e+00],
[-1.4869e+00, -1.0026e+00, -7.6479e-01, ..., -5.7120e-01,
-2.8358e-01, -3.3714e-01],
[-1.6394e+00, -1.1599e+00, -8.6321e-01, ..., -9.3347e-01,
-8.0715e-01, -7.8489e-01]],
...,
[[-2.2492e+00, -1.2657e+00, -2.2987e+00, ..., -2.7523e+00,
-2.1913e+00, -1.4576e+00],
[-6.5210e-01, 1.1009e+00, -3.7204e-01, ..., -1.3244e+00,
-1.3732e+00, -1.1577e+00],
[-3.8420e-01, 2.1245e+00, 8.6821e-01, ..., -1.4049e-01,
2.1441e-01, 3.5048e-01],
...,
[-1.7221e+00, 1.0079e+00, -7.1678e-02, ..., -1.1943e+00,
-5.1569e-01, 5.6861e-02],
[-8.2655e-01, 1.6196e+00, 1.4296e+00, ..., 4.2251e-01,
4.4634e-01, 7.3058e-01],
[-2.1758e+00, -1.1726e+00, -9.7947e-01, ..., -2.1044e+00,
-1.1039e+00, -3.6313e-01]],
[[-1.3526e+00, -1.2483e+00, -2.7576e+00, ..., -1.1397e+00,
-6.3038e-01, -5.0740e-01],
[-6.3973e-01, -9.3971e-02, -3.5945e+00, ..., -1.7948e+00,
-1.5023e+00, -1.6575e+00],
[-2.0680e+00, -1.2865e+00, -5.9576e+00, ..., -3.6799e+00,
-3.1252e+00, -2.4849e+00],
...,
[-2.5480e+00, -1.8420e+00, -5.2815e+00, ..., -2.9851e+00,
-2.6095e+00, -2.0856e+00],
[-2.2498e+00, -1.6530e+00, -4.1531e+00, ..., -2.6271e+00,
-2.2107e+00, -1.5432e+00],
[-2.3250e+00, -1.8511e+00, -3.2706e+00, ..., -3.0223e+00,
-2.4275e+00, -2.1322e+00]],
[[-1.1665e+00, -1.7309e+00, -1.4991e+00, ..., -1.6280e+00,
-1.3485e+00, -1.0206e+00],
[-1.1804e+00, -1.6541e+00, -1.1639e+00, ..., -1.3835e+00,
-1.1212e+00, -8.9465e-01],
[-1.5406e+00, -1.8272e+00, -1.3404e+00, ..., -1.5919e+00,
-1.2054e+00, -8.8392e-01],
...,
[-3.0406e+00, -3.8153e+00, -3.6915e+00, ..., -4.0228e+00,
-3.2366e+00, -2.4934e+00],
[-2.0348e+00, -2.6356e+00, -2.6636e+00, ..., -2.8123e+00,
-2.3316e+00, -1.9021e+00],
[-1.4967e+00, -1.8218e+00, -1.9184e+00, ..., -1.8769e+00,
-1.5367e+00, -1.1756e+00]]]], device='cuda:0'),
tensor([[[[ 4.8767e-01, 9.4503e-01, 6.8745e-01, ..., 7.8914e-01,
1.0530e+00, 8.1321e-01],
[-1.0569e+00, -1.3102e-01, -1.5012e+00, ..., -1.1937e+00,
8.2076e-02, 6.2844e-01],
[ 7.5389e-01, 5.2649e-01, 1.9464e-01, ..., 5.1534e-01,
8.7425e-01, 8.8936e-01],
...,
[-8.2640e-01, -1.8812e+00, -1.9750e+00, ..., -1.7071e+00,
-1.2282e+00, -8.5394e-01],
[ 8.6666e-01, 1.0722e-01, -5.9891e-02, ..., -1.8448e-02,
7.1667e-01, 8.1151e-01],
[ 2.2200e-01, -4.7349e-01, -4.2318e-01, ..., -1.4628e-01,
4.5503e-01, 3.6871e-01]],
[[-2.6503e+00, -1.4378e+00, -4.8223e+00, ..., -5.8943e+00,
-5.0237e+00, -3.9043e+00],
[-6.6416e-01, 1.3841e+00, -3.3744e+00, ..., -4.7450e+00,
-4.5459e+00, -3.6822e+00],
[ 3.0962e-02, 1.1930e+00, -6.1013e+00, ..., -6.8826e+00,
-6.5515e+00, -5.9622e+00],
...,
[ 4.9144e-01, 7.4338e-01, -3.8601e+00, ..., -4.6795e+00,
-4.2301e+00, -3.5244e+00],
[ 2.7014e+00, 2.7699e+00, 3.4240e-01, ..., -9.2275e-01,
-1.2709e+00, -1.2991e+00],
[ 1.6231e+00, 1.4777e+00, -1.5532e+00, ..., -2.2807e+00,
-2.5387e+00, -2.4125e+00]],
[[-5.3525e-01, -1.5685e+00, -1.3963e+00, ..., -7.5582e-01,
-5.1106e-01, -7.3438e-01],
[-1.2964e+00, -1.7113e+00, -1.1116e+00, ..., -6.9408e-01,
-3.6440e-01, -3.2791e-01],
[-1.1697e+00, -1.0857e+00, -5.0631e-01, ..., -4.3526e-01,
-8.2158e-02, -8.5639e-02],
...,
[-2.9286e-01, -7.9390e-01, -5.0078e-01, ..., -4.8786e-01,
-1.9628e-01, -4.3865e-01],
[-8.7815e-01, -3.2277e-01, -4.8603e-01, ..., -1.9695e-01,
2.4288e-01, 2.3177e-01],
[-1.0174e+00, -3.1146e-01, -3.6779e-01, ..., -3.9541e-01,
-1.9067e-01, -2.2600e-01]],
...,
[[-2.0807e+00, -1.1680e+00, -2.4966e+00, ..., -3.2854e+00,
-2.7040e+00, -1.8412e+00],
[-5.3042e-01, 1.1601e+00, -6.7696e-01, ..., -1.6320e+00,
-1.8036e+00, -1.5651e+00],
[-2.9408e-01, 2.1699e+00, 5.1507e-01, ..., -3.9432e-01,
-3.7833e-02, 1.3433e-01],
...,
[-2.1903e+00, 1.6462e-01, -1.1329e-01, ..., -1.2933e+00,
-4.0115e-01, 7.8274e-02],
[-1.2256e+00, 1.0214e+00, 1.0615e+00, ..., 3.9792e-02,
2.9432e-01, 4.5717e-01],
[-2.1472e+00, -1.4266e+00, -1.6948e+00, ..., -2.3553e+00,
-1.4246e+00, -8.9767e-01]],
[[-1.4605e+00, -1.4098e+00, -2.8190e+00, ..., -1.1489e+00,
-5.9901e-01, -4.7278e-01],
[-8.7350e-01, -4.8393e-01, -3.7983e+00, ..., -1.9410e+00,
-1.6585e+00, -1.8420e+00],
[-2.4349e+00, -1.9266e+00, -6.3938e+00, ..., -4.2385e+00,
-3.6576e+00, -2.8800e+00],
...,
[-1.3245e+00, 1.8343e-01, -3.6024e+00, ..., -1.4820e+00,
-1.0871e+00, -9.6984e-01],
[-1.6517e+00, -4.7269e-01, -3.1943e+00, ..., -1.7424e+00,
-1.1973e+00, -8.4407e-01],
[-2.3244e+00, -1.4032e+00, -3.1588e+00, ..., -2.9248e+00,
-2.1256e+00, -1.9601e+00]],
[[-1.2416e+00, -1.8272e+00, -1.6444e+00, ..., -1.8578e+00,
-1.5436e+00, -1.1552e+00],
[-1.2983e+00, -1.8124e+00, -1.4064e+00, ..., -1.6382e+00,
-1.3278e+00, -1.0381e+00],
[-1.7129e+00, -2.0714e+00, -1.6920e+00, ..., -1.9307e+00,
-1.4800e+00, -1.0839e+00],
...,
[-2.4965e+00, -3.0774e+00, -2.7759e+00, ..., -2.8086e+00,
-2.2399e+00, -1.7226e+00],
[-1.6187e+00, -2.0913e+00, -1.9670e+00, ..., -1.9101e+00,
-1.5706e+00, -1.2670e+00],
[-1.0586e+00, -1.3427e+00, -1.3438e+00, ..., -1.1732e+00,
-9.9525e-01, -7.5753e-01]]],
[[[ 8.0368e-02, 2.3355e-01, -1.3850e+00, ..., -5.9856e-01,
-1.9480e-01, -5.6836e-02],
[-2.0598e+00, -1.6100e+00, -4.4903e+00, ..., -3.7496e+00,
-1.5427e+00, -5.6856e-01],
[ 6.0209e-01, -6.2829e-02, -7.9790e-01, ..., -1.9956e-01,
-2.6486e-01, 2.3511e-01],
...,
[-3.1399e-01, -1.0979e+00, -1.4097e+00, ..., -1.3604e+00,
-9.0988e-01, -6.7773e-01],
[ 9.5573e-01, 5.6487e-01, 3.4449e-01, ..., 3.7616e-01,
7.0628e-01, 8.0447e-01],
[ 6.6194e-01, 4.1455e-01, 3.2222e-01, ..., 4.9810e-01,
7.6593e-01, 7.0765e-01]],
[[-3.2816e+00, -1.8018e+00, -6.2534e+00, ..., -7.1586e+00,
-6.2522e+00, -4.6171e+00],
[-8.0566e-01, 1.5290e+00, -4.3593e+00, ..., -5.4671e+00,
-5.5266e+00, -4.2605e+00],
[-3.5579e-01, 1.0920e+00, -7.4023e+00, ..., -7.9274e+00,
-7.8108e+00, -6.6793e+00],
...,
[-4.5366e-01, 2.8032e-01, -4.4326e+00, ..., -5.3391e+00,
-4.6653e+00, -3.8972e+00],
[ 1.9298e+00, 2.4021e+00, -1.7200e-01, ..., -1.1080e+00,
-1.2855e+00, -1.4124e+00],
[ 1.0943e+00, 1.2857e+00, -1.7528e+00, ..., -2.4777e+00,
-2.6085e+00, -2.5693e+00]],
[[-1.3302e+00, -2.3473e+00, -2.3249e+00, ..., -1.7539e+00,
-1.2384e+00, -1.3829e+00],
[-1.6786e+00, -1.9928e+00, -1.5419e+00, ..., -1.2655e+00,
-7.4112e-01, -8.1882e-01],
[-2.1589e+00, -1.9324e+00, -1.0063e+00, ..., -1.1023e+00,
-4.5996e-01, -1.8146e-01],
...,
[-6.0764e-01, -1.0713e+00, -5.7017e-01, ..., -5.8884e-01,
-4.5858e-01, -6.2135e-01],
[-1.0084e+00, -6.8217e-01, -5.7247e-01, ..., -2.6958e-01,
-7.3408e-02, -5.6454e-02],
[-1.1737e+00, -6.6381e-01, -3.9986e-01, ..., -3.2955e-01,
-2.7520e-01, -2.4608e-01]],
...,
[[-2.0388e+00, -1.1009e+00, -2.3484e+00, ..., -3.2064e+00,
-2.6384e+00, -1.8333e+00],
[-1.7690e+00, 3.6734e-02, -1.3498e+00, ..., -2.4775e+00,
-2.5089e+00, -1.9526e+00],
[-3.9581e-01, 2.3604e+00, 1.0463e+00, ..., 1.0936e-01,
2.8305e-01, 5.3654e-01],
...,
[-1.5608e+00, 9.1630e-01, -1.5157e-01, ..., -1.1475e+00,
-4.5442e-01, 3.1568e-02],
[-5.6942e-01, 1.3809e+00, 7.5053e-01, ..., -3.5250e-03,
1.8570e-01, 3.0710e-01],
[-1.5971e+00, -1.0327e+00, -1.4421e+00, ..., -2.2676e+00,
-1.3101e+00, -6.8995e-01]],
[[-2.4191e+00, -2.8327e+00, -4.2312e+00, ..., -2.3976e+00,
-1.8814e+00, -1.6905e+00],
[-1.7790e+00, -1.6455e+00, -4.8920e+00, ..., -2.8748e+00,
-2.7415e+00, -2.7572e+00],
[-2.8338e+00, -2.4735e+00, -6.6936e+00, ..., -4.4210e+00,
-4.3509e+00, -3.4908e+00],
...,
[-1.7020e+00, -5.7062e-01, -4.0669e+00, ..., -2.1114e+00,
-1.5864e+00, -1.2378e+00],
[-1.6667e+00, -7.5146e-01, -3.3624e+00, ..., -2.0644e+00,
-1.3727e+00, -8.4268e-01],
[-2.0166e+00, -1.2691e+00, -2.8068e+00, ..., -2.6768e+00,
-1.8047e+00, -1.5472e+00]],
[[-1.8028e+00, -2.5248e+00, -2.4470e+00, ..., -2.7045e+00,
-2.2205e+00, -1.6994e+00],
[-1.7997e+00, -2.4996e+00, -2.1411e+00, ..., -2.4134e+00,
-1.9057e+00, -1.5400e+00],
[-2.2532e+00, -2.8080e+00, -2.5029e+00, ..., -2.7475e+00,
-2.1444e+00, -1.6248e+00],
...,
[-2.2744e+00, -2.8481e+00, -2.6614e+00, ..., -2.8645e+00,
-2.2867e+00, -1.7237e+00],
[-1.5426e+00, -1.9802e+00, -1.9477e+00, ..., -2.0463e+00,
-1.6803e+00, -1.3338e+00],
[-1.0496e+00, -1.2847e+00, -1.3612e+00, ..., -1.2837e+00,
-1.0884e+00, -8.0235e-01]]],
[[[ 3.2656e-02, 3.6070e-01, 4.1413e-01, ..., -3.4114e-01,
-7.7711e-02, 1.5515e-02],
[-3.3365e+00, -2.7534e+00, -3.7885e+00, ..., -3.2425e+00,
-1.2995e+00, -4.4532e-01],
[-3.3822e+00, -3.7203e+00, -3.6913e+00, ..., -1.6125e-01,
-3.4484e-01, 1.8489e-01],
...,
[-1.2214e-01, -8.0048e-01, -1.3988e+00, ..., -1.9287e+00,
-1.9298e+00, -1.2920e+00],
[ 1.0827e+00, 7.5514e-01, 2.9627e-01, ..., 2.5504e-02,
2.4445e-01, 4.5416e-01],
[ 7.0340e-01, 4.5541e-01, 1.5331e-01, ..., 2.4023e-01,
4.5663e-01, 4.9591e-01]],
[[-2.2544e+00, -8.2807e-01, -3.9870e+00, ..., -6.8477e+00,
-6.0543e+00, -4.5202e+00],
[ 9.6071e-03, 1.8917e+00, -2.5000e+00, ..., -5.4412e+00,
-5.5303e+00, -4.2866e+00],
[-2.1456e-01, 1.9795e-01, -6.6649e+00, ..., -7.8317e+00,
-7.7672e+00, -6.5896e+00],
...,
[-4.7185e-01, 5.9973e-01, -4.1462e+00, ..., -7.0752e+00,
-6.6173e+00, -5.1752e+00],
[ 1.9693e+00, 2.6300e+00, 9.4575e-02, ..., -1.8913e+00,
-2.5016e+00, -2.0449e+00],
[ 1.1354e+00, 1.4216e+00, -1.6118e+00, ..., -2.8938e+00,
-3.4390e+00, -3.0355e+00]],
[[-6.0675e-01, -1.6414e+00, -1.4780e+00, ..., -1.4002e+00,
-1.0383e+00, -1.3322e+00],
[-1.3834e+00, -1.9147e+00, -1.3531e+00, ..., -9.8746e-01,
-6.0940e-01, -8.9906e-01],
[-9.8889e-01, -9.7991e-01, -3.9590e-02, ..., -1.0860e+00,
-5.7496e-01, -4.6452e-01],
...,
[-9.0108e-01, -1.3056e+00, -5.0519e-01, ..., -1.0355e+00,
-8.7716e-01, -1.2513e+00],
[-1.2505e+00, -8.5331e-01, -5.2860e-01, ..., -5.4485e-01,
-2.5157e-01, -4.1865e-01],
[-1.3300e+00, -7.4906e-01, -3.4340e-01, ..., -4.7848e-01,
-4.3706e-01, -6.0174e-01]],
...,
[[-2.5905e+00, -1.4647e+00, -2.8588e+00, ..., -3.0796e+00,
-2.4973e+00, -1.8031e+00],
[-1.1860e+00, 1.0607e+00, -1.0513e+00, ..., -2.4273e+00,
-2.4819e+00, -1.9927e+00],
[-2.0546e+00, 2.6957e+00, 2.0117e-01, ..., -1.4087e-01,
3.2270e-02, 3.0498e-01],
...,
[-1.4404e+00, 1.3085e+00, -1.1238e-01, ..., -9.2675e-01,
-2.4142e-01, 2.9499e-01],
[-4.9590e-01, 1.6020e+00, 8.2803e-01, ..., 4.1031e-01,
4.9138e-01, 8.2177e-01],
[-1.5624e+00, -9.6416e-01, -1.4236e+00, ..., -2.2389e+00,
-1.1801e+00, -3.6082e-01]],
[[-1.3312e+00, -1.1978e+00, -2.6802e+00, ..., -2.2344e+00,
-1.8962e+00, -1.7960e+00],
[-2.1747e-01, 5.4239e-01, -3.1165e+00, ..., -2.8586e+00,
-2.8495e+00, -2.8790e+00],
[-1.0358e+00, 1.2949e-02, -5.0407e+00, ..., -4.4643e+00,
-4.4980e+00, -3.6971e+00],
...,
[-1.6274e+00, -5.5219e-01, -4.1341e+00, ..., -2.7979e+00,
-2.7068e+00, -2.3103e+00],
[-1.6212e+00, -7.4458e-01, -3.4128e+00, ..., -2.4236e+00,
-2.2314e+00, -1.6588e+00],
[-2.0395e+00, -1.3314e+00, -2.9098e+00, ..., -2.8598e+00,
-2.4198e+00, -2.1927e+00]],
[[-1.3204e+00, -1.9852e+00, -1.6964e+00, ..., -2.5642e+00,
-2.1446e+00, -1.6685e+00],
[-1.5997e+00, -2.3165e+00, -1.5546e+00, ..., -2.3941e+00,
-1.9315e+00, -1.5848e+00],
[-2.3978e+00, -2.9048e+00, -2.1535e+00, ..., -2.8110e+00,
-2.2474e+00, -1.7494e+00],
...,
[-2.3036e+00, -2.8561e+00, -2.6463e+00, ..., -3.9981e+00,
-3.2383e+00, -2.5346e+00],
[-1.5575e+00, -1.9723e+00, -1.9131e+00, ..., -2.7514e+00,
-2.3079e+00, -1.9060e+00],
[-1.0510e+00, -1.2615e+00, -1.3244e+00, ..., -1.8180e+00,
-1.5169e+00, -1.1738e+00]]],
[[[ 4.6851e-01, 9.5060e-01, 8.5835e-01, ..., -3.0486e-01,
-5.9292e-02, 5.5138e-03],
[-1.2164e+00, -3.0756e-01, -1.4746e+00, ..., -3.3326e+00,
-1.3697e+00, -4.9887e-01],
[ 5.9128e-01, 4.2455e-01, 3.5363e-01, ..., -1.2211e-01,
-3.7710e-01, 1.3628e-01],
...,
[-1.6010e-01, -7.8305e-01, -1.4243e+00, ..., -1.7273e+00,
-1.7249e+00, -1.1207e+00],
[ 1.0789e+00, 8.2482e-01, 3.7679e-01, ..., 3.9182e-02,
2.5565e-01, 4.4082e-01],
[ 7.0787e-01, 5.1864e-01, 2.3883e-01, ..., 2.0926e-01,
4.2952e-01, 4.8164e-01]],
[[-2.2978e+00, -1.0854e+00, -4.1100e+00, ..., -7.0879e+00,
-6.2272e+00, -4.6334e+00],
[-2.9937e-01, 1.7029e+00, -2.5260e+00, ..., -5.5232e+00,
-5.6090e+00, -4.3309e+00],
[ 4.0287e-01, 1.4839e+00, -5.0498e+00, ..., -8.0237e+00,
-7.9266e+00, -6.7251e+00],
...,
[-4.3167e-01, 6.8142e-01, -4.0948e+00, ..., -7.1268e+00,
-6.7207e+00, -5.2510e+00],
[ 2.0858e+00, 2.7730e+00, 2.4179e-01, ..., -2.2574e+00,
-2.8299e+00, -2.2649e+00],
[ 1.2124e+00, 1.5124e+00, -1.5457e+00, ..., -3.0042e+00,
-3.5058e+00, -3.0419e+00]],
[[-5.5893e-01, -1.5574e+00, -1.3819e+00, ..., -1.5091e+00,
-1.1085e+00, -1.3900e+00],
[-1.2623e+00, -1.6246e+00, -9.9668e-01, ..., -1.0731e+00,
-6.4649e-01, -9.0565e-01],
[-1.1221e+00, -1.0034e+00, -4.0731e-01, ..., -1.1063e+00,
-5.7861e-01, -4.5459e-01],
...,
[-9.2813e-01, -1.3394e+00, -5.3373e-01, ..., -9.8916e-01,
-8.5412e-01, -1.2057e+00],
[-1.2939e+00, -8.8401e-01, -5.5042e-01, ..., -5.6133e-01,
-2.7627e-01, -4.4035e-01],
[-1.3897e+00, -7.7997e-01, -3.5320e-01, ..., -4.9052e-01,
-4.4849e-01, -6.1248e-01]],
...,
[[-2.1683e+00, -1.2048e+00, -2.5143e+00, ..., -3.3491e+00,
-2.6662e+00, -1.9080e+00],
[-7.1032e-01, 9.6115e-01, -9.5093e-01, ..., -2.5415e+00,
-2.5667e+00, -2.0829e+00],
[-2.3575e-01, 2.2332e+00, 5.0251e-01, ..., -1.0717e-01,
1.0056e-01, 3.5612e-01],
...,
[-1.4828e+00, 1.2600e+00, -1.8504e-01, ..., -8.7461e-01,
-2.5950e-01, 2.3467e-01],
[-4.9002e-01, 1.5759e+00, 7.7860e-01, ..., 3.8646e-01,
4.3426e-01, 7.5197e-01],
[-1.5557e+00, -1.0050e+00, -1.5400e+00, ..., -2.0147e+00,
-1.0639e+00, -3.0651e-01]],
[[-1.2983e+00, -1.2075e+00, -2.6207e+00, ..., -2.1689e+00,
-1.8146e+00, -1.7299e+00],
[-6.2790e-01, -1.3571e-01, -3.3640e+00, ..., -2.8431e+00,
-2.8167e+00, -2.8931e+00],
[-2.0834e+00, -1.4127e+00, -5.7340e+00, ..., -4.5764e+00,
-4.5665e+00, -3.7449e+00],
...,
[-1.6284e+00, -5.3339e-01, -4.1831e+00, ..., -2.9539e+00,
-2.8555e+00, -2.3939e+00],
[-1.6329e+00, -7.4262e-01, -3.4833e+00, ..., -2.5098e+00,
-2.3363e+00, -1.7465e+00],
[-2.0653e+00, -1.3622e+00, -2.9765e+00, ..., -2.8011e+00,
-2.4112e+00, -2.1767e+00]],
[[-1.1424e+00, -1.6990e+00, -1.5105e+00, ..., -2.6438e+00,
-2.2079e+00, -1.6996e+00],
[-1.1495e+00, -1.6179e+00, -1.1982e+00, ..., -2.4236e+00,
-1.9544e+00, -1.5969e+00],
[-1.4996e+00, -1.8018e+00, -1.4068e+00, ..., -2.8207e+00,
-2.2504e+00, -1.7374e+00],
...,
[-2.3135e+00, -2.8722e+00, -2.6526e+00, ..., -3.9148e+00,
-3.1792e+00, -2.4928e+00],
[-1.5606e+00, -1.9750e+00, -1.9031e+00, ..., -2.7187e+00,
-2.2760e+00, -1.8766e+00],
[-1.0520e+00, -1.2563e+00, -1.3043e+00, ..., -1.8223e+00,
-1.5070e+00, -1.1667e+00]]],
[[[-7.6347e-02, 2.3304e-02, -2.0325e+00, ..., -1.0043e+00,
-5.9583e-01, -2.7265e-01],
[-2.3856e+00, -1.9978e+00, -5.4695e+00, ..., -4.4772e+00,
-2.0237e+00, -8.5318e-01],
[ 4.2169e-01, -3.5860e-01, -1.3781e+00, ..., -4.8543e-01,
-6.4359e-01, 8.4328e-02],
...,
[-6.2427e-01, -1.4049e+00, -1.5464e+00, ..., -1.5358e+00,
-1.1326e+00, -8.1194e-01],
[ 8.7152e-01, 3.2511e-01, 1.5785e-01, ..., 2.1791e-01,
7.9645e-01, 8.5193e-01],
[ 3.2005e-01, -1.9157e-01, -1.6067e-01, ..., 1.3421e-01,
6.0668e-01, 4.8128e-01]],
[[-3.5174e+00, -1.9702e+00, -6.7089e+00, ..., -7.5263e+00,
-6.6630e+00, -4.9051e+00],
[-9.2944e-01, 1.4771e+00, -4.7582e+00, ..., -5.7408e+00,
-5.9302e+00, -4.5630e+00],
[-5.6479e-01, 9.7851e-01, -7.9620e+00, ..., -8.3726e+00,
-8.3580e+00, -7.0347e+00],
...,
[ 9.8669e-03, 3.3174e-01, -4.1191e+00, ..., -4.8687e+00,
-4.3614e+00, -3.6254e+00],
[ 2.1421e+00, 2.2827e+00, -1.6974e-01, ..., -9.2834e-01,
-1.2491e+00, -1.3033e+00],
[ 1.2383e+00, 1.1795e+00, -1.7722e+00, ..., -2.3247e+00,
-2.5720e+00, -2.4772e+00]],
[[-1.6400e+00, -2.5836e+00, -2.5222e+00, ..., -1.9782e+00,
-1.4359e+00, -1.5611e+00],
[-1.9287e+00, -2.1244e+00, -1.6217e+00, ..., -1.4462e+00,
-9.0845e-01, -1.0162e+00],
[-2.5777e+00, -2.2284e+00, -1.1171e+00, ..., -1.3158e+00,
-6.2205e-01, -2.7662e-01],
...,
[-3.2153e-01, -7.9442e-01, -4.7205e-01, ..., -5.0424e-01,
-2.5193e-01, -4.7084e-01],
[-7.9666e-01, -3.3818e-01, -4.2628e-01, ..., -1.8215e-01,
2.1313e-01, 2.1489e-01],
[-8.9718e-01, -2.9840e-01, -3.0593e-01, ..., -3.2078e-01,
-1.6154e-01, -1.8750e-01]],
...,
[[-2.0592e+00, -1.1280e+00, -2.3182e+00, ..., -3.1546e+00,
-2.6356e+00, -1.8692e+00],
[-2.1672e+00, -4.0814e-01, -1.6118e+00, ..., -2.7582e+00,
-2.7758e+00, -2.1300e+00],
[-5.4819e-01, 2.1680e+00, 1.0834e+00, ..., 1.6771e-01,
2.4872e-01, 5.1547e-01],
...,
[-1.7962e+00, 3.2228e-01, -1.0132e-01, ..., -1.2353e+00,
-3.2396e-01, 1.9579e-01],
[-9.5484e-01, 1.0325e+00, 9.5309e-01, ..., 5.0943e-02,
3.2667e-01, 4.8271e-01],
[-1.9074e+00, -1.2691e+00, -1.5294e+00, ..., -2.3623e+00,
-1.3744e+00, -8.3630e-01]],
[[-2.6820e+00, -3.2246e+00, -4.4871e+00, ..., -2.7007e+00,
-2.2272e+00, -2.0273e+00],
[-2.0565e+00, -2.0497e+00, -5.1185e+00, ..., -3.1417e+00,
-3.0926e+00, -3.0444e+00],
[-3.0638e+00, -2.7973e+00, -6.7663e+00, ..., -4.5877e+00,
-4.6783e+00, -3.7448e+00],
...,
[-1.4208e+00, -2.2444e-01, -3.6278e+00, ..., -1.6505e+00,
-1.2201e+00, -1.0441e+00],
[-1.6294e+00, -6.8153e-01, -3.1640e+00, ..., -1.8130e+00,
-1.2355e+00, -8.3645e-01],
[-2.1948e+00, -1.4398e+00, -3.0487e+00, ..., -2.8933e+00,
-2.0730e+00, -1.8885e+00]],
[[-2.0094e+00, -2.7970e+00, -2.7409e+00, ..., -2.9931e+00,
-2.4486e+00, -1.8756e+00],
[-2.0113e+00, -2.8008e+00, -2.4629e+00, ..., -2.7228e+00,
-2.1376e+00, -1.7286e+00],
[-2.5023e+00, -3.1621e+00, -2.8963e+00, ..., -3.1162e+00,
-2.4322e+00, -1.8540e+00],
...,
[-2.3428e+00, -2.8970e+00, -2.6770e+00, ..., -2.8415e+00,
-2.2764e+00, -1.7345e+00],
[-1.5426e+00, -1.9855e+00, -1.9185e+00, ..., -1.9497e+00,
-1.6107e+00, -1.2877e+00],
[-1.0202e+00, -1.2878e+00, -1.3248e+00, ..., -1.1936e+00,
-1.0250e+00, -7.6755e-01]]]], device='cuda:0'),
tensor([[[[ 2.9474e-01, 5.5545e-01, 6.9008e-01, ..., 4.6638e-01,
1.0031e+00, 6.0410e-01],
[-2.1973e+00, -1.5270e+00, -3.2025e+00, ..., -2.6728e+00,
-8.6654e-01, -1.9336e-01],
[-1.9190e-01, -1.0042e+00, -2.2013e-01, ..., -1.3492e-01,
5.7878e-01, 6.7258e-01],
...,
[ 8.0109e-02, -6.9804e-01, -1.5174e+00, ..., -9.8992e-01,
-7.1879e-01, -5.2186e-01],
[ 1.4935e+00, 1.2450e+00, 6.2898e-01, ..., 8.0914e-01,
9.5810e-01, 9.3854e-01],
[ 1.1165e+00, 9.9756e-01, 6.8746e-01, ..., 9.4562e-01,
1.0317e+00, 8.5570e-01]],
[[-1.6637e+00, -1.1509e+00, -4.1739e+00, ..., -4.8724e+00,
-4.4393e+00, -3.4047e+00],
[ 6.0975e-01, 1.9451e+00, -2.2467e+00, ..., -3.4950e+00,
-3.6239e+00, -2.9861e+00],
[ 1.3527e+00, 1.7033e+00, -4.8938e+00, ..., -5.4565e+00,
-5.3049e+00, -4.9538e+00],
...,
[-6.2510e-01, 5.7761e-01, -4.6861e+00, ..., -5.8726e+00,
-5.2319e+00, -4.2414e+00],
[ 2.1564e+00, 2.8817e+00, 1.6897e-01, ..., -1.6160e+00,
-1.7923e+00, -1.7481e+00],
[ 1.2641e+00, 1.5565e+00, -1.6747e+00, ..., -2.7574e+00,
-2.8850e+00, -2.7319e+00]],
[[-1.0434e-01, -1.1417e+00, -1.3141e+00, ..., -4.8692e-01,
-1.6423e-01, -4.7738e-01],
[-5.8232e-01, -8.4265e-01, -7.6430e-01, ..., -2.6691e-01,
3.7747e-02, -7.0541e-02],
[-5.3889e-01, -3.3324e-01, -3.8742e-01, ..., -3.3895e-01,
9.0380e-02, 7.7075e-02],
...,
[-1.0947e+00, -1.5639e+00, -6.3320e-01, ..., -6.0786e-01,
-4.4699e-01, -6.0306e-01],
[-1.3282e+00, -8.8073e-01, -4.5511e-01, ..., -9.5212e-02,
1.0058e-01, 8.4965e-02],
[-1.3914e+00, -7.0183e-01, -1.6700e-01, ..., -5.4563e-02,
-2.6497e-02, -3.9658e-02]],
...,
[[-3.0528e+00, -2.5570e+00, -2.8559e+00, ..., -3.5007e+00,
-2.6414e+00, -1.7637e+00],
[-2.3050e+00, -8.9051e-01, -1.6455e+00, ..., -2.6753e+00,
-2.3504e+00, -1.8331e+00],
[-1.0777e+00, 1.1289e+00, 5.4949e-01, ..., -4.9828e-01,
-3.2162e-02, 1.4313e-01],
...,
[-1.5601e+00, 1.6118e+00, 2.9427e-01, ..., -8.3499e-01,
-1.4218e-01, 3.8052e-01],
[-5.7276e-01, 1.7971e+00, 1.1087e+00, ..., 1.0767e-01,
2.9337e-01, 4.5702e-01],
[-1.6652e+00, -9.8191e-01, -1.3775e+00, ..., -2.0606e+00,
-1.1322e+00, -5.3962e-01]],
[[-1.5926e+00, -1.5891e+00, -2.9068e+00, ..., -1.2752e+00,
-8.3417e-01, -8.3692e-01],
[-5.8623e-01, 9.1325e-02, -3.2314e+00, ..., -1.3829e+00,
-1.2310e+00, -1.5925e+00],
[-2.0787e+00, -8.9488e-01, -5.3591e+00, ..., -3.2200e+00,
-2.7302e+00, -2.2251e+00],
...,
[-1.7259e+00, -6.5702e-01, -4.6858e+00, ..., -2.5222e+00,
-2.0710e+00, -1.6222e+00],
[-1.6382e+00, -7.6675e-01, -3.7766e+00, ..., -2.3152e+00,
-1.6990e+00, -1.1182e+00],
[-2.0330e+00, -1.3106e+00, -3.0247e+00, ..., -2.7487e+00,
-1.9518e+00, -1.6747e+00]],
[[-1.3512e+00, -1.9137e+00, -1.7175e+00, ..., -1.7688e+00,
-1.4675e+00, -1.1162e+00],
[-1.2757e+00, -1.7626e+00, -1.2374e+00, ..., -1.3731e+00,
-1.0864e+00, -8.6687e-01],
[-1.5962e+00, -1.9435e+00, -1.4838e+00, ..., -1.5951e+00,
-1.1987e+00, -8.8763e-01],
...,
[-2.4893e+00, -3.1055e+00, -2.8235e+00, ..., -2.9721e+00,
-2.3763e+00, -1.8140e+00],
[-1.6784e+00, -2.1299e+00, -2.0111e+00, ..., -2.0913e+00,
-1.7141e+00, -1.3676e+00],
[-1.1265e+00, -1.3472e+00, -1.3816e+00, ..., -1.2982e+00,
-1.1024e+00, -8.1236e-01]]],
[[[ 2.9699e-01, 5.5826e-01, 6.8437e-01, ..., 4.7990e-01,
1.0181e+00, 6.0785e-01],
[-2.1340e+00, -1.5102e+00, -3.1804e+00, ..., -2.6630e+00,
-8.5425e-01, -1.7925e-01],
[-1.8519e-01, -1.0010e+00, -2.3151e-01, ..., -1.5094e-01,
5.4382e-01, 6.5416e-01],
...,
[ 9.7600e-02, -5.7606e-01, -1.4074e+00, ..., -7.6325e+00,
-9.4622e-01, -7.4769e-01],
[ 1.4313e+00, 1.1946e+00, 5.9933e-01, ..., -5.0038e+00,
9.1860e-01, 9.3338e-01],
[ 1.0815e+00, 9.6642e-01, 6.9022e-01, ..., -4.5173e+00,
1.0301e+00, 8.7317e-01]],
[[-1.6821e+00, -1.1676e+00, -4.1718e+00, ..., -4.8874e+00,
-4.4478e+00, -3.4044e+00],
[ 5.6815e-01, 1.9033e+00, -2.2358e+00, ..., -3.5198e+00,
-3.6322e+00, -2.9805e+00],
[ 1.3029e+00, 1.7011e+00, -4.8416e+00, ..., -5.4764e+00,
-5.2952e+00, -4.9496e+00],
...,
[-8.7497e-01, 1.5697e-01, -4.9353e+00, ..., -7.3108e+00,
-5.2285e+00, -4.1991e+00],
[ 1.7556e+00, 2.3995e+00, -2.9311e-01, ..., -2.1172e+00,
-1.5266e+00, -1.5098e+00],
[ 1.0049e+00, 1.2749e+00, -1.8460e+00, ..., -2.9728e+00,
-2.8098e+00, -2.7031e+00]],
[[-9.6133e-02, -1.1115e+00, -1.2966e+00, ..., -4.8746e-01,
-1.5573e-01, -4.8353e-01],
[-5.6731e-01, -8.3690e-01, -7.8161e-01, ..., -2.8143e-01,
3.7585e-02, -7.8753e-02],
[-5.3975e-01, -3.6018e-01, -4.3233e-01, ..., -3.5283e-01,
7.9615e-02, 7.5730e-02],
...,
[-9.7826e-01, -1.4237e+00, -6.2803e-01, ..., -6.5416e-01,
-4.9409e-01, -6.7257e-01],
[-1.1777e+00, -7.7089e-01, -4.1917e-01, ..., 3.9826e-01,
1.1643e-01, 1.0120e-01],
[-1.1965e+00, -5.8622e-01, -1.1341e-01, ..., 5.1205e-01,
-1.3623e-02, -4.7337e-02]],
...,
[[-2.9900e+00, -2.4733e+00, -2.8002e+00, ..., -3.4678e+00,
-2.6505e+00, -1.7632e+00],
[-2.2907e+00, -8.4840e-01, -1.5836e+00, ..., -2.6218e+00,
-2.3608e+00, -1.8514e+00],
[-1.0978e+00, 1.1580e+00, 6.2784e-01, ..., -4.4343e-01,
-9.7639e-03, 1.3420e-01],
...,
[-1.4376e+00, 1.4255e+00, 2.6903e-01, ..., -1.1960e+00,
-1.4757e-01, 4.3435e-01],
[-5.7139e-01, 1.5855e+00, 1.0166e+00, ..., 5.3945e-01,
3.9023e-01, 5.5298e-01],
[-1.6509e+00, -1.0197e+00, -1.3876e+00, ..., -1.3746e+00,
-1.2575e+00, -5.8466e-01]],
[[-1.6022e+00, -1.5941e+00, -2.9067e+00, ..., -1.2559e+00,
-8.2186e-01, -8.3665e-01],
[-6.1735e-01, 7.4427e-02, -3.1952e+00, ..., -1.3703e+00,
-1.2080e+00, -1.5691e+00],
[-2.0965e+00, -9.0509e-01, -5.3291e+00, ..., -3.1842e+00,
-2.7025e+00, -2.2031e+00],
...,
[-1.8331e+00, -9.2888e-01, -4.5169e+00, ..., -2.4993e+00,
-1.9370e+00, -1.5155e+00],
[-1.7140e+00, -9.5775e-01, -3.6461e+00, ..., -2.6497e+00,
-1.6106e+00, -1.0083e+00],
[-2.0414e+00, -1.4138e+00, -2.9795e+00, ..., -2.5048e+00,
-1.9689e+00, -1.6842e+00]],
[[-1.3491e+00, -1.9094e+00, -1.7149e+00, ..., -1.7744e+00,
-1.4636e+00, -1.1185e+00],
[-1.2753e+00, -1.7686e+00, -1.2424e+00, ..., -1.3765e+00,
-1.0824e+00, -8.6829e-01],
[-1.6077e+00, -1.9612e+00, -1.4895e+00, ..., -1.6018e+00,
-1.1974e+00, -8.8662e-01],
...,
[-2.3823e+00, -2.9854e+00, -2.7862e+00, ..., -3.4371e+00,
-2.4668e+00, -1.8854e+00],
[-1.6039e+00, -2.0391e+00, -1.9779e+00, ..., -2.4707e+00,
-1.7584e+00, -1.4129e+00],
[-1.0721e+00, -1.2921e+00, -1.3486e+00, ..., -1.4398e+00,
-1.1213e+00, -8.2882e-01]]],
[[[ 2.8773e-01, 5.4837e-01, 6.7719e-01, ..., 4.7562e-01,
1.0271e+00, 6.0708e-01],
[-2.1279e+00, -1.4762e+00, -3.1180e+00, ..., -2.8394e+00,
-9.4667e-01, -2.1167e-01],
[-1.8621e-01, -9.4748e-01, -2.0293e-01, ..., -1.9900e-01,
5.2184e-01, 6.3371e-01],
...,
[ 9.3659e-02, -6.8569e-01, -1.6119e+00, ..., -1.0853e+00,
-7.8763e-01, -5.7301e-01],
[ 1.5481e+00, 1.2517e+00, 5.6230e-01, ..., 7.1383e-01,
8.9719e-01, 9.1985e-01],
[ 1.1709e+00, 1.0112e+00, 6.6019e-01, ..., 9.0431e-01,
9.9119e-01, 8.3723e-01]],
[[-1.6782e+00, -1.2335e+00, -4.1850e+00, ..., -5.0574e+00,
-4.5888e+00, -3.4829e+00],
[ 4.9124e-01, 1.7092e+00, -2.3596e+00, ..., -3.5025e+00,
-3.6293e+00, -2.9807e+00],
[ 1.1781e+00, 1.4152e+00, -4.9617e+00, ..., -5.5196e+00,
-5.3606e+00, -5.0631e+00],
...,
[-6.5645e-01, 5.4196e-01, -4.7438e+00, ..., -5.8685e+00,
-5.2485e+00, -4.2521e+00],
[ 2.1661e+00, 2.8992e+00, 1.8231e-01, ..., -1.5677e+00,
-1.7547e+00, -1.6958e+00],
[ 1.2908e+00, 1.5978e+00, -1.6553e+00, ..., -2.7289e+00,
-2.8494e+00, -2.7160e+00]],
[[-1.0223e-01, -1.0816e+00, -1.2555e+00, ..., -5.6833e-01,
-2.0660e-01, -5.3844e-01],
[-5.4550e-01, -7.8802e-01, -7.4632e-01, ..., -3.5228e-01,
-2.7119e-03, -9.7054e-02],
[-4.8614e-01, -2.8955e-01, -4.0134e-01, ..., -3.8036e-01,
8.8111e-02, 8.3210e-02],
...,
[-1.1182e+00, -1.6070e+00, -6.7138e-01, ..., -6.3103e-01,
-4.8504e-01, -6.3101e-01],
[-1.3633e+00, -9.0666e-01, -4.8172e-01, ..., -1.0808e-01,
8.9209e-02, 7.4747e-02],
[-1.3995e+00, -6.9980e-01, -1.5285e-01, ..., -4.4932e-02,
-3.2181e-02, -6.0735e-02]],
...,
[[-2.9447e+00, -2.5063e+00, -2.7588e+00, ..., -3.7467e+00,
-2.8621e+00, -1.8985e+00],
[-2.2300e+00, -9.2049e-01, -1.5935e+00, ..., -2.7333e+00,
-2.4477e+00, -1.9216e+00],
[-1.0596e+00, 9.8959e-01, 5.5291e-01, ..., -4.5006e-01,
3.1665e-02, 2.0341e-01],
...,
[-1.5885e+00, 1.6709e+00, 3.9750e-01, ..., -7.6338e-01,
-1.2443e-01, 3.8597e-01],
[-6.0448e-01, 1.8115e+00, 1.1626e+00, ..., 1.7411e-01,
3.2780e-01, 5.0012e-01],
[-1.7199e+00, -1.0145e+00, -1.3988e+00, ..., -2.0846e+00,
-1.1724e+00, -5.3777e-01]],
[[-1.5988e+00, -1.6001e+00, -2.8016e+00, ..., -1.1430e+00,
-7.0086e-01, -7.4082e-01],
[-6.5176e-01, -1.4287e-02, -3.0952e+00, ..., -1.2569e+00,
-1.1403e+00, -1.5754e+00],
[-2.1226e+00, -1.0085e+00, -5.1407e+00, ..., -3.2251e+00,
-2.7565e+00, -2.2650e+00],
...,
[-1.7630e+00, -6.7979e-01, -4.7094e+00, ..., -2.4839e+00,
-2.0510e+00, -1.6095e+00],
[-1.6728e+00, -7.5533e-01, -3.7380e+00, ..., -2.3186e+00,
-1.6861e+00, -1.0918e+00],
[-2.0763e+00, -1.3052e+00, -2.9886e+00, ..., -2.7776e+00,
-1.9611e+00, -1.6756e+00]],
[[-1.3251e+00, -1.8782e+00, -1.7057e+00, ..., -1.8267e+00,
-1.5092e+00, -1.1419e+00],
[-1.2544e+00, -1.7326e+00, -1.2578e+00, ..., -1.3774e+00,
-1.0848e+00, -8.6332e-01],
[-1.5698e+00, -1.9153e+00, -1.5152e+00, ..., -1.5814e+00,
-1.1788e+00, -8.6742e-01],
...,
[-2.4986e+00, -3.1228e+00, -2.8471e+00, ..., -3.0229e+00,
-2.3992e+00, -1.8352e+00],
[-1.6767e+00, -2.1376e+00, -2.0316e+00, ..., -2.1096e+00,
-1.7243e+00, -1.3770e+00],
[-1.1178e+00, -1.3462e+00, -1.3855e+00, ..., -1.3012e+00,
-1.1001e+00, -8.1598e-01]]],
[[[ 3.7320e-01, 8.4666e-01, 7.2338e-01, ..., 6.8573e-01,
9.6755e-01, 7.5337e-01],
[-1.1946e+00, -2.3947e-01, -1.3509e+00, ..., -1.0842e+00,
8.4962e-02, 5.7987e-01],
[ 7.0469e-01, 5.2811e-01, 5.3351e-01, ..., 5.9454e-01,
9.5633e-01, 8.8528e-01],
...,
[-1.9025e-01, -1.1845e+00, -2.6976e+00, ..., -1.5445e+00,
-1.4483e+00, -8.6578e-01],
[ 1.4297e+00, 8.0940e-01, -1.0518e+00, ..., -3.7361e-01,
2.9013e-02, 3.3463e-01],
[ 8.3091e-01, 2.4932e-01, -7.6144e-01, ..., -2.4983e-01,
1.6342e-01, 3.5199e-01]],
[[-2.4145e+00, -1.2552e+00, -4.2938e+00, ..., -5.2977e+00,
-4.5433e+00, -3.5561e+00],
[-2.1938e-01, 1.7004e+00, -2.5967e+00, ..., -4.1849e+00,
-4.0489e+00, -3.3206e+00],
[ 6.3199e-01, 1.5434e+00, -5.1460e+00, ..., -6.0776e+00,
-5.8248e+00, -5.3535e+00],
...,
[-1.3162e+00, 1.2416e-01, -6.8303e+00, ..., -7.6918e+00,
-7.0253e+00, -5.5187e+00],
[ 1.7738e+00, 2.8221e+00, -1.1000e+00, ..., -2.6172e+00,
-3.0309e+00, -2.4296e+00],
[ 1.2092e+00, 1.8380e+00, -2.3077e+00, ..., -3.2495e+00,
-3.6702e+00, -3.1655e+00]],
[[-6.7033e-01, -1.8146e+00, -1.6788e+00, ..., -9.1744e-01,
-6.5483e-01, -7.7914e-01],
[-1.3789e+00, -1.8644e+00, -1.2695e+00, ..., -7.2807e-01,
-3.9745e-01, -2.9563e-01],
[-1.1040e+00, -1.0618e+00, -5.0026e-01, ..., -3.4394e-01,
7.6014e-03, 3.2974e-02],
...,
[-1.4415e+00, -1.7894e+00, -7.5477e-01, ..., -8.6626e-01,
-7.7059e-01, -1.0427e+00],
[-1.7821e+00, -1.1889e+00, -8.2642e-01, ..., -5.6177e-01,
-2.9325e-01, -3.6391e-01],
[-1.9383e+00, -1.3399e+00, -9.3504e-01, ..., -1.0128e+00,
-8.8092e-01, -8.5621e-01]],
...,
[[-2.2139e+00, -1.1935e+00, -2.2888e+00, ..., -2.8748e+00,
-2.2932e+00, -1.5120e+00],
[-6.1602e-01, 1.1394e+00, -4.1352e-01, ..., -1.3410e+00,
-1.4297e+00, -1.2211e+00],
[-3.4345e-01, 2.1997e+00, 8.5145e-01, ..., -1.4182e-01,
2.0737e-01, 3.3433e-01],
...,
[-1.8667e+00, 1.2543e+00, 2.2612e-02, ..., -1.1228e+00,
-5.4670e-01, -7.9200e-03],
[-7.9295e-01, 1.9595e+00, 1.6958e+00, ..., 5.0076e-01,
4.3489e-01, 6.9903e-01],
[-2.2110e+00, -1.1298e+00, -9.4115e-01, ..., -1.8418e+00,
-9.4867e-01, -2.8128e-01]],
[[-1.3658e+00, -1.2602e+00, -2.7608e+00, ..., -1.1248e+00,
-6.2881e-01, -5.0993e-01],
[-6.6163e-01, -1.3692e-01, -3.6131e+00, ..., -1.8196e+00,
-1.5536e+00, -1.7120e+00],
[-2.0999e+00, -1.3718e+00, -6.0051e+00, ..., -3.7631e+00,
-3.2390e+00, -2.5754e+00],
...,
[-2.4922e+00, -1.5942e+00, -5.4916e+00, ..., -3.0742e+00,
-2.7586e+00, -2.2146e+00],
[-2.2279e+00, -1.4861e+00, -4.2861e+00, ..., -2.6823e+00,
-2.3530e+00, -1.6769e+00],
[-2.3902e+00, -1.8003e+00, -3.3317e+00, ..., -2.9827e+00,
-2.4736e+00, -2.1546e+00]],
[[-1.1604e+00, -1.7263e+00, -1.5021e+00, ..., -1.6461e+00,
-1.3607e+00, -1.0277e+00],
[-1.1704e+00, -1.6437e+00, -1.1733e+00, ..., -1.3776e+00,
-1.1125e+00, -8.8272e-01],
[-1.5290e+00, -1.8210e+00, -1.3485e+00, ..., -1.5708e+00,
-1.1958e+00, -8.7174e-01],
...,
[-3.2377e+00, -4.0450e+00, -3.8268e+00, ..., -4.0185e+00,
-3.2292e+00, -2.4926e+00],
[-2.1517e+00, -2.7779e+00, -2.7375e+00, ..., -2.8087e+00,
-2.3243e+00, -1.8948e+00],
[-1.5679e+00, -1.9016e+00, -1.9596e+00, ..., -1.8968e+00,
-1.5410e+00, -1.1823e+00]]],
[[[ 5.2916e-01, 1.0062e+00, 8.9986e-01, ..., -3.3566e-01,
-7.1067e-02, 2.2552e-02],
[-1.3449e+00, -4.0144e-01, -1.6886e+00, ..., -3.2180e+00,
-1.2680e+00, -4.2750e-01],
[ 6.8994e-01, 4.2938e-01, 3.2696e-01, ..., -1.4626e-01,
-3.1668e-01, 1.8784e-01],
...,
[-1.5772e-01, -7.7798e-01, -1.3933e+00, ..., -1.9401e+00,
-1.9135e+00, -1.2979e+00],
[ 1.0335e+00, 7.6398e-01, 2.6916e-01, ..., 2.6660e-02,
2.7163e-01, 4.5750e-01],
[ 6.7194e-01, 4.5685e-01, 1.4290e-01, ..., 2.2757e-01,
4.7500e-01, 5.1392e-01]],
[[-2.3746e+00, -9.9725e-01, -4.1223e+00, ..., -6.8359e+00,
-6.0673e+00, -4.5368e+00],
[-7.8613e-02, 2.1711e+00, -2.2564e+00, ..., -5.4461e+00,
-5.5613e+00, -4.3171e+00],
[ 7.6493e-01, 1.9956e+00, -4.9800e+00, ..., -7.8809e+00,
-7.8345e+00, -6.6106e+00],
...,
[-5.1896e-01, 5.0230e-01, -4.1493e+00, ..., -7.0960e+00,
-6.6720e+00, -5.1866e+00],
[ 1.9111e+00, 2.5626e+00, 7.1906e-02, ..., -1.8935e+00,
-2.5349e+00, -2.0387e+00],
[ 1.0946e+00, 1.3778e+00, -1.6220e+00, ..., -2.9179e+00,
-3.4817e+00, -3.0494e+00]],
[[-6.8044e-01, -1.7933e+00, -1.5726e+00, ..., -1.4053e+00,
-1.0642e+00, -1.3508e+00],
[-1.4952e+00, -1.8567e+00, -1.1154e+00, ..., -9.8561e-01,
-6.1551e-01, -9.0165e-01],
[-1.3116e+00, -1.1095e+00, -4.0231e-01, ..., -1.0796e+00,
-5.7794e-01, -4.7257e-01],
...,
[-8.8981e-01, -1.2762e+00, -4.8156e-01, ..., -1.0483e+00,
-8.8777e-01, -1.2579e+00],
[-1.2307e+00, -8.3146e-01, -5.0445e-01, ..., -5.5709e-01,
-2.5069e-01, -4.2584e-01],
[-1.2986e+00, -7.3537e-01, -3.4325e-01, ..., -4.9366e-01,
-4.3571e-01, -6.0539e-01]],
...,
[[-2.4720e+00, -1.3296e+00, -2.7522e+00, ..., -3.0853e+00,
-2.5110e+00, -1.8112e+00],
[-7.6199e-01, 1.1384e+00, -9.5750e-01, ..., -2.3985e+00,
-2.5070e+00, -2.0198e+00],
[-2.3583e-01, 2.5472e+00, 6.4464e-01, ..., -5.2161e-02,
3.9076e-02, 2.6979e-01],
...,
[-1.4423e+00, 1.1956e+00, -1.6328e-01, ..., -9.1258e-01,
-2.3583e-01, 2.9873e-01],
[-4.7860e-01, 1.4824e+00, 7.2436e-01, ..., 4.4329e-01,
4.9853e-01, 8.2179e-01],
[-1.5418e+00, -1.0119e+00, -1.5215e+00, ..., -2.2258e+00,
-1.1680e+00, -3.5412e-01]],
[[-1.2800e+00, -1.1399e+00, -2.7135e+00, ..., -2.2161e+00,
-1.9148e+00, -1.8225e+00],
[-4.7433e-01, 1.4676e-01, -3.4607e+00, ..., -2.8686e+00,
-2.8919e+00, -2.9120e+00],
[-2.0665e+00, -1.2162e+00, -6.0000e+00, ..., -4.4588e+00,
-4.5316e+00, -3.7204e+00],
...,
[-1.6608e+00, -5.6993e-01, -4.0434e+00, ..., -2.8223e+00,
-2.7374e+00, -2.3025e+00],
[-1.6534e+00, -7.4906e-01, -3.3495e+00, ..., -2.4392e+00,
-2.2570e+00, -1.6557e+00],
[-2.0637e+00, -1.3582e+00, -2.9058e+00, ..., -2.8733e+00,
-2.4380e+00, -2.2019e+00]],
[[-1.2050e+00, -1.7968e+00, -1.5496e+00, ..., -2.5770e+00,
-2.1509e+00, -1.6701e+00],
[-1.1869e+00, -1.6810e+00, -1.1588e+00, ..., -2.4070e+00,
-1.9356e+00, -1.5884e+00],
[-1.5373e+00, -1.8327e+00, -1.3394e+00, ..., -2.8232e+00,
-2.2556e+00, -1.7534e+00],
...,
[-2.2782e+00, -2.8404e+00, -2.6600e+00, ..., -4.0038e+00,
-3.2476e+00, -2.5418e+00],
[-1.5409e+00, -1.9572e+00, -1.9140e+00, ..., -2.7652e+00,
-2.3169e+00, -1.9149e+00],
[-1.0346e+00, -1.2440e+00, -1.3142e+00, ..., -1.8243e+00,
-1.5222e+00, -1.1772e+00]]]], device='cuda:0'),
tensor([[[[ 1.5647e-01, 3.8747e-01, -1.1601e+00, ..., 7.0079e-01,
9.2476e-01, 6.8261e-01],
[-2.0866e+00, -1.4330e+00, -4.3512e+00, ..., -1.3961e+00,
-1.9422e-01, 3.8182e-01],
[ 4.3357e-01, -2.6965e-01, -1.1083e+00, ..., 6.3242e-01,
9.0007e-01, 8.4790e-01],
...,
[-2.6376e-01, -1.2188e+00, -2.5315e+00, ..., -1.2179e+00,
-8.5622e-01, -6.7017e-01],
[ 1.2535e+00, 7.2392e-01, -8.2989e-01, ..., 5.0022e-01,
7.4740e-01, 7.9240e-01],
[ 8.0638e-01, 3.3266e-01, -4.0025e-01, ..., 5.2865e-01,
7.7490e-01, 6.8248e-01]],
[[-3.2413e+00, -1.4775e+00, -6.1230e+00, ..., -5.5221e+00,
-4.6405e+00, -3.6450e+00],
[-5.8459e-01, 2.1250e+00, -4.1633e+00, ..., -4.3320e+00,
-4.0300e+00, -3.3255e+00],
[ 2.9596e-02, 1.8453e+00, -7.3539e+00, ..., -6.2459e+00,
-5.8344e+00, -5.4273e+00],
...,
[-1.2315e+00, 2.9477e-01, -6.5254e+00, ..., -5.3777e+00,
-4.7095e+00, -3.9409e+00],
[ 1.5247e+00, 2.5873e+00, -1.3853e+00, ..., -1.2581e+00,
-1.3957e+00, -1.4934e+00],
[ 9.8906e-01, 1.6173e+00, -2.4270e+00, ..., -2.4885e+00,
-2.6024e+00, -2.5604e+00]],
[[-1.3612e+00, -2.3232e+00, -2.1495e+00, ..., -8.6218e-01,
-6.5034e-01, -8.1848e-01],
[-2.0066e+00, -2.1958e+00, -1.5625e+00, ..., -6.8323e-01,
-4.3202e-01, -3.6798e-01],
[-2.5129e+00, -2.0013e+00, -8.4381e-01, ..., -4.0429e-01,
-1.0576e-01, -1.0524e-01],
...,
[-1.5682e+00, -1.8208e+00, -7.7385e-01, ..., -6.4352e-01,
-5.4219e-01, -7.0855e-01],
[-1.7417e+00, -1.1844e+00, -7.7030e-01, ..., -2.3430e-01,
-6.1732e-02, -5.9875e-02],
[-1.7268e+00, -1.1240e+00, -6.0243e-01, ..., -3.2633e-01,
-2.8893e-01, -2.7114e-01]],
...,
[[-2.3518e+00, -1.3163e+00, -2.7498e+00, ..., -3.0736e+00,
-2.3854e+00, -1.6268e+00],
[-1.8679e+00, 2.1071e-01, -1.3185e+00, ..., -1.5147e+00,
-1.4880e+00, -1.2985e+00],
[-4.8933e-01, 2.5085e+00, 9.4244e-01, ..., -1.5745e-01,
2.2976e-01, 3.3685e-01],
...,
[-1.7564e+00, 1.1399e+00, 2.1648e-01, ..., -1.0090e+00,
-2.9983e-01, 1.7194e-01],
[-8.8616e-01, 1.5936e+00, 1.4273e+00, ..., 1.5959e-01,
3.0987e-01, 4.0961e-01],
[-2.0769e+00, -1.0462e+00, -8.3721e-01, ..., -2.1685e+00,
-1.2535e+00, -6.4670e-01]],
[[-2.3611e+00, -2.6099e+00, -3.9997e+00, ..., -1.1803e+00,
-6.2167e-01, -4.8690e-01],
[-1.6532e+00, -1.3155e+00, -4.7335e+00, ..., -1.7549e+00,
-1.4951e+00, -1.7195e+00],
[-3.0076e+00, -2.4235e+00, -6.8891e+00, ..., -3.7792e+00,
-3.2736e+00, -2.6437e+00],
...,
[-2.5096e+00, -1.7476e+00, -5.0517e+00, ..., -2.1707e+00,
-1.7253e+00, -1.3981e+00],
[-2.1944e+00, -1.5467e+00, -3.9123e+00, ..., -2.1238e+00,
-1.4830e+00, -9.6506e-01],
[-2.2656e+00, -1.7496e+00, -3.0826e+00, ..., -2.8271e+00,
-1.9598e+00, -1.6761e+00]],
[[-1.8847e+00, -2.6078e+00, -2.3948e+00, ..., -1.6986e+00,
-1.4033e+00, -1.0502e+00],
[-1.9071e+00, -2.6156e+00, -2.0806e+00, ..., -1.3882e+00,
-1.1087e+00, -8.7779e-01],
[-2.4313e+00, -2.9836e+00, -2.4931e+00, ..., -1.5975e+00,
-1.1994e+00, -8.7476e-01],
...,
[-3.1011e+00, -3.8681e+00, -3.6330e+00, ..., -2.7834e+00,
-2.2223e+00, -1.6790e+00],
[-2.0726e+00, -2.6747e+00, -2.6347e+00, ..., -1.9507e+00,
-1.6021e+00, -1.2767e+00],
[-1.4963e+00, -1.8142e+00, -1.8598e+00, ..., -1.2144e+00,
-1.0368e+00, -7.6414e-01]]],
[[[ 3.0117e-01, 5.6757e-01, 7.1393e-01, ..., 4.7212e-01,
1.0181e+00, 6.0197e-01],
[-2.1545e+00, -1.5062e+00, -3.1682e+00, ..., -2.6776e+00,
-8.5910e-01, -1.8736e-01],
[-1.6831e-01, -1.0046e+00, -2.1651e-01, ..., -1.5226e-01,
5.5609e-01, 6.6917e-01],
...,
[-1.6238e-01, -1.1269e+00, -2.6349e+00, ..., -1.5088e+00,
-1.4121e+00, -8.6334e-01],
[ 1.4287e+00, 8.2838e-01, -9.5673e-01, ..., -2.7829e-01,
1.0100e-01, 3.7761e-01],
[ 8.5741e-01, 3.0075e-01, -6.7395e-01, ..., -1.6867e-01,
2.1975e-01, 3.8299e-01]],
[[-1.6534e+00, -1.1355e+00, -4.1766e+00, ..., -4.8663e+00,
-4.4485e+00, -3.3850e+00],
[ 6.1390e-01, 1.9475e+00, -2.2513e+00, ..., -3.4851e+00,
-3.6132e+00, -2.9662e+00],
[ 1.3576e+00, 1.7151e+00, -4.8913e+00, ..., -5.4204e+00,
-5.2614e+00, -4.9467e+00],
...,
[-1.2946e+00, 1.4132e-01, -6.7147e+00, ..., -7.6190e+00,
-6.9252e+00, -5.4275e+00],
[ 1.7663e+00, 2.7879e+00, -1.0840e+00, ..., -2.6194e+00,
-2.9878e+00, -2.3934e+00],
[ 1.1888e+00, 1.7859e+00, -2.2944e+00, ..., -3.2706e+00,
-3.6427e+00, -3.1376e+00]],
[[-1.1883e-01, -1.1516e+00, -1.3148e+00, ..., -4.8204e-01,
-1.4513e-01, -4.8419e-01],
[-6.0234e-01, -8.6195e-01, -7.8002e-01, ..., -2.7387e-01,
4.9730e-02, -8.4618e-02],
[-5.5243e-01, -3.5624e-01, -4.0072e-01, ..., -3.3292e-01,
9.4977e-02, 8.6209e-02],
...,
[-1.3529e+00, -1.7316e+00, -7.5959e-01, ..., -8.8045e-01,
-7.6476e-01, -1.0092e+00],
[-1.7048e+00, -1.1469e+00, -8.2241e-01, ..., -5.5927e-01,
-2.9524e-01, -3.5987e-01],
[-1.8517e+00, -1.2757e+00, -8.9576e-01, ..., -9.5230e-01,
-8.2374e-01, -7.9243e-01]],
...,
[[-3.0509e+00, -2.5092e+00, -2.8220e+00, ..., -3.4504e+00,
-2.6367e+00, -1.7490e+00],
[-2.2965e+00, -8.4853e-01, -1.6301e+00, ..., -2.6076e+00,
-2.3576e+00, -1.8453e+00],
[-1.0753e+00, 1.1652e+00, 5.8536e-01, ..., -4.6954e-01,
-8.5403e-03, 1.4332e-01],
...,
[-1.8482e+00, 1.2552e+00, 7.6533e-04, ..., -1.0846e+00,
-5.1860e-01, 1.2988e-03],
[-8.2032e-01, 1.9089e+00, 1.6154e+00, ..., 4.5023e-01,
4.1762e-01, 6.7954e-01],
[-2.2504e+00, -1.1639e+00, -1.0026e+00, ..., -1.8715e+00,
-9.6592e-01, -2.9557e-01]],
[[-1.5978e+00, -1.5869e+00, -2.9320e+00, ..., -1.2585e+00,
-8.2988e-01, -8.3536e-01],
[-5.9586e-01, 7.3739e-02, -3.2309e+00, ..., -1.3574e+00,
-1.2162e+00, -1.5829e+00],
[-2.0911e+00, -8.9956e-01, -5.3537e+00, ..., -3.1658e+00,
-2.7147e+00, -2.2139e+00],
...,
[-2.4553e+00, -1.5619e+00, -5.4501e+00, ..., -3.1133e+00,
-2.7384e+00, -2.1787e+00],
[-2.2148e+00, -1.4723e+00, -4.2500e+00, ..., -2.6968e+00,
-2.3175e+00, -1.6444e+00],
[-2.3859e+00, -1.7687e+00, -3.3066e+00, ..., -2.9713e+00,
-2.4345e+00, -2.1234e+00]],
[[-1.3565e+00, -1.9113e+00, -1.7086e+00, ..., -1.7944e+00,
-1.4798e+00, -1.1279e+00],
[-1.2767e+00, -1.7659e+00, -1.2341e+00, ..., -1.3996e+00,
-1.1017e+00, -8.7739e-01],
[-1.6046e+00, -1.9519e+00, -1.4832e+00, ..., -1.6218e+00,
-1.2182e+00, -8.9951e-01],
...,
[-3.1741e+00, -3.9677e+00, -3.7659e+00, ..., -3.9495e+00,
-3.1674e+00, -2.4414e+00],
[-2.1079e+00, -2.7245e+00, -2.6973e+00, ..., -2.7734e+00,
-2.2931e+00, -1.8645e+00],
[-1.5333e+00, -1.8616e+00, -1.9299e+00, ..., -1.8737e+00,
-1.5218e+00, -1.1667e+00]]],
[[[ 2.9405e-01, 5.9205e-01, 6.7403e-01, ..., -5.2513e-01,
-2.7507e-01, -1.1159e-01],
[-1.9524e+00, -1.1735e+00, -2.7556e+00, ..., -3.5907e+00,
-1.5572e+00, -6.2473e-01],
[ 9.1517e-02, -6.6145e-01, 9.5640e-02, ..., -2.8101e-01,
-5.7284e-01, 7.6558e-02],
...,
[-1.0953e+00, -2.6784e+00, -2.5562e+00, ..., -2.0012e+00,
-1.9680e+00, -1.2897e+00],
[ 8.1449e-01, 5.6851e-02, 1.6087e-02, ..., -1.6232e-01,
1.1532e-01, 3.5817e-01],
[ 2.6871e-01, -3.9403e-01, -2.4029e-01, ..., 3.5198e-02,
3.5529e-01, 4.4897e-01]],
[[-1.8443e+00, -1.2468e+00, -4.2836e+00, ..., -7.1242e+00,
-6.3224e+00, -4.6941e+00],
[ 5.3302e-01, 1.8976e+00, -2.3622e+00, ..., -5.7088e+00,
-5.8307e+00, -4.4759e+00],
[ 1.3993e+00, 1.6501e+00, -5.0583e+00, ..., -8.2095e+00,
-8.1389e+00, -6.7991e+00],
...,
[ 2.2257e-02, 1.7746e-01, -4.2264e+00, ..., -7.4304e+00,
-7.0459e+00, -5.3943e+00],
[ 2.2506e+00, 2.2272e+00, -1.7328e-01, ..., -2.0851e+00,
-2.7859e+00, -2.1439e+00],
[ 1.3266e+00, 1.1658e+00, -1.7571e+00, ..., -3.0040e+00,
-3.6313e+00, -3.1073e+00]],
[[-2.8720e-01, -1.4566e+00, -1.6120e+00, ..., -1.5494e+00,
-1.1617e+00, -1.4337e+00],
[-8.1863e-01, -1.1875e+00, -1.0647e+00, ..., -1.0896e+00,
-6.9065e-01, -1.0055e+00],
[-6.1654e-01, -4.6132e-01, -4.7528e-01, ..., -1.2119e+00,
-6.7235e-01, -5.2136e-01],
...,
[-7.4663e-01, -1.1557e+00, -7.2439e-01, ..., -1.0577e+00,
-8.9535e-01, -1.2722e+00],
[-7.0408e-01, -2.7134e-01, -5.4885e-01, ..., -5.6019e-01,
-2.3582e-01, -4.1799e-01],
[-8.5459e-01, -2.8868e-01, -4.3148e-01, ..., -6.3556e-01,
-5.8869e-01, -7.4497e-01]],
...,
[[-2.9936e+00, -2.3349e+00, -2.5859e+00, ..., -3.0790e+00,
-2.4802e+00, -1.8218e+00],
[-1.9178e+00, -3.2618e-01, -9.6255e-01, ..., -2.6057e+00,
-2.6505e+00, -2.1239e+00],
[-9.7884e-01, 1.3317e+00, 9.4396e-01, ..., -9.3979e-02,
8.6857e-03, 2.6170e-01],
...,
[-1.8409e+00, -2.2084e-02, -5.6961e-02, ..., -9.6486e-01,
-3.2218e-01, 2.3977e-01],
[-1.0927e+00, 8.5388e-01, 9.6768e-01, ..., 5.3397e-01,
5.1574e-01, 8.6607e-01],
[-1.9905e+00, -1.3666e+00, -1.5849e+00, ..., -2.1263e+00,
-1.0969e+00, -2.8723e-01]],
[[-1.5741e+00, -1.5148e+00, -2.8885e+00, ..., -2.4306e+00,
-2.1274e+00, -1.9981e+00],
[-5.6494e-01, 1.1728e-01, -3.3385e+00, ..., -3.0696e+00,
-3.0899e+00, -3.0504e+00],
[-2.0443e+00, -8.8875e-01, -5.4887e+00, ..., -4.5996e+00,
-4.7009e+00, -3.8462e+00],
...,
[-1.5129e+00, -3.6865e-01, -3.5783e+00, ..., -2.8792e+00,
-2.8180e+00, -2.3780e+00],
[-1.6687e+00, -6.0679e-01, -3.0819e+00, ..., -2.4908e+00,
-2.3509e+00, -1.7580e+00],
[-2.2307e+00, -1.3803e+00, -3.0030e+00, ..., -2.9123e+00,
-2.5164e+00, -2.2828e+00]],
[[-1.3347e+00, -1.8943e+00, -1.6562e+00, ..., -2.7135e+00,
-2.2675e+00, -1.7599e+00],
[-1.2688e+00, -1.7569e+00, -1.1856e+00, ..., -2.5598e+00,
-2.0569e+00, -1.6858e+00],
[-1.5960e+00, -1.9271e+00, -1.4178e+00, ..., -3.0185e+00,
-2.4073e+00, -1.8729e+00],
...,
[-2.4900e+00, -3.0607e+00, -2.8529e+00, ..., -4.1871e+00,
-3.4002e+00, -2.6773e+00],
[-1.5538e+00, -2.0294e+00, -1.9504e+00, ..., -2.8802e+00,
-2.4090e+00, -1.9983e+00],
[-1.0281e+00, -1.3205e+00, -1.3548e+00, ..., -1.9127e+00,
-1.5774e+00, -1.2318e+00]]],
[[[ 2.7507e-01, 5.3016e-01, 6.8119e-01, ..., 4.8145e-01,
1.0329e+00, 6.1561e-01],
[-2.0973e+00, -1.4882e+00, -3.0730e+00, ..., -2.8243e+00,
-9.3928e-01, -1.9993e-01],
[-2.1314e-01, -9.9180e-01, -2.1362e-01, ..., -1.7133e-01,
5.4327e-01, 6.2930e-01],
...,
[-1.6495e-01, -1.1253e+00, -2.6371e+00, ..., -1.4973e+00,
-1.4330e+00, -8.5591e-01],
[ 1.4401e+00, 8.5270e-01, -9.5754e-01, ..., -2.8462e-01,
6.5959e-02, 3.6401e-01],
[ 8.4475e-01, 2.8458e-01, -7.0465e-01, ..., -1.9515e-01,
1.9387e-01, 3.7063e-01]],
[[-1.6774e+00, -1.3072e+00, -4.2247e+00, ..., -5.0405e+00,
-4.6215e+00, -3.4884e+00],
[ 4.3958e-01, 1.5625e+00, -2.4316e+00, ..., -3.5025e+00,
-3.6732e+00, -2.9722e+00],
[ 1.0736e+00, 1.2697e+00, -4.9724e+00, ..., -5.5327e+00,
-5.4011e+00, -5.0226e+00],
...,
[-1.2920e+00, 1.1821e-01, -6.7488e+00, ..., -7.6595e+00,
-6.9357e+00, -5.4479e+00],
[ 1.7642e+00, 2.7635e+00, -1.1129e+00, ..., -2.6304e+00,
-2.9806e+00, -2.4061e+00],
[ 1.1926e+00, 1.7811e+00, -2.3099e+00, ..., -3.2687e+00,
-3.6435e+00, -3.1421e+00]],
[[-9.7836e-02, -1.0298e+00, -1.2252e+00, ..., -5.7413e-01,
-2.0232e-01, -5.3395e-01],
[-4.9548e-01, -7.1395e-01, -6.9936e-01, ..., -3.5893e-01,
5.0142e-03, -7.8827e-02],
[-4.5452e-01, -2.6123e-01, -3.7537e-01, ..., -3.8582e-01,
8.8562e-02, 1.0153e-01],
...,
[-1.3946e+00, -1.7388e+00, -7.4169e-01, ..., -8.9124e-01,
-7.7035e-01, -1.0237e+00],
[-1.7409e+00, -1.1490e+00, -8.1139e-01, ..., -5.6794e-01,
-3.0632e-01, -3.6506e-01],
[-1.8890e+00, -1.2919e+00, -9.0237e-01, ..., -9.6446e-01,
-8.4649e-01, -8.2179e-01]],
...,
[[-2.8866e+00, -2.4685e+00, -2.7124e+00, ..., -3.7231e+00,
-2.8424e+00, -1.8738e+00],
[-2.2193e+00, -9.6474e-01, -1.6029e+00, ..., -2.7186e+00,
-2.4581e+00, -1.9402e+00],
[-1.0854e+00, 9.0615e-01, 4.8411e-01, ..., -4.5081e-01,
2.4976e-02, 1.7851e-01],
...,
[-1.8618e+00, 1.2355e+00, 3.1310e-02, ..., -1.0829e+00,
-5.3746e-01, -8.2098e-03],
[-8.1584e-01, 1.8783e+00, 1.6218e+00, ..., 4.8946e-01,
4.2902e-01, 6.7760e-01],
[-2.2078e+00, -1.1511e+00, -9.8154e-01, ..., -1.8433e+00,
-9.6405e-01, -2.9453e-01]],
[[-1.6103e+00, -1.6195e+00, -2.7483e+00, ..., -1.1611e+00,
-7.1644e-01, -7.5571e-01],
[-6.8732e-01, -7.0442e-02, -3.0197e+00, ..., -1.2710e+00,
-1.1389e+00, -1.5539e+00],
[-2.1461e+00, -1.0584e+00, -5.0442e+00, ..., -3.2286e+00,
-2.7518e+00, -2.2198e+00],
...,
[-2.4912e+00, -1.6252e+00, -5.5155e+00, ..., -3.0826e+00,
-2.7219e+00, -2.1926e+00],
[-2.2189e+00, -1.4915e+00, -4.2766e+00, ..., -2.6768e+00,
-2.3215e+00, -1.6575e+00],
[-2.3753e+00, -1.7762e+00, -3.3107e+00, ..., -2.9769e+00,
-2.4548e+00, -2.1251e+00]],
[[-1.3183e+00, -1.8548e+00, -1.7081e+00, ..., -1.8338e+00,
-1.5115e+00, -1.1484e+00],
[-1.2456e+00, -1.7153e+00, -1.2748e+00, ..., -1.3803e+00,
-1.0952e+00, -8.7201e-01],
[-1.5542e+00, -1.9019e+00, -1.5259e+00, ..., -1.5872e+00,
-1.1852e+00, -8.6695e-01],
...,
[-3.1893e+00, -3.9778e+00, -3.7683e+00, ..., -3.9766e+00,
-3.1835e+00, -2.4514e+00],
[-2.1278e+00, -2.7436e+00, -2.7147e+00, ..., -2.7932e+00,
-2.3014e+00, -1.8680e+00],
[-1.5530e+00, -1.8834e+00, -1.9467e+00, ..., -1.8877e+00,
-1.5278e+00, -1.1696e+00]]],
[[[ 3.1395e-01, 6.1408e-01, 7.1995e-01, ..., 5.3213e-01,
1.0163e+00, 6.3560e-01],
[-2.1108e+00, -1.4141e+00, -3.0949e+00, ..., -2.4351e+00,
-7.5847e-01, -9.6474e-02],
[-9.8702e-02, -8.8418e-01, -1.8279e-01, ..., -2.4734e-03,
6.0711e-01, 7.1042e-01],
...,
[-2.0345e-01, -1.1818e+00, -2.7041e+00, ..., -1.5287e+00,
-1.4363e+00, -8.8543e-01],
[ 1.4169e+00, 7.9294e-01, -1.0437e+00, ..., -3.3272e-01,
5.7896e-02, 3.2749e-01],
[ 8.1587e-01, 2.4247e-01, -7.6570e-01, ..., -2.4318e-01,
1.8078e-01, 3.5009e-01]],
[[-1.7137e+00, -1.1317e+00, -4.1582e+00, ..., -4.9126e+00,
-4.4350e+00, -3.4178e+00],
[ 5.6091e-01, 1.9720e+00, -2.2395e+00, ..., -3.6141e+00,
-3.6815e+00, -3.0282e+00],
[ 1.3121e+00, 1.7555e+00, -4.9075e+00, ..., -5.5652e+00,
-5.3503e+00, -4.9676e+00],
...,
[-1.4357e+00, -1.9954e-02, -6.8568e+00, ..., -7.7405e+00,
-7.0636e+00, -5.4809e+00],
[ 1.6775e+00, 2.7223e+00, -1.0916e+00, ..., -2.6605e+00,
-3.0471e+00, -2.3981e+00],
[ 1.1357e+00, 1.7720e+00, -2.2510e+00, ..., -3.3002e+00,
-3.6853e+00, -3.1517e+00]],
[[-1.3219e-01, -1.1811e+00, -1.3233e+00, ..., -5.1768e-01,
-1.9752e-01, -4.8731e-01],
[-6.5144e-01, -9.2711e-01, -8.0945e-01, ..., -2.8222e-01,
2.7126e-02, -7.1808e-02],
[-6.0230e-01, -4.0975e-01, -3.9887e-01, ..., -3.5656e-01,
7.2743e-02, 8.2374e-02],
...,
[-1.3501e+00, -1.7050e+00, -7.1393e-01, ..., -8.8931e-01,
-7.8196e-01, -1.0417e+00],
[-1.6854e+00, -1.1272e+00, -8.2200e-01, ..., -5.5536e-01,
-2.9595e-01, -3.7469e-01],
[-1.8763e+00, -1.3258e+00, -9.5425e-01, ..., -1.0122e+00,
-8.7806e-01, -8.6449e-01]],
...,
[[-3.0234e+00, -2.4534e+00, -2.7985e+00, ..., -3.4403e+00,
-2.5631e+00, -1.7085e+00],
[-2.2227e+00, -7.2601e-01, -1.5327e+00, ..., -2.6045e+00,
-2.2656e+00, -1.7276e+00],
[-1.0283e+00, 1.2569e+00, 6.2318e-01, ..., -5.2194e-01,
-5.8406e-02, 1.2671e-01],
...,
[-1.9220e+00, 1.1000e+00, -8.2291e-02, ..., -1.0792e+00,
-5.5497e-01, -2.2118e-02],
[-9.0635e-01, 1.8233e+00, 1.6301e+00, ..., 5.4274e-01,
4.3579e-01, 7.0173e-01],
[-2.3135e+00, -1.2509e+00, -9.9295e-01, ..., -1.8014e+00,
-9.3666e-01, -2.7258e-01]],
[[-1.5829e+00, -1.5527e+00, -2.8921e+00, ..., -1.3145e+00,
-8.5677e-01, -8.3528e-01],
[-5.8890e-01, 1.1717e-01, -3.2380e+00, ..., -1.4697e+00,
-1.2776e+00, -1.5950e+00],
[-2.0932e+00, -8.9049e-01, -5.3805e+00, ..., -3.2866e+00,
-2.7665e+00, -2.2383e+00],
...,
[-2.5332e+00, -1.6692e+00, -5.3951e+00, ..., -3.1023e+00,
-2.7557e+00, -2.1918e+00],
[-2.2639e+00, -1.5697e+00, -4.2421e+00, ..., -2.7047e+00,
-2.3410e+00, -1.6694e+00],
[-2.4201e+00, -1.8435e+00, -3.3076e+00, ..., -2.9818e+00,
-2.4689e+00, -2.1572e+00]],
[[-1.3480e+00, -1.9209e+00, -1.7060e+00, ..., -1.7545e+00,
-1.4489e+00, -1.1027e+00],
[-1.2742e+00, -1.7696e+00, -1.2338e+00, ..., -1.3872e+00,
-1.0945e+00, -8.6789e-01],
[-1.6034e+00, -1.9519e+00, -1.4766e+00, ..., -1.6141e+00,
-1.2110e+00, -8.9266e-01],
...,
[-3.1985e+00, -4.0171e+00, -3.8372e+00, ..., -4.0230e+00,
-3.2316e+00, -2.4879e+00],
[-2.1133e+00, -2.7522e+00, -2.7358e+00, ..., -2.8239e+00,
-2.3320e+00, -1.8934e+00],
[-1.5413e+00, -1.8849e+00, -1.9524e+00, ..., -1.9115e+00,
-1.5475e+00, -1.1852e+00]]]], device='cuda:0'),
tensor([[[[ 1.6223e-01, 3.9635e-01, -1.1531e+00, ..., 4.6257e-01,
1.0256e+00, 6.5520e-01],
[-2.0686e+00, -1.4092e+00, -4.3515e+00, ..., -2.6229e+00,
-7.5600e-01, -6.3010e-02],
[ 4.6539e-01, -2.5259e-01, -1.1078e+00, ..., -3.1385e-01,
5.0710e-01, 6.3721e-01],
...,
[-2.7640e-01, -1.2680e+00, -2.5812e+00, ..., -2.0509e+00,
-1.4377e+00, -9.3809e-01],
[ 1.2299e+00, 6.5850e-01, -8.8056e-01, ..., -3.2974e-01,
4.8108e-01, 6.8870e-01],
[ 8.0549e-01, 3.1456e-01, -3.9418e-01, ..., -3.5388e-01,
2.7332e-01, 2.6879e-01]],
[[-3.2348e+00, -1.4547e+00, -6.1247e+00, ..., -4.8390e+00,
-4.4181e+00, -3.3292e+00],
[-5.6702e-01, 2.1491e+00, -4.1639e+00, ..., -3.4136e+00,
-3.5508e+00, -2.8813e+00],
[ 3.2996e-02, 1.8559e+00, -7.3855e+00, ..., -5.3983e+00,
-5.2579e+00, -4.9119e+00],
...,
[-1.2402e+00, 2.3616e-01, -6.5049e+00, ..., -4.5663e+00,
-4.1241e+00, -3.4561e+00],
[ 1.5316e+00, 2.5894e+00, -1.3045e+00, ..., -8.6446e-01,
-1.1839e+00, -1.2443e+00],
[ 9.9346e-01, 1.6160e+00, -2.3768e+00, ..., -2.2379e+00,
-2.4635e+00, -2.3613e+00]],
[[-1.3835e+00, -2.3574e+00, -2.1777e+00, ..., -5.7172e-01,
-1.7865e-01, -4.6670e-01],
[-2.0155e+00, -2.1949e+00, -1.5556e+00, ..., -3.3161e-01,
8.7603e-02, 3.2093e-02],
[-2.5138e+00, -1.9928e+00, -8.2294e-01, ..., -2.7886e-01,
2.6042e-01, 2.8130e-01],
...,
[-1.5871e+00, -1.8084e+00, -7.6344e-01, ..., -4.9924e-01,
-1.5895e-01, -3.3896e-01],
[-1.7410e+00, -1.1597e+00, -7.3749e-01, ..., -3.2718e-01,
1.4703e-01, 2.2131e-01],
[-1.7088e+00, -1.1044e+00, -5.8117e-01, ..., -5.0450e-01,
-2.7974e-01, -2.5170e-01]],
...,
[[-2.3747e+00, -1.3188e+00, -2.7393e+00, ..., -3.6697e+00,
-2.8858e+00, -1.8595e+00],
[-1.8698e+00, 2.1627e-01, -1.3089e+00, ..., -2.6538e+00,
-2.4779e+00, -1.8745e+00],
[-5.0956e-01, 2.4965e+00, 9.2898e-01, ..., -5.2135e-01,
-1.0282e-01, 1.5587e-01],
...,
[-1.7250e+00, 1.1242e+00, 1.9453e-01, ..., -1.5741e+00,
-6.9727e-01, -1.4438e-01],
[-8.6491e-01, 1.6072e+00, 1.4509e+00, ..., -1.5101e-01,
1.0251e-01, 3.2620e-01],
[-2.1216e+00, -1.1227e+00, -9.5415e-01, ..., -2.3542e+00,
-1.4503e+00, -9.1516e-01]],
[[-2.3641e+00, -2.6225e+00, -3.9987e+00, ..., -1.1169e+00,
-7.4741e-01, -7.8626e-01],
[-1.6473e+00, -1.3225e+00, -4.7318e+00, ..., -1.3450e+00,
-1.2208e+00, -1.6247e+00],
[-2.9968e+00, -2.4339e+00, -6.8920e+00, ..., -3.2433e+00,
-2.7716e+00, -2.2611e+00],
...,
[-2.5124e+00, -1.7421e+00, -4.9352e+00, ..., -1.4466e+00,
-1.0426e+00, -8.8393e-01],
[-2.2187e+00, -1.5602e+00, -3.8697e+00, ..., -1.7137e+00,
-1.1466e+00, -7.6128e-01],
[-2.3138e+00, -1.7866e+00, -3.0665e+00, ..., -2.8114e+00,
-2.0252e+00, -1.8436e+00]],
[[-1.8916e+00, -2.6157e+00, -2.4005e+00, ..., -1.7968e+00,
-1.4902e+00, -1.1379e+00],
[-1.9160e+00, -2.6273e+00, -2.0875e+00, ..., -1.4001e+00,
-1.1013e+00, -8.8690e-01],
[-2.4356e+00, -2.9898e+00, -2.4969e+00, ..., -1.6054e+00,
-1.2041e+00, -8.9568e-01],
...,
[-3.0983e+00, -3.8650e+00, -3.6564e+00, ..., -2.8814e+00,
-2.2996e+00, -1.7699e+00],
[-2.0509e+00, -2.6537e+00, -2.6242e+00, ..., -1.9836e+00,
-1.6281e+00, -1.3144e+00],
[-1.4730e+00, -1.7917e+00, -1.8484e+00, ..., -1.2262e+00,
-1.0352e+00, -7.9289e-01]]],
[[[ 5.1674e-01, 9.9861e-01, 8.8991e-01, ..., -3.2425e-01,
-8.6214e-02, 7.8490e-03],
[-1.3118e+00, -3.6714e-01, -1.6572e+00, ..., -3.1437e+00,
-1.2799e+00, -4.3674e-01],
[ 6.7771e-01, 4.2779e-01, 3.3559e-01, ..., -1.1947e-01,
-3.4496e-01, 1.8227e-01],
...,
[-1.3867e-01, -8.0732e-01, -1.3458e+00, ..., -1.8944e+00,
-1.8704e+00, -1.2408e+00],
[ 9.9520e-01, 7.2754e-01, 2.9816e-01, ..., 7.7634e-02,
2.7506e-01, 4.6664e-01],
[ 6.4035e-01, 4.5972e-01, 1.5054e-01, ..., 2.7579e-01,
4.8104e-01, 5.2433e-01]],
[[-2.3464e+00, -9.4415e-01, -4.0883e+00, ..., -6.8349e+00,
-6.0301e+00, -4.5038e+00],
[-9.1399e-02, 2.1825e+00, -2.2798e+00, ..., -5.4789e+00,
-5.5316e+00, -4.2648e+00],
[ 7.5565e-01, 2.0056e+00, -4.9914e+00, ..., -7.8342e+00,
-7.7520e+00, -6.5338e+00],
...,
[-4.5354e-01, 5.7483e-01, -4.1341e+00, ..., -7.1056e+00,
-6.6599e+00, -5.1959e+00],
[ 1.9342e+00, 2.5408e+00, 4.9780e-03, ..., -2.0281e+00,
-2.6148e+00, -2.1142e+00],
[ 1.0936e+00, 1.3546e+00, -1.6597e+00, ..., -2.9733e+00,
-3.4808e+00, -3.0687e+00]],
[[-6.3665e-01, -1.7600e+00, -1.5436e+00, ..., -1.4011e+00,
-1.0386e+00, -1.3253e+00],
[-1.4553e+00, -1.8392e+00, -1.1057e+00, ..., -9.8185e-01,
-6.1297e-01, -8.9642e-01],
[-1.2928e+00, -1.1192e+00, -4.2559e-01, ..., -1.0919e+00,
-5.8550e-01, -4.5805e-01],
...,
[-8.7985e-01, -1.2729e+00, -4.7170e-01, ..., -1.0210e+00,
-8.5857e-01, -1.2213e+00],
[-1.2291e+00, -8.4791e-01, -5.2617e-01, ..., -5.5685e-01,
-2.6969e-01, -4.3725e-01],
[-1.3171e+00, -7.6630e-01, -3.8198e-01, ..., -4.7928e-01,
-4.3826e-01, -5.9233e-01]],
...,
[[-2.4148e+00, -1.2686e+00, -2.6670e+00, ..., -3.0086e+00,
-2.4460e+00, -1.7938e+00],
[-7.8594e-01, 1.1503e+00, -9.1975e-01, ..., -2.3594e+00,
-2.4316e+00, -1.9769e+00],
[-2.2833e-01, 2.5806e+00, 6.9134e-01, ..., -1.0870e-01,
1.5943e-02, 2.7066e-01],
...,
[-1.4036e+00, 1.2514e+00, -2.0128e-01, ..., -9.2042e-01,
-2.3709e-01, 2.9162e-01],
[-4.5981e-01, 1.5482e+00, 7.2951e-01, ..., 3.7302e-01,
4.8505e-01, 8.0340e-01],
[-1.4780e+00, -9.0605e-01, -1.4021e+00, ..., -2.1741e+00,
-1.1302e+00, -3.3210e-01]],
[[-1.2908e+00, -1.1596e+00, -2.7349e+00, ..., -2.2859e+00,
-1.9278e+00, -1.8072e+00],
[-4.9527e-01, 1.2734e-01, -3.5022e+00, ..., -2.8910e+00,
-2.8628e+00, -2.8722e+00],
[-2.0542e+00, -1.1995e+00, -6.0120e+00, ..., -4.4546e+00,
-4.4997e+00, -3.6934e+00],
...,
[-1.6320e+00, -5.6049e-01, -4.1237e+00, ..., -2.8773e+00,
-2.7715e+00, -2.3400e+00],
[-1.6273e+00, -7.4688e-01, -3.4099e+00, ..., -2.4585e+00,
-2.2831e+00, -1.7028e+00],
[-2.0307e+00, -1.3243e+00, -2.9037e+00, ..., -2.8446e+00,
-2.4297e+00, -2.1952e+00]],
[[-1.2032e+00, -1.7975e+00, -1.5509e+00, ..., -2.5606e+00,
-2.1339e+00, -1.6622e+00],
[-1.1948e+00, -1.6889e+00, -1.1715e+00, ..., -2.4094e+00,
-1.9343e+00, -1.5811e+00],
[-1.5577e+00, -1.8522e+00, -1.3625e+00, ..., -2.8284e+00,
-2.2540e+00, -1.7433e+00],
...,
[-2.2816e+00, -2.8371e+00, -2.6420e+00, ..., -3.9844e+00,
-3.2342e+00, -2.5317e+00],
[-1.5435e+00, -1.9592e+00, -1.9148e+00, ..., -2.7666e+00,
-2.3136e+00, -1.9115e+00],
[-1.0438e+00, -1.2550e+00, -1.3336e+00, ..., -1.8281e+00,
-1.5297e+00, -1.1849e+00]]],
[[[-4.0198e-02, 6.2143e-02, -1.9403e+00, ..., -9.4295e-01,
-5.2369e-01, -2.5562e-01],
[-2.3196e+00, -1.9516e+00, -5.3388e+00, ..., -4.3303e+00,
-1.9072e+00, -8.3181e-01],
[ 4.5062e-01, -3.5263e-01, -1.2996e+00, ..., -3.7364e-01,
-5.6910e-01, 1.0766e-01],
...,
[-1.8544e-01, -8.1990e-01, -1.3247e+00, ..., -1.2852e+00,
-8.9821e-01, -6.5686e-01],
[ 1.0775e+00, 8.6176e-01, 3.9168e-01, ..., 4.7257e-01,
7.6709e-01, 8.5059e-01],
[ 7.7130e-01, 6.3995e-01, 3.6346e-01, ..., 5.7998e-01,
8.3463e-01, 7.4810e-01]],
[[-3.4895e+00, -1.8842e+00, -6.6482e+00, ..., -7.5082e+00,
-6.6115e+00, -4.8533e+00],
[-8.9504e-01, 1.5797e+00, -4.6723e+00, ..., -5.8142e+00,
-5.8982e+00, -4.5189e+00],
[-5.0244e-01, 1.1073e+00, -7.8438e+00, ..., -8.4284e+00,
-8.2625e+00, -6.9461e+00],
...,
[-7.3737e-01, 1.6494e-01, -4.7030e+00, ..., -5.6760e+00,
-4.9347e+00, -4.1098e+00],
[ 1.7307e+00, 2.3157e+00, -3.4096e-01, ..., -1.3043e+00,
-1.4479e+00, -1.5300e+00],
[ 9.7264e-01, 1.2428e+00, -1.8742e+00, ..., -2.5856e+00,
-2.7232e+00, -2.6611e+00]],
[[-1.5908e+00, -2.5672e+00, -2.4973e+00, ..., -1.9708e+00,
-1.4067e+00, -1.5190e+00],
[-1.8980e+00, -2.1368e+00, -1.6166e+00, ..., -1.3946e+00,
-8.4628e-01, -9.6135e-01],
[-2.5493e+00, -2.2276e+00, -1.1102e+00, ..., -1.2832e+00,
-5.8391e-01, -2.5227e-01],
...,
[-7.6440e-01, -1.2414e+00, -5.6893e-01, ..., -5.7681e-01,
-4.6364e-01, -6.3944e-01],
[-1.1577e+00, -8.2162e-01, -5.7548e-01, ..., -2.8763e-01,
-7.4466e-02, -4.4673e-02],
[-1.2851e+00, -7.8272e-01, -4.2208e-01, ..., -3.5499e-01,
-2.9632e-01, -2.5891e-01]],
...,
[[-2.0413e+00, -1.0706e+00, -2.3396e+00, ..., -3.1243e+00,
-2.5735e+00, -1.8211e+00],
[-2.1101e+00, -2.6395e-01, -1.5983e+00, ..., -2.7180e+00,
-2.7185e+00, -2.1060e+00],
[-4.6973e-01, 2.3470e+00, 1.0808e+00, ..., 1.2053e-01,
2.1885e-01, 4.9627e-01],
...,
[-1.3574e+00, 1.1586e+00, -2.0604e-01, ..., -1.1987e+00,
-4.7506e-01, 5.4543e-02],
[-4.0638e-01, 1.5038e+00, 7.2380e-01, ..., 2.2326e-02,
1.8457e-01, 3.1828e-01],
[-1.5132e+00, -9.8468e-01, -1.4599e+00, ..., -2.3109e+00,
-1.3284e+00, -7.0640e-01]],
[[-2.6393e+00, -3.1546e+00, -4.4980e+00, ..., -2.7160e+00,
-2.2174e+00, -1.9928e+00],
[-1.9978e+00, -1.9543e+00, -5.1556e+00, ..., -3.2165e+00,
-3.0802e+00, -3.0116e+00],
[-2.9982e+00, -2.7024e+00, -6.8262e+00, ..., -4.6427e+00,
-4.6265e+00, -3.7098e+00],
...,
[-1.7447e+00, -7.4487e-01, -4.3587e+00, ..., -2.3165e+00,
-1.7451e+00, -1.3334e+00],
[-1.7038e+00, -8.8719e-01, -3.5697e+00, ..., -2.2264e+00,
-1.4991e+00, -9.0627e-01],
[-2.0505e+00, -1.3915e+00, -2.9706e+00, ..., -2.8116e+00,
-1.9097e+00, -1.6161e+00]],
[[-1.9754e+00, -2.7511e+00, -2.6749e+00, ..., -2.9365e+00,
-2.4073e+00, -1.8497e+00],
[-1.9795e+00, -2.7539e+00, -2.3855e+00, ..., -2.6768e+00,
-2.1092e+00, -1.7076e+00],
[-2.4667e+00, -3.1124e+00, -2.8040e+00, ..., -3.0813e+00,
-2.4021e+00, -1.8369e+00],
...,
[-2.3055e+00, -2.8749e+00, -2.6934e+00, ..., -2.9435e+00,
-2.3498e+00, -1.7734e+00],
[-1.5716e+00, -1.9948e+00, -1.9624e+00, ..., -2.0882e+00,
-1.7168e+00, -1.3677e+00],
[-1.0815e+00, -1.3015e+00, -1.3733e+00, ..., -1.3156e+00,
-1.1156e+00, -8.2197e-01]]],
[[[ 4.6377e-01, 9.3986e-01, 8.1352e-01, ..., -3.1308e-01,
-2.6708e-02, 3.8123e-02],
[-1.2496e+00, -3.7738e-01, -1.5826e+00, ..., -3.3818e+00,
-1.3302e+00, -4.4225e-01],
[ 5.3123e-01, 3.8766e-01, 2.9153e-01, ..., -1.7480e-01,
-3.0373e-01, 1.8798e-01],
...,
[-1.7364e-01, -8.5493e-01, -1.4939e+00, ..., -1.7539e+00,
-1.6840e+00, -1.1253e+00],
[ 1.0857e+00, 7.7746e-01, 2.9089e-01, ..., 4.0052e-02,
2.7913e-01, 4.0989e-01],
[ 7.0090e-01, 4.8680e-01, 1.6341e-01, ..., 2.0515e-01,
4.4029e-01, 4.7821e-01]],
[[-2.3601e+00, -1.1907e+00, -4.1117e+00, ..., -7.0104e+00,
-6.2263e+00, -4.6330e+00],
[-3.7294e-01, 1.5955e+00, -2.4987e+00, ..., -5.4105e+00,
-5.5842e+00, -4.3079e+00],
[ 3.1018e-01, 1.4013e+00, -4.9891e+00, ..., -7.9286e+00,
-7.9219e+00, -6.7366e+00],
...,
[-4.5272e-01, 6.4729e-01, -4.0331e+00, ..., -7.1305e+00,
-6.7133e+00, -5.2524e+00],
[ 2.1316e+00, 2.8307e+00, 4.5059e-01, ..., -2.3051e+00,
-2.8818e+00, -2.2957e+00],
[ 1.2382e+00, 1.5444e+00, -1.4240e+00, ..., -3.0335e+00,
-3.5398e+00, -3.0488e+00]],
[[-5.6310e-01, -1.5106e+00, -1.3557e+00, ..., -1.4712e+00,
-1.1052e+00, -1.4094e+00],
[-1.2180e+00, -1.5423e+00, -9.8703e-01, ..., -1.0507e+00,
-6.4352e-01, -9.1218e-01],
[-1.0402e+00, -8.9843e-01, -3.8211e-01, ..., -1.0904e+00,
-5.5194e-01, -4.3785e-01],
...,
[-9.3836e-01, -1.3640e+00, -5.0314e-01, ..., -1.0310e+00,
-8.6193e-01, -1.1791e+00],
[-1.2946e+00, -8.7513e-01, -5.2004e-01, ..., -5.8786e-01,
-2.7769e-01, -4.3998e-01],
[-1.3933e+00, -7.9593e-01, -3.5721e-01, ..., -4.9547e-01,
-4.3635e-01, -5.8852e-01]],
...,
[[-2.0963e+00, -1.2317e+00, -2.5669e+00, ..., -3.3752e+00,
-2.7520e+00, -1.9593e+00],
[-6.5099e-01, 8.9929e-01, -9.8534e-01, ..., -2.5152e+00,
-2.6338e+00, -2.1213e+00],
[-2.5593e-01, 2.1158e+00, 4.9324e-01, ..., -4.9069e-02,
5.8856e-02, 3.1089e-01],
...,
[-1.5622e+00, 1.2834e+00, -1.8170e-01, ..., -7.9151e-01,
-2.7800e-01, 2.3066e-01],
[-5.7169e-01, 1.5603e+00, 7.8312e-01, ..., 4.3229e-01,
4.1416e-01, 7.3114e-01],
[-1.6516e+00, -1.1113e+00, -1.6474e+00, ..., -1.9865e+00,
-1.0549e+00, -3.1158e-01]],
[[-1.2871e+00, -1.1979e+00, -2.4594e+00, ..., -2.0973e+00,
-1.7858e+00, -1.7269e+00],
[-6.1566e-01, -1.1242e-01, -3.1677e+00, ..., -2.7582e+00,
-2.8096e+00, -2.9062e+00],
[-2.0836e+00, -1.3964e+00, -5.5721e+00, ..., -4.4923e+00,
-4.5724e+00, -3.7797e+00],
...,
[-1.6128e+00, -4.3564e-01, -4.0779e+00, ..., -2.9240e+00,
-2.8209e+00, -2.4127e+00],
[-1.6251e+00, -6.5197e-01, -3.4077e+00, ..., -2.4744e+00,
-2.3108e+00, -1.7542e+00],
[-2.1009e+00, -1.3301e+00, -2.9554e+00, ..., -2.7835e+00,
-2.3922e+00, -2.1606e+00]],
[[-1.1414e+00, -1.7075e+00, -1.5390e+00, ..., -2.6466e+00,
-2.2050e+00, -1.7115e+00],
[-1.1401e+00, -1.6171e+00, -1.2240e+00, ..., -2.4138e+00,
-1.9394e+00, -1.5945e+00],
[-1.4822e+00, -1.7915e+00, -1.4283e+00, ..., -2.8016e+00,
-2.2299e+00, -1.7262e+00],
...,
[-2.3304e+00, -2.9114e+00, -2.6941e+00, ..., -3.9257e+00,
-3.1922e+00, -2.5087e+00],
[-1.5619e+00, -1.9922e+00, -1.9189e+00, ..., -2.7376e+00,
-2.2938e+00, -1.8909e+00],
[-1.0419e+00, -1.2497e+00, -1.3046e+00, ..., -1.8308e+00,
-1.5157e+00, -1.1752e+00]]],
[[[ 9.5655e-02, 3.0922e-01, -1.4562e+00, ..., 6.4586e-01,
9.2202e-01, 7.0252e-01],
[-2.2076e+00, -1.5928e+00, -4.8089e+00, ..., -1.4085e+00,
-9.9324e-02, 4.6937e-01],
[ 3.7426e-01, -3.9816e-01, -1.3523e+00, ..., 5.7352e-01,
9.4963e-01, 8.5444e-01],
...,
[-3.1939e-01, -1.3526e+00, -2.7623e+00, ..., -1.0676e+00,
-7.1464e-01, -5.4807e-01],
[ 1.2589e+00, 5.7642e-01, -1.1819e+00, ..., 5.8617e-01,
8.1601e-01, 8.5770e-01],
[ 7.7844e-01, 1.7407e-01, -6.6023e-01, ..., 6.5519e-01,
8.7210e-01, 7.5706e-01]],
[[-3.3449e+00, -1.5255e+00, -6.3194e+00, ..., -5.5311e+00,
-4.7400e+00, -3.6724e+00],
[-6.3442e-01, 2.1373e+00, -4.3439e+00, ..., -4.3071e+00,
-4.1723e+00, -3.3908e+00],
[-6.7781e-02, 1.8283e+00, -7.6118e+00, ..., -6.2986e+00,
-6.0221e+00, -5.5225e+00],
...,
[-1.3981e+00, 2.0400e-01, -6.7751e+00, ..., -5.4886e+00,
-4.7980e+00, -4.0015e+00],
[ 1.4224e+00, 2.5818e+00, -1.4869e+00, ..., -1.3553e+00,
-1.4898e+00, -1.5672e+00],
[ 9.5474e-01, 1.6500e+00, -2.4823e+00, ..., -2.5535e+00,
-2.6582e+00, -2.6278e+00]],
[[-1.5076e+00, -2.4408e+00, -2.2523e+00, ..., -8.8405e-01,
-6.6399e-01, -8.4687e-01],
[-2.1047e+00, -2.2351e+00, -1.5829e+00, ..., -7.1705e-01,
-4.4230e-01, -3.7921e-01],
[-2.6962e+00, -2.1211e+00, -8.8264e-01, ..., -4.0054e-01,
-9.3295e-02, -1.1483e-01],
...,
[-1.6883e+00, -1.8708e+00, -7.5230e-01, ..., -6.1994e-01,
-5.3810e-01, -7.1577e-01],
[-1.8402e+00, -1.2266e+00, -7.5653e-01, ..., -2.1434e-01,
-4.2279e-02, -4.4611e-02],
[-1.8210e+00, -1.2289e+00, -7.0749e-01, ..., -3.1853e-01,
-2.8067e-01, -2.4490e-01]],
...,
[[-2.3614e+00, -1.3227e+00, -2.7409e+00, ..., -3.1078e+00,
-2.4892e+00, -1.6781e+00],
[-2.0547e+00, 1.0534e-02, -1.4605e+00, ..., -1.4575e+00,
-1.5495e+00, -1.3447e+00],
[-5.6446e-01, 2.4368e+00, 9.0306e-01, ..., -1.2529e-01,
1.2369e-01, 2.8324e-01],
...,
[-1.7882e+00, 1.1364e+00, 2.7112e-01, ..., -1.0175e+00,
-3.5929e-01, 1.5948e-01],
[-8.6635e-01, 1.7318e+00, 1.6523e+00, ..., 1.3973e-01,
2.3122e-01, 3.2083e-01],
[-2.1023e+00, -1.0028e+00, -7.1299e-01, ..., -2.1192e+00,
-1.2249e+00, -6.1589e-01]],
[[-2.4710e+00, -2.7854e+00, -4.1229e+00, ..., -1.0973e+00,
-5.8966e-01, -4.8576e-01],
[-1.7589e+00, -1.4814e+00, -4.8537e+00, ..., -1.7434e+00,
-1.5444e+00, -1.8004e+00],
[-3.0722e+00, -2.5499e+00, -6.9436e+00, ..., -3.8353e+00,
-3.3673e+00, -2.7475e+00],
...,
[-2.6032e+00, -1.8542e+00, -5.0094e+00, ..., -2.2824e+00,
-1.8343e+00, -1.5179e+00],
[-2.2513e+00, -1.6323e+00, -3.9230e+00, ..., -2.1833e+00,
-1.5610e+00, -1.0644e+00],
[-2.3152e+00, -1.8302e+00, -3.0896e+00, ..., -2.7766e+00,
-1.9343e+00, -1.6703e+00]],
[[-1.9778e+00, -2.7256e+00, -2.5186e+00, ..., -1.7062e+00,
-1.4191e+00, -1.0656e+00],
[-2.0047e+00, -2.7523e+00, -2.2180e+00, ..., -1.4026e+00,
-1.1397e+00, -9.0449e-01],
[-2.5445e+00, -3.1408e+00, -2.6547e+00, ..., -1.6088e+00,
-1.2342e+00, -8.9924e-01],
...,
[-3.2040e+00, -4.0168e+00, -3.7839e+00, ..., -2.7957e+00,
-2.2324e+00, -1.6834e+00],
[-2.1355e+00, -2.7664e+00, -2.7308e+00, ..., -1.9697e+00,
-1.6108e+00, -1.2846e+00],
[-1.5437e+00, -1.8879e+00, -1.9353e+00, ..., -1.2288e+00,
-1.0508e+00, -7.6751e-01]]]], device='cuda:0'),
tensor([[[[ 2.8686e-01, 5.3797e-01, 6.7799e-01, ..., 4.7325e-01,
1.0236e+00, 6.1285e-01],
[-2.2051e+00, -1.5488e+00, -3.1960e+00, ..., -2.6856e+00,
-8.6812e-01, -1.8652e-01],
[-1.8087e-01, -9.8768e-01, -2.2269e-01, ..., -1.7387e-01,
5.4019e-01, 6.5049e-01],
...,
[-1.6949e-01, -1.1487e+00, -2.6515e+00, ..., -1.5281e+00,
-1.4221e+00, -8.8946e-01],
[ 1.4571e+00, 8.4812e-01, -9.4539e-01, ..., -3.0814e-01,
8.8561e-02, 3.5379e-01],
[ 8.6728e-01, 2.8884e-01, -6.9046e-01, ..., -1.8460e-01,
2.1783e-01, 3.6891e-01]],
[[-1.7077e+00, -1.1887e+00, -4.1740e+00, ..., -4.8744e+00,
-4.4608e+00, -3.3956e+00],
[ 5.5032e-01, 1.8979e+00, -2.2648e+00, ..., -3.5141e+00,
-3.6338e+00, -2.9870e+00],
[ 1.2864e+00, 1.6747e+00, -4.8908e+00, ..., -5.4816e+00,
-5.2950e+00, -4.9589e+00],
...,
[-1.2709e+00, 1.6232e-01, -6.7006e+00, ..., -7.5978e+00,
-6.8988e+00, -5.4102e+00],
[ 1.8039e+00, 2.8318e+00, -1.0368e+00, ..., -2.5874e+00,
-2.9399e+00, -2.3694e+00],
[ 1.2177e+00, 1.8166e+00, -2.2795e+00, ..., -3.2337e+00,
-3.5988e+00, -3.1243e+00]],
[[-1.1237e-01, -1.1188e+00, -1.2888e+00, ..., -4.9674e-01,
-1.5966e-01, -4.7903e-01],
[-5.6632e-01, -8.3268e-01, -7.5705e-01, ..., -2.8054e-01,
4.1758e-02, -6.5543e-02],
[-5.1090e-01, -3.3214e-01, -3.9632e-01, ..., -3.5700e-01,
7.2892e-02, 7.5485e-02],
...,
[-1.3900e+00, -1.7704e+00, -7.8546e-01, ..., -8.6922e-01,
-7.6298e-01, -1.0231e+00],
[-1.7332e+00, -1.1615e+00, -8.3443e-01, ..., -5.5655e-01,
-2.9392e-01, -3.6335e-01],
[-1.8773e+00, -1.2764e+00, -8.8312e-01, ..., -9.4990e-01,
-8.2030e-01, -8.0131e-01]],
...,
[[-3.0451e+00, -2.5311e+00, -2.8231e+00, ..., -3.4974e+00,
-2.6708e+00, -1.7709e+00],
[-2.2903e+00, -8.6752e-01, -1.6172e+00, ..., -2.6400e+00,
-2.3849e+00, -1.8741e+00],
[-1.1058e+00, 1.1310e+00, 6.0653e-01, ..., -4.1510e-01,
-2.3386e-02, 1.0937e-01],
...,
[-1.8239e+00, 1.3211e+00, 3.7104e-02, ..., -1.1022e+00,
-5.2103e-01, 1.9557e-03],
[-8.0463e-01, 1.9634e+00, 1.6723e+00, ..., 4.5752e-01,
4.1365e-01, 6.7053e-01],
[-2.2370e+00, -1.1557e+00, -9.9906e-01, ..., -1.8780e+00,
-9.7944e-01, -3.1551e-01]],
[[-1.5949e+00, -1.5744e+00, -2.8667e+00, ..., -1.2562e+00,
-8.3327e-01, -8.4579e-01],
[-6.0838e-01, 6.7331e-02, -3.2101e+00, ..., -1.3598e+00,
-1.2221e+00, -1.5854e+00],
[-2.1166e+00, -9.2502e-01, -5.3327e+00, ..., -3.1771e+00,
-2.7080e+00, -2.2148e+00],
...,
[-2.4676e+00, -1.5479e+00, -5.4686e+00, ..., -3.0806e+00,
-2.7121e+00, -2.1596e+00],
[-2.2069e+00, -1.4475e+00, -4.2585e+00, ..., -2.6778e+00,
-2.2943e+00, -1.6257e+00],
[-2.3797e+00, -1.7541e+00, -3.3172e+00, ..., -2.9679e+00,
-2.4224e+00, -2.1171e+00]],
[[-1.3457e+00, -1.9069e+00, -1.7085e+00, ..., -1.7701e+00,
-1.4602e+00, -1.1101e+00],
[-1.2680e+00, -1.7585e+00, -1.2362e+00, ..., -1.3708e+00,
-1.0787e+00, -8.5995e-01],
[-1.5873e+00, -1.9440e+00, -1.4868e+00, ..., -1.5985e+00,
-1.1931e+00, -8.8269e-01],
...,
[-3.1861e+00, -3.9802e+00, -3.7668e+00, ..., -3.9486e+00,
-3.1705e+00, -2.4423e+00],
[-2.1212e+00, -2.7388e+00, -2.7055e+00, ..., -2.7663e+00,
-2.2863e+00, -1.8601e+00],
[-1.5466e+00, -1.8738e+00, -1.9339e+00, ..., -1.8615e+00,
-1.5172e+00, -1.1603e+00]]],
[[[ 4.0887e-01, 8.6648e-01, 5.2482e-01, ..., 5.6377e-01,
8.6218e-01, 6.4465e-01],
[-1.2303e+00, -2.4681e-01, -1.6335e+00, ..., -1.2622e+00,
-1.1408e-02, 4.8495e-01],
[ 8.1731e-01, 6.2583e-01, 5.3921e-01, ..., 6.7011e-01,
9.2068e-01, 8.5663e-01],
...,
[-1.7336e-01, -1.2153e+00, -2.8087e+00, ..., -1.5796e+00,
-1.5357e+00, -8.7601e-01],
[ 1.4312e+00, 7.2198e-01, -1.3197e+00, ..., -5.1738e-01,
-1.1190e-01, 2.4103e-01],
[ 7.5258e-01, 6.0649e-02, -1.0390e+00, ..., -4.4524e-01,
2.5537e-02, 2.8032e-01]],
[[-2.5652e+00, -1.2921e+00, -4.4760e+00, ..., -5.4850e+00,
-4.6996e+00, -3.6535e+00],
[-3.2888e-01, 1.7208e+00, -2.7334e+00, ..., -4.3860e+00,
-4.1779e+00, -3.4050e+00],
[ 4.7614e-01, 1.5040e+00, -5.3924e+00, ..., -6.3182e+00,
-5.9636e+00, -5.4464e+00],
...,
[-1.4742e+00, 3.9512e-02, -7.2193e+00, ..., -8.1021e+00,
-7.4402e+00, -5.7621e+00],
[ 1.6559e+00, 2.7507e+00, -1.3960e+00, ..., -2.8736e+00,
-3.3201e+00, -2.5640e+00],
[ 1.1583e+00, 1.8467e+00, -2.4455e+00, ..., -3.3899e+00,
-3.8507e+00, -3.2404e+00]],
[[-7.6837e-01, -1.9163e+00, -1.7416e+00, ..., -9.6377e-01,
-6.7328e-01, -7.8857e-01],
[-1.4623e+00, -1.9106e+00, -1.2538e+00, ..., -7.0089e-01,
-3.8922e-01, -2.7855e-01],
[-1.3015e+00, -1.2134e+00, -5.2793e-01, ..., -3.7453e-01,
-2.1501e-02, 5.2305e-02],
...,
[-1.5528e+00, -1.8076e+00, -6.7317e-01, ..., -8.9079e-01,
-7.9725e-01, -1.0772e+00],
[-1.8616e+00, -1.2088e+00, -7.5451e-01, ..., -5.3994e-01,
-2.8518e-01, -3.7695e-01],
[-2.0425e+00, -1.4910e+00, -1.0668e+00, ..., -1.1765e+00,
-1.0484e+00, -1.0183e+00]],
...,
[[-2.1823e+00, -1.0399e+00, -2.3142e+00, ..., -2.8453e+00,
-2.2676e+00, -1.5286e+00],
[-6.4566e-01, 1.2574e+00, -4.8283e-01, ..., -1.4030e+00,
-1.4700e+00, -1.3074e+00],
[-2.3046e-01, 2.4721e+00, 9.1653e-01, ..., -1.1729e-01,
2.3944e-01, 3.5071e-01],
...,
[-2.0254e+00, 1.0646e+00, -1.3724e-01, ..., -1.1120e+00,
-5.9129e-01, -5.8848e-02],
[-8.5815e-01, 1.9510e+00, 1.7488e+00, ..., 6.5215e-01,
4.9665e-01, 7.5818e-01],
[-2.2835e+00, -1.1161e+00, -8.0870e-01, ..., -1.6963e+00,
-8.1897e-01, -1.7927e-01]],
[[-1.4583e+00, -1.4070e+00, -2.9461e+00, ..., -1.3134e+00,
-7.7733e-01, -6.0785e-01],
[-7.4002e-01, -2.3954e-01, -3.7734e+00, ..., -1.9829e+00,
-1.6550e+00, -1.7616e+00],
[-2.1550e+00, -1.4570e+00, -6.1643e+00, ..., -3.9072e+00,
-3.3353e+00, -2.6092e+00],
...,
[-2.6413e+00, -1.7963e+00, -5.5546e+00, ..., -3.1817e+00,
-2.8912e+00, -2.3113e+00],
[-2.3348e+00, -1.6486e+00, -4.3392e+00, ..., -2.7728e+00,
-2.4925e+00, -1.7975e+00],
[-2.4426e+00, -1.8955e+00, -3.3667e+00, ..., -3.0397e+00,
-2.5921e+00, -2.2591e+00]],
[[-1.1943e+00, -1.7813e+00, -1.5592e+00, ..., -1.6998e+00,
-1.4132e+00, -1.0664e+00],
[-1.1889e+00, -1.6884e+00, -1.2035e+00, ..., -1.4344e+00,
-1.1655e+00, -9.2410e-01],
[-1.5588e+00, -1.8639e+00, -1.3962e+00, ..., -1.6532e+00,
-1.2635e+00, -9.2446e-01],
...,
[-3.3698e+00, -4.2419e+00, -4.0324e+00, ..., -4.2188e+00,
-3.3880e+00, -2.6239e+00],
[-2.2322e+00, -2.9065e+00, -2.8812e+00, ..., -2.9453e+00,
-2.4321e+00, -1.9853e+00],
[-1.6388e+00, -2.0099e+00, -2.0669e+00, ..., -2.0030e+00,
-1.6137e+00, -1.2439e+00]]],
[[[ 1.2601e-01, 3.9260e-01, -1.2506e+00, ..., 4.3083e-01,
9.7114e-01, 6.3815e-01],
[-2.1456e+00, -1.4520e+00, -4.4991e+00, ..., -2.4885e+00,
-7.4375e-01, -6.0945e-02],
[ 3.9755e-01, -3.0405e-01, -1.1804e+00, ..., -2.5609e-01,
4.6799e-01, 6.2936e-01],
...,
[-2.6756e-01, -1.2924e+00, -2.6416e+00, ..., -2.1082e+00,
-1.5395e+00, -1.0837e+00],
[ 1.2722e+00, 6.5629e-01, -9.4677e-01, ..., -2.8098e-01,
5.0509e-01, 6.9503e-01],
[ 8.2421e-01, 2.9271e-01, -4.5321e-01, ..., -2.9505e-01,
3.2610e-01, 2.9749e-01]],
[[-3.3120e+00, -1.5175e+00, -6.1734e+00, ..., -4.7257e+00,
-4.3237e+00, -3.3156e+00],
[-6.5208e-01, 2.0762e+00, -4.2171e+00, ..., -3.3762e+00,
-3.5104e+00, -2.8898e+00],
[-6.5128e-02, 1.7710e+00, -7.4799e+00, ..., -5.2881e+00,
-5.1700e+00, -4.8411e+00],
...,
[-1.2438e+00, 3.0908e-01, -6.5452e+00, ..., -4.5756e+00,
-4.0999e+00, -3.3908e+00],
[ 1.5582e+00, 2.6453e+00, -1.3384e+00, ..., -5.7318e-01,
-9.3603e-01, -1.0270e+00],
[ 1.0319e+00, 1.6638e+00, -2.4129e+00, ..., -2.1589e+00,
-2.4165e+00, -2.3503e+00]],
[[-1.3769e+00, -2.3430e+00, -2.2067e+00, ..., -5.5625e-01,
-1.7532e-01, -4.3104e-01],
[-1.9882e+00, -2.1562e+00, -1.5769e+00, ..., -3.1791e-01,
6.2923e-02, 3.6853e-02],
[-2.4961e+00, -1.9505e+00, -8.3135e-01, ..., -2.7368e-01,
2.2914e-01, 2.7359e-01],
...,
[-1.6212e+00, -1.8574e+00, -7.7976e-01, ..., -5.3266e-01,
-1.9400e-01, -4.0151e-01],
[-1.7897e+00, -1.2080e+00, -7.7945e-01, ..., -3.0941e-01,
1.8568e-01, 2.4211e-01],
[-1.7561e+00, -1.1468e+00, -6.2434e-01, ..., -4.9422e-01,
-2.5897e-01, -2.3777e-01]],
...,
[[-2.3097e+00, -1.3164e+00, -2.7350e+00, ..., -3.4661e+00,
-2.6789e+00, -1.7500e+00],
[-1.8792e+00, 1.5121e-01, -1.3053e+00, ..., -2.5522e+00,
-2.3064e+00, -1.7806e+00],
[-5.4016e-01, 2.4069e+00, 9.2508e-01, ..., -5.7102e-01,
-6.2967e-02, 1.5185e-01],
...,
[-1.7505e+00, 1.2079e+00, 2.4035e-01, ..., -1.6429e+00,
-6.8152e-01, -7.8948e-02],
[-8.6807e-01, 1.6936e+00, 1.5148e+00, ..., -1.4119e-01,
1.4968e-01, 3.7905e-01],
[-2.1039e+00, -1.0241e+00, -8.0453e-01, ..., -2.5811e+00,
-1.5567e+00, -9.6986e-01]],
[[-2.3784e+00, -2.6442e+00, -3.9268e+00, ..., -1.1876e+00,
-7.6824e-01, -7.7943e-01],
[-1.6583e+00, -1.3276e+00, -4.6792e+00, ..., -1.4106e+00,
-1.2281e+00, -1.5619e+00],
[-3.0003e+00, -2.4412e+00, -6.8644e+00, ..., -3.2628e+00,
-2.7492e+00, -2.1859e+00],
...,
[-2.5309e+00, -1.7344e+00, -5.0210e+00, ..., -1.3771e+00,
-9.3463e-01, -7.8051e-01],
[-2.2193e+00, -1.5431e+00, -3.8930e+00, ..., -1.6985e+00,
-1.0716e+00, -6.6059e-01],
[-2.3110e+00, -1.7609e+00, -3.0604e+00, ..., -2.8779e+00,
-2.0185e+00, -1.8533e+00]],
[[-1.9092e+00, -2.6537e+00, -2.4485e+00, ..., -1.7673e+00,
-1.4657e+00, -1.1226e+00],
[-1.9373e+00, -2.6515e+00, -2.1402e+00, ..., -1.4012e+00,
-1.1063e+00, -8.8715e-01],
[-2.4548e+00, -3.0109e+00, -2.5444e+00, ..., -1.6150e+00,
-1.2147e+00, -9.0344e-01],
...,
[-3.1221e+00, -3.9018e+00, -3.6636e+00, ..., -2.9534e+00,
-2.3675e+00, -1.8194e+00],
[-2.0763e+00, -2.6867e+00, -2.6477e+00, ..., -2.0187e+00,
-1.6627e+00, -1.3445e+00],
[-1.4922e+00, -1.8200e+00, -1.8668e+00, ..., -1.2213e+00,
-1.0471e+00, -7.9839e-01]]],
[[[ 1.3109e-01, 3.5431e-01, -1.2380e+00, ..., 6.5460e-01,
9.2770e-01, 6.7323e-01],
[-2.1377e+00, -1.4955e+00, -4.4354e+00, ..., -1.5142e+00,
-1.9058e-01, 3.7328e-01],
[ 4.0638e-01, -3.5664e-01, -1.1680e+00, ..., 5.0197e-01,
8.9991e-01, 8.1583e-01],
...,
[-2.7927e-01, -1.2728e+00, -2.5946e+00, ..., -1.1613e+00,
-8.0298e-01, -6.1896e-01],
[ 1.2585e+00, 6.6548e-01, -9.3684e-01, ..., 5.1210e-01,
7.6889e-01, 8.1087e-01],
[ 8.1400e-01, 2.9658e-01, -4.4897e-01, ..., 5.5501e-01,
8.0192e-01, 7.0707e-01]],
[[-3.2482e+00, -1.4467e+00, -6.1605e+00, ..., -5.4828e+00,
-4.6890e+00, -3.6452e+00],
[-5.5657e-01, 2.1782e+00, -4.2061e+00, ..., -4.2114e+00,
-4.0730e+00, -3.3472e+00],
[ 3.0432e-02, 1.8525e+00, -7.4365e+00, ..., -6.1395e+00,
-5.8658e+00, -5.4728e+00],
...,
[-1.2668e+00, 2.7328e-01, -6.5713e+00, ..., -5.3871e+00,
-4.7475e+00, -3.9618e+00],
[ 1.5235e+00, 2.6182e+00, -1.3708e+00, ..., -1.2643e+00,
-1.4265e+00, -1.5119e+00],
[ 1.0071e+00, 1.6465e+00, -2.4169e+00, ..., -2.4948e+00,
-2.6272e+00, -2.5802e+00]],
[[-1.4122e+00, -2.3624e+00, -2.1792e+00, ..., -8.5211e-01,
-6.3230e-01, -8.1838e-01],
[-2.0426e+00, -2.1891e+00, -1.5375e+00, ..., -7.1139e-01,
-4.3708e-01, -3.7299e-01],
[-2.5759e+00, -2.0228e+00, -8.2964e-01, ..., -4.0340e-01,
-1.0899e-01, -1.0145e-01],
...,
[-1.6233e+00, -1.8495e+00, -7.7466e-01, ..., -6.3739e-01,
-5.5991e-01, -7.1312e-01],
[-1.7922e+00, -1.2211e+00, -7.8740e-01, ..., -2.2881e-01,
-6.0166e-02, -5.5229e-02],
[-1.7555e+00, -1.1448e+00, -6.2479e-01, ..., -3.2879e-01,
-2.9059e-01, -2.6059e-01]],
...,
[[-2.3728e+00, -1.3088e+00, -2.7646e+00, ..., -3.0403e+00,
-2.4310e+00, -1.6353e+00],
[-1.9063e+00, 1.7031e-01, -1.3970e+00, ..., -1.4438e+00,
-1.5330e+00, -1.3339e+00],
[-4.7994e-01, 2.5068e+00, 9.0797e-01, ..., -9.0972e-02,
2.6704e-01, 3.4134e-01],
...,
[-1.7484e+00, 1.1550e+00, 1.6926e-01, ..., -1.0092e+00,
-3.0968e-01, 1.7335e-01],
[-8.6764e-01, 1.6723e+00, 1.4945e+00, ..., 1.6585e-01,
2.9987e-01, 3.9824e-01],
[-2.0940e+00, -1.0411e+00, -8.2663e-01, ..., -2.1433e+00,
-1.2401e+00, -6.3727e-01]],
[[-2.3851e+00, -2.6490e+00, -4.0248e+00, ..., -1.1244e+00,
-5.9439e-01, -4.7187e-01],
[-1.6597e+00, -1.3402e+00, -4.7648e+00, ..., -1.6867e+00,
-1.4591e+00, -1.7051e+00],
[-2.9980e+00, -2.4402e+00, -6.8779e+00, ..., -3.7439e+00,
-3.2423e+00, -2.6236e+00],
...,
[-2.5425e+00, -1.7609e+00, -5.0047e+00, ..., -2.1925e+00,
-1.7657e+00, -1.4246e+00],
[-2.2230e+00, -1.5701e+00, -3.9082e+00, ..., -2.1340e+00,
-1.5070e+00, -9.8288e-01],
[-2.3080e+00, -1.7842e+00, -3.0912e+00, ..., -2.8105e+00,
-1.9528e+00, -1.6610e+00]],
[[-1.9155e+00, -2.6479e+00, -2.4351e+00, ..., -1.7058e+00,
-1.4206e+00, -1.0605e+00],
[-1.9414e+00, -2.6663e+00, -2.1290e+00, ..., -1.3910e+00,
-1.1354e+00, -8.8722e-01],
[-2.4710e+00, -3.0430e+00, -2.5510e+00, ..., -1.6006e+00,
-1.2238e+00, -8.8194e-01],
...,
[-3.1308e+00, -3.9162e+00, -3.6809e+00, ..., -2.7839e+00,
-2.2179e+00, -1.6781e+00],
[-2.0847e+00, -2.6970e+00, -2.6528e+00, ..., -1.9512e+00,
-1.6005e+00, -1.2741e+00],
[-1.5022e+00, -1.8237e+00, -1.8656e+00, ..., -1.2183e+00,
-1.0364e+00, -7.6477e-01]]],
[[[ 2.8455e-01, 5.5332e-01, 6.9708e-01, ..., 4.5823e-01,
9.8847e-01, 5.9263e-01],
[-2.1293e+00, -1.4636e+00, -3.0889e+00, ..., -2.6774e+00,
-8.8307e-01, -2.0163e-01],
[-1.7234e-01, -9.9212e-01, -2.3576e-01, ..., -1.9852e-01,
4.9916e-01, 6.3372e-01],
...,
[-2.6360e+00, -3.8029e+00, -6.5935e+00, ..., -1.5102e+00,
-1.4404e+00, -8.7559e-01],
[ 2.8259e-01, -7.1532e-01, -2.7970e+00, ..., -2.9944e-01,
7.0920e-02, 3.6609e-01],
[ 7.3264e-01, 7.7325e-02, -7.9997e-01, ..., -2.0376e-01,
1.8652e-01, 3.7112e-01]],
[[-1.6465e+00, -1.0966e+00, -4.1270e+00, ..., -4.8691e+00,
-4.4026e+00, -3.4091e+00],
[ 5.6518e-01, 1.9174e+00, -2.3088e+00, ..., -3.5014e+00,
-3.5688e+00, -3.0097e+00],
[ 1.2831e+00, 1.6596e+00, -4.9481e+00, ..., -5.4173e+00,
-5.2189e+00, -4.9701e+00],
...,
[-8.9056e-01, 1.4073e-01, -6.6532e+00, ..., -7.6728e+00,
-6.9561e+00, -5.4543e+00],
[ 1.9386e+00, 3.0110e+00, -8.5492e-01, ..., -2.6299e+00,
-2.9901e+00, -2.3885e+00],
[ 1.2047e+00, 1.7909e+00, -2.3098e+00, ..., -3.2494e+00,
-3.6426e+00, -3.1325e+00]],
[[-1.0836e-01, -1.1343e+00, -1.2866e+00, ..., -5.0198e-01,
-1.7967e-01, -4.7782e-01],
[-5.8301e-01, -8.4853e-01, -7.4911e-01, ..., -2.7553e-01,
3.7326e-03, -9.2007e-02],
[-5.4337e-01, -3.6614e-01, -4.0235e-01, ..., -3.2738e-01,
5.2655e-02, 6.3630e-02],
...,
[-1.7565e+00, -1.8301e+00, -2.9552e-01, ..., -8.9352e-01,
-7.8806e-01, -1.0302e+00],
[-2.2764e+00, -1.6069e+00, -9.0528e-01, ..., -5.5477e-01,
-2.9155e-01, -3.6281e-01],
[-1.9914e+00, -1.3792e+00, -9.1075e-01, ..., -9.7132e-01,
-8.3094e-01, -8.0894e-01]],
...,
[[-2.9801e+00, -2.4160e+00, -2.7366e+00, ..., -3.3869e+00,
-2.5830e+00, -1.7543e+00],
[-2.2391e+00, -8.0075e-01, -1.5760e+00, ..., -2.5552e+00,
-2.2575e+00, -1.7990e+00],
[-1.0115e+00, 1.2137e+00, 6.2699e-01, ..., -4.2634e-01,
4.2152e-02, 1.4412e-01],
...,
[-1.7896e+00, -6.8599e-02, -2.8304e+00, ..., -1.0706e+00,
-5.0559e-01, 2.0137e-02],
[-1.0596e-01, 2.4980e+00, 2.0649e+00, ..., 4.9337e-01,
4.3329e-01, 7.0089e-01],
[-2.1897e+00, -1.1221e+00, -9.6573e-01, ..., -1.8603e+00,
-9.5416e-01, -2.9500e-01]],
[[-1.5857e+00, -1.5874e+00, -2.9182e+00, ..., -1.2482e+00,
-8.0882e-01, -8.2580e-01],
[-6.2140e-01, 3.2722e-02, -3.2452e+00, ..., -1.3588e+00,
-1.1917e+00, -1.5634e+00],
[-2.0988e+00, -9.1451e-01, -5.3171e+00, ..., -3.1717e+00,
-2.6893e+00, -2.2093e+00],
...,
[-2.3733e+00, -2.5599e+00, -6.1095e+00, ..., -3.0931e+00,
-2.7379e+00, -2.1870e+00],
[-2.2319e+00, -2.0185e+00, -4.7688e+00, ..., -2.6843e+00,
-2.3117e+00, -1.6481e+00],
[-2.3512e+00, -1.7963e+00, -3.3369e+00, ..., -2.9620e+00,
-2.4465e+00, -2.1328e+00]],
[[-1.3438e+00, -1.8963e+00, -1.6984e+00, ..., -1.7760e+00,
-1.4650e+00, -1.1093e+00],
[-1.2738e+00, -1.7651e+00, -1.2428e+00, ..., -1.3944e+00,
-1.1012e+00, -8.6522e-01],
[-1.6051e+00, -1.9653e+00, -1.5035e+00, ..., -1.6179e+00,
-1.2267e+00, -8.9349e-01],
...,
[-3.6500e+00, -4.7005e+00, -4.7043e+00, ..., -3.9675e+00,
-3.1852e+00, -2.4555e+00],
[-2.3024e+00, -3.0660e+00, -3.1498e+00, ..., -2.7854e+00,
-2.3028e+00, -1.8717e+00],
[-1.5880e+00, -1.9246e+00, -1.9771e+00, ..., -1.8863e+00,
-1.5277e+00, -1.1736e+00]]]], device='cuda:0'),
tensor([[[[ 3.8094e-01, 8.3918e-01, 6.8240e-01, ..., 6.8934e-01,
9.5578e-01, 7.3645e-01],
[-1.1754e+00, -2.4232e-01, -1.3684e+00, ..., -1.2135e+00,
1.0503e-02, 5.6124e-01],
[ 6.9021e-01, 5.4896e-01, 5.7732e-01, ..., 6.6958e-01,
9.7432e-01, 9.0078e-01],
...,
[-2.1474e-01, -1.1797e+00, -2.6645e+00, ..., -1.4730e+00,
-1.4417e+00, -8.5974e-01],
[ 1.4534e+00, 8.6760e-01, -9.7994e-01, ..., -3.8355e-01,
-5.6318e-03, 2.9309e-01],
[ 8.1357e-01, 2.3413e-01, -7.5923e-01, ..., -2.7570e-01,
1.4295e-01, 3.3214e-01]],
[[-2.4435e+00, -1.3772e+00, -4.3585e+00, ..., -5.5094e+00,
-4.6923e+00, -3.6677e+00],
[-3.3415e-01, 1.4581e+00, -2.7382e+00, ..., -4.1676e+00,
-4.0682e+00, -3.3489e+00],
[ 4.4445e-01, 1.2482e+00, -5.2546e+00, ..., -6.0972e+00,
-5.9198e+00, -5.4869e+00],
...,
[-1.3625e+00, 1.1944e-01, -6.9949e+00, ..., -7.8250e+00,
-7.1336e+00, -5.5569e+00],
[ 1.7068e+00, 2.7772e+00, -1.2942e+00, ..., -2.7563e+00,
-3.1459e+00, -2.4597e+00],
[ 1.1650e+00, 1.8159e+00, -2.4129e+00, ..., -3.2841e+00,
-3.6996e+00, -3.1420e+00]],
[[-6.8842e-01, -1.7648e+00, -1.6269e+00, ..., -9.8352e-01,
-6.9325e-01, -8.2781e-01],
[-1.3435e+00, -1.7791e+00, -1.2079e+00, ..., -7.8208e-01,
-4.2540e-01, -2.9293e-01],
[-1.0597e+00, -9.9002e-01, -4.6101e-01, ..., -3.4805e-01,
3.0143e-02, 6.4609e-02],
...,
[-1.4429e+00, -1.7820e+00, -7.3040e-01, ..., -8.6295e-01,
-7.5691e-01, -1.0503e+00],
[-1.7587e+00, -1.1896e+00, -8.2219e-01, ..., -5.5891e-01,
-2.7749e-01, -3.6187e-01],
[-1.9395e+00, -1.3766e+00, -9.6898e-01, ..., -1.0341e+00,
-9.0390e-01, -8.8461e-01]],
...,
[[-2.1093e+00, -1.1864e+00, -2.2916e+00, ..., -3.1481e+00,
-2.4360e+00, -1.6028e+00],
[-5.5338e-01, 1.0517e+00, -4.9853e-01, ..., -1.4710e+00,
-1.5083e+00, -1.3080e+00],
[-3.3219e-01, 2.0071e+00, 6.9012e-01, ..., -2.1671e-01,
2.3522e-01, 4.0001e-01],
...,
[-1.8837e+00, 1.1932e+00, -5.2642e-02, ..., -1.1088e+00,
-5.5280e-01, -2.1559e-03],
[-8.2752e-01, 1.9307e+00, 1.6665e+00, ..., 5.4267e-01,
4.3334e-01, 7.0365e-01],
[-2.2194e+00, -1.1067e+00, -9.0027e-01, ..., -1.8033e+00,
-9.0617e-01, -2.6510e-01]],
[[-1.3808e+00, -1.3155e+00, -2.6882e+00, ..., -1.0975e+00,
-5.5634e-01, -4.2687e-01],
[-7.0647e-01, -2.6752e-01, -3.5062e+00, ..., -1.7744e+00,
-1.5229e+00, -1.7384e+00],
[-2.1168e+00, -1.5020e+00, -5.8649e+00, ..., -3.8627e+00,
-3.3404e+00, -2.6523e+00],
...,
[-2.5437e+00, -1.7119e+00, -5.5739e+00, ..., -3.1273e+00,
-2.7823e+00, -2.2222e+00],
[-2.2422e+00, -1.5863e+00, -4.3630e+00, ..., -2.7397e+00,
-2.3839e+00, -1.6886e+00],
[-2.3738e+00, -1.8619e+00, -3.3845e+00, ..., -3.0020e+00,
-2.4833e+00, -2.1493e+00]],
[[-1.1371e+00, -1.6924e+00, -1.5095e+00, ..., -1.7121e+00,
-1.4242e+00, -1.0688e+00],
[-1.1445e+00, -1.6140e+00, -1.1971e+00, ..., -1.3913e+00,
-1.1324e+00, -8.9880e-01],
[-1.4890e+00, -1.7786e+00, -1.3804e+00, ..., -1.5630e+00,
-1.1931e+00, -8.6896e-01],
...,
[-3.2719e+00, -4.0869e+00, -3.8719e+00, ..., -4.0621e+00,
-3.2615e+00, -2.5236e+00],
[-2.1848e+00, -2.8198e+00, -2.7849e+00, ..., -2.8364e+00,
-2.3414e+00, -1.9116e+00],
[-1.6031e+00, -1.9453e+00, -1.9956e+00, ..., -1.9212e+00,
-1.5579e+00, -1.1959e+00]]],
[[[ 2.1830e-01, 4.8802e-01, 6.3050e-01, ..., 5.9202e-01,
9.7830e-01, 7.8476e-01],
[-1.9992e+00, -1.3084e+00, -2.8360e+00, ..., -1.3359e+00,
9.4805e-02, 6.3797e-01],
[-1.7742e-01, -9.1804e-01, -1.7657e-02, ..., 7.3047e-02,
9.6783e-01, 9.6099e-01],
...,
[-1.0052e+00, -2.2247e+00, -2.1396e+00, ..., -1.3872e+00,
-9.2082e-01, -6.5945e-01],
[ 7.8089e-01, -1.4577e-01, -1.7771e-01, ..., 4.3743e-01,
7.9979e-01, 9.0653e-01],
[ 1.9124e-01, -5.8618e-01, -4.5036e-01, ..., 6.2211e-01,
8.8023e-01, 8.0081e-01]],
[[-1.6501e+00, -1.3805e+00, -4.2417e+00, ..., -5.7499e+00,
-4.8847e+00, -3.7963e+00],
[ 4.6582e-01, 1.3906e+00, -2.5438e+00, ..., -4.7386e+00,
-4.4425e+00, -3.6286e+00],
[ 1.1543e+00, 1.0576e+00, -5.0308e+00, ..., -6.9847e+00,
-6.3915e+00, -5.7773e+00],
...,
[ 3.9269e-01, 3.7946e-01, -4.0865e+00, ..., -6.0209e+00,
-5.2061e+00, -4.2796e+00],
[ 2.5619e+00, 2.5210e+00, 1.8134e-01, ..., -1.3188e+00,
-1.4464e+00, -1.5020e+00],
[ 1.5263e+00, 1.3419e+00, -1.5944e+00, ..., -2.6939e+00,
-2.8009e+00, -2.7585e+00]],
[[-1.2251e-01, -1.1452e+00, -1.3914e+00, ..., -7.9173e-01,
-5.3131e-01, -6.6422e-01],
[-4.8735e-01, -7.9622e-01, -8.8303e-01, ..., -6.8403e-01,
-3.5586e-01, -2.2386e-01],
[-3.4044e-01, -1.8309e-01, -4.4661e-01, ..., -4.4802e-01,
-8.0870e-03, 9.7669e-02],
...,
[-4.7745e-02, -6.1177e-01, -6.1031e-01, ..., -6.7219e-01,
-4.8416e-01, -5.7876e-01],
[-7.0771e-01, -2.3892e-01, -6.0123e-01, ..., -2.9070e-01,
3.1775e-03, 9.3380e-02],
[-8.9061e-01, -2.8362e-01, -5.1133e-01, ..., -2.7595e-01,
-1.9607e-01, -1.3347e-01]],
...,
[[-2.7996e+00, -2.4248e+00, -2.4105e+00, ..., -3.1359e+00,
-2.5927e+00, -1.7289e+00],
[-2.1087e+00, -8.7817e-01, -1.1062e+00, ..., -1.6558e+00,
-1.7472e+00, -1.4525e+00],
[-1.0946e+00, 7.4693e-01, 7.7648e-01, ..., -4.6985e-01,
-3.6486e-02, 1.4206e-01],
...,
[-2.3055e+00, -1.7595e-01, -1.2220e-01, ..., -1.0748e+00,
-3.6484e-01, 1.7902e-01],
[-1.2691e+00, 8.8661e-01, 1.1267e+00, ..., 2.3781e-01,
3.7389e-01, 4.5047e-01],
[-2.1719e+00, -1.4879e+00, -1.6455e+00, ..., -2.2880e+00,
-1.2744e+00, -6.3984e-01]],
[[-1.6468e+00, -1.6605e+00, -2.7852e+00, ..., -1.3206e+00,
-7.2307e-01, -5.2054e-01],
[-7.2638e-01, -1.2249e-01, -3.0879e+00, ..., -1.9767e+00,
-1.6182e+00, -1.7051e+00],
[-2.0743e+00, -9.9153e-01, -4.9741e+00, ..., -3.9185e+00,
-3.3451e+00, -2.5671e+00],
...,
[-1.3196e+00, 2.3425e-01, -3.3598e+00, ..., -2.3906e+00,
-1.8003e+00, -1.3019e+00],
[-1.6565e+00, -4.3237e-01, -3.0122e+00, ..., -2.2972e+00,
-1.5085e+00, -8.3694e-01],
[-2.3182e+00, -1.3407e+00, -3.0276e+00, ..., -2.9284e+00,
-1.9372e+00, -1.6120e+00]],
[[-1.3023e+00, -1.8334e+00, -1.6594e+00, ..., -1.8494e+00,
-1.4949e+00, -1.0974e+00],
[-1.2454e+00, -1.7105e+00, -1.2426e+00, ..., -1.6296e+00,
-1.2779e+00, -9.7420e-01],
[-1.5623e+00, -1.9044e+00, -1.5059e+00, ..., -1.9343e+00,
-1.4272e+00, -1.0038e+00],
...,
[-2.4518e+00, -3.0707e+00, -2.7914e+00, ..., -3.1827e+00,
-2.4946e+00, -1.8518e+00],
[-1.5872e+00, -2.0929e+00, -1.9893e+00, ..., -2.2366e+00,
-1.8069e+00, -1.4145e+00],
[-1.0379e+00, -1.3515e+00, -1.3697e+00, ..., -1.3739e+00,
-1.1557e+00, -8.3550e-01]]],
[[[ 4.8117e-01, 9.6192e-01, 8.6089e-01, ..., -3.5794e-01,
-8.9385e-02, 1.4520e-02],
[-1.2112e+00, -3.2989e-01, -1.5019e+00, ..., -3.2323e+00,
-1.2826e+00, -4.4345e-01],
[ 5.8886e-01, 4.3853e-01, 3.4785e-01, ..., -2.1983e-01,
-3.8347e-01, 1.4757e-01],
...,
[-1.6422e-01, -8.5243e-01, -1.5415e+00, ..., -1.8577e+00,
-1.8791e+00, -1.3296e+00],
[ 1.0930e+00, 7.8809e-01, 2.7901e-01, ..., 3.6857e-02,
2.6198e-01, 3.7375e-01],
[ 7.2949e-01, 5.3042e-01, 1.8940e-01, ..., 2.0375e-01,
4.4733e-01, 4.4115e-01]],
[[-2.3083e+00, -1.1430e+00, -4.1272e+00, ..., -6.8389e+00,
-6.0501e+00, -4.5263e+00],
[-3.1541e-01, 1.6571e+00, -2.5196e+00, ..., -5.4974e+00,
-5.5875e+00, -4.3401e+00],
[ 3.7706e-01, 1.4737e+00, -5.0279e+00, ..., -7.9117e+00,
-7.8486e+00, -6.6310e+00],
...,
[-4.3591e-01, 6.7269e-01, -4.0363e+00, ..., -7.1774e+00,
-6.7082e+00, -5.1238e+00],
[ 2.1509e+00, 2.8461e+00, 4.3378e-01, ..., -2.0268e+00,
-2.5700e+00, -1.9791e+00],
[ 1.2508e+00, 1.5556e+00, -1.4533e+00, ..., -2.9545e+00,
-3.4326e+00, -2.9853e+00]],
[[-5.7509e-01, -1.5514e+00, -1.3699e+00, ..., -1.3823e+00,
-1.0472e+00, -1.3407e+00],
[-1.2669e+00, -1.6154e+00, -9.9838e-01, ..., -9.7710e-01,
-6.1174e-01, -9.0062e-01],
[-1.0970e+00, -9.7403e-01, -3.8896e-01, ..., -1.1063e+00,
-6.1615e-01, -4.8443e-01],
...,
[-9.4940e-01, -1.3860e+00, -5.3222e-01, ..., -9.9944e-01,
-8.6081e-01, -1.2630e+00],
[-1.3046e+00, -8.9368e-01, -5.5371e-01, ..., -5.0042e-01,
-2.1230e-01, -4.3625e-01],
[-1.3975e+00, -7.9269e-01, -3.6712e-01, ..., -4.6390e-01,
-4.0862e-01, -6.1711e-01]],
...,
[[-2.1502e+00, -1.2229e+00, -2.5351e+00, ..., -3.0259e+00,
-2.4831e+00, -1.7853e+00],
[-6.9063e-01, 9.4012e-01, -9.4931e-01, ..., -2.4486e+00,
-2.5316e+00, -2.0313e+00],
[-2.7009e-01, 2.1790e+00, 4.7084e-01, ..., -4.6023e-02,
2.2775e-02, 2.5086e-01],
...,
[-1.5166e+00, 1.3168e+00, -1.3896e-01, ..., -8.6190e-01,
-1.9542e-01, 3.2170e-01],
[-5.0376e-01, 1.6427e+00, 8.7625e-01, ..., 4.1055e-01,
4.7413e-01, 8.5002e-01],
[-1.6055e+00, -1.0473e+00, -1.5483e+00, ..., -2.2233e+00,
-1.1976e+00, -3.5186e-01]],
[[-1.3052e+00, -1.2131e+00, -2.5666e+00, ..., -2.2562e+00,
-1.9599e+00, -1.8661e+00],
[-6.3653e-01, -1.2791e-01, -3.2881e+00, ..., -2.8859e+00,
-2.8922e+00, -2.9137e+00],
[-2.0994e+00, -1.4112e+00, -5.6965e+00, ..., -4.4284e+00,
-4.4932e+00, -3.7063e+00],
...,
[-1.6128e+00, -4.4813e-01, -4.1006e+00, ..., -2.8622e+00,
-2.7388e+00, -2.2664e+00],
[-1.6382e+00, -6.8357e-01, -3.4243e+00, ..., -2.4584e+00,
-2.2171e+00, -1.6446e+00],
[-2.1066e+00, -1.3431e+00, -2.9460e+00, ..., -2.8431e+00,
-2.3795e+00, -2.2222e+00]],
[[-1.1411e+00, -1.7009e+00, -1.5258e+00, ..., -2.5724e+00,
-2.1441e+00, -1.6651e+00],
[-1.1457e+00, -1.6155e+00, -1.2080e+00, ..., -2.4128e+00,
-1.9429e+00, -1.5975e+00],
[-1.4931e+00, -1.7953e+00, -1.4046e+00, ..., -2.8399e+00,
-2.2710e+00, -1.7660e+00],
...,
[-2.3178e+00, -2.8905e+00, -2.6695e+00, ..., -4.0386e+00,
-3.2880e+00, -2.5649e+00],
[-1.5507e+00, -1.9731e+00, -1.8992e+00, ..., -2.7945e+00,
-2.3509e+00, -1.9354e+00],
[-1.0362e+00, -1.2438e+00, -1.2971e+00, ..., -1.8520e+00,
-1.5419e+00, -1.1923e+00]]],
[[[ 7.2620e-02, 2.1907e-01, -1.3890e+00, ..., -6.2576e-01,
-1.9665e-01, -4.8176e-02],
[-2.0568e+00, -1.5820e+00, -4.4839e+00, ..., -3.7748e+00,
-1.5351e+00, -5.5611e-01],
[ 5.9598e-01, -7.4761e-02, -8.1243e-01, ..., -2.0549e-01,
-2.5816e-01, 2.4123e-01],
...,
[-8.7837e-01, -1.8587e+00, -1.8613e+00, ..., -1.8879e+00,
-1.3480e+00, -9.6390e-01],
[ 7.3574e-01, -9.5565e-03, -1.0769e-01, ..., -5.1165e-02,
7.0715e-01, 7.9580e-01],
[ 1.1559e-01, -5.9162e-01, -4.8143e-01, ..., -1.7935e-01,
4.3109e-01, 3.3208e-01]],
[[-3.3320e+00, -1.8196e+00, -6.2708e+00, ..., -7.1144e+00,
-6.2437e+00, -4.6132e+00],
[-8.2760e-01, 1.5119e+00, -4.3597e+00, ..., -5.4321e+00,
-5.5287e+00, -4.2642e+00],
[-3.8640e-01, 1.0527e+00, -7.4053e+00, ..., -7.9341e+00,
-7.8257e+00, -6.6712e+00],
...,
[ 3.0560e-01, 3.5882e-01, -4.0436e+00, ..., -4.6196e+00,
-4.2037e+00, -3.4775e+00],
[ 2.2911e+00, 2.2591e+00, -1.9714e-01, ..., -7.8221e-01,
-1.1721e+00, -1.1985e+00],
[ 1.3428e+00, 1.1613e+00, -1.7870e+00, ..., -2.2288e+00,
-2.5130e+00, -2.3907e+00]],
[[-1.3063e+00, -2.3104e+00, -2.3055e+00, ..., -1.7501e+00,
-1.2345e+00, -1.3730e+00],
[-1.6621e+00, -1.9733e+00, -1.5359e+00, ..., -1.2536e+00,
-7.2628e-01, -8.1034e-01],
[-2.1353e+00, -1.9136e+00, -1.0192e+00, ..., -1.0956e+00,
-4.6050e-01, -1.8096e-01],
...,
[-1.4269e-01, -5.8790e-01, -4.6548e-01, ..., -5.0698e-01,
-1.8509e-01, -4.4247e-01],
[-6.9728e-01, -1.7549e-01, -4.2463e-01, ..., -2.1893e-01,
2.5748e-01, 2.5415e-01],
[-7.8459e-01, -1.6576e-01, -3.3022e-01, ..., -4.1161e-01,
-1.9201e-01, -2.1743e-01]],
...,
[[-2.0339e+00, -1.1095e+00, -2.3575e+00, ..., -3.1910e+00,
-2.6288e+00, -1.8242e+00],
[-1.7581e+00, 4.0235e-02, -1.3467e+00, ..., -2.4735e+00,
-2.5251e+00, -1.9664e+00],
[-4.0069e-01, 2.3596e+00, 1.0870e+00, ..., 1.2169e-01,
2.7526e-01, 5.2730e-01],
...,
[-2.0549e+00, -1.0218e-01, -1.9190e-01, ..., -1.3587e+00,
-4.0005e-01, 1.0553e-01],
[-1.1976e+00, 7.7861e-01, 9.2311e-01, ..., 4.5558e-03,
3.1232e-01, 4.7730e-01],
[-2.0857e+00, -1.4214e+00, -1.6391e+00, ..., -2.4920e+00,
-1.5090e+00, -9.5861e-01]],
[[-2.4253e+00, -2.8136e+00, -4.1854e+00, ..., -2.3692e+00,
-1.8797e+00, -1.6928e+00],
[-1.7633e+00, -1.6345e+00, -4.8493e+00, ..., -2.8834e+00,
-2.7603e+00, -2.7523e+00],
[-2.8228e+00, -2.4666e+00, -6.6579e+00, ..., -4.4367e+00,
-4.3602e+00, -3.4751e+00],
...,
[-1.4157e+00, -5.1470e-02, -3.3889e+00, ..., -1.3983e+00,
-9.9387e-01, -9.2204e-01],
[-1.7135e+00, -6.5161e-01, -3.0361e+00, ..., -1.7013e+00,
-1.1422e+00, -7.9642e-01],
[-2.3212e+00, -1.4942e+00, -3.1128e+00, ..., -2.9598e+00,
-2.1301e+00, -1.9720e+00]],
[[-1.8052e+00, -2.5369e+00, -2.4572e+00, ..., -2.6969e+00,
-2.2110e+00, -1.6918e+00],
[-1.7966e+00, -2.5061e+00, -2.1468e+00, ..., -2.3971e+00,
-1.8984e+00, -1.5334e+00],
[-2.2429e+00, -2.8046e+00, -2.5045e+00, ..., -2.7349e+00,
-2.1339e+00, -1.6183e+00],
...,
[-2.3860e+00, -2.9561e+00, -2.7313e+00, ..., -2.8712e+00,
-2.2987e+00, -1.7691e+00],
[-1.5520e+00, -2.0176e+00, -1.9448e+00, ..., -1.9463e+00,
-1.6078e+00, -1.3018e+00],
[-1.0121e+00, -1.3081e+00, -1.3386e+00, ..., -1.1867e+00,
-1.0181e+00, -7.7719e-01]]],
[[[ 3.4274e-01, 8.2620e-01, 5.2789e-01, ..., 4.6083e-01,
1.0226e+00, 6.6562e-01],
[-1.3025e+00, -2.9628e-01, -1.6808e+00, ..., -2.5627e+00,
-7.3163e-01, -4.7266e-02],
[ 6.6142e-01, 6.2117e-01, 4.0208e-01, ..., -2.8629e-01,
4.9151e-01, 6.5457e-01],
...,
[-4.0601e-03, -7.2014e-01, -1.5675e+00, ..., -1.9286e+00,
-1.4069e+00, -9.3457e-01],
[ 1.4342e+00, 1.1963e+00, 5.8779e-01, ..., -2.6606e-01,
5.0058e-01, 7.1267e-01],
[ 1.0650e+00, 9.4549e-01, 6.1612e-01, ..., -3.2857e-01,
2.6264e-01, 2.6908e-01]],
[[-2.6299e+00, -1.5413e+00, -4.6259e+00, ..., -4.8145e+00,
-4.4054e+00, -3.3515e+00],
[-6.1285e-01, 1.3163e+00, -3.0281e+00, ..., -3.3879e+00,
-3.5527e+00, -2.9086e+00],
[ 5.6623e-02, 1.0692e+00, -5.7095e+00, ..., -5.3926e+00,
-5.2694e+00, -4.9310e+00],
...,
[-7.7082e-01, 2.5765e-01, -4.6680e+00, ..., -4.5919e+00,
-4.1485e+00, -3.4540e+00],
[ 2.0449e+00, 2.7249e+00, 2.3345e-01, ..., -9.4165e-01,
-1.2663e+00, -1.2768e+00],
[ 1.2294e+00, 1.5255e+00, -1.5667e+00, ..., -2.2363e+00,
-2.4913e+00, -2.3213e+00]],
[[-5.5272e-01, -1.6650e+00, -1.5839e+00, ..., -5.7371e-01,
-1.9071e-01, -4.6250e-01],
[-1.2551e+00, -1.6959e+00, -1.1427e+00, ..., -3.2909e-01,
8.6061e-02, 3.3080e-02],
[-1.2254e+00, -1.2315e+00, -7.0738e-01, ..., -2.7847e-01,
2.5258e-01, 2.7196e-01],
...,
[-1.0181e+00, -1.5751e+00, -7.3992e-01, ..., -4.8486e-01,
-1.6968e-01, -3.6262e-01],
[-1.3357e+00, -9.7675e-01, -6.3745e-01, ..., -2.9201e-01,
1.8137e-01, 2.3733e-01],
[-1.4012e+00, -7.8904e-01, -3.0804e-01, ..., -4.7973e-01,
-2.4269e-01, -2.1143e-01]],
...,
[[-2.0086e+00, -1.0925e+00, -2.3404e+00, ..., -3.6571e+00,
-2.8298e+00, -1.8275e+00],
[-5.4193e-01, 1.0594e+00, -7.0987e-01, ..., -2.6708e+00,
-2.4461e+00, -1.8499e+00],
[-1.2893e-01, 2.2930e+00, 6.9250e-01, ..., -5.3598e-01,
-8.1016e-02, 1.5467e-01],
...,
[-1.5060e+00, 1.4560e+00, 7.4818e-02, ..., -1.4948e+00,
-6.1989e-01, -7.1635e-02],
[-5.6943e-01, 1.6334e+00, 9.3602e-01, ..., -1.5040e-01,
1.2317e-01, 3.5527e-01],
[-1.7275e+00, -1.1823e+00, -1.6553e+00, ..., -2.3732e+00,
-1.4241e+00, -9.2601e-01]],
[[-1.3598e+00, -1.4058e+00, -3.0101e+00, ..., -1.1329e+00,
-7.5078e-01, -7.7444e-01],
[-7.0861e-01, -3.8401e-01, -3.9294e+00, ..., -1.3573e+00,
-1.2369e+00, -1.6094e+00],
[-2.0916e+00, -1.5902e+00, -6.2613e+00, ..., -3.2574e+00,
-2.7883e+00, -2.2513e+00],
...,
[-1.7598e+00, -7.4486e-01, -4.7414e+00, ..., -1.5435e+00,
-1.0592e+00, -8.8894e-01],
[-1.6962e+00, -8.3960e-01, -3.8036e+00, ..., -1.7664e+00,
-1.1656e+00, -7.6254e-01],
[-2.0944e+00, -1.3791e+00, -3.0883e+00, ..., -2.8051e+00,
-2.0309e+00, -1.8262e+00]],
[[-1.2109e+00, -1.8301e+00, -1.7077e+00, ..., -1.7942e+00,
-1.4880e+00, -1.1366e+00],
[-1.2467e+00, -1.7828e+00, -1.4329e+00, ..., -1.4003e+00,
-1.1045e+00, -8.8756e-01],
[-1.6328e+00, -1.9996e+00, -1.6837e+00, ..., -1.6021e+00,
-1.2028e+00, -8.9474e-01],
...,
[-2.4221e+00, -3.0711e+00, -2.9012e+00, ..., -2.8934e+00,
-2.3011e+00, -1.7811e+00],
[-1.6271e+00, -2.0968e+00, -2.0581e+00, ..., -2.0030e+00,
-1.6457e+00, -1.3251e+00],
[-1.1034e+00, -1.3494e+00, -1.4278e+00, ..., -1.2498e+00,
-1.0445e+00, -8.0561e-01]]]], device='cuda:0'),
tensor([[[[ 6.4493e-02, 2.1455e-01, -1.3955e+00, ..., -6.1294e-01,
-1.9205e-01, -4.1845e-02],
[-2.0399e+00, -1.5980e+00, -4.4862e+00, ..., -3.7877e+00,
-1.5372e+00, -5.5257e-01],
[ 5.9095e-01, -6.5078e-02, -8.4903e-01, ..., -2.3968e-01,
-2.8035e-01, 2.3241e-01],
...,
[-8.7243e-01, -1.8630e+00, -1.8751e+00, ..., -1.8387e+00,
-1.3517e+00, -9.8816e-01],
[ 7.2916e-01, -3.6963e-02, -1.4013e-01, ..., -4.7606e-02,
6.9670e-01, 7.8086e-01],
[ 1.2315e-01, -6.0678e-01, -4.9997e-01, ..., -1.7664e-01,
4.2804e-01, 3.3538e-01]],
[[-3.2905e+00, -1.8075e+00, -6.2512e+00, ..., -7.1334e+00,
-6.2384e+00, -4.6243e+00],
[-8.4457e-01, 1.4559e+00, -4.3712e+00, ..., -5.4615e+00,
-5.5322e+00, -4.2747e+00],
[-4.3573e-01, 1.0179e+00, -7.3567e+00, ..., -7.9679e+00,
-7.8414e+00, -6.6852e+00],
...,
[ 3.0935e-01, 3.5448e-01, -3.9932e+00, ..., -4.6213e+00,
-4.2056e+00, -3.4812e+00],
[ 2.2988e+00, 2.2799e+00, -1.1964e-01, ..., -7.9541e-01,
-1.1668e+00, -1.2036e+00],
[ 1.3406e+00, 1.1757e+00, -1.7535e+00, ..., -2.2131e+00,
-2.4993e+00, -2.4078e+00]],
[[-1.3101e+00, -2.2952e+00, -2.2666e+00, ..., -1.7506e+00,
-1.2435e+00, -1.3806e+00],
[-1.6424e+00, -1.9471e+00, -1.5157e+00, ..., -1.2495e+00,
-7.3306e-01, -8.1972e-01],
[-2.1280e+00, -1.9182e+00, -1.0233e+00, ..., -1.1012e+00,
-4.6927e-01, -1.9380e-01],
...,
[-1.4941e-01, -5.9708e-01, -4.7602e-01, ..., -4.8503e-01,
-1.9559e-01, -4.4977e-01],
[-6.9305e-01, -1.6983e-01, -4.2019e-01, ..., -1.9974e-01,
2.4994e-01, 2.4314e-01],
[-7.8019e-01, -1.5520e-01, -3.3036e-01, ..., -3.9344e-01,
-1.9325e-01, -2.2599e-01]],
...,
[[-1.9808e+00, -1.0382e+00, -2.2913e+00, ..., -3.1947e+00,
-2.6259e+00, -1.8357e+00],
[-1.7462e+00, 4.2640e-02, -1.3165e+00, ..., -2.4639e+00,
-2.4977e+00, -1.9442e+00],
[-3.9946e-01, 2.3350e+00, 1.0558e+00, ..., 1.5358e-01,
3.2125e-01, 5.7211e-01],
...,
[-2.0593e+00, -8.3405e-02, -1.7932e-01, ..., -1.3771e+00,
-4.0167e-01, 1.0185e-01],
[-1.1876e+00, 8.1103e-01, 9.5423e-01, ..., 5.7522e-03,
3.1651e-01, 4.7741e-01],
[-2.1021e+00, -1.4222e+00, -1.6394e+00, ..., -2.4742e+00,
-1.4712e+00, -9.3883e-01]],
[[-2.4224e+00, -2.8297e+00, -4.2082e+00, ..., -2.3637e+00,
-1.8814e+00, -1.6986e+00],
[-1.8051e+00, -1.6864e+00, -4.8716e+00, ..., -2.8890e+00,
-2.7615e+00, -2.7566e+00],
[-2.8405e+00, -2.4884e+00, -6.6268e+00, ..., -4.4501e+00,
-4.3548e+00, -3.4803e+00],
...,
[-1.3863e+00, -1.4646e-02, -3.3346e+00, ..., -1.4344e+00,
-1.0112e+00, -9.2084e-01],
[-1.7028e+00, -6.2179e-01, -3.0161e+00, ..., -1.7208e+00,
-1.1418e+00, -8.0745e-01],
[-2.3148e+00, -1.4742e+00, -3.1194e+00, ..., -2.9544e+00,
-2.1415e+00, -1.9986e+00]],
[[-1.7888e+00, -2.5161e+00, -2.4391e+00, ..., -2.6949e+00,
-2.2069e+00, -1.6929e+00],
[-1.7900e+00, -2.4960e+00, -2.1489e+00, ..., -2.3966e+00,
-1.8979e+00, -1.5379e+00],
[-2.2427e+00, -2.8134e+00, -2.5164e+00, ..., -2.7399e+00,
-2.1398e+00, -1.6283e+00],
...,
[-2.3914e+00, -2.9650e+00, -2.7364e+00, ..., -2.8684e+00,
-2.2959e+00, -1.7574e+00],
[-1.5513e+00, -2.0153e+00, -1.9428e+00, ..., -1.9427e+00,
-1.6054e+00, -1.2941e+00],
[-1.0102e+00, -1.3018e+00, -1.3333e+00, ..., -1.1855e+00,
-1.0168e+00, -7.6961e-01]]],
[[[ 4.8536e-01, 9.6654e-01, 8.8625e-01, ..., -3.5872e-01,
-1.2284e-01, -1.6935e-02],
[-1.2203e+00, -3.0335e-01, -1.4359e+00, ..., -3.2538e+00,
-1.3406e+00, -4.9186e-01],
[ 5.9652e-01, 4.4756e-01, 3.9960e-01, ..., -1.6006e-01,
-3.8465e-01, 1.2403e-01],
...,
[-1.0268e-01, -7.7303e-01, -1.4175e+00, ..., -1.9357e+00,
-1.8910e+00, -1.2398e+00],
[ 1.0769e+00, 8.0011e-01, 3.2228e-01, ..., 1.6642e-02,
2.8078e-01, 4.6852e-01],
[ 7.1735e-01, 5.5296e-01, 2.2174e-01, ..., 2.3166e-01,
4.7559e-01, 5.2042e-01]],
[[-2.2876e+00, -1.0742e+00, -4.1338e+00, ..., -6.8812e+00,
-6.0842e+00, -4.5418e+00],
[-2.7888e-01, 1.7277e+00, -2.5481e+00, ..., -5.4844e+00,
-5.5828e+00, -4.3044e+00],
[ 4.2540e-01, 1.5166e+00, -5.0836e+00, ..., -7.8623e+00,
-7.8251e+00, -6.5784e+00],
...,
[-4.5753e-01, 5.4782e-01, -4.0852e+00, ..., -7.0999e+00,
-6.7167e+00, -5.2659e+00],
[ 2.0344e+00, 2.6900e+00, 2.8163e-01, ..., -1.9209e+00,
-2.6035e+00, -2.1158e+00],
[ 1.1591e+00, 1.4580e+00, -1.5036e+00, ..., -2.9374e+00,
-3.5261e+00, -3.0871e+00]],
[[-5.7315e-01, -1.5883e+00, -1.4035e+00, ..., -1.4348e+00,
-1.0702e+00, -1.3421e+00],
[-1.2843e+00, -1.6467e+00, -1.0149e+00, ..., -1.0193e+00,
-6.3281e-01, -8.9511e-01],
[-1.1287e+00, -9.9632e-01, -3.9827e-01, ..., -1.1151e+00,
-6.0545e-01, -4.6149e-01],
...,
[-9.3783e-01, -1.2971e+00, -4.9678e-01, ..., -1.0311e+00,
-8.9943e-01, -1.2673e+00],
[-1.2740e+00, -8.3916e-01, -5.0258e-01, ..., -5.6240e-01,
-2.6242e-01, -4.3223e-01],
[-1.3296e+00, -7.2832e-01, -3.1597e-01, ..., -5.0835e-01,
-4.6373e-01, -6.2430e-01]],
...,
[[-2.1933e+00, -1.2386e+00, -2.5372e+00, ..., -3.0231e+00,
-2.4418e+00, -1.7833e+00],
[-7.0241e-01, 9.7649e-01, -9.3630e-01, ..., -2.3620e+00,
-2.4600e+00, -2.0070e+00],
[-2.4205e-01, 2.2350e+00, 5.1201e-01, ..., -6.6945e-02,
4.1565e-02, 2.7403e-01],
...,
[-1.4881e+00, 1.2036e+00, -2.1046e-01, ..., -9.0749e-01,
-2.5063e-01, 3.0018e-01],
[-5.1074e-01, 1.5115e+00, 7.7048e-01, ..., 4.7565e-01,
5.1390e-01, 8.2732e-01],
[-1.6120e+00, -1.1019e+00, -1.6225e+00, ..., -2.1743e+00,
-1.1337e+00, -3.2862e-01]],
[[-1.2922e+00, -1.2186e+00, -2.6223e+00, ..., -2.2917e+00,
-1.9367e+00, -1.8255e+00],
[-6.2423e-01, -1.3973e-01, -3.3763e+00, ..., -2.9048e+00,
-2.8849e+00, -2.8862e+00],
[-2.0870e+00, -1.4215e+00, -5.7487e+00, ..., -4.4915e+00,
-4.5224e+00, -3.6934e+00],
...,
[-1.6491e+00, -5.1330e-01, -4.0623e+00, ..., -2.8233e+00,
-2.7659e+00, -2.3590e+00],
[-1.6791e+00, -7.4754e-01, -3.4062e+00, ..., -2.4491e+00,
-2.2890e+00, -1.7022e+00],
[-2.1167e+00, -1.3840e+00, -2.9793e+00, ..., -2.8605e+00,
-2.4503e+00, -2.2028e+00]],
[[-1.1462e+00, -1.7027e+00, -1.5179e+00, ..., -2.5898e+00,
-2.1571e+00, -1.6814e+00],
[-1.1580e+00, -1.6165e+00, -1.2027e+00, ..., -2.4301e+00,
-1.9598e+00, -1.6004e+00],
[-1.5085e+00, -1.8068e+00, -1.4116e+00, ..., -2.8602e+00,
-2.2864e+00, -1.7659e+00],
...,
[-2.3059e+00, -2.8700e+00, -2.6722e+00, ..., -4.0315e+00,
-3.2638e+00, -2.5588e+00],
[-1.5399e+00, -1.9512e+00, -1.8952e+00, ..., -2.7886e+00,
-2.3335e+00, -1.9309e+00],
[-1.0235e+00, -1.2207e+00, -1.2855e+00, ..., -1.8524e+00,
-1.5403e+00, -1.1944e+00]]],
[[[ 3.7296e-01, 8.5150e-01, 7.8578e-01, ..., -2.3126e-01,
4.8206e-01, 7.2616e-01],
[-1.1084e+00, -2.0078e-01, -1.2172e+00, ..., -6.1307e+00,
-2.7029e+00, 5.0361e-01],
[ 6.9639e-01, 4.8878e-01, 5.7480e-01, ..., -7.4715e+00,
-3.1261e+00, 8.2917e-01],
...,
[-1.5703e-01, -1.1530e+00, -2.7019e+00, ..., -1.5098e+00,
-1.4290e+00, -8.6679e-01],
[ 1.4528e+00, 8.2014e-01, -1.0491e+00, ..., -2.7979e-01,
5.4116e-02, 3.2995e-01],
[ 8.5100e-01, 2.4915e-01, -7.6029e-01, ..., -1.9495e-01,
1.6858e-01, 3.3360e-01]],
[[-2.3568e+00, -1.2935e+00, -4.3180e+00, ..., -5.9022e+00,
-4.7873e+00, -3.6336e+00],
[-3.0370e-01, 1.4637e+00, -2.7400e+00, ..., -5.8875e+00,
-4.7241e+00, -3.2930e+00],
[ 4.7352e-01, 1.2625e+00, -5.2257e+00, ..., -8.8038e+00,
-6.8934e+00, -5.4000e+00],
...,
[-1.3152e+00, 1.1574e-01, -6.7942e+00, ..., -7.7505e+00,
-7.0624e+00, -5.5366e+00],
[ 1.7614e+00, 2.7981e+00, -1.1067e+00, ..., -2.7273e+00,
-3.0965e+00, -2.4723e+00],
[ 1.2015e+00, 1.8183e+00, -2.3140e+00, ..., -3.2992e+00,
-3.6720e+00, -3.1496e+00]],
[[-6.2858e-01, -1.7248e+00, -1.6225e+00, ..., -1.6285e+00,
-8.4547e-01, -8.2260e-01],
[-1.2971e+00, -1.7554e+00, -1.2270e+00, ..., -1.3365e+00,
-5.0957e-01, -2.8412e-01],
[-1.0009e+00, -9.5084e-01, -4.4673e-01, ..., -3.3995e-01,
1.2711e-02, 7.0201e-02],
...,
[-1.4358e+00, -1.7751e+00, -7.5244e-01, ..., -8.8637e-01,
-7.7708e-01, -1.0462e+00],
[-1.7674e+00, -1.1719e+00, -8.2093e-01, ..., -5.7151e-01,
-3.1496e-01, -3.7143e-01],
[-1.9087e+00, -1.3159e+00, -9.1947e-01, ..., -9.8609e-01,
-8.6526e-01, -8.4571e-01]],
...,
[[-2.1035e+00, -1.1319e+00, -2.1905e+00, ..., -3.5334e+00,
-2.6789e+00, -1.6037e+00],
[-6.1356e-01, 1.0801e+00, -3.9139e-01, ..., -4.4050e+00,
-2.4927e+00, -1.3198e+00],
[-4.1873e-01, 1.9653e+00, 7.3747e-01, ..., -3.6206e+00,
3.5089e-02, 3.9124e-01],
...,
[-1.8745e+00, 1.2435e+00, 1.0561e-02, ..., -1.0576e+00,
-5.2958e-01, -2.7391e-04],
[-8.1296e-01, 1.9402e+00, 1.6947e+00, ..., 5.3997e-01,
4.5332e-01, 7.1407e-01],
[-2.2345e+00, -1.1369e+00, -9.4661e-01, ..., -1.7975e+00,
-9.2552e-01, -2.7836e-01]],
[[-1.3812e+00, -1.3162e+00, -2.7165e+00, ..., -6.6644e-01,
-3.6372e-01, -3.9120e-01],
[-7.2279e-01, -2.9411e-01, -3.5103e+00, ..., -1.4169e+00,
-1.3938e+00, -1.6767e+00],
[-2.0913e+00, -1.4646e+00, -5.7914e+00, ..., -5.3634e+00,
-3.4058e+00, -2.5816e+00],
...,
[-2.5111e+00, -1.6134e+00, -5.4811e+00, ..., -3.1283e+00,
-2.7982e+00, -2.2015e+00],
[-2.2403e+00, -1.4952e+00, -4.2685e+00, ..., -2.7452e+00,
-2.3827e+00, -1.6710e+00],
[-2.3911e+00, -1.7966e+00, -3.3299e+00, ..., -3.0071e+00,
-2.4781e+00, -2.1495e+00]],
[[-1.1498e+00, -1.6944e+00, -1.5024e+00, ..., -1.8717e+00,
-1.4727e+00, -1.0615e+00],
[-1.1670e+00, -1.6332e+00, -1.2105e+00, ..., -1.8017e+00,
-1.2751e+00, -8.9252e-01],
[-1.5215e+00, -1.8171e+00, -1.4052e+00, ..., -2.2815e+00,
-1.4606e+00, -8.5500e-01],
...,
[-3.2179e+00, -4.0242e+00, -3.8147e+00, ..., -4.0349e+00,
-3.2379e+00, -2.4836e+00],
[-2.1386e+00, -2.7654e+00, -2.7327e+00, ..., -2.8342e+00,
-2.3386e+00, -1.8916e+00],
[-1.5591e+00, -1.8907e+00, -1.9564e+00, ..., -1.9224e+00,
-1.5550e+00, -1.1808e+00]]],
[[[ 2.6221e-01, 5.3689e-01, 6.5687e-01, ..., 5.8547e-01,
9.7379e-01, 7.8050e-01],
[-2.0995e+00, -1.3855e+00, -3.0268e+00, ..., -1.3370e+00,
7.1362e-02, 6.2022e-01],
[-7.2523e-02, -9.0994e-01, -2.5018e-02, ..., 4.4654e-01,
9.5370e-01, 9.6089e-01],
...,
[-9.5755e-01, -2.1345e+00, -1.9779e+00, ..., -1.3989e+00,
-8.9601e-01, -6.7120e-01],
[ 7.5317e-01, -1.6907e-01, -1.4025e-01, ..., 4.2436e-01,
8.1337e-01, 9.0780e-01],
[ 1.5888e-01, -6.3602e-01, -4.4462e-01, ..., 6.2356e-01,
9.0182e-01, 8.0346e-01]],
[[-1.7466e+00, -1.2782e+00, -4.2636e+00, ..., -5.7813e+00,
-4.9062e+00, -3.8225e+00],
[ 6.4347e-01, 1.8819e+00, -2.2686e+00, ..., -4.7195e+00,
-4.4569e+00, -3.6497e+00],
[ 1.5025e+00, 1.6626e+00, -4.9522e+00, ..., -6.8401e+00,
-6.3869e+00, -5.8070e+00],
...,
[ 3.5929e-01, 2.7154e-01, -4.2178e+00, ..., -6.0158e+00,
-5.1732e+00, -4.2414e+00],
[ 2.3944e+00, 2.2826e+00, -1.8485e-01, ..., -1.3274e+00,
-1.4117e+00, -1.4992e+00],
[ 1.4112e+00, 1.1928e+00, -1.7739e+00, ..., -2.7245e+00,
-2.8059e+00, -2.7556e+00]],
[[-2.1377e-01, -1.3536e+00, -1.5666e+00, ..., -8.0052e-01,
-5.4461e-01, -6.7247e-01],
[-6.8942e-01, -1.0248e+00, -1.0075e+00, ..., -6.8504e-01,
-3.6796e-01, -2.3143e-01],
[-5.0841e-01, -3.4695e-01, -4.7372e-01, ..., -3.9048e-01,
-3.0693e-03, 9.5398e-02],
...,
[-3.0189e-02, -5.4523e-01, -5.8215e-01, ..., -6.7929e-01,
-4.5583e-01, -5.7283e-01],
[-6.6821e-01, -2.0004e-01, -5.6862e-01, ..., -2.7887e-01,
3.0737e-03, 7.7965e-02],
[-8.2069e-01, -2.4458e-01, -4.8312e-01, ..., -2.8640e-01,
-1.8504e-01, -1.4157e-01]],
...,
[[-3.0770e+00, -2.5251e+00, -2.6287e+00, ..., -3.1766e+00,
-2.6100e+00, -1.7434e+00],
[-2.1701e+00, -6.7725e-01, -1.1097e+00, ..., -1.6803e+00,
-1.7617e+00, -1.4613e+00],
[-1.1238e+00, 1.0809e+00, 9.0432e-01, ..., -2.5047e-01,
2.7315e-03, 1.5426e-01],
...,
[-2.2002e+00, -2.0297e-01, -1.1863e-01, ..., -1.0721e+00,
-3.7998e-01, 1.8390e-01],
[-1.2120e+00, 8.1852e-01, 1.0201e+00, ..., 2.1312e-01,
3.5262e-01, 4.6839e-01],
[-2.0875e+00, -1.3925e+00, -1.5649e+00, ..., -2.2540e+00,
-1.2771e+00, -6.2618e-01]],
[[-1.6271e+00, -1.5885e+00, -2.9317e+00, ..., -1.3169e+00,
-7.2073e-01, -5.1858e-01],
[-5.7456e-01, 1.3108e-01, -3.2756e+00, ..., -2.0013e+00,
-1.6229e+00, -1.7113e+00],
[-2.0491e+00, -8.1384e-01, -5.3943e+00, ..., -4.0048e+00,
-3.3595e+00, -2.5730e+00],
...,
[-1.3701e+00, 1.0620e-01, -3.3564e+00, ..., -2.3894e+00,
-1.7775e+00, -1.3074e+00],
[-1.6785e+00, -5.0082e-01, -2.9596e+00, ..., -2.2756e+00,
-1.4955e+00, -8.5426e-01],
[-2.2853e+00, -1.3455e+00, -2.9758e+00, ..., -2.8953e+00,
-1.9412e+00, -1.6219e+00]],
[[-1.3549e+00, -1.9152e+00, -1.6823e+00, ..., -1.8583e+00,
-1.5023e+00, -1.1016e+00],
[-1.2797e+00, -1.7695e+00, -1.1924e+00, ..., -1.6233e+00,
-1.2773e+00, -9.7705e-01],
[-1.6050e+00, -1.9399e+00, -1.4192e+00, ..., -1.9016e+00,
-1.4154e+00, -1.0061e+00],
...,
[-2.4176e+00, -3.0205e+00, -2.7665e+00, ..., -3.1677e+00,
-2.4944e+00, -1.8554e+00],
[-1.5781e+00, -2.0756e+00, -1.9979e+00, ..., -2.2292e+00,
-1.8123e+00, -1.4240e+00],
[-1.0372e+00, -1.3530e+00, -1.3806e+00, ..., -1.3768e+00,
-1.1596e+00, -8.4622e-01]]],
[[[ 1.4601e-01, 3.7590e-01, -1.2345e+00, ..., 4.3473e-01,
9.8399e-01, 6.4046e-01],
[-2.1139e+00, -1.4684e+00, -4.4950e+00, ..., -2.4639e+00,
-7.0647e-01, -3.3571e-02],
[ 4.3311e-01, -3.0603e-01, -1.1861e+00, ..., -2.4596e-01,
4.8013e-01, 6.2828e-01],
...,
[-2.7544e-01, -1.2924e+00, -2.6409e+00, ..., -2.3675e+01,
-2.3575e+01, -2.3151e+01],
[ 1.2621e+00, 6.3464e-01, -9.3686e-01, ..., -2.0512e+01,
-1.9437e+01, -1.7389e+01],
[ 8.1865e-01, 2.7111e-01, -4.5576e-01, ..., -1.4421e+01,
-1.4469e+01, -1.0243e+01]],
[[-3.2649e+00, -1.4671e+00, -6.1910e+00, ..., -4.7476e+00,
-4.3402e+00, -3.3225e+00],
[-5.8036e-01, 2.1701e+00, -4.2093e+00, ..., -3.3861e+00,
-3.5256e+00, -2.9058e+00],
[ 3.3252e-02, 1.8752e+00, -7.4392e+00, ..., -5.2975e+00,
-5.1889e+00, -4.8663e+00],
...,
[-1.2681e+00, 2.6271e-01, -6.5967e+00, ..., -6.0002e+00,
-9.5441e+00, -6.7515e+00],
[ 1.5256e+00, 2.6145e+00, -1.3691e+00, ..., -2.3223e+00,
-4.5626e+00, -3.4496e+00],
[ 1.0028e+00, 1.6404e+00, -2.4201e+00, ..., -3.6967e+00,
-5.0475e+00, -5.0239e+00]],
[[-1.4143e+00, -2.3983e+00, -2.2087e+00, ..., -5.5545e-01,
-1.7474e-01, -4.3734e-01],
[-2.0457e+00, -2.2334e+00, -1.5815e+00, ..., -3.2913e-01,
7.5059e-02, 5.0091e-02],
[-2.5791e+00, -2.0518e+00, -8.5467e-01, ..., -2.7746e-01,
2.2595e-01, 2.7591e-01],
...,
[-1.6188e+00, -1.8618e+00, -8.0797e-01, ..., -5.8242e-01,
2.9566e-01, -1.3085e+00],
[-1.7785e+00, -1.2053e+00, -7.7945e-01, ..., 4.3278e-01,
1.3321e+00, -2.8125e-01],
[-1.7403e+00, -1.1409e+00, -6.1851e-01, ..., -2.0143e-01,
-1.9648e-01, -7.3643e-01]],
...,
[[-2.3736e+00, -1.3078e+00, -2.7486e+00, ..., -3.5059e+00,
-2.7137e+00, -1.7487e+00],
[-1.9276e+00, 2.0087e-01, -1.3251e+00, ..., -2.5793e+00,
-2.3582e+00, -1.8246e+00],
[-5.1432e-01, 2.5167e+00, 9.5141e-01, ..., -5.6557e-01,
-8.9611e-02, 1.2670e-01],
...,
[-1.7430e+00, 1.2039e+00, 2.5401e-01, ..., -1.1872e+01,
-1.4116e+01, -4.5075e+00],
[-8.6137e-01, 1.7080e+00, 1.5470e+00, ..., -1.0173e+01,
-8.3960e+00, -1.3872e+00],
[-2.1028e+00, -1.0244e+00, -8.2557e-01, ..., -9.3692e+00,
-5.3892e+00, -1.6444e+00]],
[[-2.3955e+00, -2.6625e+00, -4.0511e+00, ..., -1.1868e+00,
-7.7501e-01, -7.8875e-01],
[-1.6724e+00, -1.3486e+00, -4.7970e+00, ..., -1.3910e+00,
-1.2321e+00, -1.5732e+00],
[-3.0227e+00, -2.4550e+00, -6.9550e+00, ..., -3.2383e+00,
-2.7526e+00, -2.1953e+00],
...,
[-2.5432e+00, -1.7486e+00, -4.9571e+00, ..., 2.9615e+00,
2.4887e+00, 1.7336e+00],
[-2.2296e+00, -1.5647e+00, -3.8608e+00, ..., 9.2508e-01,
2.1163e+00, 1.8904e+00],
[-2.3083e+00, -1.7774e+00, -3.0751e+00, ..., -7.1944e-01,
1.1987e+00, -1.0501e-01]],
[[-1.9105e+00, -2.6426e+00, -2.4234e+00, ..., -1.7722e+00,
-1.4717e+00, -1.1215e+00],
[-1.9355e+00, -2.6565e+00, -2.1114e+00, ..., -1.3996e+00,
-1.1091e+00, -8.9016e-01],
[-2.4649e+00, -3.0226e+00, -2.5262e+00, ..., -1.6123e+00,
-1.2134e+00, -9.0300e-01],
...,
[-3.1249e+00, -3.9065e+00, -3.6776e+00, ..., -5.2476e+00,
-5.1970e+00, -3.8544e+00],
[-2.0758e+00, -2.6858e+00, -2.6465e+00, ..., -4.0137e+00,
-3.5879e+00, -2.4414e+00],
[-1.4958e+00, -1.8211e+00, -1.8613e+00, ..., -2.3622e+00,
-2.1194e+00, -1.3259e+00]]]], device='cuda:0'),
tensor([[[[ 2.8779e-01, 5.5382e-01, 6.9811e-01, ..., 4.8419e-01,
1.0077e+00, 5.9570e-01],
[-2.1973e+00, -1.5290e+00, -3.2083e+00, ..., -2.6374e+00,
-8.6066e-01, -1.7812e-01],
[-1.9192e-01, -1.0338e+00, -2.3896e-01, ..., -1.2560e-01,
5.9969e-01, 6.8628e-01],
...,
[ 8.6792e-02, -6.5799e-01, -1.4880e+00, ..., -1.0615e+00,
-7.6867e-01, -5.6839e-01],
[ 1.5462e+00, 1.2725e+00, 6.5415e-01, ..., 7.6427e-01,
9.2605e-01, 9.4683e-01],
[ 1.1705e+00, 1.0192e+00, 7.1120e-01, ..., 9.2462e-01,
1.0234e+00, 8.6853e-01]],
[[-1.6670e+00, -1.1231e+00, -4.1649e+00, ..., -4.8509e+00,
-4.4457e+00, -3.3709e+00],
[ 6.2445e-01, 1.9872e+00, -2.2505e+00, ..., -3.4856e+00,
-3.6357e+00, -2.9427e+00],
[ 1.3800e+00, 1.7459e+00, -4.9437e+00, ..., -5.3841e+00,
-5.2782e+00, -4.9106e+00],
...,
[-6.5010e-01, 5.6070e-01, -4.6818e+00, ..., -5.9541e+00,
-5.2247e+00, -4.2517e+00],
[ 2.1760e+00, 2.8973e+00, 2.0923e-01, ..., -1.5594e+00,
-1.7118e+00, -1.7106e+00],
[ 1.2954e+00, 1.5824e+00, -1.6423e+00, ..., -2.7058e+00,
-2.8533e+00, -2.7447e+00]],
[[-1.1488e-01, -1.1618e+00, -1.3328e+00, ..., -4.5737e-01,
-1.3108e-01, -4.6989e-01],
[-5.9497e-01, -8.6148e-01, -7.7803e-01, ..., -2.7258e-01,
3.4591e-02, -7.8726e-02],
[-5.3831e-01, -3.5042e-01, -4.1707e-01, ..., -3.4278e-01,
8.8684e-02, 9.0338e-02],
...,
[-1.0735e+00, -1.5879e+00, -6.6460e-01, ..., -5.2811e-01,
-4.5601e-01, -6.2843e-01],
[-1.3455e+00, -8.9589e-01, -4.6045e-01, ..., -7.1853e-02,
9.0855e-02, 6.9803e-02],
[-1.3964e+00, -7.0413e-01, -1.4028e-01, ..., -5.2373e-02,
-3.0329e-02, -5.5533e-02]],
...,
[[-3.0689e+00, -2.5329e+00, -2.8289e+00, ..., -3.4666e+00,
-2.6540e+00, -1.7606e+00],
[-2.3164e+00, -8.6442e-01, -1.6371e+00, ..., -2.6432e+00,
-2.3712e+00, -1.8418e+00],
[-1.0720e+00, 1.1978e+00, 6.2028e-01, ..., -4.6355e-01,
-4.8448e-02, 1.0869e-01],
...,
[-1.6537e+00, 1.5918e+00, 2.7668e-01, ..., -6.1510e-01,
-1.1383e-01, 3.6604e-01],
[-6.2808e-01, 1.7842e+00, 1.0795e+00, ..., 1.6642e-01,
3.3051e-01, 4.8897e-01],
[-1.7163e+00, -1.0116e+00, -1.4371e+00, ..., -2.0705e+00,
-1.1493e+00, -5.3312e-01]],
[[-1.5968e+00, -1.5794e+00, -2.9151e+00, ..., -1.3204e+00,
-8.5559e-01, -8.5028e-01],
[-5.8590e-01, 1.0100e-01, -3.2440e+00, ..., -1.3769e+00,
-1.2153e+00, -1.5762e+00],
[-2.0957e+00, -8.9040e-01, -5.3810e+00, ..., -3.1663e+00,
-2.6998e+00, -2.2039e+00],
...,
[-1.7533e+00, -6.7395e-01, -4.7020e+00, ..., -2.3558e+00,
-2.0303e+00, -1.6133e+00],
[-1.6723e+00, -7.8656e-01, -3.7603e+00, ..., -2.2612e+00,
-1.7012e+00, -1.1128e+00],
[-2.0800e+00, -1.3257e+00, -3.0122e+00, ..., -2.7540e+00,
-1.9724e+00, -1.6917e+00]],
[[-1.3596e+00, -1.9214e+00, -1.7134e+00, ..., -1.7723e+00,
-1.4676e+00, -1.1226e+00],
[-1.2835e+00, -1.7712e+00, -1.2338e+00, ..., -1.3861e+00,
-1.0864e+00, -8.6908e-01],
[-1.6116e+00, -1.9590e+00, -1.4874e+00, ..., -1.6131e+00,
-1.2034e+00, -8.8930e-01],
...,
[-2.4779e+00, -3.0936e+00, -2.8165e+00, ..., -3.0123e+00,
-2.3791e+00, -1.8062e+00],
[-1.6619e+00, -2.1171e+00, -2.0030e+00, ..., -2.0859e+00,
-1.7023e+00, -1.3580e+00],
[-1.1126e+00, -1.3384e+00, -1.3631e+00, ..., -1.2850e+00,
-1.0915e+00, -8.0311e-01]]],
[[[ 7.3179e-02, 2.3682e-01, -1.3887e+00, ..., -6.1898e-01,
-1.9925e-01, -5.2681e-02],
[-2.0961e+00, -1.5904e+00, -4.5007e+00, ..., -3.7814e+00,
-1.5360e+00, -5.6499e-01],
[ 5.6692e-01, -6.4983e-02, -8.2779e-01, ..., -2.0225e-01,
-2.7699e-01, 2.3101e-01],
...,
[-8.4290e-01, -1.8330e+00, -1.7699e+00, ..., -1.6847e+00,
-1.2852e+00, -9.1760e-01],
[ 7.4438e-01, -2.9852e-02, -4.2836e-02, ..., 7.3813e-02,
6.9844e-01, 7.9128e-01],
[ 9.2843e-02, -6.4476e-01, -4.5858e-01, ..., -9.8820e-02,
4.2507e-01, 3.1215e-01]],
[[-3.2972e+00, -1.7975e+00, -6.2409e+00, ..., -7.1291e+00,
-6.2512e+00, -4.6305e+00],
[-8.0243e-01, 1.5365e+00, -4.3515e+00, ..., -5.4515e+00,
-5.5360e+00, -4.2623e+00],
[-3.7349e-01, 1.0551e+00, -7.4575e+00, ..., -7.9418e+00,
-7.8145e+00, -6.6542e+00],
...,
[ 2.9296e-01, 3.1931e-01, -4.1240e+00, ..., -4.6820e+00,
-4.1799e+00, -3.4428e+00],
[ 2.2248e+00, 2.1684e+00, -3.4537e-01, ..., -8.6263e-01,
-1.1961e+00, -1.1911e+00],
[ 1.2894e+00, 1.0937e+00, -1.8790e+00, ..., -2.2833e+00,
-2.4995e+00, -2.3499e+00]],
[[-1.3467e+00, -2.3707e+00, -2.3320e+00, ..., -1.7589e+00,
-1.2405e+00, -1.3781e+00],
[-1.6955e+00, -1.9951e+00, -1.5381e+00, ..., -1.2568e+00,
-7.2982e-01, -8.1458e-01],
[-2.1682e+00, -1.9239e+00, -1.0159e+00, ..., -1.1054e+00,
-4.5685e-01, -1.8621e-01],
...,
[-1.2837e-01, -5.5282e-01, -4.4724e-01, ..., -4.8706e-01,
-1.6911e-01, -4.3346e-01],
[-6.6845e-01, -1.5179e-01, -3.9056e-01, ..., -1.7081e-01,
2.9373e-01, 2.7969e-01],
[-7.5178e-01, -1.4764e-01, -2.9973e-01, ..., -3.8918e-01,
-1.6659e-01, -2.0095e-01]],
...,
[[-2.0463e+00, -1.1138e+00, -2.3455e+00, ..., -3.1944e+00,
-2.6155e+00, -1.8207e+00],
[-1.7387e+00, 4.0559e-02, -1.3504e+00, ..., -2.4593e+00,
-2.5061e+00, -1.9690e+00],
[-3.5820e-01, 2.3681e+00, 1.0726e+00, ..., 1.3208e-01,
2.9972e-01, 5.3935e-01],
...,
[-2.0543e+00, -1.2699e-01, -2.0694e-01, ..., -1.4046e+00,
-3.5719e-01, 1.4038e-01],
[-1.2298e+00, 7.4404e-01, 8.7032e-01, ..., -1.0080e-01,
2.8796e-01, 4.8135e-01],
[-2.0888e+00, -1.3945e+00, -1.6466e+00, ..., -2.5466e+00,
-1.5429e+00, -9.6342e-01]],
[[-2.4096e+00, -2.8203e+00, -4.1991e+00, ..., -2.3734e+00,
-1.8923e+00, -1.7009e+00],
[-1.7537e+00, -1.6281e+00, -4.8865e+00, ..., -2.8905e+00,
-2.7671e+00, -2.7484e+00],
[-2.8342e+00, -2.4781e+00, -6.6716e+00, ..., -4.4380e+00,
-4.3572e+00, -3.4647e+00],
...,
[-1.4268e+00, -1.1397e-01, -3.3907e+00, ..., -1.4557e+00,
-1.0050e+00, -8.9944e-01],
[-1.7016e+00, -6.7909e-01, -3.0257e+00, ..., -1.7457e+00,
-1.1556e+00, -7.8847e-01],
[-2.2865e+00, -1.5077e+00, -3.1352e+00, ..., -2.9696e+00,
-2.1082e+00, -1.9572e+00]],
[[-1.8014e+00, -2.5333e+00, -2.4503e+00, ..., -2.6967e+00,
-2.2144e+00, -1.6948e+00],
[-1.7983e+00, -2.5021e+00, -2.1415e+00, ..., -2.3978e+00,
-1.9032e+00, -1.5358e+00],
[-2.2480e+00, -2.8045e+00, -2.5085e+00, ..., -2.7469e+00,
-2.1428e+00, -1.6203e+00],
...,
[-2.3930e+00, -2.9570e+00, -2.7448e+00, ..., -2.8808e+00,
-2.3105e+00, -1.7682e+00],
[-1.5707e+00, -2.0285e+00, -1.9631e+00, ..., -1.9541e+00,
-1.6146e+00, -1.2989e+00],
[-1.0248e+00, -1.3188e+00, -1.3437e+00, ..., -1.1911e+00,
-1.0208e+00, -7.7979e-01]]],
[[[ 9.5010e-02, 2.5456e-01, -1.4145e+00, ..., -6.0680e-01,
-2.0736e-01, -3.1949e-02],
[-2.2072e+00, -1.6888e+00, -4.7070e+00, ..., -3.6312e+00,
-1.4655e+00, -5.1506e-01],
[ 6.0055e-01, -1.3765e-01, -8.7122e-01, ..., -1.6255e-01,
-2.4295e-01, 2.9093e-01],
...,
[-3.4050e-01, -1.1248e+00, -1.4432e+00, ..., -1.3957e+00,
-8.9321e-01, -6.9004e-01],
[ 9.6830e-01, 5.8329e-01, 3.3940e-01, ..., 3.4500e-01,
7.1519e-01, 7.8697e-01],
[ 6.8131e-01, 4.5285e-01, 3.4246e-01, ..., 4.9382e-01,
7.8246e-01, 6.9554e-01]],
[[-3.3353e+00, -1.6371e+00, -6.2048e+00, ..., -7.0096e+00,
-6.1159e+00, -4.5468e+00],
[-5.8884e-01, 2.0086e+00, -4.0773e+00, ..., -5.5092e+00,
-5.5401e+00, -4.2926e+00],
[-3.1140e-02, 1.5974e+00, -7.3428e+00, ..., -7.9750e+00,
-7.8032e+00, -6.6229e+00],
...,
[-4.6832e-01, 2.5922e-01, -4.4104e+00, ..., -5.3647e+00,
-4.6734e+00, -3.8839e+00],
[ 1.9695e+00, 2.4591e+00, -9.8965e-03, ..., -1.1146e+00,
-1.2541e+00, -1.3744e+00],
[ 1.1310e+00, 1.3284e+00, -1.6649e+00, ..., -2.4877e+00,
-2.5958e+00, -2.5670e+00]],
[[-1.4028e+00, -2.5210e+00, -2.4538e+00, ..., -1.6808e+00,
-1.2052e+00, -1.3361e+00],
[-1.8630e+00, -2.1682e+00, -1.6251e+00, ..., -1.1785e+00,
-7.0326e-01, -8.1628e-01],
[-2.3306e+00, -2.0356e+00, -1.0137e+00, ..., -1.0830e+00,
-4.6545e-01, -2.0979e-01],
...,
[-6.5433e-01, -1.1206e+00, -5.9137e-01, ..., -6.1493e-01,
-4.6907e-01, -6.4869e-01],
[-1.0552e+00, -7.0930e-01, -5.8097e-01, ..., -2.8580e-01,
-6.8893e-02, -5.6902e-02],
[-1.2080e+00, -6.7488e-01, -3.8711e-01, ..., -3.4534e-01,
-2.7897e-01, -2.5597e-01]],
...,
[[-2.2412e+00, -1.1315e+00, -2.5116e+00, ..., -3.0157e+00,
-2.4851e+00, -1.7504e+00],
[-1.7989e+00, 2.2561e-01, -1.3735e+00, ..., -2.4577e+00,
-2.4585e+00, -1.8725e+00],
[-3.3036e-01, 2.7286e+00, 1.1954e+00, ..., 9.2481e-02,
1.8301e-01, 4.7909e-01],
...,
[-1.5492e+00, 9.5021e-01, -1.5275e-01, ..., -1.1186e+00,
-4.8553e-01, 4.9168e-02],
[-5.4687e-01, 1.3933e+00, 7.6619e-01, ..., 7.8703e-02,
1.8485e-01, 3.1281e-01],
[-1.5995e+00, -1.0875e+00, -1.5371e+00, ..., -2.2429e+00,
-1.3252e+00, -6.9422e-01]],
[[-2.4060e+00, -2.7630e+00, -4.3532e+00, ..., -2.4598e+00,
-2.0068e+00, -1.7959e+00],
[-1.6340e+00, -1.3670e+00, -5.0329e+00, ..., -2.9913e+00,
-2.8684e+00, -2.8235e+00],
[-2.7771e+00, -2.2250e+00, -6.9503e+00, ..., -4.4231e+00,
-4.3559e+00, -3.5019e+00],
...,
[-1.6892e+00, -5.4231e-01, -4.0705e+00, ..., -2.0379e+00,
-1.5521e+00, -1.2315e+00],
[-1.6757e+00, -7.2384e-01, -3.3612e+00, ..., -2.0283e+00,
-1.3390e+00, -8.1939e-01],
[-2.0556e+00, -1.2745e+00, -2.8320e+00, ..., -2.6681e+00,
-1.7884e+00, -1.5420e+00]],
[[-1.8668e+00, -2.6398e+00, -2.4939e+00, ..., -2.6465e+00,
-2.1796e+00, -1.6689e+00],
[-1.8565e+00, -2.6004e+00, -2.1435e+00, ..., -2.4052e+00,
-1.9018e+00, -1.5407e+00],
[-2.3267e+00, -2.8975e+00, -2.4902e+00, ..., -2.7717e+00,
-2.1632e+00, -1.6493e+00],
...,
[-2.2750e+00, -2.8544e+00, -2.6590e+00, ..., -2.8819e+00,
-2.2990e+00, -1.7320e+00],
[-1.5330e+00, -1.9772e+00, -1.9289e+00, ..., -2.0466e+00,
-1.6827e+00, -1.3362e+00],
[-1.0395e+00, -1.2705e+00, -1.3343e+00, ..., -1.2776e+00,
-1.0876e+00, -8.0130e-01]]],
[[[ 7.1308e-02, 2.4337e-01, -1.3840e+00, ..., -6.0709e-01,
-1.7604e-01, -1.9460e-02],
[-2.1052e+00, -1.6088e+00, -4.5029e+00, ..., -3.7729e+00,
-1.5074e+00, -5.0948e-01],
[ 5.5596e-01, -5.9284e-02, -8.4165e-01, ..., -2.0351e-01,
-2.4136e-01, 2.7372e-01],
...,
[-3.6954e-01, -1.2534e+00, -1.6898e+00, ..., -1.2436e+00,
-8.0437e-01, -5.6424e-01],
[ 1.0857e+00, 6.3404e-01, 2.9209e-01, ..., 4.0287e-01,
7.0775e-01, 8.1073e-01],
[ 7.5830e-01, 4.9186e-01, 2.9786e-01, ..., 5.3099e-01,
7.7049e-01, 7.0552e-01]],
[[-3.2982e+00, -1.8715e+00, -6.2519e+00, ..., -7.1200e+00,
-6.2292e+00, -4.6355e+00],
[-8.2451e-01, 1.4277e+00, -4.3407e+00, ..., -5.4720e+00,
-5.5244e+00, -4.2941e+00],
[-4.0297e-01, 9.5454e-01, -7.3933e+00, ..., -7.9864e+00,
-7.8332e+00, -6.7258e+00],
...,
[-2.9611e-01, 6.1738e-01, -4.1892e+00, ..., -5.3802e+00,
-4.6921e+00, -3.9510e+00],
[ 2.3652e+00, 2.9689e+00, 5.3030e-01, ..., -1.2055e+00,
-1.3448e+00, -1.4664e+00],
[ 1.4014e+00, 1.6456e+00, -1.4334e+00, ..., -2.5243e+00,
-2.6102e+00, -2.5960e+00]],
[[-1.3493e+00, -2.3468e+00, -2.3189e+00, ..., -1.7361e+00,
-1.2519e+00, -1.3951e+00],
[-1.6972e+00, -1.9668e+00, -1.5275e+00, ..., -1.2240e+00,
-7.3577e-01, -8.1798e-01],
[-2.1577e+00, -1.8715e+00, -9.7015e-01, ..., -1.1024e+00,
-4.7170e-01, -1.8998e-01],
...,
[-7.6502e-01, -1.2723e+00, -6.3761e-01, ..., -5.9715e-01,
-4.8507e-01, -6.3334e-01],
[-1.2066e+00, -8.0599e-01, -6.3442e-01, ..., -2.7070e-01,
-7.8432e-02, -5.0571e-02],
[-1.3858e+00, -7.6708e-01, -4.2285e-01, ..., -3.4288e-01,
-2.9971e-01, -2.6040e-01]],
...,
[[-2.0254e+00, -1.1267e+00, -2.3402e+00, ..., -3.2385e+00,
-2.6770e+00, -1.8628e+00],
[-1.7334e+00, 1.4078e-02, -1.3148e+00, ..., -2.5205e+00,
-2.5353e+00, -1.9522e+00],
[-3.8567e-01, 2.2650e+00, 1.0635e+00, ..., 1.1275e-01,
2.4936e-01, 5.1937e-01],
...,
[-1.7234e+00, 1.1118e+00, -6.1431e-02, ..., -1.1845e+00,
-4.7542e-01, 2.1458e-02],
[-6.1271e-01, 1.5738e+00, 9.2441e-01, ..., 2.6595e-02,
1.6339e-01, 2.7303e-01],
[-1.7057e+00, -1.1526e+00, -1.6130e+00, ..., -2.1918e+00,
-1.2763e+00, -6.7957e-01]],
[[-2.4209e+00, -2.8346e+00, -4.1725e+00, ..., -2.3671e+00,
-1.9081e+00, -1.7339e+00],
[-1.7787e+00, -1.6629e+00, -4.8168e+00, ..., -2.8898e+00,
-2.7835e+00, -2.8197e+00],
[-2.8575e+00, -2.4971e+00, -6.5891e+00, ..., -4.4406e+00,
-4.3701e+00, -3.5448e+00],
...,
[-1.6098e+00, -2.7380e-01, -4.1745e+00, ..., -2.0936e+00,
-1.6294e+00, -1.2566e+00],
[-1.6358e+00, -5.3487e-01, -3.4411e+00, ..., -2.0672e+00,
-1.3976e+00, -8.4375e-01],
[-2.1003e+00, -1.1998e+00, -2.8785e+00, ..., -2.6661e+00,
-1.7968e+00, -1.5331e+00]],
[[-1.7975e+00, -2.5249e+00, -2.4569e+00, ..., -2.6919e+00,
-2.2081e+00, -1.6915e+00],
[-1.7923e+00, -2.4914e+00, -2.1568e+00, ..., -2.4031e+00,
-1.8957e+00, -1.5366e+00],
[-2.2368e+00, -2.7935e+00, -2.5235e+00, ..., -2.7506e+00,
-2.1402e+00, -1.6248e+00],
...,
[-2.3830e+00, -2.9896e+00, -2.7218e+00, ..., -2.8394e+00,
-2.2601e+00, -1.6965e+00],
[-1.5981e+00, -2.0597e+00, -1.9634e+00, ..., -2.0142e+00,
-1.6498e+00, -1.3057e+00],
[-1.0738e+00, -1.3120e+00, -1.3464e+00, ..., -1.2569e+00,
-1.0656e+00, -7.8165e-01]]],
[[[-1.0256e-01, 2.7837e-02, -2.3458e+00, ..., -2.5251e+00,
1.0858e+00, 7.6847e-01],
[-2.6258e+00, -2.1947e+00, -6.1607e+00, ..., -5.9209e+00,
-2.1455e-01, 3.5793e-01],
[ 2.2567e-01, -7.7667e-01, -2.0780e+00, ..., -3.2459e+00,
7.7840e-01, 7.9024e-01],
...,
[-2.9686e-01, -1.4605e+00, -3.0580e+00, ..., -1.3599e+00,
-9.4336e-01, -6.1222e-01],
[ 1.3118e+00, 4.3499e-01, -1.7035e+00, ..., 1.9246e-01,
7.1826e-01, 8.5108e-01],
[ 6.4856e-01, -1.7623e-01, -1.1338e+00, ..., 2.1433e-01,
6.2491e-01, 5.6815e-01]],
[[-3.6857e+00, -1.7085e+00, -7.0339e+00, ..., -5.6256e+00,
-4.4975e+00, -3.4587e+00],
[-8.1695e-01, 2.1513e+00, -4.9227e+00, ..., -3.8232e+00,
-3.7797e+00, -3.0946e+00],
[-3.5202e-01, 1.7751e+00, -8.3853e+00, ..., -5.3147e+00,
-5.5517e+00, -5.1815e+00],
...,
[-1.8146e+00, -5.8817e-02, -7.6889e+00, ..., -5.0186e+00,
-4.4223e+00, -3.6949e+00],
[ 1.1851e+00, 2.5473e+00, -1.9618e+00, ..., -1.0962e+00,
-1.3141e+00, -1.3771e+00],
[ 8.7910e-01, 1.7817e+00, -2.6608e+00, ..., -2.4155e+00,
-2.5665e+00, -2.4676e+00]],
[[-1.9527e+00, -2.8309e+00, -2.5926e+00, ..., -7.8723e-01,
-3.6533e-01, -5.6463e-01],
[-2.5179e+00, -2.5530e+00, -1.7878e+00, ..., -4.9742e-01,
-1.6392e-01, -8.8039e-02],
[-3.3848e+00, -2.6583e+00, -1.0508e+00, ..., 3.4144e-01,
1.4247e-01, 1.9516e-01],
...,
[-1.9844e+00, -2.0182e+00, -6.3702e-01, ..., -4.8079e-01,
-2.5331e-01, -3.7485e-01],
[-2.0476e+00, -1.3035e+00, -6.2381e-01, ..., -2.1432e-01,
1.0839e-01, 1.7268e-01],
[-2.0983e+00, -1.5704e+00, -9.5990e-01, ..., -3.2050e-01,
-1.8938e-01, -1.6541e-01]],
...,
[[-2.4021e+00, -1.3025e+00, -2.7352e+00, ..., -4.3687e+00,
-2.7564e+00, -1.8061e+00],
[-2.6468e+00, -4.8862e-01, -1.8472e+00, ..., -1.6828e+00,
-2.0636e+00, -1.6434e+00],
[-7.2823e-01, 2.3912e+00, 9.5865e-01, ..., 2.2791e-01,
3.1307e-03, 2.3202e-01],
...,
[-2.0915e+00, 8.7338e-01, 7.6856e-02, ..., -1.3292e+00,
-5.3270e-01, 2.3512e-03],
[-9.9147e-01, 1.7929e+00, 1.9021e+00, ..., -1.0082e-01,
1.4934e-01, 3.4034e-01],
[-2.3093e+00, -1.0560e+00, -4.6773e-01, ..., -2.2116e+00,
-1.3015e+00, -7.3783e-01]],
[[-2.8667e+00, -3.3503e+00, -4.6158e+00, ..., -6.6702e-01,
-5.1439e-01, -4.9460e-01],
[-2.1653e+00, -2.0417e+00, -5.2890e+00, ..., -1.5726e+00,
-1.2504e+00, -1.5402e+00],
[-3.3870e+00, -2.9866e+00, -7.1647e+00, ..., -3.4163e+00,
-2.9799e+00, -2.3117e+00],
...,
[-2.9010e+00, -2.2645e+00, -5.1595e+00, ..., -1.9469e+00,
-1.4055e+00, -1.0553e+00],
[-2.4708e+00, -1.9841e+00, -4.0682e+00, ..., -1.9849e+00,
-1.3183e+00, -7.8338e-01],
[-2.4558e+00, -2.1076e+00, -3.2278e+00, ..., -2.7263e+00,
-1.9035e+00, -1.6300e+00]],
[[-2.2670e+00, -3.1001e+00, -2.9106e+00, ..., -1.8982e+00,
-1.4511e+00, -1.0870e+00],
[-2.3079e+00, -3.1683e+00, -2.6238e+00, ..., -1.6537e+00,
-1.1254e+00, -8.8388e-01],
[-2.8993e+00, -3.6425e+00, -3.1552e+00, ..., -1.7255e+00,
-1.2230e+00, -8.9029e-01],
...,
[-3.5408e+00, -4.4649e+00, -4.2262e+00, ..., -2.8277e+00,
-2.2594e+00, -1.7201e+00],
[-2.3165e+00, -3.0435e+00, -3.0005e+00, ..., -1.9856e+00,
-1.6272e+00, -1.2995e+00],
[-1.6966e+00, -2.0940e+00, -2.1171e+00, ..., -1.2320e+00,
-1.0441e+00, -7.8569e-01]]]], device='cuda:0'),
tensor([[[[ 3.0560e-02, 1.9816e-01, -1.8008e+00, ..., 5.5855e-01,
1.0392e+00, 7.0422e-01],
[-2.3823e+00, -1.8267e+00, -5.3125e+00, ..., -2.1007e+00,
-4.5128e-01, 1.7365e-01],
[ 2.8984e-01, -5.4476e-01, -1.6395e+00, ..., -1.4080e-03,
6.6403e-01, 7.4318e-01],
...,
[-2.6592e-01, -1.3250e+00, -2.7604e+00, ..., -1.6773e+00,
-1.1800e+00, -7.8099e-01],
[ 1.2597e+00, 5.5400e-01, -1.2960e+00, ..., -3.6924e-02,
6.2182e-01, 7.7796e-01],
[ 7.0684e-01, 4.6015e-02, -7.9055e-01, ..., -4.3517e-02,
4.6094e-01, 4.2256e-01]],
[[-3.4641e+00, -1.5561e+00, -6.5947e+00, ..., -4.9155e+00,
-4.4241e+00, -3.3964e+00],
[-7.0061e-01, 2.1695e+00, -4.5745e+00, ..., -3.5409e+00,
-3.6321e+00, -2.9885e+00],
[-1.6956e-01, 1.7925e+00, -7.9209e+00, ..., -5.4907e+00,
-5.3609e+00, -5.0332e+00],
...,
[-1.4765e+00, 1.5055e-01, -7.0965e+00, ..., -4.7758e+00,
-4.2555e+00, -3.5672e+00],
[ 1.3705e+00, 2.5429e+00, -1.7308e+00, ..., -9.7335e-01,
-1.2526e+00, -1.3098e+00],
[ 9.4977e-01, 1.6918e+00, -2.5847e+00, ..., -2.2991e+00,
-2.4951e+00, -2.4060e+00]],
[[-1.6784e+00, -2.6185e+00, -2.3967e+00, ..., -6.4261e-01,
-2.7763e-01, -5.0838e-01],
[-2.2797e+00, -2.3953e+00, -1.6912e+00, ..., -4.2940e-01,
-4.1229e-02, -2.4780e-02],
[-2.9826e+00, -2.3447e+00, -9.7696e-01, ..., -2.6996e-01,
1.9745e-01, 2.4294e-01],
...,
[-1.7916e+00, -1.9116e+00, -6.6870e-01, ..., -4.6997e-01,
-2.0512e-01, -3.6327e-01],
[-1.9259e+00, -1.2505e+00, -6.8593e-01, ..., -2.6301e-01,
1.3047e-01, 1.9975e-01],
[-1.9385e+00, -1.3635e+00, -7.8611e-01, ..., -4.0540e-01,
-2.2699e-01, -1.9532e-01]],
...,
[[-2.3763e+00, -1.3031e+00, -2.7522e+00, ..., -3.5338e+00,
-2.7865e+00, -1.8141e+00],
[-2.2264e+00, -1.2886e-01, -1.6017e+00, ..., -2.3437e+00,
-2.2170e+00, -1.7351e+00],
[-5.3060e-01, 2.4938e+00, 9.8856e-01, ..., -5.2742e-01,
-5.9860e-02, 1.7688e-01],
...,
[-1.8946e+00, 1.0143e+00, 1.0081e-01, ..., -1.4690e+00,
-6.2443e-01, -5.0804e-02],
[-9.2246e-01, 1.7132e+00, 1.6521e+00, ..., -1.4635e-01,
1.2090e-01, 3.3914e-01],
[-2.1929e+00, -1.0343e+00, -6.4642e-01, ..., -2.3087e+00,
-1.3852e+00, -8.2945e-01]],
[[-2.6078e+00, -3.0026e+00, -4.3421e+00, ..., -1.0465e+00,
-6.1526e-01, -6.2275e-01],
[-1.9095e+00, -1.7037e+00, -5.0348e+00, ..., -1.3985e+00,
-1.2429e+00, -1.5744e+00],
[-3.1949e+00, -2.7267e+00, -7.0024e+00, ..., -3.3717e+00,
-2.8981e+00, -2.3006e+00],
...,
[-2.7074e+00, -2.0148e+00, -5.1320e+00, ..., -1.6916e+00,
-1.2178e+00, -9.7373e-01],
[-2.3295e+00, -1.7601e+00, -4.0132e+00, ..., -1.8506e+00,
-1.2213e+00, -7.7350e-01],
[-2.3465e+00, -1.8916e+00, -3.1389e+00, ..., -2.7484e+00,
-1.9356e+00, -1.7288e+00]],
[[-2.0858e+00, -2.8655e+00, -2.6611e+00, ..., -1.7624e+00,
-1.4651e+00, -1.1132e+00],
[-2.1214e+00, -2.9110e+00, -2.3744e+00, ..., -1.3977e+00,
-1.1172e+00, -8.8586e-01],
[-2.6888e+00, -3.3402e+00, -2.8622e+00, ..., -1.5998e+00,
-1.2126e+00, -8.9412e-01],
...,
[-3.3471e+00, -4.1997e+00, -3.9673e+00, ..., -2.8466e+00,
-2.2836e+00, -1.7513e+00],
[-2.2194e+00, -2.8880e+00, -2.8528e+00, ..., -1.9797e+00,
-1.6295e+00, -1.3131e+00],
[-1.6183e+00, -1.9810e+00, -2.0226e+00, ..., -1.2323e+00,
-1.0457e+00, -7.9482e-01]]],
[[[ 9.5470e-02, 3.1555e-01, -1.1922e+00, ..., 4.6914e-01,
1.0231e+00, 6.6372e-01],
[-1.9344e+00, -1.3427e+00, -4.0827e+00, ..., -2.5563e+00,
-7.3835e-01, -4.6750e-02],
[ 3.1684e-01, -2.7576e-01, -1.0811e+00, ..., -2.5325e-01,
4.9633e-01, 6.5312e-01],
...,
[-2.5031e-01, -1.3907e+00, -2.8005e+00, ..., -2.0015e+00,
-1.4082e+00, -9.1254e-01],
[ 1.4219e+00, 7.2778e-01, -8.8201e-01, ..., -3.0393e-01,
5.3247e-01, 7.2357e-01],
[ 9.3481e-01, 3.7690e-01, -3.8985e-01, ..., -3.6085e-01,
2.8911e-01, 2.9719e-01]],
[[-3.1116e+00, -1.5887e+00, -6.1061e+00, ..., -4.8343e+00,
-4.4182e+00, -3.3630e+00],
[-7.9000e-01, 1.5524e+00, -4.4875e+00, ..., -3.3948e+00,
-3.5394e+00, -2.9022e+00],
[-3.7306e-01, 1.2237e+00, -7.4533e+00, ..., -5.3742e+00,
-5.2488e+00, -4.9248e+00],
...,
[-1.0676e+00, 6.0734e-01, -6.3386e+00, ..., -4.5781e+00,
-4.1272e+00, -3.4734e+00],
[ 1.9219e+00, 3.1040e+00, -7.8706e-01, ..., -9.1244e-01,
-1.2391e+00, -1.2933e+00],
[ 1.2723e+00, 1.9513e+00, -2.1467e+00, ..., -2.2569e+00,
-2.4962e+00, -2.3919e+00]],
[[-1.2713e+00, -2.0947e+00, -1.9462e+00, ..., -5.8421e-01,
-1.8824e-01, -4.6083e-01],
[-1.7664e+00, -1.9169e+00, -1.3813e+00, ..., -3.3471e-01,
7.7264e-02, 4.0416e-02],
[-2.3070e+00, -1.8482e+00, -7.8507e-01, ..., -2.7642e-01,
2.5412e-01, 2.9401e-01],
...,
[-1.7312e+00, -1.9937e+00, -8.2909e-01, ..., -4.8900e-01,
-1.6852e-01, -3.4353e-01],
[-1.9268e+00, -1.2977e+00, -8.3193e-01, ..., -3.1304e-01,
1.4261e-01, 1.9872e-01],
[-1.8941e+00, -1.1984e+00, -6.0765e-01, ..., -5.0455e-01,
-2.7867e-01, -2.5081e-01]],
...,
[[-2.0118e+00, -1.1738e+00, -2.4772e+00, ..., -3.6761e+00,
-2.8365e+00, -1.8314e+00],
[-1.8031e+00, -6.8812e-03, -1.3114e+00, ..., -2.6965e+00,
-2.4360e+00, -1.8654e+00],
[-5.4495e-01, 2.0943e+00, 7.7195e-01, ..., -6.1351e-01,
-9.8225e-02, 1.4285e-01],
...,
[-1.8891e+00, 1.3545e+00, 2.5869e-01, ..., -1.4986e+00,
-6.8708e-01, -1.4323e-01],
[-9.3739e-01, 1.8589e+00, 1.6218e+00, ..., -1.2752e-01,
1.0826e-01, 3.2796e-01],
[-2.2514e+00, -1.1390e+00, -9.7729e-01, ..., -2.3051e+00,
-1.4014e+00, -8.8834e-01]],
[[-2.3941e+00, -2.7212e+00, -3.8906e+00, ..., -1.1496e+00,
-7.4125e-01, -7.6107e-01],
[-1.8410e+00, -1.6545e+00, -4.5866e+00, ..., -1.3658e+00,
-1.2294e+00, -1.5960e+00],
[-3.0423e+00, -2.6420e+00, -6.5289e+00, ..., -3.2756e+00,
-2.8050e+00, -2.2443e+00],
...,
[-2.4781e+00, -1.5029e+00, -5.1087e+00, ..., -1.4874e+00,
-1.0787e+00, -9.2602e-01],
[-2.1955e+00, -1.3975e+00, -3.9785e+00, ..., -1.7204e+00,
-1.1767e+00, -8.0517e-01],
[-2.3562e+00, -1.7261e+00, -3.1472e+00, ..., -2.7762e+00,
-2.0363e+00, -1.8596e+00]],
[[-1.8226e+00, -2.5030e+00, -2.3672e+00, ..., -1.8028e+00,
-1.4963e+00, -1.1433e+00],
[-1.8714e+00, -2.5569e+00, -2.1419e+00, ..., -1.4001e+00,
-1.1100e+00, -8.8547e-01],
[-2.3925e+00, -2.9583e+00, -2.5855e+00, ..., -1.5962e+00,
-1.1997e+00, -8.9023e-01],
...,
[-3.2156e+00, -4.0104e+00, -3.7178e+00, ..., -2.8682e+00,
-2.2867e+00, -1.7619e+00],
[-2.1295e+00, -2.7508e+00, -2.6618e+00, ..., -1.9894e+00,
-1.6329e+00, -1.3179e+00],
[-1.5132e+00, -1.8399e+00, -1.8568e+00, ..., -1.2422e+00,
-1.0443e+00, -8.0304e-01]]],
[[[ 1.5216e-01, 3.9175e-01, -1.1533e+00, ..., 4.2349e-01,
9.7172e-01, 6.1802e-01],
[-2.0952e+00, -1.4239e+00, -4.3338e+00, ..., -2.5615e+00,
-7.8100e-01, -1.1246e-01],
[ 4.3685e-01, -3.0759e-01, -1.0937e+00, ..., -2.8858e-01,
4.7173e-01, 6.1915e-01],
...,
[-2.6597e-01, -1.2655e+00, -2.6020e+00, ..., -2.2371e+00,
-1.5780e+00, -1.1006e+00],
[ 1.2470e+00, 6.5232e-01, -8.9743e-01, ..., -3.9438e-01,
5.3589e-01, 7.0893e-01],
[ 8.2490e-01, 3.1489e-01, -3.9887e-01, ..., -4.1684e-01,
3.1878e-01, 2.8497e-01]],
[[-3.2307e+00, -1.4197e+00, -6.1071e+00, ..., -4.7564e+00,
-4.3252e+00, -3.3061e+00],
[-5.5378e-01, 2.1873e+00, -4.1656e+00, ..., -3.4040e+00,
-3.4952e+00, -2.8855e+00],
[ 5.9508e-02, 1.8784e+00, -7.3942e+00, ..., -5.3153e+00,
-5.1769e+00, -4.8516e+00],
...,
[-1.2918e+00, 2.0816e-01, -6.5048e+00, ..., -4.4926e+00,
-4.1082e+00, -3.3716e+00],
[ 1.4875e+00, 2.5754e+00, -1.2997e+00, ..., -5.1426e-01,
-9.6355e-01, -1.0232e+00],
[ 9.7319e-01, 1.6131e+00, -2.3817e+00, ..., -2.1384e+00,
-2.4616e+00, -2.3673e+00]],
[[-1.3729e+00, -2.3501e+00, -2.1762e+00, ..., -5.5188e-01,
-1.8118e-01, -4.4652e-01],
[-2.0253e+00, -2.2028e+00, -1.5608e+00, ..., -3.1436e-01,
5.4516e-02, 1.4278e-02],
[-2.5407e+00, -2.0051e+00, -8.2904e-01, ..., -2.6719e-01,
2.3818e-01, 2.4899e-01],
...,
[-1.5534e+00, -1.7954e+00, -7.6846e-01, ..., -5.1386e-01,
-1.6032e-01, -3.7508e-01],
[-1.7152e+00, -1.1554e+00, -7.6090e-01, ..., -3.1360e-01,
1.8951e-01, 2.3104e-01],
[-1.6866e+00, -1.0878e+00, -5.7039e-01, ..., -5.2453e-01,
-2.6196e-01, -2.4185e-01]],
...,
[[-2.3659e+00, -1.2963e+00, -2.7592e+00, ..., -3.4917e+00,
-2.6888e+00, -1.7657e+00],
[-1.8587e+00, 2.5849e-01, -1.3231e+00, ..., -2.5692e+00,
-2.2880e+00, -1.7627e+00],
[-4.6420e-01, 2.5688e+00, 9.3795e-01, ..., -5.2719e-01,
4.6448e-03, 2.1971e-01],
...,
[-1.7479e+00, 1.1282e+00, 2.0053e-01, ..., -1.6029e+00,
-7.3310e-01, -9.5757e-02],
[-9.1123e-01, 1.6020e+00, 1.4737e+00, ..., -1.2111e-01,
1.1979e-01, 4.2834e-01],
[-2.1589e+00, -1.1036e+00, -9.0552e-01, ..., -2.5578e+00,
-1.5936e+00, -9.5760e-01]],
[[-2.3545e+00, -2.6033e+00, -4.0252e+00, ..., -1.1913e+00,
-7.7874e-01, -7.8412e-01],
[-1.6303e+00, -1.2885e+00, -4.7817e+00, ..., -1.4049e+00,
-1.2264e+00, -1.5750e+00],
[-2.9857e+00, -2.4029e+00, -6.9245e+00, ..., -3.2618e+00,
-2.7461e+00, -2.2030e+00],
...,
[-2.5259e+00, -1.7410e+00, -4.9357e+00, ..., -1.3298e+00,
-9.2473e-01, -8.1845e-01],
[-2.2183e+00, -1.5486e+00, -3.8706e+00, ..., -1.6450e+00,
-1.0541e+00, -7.0740e-01],
[-2.3004e+00, -1.7727e+00, -3.0920e+00, ..., -2.8461e+00,
-2.0198e+00, -1.8844e+00]],
[[-1.8894e+00, -2.6144e+00, -2.3892e+00, ..., -1.7789e+00,
-1.4825e+00, -1.1301e+00],
[-1.9151e+00, -2.6274e+00, -2.0738e+00, ..., -1.4185e+00,
-1.1262e+00, -9.0238e-01],
[-2.4392e+00, -2.9933e+00, -2.4905e+00, ..., -1.6461e+00,
-1.2410e+00, -9.3101e-01],
...,
[-3.0907e+00, -3.8641e+00, -3.6492e+00, ..., -2.9814e+00,
-2.3919e+00, -1.8475e+00],
[-2.0495e+00, -2.6533e+00, -2.6198e+00, ..., -2.0547e+00,
-1.6972e+00, -1.3791e+00],
[-1.4682e+00, -1.7886e+00, -1.8390e+00, ..., -1.2606e+00,
-1.0705e+00, -8.2628e-01]]],
[[[ 2.0712e-01, 4.7755e-01, 6.0369e-01, ..., -3.5761e-01,
-9.5933e-02, 5.4075e-03],
[-2.0165e+00, -1.2958e+00, -2.8290e+00, ..., -3.2685e+00,
-1.3194e+00, -4.5434e-01],
[-1.5829e-01, -8.5776e-01, 9.5916e-03, ..., -1.7970e-01,
-3.6302e-01, 1.5788e-01],
...,
[-9.8905e-01, -2.1403e+00, -2.0174e+00, ..., -1.9741e+00,
-1.8905e+00, -1.2729e+00],
[ 7.7818e-01, -1.1466e-01, -7.7689e-02, ..., 1.2249e-03,
2.6813e-01, 4.5563e-01],
[ 1.9550e-01, -5.8018e-01, -3.7007e-01, ..., 2.0628e-01,
4.6363e-01, 5.0565e-01]],
[[-1.6672e+00, -1.3821e+00, -4.2484e+00, ..., -6.8570e+00,
-6.0737e+00, -4.5346e+00],
[ 4.9164e-01, 1.4319e+00, -2.5549e+00, ..., -5.4314e+00,
-5.5494e+00, -4.3014e+00],
[ 1.2095e+00, 1.1375e+00, -5.1078e+00, ..., -7.8282e+00,
-7.8086e+00, -6.6128e+00],
...,
[ 3.7689e-01, 3.4820e-01, -4.1135e+00, ..., -7.1417e+00,
-6.6809e+00, -5.2175e+00],
[ 2.5556e+00, 2.5045e+00, 1.3804e-01, ..., -1.9213e+00,
-2.5450e+00, -2.0771e+00],
[ 1.5168e+00, 1.3313e+00, -1.6237e+00, ..., -2.9272e+00,
-3.4792e+00, -3.0653e+00]],
[[-1.3264e-01, -1.1584e+00, -1.3961e+00, ..., -1.4035e+00,
-1.0466e+00, -1.3431e+00],
[-4.7926e-01, -7.8867e-01, -8.6285e-01, ..., -1.0031e+00,
-6.2340e-01, -9.0327e-01],
[-3.0111e-01, -1.7262e-01, -4.1811e-01, ..., -1.1002e+00,
-5.9393e-01, -4.6797e-01],
...,
[-4.0569e-02, -6.1035e-01, -6.1609e-01, ..., -1.0539e+00,
-8.7899e-01, -1.2640e+00],
[-7.1035e-01, -2.3334e-01, -5.8424e-01, ..., -5.7979e-01,
-2.6101e-01, -4.2624e-01],
[-8.9562e-01, -2.8379e-01, -4.8630e-01, ..., -5.1429e-01,
-4.5919e-01, -6.1985e-01]],
...,
[[-2.8286e+00, -2.4496e+00, -2.4014e+00, ..., -3.0939e+00,
-2.5169e+00, -1.8270e+00],
[-2.1274e+00, -9.0340e-01, -1.0965e+00, ..., -2.4375e+00,
-2.5305e+00, -2.0432e+00],
[-1.1699e+00, 6.8546e-01, 7.4096e-01, ..., -1.0088e-01,
2.3631e-02, 2.6901e-01],
...,
[-2.2996e+00, -2.5631e-01, -2.0691e-01, ..., -8.8687e-01,
-2.6154e-01, 2.8794e-01],
[-1.2325e+00, 8.4530e-01, 1.0297e+00, ..., 4.6809e-01,
5.0839e-01, 8.1440e-01],
[-2.1558e+00, -1.4858e+00, -1.6994e+00, ..., -2.1947e+00,
-1.1624e+00, -3.5102e-01]],
[[-1.6571e+00, -1.6667e+00, -2.7745e+00, ..., -2.2306e+00,
-1.9054e+00, -1.8080e+00],
[-7.2464e-01, -1.2638e-01, -3.1353e+00, ..., -2.8425e+00,
-2.8592e+00, -2.8894e+00],
[-2.0784e+00, -1.0096e+00, -5.0983e+00, ..., -4.4479e+00,
-4.5075e+00, -3.7098e+00],
...,
[-1.3148e+00, 1.5815e-01, -3.4187e+00, ..., -2.7980e+00,
-2.7279e+00, -2.3202e+00],
[-1.6569e+00, -4.7998e-01, -3.0507e+00, ..., -2.4206e+00,
-2.2548e+00, -1.6683e+00],
[-2.3073e+00, -1.3702e+00, -3.0553e+00, ..., -2.8627e+00,
-2.4378e+00, -2.1983e+00]],
[[-1.3101e+00, -1.8451e+00, -1.6725e+00, ..., -2.5790e+00,
-2.1522e+00, -1.6726e+00],
[-1.2505e+00, -1.7293e+00, -1.2594e+00, ..., -2.4006e+00,
-1.9339e+00, -1.5840e+00],
[-1.5710e+00, -1.9165e+00, -1.5240e+00, ..., -2.8146e+00,
-2.2463e+00, -1.7448e+00],
...,
[-2.4458e+00, -3.0524e+00, -2.7737e+00, ..., -4.0166e+00,
-3.2534e+00, -2.5490e+00],
[-1.5886e+00, -2.0843e+00, -1.9774e+00, ..., -2.7796e+00,
-2.3231e+00, -1.9177e+00],
[-1.0400e+00, -1.3490e+00, -1.3560e+00, ..., -1.8377e+00,
-1.5272e+00, -1.1808e+00]]],
[[[ 1.5360e-01, 3.8567e-01, -1.1645e+00, ..., 6.5203e-01,
9.2952e-01, 6.8766e-01],
[-2.0994e+00, -1.4361e+00, -4.3601e+00, ..., -1.5272e+00,
-1.8409e-01, 3.8962e-01],
[ 4.4256e-01, -2.7090e-01, -1.0960e+00, ..., 4.8660e-01,
9.3299e-01, 8.3596e-01],
...,
[-2.4151e-01, -1.2480e+00, -2.5490e+00, ..., -1.1932e+00,
-7.9551e-01, -6.0000e-01],
[ 1.2455e+00, 6.6563e-01, -8.7585e-01, ..., 5.2746e-01,
7.8276e-01, 8.3867e-01],
[ 8.1667e-01, 3.2228e-01, -3.8760e-01, ..., 5.4434e-01,
7.6742e-01, 6.8748e-01]],
[[-3.2315e+00, -1.4497e+00, -6.1237e+00, ..., -5.4460e+00,
-4.6558e+00, -3.6209e+00],
[-5.4299e-01, 2.1604e+00, -4.1695e+00, ..., -4.1915e+00,
-4.0794e+00, -3.3586e+00],
[ 6.0603e-02, 1.8482e+00, -7.3889e+00, ..., -6.1488e+00,
-5.9143e+00, -5.4929e+00],
...,
[-1.2223e+00, 2.9176e-01, -6.5117e+00, ..., -5.3931e+00,
-4.7155e+00, -3.9815e+00],
[ 1.5367e+00, 2.6176e+00, -1.3473e+00, ..., -1.3292e+00,
-1.4418e+00, -1.5570e+00],
[ 9.9355e-01, 1.6310e+00, -2.4057e+00, ..., -2.5151e+00,
-2.6114e+00, -2.5777e+00]],
[[-1.3868e+00, -2.3649e+00, -2.1817e+00, ..., -8.2550e-01,
-6.2257e-01, -8.1895e-01],
[-2.0286e+00, -2.2132e+00, -1.5748e+00, ..., -6.7462e-01,
-4.0558e-01, -3.5479e-01],
[-2.5343e+00, -1.9927e+00, -8.3566e-01, ..., -3.9207e-01,
-1.0672e-01, -1.1081e-01],
...,
[-1.5945e+00, -1.8247e+00, -7.9342e-01, ..., -6.4193e-01,
-5.5269e-01, -7.1182e-01],
[-1.7477e+00, -1.1716e+00, -7.6098e-01, ..., -2.2221e-01,
-6.0677e-02, -6.0516e-02],
[-1.6999e+00, -1.0960e+00, -5.8227e-01, ..., -3.2526e-01,
-2.8883e-01, -2.6683e-01]],
...,
[[-2.3788e+00, -1.3027e+00, -2.7538e+00, ..., -3.0817e+00,
-2.4707e+00, -1.6358e+00],
[-1.8440e+00, 2.5182e-01, -1.3074e+00, ..., -1.5181e+00,
-1.6192e+00, -1.3801e+00],
[-4.4548e-01, 2.5414e+00, 9.4538e-01, ..., -3.3378e-02,
2.3680e-01, 3.3118e-01],
...,
[-1.6947e+00, 1.1845e+00, 2.0516e-01, ..., -9.7313e-01,
-2.7392e-01, 1.6623e-01],
[-8.2867e-01, 1.6691e+00, 1.4873e+00, ..., 1.6261e-01,
3.3905e-01, 4.0586e-01],
[-2.1039e+00, -1.0590e+00, -8.7750e-01, ..., -2.1595e+00,
-1.2316e+00, -6.1659e-01]],
[[-2.3622e+00, -2.6097e+00, -4.0115e+00, ..., -1.0885e+00,
-6.2371e-01, -5.2386e-01],
[-1.6425e+00, -1.2972e+00, -4.7457e+00, ..., -1.6758e+00,
-1.5037e+00, -1.7592e+00],
[-3.0010e+00, -2.4206e+00, -6.8665e+00, ..., -3.7169e+00,
-3.2548e+00, -2.6463e+00],
...,
[-2.5000e+00, -1.7023e+00, -4.9375e+00, ..., -2.1764e+00,
-1.7524e+00, -1.4263e+00],
[-2.2125e+00, -1.5499e+00, -3.8760e+00, ..., -2.1100e+00,
-1.5174e+00, -9.8663e-01],
[-2.2993e+00, -1.7773e+00, -3.1019e+00, ..., -2.7901e+00,
-1.9562e+00, -1.6741e+00]],
[[-1.8949e+00, -2.6146e+00, -2.4037e+00, ..., -1.6929e+00,
-1.4020e+00, -1.0481e+00],
[-1.9216e+00, -2.6272e+00, -2.0941e+00, ..., -1.3672e+00,
-1.1080e+00, -8.8440e-01],
[-2.4456e+00, -2.9962e+00, -2.5106e+00, ..., -1.5804e+00,
-1.2081e+00, -8.8516e-01],
...,
[-3.1007e+00, -3.8705e+00, -3.6446e+00, ..., -2.7950e+00,
-2.2271e+00, -1.6870e+00],
[-2.0580e+00, -2.6545e+00, -2.6149e+00, ..., -1.9634e+00,
-1.6161e+00, -1.2859e+00],
[-1.4764e+00, -1.7899e+00, -1.8337e+00, ..., -1.2298e+00,
-1.0514e+00, -7.7802e-01]]]], device='cuda:0'),
tensor([[[[ 2.2670e-01, 4.8847e-01, 6.2971e-01, ..., -3.1891e-01,
-6.7757e-02, 1.7796e-02],
[-2.0477e+00, -1.3618e+00, -2.8835e+00, ..., -3.2283e+00,
-1.2858e+00, -4.4464e-01],
[-1.5703e-01, -9.0620e-01, 1.1115e-02, ..., -1.7822e-01,
-3.3749e-01, 1.6891e-01],
...,
[-9.7476e-01, -2.1824e+00, -2.0614e+00, ..., -1.8952e+00,
-1.8853e+00, -1.2686e+00],
[ 7.9749e-01, -1.3465e-01, -1.1987e-01, ..., 6.4880e-02,
2.8421e-01, 4.8355e-01],
[ 1.9451e-01, -5.8932e-01, -4.1619e-01, ..., 2.5064e-01,
4.7063e-01, 5.1489e-01]],
[[-1.6769e+00, -1.3723e+00, -4.2687e+00, ..., -6.8640e+00,
-6.0700e+00, -4.5258e+00],
[ 5.0653e-01, 1.5070e+00, -2.5149e+00, ..., -5.4623e+00,
-5.5585e+00, -4.2872e+00],
[ 1.2613e+00, 1.2316e+00, -5.0457e+00, ..., -7.8609e+00,
-7.8251e+00, -6.5960e+00],
...,
[ 4.2149e-01, 4.0432e-01, -4.0676e+00, ..., -7.1380e+00,
-6.6911e+00, -5.2318e+00],
[ 2.5840e+00, 2.5383e+00, 1.7501e-01, ..., -1.9711e+00,
-2.5822e+00, -2.1020e+00],
[ 1.5369e+00, 1.3471e+00, -1.6002e+00, ..., -2.9623e+00,
-3.4805e+00, -3.0518e+00]],
[[-1.4746e-01, -1.1910e+00, -1.4153e+00, ..., -1.4046e+00,
-1.0502e+00, -1.3342e+00],
[-5.1555e-01, -8.3757e-01, -8.9004e-01, ..., -9.9208e-01,
-6.1809e-01, -8.9529e-01],
[-3.2835e-01, -2.1421e-01, -4.4881e-01, ..., -1.1051e+00,
-5.8311e-01, -4.6314e-01],
...,
[-5.3074e-02, -5.7509e-01, -5.9359e-01, ..., -1.0349e+00,
-8.7619e-01, -1.2556e+00],
[-7.1134e-01, -2.0865e-01, -5.6549e-01, ..., -5.6238e-01,
-2.6521e-01, -4.2961e-01],
[-8.8596e-01, -2.6453e-01, -4.7629e-01, ..., -4.8195e-01,
-4.3249e-01, -6.0739e-01]],
...,
[[-2.8707e+00, -2.4775e+00, -2.4862e+00, ..., -3.0994e+00,
-2.5194e+00, -1.8313e+00],
[-2.1253e+00, -8.4937e-01, -1.1330e+00, ..., -2.4253e+00,
-2.5044e+00, -2.0173e+00],
[-1.1422e+00, 7.8245e-01, 7.8062e-01, ..., -9.8526e-02,
1.2105e-02, 2.8217e-01],
...,
[-2.3083e+00, -1.8570e-01, -1.5510e-01, ..., -9.1535e-01,
-2.4205e-01, 2.8221e-01],
[-1.3015e+00, 8.7148e-01, 1.1040e+00, ..., 4.2587e-01,
5.1676e-01, 8.1387e-01],
[-2.2228e+00, -1.5269e+00, -1.6616e+00, ..., -2.2065e+00,
-1.1555e+00, -3.4707e-01]],
[[-1.6361e+00, -1.6356e+00, -2.7968e+00, ..., -2.2271e+00,
-1.8952e+00, -1.7961e+00],
[-6.8754e-01, -7.0385e-02, -3.1436e+00, ..., -2.8523e+00,
-2.8574e+00, -2.8796e+00],
[-2.0671e+00, -9.6409e-01, -5.1376e+00, ..., -4.4662e+00,
-4.5120e+00, -3.7052e+00],
...,
[-1.3304e+00, 2.0917e-01, -3.3719e+00, ..., -2.8746e+00,
-2.7708e+00, -2.3169e+00],
[-1.6583e+00, -4.4334e-01, -3.0160e+00, ..., -2.4765e+00,
-2.2963e+00, -1.6580e+00],
[-2.3233e+00, -1.3329e+00, -3.0347e+00, ..., -2.8831e+00,
-2.4524e+00, -2.1869e+00]],
[[-1.3079e+00, -1.8417e+00, -1.6587e+00, ..., -2.5661e+00,
-2.1412e+00, -1.6648e+00],
[-1.2467e+00, -1.7112e+00, -1.2274e+00, ..., -2.3899e+00,
-1.9236e+00, -1.5774e+00],
[-1.5677e+00, -1.8973e+00, -1.4795e+00, ..., -2.8038e+00,
-2.2350e+00, -1.7324e+00],
...,
[-2.4643e+00, -3.0799e+00, -2.7921e+00, ..., -4.0069e+00,
-3.2406e+00, -2.5402e+00],
[-1.5967e+00, -2.1057e+00, -1.9924e+00, ..., -2.7761e+00,
-2.3167e+00, -1.9124e+00],
[-1.0426e+00, -1.3594e+00, -1.3720e+00, ..., -1.8308e+00,
-1.5249e+00, -1.1796e+00]]],
[[[ 2.8426e-01, 5.4541e-01, 6.5932e-01, ..., -1.4874e+00,
6.3277e-01, 5.9603e-01],
[-2.0981e+00, -1.4849e+00, -3.1330e+00, ..., -4.7820e+00,
-1.3204e+00, -2.1378e-01],
[-1.7144e-01, -9.7695e-01, -2.1076e-01, ..., -2.4471e+00,
1.7901e-01, 6.2311e-01],
...,
[-2.1927e-01, -1.1634e+00, -2.6747e+00, ..., -1.5154e+00,
-1.4177e+00, -8.6511e-01],
[ 1.4353e+00, 8.8273e-01, -9.7127e-01, ..., -3.1610e-01,
6.5129e-02, 3.5158e-01],
[ 8.5675e-01, 2.9873e-01, -6.9514e-01, ..., -1.9677e-01,
1.9872e-01, 3.7039e-01]],
[[-1.6819e+00, -1.2376e+00, -4.1958e+00, ..., -5.7432e+00,
-4.7690e+00, -3.4707e+00],
[ 4.7046e-01, 1.7046e+00, -2.3746e+00, ..., -3.8530e+00,
-3.6105e+00, -2.9380e+00],
[ 1.1491e+00, 1.4432e+00, -4.9578e+00, ..., -5.6147e+00,
-5.2324e+00, -5.0140e+00],
...,
[-1.4040e+00, 5.3391e-02, -6.7417e+00, ..., -7.6408e+00,
-6.9412e+00, -5.4398e+00],
[ 1.7056e+00, 2.7659e+00, -1.0509e+00, ..., -2.5839e+00,
-2.9428e+00, -2.3716e+00],
[ 1.1746e+00, 1.7944e+00, -2.3002e+00, ..., -3.2457e+00,
-3.6245e+00, -3.1427e+00]],
[[-9.3117e-02, -1.0652e+00, -1.2496e+00, ..., -7.9447e-01,
-3.0327e-01, -5.3856e-01],
[-5.1322e-01, -7.5843e-01, -7.1911e-01, ..., -5.2804e-01,
-1.5777e-01, -7.4062e-02],
[-4.6405e-01, -2.8756e-01, -3.7648e-01, ..., -4.7124e-01,
-1.7144e-02, 1.0152e-01],
...,
[-1.3313e+00, -1.7472e+00, -7.9812e-01, ..., -8.8704e-01,
-7.7733e-01, -1.0390e+00],
[-1.6813e+00, -1.1453e+00, -8.4959e-01, ..., -5.5783e-01,
-2.9137e-01, -3.6087e-01],
[-1.8586e+00, -1.2862e+00, -9.0850e-01, ..., -9.6579e-01,
-8.3728e-01, -8.1834e-01]],
...,
[[-2.9285e+00, -2.4768e+00, -2.7452e+00, ..., -3.3177e+00,
-2.8603e+00, -1.8713e+00],
[-2.2608e+00, -9.1618e-01, -1.5926e+00, ..., -2.5790e+00,
-2.4178e+00, -1.9118e+00],
[-1.1082e+00, 9.9568e-01, 5.0982e-01, ..., -4.6728e-01,
6.0424e-02, 2.0144e-01],
...,
[-1.8982e+00, 1.1699e+00, -8.7415e-03, ..., -1.1112e+00,
-5.4425e-01, -3.5615e-03],
[-8.5022e-01, 1.8718e+00, 1.6375e+00, ..., 4.9305e-01,
4.2862e-01, 6.8777e-01],
[-2.2597e+00, -1.1596e+00, -9.7873e-01, ..., -1.8752e+00,
-9.7814e-01, -3.0246e-01]],
[[-1.6121e+00, -1.6062e+00, -2.7921e+00, ..., -7.4593e-01,
-6.4501e-01, -7.2769e-01],
[-6.6672e-01, -1.8198e-02, -3.0853e+00, ..., -1.3900e+00,
-1.0450e+00, -1.5596e+00],
[-2.1246e+00, -1.0001e+00, -5.1587e+00, ..., -3.4058e+00,
-2.7883e+00, -2.2587e+00],
...,
[-2.5034e+00, -1.6360e+00, -5.4428e+00, ..., -3.0568e+00,
-2.7284e+00, -2.1675e+00],
[-2.2326e+00, -1.4879e+00, -4.2490e+00, ..., -2.6705e+00,
-2.3060e+00, -1.6379e+00],
[-2.3877e+00, -1.7913e+00, -3.3317e+00, ..., -2.9746e+00,
-2.4411e+00, -2.1310e+00]],
[[-1.3195e+00, -1.8726e+00, -1.7027e+00, ..., -2.2140e+00,
-1.5792e+00, -1.1546e+00],
[-1.2498e+00, -1.7341e+00, -1.2568e+00, ..., -1.8728e+00,
-1.2112e+00, -8.8149e-01],
[-1.5653e+00, -1.9145e+00, -1.5058e+00, ..., -2.0083e+00,
-1.2842e+00, -8.7411e-01],
...,
[-3.1672e+00, -3.9694e+00, -3.7742e+00, ..., -3.9737e+00,
-3.1859e+00, -2.4525e+00],
[-2.1048e+00, -2.7340e+00, -2.7067e+00, ..., -2.7715e+00,
-2.2901e+00, -1.8615e+00],
[-1.5405e+00, -1.8728e+00, -1.9371e+00, ..., -1.8679e+00,
-1.5174e+00, -1.1611e+00]]],
[[[ 1.2828e-01, 3.6563e-01, -1.1719e+00, ..., 4.4983e-01,
1.0072e+00, 6.2918e-01],
[-2.1107e+00, -1.4182e+00, -4.3147e+00, ..., -2.6065e+00,
-7.4524e-01, -8.2975e-02],
[ 4.3079e-01, -2.4535e-01, -1.0408e+00, ..., -2.6048e-01,
5.4001e-01, 6.6044e-01],
...,
[-2.6117e-01, -1.2646e+00, -2.6184e+00, ..., -2.0253e+00,
-1.3760e+00, -8.7696e-01],
[ 1.2536e+00, 6.8750e-01, -9.0504e-01, ..., -3.5848e-01,
5.3060e-01, 7.1581e-01],
[ 8.1771e-01, 3.4493e-01, -4.1280e-01, ..., -4.0676e-01,
2.8121e-01, 2.6233e-01]],
[[-3.2144e+00, -1.4213e+00, -6.0993e+00, ..., -4.8500e+00,
-4.4333e+00, -3.3723e+00],
[-5.4332e-01, 2.1770e+00, -4.1850e+00, ..., -3.3718e+00,
-3.5293e+00, -2.8930e+00],
[ 7.1536e-02, 1.8701e+00, -7.4236e+00, ..., -5.3415e+00,
-5.2556e+00, -4.9461e+00],
...,
[-1.2457e+00, 2.6531e-01, -6.4729e+00, ..., -4.5530e+00,
-4.0962e+00, -3.4209e+00],
[ 1.5481e+00, 2.6334e+00, -1.2751e+00, ..., -9.3307e-01,
-1.2335e+00, -1.2866e+00],
[ 1.0102e+00, 1.6487e+00, -2.3669e+00, ..., -2.2711e+00,
-2.4928e+00, -2.4045e+00]],
[[-1.3487e+00, -2.3288e+00, -2.1722e+00, ..., -5.8859e-01,
-2.1645e-01, -4.7939e-01],
[-1.9876e+00, -2.1851e+00, -1.5813e+00, ..., -3.3000e-01,
5.5595e-02, 1.9480e-02],
[-2.5123e+00, -2.0014e+00, -8.6326e-01, ..., -2.6134e-01,
2.6155e-01, 2.7917e-01],
...,
[-1.6011e+00, -1.8259e+00, -7.8338e-01, ..., -4.6037e-01,
-1.4111e-01, -3.3226e-01],
[-1.7549e+00, -1.1730e+00, -7.6522e-01, ..., -3.0967e-01,
1.5134e-01, 2.0126e-01],
[-1.7174e+00, -1.0939e+00, -5.9120e-01, ..., -5.2333e-01,
-2.9058e-01, -2.5484e-01]],
...,
[[-2.3729e+00, -1.3235e+00, -2.7571e+00, ..., -3.6884e+00,
-2.8383e+00, -1.8501e+00],
[-1.9062e+00, 1.8920e-01, -1.3238e+00, ..., -2.6678e+00,
-2.4047e+00, -1.8401e+00],
[-5.2068e-01, 2.4784e+00, 9.5009e-01, ..., -5.6355e-01,
-5.9910e-02, 1.7572e-01],
...,
[-1.7163e+00, 1.1709e+00, 2.2957e-01, ..., -1.5033e+00,
-7.3806e-01, -1.5197e-01],
[-8.5722e-01, 1.6418e+00, 1.5228e+00, ..., -1.3231e-01,
6.7858e-02, 3.3499e-01],
[-2.1142e+00, -1.1102e+00, -8.8289e-01, ..., -2.2728e+00,
-1.4220e+00, -8.6661e-01]],
[[-2.3621e+00, -2.6191e+00, -4.0523e+00, ..., -1.1361e+00,
-7.1162e-01, -7.3147e-01],
[-1.6431e+00, -1.3245e+00, -4.8165e+00, ..., -1.3522e+00,
-1.2007e+00, -1.5963e+00],
[-2.9818e+00, -2.4377e+00, -6.9495e+00, ..., -3.2903e+00,
-2.7993e+00, -2.2730e+00],
...,
[-2.5112e+00, -1.7214e+00, -4.9517e+00, ..., -1.4459e+00,
-1.0577e+00, -9.4390e-01],
[-2.2142e+00, -1.5403e+00, -3.8844e+00, ..., -1.6660e+00,
-1.1479e+00, -8.3118e-01],
[-2.3102e+00, -1.7785e+00, -3.0867e+00, ..., -2.7308e+00,
-2.0099e+00, -1.8750e+00]],
[[-1.8996e+00, -2.6234e+00, -2.3952e+00, ..., -1.8114e+00,
-1.5117e+00, -1.1553e+00],
[-1.9302e+00, -2.6461e+00, -2.1008e+00, ..., -1.4149e+00,
-1.1239e+00, -9.0118e-01],
[-2.4650e+00, -3.0086e+00, -2.5213e+00, ..., -1.6135e+00,
-1.2121e+00, -8.9884e-01],
...,
[-3.0996e+00, -3.8739e+00, -3.6469e+00, ..., -2.8826e+00,
-2.2975e+00, -1.7697e+00],
[-2.0587e+00, -2.6596e+00, -2.6220e+00, ..., -2.0064e+00,
-1.6492e+00, -1.3291e+00],
[-1.4801e+00, -1.7921e+00, -1.8450e+00, ..., -1.2585e+00,
-1.0568e+00, -8.1218e-01]]],
[[[ 2.3289e-01, 5.1135e-01, 6.3221e-01, ..., 5.9849e-01,
9.4903e-01, 7.6480e-01],
[-1.9700e+00, -1.2642e+00, -2.7668e+00, ..., -1.3131e+00,
7.4639e-02, 6.0006e-01],
[-1.0569e-01, -8.3176e-01, 3.5430e-02, ..., 4.7144e-01,
9.1508e-01, 9.4789e-01],
...,
[-9.5143e-01, -2.1006e+00, -2.0122e+00, ..., -1.4493e+00,
-9.2231e-01, -6.5846e-01],
[ 8.0582e-01, -7.6805e-02, -9.2104e-02, ..., 3.6664e-01,
7.9559e-01, 8.8150e-01],
[ 2.4720e-01, -5.0344e-01, -3.5242e-01, ..., 5.7566e-01,
8.9098e-01, 7.9056e-01]],
[[-1.7154e+00, -1.3679e+00, -4.2647e+00, ..., -5.8140e+00,
-4.9286e+00, -3.8516e+00],
[ 4.4307e-01, 1.4862e+00, -2.5496e+00, ..., -4.7949e+00,
-4.5070e+00, -3.7040e+00],
[ 1.1768e+00, 1.2235e+00, -5.0926e+00, ..., -6.8916e+00,
-6.4168e+00, -5.8440e+00],
...,
[ 3.0162e-01, 3.2644e-01, -4.1347e+00, ..., -6.0698e+00,
-5.2612e+00, -4.3147e+00],
[ 2.5127e+00, 2.5027e+00, 1.5477e-01, ..., -1.3676e+00,
-1.5160e+00, -1.5626e+00],
[ 1.5036e+00, 1.3376e+00, -1.6104e+00, ..., -2.7599e+00,
-2.8831e+00, -2.8046e+00]],
[[-1.6198e-01, -1.2056e+00, -1.4176e+00, ..., -8.0127e-01,
-5.4919e-01, -6.6582e-01],
[-5.5808e-01, -8.9549e-01, -9.1369e-01, ..., -6.8231e-01,
-3.6710e-01, -2.3398e-01],
[-3.8498e-01, -2.8830e-01, -4.6012e-01, ..., -3.9205e-01,
-1.4391e-02, 8.4302e-02],
...,
[-7.9450e-02, -6.5739e-01, -6.2478e-01, ..., -7.0419e-01,
-4.9774e-01, -5.8365e-01],
[-7.2568e-01, -2.7187e-01, -5.8358e-01, ..., -3.2765e-01,
-2.9629e-02, 5.4717e-02],
[-9.0433e-01, -3.0473e-01, -4.8351e-01, ..., -3.3006e-01,
-2.1883e-01, -1.5360e-01]],
...,
[[-2.8259e+00, -2.4020e+00, -2.4532e+00, ..., -3.1094e+00,
-2.5608e+00, -1.7221e+00],
[-2.0354e+00, -7.3676e-01, -1.0897e+00, ..., -1.6610e+00,
-1.7559e+00, -1.4592e+00],
[-1.0826e+00, 8.8300e-01, 7.7329e-01, ..., -2.5613e-01,
-8.2746e-03, 1.4911e-01],
...,
[-2.2593e+00, -1.4256e-01, -1.6376e-01, ..., -1.0542e+00,
-3.8214e-01, 1.9160e-01],
[-1.2066e+00, 9.0523e-01, 1.0660e+00, ..., 2.5126e-01,
3.4956e-01, 4.7285e-01],
[-2.1353e+00, -1.4513e+00, -1.6377e+00, ..., -2.2037e+00,
-1.2436e+00, -6.1284e-01]],
[[-1.6141e+00, -1.6052e+00, -2.7907e+00, ..., -1.4327e+00,
-8.4176e-01, -5.8854e-01],
[-6.8663e-01, -9.2370e-02, -3.1773e+00, ..., -2.0937e+00,
-1.7112e+00, -1.7610e+00],
[-2.0638e+00, -1.0037e+00, -5.1718e+00, ..., -4.0316e+00,
-3.3925e+00, -2.5998e+00],
...,
[-1.3349e+00, 1.0260e-01, -3.4454e+00, ..., -2.4212e+00,
-1.8269e+00, -1.3749e+00],
[-1.6582e+00, -4.9536e-01, -3.0532e+00, ..., -2.2798e+00,
-1.5172e+00, -8.9104e-01],
[-2.3110e+00, -1.3576e+00, -3.0193e+00, ..., -2.8896e+00,
-1.9454e+00, -1.6295e+00]],
[[-1.2945e+00, -1.8231e+00, -1.6377e+00, ..., -1.8803e+00,
-1.5288e+00, -1.1154e+00],
[-1.2350e+00, -1.6982e+00, -1.2144e+00, ..., -1.6635e+00,
-1.3193e+00, -1.0025e+00],
[-1.5531e+00, -1.8886e+00, -1.4621e+00, ..., -1.9591e+00,
-1.4732e+00, -1.0405e+00],
...,
[-2.4265e+00, -3.0318e+00, -2.7631e+00, ..., -3.2130e+00,
-2.5288e+00, -1.8825e+00],
[-1.5723e+00, -2.0740e+00, -1.9694e+00, ..., -2.2684e+00,
-1.8407e+00, -1.4479e+00],
[-1.0312e+00, -1.3405e+00, -1.3587e+00, ..., -1.4096e+00,
-1.1807e+00, -8.6078e-01]]],
[[[-6.0304e+00, -1.4736e+01, -1.7942e+01, ..., -1.0434e+01,
-7.7271e+00, -4.1377e+00],
[-1.9483e+01, -2.6205e+01, -2.7023e+01, ..., -1.7672e+01,
-2.0099e+01, -1.6706e+01],
[-2.2749e+01, -2.8860e+01, -2.8940e+01, ..., -1.8718e+01,
-2.4770e+01, -1.9284e+01],
...,
[-2.9049e+00, -3.1201e+00, -4.0561e+00, ..., -1.9972e+00,
-2.0032e+00, -1.2474e+00],
[ 1.5901e-01, -5.0296e-01, -1.8178e+00, ..., -2.5505e-01,
9.7909e-03, 3.0028e-01],
[-8.0200e-01, -2.2460e+00, -3.2294e+00, ..., -9.0872e-02,
2.4712e-01, 3.9665e-01]],
[[-2.1963e+00, -2.1736e+00, -9.3855e+00, ..., -6.9382e+00,
-1.0641e+01, -8.1814e+00],
[-2.0943e+00, -8.3996e-01, -1.0693e+01, ..., -5.3411e+00,
-1.2772e+01, -1.0412e+01],
[-4.1972e+00, -8.7846e+00, -2.4049e+01, ..., -8.5790e+00,
-1.5970e+01, -1.1771e+01],
...,
[-1.1645e+00, -4.4484e-01, -5.4949e+00, ..., -7.7911e+00,
-7.4139e+00, -5.6661e+00],
[ 1.4447e+00, 2.4084e+00, -2.3569e-01, ..., -2.3407e+00,
-3.0676e+00, -2.3286e+00],
[ 1.3883e+00, 1.7161e+00, -1.6815e+00, ..., -3.1188e+00,
-3.7993e+00, -3.2077e+00]],
[[-1.7032e+00, -1.3262e+00, -2.0613e+00, ..., -3.2345e+00,
-3.9959e+00, -3.0553e+00],
[-1.6027e+00, 8.5752e-02, -1.7646e+00, ..., -4.3788e+00,
-7.4840e+00, -4.9188e+00],
[-1.8389e-01, -4.0877e-01, -4.0010e+00, ..., -5.6253e+00,
-6.6783e+00, -5.8334e+00],
...,
[ 2.9274e-01, -5.7571e-01, -5.7387e-01, ..., -1.0578e+00,
-9.1411e-01, -1.3185e+00],
[-1.9041e-01, -1.4481e-01, -7.9483e-01, ..., -5.7019e-01,
-2.5628e-01, -4.3565e-01],
[-6.2819e-01, -1.6090e-01, -4.6000e-01, ..., -7.6394e-01,
-7.3396e-01, -8.8359e-01]],
...,
[[-3.6910e+00, -2.8190e+00, -2.9859e+00, ..., -8.8592e+00,
-1.2598e+01, -5.6870e+00],
[-1.0370e+01, -8.1969e+00, -5.3863e+00, ..., -1.4279e+01,
-2.2587e+01, -8.2287e+00],
[-1.7867e+01, -1.4130e+01, -1.0669e+01, ..., -1.6361e+01,
-1.4216e+01, -1.4494e+00],
...,
[-2.5367e+00, -1.1301e+00, -6.3367e-01, ..., -9.4851e-01,
-3.3647e-01, 2.0558e-01],
[-3.2002e+00, -7.3921e-01, 2.0475e-01, ..., 6.7352e-01,
5.9161e-01, 9.0869e-01],
[-3.8399e+00, -2.5696e+00, -1.7541e+00, ..., -1.9881e+00,
-9.7214e-01, -1.9785e-01]],
[[-8.2477e-01, -1.0665e-01, 1.0128e+00, ..., 1.2106e+00,
1.2055e+00, -7.0930e-01],
[ 3.5265e+00, 5.9607e+00, 4.9080e+00, ..., -1.4997e+00,
-2.2858e+00, -2.2140e+00],
[ 6.0389e+00, 6.1707e+00, 2.0891e-01, ..., -1.2994e+01,
-8.7101e+00, -4.0526e+00],
...,
[-1.0392e+00, -1.8331e-01, -4.1606e+00, ..., -2.9826e+00,
-2.9648e+00, -2.4812e+00],
[-9.5111e-01, -2.3539e-01, -2.9139e+00, ..., -2.5680e+00,
-2.4977e+00, -1.8629e+00],
[-1.3110e+00, -4.3098e-01, -2.2153e+00, ..., -2.9584e+00,
-2.6295e+00, -2.3676e+00]],
[[-2.3303e+00, -4.1644e+00, -3.9458e+00, ..., -4.6264e+00,
-3.8895e+00, -2.4285e+00],
[-3.3215e+00, -6.2475e+00, -5.5935e+00, ..., -7.2368e+00,
-6.5007e+00, -3.5736e+00],
[-4.4975e+00, -8.0179e+00, -7.4510e+00, ..., -1.1186e+01,
-9.9970e+00, -5.1640e+00],
...,
[-2.5699e+00, -3.0768e+00, -2.8516e+00, ..., -4.3394e+00,
-3.5180e+00, -2.7653e+00],
[-1.5636e+00, -2.0632e+00, -1.9744e+00, ..., -2.9830e+00,
-2.4927e+00, -2.0660e+00],
[-1.2626e+00, -1.4514e+00, -1.4690e+00, ..., -1.9979e+00,
-1.6355e+00, -1.2746e+00]]]], device='cuda:0'),
tensor([[[[ 3.1484e-01, 6.0167e-01, 5.8440e-01, ..., 5.2642e-01,
1.0235e+00, 6.1980e-01],
[-1.9732e+00, -1.2812e+00, -3.2076e+00, ..., -2.5750e+00,
-7.9636e-01, -1.2148e-01],
[-7.3884e-02, -7.6094e-01, -7.1858e-01, ..., -1.6295e-02,
6.2878e-01, 6.9146e-01],
...,
[-2.1178e-01, -1.2132e+00, -2.7914e+00, ..., -1.5149e+00,
-1.4597e+00, -8.5020e-01],
[ 1.4585e+00, 7.9277e-01, -1.1656e+00, ..., -4.2702e-01,
-5.2938e-02, 2.8974e-01],
[ 8.0682e-01, 1.6044e-01, -9.0253e-01, ..., -3.4008e-01,
9.4559e-02, 3.1620e-01]],
[[-1.7435e+00, -1.2234e+00, -3.9757e+00, ..., -5.0916e+00,
-4.5799e+00, -3.5198e+00],
[ 3.5100e-01, 1.6718e+00, -2.1949e+00, ..., -3.5653e+00,
-3.6650e+00, -3.0283e+00],
[ 1.0179e+00, 1.4114e+00, -4.7881e+00, ..., -5.5481e+00,
-5.4093e+00, -5.0976e+00],
...,
[-1.4181e+00, 7.3072e-02, -7.0249e+00, ..., -7.9251e+00,
-7.2477e+00, -5.6301e+00],
[ 1.7131e+00, 2.8191e+00, -1.2133e+00, ..., -2.7738e+00,
-3.1763e+00, -2.4796e+00],
[ 1.1859e+00, 1.8609e+00, -2.3487e+00, ..., -3.3375e+00,
-3.7540e+00, -3.1881e+00]],
[[-1.4991e-01, -1.1210e+00, -1.2408e+00, ..., -5.8359e-01,
-2.5157e-01, -5.6248e-01],
[-6.0544e-01, -8.7213e-01, -7.8906e-01, ..., -4.0624e-01,
-7.2204e-02, -1.2266e-01],
[-5.4021e-01, -3.8464e-01, -4.4173e-01, ..., -3.6703e-01,
7.9141e-02, 7.9172e-02],
...,
[-1.5245e+00, -1.8075e+00, -7.4473e-01, ..., -8.7709e-01,
-7.7247e-01, -1.0550e+00],
[-1.8375e+00, -1.1953e+00, -8.0442e-01, ..., -5.3589e-01,
-2.7550e-01, -3.5602e-01],
[-1.9946e+00, -1.3964e+00, -9.7930e-01, ..., -1.0909e+00,
-9.6440e-01, -9.3710e-01]],
...,
[[-2.8045e+00, -2.3073e+00, -2.5843e+00, ..., -3.6580e+00,
-2.7913e+00, -1.8569e+00],
[-2.0016e+00, -6.2418e-01, -1.5351e+00, ..., -2.5665e+00,
-2.3137e+00, -1.8506e+00],
[-9.3708e-01, 1.2048e+00, 4.0637e-01, ..., -4.5105e-01,
6.0831e-02, 2.0249e-01],
...,
[-1.9190e+00, 1.1743e+00, -3.6146e-02, ..., -1.1500e+00,
-5.7072e-01, -3.6912e-02],
[-8.0573e-01, 1.9756e+00, 1.7597e+00, ..., 5.4615e-01,
4.6370e-01, 7.3498e-01],
[-2.2499e+00, -1.1410e+00, -9.0679e-01, ..., -1.7944e+00,
-9.0419e-01, -2.3033e-01]],
[[-1.5404e+00, -1.5254e+00, -2.6129e+00, ..., -1.1240e+00,
-6.5767e-01, -6.6202e-01],
[-6.2900e-01, -2.2784e-02, -3.0005e+00, ..., -1.2882e+00,
-1.1419e+00, -1.5400e+00],
[-2.0786e+00, -1.0282e+00, -5.1129e+00, ..., -3.2856e+00,
-2.8080e+00, -2.2619e+00],
...,
[-2.5822e+00, -1.7183e+00, -5.5367e+00, ..., -3.1411e+00,
-2.8286e+00, -2.2479e+00],
[-2.2871e+00, -1.5802e+00, -4.3150e+00, ..., -2.7492e+00,
-2.4207e+00, -1.7346e+00],
[-2.4264e+00, -1.8574e+00, -3.3753e+00, ..., -3.0248e+00,
-2.5226e+00, -2.2063e+00]],
[[-1.2951e+00, -1.8416e+00, -1.7477e+00, ..., -1.8058e+00,
-1.5014e+00, -1.1347e+00],
[-1.2321e+00, -1.7123e+00, -1.3040e+00, ..., -1.3757e+00,
-1.0942e+00, -8.6801e-01],
[-1.5517e+00, -1.8970e+00, -1.5805e+00, ..., -1.5750e+00,
-1.1856e+00, -8.6374e-01],
...,
[-3.3096e+00, -4.1492e+00, -3.9279e+00, ..., -4.1211e+00,
-3.3127e+00, -2.5550e+00],
[-2.1976e+00, -2.8501e+00, -2.8115e+00, ..., -2.8717e+00,
-2.3747e+00, -1.9328e+00],
[-1.6096e+00, -1.9602e+00, -2.0096e+00, ..., -1.9477e+00,
-1.5716e+00, -1.2102e+00]]],
[[[ 1.3751e-01, 3.6322e-01, -1.1661e+00, ..., 4.3874e-01,
9.9775e-01, 6.4667e-01],
[-2.0938e+00, -1.4312e+00, -4.3045e+00, ..., -2.6145e+00,
-7.8341e-01, -8.3169e-02],
[ 4.2630e-01, -3.0974e-01, -1.1298e+00, ..., -2.7116e-01,
4.7608e-01, 6.2761e-01],
...,
[-2.6190e-01, -1.2880e+00, -2.5737e+00, ..., -1.9917e+00,
-1.4198e+00, -9.5349e-01],
[ 1.2644e+00, 6.4132e-01, -8.6497e-01, ..., -3.2139e-01,
4.8733e-01, 6.8604e-01],
[ 8.2616e-01, 2.8701e-01, -3.9433e-01, ..., -3.5623e-01,
2.7584e-01, 2.7023e-01]],
[[-3.2294e+00, -1.4188e+00, -6.0878e+00, ..., -4.8096e+00,
-4.4047e+00, -3.3600e+00],
[-5.8920e-01, 2.1509e+00, -4.2049e+00, ..., -3.3377e+00,
-3.5105e+00, -2.8871e+00],
[ 8.5981e-03, 1.8363e+00, -7.4325e+00, ..., -5.3227e+00,
-5.2402e+00, -4.9162e+00],
...,
[-1.2194e+00, 3.0561e-01, -6.5212e+00, ..., -4.5845e+00,
-4.0856e+00, -3.4181e+00],
[ 1.5267e+00, 2.6014e+00, -1.3972e+00, ..., -8.7015e-01,
-1.1504e+00, -1.2174e+00],
[ 1.0021e+00, 1.6235e+00, -2.4457e+00, ..., -2.2044e+00,
-2.4356e+00, -2.3550e+00]],
[[-1.3616e+00, -2.3240e+00, -2.1441e+00, ..., -5.8235e-01,
-2.0301e-01, -4.6851e-01],
[-2.0018e+00, -2.1793e+00, -1.5290e+00, ..., -3.4274e-01,
6.7045e-02, 3.7410e-02],
[-2.5193e+00, -2.0083e+00, -8.3539e-01, ..., -2.7040e-01,
2.6079e-01, 2.9247e-01],
...,
[-1.5946e+00, -1.8125e+00, -7.5665e-01, ..., -4.8864e-01,
-1.6288e-01, -3.5019e-01],
[-1.7711e+00, -1.1981e+00, -7.6209e-01, ..., -3.1257e-01,
1.5515e-01, 2.1463e-01],
[-1.7244e+00, -1.1112e+00, -5.8437e-01, ..., -4.9712e-01,
-2.7180e-01, -2.4907e-01]],
...,
[[-2.3385e+00, -1.2769e+00, -2.7022e+00, ..., -3.6579e+00,
-2.8248e+00, -1.8290e+00],
[-1.8593e+00, 2.2763e-01, -1.3185e+00, ..., -2.6541e+00,
-2.4218e+00, -1.8434e+00],
[-4.6138e-01, 2.5578e+00, 9.6219e-01, ..., -5.7743e-01,
-7.8987e-02, 1.5508e-01],
...,
[-1.6970e+00, 1.2183e+00, 1.9231e-01, ..., -1.5535e+00,
-6.8569e-01, -1.3537e-01],
[-8.5788e-01, 1.6826e+00, 1.4510e+00, ..., -1.6292e-01,
1.1116e-01, 3.2861e-01],
[-2.0831e+00, -1.0210e+00, -8.6191e-01, ..., -2.3842e+00,
-1.4610e+00, -9.1999e-01]],
[[-2.3516e+00, -2.6134e+00, -4.0042e+00, ..., -1.1238e+00,
-7.1278e-01, -7.5299e-01],
[-1.6436e+00, -1.3360e+00, -4.7742e+00, ..., -1.3399e+00,
-1.1943e+00, -1.5995e+00],
[-2.9834e+00, -2.4191e+00, -6.8959e+00, ..., -3.2646e+00,
-2.7808e+00, -2.2467e+00],
...,
[-2.5353e+00, -1.7409e+00, -5.0508e+00, ..., -1.4627e+00,
-1.0141e+00, -8.7003e-01],
[-2.2125e+00, -1.5553e+00, -3.9241e+00, ..., -1.7173e+00,
-1.1296e+00, -7.5765e-01],
[-2.2864e+00, -1.7606e+00, -3.0713e+00, ..., -2.7965e+00,
-2.0131e+00, -1.8470e+00]],
[[-1.8874e+00, -2.6093e+00, -2.3903e+00, ..., -1.8175e+00,
-1.5105e+00, -1.1481e+00],
[-1.9122e+00, -2.6264e+00, -2.0858e+00, ..., -1.4164e+00,
-1.1191e+00, -8.9495e-01],
[-2.4375e+00, -3.0005e+00, -2.5098e+00, ..., -1.6148e+00,
-1.2071e+00, -8.9585e-01],
...,
[-3.1128e+00, -3.8868e+00, -3.6568e+00, ..., -2.8640e+00,
-2.2902e+00, -1.7588e+00],
[-2.0793e+00, -2.6852e+00, -2.6576e+00, ..., -1.9695e+00,
-1.6200e+00, -1.3087e+00],
[-1.4974e+00, -1.8260e+00, -1.8815e+00, ..., -1.2170e+00,
-1.0334e+00, -7.8993e-01]]],
[[[ 5.5940e-02, 1.9206e-01, -1.4701e+00, ..., -6.2629e-01,
-2.2227e-01, -7.2567e-02],
[-2.1092e+00, -1.6637e+00, -4.5975e+00, ..., -3.7885e+00,
-1.5544e+00, -5.7980e-01],
[ 5.6545e-01, -1.2009e-01, -8.7383e-01, ..., -1.9524e-01,
-2.8805e-01, 2.2767e-01],
...,
[-8.5265e-01, -1.9249e+00, -2.0220e+00, ..., -1.7576e+00,
-1.2242e+00, -8.3927e-01],
[ 9.1216e-01, 8.4848e-02, -8.6829e-02, ..., -2.0603e-02,
7.3124e-01, 8.2164e-01],
[ 2.5952e-01, -4.9636e-01, -4.4056e-01, ..., -1.5929e-01,
4.3347e-01, 3.4561e-01]],
[[-3.3160e+00, -1.7975e+00, -6.2949e+00, ..., -7.1794e+00,
-6.2769e+00, -4.6460e+00],
[-8.2517e-01, 1.5321e+00, -4.3899e+00, ..., -5.5288e+00,
-5.5738e+00, -4.3101e+00],
[-3.9380e-01, 1.0673e+00, -7.4526e+00, ..., -8.0231e+00,
-7.8579e+00, -6.7060e+00],
...,
[ 4.7895e-01, 7.5079e-01, -3.8505e+00, ..., -4.6708e+00,
-4.2259e+00, -3.5734e+00],
[ 2.7072e+00, 2.8078e+00, 3.7385e-01, ..., -9.6508e-01,
-1.3030e+00, -1.3740e+00],
[ 1.6277e+00, 1.5084e+00, -1.5557e+00, ..., -2.3232e+00,
-2.5637e+00, -2.4632e+00]],
[[-1.3636e+00, -2.3681e+00, -2.3233e+00, ..., -1.7655e+00,
-1.2503e+00, -1.3885e+00],
[-1.7081e+00, -2.0060e+00, -1.5284e+00, ..., -1.2615e+00,
-7.5586e-01, -8.4127e-01],
[-2.2059e+00, -1.9702e+00, -1.0267e+00, ..., -1.1215e+00,
-4.8299e-01, -2.0587e-01],
...,
[-3.1109e-01, -8.0410e-01, -5.2434e-01, ..., -4.9115e-01,
-2.1002e-01, -4.5075e-01],
[-8.8650e-01, -3.2252e-01, -4.7575e-01, ..., -2.1364e-01,
2.1801e-01, 2.2023e-01],
[-9.8820e-01, -2.9306e-01, -3.5355e-01, ..., -4.1292e-01,
-2.1294e-01, -2.4348e-01]],
...,
[[-2.0279e+00, -1.0775e+00, -2.3410e+00, ..., -3.1848e+00,
-2.5995e+00, -1.8281e+00],
[-1.7811e+00, 2.4132e-02, -1.3941e+00, ..., -2.5265e+00,
-2.5261e+00, -1.9752e+00],
[-3.8114e-01, 2.3841e+00, 1.0484e+00, ..., 1.0477e-01,
2.8007e-01, 5.3046e-01],
...,
[-2.2000e+00, 1.6104e-01, -1.0661e-01, ..., -1.2910e+00,
-4.3007e-01, 8.2821e-02],
[-1.2402e+00, 1.0413e+00, 1.0813e+00, ..., 7.6828e-02,
2.9113e-01, 4.4672e-01],
[-2.2050e+00, -1.4531e+00, -1.7162e+00, ..., -2.2749e+00,
-1.3748e+00, -8.7753e-01]],
[[-2.4423e+00, -2.8557e+00, -4.2513e+00, ..., -2.4476e+00,
-1.9518e+00, -1.7449e+00],
[-1.7971e+00, -1.6835e+00, -4.9223e+00, ..., -2.9507e+00,
-2.8096e+00, -2.7854e+00],
[-2.8502e+00, -2.4981e+00, -6.6925e+00, ..., -4.4800e+00,
-4.3893e+00, -3.4987e+00],
...,
[-1.3368e+00, 1.8870e-01, -3.5253e+00, ..., -1.4600e+00,
-1.0883e+00, -1.0295e+00],
[-1.6688e+00, -4.7277e-01, -3.1475e+00, ..., -1.7130e+00,
-1.1970e+00, -8.8486e-01],
[-2.3618e+00, -1.4281e+00, -3.1701e+00, ..., -2.8888e+00,
-2.1490e+00, -1.9817e+00]],
[[-1.8185e+00, -2.5509e+00, -2.4710e+00, ..., -2.7170e+00,
-2.2316e+00, -1.7073e+00],
[-1.8142e+00, -2.5307e+00, -2.1721e+00, ..., -2.4298e+00,
-1.9262e+00, -1.5543e+00],
[-2.2680e+00, -2.8444e+00, -2.5413e+00, ..., -2.7794e+00,
-2.1719e+00, -1.6465e+00],
...,
[-2.4999e+00, -3.0959e+00, -2.7927e+00, ..., -2.8320e+00,
-2.2544e+00, -1.7296e+00],
[-1.6193e+00, -2.1021e+00, -1.9752e+00, ..., -1.9404e+00,
-1.5925e+00, -1.2807e+00],
[-1.0459e+00, -1.3442e+00, -1.3417e+00, ..., -1.2016e+00,
-1.0148e+00, -7.6997e-01]]],
[[[ 4.7762e-01, 9.3282e-01, 7.3178e-01, ..., 7.9630e-01,
1.0591e+00, 8.2723e-01],
[-1.0645e+00, -1.5101e-01, -1.4578e+00, ..., -1.2151e+00,
5.0279e-02, 6.1745e-01],
[ 7.5067e-01, 5.4731e-01, 2.4345e-01, ..., 4.9718e-01,
8.4991e-01, 8.6187e-01],
...,
[-8.7006e-01, -1.9574e+00, -2.0129e+00, ..., -1.7550e+00,
-1.2567e+00, -8.6555e-01],
[ 8.5645e-01, 2.1238e-02, -1.0917e-01, ..., -1.8694e-02,
7.1266e-01, 8.1088e-01],
[ 2.1572e-01, -5.3417e-01, -4.5738e-01, ..., -1.5806e-01,
4.2353e-01, 3.4095e-01]],
[[-2.6166e+00, -1.4743e+00, -4.7602e+00, ..., -5.8523e+00,
-4.9991e+00, -3.8786e+00],
[-6.5131e-01, 1.2983e+00, -3.3397e+00, ..., -4.7047e+00,
-4.5141e+00, -3.6617e+00],
[ 1.7056e-02, 1.0676e+00, -6.0754e+00, ..., -6.8388e+00,
-6.4969e+00, -5.9115e+00],
...,
[ 4.9037e-01, 7.0320e-01, -3.8494e+00, ..., -4.6489e+00,
-4.2219e+00, -3.5229e+00],
[ 2.6882e+00, 2.7396e+00, 3.4232e-01, ..., -9.3191e-01,
-1.2936e+00, -1.3295e+00],
[ 1.6034e+00, 1.4586e+00, -1.5436e+00, ..., -2.2825e+00,
-2.5397e+00, -2.4078e+00]],
[[-5.7022e-01, -1.5861e+00, -1.4184e+00, ..., -7.5372e-01,
-4.9971e-01, -7.2403e-01],
[-1.3295e+00, -1.6978e+00, -1.0956e+00, ..., -6.9331e-01,
-3.6331e-01, -3.0545e-01],
[-1.1763e+00, -1.0432e+00, -4.6791e-01, ..., -4.1655e-01,
-6.8795e-02, -6.8293e-02],
...,
[-2.7196e-01, -7.4936e-01, -4.7430e-01, ..., -4.8186e-01,
-1.7296e-01, -4.2486e-01],
[-8.5218e-01, -2.9834e-01, -4.6382e-01, ..., -2.1003e-01,
2.3309e-01, 2.3864e-01],
[-9.6815e-01, -2.8492e-01, -3.7463e-01, ..., -3.9695e-01,
-1.9422e-01, -2.2118e-01]],
...,
[[-2.0772e+00, -1.1867e+00, -2.4754e+00, ..., -3.2894e+00,
-2.6977e+00, -1.8294e+00],
[-4.7901e-01, 1.1243e+00, -6.7009e-01, ..., -1.6233e+00,
-1.8097e+00, -1.5923e+00],
[-2.0929e-01, 2.1226e+00, 4.7703e-01, ..., -3.3066e-01,
-3.1599e-02, 1.3030e-01],
...,
[-2.2080e+00, 7.8724e-02, -1.5407e-01, ..., -1.3271e+00,
-3.9041e-01, 8.5068e-02],
[-1.2530e+00, 9.9170e-01, 1.0265e+00, ..., 8.6878e-05,
3.0652e-01, 4.5250e-01],
[-2.2140e+00, -1.4669e+00, -1.7212e+00, ..., -2.3931e+00,
-1.4258e+00, -9.0773e-01]],
[[-1.4300e+00, -1.3904e+00, -2.7582e+00, ..., -1.0858e+00,
-5.6858e-01, -4.5375e-01],
[-8.5062e-01, -4.6529e-01, -3.6823e+00, ..., -1.8906e+00,
-1.6096e+00, -1.8006e+00],
[-2.4270e+00, -1.9276e+00, -6.2476e+00, ..., -4.1928e+00,
-3.5891e+00, -2.8199e+00],
...,
[-1.3190e+00, 2.0052e-01, -3.5046e+00, ..., -1.4911e+00,
-1.0745e+00, -9.7065e-01],
[-1.6725e+00, -4.8161e-01, -3.1426e+00, ..., -1.7453e+00,
-1.1976e+00, -8.5164e-01],
[-2.3653e+00, -1.4363e+00, -3.1497e+00, ..., -2.9426e+00,
-2.1369e+00, -1.9725e+00]],
[[-1.2189e+00, -1.7946e+00, -1.6180e+00, ..., -1.8428e+00,
-1.5258e+00, -1.1318e+00],
[-1.2761e+00, -1.7773e+00, -1.3875e+00, ..., -1.6025e+00,
-1.3039e+00, -1.0199e+00],
[-1.6801e+00, -2.0306e+00, -1.6748e+00, ..., -1.9007e+00,
-1.4586e+00, -1.0646e+00],
...,
[-2.4895e+00, -3.0793e+00, -2.7873e+00, ..., -2.8204e+00,
-2.2537e+00, -1.7320e+00],
[-1.6025e+00, -2.0860e+00, -1.9630e+00, ..., -1.9256e+00,
-1.5893e+00, -1.2794e+00],
[-1.0325e+00, -1.3329e+00, -1.3435e+00, ..., -1.1890e+00,
-1.0125e+00, -7.6925e-01]]],
[[[ 3.8924e-01, 8.1160e-01, 5.1300e-01, ..., 4.5298e-01,
1.0114e+00, 6.6367e-01],
[-1.2265e+00, -3.8101e-01, -1.7821e+00, ..., -2.5171e+00,
-7.4328e-01, -2.6599e-02],
[ 7.8681e-01, 6.0088e-01, 3.6392e-01, ..., -2.2948e-01,
5.4980e-01, 6.8449e-01],
...,
[-1.2248e-02, -7.5756e-01, -1.5693e+00, ..., -1.9009e+00,
-1.3171e+00, -8.9774e-01],
[ 1.4537e+00, 1.1851e+00, 5.7078e-01, ..., -2.2122e-01,
5.6236e-01, 7.0069e-01],
[ 1.0823e+00, 9.5053e-01, 6.1158e-01, ..., -3.1349e-01,
3.1998e-01, 2.6990e-01]],
[[-2.6867e+00, -1.5813e+00, -4.7327e+00, ..., -4.7928e+00,
-4.3985e+00, -3.3523e+00],
[-6.3685e-01, 1.3171e+00, -3.0713e+00, ..., -3.3788e+00,
-3.5783e+00, -2.9382e+00],
[ 6.9942e-02, 1.0881e+00, -5.7089e+00, ..., -5.3269e+00,
-5.3192e+00, -4.9362e+00],
...,
[-7.1512e-01, 3.9275e-01, -4.6184e+00, ..., -4.6084e+00,
-4.1708e+00, -3.4270e+00],
[ 2.1634e+00, 2.8651e+00, 3.3413e-01, ..., -9.3579e-01,
-1.2609e+00, -1.2412e+00],
[ 1.3004e+00, 1.6051e+00, -1.5380e+00, ..., -2.2903e+00,
-2.5225e+00, -2.3450e+00]],
[[-6.2693e-01, -1.7338e+00, -1.6066e+00, ..., -5.8143e-01,
-1.9599e-01, -4.5772e-01],
[-1.3208e+00, -1.7802e+00, -1.1658e+00, ..., -2.9062e-01,
6.9398e-02, 4.1857e-02],
[-1.2261e+00, -1.2607e+00, -7.0947e-01, ..., -2.4051e-01,
2.5781e-01, 2.9109e-01],
...,
[-1.0423e+00, -1.6227e+00, -7.8096e-01, ..., -4.9129e-01,
-1.5481e-01, -3.3126e-01],
[-1.3732e+00, -1.0126e+00, -6.5189e-01, ..., -3.0344e-01,
1.7174e-01, 2.5032e-01],
[-1.4510e+00, -8.3896e-01, -3.4858e-01, ..., -5.0187e-01,
-2.6284e-01, -2.3347e-01]],
...,
[[-2.0667e+00, -1.0784e+00, -2.3504e+00, ..., -3.6102e+00,
-2.7608e+00, -1.7857e+00],
[-5.3978e-01, 1.1361e+00, -7.3168e-01, ..., -2.6932e+00,
-2.3618e+00, -1.8027e+00],
[-1.7195e-01, 2.3772e+00, 7.1374e-01, ..., -6.3347e-01,
-1.0516e-01, 1.7027e-01],
...,
[-1.5129e+00, 1.5402e+00, 1.1266e-01, ..., -1.5941e+00,
-6.6755e-01, -1.2354e-01],
[-5.3518e-01, 1.7408e+00, 9.7085e-01, ..., -2.0333e-01,
8.8400e-02, 3.1371e-01],
[-1.7125e+00, -1.1249e+00, -1.6090e+00, ..., -2.3644e+00,
-1.4672e+00, -9.3673e-01]],
[[-1.3907e+00, -1.4115e+00, -2.9555e+00, ..., -1.1770e+00,
-7.7385e-01, -7.9246e-01],
[-7.4129e-01, -3.8320e-01, -3.8184e+00, ..., -1.4326e+00,
-1.2840e+00, -1.6205e+00],
[-2.1583e+00, -1.6379e+00, -6.2041e+00, ..., -3.3102e+00,
-2.8406e+00, -2.2650e+00],
...,
[-1.7099e+00, -6.3853e-01, -4.7739e+00, ..., -1.5358e+00,
-1.0463e+00, -8.7518e-01],
[-1.6746e+00, -7.8084e-01, -3.8326e+00, ..., -1.7921e+00,
-1.1517e+00, -7.7338e-01],
[-2.1076e+00, -1.3632e+00, -3.0848e+00, ..., -2.8521e+00,
-2.0064e+00, -1.8346e+00]],
[[-1.2058e+00, -1.8210e+00, -1.6920e+00, ..., -1.8005e+00,
-1.4949e+00, -1.1485e+00],
[-1.2196e+00, -1.7570e+00, -1.4007e+00, ..., -1.4195e+00,
-1.1278e+00, -9.1327e-01],
[-1.5875e+00, -1.9674e+00, -1.6440e+00, ..., -1.6266e+00,
-1.2383e+00, -9.3869e-01],
...,
[-2.4450e+00, -3.0971e+00, -2.9186e+00, ..., -2.8845e+00,
-2.3267e+00, -1.7743e+00],
[-1.6399e+00, -2.1118e+00, -2.0548e+00, ..., -1.9980e+00,
-1.6478e+00, -1.3202e+00],
[-1.1095e+00, -1.3566e+00, -1.4361e+00, ..., -1.2497e+00,
-1.0465e+00, -8.0150e-01]]]], device='cuda:0'),
tensor([[[[ 2.2819e-01, 5.0322e-01, 6.1979e-01, ..., 5.7969e-01,
9.4943e-01, 7.7269e-01],
[-1.9811e+00, -1.2582e+00, -2.7810e+00, ..., -1.8486e+00,
-1.5878e+00, -6.0609e-01],
[-9.6531e-02, -7.9648e-01, 3.3562e-02, ..., -1.6889e+00,
-2.6996e+00, -1.8133e+00],
...,
[-9.5482e-01, -2.1046e+00, -2.0355e+00, ..., -1.3824e+00,
-9.0883e-01, -6.2076e-01],
[ 8.3385e-01, -2.7843e-02, -7.0429e-02, ..., 4.2118e-01,
8.0062e-01, 9.1033e-01],
[ 2.5702e-01, -4.9132e-01, -3.5089e-01, ..., 6.1629e-01,
8.8679e-01, 8.0453e-01]],
[[-1.7285e+00, -1.3943e+00, -4.2825e+00, ..., -5.8374e+00,
-4.9442e+00, -3.8782e+00],
[ 4.2693e-01, 1.4366e+00, -2.5560e+00, ..., -4.5453e+00,
-4.3589e+00, -3.6517e+00],
[ 1.1527e+00, 1.1353e+00, -5.0887e+00, ..., -6.1966e+00,
-6.4018e+00, -5.2397e+00],
...,
[ 2.8186e-01, 2.8769e-01, -4.1204e+00, ..., -6.0918e+00,
-5.2768e+00, -4.3630e+00],
[ 2.5074e+00, 2.4947e+00, 2.1047e-01, ..., -1.3888e+00,
-1.5292e+00, -1.6042e+00],
[ 1.4920e+00, 1.3356e+00, -1.5803e+00, ..., -2.7673e+00,
-2.9001e+00, -2.8210e+00]],
[[-1.7384e-01, -1.2147e+00, -1.4206e+00, ..., -8.1451e-01,
-5.5799e-01, -6.8015e-01],
[-5.6866e-01, -8.9016e-01, -9.1288e-01, ..., -8.1370e-01,
-3.2780e-01, -4.7743e-01],
[-3.7713e-01, -2.4984e-01, -4.4560e-01, ..., -4.5915e-01,
6.0994e-01, 1.3105e-01],
...,
[-6.6162e-02, -6.5009e-01, -6.3546e-01, ..., -7.1314e-01,
-5.0667e-01, -5.8735e-01],
[-7.0856e-01, -2.4407e-01, -5.9123e-01, ..., -3.2767e-01,
-3.2899e-02, 6.2908e-02],
[-8.9250e-01, -3.0276e-01, -4.7864e-01, ..., -3.2542e-01,
-2.2175e-01, -1.4530e-01]],
...,
[[-2.8186e+00, -2.4073e+00, -2.4437e+00, ..., -3.1974e+00,
-2.5937e+00, -1.7461e+00],
[-2.0155e+00, -7.4016e-01, -1.0651e+00, ..., -1.5188e+00,
-1.5230e+00, -1.3987e+00],
[-1.0749e+00, 8.3581e-01, 7.9631e-01, ..., -5.2932e-01,
-2.2330e-01, -6.5621e-01],
...,
[-2.2831e+00, -1.9503e-01, -1.3978e-01, ..., -1.0756e+00,
-3.5885e-01, 1.9157e-01],
[-1.2294e+00, 8.5626e-01, 1.0481e+00, ..., 2.3243e-01,
3.6689e-01, 4.6851e-01],
[-2.1886e+00, -1.5003e+00, -1.7011e+00, ..., -2.2290e+00,
-1.2307e+00, -6.0334e-01]],
[[-1.6271e+00, -1.6185e+00, -2.7662e+00, ..., -1.4162e+00,
-7.9753e-01, -5.6289e-01],
[-7.0292e-01, -1.1171e-01, -3.1216e+00, ..., -1.6629e+00,
-1.2018e+00, -1.2090e+00],
[-2.0869e+00, -1.0292e+00, -5.0802e+00, ..., -2.7550e+00,
-2.1950e+00, -1.3539e+00],
...,
[-1.3660e+00, 1.3587e-01, -3.3979e+00, ..., -2.4359e+00,
-1.8567e+00, -1.3858e+00],
[-1.6744e+00, -4.8260e-01, -3.0133e+00, ..., -2.2931e+00,
-1.5477e+00, -8.9822e-01],
[-2.3138e+00, -1.3719e+00, -3.0208e+00, ..., -2.9070e+00,
-1.9554e+00, -1.6329e+00]],
[[-1.2927e+00, -1.8260e+00, -1.6479e+00, ..., -1.8933e+00,
-1.5357e+00, -1.1267e+00],
[-1.2329e+00, -1.6994e+00, -1.2270e+00, ..., -1.9005e+00,
-1.5326e+00, -1.1911e+00],
[-1.5462e+00, -1.8835e+00, -1.4809e+00, ..., -2.9630e+00,
-2.4602e+00, -1.8932e+00],
...,
[-2.4346e+00, -3.0356e+00, -2.7830e+00, ..., -3.2169e+00,
-2.5270e+00, -1.8835e+00],
[-1.5708e+00, -2.0769e+00, -1.9847e+00, ..., -2.2679e+00,
-1.8421e+00, -1.4452e+00],
[-1.0278e+00, -1.3391e+00, -1.3580e+00, ..., -1.4086e+00,
-1.1813e+00, -8.5947e-01]]],
[[[ 2.7934e-01, 5.4103e-01, 6.8346e-01, ..., 4.6797e-01,
1.0157e+00, 5.9403e-01],
[-2.1129e+00, -1.4719e+00, -3.1329e+00, ..., -2.8466e+00,
-9.6364e-01, -2.3054e-01],
[-1.9821e-01, -1.0092e+00, -2.3999e-01, ..., -1.6968e-01,
5.5152e-01, 6.4881e-01],
...,
[ 9.3912e-02, -7.0552e-01, -1.6029e+00, ..., -1.0541e+00,
-7.5632e-01, -5.6090e-01],
[ 1.5443e+00, 1.2626e+00, 6.1285e-01, ..., 7.3918e-01,
9.1315e-01, 9.4576e-01],
[ 1.1555e+00, 1.0204e+00, 7.0371e-01, ..., 9.1028e-01,
1.0051e+00, 8.5714e-01]],
[[-1.6457e+00, -1.1847e+00, -4.1843e+00, ..., -5.0572e+00,
-4.5800e+00, -3.4550e+00],
[ 5.2004e-01, 1.7429e+00, -2.3716e+00, ..., -3.5127e+00,
-3.6236e+00, -2.9470e+00],
[ 1.1921e+00, 1.4534e+00, -4.9725e+00, ..., -5.5169e+00,
-5.3434e+00, -5.0228e+00],
...,
[-6.2622e-01, 5.5648e-01, -4.7368e+00, ..., -5.8929e+00,
-5.2355e+00, -4.2523e+00],
[ 2.1568e+00, 2.8743e+00, 1.4706e-01, ..., -1.5362e+00,
-1.7099e+00, -1.6847e+00],
[ 1.2790e+00, 1.5644e+00, -1.6697e+00, ..., -2.7098e+00,
-2.8367e+00, -2.7269e+00]],
[[-7.6923e-02, -1.0647e+00, -1.2472e+00, ..., -5.6393e-01,
-1.8802e-01, -5.2977e-01],
[-5.0842e-01, -7.5283e-01, -7.2929e-01, ..., -3.3741e-01,
1.9026e-02, -9.0457e-02],
[-4.5580e-01, -2.7791e-01, -3.9517e-01, ..., -3.6956e-01,
1.0139e-01, 1.0236e-01],
...,
[-1.1363e+00, -1.6022e+00, -6.7849e-01, ..., -6.3636e-01,
-4.7984e-01, -6.3252e-01],
[-1.3570e+00, -8.9187e-01, -4.7255e-01, ..., -1.0397e-01,
1.0194e-01, 8.4642e-02],
[-1.3881e+00, -6.7963e-01, -1.3574e-01, ..., -6.0504e-02,
-3.4029e-02, -5.3500e-02]],
...,
[[-2.9315e+00, -2.4722e+00, -2.7282e+00, ..., -3.7178e+00,
-2.8585e+00, -1.8931e+00],
[-2.2625e+00, -9.2697e-01, -1.6035e+00, ..., -2.7226e+00,
-2.4616e+00, -1.9317e+00],
[-1.0886e+00, 1.0079e+00, 5.6827e-01, ..., -4.6035e-01,
1.6224e-02, 1.9500e-01],
...,
[-1.5508e+00, 1.6932e+00, 3.8345e-01, ..., -7.9725e-01,
-1.4716e-01, 3.7173e-01],
[-5.8097e-01, 1.8259e+00, 1.1726e+00, ..., 1.7349e-01,
3.2715e-01, 4.9030e-01],
[-1.6879e+00, -9.9584e-01, -1.4042e+00, ..., -2.1008e+00,
-1.1799e+00, -5.4396e-01]],
[[-1.6054e+00, -1.6150e+00, -2.8329e+00, ..., -1.1768e+00,
-7.1577e-01, -7.5279e-01],
[-6.5222e-01, -3.2479e-02, -3.1175e+00, ..., -1.2703e+00,
-1.1463e+00, -1.5793e+00],
[-2.1030e+00, -9.8467e-01, -5.1597e+00, ..., -3.2132e+00,
-2.7492e+00, -2.2588e+00],
...,
[-1.7568e+00, -6.6138e-01, -4.6900e+00, ..., -2.4380e+00,
-2.0289e+00, -1.5973e+00],
[-1.6671e+00, -7.7814e-01, -3.7608e+00, ..., -2.2609e+00,
-1.6699e+00, -1.0856e+00],
[-2.0636e+00, -1.3273e+00, -3.0346e+00, ..., -2.7414e+00,
-1.9490e+00, -1.6724e+00]],
[[-1.3379e+00, -1.8871e+00, -1.7117e+00, ..., -1.8292e+00,
-1.5068e+00, -1.1464e+00],
[-1.2666e+00, -1.7469e+00, -1.2695e+00, ..., -1.3890e+00,
-1.0864e+00, -8.6275e-01],
[-1.5842e+00, -1.9394e+00, -1.5227e+00, ..., -1.5850e+00,
-1.1787e+00, -8.6448e-01],
...,
[-2.4937e+00, -3.1112e+00, -2.8357e+00, ..., -2.9971e+00,
-2.3893e+00, -1.8229e+00],
[-1.6789e+00, -2.1258e+00, -2.0150e+00, ..., -2.0889e+00,
-1.7083e+00, -1.3673e+00],
[-1.1234e+00, -1.3420e+00, -1.3657e+00, ..., -1.2852e+00,
-1.0929e+00, -8.0723e-01]]],
[[[ 5.7934e-02, 2.0544e-01, -1.4523e+00, ..., -6.5164e-01,
-2.1164e-01, -6.0029e-02],
[-2.1466e+00, -1.6737e+00, -4.6328e+00, ..., -3.8393e+00,
-1.5384e+00, -5.7397e-01],
[ 5.5106e-01, -9.7623e-02, -8.7319e-01, ..., -2.7720e-01,
-2.6206e-01, 2.3823e-01],
...,
[-8.9003e-01, -2.6573e+00, -4.1576e+00, ..., -1.7178e+00,
-1.2550e+00, -8.3782e-01],
[ 8.8372e-01, -6.4717e-01, -1.8782e+00, ..., -1.2738e-02,
7.0175e-01, 8.2492e-01],
[ 2.3574e-01, -1.1007e+00, -1.7530e+00, ..., -1.5504e-01,
4.3277e-01, 3.6683e-01]],
[[-3.3197e+00, -1.8320e+00, -6.3117e+00, ..., -7.1539e+00,
-6.2974e+00, -4.6236e+00],
[-8.0114e-01, 1.5308e+00, -4.3777e+00, ..., -5.4958e+00,
-5.6031e+00, -4.2554e+00],
[-3.7187e-01, 1.0769e+00, -7.4778e+00, ..., -8.0030e+00,
-7.8952e+00, -6.6450e+00],
...,
[ 4.9188e-01, 9.9914e-01, -4.2549e+00, ..., -4.6315e+00,
-4.1952e+00, -3.5433e+00],
[ 2.7381e+00, 2.9156e+00, 3.9517e-02, ..., -9.1629e-01,
-1.2673e+00, -1.3346e+00],
[ 1.6410e+00, 1.4885e+00, -1.5074e+00, ..., -2.2831e+00,
-2.5480e+00, -2.4366e+00]],
[[-1.3823e+00, -2.3982e+00, -2.3570e+00, ..., -1.7635e+00,
-1.2483e+00, -1.3786e+00],
[-1.7284e+00, -2.0122e+00, -1.5433e+00, ..., -1.2421e+00,
-7.2088e-01, -8.1377e-01],
[-2.2054e+00, -1.9443e+00, -9.9400e-01, ..., -1.1048e+00,
-4.4588e-01, -1.7722e-01],
...,
[-2.8589e-01, -7.4978e-01, -3.5945e-01, ..., -4.8554e-01,
-1.8818e-01, -4.3243e-01],
[-8.8809e-01, -3.9925e-01, -4.6400e-01, ..., -1.9995e-01,
2.4122e-01, 2.2872e-01],
[-1.0216e+00, -5.1935e-01, -4.8362e-01, ..., -4.0413e-01,
-1.9902e-01, -2.2499e-01]],
...,
[[-2.0552e+00, -1.1380e+00, -2.4008e+00, ..., -3.1780e+00,
-2.6341e+00, -1.8395e+00],
[-1.7638e+00, -7.5295e-03, -1.3966e+00, ..., -2.4947e+00,
-2.5601e+00, -1.9610e+00],
[-3.7584e-01, 2.3477e+00, 1.0357e+00, ..., 1.3889e-01,
1.9746e-01, 4.9242e-01],
...,
[-2.2259e+00, 3.5992e-01, -3.0842e-01, ..., -1.3109e+00,
-3.9681e-01, 7.8550e-02],
[-1.2383e+00, 1.2093e+00, 7.1301e-01, ..., 1.1256e-02,
2.9176e-01, 4.5189e-01],
[-2.1978e+00, -1.1057e+00, -1.8039e+00, ..., -2.3600e+00,
-1.4048e+00, -8.8644e-01]],
[[-2.4344e+00, -2.8538e+00, -4.2223e+00, ..., -2.4276e+00,
-1.9266e+00, -1.7432e+00],
[-1.7811e+00, -1.6602e+00, -4.8752e+00, ..., -2.9405e+00,
-2.7846e+00, -2.7948e+00],
[-2.8528e+00, -2.4898e+00, -6.6680e+00, ..., -4.4530e+00,
-4.3568e+00, -3.5172e+00],
...,
[-1.3102e+00, 1.9713e-01, -3.0260e+00, ..., -1.4581e+00,
-1.0978e+00, -9.9652e-01],
[-1.6374e+00, -3.9669e-01, -2.5413e+00, ..., -1.7187e+00,
-1.2146e+00, -8.6812e-01],
[-2.3445e+00, -1.2005e+00, -2.5283e+00, ..., -2.9062e+00,
-2.1455e+00, -1.9794e+00]],
[[-1.8222e+00, -2.5558e+00, -2.4782e+00, ..., -2.7138e+00,
-2.2300e+00, -1.7186e+00],
[-1.8155e+00, -2.5314e+00, -2.1700e+00, ..., -2.4293e+00,
-1.9251e+00, -1.5686e+00],
[-2.2658e+00, -2.8351e+00, -2.5327e+00, ..., -2.7847e+00,
-2.1773e+00, -1.6609e+00],
...,
[-2.5136e+00, -3.1916e+00, -3.1376e+00, ..., -2.8175e+00,
-2.2420e+00, -1.7288e+00],
[-1.6297e+00, -2.2496e+00, -2.4044e+00, ..., -1.9220e+00,
-1.5789e+00, -1.2734e+00],
[-1.0560e+00, -1.4876e+00, -1.7008e+00, ..., -1.1837e+00,
-1.0036e+00, -7.6228e-01]]],
[[[ 1.4922e-01, 3.8382e-01, -1.1358e+00, ..., 4.4928e-01,
1.0230e+00, 6.6256e-01],
[-2.0372e+00, -1.3875e+00, -4.2670e+00, ..., -2.6277e+00,
-7.6210e-01, -6.3224e-02],
[ 4.4022e-01, -2.8076e-01, -1.1138e+00, ..., -3.0700e-01,
4.8658e-01, 6.4342e-01],
...,
[-2.6273e-01, -1.2326e+00, -2.5486e+00, ..., -1.9861e+00,
-1.3800e+00, -9.0229e-01],
[ 1.2558e+00, 7.3842e-01, -8.0365e-01, ..., -3.2164e-01,
5.2605e-01, 7.1575e-01],
[ 8.1584e-01, 3.4390e-01, -3.8416e-01, ..., -3.6298e-01,
2.8343e-01, 2.7691e-01]],
[[-3.1954e+00, -1.4394e+00, -6.1051e+00, ..., -4.8268e+00,
-4.4470e+00, -3.3673e+00],
[-5.5645e-01, 2.1132e+00, -4.1669e+00, ..., -3.3624e+00,
-3.5477e+00, -2.8943e+00],
[ 3.4228e-02, 1.8200e+00, -7.3495e+00, ..., -5.3520e+00,
-5.2700e+00, -4.9088e+00],
...,
[-1.2334e+00, 3.0463e-01, -6.5405e+00, ..., -4.5710e+00,
-4.1177e+00, -3.4446e+00],
[ 1.5111e+00, 2.5729e+00, -1.4181e+00, ..., -8.7267e-01,
-1.1925e+00, -1.2390e+00],
[ 9.9277e-01, 1.6063e+00, -2.4715e+00, ..., -2.2426e+00,
-2.4621e+00, -2.3666e+00]],
[[-1.3620e+00, -2.3259e+00, -2.1521e+00, ..., -5.9530e-01,
-1.9401e-01, -4.5887e-01],
[-2.0070e+00, -2.1855e+00, -1.5465e+00, ..., -3.3784e-01,
8.1078e-02, 4.3400e-02],
[-2.5407e+00, -2.0145e+00, -8.4708e-01, ..., -2.7210e-01,
2.5845e-01, 2.8799e-01],
...,
[-1.5633e+00, -1.8323e+00, -7.7746e-01, ..., -4.8298e-01,
-1.5703e-01, -3.5380e-01],
[-1.7350e+00, -1.1910e+00, -7.8807e-01, ..., -3.0660e-01,
1.6159e-01, 2.2101e-01],
[-1.7211e+00, -1.1236e+00, -5.9751e-01, ..., -5.0000e-01,
-2.7251e-01, -2.4787e-01]],
...,
[[-2.3294e+00, -1.2574e+00, -2.6878e+00, ..., -3.6761e+00,
-2.8896e+00, -1.8716e+00],
[-1.8656e+00, 2.2418e-01, -1.3130e+00, ..., -2.6699e+00,
-2.4759e+00, -1.9130e+00],
[-4.8635e-01, 2.5205e+00, 9.4328e-01, ..., -5.4090e-01,
-1.0292e-01, 1.3102e-01],
...,
[-1.7352e+00, 1.1481e+00, 2.0524e-01, ..., -1.5572e+00,
-6.9023e-01, -1.3174e-01],
[-8.9972e-01, 1.6096e+00, 1.4346e+00, ..., -1.4961e-01,
1.2557e-01, 3.3561e-01],
[-2.0783e+00, -1.0022e+00, -7.9203e-01, ..., -2.3935e+00,
-1.4608e+00, -9.1372e-01]],
[[-2.3698e+00, -2.6323e+00, -4.0510e+00, ..., -1.1175e+00,
-7.2367e-01, -7.6161e-01],
[-1.6598e+00, -1.3392e+00, -4.7534e+00, ..., -1.3294e+00,
-1.2166e+00, -1.6019e+00],
[-2.9956e+00, -2.4353e+00, -6.8549e+00, ..., -3.2389e+00,
-2.7930e+00, -2.2465e+00],
...,
[-2.5391e+00, -1.7850e+00, -5.0832e+00, ..., -1.4798e+00,
-1.0468e+00, -8.8370e-01],
[-2.1901e+00, -1.5579e+00, -3.9552e+00, ..., -1.7441e+00,
-1.1546e+00, -7.6192e-01],
[-2.2323e+00, -1.7417e+00, -3.1178e+00, ..., -2.8192e+00,
-2.0177e+00, -1.8441e+00]],
[[-1.8974e+00, -2.6079e+00, -2.4040e+00, ..., -1.8135e+00,
-1.5003e+00, -1.1457e+00],
[-1.9205e+00, -2.6251e+00, -2.0944e+00, ..., -1.4091e+00,
-1.1132e+00, -8.8880e-01],
[-2.4463e+00, -3.0005e+00, -2.5129e+00, ..., -1.6092e+00,
-1.2053e+00, -8.9426e-01],
...,
[-3.1065e+00, -3.8759e+00, -3.6317e+00, ..., -2.8747e+00,
-2.2947e+00, -1.7623e+00],
[-2.0830e+00, -2.6862e+00, -2.6427e+00, ..., -1.9839e+00,
-1.6305e+00, -1.3135e+00],
[-1.5087e+00, -1.8301e+00, -1.8744e+00, ..., -1.2290e+00,
-1.0407e+00, -7.9599e-01]]],
[[[ 8.2996e-02, 2.4654e-01, -1.3821e+00, ..., -6.1492e-01,
-1.8396e-01, -3.1457e-02],
[-2.0931e+00, -1.5837e+00, -4.5013e+00, ..., -3.7519e+00,
-1.5034e+00, -5.3021e-01],
[ 5.6626e-01, -6.3290e-02, -8.4895e-01, ..., -1.9676e-01,
-2.5548e-01, 2.4534e-01],
...,
[-3.2990e-01, -1.1460e+00, -1.4808e+00, ..., -1.2865e+00,
-8.6340e-01, -6.3438e-01],
[ 9.7142e-01, 5.4704e-01, 3.0312e-01, ..., 4.1777e-01,
6.8928e-01, 8.1135e-01],
[ 6.8352e-01, 4.2782e-01, 3.1767e-01, ..., 5.3780e-01,
7.5581e-01, 6.9460e-01]],
[[-3.2917e+00, -1.8346e+00, -6.2259e+00, ..., -7.1263e+00,
-6.2476e+00, -4.6242e+00],
[-8.1832e-01, 1.4781e+00, -4.3453e+00, ..., -5.4588e+00,
-5.5546e+00, -4.2916e+00],
[-3.9193e-01, 1.0010e+00, -7.4684e+00, ..., -7.9817e+00,
-7.8590e+00, -6.6985e+00],
...,
[-4.3931e-01, 3.0615e-01, -4.4082e+00, ..., -5.3831e+00,
-4.6865e+00, -3.8721e+00],
[ 1.9942e+00, 2.4897e+00, -2.2108e-02, ..., -1.1725e+00,
-1.2808e+00, -1.3703e+00],
[ 1.1391e+00, 1.3427e+00, -1.6752e+00, ..., -2.5032e+00,
-2.5862e+00, -2.5433e+00]],
[[-1.3470e+00, -2.3641e+00, -2.3435e+00, ..., -1.7534e+00,
-1.2457e+00, -1.3773e+00],
[-1.7030e+00, -1.9809e+00, -1.5282e+00, ..., -1.2523e+00,
-7.2168e-01, -8.0456e-01],
[-2.1573e+00, -1.8843e+00, -9.7727e-01, ..., -1.0985e+00,
-4.6174e-01, -1.8781e-01],
...,
[-6.4245e-01, -1.1115e+00, -5.8062e-01, ..., -5.9894e-01,
-4.5473e-01, -6.2286e-01],
[-1.0424e+00, -7.0249e-01, -5.8042e-01, ..., -2.5371e-01,
-4.5656e-02, -1.6350e-02],
[-1.1959e+00, -6.7215e-01, -3.9623e-01, ..., -3.2738e-01,
-2.7986e-01, -2.3341e-01]],
...,
[[-2.0351e+00, -1.1177e+00, -2.3316e+00, ..., -3.2084e+00,
-2.6385e+00, -1.8212e+00],
[-1.7296e+00, 1.5943e-02, -1.3221e+00, ..., -2.4812e+00,
-2.5268e+00, -1.9578e+00],
[-3.4908e-01, 2.3140e+00, 1.0637e+00, ..., 1.6044e-01,
2.7111e-01, 5.1952e-01],
...,
[-1.5742e+00, 9.5392e-01, -1.2378e-01, ..., -1.2267e+00,
-4.7504e-01, 4.3735e-02],
[-5.7262e-01, 1.4089e+00, 7.8080e-01, ..., -3.3945e-02,
1.8126e-01, 2.9345e-01],
[-1.6273e+00, -1.0736e+00, -1.5088e+00, ..., -2.3159e+00,
-1.3476e+00, -7.0863e-01]],
[[-2.4124e+00, -2.8369e+00, -4.1758e+00, ..., -2.3760e+00,
-1.8946e+00, -1.7132e+00],
[-1.7592e+00, -1.6376e+00, -4.8056e+00, ..., -2.8993e+00,
-2.7868e+00, -2.7801e+00],
[-2.8283e+00, -2.4573e+00, -6.5916e+00, ..., -4.4338e+00,
-4.3707e+00, -3.4869e+00],
...,
[-1.6988e+00, -5.1051e-01, -4.0295e+00, ..., -2.1169e+00,
-1.5736e+00, -1.2039e+00],
[-1.6819e+00, -7.0509e-01, -3.3468e+00, ..., -2.0854e+00,
-1.3632e+00, -7.9050e-01],
[-2.0543e+00, -1.2646e+00, -2.8296e+00, ..., -2.6977e+00,
-1.7876e+00, -1.5101e+00]],
[[-1.8049e+00, -2.5352e+00, -2.4570e+00, ..., -2.6952e+00,
-2.2094e+00, -1.6909e+00],
[-1.7987e+00, -2.5044e+00, -2.1590e+00, ..., -2.3985e+00,
-1.8949e+00, -1.5358e+00],
[-2.2501e+00, -2.8003e+00, -2.5241e+00, ..., -2.7403e+00,
-2.1382e+00, -1.6252e+00],
...,
[-2.2803e+00, -2.8710e+00, -2.6682e+00, ..., -2.8766e+00,
-2.2873e+00, -1.7200e+00],
[-1.5409e+00, -1.9866e+00, -1.9367e+00, ..., -2.0260e+00,
-1.6635e+00, -1.3174e+00],
[-1.0392e+00, -1.2754e+00, -1.3391e+00, ..., -1.2538e+00,
-1.0703e+00, -7.8853e-01]]]], device='cuda:0'),
tensor([[[[ 4.7076e-01, 9.3487e-01, 7.3110e-01, ..., 7.8649e-01,
1.0502e+00, 8.2774e-01],
[-1.0565e+00, -1.3818e-01, -1.4317e+00, ..., -1.2550e+00,
3.6400e-02, 6.1259e-01],
[ 7.3932e-01, 5.3971e-01, 2.5137e-01, ..., 5.0245e-01,
8.8564e-01, 8.9221e-01],
...,
[-8.8102e-01, -1.9805e+00, -2.0890e+00, ..., -1.7529e+00,
-1.2233e+00, -8.6567e-01],
[ 8.8080e-01, 4.0715e-02, -1.2463e-01, ..., -4.5298e-02,
7.2741e-01, 8.1207e-01],
[ 2.3756e-01, -5.3307e-01, -4.7640e-01, ..., -1.9944e-01,
4.2086e-01, 3.4806e-01]],
[[-2.6124e+00, -1.4255e+00, -4.7482e+00, ..., -5.8516e+00,
-4.9799e+00, -3.8859e+00],
[-6.3844e-01, 1.3714e+00, -3.3174e+00, ..., -4.6616e+00,
-4.4782e+00, -3.6415e+00],
[ 4.4453e-02, 1.1742e+00, -6.0092e+00, ..., -6.8081e+00,
-6.4799e+00, -5.9198e+00],
...,
[ 5.1505e-01, 7.4686e-01, -3.8304e+00, ..., -4.6322e+00,
-4.2343e+00, -3.5314e+00],
[ 2.7378e+00, 2.8027e+00, 3.8954e-01, ..., -8.9610e-01,
-1.2697e+00, -1.2997e+00],
[ 1.6450e+00, 1.4979e+00, -1.5358e+00, ..., -2.2698e+00,
-2.5418e+00, -2.4284e+00]],
[[-5.3928e-01, -1.5535e+00, -1.3734e+00, ..., -7.3811e-01,
-5.0925e-01, -7.3000e-01],
[-1.3023e+00, -1.6980e+00, -1.0839e+00, ..., -6.8544e-01,
-3.6650e-01, -3.1224e-01],
[-1.1711e+00, -1.0858e+00, -4.8947e-01, ..., -4.0470e-01,
-5.7178e-02, -6.2259e-02],
...,
[-3.0116e-01, -7.7109e-01, -5.2338e-01, ..., -4.8984e-01,
-1.8966e-01, -4.4511e-01],
[-8.8946e-01, -2.9905e-01, -4.6617e-01, ..., -2.0854e-01,
2.5015e-01, 2.3714e-01],
[-9.9245e-01, -2.8023e-01, -3.6014e-01, ..., -4.1090e-01,
-1.9352e-01, -2.2727e-01]],
...,
[[-2.0658e+00, -1.1592e+00, -2.5042e+00, ..., -3.3073e+00,
-2.7375e+00, -1.8677e+00],
[-5.0516e-01, 1.1447e+00, -7.1143e-01, ..., -1.6390e+00,
-1.8215e+00, -1.5785e+00],
[-2.1749e-01, 2.1956e+00, 4.8177e-01, ..., -3.7153e-01,
-6.2599e-02, 1.1186e-01],
...,
[-2.2317e+00, 1.3435e-01, -7.2184e-02, ..., -1.3161e+00,
-4.0417e-01, 7.4171e-02],
[-1.2418e+00, 1.0216e+00, 1.0966e+00, ..., 3.4326e-02,
3.0875e-01, 4.5145e-01],
[-2.2131e+00, -1.4554e+00, -1.7299e+00, ..., -2.3948e+00,
-1.4300e+00, -9.0609e-01]],
[[-1.4295e+00, -1.3784e+00, -2.7952e+00, ..., -1.0730e+00,
-5.4345e-01, -4.4061e-01],
[-8.5048e-01, -4.5137e-01, -3.7580e+00, ..., -1.8493e+00,
-1.5879e+00, -1.8232e+00],
[-2.4196e+00, -1.8959e+00, -6.3396e+00, ..., -4.1651e+00,
-3.5999e+00, -2.8571e+00],
...,
[-1.3232e+00, 2.2435e-01, -3.5173e+00, ..., -1.4508e+00,
-1.0696e+00, -9.7076e-01],
[-1.6766e+00, -4.4785e-01, -3.1248e+00, ..., -1.7252e+00,
-1.1908e+00, -8.3945e-01],
[-2.3752e+00, -1.4110e+00, -3.1466e+00, ..., -2.9223e+00,
-2.1412e+00, -1.9842e+00]],
[[-1.2153e+00, -1.7938e+00, -1.6098e+00, ..., -1.8404e+00,
-1.5345e+00, -1.1394e+00],
[-1.2705e+00, -1.7808e+00, -1.3695e+00, ..., -1.6068e+00,
-1.3071e+00, -1.0218e+00],
[-1.6859e+00, -2.0400e+00, -1.6594e+00, ..., -1.8955e+00,
-1.4561e+00, -1.0575e+00],
...,
[-2.5090e+00, -3.0977e+00, -2.7976e+00, ..., -2.8361e+00,
-2.2668e+00, -1.7429e+00],
[-1.6187e+00, -2.1045e+00, -1.9805e+00, ..., -1.9309e+00,
-1.5918e+00, -1.2802e+00],
[-1.0456e+00, -1.3471e+00, -1.3493e+00, ..., -1.1850e+00,
-1.0088e+00, -7.6404e-01]]],
[[[ 1.5324e-01, 3.7897e-01, -1.2074e+00, ..., 4.7470e-01,
1.0151e+00, 6.4607e-01],
[-2.1209e+00, -1.4564e+00, -4.4028e+00, ..., -2.5623e+00,
-7.4592e-01, -6.1335e-02],
[ 4.1261e-01, -2.8494e-01, -1.1458e+00, ..., -2.4112e-01,
4.8425e-01, 6.2675e-01],
...,
[-2.2288e-01, -1.2418e+00, -2.5470e+00, ..., -1.9641e+00,
-1.3870e+00, -9.1879e-01],
[ 1.2700e+00, 6.5664e-01, -9.0230e-01, ..., -3.0102e-01,
4.9078e-01, 6.8885e-01],
[ 8.3204e-01, 2.9954e-01, -4.1776e-01, ..., -3.3188e-01,
2.7851e-01, 2.7141e-01]],
[[-3.2461e+00, -1.4695e+00, -6.1542e+00, ..., -4.8754e+00,
-4.4279e+00, -3.3623e+00],
[-5.5817e-01, 2.1492e+00, -4.1802e+00, ..., -3.4191e+00,
-3.5539e+00, -2.9152e+00],
[ 4.5265e-02, 1.8389e+00, -7.4171e+00, ..., -5.4072e+00,
-5.2568e+00, -4.9418e+00],
...,
[-1.2484e+00, 2.2103e-01, -6.5867e+00, ..., -4.6075e+00,
-4.1431e+00, -3.4555e+00],
[ 1.4584e+00, 2.5209e+00, -1.4732e+00, ..., -9.3228e-01,
-1.2316e+00, -1.2555e+00],
[ 9.6473e-01, 1.5861e+00, -2.4695e+00, ..., -2.2272e+00,
-2.4412e+00, -2.3453e+00]],
[[-1.3968e+00, -2.3744e+00, -2.1821e+00, ..., -6.0185e-01,
-1.9272e-01, -4.6526e-01],
[-2.0361e+00, -2.2217e+00, -1.5787e+00, ..., -3.5423e-01,
7.8990e-02, 4.1390e-02],
[-2.5693e+00, -2.0236e+00, -8.4701e-01, ..., -2.7546e-01,
2.4926e-01, 2.7889e-01],
...,
[-1.6290e+00, -1.8226e+00, -7.8536e-01, ..., -4.8427e-01,
-1.6514e-01, -3.5196e-01],
[-1.7832e+00, -1.1810e+00, -7.5047e-01, ..., -3.2681e-01,
1.4640e-01, 2.1881e-01],
[-1.7095e+00, -1.0923e+00, -5.6584e-01, ..., -4.7903e-01,
-2.6812e-01, -2.3941e-01]],
...,
[[-2.3763e+00, -1.3292e+00, -2.7682e+00, ..., -3.6824e+00,
-2.8355e+00, -1.8233e+00],
[-1.8891e+00, 1.7839e-01, -1.3550e+00, ..., -2.6574e+00,
-2.4396e+00, -1.8876e+00],
[-5.0052e-01, 2.4863e+00, 9.4560e-01, ..., -5.5771e-01,
-4.9569e-02, 1.7918e-01],
...,
[-1.6854e+00, 1.1714e+00, 2.3347e-01, ..., -1.5339e+00,
-6.7773e-01, -1.2802e-01],
[-8.3855e-01, 1.6456e+00, 1.4853e+00, ..., -1.4919e-01,
1.2198e-01, 3.3247e-01],
[-2.0813e+00, -1.0321e+00, -8.3549e-01, ..., -2.3649e+00,
-1.4478e+00, -9.1852e-01]],
[[-2.3732e+00, -2.6412e+00, -4.0318e+00, ..., -1.1208e+00,
-7.3827e-01, -7.5352e-01],
[-1.6593e+00, -1.3347e+00, -4.7663e+00, ..., -1.3694e+00,
-1.2321e+00, -1.6004e+00],
[-3.0254e+00, -2.4516e+00, -6.9084e+00, ..., -3.2983e+00,
-2.7983e+00, -2.2355e+00],
...,
[-2.5602e+00, -1.7975e+00, -4.9917e+00, ..., -1.5067e+00,
-1.0571e+00, -8.7317e-01],
[-2.2431e+00, -1.5970e+00, -3.8834e+00, ..., -1.7515e+00,
-1.1557e+00, -7.5061e-01],
[-2.3031e+00, -1.7972e+00, -3.0922e+00, ..., -2.8102e+00,
-2.0042e+00, -1.8222e+00]],
[[-1.9038e+00, -2.6310e+00, -2.4202e+00, ..., -1.8014e+00,
-1.4965e+00, -1.1340e+00],
[-1.9267e+00, -2.6367e+00, -2.1052e+00, ..., -1.3949e+00,
-1.1082e+00, -8.8476e-01],
[-2.4510e+00, -3.0060e+00, -2.5192e+00, ..., -1.6003e+00,
-1.2010e+00, -8.9256e-01],
...,
[-3.1105e+00, -3.8809e+00, -3.6573e+00, ..., -2.8617e+00,
-2.2886e+00, -1.7603e+00],
[-2.0718e+00, -2.6702e+00, -2.6389e+00, ..., -1.9724e+00,
-1.6196e+00, -1.3058e+00],
[-1.4901e+00, -1.8063e+00, -1.8524e+00, ..., -1.2206e+00,
-1.0342e+00, -7.9135e-01]]],
[[[ 2.6495e-01, 5.4032e-01, 6.5888e-01, ..., 6.0807e-01,
9.5282e-01, 7.6482e-01],
[-2.0639e+00, -1.3594e+00, -2.9695e+00, ..., -1.2974e+00,
6.0024e-02, 6.0384e-01],
[-4.6580e-02, -8.8894e-01, -3.1522e-03, ..., 4.7913e-01,
9.3139e-01, 9.6344e-01],
...,
[-9.5352e-01, -2.1016e+00, -1.9708e+00, ..., -1.4282e+00,
-8.9648e-01, -6.6012e-01],
[ 7.3847e-01, -1.6920e-01, -1.4369e-01, ..., 3.7840e-01,
7.8123e-01, 8.8282e-01],
[ 1.5668e-01, -6.2514e-01, -4.3120e-01, ..., 5.7981e-01,
8.7391e-01, 7.9620e-01]],
[[-1.7285e+00, -1.2402e+00, -4.2693e+00, ..., -5.7934e+00,
-4.8953e+00, -3.8481e+00],
[ 6.3898e-01, 1.8950e+00, -2.3236e+00, ..., -4.7172e+00,
-4.4360e+00, -3.6845e+00],
[ 1.4905e+00, 1.6508e+00, -4.9881e+00, ..., -6.8140e+00,
-6.3745e+00, -5.8545e+00],
...,
[ 3.3494e-01, 2.8512e-01, -4.2240e+00, ..., -6.0141e+00,
-5.2178e+00, -4.2674e+00],
[ 2.3898e+00, 2.2962e+00, -1.9250e-01, ..., -1.3117e+00,
-1.4616e+00, -1.5257e+00],
[ 1.4043e+00, 1.1936e+00, -1.7940e+00, ..., -2.6899e+00,
-2.8341e+00, -2.7790e+00]],
[[-2.2050e-01, -1.3637e+00, -1.5520e+00, ..., -7.8884e-01,
-5.4536e-01, -6.7225e-01],
[-7.0306e-01, -1.0578e+00, -9.9426e-01, ..., -6.6302e-01,
-3.7617e-01, -2.4649e-01],
[-5.2241e-01, -3.8265e-01, -4.6120e-01, ..., -3.5992e-01,
-1.3380e-02, 7.6931e-02],
...,
[-1.9998e-02, -5.5150e-01, -5.7908e-01, ..., -6.9918e-01,
-4.9260e-01, -5.7260e-01],
[-6.5619e-01, -1.9528e-01, -5.4935e-01, ..., -3.1059e-01,
-1.8462e-02, 7.5805e-02],
[-8.2379e-01, -2.6211e-01, -4.8854e-01, ..., -3.0529e-01,
-2.0640e-01, -1.4384e-01]],
...,
[[-3.0513e+00, -2.4694e+00, -2.6007e+00, ..., -3.1536e+00,
-2.5783e+00, -1.7421e+00],
[-2.1367e+00, -6.1022e-01, -1.0874e+00, ..., -1.6976e+00,
-1.7636e+00, -1.4771e+00],
[-1.0894e+00, 1.1384e+00, 8.8130e-01, ..., -2.9186e-01,
1.3873e-02, 1.6892e-01],
...,
[-2.1639e+00, -1.5768e-01, -9.3106e-02, ..., -1.0472e+00,
-3.7242e-01, 1.6855e-01],
[-1.1808e+00, 8.6011e-01, 1.0569e+00, ..., 2.3341e-01,
3.4875e-01, 4.5171e-01],
[-2.0762e+00, -1.3641e+00, -1.5209e+00, ..., -2.2561e+00,
-1.2644e+00, -6.2912e-01]],
[[-1.6290e+00, -1.5862e+00, -2.9516e+00, ..., -1.3543e+00,
-7.5544e-01, -5.3781e-01],
[-6.0097e-01, 1.0797e-01, -3.3131e+00, ..., -2.0223e+00,
-1.6475e+00, -1.7358e+00],
[-2.0571e+00, -8.3693e-01, -5.4246e+00, ..., -4.0053e+00,
-3.3917e+00, -2.6102e+00],
...,
[-1.3558e+00, 9.2895e-02, -3.3789e+00, ..., -2.3870e+00,
-1.7875e+00, -1.3088e+00],
[-1.6607e+00, -5.0392e-01, -2.9766e+00, ..., -2.2563e+00,
-1.5061e+00, -8.5772e-01],
[-2.2693e+00, -1.3399e+00, -2.9596e+00, ..., -2.8935e+00,
-1.9466e+00, -1.6209e+00]],
[[-1.3475e+00, -1.9028e+00, -1.6747e+00, ..., -1.8805e+00,
-1.5300e+00, -1.1114e+00],
[-1.2763e+00, -1.7662e+00, -1.1927e+00, ..., -1.6471e+00,
-1.3103e+00, -9.9494e-01],
[-1.6049e+00, -1.9452e+00, -1.4288e+00, ..., -1.9318e+00,
-1.4542e+00, -1.0292e+00],
...,
[-2.4133e+00, -3.0050e+00, -2.7492e+00, ..., -3.1901e+00,
-2.5133e+00, -1.8721e+00],
[-1.5747e+00, -2.0688e+00, -1.9906e+00, ..., -2.2433e+00,
-1.8229e+00, -1.4328e+00],
[-1.0371e+00, -1.3536e+00, -1.3840e+00, ..., -1.3837e+00,
-1.1652e+00, -8.4543e-01]]],
[[[ 4.8623e-01, 9.6631e-01, 8.6749e-01, ..., -3.2275e-01,
-5.7285e-02, 1.3945e-02],
[-1.2300e+00, -3.2048e-01, -1.5214e+00, ..., -3.1499e+00,
-1.2390e+00, -4.2617e-01],
[ 5.9998e-01, 4.2347e-01, 3.2586e-01, ..., -1.4289e-01,
-3.3319e-01, 1.6577e-01],
...,
[-1.9979e-01, -8.4280e-01, -1.4068e+00, ..., -1.9399e+00,
-1.8967e+00, -1.2867e+00],
[ 1.0400e+00, 7.8737e-01, 3.7300e-01, ..., 2.5201e-02,
2.8279e-01, 4.5622e-01],
[ 6.9905e-01, 5.1274e-01, 2.3161e-01, ..., 2.2433e-01,
4.7675e-01, 5.1387e-01]],
[[-2.2993e+00, -1.0961e+00, -4.1014e+00, ..., -6.8189e+00,
-6.0851e+00, -4.5157e+00],
[-2.6796e-01, 1.7345e+00, -2.4902e+00, ..., -5.4782e+00,
-5.5807e+00, -4.2675e+00],
[ 4.4458e-01, 1.5291e+00, -5.0342e+00, ..., -7.8657e+00,
-7.7855e+00, -6.5429e+00],
...,
[-4.4184e-01, 6.0490e-01, -4.0238e+00, ..., -7.0932e+00,
-6.6652e+00, -5.1790e+00],
[ 2.0830e+00, 2.7347e+00, 3.9112e-01, ..., -1.9309e+00,
-2.5569e+00, -2.0521e+00],
[ 1.2026e+00, 1.4842e+00, -1.4754e+00, ..., -2.9411e+00,
-3.4758e+00, -3.0536e+00]],
[[-5.8173e-01, -1.5811e+00, -1.4042e+00, ..., -1.4032e+00,
-1.0385e+00, -1.3168e+00],
[-1.2947e+00, -1.6432e+00, -1.0102e+00, ..., -9.8332e-01,
-6.0567e-01, -8.7780e-01],
[-1.1396e+00, -9.9892e-01, -4.1314e-01, ..., -1.1045e+00,
-5.9823e-01, -4.5045e-01],
...,
[-9.0459e-01, -1.3472e+00, -5.0742e-01, ..., -1.0385e+00,
-8.7456e-01, -1.2384e+00],
[-1.2746e+00, -8.6500e-01, -5.2734e-01, ..., -5.6142e-01,
-2.5625e-01, -4.2962e-01],
[-1.3880e+00, -8.0214e-01, -3.5584e-01, ..., -4.9343e-01,
-4.3987e-01, -6.0636e-01]],
...,
[[-2.2032e+00, -1.2344e+00, -2.5381e+00, ..., -3.0309e+00,
-2.4475e+00, -1.7766e+00],
[-7.2003e-01, 9.4608e-01, -9.5407e-01, ..., -2.4244e+00,
-2.4559e+00, -2.0034e+00],
[-2.4573e-01, 2.2382e+00, 5.2474e-01, ..., -1.4626e-01,
-4.8585e-03, 2.5881e-01],
...,
[-1.5573e+00, 1.1597e+00, -3.3634e-01, ..., -9.2774e-01,
-2.7120e-01, 2.7715e-01],
[-5.1087e-01, 1.5497e+00, 6.5677e-01, ..., 4.2552e-01,
4.7989e-01, 8.0790e-01],
[-1.5968e+00, -1.0639e+00, -1.6692e+00, ..., -2.2036e+00,
-1.1642e+00, -3.5051e-01]],
[[-1.2965e+00, -1.2051e+00, -2.6043e+00, ..., -2.2948e+00,
-1.9328e+00, -1.8023e+00],
[-6.1341e-01, -1.0050e-01, -3.3354e+00, ..., -2.8961e+00,
-2.8487e+00, -2.8445e+00],
[-2.0903e+00, -1.3993e+00, -5.7235e+00, ..., -4.4808e+00,
-4.4842e+00, -3.6663e+00],
...,
[-1.5994e+00, -4.6976e-01, -4.1509e+00, ..., -2.8369e+00,
-2.7360e+00, -2.3187e+00],
[-1.6261e+00, -6.9218e-01, -3.4870e+00, ..., -2.4493e+00,
-2.2441e+00, -1.6731e+00],
[-2.0757e+00, -1.3391e+00, -2.9814e+00, ..., -2.8591e+00,
-2.4190e+00, -2.2038e+00]],
[[-1.1506e+00, -1.7070e+00, -1.5199e+00, ..., -2.5628e+00,
-2.1405e+00, -1.6723e+00],
[-1.1511e+00, -1.6176e+00, -1.1934e+00, ..., -2.3969e+00,
-1.9406e+00, -1.5927e+00],
[-1.5006e+00, -1.7969e+00, -1.3966e+00, ..., -2.8125e+00,
-2.2460e+00, -1.7507e+00],
...,
[-2.3147e+00, -2.8727e+00, -2.6693e+00, ..., -3.9877e+00,
-3.2375e+00, -2.5355e+00],
[-1.5390e+00, -1.9584e+00, -1.9044e+00, ..., -2.7606e+00,
-2.3127e+00, -1.9111e+00],
[-1.0330e+00, -1.2400e+00, -1.3062e+00, ..., -1.8262e+00,
-1.5216e+00, -1.1779e+00]]],
[[[ 1.1968e-01, 3.3848e-01, -1.3281e+00, ..., -2.3381e+00,
-2.9379e+00, -2.5309e+00],
[-2.1635e+00, -1.5470e+00, -4.6202e+00, ..., -9.5584e+00,
-7.3034e+00, -4.3896e+00],
[ 4.1290e-01, -3.6945e-01, -1.2512e+00, ..., -5.4802e+00,
-5.4146e+00, -2.4517e+00],
...,
[-3.0052e-01, -1.3238e+00, -2.6730e+00, ..., -2.0241e+00,
-1.4660e+00, -1.0585e+00],
[ 1.2519e+00, 6.2263e-01, -1.0221e+00, ..., -2.3861e-01,
5.3063e-01, 6.8167e-01],
[ 8.0557e-01, 2.4613e-01, -5.1622e-01, ..., -2.6142e-01,
3.7154e-01, 3.3261e-01]],
[[-3.2875e+00, -1.4648e+00, -6.2205e+00, ..., -4.6321e+00,
-4.3531e+00, -3.9156e+00],
[-5.8328e-01, 2.1600e+00, -4.2529e+00, ..., -4.5281e+00,
-4.9460e+00, -4.8541e+00],
[ 1.5948e-02, 1.8549e+00, -7.5100e+00, ..., -8.8202e+00,
-8.5950e+00, -7.7337e+00],
...,
[-1.3019e+00, 2.6177e-01, -6.6441e+00, ..., -4.6146e+00,
-4.0997e+00, -3.3774e+00],
[ 1.5186e+00, 2.6423e+00, -1.3682e+00, ..., -5.8242e-01,
-9.2542e-01, -1.0049e+00],
[ 1.0122e+00, 1.6770e+00, -2.4140e+00, ..., -2.1562e+00,
-2.4057e+00, -2.3411e+00]],
[[-1.4617e+00, -2.4149e+00, -2.2255e+00, ..., -5.4542e-01,
-4.3363e-01, -7.5651e-01],
[-2.0775e+00, -2.2270e+00, -1.5878e+00, ..., 3.3428e-03,
-1.4323e-01, -3.3958e-01],
[-2.6109e+00, -2.0421e+00, -8.4928e-01, ..., 2.5977e-01,
2.2027e-01, -4.5222e-02],
...,
[-1.6666e+00, -1.8825e+00, -7.9179e-01, ..., -5.4577e-01,
-1.7965e-01, -3.8604e-01],
[-1.8233e+00, -1.2185e+00, -7.7445e-01, ..., -2.9948e-01,
1.8874e-01, 2.3907e-01],
[-1.7980e+00, -1.1880e+00, -6.5806e-01, ..., -4.5997e-01,
-2.3859e-01, -2.4045e-01]],
...,
[[-2.3574e+00, -1.2689e+00, -2.7217e+00, ..., -3.4582e+00,
-2.4600e+00, -1.2696e+00],
[-1.9572e+00, 1.2937e-01, -1.3813e+00, ..., -3.4987e+00,
-1.8745e+00, -2.0200e+00],
[-5.1553e-01, 2.4596e+00, 9.1845e-01, ..., -3.2525e+00,
1.2388e+00, -3.5922e-01],
...,
[-1.7541e+00, 1.1766e+00, 2.4683e-01, ..., -1.6311e+00,
-6.9955e-01, -9.0183e-02],
[-8.4142e-01, 1.7182e+00, 1.5634e+00, ..., -1.2605e-01,
1.3578e-01, 3.6327e-01],
[-2.1055e+00, -1.0322e+00, -7.9614e-01, ..., -2.5996e+00,
-1.5734e+00, -9.4657e-01]],
[[-2.4259e+00, -2.7129e+00, -4.0877e+00, ..., -1.4009e+00,
-7.1943e-01, -7.6853e-01],
[-1.7066e+00, -1.4029e+00, -4.8088e+00, ..., -1.5952e+00,
-8.4973e-01, -1.3441e+00],
[-3.0259e+00, -2.4816e+00, -6.9202e+00, ..., -2.3389e+00,
-2.9634e+00, -2.5543e+00],
...,
[-2.5261e+00, -1.7516e+00, -4.9849e+00, ..., -1.3882e+00,
-9.3369e-01, -7.5375e-01],
[-2.2190e+00, -1.5637e+00, -3.8934e+00, ..., -1.7133e+00,
-1.0800e+00, -6.4031e-01],
[-2.3130e+00, -1.7903e+00, -3.0800e+00, ..., -2.8839e+00,
-2.0046e+00, -1.8267e+00]],
[[-1.9438e+00, -2.6826e+00, -2.4649e+00, ..., -2.1848e+00,
-1.8182e+00, -1.2351e+00],
[-1.9717e+00, -2.7078e+00, -2.1707e+00, ..., -2.5672e+00,
-2.0278e+00, -1.5454e+00],
[-2.5114e+00, -3.0940e+00, -2.6082e+00, ..., -3.4580e+00,
-2.9767e+00, -1.9617e+00],
...,
[-3.1755e+00, -3.9649e+00, -3.7259e+00, ..., -2.9567e+00,
-2.3760e+00, -1.8238e+00],
[-2.1024e+00, -2.7232e+00, -2.6783e+00, ..., -2.0210e+00,
-1.6624e+00, -1.3493e+00],
[-1.5150e+00, -1.8439e+00, -1.8835e+00, ..., -1.2193e+00,
-1.0483e+00, -7.9622e-01]]]], device='cuda:0'),
tensor([[[[ 1.4766e-01, 3.9454e-01, -1.1567e+00, ..., 4.4886e-01,
1.0096e+00, 6.5090e-01],
[-2.0672e+00, -1.3804e+00, -4.3230e+00, ..., -2.6067e+00,
-7.6222e-01, -4.5239e-02],
[ 4.5859e-01, -2.7398e-01, -1.0991e+00, ..., -2.7554e-01,
4.8828e-01, 6.4824e-01],
...,
[-2.8217e-01, -1.2572e+00, -2.6013e+00, ..., -2.0463e+00,
-1.4200e+00, -9.6312e-01],
[ 1.2499e+00, 6.6841e-01, -9.0418e-01, ..., -3.3904e-01,
5.0745e-01, 6.8124e-01],
[ 8.2295e-01, 3.2292e-01, -4.0232e-01, ..., -3.6854e-01,
2.9517e-01, 2.7653e-01]],
[[-3.2353e+00, -1.4590e+00, -6.1385e+00, ..., -4.8343e+00,
-4.4210e+00, -3.3564e+00],
[-5.6549e-01, 2.1416e+00, -4.2196e+00, ..., -3.3629e+00,
-3.5125e+00, -2.8724e+00],
[ 3.5887e-02, 1.8224e+00, -7.4320e+00, ..., -5.3383e+00,
-5.2219e+00, -4.9075e+00],
...,
[-1.2637e+00, 2.6369e-01, -6.5130e+00, ..., -4.5628e+00,
-4.1033e+00, -3.4229e+00],
[ 1.5280e+00, 2.6188e+00, -1.3047e+00, ..., -8.4152e-01,
-1.1654e+00, -1.2150e+00],
[ 9.9819e-01, 1.6349e+00, -2.3786e+00, ..., -2.2175e+00,
-2.4544e+00, -2.3513e+00]],
[[-1.3754e+00, -2.3376e+00, -2.1599e+00, ..., -5.8176e-01,
-1.8138e-01, -4.6562e-01],
[-2.0182e+00, -2.1896e+00, -1.5378e+00, ..., -3.3012e-01,
9.2265e-02, 5.3016e-02],
[-2.5203e+00, -1.9921e+00, -8.1709e-01, ..., -2.6058e-01,
2.6795e-01, 2.9285e-01],
...,
[-1.5796e+00, -1.8278e+00, -7.7955e-01, ..., -5.1076e-01,
-1.6170e-01, -3.5493e-01],
[-1.7429e+00, -1.1829e+00, -7.7901e-01, ..., -3.3415e-01,
1.5292e-01, 2.1932e-01],
[-1.7195e+00, -1.1135e+00, -6.0034e-01, ..., -5.0946e-01,
-2.7956e-01, -2.4917e-01]],
...,
[[-2.3773e+00, -1.3030e+00, -2.7181e+00, ..., -3.6808e+00,
-2.8560e+00, -1.8363e+00],
[-1.8924e+00, 2.0536e-01, -1.3418e+00, ..., -2.6838e+00,
-2.4519e+00, -1.8894e+00],
[-4.8422e-01, 2.4971e+00, 9.0212e-01, ..., -6.1389e-01,
-8.4536e-02, 1.6782e-01],
...,
[-1.7215e+00, 1.1698e+00, 2.0396e-01, ..., -1.5498e+00,
-7.2778e-01, -1.3830e-01],
[-8.7708e-01, 1.6389e+00, 1.4824e+00, ..., -1.1207e-01,
1.0815e-01, 3.1659e-01],
[-2.1085e+00, -1.0861e+00, -8.8814e-01, ..., -2.3676e+00,
-1.4578e+00, -9.2957e-01]],
[[-2.3728e+00, -2.6242e+00, -4.0253e+00, ..., -1.1325e+00,
-7.2581e-01, -7.5203e-01],
[-1.6563e+00, -1.3418e+00, -4.7645e+00, ..., -1.3415e+00,
-1.2088e+00, -1.5821e+00],
[-3.0096e+00, -2.4438e+00, -6.8834e+00, ..., -3.2478e+00,
-2.7855e+00, -2.2436e+00],
...,
[-2.5093e+00, -1.7231e+00, -4.9763e+00, ..., -1.4316e+00,
-9.9113e-01, -8.6529e-01],
[-2.2006e+00, -1.5437e+00, -3.9082e+00, ..., -1.7108e+00,
-1.1253e+00, -7.3928e-01],
[-2.3002e+00, -1.7832e+00, -3.0860e+00, ..., -2.8122e+00,
-2.0178e+00, -1.8304e+00]],
[[-1.8925e+00, -2.6154e+00, -2.3976e+00, ..., -1.8151e+00,
-1.5058e+00, -1.1537e+00],
[-1.9087e+00, -2.6222e+00, -2.0904e+00, ..., -1.4061e+00,
-1.1170e+00, -8.9775e-01],
[-2.4338e+00, -2.9874e+00, -2.5086e+00, ..., -1.6049e+00,
-1.2095e+00, -8.9906e-01],
...,
[-3.0971e+00, -3.8675e+00, -3.6413e+00, ..., -2.8772e+00,
-2.2998e+00, -1.7717e+00],
[-2.0558e+00, -2.6549e+00, -2.6165e+00, ..., -1.9801e+00,
-1.6272e+00, -1.3154e+00],
[-1.4750e+00, -1.7918e+00, -1.8394e+00, ..., -1.2214e+00,
-1.0348e+00, -7.9275e-01]]],
[[[ 7.6810e-02, 2.4777e-01, -1.3982e+00, ..., -6.0864e-01,
-1.7709e-01, -3.8867e-02],
[-2.1092e+00, -1.5967e+00, -4.5386e+00, ..., -3.7968e+00,
-1.5300e+00, -5.4526e-01],
[ 5.5078e-01, -6.4580e-02, -8.3269e-01, ..., -2.0112e-01,
-2.3647e-01, 2.6189e-01],
...,
[-8.8275e-01, -1.8951e+00, -1.8726e+00, ..., -1.8266e+00,
-1.3540e+00, -9.6693e-01],
[ 7.3252e-01, -5.5576e-02, -1.4064e-01, ..., -1.5341e-02,
7.0446e-01, 7.9150e-01],
[ 1.2633e-01, -6.1633e-01, -5.0039e-01, ..., -1.8107e-01,
4.2332e-01, 3.3999e-01]],
[[-3.2984e+00, -1.8426e+00, -6.2404e+00, ..., -7.1557e+00,
-6.2711e+00, -4.6283e+00],
[-8.2659e-01, 1.4658e+00, -4.3660e+00, ..., -5.4532e+00,
-5.5392e+00, -4.2453e+00],
[-4.0022e-01, 1.0031e+00, -7.4903e+00, ..., -7.9651e+00,
-7.8264e+00, -6.6634e+00],
...,
[ 3.2935e-01, 3.8483e-01, -4.0357e+00, ..., -4.6510e+00,
-4.2251e+00, -3.5164e+00],
[ 2.3214e+00, 2.2752e+00, -1.6641e-01, ..., -8.4674e-01,
-1.2224e+00, -1.2601e+00],
[ 1.3626e+00, 1.1654e+00, -1.7790e+00, ..., -2.2363e+00,
-2.5246e+00, -2.4161e+00]],
[[-1.3514e+00, -2.3642e+00, -2.3516e+00, ..., -1.7582e+00,
-1.2413e+00, -1.3813e+00],
[-1.6892e+00, -1.9603e+00, -1.5181e+00, ..., -1.2477e+00,
-7.2160e-01, -8.0865e-01],
[-2.1098e+00, -1.8505e+00, -9.4782e-01, ..., -1.0968e+00,
-4.4147e-01, -1.7255e-01],
...,
[-1.3912e-01, -6.1512e-01, -4.7420e-01, ..., -4.8818e-01,
-1.9661e-01, -4.4649e-01],
[-7.1029e-01, -2.0649e-01, -4.3620e-01, ..., -2.1432e-01,
2.3542e-01, 2.3114e-01],
[-8.0791e-01, -1.9818e-01, -3.5060e-01, ..., -3.9578e-01,
-1.9068e-01, -2.2566e-01]],
...,
[[-2.0035e+00, -1.1199e+00, -2.3556e+00, ..., -3.2667e+00,
-2.6799e+00, -1.8648e+00],
[-1.6944e+00, -4.1996e-03, -1.3491e+00, ..., -2.5094e+00,
-2.5508e+00, -1.9804e+00],
[-3.6130e-01, 2.2875e+00, 1.0083e+00, ..., 1.0248e-01,
2.3201e-01, 5.1069e-01],
...,
[-2.0834e+00, -5.5321e-02, -1.7526e-01, ..., -1.3610e+00,
-3.8847e-01, 1.0326e-01],
[-1.2080e+00, 8.3601e-01, 9.5578e-01, ..., 1.5215e-02,
3.2759e-01, 4.8006e-01],
[-2.0890e+00, -1.3835e+00, -1.6072e+00, ..., -2.4384e+00,
-1.4460e+00, -9.3133e-01]],
[[-2.4106e+00, -2.8294e+00, -4.1362e+00, ..., -2.3691e+00,
-1.8808e+00, -1.6978e+00],
[-1.7655e+00, -1.6219e+00, -4.7590e+00, ..., -2.8775e+00,
-2.7563e+00, -2.7833e+00],
[-2.8294e+00, -2.4504e+00, -6.5959e+00, ..., -4.4418e+00,
-4.3669e+00, -3.5042e+00],
...,
[-1.3825e+00, 1.9864e-04, -3.3645e+00, ..., -1.4576e+00,
-1.0437e+00, -9.4494e-01],
[-1.6926e+00, -6.1358e-01, -3.0410e+00, ..., -1.7341e+00,
-1.1824e+00, -8.2864e-01],
[-2.3130e+00, -1.4660e+00, -3.1003e+00, ..., -2.9664e+00,
-2.1599e+00, -1.9946e+00]],
[[-1.8046e+00, -2.5377e+00, -2.4654e+00, ..., -2.6933e+00,
-2.2170e+00, -1.6975e+00],
[-1.7986e+00, -2.5067e+00, -2.1660e+00, ..., -2.3916e+00,
-1.8937e+00, -1.5379e+00],
[-2.2471e+00, -2.8001e+00, -2.5219e+00, ..., -2.7348e+00,
-2.1296e+00, -1.6176e+00],
...,
[-2.3977e+00, -2.9754e+00, -2.7493e+00, ..., -2.8630e+00,
-2.2897e+00, -1.7543e+00],
[-1.5597e+00, -2.0280e+00, -1.9503e+00, ..., -1.9518e+00,
-1.6079e+00, -1.2934e+00],
[-1.0172e+00, -1.3109e+00, -1.3421e+00, ..., -1.1973e+00,
-1.0215e+00, -7.7152e-01]]],
[[[ 1.1311e-01, 3.1278e-01, -1.4270e+00, ..., 6.4362e-01,
8.8495e-01, 6.6906e-01],
[-2.1868e+00, -1.5726e+00, -4.7362e+00, ..., -1.3146e+00,
-1.1554e-01, 4.3626e-01],
[ 3.8296e-01, -3.8103e-01, -1.3100e+00, ..., 5.8437e-01,
8.7678e-01, 8.1109e-01],
...,
[-2.8944e-01, -1.3104e+00, -2.6842e+00, ..., -1.2519e+00,
-9.3292e-01, -7.5523e-01],
[ 1.2461e+00, 5.9681e-01, -1.0869e+00, ..., 5.9800e-01,
7.8471e-01, 8.2946e-01],
[ 7.7622e-01, 2.0116e-01, -5.5743e-01, ..., 6.8752e-01,
8.5804e-01, 7.5106e-01]],
[[-3.3493e+00, -1.5706e+00, -6.2989e+00, ..., -5.4367e+00,
-4.6360e+00, -3.6297e+00],
[-6.5339e-01, 2.0640e+00, -4.3292e+00, ..., -4.3071e+00,
-4.1221e+00, -3.3863e+00],
[-8.1984e-02, 1.7401e+00, -7.6184e+00, ..., -6.2072e+00,
-5.8856e+00, -5.4327e+00],
...,
[-1.4347e+00, 9.8875e-02, -6.7317e+00, ..., -5.4845e+00,
-4.8008e+00, -3.9865e+00],
[ 1.4071e+00, 2.5374e+00, -1.4325e+00, ..., -1.0682e+00,
-1.2702e+00, -1.3650e+00],
[ 9.4178e-01, 1.6327e+00, -2.4307e+00, ..., -2.4278e+00,
-2.5985e+00, -2.5534e+00]],
[[-1.5036e+00, -2.4353e+00, -2.2739e+00, ..., -8.4206e-01,
-6.2580e-01, -8.0136e-01],
[-2.0924e+00, -2.2057e+00, -1.5769e+00, ..., -7.0614e-01,
-4.3900e-01, -3.7583e-01],
[-2.6626e+00, -2.0623e+00, -8.3153e-01, ..., -4.1655e-01,
-1.3232e-01, -1.2411e-01],
...,
[-1.6325e+00, -1.8337e+00, -7.5242e-01, ..., -6.7829e-01,
-5.6987e-01, -7.5181e-01],
[-1.7901e+00, -1.2011e+00, -7.5747e-01, ..., -2.2616e-01,
-2.7688e-02, -1.8058e-02],
[-1.7766e+00, -1.1978e+00, -6.8142e-01, ..., -2.8689e-01,
-2.4086e-01, -2.2424e-01]],
...,
[[-2.3569e+00, -1.3284e+00, -2.7266e+00, ..., -2.8874e+00,
-2.3019e+00, -1.5660e+00],
[-2.0232e+00, -3.1131e-03, -1.4289e+00, ..., -1.3381e+00,
-1.4367e+00, -1.2986e+00],
[-5.6679e-01, 2.3722e+00, 8.9453e-01, ..., -7.9007e-02,
2.3414e-01, 2.9372e-01],
...,
[-1.8315e+00, 1.0017e+00, 1.2677e-01, ..., -1.0314e+00,
-2.6649e-01, 2.3095e-01],
[-9.0957e-01, 1.6301e+00, 1.5399e+00, ..., 1.6041e-01,
3.3843e-01, 4.2824e-01],
[-2.1619e+00, -1.1004e+00, -7.9915e-01, ..., -2.3666e+00,
-1.3651e+00, -6.9911e-01]],
[[-2.4613e+00, -2.7860e+00, -4.0542e+00, ..., -1.1853e+00,
-6.6468e-01, -5.2418e-01],
[-1.7541e+00, -1.4727e+00, -4.7724e+00, ..., -1.7971e+00,
-1.5453e+00, -1.7272e+00],
[-3.0856e+00, -2.5541e+00, -6.8938e+00, ..., -3.7925e+00,
-3.2855e+00, -2.6265e+00],
...,
[-2.6007e+00, -1.8830e+00, -4.9507e+00, ..., -2.2169e+00,
-1.7341e+00, -1.3793e+00],
[-2.2693e+00, -1.6705e+00, -3.8859e+00, ..., -2.1856e+00,
-1.4933e+00, -9.1433e-01],
[-2.3390e+00, -1.8698e+00, -3.0883e+00, ..., -2.9034e+00,
-1.9635e+00, -1.6603e+00]],
[[-1.9650e+00, -2.7113e+00, -2.5113e+00, ..., -1.6670e+00,
-1.3861e+00, -1.0383e+00],
[-1.9875e+00, -2.7321e+00, -2.2132e+00, ..., -1.3885e+00,
-1.1346e+00, -8.9235e-01],
[-2.5193e+00, -3.1019e+00, -2.6370e+00, ..., -1.6112e+00,
-1.2383e+00, -8.9744e-01],
...,
[-3.1630e+00, -3.9669e+00, -3.7571e+00, ..., -2.8841e+00,
-2.3020e+00, -1.7443e+00],
[-2.0920e+00, -2.7186e+00, -2.6886e+00, ..., -2.0127e+00,
-1.6518e+00, -1.3224e+00],
[-1.5046e+00, -1.8429e+00, -1.8902e+00, ..., -1.2309e+00,
-1.0564e+00, -7.8115e-01]]],
[[[ 2.2233e-01, 4.9031e-01, 6.5876e-01, ..., -4.4863e-01,
-1.7201e-01, -4.7530e-02],
[-1.9806e+00, -1.2797e+00, -2.6406e+00, ..., -3.4017e+00,
-1.4105e+00, -5.3229e-01],
[-1.4100e-01, -8.4560e-01, 9.5003e-02, ..., -2.5109e-01,
-4.3229e-01, 1.2445e-01],
...,
[-9.0285e-01, -2.0593e+00, -2.0213e+00, ..., -1.8754e+00,
-1.8769e+00, -1.2225e+00],
[ 8.5516e-01, -3.1722e-02, -7.7209e-02, ..., 4.1651e-02,
2.4859e-01, 4.4266e-01],
[ 2.6148e-01, -4.9537e-01, -3.5298e-01, ..., 2.1028e-01,
4.2136e-01, 4.7543e-01]],
[[-1.6944e+00, -1.4146e+00, -4.2665e+00, ..., -6.9322e+00,
-6.1707e+00, -4.5854e+00],
[ 4.0787e-01, 1.3596e+00, -2.5806e+00, ..., -5.5392e+00,
-5.6666e+00, -4.3515e+00],
[ 1.1070e+00, 1.1077e+00, -5.0849e+00, ..., -7.9871e+00,
-7.9398e+00, -6.6349e+00],
...,
[ 3.3840e-01, 3.9557e-01, -4.1337e+00, ..., -7.2911e+00,
-6.9047e+00, -5.3119e+00],
[ 2.5508e+00, 2.5588e+00, 1.8497e-01, ..., -2.1563e+00,
-2.8058e+00, -2.1403e+00],
[ 1.5229e+00, 1.3714e+00, -1.5915e+00, ..., -3.0297e+00,
-3.5811e+00, -3.0233e+00]],
[[-1.4951e-01, -1.1680e+00, -1.3958e+00, ..., -1.4651e+00,
-1.0974e+00, -1.3794e+00],
[-5.1192e-01, -8.3217e-01, -8.5022e-01, ..., -1.0217e+00,
-6.5266e-01, -9.5172e-01],
[-3.5040e-01, -2.5160e-01, -3.9568e-01, ..., -1.1290e+00,
-6.1862e-01, -4.9676e-01],
...,
[-1.0989e-01, -6.7222e-01, -6.1958e-01, ..., -1.0261e+00,
-8.8010e-01, -1.2534e+00],
[-7.4083e-01, -2.7424e-01, -5.8494e-01, ..., -5.5229e-01,
-2.1769e-01, -3.9941e-01],
[-9.0719e-01, -2.9431e-01, -4.6489e-01, ..., -5.1898e-01,
-4.6626e-01, -6.2509e-01]],
...,
[[-2.7979e+00, -2.4159e+00, -2.4783e+00, ..., -3.0305e+00,
-2.4695e+00, -1.8051e+00],
[-2.0239e+00, -8.1364e-01, -1.1692e+00, ..., -2.4700e+00,
-2.5218e+00, -2.0389e+00],
[-1.1104e+00, 7.5943e-01, 5.5851e-01, ..., -6.3524e-02,
4.2024e-02, 2.8812e-01],
...,
[-2.2492e+00, -6.1910e-02, -1.3023e-01, ..., -9.0327e-01,
-1.8355e-01, 3.6110e-01],
[-1.2358e+00, 9.2943e-01, 1.0955e+00, ..., 4.6345e-01,
4.9546e-01, 8.4441e-01],
[-2.1863e+00, -1.4924e+00, -1.6827e+00, ..., -2.1701e+00,
-1.0997e+00, -3.3219e-01]],
[[-1.6033e+00, -1.6260e+00, -2.8215e+00, ..., -2.3129e+00,
-1.9902e+00, -1.8862e+00],
[-7.2202e-01, -1.3875e-01, -3.2076e+00, ..., -2.9309e+00,
-2.9390e+00, -2.9361e+00],
[-2.0804e+00, -1.0408e+00, -5.1937e+00, ..., -4.4825e+00,
-4.5637e+00, -3.7456e+00],
...,
[-1.3485e+00, 1.5304e-01, -3.4629e+00, ..., -3.0071e+00,
-2.8406e+00, -2.3624e+00],
[-1.6586e+00, -4.6863e-01, -3.0694e+00, ..., -2.5769e+00,
-2.3421e+00, -1.6892e+00],
[-2.3009e+00, -1.3447e+00, -3.0539e+00, ..., -2.9040e+00,
-2.4613e+00, -2.1901e+00]],
[[-1.2772e+00, -1.7992e+00, -1.6362e+00, ..., -2.6366e+00,
-2.2022e+00, -1.7123e+00],
[-1.2281e+00, -1.6881e+00, -1.2282e+00, ..., -2.4772e+00,
-2.0033e+00, -1.6426e+00],
[-1.5532e+00, -1.8859e+00, -1.4880e+00, ..., -2.9123e+00,
-2.3365e+00, -1.8187e+00],
...,
[-2.4484e+00, -3.0603e+00, -2.7796e+00, ..., -4.1024e+00,
-3.3146e+00, -2.6182e+00],
[-1.5897e+00, -2.0882e+00, -1.9806e+00, ..., -2.8503e+00,
-2.3780e+00, -1.9776e+00],
[-1.0400e+00, -1.3454e+00, -1.3573e+00, ..., -1.9073e+00,
-1.5696e+00, -1.2228e+00]]],
[[[-1.0540e-01, -3.6578e-02, -2.1480e+00, ..., -5.9255e+00,
-6.3884e-01, -3.1174e-01],
[-2.4186e+00, -2.0639e+00, -5.5520e+00, ..., -3.0071e+01,
-2.0830e+00, -9.2313e-01],
[ 3.6720e-01, -4.2394e-01, -1.3794e+00, ..., -3.5456e+01,
-7.1403e-01, 5.2293e-02],
...,
[-4.7206e-01, -1.2946e+00, -1.4740e+00, ..., -1.3408e+00,
-1.0386e+00, -7.5748e-01],
[ 9.8437e-01, 3.9397e-01, 2.0860e-01, ..., 3.7846e-01,
8.3966e-01, 8.5997e-01],
[ 4.1886e-01, -1.1404e-01, -9.4402e-02, ..., 2.7238e-01,
6.5988e-01, 5.0435e-01]],
[[-3.5249e+00, -1.9549e+00, -6.8168e+00, ..., -1.2728e+01,
-6.7728e+00, -4.9563e+00],
[-9.7947e-01, 1.4756e+00, -4.8821e+00, ..., -1.4639e+01,
-6.1100e+00, -4.6511e+00],
[-6.5220e-01, 9.8109e-01, -8.0378e+00, ..., -1.6545e+01,
-8.5449e+00, -7.1034e+00],
...,
[-4.8101e-02, 3.1703e-01, -4.1334e+00, ..., -4.9598e+00,
-4.4220e+00, -3.6413e+00],
[ 2.1290e+00, 2.3090e+00, -1.0749e-01, ..., -1.0440e+00,
-1.3147e+00, -1.2997e+00],
[ 1.2448e+00, 1.2039e+00, -1.7341e+00, ..., -2.3576e+00,
-2.5595e+00, -2.4215e+00]],
[[-1.6720e+00, -2.6020e+00, -2.5403e+00, ..., -1.5515e+00,
-1.4699e+00, -1.5891e+00],
[-1.9175e+00, -2.1253e+00, -1.6087e+00, ..., -1.1953e-01,
-9.0419e-01, -1.0539e+00],
[-2.6592e+00, -2.3057e+00, -1.1120e+00, ..., -1.8835e+00,
-6.4993e-01, -3.0367e-01],
...,
[-4.0574e-01, -8.7423e-01, -4.8990e-01, ..., -4.7935e-01,
-2.4757e-01, -4.6879e-01],
[-8.5826e-01, -3.8214e-01, -3.9736e-01, ..., -1.5625e-01,
2.1998e-01, 2.2274e-01],
[-9.3480e-01, -3.2266e-01, -2.6631e-01, ..., -2.6050e-01,
-1.2433e-01, -1.7144e-01]],
...,
[[-2.0212e+00, -1.0873e+00, -2.3116e+00, ..., -6.6400e+00,
-2.6129e+00, -1.8659e+00],
[-2.2779e+00, -4.6697e-01, -1.7523e+00, ..., -5.9069e+00,
-2.8424e+00, -2.1507e+00],
[-5.6086e-01, 2.2118e+00, 9.8922e-01, ..., 1.6325e+00,
2.3768e-01, 5.3678e-01],
...,
[-1.8124e+00, 4.3736e-01, -3.3945e-02, ..., -1.2642e+00,
-2.8492e-01, 2.1238e-01],
[-9.6091e-01, 1.0763e+00, 9.3861e-01, ..., -6.0639e-03,
3.3245e-01, 4.8935e-01],
[-1.8959e+00, -1.2539e+00, -1.5495e+00, ..., -2.3882e+00,
-1.3911e+00, -8.1720e-01]],
[[-2.7379e+00, -3.3422e+00, -4.6318e+00, ..., 7.3545e-02,
-2.3496e+00, -2.1375e+00],
[-2.1253e+00, -2.2061e+00, -5.3136e+00, ..., -3.8022e-01,
-3.2234e+00, -3.1461e+00],
[-3.0641e+00, -2.8736e+00, -6.9013e+00, ..., -6.6849e+00,
-4.7354e+00, -3.8113e+00],
...,
[-1.4710e+00, -2.8271e-01, -3.7018e+00, ..., -1.7942e+00,
-1.2908e+00, -1.0266e+00],
[-1.6473e+00, -6.8748e-01, -3.1863e+00, ..., -1.9415e+00,
-1.2799e+00, -8.0405e-01],
[-2.2011e+00, -1.4239e+00, -3.0440e+00, ..., -2.9393e+00,
-2.0401e+00, -1.8310e+00]],
[[-2.0446e+00, -2.8320e+00, -2.7797e+00, ..., -3.9604e+00,
-2.4852e+00, -1.9068e+00],
[-2.0564e+00, -2.8640e+00, -2.5178e+00, ..., -4.9904e+00,
-2.1851e+00, -1.7756e+00],
[-2.5637e+00, -3.2465e+00, -2.9728e+00, ..., -5.8370e+00,
-2.5119e+00, -1.9269e+00],
...,
[-2.3331e+00, -2.8876e+00, -2.6654e+00, ..., -2.8302e+00,
-2.2770e+00, -1.7334e+00],
[-1.5375e+00, -1.9763e+00, -1.9002e+00, ..., -1.9423e+00,
-1.6097e+00, -1.2851e+00],
[-1.0139e+00, -1.2751e+00, -1.3098e+00, ..., -1.1922e+00,
-1.0257e+00, -7.6771e-01]]]], device='cuda:0'),
tensor([[[[ 4.1945e-01, 8.4385e-01, 5.1346e-01, ..., 4.3523e-01,
1.0138e+00, 6.4148e-01],
[-1.3441e+00, -4.2088e-01, -1.9733e+00, ..., -2.6242e+00,
-7.7912e-01, -8.3245e-02],
[ 8.8608e-01, 6.2921e-01, 3.0814e-01, ..., -3.0371e-01,
4.6446e-01, 6.2509e-01],
...,
[ 4.4656e-02, -5.6340e-01, -1.5363e+00, ..., -2.0568e+00,
-1.3654e+00, -9.4813e-01],
[ 1.3238e+00, 1.1151e+00, 5.6059e-01, ..., -3.6167e-01,
5.2431e-01, 6.7760e-01],
[ 9.8233e-01, 8.8507e-01, 5.8823e-01, ..., -3.8611e-01,
2.8074e-01, 2.5901e-01]],
[[-2.7845e+00, -1.5076e+00, -4.7511e+00, ..., -4.8404e+00,
-4.4325e+00, -3.3623e+00],
[-4.7224e-01, 1.6995e+00, -2.8432e+00, ..., -3.3689e+00,
-3.5311e+00, -2.8836e+00],
[ 3.5859e-01, 1.4708e+00, -5.6779e+00, ..., -5.3471e+00,
-5.2249e+00, -4.9050e+00],
...,
[-9.2101e-01, -1.6561e-02, -4.8728e+00, ..., -4.5936e+00,
-4.1315e+00, -3.4592e+00],
[ 1.6863e+00, 2.2744e+00, -3.0309e-01, ..., -9.2636e-01,
-1.2501e+00, -1.2785e+00],
[ 9.6838e-01, 1.2364e+00, -1.8026e+00, ..., -2.2606e+00,
-2.4852e+00, -2.3626e+00]],
[[-7.1767e-01, -1.9348e+00, -1.7700e+00, ..., -5.9752e-01,
-1.8648e-01, -4.6506e-01],
[-1.5211e+00, -1.9903e+00, -1.2722e+00, ..., -3.4655e-01,
7.7926e-02, 3.9407e-02],
[-1.4043e+00, -1.3638e+00, -7.2508e-01, ..., -2.7318e-01,
2.5415e-01, 2.8609e-01],
...,
[-9.0613e-01, -1.4267e+00, -7.8631e-01, ..., -5.2622e-01,
-1.6334e-01, -3.4737e-01],
[-1.1732e+00, -8.6484e-01, -6.1293e-01, ..., -3.4215e-01,
1.5318e-01, 2.1272e-01],
[-1.2087e+00, -6.9723e-01, -3.1043e-01, ..., -5.1298e-01,
-2.7230e-01, -2.4825e-01]],
...,
[[-2.3533e+00, -1.1881e+00, -2.4869e+00, ..., -3.6690e+00,
-2.8435e+00, -1.8380e+00],
[-6.0999e-01, 1.2843e+00, -6.9221e-01, ..., -2.6623e+00,
-2.4448e+00, -1.8960e+00],
[-1.8230e-01, 2.6286e+00, 8.9409e-01, ..., -5.5909e-01,
-6.4168e-02, 1.5641e-01],
...,
[-1.3393e+00, 1.3569e+00, 1.0017e-01, ..., -1.4199e+00,
-7.0546e-01, -1.5200e-01],
[-5.0855e-01, 1.4880e+00, 8.3178e-01, ..., -7.7059e-02,
1.0307e-01, 3.2420e-01],
[-1.6342e+00, -1.1044e+00, -1.5182e+00, ..., -2.3118e+00,
-1.4350e+00, -9.0569e-01]],
[[-1.3896e+00, -1.3600e+00, -3.0502e+00, ..., -1.1231e+00,
-7.2052e-01, -7.5042e-01],
[-6.1789e-01, -1.6663e-01, -3.9458e+00, ..., -1.3397e+00,
-1.1981e+00, -1.5725e+00],
[-2.1538e+00, -1.5061e+00, -6.4371e+00, ..., -3.2607e+00,
-2.7746e+00, -2.2250e+00],
...,
[-1.7900e+00, -9.1619e-01, -4.6212e+00, ..., -1.4199e+00,
-1.0209e+00, -8.9190e-01],
[-1.7148e+00, -9.7067e-01, -3.6893e+00, ..., -1.6979e+00,
-1.1419e+00, -7.7033e-01],
[-2.0495e+00, -1.4280e+00, -3.0273e+00, ..., -2.7826e+00,
-2.0096e+00, -1.8395e+00]],
[[-1.2644e+00, -1.9203e+00, -1.7390e+00, ..., -1.8149e+00,
-1.5014e+00, -1.1469e+00],
[-1.2547e+00, -1.8204e+00, -1.3847e+00, ..., -1.4063e+00,
-1.1099e+00, -8.9051e-01],
[-1.6204e+00, -2.0025e+00, -1.6186e+00, ..., -1.6030e+00,
-1.2023e+00, -8.9183e-01],
...,
[-2.3282e+00, -2.9653e+00, -2.8905e+00, ..., -2.8892e+00,
-2.3137e+00, -1.7824e+00],
[-1.5678e+00, -2.0190e+00, -2.0405e+00, ..., -1.9995e+00,
-1.6437e+00, -1.3271e+00],
[-1.0653e+00, -1.3068e+00, -1.4224e+00, ..., -1.2421e+00,
-1.0465e+00, -8.0159e-01]]],
[[[ 3.0681e-01, 5.8352e-01, 7.2027e-01, ..., 5.2454e-01,
1.0454e+00, 6.3995e-01],
[-2.0875e+00, -1.4275e+00, -3.0675e+00, ..., -2.5302e+00,
-7.5970e-01, -1.0762e-01],
[-1.0489e-01, -9.0509e-01, -1.8862e-01, ..., -9.7125e-02,
6.0301e-01, 6.9500e-01],
...,
[-1.8216e-01, -1.0311e+00, -2.4767e+00, ..., -1.7081e+00,
-1.6439e+00, -1.0764e+00],
[ 1.3066e+00, 7.6916e-01, -1.0233e+00, ..., -3.4355e-01,
2.3980e-02, 3.1033e-01],
[ 7.6077e-01, 2.1945e-01, -7.4423e-01, ..., -2.2267e-01,
1.8776e-01, 3.5482e-01]],
[[-1.6954e+00, -1.1536e+00, -4.1873e+00, ..., -4.8959e+00,
-4.4388e+00, -3.3921e+00],
[ 5.3914e-01, 1.9016e+00, -2.2969e+00, ..., -3.5692e+00,
-3.6466e+00, -2.9827e+00],
[ 1.2758e+00, 1.6468e+00, -4.9083e+00, ..., -5.5255e+00,
-5.3266e+00, -4.9613e+00],
...,
[-1.6194e+00, -3.4999e-01, -7.0398e+00, ..., -7.7302e+00,
-6.9968e+00, -5.4411e+00],
[ 1.2968e+00, 2.2615e+00, -1.5893e+00, ..., -2.4054e+00,
-2.8046e+00, -2.2248e+00],
[ 8.8144e-01, 1.4880e+00, -2.4922e+00, ..., -3.1781e+00,
-3.6129e+00, -3.1253e+00]],
[[-1.3078e-01, -1.1630e+00, -1.3112e+00, ..., -4.9353e-01,
-1.8548e-01, -4.9584e-01],
[-6.3267e-01, -9.0842e-01, -7.9848e-01, ..., -2.6652e-01,
2.9075e-02, -8.5587e-02],
[-6.0338e-01, -4.0493e-01, -4.1292e-01, ..., -3.4534e-01,
9.0520e-02, 7.4209e-02],
...,
[-1.2109e+00, -1.5760e+00, -7.1778e-01, ..., -9.3207e-01,
-7.7965e-01, -1.0654e+00],
[-1.5186e+00, -1.0216e+00, -7.5917e-01, ..., -5.6783e-01,
-2.6667e-01, -3.4311e-01],
[-1.6775e+00, -1.2165e+00, -9.0711e-01, ..., -9.9098e-01,
-8.5702e-01, -8.3421e-01]],
...,
[[-2.9919e+00, -2.4200e+00, -2.7540e+00, ..., -3.4727e+00,
-2.6471e+00, -1.7829e+00],
[-2.2130e+00, -7.4314e-01, -1.5386e+00, ..., -2.5983e+00,
-2.3214e+00, -1.7892e+00],
[-1.0080e+00, 1.2502e+00, 6.3087e-01, ..., -4.5668e-01,
-7.9968e-02, 1.0727e-01],
...,
[-1.7705e+00, 9.2964e-01, -1.1647e-01, ..., -1.1765e+00,
-5.5287e-01, 5.1717e-02],
[-8.1241e-01, 1.6641e+00, 1.4980e+00, ..., 5.1389e-01,
4.8449e-01, 7.4940e-01],
[-2.2129e+00, -1.1874e+00, -9.4458e-01, ..., -2.0513e+00,
-1.0576e+00, -3.4269e-01]],
[[-1.5884e+00, -1.5721e+00, -2.9042e+00, ..., -1.2422e+00,
-8.2934e-01, -8.1795e-01],
[-6.2308e-01, 5.3254e-02, -3.2166e+00, ..., -1.3838e+00,
-1.2509e+00, -1.6068e+00],
[-2.1172e+00, -9.5219e-01, -5.3050e+00, ..., -3.2105e+00,
-2.7448e+00, -2.2692e+00],
...,
[-2.5841e+00, -1.8938e+00, -5.2712e+00, ..., -3.0195e+00,
-2.6161e+00, -2.1164e+00],
[-2.2958e+00, -1.7101e+00, -4.1590e+00, ..., -2.6527e+00,
-2.2537e+00, -1.5770e+00],
[-2.3727e+00, -1.9075e+00, -3.2824e+00, ..., -3.0422e+00,
-2.4670e+00, -2.1499e+00]],
[[-1.3425e+00, -1.8935e+00, -1.6971e+00, ..., -1.7512e+00,
-1.4531e+00, -1.1061e+00],
[-1.2677e+00, -1.7497e+00, -1.2330e+00, ..., -1.3696e+00,
-1.0790e+00, -8.5640e-01],
[-1.5954e+00, -1.9435e+00, -1.4884e+00, ..., -1.5888e+00,
-1.1884e+00, -8.7834e-01],
...,
[-3.0791e+00, -3.8678e+00, -3.7516e+00, ..., -4.1086e+00,
-3.3027e+00, -2.5472e+00],
[-2.0392e+00, -2.6537e+00, -2.6815e+00, ..., -2.8590e+00,
-2.3677e+00, -1.9329e+00],
[-1.4981e+00, -1.8299e+00, -1.9248e+00, ..., -1.9051e+00,
-1.5654e+00, -1.2000e+00]]],
[[[ 5.2067e-01, 9.8555e-01, 7.7172e-01, ..., 8.0341e-01,
1.0704e+00, 8.5672e-01],
[-1.1572e+00, -2.0041e-01, -1.6325e+00, ..., -1.0721e+00,
1.4469e-01, 6.8056e-01],
[ 8.3439e-01, 5.3287e-01, 2.1506e-01, ..., 5.1344e-01,
9.0065e-01, 9.2846e-01],
...,
[-8.9420e-01, -1.8027e+00, -1.8242e+00, ..., -1.6877e+00,
-1.2512e+00, -9.0338e-01],
[ 6.9985e-01, -1.2845e-02, -1.0488e-01, ..., 1.0898e-03,
7.3334e-01, 8.0150e-01],
[ 9.2023e-02, -5.9360e-01, -4.9458e-01, ..., -1.8038e-01,
4.3312e-01, 3.4928e-01]],
[[-2.6737e+00, -1.3006e+00, -4.7349e+00, ..., -5.6947e+00,
-4.8817e+00, -3.8196e+00],
[-4.4277e-01, 1.8303e+00, -3.0536e+00, ..., -4.7066e+00,
-4.4875e+00, -3.6748e+00],
[ 3.8528e-01, 1.6801e+00, -5.9218e+00, ..., -6.8176e+00,
-6.4567e+00, -5.8571e+00],
...,
[ 3.2529e-01, 3.8273e-01, -4.0732e+00, ..., -4.7148e+00,
-4.3039e+00, -3.5185e+00],
[ 2.3002e+00, 2.2386e+00, -2.5389e-01, ..., -9.5582e-01,
-1.3350e+00, -1.2954e+00],
[ 1.3236e+00, 1.1285e+00, -1.8171e+00, ..., -2.3231e+00,
-2.5685e+00, -2.4146e+00]],
[[-6.2933e-01, -1.7804e+00, -1.5648e+00, ..., -6.9840e-01,
-4.9142e-01, -6.8810e-01],
[-1.4985e+00, -1.9191e+00, -1.2077e+00, ..., -6.1940e-01,
-3.4469e-01, -2.9810e-01],
[-1.3529e+00, -1.2009e+00, -5.0310e-01, ..., -3.9352e-01,
-7.8390e-02, -8.8708e-02],
...,
[-1.3283e-01, -5.6228e-01, -4.4836e-01, ..., -4.5697e-01,
-1.7036e-01, -4.2849e-01],
[-6.9002e-01, -1.6369e-01, -4.1066e-01, ..., -1.9755e-01,
2.5437e-01, 2.4360e-01],
[-7.9193e-01, -1.8174e-01, -3.4653e-01, ..., -3.9229e-01,
-1.8669e-01, -2.2167e-01]],
...,
[[-2.3178e+00, -1.2099e+00, -2.6265e+00, ..., -3.1185e+00,
-2.5826e+00, -1.7720e+00],
[-5.6555e-01, 1.3721e+00, -6.1423e-01, ..., -1.5897e+00,
-1.7454e+00, -1.5095e+00],
[-2.0254e-01, 2.5549e+00, 7.2817e-01, ..., -3.5659e-01,
-1.0494e-01, 6.8898e-02],
...,
[-2.0702e+00, -1.8125e-01, -2.1756e-01, ..., -1.3582e+00,
-3.8567e-01, 9.4736e-02],
[-1.1920e+00, 7.4576e-01, 9.1002e-01, ..., -1.4546e-02,
2.9854e-01, 4.6033e-01],
[-2.0744e+00, -1.4107e+00, -1.6138e+00, ..., -2.4250e+00,
-1.4367e+00, -9.1855e-01]],
[[-1.4131e+00, -1.3190e+00, -2.9103e+00, ..., -1.1483e+00,
-6.5224e-01, -5.2009e-01],
[-7.1697e-01, -1.8909e-01, -3.8888e+00, ..., -1.9542e+00,
-1.6880e+00, -1.8294e+00],
[-2.3871e+00, -1.6970e+00, -6.6104e+00, ..., -4.1598e+00,
-3.5971e+00, -2.8354e+00],
...,
[-1.3580e+00, -2.4747e-02, -3.4091e+00, ..., -1.5244e+00,
-1.0807e+00, -9.5556e-01],
[-1.6689e+00, -6.4520e-01, -3.0726e+00, ..., -1.7880e+00,
-1.2025e+00, -8.5022e-01],
[-2.2910e+00, -1.4876e+00, -3.1203e+00, ..., -2.9854e+00,
-2.1525e+00, -1.9867e+00]],
[[-1.2716e+00, -1.8828e+00, -1.6384e+00, ..., -1.7860e+00,
-1.4791e+00, -1.1024e+00],
[-1.3173e+00, -1.8433e+00, -1.3383e+00, ..., -1.5912e+00,
-1.2907e+00, -1.0077e+00],
[-1.7419e+00, -2.0864e+00, -1.6226e+00, ..., -1.9000e+00,
-1.4574e+00, -1.0715e+00],
...,
[-2.4179e+00, -2.9787e+00, -2.7514e+00, ..., -2.8792e+00,
-2.2984e+00, -1.7590e+00],
[-1.5795e+00, -2.0362e+00, -1.9603e+00, ..., -1.9696e+00,
-1.6187e+00, -1.2920e+00],
[-1.0317e+00, -1.3218e+00, -1.3506e+00, ..., -1.2065e+00,
-1.0214e+00, -7.7197e-01]]],
[[[ 1.5616e-01, 3.9651e-01, -1.1218e+00, ..., 6.2377e-01,
8.9662e-01, 6.5256e-01],
[-2.0287e+00, -1.4062e+00, -4.2640e+00, ..., -1.4087e+00,
-1.8422e-01, 3.6454e-01],
[ 4.2486e-01, -2.8513e-01, -1.0948e+00, ..., 5.7712e-01,
8.9055e-01, 8.3074e-01],
...,
[-2.4224e-01, -1.2645e+00, -2.5364e+00, ..., -1.3704e+00,
-1.0230e+00, -8.2913e-01],
[ 1.2464e+00, 6.2686e-01, -8.7563e-01, ..., 4.9007e-01,
7.4869e-01, 7.9589e-01],
[ 8.0968e-01, 2.8379e-01, -4.1275e-01, ..., 5.2672e-01,
7.8684e-01, 6.8978e-01]],
[[-3.1814e+00, -1.4276e+00, -6.1045e+00, ..., -5.3269e+00,
-4.5717e+00, -3.5795e+00],
[-5.7685e-01, 2.0810e+00, -4.2102e+00, ..., -4.1970e+00,
-4.0296e+00, -3.3155e+00],
[ 1.9739e-03, 1.8037e+00, -7.3509e+00, ..., -6.0613e+00,
-5.7362e+00, -5.3358e+00],
...,
[-1.2111e+00, 3.2242e-01, -6.4689e+00, ..., -5.3457e+00,
-4.6937e+00, -3.9047e+00],
[ 1.5169e+00, 2.5787e+00, -1.3979e+00, ..., -9.6561e-01,
-1.1769e+00, -1.3059e+00],
[ 9.8410e-01, 1.5942e+00, -2.4410e+00, ..., -2.3742e+00,
-2.5664e+00, -2.5489e+00]],
[[-1.3474e+00, -2.2939e+00, -2.1109e+00, ..., -8.0541e-01,
-5.8805e-01, -7.6858e-01],
[-1.9777e+00, -2.1690e+00, -1.5365e+00, ..., -6.7526e-01,
-4.1753e-01, -3.5541e-01],
[-2.4990e+00, -1.9981e+00, -8.2957e-01, ..., -4.0381e-01,
-1.2212e-01, -1.0415e-01],
...,
[-1.5925e+00, -1.8062e+00, -7.3615e-01, ..., -6.7087e-01,
-5.6643e-01, -7.5058e-01],
[-1.7603e+00, -1.1882e+00, -7.5832e-01, ..., -2.4396e-01,
-3.9310e-02, -3.5382e-02],
[-1.7261e+00, -1.1219e+00, -6.0366e-01, ..., -3.2246e-01,
-2.7879e-01, -2.6017e-01]],
...,
[[-2.2965e+00, -1.2479e+00, -2.7104e+00, ..., -2.8568e+00,
-2.2764e+00, -1.5409e+00],
[-1.8650e+00, 2.2695e-01, -1.3181e+00, ..., -1.3999e+00,
-1.4503e+00, -1.2961e+00],
[-4.9281e-01, 2.5042e+00, 9.1604e-01, ..., -1.1960e-01,
2.3544e-01, 3.0041e-01],
...,
[-1.6942e+00, 1.1980e+00, 1.7073e-01, ..., -1.0759e+00,
-2.8253e-01, 2.3028e-01],
[-8.5813e-01, 1.6557e+00, 1.4191e+00, ..., 1.7459e-01,
3.7090e-01, 4.6590e-01],
[-2.0624e+00, -9.9852e-01, -8.2794e-01, ..., -2.3902e+00,
-1.3570e+00, -7.0182e-01]],
[[-2.3686e+00, -2.6282e+00, -4.0404e+00, ..., -1.2241e+00,
-6.9407e-01, -5.3719e-01],
[-1.6822e+00, -1.3624e+00, -4.7589e+00, ..., -1.7629e+00,
-1.4968e+00, -1.6854e+00],
[-2.9978e+00, -2.4392e+00, -6.8632e+00, ..., -3.7092e+00,
-3.1987e+00, -2.5627e+00],
...,
[-2.5074e+00, -1.7272e+00, -5.0207e+00, ..., -2.0821e+00,
-1.6308e+00, -1.3205e+00],
[-2.1939e+00, -1.5329e+00, -3.8991e+00, ..., -2.0806e+00,
-1.4141e+00, -8.7650e-01],
[-2.2741e+00, -1.7329e+00, -3.0593e+00, ..., -2.8838e+00,
-1.9689e+00, -1.6948e+00]],
[[-1.8815e+00, -2.5941e+00, -2.3800e+00, ..., -1.6605e+00,
-1.3790e+00, -1.0329e+00],
[-1.9106e+00, -2.6201e+00, -2.0850e+00, ..., -1.3801e+00,
-1.1155e+00, -8.7876e-01],
[-2.4421e+00, -2.9991e+00, -2.5116e+00, ..., -1.6063e+00,
-1.2215e+00, -8.8554e-01],
...,
[-3.0983e+00, -3.8760e+00, -3.6436e+00, ..., -2.8698e+00,
-2.2960e+00, -1.7370e+00],
[-2.0690e+00, -2.6793e+00, -2.6416e+00, ..., -2.0003e+00,
-1.6508e+00, -1.3193e+00],
[-1.4907e+00, -1.8149e+00, -1.8708e+00, ..., -1.2293e+00,
-1.0609e+00, -7.8039e-01]]],
[[[ 3.1617e-01, 5.8846e-01, 7.1733e-01, ..., 4.8372e-01,
1.0080e+00, 6.1720e-01],
[-2.1125e+00, -1.4340e+00, -3.0798e+00, ..., -2.5883e+00,
-8.1107e-01, -1.3137e-01],
[-1.3676e-01, -9.1852e-01, -1.7816e-01, ..., -1.0338e-01,
5.8234e-01, 6.8611e-01],
...,
[-2.1145e-01, -1.1928e+00, -2.7334e+00, ..., -1.5162e+00,
-1.4293e+00, -8.3124e-01],
[ 1.4475e+00, 8.1443e-01, -1.0313e+00, ..., -3.0884e-01,
5.6073e-02, 3.5908e-01],
[ 8.5229e-01, 2.3762e-01, -7.6062e-01, ..., -2.2046e-01,
1.7469e-01, 3.5466e-01]],
[[-1.6958e+00, -1.1216e+00, -4.1402e+00, ..., -4.8837e+00,
-4.4260e+00, -3.4123e+00],
[ 5.6990e-01, 1.9560e+00, -2.2470e+00, ..., -3.5256e+00,
-3.6166e+00, -2.9828e+00],
[ 1.3178e+00, 1.7248e+00, -4.9083e+00, ..., -5.4251e+00,
-5.2830e+00, -4.9435e+00],
...,
[-1.3491e+00, 1.1870e-01, -6.8507e+00, ..., -7.7604e+00,
-7.0723e+00, -5.5263e+00],
[ 1.7675e+00, 2.8255e+00, -1.1066e+00, ..., -2.7441e+00,
-3.1262e+00, -2.4760e+00],
[ 1.2050e+00, 1.8331e+00, -2.3075e+00, ..., -3.3382e+00,
-3.7275e+00, -3.1621e+00]],
[[-1.3552e-01, -1.1708e+00, -1.3174e+00, ..., -4.8735e-01,
-1.7789e-01, -4.9130e-01],
[-6.2755e-01, -8.9350e-01, -7.7753e-01, ..., -2.8346e-01,
2.5956e-03, -9.4513e-02],
[-5.7908e-01, -3.8282e-01, -3.8038e-01, ..., -3.4902e-01,
8.0719e-02, 7.8759e-02],
...,
[-1.4244e+00, -1.7815e+00, -7.5870e-01, ..., -8.7469e-01,
-7.6193e-01, -1.0171e+00],
[-1.7658e+00, -1.1843e+00, -8.2950e-01, ..., -5.6482e-01,
-2.9241e-01, -3.6642e-01],
[-1.9264e+00, -1.3446e+00, -9.4355e-01, ..., -1.0196e+00,
-8.7150e-01, -8.3996e-01]],
...,
[[-3.0023e+00, -2.4405e+00, -2.8153e+00, ..., -3.4254e+00,
-2.6114e+00, -1.7701e+00],
[-2.1960e+00, -7.2027e-01, -1.5418e+00, ..., -2.5757e+00,
-2.2855e+00, -1.7973e+00],
[-1.0119e+00, 1.2449e+00, 6.2018e-01, ..., -4.9436e-01,
-4.8912e-02, 1.2984e-01],
...,
[-1.8893e+00, 1.2255e+00, 4.1245e-03, ..., -1.1051e+00,
-5.1469e-01, 7.9207e-03],
[-8.0584e-01, 1.9514e+00, 1.6705e+00, ..., 4.7476e-01,
4.3092e-01, 7.0627e-01],
[-2.2345e+00, -1.1234e+00, -9.5581e-01, ..., -1.7955e+00,
-9.0741e-01, -2.5847e-01]],
[[-1.5676e+00, -1.5452e+00, -2.8851e+00, ..., -1.2542e+00,
-8.1133e-01, -8.1355e-01],
[-5.7509e-01, 1.0580e-01, -3.2381e+00, ..., -1.3683e+00,
-1.2096e+00, -1.5709e+00],
[-2.0632e+00, -8.8033e-01, -5.3817e+00, ..., -3.2047e+00,
-2.7188e+00, -2.2255e+00],
...,
[-2.5123e+00, -1.6377e+00, -5.5204e+00, ..., -3.1717e+00,
-2.8141e+00, -2.2375e+00],
[-2.2398e+00, -1.5212e+00, -4.3028e+00, ..., -2.7301e+00,
-2.3854e+00, -1.7002e+00],
[-2.4017e+00, -1.8085e+00, -3.3299e+00, ..., -2.9842e+00,
-2.4779e+00, -2.1464e+00]],
[[-1.3429e+00, -1.9050e+00, -1.6932e+00, ..., -1.7621e+00,
-1.4601e+00, -1.1132e+00],
[-1.2736e+00, -1.7707e+00, -1.2345e+00, ..., -1.3793e+00,
-1.0894e+00, -8.6816e-01],
[-1.6041e+00, -1.9521e+00, -1.4829e+00, ..., -1.6103e+00,
-1.2006e+00, -8.8742e-01],
...,
[-3.2279e+00, -4.0385e+00, -3.8268e+00, ..., -4.0071e+00,
-3.2156e+00, -2.4860e+00],
[-2.1476e+00, -2.7796e+00, -2.7438e+00, ..., -2.8195e+00,
-2.3298e+00, -1.8985e+00],
[-1.5665e+00, -1.9099e+00, -1.9707e+00, ..., -1.9135e+00,
-1.5511e+00, -1.1920e+00]]]], device='cuda:0'),
tensor([[[[ 4.3866e-01, 7.2026e-01, 5.6183e-01, ..., 7.8656e-01,
1.0603e+00, 8.3471e-01],
[-1.1765e+00, -6.3016e-01, -2.2340e+00, ..., -1.2428e+00,
7.9690e-02, 6.3889e-01],
[ 6.8354e-01, 2.9495e-01, -2.5558e-01, ..., 5.3488e-01,
9.4993e-01, 9.2746e-01],
...,
[-8.7674e-01, -1.8868e+00, -1.9382e+00, ..., -1.9007e+00,
-1.3196e+00, -9.5730e-01],
[ 7.4313e-01, -3.6011e-02, -2.1212e-01, ..., -1.2896e-01,
7.2470e-01, 7.9055e-01],
[ 1.5588e-01, -5.8035e-01, -5.4374e-01, ..., -2.3875e-01,
4.4673e-01, 3.2365e-01]],
[[-2.6103e+00, -1.4036e+00, -4.7379e+00, ..., -5.8230e+00,
-5.0082e+00, -3.8751e+00],
[-6.4074e-01, 1.3583e+00, -3.3165e+00, ..., -4.6661e+00,
-4.4916e+00, -3.6348e+00],
[ 2.0023e-02, 1.1030e+00, -6.0921e+00, ..., -6.8073e+00,
-6.5160e+00, -5.8869e+00],
...,
[ 2.9184e-01, 3.4616e-01, -4.0135e+00, ..., -4.6411e+00,
-4.2501e+00, -3.4741e+00],
[ 2.2902e+00, 2.2572e+00, -1.6143e-01, ..., -8.1793e-01,
-1.2117e+00, -1.2038e+00],
[ 1.3444e+00, 1.1651e+00, -1.7783e+00, ..., -2.2391e+00,
-2.5447e+00, -2.3959e+00]],
[[-5.2314e-01, -1.5144e+00, -1.1558e+00, ..., -7.5920e-01,
-5.2999e-01, -7.3855e-01],
[-1.2750e+00, -1.6123e+00, -8.6223e-01, ..., -6.7922e-01,
-3.5630e-01, -3.0167e-01],
[-1.1661e+00, -1.1294e+00, -5.9671e-01, ..., -3.9231e-01,
-2.4416e-02, -4.2701e-02],
...,
[-1.2794e-01, -5.9516e-01, -4.7243e-01, ..., -4.9471e-01,
-1.8562e-01, -4.3727e-01],
[-7.0201e-01, -1.9701e-01, -4.4779e-01, ..., -2.1654e-01,
2.5555e-01, 2.4129e-01],
[-7.8392e-01, -1.7772e-01, -3.6278e-01, ..., -4.2669e-01,
-1.9151e-01, -2.2722e-01]],
...,
[[-2.0625e+00, -1.3484e+00, -2.7504e+00, ..., -3.3327e+00,
-2.7651e+00, -1.8630e+00],
[-5.0846e-01, 8.5010e-01, -1.3453e+00, ..., -1.6372e+00,
-1.8438e+00, -1.5788e+00],
[-2.2152e-01, 2.0305e+00, 2.2127e-01, ..., -3.5339e-01,
-1.2995e-01, 7.8530e-02],
...,
[-2.0858e+00, -1.0144e-01, -1.3913e-01, ..., -1.3133e+00,
-4.3858e-01, 7.8729e-02],
[-1.1891e+00, 8.2189e-01, 9.8619e-01, ..., 7.0077e-02,
2.8867e-01, 4.9880e-01],
[-2.1169e+00, -1.4094e+00, -1.5814e+00, ..., -2.4085e+00,
-1.4916e+00, -9.3018e-01]],
[[-1.4096e+00, -1.3313e+00, -2.7696e+00, ..., -1.0778e+00,
-5.4505e-01, -4.4803e-01],
[-8.4414e-01, -4.0648e-01, -3.6770e+00, ..., -1.8687e+00,
-1.6172e+00, -1.8332e+00],
[-2.4175e+00, -1.9026e+00, -6.3392e+00, ..., -4.1730e+00,
-3.6246e+00, -2.8832e+00],
...,
[-1.3873e+00, -2.4363e-02, -3.3603e+00, ..., -1.4150e+00,
-1.0007e+00, -9.4202e-01],
[-1.7082e+00, -6.4243e-01, -3.0002e+00, ..., -1.7060e+00,
-1.1330e+00, -8.2181e-01],
[-2.3271e+00, -1.4842e+00, -3.0710e+00, ..., -2.9374e+00,
-2.1146e+00, -1.9881e+00]],
[[-1.2101e+00, -1.8231e+00, -1.7133e+00, ..., -1.8394e+00,
-1.5241e+00, -1.1383e+00],
[-1.2735e+00, -1.8120e+00, -1.4608e+00, ..., -1.6001e+00,
-1.2991e+00, -1.0165e+00],
[-1.6779e+00, -2.0516e+00, -1.7199e+00, ..., -1.8878e+00,
-1.4523e+00, -1.0619e+00],
...,
[-2.3932e+00, -2.9629e+00, -2.7506e+00, ..., -2.8862e+00,
-2.3096e+00, -1.7788e+00],
[-1.5498e+00, -2.0177e+00, -1.9651e+00, ..., -1.9695e+00,
-1.6226e+00, -1.3108e+00],
[-1.0067e+00, -1.3070e+00, -1.3517e+00, ..., -1.2145e+00,
-1.0311e+00, -7.8688e-01]]],
[[[ 2.1731e-01, 4.9094e-01, 6.2828e-01, ..., -3.1076e-01,
-7.5716e-02, 4.0922e-02],
[-2.0134e+00, -1.3125e+00, -2.8408e+00, ..., -3.1527e+00,
-1.2602e+00, -4.1362e-01],
[-1.2262e-01, -8.7658e-01, 4.1424e-02, ..., -1.1682e-01,
-3.4409e-01, 1.8320e-01],
...,
[-9.7455e-01, -2.1210e+00, -2.0989e+00, ..., -1.9443e+00,
-1.8479e+00, -1.2531e+00],
[ 7.9504e-01, -1.1202e-01, -1.6993e-01, ..., 3.4756e-03,
2.9369e-01, 4.6415e-01],
[ 2.1636e-01, -5.5351e-01, -3.9843e-01, ..., 2.1958e-01,
4.8784e-01, 5.1275e-01]],
[[-1.6995e+00, -1.3766e+00, -4.2629e+00, ..., -6.8626e+00,
-6.0661e+00, -4.5391e+00],
[ 4.5968e-01, 1.4918e+00, -2.5388e+00, ..., -5.4922e+00,
-5.5811e+00, -4.3143e+00],
[ 1.1919e+00, 1.1988e+00, -5.0915e+00, ..., -7.8916e+00,
-7.8556e+00, -6.6034e+00],
...,
[ 3.5557e-01, 3.2221e-01, -4.1333e+00, ..., -7.1183e+00,
-6.7087e+00, -5.2046e+00],
[ 2.5369e+00, 2.4971e+00, 1.4441e-01, ..., -1.9440e+00,
-2.5899e+00, -2.0811e+00],
[ 1.5037e+00, 1.3237e+00, -1.5929e+00, ..., -2.9600e+00,
-3.4954e+00, -3.0663e+00]],
[[-1.5236e-01, -1.1906e+00, -1.4322e+00, ..., -1.3942e+00,
-1.0458e+00, -1.3400e+00],
[-5.2656e-01, -8.4270e-01, -8.9235e-01, ..., -9.6062e-01,
-6.0879e-01, -8.9784e-01],
[-3.4211e-01, -2.2400e-01, -4.3579e-01, ..., -1.0876e+00,
-5.8460e-01, -4.6338e-01],
...,
[-6.9170e-02, -6.0419e-01, -6.1187e-01, ..., -1.0122e+00,
-8.7718e-01, -1.2723e+00],
[-7.0739e-01, -2.2076e-01, -5.7042e-01, ..., -5.6195e-01,
-2.5296e-01, -4.3448e-01],
[-8.5583e-01, -2.6233e-01, -4.8413e-01, ..., -5.0969e-01,
-4.5187e-01, -6.3144e-01]],
...,
[[-2.8577e+00, -2.4692e+00, -2.4627e+00, ..., -3.0814e+00,
-2.4671e+00, -1.8025e+00],
[-2.1015e+00, -8.2138e-01, -1.1218e+00, ..., -2.5032e+00,
-2.5002e+00, -1.9947e+00],
[-1.1058e+00, 8.2428e-01, 7.6733e-01, ..., -1.9468e-01,
-4.5206e-02, 2.6869e-01],
...,
[-2.2921e+00, -2.6632e-01, -2.1940e-01, ..., -9.2580e-01,
-2.7589e-01, 2.9053e-01],
[-1.2410e+00, 8.2795e-01, 1.0328e+00, ..., 4.8623e-01,
4.9169e-01, 8.2384e-01],
[-2.2097e+00, -1.5486e+00, -1.7228e+00, ..., -2.1696e+00,
-1.1345e+00, -3.1894e-01]],
[[-1.6339e+00, -1.6315e+00, -2.7848e+00, ..., -2.2743e+00,
-1.9475e+00, -1.8404e+00],
[-6.8626e-01, -9.4298e-02, -3.1354e+00, ..., -2.9068e+00,
-2.9060e+00, -2.9093e+00],
[-2.0613e+00, -9.8981e-01, -5.1158e+00, ..., -4.4742e+00,
-4.5270e+00, -3.7321e+00],
...,
[-1.3445e+00, 1.9262e-01, -3.3159e+00, ..., -2.8158e+00,
-2.7490e+00, -2.3260e+00],
[-1.6852e+00, -4.8496e-01, -2.9875e+00, ..., -2.4529e+00,
-2.2772e+00, -1.6902e+00],
[-2.3525e+00, -1.3956e+00, -3.0329e+00, ..., -2.8606e+00,
-2.4317e+00, -2.2082e+00]],
[[-1.3023e+00, -1.8373e+00, -1.6508e+00, ..., -2.5932e+00,
-2.1704e+00, -1.6909e+00],
[-1.2434e+00, -1.7126e+00, -1.2224e+00, ..., -2.4243e+00,
-1.9653e+00, -1.6119e+00],
[-1.5582e+00, -1.8982e+00, -1.4772e+00, ..., -2.8454e+00,
-2.2820e+00, -1.7755e+00],
...,
[-2.4476e+00, -3.0547e+00, -2.8007e+00, ..., -4.0232e+00,
-3.2659e+00, -2.5615e+00],
[-1.5729e+00, -2.0736e+00, -1.9801e+00, ..., -2.7864e+00,
-2.3333e+00, -1.9328e+00],
[-1.0169e+00, -1.3251e+00, -1.3471e+00, ..., -1.8515e+00,
-1.5410e+00, -1.1900e+00]]],
[[[ 2.9015e-01, 5.7759e-01, 7.1438e-01, ..., 5.3416e-01,
1.0369e+00, 6.5697e-01],
[-2.0320e+00, -1.2902e+00, -2.8846e+00, ..., -2.4145e+00,
-7.1796e-01, -6.5060e-02],
[-1.1629e-01, -7.9156e-01, -8.2268e-02, ..., -6.2872e-02,
5.8714e-01, 7.1939e-01],
...,
[-2.0605e-01, -1.2031e+00, -2.7292e+00, ..., -1.5604e+00,
-1.5009e+00, -8.9123e-01],
[ 1.4483e+00, 7.8365e-01, -1.1033e+00, ..., -3.9075e-01,
-1.3769e-02, 3.0616e-01],
[ 8.3041e-01, 1.8233e-01, -8.2211e-01, ..., -2.8751e-01,
1.2734e-01, 3.2325e-01]],
[[-1.7008e+00, -1.1539e+00, -4.1299e+00, ..., -4.8734e+00,
-4.4094e+00, -3.4441e+00],
[ 4.8191e-01, 1.8489e+00, -2.3122e+00, ..., -3.5732e+00,
-3.6547e+00, -3.0748e+00],
[ 1.1927e+00, 1.6302e+00, -4.9128e+00, ..., -5.5340e+00,
-5.3465e+00, -5.0225e+00],
...,
[-1.4077e+00, 8.0629e-02, -6.9937e+00, ..., -7.8633e+00,
-7.1652e+00, -5.5970e+00],
[ 1.7011e+00, 2.7787e+00, -1.2264e+00, ..., -2.7308e+00,
-3.1197e+00, -2.4707e+00],
[ 1.1804e+00, 1.8225e+00, -2.3617e+00, ..., -3.3135e+00,
-3.7223e+00, -3.1755e+00]],
[[-1.2843e-01, -1.1389e+00, -1.2810e+00, ..., -4.9751e-01,
-1.9921e-01, -4.9091e-01],
[-5.9528e-01, -8.6912e-01, -7.4644e-01, ..., -2.8010e-01,
-6.2795e-03, -1.0349e-01],
[-5.7178e-01, -4.0208e-01, -3.9200e-01, ..., -3.3757e-01,
4.6842e-02, 5.1582e-02],
...,
[-1.4605e+00, -1.8019e+00, -7.5931e-01, ..., -8.8008e-01,
-7.7442e-01, -1.0557e+00],
[-1.7923e+00, -1.1967e+00, -8.1811e-01, ..., -5.5641e-01,
-2.9547e-01, -3.7678e-01],
[-1.9587e+00, -1.3868e+00, -9.6807e-01, ..., -1.0463e+00,
-9.2735e-01, -9.0888e-01]],
...,
[[-2.9178e+00, -2.3795e+00, -2.7041e+00, ..., -3.3838e+00,
-2.5857e+00, -1.7487e+00],
[-2.1198e+00, -7.0205e-01, -1.5160e+00, ..., -2.5323e+00,
-2.2500e+00, -1.7866e+00],
[-9.7824e-01, 1.2610e+00, 5.7244e-01, ..., -4.3945e-01,
-2.5179e-03, 1.2410e-01],
...,
[-1.8907e+00, 1.2237e+00, -5.6819e-03, ..., -1.0918e+00,
-5.4121e-01, -2.3442e-02],
[-8.1849e-01, 1.9652e+00, 1.7167e+00, ..., 5.5922e-01,
4.8349e-01, 7.3001e-01],
[-2.2378e+00, -1.1009e+00, -9.0018e-01, ..., -1.8088e+00,
-8.9486e-01, -2.3654e-01]],
[[-1.5572e+00, -1.5666e+00, -2.8927e+00, ..., -1.2423e+00,
-8.1805e-01, -8.1747e-01],
[-6.1974e-01, 1.2164e-03, -3.2955e+00, ..., -1.4009e+00,
-1.2611e+00, -1.5850e+00],
[-2.0738e+00, -9.7680e-01, -5.4039e+00, ..., -3.2115e+00,
-2.7670e+00, -2.2395e+00],
...,
[-2.5423e+00, -1.6918e+00, -5.5115e+00, ..., -3.1586e+00,
-2.8046e+00, -2.2316e+00],
[-2.2625e+00, -1.5581e+00, -4.3018e+00, ..., -2.7456e+00,
-2.3984e+00, -1.7039e+00],
[-2.4092e+00, -1.8319e+00, -3.3519e+00, ..., -3.0140e+00,
-2.5100e+00, -2.1834e+00]],
[[-1.3198e+00, -1.8730e+00, -1.6833e+00, ..., -1.7508e+00,
-1.4446e+00, -1.0947e+00],
[-1.2586e+00, -1.7564e+00, -1.2458e+00, ..., -1.3777e+00,
-1.0903e+00, -8.6487e-01],
[-1.6019e+00, -1.9532e+00, -1.5126e+00, ..., -1.6082e+00,
-1.2083e+00, -8.9387e-01],
...,
[-3.2649e+00, -4.0932e+00, -3.8788e+00, ..., -4.0746e+00,
-3.2700e+00, -2.5213e+00],
[-2.1674e+00, -2.8154e+00, -2.7758e+00, ..., -2.8509e+00,
-2.3576e+00, -1.9186e+00],
[-1.5863e+00, -1.9353e+00, -1.9865e+00, ..., -1.9339e+00,
-1.5700e+00, -1.2004e+00]]],
[[[ 1.2392e-01, 3.4801e-01, -1.3089e+00, ..., 4.7040e-01,
1.0048e+00, 6.5331e-01],
[-2.1847e+00, -1.5366e+00, -4.5739e+00, ..., -2.4891e+00,
-7.0665e-01, -2.6286e-02],
[ 3.8049e-01, -3.4264e-01, -1.2040e+00, ..., -1.9346e-01,
5.3046e-01, 6.5986e-01],
...,
[-2.7410e-01, -1.2896e+00, -2.6673e+00, ..., -1.9940e+00,
-1.3505e+00, -8.9014e-01],
[ 1.2393e+00, 6.2527e-01, -1.0237e+00, ..., -2.8262e-01,
5.4139e-01, 6.9991e-01],
[ 7.9356e-01, 2.6705e-01, -4.9777e-01, ..., -2.8525e-01,
3.3395e-01, 3.0010e-01]],
[[-3.2866e+00, -1.4555e+00, -6.2318e+00, ..., -4.8799e+00,
-4.4323e+00, -3.3721e+00],
[-5.8657e-01, 2.1922e+00, -4.2700e+00, ..., -3.4377e+00,
-3.5662e+00, -2.9137e+00],
[ 8.6808e-03, 1.8895e+00, -7.5379e+00, ..., -5.4250e+00,
-5.2889e+00, -4.9470e+00],
...,
[-1.3777e+00, 1.4848e-01, -6.6745e+00, ..., -4.6698e+00,
-4.1986e+00, -3.4899e+00],
[ 1.4160e+00, 2.5309e+00, -1.4429e+00, ..., -9.6801e-01,
-1.2494e+00, -1.2791e+00],
[ 9.2883e-01, 1.6067e+00, -2.4268e+00, ..., -2.2794e+00,
-2.4840e+00, -2.3628e+00]],
[[-1.4409e+00, -2.4115e+00, -2.2334e+00, ..., -6.1863e-01,
-2.2340e-01, -4.8434e-01],
[-2.0745e+00, -2.2447e+00, -1.5882e+00, ..., -3.7478e-01,
3.5326e-02, 7.3728e-03],
[-2.6285e+00, -2.0860e+00, -8.6730e-01, ..., -2.9025e-01,
2.4131e-01, 2.6595e-01],
...,
[-1.6217e+00, -1.8081e+00, -7.4914e-01, ..., -5.0876e-01,
-1.6071e-01, -3.4535e-01],
[-1.7467e+00, -1.1580e+00, -7.4133e-01, ..., -3.1571e-01,
1.5268e-01, 2.1315e-01],
[-1.7180e+00, -1.1445e+00, -6.5073e-01, ..., -4.8143e-01,
-2.5735e-01, -2.3070e-01]],
...,
[[-2.3704e+00, -1.3249e+00, -2.7798e+00, ..., -3.6368e+00,
-2.8167e+00, -1.8316e+00],
[-1.9532e+00, 1.3691e-01, -1.4135e+00, ..., -2.5779e+00,
-2.3587e+00, -1.8133e+00],
[-5.0071e-01, 2.5257e+00, 9.2545e-01, ..., -5.3579e-01,
-3.4913e-02, 1.9827e-01],
...,
[-1.7490e+00, 1.0932e+00, 1.8678e-01, ..., -1.4654e+00,
-6.7425e-01, -1.1008e-01],
[-8.8467e-01, 1.6202e+00, 1.5057e+00, ..., -1.0796e-01,
1.1877e-01, 3.2828e-01],
[-2.1617e+00, -1.1101e+00, -8.3453e-01, ..., -2.3220e+00,
-1.4186e+00, -8.8622e-01]],
[[-2.4120e+00, -2.6987e+00, -4.0957e+00, ..., -1.1012e+00,
-7.0022e-01, -7.1427e-01],
[-1.6937e+00, -1.3788e+00, -4.8480e+00, ..., -1.3646e+00,
-1.2245e+00, -1.5825e+00],
[-3.0300e+00, -2.4687e+00, -6.9575e+00, ..., -3.3156e+00,
-2.8276e+00, -2.2491e+00],
...,
[-2.5772e+00, -1.8022e+00, -4.9112e+00, ..., -1.5138e+00,
-1.0924e+00, -9.0038e-01],
[-2.2634e+00, -1.6114e+00, -3.8670e+00, ..., -1.7455e+00,
-1.1647e+00, -7.5829e-01],
[-2.3335e+00, -1.8349e+00, -3.0891e+00, ..., -2.7829e+00,
-1.9898e+00, -1.8078e+00]],
[[-1.9422e+00, -2.6809e+00, -2.4644e+00, ..., -1.7956e+00,
-1.4928e+00, -1.1355e+00],
[-1.9716e+00, -2.7017e+00, -2.1554e+00, ..., -1.4045e+00,
-1.1144e+00, -8.8748e-01],
[-2.5079e+00, -3.0868e+00, -2.5830e+00, ..., -1.6080e+00,
-1.2108e+00, -8.9175e-01],
...,
[-3.1341e+00, -3.9303e+00, -3.7112e+00, ..., -2.8871e+00,
-2.3060e+00, -1.7727e+00],
[-2.0713e+00, -2.6892e+00, -2.6603e+00, ..., -1.9931e+00,
-1.6405e+00, -1.3202e+00],
[-1.4863e+00, -1.8174e+00, -1.8681e+00, ..., -1.2335e+00,
-1.0448e+00, -7.9825e-01]]],
[[[ 1.4344e-01, 3.7878e-01, -1.2059e+00, ..., 4.1931e-01,
9.6743e-01, 6.1530e-01],
[-2.1065e+00, -1.4296e+00, -4.4001e+00, ..., -2.5168e+00,
-7.2463e-01, -7.7716e-02],
[ 4.2693e-01, -3.0839e-01, -1.1561e+00, ..., -3.8098e-01,
4.2614e-01, 6.1493e-01],
...,
[-3.1186e-01, -1.2928e+00, -2.5559e+00, ..., -2.4858e+00,
-1.4674e+00, -1.0881e+00],
[ 1.1824e+00, 6.1718e-01, -9.0115e-01, ..., -5.4396e-01,
6.1143e-01, 6.5221e-01],
[ 7.6279e-01, 2.5942e-01, -4.2385e-01, ..., -4.3776e-01,
3.9921e-01, 2.2845e-01]],
[[-3.2398e+00, -1.4283e+00, -6.1634e+00, ..., -4.7183e+00,
-4.3166e+00, -3.2837e+00],
[-5.6678e-01, 2.1740e+00, -4.2224e+00, ..., -3.3320e+00,
-3.4826e+00, -2.8312e+00],
[ 3.6957e-02, 1.8610e+00, -7.4527e+00, ..., -5.1838e+00,
-5.1581e+00, -4.8296e+00],
...,
[-1.2404e+00, 2.8222e-01, -6.5786e+00, ..., -4.4964e+00,
-4.1777e+00, -3.2837e+00],
[ 1.5193e+00, 2.5867e+00, -1.4262e+00, ..., -5.4645e-01,
-9.7691e-01, -9.4666e-01],
[ 9.9441e-01, 1.6183e+00, -2.4736e+00, ..., -2.1545e+00,
-2.4262e+00, -2.3237e+00]],
[[-1.3923e+00, -2.3712e+00, -2.1819e+00, ..., -5.4559e-01,
-1.7167e-01, -4.2984e-01],
[-2.0324e+00, -2.2165e+00, -1.5561e+00, ..., -3.0154e-01,
7.9207e-02, 4.6016e-02],
[-2.5767e+00, -2.0435e+00, -8.3909e-01, ..., -2.4397e-01,
2.3930e-01, 2.7906e-01],
...,
[-1.5740e+00, -1.8232e+00, -7.6108e-01, ..., -5.4749e-01,
-1.4050e-01, -3.7108e-01],
[-1.7625e+00, -1.2046e+00, -7.7456e-01, ..., -4.0211e-01,
1.6356e-01, 2.7655e-01],
[-1.7489e+00, -1.1640e+00, -6.3278e-01, ..., -5.6573e-01,
-3.0941e-01, -2.1448e-01]],
...,
[[-2.3657e+00, -1.2777e+00, -2.7160e+00, ..., -3.3969e+00,
-2.6479e+00, -1.7330e+00],
[-1.9220e+00, 1.9547e-01, -1.3483e+00, ..., -2.5136e+00,
-2.2893e+00, -1.7568e+00],
[-4.9901e-01, 2.5316e+00, 9.3328e-01, ..., -6.7663e-01,
-1.3509e-01, 1.2731e-01],
...,
[-1.7404e+00, 1.1332e+00, 1.3692e-01, ..., -1.5164e+00,
-8.0400e-01, -3.7399e-02],
[-8.3142e-01, 1.6872e+00, 1.4398e+00, ..., 1.2195e-02,
8.2879e-02, 3.2655e-01],
[-2.0789e+00, -9.9076e-01, -8.1259e-01, ..., -2.4561e+00,
-1.6082e+00, -1.0808e+00]],
[[-2.3895e+00, -2.6568e+00, -4.0695e+00, ..., -1.1993e+00,
-7.6032e-01, -7.6032e-01],
[-1.6760e+00, -1.3462e+00, -4.8266e+00, ..., -1.3559e+00,
-1.1857e+00, -1.5493e+00],
[-3.0091e+00, -2.4286e+00, -6.9439e+00, ..., -3.2220e+00,
-2.7452e+00, -2.2153e+00],
...,
[-2.5123e+00, -1.7209e+00, -5.0074e+00, ..., -1.3298e+00,
-7.9690e-01, -7.8385e-01],
[-2.1816e+00, -1.5164e+00, -3.9086e+00, ..., -1.6038e+00,
-9.3560e-01, -5.7381e-01],
[-2.2542e+00, -1.7402e+00, -3.0813e+00, ..., -2.8075e+00,
-1.9397e+00, -1.7477e+00]],
[[-1.9121e+00, -2.6440e+00, -2.4221e+00, ..., -1.7699e+00,
-1.4828e+00, -1.1371e+00],
[-1.9368e+00, -2.6587e+00, -2.1117e+00, ..., -1.4190e+00,
-1.1274e+00, -9.0260e-01],
[-2.4723e+00, -3.0340e+00, -2.5361e+00, ..., -1.6656e+00,
-1.2447e+00, -9.2012e-01],
...,
[-3.1476e+00, -3.9216e+00, -3.6904e+00, ..., -3.0193e+00,
-2.4040e+00, -1.8769e+00],
[-2.0982e+00, -2.7132e+00, -2.6759e+00, ..., -2.0894e+00,
-1.7123e+00, -1.4145e+00],
[-1.5167e+00, -1.8397e+00, -1.8893e+00, ..., -1.2664e+00,
-1.0951e+00, -8.5061e-01]]]], device='cuda:0'),
tensor([[[[ 1.1733e-01, 3.5309e-01, -1.1640e+00, ..., 6.4320e-01,
9.3191e-01, 6.8202e-01],
[-1.9620e+00, -1.3368e+00, -4.1441e+00, ..., -1.5474e+00,
-1.9896e-01, 3.7868e-01],
[ 3.5712e-01, -2.8919e-01, -1.0985e+00, ..., 4.9446e-01,
9.1518e-01, 8.3518e-01],
...,
[-2.9318e-01, -1.3990e+00, -2.8250e+00, ..., -1.1933e+00,
-8.2849e-01, -6.3564e-01],
[ 1.3804e+00, 7.1589e-01, -8.9751e-01, ..., 5.0471e-01,
7.3901e-01, 7.7363e-01],
[ 9.2866e-01, 3.8496e-01, -3.8866e-01, ..., 5.3643e-01,
7.6098e-01, 6.6568e-01]],
[[-3.1290e+00, -1.4918e+00, -6.0714e+00, ..., -5.4548e+00,
-4.6852e+00, -3.6374e+00],
[-7.1167e-01, 1.7750e+00, -4.3825e+00, ..., -4.1824e+00,
-4.0612e+00, -3.3335e+00],
[-2.2958e-01, 1.4382e+00, -7.4357e+00, ..., -6.1404e+00,
-5.8867e+00, -5.4447e+00],
...,
[-1.0754e+00, 6.0564e-01, -6.3074e+00, ..., -5.3454e+00,
-4.6907e+00, -3.9192e+00],
[ 1.9453e+00, 3.1413e+00, -7.3982e-01, ..., -1.2426e+00,
-1.3889e+00, -1.4775e+00],
[ 1.2836e+00, 1.9605e+00, -2.1333e+00, ..., -2.4896e+00,
-2.5971e+00, -2.5584e+00]],
[[-1.2860e+00, -2.1610e+00, -1.9993e+00, ..., -8.4498e-01,
-6.2546e-01, -8.0820e-01],
[-1.8344e+00, -2.0028e+00, -1.4386e+00, ..., -6.8440e-01,
-4.1794e-01, -3.5924e-01],
[-2.3794e+00, -1.9053e+00, -8.1804e-01, ..., -4.1655e-01,
-1.0663e-01, -9.4649e-02],
...,
[-1.7211e+00, -1.9873e+00, -8.3343e-01, ..., -6.1109e-01,
-5.4054e-01, -7.0202e-01],
[-1.9284e+00, -1.2976e+00, -8.1636e-01, ..., -2.2408e-01,
-5.3838e-02, -4.8035e-02],
[-1.8867e+00, -1.1894e+00, -5.9860e-01, ..., -3.2667e-01,
-2.9271e-01, -2.6870e-01]],
...,
[[-2.1206e+00, -1.1876e+00, -2.5307e+00, ..., -3.0924e+00,
-2.4713e+00, -1.6531e+00],
[-1.8254e+00, 7.4906e-02, -1.3112e+00, ..., -1.5331e+00,
-1.5897e+00, -1.3641e+00],
[-4.7236e-01, 2.2667e+00, 8.2562e-01, ..., -8.8889e-02,
2.2279e-01, 3.1969e-01],
...,
[-1.9139e+00, 1.3286e+00, 2.5006e-01, ..., -1.0278e+00,
-2.9185e-01, 1.8668e-01],
[-9.1244e-01, 1.8489e+00, 1.6176e+00, ..., 1.2604e-01,
2.9840e-01, 4.0562e-01],
[-2.2311e+00, -1.1583e+00, -1.0334e+00, ..., -2.1852e+00,
-1.2786e+00, -6.4679e-01]],
[[-2.3675e+00, -2.6722e+00, -3.9428e+00, ..., -1.1136e+00,
-5.8968e-01, -4.8955e-01],
[-1.7676e+00, -1.5356e+00, -4.6736e+00, ..., -1.6687e+00,
-1.4611e+00, -1.7151e+00],
[-3.0020e+00, -2.5524e+00, -6.6515e+00, ..., -3.7139e+00,
-3.2517e+00, -2.6177e+00],
...,
[-2.4553e+00, -1.4855e+00, -5.0832e+00, ..., -2.1452e+00,
-1.7198e+00, -1.3923e+00],
[-2.1993e+00, -1.3878e+00, -3.9657e+00, ..., -2.0908e+00,
-1.4656e+00, -9.6645e-01],
[-2.3809e+00, -1.7298e+00, -3.1561e+00, ..., -2.8079e+00,
-1.9414e+00, -1.6723e+00]],
[[-1.8364e+00, -2.5281e+00, -2.3520e+00, ..., -1.7017e+00,
-1.4091e+00, -1.0562e+00],
[-1.8792e+00, -2.5700e+00, -2.1064e+00, ..., -1.3657e+00,
-1.1111e+00, -8.7816e-01],
[-2.4075e+00, -2.9665e+00, -2.5470e+00, ..., -1.5680e+00,
-1.1944e+00, -8.6858e-01],
...,
[-3.2052e+00, -4.0010e+00, -3.7021e+00, ..., -2.7932e+00,
-2.2353e+00, -1.6879e+00],
[-2.1161e+00, -2.7349e+00, -2.6476e+00, ..., -1.9601e+00,
-1.6149e+00, -1.2871e+00],
[-1.5036e+00, -1.8277e+00, -1.8445e+00, ..., -1.2222e+00,
-1.0420e+00, -7.6982e-01]]],
[[[ 8.8602e-02, 2.8311e-01, -1.5358e+00, ..., 5.2272e-01,
1.0174e+00, 6.8224e-01],
[-2.2625e+00, -1.6718e+00, -4.9310e+00, ..., -2.1745e+00,
-5.2990e-01, 8.8487e-02],
[ 3.7696e-01, -4.1439e-01, -1.4064e+00, ..., -7.0645e-01,
2.2905e-01, 5.0009e-01],
...,
[-2.5312e-01, -1.2928e+00, -2.6782e+00, ..., -2.0215e+00,
-1.4373e+00, -1.0106e+00],
[ 1.2721e+00, 6.2060e-01, -1.1178e+00, ..., -1.9266e-01,
5.5796e-01, 7.3174e-01],
[ 7.8373e-01, 1.9924e-01, -5.9754e-01, ..., -1.6414e-01,
4.0779e-01, 3.5882e-01]],
[[-3.3868e+00, -1.5180e+00, -6.3969e+00, ..., -4.8027e+00,
-4.3578e+00, -3.3450e+00],
[-6.3963e-01, 2.1790e+00, -4.3784e+00, ..., -3.5337e+00,
-3.6095e+00, -2.9593e+00],
[-6.3362e-02, 1.8709e+00, -7.6798e+00, ..., -5.4328e+00,
-5.4016e+00, -5.2166e+00],
...,
[-1.3895e+00, 1.7224e-01, -6.8558e+00, ..., -4.6471e+00,
-4.1469e+00, -3.4484e+00],
[ 1.4404e+00, 2.5853e+00, -1.5254e+00, ..., -5.7883e-01,
-9.3505e-01, -1.0534e+00],
[ 9.7145e-01, 1.6741e+00, -2.4715e+00, ..., -2.1560e+00,
-2.4278e+00, -2.3679e+00]],
[[-1.5353e+00, -2.4921e+00, -2.2955e+00, ..., -5.8300e-01,
-2.2201e-01, -4.6302e-01],
[-2.1547e+00, -2.3145e+00, -1.6435e+00, ..., -3.6750e-01,
5.8081e-03, 1.7630e-04],
[-2.7733e+00, -2.2036e+00, -9.1821e-01, ..., -4.0083e-01,
-2.8429e-02, 2.1051e-02],
...,
[-1.7108e+00, -1.8842e+00, -7.5951e-01, ..., -5.3707e-01,
-2.1764e-01, -4.1326e-01],
[-1.8667e+00, -1.2333e+00, -7.4377e-01, ..., -2.9095e-01,
1.7370e-01, 2.4342e-01],
[-1.8454e+00, -1.2448e+00, -6.9944e-01, ..., -4.5221e-01,
-2.3904e-01, -2.0961e-01]],
...,
[[-2.3974e+00, -1.3207e+00, -2.7668e+00, ..., -3.4531e+00,
-2.6640e+00, -1.7354e+00],
[-2.1045e+00, 1.8258e-02, -1.4659e+00, ..., -2.4556e+00,
-2.2178e+00, -1.7102e+00],
[-5.7165e-01, 2.4907e+00, 9.6597e-01, ..., -4.6103e-01,
-3.2294e-01, -1.7555e-01],
...,
[-1.8542e+00, 1.0544e+00, 1.5480e-01, ..., -1.5455e+00,
-6.2567e-01, -1.8956e-02],
[-8.9496e-01, 1.6767e+00, 1.5908e+00, ..., -1.0998e-01,
1.7633e-01, 3.9459e-01],
[-2.1380e+00, -1.0663e+00, -7.7083e-01, ..., -2.5725e+00,
-1.5510e+00, -9.3499e-01]],
[[-2.5035e+00, -2.8208e+00, -4.1875e+00, ..., -1.1616e+00,
-7.4588e-01, -7.2557e-01],
[-1.7943e+00, -1.5195e+00, -4.9271e+00, ..., -1.4670e+00,
-1.2723e+00, -1.5571e+00],
[-3.1151e+00, -2.5833e+00, -7.0042e+00, ..., -3.2031e+00,
-2.6057e+00, -2.0004e+00],
...,
[-2.6447e+00, -1.9032e+00, -5.0496e+00, ..., -1.4373e+00,
-9.9948e-01, -8.2187e-01],
[-2.3033e+00, -1.6986e+00, -3.9602e+00, ..., -1.7152e+00,
-1.0950e+00, -6.5206e-01],
[-2.3523e+00, -1.8938e+00, -3.1380e+00, ..., -2.8469e+00,
-1.9788e+00, -1.7846e+00]],
[[-2.0010e+00, -2.7592e+00, -2.5469e+00, ..., -1.7440e+00,
-1.4494e+00, -1.1068e+00],
[-2.0304e+00, -2.7837e+00, -2.2409e+00, ..., -1.3937e+00,
-1.1097e+00, -8.8563e-01],
[-2.5736e+00, -3.1816e+00, -2.6858e+00, ..., -1.6980e+00,
-1.3082e+00, -9.5772e-01],
...,
[-3.2267e+00, -4.0403e+00, -3.8096e+00, ..., -2.9501e+00,
-2.3624e+00, -1.8135e+00],
[-2.1380e+00, -2.7655e+00, -2.7277e+00, ..., -2.0141e+00,
-1.6680e+00, -1.3499e+00],
[-1.5472e+00, -1.8793e+00, -1.9140e+00, ..., -1.2220e+00,
-1.0505e+00, -8.0378e-01]]],
[[[ 3.1224e-01, 5.7352e-01, 6.8725e-01, ..., 4.9137e-01,
1.0201e+00, 6.1454e-01],
[-2.1181e+00, -1.4592e+00, -3.1372e+00, ..., -2.5930e+00,
-7.9852e-01, -1.4436e-01],
[-1.0668e-01, -9.0901e-01, -2.1698e-01, ..., -1.1981e-01,
5.9367e-01, 6.6460e-01],
...,
[ 9.8508e-02, -6.6587e-01, -1.5843e+00, ..., -1.0841e+00,
-8.3464e-01, -6.0019e-01],
[ 1.5510e+00, 1.2747e+00, 6.0548e-01, ..., 7.5253e-01,
9.1274e-01, 9.4181e-01],
[ 1.1689e+00, 1.0344e+00, 6.9588e-01, ..., 9.0936e-01,
1.0059e+00, 8.6753e-01]],
[[-1.7232e+00, -1.1752e+00, -4.1794e+00, ..., -4.8851e+00,
-4.4547e+00, -3.3931e+00],
[ 5.4689e-01, 1.9157e+00, -2.2617e+00, ..., -3.5377e+00,
-3.6574e+00, -2.9641e+00],
[ 1.2876e+00, 1.6890e+00, -4.9115e+00, ..., -5.4592e+00,
-5.3253e+00, -4.9104e+00],
...,
[-6.9042e-01, 5.2114e-01, -4.8244e+00, ..., -5.9352e+00,
-5.2826e+00, -4.2922e+00],
[ 2.1515e+00, 2.8807e+00, 1.3269e-01, ..., -1.5484e+00,
-1.7563e+00, -1.7233e+00],
[ 1.2762e+00, 1.5811e+00, -1.6797e+00, ..., -2.7367e+00,
-2.8878e+00, -2.7622e+00]],
[[-1.5062e-01, -1.1730e+00, -1.3070e+00, ..., -4.9159e-01,
-1.6620e-01, -4.8834e-01],
[-6.3856e-01, -9.0724e-01, -7.9377e-01, ..., -2.8826e-01,
3.1665e-02, -6.6695e-02],
[-5.6934e-01, -3.8744e-01, -4.1012e-01, ..., -3.4380e-01,
9.6672e-02, 1.0338e-01],
...,
[-1.0779e+00, -1.5886e+00, -6.8297e-01, ..., -6.4751e-01,
-5.0163e-01, -6.5453e-01],
[-1.3373e+00, -8.9670e-01, -4.9355e-01, ..., -1.2717e-01,
6.4656e-02, 4.4572e-02],
[-1.3852e+00, -6.9703e-01, -1.5784e-01, ..., -6.5322e-02,
-4.5250e-02, -7.4551e-02]],
...,
[[-3.0138e+00, -2.4568e+00, -2.8003e+00, ..., -3.4318e+00,
-2.6477e+00, -1.7542e+00],
[-2.2035e+00, -7.5417e-01, -1.5669e+00, ..., -2.5625e+00,
-2.3460e+00, -1.8372e+00],
[-1.0363e+00, 1.2332e+00, 6.1759e-01, ..., -4.3068e-01,
-7.5287e-02, 9.5557e-02],
...,
[-1.6342e+00, 1.5867e+00, 3.3036e-01, ..., -7.6672e-01,
-1.3131e-01, 3.8301e-01],
[-6.4650e-01, 1.7711e+00, 1.1571e+00, ..., 2.0057e-01,
3.6000e-01, 5.2569e-01],
[-1.7655e+00, -1.0473e+00, -1.4059e+00, ..., -2.0628e+00,
-1.1449e+00, -5.1327e-01]],
[[-1.5845e+00, -1.5576e+00, -2.8653e+00, ..., -1.2612e+00,
-8.2002e-01, -8.2922e-01],
[-6.0267e-01, 7.1234e-02, -3.2025e+00, ..., -1.3730e+00,
-1.2049e+00, -1.5801e+00],
[-2.1122e+00, -9.4310e-01, -5.3495e+00, ..., -3.1912e+00,
-2.7003e+00, -2.2189e+00],
...,
[-1.7996e+00, -7.6895e-01, -4.7359e+00, ..., -2.4859e+00,
-2.0932e+00, -1.6515e+00],
[-1.6875e+00, -8.2475e-01, -3.7864e+00, ..., -2.2767e+00,
-1.7143e+00, -1.1296e+00],
[-2.0747e+00, -1.3408e+00, -3.0258e+00, ..., -2.7477e+00,
-1.9860e+00, -1.6988e+00]],
[[-1.3437e+00, -1.9023e+00, -1.7058e+00, ..., -1.7650e+00,
-1.4585e+00, -1.1177e+00],
[-1.2636e+00, -1.7546e+00, -1.2332e+00, ..., -1.3809e+00,
-1.0863e+00, -8.7521e-01],
[-1.5829e+00, -1.9352e+00, -1.4812e+00, ..., -1.6063e+00,
-1.2048e+00, -9.0290e-01],
...,
[-2.5125e+00, -3.1395e+00, -2.8765e+00, ..., -3.0376e+00,
-2.4155e+00, -1.8428e+00],
[-1.6914e+00, -2.1487e+00, -2.0405e+00, ..., -2.1209e+00,
-1.7336e+00, -1.3858e+00],
[-1.1327e+00, -1.3601e+00, -1.3924e+00, ..., -1.3156e+00,
-1.1098e+00, -8.2403e-01]]],
[[[ 2.6943e-01, 5.3669e-01, 6.6306e-01, ..., 4.8912e-01,
1.0297e+00, 6.0898e-01],
[-2.1648e+00, -1.5062e+00, -3.1837e+00, ..., -2.8417e+00,
-9.5883e-01, -2.2588e-01],
[-2.0137e-01, -9.6586e-01, -2.1433e-01, ..., -2.0873e-01,
5.0727e-01, 6.2418e-01],
...,
[ 8.7333e-02, -6.8361e-01, -1.5572e+00, ..., -1.0309e+00,
-7.8502e-01, -5.8611e-01],
[ 1.5248e+00, 1.2461e+00, 6.0846e-01, ..., 7.8101e-01,
9.2066e-01, 9.3781e-01],
[ 1.1596e+00, 1.0101e+00, 6.9273e-01, ..., 9.2141e-01,
1.0165e+00, 8.6914e-01]],
[[-1.6858e+00, -1.2572e+00, -4.2080e+00, ..., -5.0605e+00,
-4.5640e+00, -3.4725e+00],
[ 5.0568e-01, 1.7232e+00, -2.3644e+00, ..., -3.4991e+00,
-3.6164e+00, -2.9669e+00],
[ 1.1934e+00, 1.4503e+00, -4.9637e+00, ..., -5.5146e+00,
-5.3425e+00, -5.0378e+00],
...,
[-6.3676e-01, 5.4560e-01, -4.6929e+00, ..., -5.8722e+00,
-5.2051e+00, -4.2420e+00],
[ 2.1814e+00, 2.8932e+00, 2.1860e-01, ..., -1.5217e+00,
-1.7018e+00, -1.6851e+00],
[ 1.2985e+00, 1.5805e+00, -1.6338e+00, ..., -2.7062e+00,
-2.8444e+00, -2.7355e+00]],
[[-1.0153e-01, -1.0844e+00, -1.2643e+00, ..., -5.5113e-01,
-1.9335e-01, -5.3573e-01],
[-5.2289e-01, -7.6371e-01, -7.2856e-01, ..., -3.3259e-01,
1.8907e-02, -9.0962e-02],
[-4.5514e-01, -2.5942e-01, -3.6606e-01, ..., -3.6508e-01,
1.0061e-01, 8.4869e-02],
...,
[-1.1047e+00, -1.5886e+00, -6.5153e-01, ..., -6.2094e-01,
-4.7754e-01, -6.2862e-01],
[-1.3554e+00, -8.9185e-01, -4.6551e-01, ..., -9.4213e-02,
9.3234e-02, 7.3647e-02],
[-1.4046e+00, -7.0492e-01, -1.4137e-01, ..., -4.4499e-02,
-2.8688e-02, -5.5809e-02]],
...,
[[-2.9663e+00, -2.5463e+00, -2.7716e+00, ..., -3.6983e+00,
-2.8171e+00, -1.8699e+00],
[-2.2675e+00, -9.5975e-01, -1.6170e+00, ..., -2.7306e+00,
-2.4584e+00, -1.9313e+00],
[-1.1073e+00, 9.3621e-01, 5.1412e-01, ..., -4.3547e-01,
3.5744e-02, 2.0096e-01],
...,
[-1.5929e+00, 1.6564e+00, 3.4193e-01, ..., -7.8514e-01,
-1.3425e-01, 3.7686e-01],
[-5.9579e-01, 1.7999e+00, 1.1207e+00, ..., 1.6326e-01,
3.3129e-01, 4.8637e-01],
[-1.7173e+00, -9.9213e-01, -1.4185e+00, ..., -2.0900e+00,
-1.1621e+00, -5.4216e-01]],
[[-1.6174e+00, -1.6173e+00, -2.7969e+00, ..., -1.1588e+00,
-7.2718e-01, -7.5728e-01],
[-6.5307e-01, -1.2735e-02, -3.1048e+00, ..., -1.2469e+00,
-1.1377e+00, -1.5691e+00],
[-2.1302e+00, -1.0079e+00, -5.1772e+00, ..., -3.1956e+00,
-2.7323e+00, -2.2358e+00],
...,
[-1.7558e+00, -6.3571e-01, -4.6729e+00, ..., -2.4719e+00,
-2.0350e+00, -1.6030e+00],
[-1.6720e+00, -7.4886e-01, -3.7437e+00, ..., -2.2825e+00,
-1.6871e+00, -1.0989e+00],
[-2.0748e+00, -1.3031e+00, -3.0100e+00, ..., -2.7514e+00,
-1.9779e+00, -1.6861e+00]],
[[-1.3298e+00, -1.8896e+00, -1.7152e+00, ..., -1.8296e+00,
-1.5123e+00, -1.1430e+00],
[-1.2560e+00, -1.7418e+00, -1.2620e+00, ..., -1.3834e+00,
-1.0874e+00, -8.6788e-01],
[-1.5661e+00, -1.9211e+00, -1.5168e+00, ..., -1.5878e+00,
-1.1800e+00, -8.6639e-01],
...,
[-2.4925e+00, -3.1086e+00, -2.8378e+00, ..., -2.9936e+00,
-2.3842e+00, -1.8171e+00],
[-1.6743e+00, -2.1281e+00, -2.0181e+00, ..., -2.0968e+00,
-1.7111e+00, -1.3667e+00],
[-1.1158e+00, -1.3418e+00, -1.3687e+00, ..., -1.2922e+00,
-1.0955e+00, -8.0779e-01]]],
[[[ 3.5655e-01, 8.2999e-01, 7.5616e-01, ..., 6.7323e-01,
9.3660e-01, 7.1782e-01],
[-1.2370e+00, -2.8211e-01, -1.3627e+00, ..., -1.0978e+00,
5.7992e-02, 5.4033e-01],
[ 6.4850e-01, 4.4735e-01, 5.4243e-01, ..., 6.0675e-01,
9.7254e-01, 8.8150e-01],
...,
[-1.8127e-01, -1.1183e+00, -2.5982e+00, ..., -1.5402e+00,
-1.4509e+00, -8.8583e-01],
[ 1.4519e+00, 8.7354e-01, -8.5228e-01, ..., -2.7213e-01,
8.2935e-02, 3.5808e-01],
[ 8.5544e-01, 2.8472e-01, -6.3360e-01, ..., -1.5328e-01,
2.1685e-01, 3.7470e-01]],
[[-2.3223e+00, -1.2201e+00, -4.2383e+00, ..., -5.2292e+00,
-4.5051e+00, -3.5118e+00],
[-1.2328e-01, 1.7341e+00, -2.5235e+00, ..., -4.0747e+00,
-3.9781e+00, -3.2569e+00],
[ 7.5280e-01, 1.5771e+00, -5.0854e+00, ..., -5.9511e+00,
-5.7163e+00, -5.2694e+00],
...,
[-1.2844e+00, 1.4139e-01, -6.7297e+00, ..., -7.5912e+00,
-6.8737e+00, -5.3941e+00],
[ 1.7591e+00, 2.7643e+00, -1.1072e+00, ..., -2.5739e+00,
-2.9123e+00, -2.3431e+00],
[ 1.1959e+00, 1.7660e+00, -2.3220e+00, ..., -3.2482e+00,
-3.5929e+00, -3.1077e+00]],
[[-6.6092e-01, -1.8186e+00, -1.7126e+00, ..., -9.3968e-01,
-6.6104e-01, -7.5998e-01],
[-1.3759e+00, -1.8469e+00, -1.2724e+00, ..., -7.2303e-01,
-3.8955e-01, -2.7977e-01],
[-1.0727e+00, -9.8782e-01, -4.5096e-01, ..., -3.0531e-01,
4.7388e-02, 5.3543e-02],
...,
[-1.3746e+00, -1.7435e+00, -7.7260e-01, ..., -8.8583e-01,
-7.6121e-01, -1.0273e+00],
[-1.7228e+00, -1.1511e+00, -8.2930e-01, ..., -5.5868e-01,
-2.8959e-01, -3.6673e-01],
[-1.8769e+00, -1.2637e+00, -8.5837e-01, ..., -9.3941e-01,
-8.0938e-01, -7.9893e-01]],
...,
[[-2.2667e+00, -1.2809e+00, -2.3115e+00, ..., -2.8443e+00,
-2.2623e+00, -1.4991e+00],
[-6.3903e-01, 1.1068e+00, -3.9840e-01, ..., -1.2955e+00,
-1.3696e+00, -1.1831e+00],
[-3.7007e-01, 2.1352e+00, 8.3204e-01, ..., -1.9019e-01,
2.0797e-01, 3.4051e-01],
...,
[-1.7989e+00, 1.2883e+00, 3.6376e-03, ..., -1.0935e+00,
-5.1630e-01, 1.4146e-02],
[-8.2321e-01, 1.9034e+00, 1.5998e+00, ..., 4.6538e-01,
4.2755e-01, 6.8697e-01],
[-2.2101e+00, -1.1271e+00, -1.0242e+00, ..., -1.8894e+00,
-9.9459e-01, -3.0934e-01]],
[[-1.3483e+00, -1.2377e+00, -2.7537e+00, ..., -1.1063e+00,
-5.8119e-01, -4.5338e-01],
[-6.2760e-01, -1.0128e-01, -3.5912e+00, ..., -1.7885e+00,
-1.5028e+00, -1.6822e+00],
[-2.0567e+00, -1.2955e+00, -5.9522e+00, ..., -3.7270e+00,
-3.2002e+00, -2.5426e+00],
...,
[-2.4963e+00, -1.6365e+00, -5.5717e+00, ..., -3.0428e+00,
-2.7046e+00, -2.1563e+00],
[-2.1987e+00, -1.4661e+00, -4.3082e+00, ..., -2.6578e+00,
-2.2870e+00, -1.6186e+00],
[-2.3396e+00, -1.7351e+00, -3.3305e+00, ..., -2.9646e+00,
-2.4222e+00, -2.1110e+00]],
[[-1.1668e+00, -1.7226e+00, -1.4975e+00, ..., -1.6352e+00,
-1.3649e+00, -1.0251e+00],
[-1.1782e+00, -1.6459e+00, -1.1663e+00, ..., -1.3701e+00,
-1.1157e+00, -8.7978e-01],
[-1.5326e+00, -1.8165e+00, -1.3387e+00, ..., -1.5565e+00,
-1.1893e+00, -8.6396e-01],
...,
[-3.1796e+00, -3.9685e+00, -3.7509e+00, ..., -3.9436e+00,
-3.1637e+00, -2.4343e+00],
[-2.1337e+00, -2.7522e+00, -2.7144e+00, ..., -2.7596e+00,
-2.2822e+00, -1.8534e+00],
[-1.5656e+00, -1.8953e+00, -1.9513e+00, ..., -1.8594e+00,
-1.5128e+00, -1.1589e+00]]]], device='cuda:0'),
tensor([[[[ 2.7120e-01, 5.4375e-01, 6.6826e-01, ..., 4.9107e-01,
1.0319e+00, 6.1929e-01],
[-2.1874e+00, -1.5338e+00, -3.2131e+00, ..., -2.6909e+00,
-8.6569e-01, -1.7910e-01],
[-2.0517e-01, -1.0045e+00, -2.6013e-01, ..., -1.4747e-01,
6.0072e-01, 6.9776e-01],
...,
[ 9.2916e-02, -5.6786e-01, -1.3466e+00, ..., -1.2680e+00,
-9.5054e-01, -7.3359e-01],
[ 1.4145e+00, 1.1744e+00, 6.1991e-01, ..., 6.9967e-01,
9.2282e-01, 9.3160e-01],
[ 1.0680e+00, 9.4369e-01, 6.6959e-01, ..., 8.9698e-01,
1.0448e+00, 8.6686e-01]],
[[-1.6706e+00, -1.1795e+00, -4.1353e+00, ..., -4.8663e+00,
-4.4523e+00, -3.3724e+00],
[ 5.5373e-01, 1.8808e+00, -2.2384e+00, ..., -3.5060e+00,
-3.6355e+00, -2.9479e+00],
[ 1.2762e+00, 1.6600e+00, -4.8598e+00, ..., -5.4564e+00,
-5.2888e+00, -4.9174e+00],
...,
[-9.2096e-01, 9.1263e-02, -4.9861e+00, ..., -5.8089e+00,
-5.1891e+00, -4.1627e+00],
[ 1.7101e+00, 2.3517e+00, -3.0235e-01, ..., -1.2274e+00,
-1.4693e+00, -1.4721e+00],
[ 9.7081e-01, 1.2423e+00, -1.8202e+00, ..., -2.6004e+00,
-2.8104e+00, -2.7038e+00]],
[[-8.4112e-02, -1.0892e+00, -1.2796e+00, ..., -4.7531e-01,
-1.2691e-01, -4.6586e-01],
[-5.4855e-01, -8.1645e-01, -7.6706e-01, ..., -2.6294e-01,
5.5657e-02, -6.3564e-02],
[-5.2035e-01, -3.3173e-01, -4.0636e-01, ..., -3.3829e-01,
1.0048e-01, 8.3752e-02],
...,
[-8.9986e-01, -1.3802e+00, -5.9684e-01, ..., -6.5198e-01,
-4.7810e-01, -6.5769e-01],
[-1.1231e+00, -7.4902e-01, -3.9892e-01, ..., -9.4939e-02,
1.2524e-01, 1.0171e-01],
[-1.1696e+00, -5.9398e-01, -1.3134e-01, ..., -5.7764e-02,
-1.9155e-02, -5.2179e-02]],
...,
[[-3.0032e+00, -2.5109e+00, -2.7890e+00, ..., -3.5142e+00,
-2.6796e+00, -1.7786e+00],
[-2.3002e+00, -8.8295e-01, -1.5959e+00, ..., -2.6814e+00,
-2.3911e+00, -1.8642e+00],
[-1.0835e+00, 1.1183e+00, 6.1268e-01, ..., -5.2970e-01,
-7.5715e-02, 1.2924e-01],
...,
[-1.5403e+00, 1.2948e+00, 1.6052e-01, ..., -8.5602e-01,
-1.4934e-01, 4.2653e-01],
[-6.5312e-01, 1.4612e+00, 8.9819e-01, ..., 1.9365e-01,
3.5420e-01, 5.5448e-01],
[-1.6712e+00, -1.0767e+00, -1.4432e+00, ..., -2.2649e+00,
-1.2779e+00, -5.8373e-01]],
[[-1.6026e+00, -1.5887e+00, -2.8967e+00, ..., -1.2777e+00,
-8.3983e-01, -8.5356e-01],
[-6.1461e-01, 6.5075e-02, -3.2051e+00, ..., -1.3790e+00,
-1.2248e+00, -1.6029e+00],
[-2.0971e+00, -8.9744e-01, -5.2972e+00, ..., -3.2014e+00,
-2.7127e+00, -2.2305e+00],
...,
[-1.8349e+00, -9.8286e-01, -4.5074e+00, ..., -2.3428e+00,
-1.9157e+00, -1.5078e+00],
[-1.7265e+00, -1.0180e+00, -3.6320e+00, ..., -2.2240e+00,
-1.5927e+00, -1.0050e+00],
[-2.0468e+00, -1.4360e+00, -2.9638e+00, ..., -2.8151e+00,
-1.9707e+00, -1.6969e+00]],
[[-1.3498e+00, -1.9084e+00, -1.7145e+00, ..., -1.7819e+00,
-1.4772e+00, -1.1257e+00],
[-1.2774e+00, -1.7658e+00, -1.2470e+00, ..., -1.3914e+00,
-1.1002e+00, -8.7827e-01],
[-1.5986e+00, -1.9563e+00, -1.4966e+00, ..., -1.6270e+00,
-1.2192e+00, -9.0108e-01],
...,
[-2.3603e+00, -2.9618e+00, -2.7801e+00, ..., -3.0730e+00,
-2.4511e+00, -1.8751e+00],
[-1.5878e+00, -2.0216e+00, -1.9723e+00, ..., -2.1343e+00,
-1.7552e+00, -1.4106e+00],
[-1.0654e+00, -1.2871e+00, -1.3414e+00, ..., -1.3020e+00,
-1.1118e+00, -8.2538e-01]]],
[[[ 2.8715e-01, 5.5746e-01, 6.9286e-01, ..., 4.5692e-01,
1.0091e+00, 5.9761e-01],
[-2.1858e+00, -1.5408e+00, -3.2343e+00, ..., -2.7189e+00,
-8.8420e-01, -2.1042e-01],
[-1.7344e-01, -1.0086e+00, -2.4647e-01, ..., -1.6091e-01,
5.3478e-01, 6.3797e-01],
...,
[-1.8175e-01, -1.1183e+00, -2.6323e+00, ..., -1.5635e+00,
-1.4117e+00, -8.5396e-01],
[ 1.4343e+00, 8.7802e-01, -9.2434e-01, ..., -3.3124e-01,
1.0907e-01, 3.9165e-01],
[ 8.5311e-01, 3.2087e-01, -6.6744e-01, ..., -1.8715e-01,
2.3110e-01, 3.8938e-01]],
[[-1.6758e+00, -1.1611e+00, -4.1727e+00, ..., -4.8937e+00,
-4.4649e+00, -3.4028e+00],
[ 6.1247e-01, 1.9482e+00, -2.2571e+00, ..., -3.5208e+00,
-3.6171e+00, -2.9690e+00],
[ 1.3626e+00, 1.7084e+00, -4.9059e+00, ..., -5.4814e+00,
-5.2833e+00, -4.9386e+00],
...,
[-1.3571e+00, 4.4427e-02, -6.7333e+00, ..., -7.5701e+00,
-6.9015e+00, -5.4228e+00],
[ 1.7210e+00, 2.7385e+00, -1.0559e+00, ..., -2.5555e+00,
-2.9532e+00, -2.3886e+00],
[ 1.1676e+00, 1.7674e+00, -2.2616e+00, ..., -3.2496e+00,
-3.6367e+00, -3.1386e+00]],
[[-1.1985e-01, -1.1462e+00, -1.3264e+00, ..., -5.0704e-01,
-1.6553e-01, -4.8435e-01],
[-6.0067e-01, -8.5771e-01, -7.9100e-01, ..., -2.8821e-01,
3.5894e-02, -7.0902e-02],
[-5.5004e-01, -3.5793e-01, -4.2247e-01, ..., -3.6167e-01,
8.3187e-02, 8.3922e-02],
...,
[-1.3214e+00, -1.7292e+00, -7.8073e-01, ..., -8.8016e-01,
-7.5859e-01, -1.0259e+00],
[-1.6797e+00, -1.1481e+00, -8.5110e-01, ..., -5.7632e-01,
-2.9197e-01, -3.5790e-01],
[-1.8447e+00, -1.2628e+00, -8.9651e-01, ..., -9.6458e-01,
-8.2466e-01, -7.9951e-01]],
...,
[[-3.0515e+00, -2.5327e+00, -2.8271e+00, ..., -3.4937e+00,
-2.6631e+00, -1.7704e+00],
[-2.3037e+00, -8.6403e-01, -1.6078e+00, ..., -2.6425e+00,
-2.3602e+00, -1.8532e+00],
[-1.0769e+00, 1.1675e+00, 6.4880e-01, ..., -4.5149e-01,
-8.5557e-03, 1.3003e-01],
...,
[-1.8777e+00, 1.1456e+00, -3.6021e-02, ..., -1.0768e+00,
-5.5359e-01, -6.6504e-03],
[-8.4790e-01, 1.8244e+00, 1.5866e+00, ..., 4.9290e-01,
4.1499e-01, 6.7791e-01],
[-2.2305e+00, -1.2012e+00, -1.0370e+00, ..., -1.8635e+00,
-9.9094e-01, -3.0259e-01]],
[[-1.6025e+00, -1.5866e+00, -2.9060e+00, ..., -1.2624e+00,
-8.2243e-01, -8.3204e-01],
[-5.9845e-01, 8.5618e-02, -3.2119e+00, ..., -1.3672e+00,
-1.2036e+00, -1.5677e+00],
[-2.1006e+00, -9.1425e-01, -5.3322e+00, ..., -3.1973e+00,
-2.7056e+00, -2.2003e+00],
...,
[-2.4974e+00, -1.6433e+00, -5.4470e+00, ..., -3.0215e+00,
-2.7040e+00, -2.1882e+00],
[-2.2263e+00, -1.5129e+00, -4.2459e+00, ..., -2.6404e+00,
-2.3033e+00, -1.6451e+00],
[-2.3865e+00, -1.8018e+00, -3.3167e+00, ..., -2.9529e+00,
-2.4258e+00, -2.1052e+00]],
[[-1.3637e+00, -1.9196e+00, -1.7164e+00, ..., -1.7769e+00,
-1.4662e+00, -1.1171e+00],
[-1.2841e+00, -1.7733e+00, -1.2403e+00, ..., -1.3742e+00,
-1.0814e+00, -8.6033e-01],
[-1.6085e+00, -1.9580e+00, -1.4931e+00, ..., -1.5949e+00,
-1.1950e+00, -8.7866e-01],
...,
[-3.1434e+00, -3.9339e+00, -3.7454e+00, ..., -3.9683e+00,
-3.1733e+00, -2.4475e+00],
[-2.0931e+00, -2.7055e+00, -2.6833e+00, ..., -2.7790e+00,
-2.2927e+00, -1.8654e+00],
[-1.5267e+00, -1.8504e+00, -1.9161e+00, ..., -1.8739e+00,
-1.5215e+00, -1.1666e+00]]],
[[[ 6.9635e-02, 2.2573e-01, -1.4009e+00, ..., -6.3519e-01,
-2.0444e-01, -5.3996e-02],
[-2.0772e+00, -1.6024e+00, -4.5212e+00, ..., -3.7641e+00,
-1.5298e+00, -5.5472e-01],
[ 5.6227e-01, -9.0430e-02, -8.2667e-01, ..., -2.1153e-01,
-2.5084e-01, 2.4927e-01],
...,
[-8.8438e-01, -1.8545e+00, -1.8619e+00, ..., -1.7837e+00,
-1.3608e+00, -9.6217e-01],
[ 7.2159e-01, -2.6808e-02, -1.1373e-01, ..., 2.1600e-02,
6.8970e-01, 7.9963e-01],
[ 1.1278e-01, -5.7864e-01, -4.8713e-01, ..., -1.2211e-01,
4.3745e-01, 3.5872e-01]],
[[-3.2939e+00, -1.7731e+00, -6.2777e+00, ..., -7.1511e+00,
-6.2623e+00, -4.6171e+00],
[-7.9894e-01, 1.5470e+00, -4.3776e+00, ..., -5.4947e+00,
-5.5572e+00, -4.2575e+00],
[-3.6503e-01, 1.0980e+00, -7.4314e+00, ..., -7.9830e+00,
-7.8525e+00, -6.6493e+00],
...,
[ 2.8071e-01, 3.7162e-01, -4.0002e+00, ..., -4.6712e+00,
-4.1659e+00, -3.5022e+00],
[ 2.2980e+00, 2.2889e+00, -1.2918e-01, ..., -8.1928e-01,
-1.1539e+00, -1.2368e+00],
[ 1.3330e+00, 1.1694e+00, -1.7533e+00, ..., -2.2448e+00,
-2.5050e+00, -2.4192e+00]],
[[-1.3323e+00, -2.3559e+00, -2.3247e+00, ..., -1.7675e+00,
-1.2452e+00, -1.3805e+00],
[-1.6777e+00, -1.9892e+00, -1.5310e+00, ..., -1.2369e+00,
-7.2637e-01, -8.0833e-01],
[-2.1742e+00, -1.9468e+00, -1.0108e+00, ..., -1.1003e+00,
-4.4892e-01, -1.6456e-01],
...,
[-1.0712e-01, -5.5847e-01, -4.3350e-01, ..., -4.9286e-01,
-1.9642e-01, -4.4774e-01],
[-6.6923e-01, -1.7142e-01, -4.2362e-01, ..., -1.9956e-01,
2.4031e-01, 2.3892e-01],
[-7.9435e-01, -1.8688e-01, -3.5813e-01, ..., -3.8902e-01,
-1.9899e-01, -2.3201e-01]],
...,
[[-2.0205e+00, -1.0828e+00, -2.3475e+00, ..., -3.1758e+00,
-2.6130e+00, -1.8188e+00],
[-1.7809e+00, 3.0132e-02, -1.3610e+00, ..., -2.4844e+00,
-2.5098e+00, -1.9545e+00],
[-3.9541e-01, 2.3634e+00, 1.0433e+00, ..., 1.1963e-01,
2.4449e-01, 5.1147e-01],
...,
[-2.1057e+00, -1.2876e-01, -2.0161e-01, ..., -1.3996e+00,
-4.0476e-01, 9.5731e-02],
[-1.2578e+00, 7.7052e-01, 9.5381e-01, ..., -1.5315e-02,
3.2377e-01, 4.6650e-01],
[-2.1341e+00, -1.4740e+00, -1.6310e+00, ..., -2.4725e+00,
-1.4568e+00, -9.1583e-01]],
[[-2.4306e+00, -2.8462e+00, -4.2534e+00, ..., -2.4111e+00,
-1.9084e+00, -1.7141e+00],
[-1.7796e+00, -1.6592e+00, -4.9357e+00, ..., -2.9291e+00,
-2.7901e+00, -2.7789e+00],
[-2.8277e+00, -2.4825e+00, -6.7310e+00, ..., -4.4646e+00,
-4.3741e+00, -3.4967e+00],
...,
[-1.3940e+00, -6.1391e-02, -3.3448e+00, ..., -1.4424e+00,
-1.0131e+00, -9.3087e-01],
[-1.6992e+00, -6.5187e-01, -3.0317e+00, ..., -1.7268e+00,
-1.1718e+00, -8.1906e-01],
[-2.3122e+00, -1.4706e+00, -3.0886e+00, ..., -2.9706e+00,
-2.1613e+00, -1.9886e+00]],
[[-1.8066e+00, -2.5390e+00, -2.4531e+00, ..., -2.7033e+00,
-2.2199e+00, -1.7030e+00],
[-1.8083e+00, -2.5201e+00, -2.1515e+00, ..., -2.4177e+00,
-1.9096e+00, -1.5480e+00],
[-2.2629e+00, -2.8281e+00, -2.5154e+00, ..., -2.7558e+00,
-2.1574e+00, -1.6399e+00],
...,
[-2.3880e+00, -2.9788e+00, -2.7566e+00, ..., -2.8544e+00,
-2.2799e+00, -1.7454e+00],
[-1.5581e+00, -2.0341e+00, -1.9674e+00, ..., -1.9350e+00,
-1.5957e+00, -1.2850e+00],
[-1.0222e+00, -1.3237e+00, -1.3627e+00, ..., -1.1761e+00,
-1.0095e+00, -7.6352e-01]]],
[[[ 4.2741e-01, 8.5297e-01, 5.3321e-01, ..., 4.3312e-01,
9.7968e-01, 6.3528e-01],
[-1.3337e+00, -4.3656e-01, -1.9658e+00, ..., -2.4983e+00,
-7.3919e-01, -6.6429e-02],
[ 8.9073e-01, 6.1738e-01, 3.6160e-01, ..., -2.5372e-01,
4.5823e-01, 6.3171e-01],
...,
[ 2.5683e-02, -5.9012e-01, -1.4137e+00, ..., -2.2042e+00,
-1.5702e+00, -1.0806e+00],
[ 1.3291e+00, 1.1364e+00, 5.3092e-01, ..., -3.4941e-01,
5.1533e-01, 7.0540e-01],
[ 9.8997e-01, 8.9011e-01, 5.7366e-01, ..., -3.5836e-01,
3.2059e-01, 2.8661e-01]],
[[-2.7955e+00, -1.5244e+00, -4.8080e+00, ..., -4.7302e+00,
-4.3467e+00, -3.3217e+00],
[-4.6469e-01, 1.7102e+00, -2.8632e+00, ..., -3.3692e+00,
-3.5136e+00, -2.8954e+00],
[ 3.6194e-01, 1.5050e+00, -5.6618e+00, ..., -5.2961e+00,
-5.1646e+00, -4.8527e+00],
...,
[-9.3163e-01, -7.7097e-03, -4.8337e+00, ..., -4.5135e+00,
-4.0781e+00, -3.3898e+00],
[ 1.6967e+00, 2.2991e+00, -2.5944e-01, ..., -5.5848e-01,
-9.5338e-01, -1.0571e+00],
[ 9.8568e-01, 1.2536e+00, -1.7959e+00, ..., -2.1562e+00,
-2.4383e+00, -2.3656e+00]],
[[-7.2596e-01, -1.9404e+00, -1.7762e+00, ..., -5.6031e-01,
-1.6970e-01, -4.3386e-01],
[-1.5167e+00, -1.9831e+00, -1.2855e+00, ..., -3.2736e-01,
7.2009e-02, 4.3090e-02],
[-1.3959e+00, -1.3647e+00, -7.3649e-01, ..., -2.7592e-01,
2.2857e-01, 2.7170e-01],
...,
[-9.0635e-01, -1.4351e+00, -7.0236e-01, ..., -5.3017e-01,
-1.8756e-01, -3.8248e-01],
[-1.1822e+00, -8.7648e-01, -5.8886e-01, ..., -3.1673e-01,
1.7053e-01, 2.2592e-01],
[-1.2262e+00, -7.0657e-01, -3.1346e-01, ..., -5.1192e-01,
-2.7078e-01, -2.5385e-01]],
...,
[[-2.3459e+00, -1.1827e+00, -2.5351e+00, ..., -3.5069e+00,
-2.7210e+00, -1.7632e+00],
[-6.1670e-01, 1.2825e+00, -7.3422e-01, ..., -2.5900e+00,
-2.3620e+00, -1.8274e+00],
[-2.0115e-01, 2.6222e+00, 8.2171e-01, ..., -5.4983e-01,
-5.7219e-02, 1.3973e-01],
...,
[-1.3505e+00, 1.3114e+00, 6.4641e-02, ..., -1.6163e+00,
-7.1131e-01, -1.1095e-01],
[-5.0221e-01, 1.4880e+00, 8.3020e-01, ..., -1.1015e-01,
1.5049e-01, 3.8490e-01],
[-1.6283e+00, -1.1017e+00, -1.5138e+00, ..., -2.5351e+00,
-1.5581e+00, -9.5431e-01]],
[[-1.3952e+00, -1.3646e+00, -3.0509e+00, ..., -1.1865e+00,
-7.8651e-01, -7.9345e-01],
[-6.2661e-01, -1.5135e-01, -3.9334e+00, ..., -1.3947e+00,
-1.2334e+00, -1.5707e+00],
[-2.1699e+00, -1.5021e+00, -6.4617e+00, ..., -3.2367e+00,
-2.7538e+00, -2.1943e+00],
...,
[-1.7949e+00, -9.3067e-01, -4.6016e+00, ..., -1.3148e+00,
-9.2073e-01, -8.1943e-01],
[-1.7080e+00, -9.8574e-01, -3.7237e+00, ..., -1.6391e+00,
-1.0765e+00, -6.9882e-01],
[-2.0490e+00, -1.4421e+00, -3.0393e+00, ..., -2.8374e+00,
-2.0252e+00, -1.8671e+00]],
[[-1.2595e+00, -1.9229e+00, -1.7410e+00, ..., -1.7839e+00,
-1.4771e+00, -1.1262e+00],
[-1.2527e+00, -1.8176e+00, -1.3781e+00, ..., -1.4112e+00,
-1.1132e+00, -8.8581e-01],
[-1.6230e+00, -2.0021e+00, -1.5963e+00, ..., -1.6201e+00,
-1.2150e+00, -9.0010e-01],
...,
[-2.3318e+00, -2.9537e+00, -2.8474e+00, ..., -2.9673e+00,
-2.3750e+00, -1.8298e+00],
[-1.5703e+00, -2.0195e+00, -2.0253e+00, ..., -2.0365e+00,
-1.6793e+00, -1.3591e+00],
[-1.0692e+00, -1.3087e+00, -1.4230e+00, ..., -1.2430e+00,
-1.0624e+00, -8.1880e-01]]],
[[[ 2.1288e-01, 4.9040e-01, 6.4399e-01, ..., -3.5973e-01,
-1.0169e-01, -2.0804e-03],
[-1.9723e+00, -1.2863e+00, -2.8044e+00, ..., -3.3914e+00,
-1.4018e+00, -5.0174e-01],
[-1.6955e-01, -8.9662e-01, -9.7521e-03, ..., -1.9299e-01,
-4.0408e-01, 1.2607e-01],
...,
[-9.8689e-01, -2.1720e+00, -2.0769e+00, ..., -1.7064e+00,
-1.6983e+00, -1.0893e+00],
[ 7.9903e-01, -1.1539e-01, -1.3041e-01, ..., 1.9183e-02,
2.5931e-01, 4.1883e-01],
[ 2.0537e-01, -5.7374e-01, -4.0846e-01, ..., 1.9297e-01,
4.2813e-01, 4.3775e-01]],
[[-1.6331e+00, -1.3664e+00, -4.2483e+00, ..., -7.0424e+00,
-6.1873e+00, -4.6259e+00],
[ 4.6722e-01, 1.4103e+00, -2.5678e+00, ..., -5.4204e+00,
-5.5252e+00, -4.2880e+00],
[ 1.1732e+00, 1.1383e+00, -5.0517e+00, ..., -7.8429e+00,
-7.8355e+00, -6.6957e+00],
...,
[ 4.2763e-01, 4.1012e-01, -4.1132e+00, ..., -7.1691e+00,
-6.7186e+00, -5.2198e+00],
[ 2.5918e+00, 2.5492e+00, 1.5220e-01, ..., -2.2965e+00,
-2.8391e+00, -2.2404e+00],
[ 1.5443e+00, 1.3551e+00, -1.6083e+00, ..., -3.0187e+00,
-3.5207e+00, -3.0181e+00]],
[[-1.0919e-01, -1.1260e+00, -1.3645e+00, ..., -1.4832e+00,
-1.0945e+00, -1.3971e+00],
[-4.8029e-01, -8.0528e-01, -8.7094e-01, ..., -1.0461e+00,
-6.4823e-01, -9.1828e-01],
[-3.3090e-01, -2.3506e-01, -4.5007e-01, ..., -1.0944e+00,
-5.6893e-01, -4.4994e-01],
...,
[-7.0920e-02, -6.2322e-01, -6.2068e-01, ..., -9.9278e-01,
-8.3109e-01, -1.1950e+00],
[-7.2742e-01, -2.4378e-01, -5.9189e-01, ..., -5.6997e-01,
-2.5029e-01, -4.2597e-01],
[-9.0065e-01, -2.8736e-01, -4.9713e-01, ..., -5.1600e-01,
-4.3618e-01, -5.9531e-01]],
...,
[[-2.7955e+00, -2.4187e+00, -2.3985e+00, ..., -3.2781e+00,
-2.6523e+00, -1.9232e+00],
[-2.1304e+00, -8.9337e-01, -1.1192e+00, ..., -2.4981e+00,
-2.5721e+00, -2.1016e+00],
[-1.1102e+00, 7.5406e-01, 7.3773e-01, ..., -1.1640e-01,
8.6078e-02, 3.5672e-01],
...,
[-2.2792e+00, -1.7808e-01, -1.3597e-01, ..., -8.6396e-01,
-2.6541e-01, 2.7750e-01],
[-1.2386e+00, 8.8880e-01, 1.0887e+00, ..., 3.7849e-01,
3.8086e-01, 7.6718e-01],
[-2.1649e+00, -1.4952e+00, -1.6822e+00, ..., -2.0001e+00,
-1.1088e+00, -3.3497e-01]],
[[-1.6507e+00, -1.6637e+00, -2.8332e+00, ..., -2.1465e+00,
-1.7914e+00, -1.7148e+00],
[-7.3935e-01, -1.3777e-01, -3.1286e+00, ..., -2.7499e+00,
-2.7741e+00, -2.8642e+00],
[-2.0788e+00, -9.8163e-01, -5.0573e+00, ..., -4.4794e+00,
-4.5344e+00, -3.7351e+00],
...,
[-1.3150e+00, 2.1742e-01, -3.4005e+00, ..., -2.9524e+00,
-2.8319e+00, -2.4139e+00],
[-1.6643e+00, -4.5733e-01, -3.0327e+00, ..., -2.4897e+00,
-2.2874e+00, -1.7401e+00],
[-2.3264e+00, -1.3657e+00, -3.0360e+00, ..., -2.7774e+00,
-2.3653e+00, -2.1557e+00]],
[[-1.3002e+00, -1.8253e+00, -1.6509e+00, ..., -2.6477e+00,
-2.2058e+00, -1.7085e+00],
[-1.2419e+00, -1.7077e+00, -1.2375e+00, ..., -2.4255e+00,
-1.9554e+00, -1.5963e+00],
[-1.5682e+00, -1.9113e+00, -1.4939e+00, ..., -2.8225e+00,
-2.2462e+00, -1.7355e+00],
...,
[-2.4572e+00, -3.0628e+00, -2.7806e+00, ..., -3.9226e+00,
-3.1886e+00, -2.5073e+00],
[-1.5864e+00, -2.0860e+00, -1.9790e+00, ..., -2.7347e+00,
-2.2891e+00, -1.8889e+00],
[-1.0360e+00, -1.3436e+00, -1.3568e+00, ..., -1.8368e+00,
-1.5183e+00, -1.1784e+00]]]], device='cuda:0'),
tensor([[[[ 2.8325e-01, 5.1698e-01, 6.5780e-01, ..., 4.6383e-01,
9.9463e-01, 5.7727e-01],
[-2.2133e+00, -1.5753e+00, -3.1967e+00, ..., -2.6335e+00,
-8.6018e-01, -2.1591e-01],
[-1.9094e-01, -1.0206e+00, -2.2879e-01, ..., -1.4450e-01,
5.4017e-01, 6.1870e-01],
...,
[-1.7749e-01, -1.1562e+00, -2.6453e+00, ..., -1.5656e+00,
-1.4273e+00, -8.7037e-01],
[ 1.4407e+00, 8.4784e-01, -9.2324e-01, ..., -3.1902e-01,
8.5155e-02, 3.7074e-01],
[ 8.5767e-01, 2.8966e-01, -6.8511e-01, ..., -2.0235e-01,
2.1168e-01, 3.8118e-01]],
[[-1.6784e+00, -1.1409e+00, -4.1234e+00, ..., -4.8662e+00,
-4.4200e+00, -3.3708e+00],
[ 6.0124e-01, 1.9422e+00, -2.2346e+00, ..., -3.5033e+00,
-3.6065e+00, -2.9291e+00],
[ 1.3417e+00, 1.6904e+00, -4.9132e+00, ..., -5.4113e+00,
-5.2467e+00, -4.8766e+00],
...,
[-1.2927e+00, 1.4925e-01, -6.7066e+00, ..., -7.6102e+00,
-6.9052e+00, -5.4320e+00],
[ 1.7868e+00, 2.8111e+00, -1.0733e+00, ..., -2.5511e+00,
-2.9289e+00, -2.3632e+00],
[ 1.2120e+00, 1.8073e+00, -2.3022e+00, ..., -3.2341e+00,
-3.6157e+00, -3.1290e+00]],
[[-1.2266e-01, -1.1468e+00, -1.3182e+00, ..., -4.7438e-01,
-1.4740e-01, -4.6784e-01],
[-5.9418e-01, -8.3942e-01, -7.6491e-01, ..., -2.5887e-01,
5.1229e-02, -7.2416e-02],
[-5.4134e-01, -3.3285e-01, -3.9769e-01, ..., -3.3241e-01,
9.0859e-02, 8.3701e-02],
...,
[-1.3859e+00, -1.7643e+00, -7.7919e-01, ..., -8.9484e-01,
-7.7471e-01, -1.0275e+00],
[-1.7315e+00, -1.1609e+00, -8.3362e-01, ..., -5.7319e-01,
-2.9527e-01, -3.6400e-01],
[-1.8877e+00, -1.2902e+00, -8.9789e-01, ..., -9.5749e-01,
-8.2238e-01, -8.0041e-01]],
...,
[[-3.0503e+00, -2.5329e+00, -2.8129e+00, ..., -3.3786e+00,
-2.5704e+00, -1.7271e+00],
[-2.2715e+00, -8.4500e-01, -1.5861e+00, ..., -2.5788e+00,
-2.3105e+00, -1.8062e+00],
[-1.0569e+00, 1.1894e+00, 6.7192e-01, ..., -4.4696e-01,
-1.5145e-02, 1.3766e-01],
...,
[-1.8282e+00, 1.2821e+00, 1.9806e-02, ..., -1.0459e+00,
-5.2090e-01, -9.1359e-04],
[-7.9692e-01, 1.9405e+00, 1.6255e+00, ..., 5.1444e-01,
4.3390e-01, 6.8981e-01],
[-2.2100e+00, -1.1174e+00, -9.4977e-01, ..., -1.8917e+00,
-9.8942e-01, -3.0742e-01]],
[[-1.5829e+00, -1.5768e+00, -2.8817e+00, ..., -1.3074e+00,
-8.3024e-01, -8.2862e-01],
[-5.8616e-01, 8.6622e-02, -3.2071e+00, ..., -1.3832e+00,
-1.1802e+00, -1.5466e+00],
[-2.0775e+00, -8.8784e-01, -5.3209e+00, ..., -3.1666e+00,
-2.6418e+00, -2.1834e+00],
...,
[-2.4704e+00, -1.5773e+00, -5.5067e+00, ..., -3.0484e+00,
-2.7021e+00, -2.1623e+00],
[-2.2020e+00, -1.4507e+00, -4.2732e+00, ..., -2.6524e+00,
-2.2945e+00, -1.6198e+00],
[-2.3715e+00, -1.7506e+00, -3.3104e+00, ..., -2.9587e+00,
-2.4303e+00, -2.1138e+00]],
[[-1.3564e+00, -1.9202e+00, -1.7165e+00, ..., -1.7757e+00,
-1.4801e+00, -1.1237e+00],
[-1.2836e+00, -1.7766e+00, -1.2498e+00, ..., -1.3997e+00,
-1.1064e+00, -8.8229e-01],
[-1.6040e+00, -1.9647e+00, -1.5109e+00, ..., -1.6334e+00,
-1.2222e+00, -9.0522e-01],
...,
[-3.1778e+00, -3.9710e+00, -3.7497e+00, ..., -3.9592e+00,
-3.1711e+00, -2.4521e+00],
[-2.1191e+00, -2.7388e+00, -2.7028e+00, ..., -2.7790e+00,
-2.2915e+00, -1.8688e+00],
[-1.5448e+00, -1.8781e+00, -1.9387e+00, ..., -1.8739e+00,
-1.5209e+00, -1.1669e+00]]],
[[[ 6.3430e-02, 1.9914e-01, -1.4194e+00, ..., -6.3796e-01,
-2.1799e-01, -6.1755e-02],
[-2.0902e+00, -1.6045e+00, -4.5331e+00, ..., -3.7762e+00,
-1.5442e+00, -5.7078e-01],
[ 5.6827e-01, -8.2526e-02, -8.6495e-01, ..., -1.8064e-01,
-2.6939e-01, 2.3151e-01],
...,
[-3.1842e-01, -1.1033e+00, -1.4225e+00, ..., -1.2966e+00,
-8.4217e-01, -6.0552e-01],
[ 9.7261e-01, 5.6242e-01, 3.5128e-01, ..., 4.2697e-01,
7.4212e-01, 8.2951e-01],
[ 6.8915e-01, 4.4668e-01, 3.6240e-01, ..., 5.4239e-01,
7.8697e-01, 7.1676e-01]],
[[-3.2929e+00, -1.8174e+00, -6.3032e+00, ..., -7.1650e+00,
-6.2462e+00, -4.6216e+00],
[-8.0054e-01, 1.4962e+00, -4.4072e+00, ..., -5.5251e+00,
-5.5516e+00, -4.2739e+00],
[-3.7154e-01, 1.0149e+00, -7.4850e+00, ..., -8.0071e+00,
-7.8413e+00, -6.6715e+00],
...,
[-4.7974e-01, 2.5796e-01, -4.4618e+00, ..., -5.3482e+00,
-4.6976e+00, -3.9287e+00],
[ 1.9478e+00, 2.4447e+00, -5.2227e-02, ..., -1.1341e+00,
-1.3153e+00, -1.4491e+00],
[ 1.1023e+00, 1.3098e+00, -1.6718e+00, ..., -2.5160e+00,
-2.6429e+00, -2.5981e+00]],
[[-1.3532e+00, -2.3658e+00, -2.3387e+00, ..., -1.7457e+00,
-1.2340e+00, -1.3822e+00],
[-1.6912e+00, -1.9905e+00, -1.5218e+00, ..., -1.2397e+00,
-7.2104e-01, -8.2545e-01],
[-2.1873e+00, -1.9500e+00, -1.0263e+00, ..., -1.0997e+00,
-4.5792e-01, -1.8558e-01],
...,
[-6.3861e-01, -1.1169e+00, -5.7048e-01, ..., -6.0026e-01,
-4.9805e-01, -6.4004e-01],
[-1.0398e+00, -7.0104e-01, -5.6541e-01, ..., -2.6976e-01,
-8.1606e-02, -4.2978e-02],
[-1.1979e+00, -6.8276e-01, -3.8964e-01, ..., -3.4683e-01,
-2.9843e-01, -2.5843e-01]],
...,
[[-2.0497e+00, -1.1210e+00, -2.3437e+00, ..., -3.1751e+00,
-2.5986e+00, -1.8112e+00],
[-1.8058e+00, -4.5629e-02, -1.3932e+00, ..., -2.5165e+00,
-2.5282e+00, -1.9631e+00],
[-4.1270e-01, 2.2684e+00, 9.9855e-01, ..., 9.1104e-02,
2.5779e-01, 5.1989e-01],
...,
[-1.5708e+00, 9.5410e-01, -1.6228e-01, ..., -1.1836e+00,
-4.7049e-01, 3.5143e-02],
[-5.7632e-01, 1.3810e+00, 7.3281e-01, ..., 2.8851e-02,
1.8396e-01, 2.7813e-01],
[-1.6173e+00, -1.1142e+00, -1.5726e+00, ..., -2.1995e+00,
-1.2819e+00, -6.8964e-01]],
[[-2.4397e+00, -2.8718e+00, -4.2499e+00, ..., -2.4324e+00,
-1.9323e+00, -1.7342e+00],
[-1.7939e+00, -1.6984e+00, -4.9173e+00, ..., -2.9424e+00,
-2.8034e+00, -2.7818e+00],
[-2.8416e+00, -2.5286e+00, -6.6960e+00, ..., -4.4497e+00,
-4.3611e+00, -3.5005e+00],
...,
[-1.6956e+00, -5.4345e-01, -4.0494e+00, ..., -2.1155e+00,
-1.6038e+00, -1.2491e+00],
[-1.6753e+00, -7.4792e-01, -3.3785e+00, ..., -2.0827e+00,
-1.3918e+00, -8.3926e-01],
[-2.0556e+00, -1.3011e+00, -2.8373e+00, ..., -2.6957e+00,
-1.8247e+00, -1.5269e+00]],
[[-1.8192e+00, -2.5490e+00, -2.4712e+00, ..., -2.7081e+00,
-2.2266e+00, -1.7080e+00],
[-1.8119e+00, -2.5260e+00, -2.1725e+00, ..., -2.4201e+00,
-1.9213e+00, -1.5544e+00],
[-2.2667e+00, -2.8314e+00, -2.5491e+00, ..., -2.7686e+00,
-2.1739e+00, -1.6466e+00],
...,
[-2.2740e+00, -2.8589e+00, -2.6737e+00, ..., -2.8603e+00,
-2.2768e+00, -1.7157e+00],
[-1.5322e+00, -1.9727e+00, -1.9288e+00, ..., -2.0348e+00,
-1.6682e+00, -1.3220e+00],
[-1.0357e+00, -1.2669e+00, -1.3325e+00, ..., -1.2768e+00,
-1.0780e+00, -7.9605e-01]]],
[[[ 2.9695e-01, 5.7149e-01, 6.8784e-01, ..., 4.6039e-01,
1.0262e+00, 6.1547e-01],
[-2.1745e+00, -1.4956e+00, -3.1935e+00, ..., -2.6988e+00,
-8.5076e-01, -1.8054e-01],
[-1.6738e-01, -9.8349e-01, -2.3935e-01, ..., -1.8323e-01,
5.8298e-01, 6.7517e-01],
...,
[-1.4719e-01, -9.6301e-01, -2.3558e+00, ..., -1.7251e+00,
-1.6089e+00, -1.0282e+00],
[ 1.2855e+00, 8.1335e-01, -8.7758e-01, ..., -3.0653e-01,
8.7774e-02, 3.7837e-01],
[ 7.5941e-01, 2.6582e-01, -6.2857e-01, ..., -1.5290e-01,
2.4818e-01, 4.0278e-01]],
[[-1.6685e+00, -1.1538e+00, -4.1738e+00, ..., -4.8444e+00,
-4.4458e+00, -3.3721e+00],
[ 5.8775e-01, 1.9205e+00, -2.2837e+00, ..., -3.5097e+00,
-3.6460e+00, -2.9749e+00],
[ 1.3269e+00, 1.6710e+00, -4.9301e+00, ..., -5.4756e+00,
-5.3103e+00, -4.9493e+00],
...,
[-1.5901e+00, -3.8974e-01, -6.9746e+00, ..., -7.5492e+00,
-6.8541e+00, -5.3724e+00],
[ 1.2307e+00, 2.1463e+00, -1.7075e+00, ..., -2.2684e+00,
-2.6820e+00, -2.1796e+00],
[ 8.1839e-01, 1.3975e+00, -2.5605e+00, ..., -3.1424e+00,
-3.5750e+00, -3.1298e+00]],
[[-1.0962e-01, -1.1498e+00, -1.3229e+00, ..., -4.7256e-01,
-1.4709e-01, -4.7752e-01],
[-5.9028e-01, -8.5693e-01, -7.8086e-01, ..., -2.6832e-01,
4.9594e-02, -7.0307e-02],
[-5.4476e-01, -3.4442e-01, -4.0321e-01, ..., -3.6215e-01,
7.9263e-02, 8.2309e-02],
...,
[-1.1450e+00, -1.5020e+00, -7.1398e-01, ..., -9.1871e-01,
-7.9595e-01, -1.0679e+00],
[-1.4518e+00, -9.7255e-01, -7.4247e-01, ..., -5.6864e-01,
-2.8717e-01, -3.5425e-01],
[-1.5867e+00, -1.1276e+00, -8.4659e-01, ..., -9.4305e-01,
-8.1130e-01, -7.9871e-01]],
...,
[[-3.0443e+00, -2.5232e+00, -2.7980e+00, ..., -3.4669e+00,
-2.6748e+00, -1.7593e+00],
[-2.2967e+00, -8.6482e-01, -1.5880e+00, ..., -2.6415e+00,
-2.3947e+00, -1.8435e+00],
[-1.0866e+00, 1.1514e+00, 6.3060e-01, ..., -4.2870e-01,
-4.5113e-02, 1.0125e-01],
...,
[-1.7232e+00, 8.8610e-01, -1.2707e-01, ..., -1.1647e+00,
-5.2186e-01, 6.2365e-02],
[-7.9937e-01, 1.5299e+00, 1.3633e+00, ..., 5.2050e-01,
4.7356e-01, 7.3647e-01],
[-2.1620e+00, -1.2101e+00, -1.0258e+00, ..., -2.0541e+00,
-1.0538e+00, -3.4274e-01]],
[[-1.6028e+00, -1.6001e+00, -2.9208e+00, ..., -1.2657e+00,
-8.4426e-01, -8.6468e-01],
[-6.1740e-01, 5.5360e-02, -3.2249e+00, ..., -1.3745e+00,
-1.2320e+00, -1.6022e+00],
[-2.1072e+00, -9.2554e-01, -5.3308e+00, ..., -3.1741e+00,
-2.7079e+00, -2.2158e+00],
...,
[-2.5662e+00, -1.9330e+00, -5.2543e+00, ..., -2.9531e+00,
-2.5902e+00, -2.0934e+00],
[-2.2721e+00, -1.7327e+00, -4.1379e+00, ..., -2.6090e+00,
-2.2209e+00, -1.5458e+00],
[-2.3302e+00, -1.9038e+00, -3.2810e+00, ..., -3.0101e+00,
-2.4567e+00, -2.1318e+00]],
[[-1.3544e+00, -1.9104e+00, -1.7140e+00, ..., -1.7619e+00,
-1.4557e+00, -1.1140e+00],
[-1.2758e+00, -1.7629e+00, -1.2391e+00, ..., -1.3758e+00,
-1.0762e+00, -8.6525e-01],
[-1.5989e+00, -1.9481e+00, -1.4862e+00, ..., -1.6025e+00,
-1.1962e+00, -8.8812e-01],
...,
[-3.0212e+00, -3.7848e+00, -3.6973e+00, ..., -4.0442e+00,
-3.2403e+00, -2.5028e+00],
[-2.0149e+00, -2.6050e+00, -2.6530e+00, ..., -2.8252e+00,
-2.3353e+00, -1.9078e+00],
[-1.4790e+00, -1.8027e+00, -1.9075e+00, ..., -1.8856e+00,
-1.5442e+00, -1.1832e+00]]],
[[[ 2.9134e-01, 5.4995e-01, 6.9185e-01, ..., 4.7681e-01,
1.0222e+00, 6.0085e-01],
[-2.1316e+00, -1.5108e+00, -3.1972e+00, ..., -2.6956e+00,
-8.7558e-01, -2.1212e-01],
[-1.6893e-01, -1.0181e+00, -2.6392e-01, ..., -1.6446e-01,
5.5468e-01, 6.5097e-01],
...,
[-1.6635e-01, -1.1375e+00, -2.6126e+00, ..., -1.4732e+00,
-1.3945e+00, -8.3440e-01],
[ 1.4546e+00, 8.6953e-01, -9.0256e-01, ..., -2.4561e-01,
8.5694e-02, 3.6725e-01],
[ 8.7615e-01, 3.1945e-01, -6.4825e-01, ..., -1.4309e-01,
2.1061e-01, 3.7600e-01]],
[[-1.6545e+00, -1.1353e+00, -4.1668e+00, ..., -4.8920e+00,
-4.4804e+00, -3.4024e+00],
[ 5.7933e-01, 1.8919e+00, -2.2719e+00, ..., -3.5253e+00,
-3.6429e+00, -2.9714e+00],
[ 1.3009e+00, 1.6640e+00, -4.8814e+00, ..., -5.4751e+00,
-5.2921e+00, -4.9349e+00],
...,
[-1.2690e+00, 1.4568e-01, -6.7048e+00, ..., -7.5912e+00,
-6.8840e+00, -5.3993e+00],
[ 1.7965e+00, 2.8032e+00, -1.0424e+00, ..., -2.5983e+00,
-2.9527e+00, -2.3679e+00],
[ 1.2085e+00, 1.7989e+00, -2.2753e+00, ..., -3.2484e+00,
-3.6116e+00, -3.1137e+00]],
[[-8.5811e-02, -1.1037e+00, -1.2770e+00, ..., -5.0564e-01,
-1.6223e-01, -4.7749e-01],
[-5.5712e-01, -8.2492e-01, -7.6935e-01, ..., -2.8590e-01,
3.8339e-02, -7.1500e-02],
[-5.1831e-01, -3.4619e-01, -4.2122e-01, ..., -3.5250e-01,
8.6767e-02, 8.0407e-02],
...,
[-1.3780e+00, -1.7528e+00, -7.7870e-01, ..., -8.8208e-01,
-7.5849e-01, -1.0037e+00],
[-1.7257e+00, -1.1525e+00, -8.3166e-01, ..., -5.6992e-01,
-2.9173e-01, -3.4599e-01],
[-1.8706e+00, -1.2729e+00, -8.7940e-01, ..., -9.4982e-01,
-8.1591e-01, -7.8201e-01]],
...,
[[-2.9966e+00, -2.4629e+00, -2.7554e+00, ..., -3.4914e+00,
-2.6666e+00, -1.7754e+00],
[-2.3096e+00, -8.6470e-01, -1.5702e+00, ..., -2.6561e+00,
-2.3797e+00, -1.8574e+00],
[-1.0946e+00, 1.1530e+00, 6.2215e-01, ..., -4.4524e-01,
-2.6238e-02, 1.2456e-01],
...,
[-1.8286e+00, 1.2805e+00, 3.4676e-02, ..., -1.1258e+00,
-5.1084e-01, 2.0578e-02],
[-8.1276e-01, 1.9172e+00, 1.6116e+00, ..., 4.0349e-01,
3.9022e-01, 6.5857e-01],
[-2.2262e+00, -1.1476e+00, -1.0129e+00, ..., -1.9051e+00,
-9.9319e-01, -3.1815e-01]],
[[-1.6097e+00, -1.6020e+00, -2.9215e+00, ..., -1.2760e+00,
-8.4111e-01, -8.3889e-01],
[-6.3954e-01, 4.3363e-02, -3.2162e+00, ..., -1.3765e+00,
-1.2121e+00, -1.5719e+00],
[-2.1091e+00, -9.1175e-01, -5.3127e+00, ..., -3.1915e+00,
-2.7039e+00, -2.2105e+00],
...,
[-2.4594e+00, -1.5619e+00, -5.4925e+00, ..., -3.0844e+00,
-2.7078e+00, -2.1525e+00],
[-2.2081e+00, -1.4553e+00, -4.2721e+00, ..., -2.6886e+00,
-2.2897e+00, -1.6160e+00],
[-2.3758e+00, -1.7630e+00, -3.3120e+00, ..., -2.9792e+00,
-2.4161e+00, -2.0937e+00]],
[[-1.3485e+00, -1.9056e+00, -1.7078e+00, ..., -1.7752e+00,
-1.4681e+00, -1.1148e+00],
[-1.2778e+00, -1.7626e+00, -1.2444e+00, ..., -1.3786e+00,
-1.0854e+00, -8.6337e-01],
[-1.6051e+00, -1.9617e+00, -1.4947e+00, ..., -1.6049e+00,
-1.1987e+00, -8.8717e-01],
...,
[-3.1757e+00, -3.9605e+00, -3.7472e+00, ..., -3.9293e+00,
-3.1568e+00, -2.4378e+00],
[-2.1146e+00, -2.7240e+00, -2.6887e+00, ..., -2.7581e+00,
-2.2790e+00, -1.8557e+00],
[-1.5374e+00, -1.8651e+00, -1.9256e+00, ..., -1.8576e+00,
-1.5121e+00, -1.1583e+00]]],
[[[ 6.2456e-02, 2.2964e-01, -1.4021e+00, ..., -5.7679e-01,
-1.7218e-01, -4.2488e-02],
[-2.1130e+00, -1.6015e+00, -4.4968e+00, ..., -3.6084e+00,
-1.4488e+00, -5.2423e-01],
[ 5.5964e-01, -9.7063e-02, -8.4553e-01, ..., -1.3585e-01,
-2.1714e-01, 2.7365e-01],
...,
[-8.8704e-01, -1.8671e+00, -1.8608e+00, ..., -1.8110e+00,
-1.3341e+00, -9.4648e-01],
[ 7.4352e-01, -1.0957e-02, -1.0854e-01, ..., -2.7482e-02,
6.9713e-01, 7.9734e-01],
[ 1.3403e-01, -5.8715e-01, -4.7952e-01, ..., -1.6414e-01,
4.1930e-01, 3.4643e-01]],
[[-3.2971e+00, -1.8063e+00, -6.2158e+00, ..., -7.1111e+00,
-6.2143e+00, -4.6144e+00],
[-7.9523e-01, 1.5418e+00, -4.3263e+00, ..., -5.5486e+00,
-5.5682e+00, -4.2874e+00],
[-3.4367e-01, 1.0908e+00, -7.3977e+00, ..., -8.0272e+00,
-7.8126e+00, -6.6359e+00],
...,
[ 3.0946e-01, 3.4591e-01, -4.0329e+00, ..., -4.6477e+00,
-4.2033e+00, -3.5264e+00],
[ 2.3437e+00, 2.2957e+00, -1.1347e-01, ..., -8.3475e-01,
-1.1868e+00, -1.2479e+00],
[ 1.3779e+00, 1.1900e+00, -1.7487e+00, ..., -2.2456e+00,
-2.5031e+00, -2.4156e+00]],
[[-1.3348e+00, -2.3372e+00, -2.3065e+00, ..., -1.7334e+00,
-1.2329e+00, -1.3601e+00],
[-1.6816e+00, -1.9635e+00, -1.5089e+00, ..., -1.2170e+00,
-7.3411e-01, -8.1623e-01],
[-2.1494e+00, -1.8893e+00, -9.6564e-01, ..., -1.0929e+00,
-4.6888e-01, -1.8903e-01],
...,
[-1.4031e-01, -6.2264e-01, -4.8200e-01, ..., -5.0204e-01,
-1.9056e-01, -4.4459e-01],
[-7.0135e-01, -1.9218e-01, -4.2487e-01, ..., -2.1042e-01,
2.4138e-01, 2.4292e-01],
[-7.9694e-01, -1.8352e-01, -3.3384e-01, ..., -3.9235e-01,
-1.9399e-01, -2.2933e-01]],
...,
[[-2.0276e+00, -1.1076e+00, -2.3631e+00, ..., -3.1062e+00,
-2.5481e+00, -1.7955e+00],
[-1.7145e+00, 5.6559e-02, -1.3760e+00, ..., -2.4487e+00,
-2.4526e+00, -1.9289e+00],
[-3.4012e-01, 2.3963e+00, 1.0265e+00, ..., 1.4166e-01,
2.7936e-01, 5.0837e-01],
...,
[-2.1003e+00, -1.1230e-01, -1.8046e-01, ..., -1.3777e+00,
-4.0074e-01, 9.4847e-02],
[-1.2037e+00, 8.0529e-01, 9.4993e-01, ..., 3.2674e-03,
3.1788e-01, 4.6809e-01],
[-2.1305e+00, -1.4367e+00, -1.6510e+00, ..., -2.4769e+00,
-1.4620e+00, -9.3397e-01]],
[[-2.4040e+00, -2.7952e+00, -4.1942e+00, ..., -2.4548e+00,
-1.9816e+00, -1.7470e+00],
[-1.7535e+00, -1.6057e+00, -4.8725e+00, ..., -2.9898e+00,
-2.8187e+00, -2.7738e+00],
[-2.8175e+00, -2.4300e+00, -6.6701e+00, ..., -4.4812e+00,
-4.3502e+00, -3.4757e+00],
...,
[-1.3946e+00, -4.5284e-02, -3.3641e+00, ..., -1.4415e+00,
-1.0294e+00, -9.2791e-01],
[-1.7133e+00, -6.4403e-01, -3.0172e+00, ..., -1.7545e+00,
-1.1732e+00, -8.0916e-01],
[-2.3365e+00, -1.5000e+00, -3.1039e+00, ..., -2.9852e+00,
-2.1570e+00, -1.9886e+00]],
[[-1.8029e+00, -2.5272e+00, -2.4504e+00, ..., -2.6718e+00,
-2.2034e+00, -1.6812e+00],
[-1.7992e+00, -2.5033e+00, -2.1439e+00, ..., -2.4143e+00,
-1.9148e+00, -1.5407e+00],
[-2.2473e+00, -2.8074e+00, -2.5140e+00, ..., -2.7699e+00,
-2.1738e+00, -1.6361e+00],
...,
[-2.3893e+00, -2.9549e+00, -2.7324e+00, ..., -2.8609e+00,
-2.2828e+00, -1.7568e+00],
[-1.5445e+00, -2.0109e+00, -1.9371e+00, ..., -1.9411e+00,
-1.5970e+00, -1.2913e+00],
[-1.0034e+00, -1.2971e+00, -1.3317e+00, ..., -1.1831e+00,
-1.0121e+00, -7.6526e-01]]]], device='cuda:0')],
4: [tensor([[[[ 1.7784e-02, -2.4517e-01, -9.2729e-03, ..., -9.4209e-01,
-5.3496e-01, -1.4207e+00],
[ 2.9196e-01, -1.0864e+00, -7.5846e-01, ..., -1.1198e+00,
-1.2938e+00, -1.4589e+00],
[ 1.8595e-01, -7.0268e-01, -5.3027e-01, ..., -8.9486e-01,
-1.0860e+00, -1.8073e+00],
...,
[ 5.6449e-02, -5.7945e-01, -4.2217e-01, ..., -9.7206e-01,
-1.0139e+00, -1.9238e+00],
[-4.5125e-01, -5.9207e-01, -7.7416e-02, ..., -4.3960e-01,
-5.4559e-01, -1.7143e+00],
[-6.0414e-01, -1.2255e+00, -1.0701e+00, ..., -1.1109e+00,
-1.5398e+00, -7.6773e-01]],
[[-7.1815e-01, -2.1219e+00, -1.7078e+00, ..., -3.9206e+00,
-5.4370e+00, -1.8476e+00],
[-2.2349e+00, -4.1117e+00, -3.0213e+00, ..., -5.8840e+00,
-8.0092e+00, -3.3813e+00],
[-1.6252e+00, -2.7816e+00, -2.5097e+00, ..., -4.9893e+00,
-6.1647e+00, -2.5484e+00],
...,
[-1.8045e+00, -2.4773e+00, -2.4860e+00, ..., -5.0666e+00,
-6.3060e+00, -2.4850e+00],
[-3.5335e+00, -4.0659e+00, -3.4022e+00, ..., -6.4456e+00,
-7.9263e+00, -2.5614e+00],
[ 3.0177e-01, -2.7368e-01, 3.4708e-01, ..., 4.3201e-01,
4.6529e-01, 2.1058e-01]],
[[-3.1910e-01, 5.2476e-02, 7.8333e-01, ..., 6.4821e-01,
2.5765e-01, 1.5251e+00],
[ 1.0058e-01, -4.3382e-01, 6.3368e-01, ..., 6.0179e-01,
-3.3227e-02, 1.4589e-01],
[-2.1918e-01, -2.5854e-01, 2.4888e-01, ..., 5.1823e-02,
-3.3608e-02, -1.1609e-01],
...,
[-2.4905e-01, -2.1840e-01, 1.5588e-01, ..., 7.3594e-02,
-9.8207e-02, -1.4966e-01],
[-2.2265e-01, -1.8419e-01, 1.7049e-01, ..., -1.0175e-01,
2.9216e-03, -1.6549e-01],
[-8.1247e-02, -4.9902e-01, 9.6652e-02, ..., 8.5773e-02,
2.0166e-01, -5.4309e-01]],
...,
[[ 4.5030e-01, -3.9508e-01, -7.4701e-02, ..., 4.1318e-01,
3.4338e-01, -1.2942e+00],
[-7.0379e-01, -5.9941e-01, -6.1534e-01, ..., -1.4559e-01,
-7.6507e-01, -1.3955e+00],
[-4.4288e-01, -2.0376e-01, -2.3766e-01, ..., -1.0918e-01,
-7.3434e-02, -9.4657e-01],
...,
[-3.6216e-01, -1.7167e-01, -1.8055e-01, ..., -3.1962e-02,
-1.2572e-01, -9.3364e-01],
[-2.0773e+00, -1.5138e+00, -1.1724e+00, ..., -1.5116e+00,
-1.5665e+00, -1.8170e+00],
[-4.6691e-01, -6.9027e-01, -4.0725e-01, ..., -3.4721e-01,
-5.3729e-01, -1.2013e+00]],
[[ 7.0427e-02, 2.4086e-01, 7.0477e-01, ..., -9.7050e-02,
3.6284e-01, -4.0074e-01],
[ 2.8358e-02, 2.6342e-01, 4.3217e-01, ..., 2.0031e-01,
9.5580e-02, -8.1006e-01],
[-2.3449e-01, -1.4131e-02, 2.6219e-01, ..., 2.3652e-01,
1.3879e-01, -4.5197e-01],
...,
[-1.1629e-01, 5.2732e-02, 2.8197e-01, ..., 2.3178e-01,
1.3496e-01, -5.3453e-01],
[-4.4699e-01, -5.5150e-01, 6.9751e-03, ..., -1.3005e-01,
-2.5857e-01, -9.3917e-01],
[ 5.2450e-01, 4.5552e-01, 9.4590e-02, ..., 3.6548e-01,
3.3946e-01, 2.7734e-01]],
[[-9.9773e-01, 2.9779e-01, 3.9981e-01, ..., -5.0672e-02,
-4.8034e-02, 5.6396e-01],
[-1.2146e+00, 2.8196e-01, -3.6062e-01, ..., -9.2297e-01,
-4.3943e-01, 8.7036e-01],
[-9.2025e-01, 4.0123e-01, -1.2339e-01, ..., -5.8355e-01,
-5.3378e-01, 6.0786e-01],
...,
[-9.8483e-01, 2.5136e-01, -8.6992e-02, ..., -4.7591e-01,
-5.1594e-01, 6.1268e-01],
[-1.6050e+00, -2.7456e-01, -3.1492e-01, ..., -6.9848e-01,
-6.5170e-01, 4.1165e-01],
[-6.8720e-01, 3.6564e-02, -2.2706e-01, ..., -4.8926e-01,
-4.3270e-01, 9.9477e-01]]],
[[[-3.2902e-01, -5.0956e-01, -1.3676e-01, ..., -9.3642e-01,
-5.3027e-01, -1.4273e+00],
[ 1.6731e-02, -1.4123e+00, -6.9002e-01, ..., -1.1308e+00,
-1.3367e+00, -1.4839e+00],
[-9.5800e-02, -1.2769e+00, -5.0577e-01, ..., -9.1305e-01,
-1.1013e+00, -1.8315e+00],
...,
[-2.9137e-01, -1.3107e+00, -4.7235e-01, ..., -1.1198e+00,
-1.0831e+00, -1.9319e+00],
[-5.0447e-01, -1.1170e+00, -7.9764e-02, ..., -5.0188e-01,
-5.6468e-01, -1.7148e+00],
[-5.8192e-01, -1.5840e+00, -1.2656e+00, ..., -1.1364e+00,
-1.5314e+00, -7.8077e-01]],
[[-6.8816e-01, -1.9507e+00, -7.9561e-01, ..., -3.8968e+00,
-5.4183e+00, -1.8208e+00],
[-2.1689e+00, -3.9234e+00, -2.0263e+00, ..., -5.8830e+00,
-8.0398e+00, -3.3915e+00],
[-1.6831e+00, -2.9129e+00, -1.5586e+00, ..., -5.0373e+00,
-6.1922e+00, -2.5839e+00],
...,
[-1.6724e+00, -2.7495e+00, -1.4951e+00, ..., -5.1352e+00,
-6.3820e+00, -2.4768e+00],
[-3.7439e+00, -4.5780e+00, -2.9049e+00, ..., -6.4956e+00,
-7.9236e+00, -2.5316e+00],
[ 5.7227e-01, -6.8323e-01, 3.9638e-01, ..., 4.3459e-01,
4.8475e-01, 2.2290e-01]],
[[-3.6012e-01, 7.4578e-01, 1.0255e+00, ..., 6.3954e-01,
2.9328e-01, 1.5497e+00],
[-7.3733e-01, 2.5685e-01, 5.3192e-01, ..., 5.6933e-01,
-4.1161e-02, 1.1957e-01],
[-1.2610e+00, 3.1895e-01, 1.8597e-01, ..., 3.4530e-02,
1.1028e-02, -1.4257e-01],
...,
[-1.4261e+00, 3.2461e-01, 1.3213e-01, ..., -6.0439e-02,
-9.3691e-02, -1.7884e-01],
[-1.4091e+00, 3.2769e-01, 2.1495e-01, ..., -8.6817e-02,
2.8126e-02, -1.7769e-01],
[-9.0618e-01, 6.6984e-02, -1.3116e-01, ..., 6.9695e-02,
2.3145e-01, -5.3755e-01]],
...,
[[ 4.0471e-01, -1.2250e-01, -1.1904e-01, ..., 4.6223e-01,
3.3780e-01, -1.3455e+00],
[-1.3281e+00, -1.0922e+00, -6.2995e-01, ..., -1.5782e-01,
-7.9760e-01, -1.4281e+00],
[-6.9585e-01, -4.8519e-01, -1.1637e-01, ..., -8.7951e-02,
-1.0932e-01, -9.7165e-01],
...,
[-9.5885e-01, -5.8650e-01, -1.2296e-01, ..., -2.3854e-01,
-7.7780e-02, -9.2944e-01],
[-2.5499e+00, -1.7706e+00, -8.4613e-01, ..., -1.4842e+00,
-1.5609e+00, -1.8117e+00],
[-6.3126e-01, -1.1344e+00, -8.7307e-02, ..., -3.4684e-01,
-5.3235e-01, -1.1795e+00]],
[[ 4.3911e-02, 2.3574e-01, 5.0063e-01, ..., -8.6400e-02,
4.1669e-01, -3.4478e-01],
[-1.0710e-01, -6.5051e-01, -2.6543e-02, ..., 2.1626e-01,
1.2318e-01, -8.2299e-01],
[-4.9040e-01, -1.0259e+00, -1.4856e-01, ..., 2.4951e-01,
1.6163e-01, -4.7014e-01],
...,
[-5.9287e-01, -1.1082e+00, -1.3584e-01, ..., 3.1583e-01,
7.3446e-02, -5.0929e-01],
[-1.0591e+00, -1.9595e+00, -4.6210e-01, ..., -2.1823e-01,
-3.0135e-01, -9.0120e-01],
[ 2.5910e-01, -4.4474e-01, -2.5014e-01, ..., 3.5086e-01,
3.5704e-01, 3.0721e-01]],
[[-3.4554e-01, 1.1003e-01, 5.2229e-01, ..., -8.7611e-02,
-6.1139e-02, 5.9531e-01],
[-3.5650e-01, -1.3460e-02, -1.3085e-03, ..., -9.2267e-01,
-3.9982e-01, 8.9719e-01],
[ 4.3007e-02, 5.5124e-02, 3.2157e-02, ..., -5.9175e-01,
-5.0496e-01, 6.2675e-01],
...,
[ 4.4295e-02, -6.0400e-02, 9.6896e-02, ..., -2.0752e-01,
-4.3739e-01, 6.0030e-01],
[-3.8831e-01, -5.0409e-01, -1.4154e-01, ..., -6.0110e-01,
-6.7962e-01, 3.8789e-01],
[ 2.1076e-02, -5.0601e-01, -1.8141e-01, ..., -4.7656e-01,
-4.2234e-01, 1.0048e+00]]],
[[[ 1.7751e-02, -2.6959e-01, -4.8824e-02, ..., -1.2008e+00,
-6.4915e-01, -1.0538e+00],
[ 2.9938e-01, -9.5320e-01, -7.1955e-01, ..., -1.6835e+00,
-1.3563e+00, -1.3361e+00],
[ 2.6389e-01, -5.7638e-01, -5.4039e-01, ..., -1.2402e+00,
-1.3713e+00, -2.4150e+00],
...,
[-4.0888e-03, 2.3781e-01, 6.0403e-01, ..., -3.8645e+00,
-7.5241e-01, -1.4432e+00],
[-6.0411e-01, -2.9720e-01, 1.0008e+00, ..., -8.4806e-01,
-3.7772e-01, -1.5963e+00],
[-7.1396e-01, -1.3026e+00, -1.0144e+00, ..., -1.2621e+00,
-1.4749e+00, -9.0760e-01]],
[[-8.2369e-01, -2.2467e+00, -2.0197e+00, ..., -4.1695e+00,
-4.9636e+00, -1.8440e+00],
[-2.0841e+00, -3.8141e+00, -3.0532e+00, ..., -6.3707e+00,
-7.1588e+00, -2.8833e+00],
[-1.6264e+00, -2.6188e+00, -2.6806e+00, ..., -7.4844e+00,
-8.2872e+00, -3.8996e+00],
...,
[-1.8963e+00, -2.0069e+00, -2.6388e+00, ..., -6.7154e+00,
-6.4634e+00, -3.2781e+00],
[-3.8701e+00, -4.6263e+00, -3.7299e+00, ..., -6.8487e+00,
-7.5531e+00, -2.9493e+00],
[ 2.9425e-01, -3.3413e-01, 3.7493e-01, ..., 5.6034e-01,
6.5313e-01, 4.8687e-02]],
[[-1.9175e-01, 1.7388e-01, 7.3822e-01, ..., 9.7254e-01,
6.9634e-01, 1.5301e+00],
[ 1.0004e-01, -3.6813e-01, 5.4011e-01, ..., 2.8026e-01,
3.4221e-01, 2.6692e-01],
[-1.3458e-01, -1.9071e-01, 2.1942e-01, ..., -1.6338e-01,
1.9856e-02, -1.2171e-02],
...,
[-3.1868e-01, 5.0462e-01, -2.1695e+00, ..., 1.9218e+00,
4.0528e-01, -1.0102e+00],
[-2.7106e-01, 6.8113e-01, 8.0114e-02, ..., 5.8589e-01,
3.4281e-01, 2.6557e-02],
[-1.1659e-01, -4.1430e-01, 2.2063e-01, ..., -3.8361e-01,
-7.8612e-02, -3.7551e-01]],
...,
[[ 5.9866e-01, -2.1652e-01, 4.6972e-02, ..., -1.1464e-01,
-2.7540e-01, -1.5283e+00],
[-5.0845e-01, -5.0644e-01, -4.7388e-01, ..., -7.8088e-01,
-1.5332e+00, -1.7452e+00],
[-3.4309e-01, -2.2096e-01, -2.2898e-01, ..., -6.9131e-01,
-8.8003e-01, -1.1946e+00],
...,
[-4.2839e-01, 2.1430e-01, 1.6190e+00, ..., -2.4030e+00,
-4.3045e+00, -3.5321e+00],
[-2.3393e+00, -2.3642e+00, -2.0144e+00, ..., -3.2772e+00,
-2.5900e+00, -1.6334e+00],
[-5.0947e-01, -8.5914e-01, -5.6287e-01, ..., -7.1893e-01,
-7.8228e-01, -8.8791e-01]],
[[ 1.7531e-01, 2.7333e-01, 6.5526e-01, ..., 4.3833e-01,
7.6719e-01, -5.0378e-02],
[ 1.7520e-01, 2.4441e-01, 4.2832e-01, ..., 2.6778e-01,
3.8757e-01, -3.3858e-01],
[-1.0057e-01, 8.2056e-02, 2.5909e-01, ..., 6.3852e-01,
4.7543e-01, -1.2626e+00],
...,
[-1.9096e-01, 2.9385e-01, 1.5872e-01, ..., -1.2012e+00,
1.6193e+00, 2.3369e-01],
[-5.4960e-01, -4.0956e-01, 8.4815e-02, ..., -9.7764e-01,
-2.7214e-01, -6.3506e-01],
[ 5.1007e-01, 3.9625e-01, 1.9699e-01, ..., 2.6902e-01,
1.9459e-01, 3.2237e-01]],
[[-8.6064e-01, 2.3332e-01, 3.0230e-01, ..., 7.6438e-02,
1.8715e-01, 8.4100e-01],
[-1.1384e+00, 1.9492e-01, -3.6790e-01, ..., -1.0584e-01,
1.1036e-01, 1.3961e+00],
[-9.3177e-01, 3.1456e-01, -1.3860e-01, ..., 4.7355e-02,
3.7689e-01, 1.3740e+00],
...,
[-9.4098e-01, -6.8877e-02, 6.1770e-01, ..., -1.3666e+00,
-1.4283e-01, 2.3930e+00],
[-1.6781e+00, -5.6445e-01, 2.7961e-01, ..., 9.1403e-01,
1.0657e+00, 1.5119e+00],
[-7.2956e-01, -1.0165e-01, -2.6993e-01, ..., 2.6469e-01,
1.3906e-01, 1.1177e+00]]],
[[[-2.0673e-01, -5.8630e-01, -2.4755e-01, ..., -7.5511e-01,
1.0660e+00, -2.0410e-01],
[ 2.5138e-01, -1.3514e+00, -6.7190e-01, ..., 5.7608e-01,
-1.1085e+00, -3.0873e-01],
[ 1.0750e-01, -1.0683e+00, -4.0493e-01, ..., -5.7448e-01,
-1.9308e+00, -6.4106e-01],
...,
[-3.8601e-01, -5.9391e-01, -5.5992e-02, ..., -5.1806e+00,
-2.7936e+00, -9.4399e-01],
[-1.2880e+00, -1.8137e+00, -7.6015e-01, ..., -5.9575e-01,
1.8387e+00, -1.3622e+00],
[-1.3502e+00, -2.3250e+00, -1.7402e+00, ..., -1.1842e+00,
-1.5734e+00, -1.8190e+00]],
[[-1.7605e+00, -3.4842e+00, -2.9431e+00, ..., -7.1001e+00,
-6.4486e+00, -1.4057e+00],
[-3.3062e+00, -5.2448e+00, -3.9230e+00, ..., -1.0847e+01,
-7.2804e+00, -2.1238e+00],
[-2.2755e+00, -3.3932e+00, -2.6753e+00, ..., -6.1288e+00,
-4.3386e+00, -1.4146e+00],
...,
[-4.0380e+00, -6.0593e+00, -5.1802e+00, ..., -1.3490e+01,
-9.7347e+00, -6.5358e+00],
[-6.3739e+00, -7.3739e+00, -6.3620e+00, ..., -7.9255e+00,
-8.4983e+00, -4.1580e+00],
[-8.6519e-02, -9.2951e-01, 1.6239e-01, ..., -7.2653e-01,
-5.4809e-01, -5.9165e-01]],
[[-4.9503e-01, 2.8697e-01, 8.3722e-01, ..., -8.3374e-01,
1.4113e+00, 1.3287e+00],
[-5.6463e-01, -2.4095e-02, 5.9535e-01, ..., -2.3807e+00,
1.9602e+00, 3.3668e-01],
[-8.9945e-01, 7.3719e-03, 3.0098e-01, ..., -1.3568e+00,
1.3893e+00, 3.9769e-01],
...,
[-1.4565e+00, 1.5977e+00, -1.3373e+00, ..., -1.6629e+00,
-3.9506e+00, -4.0774e+00],
[-1.1747e+00, 2.0476e-01, -1.6618e-01, ..., 1.8064e+00,
-1.2704e+00, -1.5880e+00],
[-1.1673e+00, -1.4572e-01, -5.7370e-02, ..., 1.1655e+00,
-7.8491e-01, -2.8653e-01]],
...,
[[ 8.2522e-01, 1.9152e-01, 2.4287e-01, ..., -2.4039e+00,
-2.4047e+00, -9.1263e-01],
[-1.0335e+00, -8.0486e-01, -6.1808e-01, ..., -7.4345e+00,
-4.8877e+00, -1.1266e+00],
[-6.1408e-01, -3.6050e-01, -3.0104e-01, ..., -6.6294e+00,
-1.7636e+00, -6.7612e-01],
...,
[-5.7096e-01, -1.5295e+00, -1.6592e+00, ..., -2.0565e+00,
-2.8844e+00, -6.9579e+00],
[-3.1751e+00, -3.2297e+00, -2.3216e+00, ..., -6.2387e+00,
-8.0763e+00, -6.7475e+00],
[-5.7533e-01, -1.7293e+00, -1.1342e+00, ..., -3.7658e+00,
-3.4776e+00, -2.3882e+00]],
[[ 6.9225e-02, 2.3503e-01, 5.8263e-01, ..., -6.8749e-01,
1.3745e+00, 2.5285e-01],
[-3.3279e-03, -4.2597e-01, 2.1323e-01, ..., 6.5606e-01,
5.1159e-01, -1.6961e-01],
[-2.7887e-01, -6.1753e-01, 1.9268e-01, ..., 4.0484e-01,
-8.3737e-01, -3.1837e-02],
...,
[-1.1048e-01, -4.0741e-01, 3.3553e-02, ..., -1.8756e+00,
-1.4368e+00, 3.1010e-01],
[-5.0853e-01, -1.6660e+00, 3.1898e-03, ..., -4.1985e-01,
1.3117e+00, -4.5514e-01],
[ 4.4217e-01, -1.2008e-01, 1.7154e-01, ..., -6.5946e-01,
2.2293e-01, -2.7040e-01]],
[[-8.7943e-01, -2.7343e-01, 1.2516e-01, ..., -2.8070e-01,
3.0457e-01, 7.5097e-01],
[-1.1184e+00, -2.8877e-01, -3.7704e-01, ..., 1.5544e+00,
1.1732e+00, 1.1680e+00],
[-7.8773e-01, -2.5722e-01, -2.8092e-01, ..., 2.3033e+00,
8.9769e-01, 7.2079e-01],
...,
[-1.3429e+00, -1.6982e+00, 1.3675e-01, ..., -8.9583e+00,
-5.3142e+00, 1.9891e+00],
[-2.1566e+00, -2.0757e+00, -8.3488e-01, ..., -5.5768e+00,
-2.3117e+00, 2.2746e+00],
[-1.0757e+00, -1.0308e+00, -2.6045e-01, ..., 1.9430e-01,
7.0681e-01, 2.0532e+00]]],
[[[-3.0262e-01, -9.8182e-01, -3.4237e-01, ..., -3.5017e-01,
-4.8742e-02, -7.6910e-01],
[ 7.2217e-03, -1.7853e+00, -3.5031e-01, ..., -1.0759e+00,
-8.2953e-01, -1.1807e+00],
[-4.1835e-04, -1.3734e+00, -5.2212e-01, ..., -1.0750e+00,
-8.0868e-01, -1.7511e+00],
...,
[ 1.3970e-01, -5.0049e-01, -1.9582e-01, ..., -2.3074e-01,
-1.7347e-01, -4.5979e-01],
[-3.2570e-01, -4.7694e-01, -3.8521e-03, ..., 1.4553e-01,
2.6952e-01, -1.7466e-01],
[-6.4213e-01, -1.3334e+00, -1.0990e+00, ..., -1.0887e+00,
-1.1727e+00, -4.9513e-01]],
[[-1.9794e+00, -3.8755e+00, -6.0585e+00, ..., -4.3922e+00,
-4.9552e+00, -1.9751e+00],
[-3.7549e+00, -5.8997e+00, -5.0564e+00, ..., -5.9308e+00,
-6.8897e+00, -2.9668e+00],
[-3.4383e+00, -5.2089e+00, -5.1119e+00, ..., -5.4531e+00,
-5.7758e+00, -2.7939e+00],
...,
[-1.6854e+00, -2.1811e+00, -1.8183e+00, ..., -2.2722e+00,
-2.4469e+00, -9.3234e-01],
[-2.7677e+00, -2.9921e+00, -1.9794e+00, ..., -2.2799e+00,
-3.3878e+00, -1.0309e+00],
[ 3.1625e-01, -2.6518e-01, 3.3310e-01, ..., 3.2074e-01,
4.7413e-01, 2.0359e-01]],
[[-3.8713e-01, 5.3113e-01, 4.1560e-01, ..., 8.9197e-01,
5.5517e-01, 1.4851e+00],
[-5.9178e-01, 2.1236e-02, 1.1031e+00, ..., 6.3621e-01,
2.4179e-01, 3.9573e-01],
[-1.0889e+00, 1.1912e-01, 1.0961e+00, ..., 4.4649e-01,
1.4853e-01, 3.1554e-01],
...,
[-6.1686e-01, 1.0516e-01, 2.4990e-01, ..., 6.9186e-04,
2.6734e-01, 3.7492e-01],
[-4.5083e-01, 7.0874e-02, 1.5011e-01, ..., 1.6753e-01,
2.1290e-01, 5.3142e-01],
[-1.5420e-01, -2.6098e-01, 1.2333e-01, ..., 4.9489e-03,
8.8479e-02, 3.4340e-01]],
...,
[[ 1.1478e+00, 3.7476e-01, 1.4392e+00, ..., -3.2468e-01,
2.3305e-01, -1.0285e+00],
[-6.5240e-01, -5.4339e-01, -8.6996e-02, ..., -1.1149e+00,
-1.2290e+00, -1.3573e+00],
[-1.8533e-01, -4.5476e-01, -1.1061e+00, ..., -4.4526e-01,
-5.6475e-01, -1.1469e+00],
...,
[-6.0657e-01, -4.7374e-01, -3.3112e-01, ..., -5.1296e-01,
-4.8871e-01, -5.5475e-01],
[-2.2825e+00, -1.8332e+00, -1.3096e+00, ..., -1.3205e+00,
-1.3642e+00, -1.0786e+00],
[-5.9743e-01, -7.8661e-01, -4.3071e-01, ..., -2.4114e-01,
-3.6342e-01, -4.0310e-01]],
[[ 2.2187e-01, 2.2866e-01, 5.9778e-01, ..., 1.9986e-01,
7.1571e-01, -2.5141e-01],
[ 4.2557e-01, -6.3789e-01, 2.0084e-01, ..., 4.6943e-01,
3.7288e-01, -7.7507e-01],
[ 1.4116e-02, -6.9498e-01, 3.1594e-01, ..., -3.8728e-04,
1.3616e-01, -6.8119e-01],
...,
[-1.8501e-01, -3.8236e-01, 2.9983e-01, ..., 3.1957e-01,
2.1968e-01, 6.7438e-02],
[-5.0084e-01, -8.7446e-01, 5.4371e-02, ..., 3.0119e-02,
-3.3895e-02, -1.7735e-01],
[ 2.9542e-01, -2.0096e-02, -1.9255e-01, ..., -1.3103e-01,
-9.6940e-02, 1.7995e-01]],
[[-6.7506e-01, -2.8592e-03, -8.2013e-03, ..., 4.2672e-01,
3.9987e-01, 9.4711e-01],
[-1.2178e+00, -6.6958e-01, -1.2088e+00, ..., -5.3212e-01,
-2.5657e-01, 9.7962e-01],
[-9.2603e-01, -6.4702e-01, -7.1158e-01, ..., -2.9672e-01,
-9.3292e-02, 7.5101e-01],
...,
[-6.6811e-01, -1.8144e-01, -8.6274e-02, ..., 2.1857e-01,
4.2751e-02, 6.2974e-01],
[-1.1514e+00, -5.6725e-01, -3.4310e-01, ..., -3.2960e-01,
-1.8602e-01, 1.7594e-01],
[-3.6907e-01, -2.3795e-02, -1.8189e-01, ..., -2.3704e-01,
-2.1938e-01, 3.2892e-01]]]], device='cuda:0'),
tensor([[[[-7.2425e-01, -1.4776e+00, -1.4652e+00, ..., -1.2610e+00,
-8.3426e-01, -1.6545e+00],
[-3.3571e-01, -2.4577e+00, -2.2881e+00, ..., -1.7975e+00,
-1.8995e+00, -1.9579e+00],
[-1.1403e-01, -1.9142e+00, -1.7180e+00, ..., -5.6991e-01,
-1.3690e+00, -2.1634e+00],
...,
[ 7.2834e-02, -5.4750e-01, -2.1973e-01, ..., -1.3025e-01,
-6.0575e-02, -4.2124e-01],
[-3.6293e-01, -4.5559e-01, 6.4073e-02, ..., 1.4326e-01,
2.7506e-01, -1.6147e-01],
[-6.3202e-01, -1.3184e+00, -1.0833e+00, ..., -1.1042e+00,
-1.1842e+00, -4.9627e-01]],
[[-2.6871e+00, -5.1095e+00, -4.5745e+00, ..., -4.7573e+00,
-5.9701e+00, -2.3759e+00],
[-5.3727e+00, -8.1914e+00, -7.5349e+00, ..., -8.6863e+00,
-9.4609e+00, -4.4225e+00],
[-4.0755e+00, -5.8870e+00, -5.7851e+00, ..., -6.8273e+00,
-6.7497e+00, -3.3223e+00],
...,
[-1.7449e+00, -2.2326e+00, -1.7706e+00, ..., -1.8300e+00,
-2.2804e+00, -9.3529e-01],
[-2.7823e+00, -2.9637e+00, -2.0217e+00, ..., -2.2569e+00,
-3.3990e+00, -1.0103e+00],
[ 3.2902e-01, -2.8276e-01, 3.7411e-01, ..., 3.0294e-01,
4.3151e-01, 1.9240e-01]],
[[-6.5179e-01, 4.8477e-01, 8.6535e-01, ..., 8.2399e-01,
5.8981e-01, 1.9106e+00],
[-9.6285e-01, 7.1204e-02, 4.2449e-01, ..., 7.9731e-01,
4.4110e-03, 1.5769e-01],
[-1.7547e+00, -2.0690e-01, -3.7570e-02, ..., 4.2797e-01,
7.1222e-02, -7.7498e-02],
...,
[-6.5676e-01, 6.7040e-02, 2.5995e-01, ..., 1.8482e-01,
2.0646e-01, 4.4527e-01],
[-4.5485e-01, 7.3218e-02, 2.0946e-01, ..., 1.7173e-01,
2.0078e-01, 5.6964e-01],
[-1.4087e-01, -2.0661e-01, 9.8588e-02, ..., -2.3456e-02,
9.6193e-02, 3.4191e-01]],
...,
[[ 1.2340e+00, -3.6872e-01, -5.8801e-02, ..., 2.2640e-01,
-1.1551e-01, -2.1384e+00],
[-7.1809e-01, -9.0779e-01, -5.2038e-01, ..., -1.6119e-01,
-1.4721e+00, -2.0756e+00],
[-6.7507e-01, -8.7870e-01, -2.7310e-01, ..., -8.4251e-01,
-5.7192e-01, -1.3753e+00],
...,
[-6.7302e-01, -4.8412e-01, -3.6523e-01, ..., -3.4523e-01,
-4.7276e-01, -5.7062e-01],
[-2.2829e+00, -1.7859e+00, -1.3308e+00, ..., -1.3446e+00,
-1.3851e+00, -1.1043e+00],
[-6.1762e-01, -8.5878e-01, -4.6015e-01, ..., -2.4787e-01,
-3.5998e-01, -3.9322e-01]],
[[ 5.0229e-02, 9.3040e-03, 4.4304e-01, ..., 4.6148e-01,
9.4473e-01, -3.1334e-03],
[ 2.5591e-01, -7.9483e-01, -7.8157e-02, ..., -2.0090e-01,
2.9969e-01, -8.7492e-01],
[ 2.0570e-01, -6.1336e-01, 2.8458e-01, ..., 6.2789e-01,
5.3376e-01, -4.1315e-01],
...,
[-2.4690e-01, -4.7359e-01, 2.2780e-01, ..., 3.5878e-01,
3.3898e-01, 3.7907e-02],
[-5.2859e-01, -9.2119e-01, 9.1164e-02, ..., 3.5089e-02,
-3.3527e-02, -1.8974e-01],
[ 3.2125e-01, -3.5184e-02, -1.8914e-01, ..., -1.3438e-01,
-1.2349e-01, 1.7536e-01]],
[[-1.5422e+00, -7.8110e-01, -4.2986e-01, ..., -4.6523e-01,
-2.9306e-01, 8.7026e-01],
[-2.1272e+00, -7.3323e-01, -5.4197e-01, ..., -8.1465e-01,
-1.7815e-01, 1.6600e+00],
[-1.4802e+00, -5.1257e-01, -2.1912e-01, ..., -6.2125e-01,
-8.3922e-02, 1.3952e+00],
...,
[-6.6418e-01, -2.3229e-01, -1.3171e-01, ..., -1.8441e-01,
3.2057e-02, 6.0112e-01],
[-1.1454e+00, -5.8691e-01, -2.8603e-01, ..., -3.3620e-01,
-2.1175e-01, 1.5576e-01],
[-3.5027e-01, -1.3284e-02, -1.5136e-01, ..., -2.3665e-01,
-2.4551e-01, 3.2723e-01]]],
[[[-1.1111e-01, -5.0541e-01, -2.0956e-01, ..., 1.1121e-01,
3.7937e-01, 2.4666e-02],
[ 3.0987e-01, -1.0768e+00, -6.6914e-01, ..., -6.2074e-01,
-2.6648e-01, -1.3137e-01],
[ 2.3672e-01, -7.2513e-01, -3.1079e-01, ..., -4.7202e-01,
-2.0218e-01, -5.1844e-01],
...,
[-2.1019e-01, -1.7075e+00, -1.2368e+00, ..., -9.1728e-01,
-9.0660e-01, -1.9980e+00],
[-1.0120e+00, -1.6967e+00, -8.0665e-01, ..., -5.2331e-01,
-4.1130e-01, -1.7651e+00],
[-1.2171e+00, -2.1403e+00, -1.6133e+00, ..., -1.4447e+00,
-1.5712e+00, -9.4971e-01]],
[[-1.4262e+00, -2.9621e+00, -2.5543e+00, ..., -3.0609e+00,
-3.3034e+00, -1.2365e+00],
[-2.9646e+00, -4.7177e+00, -3.6328e+00, ..., -4.2624e+00,
-4.8901e+00, -2.1279e+00],
[-1.9242e+00, -2.7746e+00, -2.2507e+00, ..., -2.6401e+00,
-2.6875e+00, -1.2335e+00],
...,
[-3.7091e+00, -5.5161e+00, -5.1585e+00, ..., -5.3485e+00,
-6.1132e+00, -2.8708e+00],
[-6.1146e+00, -6.9857e+00, -6.0639e+00, ..., -6.0436e+00,
-7.3296e+00, -2.6374e+00],
[ 9.9360e-03, -8.1039e-01, 2.9036e-01, ..., 3.8872e-01,
6.7955e-01, 1.9789e-01]],
[[-3.7774e-01, 2.6225e-01, 8.2257e-01, ..., 1.1084e+00,
5.5516e-01, 1.3851e+00],
[-2.4107e-01, -2.0120e-04, 5.8009e-01, ..., 7.8151e-01,
4.1654e-01, 2.7836e-01],
[-6.3868e-01, 8.8346e-02, 3.1190e-01, ..., 2.3878e-01,
3.7828e-01, 4.3343e-01],
...,
[-1.3808e+00, -5.8006e-03, 7.1915e-03, ..., 6.5158e-02,
7.4606e-03, 3.1371e-01],
[-1.2034e+00, 1.5599e-01, -1.6472e-02, ..., -8.7142e-03,
1.7906e-02, 2.9484e-01],
[-1.1306e+00, -2.7012e-02, 1.0950e-01, ..., 1.7450e-02,
-1.2409e-02, -3.2288e-01]],
...,
[[ 9.9947e-01, 2.9659e-01, 2.2655e-01, ..., -3.1833e-01,
1.0196e-01, -9.8482e-01],
[-9.0998e-01, -8.6718e-01, -6.8815e-01, ..., -1.5483e+00,
-1.6661e+00, -1.1935e+00],
[-4.1435e-01, -3.3241e-01, -2.5045e-01, ..., -4.6568e-01,
-5.1613e-01, -6.2434e-01],
...,
[-6.0325e-01, -5.9673e-01, -3.2381e-01, ..., -3.4642e-01,
-5.6465e-01, -1.1879e+00],
[-3.0439e+00, -2.8875e+00, -2.0485e+00, ..., -1.9317e+00,
-2.0415e+00, -2.0642e+00],
[-5.3031e-01, -1.5896e+00, -1.0759e+00, ..., -8.6252e-01,
-9.5882e-01, -1.3575e+00]],
[[ 1.0037e-01, 3.4142e-01, 6.8527e-01, ..., 1.0668e+00,
1.0828e+00, 4.5648e-01],
[ 1.5559e-01, -3.3749e-01, 1.6907e-01, ..., 4.5860e-01,
5.8540e-01, -2.1182e-01],
[-1.5028e-01, -4.4959e-01, 2.1830e-01, ..., 8.9943e-02,
3.2163e-01, 3.7062e-02],
...,
[-9.4736e-02, -9.9161e-01, 5.5256e-02, ..., 2.5396e-01,
1.5320e-01, -6.8983e-01],
[-4.9261e-01, -1.8202e+00, -2.0527e-01, ..., 2.8817e-02,
-1.8912e-02, -9.0990e-01],
[ 4.5736e-01, -1.7578e-01, 1.1168e-01, ..., 2.4354e-01,
2.7550e-01, 3.1527e-01]],
[[-8.7680e-01, -2.1084e-01, 1.3212e-01, ..., 2.5346e-01,
6.1845e-02, 7.7017e-01],
[-1.0729e+00, -2.5763e-01, -3.1230e-01, ..., 1.2729e-01,
-9.7926e-02, 1.2243e+00],
[-6.8164e-01, -1.7883e-01, -2.1838e-01, ..., 3.6148e-02,
-1.2414e-01, 6.8311e-01],
...,
[-1.0569e+00, -9.1550e-01, -4.3830e-01, ..., -3.9785e-01,
-6.1474e-02, 8.3459e-01],
[-1.7854e+00, -1.7894e+00, -9.7422e-01, ..., -8.1826e-01,
-4.8796e-01, 4.4316e-01],
[-9.2328e-01, -9.4746e-01, -2.4181e-01, ..., -2.7418e-01,
-1.5757e-01, 1.1120e+00]]],
[[[-6.4689e-01, -1.3788e+00, -2.0163e+00, ..., -1.2657e+00,
-9.0365e-01, -1.7783e+00],
[-2.6965e-01, -2.5308e+00, -1.3138e+00, ..., -1.7839e+00,
-1.9007e+00, -1.9748e+00],
[-1.9794e-01, -1.9178e+00, -6.9466e-01, ..., -1.3616e+00,
-1.4739e+00, -2.1378e+00],
...,
[ 1.1466e-01, -6.1608e-01, -2.4018e-01, ..., -9.8446e+00,
-6.0898e+00, -9.2191e-01],
[-2.6849e-01, -4.6085e-01, 3.1664e-02, ..., -4.2299e+00,
-2.1592e+00, -4.1600e-02],
[-6.0908e-01, -1.2958e+00, -1.1048e+00, ..., -1.4103e+00,
-1.9021e+00, -8.3679e-01]],
[[-2.5669e+00, -6.0527e+00, -7.6988e+00, ..., -4.8530e+00,
-6.1483e+00, -2.5003e+00],
[-5.7642e+00, -8.6272e+00, -7.2124e+00, ..., -7.8020e+00,
-9.2441e+00, -4.3939e+00],
[-4.2235e+00, -6.2290e+00, -5.4850e+00, ..., -6.7017e+00,
-6.5637e+00, -3.2547e+00],
...,
[-1.8450e+00, -2.3728e+00, -2.0732e+00, ..., -6.5677e+00,
-1.3694e+00, -1.3958e-01],
[-2.8777e+00, -3.0482e+00, -2.1081e+00, ..., -2.5542e+00,
-1.8264e+00, -3.2844e-01],
[ 3.7246e-01, -1.6972e-01, 3.5315e-01, ..., 4.0732e-01,
2.8594e-01, 5.7662e-02]],
[[-7.1481e-01, 3.7019e-01, 3.7280e-01, ..., 7.8750e-01,
6.2561e-01, 1.8912e+00],
[-9.3632e-01, 3.0120e-01, 1.0061e+00, ..., 5.6178e-01,
1.2304e-02, 9.8151e-02],
[-1.7173e+00, 2.3006e-01, 1.2400e+00, ..., 1.8992e-01,
1.0127e-02, -6.8008e-02],
...,
[-7.0050e-01, 1.1045e-01, 2.5630e-01, ..., -1.5525e+01,
-7.3520e+00, -1.2755e+00],
[-4.9880e-01, 1.3548e-01, 2.2670e-01, ..., -1.1788e+00,
1.1417e+00, 1.0046e+00],
[-1.5676e-01, -2.1543e-01, 1.4164e-01, ..., 1.0710e-01,
6.6718e-02, 5.3230e-04]],
...,
[[ 1.1043e+00, -1.0978e-01, 4.5452e-02, ..., 1.9964e-01,
-4.4520e-02, -2.1104e+00],
[-9.7521e-01, -8.2102e-01, 7.5190e-01, ..., -5.8061e-01,
-1.4884e+00, -2.0890e+00],
[-7.2745e-01, -1.1476e+00, -1.1883e+00, ..., -2.9569e-01,
-6.0279e-01, -1.4267e+00],
...,
[-6.3681e-01, -5.3872e-01, -3.5967e-01, ..., -9.1668e+00,
-9.9012e+00, -8.6066e+00],
[-2.1763e+00, -1.7421e+00, -1.2426e+00, ..., -5.0206e+00,
-6.1046e+00, -3.4483e+00],
[-6.1502e-01, -8.4526e-01, -4.7009e-01, ..., -1.5089e+00,
-1.3174e+00, -5.0360e-01]],
[[-9.7410e-02, 3.3745e-02, 1.3209e-01, ..., 5.6378e-01,
9.2693e-01, -1.5128e-02],
[-8.6165e-03, -1.1692e+00, -6.3752e-01, ..., 4.3438e-01,
3.7787e-01, -8.3497e-01],
[-3.4738e-03, -8.1323e-01, -4.6723e-04, ..., 2.7826e-01,
5.2879e-01, -3.6639e-01],
...,
[-1.9392e-01, -5.1301e-01, 2.3083e-01, ..., -6.1626e+00,
-2.0756e+00, -1.0267e+00],
[-4.8430e-01, -9.9601e-01, 5.1247e-02, ..., -2.1525e+00,
1.6657e+00, 2.1644e-01],
[ 3.6619e-01, -6.2384e-02, -1.8563e-01, ..., -1.1547e+00,
-1.1168e+00, -6.7359e-01]],
[[-1.7209e+00, -9.8966e-01, -7.3440e-01, ..., -5.1239e-01,
-3.2237e-01, 8.5016e-01],
[-2.1348e+00, -9.5077e-01, -1.1697e+00, ..., -8.2997e-01,
-2.1394e-01, 1.6576e+00],
[-1.5338e+00, -6.2218e-01, -7.7988e-01, ..., -2.8469e-01,
-1.6695e-01, 1.3866e+00],
...,
[-6.1056e-01, -2.5708e-01, -1.5370e-01, ..., -9.2191e+00,
-5.9182e+00, 2.4428e+00],
[-1.0940e+00, -6.1073e-01, -3.5951e-01, ..., 7.8257e-01,
-2.3802e-01, 1.2231e+00],
[-3.7994e-01, -6.6253e-02, -1.6009e-01, ..., 2.9860e-01,
2.1204e-01, 6.3341e-01]]],
[[[-2.0644e-01, -2.5885e-01, 1.9214e-01, ..., 6.3103e-02,
5.0689e-01, 1.5390e-01],
[-1.3150e-01, -1.5619e+00, -8.7474e-01, ..., -7.2106e-01,
-4.1640e-01, -3.2114e-01],
[-1.2651e-01, -1.3437e+00, -5.8968e-01, ..., -4.5149e-01,
-3.5201e-01, -6.6602e-01],
...,
[-1.4593e-01, -1.4985e+00, -1.2535e+00, ..., -1.0023e+00,
-9.7947e-01, -2.0528e+00],
[-8.0482e-01, -1.4535e+00, -8.2590e-01, ..., -5.2869e-01,
-4.4218e-01, -1.8115e+00],
[-9.9739e-01, -1.7883e+00, -1.4594e+00, ..., -1.5348e+00,
-1.6674e+00, -1.0602e+00]],
[[-1.7641e-01, -1.4388e+00, -5.7482e-01, ..., -5.4605e-01,
-1.2867e+00, -2.2760e-01],
[-2.0521e+00, -4.0554e+00, -2.3955e+00, ..., -2.3149e+00,
-3.2299e+00, -1.0929e+00],
[-1.8884e+00, -3.5096e+00, -3.0956e+00, ..., -2.5025e+00,
-2.9283e+00, -1.2108e+00],
...,
[-3.7171e+00, -5.3800e+00, -5.2589e+00, ..., -5.5589e+00,
-6.2223e+00, -2.9250e+00],
[-5.8545e+00, -6.6192e+00, -6.0461e+00, ..., -6.4699e+00,
-7.7735e+00, -2.8332e+00],
[ 4.6915e-02, -5.5630e-01, 3.0724e-01, ..., 3.9251e-01,
7.3032e-01, 1.7709e-01]],
[[-5.3818e-01, -1.8866e-01, 3.3931e-01, ..., 3.4177e-01,
3.5741e-02, 8.6019e-01],
[-3.5681e-01, -4.7174e-01, 4.7203e-01, ..., 3.6627e-01,
2.6981e-01, 4.9940e-01],
[-8.7815e-01, -2.3973e-01, 1.7078e-01, ..., 6.1805e-02,
1.9116e-01, 4.1605e-01],
...,
[-1.3302e+00, 9.9788e-02, 6.9312e-03, ..., 1.6295e-02,
5.6950e-03, 2.4982e-01],
[-1.2497e+00, 1.0877e-01, -2.4877e-02, ..., -1.8209e-02,
-2.9054e-03, 2.9367e-01],
[-1.0967e+00, -4.7059e-02, 5.9414e-02, ..., 2.5217e-02,
-6.0065e-02, -3.3535e-01]],
...,
[[ 5.5370e-02, -6.3052e-01, -4.0989e-01, ..., -7.8491e-02,
-2.2841e-01, -7.2053e-01],
[-8.8561e-01, -6.3026e-01, -5.4476e-01, ..., -2.7763e-01,
-9.3578e-01, -7.0937e-01],
[-5.1953e-01, -1.0919e-01, -1.6805e-02, ..., 1.2579e-01,
-2.0471e-01, -5.1559e-01],
...,
[-4.0109e-01, -6.9596e-01, -3.6939e-01, ..., -3.8442e-01,
-6.1760e-01, -1.2023e+00],
[-2.6241e+00, -2.7722e+00, -1.9360e+00, ..., -2.0013e+00,
-2.1355e+00, -2.1245e+00],
[-5.1688e-01, -1.5666e+00, -9.9482e-01, ..., -9.1128e-01,
-9.7116e-01, -1.4112e+00]],
[[-1.9263e-01, 3.8253e-02, 4.3634e-01, ..., 1.0678e-01,
5.3144e-01, 1.7857e-01],
[-4.5331e-01, -2.7148e-01, -3.1449e-03, ..., -8.7828e-02,
7.5034e-02, -2.2183e-02],
[-6.8132e-01, -5.7929e-01, 1.2021e-01, ..., 3.8548e-02,
1.3644e-02, -2.7541e-02],
...,
[ 6.8765e-02, -8.9377e-01, 6.7234e-02, ..., 1.7162e-01,
1.1536e-01, -7.4362e-01],
[-3.1841e-01, -1.6878e+00, -1.8806e-01, ..., -9.7105e-02,
-1.2632e-01, -1.0345e+00],
[ 5.5594e-01, -4.8206e-02, 1.9789e-01, ..., 1.9351e-01,
2.6322e-01, 2.8440e-01]],
[[-7.0111e-01, 4.2026e-01, 5.6652e-01, ..., 4.7129e-01,
3.6813e-01, 5.7583e-01],
[-8.6027e-01, 2.9776e-01, -2.3198e-01, ..., -2.6030e-01,
-1.9516e-01, 6.2997e-01],
[-4.5524e-01, 3.7374e-01, 3.8314e-02, ..., 7.0173e-02,
7.3169e-02, 5.9211e-01],
...,
[-1.0891e+00, -8.6112e-01, -4.1428e-01, ..., -3.8092e-01,
-6.8064e-02, 8.1355e-01],
[-1.7359e+00, -1.5482e+00, -7.9412e-01, ..., -7.7956e-01,
-4.4821e-01, 3.7976e-01],
[-9.3370e-01, -8.2528e-01, -1.5846e-01, ..., -3.2564e-01,
-1.8282e-01, 1.1127e+00]]],
[[[-2.0772e-01, -3.2632e-01, 1.1754e-01, ..., 1.9424e-02,
4.5824e-01, 1.0229e-01],
[-1.4262e-01, -1.5849e+00, -9.0706e-01, ..., -7.5578e-01,
-4.8696e-01, -3.8667e-01],
[-1.2975e-01, -1.3981e+00, -6.8997e-01, ..., -5.1561e-01,
-4.1858e-01, -7.5237e-01],
...,
[-1.9950e-01, -1.9742e+00, -1.4945e+00, ..., -1.1402e-01,
2.0090e-01, -1.8074e+00],
[-4.6297e-01, -1.5735e+00, -7.8116e-01, ..., 1.1334e-01,
2.6662e+00, -5.1101e-01],
[-2.3395e-01, -1.3344e+00, -1.1491e+00, ..., 1.5447e-02,
-7.6210e-01, -1.6037e+00]],
[[-3.1553e-01, -1.6111e+00, -8.2025e-01, ..., -8.4924e-01,
-1.5378e+00, -3.4234e-01],
[-2.1373e+00, -4.1455e+00, -2.5916e+00, ..., -2.5703e+00,
-3.5766e+00, -1.2397e+00],
[-1.9988e+00, -3.5703e+00, -2.8978e+00, ..., -2.7554e+00,
-3.1918e+00, -1.3283e+00],
...,
[-3.7813e+00, -5.2657e+00, -5.4579e+00, ..., -1.4023e+01,
-1.1849e+01, -6.6347e+00],
[-5.6448e+00, -6.8057e+00, -6.0188e+00, ..., -1.1800e+01,
-1.4919e+01, -7.8490e+00],
[ 5.2164e-01, -5.3865e-01, 5.1761e-01, ..., -5.5634e-01,
-1.9718e+00, -2.7591e+00]],
[[-5.3464e-01, -1.2877e-01, 4.2423e-01, ..., 4.1716e-01,
1.2848e-01, 9.2982e-01],
[-3.9249e-01, -4.7164e-01, 4.5112e-01, ..., 3.5303e-01,
2.9149e-01, 4.9442e-01],
[-9.1103e-01, -2.5172e-01, 1.1324e-01, ..., 6.7150e-02,
1.7173e-01, 4.1256e-01],
...,
[-1.7172e+00, -3.9606e-01, -1.7849e-01, ..., -4.9670e+00,
-3.7624e+00, 8.7169e-02],
[-1.4804e+00, -2.9910e-01, -1.3424e-01, ..., -3.6543e+00,
-4.1914e+00, -2.5012e-01],
[-6.4171e-01, -1.4017e-01, 9.9065e-02, ..., -5.4528e+00,
-3.5804e+00, -2.0605e+00]],
...,
[[ 8.0185e-02, -5.7163e-01, -3.3883e-01, ..., -3.2172e-02,
-1.7989e-01, -7.2002e-01],
[-8.5205e-01, -5.8878e-01, -5.1005e-01, ..., -2.5214e-01,
-9.8874e-01, -7.7702e-01],
[-5.4662e-01, -1.3979e-01, -2.7804e-02, ..., 9.0285e-02,
-1.8375e-01, -5.3950e-01],
...,
[-1.0307e+00, -9.2544e-01, -3.3090e-01, ..., -1.0279e+01,
-6.6672e+00, -4.2978e+00],
[-2.6170e+00, -2.0545e+00, -1.3818e+00, ..., -1.2491e+01,
-1.0250e+01, -6.5463e+00],
[-5.5493e-01, -1.1592e+00, -1.8719e-01, ..., -1.0826e+01,
-9.2543e+00, -5.1457e+00]],
[[-1.7584e-01, 3.8311e-04, 3.8638e-01, ..., 8.7560e-02,
5.4114e-01, 1.3934e-01],
[-4.4724e-01, -2.7078e-01, 2.7955e-02, ..., -4.4325e-02,
1.5323e-01, -9.9752e-02],
[-6.6447e-01, -5.6304e-01, 2.7669e-02, ..., 7.3513e-02,
4.5181e-02, -6.0123e-02],
...,
[-1.6122e-01, -8.5884e-01, 3.6712e-01, ..., -2.5325e+00,
-2.5432e-01, -7.8816e-01],
[-7.7111e-01, -2.0011e+00, -1.1315e-01, ..., 3.4454e-02,
2.1302e+00, 8.6238e-01],
[ 7.6489e-01, 4.3843e-02, 3.9036e-01, ..., 1.7415e+00,
2.6389e+00, 1.7012e+00]],
[[-6.4910e-01, 4.1052e-01, 5.2663e-01, ..., 4.7643e-01,
3.8510e-01, 6.3609e-01],
[-8.8954e-01, 2.6769e-01, -1.9159e-01, ..., -2.5828e-01,
-1.4962e-01, 6.9965e-01],
[-5.0615e-01, 2.8858e-01, 2.9223e-02, ..., 7.4637e-02,
1.2959e-01, 6.3463e-01],
...,
[-1.4037e+00, -7.6785e-01, -2.1501e-01, ..., -4.9079e+00,
-1.3272e+00, 3.2150e+00],
[-1.4870e+00, -9.1621e-01, -2.2673e-01, ..., -9.9470e+00,
-4.3962e+00, 1.9428e+00],
[-7.0254e-01, -4.9796e-01, 1.4576e-01, ..., -7.0956e+00,
-2.2380e+00, 2.2173e+00]]]], device='cuda:0'),
tensor([[[[-7.3811e-01, -1.4916e+00, -1.4753e+00, ..., -1.3324e+00,
-8.8557e-01, -1.6787e+00],
[-3.9904e-01, -2.3734e+00, -2.2298e+00, ..., -3.2870e+00,
-4.0502e+00, -3.1854e+00],
[-1.4159e-01, -1.8299e+00, -1.6125e+00, ..., -7.6062e-01,
-5.7007e-01, 1.4432e+00],
...,
[ 1.9265e+00, -2.0661e+00, -6.4786e+00, ..., 2.8346e-01,
-2.3268e-01, -8.0784e-01],
[ 1.3031e+00, 1.7117e+00, -3.1893e+00, ..., 1.9602e-01,
1.7131e-01, -4.1848e-01],
[ 2.3410e-02, -1.3472e-01, -1.8420e+00, ..., -1.3370e+00,
-1.1849e+00, -7.0129e-01]],
[[-2.8950e+00, -5.3015e+00, -4.7670e+00, ..., -5.1044e+00,
-6.1387e+00, -2.4244e+00],
[-5.3710e+00, -8.3364e+00, -7.9885e+00, ..., -1.3629e+01,
-1.4705e+01, -6.4028e+00],
[-4.1099e+00, -5.9999e+00, -6.6688e+00, ..., -5.7485e+00,
-6.9675e+00, -2.4894e+00],
...,
[-5.3231e+00, -9.5834e+00, -1.1833e+01, ..., -2.6830e+00,
-2.5699e+00, -1.3499e+00],
[-3.2164e+00, -4.4032e+00, -4.8488e+00, ..., -2.3314e+00,
-3.0225e+00, -9.4696e-01],
[-2.5092e-01, -6.6358e-01, 1.1772e-02, ..., -2.5590e-01,
2.9945e-01, 1.3724e-01]],
[[-6.5490e-01, 5.2775e-01, 8.7767e-01, ..., 9.5207e-01,
6.4621e-01, 1.9280e+00],
[-9.5692e-01, 7.5847e-02, 5.1954e-01, ..., 1.5132e+00,
9.2258e-01, 1.0062e+00],
[-1.7932e+00, -3.0141e-01, -3.8717e-02, ..., -3.3938e+00,
-3.7042e+00, -1.5608e+00],
...,
[-2.4016e+00, -1.4592e+01, -1.0802e+01, ..., 2.8126e-02,
-9.9421e-02, 6.2815e-01],
[ 5.3647e-01, -7.5421e+00, -6.9898e+00, ..., -3.2137e-01,
-4.0687e-01, 4.9015e-01],
[ 1.0323e+00, -1.4885e+00, -1.3446e+00, ..., -4.1280e-01,
-5.7119e-01, 8.2615e-02]],
...,
[[ 1.3205e+00, -3.2675e-01, -7.4487e-02, ..., 1.2000e-01,
-1.3898e-01, -2.1438e+00],
[-5.6427e-01, -8.5746e-01, -5.2265e-01, ..., -1.6062e+00,
-2.4327e+00, -3.0067e+00],
[-7.3029e-01, -9.0336e-01, -2.2360e-01, ..., -1.3168e+00,
-1.8060e+00, -4.9882e+00],
...,
[ 1.3466e+00, 1.0592e+00, -6.3371e+00, ..., -6.1665e-01,
-9.6560e-01, -9.3928e-01],
[-2.9976e+00, -1.6765e+00, -4.5021e+00, ..., -1.2303e+00,
-1.3216e+00, -1.1895e+00],
[-1.8859e+00, -2.6090e+00, -1.7633e+00, ..., -6.8375e-01,
-7.1117e-01, -3.5846e-01]],
[[ 1.2081e-01, -6.6550e-03, 3.9768e-01, ..., 3.7673e-01,
9.0281e-01, -4.2831e-03],
[ 3.4738e-01, -7.2533e-01, 9.5717e-03, ..., 6.4844e-01,
-1.1988e-01, -1.5954e+00],
[ 3.1582e-01, -5.0687e-01, 4.7070e-01, ..., 1.9511e+00,
1.3193e+00, 9.0440e-01],
...,
[-2.5156e-01, -4.2205e+00, -4.7451e+00, ..., 1.6513e-01,
2.9437e-01, -9.8282e-02],
[-7.4061e-01, -2.7498e+00, -4.2115e+00, ..., -2.9757e-01,
-7.2108e-03, -1.3664e-01],
[ 6.2824e-01, -1.0017e-01, -1.7875e+00, ..., -5.0818e-01,
-1.1643e-01, 7.1725e-02]],
[[-1.5101e+00, -7.8558e-01, -4.6267e-01, ..., -5.1502e-01,
-2.8364e-01, 8.8873e-01],
[-2.1683e+00, -7.3378e-01, -5.4427e-01, ..., -2.2237e+00,
-1.1391e+00, 1.1536e+00],
[-1.5707e+00, -5.3912e-01, -2.6081e-01, ..., -4.7883e+00,
-4.7921e+00, -2.4621e+00],
...,
[-5.8094e-03, 1.0529e-01, -6.2572e+00, ..., -8.7111e-01,
1.9925e-02, 7.0151e-01],
[-1.8509e+00, 1.3019e+00, -2.5068e-01, ..., -7.7903e-01,
3.3990e-02, 4.2368e-01],
[-7.6797e-01, 7.9496e-01, 1.5941e+00, ..., -4.1535e-01,
4.2664e-02, 5.2102e-01]]],
[[[-8.2801e-01, -1.3919e+00, -1.3940e+00, ..., 7.8056e-03,
6.1287e-01, 5.5119e-01],
[-3.9947e-01, -2.2854e+00, -2.1406e+00, ..., -4.9561e-01,
-2.5783e-01, 1.2275e-01],
[-3.9860e-01, -1.9587e+00, -1.8169e+00, ..., -5.0577e-01,
-1.7331e-01, -9.2291e-02],
...,
[-4.4038e-02, -1.8617e+00, -1.7927e+00, ..., -3.2494e-01,
-6.8978e-02, -1.3222e-01],
[-1.2174e+00, -1.8564e+00, -1.0890e+00, ..., 2.6507e-02,
2.5942e-01, -1.2335e-02],
[-1.0719e+00, -1.9434e+00, -1.4390e+00, ..., -1.0202e+00,
-1.0594e+00, -3.8428e-01]],
[[-2.5150e+00, -4.9345e+00, -4.3312e+00, ..., -4.3375e-01,
-1.0140e+00, -7.5628e-05],
[-5.2047e+00, -7.8442e+00, -6.6068e+00, ..., -1.8302e+00,
-2.6730e+00, -6.6639e-01],
[-4.3544e+00, -6.1358e+00, -6.7896e+00, ..., -1.9424e+00,
-1.5739e+00, -3.7777e-01],
...,
[-4.5360e+00, -6.4653e+00, -5.7427e+00, ..., -1.2974e+00,
-1.5757e+00, -3.6085e-01],
[-6.5570e+00, -7.7611e+00, -6.5229e+00, ..., -2.4205e+00,
-3.2901e+00, -9.5235e-01],
[-7.1737e-02, -1.0445e+00, 4.3982e-01, ..., 3.4041e-01,
3.6216e-01, 9.0093e-02]],
[[-5.6889e-01, 7.3069e-01, 1.0417e+00, ..., 8.1727e-01,
2.9713e-01, 1.0952e+00],
[-9.5203e-01, 1.3617e-01, 3.5273e-01, ..., 4.0770e-01,
4.9697e-02, 4.7643e-01],
[-1.6884e+00, -4.5080e-02, -2.0779e-01, ..., 1.1861e-01,
1.0757e-01, 4.1613e-01],
...,
[-1.3979e+00, -5.4339e-01, -3.6394e-01, ..., 4.6608e-02,
6.9704e-02, 4.1460e-01],
[-1.4543e+00, -3.0509e-02, 6.4241e-03, ..., 1.3260e-01,
1.1900e-01, 6.0761e-01],
[-1.3230e+00, -5.8812e-01, -3.2444e-01, ..., -8.4911e-02,
6.5335e-02, 3.5683e-01]],
...,
[[ 1.2047e+00, -3.4659e-01, -1.8265e-01, ..., -1.1276e-01,
-3.9606e-01, -9.3826e-01],
[-6.1309e-01, -9.8550e-01, -7.9619e-01, ..., -6.8049e-01,
-1.2894e+00, -8.3689e-01],
[-6.0679e-01, -1.0406e+00, -5.9539e-01, ..., -4.8849e-02,
-2.0544e-01, -3.8176e-01],
...,
[-1.3942e+00, -1.4435e+00, -1.1644e+00, ..., -9.8173e-02,
-1.8187e-01, -3.6404e-01],
[-3.1343e+00, -2.9804e+00, -2.0076e+00, ..., -8.1828e-01,
-8.7357e-01, -7.9371e-01],
[-7.0224e-01, -1.7449e+00, -1.0123e+00, ..., -2.0242e-02,
-1.2132e-01, -3.0476e-01]],
[[ 1.2346e-01, 9.9441e-02, 5.8893e-01, ..., 3.3103e-01,
6.8075e-01, 2.8489e-01],
[ 1.9672e-01, -7.0360e-01, 1.1529e-01, ..., -3.8292e-02,
1.0760e-01, -5.8243e-02],
[ 9.2442e-02, -6.7834e-01, 5.6122e-01, ..., -5.4339e-01,
-1.3739e-01, 1.7848e-01],
...,
[-2.0565e-01, -9.2096e-01, 1.8686e-01, ..., -1.3490e-01,
-9.9954e-02, 1.6316e-01],
[-4.5358e-01, -1.5230e+00, 4.1401e-02, ..., -3.9140e-01,
-3.0569e-01, -3.8831e-02],
[ 5.7749e-01, 1.6243e-01, 2.3263e-01, ..., -2.3730e-01,
-2.0188e-01, 3.2797e-01]],
[[-1.5049e+00, -5.9758e-01, -1.0746e-02, ..., 5.6804e-01,
3.7912e-01, 5.4825e-01],
[-2.1232e+00, -3.6344e-01, -4.0382e-02, ..., -6.6124e-02,
-1.3981e-01, 4.1581e-01],
[-1.8057e+00, -3.4077e-01, 1.2138e-01, ..., 2.0142e-01,
-1.6169e-01, 7.9790e-02],
...,
[-1.3812e+00, -1.5727e-01, 6.4699e-01, ..., 4.2574e-02,
-9.6261e-02, 1.1013e-01],
[-2.3700e+00, -1.3466e+00, -1.5757e-01, ..., -1.2182e-01,
-2.6232e-01, -1.8958e-01],
[-1.0923e+00, -4.1380e-01, 3.0412e-01, ..., -1.0626e-01,
-2.4908e-01, 1.2589e-01]]],
[[[-8.0305e-01, -1.5487e+00, -1.5066e+00, ..., -1.2653e+00,
-8.9753e-01, -1.7783e+00],
[-4.7177e-01, -2.5178e+00, -2.3859e+00, ..., -2.8902e+00,
-2.3971e+00, -2.0131e+00],
[-2.4542e-01, -2.0127e+00, -9.9468e-01, ..., -1.3913e+00,
-1.6744e+00, -2.2376e+00],
...,
[ 9.8912e-02, -7.1699e-01, -3.0778e-01, ..., -1.6797e-01,
-5.3386e-01, -3.4105e-01],
[-2.7636e-01, -5.5204e-01, -3.2564e-02, ..., -5.0335e-01,
-7.6807e-01, 4.1668e-01],
[-5.4127e-01, -1.3241e+00, -1.1076e+00, ..., -5.4085e+00,
-2.9288e+00, 9.5791e-01]],
[[-2.8101e+00, -5.2391e+00, -4.6200e+00, ..., -4.7357e+00,
-6.0904e+00, -2.4175e+00],
[-5.5623e+00, -8.4319e+00, -7.7944e+00, ..., -1.1904e+01,
-1.1543e+01, -4.5076e+00],
[-4.3268e+00, -6.2101e+00, -1.1372e+01, ..., -7.4427e+00,
-8.8679e+00, -3.3187e+00],
...,
[-1.9948e+00, -2.5994e+00, -2.2280e+00, ..., -5.7590e+00,
-7.6537e+00, -2.7812e+00],
[-3.0582e+00, -3.2843e+00, -2.2780e+00, ..., -8.6776e+00,
-9.2831e+00, -2.8022e+00],
[ 4.0041e-01, -2.0289e-01, 4.1143e-01, ..., 2.2905e+00,
2.0376e+00, 3.0691e-02]],
[[-6.6960e-01, 5.2400e-01, 8.8081e-01, ..., 8.4849e-01,
6.2544e-01, 1.9468e+00],
[-9.2458e-01, 1.7273e-01, 4.2554e-01, ..., 1.3201e+00,
1.5376e-01, 7.4586e-02],
[-1.8180e+00, -1.5370e-01, -5.4358e-01, ..., -8.7375e-01,
8.3195e-01, -1.0493e-01],
...,
[-7.7594e-01, 1.3515e-01, 2.6190e-01, ..., -1.2444e+01,
-5.5633e+00, -7.2307e-01],
[-5.2785e-01, 1.1501e-01, 2.1806e-01, ..., -1.2047e+01,
-6.9531e+00, -3.4435e+00],
[-1.3938e-01, -1.5433e-01, 1.1115e-01, ..., -5.2173e+00,
-2.6393e+00, -2.2196e+00]],
...,
[[ 1.3722e+00, -3.1171e-01, -4.5875e-02, ..., 2.4434e-01,
-8.8841e-02, -2.2162e+00],
[-6.7608e-01, -1.0108e+00, -6.1051e-01, ..., -1.7894e-01,
-9.1383e-01, -2.1935e+00],
[-7.1827e-01, -9.4874e-01, -2.6122e-01, ..., -3.4958e+00,
-2.1756e+00, -1.5040e+00],
...,
[-6.7099e-01, -5.2079e-01, -3.5700e-01, ..., -7.3717e+00,
-5.5625e+00, -2.7453e+00],
[-2.2613e+00, -1.7975e+00, -1.2906e+00, ..., -1.0609e+01,
-9.2957e+00, -6.5514e+00],
[-6.4375e-01, -8.6242e-01, -4.5261e-01, ..., -2.8882e+00,
-4.8680e+00, -5.0424e+00]],
[[ 7.7288e-02, 2.5727e-02, 5.4305e-01, ..., 6.2332e-01,
1.0305e+00, 4.4509e-02],
[ 3.1176e-01, -8.4012e-01, -3.1871e-02, ..., -1.4385e+00,
-1.3798e+00, -8.7339e-01],
[ 2.7336e-01, -6.6535e-01, 1.1214e+00, ..., 1.1562e+00,
-8.0007e-01, -4.1998e-01],
...,
[-1.7892e-01, -5.6851e-01, 2.3516e-01, ..., -2.2360e+00,
-1.2256e+00, -4.9361e-01],
[-5.0348e-01, -1.0522e+00, 1.3491e-02, ..., -5.0161e+00,
-4.7220e+00, -1.5346e+00],
[ 3.9691e-01, -8.8041e-02, -1.6616e-01, ..., -2.0605e+00,
-1.8199e+00, 6.3726e-01]],
[[-1.6817e+00, -9.1766e-01, -5.2955e-01, ..., -5.4463e-01,
-3.3270e-01, 8.6865e-01],
[-2.3207e+00, -8.3541e-01, -6.2634e-01, ..., -1.5709e+00,
5.9752e-01, 1.6958e+00],
[-1.7096e+00, -6.4513e-01, -5.0351e-01, ..., -3.0535e+00,
8.3447e-01, 1.3920e+00],
...,
[-7.2917e-01, -3.4752e-01, -2.1330e-01, ..., -6.2978e+00,
-2.3253e+00, -2.5140e-01],
[-1.1454e+00, -6.6349e-01, -3.7773e-01, ..., -8.4932e+00,
-5.1194e+00, -8.6417e-01],
[-3.6651e-01, -1.1230e-01, -1.4920e-01, ..., -6.7553e+00,
-5.5243e+00, -8.8766e-01]]],
[[[-7.3638e-01, -1.4206e+00, -1.4360e+00, ..., -3.0019e-01,
6.0819e-01, 5.1721e-01],
[-3.1239e-01, -2.4126e+00, -2.2161e+00, ..., -6.7982e-01,
-2.3289e-01, 1.1469e-01],
[-3.9998e-01, -2.1385e+00, -1.8083e+00, ..., -5.4580e-02,
-1.5853e-01, -1.0896e-01],
...,
[-5.0730e-01, -1.9721e+00, -1.9031e+00, ..., -3.8587e-01,
-1.0268e-01, -1.5122e-01],
[-9.5139e-01, -1.7456e+00, -1.2042e+00, ..., 1.8018e-02,
2.3729e-01, -4.0513e-02],
[-7.7630e-01, -1.6551e+00, -1.2560e+00, ..., -1.0107e+00,
-1.0783e+00, -3.6910e-01]],
[[-2.3849e+00, -4.8561e+00, -4.1424e+00, ..., -1.2395e+00,
-1.0607e+00, -9.9696e-03],
[-5.4580e+00, -8.2677e+00, -6.6088e+00, ..., -3.9283e+00,
-2.7479e+00, -7.0960e-01],
[-4.4471e+00, -6.4080e+00, -6.2449e+00, ..., -5.1450e+00,
-1.6213e+00, -4.1877e-01],
...,
[-4.4297e+00, -6.2176e+00, -5.5888e+00, ..., -1.4133e+00,
-1.6580e+00, -4.0414e-01],
[-6.4490e+00, -7.5034e+00, -6.4468e+00, ..., -2.5000e+00,
-3.3715e+00, -9.9602e-01],
[-2.3481e-02, -8.0551e-01, 3.6438e-01, ..., 3.7046e-01,
3.8383e-01, 8.7970e-02]],
[[-5.8331e-01, 6.0738e-01, 1.1061e+00, ..., 9.4112e-01,
3.4852e-01, 1.1218e+00],
[-8.6648e-01, 6.5151e-03, 4.2420e-01, ..., 9.3159e-01,
8.6267e-02, 4.6973e-01],
[-1.7059e+00, -7.6098e-02, -1.0463e-01, ..., 1.2771e+00,
1.3734e-01, 4.1404e-01],
...,
[-1.7210e+00, -7.7599e-02, -9.5328e-02, ..., 6.0044e-02,
8.7684e-02, 4.2849e-01],
[-1.5403e+00, 9.6343e-02, 6.4809e-02, ..., 1.7346e-01,
1.3247e-01, 5.9632e-01],
[-1.4542e+00, -5.4149e-01, -2.6824e-01, ..., -6.2294e-02,
8.3161e-02, 3.5558e-01]],
...,
[[ 1.0759e+00, -4.3001e-01, -2.2695e-01, ..., -5.8734e-02,
-3.7542e-01, -9.2986e-01],
[-8.1313e-01, -9.8808e-01, -8.3255e-01, ..., -1.0875e+00,
-1.2948e+00, -8.3262e-01],
[-6.3833e-01, -9.5359e-01, -7.1917e-01, ..., -1.4462e+00,
-2.2584e-01, -4.1053e-01],
...,
[-7.3779e-01, -1.1595e+00, -7.2771e-01, ..., -8.0806e-02,
-1.9873e-01, -3.8539e-01],
[-2.7159e+00, -2.8313e+00, -1.8220e+00, ..., -8.1921e-01,
-8.9502e-01, -8.1275e-01],
[-6.4955e-01, -1.6567e+00, -1.0099e+00, ..., 2.1702e-02,
-1.1921e-01, -3.3389e-01]],
[[-9.4573e-04, -3.3079e-02, 5.9252e-01, ..., -4.5442e-01,
6.9436e-01, 3.0183e-01],
[ 1.5654e-02, -7.5781e-01, 9.7291e-02, ..., -9.8080e-01,
1.1409e-01, -5.9798e-02],
[-2.3669e-02, -8.8035e-01, 3.3951e-01, ..., -7.7369e-01,
-1.5819e-01, 1.6473e-01],
...,
[ 6.0941e-02, -8.4722e-01, 2.7879e-01, ..., -1.5888e-01,
-1.3867e-01, 1.6043e-01],
[-2.0870e-01, -1.4456e+00, 2.3605e-02, ..., -3.8348e-01,
-3.2290e-01, -6.5258e-02],
[ 6.8259e-01, 1.3116e-01, 2.8415e-01, ..., -2.0810e-01,
-2.1695e-01, 3.1854e-01]],
[[-1.6458e+00, -6.1834e-01, 1.4914e-02, ..., 5.8165e-01,
3.7925e-01, 5.8527e-01],
[-2.1235e+00, -3.0185e-01, -4.6244e-02, ..., -9.6094e-03,
-1.3494e-01, 4.4740e-01],
[-1.7589e+00, -3.1178e-01, 2.2792e-01, ..., 5.1748e-01,
-1.6892e-01, 8.4370e-02],
...,
[-1.7957e+00, -5.1589e-01, 2.5916e-01, ..., 1.4489e-01,
-9.3537e-02, 1.1227e-01],
[-2.1797e+00, -1.1122e+00, -1.5712e-01, ..., -1.0065e-01,
-2.6955e-01, -1.8919e-01],
[-1.0806e+00, -1.6646e-01, 4.6564e-01, ..., -9.3710e-02,
-2.5576e-01, 1.3729e-01]]],
[[[-2.2010e-01, -2.7539e-01, 2.1597e-01, ..., -1.0620e-02,
4.8337e-01, 1.6121e-01],
[-1.1821e-01, -1.5350e+00, -8.7377e-01, ..., -6.9753e-01,
-4.1914e-01, -3.0869e-01],
[-1.0826e-01, -1.3189e+00, -6.0999e-01, ..., -5.2766e-01,
-4.1671e-01, -6.5252e-01],
...,
[-1.7152e-01, -1.5366e+00, -1.2811e+00, ..., -1.0055e+00,
-9.8140e-01, -2.0640e+00],
[-7.6272e-01, -1.4606e+00, -8.1335e-01, ..., -5.5751e-01,
-4.4123e-01, -1.8133e+00],
[-9.6865e-01, -1.8324e+00, -1.5017e+00, ..., -1.5366e+00,
-1.6826e+00, -1.0624e+00]],
[[-2.0236e-01, -1.3923e+00, -5.8758e-01, ..., -6.8761e-01,
-1.3936e+00, -2.1443e-01],
[-2.0101e+00, -3.9853e+00, -2.4208e+00, ..., -2.4052e+00,
-3.3645e+00, -1.1465e+00],
[-1.8681e+00, -3.3524e+00, -2.9033e+00, ..., -2.5544e+00,
-2.9331e+00, -1.2011e+00],
...,
[-3.7243e+00, -5.4145e+00, -5.2766e+00, ..., -5.5029e+00,
-6.1368e+00, -2.8745e+00],
[-5.8720e+00, -6.5879e+00, -5.9734e+00, ..., -6.4783e+00,
-7.7532e+00, -2.8224e+00],
[ 5.6777e-02, -5.5054e-01, 3.0541e-01, ..., 3.4896e-01,
6.9475e-01, 1.6568e-01]],
[[-5.2568e-01, -1.5561e-01, 3.3867e-01, ..., 3.7268e-01,
5.3003e-02, 8.7479e-01],
[-3.7648e-01, -4.5916e-01, 4.9343e-01, ..., 4.4351e-01,
3.3496e-01, 5.0625e-01],
[-8.3849e-01, -2.2092e-01, 1.5110e-01, ..., 2.3257e-02,
1.8789e-01, 4.3060e-01],
...,
[-1.3604e+00, 1.0824e-01, 8.0256e-02, ..., 8.4444e-02,
3.7601e-02, 2.8474e-01],
[-1.2586e+00, 1.3500e-01, -5.3893e-02, ..., 1.8861e-02,
1.9703e-02, 2.8101e-01],
[-1.0822e+00, -2.1099e-02, 7.2517e-02, ..., 4.3951e-03,
-5.5713e-02, -3.5673e-01]],
...,
[[ 9.0798e-02, -5.9360e-01, -4.0345e-01, ..., -6.4509e-02,
-2.4294e-01, -7.3412e-01],
[-8.4099e-01, -6.5746e-01, -5.6074e-01, ..., -3.1085e-01,
-1.0094e+00, -7.3666e-01],
[-4.9445e-01, -1.2897e-01, -7.3877e-02, ..., 5.6503e-02,
-2.1661e-01, -5.1255e-01],
...,
[-4.5659e-01, -7.2820e-01, -4.2985e-01, ..., -3.7469e-01,
-6.2331e-01, -1.2321e+00],
[-2.6097e+00, -2.7694e+00, -1.9154e+00, ..., -2.0238e+00,
-2.1515e+00, -2.1225e+00],
[-5.4870e-01, -1.5857e+00, -1.0207e+00, ..., -8.4623e-01,
-9.5224e-01, -1.4025e+00]],
[[-1.9160e-01, 3.6812e-02, 4.4953e-01, ..., 3.8779e-02,
5.3054e-01, 1.9599e-01],
[-4.2791e-01, -2.7337e-01, 2.2439e-03, ..., -2.0333e-02,
1.4558e-01, -6.0027e-02],
[-6.5506e-01, -5.6292e-01, 4.1421e-02, ..., 3.3270e-02,
5.3591e-03, 2.2648e-03],
...,
[ 3.7252e-02, -9.2191e-01, 5.1543e-02, ..., 1.9038e-01,
1.1686e-01, -7.6225e-01],
[-3.0645e-01, -1.6575e+00, -1.8532e-01, ..., -4.9727e-02,
-1.1022e-01, -1.0334e+00],
[ 5.3310e-01, -9.3521e-02, 1.6310e-01, ..., 2.3162e-01,
2.5581e-01, 2.7851e-01]],
[[-6.7728e-01, 4.0329e-01, 5.5876e-01, ..., 4.7858e-01,
3.6846e-01, 5.8897e-01],
[-8.6956e-01, 2.7225e-01, -2.4998e-01, ..., -2.4409e-01,
-1.5551e-01, 6.4486e-01],
[-4.6990e-01, 3.4412e-01, 8.9113e-02, ..., 6.7363e-02,
1.0783e-01, 6.0540e-01],
...,
[-1.0847e+00, -8.8153e-01, -4.3028e-01, ..., -4.1349e-01,
-7.9682e-02, 8.2377e-01],
[-1.7002e+00, -1.5604e+00, -8.3130e-01, ..., -7.8909e-01,
-4.5181e-01, 4.0780e-01],
[-9.0511e-01, -7.6224e-01, -1.6884e-01, ..., -2.4420e-01,
-1.6006e-01, 1.1152e+00]]]], device='cuda:0'),
tensor([[[[-7.1695e-01, -1.4501e+00, -1.4553e+00, ..., -1.2279e+00,
-8.5996e-01, -1.6802e+00],
[-3.2334e-01, -2.4401e+00, -2.2668e+00, ..., -1.7828e+00,
-1.8905e+00, -1.9027e+00],
[-1.1786e-01, -1.9110e+00, -1.7072e+00, ..., -1.2511e+00,
-1.4003e+00, -2.1111e+00],
...,
[-8.4028e-02, -1.1237e+00, -6.0405e-01, ..., -5.3950e-01,
-4.6275e-01, -9.0684e-01],
[-1.2799e-01, -1.0006e+00, -3.3220e-01, ..., -2.3552e-01,
-8.0148e-02, -9.6440e-01],
[-1.4401e-01, -1.1061e+00, -8.2585e-01, ..., -8.9835e-01,
-1.1006e+00, -6.2625e-01]],
[[-2.6610e+00, -5.0993e+00, -4.6017e+00, ..., -4.7956e+00,
-6.0518e+00, -2.4438e+00],
[-5.3529e+00, -8.1758e+00, -7.5022e+00, ..., -7.8935e+00,
-9.2423e+00, -4.3864e+00],
[-4.0543e+00, -5.8430e+00, -5.8085e+00, ..., -7.0370e+00,
-6.4553e+00, -3.2050e+00],
...,
[-2.0081e+00, -3.1643e+00, -2.8439e+00, ..., -2.9737e+00,
-3.4184e+00, -1.4828e+00],
[-4.5027e+00, -5.4454e+00, -4.8140e+00, ..., -5.0851e+00,
-6.2180e+00, -2.3764e+00],
[ 6.9248e-01, -2.9398e-01, 5.5036e-01, ..., 4.8519e-01,
5.7092e-01, 2.7638e-03]],
[[-6.5591e-01, 4.8565e-01, 8.7367e-01, ..., 8.2313e-01,
6.1859e-01, 1.9019e+00],
[-9.8931e-01, 4.4274e-02, 4.2098e-01, ..., 5.6708e-01,
-9.8533e-03, 1.4617e-01],
[-1.7777e+00, -2.2529e-01, -5.8244e-02, ..., 3.2830e-01,
-5.0520e-02, -1.1982e-01],
...,
[-9.3675e-01, -1.0619e-01, 7.4427e-02, ..., 3.1891e-02,
8.1962e-02, 3.5729e-01],
[-9.6356e-01, 4.9097e-02, 6.0013e-02, ..., 1.0233e-02,
1.1434e-01, 4.0629e-01],
[-4.5781e-01, -1.7404e-02, 8.3819e-02, ..., -2.2258e-02,
9.4176e-02, 5.6716e-02]],
...,
[[ 1.2248e+00, -3.7297e-01, -7.3308e-02, ..., 2.3583e-01,
-1.1194e-01, -2.1351e+00],
[-7.3449e-01, -9.0013e-01, -4.8999e-01, ..., -5.1765e-01,
-1.4698e+00, -2.0749e+00],
[-7.0375e-01, -8.4661e-01, -2.7007e-01, ..., -1.2052e-01,
-5.7955e-01, -1.3910e+00],
...,
[-4.3637e-01, -1.2064e-01, 9.0612e-02, ..., 8.4504e-02,
-1.2557e-01, -5.4124e-01],
[-2.0475e+00, -1.3646e+00, -8.5354e-01, ..., -9.6408e-01,
-1.1266e+00, -1.1005e+00],
[-2.9944e-01, -7.5135e-01, -1.5380e-01, ..., -1.7241e-01,
-3.1643e-01, -6.1231e-01]],
[[ 3.8959e-02, 1.3733e-02, 4.3167e-01, ..., 4.9854e-01,
9.3359e-01, -6.2896e-02],
[ 2.2855e-01, -8.0142e-01, -6.7159e-02, ..., 3.2304e-01,
4.1243e-01, -8.2218e-01],
[ 1.6029e-01, -6.1725e-01, 3.0246e-01, ..., 2.2007e-01,
5.8016e-01, -3.6119e-01],
...,
[-3.9745e-01, -5.4062e-01, 7.8510e-02, ..., 8.9200e-02,
5.3413e-02, -1.0541e-01],
[-7.5047e-01, -1.5178e+00, -3.1708e-01, ..., -3.5130e-01,
-3.5894e-01, -6.8634e-01],
[ 5.1302e-01, 5.8206e-02, 1.9158e-01, ..., 1.8458e-01,
1.5753e-01, 3.4811e-01]],
[[-1.5309e+00, -7.8323e-01, -4.1001e-01, ..., -4.3882e-01,
-2.6831e-01, 8.7519e-01],
[-2.0929e+00, -7.3966e-01, -5.3252e-01, ..., -8.0452e-01,
-1.9411e-01, 1.6534e+00],
[-1.4754e+00, -5.3901e-01, -1.8809e-01, ..., -3.7998e-01,
-1.0003e-01, 1.4117e+00],
...,
[-4.5866e-01, 1.3357e-01, 1.1808e-01, ..., 1.4014e-01,
2.2067e-01, 7.0836e-01],
[-8.9419e-01, -2.9182e-01, -1.3121e-01, ..., -8.1365e-02,
6.6166e-02, 4.2153e-01],
[-4.1730e-01, -3.0537e-01, -1.4900e-01, ..., -2.3211e-01,
-2.5275e-01, 5.8871e-01]]],
[[[-7.3186e-01, -1.4297e+00, -1.4084e+00, ..., -4.3904e-02,
5.8971e-01, 5.5072e-01],
[-2.7995e-01, -2.2423e+00, -2.3339e+00, ..., -5.1139e-01,
-3.0089e-01, 1.2026e-01],
[-3.8035e-01, -1.7560e+00, -2.8264e+00, ..., -3.4917e-01,
-1.5414e-01, -9.5688e-02],
...,
[-4.8901e-01, -1.9490e+00, -1.8545e+00, ..., -3.1422e-01,
-6.2144e-02, -1.4807e-01],
[-9.6832e-01, -1.6883e+00, -1.1644e+00, ..., 3.4102e-02,
2.7035e-01, -3.2740e-02],
[-7.9909e-01, -1.6676e+00, -1.2809e+00, ..., -1.0254e+00,
-1.1122e+00, -3.9427e-01]],
[[-2.4162e+00, -4.7951e+00, -4.1381e+00, ..., -5.3001e-01,
-1.0325e+00, 1.5520e-02],
[-5.3672e+00, -9.2302e+00, -8.2004e+00, ..., -1.9571e+00,
-2.7511e+00, -7.0270e-01],
[-4.3827e+00, -7.6526e+00, -6.9850e+00, ..., -1.4080e+00,
-1.5993e+00, -4.1310e-01],
...,
[-4.3744e+00, -6.1859e+00, -5.6313e+00, ..., -1.3125e+00,
-1.5888e+00, -3.8221e-01],
[-6.3903e+00, -7.4804e+00, -6.2899e+00, ..., -2.4372e+00,
-3.3510e+00, -9.5262e-01],
[-8.8629e-03, -8.3614e-01, 3.8632e-01, ..., 3.3770e-01,
3.6771e-01, 8.3855e-02]],
[[-5.7288e-01, 6.2529e-01, 1.1067e+00, ..., 8.4675e-01,
3.1003e-01, 1.1016e+00],
[-9.0734e-01, -9.1774e-03, 4.3777e-01, ..., 4.3586e-01,
8.3777e-02, 4.8060e-01],
[-1.6537e+00, -3.8256e-01, -8.5477e-01, ..., 5.9835e-02,
1.3385e-01, 4.2456e-01],
...,
[-1.6822e+00, -1.0212e-01, -1.3138e-01, ..., 8.3277e-02,
9.5724e-03, 4.2445e-01],
[-1.5134e+00, 7.4080e-02, 3.9462e-02, ..., 1.3669e-01,
2.9325e-02, 5.6153e-01],
[-1.4446e+00, -4.9095e-01, -2.7328e-01, ..., -6.3163e-02,
3.4672e-02, 3.3921e-01]],
...,
[[ 1.0715e+00, -4.1430e-01, -2.1349e-01, ..., -1.2943e-01,
-3.9633e-01, -9.4106e-01],
[-7.7747e-01, -9.0500e-01, -7.3513e-01, ..., -7.2336e-01,
-1.3177e+00, -8.5062e-01],
[-6.8129e-01, -4.3204e-01, 1.3882e-01, ..., -1.0147e-01,
-2.1389e-01, -3.9541e-01],
...,
[-7.5222e-01, -1.1213e+00, -7.1411e-01, ..., -1.0110e-01,
-2.0274e-01, -3.9479e-01],
[-2.7899e+00, -2.8166e+00, -1.8179e+00, ..., -8.4167e-01,
-9.1201e-01, -8.1543e-01],
[-6.6513e-01, -1.6777e+00, -1.0962e+00, ..., -2.4121e-02,
-1.5582e-01, -3.4313e-01]],
[[ 2.4686e-02, -1.9118e-02, 5.8055e-01, ..., 3.0819e-01,
6.7009e-01, 2.9551e-01],
[ 1.4422e-02, -4.6631e-01, 3.4091e-01, ..., -3.3466e-02,
7.6112e-02, -6.2363e-02],
[-7.1548e-02, -4.2557e-01, -4.9323e-04, ..., -1.4880e-01,
-1.2483e-01, 1.8634e-01],
...,
[ 4.3910e-02, -7.8943e-01, 2.8790e-01, ..., -1.1091e-01,
-9.8809e-02, 1.3340e-01],
[-2.3614e-01, -1.4293e+00, 6.6152e-02, ..., -3.8169e-01,
-3.2027e-01, -8.8239e-02],
[ 6.5690e-01, 1.0823e-01, 3.0079e-01, ..., -2.3031e-01,
-2.4678e-01, 3.1122e-01]],
[[-1.5984e+00, -6.0356e-01, 2.6111e-02, ..., 5.5887e-01,
3.8020e-01, 5.6259e-01],
[-2.0812e+00, -3.5951e-01, -9.6368e-02, ..., -2.8946e-02,
-1.4952e-01, 4.1371e-01],
[-1.7655e+00, -6.7548e-01, -4.5429e-01, ..., 7.8314e-02,
-1.5514e-01, 9.0240e-02],
...,
[-1.7715e+00, -5.0985e-01, 2.3138e-01, ..., 6.6435e-03,
-7.8150e-02, 5.8659e-02],
[-2.1898e+00, -1.1887e+00, -1.9306e-01, ..., -1.3467e-01,
-2.3073e-01, -2.2327e-01],
[-1.0520e+00, -2.3128e-01, 4.4887e-01, ..., -1.0634e-01,
-2.7178e-01, 1.1250e-01]]],
[[[-6.1375e-01, -1.4434e+00, -3.1396e+00, ..., -1.2688e+00,
-8.8746e-01, -1.7142e+00],
[-2.4795e-01, -2.4988e+00, -3.6946e+00, ..., -1.8592e+00,
-1.8540e+00, -1.9300e+00],
[-1.5742e-01, -2.0543e+00, -2.1130e+00, ..., -9.2468e-01,
-1.2500e+00, -2.1040e+00],
...,
[ 7.4813e-02, -5.5785e-01, -2.1407e-01, ..., -8.6977e-02,
-1.6227e-02, -4.2186e-01],
[-3.2922e-01, -4.4671e-01, 2.2898e-02, ..., 1.4196e-01,
2.8525e-01, -1.6328e-01],
[-6.2446e-01, -1.3294e+00, -1.1209e+00, ..., -1.0715e+00,
-1.1595e+00, -4.8522e-01]],
[[-2.4994e+00, -7.1715e+00, -6.9168e+00, ..., -4.8009e+00,
-6.0504e+00, -2.5037e+00],
[-5.4739e+00, -1.0936e+01, -1.0220e+01, ..., -8.4987e+00,
-9.4888e+00, -4.3468e+00],
[-3.9747e+00, -8.1781e+00, -9.5249e+00, ..., -6.5679e+00,
-7.0869e+00, -3.2009e+00],
...,
[-1.7680e+00, -2.2065e+00, -1.7815e+00, ..., -1.8346e+00,
-2.2457e+00, -9.0273e-01],
[-2.7925e+00, -2.9562e+00, -1.9595e+00, ..., -2.1743e+00,
-3.3293e+00, -9.8939e-01],
[ 3.5637e-01, -2.4834e-01, 2.9456e-01, ..., 3.0137e-01,
4.4685e-01, 2.1327e-01]],
[[-6.8053e-01, 2.5558e-01, -4.9386e-01, ..., 7.6661e-01,
5.9655e-01, 1.8664e+00],
[-9.5194e-01, 3.3212e-01, -1.6214e+00, ..., 7.8438e-01,
1.0733e-01, 6.6326e-02],
[-1.7573e+00, 1.7960e-01, -3.0905e+00, ..., 4.4521e-01,
3.3822e-01, -1.0136e-01],
...,
[-6.4308e-01, 6.8594e-02, 2.4195e-01, ..., 1.7989e-01,
2.5328e-01, 4.0876e-01],
[-4.3912e-01, 1.0930e-01, 2.5399e-01, ..., 1.4902e-01,
2.2346e-01, 5.5762e-01],
[-1.7571e-01, -2.6456e-01, 1.4143e-01, ..., 1.7602e-02,
1.0534e-01, 3.4557e-01]],
...,
[[ 1.0728e+00, -1.3607e-01, 4.1760e-01, ..., 2.3490e-01,
-2.2629e-02, -2.0619e+00],
[-9.8756e-01, -6.4960e-01, -8.5328e-01, ..., -3.2671e-01,
-1.3712e+00, -2.0358e+00],
[-8.0854e-01, -2.9613e-01, 6.0858e-01, ..., -5.9018e-01,
-6.4417e-01, -1.3697e+00],
...,
[-6.5755e-01, -5.2504e-01, -3.5193e-01, ..., -3.3343e-01,
-4.9495e-01, -5.4327e-01],
[-2.2257e+00, -1.7631e+00, -1.3011e+00, ..., -1.3041e+00,
-1.3608e+00, -1.0980e+00],
[-6.1801e-01, -8.3485e-01, -4.6789e-01, ..., -2.3265e-01,
-3.5455e-01, -4.0619e-01]],
[[-6.7695e-02, 2.1683e-01, -4.9482e-03, ..., 4.8493e-01,
9.0418e-01, -8.0593e-03],
[-3.1610e-03, -1.0812e+00, -1.6492e+00, ..., 1.2929e-02,
1.5177e-01, -8.2970e-01],
[-1.4321e-01, -1.0962e+00, -9.5428e-01, ..., 8.6466e-01,
4.4370e-01, -3.7489e-01],
...,
[-2.3445e-01, -4.7205e-01, 2.0774e-01, ..., 3.4434e-01,
3.2833e-01, 7.1732e-02],
[-5.1099e-01, -9.0050e-01, 4.6782e-02, ..., -2.8983e-04,
-2.8814e-02, -1.6072e-01],
[ 3.3317e-01, -1.1393e-02, -2.0698e-01, ..., -1.3644e-01,
-1.0762e-01, 1.7147e-01]],
[[-1.6426e+00, -1.1685e+00, -1.6434e+00, ..., -4.8756e-01,
-3.2225e-01, 8.4371e-01],
[-2.1473e+00, -1.7960e+00, -2.5738e+00, ..., -7.9172e-01,
-1.7941e-01, 1.6509e+00],
[-1.4706e+00, -1.6574e+00, -1.9703e+00, ..., -6.7951e-01,
-1.4643e-01, 1.4033e+00],
...,
[-6.5787e-01, -2.0706e-01, -1.4764e-01, ..., -1.4371e-01,
-1.1923e-02, 6.3341e-01],
[-1.1278e+00, -5.4758e-01, -3.6021e-01, ..., -3.4482e-01,
-2.0100e-01, 1.7121e-01],
[-3.6494e-01, 1.3689e-02, -1.7283e-01, ..., -2.5114e-01,
-2.3963e-01, 3.2011e-01]]],
[[[-6.5483e-01, -1.9274e+00, -2.0227e+00, ..., -1.1094e-02,
6.1353e-01, 5.5769e-01],
[-1.5911e-01, -3.8411e+00, -3.0954e+00, ..., -4.7024e-01,
-2.6106e-01, 1.2952e-01],
[-4.6116e-01, -4.0985e+00, -1.9250e+00, ..., -3.1955e-01,
-1.6679e-01, -8.6585e-02],
...,
[-4.8751e-01, -1.8746e+00, -1.7827e+00, ..., -3.7349e-01,
-3.0239e-01, -1.2865e-01],
[-9.4202e-01, -1.6370e+00, -1.1790e+00, ..., 5.4280e-03,
2.8621e-01, -1.2246e-02],
[-7.9928e-01, -1.6493e+00, -1.2911e+00, ..., -1.0195e+00,
-1.0842e+00, -3.7605e-01]],
[[-3.5694e+00, -6.3247e+00, -6.5819e+00, ..., -4.4807e-01,
-1.0223e+00, 4.5517e-03],
[-6.6557e+00, -9.9399e+00, -8.8978e+00, ..., -1.8108e+00,
-2.6720e+00, -6.8651e-01],
[-5.4569e+00, -7.0590e+00, -5.8193e+00, ..., -1.3126e+00,
-1.5146e+00, -3.9083e-01],
...,
[-4.3723e+00, -6.1666e+00, -5.5602e+00, ..., -1.3305e+00,
-1.5935e+00, -3.5978e-01],
[-6.3648e+00, -7.4392e+00, -6.2740e+00, ..., -2.4872e+00,
-3.3411e+00, -9.4426e-01],
[-1.0190e-02, -8.3494e-01, 4.1383e-01, ..., 3.4536e-01,
3.6514e-01, 8.4719e-02]],
[[-6.0078e-01, 6.6243e-01, 1.4045e+00, ..., 8.2483e-01,
3.1150e-01, 1.1016e+00],
[-1.4150e+00, -1.1350e+00, 2.2964e-02, ..., 4.1103e-01,
5.9894e-02, 4.8563e-01],
[-1.6848e+00, -3.4089e+00, -4.3005e+00, ..., 5.0252e-02,
8.9734e-02, 4.2007e-01],
...,
[-1.6969e+00, -7.7038e-02, 3.4577e-04, ..., -2.7804e-01,
1.8683e-01, 4.1544e-01],
[-1.5473e+00, 8.5821e-02, -4.5255e-02, ..., 9.8090e-02,
1.1525e-01, 6.2059e-01],
[-1.4122e+00, -4.3826e-01, -2.4950e-01, ..., -7.7995e-02,
8.8658e-02, 3.8594e-01]],
...,
[[ 1.1053e+00, -2.5111e-01, -1.8516e-02, ..., -9.4449e-02,
-4.1054e-01, -9.4245e-01],
[-4.5281e-02, 3.7453e-01, -4.3576e-01, ..., -6.7289e-01,
-1.2995e+00, -8.3248e-01],
[-3.0685e-01, -7.4616e-01, -2.6031e+00, ..., -1.1398e-01,
-2.1053e-01, -3.8526e-01],
...,
[-7.5292e-01, -1.1174e+00, -7.9106e-01, ..., -3.2562e-01,
-7.9828e-02, -3.6832e-01],
[-2.7887e+00, -2.8643e+00, -1.8575e+00, ..., -8.2804e-01,
-8.6162e-01, -7.8241e-01],
[-6.8104e-01, -1.6603e+00, -9.8528e-01, ..., -2.4521e-02,
-1.4153e-01, -3.4029e-01]],
[[ 2.4448e-01, 2.3731e-01, 6.6044e-01, ..., 3.1374e-01,
6.8873e-01, 3.0448e-01],
[ 1.2939e-01, -7.2050e-01, 4.0075e-01, ..., -4.8404e-03,
1.1681e-01, -5.9038e-02],
[-1.5144e-01, -2.6616e+00, -1.8130e+00, ..., -1.2292e-01,
-1.0526e-01, 1.9033e-01],
...,
[ 3.9906e-02, -7.6121e-01, 3.4622e-01, ..., -2.3363e-01,
-2.8924e-01, 1.4041e-01],
[-2.6839e-01, -1.4467e+00, 3.8255e-02, ..., -4.0995e-01,
-3.1689e-01, -8.4551e-02],
[ 6.5669e-01, 1.2956e-01, 2.5347e-01, ..., -2.4238e-01,
-2.0999e-01, 3.0866e-01]],
[[-1.7093e+00, -7.6316e-01, -1.1950e-01, ..., 5.7055e-01,
3.8391e-01, 5.4623e-01],
[-2.6671e+00, -1.4366e+00, -2.0834e+00, ..., -4.7339e-02,
-1.1565e-01, 4.0436e-01],
[-2.5396e+00, -1.4950e+00, -2.6363e+00, ..., 4.4038e-02,
-1.3677e-01, 8.2586e-02],
...,
[-1.7143e+00, -5.5459e-01, 1.7110e-01, ..., 8.6296e-01,
1.0633e-01, 9.1526e-02],
[-2.1583e+00, -1.1531e+00, -1.7830e-01, ..., -8.9208e-02,
-2.1729e-01, -1.9830e-01],
[-1.0362e+00, -1.8618e-01, 4.5428e-01, ..., -1.0368e-01,
-2.6777e-01, 1.0325e-01]]],
[[[ 1.3169e-02, -2.5959e-01, -2.6487e-02, ..., -9.0192e-01,
-5.1616e-01, -1.4399e+00],
[ 2.8743e-01, -9.4952e-01, -6.7932e-01, ..., -1.0714e+00,
-1.3074e+00, -1.4833e+00],
[ 2.1282e-01, -5.8875e-01, -4.5623e-01, ..., -8.4608e-01,
-1.0818e+00, -1.8031e+00],
...,
[-1.8861e-02, -6.5481e-01, -4.0614e-01, ..., -9.2076e-01,
-1.0530e+00, -1.9137e+00],
[-6.5733e-01, -7.4812e-01, -6.2671e-02, ..., -4.7209e-01,
-5.7425e-01, -1.7035e+00],
[-7.2204e-01, -1.3817e+00, -1.1812e+00, ..., -1.1713e+00,
-1.5605e+00, -7.7723e-01]],
[[-7.3992e-01, -2.1198e+00, -1.8213e+00, ..., -4.0382e+00,
-5.4549e+00, -1.8738e+00],
[-1.9820e+00, -3.6897e+00, -2.9945e+00, ..., -6.0131e+00,
-8.0083e+00, -3.4089e+00],
[-1.5852e+00, -2.5587e+00, -2.5590e+00, ..., -5.1106e+00,
-6.1112e+00, -2.5463e+00],
...,
[-1.8294e+00, -2.5273e+00, -2.4756e+00, ..., -4.9804e+00,
-6.2472e+00, -2.5342e+00],
[-3.7878e+00, -4.4747e+00, -3.6764e+00, ..., -6.4604e+00,
-7.9450e+00, -2.5996e+00],
[ 2.6950e-01, -3.6782e-01, 3.9994e-01, ..., 4.2886e-01,
4.6531e-01, 1.8786e-01]],
[[-2.1921e-01, 2.0171e-01, 7.2827e-01, ..., 6.4259e-01,
2.5903e-01, 1.5270e+00],
[ 2.4537e-02, -3.5637e-01, 5.6624e-01, ..., 6.6141e-01,
-7.9596e-02, 1.0668e-01],
[-1.5495e-01, -2.5086e-01, 2.2304e-01, ..., 1.5451e-01,
-2.5317e-02, -1.3684e-01],
...,
[-2.8946e-01, -2.2286e-01, 1.5538e-01, ..., -7.4086e-02,
-1.6857e-01, -2.1255e-01],
[-2.7877e-01, -2.2619e-01, 1.6489e-01, ..., -1.3259e-01,
-9.0620e-02, -2.5421e-01],
[-1.7702e-01, -5.4531e-01, 1.3412e-01, ..., 7.3267e-02,
2.1160e-01, -5.8670e-01]],
...,
[[ 6.2707e-01, -2.3005e-01, 3.3708e-03, ..., 5.0206e-01,
3.4970e-01, -1.3202e+00],
[-4.7249e-01, -5.4578e-01, -5.2735e-01, ..., -2.2467e-01,
-7.5451e-01, -1.4081e+00],
[-3.5247e-01, -2.0845e-01, -2.5586e-01, ..., -1.5968e-01,
-8.3689e-02, -9.5729e-01],
...,
[-4.1924e-01, -1.7075e-01, -1.9283e-01, ..., -1.3449e-01,
-1.0409e-01, -9.5118e-01],
[-2.3208e+00, -1.6997e+00, -1.2357e+00, ..., -1.5634e+00,
-1.5809e+00, -1.8237e+00],
[-4.6815e-01, -7.9965e-01, -4.2532e-01, ..., -3.7018e-01,
-5.5516e-01, -1.1734e+00]],
[[ 1.6968e-01, 2.5828e-01, 6.3906e-01, ..., -1.4247e-01,
4.1926e-01, -3.8968e-01],
[ 1.9058e-01, 2.3053e-01, 4.5991e-01, ..., 2.4436e-01,
1.2736e-01, -8.4114e-01],
[-1.0623e-01, 9.6383e-02, 3.0504e-01, ..., 2.9595e-01,
1.9490e-01, -4.5234e-01],
...,
[-1.9696e-01, 7.5018e-03, 3.1570e-01, ..., 2.0355e-01,
1.5748e-01, -4.5853e-01],
[-5.2157e-01, -6.6059e-01, -4.0724e-02, ..., -2.0386e-01,
-3.0453e-01, -9.0555e-01],
[ 4.8284e-01, 3.9117e-01, 6.9160e-02, ..., 3.1110e-01,
3.1537e-01, 3.0396e-01]],
[[-8.5044e-01, 2.5890e-01, 3.2332e-01, ..., -6.5903e-02,
-4.0946e-02, 5.9438e-01],
[-1.1561e+00, 1.9095e-01, -3.6657e-01, ..., -9.3372e-01,
-4.0616e-01, 8.9091e-01],
[-9.8523e-01, 3.5049e-01, -1.1732e-01, ..., -5.8160e-01,
-4.9654e-01, 6.2922e-01],
...,
[-9.5889e-01, 2.2871e-01, -8.1888e-02, ..., -6.2966e-01,
-4.9791e-01, 6.3161e-01],
[-1.7022e+00, -3.4564e-01, -2.9940e-01, ..., -7.5564e-01,
-6.7167e-01, 4.2626e-01],
[-7.2886e-01, -9.3886e-02, -2.8131e-01, ..., -5.0715e-01,
-4.4692e-01, 1.0015e+00]]]], device='cuda:0'),
tensor([[[[ 3.7384e-02, -2.2703e-01, -4.6174e-02, ..., -1.0234e+00,
-7.7423e-01, -5.1945e-01],
[ 3.0576e-01, -9.3195e-01, -5.7044e-01, ..., -1.3353e+00,
-1.6350e+00, -1.3356e+00],
[ 1.9887e-01, -5.9421e-01, -3.5756e-01, ..., -9.0437e-01,
-1.3398e+00, -2.2201e+00],
...,
[-3.9271e-03, -5.7317e-01, -4.2170e-01, ..., -1.3656e+00,
-1.1902e+00, -1.6504e+00],
[-6.3142e-01, -6.7761e-01, -8.3285e-02, ..., -7.3353e-01,
-5.7871e-01, -1.4426e+00],
[-6.4243e-01, -1.2970e+00, -1.1622e+00, ..., -1.2519e+00,
-1.4667e+00, -9.1039e-01]],
[[-8.2965e-01, -2.2434e+00, -1.8793e+00, ..., -4.4750e+00,
-4.3826e+00, -1.9669e+00],
[-2.1152e+00, -3.8435e+00, -3.7708e+00, ..., -6.8223e+00,
-7.1474e+00, -3.5917e+00],
[-1.6634e+00, -2.6135e+00, -3.2478e+00, ..., -5.7584e+00,
-6.1075e+00, -3.5275e+00],
...,
[-1.7850e+00, -2.4134e+00, -2.4369e+00, ..., -5.2080e+00,
-5.4443e+00, -2.3102e+00],
[-3.7533e+00, -4.3845e+00, -3.6484e+00, ..., -6.1747e+00,
-7.1106e+00, -2.7890e+00],
[ 2.6952e-01, -3.1862e-01, 4.2398e-01, ..., 5.6442e-01,
6.8704e-01, 7.3511e-02]],
[[-2.4680e-01, 1.3904e-01, 7.2964e-01, ..., 9.9517e-01,
7.0887e-01, 1.2928e+00],
[ 3.9780e-02, -3.7881e-01, 6.4074e-01, ..., 4.7385e-01,
-4.0331e-02, -1.2591e-01],
[-1.4516e-01, -2.9626e-01, 2.4720e-01, ..., 1.9553e-01,
-7.9542e-01, -8.8582e-02],
...,
[-2.5559e-01, -2.2520e-01, 1.4886e-01, ..., -1.6181e-01,
1.0848e-01, 3.6603e-02],
[-2.4499e-01, -2.0883e-01, 1.4860e-01, ..., -6.8217e-02,
9.8574e-02, 1.3372e-01],
[-1.0061e-01, -4.4588e-01, 1.0954e-01, ..., -3.5097e-01,
-7.5839e-02, -3.0790e-01]],
...,
[[ 6.0046e-01, -2.9327e-01, -2.2854e-02, ..., 6.5004e-03,
-2.8097e-01, -1.9076e+00],
[-4.6141e-01, -5.6713e-01, -4.9614e-01, ..., -6.2004e-01,
-2.1058e+00, -2.4567e+00],
[-3.2724e-01, -2.2558e-01, 8.6433e-02, ..., -3.6421e-01,
-5.0522e-01, -1.7472e+00],
...,
[-4.0994e-01, -1.1744e-01, -1.7923e-01, ..., -7.2501e-01,
-9.1247e-01, -1.1985e+00],
[-2.3162e+00, -1.6681e+00, -1.2575e+00, ..., -1.8253e+00,
-1.9309e+00, -1.7740e+00],
[-4.8074e-01, -7.8395e-01, -4.0986e-01, ..., -6.7292e-01,
-7.7545e-01, -8.3955e-01]],
[[ 1.6375e-01, 2.1799e-01, 6.0484e-01, ..., 6.1836e-01,
8.2557e-01, -2.9451e-01],
[ 1.2528e-01, 2.5523e-01, 7.0460e-01, ..., 6.1898e-01,
5.1412e-01, -9.9501e-01],
[-1.2615e-01, 1.4845e-01, 5.5888e-01, ..., 8.5883e-01,
4.5259e-01, -1.1336e+00],
...,
[-1.8767e-01, 2.4272e-02, 3.0249e-01, ..., 3.1798e-01,
2.1180e-01, -3.4652e-01],
[-5.2448e-01, -6.3567e-01, -2.4680e-02, ..., -5.1412e-02,
-8.0470e-02, -5.7281e-01],
[ 5.0186e-01, 4.1421e-01, 6.4259e-02, ..., 2.4077e-01,
1.9473e-01, 3.0801e-01]],
[[-8.3067e-01, 2.6295e-01, 3.2467e-01, ..., 5.6589e-02,
2.1586e-01, 5.9932e-01],
[-1.1705e+00, 2.4802e-01, -3.8684e-01, ..., -2.1627e-01,
5.5323e-01, 7.5701e-01],
[-1.0478e+00, 3.8267e-01, -4.0717e-01, ..., 5.8196e-02,
1.0102e+00, 4.6804e-01],
...,
[-9.7542e-01, 2.2217e-01, -9.5217e-02, ..., 1.9822e-01,
4.0462e-01, 1.2329e+00],
[-1.6945e+00, -3.5039e-01, -3.1453e-01, ..., -1.4763e-01,
1.4382e-01, 8.3814e-01],
[-6.8725e-01, -4.1746e-02, -2.7486e-01, ..., 2.3285e-01,
1.5539e-01, 1.0497e+00]]],
[[[-7.3600e-01, -1.4595e+00, -1.4136e+00, ..., 4.4830e-01,
4.9773e-01, 4.9396e-01],
[-3.0277e-01, -2.4498e+00, -2.1904e+00, ..., 8.0134e-01,
2.5415e-02, 9.5337e-02],
[-3.9789e-01, -2.1559e+00, -1.9051e+00, ..., -9.4972e-02,
1.2136e-01, -1.1956e-01],
...,
[-2.4718e-01, -1.1460e+00, -1.3708e+00, ..., -9.4948e-01,
1.2201e-01, -1.5006e-02],
[-9.3684e-01, -1.2429e+00, -3.7585e-01, ..., -6.2100e-02,
-2.0279e-01, -2.7070e-01],
[-8.3454e-01, -1.6759e+00, -1.2395e+00, ..., -8.6944e-01,
-1.5162e+00, -6.0887e-01]],
[[-2.3405e+00, -4.8421e+00, -4.2770e+00, ..., -2.3285e+00,
-1.6179e+00, -5.5582e-02],
[-5.3689e+00, -8.0694e+00, -6.6520e+00, ..., -2.5551e+00,
-3.3244e+00, -7.2200e-01],
[-4.3869e+00, -6.2767e+00, -5.5057e+00, ..., -1.6344e+00,
-2.1381e+00, -4.4618e-01],
...,
[-4.4594e+00, -5.9335e+00, -5.9228e+00, ..., -2.1953e+00,
-3.1902e+00, 1.3504e-01],
[-6.4606e+00, -7.7136e+00, -6.7228e+00, ..., -3.2221e+00,
-4.2970e+00, -6.0672e-01],
[-3.6175e-02, -8.6504e-01, 3.5266e-01, ..., 2.2937e-01,
1.3725e-01, 6.6060e-03]],
[[-6.3851e-01, 5.4933e-01, 1.1184e+00, ..., 1.8181e+00,
4.3300e-01, 1.0953e+00],
[-9.1919e-01, -5.6425e-02, 2.9198e-01, ..., 6.9049e-01,
2.2698e-01, 5.1411e-01],
[-1.6413e+00, -1.6102e-01, -9.6517e-02, ..., -1.0497e+00,
1.8678e-01, 4.3941e-01],
...,
[-1.1196e+00, 7.9269e-01, -2.7325e+00, ..., -8.5897e-01,
-1.9462e+00, -9.9616e-01],
[-1.3895e+00, 1.2550e+00, 4.2173e-01, ..., -7.3833e-01,
-1.1347e+00, -5.2010e-01],
[-1.4696e+00, -5.4438e-01, -2.8476e-01, ..., -5.6408e-01,
-3.0203e-02, 4.3479e-01]],
...,
[[ 1.0712e+00, -4.1333e-01, -2.0776e-01, ..., -2.3745e-01,
-2.3528e-01, -8.9403e-01],
[-8.1478e-01, -1.0018e+00, -8.1411e-01, ..., -2.6539e+00,
-1.4046e+00, -8.1665e-01],
[-6.8115e-01, -9.8479e-01, -6.9764e-01, ..., -2.9996e+00,
-5.5632e-01, -3.8997e-01],
...,
[-9.7097e-01, -1.0327e+00, -5.4984e-01, ..., -3.8515e-01,
-8.3572e-01, -1.2655e+00],
[-2.9719e+00, -3.7976e+00, -3.5910e+00, ..., -7.6504e-01,
-1.3451e+00, -1.5781e+00],
[-6.3819e-01, -1.6717e+00, -1.0246e+00, ..., 1.6852e-01,
4.0822e-02, -7.8525e-01]],
[[ 1.2964e-02, 3.1377e-02, 6.1600e-01, ..., -1.9637e-01,
3.0997e-01, 2.8699e-01],
[-6.2844e-02, -7.6496e-01, 1.5355e-01, ..., 1.6096e-01,
4.5536e-02, -3.3921e-02],
[-1.2763e-01, -8.4406e-01, 3.1356e-01, ..., -2.7843e-01,
-1.6819e-01, 1.7397e-01],
...,
[ 2.4059e-01, -7.7522e-01, -1.5704e+00, ..., -3.0451e-01,
-6.1448e-01, -6.7931e-01],
[-1.2915e-01, -1.1583e+00, -4.8303e-01, ..., -4.8878e-01,
-4.5636e-01, -7.3088e-01],
[ 6.6793e-01, 1.5140e-01, 3.0352e-01, ..., -5.1738e-01,
-4.8332e-01, 4.4859e-01]],
[[-1.6458e+00, -6.5470e-01, 2.4380e-02, ..., 5.8158e-01,
3.4152e-01, 5.2403e-01],
[-2.1460e+00, -3.8251e-01, -3.7071e-03, ..., -1.2248e-01,
-5.3549e-02, 4.0967e-01],
[-1.7817e+00, -4.2913e-01, 2.2453e-01, ..., 3.1525e-01,
4.3712e-03, 1.1008e-01],
...,
[-2.0668e+00, -1.4042e+00, 6.8230e-01, ..., 3.8427e-01,
-4.2666e-02, -1.0593e+00],
[-2.2911e+00, -1.3849e+00, 3.0930e-01, ..., 2.7972e-01,
-2.0259e-01, -1.0637e+00],
[-1.0858e+00, -2.2530e-01, 4.4637e-01, ..., 4.7453e-01,
1.5959e-01, -6.8300e-03]]],
[[[ 4.8412e-02, -2.9300e-01, -6.4020e-02, ..., -9.7093e-01,
-5.2969e-01, -1.4442e+00],
[-6.7977e-01, -2.1637e+00, -1.9722e+00, ..., -1.1611e+00,
-1.3259e+00, -1.4917e+00],
[-2.7364e+00, -2.4285e+00, -8.5902e-01, ..., -9.5980e-01,
-1.1071e+00, -1.8281e+00],
...,
[ 1.0054e-01, -4.2401e-01, -3.5687e-01, ..., -2.7742e+00,
1.3535e-02, -8.9892e-01],
[-4.2591e-01, -4.6536e-01, -1.6523e-03, ..., -1.2986e+00,
1.1884e+00, -1.6433e+00],
[-6.0783e-01, -1.2582e+00, -1.0430e+00, ..., 1.0341e+00,
-1.8166e+00, -2.2972e+00]],
[[-7.7143e-01, -2.4238e+00, -1.9739e+00, ..., -3.9487e+00,
-5.4545e+00, -1.8410e+00],
[-3.8742e+00, -8.4819e+00, -8.9539e+00, ..., -5.9684e+00,
-8.1316e+00, -3.4346e+00],
[-4.0961e+00, -7.9171e+00, -5.8775e+00, ..., -5.4448e+00,
-6.2191e+00, -2.5629e+00],
...,
[-1.7574e+00, -2.4527e+00, -2.7175e+00, ..., -8.0989e+00,
-9.4924e+00, -5.9195e+00],
[-3.6015e+00, -4.1408e+00, -3.5660e+00, ..., -6.7154e+00,
-1.0318e+01, -4.0151e+00],
[ 2.8070e-01, -2.9545e-01, 3.9008e-01, ..., -2.7755e-01,
-1.2318e+00, -7.1521e-01]],
[[-2.7233e-01, 1.0279e-01, 8.0938e-01, ..., 6.0722e-01,
2.6628e-01, 1.5476e+00],
[-1.1807e-01, -6.1444e-01, 8.5091e-01, ..., 5.4134e-01,
-6.9015e-02, 1.1052e-01],
[-4.1214e+00, -6.1315e+00, -4.1212e+00, ..., 1.1462e-01,
-1.2195e-03, -1.4518e-01],
...,
[-2.5082e-01, -2.1495e-01, 4.5083e-02, ..., -5.8062e+00,
-3.5414e+00, 7.5962e-01],
[-8.5610e-02, -8.3956e-02, 1.1576e-01, ..., -3.9618e+00,
-3.5969e+00, 1.3909e+00],
[-8.4265e-02, -3.9208e-01, 1.2888e-01, ..., -5.5957e-01,
-1.5937e+00, 3.1317e-01]],
...,
[[ 4.4320e-01, -3.7106e-01, -1.5632e-03, ..., 4.6955e-01,
3.4941e-01, -1.3357e+00],
[-7.0964e-01, -5.2579e-01, -6.1397e-01, ..., -1.6404e-01,
-7.9544e-01, -1.4229e+00],
[ 1.1782e-01, -7.0010e-01, -2.1185e+00, ..., -3.7978e-02,
-7.4564e-02, -9.5920e-01],
...,
[-1.4547e-01, 5.5776e-02, -3.0360e-01, ..., -3.7940e+00,
-5.2439e+00, -3.3672e+00],
[-2.1998e+00, -1.6257e+00, -1.2091e+00, ..., -4.1799e+00,
-8.0286e+00, -5.0963e+00],
[-4.7292e-01, -6.8744e-01, -4.1486e-01, ..., -6.1160e+00,
-4.5494e+00, -2.0528e+00]],
[[ 1.3351e-01, 2.7303e-01, 6.6669e-01, ..., -7.7142e-02,
4.0846e-01, -3.3975e-01],
[-2.6881e-01, 1.8531e-01, 5.8133e-01, ..., 2.1630e-01,
1.1379e-01, -8.0622e-01],
[-2.0569e+00, -1.7960e+00, -6.2606e-01, ..., 7.6756e-02,
1.2718e-01, -4.4945e-01],
...,
[-2.3264e-01, 6.1447e-02, 6.1971e-01, ..., -1.6368e+00,
-8.5137e-01, -8.9147e-01],
[-4.4272e-01, -5.7620e-01, 6.6853e-02, ..., -7.7848e-01,
1.4166e+00, -9.1715e-01],
[ 5.1134e-01, 3.8702e-01, 9.0788e-02, ..., 2.1808e+00,
7.0510e-01, -7.3440e-01]],
[[-9.6855e-01, 2.6962e-01, 3.5581e-01, ..., -1.1887e-01,
-9.3812e-02, 5.7784e-01],
[-1.7397e+00, -5.8913e-01, -1.4584e+00, ..., -9.2026e-01,
-4.1389e-01, 9.1712e-01],
[-2.2203e+00, -6.3395e+00, -6.6622e+00, ..., -5.6633e-01,
-5.2894e-01, 6.3951e-01],
...,
[-9.2314e-01, 1.6586e-01, -3.2099e-01, ..., -8.5814e+00,
-1.1256e+00, 1.2372e+00],
[-1.5679e+00, -1.6505e-01, -1.3796e-01, ..., -7.2009e+00,
-1.0798e-01, 1.4244e+00],
[-6.3441e-01, 1.2235e-02, -2.7575e-01, ..., -2.2861e+00,
1.8045e+00, 2.0611e+00]]],
[[[-7.3439e-01, -1.4221e+00, -1.4189e+00, ..., -2.0840e-01,
2.3855e-01, -3.9222e-01],
[-2.9801e-01, -2.3880e+00, -2.1401e+00, ..., -6.0014e-01,
-3.8954e-01, -5.6151e-01],
[-4.0389e-01, -2.0082e+00, -1.9903e+00, ..., -3.6676e-01,
-3.6542e-01, -1.0425e+00],
...,
[-4.6427e-01, -1.9600e+00, -1.8416e+00, ..., -2.6230e-01,
-1.8630e-01, -1.1630e+00],
[-9.8398e-01, -1.7556e+00, -1.1596e+00, ..., 5.9226e-03,
9.1184e-02, -9.5668e-01],
[-8.3414e-01, -1.6819e+00, -1.2549e+00, ..., -9.0109e-01,
-1.1247e+00, -6.5670e-01]],
[[-2.3771e+00, -4.8108e+00, -4.2231e+00, ..., -2.1188e+00,
-3.4367e+00, -1.2620e+00],
[-5.3305e+00, -8.0925e+00, -6.7287e+00, ..., -3.4275e+00,
-5.1164e+00, -2.1166e+00],
[-4.3065e+00, -6.7796e+00, -6.8698e+00, ..., -3.0172e+00,
-3.8358e+00, -1.7431e+00],
...,
[-4.3647e+00, -6.1240e+00, -5.5059e+00, ..., -3.0159e+00,
-3.8196e+00, -1.7512e+00],
[-6.3982e+00, -7.4191e+00, -6.3544e+00, ..., -3.9002e+00,
-5.3867e+00, -1.8985e+00],
[-2.7942e-02, -8.4612e-01, 4.1015e-01, ..., 4.2298e-01,
6.1066e-01, 1.7195e-01]],
[[-5.8963e-01, 6.0826e-01, 1.1136e+00, ..., 8.3161e-01,
5.5846e-01, 1.3816e+00],
[-9.0976e-01, -5.7461e-02, 3.8214e-01, ..., 4.6552e-01,
3.5667e-01, 2.2329e-01],
[-1.6233e+00, -1.1661e-01, -1.4917e-01, ..., 9.0097e-02,
2.7104e-01, 7.7224e-02],
...,
[-1.7300e+00, -3.5612e-02, -1.4741e-01, ..., 2.4284e-01,
1.4136e-01, -4.6649e-02],
[-1.5591e+00, 9.0167e-02, 3.8766e-02, ..., 1.1462e-01,
2.3522e-01, 1.4787e-01],
[-1.4221e+00, -5.0811e-01, -2.7919e-01, ..., 2.3300e-02,
1.8860e-01, -1.4969e-01]],
...,
[[ 1.0785e+00, -4.0172e-01, -2.3225e-01, ..., 3.6411e-01,
2.1251e-01, -8.1459e-01],
[-8.1850e-01, -9.8781e-01, -7.9583e-01, ..., -3.2390e-01,
-1.0848e+00, -9.6567e-01],
[-6.7852e-01, -9.3743e-01, -5.2065e-01, ..., -9.5127e-02,
-2.3755e-01, -5.9944e-01],
...,
[-7.4973e-01, -1.1507e+00, -7.4773e-01, ..., -2.8047e-01,
-4.2651e-01, -7.0062e-01],
[-2.8099e+00, -2.8186e+00, -1.8603e+00, ..., -1.0561e+00,
-1.1187e+00, -1.1083e+00],
[-6.5786e-01, -1.6739e+00, -1.0115e+00, ..., -1.7489e-01,
-2.6173e-01, -4.7853e-01]],
[[ 1.8178e-02, 8.8843e-03, 6.0811e-01, ..., 3.1022e-01,
8.4747e-01, 9.5681e-02],
[-1.6983e-02, -7.3394e-01, 1.9356e-01, ..., 3.9262e-01,
6.7181e-01, -2.5812e-01],
[-1.1570e-01, -6.7161e-01, 5.1000e-01, ..., 3.7592e-01,
4.3453e-01, -1.0763e-01],
...,
[ 7.5665e-02, -8.0707e-01, 2.7597e-01, ..., 4.9234e-01,
4.5308e-01, -2.7279e-01],
[-2.4328e-01, -1.4757e+00, 1.6530e-02, ..., 1.4352e-02,
7.1005e-02, -4.6053e-01],
[ 6.5491e-01, 1.5223e-01, 2.6843e-01, ..., 2.3290e-01,
2.9718e-01, 3.4628e-01]],
[[-1.6094e+00, -5.9661e-01, 8.2739e-02, ..., 3.0944e-01,
2.4350e-01, 8.6006e-01],
[-2.1064e+00, -3.4416e-01, -2.0207e-02, ..., -4.5340e-01,
-1.5998e-01, 1.2175e+00],
[-1.7312e+00, -4.0435e-01, 3.2270e-02, ..., -1.0252e-01,
-7.6040e-03, 9.6108e-01],
...,
[-1.7480e+00, -4.7441e-01, 2.3768e-01, ..., -1.1445e-01,
1.3528e-01, 1.0724e+00],
[-2.2159e+00, -1.1409e+00, -1.7516e-01, ..., -2.5715e-01,
-9.1357e-02, 6.3370e-01],
[-1.0675e+00, -2.3096e-01, 4.4615e-01, ..., -2.2004e-01,
-2.4579e-01, 6.8903e-01]]],
[[[ 3.9284e-02, -2.4926e-01, -5.2763e-02, ..., -1.8961e+00,
-2.5202e+00, 5.3854e-01],
[ 3.1687e-01, -9.4134e-01, -7.4400e-01, ..., -4.0860e+00,
-5.8405e+00, 1.0505e+00],
[ 2.5200e-01, -5.5812e-01, -3.8536e-01, ..., -3.0902e+00,
-5.0754e+00, 5.0819e-01],
...,
[-1.3506e-02, -5.9454e-01, -4.1867e-01, ..., -1.4200e+00,
-1.1899e+00, -1.6956e+00],
[-5.9714e-01, -6.8577e-01, -6.1168e-02, ..., -7.9082e-01,
-5.8384e-01, -1.4175e+00],
[-7.1676e-01, -1.3562e+00, -1.1870e+00, ..., -1.1223e+00,
-1.3278e+00, -8.0126e-01]],
[[-7.8515e-01, -2.1947e+00, -1.9004e+00, ..., -6.7127e+00,
-5.8038e+00, -1.8130e+00],
[-2.0562e+00, -3.8245e+00, -3.0964e+00, ..., -1.0000e+01,
-9.2195e+00, -4.5035e+00],
[-1.5870e+00, -2.6565e+00, -6.0312e+00, ..., -8.6830e+00,
-9.8262e+00, -4.5945e+00],
...,
[-1.7708e+00, -2.4156e+00, -2.4365e+00, ..., -5.2953e+00,
-5.5328e+00, -2.3380e+00],
[-3.8486e+00, -4.4545e+00, -3.7055e+00, ..., -6.0227e+00,
-6.9016e+00, -2.6598e+00],
[ 2.8910e-01, -3.1793e-01, 3.8741e-01, ..., 4.7379e-01,
6.1455e-01, 8.5657e-02]],
[[-2.1941e-01, 1.5229e-01, 7.2367e-01, ..., 2.4095e-01,
-4.0891e-01, 1.8432e+00],
[ 6.3777e-02, -3.8126e-01, 5.2666e-01, ..., -1.4072e+00,
-5.1444e+00, -4.8962e-01],
[-1.3805e-01, -1.9728e-01, -4.4819e-02, ..., -3.9451e+00,
-1.0814e+01, -9.6237e-01],
...,
[-2.7160e-01, -1.7936e-01, 1.1982e-01, ..., -1.4177e-01,
1.0964e-01, 1.0664e-02],
[-2.6548e-01, -1.6013e-01, 1.6185e-01, ..., -3.0998e-02,
1.6289e-01, 1.0738e-01],
[-1.4942e-01, -4.7553e-01, 1.2091e-01, ..., -4.1766e-01,
-5.7583e-02, -3.0295e-01]],
...,
[[ 5.8447e-01, -2.5897e-01, -2.4754e-03, ..., 7.2561e-01,
-3.3866e-01, -2.6397e+00],
[-5.2048e-01, -5.1808e-01, -5.0956e-01, ..., -3.3157e-01,
-4.3995e+00, -5.7016e+00],
[-3.7147e-01, -1.7267e-01, 1.7438e-01, ..., 4.6939e-01,
-4.2166e+00, -7.0280e+00],
...,
[-4.4560e-01, -1.5050e-01, -1.4939e-01, ..., -7.1009e-01,
-8.7811e-01, -1.2028e+00],
[-2.3322e+00, -1.7208e+00, -1.2414e+00, ..., -1.7412e+00,
-1.8245e+00, -1.6814e+00],
[-4.8623e-01, -7.9052e-01, -4.0255e-01, ..., -7.5846e-01,
-7.7568e-01, -8.2359e-01]],
[[ 1.5486e-01, 2.5657e-01, 6.2640e-01, ..., 8.3453e-01,
-2.9782e-01, -1.5620e-01],
[ 1.3285e-01, 2.5222e-01, 4.5370e-01, ..., 6.5310e-01,
-3.0024e+00, -9.7185e-01],
[-1.2008e-01, 1.2631e-01, 7.9169e-01, ..., 4.4350e-01,
-4.9166e+00, -2.8600e-01],
...,
[-1.9700e-01, 1.2165e-02, 3.3845e-01, ..., 3.0687e-01,
2.0743e-01, -3.0620e-01],
[-5.4377e-01, -7.0558e-01, -4.6080e-02, ..., 4.8601e-02,
-2.7964e-02, -5.1511e-01],
[ 4.9980e-01, 3.7574e-01, 5.0437e-02, ..., 2.9324e-01,
1.9767e-01, 3.2887e-01]],
[[-8.4360e-01, 2.6168e-01, 3.5219e-01, ..., -4.1411e-01,
3.1651e-01, -5.3246e-02],
[-1.1380e+00, 2.3380e-01, -3.0998e-01, ..., -2.1439e+00,
-1.7730e+00, -3.1800e+00],
[-9.4467e-01, 3.4400e-01, -2.4293e-01, ..., -1.2779e+00,
-2.7153e+00, -7.5811e+00],
...,
[-9.5696e-01, 1.7661e-01, -1.2113e-01, ..., 2.1123e-01,
3.9910e-01, 1.2454e+00],
[-1.6754e+00, -3.7066e-01, -3.5240e-01, ..., -7.8290e-02,
1.9212e-01, 8.8755e-01],
[-7.2959e-01, -6.5229e-02, -3.0808e-01, ..., 3.2994e-01,
2.1281e-01, 1.0956e+00]]]], device='cuda:0'),
tensor([[[[-1.8962e+00, -2.4127e+00, -2.0905e+00, ..., -5.7955e-01,
-1.0472e-01, -8.3898e-01],
[-4.2469e-01, -3.4173e+00, -2.6180e+00, ..., -1.0034e+00,
-8.4297e-01, -1.1920e+00],
[ 3.1618e-01, -1.2060e+00, 8.3428e-02, ..., -7.9501e-01,
-8.3223e-01, -1.7402e+00],
...,
[ 6.4906e-02, -5.7666e-01, 3.8240e-01, ..., -1.0117e-01,
-4.4374e-02, -4.5396e-01],
[-3.8702e-01, -4.9512e-01, 6.3552e-02, ..., 1.3805e-01,
2.6494e-01, -1.6611e-01],
[-6.3098e-01, -1.3311e+00, -1.1061e+00, ..., -1.0894e+00,
-1.1751e+00, -5.0822e-01]],
[[-5.6068e+00, -1.1296e+01, -1.0900e+01, ..., -4.0103e+00,
-4.9534e+00, -2.0010e+00],
[-6.9236e+00, -1.2421e+01, -1.1347e+01, ..., -5.4422e+00,
-6.7341e+00, -2.8779e+00],
[-3.8105e+00, -5.1884e+00, -4.8831e+00, ..., -5.1827e+00,
-5.7294e+00, -2.7860e+00],
...,
[-1.7574e+00, -2.2255e+00, -1.7745e+00, ..., -1.8466e+00,
-2.2615e+00, -9.2105e-01],
[-2.8621e+00, -3.0506e+00, -1.9691e+00, ..., -2.2489e+00,
-3.3822e+00, -1.0129e+00],
[ 3.3469e-01, -2.7243e-01, 3.4154e-01, ..., 3.0856e-01,
4.5286e-01, 1.9811e-01]],
[[-3.7748e+00, -4.8097e+00, -6.2533e+00, ..., 7.9341e-01,
5.8013e-01, 1.4787e+00],
[-4.5645e+00, -4.5635e+00, -4.9560e+00, ..., 4.1331e-01,
3.1490e-01, 4.5193e-01],
[-1.2655e+00, 4.7224e-01, -5.7603e-01, ..., 1.6153e-01,
1.9126e-01, 3.5853e-01],
...,
[-6.7694e-01, 7.8487e-02, 1.3468e+00, ..., 1.8164e-01,
2.0854e-01, 4.0295e-01],
[-4.8197e-01, 8.0777e-02, 2.8159e-01, ..., 1.6749e-01,
2.0821e-01, 5.8967e-01],
[-1.4342e-01, -2.3476e-01, 1.4387e-01, ..., -6.3400e-03,
7.3087e-02, 3.4491e-01]],
...,
[[ 2.6526e-01, -2.4672e+00, -4.1567e+00, ..., 4.4829e-01,
2.2459e-01, -9.9777e-01],
[ 9.7231e-01, -1.9478e-01, -2.5344e+00, ..., -3.2170e-01,
-1.1307e+00, -1.3007e+00],
[ 2.3807e-01, -5.6549e-01, -3.0106e+00, ..., -3.1334e-01,
-5.7327e-01, -1.1642e+00],
...,
[-6.7305e-01, -5.2730e-01, -8.9012e-01, ..., -3.5320e-01,
-5.0495e-01, -5.6264e-01],
[-2.2840e+00, -1.8060e+00, -1.4010e+00, ..., -1.3202e+00,
-1.3654e+00, -1.0896e+00],
[-6.0582e-01, -8.6810e-01, -4.4487e-01, ..., -2.5321e-01,
-3.4701e-01, -3.8662e-01]],
[[-1.1194e+00, -2.2965e+00, -2.9918e+00, ..., 2.1812e-01,
6.7462e-01, -3.1636e-01],
[-7.3512e-01, -2.2729e+00, -3.8567e-01, ..., 2.0154e-01,
3.8753e-01, -7.4800e-01],
[-3.7567e-01, -2.1639e+00, -4.8144e-01, ..., 2.6888e-01,
1.7040e-01, -6.8413e-01],
...,
[-2.2910e-01, -5.0048e-01, 3.5839e-01, ..., 3.5517e-01,
3.3604e-01, 2.2642e-02],
[-5.4656e-01, -9.7248e-01, 1.0744e-01, ..., 1.1679e-02,
-3.8991e-02, -1.6547e-01],
[ 3.1947e-01, -4.8670e-02, -1.8917e-01, ..., -1.4014e-01,
-9.5463e-02, 1.8890e-01]],
[[-1.2765e+00, -4.5575e+00, -5.1145e+00, ..., 3.1798e-01,
4.2017e-01, 9.4425e-01],
[-1.3802e+00, -4.6042e+00, -5.9628e+00, ..., -6.1769e-01,
-2.4586e-01, 9.6343e-01],
[-1.2014e+00, -1.6595e+00, -9.2736e-01, ..., -3.4984e-01,
-3.2765e-02, 7.7697e-01],
...,
[-6.6217e-01, -2.6707e-01, -6.0184e-01, ..., -1.6580e-01,
3.9896e-02, 6.2248e-01],
[-1.1682e+00, -6.2412e-01, -3.2450e-01, ..., -3.5349e-01,
-1.8942e-01, 1.6153e-01],
[-3.8264e-01, -3.7001e-02, -1.5359e-01, ..., -2.4998e-01,
-2.1972e-01, 3.2920e-01]]],
[[[-7.3350e-01, -1.4706e+00, -1.4458e+00, ..., -1.2512e+00,
-8.2054e-01, -1.6408e+00],
[-4.4651e-01, -2.9490e+00, -2.9706e+00, ..., -1.7332e+00,
-1.8345e+00, -1.8853e+00],
[-2.5956e+00, -3.4302e+00, -1.8036e+00, ..., -1.2331e+00,
-1.4020e+00, -2.1149e+00],
...,
[ 6.1841e-02, -5.4302e-01, -2.2684e-01, ..., 1.5391e+00,
-1.9339e-01, -1.2602e+00],
[-3.9229e-01, -4.4596e-01, 4.3085e-02, ..., 1.3891e-01,
2.9208e-01, -1.5846e-01],
[-6.5835e-01, -1.3399e+00, -1.0998e+00, ..., -1.0875e+00,
-1.1803e+00, -5.1160e-01]],
[[-2.6934e+00, -5.0762e+00, -4.5891e+00, ..., -4.7200e+00,
-5.9573e+00, -2.4086e+00],
[-7.7330e+00, -1.2137e+01, -1.4683e+01, ..., -7.6838e+00,
-9.1892e+00, -4.3272e+00],
[-6.2346e+00, -8.6871e+00, -1.0300e+01, ..., -5.9481e+00,
-6.4484e+00, -3.1527e+00],
...,
[-1.7349e+00, -2.1829e+00, -1.7620e+00, ..., -1.8217e+00,
-3.1882e+00, -1.9457e+00],
[-2.8019e+00, -3.0163e+00, -1.9910e+00, ..., -2.1914e+00,
-3.3545e+00, -1.0091e+00],
[ 3.3024e-01, -2.9660e-01, 3.2088e-01, ..., 3.1283e-01,
4.7088e-01, 2.1211e-01]],
[[-6.2346e-01, 5.1914e-01, 8.7654e-01, ..., 8.2923e-01,
6.1113e-01, 1.8895e+00],
[-1.2235e+00, 3.4266e-01, 1.0150e+00, ..., 5.5641e-01,
-2.1495e-02, 1.2322e-01],
[-4.4582e+00, -4.3645e+00, -2.0330e+00, ..., 4.2411e-02,
1.3725e-02, -1.1084e-01],
...,
[-6.4460e-01, 3.4555e-02, 2.5326e-01, ..., 5.8626e-01,
-7.1227e-01, 3.3994e-01],
[-4.3318e-01, 5.6400e-02, 2.0307e-01, ..., 1.7200e-01,
2.2475e-01, 5.8195e-01],
[-1.5796e-01, -2.6847e-01, 1.3120e-01, ..., -1.8650e-02,
8.6167e-02, 3.3554e-01]],
...,
[[ 1.2594e+00, -3.4402e-01, -8.8372e-02, ..., 2.0981e-01,
-1.1186e-01, -2.1001e+00],
[-6.0398e-01, -8.5280e-01, -7.1082e-01, ..., -5.3312e-01,
-1.4872e+00, -2.0492e+00],
[ 1.1583e+00, 3.7664e-01, -6.4304e-01, ..., -3.3226e-01,
-5.9546e-01, -1.3818e+00],
...,
[-6.5191e-01, -5.1635e-01, -3.6610e-01, ..., -3.5164e+00,
-2.5465e+00, -1.0708e+00],
[-2.2740e+00, -1.8316e+00, -1.3152e+00, ..., -1.3145e+00,
-1.3880e+00, -1.1003e+00],
[-6.1769e-01, -8.4541e-01, -4.3669e-01, ..., -2.5199e-01,
-3.4518e-01, -3.5859e-01]],
[[ 7.7913e-02, 3.0020e-02, 4.4592e-01, ..., 4.7114e-01,
9.2452e-01, -3.0361e-02],
[ 8.2206e-01, -3.6976e-01, 5.3104e-01, ..., 4.2926e-01,
3.8050e-01, -8.0874e-01],
[-6.7010e-01, -1.2453e+00, 1.7742e-01, ..., 6.4372e-01,
5.2934e-01, -3.8555e-01],
...,
[-2.3739e-01, -4.8226e-01, 2.3723e-01, ..., -1.5986e+00,
-1.0875e-01, 1.2706e-01],
[-5.3446e-01, -8.7972e-01, 7.2579e-02, ..., 6.7284e-03,
-2.9978e-02, -1.3689e-01],
[ 3.1014e-01, -3.8017e-03, -1.8237e-01, ..., -1.4964e-01,
-1.1039e-01, 1.7519e-01]],
[[-1.5160e+00, -7.8540e-01, -4.1747e-01, ..., -4.3693e-01,
-2.8421e-01, 8.7264e-01],
[-2.1672e+00, -1.7705e+00, -1.7679e+00, ..., -7.9565e-01,
-2.1041e-01, 1.6636e+00],
[-2.2132e+00, -4.7885e+00, -4.9867e+00, ..., -3.8629e-01,
-1.1854e-01, 1.3915e+00],
...,
[-6.8073e-01, -1.9919e-01, -1.0413e-01, ..., 2.5239e+00,
2.4548e+00, 2.2269e+00],
[-1.2140e+00, -5.8274e-01, -3.1097e-01, ..., -3.1412e-01,
-1.8322e-01, 1.6238e-01],
[-3.7682e-01, -1.8379e-02, -1.7998e-01, ..., -2.5268e-01,
-2.3546e-01, 3.2181e-01]]],
[[[-1.8549e-01, -2.3874e-01, 1.9046e-01, ..., 5.5257e-02,
4.6365e-01, 1.4533e-01],
[-1.0870e-01, -1.5482e+00, -8.7648e-01, ..., -1.0755e+00,
-4.4606e-01, -3.3844e-01],
[-1.3620e-01, -1.3774e+00, -6.7436e-01, ..., -2.6527e-01,
-3.9549e-01, -6.6972e-01],
...,
[ 2.9982e-01, -1.9283e+00, -3.1791e+00, ..., -9.2557e-01,
-9.8901e-01, -2.0538e+00],
[-7.9847e-01, -1.3103e+00, -1.5968e+00, ..., -5.6248e-01,
-5.0096e-01, -1.8029e+00],
[-1.1745e+00, -1.9006e+00, -1.5440e+00, ..., -1.4463e+00,
-1.5662e+00, -9.8442e-01]],
[[-1.7708e-01, -1.4635e+00, -6.4079e-01, ..., -5.6925e-01,
-1.3540e+00, -2.7846e-01],
[-2.0468e+00, -4.0396e+00, -2.4047e+00, ..., -3.2156e+00,
-3.3049e+00, -1.1121e+00],
[-1.8970e+00, -3.3976e+00, -2.5101e+00, ..., -4.7556e+00,
-2.9130e+00, -1.1887e+00],
...,
[-4.8812e+00, -7.5743e+00, -8.4958e+00, ..., -5.5648e+00,
-6.2049e+00, -2.9467e+00],
[-6.2280e+00, -6.9804e+00, -6.4230e+00, ..., -6.1005e+00,
-7.4502e+00, -2.7461e+00],
[-1.8312e-01, -9.8542e-01, 2.0795e-01, ..., 4.0858e-01,
7.0078e-01, 1.7054e-01]],
[[-5.4488e-01, -1.8103e-01, 3.6948e-01, ..., 3.2629e-01,
2.5225e-02, 8.7337e-01],
[-3.9099e-01, -4.9593e-01, 4.5486e-01, ..., 4.6629e-01,
2.3662e-01, 4.6358e-01],
[-8.9732e-01, -2.4540e-01, 1.7592e-01, ..., 6.7553e-01,
1.8762e-01, 3.8394e-01],
...,
[-1.0286e+00, -2.1867e+00, -4.4502e+00, ..., 5.3448e-02,
3.3628e-02, 3.3043e-01],
[-8.0565e-01, -1.2662e+00, -1.8450e+00, ..., -7.1712e-02,
-2.8055e-02, 3.1766e-01],
[-7.0143e-01, -1.6593e-01, -1.8789e-01, ..., -3.5971e-05,
-1.0702e-01, -3.7293e-01]],
...,
[[ 4.6439e-02, -6.3528e-01, -3.8824e-01, ..., -6.5063e-02,
-2.4601e-01, -7.4931e-01],
[-8.8627e-01, -6.2770e-01, -5.5493e-01, ..., -2.1761e-01,
-9.8546e-01, -7.4662e-01],
[-5.3070e-01, -1.1097e-01, -3.1559e-02, ..., -1.3241e-01,
-2.1189e-01, -5.1636e-01],
...,
[-3.5598e-01, -1.5751e-01, -1.4721e+00, ..., -3.6422e-01,
-6.1876e-01, -1.2458e+00],
[-3.0400e+00, -1.9754e+00, -1.7578e+00, ..., -1.9434e+00,
-2.0465e+00, -2.0924e+00],
[-7.8499e-01, -1.9536e+00, -1.5848e+00, ..., -8.8775e-01,
-9.5610e-01, -1.3488e+00]],
[[-1.8036e-01, 2.9504e-02, 4.1476e-01, ..., 9.7568e-02,
5.3154e-01, 1.6182e-01],
[-4.6145e-01, -2.4187e-01, -2.3096e-03, ..., -7.4282e-01,
8.6659e-02, -3.6129e-02],
[-6.9296e-01, -5.9640e-01, -2.0074e-02, ..., -6.4552e-01,
6.8521e-03, -2.0758e-02],
...,
[ 2.9601e-01, -9.8393e-01, -2.4998e+00, ..., 3.1858e-01,
1.9162e-01, -7.5475e-01],
[-3.7008e-01, -1.9821e+00, -1.7878e+00, ..., 8.8728e-03,
-3.9843e-02, -9.7843e-01],
[ 5.7548e-01, -9.3084e-02, -5.6535e-01, ..., 2.4953e-01,
3.1014e-01, 3.1849e-01]],
[[-6.8267e-01, 4.0965e-01, 5.7792e-01, ..., 4.7083e-01,
3.9130e-01, 5.9870e-01],
[-8.6239e-01, 2.8491e-01, -2.4761e-01, ..., -3.0686e-01,
-1.9957e-01, 6.3825e-01],
[-4.7970e-01, 3.4053e-01, 2.7840e-02, ..., 2.0582e-01,
5.1550e-02, 6.0278e-01],
...,
[-1.7762e+00, -9.1640e-01, -1.6190e+00, ..., -4.0031e-01,
-1.4308e-02, 8.4584e-01],
[-2.3960e+00, -1.8384e+00, -1.2240e+00, ..., -8.4538e-01,
-4.4036e-01, 4.3811e-01],
[-1.2450e+00, -8.5013e-01, 2.7326e-01, ..., -3.1838e-01,
-1.2998e-01, 1.1317e+00]]],
[[[-3.7463e-03, -2.3689e-01, -3.5742e-02, ..., -9.5928e-01,
-5.2767e-01, -1.4212e+00],
[ 2.6775e-01, -9.2001e-01, -6.8325e-01, ..., -1.1593e+00,
-1.3123e+00, -1.4834e+00],
[ 2.2047e-01, -5.6498e-01, -4.7018e-01, ..., -8.8996e-01,
-1.0787e+00, -1.8072e+00],
...,
[-1.4519e-02, -5.8906e-01, -4.5333e-01, ..., -9.0891e-01,
-1.0842e+00, -1.9407e+00],
[-6.5089e-01, -6.9552e-01, -7.5729e-02, ..., -4.4117e-01,
-6.0559e-01, -1.7179e+00],
[-7.8853e-01, -1.3524e+00, -1.1915e+00, ..., -1.1040e+00,
-1.5524e+00, -7.8744e-01]],
[[-7.4772e-01, -2.1010e+00, -1.8378e+00, ..., -3.9230e+00,
-5.4498e+00, -1.8435e+00],
[-1.9619e+00, -3.6260e+00, -2.9544e+00, ..., -5.9000e+00,
-8.0556e+00, -3.4053e+00],
[-1.5854e+00, -2.5463e+00, -2.5069e+00, ..., -5.0895e+00,
-6.1917e+00, -2.5735e+00],
...,
[-1.7666e+00, -2.4247e+00, -2.4692e+00, ..., -4.9187e+00,
-6.1981e+00, -2.4789e+00],
[-3.8406e+00, -4.4228e+00, -3.6446e+00, ..., -6.3457e+00,
-7.9447e+00, -2.6281e+00],
[ 2.5116e-01, -3.4203e-01, 3.5131e-01, ..., 4.5118e-01,
4.9255e-01, 1.4592e-01]],
[[-2.5015e-01, 1.5979e-01, 7.1751e-01, ..., 6.4434e-01,
2.7479e-01, 1.5265e+00],
[ 1.0401e-03, -3.4289e-01, 5.5254e-01, ..., 5.7753e-01,
-5.1073e-02, 9.3572e-02],
[-1.3180e-01, -2.1231e-01, 2.1801e-01, ..., 3.4598e-02,
2.2318e-02, -1.1449e-01],
...,
[-2.3569e-01, -2.7121e-01, 1.1861e-01, ..., -2.2913e-03,
-1.7163e-01, -2.1295e-01],
[-2.6718e-01, -2.2198e-01, 1.5973e-01, ..., -1.2179e-01,
-7.8962e-02, -2.5677e-01],
[-1.2771e-01, -5.2450e-01, 8.0507e-02, ..., 8.8666e-02,
1.3519e-01, -6.8377e-01]],
...,
[[ 6.3355e-01, -2.0501e-01, -1.2455e-02, ..., 4.3886e-01,
3.3399e-01, -1.3097e+00],
[-4.9334e-01, -5.7075e-01, -5.2133e-01, ..., -1.6589e-01,
-7.9898e-01, -1.3969e+00],
[-3.8821e-01, -2.3874e-01, -2.4681e-01, ..., -9.8883e-02,
-8.0661e-02, -9.4508e-01],
...,
[-4.3795e-01, -1.4955e-01, -1.6661e-01, ..., -7.9329e-02,
-6.2624e-02, -9.9750e-01],
[-2.3460e+00, -1.7078e+00, -1.2400e+00, ..., -1.5256e+00,
-1.5236e+00, -1.8146e+00],
[-4.8380e-01, -8.3818e-01, -4.1105e-01, ..., -3.4148e-01,
-5.2620e-01, -1.2004e+00]],
[[ 1.5125e-01, 2.6235e-01, 6.4644e-01, ..., -7.7132e-02,
3.8528e-01, -3.7407e-01],
[ 1.4450e-01, 1.9516e-01, 4.9019e-01, ..., 1.9657e-01,
5.9683e-02, -8.4874e-01],
[-1.1762e-01, 9.2984e-02, 2.9320e-01, ..., 2.5696e-01,
1.0655e-01, -4.7791e-01],
...,
[-2.0237e-01, 5.3503e-02, 2.9811e-01, ..., 2.6960e-01,
2.0445e-01, -4.9478e-01],
[-5.7636e-01, -7.0753e-01, -3.5450e-02, ..., -1.2344e-01,
-2.3147e-01, -9.2872e-01],
[ 5.0648e-01, 4.1990e-01, 8.8044e-02, ..., 3.4418e-01,
3.4157e-01, 3.0635e-01]],
[[-8.4728e-01, 2.3306e-01, 3.3777e-01, ..., -5.7460e-02,
-6.2754e-02, 5.8257e-01],
[-1.1237e+00, 1.5129e-01, -3.4667e-01, ..., -8.9576e-01,
-4.4836e-01, 8.7930e-01],
[-9.5750e-01, 3.3919e-01, -1.1301e-01, ..., -5.4975e-01,
-5.2006e-01, 6.3582e-01],
...,
[-1.0178e+00, 1.9806e-01, -1.0040e-01, ..., -5.9267e-01,
-5.0608e-01, 5.9314e-01],
[-1.7307e+00, -3.9609e-01, -3.5391e-01, ..., -7.3026e-01,
-6.2525e-01, 3.8819e-01],
[-7.7549e-01, -8.7671e-02, -2.9224e-01, ..., -5.1580e-01,
-4.3767e-01, 9.8964e-01]]],
[[[-1.9491e-01, -2.3479e-01, 2.0625e-01, ..., 7.5927e-02,
5.8119e-01, 2.4260e-01],
[-6.2051e-02, -1.4198e+00, -8.4585e-01, ..., -7.3250e-01,
-4.7001e-01, -2.8718e-01],
[ 1.8846e-03, -1.2184e+00, -6.8071e-01, ..., -5.4542e-01,
-4.6699e-01, -6.8365e-01],
...,
[-2.4080e-01, -1.6693e+00, -1.1654e+00, ..., -9.4922e-01,
-9.7226e-01, -2.0268e+00],
[-1.1007e+00, -1.7204e+00, -7.8544e-01, ..., -5.3108e-01,
-4.4762e-01, -1.7638e+00],
[-1.2925e+00, -2.1677e+00, -1.6050e+00, ..., -1.4266e+00,
-1.5386e+00, -9.6790e-01]],
[[-1.0808e-01, -1.3842e+00, -6.7786e-01, ..., -5.2839e-01,
-1.3812e+00, -2.1067e-01],
[-1.9544e+00, -3.9082e+00, -2.5124e+00, ..., -2.4231e+00,
-3.5780e+00, -1.2599e+00],
[-1.8032e+00, -3.2657e+00, -2.6179e+00, ..., -2.5940e+00,
-3.0160e+00, -1.2338e+00],
...,
[-3.7284e+00, -5.5068e+00, -5.2083e+00, ..., -5.4058e+00,
-6.1452e+00, -2.8935e+00],
[-6.1718e+00, -7.0690e+00, -6.1131e+00, ..., -6.0945e+00,
-7.3875e+00, -2.6883e+00],
[-1.8954e-02, -9.1105e-01, 2.5666e-01, ..., 4.2136e-01,
7.2157e-01, 2.0892e-01]],
[[-4.9307e-01, -1.7843e-01, 3.6449e-01, ..., 3.8169e-01,
5.5034e-02, 9.8125e-01],
[-3.6556e-01, -5.0021e-01, 3.7207e-01, ..., 3.9301e-01,
2.1004e-01, 4.8036e-01],
[-7.6065e-01, -2.1033e-01, 1.0387e-01, ..., 4.3066e-02,
1.5910e-01, 3.7308e-01],
...,
[-1.4024e+00, 1.3133e-02, 2.7621e-01, ..., 4.8414e-02,
2.2480e-02, 2.8066e-01],
[-1.2392e+00, 1.2157e-01, -2.1826e-02, ..., -5.4436e-03,
3.1185e-02, 3.1024e-01],
[-1.1735e+00, -7.8003e-02, 6.9034e-02, ..., 4.8027e-03,
-4.9234e-02, -3.1694e-01]],
...,
[[ 1.6815e-01, -5.5447e-01, -3.8302e-01, ..., -9.0897e-02,
-3.0655e-01, -8.0726e-01],
[-7.3459e-01, -6.6943e-01, -5.8698e-01, ..., -3.4557e-01,
-1.1957e+00, -8.7022e-01],
[-3.4385e-01, -1.6170e-01, -8.5516e-02, ..., 9.1650e-02,
-1.9409e-01, -5.7168e-01],
...,
[-5.8331e-01, -6.2757e-01, -5.0246e-01, ..., -3.7963e-01,
-5.9887e-01, -1.2025e+00],
[-3.1453e+00, -2.8989e+00, -2.0417e+00, ..., -1.8746e+00,
-2.0329e+00, -2.0556e+00],
[-5.2359e-01, -1.6394e+00, -1.1222e+00, ..., -8.4182e-01,
-9.2987e-01, -1.3346e+00]],
[[-1.2047e-01, 9.7847e-02, 4.4570e-01, ..., 1.3221e-01,
5.8668e-01, 2.0817e-01],
[-3.4580e-01, -2.2042e-01, 3.4186e-02, ..., -4.7253e-02,
1.3291e-01, -1.4980e-01],
[-5.5115e-01, -4.6608e-01, -4.7275e-03, ..., -9.6079e-03,
-1.7373e-02, -4.6731e-02],
...,
[-9.8321e-02, -1.0245e+00, 8.7977e-02, ..., 2.6556e-01,
1.2989e-01, -7.4493e-01],
[-4.9007e-01, -1.8462e+00, -2.3568e-01, ..., 3.0356e-02,
-4.9820e-02, -9.4102e-01],
[ 4.5427e-01, -1.3629e-01, 9.6555e-02, ..., 2.7531e-01,
3.1850e-01, 3.2579e-01]],
[[-6.4104e-01, 4.1731e-01, 5.4856e-01, ..., 5.1065e-01,
4.1151e-01, 6.6295e-01],
[-7.6400e-01, 2.7470e-01, -2.4545e-01, ..., -2.8848e-01,
-1.9058e-01, 6.7672e-01],
[-4.3870e-01, 3.3230e-01, 1.5762e-02, ..., 8.3166e-02,
4.1598e-02, 5.4909e-01],
...,
[-1.0453e+00, -9.3906e-01, -5.3451e-01, ..., -4.0403e-01,
-8.0338e-02, 8.1153e-01],
[-1.8373e+00, -1.8288e+00, -9.8199e-01, ..., -7.9775e-01,
-4.5784e-01, 4.2378e-01],
[-9.6269e-01, -9.9503e-01, -2.7026e-01, ..., -2.9624e-01,
-1.4264e-01, 1.1251e+00]]]], device='cuda:0'),
tensor([[[[-3.7849e-01, -1.0172e+00, -8.2998e-01, ..., -2.0944e-01,
6.3202e-01, 5.4904e-01],
[ 9.8486e-02, -1.7657e+00, -1.2604e+00, ..., -6.5096e-01,
-2.3026e-01, 1.2921e-01],
[ 6.2920e-01, -3.8096e-01, -7.4486e-01, ..., -6.3468e-01,
-1.0443e-01, -8.6163e-02],
...,
[ 6.3584e-02, -1.3380e+00, -9.2042e-01, ..., -3.1374e-01,
-7.0639e-02, -1.0925e-01],
[-2.1515e-01, -1.0828e+00, -4.5383e-01, ..., 1.3941e-02,
2.7863e-01, -1.2822e-02],
[-4.0858e-01, -1.2747e+00, -1.1312e+00, ..., -1.0393e+00,
-1.0792e+00, -3.7941e-01]],
[[-2.1754e+00, -4.0729e+00, -3.8084e+00, ..., -1.3449e+00,
-9.7222e-01, 1.6974e-02],
[-4.5458e+00, -6.2847e+00, -5.8929e+00, ..., -3.0056e+00,
-2.6406e+00, -6.6376e-01],
[-3.3190e+00, -3.8318e+00, -5.1023e+00, ..., -1.9427e+00,
-1.4928e+00, -3.7813e-01],
...,
[-3.1727e+00, -4.7030e+00, -4.6001e+00, ..., -1.2538e+00,
-1.4874e+00, -3.5015e-01],
[-5.5026e+00, -6.7247e+00, -6.0310e+00, ..., -2.4986e+00,
-3.3293e+00, -9.6742e-01],
[ 6.6613e-01, -3.2019e-01, 4.6546e-01, ..., 3.3895e-01,
3.6748e-01, 7.9979e-02]],
[[-5.6815e-01, 4.4880e-01, 7.6116e-01, ..., 9.0320e-01,
3.0549e-01, 1.0834e+00],
[-9.7441e-01, 1.3296e-01, 6.1663e-01, ..., 5.6450e-01,
5.9956e-02, 4.8136e-01],
[-1.5235e+00, 1.4141e-01, 1.8011e-01, ..., 2.1608e-01,
1.0924e-01, 4.1563e-01],
...,
[-1.4395e+00, 4.7671e-02, 5.8453e-03, ..., 6.4796e-02,
9.4473e-02, 4.2545e-01],
[-1.3583e+00, 1.4388e-01, -2.3517e-02, ..., 1.3059e-01,
1.2130e-01, 5.9410e-01],
[-5.7892e-01, 2.3831e-01, 1.2505e-01, ..., -6.6142e-02,
8.2596e-02, 3.6508e-01]],
...,
[[ 7.9046e-01, 1.7072e-01, 3.0618e-01, ..., -1.6538e-01,
-3.9007e-01, -9.2650e-01],
[-4.6475e-01, -1.8341e-01, 4.6782e-02, ..., -1.0357e+00,
-1.2778e+00, -8.2150e-01],
[ 3.7949e-02, -3.7940e-01, -2.3334e-01, ..., -4.8788e-01,
-1.9733e-01, -3.8166e-01],
...,
[-7.0731e-01, -5.0829e-01, -8.3011e-02, ..., -9.9457e-02,
-1.9068e-01, -3.6549e-01],
[-2.4936e+00, -1.9320e+00, -1.3307e+00, ..., -8.2190e-01,
-8.8978e-01, -7.9054e-01],
[-4.7237e-01, -1.2126e+00, -3.3724e-01, ..., 2.7644e-03,
-1.2781e-01, -3.2371e-01]],
[[ 5.1680e-03, -2.5115e-01, -8.5751e-02, ..., 7.5059e-02,
7.0211e-01, 3.0329e-01],
[ 3.3022e-01, -1.0268e+00, -3.2283e-01, ..., 1.0371e-01,
1.1163e-01, -4.6649e-02],
[ 2.1360e-01, -2.3739e-01, -1.1392e-02, ..., -7.2887e-02,
-7.2481e-02, 1.9817e-01],
...,
[-1.5423e-01, -1.0662e+00, 4.1544e-02, ..., -1.1774e-01,
-9.9259e-02, 1.7409e-01],
[-6.8463e-01, -2.1348e+00, -4.1275e-01, ..., -4.0470e-01,
-3.4154e-01, -7.3156e-02],
[ 7.5999e-01, -7.4476e-02, 2.2353e-01, ..., -2.4226e-01,
-2.1632e-01, 3.1307e-01]],
[[-5.7009e-01, -1.0872e-01, -1.1533e-02, ..., 4.2757e-01,
3.7313e-01, 5.4622e-01],
[-1.1389e+00, -7.6906e-01, -8.7939e-01, ..., 1.9841e-01,
-1.5206e-01, 4.1280e-01],
[-1.0314e+00, -8.0209e-01, -5.0388e-01, ..., 8.0159e-01,
-1.6664e-01, 9.4938e-02],
...,
[-6.3243e-01, -8.0829e-01, -5.1898e-01, ..., 4.6988e-02,
-8.1923e-02, 9.8762e-02],
[-1.0308e+00, -1.1580e+00, -6.4548e-01, ..., -1.1447e-01,
-2.6324e-01, -1.9575e-01],
[-6.3278e-01, -9.7160e-01, -4.2678e-01, ..., -1.1660e-01,
-2.7896e-01, 1.2228e-01]]],
[[[-1.3143e-01, -5.9954e-01, -3.4819e-01, ..., -1.3668e+00,
-8.2501e-01, -1.5613e+00],
[ 3.9095e-02, -1.3247e+00, -8.7682e-01, ..., -9.8978e-01,
-5.6524e-01, -9.0870e-01],
[-9.2731e-01, -6.0800e+00, -7.0532e+00, ..., -5.4248e+00,
-7.4801e-01, -9.3349e-01],
...,
[-1.0202e+00, -5.4208e+00, -1.0224e+00, ..., 8.3306e-01,
-1.1662e+00, -2.9945e+00],
[-1.6550e+00, -4.7947e+00, -2.9910e+00, ..., -4.3852e-01,
1.7970e+00, -2.4846e+00],
[-2.6086e+00, -3.4854e+00, -7.7077e+00, ..., -8.3291e+00,
-3.6319e+00, -1.3044e+00]],
[[-1.4812e+00, -3.0801e+00, -2.9782e+00, ..., -7.6391e+00,
-7.2085e+00, -2.7661e+00],
[-2.9179e+00, -4.8106e+00, -4.5986e+00, ..., -1.2598e+01,
-1.1452e+01, -5.4240e+00],
[-1.1237e+01, -1.7525e+01, -2.0294e+01, ..., -1.4957e+01,
-1.2261e+01, -5.6275e+00],
...,
[-5.8715e+00, -7.3683e+00, -6.6556e+00, ..., -7.5271e+00,
-1.0871e+01, -2.9171e+00],
[-6.3797e+00, -5.4256e+00, -1.1226e+01, ..., -1.0033e+01,
-1.1765e+01, -2.4828e+00],
[-2.1113e+00, -1.7664e+00, 1.2427e+00, ..., 4.9734e-01,
-7.5511e-01, 3.3387e-01]],
[[-2.8458e-01, 2.4497e-01, 7.3811e-01, ..., 1.1553e+00,
3.4758e-01, 1.8662e+00],
[-1.9663e-01, -2.7192e-01, 4.4966e-01, ..., -9.7709e-01,
4.9488e-01, 6.6050e-01],
[-3.3154e+00, -2.2138e+00, 1.6367e+00, ..., -3.5061e+00,
6.3209e-01, 1.2894e+00],
...,
[-4.2410e+00, -7.8591e+00, -1.2432e+01, ..., -1.2267e+00,
-1.7233e-01, -2.1144e-01],
[-2.7388e+00, -1.2987e+01, -1.5027e+01, ..., -3.0213e+00,
1.0632e+00, -3.2754e-01],
[-3.5673e+00, -1.3176e+01, -1.6108e+01, ..., -8.5978e+00,
5.2330e-01, -8.9298e-01]],
...,
[[ 7.8874e-01, 2.4955e-01, 2.9242e-01, ..., 1.0542e+00,
1.0977e+00, -1.0333e+00],
[-4.4007e-01, -3.9948e-01, -2.6394e-01, ..., -7.3986e-01,
-1.2890e+00, -1.5929e+00],
[ 5.8369e-01, 4.2127e-01, -6.4003e-02, ..., -2.1801e+00,
-1.7713e+00, -1.5901e+00],
...,
[ 1.7603e+00, -3.0093e-01, -8.0290e+00, ..., -7.8643e+00,
-2.9633e+00, -8.9867e-01],
[-8.5155e-01, -4.3638e+00, -1.4545e+01, ..., -1.6527e+01,
-6.9672e+00, -2.2744e+00],
[-8.1377e-01, -2.9333e+00, -6.3575e+00, ..., -7.8164e+00,
-5.0193e+00, -1.4421e+00]],
[[ 1.1440e-01, 1.7092e-01, 4.4557e-01, ..., 2.6466e-02,
-2.5037e-01, -6.4807e-01],
[ 1.4291e-01, -1.0211e-01, 2.4074e-01, ..., 7.8723e-01,
-2.4244e-01, -1.1157e+00],
[ 6.5782e-01, -1.1776e+00, -2.7137e+00, ..., -3.2272e+00,
-7.0388e-01, -2.2710e-01],
...,
[-1.2084e-01, -3.9007e+00, -6.3366e+00, ..., 1.2370e+00,
-1.6068e+00, -8.4706e-01],
[-8.3387e-01, -5.1045e+00, -5.6135e+00, ..., -2.9532e+00,
-8.4310e-01, -7.2178e-01],
[-4.7984e-01, -3.7698e+00, -6.8395e+00, ..., -2.2501e-01,
3.1941e-01, 5.6498e-01]],
[[-7.4028e-01, 8.2609e-02, 3.2371e-01, ..., -5.4603e-01,
1.4232e-02, 6.7245e-01],
[-1.2199e+00, -1.2857e-01, -2.4570e-01, ..., -3.8934e+00,
2.3881e-01, 1.2937e+00],
[-2.0192e+00, -1.7742e+00, -1.3758e+00, ..., -7.2968e+00,
-3.4943e-01, 1.1236e+00],
...,
[-3.8234e+00, -9.6667e+00, -1.1602e+01, ..., -4.1013e+00,
1.3700e+00, 1.7423e+00],
[-3.1738e+00, -9.0558e+00, -1.2939e+01, ..., -2.8151e+00,
-1.3104e+00, 4.0727e-01],
[-4.9335e-02, -2.4708e+00, -6.8630e+00, ..., -1.8300e+00,
-7.1059e-01, 1.1278e+00]]],
[[[ 3.4447e-02, -2.3346e-01, -2.7106e-02, ..., -9.5228e-01,
-4.5315e-01, -1.3492e+00],
[ 3.0860e-01, -9.0638e-01, -6.8828e-01, ..., -1.1349e+00,
-1.2758e+00, -1.4152e+00],
[ 2.5036e-01, -5.4669e-01, -4.8276e-01, ..., -8.6051e-01,
-1.0247e+00, -1.8590e+00],
...,
[-2.7809e-02, -5.8871e-01, -2.9686e-01, ..., -9.0417e-01,
-9.9731e-01, -1.8978e+00],
[-7.0557e-01, -7.1703e-01, -5.8758e-02, ..., -4.5739e-01,
-5.7643e-01, -1.6981e+00],
[-8.0260e-01, -1.3588e+00, -1.2079e+00, ..., -9.5205e-01,
-1.3496e+00, -6.5428e-01]],
[[-7.2219e-01, -2.1626e+00, -1.8620e+00, ..., -3.8398e+00,
-5.4285e+00, -1.7728e+00],
[-1.9331e+00, -3.6716e+00, -2.9456e+00, ..., -5.9316e+00,
-8.2116e+00, -3.4355e+00],
[-1.5525e+00, -2.5185e+00, -2.5385e+00, ..., -5.0692e+00,
-6.3733e+00, -2.6135e+00],
...,
[-1.7617e+00, -2.3567e+00, -2.4759e+00, ..., -5.0129e+00,
-6.2958e+00, -2.4879e+00],
[-3.8653e+00, -4.4436e+00, -3.6484e+00, ..., -6.0568e+00,
-7.5194e+00, -2.3684e+00],
[ 2.4862e-01, -3.5761e-01, 3.4332e-01, ..., 4.1810e-01,
4.4304e-01, 2.5554e-01]],
[[-2.1163e-01, 1.7970e-01, 7.3171e-01, ..., 6.9407e-01,
2.8339e-01, 1.6000e+00],
[ 2.6408e-02, -3.9105e-01, 5.4440e-01, ..., 5.8345e-01,
-1.0705e-01, 1.3848e-01],
[-1.2264e-01, -2.1353e-01, 2.0018e-01, ..., 3.9188e-02,
-1.7814e-02, -1.4972e-01],
...,
[-2.4017e-01, -2.0620e-01, 3.3656e-01, ..., 1.2250e-02,
-7.2347e-02, -1.4349e-01],
[-2.4847e-01, -2.0748e-01, 1.6949e-01, ..., -8.5353e-02,
3.2611e-02, -1.5447e-01],
[-1.4092e-01, -4.8913e-01, 1.1053e-01, ..., 7.5554e-02,
2.3322e-01, -5.5936e-01]],
...,
[[ 6.4511e-01, -2.5963e-01, -1.1969e-02, ..., 3.9703e-01,
2.4477e-01, -1.3564e+00],
[-4.7861e-01, -5.6361e-01, -5.1688e-01, ..., -1.8451e-01,
-8.7662e-01, -1.4206e+00],
[-3.6760e-01, -2.1528e-01, -2.5871e-01, ..., -1.0521e-01,
-8.8549e-02, -9.4078e-01],
...,
[-4.4121e-01, -1.7494e-01, -3.0990e-01, ..., -1.2903e-01,
-1.0970e-01, -9.0996e-01],
[-2.3901e+00, -1.7531e+00, -1.2390e+00, ..., -1.3829e+00,
-1.4271e+00, -1.7341e+00],
[-4.6628e-01, -8.4243e-01, -4.0607e-01, ..., -3.5293e-01,
-4.9755e-01, -1.1314e+00]],
[[ 1.8639e-01, 2.7980e-01, 6.3075e-01, ..., -9.4767e-02,
4.1912e-01, -3.6556e-01],
[ 1.5728e-01, 2.4766e-01, 4.7826e-01, ..., 1.7393e-01,
8.5768e-02, -8.3357e-01],
[-1.1482e-01, 1.1507e-01, 2.9841e-01, ..., 2.6717e-01,
1.3348e-01, -5.5793e-01],
...,
[-2.0313e-01, 2.1123e-02, 3.6608e-01, ..., 2.4329e-01,
1.2610e-01, -5.0993e-01],
[-5.7195e-01, -7.3157e-01, -7.5563e-02, ..., -6.5200e-02,
-1.7702e-01, -8.2955e-01],
[ 5.1827e-01, 3.9219e-01, 5.8747e-02, ..., 3.9443e-01,
3.6787e-01, 3.2125e-01]],
[[-8.2931e-01, 2.7145e-01, 3.3631e-01, ..., -1.2340e-02,
-3.9170e-02, 6.2035e-01],
[-1.1385e+00, 1.9656e-01, -3.6236e-01, ..., -9.1288e-01,
-3.8934e-01, 9.4069e-01],
[-9.5888e-01, 3.5458e-01, -1.4494e-01, ..., -5.5580e-01,
-5.1309e-01, 6.0660e-01],
...,
[-9.6414e-01, 2.0208e-01, -1.8227e-01, ..., -5.2683e-01,
-5.3746e-01, 6.1069e-01],
[-1.7095e+00, -4.1300e-01, -3.7256e-01, ..., -6.2980e-01,
-6.7303e-01, 4.2309e-01],
[-7.2112e-01, -8.9611e-02, -3.0239e-01, ..., -4.1115e-01,
-3.8276e-01, 9.9644e-01]]],
[[[-5.1779e-01, -2.2901e+00, -6.2053e-01, ..., -9.6749e-01,
-4.6024e-01, -1.3600e+00],
[-2.6220e+00, -2.9867e+00, -3.6122e-02, ..., -1.1707e+00,
-1.3571e+00, -1.4977e+00],
[-1.3558e+00, -1.1059e+00, -1.3450e+00, ..., -1.0342e+00,
-1.2598e+00, -2.0078e+00],
...,
[-3.4769e-02, -1.5688e+00, -9.5269e-01, ..., -1.1574e+00,
-6.7447e-01, -2.1352e+00],
[-2.8536e-01, -8.7491e-01, -3.6477e-01, ..., -4.4401e-01,
-5.6628e-01, -1.7118e+00],
[-4.8784e-01, -1.0880e+00, -1.0475e+00, ..., -9.8770e-01,
-1.3844e+00, -6.8043e-01]],
[[-3.7632e+00, -5.0024e+00, -4.5409e+00, ..., -3.8072e+00,
-5.4084e+00, -1.7532e+00],
[-6.8113e+00, -8.2820e+00, -7.4482e+00, ..., -6.3904e+00,
-8.5695e+00, -3.6074e+00],
[-4.8325e+00, -7.0277e+00, -5.6396e+00, ..., -9.4585e+00,
-9.6021e+00, -4.5466e+00],
...,
[-2.3420e+00, -3.4363e+00, -2.7117e+00, ..., -6.0989e+00,
-6.9499e+00, -2.8874e+00],
[-3.8886e+00, -4.5350e+00, -3.0697e+00, ..., -6.1200e+00,
-7.5670e+00, -2.3941e+00],
[ 4.4330e-01, -5.9502e-01, 2.6597e-01, ..., 3.9250e-01,
4.3763e-01, 2.5140e-01]],
[[-1.3876e+00, -1.4778e+00, -6.6931e-01, ..., 6.8449e-01,
2.6466e-01, 1.5890e+00],
[-2.6055e+00, -5.8019e+00, -6.3517e+00, ..., 5.6950e-01,
-1.7243e-01, 1.4143e-01],
[-1.0079e+01, -1.1284e+01, -8.5089e+00, ..., -5.6239e-01,
3.7269e-01, 3.0767e-01],
...,
[-1.2533e+00, -1.3170e+00, -9.2461e-01, ..., 1.6850e-01,
-1.0229e+00, -1.8460e-01],
[-1.0265e+00, -3.2302e-01, -8.4993e-01, ..., -6.6656e-02,
-1.4551e-01, -1.6339e-01],
[-5.4692e-01, -1.1278e-01, -7.1595e-01, ..., 6.4877e-02,
1.8588e-01, -6.4784e-01]],
...,
[[ 1.2168e+00, 1.0881e+00, -5.8226e-01, ..., 3.9794e-01,
2.5717e-01, -1.3663e+00],
[-7.0080e-01, -3.3970e+00, -5.9290e+00, ..., -1.5267e-01,
-8.0558e-01, -1.3776e+00],
[-2.6738e+00, -7.8277e+00, -8.7827e+00, ..., 6.5249e-01,
2.2543e-01, -8.2095e-01],
...,
[-5.8260e-01, -1.2175e-01, -1.0630e+00, ..., -1.4736e+00,
-1.1774e+00, -1.2871e+00],
[-2.3038e+00, -1.2537e+00, -1.2925e+00, ..., -1.6151e+00,
-1.6350e+00, -1.7932e+00],
[-8.5403e-01, -1.2163e+00, 3.9907e-01, ..., -3.4312e-01,
-4.6928e-01, -1.1496e+00]],
[[ 3.6902e-01, -1.1162e+00, -3.9236e-01, ..., -8.1946e-02,
4.2533e-01, -3.3389e-01],
[-1.1112e+00, -5.0617e+00, -3.3037e+00, ..., 5.6315e-02,
-1.0957e-01, -9.4143e-01],
[-4.5113e+00, -7.1365e+00, -5.2300e+00, ..., -9.8377e-02,
-4.3127e-01, -1.5541e+00],
...,
[-6.4378e-01, -1.2579e+00, -2.8189e-01, ..., -4.6668e-02,
4.0293e-01, -7.2031e-01],
[-1.1827e+00, -1.9742e+00, -2.7835e-01, ..., -1.2986e-01,
-2.9314e-01, -8.8742e-01],
[ 2.9729e-01, -4.7919e-01, -3.8715e-01, ..., 3.8136e-01,
3.4983e-01, 3.3364e-01]],
[[-1.2408e+00, -1.3966e+00, -9.3839e-01, ..., -4.6057e-02,
-6.6713e-02, 6.2542e-01],
[-3.6952e+00, -6.4988e+00, -6.7996e+00, ..., -9.1968e-01,
-3.2780e-01, 1.0007e+00],
[-5.0360e+00, -8.6893e+00, -9.0330e+00, ..., -1.0311e+00,
-2.5821e-01, 9.8484e-01],
...,
[-2.8531e-01, 4.2544e-02, -1.0684e+00, ..., -1.2208e+00,
-2.3367e-01, 1.2474e+00],
[-6.2212e-01, -2.7028e-01, 1.4618e-01, ..., -6.8140e-01,
-5.5286e-01, 5.0657e-01],
[-4.6522e-02, 1.1425e-01, 7.7641e-01, ..., -4.3608e-01,
-4.3187e-01, 1.0320e+00]]],
[[[-7.3180e-01, -1.4741e+00, -1.4321e+00, ..., -1.2411e+00,
-8.4069e-01, -1.6774e+00],
[-3.3062e-01, -2.4782e+00, -2.2713e+00, ..., -1.7669e+00,
-1.9047e+00, -1.9302e+00],
[-1.3995e-01, -1.9066e+00, -1.4742e+00, ..., -1.2490e+00,
-1.4481e+00, -2.1248e+00],
...,
[-2.3256e-01, -1.3100e+00, -5.2326e-01, ..., -5.5866e-01,
-4.5231e-01, -8.9228e-01],
[-6.0466e-01, -1.3260e+00, -1.3812e+00, ..., -2.2396e-01,
-5.9600e-02, -9.6321e-01],
[-5.6470e-01, -1.3764e+00, -6.3932e-01, ..., -8.5683e-01,
-1.0396e+00, -5.9797e-01]],
[[-2.6905e+00, -5.0718e+00, -4.5949e+00, ..., -4.7826e+00,
-5.9907e+00, -2.3800e+00],
[-5.3424e+00, -8.1107e+00, -7.4761e+00, ..., -7.7983e+00,
-9.2806e+00, -4.3770e+00],
[-4.0236e+00, -5.8287e+00, -8.0410e+00, ..., -6.0506e+00,
-6.4583e+00, -3.2305e+00],
...,
[-2.1019e+00, -3.4143e+00, -8.1324e+00, ..., -2.9647e+00,
-3.3914e+00, -1.4803e+00],
[-4.9749e+00, -6.2298e+00, -7.2104e+00, ..., -4.9234e+00,
-6.0283e+00, -2.3102e+00],
[ 4.8947e-01, -6.1014e-01, 1.1836e+00, ..., 4.8629e-01,
5.7721e-01, 1.3877e-02]],
[[-6.6680e-01, 4.3969e-01, 8.5943e-01, ..., 8.2105e-01,
6.2122e-01, 1.8945e+00],
[-9.5527e-01, 1.4784e-02, 4.1716e-01, ..., 5.4269e-01,
2.5555e-02, 1.3252e-01],
[-1.7165e+00, -2.4244e-01, -8.0992e-02, ..., 5.6415e-03,
2.3410e-02, -1.0829e-01],
...,
[-1.0518e+00, -1.7557e-01, -7.9642e-01, ..., 1.7988e-02,
8.2665e-02, 3.7439e-01],
[-1.0349e+00, -6.0893e-02, 1.6983e-01, ..., 2.4943e-02,
1.0493e-01, 4.0703e-01],
[-5.1259e-01, -4.0907e-02, 1.0500e+00, ..., -1.6828e-02,
8.9382e-02, 2.5923e-02]],
...,
[[ 1.2179e+00, -3.6796e-01, -4.8074e-02, ..., 2.3270e-01,
-1.0976e-01, -2.1431e+00],
[-7.4825e-01, -9.0918e-01, -4.8044e-01, ..., -5.2022e-01,
-1.4976e+00, -2.0754e+00],
[-7.0573e-01, -8.3085e-01, -1.4984e-01, ..., -3.2520e-01,
-6.0226e-01, -1.3847e+00],
...,
[-5.6857e-01, -9.5209e-02, 1.0875e+00, ..., 8.6731e-02,
-1.3361e-01, -5.1783e-01],
[-2.4849e+00, -1.6891e+00, -1.1088e-01, ..., -8.8273e-01,
-1.0729e+00, -1.0561e+00],
[-3.9183e-01, -1.0855e+00, -5.9217e-01, ..., -1.7249e-01,
-3.1896e-01, -5.9553e-01]],
[[ 2.8246e-02, 6.0541e-03, 4.7225e-01, ..., 4.6817e-01,
9.3246e-01, -5.4643e-02],
[ 2.1770e-01, -8.2993e-01, -4.9412e-02, ..., 4.2572e-01,
3.8883e-01, -8.8135e-01],
[ 1.3308e-01, -6.4353e-01, 7.0932e-01, ..., 6.9394e-01,
5.4256e-01, -4.2852e-01],
...,
[-5.6130e-01, -6.0063e-01, 3.2887e-01, ..., 1.0727e-01,
6.1668e-02, -1.1331e-01],
[-1.0172e+00, -1.7104e+00, -1.3511e+00, ..., -3.2895e-01,
-2.8844e-01, -6.2875e-01],
[ 4.9053e-01, 1.1733e-01, -2.3215e-01, ..., 2.0000e-01,
1.8175e-01, 3.5702e-01]],
[[-1.5610e+00, -7.9536e-01, -4.2821e-01, ..., -4.3661e-01,
-2.8705e-01, 8.6960e-01],
[-2.1187e+00, -7.7346e-01, -5.5928e-01, ..., -7.7035e-01,
-1.7596e-01, 1.6518e+00],
[-1.4796e+00, -6.0097e-01, -1.9529e-01, ..., -3.6841e-01,
-4.6398e-02, 1.3974e+00],
...,
[-5.3729e-01, 1.5577e-01, -6.1991e-01, ..., 1.4528e-01,
2.1714e-01, 7.2824e-01],
[-1.1294e+00, -4.7856e-01, -2.0916e+00, ..., -9.6428e-02,
4.7103e-02, 4.4489e-01],
[-5.7507e-01, -4.6168e-01, -8.8698e-01, ..., -2.1775e-01,
-2.4365e-01, 6.0581e-01]]]], device='cuda:0'),
tensor([[[[ 6.0631e-02, 3.7836e-02, -2.1653e-02, ..., -9.9667e-01,
-5.8017e-01, -1.4497e+00],
[ 4.9357e-02, -4.4799e-01, -5.2146e-01, ..., -1.7068e+00,
-1.9154e+00, -1.6107e+00],
[ 5.8313e-03, -1.0740e-01, -2.7297e-01, ..., -8.2224e-01,
-1.2261e+00, -1.9966e+00],
...,
[ 3.3101e-02, -2.5126e-01, -1.9031e+00, ..., -9.1251e-01,
-1.0035e+00, -1.9145e+00],
[-4.9936e-01, -4.1297e-01, -2.9684e-01, ..., -4.4511e-01,
-5.4507e-01, -1.7668e+00],
[-6.2679e-01, -1.1131e+00, -3.5228e-01, ..., -1.1177e+00,
-1.4949e+00, -7.5565e-01]],
[[-5.3807e-02, -2.3497e+00, -1.6786e+00, ..., -3.9451e+00,
-5.3918e+00, -1.8218e+00],
[-2.0306e+00, -4.4123e+00, -2.9497e+00, ..., -8.4883e+00,
-9.6312e+00, -3.7158e+00],
[-2.1486e+00, -3.6441e+00, -3.2567e+00, ..., -7.2567e+00,
-8.7395e+00, -3.4852e+00],
...,
[-1.7599e+00, -3.2599e+00, -5.8482e+00, ..., -4.9881e+00,
-6.3488e+00, -2.4718e+00],
[-3.5587e+00, -4.1821e+00, -4.8522e+00, ..., -6.4961e+00,
-7.9481e+00, -2.5089e+00],
[ 2.8480e-01, -4.6973e-01, 2.3945e-01, ..., 4.1278e-01,
4.7099e-01, 2.5943e-01]],
[[-1.2751e-01, 1.9169e-01, 7.3526e-01, ..., 6.0368e-01,
2.5698e-01, 1.5273e+00],
[-4.3330e-02, -8.9119e-01, 1.1621e-01, ..., 5.2502e-01,
-5.3724e-02, 8.5411e-02],
[-8.9160e-01, -1.4958e+00, -8.3463e-01, ..., -1.1728e-02,
-5.6940e-02, -8.4414e-02],
...,
[-2.3513e-01, 7.9861e-03, -4.6524e+00, ..., -9.2713e-03,
-4.2338e-02, -1.5135e-01],
[-1.6820e-01, 8.9802e-02, -3.9378e+00, ..., -1.3997e-01,
3.8081e-02, -1.7544e-01],
[-8.1459e-02, -4.8340e-02, -3.8124e-01, ..., 5.0026e-02,
2.2427e-01, -5.1207e-01]],
...,
[[ 6.5713e-01, -3.2121e-01, 1.7686e-01, ..., 4.5833e-01,
3.7419e-01, -1.3574e+00],
[-8.2665e-01, -1.2013e+00, -7.2177e-01, ..., 8.7403e-02,
-5.5091e-01, -1.3886e+00],
[-2.7514e-01, -1.4045e+00, -8.8638e-01, ..., -1.8879e+00,
-1.1413e+00, -9.5455e-01],
...,
[-3.8747e-01, 3.0518e-03, -2.9038e-01, ..., -1.1318e-01,
-1.1595e-01, -9.2449e-01],
[-2.1416e+00, -1.4227e+00, -2.4950e-02, ..., -1.5485e+00,
-1.6069e+00, -1.8467e+00],
[-4.6938e-01, -8.9208e-01, 1.1929e-01, ..., -3.2967e-01,
-5.1503e-01, -1.1959e+00]],
[[ 3.7304e-01, 7.0552e-01, 9.5230e-01, ..., -1.0137e-01,
3.9959e-01, -3.2798e-01],
[-5.6727e-02, 3.3842e-01, 7.2524e-01, ..., -9.5400e-01,
-1.0462e+00, -1.1887e+00],
[-7.2961e-01, -7.6872e-02, 2.9453e-01, ..., -1.7791e-01,
-6.1708e-01, -1.1430e+00],
...,
[-1.2550e-01, 3.5722e-01, -5.8942e-01, ..., 3.0516e-01,
1.5013e-01, -5.0077e-01],
[-4.4375e-01, -4.5438e-01, -1.3672e+00, ..., -9.4446e-02,
-2.3596e-01, -9.4086e-01],
[ 5.1234e-01, 5.4589e-01, -4.1685e-01, ..., 3.7470e-01,
3.5880e-01, 2.9574e-01]],
[[-9.0203e-01, 5.0681e-02, 3.4041e-02, ..., -1.1382e-01,
-1.1866e-01, 5.6998e-01],
[-1.3411e+00, -7.2283e-01, -1.2254e+00, ..., -1.0017e+00,
-9.7611e-02, 9.4618e-01],
[-1.2842e+00, -1.0249e+00, -5.5411e-01, ..., -1.5365e+00,
-3.6131e-01, 8.6219e-01],
...,
[-9.7359e-01, 3.0943e-02, 3.1045e-01, ..., -5.6620e-01,
-5.4773e-01, 6.0050e-01],
[-1.5781e+00, -5.8266e-01, 6.9211e-01, ..., -7.1590e-01,
-6.9508e-01, 3.9401e-01],
[-6.4936e-01, -1.4828e-01, 1.4066e-01, ..., -4.9015e-01,
-4.0650e-01, 1.0235e+00]]],
[[[-8.4388e+00, -5.5290e+00, 1.1003e+00, ..., -1.9680e+00,
-9.0045e-01, -1.8817e+00],
[-7.6198e+00, -9.7895e+00, -6.7305e+00, ..., -4.8786e+00,
-2.0401e+00, -2.0495e+00],
[-1.1984e+00, -4.0488e+00, -8.7502e+00, ..., -6.2162e+00,
-1.6715e+00, -2.3676e+00],
...,
[-1.6457e+00, -3.5023e-01, 1.8232e-01, ..., -3.8694e-01,
-1.3392e+00, -1.3603e+00],
[ 6.1312e-01, 5.7170e-01, -2.2866e+00, ..., 3.3365e-01,
-5.2025e-02, -3.5382e-01],
[-2.4660e+00, -2.4840e+00, -1.2666e+00, ..., -7.8845e-01,
-9.9391e-01, -2.2720e-01]],
[[-1.6425e+01, -1.4068e+01, -1.0635e+01, ..., -1.0018e+01,
-6.2942e+00, -2.4407e+00],
[-2.2172e+01, -2.6824e+01, -1.8750e+01, ..., -1.5290e+01,
-9.7822e+00, -4.5455e+00],
[-1.1535e+01, -1.6131e+01, -1.3731e+01, ..., -1.6116e+01,
-7.1190e+00, -3.3623e+00],
...,
[-3.1515e+00, -5.5466e+00, -3.7738e+00, ..., -5.6232e+00,
-5.0399e+00, -2.7705e+00],
[-3.5881e+00, -4.1038e+00, -3.0243e+00, ..., -1.5589e+00,
-1.9558e+00, -5.0353e-01],
[ 2.4188e+00, -3.3553e-01, -5.0003e-01, ..., 3.3385e-02,
2.3529e-02, -3.6928e-02]],
[[-1.9922e+01, -2.4467e+01, -1.6412e+01, ..., 2.9369e+00,
5.7913e-01, 1.9807e+00],
[-3.0244e+01, -4.0533e+01, -3.2320e+01, ..., 3.7978e+00,
-1.3170e-01, 1.5057e-02],
[-2.6084e+01, -4.1932e+01, -3.6194e+01, ..., 3.7531e+00,
-5.5814e-02, -1.5733e-01],
...,
[-4.4195e+00, -4.8452e+00, -2.4888e+00, ..., 8.1257e-01,
7.2030e-01, -2.2516e+00],
[-3.4458e+00, -5.4099e+00, -3.1319e+00, ..., 1.0423e-01,
2.9684e-01, 2.7682e-01],
[-1.8674e+00, -9.0030e-01, -5.8062e-01, ..., -1.8631e-01,
-7.5471e-02, -8.1761e-02]],
...,
[[ 9.1003e-01, -8.6016e+00, -1.5474e+01, ..., -3.0834e+00,
-4.8917e-02, -2.2722e+00],
[-4.5900e-01, -1.6070e+01, -2.6610e+01, ..., -6.3868e+00,
-1.5942e+00, -2.2590e+00],
[ 2.5470e+00, -1.1551e+01, -1.8994e+01, ..., -5.0993e+00,
-6.7373e-01, -1.6278e+00],
...,
[-2.6922e-01, -1.2038e+00, -1.5203e+00, ..., -2.0535e+00,
-3.3724e+00, -2.7588e+00],
[-5.8997e+00, -4.9908e+00, -2.9647e+00, ..., -8.8865e-01,
-8.9103e-01, -9.3860e-01],
[-4.5608e-01, -2.3114e+00, -1.2520e+00, ..., -5.1690e-01,
-7.9706e-01, -5.9510e-01]],
[[-5.5089e+00, -7.0645e+00, -5.0687e+00, ..., 1.2697e+00,
1.0991e+00, 1.0819e-01],
[-1.0104e+01, -1.9037e+01, -1.2596e+01, ..., 1.6847e+00,
4.4758e-01, -8.2296e-01],
[-1.1239e+01, -1.5351e+01, -9.6745e+00, ..., 1.1175e-01,
5.5066e-01, -4.7663e-01],
...,
[-3.3232e+00, -1.1289e+00, 1.6338e+00, ..., -5.0013e-01,
2.0425e-01, 8.0343e-01],
[-1.0817e+00, -1.9761e+00, -2.3386e+00, ..., 4.4621e-01,
2.5269e-01, 9.7044e-02],
[ 1.5854e-02, 2.7086e-02, 5.3811e-01, ..., -9.5081e-02,
-1.8576e-01, -4.1162e-02]],
[[-9.2264e+00, -1.8249e+01, -2.0577e+01, ..., -1.5941e+00,
-4.2745e-01, 8.4108e-01],
[-8.2577e+00, -2.1602e+01, -2.1142e+01, ..., -5.1531e-01,
-2.5041e-01, 1.7307e+00],
[-4.3060e-01, -1.3147e+01, -2.1649e+01, ..., 3.1352e+00,
-2.8018e-01, 1.4490e+00],
...,
[-3.8313e+00, -3.7304e+00, -4.0885e+00, ..., 2.3055e+00,
1.2073e+00, 2.9532e+00],
[-2.5997e+00, -2.9787e+00, -2.3762e+00, ..., -4.4450e-01,
-3.0644e-01, 3.0810e-01],
[-1.8321e+00, -2.9754e+00, -1.9315e+00, ..., -1.6506e-01,
1.1311e-01, 5.5441e-01]]],
[[[-2.3347e-01, -2.7191e-01, 2.1416e-01, ..., 5.8343e-02,
4.6776e-01, 1.3124e-01],
[-1.3067e-01, -1.5419e+00, -8.9624e-01, ..., -6.9889e-01,
-4.6466e-01, -3.6424e-01],
[-1.2358e-01, -1.3400e+00, -6.6804e-01, ..., -4.5455e-01,
-4.0109e-01, -6.8638e-01],
...,
[-2.6963e-01, -1.7064e+00, -1.2644e+00, ..., -9.4132e-01,
-9.5544e-01, -2.0357e+00],
[-1.0580e+00, -1.6542e+00, -8.3567e-01, ..., -5.3345e-01,
-4.6386e-01, -1.7903e+00],
[-1.2386e+00, -2.1187e+00, -1.5792e+00, ..., -1.4088e+00,
-1.5554e+00, -9.6662e-01]],
[[-1.5981e-01, -1.4029e+00, -5.7023e-01, ..., -5.7281e-01,
-1.3469e+00, -2.9390e-01],
[-2.0067e+00, -4.0590e+00, -2.4421e+00, ..., -2.3324e+00,
-3.2540e+00, -1.1293e+00],
[-1.8855e+00, -3.4212e+00, -2.6005e+00, ..., -2.4863e+00,
-2.9157e+00, -1.2003e+00],
...,
[-3.7664e+00, -5.5704e+00, -5.1879e+00, ..., -5.4058e+00,
-6.1786e+00, -2.8973e+00],
[-6.0800e+00, -6.9807e+00, -6.1477e+00, ..., -6.1016e+00,
-7.4215e+00, -2.7014e+00],
[ 1.1197e-02, -8.6830e-01, 2.5477e-01, ..., 3.9482e-01,
6.8723e-01, 1.8984e-01]],
[[-5.3751e-01, -1.8495e-01, 3.7022e-01, ..., 2.9382e-01,
3.9593e-03, 8.7625e-01],
[-3.6138e-01, -4.8692e-01, 4.7090e-01, ..., 3.7567e-01,
2.0027e-01, 4.6854e-01],
[-8.6915e-01, -2.3385e-01, 1.7694e-01, ..., 6.8609e-02,
1.3318e-01, 4.0272e-01],
...,
[-1.4204e+00, 2.8176e-02, 5.0685e-02, ..., 7.7382e-02,
2.5517e-02, 3.0547e-01],
[-1.2478e+00, 1.1524e-01, -6.0390e-02, ..., -4.4387e-02,
1.6018e-02, 2.9289e-01],
[-1.1495e+00, -6.9998e-02, 4.7182e-02, ..., -2.4121e-02,
-4.8794e-02, -3.4209e-01]],
...,
[[ 4.3291e-02, -6.2078e-01, -3.8547e-01, ..., -3.7815e-02,
-2.3167e-01, -7.6945e-01],
[-8.5982e-01, -6.3735e-01, -5.5824e-01, ..., -2.9643e-01,
-9.7691e-01, -7.7380e-01],
[-4.9423e-01, -1.2482e-01, -2.1571e-02, ..., 7.3523e-02,
-1.8529e-01, -5.4332e-01],
...,
[-5.8044e-01, -6.0642e-01, -3.6067e-01, ..., -3.7575e-01,
-5.7286e-01, -1.1832e+00],
[-3.0339e+00, -2.8610e+00, -2.0465e+00, ..., -1.9127e+00,
-2.0307e+00, -2.0584e+00],
[-5.2557e-01, -1.5717e+00, -1.0504e+00, ..., -8.7145e-01,
-9.4810e-01, -1.3476e+00]],
[[-2.0483e-01, 4.1974e-02, 4.8362e-01, ..., 1.2609e-01,
5.3277e-01, 1.2793e-01],
[-4.5959e-01, -2.7625e-01, -5.1500e-04, ..., -6.9810e-02,
1.0348e-01, -7.4217e-02],
[-6.7843e-01, -5.8335e-01, -1.0584e-02, ..., 5.2966e-03,
2.4917e-02, -2.5454e-02],
...,
[-9.9807e-02, -1.0262e+00, 2.5846e-02, ..., 2.7486e-01,
1.6489e-01, -6.9850e-01],
[-5.1652e-01, -1.7851e+00, -1.7064e-01, ..., 7.0785e-03,
-3.9721e-02, -9.3260e-01],
[ 4.7603e-01, -9.5031e-02, 1.8662e-01, ..., 2.6525e-01,
2.9476e-01, 3.3388e-01]],
[[-6.9582e-01, 3.9652e-01, 5.3943e-01, ..., 4.8299e-01,
3.9412e-01, 6.1474e-01],
[-8.6014e-01, 2.9942e-01, -2.1288e-01, ..., -2.9987e-01,
-2.1043e-01, 6.3441e-01],
[-4.8328e-01, 3.1697e-01, 7.9479e-02, ..., 7.7978e-02,
6.0599e-02, 5.9388e-01],
...,
[-1.0796e+00, -9.0304e-01, -4.1046e-01, ..., -3.8118e-01,
-6.5543e-02, 8.2358e-01],
[-1.8594e+00, -1.8185e+00, -8.9280e-01, ..., -7.8229e-01,
-4.6424e-01, 4.2988e-01],
[-9.4378e-01, -9.4299e-01, -2.3423e-01, ..., -2.8118e-01,
-1.4318e-01, 1.1293e+00]]],
[[[-4.4847e+00, -3.1180e+00, -3.0530e+00, ..., -5.3857e-01,
-2.9006e-02, -1.2837e+00],
[-6.3811e+00, -2.6613e+00, 1.3774e-02, ..., -5.6121e-01,
-3.9582e-01, -1.4872e+00],
[-8.4768e+00, -1.0197e+01, -1.1273e+01, ..., -3.6697e-01,
-3.2539e-01, -1.6047e+00],
...,
[ 1.3622e-01, -8.0391e-01, 1.7217e-01, ..., 5.9203e-01,
-2.3326e-01, -3.6688e+00],
[-1.5327e-01, -7.0846e-01, -3.8659e-02, ..., 1.8062e+00,
1.4863e+00, -9.2993e-01],
[-4.0148e-01, -1.3089e+00, -1.0638e+00, ..., -1.0311e+00,
-1.2363e+00, -6.1722e-01]],
[[-8.2989e+00, -1.1364e+01, -7.0174e+00, ..., -4.2224e+00,
-5.9365e+00, -2.0378e+00],
[-6.6019e+00, -1.0525e+01, -7.6473e+00, ..., -6.3713e+00,
-8.1421e+00, -3.1237e+00],
[-1.4321e+00, -5.6128e+00, -7.7646e+00, ..., -5.1932e+00,
-5.4827e+00, -2.4213e+00],
...,
[-2.4969e+00, -3.5801e+00, -3.2169e+00, ..., -5.1981e+00,
-3.4576e+00, -3.3536e+00],
[-3.4159e+00, -3.8979e+00, -2.9145e+00, ..., -4.8989e+00,
-6.3081e+00, -4.1510e+00],
[ 5.5915e-01, -5.1941e-02, 5.6923e-01, ..., 4.8175e-01,
5.2818e-01, 1.8112e-01]],
[[-7.6685e+00, -5.6558e+00, -1.3856e+01, ..., 9.1521e-01,
9.8899e-01, 1.6567e+00],
[-2.0415e+01, -1.4250e+01, -2.4387e+01, ..., -1.0318e-01,
5.1561e-01, 5.7471e-01],
[-2.2703e+01, -2.8090e+01, -3.8830e+01, ..., -8.4538e-01,
1.1132e-01, -4.2897e-01],
...,
[-1.0627e+00, 5.0369e-01, 5.3566e-01, ..., -2.4508e+00,
-2.0737e+00, -2.1464e+00],
[-7.7435e-01, 2.3173e-01, 2.5827e-01, ..., 2.1975e-02,
1.8230e-01, -1.7600e+00],
[-1.7504e-01, 2.3342e-02, 8.4885e-02, ..., 1.0962e-03,
1.2629e-01, 1.9933e-01]],
...,
[[-6.1225e-01, -2.2003e+00, -4.0814e+00, ..., 4.9168e-03,
-2.7213e-01, -1.4245e+00],
[-2.8644e+00, -1.2838e+01, -1.1122e+01, ..., -1.2598e+00,
-1.5520e+00, -1.5463e+00],
[-3.0528e+00, -1.0832e+01, -1.1558e+01, ..., -1.5627e+00,
-9.0640e-01, -8.5587e-01],
...,
[-6.7318e-01, -7.8591e-01, -8.8155e-01, ..., 9.5226e-02,
5.1587e-02, -2.9024e+00],
[-2.1642e+00, -1.7004e+00, -1.2591e+00, ..., -5.3065e+00,
-5.6967e+00, -2.7397e+00],
[-6.4827e-01, -9.1776e-01, -4.1257e-01, ..., -3.9136e-01,
-5.3757e-01, -4.8000e-01]],
[[-3.7286e+00, 2.8455e-01, -2.3506e+00, ..., 3.2097e-01,
6.0469e-01, -5.2154e-01],
[-8.1570e+00, -3.6527e+00, -6.3027e+00, ..., 1.2822e+00,
7.6012e-01, -6.6728e-01],
[-1.0180e+01, -9.2142e+00, -1.2245e+01, ..., 5.2187e-01,
5.0671e-01, -4.9922e-01],
...,
[-8.5286e-02, -6.6843e-01, 2.0401e-01, ..., 2.7687e+00,
3.0292e+00, 3.0424e+00],
[-4.9091e-01, -1.3258e+00, 1.8891e-02, ..., 3.0214e-01,
-5.3507e-01, -1.9031e+00],
[ 5.2373e-01, -1.2403e-01, -6.0212e-02, ..., 6.1986e-02,
7.0001e-02, 1.7360e-01]],
[[-8.4270e+00, -9.9461e+00, -5.8100e+00, ..., 1.0153e-01,
6.0996e-02, 6.7497e-01],
[-1.5744e+01, -1.3443e+01, -6.9254e+00, ..., -7.2056e-01,
-5.0860e-01, 8.8682e-01],
[-1.7071e+01, -1.2261e+01, -1.4252e+01, ..., 1.6490e-01,
4.4536e-02, 6.6687e-01],
...,
[-6.5022e-01, -4.6768e-01, 1.4197e-01, ..., 1.8784e+00,
1.9330e+00, 2.2033e+00],
[-1.0269e+00, -7.6718e-01, -4.1472e-01, ..., 2.4429e+00,
3.5419e+00, 3.4342e+00],
[-3.6699e-01, -2.7168e-01, -1.4337e-01, ..., -2.2425e-01,
-1.6368e-01, 5.9962e-01]]],
[[[-2.8554e-01, -9.6317e-01, -8.2223e-01, ..., -3.9723e-02,
6.0669e-01, 5.2808e-01],
[ 1.6891e-01, -1.8694e+00, -1.2473e+00, ..., -4.6279e-01,
-2.2631e-01, 1.0635e-01],
[ 1.5059e-01, -1.4200e+00, -9.4226e-01, ..., -3.1136e-01,
-1.3858e-01, -1.0504e-01],
...,
[ 1.5614e-01, -1.1400e+00, -7.8709e-01, ..., -5.8325e-01,
-9.5447e-02, -1.5664e-01],
[-1.5664e-01, -7.5226e-01, -4.4399e-01, ..., -1.4631e-01,
2.6722e-01, -4.3049e-02],
[-4.7629e-01, -9.1861e-01, -1.0238e+00, ..., -1.0310e+00,
-1.0903e+00, -4.0155e-01]],
[[-2.0780e+00, -4.0802e+00, -3.7131e+00, ..., -4.9382e-01,
-1.0517e+00, -2.9874e-02],
[-3.9188e+00, -6.3762e+00, -5.5417e+00, ..., -1.7451e+00,
-2.6241e+00, -6.2980e-01],
[-3.2201e+00, -4.9874e+00, -4.6751e+00, ..., -1.2890e+00,
-1.5019e+00, -4.0328e-01],
...,
[-2.8371e+00, -3.9402e+00, -5.0345e+00, ..., -1.7440e+00,
-1.6035e+00, -4.1643e-01],
[-5.2568e+00, -5.5754e+00, -5.8987e+00, ..., -2.5880e+00,
-3.3526e+00, -9.4775e-01],
[ 4.6334e-01, -2.4397e-01, 3.7779e-01, ..., 3.3131e-01,
3.6431e-01, 8.8769e-02]],
[[-6.0898e-01, 3.4424e-01, 8.1110e-01, ..., 8.0339e-01,
2.9913e-01, 1.0437e+00],
[-9.1052e-01, 6.0478e-04, 6.0153e-01, ..., 3.8843e-01,
3.8152e-02, 4.6504e-01],
[-1.2957e+00, -1.4063e-02, 2.3062e-02, ..., 3.9953e-02,
7.1769e-02, 4.2093e-01],
...,
[-9.6457e-01, -4.9793e-01, 1.6601e-01, ..., 2.5452e-01,
-7.3044e-03, 4.1110e-01],
[-8.9646e-01, -7.9896e-01, 2.7235e-02, ..., 1.6021e-01,
5.4477e-03, 5.4539e-01],
[-1.5909e-01, -3.9028e-01, 1.2205e-01, ..., -7.6588e-02,
2.2146e-02, 3.3650e-01]],
...,
[[ 6.7172e-01, 7.3186e-02, 2.9771e-01, ..., -1.2333e-01,
-3.7799e-01, -8.6973e-01],
[-7.9641e-01, -3.9732e-01, -1.0470e-01, ..., -6.5133e-01,
-1.2700e+00, -8.1445e-01],
[-5.2254e-01, -4.1077e-01, -6.1936e-02, ..., -1.1762e-01,
-1.9656e-01, -3.9436e-01],
...,
[-3.1624e-01, -2.2490e-01, -1.5921e-01, ..., -3.0870e-01,
-1.7118e-01, -3.8563e-01],
[-1.9337e+00, -1.5708e+00, -1.3244e+00, ..., -6.7444e-01,
-8.8159e-01, -8.1266e-01],
[-4.4092e-01, -1.0526e+00, -3.7298e-01, ..., -2.8488e-02,
-1.2736e-01, -3.2982e-01]],
[[ 2.8515e-02, -1.7665e-01, -3.8797e-02, ..., 3.0703e-01,
6.5869e-01, 2.9752e-01],
[ 4.9428e-02, -8.4435e-01, -4.0371e-02, ..., -1.9447e-02,
1.2133e-01, -3.0431e-02],
[-1.9745e-01, -9.5607e-01, 1.2130e-02, ..., -1.1200e-01,
-9.1855e-02, 1.6346e-01],
...,
[ 5.2961e-01, 4.6847e-01, 1.7548e-01, ..., -1.7065e-01,
-1.0854e-01, 1.2259e-01],
[-2.8009e-01, -1.0674e+00, -2.3417e-01, ..., -4.9880e-01,
-3.1857e-01, -7.8301e-02],
[ 7.7025e-01, 2.9934e-01, 1.7230e-01, ..., -2.2487e-01,
-2.1678e-01, 2.9195e-01]],
[[-6.4958e-01, -1.5874e-01, 5.8578e-02, ..., 5.4404e-01,
3.0915e-01, 5.1583e-01],
[-1.0764e+00, -7.5775e-01, -8.9206e-01, ..., -3.1762e-02,
-1.2341e-01, 3.7870e-01],
[-6.6021e-01, -6.6360e-01, -5.2506e-01, ..., 5.9975e-02,
-1.3417e-01, 6.3967e-02],
...,
[-1.4209e+00, -2.8721e-01, -4.2207e-01, ..., 5.6188e-01,
-8.7299e-02, 6.0916e-02],
[-1.8060e+00, -3.6111e-01, -3.7652e-01, ..., 1.4215e-01,
-2.4773e-01, -2.0926e-01],
[-1.1696e+00, -6.4228e-01, -1.8951e-01, ..., -9.3993e-02,
-2.6030e-01, 1.0781e-01]]]], device='cuda:0'),
tensor([[[[-5.4953e-02, -4.6841e-01, -2.4568e-01, ..., -3.4523e+00,
-4.9029e-01, -1.5995e+00],
[ 1.4877e-01, -1.1846e+00, -1.8308e+00, ..., -8.5328e+00,
-3.2741e+00, -2.4486e+00],
[ 2.3679e-01, -7.6805e-01, -2.0561e+00, ..., -9.6563e+00,
-3.0518e+00, -2.5434e+00],
...,
[ 9.0996e-02, -3.5165e-01, 1.3931e+00, ..., -2.7899e+00,
-1.3550e+00, -2.1538e+00],
[-4.0613e-01, -8.4635e-01, -1.7284e-01, ..., -1.2715e+00,
-8.9598e-01, -2.0832e+00],
[-5.5946e-01, -1.2430e+00, -1.0611e+00, ..., -1.4046e+00,
-1.8815e+00, -1.1626e+00]],
[[-1.3041e+00, -2.8517e+00, -3.0269e+00, ..., -7.3380e+00,
-5.6559e+00, -1.7154e+00],
[-2.7665e+00, -4.5286e+00, -1.0330e+01, ..., -2.0689e+01,
-1.4607e+01, -6.9501e+00],
[-2.3478e+00, -3.4780e+00, -8.3528e+00, ..., -1.4336e+01,
-1.0026e+01, -7.3975e+00],
...,
[-2.3658e+00, -3.7454e+00, -3.1688e+00, ..., -6.8523e+00,
-6.8497e+00, -2.3967e+00],
[-4.2794e+00, -5.1464e+00, -4.5892e+00, ..., -7.0284e+00,
-8.2868e+00, -2.3521e+00],
[ 4.7548e-01, -1.1208e-01, 6.1744e-01, ..., 2.3364e-01,
2.2313e-01, 3.2173e-01]],
[[-2.2198e-01, 2.1502e-01, 5.8952e-01, ..., -7.8978e-01,
5.8780e-02, 1.5480e+00],
[-4.2416e-02, -1.8729e-01, -1.2232e+00, ..., 8.2704e-01,
-2.6849e-01, 3.9932e-02],
[-4.5693e-01, -1.5167e-01, -2.6193e+00, ..., -3.5689e+00,
-2.6568e-02, 1.2722e+00],
...,
[-5.9865e-01, 1.2312e+00, 1.8117e+00, ..., 9.5969e-01,
-1.7758e-01, -2.2411e-01],
[-4.4219e-01, -1.0143e-01, 7.6066e-02, ..., -1.8312e-01,
-4.9094e-02, -2.9337e-01],
[ 3.1797e-02, -9.2448e-02, 1.4361e-01, ..., -6.9679e-02,
4.7457e-02, -8.9288e-01]],
...,
[[ 7.3209e-01, 1.8662e-01, 2.9116e-01, ..., -6.1910e+00,
4.8242e-01, -1.4036e+00],
[-4.9506e-01, -4.5944e-01, 2.9896e-01, ..., -1.1748e+01,
-1.6872e+00, -1.3419e+00],
[-2.4813e-01, -1.4366e-01, 2.8148e+00, ..., -1.0278e+01,
-2.2611e+00, -2.1632e+00],
...,
[-4.4744e-01, -1.5115e+00, -3.3890e+00, ..., -1.7141e+00,
4.1573e-02, -9.7759e-01],
[-2.2928e+00, -1.6429e+00, -1.2366e+00, ..., -1.7021e+00,
-1.9870e+00, -2.2184e+00],
[-4.9324e-01, -7.7082e-01, -1.2774e-01, ..., -4.9380e-01,
-3.8146e-01, -1.3040e+00]],
[[ 1.4019e-01, 1.9862e-01, 5.0141e-01, ..., 2.2049e+00,
4.9246e-01, -5.6146e-02],
[ 1.6926e-01, -7.8060e-03, 6.4627e-01, ..., 1.8718e-01,
-3.7202e+00, -3.0392e+00],
[-8.7984e-02, -1.4361e-01, -5.2787e-01, ..., 2.0252e+00,
1.2742e-01, -2.1901e+00],
...,
[-1.0225e-01, 7.4541e-01, 4.0612e-01, ..., -5.7089e-01,
1.4114e-01, -2.4791e-01],
[-5.8994e-01, -1.1110e+00, -7.9555e-03, ..., -5.7267e-01,
-1.7289e-01, -6.1148e-01],
[ 6.4891e-01, 3.0643e-01, 2.5279e-01, ..., 2.5726e-01,
1.7191e-01, 4.7028e-01]],
[[-7.6317e-01, 5.9893e-02, 2.8279e-01, ..., 3.4173e+00,
-2.9424e-01, 5.0028e-01],
[-1.2015e+00, -1.5024e-01, -1.3973e+00, ..., 3.3211e+00,
-3.3534e-01, 1.7089e+00],
[-8.8871e-01, 1.2163e-01, -3.1570e+00, ..., -1.8339e+00,
-1.1393e+00, 1.4569e+00],
...,
[-9.3454e-01, -9.6573e-01, -2.6508e-01, ..., 2.0596e-01,
-8.6883e-01, 6.7586e-01],
[-1.5471e+00, -5.3569e-01, -2.7969e-01, ..., -6.7697e-01,
-1.1625e+00, 2.8328e-01],
[-7.2958e-01, -2.8637e-01, -1.6200e-01, ..., -5.2554e-01,
-6.8200e-01, 1.1185e+00]]],
[[[-3.1187e-01, -1.0086e+00, -8.4203e-01, ..., -1.7097e-02,
6.1726e-01, 5.4687e-01],
[ 1.7545e-01, -1.8516e+00, -1.2839e+00, ..., -5.8057e-01,
-2.4805e-01, 1.2464e-01],
[-3.1710e-03, -1.6016e+00, -1.0129e+00, ..., -2.4885e-01,
-1.4776e-01, -9.9219e-02],
...,
[ 2.7635e-01, -1.4711e+00, -1.5666e+00, ..., -2.7838e-01,
-1.1599e-01, -1.6678e-01],
[ 7.8353e-02, -8.7340e-01, -4.9694e-01, ..., 4.8658e-02,
2.4109e-01, -5.0550e-02],
[-1.4499e-01, -1.0573e+00, -9.2267e-01, ..., -1.0672e+00,
-1.0948e+00, -4.1348e-01]],
[[-2.1094e+00, -4.0768e+00, -3.6910e+00, ..., -5.0331e-01,
-1.0563e+00, -1.7162e-02],
[-3.9634e+00, -6.3665e+00, -5.5241e+00, ..., -2.2732e+00,
-2.7701e+00, -7.1962e-01],
[-4.3657e+00, -5.4935e+00, -5.2020e+00, ..., -2.3463e+00,
-1.5602e+00, -4.0355e-01],
...,
[-3.2946e+00, -5.2904e+00, -4.8800e+00, ..., -1.3819e+00,
-1.6424e+00, -4.6342e-01],
[-5.0271e+00, -6.0420e+00, -5.7401e+00, ..., -2.3858e+00,
-3.3302e+00, -9.7779e-01],
[ 6.8341e-01, -1.7162e-01, 4.6691e-01, ..., 3.3114e-01,
3.5675e-01, 7.9839e-02]],
[[-5.8255e-01, 4.0425e-01, 8.2839e-01, ..., 8.5532e-01,
3.3310e-01, 1.1204e+00],
[-9.2867e-01, 1.1613e-01, 6.4446e-01, ..., 5.1812e-01,
5.2013e-02, 4.8835e-01],
[-1.4197e+00, 8.3922e-02, 1.3261e-01, ..., 5.0303e-01,
9.7395e-02, 4.2497e-01],
...,
[-1.9599e+00, -3.8122e-01, 3.4642e-01, ..., 4.6734e-02,
6.2097e-02, 4.5903e-01],
[-1.3240e+00, 1.7354e-01, -4.5375e-02, ..., 5.8101e-02,
4.4870e-02, 6.4607e-01],
[-6.0342e-01, 2.5581e-01, 1.4102e-01, ..., -9.5802e-02,
1.0247e-03, 3.5419e-01]],
...,
[[ 6.6503e-01, 7.3737e-02, 2.6948e-01, ..., -8.3537e-02,
-4.0602e-01, -9.4562e-01],
[-8.1684e-01, -3.7394e-01, -8.9846e-02, ..., -6.7526e-01,
-1.3024e+00, -8.4364e-01],
[-3.5805e-01, -3.7384e-01, -8.6264e-03, ..., -1.5512e-01,
-1.9292e-01, -3.9042e-01],
...,
[-1.5515e+00, -1.6399e+00, -1.2138e-01, ..., -1.2809e-01,
-2.0728e-01, -3.9151e-01],
[-2.0996e+00, -1.7490e+00, -1.2152e+00, ..., -8.3959e-01,
-8.7513e-01, -7.9564e-01],
[-4.0698e-01, -9.7431e-01, -3.3163e-01, ..., -6.4596e-02,
-1.6502e-01, -3.2355e-01]],
[[ 1.4857e-02, -2.5079e-01, -5.4690e-02, ..., 3.5707e-01,
7.1375e-01, 3.1079e-01],
[ 1.3652e-01, -9.1628e-01, -8.8663e-02, ..., -2.2706e-01,
1.1480e-01, -6.5915e-02],
[-2.1258e-01, -1.4228e+00, -3.6563e-01, ..., -3.1648e-01,
-1.0256e-01, 1.9002e-01],
...,
[-2.3794e-01, -1.1657e+00, -4.8019e-01, ..., -1.3085e-01,
-7.8056e-02, 1.1253e-01],
[-5.0763e-01, -1.8412e+00, -2.6937e-01, ..., -4.0738e-01,
-3.2207e-01, -6.3613e-02],
[ 6.8633e-01, -7.9077e-02, 2.7058e-01, ..., -2.9376e-01,
-2.3749e-01, 2.8540e-01]],
[[-6.4528e-01, -1.5037e-01, 1.8495e-02, ..., 5.7983e-01,
3.8845e-01, 5.7254e-01],
[-1.0400e+00, -8.1397e-01, -9.4754e-01, ..., -5.4091e-02,
-1.2757e-01, 4.4266e-01],
[-4.8336e-01, -5.9882e-01, -3.9908e-01, ..., 6.5345e-02,
-1.6544e-01, 1.0429e-01],
...,
[-3.6419e-01, -1.9807e-01, -3.5042e-01, ..., 2.3148e-02,
-3.4029e-02, 6.9051e-02],
[-7.9573e-01, -9.5978e-01, -5.6006e-01, ..., -1.3909e-01,
-2.0960e-01, -2.1068e-01],
[-4.4687e-01, -8.2950e-01, -3.3904e-01, ..., -1.5544e-01,
-2.5477e-01, 1.0912e-01]]],
[[[-7.0198e-01, -1.3869e+00, -1.3856e+00, ..., -2.2619e-01,
2.4515e-01, -4.1623e-01],
[-2.6636e-01, -2.4234e+00, -2.1160e+00, ..., -5.7791e-01,
-3.9243e-01, -5.6465e-01],
[-3.9346e-01, -2.1156e+00, -1.8445e+00, ..., -3.5728e-01,
-3.8148e-01, -1.0624e+00],
...,
[-4.7990e-01, -1.9284e+00, -1.4611e+00, ..., 4.8941e-01,
3.6591e-02, -1.7731e+00],
[-9.3915e-01, -1.6708e+00, -1.1885e+00, ..., 8.9773e-02,
1.8273e-01, -1.0084e+00],
[-7.9378e-01, -1.6314e+00, -1.2586e+00, ..., -1.0510e+00,
-1.2945e+00, -7.5351e-01]],
[[-2.3365e+00, -4.8275e+00, -4.2784e+00, ..., -2.2648e+00,
-3.5781e+00, -1.3052e+00],
[-5.3456e+00, -8.1179e+00, -6.6020e+00, ..., -3.3730e+00,
-5.0402e+00, -2.0463e+00],
[-4.3080e+00, -6.2274e+00, -5.4414e+00, ..., -3.0166e+00,
-3.8859e+00, -1.7546e+00],
...,
[-4.3673e+00, -6.0778e+00, -5.6952e+00, ..., -3.0531e+00,
-4.5399e+00, -2.7187e+00],
[-6.3502e+00, -7.3791e+00, -6.2817e+00, ..., -4.3242e+00,
-5.9350e+00, -2.1333e+00],
[-1.1201e-02, -8.2006e-01, 3.9175e-01, ..., 4.6086e-01,
6.1835e-01, 1.4159e-01]],
[[-5.9255e-01, 5.8250e-01, 1.0888e+00, ..., 7.8376e-01,
5.5989e-01, 1.3152e+00],
[-9.0876e-01, -4.2738e-02, 3.3827e-01, ..., 3.8381e-01,
4.0024e-01, 2.6110e-01],
[-1.6330e+00, -1.2779e-01, -7.2380e-02, ..., 6.6061e-02,
2.8455e-01, 8.2537e-02],
...,
[-1.7366e+00, -3.4582e-02, 9.1921e-01, ..., 4.4775e-01,
-3.0288e-01, 2.5888e-01],
[-1.5670e+00, 7.5729e-02, 4.8293e-02, ..., 1.8251e-01,
2.6163e-01, 1.2257e-01],
[-1.3930e+00, -5.1171e-01, -2.5740e-01, ..., 3.6327e-03,
1.9857e-01, -1.2093e-01]],
...,
[[ 1.0318e+00, -3.9213e-01, -2.3759e-01, ..., 3.4396e-01,
2.4290e-01, -7.5240e-01],
[-8.4938e-01, -9.4906e-01, -7.6528e-01, ..., -2.5803e-01,
-9.5794e-01, -9.3675e-01],
[-6.9690e-01, -9.6159e-01, -6.9172e-01, ..., -1.1971e-01,
-2.4660e-01, -6.1482e-01],
...,
[-7.4913e-01, -1.1501e+00, -1.4289e+00, ..., -1.1926e+00,
-1.4801e+00, -2.1410e+00],
[-2.7430e+00, -2.8360e+00, -1.8419e+00, ..., -1.2522e+00,
-1.3330e+00, -1.2888e+00],
[-6.7665e-01, -1.6402e+00, -9.5099e-01, ..., -1.2269e-01,
-2.5436e-01, -5.4077e-01]],
[[ 1.8849e-02, 1.5782e-02, 6.2766e-01, ..., 2.9656e-01,
7.6079e-01, -1.5866e-02],
[-2.7048e-02, -6.9280e-01, 1.9575e-01, ..., 4.2725e-01,
6.5072e-01, -2.6916e-01],
[-1.2542e-01, -8.2238e-01, 3.4003e-01, ..., 4.0334e-01,
4.2428e-01, -1.5048e-01],
...,
[ 4.5524e-02, -8.0118e-01, 4.1668e-01, ..., 4.0131e-01,
-7.6647e-01, -1.1006e+00],
[-2.6908e-01, -1.4802e+00, 8.8721e-03, ..., -1.5282e-03,
2.6484e-02, -6.6738e-01],
[ 6.6985e-01, 1.3312e-01, 2.5345e-01, ..., 1.9931e-01,
2.4982e-01, 3.4658e-01]],
[[-1.5966e+00, -6.1525e-01, 1.0562e-01, ..., 2.9949e-01,
2.4433e-01, 8.1636e-01],
[-2.0686e+00, -3.9844e-01, -6.2853e-02, ..., -4.1138e-01,
-1.6377e-01, 1.1265e+00],
[-1.7203e+00, -4.3354e-01, 1.3008e-01, ..., -6.1198e-02,
2.1782e-02, 9.2695e-01],
...,
[-1.7311e+00, -4.6846e-01, -5.8358e-02, ..., 7.5553e-02,
1.0369e+00, 1.6105e+00],
[-2.1679e+00, -1.0964e+00, -1.5847e-01, ..., -2.9856e-01,
-8.3499e-02, 6.4895e-01],
[-1.0191e+00, -1.6001e-01, 4.6399e-01, ..., -3.0114e-01,
-3.1598e-01, 6.9915e-01]]],
[[[-7.0843e-01, -1.4220e+00, -1.4188e+00, ..., -1.1872e-02,
5.8904e-01, 4.9452e-01],
[-2.7228e-01, -2.4303e+00, -2.1969e+00, ..., -4.7164e-01,
-2.6697e-01, 8.6239e-02],
[-3.8363e-01, -2.1155e+00, -1.9078e+00, ..., -2.9581e-01,
-1.3909e-01, -9.1257e-02],
...,
[-4.5826e-01, -1.9376e+00, -1.8411e+00, ..., -3.2263e-01,
-8.5235e-02, -1.2635e-01],
[-9.8585e-01, -1.7364e+00, -1.1614e+00, ..., 2.4952e-03,
2.5860e-01, -3.1246e-02],
[-8.4291e-01, -1.6686e+00, -1.2731e+00, ..., -1.0287e+00,
-1.0554e+00, -3.9237e-01]],
[[-2.3644e+00, -4.8038e+00, -4.1519e+00, ..., -4.7195e-01,
-1.0031e+00, -1.2771e-02],
[-5.3780e+00, -8.1992e+00, -6.6476e+00, ..., -1.8668e+00,
-2.6411e+00, -6.5307e-01],
[-4.2984e+00, -6.3278e+00, -5.6663e+00, ..., -1.3634e+00,
-1.5506e+00, -3.9092e-01],
...,
[-4.4414e+00, -6.3002e+00, -5.6726e+00, ..., -1.3067e+00,
-1.5559e+00, -3.6696e-01],
[-6.3987e+00, -7.4410e+00, -6.4233e+00, ..., -2.5147e+00,
-3.3093e+00, -9.6847e-01],
[-2.2345e-02, -7.9101e-01, 3.9075e-01, ..., 3.1727e-01,
3.5191e-01, 7.7064e-02]],
[[-6.0169e-01, 5.7699e-01, 1.0958e+00, ..., 8.0113e-01,
2.9794e-01, 1.0665e+00],
[-9.1402e-01, -7.9813e-02, 3.9466e-01, ..., 4.0673e-01,
6.4501e-02, 4.6542e-01],
[-1.6345e+00, -1.7843e-01, -7.7767e-02, ..., 8.3692e-02,
1.0382e-01, 4.1842e-01],
...,
[-1.7921e+00, -1.7262e-02, -3.7165e-02, ..., 7.9255e-02,
6.6890e-02, 4.4563e-01],
[-1.5484e+00, 3.0902e-02, -3.3076e-02, ..., 1.1914e-01,
9.6334e-02, 6.2985e-01],
[-1.3787e+00, -5.6516e-01, -4.2500e-01, ..., -9.3846e-02,
6.3405e-02, 3.6025e-01]],
...,
[[ 1.0393e+00, -4.1589e-01, -2.3872e-01, ..., -1.1181e-01,
-3.5799e-01, -9.2051e-01],
[-8.2873e-01, -9.3412e-01, -8.1566e-01, ..., -6.5815e-01,
-1.2384e+00, -8.2511e-01],
[-6.7801e-01, -9.0792e-01, -6.6278e-01, ..., -8.5021e-02,
-1.9281e-01, -4.0256e-01],
...,
[-6.6101e-01, -1.0714e+00, -8.2856e-01, ..., -1.2345e-01,
-1.9689e-01, -3.7578e-01],
[-2.7277e+00, -2.7834e+00, -1.8955e+00, ..., -8.2593e-01,
-8.6170e-01, -7.9640e-01],
[-6.8449e-01, -1.6588e+00, -1.0239e+00, ..., -3.1831e-02,
-1.5795e-01, -3.2355e-01]],
[[ 4.2053e-03, -1.9250e-02, 5.6951e-01, ..., 3.1124e-01,
6.3135e-01, 2.9722e-01],
[-2.9246e-02, -7.2814e-01, 9.2199e-02, ..., -3.1930e-02,
5.6322e-02, -4.3891e-02],
[-9.9415e-02, -7.6138e-01, 3.2811e-01, ..., -1.1498e-01,
-9.9680e-02, 1.7942e-01],
...,
[ 1.3465e-01, -7.2301e-01, 2.9109e-01, ..., -1.5354e-01,
-7.4781e-02, 1.4160e-01],
[-2.3157e-01, -1.3213e+00, 9.3443e-02, ..., -4.5736e-01,
-3.3333e-01, -5.5774e-02],
[ 6.5348e-01, 2.1716e-01, 3.0682e-01, ..., -2.5976e-01,
-2.3591e-01, 3.1953e-01]],
[[-1.5968e+00, -5.9845e-01, 7.3925e-02, ..., 5.5691e-01,
3.6299e-01, 5.6912e-01],
[-2.0691e+00, -3.4420e-01, 1.0108e-03, ..., -5.3670e-02,
-1.4279e-01, 4.1721e-01],
[-1.7421e+00, -4.2696e-01, 1.8505e-01, ..., 5.5295e-02,
-1.3607e-01, 6.5142e-02],
...,
[-1.8071e+00, -4.4142e-01, 3.4620e-01, ..., 2.0083e-02,
-4.8282e-02, 8.2600e-02],
[-2.3091e+00, -1.1584e+00, -4.2003e-02, ..., -1.1798e-01,
-2.3669e-01, -2.1359e-01],
[-1.1310e+00, -2.5738e-01, 4.5419e-01, ..., -1.1505e-01,
-2.5195e-01, 1.2728e-01]]],
[[[ 1.5186e-02, -3.4204e-01, -1.3836e-01, ..., -3.2496e+00,
-9.4998e-01, -1.2969e+00],
[ 2.6547e-01, -1.0280e+00, -7.5382e-01, ..., -6.6313e+00,
-2.3202e+00, -1.5143e+00],
[ 2.3690e-01, -6.4553e-01, -5.2149e-01, ..., -9.6030e+00,
-3.4631e+00, -1.8513e+00],
...,
[-1.6359e+00, 5.2082e-01, 4.0304e-01, ..., -4.4459e+00,
-6.7717e+00, -2.2535e+00],
[-3.7120e-01, -8.6553e-01, -6.3902e-01, ..., -6.4822e-01,
-2.0498e+00, -2.4566e+00],
[-6.7402e-01, -1.3014e+00, -1.1302e+00, ..., -1.4225e+00,
-1.5719e+00, -9.3145e-01]],
[[-9.6302e-01, -2.5017e+00, -2.3042e+00, ..., -9.3223e+00,
-9.3304e+00, -3.1110e+00],
[-2.2804e+00, -4.0704e+00, -3.4419e+00, ..., -1.6724e+01,
-1.4424e+01, -4.9528e+00],
[-1.8686e+00, -2.9194e+00, -3.0222e+00, ..., -2.1422e+01,
-1.5232e+01, -4.0215e+00],
...,
[-2.6681e+00, -4.2094e+00, -5.7614e+00, ..., -9.7479e+00,
-8.2717e+00, -2.3133e+00],
[-4.0192e+00, -5.4832e+00, -4.6694e+00, ..., -6.1187e+00,
-7.3527e+00, -2.7766e+00],
[ 3.5446e-01, -1.9106e-01, 5.1514e-01, ..., 4.2748e-01,
6.0819e-01, 6.1388e-02]],
[[-1.8872e-01, 2.4813e-01, 7.6587e-01, ..., -3.8886e-01,
9.1108e-01, 2.1378e+00],
[ 1.2099e-02, -3.4077e-01, 5.3525e-01, ..., -7.3418e+00,
-1.5818e+00, 1.7234e+00],
[-2.3847e-01, -2.3684e-01, 1.8518e-01, ..., -1.5431e+01,
-6.8768e+00, -8.5696e-01],
...,
[ 2.0033e+00, -3.3950e+00, -2.4612e+00, ..., -6.2399e+00,
3.1052e+00, -2.2037e+00],
[ 1.2455e+00, -9.8411e-01, -1.3520e-01, ..., -9.9379e-01,
1.6046e+00, 5.5240e-01],
[-2.6971e-02, -3.0022e-01, 1.2506e-01, ..., -4.5685e-01,
-1.3701e-01, -4.5136e-01]],
...,
[[ 6.8467e-01, -4.7336e-02, 1.9178e-01, ..., 9.9223e-01,
8.0104e-01, -9.9366e-01],
[-4.4716e-01, -4.2805e-01, -4.3292e-01, ..., -3.8747e-01,
-2.0304e+00, -3.2448e+00],
[-2.8895e-01, -1.4548e-01, -1.5858e-01, ..., -5.5841e+00,
-4.4490e+00, -5.2487e+00],
...,
[ 7.1776e-02, -5.0923e+00, -7.6590e+00, ..., -2.9474e+00,
-7.1911e+00, -6.6404e+00],
[-3.6330e+00, -3.1655e+00, -1.9103e+00, ..., -4.1476e+00,
-4.8420e+00, -2.5977e+00],
[-4.7936e-01, -8.0383e-01, -2.7248e-01, ..., -8.7160e-01,
-9.5138e-01, -1.0271e+00]],
[[ 2.1252e-01, 2.8819e-01, 5.7342e-01, ..., -6.8430e-01,
4.3025e-01, -9.4843e-01],
[ 1.7770e-01, 2.4009e-01, 4.1917e-01, ..., -4.9175e+00,
-2.6142e-01, -7.5057e-01],
[-1.1141e-01, 9.7201e-02, 3.1817e-01, ..., -1.1160e+01,
-4.8218e-02, 1.5751e+00],
...,
[-2.0499e+00, 2.0552e+00, 1.4190e+00, ..., -4.6211e+00,
-2.0446e+00, 1.0831e+00],
[-9.7697e-01, -3.4077e-01, -3.5461e-01, ..., -1.5337e+00,
-1.9062e+00, -8.3352e-01],
[ 5.6522e-01, 3.5046e-01, 1.6266e-01, ..., 1.8231e-01,
1.7035e-01, 2.9957e-01]],
[[-7.7696e-01, 1.9745e-01, 2.7481e-01, ..., 2.2942e-01,
5.9975e-01, 1.5443e+00],
[-1.1056e+00, 1.2812e-01, -3.5940e-01, ..., -4.6095e+00,
-3.0619e+00, 2.9429e+00],
[-9.5606e-01, 2.4218e-01, -5.5794e-02, ..., -1.1245e+01,
-9.4296e+00, 3.5152e+00],
...,
[-5.0799e+00, 9.0932e-01, 4.5413e+00, ..., 2.3131e+00,
-3.5457e+00, 1.7456e+00],
[-1.9656e+00, 3.8353e-01, 6.5345e-01, ..., 1.3112e+00,
1.0934e+00, 1.9780e+00],
[-7.5590e-01, -1.6409e-01, -2.5239e-01, ..., 2.5616e-01,
2.0690e-01, 1.2319e+00]]]], device='cuda:0'),
tensor([[[[-7.8097e-01, -1.3823e+00, -1.4173e+00, ..., 9.8233e-02,
1.1676e-01, -5.0561e-01],
[-3.3616e-01, -2.3961e+00, -2.3411e+00, ..., -3.3146e-01,
-3.2015e-01, -5.3162e-01],
[-3.4213e-01, -2.6973e+00, -2.8191e+00, ..., 5.9426e-01,
-1.0690e-01, -1.4552e+00],
...,
[ 2.8919e-01, -1.6459e+00, -2.4112e+00, ..., -2.0947e+00,
-3.4526e+00, -2.2742e+00],
[-1.1436e+00, -1.7807e+00, -1.1661e+00, ..., -3.2145e+00,
-1.0503e+00, -1.2334e+00],
[-1.1008e+00, -1.9766e+00, -1.4653e+00, ..., -2.2514e+00,
-2.8024e-01, -1.3299e+00]],
[[-2.5458e+00, -4.9737e+00, -4.3482e+00, ..., -3.6416e+00,
-4.3447e+00, -1.8913e+00],
[-5.6440e+00, -8.6609e+00, -7.6668e+00, ..., -4.2088e+00,
-4.7686e+00, -2.5664e+00],
[-6.0053e+00, -8.2261e+00, -8.5053e+00, ..., -4.2939e+00,
-5.0721e+00, -3.1265e+00],
...,
[-4.6022e+00, -6.3936e+00, -6.3543e+00, ..., -1.0400e+01,
-6.5074e+00, -2.1392e+00],
[-6.7420e+00, -7.9576e+00, -6.8192e+00, ..., -7.6928e+00,
-5.6143e+00, -2.8674e+00],
[-9.6522e-02, -1.0655e+00, 3.8954e-01, ..., -1.1892e+00,
2.0547e-01, -8.2597e-01]],
[[-5.8218e-01, 6.8585e-01, 1.0588e+00, ..., 7.6952e-01,
4.7268e-01, 1.2747e+00],
[-9.3067e-01, -2.7841e-02, 3.6350e-01, ..., -4.9397e-01,
-3.5178e-01, 8.6795e-02],
[-1.9461e+00, -7.6876e-01, -2.3333e-01, ..., -1.4094e-01,
-4.0533e-02, 4.5329e-01],
...,
[-8.0989e-01, -1.1768e+00, 6.8225e-01, ..., -9.1729e+00,
-7.0757e+00, -1.6468e+00],
[-1.3780e+00, 3.1864e-02, -1.7479e-02, ..., -4.0325e+00,
-3.7971e+00, -1.8089e+00],
[-1.4403e+00, -6.3118e-01, -2.4967e-01, ..., 2.4192e+00,
-5.6953e-01, -1.4365e+00]],
...,
[[ 1.2282e+00, -3.5091e-01, -2.3112e-01, ..., 1.2687e-01,
4.8493e-01, -6.3470e-01],
[-5.7472e-01, -9.5174e-01, -8.9391e-01, ..., -2.3235e-01,
-1.1099e+00, -1.2844e+00],
[-4.4840e-01, -8.4126e-01, -8.0628e-01, ..., -1.5506e+00,
-8.4434e-01, -9.2041e-01],
...,
[-1.6950e+00, -1.6694e+00, -2.5472e+00, ..., -4.3520e+00,
-2.6491e+00, -5.9168e+00],
[-3.3885e+00, -3.3010e+00, -2.2994e+00, ..., -3.3804e+00,
-4.1354e+00, -6.3251e+00],
[-6.5655e-01, -1.7488e+00, -1.0674e+00, ..., -2.7373e+00,
-4.0277e+00, -3.0412e+00]],
[[ 1.0658e-01, 5.7827e-02, 5.9486e-01, ..., 8.7839e-01,
5.4351e-01, -3.2913e-01],
[ 2.6845e-01, -5.5748e-01, 3.6029e-02, ..., 4.1062e-01,
8.8423e-01, -5.3170e-01],
[ 4.1585e-01, -4.2442e-01, 2.5669e-01, ..., 1.6102e-01,
7.1113e-01, -5.7612e-01],
...,
[ 2.7982e-01, -1.5520e+00, -7.4013e-01, ..., -4.0139e+00,
-3.8784e+00, 1.6565e-02],
[-3.2055e-01, -1.6606e+00, -3.0105e-01, ..., -2.9574e+00,
-1.3788e+00, 3.2629e-01],
[ 5.6441e-01, 1.2519e-01, 1.8892e-01, ..., -5.7349e-01,
2.0862e+00, 3.5958e-01]],
[[-1.5614e+00, -6.5970e-01, -4.7345e-03, ..., -2.1519e-01,
-9.8754e-02, 8.2113e-01],
[-2.1024e+00, -3.9096e-01, -8.8544e-02, ..., -1.5806e+00,
-9.4007e-01, 1.0447e+00],
[-1.8606e+00, -9.0192e-01, -5.3573e-01, ..., -1.5755e+00,
-3.9981e-01, 1.0047e+00],
...,
[-2.0601e+00, 2.2742e-01, 2.8012e-01, ..., -8.3641e+00,
-5.6680e+00, -1.7046e+00],
[-2.4470e+00, -1.2242e+00, -1.3747e-01, ..., -6.1636e+00,
-2.8481e+00, 2.0798e+00],
[-1.1408e+00, -4.4312e-01, 3.3309e-01, ..., -2.3016e+00,
-3.6467e-01, 2.6661e+00]]],
[[[-2.1460e-01, -2.7474e-01, 2.0240e-01, ..., 4.2379e-02,
5.0920e-01, 1.6955e-01],
[-1.0901e-01, -1.5413e+00, -8.8036e-01, ..., -7.1863e-01,
-4.5561e-01, -3.1130e-01],
[-1.0778e-01, -1.3546e+00, -5.6739e-01, ..., -5.0156e-01,
-3.8669e-01, -6.5830e-01],
...,
[-2.6379e-01, -1.7088e+00, -1.2166e+00, ..., -9.1703e-01,
-9.3672e-01, -2.0292e+00],
[-1.0500e+00, -1.7084e+00, -7.9282e-01, ..., -5.2856e-01,
-4.6216e-01, -1.7845e+00],
[-1.2529e+00, -2.1473e+00, -1.6131e+00, ..., -1.4223e+00,
-1.5584e+00, -9.6915e-01]],
[[-1.7149e-01, -1.4211e+00, -5.5215e-01, ..., -6.2246e-01,
-1.3450e+00, -2.1944e-01],
[-2.0012e+00, -3.9572e+00, -2.4026e+00, ..., -2.3830e+00,
-3.3785e+00, -1.1463e+00],
[-1.8678e+00, -3.3938e+00, -3.1590e+00, ..., -2.4886e+00,
-2.9405e+00, -1.2376e+00],
...,
[-3.7434e+00, -5.5250e+00, -5.4686e+00, ..., -5.3871e+00,
-6.1519e+00, -2.8775e+00],
[-6.1791e+00, -7.0315e+00, -6.0676e+00, ..., -6.0735e+00,
-7.4108e+00, -2.7126e+00],
[-9.8397e-03, -8.4686e-01, 2.7698e-01, ..., 4.1083e-01,
6.9792e-01, 1.9043e-01]],
[[-5.1378e-01, -1.5198e-01, 3.4183e-01, ..., 3.6954e-01,
5.5617e-02, 8.8664e-01],
[-3.8243e-01, -4.4190e-01, 4.4724e-01, ..., 3.7104e-01,
2.7392e-01, 4.9778e-01],
[-8.7279e-01, -2.0400e-01, 1.2153e-01, ..., 1.2844e-02,
1.6936e-01, 4.1149e-01],
...,
[-1.3795e+00, 2.8926e-02, 1.0087e+00, ..., 4.3893e-02,
2.9767e-02, 3.0846e-01],
[-1.2324e+00, 1.8339e-01, 1.7610e-03, ..., -4.6487e-02,
1.4923e-02, 2.9336e-01],
[-1.1960e+00, -5.1079e-02, 1.0884e-01, ..., -7.0148e-03,
-4.3636e-02, -3.3357e-01]],
...,
[[ 7.1589e-02, -6.0397e-01, -3.8602e-01, ..., -5.7970e-02,
-2.4834e-01, -7.4044e-01],
[-8.5485e-01, -6.3309e-01, -5.3462e-01, ..., -2.9381e-01,
-9.9433e-01, -7.4705e-01],
[-5.0334e-01, -1.2821e-01, -5.3117e-02, ..., 1.1364e-01,
-1.8364e-01, -5.1919e-01],
...,
[-6.1562e-01, -6.1062e-01, -7.9873e-01, ..., -3.6009e-01,
-5.7772e-01, -1.1821e+00],
[-3.0736e+00, -2.8781e+00, -1.9952e+00, ..., -1.9089e+00,
-2.0197e+00, -2.0560e+00],
[-5.3519e-01, -1.6346e+00, -1.1121e+00, ..., -8.5947e-01,
-9.3502e-01, -1.3420e+00]],
[[-1.8349e-01, 2.7717e-02, 4.3010e-01, ..., 9.5295e-02,
5.4723e-01, 1.8938e-01],
[-4.1911e-01, -2.8206e-01, -1.2934e-02, ..., -4.5601e-02,
1.0424e-01, -6.5999e-02],
[-6.5279e-01, -5.9055e-01, 3.5494e-02, ..., 5.1004e-02,
5.0864e-02, -2.8441e-02],
...,
[-1.4441e-01, -1.0185e+00, 2.3168e-01, ..., 2.7099e-01,
1.4756e-01, -7.0804e-01],
[-5.0123e-01, -1.8326e+00, -1.9540e-01, ..., 7.6754e-03,
-3.5899e-02, -9.3292e-01],
[ 4.5819e-01, -1.5927e-01, 1.1651e-01, ..., 2.6644e-01,
3.1079e-01, 3.3146e-01]],
[[-6.6118e-01, 3.9982e-01, 5.5040e-01, ..., 4.6735e-01,
3.7918e-01, 6.0848e-01],
[-8.1018e-01, 2.8396e-01, -2.6157e-01, ..., -2.7157e-01,
-1.6159e-01, 6.5956e-01],
[-4.4764e-01, 3.4649e-01, 2.0679e-02, ..., 8.5962e-02,
9.8434e-02, 6.2391e-01],
...,
[-1.0865e+00, -9.4604e-01, -8.6069e-01, ..., -3.9619e-01,
-6.0868e-02, 8.3403e-01],
[-1.7751e+00, -1.8073e+00, -9.7682e-01, ..., -8.2507e-01,
-4.7248e-01, 4.3184e-01],
[-9.3546e-01, -9.5104e-01, -2.4638e-01, ..., -2.9299e-01,
-1.5625e-01, 1.1186e+00]]],
[[[-1.2661e+01, -1.5497e+01, -7.2584e+00, ..., -2.6609e-01,
3.4194e-01, 2.0633e-01],
[-1.0041e+01, -8.1396e+00, -5.6933e+00, ..., -1.1903e+00,
-3.8398e-01, -8.7672e-02],
[-8.4725e+00, -2.7275e+01, -2.6681e+01, ..., -6.3068e-01,
-3.7448e-01, -4.1207e-01],
...,
[-6.5289e-01, -2.2775e+00, -2.2263e+00, ..., -5.3547e-01,
-3.7815e-01, -4.5052e-01],
[-1.3661e+00, -2.1790e+00, -1.4244e+00, ..., -1.4101e-01,
9.1998e-02, -2.1461e-01],
[-9.9368e-01, -2.0077e+00, -1.5608e+00, ..., -1.0271e+00,
-1.1720e+00, -4.3915e-01]],
[[-2.4142e+01, -2.9924e+01, -2.6097e+01, ..., -1.4093e+00,
-1.8754e+00, -4.1033e-01],
[-7.1227e+00, -1.1853e+01, -1.1701e+01, ..., -4.0594e+00,
-3.6155e+00, -1.1541e+00],
[-2.7387e+01, -2.3550e+01, -1.7323e+01, ..., -5.8765e+00,
-2.3601e+00, -8.2143e-01],
...,
[-5.0431e+00, -6.9608e+00, -6.1816e+00, ..., -2.3189e+00,
-2.4782e+00, -8.4969e-01],
[-6.9356e+00, -8.1860e+00, -7.1180e+00, ..., -3.4345e+00,
-4.2390e+00, -1.4030e+00],
[-2.8042e-01, -1.0155e+00, 9.6944e-02, ..., 4.5655e-01,
4.5903e-01, 1.0320e-01]],
[[-1.3722e+01, -1.6170e+01, -2.4283e+01, ..., 9.9235e-01,
6.1114e-01, 1.3349e+00],
[-1.8849e+01, -3.2541e+01, -5.2303e+01, ..., 5.8004e-01,
3.0259e-01, 4.5397e-01],
[-2.7611e+01, -3.5105e+01, -5.6659e+01, ..., 8.5291e-01,
2.5624e-01, 4.4425e-01],
...,
[-1.8354e+00, -1.5815e-01, -1.9462e-01, ..., 7.1770e-02,
2.6547e-01, 4.8193e-01],
[-1.5248e+00, -1.2972e-02, -8.3247e-02, ..., 1.1459e-01,
2.8270e-01, 6.3601e-01],
[-1.5507e+00, -7.2169e-01, -3.9344e-01, ..., -1.3753e-01,
6.6362e-02, 3.0054e-01]],
...,
[[ 3.2099e+00, -8.0544e+00, -1.5159e+01, ..., -1.1048e-02,
-2.2487e-01, -1.0315e+00],
[-7.7922e+00, -2.7352e+01, -3.5280e+01, ..., -7.8432e-01,
-1.4505e+00, -1.0580e+00],
[-1.1623e+01, -2.7702e+01, -3.0694e+01, ..., -8.2732e-01,
-3.4870e-01, -5.8452e-01],
...,
[-6.6528e-01, -1.2439e+00, -7.8193e-01, ..., -3.0635e-01,
-3.3814e-01, -5.4772e-01],
[-3.0123e+00, -3.4165e+00, -2.1647e+00, ..., -1.0201e+00,
-1.0194e+00, -9.2561e-01],
[-6.7000e-01, -1.9627e+00, -1.2191e+00, ..., -1.7622e-01,
-2.2598e-01, -4.0221e-01]],
[[-8.2201e+00, -1.5997e+01, -1.7478e+01, ..., 4.3942e-01,
7.9361e-01, 3.4156e-01],
[-9.6881e+00, -1.3401e+01, -2.0695e+01, ..., -8.6729e-01,
2.4256e-01, -1.0495e-01],
[-1.2089e+01, -2.8957e+01, -3.2169e+01, ..., -1.6539e+00,
-1.6185e-01, -3.7453e-02],
...,
[ 2.0665e-01, -9.7302e-01, 2.9468e-01, ..., -8.2652e-03,
-2.0335e-01, -1.0063e-01],
[-2.6135e-02, -1.3774e+00, 1.5065e-01, ..., -3.5556e-01,
-4.4935e-01, -3.3599e-01],
[ 6.6626e-01, 1.4581e-01, 2.3046e-01, ..., -8.9789e-02,
-1.7229e-01, 2.0705e-01]],
[[-1.4539e+01, -1.4308e+01, -2.4854e+01, ..., 6.1811e-01,
3.8190e-01, 7.5962e-01],
[-2.0117e+01, -1.8950e+01, -2.7873e+01, ..., 1.6071e-01,
-6.0716e-02, 7.2023e-01],
[-2.0596e+01, -1.9721e+01, -2.3042e+01, ..., 5.7748e-01,
-1.1551e-01, 3.0141e-01],
...,
[-2.2610e+00, -5.7971e-01, 1.1656e-01, ..., 2.1501e-01,
2.3596e-02, 3.0790e-01],
[-2.9159e+00, -1.4814e+00, -4.0971e-01, ..., 2.5388e-02,
-1.4255e-01, 4.5433e-02],
[-1.3619e+00, -2.1950e-01, 4.3918e-01, ..., -1.3467e-01,
-3.1932e-01, 3.0026e-01]]],
[[[-3.1753e-01, -1.0283e+00, -8.4891e-01, ..., -5.4117e-02,
6.1914e-01, 5.5223e-01],
[ 1.3971e-01, -1.8723e+00, -1.2698e+00, ..., -5.0130e-01,
-2.3044e-01, 1.3239e-01],
[ 1.2491e-01, -1.4034e+00, -9.0810e-01, ..., -4.8672e-01,
-1.2298e-01, -7.7408e-02],
...,
[ 5.8167e-01, -1.6194e+00, -1.4705e+00, ..., -4.2549e-01,
-1.8255e-01, -1.6537e-01],
[ 1.3338e-01, -8.3855e-01, -4.6064e-01, ..., 2.7959e-02,
2.3801e-01, -3.8411e-02],
[-1.2413e-01, -1.0722e+00, -9.2105e-01, ..., -1.0370e+00,
-1.0880e+00, -3.9566e-01]],
[[-2.0771e+00, -4.0231e+00, -3.6615e+00, ..., -5.1589e-01,
-1.0429e+00, -8.8277e-03],
[-3.9231e+00, -6.3020e+00, -5.5317e+00, ..., -1.8527e+00,
-2.6764e+00, -6.7589e-01],
[-3.2053e+00, -5.0730e+00, -5.0562e+00, ..., -2.2198e+00,
-1.5045e+00, -3.6809e-01],
...,
[-3.2071e+00, -4.7538e+00, -4.7313e+00, ..., -1.5477e+00,
-1.6520e+00, -4.6223e-01],
[-5.0542e+00, -5.9247e+00, -5.5176e+00, ..., -2.4176e+00,
-3.3186e+00, -9.8854e-01],
[ 7.2809e-01, -1.1211e-01, 4.9892e-01, ..., 3.4682e-01,
3.6753e-01, 8.3482e-02]],
[[-5.7712e-01, 4.3960e-01, 8.2467e-01, ..., 8.2555e-01,
3.1261e-01, 1.1111e+00],
[-9.4558e-01, 1.0905e-01, 6.1689e-01, ..., 4.3441e-01,
4.8427e-02, 4.7995e-01],
[-1.3333e+00, 2.2233e-02, -1.9535e-02, ..., 2.1457e-01,
1.0547e-01, 4.1836e-01],
...,
[-1.6343e+00, -2.0569e-01, 3.4530e-01, ..., -3.1071e-02,
4.5560e-03, 4.4327e-01],
[-1.2275e+00, 2.5491e-01, -1.1875e-01, ..., 1.1875e-01,
6.0615e-02, 6.2080e-01],
[-5.8161e-01, 2.9229e-01, 1.6723e-01, ..., -9.0299e-02,
3.5156e-02, 3.6393e-01]],
...,
[[ 6.8069e-01, 1.0640e-01, 2.7334e-01, ..., -1.0192e-01,
-3.8887e-01, -9.3825e-01],
[-8.2237e-01, -3.5557e-01, -7.2431e-02, ..., -6.9674e-01,
-1.2892e+00, -8.3634e-01],
[-5.1607e-01, -3.6970e-01, -3.0320e-02, ..., -7.8294e-02,
-2.0309e-01, -3.8356e-01],
...,
[-9.6056e-01, -7.9345e-01, -5.2074e-01, ..., -2.7496e-01,
-2.0715e-01, -3.7106e-01],
[-2.1938e+00, -1.7884e+00, -1.3008e+00, ..., -8.2401e-01,
-8.7449e-01, -8.0213e-01],
[-4.4174e-01, -1.0301e+00, -3.4719e-01, ..., -4.5631e-02,
-1.3954e-01, -3.1891e-01]],
[[ 1.2667e-02, -2.5876e-01, -6.7568e-02, ..., 3.0037e-01,
6.7976e-01, 3.0815e-01],
[ 1.2404e-01, -9.2098e-01, -7.0189e-02, ..., -1.4757e-02,
1.2274e-01, -4.6653e-02],
[-2.1298e-01, -9.4067e-01, 1.2201e-01, ..., -8.3418e-01,
-1.0394e-01, 2.0968e-01],
...,
[-3.2573e-02, -1.5609e+00, 6.7764e-03, ..., -3.4291e-01,
-1.9416e-01, 1.0807e-01],
[-4.7560e-01, -1.9372e+00, -3.4209e-01, ..., -3.7347e-01,
-3.2312e-01, -7.1924e-02],
[ 7.2362e-01, -1.3674e-01, 2.3919e-01, ..., -2.4370e-01,
-2.1002e-01, 3.0243e-01]],
[[-6.2630e-01, -1.1664e-01, 1.3891e-02, ..., 5.6774e-01,
3.6724e-01, 5.5055e-01],
[-1.0688e+00, -8.2922e-01, -9.4691e-01, ..., -3.1699e-02,
-1.5250e-01, 4.1384e-01],
[-6.5966e-01, -7.2647e-01, -5.6709e-01, ..., 3.7380e-02,
-1.7263e-01, 1.0463e-01],
...,
[-9.4160e-02, -5.0706e-01, -4.7586e-01, ..., 1.3781e-01,
1.5101e-02, 7.7757e-02],
[-8.2634e-01, -9.8790e-01, -5.4255e-01, ..., -1.4467e-01,
-2.2452e-01, -2.1082e-01],
[-4.7729e-01, -8.5469e-01, -3.4894e-01, ..., -1.2862e-01,
-2.3758e-01, 1.1689e-01]]],
[[[-7.1950e-01, -1.4667e+00, -1.4429e+00, ..., -1.2499e+00,
-8.8998e-01, -1.6754e+00],
[-3.1890e-01, -2.3698e+00, -2.2280e+00, ..., -1.7513e+00,
-1.8936e+00, -1.9155e+00],
[-1.3731e-01, -1.8674e+00, -1.6953e+00, ..., -1.3313e+00,
-1.4308e+00, -2.0871e+00],
...,
[-6.8792e-02, -1.1081e+00, -5.2172e-01, ..., -6.0377e-01,
-4.5892e-01, -9.0560e-01],
[-1.3303e-01, -1.0170e+00, -1.8719e-01, ..., -1.9554e-01,
-8.7954e-02, -9.7951e-01],
[-1.3827e-01, -1.1033e+00, -7.2412e-01, ..., -8.6569e-01,
-1.1077e+00, -6.1956e-01]],
[[-2.6940e+00, -5.0910e+00, -4.5873e+00, ..., -4.8017e+00,
-5.9980e+00, -2.4395e+00],
[-5.2560e+00, -8.0603e+00, -7.4417e+00, ..., -7.7948e+00,
-9.1904e+00, -4.3484e+00],
[-3.9889e+00, -5.7646e+00, -5.7102e+00, ..., -6.1875e+00,
-6.5883e+00, -3.2660e+00],
...,
[-1.9851e+00, -3.1099e+00, -3.1530e+00, ..., -3.0810e+00,
-3.4286e+00, -1.4759e+00],
[-4.5153e+00, -5.5058e+00, -5.2022e+00, ..., -5.0422e+00,
-6.2319e+00, -2.3774e+00],
[ 7.0216e-01, -2.8849e-01, 6.4173e-01, ..., 5.0031e-01,
5.7134e-01, -3.1786e-03]],
[[-6.3255e-01, 4.9365e-01, 8.7464e-01, ..., 8.3170e-01,
6.4711e-01, 1.9253e+00],
[-9.3720e-01, 6.5026e-02, 4.4001e-01, ..., 5.1749e-01,
-5.5303e-03, 1.1917e-01],
[-1.7383e+00, -2.0845e-01, -6.4615e-02, ..., 7.8132e-02,
2.6109e-02, -8.4138e-02],
...,
[-8.8691e-01, -1.8997e-01, 4.8507e-02, ..., 6.7785e-02,
1.0205e-01, 3.4727e-01],
[-9.4458e-01, 1.6808e-02, 1.7586e-01, ..., 1.0210e-02,
1.0951e-01, 3.8018e-01],
[-4.4101e-01, -1.8632e-02, 2.1205e-01, ..., -1.7048e-02,
9.8431e-02, 4.0044e-02]],
...,
[[ 1.2503e+00, -3.5552e-01, -7.3241e-02, ..., 2.4288e-01,
-8.3627e-02, -2.1049e+00],
[-6.6859e-01, -9.1248e-01, -5.1260e-01, ..., -4.9742e-01,
-1.4710e+00, -2.0708e+00],
[-7.2468e-01, -8.8749e-01, -2.7837e-01, ..., -2.6383e-01,
-5.9929e-01, -1.4208e+00],
...,
[-4.3000e-01, -1.5659e-01, 2.7897e-01, ..., 8.5345e-02,
-1.2601e-01, -5.3009e-01],
[-2.0006e+00, -1.3750e+00, -7.5214e-01, ..., -9.6719e-01,
-1.1426e+00, -1.1099e+00],
[-2.7198e-01, -7.3324e-01, -9.0425e-02, ..., -1.4199e-01,
-2.8299e-01, -6.1445e-01]],
[[ 7.4539e-02, 2.4826e-02, 4.2131e-01, ..., 4.8840e-01,
9.2464e-01, 1.0897e-03],
[ 2.9217e-01, -7.3918e-01, -2.4372e-02, ..., 4.3771e-01,
3.5135e-01, -8.3076e-01],
[ 2.0498e-01, -6.2984e-01, 2.7250e-01, ..., 6.3656e-01,
4.6562e-01, -4.0659e-01],
...,
[-3.6973e-01, -4.6389e-01, 2.1008e-01, ..., 6.3951e-02,
6.6783e-02, -1.1309e-01],
[-7.4906e-01, -1.5212e+00, -1.2735e-01, ..., -3.2333e-01,
-3.5389e-01, -7.0772e-01],
[ 5.5268e-01, 5.2868e-02, 2.2916e-01, ..., 2.4076e-01,
1.6445e-01, 3.2794e-01]],
[[-1.5046e+00, -7.8354e-01, -4.2350e-01, ..., -4.5003e-01,
-2.7922e-01, 8.7868e-01],
[-2.1087e+00, -7.6172e-01, -5.6393e-01, ..., -8.0649e-01,
-1.8743e-01, 1.6923e+00],
[-1.4930e+00, -5.3823e-01, -2.2168e-01, ..., -4.5214e-01,
-9.9110e-02, 1.4298e+00],
...,
[-4.9758e-01, 1.5090e-01, 1.8122e-03, ..., 2.0505e-01,
2.0700e-01, 6.9979e-01],
[-9.4519e-01, -3.0525e-01, -3.6535e-01, ..., -7.3886e-02,
9.7859e-02, 4.2935e-01],
[-4.3717e-01, -3.3645e-01, -2.5608e-01, ..., -2.3003e-01,
-2.3951e-01, 5.7080e-01]]]], device='cuda:0'),
tensor([[[[-2.8075e-01, -8.1632e-01, -2.2100e+00, ..., -3.1647e+00,
-5.5596e-01, -2.7326e-01],
[ 2.1167e-01, -1.3987e+00, -2.1558e+00, ..., -2.0048e+00,
-2.1412e+00, -1.4574e+00],
[ 8.9853e-02, -1.0927e+00, -2.1024e+00, ..., -1.2512e-01,
-1.1875e+00, -2.0024e+00],
...,
[-1.9074e+00, -1.0098e+01, -6.0224e+00, ..., -6.4092e+00,
-1.2437e+00, -2.4697e+00],
[-4.7080e+00, -9.7462e+00, -2.9105e+00, ..., -6.8185e+00,
-8.4267e-01, -2.1739e+00],
[-1.6108e+00, -2.7797e+00, -1.3453e+00, ..., -3.7349e+00,
-1.8979e+00, -1.2426e+00]],
[[-1.9650e+00, -6.4422e+00, -6.7284e+00, ..., -5.4054e+00,
-8.4914e+00, -4.0159e+00],
[-3.8132e+00, -1.0554e+01, -7.3133e+00, ..., -2.6169e+00,
-1.1226e+01, -5.0143e+00],
[-2.8917e+00, -8.5410e+00, -6.2838e+00, ..., -1.9447e+00,
-8.1777e+00, -2.7560e+00],
...,
[-1.1803e+01, -1.9972e+01, -1.0797e+01, ..., -1.3470e+01,
-7.3164e+00, -3.1254e+00],
[-1.5695e+01, -2.3167e+01, -1.1905e+01, ..., -1.2708e+01,
-8.6221e+00, -2.9098e+00],
[-9.8836e-02, -6.8487e+00, -3.3791e+00, ..., 1.4510e+00,
4.5686e-01, 3.4850e-01]],
[[-5.6198e-01, 1.7480e-01, 3.0751e-01, ..., -3.8987e+00,
6.0735e-01, 2.7740e+00],
[-9.3795e-01, -1.9097e-02, 2.1982e-01, ..., -6.4507e+00,
-1.8674e-01, 1.5110e+00],
[-1.2026e+00, 6.8722e-01, -3.2678e+00, ..., -6.6180e+00,
1.8430e-01, 6.6685e-01],
...,
[-1.7198e+01, -1.7529e+01, -5.1706e+00, ..., 5.0617e+00,
-1.4122e-01, 6.9647e-02],
[-1.7730e+01, -1.8649e+01, -7.9228e+00, ..., 3.4534e+00,
4.5607e-02, 1.3608e-01],
[-9.8168e+00, -1.5307e+01, -1.3674e+01, ..., 2.8152e+00,
-5.5295e-02, -5.6553e-01]],
...,
[[ 7.7643e-01, 2.4139e-01, 1.6649e+00, ..., -1.6701e-01,
-3.2912e+00, -2.4641e+00],
[-1.0507e+00, 3.6587e-02, -8.0710e-01, ..., -3.2182e+00,
-6.0813e+00, -2.4936e+00],
[-6.8659e-01, 8.2100e-01, -2.4970e-01, ..., -5.4345e+00,
-5.2992e+00, -1.1442e+00],
...,
[-8.3486e+00, -1.3644e+01, 3.3514e+00, ..., -6.5229e+00,
-4.7674e-01, -1.1807e+00],
[-2.6270e+00, -1.2899e+01, -1.1639e+01, ..., -1.0421e+01,
-2.4374e+00, -2.4333e+00],
[-1.2910e+00, -9.6959e+00, -6.4900e+00, ..., -6.3279e+00,
-9.8646e-01, -1.6591e+00]],
[[ 1.2404e-02, 4.9707e-01, 1.3057e+00, ..., 8.7359e-01,
2.1816e+00, 4.4993e-01],
[-5.2147e-02, -3.9506e-01, -3.4904e-01, ..., 3.6941e+00,
1.2654e+00, -7.6339e-01],
[-2.1810e-01, -1.5364e+00, -1.9992e+00, ..., 3.6615e+00,
9.0782e-01, -1.0258e+00],
...,
[-7.8181e+00, -8.1988e+00, -4.0136e+00, ..., -8.3916e-01,
5.2613e-02, -7.8355e-01],
[-8.0684e+00, -7.6683e+00, -6.5783e+00, ..., -5.0912e+00,
-2.3895e-02, -8.6991e-01],
[-6.0306e-01, -7.1173e+00, -6.7103e+00, ..., -7.5867e-01,
4.1051e-01, 5.9253e-01]],
[[-8.9700e-01, -5.6755e-01, -1.4527e+00, ..., -4.6408e+00,
-7.8390e-01, 7.6402e-01],
[-1.0912e+00, -1.3233e+00, -2.6891e+00, ..., -4.6763e+00,
1.7141e+00, 2.1742e+00],
[-7.0850e-01, -1.7625e+00, -1.2441e+00, ..., -3.2474e+00,
1.9015e+00, 1.8985e+00],
...,
[-4.0956e+00, -2.4728e+00, -1.1353e+01, ..., 1.0737e+00,
-3.3057e-01, 8.3052e-01],
[-2.6426e+00, -6.4505e+00, -1.2932e+01, ..., 1.9490e+00,
-1.0037e+00, 3.6271e-01],
[ 1.1368e+00, -8.4981e-01, -6.1972e+00, ..., 1.8400e+00,
-3.0687e-01, 1.3684e+00]]],
[[[-1.3966e-01, -5.2039e-01, -2.1770e-01, ..., 4.5061e-02,
2.4818e-01, -6.4212e-02],
[ 2.8323e-01, -1.2422e+00, -7.3156e-01, ..., -5.1064e-01,
-3.1095e-01, -2.0457e-01],
[ 1.3680e-01, -9.4189e-01, -3.6078e-01, ..., -4.3204e-01,
-1.9996e-01, -5.2363e-01],
...,
[-1.6701e-01, -1.5092e+00, -1.2812e+00, ..., -9.5457e-01,
-9.6375e-01, -2.0533e+00],
[-8.2415e-01, -1.5126e+00, -8.2915e-01, ..., -4.7941e-01,
-4.1323e-01, -1.8089e+00],
[-1.0223e+00, -1.8741e+00, -1.5155e+00, ..., -1.5141e+00,
-1.6809e+00, -1.0604e+00]],
[[-1.4041e+00, -3.0651e+00, -2.5403e+00, ..., -3.1237e+00,
-3.3822e+00, -1.3065e+00],
[-3.1594e+00, -5.0178e+00, -3.5202e+00, ..., -4.5243e+00,
-4.7797e+00, -2.0661e+00],
[-2.0410e+00, -3.0353e+00, -2.2360e+00, ..., -2.6912e+00,
-2.7030e+00, -1.2802e+00],
...,
[-3.7016e+00, -5.3781e+00, -5.2997e+00, ..., -5.4142e+00,
-6.1407e+00, -2.8723e+00],
[-5.9044e+00, -6.6130e+00, -5.9973e+00, ..., -6.3745e+00,
-7.7462e+00, -2.8300e+00],
[ 3.4613e-02, -5.9178e-01, 3.0409e-01, ..., 4.1884e-01,
7.1643e-01, 1.7539e-01]],
[[-4.6083e-01, 1.7959e-01, 8.3421e-01, ..., 9.4762e-01,
5.6948e-01, 1.3510e+00],
[-3.2968e-01, -9.1462e-02, 5.9995e-01, ..., 6.3631e-01,
4.5397e-01, 3.3290e-01],
[-7.7678e-01, 4.5608e-02, 3.0430e-01, ..., 5.1015e-01,
4.1845e-01, 4.5532e-01],
...,
[-1.3260e+00, 5.8045e-02, -3.0824e-02, ..., 7.2360e-02,
-9.0207e-03, 2.9267e-01],
[-1.2400e+00, 1.4679e-01, -5.9893e-02, ..., 1.1475e-02,
-1.0276e-03, 2.9944e-01],
[-1.1023e+00, 1.7465e-02, 9.8784e-02, ..., 1.4598e-02,
-5.4268e-02, -3.4029e-01]],
...,
[[ 8.6314e-01, 1.8951e-01, 2.1496e-01, ..., -1.6969e-01,
1.0706e-01, -9.0488e-01],
[-1.1883e+00, -8.6604e-01, -6.8362e-01, ..., -1.8252e+00,
-1.5226e+00, -1.1141e+00],
[-6.0234e-01, -3.0200e-01, -2.6577e-01, ..., -5.5027e-01,
-5.2325e-01, -6.0389e-01],
...,
[-4.5558e-01, -7.5885e-01, -3.4553e-01, ..., -3.5168e-01,
-5.9175e-01, -1.2134e+00],
[-2.7044e+00, -2.8285e+00, -1.9438e+00, ..., -2.0469e+00,
-2.1570e+00, -2.1196e+00],
[-5.2128e-01, -1.5979e+00, -1.0302e+00, ..., -8.6645e-01,
-9.4875e-01, -1.3960e+00]],
[[ 4.0527e-02, 2.8652e-01, 7.0954e-01, ..., 4.3670e-01,
9.5472e-01, 3.7920e-01],
[-2.0474e-02, -4.1064e-01, 1.5341e-01, ..., 3.4365e-01,
5.6702e-01, -1.8079e-01],
[-2.9015e-01, -5.8848e-01, 1.8964e-01, ..., 2.1277e-01,
3.5191e-01, 3.2498e-02],
...,
[ 3.5630e-02, -9.0400e-01, 5.8191e-02, ..., 2.3395e-01,
1.2602e-01, -7.4307e-01],
[-3.2280e-01, -1.7179e+00, -1.8619e-01, ..., -5.7901e-02,
-1.1490e-01, -1.0311e+00],
[ 5.3190e-01, -9.2105e-02, 1.7306e-01, ..., 2.0187e-01,
2.5213e-01, 3.0115e-01]],
[[-9.8483e-01, -2.3529e-01, 1.4509e-01, ..., 1.2632e-01,
5.9546e-02, 7.2102e-01],
[-1.1090e+00, -1.8607e-01, -2.9040e-01, ..., -8.8334e-02,
-6.6315e-02, 1.1597e+00],
[-7.0743e-01, -1.8527e-01, -2.1306e-01, ..., 6.0032e-02,
-1.4428e-01, 6.8628e-01],
...,
[-1.1250e+00, -8.9396e-01, -3.9762e-01, ..., -4.1453e-01,
-8.9512e-02, 7.9299e-01],
[-1.7280e+00, -1.6509e+00, -8.9721e-01, ..., -8.2218e-01,
-4.7009e-01, 3.7590e-01],
[-9.0114e-01, -8.4418e-01, -2.0650e-01, ..., -3.3957e-01,
-1.9601e-01, 1.1105e+00]]],
[[[-6.9626e-01, -1.4311e+00, -1.4203e+00, ..., -1.2212e+00,
-8.5929e-01, -1.6438e+00],
[-2.9993e-01, -2.4093e+00, -2.2596e+00, ..., -1.7156e+00,
-1.8744e+00, -1.8890e+00],
[-1.2446e-01, -1.8621e+00, -1.6809e+00, ..., -1.2186e+00,
-1.4064e+00, -2.1303e+00],
...,
[-2.2707e-01, -1.3414e+00, -9.6732e-01, ..., -5.5696e-01,
-4.6171e-01, -9.0109e-01],
[-3.4164e-01, 4.4761e-01, 8.8152e-01, ..., -1.8814e-01,
-6.2937e-02, -9.6395e-01],
[-6.8168e-01, -1.3923e+00, -8.4857e-01, ..., -8.4982e-01,
-1.0669e+00, -5.9805e-01]],
[[-2.6232e+00, -5.0726e+00, -4.5836e+00, ..., -4.7671e+00,
-6.0105e+00, -2.4165e+00],
[-5.3297e+00, -8.1665e+00, -7.5478e+00, ..., -7.8267e+00,
-9.2496e+00, -4.3556e+00],
[-3.9904e+00, -5.8108e+00, -5.7768e+00, ..., -5.9866e+00,
-6.4977e+00, -3.2159e+00],
...,
[-3.0100e+00, -4.6702e+00, -5.7409e+00, ..., -2.9754e+00,
-3.3739e+00, -1.4528e+00],
[-2.5362e+00, -2.6987e+00, -7.5106e-01, ..., -4.9754e+00,
-6.1456e+00, -2.3018e+00],
[-3.3498e-02, -1.5445e+00, -3.4536e-01, ..., 5.0411e-01,
5.3592e-01, 7.6825e-04]],
[[-6.2763e-01, 4.8592e-01, 8.8688e-01, ..., 8.2641e-01,
6.2899e-01, 1.9256e+00],
[-9.6526e-01, 1.1764e-02, 3.8123e-01, ..., 4.8948e-01,
3.8467e-03, 1.3741e-01],
[-1.7407e+00, -1.8645e-01, -4.8975e-02, ..., -5.4887e-03,
-4.0015e-02, -1.1866e-01],
...,
[-9.1050e-01, -5.5942e-01, 5.9329e-01, ..., 2.9998e-02,
9.0141e-02, 3.4633e-01],
[-1.6903e+00, -3.0760e+00, -2.6046e+00, ..., 6.6248e-03,
9.5185e-02, 4.0814e-01],
[-1.9100e+00, -3.7462e+00, -4.4372e+00, ..., -7.0855e-02,
9.6717e-02, 6.3471e-02]],
...,
[[ 1.2273e+00, -3.5600e-01, -3.5096e-02, ..., 2.4229e-01,
-9.1815e-02, -2.1157e+00],
[-7.5443e-01, -9.1976e-01, -4.8609e-01, ..., -4.8984e-01,
-1.4914e+00, -2.0990e+00],
[-6.8846e-01, -8.5290e-01, -2.3645e-01, ..., -2.6130e-01,
-5.6965e-01, -1.4111e+00],
...,
[ 5.3234e-02, 4.5720e-01, -7.8807e-02, ..., 6.4378e-02,
-1.3140e-01, -4.9982e-01],
[-1.8357e+00, -8.8187e-01, -9.7410e-01, ..., -9.2048e-01,
-1.0596e+00, -1.0435e+00],
[-1.7582e+00, -2.4990e+00, -2.9452e+00, ..., -2.1513e-01,
-3.3849e-01, -6.0218e-01]],
[[ 5.1953e-02, 2.4829e-02, 4.7559e-01, ..., 4.9368e-01,
9.1876e-01, -2.3190e-02],
[ 2.3442e-01, -7.8801e-01, -7.9082e-02, ..., 4.2830e-01,
3.9212e-01, -8.3901e-01],
[ 1.5262e-01, -5.9928e-01, 3.2133e-01, ..., 6.8549e-01,
5.7029e-01, -3.8535e-01],
...,
[ 1.3489e-02, 4.0367e-01, 8.5099e-01, ..., 7.4262e-02,
1.8431e-02, -1.0588e-01],
[-4.6921e-01, -5.4610e-01, 2.6765e-01, ..., -2.8071e-01,
-3.8380e-01, -6.7256e-01],
[ 4.7159e-02, -1.0644e+00, -1.6628e+00, ..., 2.2404e-01,
1.3967e-01, 3.4316e-01]],
[[-1.5340e+00, -7.8389e-01, -4.0251e-01, ..., -4.3998e-01,
-2.9191e-01, 8.6699e-01],
[-2.1033e+00, -7.7416e-01, -5.4611e-01, ..., -7.9673e-01,
-1.9789e-01, 1.6730e+00],
[-1.4625e+00, -5.8660e-01, -2.3758e-01, ..., -4.4282e-01,
-1.4875e-01, 1.4162e+00],
...,
[-1.2654e+00, -1.5145e+00, -1.5589e+00, ..., 1.8236e-01,
2.0733e-01, 7.1385e-01],
[-2.0402e+00, -3.3854e+00, -3.3729e+00, ..., -3.7273e-02,
7.9379e-02, 4.3923e-01],
[-1.6328e-01, -1.6975e+00, -1.4640e+00, ..., -1.9909e-01,
-2.4760e-01, 5.8485e-01]]],
[[[ 3.4468e-02, -2.1859e-01, -2.9638e-02, ..., -9.3844e-01,
-5.4042e-01, -1.4200e+00],
[ 3.3181e-01, -9.2107e-01, -6.9845e-01, ..., -1.1412e+00,
-1.3503e+00, -1.4667e+00],
[ 2.4157e-01, -5.6623e-01, -4.8117e-01, ..., -8.8419e-01,
-1.1600e+00, -1.8110e+00],
...,
[-1.6395e-02, -6.0584e-01, -3.9659e-01, ..., -8.9673e-01,
-1.0247e+00, -1.9233e+00],
[-5.6793e-01, -6.3782e-01, 1.3340e-04, ..., -4.7094e-01,
-5.5346e-01, -1.7150e+00],
[-7.1078e-01, -1.3418e+00, -1.1994e+00, ..., -1.1236e+00,
-1.5120e+00, -7.7081e-01]],
[[-7.3244e-01, -2.1501e+00, -1.7976e+00, ..., -3.9624e+00,
-5.4794e+00, -1.8650e+00],
[-1.9921e+00, -3.7543e+00, -2.9244e+00, ..., -5.9283e+00,
-8.0758e+00, -3.4582e+00],
[-1.5241e+00, -2.5620e+00, -2.4919e+00, ..., -5.1042e+00,
-6.2124e+00, -2.6172e+00],
...,
[-1.7707e+00, -2.4254e+00, -2.4069e+00, ..., -5.0585e+00,
-6.3512e+00, -2.5027e+00],
[-3.8122e+00, -4.4114e+00, -3.7027e+00, ..., -6.4357e+00,
-7.9203e+00, -2.5439e+00],
[ 3.2369e-01, -2.7561e-01, 3.8633e-01, ..., 4.3521e-01,
4.8268e-01, 2.3227e-01]],
[[-2.3233e-01, 1.3804e-01, 7.2465e-01, ..., 6.3718e-01,
2.4263e-01, 1.5252e+00],
[ 7.0444e-02, -3.7429e-01, 5.7909e-01, ..., 6.1811e-01,
-1.0271e-01, 1.0055e-01],
[-1.2712e-01, -2.0996e-01, 2.3456e-01, ..., 3.4379e-02,
-1.0997e-02, -1.1346e-01],
...,
[-2.5207e-01, -1.8188e-01, 1.0534e-01, ..., 2.1250e-02,
-1.0006e-01, -1.6464e-01],
[-2.5133e-01, -1.3671e-01, 1.4310e-01, ..., -1.0270e-01,
-6.6541e-03, -1.8193e-01],
[-1.6381e-01, -4.9405e-01, 1.1864e-01, ..., 3.9741e-02,
2.4674e-01, -5.3830e-01]],
...,
[[ 5.8442e-01, -2.8694e-01, -2.4254e-02, ..., 4.4694e-01,
3.5967e-01, -1.3223e+00],
[-5.3505e-01, -5.5167e-01, -5.5349e-01, ..., -1.4901e-01,
-7.7445e-01, -1.4333e+00],
[-3.7665e-01, -2.0950e-01, -2.5360e-01, ..., -6.4180e-02,
-8.2898e-02, -1.0124e+00],
...,
[-4.2606e-01, -1.5658e-01, -1.8588e-01, ..., -1.4332e-01,
-9.9330e-02, -9.3037e-01],
[-2.2742e+00, -1.7002e+00, -1.2817e+00, ..., -1.5757e+00,
-1.5864e+00, -1.8257e+00],
[-4.8422e-01, -7.9373e-01, -3.9155e-01, ..., -3.3609e-01,
-5.1834e-01, -1.1639e+00]],
[[ 1.5266e-01, 2.6444e-01, 6.3995e-01, ..., -9.2683e-02,
3.7494e-01, -3.7445e-01],
[ 1.3240e-01, 2.3718e-01, 4.5832e-01, ..., 2.0538e-01,
7.1334e-02, -8.5520e-01],
[-1.2862e-01, 1.0033e-01, 2.8948e-01, ..., 2.9589e-01,
1.5227e-01, -4.8282e-01],
...,
[-1.8622e-01, -1.3330e-03, 3.2146e-01, ..., 2.3433e-01,
1.0535e-01, -5.0764e-01],
[-5.3011e-01, -6.9544e-01, -3.0550e-02, ..., -1.6001e-01,
-3.0095e-01, -9.1686e-01],
[ 5.1447e-01, 3.9327e-01, 5.8424e-02, ..., 3.6280e-01,
3.4788e-01, 3.0866e-01]],
[[-8.5131e-01, 2.5176e-01, 3.3524e-01, ..., -7.5595e-02,
-6.8303e-02, 5.9196e-01],
[-1.1303e+00, 2.2607e-01, -3.4272e-01, ..., -9.4501e-01,
-4.4834e-01, 8.8801e-01],
[-9.5431e-01, 3.6443e-01, -1.1577e-01, ..., -5.8969e-01,
-5.7316e-01, 6.2937e-01],
...,
[-9.5332e-01, 1.8789e-01, -9.0346e-02, ..., -5.5194e-01,
-5.3742e-01, 5.8803e-01],
[-1.6559e+00, -3.5226e-01, -3.4397e-01, ..., -6.7501e-01,
-7.0441e-01, 3.7551e-01],
[-7.3009e-01, -2.4808e-02, -3.0016e-01, ..., -4.4085e-01,
-4.3226e-01, 1.0018e+00]]],
[[[-8.2166e-01, -1.5955e+00, -3.8838e+00, ..., -3.9342e+00,
-3.6088e+00, -1.3422e+00],
[-5.5278e-01, -2.5872e+00, -3.3977e+00, ..., -8.2071e+00,
-4.7306e+00, -1.4850e+00],
[-3.8712e-01, -2.0963e+00, 1.4066e+00, ..., 9.3763e-01,
-4.1291e+00, -4.1447e+00],
...,
[ 5.6303e-01, -3.0369e-01, -6.9341e+00, ..., -1.7989e+00,
-1.1862e+00, -1.2936e+00],
[ 7.8272e-01, -9.4287e-01, -7.4959e+00, ..., -1.8156e+00,
-1.6166e+00, -1.1004e+00],
[ 7.4484e-01, -5.4191e-01, -5.0408e+00, ..., -1.2107e+00,
-2.2363e+00, -7.1774e-01]],
[[-2.9175e+00, -5.3963e+00, -1.6915e+01, ..., -9.8575e+00,
-1.3143e+01, -1.0473e+00],
[-6.0211e+00, -8.9100e+00, -2.2201e+01, ..., -5.0271e+00,
-1.4455e+01, -3.6876e+00],
[-4.7502e+00, -6.6566e+00, -2.0514e+01, ..., -1.3438e+01,
-1.5831e+01, -9.7991e-01],
...,
[-4.6579e+00, -8.9528e+00, -8.9272e+00, ..., -6.4047e+00,
-7.3703e+00, -2.0378e+00],
[-6.8903e+00, -8.9756e+00, -9.6076e+00, ..., -1.0572e+01,
-9.2450e+00, -2.5906e+00],
[ 1.7178e+00, 1.0756e+00, -3.3523e+00, ..., -1.6919e+00,
-5.2030e-01, 1.0206e-01]],
[[-7.5765e-01, 4.7875e-01, -1.9033e+00, ..., -1.4500e+01,
-8.0942e+00, -5.9436e-01],
[-8.3144e-01, 1.7696e-01, 2.5076e+00, ..., -1.2655e+01,
-1.2099e+01, -7.3493e+00],
[-1.7824e+00, -1.7147e-01, 1.2264e+00, ..., -1.5342e+00,
-7.6618e+00, -8.1095e+00],
...,
[-1.1731e+00, 1.8807e+00, -1.3028e+01, ..., -2.9253e+00,
1.0762e+00, 8.5618e-02],
[-7.0080e-01, 2.2063e+00, -1.4259e+01, ..., -3.2748e+00,
1.3329e+00, 1.5867e-01],
[ 1.6463e-01, 1.4229e+00, -1.0842e+01, ..., -2.2096e+00,
1.1045e+00, -4.7100e-02]],
...,
[[ 1.5599e+00, -3.5717e-01, 1.2498e+00, ..., -5.7897e+00,
1.5754e+00, -3.5318e+00],
[-7.5365e-01, -1.1959e+00, -1.2335e+00, ..., -1.0562e+01,
-9.7968e+00, -1.1621e+01],
[-7.1050e-01, -9.9797e-01, -1.7097e+00, ..., -1.6898e+01,
-1.4167e+01, -1.1212e+01],
...,
[-2.4858e-01, 5.5878e-01, -1.5130e+00, ..., -4.6952e+00,
-2.5975e+00, -7.0896e-01],
[-1.6442e+00, -9.5025e-01, -5.6466e+00, ..., -7.9075e+00,
-3.0338e+00, -1.3883e+00],
[ 2.2713e-02, -8.0914e-01, -2.4601e+00, ..., -5.7086e+00,
-1.6079e+00, -7.9382e-01]],
[[ 4.4816e-02, -1.5955e-03, 9.1390e-01, ..., -9.1777e+00,
-2.1467e+00, 4.8010e-01],
[ 2.6858e-01, -9.1406e-01, 2.6655e+00, ..., -8.8733e+00,
-1.8356e+00, 2.1142e-01],
[ 3.1128e-01, -7.7777e-01, 2.9745e+00, ..., 1.4493e+00,
-5.3268e+00, -1.8914e+00],
...,
[ 1.1272e-01, 9.5571e-01, -5.4291e+00, ..., 2.1021e+00,
5.2683e-01, -4.9308e-01],
[ 2.3927e-01, -3.6655e-01, -6.7083e+00, ..., 2.7509e-01,
-5.6493e-01, -8.8208e-01],
[ 1.3662e+00, 2.2187e+00, -3.4363e+00, ..., 2.1596e+00,
5.6841e-02, 1.5928e-01]],
[[-2.0164e+00, -1.1754e+00, -5.6843e+00, ..., -1.5188e+01,
-4.7946e+00, -3.1312e-01],
[-2.7154e+00, -9.1105e-01, -6.0115e+00, ..., -1.6908e+01,
-8.9771e+00, -2.4235e+00],
[-2.1053e+00, -8.6707e-01, -4.1600e+00, ..., -1.8276e+01,
-1.1349e+01, -4.1207e+00],
...,
[-4.4416e-01, -1.6491e+00, 2.0180e-01, ..., 3.9622e-01,
1.2310e+00, 9.2308e-01],
[-1.0697e+00, -1.2045e+00, 5.0805e-01, ..., 1.2931e+00,
1.4627e+00, 5.5073e-01],
[-7.5607e-01, -1.6023e-01, 1.8737e+00, ..., 5.1909e-01,
9.7988e-01, 6.3110e-01]]]], device='cuda:0'),
tensor([[[[-3.5569e-01, -5.0640e-01, -9.8225e-02, ..., -9.3600e-01,
-4.9929e-01, -1.4238e+00],
[-5.0521e-02, -1.4664e+00, -6.4386e-01, ..., -1.1660e+00,
-1.3101e+00, -1.4818e+00],
[-3.1536e-01, -1.4162e+00, -5.9918e-01, ..., -9.0397e-01,
-1.0994e+00, -1.8142e+00],
...,
[-2.7462e-01, -1.2724e+00, -4.5911e-01, ..., -9.5481e-01,
-1.0432e+00, -1.9070e+00],
[-4.8988e-01, -1.0666e+00, -9.5844e-02, ..., -5.4201e-01,
-5.6094e-01, -1.6984e+00],
[-6.2011e-01, -1.5700e+00, -1.2548e+00, ..., -1.1992e+00,
-1.5562e+00, -7.5964e-01]],
[[-7.1633e-01, -1.9775e+00, -9.2843e-01, ..., -3.9747e+00,
-5.4434e+00, -1.8589e+00],
[-2.2309e+00, -3.9578e+00, -2.1106e+00, ..., -5.9335e+00,
-8.1035e+00, -3.4501e+00],
[-2.2959e+00, -3.3383e+00, -1.7711e+00, ..., -5.0148e+00,
-6.2049e+00, -2.5964e+00],
...,
[-1.6740e+00, -2.7175e+00, -1.4968e+00, ..., -5.4135e+00,
-6.2942e+00, -2.5107e+00],
[-3.7526e+00, -4.5432e+00, -2.8844e+00, ..., -6.6211e+00,
-7.9286e+00, -2.5450e+00],
[ 5.5326e-01, -6.6577e-01, 4.1488e-01, ..., 3.8869e-01,
4.6784e-01, 2.2015e-01]],
[[-3.6462e-01, 7.0825e-01, 9.9373e-01, ..., 6.3800e-01,
2.8521e-01, 1.5432e+00],
[-7.6356e-01, 2.1754e-01, 5.4460e-01, ..., 5.6117e-01,
-1.0676e-01, 1.1810e-01],
[-1.3140e+00, 3.3031e-01, 2.0457e-01, ..., 1.8151e-02,
-4.9921e-02, -1.3969e-01],
...,
[-1.4088e+00, 3.1123e-01, 2.0157e-01, ..., 1.8846e-01,
-1.3840e-01, -2.1299e-01],
[-1.3870e+00, 3.2157e-01, 2.5092e-01, ..., -2.0693e-02,
-1.5926e-02, -2.1341e-01],
[-8.8607e-01, 5.3575e-02, -1.3572e-01, ..., 5.4781e-02,
2.2607e-01, -5.3861e-01]],
...,
[[ 4.0251e-01, -1.0806e-01, -9.5131e-02, ..., 4.8380e-01,
3.4810e-01, -1.3246e+00],
[-1.3909e+00, -1.0457e+00, -5.6269e-01, ..., -1.7013e-01,
-7.9030e-01, -1.4001e+00],
[-7.4937e-01, -4.8611e-01, -1.2126e-01, ..., -9.2423e-02,
-1.0424e-01, -9.6122e-01],
...,
[-9.5505e-01, -6.0437e-01, -1.3671e-01, ..., -2.7193e-01,
-8.8628e-02, -9.4054e-01],
[-2.5558e+00, -1.7843e+00, -8.6508e-01, ..., -1.5583e+00,
-1.5695e+00, -1.8300e+00],
[-6.3971e-01, -1.1372e+00, -6.2875e-02, ..., -3.4154e-01,
-5.6336e-01, -1.1951e+00]],
[[ 2.6994e-02, 2.2817e-01, 4.8088e-01, ..., -5.9770e-02,
4.1595e-01, -3.6102e-01],
[-1.4523e-01, -6.8603e-01, -1.1130e-02, ..., 2.3428e-01,
1.0560e-01, -8.3458e-01],
[-6.1740e-01, -1.2025e+00, -2.4226e-01, ..., 3.0047e-01,
1.5945e-01, -4.5630e-01],
...,
[-5.9655e-01, -1.1350e+00, -1.1483e-01, ..., 2.0549e-01,
9.6094e-02, -5.0320e-01],
[-1.0595e+00, -1.9782e+00, -4.4539e-01, ..., -2.0463e-01,
-2.8878e-01, -9.0923e-01],
[ 2.5182e-01, -4.3424e-01, -2.2572e-01, ..., 2.5691e-01,
3.0215e-01, 3.0059e-01]],
[[-3.6100e-01, 1.2198e-01, 5.0560e-01, ..., -7.3043e-02,
-5.8778e-02, 5.9667e-01],
[-3.8157e-01, -6.5822e-02, -7.7027e-02, ..., -9.2905e-01,
-4.1623e-01, 9.1121e-01],
[ 4.1084e-03, -2.6129e-02, 2.1733e-02, ..., -5.8204e-01,
-5.1839e-01, 6.4688e-01],
...,
[ 7.4226e-03, -5.5345e-02, 5.8237e-02, ..., -3.3392e-01,
-5.4191e-01, 6.1403e-01],
[-4.1697e-01, -5.1561e-01, -1.6628e-01, ..., -5.0875e-01,
-6.9213e-01, 4.0092e-01],
[ 6.2841e-03, -5.4359e-01, -1.8616e-01, ..., -4.3525e-01,
-4.5385e-01, 9.8147e-01]]],
[[[-7.2523e-01, -1.4827e+00, -1.4874e+00, ..., -2.8387e+00,
-1.4360e+00, -1.7904e+00],
[-3.7765e-01, -2.4782e+00, -2.3547e+00, ..., -3.5972e+00,
-2.5748e+00, -1.9792e+00],
[-2.0754e-01, -1.9772e+00, -1.8363e+00, ..., -1.2789e+00,
-1.3763e+00, -2.1855e+00],
...,
[-1.4676e-01, -9.0532e-01, -9.2388e-01, ..., -6.5443e-01,
-5.3171e-01, -1.0356e+00],
[-4.1504e-01, -1.2466e+00, -3.4099e+00, ..., 4.1413e-02,
-9.5316e-02, -1.0910e+00],
[-4.2171e-01, -9.9780e-01, 3.7000e-01, ..., -1.5950e+00,
-9.8350e-01, -6.0346e-01]],
[[-2.8256e+00, -5.3033e+00, -4.7266e+00, ..., -8.0801e+00,
-7.6336e+00, -2.4697e+00],
[-5.5679e+00, -8.4168e+00, -7.6524e+00, ..., -1.3132e+01,
-1.2805e+01, -4.4829e+00],
[-4.2548e+00, -6.0964e+00, -5.9535e+00, ..., -7.3317e+00,
-9.1430e+00, -3.3619e+00],
...,
[-2.2627e+00, -6.3057e+00, -1.3531e+01, ..., -5.1541e+00,
-3.7721e+00, -1.6664e+00],
[-5.0084e+00, -9.0490e+00, -8.3479e+00, ..., -8.1834e+00,
-6.3343e+00, -2.4506e+00],
[ 5.9923e-01, -8.6383e-01, 1.7171e+00, ..., -4.5243e-01,
6.1892e-01, 2.2066e-02]],
[[-7.1683e-01, 3.9025e-01, 8.9586e-01, ..., 8.9590e-01,
8.1639e-01, 1.9568e+00],
[-9.2312e-01, -1.4594e-02, 3.4527e-01, ..., 9.3240e-01,
5.5015e-01, 3.2796e-02],
[-1.7214e+00, -2.1283e-01, -1.2554e-01, ..., -3.0386e+00,
2.2947e-01, -1.6978e-01],
...,
[-1.0877e+00, -3.0485e-01, -1.8772e+00, ..., 8.5515e-01,
7.5312e-02, 2.8331e-01],
[-1.1143e+00, -1.2129e-01, -2.8496e-01, ..., 1.5947e+00,
8.8849e-02, 3.3166e-01],
[-6.0704e-01, 5.9409e-01, -7.5504e-01, ..., 4.8381e-01,
1.2523e-01, -3.3339e-03]],
...,
[[ 1.3426e+00, -3.0247e-01, -4.0075e-02, ..., 2.0353e-01,
2.9639e-02, -2.1758e+00],
[-7.2540e-01, -9.5811e-01, -5.8994e-01, ..., -2.3508e+00,
-1.7801e+00, -2.1531e+00],
[-7.2128e-01, -9.2459e-01, -3.5587e-01, ..., -3.8589e+00,
-2.2651e+00, -1.4517e+00],
...,
[-5.6852e-01, -2.8692e-01, 1.8455e+00, ..., -3.1633e-01,
-1.8875e-01, -6.1642e-01],
[-2.3855e+00, -5.5501e-01, 1.0905e+00, ..., -2.2908e+00,
-1.0919e+00, -1.1099e+00],
[-3.7831e-01, -7.3120e-01, -1.6861e+00, ..., -1.4741e+00,
-3.6867e-01, -6.1436e-01]],
[[ 3.6701e-02, 2.0826e-02, 5.8033e-01, ..., -2.1148e+00,
-8.2344e-02, 3.8972e-02],
[ 1.9316e-01, -7.8807e-01, -2.5993e-03, ..., -1.8300e+00,
-1.0644e+00, -8.6867e-01],
[ 2.0864e-01, -6.1841e-01, 2.6894e-01, ..., 6.9866e-01,
3.0440e-01, -4.2256e-01],
...,
[-4.4762e-01, -4.3607e-01, 6.8295e-01, ..., -4.9186e-01,
9.2331e-02, -2.1727e-01],
[-9.0035e-01, -1.1530e+00, -7.5736e-01, ..., -2.5770e-01,
-3.3584e-01, -7.0414e-01],
[ 5.4076e-01, 2.9136e-01, 6.4215e-01, ..., 4.5613e-01,
1.9413e-01, 3.2857e-01]],
[[-1.7047e+00, -9.5239e-01, -4.9765e-01, ..., -4.8128e-01,
-1.7026e-01, 8.7151e-01],
[-2.3052e+00, -8.2088e-01, -5.7602e-01, ..., -1.4124e+00,
3.8107e-01, 1.6817e+00],
[-1.6867e+00, -6.7514e-01, -2.2611e-01, ..., -1.1622e+00,
3.2405e-01, 1.3863e+00],
...,
[-4.4305e-01, 1.4104e-01, -1.6724e+00, ..., 4.9197e-02,
2.3092e-01, 7.7311e-01],
[-9.9222e-01, -1.0115e+00, -4.6113e+00, ..., -4.0084e-02,
1.1489e-01, 4.9562e-01],
[-4.9633e-01, -1.2679e+00, -1.5447e+00, ..., 6.4239e-01,
-1.9091e-01, 6.5671e-01]]],
[[[-3.1648e-01, -9.6492e-01, -7.6398e-01, ..., -5.5008e-01,
-3.2901e-02, -7.7527e-01],
[ 3.3086e-02, -1.8560e+00, -7.8167e-01, ..., -1.0266e+00,
-8.1557e-01, -1.1860e+00],
[ 5.7849e-02, -1.4173e+00, -6.4953e-01, ..., -8.4987e-01,
-8.0065e-01, -1.7429e+00],
...,
[ 1.3494e-01, -5.1595e-01, -2.3842e-02, ..., -5.6462e-01,
-4.5007e-02, -4.4379e-01],
[-3.4802e-01, -4.7903e-01, 7.6240e-02, ..., 1.5192e-01,
2.6712e-01, -1.7945e-01],
[-6.3814e-01, -1.3469e+00, -1.0809e+00, ..., -1.0834e+00,
-1.1639e+00, -4.9701e-01]],
[[-2.0084e+00, -3.9062e+00, -4.6960e+00, ..., -3.9155e+00,
-4.9193e+00, -1.9508e+00],
[-3.8307e+00, -5.9996e+00, -6.0991e+00, ..., -5.5472e+00,
-6.8450e+00, -2.9420e+00],
[-3.5902e+00, -5.4279e+00, -6.2115e+00, ..., -5.4845e+00,
-5.8038e+00, -2.8097e+00],
...,
[-1.6704e+00, -2.1459e+00, -1.8062e+00, ..., -2.2004e+00,
-2.2534e+00, -9.0090e-01],
[-2.8140e+00, -2.9895e+00, -1.9581e+00, ..., -2.2140e+00,
-3.3374e+00, -9.9325e-01],
[ 3.2214e-01, -2.8707e-01, 2.9981e-01, ..., 3.1634e-01,
4.5551e-01, 2.1272e-01]],
[[-4.3732e-01, 4.7580e-01, 9.6329e-01, ..., 8.3862e-01,
5.6354e-01, 1.4651e+00],
[-5.6827e-01, 6.0558e-02, 8.5267e-01, ..., 4.4932e-01,
3.0953e-01, 4.0905e-01],
[-1.0807e+00, 1.7926e-01, 2.6016e-01, ..., 2.0345e-01,
1.8257e-01, 3.2276e-01],
...,
[-6.2333e-01, 1.1498e-01, 3.6571e-01, ..., 2.8696e-01,
2.4723e-01, 3.9323e-01],
[-4.6631e-01, 5.7088e-02, 2.0302e-01, ..., 1.7541e-01,
2.2648e-01, 5.5120e-01],
[-1.4182e-01, -2.4328e-01, 1.5967e-01, ..., 1.0896e-02,
9.3576e-02, 3.3628e-01]],
...,
[[ 1.0789e+00, 3.4427e-01, 1.1907e+00, ..., 4.1344e-01,
2.3351e-01, -1.0232e+00],
[-6.7849e-01, -5.1332e-01, -2.5096e-01, ..., -3.9716e-01,
-1.2264e+00, -1.3498e+00],
[-1.7879e-01, -4.1862e-01, -2.3619e-01, ..., -3.0887e-01,
-5.4554e-01, -1.1405e+00],
...,
[-6.1649e-01, -4.3095e-01, -3.3687e-01, ..., -4.8428e-01,
-4.9961e-01, -5.5968e-01],
[-2.2979e+00, -1.8653e+00, -1.3939e+00, ..., -1.3241e+00,
-1.3572e+00, -1.0903e+00],
[-5.8400e-01, -8.3543e-01, -5.0503e-01, ..., -2.5792e-01,
-3.3969e-01, -3.9213e-01]],
[[ 2.0568e-01, 2.3059e-01, 4.8845e-01, ..., 2.6784e-01,
7.1402e-01, -2.7795e-01],
[ 4.1816e-01, -6.1002e-01, 9.0153e-02, ..., 1.0301e-01,
3.7265e-01, -7.7274e-01],
[ 4.4475e-02, -7.1980e-01, 2.8918e-01, ..., 1.1730e-01,
1.3743e-01, -6.7191e-01],
...,
[-1.7744e-01, -4.1923e-01, 4.6131e-01, ..., -4.9225e-02,
3.1278e-01, 5.1257e-02],
[-5.3041e-01, -9.3174e-01, 1.5262e-01, ..., 1.8735e-02,
-3.5003e-02, -1.6963e-01],
[ 3.0374e-01, -5.4412e-02, -1.6289e-01, ..., -1.3175e-01,
-1.1402e-01, 1.6918e-01]],
[[-6.8623e-01, -6.1091e-02, -4.3955e-01, ..., 3.2861e-01,
4.1057e-01, 9.6063e-01],
[-1.1948e+00, -7.3549e-01, -9.9717e-01, ..., -6.3136e-01,
-2.9048e-01, 9.7106e-01],
[-8.4991e-01, -6.0189e-01, -8.1901e-01, ..., -3.3773e-01,
-6.6409e-02, 7.8560e-01],
...,
[-6.6562e-01, -2.7577e-01, -1.3051e-01, ..., 2.2080e-01,
-1.5723e-02, 6.4160e-01],
[-1.1482e+00, -6.0680e-01, -4.0374e-01, ..., -3.3049e-01,
-2.0302e-01, 1.6969e-01],
[-3.7938e-01, -3.5954e-02, -2.1006e-01, ..., -2.2844e-01,
-2.4510e-01, 3.2077e-01]]],
[[[ 4.3039e-03, -2.7018e-01, -4.7566e-02, ..., -9.5195e-01,
-5.6634e-01, -1.5129e+00],
[ 2.9669e-01, -9.4107e-01, -7.0546e-01, ..., -1.1750e+00,
-1.3293e+00, -1.5182e+00],
[ 2.5279e-01, -5.6025e-01, -5.0829e-01, ..., -9.4126e-01,
-1.0985e+00, -1.8210e+00],
...,
[-1.2834e-02, -6.0595e-01, 4.3725e-01, ..., -1.0793e+00,
-1.0451e+00, -1.9427e+00],
[-6.1422e-01, -6.8392e-01, 1.0856e-01, ..., -5.3438e-01,
-5.9192e-01, -1.7460e+00],
[-7.5068e-01, -1.3419e+00, -1.0889e+00, ..., -1.1739e+00,
-1.5660e+00, -7.7788e-01]],
[[-7.8818e-01, -2.1988e+00, -1.9758e+00, ..., -3.9302e+00,
-5.5084e+00, -1.8644e+00],
[-2.0598e+00, -3.7390e+00, -3.0980e+00, ..., -5.9003e+00,
-8.0744e+00, -3.3946e+00],
[-1.6360e+00, -2.6210e+00, -2.7776e+00, ..., -5.0157e+00,
-6.2041e+00, -2.5331e+00],
...,
[-1.8208e+00, -2.5193e+00, -3.3768e+00, ..., -5.0548e+00,
-6.2928e+00, -2.4864e+00],
[-3.9001e+00, -4.5195e+00, -4.2535e+00, ..., -6.5508e+00,
-8.0425e+00, -2.5789e+00],
[ 2.9550e-01, -3.0519e-01, 1.1788e-01, ..., 4.0916e-01,
4.4992e-01, 2.1656e-01]],
[[-2.0797e-01, 1.6212e-01, 7.3440e-01, ..., 6.4775e-01,
2.2408e-01, 1.4507e+00],
[ 7.2250e-02, -3.6826e-01, 5.5140e-01, ..., 5.8486e-01,
-3.2460e-02, 2.8837e-02],
[-1.3368e-01, -1.9779e-01, 2.2565e-01, ..., 1.5395e-02,
8.2877e-03, -1.5558e-01],
...,
[-2.5234e-01, -2.3834e-01, 1.5212e+00, ..., 1.9760e-02,
-2.0909e-01, -2.8691e-01],
[-2.7157e-01, -1.8033e-01, 1.0985e+00, ..., -9.4236e-02,
-5.5823e-02, -2.6775e-01],
[-1.2858e-01, -4.3565e-01, 4.3109e-01, ..., 9.5543e-02,
2.0889e-01, -5.8560e-01]],
...,
[[ 6.0343e-01, -2.2225e-01, 4.4697e-02, ..., 4.2702e-01,
3.9845e-01, -1.2814e+00],
[-4.9967e-01, -5.1799e-01, -4.9406e-01, ..., -1.8431e-01,
-7.3600e-01, -1.3795e+00],
[-3.5507e-01, -2.1901e-01, -2.2435e-01, ..., -8.3093e-02,
-8.7542e-02, -9.5643e-01],
...,
[-4.3996e-01, -1.7912e-01, -2.5100e-01, ..., -3.5964e-02,
-1.2761e-01, -9.5762e-01],
[-2.3398e+00, -1.7360e+00, -1.7638e+00, ..., -1.5863e+00,
-1.6330e+00, -1.8231e+00],
[-4.8767e-01, -8.2106e-01, -6.9012e-01, ..., -3.3829e-01,
-5.3210e-01, -1.2114e+00]],
[[ 1.6247e-01, 2.6510e-01, 6.3438e-01, ..., -8.0469e-02,
3.5387e-01, -3.7061e-01],
[ 1.5379e-01, 2.3561e-01, 4.6336e-01, ..., 2.0141e-01,
8.0754e-02, -8.3363e-01],
[-1.0801e-01, 8.8203e-02, 3.2507e-01, ..., 3.0195e-01,
1.5704e-01, -4.4203e-01],
...,
[-2.1015e-01, 5.6394e-02, 2.3739e-01, ..., 1.1317e-01,
1.5185e-01, -4.3168e-01],
[-5.8057e-01, -7.5300e-01, 5.0569e-02, ..., -2.0452e-01,
-3.0270e-01, -9.0858e-01],
[ 5.2203e-01, 3.8853e-01, 2.0249e-01, ..., 3.2961e-01,
3.1163e-01, 3.1967e-01]],
[[-8.3039e-01, 2.4013e-01, 2.9516e-01, ..., -1.1615e-01,
-1.0207e-01, 5.5421e-01],
[-1.1197e+00, 2.0607e-01, -3.3881e-01, ..., -9.2937e-01,
-4.4892e-01, 8.8264e-01],
[-9.3269e-01, 3.3059e-01, -8.6930e-02, ..., -5.8184e-01,
-5.6520e-01, 6.1930e-01],
...,
[-9.9128e-01, 1.6003e-01, -1.6478e+00, ..., -5.6115e-01,
-5.8192e-01, 6.8514e-01],
[-1.6716e+00, -4.1192e-01, -1.1084e+00, ..., -7.6366e-01,
-7.4314e-01, 4.3795e-01],
[-7.5105e-01, -1.1220e-01, -5.1543e-01, ..., -4.8815e-01,
-4.6916e-01, 1.0247e+00]]],
[[[-3.4254e-01, -5.1198e-01, -1.1380e-01, ..., -9.4250e-01,
-4.7420e-01, -1.3697e+00],
[-3.8811e-02, -1.4175e+00, -7.0304e-01, ..., -1.1485e+00,
-1.3061e+00, -1.4396e+00],
[-5.0449e-01, -1.4052e+00, -5.7763e-01, ..., -9.0229e-01,
-1.0621e+00, -1.8122e+00],
...,
[-2.9750e-01, -1.2654e+00, -4.6666e-01, ..., -8.9998e-01,
-9.9923e-01, -1.9135e+00],
[-5.3100e-01, -1.0971e+00, -7.7018e-02, ..., -4.6464e-01,
-5.5945e-01, -1.7042e+00],
[-6.2619e-01, -1.6009e+00, -1.2541e+00, ..., -9.5232e-01,
-1.3513e+00, -6.6011e-01]],
[[-6.8323e-01, -1.9866e+00, -8.7975e-01, ..., -3.8471e+00,
-5.4565e+00, -1.7911e+00],
[-2.2867e+00, -4.1231e+00, -2.2055e+00, ..., -5.9488e+00,
-8.2428e+00, -3.4726e+00],
[-3.4464e+00, -4.1782e+00, -2.3869e+00, ..., -4.9782e+00,
-6.1394e+00, -2.5251e+00],
...,
[-1.6293e+00, -2.7102e+00, -1.4902e+00, ..., -5.0151e+00,
-6.3028e+00, -2.5012e+00],
[-3.7557e+00, -4.5957e+00, -2.8913e+00, ..., -6.0959e+00,
-7.5743e+00, -2.4343e+00],
[ 5.3937e-01, -7.1732e-01, 4.0199e-01, ..., 4.2569e-01,
4.6383e-01, 2.4325e-01]],
[[-3.9426e-01, 7.1179e-01, 1.0191e+00, ..., 7.1290e-01,
2.8181e-01, 1.6009e+00],
[-8.1014e-01, 2.2154e-01, 5.1941e-01, ..., 5.7886e-01,
-1.4307e-01, 1.2051e-01],
[-1.5791e+00, 2.6597e-01, 3.0868e-01, ..., 4.4822e-02,
2.0456e-02, -1.2893e-01],
...,
[-1.4185e+00, 2.9579e-01, 1.5104e-01, ..., -1.4801e-02,
-7.2490e-02, -1.5875e-01],
[-1.3896e+00, 3.0482e-01, 2.1925e-01, ..., -8.1871e-02,
-2.1035e-02, -1.9334e-01],
[-9.0372e-01, 6.3405e-02, -1.6083e-01, ..., 8.7281e-02,
1.9630e-01, -6.1170e-01]],
...,
[[ 4.0307e-01, -6.3644e-02, -1.2612e-01, ..., 4.0733e-01,
2.3965e-01, -1.3543e+00],
[-1.3663e+00, -1.0204e+00, -6.3039e-01, ..., -1.6933e-01,
-8.3458e-01, -1.4115e+00],
[-5.2908e-01, -1.4468e-01, 5.1080e-02, ..., -8.9627e-02,
-1.3270e-01, -9.8364e-01],
...,
[-9.5939e-01, -5.9168e-01, -1.3758e-01, ..., -9.4536e-02,
-1.0234e-01, -9.3247e-01],
[-2.5654e+00, -1.7861e+00, -8.4824e-01, ..., -1.3723e+00,
-1.4160e+00, -1.7222e+00],
[-6.3140e-01, -1.1670e+00, -7.6771e-02, ..., -3.4854e-01,
-4.9003e-01, -1.1343e+00]],
[[ 1.8147e-02, 2.2524e-01, 4.9446e-01, ..., -9.2963e-02,
4.3323e-01, -3.6695e-01],
[-1.6629e-01, -7.0931e-01, -4.0350e-02, ..., 1.8688e-01,
6.6822e-02, -8.6048e-01],
[-7.5174e-01, -1.3665e+00, -5.3947e-01, ..., 2.7034e-01,
1.4817e-01, -4.7713e-01],
...,
[-5.9461e-01, -1.1308e+00, -1.0752e-01, ..., 2.4441e-01,
1.7372e-01, -4.7479e-01],
[-1.0781e+00, -1.9716e+00, -4.3053e-01, ..., -4.0391e-02,
-1.5267e-01, -8.3721e-01],
[ 2.4844e-01, -4.4400e-01, -2.1303e-01, ..., 3.9226e-01,
3.6859e-01, 3.1782e-01]],
[[-3.7408e-01, 1.0923e-01, 5.2425e-01, ..., -1.4237e-02,
-2.4163e-02, 6.2873e-01],
[-4.0186e-01, -5.3192e-02, 1.1899e-02, ..., -8.9557e-01,
-3.9794e-01, 9.2626e-01],
[ 1.2164e-01, 1.3658e-01, 2.2407e-01, ..., -5.6699e-01,
-5.0753e-01, 6.3058e-01],
...,
[-4.3127e-04, -8.7027e-02, 9.8768e-02, ..., -5.7102e-01,
-5.1717e-01, 6.1149e-01],
[-4.1138e-01, -5.3773e-01, -1.1122e-01, ..., -6.4848e-01,
-6.5135e-01, 4.2998e-01],
[ 2.0712e-02, -5.3770e-01, -1.7803e-01, ..., -4.2055e-01,
-4.0319e-01, 1.0138e+00]]]], device='cuda:0'),
tensor([[[[-2.0655e-01, -3.8772e-01, 3.8992e-02, ..., -8.9641e-02,
3.8158e-01, 9.0633e-03],
[-6.9403e-02, -1.6119e+00, -9.8803e-01, ..., -8.0042e-01,
-5.7489e-01, -5.2465e-01],
[ 3.8678e-02, -4.4295e-01, -1.5408e+00, ..., -2.3732e+00,
-2.4663e+00, -1.7569e+00],
...,
[-1.5953e-01, -1.9236e+00, -2.8621e+00, ..., -2.1006e+00,
-2.5861e+00, -2.0963e+00],
[-4.5982e-01, -1.4024e+00, -1.1043e+00, ..., -1.0730e+00,
-1.4721e+00, -1.5630e+00],
[-2.5958e-01, -1.1798e+00, -7.7237e-01, ..., -2.9369e+00,
-1.2543e+00, -7.9643e-01]],
[[-4.4165e-01, -1.8022e+00, -1.0879e+00, ..., -1.1090e+00,
-1.7751e+00, -4.9373e-01],
[-2.3041e+00, -4.4003e+00, -2.9825e+00, ..., -2.9542e+00,
-3.9227e+00, -1.4230e+00],
[-2.3853e+00, -9.1801e+00, -9.7343e+00, ..., -1.2142e+01,
-9.2808e+00, -4.1125e+00],
...,
[-3.9525e+00, -7.6078e+00, -8.1915e+00, ..., -1.0647e+01,
-7.4529e+00, -3.0814e+00],
[-5.8593e+00, -7.7860e+00, -8.7536e+00, ..., -1.1698e+01,
-7.3014e+00, -2.9030e+00],
[ 4.4533e-01, -5.2959e-01, 6.8411e-01, ..., -1.3809e+00,
6.5333e-01, 1.2974e-01]],
[[-4.9168e-01, -3.3255e-02, 5.1434e-01, ..., 5.2557e-01,
2.2171e-01, 1.0206e+00],
[-3.6603e-01, -3.5535e-01, 4.9887e-01, ..., 4.7381e-01,
3.4024e-01, 5.0237e-01],
[-9.4163e-01, -7.4026e-01, -8.1556e-01, ..., -2.5856e-01,
1.5741e-01, 3.8066e-01],
...,
[-1.8177e+00, -3.5767e-03, -8.8962e-01, ..., -1.0175e+00,
1.0628e+00, -1.5946e-01],
[-1.5498e+00, 5.1839e-01, -1.2430e+00, ..., 8.7974e-01,
-2.1168e-01, -2.4858e-01],
[-7.7804e-01, 4.1992e-01, -1.5166e-01, ..., 1.4255e+00,
-1.1923e-01, -4.7505e-01]],
...,
[[ 1.6359e-01, -4.4306e-01, -2.6540e-01, ..., 4.2802e-03,
-1.4277e-01, -7.1804e-01],
[-7.9382e-01, -5.9375e-01, -5.0090e-01, ..., -2.7623e-01,
-1.0117e+00, -8.3035e-01],
[-4.7485e-01, -1.5648e-01, 5.0523e-01, ..., -3.5890e-01,
-1.2441e-01, -5.9502e-01],
...,
[-9.9948e-01, -6.0125e-01, -6.0055e-01, ..., -5.9218e+00,
-1.4768e+00, -1.3183e+00],
[-2.6211e+00, -2.1113e+00, -1.5668e+00, ..., -7.9511e+00,
-1.5730e+00, -1.7089e+00],
[-5.6279e-01, -1.4793e+00, -3.1810e-03, ..., -4.3597e+00,
-4.5191e-01, -6.5995e-01]],
[[-1.3113e-01, 4.2933e-02, 3.7022e-01, ..., 5.4850e-02,
5.2749e-01, 1.0140e-01],
[-3.3168e-01, -2.4227e-01, 3.3291e-02, ..., 1.3580e-02,
1.3458e-01, -1.7696e-01],
[-5.4498e-01, 4.7921e-01, 9.9076e-02, ..., -4.6315e-01,
-2.6354e+00, -2.0554e+00],
...,
[-5.5771e-02, -6.7041e-01, 8.9478e-02, ..., -1.3613e-01,
-3.8667e-01, -3.2012e-01],
[-6.5505e-01, -1.9519e+00, 9.0545e-02, ..., -9.4255e-01,
-7.5014e-01, -5.1538e-01],
[ 7.9277e-01, 5.3960e-02, 9.5561e-02, ..., -6.0976e-01,
5.3607e-01, 3.5249e-01]],
[[-5.7511e-01, 4.0131e-01, 5.7573e-01, ..., 4.9403e-01,
4.2583e-01, 7.3120e-01],
[-8.6028e-01, 2.1052e-01, -1.9462e-01, ..., -2.7768e-01,
-1.3382e-01, 7.8618e-01],
[-4.6689e-01, 3.7849e-01, -2.3893e-01, ..., -9.4034e-01,
8.0467e-02, 1.0701e+00],
...,
[-1.4457e+00, -1.7128e+00, -1.7981e+00, ..., 1.6960e+00,
1.2638e+00, 1.3794e+00],
[-1.6016e+00, -1.8339e+00, -6.9234e-01, ..., 1.4153e+00,
7.0240e-01, 1.2320e+00],
[-7.8418e-01, -8.4446e-01, 1.7866e-01, ..., 1.0564e+00,
2.2613e-01, 1.4362e+00]]],
[[[-2.0695e-01, -4.6961e-01, -9.0327e-01, ..., 5.6618e-02,
5.8965e-01, 2.6664e-01],
[-3.1474e-02, -9.5601e-01, -3.4079e-01, ..., -8.1046e-01,
-5.5747e-01, -2.4644e-01],
[ 3.0046e-02, -9.2299e-01, -1.4571e-01, ..., -5.5766e-01,
-4.4606e-01, -6.7953e-01],
...,
[-2.0878e-01, -1.9087e+00, -1.4629e+00, ..., -1.0222e+00,
-1.0947e+00, -1.9375e+00],
[-4.6226e-01, -1.4999e+00, -7.3322e-01, ..., -1.6727e-01,
-1.9422e-01, -1.3655e+00],
[-2.0515e-01, -1.3416e+00, -1.1360e+00, ..., -8.3748e-01,
-1.1212e+00, -7.6450e-01]],
[[-2.5247e-01, -2.6595e+00, -3.2348e+00, ..., -5.8182e-01,
-1.4082e+00, -2.0847e-01],
[-1.8659e+00, -3.2077e+00, -1.4896e+00, ..., -2.4154e+00,
-3.6320e+00, -1.2257e+00],
[-1.7805e+00, -2.9691e+00, -2.3881e+00, ..., -2.6410e+00,
-3.0798e+00, -1.2699e+00],
...,
[-3.7101e+00, -5.2752e+00, -5.2015e+00, ..., -5.4654e+00,
-5.9406e+00, -2.8832e+00],
[-5.4026e+00, -6.5893e+00, -5.6584e+00, ..., -5.4234e+00,
-6.3782e+00, -2.6729e+00],
[ 5.3892e-01, -5.1610e-01, 5.5178e-01, ..., 4.5517e-01,
6.3934e-01, 1.0213e-01]],
[[-4.4892e-01, -6.6051e-01, -1.5690e+00, ..., 1.5440e-01,
2.1460e-02, 9.4584e-01],
[-2.1306e-01, -3.3432e-01, -6.6296e-01, ..., 1.1872e-01,
2.1836e-01, 5.1362e-01],
[-6.4498e-01, 1.9161e-01, 1.8600e-01, ..., 1.1445e-01,
1.8096e-01, 3.7872e-01],
...,
[-1.7727e+00, -4.2699e-01, -4.7207e-02, ..., -3.5949e-02,
-5.5082e-02, -1.6612e-01],
[-1.4483e+00, -3.4733e-01, -1.3173e-01, ..., 6.1119e-02,
-9.5593e-02, -2.7365e-01],
[-5.6953e-01, -5.5450e-02, -1.8921e-02, ..., -2.0028e-01,
-9.6232e-02, -4.9748e-01]],
...,
[[ 2.2011e-01, -4.2576e-01, -6.2680e-01, ..., 2.8572e-02,
-3.4486e-01, -8.1728e-01],
[-6.3971e-01, -2.6798e-01, -9.9084e-02, ..., -1.8286e-01,
-1.1106e+00, -8.4148e-01],
[-3.6895e-01, -5.4359e-01, -8.8772e-01, ..., 6.0655e-02,
-2.3998e-01, -5.5673e-01],
...,
[-9.9590e-01, -8.3979e-01, -3.0480e-01, ..., -3.5103e-01,
-6.3266e-01, -1.2221e+00],
[-2.5958e+00, -1.9906e+00, -1.3464e+00, ..., -1.5343e+00,
-1.5711e+00, -1.5298e+00],
[-5.8969e-01, -1.2280e+00, -2.1242e-01, ..., -2.8421e-01,
-2.8177e-01, -4.5696e-01]],
[[-1.3317e-01, 3.9982e-02, -4.4479e-01, ..., 3.0473e-01,
6.7684e-01, 2.3657e-01],
[-3.0224e-01, -3.2050e-01, -5.0116e-01, ..., -5.9363e-02,
4.3683e-02, -8.2839e-02],
[-5.2666e-01, -5.1349e-01, -1.9998e-01, ..., -3.5154e-02,
-2.1904e-02, -7.1979e-02],
...,
[-1.9361e-01, -8.6566e-01, 3.3556e-01, ..., 6.4112e-01,
5.2278e-01, -3.0717e-01],
[-7.9034e-01, -1.9356e+00, -8.9290e-02, ..., 3.8595e-01,
2.4977e-01, -4.7674e-01],
[ 7.6746e-01, 1.9131e-02, 3.8724e-01, ..., 6.1820e-01,
4.8279e-01, 3.2474e-01]],
[[-6.0612e-01, 3.5926e-01, -1.4737e-01, ..., 2.9149e-01,
3.6194e-01, 6.4229e-01],
[-8.1023e-01, -1.6643e-01, -3.4709e-01, ..., -2.6306e-02,
-1.4985e-01, 6.9917e-01],
[-4.4601e-01, 3.0111e-01, 4.6278e-01, ..., 1.2398e-01,
9.6001e-02, 5.6702e-01],
...,
[-1.2982e+00, -6.7378e-01, -1.5098e-01, ..., -2.2976e-01,
-3.5571e-02, 1.4332e+00],
[-1.4518e+00, -9.1145e-01, -1.0988e-01, ..., -3.0107e-01,
-3.5075e-02, 1.2365e+00],
[-6.2462e-01, -5.0290e-01, 1.6000e-01, ..., -5.9443e-02,
-3.3453e-02, 1.2568e+00]]],
[[[-7.0739e-01, -1.4610e+00, -1.5269e+00, ..., -2.3482e-01,
2.4557e-01, -4.1298e-01],
[-1.0440e-01, -4.4390e+00, -6.0929e+00, ..., -7.0781e-01,
-4.5848e-01, -6.0722e-01],
[-1.9062e-01, -6.4093e+00, -5.5281e+00, ..., -8.2059e-01,
-1.0538e+00, -1.5446e+00],
...,
[-4.4472e-01, -1.9165e+00, -1.9355e+00, ..., -2.6356e-01,
-6.6329e-02, -1.2075e+00],
[-9.4770e-01, -1.7506e+00, -1.1771e+00, ..., 1.6563e-01,
8.7397e-02, -1.0655e+00],
[-7.8591e-01, -1.6765e+00, -1.2556e+00, ..., -8.7982e-01,
-1.1044e+00, -6.6733e-01]],
[[-2.4390e+00, -5.0284e+00, -4.5386e+00, ..., -2.1856e+00,
-3.4888e+00, -1.2745e+00],
[-7.4724e+00, -1.3916e+01, -1.0574e+01, ..., -3.7375e+00,
-5.2576e+00, -2.1203e+00],
[-7.4636e+00, -1.1541e+01, -5.6631e+00, ..., -7.0286e+00,
-7.4784e+00, -3.6638e+00],
...,
[-4.4667e+00, -6.3530e+00, -5.6239e+00, ..., -3.7849e+00,
-4.5034e+00, -1.9804e+00],
[-6.3473e+00, -7.4494e+00, -6.5033e+00, ..., -4.0838e+00,
-5.5987e+00, -2.0431e+00],
[ 1.2280e-02, -8.0671e-01, 3.8211e-01, ..., 4.3050e-01,
6.3222e-01, 1.8378e-01]],
[[-5.8190e-01, 5.9052e-01, 1.0660e+00, ..., 8.1878e-01,
5.7838e-01, 1.3534e+00],
[-1.0876e+00, -2.9743e+00, -2.7523e+00, ..., 4.6569e-01,
3.7942e-01, 2.2613e-01],
[-1.9509e+00, -3.3066e+00, -5.2370e+00, ..., 1.0650e+00,
1.1359e+00, 6.6487e-01],
...,
[-1.7324e+00, -3.9366e-02, -3.5532e-02, ..., -1.7825e-02,
-1.3434e-01, 8.3354e-02],
[-1.5364e+00, 4.6612e-02, 8.5998e-02, ..., 5.4210e-02,
8.9320e-02, 5.6509e-03],
[-1.4261e+00, -5.8927e-01, -4.0142e-01, ..., 1.5563e-02,
1.8806e-01, -1.5090e-01]],
...,
[[ 1.0859e+00, -4.1337e-01, -2.5442e-01, ..., 3.7011e-01,
2.2932e-01, -7.9112e-01],
[-9.0838e-01, -7.6142e-01, -9.8891e-01, ..., -3.5363e-01,
-1.0505e+00, -9.6024e-01],
[ 2.3096e-01, 4.7157e-01, -2.7102e+00, ..., -5.5605e-01,
-7.3455e-01, -6.8646e-01],
...,
[-6.9240e-01, -1.1290e+00, -8.1187e-01, ..., -7.0820e-01,
-1.1271e+00, -1.4200e+00],
[-2.7671e+00, -2.7816e+00, -1.8730e+00, ..., -1.4852e+00,
-1.3522e+00, -1.2036e+00],
[-6.6899e-01, -1.6521e+00, -1.0333e+00, ..., -1.6272e-01,
-2.4967e-01, -4.8744e-01]],
[[ 4.1198e-02, 2.8846e-02, 6.3837e-01, ..., 3.1650e-01,
8.1375e-01, 5.5222e-02],
[ 3.5200e-01, -5.6445e-01, -2.6146e+00, ..., 3.5076e-01,
5.6387e-01, -3.0708e-01],
[ 6.0997e-01, -3.2965e+00, -5.0954e+00, ..., 7.5660e-01,
1.2132e-01, -8.2511e-01],
...,
[ 5.6495e-02, -8.2114e-01, 2.1506e-01, ..., -3.3725e-02,
6.7301e-01, 1.0672e-01],
[-1.5108e-01, -1.3533e+00, 1.4368e-02, ..., -7.3183e-02,
-4.4089e-02, -6.2246e-01],
[ 6.9000e-01, 2.4108e-01, 3.0626e-01, ..., 2.6203e-01,
3.3410e-01, 3.5749e-01]],
[[-1.5986e+00, -6.0340e-01, -1.3775e-02, ..., 3.2808e-01,
2.7676e-01, 8.8606e-01],
[-2.1553e+00, -1.4466e+00, -1.4757e+00, ..., -4.7276e-01,
-1.5263e-01, 1.1863e+00],
[-2.1934e+00, -4.4566e+00, -4.0435e+00, ..., -1.5071e+00,
-9.2228e-01, 6.5565e-01],
...,
[-1.7841e+00, -3.9877e-01, 3.3136e-01, ..., 4.9320e-01,
9.2886e-02, 1.5240e+00],
[-2.3010e+00, -1.0899e+00, -3.9411e-02, ..., 6.8924e-02,
1.8590e-01, 8.1986e-01],
[-1.1395e+00, -2.3712e-01, 3.9164e-01, ..., -2.2657e-01,
-2.4198e-01, 7.1202e-01]]],
[[[-4.9968e+00, -4.9511e+00, -4.5024e+00, ..., -5.7779e-01,
-2.9447e-01, -9.0785e-01],
[-2.1091e+00, -1.9702e+00, -4.9104e+00, ..., -9.9696e-01,
-1.1053e+00, -1.3107e+00],
[-4.0501e+00, -2.2707e+00, 9.3713e-01, ..., -6.1793e-01,
-9.0866e-01, -1.5276e+00],
...,
[-3.1220e-01, -1.8399e+00, -2.1507e+00, ..., -7.2058e+00,
-6.5358e+00, 1.9364e+00],
[-6.5051e-01, -1.1030e+00, -1.4255e+00, ..., -2.9324e+00,
-6.3419e+00, -5.1916e+00],
[-4.9804e-01, -1.2682e+00, -1.0876e+00, ..., -5.8454e+00,
-9.5695e+00, -8.5945e+00]],
[[-1.1001e+01, -1.6126e+01, -1.2779e+01, ..., -3.3605e+00,
-4.1680e+00, -1.5984e+00],
[-1.0460e+01, -1.4487e+01, -1.2429e+01, ..., -5.3498e+00,
-7.4243e+00, -3.1827e+00],
[-7.9202e+00, -7.0425e+00, -6.4828e+00, ..., -7.0516e+00,
-7.3516e+00, -3.0707e+00],
...,
[-4.5116e+00, -7.1503e+00, -8.2953e+00, ..., -1.0351e+01,
-8.4269e+00, 2.5534e-01],
[-6.4704e+00, -8.0385e+00, -6.9123e+00, ..., 1.8665e+00,
4.9855e+00, 9.6671e+00],
[ 6.0774e-02, -8.2172e-01, 5.1726e-01, ..., 5.7345e+00,
4.5915e+00, 8.7125e+00]],
[[ 1.0346e+00, 1.9095e+00, 2.0130e+00, ..., 1.5986e-01,
5.8835e-01, 1.7073e+00],
[-1.0910e+01, -5.3213e+00, 1.0984e-01, ..., -9.8140e-01,
-5.8910e-01, 2.7523e-01],
[-2.7688e+01, -2.2831e+01, -2.1086e+01, ..., -1.2239e+00,
-9.7225e-01, -2.5746e-01],
...,
[-1.5198e+00, -6.1207e-01, -2.3687e-01, ..., -3.4485e+01,
-2.9086e+01, -7.8541e+00],
[-1.3630e+00, -6.4783e-02, -5.2306e-02, ..., -2.3927e+01,
-2.4335e+01, -7.6306e+00],
[-1.1225e+00, -1.6535e-01, -9.0898e-01, ..., -1.5837e+01,
-1.6885e+01, -5.6316e+00]],
...,
[[-1.1745e-01, -4.1607e-01, -1.0170e+00, ..., 1.6648e-01,
8.3526e-02, -8.7017e-01],
[ 2.0121e+00, -3.0614e+00, -3.4907e+00, ..., -5.6261e-01,
-1.2785e+00, -1.3302e+00],
[-6.3338e+00, -1.4935e+01, -1.7409e+01, ..., -6.0418e-01,
-3.7156e-01, -8.9174e-01],
...,
[-5.1737e-01, -6.7918e-01, -8.8464e-01, ..., -1.7307e+01,
-2.1835e+01, -2.2615e+01],
[-2.6431e+00, -2.3958e+00, -1.7421e+00, ..., -9.4031e+00,
-1.2314e+01, -1.3640e+01],
[-6.5182e-01, -1.3439e+00, -6.7881e-01, ..., 1.4952e+00,
-3.1635e+00, -5.2650e+00]],
[[ 5.9653e-01, 4.7975e-01, -1.1686e+00, ..., 8.6780e-01,
7.6951e-01, 1.5480e-01],
[ 1.5915e+00, 1.9917e+00, 1.9393e+00, ..., 2.7395e-01,
2.1591e-01, -9.1674e-01],
[-9.4496e+00, -9.7215e+00, -4.6689e+00, ..., -5.7074e-01,
-9.1357e-01, -1.2130e+00],
...,
[-2.3829e-01, -5.4601e-01, 3.1036e-01, ..., -1.3373e+01,
-1.7557e+01, -4.3527e+00],
[-6.0688e-01, -1.3050e+00, 1.1372e-01, ..., -2.0713e+00,
-7.3994e+00, 1.6282e+00],
[ 7.9440e-01, 4.2832e-01, 7.3707e-03, ..., -4.0022e+00,
-4.5863e+00, 2.4948e+00]],
[[-3.1117e+00, -2.5435e+00, -9.4861e-01, ..., -4.4001e-02,
-7.3677e-02, 1.3574e+00],
[-1.1463e+01, -1.4754e+01, -9.0975e+00, ..., -1.2396e+00,
-8.1200e-01, 1.3281e+00],
[-1.5239e+01, -1.8341e+01, -1.2518e+01, ..., -4.8957e-01,
-1.6944e-02, 1.2367e+00],
...,
[-2.2405e+00, -1.6510e+00, -1.6392e+00, ..., -1.4194e+01,
-1.2129e+01, -1.4446e+01],
[-2.2087e+00, -1.3448e+00, -3.5932e-01, ..., -1.3212e+01,
-1.6601e+01, -1.9096e+01],
[-1.0624e+00, -5.5681e-01, 8.3584e-01, ..., -1.5502e+00,
-5.9553e+00, -1.0038e+01]]],
[[[ 4.8965e-02, -2.3355e-01, 2.1965e-02, ..., -1.1586e+00,
-6.3197e-01, -9.7072e-01],
[ 3.0001e-01, -1.0925e+00, -7.7341e-01, ..., -1.5676e+00,
-1.2702e+00, -1.2393e+00],
[-4.0307e-02, -1.1304e+00, -5.4798e-01, ..., -1.3363e+00,
-1.2125e+00, -1.8584e+00],
...,
[ 2.6183e-02, -5.4856e-01, -4.3488e-01, ..., -1.1746e+00,
-9.6855e-01, -1.7894e+00],
[-4.9936e-01, -5.9707e-01, -3.1785e-02, ..., -7.5875e-01,
-5.7122e-01, -1.4233e+00],
[-6.0493e-01, -1.2634e+00, -1.0206e+00, ..., -1.2817e+00,
-1.4624e+00, -8.9165e-01]],
[[-7.4174e-01, -2.1408e+00, -1.7065e+00, ..., -4.0655e+00,
-4.7882e+00, -1.7909e+00],
[-2.3141e+00, -4.2098e+00, -3.0660e+00, ..., -6.1533e+00,
-6.8572e+00, -2.7506e+00],
[-2.3894e+00, -3.2304e+00, -2.9205e+00, ..., -5.8575e+00,
-6.3012e+00, -2.8542e+00],
...,
[-1.8405e+00, -2.5352e+00, -2.5643e+00, ..., -5.3246e+00,
-5.5498e+00, -2.5859e+00],
[-3.5839e+00, -4.1386e+00, -3.4004e+00, ..., -6.1852e+00,
-7.1324e+00, -2.7644e+00],
[ 2.8575e-01, -3.1832e-01, 3.9535e-01, ..., 5.3714e-01,
6.5971e-01, 6.9456e-02]],
[[-2.7255e-01, 4.7625e-02, 7.8402e-01, ..., 9.9195e-01,
7.1033e-01, 1.5146e+00],
[ 1.2285e-01, -4.3725e-01, 6.2074e-01, ..., 3.3790e-01,
3.4354e-01, 2.6515e-01],
[-2.3954e-01, -1.6552e-01, 3.5370e-01, ..., -6.8182e-02,
7.2337e-02, -2.1891e-02],
...,
[-2.4904e-01, -2.7314e-01, 1.8012e-01, ..., 2.1182e-01,
1.5961e-02, -2.1651e-01],
[-2.6083e-01, -2.0952e-01, 6.6861e-02, ..., -8.8658e-02,
1.1019e-01, 1.7009e-01],
[-8.2644e-02, -3.9934e-01, 4.4053e-02, ..., -3.7438e-01,
-9.8374e-02, -2.6677e-01]],
...,
[[ 4.6696e-01, -4.3003e-01, -7.6074e-02, ..., -1.2408e-01,
-2.8752e-01, -1.4635e+00],
[-7.1536e-01, -6.5219e-01, -6.5041e-01, ..., -8.0618e-01,
-1.5493e+00, -1.6932e+00],
[-6.8886e-01, -3.5945e-01, -1.9329e-01, ..., -7.1602e-01,
-8.7401e-01, -1.2470e+00],
...,
[-4.4142e-01, -6.7329e-02, -2.6068e-01, ..., -1.0910e+00,
-1.2782e+00, -1.5175e+00],
[-2.1923e+00, -1.5395e+00, -1.1802e+00, ..., -1.8665e+00,
-1.8980e+00, -1.7319e+00],
[-4.8843e-01, -7.5161e-01, -4.6047e-01, ..., -7.3980e-01,
-7.6361e-01, -8.7244e-01]],
[[ 1.0035e-01, 2.6772e-01, 7.0143e-01, ..., 4.1204e-01,
7.1841e-01, -3.6031e-03],
[ 2.2140e-02, 2.6130e-01, 3.7799e-01, ..., 2.9588e-01,
3.7783e-01, -2.9366e-01],
[-2.2649e-01, -2.3004e-01, 8.7734e-02, ..., 3.5122e-01,
2.6198e-01, -6.1433e-01],
...,
[-1.8400e-01, 3.0670e-02, 3.6340e-01, ..., 4.2498e-01,
1.4721e-01, -8.2461e-01],
[-4.5603e-01, -6.5400e-01, 9.5147e-02, ..., -5.8302e-02,
-1.1425e-01, -5.6992e-01],
[ 5.0583e-01, 3.8470e-01, 1.2799e-01, ..., 1.9491e-01,
1.5278e-01, 3.2406e-01]],
[[-9.7408e-01, 2.8960e-01, 4.0527e-01, ..., 1.4339e-01,
2.1954e-01, 8.3993e-01],
[-1.2260e+00, 2.8650e-01, -3.8512e-01, ..., -1.0329e-03,
1.4885e-01, 1.3577e+00],
[-1.3949e+00, 1.6107e-01, -1.9169e-01, ..., 2.2941e-01,
3.2566e-01, 1.2012e+00],
...,
[-9.6527e-01, 2.0234e-01, -1.2144e-01, ..., 2.1130e-01,
6.6852e-01, 1.5266e+00],
[-1.6135e+00, -3.6221e-01, -3.0841e-01, ..., -1.4863e-01,
1.4611e-01, 8.2056e-01],
[-6.5855e-01, -3.1124e-02, -2.5194e-01, ..., 2.4894e-01,
1.3492e-01, 1.0745e+00]]]], device='cuda:0'),
tensor([[[[-3.5186e-01, -9.5632e-01, -6.0603e-01, ..., -3.3946e-01,
-3.4359e-02, -7.7747e-01],
[ 6.7155e-03, -1.8002e+00, -1.2494e+00, ..., -1.0880e+00,
-8.4636e-01, -1.1796e+00],
[ 4.0930e-02, -1.4133e+00, -9.8518e-01, ..., -8.9175e-01,
-8.1387e-01, -1.7480e+00],
...,
[-8.0804e-02, -7.4461e-01, -2.7884e-01, ..., -1.6287e-01,
-5.8525e-02, -4.5087e-01],
[-7.6204e-01, -7.4832e-01, -1.8572e-01, ..., 1.0605e-01,
2.7328e-01, -1.6759e-01],
[-8.9579e-01, -1.5369e+00, -1.4251e+00, ..., -1.0363e+00,
-1.1261e+00, -4.6547e-01]],
[[-1.9832e+00, -3.9315e+00, -3.8732e+00, ..., -4.1121e+00,
-4.8831e+00, -1.9313e+00],
[-3.7221e+00, -5.9559e+00, -5.4468e+00, ..., -5.8271e+00,
-6.8269e+00, -2.9201e+00],
[-3.4599e+00, -5.2509e+00, -5.0529e+00, ..., -5.2697e+00,
-5.7605e+00, -2.7619e+00],
...,
[-1.8510e+00, -2.4650e+00, -2.1519e+00, ..., -1.9156e+00,
-2.2859e+00, -9.1002e-01],
[-3.1714e+00, -3.5896e+00, -2.3811e+00, ..., -2.0769e+00,
-3.1862e+00, -9.1887e-01],
[ 2.1568e-01, -5.2885e-01, 2.2965e-01, ..., 2.9051e-01,
4.2541e-01, 2.1782e-01]],
[[-4.6642e-01, 4.6531e-01, 7.5991e-01, ..., 9.4701e-01,
5.7621e-01, 1.4605e+00],
[-6.7286e-01, 1.9734e-02, 5.1016e-01, ..., 5.4822e-01,
2.6715e-01, 3.5979e-01],
[-1.1430e+00, 1.3501e-01, 1.0063e-01, ..., 2.4731e-01,
1.5127e-01, 3.0800e-01],
...,
[-7.8248e-01, -2.8568e-02, 2.5785e-01, ..., 1.3150e-01,
2.4907e-01, 4.0693e-01],
[-5.6122e-01, -7.4507e-02, 2.2676e-01, ..., 1.2035e-01,
2.0025e-01, 5.2615e-01],
[-2.1733e-01, -3.4018e-01, 4.7116e-02, ..., -3.2374e-02,
1.0276e-01, 3.1151e-01]],
...,
[[ 1.0771e+00, 3.7133e-01, 4.0035e-01, ..., 9.8901e-02,
2.1618e-01, -1.0251e+00],
[-7.2603e-01, -5.4131e-01, -3.5248e-01, ..., -8.6446e-01,
-1.2264e+00, -1.3193e+00],
[-2.4548e-01, -4.0487e-01, -2.8237e-01, ..., -2.8517e-01,
-5.2920e-01, -1.1455e+00],
...,
[-8.0404e-01, -4.6869e-01, -2.7382e-01, ..., -3.1822e-01,
-4.7945e-01, -5.5143e-01],
[-2.6482e+00, -2.1056e+00, -1.3499e+00, ..., -1.2665e+00,
-1.3160e+00, -1.0457e+00],
[-6.8543e-01, -1.0666e+00, -5.7814e-01, ..., -2.8056e-01,
-3.3381e-01, -3.7512e-01]],
[[ 1.6696e-01, 2.0769e-01, 3.2897e-01, ..., 4.6436e-01,
7.3327e-01, -2.5517e-01],
[ 3.4768e-01, -6.1973e-01, 2.4204e-02, ..., 2.0985e-01,
3.6939e-01, -7.8403e-01],
[ 4.4682e-03, -7.2219e-01, 1.5067e-01, ..., 1.6676e-01,
1.2792e-01, -6.9877e-01],
...,
[-4.3426e-01, -5.5020e-01, 3.0961e-01, ..., 2.4499e-01,
2.9431e-01, 7.6042e-02],
[-7.7314e-01, -1.0716e+00, -9.1996e-02, ..., 4.4852e-02,
-1.8070e-02, -1.0237e-01],
[ 3.1119e-01, -4.0808e-02, -2.9745e-01, ..., -1.5066e-01,
-1.0299e-01, 1.9499e-01]],
[[-6.6111e-01, -3.9302e-02, 2.8592e-01, ..., 3.8841e-01,
4.3580e-01, 9.9915e-01],
[-1.1468e+00, -6.7860e-01, -7.0498e-01, ..., -4.5860e-01,
-2.6480e-01, 9.8850e-01],
[-8.6073e-01, -6.6417e-01, -4.5031e-01, ..., -2.6013e-01,
-9.4447e-02, 7.4207e-01],
...,
[-7.1175e-01, -2.3659e-01, -1.9867e-01, ..., -7.7942e-02,
3.2073e-02, 6.2588e-01],
[-1.3589e+00, -7.6785e-01, -6.2486e-01, ..., -3.1921e-01,
-2.1832e-01, 1.6400e-01],
[-4.7410e-01, -8.1551e-02, -2.2093e-01, ..., -2.0082e-01,
-2.1246e-01, 3.3886e-01]]],
[[[-7.6304e-01, -1.4900e+00, -1.5116e+00, ..., -3.3880e+00,
-4.8675e-01, 1.5819e+00],
[-3.3803e-01, -2.2984e+00, -4.6974e+00, ..., -5.0563e+00,
-1.8481e+00, 1.2123e+00],
[-4.4939e-01, -1.3117e+00, -8.3009e+00, ..., -1.7281e+00,
-9.6813e-01, 6.5499e-02],
...,
[-5.6625e-01, -2.0498e+00, -4.0880e-01, ..., -4.5151e-01,
-2.0149e-01, -2.4373e-01],
[-1.1022e+00, -1.8599e+00, -1.3045e+00, ..., -3.9139e-02,
2.3212e-01, -7.5814e-02],
[-8.3928e-01, -1.7379e+00, -1.3210e+00, ..., -1.0469e+00,
-1.1355e+00, -4.0230e-01]],
[[-2.4568e+00, -4.9901e+00, -4.2686e+00, ..., -5.9750e+00,
-7.7613e+00, -1.6889e+00],
[-5.6309e+00, -1.3900e+01, -1.2363e+01, ..., -1.0813e+01,
-1.0576e+01, -1.8594e+00],
[-4.5279e+00, -1.3134e+01, -1.0456e+01, ..., -1.2571e+01,
-9.5738e+00, -5.1561e-02],
...,
[-4.6224e+00, -6.4759e+00, -6.3498e+00, ..., -1.7105e+00,
-1.8737e+00, -5.4425e-01],
[-6.6298e+00, -7.7265e+00, -6.6403e+00, ..., -2.8541e+00,
-3.6581e+00, -1.1270e+00],
[-1.1087e-01, -8.6576e-01, 2.9302e-01, ..., 3.9674e-01,
3.9466e-01, 7.5905e-02]],
[[-6.5599e-01, 5.4101e-01, 1.1213e+00, ..., -2.2129e+00,
-2.5886e+00, 5.1748e-01],
[-8.7810e-01, -1.2932e+00, -1.6037e+00, ..., -6.1961e+00,
-8.7489e+00, -4.1134e+00],
[-1.6536e+00, -2.2903e+00, -6.1209e+00, ..., -1.3465e+01,
-1.1864e+01, -4.3084e+00],
...,
[-1.7585e+00, -9.2724e-02, 2.4394e+00, ..., 1.2600e-01,
1.3516e-01, 4.6209e-01],
[-1.5329e+00, 7.1266e-02, 2.2170e-02, ..., 2.2176e-01,
2.0006e-01, 6.5060e-01],
[-1.5162e+00, -5.5480e-01, -3.1356e-01, ..., -1.2117e-01,
8.3769e-02, 3.7892e-01]],
...,
[[ 1.1273e+00, -4.6694e-01, -2.6038e-01, ..., -1.6843e-02,
-3.5671e+00, -5.4443e+00],
[-7.8823e-01, -8.3467e-01, -3.7517e-01, ..., -4.3216e+00,
-9.8032e+00, -9.9756e+00],
[-6.5231e-01, 1.1589e+00, 1.1237e+00, ..., -5.6761e+00,
-9.4484e+00, -9.2868e+00],
...,
[-7.1207e-01, -1.1987e+00, -3.3346e+00, ..., -1.6488e-01,
-2.6308e-01, -4.3540e-01],
[-2.8345e+00, -3.0358e+00, -1.9468e+00, ..., -8.4287e-01,
-9.3656e-01, -8.4357e-01],
[-6.6336e-01, -1.7613e+00, -1.0659e+00, ..., -9.1323e-02,
-1.8454e-01, -3.7223e-01]],
[[-1.6706e-02, -7.5292e-02, 6.0361e-01, ..., -2.6823e+00,
-1.7328e+00, -9.7703e-01],
[-4.3526e-02, -2.9918e-01, -2.8180e-01, ..., -6.5827e+00,
-6.0416e+00, -2.3945e+00],
[-7.6207e-02, -5.2753e-01, -9.4529e-01, ..., -5.5081e+00,
-5.8985e+00, -4.0716e-01],
...,
[ 1.0106e-01, -8.4659e-01, 1.1665e+00, ..., -1.4711e-01,
-1.7302e-01, 1.0605e-01],
[-1.4705e-01, -1.4043e+00, 6.6301e-02, ..., -4.1445e-01,
-3.7904e-01, -1.3247e-01],
[ 6.9218e-01, 1.6716e-01, 2.8801e-01, ..., -2.0569e-01,
-2.2984e-01, 2.7847e-01]],
[[-1.7838e+00, -7.3582e-01, -5.6358e-02, ..., -1.7037e+00,
-3.2170e+00, -2.9684e+00],
[-2.3408e+00, -3.2379e-01, -4.7903e-01, ..., -4.0664e+00,
-7.3264e+00, -6.5391e+00],
[-1.9840e+00, -1.9731e+00, -2.8347e+00, ..., -3.6332e+00,
-8.4381e+00, -8.9987e+00],
...,
[-1.9668e+00, -5.5994e-01, 1.8149e-01, ..., 7.3640e-02,
-4.9233e-02, 1.8525e-01],
[-2.4300e+00, -1.2428e+00, -2.4148e-01, ..., -7.9374e-02,
-2.1149e-01, -1.1773e-01],
[-1.1788e+00, -1.7715e-01, 4.4807e-01, ..., -1.6195e-01,
-2.6755e-01, 1.8522e-01]]],
[[[-7.9003e-01, -1.4303e+00, -1.4237e+00, ..., -2.5391e-01,
4.3219e-01, 3.6698e-01],
[-3.5635e-01, -2.3341e+00, -2.2178e+00, ..., -8.8562e-01,
1.9983e-01, 3.2521e-01],
[-3.9209e-01, -1.9947e+00, -1.8095e+00, ..., 2.0967e+00,
1.8205e+00, -6.8522e-02],
...,
[ 6.8139e-01, -1.8131e+00, -2.5928e+00, ..., -3.3739e-01,
-1.3728e-01, -1.5984e-01],
[-9.5225e-01, -2.1093e+00, -1.5293e+00, ..., 6.7641e-03,
2.3774e-01, -3.8149e-02],
[-1.1359e+00, -2.0156e+00, -1.4150e+00, ..., -1.0088e+00,
-1.0761e+00, -3.8585e-01]],
[[-2.5880e+00, -5.0202e+00, -4.4152e+00, ..., -5.0180e+00,
-3.3935e+00, -1.0966e+00],
[-5.3321e+00, -7.9475e+00, -6.7211e+00, ..., -3.1986e+00,
-3.5870e+00, -1.4645e+00],
[-4.4482e+00, -6.2624e+00, -6.2882e+00, ..., -3.5562e+00,
-4.7693e+00, -1.3228e+00],
...,
[-5.0484e+00, -7.6492e+00, -6.8550e+00, ..., -1.4451e+00,
-1.6677e+00, -4.2807e-01],
[-7.0537e+00, -7.9730e+00, -6.6215e+00, ..., -2.5050e+00,
-3.3764e+00, -1.0070e+00],
[-9.5837e-02, -1.1158e+00, 3.9373e-01, ..., 3.7131e-01,
3.7135e-01, 8.1856e-02]],
[[-6.1886e-01, 6.4330e-01, 1.0471e+00, ..., 1.1144e-01,
1.2855e+00, 1.1321e+00],
[-9.3277e-01, 3.1852e-02, 3.5282e-01, ..., -4.9359e+00,
-4.4440e-01, 5.5445e-01],
[-1.6608e+00, -1.0375e-01, -2.7717e-02, ..., -4.9208e+00,
-7.0458e-01, 9.9706e-01],
...,
[-2.3965e+00, -1.2924e+00, 2.7496e-01, ..., 9.8396e-02,
9.1114e-02, 4.4842e-01],
[-2.1972e+00, -8.9427e-01, -1.7584e-01, ..., 1.7432e-01,
1.3410e-01, 6.3401e-01],
[-1.4605e+00, -6.0878e-01, -2.5819e-01, ..., -8.1652e-02,
6.4949e-02, 3.5730e-01]],
...,
[[ 1.2339e+00, -3.6357e-01, -2.0946e-01, ..., -1.4151e+00,
-1.1487e+00, -7.0843e-01],
[-6.4972e-01, -1.0046e+00, -8.2435e-01, ..., -7.8080e+00,
-5.4092e+00, -1.7780e+00],
[-6.2578e-01, -1.0507e+00, -7.2917e-01, ..., -1.4133e+01,
-7.7751e+00, -1.1746e+00],
...,
[-2.0702e+00, -3.2755e+00, -2.1401e+00, ..., -1.3787e-01,
-2.2035e-01, -3.9449e-01],
[-4.0790e+00, -3.2863e+00, -1.9507e+00, ..., -8.1644e-01,
-8.7929e-01, -8.0603e-01],
[-7.2287e-01, -1.8339e+00, -1.0784e+00, ..., -2.2496e-02,
-1.3797e-01, -3.1396e-01]],
[[ 8.7528e-02, 4.0634e-02, 5.9488e-01, ..., -7.0981e-02,
-6.5495e-01, -7.2298e-01],
[ 1.0829e-01, -7.4638e-01, 9.0497e-02, ..., -5.3907e-02,
7.4256e-01, -5.8217e-01],
[ 4.4169e-02, -7.1010e-01, 4.8250e-01, ..., -2.0954e+00,
5.7838e-01, -4.5580e-01],
...,
[ 4.2827e-01, 2.3393e-01, 1.7520e-01, ..., -1.0212e-01,
-1.3103e-01, 1.1698e-01],
[ 2.9051e-02, -1.7235e+00, -3.4134e-01, ..., -4.0207e-01,
-3.2304e-01, -6.2267e-02],
[ 5.8124e-01, 9.0785e-02, 2.6247e-01, ..., -2.1064e-01,
-2.1599e-01, 3.0684e-01]],
[[-1.6210e+00, -6.8150e-01, -5.2694e-02, ..., -8.1810e-01,
2.2561e+00, 9.5762e-01],
[-2.1915e+00, -3.9062e-01, -2.7061e-02, ..., -6.6889e+00,
1.6758e+00, 4.7829e-01],
[-1.8490e+00, -4.1799e-01, 1.3769e-01, ..., -4.1656e+00,
1.7388e+00, 4.9558e-01],
...,
[-2.4167e+00, 1.9132e+00, 1.6681e+00, ..., 4.0925e-02,
-5.3230e-02, 1.0322e-01],
[-2.4755e+00, -7.3392e-01, 1.3109e-01, ..., -1.1352e-01,
-2.3416e-01, -1.9959e-01],
[-1.1820e+00, -3.9845e-01, 3.5203e-01, ..., -1.0761e-01,
-2.2968e-01, 1.3051e-01]]],
[[[ 3.2610e-02, -2.4505e-01, -2.2422e-02, ..., -9.5516e-01,
-5.2674e-01, -1.4400e+00],
[ 2.8525e-01, -1.1278e+00, -7.7488e-01, ..., -1.1535e+00,
-1.3317e+00, -1.4997e+00],
[ 1.0296e-01, -1.1360e+00, -1.0667e+00, ..., -8.9733e-01,
-1.1052e+00, -1.8395e+00],
...,
[ 2.0279e-01, -1.9403e-01, -8.9263e-02, ..., -2.9675e+00,
-7.5515e-01, -1.9431e+00],
[-4.2824e-01, -5.9214e-01, -8.7277e-02, ..., -5.5809e-01,
-8.7763e-01, -2.3617e+00],
[-5.5115e-01, -1.1968e+00, -1.0167e+00, ..., -1.1021e+00,
-1.5361e+00, -7.6145e-01]],
[[-6.7948e-01, -2.1576e+00, -1.7790e+00, ..., -3.9332e+00,
-5.4666e+00, -1.8585e+00],
[-2.2589e+00, -4.2106e+00, -3.0848e+00, ..., -5.8964e+00,
-8.1064e+00, -3.4208e+00],
[-2.5811e+00, -4.6389e+00, -5.2117e+00, ..., -4.9739e+00,
-6.1915e+00, -2.5519e+00],
...,
[-1.8033e+00, -2.4338e+00, -2.5461e+00, ..., -6.9475e+00,
-7.1975e+00, -4.1718e+00],
[-3.5313e+00, -4.0065e+00, -3.4062e+00, ..., -6.8634e+00,
-8.4831e+00, -3.3356e+00],
[ 2.9982e-01, -2.1815e-01, 3.9488e-01, ..., 3.9835e-01,
4.4983e-01, 2.3752e-01]],
[[-2.8162e-01, 6.2374e-02, 7.6546e-01, ..., 6.3722e-01,
2.3709e-01, 1.5148e+00],
[ 7.8324e-02, -4.1148e-01, 6.3854e-01, ..., 6.1008e-01,
-1.0599e-01, 1.0225e-01],
[-1.8859e-01, 1.1188e-01, 8.4703e-01, ..., 6.2052e-02,
-6.6164e-02, -1.3919e-01],
...,
[ 2.9513e-01, 1.4520e-01, 5.1186e-01, ..., 1.0074e+00,
-4.8290e+00, -2.8941e+00],
[-1.6523e-01, -3.0863e-02, 1.2269e-01, ..., 2.0838e+00,
-8.7785e-01, -1.0315e+00],
[-6.7243e-02, -3.6811e-01, 1.2815e-01, ..., 6.3527e-02,
2.4447e-01, -5.3500e-01]],
...,
[[ 4.4368e-01, -3.7407e-01, -1.5917e-02, ..., 4.4068e-01,
3.4492e-01, -1.3095e+00],
[-7.3259e-01, -5.6372e-01, -5.7223e-01, ..., -1.6394e-01,
-7.8213e-01, -1.3931e+00],
[-3.4921e-01, -1.1606e-01, -2.4447e-01, ..., -9.4806e-02,
-5.6854e-02, -9.3688e-01],
...,
[-7.6112e-01, -7.9474e-01, -1.1303e+00, ..., -2.7447e+00,
-2.8193e+00, -3.7350e+00],
[-2.0708e+00, -1.4848e+00, -1.1411e+00, ..., -4.5760e+00,
-3.3786e+00, -2.4900e+00],
[-4.3994e-01, -6.2368e-01, -3.4404e-01, ..., -3.6154e-01,
-5.4798e-01, -1.2031e+00]],
[[ 9.3707e-02, 2.5712e-01, 6.9195e-01, ..., -7.2685e-02,
3.8810e-01, -3.8587e-01],
[ 2.9107e-02, 2.3518e-01, 4.0811e-01, ..., 2.0767e-01,
9.0702e-02, -8.2634e-01],
[-7.2994e-02, 2.6637e-01, 4.7046e-01, ..., 3.0491e-01,
1.6711e-01, -4.2607e-01],
...,
[-1.1818e-02, 6.5330e-03, -1.1389e-01, ..., -5.3160e-01,
8.9383e-01, 6.7187e-01],
[-4.2526e-01, -6.1279e-01, 3.5547e-02, ..., -1.6415e+00,
-1.3397e+00, -1.4162e+00],
[ 5.4329e-01, 4.2333e-01, 1.0835e-01, ..., 3.3808e-01,
3.1618e-01, 3.0792e-01]],
[[-9.7280e-01, 2.9209e-01, 3.7652e-01, ..., -8.1672e-02,
-7.9590e-02, 5.6179e-01],
[-1.1806e+00, 2.4500e-01, -4.1531e-01, ..., -9.2818e-01,
-4.3016e-01, 8.8138e-01],
[-9.8475e-01, 1.7638e-01, -4.7055e-01, ..., -5.7535e-01,
-5.2926e-01, 6.0919e-01],
...,
[-1.0505e+00, 5.0225e-01, 8.1164e-01, ..., -5.0907e+00,
-3.2192e+00, 1.7882e+00],
[-1.5030e+00, -2.7639e-01, -3.1139e-01, ..., -2.0901e-01,
-1.4647e-01, 1.4283e+00],
[-6.0259e-01, 7.9271e-02, -2.1853e-01, ..., -4.5457e-01,
-4.4220e-01, 1.0136e+00]]],
[[[ 1.3619e-02, -2.5729e-01, -2.9208e-02, ..., -9.7757e-01,
-5.7852e-01, -1.4846e+00],
[ 3.1229e-01, -9.3305e-01, -6.9254e-01, ..., -1.1951e+00,
-1.3304e+00, -1.5146e+00],
[ 2.4230e-01, -5.6162e-01, -5.1459e-01, ..., -1.2703e+00,
-1.1672e+00, -1.8458e+00],
...,
[-1.7201e-02, -6.4679e-01, 1.6881e-01, ..., -9.2297e-01,
-1.0392e+00, -1.9344e+00],
[-6.4768e-01, -7.3301e-01, 2.7317e-01, ..., -4.5835e-01,
-5.7221e-01, -1.7299e+00],
[-7.5608e-01, -1.3564e+00, -1.0459e+00, ..., -1.1335e+00,
-1.5393e+00, -7.6910e-01]],
[[-7.8935e-01, -2.1911e+00, -1.8866e+00, ..., -3.9810e+00,
-5.4898e+00, -1.8835e+00],
[-2.0274e+00, -3.7440e+00, -2.9759e+00, ..., -5.9504e+00,
-8.0785e+00, -3.4320e+00],
[-1.6118e+00, -2.5840e+00, -2.5947e+00, ..., -6.2530e+00,
-7.0821e+00, -2.6796e+00],
...,
[-1.7849e+00, -2.4707e+00, -2.9420e+00, ..., -4.9803e+00,
-6.2852e+00, -2.4550e+00],
[-3.8578e+00, -4.4721e+00, -3.8482e+00, ..., -6.4873e+00,
-7.9425e+00, -2.5683e+00],
[ 2.7014e-01, -3.1797e-01, 2.6986e-01, ..., 4.1440e-01,
4.7158e-01, 2.1290e-01]],
[[-2.1385e-01, 1.3479e-01, 7.1607e-01, ..., 6.2546e-01,
2.5573e-01, 1.4717e+00],
[ 8.1953e-02, -3.8844e-01, 5.4438e-01, ..., 5.4480e-01,
-3.0041e-02, 7.2071e-02],
[-1.2691e-01, -2.2847e-01, 2.4224e-01, ..., 1.5987e-01,
1.7987e-01, -1.2170e-01],
...,
[-2.6692e-01, -2.0149e-01, 5.4232e-01, ..., 5.1155e-03,
-9.5167e-02, -1.7174e-01],
[-2.6359e-01, -1.6630e-01, 6.1099e-01, ..., -9.6629e-02,
-8.8198e-03, -2.1960e-01],
[-1.3256e-01, -4.8380e-01, 4.3182e-01, ..., 9.6360e-02,
2.1845e-01, -5.7166e-01]],
...,
[[ 5.9712e-01, -2.8788e-01, -7.3053e-03, ..., 4.2445e-01,
3.8042e-01, -1.2965e+00],
[-5.0038e-01, -5.5846e-01, -5.2208e-01, ..., -1.5369e-01,
-7.3973e-01, -1.3969e+00],
[-3.5704e-01, -2.2088e-01, -2.4801e-01, ..., 4.5401e-02,
3.9186e-02, -9.5743e-01],
...,
[-4.4817e-01, -1.4005e-01, -5.4489e-02, ..., -1.0366e-01,
-1.0723e-01, -9.4521e-01],
[-2.3490e+00, -1.7251e+00, -1.3976e+00, ..., -1.5281e+00,
-1.5509e+00, -1.8149e+00],
[-4.8181e-01, -8.1970e-01, -4.7498e-01, ..., -3.4821e-01,
-5.1627e-01, -1.1992e+00]],
[[ 1.6369e-01, 2.4324e-01, 6.3871e-01, ..., -1.0319e-01,
3.2731e-01, -3.8463e-01],
[ 1.3990e-01, 2.4366e-01, 4.4945e-01, ..., 2.0517e-01,
4.8160e-02, -8.4608e-01],
[-1.1288e-01, 8.9059e-02, 2.5309e-01, ..., -3.5838e-01,
-3.2126e-01, -5.9545e-01],
...,
[-2.0879e-01, 1.9899e-02, 6.1502e-01, ..., 2.2317e-01,
1.3309e-01, -4.5949e-01],
[-5.6528e-01, -7.2410e-01, 1.0064e-01, ..., -1.7075e-01,
-2.7305e-01, -9.0412e-01],
[ 4.9718e-01, 3.7284e-01, 1.6661e-01, ..., 3.2445e-01,
3.2779e-01, 3.0935e-01]],
[[-8.2302e-01, 2.5493e-01, 3.0639e-01, ..., -9.1835e-02,
-8.4400e-02, 5.4859e-01],
[-1.1215e+00, 2.4417e-01, -3.4662e-01, ..., -9.3098e-01,
-4.0858e-01, 8.8802e-01],
[-9.4889e-01, 3.5969e-01, -1.1450e-01, ..., -6.2175e-01,
-3.0835e-01, 6.4789e-01],
...,
[-9.6803e-01, 1.1581e-01, -5.9705e-01, ..., -5.4907e-01,
-5.4544e-01, 6.1458e-01],
[-1.6975e+00, -4.2771e-01, -6.0777e-01, ..., -7.0984e-01,
-7.0866e-01, 3.9526e-01],
[-7.3304e-01, -9.2838e-02, -4.4068e-01, ..., -4.8411e-01,
-4.5079e-01, 1.0106e+00]]]], device='cuda:0'),
tensor([[[[-3.1313e-01, -1.0057e+00, -8.7053e-01, ..., -2.5326e-02,
5.8540e-01, 5.0417e-01],
[ 1.6647e-01, -1.8772e+00, -1.2921e+00, ..., -4.4667e-01,
-2.1231e-01, 1.0826e-01],
[ 1.4799e-01, -1.2774e+00, -1.4019e+00, ..., -3.3054e-01,
-1.2402e-01, -8.3918e-02],
...,
[ 4.1713e-01, -1.9538e+00, -1.1392e+00, ..., -2.9170e-01,
3.7950e-02, -1.4385e-02],
[ 1.0099e-01, -8.8198e-01, -4.2713e-01, ..., 7.4178e-02,
3.0681e-01, -1.3973e-02],
[-1.1109e-01, -1.0368e+00, -9.0138e-01, ..., -1.1808e+00,
-1.2447e+00, -4.6703e-01]],
[[-2.1014e+00, -4.1409e+00, -3.7157e+00, ..., -5.3611e-01,
-1.0387e+00, -2.7564e-02],
[-3.9832e+00, -6.4499e+00, -5.6150e+00, ..., -1.7834e+00,
-2.6074e+00, -6.5514e-01],
[-3.1929e+00, -6.1528e+00, -7.4029e+00, ..., -1.2886e+00,
-1.4925e+00, -3.5865e-01],
...,
[-3.0129e+00, -5.2362e+00, -4.9767e+00, ..., -1.3463e+00,
-1.5824e+00, -3.4970e-01],
[-4.9414e+00, -5.9781e+00, -5.7231e+00, ..., -2.7293e+00,
-3.6795e+00, -1.0559e+00],
[ 7.1372e-01, -1.1548e-01, 5.1496e-01, ..., 3.6941e-01,
3.5745e-01, 6.1817e-02]],
[[-6.1789e-01, 4.0400e-01, 8.3107e-01, ..., 7.7704e-01,
2.5815e-01, 1.0461e+00],
[-9.5282e-01, 4.9760e-02, 6.3328e-01, ..., 4.1472e-01,
6.2096e-02, 4.7694e-01],
[-1.3336e+00, -2.9843e-03, -1.4728e-01, ..., 6.6360e-02,
8.6556e-02, 4.2915e-01],
...,
[-1.9891e+00, 6.8291e-01, -2.2625e-01, ..., 2.6372e-01,
2.2972e-01, 4.1329e-01],
[-1.3356e+00, 1.8521e-01, -1.1449e-01, ..., 1.0121e-01,
8.1943e-02, 5.8417e-01],
[-5.4968e-01, 2.8907e-01, 1.5250e-01, ..., -9.5950e-02,
7.1237e-02, 3.7255e-01]],
...,
[[ 6.5335e-01, 8.1922e-02, 2.7580e-01, ..., -7.8132e-02,
-3.2843e-01, -8.8413e-01],
[-8.2023e-01, -3.2823e-01, -9.2821e-02, ..., -5.7721e-01,
-1.2066e+00, -8.1066e-01],
[-5.2104e-01, -3.4152e-01, -7.1652e-02, ..., -1.0264e-01,
-2.0006e-01, -3.9263e-01],
...,
[-8.0111e-01, -1.5027e+00, -7.3889e-01, ..., -1.9043e-01,
-4.2791e-01, -7.3282e-01],
[-2.1590e+00, -1.7650e+00, -1.2185e+00, ..., -9.2714e-01,
-9.7741e-01, -8.7113e-01],
[-4.2564e-01, -9.9291e-01, -3.3082e-01, ..., -2.0097e-02,
-1.8041e-01, -4.0069e-01]],
[[ 7.9414e-03, -2.4877e-01, -5.3112e-02, ..., 2.7873e-01,
6.1340e-01, 2.7568e-01],
[ 1.1849e-01, -8.5667e-01, -5.9845e-02, ..., 9.9507e-03,
1.1437e-01, -2.9542e-02],
[-1.8948e-01, -6.8775e-01, 4.4584e-01, ..., -1.2443e-01,
-9.1871e-02, 1.8845e-01],
...,
[-3.4795e-01, -1.3451e+00, 4.1085e-01, ..., -1.5159e-01,
-1.0952e-01, 6.2318e-02],
[-4.8758e-01, -1.8956e+00, -2.5314e-01, ..., -4.7324e-01,
-4.2439e-01, -1.3015e-01],
[ 7.3020e-01, -6.7640e-02, 2.8108e-01, ..., -2.9903e-01,
-3.0149e-01, 2.8254e-01]],
[[-6.5935e-01, -1.2473e-01, 3.8595e-02, ..., 5.0334e-01,
3.2548e-01, 5.3309e-01],
[-1.0918e+00, -7.7396e-01, -8.8697e-01, ..., -4.4680e-02,
-1.5273e-01, 4.0908e-01],
[-6.9996e-01, -8.2897e-01, -1.1594e+00, ..., 6.4818e-02,
-1.8785e-01, 9.8073e-02],
...,
[ 1.6842e-01, -7.0697e-01, -1.4233e-01, ..., -1.3300e-02,
-1.3455e-01, 2.9290e-01],
[-7.9205e-01, -9.7432e-01, -5.1282e-01, ..., -9.8630e-02,
-3.2143e-01, -2.3632e-01],
[-4.7608e-01, -8.4673e-01, -3.6861e-01, ..., -1.7226e-01,
-3.4724e-01, 1.0237e-01]]],
[[[-2.1729e-01, -2.7435e-01, 1.8862e-01, ..., 5.3127e-02,
5.2153e-01, 1.5416e-01],
[-1.3258e-01, -1.5345e+00, -8.8158e-01, ..., -7.1230e-01,
-4.0793e-01, -3.2231e-01],
[-1.4225e-01, -1.3477e+00, -8.6672e-01, ..., -4.9098e-01,
-3.8733e-01, -6.8550e-01],
...,
[-1.8140e-01, -1.5040e+00, -1.1170e+00, ..., -1.3223e+00,
-9.9598e-01, -2.0059e+00],
[-8.0067e-01, -1.4383e+00, -8.3107e-01, ..., -1.2809e+00,
-5.1498e-01, -1.7964e+00],
[-9.7916e-01, -1.7790e+00, -1.4869e+00, ..., -1.9461e+00,
-1.7221e+00, -1.0449e+00]],
[[-2.0834e-01, -1.4610e+00, -6.1967e-01, ..., -6.0925e-01,
-1.3399e+00, -2.2879e-01],
[-2.0522e+00, -4.0487e+00, -2.4681e+00, ..., -2.3213e+00,
-3.3330e+00, -1.1104e+00],
[-1.8655e+00, -3.3830e+00, -3.9481e+00, ..., -2.5553e+00,
-2.9382e+00, -1.2247e+00],
...,
[-3.7660e+00, -5.5053e+00, -5.4425e+00, ..., -7.1937e+00,
-6.1622e+00, -2.8898e+00],
[-5.8565e+00, -6.5855e+00, -6.1224e+00, ..., -8.1529e+00,
-7.7721e+00, -2.8427e+00],
[ 5.3314e-02, -5.4996e-01, 3.0067e-01, ..., -5.6721e-01,
7.1581e-01, 1.6124e-01]],
[[-5.3346e-01, -1.8322e-01, 3.2741e-01, ..., 3.6410e-01,
5.4882e-02, 8.6796e-01],
[-3.5430e-01, -4.6929e-01, 4.2865e-01, ..., 3.9662e-01,
2.7733e-01, 5.2007e-01],
[-8.8068e-01, -2.6984e-01, -2.9444e-02, ..., 4.5335e-02,
1.5489e-01, 4.1763e-01],
...,
[-1.3627e+00, 5.7876e-02, 3.3783e-01, ..., 5.7859e-01,
-5.6743e-03, 2.2579e-01],
[-1.2766e+00, 1.4085e-01, -1.8671e-02, ..., 2.1354e-01,
-1.1392e-02, 2.2541e-01],
[-1.1108e+00, -4.9879e-02, -6.9685e-03, ..., 2.5330e-01,
-5.1570e-02, -3.6664e-01]],
...,
[[ 6.3055e-02, -6.0495e-01, -3.9965e-01, ..., -7.0508e-02,
-2.5589e-01, -7.4908e-01],
[-8.4499e-01, -5.9494e-01, -5.3024e-01, ..., -2.7870e-01,
-9.8076e-01, -7.3075e-01],
[-5.4195e-01, -1.3277e-01, 2.3994e-01, ..., 6.0081e-02,
-1.9609e-01, -5.0971e-01],
...,
[-4.4042e-01, -7.3143e-01, -5.5934e-01, ..., -9.2625e-01,
-6.1329e-01, -1.2836e+00],
[-2.5666e+00, -2.7327e+00, -1.9374e+00, ..., -2.7209e+00,
-2.1898e+00, -2.1663e+00],
[-4.9396e-01, -1.5786e+00, -9.8758e-01, ..., -1.7758e+00,
-1.0000e+00, -1.4218e+00]],
[[-1.8876e-01, 4.1602e-02, 4.3175e-01, ..., 9.3568e-02,
5.4921e-01, 1.7852e-01],
[-4.4989e-01, -2.7517e-01, 1.9689e-02, ..., -7.0278e-02,
1.4611e-01, -6.4013e-02],
[-6.9238e-01, -5.9250e-01, 1.5514e-01, ..., 1.7345e-02,
1.7572e-02, -4.5697e-02],
...,
[ 5.2726e-02, -9.0877e-01, 2.1139e-01, ..., -2.6490e-02,
7.1916e-02, -7.5541e-01],
[-3.5342e-01, -1.6455e+00, -1.3543e-01, ..., -2.8845e-01,
-1.9587e-01, -1.0458e+00],
[ 5.4799e-01, -1.3627e-02, 2.2714e-01, ..., 2.4610e-01,
2.1668e-01, 2.8110e-01]],
[[-6.8073e-01, 4.0012e-01, 5.5875e-01, ..., 4.8924e-01,
3.7667e-01, 6.0742e-01],
[-8.7090e-01, 2.6681e-01, -2.4912e-01, ..., -2.8565e-01,
-1.4603e-01, 6.3650e-01],
[-5.0373e-01, 2.9785e-01, -1.8096e-01, ..., 6.7050e-02,
8.6351e-02, 5.9124e-01],
...,
[-1.1380e+00, -8.8717e-01, -5.0171e-01, ..., -1.8717e-01,
-8.5124e-02, 8.5782e-01],
[-1.7775e+00, -1.6173e+00, -7.8880e-01, ..., -1.7527e-01,
-4.9907e-01, 3.9808e-01],
[-9.6497e-01, -8.2701e-01, -1.6806e-01, ..., -4.7299e-02,
-1.9432e-01, 1.1226e+00]]],
[[[-7.2199e-01, -1.4544e+00, -1.4359e+00, ..., -1.2586e+00,
-8.6561e-01, -1.6603e+00],
[-3.2157e-01, -2.4442e+00, -2.3078e+00, ..., -1.7272e+00,
-1.8666e+00, -1.8990e+00],
[-1.2662e-01, -1.8904e+00, -1.7017e+00, ..., -1.3749e+00,
-1.4323e+00, -2.1362e+00],
...,
[ 7.9030e-02, -5.4842e-01, -2.6869e-01, ..., -1.4299e-01,
-5.4660e-02, -4.4086e-01],
[-3.6391e-01, -4.5465e-01, 3.7742e-02, ..., 1.2292e-01,
2.6113e-01, -1.6969e-01],
[-6.2112e-01, -1.2939e+00, -1.0593e+00, ..., -1.0480e+00,
-1.1802e+00, -4.9129e-01]],
[[-2.6325e+00, -5.0367e+00, -4.5645e+00, ..., -4.8329e+00,
-6.0178e+00, -2.4168e+00],
[-5.3876e+00, -8.1927e+00, -7.4603e+00, ..., -7.7046e+00,
-9.2369e+00, -4.3747e+00],
[-4.0457e+00, -5.8986e+00, -5.9421e+00, ..., -6.2966e+00,
-6.4951e+00, -3.2670e+00],
...,
[-1.7511e+00, -2.2431e+00, -1.7698e+00, ..., -1.9021e+00,
-2.2782e+00, -8.9275e-01],
[-2.7568e+00, -2.9848e+00, -2.0516e+00, ..., -2.2618e+00,
-3.4311e+00, -1.0223e+00],
[ 3.2049e-01, -2.8059e-01, 3.5653e-01, ..., 3.1391e-01,
4.3001e-01, 1.9830e-01]],
[[-6.3386e-01, 4.7066e-01, 8.6626e-01, ..., 8.0751e-01,
6.1917e-01, 1.9126e+00],
[-9.6520e-01, 1.9235e-02, 4.0591e-01, ..., 5.3093e-01,
-4.1319e-03, 1.4252e-01],
[-1.7599e+00, -2.1076e-01, -2.9242e-02, ..., 3.5470e-02,
-4.4130e-02, -9.9962e-02],
...,
[-6.5871e-01, 7.4510e-02, 2.5469e-01, ..., 1.1053e-01,
2.4882e-01, 3.9011e-01],
[-4.2747e-01, 9.9547e-02, 2.0676e-01, ..., 1.7333e-01,
2.2560e-01, 5.6475e-01],
[-1.5259e-01, -2.4149e-01, 1.1282e-01, ..., -1.8152e-02,
1.1538e-01, 3.3411e-01]],
...,
[[ 1.2262e+00, -3.8613e-01, -6.6542e-02, ..., 2.3895e-01,
-1.0625e-01, -2.1294e+00],
[-7.5246e-01, -9.0698e-01, -4.8777e-01, ..., -4.8140e-01,
-1.4931e+00, -2.0890e+00],
[-6.9722e-01, -8.4382e-01, -2.8776e-01, ..., -3.3163e-01,
-6.0676e-01, -1.3963e+00],
...,
[-6.5312e-01, -5.1402e-01, -3.7458e-01, ..., -3.9714e-01,
-5.0406e-01, -5.5978e-01],
[-2.2496e+00, -1.7600e+00, -1.2871e+00, ..., -1.3299e+00,
-1.3623e+00, -1.0838e+00],
[-6.0289e-01, -8.3966e-01, -4.3559e-01, ..., -2.5453e-01,
-3.5804e-01, -4.1373e-01]],
[[ 4.4467e-02, 1.5571e-02, 4.7021e-01, ..., 4.6127e-01,
9.1531e-01, -3.1675e-02],
[ 2.1896e-01, -8.2779e-01, -1.1372e-01, ..., 4.5872e-01,
4.0749e-01, -8.2816e-01],
[ 1.5599e-01, -6.2342e-01, 2.7384e-01, ..., 2.7528e-01,
5.3878e-01, -4.0673e-01],
...,
[-2.2861e-01, -5.0331e-01, 1.9519e-01, ..., 2.5277e-01,
2.6155e-01, 3.9029e-02],
[-4.8489e-01, -8.9174e-01, 7.0088e-02, ..., 2.0946e-02,
-5.0560e-02, -1.9140e-01],
[ 3.1415e-01, 2.1192e-02, -1.7351e-01, ..., -1.2976e-01,
-9.9408e-02, 1.6772e-01]],
[[-1.5540e+00, -7.9171e-01, -4.2179e-01, ..., -4.3662e-01,
-2.9440e-01, 8.6521e-01],
[-2.1128e+00, -7.4652e-01, -5.5351e-01, ..., -7.9980e-01,
-1.9819e-01, 1.6595e+00],
[-1.5152e+00, -5.6020e-01, -1.9246e-01, ..., -3.7179e-01,
-1.2681e-01, 1.4001e+00],
...,
[-6.2681e-01, -2.0812e-01, -1.0459e-01, ..., -8.0453e-02,
-1.1495e-02, 6.1252e-01],
[-1.1499e+00, -5.6185e-01, -2.7503e-01, ..., -2.9340e-01,
-1.7758e-01, 1.5408e-01],
[-3.6073e-01, 5.1891e-03, -1.7286e-01, ..., -2.2034e-01,
-2.3782e-01, 3.1209e-01]]],
[[[-7.2857e-01, -1.4651e+00, -1.4377e+00, ..., -1.2778e+00,
-8.2544e-01, -1.6419e+00],
[-3.2627e-01, -2.4662e+00, -2.2365e+00, ..., -1.7612e+00,
-1.8832e+00, -1.9044e+00],
[-1.2031e-01, -1.9037e+00, -1.6833e+00, ..., -1.2711e+00,
-1.4395e+00, -2.1106e+00],
...,
[ 8.7338e-02, -5.4725e-01, -2.2767e-01, ..., -8.5482e-02,
-1.5361e-02, -4.1570e-01],
[-2.7466e-01, -4.5768e-01, -2.2451e-02, ..., 1.3046e-01,
2.8779e-01, -1.3701e-01],
[-5.7985e-01, -1.3020e+00, -1.1067e+00, ..., -1.0751e+00,
-1.1865e+00, -4.8355e-01]],
[[-2.6738e+00, -5.1087e+00, -4.6466e+00, ..., -4.8575e+00,
-6.0724e+00, -2.4353e+00],
[-5.3388e+00, -8.1153e+00, -7.4798e+00, ..., -7.6746e+00,
-9.2993e+00, -4.3983e+00],
[-3.9997e+00, -5.7480e+00, -5.7081e+00, ..., -5.9307e+00,
-6.4719e+00, -3.2315e+00],
...,
[-1.6890e+00, -2.0907e+00, -1.7142e+00, ..., -1.8313e+00,
-2.2029e+00, -8.4138e-01],
[-2.7739e+00, -2.8876e+00, -1.9602e+00, ..., -2.2441e+00,
-3.3909e+00, -9.9401e-01],
[ 3.5452e-01, -2.1796e-01, 2.9904e-01, ..., 2.9524e-01,
4.2495e-01, 1.9000e-01]],
[[-6.5747e-01, 4.5559e-01, 8.5377e-01, ..., 8.0781e-01,
6.6364e-01, 1.9568e+00],
[-9.8761e-01, 2.2927e-02, 4.1429e-01, ..., 4.5140e-01,
-1.3874e-01, 9.6361e-02],
[-1.7555e+00, -2.3813e-01, -4.6980e-02, ..., -5.8788e-02,
-1.1051e-01, -1.5826e-01],
...,
[-6.1452e-01, 5.4526e-02, 2.5611e-01, ..., 1.9854e-01,
2.6258e-01, 4.1357e-01],
[-4.3725e-01, 3.6698e-02, 1.9714e-01, ..., 1.6519e-01,
2.2674e-01, 5.7408e-01],
[-1.1613e-01, -2.5308e-01, 1.4417e-01, ..., 5.6554e-04,
1.0344e-01, 3.4111e-01]],
...,
[[ 1.2151e+00, -3.7950e-01, -5.6179e-02, ..., 3.2995e-01,
-8.1749e-02, -2.1226e+00],
[-7.6702e-01, -9.0142e-01, -4.6835e-01, ..., -5.6682e-01,
-1.4819e+00, -2.0655e+00],
[-7.1440e-01, -8.6392e-01, -2.5624e-01, ..., -3.3449e-01,
-5.6652e-01, -1.3702e+00],
...,
[-6.2303e-01, -5.1195e-01, -3.5139e-01, ..., -3.3480e-01,
-4.7419e-01, -5.4106e-01],
[-2.1708e+00, -1.7033e+00, -1.2974e+00, ..., -1.3339e+00,
-1.3614e+00, -1.0887e+00],
[-5.5990e-01, -7.5550e-01, -4.3269e-01, ..., -2.3537e-01,
-3.5878e-01, -4.3030e-01]],
[[ 2.9677e-02, 3.9979e-03, 4.4879e-01, ..., 5.4910e-01,
9.8902e-01, 2.6012e-02],
[ 2.0432e-01, -8.0528e-01, -4.2174e-02, ..., 5.2764e-01,
4.3439e-01, -8.4391e-01],
[ 1.2427e-01, -6.2144e-01, 3.3523e-01, ..., 6.6759e-01,
5.2910e-01, -3.9258e-01],
...,
[-2.1521e-01, -4.3687e-01, 2.3669e-01, ..., 3.6092e-01,
3.3548e-01, 6.2381e-02],
[-5.0918e-01, -9.1248e-01, 2.6342e-02, ..., 1.0245e-02,
-5.0146e-02, -1.9466e-01],
[ 3.3492e-01, -3.5745e-02, -2.2098e-01, ..., -1.3428e-01,
-1.1896e-01, 1.6460e-01]],
[[-1.5549e+00, -7.8944e-01, -4.2503e-01, ..., -4.2405e-01,
-2.4558e-01, 9.2297e-01],
[-2.1164e+00, -7.5899e-01, -5.4768e-01, ..., -8.1699e-01,
-2.0825e-01, 1.6345e+00],
[-1.4863e+00, -5.7890e-01, -2.0467e-01, ..., -4.2456e-01,
-1.1630e-01, 1.3892e+00],
...,
[-6.5024e-01, -2.0861e-01, -1.3108e-01, ..., -1.2987e-01,
7.4229e-03, 6.0623e-01],
[-1.0859e+00, -5.3304e-01, -3.2901e-01, ..., -2.8275e-01,
-1.7671e-01, 1.4319e-01],
[-3.1823e-01, 3.8010e-02, -1.4290e-01, ..., -2.3148e-01,
-2.5369e-01, 2.9863e-01]]],
[[[ 2.9852e-02, -2.5259e-01, -5.1990e-02, ..., -1.1935e+00,
-6.5947e-01, -1.0194e+00],
[ 3.0260e-01, -9.4869e-01, -7.0747e-01, ..., -1.5968e+00,
-1.3214e+00, -1.2811e+00],
[ 2.0906e-01, -6.2099e-01, -5.1236e-01, ..., -9.8921e-01,
-1.7544e+00, -2.8481e+00],
...,
[-2.3743e-02, -6.3179e-01, 2.3802e-02, ..., -9.9955e-01,
-7.2648e-01, -1.9079e+00],
[-6.5497e-01, -7.0322e-01, 8.1805e-02, ..., -7.5886e-01,
-5.3554e-01, -1.4526e+00],
[-7.7369e-01, -1.3503e+00, -1.1575e+00, ..., -1.3027e+00,
-1.4883e+00, -9.0863e-01]],
[[-8.1502e-01, -2.2943e+00, -1.9268e+00, ..., -4.0948e+00,
-4.9079e+00, -1.8899e+00],
[-2.0780e+00, -3.9214e+00, -3.1329e+00, ..., -6.2042e+00,
-6.8842e+00, -2.8157e+00],
[-1.6437e+00, -2.6843e+00, -2.6923e+00, ..., -8.4116e+00,
-1.0057e+01, -4.9882e+00],
...,
[-1.8133e+00, -2.4964e+00, -2.3523e+00, ..., -5.2917e+00,
-5.5125e+00, -2.4765e+00],
[-3.8234e+00, -4.4317e+00, -3.7365e+00, ..., -6.3098e+00,
-7.2578e+00, -2.7727e+00],
[ 2.6979e-01, -3.5710e-01, 3.8021e-01, ..., 5.3746e-01,
6.4562e-01, 7.4860e-02]],
[[-2.3261e-01, 1.5189e-01, 7.5203e-01, ..., 9.5774e-01,
7.1324e-01, 1.5175e+00],
[ 1.3752e-02, -4.3368e-01, 5.5633e-01, ..., 2.5924e-01,
2.6488e-01, 2.3452e-01],
[-1.5370e-01, -3.2333e-01, 1.4276e-01, ..., -2.3455e-01,
-9.1753e-03, 3.8661e-01],
...,
[-2.7841e-01, -2.0122e-01, 4.8666e-01, ..., 6.3252e-01,
1.5049e-02, -1.0753e-01],
[-2.7403e-01, -1.9905e-01, 3.8644e-01, ..., -9.0375e-02,
1.1816e-01, 1.6316e-01],
[-1.3879e-01, -4.7362e-01, 1.0729e-01, ..., -4.0855e-01,
-8.0354e-02, -2.5785e-01]],
...,
[[ 6.1066e-01, -2.2500e-01, 2.0875e-02, ..., -1.0819e-01,
-2.6557e-01, -1.4538e+00],
[-4.3410e-01, -4.9480e-01, -5.0213e-01, ..., -7.7039e-01,
-1.4971e+00, -1.6748e+00],
[-3.1638e-01, -1.2178e-01, -2.3302e-01, ..., -5.1482e-01,
-6.5950e-01, -1.0540e+00],
...,
[-4.3597e-01, -1.3511e-01, -2.0909e-01, ..., -1.2822e+00,
-1.5989e+00, -2.2416e+00],
[-2.3398e+00, -1.7167e+00, -1.5500e+00, ..., -1.9021e+00,
-1.9521e+00, -1.7515e+00],
[-5.0760e-01, -8.5277e-01, -3.9944e-01, ..., -7.3663e-01,
-7.8054e-01, -8.7549e-01]],
[[ 1.8235e-01, 2.3600e-01, 5.9086e-01, ..., 4.6312e-01,
7.7070e-01, -3.6014e-02],
[ 1.4754e-01, 2.9023e-01, 4.8138e-01, ..., 3.1824e-01,
4.2994e-01, -2.8980e-01],
[-1.3198e-01, 1.4587e-01, 3.5898e-01, ..., 9.7399e-01,
6.1587e-01, -1.7554e+00],
...,
[-2.0578e-01, -4.8141e-03, 6.2759e-01, ..., 3.6567e-01,
1.6117e-01, -8.9304e-01],
[-5.6054e-01, -7.2491e-01, 1.2473e-01, ..., -8.7602e-02,
-1.3092e-01, -5.9510e-01],
[ 5.0323e-01, 3.7511e-01, 9.4528e-02, ..., 2.0483e-01,
1.4457e-01, 3.3114e-01]],
[[-8.1799e-01, 2.3982e-01, 3.3099e-01, ..., 1.0128e-01,
2.1898e-01, 8.4854e-01],
[-1.1331e+00, 1.9466e-01, -3.6216e-01, ..., -8.0904e-02,
1.9118e-01, 1.4270e+00],
[-1.0076e+00, 3.1009e-01, -1.0008e-01, ..., 9.1438e-02,
2.7757e-01, 1.2599e+00],
...,
[-9.4571e-01, 1.8356e-01, -2.0642e-01, ..., -8.9287e-02,
5.6251e-01, 1.6191e+00],
[-1.6968e+00, -4.3080e-01, -4.4543e-01, ..., -1.4000e-01,
1.5105e-01, 8.5116e-01],
[-7.3215e-01, -1.0391e-01, -2.9095e-01, ..., 2.3410e-01,
1.3324e-01, 1.0972e+00]]]], device='cuda:0'),
tensor([[[[ 5.6704e-03, -2.4741e-01, -6.9683e-02, ..., -1.0015e+00,
-5.3187e-01, -1.4849e+00],
[ 2.9614e-01, -9.3166e-01, -7.4748e-01, ..., -1.9900e+00,
-1.7174e+00, -1.5124e+00],
[ 2.6374e-01, -5.5080e-01, -5.1110e-01, ..., -1.7850e+00,
-9.6753e-01, -1.8440e+00],
...,
[-1.8976e-03, -2.7731e-01, 1.1500e+00, ..., -1.6457e+00,
-2.3788e+00, -2.5935e+00],
[-6.3256e-01, -6.6845e-01, 7.8694e-01, ..., -5.1802e-01,
-5.4465e-01, -1.7630e+00],
[-7.8541e-01, -1.3390e+00, -1.1268e+00, ..., -1.1647e+00,
-1.5717e+00, -8.1400e-01]],
[[-8.4369e-01, -2.2494e+00, -1.9722e+00, ..., -3.9361e+00,
-5.4614e+00, -1.8655e+00],
[-2.1267e+00, -3.8611e+00, -3.1162e+00, ..., -7.8215e+00,
-9.6178e+00, -3.4194e+00],
[-1.7043e+00, -2.7262e+00, -2.7567e+00, ..., -8.2067e+00,
-9.7125e+00, -2.5478e+00],
...,
[-1.9403e+00, -2.6987e+00, -3.4599e+00, ..., -7.7854e+00,
-7.1506e+00, -2.5673e+00],
[-3.9093e+00, -4.7570e+00, -3.8806e+00, ..., -6.5441e+00,
-8.0692e+00, -2.5213e+00],
[ 2.6735e-01, -2.9391e-01, 1.2364e-01, ..., 4.0123e-01,
4.4022e-01, 2.5351e-01]],
[[-1.8534e-01, 1.9169e-01, 7.3618e-01, ..., 6.0510e-01,
2.1476e-01, 1.5126e+00],
[ 1.1270e-01, -3.1467e-01, 5.7623e-01, ..., 4.9703e-01,
1.8471e-01, 7.8032e-02],
[-1.4074e-01, -1.7459e-01, 2.3555e-01, ..., -1.3561e+00,
2.6874e-01, -1.2837e-01],
...,
[-3.1246e-01, 7.8746e-01, -3.7419e-01, ..., 1.0112e+00,
1.6499e-01, -6.0811e-03],
[-2.4082e-01, 1.1219e-01, 9.1414e-01, ..., -1.3265e-01,
-1.5526e-04, -1.9578e-01],
[-3.8954e-02, -4.8056e-01, 4.9896e-01, ..., 4.0625e-02,
1.8275e-01, -5.9911e-01]],
...,
[[ 6.4229e-01, -2.0925e-01, 7.7555e-02, ..., 4.2123e-01,
3.7220e-01, -1.3111e+00],
[-5.1076e-01, -5.5706e-01, -5.2060e-01, ..., 2.4895e-01,
-4.4223e-01, -1.4005e+00],
[-3.1109e-01, -2.3123e-01, -2.4602e-01, ..., -4.6426e-01,
-4.9157e-01, -9.5225e-01],
...,
[-3.4002e-01, -4.2642e-01, 2.3874e-01, ..., -4.6952e+00,
-7.1353e-01, -7.6350e-01],
[-2.2881e+00, -2.1578e+00, -2.1809e+00, ..., -1.5053e+00,
-1.5797e+00, -1.8420e+00],
[-5.0153e-01, -8.2201e-01, -7.1331e-01, ..., -3.8720e-01,
-5.3683e-01, -1.2304e+00]],
[[ 1.7627e-01, 2.6795e-01, 6.3028e-01, ..., -1.0489e-01,
4.1600e-01, -3.5662e-01],
[ 1.7898e-01, 2.1319e-01, 4.5381e-01, ..., -1.3115e+00,
-8.0248e-01, -8.2432e-01],
[-8.2020e-02, 4.1668e-02, 2.5984e-01, ..., -2.0112e+00,
-1.8486e+00, -4.6043e-01],
...,
[-1.8200e-01, 4.8466e-02, 3.9850e-01, ..., -6.6797e-01,
-7.9724e-01, -1.2749e+00],
[-5.7809e-01, -4.8586e-01, -8.1952e-02, ..., -2.0023e-01,
-2.3853e-01, -8.9471e-01],
[ 5.0965e-01, 4.0372e-01, 2.5707e-01, ..., 3.4343e-01,
3.3175e-01, 3.2198e-01]],
[[-8.0564e-01, 2.3863e-01, 2.7577e-01, ..., -1.5835e-01,
-1.1118e-01, 5.5529e-01],
[-1.1445e+00, 1.6047e-01, -4.2624e-01, ..., -9.8063e-02,
1.9482e-01, 9.1723e-01],
[-9.1903e-01, 3.4220e-01, -1.0210e-01, ..., -1.7042e-01,
3.8567e-01, 6.2479e-01],
...,
[-1.0509e+00, -5.5282e-01, -1.5967e-01, ..., 1.7839e+00,
9.5489e-01, 7.8703e-01],
[-1.7677e+00, -6.7400e-01, -6.4640e-01, ..., -7.1523e-01,
-7.0564e-01, 4.0145e-01],
[-8.0876e-01, -1.3846e-01, -4.7998e-01, ..., -5.0170e-01,
-4.7640e-01, 1.0374e+00]]],
[[[-7.1108e-01, -1.4150e+00, -1.4241e+00, ..., -2.2975e-01,
2.3475e-01, -4.0684e-01],
[-2.7795e-01, -2.4619e+00, -2.1979e+00, ..., -6.0962e-01,
-4.1220e-01, -5.8254e-01],
[-3.9027e-01, -2.0603e+00, -1.4427e+00, ..., -3.5839e-01,
-3.9459e-01, -1.0475e+00],
...,
[-4.8858e-01, -1.9523e+00, -1.8419e+00, ..., -2.9976e-01,
-2.4944e-01, -1.0881e+00],
[-9.8020e-01, -1.7175e+00, -1.1782e+00, ..., -4.2799e-03,
1.1637e-01, -9.6781e-01],
[-8.1508e-01, -1.6757e+00, -1.2632e+00, ..., -9.1003e-01,
-1.1120e+00, -6.5960e-01]],
[[-2.3661e+00, -4.8411e+00, -4.2284e+00, ..., -2.1668e+00,
-3.4937e+00, -1.2728e+00],
[-5.3634e+00, -8.0904e+00, -6.4502e+00, ..., -3.4278e+00,
-5.1004e+00, -2.0913e+00],
[-4.2982e+00, -6.6893e+00, -8.7612e+00, ..., -3.0331e+00,
-3.8484e+00, -1.7476e+00],
...,
[-4.3456e+00, -6.0742e+00, -5.4712e+00, ..., -2.9712e+00,
-3.8844e+00, -1.6991e+00],
[-6.4150e+00, -7.4790e+00, -6.3376e+00, ..., -3.9190e+00,
-5.3858e+00, -1.9441e+00],
[-2.8732e-02, -8.4662e-01, 4.2570e-01, ..., 4.1776e-01,
6.0196e-01, 1.7685e-01]],
[[-6.0118e-01, 5.6509e-01, 1.1054e+00, ..., 8.3052e-01,
5.6908e-01, 1.3414e+00],
[-8.9531e-01, -7.2810e-02, 3.5114e-01, ..., 4.0351e-01,
3.6545e-01, 2.1792e-01],
[-1.6122e+00, -1.6412e-01, -1.7677e-01, ..., 8.8164e-02,
3.0174e-01, 7.4311e-02],
...,
[-1.7237e+00, -9.8958e-02, -1.1233e-01, ..., 1.3021e-01,
2.3191e-01, 7.4618e-02],
[-1.5061e+00, 1.2219e-01, 5.0541e-02, ..., 6.6859e-02,
2.2484e-01, 1.4526e-01],
[-1.4190e+00, -4.8570e-01, -2.4108e-01, ..., 2.9871e-02,
2.0181e-01, -1.2332e-01]],
...,
[[ 1.0165e+00, -4.0671e-01, -2.0218e-01, ..., 3.4300e-01,
2.0742e-01, -7.8561e-01],
[-8.5075e-01, -9.5521e-01, -8.1730e-01, ..., -3.2288e-01,
-1.0382e+00, -9.4914e-01],
[-6.8941e-01, -9.6466e-01, -5.6186e-01, ..., -9.9886e-02,
-2.1355e-01, -6.0256e-01],
...,
[-7.8703e-01, -1.1469e+00, -7.3172e-01, ..., -1.7310e-01,
-2.5474e-01, -6.1697e-01],
[-2.7909e+00, -2.8295e+00, -1.8754e+00, ..., -1.0833e+00,
-1.1514e+00, -1.1098e+00],
[-6.6637e-01, -1.6548e+00, -9.8020e-01, ..., -1.5545e-01,
-2.6280e-01, -4.7342e-01]],
[[ 1.0972e-02, 3.2488e-03, 6.1252e-01, ..., 3.3192e-01,
8.0983e-01, 3.6754e-02],
[-5.0760e-02, -7.2791e-01, 1.2745e-01, ..., 4.0648e-01,
6.3854e-01, -2.8545e-01],
[-1.2336e-01, -8.0570e-01, 1.0266e+00, ..., 4.1196e-01,
4.3685e-01, -1.1763e-01],
...,
[ 1.8607e-02, -8.0278e-01, 3.2711e-01, ..., 4.4627e-01,
4.5971e-01, -1.6483e-01],
[-2.8506e-01, -1.4823e+00, 5.0840e-02, ..., -2.2624e-02,
9.9874e-02, -4.6039e-01],
[ 6.3753e-01, 1.2171e-01, 2.8295e-01, ..., 2.2683e-01,
3.1514e-01, 3.5566e-01]],
[[-1.6254e+00, -6.1095e-01, 7.6413e-02, ..., 3.4093e-01,
2.5824e-01, 8.3937e-01],
[-2.0829e+00, -3.6751e-01, -2.1613e-02, ..., -4.1917e-01,
-1.3648e-01, 1.1874e+00],
[-1.7208e+00, -3.7639e-01, 1.7608e-01, ..., -9.6068e-02,
5.2831e-03, 9.5399e-01],
...,
[-1.7357e+00, -5.5040e-01, 1.6899e-01, ..., -1.0033e-01,
5.2628e-02, 9.7474e-01],
[-2.1768e+00, -1.1996e+00, -2.2526e-01, ..., -2.8372e-01,
-1.3402e-01, 6.3675e-01],
[-1.0439e+00, -2.1691e-01, 4.4275e-01, ..., -2.4034e-01,
-2.4567e-01, 7.2352e-01]]],
[[[-7.5461e-01, -1.5168e+00, -1.5318e+00, ..., -1.2861e+00,
-8.9799e-01, -1.7818e+00],
[-4.1035e-01, -2.5141e+00, -2.1742e+00, ..., -3.3241e+00,
-2.6540e+00, -2.0069e+00],
[-2.1196e-01, -1.5952e+00, -2.4207e+00, ..., -1.8390e+00,
-1.7847e+00, -2.2532e+00],
...,
[-1.5848e-02, -7.4687e-01, 2.5513e-02, ..., -1.3334e+00,
4.9094e-01, -1.2770e-02],
[-6.6426e-01, -7.8035e-01, 7.9301e-02, ..., 5.3534e-01,
7.7338e-01, 6.1684e-01],
[-8.5424e-01, -1.5201e+00, -1.2546e+00, ..., -3.3521e+00,
-7.4652e-01, -5.0326e-01]],
[[-2.8041e+00, -5.2925e+00, -4.7379e+00, ..., -4.7760e+00,
-6.0615e+00, -2.4064e+00],
[-5.6076e+00, -8.4159e+00, -8.6014e+00, ..., -1.1422e+01,
-1.1945e+01, -4.5438e+00],
[-4.2909e+00, -7.7529e+00, -1.5653e+01, ..., -6.1473e+00,
-8.4784e+00, -3.3310e+00],
...,
[-2.1187e+00, -2.9979e+00, -2.4724e+00, ..., -9.7928e+00,
-7.0810e+00, -2.7964e+00],
[-3.4347e+00, -3.8998e+00, -2.6072e+00, ..., -9.5655e+00,
-5.5788e+00, -2.0904e+00],
[ 2.8556e-01, -4.8283e-01, 2.8328e-01, ..., -5.9726e-01,
-9.4685e-01, -5.3388e-01]],
[[-6.9328e-01, 4.8295e-01, 8.8396e-01, ..., 8.3481e-01,
6.1520e-01, 1.9754e+00],
[-9.4668e-01, 9.6304e-02, 1.8611e-01, ..., 1.5516e+00,
4.5009e-01, 9.5061e-02],
[-1.8167e+00, -5.4595e-01, -2.4409e+00, ..., -8.0797e-02,
1.6701e+00, -1.1753e-01],
...,
[-8.5163e-01, 2.5134e-01, 2.0447e-01, ..., -5.0660e+00,
1.2172e+00, 1.4695e+00],
[-5.7360e-01, 9.3928e-03, 2.9917e-01, ..., -1.0173e+01,
-2.7363e+00, 5.9613e-01],
[-1.5997e-01, -2.2852e-01, 9.4409e-02, ..., -3.8510e+00,
-1.9650e+00, 4.8181e-01]],
...,
[[ 1.3303e+00, -3.4373e-01, -4.7900e-02, ..., 2.6663e-01,
-8.7618e-02, -2.2232e+00],
[-7.6467e-01, -9.8810e-01, -6.5998e-01, ..., -1.0267e+00,
-1.5586e+00, -2.2073e+00],
[-7.3124e-01, -1.0362e+00, 5.8287e-01, ..., -1.9647e+00,
-1.7155e+00, -1.5295e+00],
...,
[-7.8682e-01, -6.0663e-01, -3.5970e-01, ..., -5.5111e+00,
-3.5175e+00, -1.0756e+00],
[-2.6034e+00, -2.0611e+00, -1.3016e+00, ..., -1.1682e+01,
-8.7721e+00, -3.7023e+00],
[-7.2065e-01, -1.0965e+00, -4.0788e-01, ..., -7.1371e+00,
-5.3225e+00, -2.7314e+00]],
[[ 5.1694e-02, 1.4595e-02, 5.4208e-01, ..., 6.1164e-01,
1.0421e+00, 7.6211e-02],
[ 2.4280e-01, -8.3138e-01, -4.9609e-02, ..., -1.2800e+00,
-1.5404e+00, -8.8706e-01],
[ 1.9610e-01, -7.1407e-01, 5.3044e-01, ..., 1.3999e+00,
-8.5108e-01, -4.3409e-01],
...,
[-3.2322e-01, -6.4464e-01, 2.9630e-01, ..., 4.7702e-01,
1.4059e+00, -7.1742e-02],
[-7.0561e-01, -1.2149e+00, -2.5052e-02, ..., -2.5054e+00,
-5.5384e-01, 1.3471e-01],
[ 4.0261e-01, -5.1163e-02, -1.7880e-01, ..., -2.4351e+00,
-1.3426e+00, 4.0305e-01]],
[[-1.7232e+00, -9.2001e-01, -5.1321e-01, ..., -5.5797e-01,
-3.5785e-01, 8.5971e-01],
[-2.3387e+00, -8.3658e-01, -6.4261e-01, ..., -2.3992e+00,
3.2914e-02, 1.7259e+00],
[-1.6939e+00, -7.9920e-01, -1.7916e+00, ..., -2.9911e+00,
3.3952e-01, 1.4183e+00],
...,
[-7.1498e-01, -4.8159e-01, -3.1360e-02, ..., -3.2632e-01,
1.5202e-01, 8.9099e-01],
[-1.3227e+00, -9.2732e-01, -4.7132e-01, ..., -3.3975e+00,
-5.9275e-01, 3.9778e-01],
[-4.6903e-01, -2.5057e-01, -3.3443e-01, ..., -3.4397e+00,
-1.1710e+00, 5.1252e-01]]],
[[[-1.5579e-01, -5.2675e-01, 2.6945e-01, ..., -4.3150e-01,
1.2784e-01, -8.4980e-02],
[ 2.4573e-01, -1.1925e+00, -4.7461e-01, ..., -7.6064e-01,
-4.3924e-01, -2.2373e-01],
[ 1.1480e-01, -8.2868e-01, -3.9832e-01, ..., -2.9859e-01,
-4.5221e-01, -5.3773e-01],
...,
[-2.1315e-01, -1.0451e+00, -2.9851e+00, ..., -9.4752e-01,
-1.0093e+00, -2.0587e+00],
[-1.1025e+00, -2.3004e+00, -1.0644e+00, ..., -5.2730e-01,
-4.8259e-01, -1.8121e+00],
[-1.2887e+00, -2.2659e+00, -1.1177e+00, ..., -1.4691e+00,
-1.5860e+00, -9.8280e-01]],
[[-1.4305e+00, -3.0645e+00, -5.1489e+00, ..., -3.6574e+00,
-3.7004e+00, -1.3203e+00],
[-3.1175e+00, -5.1583e+00, -5.8992e+00, ..., -5.1015e+00,
-5.3955e+00, -2.0832e+00],
[-2.0888e+00, -3.3798e+00, -2.5531e+00, ..., -4.7803e+00,
-3.4531e+00, -1.3004e+00],
...,
[-3.7984e+00, -8.5312e+00, -1.1151e+01, ..., -5.5204e+00,
-6.1772e+00, -2.9092e+00],
[-6.2109e+00, -7.7826e+00, -6.4137e+00, ..., -6.0534e+00,
-7.3778e+00, -2.7261e+00],
[-5.0595e-02, -1.2330e+00, 1.9230e-02, ..., 3.9811e-01,
7.2332e-01, 1.5889e-01]],
[[-4.2032e-01, 2.7246e-01, 8.3930e-01, ..., 1.1631e+00,
6.0251e-01, 1.3634e+00],
[-3.6282e-01, -1.9884e-02, 8.8408e-01, ..., 9.8020e-01,
5.6233e-01, 3.1000e-01],
[-8.3928e-01, 9.3871e-02, 9.9041e-01, ..., 6.6473e-01,
6.5201e-01, 4.5439e-01],
...,
[-1.3387e+00, -2.0803e-01, -5.2933e+00, ..., 5.1931e-02,
1.0141e-02, 2.8224e-01],
[-1.1647e+00, 1.2973e+00, -8.4052e-01, ..., 8.7746e-03,
-2.3136e-02, 2.3139e-01],
[-1.1791e+00, 8.5870e-01, 9.6060e-01, ..., 3.6478e-02,
-7.8459e-02, -4.2303e-01]],
...,
[[ 9.3950e-01, 2.5011e-01, 7.4689e-01, ..., 2.8082e-01,
1.5457e-01, -9.0136e-01],
[-1.1235e+00, -8.1926e-01, -1.3543e-02, ..., -9.8632e-01,
-1.6591e+00, -1.1107e+00],
[-5.9015e-01, -1.6179e-01, -1.4029e-01, ..., -9.6459e-01,
-6.0951e-01, -6.1721e-01],
...,
[-6.0887e-01, 2.8875e-01, 5.0499e-01, ..., -3.0996e-01,
-5.7682e-01, -1.2540e+00],
[-3.1946e+00, -2.9624e+00, -2.6066e+00, ..., -1.9046e+00,
-2.0604e+00, -2.1171e+00],
[-5.3522e-01, -2.1982e+00, -2.5247e+00, ..., -8.9083e-01,
-9.4066e-01, -1.3415e+00]],
[[ 4.7652e-02, 3.0703e-01, 1.5063e+00, ..., 3.7260e-01,
7.2409e-01, 3.7261e-01],
[ 3.2884e-02, -3.4029e-01, 1.0766e+00, ..., 5.7257e-01,
5.3370e-01, -2.1523e-01],
[-2.9344e-01, -4.4283e-01, 8.0853e-01, ..., 4.0224e-01,
3.7548e-01, 1.0172e-02],
...,
[-1.3318e-01, -1.4867e+00, -1.3392e+00, ..., 2.7043e-01,
1.5189e-01, -7.1784e-01],
[-5.1073e-01, -2.1407e+00, -1.8027e+00, ..., 4.0262e-02,
-8.3275e-03, -9.3162e-01],
[ 4.3639e-01, 8.0887e-02, -7.4742e-02, ..., 2.4578e-01,
2.8960e-01, 3.0422e-01]],
[[-9.3509e-01, -2.1304e-01, 4.2652e-03, ..., 3.2160e-01,
1.1378e-01, 7.6073e-01],
[-1.0997e+00, -2.9898e-01, -1.0954e+00, ..., -1.3637e-01,
-4.1877e-02, 1.1723e+00],
[-6.8044e-01, -2.9201e-01, -7.8590e-01, ..., 3.4724e-01,
3.0515e-02, 6.6649e-01],
...,
[-1.2192e+00, -1.8883e+00, -9.6186e-01, ..., -4.6531e-01,
-7.6029e-02, 8.5969e-01],
[-1.9004e+00, -3.2470e+00, -1.1457e+00, ..., -8.8897e-01,
-5.0646e-01, 4.3809e-01],
[-9.7130e-01, -1.4599e+00, 3.0303e-01, ..., -3.1871e-01,
-1.5171e-01, 1.1343e+00]]],
[[[-6.8556e-01, -1.4475e+00, -1.4849e+00, ..., -1.2728e+00,
-8.5824e-01, -1.6418e+00],
[-3.1881e-01, -2.4282e+00, -2.2587e+00, ..., -1.7672e+00,
-1.8938e+00, -1.8985e+00],
[-1.4074e-01, -1.9214e+00, -1.7128e+00, ..., -1.2926e+00,
-1.4005e+00, -2.1157e+00],
...,
[ 9.1168e-02, -5.2856e-01, -2.2734e-01, ..., -9.5747e-02,
-3.7976e-02, -4.1685e-01],
[-3.6356e-01, -4.4855e-01, 4.0465e-02, ..., 1.4595e-01,
3.0102e-01, -1.5103e-01],
[-6.4938e-01, -1.3291e+00, -1.1206e+00, ..., -1.0787e+00,
-1.1567e+00, -4.9467e-01]],
[[-2.7107e+00, -5.2334e+00, -4.7636e+00, ..., -4.7605e+00,
-5.9681e+00, -2.3933e+00],
[-5.4202e+00, -8.1976e+00, -7.5041e+00, ..., -7.7978e+00,
-9.3086e+00, -4.3864e+00],
[-4.0678e+00, -5.7643e+00, -5.7214e+00, ..., -6.0520e+00,
-6.4885e+00, -3.2232e+00],
...,
[-1.7084e+00, -2.1744e+00, -1.8160e+00, ..., -1.9402e+00,
-2.3403e+00, -9.7805e-01],
[-2.8025e+00, -2.9778e+00, -1.9685e+00, ..., -2.2363e+00,
-3.3193e+00, -9.9246e-01],
[ 3.3017e-01, -2.6367e-01, 3.1851e-01, ..., 2.7310e-01,
4.3607e-01, 2.0152e-01]],
[[-6.8603e-01, 4.3291e-01, 8.7309e-01, ..., 8.3203e-01,
6.1224e-01, 1.9124e+00],
[-9.4253e-01, 1.0165e-02, 4.0604e-01, ..., 5.9198e-01,
-4.0120e-03, 1.1734e-01],
[-1.7380e+00, -2.3907e-01, -1.0934e-01, ..., 1.2801e-02,
2.4193e-02, -9.6661e-02],
...,
[-6.3090e-01, 3.2432e-02, 2.4716e-01, ..., 2.1194e-01,
2.5993e-01, 4.1856e-01],
[-4.5134e-01, 6.2401e-02, 1.9910e-01, ..., 1.6656e-01,
1.9448e-01, 5.8205e-01],
[-1.5706e-01, -2.8105e-01, 1.0674e-01, ..., -8.3075e-03,
7.8491e-02, 3.6052e-01]],
...,
[[ 1.2384e+00, -3.1830e-01, -5.8583e-02, ..., 2.2009e-01,
-1.1141e-01, -2.1167e+00],
[-7.1826e-01, -8.4826e-01, -4.6343e-01, ..., -5.3979e-01,
-1.5219e+00, -2.0739e+00],
[-7.0617e-01, -8.8002e-01, -2.8169e-01, ..., -3.0457e-01,
-5.9148e-01, -1.3803e+00],
...,
[-6.2318e-01, -5.2537e-01, -3.5761e-01, ..., -3.1626e-01,
-5.0920e-01, -5.5054e-01],
[-2.2431e+00, -1.7971e+00, -1.3387e+00, ..., -1.3202e+00,
-1.3755e+00, -1.0844e+00],
[-6.1850e-01, -8.6325e-01, -4.4775e-01, ..., -2.8217e-01,
-3.8435e-01, -4.0607e-01]],
[[ 2.9111e-02, 1.6778e-02, 4.3011e-01, ..., 4.5432e-01,
9.2216e-01, -1.3217e-02],
[ 2.2052e-01, -7.2233e-01, 6.3100e-03, ..., 4.4213e-01,
3.5227e-01, -8.4850e-01],
[ 1.7340e-01, -5.5141e-01, 3.4987e-01, ..., 6.3867e-01,
5.1930e-01, -3.8678e-01],
...,
[-2.0301e-01, -4.0463e-01, 2.4983e-01, ..., 3.2249e-01,
3.0317e-01, 6.9569e-02],
[-5.3661e-01, -9.0765e-01, 8.9334e-02, ..., -1.2512e-03,
-3.3981e-02, -1.6178e-01],
[ 3.0726e-01, -3.1418e-02, -2.0173e-01, ..., -1.5157e-01,
-1.1666e-01, 1.6327e-01]],
[[-1.5702e+00, -7.8767e-01, -3.7579e-01, ..., -4.5586e-01,
-3.0198e-01, 8.6315e-01],
[-2.1802e+00, -7.6828e-01, -5.2615e-01, ..., -7.8466e-01,
-2.1544e-01, 1.6525e+00],
[-1.5553e+00, -5.7156e-01, -2.2582e-01, ..., -3.3846e-01,
-1.1611e-01, 1.4081e+00],
...,
[-6.5980e-01, -2.3852e-01, -1.2525e-01, ..., -1.7812e-01,
-2.1115e-02, 6.1748e-01],
[-1.1830e+00, -6.1465e-01, -3.1214e-01, ..., -3.6794e-01,
-1.8950e-01, 1.6615e-01],
[-3.8922e-01, -2.9478e-02, -1.5222e-01, ..., -2.3707e-01,
-2.2028e-01, 3.3329e-01]]]], device='cuda:0'),
tensor([[[[-7.3457e-01, -1.4297e+00, -1.4316e+00, ..., -1.3232e+00,
-9.1069e-01, -1.6573e+00],
[-3.5156e-01, -2.4561e+00, -2.2807e+00, ..., -1.7592e+00,
-1.8883e+00, -1.9217e+00],
[-1.3869e-01, -1.9966e+00, -1.7412e+00, ..., -1.2871e+00,
-1.4223e+00, -2.1310e+00],
...,
[ 6.6455e-02, -5.4200e-01, -2.3184e-01, ..., -6.8624e-01,
-5.6291e-02, -4.3452e-01],
[-3.3432e-01, -4.4894e-01, 3.0786e-02, ..., -4.6889e-01,
2.6265e-01, -1.6071e-01],
[-6.4061e-01, -1.3424e+00, -1.1249e+00, ..., -1.7185e+00,
-1.1889e+00, -5.1244e-01]],
[[-2.6887e+00, -5.1887e+00, -4.7258e+00, ..., -4.9807e+00,
-6.0598e+00, -2.4268e+00],
[-5.3593e+00, -8.1007e+00, -7.4197e+00, ..., -7.8760e+00,
-9.3661e+00, -4.4619e+00],
[-4.1099e+00, -5.8333e+00, -5.7237e+00, ..., -5.9673e+00,
-6.4766e+00, -3.2575e+00],
...,
[-1.7183e+00, -2.1444e+00, -1.7355e+00, ..., -3.4831e+00,
-2.2761e+00, -9.1624e-01],
[-2.7851e+00, -2.9527e+00, -1.9683e+00, ..., -3.1929e+00,
-3.4131e+00, -1.0196e+00],
[ 3.3717e-01, -2.4714e-01, 3.2933e-01, ..., -1.7715e-01,
4.2783e-01, 2.0189e-01]],
[[-6.8998e-01, 3.9783e-01, 8.1731e-01, ..., 8.4205e-01,
6.3028e-01, 1.9268e+00],
[-9.5548e-01, 6.3619e-02, 3.2571e-01, ..., 5.7379e-01,
4.8916e-02, 1.3497e-01],
[-1.7346e+00, -1.6093e-01, -1.0411e-01, ..., -1.7339e-02,
1.6356e-03, -8.8510e-02],
...,
[-6.0140e-01, 2.6246e-02, 1.9663e-01, ..., 1.9436e-01,
2.5725e-01, 4.1872e-01],
[-4.3143e-01, 1.0397e-01, 1.9081e-01, ..., 5.4471e-01,
2.1510e-01, 5.7343e-01],
[-1.5535e-01, -2.4010e-01, 1.2150e-01, ..., 3.0668e-01,
8.0631e-02, 3.4744e-01]],
...,
[[ 1.1955e+00, -2.9463e-01, -2.4898e-03, ..., 2.0544e-01,
-1.2337e-01, -2.1261e+00],
[-7.4747e-01, -9.1898e-01, -5.0637e-01, ..., -5.3684e-01,
-1.5362e+00, -2.0856e+00],
[-7.1110e-01, -9.2732e-01, -3.3127e-01, ..., -3.2306e-01,
-6.1877e-01, -1.3834e+00],
...,
[-6.9081e-01, -5.1759e-01, -3.3292e-01, ..., -1.1010e+00,
-5.2185e-01, -5.6156e-01],
[-2.2368e+00, -1.7728e+00, -1.3140e+00, ..., -2.3045e+00,
-1.3636e+00, -1.1132e+00],
[-5.9372e-01, -8.1927e-01, -4.5995e-01, ..., -6.9766e-01,
-3.4903e-01, -3.9078e-01]],
[[ 2.9021e-02, 2.4696e-02, 4.6635e-01, ..., 4.1507e-01,
9.0750e-01, -1.8177e-02],
[ 2.4291e-01, -7.9908e-01, -2.7829e-02, ..., 4.5263e-01,
4.0030e-01, -8.7491e-01],
[ 1.4998e-01, -6.5501e-01, 2.5804e-01, ..., 6.5100e-01,
5.5131e-01, -4.0181e-01],
...,
[-2.3085e-01, -4.1116e-01, 2.3430e-01, ..., 5.1536e-01,
3.1983e-01, 4.1612e-02],
[-5.1694e-01, -9.2189e-01, 5.0173e-02, ..., 1.2749e-01,
-4.7681e-02, -1.7107e-01],
[ 3.3817e-01, -4.9414e-02, -2.1419e-01, ..., -3.2216e-01,
-1.0951e-01, 1.7560e-01]],
[[-1.5369e+00, -8.4522e-01, -4.3212e-01, ..., -4.1505e-01,
-2.8974e-01, 8.6820e-01],
[-2.1321e+00, -8.5872e-01, -5.8231e-01, ..., -7.6102e-01,
-1.6828e-01, 1.6877e+00],
[-1.5119e+00, -6.3811e-01, -2.5582e-01, ..., -4.1212e-01,
-1.0677e-01, 1.4203e+00],
...,
[-7.0268e-01, -1.9232e-01, -1.1382e-01, ..., 5.7848e-01,
7.9297e-03, 6.0164e-01],
[-1.1227e+00, -5.5708e-01, -3.1875e-01, ..., 6.0058e-01,
-2.0253e-01, 1.5165e-01],
[-3.6180e-01, -3.4619e-02, -1.7431e-01, ..., 5.4167e-01,
-2.2056e-01, 3.2238e-01]]],
[[[-1.3536e-01, -5.1308e-01, -2.0699e-01, ..., 1.1410e-01,
2.8922e-01, -7.4343e-02],
[ 2.7578e-01, -1.2566e+00, -7.4293e-01, ..., -6.6810e-01,
-2.7175e-01, -1.8572e-01],
[ 1.3277e-01, -9.2617e-01, -3.9583e-01, ..., -5.1645e-01,
-1.9612e-01, -5.1682e-01],
...,
[-2.0011e-01, -1.5621e+00, -8.5736e-01, ..., -9.3468e-01,
-9.5949e-01, -2.0587e+00],
[-1.0906e+00, -1.6417e+00, -5.7333e-01, ..., -4.9450e-01,
-4.4818e-01, -1.7913e+00],
[-1.2706e+00, -2.1250e+00, -1.5841e+00, ..., -1.4212e+00,
-1.5429e+00, -9.7597e-01]],
[[-1.4084e+00, -3.0668e+00, -2.5294e+00, ..., -3.2141e+00,
-3.3869e+00, -1.3484e+00],
[-3.1597e+00, -5.0700e+00, -3.6073e+00, ..., -4.5400e+00,
-4.7285e+00, -2.0875e+00],
[-2.0628e+00, -3.1093e+00, -2.2918e+00, ..., -2.7179e+00,
-2.7056e+00, -1.2589e+00],
...,
[-3.7704e+00, -5.5784e+00, -5.5183e+00, ..., -5.4370e+00,
-6.1957e+00, -2.9138e+00],
[-6.1214e+00, -6.9657e+00, -6.1316e+00, ..., -6.1155e+00,
-7.4112e+00, -2.7118e+00],
[-1.9356e-02, -8.7717e-01, 2.0452e-01, ..., 4.0747e-01,
7.1432e-01, 2.1383e-01]],
[[-4.5080e-01, 2.1477e-01, 8.5963e-01, ..., 1.1337e+00,
5.6917e-01, 1.3343e+00],
[-3.7242e-01, -1.0457e-01, 5.8695e-01, ..., 8.4314e-01,
4.4803e-01, 3.0882e-01],
[-7.9679e-01, 5.6056e-02, 3.0784e-01, ..., 3.7364e-01,
3.7937e-01, 4.5706e-01],
...,
[-1.3766e+00, 2.9214e-01, -5.4345e-02, ..., 7.6351e-02,
1.8224e-02, 2.7931e-01],
[-1.2022e+00, 1.6480e-01, 2.8286e-01, ..., 1.3144e-02,
2.1019e-02, 3.2178e-01],
[-1.1542e+00, -1.1054e-01, 1.8926e-01, ..., 1.1252e-02,
-5.4963e-02, -3.0777e-01]],
...,
[[ 8.9790e-01, 2.0494e-01, 2.2767e-01, ..., -2.5227e-01,
1.5452e-01, -8.8927e-01],
[-1.1775e+00, -8.6762e-01, -6.8550e-01, ..., -1.4585e+00,
-1.4916e+00, -1.0906e+00],
[-5.7170e-01, -3.0466e-01, -2.9109e-01, ..., -5.0392e-01,
-4.9875e-01, -5.9330e-01],
...,
[-5.3688e-01, -7.4981e-01, -2.5725e-01, ..., -3.3618e-01,
-5.5780e-01, -1.1662e+00],
[-3.0905e+00, -3.0235e+00, -2.2548e+00, ..., -1.9039e+00,
-2.0087e+00, -2.0535e+00],
[-5.1390e-01, -1.6324e+00, -1.2099e+00, ..., -8.4765e-01,
-9.2486e-01, -1.3498e+00]],
[[ 6.1942e-02, 2.9795e-01, 7.1104e-01, ..., 9.8730e-01,
9.6033e-01, 3.3839e-01],
[-2.0374e-02, -4.0707e-01, 1.3731e-01, ..., 3.9755e-01,
5.6619e-01, -2.1214e-01],
[-2.9063e-01, -5.5307e-01, 1.7693e-01, ..., 1.6783e-01,
3.3474e-01, 7.6994e-03],
...,
[-3.8521e-02, -8.3550e-01, 1.2271e-01, ..., 2.5391e-01,
1.4411e-01, -7.1343e-01],
[-4.3842e-01, -1.7543e+00, -3.6950e-02, ..., 3.4122e-02,
-2.1549e-02, -9.2896e-01],
[ 4.8302e-01, -1.1189e-01, 1.9765e-01, ..., 2.5437e-01,
3.3864e-01, 3.5654e-01]],
[[-9.8315e-01, -2.2875e-01, 1.6194e-01, ..., 1.0864e-01,
5.4512e-02, 7.2361e-01],
[-1.1099e+00, -1.8393e-01, -3.0440e-01, ..., 6.4356e-02,
-9.7004e-02, 1.1360e+00],
[-7.2687e-01, -1.4862e-01, -1.8185e-01, ..., 1.2832e-01,
-1.1040e-01, 6.6500e-01],
...,
[-1.1186e+00, -1.0631e+00, -6.7094e-01, ..., -4.3236e-01,
-8.6039e-02, 8.1523e-01],
[-1.9220e+00, -1.8398e+00, -1.0847e+00, ..., -8.5974e-01,
-4.8603e-01, 4.2164e-01],
[-9.8276e-01, -9.7673e-01, -2.8920e-01, ..., -2.9672e-01,
-1.4789e-01, 1.1376e+00]]],
[[[ 2.0078e-02, -2.6881e-01, -4.3743e-02, ..., -9.1660e-01,
-5.3338e-01, -1.4517e+00],
[ 3.1314e-01, -9.5879e-01, -7.2907e-01, ..., -1.1146e+00,
-1.3189e+00, -1.4874e+00],
[ 2.4483e-01, -5.7525e-01, -5.3579e-01, ..., -8.8203e-01,
-1.0852e+00, -1.8143e+00],
...,
[-2.1066e-02, -6.0557e-01, -3.4334e-01, ..., -1.1378e+00,
-1.0212e+00, -1.9122e+00],
[-5.1690e-01, -6.0930e-01, -5.9914e-02, ..., -4.2900e-01,
-5.5460e-01, -1.6871e+00],
[-7.0460e-01, -1.2941e+00, -1.1966e+00, ..., -1.1384e+00,
-1.5454e+00, -7.6340e-01]],
[[-7.2430e-01, -2.1516e+00, -1.8697e+00, ..., -3.9717e+00,
-5.4714e+00, -1.8741e+00],
[-1.9958e+00, -3.8204e+00, -2.9863e+00, ..., -5.9063e+00,
-8.0871e+00, -3.4395e+00],
[-1.6161e+00, -2.6584e+00, -2.6126e+00, ..., -5.0338e+00,
-6.1968e+00, -2.5798e+00],
...,
[-1.8434e+00, -2.4604e+00, -2.6635e+00, ..., -5.0724e+00,
-6.3326e+00, -2.4771e+00],
[-3.8229e+00, -4.3790e+00, -3.6640e+00, ..., -6.4193e+00,
-7.8926e+00, -2.5221e+00],
[ 3.2658e-01, -2.3080e-01, 3.5846e-01, ..., 4.3186e-01,
4.7145e-01, 2.3596e-01]],
[[-2.2234e-01, 1.3588e-01, 7.1880e-01, ..., 6.2387e-01,
2.8197e-01, 1.5011e+00],
[ 9.5648e-02, -3.5706e-01, 5.2424e-01, ..., 5.6780e-01,
-3.7780e-02, 1.0796e-01],
[-1.1471e-01, -2.0093e-01, 2.1150e-01, ..., 1.4316e-02,
-1.4422e-02, -1.3206e-01],
...,
[-2.2988e-01, -1.9613e-01, 4.4240e-01, ..., 5.8106e-02,
-1.2780e-01, -1.6047e-01],
[-2.6144e-01, -1.6921e-01, 1.6702e-01, ..., -8.4660e-02,
-5.1394e-03, -1.7474e-01],
[-1.2105e-01, -4.9061e-01, 1.0179e-01, ..., 7.8760e-02,
2.1710e-01, -5.3853e-01]],
...,
[[ 5.7338e-01, -2.6279e-01, -1.5857e-02, ..., 4.7097e-01,
3.2852e-01, -1.3154e+00],
[-5.2056e-01, -5.3766e-01, -5.1967e-01, ..., -1.5586e-01,
-7.8048e-01, -1.4019e+00],
[-3.5147e-01, -2.0645e-01, -2.3081e-01, ..., -9.7344e-02,
-8.1537e-02, -9.6575e-01],
...,
[-4.1718e-01, -1.8004e-01, -3.6558e-01, ..., -9.7142e-04,
-1.2800e-01, -9.4053e-01],
[-2.2681e+00, -1.7166e+00, -1.2742e+00, ..., -1.5285e+00,
-1.5789e+00, -1.8238e+00],
[-4.5427e-01, -7.5917e-01, -4.1502e-01, ..., -3.5634e-01,
-5.3440e-01, -1.1940e+00]],
[[ 1.5189e-01, 2.6637e-01, 6.5075e-01, ..., -5.3540e-02,
3.9445e-01, -4.0477e-01],
[ 1.3961e-01, 2.4140e-01, 4.6811e-01, ..., 2.7327e-01,
1.0249e-01, -8.8103e-01],
[-1.0400e-01, 8.3867e-02, 2.9608e-01, ..., 3.4840e-01,
1.8242e-01, -4.9128e-01],
...,
[-1.8968e-01, 2.4229e-02, 4.2073e-01, ..., 3.5801e-02,
1.1514e-01, -5.1780e-01],
[-5.1927e-01, -7.1910e-01, -5.1231e-02, ..., -1.3748e-01,
-2.9792e-01, -9.1528e-01],
[ 5.4638e-01, 3.8736e-01, 6.1955e-02, ..., 3.2920e-01,
3.2025e-01, 2.9932e-01]],
[[-8.4707e-01, 2.3681e-01, 3.2515e-01, ..., -6.4263e-02,
-3.5409e-02, 5.8564e-01],
[-1.1702e+00, 2.1712e-01, -3.3729e-01, ..., -9.0653e-01,
-3.7469e-01, 8.9483e-01],
[-9.5245e-01, 3.4649e-01, -1.3603e-01, ..., -5.7679e-01,
-5.0189e-01, 6.2453e-01],
...,
[-9.6369e-01, 2.0206e-01, -3.0928e-01, ..., -2.6132e-01,
-5.3829e-01, 5.9739e-01],
[-1.6303e+00, -3.4544e-01, -3.3475e-01, ..., -7.0221e-01,
-6.9883e-01, 3.9834e-01],
[-7.5490e-01, -5.0444e-02, -2.7234e-01, ..., -4.9247e-01,
-4.5957e-01, 9.9360e-01]]],
[[[-7.1320e-01, -1.3759e+00, -1.4465e+00, ..., -1.2909e-01,
6.0354e-01, 5.2090e-01],
[-3.2791e-01, -2.3089e+00, -2.2307e+00, ..., -1.0054e+00,
-2.5088e-01, 9.7794e-02],
[-4.7095e-01, -2.0342e+00, -1.6823e+00, ..., -2.4901e-01,
4.6670e-01, -5.4181e-02],
...,
[-5.0559e-01, -2.1557e+00, -1.5485e+00, ..., -3.8094e-01,
-1.1186e-01, -1.2727e-01],
[-1.2465e+00, -1.9738e+00, -1.1430e+00, ..., 1.6880e-02,
2.6161e-01, -3.4882e-02],
[-1.0988e+00, -2.0373e+00, -1.4466e+00, ..., -1.0517e+00,
-1.0682e+00, -3.7365e-01]],
[[-2.4924e+00, -5.0047e+00, -4.4762e+00, ..., -7.4337e-01,
-1.2013e+00, 9.9791e-04],
[-5.3182e+00, -7.8128e+00, -6.6198e+00, ..., -4.5728e+00,
-4.3033e+00, -8.1025e-01],
[-4.6043e+00, -6.4936e+00, -8.2063e+00, ..., -2.8824e+00,
-2.7863e+00, -5.0548e-01],
...,
[-4.4747e+00, -6.3659e+00, -5.7096e+00, ..., -1.3747e+00,
-1.6330e+00, -3.7398e-01],
[-6.7039e+00, -7.9701e+00, -6.7720e+00, ..., -2.5161e+00,
-3.3749e+00, -9.8655e-01],
[-4.1003e-02, -1.0761e+00, 3.5516e-01, ..., 3.4811e-01,
3.5432e-01, 7.9869e-02]],
[[-5.8569e-01, 7.1496e-01, 1.0981e+00, ..., 9.6893e-01,
3.6822e-01, 1.1035e+00],
[-9.6935e-01, -2.8915e-03, 2.2832e-01, ..., 1.4323e+00,
7.9972e-01, 5.2213e-01],
[-1.6703e+00, -2.0832e-01, -4.3671e-01, ..., -1.6711e+00,
8.7644e-01, 4.8119e-01],
...,
[-1.6313e+00, -2.0945e-01, 7.1986e-01, ..., 2.0545e-02,
9.3555e-02, 4.3637e-01],
[-1.4971e+00, -1.2248e-02, 3.2358e-02, ..., 1.0869e-01,
1.3432e-01, 6.0828e-01],
[-1.4791e+00, -6.5928e-01, -3.0301e-01, ..., -1.1793e-01,
8.8197e-02, 3.6508e-01]],
...,
[[ 1.2330e+00, -3.1556e-01, -2.0423e-01, ..., -1.5432e-01,
-3.8894e-01, -9.2354e-01],
[-6.9606e-01, -9.9487e-01, -8.1580e-01, ..., -1.1902e+00,
-1.2394e+00, -8.4177e-01],
[-6.1582e-01, -1.1147e+00, -7.0985e-01, ..., -2.8328e+00,
-1.5017e+00, -4.1046e-01],
...,
[-1.0171e+00, -1.2132e+00, -1.2037e+00, ..., -1.1918e-01,
-1.7890e-01, -3.6548e-01],
[-3.2008e+00, -3.0438e+00, -2.0129e+00, ..., -8.1737e-01,
-8.4704e-01, -7.8577e-01],
[-6.5014e-01, -1.6973e+00, -1.0508e+00, ..., -1.4947e-02,
-1.5862e-01, -3.3054e-01]],
[[ 1.3234e-01, 1.0736e-01, 6.0730e-01, ..., 2.9785e-01,
6.4168e-01, 2.9990e-01],
[ 9.8561e-02, -6.8738e-01, 9.9303e-02, ..., -1.3431e-01,
-3.3672e-01, -1.4092e-01],
[-7.6891e-03, -7.2508e-01, 5.9553e-01, ..., 7.1307e-01,
3.0548e-01, 2.1349e-01],
...,
[ 7.2001e-02, -9.7853e-01, 2.6711e-01, ..., -1.7591e-01,
-1.3231e-01, 1.6730e-01],
[-3.3606e-01, -1.5970e+00, -3.3790e-02, ..., -4.2317e-01,
-3.3849e-01, -5.1512e-02],
[ 6.0557e-01, 1.0546e-01, 2.3305e-01, ..., -2.6452e-01,
-2.3045e-01, 3.3214e-01]],
[[-1.6032e+00, -5.8077e-01, 1.2017e-02, ..., 5.0691e-01,
3.3510e-01, 5.7125e-01],
[-2.2899e+00, -3.8140e-01, 1.7980e-03, ..., -3.2835e-01,
4.9235e-01, 4.4859e-01],
[-1.8958e+00, -4.3881e-01, -4.8213e-02, ..., -1.8041e+00,
5.5123e-01, 1.0621e-01],
...,
[-1.8000e+00, -2.9693e-01, 1.5209e-01, ..., 9.1202e-02,
-9.6633e-02, 1.0799e-01],
[-2.3988e+00, -1.3042e+00, -2.3785e-01, ..., -9.7705e-02,
-2.7209e-01, -1.8278e-01],
[-1.1537e+00, -4.1649e-01, 3.1213e-01, ..., -1.2495e-01,
-2.8233e-01, 1.4576e-01]]],
[[[-2.0720e-01, -2.7600e-01, 1.8749e-01, ..., 5.6711e-02,
4.7570e-01, 1.4856e-01],
[-1.1521e-01, -1.5513e+00, -9.1717e-01, ..., -7.4091e-01,
-4.7543e-01, -3.1168e-01],
[-1.5611e-01, -1.3662e+00, -6.8007e-01, ..., -5.0974e-01,
-4.2118e-01, -6.6114e-01],
...,
[-1.7562e-01, -1.8961e+00, -1.3875e+00, ..., -6.9224e-01,
-1.1030e+00, -1.9705e+00],
[-4.2176e-01, -1.4775e+00, -6.1529e-01, ..., -5.6398e-01,
-3.7956e-01, -1.4213e+00],
[-1.7541e-01, -1.3211e+00, -1.1457e+00, ..., -8.2085e-01,
-1.1843e+00, -8.0262e-01]],
[[-1.5352e-01, -1.3647e+00, -5.2756e-01, ..., -5.8617e-01,
-1.3884e+00, -2.5427e-01],
[-2.0850e+00, -4.0657e+00, -2.4647e+00, ..., -2.3291e+00,
-3.3190e+00, -1.1424e+00],
[-1.8652e+00, -3.4342e+00, -2.5702e+00, ..., -2.5473e+00,
-2.9953e+00, -1.2093e+00],
...,
[-3.6773e+00, -5.1364e+00, -5.1954e+00, ..., -6.6313e+00,
-6.1181e+00, -2.9213e+00],
[-5.4305e+00, -6.5652e+00, -5.9296e+00, ..., -6.7366e+00,
-6.7476e+00, -2.7175e+00],
[ 5.6600e-01, -4.9934e-01, 5.1694e-01, ..., -1.1092e-01,
6.3782e-01, 1.1089e-01]],
[[-5.1154e-01, -1.7227e-01, 3.4646e-01, ..., 3.3139e-01,
6.8737e-02, 8.8934e-01],
[-3.7242e-01, -4.5889e-01, 4.5669e-01, ..., 3.1616e-01,
2.3820e-01, 4.7179e-01],
[-8.7792e-01, -2.5117e-01, 2.0209e-01, ..., 1.0983e-02,
1.1108e-01, 3.6784e-01],
...,
[-1.7234e+00, -4.2398e-01, 3.0990e-01, ..., 2.1011e-01,
5.9839e-03, -1.5851e-01],
[-1.4134e+00, -3.1848e-01, 3.1814e-01, ..., -3.3055e-01,
-1.4270e-01, -2.7805e-01],
[-6.2993e-01, -7.5362e-02, 4.4570e-02, ..., -6.5243e-02,
-9.8849e-02, -4.8782e-01]],
...,
[[ 6.4885e-02, -6.0581e-01, -4.0493e-01, ..., -8.2129e-02,
-2.3853e-01, -7.2014e-01],
[-8.5326e-01, -6.2268e-01, -5.5568e-01, ..., -2.5321e-01,
-9.5711e-01, -7.3357e-01],
[-5.0554e-01, -9.3248e-02, -1.5557e-03, ..., 1.0033e-01,
-1.7930e-01, -5.6047e-01],
...,
[-9.9864e-01, -8.6603e-01, -1.2730e-01, ..., -1.2999e+00,
-6.2970e-01, -1.2516e+00],
[-2.5911e+00, -2.0182e+00, -1.6419e+00, ..., -2.4346e+00,
-1.5771e+00, -1.5862e+00],
[-5.8450e-01, -1.1426e+00, -2.1248e-01, ..., -1.5876e+00,
-4.5317e-01, -4.8384e-01]],
[[-1.8211e-01, 3.9014e-02, 4.4311e-01, ..., 1.1417e-01,
5.1004e-01, 1.4447e-01],
[-4.3496e-01, -2.6646e-01, -3.7353e-02, ..., -3.3933e-02,
1.1112e-01, -7.0572e-02],
[-6.9221e-01, -5.7088e-01, -5.1578e-03, ..., 5.0933e-02,
5.0290e-02, -2.0287e-02],
...,
[-1.7268e-01, -8.4979e-01, 2.7059e-01, ..., 6.6894e-01,
5.5596e-01, -3.5457e-01],
[-7.6751e-01, -1.8974e+00, 5.4353e-03, ..., 1.9673e-01,
1.7676e-01, -5.3229e-01],
[ 7.3166e-01, 4.8627e-02, 3.5858e-01, ..., 1.1535e+00,
5.1578e-01, 2.8212e-01]],
[[-6.9518e-01, 4.2491e-01, 5.6394e-01, ..., 4.7247e-01,
3.9839e-01, 6.0559e-01],
[-8.8568e-01, 2.8031e-01, -2.4892e-01, ..., -2.6532e-01,
-1.3872e-01, 6.5395e-01],
[-5.1561e-01, 3.1985e-01, -6.3203e-03, ..., 6.9961e-02,
6.0689e-02, 5.7583e-01],
...,
[-1.3078e+00, -6.6482e-01, -6.4856e-01, ..., 1.0779e-01,
8.2857e-03, 1.4178e+00],
[-1.4062e+00, -8.2138e-01, -3.9681e-01, ..., 3.5786e-01,
9.2904e-02, 1.2058e+00],
[-6.2393e-01, -4.4160e-01, 1.4297e-01, ..., 5.2471e-01,
2.1424e-01, 1.2590e+00]]]], device='cuda:0'),
tensor([[[[-7.6733e-01, -1.3950e+00, -1.3906e+00, ..., -2.3015e-01,
2.5123e-01, -3.9337e-01],
[-3.2642e-01, -2.3146e+00, -2.1708e+00, ..., -5.7623e-01,
-3.8377e-01, -5.6953e-01],
[-3.7559e-01, -1.9703e+00, -1.7774e+00, ..., -3.5251e-01,
-3.4434e-01, -1.0413e+00],
...,
[-4.7134e-01, -2.0648e+00, -1.8852e+00, ..., -1.7260e-01,
-8.2336e-02, -1.2136e+00],
[-1.2154e+00, -1.8605e+00, -1.0665e+00, ..., 6.3317e-03,
7.1949e-02, -9.7217e-01],
[-1.1042e+00, -1.9919e+00, -1.4561e+00, ..., -9.0216e-01,
-1.1265e+00, -6.4880e-01]],
[[-2.4796e+00, -4.9374e+00, -4.4047e+00, ..., -2.1972e+00,
-3.5752e+00, -1.3081e+00],
[-5.2398e+00, -7.8949e+00, -6.6349e+00, ..., -3.3806e+00,
-5.0819e+00, -2.0806e+00],
[-4.3407e+00, -6.1826e+00, -7.7165e+00, ..., -3.0007e+00,
-3.8249e+00, -1.7594e+00],
...,
[-4.4799e+00, -6.1932e+00, -5.5390e+00, ..., -3.0173e+00,
-3.9654e+00, -1.9240e+00],
[-6.7409e+00, -7.9552e+00, -6.7066e+00, ..., -3.9218e+00,
-5.4527e+00, -1.9860e+00],
[-3.2056e-02, -1.0551e+00, 3.8642e-01, ..., 4.3985e-01,
6.0497e-01, 1.5707e-01]],
[[-5.7780e-01, 6.7851e-01, 1.0249e+00, ..., 8.2733e-01,
5.8040e-01, 1.3292e+00],
[-9.2811e-01, 5.6732e-02, 3.2002e-01, ..., 3.9858e-01,
3.4695e-01, 2.4524e-01],
[-1.6429e+00, -5.1289e-02, -2.9869e-01, ..., 7.7921e-02,
2.8741e-01, 7.8926e-02],
...,
[-1.7348e+00, -2.3115e-01, -1.3310e-01, ..., 2.5997e-01,
1.5247e-01, -1.2754e-01],
[-1.4873e+00, -5.2169e-02, 1.8496e-02, ..., 5.5861e-02,
2.3336e-01, 1.6354e-01],
[-1.3820e+00, -6.9375e-01, -3.0814e-01, ..., -2.6834e-02,
1.7762e-01, -1.1439e-01]],
...,
[[ 1.1827e+00, -3.5806e-01, -2.2755e-01, ..., 3.3911e-01,
2.2568e-01, -7.6693e-01],
[-6.7587e-01, -9.8168e-01, -7.8526e-01, ..., -3.1429e-01,
-9.8600e-01, -9.2973e-01],
[-6.1948e-01, -1.0280e+00, -5.6439e-01, ..., -1.1765e-01,
-2.1850e-01, -6.0372e-01],
...,
[-8.5185e-01, -1.1318e+00, -7.6086e-01, ..., -3.6423e-01,
-5.1356e-01, -7.1007e-01],
[-3.1121e+00, -3.0143e+00, -2.0355e+00, ..., -1.0867e+00,
-1.1497e+00, -1.1277e+00],
[-7.0227e-01, -1.7292e+00, -1.0385e+00, ..., -1.6527e-01,
-2.6081e-01, -5.0159e-01]],
[[ 1.0134e-01, 6.9515e-02, 5.9300e-01, ..., 3.2963e-01,
8.0645e-01, 1.3668e-02],
[ 1.1277e-01, -7.0451e-01, 1.3990e-01, ..., 4.3366e-01,
6.7904e-01, -3.0491e-01],
[ 3.7587e-02, -6.9055e-01, 6.3259e-01, ..., 4.2088e-01,
4.6720e-01, -1.2045e-01],
...,
[-7.0602e-02, -8.8461e-01, 2.5386e-01, ..., 5.4296e-01,
4.9890e-01, -3.6081e-01],
[-4.5531e-01, -1.6465e+00, -3.0002e-02, ..., 6.7965e-02,
8.7950e-02, -4.7984e-01],
[ 5.7858e-01, 1.2318e-01, 1.9858e-01, ..., 2.5467e-01,
2.8584e-01, 3.6055e-01]],
[[-1.5560e+00, -6.3536e-01, 1.6455e-02, ..., 3.2177e-01,
2.4427e-01, 8.0977e-01],
[-2.0845e+00, -3.6207e-01, -2.1188e-02, ..., -4.0758e-01,
-1.2128e-01, 1.1441e+00],
[-1.7544e+00, -3.6230e-01, 6.4530e-02, ..., -5.4831e-02,
1.6146e-02, 9.5467e-01],
...,
[-1.8370e+00, -4.0312e-01, 2.8520e-01, ..., 2.4513e-02,
3.6664e-01, 1.1420e+00],
[-2.4050e+00, -1.2713e+00, -2.4812e-01, ..., -2.3750e-01,
-6.3528e-02, 5.9103e-01],
[-1.1892e+00, -4.1327e-01, 3.0356e-01, ..., -2.2299e-01,
-2.2311e-01, 7.0282e-01]]],
[[[-7.2783e-01, -1.3935e+00, -1.4152e+00, ..., -7.8111e-02,
6.1351e-01, 5.8517e-01],
[-2.8510e-01, -2.3992e+00, -2.4214e+00, ..., -5.2750e-01,
-2.6128e-01, 1.3569e-01],
[-3.8812e-01, -1.9764e+00, -2.8876e+00, ..., -3.2424e-01,
-1.3606e-01, -9.4711e-02],
...,
[-4.8850e-01, -1.9434e+00, -1.8636e+00, ..., -3.3329e-01,
-1.1389e-01, -1.8419e-01],
[-9.9387e-01, -1.7179e+00, -1.1790e+00, ..., 1.6022e-02,
2.6575e-01, -3.1209e-02],
[-8.3383e-01, -1.6790e+00, -1.2521e+00, ..., -1.0406e+00,
-1.0957e+00, -3.8864e-01]],
[[-2.3958e+00, -4.8855e+00, -4.2820e+00, ..., -5.0413e-01,
-9.9630e-01, 2.3623e-02],
[-5.3410e+00, -8.3677e+00, -7.0025e+00, ..., -1.9558e+00,
-2.7380e+00, -6.9428e-01],
[-4.3856e+00, -7.4436e+00, -6.9060e+00, ..., -1.3974e+00,
-1.5641e+00, -4.1223e-01],
...,
[-4.3844e+00, -6.1693e+00, -5.5729e+00, ..., -1.3357e+00,
-1.5821e+00, -3.2951e-01],
[-6.3760e+00, -7.4355e+00, -6.2606e+00, ..., -2.4935e+00,
-3.3828e+00, -9.5265e-01],
[-3.7179e-02, -8.5794e-01, 3.9749e-01, ..., 3.4462e-01,
3.6171e-01, 8.0714e-02]],
[[-6.2628e-01, 5.8002e-01, 1.1199e+00, ..., 8.2475e-01,
2.9465e-01, 1.1170e+00],
[-9.2989e-01, -5.4304e-02, 2.2521e-01, ..., 4.7984e-01,
7.1638e-02, 4.8826e-01],
[-1.6411e+00, -8.1082e-02, -7.9608e-01, ..., 7.9687e-02,
1.3568e-01, 4.1315e-01],
...,
[-1.7387e+00, -9.2059e-02, -1.3654e-01, ..., -8.1289e-03,
-7.1040e-02, 3.6057e-01],
[-1.5125e+00, 9.3482e-02, 2.1233e-02, ..., 1.1444e-01,
6.0936e-02, 5.9874e-01],
[-1.4260e+00, -4.9280e-01, -2.5926e-01, ..., -7.3965e-02,
6.5483e-02, 3.5436e-01]],
...,
[[ 1.0452e+00, -4.2066e-01, -2.1337e-01, ..., -1.1774e-01,
-4.1199e-01, -9.4166e-01],
[-8.2052e-01, -9.1648e-01, -7.0680e-01, ..., -7.4313e-01,
-1.3269e+00, -8.4310e-01],
[-6.8351e-01, -7.3544e-01, -2.7406e-01, ..., -1.0613e-01,
-2.1795e-01, -3.9186e-01],
...,
[-7.6369e-01, -1.1346e+00, -7.3432e-01, ..., -1.0641e-01,
-1.9522e-01, -4.1806e-01],
[-2.8128e+00, -2.8188e+00, -1.7869e+00, ..., -8.4858e-01,
-8.8253e-01, -8.1817e-01],
[-6.7998e-01, -1.6900e+00, -1.0137e+00, ..., -8.1807e-03,
-1.4287e-01, -3.3985e-01]],
[[ 5.4407e-03, 2.2005e-02, 6.1845e-01, ..., 2.6347e-01,
6.9081e-01, 3.3187e-01],
[-6.0018e-02, -6.5141e-01, 2.1656e-01, ..., -5.1931e-02,
8.9248e-02, -4.9655e-02],
[-1.1838e-01, -4.8279e-01, 2.8589e-01, ..., -1.6791e-01,
-1.1214e-01, 1.8437e-01],
...,
[ 1.8923e-02, -8.0179e-01, 3.1507e-01, ..., -1.5617e-01,
-1.3601e-01, 1.2131e-01],
[-2.6715e-01, -1.4465e+00, 6.1733e-02, ..., -3.9905e-01,
-3.1302e-01, -7.5770e-02],
[ 6.4378e-01, 1.2659e-01, 2.7358e-01, ..., -2.4244e-01,
-2.2590e-01, 3.0941e-01]],
[[-1.6245e+00, -6.2226e-01, 2.4045e-02, ..., 5.6322e-01,
3.8498e-01, 5.5035e-01],
[-2.0961e+00, -3.7894e-01, -1.6703e-01, ..., -1.4108e-02,
-1.5143e-01, 4.1642e-01],
[-1.7457e+00, -5.6172e-01, -4.4961e-01, ..., 6.9974e-02,
-1.9190e-01, 9.7708e-02],
...,
[-1.7513e+00, -4.9070e-01, 2.3742e-01, ..., 7.7289e-02,
-2.8303e-02, 8.6746e-02],
[-2.2076e+00, -1.1846e+00, -2.3167e-01, ..., -1.3127e-01,
-2.4919e-01, -2.0339e-01],
[-1.0469e+00, -2.2174e-01, 4.2816e-01, ..., -1.0908e-01,
-2.7416e-01, 1.1383e-01]]],
[[[-7.6153e-01, -1.5040e+00, -1.4398e+00, ..., -1.2985e+00,
-8.4503e-01, -1.6761e+00],
[-3.5589e-01, -2.4851e+00, -2.2983e+00, ..., -1.7643e+00,
-1.9452e+00, -1.9489e+00],
[-1.2883e-01, -1.9153e+00, -1.7164e+00, ..., -1.0801e+00,
-1.4429e+00, -2.1363e+00],
...,
[-9.1032e-02, -1.1406e+00, -4.8958e-01, ..., -5.6277e-01,
-5.2727e-01, -9.4437e-01],
[-1.1029e-01, -1.0034e+00, -2.1941e-01, ..., -2.2181e-01,
-1.1582e-01, -1.0243e+00],
[-1.3983e-01, -1.1286e+00, -6.2966e-01, ..., -8.8618e-01,
-1.1186e+00, -6.2792e-01]],
[[-2.6772e+00, -5.0980e+00, -4.5643e+00, ..., -4.9385e+00,
-5.9792e+00, -2.3896e+00],
[-5.3752e+00, -8.1265e+00, -7.4894e+00, ..., -8.7096e+00,
-9.4473e+00, -4.3654e+00],
[-4.0689e+00, -5.7691e+00, -5.6902e+00, ..., -6.9759e+00,
-7.0176e+00, -3.2231e+00],
...,
[-2.0362e+00, -3.2174e+00, -3.5413e+00, ..., -2.9574e+00,
-3.4569e+00, -1.5235e+00],
[-4.5263e+00, -5.5116e+00, -5.6317e+00, ..., -5.0821e+00,
-6.3037e+00, -2.4409e+00],
[ 6.9572e-01, -3.0835e-01, 7.5571e-01, ..., 5.1153e-01,
5.8596e-01, -1.7038e-02]],
[[-6.7146e-01, 4.9503e-01, 8.6845e-01, ..., 8.7724e-01,
6.1036e-01, 1.8977e+00],
[-9.7710e-01, 3.8440e-02, 4.1753e-01, ..., 7.4177e-01,
4.9859e-03, 8.7283e-02],
[-1.7794e+00, -2.4836e-01, -5.4213e-02, ..., 3.8392e-01,
1.6221e-01, -1.5483e-01],
...,
[-9.4536e-01, -1.1449e-01, 2.5549e-02, ..., 1.7882e-02,
-6.2082e-03, 3.2779e-01],
[-9.7059e-01, 5.3860e-02, 1.1047e-01, ..., -8.0203e-03,
4.1218e-02, 3.2613e-01],
[-4.7222e-01, -5.1723e-03, 2.4534e-01, ..., -3.0972e-03,
9.3487e-02, 4.2233e-03]],
...,
[[ 1.2568e+00, -3.8473e-01, -7.1107e-02, ..., 2.4768e-01,
-9.3946e-02, -2.1267e+00],
[-7.5463e-01, -9.6003e-01, -5.2814e-01, ..., -3.4626e-01,
-1.4821e+00, -2.1064e+00],
[-7.1498e-01, -8.9804e-01, -2.8387e-01, ..., -4.7966e-01,
-5.4970e-01, -1.4094e+00],
...,
[-4.5566e-01, -1.2777e-01, 9.5693e-02, ..., 1.0005e-01,
-1.1570e-01, -5.2978e-01],
[-2.0456e+00, -1.3790e+00, -5.6736e-01, ..., -9.7918e-01,
-1.1594e+00, -1.1193e+00],
[-3.1561e-01, -7.6952e-01, 3.6583e-02, ..., -1.2439e-01,
-2.8314e-01, -6.0783e-01]],
[[ 3.9749e-02, 1.2896e-02, 4.5134e-01, ..., 3.4749e-01,
9.2849e-01, -2.2751e-02],
[ 2.2474e-01, -8.2713e-01, -7.3287e-02, ..., 4.2794e-02,
2.5655e-01, -8.3716e-01],
[ 1.3602e-01, -6.5125e-01, 2.7936e-01, ..., 4.6371e-01,
3.2646e-01, -3.6959e-01],
...,
[-4.1148e-01, -5.4851e-01, 1.4983e-01, ..., 1.0920e-01,
8.2567e-02, -8.7394e-02],
[-7.4408e-01, -1.5305e+00, -5.6923e-02, ..., -3.4686e-01,
-3.4193e-01, -6.8435e-01],
[ 5.0253e-01, 1.5736e-02, 3.7109e-01, ..., 2.0812e-01,
1.5690e-01, 3.6118e-01]],
[[-1.5753e+00, -8.0724e-01, -4.5315e-01, ..., -4.0649e-01,
-3.0058e-01, 8.9643e-01],
[-2.1796e+00, -7.6724e-01, -5.6753e-01, ..., -7.7582e-01,
-1.8822e-01, 1.6676e+00],
[-1.5052e+00, -5.5161e-01, -2.1407e-01, ..., -3.4038e-01,
-8.4136e-02, 1.4286e+00],
...,
[-4.5293e-01, 1.3690e-01, 9.0216e-02, ..., 1.1108e-01,
2.0161e-01, 7.1004e-01],
[-8.9408e-01, -3.0530e-01, -3.0956e-01, ..., -1.0156e-01,
5.7196e-02, 4.3085e-01],
[-4.1113e-01, -3.1129e-01, -3.8425e-01, ..., -2.4337e-01,
-2.5997e-01, 5.9656e-01]]],
[[[-7.6035e-01, -1.4549e+00, -1.4346e+00, ..., -2.4472e-01,
2.3136e-01, -4.4289e-01],
[-3.2713e-01, -2.4359e+00, -2.2049e+00, ..., -5.8443e-01,
-4.1662e-01, -6.3306e-01],
[-3.8929e-01, -2.1060e+00, -1.9115e+00, ..., -3.2917e-01,
-4.2267e-01, -1.1701e+00],
...,
[-4.7761e-01, -1.9522e+00, -1.3592e+00, ..., -3.1808e+00,
-1.6234e+00, -1.2348e-01],
[-1.0073e+00, -1.7482e+00, -1.1471e+00, ..., -3.5429e+00,
-2.7003e+00, -4.4912e-01],
[-8.3849e-01, -1.6823e+00, -1.2823e+00, ..., -2.7248e+00,
-3.0868e+00, -1.0252e-01]],
[[-2.4176e+00, -4.8490e+00, -4.1708e+00, ..., -2.3652e+00,
-3.6956e+00, -1.3752e+00],
[-5.4595e+00, -8.2128e+00, -6.6465e+00, ..., -3.5462e+00,
-5.1970e+00, -2.1537e+00],
[-4.4118e+00, -6.3647e+00, -5.5141e+00, ..., -3.4263e+00,
-4.3698e+00, -2.0250e+00],
...,
[-4.4459e+00, -6.2201e+00, -5.5707e+00, ..., -7.8853e+00,
-1.0184e+01, 2.0802e+00],
[-6.4843e+00, -7.5425e+00, -6.4869e+00, ..., -1.0151e+01,
-9.7774e+00, -1.5230e+00],
[-5.1861e-02, -8.4675e-01, 3.3461e-01, ..., -3.3314e-01,
-4.2092e-01, 7.2142e-02]],
[[-5.9794e-01, 5.9929e-01, 1.1094e+00, ..., 7.9233e-01,
5.5674e-01, 1.3111e+00],
[-8.7371e-01, -1.9225e-02, 4.1616e-01, ..., 3.4510e-01,
3.5318e-01, 2.1364e-01],
[-1.6397e+00, -1.2228e-01, -3.5791e-02, ..., 7.4843e-02,
2.2581e-01, 4.1148e-02],
...,
[-1.7677e+00, -2.7329e-02, 8.3977e-01, ..., -1.1870e+01,
-5.8474e+00, -1.4602e+00],
[-1.5829e+00, 8.0697e-02, 1.8562e-01, ..., -8.0962e+00,
-2.8487e+00, -4.5427e+00],
[-1.4673e+00, -5.6094e-01, -2.9231e-01, ..., -2.2268e+00,
2.3289e+00, -2.2697e+00]],
...,
[[ 1.0984e+00, -4.4777e-01, -2.4655e-01, ..., 3.8297e-01,
2.9484e-01, -7.4629e-01],
[-7.7207e-01, -1.0025e+00, -8.2369e-01, ..., -2.7225e-01,
-9.5446e-01, -9.2442e-01],
[-6.5970e-01, -1.0113e+00, -7.1894e-01, ..., -7.1172e-02,
-2.2295e-01, -6.3915e-01],
...,
[-6.9099e-01, -1.1865e+00, -1.3924e+00, ..., -4.8025e+00,
-4.8044e+00, -5.6014e+00],
[-2.8045e+00, -2.9267e+00, -2.0049e+00, ..., -4.7779e+00,
-4.5709e+00, -5.4576e+00],
[-6.7683e-01, -1.7337e+00, -1.0686e+00, ..., -2.2788e-02,
-2.2794e+00, -4.0636e+00]],
[[-1.3159e-03, -3.0771e-02, 5.9663e-01, ..., 3.2161e-01,
7.5062e-01, 1.9882e-03],
[ 1.3853e-02, -7.6754e-01, 1.0800e-01, ..., 4.2410e-01,
6.0724e-01, -3.2161e-01],
[-2.9383e-02, -8.0693e-01, 2.6049e-01, ..., 4.7330e-01,
4.8739e-01, -2.7390e-01],
...,
[ 1.1553e-01, -8.2946e-01, 5.8538e-01, ..., -5.3048e+00,
-3.9691e+00, -1.7964e+00],
[-1.9741e-01, -1.4793e+00, 5.9731e-02, ..., -3.7960e+00,
-4.6537e+00, -2.1567e+00],
[ 6.6288e-01, 1.3903e-01, 2.5500e-01, ..., -6.9807e-01,
-5.3427e-02, 1.8578e+00]],
[[-1.6493e+00, -6.5482e-01, -2.8088e-04, ..., 2.7792e-01,
2.2348e-01, 8.2853e-01],
[-2.1478e+00, -3.6118e-01, -5.0887e-02, ..., -4.0622e-01,
-1.4596e-01, 1.1319e+00],
[-1.7699e+00, -3.5859e-01, 1.8295e-01, ..., -7.0304e-02,
-3.0665e-03, 9.1740e-01],
...,
[-1.7662e+00, -4.6816e-01, 1.6053e-01, ..., -3.9174e+00,
-6.3065e+00, -5.0635e+00],
[-2.2953e+00, -1.1642e+00, -2.0099e-01, ..., -2.9255e+00,
-5.0689e+00, -1.6578e+00],
[-1.1316e+00, -2.2872e-01, 4.4650e-01, ..., -1.2240e+00,
-3.0234e+00, 5.4904e-01]]],
[[[-7.1690e-01, -1.4723e+00, -1.4587e+00, ..., -1.2861e+00,
-8.5257e-01, -1.6603e+00],
[-3.2561e-01, -2.4640e+00, -2.3048e+00, ..., -1.7799e+00,
-1.8836e+00, -1.9350e+00],
[-1.2756e-01, -1.9369e+00, -1.7705e+00, ..., -2.1564e+00,
-1.7124e+00, -2.5841e+00],
...,
[ 3.5888e-01, -2.2351e-01, -1.5658e+00, ..., 4.7017e-01,
-7.1368e-01, -3.6699e+00],
[-1.9102e+00, -4.2896e-01, -8.5826e-01, ..., 1.2601e-01,
2.5208e-01, -2.0129e-01],
[ 5.2543e-01, 2.4935e-02, -2.6322e-01, ..., -1.0354e+00,
-1.0892e+00, -4.6208e-01]],
[[-2.7032e+00, -5.1514e+00, -4.6429e+00, ..., -4.7728e+00,
-5.9969e+00, -2.3902e+00],
[-5.4445e+00, -8.2944e+00, -7.6222e+00, ..., -7.8589e+00,
-9.3544e+00, -4.4027e+00],
[-4.2057e+00, -5.9719e+00, -5.8830e+00, ..., -9.5064e+00,
-8.7044e+00, -4.3020e+00],
...,
[-4.8851e+00, -8.1158e+00, -9.0358e+00, ..., -2.0623e+00,
-2.5093e+00, -1.5485e+00],
[-3.8718e+00, -6.1628e+00, -7.5367e+00, ..., -2.1585e+00,
-3.2338e+00, -9.9154e-01],
[ 5.6458e-01, 8.2755e-01, 2.5955e+00, ..., 3.1843e-01,
4.6812e-01, 2.2123e-01]],
[[-6.8158e-01, 4.6235e-01, 8.7824e-01, ..., 8.3383e-01,
6.2673e-01, 1.9286e+00],
[-9.9210e-01, 3.5306e-02, 4.3858e-01, ..., 5.8467e-01,
1.1991e-03, 1.1766e-01],
[-1.7880e+00, -2.5929e-01, -5.5851e-02, ..., -3.5856e-01,
-1.1988e-01, -8.4869e-02],
...,
[-1.7784e+00, -6.7014e-01, -1.3813e+00, ..., -6.6135e-01,
-1.3212e+00, -8.9494e-01],
[-3.4552e+00, -4.1660e+00, -3.7329e+00, ..., 1.3661e-01,
1.8559e-01, 5.7230e-01],
[-4.4085e+00, -7.4895e+00, -7.5614e+00, ..., 4.5024e-03,
7.4428e-02, 3.4887e-01]],
...,
[[ 1.2510e+00, -3.6768e-01, -6.4226e-02, ..., 2.2674e-01,
-1.2855e-01, -2.1615e+00],
[-7.3725e-01, -8.8978e-01, -5.1058e-01, ..., -5.7325e-01,
-1.5523e+00, -2.1116e+00],
[-6.7482e-01, -8.2928e-01, -2.8703e-01, ..., -3.8518e-02,
-3.7611e-01, -1.3060e+00],
...,
[-6.8450e-01, -2.3470e+00, -2.0530e+00, ..., -3.1375e+00,
-3.5336e+00, -3.3480e+00],
[-2.7673e+00, -4.6196e+00, -5.4730e+00, ..., -1.2677e+00,
-1.3109e+00, -1.0499e+00],
[-2.2330e+00, -5.4469e+00, -5.3366e+00, ..., -2.9096e-01,
-3.4952e-01, -3.7291e-01]],
[[ 4.4620e-02, 6.4643e-04, 4.5338e-01, ..., 4.7727e-01,
9.6037e-01, -5.3625e-03],
[ 2.3624e-01, -7.7533e-01, -4.4428e-02, ..., 4.5163e-01,
3.8215e-01, -9.0528e-01],
[ 1.6863e-01, -5.9420e-01, 2.9934e-01, ..., -5.9740e-02,
1.8254e-02, -1.1700e+00],
...,
[ 2.1926e-01, 2.9674e-01, -5.5867e-01, ..., -9.0076e-01,
-2.0316e+00, -2.2389e+00],
[-1.4338e+00, -4.6612e-01, -2.1700e+00, ..., 3.8165e-02,
2.4215e-02, -1.3781e-01],
[-9.6683e-01, -1.5830e+00, -3.2572e+00, ..., -1.3298e-01,
-6.0867e-02, 1.9258e-01]],
[[-1.5843e+00, -8.0085e-01, -4.3394e-01, ..., -4.7438e-01,
-2.9491e-01, 8.6948e-01],
[-2.1810e+00, -7.3198e-01, -5.4817e-01, ..., -7.7994e-01,
-1.8036e-01, 1.6619e+00],
[-1.5649e+00, -5.3564e-01, -1.7279e-01, ..., -2.3209e-01,
-2.2130e-01, 1.7023e+00],
...,
[-3.8756e+00, -2.5283e+00, -4.1735e+00, ..., 1.0373e+00,
1.6103e+00, 1.9333e+00],
[-7.5353e+00, -7.1016e+00, -7.4175e+00, ..., -3.8064e-01,
-1.5268e-01, 2.0337e-01],
[-2.6685e+00, -3.0313e+00, -1.8298e+00, ..., -2.3141e-01,
-1.7796e-01, 3.5638e-01]]]], device='cuda:0'),
tensor([[[[-1.6904e-01, -2.9233e-01, 9.7696e-02, ..., -6.7792e-03,
4.5463e-01, 1.2590e-01],
[-9.8406e-02, -1.4923e+00, -2.3339e+00, ..., -7.6252e-01,
-4.6900e-01, -3.5954e-01],
[-8.6439e-02, -1.5869e+00, -5.3041e+00, ..., -5.3941e-01,
-4.5480e-01, -7.6643e-01],
...,
[-1.6941e-01, -1.9007e+00, -1.4206e+00, ..., -1.2367e+00,
-1.2487e+00, -2.0044e+00],
[-4.5074e-01, -1.5246e+00, -7.4374e-01, ..., -2.1325e-01,
4.4739e-01, -1.4600e+00],
[-2.5053e-01, -1.3380e+00, -1.1630e+00, ..., -2.7477e+00,
-1.0965e+00, -7.7225e-01]],
[[-2.5874e-01, -1.6145e+00, -8.8665e-01, ..., -8.0594e-01,
-1.5001e+00, -3.2085e-01],
[-2.1308e+00, -6.9944e+00, -9.2700e+00, ..., -2.5405e+00,
-3.5670e+00, -1.2413e+00],
[-2.0281e+00, -7.9064e+00, -9.8836e+00, ..., -2.7864e+00,
-3.2187e+00, -1.3747e+00],
...,
[-3.7679e+00, -5.3620e+00, -5.5418e+00, ..., -9.0437e+00,
-8.3089e+00, -2.9594e+00],
[-5.6536e+00, -6.7700e+00, -6.0396e+00, ..., -7.1823e+00,
-8.0262e+00, -2.7556e+00],
[ 5.1415e-01, -5.1727e-01, 5.4715e-01, ..., -8.3601e-01,
-2.9415e-01, 1.4964e-01]],
[[-5.3834e-01, -1.3820e-01, 4.1638e-01, ..., 4.0977e-01,
1.3908e-01, 9.3966e-01],
[-3.6364e-01, -7.0154e-01, -4.6833e-01, ..., 3.6509e-01,
2.9601e-01, 5.2296e-01],
[-9.1306e-01, -1.5474e+00, -5.8859e+00, ..., 2.6938e-02,
1.4049e-01, 3.9911e-01],
...,
[-1.8086e+00, -3.7128e-01, 2.4787e-01, ..., 1.0792e+00,
3.6017e-01, -1.7403e-01],
[-1.4997e+00, -3.3098e-01, -2.0324e-02, ..., -9.6791e-01,
3.3683e-01, -2.5945e-01],
[-6.4006e-01, -1.0507e-01, 4.7490e-02, ..., -2.9074e+00,
2.7741e-01, -4.6562e-01]],
...,
[[ 1.0359e-01, -5.6102e-01, -3.3678e-01, ..., -3.5709e-02,
-2.0998e-01, -7.1989e-01],
[-8.5644e-01, -4.2241e-01, -1.0480e-02, ..., -2.5081e-01,
-9.9849e-01, -7.6405e-01],
[-5.1288e-01, 1.6574e+00, 4.5279e+00, ..., 9.5645e-02,
-1.8581e-01, -5.3428e-01],
...,
[-1.0177e+00, -9.2942e-01, -5.6975e-01, ..., -2.5472e+00,
-8.9596e-01, -1.2462e+00],
[-2.6294e+00, -2.0355e+00, -1.4230e+00, ..., -7.6250e+00,
-2.5451e+00, -1.6082e+00],
[-5.6654e-01, -1.2046e+00, -2.1154e-01, ..., -5.0355e+00,
-2.1373e+00, -5.3271e-01]],
[[-1.5543e-01, 4.8160e-02, 4.0299e-01, ..., 9.0463e-02,
5.4082e-01, 1.6731e-01],
[-4.2823e-01, -7.3962e-02, 2.9731e-01, ..., -1.9171e-02,
1.4770e-01, -8.6542e-02],
[-6.3896e-01, -6.4285e-01, -2.8395e+00, ..., 4.3885e-02,
5.1286e-02, -8.4887e-02],
...,
[-1.4174e-01, -7.9784e-01, 4.4930e-01, ..., -2.4447e-01,
-3.1624e-01, -3.0546e-01],
[-7.6673e-01, -1.9702e+00, -7.3021e-02, ..., -1.8852e+00,
1.8565e-01, -4.7329e-01],
[ 7.5061e-01, 3.8889e-02, 3.7405e-01, ..., -8.0127e-02,
7.5833e-01, 3.3189e-01]],
[[-6.3025e-01, 4.1994e-01, 5.6557e-01, ..., 4.8682e-01,
3.9862e-01, 6.4217e-01],
[-8.6572e-01, 1.5243e-02, -1.0886e+00, ..., -2.7067e-01,
-1.2742e-01, 7.2083e-01],
[-5.0360e-01, -1.2918e+00, -5.1876e+00, ..., 4.7704e-02,
1.0816e-01, 6.5394e-01],
...,
[-1.3564e+00, -7.5309e-01, -3.6300e-01, ..., -1.1950e+00,
3.8253e-01, 1.3817e+00],
[-1.4778e+00, -9.3405e-01, -3.2168e-01, ..., 2.4094e-01,
5.5342e-01, 1.2192e+00],
[-6.9719e-01, -5.1317e-01, 1.3929e-01, ..., 1.2731e+00,
7.8116e-01, 1.3481e+00]]],
[[[-1.0036e-01, -2.1728e+00, -1.8002e+00, ..., -5.5963e-01,
-7.0414e-02, -7.5743e-01],
[ 3.7692e-01, -2.6540e+00, -3.3344e+00, ..., -1.6473e+00,
-1.0149e+00, -1.3308e+00],
[ 4.1991e-01, -9.7328e-01, -2.1798e+00, ..., -7.4886e-01,
-1.1166e+00, -1.9934e+00],
...,
[-6.4482e-02, -7.7328e-01, -2.3152e-01, ..., -4.7582e-01,
-3.6966e-02, -4.5691e-01],
[-7.0894e-01, -7.2795e-01, 1.6866e-01, ..., 2.4357e-01,
2.5089e-01, -1.8395e-01],
[-9.2875e-01, -1.5612e+00, -1.7418e+00, ..., -1.4863e+00,
-1.1074e+00, -4.5408e-01]],
[[-2.7971e+00, -4.9448e+00, -4.2618e+00, ..., -3.9553e+00,
-4.9966e+00, -1.9763e+00],
[-4.9865e+00, -8.6400e+00, -9.0543e+00, ..., -8.4365e+00,
-8.9469e+00, -3.3348e+00],
[-3.7016e+00, -6.8263e+00, -7.9645e+00, ..., -7.4536e+00,
-7.8229e+00, -3.6854e+00],
...,
[-1.8801e+00, -2.4628e+00, -2.7098e+00, ..., -2.7761e+00,
-2.3603e+00, -9.6948e-01],
[-3.2569e+00, -3.6213e+00, -4.2720e+00, ..., -3.1290e+00,
-3.2456e+00, -9.5387e-01],
[ 2.5036e-01, -5.0110e-01, 3.1434e-01, ..., -2.4482e-02,
4.3000e-01, 2.2671e-01]],
[[-5.0838e-01, -9.3777e-01, -2.5669e+00, ..., 8.3552e-01,
5.4255e-01, 1.4968e+00],
[-3.9912e-01, -1.8748e+00, -3.1675e+00, ..., 7.4514e-01,
6.7120e-01, 4.2742e-01],
[-9.6117e-01, -1.5772e+00, -2.9457e+00, ..., -9.1593e-01,
1.3190e-02, 2.9319e-01],
...,
[-8.0819e-01, 2.7854e-02, 4.0331e-01, ..., 2.7523e-01,
2.4538e-01, 3.9744e-01],
[-5.6397e-01, -7.0193e-02, -2.3187e-01, ..., 2.9245e-01,
2.4177e-01, 5.6988e-01],
[-2.2316e-01, -3.3862e-01, -1.8808e-01, ..., 3.2539e-01,
9.8570e-02, 3.3939e-01]],
...,
[[ 1.3161e+00, 5.8701e-01, -1.1619e+00, ..., 4.3229e-01,
2.2429e-01, -1.0508e+00],
[-4.5846e-01, -6.0853e-01, -1.9924e+00, ..., 6.1975e-02,
-9.2376e-01, -1.3643e+00],
[ 8.0183e-02, 8.9295e-01, -2.4768e-01, ..., -1.5461e+00,
-9.7896e-01, -1.1117e+00],
...,
[-7.9008e-01, -5.2535e-01, -5.1523e-01, ..., -2.3989e-01,
-5.3262e-01, -5.5777e-01],
[-2.6307e+00, -2.0801e+00, -8.1406e-01, ..., -1.6728e+00,
-1.3066e+00, -1.0375e+00],
[-6.9865e-01, -1.0947e+00, -3.7386e-01, ..., -1.4881e+00,
-3.7455e-01, -3.8522e-01]],
[[ 3.9869e-01, -4.9203e-01, -1.9859e+00, ..., 2.9303e-01,
7.2066e-01, -2.4414e-01],
[ 6.9565e-01, -1.4062e+00, -3.0224e+00, ..., -2.5587e-01,
-4.6145e-01, -1.1240e+00],
[ 3.4489e-01, -9.6053e-01, -1.4058e+00, ..., 9.9552e-01,
-2.7334e-01, -1.2630e+00],
...,
[-4.0521e-01, -5.8771e-01, 3.9442e-01, ..., -1.2155e-01,
2.8340e-01, 6.2404e-02],
[-7.6210e-01, -1.1334e+00, 4.4723e-01, ..., 3.9908e-02,
-8.2679e-03, -1.3737e-01],
[ 3.3252e-01, -4.1598e-02, -3.2084e-01, ..., -7.8113e-01,
-8.2712e-02, 2.0197e-01]],
[[-9.6436e-01, -6.6114e-01, -1.6121e+00, ..., 3.4811e-01,
3.9991e-01, 9.4886e-01],
[-1.5043e+00, -6.7575e-01, -1.8579e+00, ..., -1.4086e+00,
-5.7730e-01, 1.0099e+00],
[-1.3350e+00, -5.7563e-01, -7.0519e-01, ..., -2.9932e+00,
-1.2401e+00, 8.6066e-01],
...,
[-6.9279e-01, -2.5226e-01, -6.6654e-02, ..., -1.9209e-01,
-1.2572e-02, 6.5149e-01],
[-1.3555e+00, -7.9169e-01, -5.2329e-01, ..., -7.9192e-01,
-2.0743e-01, 2.0527e-01],
[-4.9376e-01, -1.8278e-01, -1.0431e+00, ..., 1.9274e-02,
-1.6635e-01, 3.6101e-01]]],
[[[-6.3287e-01, -1.4533e+00, -1.4109e+00, ..., -1.2492e+00,
-8.4959e-01, -1.7010e+00],
[-2.4757e-01, -2.6000e+00, -2.2751e+00, ..., -1.7417e+00,
-1.8386e+00, -1.9107e+00],
[-1.5615e-01, -2.0662e+00, -1.6814e+00, ..., -1.2423e+00,
-1.3796e+00, -2.0695e+00],
...,
[-1.0073e-01, -1.1062e+00, -3.0088e-01, ..., -5.2640e-01,
-4.5477e-01, -8.9595e-01],
[-1.5483e-01, -9.8631e-01, -2.8972e-01, ..., -2.0930e-01,
-9.6418e-02, -9.7915e-01],
[-1.5357e-01, -1.1376e+00, -7.7953e-01, ..., -8.7246e-01,
-1.1233e+00, -6.3714e-01]],
[[-2.5112e+00, -4.8114e+00, -4.3884e+00, ..., -4.8388e+00,
-6.0468e+00, -2.4789e+00],
[-5.4510e+00, -8.1954e+00, -7.3115e+00, ..., -7.6835e+00,
-9.1219e+00, -4.3483e+00],
[-3.9405e+00, -5.6912e+00, -5.5062e+00, ..., -5.9637e+00,
-6.4174e+00, -3.2134e+00],
...,
[-1.9727e+00, -3.1764e+00, -4.5732e+00, ..., -2.9175e+00,
-3.4450e+00, -1.4724e+00],
[-4.5644e+00, -5.5499e+00, -5.1985e+00, ..., -4.9887e+00,
-6.2488e+00, -2.3724e+00],
[ 6.6105e-01, -3.2925e-01, 5.6635e-01, ..., 5.1364e-01,
5.7950e-01, 3.3870e-03]],
[[-7.0057e-01, 2.9150e-01, 9.1328e-01, ..., 8.1169e-01,
6.2590e-01, 1.8690e+00],
[-9.3775e-01, -1.9944e-01, 4.3877e-01, ..., 5.4492e-01,
1.6083e-02, 1.2306e-01],
[-1.7031e+00, -3.6908e-01, -1.4447e-02, ..., 2.9069e-02,
4.7364e-02, -7.5322e-02],
...,
[-9.0953e-01, -1.3101e-01, -4.3223e-02, ..., 4.2907e-02,
9.4065e-02, 3.6464e-01],
[-9.5551e-01, 4.6952e-02, -1.9422e-02, ..., 2.1179e-02,
1.0200e-01, 3.9509e-01],
[-4.9704e-01, -2.5178e-02, 6.2672e-01, ..., -3.0257e-02,
9.4451e-02, 2.4688e-02]],
...,
[[ 1.0294e+00, -4.8849e-01, -6.8401e-02, ..., 2.3351e-01,
-5.1370e-02, -2.0694e+00],
[-9.9305e-01, -8.7466e-01, -4.6697e-01, ..., -5.3090e-01,
-1.4303e+00, -2.0209e+00],
[-8.1506e-01, -7.4708e-01, -2.6222e-01, ..., -3.2388e-01,
-6.1198e-01, -1.3638e+00],
...,
[-4.2946e-01, -6.6784e-02, 6.8246e-01, ..., 1.1239e-01,
-1.2301e-01, -5.6018e-01],
[-2.0625e+00, -1.3814e+00, -4.0497e-01, ..., -9.8926e-01,
-1.1520e+00, -1.1348e+00],
[-2.8126e-01, -7.7133e-01, -3.2795e-01, ..., -1.2881e-01,
-2.8787e-01, -6.1036e-01]],
[[-1.0999e-01, -6.4647e-02, 5.0729e-01, ..., 4.8015e-01,
9.1390e-01, -2.9768e-02],
[-4.9068e-02, -8.6942e-01, -6.6593e-02, ..., 4.2346e-01,
3.5118e-01, -8.2671e-01],
[-1.4562e-01, -8.0122e-01, 2.8229e-01, ..., 7.0287e-01,
5.3430e-01, -3.8999e-01],
...,
[-4.2497e-01, -4.9941e-01, 6.1206e-01, ..., 1.1389e-01,
8.7188e-02, -1.2125e-01],
[-7.7686e-01, -1.4955e+00, -2.2309e-01, ..., -3.0946e-01,
-3.1158e-01, -7.0496e-01],
[ 4.8374e-01, 5.2173e-02, 1.9347e-01, ..., 2.2094e-01,
1.6993e-01, 3.4041e-01]],
[[-1.6576e+00, -7.8302e-01, -3.7494e-01, ..., -4.6658e-01,
-3.1178e-01, 8.5113e-01],
[-2.1338e+00, -7.5952e-01, -5.7830e-01, ..., -7.7517e-01,
-2.0043e-01, 1.6474e+00],
[-1.4834e+00, -6.8176e-01, -2.8784e-01, ..., -3.5946e-01,
-1.1471e-01, 1.4130e+00],
...,
[-4.8432e-01, 1.1211e-01, -2.3294e-01, ..., 9.6548e-02,
2.2372e-01, 7.1343e-01],
[-9.1986e-01, -3.2186e-01, -9.0413e-01, ..., -1.0858e-01,
8.9242e-02, 4.1364e-01],
[-4.5263e-01, -3.6599e-01, -4.1576e-01, ..., -2.4714e-01,
-2.5568e-01, 5.7064e-01]]],
[[[-1.9539e-01, -2.4712e-01, 1.9488e-01, ..., 6.1631e-02,
5.0431e-01, 1.5127e-01],
[-1.0459e-01, -1.5266e+00, -8.5610e-01, ..., -7.0209e-01,
-4.1130e-01, -3.1064e-01],
[-1.1406e-01, -1.3295e+00, -6.7854e-01, ..., -4.9396e-01,
-3.5334e-01, -6.5154e-01],
...,
[-2.6492e-01, -1.7265e+00, -1.2562e+00, ..., -9.4886e-01,
-9.6420e-01, -2.0208e+00],
[-1.1171e+00, -1.7423e+00, -8.0405e-01, ..., -5.3843e-01,
-4.4353e-01, -1.7719e+00],
[-1.2885e+00, -2.1399e+00, -1.5972e+00, ..., -1.4175e+00,
-1.5389e+00, -9.8289e-01]],
[[-1.8396e-01, -1.4049e+00, -5.7355e-01, ..., -5.8395e-01,
-1.3074e+00, -2.2170e-01],
[-2.0012e+00, -3.9853e+00, -2.3604e+00, ..., -2.3180e+00,
-3.2919e+00, -1.0841e+00],
[-1.8565e+00, -3.3457e+00, -2.4697e+00, ..., -2.5017e+00,
-2.8906e+00, -1.1971e+00],
...,
[-3.7397e+00, -5.4808e+00, -5.1657e+00, ..., -5.4020e+00,
-6.1496e+00, -2.9010e+00],
[-6.1419e+00, -6.9483e+00, -6.0104e+00, ..., -6.0361e+00,
-7.4164e+00, -2.7036e+00],
[-3.8108e-02, -8.9646e-01, 3.0186e-01, ..., 4.2412e-01,
7.1766e-01, 2.1064e-01]],
[[-5.2773e-01, -1.8381e-01, 3.2712e-01, ..., 3.3548e-01,
2.6792e-02, 8.4512e-01],
[-3.6562e-01, -4.6637e-01, 4.8530e-01, ..., 3.9677e-01,
2.8475e-01, 4.9001e-01],
[-8.6260e-01, -2.4854e-01, 1.9594e-01, ..., 8.0334e-02,
1.9605e-01, 4.1658e-01],
...,
[-1.4021e+00, 1.6872e-02, 2.7602e-02, ..., 3.8763e-02,
1.0965e-03, 2.5209e-01],
[-1.2226e+00, 1.3374e-01, 1.8970e-03, ..., -2.8162e-02,
1.3895e-02, 3.0774e-01],
[-1.1527e+00, -3.9032e-02, 9.9892e-02, ..., 3.2696e-04,
-1.6771e-02, -3.1175e-01]],
...,
[[ 6.1813e-02, -6.4236e-01, -4.1287e-01, ..., -6.5979e-02,
-2.3849e-01, -7.2413e-01],
[-8.6580e-01, -6.4193e-01, -5.6238e-01, ..., -2.7467e-01,
-9.6949e-01, -7.2206e-01],
[-5.2433e-01, -1.5358e-01, -5.1429e-02, ..., 7.6724e-02,
-1.9373e-01, -5.0690e-01],
...,
[-6.2428e-01, -6.0945e-01, -3.3616e-01, ..., -3.8030e-01,
-6.0684e-01, -1.1919e+00],
[-3.1162e+00, -2.8736e+00, -1.9822e+00, ..., -1.9021e+00,
-2.0239e+00, -2.0610e+00],
[-5.4919e-01, -1.6307e+00, -1.0531e+00, ..., -8.6036e-01,
-9.2209e-01, -1.3054e+00]],
[[-1.8649e-01, 4.6171e-02, 4.4281e-01, ..., 8.6844e-02,
5.4487e-01, 1.7499e-01],
[-4.5268e-01, -2.6438e-01, 1.4548e-03, ..., -7.1862e-02,
1.1024e-01, -4.4912e-02],
[-6.9081e-01, -5.6623e-01, -4.1128e-02, ..., 2.9741e-02,
2.6900e-02, -2.3278e-02],
...,
[-1.3156e-01, -1.0156e+00, 3.2458e-02, ..., 2.6841e-01,
1.4151e-01, -7.2103e-01],
[-5.3960e-01, -1.8437e+00, -1.6192e-01, ..., 2.9981e-02,
-4.9621e-02, -9.4669e-01],
[ 4.4141e-01, -1.2377e-01, 1.4375e-01, ..., 2.5989e-01,
3.2219e-01, 3.2388e-01]],
[[-6.8950e-01, 4.1288e-01, 5.5529e-01, ..., 4.7619e-01,
3.7303e-01, 5.9240e-01],
[-8.7033e-01, 2.8588e-01, -2.4452e-01, ..., -2.6165e-01,
-1.7288e-01, 6.3817e-01],
[-4.9506e-01, 3.2784e-01, 2.2979e-02, ..., 8.4623e-02,
8.6310e-02, 5.8788e-01],
...,
[-1.0878e+00, -9.3497e-01, -4.4788e-01, ..., -4.0815e-01,
-5.8793e-02, 8.1733e-01],
[-1.8295e+00, -1.8303e+00, -9.7555e-01, ..., -7.9951e-01,
-4.6074e-01, 4.2166e-01],
[-9.2402e-01, -9.5412e-01, -2.7047e-01, ..., -2.9121e-01,
-1.4999e-01, 1.1191e+00]]],
[[[-7.5751e-01, -1.4079e+00, -1.3705e+00, ..., -2.0915e-01,
2.7617e-01, -3.7716e-01],
[-2.9339e-01, -2.3904e+00, -2.1113e+00, ..., -5.7349e-01,
-3.6133e-01, -5.3692e-01],
[-3.8095e-01, -2.0959e+00, -1.7963e+00, ..., -7.8840e-01,
-3.4412e-01, -1.0336e+00],
...,
[-4.7887e-01, -1.9178e+00, -1.8786e+00, ..., -3.0615e-01,
-2.2816e-01, -1.1263e+00],
[-9.7261e-01, -1.6505e+00, -1.2133e+00, ..., 3.8129e-02,
1.5532e-01, -9.4813e-01],
[-7.9814e-01, -1.6194e+00, -1.2336e+00, ..., -9.4146e-01,
-1.1602e+00, -6.6393e-01]],
[[-2.4004e+00, -4.8251e+00, -4.2781e+00, ..., -2.1298e+00,
-3.4513e+00, -1.2382e+00],
[-5.3194e+00, -8.0727e+00, -6.7622e+00, ..., -3.3766e+00,
-5.0732e+00, -2.0480e+00],
[-4.3007e+00, -6.2210e+00, -5.4734e+00, ..., -4.1714e+00,
-3.7915e+00, -1.7038e+00],
...,
[-4.3779e+00, -6.0674e+00, -5.5066e+00, ..., -3.2737e+00,
-4.0478e+00, -1.8302e+00],
[-6.2668e+00, -7.4279e+00, -6.3224e+00, ..., -3.9809e+00,
-5.4448e+00, -1.9556e+00],
[-2.0791e-02, -8.3285e-01, 4.2281e-01, ..., 3.8367e-01,
5.4837e-01, 1.5621e-01]],
[[-5.8677e-01, 6.4513e-01, 1.1002e+00, ..., 8.2591e-01,
5.7921e-01, 1.3595e+00],
[-9.2208e-01, -8.6842e-03, 4.7410e-01, ..., 4.2276e-01,
3.7345e-01, 2.5856e-01],
[-1.6667e+00, -1.5228e-01, -2.0144e-02, ..., 2.5913e-01,
3.0386e-01, 1.0542e-01],
...,
[-1.7043e+00, -7.5213e-02, -8.9532e-02, ..., 2.9809e-01,
2.6573e-01, 6.5578e-02],
[-1.5267e+00, 5.5480e-02, 4.4540e-02, ..., 2.1401e-02,
2.6974e-01, 1.8992e-01],
[-1.3932e+00, -4.7672e-01, -2.8066e-01, ..., -1.2293e-01,
1.3094e-01, -1.3396e-01]],
...,
[[ 1.0633e+00, -4.3042e-01, -2.2394e-01, ..., 3.3850e-01,
2.0179e-01, -7.9664e-01],
[-7.8165e-01, -9.5792e-01, -7.5011e-01, ..., -3.4133e-01,
-1.0470e+00, -9.6132e-01],
[-6.6307e-01, -9.7742e-01, -6.4055e-01, ..., -2.2755e-01,
-2.3999e-01, -5.9950e-01],
...,
[-7.5787e-01, -1.1576e+00, -7.4678e-01, ..., -2.7150e-01,
-4.3199e-01, -8.0990e-01],
[-2.7569e+00, -2.8184e+00, -1.8406e+00, ..., -1.0702e+00,
-1.1390e+00, -1.1072e+00],
[-7.0186e-01, -1.6236e+00, -9.1160e-01, ..., -2.0145e-01,
-2.8798e-01, -4.8835e-01]],
[[ 1.0319e-02, -1.9478e-02, 5.5917e-01, ..., 3.4728e-01,
8.3658e-01, 6.8319e-02],
[ 2.4398e-02, -7.1925e-01, 1.8635e-01, ..., 4.1824e-01,
6.7144e-01, -2.6059e-01],
[-7.5288e-02, -7.8719e-01, 3.6459e-01, ..., -1.2350e-01,
4.3048e-01, -1.0633e-01],
...,
[ 2.6671e-02, -8.0618e-01, 2.8235e-01, ..., 3.6036e-01,
2.0236e-01, -2.6300e-01],
[-2.6427e-01, -1.4404e+00, -4.8727e-03, ..., 1.1770e-01,
9.1427e-02, -4.8260e-01],
[ 6.9871e-01, 1.7592e-01, 2.8860e-01, ..., 2.6120e-01,
2.5285e-01, 3.3380e-01]],
[[-1.5703e+00, -5.7027e-01, 5.6075e-02, ..., 3.3557e-01,
2.6561e-01, 8.6212e-01],
[-2.0814e+00, -3.0621e-01, -5.5503e-02, ..., -4.1676e-01,
-1.3849e-01, 1.1911e+00],
[-1.7432e+00, -3.5150e-01, 1.2199e-01, ..., -1.2344e-01,
9.2828e-03, 9.5031e-01],
...,
[-1.7529e+00, -4.8803e-01, 2.5626e-01, ..., 2.3183e-02,
1.3971e-01, 1.0609e+00],
[-2.1775e+00, -1.0938e+00, -1.2923e-01, ..., -1.2450e-01,
1.6542e-02, 6.5924e-01],
[-1.0249e+00, -1.8005e-01, 4.3307e-01, ..., -2.1051e-01,
-2.4176e-01, 7.0967e-01]]]], device='cuda:0'),
tensor([[[[-7.3829e-01, -1.3873e+00, -1.3896e+00, ..., 6.9767e-04,
6.5768e-01, 5.3655e-01],
[-2.8341e-01, -2.4054e+00, -2.1324e+00, ..., -4.7012e-01,
-2.4157e-01, 1.2379e-01],
[-4.0113e-01, -2.0843e+00, -2.0871e+00, ..., -5.6869e-01,
-1.3279e-01, -7.6363e-02],
...,
[-4.6999e-01, -1.9172e+00, -1.8670e+00, ..., -3.2887e-01,
-8.3011e-02, -1.3753e-01],
[-9.5356e-01, -1.6966e+00, -1.2118e+00, ..., 1.0894e-02,
2.6275e-01, -1.2126e-02],
[-8.0905e-01, -1.6133e+00, -1.2241e+00, ..., -1.0430e+00,
-1.1041e+00, -3.8717e-01]],
[[-2.3658e+00, -4.8101e+00, -4.1202e+00, ..., -4.2582e-01,
-1.1163e+00, -3.0991e-02],
[-5.3057e+00, -8.0492e+00, -6.5025e+00, ..., -1.7856e+00,
-2.6987e+00, -6.6650e-01],
[-4.3144e+00, -6.2513e+00, -7.1134e+00, ..., -1.8201e+00,
-1.4904e+00, -3.6435e-01],
...,
[-4.4042e+00, -6.1059e+00, -5.5100e+00, ..., -1.3195e+00,
-1.6063e+00, -3.1130e-01],
[-6.3283e+00, -7.3567e+00, -6.2762e+00, ..., -2.5111e+00,
-3.4044e+00, -9.6795e-01],
[-3.5756e-02, -8.1916e-01, 4.1895e-01, ..., 3.6198e-01,
3.4684e-01, 7.3266e-02]],
[[-5.9709e-01, 6.1768e-01, 1.0907e+00, ..., 7.9743e-01,
2.8218e-01, 1.0833e+00],
[-9.1774e-01, -5.6321e-02, 4.2456e-01, ..., 4.2408e-01,
8.4738e-02, 5.1350e-01],
[-1.6353e+00, -1.3869e-01, -4.6583e-01, ..., 8.7005e-02,
1.1791e-01, 3.9805e-01],
...,
[-1.7254e+00, -5.0074e-02, -6.9145e-02, ..., 3.1885e-02,
8.2771e-03, 3.7707e-01],
[-1.5464e+00, 3.2954e-02, 4.1504e-02, ..., 1.0178e-01,
9.0360e-02, 5.6409e-01],
[-1.3674e+00, -5.1516e-01, -2.6110e-01, ..., -9.7012e-02,
6.9592e-02, 3.3091e-01]],
...,
[[ 1.0549e+00, -4.5003e-01, -2.2293e-01, ..., -1.0700e-01,
-3.6244e-01, -9.4527e-01],
[-8.3626e-01, -9.7222e-01, -8.0069e-01, ..., -6.8682e-01,
-1.2237e+00, -8.2245e-01],
[-6.8905e-01, -9.9755e-01, -5.3144e-01, ..., -1.1433e-01,
-2.1742e-01, -4.0858e-01],
...,
[-7.5085e-01, -1.1548e+00, -7.5748e-01, ..., -1.2304e-01,
-1.9587e-01, -3.9592e-01],
[-2.7452e+00, -2.7800e+00, -1.8168e+00, ..., -8.3586e-01,
-8.8717e-01, -8.1876e-01],
[-6.7878e-01, -1.6533e+00, -9.6550e-01, ..., 1.9776e-02,
-1.5087e-01, -3.5161e-01]],
[[ 9.1168e-03, -9.1499e-03, 5.7807e-01, ..., 3.3366e-01,
6.7677e-01, 2.8108e-01],
[-1.3665e-02, -7.3271e-01, 1.2332e-01, ..., 7.8958e-04,
1.4659e-01, -4.9460e-02],
[-1.1946e-01, -8.1407e-01, 4.8848e-01, ..., -7.2997e-01,
-7.5284e-02, 1.9317e-01],
...,
[ 2.8044e-02, -7.6730e-01, 2.6435e-01, ..., -1.5031e-01,
-1.2697e-01, 1.6435e-01],
[-2.7680e-01, -1.4178e+00, 1.0752e-02, ..., -3.8088e-01,
-3.7701e-01, -6.5872e-02],
[ 6.4507e-01, 1.6570e-01, 2.9441e-01, ..., -2.3600e-01,
-2.5553e-01, 3.2593e-01]],
[[-1.5842e+00, -5.7612e-01, 5.1811e-02, ..., 5.4220e-01,
3.5387e-01, 5.8655e-01],
[-2.0595e+00, -3.4539e-01, -3.6945e-02, ..., -7.6771e-02,
-1.6389e-01, 4.0509e-01],
[-1.7021e+00, -4.0279e-01, 3.0648e-01, ..., 4.3535e-02,
-1.9069e-01, 9.6250e-02],
...,
[-1.7598e+00, -4.8642e-01, 2.9115e-01, ..., 1.1559e-01,
-1.0066e-01, 6.2552e-02],
[-2.2061e+00, -1.1136e+00, -1.2325e-01, ..., -4.1119e-02,
-3.0121e-01, -2.0056e-01],
[-1.0596e+00, -1.9746e-01, 4.7341e-01, ..., -9.5167e-02,
-3.1761e-01, 1.2307e-01]]],
[[[ 2.5244e-02, -2.2212e-01, -3.4934e-02, ..., -6.3730e-01,
-4.0277e-01, -2.1772e+00],
[ 3.7533e-01, -9.1178e-01, -6.7330e-01, ..., -3.3583e+00,
-2.0164e+00, -1.5656e+00],
[ 2.4193e-01, -4.9575e-01, -1.3850e+00, ..., -7.4399e+00,
-5.8859e+00, -2.0126e+00],
...,
[ 1.7917e-02, -6.2386e-01, -4.4256e-01, ..., -1.3766e+00,
-1.1524e+00, -1.6409e+00],
[-5.1682e-01, -6.4484e-01, -5.5889e-02, ..., -7.7408e-01,
-5.5747e-01, -1.3931e+00],
[-6.3819e-01, -1.3535e+00, -1.1919e+00, ..., -1.1147e+00,
-1.2870e+00, -7.8922e-01]],
[[-7.3991e-01, -2.2939e+00, -1.8997e+00, ..., -9.3076e+00,
-1.1042e+01, -3.7790e+00],
[-1.9821e+00, -3.8650e+00, -3.1997e+00, ..., -1.3168e+01,
-1.6687e+01, -5.2429e+00],
[-1.5212e+00, -3.5854e+00, -4.8961e+00, ..., -1.2540e+01,
-1.5992e+01, -5.9052e+00],
...,
[-1.7431e+00, -2.4148e+00, -2.3797e+00, ..., -5.2412e+00,
-5.4513e+00, -2.3170e+00],
[-3.6883e+00, -4.4220e+00, -3.6892e+00, ..., -6.0196e+00,
-6.8268e+00, -2.6490e+00],
[ 3.2746e-01, -2.8911e-01, 4.0076e-01, ..., 4.6027e-01,
6.0209e-01, 7.3255e-02]],
[[-1.9616e-01, 1.5672e-01, 7.6490e-01, ..., 6.7793e-01,
5.5619e-01, 1.8893e+00],
[-1.9940e-02, -3.9303e-01, 5.5872e-01, ..., -3.6950e+00,
-3.1218e+00, -5.3981e-01],
[-2.0424e-01, -3.0573e-01, -7.7508e-01, ..., -7.2302e+00,
-7.5075e+00, -2.7122e+00],
...,
[-2.3011e-01, -1.8508e-01, 1.4143e-01, ..., -1.1914e-01,
1.1925e-01, 2.9581e-02],
[-2.2779e-01, -1.1541e-01, 1.7450e-01, ..., -1.8894e-02,
1.3510e-01, 1.4837e-01],
[-1.5179e-01, -5.0356e-01, 1.3421e-01, ..., -3.9520e-01,
-7.1491e-02, -2.6120e-01]],
...,
[[ 5.8130e-01, -2.9987e-01, -1.6932e-02, ..., -1.1994e-01,
-3.8594e-01, -1.6242e+00],
[-4.4653e-01, -5.7023e-01, -5.1693e-01, ..., -1.1146e+00,
-1.8264e+00, -4.7570e+00],
[-3.2232e-01, -1.4813e-01, 3.1724e-01, ..., -1.3543e+00,
-3.8838e+00, -8.3608e+00],
...,
[-4.5255e-01, -1.5017e-01, -1.6087e-01, ..., -7.0703e-01,
-9.0138e-01, -1.1942e+00],
[-2.2145e+00, -1.6286e+00, -1.2545e+00, ..., -1.7363e+00,
-1.7893e+00, -1.6577e+00],
[-4.6101e-01, -7.4247e-01, -3.9540e-01, ..., -7.1361e-01,
-7.6449e-01, -8.1904e-01]],
[[ 2.0120e-01, 2.6874e-01, 6.2073e-01, ..., 6.4709e-01,
1.0518e+00, -1.4243e+00],
[ 2.0704e-01, 3.4880e-01, 5.4593e-01, ..., -1.4282e+00,
-5.0796e-01, -1.4670e+00],
[-1.0751e-01, 3.1899e-01, 2.7999e-01, ..., -6.6057e+00,
-4.4711e+00, -2.2550e+00],
...,
[-1.5735e-01, -2.8224e-02, 3.1807e-01, ..., 3.2265e-01,
1.8271e-01, -3.1076e-01],
[-4.6311e-01, -6.9173e-01, -6.7167e-02, ..., 1.9311e-02,
-5.8132e-02, -5.1036e-01],
[ 5.2697e-01, 3.6816e-01, 2.6574e-02, ..., 2.8865e-01,
1.8191e-01, 3.2654e-01]],
[[-8.1728e-01, 2.4261e-01, 3.6752e-01, ..., -1.1429e+00,
-1.6300e+00, -6.6107e-01],
[-1.0689e+00, 1.7040e-01, -4.0550e-01, ..., -4.9067e+00,
-4.6010e+00, -9.5144e-02],
[-9.3243e-01, 3.8589e-01, -4.2406e-01, ..., -6.0225e+00,
-6.3798e+00, 7.0403e-02],
...,
[-9.6759e-01, 2.0156e-01, -1.2139e-01, ..., 2.2106e-01,
3.9590e-01, 1.2146e+00],
[-1.5864e+00, -2.6331e-01, -3.1038e-01, ..., -5.7756e-02,
1.8873e-01, 8.4588e-01],
[-6.9479e-01, -5.7892e-02, -3.2062e-01, ..., 3.1814e-01,
2.0779e-01, 1.0747e+00]]],
[[[ 1.3184e-02, -2.5314e-01, -3.3496e-02, ..., -1.1344e+00,
-6.0232e-01, -9.6898e-01],
[ 3.1941e-01, -9.4566e-01, -6.9284e-01, ..., -1.6078e+00,
-1.2847e+00, -1.2600e+00],
[ 2.3163e-01, -6.2793e-01, -6.2013e-01, ..., -1.9532e+00,
-2.0971e+00, -2.4283e+00],
...,
[-1.1991e-02, -5.8133e-01, 9.0408e-02, ..., -1.4219e+00,
-1.2098e+00, -1.6825e+00],
[-6.2372e-01, -6.8779e-01, -6.3346e-02, ..., -7.6364e-01,
-5.3332e-01, -1.4328e+00],
[-7.4656e-01, -1.3469e+00, -1.1831e+00, ..., -1.3259e+00,
-1.4994e+00, -9.0063e-01]],
[[-7.8469e-01, -2.2619e+00, -1.9168e+00, ..., -4.0885e+00,
-4.8034e+00, -1.7860e+00],
[-2.0698e+00, -3.8646e+00, -3.0696e+00, ..., -6.1736e+00,
-6.9761e+00, -2.8026e+00],
[-1.7176e+00, -2.9069e+00, -3.1050e+00, ..., -8.5216e+00,
-9.0541e+00, -3.8523e+00],
...,
[-1.7807e+00, -2.4801e+00, -2.7045e+00, ..., -5.2759e+00,
-5.5253e+00, -2.3461e+00],
[-3.8114e+00, -4.4435e+00, -3.6817e+00, ..., -6.2471e+00,
-7.1815e+00, -2.7706e+00],
[ 2.7953e-01, -3.3542e-01, 3.9282e-01, ..., 5.2702e-01,
6.4777e-01, 7.1369e-02]],
[[-2.4092e-01, 1.4741e-01, 7.4677e-01, ..., 9.8647e-01,
7.1577e-01, 1.5253e+00],
[ 6.4421e-03, -4.2671e-01, 5.6868e-01, ..., 2.7476e-01,
3.5786e-01, 2.9351e-01],
[-1.5877e-01, -2.7176e-01, 2.4311e-01, ..., 2.5085e-01,
8.7906e-01, 7.7400e-01],
...,
[-2.6369e-01, -2.0600e-01, 9.1155e-01, ..., -2.0037e-01,
7.6113e-02, 2.0527e-02],
[-2.8382e-01, -1.4516e-01, 2.0041e-01, ..., -7.7776e-02,
8.2850e-02, 1.4465e-01],
[-1.6298e-01, -4.7753e-01, 9.4574e-02, ..., -4.1670e-01,
-8.7438e-02, -2.7328e-01]],
...,
[[ 5.9114e-01, -2.5282e-01, 9.3799e-03, ..., -6.3255e-02,
-2.5990e-01, -1.4889e+00],
[-4.6328e-01, -5.1892e-01, -5.2543e-01, ..., -7.6520e-01,
-1.5572e+00, -1.6907e+00],
[-3.0949e-01, -1.3720e-01, -2.1911e-01, ..., -5.1191e-01,
-1.0997e+00, -1.7476e+00],
...,
[-4.6787e-01, -1.5560e-01, -1.0202e+00, ..., -7.1628e-01,
-8.6522e-01, -1.2191e+00],
[-2.3575e+00, -1.7068e+00, -1.2106e+00, ..., -1.9090e+00,
-1.9388e+00, -1.7423e+00],
[-4.7864e-01, -8.3544e-01, -4.1427e-01, ..., -7.4137e-01,
-7.9073e-01, -8.7989e-01]],
[[ 1.7500e-01, 2.4986e-01, 6.0747e-01, ..., 4.6192e-01,
7.7073e-01, 4.4459e-03],
[ 1.3634e-01, 2.6828e-01, 4.7651e-01, ..., 2.8940e-01,
4.0949e-01, -2.9684e-01],
[-1.1461e-01, 1.5915e-01, 3.5891e-01, ..., 3.0412e-01,
1.8559e-01, -8.5086e-01],
...,
[-2.1144e-01, 1.1969e-03, 5.4974e-01, ..., 2.7363e-01,
1.7171e-01, -3.1677e-01],
[-5.5588e-01, -7.1300e-01, -4.5930e-02, ..., -6.1897e-02,
-1.0944e-01, -5.7384e-01],
[ 5.1228e-01, 3.8415e-01, 7.7604e-02, ..., 1.8665e-01,
1.3288e-01, 3.3045e-01]],
[[-8.1396e-01, 2.3630e-01, 3.2082e-01, ..., 7.2601e-02,
1.6509e-01, 8.4084e-01],
[-1.1108e+00, 2.3612e-01, -3.4410e-01, ..., -4.6211e-02,
1.4295e-01, 1.3849e+00],
[-9.9353e-01, 3.0514e-01, -1.6426e-01, ..., -3.3743e-01,
-4.3029e-01, 6.2604e-01],
...,
[-9.2434e-01, 1.8721e-01, -1.4842e-01, ..., 3.5294e-01,
4.4521e-01, 1.2577e+00],
[-1.6584e+00, -4.0306e-01, -3.3933e-01, ..., -1.4510e-01,
1.1877e-01, 8.3796e-01],
[-7.2178e-01, -9.7713e-02, -2.9845e-01, ..., 2.4160e-01,
1.1665e-01, 1.1007e+00]]],
[[[-2.7930e+00, -1.2510e+00, 7.8756e-01, ..., -1.7528e-02,
6.2416e-01, 5.2724e-01],
[-2.0559e+00, -5.1187e+00, -2.7730e+00, ..., -4.9395e-01,
-2.2166e-01, 1.1632e-01],
[-2.2780e-01, -7.1923e+00, -6.5970e+00, ..., -3.5627e-01,
-1.2222e-01, -1.0577e-01],
...,
[-4.7424e-01, -1.7528e+00, -1.6134e+00, ..., -2.9683e+00,
1.1201e+00, 9.8775e-01],
[-1.0079e+00, -1.6802e+00, -1.0597e+00, ..., -2.3603e-01,
1.3161e+00, -4.7465e-01],
[-8.2645e-01, -1.6604e+00, -1.2458e+00, ..., -1.3084e+00,
-1.3966e+00, -6.0478e-01]],
[[-4.4057e+00, -6.7208e+00, -5.8182e+00, ..., -4.8657e-01,
-1.0385e+00, -2.1518e-02],
[-7.2764e+00, -1.0752e+01, -1.1409e+01, ..., -1.8141e+00,
-2.6935e+00, -6.8515e-01],
[-7.7902e+00, -1.0666e+01, -1.1874e+01, ..., -1.3326e+00,
-1.5637e+00, -4.0170e-01],
...,
[-4.3865e+00, -6.2164e+00, -5.5032e+00, ..., -5.7660e+00,
-3.0301e+00, -1.3107e+00],
[-6.3992e+00, -7.4705e+00, -6.3886e+00, ..., -4.2496e+00,
-4.4475e+00, -1.8055e+00],
[-4.7843e-02, -8.4334e-01, 3.9127e-01, ..., -1.9048e-01,
8.8730e-02, 7.1793e-02]],
[[-3.4214e+00, -5.0615e+00, -4.3538e+00, ..., 8.4275e-01,
3.2279e-01, 1.1068e+00],
[-5.5289e+00, -7.5637e+00, -6.5549e+00, ..., 4.2389e-01,
4.9557e-02, 4.7446e-01],
[-6.2178e+00, -9.8824e+00, -7.3063e+00, ..., 6.0400e-02,
7.3702e-02, 4.1086e-01],
...,
[-1.7440e+00, 2.2316e-01, -1.9437e-01, ..., -4.9481e-02,
-3.1505e+00, -1.3165e+00],
[-1.5551e+00, 1.2136e-01, 1.1377e-01, ..., 2.5584e+00,
-1.4763e+00, -7.0350e-02],
[-1.4358e+00, -5.3993e-01, -2.8927e-01, ..., -2.2489e-01,
-3.5801e-01, 4.8106e-01]],
...,
[[-2.3801e-01, -4.0830e+00, -3.5890e+00, ..., -1.0316e-01,
-3.6779e-01, -9.3336e-01],
[-5.9996e-01, -5.9266e+00, -5.2968e+00, ..., -6.8172e-01,
-1.2577e+00, -8.0645e-01],
[ 3.3284e-01, -3.5611e+00, -2.0853e+00, ..., -1.1107e-01,
-1.8692e-01, -3.6768e-01],
...,
[-7.3007e-01, -1.3384e+00, -7.0699e-01, ..., -5.7963e+00,
-6.5049e+00, -6.7879e+00],
[-2.7950e+00, -2.9264e+00, -2.0331e+00, ..., -5.8970e+00,
-4.9061e+00, -3.3528e+00],
[-6.6449e-01, -1.6973e+00, -1.0099e+00, ..., -1.0475e+00,
-2.6055e-01, -1.8212e-01]],
[[-1.4987e+00, -3.1479e+00, -1.8752e+00, ..., 3.3865e-01,
6.8747e-01, 3.0691e-01],
[-2.3769e+00, -5.7111e+00, -4.0727e+00, ..., -5.8451e-03,
1.2358e-01, -3.3478e-02],
[-1.0973e+00, -6.3781e+00, -5.5708e+00, ..., -1.7120e-01,
-8.7336e-02, 2.0504e-01],
...,
[ 9.3948e-02, -6.8869e-01, 2.6600e-01, ..., -6.1313e-01,
-1.1040e+00, -7.2132e-02],
[-2.3284e-01, -1.4197e+00, 7.3008e-02, ..., -1.1477e-01,
4.2740e-01, -8.1993e-01],
[ 6.6159e-01, 1.6507e-01, 2.9447e-01, ..., -9.3435e-01,
-7.5831e-01, 1.8064e-02]],
[[-1.5249e+00, -3.9802e+00, -5.1286e+00, ..., 5.7190e-01,
3.4714e-01, 5.8106e-01],
[-1.5720e+00, -3.6188e+00, -7.4450e+00, ..., -5.1382e-02,
-1.7006e-01, 4.4512e-01],
[-2.7690e-01, -1.1120e+00, -8.4644e+00, ..., 5.3387e-02,
-1.8365e-01, 1.2593e-01],
...,
[-1.8066e+00, -5.8015e-01, 4.4171e-01, ..., -6.5059e+00,
-2.4717e+00, 4.6615e-01],
[-2.2713e+00, -1.1428e+00, -5.9760e-02, ..., -6.7365e-02,
6.6953e-01, 8.4640e-01],
[-1.0848e+00, -2.1340e-01, 4.4795e-01, ..., 1.0261e+00,
4.3367e-01, 1.4353e-01]]],
[[[-6.3266e-01, -1.4577e+00, -1.4477e+00, ..., -1.2236e+00,
-8.7558e-01, -1.7389e+00],
[-2.6816e-01, -2.6053e+00, -2.3412e+00, ..., -1.6952e+00,
-1.8290e+00, -1.9139e+00],
[-1.5815e-01, -2.0714e+00, -1.7021e+00, ..., -1.2440e+00,
-1.4199e+00, -2.1104e+00],
...,
[ 8.8776e-02, -5.5291e-01, -2.1657e-01, ..., -1.8548e-01,
-6.3547e-02, -4.3152e-01],
[-3.3427e-01, -4.4850e-01, 9.0621e-03, ..., 1.5095e-01,
2.9499e-01, -1.4131e-01],
[-6.1224e-01, -1.3217e+00, -1.0978e+00, ..., -1.1001e+00,
-1.2017e+00, -5.2055e-01]],
[[-2.5237e+00, -4.8583e+00, -4.4427e+00, ..., -4.8525e+00,
-6.0615e+00, -2.4969e+00],
[-5.5000e+00, -8.2630e+00, -7.3873e+00, ..., -7.5900e+00,
-8.9808e+00, -4.2770e+00],
[-4.0079e+00, -5.7954e+00, -5.5982e+00, ..., -5.9119e+00,
-6.4011e+00, -3.1997e+00],
...,
[-1.7169e+00, -2.1003e+00, -1.7175e+00, ..., -1.9153e+00,
-2.2575e+00, -9.0510e-01],
[-2.7455e+00, -2.9186e+00, -1.9361e+00, ..., -2.1977e+00,
-3.3940e+00, -1.0100e+00],
[ 3.2617e-01, -2.5107e-01, 3.1287e-01, ..., 2.9599e-01,
4.5529e-01, 1.9360e-01]],
[[-6.9953e-01, 2.9084e-01, 9.0841e-01, ..., 7.5956e-01,
6.1539e-01, 1.8725e+00],
[-9.3661e-01, -1.9736e-01, 4.9650e-01, ..., 5.2642e-01,
-1.4420e-02, 8.6639e-02],
[-1.7307e+00, -3.8398e-01, 2.8291e-03, ..., 7.7047e-02,
-8.6741e-03, -1.3944e-01],
...,
[-6.0965e-01, 8.6216e-02, 2.0981e-01, ..., 1.1704e-01,
2.7053e-01, 4.2746e-01],
[-4.2526e-01, 1.0499e-01, 1.9852e-01, ..., 1.5632e-01,
1.9303e-01, 5.5933e-01],
[-1.5266e-01, -2.5610e-01, 1.3752e-01, ..., -2.0675e-02,
6.4005e-02, 3.4666e-01]],
...,
[[ 1.0426e+00, -4.6800e-01, -5.8381e-02, ..., 2.6578e-01,
-1.6892e-02, -2.0595e+00],
[-9.8070e-01, -8.6431e-01, -4.8944e-01, ..., -5.0258e-01,
-1.4248e+00, -2.0327e+00],
[-8.1488e-01, -7.4560e-01, -2.6030e-01, ..., -3.1046e-01,
-6.0157e-01, -1.3792e+00],
...,
[-6.4524e-01, -5.2606e-01, -3.7045e-01, ..., -3.0335e-01,
-4.9750e-01, -5.5609e-01],
[-2.2140e+00, -1.7759e+00, -1.3075e+00, ..., -1.3298e+00,
-1.4009e+00, -1.1155e+00],
[-5.8495e-01, -7.7918e-01, -4.3907e-01, ..., -2.5991e-01,
-3.4848e-01, -3.8065e-01]],
[[-1.0175e-01, -5.4919e-02, 4.5341e-01, ..., 4.9627e-01,
9.0648e-01, -4.7371e-02],
[-3.9524e-02, -8.5381e-01, -1.0617e-01, ..., 4.3012e-01,
4.0218e-01, -8.1431e-01],
[-1.2771e-01, -7.6911e-01, 2.8773e-01, ..., 6.4606e-01,
5.3948e-01, -3.9214e-01],
...,
[-2.2059e-01, -4.6992e-01, 2.2495e-01, ..., 2.6306e-01,
3.3025e-01, 6.8837e-02],
[-5.0312e-01, -9.0559e-01, 3.2770e-02, ..., 2.0763e-02,
-1.1265e-02, -1.6024e-01],
[ 3.3461e-01, -1.9491e-02, -2.0128e-01, ..., -1.5592e-01,
-1.2159e-01, 1.6422e-01]],
[[-1.6558e+00, -7.8426e-01, -4.4360e-01, ..., -4.8484e-01,
-3.1549e-01, 8.5716e-01],
[-2.1361e+00, -7.4030e-01, -5.7574e-01, ..., -8.0576e-01,
-2.1219e-01, 1.6500e+00],
[-1.4751e+00, -6.2375e-01, -2.3646e-01, ..., -4.0249e-01,
-1.3810e-01, 1.4069e+00],
...,
[-6.3771e-01, -2.3232e-01, -1.1510e-01, ..., -6.6546e-02,
-1.5736e-04, 6.1199e-01],
[-1.0984e+00, -5.5139e-01, -3.2582e-01, ..., -3.5158e-01,
-1.8240e-01, 1.5216e-01],
[-3.4245e-01, -7.1537e-03, -1.6058e-01, ..., -2.5130e-01,
-2.1926e-01, 3.1351e-01]]]], device='cuda:0'),
tensor([[[[-3.5245e-01, -9.5174e-01, -6.4881e-01, ..., -5.7265e-01,
-7.9636e-02, -7.6861e-01],
[-5.3887e-03, -1.7922e+00, -1.3059e+00, ..., -1.0015e+00,
-8.5782e-01, -1.1791e+00],
[ 1.7789e-02, -1.4128e+00, -1.0027e+00, ..., -7.5523e-01,
-8.2064e-01, -1.7443e+00],
...,
[ 8.4614e-02, -5.0224e-01, -2.3661e-01, ..., -2.8536e-01,
-5.5899e-01, -4.5094e-01],
[-3.8957e-01, -4.0624e-01, 5.0319e-02, ..., -3.6721e-01,
-1.1860e-01, -1.7344e-01],
[-6.7590e-01, -1.3169e+00, -1.1007e+00, ..., -1.1213e+00,
-1.1954e+00, -5.0605e-01]],
[[-2.0331e+00, -3.8947e+00, -3.8424e+00, ..., -3.9857e+00,
-4.9466e+00, -1.9898e+00],
[-3.7151e+00, -5.8906e+00, -5.3280e+00, ..., -5.5117e+00,
-6.9222e+00, -2.9740e+00],
[-3.4181e+00, -5.1690e+00, -4.9971e+00, ..., -5.1076e+00,
-5.7309e+00, -2.8104e+00],
...,
[-1.6712e+00, -2.0352e+00, -1.7197e+00, ..., -2.7308e+00,
-3.1581e+00, -9.4842e-01],
[-2.7503e+00, -2.9203e+00, -1.9693e+00, ..., -2.8297e+00,
-3.8355e+00, -1.0010e+00],
[ 3.0885e-01, -2.8026e-01, 3.1056e-01, ..., 2.4511e-01,
4.1534e-01, 1.8797e-01]],
[[-4.6723e-01, 4.7388e-01, 8.4314e-01, ..., 8.0502e-01,
5.5095e-01, 1.4917e+00],
[-6.2739e-01, 5.9881e-02, 4.1115e-01, ..., 3.9167e-01,
1.8197e-01, 3.5143e-01],
[-1.1151e+00, 1.3722e-01, 1.3331e-01, ..., 1.9024e-01,
1.1961e-01, 3.0853e-01],
...,
[-5.8849e-01, 4.4693e-02, 2.5704e-01, ..., -1.5039e+00,
9.0442e-01, 4.3979e-01],
[-3.9266e-01, 6.2645e-02, 1.8694e-01, ..., -8.2867e-01,
3.8310e-01, 5.4927e-01],
[-1.6076e-01, -2.8515e-01, 1.3089e-01, ..., -7.4268e-02,
5.8770e-02, 3.2779e-01]],
...,
[[ 1.0713e+00, 3.7138e-01, 3.4172e-01, ..., 4.6648e-01,
2.2412e-01, -1.0215e+00],
[-6.9080e-01, -5.6040e-01, -3.3722e-01, ..., -4.1116e-01,
-1.2462e+00, -1.3522e+00],
[-2.5468e-01, -4.1937e-01, -2.8212e-01, ..., -3.1228e-01,
-5.5720e-01, -1.1698e+00],
...,
[-6.2626e-01, -5.2928e-01, -3.3514e-01, ..., -2.5869e+00,
-1.1010e+00, -5.7697e-01],
[-2.2211e+00, -1.7994e+00, -1.3024e+00, ..., -1.7247e+00,
-1.2890e+00, -1.1003e+00],
[-5.9635e-01, -8.0821e-01, -4.2497e-01, ..., -3.3903e-01,
-3.1848e-01, -3.8403e-01]],
[[ 1.5805e-01, 1.8097e-01, 3.0676e-01, ..., 2.6821e-01,
7.7353e-01, -2.3602e-01],
[ 3.4992e-01, -6.6632e-01, -7.4045e-03, ..., 1.5325e-01,
3.9669e-01, -7.6431e-01],
[ 8.8269e-03, -7.1860e-01, 1.1115e-01, ..., 2.5641e-01,
1.6167e-01, -6.9400e-01],
...,
[-1.9581e-01, -4.2228e-01, 2.3736e-01, ..., 7.2799e-01,
3.0111e-01, 6.0004e-02],
[-5.1276e-01, -8.1942e-01, 8.7154e-02, ..., 5.6068e-02,
-2.9096e-01, -1.7383e-01],
[ 3.1999e-01, 5.6421e-03, -1.8350e-01, ..., -2.4626e-01,
-1.2408e-01, 1.6477e-01]],
[[-6.4658e-01, -6.6959e-02, 3.3739e-01, ..., 3.6000e-01,
4.3527e-01, 9.9019e-01],
[-1.1451e+00, -6.5242e-01, -5.6458e-01, ..., -6.1375e-01,
-2.1927e-01, 1.0216e+00],
[-8.7641e-01, -6.3074e-01, -3.9279e-01, ..., -3.6083e-01,
-6.1630e-02, 7.6149e-01],
...,
[-6.9172e-01, -2.0792e-01, -8.9937e-02, ..., 1.5075e+00,
7.2842e-01, 6.0077e-01],
[-1.1758e+00, -5.8056e-01, -3.0070e-01, ..., 7.8801e-01,
3.2512e-01, 1.5708e-01],
[-3.5036e-01, -4.1401e-02, -1.6082e-01, ..., -2.0500e-01,
-1.9475e-01, 3.2637e-01]]],
[[[-7.6442e-01, -1.4346e+00, -1.4076e+00, ..., 2.1377e-03,
6.3053e-01, 5.5171e-01],
[-3.4219e-01, -2.3759e+00, -2.1541e+00, ..., -4.8357e-01,
-2.3404e-01, 1.1893e-01],
[-3.5806e-01, -2.3316e+00, -2.2027e+00, ..., -3.0435e-01,
-1.1362e-01, -9.5291e-02],
...,
[-4.6879e-01, -1.9004e+00, -1.8612e+00, ..., -3.3368e-01,
-8.8859e-02, -1.2704e-01],
[-9.8115e-01, -1.7206e+00, -1.1445e+00, ..., 1.4736e-02,
2.7274e-01, -1.7162e-02],
[-8.3612e-01, -1.6926e+00, -1.2739e+00, ..., -1.0333e+00,
-1.0883e+00, -3.7451e-01]],
[[-2.4066e+00, -4.7589e+00, -4.1228e+00, ..., -4.0466e-01,
-9.9035e-01, 1.5678e-02],
[-5.2292e+00, -7.9852e+00, -6.5467e+00, ..., -1.7573e+00,
-2.6341e+00, -6.5585e-01],
[-4.9134e+00, -7.2447e+00, -7.2067e+00, ..., -1.2793e+00,
-1.4914e+00, -3.8384e-01],
...,
[-4.3928e+00, -6.1552e+00, -5.6649e+00, ..., -1.2881e+00,
-1.5257e+00, -3.5306e-01],
[-6.3729e+00, -7.3132e+00, -6.3177e+00, ..., -2.4440e+00,
-3.3271e+00, -9.5973e-01],
[-2.0946e-02, -8.2071e-01, 4.2055e-01, ..., 3.5520e-01,
3.7340e-01, 7.9205e-02]],
[[-5.8849e-01, 6.2675e-01, 1.0656e+00, ..., 7.9992e-01,
2.8962e-01, 1.0815e+00],
[-9.4401e-01, 2.5641e-02, 4.4606e-01, ..., 3.9021e-01,
2.3938e-02, 4.7217e-01],
[-1.6285e+00, -4.5429e-02, 2.9045e-01, ..., 6.0940e-02,
7.3553e-02, 4.0640e-01],
...,
[-1.7569e+00, -5.8269e-02, -1.0306e-01, ..., 4.8546e-02,
8.5472e-02, 4.2071e-01],
[-1.5220e+00, 5.4029e-02, 1.1926e-02, ..., 1.3403e-01,
1.0173e-01, 5.9977e-01],
[-1.3814e+00, -5.2628e-01, -2.9386e-01, ..., -6.7258e-02,
7.1270e-02, 3.7201e-01]],
...,
[[ 1.1079e+00, -3.6300e-01, -2.0268e-01, ..., -1.0924e-01,
-3.8448e-01, -9.2371e-01],
[-7.4626e-01, -9.8064e-01, -8.3064e-01, ..., -6.8461e-01,
-1.2643e+00, -8.2437e-01],
[-6.2299e-01, -9.4475e-01, -5.7589e-01, ..., -1.0772e-01,
-1.7250e-01, -3.8664e-01],
...,
[-7.1914e-01, -1.1105e+00, -7.4063e-01, ..., -9.6487e-02,
-1.9915e-01, -3.6316e-01],
[-2.7550e+00, -2.7717e+00, -1.8006e+00, ..., -8.2336e-01,
-8.8682e-01, -7.8634e-01],
[-6.6186e-01, -1.7005e+00, -1.0306e+00, ..., 1.9615e-03,
-1.1786e-01, -3.2986e-01]],
[[ 1.3189e-02, -6.6330e-03, 5.6977e-01, ..., 3.2740e-01,
6.8178e-01, 2.9198e-01],
[ 1.7319e-02, -8.0959e-01, 1.4149e-01, ..., -4.3228e-02,
1.0170e-01, -4.4460e-02],
[ 9.9050e-02, -6.2005e-01, 3.4463e-01, ..., -1.5628e-01,
-8.5210e-02, 1.9650e-01],
...,
[ 8.4645e-02, -7.3995e-01, 2.9612e-01, ..., -1.3635e-01,
-1.1165e-01, 1.6615e-01],
[-2.4406e-01, -1.3916e+00, 9.1300e-02, ..., -3.6206e-01,
-3.1887e-01, -8.5690e-02],
[ 6.3795e-01, 1.6193e-01, 2.8347e-01, ..., -2.1586e-01,
-2.3355e-01, 2.9384e-01]],
[[-1.5347e+00, -5.8579e-01, 2.2244e-02, ..., 5.6114e-01,
3.5964e-01, 5.4760e-01],
[-2.0840e+00, -3.5289e-01, -4.5011e-02, ..., -5.3893e-02,
-1.6068e-01, 3.9973e-01],
[-1.7881e+00, -5.5075e-01, -9.0962e-02, ..., 4.1396e-02,
-1.8301e-01, 8.5585e-02],
...,
[-1.7640e+00, -4.9152e-01, 2.5672e-01, ..., 5.5895e-02,
-6.8209e-02, 9.1512e-02],
[-2.2235e+00, -1.1648e+00, -1.8844e-01, ..., -1.0149e-01,
-2.3191e-01, -2.1180e-01],
[-1.0734e+00, -2.3654e-01, 4.0327e-01, ..., -9.4940e-02,
-2.7382e-01, 9.7796e-02]]],
[[[-1.4195e-01, -7.7752e-01, -5.6520e-01, ..., 1.0474e+00,
1.5503e+00, 4.3806e-01],
[-3.8897e-02, -1.7647e+00, -9.6773e-01, ..., -1.7221e-01,
-8.4507e-02, 6.8843e-02],
[-4.2932e-01, -1.0977e+00, -2.0667e-01, ..., -1.5428e+00,
-1.3817e+00, -1.2771e-01],
...,
[-1.8035e-01, -1.3925e+00, -1.1964e+00, ..., -7.9241e-01,
-9.3326e-01, -2.0833e+00],
[-8.6565e-01, -1.4955e+00, -7.9734e-01, ..., -5.1459e-01,
-4.9121e-01, -1.8192e+00],
[-1.0275e+00, -1.8686e+00, -1.2768e+00, ..., -1.8981e+00,
-1.7562e+00, -1.0595e+00]],
[[-2.6602e+00, -4.7861e+00, -5.2168e+00, ..., -4.1495e+00,
-6.2313e+00, -2.9754e+00],
[-4.2059e+00, -6.3553e+00, -5.3733e+00, ..., -5.2450e+00,
-7.4036e+00, -2.8521e+00],
[-1.8534e+00, -2.4388e+00, -2.5124e+00, ..., -7.3625e+00,
-5.9742e+00, -2.4268e+00],
...,
[-3.7523e+00, -5.5256e+00, -5.8673e+00, ..., -6.7705e+00,
-6.5967e+00, -2.9146e+00],
[-5.8335e+00, -6.6003e+00, -6.0618e+00, ..., -7.1669e+00,
-7.9847e+00, -2.8337e+00],
[ 8.0463e-03, -6.6048e-01, 2.1738e-01, ..., 2.4821e-02,
6.0657e-01, 1.9135e-01]],
[[-4.4280e-01, 2.0399e-01, 8.7073e-01, ..., -2.5293e+00,
2.8761e-01, 1.8108e+00],
[-5.0059e-01, -8.1505e-01, 1.6419e-02, ..., -7.3756e+00,
-4.9891e+00, -3.8042e-01],
[-9.4726e-01, -1.8042e+00, -1.5108e+00, ..., -7.1952e+00,
-4.7841e+00, -1.8258e+00],
...,
[-1.3802e+00, 1.1371e-01, -1.7136e-01, ..., 2.5670e-01,
1.1490e-01, 2.5655e-01],
[-1.2451e+00, 1.3234e-01, -2.1403e-01, ..., 2.3341e-01,
5.6871e-02, 2.9449e-01],
[-1.0747e+00, 3.7657e-02, 2.5250e-01, ..., 9.5401e-02,
-6.6327e-02, -3.5196e-01]],
...,
[[ 9.5096e-01, 2.9837e-01, 5.4104e-01, ..., -2.2930e+00,
-2.4930e+00, -1.4029e+00],
[-5.9286e-01, -9.6314e-02, -7.3717e-02, ..., -6.6403e+00,
-5.5640e+00, -3.2721e+00],
[-6.2262e-01, -5.8398e-01, -4.9121e-01, ..., -6.0490e+00,
-5.3651e+00, -4.6968e+00],
...,
[-4.5378e-01, -7.0912e-01, -4.2314e-01, ..., -2.2232e+00,
-8.0033e-01, -1.1984e+00],
[-2.7344e+00, -2.8737e+00, -1.3972e+00, ..., -3.8366e+00,
-2.1875e+00, -2.1032e+00],
[-5.7816e-01, -1.6206e+00, -1.1581e+00, ..., -1.2710e+00,
-9.8262e-01, -1.3964e+00]],
[[ 2.7406e-01, 3.8940e-01, 2.5825e-01, ..., -8.7756e-01,
1.3369e+00, -3.9595e-01],
[ 3.1760e-02, -2.0969e-01, -1.8476e-01, ..., -1.7795e+00,
1.1462e-01, -6.3290e-01],
[-4.4966e-01, -6.8794e-01, -4.0281e-01, ..., -2.7756e+00,
-2.1542e+00, -1.1584e-01],
...,
[ 5.7880e-02, -8.1534e-01, 1.2918e-01, ..., -8.9385e-02,
3.5404e-02, -7.5287e-01],
[-3.1650e-01, -1.6521e+00, -2.3341e-01, ..., 3.5324e-01,
-2.3207e-01, -1.0160e+00],
[ 5.3386e-01, -9.7921e-02, 1.1489e-01, ..., 1.6534e-01,
1.8420e-01, 3.3241e-01]],
[[-9.1251e-01, -2.7606e-01, -3.1363e-01, ..., -4.7443e+00,
-2.6901e-01, 1.2296e+00],
[-1.4409e+00, -1.1469e+00, -2.1425e+00, ..., -9.0409e+00,
-2.9036e+00, 7.4424e-01],
[-1.0127e+00, -1.0680e+00, -2.0237e+00, ..., -9.6606e+00,
-4.0208e+00, 2.3261e-01],
...,
[-1.1632e+00, -9.9065e-01, -3.9267e-01, ..., 6.9650e-01,
7.0999e-02, 7.9854e-01],
[-1.8515e+00, -1.7403e+00, -9.4021e-01, ..., 2.2966e-01,
-3.0280e-01, 4.0475e-01],
[-8.9143e-01, -8.3837e-01, -7.6032e-02, ..., 2.8944e-01,
-1.1515e-01, 1.1556e+00]]],
[[[-6.9752e-01, -1.3936e+00, -1.3852e+00, ..., -8.7281e-04,
5.9093e-01, 5.3607e-01],
[-2.6835e-01, -2.3817e+00, -2.1493e+00, ..., -4.8356e-01,
-2.7443e-01, 1.2980e-01],
[-3.9222e-01, -2.1375e+00, -1.8961e+00, ..., -3.1817e-01,
-1.5225e-01, -1.1194e-01],
...,
[-4.7562e-01, -1.9213e+00, -1.7651e+00, ..., -3.2410e-01,
-5.6350e-02, -1.3063e-01],
[-9.6139e-01, -1.6872e+00, -1.1373e+00, ..., 1.8334e-02,
2.5192e-01, -2.8700e-02],
[-7.9753e-01, -1.5930e+00, -1.1734e+00, ..., -1.0432e+00,
-1.0592e+00, -3.7963e-01]],
[[-2.3883e+00, -4.8482e+00, -4.1206e+00, ..., -4.5031e-01,
-1.0328e+00, -1.9758e-02],
[-5.3508e+00, -8.1710e+00, -6.5039e+00, ..., -1.7895e+00,
-2.6556e+00, -6.7355e-01],
[-4.3983e+00, -6.4677e+00, -5.5056e+00, ..., -1.2949e+00,
-1.4945e+00, -3.4395e-01],
...,
[-4.3854e+00, -6.1205e+00, -5.6089e+00, ..., -1.2571e+00,
-1.5580e+00, -3.7477e-01],
[-6.3156e+00, -7.3378e+00, -6.1769e+00, ..., -2.4465e+00,
-3.2903e+00, -9.4650e-01],
[-2.2390e-02, -8.2191e-01, 3.9387e-01, ..., 3.2911e-01,
3.5091e-01, 9.6338e-02]],
[[-5.6500e-01, 6.3209e-01, 1.1267e+00, ..., 8.0729e-01,
2.9808e-01, 1.0679e+00],
[-8.9997e-01, -1.0658e-02, 5.0631e-01, ..., 3.8728e-01,
-1.3614e-02, 4.3030e-01],
[-1.6808e+00, -1.1943e-01, -2.5133e-02, ..., 2.3142e-02,
-3.1255e-02, 3.6813e-01],
...,
[-1.7362e+00, -5.1420e-02, -3.8254e-02, ..., 2.2001e-02,
5.9882e-02, 3.9451e-01],
[-1.5496e+00, 9.9081e-02, 1.0882e-01, ..., 9.6582e-02,
9.1768e-02, 5.8288e-01],
[-1.4060e+00, -4.8553e-01, -2.2423e-01, ..., -7.5444e-02,
6.8843e-02, 3.5080e-01]],
...,
[[ 1.0542e+00, -4.1870e-01, -2.3498e-01, ..., -1.0518e-01,
-3.8916e-01, -9.1704e-01],
[-7.9098e-01, -9.1853e-01, -8.3897e-01, ..., -6.5819e-01,
-1.2814e+00, -8.5400e-01],
[-6.4975e-01, -9.1303e-01, -7.2645e-01, ..., -7.3675e-02,
-2.0229e-01, -4.1315e-01],
...,
[-7.5031e-01, -1.1459e+00, -7.2180e-01, ..., -9.2891e-02,
-1.9369e-01, -3.6831e-01],
[-2.7365e+00, -2.7977e+00, -1.8454e+00, ..., -8.2334e-01,
-8.4906e-01, -7.8395e-01],
[-6.7084e-01, -1.6459e+00, -1.0469e+00, ..., -1.6925e-02,
-1.2850e-01, -3.3069e-01]],
[[ 2.7082e-02, -2.6567e-02, 5.6232e-01, ..., 3.2986e-01,
6.6173e-01, 2.7041e-01],
[ 2.5855e-02, -7.1235e-01, 1.0147e-01, ..., -3.3058e-02,
8.8321e-02, -6.1823e-02],
[-3.7206e-02, -7.9278e-01, 2.1186e-01, ..., -1.4857e-01,
-8.9960e-02, 1.5858e-01],
...,
[ 4.7410e-02, -7.8555e-01, 3.1212e-01, ..., -1.2373e-01,
-1.1301e-01, 1.6038e-01],
[-2.5112e-01, -1.4328e+00, 5.6075e-02, ..., -4.0839e-01,
-3.3950e-01, -5.7546e-02],
[ 6.6953e-01, 1.8452e-01, 3.3661e-01, ..., -2.5059e-01,
-2.1436e-01, 3.3094e-01]],
[[-1.5729e+00, -5.3933e-01, 6.4134e-02, ..., 5.6864e-01,
3.4378e-01, 5.1686e-01],
[-2.0487e+00, -2.6214e-01, 5.2007e-03, ..., -4.0056e-02,
-1.3671e-01, 3.5488e-01],
[-1.7336e+00, -3.1212e-01, 2.6008e-01, ..., 6.7451e-02,
-1.1154e-01, 3.3674e-02],
...,
[-1.7193e+00, -4.7198e-01, 2.5067e-01, ..., 7.8483e-02,
-1.0317e-01, 9.7108e-02],
[-2.1859e+00, -1.1155e+00, -1.4216e-01, ..., -1.3177e-01,
-2.7671e-01, -1.9019e-01],
[-1.0474e+00, -1.6482e-01, 4.5777e-01, ..., -1.1090e-01,
-2.5641e-01, 1.2814e-01]]],
[[[-7.1891e-01, -1.4824e+00, -1.4340e+00, ..., -1.2941e+00,
-8.8862e-01, -1.6603e+00],
[-3.3354e-01, -2.4757e+00, -2.3021e+00, ..., -1.7457e+00,
-1.8741e+00, -1.9064e+00],
[-1.3581e-01, -1.9201e+00, -1.7399e+00, ..., -1.4048e+00,
-1.4051e+00, -2.1110e+00],
...,
[-5.5559e-02, -7.0930e-01, -1.9892e-01, ..., -9.2889e-02,
-3.8786e-02, -4.4605e-01],
[-7.9324e-01, -7.4052e-01, 7.4709e-02, ..., 1.4587e-01,
2.7533e-01, -1.7100e-01],
[-9.7501e-01, -1.5910e+00, -1.2950e+00, ..., -1.0024e+00,
-1.0694e+00, -4.2873e-01]],
[[-2.6430e+00, -5.1150e+00, -4.6777e+00, ..., -4.8550e+00,
-6.0427e+00, -2.4224e+00],
[-5.3583e+00, -8.1268e+00, -7.4310e+00, ..., -7.7379e+00,
-9.3039e+00, -4.4147e+00],
[-4.0443e+00, -5.8219e+00, -5.7347e+00, ..., -6.3066e+00,
-6.4693e+00, -3.2498e+00],
...,
[-1.8208e+00, -2.4000e+00, -1.6584e+00, ..., -1.8608e+00,
-2.2168e+00, -8.7922e-01],
[-3.1891e+00, -3.5882e+00, -2.0772e+00, ..., -2.0731e+00,
-3.1782e+00, -9.3006e-01],
[ 2.0772e-01, -5.9358e-01, 2.1389e-01, ..., 2.8307e-01,
4.4967e-01, 2.2297e-01]],
[[-6.4531e-01, 4.5404e-01, 8.5994e-01, ..., 8.1851e-01,
6.0140e-01, 1.8795e+00],
[-9.3937e-01, 8.8525e-03, 3.8513e-01, ..., 5.4803e-01,
1.9116e-02, 1.3656e-01],
[-1.7666e+00, -2.1581e-01, -5.0873e-02, ..., -2.3789e-02,
-6.0194e-03, -9.9027e-02],
...,
[-7.7006e-01, -2.3527e-02, 2.4820e-01, ..., 1.9064e-01,
2.4522e-01, 4.0578e-01],
[-5.5281e-01, -1.1142e-01, 2.7630e-01, ..., 1.8563e-01,
2.0773e-01, 5.4111e-01],
[-2.3101e-01, -3.4497e-01, 4.3338e-02, ..., 1.3337e-02,
1.1291e-01, 3.3027e-01]],
...,
[[ 1.2328e+00, -3.4646e-01, 9.8699e-03, ..., 2.2380e-01,
-1.1425e-01, -2.1131e+00],
[-7.3535e-01, -9.0777e-01, -5.0707e-01, ..., -5.0319e-01,
-1.5041e+00, -2.0724e+00],
[-6.9157e-01, -8.3767e-01, -3.0044e-01, ..., -3.3930e-01,
-5.9155e-01, -1.3798e+00],
...,
[-8.2270e-01, -5.0494e-01, -3.3260e-01, ..., -3.2758e-01,
-4.9066e-01, -5.2852e-01],
[-2.7134e+00, -2.1140e+00, -1.4069e+00, ..., -1.2276e+00,
-1.2967e+00, -1.0421e+00],
[-7.1247e-01, -1.1383e+00, -4.7977e-01, ..., -2.7669e-01,
-3.4962e-01, -3.6379e-01]],
[[ 5.4315e-02, 3.4916e-02, 4.9468e-01, ..., 4.3377e-01,
8.9383e-01, -3.8443e-02],
[ 2.3383e-01, -7.9091e-01, -3.6699e-02, ..., 4.7263e-01,
4.0467e-01, -8.4293e-01],
[ 1.4220e-01, -6.3042e-01, 2.8970e-01, ..., 3.0147e-01,
5.7474e-01, -3.7824e-01],
...,
[-4.1758e-01, -5.5116e-01, 2.2887e-01, ..., 3.3714e-01,
3.0917e-01, 8.4383e-02],
[-7.8383e-01, -1.1003e+00, 2.2142e-02, ..., 2.9836e-02,
-1.3942e-02, -1.0001e-01],
[ 3.1336e-01, -3.1705e-02, -2.0293e-01, ..., -1.0139e-01,
-8.2051e-02, 1.8802e-01]],
[[-1.5716e+00, -7.9379e-01, -4.4493e-01, ..., -4.0888e-01,
-2.8797e-01, 8.6067e-01],
[-2.1466e+00, -7.6667e-01, -5.8939e-01, ..., -7.5022e-01,
-1.9332e-01, 1.6536e+00],
[-1.4913e+00, -5.8384e-01, -2.3076e-01, ..., -3.5504e-01,
-1.2000e-01, 1.3899e+00],
...,
[-6.7763e-01, -2.4655e-01, -1.4032e-01, ..., -1.2526e-01,
1.0633e-02, 6.3418e-01],
[-1.3852e+00, -8.4251e-01, -4.3259e-01, ..., -3.2278e-01,
-1.5655e-01, 2.1309e-01],
[-4.8447e-01, -1.7016e-01, -2.4915e-01, ..., -1.8352e-01,
-1.8719e-01, 3.2673e-01]]]], device='cuda:0'),
tensor([[[[-2.8041e-01, -4.2923e+00, -6.8216e+00, ..., 9.5784e-01,
-1.8854e-01, -9.3445e-01],
[ 1.8100e-01, -5.2052e+00, -7.2258e+00, ..., -1.5689e+00,
-9.2841e-01, -1.2908e+00],
[ 2.4887e-01, -3.5565e+00, -4.0545e+00, ..., -1.8658e+00,
-8.7856e-01, -1.8158e+00],
...,
[ 7.6233e-02, -6.8168e-01, -2.7382e-01, ..., 3.5186e-01,
4.8228e-01, -6.7396e-01],
[-3.0624e-01, -5.6793e-01, -3.8907e-02, ..., -4.3871e-01,
-1.8091e-02, -4.6836e-01],
[-5.6994e-01, -1.3340e+00, -1.0905e+00, ..., -1.0578e+00,
-1.1846e+00, -5.4123e-01]],
[[-4.0693e+00, -7.9625e+00, -8.0479e+00, ..., -5.7706e+00,
-5.1217e+00, -2.0840e+00],
[-6.7639e+00, -9.2201e+00, -1.2173e+01, ..., -7.0087e+00,
-6.9386e+00, -2.9956e+00],
[-7.3372e+00, -1.4712e+01, -1.8556e+01, ..., -6.1141e+00,
-5.8562e+00, -2.7939e+00],
...,
[-1.9323e+00, -2.5271e+00, -2.1526e+00, ..., -2.2753e+00,
-2.5641e+00, -1.3311e+00],
[-3.0623e+00, -3.2948e+00, -2.2927e+00, ..., -2.3613e+00,
-3.6305e+00, -1.2873e+00],
[ 3.8966e-01, -2.3039e-01, 4.0257e-01, ..., 3.6354e-01,
4.9688e-01, 2.0188e-01]],
[[-5.5327e-01, -3.6694e+00, -1.3215e+01, ..., 2.8415e+00,
5.9180e-01, 1.5411e+00],
[-5.1119e-01, -4.7703e+00, -2.0655e+01, ..., 1.9386e+00,
3.1884e-01, 5.6826e-01],
[-1.0331e+00, -6.4810e+00, -1.7871e+01, ..., 1.8707e+00,
1.8182e-01, 3.6694e-01],
...,
[-7.3326e-01, 1.2947e-01, 2.0181e-01, ..., 7.8167e-01,
6.1735e-01, 7.0559e-01],
[-5.3606e-01, 1.5362e-01, 1.7885e-01, ..., 3.9589e-01,
1.7302e-01, 5.5873e-01],
[-1.6378e-01, -1.6186e-01, 1.2306e-01, ..., -1.3521e-02,
1.1419e-01, 3.2252e-01]],
...,
[[ 1.6096e+00, 1.7665e+00, -9.7682e-01, ..., -2.5432e+00,
1.4974e-01, -1.0972e+00],
[ 3.0465e-01, 7.3774e-01, -3.7464e+00, ..., -4.4669e+00,
-1.1138e+00, -1.4003e+00],
[ 2.1190e-01, -6.8567e-01, -4.0582e+00, ..., -3.4922e+00,
-5.8866e-01, -1.1724e+00],
...,
[-6.7700e-01, -5.2173e-01, -4.5560e-01, ..., -3.4897e+00,
-1.5484e+00, -7.3891e-01],
[-2.2760e+00, -1.8388e+00, -1.3047e+00, ..., -1.9827e+00,
-1.3022e+00, -1.1137e+00],
[-6.4390e-01, -8.5839e-01, -4.3506e-01, ..., -3.2656e-01,
-4.0034e-01, -4.1750e-01]],
[[ 5.2362e-01, 1.2753e-01, -4.3492e+00, ..., 6.4187e-01,
6.1861e-01, -3.8308e-01],
[ 5.1925e-01, -2.2865e+00, -8.1465e+00, ..., -2.2048e+00,
3.3679e-01, -6.9293e-01],
[ 2.0473e-01, -2.8325e+00, -8.1163e+00, ..., -1.3863e+00,
2.1765e-01, -6.3201e-01],
...,
[-2.0045e-01, -5.2366e-01, 3.1431e-01, ..., 1.1117e+00,
6.0084e-01, -2.0476e-01],
[-5.2930e-01, -1.0447e+00, 9.5752e-03, ..., -7.8057e-01,
-4.6612e-01, -5.3448e-01],
[ 3.7843e-01, -9.3319e-02, -1.5007e-01, ..., -7.8164e-02,
-3.7318e-02, 1.5649e-01]],
[[-1.5473e+00, -3.0703e+00, -5.7270e+00, ..., 2.3642e-01,
3.5877e-01, 8.6341e-01],
[-2.2946e+00, -2.8825e+00, -3.1049e+00, ..., 2.9587e-01,
-2.9286e-01, 9.2265e-01],
[-1.6007e+00, -1.3403e+00, -2.6845e+00, ..., 6.5414e-01,
-6.2417e-02, 7.9864e-01],
...,
[-7.1364e-01, -3.9996e-01, -2.4387e-01, ..., 3.8116e-01,
3.4771e-01, 6.6295e-01],
[-1.1297e+00, -6.9482e-01, -4.1116e-01, ..., 7.0608e-01,
1.8991e-01, 2.0484e-01],
[-3.7086e-01, -1.3286e-01, -1.5691e-01, ..., -2.3496e-01,
-2.0139e-01, 4.1020e-01]]],
[[[-4.2179e+00, -2.5565e+00, -1.6769e+00, ..., -7.4853e-01,
-1.8982e-01, -1.0810e+00],
[-6.2603e+00, -7.5417e+00, -7.0931e+00, ..., -1.0933e+00,
-1.1386e+00, -1.3128e+00],
[-9.5176e+00, -1.3425e+01, -1.3807e+01, ..., -9.4853e-01,
-3.6710e-01, -1.5596e+00],
...,
[-2.2911e-01, -9.7798e-01, -7.5746e-01, ..., -1.1954e+01,
-1.1883e+01, -7.2944e+00],
[-3.2063e-01, -1.1767e+00, -8.1548e-01, ..., -9.2205e-01,
-1.6195e+00, -3.2270e+00],
[-2.4613e-01, -1.5256e+00, -1.4958e+00, ..., 6.0782e-01,
-4.9664e-01, -2.2316e+00]],
[[-1.0144e+01, -7.6558e+00, -3.6551e+00, ..., -4.4730e+00,
-5.8467e+00, -2.0862e+00],
[-5.8898e+00, -8.8226e+00, -9.3690e+00, ..., -7.1103e+00,
-7.9448e+00, -3.1862e+00],
[-9.1725e+00, -1.5806e+01, -1.8391e+01, ..., -5.3632e+00,
-5.7271e+00, -2.1892e+00],
...,
[-2.6501e+00, -5.5798e+00, -4.1863e+00, ..., -1.1379e+01,
-1.1936e+01, 6.4569e-01],
[-4.1562e+00, -5.9716e+00, -4.7364e+00, ..., -5.2545e+00,
-4.4019e+00, -1.6643e+00],
[ 5.3367e-01, -5.0346e-01, 4.1174e-01, ..., -1.9483e+00,
-1.6124e+00, -2.5640e+00]],
[[-1.5482e+01, -1.4235e+01, -1.6573e+01, ..., 8.6695e-01,
8.6482e-01, 1.6736e+00],
[-2.5618e+01, -3.3695e+01, -3.3274e+01, ..., -3.0473e-02,
3.5895e-01, 6.4781e-01],
[-3.0300e+01, -4.3875e+01, -4.5084e+01, ..., -5.9393e-01,
-1.8141e-01, -1.0124e-01],
...,
[-1.3405e+00, -1.2614e-01, -1.7068e-01, ..., -2.0508e+01,
-1.6730e+01, -8.4115e+00],
[-1.1421e+00, 3.3143e-01, -4.5917e-01, ..., -4.6033e+00,
-6.2368e+00, -1.6372e+00],
[-4.3848e-01, 1.4994e-01, -8.1330e-01, ..., -2.9194e-01,
-1.0398e+00, -2.5630e+00]],
...,
[[-4.3295e+00, -9.8651e+00, -7.6189e+00, ..., 9.3178e-02,
-1.2108e-01, -1.2760e+00],
[-5.9615e+00, -1.2631e+01, -1.1713e+01, ..., -8.0370e-01,
-1.5044e+00, -1.4978e+00],
[-2.7084e+00, -7.9709e+00, -1.7175e+00, ..., -8.0298e-01,
-1.0449e+00, -1.3615e+00],
...,
[-1.0333e+00, -7.9802e-01, -1.0328e+00, ..., -8.3225e-03,
-5.2609e+00, -9.3379e+00],
[-2.5198e+00, -1.6529e+00, -1.0937e+00, ..., 1.2096e+00,
8.3568e-01, -4.3524e+00],
[-7.4387e-02, -9.3196e-01, -1.5743e-01, ..., -4.4704e+00,
-4.3657e+00, -2.0671e+00]],
[[-4.9977e+00, -6.2395e+00, -7.9278e+00, ..., -1.3267e-01,
6.3142e-01, -4.3549e-01],
[-6.3008e+00, -8.1609e+00, -1.1716e+01, ..., 9.1191e-01,
2.6875e-01, -6.4327e-01],
[-6.6059e+00, -1.1382e+01, -1.0767e+01, ..., 6.5800e-01,
5.6920e-01, -3.5310e-01],
...,
[-4.8420e-01, -1.0858e+00, -9.6387e-01, ..., -3.2934e+00,
-3.6764e+00, -3.4875e+00],
[-6.2930e-01, -1.6278e+00, -5.5151e-01, ..., 4.3010e-01,
1.3010e+00, 3.1476e+00],
[ 6.6340e-01, 1.0831e-01, 3.6684e-01, ..., -8.6441e-01,
-6.7799e-01, -1.8406e+00]],
[[-1.8617e+01, -1.4856e+01, -1.9240e+01, ..., 1.0108e-01,
2.1551e-01, 7.7037e-01],
[-1.7800e+01, -1.4425e+01, -2.0076e+01, ..., -3.2862e-01,
-2.2559e-01, 7.4407e-01],
[-7.2591e+00, -1.2319e+01, -1.7919e+01, ..., -4.4999e-01,
-2.9664e-01, 7.2783e-01],
...,
[-8.1880e-01, -5.7301e-01, 1.5375e-01, ..., -7.8514e+00,
-9.8936e+00, -1.0390e+01],
[-1.1948e+00, -1.4937e+00, -6.2860e-02, ..., 1.5053e+00,
5.0275e-01, -2.1042e+00],
[-2.9067e-01, -7.3529e-01, 3.9375e-01, ..., 2.1240e+00,
4.2361e+00, 3.3813e+00]]],
[[[-1.3393e-01, -5.0778e-01, -2.3127e-01, ..., 6.1683e-02,
3.1204e-01, -5.7735e-02],
[ 3.0190e-01, -1.2486e+00, -7.4043e-01, ..., -5.3198e-01,
-2.8597e-01, -1.6792e-01],
[ 1.4349e-01, -9.4894e-01, -3.6086e-01, ..., -3.3996e-01,
-1.9800e-01, -5.0135e-01],
...,
[-2.3882e-01, -1.6684e+00, -1.2402e+00, ..., -9.6939e-01,
-9.5936e-01, -2.0525e+00],
[-1.0374e+00, -1.6988e+00, -7.9646e-01, ..., -5.3083e-01,
-4.4943e-01, -1.7889e+00],
[-1.2321e+00, -2.1411e+00, -1.6132e+00, ..., -1.4407e+00,
-1.5448e+00, -9.6461e-01]],
[[-1.4502e+00, -3.1209e+00, -2.5417e+00, ..., -2.7113e+00,
-3.3499e+00, -1.3286e+00],
[-3.2018e+00, -5.0947e+00, -3.5807e+00, ..., -4.0494e+00,
-4.7300e+00, -2.0389e+00],
[-2.0673e+00, -3.0798e+00, -2.2122e+00, ..., -2.6188e+00,
-2.6379e+00, -1.2545e+00],
...,
[-3.7106e+00, -5.4713e+00, -5.1413e+00, ..., -5.5018e+00,
-6.2185e+00, -2.9209e+00],
[-6.1432e+00, -7.0329e+00, -6.1086e+00, ..., -6.1394e+00,
-7.4530e+00, -2.7173e+00],
[ 1.5439e-04, -8.2485e-01, 2.6851e-01, ..., 3.8150e-01,
6.8085e-01, 1.9489e-01]],
[[-4.7223e-01, 1.7967e-01, 8.2022e-01, ..., 7.1770e-01,
5.8791e-01, 1.3501e+00],
[-3.5800e-01, -6.1701e-02, 5.7478e-01, ..., 5.1310e-01,
4.1219e-01, 3.1181e-01],
[-8.1030e-01, 5.9369e-02, 3.0431e-01, ..., 4.2985e-01,
3.6066e-01, 4.3123e-01],
...,
[-1.3678e+00, 2.6878e-02, 4.6861e-02, ..., 8.4178e-02,
3.9434e-02, 3.0881e-01],
[-1.2367e+00, 1.6174e-01, 2.5787e-02, ..., -1.1353e-02,
3.0734e-02, 3.2230e-01],
[-1.1779e+00, -7.6841e-02, 1.0255e-01, ..., -3.1005e-02,
-5.1447e-02, -3.3742e-01]],
...,
[[ 8.5115e-01, 2.0385e-01, 2.1050e-01, ..., -2.7764e-01,
9.8758e-02, -8.8686e-01],
[-1.1980e+00, -8.6411e-01, -6.9694e-01, ..., -1.2740e+00,
-1.5229e+00, -1.0879e+00],
[-6.3339e-01, -2.7716e-01, -2.4989e-01, ..., -4.9266e-01,
-4.9663e-01, -5.9466e-01],
...,
[-5.8198e-01, -6.1984e-01, -3.1558e-01, ..., -3.8214e-01,
-5.8258e-01, -1.1733e+00],
[-3.0546e+00, -2.9245e+00, -2.0451e+00, ..., -1.9194e+00,
-2.0059e+00, -2.0397e+00],
[-5.2858e-01, -1.6096e+00, -1.0904e+00, ..., -8.6166e-01,
-9.4761e-01, -1.3506e+00]],
[[ 4.7990e-02, 2.8004e-01, 6.9927e-01, ..., 9.4625e-01,
1.0148e+00, 3.5424e-01],
[-1.1044e-02, -4.3160e-01, 1.1881e-01, ..., 5.6086e-01,
5.5257e-01, -1.9258e-01],
[-2.8813e-01, -6.2353e-01, 1.4361e-01, ..., 1.5066e-01,
3.2892e-01, 4.1072e-04],
...,
[-9.7629e-02, -9.9831e-01, 3.8375e-02, ..., 2.3437e-01,
1.3569e-01, -7.1959e-01],
[-4.8665e-01, -1.8375e+00, -2.2074e-01, ..., 4.2084e-02,
-2.7010e-02, -9.3009e-01],
[ 4.5567e-01, -1.6713e-01, 1.1058e-01, ..., 2.6731e-01,
3.2124e-01, 3.4027e-01]],
[[-9.8931e-01, -2.6664e-01, 1.3496e-01, ..., 1.6361e-01,
7.5768e-02, 7.3681e-01],
[-1.0972e+00, -1.9196e-01, -3.0773e-01, ..., -2.3717e-01,
-5.6947e-02, 1.1457e+00],
[-7.0651e-01, -1.9737e-01, -2.3228e-01, ..., -1.8533e-01,
-9.7157e-02, 6.7158e-01],
...,
[-1.0741e+00, -9.0523e-01, -4.5733e-01, ..., -3.6495e-01,
-6.7819e-02, 8.2345e-01],
[-1.7818e+00, -1.7719e+00, -9.9029e-01, ..., -7.8942e-01,
-4.6590e-01, 4.3607e-01],
[-9.4436e-01, -9.4521e-01, -2.6669e-01, ..., -2.8338e-01,
-1.4945e-01, 1.1354e+00]]],
[[[-6.0925e-01, -1.4145e+00, -1.4436e+00, ..., -1.2745e+00,
-9.3176e-01, -1.8014e+00],
[-2.5648e-01, -2.5788e+00, -2.2773e+00, ..., -1.8637e+00,
-1.8965e+00, -1.9611e+00],
[-1.8791e-01, -2.0957e+00, -1.4232e+00, ..., -1.3511e+00,
-1.4486e+00, -2.1556e+00],
...,
[ 8.7396e-02, -5.9661e-01, -2.3190e-01, ..., -7.5936e-01,
6.0239e-02, -5.6219e-01],
[-3.1681e-01, -4.6429e-01, 5.0190e-02, ..., -2.2649e-01,
-2.6345e-01, -3.0508e-01],
[-6.0159e-01, -1.3355e+00, -1.0979e+00, ..., -1.1421e+00,
-1.2175e+00, -5.1924e-01]],
[[-2.5956e+00, -4.9944e+00, -4.9303e+00, ..., -5.0807e+00,
-6.1254e+00, -2.5070e+00],
[-5.4794e+00, -8.2899e+00, -7.7700e+00, ..., -8.1290e+00,
-9.1617e+00, -4.3890e+00],
[-4.0709e+00, -5.8389e+00, -7.2787e+00, ..., -7.7105e+00,
-6.4789e+00, -3.2515e+00],
...,
[-1.7898e+00, -2.2909e+00, -1.9152e+00, ..., -2.0742e+00,
-3.1009e+00, -1.1713e+00],
[-2.8054e+00, -3.0228e+00, -2.0554e+00, ..., -2.3040e+00,
-3.5959e+00, -1.1941e+00],
[ 3.3818e-01, -2.3372e-01, 3.4344e-01, ..., 2.1408e-01,
4.2793e-01, 2.1542e-01]],
[[-7.0747e-01, 3.3706e-01, 7.2249e-01, ..., 9.0004e-01,
6.1834e-01, 1.8879e+00],
[-9.8128e-01, -1.7557e-01, 5.8908e-01, ..., 6.2118e-01,
-3.8690e-02, 6.6816e-02],
[-1.7491e+00, -3.8434e-01, 1.4696e-01, ..., 7.5823e-01,
-9.3020e-02, -2.0952e-01],
...,
[-6.7483e-01, 6.8823e-02, 2.6385e-01, ..., -3.2388e+00,
2.0645e-01, 4.8941e-01],
[-4.6707e-01, 8.9199e-02, 2.2147e-01, ..., -2.4903e-01,
2.4417e-01, 5.5511e-01],
[-1.6927e-01, -2.3659e-01, 1.4329e-01, ..., -1.2107e-01,
6.8082e-02, 3.4085e-01]],
...,
[[ 1.1051e+00, -4.1735e-01, 3.6796e-01, ..., 2.1658e-01,
-3.7936e-02, -2.0919e+00],
[-9.3025e-01, -8.6850e-01, -3.8693e-01, ..., -5.6190e-01,
-1.4125e+00, -2.0815e+00],
[-8.0592e-01, -8.0102e-01, -1.6457e-01, ..., -4.3048e-01,
-5.7244e-01, -1.3903e+00],
...,
[-6.7943e-01, -5.3332e-01, -3.8481e-01, ..., -4.2855e+00,
-2.6230e+00, -7.7410e-01],
[-2.2288e+00, -1.7849e+00, -1.2741e+00, ..., -2.8560e+00,
-1.5587e+00, -1.1279e+00],
[-6.1467e-01, -8.2559e-01, -4.5131e-01, ..., -3.7797e-01,
-3.2183e-01, -3.7077e-01]],
[[-3.4698e-02, -2.7888e-02, 5.7488e-01, ..., 5.6348e-01,
9.2479e-01, -4.0001e-02],
[ 8.7104e-03, -7.9698e-01, 1.0416e-01, ..., 2.5460e-01,
4.0168e-01, -8.0631e-01],
[-1.0438e-01, -7.7941e-01, 7.7795e-01, ..., -8.0235e-02,
5.6317e-01, -3.7632e-01],
...,
[-2.3358e-01, -5.0145e-01, 2.3031e-01, ..., 5.5167e-01,
6.6398e-01, 7.7078e-02],
[-5.0501e-01, -9.2068e-01, 8.4433e-02, ..., 8.5158e-01,
-7.9893e-02, -2.5431e-01],
[ 3.1791e-01, -5.0911e-02, -1.8360e-01, ..., -2.1869e-01,
-1.5275e-01, 1.6944e-01]],
[[-1.6704e+00, -8.3054e-01, -6.6931e-01, ..., -4.2093e-01,
-3.1781e-01, 8.4820e-01],
[-2.2180e+00, -8.0811e-01, -1.0079e+00, ..., -7.2308e-01,
-1.8969e-01, 1.6467e+00],
[-1.5538e+00, -6.9058e-01, -7.1714e-01, ..., -5.2279e-01,
-1.4185e-01, 1.4459e+00],
...,
[-6.6581e-01, -2.3090e-01, -1.4618e-01, ..., -4.3416e-01,
1.2070e+00, 8.9465e-01],
[-1.1038e+00, -5.6735e-01, -3.3985e-01, ..., 2.5092e-01,
6.5800e-01, 3.8802e-01],
[-3.4800e-01, -1.5113e-02, -1.7775e-01, ..., -1.2748e-01,
-1.9548e-01, 3.4086e-01]]],
[[[-1.4428e-01, -5.3535e-01, -2.1199e-01, ..., -6.6458e-02,
6.8572e-01, -5.8521e-02],
[ 2.5961e-01, -1.4393e+00, -1.4406e+00, ..., -1.2232e+00,
-2.9334e-01, -1.5966e-01],
[ 1.2453e-01, -1.5540e+00, -1.7812e+00, ..., -1.0761e+00,
-5.2063e-01, -5.3019e-01],
...,
[-1.7690e-01, -1.5244e+00, -1.2504e+00, ..., -1.0635e+00,
-1.0220e+00, -2.0818e+00],
[-8.6569e-01, -1.5170e+00, -7.4225e-01, ..., -6.4458e-01,
-4.3777e-01, -1.8588e+00],
[-1.0457e+00, -1.8582e+00, -1.4147e+00, ..., -1.5883e+00,
-1.7019e+00, -1.1061e+00]],
[[-1.4547e+00, -3.0222e+00, -2.5862e+00, ..., -3.1393e+00,
-4.7726e+00, -1.3142e+00],
[-3.1898e+00, -6.0342e+00, -5.9405e+00, ..., -4.2937e+00,
-6.1129e+00, -2.0313e+00],
[-2.0858e+00, -4.6033e+00, -4.6707e+00, ..., -3.0216e+00,
-3.6821e+00, -1.2960e+00],
...,
[-3.7275e+00, -5.3763e+00, -5.8890e+00, ..., -5.9961e+00,
-6.3696e+00, -3.0242e+00],
[-5.8918e+00, -6.5675e+00, -6.0665e+00, ..., -6.4435e+00,
-7.7542e+00, -2.8722e+00],
[ 2.1462e-02, -6.1126e-01, 1.6394e-01, ..., 4.0589e-01,
6.7036e-01, 9.2728e-02]],
[[-4.3865e-01, 2.0965e-01, 8.4390e-01, ..., 8.4173e-01,
1.3869e+00, 1.3599e+00],
[-3.2732e-01, -4.4525e-01, -2.1619e-01, ..., -2.0910e+00,
1.1777e+00, 3.2904e-01],
[-7.6313e-01, -1.0169e+00, -2.0969e+00, ..., -2.0817e+00,
6.7609e-01, 4.4335e-01],
...,
[-1.3474e+00, 6.8418e-02, 4.7117e-01, ..., 9.6900e-02,
-5.6376e-02, 2.2244e-01],
[-1.2514e+00, 1.2607e-01, 2.9050e-01, ..., -9.7519e-02,
-1.3986e-01, 2.4273e-01],
[-1.1050e+00, -2.9374e-02, 2.9341e-01, ..., 2.3790e-02,
-1.6720e-01, -4.1956e-01]],
...,
[[ 9.3605e-01, 2.7066e-01, 2.2947e-01, ..., -1.4373e+00,
-4.9013e-01, -8.9176e-01],
[-1.1030e+00, -5.8606e-01, -4.8726e-01, ..., -2.5274e+00,
-2.3141e+00, -1.1083e+00],
[-5.6443e-01, 4.2286e-01, 2.5664e-01, ..., -1.9096e+00,
-1.2630e+00, -6.5004e-01],
...,
[-4.4831e-01, -7.2474e-01, -3.7255e-01, ..., -5.9002e-01,
-5.8453e-01, -1.2132e+00],
[-2.7125e+00, -2.8129e+00, -1.8140e+00, ..., -2.0257e+00,
-2.1325e+00, -2.1679e+00],
[-5.3223e-01, -1.6164e+00, -1.0408e+00, ..., -9.5842e-01,
-1.0216e+00, -1.4240e+00]],
[[ 4.6575e-02, 2.9161e-01, 7.2761e-01, ..., 1.0320e+00,
1.4390e+00, 3.6992e-01],
[ 4.2945e-02, -5.0577e-01, 2.2312e-01, ..., 5.1939e-01,
5.4908e-01, -1.7819e-01],
[-2.6468e-01, -1.0669e+00, -3.5549e-01, ..., 3.9546e-01,
1.0848e-01, -2.2211e-02],
...,
[ 4.6520e-02, -9.1440e-01, 7.4849e-02, ..., 2.2954e-01,
8.1467e-02, -8.2827e-01],
[-3.3115e-01, -1.6953e+00, -2.2320e-01, ..., -3.2222e-02,
9.5583e-03, -1.0132e+00],
[ 5.3157e-01, -7.2216e-02, 1.8642e-01, ..., 1.9454e-01,
2.8622e-01, 2.2846e-01]],
[[-9.6186e-01, -2.2813e-01, 1.3454e-01, ..., -1.5100e-01,
1.3677e-01, 7.2860e-01],
[-1.1422e+00, -4.4885e-01, -1.3201e+00, ..., 3.4749e-01,
3.4751e-01, 1.1420e+00],
[-7.3110e-01, -9.0185e-01, -2.9292e+00, ..., 2.1158e-01,
4.7377e-01, 6.8049e-01],
...,
[-1.1148e+00, -8.8493e-01, -8.7281e-01, ..., -9.8385e-02,
9.5158e-03, 8.6834e-01],
[-1.7622e+00, -1.6385e+00, -1.0793e+00, ..., -7.1356e-01,
-4.6394e-01, 4.2755e-01],
[-9.1924e-01, -8.4454e-01, -2.2658e-01, ..., -3.8185e-01,
-1.5125e-01, 1.1159e+00]]]], device='cuda:0'),
tensor([[[[-1.8698e-01, -3.7362e-01, 2.1170e-02, ..., -8.6449e-02,
3.5429e-01, 4.2164e-02],
[ 1.3987e-02, -2.0962e+00, -1.8548e+00, ..., -8.0785e-01,
-5.5663e-01, -4.8756e-01],
[ 1.4956e-01, -2.5075e+00, -4.1715e+00, ..., -6.1371e-01,
-5.2746e-01, -8.6158e-01],
...,
[-1.3966e-01, -1.8630e+00, -1.9447e+00, ..., -1.4458e+00,
-3.9551e-02, -1.1801e+00],
[-2.4247e-01, -1.3392e+00, -9.0021e-01, ..., -4.0005e+00,
-4.0381e-01, -1.2285e+00],
[-1.5371e-01, -1.1657e+00, -7.8802e-02, ..., -4.3010e+00,
-2.9847e+00, -1.5852e+00]],
[[-4.6601e-01, -1.8189e+00, -1.0367e+00, ..., -1.0830e+00,
-1.7593e+00, -4.5472e-01],
[-3.7290e+00, -7.0931e+00, -7.4380e+00, ..., -2.9245e+00,
-3.9336e+00, -1.4075e+00],
[-3.7438e+00, -7.1393e+00, -9.2203e+00, ..., -3.2062e+00,
-3.6050e+00, -1.5155e+00],
...,
[-3.9456e+00, -5.4466e+00, -7.7371e+00, ..., -9.5853e+00,
-7.0165e+00, -4.0874e+00],
[-5.5003e+00, -6.4286e+00, -1.3018e+01, ..., -9.4367e+00,
-1.0665e+01, -5.0596e+00],
[ 4.5560e-01, -3.4735e-01, 1.4640e+00, ..., 6.6887e-01,
-2.3868e-01, 9.4344e-02]],
[[-5.0326e-01, -2.7823e-02, 5.1131e-01, ..., 5.1141e-01,
2.1882e-01, 1.0154e+00],
[-4.0168e-01, -2.6885e-01, 7.9235e-01, ..., 4.4803e-01,
3.6349e-01, 4.9038e-01],
[-1.1051e+00, -2.1032e+00, -2.2930e+00, ..., 4.6580e-02,
2.3390e-01, 4.2083e-01],
...,
[-1.8933e+00, -3.0690e-01, -1.0731e+00, ..., -6.7227e+00,
-1.8817e+00, 3.5280e-01],
[-1.6187e+00, -1.8337e-01, 5.2979e-01, ..., -1.9755e+00,
-8.9034e-01, 7.5336e-01],
[-8.5332e-01, -1.3396e-01, 2.3484e-01, ..., -2.6623e+00,
-1.5549e-01, 4.5331e-01]],
...,
[[ 1.1948e-01, -4.7161e-01, -2.8353e-01, ..., 7.2957e-03,
-1.6413e-01, -7.3090e-01],
[-7.6766e-01, -4.0609e-01, -1.7435e-01, ..., -2.6500e-01,
-1.0353e+00, -8.1975e-01],
[ 3.6847e-01, 1.7046e+00, 1.1460e+00, ..., 5.2416e-02,
-2.0533e-01, -5.7590e-01],
...,
[-9.3825e-01, -1.0155e+00, 1.5561e+00, ..., -8.1970e+00,
-4.3698e+00, -2.6523e+00],
[-2.3377e+00, -1.9935e+00, -1.9033e+00, ..., -1.1975e+01,
-8.5855e+00, -3.8251e+00],
[-5.7537e-01, -1.0126e+00, 2.3249e+00, ..., -3.2566e+00,
-3.9824e+00, -1.3397e+00]],
[[-1.4654e-01, 2.6910e-02, 3.6094e-01, ..., 6.0220e-02,
5.2368e-01, 1.2568e-01],
[-7.3736e-02, 2.4227e-01, 4.3113e-01, ..., 1.3102e-02,
1.6943e-01, -1.5843e-01],
[-1.9939e-01, 1.0824e-01, 3.7179e-01, ..., 8.3564e-02,
3.3205e-02, -1.1148e-01],
...,
[ 1.5718e-01, -6.7890e-01, 9.0374e-02, ..., -3.5289e+00,
-2.0550e+00, -1.2445e+00],
[-4.6803e-01, -1.7444e+00, 9.5894e-02, ..., -4.1246e-01,
-1.6738e-02, -1.9231e+00],
[ 8.2493e-01, 1.1714e-01, -1.1858e-02, ..., 7.8323e-01,
1.9721e+00, -3.0781e-01]],
[[-6.1052e-01, 4.0526e-01, 6.0064e-01, ..., 4.9612e-01,
4.1182e-01, 6.9243e-01],
[-9.6385e-01, 3.3966e-02, -7.9878e-02, ..., -2.3621e-01,
-8.9832e-02, 7.7041e-01],
[-1.1670e+00, -1.2548e+00, -2.8648e+00, ..., 1.3606e-01,
2.0443e-01, 6.8524e-01],
...,
[-1.4551e+00, -6.5596e-01, -2.7867e+00, ..., -4.4835e+00,
-8.5992e-01, 1.7588e+00],
[-1.3891e+00, -8.2504e-01, -1.8892e+00, ..., -6.0227e+00,
-4.8722e-01, 1.4245e+00],
[-6.2937e-01, -3.8155e-01, -1.0595e+00, ..., -4.2035e+00,
3.2563e-01, 1.6584e+00]]],
[[[-1.3091e+00, -8.8097e-01, -3.2082e-01, ..., 2.8074e-01,
1.5168e-01, 1.3781e-01],
[-2.2604e-01, -8.8248e-01, 8.0146e-02, ..., 4.2662e-01,
-1.6469e-01, 1.1457e-01],
[ 1.4767e-01, -1.9814e+00, 4.0551e-02, ..., -9.0865e-01,
4.2127e-01, 4.0079e-02],
...,
[-7.2874e-01, -2.9655e+00, -1.0191e+00, ..., -1.1553e+00,
-1.0368e+00, -2.1110e+00],
[ 2.7396e-01, -2.6277e+00, -2.3884e+00, ..., -5.4817e-01,
-4.7973e-01, -1.8418e+00],
[-1.0680e+00, -2.1360e+00, -1.5589e+00, ..., -1.5117e+00,
-1.6107e+00, -1.0195e+00]],
[[-1.1634e+00, -3.3339e+00, -2.0256e+00, ..., -4.0166e+00,
-2.5005e+00, -1.3209e+00],
[-2.7148e+00, -4.4840e+00, -2.4897e+00, ..., -6.7002e+00,
-5.0022e+00, -2.7349e+00],
[-2.9825e+00, -3.3711e+00, -2.7430e+00, ..., -6.0710e+00,
-6.1618e+00, -2.5598e+00],
...,
[-4.4471e+00, -1.0882e+01, -9.2287e+00, ..., -5.6683e+00,
-6.3339e+00, -2.9581e+00],
[-5.8603e+00, -8.7760e+00, -8.2202e+00, ..., -6.2990e+00,
-7.6286e+00, -2.7693e+00],
[-1.5852e-01, -8.3776e-01, 1.9675e-01, ..., 3.5477e-01,
6.5289e-01, 1.9451e-01]],
[[-1.9530e+00, -1.8513e+00, -1.3675e+00, ..., 3.3174e-01,
4.0469e-01, 1.3329e+00],
[-1.5117e+00, -1.7700e+00, -1.8668e+00, ..., -2.1459e+00,
-1.0328e+00, 6.0985e-01],
[-1.6947e+00, -8.5705e-01, -1.4906e+00, ..., -5.3512e+00,
-4.2430e+00, 1.5928e-01],
...,
[-8.2158e+00, -6.8060e+00, -9.2643e-01, ..., -1.1676e-02,
1.2496e-03, 2.8561e-01],
[-1.7345e+00, -8.5087e-01, 3.7416e+00, ..., -2.1342e-02,
9.9560e-03, 2.8553e-01],
[-6.3241e-01, -1.8248e-01, 9.1306e-01, ..., -5.5001e-02,
-7.2073e-02, -3.6701e-01]],
...,
[[-4.7183e-01, -1.1943e+00, -1.2665e+00, ..., 1.7038e-02,
-1.2685e-01, -5.6031e-01],
[-5.5421e-01, -1.7756e+00, -1.7704e+00, ..., -2.9228e+00,
-2.7183e+00, -1.7512e+00],
[-5.6568e-01, -9.7312e-01, -1.9531e+00, ..., -5.6355e+00,
-3.8577e+00, -2.8447e+00],
...,
[ 9.3539e-01, -6.8717e+00, -5.1847e+00, ..., -3.7899e-01,
-6.0786e-01, -1.1840e+00],
[-4.1552e+00, -5.7891e+00, -6.5274e+00, ..., -1.9495e+00,
-2.0629e+00, -2.1064e+00],
[-8.2672e-01, -1.8605e+00, -2.5896e+00, ..., -8.9058e-01,
-9.3020e-01, -1.3712e+00]],
[[-9.0959e-01, -9.3684e-01, -9.1445e-01, ..., -7.4015e-01,
-3.1264e-01, -4.6350e-01],
[-4.8836e-01, -6.1584e-01, -1.1622e+00, ..., -1.8059e+00,
-7.0080e-01, -4.5109e-01],
[-6.3137e-01, -1.0002e+00, -9.9792e-01, ..., -2.6518e+00,
-7.6610e-01, 9.4282e-02],
...,
[-3.5295e+00, -1.5015e+00, 9.0085e-02, ..., 1.6223e-02,
8.1944e-02, -7.3784e-01],
[-1.7197e+00, -4.5316e+00, -2.8894e+00, ..., 4.1711e-02,
-3.5346e-02, -9.4995e-01],
[ 5.7960e-01, -4.2800e-01, -5.4315e-01, ..., 2.5880e-01,
3.2032e-01, 3.5013e-01]],
[[-1.3649e+00, -1.2775e+00, -1.0245e+00, ..., -3.3409e-01,
2.0268e-01, 9.3136e-01],
[-1.0109e+00, -1.1371e+00, -1.0942e+00, ..., -5.0632e+00,
-1.3663e+00, 7.6502e-01],
[-4.0844e-01, -5.1764e-01, -9.7092e-01, ..., -6.8881e+00,
-1.7973e+00, -1.0531e-01],
...,
[-9.5233e-01, -2.6685e+00, -7.6018e+00, ..., -3.7714e-01,
-1.1762e-01, 8.2923e-01],
[-1.6309e+00, -7.1870e-01, -3.8440e+00, ..., -8.0707e-01,
-5.0233e-01, 4.4773e-01],
[-1.1344e+00, -7.0996e-01, -4.0538e-01, ..., -2.9609e-01,
-1.5082e-01, 1.1798e+00]]],
[[[-8.2407e-01, -1.3618e+00, -1.3607e+00, ..., -2.2949e-01,
2.7196e-01, -3.9386e-01],
[-4.6311e-01, -2.4126e+00, -2.3796e+00, ..., -6.0615e-01,
-4.1565e-01, -5.3743e-01],
[-6.4846e-01, -1.9478e+00, -1.6843e+00, ..., -3.5250e-01,
-3.9130e-01, -1.0593e+00],
...,
[-1.0144e-02, -1.5693e+00, -1.6830e+00, ..., -3.5614e-01,
-3.3350e-01, -1.0791e+00],
[-1.1843e+00, -1.6706e+00, -8.6820e-01, ..., 2.3306e-02,
1.3861e-01, -8.8093e-01],
[-1.0710e+00, -1.9644e+00, -1.4279e+00, ..., -9.1365e-01,
-1.1188e+00, -6.6208e-01]],
[[-2.5444e+00, -4.9178e+00, -4.2895e+00, ..., -2.1631e+00,
-3.5677e+00, -1.2713e+00],
[-5.6198e+00, -8.6029e+00, -7.5497e+00, ..., -3.4244e+00,
-5.1647e+00, -2.0960e+00],
[-4.3001e+00, -6.5526e+00, -6.5039e+00, ..., -2.9987e+00,
-3.9037e+00, -1.7730e+00],
...,
[-4.3950e+00, -6.0719e+00, -5.3430e+00, ..., -3.0476e+00,
-3.9358e+00, -1.6765e+00],
[-6.6093e+00, -7.7948e+00, -6.6162e+00, ..., -3.9033e+00,
-5.3642e+00, -1.9452e+00],
[-6.2873e-02, -1.0739e+00, 4.2069e-01, ..., 4.3400e-01,
6.1701e-01, 1.6683e-01]],
[[-5.8228e-01, 6.9708e-01, 1.0371e+00, ..., 8.1870e-01,
5.9279e-01, 1.3567e+00],
[-9.0320e-01, 2.6978e-01, 5.8690e-01, ..., 3.9596e-01,
3.7933e-01, 2.4782e-01],
[-1.7054e+00, 7.3077e-02, 1.8402e-01, ..., 5.6724e-02,
2.7231e-01, 6.0470e-02],
...,
[-1.4985e+00, -8.1061e-02, -6.2742e-01, ..., 7.5357e-02,
2.0835e-01, 8.3306e-02],
[-1.4711e+00, 1.6676e-01, 3.3077e-02, ..., 1.0432e-01,
2.9314e-01, 2.4766e-01],
[-1.3768e+00, -5.5160e-01, -2.5869e-01, ..., -2.7051e-02,
1.7446e-01, -2.4137e-02]],
...,
[[ 1.2142e+00, -3.5016e-01, -2.1732e-01, ..., 3.4027e-01,
2.0352e-01, -8.0059e-01],
[-6.0765e-01, -9.4494e-01, -8.0917e-01, ..., -2.9991e-01,
-1.0183e+00, -9.6720e-01],
[-2.9796e-01, -7.7430e-01, -7.8242e-01, ..., -9.7375e-02,
-2.4044e-01, -6.2556e-01],
...,
[-1.7335e+00, -1.9110e+00, -3.5803e-01, ..., -1.8241e-01,
-2.6356e-01, -5.9000e-01],
[-3.2139e+00, -3.2799e+00, -2.3698e+00, ..., -1.0746e+00,
-1.1524e+00, -1.0831e+00],
[-6.7234e-01, -1.6913e+00, -9.6536e-01, ..., -1.8421e-01,
-3.2271e-01, -4.9607e-01]],
[[ 9.5105e-02, 9.8840e-02, 6.0251e-01, ..., 3.2909e-01,
8.1994e-01, 1.3083e-02],
[ 2.1957e-01, -5.6957e-01, 1.6521e-01, ..., 4.3017e-01,
6.7956e-01, -2.8511e-01],
[ 1.4552e-01, -2.8285e-01, 9.1802e-01, ..., 4.1783e-01,
4.6305e-01, -1.4540e-01],
...,
[ 1.2272e-01, -1.2122e+00, -2.2040e-01, ..., 3.8198e-01,
3.3576e-01, -1.6586e-01],
[-4.7999e-01, -1.6193e+00, -3.1294e-03, ..., 5.0220e-02,
7.2957e-02, -4.6485e-01],
[ 5.5952e-01, 8.6166e-02, 1.8971e-01, ..., 2.3776e-01,
3.0192e-01, 3.6862e-01]],
[[-1.5255e+00, -6.2541e-01, -5.7960e-03, ..., 3.4450e-01,
2.6675e-01, 8.5152e-01],
[-2.1541e+00, -4.6091e-01, -1.6293e-01, ..., -4.2735e-01,
-1.3692e-01, 1.1813e+00],
[-2.0709e+00, -9.1790e-01, -6.5346e-01, ..., -8.6107e-02,
1.7204e-02, 9.5009e-01],
...,
[-1.1629e+00, 3.8415e-01, 1.1247e+00, ..., -5.5639e-02,
3.8840e-02, 9.2082e-01],
[-2.3152e+00, -1.2154e+00, 6.9186e-02, ..., -2.4739e-01,
-8.4087e-02, 6.0758e-01],
[-1.0467e+00, -3.8481e-01, 3.4940e-01, ..., -2.2149e-01,
-2.3539e-01, 7.0049e-01]]],
[[[-7.2208e-01, -1.4066e+00, -1.3903e+00, ..., -3.1632e-02,
5.7234e-01, 4.9011e-01],
[-3.0895e-01, -2.4035e+00, -2.1414e+00, ..., -4.6737e-01,
-2.1247e-01, 1.0268e-01],
[-4.1322e-01, -2.0887e+00, -1.7323e+00, ..., -3.1470e-01,
-1.2809e-01, -9.0931e-02],
...,
[-4.8340e-01, -1.7682e+00, -1.4277e+00, ..., -1.6035e-01,
1.3571e-01, -1.6291e-01],
[-9.7740e-01, -1.7250e+00, -1.2059e+00, ..., 8.1741e-02,
3.2402e-01, 4.8031e-02],
[-7.9838e-01, -1.6588e+00, -1.2665e+00, ..., -1.1752e+00,
-1.2404e+00, -4.7123e-01]],
[[-2.3982e+00, -4.7623e+00, -4.0730e+00, ..., -5.3290e-01,
-1.0393e+00, -2.5319e-02],
[-5.3918e+00, -8.0689e+00, -6.4984e+00, ..., -1.7381e+00,
-2.5459e+00, -6.0643e-01],
[-4.4748e+00, -6.4218e+00, -6.0879e+00, ..., -1.2923e+00,
-1.4930e+00, -3.7206e-01],
...,
[-4.3610e+00, -6.1731e+00, -5.6363e+00, ..., -1.4227e+00,
-1.5806e+00, -6.0099e-01],
[-6.3518e+00, -7.4172e+00, -6.2772e+00, ..., -2.6802e+00,
-3.5766e+00, -1.0133e+00],
[-3.3999e-02, -8.5548e-01, 4.0859e-01, ..., 3.6578e-01,
3.7524e-01, 6.6940e-02]],
[[-5.7351e-01, 6.1320e-01, 1.0792e+00, ..., 7.8062e-01,
2.8749e-01, 1.0171e+00],
[-8.9704e-01, -2.3272e-02, 3.9092e-01, ..., 3.9080e-01,
9.6715e-02, 4.6442e-01],
[-1.6410e+00, -7.3345e-02, -7.3022e-02, ..., 4.2316e-02,
9.8985e-02, 4.2062e-01],
...,
[-1.7117e+00, 2.9229e-01, 4.9948e-01, ..., 4.0973e-01,
-2.3866e-03, 3.3687e-01],
[-1.5438e+00, 1.0249e-01, 4.8854e-02, ..., 1.1864e-01,
1.3182e-01, 6.5614e-01],
[-1.4192e+00, -4.8180e-01, -2.3936e-01, ..., -9.6213e-02,
5.7191e-02, 3.9116e-01]],
...,
[[ 1.0765e+00, -4.1963e-01, -2.1681e-01, ..., -8.3329e-02,
-3.3054e-01, -8.6625e-01],
[-7.9337e-01, -9.6481e-01, -8.3294e-01, ..., -5.8264e-01,
-1.1555e+00, -7.7615e-01],
[-6.4656e-01, -9.6069e-01, -7.1677e-01, ..., -9.8946e-02,
-1.8102e-01, -3.8514e-01],
...,
[-7.7136e-01, -1.4415e+00, -1.4872e+00, ..., -3.2640e-01,
-5.0967e-01, -9.1906e-01],
[-2.7936e+00, -2.8341e+00, -1.8543e+00, ..., -9.2564e-01,
-1.0285e+00, -9.3327e-01],
[-6.5890e-01, -1.6522e+00, -9.9296e-01, ..., -5.8433e-02,
-1.4774e-01, -3.8766e-01]],
[[ 2.6697e-02, -1.6949e-02, 5.9297e-01, ..., 2.7419e-01,
6.0013e-01, 2.6566e-01],
[ 2.1154e-02, -7.6339e-01, 1.1665e-01, ..., 2.4382e-03,
1.1581e-01, -1.4595e-02],
[-8.1300e-02, -8.7978e-01, 3.5873e-01, ..., -1.3538e-01,
-8.2277e-02, 1.7338e-01],
...,
[ 1.9484e-02, -6.3254e-01, 3.6432e-01, ..., -1.4664e-01,
-2.4583e-01, -3.4158e-01],
[-2.7693e-01, -1.4674e+00, 1.5850e-02, ..., -4.4099e-01,
-4.1010e-01, -1.4139e-01],
[ 6.4339e-01, 1.2804e-01, 2.7366e-01, ..., -3.0040e-01,
-2.8122e-01, 2.7239e-01]],
[[-1.5804e+00, -5.9288e-01, 1.7775e-02, ..., 5.1296e-01,
3.2788e-01, 5.1239e-01],
[-2.0610e+00, -3.2517e-01, -1.0565e-01, ..., -4.4633e-02,
-1.4317e-01, 3.8881e-01],
[-1.6765e+00, -2.9298e-01, 8.2121e-02, ..., 5.4948e-02,
-1.3618e-01, 7.4467e-02],
...,
[-1.7363e+00, -6.1674e-01, 1.8466e-01, ..., -1.0298e-01,
1.5055e-01, 4.2424e-01],
[-2.1985e+00, -1.1741e+00, -2.0011e-01, ..., -1.6951e-01,
-3.1540e-01, -2.4248e-01],
[-1.0292e+00, -1.8720e-01, 4.6389e-01, ..., -1.6643e-01,
-3.2145e-01, 8.9365e-02]]],
[[[ 1.5963e-02, -2.4352e-01, -2.0064e-02, ..., -9.0662e-01,
-4.9845e-01, -1.3756e+00],
[ 2.8227e-01, -9.1996e-01, -6.7989e-01, ..., -1.1458e+00,
-1.3374e+00, -1.4530e+00],
[ 2.1173e-01, -5.7580e-01, -4.7240e-01, ..., -8.7756e-01,
-1.0914e+00, -1.8583e+00],
...,
[-4.0616e-02, -6.0722e-01, -4.0311e-01, ..., -9.1280e-01,
-1.0281e+00, -1.9196e+00],
[-6.6372e-01, -6.9432e-01, -3.6360e-02, ..., -4.5653e-01,
-5.7160e-01, -1.7034e+00],
[-7.4053e-01, -1.3415e+00, -1.1877e+00, ..., -9.4661e-01,
-1.3457e+00, -6.4951e-01]],
[[-7.8519e-01, -2.0838e+00, -1.8014e+00, ..., -3.8990e+00,
-5.4375e+00, -1.8189e+00],
[-2.0417e+00, -3.6644e+00, -2.9575e+00, ..., -5.9825e+00,
-8.2263e+00, -3.4682e+00],
[-1.6365e+00, -2.5677e+00, -2.5526e+00, ..., -5.0998e+00,
-6.2978e+00, -2.6098e+00],
...,
[-1.8087e+00, -2.4063e+00, -2.4426e+00, ..., -5.0090e+00,
-6.2943e+00, -2.5440e+00],
[-3.8031e+00, -4.3914e+00, -3.6502e+00, ..., -6.0540e+00,
-7.5255e+00, -2.4158e+00],
[ 2.7964e-01, -3.4697e-01, 3.8243e-01, ..., 4.0828e-01,
4.3638e-01, 2.3627e-01]],
[[-2.2636e-01, 1.5718e-01, 7.2604e-01, ..., 7.0372e-01,
3.0811e-01, 1.6165e+00],
[ 2.6306e-02, -3.6570e-01, 5.7713e-01, ..., 5.3225e-01,
-1.1474e-01, 1.7454e-01],
[-1.3127e-01, -2.6376e-01, 2.3835e-01, ..., -2.8577e-02,
-2.3908e-02, -9.7956e-02],
...,
[-2.6954e-01, -2.3795e-01, 1.6408e-01, ..., 1.0255e-02,
-1.0781e-01, -1.9584e-01],
[-2.5420e-01, -2.1461e-01, 1.9695e-01, ..., -1.0302e-01,
5.0875e-03, -2.0167e-01],
[-1.4118e-01, -5.0751e-01, 1.3696e-01, ..., 4.2878e-02,
2.0900e-01, -6.1744e-01]],
...,
[[ 6.1588e-01, -2.3883e-01, -3.4711e-02, ..., 4.7599e-01,
2.6005e-01, -1.3576e+00],
[-4.9465e-01, -5.6753e-01, -5.2921e-01, ..., -1.8712e-01,
-8.5552e-01, -1.4397e+00],
[-3.5977e-01, -2.2328e-01, -2.5822e-01, ..., -1.2226e-01,
-9.8861e-02, -9.8756e-01],
...,
[-4.2463e-01, -1.5957e-01, -1.9176e-01, ..., -1.3310e-01,
-9.8291e-02, -9.2567e-01],
[-2.3441e+00, -1.7321e+00, -1.2514e+00, ..., -1.3741e+00,
-1.3844e+00, -1.7048e+00],
[-5.0450e-01, -8.2471e-01, -3.7819e-01, ..., -3.6023e-01,
-5.1420e-01, -1.1458e+00]],
[[ 1.4440e-01, 2.4458e-01, 6.2301e-01, ..., -4.5170e-02,
4.2785e-01, -3.7118e-01],
[ 1.5034e-01, 2.0927e-01, 4.2630e-01, ..., 2.5234e-01,
1.0527e-01, -8.2518e-01],
[-1.1177e-01, 9.3161e-02, 2.5231e-01, ..., 3.2562e-01,
1.6749e-01, -5.0777e-01],
...,
[-2.1110e-01, 2.5252e-02, 3.1356e-01, ..., 2.2259e-01,
1.3418e-01, -4.9894e-01],
[-5.6441e-01, -6.7619e-01, -2.4553e-02, ..., -6.6259e-02,
-1.4265e-01, -8.0105e-01],
[ 4.9636e-01, 4.0928e-01, 7.5249e-02, ..., 3.8569e-01,
3.6910e-01, 3.3880e-01]],
[[-8.2976e-01, 2.6588e-01, 3.5044e-01, ..., -8.8222e-03,
1.4695e-03, 6.2400e-01],
[-1.1465e+00, 2.1214e-01, -3.3166e-01, ..., -8.8128e-01,
-3.7370e-01, 9.2178e-01],
[-9.6232e-01, 3.5368e-01, -1.2910e-01, ..., -5.6955e-01,
-5.0240e-01, 6.1160e-01],
...,
[-9.8118e-01, 2.2187e-01, -1.3711e-01, ..., -5.8318e-01,
-5.3433e-01, 6.2678e-01],
[-1.7175e+00, -3.5610e-01, -3.8013e-01, ..., -6.6068e-01,
-6.3860e-01, 4.5985e-01],
[-7.4160e-01, -5.0025e-02, -3.0768e-01, ..., -4.1946e-01,
-3.9137e-01, 1.0217e+00]]]], device='cuda:0'),
tensor([[[[-3.7118e-01, -1.0921e+00, -9.6980e-01, ..., -1.9619e-01,
5.1794e-01, 4.2226e-01],
[ 7.9489e-02, -2.1433e+00, -1.9379e+00, ..., -8.5315e-01,
-2.8322e-01, 6.4803e-02],
[-1.2692e+00, -1.9601e+00, -1.0366e+00, ..., 4.4746e-01,
-2.4771e-01, -2.0486e-01],
...,
[ 2.0286e+00, -4.9237e-02, -2.5101e+00, ..., -1.0207e+00,
-2.5827e-01, -2.2687e-01],
[-2.4119e-04, -3.5324e-01, 6.2872e-02, ..., -2.1742e+00,
1.7396e-01, -7.4372e-02],
[-2.2256e-01, -1.0681e+00, -1.0071e+00, ..., -2.0609e+00,
-1.1118e+00, -3.8795e-01]],
[[-2.1657e+00, -4.1548e+00, -3.6851e+00, ..., -1.0647e+00,
-1.3454e+00, -1.8002e-01],
[-6.2672e+00, -1.0155e+01, -9.2525e+00, ..., -5.5080e+00,
-2.9737e+00, -8.2700e-01],
[-7.2461e+00, -1.0662e+01, -1.0087e+01, ..., -4.8456e+00,
-1.8922e+00, -5.6580e-01],
...,
[-3.7762e+00, -6.3380e+00, -7.6327e+00, ..., -4.5883e+00,
-1.9292e+00, -5.6210e-01],
[-5.1794e+00, -6.1134e+00, -6.0110e+00, ..., -4.3494e+00,
-3.6640e+00, -1.1261e+00],
[ 5.7393e-01, -2.2291e-01, 3.7730e-01, ..., 1.0445e-01,
3.8170e-01, 7.7092e-02]],
[[-6.2697e-01, 3.5800e-01, 7.7679e-01, ..., 9.7431e-01,
4.7972e-01, 1.2078e+00],
[-9.5907e-01, 1.8604e-01, 4.4748e-01, ..., 9.6927e-01,
1.4611e-01, 5.0021e-01],
[-2.2063e+00, 3.3256e-01, 8.4698e-01, ..., 1.3816e+00,
1.5070e-01, 4.6077e-01],
...,
[-1.4260e+00, -1.0228e+00, -2.0023e+00, ..., 1.8898e+00,
9.9898e-02, 4.4205e-01],
[-1.2998e+00, 7.6892e-01, 6.1224e-01, ..., 2.1430e+00,
1.9763e-01, 6.2452e-01],
[-6.2490e-01, 2.5716e-01, 1.0082e-01, ..., 1.8748e-01,
9.6005e-02, 3.5879e-01]],
...,
[[ 6.6813e-01, 8.0406e-02, 3.5527e-01, ..., -9.8226e-02,
-3.0647e-01, -9.1471e-01],
[-8.8427e-01, -5.1961e-01, -3.5668e-01, ..., -7.9605e-01,
-1.3097e+00, -8.4401e-01],
[ 6.1101e-01, -6.7802e-01, -7.0363e-01, ..., -1.7092e+00,
-2.5764e-01, -4.3349e-01],
...,
[-4.6937e+00, -4.8558e+00, -5.2172e-01, ..., -5.2030e+00,
-2.4971e-01, -4.0805e-01],
[-2.1747e+00, -2.7357e+00, -3.4006e+00, ..., -3.1687e+00,
-8.9746e-01, -8.2466e-01],
[-3.8596e-01, -9.7419e-01, -4.0292e-01, ..., -2.9732e-01,
-1.8802e-01, -3.7410e-01]],
[[-4.0628e-02, -3.0402e-01, -9.0096e-02, ..., 2.0873e-01,
7.3557e-01, 3.3371e-01],
[ 1.9804e-01, -1.2569e+00, -8.6221e-01, ..., -2.0467e+00,
1.6659e-01, -2.8263e-02],
[-4.2151e-01, -7.7109e-01, 7.5403e-01, ..., -3.1016e-01,
-1.7951e-01, 1.5585e-01],
...,
[ 6.2938e-01, -1.6418e+00, -2.3983e+00, ..., -1.2105e-01,
-2.0980e-01, 1.1014e-01],
[-5.4908e-01, -1.3843e+00, 4.3498e-01, ..., -1.4880e+00,
-4.5102e-01, -1.1956e-01],
[ 6.6829e-01, -1.2568e-01, 2.6250e-01, ..., -1.1571e+00,
-2.4661e-01, 3.0940e-01]],
[[-6.8817e-01, -2.5277e-01, -7.7313e-02, ..., 5.7190e-01,
3.2911e-01, 6.1429e-01],
[-1.3362e+00, -1.3393e+00, -1.5981e+00, ..., 2.2830e-01,
-1.1337e-01, 5.2758e-01],
[-2.7520e+00, -3.3246e+00, -2.9390e+00, ..., -5.5268e-03,
-1.6611e-01, 1.8067e-01],
...,
[ 6.6367e-01, 1.8186e+00, 1.0495e+00, ..., 1.2168e+00,
-1.9892e-02, 1.8537e-01],
[-9.0943e-01, -1.1325e+00, -2.0228e-01, ..., 1.5195e+00,
-2.3575e-01, -1.1857e-01],
[-5.2198e-01, -9.7781e-01, -3.4310e-01, ..., 1.4376e-01,
-2.7676e-01, 2.1269e-01]]],
[[[ 1.0053e-02, -3.3592e-01, -1.4345e-01, ..., -1.0277e+00,
-5.4674e-01, -1.5050e+00],
[ 2.5234e-01, -1.0152e+00, -7.3935e-01, ..., -1.2770e+00,
-1.3664e+00, -1.5545e+00],
[ 2.2487e-01, -6.3094e-01, -5.4761e-01, ..., -9.3748e-01,
-1.1243e+00, -1.8688e+00],
...,
[-2.5337e+00, -2.3398e+00, -1.2360e+00, ..., 1.6984e+00,
-2.4498e+00, -2.0176e+00],
[-4.4100e+00, -5.3167e+00, -1.0481e+00, ..., -5.7348e-01,
-1.2002e+00, -1.8583e+00],
[-2.0980e+00, -4.3554e+00, -1.2781e+00, ..., -1.2395e+00,
-1.6430e+00, -8.9442e-01]],
[[-9.0049e-01, -2.3886e+00, -2.2917e+00, ..., -3.9057e+00,
-5.5530e+00, -1.8431e+00],
[-2.2531e+00, -3.9782e+00, -3.3918e+00, ..., -6.0670e+00,
-8.3284e+00, -3.4567e+00],
[-1.8045e+00, -2.8068e+00, -2.8593e+00, ..., -7.7837e+00,
-6.3638e+00, -2.4889e+00],
...,
[-5.7349e+00, -1.0012e+01, -2.5078e+00, ..., -4.0334e+00,
-6.2481e+00, -2.4722e+00],
[-8.1490e+00, -1.0496e+01, -4.1840e+00, ..., -5.7631e+00,
-8.3209e+00, -2.5131e+00],
[-8.0202e-01, -2.8987e+00, -2.3659e-01, ..., 3.5460e-01,
3.9295e-01, 2.7200e-01]],
[[-1.9436e-01, 2.3524e-01, 7.5240e-01, ..., 6.0701e-01,
1.4230e-01, 1.4931e+00],
[ 3.0087e-02, -3.1718e-01, 5.5809e-01, ..., 5.6615e-01,
-1.7025e-01, -1.3266e-02],
[-2.2524e-01, -2.3092e-01, 2.3739e-01, ..., 8.0064e-01,
-5.0705e-02, -1.6288e-01],
...,
[-1.0478e+01, -7.7651e+00, -4.4538e+00, ..., -6.0322e+00,
7.2740e-01, -1.5909e-01],
[-9.7644e+00, -6.2191e+00, -6.1914e+00, ..., -1.1481e+00,
4.3630e-01, -2.2296e-01],
[-3.7731e+00, 1.2262e-01, -3.7987e+00, ..., 4.2976e-02,
1.6285e-01, -6.2940e-01]],
...,
[[ 6.4465e-01, -9.4827e-02, 1.4314e-01, ..., 4.7809e-01,
4.5192e-01, -1.3154e+00],
[-4.8902e-01, -4.4502e-01, -3.8404e-01, ..., -2.0229e-01,
-7.5315e-01, -1.4047e+00],
[-3.2395e-01, -1.7914e-01, -1.6221e-01, ..., 9.4341e-02,
-5.1632e-02, -9.4670e-01],
...,
[-3.0533e+00, -3.0099e+00, -2.5953e+00, ..., -7.8139e+00,
-2.0361e+00, -9.7752e-01],
[-2.8584e+00, -5.4219e+00, -7.3365e+00, ..., -3.0952e+00,
-1.7933e+00, -1.9772e+00],
[ 1.7758e+00, -3.2327e+00, -4.5779e+00, ..., -4.3086e-01,
-5.3426e-01, -1.2665e+00]],
[[ 1.8116e-01, 2.7311e-01, 5.8333e-01, ..., -4.5973e-02,
4.1918e-01, -2.6717e-01],
[ 1.5940e-01, 2.2593e-01, 4.3871e-01, ..., 1.6300e-01,
8.2653e-02, -7.7834e-01],
[-1.2056e-01, 7.2890e-02, 3.0393e-01, ..., -1.4167e+00,
1.8503e-01, -3.3601e-01],
...,
[-3.7974e+00, -1.1662e+00, -1.5395e+00, ..., 3.7313e+00,
-1.8355e+00, -4.6854e-01],
[-3.9428e+00, -2.2808e+00, -1.5395e+00, ..., 3.2432e-01,
-1.1858e+00, -8.9639e-01],
[-2.1168e+00, 4.9637e-01, 1.4497e+00, ..., 3.1086e-01,
2.6816e-01, 3.1360e-01]],
[[-7.9715e-01, 2.2855e-01, 2.9072e-01, ..., -2.2827e-01,
-1.7382e-01, 5.2226e-01],
[-1.1237e+00, 1.4206e-01, -3.9271e-01, ..., -9.7059e-01,
-4.6182e-01, 9.2692e-01],
[-9.4234e-01, 2.6333e-01, -1.1598e-01, ..., -9.1712e-01,
-6.3978e-01, 6.4241e-01],
...,
[-9.7101e+00, -7.7984e+00, -6.9371e+00, ..., 1.3477e+00,
1.4091e+00, 6.4807e-01],
[-6.0165e+00, -6.1581e+00, -1.3323e+00, ..., 2.8592e-01,
-4.9013e-01, 3.7113e-01],
[-4.2775e-01, -2.7863e+00, 4.3905e-01, ..., -5.1960e-01,
-5.2007e-01, 1.0561e+00]]],
[[[-7.3466e-01, -1.4386e+00, -1.4738e+00, ..., -2.4221e-01,
2.1914e-01, -4.2601e-01],
[-2.9384e-01, -2.4714e+00, -2.2060e+00, ..., -6.2818e-01,
-3.9924e-01, -6.2969e-01],
[-4.0830e-01, -2.1617e+00, -1.7833e+00, ..., -1.3317e+00,
-9.0877e-01, -1.6839e+00],
...,
[-5.4270e-01, -2.0066e+00, -1.8345e+00, ..., -3.3634e-01,
-1.1701e+00, -1.8736e+00],
[-1.1007e+00, -1.8258e+00, -1.1880e+00, ..., 1.3657e-01,
1.4073e-01, -1.2466e+00],
[-8.7112e-01, -1.7394e+00, -1.2944e+00, ..., -8.6179e-01,
-1.0882e+00, -6.5273e-01]],
[[-2.4206e+00, -4.9837e+00, -4.3115e+00, ..., -2.3070e+00,
-3.5934e+00, -1.3326e+00],
[-5.4714e+00, -8.3170e+00, -6.8707e+00, ..., -3.6665e+00,
-5.1827e+00, -2.1417e+00],
[-4.4156e+00, -6.3436e+00, -6.3573e+00, ..., -7.8941e+00,
-8.8910e+00, -3.2747e+00],
...,
[-4.4813e+00, -6.3236e+00, -5.7264e+00, ..., -3.1582e+00,
-4.8411e+00, -2.6998e+00],
[-6.4872e+00, -7.6502e+00, -6.6068e+00, ..., -4.1194e+00,
-5.4216e+00, -2.1198e+00],
[-6.8760e-02, -8.9327e-01, 2.8748e-01, ..., 4.4303e-01,
6.4461e-01, 1.8184e-01]],
[[-6.1621e-01, 5.6090e-01, 1.1137e+00, ..., 8.0921e-01,
5.5118e-01, 1.3511e+00],
[-9.2142e-01, -5.9545e-02, 3.4133e-01, ..., 3.9900e-01,
3.2375e-01, 2.0863e-01],
[-1.6356e+00, -1.6912e-01, -2.1254e-01, ..., -3.4443e-01,
1.0398e-01, 1.1720e-01],
...,
[-1.8133e+00, -9.6364e-02, -1.2156e-01, ..., -1.5744e+00,
-2.6569e-01, -1.8613e-02],
[-1.5682e+00, -2.5440e-02, -4.1670e-02, ..., 4.3831e-01,
3.7300e-01, -8.6703e-02],
[-1.4713e+00, -5.9524e-01, -3.3511e-01, ..., 2.2509e-02,
2.0842e-01, -1.0502e-01]],
...,
[[ 1.0645e+00, -4.1793e-01, -2.0870e-01, ..., 3.7716e-01,
2.1962e-01, -8.0437e-01],
[-7.9921e-01, -9.5013e-01, -7.9632e-01, ..., -3.2837e-01,
-1.0287e+00, -9.5044e-01],
[-6.7192e-01, -9.5318e-01, -6.6824e-01, ..., -4.6774e-01,
6.0618e-01, 1.7639e-01],
...,
[-6.4574e-01, -1.1234e+00, -8.1096e-01, ..., 2.3995e-01,
-1.6217e+00, -2.9117e+00],
[-2.8714e+00, -2.9859e+00, -1.9380e+00, ..., -1.4720e+00,
-1.9743e+00, -1.5974e+00],
[-6.8244e-01, -1.7114e+00, -1.0475e+00, ..., -1.5269e-01,
-2.5217e-01, -4.8959e-01]],
[[ 1.1099e-02, -2.5800e-02, 6.0327e-01, ..., 3.0680e-01,
8.1509e-01, 2.5485e-02],
[-1.4334e-02, -7.1795e-01, 2.1503e-01, ..., 2.1620e-01,
6.2107e-01, -3.3237e-01],
[-8.8148e-02, -8.1629e-01, 4.8961e-01, ..., -1.1026e-02,
-2.0484e-01, -1.2645e+00],
...,
[ 1.5769e-01, -6.9559e-01, 3.4290e-01, ..., -7.1165e-01,
-1.2522e+00, -5.8309e-01],
[-1.7844e-01, -1.3787e+00, 6.0720e-02, ..., 2.0927e-01,
-2.2602e-01, -1.1539e+00],
[ 6.7432e-01, 1.4949e-01, 2.9211e-01, ..., 2.8190e-01,
3.4896e-01, 3.7389e-01]],
[[-1.6617e+00, -6.7368e-01, 6.2475e-02, ..., 3.3321e-01,
2.8111e-01, 8.6168e-01],
[-2.1793e+00, -4.3686e-01, -3.6084e-02, ..., -4.1568e-01,
-1.2974e-01, 1.1916e+00],
[-1.8434e+00, -4.8027e-01, 1.8175e-01, ..., -3.6631e-01,
2.6065e-01, 7.5459e-01],
...,
[-1.9574e+00, -4.8170e-01, 2.1534e-01, ..., 8.9131e-01,
3.6167e-01, 9.9507e-01],
[-2.4173e+00, -1.1951e+00, -2.2659e-01, ..., -1.5269e-01,
2.1410e-01, 1.0608e+00],
[-1.1404e+00, -2.0325e-01, 4.2614e-01, ..., -1.8499e-01,
-1.9717e-01, 7.1434e-01]]],
[[[-3.3966e+00, -1.5366e+00, 2.5493e-01, ..., -7.9288e-02,
3.5256e-01, 7.4718e-03],
[-1.4452e+00, -4.1310e+00, -6.0598e+00, ..., -7.9206e-01,
-5.6814e-01, -4.9884e-01],
[ 2.5468e-01, -3.2970e+00, -5.8651e+00, ..., -5.6022e-01,
-4.9815e-01, -8.7505e-01],
...,
[-3.7661e-01, -4.3644e-01, -6.0814e-01, ..., -2.0206e+00,
-1.0800e+00, -2.1835e+00],
[-1.3609e+00, -8.8036e-01, 7.3347e-01, ..., -6.3802e-01,
-5.4698e-01, -1.9049e+00],
[-1.3972e+00, -2.3765e+00, -1.7770e+00, ..., -1.5230e+00,
-1.6360e+00, -1.0369e+00]],
[[-3.0124e+00, -6.8962e+00, -5.1056e+00, ..., -1.1335e+00,
-1.8107e+00, -5.0523e-01],
[-7.7510e+00, -1.4512e+01, -9.4546e+00, ..., -2.9457e+00,
-3.9641e+00, -1.4343e+00],
[-4.4899e+00, -1.1690e+01, -1.1723e+01, ..., -3.1618e+00,
-3.6127e+00, -1.5494e+00],
...,
[-3.9749e+00, -5.8165e+00, -5.1216e+00, ..., -6.2223e+00,
-6.5226e+00, -3.0022e+00],
[-6.4045e+00, -7.5148e+00, -6.5054e+00, ..., -6.3689e+00,
-7.6835e+00, -2.6995e+00],
[-1.3316e-01, -9.7788e-01, 2.0127e-01, ..., 3.5698e-01,
6.3986e-01, 2.6438e-01]],
[[-4.2287e+00, -7.6214e+00, -9.6098e+00, ..., 5.1787e-01,
2.3383e-01, 1.0230e+00],
[-5.1932e+00, -7.9720e+00, -8.7188e+00, ..., 4.3955e-01,
3.4551e-01, 4.7996e-01],
[-4.3193e+00, -5.4203e+00, -1.0064e+01, ..., 9.4858e-02,
2.2379e-01, 4.0464e-01],
...,
[-1.4838e+00, -6.3871e-01, -7.4893e+00, ..., 6.3219e-01,
-1.1073e-01, 2.2520e-01],
[-1.2209e+00, 1.7053e+00, 8.0425e-02, ..., -5.2143e-02,
-1.8419e-02, 2.5457e-01],
[-1.2387e+00, -1.4612e-01, 4.3701e-02, ..., -2.0635e-02,
-1.9812e-02, -3.8630e-01]],
...,
[[-5.9177e-01, -4.0257e+00, -6.1623e+00, ..., 9.6814e-03,
-1.4907e-01, -7.1947e-01],
[-1.3832e+00, -5.3212e+00, -6.5933e+00, ..., -2.4358e-01,
-1.0370e+00, -8.3574e-01],
[ 1.4808e+00, -6.2437e-01, -4.1083e+00, ..., 7.1381e-02,
-1.9793e-01, -5.8684e-01],
...,
[-6.3561e-01, 7.0063e-01, 3.8748e+00, ..., -7.1174e-01,
-5.4222e-01, -1.1631e+00],
[-3.3249e+00, -4.1740e+00, -4.5184e+00, ..., -1.9922e+00,
-2.0774e+00, -2.1535e+00],
[-5.3365e-01, -1.7128e+00, -1.1066e+00, ..., -8.2669e-01,
-9.1676e-01, -1.3954e+00]],
[[-3.7833e+00, -4.0986e+00, -4.8401e+00, ..., 7.8961e-02,
5.3520e-01, 9.1883e-02],
[-1.9845e+00, -1.8565e+00, -6.3524e+00, ..., 2.1258e-02,
1.6320e-01, -1.7862e-01],
[-1.6861e+00, -8.0710e-01, -2.7675e+00, ..., 1.0057e-01,
3.5381e-02, -1.4870e-01],
...,
[-1.4723e-01, -1.4198e+00, -2.0524e+00, ..., -6.4548e-01,
1.0115e-01, -7.2750e-01],
[-5.1108e-01, -1.5176e+00, 2.2096e-02, ..., 3.8223e-02,
-2.2495e-02, -9.0997e-01],
[ 4.3764e-01, -1.3890e-01, 1.3890e-01, ..., 2.9787e-01,
3.7049e-01, 4.1599e-01]],
[[-2.7725e+00, -5.8614e+00, -7.0495e+00, ..., 5.0199e-01,
4.1765e-01, 7.1504e-01],
[-2.1369e+00, -4.7158e+00, -6.1694e+00, ..., -2.1847e-01,
-8.6944e-02, 7.7569e-01],
[-1.5020e+00, -3.8926e+00, -5.1042e+00, ..., 1.1362e-01,
1.9383e-01, 6.7579e-01],
...,
[-1.2952e+00, -1.5443e+00, -1.1382e+00, ..., -3.7874e-01,
-1.2993e-01, 8.1885e-01],
[-2.1874e+00, -2.3655e+00, -2.3967e-01, ..., -9.2241e-01,
-5.9101e-01, 4.5627e-01],
[-1.0631e+00, -1.0062e+00, -3.0044e-01, ..., -2.5857e-01,
-1.4637e-01, 1.2496e+00]]],
[[[-7.1427e-01, -1.4208e+00, -1.4064e+00, ..., 1.2093e-02,
6.1837e-01, 5.3248e-01],
[-2.9722e-01, -2.3939e+00, -2.1632e+00, ..., -7.2457e-01,
-2.3878e-01, 1.2268e-01],
[-3.9089e-01, -2.0652e+00, -1.7048e+00, ..., -4.3411e-01,
-1.3143e-01, -9.8666e-02],
...,
[-4.7602e-01, -1.8767e+00, -1.3485e+00, ..., -3.2802e-01,
-7.4339e-02, -1.3972e-01],
[-9.6963e-01, -1.6954e+00, -1.1851e+00, ..., 1.9975e-02,
2.6792e-01, -1.5964e-02],
[-7.8973e-01, -1.6217e+00, -1.2393e+00, ..., -1.0405e+00,
-1.0967e+00, -3.8501e-01]],
[[-2.3656e+00, -4.8448e+00, -4.1975e+00, ..., -4.5526e-01,
-1.0094e+00, 1.1514e-02],
[-5.3442e+00, -8.0967e+00, -6.5959e+00, ..., -2.6822e+00,
-2.7435e+00, -7.0713e-01],
[-4.3543e+00, -6.2961e+00, -6.2385e+00, ..., -3.1826e+00,
-1.5429e+00, -3.9151e-01],
...,
[-4.4131e+00, -6.1432e+00, -5.6709e+00, ..., -1.3074e+00,
-1.5905e+00, -3.7732e-01],
[-6.3747e+00, -7.4652e+00, -6.4171e+00, ..., -2.4834e+00,
-3.3940e+00, -9.8849e-01],
[ 6.6376e-03, -7.8619e-01, 3.7280e-01, ..., 3.3484e-01,
3.5859e-01, 7.4675e-02]],
[[-5.5720e-01, 6.0749e-01, 1.1058e+00, ..., 8.3763e-01,
3.1256e-01, 1.1112e+00],
[-8.6238e-01, -5.2138e-02, 3.7923e-01, ..., 4.9786e-01,
5.1481e-02, 4.8348e-01],
[-1.6248e+00, -9.3540e-02, -1.5996e-01, ..., 5.0774e-01,
9.8048e-02, 4.0284e-01],
...,
[-1.7478e+00, -3.2627e-03, 1.6112e+00, ..., 7.0484e-02,
6.1221e-02, 4.0688e-01],
[-1.5821e+00, 7.5133e-02, 6.5383e-02, ..., 1.5177e-01,
1.1912e-01, 5.9736e-01],
[-1.4269e+00, -5.4430e-01, -2.8575e-01, ..., -7.2988e-02,
5.8950e-02, 3.5413e-01]],
...,
[[ 1.0910e+00, -3.8006e-01, -2.1904e-01, ..., -1.0024e-01,
-3.9305e-01, -9.5895e-01],
[-8.1232e-01, -9.7401e-01, -8.0986e-01, ..., -6.2792e-01,
-1.2872e+00, -8.5268e-01],
[-6.4512e-01, -1.0028e+00, -7.5481e-01, ..., -7.0888e-01,
-1.9000e-01, -4.0475e-01],
...,
[-6.9957e-01, -1.1701e+00, -1.6312e+00, ..., -1.1961e-01,
-1.9034e-01, -3.6950e-01],
[-2.7373e+00, -2.8479e+00, -1.8440e+00, ..., -8.2095e-01,
-8.8995e-01, -7.9916e-01],
[-6.2807e-01, -1.6465e+00, -1.0077e+00, ..., -2.9527e-03,
-1.4181e-01, -3.4920e-01]],
[[ 2.7727e-02, -1.7129e-03, 6.0314e-01, ..., 3.4973e-01,
6.8240e-01, 3.0482e-01],
[ 1.2614e-02, -7.0539e-01, 1.4495e-01, ..., -8.3473e-01,
9.9671e-02, -6.6072e-02],
[-7.3232e-02, -7.9538e-01, 3.6486e-01, ..., -6.9890e-01,
-9.4612e-02, 1.7606e-01],
...,
[ 1.2214e-01, -7.8715e-01, 2.2658e-01, ..., -1.2133e-01,
-1.1230e-01, 1.4221e-01],
[-1.8343e-01, -1.4330e+00, 2.0527e-02, ..., -3.9340e-01,
-3.4362e-01, -6.3023e-02],
[ 6.9562e-01, 1.6189e-01, 3.0007e-01, ..., -2.5056e-01,
-2.3369e-01, 3.0042e-01]],
[[-1.6132e+00, -6.1179e-01, 3.6203e-02, ..., 5.6399e-01,
3.8261e-01, 5.8514e-01],
[-2.1218e+00, -3.8591e-01, -4.2285e-02, ..., -1.5671e-02,
-1.5821e-01, 4.2376e-01],
[-1.7039e+00, -3.7589e-01, 2.0435e-01, ..., 1.1573e-01,
-1.8377e-01, 8.8239e-02],
...,
[-1.7257e+00, -4.4554e-01, -1.6811e-02, ..., 6.4020e-02,
-1.2133e-01, 1.0649e-01],
[-2.2189e+00, -1.1208e+00, -1.5872e-01, ..., -1.0215e-01,
-2.7964e-01, -1.9327e-01],
[-1.0839e+00, -2.1528e-01, 4.5815e-01, ..., -9.5244e-02,
-2.7133e-01, 1.1299e-01]]]], device='cuda:0'),
tensor([[[[ 2.7908e-01, -2.6082e+00, -4.2728e+00, ..., -9.5650e-01,
-4.9645e-01, -1.4641e+00],
[-2.8974e-01, -6.1541e+00, -4.6265e+00, ..., -1.1892e+00,
-1.2758e+00, -1.5240e+00],
[-1.3723e+00, -4.9580e+00, -3.5802e+00, ..., -1.0387e+00,
-1.0749e+00, -1.8398e+00],
...,
[ 3.7626e-01, -4.3116e-01, -4.9616e-01, ..., 5.0372e-01,
-4.1458e-01, -2.0688e+00],
[-8.8144e-02, -5.9887e-01, -7.1297e-02, ..., 2.0077e+00,
4.3599e-01, -1.8641e+00],
[-4.6286e-01, -1.1721e+00, -7.7070e-01, ..., -1.3946e+00,
-2.1553e+00, -7.9199e-01]],
[[-4.8460e+00, -1.1203e+01, -1.3876e+01, ..., -4.0557e+00,
-5.5466e+00, -1.9430e+00],
[-6.0742e+00, -1.1176e+01, -1.1099e+01, ..., -5.9567e+00,
-8.1827e+00, -3.4823e+00],
[-3.3148e+00, -5.5380e+00, -7.6476e+00, ..., -5.6289e+00,
-6.3373e+00, -2.5735e+00],
...,
[-1.7741e+00, -4.1576e+00, -4.3196e+00, ..., -4.4907e+00,
-9.5177e+00, -2.8936e+00],
[-3.2887e+00, -4.2702e+00, -3.4718e+00, ..., -7.1033e+00,
-1.0976e+01, -2.7480e+00],
[ 3.3311e-01, -4.8540e-01, 3.2816e-01, ..., 1.4914e-01,
-1.2650e+00, 2.1040e-01]],
[[-7.1851e-01, -1.5584e+00, -8.4398e-01, ..., 5.8910e-01,
2.4381e-01, 1.5560e+00],
[-1.2720e+00, -6.2456e+00, -7.9973e+00, ..., 5.4097e-01,
-1.8139e-01, 1.1039e-01],
[-6.7005e-01, -9.2716e+00, -1.5317e+01, ..., 1.0423e-01,
-1.0222e-01, -1.6140e-01],
...,
[-4.5326e-01, -5.6831e-01, -3.2163e-01, ..., -2.1511e+00,
-2.6861e-01, -5.6230e-02],
[-1.7132e-01, -1.2924e-01, -2.2820e-01, ..., -2.4172e+00,
4.8304e-01, -2.0629e-01],
[-1.2669e-01, -1.4579e-01, -1.4557e-01, ..., -5.0994e+00,
5.8896e-01, -5.9081e-01]],
...,
[[ 8.9198e-01, 7.9526e-01, 1.9595e+00, ..., 5.4545e-01,
4.2088e-01, -1.3176e+00],
[ 1.4809e+00, 3.4938e+00, -1.3810e+00, ..., -1.7240e-01,
-7.7093e-01, -1.4110e+00],
[ 6.8699e-01, -1.0056e+00, -8.2400e+00, ..., -9.4224e-02,
-1.4097e-01, -9.5508e-01],
...,
[-2.6653e-01, -2.1815e-01, -1.1922e-01, ..., -2.7249e+00,
-2.0834e+00, -1.0865e+00],
[-2.1236e+00, -1.3627e+00, -1.2149e+00, ..., -8.5805e+00,
-4.9784e+00, -1.8557e+00],
[-3.9901e-01, -8.9538e-01, -9.8421e-01, ..., -4.7789e+00,
-2.8095e+00, -1.1655e+00]],
[[ 5.0766e-01, 1.4985e-01, -2.0269e+00, ..., -3.6299e-02,
4.6626e-01, -3.7265e-01],
[-1.6774e-01, -1.7493e+00, -2.9258e+00, ..., 2.5624e-01,
1.4973e-01, -8.4977e-01],
[-8.0780e-01, -3.5200e+00, -3.7139e+00, ..., 2.1904e-02,
1.0866e-01, -4.8633e-01],
...,
[-1.0617e-01, 6.3268e-02, 3.9555e-01, ..., 2.3036e+00,
1.2085e+00, -4.7447e-01],
[-3.0622e-01, -2.5972e-01, 4.0029e-01, ..., 2.4243e+00,
1.0608e+00, -9.8177e-01],
[ 4.8230e-01, 5.6089e-01, 4.0776e-01, ..., 1.2617e+00,
1.0711e+00, 2.8418e-01]],
[[-1.2050e+00, -1.3511e+00, -2.9193e+00, ..., -1.4439e-01,
-7.2401e-02, 6.0071e-01],
[-3.1816e+00, -5.9504e+00, -1.0093e+01, ..., -9.7888e-01,
-4.0411e-01, 9.0170e-01],
[-3.3627e+00, -6.5944e+00, -1.0438e+01, ..., -6.5073e-01,
-4.9450e-01, 6.6664e-01],
...,
[-1.0464e+00, 2.1383e-01, 1.9426e-01, ..., -7.6271e-01,
8.1965e-01, 1.0400e+00],
[-1.6212e+00, -7.0906e-01, -7.4312e-01, ..., -1.5698e+00,
3.5611e-01, 6.4583e-01],
[-4.6551e-01, -1.9421e-02, -3.2444e-01, ..., -8.1506e-01,
3.7917e-01, 1.0486e+00]]],
[[[-3.5407e-01, -5.0517e-01, -1.6214e-01, ..., -9.7895e-01,
-5.4613e-01, -1.4913e+00],
[-1.4103e-01, -1.5206e+00, -9.0923e-01, ..., -1.1995e+00,
-1.3681e+00, -1.5550e+00],
[-1.5102e+00, -1.6229e+00, -4.0048e-01, ..., -9.4241e-01,
-1.1462e+00, -1.8772e+00],
...,
[ 5.2795e-01, -1.6900e+00, -5.9923e-01, ..., -9.8231e-01,
-1.0497e+00, -1.9629e+00],
[-4.1950e-01, -1.0834e+00, -1.1380e-01, ..., -4.9800e-01,
-5.5166e-01, -1.7482e+00],
[-4.9969e-01, -1.5377e+00, -1.2593e+00, ..., -1.2012e+00,
-1.5918e+00, -7.9620e-01]],
[[-8.0551e-01, -2.1151e+00, -9.7713e-01, ..., -4.0015e+00,
-5.5341e+00, -1.8725e+00],
[-3.4474e+00, -5.1209e+00, -2.8287e+00, ..., -6.1661e+00,
-8.3210e+00, -3.5429e+00],
[-4.1041e+00, -6.3378e+00, -3.8223e+00, ..., -5.2138e+00,
-6.2477e+00, -2.5934e+00],
...,
[-1.6196e+00, -3.2553e+00, -1.4794e+00, ..., -5.1582e+00,
-6.4568e+00, -2.4802e+00],
[-3.8668e+00, -4.7055e+00, -3.0882e+00, ..., -6.5444e+00,
-8.0176e+00, -2.5181e+00],
[ 6.1763e-01, -6.4070e-01, 4.4280e-01, ..., 3.7413e-01,
4.5507e-01, 2.5703e-01]],
[[-4.1011e-01, 7.4079e-01, 1.0614e+00, ..., 5.8149e-01,
2.1234e-01, 1.5237e+00],
[-8.9779e-01, 2.4501e-01, 5.9259e-01, ..., 5.3087e-01,
-1.4327e-01, 3.8032e-02],
[-2.3209e+00, -1.3618e-01, 9.3187e-01, ..., 7.5694e-03,
-4.6890e-02, -1.3601e-01],
...,
[-1.0112e+00, 8.4180e-01, 1.1299e+00, ..., -3.9872e-02,
-7.7706e-02, -1.6431e-01],
[-1.3959e+00, 3.0338e-01, 2.7350e-01, ..., -1.2181e-01,
1.5763e-02, -1.9370e-01],
[-8.8285e-01, 8.1233e-02, -1.2799e-01, ..., 3.2132e-02,
2.1572e-01, -5.6409e-01]],
...,
[[ 4.6447e-01, -1.0721e-01, -9.8432e-02, ..., 5.0448e-01,
4.0872e-01, -1.3391e+00],
[-1.2962e+00, -9.9660e-01, -5.4419e-01, ..., -1.4974e-01,
-7.8905e-01, -1.4703e+00],
[-6.0519e-01, -1.5900e-01, -2.6815e-01, ..., -4.6621e-02,
-7.5192e-02, -1.0179e+00],
...,
[-3.0912e+00, -3.5292e+00, -2.1058e+00, ..., -1.6104e-01,
-1.3424e-01, -9.5093e-01],
[-2.5099e+00, -1.7804e+00, -9.6636e-01, ..., -1.5952e+00,
-1.6347e+00, -1.8722e+00],
[-6.3673e-01, -1.1419e+00, -1.1968e-01, ..., -4.1665e-01,
-5.5013e-01, -1.2267e+00]],
[[ 2.2013e-02, 2.6615e-01, 5.0132e-01, ..., -7.9796e-02,
4.1156e-01, -3.6544e-01],
[-1.1718e-01, -8.6731e-01, -5.1772e-01, ..., 1.8288e-01,
6.6863e-02, -8.8118e-01],
[-1.3220e+00, -9.8242e-01, -7.1103e-01, ..., 2.5428e-01,
1.4017e-01, -4.3141e-01],
...,
[-8.2400e-01, -2.4109e+00, -1.3107e+00, ..., 2.4203e-01,
1.3227e-01, -5.0522e-01],
[-1.0341e+00, -1.9622e+00, -4.1232e-01, ..., -1.3074e-01,
-2.6541e-01, -9.2044e-01],
[ 2.9991e-01, -4.2067e-01, -2.2480e-01, ..., 3.3825e-01,
3.0338e-01, 2.8917e-01]],
[[-4.0201e-01, 1.0716e-01, 5.4722e-01, ..., -1.5963e-01,
-1.1963e-01, 5.5985e-01],
[-3.3767e-01, 2.8301e-02, 4.6064e-02, ..., -1.0066e+00,
-4.5029e-01, 9.0308e-01],
[-2.9117e+00, -1.4730e+00, 2.3761e-02, ..., -7.1436e-01,
-5.4085e-01, 6.5419e-01],
...,
[ 5.4471e-01, 1.4887e+00, 1.3226e+00, ..., -4.4028e-01,
-5.3112e-01, 6.1607e-01],
[-3.8858e-01, -4.4900e-01, -1.8623e-01, ..., -7.0839e-01,
-7.3319e-01, 4.0061e-01],
[ 1.1392e-02, -4.9558e-01, -1.8650e-01, ..., -4.7302e-01,
-4.6024e-01, 1.0164e+00]]],
[[[-7.4541e-01, -1.4807e+00, -1.4728e+00, ..., -1.8363e+00,
-8.7078e-01, -1.7156e+00],
[-3.7154e-01, -2.6243e+00, -2.6416e+00, ..., -2.3365e+00,
-1.9380e+00, -1.9575e+00],
[-1.6457e-01, -2.2942e+00, -5.4066e+00, ..., -6.2763e-01,
-1.5958e+00, -2.1589e+00],
...,
[ 8.8874e-02, -6.2946e-01, -2.5235e-01, ..., -3.2786e-01,
-9.0123e-01, -7.6674e-01],
[-3.0463e-01, -4.7907e-01, 9.9999e-03, ..., -3.1761e-01,
2.1677e-02, -2.1257e-01],
[-5.7490e-01, -1.2694e+00, -1.0620e+00, ..., -1.0568e+00,
-1.1844e+00, -5.2058e-01]],
[[-2.7507e+00, -5.1942e+00, -4.6590e+00, ..., -5.9435e+00,
-6.0355e+00, -2.4033e+00],
[-5.4357e+00, -9.9327e+00, -1.3248e+01, ..., -1.1146e+01,
-9.4191e+00, -4.4237e+00],
[-4.1870e+00, -8.3633e+00, -1.1115e+01, ..., -1.0749e+01,
-6.9488e+00, -3.2898e+00],
...,
[-1.9029e+00, -2.3886e+00, -1.9415e+00, ..., -3.5779e+00,
-3.3976e+00, -1.2611e+00],
[-2.8648e+00, -3.0530e+00, -2.1000e+00, ..., -2.8961e+00,
-3.7315e+00, -1.1159e+00],
[ 3.7801e-01, -2.4242e-01, 3.4836e-01, ..., 3.3978e-01,
4.7130e-01, 1.8819e-01]],
[[-6.9369e-01, 4.5982e-01, 8.3857e-01, ..., 7.9671e-01,
6.2309e-01, 1.9465e+00],
[-9.5118e-01, -3.5830e-01, -5.0568e-01, ..., 9.4089e-01,
-4.1801e-02, 7.7865e-02],
[-1.7827e+00, -2.2093e+00, -4.1233e+00, ..., 1.1708e+00,
-3.3230e-05, -1.3653e-01],
...,
[-7.0935e-01, 1.0503e-01, 2.5273e-01, ..., -8.6911e-01,
3.2222e-01, 3.6138e-01],
[-4.9851e-01, 7.8335e-02, 2.3533e-01, ..., 1.3217e-02,
1.2640e-01, 5.5805e-01],
[-1.2414e-01, -1.8046e-01, 1.2685e-01, ..., -3.1305e-02,
8.4825e-02, 3.4292e-01]],
...,
[[ 1.2835e+00, -3.5254e-01, -6.4701e-02, ..., 4.3482e-01,
-9.4802e-02, -2.1784e+00],
[-7.3000e-01, -6.3943e-01, 1.1577e-02, ..., -6.8519e-01,
-1.5120e+00, -2.1125e+00],
[-7.3968e-01, 4.9073e-01, 2.3350e+00, ..., -2.0847e+00,
-5.8738e-01, -1.4105e+00],
...,
[-6.7391e-01, -5.5317e-01, -3.4568e-01, ..., -2.3591e+00,
-1.2891e+00, -6.2850e-01],
[-2.2077e+00, -1.7494e+00, -1.2730e+00, ..., -1.7761e+00,
-1.5143e+00, -1.1194e+00],
[-6.2419e-01, -8.5004e-01, -4.6612e-01, ..., -3.0345e-01,
-3.6905e-01, -4.0352e-01]],
[[ 3.0016e-02, 8.3388e-04, 4.7814e-01, ..., -7.8625e-01,
1.0023e+00, 2.5947e-02],
[ 2.3150e-01, -5.2318e-01, 8.7983e-01, ..., -1.5248e+00,
3.8755e-01, -8.4700e-01],
[ 1.7467e-01, -7.1893e-01, -1.0156e+00, ..., -5.0502e-01,
1.1572e-01, -3.7844e-01],
...,
[-2.1055e-01, -5.3691e-01, 2.1413e-01, ..., 5.3816e-01,
-1.7477e-01, -2.1525e-01],
[-5.0527e-01, -1.0158e+00, 3.4400e-02, ..., -4.2963e-01,
-2.9966e-01, -1.9424e-01],
[ 3.8651e-01, -4.5610e-02, -1.7442e-01, ..., -1.1770e-01,
-8.0500e-02, 1.6991e-01]],
[[-1.6108e+00, -8.5049e-01, -4.7012e-01, ..., -3.1693e-01,
-3.3022e-01, 8.7326e-01],
[-2.2058e+00, -9.0203e-01, -1.1465e+00, ..., -1.1722e-01,
-2.2518e-01, 1.6812e+00],
[-1.5697e+00, -1.1849e+00, -3.9889e+00, ..., 4.1859e-01,
-1.3795e-01, 1.4013e+00],
...,
[-6.4219e-01, -2.6651e-01, -1.7735e-01, ..., 1.0301e+00,
1.0460e+00, 9.0387e-01],
[-1.1050e+00, -5.7930e-01, -3.1883e-01, ..., 3.3772e-01,
4.7943e-02, 1.9868e-01],
[-3.5283e-01, -4.7500e-02, -1.5115e-01, ..., -2.4661e-01,
-2.2809e-01, 3.6339e-01]]],
[[[-2.5269e-01, -2.6345e+00, -2.8368e+00, ..., -2.0661e+00,
-8.8482e-01, -1.9649e+00],
[-1.4142e-01, -6.2020e+00, -3.5088e+00, ..., -4.0777e+00,
-5.7843e+00, -3.4094e+00],
[-1.1690e+00, -8.2522e+00, -1.7858e+00, ..., -4.2222e+00,
-4.8491e+00, 2.3056e+00],
...,
[ 1.1690e-01, -9.6665e-01, 1.2805e+00, ..., -8.3043e+00,
-1.1306e+01, -3.3364e+00],
[-1.3492e-01, -8.2290e-01, 1.7225e+00, ..., -5.2213e-01,
-5.7478e+00, -3.5553e+00],
[-3.1077e-01, -1.2825e+00, 2.8504e-01, ..., 5.6886e-01,
-3.0877e+00, -3.1193e+00]],
[[-6.4829e+00, -9.9657e+00, -1.3053e+01, ..., -6.7352e+00,
-6.4581e+00, -2.3343e+00],
[-1.1166e+01, -1.5967e+01, -2.0524e+01, ..., -1.8745e+01,
-1.9100e+01, -6.0616e+00],
[-8.2756e+00, -1.0038e+01, -1.2009e+01, ..., -1.4876e+01,
-1.4497e+01, -1.3750e+00],
...,
[-2.7537e+00, -3.7666e+00, -9.2332e+00, ..., -1.8900e+01,
-1.8541e+01, -2.0570e+00],
[-3.6028e+00, -4.2487e+00, -3.9938e+00, ..., -7.1041e+00,
-8.9176e+00, -5.9431e-01],
[ 6.0945e-01, -5.2329e-02, 6.1467e-01, ..., 3.5240e-01,
1.6425e+00, 9.3283e-01]],
[[-9.3975e-01, 2.3754e-01, 1.8603e+00, ..., 9.8649e-01,
5.4520e-01, 2.0655e+00],
[-1.9661e+00, -3.5025e+00, -2.0445e+00, ..., -3.5818e-01,
-2.1071e+00, 6.2067e-01],
[-1.7542e+00, -7.9750e+00, -1.0672e+01, ..., -8.2041e+00,
-1.0554e+01, -2.3967e+00],
...,
[-1.1183e+00, 3.2859e-01, 7.6868e-01, ..., -1.9155e+01,
-1.4640e+01, -7.5090e+00],
[-9.1445e-01, 3.0320e-01, 1.3050e+00, ..., -1.2062e+01,
-1.2011e+01, -1.2897e+00],
[-2.6109e-01, 1.4429e-01, 2.4279e+00, ..., -2.2648e+00,
-1.9183e+00, 1.7599e+00]],
...,
[[ 1.8079e+00, 1.3371e-01, 9.6679e-01, ..., -2.7505e-02,
-4.5292e-02, -2.3597e+00],
[ 6.7751e-01, 5.3589e-01, 2.6374e-01, ..., -5.5443e-01,
-1.0775e+00, -2.6449e+00],
[ 3.6154e-02, -1.8958e+00, -5.2531e+00, ..., -9.0002e-01,
-4.6843e-01, -6.1799e+00],
...,
[-7.4441e-01, -8.1369e-01, -3.7493e-02, ..., -4.0680e+00,
-8.1300e+00, -8.5105e+00],
[-2.1836e+00, -1.7130e+00, 1.3338e-01, ..., 5.0383e-01,
-3.0528e+00, -6.5352e+00],
[-6.4210e-01, -9.1454e-01, -1.9534e+00, ..., -2.3307e+00,
-1.9858e+00, -3.6250e+00]],
[[ 8.1763e-01, 7.3080e-02, 4.4819e-01, ..., -5.2819e-01,
1.1646e+00, 3.0317e-01],
[ 7.9472e-01, -2.3608e+00, -1.6753e+00, ..., -1.1177e+00,
-1.6771e+00, -2.2740e+00],
[ 3.9204e-01, -6.5205e+00, -4.7049e+00, ..., -2.4843e+00,
-3.8285e+00, -6.0843e-01],
...,
[-5.7029e-02, -7.0144e-01, 8.4373e-01, ..., -5.2812e+00,
-5.8603e+00, -6.0057e-01],
[-4.9190e-01, -1.4933e+00, -2.9538e-01, ..., -3.7413e+00,
-5.1837e+00, 3.3677e-02],
[ 5.6687e-01, -1.7042e-01, 3.7605e-01, ..., -6.1926e-03,
-2.4261e+00, -8.2818e-01]],
[[-2.1740e+00, -1.6153e+00, -1.4370e+00, ..., -5.3503e-01,
-5.0693e-01, 8.4008e-01],
[-4.2293e+00, -3.5647e+00, -6.1822e+00, ..., -2.7699e+00,
-2.4025e+00, -8.2159e-01],
[-4.5474e+00, -4.0692e+00, -8.5807e+00, ..., -4.7080e+00,
-8.8696e+00, -5.6379e+00],
...,
[-6.1351e-01, -4.6972e-01, -8.9270e-01, ..., -1.9525e+00,
-1.1152e+01, -9.1820e+00],
[-9.4902e-01, -7.8422e-01, -1.7572e+00, ..., 2.8613e+00,
-1.5089e+00, -4.7465e+00],
[-3.1663e-01, -3.5176e-01, -1.4761e+00, ..., 1.2865e+00,
2.2603e+00, 9.6536e-01]]],
[[[-2.0909e-01, -5.8466e-01, -2.0808e-01, ..., -2.8028e-01,
1.7413e-01, -2.3592e-01],
[-8.2154e-02, -1.7236e+00, -1.0260e+00, ..., -8.5561e-01,
-6.2511e-01, -6.6480e-01],
[-3.2993e-02, -1.4276e+00, -1.5034e+00, ..., -6.3635e-01,
-5.2337e-01, -1.0648e+00],
...,
[-9.1559e+00, -2.7905e+01, -1.8677e+01, ..., -1.2904e+00,
-1.2033e+00, -2.3998e+00],
[-1.1950e+01, -2.2068e+01, -1.5538e+01, ..., -8.1459e-01,
-7.0983e-01, -2.0436e+00],
[-6.5002e+00, -1.3818e+01, -9.1341e+00, ..., -1.6932e+00,
-1.7798e+00, -1.1472e+00]],
[[-9.0015e-01, -2.5045e+00, -1.8513e+00, ..., -1.8868e+00,
-2.6390e+00, -8.9101e-01],
[-2.6656e+00, -4.9016e+00, -3.5172e+00, ..., -3.5552e+00,
-4.6443e+00, -1.7593e+00],
[-2.3796e+00, -4.8322e+00, -7.8386e+00, ..., -3.5758e+00,
-4.0620e+00, -1.7698e+00],
...,
[-3.0397e+01, -3.3544e+01, -2.4575e+01, ..., -6.0255e+00,
-6.9494e+00, -3.0736e+00],
[-3.2210e+01, -4.5284e+01, -3.3519e+01, ..., -7.0010e+00,
-8.3252e+00, -2.8999e+00],
[-6.5437e+00, -8.4629e+00, 6.6740e+00, ..., 3.4029e-01,
5.5002e-01, 2.9311e-01]],
[[-4.4125e-01, 1.7062e-01, 6.9952e-01, ..., 6.9849e-01,
4.0004e-01, 1.1221e+00],
[-5.0861e-01, -2.7568e-01, 5.1709e-01, ..., 4.4365e-01,
3.8539e-01, 4.1083e-01],
[-1.0057e+00, -2.0445e-01, -5.9439e-01, ..., 8.0219e-02,
1.7334e-01, 2.3409e-01],
...,
[-1.7036e+01, -2.9850e+01, -2.8059e+01, ..., -8.2134e-02,
-1.2423e-01, 1.7302e-01],
[-1.8105e+01, -3.2487e+01, -1.4166e+01, ..., -8.4540e-02,
3.7788e-03, 2.3081e-01],
[-1.5562e+01, -2.4781e+01, 1.6084e+00, ..., -1.2383e-01,
-1.4062e-01, -5.4278e-01]],
...,
[[ 1.7278e-01, -2.6393e-01, -1.1684e-01, ..., 6.6262e-02,
-6.3939e-02, -7.7695e-01],
[-7.6133e-01, -4.4641e-01, -3.5329e-01, ..., -2.5043e-01,
-1.0317e+00, -9.3429e-01],
[-5.2408e-01, -2.2025e-01, 1.2335e-01, ..., -2.8114e-02,
-2.2110e-01, -6.4922e-01],
...,
[ 6.4945e+00, -6.8110e+00, -5.8973e+00, ..., -3.4777e-01,
-5.3978e-01, -1.1815e+00],
[ 3.2098e+00, -6.6187e+00, -9.5538e+00, ..., -2.1745e+00,
-2.2402e+00, -2.3288e+00],
[ 6.3949e+00, -1.8338e+00, -5.8038e+00, ..., -8.1941e-01,
-9.2446e-01, -1.5506e+00]],
[[-7.2824e-02, 1.5519e-02, 2.6275e-01, ..., 6.2877e-02,
5.0957e-01, -2.7835e-02],
[-2.7831e-01, -2.5849e-01, 7.1877e-02, ..., 8.2973e-02,
2.6703e-01, -2.2572e-01],
[-4.4900e-01, -4.6471e-01, 4.4143e-01, ..., 1.0069e-01,
9.9981e-02, -2.3444e-01],
...,
[-1.1716e+01, -9.9181e+00, 1.3550e+01, ..., 3.2458e-01,
1.0543e-01, -7.6657e-01],
[-1.0218e+01, -1.3483e+01, 7.0445e+00, ..., 1.7157e-01,
4.3695e-02, -9.1649e-01],
[-4.9234e+00, -5.5535e+00, -3.2067e-01, ..., 4.1336e-01,
4.9076e-01, 5.3059e-01]],
[[-5.1478e-01, 3.5668e-01, 5.6164e-01, ..., 5.1105e-01,
4.0297e-01, 8.2389e-01],
[-8.8277e-01, 9.6136e-02, -1.8278e-01, ..., -2.2210e-01,
-1.1595e-01, 8.8136e-01],
[-4.9726e-01, 2.4624e-01, -6.0091e-01, ..., 1.2407e-01,
2.1949e-01, 8.2206e-01],
...,
[ 2.4968e+00, -9.2664e+00, -1.5359e+01, ..., -5.8618e-01,
-2.4161e-01, 8.5304e-01],
[ 4.2146e+00, 1.5674e+00, 4.5879e+00, ..., -1.1099e+00,
-7.7922e-01, 4.7728e-01],
[ 1.3583e+00, 3.9249e+00, -1.5453e+00, ..., -3.7351e-01,
-2.3303e-01, 1.3560e+00]]]], device='cuda:0'),
tensor([[[[-7.2179e-01, -1.4751e+00, -1.4749e+00, ..., -1.2838e+00,
-8.8847e-01, -1.7192e+00],
[-3.6413e-01, -2.4719e+00, -2.3264e+00, ..., -1.8098e+00,
-1.9486e+00, -1.9630e+00],
[-1.7742e-01, -1.9676e+00, -1.8207e+00, ..., -1.4776e+00,
-1.5041e+00, -2.1915e+00],
...,
[-8.7275e-02, -1.1508e+00, -1.5618e+00, ..., -2.7760e-01,
-5.2784e-01, -9.7527e-01],
[-6.5200e-02, -9.0996e-01, -2.3208e+00, ..., 1.2794e+00,
-7.7514e-02, -1.0137e+00],
[-1.2108e-01, -1.0061e+00, -1.3230e+00, ..., -4.5481e-01,
-1.1017e+00, -6.3537e-01]],
[[-2.6704e+00, -5.1600e+00, -4.7266e+00, ..., -4.8243e+00,
-6.0489e+00, -2.4142e+00],
[-5.4567e+00, -8.2689e+00, -7.5520e+00, ..., -7.9323e+00,
-9.4298e+00, -4.4649e+00],
[-4.1746e+00, -5.9992e+00, -5.8572e+00, ..., -6.4331e+00,
-6.6608e+00, -3.3282e+00],
...,
[-2.1380e+00, -3.8671e+00, -6.1926e+00, ..., -4.8040e+00,
-3.7173e+00, -1.5677e+00],
[-4.5963e+00, -6.3572e+00, -7.8294e+00, ..., -6.3002e+00,
-6.3910e+00, -2.4622e+00],
[ 7.2563e-01, -1.3948e-01, 5.2595e-01, ..., -1.3025e-01,
6.0979e-01, 7.6414e-03]],
[[-6.6356e-01, 4.3975e-01, 8.2640e-01, ..., 8.2918e-01,
6.2626e-01, 1.9096e+00],
[-9.2599e-01, 3.0426e-02, 4.1829e-01, ..., 5.5390e-01,
1.0616e-02, 1.0241e-01],
[-1.7743e+00, -2.0422e-01, -7.2179e-02, ..., -5.4114e-03,
-4.3854e-03, -1.0347e-01],
...,
[-9.5545e-01, -1.2512e-01, -2.0967e+00, ..., 6.6611e-01,
1.1306e-01, 3.3469e-01],
[-1.0224e+00, 1.4530e-01, -2.8570e+00, ..., 1.8455e-01,
1.4825e-01, 3.7078e-01],
[-5.0043e-01, 1.9469e-01, -3.6141e+00, ..., 8.6540e-01,
9.9680e-02, 1.6710e-02]],
...,
[[ 1.2848e+00, -3.3910e-01, -4.4403e-02, ..., 2.2858e-01,
-9.8562e-02, -2.1639e+00],
[-7.3561e-01, -9.2983e-01, -5.0054e-01, ..., -5.6246e-01,
-1.5333e+00, -2.1301e+00],
[-7.0749e-01, -9.0030e-01, -3.4181e-01, ..., -3.2119e-01,
-6.0188e-01, -1.4189e+00],
...,
[-4.6593e-01, 3.6704e-02, 1.3914e+00, ..., -1.2593e+00,
-9.0374e-02, -5.6465e-01],
[-2.0554e+00, -1.1708e+00, -5.8595e-01, ..., -2.9781e+00,
-1.1636e+00, -1.1310e+00],
[-3.2753e-01, -4.6150e-01, 4.8698e-01, ..., -2.9118e+00,
-3.5630e-01, -6.3800e-01]],
[[ 4.8816e-02, 1.4030e-02, 4.9381e-01, ..., 5.1190e-01,
9.4812e-01, -3.1194e-02],
[ 2.2685e-01, -7.9052e-01, -2.4749e-02, ..., 4.5243e-01,
3.8568e-01, -9.0410e-01],
[ 1.6191e-01, -6.3874e-01, 2.9225e-01, ..., 2.8460e-01,
5.4659e-01, -4.2506e-01],
...,
[-3.9828e-01, -4.4572e-01, -7.3117e-02, ..., -5.5968e-01,
-4.4332e-03, -1.6991e-01],
[-7.3324e-01, -1.3717e+00, -1.3112e+00, ..., -1.4964e-02,
-3.5638e-01, -7.2078e-01],
[ 5.2144e-01, 1.4509e-02, -1.4168e+00, ..., 9.7580e-01,
1.5814e-01, 3.5481e-01]],
[[-1.6281e+00, -8.4866e-01, -4.7512e-01, ..., -4.6532e-01,
-3.0770e-01, 8.5858e-01],
[-2.2491e+00, -8.1283e-01, -6.1895e-01, ..., -7.9347e-01,
-1.8167e-01, 1.6724e+00],
[-1.6134e+00, -6.2232e-01, -2.7202e-01, ..., -3.5119e-01,
-1.0551e-01, 1.3960e+00],
...,
[-4.5602e-01, -1.3060e-01, -1.8550e+00, ..., 1.3742e+00,
3.0178e-01, 7.4358e-01],
[-8.6560e-01, -6.5607e-01, -1.8743e+00, ..., 7.8447e-01,
7.1066e-02, 4.4416e-01],
[-3.8273e-01, -5.3429e-01, 2.2947e-01, ..., 4.0784e-01,
-2.5972e-01, 6.0885e-01]]],
[[[-4.2556e-01, -9.9945e-01, 2.5334e-01, ..., -1.0139e+00,
-5.3842e-01, -8.9760e-01],
[-1.3970e-01, -1.5712e+00, 1.7368e-01, ..., -1.3247e+00,
-1.2459e+00, -1.2768e+00],
[-3.7086e-02, -9.9845e-01, -1.2440e+00, ..., -9.9058e-01,
-1.3275e+00, -1.8434e+00],
...,
[ 8.0566e-01, 1.1191e+00, 9.3580e-01, ..., -1.5373e-02,
-9.2559e-01, -6.7333e-01],
[-2.3741e-01, -4.9507e-01, 1.1208e-02, ..., 9.7428e-02,
2.2974e-01, -2.8509e-01],
[-5.2395e-01, -1.2986e+00, -1.0781e+00, ..., -1.0589e+00,
-1.1639e+00, -5.2673e-01]],
[[-2.1642e+00, -4.1109e+00, -1.2163e+01, ..., -4.5617e+00,
-5.9959e+00, -2.0321e+00],
[-3.8855e+00, -7.9694e+00, -1.1105e+01, ..., -6.9132e+00,
-8.1069e+00, -3.1257e+00],
[-3.6551e+00, -8.2319e+00, -1.0304e+01, ..., -7.5200e+00,
-6.4793e+00, -2.8777e+00],
...,
[-2.2489e+00, -2.5531e+00, -2.8615e+00, ..., -3.2579e+00,
-3.5828e+00, -1.1657e+00],
[-3.0409e+00, -3.2290e+00, -2.2965e+00, ..., -2.5777e+00,
-3.6462e+00, -1.1549e+00],
[ 4.1198e-01, -1.9490e-01, 3.6662e-01, ..., 3.6948e-01,
5.0327e-01, 2.0587e-01]],
[[-3.7583e-01, 6.1840e-01, -2.9576e-01, ..., 7.2376e-01,
4.8815e-01, 1.5864e+00],
[-6.1927e-01, -1.3380e-01, 1.2522e+00, ..., 2.5762e-01,
3.5055e-01, 5.2375e-01],
[-1.1842e+00, -1.0604e-01, 2.0503e+00, ..., 6.8866e-02,
4.8013e-01, 3.0941e-01],
...,
[ 6.7344e-01, 7.8518e-01, -5.5918e-01, ..., -2.9832e-01,
7.7941e-01, 4.2862e-01],
[-5.1683e-01, 1.2069e-01, 2.4648e-01, ..., 1.9403e-01,
2.3484e-01, 5.3730e-01],
[-1.3105e-01, -1.4381e-01, 1.3953e-01, ..., 5.2929e-03,
1.2819e-01, 3.5378e-01]],
...,
[[ 1.0465e+00, 3.7660e-01, 2.5815e+00, ..., 5.1594e-02,
1.4131e-01, -1.1374e+00],
[-4.8829e-01, -5.1608e-01, 7.0456e-01, ..., -1.7571e+00,
-1.7510e+00, -1.4467e+00],
[-1.9584e-01, 3.5673e-01, -2.8822e-02, ..., -2.0467e+00,
-8.2704e-01, -1.1594e+00],
...,
[-2.3290e+00, -3.7397e+00, -3.7477e+00, ..., -1.6485e+00,
-6.5650e-01, -5.8416e-01],
[-2.1704e+00, -1.7409e+00, -1.2467e+00, ..., -1.3439e+00,
-1.4145e+00, -1.1133e+00],
[-6.3616e-01, -8.4793e-01, -4.5645e-01, ..., -2.7998e-01,
-3.9907e-01, -4.2594e-01]],
[[ 1.9688e-01, 1.0156e-01, 1.5775e+00, ..., 6.7868e-01,
4.6486e-01, -3.4701e-01],
[ 4.3907e-01, -5.2118e-01, 2.1353e+00, ..., 1.2068e+00,
3.3822e-01, -7.4024e-01],
[ 1.1822e-01, -1.9746e-01, 2.2727e+00, ..., 1.3473e+00,
4.2734e-02, -6.7952e-01],
...,
[ 4.7821e-01, -4.7644e-01, -1.8724e+00, ..., 1.1937e-01,
-1.7389e-01, -1.8378e-01],
[-4.8848e-01, -1.0489e+00, 1.7081e-02, ..., 5.1057e-03,
-5.8102e-02, -2.4350e-01],
[ 4.0730e-01, -1.0254e-01, -1.8975e-01, ..., -9.3443e-02,
-4.3087e-02, 1.7367e-01]],
[[-6.3507e-01, -9.0885e-02, -6.8809e-01, ..., -3.3202e-01,
3.1466e-01, 8.9357e-01],
[-1.2931e+00, -6.4487e-01, -2.0375e+00, ..., -7.4383e-01,
-1.8003e-01, 9.4833e-01],
[-1.1051e+00, -1.3246e+00, -2.2094e+00, ..., 5.5299e-01,
4.9465e-01, 7.7758e-01],
...,
[-1.1684e+00, 2.8051e-01, 2.3721e+00, ..., 1.3873e+00,
7.7195e-01, 6.5340e-01],
[-1.0740e+00, -6.7674e-01, -4.3546e-01, ..., -3.6777e-01,
-2.1766e-01, 2.5636e-01],
[-3.4775e-01, -9.8025e-02, -1.7679e-01, ..., -2.2723e-01,
-2.0538e-01, 4.3161e-01]]],
[[[-1.9398e-01, -2.6467e-01, 1.7699e-01, ..., 2.4811e-02,
4.7804e-01, 1.4106e-01],
[-7.1466e-02, -1.5438e+00, -9.1695e-01, ..., -7.6273e-01,
-5.4822e-01, -3.4795e-01],
[-6.9132e-02, -1.2491e+00, -6.5090e-01, ..., -5.0463e-01,
-4.2829e-01, -6.8966e-01],
...,
[-1.8504e-01, -1.9105e+00, -1.3738e+00, ..., -1.2019e+00,
-1.1568e+00, -1.9741e+00],
[-4.4806e-01, -1.5211e+00, -5.1329e-01, ..., 2.7216e-01,
-4.2009e-01, -1.4616e+00],
[-2.3513e-01, -1.3249e+00, -1.0549e+00, ..., -9.4909e-01,
-1.1417e+00, -7.6728e-01]],
[[-2.0657e-01, -1.4948e+00, -6.6055e-01, ..., -7.1884e-01,
-1.4084e+00, -2.9099e-01],
[-2.0882e+00, -4.1372e+00, -2.5346e+00, ..., -2.6088e+00,
-3.5113e+00, -1.2474e+00],
[-1.9155e+00, -3.9597e+00, -4.2279e+00, ..., -2.6530e+00,
-3.0926e+00, -1.3153e+00],
...,
[-3.6715e+00, -5.1444e+00, -6.2425e+00, ..., -6.4500e+00,
-6.0197e+00, -2.8810e+00],
[-5.4794e+00, -6.5822e+00, -6.7033e+00, ..., -6.9834e+00,
-6.6151e+00, -2.7332e+00],
[ 5.3707e-01, -5.0658e-01, 6.1662e-01, ..., 1.5447e-01,
6.6737e-01, 1.0788e-01]],
[[-5.4213e-01, -1.5757e-01, 4.0019e-01, ..., 3.7369e-01,
1.1358e-01, 9.1368e-01],
[-3.6104e-01, -4.6748e-01, 5.0280e-01, ..., 3.2113e-01,
2.6612e-01, 5.4297e-01],
[-8.4741e-01, -2.2919e-01, 1.5007e-01, ..., 1.5716e-02,
1.0964e-01, 4.2913e-01],
...,
[-1.7345e+00, -4.0023e-01, -1.0217e-01, ..., 9.1680e-02,
-8.0418e-02, -1.7836e-01],
[-1.4422e+00, -3.1148e-01, -2.6965e-01, ..., 4.9362e-01,
-1.9345e-01, -3.2742e-01],
[-6.0917e-01, -8.5201e-02, 2.1833e-01, ..., 6.2298e-02,
-5.7028e-02, -4.9032e-01]],
...,
[[ 7.0012e-02, -6.0966e-01, -3.7751e-01, ..., -7.3273e-02,
-2.3923e-01, -7.2141e-01],
[-8.3520e-01, -6.3149e-01, -5.7446e-01, ..., -2.6746e-01,
-1.0108e+00, -7.7722e-01],
[-4.7204e-01, -9.6485e-02, 5.1627e-02, ..., 1.6768e-01,
-1.9159e-01, -5.6793e-01],
...,
[-1.0343e+00, -8.9055e-01, -2.5411e-01, ..., -2.7716e-01,
-5.8891e-01, -1.2296e+00],
[-2.5979e+00, -2.0152e+00, -7.7723e-01, ..., -2.0230e+00,
-1.5122e+00, -1.5746e+00],
[-5.6044e-01, -1.1544e+00, -9.1546e-03, ..., -1.2603e+00,
-4.1479e-01, -5.2604e-01]],
[[-1.7176e-01, 5.4366e-02, 4.2622e-01, ..., 7.5185e-02,
5.2104e-01, 1.3135e-01],
[-4.1322e-01, -2.2579e-01, -1.7695e-02, ..., -6.8753e-02,
7.4689e-02, -1.2504e-01],
[-6.1937e-01, -4.2070e-01, 2.8191e-01, ..., 7.0937e-02,
1.1304e-01, -4.4117e-02],
...,
[-1.6575e-01, -8.4765e-01, 4.5347e-01, ..., -3.0504e-02,
5.2245e-01, -3.2833e-01],
[-7.8589e-01, -1.9570e+00, 1.6849e-01, ..., -1.9501e-01,
1.2505e-01, -5.4536e-01],
[ 7.4008e-01, 3.6174e-02, 3.2591e-01, ..., 6.0849e-01,
5.0941e-01, 2.8047e-01]],
[[-6.8431e-01, 4.1257e-01, 5.7876e-01, ..., 4.9428e-01,
4.0020e-01, 6.1210e-01],
[-8.4539e-01, 2.9640e-01, -2.3282e-01, ..., -2.7740e-01,
-1.4949e-01, 6.7206e-01],
[-4.8516e-01, 3.2946e-01, 3.4905e-02, ..., 5.0494e-03,
9.5485e-02, 6.0810e-01],
...,
[-1.3058e+00, -6.9885e-01, -2.3933e-01, ..., -1.5920e-01,
-4.1798e-02, 1.4057e+00],
[-1.4259e+00, -8.8489e-01, -4.8911e-01, ..., -1.9557e-01,
4.2992e-02, 1.2288e+00],
[-6.4560e-01, -4.6468e-01, -2.1896e-01, ..., 5.3856e-02,
1.6903e-01, 1.2919e+00]]],
[[[-2.9508e-01, -1.0048e+00, -8.4198e-01, ..., 3.9179e-03,
5.9539e-01, 5.4672e-01],
[ 1.8991e-01, -1.8439e+00, -1.2990e+00, ..., -1.0127e+00,
-4.0887e-01, 9.7961e-02],
[ 1.7673e-01, -1.3873e+00, -9.6325e-01, ..., -7.9415e-01,
-1.9170e-01, -1.0519e-01],
...,
[ 1.5724e-01, -1.2283e+00, -8.9586e-01, ..., -3.2530e-01,
-7.7435e-02, -1.2214e-01],
[ 1.7566e-01, -8.4965e-01, -4.4186e-01, ..., 2.3012e-02,
2.6561e-01, -1.7562e-02],
[-8.2935e-02, -1.0601e+00, -9.4135e-01, ..., -1.0012e+00,
-1.0587e+00, -3.8069e-01]],
[[-2.1065e+00, -4.0779e+00, -3.6893e+00, ..., -4.6604e-01,
-1.0224e+00, -1.4759e-02],
[-3.9379e+00, -6.3232e+00, -5.5159e+00, ..., -3.0781e+00,
-3.6305e+00, -7.2062e-01],
[-3.1825e+00, -4.8702e+00, -4.6514e+00, ..., -2.7546e+00,
-2.6621e+00, -4.4357e-01],
...,
[-3.1291e+00, -4.5678e+00, -4.6344e+00, ..., -1.2984e+00,
-1.5302e+00, -3.6856e-01],
[-5.0320e+00, -6.0204e+00, -5.7339e+00, ..., -2.4073e+00,
-3.3182e+00, -9.5409e-01],
[ 7.7329e-01, -3.5545e-02, 4.9591e-01, ..., 3.6037e-01,
3.6844e-01, 8.1556e-02]],
[[-5.6844e-01, 4.1238e-01, 8.1877e-01, ..., 8.1679e-01,
3.0469e-01, 1.0994e+00],
[-8.9067e-01, 8.5077e-02, 6.6098e-01, ..., 7.3186e-01,
2.5820e-01, 4.7865e-01],
[-1.2599e+00, 1.8669e-02, 4.8628e-02, ..., -7.8100e-03,
4.5473e-01, 4.1375e-01],
...,
[-1.3364e+00, 6.4650e-02, -6.8465e-03, ..., 5.1950e-02,
7.3695e-02, 4.1589e-01],
[-1.2781e+00, 2.3495e-01, -7.6900e-02, ..., 1.0951e-01,
1.0621e-01, 5.9955e-01],
[-5.4437e-01, 2.8794e-01, 1.3770e-01, ..., -7.6983e-02,
6.0867e-02, 3.5571e-01]],
...,
[[ 6.7083e-01, 6.3972e-02, 2.4888e-01, ..., -1.1852e-01,
-3.9144e-01, -9.3600e-01],
[-7.9940e-01, -3.7082e-01, -1.1660e-01, ..., -7.4709e-01,
-1.1801e+00, -8.3734e-01],
[-4.9476e-01, -3.9124e-01, -8.5910e-02, ..., -1.5153e+00,
-8.9801e-01, -3.7051e-01],
...,
[-6.0489e-01, -5.2506e-01, -6.7554e-02, ..., -1.0055e-01,
-1.8177e-01, -3.5650e-01],
[-2.0484e+00, -1.7014e+00, -1.2364e+00, ..., -8.1741e-01,
-8.8270e-01, -7.8572e-01],
[-3.8417e-01, -9.3779e-01, -3.3709e-01, ..., 7.5756e-03,
-1.2885e-01, -3.2114e-01]],
[[ 4.1876e-02, -2.2731e-01, -7.7944e-02, ..., 3.2229e-01,
6.7253e-01, 2.8628e-01],
[ 1.6040e-01, -8.5589e-01, -8.3716e-02, ..., -5.8830e-01,
-2.8483e-01, -1.0494e-01],
[-1.5134e-01, -9.1993e-01, 2.8087e-02, ..., -4.0185e-02,
-2.3846e-01, 1.4044e-01],
...,
[-5.8142e-02, -9.7102e-01, 8.2763e-02, ..., -1.5293e-01,
-9.8675e-02, 1.5837e-01],
[-4.9176e-01, -1.8947e+00, -2.4229e-01, ..., -3.7446e-01,
-3.1172e-01, -6.1810e-02],
[ 7.6475e-01, -1.1637e-01, 2.4777e-01, ..., -2.2812e-01,
-2.2607e-01, 3.0130e-01]],
[[-6.4366e-01, -1.0688e-01, 1.9171e-02, ..., 5.5902e-01,
3.7342e-01, 5.3011e-01],
[-1.0743e+00, -7.1690e-01, -9.2236e-01, ..., -4.1503e-01,
-7.4508e-02, 4.0392e-01],
[-6.8500e-01, -6.1595e-01, -5.0906e-01, ..., -1.0521e+00,
2.3325e-01, 7.7541e-02],
...,
[-6.0843e-01, -8.3918e-01, -5.2085e-01, ..., 6.1409e-02,
-7.2094e-02, 9.4230e-02],
[-7.9770e-01, -1.0190e+00, -5.7049e-01, ..., -6.5112e-02,
-2.4472e-01, -2.0839e-01],
[-4.6870e-01, -8.2774e-01, -3.7290e-01, ..., -9.3513e-02,
-2.7813e-01, 1.1004e-01]]],
[[[-1.9691e+00, -1.9769e+00, -1.1134e+00, ..., -1.1391e+00,
-4.8814e-01, -1.6233e+00],
[-3.8358e+00, -3.7915e+00, -3.3701e+00, ..., -1.4925e+00,
-1.3844e+00, -1.6858e+00],
[ 2.5018e+00, 6.3969e-01, 1.3813e-01, ..., -1.2965e+01,
-9.6198e+00, -5.7898e+00],
...,
[ 1.7700e+00, -1.7415e+00, -4.5704e-01, ..., -2.4512e+00,
-1.6506e+01, -6.5530e+00],
[ 4.9236e-01, -7.3229e-01, 5.7543e-01, ..., -1.8258e+01,
-2.1172e+01, -1.0363e+01],
[-2.8670e-01, -1.3416e+00, -1.1283e+00, ..., -1.0976e+01,
-6.5956e+00, -5.3354e+00]],
[[-5.6690e+00, -7.1807e+00, -4.0138e+00, ..., -3.7139e+00,
-5.5648e+00, -1.7032e+00],
[-1.1561e+01, -1.6635e+01, -1.4387e+01, ..., -6.4029e+00,
-8.8482e+00, -3.4229e+00],
[-6.2899e+00, -8.0617e+00, -7.3653e+00, ..., -3.9795e+01,
-3.0786e+01, -9.1532e+00],
...,
[-1.6578e+00, -4.9059e+00, -3.6868e+00, ..., -1.7824e+01,
-2.2671e+01, -1.1728e+00],
[-4.8122e+00, -6.0052e+00, -4.7268e+00, ..., -2.8648e+01,
-1.9095e+01, -2.6338e+00],
[ 6.6872e-01, -5.2174e-01, 5.4751e-01, ..., 2.6808e+00,
3.8108e+00, -2.8324e+00]],
[[-1.2427e-01, 8.3779e-01, 9.7890e-01, ..., 6.5198e-01,
6.2306e-02, 1.5085e+00],
[-2.3767e-01, 2.4585e+00, 2.9086e+00, ..., 5.2466e-01,
-1.8169e-01, -9.3138e-02],
[-1.2012e+01, -6.3702e+00, -2.4671e+00, ..., 3.1410e+00,
1.9652e+00, 1.0762e+00],
...,
[-4.5156e+00, -1.7317e+00, -3.2492e+00, ..., -4.4182e+01,
-3.2896e+01, -7.7837e+00],
[-9.8348e-01, 2.3938e-01, -3.7202e-01, ..., -3.5704e+01,
-2.7051e+01, -8.9458e+00],
[-8.3459e-01, -1.5524e-01, 5.7779e-02, ..., -7.7294e+00,
-7.6391e+00, -2.4956e+00]],
...,
[[ 5.0596e-01, -6.2072e-01, -4.6909e-01, ..., 4.5011e-01,
5.0262e-01, -1.3450e+00],
[-6.4790e-01, -3.7285e+00, -3.4780e+00, ..., -3.3515e-01,
-8.1830e-01, -1.4315e+00],
[-2.7797e+00, -8.4010e+00, -5.9478e+00, ..., -2.5735e+00,
-1.5808e+00, -8.3904e-02],
...,
[-7.6545e-01, -2.1264e+00, -1.6163e+00, ..., -2.1731e+01,
-1.6143e+01, -1.7474e+01],
[-3.3711e+00, -3.5386e+00, -3.6716e+00, ..., -7.7142e+00,
-5.2540e+00, -1.4112e+01],
[-5.7773e-01, -1.2486e+00, -3.6494e-01, ..., 2.6232e-01,
-4.3488e+00, -1.2157e+01]],
[[ 2.4018e-01, -6.3018e-01, -6.6948e-01, ..., 1.3935e-01,
5.3699e-01, -7.1132e-02],
[ 1.2695e+00, 7.6734e-01, 3.5449e-02, ..., 1.9954e-01,
1.0255e-01, -7.3935e-01],
[-1.8500e+00, -9.9844e-01, 3.2453e-01, ..., -5.1731e+00,
-6.2669e+00, -5.4731e+00],
...,
[-1.0390e+00, -2.6512e+00, -2.5812e-01, ..., -2.0133e+01,
-2.4149e+01, -1.0815e+01],
[-6.8733e-01, -2.1577e+00, -1.1409e+00, ..., -1.5653e+01,
-1.2661e+01, -5.4494e+00],
[ 5.1316e-01, -7.0868e-02, -1.8680e-02, ..., 5.2624e-01,
2.1776e+00, 3.2509e+00]],
[[-1.2075e+00, -4.8900e-01, 4.9445e-01, ..., -3.7152e-01,
-3.0828e-01, 4.6984e-01],
[-4.5907e+00, -4.2786e+00, -2.4274e+00, ..., -1.0174e+00,
-4.9566e-01, 1.0193e+00],
[-9.5940e+00, -1.0327e+01, -9.0292e+00, ..., -1.0217e+01,
-9.1138e+00, -4.9020e+00],
...,
[ 1.7237e+00, 1.3237e+00, 9.2583e-01, ..., -7.7271e+00,
-1.6635e+01, -1.4656e+01],
[-2.0515e-01, 1.0347e+00, 1.9198e+00, ..., -1.0259e+01,
-9.3393e+00, -4.8488e+00],
[-2.3601e-01, -2.7852e-01, -8.4845e-03, ..., -3.3241e+00,
-1.0936e+00, 1.8397e+00]]]], device='cuda:0'),
tensor([[[[-5.8230e+00, -4.8657e+00, 6.0003e+00, ..., -4.3084e-01,
1.1042e-01, -6.3754e-01],
[-1.2558e+00, -2.8663e+00, -6.4036e+00, ..., -1.9722e+00,
-1.3527e+00, -1.6139e+00],
[-7.8312e+00, -1.2797e+01, -2.3916e+01, ..., -1.0747e+00,
-1.3866e+00, -2.8347e+00],
...,
[ 1.1713e+00, -4.1500e+00, -3.8209e+00, ..., -3.6647e+00,
-6.3828e+00, -1.0086e+01],
[-1.5048e+00, -2.3343e+00, -1.4926e+00, ..., 7.5692e-01,
-4.1297e-01, -4.8108e+00],
[-1.0383e+00, -2.1823e+00, -1.6766e+00, ..., -1.7168e-01,
-3.8274e-01, -1.2724e+00]],
[[-4.3727e+00, -8.6496e+00, -5.9798e+00, ..., -3.2920e+00,
-4.5000e+00, -1.7233e+00],
[-9.9353e+00, -1.8620e+01, -2.3613e+01, ..., -1.0744e+01,
-1.1137e+01, -4.2640e+00],
[-9.6273e+00, -2.1291e+01, -1.5923e+01, ..., -8.3896e+00,
-1.2524e+01, -4.7033e+00],
...,
[-4.5179e+00, -8.9259e+00, -9.0424e+00, ..., -8.7711e+00,
-1.0682e+01, -3.8840e+00],
[-7.0090e+00, -8.3166e+00, -7.1942e+00, ..., -5.9497e+00,
-6.7973e+00, -5.1554e+00],
[-3.4345e-01, -1.0645e+00, 6.8154e-02, ..., 6.1847e-01,
7.8288e-01, -7.4494e-01]],
[[-8.2384e+00, -1.5536e+01, -1.6432e+01, ..., 7.6842e-01,
4.6306e-01, 1.3022e+00],
[-1.8955e+01, -2.6371e+01, -3.3039e+01, ..., 9.1997e-01,
6.2725e-01, 2.9748e-01],
[-2.6768e+01, -2.8251e+01, -4.1274e+01, ..., 5.8786e-01,
1.1074e+00, 1.2083e+00],
...,
[-2.8327e+00, 3.0296e+00, -2.0113e+00, ..., -7.2327e+00,
-9.1638e+00, -3.0452e+00],
[-1.4627e+00, -1.6019e-02, -1.0343e-01, ..., -1.0414e+00,
-2.1295e+00, -2.3112e+00],
[-1.5588e+00, -6.7242e-01, -4.2599e-01, ..., 6.2993e-01,
-1.3786e-01, -1.3094e+00]],
...,
[[ 3.1125e+00, -9.5790e+00, -2.0247e+01, ..., 4.0600e-01,
1.8523e-01, -9.2057e-01],
[-4.6526e+00, -2.1537e+01, -2.5741e+01, ..., -1.0551e+00,
-1.9610e+00, -1.5387e+00],
[-3.5182e+00, -1.3113e+01, -1.2528e+01, ..., -2.2082e+00,
-3.7642e+00, -3.6205e+00],
...,
[-5.3394e+00, -9.5466e+00, -7.1233e+00, ..., 3.1880e+00,
2.0412e+00, -3.3386e+00],
[-3.2584e+00, -3.6626e+00, -2.3045e+00, ..., -3.6745e+00,
-2.2004e+00, -3.6252e+00],
[-7.2650e-01, -2.0397e+00, -1.2492e+00, ..., -1.3200e+00,
-2.0496e+00, -1.5074e+00]],
[[-2.7347e+00, -6.7433e+00, -3.3749e+00, ..., 1.8644e-01,
6.8656e-01, -3.4000e-01],
[-5.4096e+00, -1.4167e+01, -1.8296e+01, ..., -1.0853e-01,
-9.9446e-02, -1.2460e+00],
[-1.1631e+01, -1.4374e+01, -2.4275e+01, ..., 1.8831e+00,
2.1158e+00, -2.8235e-01],
...,
[ 4.8814e-01, -4.9237e+00, -2.2791e+00, ..., -1.2811e+00,
-1.4319e+00, 1.1962e+00],
[-1.5555e-02, -1.3634e+00, 2.1296e-01, ..., -1.9033e+00,
-1.8166e+00, -2.3373e+00],
[ 6.3096e-01, 6.1927e-02, 1.9643e-01, ..., 1.1007e+00,
5.7693e-01, -1.0336e+00]],
[[-9.8834e+00, -1.5730e+01, -1.6712e+01, ..., 4.2577e-01,
3.3520e-01, 7.7533e-01],
[-6.8339e+00, -5.4070e+00, -8.0402e+00, ..., -1.6172e+00,
-1.0921e+00, 3.3022e-01],
[-1.0432e+01, -9.6851e+00, -1.0876e+01, ..., -3.9509e+00,
-3.9029e+00, -1.9297e+00],
...,
[ 4.8466e-01, 3.3207e+00, 6.8923e+00, ..., 5.4842e-01,
-7.5269e-01, -3.0015e+00],
[-3.1599e+00, -1.6667e+00, -5.7635e-01, ..., 1.6121e+00,
2.5148e+00, 1.8101e+00],
[-1.3606e+00, -2.1286e-01, 3.4622e-01, ..., 1.8205e-01,
9.2223e-01, 1.6100e+00]]],
[[[-1.5188e-01, -4.9760e-01, -1.9643e-01, ..., -2.5547e-01,
2.9655e-01, -5.5579e-02],
[ 2.4140e-01, -1.2381e+00, -7.0888e-01, ..., -5.7404e-01,
-2.3220e-01, -1.8321e-01],
[ 1.1597e-01, -9.4022e-01, -3.6816e-01, ..., -3.0696e-01,
-1.8710e-01, -5.2726e-01],
...,
[-2.7632e-01, -1.6898e+00, -1.1435e+00, ..., -1.0138e+00,
-9.8059e-01, -2.0448e+00],
[-1.1297e+00, -1.7115e+00, -7.7484e-01, ..., -5.3895e-01,
-4.6426e-01, -1.7702e+00],
[-1.3041e+00, -2.1512e+00, -1.6210e+00, ..., -1.4300e+00,
-1.5768e+00, -9.7155e-01]],
[[-1.4295e+00, -3.0401e+00, -2.5009e+00, ..., -2.7444e+00,
-3.3241e+00, -1.3177e+00],
[-3.1236e+00, -4.9525e+00, -3.5232e+00, ..., -3.7890e+00,
-4.6251e+00, -2.0023e+00],
[-2.0784e+00, -3.0891e+00, -2.2205e+00, ..., -2.4652e+00,
-2.6708e+00, -1.2412e+00],
...,
[-3.6970e+00, -5.4547e+00, -5.1591e+00, ..., -5.5061e+00,
-6.1854e+00, -2.8960e+00],
[-6.1771e+00, -7.0583e+00, -6.1391e+00, ..., -6.0523e+00,
-7.4017e+00, -2.6951e+00],
[-4.4201e-02, -8.9122e-01, 2.8775e-01, ..., 3.8944e-01,
6.9290e-01, 1.7849e-01]],
[[-4.5438e-01, 1.8477e-01, 8.1799e-01, ..., 7.8690e-01,
5.7410e-01, 1.3404e+00],
[-3.3875e-01, -7.4326e-02, 5.5124e-01, ..., 4.8319e-01,
3.7370e-01, 2.8455e-01],
[-8.0851e-01, 4.4041e-02, 3.0262e-01, ..., 2.6509e-01,
3.6509e-01, 4.3428e-01],
...,
[-1.3817e+00, 1.6128e-02, 1.7443e-01, ..., 1.5252e-01,
3.7615e-02, 2.8863e-01],
[-1.2041e+00, 1.2507e-01, -2.7091e-02, ..., -6.8465e-02,
-5.5038e-03, 2.6531e-01],
[-1.1571e+00, -5.5927e-02, 6.0621e-02, ..., -8.3649e-04,
-3.5119e-02, -3.8937e-01]],
...,
[[ 8.4510e-01, 2.3179e-01, 2.5600e-01, ..., 2.3297e-01,
1.9833e-01, -8.5923e-01],
[-1.1824e+00, -8.8104e-01, -6.7225e-01, ..., -6.9759e-01,
-1.4650e+00, -1.0856e+00],
[-6.2266e-01, -2.7587e-01, -2.6652e-01, ..., -2.2928e-01,
-4.8008e-01, -5.9681e-01],
...,
[-5.5228e-01, -6.0988e-01, -6.0193e-01, ..., -3.3720e-01,
-5.7562e-01, -1.1853e+00],
[-3.1139e+00, -2.8728e+00, -2.0414e+00, ..., -1.9149e+00,
-2.0229e+00, -2.0695e+00],
[-5.2531e-01, -1.6260e+00, -1.0505e+00, ..., -8.7798e-01,
-9.3101e-01, -1.3402e+00]],
[[ 3.3223e-02, 2.7583e-01, 7.1531e-01, ..., 4.6567e-01,
9.6768e-01, 3.9888e-01],
[-1.5807e-02, -4.3308e-01, 1.4192e-01, ..., 4.9043e-01,
5.9762e-01, -1.7700e-01],
[-3.0655e-01, -6.3185e-01, 1.4395e-01, ..., 2.3784e-01,
3.2582e-01, 1.9416e-02],
...,
[-1.1101e-01, -9.7068e-01, 1.7475e-01, ..., 1.4194e-01,
1.3932e-01, -6.9352e-01],
[-5.3220e-01, -1.8179e+00, -1.7408e-01, ..., 1.1515e-02,
-2.8936e-02, -9.1330e-01],
[ 4.4223e-01, -1.0226e-01, 1.4443e-01, ..., 2.6532e-01,
2.8511e-01, 3.1531e-01]],
[[-9.7295e-01, -2.6362e-01, 1.5155e-01, ..., 9.8704e-02,
2.7156e-02, 7.2288e-01],
[-1.1116e+00, -2.6359e-01, -3.1675e-01, ..., -3.2693e-01,
-8.1859e-02, 1.1544e+00],
[-6.8698e-01, -2.0790e-01, -2.3556e-01, ..., -2.0519e-01,
-1.1410e-01, 6.9084e-01],
...,
[-1.1524e+00, -9.3192e-01, -4.6094e-01, ..., -3.5690e-01,
-9.2680e-02, 8.3661e-01],
[-1.8448e+00, -1.8314e+00, -9.8085e-01, ..., -7.9090e-01,
-4.7538e-01, 4.3664e-01],
[-9.6765e-01, -9.7858e-01, -2.7630e-01, ..., -2.6734e-01,
-1.5626e-01, 1.1201e+00]]],
[[[-6.9667e-01, -1.4386e+00, -1.3216e+00, ..., -2.3828e-01,
2.1617e-01, -4.1724e-01],
[-2.6325e-01, -2.4531e+00, -1.9039e+00, ..., -5.5433e-01,
-4.5595e-01, -8.3332e-01],
[-4.1804e-01, -2.1577e+00, -1.5304e+00, ..., -1.5730e+00,
-2.8157e+00, -1.4746e+00],
...,
[-4.6724e-01, -1.8876e+00, -1.8504e+00, ..., -2.9657e-01,
-1.6138e-01, -1.0635e+00],
[-9.6330e-01, -1.6647e+00, -1.1605e+00, ..., 4.2911e-02,
1.0059e-01, -9.9476e-01],
[-8.1755e-01, -1.6393e+00, -1.2533e+00, ..., -1.0810e+00,
-1.3374e+00, -7.7034e-01]],
[[-2.4849e+00, -5.0307e+00, -5.7795e+00, ..., -2.2987e+00,
-3.6247e+00, -1.3545e+00],
[-5.3664e+00, -8.1017e+00, -8.1739e+00, ..., -3.8050e+00,
-6.0610e+00, -2.6077e+00],
[-4.3436e+00, -6.3085e+00, -7.6200e+00, ..., -8.4068e+00,
-1.2186e+01, -7.2480e+00],
...,
[-4.3609e+00, -6.0498e+00, -5.5039e+00, ..., -3.1236e+00,
-3.8945e+00, -1.5546e+00],
[-6.3663e+00, -7.4410e+00, -6.3576e+00, ..., -4.2716e+00,
-5.8614e+00, -2.1343e+00],
[-9.3375e-03, -8.1892e-01, 4.0084e-01, ..., 4.7816e-01,
6.1453e-01, 1.1816e-01]],
[[-5.8070e-01, 6.4432e-01, 1.1179e+00, ..., 7.8885e-01,
5.4962e-01, 1.2934e+00],
[-9.4687e-01, -1.2783e-01, -6.8408e-02, ..., 3.3657e-01,
2.9248e-01, 3.2398e-01],
[-1.6600e+00, -2.0794e-01, 3.0149e-01, ..., -2.0738e+00,
-2.7629e+00, 2.3668e-01],
...,
[-1.7479e+00, -6.7995e-02, -1.2600e-01, ..., 1.3332e-01,
2.0316e-01, -1.8726e-02],
[-1.5469e+00, 7.8807e-02, 6.1527e-02, ..., 6.0140e-02,
1.4502e-01, 1.4576e-01],
[-1.4021e+00, -5.1689e-01, -2.8606e-01, ..., 4.2157e-03,
1.3539e-01, -1.4503e-01]],
...,
[[ 1.1103e+00, -3.5357e-01, -9.1663e-02, ..., 3.4983e-01,
2.3204e-01, -7.3953e-01],
[-7.5660e-01, -8.6927e-01, -3.0131e-01, ..., -2.4474e-01,
-9.2818e-01, -9.0672e-01],
[-6.2611e-01, -9.2359e-01, -4.0704e-01, ..., 1.1442e+00,
2.2325e-01, -2.0404e+00],
...,
[-7.3690e-01, -1.2220e+00, -7.4341e-01, ..., -2.2908e-01,
-5.1433e-01, -7.9919e-01],
[-2.7604e+00, -2.8230e+00, -1.8365e+00, ..., -1.2193e+00,
-1.2929e+00, -1.2240e+00],
[-6.7215e-01, -1.6849e+00, -1.0071e+00, ..., -1.5346e-01,
-3.1927e-01, -5.9200e-01]],
[[ 5.3310e-02, 2.2245e-02, 8.9713e-01, ..., 3.1855e-01,
7.4908e-01, -3.8721e-02],
[ 4.6832e-02, -6.3143e-01, 4.7687e-01, ..., 5.1830e-01,
6.9786e-01, -5.8057e-01],
[-9.8916e-02, -7.5795e-01, 7.8877e-01, ..., 3.5729e-01,
-8.7062e-01, -1.8739e+00],
...,
[ 5.8796e-02, -7.9211e-01, 2.6232e-01, ..., 4.0117e-01,
3.9809e-01, -1.1062e-01],
[-2.5564e-01, -1.4930e+00, -1.6086e-02, ..., -5.4233e-02,
-3.3339e-02, -5.6470e-01],
[ 6.4422e-01, 1.2452e-01, 2.4405e-01, ..., 1.6880e-01,
2.1345e-01, 3.3484e-01]],
[[-1.5822e+00, -5.2161e-01, 1.4025e-01, ..., 2.9756e-01,
2.3553e-01, 7.9306e-01],
[-2.1572e+00, -3.3895e-01, -2.7100e-01, ..., -4.1999e-01,
-1.2575e-01, 1.0601e+00],
[-1.8048e+00, -4.2614e-01, -2.7931e-01, ..., -1.0043e+00,
-3.9582e+00, -1.4024e+00],
...,
[-1.7359e+00, -4.6353e-01, 2.2114e-01, ..., -1.6963e-01,
-3.5837e-02, 1.1421e+00],
[-2.1682e+00, -1.1223e+00, -1.6213e-01, ..., -3.3213e-01,
-1.1845e-01, 6.1143e-01],
[-1.0648e+00, -2.4329e-01, 4.5643e-01, ..., -3.4823e-01,
-3.3806e-01, 6.7296e-01]]],
[[[-3.3478e-01, -1.0723e+00, -8.8700e-01, ..., -4.9149e-02,
5.9341e-01, 5.0721e-01],
[ 1.4440e-01, -1.9620e+00, -1.3318e+00, ..., -5.0015e-01,
-2.4141e-01, 1.1929e-01],
[ 2.0139e-01, -1.5428e+00, -1.0586e+00, ..., -4.3031e-01,
-1.5728e-01, -1.2163e-01],
...,
[-2.1415e+00, -2.0454e+00, -3.7517e+00, ..., -3.6797e-01,
-1.1259e-01, -1.5771e-01],
[-1.0577e+00, -2.6657e+00, -3.6130e+00, ..., -3.8264e-03,
2.4199e-01, -4.0249e-02],
[ 1.7268e-02, -1.7591e+00, -3.0710e+00, ..., -1.0515e+00,
-1.0922e+00, -3.7052e-01]],
[[-2.1444e+00, -4.1309e+00, -3.6634e+00, ..., -5.6334e-01,
-1.0950e+00, -1.4454e-02],
[-4.0658e+00, -6.5245e+00, -5.7956e+00, ..., -1.9618e+00,
-2.8226e+00, -7.3292e-01],
[-3.6327e+00, -5.2495e+00, -5.2513e+00, ..., -1.6889e+00,
-1.6339e+00, -4.2123e-01],
...,
[-1.0316e+01, -1.1198e+01, -9.3082e+00, ..., -1.3878e+00,
-1.6762e+00, -4.0347e-01],
[-6.1340e+00, -6.7707e+00, -6.2733e+00, ..., -2.5703e+00,
-3.4461e+00, -1.0274e+00],
[ 8.0502e-01, 4.1143e-01, 4.7132e-01, ..., 3.7802e-01,
3.7423e-01, 6.8479e-02]],
[[-5.7519e-01, 4.2651e-01, 8.0932e-01, ..., 8.7146e-01,
3.6105e-01, 1.1353e+00],
[-9.7193e-01, 1.3064e-01, 6.1302e-01, ..., 4.4708e-01,
8.6496e-02, 4.9016e-01],
[-1.3354e+00, 1.9429e-02, 7.8586e-04, ..., 1.3214e-01,
9.9534e-02, 4.3739e-01],
...,
[-1.0878e+01, -9.4695e+00, -2.6275e+00, ..., 7.1366e-02,
9.2560e-02, 4.3694e-01],
[-5.4442e+00, -5.6501e+00, 1.2494e+00, ..., 1.4338e-01,
1.4391e-01, 6.1768e-01],
[-5.4295e-01, -1.9850e+00, 1.2310e+00, ..., -9.1057e-02,
8.0670e-02, 3.6026e-01]],
...,
[[ 6.4730e-01, 7.2533e-02, 2.8929e-01, ..., -1.1069e-01,
-3.7401e-01, -9.4060e-01],
[-8.1143e-01, -3.2140e-01, -8.6969e-02, ..., -7.0570e-01,
-1.3170e+00, -8.5333e-01],
[-5.1109e-01, -3.7514e-01, 2.6439e-02, ..., -1.7280e-01,
-1.9779e-01, -4.0673e-01],
...,
[-2.2437e+00, -1.0488e+01, -9.0101e+00, ..., -1.1621e-01,
-2.0016e-01, -3.6035e-01],
[-4.4653e+00, -9.0599e+00, -7.5678e+00, ..., -8.4349e-01,
-8.8556e-01, -7.8441e-01],
[-2.1592e+00, -3.1855e+00, -3.2301e+00, ..., 1.2469e-02,
-1.6350e-01, -3.5367e-01]],
[[-5.9591e-04, -2.9093e-01, -8.9674e-02, ..., 3.5822e-01,
6.8663e-01, 3.0135e-01],
[ 1.0072e-01, -9.7834e-01, -8.2547e-02, ..., -1.3112e-02,
1.2075e-01, -4.5832e-02],
[-1.5600e-01, -1.0420e+00, -9.7675e-04, ..., -3.4644e-01,
-1.1506e-01, 1.9179e-01],
...,
[-3.3591e+00, -2.7333e+00, 2.6530e-01, ..., -1.2990e-01,
-1.3533e-01, 1.6912e-01],
[-1.5302e+00, -4.7880e+00, -2.1861e+00, ..., -3.7454e-01,
-3.5565e-01, -5.7901e-02],
[ 1.4285e+00, -7.7361e-01, -1.3520e+00, ..., -2.2938e-01,
-2.4011e-01, 3.1369e-01]],
[[-6.4451e-01, -1.5360e-01, -1.7068e-02, ..., 5.8722e-01,
3.7640e-01, 6.0073e-01],
[-1.0670e+00, -8.6808e-01, -9.4078e-01, ..., -4.3716e-02,
-1.4539e-01, 4.5359e-01],
[-6.5519e-01, -7.8583e-01, -5.3491e-01, ..., 5.7679e-02,
-1.8018e-01, 1.0429e-01],
...,
[-5.6799e+00, -4.7194e+00, -9.0847e+00, ..., 9.9923e-02,
-9.0453e-02, 1.4666e-01],
[-2.8967e+00, 1.7970e+00, -3.5939e+00, ..., -7.0516e-02,
-2.9255e-01, -1.7054e-01],
[-1.5485e+00, 1.4868e+00, 4.2105e-01, ..., -9.2082e-02,
-2.8587e-01, 1.5218e-01]]],
[[[-2.0539e-01, -3.2756e-01, 1.8219e-01, ..., 5.1252e-01,
5.0467e-01, 1.7170e-01],
[-7.9673e-02, -1.4197e+00, -1.2875e+00, ..., -5.9206e-01,
-5.7775e-01, -3.6024e-01],
[-1.7173e-02, -1.1025e+00, -1.0959e+00, ..., -4.4347e-01,
-5.3133e-01, -7.5249e-01],
...,
[-3.0477e-01, -1.5659e+00, -2.0208e+00, ..., -1.0175e+00,
-1.0211e+00, -2.0908e+00],
[-1.1675e+00, -1.5560e+00, -2.6552e-01, ..., -6.2418e-01,
-5.2560e-01, -1.8296e+00],
[-1.3006e+00, -2.0661e+00, -6.6961e-01, ..., -1.4385e+00,
-1.5764e+00, -9.8494e-01]],
[[-2.7158e-01, -1.5264e+00, -2.6274e+00, ..., -1.1563e+00,
-1.5843e+00, -3.0937e-01],
[-2.0262e+00, -4.1977e+00, -5.7674e+00, ..., -3.9078e+00,
-3.8069e+00, -1.3470e+00],
[-1.8951e+00, -4.0123e+00, -6.8175e+00, ..., -4.3312e+00,
-3.2415e+00, -1.3833e+00],
...,
[-3.8163e+00, -6.5134e+00, -7.2156e+00, ..., -5.4945e+00,
-6.3274e+00, -2.9345e+00],
[-6.2815e+00, -7.2419e+00, -7.6343e+00, ..., -6.2257e+00,
-7.4980e+00, -2.7241e+00],
[-5.8441e-02, -1.1280e+00, 1.0899e+00, ..., 3.3780e-01,
6.6405e-01, 2.1418e-01]],
[[-4.4218e-01, -4.3004e-02, 1.7434e-01, ..., 2.2149e-01,
1.3474e-01, 1.0179e+00],
[-3.2288e-01, -3.7181e-01, 7.1715e-02, ..., 2.2780e-01,
3.1106e-01, 4.7792e-01],
[-7.7209e-01, -1.8007e-01, -7.3194e-01, ..., -6.5226e-02,
1.7527e-01, 3.5027e-01],
...,
[-1.4163e+00, 2.9686e-01, 5.2340e-01, ..., 1.5367e-02,
-1.8946e-03, 2.8426e-01],
[-1.2023e+00, 1.7762e-01, -3.9683e-01, ..., -6.1791e-02,
1.3573e-02, 2.9517e-01],
[-1.1994e+00, 3.2376e-01, 1.1369e+00, ..., -1.0151e-02,
-4.4001e-02, -3.5726e-01]],
...,
[[ 2.3943e-01, -4.3428e-01, 1.2870e-01, ..., -5.2308e-01,
-3.0289e-01, -7.9137e-01],
[-6.8131e-01, -6.2517e-01, -1.1071e-01, ..., -1.5231e+00,
-1.1770e+00, -8.4241e-01],
[-3.6560e-01, 4.1355e-02, 1.8625e+00, ..., -1.3489e+00,
-1.8768e-01, -5.6289e-01],
...,
[-6.2043e-01, -4.8794e-01, -1.7592e+00, ..., -3.7749e-01,
-5.7507e-01, -1.1937e+00],
[-3.1116e+00, -2.6785e+00, 3.5340e-01, ..., -1.9388e+00,
-2.0751e+00, -2.1015e+00],
[-5.2519e-01, -1.6889e+00, -5.0923e-01, ..., -8.4299e-01,
-9.5386e-01, -1.3703e+00]],
[[-1.2138e-01, 8.3720e-02, 7.1031e-01, ..., 5.1056e-01,
6.5881e-01, 1.7542e-01],
[-2.7211e-01, -2.0456e-01, 1.4264e-01, ..., -6.7078e-01,
1.0755e-01, -1.3779e-01],
[-5.1326e-01, -2.3798e-01, -1.2053e-01, ..., -3.0637e-01,
-7.0170e-03, -5.0967e-02],
...,
[-1.4162e-01, -8.4138e-01, -6.2785e-01, ..., 2.8267e-01,
1.0307e-01, -7.3049e-01],
[-5.4910e-01, -1.8846e+00, -8.2871e-01, ..., 7.1713e-03,
-4.1010e-02, -9.2618e-01],
[ 4.5381e-01, -7.2137e-02, -5.3298e-01, ..., 2.8323e-01,
3.1884e-01, 3.5274e-01]],
[[-5.6789e-01, 4.3917e-01, -1.9398e-01, ..., 2.6791e-01,
4.0999e-01, 6.9284e-01],
[-8.0562e-01, 2.4365e-01, -7.9128e-01, ..., -8.1602e-02,
-1.3764e-01, 7.4880e-01],
[-4.6824e-01, 2.0338e-01, -8.2454e-01, ..., 2.9787e-01,
1.3420e-01, 6.3737e-01],
...,
[-1.1987e+00, -1.6727e+00, -3.3148e+00, ..., -4.3924e-01,
-9.8177e-02, 8.1582e-01],
[-1.8820e+00, -2.1687e+00, -3.0709e+00, ..., -8.3819e-01,
-5.2611e-01, 4.2978e-01],
[-9.6283e-01, -1.0247e+00, -1.3600e+00, ..., -2.6373e-01,
-1.5676e-01, 1.1760e+00]]]], device='cuda:0'),
tensor([[[[ 1.2770e-02, -2.8623e-01, -8.4696e-02, ..., -2.5111e+00,
1.1327e+00, 5.1174e-02],
[ 3.0346e-01, -9.4112e-01, -7.3523e-01, ..., -4.0870e+00,
-1.4842e+00, 5.5628e-01],
[ 2.6257e-01, -5.6654e-01, -5.1750e-01, ..., -7.7284e+00,
-8.7964e+00, -2.0717e+00],
...,
[ 3.9273e-02, -5.7642e-01, -1.1141e-01, ..., -3.5872e+00,
-3.2644e+00, 4.4296e-01],
[-5.4419e-01, -6.6654e-01, 3.4070e-01, ..., 2.4582e-01,
-7.7508e-01, -1.6766e+00],
[-6.8625e-01, -1.2993e+00, -8.0774e-01, ..., -1.1473e+00,
-1.5843e+00, -9.2469e-01]],
[[-8.3711e-01, -2.2769e+00, -2.0416e+00, ..., -9.5687e+00,
-1.4118e+01, -5.5459e+00],
[-2.1147e+00, -3.8484e+00, -3.2167e+00, ..., -1.6135e+01,
-1.6202e+01, -3.9366e+00],
[-1.7150e+00, -2.6906e+00, -2.7733e+00, ..., -1.7156e+01,
-1.4344e+01, -5.5030e-01],
...,
[-1.8731e+00, -2.5761e+00, -4.8724e+00, ..., -8.6300e+00,
-7.5643e+00, -2.3934e+00],
[-3.8809e+00, -4.4877e+00, -4.9429e+00, ..., -8.4543e+00,
-7.4776e+00, -3.2043e+00],
[ 3.1461e-01, -2.6865e-01, 2.3825e-01, ..., 8.2747e-02,
5.2115e-01, 7.9880e-02]],
[[-1.9414e-01, 1.9196e-01, 7.4023e-01, ..., -3.2552e-01,
-7.3962e-02, 2.9920e+00],
[ 8.3278e-02, -3.1871e-01, 5.6388e-01, ..., -1.1758e+01,
-1.0250e+01, -4.2395e+00],
[-1.6935e-01, -1.7394e-01, 1.9376e-01, ..., -2.0150e+01,
-1.6463e+01, -8.6517e+00],
...,
[-2.9827e-01, -1.7042e-01, 1.3088e+00, ..., -7.7266e+00,
2.9456e+00, -2.7979e+00],
[-2.5854e-01, -9.6955e-02, 9.0188e-01, ..., 1.6301e-01,
2.8704e+00, -1.2665e+00],
[-9.7434e-02, -3.9292e-01, 1.4926e+00, ..., -5.9487e-01,
2.4663e-01, -4.2579e-01]],
...,
[[ 6.5516e-01, -1.6566e-01, 8.3059e-02, ..., -1.7573e+00,
-2.2773e+00, -4.3544e+00],
[-4.8815e-01, -5.1598e-01, -5.1763e-01, ..., -8.4919e+00,
-1.1331e+01, -1.0780e+01],
[-3.3775e-01, -2.1069e-01, -1.9727e-01, ..., -9.2295e+00,
-1.2893e+01, -1.2435e+01],
...,
[-4.2431e-01, -1.4099e-01, -4.1459e-01, ..., -5.0325e-01,
-4.9074e+00, -6.8760e+00],
[-2.3060e+00, -1.6647e+00, -1.8889e-01, ..., -6.3181e+00,
-7.5040e+00, -3.9406e+00],
[-4.8028e-01, -7.9125e-01, -6.1278e-01, ..., -2.0264e+00,
-1.3242e+00, -9.3018e-01]],
[[ 1.7931e-01, 2.7967e-01, 6.1263e-01, ..., -2.3602e-01,
2.8400e+00, -4.4874e-01],
[ 1.9249e-01, 2.3135e-01, 4.4777e-01, ..., -6.2553e+00,
-5.3841e+00, -4.1297e+00],
[-1.0813e-01, 4.6417e-02, 3.0387e-01, ..., -1.0404e+01,
-1.1521e+01, -3.2957e+00],
...,
[-1.5031e-01, -2.3518e-02, 1.0294e+00, ..., -4.9135e+00,
-1.1435e+00, 2.3972e+00],
[-5.2350e-01, -7.8104e-01, -2.8644e-01, ..., -5.9306e-02,
-6.5491e-01, -1.4679e+00],
[ 5.3154e-01, 3.7402e-01, 4.6346e-02, ..., 7.1457e-01,
4.4148e-02, 3.1600e-01]],
[[-8.0028e-01, 2.4423e-01, 2.9157e-01, ..., -3.1588e+00,
-3.5959e+00, -2.7476e+00],
[-1.1005e+00, 1.4211e-01, -3.8849e-01, ..., -7.8984e+00,
-8.2871e+00, -4.9302e+00],
[-9.2227e-01, 2.8389e-01, -1.1726e-01, ..., -9.5393e+00,
-1.0413e+01, -7.6018e+00],
...,
[-9.2561e-01, 1.5532e-01, -1.4086e+00, ..., -1.9223e+00,
-4.4230e+00, 3.0102e-01],
[-1.6172e+00, -4.0160e-01, -1.0262e+00, ..., 5.3003e-03,
7.6744e-01, 2.7046e+00],
[-7.2206e-01, -1.0444e-01, -6.6608e-01, ..., 4.6032e-01,
2.8423e-01, 1.1956e+00]]],
[[[-2.5039e-01, -6.1831e-01, -2.3192e-01, ..., 5.6281e-02,
7.8740e-02, 1.0648e-01],
[-1.0471e-01, -1.4741e+00, -9.7069e-01, ..., -2.1628e+00,
-1.2697e+00, -4.6639e-01],
[ 4.1272e-02, -1.0407e+00, -8.7117e-01, ..., -1.0738e+00,
-7.2379e-01, -9.2665e-01],
...,
[-2.5968e-01, -1.3407e+00, -6.0172e-01, ..., -7.1980e+00,
-6.7295e+00, -3.3745e+00],
[-1.3291e+00, -1.8112e+00, -5.2336e-01, ..., -6.1118e+00,
-3.2044e+00, -2.4077e+00],
[-1.3817e+00, -2.3445e+00, -1.7580e+00, ..., -3.6190e+00,
-2.4119e+00, -1.4357e+00]],
[[-5.7681e-01, -2.2037e+00, -2.1253e+00, ..., -1.1036e+00,
-2.1863e+00, -5.6076e-01],
[-2.2334e+00, -3.9987e+00, -2.9423e+00, ..., -4.6916e+00,
-4.5684e+00, -1.5727e+00],
[-2.2529e+00, -4.4202e+00, -4.2846e+00, ..., -3.8196e+00,
-3.5977e+00, -1.5583e+00],
...,
[-4.1035e+00, -5.8752e+00, -5.1895e+00, ..., -1.0950e+01,
-6.9209e+00, -3.8703e+00],
[-6.4080e+00, -7.4386e+00, -6.6819e+00, ..., -7.4560e+00,
-7.6222e+00, -3.6273e+00],
[-1.2529e-01, -9.8541e-01, 1.9527e-01, ..., -1.6389e+00,
-1.0096e+00, -5.9810e-01]],
[[-5.3274e-01, -1.3289e-01, -1.1801e-01, ..., -3.8692e-01,
3.8097e-01, 1.2149e+00],
[-3.4591e-01, -3.5193e-01, 1.8082e-01, ..., -3.8750e-01,
-1.1278e-01, 5.7393e-01],
[-7.8054e-01, -1.1037e-01, 3.0153e-01, ..., 4.4003e-02,
-8.8948e-02, 3.4224e-01],
...,
[-1.1335e+00, -1.4681e-01, -9.0949e-02, ..., -1.3991e+01,
-4.5750e+00, 2.9902e-01],
[-1.2283e+00, 3.0168e-01, 3.7322e-01, ..., -6.5047e+00,
-3.5924e+00, -6.2024e-01],
[-1.2388e+00, -1.1586e-01, 6.6497e-02, ..., 5.5436e-01,
-5.3990e-01, -8.8258e-01]],
...,
[[ 2.0079e-01, -6.2667e-01, -8.4546e-01, ..., -2.1350e+00,
-9.1572e-01, -8.2002e-01],
[-5.0409e-01, -9.8141e-02, 3.0351e-01, ..., -2.2374e+00,
-1.6025e+00, -9.4645e-01],
[-4.4074e-01, -2.8354e-01, -1.1411e-01, ..., -1.1465e+00,
-4.1982e-01, -6.1224e-01],
...,
[-8.1889e-01, -7.2886e-01, 2.5247e-01, ..., -9.5719e+00,
-6.1615e+00, -3.6495e+00],
[-3.3304e+00, -3.5134e+00, -3.1878e+00, ..., -4.8393e+00,
-5.8583e+00, -4.9418e+00],
[-5.5468e-01, -1.7094e+00, -1.0967e+00, ..., -2.6042e+00,
-2.7738e+00, -2.6603e+00]],
[[-1.7297e-01, 6.0570e-04, -4.8241e-03, ..., 2.2871e-01,
2.4919e-01, 2.0941e-01],
[-3.2755e-01, -3.7519e-01, -1.7724e-01, ..., -2.4591e-02,
-1.0279e-03, -2.2297e-01],
[-4.6201e-01, -2.9812e-01, 6.2106e-02, ..., 6.1291e-01,
1.9074e-01, -1.4801e-01],
...,
[-4.4175e-02, -1.1490e+00, -1.6281e-01, ..., -9.4517e+00,
-6.2019e+00, -3.1463e+00],
[-4.8051e-01, -1.7172e+00, 2.2835e-01, ..., -3.3607e+00,
-8.1852e-01, -5.6496e-01],
[ 4.4705e-01, -1.3146e-01, 1.7482e-01, ..., 3.8058e-01,
2.3992e+00, 1.3538e+00]],
[[-4.8552e-01, 4.8074e-01, 4.3390e-01, ..., -2.6483e-01,
8.3219e-01, 7.8934e-01],
[-8.2389e-01, 1.9246e-01, -2.2145e-01, ..., -4.9255e-01,
4.5711e-01, 8.4750e-01],
[-4.7199e-01, 2.6732e-01, 1.0807e-01, ..., 1.6852e-01,
4.6982e-01, 7.7233e-01],
...,
[-1.3193e+00, -8.2559e-01, -4.8059e-01, ..., -7.5603e+00,
-2.1684e+00, 2.8590e+00],
[-2.1670e+00, -2.0818e+00, -9.9441e-01, ..., -8.5114e+00,
-2.5306e+00, 2.8189e+00],
[-1.0494e+00, -1.0104e+00, -2.9451e-01, ..., -9.5580e-01,
-5.2058e-02, 2.4094e+00]]],
[[[-3.3118e-01, -4.8632e-01, -1.4432e-01, ..., -9.6117e-01,
-5.2051e-01, -1.4291e+00],
[-1.5186e-02, -1.3948e+00, -7.1165e-01, ..., -1.1622e+00,
-1.3012e+00, -1.4703e+00],
[-1.1411e-01, -1.2571e+00, -4.8060e-01, ..., -9.0862e-01,
-1.0881e+00, -1.8294e+00],
...,
[-3.0422e-01, -1.3215e+00, -5.0064e-01, ..., -1.2548e+00,
-1.1126e+00, -1.9250e+00],
[-4.6970e-01, -1.1190e+00, -8.9809e-02, ..., -4.5242e-01,
-5.6502e-01, -1.7036e+00],
[-5.2563e-01, -1.6207e+00, -1.2561e+00, ..., -1.1200e+00,
-1.5312e+00, -7.5430e-01]],
[[-6.8099e-01, -1.9289e+00, -7.7701e-01, ..., -3.9439e+00,
-5.4536e+00, -1.8468e+00],
[-2.1469e+00, -3.9134e+00, -2.0415e+00, ..., -5.9204e+00,
-8.0878e+00, -3.4226e+00],
[-1.6095e+00, -2.8120e+00, -1.4990e+00, ..., -5.0958e+00,
-6.1828e+00, -2.5681e+00],
...,
[-1.6762e+00, -2.7709e+00, -1.4797e+00, ..., -5.2911e+00,
-6.3549e+00, -2.4974e+00],
[-3.7167e+00, -4.5887e+00, -2.8736e+00, ..., -6.4464e+00,
-7.9423e+00, -2.5519e+00],
[ 5.8258e-01, -6.7973e-01, 4.5326e-01, ..., 4.2918e-01,
4.5776e-01, 2.1247e-01]],
[[-3.6453e-01, 7.2988e-01, 1.0217e+00, ..., 6.4842e-01,
2.5531e-01, 1.5270e+00],
[-7.6175e-01, 2.4917e-01, 5.4898e-01, ..., 5.9578e-01,
-6.4730e-02, 1.4769e-01],
[-1.2810e+00, 3.0440e-01, 1.7534e-01, ..., 4.3837e-02,
4.0368e-03, -1.2184e-01],
...,
[-1.4219e+00, 3.1406e-01, 1.2708e-01, ..., -1.7364e-01,
-1.1573e-01, -1.9719e-01],
[-1.4073e+00, 3.2943e-01, 1.9582e-01, ..., -1.1690e-01,
-9.5201e-03, -2.1700e-01],
[-9.1660e-01, 6.8536e-02, -1.3331e-01, ..., 5.4242e-02,
2.2279e-01, -5.6593e-01]],
...,
[[ 4.2796e-01, -1.0379e-01, -1.0817e-01, ..., 4.2340e-01,
3.1583e-01, -1.3185e+00],
[-1.3536e+00, -1.0514e+00, -6.2698e-01, ..., -1.5308e-01,
-7.8061e-01, -1.3904e+00],
[-7.3965e-01, -4.4662e-01, -9.7583e-02, ..., -1.3545e-01,
-1.0823e-01, -9.5899e-01],
...,
[-9.6379e-01, -5.7615e-01, -1.0641e-01, ..., -2.3524e-01,
-6.2441e-02, -9.3382e-01],
[-2.5469e+00, -1.7654e+00, -8.8600e-01, ..., -1.5209e+00,
-1.5581e+00, -1.7941e+00],
[-6.5255e-01, -1.1042e+00, -5.8349e-02, ..., -3.4396e-01,
-5.5879e-01, -1.2075e+00]],
[[ 2.8762e-02, 2.4484e-01, 4.8901e-01, ..., -8.9399e-02,
3.8306e-01, -3.7606e-01],
[-1.5101e-01, -6.7341e-01, -4.0035e-02, ..., 1.8927e-01,
1.0968e-01, -8.2269e-01],
[-5.5589e-01, -9.9039e-01, -9.9717e-02, ..., 2.3717e-01,
1.3808e-01, -4.7531e-01],
...,
[-6.0654e-01, -1.1035e+00, -1.3152e-01, ..., 1.5990e-01,
9.6203e-02, -5.1047e-01],
[-1.0596e+00, -1.9269e+00, -4.3808e-01, ..., -1.2273e-01,
-2.8394e-01, -9.3705e-01],
[ 2.3351e-01, -4.9436e-01, -2.6449e-01, ..., 3.4609e-01,
3.1015e-01, 2.7908e-01]],
[[-3.6320e-01, 1.2133e-01, 5.3799e-01, ..., -7.2045e-02,
-8.0269e-02, 5.7702e-01],
[-3.7751e-01, -3.7769e-02, 3.2232e-02, ..., -8.8790e-01,
-4.2338e-01, 8.8082e-01],
[ 2.6365e-02, 2.0743e-02, 7.9621e-02, ..., -5.3881e-01,
-5.2638e-01, 6.2331e-01],
...,
[ 2.6176e-02, -8.5109e-02, 1.0190e-01, ..., 4.8844e-02,
-3.8777e-01, 6.1371e-01],
[-3.5467e-01, -4.8980e-01, -1.3521e-01, ..., -6.3981e-01,
-6.6286e-01, 3.9706e-01],
[ 5.5329e-02, -5.2855e-01, -1.7871e-01, ..., -4.8480e-01,
-4.4459e-01, 9.9750e-01]]],
[[[-3.0206e-01, -9.9750e-01, -8.5343e-01, ..., 1.6573e-02,
6.1356e-01, 5.4459e-01],
[ 1.7253e-01, -1.8340e+00, -1.3024e+00, ..., -4.6492e-01,
-2.1277e-01, 1.2340e-01],
[ 1.5969e-01, -1.2094e+00, -1.0846e+00, ..., -3.3145e-01,
-1.3063e-01, -1.0639e-01],
...,
[ 2.4922e-01, -1.4746e+00, -6.5049e-01, ..., -3.6819e-01,
-1.4184e-01, -1.2546e-01],
[ 1.1921e-01, -9.3307e-01, -5.2327e-01, ..., 4.5255e-02,
2.4251e-01, -3.5558e-02],
[-1.2572e-01, -1.0845e+00, -9.6434e-01, ..., -1.0139e+00,
-1.0766e+00, -3.6492e-01]],
[[-2.0739e+00, -4.0681e+00, -3.7980e+00, ..., -4.5135e-01,
-1.0089e+00, -5.7053e-03],
[-3.8984e+00, -6.3098e+00, -5.5775e+00, ..., -1.7962e+00,
-2.6524e+00, -6.9910e-01],
[-3.1827e+00, -6.0358e+00, -6.9547e+00, ..., -1.3139e+00,
-1.5189e+00, -3.8859e-01],
...,
[-3.1695e+00, -5.1876e+00, -4.7688e+00, ..., -1.4180e+00,
-1.6357e+00, -3.8131e-01],
[-5.0677e+00, -6.1147e+00, -5.7877e+00, ..., -2.3851e+00,
-3.3350e+00, -9.7718e-01],
[ 7.2109e-01, -1.1212e-01, 5.1796e-01, ..., 3.6782e-01,
3.5785e-01, 8.1074e-02]],
[[-5.7798e-01, 4.3589e-01, 8.2150e-01, ..., 8.1280e-01,
3.0922e-01, 1.1013e+00],
[-9.2677e-01, 8.1834e-02, 6.4572e-01, ..., 4.0581e-01,
2.8948e-02, 4.8914e-01],
[-1.2917e+00, -7.8463e-03, -1.4999e-02, ..., 2.5737e-02,
2.2721e-02, 4.0788e-01],
...,
[-1.6876e+00, 7.5821e-01, 2.5433e-01, ..., 3.9247e-02,
1.2148e-01, 4.3560e-01],
[-1.3074e+00, 1.5011e-01, -2.1411e-01, ..., 6.1755e-02,
1.4869e-01, 6.1078e-01],
[-5.3777e-01, 3.2284e-01, 1.7628e-01, ..., -1.4869e-01,
4.8868e-02, 3.7128e-01]],
...,
[[ 6.8475e-01, 7.2025e-02, 2.7034e-01, ..., -9.8598e-02,
-3.9101e-01, -9.3449e-01],
[-8.1843e-01, -3.6700e-01, -8.7080e-02, ..., -6.7234e-01,
-1.2811e+00, -8.4037e-01],
[-5.1888e-01, -3.0860e-01, 1.3726e-01, ..., -1.1800e-01,
-1.8914e-01, -3.9557e-01],
...,
[-8.6638e-01, -1.7508e+00, -1.0548e+00, ..., -1.1192e-01,
-2.0967e-01, -3.6923e-01],
[-2.2426e+00, -1.8485e+00, -1.3357e+00, ..., -7.9004e-01,
-8.5538e-01, -7.8621e-01],
[-4.0318e-01, -9.8062e-01, -3.2739e-01, ..., -3.6564e-02,
-1.4860e-01, -3.2843e-01]],
[[ 1.1485e-02, -2.3942e-01, -5.0510e-02, ..., 3.3389e-01,
6.7343e-01, 2.8823e-01],
[ 1.3990e-01, -8.9354e-01, -5.5110e-02, ..., -4.6683e-02,
1.4597e-01, -7.2704e-02],
[-1.7755e-01, -7.5338e-01, 2.1690e-01, ..., -1.9151e-01,
-1.0305e-01, 1.6631e-01],
...,
[-1.7016e-01, -9.8601e-01, 4.2698e-01, ..., -1.4347e-01,
-1.4317e-01, 1.4602e-01],
[-5.0169e-01, -1.9388e+00, -3.5002e-01, ..., -3.1529e-01,
-3.0769e-01, -7.3284e-02],
[ 7.2635e-01, -8.7486e-02, 2.2889e-01, ..., -2.2562e-01,
-2.5026e-01, 2.8840e-01]],
[[-6.3442e-01, -9.1249e-02, 3.6145e-02, ..., 5.5592e-01,
3.7579e-01, 5.4157e-01],
[-1.0667e+00, -7.4774e-01, -9.3046e-01, ..., -1.0967e-01,
-1.1864e-01, 4.0922e-01],
[-6.5282e-01, -6.9888e-01, -7.1971e-01, ..., 2.8830e-03,
-1.7034e-01, 7.3001e-02],
...,
[-4.0726e-01, -7.2201e-01, -1.0031e-01, ..., 1.0147e-01,
-8.2250e-02, 8.4645e-02],
[-7.8271e-01, -9.6372e-01, -5.7196e-01, ..., -1.4937e-02,
-2.0497e-01, -2.0684e-01],
[-4.9059e-01, -8.2622e-01, -3.8119e-01, ..., -6.8776e-02,
-2.4303e-01, 1.0016e-01]]],
[[[-7.4677e-01, -1.4375e+00, -1.3978e+00, ..., -4.8372e-02,
5.8888e-01, 5.2355e-01],
[-3.1244e-01, -2.4500e+00, -2.2066e+00, ..., -5.3180e-01,
-2.7074e-01, 1.1008e-01],
[-4.2882e-01, -2.1973e+00, -1.7107e+00, ..., -3.3071e-01,
-1.9880e-01, -1.3828e-01],
...,
[-5.1877e-01, -1.9909e+00, -1.4639e+00, ..., -3.4552e-01,
-1.3414e-01, -1.5853e-01],
[-1.0374e+00, -1.7900e+00, -1.2117e+00, ..., 4.9355e-03,
2.0310e-01, -7.9712e-02],
[-8.6253e-01, -1.7251e+00, -1.2850e+00, ..., -1.0241e+00,
-1.0914e+00, -3.8701e-01]],
[[-2.4769e+00, -5.0415e+00, -4.2636e+00, ..., -5.6627e-01,
-1.1087e+00, -3.8697e-02],
[-5.4472e+00, -8.2358e+00, -6.6588e+00, ..., -2.1690e+00,
-2.8260e+00, -7.3126e-01],
[-4.4248e+00, -6.4244e+00, -6.5525e+00, ..., -2.3870e+00,
-1.9630e+00, -4.5527e-01],
...,
[-4.4869e+00, -6.2314e+00, -5.7338e+00, ..., -1.4560e+00,
-1.6944e+00, -4.3986e-01],
[-6.4940e+00, -7.5266e+00, -6.4311e+00, ..., -2.5936e+00,
-3.4278e+00, -1.0221e+00],
[-6.9542e-02, -8.7847e-01, 3.3910e-01, ..., 3.6838e-01,
3.4085e-01, 9.1919e-02]],
[[-6.0815e-01, 5.5456e-01, 1.1450e+00, ..., 9.0661e-01,
3.8855e-01, 1.1514e+00],
[-8.4505e-01, -1.3926e-01, 3.5647e-01, ..., 5.4736e-01,
7.9293e-02, 4.8792e-01],
[-1.6589e+00, -1.4992e-01, -1.3889e-01, ..., 4.8817e-01,
1.9371e-01, 4.4191e-01],
...,
[-1.7534e+00, -7.8847e-02, 6.1561e-01, ..., 7.3709e-02,
9.9513e-02, 4.5458e-01],
[-1.5128e+00, 2.5400e-02, 3.5357e-02, ..., 1.1128e-01,
8.9079e-02, 6.1406e-01],
[-1.4101e+00, -5.5494e-01, -2.9076e-01, ..., -6.1943e-02,
4.0658e-02, 3.4855e-01]],
...,
[[ 1.1126e+00, -3.9636e-01, -1.6871e-01, ..., -9.1689e-02,
-3.8322e-01, -9.4737e-01],
[-8.1653e-01, -9.7059e-01, -8.0392e-01, ..., -7.4514e-01,
-1.3334e+00, -8.4439e-01],
[-6.3450e-01, -9.7556e-01, -7.8297e-01, ..., -3.3497e-01,
-1.2657e-01, -4.0235e-01],
...,
[-7.3138e-01, -1.2026e+00, -1.9373e+00, ..., -1.2143e-01,
-2.2884e-01, -3.9253e-01],
[-2.8095e+00, -2.9058e+00, -1.8544e+00, ..., -8.1038e-01,
-8.6084e-01, -7.9837e-01],
[-6.8734e-01, -1.7250e+00, -1.0331e+00, ..., 3.8671e-02,
-1.2284e-01, -3.5562e-01]],
[[ 1.2269e-02, -2.5826e-02, 6.3399e-01, ..., 3.7265e-01,
7.1910e-01, 3.2498e-01],
[-2.1420e-02, -7.1991e-01, 1.1614e-01, ..., -7.9435e-02,
1.1912e-01, -4.9463e-02],
[-6.5877e-02, -8.4994e-01, 4.0254e-01, ..., -4.2423e-01,
-3.2364e-01, 1.7273e-01],
...,
[ 3.9266e-02, -8.2685e-01, 6.5881e-01, ..., -1.2332e-01,
-1.7624e-01, 1.3849e-01],
[-2.5755e-01, -1.4587e+00, 6.4949e-02, ..., -4.0627e-01,
-3.8504e-01, -7.8582e-02],
[ 6.3945e-01, 1.3017e-01, 2.9550e-01, ..., -2.2171e-01,
-2.4412e-01, 2.9603e-01]],
[[-1.7104e+00, -6.5552e-01, -2.2464e-02, ..., 5.7821e-01,
3.7960e-01, 5.8730e-01],
[-2.2039e+00, -3.3174e-01, -4.1934e-02, ..., -3.6272e-02,
-1.0425e-01, 4.7038e-01],
[-1.8230e+00, -4.2326e-01, 2.3559e-01, ..., 1.2561e-01,
-1.1030e-01, 1.1012e-01],
...,
[-1.8795e+00, -5.2426e-01, 1.7661e-01, ..., 9.2072e-02,
-9.3248e-02, 9.6726e-02],
[-2.3318e+00, -1.2089e+00, -2.1306e-01, ..., -9.9968e-02,
-2.5961e-01, -1.9429e-01],
[-1.1343e+00, -2.2594e-01, 4.4616e-01, ..., -7.4562e-02,
-2.3686e-01, 1.2559e-01]]]], device='cuda:0'),
tensor([[[[-1.1430e-01, -5.1125e-01, -2.2443e-01, ..., -8.8632e-02,
3.0999e-01, -4.7245e-02],
[ 2.6522e-01, -1.1779e+00, -6.8649e-01, ..., -4.5166e-01,
-2.5470e-01, -1.8421e-01],
[ 1.0952e-01, -8.5843e-01, -3.5755e-01, ..., -3.3906e-01,
-2.0671e-01, -5.3439e-01],
...,
[-2.5857e-01, -1.6834e+00, -1.1974e+00, ..., -9.2474e-01,
-9.2946e-01, -2.0301e+00],
[-1.0927e+00, -1.7365e+00, -8.3483e-01, ..., -5.1926e-01,
-4.3559e-01, -1.7947e+00],
[-1.2844e+00, -2.1565e+00, -1.6091e+00, ..., -1.4044e+00,
-1.5370e+00, -9.6321e-01]],
[[-1.4508e+00, -3.0372e+00, -2.5490e+00, ..., -2.6517e+00,
-3.3323e+00, -1.3026e+00],
[-3.0981e+00, -4.9526e+00, -3.6057e+00, ..., -3.7702e+00,
-4.7002e+00, -2.0268e+00],
[-1.9643e+00, -2.9361e+00, -2.2181e+00, ..., -2.5335e+00,
-2.7084e+00, -1.2365e+00],
...,
[-3.6978e+00, -5.4668e+00, -5.1442e+00, ..., -5.3809e+00,
-6.1548e+00, -2.9066e+00],
[-6.1974e+00, -7.0504e+00, -6.0902e+00, ..., -6.0997e+00,
-7.4071e+00, -2.6741e+00],
[-3.5119e-02, -8.9437e-01, 2.6737e-01, ..., 3.9759e-01,
6.9843e-01, 1.9895e-01]],
[[-4.2988e-01, 2.1052e-01, 8.4299e-01, ..., 8.2619e-01,
5.6554e-01, 1.3580e+00],
[-3.2557e-01, -9.2310e-02, 6.0661e-01, ..., 6.1180e-01,
3.8660e-01, 3.0295e-01],
[-7.3643e-01, -5.6879e-02, 2.5792e-01, ..., 2.5750e-01,
3.7570e-01, 4.3684e-01],
...,
[-1.3750e+00, -8.1494e-04, 1.0049e-01, ..., 8.2281e-02,
1.7955e-03, 3.2128e-01],
[-1.2276e+00, 1.0479e-01, -2.9829e-02, ..., -1.6047e-02,
1.9682e-02, 3.0997e-01],
[-1.1680e+00, -4.6870e-02, 1.2257e-01, ..., 7.4681e-03,
-7.0832e-03, -3.3165e-01]],
...,
[[ 9.1703e-01, 2.2571e-01, 2.2760e-01, ..., 1.3767e-01,
1.6613e-01, -9.0190e-01],
[-1.0336e+00, -8.1791e-01, -6.9596e-01, ..., -9.8196e-01,
-1.4961e+00, -1.1166e+00],
[-5.8223e-01, -2.9259e-01, -2.9498e-01, ..., -3.2818e-01,
-4.9594e-01, -6.2653e-01],
...,
[-5.6042e-01, -5.8150e-01, -4.2936e-01, ..., -3.6495e-01,
-5.8310e-01, -1.1853e+00],
[-3.1526e+00, -2.9316e+00, -2.0503e+00, ..., -1.8940e+00,
-2.0019e+00, -2.0743e+00],
[-5.1724e-01, -1.5909e+00, -1.0630e+00, ..., -8.5927e-01,
-9.3009e-01, -1.3439e+00]],
[[ 8.4530e-02, 3.2710e-01, 6.8291e-01, ..., 6.9203e-01,
9.8774e-01, 3.8657e-01],
[-6.9257e-03, -3.3470e-01, 1.9735e-01, ..., 4.2201e-01,
5.7280e-01, -2.0388e-01],
[-3.0538e-01, -4.8928e-01, 2.1137e-01, ..., 1.6619e-01,
3.1747e-01, 1.0408e-02],
...,
[-1.0493e-01, -9.6104e-01, 1.0679e-01, ..., 2.3905e-01,
1.5164e-01, -7.1462e-01],
[-5.2996e-01, -1.8524e+00, -2.4375e-01, ..., 2.8557e-03,
-3.4067e-02, -9.2609e-01],
[ 4.3640e-01, -1.4478e-01, 1.3511e-01, ..., 2.7707e-01,
3.1743e-01, 3.3638e-01]],
[[-9.3868e-01, -2.1178e-01, 1.7043e-01, ..., 8.9166e-02,
4.6104e-02, 7.2830e-01],
[-1.1549e+00, -2.3386e-01, -2.9776e-01, ..., -3.3572e-01,
-8.6018e-02, 1.1584e+00],
[-7.9474e-01, -2.1248e-01, -2.1329e-01, ..., -1.8050e-01,
-1.2690e-01, 6.7368e-01],
...,
[-1.1207e+00, -9.4444e-01, -4.5114e-01, ..., -4.0113e-01,
-8.0620e-02, 8.3067e-01],
[-1.8253e+00, -1.8212e+00, -1.0112e+00, ..., -7.9503e-01,
-4.8655e-01, 4.2937e-01],
[-9.3980e-01, -9.5980e-01, -2.4643e-01, ..., -2.8133e-01,
-1.4415e-01, 1.1254e+00]]],
[[[ 4.2132e-02, -2.2524e-01, -3.1406e-02, ..., -9.5242e-01,
-5.7341e-01, -1.4807e+00],
[ 3.1408e-01, -9.5826e-01, -7.0886e-01, ..., -1.1134e+00,
-1.3227e+00, -1.5069e+00],
[ 2.2572e-01, -5.9046e-01, -4.9829e-01, ..., -8.8405e-01,
-1.0885e+00, -1.8187e+00],
...,
[ 1.6102e-03, -5.8627e-01, -3.9981e-01, ..., -1.0956e+00,
-1.1269e+00, -1.9799e+00],
[-6.2431e-01, -6.8471e-01, -3.6860e-02, ..., -4.6069e-01,
-5.6142e-01, -1.6717e+00],
[-7.3760e-01, -1.3483e+00, -1.1799e+00, ..., -1.1164e+00,
-1.5218e+00, -7.5293e-01]],
[[-7.2212e-01, -2.1387e+00, -1.8730e+00, ..., -3.9673e+00,
-5.5575e+00, -1.8639e+00],
[-2.0710e+00, -3.7599e+00, -2.9541e+00, ..., -5.9085e+00,
-8.1588e+00, -3.4471e+00],
[-1.6379e+00, -2.5702e+00, -2.5190e+00, ..., -5.0172e+00,
-6.2376e+00, -2.6130e+00],
...,
[-1.7568e+00, -2.4111e+00, -2.3841e+00, ..., -5.3666e+00,
-6.5962e+00, -2.6651e+00],
[-3.7853e+00, -4.4241e+00, -3.6417e+00, ..., -6.3630e+00,
-7.7968e+00, -2.5236e+00],
[ 2.7560e-01, -3.3238e-01, 3.6501e-01, ..., 4.3892e-01,
4.5849e-01, 2.2087e-01]],
[[-2.2205e-01, 1.2282e-01, 7.1916e-01, ..., 6.6115e-01,
3.0997e-01, 1.5103e+00],
[ 7.3812e-02, -4.0578e-01, 5.8214e-01, ..., 5.6763e-01,
2.6097e-03, 1.2332e-01],
[-1.0808e-01, -2.2895e-01, 2.3648e-01, ..., -4.7854e-02,
6.0013e-03, -1.2844e-01],
...,
[-2.3573e-01, -1.9549e-01, 1.4120e-01, ..., -3.0477e-01,
-4.0257e-01, -1.9973e-01],
[-2.6699e-01, -1.6077e-01, 1.5704e-01, ..., -9.2431e-02,
-2.3171e-02, -1.8988e-01],
[-1.5684e-01, -5.1805e-01, 1.0469e-01, ..., 9.6350e-02,
2.3206e-01, -5.5119e-01]],
...,
[[ 5.8893e-01, -3.0328e-01, -2.7388e-02, ..., 4.2976e-01,
3.2507e-01, -1.2949e+00],
[-5.3799e-01, -5.5193e-01, -5.0791e-01, ..., -1.4087e-01,
-7.6359e-01, -1.3784e+00],
[-3.7168e-01, -2.2170e-01, -2.6620e-01, ..., -1.4320e-01,
-1.0802e-01, -9.6560e-01],
...,
[-4.3454e-01, -1.4929e-01, -1.6472e-01, ..., -4.0679e-01,
-7.4738e-02, -8.8261e-01],
[-2.3269e+00, -1.7181e+00, -1.2695e+00, ..., -1.5262e+00,
-1.5470e+00, -1.8415e+00],
[-4.7841e-01, -7.9315e-01, -4.1129e-01, ..., -3.8092e-01,
-5.9213e-01, -1.1831e+00]],
[[ 1.5705e-01, 2.5022e-01, 6.2149e-01, ..., -8.1004e-02,
3.4529e-01, -4.0945e-01],
[ 1.2779e-01, 2.4410e-01, 4.2614e-01, ..., 2.7804e-01,
1.1258e-01, -8.3748e-01],
[-1.2313e-01, 9.1951e-02, 2.8415e-01, ..., 3.5871e-01,
1.8574e-01, -4.5916e-01],
...,
[-1.7953e-01, 1.7690e-02, 3.0952e-01, ..., 1.2862e-01,
9.3511e-02, -5.6661e-01],
[-5.3822e-01, -6.8304e-01, -3.8196e-02, ..., -1.7786e-01,
-2.8778e-01, -9.0103e-01],
[ 5.0818e-01, 4.1266e-01, 6.8371e-02, ..., 3.4322e-01,
3.4329e-01, 2.9172e-01]],
[[-8.4766e-01, 2.7762e-01, 3.4971e-01, ..., -3.4131e-02,
-5.1914e-02, 5.6776e-01],
[-1.1509e+00, 2.2364e-01, -3.4262e-01, ..., -8.5745e-01,
-3.5822e-01, 8.8360e-01],
[-9.9288e-01, 3.5747e-01, -1.1237e-01, ..., -5.4224e-01,
-4.9579e-01, 6.2835e-01],
...,
[-9.3852e-01, 2.0131e-01, -9.9702e-02, ..., -4.9973e-02,
-1.8168e-01, 7.6977e-01],
[-1.6817e+00, -3.7543e-01, -3.4445e-01, ..., -7.3511e-01,
-7.1660e-01, 4.1432e-01],
[-7.2896e-01, -6.8039e-02, -2.9949e-01, ..., -4.8709e-01,
-4.2119e-01, 9.7300e-01]]],
[[[-7.3496e-01, -1.4117e+00, -1.4303e+00, ..., 4.9378e-03,
6.2954e-01, 5.5063e-01],
[-2.9346e-01, -2.4118e+00, -2.1962e+00, ..., -5.2309e-01,
-2.2835e-01, 1.2183e-01],
[-3.9211e-01, -2.1176e+00, -1.8101e+00, ..., -2.7433e-01,
-1.0849e-01, -1.0718e-01],
...,
[-4.7947e-01, -1.9069e+00, -1.7688e+00, ..., -2.9264e-01,
-8.5866e-02, -1.6799e-01],
[-9.9173e-01, -1.6845e+00, -1.1890e+00, ..., 3.6466e-02,
2.7058e-01, -1.5099e-02],
[-8.3633e-01, -1.6324e+00, -1.2297e+00, ..., -1.0253e+00,
-1.1078e+00, -4.0282e-01]],
[[-2.3742e+00, -4.8678e+00, -4.2058e+00, ..., -4.5953e-01,
-1.0355e+00, -3.1091e-02],
[-5.3571e+00, -8.1352e+00, -6.6412e+00, ..., -1.8737e+00,
-2.7123e+00, -6.8416e-01],
[-4.3360e+00, -6.2275e+00, -5.8254e+00, ..., -1.3393e+00,
-1.5556e+00, -3.5956e-01],
...,
[-4.3510e+00, -6.0874e+00, -5.5320e+00, ..., -1.3748e+00,
-1.6192e+00, -4.0314e-01],
[-6.3956e+00, -7.4414e+00, -6.3367e+00, ..., -2.4120e+00,
-3.3424e+00, -9.5293e-01],
[-4.7570e-02, -8.5170e-01, 4.2949e-01, ..., 3.2177e-01,
3.7538e-01, 1.0295e-01]],
[[-5.6794e-01, 6.4234e-01, 1.1109e+00, ..., 8.3061e-01,
2.9556e-01, 1.0824e+00],
[-9.1536e-01, -8.6435e-03, 3.9440e-01, ..., 4.3660e-01,
1.5287e-02, 4.5241e-01],
[-1.6853e+00, -1.1962e-01, -1.0894e-01, ..., 8.2371e-02,
3.6063e-02, 3.7435e-01],
...,
[-1.7561e+00, -5.3333e-02, -1.8412e-02, ..., 5.9284e-02,
5.8126e-02, 4.1995e-01],
[-1.5363e+00, 9.8776e-02, 6.5523e-02, ..., 1.3945e-01,
2.9542e-02, 5.8177e-01],
[-1.4129e+00, -5.0220e-01, -2.5027e-01, ..., -6.6341e-02,
-5.0544e-03, 3.0641e-01]],
...,
[[ 1.0577e+00, -4.3490e-01, -2.2918e-01, ..., -1.1923e-01,
-3.9072e-01, -9.3899e-01],
[-7.7749e-01, -9.6256e-01, -7.9354e-01, ..., -6.8286e-01,
-1.2774e+00, -8.4857e-01],
[-6.8058e-01, -1.0026e+00, -6.9427e-01, ..., -1.3788e-01,
-1.7232e-01, -4.2131e-01],
...,
[-7.5328e-01, -1.1794e+00, -8.5816e-01, ..., -7.5549e-02,
-2.0785e-01, -3.9145e-01],
[-2.7994e+00, -2.8625e+00, -1.8719e+00, ..., -8.2645e-01,
-9.0061e-01, -8.2239e-01],
[-6.6950e-01, -1.6698e+00, -9.5934e-01, ..., -5.6422e-02,
-1.4364e-01, -3.2530e-01]],
[[ 2.5447e-02, 4.2002e-02, 5.8458e-01, ..., 3.3096e-01,
6.8552e-01, 2.7136e-01],
[ 1.6132e-02, -6.9132e-01, 1.2634e-01, ..., -1.2290e-01,
1.0002e-01, -7.2140e-02],
[-8.8144e-02, -7.9935e-01, 3.1018e-01, ..., -1.5662e-01,
-8.8378e-02, 1.7497e-01],
...,
[ 4.9652e-02, -8.1444e-01, 3.9280e-01, ..., -9.3468e-02,
-1.1169e-01, 1.5495e-01],
[-2.7495e-01, -1.4795e+00, 4.6999e-02, ..., -3.4688e-01,
-3.1293e-01, -7.3178e-02],
[ 6.4964e-01, 1.4912e-01, 2.9794e-01, ..., -2.1805e-01,
-2.3498e-01, 2.9470e-01]],
[[-1.6005e+00, -6.0651e-01, 4.9368e-02, ..., 5.6010e-01,
3.6433e-01, 5.4661e-01],
[-2.0869e+00, -3.3040e-01, -3.6124e-02, ..., -8.0423e-02,
-1.9422e-01, 4.0179e-01],
[-1.7571e+00, -3.3887e-01, 2.0361e-01, ..., 5.2338e-02,
-1.9781e-01, 6.9070e-02],
...,
[-1.7388e+00, -5.0196e-01, 1.6888e-01, ..., 3.0717e-02,
-5.3672e-02, 8.6566e-02],
[-2.2142e+00, -1.1571e+00, -1.9657e-01, ..., -1.3731e-01,
-2.1108e-01, -2.2364e-01],
[-1.0549e+00, -2.0011e-01, 4.5795e-01, ..., -1.0557e-01,
-2.5138e-01, 1.0836e-01]]],
[[[ 5.8661e-02, -2.3018e-01, -4.1033e-02, ..., -9.7042e-01,
-5.1817e-01, -1.4187e+00],
[ 2.6775e-01, -1.0843e+00, -7.3924e-01, ..., -1.1708e+00,
-1.3088e+00, -1.4812e+00],
[-2.4456e-01, -1.3808e+00, -7.6188e-01, ..., -1.6384e+00,
-1.3690e+00, -1.8125e+00],
...,
[ 7.1715e-02, -5.4686e-01, -4.2802e-01, ..., -9.6957e-01,
-1.1191e+00, -1.9126e+00],
[-4.7332e-01, -6.0776e-01, -7.9551e-02, ..., -4.9351e-01,
-6.1107e-01, -1.6958e+00],
[-6.1792e-01, -1.2365e+00, -1.0596e+00, ..., -1.1343e+00,
-1.5191e+00, -7.6453e-01]],
[[-6.1087e-01, -2.0414e+00, -1.7420e+00, ..., -3.9615e+00,
-5.4790e+00, -1.8556e+00],
[-2.2039e+00, -4.0523e+00, -3.0335e+00, ..., -5.9043e+00,
-8.0832e+00, -3.4214e+00],
[-3.7523e+00, -3.2913e+00, -3.1737e+00, ..., -7.1004e+00,
-6.8876e+00, -2.5825e+00],
...,
[-1.7558e+00, -2.3871e+00, -2.4429e+00, ..., -5.2738e+00,
-6.2859e+00, -2.5591e+00],
[-3.5212e+00, -4.0370e+00, -3.4088e+00, ..., -6.4171e+00,
-7.8649e+00, -2.5986e+00],
[ 2.7426e-01, -2.5345e-01, 4.1037e-01, ..., 4.3399e-01,
5.0565e-01, 1.9321e-01]],
[[-2.5291e-01, 1.0096e-01, 7.7495e-01, ..., 6.3897e-01,
2.5614e-01, 1.5498e+00],
[ 5.9116e-02, -4.1323e-01, 5.9275e-01, ..., 5.8890e-01,
-9.5750e-02, 1.2877e-01],
[-5.8525e-01, -5.9308e-01, 1.4157e-01, ..., 1.2265e-01,
2.8668e-02, -1.3127e-01],
...,
[-2.2058e-01, -1.8594e-01, 1.2880e-01, ..., 1.8158e-02,
-2.4543e-01, -2.6048e-01],
[-1.7842e-01, -7.8877e-02, 1.7334e-01, ..., -5.9938e-02,
-6.8664e-02, -2.5897e-01],
[-8.8583e-02, -4.4600e-01, 1.1627e-01, ..., 1.1017e-01,
2.1210e-01, -5.9801e-01]],
...,
[[ 4.9517e-01, -3.1945e-01, -5.5361e-02, ..., 4.5657e-01,
3.5057e-01, -1.3119e+00],
[-6.9451e-01, -5.6917e-01, -5.8890e-01, ..., -1.7029e-01,
-7.7859e-01, -1.3928e+00],
[-9.2808e-01, -6.6967e-01, -2.7071e-01, ..., -2.7402e-02,
-6.9298e-02, -9.3576e-01],
...,
[-3.9850e-01, -9.5866e-02, -1.7590e-01, ..., -1.7252e-01,
-1.2608e-01, -1.0061e+00],
[-2.1501e+00, -1.5188e+00, -1.1277e+00, ..., -1.5067e+00,
-1.5766e+00, -1.8430e+00],
[-4.6762e-01, -6.7334e-01, -3.7364e-01, ..., -3.5411e-01,
-5.2409e-01, -1.1573e+00]],
[[ 1.2704e-01, 2.7877e-01, 6.6900e-01, ..., -9.5168e-02,
4.0168e-01, -3.5354e-01],
[ 5.6443e-02, 2.9625e-01, 4.2899e-01, ..., 1.7205e-01,
1.0127e-01, -8.4213e-01],
[-9.7149e-02, -1.4750e-01, -1.8750e-01, ..., -9.8186e-01,
-5.7690e-01, -4.5103e-01],
...,
[-1.2545e-01, 3.6597e-02, 2.9036e-01, ..., 1.1819e-01,
7.7254e-02, -5.2406e-01],
[-4.1818e-01, -5.8590e-01, 6.7678e-03, ..., -2.1874e-01,
-3.2776e-01, -8.9053e-01],
[ 5.3026e-01, 4.2405e-01, 6.8915e-02, ..., 3.2349e-01,
3.2833e-01, 3.1122e-01]],
[[-9.4680e-01, 2.9091e-01, 4.2419e-01, ..., -8.4998e-02,
-7.7744e-02, 5.7046e-01],
[-1.2538e+00, 1.7529e-01, -3.6950e-01, ..., -9.2500e-01,
-4.2837e-01, 8.7091e-01],
[-1.8039e+00, -2.6987e-01, -1.7719e-01, ..., -5.7030e-01,
-4.0857e-01, 6.2428e-01],
...,
[-9.6354e-01, 2.1789e-01, -1.0095e-01, ..., -5.3010e-01,
-5.6821e-01, 5.9888e-01],
[-1.5941e+00, -2.7572e-01, -3.1021e-01, ..., -7.4241e-01,
-7.4621e-01, 3.9407e-01],
[-6.3257e-01, 3.3449e-02, -2.5531e-01, ..., -4.7059e-01,
-4.2745e-01, 9.9813e-01]]],
[[[-6.9378e-01, -1.3755e+00, -1.4353e+00, ..., -2.2331e-01,
2.0577e-01, -4.4842e-01],
[-2.5393e-01, -2.4041e+00, -2.1825e+00, ..., -5.4302e-01,
-4.5282e-01, -6.2107e-01],
[-4.1011e-01, -2.1320e+00, -1.8870e+00, ..., -2.2648e-01,
-4.2879e-01, -1.2564e+00],
...,
[-5.0030e-01, -1.9290e+00, -1.8403e+00, ..., -3.9976e-01,
-1.3728e+00, -1.0075e+00],
[-1.0106e+00, -1.7128e+00, -1.1646e+00, ..., 2.8151e-01,
-5.5561e-01, -1.2336e+00],
[-8.3021e-01, -1.6529e+00, -1.2496e+00, ..., -8.6092e-01,
-2.0627e+00, -1.0336e+00]],
[[-2.3291e+00, -4.8610e+00, -4.2905e+00, ..., -2.3086e+00,
-3.5725e+00, -1.3464e+00],
[-5.3442e+00, -8.0480e+00, -6.4487e+00, ..., -3.4958e+00,
-5.1594e+00, -2.1399e+00],
[-4.3432e+00, -6.2395e+00, -5.4428e+00, ..., -3.7184e+00,
-4.9181e+00, -2.3170e+00],
...,
[-4.3809e+00, -6.0696e+00, -5.5034e+00, ..., -2.6572e+00,
-3.1604e+00, -2.3298e+00],
[-6.3828e+00, -7.3954e+00, -6.3168e+00, ..., -3.7652e+00,
-6.5280e+00, -2.0755e+00],
[-3.4576e-02, -8.4727e-01, 4.1930e-01, ..., 7.2000e-01,
-3.7632e-01, -5.8313e-01]],
[[-5.8193e-01, 5.9998e-01, 1.1030e+00, ..., 7.6821e-01,
5.4916e-01, 1.2817e+00],
[-9.1032e-01, -2.8579e-02, 3.4501e-01, ..., 4.0612e-01,
3.4594e-01, 1.9793e-01],
[-1.6613e+00, -1.2163e-01, -5.9516e-02, ..., 1.4118e-01,
1.5351e-01, -4.7711e-02],
...,
[-1.7662e+00, -4.6903e-02, -1.1172e-01, ..., -7.1932e-01,
-5.5140e+00, -1.7585e-01],
[-1.5090e+00, 7.7732e-02, 6.8444e-02, ..., -9.7021e-01,
-3.8840e+00, -8.9428e-02],
[-1.3993e+00, -5.2468e-01, -2.5763e-01, ..., -6.9992e-01,
-1.4725e+00, 1.1748e-02]],
...,
[[ 1.0444e+00, -3.8030e-01, -1.9343e-01, ..., 3.6759e-01,
2.7104e-01, -7.4132e-01],
[-8.4459e-01, -9.7658e-01, -8.0291e-01, ..., -2.4398e-01,
-9.2747e-01, -9.3933e-01],
[-6.7853e-01, -9.4656e-01, -7.2770e-01, ..., -1.1411e-01,
-2.0732e-01, -6.4182e-01],
...,
[-7.3353e-01, -1.1496e+00, -7.5097e-01, ..., 6.1898e-02,
-1.0689e+00, -1.7622e+00],
[-2.7698e+00, -2.8356e+00, -1.8323e+00, ..., -1.6146e+00,
-1.9393e+00, -1.9685e+00],
[-6.6306e-01, -1.6562e+00, -9.7356e-01, ..., 1.2458e-01,
8.2873e-02, -1.0268e+00]],
[[ 3.0608e-02, 3.3122e-02, 6.2011e-01, ..., 2.9896e-01,
7.8636e-01, -2.3076e-02],
[-2.4837e-02, -6.8810e-01, 1.5488e-01, ..., 4.3801e-01,
6.8152e-01, -3.3198e-01],
[-1.2606e-01, -8.1762e-01, 3.0134e-01, ..., 5.2997e-01,
4.7913e-01, -5.2871e-01],
...,
[ 3.5032e-02, -7.8264e-01, 2.9939e-01, ..., 1.2319e+00,
-1.2516e+00, -1.0355e+00],
[-2.7056e-01, -1.4194e+00, 5.0350e-02, ..., 2.5953e-01,
-1.4714e+00, -1.7605e+00],
[ 6.5457e-01, 1.6892e-01, 2.9685e-01, ..., 8.6540e-02,
-9.8374e-01, -1.2109e-01]],
[[-1.6107e+00, -5.9153e-01, 1.1925e-01, ..., 2.7782e-01,
2.0724e-01, 7.9704e-01],
[-2.0978e+00, -3.6646e-01, -3.2287e-02, ..., -4.0123e-01,
-1.2003e-01, 1.1293e+00],
[-1.7561e+00, -3.7733e-01, 1.7405e-01, ..., -8.6879e-02,
4.6913e-02, 8.7165e-01],
...,
[-1.7899e+00, -5.1484e-01, 2.4548e-01, ..., -2.1998e-01,
2.8409e-01, -2.4575e+00],
[-2.2250e+00, -1.1536e+00, -1.3995e-01, ..., 5.4346e-01,
5.5205e-01, -1.9847e+00],
[-1.0416e+00, -2.1020e-01, 4.5769e-01, ..., 3.9969e-01,
4.9638e-01, -4.7949e-01]]]], device='cuda:0'),
tensor([[[[-7.6003e-01, -1.4239e+00, -1.3933e+00, ..., -2.3508e-01,
2.0693e-01, -4.3605e-01],
[-3.0631e-01, -2.4417e+00, -2.0557e+00, ..., -6.0126e-01,
-4.4820e-01, -6.0363e-01],
[-3.9782e-01, -2.1175e+00, -1.5468e+00, ..., -3.5004e-01,
-4.8836e-01, -1.1473e+00],
...,
[-4.8410e-01, -1.9618e+00, -1.8688e+00, ..., -3.3292e-01,
-2.6462e-01, -1.0945e+00],
[-9.3268e-01, -1.6726e+00, -1.2420e+00, ..., 5.0703e-02,
1.2886e-01, -9.9931e-01],
[-7.4239e-01, -1.5588e+00, -1.2220e+00, ..., -1.0319e+00,
-1.3055e+00, -7.6120e-01]],
[[-2.3722e+00, -4.8143e+00, -4.1764e+00, ..., -2.2953e+00,
-3.5347e+00, -1.2591e+00],
[-5.3652e+00, -8.1177e+00, -7.2083e+00, ..., -3.5056e+00,
-5.0892e+00, -2.0477e+00],
[-4.3239e+00, -6.2453e+00, -8.1013e+00, ..., -3.1838e+00,
-4.0024e+00, -1.8331e+00],
...,
[-4.3873e+00, -6.1190e+00, -5.4799e+00, ..., -3.0133e+00,
-3.9119e+00, -1.6606e+00],
[-6.2670e+00, -7.3526e+00, -6.2775e+00, ..., -4.2732e+00,
-5.8764e+00, -2.0754e+00],
[-6.6746e-03, -7.9250e-01, 3.3904e-01, ..., 4.6711e-01,
6.1793e-01, 1.3999e-01]],
[[-6.0575e-01, 5.8355e-01, 1.1010e+00, ..., 7.9375e-01,
5.5200e-01, 1.3001e+00],
[-9.3249e-01, -8.0091e-02, 3.6773e-01, ..., 3.8754e-01,
3.9382e-01, 2.1001e-01],
[-1.6347e+00, -1.4257e-01, -3.4335e-01, ..., 3.3140e-02,
2.4779e-01, 6.8100e-02],
...,
[-1.7272e+00, -8.7203e-02, -1.0597e-01, ..., 6.7526e-02,
2.2500e-01, 7.5826e-02],
[-1.5559e+00, 5.3714e-02, -1.1186e-02, ..., 9.3615e-02,
2.3828e-01, 1.2598e-01],
[-1.3457e+00, -4.6843e-01, -3.2527e-01, ..., 5.4274e-03,
2.0125e-01, -1.4960e-01]],
...,
[[ 1.0519e+00, -4.3895e-01, -2.2435e-01, ..., 3.8396e-01,
2.6124e-01, -7.5607e-01],
[-8.4038e-01, -9.6701e-01, -8.5240e-01, ..., -2.5481e-01,
-1.0113e+00, -9.5190e-01],
[-7.0377e-01, -9.7740e-01, -3.0254e-01, ..., -4.8640e-02,
-2.4755e-01, -6.5775e-01],
...,
[-7.2968e-01, -1.1396e+00, -7.5460e-01, ..., -1.0041e-01,
-2.2782e-01, -6.0084e-01],
[-2.6614e+00, -2.7579e+00, -1.8749e+00, ..., -1.1818e+00,
-1.2574e+00, -1.2011e+00],
[-6.2735e-01, -1.5812e+00, -9.1561e-01, ..., -1.5739e-01,
-2.4883e-01, -5.4522e-01]],
[[ 5.9491e-03, 2.9162e-03, 6.0594e-01, ..., 3.0750e-01,
7.7393e-01, 3.3341e-02],
[-4.1132e-02, -7.5063e-01, 2.0292e-01, ..., 4.0281e-01,
6.1810e-01, -2.7908e-01],
[-1.2602e-01, -8.1730e-01, 7.5448e-01, ..., 4.2802e-01,
3.7633e-01, -2.1759e-01],
...,
[ 3.8099e-02, -8.0829e-01, 2.8834e-01, ..., 4.0406e-01,
4.5600e-01, -1.3568e-01],
[-2.8797e-01, -1.4727e+00, 2.4359e-02, ..., -5.7759e-02,
4.4476e-02, -5.7190e-01],
[ 7.3230e-01, 2.1158e-01, 3.2356e-01, ..., 1.9470e-01,
2.5869e-01, 3.4079e-01]],
[[-1.6200e+00, -6.3141e-01, 4.9053e-02, ..., 2.9018e-01,
2.1565e-01, 8.2257e-01],
[-2.0780e+00, -3.5387e-01, 1.7555e-02, ..., -3.7381e-01,
-1.2648e-01, 1.1669e+00],
[-1.7211e+00, -3.9999e-01, 1.4244e-01, ..., -4.9332e-02,
7.4073e-02, 9.3384e-01],
...,
[-1.8121e+00, -4.7483e-01, 2.8003e-01, ..., -7.8011e-02,
5.2820e-03, 9.6309e-01],
[-2.1604e+00, -1.0169e+00, -4.2262e-02, ..., -2.8898e-01,
-1.2027e-01, 5.7684e-01],
[-1.0665e+00, -1.2449e-01, 5.5031e-01, ..., -3.1159e-01,
-3.1344e-01, 6.8658e-01]]],
[[[-1.7224e-01, -2.6131e-01, 1.8038e-01, ..., 1.6505e-01,
5.3827e-01, 2.1581e-01],
[-8.7347e-02, -1.4148e+00, -8.3803e-01, ..., -7.1839e-01,
-5.0807e-01, -2.9998e-01],
[-3.9131e-02, -1.1808e+00, -6.3721e-01, ..., -7.2142e-01,
-4.3949e-01, -6.7488e-01],
...,
[-2.6155e-01, -1.7029e+00, -1.1809e+00, ..., -1.3138e+00,
-9.8738e-01, -2.0378e+00],
[-1.0542e+00, -1.6976e+00, -7.9343e-01, ..., -6.4584e-01,
-4.5197e-01, -1.7953e+00],
[-1.2562e+00, -2.1489e+00, -1.6007e+00, ..., -1.4357e+00,
-1.5661e+00, -9.7381e-01]],
[[-2.0051e-01, -1.3934e+00, -6.2725e-01, ..., -1.3238e+00,
-1.3416e+00, -2.0515e-01],
[-1.9144e+00, -3.8215e+00, -2.4860e+00, ..., -2.5194e+00,
-3.5295e+00, -1.1894e+00],
[-1.7999e+00, -3.2005e+00, -2.5694e+00, ..., -2.5610e+00,
-3.0140e+00, -1.2687e+00],
...,
[-3.7053e+00, -5.4800e+00, -5.2556e+00, ..., -5.9999e+00,
-6.1528e+00, -2.9234e+00],
[-6.1752e+00, -7.0422e+00, -6.0454e+00, ..., -6.1900e+00,
-7.4437e+00, -2.7198e+00],
[-4.4107e-03, -8.5606e-01, 2.6582e-01, ..., 4.0901e-01,
7.1619e-01, 2.0039e-01]],
[[-4.4621e-01, -7.8472e-02, 3.3271e-01, ..., -6.6053e-03,
3.9442e-02, 9.3797e-01],
[-3.4914e-01, -4.0314e-01, 4.6284e-01, ..., 4.0577e-01,
2.3622e-01, 4.7491e-01],
[-7.5893e-01, -2.2904e-01, 1.7960e-01, ..., 3.0360e-01,
1.6618e-01, 3.7993e-01],
...,
[-1.3879e+00, 9.9092e-03, 2.0877e-01, ..., 4.7799e-01,
-2.3550e-02, 2.9396e-01],
[-1.2295e+00, 1.5494e-01, -2.3841e-02, ..., 2.7094e-02,
-1.4756e-02, 3.0642e-01],
[-1.1733e+00, -4.0219e-02, 1.1589e-01, ..., 7.5739e-03,
-5.8956e-02, -3.4571e-01]],
...,
[[ 2.1391e-01, -5.0885e-01, -3.6366e-01, ..., -1.6220e-01,
-3.0695e-01, -8.1142e-01],
[-6.9506e-01, -6.1445e-01, -5.3159e-01, ..., -7.0979e-01,
-1.1152e+00, -8.2756e-01],
[-3.8331e-01, -1.5387e-01, -2.3857e-02, ..., 3.6019e-02,
-1.7957e-01, -5.3686e-01],
...,
[-6.4463e-01, -6.0786e-01, -4.2052e-01, ..., -6.9740e-01,
-5.8130e-01, -1.2254e+00],
[-3.0981e+00, -2.8777e+00, -1.9880e+00, ..., -1.8767e+00,
-2.0210e+00, -2.0667e+00],
[-5.2553e-01, -1.5923e+00, -1.0921e+00, ..., -8.7042e-01,
-9.4794e-01, -1.3576e+00]],
[[-1.0534e-01, 5.9632e-02, 3.8267e-01, ..., -6.6761e-02,
6.1733e-01, 1.8878e-01],
[-2.9933e-01, -2.3586e-01, -2.1242e-03, ..., -7.0297e-02,
1.1525e-01, -1.1217e-01],
[-5.4591e-01, -4.4224e-01, 3.4875e-03, ..., -1.4767e-01,
1.5821e-03, -7.5044e-02],
...,
[-1.3449e-01, -1.0019e+00, 8.5455e-02, ..., -9.2747e-02,
1.6696e-01, -7.1650e-01],
[-5.2071e-01, -1.8251e+00, -2.1774e-01, ..., -1.4351e-01,
-6.7674e-03, -9.4826e-01],
[ 4.3974e-01, -1.6034e-01, 1.2093e-01, ..., 2.5519e-01,
2.9685e-01, 3.3254e-01]],
[[-5.8250e-01, 4.3093e-01, 5.3693e-01, ..., 3.5769e-01,
4.2284e-01, 6.5933e-01],
[-8.0322e-01, 2.5564e-01, -2.2148e-01, ..., -2.2758e-01,
-1.4054e-01, 6.7508e-01],
[-4.5893e-01, 3.2639e-01, 4.4197e-02, ..., 2.7024e-01,
5.5432e-02, 5.7518e-01],
...,
[-1.0741e+00, -9.2037e-01, -5.3498e-01, ..., -5.1175e-02,
-3.8253e-02, 8.3083e-01],
[-1.7674e+00, -1.8050e+00, -9.9828e-01, ..., -8.4460e-01,
-4.6099e-01, 4.2804e-01],
[-9.2566e-01, -9.4301e-01, -2.7150e-01, ..., -3.0006e-01,
-1.6586e-01, 1.1260e+00]]],
[[[-2.4978e+00, -2.3957e+00, -1.1984e+00, ..., -1.0471e+00,
-5.3238e-01, -1.5358e+00],
[-1.0994e+00, -1.5208e-01, -3.8385e-01, ..., -1.3187e+00,
-1.3702e+00, -1.5775e+00],
[-2.5761e+00, -1.0716e+00, 3.8827e-01, ..., -1.0659e+00,
-1.1135e+00, -1.8649e+00],
...,
[-5.6444e+00, -6.3873e-01, -1.5991e+00, ..., -6.8364e+00,
-3.8795e+00, -3.0870e+00],
[-1.7019e+00, 1.7776e+00, 5.2200e-02, ..., -2.9805e+00,
-1.7607e+00, -2.7244e+00],
[ 9.9414e-02, -1.3713e+00, -2.1075e+00, ..., -1.6589e+00,
-2.2763e+00, -1.4430e+00]],
[[-6.5328e+00, -7.7335e+00, -5.5179e+00, ..., -3.8454e+00,
-5.5508e+00, -1.7887e+00],
[-1.3105e+01, -1.2190e+01, -8.0782e+00, ..., -6.2595e+00,
-8.6284e+00, -3.5246e+00],
[-1.4282e+01, -1.5063e+01, -6.6261e+00, ..., -5.1274e+00,
-6.4435e+00, -2.4227e+00],
...,
[-7.4359e+00, -5.4902e+00, -2.6965e+00, ..., -1.0007e+01,
-9.4064e+00, -4.1561e+00],
[-4.4656e+00, -4.3720e+00, -2.7454e+00, ..., -8.1465e+00,
-1.0035e+01, -3.1113e+00],
[ 6.9655e-01, -1.6485e+00, 6.5619e-01, ..., 2.2763e-01,
1.1452e-02, -2.7179e-02]],
[[-1.2621e+00, 1.1356e+00, 1.7663e+00, ..., 6.3040e-01,
1.2686e-01, 1.5223e+00],
[-7.1954e+00, -1.7496e+00, 1.8187e+00, ..., 5.5390e-01,
-2.2699e-01, 5.6592e-03],
[-1.2614e+01, -1.0473e+01, -2.7737e+00, ..., 1.7539e-02,
-1.0273e-01, -1.3833e-01],
...,
[-4.5061e+00, -2.0676e+00, 1.1301e+00, ..., -6.2796e+00,
-3.7228e+00, 1.8083e-01],
[-1.0239e+00, -7.4086e-01, 2.2352e+00, ..., 6.0486e-02,
-6.4078e-01, 5.4233e-01],
[-1.1246e+00, -5.4343e-01, 3.3875e-01, ..., -2.8722e-01,
-6.9468e-02, -2.9167e-01]],
...,
[[ 1.5997e+00, 1.3436e+00, 6.6015e-01, ..., 4.5137e-01,
4.0272e-01, -1.3643e+00],
[ 6.1676e-01, -1.5705e+00, -4.7462e-01, ..., -2.7644e-01,
-8.2242e-01, -1.4241e+00],
[-4.3529e+00, -1.0996e+01, -3.6355e+00, ..., -1.0406e-01,
-1.1996e-02, -9.5255e-01],
...,
[-1.9420e+00, -7.3530e+00, -7.4602e+00, ..., -4.7242e+00,
-5.9763e+00, -3.6550e+00],
[-4.6873e+00, -1.0069e+01, -5.8618e+00, ..., -5.8598e+00,
-5.0404e+00, -3.0899e+00],
[-2.4967e+00, -4.0830e+00, -5.2072e-01, ..., -2.3773e+00,
-1.0453e+00, -1.1594e+00]],
[[-6.3645e-01, -1.3500e+00, -1.1298e+00, ..., 2.2863e-02,
4.6484e-01, -2.3047e-01],
[-2.6524e-01, -1.6848e+00, -9.0860e-01, ..., 2.0795e-01,
9.4416e-02, -8.2712e-01],
[-5.3895e+00, -3.0681e+00, -6.7206e-01, ..., 3.1349e-01,
1.9456e-01, -2.9336e-01],
...,
[-4.1353e+00, -2.5889e+00, -1.2717e+00, ..., -4.1385e+00,
-1.4901e+00, -1.7157e+00],
[-1.5001e+00, 2.4988e-01, -4.6568e-01, ..., -8.2095e-01,
7.4223e-01, -7.9846e-01],
[ 1.2388e-01, -1.4941e-01, -7.4534e-01, ..., -1.4386e+00,
-8.3799e-01, -3.9617e-02]],
[[-1.6270e+00, 4.5838e-01, 1.0327e+00, ..., -2.4967e-01,
-1.7706e-01, 5.1899e-01],
[-7.1069e+00, -2.3288e+00, -2.3542e-01, ..., -9.4909e-01,
-3.8973e-01, 9.7827e-01],
[-1.2142e+01, -4.0935e+00, -1.2389e+00, ..., -7.0119e-01,
-7.0133e-01, 6.4667e-01],
...,
[-7.1026e+00, -3.7322e+00, -2.7754e+00, ..., -5.8595e+00,
-1.0969e-01, 2.6086e+00],
[-4.1931e+00, -1.8096e+00, 5.4437e-01, ..., -1.0516e+00,
5.5485e-02, 9.5068e-01],
[ 9.9791e-02, 5.7954e-01, 4.6132e-01, ..., 7.8065e-01,
2.7939e-01, 1.0596e+00]]],
[[[-1.7892e-01, -3.0288e-01, 1.4069e-01, ..., -8.4759e-02,
5.9999e-01, 2.3786e-01],
[-7.6341e-02, -1.3761e+00, -8.2211e-01, ..., -9.4215e-01,
-5.2889e-01, -2.6516e-01],
[-4.0375e-02, -1.1686e+00, -6.2075e-01, ..., -6.5909e-01,
-4.5174e-01, -6.8916e-01],
...,
[-1.7094e-01, -1.9069e+00, -1.4307e+00, ..., -1.1162e+00,
-1.1480e+00, -1.9275e+00],
[-3.7437e-01, -1.4667e+00, -6.5671e-01, ..., -4.0545e-01,
-3.9878e-01, -1.4072e+00],
[-1.6442e-01, -1.3166e+00, -1.0607e+00, ..., -8.7108e-01,
-1.1809e+00, -7.8273e-01]],
[[-2.6795e-01, -1.4291e+00, -7.2175e-01, ..., -1.0447e+00,
-1.4704e+00, -2.0329e-01],
[-1.8755e+00, -3.8007e+00, -2.5567e+00, ..., -2.7125e+00,
-3.5859e+00, -1.2231e+00],
[-1.8250e+00, -3.2986e+00, -2.8715e+00, ..., -2.5748e+00,
-3.0002e+00, -1.2773e+00],
...,
[-3.6744e+00, -5.1262e+00, -5.5958e+00, ..., -5.6171e+00,
-5.9987e+00, -2.8851e+00],
[-5.4714e+00, -6.6020e+00, -6.2641e+00, ..., -5.8152e+00,
-6.6741e+00, -2.7524e+00],
[ 5.8152e-01, -4.6604e-01, 6.8264e-01, ..., 4.7219e-01,
6.6495e-01, 9.4138e-02]],
[[-4.6875e-01, -8.3241e-02, 3.5274e-01, ..., 5.7074e-01,
1.8197e-01, 9.3637e-01],
[-2.9194e-01, -3.9841e-01, 5.5711e-01, ..., 2.8432e-02,
3.2720e-01, 4.7369e-01],
[-7.3186e-01, -2.8014e-01, 1.8666e-01, ..., -3.5238e-02,
1.5045e-01, 3.8379e-01],
...,
[-1.7279e+00, -4.6078e-01, -4.6064e-02, ..., -4.3470e-02,
-3.8713e-02, -1.2688e-01],
[-1.4174e+00, -3.6772e-01, -6.4299e-02, ..., -1.6522e-01,
-2.0051e-01, -2.8097e-01],
[-5.7396e-01, -1.0494e-01, 1.5108e-01, ..., -9.0642e-02,
-8.7200e-02, -4.5337e-01]],
...,
[[ 1.9719e-01, -4.9964e-01, -3.4834e-01, ..., -4.4449e-01,
-2.9022e-01, -7.8216e-01],
[-6.3367e-01, -5.7218e-01, -5.5499e-01, ..., -8.6568e-01,
-1.1555e+00, -8.1515e-01],
[-3.6734e-01, -1.2364e-01, -2.3852e-02, ..., 1.0631e-01,
-1.6474e-01, -5.3126e-01],
...,
[-9.9777e-01, -8.6697e-01, -2.2989e-01, ..., -3.5185e-01,
-6.3284e-01, -1.2427e+00],
[-2.5557e+00, -1.9846e+00, -1.1413e+00, ..., -1.3759e+00,
-1.5425e+00, -1.5883e+00],
[-5.4607e-01, -1.1096e+00, -2.1724e-02, ..., -3.8755e-01,
-3.9621e-01, -5.0379e-01]],
[[-1.3404e-01, 5.1071e-02, 3.9070e-01, ..., -4.9217e-02,
6.8205e-01, 2.1213e-01],
[-2.8085e-01, -1.9896e-01, -3.4081e-03, ..., -1.8113e-01,
7.9026e-02, -1.0439e-01],
[-5.2478e-01, -4.0427e-01, 1.4351e-02, ..., 4.7574e-02,
5.7261e-03, -6.2759e-02],
...,
[-1.5981e-01, -8.5539e-01, 3.8500e-01, ..., 5.7829e-01,
5.0404e-01, -3.1426e-01],
[-7.5816e-01, -1.9342e+00, 2.9577e-02, ..., 8.4943e-02,
1.0707e-01, -5.5651e-01],
[ 7.5563e-01, 3.7370e-02, 3.9253e-01, ..., 5.8886e-01,
4.9084e-01, 2.9454e-01]],
[[-5.9871e-01, 4.3497e-01, 5.2640e-01, ..., 8.3134e-01,
4.1198e-01, 6.4293e-01],
[-8.4645e-01, 2.7556e-01, -1.7668e-01, ..., 6.6284e-02,
-1.7280e-01, 6.8149e-01],
[-4.9189e-01, 3.2402e-01, 5.4566e-02, ..., 2.5804e-01,
1.2579e-01, 6.0219e-01],
...,
[-1.2942e+00, -6.2709e-01, -1.8869e-01, ..., -1.8136e-01,
-9.6204e-03, 1.4194e+00],
[-1.4045e+00, -7.9840e-01, -3.9591e-01, ..., -1.5328e-01,
7.8826e-02, 1.2162e+00],
[-6.3032e-01, -4.5445e-01, -1.2269e-01, ..., 3.8609e-02,
1.2617e-01, 1.2723e+00]]],
[[[ 1.8891e-02, -2.3854e-01, -2.8631e-02, ..., -1.1798e+00,
-6.2030e-01, -9.3408e-01],
[ 2.9263e-01, -9.4668e-01, -7.1908e-01, ..., -1.6147e+00,
-1.3010e+00, -1.2291e+00],
[ 2.4263e-01, -5.6885e-01, -4.9957e-01, ..., -1.3877e+00,
-1.2174e+00, -1.6185e+00],
...,
[-1.3308e-02, -1.4327e-01, -8.8635e-02, ..., -1.3595e+00,
-1.1770e+00, -1.6520e+00],
[-6.2966e-01, -6.0390e-01, 4.0414e-01, ..., -7.5354e-01,
-5.6876e-01, -1.4342e+00],
[-7.4336e-01, -1.3919e+00, -1.1931e+00, ..., -1.2932e+00,
-1.5023e+00, -9.0020e-01]],
[[-7.3390e-01, -2.1730e+00, -1.8340e+00, ..., -4.0081e+00,
-4.7454e+00, -1.7985e+00],
[-2.0466e+00, -3.7442e+00, -2.9117e+00, ..., -6.0865e+00,
-6.8269e+00, -2.7193e+00],
[-1.5860e+00, -2.5757e+00, -2.5292e+00, ..., -5.1780e+00,
-5.4390e+00, -2.3328e+00],
...,
[-1.8000e+00, -2.3815e+00, -1.7612e+00, ..., -5.1649e+00,
-5.4150e+00, -2.2990e+00],
[-3.8070e+00, -4.7056e+00, -3.9346e+00, ..., -6.2468e+00,
-7.2233e+00, -2.7821e+00],
[ 2.6786e-01, -3.4521e-01, 3.7484e-01, ..., 5.6792e-01,
6.8115e-01, 6.9255e-02]],
[[-2.2567e-01, 1.2860e-01, 7.1097e-01, ..., 9.4871e-01,
7.1367e-01, 1.5618e+00],
[ 8.5595e-02, -3.9263e-01, 5.1996e-01, ..., 2.9174e-01,
2.7563e-01, 2.8286e-01],
[-1.1174e-01, -2.0328e-01, 2.0309e-01, ..., -1.1542e-01,
1.5371e-01, 6.8745e-02],
...,
[-2.3825e-01, -2.8829e-02, -6.9337e-01, ..., -1.8206e-01,
1.0343e-01, 4.3417e-02],
[-2.6968e-01, 2.9957e-01, 2.7401e-01, ..., -6.7504e-02,
8.0654e-02, 1.3935e-01],
[-1.4425e-01, -4.8385e-01, 7.9804e-02, ..., -3.6338e-01,
-8.7943e-02, -2.7209e-01]],
...,
[[ 5.9998e-01, -2.7238e-01, -8.7629e-03, ..., -1.0034e-01,
-2.7107e-01, -1.4629e+00],
[-5.3267e-01, -5.4149e-01, -5.1769e-01, ..., -8.0412e-01,
-1.5610e+00, -1.6740e+00],
[-3.6346e-01, -2.1501e-01, -2.4492e-01, ..., -6.9985e-01,
-8.8494e-01, -1.2131e+00],
...,
[-4.5752e-01, 2.1484e-01, 3.2417e-01, ..., -6.8578e-01,
-9.2281e-01, -1.2260e+00],
[-2.3403e+00, -1.8715e+00, -1.8580e+00, ..., -1.8714e+00,
-1.9693e+00, -1.7721e+00],
[-4.8944e-01, -8.1849e-01, -4.0691e-01, ..., -7.1939e-01,
-7.5334e-01, -8.8227e-01]],
[[ 1.5345e-01, 2.4713e-01, 6.3534e-01, ..., 4.4237e-01,
7.7501e-01, 7.9883e-03],
[ 1.1987e-01, 2.4368e-01, 4.4730e-01, ..., 2.8970e-01,
4.1960e-01, -2.6174e-01],
[-1.1671e-01, 9.8796e-02, 2.7983e-01, ..., 3.3539e-01,
2.2297e-01, -2.8104e-01],
...,
[-2.3033e-01, -1.1557e-01, 1.6509e-01, ..., 3.3971e-01,
2.2503e-01, -2.9652e-01],
[-5.6192e-01, -7.1486e-01, 1.6142e-02, ..., -6.0780e-02,
-1.1877e-01, -5.9669e-01],
[ 4.8663e-01, 3.6105e-01, 5.8737e-02, ..., 2.1371e-01,
1.4294e-01, 3.0861e-01]],
[[-8.3442e-01, 2.6300e-01, 3.1479e-01, ..., 1.0367e-01,
2.1264e-01, 8.6602e-01],
[-1.1461e+00, 2.2687e-01, -3.3199e-01, ..., -8.2138e-02,
1.6908e-01, 1.3694e+00],
[-9.5072e-01, 3.5934e-01, -1.1412e-01, ..., 1.8125e-01,
3.7999e-01, 1.2174e+00],
...,
[-9.5469e-01, 8.6828e-02, -3.6166e-01, ..., 1.7779e-01,
3.7290e-01, 1.2636e+00],
[-1.6832e+00, -3.9261e-01, 2.6683e-01, ..., -1.6579e-01,
1.4265e-01, 8.2595e-01],
[-7.3921e-01, -9.5310e-02, -3.1661e-01, ..., 2.1320e-01,
1.1208e-01, 1.0586e+00]]]], device='cuda:0'),
tensor([[[[-7.3531e-01, -1.4906e+00, -1.4254e+00, ..., -1.2579e+00,
-8.2630e-01, -1.6295e+00],
[-3.3762e-01, -2.4690e+00, -2.2403e+00, ..., -1.7293e+00,
-1.8643e+00, -1.9031e+00],
[-1.2638e-01, -1.9273e+00, -1.7240e+00, ..., -1.2514e+00,
-1.3910e+00, -2.1098e+00],
...,
[-8.8846e-02, -1.0772e+00, -4.9995e-01, ..., -5.5735e-01,
-4.8109e-01, -9.3398e-01],
[-1.4523e-01, -9.7183e-01, -3.4880e-01, ..., -2.0826e-01,
-7.6560e-02, -9.5176e-01],
[-1.6358e-01, -1.0912e+00, -7.5098e-01, ..., -8.7978e-01,
-1.1255e+00, -6.2858e-01]],
[[-2.6956e+00, -5.1502e+00, -4.6994e+00, ..., -4.6937e+00,
-5.9699e+00, -2.3790e+00],
[-5.2828e+00, -8.0748e+00, -7.4816e+00, ..., -7.7097e+00,
-9.2315e+00, -4.3732e+00],
[-4.0003e+00, -5.7754e+00, -5.7021e+00, ..., -5.9882e+00,
-6.4761e+00, -3.2153e+00],
...,
[-2.0166e+00, -3.1343e+00, -3.6234e+00, ..., -2.9814e+00,
-3.5226e+00, -1.4912e+00],
[-4.5609e+00, -5.4836e+00, -5.4641e+00, ..., -5.0860e+00,
-6.2932e+00, -2.3864e+00],
[ 6.7884e-01, -2.9254e-01, 6.1301e-01, ..., 5.0062e-01,
5.6569e-01, 4.1516e-03]],
[[-6.7006e-01, 4.5893e-01, 8.3640e-01, ..., 8.1997e-01,
6.0986e-01, 1.9145e+00],
[-9.7226e-01, 1.8636e-02, 4.0580e-01, ..., 5.8529e-01,
-1.4711e-02, 1.3278e-01],
[-1.7610e+00, -2.3296e-01, -5.2177e-02, ..., 3.8131e-02,
4.4866e-03, -9.1476e-02],
...,
[-9.0769e-01, -1.3409e-01, 2.4997e-02, ..., 3.0693e-02,
1.0419e-01, 3.4550e-01],
[-9.5779e-01, 5.7552e-02, -8.5236e-02, ..., 1.0260e-03,
1.4464e-01, 4.0437e-01],
[-4.6392e-01, -2.9244e-02, 2.8016e-01, ..., -5.8748e-02,
7.2254e-02, 4.3053e-02]],
...,
[[ 1.2346e+00, -3.6891e-01, -3.9600e-02, ..., 2.1483e-01,
-1.0912e-01, -2.1291e+00],
[-7.5797e-01, -9.0386e-01, -4.4890e-01, ..., -5.6815e-01,
-1.5366e+00, -2.0631e+00],
[-7.4528e-01, -8.8998e-01, -2.7749e-01, ..., -3.2637e-01,
-5.7737e-01, -1.3614e+00],
...,
[-4.0787e-01, -1.3840e-01, 1.1803e-01, ..., 8.6575e-02,
-1.2039e-01, -5.4662e-01],
[-2.0314e+00, -1.4005e+00, -5.9012e-01, ..., -9.5161e-01,
-1.1466e+00, -1.1180e+00],
[-3.0216e-01, -7.8889e-01, -5.6376e-02, ..., -1.5390e-01,
-3.4065e-01, -6.2906e-01]],
[[ 4.6525e-02, 9.1233e-03, 4.1594e-01, ..., 4.6936e-01,
9.4026e-01, -1.1873e-02],
[ 2.2870e-01, -8.1491e-01, -4.5808e-02, ..., 4.4945e-01,
3.8750e-01, -8.3513e-01],
[ 1.3685e-01, -6.5138e-01, 2.8745e-01, ..., 6.1663e-01,
5.3858e-01, -3.9063e-01],
...,
[-3.9510e-01, -5.1213e-01, 2.3253e-01, ..., 9.4299e-02,
1.8246e-02, -1.2216e-01],
[-7.5942e-01, -1.5121e+00, -2.9820e-01, ..., -3.2221e-01,
-3.6850e-01, -6.9777e-01],
[ 5.1287e-01, 5.0513e-02, 1.1672e-01, ..., 2.1207e-01,
1.4276e-01, 3.3608e-01]],
[[-1.5479e+00, -7.9513e-01, -4.3179e-01, ..., -4.5148e-01,
-2.8605e-01, 8.7582e-01],
[-2.1197e+00, -7.6998e-01, -5.7278e-01, ..., -8.1222e-01,
-1.9262e-01, 1.6539e+00],
[-1.4714e+00, -5.8392e-01, -2.6503e-01, ..., -3.9140e-01,
-1.2713e-01, 1.3801e+00],
...,
[-4.8825e-01, 1.2358e-01, 9.2730e-02, ..., 1.2521e-01,
1.9407e-01, 6.9883e-01],
[-9.2012e-01, -3.1300e-01, -3.8373e-01, ..., -7.6322e-02,
8.7960e-02, 4.2252e-01],
[-4.3766e-01, -3.3349e-01, -3.5261e-01, ..., -2.5136e-01,
-2.4820e-01, 5.6736e-01]]],
[[[-6.1674e-01, -1.4167e+00, -1.3812e+00, ..., -1.2599e+00,
-8.8607e-01, -1.7560e+00],
[-2.3135e-01, -2.5092e+00, -2.0509e+00, ..., -1.7332e+00,
-1.8433e+00, -1.9200e+00],
[-1.3390e-01, -2.0154e+00, -1.5174e+00, ..., -1.3222e+00,
-1.4132e+00, -2.1209e+00],
...,
[ 7.2463e-02, -5.4928e-01, -2.3471e-01, ..., -4.0124e-01,
-4.5159e-02, -4.3356e-01],
[-3.2141e-01, -4.2659e-01, 4.9772e-02, ..., 1.1149e-01,
2.9580e-01, -1.5530e-01],
[-6.2294e-01, -1.3176e+00, -1.1254e+00, ..., -1.1074e+00,
-1.1784e+00, -5.0719e-01]],
[[-2.5163e+00, -4.8944e+00, -5.1798e+00, ..., -5.0346e+00,
-6.1653e+00, -2.5673e+00],
[-5.5818e+00, -8.3814e+00, -7.7597e+00, ..., -7.6869e+00,
-9.1126e+00, -4.3644e+00],
[-3.9959e+00, -5.8540e+00, -5.8086e+00, ..., -6.6149e+00,
-6.4436e+00, -3.2430e+00],
...,
[-1.7453e+00, -2.1831e+00, -1.7515e+00, ..., -2.2230e+00,
-2.2922e+00, -9.3392e-01],
[-2.8021e+00, -2.9659e+00, -2.0225e+00, ..., -2.2775e+00,
-3.3884e+00, -1.0086e+00],
[ 3.6404e-01, -2.2950e-01, 3.1561e-01, ..., 2.8927e-01,
4.4141e-01, 2.0805e-01]],
[[-6.6021e-01, 3.5208e-01, 8.3628e-01, ..., 7.2872e-01,
6.5541e-01, 1.8478e+00],
[-9.3504e-01, -1.1694e-01, 3.4765e-01, ..., 4.4548e-01,
-2.8259e-02, 1.0435e-01],
[-1.7665e+00, -3.6307e-01, 3.2281e-01, ..., 1.6811e-01,
-9.9817e-02, -8.9138e-02],
...,
[-6.2244e-01, 4.1023e-02, 2.3201e-01, ..., 2.6322e-01,
2.5297e-01, 4.1067e-01],
[-4.3619e-01, 1.0517e-01, 2.1855e-01, ..., 1.7675e-01,
2.1997e-01, 5.6145e-01],
[-1.7889e-01, -2.5903e-01, 1.4343e-01, ..., -1.9467e-02,
1.0227e-01, 3.4610e-01]],
...,
[[ 1.0933e+00, -4.3123e-01, 1.1307e-01, ..., 3.0218e-01,
-2.9886e-02, -2.0646e+00],
[-8.8141e-01, -8.8852e-01, -3.2595e-01, ..., -5.2211e-01,
-1.4231e+00, -2.0503e+00],
[-7.1059e-01, -7.0157e-01, -2.9437e-01, ..., -2.5550e-01,
-5.9478e-01, -1.3939e+00],
...,
[-6.4655e-01, -5.2731e-01, -3.5621e-01, ..., -6.2663e-01,
-5.0506e-01, -5.5270e-01],
[-2.2075e+00, -1.7665e+00, -1.3077e+00, ..., -1.3350e+00,
-1.3754e+00, -1.0905e+00],
[-6.0671e-01, -8.1480e-01, -4.3185e-01, ..., -2.5188e-01,
-3.6968e-01, -3.9088e-01]],
[[-4.6740e-02, -6.1115e-02, 6.7911e-01, ..., 5.3630e-01,
8.9148e-01, -7.4789e-02],
[ 8.9723e-02, -8.4983e-01, -9.3573e-02, ..., 4.9223e-01,
4.2437e-01, -8.8108e-01],
[ 2.2737e-02, -7.3014e-01, 2.1823e-01, ..., 4.1402e-01,
5.3690e-01, -4.5534e-01],
...,
[-2.3341e-01, -4.6751e-01, 2.1602e-01, ..., 1.7697e-01,
3.0532e-01, 4.2864e-02],
[-5.1685e-01, -9.0827e-01, 5.0509e-02, ..., -1.0647e-02,
-4.1565e-02, -1.5995e-01],
[ 3.3653e-01, -9.1205e-03, -1.9504e-01, ..., -1.6040e-01,
-1.1228e-01, 1.7472e-01]],
[[-1.5998e+00, -8.0896e-01, -7.3888e-01, ..., -4.6480e-01,
-2.6536e-01, 8.5811e-01],
[-2.0722e+00, -7.2488e-01, -9.5772e-01, ..., -7.9478e-01,
-1.5214e-01, 1.6225e+00],
[-1.4821e+00, -6.2300e-01, -5.8823e-01, ..., -1.5911e-01,
-7.2818e-02, 1.3864e+00],
...,
[-6.8883e-01, -2.2116e-01, -1.2487e-01, ..., 2.0233e-01,
-1.4903e-02, 6.0597e-01],
[-1.1229e+00, -5.5444e-01, -3.2004e-01, ..., -3.0918e-01,
-2.1549e-01, 1.5948e-01],
[-3.7419e-01, 1.1284e-02, -1.5588e-01, ..., -2.3745e-01,
-2.3534e-01, 3.2863e-01]]],
[[[-2.8474e-01, -4.4775e-01, -9.7328e-02, ..., -9.6055e-01,
-4.9037e-01, -1.4062e+00],
[ 3.2180e-02, -1.5899e+00, -7.7689e-01, ..., -1.1402e+00,
-1.2456e+00, -1.4545e+00],
[-2.6488e-01, -1.6257e+00, -5.6863e-01, ..., -8.8887e-01,
-1.0432e+00, -1.8150e+00],
...,
[-6.0088e-01, -1.2952e+00, -1.0202e+00, ..., -1.4112e+00,
-1.0626e+00, -1.9166e+00],
[-2.2582e-01, -9.4617e-01, -1.3109e-01, ..., -8.1188e-01,
-5.5278e-01, -1.7206e+00],
[-3.3186e-01, -1.4329e+00, -1.0831e+00, ..., -1.2814e+00,
-1.4951e+00, -7.8264e-01]],
[[-5.8269e-01, -1.9732e+00, -6.4984e-01, ..., -3.9305e+00,
-5.4269e+00, -1.8396e+00],
[-2.4511e+00, -4.5536e+00, -2.1514e+00, ..., -5.8626e+00,
-8.0599e+00, -3.4255e+00],
[-1.9789e+00, -3.4763e+00, -1.6301e+00, ..., -5.0231e+00,
-6.2590e+00, -2.6149e+00],
...,
[-2.0612e+00, -3.6709e+00, -2.5095e+00, ..., -5.9674e+00,
-6.3082e+00, -2.5293e+00],
[-3.4915e+00, -4.1530e+00, -2.5631e+00, ..., -6.7247e+00,
-7.7865e+00, -2.5733e+00],
[ 6.1366e-01, -6.2931e-01, 4.4561e-01, ..., 3.5327e-01,
4.9221e-01, 1.7921e-01]],
[[-4.6534e-01, 6.6076e-01, 1.0737e+00, ..., 6.3137e-01,
2.9633e-01, 1.5512e+00],
[-8.4524e-01, 2.0682e-01, 5.8201e-01, ..., 6.0421e-01,
-6.0433e-02, 1.3850e-01],
[-1.5360e+00, 2.8968e-01, 2.0179e-01, ..., 2.8870e-02,
-8.2822e-02, -1.8096e-01],
...,
[-1.2991e+00, -1.2295e+00, -7.3254e-01, ..., 5.1782e-01,
-1.0149e-01, -1.4372e-01],
[-1.1546e+00, 2.8504e-01, 1.8209e-01, ..., 1.7512e-01,
-9.7486e-02, -2.0276e-01],
[-8.5589e-01, 7.8727e-02, -8.9600e-02, ..., 9.3949e-02,
1.3327e-01, -6.4257e-01]],
...,
[[ 2.5471e-01, -2.5196e-01, -1.5999e-01, ..., 4.3892e-01,
3.1874e-01, -1.3332e+00],
[-1.6119e+00, -1.1560e+00, -7.6757e-01, ..., -1.2602e-01,
-7.8806e-01, -1.3646e+00],
[-8.0562e-01, -3.4737e-01, -9.0334e-02, ..., -1.0835e-01,
-1.1089e-01, -8.9009e-01],
...,
[-1.3349e+00, -3.0429e+00, -1.1497e+00, ..., -3.7101e-01,
-1.0679e-01, -9.8095e-01],
[-2.6535e+00, -1.8871e+00, -8.0425e-01, ..., -1.9894e+00,
-1.5606e+00, -1.8190e+00],
[-6.3967e-01, -9.9777e-01, -8.6699e-02, ..., -2.9241e-01,
-5.0507e-01, -1.1469e+00]],
[[ 7.5054e-03, 2.4829e-01, 5.5007e-01, ..., -9.9065e-02,
3.8801e-01, -3.7883e-01],
[-2.6850e-01, -7.0498e-01, -1.5073e-01, ..., 2.4211e-01,
1.1964e-01, -8.3031e-01],
[-6.7763e-01, -1.3468e+00, -2.8046e-01, ..., 2.6857e-01,
1.1162e-01, -5.0614e-01],
...,
[-1.5086e+00, -1.0271e+00, -3.3454e-01, ..., 9.3489e-02,
1.6258e-01, -4.8797e-01],
[-1.0337e+00, -2.0029e+00, -4.4809e-01, ..., -4.2689e-01,
-2.2839e-01, -8.5667e-01],
[ 2.1990e-01, -4.9273e-01, -2.4442e-01, ..., 1.7812e-01,
3.2621e-01, 2.9892e-01]],
[[-4.7695e-01, 1.1462e-01, 5.8072e-01, ..., -8.1470e-02,
-8.0063e-02, 5.8754e-01],
[-3.6015e-01, 4.5453e-02, -1.5671e-02, ..., -9.4419e-01,
-4.1633e-01, 9.1655e-01],
[-2.1950e-02, 6.5929e-02, 7.7436e-02, ..., -5.9450e-01,
-5.1466e-01, 6.3791e-01],
...,
[-4.3906e-01, -1.7784e-01, 1.0098e+00, ..., 3.2751e-01,
-5.1778e-01, 6.0221e-01],
[-2.9930e-01, -2.9933e-01, 6.4089e-02, ..., 1.6490e-02,
-6.4556e-01, 4.2654e-01],
[ 8.2885e-02, -4.3769e-01, -1.1186e-01, ..., -1.8189e-01,
-4.0757e-01, 1.0246e+00]]],
[[[-4.2846e-01, -1.2958e+00, -1.3437e+00, ..., -3.3134e+00,
-1.0700e+00, 7.6703e-01],
[-3.9613e-02, -2.3852e+00, -2.2989e+00, ..., -7.9965e+00,
-1.9610e+00, -1.0676e+00],
[ 2.8682e-02, -1.3072e+00, -1.6057e+00, ..., -7.8996e+00,
-2.9486e+00, -1.9691e+00],
...,
[-7.0943e+00, -1.0905e+01, -2.2823e+01, ..., -1.2832e+00,
-4.0759e-01, -5.7873e-01],
[-2.8725e+00, -7.5863e+00, -2.1005e+01, ..., -2.9271e-01,
9.0357e-02, -2.2757e-01],
[-5.7924e+00, -8.8267e+00, -2.0223e+01, ..., -1.5013e+00,
-1.4231e+00, -5.0683e-01]],
[[-2.3287e+00, -4.5074e+00, -4.4051e+00, ..., -1.4531e+01,
-4.8137e+00, -4.7730e+00],
[-5.0641e+00, -7.6030e+00, -7.2667e+00, ..., -8.2197e+00,
-2.7555e+00, -5.6402e+00],
[-4.3375e+00, -5.8943e+00, -6.4539e+00, ..., -3.6092e+00,
-1.1325e+00, -4.9272e+00],
...,
[-1.0773e+01, 2.1519e+00, -1.9823e+00, ..., -4.8465e+00,
-3.8940e+00, -1.2754e+00],
[-1.3394e+01, -8.6745e+00, -1.6403e+01, ..., -4.9755e+00,
-5.0313e+00, -1.2074e+00],
[-2.7472e+00, -2.7938e+00, -2.2645e+00, ..., -1.1992e-01,
-6.6619e-03, -5.8712e-02]],
[[-6.8997e-01, 2.4382e-01, 8.6606e-01, ..., -8.9581e+00,
9.4916e-01, 3.2580e+00],
[-8.4996e-01, -8.8246e-02, 3.4918e-01, ..., -1.0077e+01,
-6.2718e+00, 3.8798e+00],
[-1.2925e+00, -1.4098e-01, -6.1409e-01, ..., -6.4879e+00,
-7.4644e+00, 3.3441e+00],
...,
[-2.7599e+01, -3.9854e+01, -1.5534e+01, ..., 6.2173e-01,
-3.9130e-01, 3.3319e-01],
[-2.1243e+01, -2.5384e+01, -1.3912e+01, ..., 5.5071e-02,
-1.0216e+00, -4.9612e-01],
[-1.6648e+01, -1.6846e+01, -8.6741e+00, ..., -2.1719e-01,
-6.7801e-01, -3.8102e-01]],
...,
[[ 7.7959e-01, 3.0133e-02, 4.1161e-01, ..., -8.1448e+00,
-6.8919e+00, -3.1181e+00],
[-6.5840e-01, -8.4058e-02, -2.7858e-02, ..., -1.2260e+01,
-1.1046e+01, -4.4793e+00],
[-9.6788e-02, -1.3026e-02, 3.6464e-01, ..., -1.0145e+01,
-7.3100e+00, -3.7797e+00],
...,
[-7.2071e-01, -4.6981e+00, -5.8485e+00, ..., -6.1430e-01,
-4.8589e-01, -1.0014e+00],
[-2.0282e+00, -1.0704e+01, -1.3651e+01, ..., -1.7806e+00,
-1.5149e+00, -1.4172e+00],
[-2.0473e-01, -3.2084e+00, -2.7134e+00, ..., -6.2967e-01,
-1.1517e+00, -8.6254e-01]],
[[-9.4606e-02, -1.5544e-01, 2.6056e-02, ..., -1.9541e+00,
6.5042e-01, 9.1212e-01],
[ 1.1876e-02, -8.4980e-01, 4.5215e-02, ..., 6.8405e-03,
3.1705e+00, -1.3026e+00],
[-4.1743e-02, -3.9906e-01, 2.7396e-01, ..., 1.5644e+00,
3.1325e+00, -1.6611e+00],
...,
[-7.2418e+00, -4.3846e+00, -2.7608e+00, ..., -6.8481e-01,
-2.0220e-01, 2.9426e-01],
[-4.5602e+00, -3.8216e+00, -4.9554e+00, ..., -1.0323e+00,
-8.1480e-01, -3.6834e-01],
[-5.6472e+00, -8.2945e+00, -9.5291e+00, ..., -3.6700e-01,
-6.5038e-01, -7.5410e-02]],
[[-8.8157e-01, -3.8471e-01, -1.3195e-01, ..., -1.8872e+01,
-1.6653e+00, -5.9744e-01],
[-1.6551e+00, -1.4272e+00, -1.1489e+00, ..., -1.9073e+01,
1.5909e+00, 9.1693e-01],
[-1.5751e+00, -1.5733e+00, -1.0125e+00, ..., -1.5089e+01,
1.1094e+00, 5.8096e-01],
...,
[-1.0504e+01, -1.7730e+01, -3.1481e+01, ..., 5.0001e-02,
-4.6310e-01, 8.2611e-02],
[-6.4380e-01, -7.5699e+00, -2.4922e+01, ..., 3.5228e-01,
4.2145e-02, 2.2461e-01],
[ 6.6685e-01, -1.4756e+00, -1.3261e+01, ..., -8.7108e-02,
-2.3175e-01, 4.2902e-01]]],
[[[ 3.1130e-02, -2.2257e-01, 7.3496e-04, ..., -1.1685e+00,
-5.6262e-01, -8.3561e-01],
[ 3.1732e-01, -9.3076e-01, -7.0990e-01, ..., -1.5786e+00,
-1.2733e+00, -1.1377e+00],
[ 2.2623e-01, -5.5848e-01, -4.8703e-01, ..., -1.3560e+00,
-1.1845e+00, -1.6069e+00],
...,
[-2.8453e-02, -6.1160e-01, -2.1650e-01, ..., -1.1438e+00,
-1.2523e+00, -1.9419e+00],
[-6.6319e-01, -6.9700e-01, 3.7537e-01, ..., -6.2763e-01,
-4.4417e-01, -1.5037e+00],
[-7.5767e-01, -1.3785e+00, -1.0657e+00, ..., -1.1346e+00,
-1.3085e+00, -7.9707e-01]],
[[-7.2357e-01, -2.0970e+00, -1.7748e+00, ..., -3.9347e+00,
-4.7145e+00, -1.7236e+00],
[-2.0566e+00, -3.7253e+00, -2.9504e+00, ..., -6.1529e+00,
-6.9815e+00, -2.8233e+00],
[-1.6341e+00, -2.5540e+00, -2.5203e+00, ..., -5.2018e+00,
-5.4171e+00, -2.3257e+00],
...,
[-1.7575e+00, -2.4174e+00, -3.2741e+00, ..., -5.1705e+00,
-5.3740e+00, -2.9959e+00],
[-3.7940e+00, -4.4245e+00, -3.7729e+00, ..., -5.9880e+00,
-6.9332e+00, -2.8411e+00],
[ 2.5232e-01, -3.8136e-01, 3.4734e-01, ..., 5.4061e-01,
6.4788e-01, 7.3174e-02]],
[[-2.2216e-01, 1.4098e-01, 7.1528e-01, ..., 9.9248e-01,
6.7480e-01, 1.5563e+00],
[ 7.3127e-02, -3.8363e-01, 5.7128e-01, ..., 3.6379e-01,
3.4008e-01, 3.0776e-01],
[-1.0322e-01, -2.2084e-01, 2.4641e-01, ..., -9.7777e-02,
1.7166e-01, 7.8514e-02],
...,
[-2.7222e-01, -1.9890e-01, 7.3733e-01, ..., 1.9695e-01,
-1.6098e+00, 7.2088e-01],
[-2.7957e-01, -1.8893e-01, 8.3386e-01, ..., 3.4814e-01,
2.0208e-02, 7.2764e-02],
[-1.5676e-01, -4.9999e-01, 4.7993e-01, ..., -3.6570e-01,
-1.0857e-01, -2.5408e-01]],
...,
[[ 6.0196e-01, -2.7777e-01, -2.3707e-02, ..., -1.1157e-01,
-3.3552e-01, -1.5414e+00],
[-5.2818e-01, -6.0222e-01, -5.6535e-01, ..., -8.3295e-01,
-1.6924e+00, -1.7820e+00],
[-3.6072e-01, -2.2731e-01, -2.6183e-01, ..., -6.6747e-01,
-8.7650e-01, -1.2389e+00],
...,
[-4.5428e-01, -1.3284e-01, 1.8519e-01, ..., -4.8526e-01,
2.7979e-01, -1.8744e+00],
[-2.3608e+00, -1.7453e+00, -1.3328e+00, ..., -1.9873e+00,
-1.9815e+00, -2.1094e+00],
[-5.0735e-01, -8.4799e-01, -5.0874e-01, ..., -7.0014e-01,
-7.3907e-01, -8.1250e-01]],
[[ 1.5348e-01, 2.5591e-01, 6.5878e-01, ..., 4.3219e-01,
7.7724e-01, 3.2489e-02],
[ 1.4280e-01, 2.4901e-01, 4.4310e-01, ..., 3.4249e-01,
4.4302e-01, -2.7940e-01],
[-1.2535e-01, 1.0482e-01, 2.6460e-01, ..., 3.6509e-01,
2.3735e-01, -3.1055e-01],
...,
[-2.2018e-01, 1.8001e-02, 7.6704e-01, ..., 4.6122e-01,
-3.7573e-01, -7.3417e-01],
[-5.7561e-01, -6.9851e-01, 3.0220e-01, ..., 1.3729e-01,
-1.0766e-01, -7.1823e-01],
[ 4.8730e-01, 3.7444e-01, 1.2415e-01, ..., 2.7147e-01,
2.0359e-01, 3.2939e-01]],
[[-8.5272e-01, 2.5673e-01, 3.3326e-01, ..., 1.8297e-01,
2.4572e-01, 8.9551e-01],
[-1.1580e+00, 2.1090e-01, -3.7597e-01, ..., -6.7527e-02,
1.8187e-01, 1.4352e+00],
[-9.8438e-01, 3.7641e-01, -1.1617e-01, ..., 1.9372e-01,
3.6288e-01, 1.2080e+00],
...,
[-9.4587e-01, 1.6873e-01, -7.8427e-01, ..., -1.0763e-01,
1.0122e+00, 3.2847e-01],
[-1.6980e+00, -3.9779e-01, -5.2550e-01, ..., -2.2959e-01,
3.1002e-01, 1.0591e+00],
[-7.3195e-01, -6.8043e-02, -3.6127e-01, ..., 2.8362e-01,
1.9633e-01, 1.0764e+00]]]], device='cuda:0'),
tensor([[[[-7.2152e-01, -1.4792e+00, -1.4756e+00, ..., -3.4853e+00,
-1.4613e+00, 1.2277e+00],
[-3.3013e-01, -2.4022e+00, -2.3598e+00, ..., -6.5112e-01,
-7.7520e-01, 6.4816e-01],
[-1.3260e-01, -1.8422e+00, -1.9521e+00, ..., -1.9639e+00,
-1.6834e+00, -3.4513e-01],
...,
[ 7.7301e-01, -1.0219e+00, -5.4030e+00, ..., -1.1499e-01,
-3.3583e-02, -4.4624e-01],
[ 2.1462e-01, 8.7491e-01, -3.3845e+00, ..., 1.3960e-01,
2.5970e-01, -1.8415e-01],
[-5.2595e-01, 7.3855e-01, 7.7702e-01, ..., -1.0341e+00,
-1.1268e+00, -4.6932e-01]],
[[-2.7291e+00, -5.1649e+00, -4.7083e+00, ..., -1.1558e+01,
-1.0473e+01, -2.9628e+00],
[-5.3812e+00, -8.3219e+00, -8.2235e+00, ..., -9.9034e+00,
-1.2295e+01, -3.2833e+00],
[-4.0625e+00, -6.0734e+00, -6.5047e+00, ..., -1.2452e+01,
-1.1020e+01, -1.1475e+00],
...,
[-3.4726e+00, -9.0508e+00, -1.3540e+01, ..., -1.8888e+00,
-2.3099e+00, -9.1671e-01],
[-2.4540e+00, -6.8379e+00, -1.1589e+01, ..., -2.2510e+00,
-3.3276e+00, -9.9398e-01],
[ 9.6786e-02, -1.3711e-01, -2.6679e-01, ..., 3.0994e-01,
4.4641e-01, 2.0264e-01]],
[[-6.5820e-01, 4.4169e-01, 8.3331e-01, ..., -1.5447e+00,
-1.7595e+00, 2.0515e+00],
[-9.3529e-01, 4.1974e-02, 3.3365e-01, ..., -1.0368e+01,
-1.0190e+01, -2.7568e+00],
[-1.7528e+00, -1.9323e-01, -3.7418e-01, ..., -1.1185e+01,
-7.0355e+00, -1.8210e+00],
...,
[ 1.3838e-01, -5.2586e+00, -1.6618e+01, ..., 1.7523e-01,
2.7374e-01, 4.1888e-01],
[-2.5078e-01, -5.0980e+00, -1.5477e+01, ..., 1.7356e-01,
2.1363e-01, 5.5012e-01],
[ 3.6720e-01, -8.5817e-01, -7.2539e+00, ..., 1.2926e-02,
1.0633e-01, 3.3684e-01]],
...,
[[ 1.2347e+00, -3.7308e-01, -2.5016e-02, ..., 2.6864e-01,
-1.3841e+00, -4.6295e+00],
[-7.1719e-01, -8.9871e-01, -4.1228e-01, ..., -6.7754e+00,
-7.8282e+00, -7.2787e+00],
[-7.2604e-01, -8.2681e-01, 1.2450e-01, ..., -7.6415e+00,
-5.5483e+00, -5.8890e+00],
...,
[-3.2214e-01, -3.4036e-01, -4.9845e+00, ..., -3.3145e-01,
-5.1255e-01, -5.6641e-01],
[-1.7804e+00, 7.2375e-01, 3.5597e-01, ..., -1.3048e+00,
-1.3512e+00, -1.0839e+00],
[-1.3303e+00, -4.1251e-01, 1.8819e+00, ..., -2.2731e-01,
-3.5329e-01, -3.8047e-01]],
[[ 4.1674e-02, 4.9555e-03, 4.5994e-01, ..., -3.4465e-01,
-2.2646e-01, 5.0935e-01],
[ 2.4089e-01, -7.3811e-01, 8.2286e-02, ..., -1.8491e+00,
-2.7920e+00, -1.6314e+00],
[ 1.7902e-01, -5.5924e-01, 2.9005e-01, ..., -5.0204e+00,
-6.9667e+00, -2.7330e+00],
...,
[ 7.0134e-01, -1.1047e+00, -7.2971e+00, ..., 3.3766e-01,
3.4534e-01, 6.8358e-02],
[-1.7942e-01, -1.9591e+00, -5.6657e+00, ..., 2.3702e-02,
-2.0630e-02, -1.4856e-01],
[ 5.5294e-01, -2.0042e-01, -2.2370e+00, ..., -1.0335e-01,
-7.6257e-02, 1.8133e-01]],
[[-1.5708e+00, -8.1384e-01, -4.7292e-01, ..., -3.4856e+00,
-3.0528e+00, -8.7849e-01],
[-2.1616e+00, -7.9843e-01, -6.3183e-01, ..., -4.9448e+00,
-4.8011e+00, -3.1857e+00],
[-1.5184e+00, -6.0997e-01, -5.5671e-01, ..., -5.1202e+00,
-5.3127e+00, -4.1685e+00],
...,
[-1.2865e+00, -1.1065e+00, -4.0937e+00, ..., -1.5280e-01,
1.3393e-03, 6.3611e-01],
[-1.7053e+00, 2.6088e-01, -5.8516e-01, ..., -3.1159e-01,
-1.7126e-01, 1.9003e-01],
[-7.0005e-01, 3.2484e-01, 2.7434e+00, ..., -2.1161e-01,
-1.8197e-01, 3.4349e-01]]],
[[[-5.9890e-01, -1.0253e+00, -3.3063e-01, ..., -9.9324e-01,
-7.2740e-01, 1.0697e+00],
[-2.8263e-01, -2.0199e+00, -1.2106e+00, ..., -1.5708e+01,
-1.0958e+01, -4.3172e-01],
[-1.5247e-02, -1.3139e+00, -9.7068e-01, ..., -1.7084e+01,
-4.4809e+00, 5.3102e+00],
...,
[ 1.6808e+00, -1.0024e+01, -9.4434e+00, ..., -1.7400e+00,
-1.7862e+00, -2.5881e+00],
[ 5.8652e-01, -1.0332e+01, -1.3513e+01, ..., -7.4356e-01,
-8.3351e-01, -1.9089e+00],
[ 1.1947e-01, -9.2379e-01, -6.1244e+00, ..., -1.6889e+00,
-1.8410e+00, -1.0739e+00]],
[[-2.6557e+00, -4.8407e+00, -4.0576e+00, ..., -7.7856e+00,
-1.8426e+00, 3.9207e+00],
[-4.7755e+00, -8.0749e+00, -7.9991e+00, ..., -1.6639e+01,
-7.4268e+00, 7.1532e+00],
[-3.5247e+00, -5.6591e+00, -7.4921e+00, ..., -1.8357e+01,
-9.0597e+00, 5.0083e+00],
...,
[-8.4750e+00, -1.4546e+01, -1.6119e+01, ..., -6.5909e+00,
-6.8064e+00, -3.3943e+00],
[-7.4533e+00, -1.1937e+01, -1.9643e+01, ..., -7.6643e+00,
-8.2021e+00, -3.5416e+00],
[ 1.6858e+00, -2.2885e+00, -7.7782e+00, ..., -4.0388e-02,
2.5962e-01, -5.6411e-02]],
[[-4.9385e-01, -5.2009e-02, 4.9620e-01, ..., -2.2817e+01,
-2.4656e+01, -5.4811e+00],
[-8.6888e-01, -1.2535e+00, 2.5098e-01, ..., -1.5862e+01,
-1.5801e+01, -4.8676e+00],
[-1.3893e+00, -1.6304e+00, -2.0822e-01, ..., -9.4452e+00,
-1.5490e+01, -6.5800e+00],
...,
[ 1.1648e+00, -1.4351e+01, -2.5789e+01, ..., -4.2912e-01,
2.4059e-01, 2.0430e-01],
[ 2.3934e+00, -2.0206e+01, -2.8087e+01, ..., -6.0110e-01,
1.0222e-01, -2.1014e-01],
[ 3.4377e+00, -1.5505e+01, -2.1840e+01, ..., -9.9970e-01,
-4.0679e-01, -4.6164e-01]],
...,
[[ 7.4313e-01, 4.1909e-01, 6.3733e-01, ..., -2.3004e+01,
-1.5414e+01, -1.1117e+01],
[-4.0759e-01, -4.4279e-01, -2.7240e-02, ..., -1.9482e+01,
-1.2296e+01, -1.0382e+01],
[ 1.0664e-01, 2.0910e-01, 1.9891e-02, ..., -1.5232e+01,
-1.4009e+01, -1.3289e+01],
...,
[ 1.5489e+00, 8.8239e-01, -7.5285e+00, ..., -6.7151e-01,
-9.2077e-01, -2.2477e+00],
[-1.0142e+00, -2.3042e+00, -1.2504e+01, ..., -2.4817e+00,
-2.6699e+00, -2.3848e+00],
[-1.0682e-01, -7.0905e-01, -1.2535e+00, ..., -1.4319e+00,
-1.4306e+00, -1.3083e+00]],
[[-9.7339e-03, -9.4892e-02, 2.6916e-01, ..., -8.1109e+00,
-6.1590e+00, -4.0835e+00],
[ 1.6732e-01, -2.6178e-01, -2.3813e-01, ..., -6.7653e+00,
-2.2336e+00, 1.4690e-01],
[ 6.2382e-02, 1.1523e-01, 1.2307e-01, ..., -8.2154e+00,
-8.1519e+00, -1.5752e+00],
...,
[ 3.1278e+00, 1.6153e-01, -8.5905e+00, ..., 4.5704e-01,
4.3788e-01, -2.8255e-01],
[ 2.0583e+00, -6.3502e+00, -1.2295e+01, ..., 7.0402e-02,
2.7926e-02, -8.0731e-01],
[ 2.9863e+00, -1.5911e+00, -6.7535e+00, ..., -8.1347e-02,
1.5466e-01, 3.0107e-01]],
[[-1.0103e+00, -2.8246e-01, -9.5902e-02, ..., -1.5062e+01,
-1.8319e+01, -1.3144e+01],
[-1.5648e+00, -5.1214e-01, -1.1885e+00, ..., -7.2107e+00,
-1.1554e+01, -1.7217e+01],
[-1.4839e+00, -1.1454e+00, -1.5336e+00, ..., -1.4307e+01,
-9.8110e+00, -1.6288e+01],
...,
[-2.7137e+00, -3.3332e+00, -1.2752e+01, ..., 1.0731e-01,
6.4112e-01, 1.5207e+00],
[-1.6225e+00, -5.7755e-01, -1.5536e+01, ..., -1.1606e-01,
5.6034e-01, 1.4867e+00],
[-1.1373e+00, 5.2499e+00, -2.2183e+00, ..., 1.7673e-01,
6.3609e-01, 1.5945e+00]]],
[[[ 3.4573e-02, -2.4152e-01, -1.4794e-02, ..., -1.1729e+00,
-6.3510e-01, -1.0144e+00],
[ 2.7961e-01, -1.1554e+00, -7.6669e-01, ..., -2.4181e+00,
-2.1701e+00, -2.3569e+00],
[-8.8983e-02, -1.5863e+00, -1.3790e+00, ..., -2.8912e+00,
-1.3207e+00, -8.2184e-01],
...,
[ 8.8999e-01, -7.4622e-02, -4.1046e-01, ..., -7.2985e-01,
-1.3188e+00, -1.7125e+00],
[-5.1182e-01, -6.3627e-01, -9.0998e-02, ..., -6.9025e-01,
-7.6168e-01, -1.4609e+00],
[-6.1580e-01, -1.2158e+00, -1.0534e+00, ..., -1.3039e+00,
-1.4963e+00, -9.1236e-01]],
[[-7.8650e-01, -2.2487e+00, -1.8425e+00, ..., -4.0741e+00,
-4.8370e+00, -1.8194e+00],
[-2.4035e+00, -4.3211e+00, -3.1549e+00, ..., -1.0318e+01,
-1.1785e+01, -4.7969e+00],
[-5.8064e+00, -5.3343e+00, -6.0111e+00, ..., -1.2223e+01,
-1.2820e+01, -5.4394e+00],
...,
[-1.5631e+00, -2.1790e+00, -2.6522e+00, ..., -5.5211e+00,
-5.8130e+00, -2.3916e+00],
[-3.6217e+00, -4.1176e+00, -3.4889e+00, ..., -6.1784e+00,
-7.2335e+00, -2.8159e+00],
[ 2.9179e-01, -2.6749e-01, 3.9265e-01, ..., 5.3107e-01,
6.4279e-01, 5.3444e-02]],
[[-2.7727e-01, 8.3377e-02, 8.0483e-01, ..., 9.7511e-01,
7.1701e-01, 1.5369e+00],
[ 1.0063e-01, -4.2776e-01, 6.4109e-01, ..., 3.7507e-01,
9.0865e-01, 1.1709e+00],
[-7.5080e-01, -8.0071e-01, 1.0209e-01, ..., -1.2985e+00,
-7.8439e-01, 5.4810e-01],
...,
[ 2.9228e-01, -1.9426e-01, 6.5045e-01, ..., -1.2957e+00,
2.0062e-01, -4.1365e-02],
[-1.7822e-01, -9.2306e-02, 1.5800e-01, ..., -2.4891e-01,
2.6866e-01, 1.6394e-01],
[-7.2455e-02, -3.9327e-01, 1.4754e-01, ..., -3.7908e-01,
-1.1903e-01, -2.9378e-01]],
...,
[[ 4.9414e-01, -3.6122e-01, -1.6896e-02, ..., -9.8633e-02,
-2.9844e-01, -1.5117e+00],
[-7.3299e-01, -6.1116e-01, -5.9716e-01, ..., -4.1439e-01,
-8.6802e-01, -1.3807e+00],
[-1.2993e+00, -8.9240e-01, 7.6256e-02, ..., -3.3175e-01,
-1.3362e+00, -3.4321e+00],
...,
[-1.5355e+00, -1.5916e+00, -1.2312e+00, ..., -1.9925e+00,
-1.1654e+00, -1.2422e+00],
[-2.1458e+00, -1.5485e+00, -1.1383e+00, ..., -2.2676e+00,
-1.8739e+00, -1.8030e+00],
[-4.7269e-01, -7.1203e-01, -3.7423e-01, ..., -7.7899e-01,
-8.1917e-01, -9.1171e-01]],
[[ 1.1150e-01, 2.5813e-01, 6.7886e-01, ..., 4.6059e-01,
7.8767e-01, -9.2955e-03],
[ 5.7303e-02, 2.2712e-01, 4.0557e-01, ..., 1.0735e-01,
1.2790e-01, -1.3314e+00],
[ 2.2274e-01, 2.9861e-01, -7.3686e-01, ..., -1.2566e+00,
-5.6134e-01, -8.7976e-01],
...,
[-3.6984e-01, -9.4202e-01, -2.2335e-01, ..., 9.0334e-01,
7.6497e-02, -3.4560e-01],
[-4.5355e-01, -6.6396e-01, 6.1196e-03, ..., -3.4617e-01,
-3.6412e-01, -5.9769e-01],
[ 5.3440e-01, 3.8349e-01, 7.2322e-02, ..., 2.1031e-01,
1.5856e-01, 3.2807e-01]],
[[-9.1115e-01, 2.8124e-01, 3.8671e-01, ..., 7.4897e-02,
1.8989e-01, 8.6669e-01],
[-1.2992e+00, 1.9315e-01, -3.7616e-01, ..., -1.8519e-01,
-2.9655e-02, 1.1190e+00],
[-2.8007e+00, -1.0234e+00, -7.0215e-01, ..., -2.5708e+00,
-3.5876e+00, -1.2557e+00],
...,
[-4.6499e-02, 1.9033e+00, 8.1298e-01, ..., 8.7835e-01,
8.7959e-01, 1.2906e+00],
[-1.5942e+00, -3.3035e-01, -3.4422e-01, ..., 4.2155e-01,
4.4250e-01, 8.4785e-01],
[-6.3933e-01, 2.1963e-03, -2.3101e-01, ..., 2.3907e-01,
1.3698e-01, 1.1112e+00]]],
[[[-1.6968e-03, -3.6298e-01, -1.1651e-01, ..., -1.4501e+00,
-1.3569e-01, -1.2341e+00],
[ 2.3397e-01, -1.4073e+00, -8.7499e-01, ..., -4.5135e+00,
-1.7760e+00, -6.8604e-01],
[-6.2599e-01, -9.6478e-01, -4.6188e-01, ..., -2.0026e+00,
-2.3496e+00, -2.2659e+00],
...,
[ 1.1234e-01, -7.3206e-01, -3.8809e-01, ..., -1.2689e+00,
-1.4468e+00, -2.0109e+00],
[-4.5466e-01, -6.7491e-01, -4.2366e-02, ..., -6.6121e-01,
-7.3829e-01, -1.8476e+00],
[-5.3034e-01, -1.2108e+00, -9.9027e-01, ..., -1.3385e+00,
-1.6284e+00, -8.8884e-01]],
[[-9.0686e-01, -2.4409e+00, -2.1771e+00, ..., -9.8676e+00,
-8.0334e+00, -2.7486e+00],
[-2.9560e+00, -4.8098e+00, -3.7098e+00, ..., -1.4108e+01,
-1.0051e+01, -3.9282e+00],
[-2.5872e+00, -4.4951e+00, -3.1479e+00, ..., -1.2866e+01,
-9.3413e+00, -5.3483e+00],
...,
[-2.0357e+00, -3.2908e+00, -3.0544e+00, ..., -6.5157e+00,
-7.1513e+00, -2.5049e+00],
[-3.7442e+00, -4.5496e+00, -3.7021e+00, ..., -6.8662e+00,
-8.3027e+00, -2.4621e+00],
[ 3.0292e-01, -3.0819e-01, 4.3868e-01, ..., 2.7073e-01,
3.7567e-01, 2.9143e-01]],
[[-2.9491e-01, 1.1471e-01, 8.2918e-01, ..., -1.2362e+00,
2.3665e-01, 1.9524e+00],
[ 1.4910e-02, -3.9211e-01, 6.1133e-01, ..., -6.2004e+00,
-2.7049e+00, 3.5036e-01],
[-5.5535e-01, -4.8908e-01, 3.0324e-01, ..., -4.2270e+00,
-1.7239e+00, -1.1302e-02],
...,
[-5.4252e-01, -7.5857e-01, 2.7329e-01, ..., -1.0081e+00,
1.5783e-02, -2.5929e-01],
[-3.4711e-01, -4.4035e-01, 3.8436e-01, ..., -2.9939e-01,
-4.9984e-02, -2.1600e-01],
[-1.4956e-01, -4.3367e-01, 2.2337e-01, ..., 8.8387e-02,
1.9769e-01, -5.9017e-01]],
...,
[[ 4.5206e-01, -2.3606e-01, 7.4960e-02, ..., 2.1476e+00,
1.7056e+00, -8.1147e-01],
[-8.1078e-01, -5.6470e-01, -5.2457e-01, ..., -1.0739e+00,
-1.1568e+00, -2.7405e+00],
[-5.1472e-01, -1.0936e+00, -4.6399e-01, ..., -6.8702e+00,
-4.4061e+00, -5.3620e+00],
...,
[-4.4339e-01, -3.4175e-02, -1.7123e-02, ..., -1.3909e+00,
-2.7328e-02, -9.0271e-01],
[-2.0323e+00, -1.5031e+00, -1.1415e+00, ..., -2.0103e+00,
-1.7003e+00, -1.9725e+00],
[-4.0440e-01, -6.0406e-01, -3.6114e-01, ..., -2.8960e-01,
-5.1182e-01, -1.2427e+00]],
[[ 8.7226e-02, 2.2426e-01, 6.9248e-01, ..., -9.2830e-01,
4.3568e-01, -1.4362e+00],
[ 1.0787e-01, 1.9877e-01, 2.7243e-01, ..., -4.0921e+00,
-1.9905e-02, -5.9562e-01],
[-9.4262e-01, 3.4649e-02, 3.7543e-01, ..., -2.2828e+00,
8.7409e-02, 9.5882e-01],
...,
[-1.9488e-01, 2.8460e-01, 3.2311e-01, ..., -2.3035e-01,
-2.3175e-01, -4.7187e-01],
[-4.6440e-01, -5.7156e-01, 4.2455e-02, ..., -2.1700e-01,
-3.3452e-01, -8.5812e-01],
[ 5.0774e-01, 4.1688e-01, 1.4062e-01, ..., 2.5430e-01,
2.8740e-01, 3.3897e-01]],
[[-9.3792e-01, 2.2057e-01, 3.7376e-01, ..., -5.5711e-01,
4.5379e-01, 1.5048e+00],
[-1.3844e+00, 3.3033e-02, -4.0688e-01, ..., -7.7334e+00,
-2.8030e+00, 1.6158e+00],
[-2.0254e+00, -5.5369e-01, -3.7766e-01, ..., -1.3673e+01,
-5.6432e+00, 1.2201e+00],
...,
[-1.1832e+00, 4.1711e-01, 6.0094e-02, ..., 6.0354e-01,
-2.8293e-02, 6.7748e-01],
[-1.6316e+00, -1.8787e-01, -2.6011e-01, ..., -2.4347e-01,
-6.5209e-01, 3.6209e-01],
[-6.2007e-01, 7.1656e-02, -1.8062e-01, ..., -2.9538e-01,
-5.0158e-01, 1.0541e+00]]],
[[[-3.4914e-01, -1.0869e+00, -8.9190e-01, ..., -4.0985e-02,
5.9433e-01, 5.1007e-01],
[ 1.2469e-01, -1.9540e+00, -1.3381e+00, ..., -1.1276e+00,
-3.6004e-01, 1.0667e-01],
[ 1.0381e-01, -1.4700e+00, -1.1337e+00, ..., -1.4541e+00,
5.1317e-02, -1.2181e-01],
...,
[-7.2250e-01, -5.8646e-01, -6.0880e-01, ..., -3.7400e-01,
-1.2197e-01, -1.3701e-01],
[ 4.1967e-01, -8.8394e-01, -9.1025e-01, ..., 7.6894e-04,
2.2932e-01, -3.2780e-02],
[-1.5325e-01, -1.0595e+00, -9.5391e-01, ..., -1.0390e+00,
-1.0979e+00, -3.7552e-01]],
[[-2.1484e+00, -4.1190e+00, -3.7074e+00, ..., -6.2521e-01,
-1.1467e+00, -6.3651e-02],
[-4.0615e+00, -6.4571e+00, -5.6291e+00, ..., -3.9127e+00,
-4.0301e+00, -7.4068e-01],
[-3.4695e+00, -5.4405e+00, -6.8343e+00, ..., -5.3438e+00,
-5.3758e+00, -4.8424e-01],
...,
[-4.1523e+00, -7.2891e+00, -5.8942e+00, ..., -1.4158e+00,
-1.6697e+00, -4.2082e-01],
[-5.0563e+00, -6.0607e+00, -5.7679e+00, ..., -2.6170e+00,
-3.4613e+00, -1.0245e+00],
[ 6.7598e-01, -1.2189e-01, 4.6998e-01, ..., 3.6443e-01,
3.7835e-01, 7.7349e-02]],
[[-6.1451e-01, 3.9642e-01, 8.0389e-01, ..., 8.6765e-01,
3.6865e-01, 1.1289e+00],
[-9.8779e-01, 1.2535e-01, 5.8303e-01, ..., 9.6656e-01,
3.9390e-01, 4.6038e-01],
[-1.3656e+00, 1.5098e-03, -1.1740e-01, ..., 1.1755e+00,
1.3379e+00, 4.5111e-01],
...,
[ 1.3179e-01, 1.1419e+00, -2.1995e+00, ..., 8.1435e-02,
1.1592e-01, 4.3803e-01],
[-8.6507e-01, 1.5618e-01, -3.9192e-01, ..., 1.5458e-01,
1.2197e-01, 6.1862e-01],
[-5.8676e-01, 3.0017e-01, 1.3764e-01, ..., -8.5250e-02,
8.0795e-02, 3.8261e-01]],
...,
[[ 6.3578e-01, 9.1545e-02, 2.8748e-01, ..., -8.4767e-02,
-3.6059e-01, -9.2773e-01],
[-8.4493e-01, -3.3378e-01, -1.0103e-01, ..., -2.8238e-01,
-1.1930e+00, -8.3606e-01],
[-5.5619e-01, -3.1692e-01, 1.1948e-01, ..., -1.7886e+00,
-1.0259e+00, -4.1685e-01],
...,
[-1.4789e+00, -5.3980e+00, -3.6613e+00, ..., -1.2004e-01,
-2.2039e-01, -3.8302e-01],
[-3.2320e+00, -2.6765e+00, -1.1955e+00, ..., -8.5154e-01,
-8.9671e-01, -8.0729e-01],
[-4.1574e-01, -9.6594e-01, -3.4352e-01, ..., -1.3544e-02,
-1.3262e-01, -3.4517e-01]],
[[-2.3876e-02, -3.1372e-01, -4.9780e-02, ..., 3.6171e-01,
6.9426e-01, 2.9056e-01],
[ 6.4355e-02, -1.0190e+00, -8.3232e-02, ..., -1.1195e+00,
-7.0335e-01, -5.9103e-02],
[-2.5061e-01, -1.0608e+00, 3.0758e-01, ..., -6.5932e-01,
-8.8371e-01, 1.5616e-01],
...,
[-1.5578e+00, -1.2979e-02, 3.5952e-01, ..., -1.4784e-01,
-1.3271e-01, 1.4481e-01],
[-6.8317e-01, -2.1972e+00, -9.1551e-01, ..., -4.2075e-01,
-3.7726e-01, -8.8944e-02],
[ 6.8337e-01, -1.4897e-01, 2.3688e-01, ..., -2.4094e-01,
-2.2974e-01, 3.1133e-01]],
[[-6.7124e-01, -1.9300e-01, -2.1421e-02, ..., 5.3502e-01,
3.4050e-01, 5.7613e-01],
[-1.0963e+00, -8.9027e-01, -9.5067e-01, ..., -2.5811e-01,
-2.6100e-01, 4.5680e-01],
[-6.8833e-01, -8.3862e-01, -5.2788e-01, ..., -6.1741e-01,
-1.9104e-03, 1.0458e-01],
...,
[-2.6505e+00, -1.8483e+00, 9.8762e-01, ..., 7.0043e-02,
-7.4850e-02, 1.0610e-01],
[-7.8045e-01, -6.1783e-01, -6.7782e-02, ..., -9.9096e-02,
-2.4230e-01, -1.9896e-01],
[-4.6734e-01, -8.8473e-01, -3.4266e-01, ..., -1.3626e-01,
-2.5811e-01, 1.3456e-01]]]], device='cuda:0'),
tensor([[[[-4.6987e+00, -6.1239e+00, -3.4639e+00, ..., 6.1540e-02,
6.3684e-01, -8.4536e-01],
[-4.0427e+00, -6.4994e+00, -6.9317e+00, ..., -4.1941e+00,
-2.0226e+00, -1.4155e+00],
[-1.4218e+00, -6.2850e+00, -9.9208e+00, ..., -1.9187e+00,
-1.4018e+00, -1.9590e+00],
...,
[ 6.8930e-02, -6.5590e-01, -2.5487e-01, ..., -1.1955e+00,
-6.9938e-01, -9.8992e-01],
[-3.3121e-01, -5.1578e-01, 2.4281e-02, ..., 1.4295e-01,
2.7500e-01, -2.2171e-01],
[-5.7139e-01, -1.3167e+00, -1.0910e+00, ..., -1.0690e+00,
-1.1607e+00, -5.1844e-01]],
[[-5.0804e+00, -8.6645e+00, -1.1604e+01, ..., -3.9921e+00,
-6.7187e+00, -2.0494e+00],
[-5.6711e+00, -9.1503e+00, -1.4319e+01, ..., -9.7438e+00,
-1.1683e+01, -4.3716e+00],
[-8.7249e+00, -1.4582e+01, -1.4961e+01, ..., -8.3886e+00,
-8.1824e+00, -3.6895e+00],
...,
[-1.9157e+00, -2.4782e+00, -2.0936e+00, ..., -3.1647e+00,
-3.0036e+00, -1.4107e+00],
[-2.9565e+00, -3.1735e+00, -2.1619e+00, ..., -2.4419e+00,
-3.5263e+00, -1.0900e+00],
[ 3.6373e-01, -2.4642e-01, 3.7130e-01, ..., 3.3848e-01,
4.8113e-01, 1.9580e-01]],
[[-3.0983e+00, -1.0665e+01, -1.3825e+01, ..., -1.3678e+00,
1.3871e+00, 1.5925e+00],
[-3.3424e+00, -1.3007e+01, -1.2278e+01, ..., -5.7009e+00,
8.1645e-01, 5.2049e-01],
[-8.5408e+00, -1.1778e+01, -3.3751e+00, ..., -7.1084e+00,
2.0359e-01, 2.2905e-01],
...,
[-7.4035e-01, 6.3836e-02, 2.8136e-01, ..., 1.3123e+00,
2.2259e-01, 6.9510e-01],
[-4.9644e-01, 9.7476e-02, 2.5034e-01, ..., 1.8670e-01,
2.3696e-01, 5.5778e-01],
[-1.7502e-01, -1.8911e-01, 1.4430e-01, ..., -2.4307e-03,
6.9700e-02, 3.3398e-01]],
...,
[[ 2.7986e+00, -1.3044e+00, -7.1132e+00, ..., -4.8568e+00,
-1.8718e+00, -1.0863e+00],
[-1.2115e+00, -3.2179e+00, -5.2232e+00, ..., -7.3749e+00,
-4.7492e+00, -1.5198e+00],
[ 6.8154e-01, -3.1334e+00, -3.2973e+00, ..., -6.6218e+00,
-3.5396e+00, -1.4016e+00],
...,
[-6.6260e-01, -5.4863e-01, -3.5726e-01, ..., -2.3602e+00,
-2.4994e+00, -1.1589e+00],
[-2.2154e+00, -1.7785e+00, -1.2983e+00, ..., -1.2799e+00,
-1.3913e+00, -1.1027e+00],
[-6.5489e-01, -8.6820e-01, -4.4812e-01, ..., -3.2429e-01,
-4.0012e-01, -3.9795e-01]],
[[-2.6288e+00, -6.9097e+00, -4.0401e+00, ..., -7.2545e-01,
1.2238e+00, -3.1223e-01],
[-3.6501e+00, -7.4200e+00, -2.3878e+00, ..., -4.7788e+00,
-1.5744e+00, -1.2122e+00],
[-2.1352e+00, -4.8843e+00, -4.0108e+00, ..., -1.4534e+00,
1.6898e-01, -2.4881e-01],
...,
[-2.2258e-01, -5.2400e-01, 2.6642e-01, ..., -6.8978e-01,
-7.2413e-02, -4.3376e-01],
[-5.2014e-01, -1.0009e+00, 8.6975e-02, ..., 3.2982e-02,
-3.3420e-02, -1.8938e-01],
[ 3.6175e-01, -5.7774e-02, -1.6732e-01, ..., -1.1332e-01,
-8.8390e-02, 1.6610e-01]],
[[-5.0465e+00, -4.7196e+00, -5.5006e+00, ..., -1.8655e+00,
6.2097e-01, 9.1968e-01],
[-4.0660e+00, -3.6566e+00, -3.3791e+00, ..., -4.1548e-01,
2.8981e-01, 8.7172e-01],
[-2.1498e+00, -4.8532e+00, -7.2363e+00, ..., -1.9421e+00,
-5.4903e-01, 3.8800e-01],
...,
[-6.7381e-01, -2.7748e-01, -1.8342e-01, ..., 1.1167e+00,
9.8563e-01, 1.3677e+00],
[-1.1069e+00, -6.1366e-01, -3.4906e-01, ..., -3.4198e-01,
-1.9780e-01, 2.3380e-01],
[-3.4831e-01, -6.2007e-02, -1.5145e-01, ..., -2.3468e-01,
-2.1578e-01, 3.8612e-01]]],
[[[-6.2994e-01, -1.4586e+00, -1.0569e+00, ..., -1.0237e+00,
-7.8270e-01, -1.7114e+00],
[-2.6081e-01, -2.5438e+00, -1.9878e+00, ..., -1.5104e+00,
-1.7789e+00, -1.9350e+00],
[-1.6715e-01, -1.9621e+00, -1.5970e+00, ..., -1.0624e+00,
-1.4276e+00, -2.0909e+00],
...,
[-9.3000e-02, -1.1147e+00, -6.0668e-01, ..., -5.5541e-01,
-4.7244e-01, -8.9311e-01],
[-1.1159e-01, -9.5301e-01, -2.9971e-01, ..., -2.1420e-01,
-8.9038e-02, -9.8518e-01],
[-1.4758e-01, -1.1198e+00, -8.2776e-01, ..., -8.7865e-01,
-1.1303e+00, -6.5180e-01]],
[[-2.5588e+00, -5.0071e+00, -6.1546e+00, ..., -4.7138e+00,
-6.3336e+00, -2.4672e+00],
[-5.4693e+00, -8.7741e+00, -8.7217e+00, ..., -7.7758e+00,
-9.5407e+00, -4.3449e+00],
[-3.9972e+00, -6.1280e+00, -5.2967e+00, ..., -6.5630e+00,
-6.8891e+00, -3.2216e+00],
...,
[-2.0233e+00, -3.1678e+00, -2.8981e+00, ..., -2.9427e+00,
-3.4122e+00, -1.4692e+00],
[-4.4290e+00, -5.3841e+00, -4.7581e+00, ..., -5.0471e+00,
-6.3198e+00, -2.4176e+00],
[ 6.6999e-01, -2.9171e-01, 5.2258e-01, ..., 5.1654e-01,
5.7530e-01, -1.2865e-03]],
[[-7.1747e-01, 2.7823e-01, 8.4013e-01, ..., 9.7219e-01,
7.8802e-01, 1.8677e+00],
[-9.3687e-01, -1.6471e-01, 1.0141e+00, ..., 2.3528e-01,
2.0984e-01, 1.0132e-01],
[-1.7340e+00, -3.7859e-01, 5.6890e-01, ..., -5.3151e-01,
1.9121e-01, -9.1444e-02],
...,
[-9.2515e-01, -1.2105e-01, 5.6636e-02, ..., 1.3170e-02,
6.8684e-02, 3.6942e-01],
[-9.4153e-01, 2.6681e-02, 5.2151e-02, ..., 3.4809e-03,
1.1214e-01, 4.0783e-01],
[-4.6028e-01, 3.5715e-03, 9.0537e-02, ..., -3.6648e-02,
1.1094e-01, 4.0051e-02]],
...,
[[ 1.0441e+00, -4.6638e-01, 4.3519e-01, ..., 3.5804e-01,
-1.3578e-02, -2.0425e+00],
[-9.5784e-01, -7.7388e-01, 6.8152e-02, ..., -5.9113e-01,
-1.5212e+00, -2.0237e+00],
[-8.3621e-01, -5.6949e-01, 2.0864e-02, ..., -5.3829e-01,
-6.4848e-01, -1.3697e+00],
...,
[-4.4461e-01, -1.1117e-01, 1.3114e-01, ..., 7.0833e-02,
-1.3269e-01, -5.1429e-01],
[-2.0275e+00, -1.3508e+00, -8.5675e-01, ..., -9.5906e-01,
-1.1432e+00, -1.1117e+00],
[-3.1179e-01, -7.3875e-01, -2.2420e-01, ..., -1.3043e-01,
-2.7594e-01, -6.0684e-01]],
[[-9.9936e-02, -5.2655e-02, 7.6134e-01, ..., 7.9699e-01,
8.8844e-01, -1.8254e-02],
[-3.6113e-02, -7.4858e-01, 4.3745e-01, ..., 1.0510e+00,
4.5223e-01, -8.1990e-01],
[-1.3846e-01, -6.6730e-01, 6.7399e-01, ..., 1.1844e+00,
6.2356e-01, -3.6720e-01],
...,
[-3.9579e-01, -5.2337e-01, 6.5453e-02, ..., 1.2156e-01,
2.7258e-02, -1.3930e-01],
[-7.3257e-01, -1.4538e+00, -2.8984e-01, ..., -3.3796e-01,
-3.4002e-01, -7.0740e-01],
[ 4.8297e-01, 2.2764e-02, 1.8828e-01, ..., 2.1067e-01,
1.7184e-01, 3.5645e-01]],
[[-1.6546e+00, -7.6618e-01, -7.0641e-01, ..., -4.4164e-01,
-3.1851e-01, 8.6362e-01],
[-2.1384e+00, -7.6489e-01, -1.2571e+00, ..., -6.8798e-01,
-9.7380e-02, 1.6561e+00],
[-1.5096e+00, -7.8673e-01, -8.0975e-01, ..., -2.1019e-01,
5.0236e-02, 1.4000e+00],
...,
[-4.7444e-01, 1.1469e-01, 9.1544e-02, ..., 1.6528e-01,
2.2613e-01, 7.0099e-01],
[-8.6408e-01, -3.2272e-01, -1.7869e-01, ..., -7.8687e-02,
6.9761e-02, 4.1276e-01],
[-3.7826e-01, -3.1908e-01, -1.6553e-01, ..., -2.4226e-01,
-2.7233e-01, 5.7688e-01]]],
[[[ 2.3283e-02, -2.3290e-01, -3.0181e-02, ..., -5.8319e-01,
-2.6563e-01, -1.4470e+00],
[ 3.0021e-01, -9.3542e-01, -6.9708e-01, ..., -1.0939e+00,
-6.2552e-01, -1.4809e+00],
[ 2.2658e-01, -5.9001e-01, -4.9941e-01, ..., -1.3682e+00,
-1.2517e+00, -1.8071e+00],
...,
[-3.4755e-02, -5.9968e-01, 4.2433e-01, ..., -9.0042e-01,
-1.0269e+00, -1.9266e+00],
[-6.5703e-01, -7.0543e-01, 9.6273e-01, ..., -4.4263e-01,
-5.5923e-01, -1.7083e+00],
[-7.3692e-01, -1.3124e+00, -6.9568e-01, ..., -1.0998e+00,
-1.4912e+00, -7.4340e-01]],
[[-7.7709e-01, -2.1844e+00, -1.8468e+00, ..., -4.0549e+00,
-6.5647e+00, -1.8530e+00],
[-2.0714e+00, -3.8041e+00, -2.9705e+00, ..., -5.1216e+00,
-9.6292e+00, -3.4341e+00],
[-1.6114e+00, -2.6077e+00, -2.5260e+00, ..., -7.0258e+00,
-8.6695e+00, -2.6075e+00],
...,
[-1.7964e+00, -2.4539e+00, -5.2394e+00, ..., -4.9838e+00,
-6.3126e+00, -2.4855e+00],
[-3.8198e+00, -4.4102e+00, -4.0861e+00, ..., -6.3807e+00,
-7.8512e+00, -2.5200e+00],
[ 2.8143e-01, -3.5699e-01, 6.1695e-01, ..., 4.3308e-01,
4.8213e-01, 2.2263e-01]],
[[-2.3890e-01, 1.3560e-01, 7.3822e-01, ..., 1.6380e-01,
2.8389e-01, 1.5231e+00],
[ 4.7158e-02, -4.1590e-01, 5.6217e-01, ..., -2.6150e+00,
-4.3704e-01, 9.1571e-02],
[-1.4284e-01, -2.3718e-01, 2.2911e-01, ..., -6.7445e+00,
-7.5719e-01, -1.5597e-01],
...,
[-2.8080e-01, -2.3564e-01, 2.1188e+00, ..., 1.7357e-02,
-1.1590e-01, -1.6850e-01],
[-3.1406e-01, -2.4317e-01, 7.7005e-01, ..., -7.9154e-02,
9.3974e-03, -1.8968e-01],
[-1.5160e-01, -5.0650e-01, 1.3423e+00, ..., 6.7631e-02,
2.3871e-01, -5.5205e-01]],
...,
[[ 5.7659e-01, -3.1134e-01, -3.4730e-02, ..., -2.8924e-01,
3.7635e-01, -1.2979e+00],
[-5.2945e-01, -5.5780e-01, -5.4026e-01, ..., -2.6830e+00,
-1.9467e+00, -1.3851e+00],
[-3.8422e-01, -2.2215e-01, -2.4483e-01, ..., -4.8912e+00,
-2.5526e+00, -9.5737e-01],
...,
[-4.4618e-01, -1.6358e-01, -6.5530e-02, ..., -1.3687e-01,
-1.2112e-01, -9.2593e-01],
[-2.3430e+00, -1.7306e+00, -1.6472e+00, ..., -1.5260e+00,
-1.5439e+00, -1.7852e+00],
[-4.7028e-01, -8.0248e-01, -7.2274e-01, ..., -3.2340e-01,
-5.3085e-01, -1.1760e+00]],
[[ 1.4073e-01, 2.4010e-01, 6.3477e-01, ..., 1.2498e-01,
4.2797e-02, -3.5453e-01],
[ 1.0140e-01, 2.3619e-01, 4.5821e-01, ..., 2.5476e-01,
4.0414e-01, -7.9079e-01],
[-1.3061e-01, 8.9331e-02, 2.7584e-01, ..., 1.3745e+00,
8.7255e-01, -4.6140e-01],
...,
[-2.3597e-01, 2.5930e-02, 9.0234e-01, ..., 2.3910e-01,
1.3290e-01, -4.8705e-01],
[-6.0534e-01, -7.0234e-01, 3.8174e-01, ..., -1.2161e-01,
-2.6753e-01, -8.8818e-01],
[ 5.0176e-01, 4.4372e-01, 4.3757e-02, ..., 3.7750e-01,
3.4096e-01, 3.1228e-01]],
[[-8.4505e-01, 2.7921e-01, 3.2940e-01, ..., -4.4791e-01,
-5.1272e-02, 5.8994e-01],
[-1.1247e+00, 2.6830e-01, -3.1458e-01, ..., -2.5507e+00,
-6.0435e-01, 9.3017e-01],
[-9.4780e-01, 3.5462e-01, -1.1288e-01, ..., -3.4824e+00,
-4.7397e-01, 6.3855e-01],
...,
[-9.6188e-01, 1.9105e-01, -5.6822e-01, ..., -5.5839e-01,
-5.4593e-01, 6.1311e-01],
[-1.7261e+00, -4.0015e-01, -4.3301e-01, ..., -6.4959e-01,
-6.8212e-01, 3.9344e-01],
[-7.6271e-01, -5.2763e-02, -8.3928e-01, ..., -4.5081e-01,
-4.3180e-01, 1.0098e+00]]],
[[[-7.4275e-01, -1.4999e+00, -1.5246e+00, ..., -1.6681e+00,
-1.2983e+00, -2.0472e+00],
[-3.5225e-01, -2.5146e+00, -2.4181e+00, ..., -1.1669e+00,
-1.2336e+00, -1.4367e+00],
[-3.1449e-01, -5.5606e+00, -4.5326e+00, ..., 7.5967e-01,
-4.7092e-01, -1.5129e+00],
...,
[ 1.2698e-01, 4.7892e-01, 9.3502e-01, ..., -1.2470e+00,
-1.5252e-01, -5.4973e-01],
[-2.6041e-01, 1.0792e-01, 1.1582e+00, ..., 1.3517e-01,
2.2975e-01, -2.6481e-01],
[-5.4311e-01, -1.2865e+00, -1.0875e+00, ..., -1.0074e+00,
-1.1635e+00, -5.3195e-01]],
[[-2.7646e+00, -5.2363e+00, -4.6440e+00, ..., -1.0749e+01,
-8.7888e+00, -5.1538e+00],
[-5.6709e+00, -8.3925e+00, -7.7038e+00, ..., -2.0503e+01,
-1.3563e+01, -7.7699e+00],
[-9.1219e+00, -1.2977e+01, -1.6741e+01, ..., -2.1332e+01,
-8.1537e+00, -5.1176e+00],
...,
[-2.0023e+00, -1.9643e+00, -2.5482e+00, ..., -2.8181e+00,
-2.5908e+00, -1.0736e+00],
[-2.9864e+00, -3.4359e+00, -1.7572e+00, ..., -2.4767e+00,
-3.6412e+00, -1.1581e+00],
[ 3.9060e-01, -1.4774e-01, 4.0047e-01, ..., 3.6214e-01,
4.5721e-01, 1.8382e-01]],
[[-6.9563e-01, 4.5233e-01, 8.7398e-01, ..., 9.1490e-01,
9.0964e-01, 2.4556e+00],
[-9.2365e-01, 6.2605e-02, 3.9445e-01, ..., -2.8224e+00,
-2.3760e-01, 1.2540e+00],
[-2.7448e+00, -2.3503e+00, -4.0514e-01, ..., -9.1066e+00,
-5.4800e+00, -4.4592e-01],
...,
[-7.5756e-01, 2.7055e+00, -3.9572e+00, ..., 6.6554e-01,
3.0269e-01, 4.0037e-01],
[-5.1250e-01, 2.0348e+00, 2.2618e-02, ..., 1.7413e-01,
2.4501e-01, 5.5613e-01],
[-1.1040e-01, -1.5254e-01, 1.1956e-01, ..., -1.7703e-02,
1.0697e-01, 3.1137e-01]],
...,
[[ 1.2914e+00, -3.7590e-01, -6.9976e-02, ..., 1.5546e+00,
1.1184e+00, -1.4614e+00],
[-7.5767e-01, -9.5625e-01, -5.7919e-01, ..., -1.8357e+00,
-2.6558e+00, -3.2799e+00],
[-5.0462e-02, 3.0809e-02, -7.9948e-01, ..., -8.5590e+00,
-6.7418e+00, -4.8630e+00],
...,
[-6.2976e-01, -4.2646e-01, -1.1976e+00, ..., -6.0431e-01,
-5.5232e-01, -6.1618e-01],
[-2.2165e+00, -3.2531e+00, -3.4611e+00, ..., -1.3010e+00,
-1.3686e+00, -1.1321e+00],
[-6.3538e-01, -8.2551e-01, -3.8461e-01, ..., -3.1803e-01,
-3.8583e-01, -4.2199e-01]],
[[ 1.9726e-02, -2.6025e-03, 5.0956e-01, ..., 2.4549e-01,
-4.2954e-02, -1.7493e+00],
[ 2.2559e-01, -8.1148e-01, -7.4752e-02, ..., -2.0432e-01,
3.4895e-01, -1.7016e+00],
[ 9.8701e-01, -3.6762e-01, -4.5365e-01, ..., -2.3084e+00,
-5.6127e-01, 1.1314e+00],
...,
[-1.6348e-01, -6.2830e-01, -7.6024e-01, ..., -5.0860e-01,
2.4864e-01, -4.8918e-02],
[-4.9952e-01, -3.6780e-01, 1.7499e-01, ..., 2.7825e-02,
-5.5094e-02, -2.5629e-01],
[ 3.9623e-01, -8.3874e-02, -1.8088e-01, ..., -5.5210e-02,
-4.1351e-02, 1.5164e-01]],
[[-1.6973e+00, -8.9426e-01, -5.0093e-01, ..., -7.8236e-01,
-3.9125e-01, 1.5097e+00],
[-2.3061e+00, -8.0043e-01, -5.5601e-01, ..., -6.5304e+00,
-9.9126e-01, 3.0002e+00],
[-2.0238e+00, -2.4648e+00, -1.2840e+00, ..., -1.1745e+01,
-2.7625e+00, 2.5484e+00],
...,
[-6.8409e-01, -3.2002e+00, -6.8872e-01, ..., 5.8461e-01,
-3.6601e-02, 6.3654e-01],
[-1.1103e+00, -1.6911e+00, 1.6404e-01, ..., -3.2489e-01,
-1.5810e-01, 2.4251e-01],
[-3.5364e-01, -7.6965e-02, -1.5590e-01, ..., -2.0747e-01,
-2.0170e-01, 3.8297e-01]]],
[[[-1.3147e-01, -3.8802e-01, -1.9527e+00, ..., -2.9286e-01,
2.6324e-01, -1.0266e-01],
[ 3.1786e-01, -9.9999e-01, -2.3997e+00, ..., -9.2450e-01,
-3.1363e-01, -2.1012e-01],
[ 1.6847e-01, -5.8296e-01, -7.6763e-01, ..., -4.9456e-02,
-3.3783e-01, -5.4533e-01],
...,
[-3.2694e-01, -4.3397e+00, -1.8586e+00, ..., -1.0025e+00,
-1.0185e+00, -2.0809e+00],
[-7.9634e-01, -3.5308e+00, -3.0073e+00, ..., -5.0492e-01,
-4.7780e-01, -1.8433e+00],
[-8.5500e-01, -2.5378e+00, -4.3106e+00, ..., -1.4745e+00,
-1.5819e+00, -9.9483e-01]],
[[-1.4498e+00, -3.5440e+00, -3.3181e+00, ..., -2.9573e+00,
-3.4636e+00, -1.3908e+00],
[-3.2240e+00, -6.4655e+00, -7.6890e+00, ..., -4.5767e+00,
-4.9832e+00, -2.0823e+00],
[-2.1195e+00, -4.2413e+00, -6.3755e+00, ..., -3.6053e+00,
-3.2886e+00, -1.3160e+00],
...,
[-5.2612e+00, -8.0367e+00, -5.8572e+00, ..., -5.7350e+00,
-6.3142e+00, -2.9520e+00],
[-8.1479e+00, -1.2239e+01, -1.0575e+01, ..., -6.2722e+00,
-7.5400e+00, -2.7244e+00],
[ 1.8876e-01, -3.2014e-01, -8.1183e-01, ..., 3.6215e-01,
6.8128e-01, 1.8484e-01]],
[[-4.6073e-01, 1.5346e-01, -1.0069e+00, ..., 8.2979e-01,
5.9628e-01, 1.3681e+00],
[-4.0213e-01, 1.3279e-02, -1.2699e+00, ..., 6.1390e-01,
4.5464e-01, 3.0123e-01],
[-8.4148e-01, 1.7225e-02, -2.6171e+00, ..., 4.0744e-01,
4.8002e-01, 4.1862e-01],
...,
[-2.5890e+00, -3.5785e+00, -6.4110e+00, ..., 1.0011e-01,
-2.4635e-04, 3.0313e-01],
[-8.1222e-01, -3.5153e+00, -5.5550e+00, ..., 9.6065e-03,
1.2354e-02, 2.7911e-01],
[-5.0796e-01, -3.4918e+00, -5.4297e+00, ..., -4.9232e-02,
-3.4072e-02, -3.4935e-01]],
...,
[[ 9.0150e-01, 6.8072e-01, 2.4912e+00, ..., 4.0919e-01,
1.5615e-01, -8.7414e-01],
[-1.1426e+00, -7.3431e-01, -1.3568e+00, ..., -8.0071e-01,
-1.5182e+00, -1.0952e+00],
[-5.4600e-01, 3.7014e-03, 1.0198e+00, ..., -1.0469e+00,
-5.6723e-01, -6.0698e-01],
...,
[ 3.7690e-01, -2.4572e-01, -4.2459e+00, ..., -4.0764e-01,
-6.1643e-01, -1.2053e+00],
[-1.7883e+00, -2.3886e+00, -7.5997e+00, ..., -1.9372e+00,
-2.0679e+00, -2.0902e+00],
[-1.7708e-01, 7.9381e-02, -3.8632e+00, ..., -8.8429e-01,
-9.4108e-01, -1.3506e+00]],
[[ 7.7610e-02, 3.6549e-01, -1.8328e-01, ..., 4.4754e-01,
9.6064e-01, 3.3951e-01],
[ 3.8063e-02, -2.2135e-01, -1.2836e+00, ..., -4.8710e-03,
5.3736e-01, -1.8765e-01],
[-2.4416e-01, -2.8621e-01, -4.4560e-01, ..., 6.9072e-01,
2.2641e-01, -5.8488e-03],
...,
[-3.0848e-01, -4.0128e+00, -5.3464e+00, ..., 2.1349e-01,
1.0952e-01, -7.4133e-01],
[-2.9381e-01, -3.4420e+00, -5.2682e+00, ..., 5.6690e-02,
-5.8150e-02, -9.6439e-01],
[ 4.8996e-01, -1.4163e+00, -2.8437e+00, ..., 2.0374e-01,
2.6737e-01, 3.3459e-01]],
[[-9.4841e-01, -6.8734e-01, -2.7854e+00, ..., 1.5235e-01,
5.4378e-02, 7.2818e-01],
[-1.0419e+00, -5.8378e-01, -2.6457e+00, ..., -6.1898e-01,
-1.6436e-01, 1.1413e+00],
[-7.1729e-01, -5.5430e-01, -1.2600e+00, ..., -5.5898e-01,
-5.9926e-02, 6.6757e-01],
...,
[-1.3073e+00, -3.3480e+00, -6.1203e+00, ..., -4.0798e-01,
-7.2174e-02, 8.4713e-01],
[-2.3196e+00, -2.8396e+00, -4.5809e+00, ..., -7.7060e-01,
-5.1906e-01, 4.2734e-01],
[-1.5032e+00, -1.2160e+00, -4.4680e-01, ..., -2.6950e-01,
-1.6296e-01, 1.1502e+00]]]], device='cuda:0'),
tensor([[[[-8.8453e-01, -1.6410e+00, -1.7156e+00, ..., -3.3583e-01,
2.7687e-01, 1.3210e-01],
[-5.0383e-01, -2.6924e+00, -2.4929e+00, ..., -7.5128e-01,
-4.5758e-01, -1.2156e-01],
[-5.2237e-01, -2.3812e+00, -2.2926e+00, ..., -2.4078e+00,
-1.4255e+00, -5.0367e-01],
...,
[ 1.3116e+00, 4.2997e-01, -8.5796e+00, ..., -1.4995e+00,
-3.7539e-01, -4.9822e-01],
[-8.4164e-01, -6.8240e-01, -2.1199e+00, ..., -1.9783e+00,
2.3740e-02, -2.9082e-01],
[-1.0780e+00, -2.1804e+00, -2.9315e+00, ..., -1.0764e+00,
-1.1642e+00, -4.3096e-01]],
[[-2.8008e+00, -5.5247e+00, -4.4856e+00, ..., -1.7491e+00,
-2.1614e+00, -5.1169e-01],
[-6.0531e+00, -9.1372e+00, -9.2535e+00, ..., -3.3677e+00,
-4.0108e+00, -1.3199e+00],
[-4.8631e+00, -6.9946e+00, -1.1254e+01, ..., -8.8295e+00,
-5.0407e+00, -1.0173e+00],
...,
[-4.6563e+00, -8.3577e+00, -2.9628e+01, ..., -4.9242e+00,
-2.5762e+00, -9.1173e-01],
[-7.0413e+00, -8.0682e+00, -2.6994e+01, ..., -4.3553e+00,
-4.3899e+00, -1.4472e+00],
[-3.5515e-01, -1.1125e+00, -1.3981e+00, ..., 3.0003e-01,
4.4240e-01, 9.3364e-02]],
[[-6.9082e-01, 5.0401e-01, 1.1010e+00, ..., 1.0141e+00,
6.7518e-01, 1.3706e+00],
[-8.0090e-01, 6.7059e-03, 9.6713e-02, ..., 5.4488e-01,
4.0685e-01, 4.7035e-01],
[-1.7203e+00, -2.0721e-01, -9.6716e-01, ..., 3.8505e-01,
3.0675e-01, 4.6301e-01],
...,
[ 2.4955e+00, -7.6649e+00, -4.9774e+00, ..., 3.2716e+00,
2.8226e-01, 4.7011e-01],
[ 1.6614e+00, 9.9017e-01, 4.9019e+00, ..., 1.8293e+00,
2.8295e-01, 6.4880e-01],
[-1.6121e+00, -6.9095e-01, 2.1531e+00, ..., -1.8752e-02,
9.9384e-02, 2.9022e-01]],
...,
[[ 1.3806e+00, -4.2948e-01, -2.7917e-01, ..., -1.3073e-02,
-2.0885e-01, -1.0440e+00],
[-7.2673e-01, -1.1139e+00, -8.8837e-01, ..., -8.2228e-01,
-1.5248e+00, -1.1530e+00],
[-6.0687e-01, -1.0431e+00, 1.7202e-01, ..., -7.2680e-01,
-2.5148e-01, -6.6716e-01],
...,
[-1.1463e+00, -1.5282e+00, -4.1919e-01, ..., -5.1418e+00,
-3.8388e-01, -6.0441e-01],
[-5.2499e+00, -7.3890e+00, -7.6844e+00, ..., -1.1156e+00,
-1.0653e+00, -9.8314e-01],
[-7.3256e-01, -2.1045e+00, -1.1777e+00, ..., -1.0295e-01,
-2.4375e-01, -4.2740e-01]],
[[ 1.6353e-02, -1.2530e-01, 6.6925e-01, ..., 3.9718e-01,
8.0899e-01, 3.1182e-01],
[ 4.5291e-02, -8.8180e-01, 1.1610e-01, ..., 3.1751e-02,
2.3735e-01, -1.7064e-01],
[ 1.0494e-01, -9.1488e-01, 8.7184e-01, ..., -3.1327e+00,
-2.3546e+00, -1.8581e-01],
...,
[-4.5437e-01, -3.7103e+00, -4.7547e+00, ..., 3.1419e-01,
-1.5036e-01, -1.2515e-01],
[ 2.2430e-01, -1.4381e+00, -1.0859e+00, ..., -1.3397e+00,
-4.5526e-01, -4.0664e-01],
[ 6.2827e-01, 1.3038e-01, -1.3578e+00, ..., -2.9285e-01,
-1.7429e-01, 2.0076e-01]],
[[-2.0951e+00, -9.8039e-01, -2.8216e-01, ..., 6.2573e-01,
3.9286e-01, 7.9331e-01],
[-2.6747e+00, -4.6872e-01, -1.2912e-01, ..., 5.8608e-02,
3.7098e-03, 7.7736e-01],
[-2.3228e+00, -5.5099e-01, -6.8673e-01, ..., 5.5428e-01,
3.3493e-01, 3.5846e-01],
...,
[-4.9502e+00, 1.0857e-01, -4.2219e+00, ..., 1.5561e+00,
5.4405e-03, 3.7969e-01],
[-4.3636e+00, -1.8082e+00, -3.4330e+00, ..., 8.9075e-01,
-1.2101e-01, 6.7577e-02],
[-1.3859e+00, -2.5587e-01, -2.6507e+00, ..., -1.0760e-01,
-2.9745e-01, 3.0990e-01]]],
[[[-3.3138e-01, -9.8898e-01, -6.5467e-01, ..., -6.0952e-01,
-2.4743e-02, -8.3171e-01],
[ 3.0905e-02, -2.0348e+00, -1.3520e+00, ..., -9.6901e-01,
-8.6662e-01, -1.2058e+00],
[-2.7210e-03, -1.6552e+00, -1.0297e+00, ..., -9.4542e-01,
-9.3891e-01, -1.7388e+00],
...,
[ 5.7603e-02, -5.6364e-01, -2.0743e-01, ..., 9.5904e-02,
-1.8092e-01, -6.9784e-01],
[-3.9512e-01, -4.7761e-01, 4.1596e-02, ..., 1.5233e-01,
2.8254e-01, -1.5913e-01],
[-6.4029e-01, -1.3379e+00, -1.0864e+00, ..., -1.1171e+00,
-1.1928e+00, -5.0289e-01]],
[[-1.9001e+00, -3.9122e+00, -3.6948e+00, ..., -4.3076e+00,
-5.2430e+00, -2.0475e+00],
[-3.9452e+00, -6.2504e+00, -5.2143e+00, ..., -5.5446e+00,
-6.8426e+00, -2.8446e+00],
[-3.4643e+00, -5.2936e+00, -4.8294e+00, ..., -5.4272e+00,
-5.9157e+00, -2.8071e+00],
...,
[-1.7252e+00, -2.1373e+00, -1.7652e+00, ..., -2.4360e+00,
-2.8677e+00, -1.2610e+00],
[-2.7829e+00, -2.9779e+00, -1.9510e+00, ..., -2.1715e+00,
-3.3244e+00, -9.8370e-01],
[ 3.1292e-01, -2.8137e-01, 3.2168e-01, ..., 2.8694e-01,
4.3813e-01, 1.9816e-01]],
[[-5.4471e-01, 3.9553e-01, 9.2746e-01, ..., -1.8513e-01,
5.8182e-01, 1.4520e+00],
[-7.2735e-01, -7.5614e-02, 4.7229e-01, ..., -3.4590e-01,
3.2754e-01, 3.4463e-01],
[-1.2246e+00, 2.0740e-02, 1.1575e-01, ..., -8.6021e-03,
2.6985e-01, 3.3356e-01],
...,
[-6.2913e-01, 4.0440e-02, 2.6098e-01, ..., 6.6779e-02,
5.6972e-02, 2.7191e-01],
[-4.4943e-01, 5.1174e-02, 2.1067e-01, ..., 1.4455e-01,
1.5576e-01, 5.3266e-01],
[-1.4698e-01, -2.7779e-01, 1.5242e-01, ..., -1.3010e-02,
8.5905e-02, 3.3301e-01]],
...,
[[ 9.2580e-01, 2.4907e-01, 2.6915e-01, ..., 7.6400e-01,
1.0161e-01, -9.4386e-01],
[-1.0068e+00, -5.6986e-01, -3.9543e-01, ..., -1.1729e+00,
-1.5192e+00, -1.2449e+00],
[-4.2482e-01, -3.0644e-01, -2.8197e-01, ..., -5.6594e-01,
-5.5613e-01, -1.1479e+00],
...,
[-6.5310e-01, -5.1246e-01, -3.3517e-01, ..., -1.4544e+00,
-1.2544e+00, -8.4719e-01],
[-2.2552e+00, -1.7921e+00, -1.3021e+00, ..., -1.3488e+00,
-1.3530e+00, -1.0795e+00],
[-6.2286e-01, -8.3976e-01, -4.2204e-01, ..., -2.6157e-01,
-3.6461e-01, -4.0483e-01]],
[[ 1.3223e-01, 2.0529e-01, 3.3839e-01, ..., -6.4060e-01,
5.7149e-01, -2.8878e-01],
[ 1.7824e-01, -6.6232e-01, -6.9501e-02, ..., 8.0325e-01,
3.3464e-01, -7.2057e-01],
[-1.6124e-01, -8.1110e-01, 5.8756e-02, ..., 3.4576e-01,
3.8532e-02, -6.9532e-01],
...,
[-2.4765e-01, -4.3270e-01, 2.5695e-01, ..., -2.1298e-01,
-2.5717e-01, -3.0182e-01],
[-5.4310e-01, -8.9105e-01, 8.3642e-02, ..., -2.0904e-02,
-3.9702e-02, -1.6840e-01],
[ 3.1980e-01, -3.0678e-02, -1.8728e-01, ..., -1.7253e-01,
-1.2611e-01, 1.6630e-01]],
[[-7.7351e-01, -6.2786e-02, 3.6733e-01, ..., -1.7761e-01,
4.4638e-01, 9.6518e-01],
[-1.1475e+00, -5.4375e-01, -5.5882e-01, ..., -8.4878e-01,
-1.8421e-01, 9.7614e-01],
[-8.5063e-01, -6.0192e-01, -3.7801e-01, ..., 1.5082e-03,
3.4214e-02, 7.8090e-01],
...,
[-6.8184e-01, -1.8945e-01, -1.0010e-01, ..., 8.0221e-01,
7.0181e-01, 1.0123e+00],
[-1.1666e+00, -5.4538e-01, -3.0837e-01, ..., -3.6424e-01,
-1.9758e-01, 1.7312e-01],
[-3.6444e-01, 3.0296e-03, -1.5856e-01, ..., -2.4692e-01,
-2.3990e-01, 3.0731e-01]]],
[[[-2.3545e-01, -5.1406e-01, -1.1650e-01, ..., -2.0162e-01,
2.0950e-01, -1.8072e-01],
[ 1.0503e-01, -1.4939e+00, -2.8277e+00, ..., -8.1385e-01,
-6.0940e-01, -6.4342e-01],
[ 2.4371e-01, -4.9755e+00, -6.7927e+00, ..., -1.3200e+00,
-8.5456e-01, -1.0069e+00],
...,
[ 1.6432e-01, -2.2268e+00, -2.1729e+00, ..., -1.2028e+00,
-1.1844e+00, -2.3403e+00],
[-8.5695e-01, -3.1923e+00, -5.4489e-01, ..., -6.9235e-01,
-6.4300e-01, -1.9891e+00],
[-8.3767e-01, -1.4657e+00, -1.6410e+00, ..., -1.6001e+00,
-1.7303e+00, -1.1087e+00]],
[[-7.6465e-01, -2.2476e+00, -1.5918e+00, ..., -1.7228e+00,
-2.4773e+00, -8.2525e-01],
[-3.2306e+00, -9.1352e+00, -1.1580e+01, ..., -3.3007e+00,
-4.4796e+00, -1.6764e+00],
[-5.4985e+00, -1.3419e+01, -1.6516e+01, ..., -6.8624e+00,
-4.9714e+00, -1.7026e+00],
...,
[-6.2572e+00, -1.1134e+01, -1.0373e+01, ..., -5.8742e+00,
-6.8304e+00, -3.0874e+00],
[-7.9946e+00, -8.1436e+00, -6.0690e+00, ..., -6.8166e+00,
-8.0584e+00, -2.8415e+00],
[ 2.1994e-01, 9.8908e-02, -2.8597e-01, ..., 3.5709e-01,
6.2987e-01, 3.2037e-01]],
[[-4.5638e-01, 1.0823e-01, 6.2984e-01, ..., 6.2758e-01,
3.6408e-01, 1.0948e+00],
[-5.7430e-01, -7.9776e-01, 4.0301e-01, ..., 3.6974e-01,
3.8793e-01, 4.7398e-01],
[-1.3219e+00, -4.3721e+00, -5.1583e+00, ..., 5.8309e-01,
2.1935e-01, 2.8267e-01],
...,
[-1.4612e+00, -9.0194e-01, 3.2103e-01, ..., -5.0953e-02,
-1.5892e-01, 1.2511e-01],
[-7.2828e-01, 3.1514e-01, -2.4182e+00, ..., -8.4399e-02,
-5.7630e-02, 1.0130e-01],
[-7.3818e-01, -1.1767e+00, -5.2703e+00, ..., -5.6542e-02,
-2.0293e-02, -5.0862e-01]],
...,
[[ 1.6336e-01, -3.4598e-01, -1.8086e-01, ..., 6.0434e-02,
-8.0548e-02, -7.5419e-01],
[-7.9836e-01, -4.5128e-01, -3.0800e-01, ..., -2.1862e-01,
-1.0183e+00, -9.1223e-01],
[ 9.2956e-03, 2.4245e+00, 2.4561e+00, ..., 2.4728e-02,
-2.7249e-01, -6.3679e-01],
...,
[-3.0549e-01, 1.8336e-01, -5.7737e-01, ..., -3.2707e-01,
-5.1918e-01, -1.2521e+00],
[-2.7541e+00, -3.1608e+00, -6.1014e+00, ..., -2.0851e+00,
-2.1919e+00, -2.3113e+00],
[-3.6598e-01, -2.9807e+00, -5.1606e+00, ..., -8.3038e-01,
-9.6857e-01, -1.5317e+00]],
[[-1.0631e-01, 5.0331e-02, 3.5819e-01, ..., 7.2624e-02,
5.2324e-01, -3.7924e-03],
[-2.6681e-01, 5.3644e-02, 3.6439e-01, ..., 9.8348e-02,
2.9195e-01, -2.0430e-01],
[-7.4350e-02, -1.0048e+00, -3.4490e+00, ..., -1.5669e+00,
-7.6660e-01, -2.0838e-01],
...,
[ 3.9369e-01, -7.7933e-01, -5.4391e-02, ..., 3.6800e-01,
1.1058e-01, -7.8760e-01],
[ 2.3040e-01, -2.0504e+00, -2.5807e+00, ..., 1.3708e-01,
-4.5805e-02, -9.3568e-01],
[ 7.5334e-01, 1.1613e-02, -2.5487e+00, ..., 3.6173e-01,
3.9919e-01, 4.4815e-01]],
[[-5.5404e-01, 3.5284e-01, 5.7055e-01, ..., 5.1581e-01,
4.3624e-01, 7.8718e-01],
[-8.7282e-01, -3.7275e-01, -1.1836e+00, ..., -2.3365e-01,
-8.2258e-02, 8.5659e-01],
[-8.7255e-01, -3.2401e+00, -4.6849e+00, ..., -2.6507e-02,
2.2677e-01, 7.8152e-01],
...,
[-1.7858e+00, -2.5015e+00, -3.4048e+00, ..., -6.1171e-01,
-2.0376e-01, 8.6246e-01],
[-3.1663e+00, -3.3966e+00, -1.9144e+00, ..., -9.8887e-01,
-6.5385e-01, 4.9624e-01],
[-1.6840e+00, -1.4093e-01, 1.7091e+00, ..., -2.9009e-01,
-1.7615e-01, 1.3082e+00]]],
[[[-3.1449e-01, -9.1223e-01, -4.6757e-01, ..., -7.4123e-01,
4.2962e-02, -1.6577e+00],
[-2.2619e-01, -1.1814e+00, -8.1224e-01, ..., -2.9943e+00,
-2.4716e+00, -1.6504e+00],
[-1.5200e+00, -2.1344e+00, -3.1167e+00, ..., -1.2942e+00,
-7.4421e-01, 5.4442e-01],
...,
[-4.6807e-01, -1.8752e+00, 2.1966e+00, ..., -1.3030e+01,
-1.0014e+01, -6.5341e+00],
[-1.5196e+00, -2.1240e+00, 1.8211e+00, ..., -2.4387e+01,
-2.2452e+01, -1.2431e+01],
[-1.4138e+00, -2.4641e+00, -9.3936e-01, ..., -1.7983e+01,
-1.7842e+01, -9.8642e+00]],
[[-1.3037e+00, -3.0620e+00, -2.4316e+00, ..., -1.3047e+00,
-5.1715e+00, -4.6451e-01],
[-2.3703e+00, -4.5706e+00, -3.5557e+00, ..., -9.0359e+00,
-8.5683e+00, -3.1969e+00],
[-9.3115e+00, -1.6517e+01, -1.6476e+01, ..., -3.6652e+00,
-3.4128e+00, -1.7330e+00],
...,
[-4.2408e+00, -6.0424e+00, -1.0034e+01, ..., -1.1835e+01,
-1.4580e+01, -5.0062e-01],
[-6.6209e+00, -7.7494e+00, -1.2050e+01, ..., -2.4715e+01,
-2.0223e+01, -1.5982e+00],
[-2.1584e-01, -1.0016e+00, 1.1489e+00, ..., 7.9290e-01,
8.4028e+00, 7.2038e+00]],
[[-5.5303e-01, -3.5205e-01, 5.7756e-02, ..., -1.0258e+00,
-3.6585e-01, -4.6470e-01],
[-4.5332e-01, -5.2706e-01, 3.4413e-01, ..., -2.3935e+00,
-3.0066e-01, -1.2437e+00],
[-1.1354e+00, 1.0023e+00, 2.3083e+00, ..., -1.6150e+00,
1.3694e-01, -3.6822e-01],
...,
[-1.4773e+00, -7.6507e-02, 2.9073e+00, ..., -3.1941e+01,
-2.2192e+01, -4.6375e+00],
[-1.1695e+00, 9.9779e-02, 3.0256e+00, ..., -4.4375e+01,
-2.7070e+01, -8.2953e+00],
[-1.2885e+00, -2.2872e-01, 1.4881e+00, ..., -2.9742e+01,
-1.9262e+01, -7.9114e+00]],
...,
[[-2.0888e-01, -5.2742e-01, -4.5898e-01, ..., -1.1625e+00,
2.7554e-01, -7.7448e-01],
[ 1.0827e-01, -7.6871e-02, -8.1408e-02, ..., -8.9176e-01,
-3.4318e-01, -2.2442e+00],
[-8.2601e-02, -7.9752e-01, -1.1261e-01, ..., 2.3720e+00,
6.8399e-01, -1.8085e+00],
...,
[-6.3068e-01, -8.2698e-01, 1.2359e+00, ..., -2.4802e+01,
-2.0136e+01, -1.8392e+01],
[-3.3953e+00, -3.5387e+00, -2.2140e+00, ..., -1.6497e+01,
-1.8680e+01, -2.0587e+01],
[-5.2969e-01, -1.7409e+00, -4.9499e-01, ..., -2.6947e+00,
-9.4150e+00, -1.2498e+01]],
[[-1.4424e-01, -1.7340e-01, -1.9462e-01, ..., -1.9400e+00,
-1.6946e+00, -1.8331e+00],
[-5.6710e-01, 1.3382e-02, 2.5433e-01, ..., -1.4837e+00,
-3.3441e+00, -3.2128e+00],
[-3.7945e-01, 1.0353e-01, -1.7729e-01, ..., 1.8429e+00,
6.5602e-01, 1.0956e+00],
...,
[-8.9374e-02, -1.1907e+00, 3.3323e+00, ..., -1.4956e+01,
-8.1158e+00, -2.5764e+00],
[-4.2420e-01, -1.7957e+00, 1.8547e+00, ..., -2.0188e+01,
-1.2834e+01, -3.3918e+00],
[ 4.7569e-01, -1.3921e-01, -7.2549e-01, ..., -6.4063e+00,
-2.3217e+00, 7.8848e-01]],
[[-6.0940e-01, 9.2167e-02, 2.0234e-01, ..., -1.5179e+00,
-8.7435e-01, -6.5178e-01],
[-1.0511e+00, -9.8238e-02, -2.0328e-01, ..., -2.0440e+00,
-1.1313e+00, -7.6134e-01],
[-1.0189e+00, -1.3840e+00, -2.1736e+00, ..., -8.1012e-01,
-8.5618e-01, -2.4590e-01],
...,
[-1.5731e+00, -1.1277e+00, -2.6841e+00, ..., -2.9278e+01,
-2.0157e+01, -6.7564e+00],
[-2.4631e+00, -2.1677e+00, -3.3321e+00, ..., -2.3484e+01,
-2.2143e+01, -7.9012e+00],
[-1.1727e+00, -9.8227e-01, -1.4129e+00, ..., -5.1243e+00,
-6.6649e+00, -4.3146e-01]]],
[[[-3.6886e-01, -1.1723e+00, -1.0411e+00, ..., -1.8065e-01,
4.8425e-01, 4.0224e-01],
[-8.4428e-01, -2.1090e+00, -1.8167e+00, ..., -9.4143e-02,
-1.7029e+00, -3.0044e-01],
[-4.2576e+00, -1.9580e+00, -1.1485e+00, ..., -5.2692e+00,
-5.5208e+00, 1.8878e+00],
...,
[-3.6044e-01, -5.2473e+00, -3.7976e+00, ..., -4.1530e-01,
-1.7905e-01, -2.7327e-01],
[ 1.2432e+00, -1.5214e+00, 1.4041e-01, ..., -6.0496e-02,
2.0244e-01, -1.1679e-01],
[ 6.4889e-01, -1.8504e-01, -5.3122e-01, ..., -1.0182e+00,
-1.1151e+00, -3.9172e-01]],
[[-2.1226e+00, -4.0444e+00, -3.6977e+00, ..., -9.3872e-01,
-1.4677e+00, -1.9473e-01],
[-7.2478e+00, -1.2263e+01, -1.1118e+01, ..., -7.3740e+00,
-8.4405e+00, -4.2937e+00],
[-6.6599e+00, -1.3866e+01, -1.6388e+01, ..., -1.6043e+01,
-1.2520e+01, -6.2183e+00],
...,
[-1.1009e+01, -1.7961e+01, -1.3587e+01, ..., -1.7711e+00,
-1.9687e+00, -5.9591e-01],
[-5.9897e+00, -8.4512e+00, -7.5431e+00, ..., -2.9137e+00,
-3.6658e+00, -1.1360e+00],
[ 7.5930e-01, 8.1017e-01, 9.1666e-01, ..., 3.9965e-01,
4.0506e-01, 8.5813e-02]],
[[-6.1007e-01, 3.8420e-01, 7.6251e-01, ..., 1.0105e+00,
4.9197e-01, 1.2414e+00],
[-1.4227e+00, -1.8867e-01, 1.6771e-01, ..., 3.7628e-01,
-2.2215e-01, 7.2009e-01],
[-3.9199e+00, -1.3395e+00, -2.2653e-01, ..., 1.0667e+00,
-1.7249e+00, -2.1433e-01],
...,
[-1.1486e+01, -9.5943e+00, 2.7959e+00, ..., 1.3518e-01,
1.7325e-01, 4.8252e-01],
[-3.7506e+00, -2.5292e+00, 2.9742e+00, ..., 1.6806e-01,
1.8664e-01, 6.3965e-01],
[ 2.7075e-01, 4.3821e-01, 1.3691e+00, ..., -8.7921e-02,
9.5175e-02, 3.6968e-01]],
...,
[[ 6.4380e-01, 3.9360e-03, 3.2943e-01, ..., -6.4265e-02,
-3.2725e-01, -9.6075e-01],
[-5.3466e-01, 2.4402e-01, 8.5921e-01, ..., -6.3616e-01,
-1.1623e+00, -7.9448e-01],
[ 2.2002e-01, -1.0473e+00, -9.2257e-01, ..., 1.4141e-01,
-2.3275e+00, -5.1113e+00],
...,
[ 2.4804e-01, -5.6435e+00, -8.0661e+00, ..., -1.6357e-01,
-2.9968e-01, -4.6768e-01],
[-2.5909e+00, -4.9353e+00, -8.3139e+00, ..., -8.7848e-01,
-9.4854e-01, -8.7185e-01],
[-1.7143e+00, -2.7337e+00, -3.0976e+00, ..., -7.5347e-02,
-1.8716e-01, -3.7692e-01]],
[[-4.4181e-02, -3.2293e-01, -9.2381e-02, ..., 3.6504e-01,
7.5135e-01, 3.3373e-01],
[ 4.3115e-01, -1.0049e+00, -8.8464e-01, ..., 8.9634e-01,
-7.7205e-01, -2.1306e+00],
[-7.1832e-01, -1.7979e+00, 2.1069e-01, ..., -1.4094e+00,
-4.7290e+00, -2.1198e+00],
...,
[-2.3677e+00, -4.8758e+00, -1.1367e+00, ..., -1.3520e-01,
-1.5750e-01, 9.0061e-02],
[-1.2081e+00, -5.9908e+00, -2.2309e+00, ..., -4.3397e-01,
-3.9432e-01, -1.6074e-01],
[ 1.2855e+00, -1.2291e+00, -1.1083e+00, ..., -1.7671e-01,
-2.0000e-01, 2.6806e-01]],
[[-6.8682e-01, -2.2257e-01, -1.1303e-01, ..., 6.3540e-01,
3.7508e-01, 6.4729e-01],
[-1.7430e+00, -6.3024e-01, -1.1013e+00, ..., -2.7374e-01,
-3.6015e-01, 1.2042e-02],
[-3.6069e+00, -5.4224e+00, -4.3569e+00, ..., -3.4295e+00,
-3.8942e+00, -4.1798e+00],
...,
[-3.2872e-01, -4.3346e+00, -1.1524e+01, ..., 8.2275e-02,
-6.0196e-02, 2.1014e-01],
[-1.2039e+00, -2.4486e+00, -5.4694e+00, ..., -8.6841e-02,
-2.0944e-01, -8.8899e-02],
[-5.1553e-01, 2.0764e-01, 9.1556e-02, ..., -1.5441e-01,
-2.5803e-01, 2.0204e-01]]]], device='cuda:0'),
tensor([[[[ 3.5505e-02, -2.6254e-01, -6.0668e-02, ..., -9.6001e-01,
-5.1026e-01, -1.4528e+00],
[ 2.8057e-01, -9.6138e-01, -7.2087e-01, ..., -9.1113e-01,
-1.7793e+00, -2.8127e+00],
[ 2.1155e-01, -6.0715e-01, -5.2322e-01, ..., -1.0620e+00,
-3.0133e+00, -3.0789e+00],
...,
[ 4.3627e-01, -4.6226e+00, -7.6619e+00, ..., -1.0203e+00,
-1.0695e+00, -1.9577e+00],
[-2.2211e-01, -5.3546e-01, -3.2998e+00, ..., -5.0243e-01,
-6.1462e-01, -1.7486e+00],
[-5.2032e-01, 1.8854e-01, -8.1108e-01, ..., -1.0086e+00,
-1.3884e+00, -6.9006e-01]],
[[-8.3285e-01, -2.2410e+00, -2.0167e+00, ..., -3.8592e+00,
-5.5364e+00, -1.7737e+00],
[-2.1039e+00, -3.7599e+00, -3.1579e+00, ..., -7.7167e+00,
-1.1137e+01, -4.8456e+00],
[-1.7382e+00, -2.6213e+00, -2.6972e+00, ..., -8.5347e+00,
-9.8385e+00, -5.3189e+00],
...,
[-3.5926e+00, -1.0978e+01, -1.3278e+01, ..., -5.1833e+00,
-6.3643e+00, -2.4897e+00],
[-3.8052e+00, -8.5223e+00, -9.1830e+00, ..., -6.1143e+00,
-7.6191e+00, -2.4010e+00],
[-7.7037e-02, -1.0398e+00, -1.0923e-01, ..., 3.6578e-01,
4.2212e-01, 2.6390e-01]],
[[-2.0435e-01, 1.9866e-01, 7.4108e-01, ..., 7.1379e-01,
3.0016e-01, 1.6129e+00],
[ 3.6096e-02, -3.4733e-01, 5.6645e-01, ..., 4.9851e-01,
-3.3864e-01, -1.1445e-01],
[-1.6990e-01, -2.3918e-01, 2.1532e-01, ..., -2.9169e-01,
-9.8604e-01, -6.8780e-01],
...,
[ 3.0163e-01, -8.3793e+00, -2.0779e+01, ..., 1.9427e-02,
-5.8110e-02, -1.4081e-01],
[ 1.5438e-02, -7.0649e+00, -1.6650e+01, ..., -8.9961e-02,
-1.0242e-02, -2.0779e-01],
[ 5.4392e-01, -5.7963e-02, -3.0155e+00, ..., 6.1040e-02,
1.5179e-01, -6.7644e-01]],
...,
[[ 6.3061e-01, -1.7828e-01, 1.0315e-01, ..., 4.4506e-01,
2.9521e-01, -1.3789e+00],
[-4.7770e-01, -4.9094e-01, -4.6807e-01, ..., -2.8891e-01,
-1.1206e+00, -1.6844e+00],
[-3.3353e-01, -1.7213e-01, -1.8403e-01, ..., 8.8521e-01,
8.1808e-01, -2.5308e+00],
...,
[ 7.3594e-02, 2.3735e+00, -2.4380e+00, ..., -2.0094e-02,
-9.4513e-02, -9.7476e-01],
[-2.1099e+00, 5.2358e-01, -2.0918e+00, ..., -1.3799e+00,
-1.4077e+00, -1.7634e+00],
[-9.0536e-01, -2.3673e-01, -5.5540e-01, ..., -3.9363e-01,
-4.8026e-01, -1.1547e+00]],
[[ 1.8398e-01, 2.5697e-01, 6.1347e-01, ..., -3.8385e-02,
4.3388e-01, -3.2868e-01],
[ 1.5301e-01, 2.2899e-01, 4.4210e-01, ..., 5.1246e-01,
2.7556e-01, -1.6692e+00],
[-1.0903e-01, 9.9660e-02, 2.7959e-01, ..., 8.7232e-01,
-4.5699e-01, -1.6165e+00],
...,
[ 1.5756e-01, -1.7237e+00, -5.5611e+00, ..., 2.1629e-01,
1.9640e-01, -4.8003e-01],
[-5.2367e-01, -1.9642e+00, -4.3373e+00, ..., -3.7771e-02,
-1.1735e-01, -7.9643e-01],
[ 6.9975e-01, 5.8049e-01, -1.1105e+00, ..., 3.6789e-01,
3.6059e-01, 3.3876e-01]],
[[-8.0481e-01, 2.4154e-01, 2.8745e-01, ..., -2.8432e-02,
-6.1062e-02, 6.1754e-01],
[-1.1137e+00, 2.0210e-01, -3.8427e-01, ..., -1.0395e+00,
-8.5924e-01, 5.8999e-01],
[-9.7273e-01, 3.0235e-01, -1.1789e-01, ..., -1.2445e+00,
-2.3024e+00, -6.6298e-01],
...,
[-1.6867e+00, -2.6069e+00, -9.0311e+00, ..., -5.0918e-01,
-5.5252e-01, 6.3092e-01],
[-1.9275e+00, -1.3853e+00, -4.8639e+00, ..., -7.3407e-01,
-6.9426e-01, 4.4744e-01],
[-9.6064e-01, -1.2381e+00, -5.1269e-01, ..., -4.5677e-01,
-4.3644e-01, 1.0535e+00]]],
[[[-7.9930e-01, -1.4182e+00, -1.3653e+00, ..., 2.1592e-02,
6.1477e-01, 5.3734e-01],
[-3.6387e-01, -2.3209e+00, -2.1738e+00, ..., -4.5597e-01,
-2.3349e-01, 1.2359e-01],
[-3.9691e-01, -1.9794e+00, -1.9015e+00, ..., -3.0620e-01,
-1.1493e-01, -8.7758e-02],
...,
[-5.0973e-01, -2.0531e+00, -1.8714e+00, ..., -2.9706e-01,
-8.0760e-02, -1.3413e-01],
[-1.1759e+00, -1.8475e+00, -1.1452e+00, ..., 2.8504e-02,
2.7781e-01, -2.9369e-02],
[-1.0618e+00, -1.9324e+00, -1.4308e+00, ..., -1.0405e+00,
-1.0810e+00, -3.7836e-01]],
[[-2.5605e+00, -4.9126e+00, -4.3282e+00, ..., -4.7699e-01,
-1.0331e+00, -1.0290e-02],
[-5.2446e+00, -7.8086e+00, -6.5524e+00, ..., -1.7765e+00,
-2.6478e+00, -6.7079e-01],
[-4.3974e+00, -6.1737e+00, -5.4623e+00, ..., -1.2984e+00,
-1.4724e+00, -3.9740e-01],
...,
[-4.4094e+00, -6.1567e+00, -5.5458e+00, ..., -1.2986e+00,
-1.5369e+00, -3.7178e-01],
[-6.5784e+00, -7.7597e+00, -6.5508e+00, ..., -2.4270e+00,
-3.2849e+00, -9.4188e-01],
[-7.3533e-02, -1.0381e+00, 4.7419e-01, ..., 3.3771e-01,
3.7440e-01, 8.5859e-02]],
[[-6.1437e-01, 6.8733e-01, 1.0964e+00, ..., 7.9833e-01,
2.8900e-01, 1.0685e+00],
[-9.5387e-01, 2.0833e-02, 3.9660e-01, ..., 3.9070e-01,
6.3134e-02, 4.6737e-01],
[-1.6503e+00, -9.4820e-02, -4.7963e-02, ..., 7.1238e-02,
8.9898e-02, 4.1820e-01],
...,
[-1.7695e+00, -2.2187e-01, -8.7580e-02, ..., 8.2757e-02,
4.0122e-02, 4.2219e-01],
[-1.4824e+00, 1.1535e-03, -7.3856e-04, ..., 1.5659e-01,
9.5809e-02, 6.0950e-01],
[-1.3379e+00, -5.7021e-01, -2.7202e-01, ..., -6.5053e-02,
7.5350e-02, 3.6981e-01]],
...,
[[ 1.2327e+00, -3.5325e-01, -2.2842e-01, ..., -9.0156e-02,
-3.7824e-01, -9.2773e-01],
[-6.5843e-01, -1.0070e+00, -8.5018e-01, ..., -6.2439e-01,
-1.2725e+00, -8.3959e-01],
[-6.4659e-01, -1.0211e+00, -7.3958e-01, ..., -1.0936e-01,
-1.9660e-01, -3.9621e-01],
...,
[-8.8265e-01, -1.1149e+00, -7.4568e-01, ..., -8.1845e-02,
-1.9173e-01, -3.6664e-01],
[-3.1599e+00, -3.0191e+00, -2.0361e+00, ..., -8.2687e-01,
-8.7609e-01, -7.9051e-01],
[-6.8029e-01, -1.7066e+00, -1.0076e+00, ..., -1.6558e-02,
-1.2395e-01, -3.2720e-01]],
[[ 1.0460e-01, 8.5371e-02, 5.9844e-01, ..., 3.4434e-01,
6.7610e-01, 2.9974e-01],
[ 1.1069e-01, -7.3405e-01, 1.2217e-01, ..., -4.1604e-03,
1.0968e-01, -5.7482e-02],
[ 1.1041e-02, -7.3161e-01, 2.8540e-01, ..., -1.3056e-01,
-6.3674e-02, 1.6353e-01],
...,
[-1.1098e-01, -9.1780e-01, 2.3236e-01, ..., -1.3226e-01,
-1.1030e-01, 1.4629e-01],
[-4.7292e-01, -1.5812e+00, -1.1257e-02, ..., -3.5876e-01,
-3.0012e-01, -8.0925e-02],
[ 5.5871e-01, 1.3044e-01, 2.2992e-01, ..., -2.2856e-01,
-2.1188e-01, 3.0580e-01]],
[[-1.5593e+00, -6.1064e-01, 2.5441e-02, ..., 5.3845e-01,
3.6400e-01, 5.4811e-01],
[-2.1218e+00, -3.5496e-01, 8.3562e-03, ..., -7.3821e-02,
-1.3943e-01, 4.0949e-01],
[-1.7837e+00, -3.6179e-01, 2.5374e-01, ..., 1.0995e-02,
-1.4492e-01, 7.1324e-02],
...,
[-1.7583e+00, -5.1952e-01, 2.5893e-01, ..., 2.8280e-02,
-4.7657e-02, 9.1463e-02],
[-2.3676e+00, -1.3775e+00, -1.8984e-01, ..., -1.4271e-01,
-2.4162e-01, -2.1183e-01],
[-1.0931e+00, -4.2331e-01, 3.1636e-01, ..., -1.0140e-01,
-2.6317e-01, 1.1495e-01]]],
[[[ 2.3238e-01, -2.6429e+00, -3.0835e+00, ..., -1.2945e+00,
-9.4646e-01, -1.8240e+00],
[-1.2943e+00, -6.1456e+00, -2.4977e+00, ..., -1.8153e+00,
-2.0052e+00, -2.0245e+00],
[-2.6766e+00, -3.5599e+00, 1.5361e+00, ..., -2.2719e+00,
-1.4813e+00, -2.1990e+00],
...,
[-4.8113e-02, -5.7758e-01, -2.9917e+00, ..., -6.1152e-01,
-5.2139e-01, -1.0573e+00],
[-1.1393e-02, -6.4056e-01, -3.9913e+00, ..., -2.1200e-01,
-1.5348e-01, -1.1063e+00],
[-1.1665e-01, -9.2011e-01, -1.5364e+00, ..., -8.4905e-01,
-1.0883e+00, -6.4176e-01]],
[[-8.7669e+00, -1.2422e+01, -1.1463e+01, ..., -4.9838e+00,
-6.2076e+00, -2.4593e+00],
[-1.3076e+01, -1.6757e+01, -1.5772e+01, ..., -8.0352e+00,
-9.5829e+00, -4.5060e+00],
[-7.6758e+00, -1.0613e+01, -1.0722e+01, ..., -9.4290e+00,
-8.5018e+00, -3.5283e+00],
...,
[-2.2331e+00, -7.8201e+00, -1.3565e+01, ..., -3.3980e+00,
-3.8900e+00, -1.7526e+00],
[-4.5358e+00, -9.6019e+00, -1.2127e+01, ..., -5.2245e+00,
-6.3661e+00, -2.5040e+00],
[ 7.0176e-01, 4.7426e-01, 9.3758e-01, ..., 5.6570e-01,
6.3523e-01, 6.0565e-03]],
[[-1.2773e+00, 8.0540e-02, 8.3059e-01, ..., 8.0759e-01,
6.4963e-01, 1.9246e+00],
[-2.7505e+00, -4.6496e+00, -3.9898e+00, ..., 5.1349e-01,
1.4653e-02, 9.8909e-02],
[-3.1586e+00, -8.8963e+00, -7.5849e+00, ..., 8.4690e-01,
9.2206e-01, -1.3689e-02],
...,
[-9.7554e-01, -2.8107e-01, -2.2155e+00, ..., 2.5034e-02,
7.6968e-02, 3.1491e-01],
[-1.0573e+00, 4.5438e-01, -5.3584e-01, ..., -4.1661e-03,
4.4239e-02, 2.5326e-01],
[-6.0743e-01, 1.7583e+00, -1.3227e+00, ..., 1.7028e-02,
8.4507e-02, -6.2393e-02]],
...,
[[ 1.5667e+00, 1.5332e-01, 9.0118e-01, ..., 2.2987e-01,
-1.2479e-01, -2.2219e+00],
[ 1.0967e+00, -7.9594e-01, -3.5104e+00, ..., -5.5254e-01,
-1.5500e+00, -2.2029e+00],
[-4.1228e-01, -6.9823e+00, -1.1546e+01, ..., 7.3070e-01,
-2.7779e-01, -1.5143e+00],
...,
[-4.8017e-01, 1.5269e-01, 1.4440e+00, ..., 8.0698e-02,
-1.7699e-01, -6.2884e-01],
[-2.0082e+00, -1.3538e-01, -8.3410e-01, ..., -1.0109e+00,
-1.2207e+00, -1.1607e+00],
[-2.9335e-01, -3.8975e-01, -1.3485e+00, ..., -2.1134e-01,
-3.5732e-01, -6.2479e-01]],
[[ 1.2543e+00, -4.3052e-01, -1.7762e+00, ..., 5.5884e-01,
9.6517e-01, -2.3852e-02],
[ 2.1119e-01, -5.3027e+00, -4.6364e+00, ..., 5.0645e-01,
3.9726e-01, -9.2788e-01],
[-1.0223e+00, -7.3345e+00, -5.9155e+00, ..., -7.0624e-01,
-1.9101e-01, -4.8943e-01],
...,
[-3.2867e-01, 3.0967e-01, -3.5285e-01, ..., 7.8918e-02,
7.8979e-02, -2.3522e-01],
[-6.1089e-01, -3.9128e-01, -1.8673e+00, ..., -3.1720e-01,
-2.9094e-01, -7.3532e-01],
[ 5.2514e-01, 1.1764e+00, 1.0714e-01, ..., 2.4245e-01,
1.9668e-01, 2.9216e-01]],
[[-1.7719e+00, -1.1919e+00, -9.6088e-01, ..., -4.8297e-01,
-3.0471e-01, 8.7216e-01],
[-5.1729e+00, -4.8687e+00, -5.5432e+00, ..., -8.6262e-01,
-1.6661e-01, 1.6921e+00],
[-5.0039e+00, -3.9174e+00, -8.2334e+00, ..., -3.5564e-01,
7.3421e-01, 1.4058e+00],
...,
[-4.8509e-01, -4.8374e-02, -2.4047e+00, ..., 1.3690e-01,
2.2926e-01, 8.0878e-01],
[-7.8904e-01, -1.1374e+00, -3.8285e+00, ..., -1.5255e-01,
6.9499e-02, 4.9911e-01],
[-3.7462e-01, -1.4915e+00, -9.5991e-01, ..., -2.1757e-01,
-2.0068e-01, 6.0311e-01]]],
[[[-4.7549e-01, -3.5987e+00, -1.3968e+00, ..., -1.3120e+00,
-9.1010e-01, -1.7755e+00],
[-1.0930e+00, -5.6569e+00, -8.8925e-01, ..., -2.0994e+00,
-2.0952e+00, -2.0383e+00],
[-8.9115e-01, -5.7013e+00, -8.1325e+00, ..., -1.1471e+00,
-1.6277e+00, -2.1611e+00],
...,
[-2.0648e-02, -1.1874e+00, -1.1313e+00, ..., -5.3423e-01,
-2.6579e-01, -6.0058e-01],
[-6.4779e-01, -1.1705e+00, -6.5974e-01, ..., 4.0430e-01,
4.1443e-01, -2.0617e-01],
[-9.2381e-01, -1.8777e+00, -9.8815e-01, ..., -8.9663e-01,
-1.0076e+00, -4.0757e-01]],
[[-4.8539e+00, -8.1063e+00, -1.0742e+01, ..., -4.9479e+00,
-6.1453e+00, -2.4274e+00],
[-8.3593e+00, -1.0738e+01, -9.3868e+00, ..., -9.4118e+00,
-9.9173e+00, -4.5114e+00],
[-1.0187e+01, -1.9131e+01, -1.9287e+01, ..., -9.2882e+00,
-7.7719e+00, -3.6551e+00],
...,
[-2.0893e+00, -3.3229e+00, -3.3646e+00, ..., -2.7464e+00,
-2.8902e+00, -1.2205e+00],
[-3.5953e+00, -4.6066e+00, -3.4855e+00, ..., -2.0083e+00,
-3.3197e+00, -9.7532e-01],
[ 3.1877e-01, -6.2753e-01, 8.0271e-02, ..., 1.8214e-01,
3.6713e-01, 1.9626e-01]],
[[-9.9960e-01, -6.9093e-01, 1.3936e+00, ..., 8.6611e-01,
6.3819e-01, 1.9675e+00],
[-1.9968e+00, -5.9609e+00, -1.0979e+01, ..., 1.0785e+00,
2.7330e-02, 1.0908e-01],
[-4.5396e+00, -1.0283e+01, -1.3578e+01, ..., 1.3903e+00,
2.7026e-01, -1.6305e-03],
...,
[-8.9922e-01, -3.0644e-01, -1.4936e-01, ..., 1.2000e-01,
1.9040e-01, 3.7721e-01],
[-5.6999e-01, -2.6865e-01, -3.4427e-01, ..., 3.6282e-01,
3.3470e-01, 5.6920e-01],
[-1.8329e-01, -4.5393e-01, -7.2460e-01, ..., -2.3957e-01,
7.4385e-03, 2.9142e-01]],
...,
[[ 2.1692e+00, 1.2303e+00, -6.9169e-01, ..., 2.6707e-01,
-1.0130e-01, -2.2351e+00],
[ 2.2753e-01, -1.6382e+00, -3.9736e+00, ..., -9.9582e-01,
-1.6340e+00, -2.2280e+00],
[-9.3467e-01, -3.4449e+00, -7.9996e+00, ..., -1.6717e+00,
-7.1624e-01, -1.3470e+00],
...,
[-7.7936e-01, -2.1340e-01, 3.3615e-01, ..., -2.4434e-01,
-5.0512e-01, -6.1003e-01],
[-2.6908e+00, -1.9642e+00, -1.9495e+00, ..., -1.1406e+00,
-1.2559e+00, -9.7923e-01],
[-7.3761e-01, -1.0631e+00, -9.6454e-01, ..., -4.3717e-01,
-3.9830e-01, -4.2360e-01]],
[[ 6.2061e-01, 1.3577e-01, 1.7106e+00, ..., 5.6240e-01,
1.0404e+00, 3.6833e-02],
[ 5.2750e-01, -3.0329e+00, -2.4091e+00, ..., -1.2109e-01,
1.8786e-01, -8.8176e-01],
[-7.1423e-01, -4.6019e+00, -8.9180e+00, ..., 5.0385e-01,
2.7759e-02, -5.7028e-01],
...,
[-3.0495e-01, -7.4662e-01, 5.9540e-02, ..., 4.2201e-01,
3.1005e-01, -7.3596e-02],
[-7.4713e-01, -1.6379e+00, -9.0026e-01, ..., 3.8379e-01,
9.9699e-02, -1.8061e-01],
[ 3.9194e-01, -2.3787e-01, -1.1230e+00, ..., -2.7518e-01,
-1.5516e-01, 1.3086e-01]],
[[-2.5488e+00, -2.4467e+00, -3.8239e+00, ..., -5.3499e-01,
-3.4769e-01, 8.6806e-01],
[-3.5090e+00, -3.6572e+00, -6.2722e+00, ..., -1.0978e+00,
-2.2898e-01, 1.7105e+00],
[-1.8474e+00, -2.5276e+00, -7.8541e+00, ..., -1.5991e+00,
-4.1737e-01, 1.4351e+00],
...,
[-7.5994e-01, -5.5385e-01, -9.5429e-01, ..., -1.0627e-01,
9.1203e-02, 7.2097e-01],
[-1.3491e+00, -1.1515e+00, -1.8677e+00, ..., -3.1470e-01,
-1.8962e-01, 2.9132e-01],
[-4.9766e-01, -2.1400e-01, -1.3936e-01, ..., -3.4523e-01,
-3.0375e-01, 4.0587e-01]]],
[[[-3.6792e-01, -1.1138e+00, -8.3201e-01, ..., 6.1529e-03,
5.9614e-01, 4.2375e-01],
[ 1.3966e-01, -1.9138e+00, -1.3204e+00, ..., -4.0411e-01,
-3.4782e-01, 1.2341e-01],
[ 1.5586e-01, -1.4341e+00, -9.5921e-01, ..., -4.9155e-01,
-1.0544e+00, -2.7940e-01],
...,
[ 7.5888e-02, -1.0948e+00, -9.4536e-01, ..., -2.7266e-01,
-7.1259e-02, -7.6384e-02],
[ 5.2131e-02, -7.8285e-01, -5.0762e-01, ..., 6.5887e-02,
2.6368e-01, 4.9582e-02],
[-1.0456e-01, -9.9538e-01, -9.8663e-01, ..., -1.2133e+00,
-1.3097e+00, -4.8207e-01]],
[[-2.0937e+00, -3.9958e+00, -3.6101e+00, ..., -6.2343e-01,
-1.3359e+00, -2.8928e-01],
[-3.9953e+00, -6.3759e+00, -5.5312e+00, ..., -3.2824e+00,
-4.5017e+00, -2.3263e+00],
[-3.1214e+00, -4.8743e+00, -4.6416e+00, ..., -3.1818e+00,
-3.8621e+00, -2.9366e+00],
...,
[-2.9805e+00, -4.0994e+00, -4.5377e+00, ..., -1.4226e+00,
-1.7054e+00, -4.3269e-01],
[-5.1129e+00, -5.8561e+00, -5.6846e+00, ..., -2.7194e+00,
-3.7159e+00, -1.0751e+00],
[ 6.6280e-01, -5.7088e-02, 5.1162e-01, ..., 3.8360e-01,
3.8289e-01, 3.8659e-02]],
[[-5.7958e-01, 4.0656e-01, 8.1513e-01, ..., 7.5908e-01,
2.7854e-01, 9.9223e-01],
[-9.0316e-01, 1.1700e-01, 6.7542e-01, ..., 2.2647e-01,
-5.3047e-01, 1.4318e-02],
[-1.2855e+00, 8.3762e-03, 8.5944e-02, ..., -4.2722e-01,
-1.6058e+00, -2.7815e-01],
...,
[-9.4573e-01, -2.9630e-01, 1.8739e-02, ..., 1.5558e-01,
1.3342e-01, 4.3981e-01],
[-1.0475e+00, -5.7962e-02, -2.8161e-02, ..., 9.2439e-02,
1.6586e-01, 6.4994e-01],
[-4.1734e-01, 1.9903e-01, 1.7080e-01, ..., -1.5400e-01,
4.3327e-02, 4.2681e-01]],
...,
[[ 6.7277e-01, 1.0671e-01, 2.5914e-01, ..., -5.1529e-02,
-2.9913e-01, -8.8253e-01],
[-7.2698e-01, -3.9526e-01, -1.1787e-01, ..., -4.8736e-01,
-9.0866e-01, -6.9517e-01],
[-4.7495e-01, -3.7343e-01, -3.2297e-02, ..., 4.1159e-01,
-3.1115e-01, -1.1739e+00],
...,
[-6.5419e-01, -4.7289e-01, -5.9250e-02, ..., -1.6960e-01,
-3.1873e-01, -5.4328e-01],
[-2.1385e+00, -1.7596e+00, -1.2344e+00, ..., -9.7625e-01,
-1.1311e+00, -1.0479e+00],
[-4.4145e-01, -9.2200e-01, -3.0138e-01, ..., -5.6974e-02,
-1.3595e-01, -4.3236e-01]],
[[-6.2687e-03, -2.5134e-01, -3.2518e-02, ..., 3.4321e-01,
6.7252e-01, 9.7782e-02],
[ 1.6864e-01, -9.5631e-01, -1.1775e-01, ..., 1.5031e-01,
8.2704e-02, -6.1563e-01],
[-9.7744e-02, -9.5748e-01, 5.5345e-03, ..., -2.2583e-01,
-5.7332e-01, -7.5351e-01],
...,
[ 1.2134e-01, -4.4589e-01, 1.2613e-01, ..., -1.1884e-01,
-1.6949e-01, 2.4029e-02],
[-4.5690e-01, -1.6783e+00, -2.6913e-01, ..., -3.8944e-01,
-4.9000e-01, -1.8970e-01],
[ 7.7485e-01, 1.5609e-02, 2.3869e-01, ..., -2.6373e-01,
-3.1631e-01, 2.4426e-01]],
[[-6.3701e-01, -1.0938e-01, -5.0632e-03, ..., 4.2796e-01,
2.4395e-01, 4.1300e-01],
[-1.0126e+00, -6.8639e-01, -9.1678e-01, ..., -6.4263e-02,
-6.8917e-01, -3.1973e-01],
[-6.8054e-01, -6.4489e-01, -5.3982e-01, ..., -2.5563e-01,
-1.3766e+00, -1.4178e+00],
...,
[-9.0762e-01, -2.9257e-01, -5.2771e-01, ..., 7.6349e-02,
-7.8940e-02, 5.6489e-02],
[-1.0364e+00, -5.7141e-01, -5.3330e-01, ..., -7.5894e-02,
-2.9190e-01, -3.0875e-01],
[-5.5669e-01, -6.7646e-01, -3.1239e-01, ..., -1.7333e-01,
-3.1314e-01, 6.2313e-02]]]], device='cuda:0'),
tensor([[[[-7.2599e-01, -1.4001e+00, -1.3557e+00, ..., -2.2945e-02,
5.8223e-01, 5.1851e-01],
[-2.8751e-01, -2.4415e+00, -2.0799e+00, ..., -4.1673e-01,
-2.9488e-01, 1.0558e-01],
[-3.8496e-01, -2.1291e+00, -1.7686e+00, ..., -2.7919e-01,
-1.7508e-01, -1.2960e-01],
...,
[-5.0093e-01, -1.8820e+00, -1.8506e+00, ..., -3.4310e-01,
-9.1811e-02, -1.1238e-01],
[-9.0411e-01, -1.6321e+00, -1.1372e+00, ..., 3.3778e-02,
2.7823e-01, -2.1103e-02],
[-7.6032e-01, -1.6166e+00, -1.2294e+00, ..., -1.0599e+00,
-1.0928e+00, -3.9400e-01]],
[[-2.3562e+00, -4.7482e+00, -4.5778e+00, ..., -5.8450e-01,
-1.1163e+00, -3.5794e-02],
[-5.3663e+00, -8.1213e+00, -6.8637e+00, ..., -1.9262e+00,
-2.7699e+00, -7.5370e-01],
[-4.3463e+00, -6.2470e+00, -5.5393e+00, ..., -1.3173e+00,
-1.5039e+00, -4.5406e-01],
...,
[-4.3577e+00, -6.0976e+00, -5.5039e+00, ..., -1.3339e+00,
-1.5599e+00, -3.6000e-01],
[-6.3430e+00, -7.3783e+00, -6.2610e+00, ..., -2.4305e+00,
-3.3204e+00, -9.5784e-01],
[ 2.0297e-02, -7.6915e-01, 3.9629e-01, ..., 3.5106e-01,
3.6748e-01, 8.3401e-02]],
[[-6.0201e-01, 5.8420e-01, 1.1454e+00, ..., 7.8645e-01,
3.1695e-01, 1.0856e+00],
[-9.2020e-01, -4.7870e-02, 4.1383e-01, ..., 3.7701e-01,
5.9965e-02, 5.0272e-01],
[-1.6396e+00, -1.5526e-01, 3.4001e-02, ..., 1.0639e-02,
7.5169e-02, 4.4548e-01],
...,
[-1.6847e+00, -8.7741e-02, -1.2866e-01, ..., 6.7154e-02,
7.4336e-02, 4.3160e-01],
[-1.5299e+00, 1.0451e-01, 4.3088e-02, ..., 1.3430e-01,
1.1482e-01, 6.0621e-01],
[-1.4498e+00, -5.0014e-01, -2.4254e-01, ..., -1.0319e-01,
6.7817e-02, 3.6600e-01]],
...,
[[ 1.0236e+00, -4.3271e-01, -2.0565e-01, ..., -1.2573e-01,
-4.0702e-01, -9.3738e-01],
[-8.2948e-01, -9.5505e-01, -6.4398e-01, ..., -6.8510e-01,
-1.3028e+00, -8.6064e-01],
[-6.7923e-01, -9.3392e-01, -6.2442e-01, ..., -1.5106e-01,
-2.2067e-01, -4.0893e-01],
...,
[-7.3848e-01, -1.1706e+00, -7.6737e-01, ..., -9.3990e-02,
-1.8709e-01, -3.7508e-01],
[-2.6870e+00, -2.7765e+00, -1.8116e+00, ..., -8.1628e-01,
-8.7854e-01, -7.9379e-01],
[-6.7658e-01, -1.6597e+00, -1.0226e+00, ..., -3.2632e-03,
-1.2487e-01, -3.2197e-01]],
[[ 3.0677e-03, -1.1128e-02, 6.9859e-01, ..., 2.7826e-01,
6.7025e-01, 2.5786e-01],
[-3.6990e-02, -7.4887e-01, 2.5704e-01, ..., -1.4651e-02,
1.1849e-01, -7.8289e-02],
[-1.1229e-01, -8.3367e-01, 3.8548e-01, ..., -1.3003e-01,
-8.0553e-02, 1.6371e-01],
...,
[ 4.9160e-02, -7.5656e-01, 2.8890e-01, ..., -1.3684e-01,
-1.1491e-01, 1.5371e-01],
[-2.3169e-01, -1.4221e+00, 1.3140e-02, ..., -3.7037e-01,
-3.2077e-01, -5.6586e-02],
[ 6.6953e-01, 1.2194e-01, 2.7224e-01, ..., -2.4254e-01,
-2.2992e-01, 3.2225e-01]],
[[-1.5946e+00, -6.1404e-01, 9.0661e-02, ..., 5.3918e-01,
3.7177e-01, 5.4471e-01],
[-2.0562e+00, -3.5936e-01, -1.0833e-01, ..., -7.1900e-02,
-1.2056e-01, 3.9634e-01],
[-1.7051e+00, -3.8347e-01, 9.3054e-02, ..., -7.6576e-03,
-1.4267e-01, 8.0761e-02],
...,
[-1.7623e+00, -5.0756e-01, 2.7239e-01, ..., 8.0857e-02,
-6.9847e-02, 8.3370e-02],
[-2.1043e+00, -1.0815e+00, -1.3810e-01, ..., -8.7501e-02,
-2.5977e-01, -2.0738e-01],
[-1.0341e+00, -1.6863e-01, 5.0181e-01, ..., -1.0492e-01,
-2.7938e-01, 1.1743e-01]]],
[[[-2.1633e-02, -4.4019e-01, -2.6387e-01, ..., -1.0686e+00,
-4.8867e-01, -1.6056e+00],
[ 1.8174e-01, -1.1365e+00, -8.7963e-01, ..., -1.4016e+00,
-1.3623e+00, -1.5883e+00],
[ 2.2012e-01, -7.6090e-01, -6.6173e-01, ..., -1.1697e+00,
-1.1562e+00, -1.9003e+00],
...,
[ 1.2905e-01, -7.3396e-01, 1.9890e-01, ..., -1.1363e+01,
-1.9755e+01, -2.0510e+01],
[-3.6150e-01, -7.3476e-01, -1.3753e-01, ..., -1.1490e+01,
-1.3889e+01, -1.6966e+01],
[-5.4174e-01, -1.1842e+00, -1.0363e+00, ..., -9.4081e-01,
-4.5131e+00, -5.9781e+00]],
[[-1.1312e+00, -2.8233e+00, -2.7323e+00, ..., -3.7705e+00,
-5.6512e+00, -1.7997e+00],
[-2.5791e+00, -4.4965e+00, -3.8708e+00, ..., -6.2594e+00,
-8.6679e+00, -3.4027e+00],
[-2.2745e+00, -3.4830e+00, -3.5703e+00, ..., -5.1116e+00,
-6.5961e+00, -2.3352e+00],
...,
[-2.1728e+00, -3.0413e+00, -3.5057e+00, ..., -1.7678e+01,
-3.1909e+01, -7.4919e+00],
[-4.1158e+00, -4.8776e+00, -4.6201e+00, ..., -2.2911e+01,
-2.0083e+01, -5.3574e+00],
[ 4.4620e-01, -8.7039e-02, 6.0991e-01, ..., -4.3063e-01,
1.0383e+00, -6.6292e-01]],
[[-2.0004e-01, 2.4265e-01, 7.3153e-01, ..., 6.3718e-01,
8.6484e-02, 1.4736e+00],
[-5.6902e-02, -2.9147e-01, 4.1131e-01, ..., 5.8127e-01,
-2.1514e-01, -1.8188e-01],
[-4.0591e-01, -2.3977e-01, 5.3238e-02, ..., -7.1781e-03,
-2.0723e-02, -2.0668e-01],
...,
[-4.8958e-01, -1.5489e-01, 2.7259e+00, ..., -4.8784e+01,
-5.5231e+01, -1.9710e+01],
[-3.7016e-01, -4.5569e-02, 5.4961e-01, ..., -4.4516e+01,
-3.8526e+01, -1.3565e+01],
[-1.9562e-02, -1.8912e-01, 1.7698e-01, ..., -1.1176e+01,
-7.1112e+00, -2.7237e+00]],
...,
[[ 7.1128e-01, 1.2880e-01, 2.7819e-01, ..., 4.7051e-01,
4.8493e-01, -1.3193e+00],
[-4.8854e-01, -3.8327e-01, -3.2425e-01, ..., -3.2809e-01,
-7.9186e-01, -1.3872e+00],
[-2.8620e-01, -8.1242e-02, -5.1388e-02, ..., -7.3123e-02,
-2.1064e-03, -9.8193e-01],
...,
[-4.0698e-01, -1.6219e-01, -9.6715e-01, ..., -1.2662e+01,
-2.1503e+01, -2.4445e+01],
[-2.2355e+00, -1.5786e+00, -1.4056e+00, ..., -8.3743e+00,
-1.4767e+01, -2.1533e+01],
[-4.6497e-01, -6.8553e-01, -1.3232e-01, ..., -9.7956e+00,
-1.3838e+01, -1.3169e+01]],
[[ 1.8933e-01, 2.5940e-01, 5.0309e-01, ..., 1.1813e-01,
5.3424e-01, -1.5148e-01],
[ 1.5192e-01, 1.3057e-01, 3.4049e-01, ..., 2.7018e-01,
5.7778e-02, -8.2393e-01],
[-1.1754e-01, -3.3461e-02, 2.9997e-01, ..., 3.3441e-01,
1.5288e-01, -2.3435e-01],
...,
[-8.0085e-02, -1.3522e-01, 3.0434e-01, ..., -2.1240e+01,
-2.4978e+01, -1.6722e+01],
[-5.1437e-01, -9.8512e-01, 1.3845e-01, ..., -1.5267e+01,
-1.3091e+01, -2.9544e+00],
[ 6.3444e-01, 3.4213e-01, 2.3063e-01, ..., 1.2760e+00,
1.4362e+00, 5.9889e+00]],
[[-7.6868e-01, 1.1421e-01, 2.9284e-01, ..., -3.4346e-01,
-2.6536e-01, 5.3389e-01],
[-1.1743e+00, -7.5676e-02, -3.1741e-01, ..., -1.0250e+00,
-5.0520e-01, 1.0315e+00],
[-9.3879e-01, 9.1655e-02, -1.4488e-02, ..., -7.5634e-01,
-8.6177e-01, 6.7371e-01],
...,
[-8.6751e-01, 4.7305e-02, -1.3317e+00, ..., -1.8747e+01,
-2.4103e+01, -1.2993e+01],
[-1.5013e+00, -4.4461e-01, -4.9065e-01, ..., -1.2851e+01,
-2.0885e+01, -5.4092e+00],
[-7.0170e-01, -2.6316e-01, -1.7164e-01, ..., -5.8203e-02,
-7.8802e-01, 4.1960e+00]]],
[[[-8.0543e-01, -8.1644e-01, -7.3330e+00, ..., -1.3644e+00,
-8.6969e-01, -1.8867e+00],
[-5.1690e-01, -3.3082e+00, -1.5035e+01, ..., -2.0077e+00,
-2.0420e+00, -2.1440e+00],
[-9.5801e+00, -1.2519e+01, -1.8017e+01, ..., -2.8531e+00,
-2.4913e+00, -2.3916e+00],
...,
[ 1.2878e-01, 3.9254e-01, 1.2470e+00, ..., -1.9589e+00,
-2.7765e-01, -8.4844e-01],
[-1.8198e-01, -7.1120e-01, -9.1782e-02, ..., -5.4923e-01,
1.5862e-01, -4.6194e-01],
[-3.8319e-01, -1.2570e+00, -1.0503e+00, ..., -1.0321e+00,
-1.2208e+00, -6.1266e-01]],
[[-2.9178e+00, -1.3220e+01, -1.9454e+01, ..., -4.7704e+00,
-6.2531e+00, -2.3487e+00],
[-6.0760e+00, -1.6446e+01, -1.7410e+01, ..., -8.2797e+00,
-1.0117e+01, -4.5940e+00],
[-1.5745e+01, -2.6050e+01, -2.4539e+01, ..., -1.2078e+01,
-9.3070e+00, -3.3025e+00],
...,
[-2.4876e+00, -3.0464e+00, -3.0681e+00, ..., -5.0855e+00,
-3.6178e+00, -1.6399e+00],
[-3.3960e+00, -3.8384e+00, -2.8745e+00, ..., -3.8509e+00,
-4.2584e+00, -1.4803e+00],
[ 5.3664e-01, -8.8351e-02, 5.6824e-01, ..., 4.8946e-01,
5.6175e-01, 2.0362e-01]],
[[-7.6836e-01, -3.7116e-01, -3.1793e+00, ..., 8.5531e-01,
5.5470e-01, 2.0682e+00],
[-8.9413e-01, -3.0836e+00, -1.2402e+01, ..., 5.7591e-01,
-1.6585e-01, 4.2686e-02],
[-6.6223e+00, -1.9425e+00, -1.7835e+01, ..., 1.6738e-01,
1.0533e-01, -7.4927e-02],
...,
[-8.3615e-01, 2.4909e+00, 1.0603e-01, ..., 2.1122e+00,
3.2377e-01, 3.7294e-01],
[-7.2880e-01, 2.3113e-01, 3.1515e-01, ..., 5.0384e-01,
2.7424e-01, 4.9901e-01],
[-2.1309e-01, 4.0630e-02, 1.6606e-01, ..., -2.2448e-02,
1.2258e-01, 2.3667e-01]],
...,
[[ 1.5318e+00, -2.9408e-01, 1.8166e-01, ..., 2.6922e-01,
-6.7992e-02, -2.3443e+00],
[-7.6173e-01, 2.3365e+00, 1.8657e+00, ..., -7.7333e-01,
-1.7298e+00, -2.3531e+00],
[-2.1715e+00, -1.7844e+00, -3.6828e+00, ..., -1.2920e+00,
-6.2186e-01, -1.6706e+00],
...,
[-8.2610e-01, -2.2804e+00, -3.3047e+00, ..., -2.2159e+00,
-5.6190e-01, -7.2210e-01],
[-2.1622e+00, -1.7000e+00, -1.2369e+00, ..., -1.2447e+00,
-1.4724e+00, -1.2277e+00],
[-6.6869e-01, -8.8728e-01, -3.5369e-01, ..., -3.9477e-01,
-5.2305e-01, -4.6669e-01]],
[[ 1.8008e-02, 1.1960e+00, 9.7211e-01, ..., 7.8822e-01,
1.1602e+00, 2.3312e-01],
[ 2.1735e-01, -1.2385e+00, -3.7975e+00, ..., 6.1102e-01,
4.6050e-01, -8.7828e-01],
[-4.4622e+00, -1.1077e+01, -1.2381e+01, ..., -1.9952e+00,
-1.2647e+00, -3.8738e-01],
...,
[ 4.0698e-02, -8.0885e-02, -6.0734e-01, ..., 1.8006e-01,
2.9626e-01, -1.8338e-01],
[-4.7308e-01, -1.2995e+00, -5.5040e-03, ..., -4.1177e-01,
-1.5248e-02, -3.1706e-01],
[ 5.3055e-01, -1.1658e-01, -4.1010e-02, ..., 4.5584e-02,
7.2371e-02, 1.8205e-01]],
[[-1.9932e+00, -2.1631e+00, -5.5646e+00, ..., -7.3203e-01,
-4.5371e-01, 8.5791e-01],
[-2.6614e+00, -4.2784e+00, -1.1635e+01, ..., -1.0468e+00,
-2.4079e-01, 1.8213e+00],
[-8.7872e+00, -9.7824e+00, -1.4909e+01, ..., -5.5259e-01,
-1.3714e-01, 1.4624e+00],
...,
[-8.2685e-01, -1.3421e+00, 5.3761e-01, ..., 1.1134e+00,
-6.0725e-02, 7.6533e-01],
[-9.9741e-01, -7.6657e-01, -4.5615e-01, ..., 9.9154e-02,
-2.1333e-01, 4.0100e-01],
[-3.1275e-01, -2.6778e-01, -1.2916e-01, ..., -2.3764e-01,
-1.5323e-01, 5.8170e-01]]],
[[[-7.0433e-01, -1.4489e+00, -1.4587e+00, ..., -1.2570e+00,
-8.6885e-01, -1.6615e+00],
[-3.3308e-01, -2.4520e+00, -2.2693e+00, ..., -1.7503e+00,
-1.9291e+00, -1.9275e+00],
[-1.4197e-01, -1.9200e+00, -1.6791e+00, ..., -1.6273e+00,
-1.4244e+00, -2.1031e+00],
...,
[ 8.3793e-02, -5.6302e-01, -2.4912e-01, ..., -4.3207e-01,
-2.0800e-01, -4.4877e-01],
[-3.7499e-01, -4.6739e-01, 2.9098e-02, ..., 1.3793e-01,
2.8954e-01, -1.7142e-01],
[-6.2073e-01, -1.3226e+00, -1.0767e+00, ..., -1.0674e+00,
-1.1676e+00, -4.9389e-01]],
[[-2.7719e+00, -5.2558e+00, -4.6522e+00, ..., -4.7857e+00,
-6.0006e+00, -2.3832e+00],
[-5.3631e+00, -8.1644e+00, -7.5113e+00, ..., -7.8145e+00,
-9.2877e+00, -4.4023e+00],
[-4.0876e+00, -5.8542e+00, -8.4129e+00, ..., -6.7266e+00,
-6.4868e+00, -3.2271e+00],
...,
[-1.7586e+00, -2.2345e+00, -1.8490e+00, ..., -2.3005e+00,
-2.3490e+00, -8.6853e-01],
[-2.8103e+00, -3.0089e+00, -1.9626e+00, ..., -2.2353e+00,
-3.3678e+00, -9.9742e-01],
[ 3.3714e-01, -2.7548e-01, 3.3598e-01, ..., 2.9027e-01,
4.2162e-01, 1.9482e-01]],
[[-6.7145e-01, 4.7952e-01, 8.9344e-01, ..., 8.3875e-01,
6.0959e-01, 1.8989e+00],
[-9.7992e-01, 7.4095e-03, 4.2229e-01, ..., 5.8348e-01,
-9.3962e-03, 1.1209e-01],
[-1.7917e+00, -2.6384e-01, -2.3350e-01, ..., 5.0438e-02,
1.8308e-02, -9.1321e-02],
...,
[-6.4202e-01, 3.8158e-02, 2.3164e-01, ..., 1.0122e-01,
3.0977e-01, 4.0749e-01],
[-4.4144e-01, 7.0709e-02, 2.1348e-01, ..., 1.9512e-01,
2.4780e-01, 5.5798e-01],
[-1.5181e-01, -2.4315e-01, 1.2520e-01, ..., -1.4377e-02,
1.1969e-01, 3.2279e-01]],
...,
[[ 1.2874e+00, -3.2190e-01, -4.5935e-02, ..., 2.2912e-01,
-8.9776e-02, -2.1347e+00],
[-7.0892e-01, -8.7022e-01, -5.1170e-01, ..., -5.3762e-01,
-1.4959e+00, -2.0910e+00],
[-7.1129e-01, -8.7023e-01, -2.2889e-01, ..., -2.8581e-01,
-5.7289e-01, -1.3953e+00],
...,
[-6.6522e-01, -5.2435e-01, -3.4912e-01, ..., -6.7193e-01,
-4.2047e-01, -5.6323e-01],
[-2.2636e+00, -1.7785e+00, -1.2809e+00, ..., -1.3071e+00,
-1.3607e+00, -1.0910e+00],
[-6.3476e-01, -8.4430e-01, -4.5292e-01, ..., -2.8518e-01,
-3.5766e-01, -3.8897e-01]],
[[ 1.0413e-01, 4.6357e-02, 4.6106e-01, ..., 4.5404e-01,
9.2901e-01, -8.6940e-03],
[ 2.6582e-01, -7.4066e-01, -1.2157e-02, ..., 4.2861e-01,
3.4410e-01, -8.4377e-01],
[ 1.6211e-01, -5.9749e-01, 8.3480e-01, ..., -1.9446e-01,
5.2104e-01, -3.5847e-01],
...,
[-2.2928e-01, -4.2724e-01, 2.3798e-01, ..., -1.8167e-01,
1.2147e-01, 6.1544e-02],
[-5.2832e-01, -9.0566e-01, 6.9887e-02, ..., 1.7562e-02,
-3.6165e-02, -1.7577e-01],
[ 3.3890e-01, -1.8185e-02, -1.7219e-01, ..., -1.2848e-01,
-9.9805e-02, 1.7112e-01]],
[[-1.5751e+00, -7.9141e-01, -4.1738e-01, ..., -4.4880e-01,
-3.1002e-01, 8.5761e-01],
[-2.1897e+00, -7.7049e-01, -5.6088e-01, ..., -8.0206e-01,
-2.3094e-01, 1.6268e+00],
[-1.5346e+00, -5.5947e-01, -2.5052e-01, ..., -2.5019e-01,
-1.4174e-01, 1.4012e+00],
...,
[-6.6716e-01, -2.0569e-01, -1.1130e-01, ..., 3.0170e-01,
9.4183e-02, 6.3133e-01],
[-1.1469e+00, -5.6012e-01, -3.0390e-01, ..., -3.2514e-01,
-2.0266e-01, 1.8570e-01],
[-3.6480e-01, -3.5013e-02, -1.5833e-01, ..., -2.3614e-01,
-2.4794e-01, 3.2273e-01]]],
[[[-1.3974e-01, -3.9175e-01, -1.5677e+00, ..., -1.9054e-01,
3.2460e-01, 3.4854e-02],
[ 3.3029e-01, -9.6211e-01, 5.1065e-02, ..., -3.9691e-01,
-2.6676e-01, -9.5102e-02],
[ 2.4036e-01, -7.3253e-01, 2.6301e-01, ..., -1.6638e-01,
-2.3816e-01, -5.1330e-01],
...,
[ 8.1851e-02, -1.0303e+00, -8.5468e-01, ..., -9.5575e-01,
-9.5426e-01, -2.0140e+00],
[-1.0795e+00, -1.6894e+00, -7.8904e-01, ..., -5.3647e-01,
-4.6368e-01, -1.7656e+00],
[-1.2507e+00, -2.0857e+00, -1.5961e+00, ..., -1.4076e+00,
-1.5536e+00, -9.6151e-01]],
[[-1.4543e+00, -3.1708e+00, -5.0376e+00, ..., -2.6082e+00,
-3.3339e+00, -1.2498e+00],
[-2.9557e+00, -4.7758e+00, -2.9737e+00, ..., -3.7327e+00,
-4.8988e+00, -2.1380e+00],
[-1.9995e+00, -2.9227e+00, -1.9677e+00, ..., -2.4041e+00,
-2.6884e+00, -1.2570e+00],
...,
[-3.9509e+00, -5.6671e+00, -5.3069e+00, ..., -5.4150e+00,
-6.1500e+00, -2.8704e+00],
[-6.1453e+00, -7.0582e+00, -6.1304e+00, ..., -6.0988e+00,
-7.4473e+00, -2.7377e+00],
[ 9.1571e-03, -8.3653e-01, 2.5801e-01, ..., 4.1678e-01,
7.1135e-01, 1.8553e-01]],
[[-3.8137e-01, 2.3654e-01, -5.3899e-01, ..., 7.9360e-01,
5.6119e-01, 1.3604e+00],
[-2.9381e-01, 8.0172e-02, -3.1622e-01, ..., 5.2579e-01,
4.5311e-01, 2.9502e-01],
[-6.7440e-01, 1.4235e-01, 6.9472e-01, ..., 1.5775e-01,
3.5830e-01, 4.6064e-01],
...,
[-8.7813e-01, 5.9349e-01, 7.7426e-01, ..., 2.4193e-02,
6.6709e-02, 3.0721e-01],
[-1.1922e+00, 9.8834e-02, 1.9419e-03, ..., -3.8314e-02,
5.1608e-02, 3.3088e-01],
[-1.1320e+00, -1.1046e-01, 8.9472e-02, ..., -2.3620e-02,
-8.3211e-02, -3.3491e-01]],
...,
[[ 9.3529e-01, 4.1861e-01, 1.1312e-01, ..., 3.5537e-01,
1.0124e-01, -9.7042e-01],
[-9.6826e-01, -7.8023e-01, 8.1491e-02, ..., -7.5306e-01,
-1.6864e+00, -1.2080e+00],
[-4.6421e-01, -3.1382e-01, 8.4089e-02, ..., -3.1763e-01,
-5.3243e-01, -6.4333e-01],
...,
[-8.5200e-01, -1.3125e+00, -1.6758e+00, ..., -3.5610e-01,
-5.9441e-01, -1.2113e+00],
[-3.0442e+00, -2.8369e+00, -2.0195e+00, ..., -1.8591e+00,
-2.0096e+00, -2.0411e+00],
[-5.0838e-01, -1.6012e+00, -1.1004e+00, ..., -8.2742e-01,
-9.5948e-01, -1.3657e+00]],
[[ 7.8970e-02, 3.8285e-01, -3.1973e-01, ..., 6.3653e-01,
1.0403e+00, 4.4592e-01],
[ 1.0033e-01, -3.4850e-01, -2.0492e-01, ..., 4.6185e-01,
5.9020e-01, -2.0620e-01],
[-1.8855e-01, -4.8068e-01, 1.9327e-01, ..., 3.8293e-01,
3.0238e-01, 1.6492e-02],
...,
[ 7.9475e-02, -9.0192e-01, -9.6413e-02, ..., 2.8048e-01,
1.2359e-01, -7.4309e-01],
[-4.6422e-01, -1.8020e+00, -2.1568e-01, ..., 6.2007e-02,
-4.2333e-02, -9.6731e-01],
[ 4.8966e-01, -8.5302e-02, 1.2246e-01, ..., 2.7883e-01,
3.0823e-01, 3.0914e-01]],
[[-8.9000e-01, -4.0550e-01, -1.6032e-01, ..., 1.6091e-01,
5.0408e-02, 7.4339e-01],
[-1.0536e+00, -3.6234e-01, -1.4369e+00, ..., -3.4639e-01,
-4.2272e-02, 1.2212e+00],
[-6.7134e-01, -2.1100e-01, -1.0371e+00, ..., -1.9157e-01,
-1.2893e-01, 6.5947e-01],
...,
[-1.1088e+00, -6.8548e-01, -2.1348e-02, ..., -3.8257e-01,
-6.9884e-02, 8.3375e-01],
[-1.8435e+00, -1.7557e+00, -9.7801e-01, ..., -7.8808e-01,
-4.2636e-01, 4.4345e-01],
[-9.7915e-01, -9.4486e-01, -2.6232e-01, ..., -3.1368e-01,
-1.3746e-01, 1.1170e+00]]]], device='cuda:0'),
tensor([[[[-7.2180e-01, -1.4806e+00, -1.4655e+00, ..., -1.7964e+00,
-7.6938e-01, -1.7456e+00],
[-3.5705e-01, -2.4601e+00, -2.3156e+00, ..., -3.1579e+00,
-3.1867e+00, -2.4196e+00],
[-1.6687e-01, -1.9512e+00, -1.7738e+00, ..., 7.8053e-01,
1.2719e+00, -2.4215e-01],
...,
[-1.1532e-01, -2.6540e+00, -1.3659e+00, ..., 1.0820e-02,
5.5874e-02, -6.0732e-01],
[-2.8289e-01, -1.1787e+00, -1.5053e+00, ..., 3.4371e-01,
3.8128e-01, -4.0228e-01],
[-4.6249e-01, -1.2442e+00, -1.6356e+00, ..., -3.2167e-01,
-5.8597e-01, 3.7072e-01]],
[[-2.7207e+00, -5.1843e+00, -4.7736e+00, ..., -8.9853e+00,
-7.0626e+00, -3.4399e+00],
[-5.4820e+00, -8.3264e+00, -7.6975e+00, ..., -1.5991e+01,
-1.3990e+01, -7.1385e+00],
[-4.1717e+00, -6.0752e+00, -6.2728e+00, ..., -1.2439e+01,
-8.7536e+00, -2.5917e+00],
...,
[-3.5223e+00, -6.2823e+00, -6.8263e+00, ..., -1.8676e+00,
-3.4830e+00, -1.3428e+00],
[-2.8790e+00, -4.4162e+00, -4.0946e+00, ..., -1.2242e+00,
-2.3871e+00, -2.1693e-01],
[ 2.9165e-01, -6.8426e-01, -5.8230e-01, ..., -1.2575e+00,
-6.9803e-01, -2.1032e-01]],
[[-6.7519e-01, 4.4028e-01, 8.4394e-01, ..., 7.1993e-01,
5.9811e-01, 1.9444e+00],
[-9.4779e-01, 3.1165e-02, 3.9198e-01, ..., -1.4574e+00,
-2.2251e+00, 3.3621e-01],
[-1.7703e+00, -2.2805e-01, -7.3188e-02, ..., -5.3060e+00,
-8.2012e+00, -3.8530e+00],
...,
[-2.3099e+00, -6.7894e+00, -6.8392e+00, ..., -1.7129e+00,
-9.0164e-01, 4.9338e-01],
[-2.1395e-01, -4.0893e+00, -1.6882e+00, ..., -1.1296e+00,
-7.9803e-01, -4.7857e-01],
[ 5.2025e-01, -9.0209e-01, 9.0840e-01, ..., -1.4038e+00,
-4.7649e-01, -6.1303e-01]],
...,
[[ 1.2554e+00, -3.5109e-01, -1.4563e-02, ..., 8.5215e-01,
3.4620e-01, -1.8076e+00],
[-7.4194e-01, -9.2035e-01, -4.6551e-01, ..., -1.6658e+00,
-2.0991e+00, -3.5214e+00],
[-7.3018e-01, -8.8707e-01, -2.6922e-01, ..., -7.7090e+00,
-5.0118e+00, -6.0829e+00],
...,
[ 4.6306e-02, -3.4264e+00, -4.5366e+00, ..., -1.3313e+00,
-1.5851e+00, -9.2121e-01],
[-2.3288e+00, -1.3664e+00, -3.1919e+00, ..., -2.2098e+00,
-2.6659e+00, -1.1125e+00],
[-8.7440e-01, -8.8258e-01, -1.7602e+00, ..., -1.7768e+00,
-1.4579e+00, -5.0875e-01]],
[[ 3.2164e-02, -9.6805e-03, 4.8674e-01, ..., -4.5656e-01,
4.9886e-01, -6.5952e-01],
[ 2.2991e-01, -7.8056e-01, 1.4567e-02, ..., -2.3108e+00,
-2.9104e-01, -1.8060e+00],
[ 1.7193e-01, -6.3446e-01, 3.7655e-01, ..., -4.4841e+00,
-6.2806e-01, 8.9863e-01],
...,
[-7.3674e-01, -4.1884e+00, -4.1571e+00, ..., -9.6094e-01,
3.3720e-01, 3.2014e-01],
[-6.6010e-01, -3.0715e+00, -2.2188e+00, ..., 2.3431e-01,
9.4776e-02, 2.0367e-01],
[ 5.0732e-01, -5.4777e-01, -9.8773e-01, ..., -3.6992e-01,
-6.3193e-01, -7.6015e-02]],
[[-1.6221e+00, -8.6575e-01, -4.5848e-01, ..., -6.3524e-01,
-1.1180e-01, 9.6226e-01],
[-2.2439e+00, -8.2563e-01, -6.2621e-01, ..., -3.1910e+00,
-7.8495e-01, 1.5272e+00],
[-1.6056e+00, -6.2891e-01, -3.3782e-01, ..., -1.0400e+01,
-6.8455e+00, 2.0511e-01],
...,
[-7.3912e-01, -6.1819e-01, -4.6490e+00, ..., -1.5681e+00,
-7.4414e-01, 4.5217e-01],
[-1.3475e+00, -7.0001e-01, -2.7768e+00, ..., -1.5301e+00,
-9.1222e-01, 5.5920e-01],
[-4.5632e-01, 2.2038e-01, -1.0993e-01, ..., -1.1739e-01,
-5.5865e-01, 5.9485e-01]]],
[[[-3.3395e-01, -1.0249e+00, -9.1070e-01, ..., -2.4180e-02,
5.5576e-01, 4.8064e-01],
[ 1.5753e-01, -2.0141e+00, -2.1961e+00, ..., -4.4022e-01,
-2.2508e-01, 1.1022e-01],
[ 1.5341e-01, -1.5675e+00, -2.4460e+00, ..., -1.6314e-01,
-4.3725e-01, -6.4749e-01],
...,
[-5.5978e-01, -3.2549e+00, -1.2488e+00, ..., -4.3670e-01,
4.3676e-03, -1.2405e-02],
[-1.3921e-01, -1.3529e+00, -2.2107e-01, ..., 4.9567e-02,
3.2487e-01, 2.0086e-02],
[-9.6548e-02, -9.1438e-01, -1.0457e+00, ..., -1.1987e+00,
-1.2713e+00, -4.6154e-01]],
[[-2.1228e+00, -4.2986e+00, -4.8159e+00, ..., -6.6047e-01,
-1.1637e+00, -9.5509e-02],
[-4.0220e+00, -7.5365e+00, -7.8953e+00, ..., -1.8220e+00,
-2.6695e+00, -6.8141e-01],
[-3.2763e+00, -6.9746e+00, -5.7788e+00, ..., -3.1123e+00,
-3.7646e+00, -1.4475e+00],
...,
[-5.2234e+00, -7.3375e+00, -5.9032e+00, ..., -1.6257e+00,
-1.5349e+00, 7.3062e-02],
[-5.1928e+00, -6.6632e+00, -5.9578e+00, ..., -2.7997e+00,
-3.7010e+00, -1.0700e+00],
[ 7.3202e-01, -1.0346e-01, 4.4530e-01, ..., 3.9138e-01,
3.7103e-01, 5.4767e-02]],
[[-6.1115e-01, 2.9884e-01, 6.3649e-01, ..., 8.1889e-01,
3.2031e-01, 1.0573e+00],
[-9.4561e-01, -1.9027e-01, -6.4897e-02, ..., 3.8778e-01,
9.2736e-02, 4.8495e-01],
[-1.3538e+00, 1.3474e-01, -1.6850e-02, ..., -7.9255e-03,
-7.5806e-02, 3.0988e-01],
...,
[-3.9860e+00, -2.6691e-01, -1.0762e+00, ..., 4.0900e-01,
5.2742e-01, 6.0755e-01],
[-1.1562e+00, 1.3368e+00, 1.2137e-02, ..., 1.6614e-01,
1.2114e-01, 6.2067e-01],
[-3.6716e-01, 2.7921e-01, -3.4461e-03, ..., -9.0576e-02,
5.9298e-02, 3.8244e-01]],
...,
[[ 6.1930e-01, 7.9676e-02, 3.2417e-01, ..., -5.1071e-02,
-2.8592e-01, -8.6482e-01],
[-8.6657e-01, -2.8969e-01, 3.6140e-01, ..., -5.6302e-01,
-1.1594e+00, -7.8288e-01],
[-5.4781e-01, 2.2787e-01, 1.0434e+00, ..., -1.8166e-02,
-3.2171e-02, -3.4419e-01],
...,
[ 4.3446e-01, -2.1460e-01, -1.6988e+00, ..., -1.3517e-01,
-7.1513e-01, -9.9975e-01],
[-1.6144e+00, -2.8443e+00, -3.1329e+00, ..., -9.4998e-01,
-1.0088e+00, -8.7512e-01],
[-7.7829e-01, -1.5873e+00, -6.5930e-01, ..., -3.0965e-02,
-1.5763e-01, -4.0962e-01]],
[[ 6.7025e-03, -2.8436e-01, 1.3001e-01, ..., 3.1630e-01,
6.5008e-01, 2.6687e-01],
[ 9.3371e-02, -6.9547e-01, 6.5202e-02, ..., 4.4149e-02,
1.6323e-01, -7.2572e-02],
[-2.0104e-01, -6.9498e-01, -1.5327e-01, ..., 3.1817e-01,
-8.3931e-02, -8.7271e-01],
...,
[-1.8698e+00, -1.7856e+00, 5.2550e-01, ..., -4.3207e-02,
4.6139e-02, -1.5140e-01],
[-1.0963e+00, -2.0451e+00, 2.4714e-01, ..., -4.4804e-01,
-4.3223e-01, -1.4982e-01],
[ 6.7943e-01, -4.0194e-01, -9.2342e-02, ..., -2.7061e-01,
-3.0274e-01, 2.7193e-01]],
[[-6.7116e-01, -2.0891e-01, -5.8942e-02, ..., 4.7218e-01,
3.3222e-01, 5.2647e-01],
[-1.0513e+00, -7.2140e-01, -1.5684e+00, ..., -4.3808e-02,
-1.2462e-01, 3.9326e-01],
[-6.4440e-01, -8.3341e-01, -1.1924e+00, ..., 9.3575e-02,
-3.7441e-01, -3.3255e-01],
...,
[-5.6301e-01, -4.1927e+00, -2.8142e+00, ..., -2.7613e-01,
-2.3465e-01, 4.2787e-01],
[-8.4049e-01, -2.2389e+00, -9.7362e-01, ..., -1.0343e-01,
-2.9722e-01, -2.6623e-01],
[-3.8233e-01, -6.9941e-01, -3.4860e-02, ..., -1.7447e-01,
-3.4822e-01, 7.8365e-02]]],
[[[-2.0773e-01, -2.5169e-01, 2.0364e-01, ..., 2.3008e-02,
4.8901e-01, 1.3990e-01],
[-1.0158e-01, -1.5590e+00, -8.9740e-01, ..., -7.6087e-01,
-4.3923e-01, -3.2596e-01],
[-9.8373e-02, -1.3551e+00, -6.5059e-01, ..., -5.4105e-01,
-4.0991e-01, -7.0192e-01],
...,
[-2.2234e-01, -1.5690e+00, -1.1601e+00, ..., -9.3542e-01,
-9.7099e-01, -2.0461e+00],
[-1.0571e+00, -1.6563e+00, -5.5092e-01, ..., -5.1851e-01,
-4.7131e-01, -1.8090e+00],
[-1.2628e+00, -2.1545e+00, -1.6356e+00, ..., -1.4328e+00,
-1.5625e+00, -9.7656e-01]],
[[-1.6312e-01, -1.4416e+00, -6.1412e-01, ..., -7.2441e-01,
-1.3849e+00, -2.4766e-01],
[-2.0874e+00, -4.1601e+00, -2.5565e+00, ..., -2.4304e+00,
-3.3872e+00, -1.1316e+00],
[-1.8834e+00, -3.4667e+00, -2.7014e+00, ..., -2.6899e+00,
-3.0125e+00, -1.2447e+00],
...,
[-3.7290e+00, -5.5383e+00, -5.3005e+00, ..., -5.4237e+00,
-6.1785e+00, -2.9393e+00],
[-6.1533e+00, -7.0461e+00, -6.2568e+00, ..., -6.1003e+00,
-7.4054e+00, -2.7074e+00],
[-8.1612e-03, -8.5402e-01, 2.4446e-01, ..., 4.1120e-01,
7.2260e-01, 1.9729e-01]],
[[-5.2917e-01, -1.7839e-01, 3.6325e-01, ..., 4.2630e-01,
9.8524e-02, 8.6869e-01],
[-4.0994e-01, -4.8573e-01, 4.5788e-01, ..., 3.7887e-01,
3.1759e-01, 4.9948e-01],
[-8.9819e-01, -2.1524e-01, 1.7962e-01, ..., -2.1845e-03,
1.8201e-01, 4.2896e-01],
...,
[-1.3422e+00, -4.0048e-02, 2.9054e-01, ..., 8.4430e-02,
8.9045e-03, 2.6279e-01],
[-1.2068e+00, 1.6427e-01, 5.0688e-01, ..., -9.0301e-03,
-4.4767e-03, 2.8813e-01],
[-1.1587e+00, -5.5829e-02, 8.3347e-02, ..., 2.8050e-02,
-4.5425e-02, -3.4522e-01]],
...,
[[ 4.8233e-02, -6.0730e-01, -3.6154e-01, ..., -5.7908e-02,
-2.1961e-01, -7.1648e-01],
[-8.6712e-01, -6.2166e-01, -5.6265e-01, ..., -2.8353e-01,
-9.9242e-01, -7.3558e-01],
[-4.7999e-01, -5.3266e-02, -3.4696e-02, ..., 4.9566e-02,
-1.8930e-01, -5.2437e-01],
...,
[-5.7735e-01, -5.4321e-01, 3.0175e-01, ..., -3.4755e-01,
-5.6556e-01, -1.2032e+00],
[-3.1283e+00, -2.9760e+00, -2.3722e+00, ..., -1.9254e+00,
-2.0226e+00, -2.0670e+00],
[-5.2748e-01, -1.6369e+00, -1.1219e+00, ..., -8.6122e-01,
-9.4173e-01, -1.3563e+00]],
[[-1.7081e-01, 3.8213e-02, 4.3542e-01, ..., 9.7848e-02,
5.2714e-01, 1.6852e-01],
[-4.4285e-01, -2.6848e-01, -2.6276e-03, ..., -5.9693e-02,
1.3753e-01, -7.8913e-02],
[-6.6977e-01, -5.7812e-01, 2.6198e-02, ..., 4.9154e-02,
3.4239e-02, -6.7778e-02],
...,
[-7.7941e-02, -8.6292e-01, -2.3393e-01, ..., 2.7014e-01,
1.5679e-01, -7.3391e-01],
[-4.8970e-01, -1.7807e+00, -2.3650e-01, ..., 2.2561e-02,
-1.0700e-02, -9.4580e-01],
[ 4.3754e-01, -1.5368e-01, 1.0952e-01, ..., 2.5993e-01,
3.1363e-01, 3.2267e-01]],
[[-6.8405e-01, 3.7926e-01, 5.5851e-01, ..., 4.8046e-01,
3.6173e-01, 6.0863e-01],
[-8.1371e-01, 2.6154e-01, -2.5497e-01, ..., -2.5071e-01,
-1.4443e-01, 6.5118e-01],
[-4.5601e-01, 3.1140e-01, 4.3604e-02, ..., 1.2148e-01,
1.3226e-01, 5.9318e-01],
...,
[-1.1540e+00, -9.9270e-01, -7.3736e-01, ..., -4.4528e-01,
-4.9229e-02, 8.6132e-01],
[-1.8458e+00, -1.8219e+00, -9.1465e-01, ..., -8.6313e-01,
-4.7973e-01, 4.1649e-01],
[-9.5410e-01, -9.4779e-01, -2.3330e-01, ..., -2.9966e-01,
-1.4669e-01, 1.1153e+00]]],
[[[ 2.7970e-02, -2.6366e-01, -6.3058e-02, ..., -9.9927e-01,
-5.4464e-01, -1.4753e+00],
[ 2.7734e-01, -1.1279e+00, -8.0223e-01, ..., -1.1879e+00,
-1.3111e+00, -1.5282e+00],
[-2.4813e-01, -1.0408e+00, -5.7667e-01, ..., -9.2868e-01,
-1.0698e+00, -1.8395e+00],
...,
[ 6.5926e-01, 6.5471e-02, 3.4182e-01, ..., -4.0490e+00,
-1.3051e+00, -2.2174e+00],
[-4.2886e-01, -5.8878e-01, 8.3236e-02, ..., -1.2027e+00,
-7.0944e-01, -1.6603e+00],
[-5.5868e-01, -1.2141e+00, -1.0273e+00, ..., -1.1811e+00,
-1.5673e+00, -7.8938e-01]],
[[-6.8225e-01, -2.1879e+00, -1.8494e+00, ..., -3.9171e+00,
-5.4756e+00, -1.8308e+00],
[-2.3224e+00, -4.3218e+00, -3.2107e+00, ..., -5.9765e+00,
-8.2233e+00, -3.4591e+00],
[-3.2527e+00, -5.3158e+00, -5.5832e+00, ..., -5.0456e+00,
-6.2638e+00, -2.5893e+00],
...,
[-1.8773e+00, -2.9547e+00, -3.0844e+00, ..., -7.5956e+00,
-7.5047e+00, -4.1791e+00],
[-3.5930e+00, -4.1386e+00, -3.5782e+00, ..., -6.5360e+00,
-7.8824e+00, -2.4138e+00],
[ 3.2019e-01, -2.0850e-01, 4.2105e-01, ..., 3.9293e-01,
4.2700e-01, 2.3902e-01]],
[[-2.9662e-01, 9.1324e-02, 7.8909e-01, ..., 6.4401e-01,
2.1766e-01, 1.4875e+00],
[ 7.2495e-02, -3.9568e-01, 5.7945e-01, ..., 5.9280e-01,
-8.9407e-02, 3.6789e-02],
[-3.3070e-01, -4.1839e-01, -5.2507e-02, ..., 3.8544e-02,
-3.9896e-02, -1.5147e-01],
...,
[ 1.6592e-01, 8.4678e-01, 4.2120e-01, ..., 1.7077e+00,
-1.4412e+00, 9.2834e-02],
[-1.5434e-01, 5.0802e-02, 5.4950e-01, ..., 1.4253e-01,
-1.4714e-01, -2.0375e-01],
[-5.7893e-02, -3.3037e-01, 1.2139e-01, ..., 8.7682e-02,
2.2059e-01, -5.3752e-01]],
...,
[[ 4.6464e-01, -2.9127e-01, 1.5677e-02, ..., 4.0436e-01,
3.8825e-01, -1.3017e+00],
[-6.9389e-01, -5.5714e-01, -5.7259e-01, ..., -1.9400e-01,
-7.5413e-01, -1.3710e+00],
[-4.8658e-01, -6.3981e-01, -3.7159e-01, ..., -9.5556e-02,
-4.9521e-02, -9.2667e-01],
...,
[-1.1090e+00, -2.1609e+00, -2.0552e+00, ..., -4.4343e+00,
-4.8371e+00, -4.1220e+00],
[-2.1049e+00, -1.5362e+00, -1.3667e+00, ..., -2.2976e+00,
-2.0132e+00, -1.7272e+00],
[-4.4956e-01, -6.4087e-01, -3.3563e-01, ..., -3.9756e-01,
-5.6932e-01, -1.2139e+00]],
[[ 9.3005e-02, 2.5492e-01, 6.8763e-01, ..., -8.2149e-02,
3.8567e-01, -3.5064e-01],
[ 3.9616e-02, 2.9084e-01, 4.1520e-01, ..., 1.8836e-01,
5.9010e-02, -8.2506e-01],
[-4.6416e-01, -1.0034e-01, 5.2051e-01, ..., 2.8479e-01,
1.5106e-01, -4.3818e-01],
...,
[-6.3810e-01, -7.3129e-01, -4.8857e-01, ..., -2.9320e+00,
2.3849e-01, -7.5214e-01],
[-4.2119e-01, -6.3393e-01, 1.6680e-01, ..., -1.0783e+00,
-3.9164e-01, -8.1307e-01],
[ 5.4891e-01, 3.9196e-01, 1.1381e-01, ..., 3.1549e-01,
3.2227e-01, 2.8861e-01]],
[[-9.4695e-01, 2.6127e-01, 3.9055e-01, ..., -1.1295e-01,
-1.2834e-01, 5.5079e-01],
[-1.2060e+00, 1.9093e-01, -3.9380e-01, ..., -9.2245e-01,
-4.5223e-01, 8.9134e-01],
[-1.3145e+00, 1.5719e-01, -2.9969e-01, ..., -5.9520e-01,
-5.6381e-01, 6.2722e-01],
...,
[-2.2702e-01, 6.0284e-01, 1.0553e+00, ..., -1.4886e+00,
1.2554e-01, 2.2654e+00],
[-1.5031e+00, -3.5507e-01, -4.2106e-01, ..., -8.2607e-02,
-2.8830e-01, 6.5783e-01],
[-5.9769e-01, 2.4882e-02, -2.1794e-01, ..., -4.8773e-01,
-4.4264e-01, 1.0113e+00]]],
[[[ 3.2095e-02, -2.3230e-01, -1.9358e-02, ..., -9.1481e-01,
-4.6869e-01, -1.3548e+00],
[ 3.0280e-01, -9.1631e-01, -6.8803e-01, ..., -1.1125e+00,
-1.2676e+00, -1.4352e+00],
[ 2.3089e-01, -5.6283e-01, -4.7906e-01, ..., -8.7343e-01,
-1.0338e+00, -1.8169e+00],
...,
[-2.1555e-02, -6.0501e-01, -3.4572e-01, ..., -9.1446e-01,
-1.0401e+00, -1.9116e+00],
[-6.0595e-01, -6.5918e-01, -6.1807e-02, ..., -4.6084e-01,
-5.5684e-01, -1.6898e+00],
[-6.6952e-01, -1.3388e+00, -1.1874e+00, ..., -9.8101e-01,
-1.3955e+00, -6.5747e-01]],
[[-7.5676e-01, -2.1748e+00, -1.8166e+00, ..., -3.9002e+00,
-5.4528e+00, -1.8464e+00],
[-2.0390e+00, -3.8048e+00, -2.9935e+00, ..., -6.0088e+00,
-8.3329e+00, -3.6121e+00],
[-1.6176e+00, -2.6178e+00, -2.5794e+00, ..., -5.0258e+00,
-6.3535e+00, -2.6538e+00],
...,
[-1.7595e+00, -2.4128e+00, -2.5219e+00, ..., -5.0890e+00,
-6.2774e+00, -2.5122e+00],
[-3.7604e+00, -4.4382e+00, -3.6771e+00, ..., -6.0801e+00,
-7.5484e+00, -2.4208e+00],
[ 2.9872e-01, -3.1122e-01, 3.7814e-01, ..., 4.1935e-01,
4.4403e-01, 2.3115e-01]],
[[-2.1393e-01, 1.6221e-01, 7.4272e-01, ..., 6.8950e-01,
2.5451e-01, 1.5891e+00],
[ 7.9781e-02, -3.7156e-01, 6.0002e-01, ..., 5.9531e-01,
-2.2181e-01, 1.0100e-01],
[-1.4057e-01, -2.2499e-01, 2.4009e-01, ..., 4.2242e-02,
-1.2050e-01, -1.2247e-01],
...,
[-2.3908e-01, -2.1607e-01, 4.0956e-01, ..., 5.7131e-03,
-5.8437e-02, -1.7300e-01],
[-2.3314e-01, -1.8123e-01, 1.3635e-01, ..., -8.4794e-02,
-1.7169e-02, -2.0088e-01],
[-1.2281e-01, -4.6194e-01, 9.8713e-02, ..., 8.0995e-02,
1.7919e-01, -6.0491e-01]],
...,
[[ 5.9236e-01, -2.9311e-01, -2.9242e-02, ..., 4.1952e-01,
2.5340e-01, -1.3566e+00],
[-5.2131e-01, -5.3319e-01, -5.2763e-01, ..., -1.9235e-01,
-8.5259e-01, -1.4060e+00],
[-3.3510e-01, -2.2245e-01, -2.5302e-01, ..., -1.2001e-01,
-6.0204e-02, -9.4903e-01],
...,
[-4.3748e-01, -1.5019e-01, -3.1028e-01, ..., -6.8428e-02,
-9.3551e-02, -9.1973e-01],
[-2.3166e+00, -1.6868e+00, -1.2532e+00, ..., -1.3619e+00,
-1.3953e+00, -1.7195e+00],
[-4.7236e-01, -7.7393e-01, -3.9342e-01, ..., -3.9171e-01,
-5.0401e-01, -1.1513e+00]],
[[ 1.5582e-01, 2.5102e-01, 6.1395e-01, ..., -5.4530e-02,
4.3956e-01, -3.9258e-01],
[ 1.4191e-01, 2.4325e-01, 4.3442e-01, ..., 2.3306e-01,
8.5064e-02, -9.2189e-01],
[-1.0643e-01, 1.0813e-01, 2.7105e-01, ..., 3.1071e-01,
1.5798e-01, -5.8491e-01],
...,
[-1.8768e-01, 3.8611e-03, 4.0895e-01, ..., 2.3776e-01,
1.4905e-01, -4.9805e-01],
[-5.3387e-01, -6.8932e-01, -5.1773e-02, ..., -3.0577e-02,
-1.4584e-01, -8.6371e-01],
[ 5.2071e-01, 3.5892e-01, 4.6354e-02, ..., 3.9059e-01,
3.3613e-01, 2.7782e-01]],
[[-8.4615e-01, 2.8249e-01, 3.3337e-01, ..., -1.5132e-02,
-9.7491e-03, 6.2404e-01],
[-1.1207e+00, 2.4107e-01, -3.4710e-01, ..., -9.2321e-01,
-3.8753e-01, 9.0989e-01],
[-9.4131e-01, 3.6857e-01, -1.1744e-01, ..., -6.2673e-01,
-5.7417e-01, 5.9155e-01],
...,
[-9.5898e-01, 2.0204e-01, -2.6452e-01, ..., -5.1786e-01,
-5.0770e-01, 6.1649e-01],
[-1.6513e+00, -3.4695e-01, -3.1364e-01, ..., -6.4815e-01,
-6.4061e-01, 4.5155e-01],
[-6.6456e-01, -6.4318e-02, -2.8906e-01, ..., -4.2785e-01,
-4.1491e-01, 9.8444e-01]]]], device='cuda:0'),
tensor([[[[-1.3612e-01, -5.0352e-01, -2.1927e-01, ..., -5.4918e-02,
3.2015e-01, -6.0973e-02],
[ 2.7862e-01, -1.2204e+00, -7.0833e-01, ..., -5.3488e-01,
-2.6657e-01, -2.1354e-01],
[ 1.0733e-01, -9.4861e-01, -3.9761e-01, ..., -3.3228e-01,
-2.3778e-01, -5.8045e-01],
...,
[-1.8859e-01, -1.5087e+00, -1.1154e+00, ..., -9.5065e-01,
-9.0259e-01, -2.0129e+00],
[-1.0472e+00, -1.6842e+00, -8.1209e-01, ..., -5.3733e-01,
-4.4278e-01, -1.7804e+00],
[-1.2559e+00, -2.0824e+00, -1.5715e+00, ..., -1.4311e+00,
-1.5664e+00, -9.6127e-01]],
[[-1.4322e+00, -3.0840e+00, -2.5499e+00, ..., -2.8682e+00,
-3.2757e+00, -1.2594e+00],
[-3.1134e+00, -4.9772e+00, -3.5265e+00, ..., -4.0387e+00,
-4.6677e+00, -1.9673e+00],
[-2.0604e+00, -3.0430e+00, -2.4192e+00, ..., -2.6056e+00,
-2.7638e+00, -1.2580e+00],
...,
[-3.7496e+00, -5.6010e+00, -5.2797e+00, ..., -5.3853e+00,
-6.1212e+00, -2.8932e+00],
[-6.1825e+00, -7.0168e+00, -6.0702e+00, ..., -6.1006e+00,
-7.3977e+00, -2.6763e+00],
[-1.8872e-02, -8.6312e-01, 2.8124e-01, ..., 3.9495e-01,
6.7046e-01, 1.8560e-01]],
[[-4.6917e-01, 1.7472e-01, 8.2079e-01, ..., 9.6347e-01,
5.6566e-01, 1.3529e+00],
[-3.3666e-01, -8.9211e-02, 5.9215e-01, ..., 7.3753e-01,
3.9252e-01, 3.2584e-01],
[-7.9214e-01, 8.0169e-03, 3.0694e-01, ..., 3.0149e-01,
3.8761e-01, 4.4320e-01],
...,
[-1.3443e+00, 1.7407e-01, 9.4183e-02, ..., 5.6087e-02,
-1.0686e-02, 2.9958e-01],
[-1.2522e+00, 1.1453e-01, -5.8795e-02, ..., -1.4623e-02,
2.1810e-02, 3.2253e-01],
[-1.1873e+00, -4.3350e-02, 9.5243e-02, ..., 1.0057e-02,
-3.9236e-02, -3.3154e-01]],
...,
[[ 8.6050e-01, 2.1140e-01, 2.3212e-01, ..., 7.0277e-02,
1.6677e-01, -8.8947e-01],
[-1.1859e+00, -8.5009e-01, -6.6436e-01, ..., -1.0263e+00,
-1.4623e+00, -1.1185e+00],
[-6.3732e-01, -3.0626e-01, -2.0444e-01, ..., -2.6206e-01,
-5.2517e-01, -6.4533e-01],
...,
[-6.6465e-01, -7.9356e-01, -6.4299e-01, ..., -3.5121e-01,
-5.9663e-01, -1.1817e+00],
[-3.0601e+00, -2.8755e+00, -2.0314e+00, ..., -1.9235e+00,
-2.0568e+00, -2.0586e+00],
[-5.3207e-01, -1.6250e+00, -1.0740e+00, ..., -8.5870e-01,
-9.6070e-01, -1.3615e+00]],
[[ 4.3659e-02, 2.8264e-01, 6.9465e-01, ..., 7.4978e-01,
9.6919e-01, 3.7658e-01],
[-3.1743e-02, -3.7955e-01, 1.7311e-01, ..., 3.6007e-01,
5.3127e-01, -2.0233e-01],
[-3.2003e-01, -5.9939e-01, 2.2489e-01, ..., 1.9967e-01,
2.9498e-01, 1.1168e-02],
...,
[-6.7621e-02, -9.1044e-01, 8.1590e-03, ..., 2.4669e-01,
1.3841e-01, -7.1567e-01],
[-5.0102e-01, -1.8564e+00, -2.4058e-01, ..., 3.2644e-02,
-2.2718e-02, -9.2654e-01],
[ 4.8066e-01, -1.2921e-01, 1.3124e-01, ..., 2.3863e-01,
2.8478e-01, 3.2501e-01]],
[[-9.9862e-01, -2.6375e-01, 1.3622e-01, ..., 1.4574e-01,
7.5629e-02, 7.5065e-01],
[-1.1348e+00, -1.9394e-01, -2.9742e-01, ..., -1.8581e-01,
-9.1085e-02, 1.1675e+00],
[-7.3322e-01, -1.8987e-01, -2.3664e-01, ..., -2.2517e-02,
-1.2782e-01, 6.7894e-01],
...,
[-1.0723e+00, -9.2140e-01, -1.3733e-01, ..., -4.0375e-01,
-5.9086e-02, 8.5910e-01],
[-1.7939e+00, -1.7616e+00, -9.4666e-01, ..., -8.3150e-01,
-4.8319e-01, 4.5597e-01],
[-9.3221e-01, -9.3177e-01, -2.2224e-01, ..., -2.8141e-01,
-1.4220e-01, 1.1190e+00]]],
[[[-3.7631e-01, -9.7351e-01, -4.8458e-01, ..., 4.1496e-01,
9.7117e-02, -8.1949e-01],
[-2.7997e-02, -1.8244e+00, -9.9427e-01, ..., -1.7462e+00,
-1.7355e-01, -1.2224e+00],
[ 1.7443e-02, -1.4734e+00, -9.7440e-01, ..., -1.9023e+00,
-1.3155e-01, -1.7678e+00],
...,
[ 8.7893e-02, -5.8760e-01, -2.6943e-01, ..., -5.4212e-01,
-9.4940e-01, -6.1792e-01],
[-3.2024e-01, -4.5289e-01, 1.2167e-02, ..., 5.8389e-02,
-1.9120e-02, -2.5664e-01],
[-5.9410e-01, -1.2809e+00, -1.0735e+00, ..., -1.0941e+00,
-1.1793e+00, -5.0331e-01]],
[[-2.0458e+00, -3.9626e+00, -4.4658e+00, ..., -5.4243e+00,
-6.4596e+00, -1.9834e+00],
[-3.8259e+00, -6.1575e+00, -5.3381e+00, ..., -5.8947e+00,
-8.5367e+00, -3.0274e+00],
[-3.5800e+00, -5.5302e+00, -5.4901e+00, ..., -7.2870e+00,
-8.0179e+00, -2.8117e+00],
...,
[-1.8416e+00, -2.3162e+00, -1.9379e+00, ..., -3.5847e+00,
-3.5075e+00, -1.1466e+00],
[-2.9154e+00, -3.0750e+00, -2.0845e+00, ..., -2.9635e+00,
-3.8033e+00, -1.0918e+00],
[ 3.6401e-01, -2.3455e-01, 3.5586e-01, ..., 3.1301e-01,
4.6086e-01, 2.0762e-01]],
[[-4.5921e-01, 5.0161e-01, 7.0999e-01, ..., 2.1870e+00,
1.0026e+00, 1.5312e+00],
[-6.3635e-01, 4.8833e-02, 8.9833e-01, ..., -4.9654e-01,
8.0990e-01, 3.7905e-01],
[-1.1615e+00, 8.2037e-02, 2.1640e-01, ..., -3.0527e+00,
6.6268e-01, 2.8783e-01],
...,
[-6.9450e-01, 9.7496e-02, 2.5001e-01, ..., -1.6467e+00,
1.6815e-01, 3.8129e-01],
[-4.7824e-01, 9.9062e-02, 2.2969e-01, ..., -1.0511e-01,
3.8423e-01, 5.4674e-01],
[-1.4916e-01, -2.2338e-01, 1.2099e-01, ..., -1.4359e-02,
1.2254e-01, 3.4572e-01]],
...,
[[ 1.0623e+00, 3.6326e-01, 8.6487e-01, ..., -7.3668e-01,
6.1113e-01, -1.0400e+00],
[-6.9459e-01, -5.2950e-01, -3.9443e-01, ..., -2.8397e+00,
-1.7976e+00, -1.3667e+00],
[-2.2847e-01, -4.0381e-01, -5.0446e-01, ..., -3.3453e+00,
-2.0137e+00, -1.1781e+00],
...,
[-6.5845e-01, -5.3323e-01, -3.4674e-01, ..., -2.6147e+00,
-1.7914e+00, -6.2166e-01],
[-2.2226e+00, -1.7481e+00, -1.2933e+00, ..., -1.4665e+00,
-1.3041e+00, -1.0833e+00],
[-6.1865e-01, -8.6571e-01, -4.4573e-01, ..., -2.5647e-01,
-3.7444e-01, -4.0706e-01]],
[[ 1.7217e-01, 1.8231e-01, 4.3962e-01, ..., 7.5030e-02,
1.4278e-01, -2.5388e-01],
[ 3.6175e-01, -6.4046e-01, 2.1413e-01, ..., -2.0230e-01,
2.8724e-01, -7.4549e-01],
[ 4.2792e-02, -7.2046e-01, 2.2952e-01, ..., -4.0023e-02,
3.3460e-01, -6.5482e-01],
...,
[-2.0990e-01, -5.1121e-01, 2.1222e-01, ..., 1.3536e+00,
1.2625e-01, -2.9115e-02],
[-5.1553e-01, -9.7313e-01, 4.3917e-02, ..., 1.0006e-01,
-2.0450e-01, -2.1200e-01],
[ 3.5222e-01, -1.6890e-02, -1.6134e-01, ..., -1.3936e-01,
-8.4951e-02, 1.7743e-01]],
[[-6.7919e-01, -6.8851e-02, 3.7476e-02, ..., 5.9814e-01,
7.2300e-01, 9.7731e-01],
[-1.1793e+00, -6.9349e-01, -9.2997e-01, ..., -5.9109e-01,
-2.5897e-02, 1.0068e+00],
[-9.1967e-01, -6.5931e-01, -4.5795e-01, ..., -2.6300e-01,
6.6684e-01, 7.9548e-01],
...,
[-6.9647e-01, -2.8514e-01, -1.5290e-01, ..., 3.6790e-01,
9.0381e-01, 8.9262e-01],
[-1.1594e+00, -6.1490e-01, -3.0842e-01, ..., 4.9009e-01,
6.4764e-02, 2.4901e-01],
[-3.8359e-01, -4.6406e-02, -1.4574e-01, ..., -2.5676e-01,
-2.1804e-01, 3.7246e-01]]],
[[[-3.6673e-01, -1.0591e+00, -8.7044e-01, ..., -7.4823e-02,
5.6969e-01, 4.9804e-01],
[ 5.8022e-01, -2.4806e+00, -1.8279e+00, ..., -8.7890e-01,
-3.1291e-01, 1.1036e-01],
[-1.2943e+00, -3.3627e+00, 1.4622e-02, ..., -2.1529e-01,
2.2189e-01, -9.9806e-02],
...,
[ 2.8999e-01, -1.4117e+00, -1.2442e+00, ..., -3.7687e-01,
-1.2704e-01, -1.4040e-01],
[ 7.6342e-02, -7.8386e-01, -4.4815e-01, ..., -8.5179e-03,
2.4068e-01, -3.4726e-02],
[-1.9238e-01, -1.0918e+00, -1.0020e+00, ..., -1.0312e+00,
-1.0898e+00, -3.7594e-01]],
[[-2.1377e+00, -4.1973e+00, -3.7391e+00, ..., -7.0791e-01,
-1.1490e+00, -6.4980e-02],
[-6.7230e+00, -9.9994e+00, -1.0802e+01, ..., -4.4930e+00,
-3.8876e+00, -7.2555e-01],
[-8.4062e+00, -1.0439e+01, -1.1034e+01, ..., -3.3428e+00,
-2.9055e+00, -4.3137e-01],
...,
[-3.3523e+00, -5.0137e+00, -4.7379e+00, ..., -1.4325e+00,
-1.6468e+00, -4.0847e-01],
[-5.1414e+00, -6.1199e+00, -5.7856e+00, ..., -2.6052e+00,
-3.4286e+00, -1.0151e+00],
[ 6.6399e-01, -1.7001e-01, 4.8597e-01, ..., 3.8741e-01,
3.8263e-01, 8.7972e-02]],
[[-6.6145e-01, 3.5400e-01, 7.9459e-01, ..., 9.0751e-01,
3.7256e-01, 1.1204e+00],
[-1.1388e+00, -9.0414e-02, 5.5858e-01, ..., 1.2652e+00,
4.0843e-01, 4.8828e-01],
[-2.8623e+00, -2.2936e+00, -1.5271e+00, ..., 5.8772e-01,
8.0932e-01, 4.4091e-01],
...,
[-1.6002e+00, -8.0693e-02, -2.3687e-02, ..., 9.4973e-02,
9.1676e-02, 4.4416e-01],
[-1.3532e+00, 2.0811e-01, 8.4807e-02, ..., 1.8909e-01,
1.6982e-01, 6.1261e-01],
[-5.8211e-01, 2.8724e-01, 1.7524e-01, ..., -8.3375e-02,
1.0160e-01, 3.6683e-01]],
...,
[[ 6.4861e-01, 7.5624e-02, 3.3080e-01, ..., -1.1704e-01,
-3.8016e-01, -9.3721e-01],
[-7.8109e-01, 7.7365e-04, 9.7803e-01, ..., -5.4941e-01,
-1.1901e+00, -8.4735e-01],
[ 3.3152e-01, 6.3285e-01, -2.3386e-01, ..., -1.5237e+00,
-4.1631e-01, -4.0411e-01],
...,
[-1.0447e+00, -9.2008e-01, -1.5298e-01, ..., -1.4277e-01,
-2.1021e-01, -3.7459e-01],
[-2.1980e+00, -1.8151e+00, -1.3712e+00, ..., -8.6476e-01,
-9.1787e-01, -7.9613e-01],
[-4.6022e-01, -1.0041e+00, -3.4392e-01, ..., 1.5812e-02,
-1.1422e-01, -3.1643e-01]],
[[-8.3997e-03, -2.7638e-01, -5.8837e-02, ..., 3.2201e-01,
6.6180e-01, 2.8976e-01],
[ 6.2612e-01, -9.2521e-01, -6.3293e-01, ..., -8.1694e-01,
-4.7784e-01, -5.4463e-02],
[-1.0908e+00, -2.9499e+00, -5.5001e-01, ..., -3.0874e-01,
-1.9852e-01, 1.7314e-01],
...,
[-1.8332e-01, -1.1585e+00, -2.2846e-01, ..., -1.6450e-01,
-1.3498e-01, 1.5329e-01],
[-5.6121e-01, -1.8925e+00, -2.5330e-01, ..., -4.1088e-01,
-3.3302e-01, -5.9952e-02],
[ 6.8338e-01, -1.7669e-01, 2.1492e-01, ..., -1.9659e-01,
-2.0316e-01, 3.2865e-01]],
[[-6.9391e-01, -2.0298e-01, -4.7150e-02, ..., 5.5670e-01,
3.6896e-01, 5.7740e-01],
[-1.2397e+00, -9.2777e-01, -8.8796e-01, ..., -3.2663e-01,
-1.9986e-01, 4.2949e-01],
[-2.1918e+00, -2.7018e+00, -3.4827e+00, ..., -6.3546e-01,
-1.0999e-01, 7.7891e-02],
...,
[-5.1273e-01, -4.7106e-01, -6.3566e-01, ..., 4.5575e-02,
-6.9751e-02, 1.1428e-01],
[-8.9430e-01, -1.0589e+00, -6.4650e-01, ..., -1.1032e-01,
-2.7376e-01, -1.8648e-01],
[-4.9697e-01, -9.1121e-01, -3.8567e-01, ..., -8.8777e-02,
-2.3977e-01, 1.4950e-01]]],
[[[-2.0010e-01, -2.8368e-01, 1.1103e-01, ..., 2.1085e-02,
4.7053e-01, 1.3208e-01],
[-1.2032e-01, -1.5829e+00, -9.4295e-01, ..., -7.3177e-01,
-4.5212e-01, -3.7026e-01],
[-1.5219e-02, -1.5136e+00, -1.0969e+00, ..., -5.5280e-01,
-4.1253e-01, -7.3965e-01],
...,
[-3.0807e-01, -1.7746e+00, -1.2575e+00, ..., -9.8117e-01,
-9.9367e-01, -2.0769e+00],
[-1.1822e+00, -1.7929e+00, -8.5985e-01, ..., -5.6134e-01,
-4.9117e-01, -1.8342e+00],
[-1.3161e+00, -2.2283e+00, -1.6712e+00, ..., -1.4591e+00,
-1.5582e+00, -9.9466e-01]],
[[-2.7774e-01, -1.5648e+00, -7.8602e-01, ..., -7.3830e-01,
-1.4627e+00, -3.1901e-01],
[-2.1183e+00, -4.1246e+00, -2.6015e+00, ..., -2.5087e+00,
-3.4842e+00, -1.1949e+00],
[-2.7852e+00, -5.9608e+00, -9.3802e+00, ..., -2.7634e+00,
-3.1251e+00, -1.2977e+00],
...,
[-3.8287e+00, -5.6251e+00, -5.3095e+00, ..., -5.4737e+00,
-6.2362e+00, -2.9690e+00],
[-6.1940e+00, -7.0833e+00, -6.1201e+00, ..., -6.0645e+00,
-7.3483e+00, -2.6837e+00],
[-7.4609e-02, -9.2093e-01, 2.7369e-01, ..., 3.7758e-01,
6.9770e-01, 1.9993e-01]],
[[-5.3118e-01, -1.8116e-01, 3.4720e-01, ..., 3.8078e-01,
9.9043e-02, 9.2576e-01],
[-3.5916e-01, -4.5194e-01, 4.0414e-01, ..., 3.9169e-01,
2.9140e-01, 5.0647e-01],
[-1.0043e+00, -4.7848e-01, -2.4035e-01, ..., 4.2368e-02,
1.7013e-01, 4.1127e-01],
...,
[-1.4281e+00, -3.3455e-02, 1.7170e-01, ..., 4.5661e-02,
-3.1291e-02, 2.7141e-01],
[-1.2052e+00, 9.7976e-02, -8.6995e-02, ..., -3.7537e-02,
-2.7542e-02, 2.8694e-01],
[-1.1501e+00, -4.2969e-02, 4.1707e-02, ..., 2.7955e-02,
-3.8921e-02, -3.5678e-01]],
...,
[[ 7.0364e-02, -5.8063e-01, -3.5551e-01, ..., -2.2673e-02,
-2.1747e-01, -7.4651e-01],
[-8.8673e-01, -5.7584e-01, -4.6523e-01, ..., -2.7399e-01,
-1.0012e+00, -7.6925e-01],
[-5.6720e-01, -8.6258e-02, 2.7192e-01, ..., 6.2282e-02,
-1.9737e-01, -5.2636e-01],
...,
[-6.1905e-01, -5.9442e-01, -4.4004e-01, ..., -3.6421e-01,
-5.7347e-01, -1.1774e+00],
[-3.1484e+00, -2.9501e+00, -2.0661e+00, ..., -1.9364e+00,
-2.0309e+00, -2.1086e+00],
[-5.6656e-01, -1.6331e+00, -1.0424e+00, ..., -8.6588e-01,
-9.0444e-01, -1.3148e+00]],
[[-1.8420e-01, 3.7187e-02, 4.2032e-01, ..., 8.3119e-02,
5.5241e-01, 1.6148e-01],
[-4.5181e-01, -2.7606e-01, 1.9549e-02, ..., -3.9198e-02,
1.5088e-01, -7.6299e-02],
[-5.0475e-01, -1.7807e-01, 6.1882e-01, ..., -4.7105e-02,
1.1388e-02, -6.8390e-02],
...,
[-1.3862e-01, -1.0417e+00, 1.8870e-01, ..., 2.7152e-01,
1.7684e-01, -7.1525e-01],
[-5.3778e-01, -1.8281e+00, -1.5082e-01, ..., 1.7882e-02,
1.1463e-02, -8.8819e-01],
[ 4.3240e-01, -1.5851e-01, 1.5829e-01, ..., 2.7207e-01,
3.5752e-01, 3.6377e-01]],
[[-6.7009e-01, 3.9727e-01, 5.7227e-01, ..., 4.6595e-01,
3.8757e-01, 6.2642e-01],
[-8.6321e-01, 2.4915e-01, -2.3642e-01, ..., -2.8519e-01,
-1.3440e-01, 6.9343e-01],
[-4.5898e-01, 2.0272e-01, -6.9618e-01, ..., 5.2945e-02,
1.3130e-01, 6.2248e-01],
...,
[-1.1975e+00, -9.8923e-01, -6.1294e-01, ..., -4.3355e-01,
-6.2154e-02, 8.3589e-01],
[-1.9524e+00, -1.9396e+00, -9.8592e-01, ..., -8.6930e-01,
-5.2697e-01, 4.3808e-01],
[-9.4969e-01, -9.8901e-01, -2.6789e-01, ..., -2.8721e-01,
-1.2476e-01, 1.1634e+00]]],
[[[-7.2560e-01, -1.4533e+00, -1.4490e+00, ..., -2.1185e+00,
-1.3647e+00, -2.0788e+00],
[-3.2504e-01, -2.4359e+00, -2.2692e+00, ..., 5.6277e-01,
-3.4667e-01, -2.0108e+00],
[-1.4300e-01, -1.8772e+00, -1.6482e+00, ..., 9.9560e-01,
8.1248e-01, -1.3860e+00],
...,
[-1.0114e-02, -1.3298e+00, -4.3166e+00, ..., -1.4055e-01,
-1.4177e-01, -4.9274e-01],
[-6.9135e-01, -9.1737e-01, -1.0677e+00, ..., 2.0672e-02,
1.4246e-01, -1.9794e-01],
[-9.2247e-01, -1.3783e+00, -1.1599e+00, ..., -1.0950e+00,
-1.1930e+00, -4.8087e-01]],
[[-2.6689e+00, -5.0804e+00, -4.5692e+00, ..., -1.1589e+01,
-1.0621e+01, -4.9956e+00],
[-5.3878e+00, -8.1330e+00, -7.5214e+00, ..., -1.4414e+01,
-1.4830e+01, -8.2127e+00],
[-4.0254e+00, -5.8481e+00, -6.2754e+00, ..., -8.6273e+00,
-8.2511e+00, -3.7260e+00],
...,
[-1.9289e+00, -3.6576e+00, -5.4500e+00, ..., -2.2416e+00,
-2.6419e+00, -9.4126e-01],
[-3.2464e+00, -5.0640e+00, -4.6349e+00, ..., -2.4030e+00,
-3.4439e+00, -9.5515e-01],
[ 2.5470e-01, -2.4919e-01, 1.9117e-01, ..., 1.2390e-01,
3.6146e-01, 2.0775e-01]],
[[-6.5068e-01, 4.7722e-01, 8.4741e-01, ..., 1.0448e+00,
1.3005e+00, 2.3164e+00],
[-9.6346e-01, 6.2018e-02, 4.2835e-01, ..., -1.3668e+00,
-4.4972e-01, 1.5950e+00],
[-1.7301e+00, -1.7544e-01, -6.0808e-03, ..., -8.5858e+00,
-6.3271e+00, -1.7715e+00],
...,
[-7.5201e-01, -3.5101e-01, -1.2206e+01, ..., 2.6758e-02,
4.8972e-01, 4.0609e-01],
[-4.9939e-01, -3.6618e-01, -9.5265e+00, ..., -1.1209e-01,
2.6871e-01, 5.2926e-01],
[-1.8460e-01, -1.1607e-01, -2.5175e+00, ..., -1.7043e-01,
1.1309e-01, 3.2459e-01]],
...,
[[ 1.2331e+00, -3.6402e-01, -6.3823e-02, ..., 1.2512e+00,
4.8422e-01, -1.9435e+00],
[-7.3376e-01, -9.4319e-01, -5.0382e-01, ..., -2.8272e+00,
-3.8911e+00, -3.7618e+00],
[-7.0622e-01, -9.2153e-01, -2.2571e-01, ..., -8.8911e+00,
-5.8849e+00, -4.9188e+00],
...,
[-7.4437e-01, 2.8757e-01, -1.4146e+00, ..., -6.4969e-01,
-7.1072e-01, -5.6451e-01],
[-2.6527e+00, -2.2582e+00, -2.7008e+00, ..., -1.5599e+00,
-1.4520e+00, -1.0439e+00],
[-6.8535e-01, -3.5447e-01, 6.0091e-01, ..., -4.1764e-01,
-3.5123e-01, -3.6518e-01]],
[[ 4.3245e-02, 1.4691e-02, 4.6290e-01, ..., -4.9167e-01,
-2.7085e-01, -1.3741e+00],
[ 2.1682e-01, -8.1347e-01, -3.6928e-02, ..., 4.9799e-01,
7.9912e-01, -2.2922e+00],
[ 1.5950e-01, -6.2071e-01, 4.3261e-01, ..., -3.6867e+00,
-1.9180e+00, -1.8259e+00],
...,
[-3.5522e-01, -1.0734e+00, -4.5031e+00, ..., 2.0819e-01,
1.8519e-01, 5.9735e-02],
[-7.1371e-01, -1.3694e+00, -3.2343e+00, ..., -1.0144e-01,
-9.9138e-02, -1.2691e-01],
[ 3.4854e-01, -1.5980e-01, -3.4596e-01, ..., -1.4640e-01,
-1.0132e-01, 1.8951e-01]],
[[-1.5688e+00, -8.0514e-01, -4.4323e-01, ..., -7.8439e-01,
-4.9133e-01, 8.5567e-01],
[-2.1512e+00, -7.7273e-01, -5.6818e-01, ..., -5.3770e+00,
-3.3106e+00, 7.8991e-01],
[-1.5260e+00, -5.4846e-01, -2.8269e-01, ..., -8.8044e+00,
-4.7433e+00, 8.5785e-01],
...,
[-7.3191e-01, -2.1427e+00, -2.2932e+00, ..., -9.6968e-02,
-1.2901e-02, 6.7882e-01],
[-1.3191e+00, -1.5753e+00, -3.8863e-01, ..., -3.7918e-01,
-1.1157e-01, 2.4945e-01],
[-4.7762e-01, -7.2774e-01, -5.9745e-01, ..., -4.7874e-02,
-4.0497e-02, 3.6044e-01]]]], device='cuda:0'),
tensor([[[[-5.6343e-02, -5.3371e-01, -3.4193e-01, ..., 1.6483e+00,
-3.1772e-01, 2.1231e+00],
[ 1.4034e-01, -1.2370e+00, -8.2776e-01, ..., -8.3885e+00,
-3.9144e+00, -4.8696e+00],
[ 2.2210e-01, -8.5819e-01, -3.6806e-01, ..., -5.4640e+00,
1.1757e+00, 1.8959e+00],
...,
[ 9.9753e-02, 6.9330e-01, -1.7475e+00, ..., -1.5955e+00,
-1.6293e+00, -2.6226e+00],
[-3.8609e-01, 6.5614e-01, -2.3376e+00, ..., -5.8358e-01,
-7.3817e-01, -1.9807e+00],
[-5.9434e-01, -2.3330e-01, -1.2303e+00, ..., -1.7225e+00,
-1.6999e+00, -1.1176e+00]],
[[-1.3172e+00, -2.9657e+00, -2.8252e+00, ..., -4.1799e+00,
-5.9398e+00, 2.2644e-01],
[-2.7360e+00, -4.6514e+00, -4.7960e+00, ..., -7.6125e+00,
-1.2796e+01, -5.4404e+00],
[-2.3652e+00, -3.6162e+00, -5.4650e+00, ..., -5.8810e+00,
-3.0434e+00, -1.7057e+00],
...,
[-2.2713e+00, -6.9464e+00, -8.4130e+00, ..., -6.2042e+00,
-6.7227e+00, -3.6555e+00],
[-4.3599e+00, -7.9657e+00, -8.4275e+00, ..., -8.2924e+00,
-8.9349e+00, -3.6429e+00],
[ 4.6121e-01, 1.0471e+00, 1.6784e-01, ..., -1.6880e-01,
2.8246e-01, -6.1718e-02]],
[[-2.6266e-01, 2.3701e-01, 7.3751e-01, ..., -2.0382e+01,
-1.5676e+01, -5.5349e+00],
[-9.0893e-02, -2.6560e-01, 4.5437e-01, ..., -1.4158e+01,
-1.3536e+01, -6.0650e+00],
[-4.6975e-01, -2.6727e-01, 1.1625e-01, ..., -4.4053e+00,
-8.4238e+00, -4.5252e+00],
...,
[-5.9060e-01, 1.0087e+00, -4.5898e+00, ..., -1.7818e-01,
3.4746e-02, 3.0448e-01],
[-4.5890e-01, 1.4930e+00, -8.3787e+00, ..., -4.3725e-01,
-1.4357e-01, -1.4336e-01],
[-9.4264e-03, 2.1530e+00, -6.5249e+00, ..., -7.0557e-01,
-6.2577e-01, -5.8498e-01]],
...,
[[ 7.3340e-01, 2.0268e-01, 2.8560e-01, ..., -1.6597e+01,
-7.4301e+00, -8.7675e+00],
[-4.1601e-01, -3.3080e-01, -2.7215e-01, ..., -1.4804e+01,
-9.5490e+00, -1.2283e+01],
[-2.3972e-01, -7.0882e-02, 3.0109e-01, ..., -9.2340e+00,
-3.1516e+00, -1.1838e+01],
...,
[-4.7746e-01, 4.3678e-01, 5.0539e-01, ..., -5.5210e-01,
-1.2854e+00, -2.5589e+00],
[-2.3038e+00, -1.2531e+00, -1.5063e+00, ..., -2.7877e+00,
-2.5055e+00, -2.2810e+00],
[-4.7271e-01, -3.3588e-01, 7.7603e-01, ..., -1.3880e+00,
-1.5774e+00, -1.5065e+00]],
[[ 1.5697e-01, 1.7897e-01, 4.4662e-01, ..., -7.0077e+00,
-7.7225e+00, -2.2991e+00],
[ 1.4876e-01, 8.6753e-02, 4.8745e-01, ..., -1.4367e+00,
-2.2545e+00, -1.6749e+00],
[-8.3490e-02, -4.2440e-02, 8.6637e-01, ..., -1.4468e+00,
-1.2344e+00, 4.0256e+00],
...,
[-9.2699e-02, 1.5315e+00, 1.7360e+00, ..., -9.1780e-02,
5.6720e-01, 3.4413e-02],
[-6.1867e-01, 5.7801e-01, 5.6798e-01, ..., -3.7616e-01,
-3.0448e-01, -9.8622e-01],
[ 6.4574e-01, 1.5285e+00, 8.7516e-01, ..., 2.0043e-02,
2.3586e-01, 1.0271e-01]],
[[-7.5972e-01, 1.0852e-01, 3.1674e-01, ..., -2.5272e+01,
-1.2260e+01, -6.4476e+00],
[-1.1993e+00, -4.4035e-02, -2.8725e-01, ..., -2.1697e+01,
-1.4172e+01, -5.6540e+00],
[-9.2358e-01, 1.2577e-01, -2.2781e-01, ..., -1.3241e+01,
-1.6174e+01, -9.1043e+00],
...,
[-9.1646e-01, -4.6525e-01, -2.2405e+00, ..., -6.3347e-01,
3.1365e-01, 1.3955e+00],
[-1.4903e+00, -9.4174e-01, -6.9523e-01, ..., -5.7219e-01,
7.6530e-01, 1.6436e+00],
[-7.2333e-01, -7.4785e-01, 2.2187e+00, ..., -3.2836e-01,
5.3410e-01, 1.6633e+00]]],
[[[-1.7587e-01, -2.5587e-01, 1.8392e-01, ..., 4.8608e-02,
4.7996e-01, 1.5051e-01],
[-1.1852e-01, -1.4673e+00, -8.2602e-01, ..., -9.1321e-01,
-6.1523e-01, -3.2598e-01],
[-1.1466e-01, -1.2855e+00, -6.3760e-01, ..., -5.7659e-01,
-6.0233e-01, -6.8991e-01],
...,
[-6.1891e-02, -1.7552e+00, -1.4655e+00, ..., -5.6799e-01,
-1.5610e+00, -1.9031e+00],
[-1.4176e-01, -1.2875e+00, -7.4374e-01, ..., -2.7782e+00,
-6.2856e-01, -1.7883e+00],
[-5.8771e-02, -1.1487e+00, -1.0797e+00, ..., 1.0855e+00,
8.8168e-02, -1.2016e+00]],
[[-1.7121e-01, -1.3825e+00, -5.6758e-01, ..., -6.0852e-01,
-1.4059e+00, -2.7512e-01],
[-1.9876e+00, -3.9935e+00, -2.4681e+00, ..., -3.0731e+00,
-3.5512e+00, -1.1816e+00],
[-1.8538e+00, -3.3429e+00, -2.5114e+00, ..., -3.8509e+00,
-3.3147e+00, -1.2604e+00],
...,
[-3.6246e+00, -4.9783e+00, -5.2086e+00, ..., -1.1479e+01,
-7.6403e+00, -4.8869e+00],
[-5.1835e+00, -6.0819e+00, -5.5422e+00, ..., -1.3392e+01,
-1.2061e+01, -5.4484e+00],
[ 5.9121e-01, -2.4877e-01, 5.6059e-01, ..., -2.4733e+00,
-2.3692e+00, -1.5395e+00]],
[[-4.6911e-01, -1.4725e-01, 3.1752e-01, ..., 3.2640e-01,
7.6665e-02, 9.2573e-01],
[-3.7219e-01, -4.5268e-01, 4.6736e-01, ..., 2.9743e-01,
1.7343e-01, 4.8436e-01],
[-8.2823e-01, -2.6149e-01, 1.6467e-01, ..., 3.3621e-01,
7.6647e-02, 3.1424e-01],
...,
[-1.7736e+00, -3.7831e-01, -1.2134e-01, ..., -4.9208e+00,
-2.5316e+00, 4.9913e-02],
[-1.4809e+00, -2.9453e-01, -1.7621e-01, ..., -2.7480e+00,
-3.7064e+00, -6.8667e-02],
[-5.4856e-01, -8.6751e-02, 9.9442e-03, ..., -1.6312e+00,
-2.4679e+00, 2.1543e-01]],
...,
[[ 9.7556e-02, -5.5701e-01, -3.4903e-01, ..., -7.4980e-02,
-2.2135e-01, -7.0602e-01],
[-7.8131e-01, -6.2900e-01, -5.7029e-01, ..., -1.9962e-01,
-9.5533e-01, -7.5886e-01],
[-4.6118e-01, -1.1185e-01, -3.8368e-02, ..., -2.7075e-01,
-2.4262e-01, -5.7946e-01],
...,
[-9.0073e-01, -9.8406e-01, -3.1904e-01, ..., -6.0769e+00,
-3.4445e+00, -2.5448e+00],
[-2.2846e+00, -1.8814e+00, -1.3635e+00, ..., -7.0976e+00,
-5.9676e+00, -3.3713e+00],
[-5.2022e-01, -1.0146e+00, -2.0121e-01, ..., -4.9569e+00,
-5.3793e+00, -2.4512e+00]],
[[-1.4342e-01, 3.1681e-02, 4.2233e-01, ..., 1.1126e-01,
5.2550e-01, 1.2190e-01],
[-3.9216e-01, -2.2586e-01, 8.5465e-03, ..., -5.3296e-01,
-5.0959e-02, -6.3384e-02],
[-6.4733e-01, -5.0480e-01, -8.3055e-03, ..., -7.3925e-02,
-1.0137e-01, -6.1911e-03],
...,
[ 7.5039e-02, -7.0846e-01, 3.4402e-01, ..., -3.3943e+00,
-1.0950e+00, -5.3037e-01],
[-5.8480e-01, -1.8142e+00, -9.4537e-02, ..., -4.5259e+00,
-5.7488e-01, -8.1314e-01],
[ 7.6406e-01, 4.1128e-02, 3.5835e-01, ..., 1.5348e+00,
2.9230e+00, 5.8922e-01]],
[[-6.6309e-01, 4.0429e-01, 5.4477e-01, ..., 4.6858e-01,
3.6663e-01, 6.0401e-01],
[-8.2923e-01, 2.3535e-01, -2.5692e-01, ..., -3.2012e-01,
-1.9733e-01, 6.1075e-01],
[-4.7283e-01, 2.9615e-01, 2.3867e-02, ..., -3.5918e-02,
7.2180e-02, 5.8120e-01],
...,
[-1.2722e+00, -5.6349e-01, -1.5046e-01, ..., -3.6417e+00,
-1.0640e+00, 2.2686e+00],
[-1.2302e+00, -7.0251e-01, -8.8821e-02, ..., -4.9655e+00,
-2.3429e+00, 1.8064e+00],
[-5.3052e-01, -4.2252e-01, 1.6741e-01, ..., -4.3203e+00,
-7.6179e-01, 1.9048e+00]]],
[[[ 1.9360e-02, -2.2695e-01, -1.2649e-02, ..., -9.6148e-01,
-5.2980e-01, -1.4491e+00],
[ 3.2009e-01, -9.0808e-01, -6.8887e-01, ..., -1.1672e+00,
-1.3130e+00, -1.4917e+00],
[ 2.5006e-01, -5.5895e-01, -4.7419e-01, ..., -8.9850e-01,
-1.0712e+00, -1.8086e+00],
...,
[-3.4560e-02, -6.2055e-01, -4.2937e-01, ..., -8.7616e-01,
-1.0251e+00, -1.8974e+00],
[-6.8797e-01, -7.0272e-01, -5.8581e-02, ..., -4.4091e-01,
-5.6970e-01, -1.6996e+00],
[-7.6143e-01, -1.3525e+00, -1.1939e+00, ..., -1.1313e+00,
-1.5425e+00, -7.5871e-01]],
[[-7.4685e-01, -2.1259e+00, -1.7999e+00, ..., -3.9828e+00,
-5.4766e+00, -1.8507e+00],
[-1.9983e+00, -3.7111e+00, -2.9425e+00, ..., -5.9531e+00,
-8.0556e+00, -3.4164e+00],
[-1.5526e+00, -2.5386e+00, -2.5001e+00, ..., -5.0212e+00,
-6.1530e+00, -2.5839e+00],
...,
[-1.7934e+00, -2.4011e+00, -2.4057e+00, ..., -4.9522e+00,
-6.2694e+00, -2.5145e+00],
[-3.8304e+00, -4.4346e+00, -3.6149e+00, ..., -6.4638e+00,
-7.9431e+00, -2.5559e+00],
[ 2.5726e-01, -3.5006e-01, 3.7428e-01, ..., 4.3918e-01,
4.7337e-01, 2.1258e-01]],
[[-2.2681e-01, 1.4013e-01, 7.2362e-01, ..., 6.5087e-01,
2.5551e-01, 1.5057e+00],
[ 5.7177e-02, -3.9160e-01, 5.8241e-01, ..., 5.9896e-01,
-5.2690e-02, 9.7498e-02],
[-1.1729e-01, -2.3285e-01, 2.5434e-01, ..., 5.9850e-02,
-1.2878e-02, -1.3108e-01],
...,
[-2.7274e-01, -2.2477e-01, 1.0858e-01, ..., 1.1440e-02,
-9.0775e-02, -1.9959e-01],
[-2.5009e-01, -1.8152e-01, 1.7291e-01, ..., -1.0997e-01,
-1.4115e-02, -2.1930e-01],
[-1.3888e-01, -4.8127e-01, 1.0755e-01, ..., 1.0158e-01,
2.4585e-01, -5.4709e-01]],
...,
[[ 5.9911e-01, -2.9399e-01, -3.1249e-02, ..., 4.3499e-01,
3.4342e-01, -1.2891e+00],
[-5.1392e-01, -5.6933e-01, -5.4125e-01, ..., -1.3825e-01,
-7.7113e-01, -1.3834e+00],
[-3.8149e-01, -2.2184e-01, -2.5434e-01, ..., -9.5260e-02,
-7.6693e-02, -9.4077e-01],
...,
[-4.4556e-01, -1.6176e-01, -1.7570e-01, ..., -1.0533e-01,
-9.9002e-02, -9.4360e-01],
[-2.3734e+00, -1.7455e+00, -1.2592e+00, ..., -1.5672e+00,
-1.5796e+00, -1.8253e+00],
[-4.8475e-01, -8.4505e-01, -4.0428e-01, ..., -3.5075e-01,
-5.4980e-01, -1.1899e+00]],
[[ 1.5495e-01, 2.5544e-01, 6.2670e-01, ..., -9.7227e-02,
3.7495e-01, -3.7963e-01],
[ 1.4737e-01, 2.4493e-01, 4.4777e-01, ..., 1.8315e-01,
8.1687e-02, -8.5722e-01],
[-1.1499e-01, 9.5628e-02, 2.8892e-01, ..., 2.6623e-01,
1.4070e-01, -4.7530e-01],
...,
[-2.1690e-01, -2.3125e-02, 3.1110e-01, ..., 2.4840e-01,
1.2391e-01, -5.0724e-01],
[-5.7184e-01, -7.2253e-01, -3.0537e-02, ..., -1.7890e-01,
-3.0875e-01, -9.4951e-01],
[ 5.2973e-01, 3.9665e-01, 7.3317e-02, ..., 3.3314e-01,
3.1909e-01, 2.8953e-01]],
[[-8.4352e-01, 2.6653e-01, 3.5031e-01, ..., -6.0439e-02,
-8.0324e-02, 5.8786e-01],
[-1.1338e+00, 2.3243e-01, -3.4136e-01, ..., -8.9544e-01,
-4.3358e-01, 8.8378e-01],
[-9.5843e-01, 3.7263e-01, -1.1989e-01, ..., -5.2090e-01,
-5.2459e-01, 6.1312e-01],
...,
[-9.7005e-01, 1.7107e-01, -9.4828e-02, ..., -5.4597e-01,
-5.3386e-01, 6.0645e-01],
[-1.7039e+00, -4.1464e-01, -3.4710e-01, ..., -7.0296e-01,
-6.9996e-01, 3.8103e-01],
[-7.0934e-01, -7.4989e-02, -2.8193e-01, ..., -4.8319e-01,
-4.4348e-01, 9.9166e-01]]],
[[[-1.1450e-01, -4.7645e-01, -1.5846e-01, ..., -1.0081e+00,
6.4062e-01, 3.3736e-01],
[ 3.2375e-01, -1.0619e+00, -5.2123e-01, ..., -1.0662e+00,
-8.9410e-01, -7.3986e-01],
[ 2.3733e-01, -7.5059e-01, -2.9864e-01, ..., -1.9514e-01,
-4.3051e-01, -8.8041e-01],
...,
[-2.3999e-01, -1.6227e+00, -1.2171e+00, ..., -9.1255e-01,
-9.6617e-01, -2.0210e+00],
[-1.1220e+00, -1.7192e+00, -7.8884e-01, ..., -5.4638e-01,
-4.6962e-01, -1.7764e+00],
[-1.3143e+00, -2.1293e+00, -1.5828e+00, ..., -1.4279e+00,
-1.5637e+00, -9.7279e-01]],
[[-1.3919e+00, -3.0024e+00, -2.6484e+00, ..., -7.3942e+00,
-6.2049e+00, -1.6803e+00],
[-2.9978e+00, -4.7987e+00, -3.4518e+00, ..., -5.8723e+00,
-7.2430e+00, -3.5038e+00],
[-2.0113e+00, -2.9875e+00, -2.3968e+00, ..., -2.4837e+00,
-3.0595e+00, -1.5962e+00],
...,
[-3.6690e+00, -5.4530e+00, -5.2461e+00, ..., -5.4755e+00,
-6.1656e+00, -2.8960e+00],
[-6.1301e+00, -6.9584e+00, -6.1530e+00, ..., -6.1363e+00,
-7.4064e+00, -2.6901e+00],
[-4.7770e-02, -9.0135e-01, 2.5330e-01, ..., 3.8594e-01,
6.8966e-01, 1.8425e-01]],
[[-3.9315e-01, 2.4922e-01, 8.3784e-01, ..., -3.3827e+00,
-2.0630e+00, 1.3493e-01],
[-2.8641e-01, -1.6177e-02, 5.8943e-01, ..., -1.7386e+00,
-1.9783e+00, -4.1696e-01],
[-7.0340e-01, 1.0056e-01, 3.3186e-01, ..., -5.5130e-02,
8.6001e-02, 4.2517e-01],
...,
[-1.4078e+00, -1.0658e-01, 5.0455e-02, ..., 1.1759e-01,
6.5566e-02, 3.1040e-01],
[-1.1955e+00, -1.1452e-02, -1.0692e-01, ..., -2.7555e-02,
4.3817e-02, 3.1731e-01],
[-1.1216e+00, -9.4414e-02, 3.4402e-02, ..., -3.2102e-02,
-3.5854e-02, -3.3309e-01]],
...,
[[ 9.4613e-01, 2.4402e-01, 2.0232e-01, ..., -1.5340e+00,
-1.1537e+00, -1.3879e+00],
[-1.0057e+00, -8.8661e-01, -6.8209e-01, ..., 3.4516e-01,
-9.0835e-01, -1.4899e+00],
[-4.6880e-01, -3.4414e-01, -2.8552e-01, ..., 8.6965e-02,
-6.9095e-01, -1.1598e+00],
...,
[-5.2349e-01, -5.7004e-01, -3.3731e-01, ..., -3.9198e-01,
-5.8447e-01, -1.1851e+00],
[-3.1233e+00, -2.9378e+00, -2.0961e+00, ..., -1.9361e+00,
-2.0242e+00, -2.0591e+00],
[-5.2528e-01, -1.6348e+00, -1.0793e+00, ..., -8.3385e-01,
-9.2344e-01, -1.3189e+00]],
[[ 8.3286e-02, 3.3509e-01, 7.6664e-01, ..., -2.5186e+00,
-2.2043e+00, -2.0657e+00],
[ 9.9095e-02, -3.4902e-01, 2.4966e-01, ..., 5.6899e-01,
-2.8134e-01, -1.7592e+00],
[-1.9363e-01, -5.2311e-01, 2.3588e-01, ..., 2.8658e-01,
8.2644e-01, 4.1089e-01],
...,
[-7.6111e-03, -8.5845e-01, 7.6276e-02, ..., 2.6596e-01,
1.3447e-01, -7.1677e-01],
[-5.0017e-01, -1.7450e+00, -1.9225e-01, ..., -1.6485e-02,
-6.0822e-02, -9.3821e-01],
[ 4.3027e-01, -9.5270e-02, 1.7695e-01, ..., 2.9454e-01,
3.0170e-01, 3.1661e-01]],
[[-8.9893e-01, -2.2559e-01, 9.3865e-02, ..., -2.8753e+00,
-2.5667e+00, -7.7006e-01],
[-1.0490e+00, -2.7603e-01, -3.5070e-01, ..., -2.2649e+00,
-1.7980e+00, 1.2752e-02],
[-6.3899e-01, -2.1455e-01, -3.1520e-01, ..., 6.4895e-01,
-5.1435e-02, 6.7889e-01],
...,
[-1.2059e+00, -8.7403e-01, -4.0124e-01, ..., -3.9176e-01,
-7.2132e-02, 8.1753e-01],
[-1.9613e+00, -1.8159e+00, -9.4065e-01, ..., -7.6164e-01,
-4.4310e-01, 4.2490e-01],
[-9.8029e-01, -9.8619e-01, -2.2092e-01, ..., -2.4299e-01,
-1.1643e-01, 1.1345e+00]]],
[[[-7.2654e-01, -1.4173e+00, -1.3878e+00, ..., -6.1818e-03,
6.3671e-01, 5.5617e-01],
[-2.8980e-01, -2.4114e+00, -2.1657e+00, ..., -4.6580e-01,
-2.3392e-01, 1.2353e-01],
[-3.7083e-01, -2.0838e+00, -1.8339e+00, ..., -3.0347e-01,
-1.4361e-01, -8.7220e-02],
...,
[-4.9495e-01, -1.9259e+00, -1.8325e+00, ..., -3.1649e-01,
-7.0301e-02, -1.1589e-01],
[-9.4478e-01, -1.6793e+00, -1.1734e+00, ..., 1.7327e-02,
2.9062e-01, -8.1002e-03],
[-7.8398e-01, -1.6132e+00, -1.2180e+00, ..., -1.0484e+00,
-1.0865e+00, -3.8463e-01]],
[[-2.3644e+00, -4.7758e+00, -4.0855e+00, ..., -4.2098e-01,
-1.0148e+00, -6.1214e-03],
[-5.3684e+00, -8.1367e+00, -6.4910e+00, ..., -1.7570e+00,
-2.6421e+00, -6.8309e-01],
[-4.3508e+00, -6.2793e+00, -5.3867e+00, ..., -1.2991e+00,
-1.4846e+00, -3.7523e-01],
...,
[-4.3518e+00, -6.0737e+00, -5.4398e+00, ..., -1.2819e+00,
-1.5472e+00, -3.4427e-01],
[-6.3518e+00, -7.3631e+00, -6.2476e+00, ..., -2.4426e+00,
-3.3295e+00, -9.4390e-01],
[-1.3553e-02, -8.3429e-01, 3.9620e-01, ..., 3.4022e-01,
3.5837e-01, 8.7255e-02]],
[[-5.7866e-01, 5.9398e-01, 1.1048e+00, ..., 7.9583e-01,
2.7735e-01, 1.0743e+00],
[-8.8432e-01, -4.9949e-02, 4.3705e-01, ..., 3.9869e-01,
3.5159e-02, 4.5291e-01],
[-1.6204e+00, -8.8051e-02, -1.2996e-02, ..., 7.1959e-02,
9.0428e-02, 4.0042e-01],
...,
[-1.6944e+00, -9.3501e-02, -6.7903e-02, ..., 3.6626e-02,
6.4984e-02, 4.1272e-01],
[-1.5124e+00, 8.3656e-02, 6.5170e-02, ..., 1.2062e-01,
1.2142e-01, 6.0475e-01],
[-1.3933e+00, -4.9623e-01, -2.3800e-01, ..., -8.4234e-02,
8.1853e-02, 3.6492e-01]],
...,
[[ 1.0327e+00, -4.3587e-01, -2.1945e-01, ..., -1.1241e-01,
-3.7447e-01, -9.2325e-01],
[-8.2845e-01, -9.5663e-01, -8.3196e-01, ..., -6.6084e-01,
-1.2783e+00, -8.2981e-01],
[-6.4600e-01, -9.6663e-01, -7.3096e-01, ..., -1.0566e-01,
-2.1005e-01, -3.9731e-01],
...,
[-7.5463e-01, -1.1534e+00, -7.4318e-01, ..., -8.6752e-02,
-1.8127e-01, -3.7627e-01],
[-2.7298e+00, -2.7969e+00, -1.8302e+00, ..., -8.2438e-01,
-8.8152e-01, -8.0877e-01],
[-6.5743e-01, -1.6511e+00, -9.7145e-01, ..., -1.9516e-02,
-1.5095e-01, -3.3824e-01]],
[[ 1.5686e-02, -1.4748e-02, 5.9148e-01, ..., 3.3164e-01,
6.7600e-01, 3.1862e-01],
[ 1.3813e-03, -7.4986e-01, 9.3818e-02, ..., -1.0684e-02,
1.0707e-01, -3.1896e-02],
[-5.0796e-02, -8.0981e-01, 2.5829e-01, ..., -1.3239e-01,
-1.0825e-01, 1.8411e-01],
...,
[ 1.7002e-02, -8.2181e-01, 2.9880e-01, ..., -1.4196e-01,
-1.1450e-01, 1.6350e-01],
[-2.8476e-01, -1.4950e+00, 4.1943e-02, ..., -3.8489e-01,
-3.2973e-01, -7.7918e-02],
[ 6.5981e-01, 1.2402e-01, 2.8054e-01, ..., -2.4226e-01,
-2.3327e-01, 3.1135e-01]],
[[-1.5923e+00, -5.9630e-01, 5.7535e-02, ..., 5.5330e-01,
3.2744e-01, 5.3930e-01],
[-2.0431e+00, -3.1311e-01, -4.8944e-03, ..., -4.5471e-02,
-1.4056e-01, 4.1421e-01],
[-1.6590e+00, -3.3544e-01, 2.3562e-01, ..., 4.5872e-02,
-1.5890e-01, 9.5726e-02],
...,
[-1.7154e+00, -5.2215e-01, 2.1172e-01, ..., 7.2184e-02,
-8.2832e-02, 9.6583e-02],
[-2.1346e+00, -1.1380e+00, -1.6933e-01, ..., -9.9735e-02,
-2.6351e-01, -2.0826e-01],
[-1.0200e+00, -1.6754e-01, 4.9519e-01, ..., -1.0096e-01,
-2.8450e-01, 1.0160e-01]]]], device='cuda:0'),
tensor([[[[-3.5735e-01, -9.5952e-01, -5.9828e-01, ..., -5.4698e-01,
-3.5265e-02, -7.4327e-01],
[ 2.1565e-02, -1.8298e+00, -1.1642e+00, ..., -9.9752e-01,
-8.3486e-01, -1.1564e+00],
[ 3.1571e-02, -1.4651e+00, -9.1674e-01, ..., -7.9731e-01,
-8.0794e-01, -1.7113e+00],
...,
[-6.5009e-02, -3.6052e-01, -9.2284e-01, ..., -5.3027e-01,
-1.5234e-01, -4.2625e-01],
[-6.9767e-01, -5.4179e-01, -9.8312e-01, ..., -1.5523e-01,
1.3286e-01, -1.5648e-01],
[-8.8263e-01, -1.4663e+00, -3.6815e+00, ..., -1.1318e+00,
-1.1602e+00, -4.5850e-01]],
[[-1.9919e+00, -3.9648e+00, -4.3849e+00, ..., -3.9302e+00,
-4.9272e+00, -1.9794e+00],
[-3.7425e+00, -6.0039e+00, -5.8149e+00, ..., -5.5736e+00,
-6.8785e+00, -2.9792e+00],
[-3.4670e+00, -5.2954e+00, -5.0757e+00, ..., -5.6036e+00,
-5.7363e+00, -2.8035e+00],
...,
[-1.8196e+00, -3.2991e+00, -4.6668e+00, ..., -2.2180e+00,
-2.5358e+00, -8.9823e-01],
[-3.2056e+00, -3.6185e+00, -6.3693e+00, ..., -2.2990e+00,
-3.3897e+00, -9.5390e-01],
[ 2.4723e-01, -6.7196e-01, 9.6153e-01, ..., 2.4402e-01,
4.1565e-01, 2.3150e-01]],
[[-4.7584e-01, 4.3783e-01, 9.0191e-01, ..., 8.0443e-01,
5.6135e-01, 1.4802e+00],
[-6.9073e-01, -1.0045e-02, 4.2407e-01, ..., 4.3487e-01,
2.5119e-01, 3.8549e-01],
[-1.1659e+00, 1.1850e-01, 1.7393e-01, ..., 3.2062e-01,
1.1494e-01, 3.0818e-01],
...,
[-7.7076e-01, 4.2221e-01, -9.7663e-01, ..., -6.0370e-01,
3.6197e-01, 3.9889e-01],
[-5.5891e-01, 4.1071e-01, 1.5145e-01, ..., -4.2734e-01,
1.9148e-01, 5.6518e-01],
[-2.2599e-01, 1.6212e-01, 1.8590e+00, ..., -1.3581e-01,
8.2160e-02, 3.5670e-01]],
...,
[[ 1.0671e+00, 3.6206e-01, 3.3881e-01, ..., 4.8497e-01,
2.0709e-01, -1.0524e+00],
[-7.5160e-01, -5.5257e-01, -1.3193e-01, ..., -3.6787e-01,
-1.2675e+00, -1.3852e+00],
[-2.2360e-01, -3.8192e-01, -2.1688e-01, ..., -2.8402e-01,
-5.5940e-01, -1.1593e+00],
...,
[-8.0225e-01, -4.0067e-01, -2.3313e-02, ..., -7.3537e-01,
-6.4865e-01, -5.3098e-01],
[-2.6400e+00, -1.9888e+00, -8.2195e-01, ..., -1.4963e+00,
-1.3675e+00, -1.0561e+00],
[-6.6910e-01, -1.4609e+00, -7.2944e-02, ..., -2.2751e-01,
-3.1021e-01, -3.7247e-01]],
[[ 1.7609e-01, 2.0741e-01, 3.9510e-01, ..., 2.6214e-01,
7.6547e-01, -2.3872e-01],
[ 3.3543e-01, -6.6202e-01, 9.4382e-02, ..., 1.7142e-01,
3.8153e-01, -7.7979e-01],
[ 2.1720e-02, -7.2793e-01, 1.5571e-01, ..., 1.4436e-01,
1.5375e-01, -6.8058e-01],
...,
[-4.0149e-01, -1.2574e-01, -1.0893e+00, ..., 7.2131e-01,
3.6168e-01, 3.8273e-02],
[-7.5303e-01, -1.1002e+00, -3.0208e+00, ..., 1.9754e-01,
-9.5571e-02, -1.3726e-01],
[ 3.0839e-01, 4.3319e-02, -1.6657e+00, ..., -1.1895e-01,
-1.3704e-01, 1.9493e-01]],
[[-6.6921e-01, -8.7290e-02, 3.5315e-01, ..., 3.4240e-01,
4.2409e-01, 9.8405e-01],
[-1.1323e+00, -7.0598e-01, -7.1093e-01, ..., -6.5665e-01,
-2.5586e-01, 1.0020e+00],
[-8.6620e-01, -6.8750e-01, -5.4215e-01, ..., -3.0490e-01,
-4.8779e-02, 7.5643e-01],
...,
[-6.8849e-01, -6.9034e-01, -1.7865e+00, ..., 6.8685e-01,
3.2679e-01, 6.1441e-01],
[-1.3235e+00, -9.3621e-01, -1.0595e+00, ..., 1.3506e-01,
1.6702e-02, 1.5836e-01],
[-4.4984e-01, -1.7671e-01, -3.0228e-01, ..., -7.6741e-02,
-1.5922e-01, 3.2135e-01]]],
[[[-1.7208e-01, -2.9653e-01, 1.2726e-01, ..., 6.9422e-01,
1.5633e+00, 2.7697e-01],
[-4.3989e-02, -1.4521e+00, -9.2199e-01, ..., -4.3091e-01,
6.8084e-02, -2.0164e-01],
[ 3.8040e-02, -1.3128e+00, -9.3299e-01, ..., 3.9829e-01,
1.8839e-01, -7.0916e-01],
...,
[ 7.8909e-02, -5.7662e+00, -6.1975e+00, ..., -1.0084e+00,
-9.4304e-01, -2.0441e+00],
[ 4.1620e-01, -2.6018e+00, -3.9700e+00, ..., -5.3633e-01,
-4.6062e-01, -1.8159e+00],
[-2.5632e-01, -1.0287e+00, -2.0870e+00, ..., -1.4316e+00,
-1.5688e+00, -9.8730e-01]],
[[-2.4145e-01, -1.6547e+00, -9.8568e-01, ..., -1.6113e+00,
-2.1506e+00, -3.1782e-01],
[-2.0382e+00, -4.2874e+00, -3.5251e+00, ..., -3.2851e+00,
-5.0573e+00, -1.3718e+00],
[-2.0089e+00, -4.0000e+00, -3.9314e+00, ..., -1.9060e+00,
-4.7541e+00, -1.3588e+00],
...,
[-8.3532e+00, -1.3337e+01, -1.2678e+01, ..., -5.5088e+00,
-6.1961e+00, -2.9051e+00],
[-6.2659e+00, -8.5631e+00, -8.1822e+00, ..., -6.1508e+00,
-7.4610e+00, -2.7231e+00],
[-5.1271e-01, -1.4457e+00, -1.6110e+00, ..., 3.8983e-01,
6.8364e-01, 1.9930e-01]],
[[-4.5246e-01, -3.5275e-02, 3.1977e-01, ..., 4.1820e-01,
7.5833e-01, 9.9789e-01],
[-3.2865e-01, -3.9796e-01, 2.7974e-01, ..., -3.5482e+00,
5.3474e-01, 5.3725e-01],
[-7.4681e-01, -3.4931e-01, -2.2174e-01, ..., -3.6677e+00,
-3.4068e-02, 4.2059e-01],
...,
[-8.1082e+00, -9.4443e+00, -9.2868e+00, ..., 1.2709e-01,
1.1855e-02, 3.1567e-01],
[-2.4978e+00, -3.3187e+00, -7.5332e-01, ..., -3.0462e-02,
-2.2967e-02, 2.9320e-01],
[ 2.2437e-01, -3.8766e-01, 6.9844e-01, ..., 3.0369e-02,
-5.6410e-02, -3.2438e-01]],
...,
[[ 1.9743e-01, -4.5458e-01, -2.8209e-01, ..., -2.0373e+00,
-1.1801e+00, -8.0021e-01],
[-6.3830e-01, -4.6854e-01, -4.1981e-01, ..., -5.6632e+00,
-3.8144e+00, -9.2204e-01],
[-2.9226e-01, 1.1746e-01, 3.3098e-01, ..., -4.9035e+00,
-3.1703e+00, -6.2256e-01],
...,
[ 2.6617e+00, -1.1846e+00, -2.9650e+00, ..., -3.1326e-01,
-5.7939e-01, -1.1884e+00],
[-1.8716e+00, -1.5040e+00, -3.0958e+00, ..., -1.9096e+00,
-2.0680e+00, -2.0742e+00],
[-1.9109e+00, -3.0765e+00, -3.5438e+00, ..., -8.5973e-01,
-9.2986e-01, -1.3627e+00]],
[[-1.0483e-01, 1.1277e-01, 4.5040e-01, ..., 9.9667e-01,
1.4111e+00, 1.7096e-01],
[-2.4165e-01, -1.1432e-01, 1.5542e-01, ..., -6.3048e-01,
1.3461e-01, -2.1576e-01],
[-4.4767e-01, -2.6806e-01, 2.0080e-01, ..., 9.5935e-01,
7.8217e-01, -1.0159e-01],
...,
[-2.3990e+00, -5.2117e+00, -4.3969e+00, ..., 2.4225e-01,
1.5384e-01, -7.3693e-01],
[-8.7649e-01, -3.6425e+00, -1.9966e+00, ..., -1.5165e-03,
-2.5602e-02, -9.4613e-01],
[ 1.0285e+00, -2.7573e-01, -1.0688e+00, ..., 2.4999e-01,
3.0929e-01, 3.2630e-01]],
[[-5.8546e-01, 4.2329e-01, 5.0098e-01, ..., -1.6803e-01,
6.0631e-01, 6.5572e-01],
[-7.8357e-01, 3.0661e-01, -1.5974e-01, ..., -2.1019e+00,
3.8846e-01, 7.1635e-01],
[-4.7151e-01, 2.6243e-01, -1.6214e-01, ..., -2.1568e+00,
3.9266e-01, 6.4599e-01],
...,
[-3.7231e-01, -3.9002e+00, -5.3254e+00, ..., -3.6234e-01,
-6.1917e-02, 8.2782e-01],
[-1.0735e+00, -1.0004e+00, -2.0176e+00, ..., -8.7067e-01,
-4.6504e-01, 4.4552e-01],
[-8.1192e-01, 3.2799e-01, 1.4330e+00, ..., -2.9803e-01,
-1.4594e-01, 1.1362e+00]]],
[[[-7.4972e-01, -1.3966e+00, -1.3880e+00, ..., -4.4531e-02,
5.6333e-01, 4.7834e-01],
[-2.8041e-01, -2.4251e+00, -2.1859e+00, ..., -5.2117e-01,
-2.5877e-01, 7.5454e-02],
[-3.6711e-01, -2.1645e+00, -1.7651e+00, ..., -3.2487e-01,
-1.3092e-01, -9.7378e-02],
...,
[-5.0003e-01, -1.8460e+00, -1.7960e+00, ..., -3.5322e-01,
-6.1858e-02, -8.4770e-02],
[-9.8341e-01, -1.6757e+00, -1.1212e+00, ..., 5.3998e-02,
3.4418e-01, 9.9842e-03],
[-7.9981e-01, -1.6250e+00, -1.2175e+00, ..., -1.2078e+00,
-1.2912e+00, -4.8026e-01]],
[[-2.3687e+00, -4.8729e+00, -4.1897e+00, ..., -4.7097e-01,
-1.0198e+00, -3.6084e-02],
[-5.4395e+00, -8.3227e+00, -6.6432e+00, ..., -1.7957e+00,
-2.5998e+00, -6.3310e-01],
[-4.4164e+00, -6.4163e+00, -6.1379e+00, ..., -1.3472e+00,
-1.5737e+00, -3.3088e-01],
...,
[-4.3243e+00, -6.1771e+00, -5.6271e+00, ..., -1.2974e+00,
-1.5051e+00, -3.3480e-01],
[-6.3750e+00, -7.4949e+00, -6.4140e+00, ..., -2.7606e+00,
-3.7008e+00, -1.0475e+00],
[-2.3551e-02, -8.7934e-01, 3.5227e-01, ..., 3.6538e-01,
3.5606e-01, 5.6287e-02]],
[[-6.0178e-01, 6.4485e-01, 1.1380e+00, ..., 7.7651e-01,
2.9784e-01, 1.0164e+00],
[-9.5797e-01, 4.6847e-03, 4.9739e-01, ..., 3.7945e-01,
3.3938e-02, 4.0613e-01],
[-1.7100e+00, -1.3765e-01, -1.2761e-01, ..., 1.9616e-02,
4.2734e-02, 3.4636e-01],
...,
[-1.7906e+00, -9.3965e-02, -9.8482e-02, ..., 4.5824e-02,
6.6647e-02, 4.4869e-01],
[-1.6510e+00, -2.4685e-02, -9.2365e-03, ..., 1.2592e-01,
1.0564e-01, 6.1339e-01],
[-1.4621e+00, -5.7387e-01, -2.8850e-01, ..., -1.0089e-01,
6.1805e-02, 3.8926e-01]],
...,
[[ 1.0397e+00, -4.2662e-01, -1.9475e-01, ..., -7.3151e-02,
-3.4467e-01, -8.8476e-01],
[-7.7768e-01, -9.3617e-01, -8.1665e-01, ..., -5.8395e-01,
-1.1520e+00, -8.0402e-01],
[-6.0096e-01, -8.7697e-01, -7.8668e-01, ..., -9.1691e-02,
-1.8206e-01, -4.1314e-01],
...,
[-6.9006e-01, -1.1380e+00, -7.7761e-01, ..., -1.5597e-01,
-2.4680e-01, -4.1989e-01],
[-2.7768e+00, -2.8548e+00, -1.8928e+00, ..., -9.5942e-01,
-1.0366e+00, -8.9145e-01],
[-6.6071e-01, -1.6620e+00, -1.0141e+00, ..., -2.8234e-02,
-1.6005e-01, -4.1106e-01]],
[[ 1.6133e-02, 5.0803e-03, 5.6418e-01, ..., 2.8975e-01,
6.0584e-01, 2.5086e-01],
[ 2.4735e-02, -7.1270e-01, 8.5765e-02, ..., -1.2512e-02,
4.1475e-02, -2.8659e-02],
[-1.6555e-02, -7.5775e-01, 2.9096e-01, ..., -1.5432e-01,
-9.7760e-02, 1.9887e-01],
...,
[ 2.3653e-01, -7.3060e-01, 2.1887e-01, ..., -1.6689e-01,
-1.4127e-01, 1.5849e-01],
[-1.6201e-01, -1.4013e+00, -4.4233e-03, ..., -4.8099e-01,
-4.1922e-01, -1.4788e-01],
[ 6.8731e-01, 1.6606e-01, 3.0426e-01, ..., -2.9966e-01,
-3.1467e-01, 2.8094e-01]],
[[-1.5702e+00, -5.6523e-01, 5.0492e-02, ..., 5.3002e-01,
3.4002e-01, 5.0411e-01],
[-2.0567e+00, -2.3195e-01, 6.3074e-03, ..., 5.0763e-03,
-1.8357e-01, 3.5845e-01],
[-1.7853e+00, -3.2773e-01, 3.3321e-01, ..., 1.2501e-01,
-1.6640e-01, 5.6621e-02],
...,
[-1.7730e+00, -3.7802e-01, 2.3231e-01, ..., 7.5266e-02,
-1.1286e-01, 6.9030e-02],
[-2.2327e+00, -1.0889e+00, -1.8305e-01, ..., -1.2143e-01,
-2.9288e-01, -2.6812e-01],
[-1.0591e+00, -1.8516e-01, 4.6406e-01, ..., -1.8451e-01,
-3.5353e-01, 7.5110e-02]]],
[[[ 2.2473e-02, -3.1591e-01, -1.2832e-01, ..., -2.5801e+00,
-6.5023e-01, -2.3626e+00],
[ 2.6817e-01, -1.0174e+00, -7.5966e-01, ..., -5.1033e+00,
-4.3974e+00, -2.4796e+00],
[ 2.3816e-01, -6.6419e-01, -5.5955e-01, ..., -5.1457e+00,
-3.4742e+00, 3.5252e+00],
...,
[ 4.6530e-02, 2.9483e-01, 3.5079e-01, ..., -1.4385e+00,
-1.6136e+00, -1.9867e+00],
[-5.4664e-01, -1.3189e-01, 8.9755e-01, ..., -8.6485e-01,
-5.7961e-01, -1.5203e+00],
[-7.0542e-01, -1.1197e+00, -7.8864e-01, ..., -1.1582e+00,
-1.3186e+00, -8.1429e-01]],
[[-8.5173e-01, -2.3508e+00, -2.1907e+00, ..., -9.3358e+00,
-1.1827e+01, -4.6428e+00],
[-2.2179e+00, -3.9938e+00, -3.3864e+00, ..., -1.4976e+01,
-1.8165e+01, -7.1269e+00],
[-1.8271e+00, -2.8431e+00, -2.9866e+00, ..., -1.0235e+01,
-6.7873e+00, -6.1228e-01],
...,
[-1.9619e+00, -4.0687e+00, -5.5294e+00, ..., -5.1955e+00,
-6.0664e+00, -3.1574e+00],
[-3.9412e+00, -4.6411e+00, -4.0957e+00, ..., -6.2988e+00,
-7.1362e+00, -2.7160e+00],
[ 3.1554e-01, -2.4822e-01, 5.7898e-01, ..., 4.0053e-01,
5.8155e-01, 8.2242e-02]],
[[-1.8964e-01, 2.1355e-01, 7.4575e-01, ..., -6.1729e-02,
3.4975e-01, 2.0092e+00],
[ 4.2022e-02, -3.4008e-01, 5.5509e-01, ..., -3.7005e+00,
-1.5166e+00, 1.7479e+00],
[-2.2005e-01, -2.2581e-01, 1.9588e-01, ..., -1.0872e+01,
-1.2050e+01, -6.2351e+00],
...,
[-3.4155e-01, 4.9512e-01, -5.7838e+00, ..., -1.1535e+00,
1.4547e+00, -1.3779e+00],
[-2.5445e-01, 4.2682e-01, -1.7267e+00, ..., -2.1835e-02,
1.5048e-01, 5.0395e-02],
[-4.8547e-02, -6.0864e-02, 4.8580e-02, ..., -4.4157e-01,
-8.3238e-02, -3.9846e-01]],
...,
[[ 6.3542e-01, -1.2200e-01, 1.5107e-01, ..., 6.1950e-01,
7.7554e-01, -1.0250e+00],
[-5.0976e-01, -4.4260e-01, -4.3384e-01, ..., -4.1534e-01,
-1.2141e+00, -3.3004e+00],
[-3.1695e-01, -1.3293e-01, -1.4396e-01, ..., -3.8386e+00,
-5.2756e+00, -8.8142e+00],
...,
[-4.3499e-01, -3.3955e-01, -5.1452e-01, ..., -2.6665e+00,
-5.4010e+00, -3.3655e+00],
[-2.3234e+00, -1.8450e+00, -6.9417e-01, ..., -1.7967e+00,
-1.8833e+00, -1.8200e+00],
[-5.0073e-01, -1.0443e+00, -5.3993e-01, ..., -8.3136e-01,
-9.0320e-01, -9.0770e-01]],
[[ 1.8672e-01, 2.7728e-01, 5.8684e-01, ..., -3.1880e-01,
6.2955e-01, -1.9244e+00],
[ 1.5558e-01, 2.2379e-01, 4.2697e-01, ..., -2.6538e+00,
-2.1816e-01, -1.8392e+00],
[-1.3144e-01, 3.5047e-02, 3.2007e-01, ..., -5.6867e+00,
-2.7295e+00, 2.3380e+00],
...,
[-1.5140e-01, 4.2609e-01, -9.0606e-01, ..., -8.6511e-01,
-1.1473e+00, -6.4127e-01],
[-5.4643e-01, -3.9552e-01, -1.2810e-01, ..., 6.8868e-02,
3.6567e-02, -4.8626e-01],
[ 5.5987e-01, 5.6383e-01, 1.1899e-01, ..., 3.3028e-01,
2.4689e-01, 3.3954e-01]],
[[-8.1423e-01, 2.3571e-01, 2.9253e-01, ..., 6.0456e-01,
6.8612e-01, 1.3840e+00],
[-1.1277e+00, 1.3660e-01, -3.6968e-01, ..., -2.5529e+00,
-3.6332e+00, -3.6571e-01],
[-9.1678e-01, 2.7183e-01, -8.3637e-02, ..., -5.7881e+00,
-8.0340e+00, -4.3124e+00],
...,
[-9.0843e-01, -8.5364e-02, 1.0547e+00, ..., 1.1224e+00,
1.3404e+00, 3.7964e+00],
[-1.6123e+00, -7.2293e-01, 4.4771e-01, ..., -1.3315e-01,
1.4706e-01, 9.8156e-01],
[-7.2244e-01, -3.0128e-01, -8.2056e-02, ..., 3.4499e-01,
3.0199e-01, 1.2356e+00]]],
[[[-6.2812e-01, -8.4976e-01, -2.9937e-01, ..., -5.6260e-01,
-1.2159e-03, -7.3623e-01],
[-7.3005e-01, -1.7766e+00, -7.6181e-01, ..., -2.0008e+00,
-1.6809e+00, -1.4369e+00],
[-4.7809e-01, -2.1265e+00, -8.1367e-01, ..., -8.3370e-01,
-8.7188e-01, -1.7738e+00],
...,
[ 1.0204e-01, -4.9723e-01, -1.8571e-01, ..., -1.0820e-01,
-6.1602e-02, -4.5157e-01],
[-3.9690e-01, -4.9643e-01, 4.8706e-03, ..., 1.5013e-01,
2.8316e-01, -1.6380e-01],
[-6.3036e-01, -1.3012e+00, -1.0689e+00, ..., -1.0788e+00,
-1.1769e+00, -5.0512e-01]],
[[-7.7458e-01, -2.0509e+00, -1.5101e+00, ..., -3.9623e+00,
-4.9343e+00, -1.9596e+00],
[-2.4983e+00, -4.7068e+00, -4.8080e+00, ..., -9.1440e+00,
-9.3600e+00, -3.9729e+00],
[-3.7614e+00, -5.8787e+00, -5.3328e+00, ..., -9.5532e+00,
-8.5108e+00, -4.0420e+00],
...,
[-1.7258e+00, -2.0252e+00, -1.8809e+00, ..., -1.7884e+00,
-2.2270e+00, -9.1558e-01],
[-2.7986e+00, -2.9619e+00, -1.8472e+00, ..., -2.1766e+00,
-3.3115e+00, -1.0112e+00],
[ 3.1643e-01, -3.0348e-01, 2.5543e-01, ..., 3.1040e-01,
4.6388e-01, 1.9757e-01]],
[[-5.0885e-01, -4.0858e-01, -4.5128e-01, ..., 8.1592e-01,
5.3805e-01, 1.4537e+00],
[-8.1954e-01, -9.1890e-01, -1.6614e+00, ..., 6.4710e-01,
6.8372e-01, 7.5569e-01],
[-1.8110e+00, -1.6086e-01, -3.7570e-01, ..., 2.4568e-01,
6.3220e-01, 1.0899e+00],
...,
[-5.8551e-01, -1.9981e-01, -7.9146e-01, ..., 1.7347e-01,
1.8942e-01, 3.9617e-01],
[-4.0321e-01, -2.9725e-02, 2.4155e-01, ..., 1.7977e-01,
1.5484e-01, 5.1981e-01],
[-1.5138e-01, -2.3883e-01, 2.0777e-01, ..., -7.0464e-03,
7.5497e-02, 3.3254e-01]],
...,
[[ 1.3175e+00, 6.8781e-01, -2.3241e-01, ..., 4.7341e-01,
2.3376e-01, -1.0291e+00],
[-1.4453e+00, -1.9767e+00, -7.7533e-01, ..., -8.5973e-02,
-1.1309e+00, -1.3908e+00],
[-3.2185e-01, -7.2754e-01, 5.7751e-01, ..., 1.5343e-01,
-5.4723e-01, -1.4384e+00],
...,
[-6.5816e-01, -5.6332e-01, -4.9851e-01, ..., -3.3701e-01,
-5.1627e-01, -5.4218e-01],
[-2.2280e+00, -1.5420e+00, -1.1541e+00, ..., -1.3180e+00,
-1.3789e+00, -1.0803e+00],
[-5.8959e-01, -7.5418e-01, -4.1526e-01, ..., -2.2991e-01,
-3.3028e-01, -3.7316e-01]],
[[-1.2452e-01, -1.9279e-01, 4.7352e-01, ..., 2.3614e-01,
7.5277e-01, -2.4235e-01],
[-2.4276e-01, -1.5812e+00, -7.8066e-01, ..., -4.7437e-01,
-2.5155e-01, -1.0971e+00],
[-4.4612e-01, -1.8457e+00, -7.7412e-01, ..., 5.6686e-01,
3.5255e-01, -3.9879e-01],
...,
[-1.7838e-01, -6.3364e-01, -2.9969e-01, ..., 3.4579e-01,
3.3582e-01, 7.3105e-02],
[-5.1335e-01, -8.9822e-01, -4.8505e-02, ..., 1.1929e-02,
-1.5999e-02, -1.4970e-01],
[ 3.4924e-01, -9.7533e-04, -2.0911e-01, ..., -1.4181e-01,
-1.0643e-01, 1.7988e-01]],
[[-8.2755e-01, -3.9044e-01, 4.0192e-01, ..., 3.2809e-01,
4.0972e-01, 9.9643e-01],
[-6.3755e-01, -7.3504e-01, -9.7224e-01, ..., -1.3033e+00,
-7.0049e-01, 7.9851e-01],
[-1.1502e-01, -1.0504e+00, -2.0088e+00, ..., -1.6679e+00,
-9.4478e-01, 7.0650e-01],
...,
[-6.6099e-01, -4.0334e-01, -3.2986e-01, ..., -1.8215e-01,
-7.1849e-04, 6.1267e-01],
[-1.1307e+00, -5.5045e-01, -3.9014e-01, ..., -3.4390e-01,
-1.7805e-01, 1.5826e-01],
[-3.5568e-01, -6.8702e-03, -1.4810e-01, ..., -2.3458e-01,
-2.1196e-01, 3.2501e-01]]]], device='cuda:0'),
tensor([[[[-3.3983e-01, -1.0557e+00, -8.1092e-01, ..., -1.2660e-02,
5.3800e-01, 4.6693e-01],
[ 1.2570e-01, -1.8751e+00, -1.2686e+00, ..., -4.4295e-01,
-2.0578e-01, 1.2499e-01],
[ 1.7818e-01, -1.5374e+00, -1.0511e+00, ..., -3.1567e-01,
-1.4535e-01, -1.2625e-01],
...,
[ 1.0375e-01, -1.2877e+00, -9.8199e-01, ..., -3.4519e-01,
-1.5221e-01, -1.8254e-01],
[ 1.0434e-01, -8.7895e-01, -4.5922e-01, ..., 6.9959e-02,
3.4809e-01, 2.9545e-02],
[-1.5345e-01, -1.0968e+00, -9.4803e-01, ..., -1.2089e+00,
-1.2467e+00, -4.7691e-01]],
[[-2.0819e+00, -4.0041e+00, -3.6688e+00, ..., -6.5450e-01,
-1.1550e+00, -7.6773e-02],
[-3.8809e+00, -6.2310e+00, -5.4679e+00, ..., -1.8368e+00,
-2.6627e+00, -6.7330e-01],
[-3.7411e+00, -5.1643e+00, -4.8924e+00, ..., -1.4924e+00,
-1.6436e+00, -4.1321e-01],
...,
[-3.1510e+00, -4.5851e+00, -4.6331e+00, ..., -1.3890e+00,
-1.5298e+00, -3.2751e-01],
[-5.0472e+00, -6.0674e+00, -5.7049e+00, ..., -2.8267e+00,
-3.6613e+00, -9.5841e-01],
[ 7.1318e-01, -1.2882e-01, 4.9312e-01, ..., 3.5151e-01,
3.6571e-01, 4.7158e-02]],
[[-5.7214e-01, 4.3688e-01, 8.3659e-01, ..., 8.1657e-01,
3.0915e-01, 1.0161e+00],
[-9.4026e-01, 1.1849e-01, 6.4057e-01, ..., 3.4794e-01,
1.5904e-01, 4.6770e-01],
[-1.4082e+00, -5.3014e-02, 7.8008e-02, ..., 3.1784e-02,
1.2319e-01, 4.3256e-01],
...,
[-1.3024e+00, 7.1816e-02, -5.5022e-02, ..., 1.2712e-01,
-4.7660e-01, 1.7745e-01],
[-1.3065e+00, 2.3129e-01, -6.5399e-02, ..., 1.0128e-01,
1.1003e-02, 5.6911e-01],
[-5.7366e-01, 2.9297e-01, 1.4959e-01, ..., -1.0459e-01,
6.8541e-02, 3.8896e-01]],
...,
[[ 6.6948e-01, 1.0064e-01, 2.5610e-01, ..., -8.1514e-02,
-3.0871e-01, -8.6764e-01],
[-7.8853e-01, -3.4472e-01, -9.8073e-02, ..., -5.9023e-01,
-1.1750e+00, -8.0054e-01],
[-5.1059e-01, -3.6651e-01, -5.5524e-02, ..., -1.2071e-01,
-2.0132e-01, -4.2121e-01],
...,
[-6.5210e-01, -5.3674e-01, -1.0410e-01, ..., -1.5719e-01,
-3.1058e-01, -6.8772e-01],
[-2.1409e+00, -1.7052e+00, -1.2357e+00, ..., -9.5743e-01,
-1.0727e+00, -9.9759e-01],
[-4.2844e-01, -9.7984e-01, -3.7964e-01, ..., -2.9319e-02,
-2.1354e-01, -4.0406e-01]],
[[-1.0319e-02, -2.5717e-01, -3.0554e-02, ..., 3.1110e-01,
5.8751e-01, 2.5998e-01],
[ 1.1529e-01, -9.1149e-01, -2.6628e-02, ..., 5.8582e-02,
9.1414e-02, -3.1239e-02],
[-1.7693e-01, -1.0753e+00, -1.4621e-01, ..., -1.4526e-01,
-9.6632e-02, 1.1063e-01],
...,
[-4.5393e-02, -1.0289e+00, -2.1402e-03, ..., -1.8826e-01,
-2.4243e-01, 1.4202e-02],
[-5.1217e-01, -1.9177e+00, -2.6877e-01, ..., -5.0473e-01,
-4.1083e-01, -1.3674e-01],
[ 7.1140e-01, -1.4631e-01, 2.4686e-01, ..., -3.3851e-01,
-2.9628e-01, 2.9144e-01]],
[[-6.2498e-01, -9.8151e-02, 4.3069e-02, ..., 4.8243e-01,
3.4636e-01, 5.2407e-01],
[-1.0492e+00, -7.9788e-01, -9.2087e-01, ..., -5.3963e-02,
-1.3107e-01, 4.0477e-01],
[-5.9985e-01, -7.2659e-01, -4.7164e-01, ..., 8.0039e-02,
-1.4643e-01, 3.7860e-02],
...,
[-5.9219e-01, -8.5296e-01, -5.7226e-01, ..., 6.3464e-02,
6.1512e-02, 2.2124e-01],
[-7.9630e-01, -1.0300e+00, -6.5217e-01, ..., -1.2368e-01,
-2.9401e-01, -1.8833e-01],
[-4.6681e-01, -8.6712e-01, -3.9480e-01, ..., -1.9503e-01,
-3.7227e-01, 9.8852e-02]]],
[[[-7.4972e-01, -1.5311e+00, -1.4829e+00, ..., -1.2844e+00,
-8.8249e-01, -1.6904e+00],
[-3.7876e-01, -2.4674e+00, -2.3121e+00, ..., -1.7883e+00,
-1.9216e+00, -1.9390e+00],
[-1.5925e-01, -1.9178e+00, -1.7367e+00, ..., -1.4727e+00,
-1.4689e+00, -2.1566e+00],
...,
[ 1.0972e-01, -5.4388e-01, -2.0784e-01, ..., -4.9400e-01,
-5.6505e-01, -5.4909e-01],
[-3.2929e-01, -4.3174e-01, 3.3158e-02, ..., 1.1091e-01,
2.7514e-01, -1.8323e-01],
[-6.1648e-01, -1.2648e+00, -1.0908e+00, ..., -1.0857e+00,
-1.1824e+00, -5.0959e-01]],
[[-2.7266e+00, -5.1812e+00, -4.6502e+00, ..., -4.8386e+00,
-6.0805e+00, -2.4429e+00],
[-5.3692e+00, -8.2115e+00, -7.5667e+00, ..., -7.8215e+00,
-9.3739e+00, -4.4534e+00],
[-4.1091e+00, -5.9325e+00, -5.8823e+00, ..., -6.5946e+00,
-6.6006e+00, -3.2686e+00],
...,
[-1.8059e+00, -2.2378e+00, -1.9474e+00, ..., -3.2522e+00,
-2.8881e+00, -1.0615e+00],
[-2.8396e+00, -2.9547e+00, -2.0211e+00, ..., -2.3749e+00,
-3.4783e+00, -1.0579e+00],
[ 3.4503e-01, -2.1666e-01, 3.3453e-01, ..., 3.0316e-01,
4.5511e-01, 1.9916e-01]],
[[-6.5309e-01, 5.3824e-01, 8.7248e-01, ..., 8.6845e-01,
6.6237e-01, 1.9461e+00],
[-9.6949e-01, 1.0453e-01, 3.9415e-01, ..., 5.1618e-01,
-4.3525e-03, 1.0350e-01],
[-1.8106e+00, -2.0959e-01, -4.9393e-02, ..., 4.1360e-02,
-2.3441e-02, -1.1318e-01],
...,
[-6.6556e-01, 1.3008e-01, 3.0364e-01, ..., 7.0987e-02,
3.7590e-01, 4.4202e-01],
[-4.5977e-01, 9.5483e-02, 2.2533e-01, ..., 1.9784e-01,
2.1560e-01, 5.8228e-01],
[-1.4003e-01, -2.4503e-01, 1.4677e-01, ..., -4.8110e-03,
1.0390e-01, 3.4928e-01]],
...,
[[ 1.3212e+00, -3.2607e-01, -7.7101e-02, ..., 2.5809e-01,
-8.2789e-02, -2.1494e+00],
[-7.0516e-01, -9.3513e-01, -5.3098e-01, ..., -5.6850e-01,
-1.5154e+00, -2.0966e+00],
[-7.0886e-01, -8.8956e-01, -2.9807e-01, ..., -3.1209e-01,
-6.0255e-01, -1.3995e+00],
...,
[-6.3283e-01, -5.2027e-01, -3.1015e-01, ..., -1.6617e+00,
-7.0035e-01, -5.2466e-01],
[-2.2146e+00, -1.7720e+00, -1.2717e+00, ..., -1.3078e+00,
-1.3735e+00, -1.1071e+00],
[-6.0251e-01, -8.4300e-01, -4.5950e-01, ..., -2.7864e-01,
-3.8095e-01, -4.1289e-01]],
[[ 8.0924e-02, 2.9614e-02, 4.5774e-01, ..., 5.3808e-01,
9.7699e-01, 5.6515e-03],
[ 3.0914e-01, -7.9563e-01, -6.7172e-02, ..., 4.8852e-01,
4.1253e-01, -8.5150e-01],
[ 2.0146e-01, -5.8468e-01, 2.9664e-01, ..., 2.0319e-01,
5.5246e-01, -3.9170e-01],
...,
[-1.7793e-01, -4.9539e-01, 2.3332e-01, ..., -2.4724e-01,
-2.2296e-01, -5.5519e-02],
[-5.0065e-01, -9.4091e-01, 4.7814e-02, ..., 1.5363e-02,
-4.6371e-02, -1.9529e-01],
[ 3.4980e-01, -1.1334e-02, -1.7441e-01, ..., -1.1274e-01,
-1.0023e-01, 1.5135e-01]],
[[-1.5677e+00, -7.9475e-01, -4.2500e-01, ..., -4.6186e-01,
-2.9846e-01, 8.6645e-01],
[-2.2022e+00, -8.1644e-01, -5.5814e-01, ..., -7.9934e-01,
-2.0991e-01, 1.6681e+00],
[-1.5427e+00, -5.9293e-01, -2.2748e-01, ..., -3.0630e-01,
-1.2884e-01, 1.4085e+00],
...,
[-6.4729e-01, -2.7812e-01, -1.7576e-01, ..., 1.1025e+00,
2.8313e-01, 6.5044e-01],
[-1.1519e+00, -5.9299e-01, -3.3099e-01, ..., -3.2022e-01,
-1.6216e-01, 1.8529e-01],
[-3.7888e-01, -3.7364e-02, -1.5616e-01, ..., -2.3327e-01,
-2.2973e-01, 3.4226e-01]]],
[[[-7.8511e-01, -1.4869e+00, -1.4630e+00, ..., -2.0920e-01,
4.1513e-01, 3.0263e-01],
[-3.6245e-01, -2.4563e+00, -2.4655e+00, ..., -5.4164e-01,
-3.6118e-01, -7.1255e-02],
[-4.4799e-01, -1.7959e+00, -2.8007e+00, ..., -3.1298e-01,
-5.2529e-01, -3.3014e-01],
...,
[-6.5254e-01, -4.5640e+00, -2.6319e+00, ..., -6.1292e-01,
-4.5107e-01, 8.8320e-03],
[-8.7535e-01, -3.0893e+00, -2.2692e+00, ..., -1.4485e-02,
1.0229e-01, -2.0043e-01],
[-8.2387e-02, -1.1996e+00, -1.1835e+00, ..., -1.1926e+00,
-1.3405e+00, -5.0590e-01]],
[[-2.4628e+00, -5.0103e+00, -5.0155e+00, ..., -9.9987e-01,
-1.4796e+00, -3.3938e-01],
[-5.6752e+00, -9.0948e+00, -8.7661e+00, ..., -2.4822e+00,
-3.0737e+00, -9.0818e-01],
[-4.5860e+00, -8.4768e+00, -8.8969e+00, ..., -2.2709e+00,
-2.2795e+00, -7.7080e-01],
...,
[-7.2246e+00, -1.3931e+01, -1.5891e+01, ..., -2.0925e+00,
-2.3033e+00, -1.7949e-01],
[-7.6267e+00, -1.0473e+01, -1.0216e+01, ..., -3.2959e+00,
-4.2887e+00, -1.2476e+00],
[-5.4165e-01, -7.6871e-01, 2.3480e-01, ..., 4.5581e-01,
4.1580e-01, 5.2875e-02]],
[[-6.3785e-01, 5.2512e-01, 1.1314e+00, ..., 9.2790e-01,
4.2657e-01, 1.1150e+00],
[-8.5575e-01, -4.9122e-02, 1.5901e-01, ..., 5.8491e-01,
9.3195e-02, 3.6218e-01],
[-1.6542e+00, -1.3472e-01, -1.7946e-01, ..., 3.5823e-01,
-1.3500e-01, 3.0277e-01],
...,
[-6.3043e+00, -1.4599e+01, -9.2500e-01, ..., -1.1737e+00,
-1.4547e+00, -5.8419e-02],
[-4.6144e+00, -6.5442e+00, 4.2320e+00, ..., -6.3579e-02,
-6.8226e-02, 8.4176e-01],
[-1.1910e+00, -1.6303e+00, 2.4888e+00, ..., -1.4930e-01,
-1.4577e-02, 3.9094e-01]],
...,
[[ 1.1360e+00, -4.7875e-01, -2.1735e-01, ..., -4.4854e-02,
-2.3771e-01, -8.2755e-01],
[-7.9571e-01, -1.0174e+00, -4.5031e-01, ..., -5.8157e-01,
-1.1979e+00, -8.6477e-01],
[-6.2488e-01, -8.2708e-01, -3.5279e-01, ..., -2.4897e-01,
-1.8223e-01, -5.4726e-01],
...,
[-1.0862e+00, -1.7733e+00, -1.6720e+00, ..., 3.9467e-02,
-4.5154e-01, -1.6356e+00],
[-1.5267e+00, -3.0130e+00, -7.6748e+00, ..., -9.8506e-01,
-1.2632e+00, -1.4068e+00],
[-7.7897e-01, -2.4958e+00, -4.5122e+00, ..., -1.0967e-01,
-2.3736e-01, -5.1935e-01]],
[[-1.7211e-02, -6.0600e-02, 7.7112e-01, ..., 2.6094e-01,
6.8084e-01, 1.9140e-01],
[-2.8575e-02, -6.7538e-01, 2.1975e-01, ..., -1.1111e-01,
2.0285e-01, -8.1122e-02],
[-6.5897e-02, -3.8136e-01, 5.2173e-01, ..., -1.7099e-01,
-1.0356e-01, -1.2468e-02],
...,
[ 4.2641e-01, -6.1813e+00, -3.9133e+00, ..., -3.1611e-01,
-2.8074e-01, 2.0183e-02],
[-1.0437e+00, -4.0914e+00, -1.6145e+00, ..., -4.8363e-01,
-5.3621e-01, -1.9441e-01],
[ 9.6759e-01, 1.8484e-01, -5.3770e-02, ..., -2.2599e-01,
-3.2220e-01, 2.1331e-01]],
[[-1.8000e+00, -7.6582e-01, -6.7444e-02, ..., 5.6460e-01,
3.0890e-01, 5.5243e-01],
[-2.3272e+00, -3.8329e-01, -4.1890e-01, ..., 2.6623e-02,
-2.0150e-01, 3.6800e-01],
[-1.9673e+00, -5.5780e-01, -7.2510e-01, ..., 1.1848e-01,
-1.3468e-01, -1.6842e-01],
...,
[-1.4395e+00, 1.0949e+00, -8.2999e+00, ..., 2.1604e-01,
-4.9679e-01, -4.8507e-01],
[-3.2503e+00, -1.9734e+00, -7.6570e+00, ..., 1.6460e-03,
-1.0427e-01, -1.3179e-01],
[-1.3832e+00, 2.5581e-01, -3.8938e-01, ..., -2.5852e-01,
-3.0285e-01, 2.1360e-01]]],
[[[-3.7903e-01, -1.0308e+00, -9.3947e-01, ..., 1.4706e-01,
6.1773e-01, 5.3186e-01],
[ 9.1278e-02, -1.7499e+00, -1.3042e+00, ..., 2.8243e-01,
-3.2645e-01, 1.1348e-01],
[ 1.5281e-01, -1.2960e+00, -9.6771e-01, ..., 1.3002e-01,
-5.1599e-01, 5.4543e-02],
...,
[ 3.0371e-01, -1.1314e+00, -2.0582e+00, ..., -3.7110e-01,
-1.1445e-01, -1.4445e-01],
[-8.6732e-02, -1.5567e+00, -1.2356e+00, ..., -2.3757e-02,
2.6126e-01, -3.4115e-02],
[-4.5024e-01, -1.3110e+00, -1.1612e+00, ..., -1.0574e+00,
-1.0395e+00, -3.7968e-01]],
[[-2.2152e+00, -4.0611e+00, -3.8176e+00, ..., -1.4558e+00,
-1.0916e+00, -4.8853e-02],
[-3.7895e+00, -5.9945e+00, -5.4523e+00, ..., -3.6882e+00,
-3.0503e+00, -7.3540e-01],
[-3.2212e+00, -4.7502e+00, -4.6356e+00, ..., -7.1394e+00,
-3.8548e+00, -1.5098e+00],
...,
[-3.2776e+00, -6.4127e+00, -5.9164e+00, ..., -1.4015e+00,
-1.6109e+00, -4.0569e-01],
[-5.6062e+00, -7.6119e+00, -6.2492e+00, ..., -2.6447e+00,
-3.3919e+00, -1.0041e+00],
[ 6.1927e-01, -3.9480e-01, 4.4810e-01, ..., 3.2402e-01,
3.4600e-01, 7.1691e-02]],
[[-5.7105e-01, 4.4131e-01, 7.2620e-01, ..., 1.4022e+00,
3.4939e-01, 1.1503e+00],
[-9.0807e-01, 1.8866e-01, 6.1044e-01, ..., 1.3101e+00,
1.3064e-01, 4.8673e-01],
[-1.3013e+00, 4.4133e-02, 3.8488e-02, ..., -1.6982e-01,
9.1511e-01, 9.8341e-01],
...,
[-1.4025e+00, -2.1523e+00, 9.6447e-01, ..., 5.5556e-02,
8.9266e-02, 4.3952e-01],
[-1.8741e+00, -2.8044e-01, 3.0969e-01, ..., 1.4362e-01,
1.3214e-01, 6.3101e-01],
[-6.2165e-01, 2.9220e-01, 1.2822e-01, ..., -8.5224e-02,
8.8351e-02, 3.6339e-01]],
...,
[[ 7.5644e-01, 1.1065e-01, 3.3161e-01, ..., 2.6921e-01,
-3.4141e-01, -9.4489e-01],
[-6.4325e-01, -2.8911e-01, -1.9242e-04, ..., -2.1058e+00,
-1.3470e+00, -8.7270e-01],
[-4.9695e-01, -3.8968e-01, 5.1167e-03, ..., -2.6320e+00,
-2.6180e-01, -2.7592e-01],
...,
[-2.9387e+00, -4.3547e+00, -1.9095e+00, ..., -1.3563e-01,
-2.0352e-01, -3.7026e-01],
[-3.5524e+00, -3.0791e+00, -1.6644e+00, ..., -8.1183e-01,
-8.5345e-01, -7.9014e-01],
[-4.9683e-01, -1.2045e+00, -3.4403e-01, ..., -3.5815e-02,
-1.8424e-01, -3.3664e-01]],
[[-1.7341e-02, -2.8097e-01, -9.8567e-02, ..., 1.0209e-01,
7.2418e-01, 3.2242e-01],
[ 1.6353e-01, -9.3196e-01, -5.6236e-02, ..., -2.9707e-01,
-1.3819e-01, -8.3992e-02],
[-1.4560e-01, -9.5109e-01, 7.3058e-02, ..., -6.0969e-01,
-1.4264e+00, -2.5478e-01],
...,
[ 8.0966e-01, -3.9200e-01, -4.8343e-01, ..., -1.3357e-01,
-1.2795e-01, 1.4104e-01],
[-8.7161e-01, -2.5207e+00, -1.0323e+00, ..., -4.7979e-01,
-3.7280e-01, -4.7602e-02],
[ 7.1623e-01, -1.5307e-01, 2.3491e-01, ..., -2.7376e-01,
-2.4324e-01, 3.2252e-01]],
[[-5.9252e-01, -1.6661e-01, -5.3106e-02, ..., 1.0481e+00,
3.5993e-01, 5.9213e-01],
[-1.1107e+00, -9.2890e-01, -9.7690e-01, ..., 5.1897e-01,
-1.9092e-01, 4.5064e-01],
[-6.8972e-01, -8.0352e-01, -5.6053e-01, ..., -6.4356e-02,
1.5067e-01, 6.2444e-01],
...,
[-2.1053e+00, 1.2059e+00, 5.8712e-01, ..., 9.0415e-02,
-9.5239e-02, 1.1859e-01],
[-7.4126e-01, -6.8144e-01, -6.6517e-01, ..., -5.2324e-02,
-3.0715e-01, -1.8181e-01],
[-5.9613e-01, -1.0274e+00, -4.0339e-01, ..., -8.8115e-02,
-2.9322e-01, 1.6069e-01]]],
[[[-1.9893e-01, -3.1478e-01, 4.8962e-02, ..., -2.6042e-01,
5.0773e-01, 1.8671e-01],
[-8.5260e-02, -1.4704e+00, -8.5996e-01, ..., -9.7255e-01,
-5.7416e-01, -3.2821e-01],
[-1.9683e-02, -1.2176e+00, -3.3920e-01, ..., -7.6908e-01,
-5.1056e-01, -7.7695e-01],
...,
[-1.7639e-01, -2.1015e+00, -1.6469e+00, ..., -2.1322e-01,
-4.7433e-01, -2.0085e+00],
[-4.4846e-01, -1.5255e+00, -4.8932e-01, ..., 2.9702e-01,
5.9039e-01, -1.4804e+00],
[-2.5411e-01, -1.3250e+00, -1.1874e+00, ..., -8.5106e-01,
-1.0821e+00, -7.9762e-01]],
[[-2.9328e-01, -1.5734e+00, -3.0946e+00, ..., -2.3163e+00,
-1.5504e+00, -2.9815e-01],
[-2.0341e+00, -3.9862e+00, -4.1383e+00, ..., -4.6059e+00,
-3.7458e+00, -1.3123e+00],
[-1.9924e+00, -3.4958e+00, -3.8995e+00, ..., -4.0980e+00,
-3.2144e+00, -1.3453e+00],
...,
[-3.7444e+00, -5.4947e+00, -5.4652e+00, ..., -8.3527e+00,
-7.8397e+00, -2.9383e+00],
[-5.5533e+00, -7.2366e+00, -6.2863e+00, ..., -8.4860e+00,
-9.1760e+00, -2.8021e+00],
[ 5.2470e-01, -5.2340e-01, 4.8373e-01, ..., -5.4041e-01,
-3.8058e-01, 1.1007e-01]],
[[-4.5788e-01, -8.1872e-02, 2.0944e-01, ..., 2.3710e-01,
2.4873e-02, 9.8753e-01],
[-3.4488e-01, -3.8918e-01, 7.7430e-01, ..., 8.2826e-01,
2.2431e-01, 4.9464e-01],
[-7.7857e-01, -2.2159e-01, 8.4620e-01, ..., 9.3270e-01,
1.6470e-01, 3.3127e-01],
...,
[-1.7342e+00, 2.1099e-01, -8.3529e-01, ..., -5.6951e-01,
7.1869e-01, -1.6129e-01],
[-1.4360e+00, 4.0768e-01, 4.7790e-01, ..., -1.6119e+00,
6.0759e-01, -2.9382e-01],
[-6.1990e-01, -1.6236e-01, 2.0334e-02, ..., -3.0691e+00,
5.2484e-01, -4.6778e-01]],
...,
[[ 1.8594e-01, -4.8902e-01, 4.7187e-01, ..., -1.1508e+00,
-3.2569e-01, -7.9419e-01],
[-6.9809e-01, -5.7361e-01, -4.8952e-01, ..., -9.9011e-01,
-1.1193e+00, -8.5922e-01],
[-3.6257e-01, -1.4606e-01, 2.1440e-01, ..., -2.9909e-01,
-1.8493e-01, -5.8639e-01],
...,
[-9.9665e-01, -6.6110e-01, 1.6146e-01, ..., -2.5489e+00,
-1.4408e+00, -1.2546e+00],
[-2.5831e+00, -2.5802e+00, -2.5644e+00, ..., -4.9139e+00,
-2.9801e+00, -1.6315e+00],
[-5.7640e-01, -1.1684e+00, -2.1413e-01, ..., -4.9241e+00,
-2.3706e+00, -5.1085e-01]],
[[-1.2500e-01, 5.4731e-02, 6.6786e-01, ..., 3.2713e-01,
6.5879e-01, 1.6042e-01],
[-2.9502e-01, -2.0545e-01, -3.3604e-01, ..., 1.1167e-01,
1.4047e-01, -1.4255e-01],
[-5.1950e-01, -4.3249e-01, -3.6980e-01, ..., -2.9855e-01,
1.5029e-02, -7.0102e-02],
...,
[-1.3831e-01, -8.1837e-01, -1.7805e-01, ..., 2.9467e-01,
4.0296e-01, -3.2303e-01],
[-7.5049e-01, -1.6482e+00, 7.6114e-02, ..., 3.8429e-01,
4.6727e-01, -5.4403e-01],
[ 7.4430e-01, 6.6292e-02, 3.6144e-01, ..., 1.7305e+00,
1.2399e+00, 3.1747e-01]],
[[-5.9621e-01, 4.2876e-01, 1.1067e-01, ..., 8.4457e-01,
4.0961e-01, 6.8002e-01],
[-8.1089e-01, 2.4855e-01, -1.0425e+00, ..., 2.8554e-01,
-1.3178e-01, 7.2670e-01],
[-4.4091e-01, 3.4899e-01, -2.1384e-01, ..., 5.4591e-01,
6.1001e-02, 5.8413e-01],
...,
[-1.3210e+00, -1.6985e+00, -8.1989e-01, ..., -1.0284e-01,
3.2466e-01, 1.4143e+00],
[-1.4788e+00, -1.3608e+00, -2.7861e-01, ..., -5.3834e-01,
5.1394e-01, 1.2206e+00],
[-7.2640e-01, -4.8335e-01, 1.7447e-01, ..., 7.1562e-01,
8.5702e-01, 1.3088e+00]]]], device='cuda:0'),
tensor([[[[-6.8355e-01, -1.4295e+00, -1.4030e+00, ..., -2.3283e-01,
2.5451e-01, -3.8007e-01],
[-2.8552e-01, -2.3715e+00, -2.1948e+00, ..., -6.1384e-01,
-4.1879e-01, -5.5752e-01],
[-3.6582e-01, -1.9528e+00, -2.1781e+00, ..., -3.8555e-01,
-3.8840e-01, -1.0509e+00],
...,
[-5.0790e-01, -1.9099e+00, -1.8763e+00, ..., 1.4643e-02,
8.3560e-02, -1.3952e+00],
[-9.4191e-01, -1.6586e+00, -1.1833e+00, ..., 1.3794e-02,
1.0814e-01, -9.5939e-01],
[-7.9092e-01, -1.6068e+00, -1.2337e+00, ..., -9.2454e-01,
-1.1014e+00, -6.4755e-01]],
[[-2.3581e+00, -4.9055e+00, -4.3217e+00, ..., -2.2039e+00,
-3.4898e+00, -1.2734e+00],
[-5.3396e+00, -8.3807e+00, -7.3132e+00, ..., -3.4951e+00,
-5.1738e+00, -2.1137e+00],
[-4.5031e+00, -7.5582e+00, -7.9019e+00, ..., -3.0700e+00,
-3.8633e+00, -1.7191e+00],
...,
[-4.3721e+00, -6.1558e+00, -5.5770e+00, ..., -3.0985e+00,
-4.0730e+00, -2.1572e+00],
[-6.4169e+00, -7.4048e+00, -6.3179e+00, ..., -3.9520e+00,
-5.3754e+00, -1.9397e+00],
[-2.6412e-02, -8.0181e-01, 3.9056e-01, ..., 4.3593e-01,
6.1377e-01, 1.7959e-01]],
[[-5.9848e-01, 5.8458e-01, 1.1212e+00, ..., 8.3810e-01,
5.6635e-01, 1.3626e+00],
[-8.7139e-01, -1.3969e-01, 3.4600e-01, ..., 4.3877e-01,
3.6482e-01, 2.1179e-01],
[-1.6075e+00, -1.6988e-01, -2.7512e-01, ..., 7.8735e-02,
3.0022e-01, 1.0952e-01],
...,
[-1.7170e+00, -1.1912e-01, -1.3800e-01, ..., 3.6386e-01,
2.9303e-02, -3.5238e-01],
[-1.5236e+00, 5.5592e-02, 3.8511e-03, ..., 1.0627e-01,
2.5058e-01, 1.8151e-01],
[-1.4320e+00, -4.9676e-01, -2.4658e-01, ..., 6.3114e-03,
1.9498e-01, -1.2202e-01]],
...,
[[ 1.0579e+00, -4.0594e-01, -1.7897e-01, ..., 3.7355e-01,
2.2209e-01, -7.9648e-01],
[-8.4921e-01, -9.6312e-01, -7.5370e-01, ..., -3.6169e-01,
-1.0931e+00, -9.6605e-01],
[-6.6235e-01, -7.7862e-01, -1.6478e-01, ..., -1.4649e-01,
-2.3614e-01, -6.0961e-01],
...,
[-7.2369e-01, -1.1564e+00, -7.6065e-01, ..., -6.7558e-01,
-8.3995e-01, -9.0626e-01],
[-2.7180e+00, -2.8337e+00, -1.8426e+00, ..., -1.0951e+00,
-1.1088e+00, -1.0977e+00],
[-6.6451e-01, -1.6875e+00, -1.0202e+00, ..., -1.6581e-01,
-2.6782e-01, -4.8763e-01]],
[[ 3.1999e-02, 8.9288e-03, 5.8032e-01, ..., 3.2135e-01,
8.4683e-01, 7.6972e-02],
[-2.2272e-02, -6.7566e-01, 1.8935e-01, ..., 3.9680e-01,
6.4627e-01, -3.0587e-01],
[-3.3589e-02, -5.1350e-01, 5.2001e-01, ..., 3.7600e-01,
4.1070e-01, -1.2569e-01],
...,
[ 4.3718e-02, -7.7092e-01, 2.9385e-01, ..., 6.1235e-01,
5.6269e-01, -5.9604e-01],
[-2.3578e-01, -1.4227e+00, 1.8337e-02, ..., 2.4275e-02,
1.1320e-01, -4.4321e-01],
[ 6.5482e-01, 1.5132e-01, 2.9946e-01, ..., 2.2792e-01,
3.1919e-01, 3.7384e-01]],
[[-1.6211e+00, -5.5919e-01, 4.0620e-02, ..., 3.1703e-01,
2.4447e-01, 8.5316e-01],
[-2.1541e+00, -3.3526e-01, -6.1336e-02, ..., -4.2865e-01,
-1.2532e-01, 1.1876e+00],
[-1.7650e+00, -5.8890e-01, -1.7322e-01, ..., -4.0447e-02,
5.1015e-02, 9.5147e-01],
...,
[-1.8094e+00, -5.1247e-01, 2.5029e-01, ..., 1.2341e-01,
6.5411e-01, 1.3250e+00],
[-2.2131e+00, -1.1349e+00, -1.4348e-01, ..., -2.5669e-01,
-9.6406e-02, 6.1590e-01],
[-1.0609e+00, -1.8226e-01, 5.0889e-01, ..., -2.4003e-01,
-2.3134e-01, 7.2467e-01]]],
[[[-1.7991e-01, -2.7328e-01, 1.5836e-01, ..., -2.0919e+00,
-8.9285e-02, 5.5939e-01],
[-5.5185e-02, -1.4324e+00, -8.6445e-01, ..., -1.6939e+00,
-2.8271e+00, -2.3194e+00],
[-1.8509e+00, -2.7987e+00, -1.8422e+00, ..., -6.9104e-01,
-1.2833e+00, -1.6150e+00],
...,
[-1.6156e-01, -1.5932e+00, -1.2700e+00, ..., -1.0094e+00,
-1.0336e+00, -2.0948e+00],
[-1.1598e+00, -1.7726e+00, -8.1929e-01, ..., -5.4028e-01,
-4.7574e-01, -1.8368e+00],
[-1.3165e+00, -2.2208e+00, -1.6637e+00, ..., -1.4578e+00,
-1.5598e+00, -9.9309e-01]],
[[-2.1653e-01, -1.5008e+00, -8.2744e-01, ..., -9.4848e+00,
-7.0869e+00, -1.1558e+00],
[-2.0000e+00, -3.9966e+00, -2.7019e+00, ..., -8.4284e+00,
-8.8097e+00, -2.9760e+00],
[-7.5547e+00, -1.0466e+01, -1.0335e+01, ..., -3.1384e+00,
-3.8809e+00, -2.0847e+00],
...,
[-3.9272e+00, -5.4812e+00, -5.3667e+00, ..., -5.4883e+00,
-6.2518e+00, -2.9808e+00],
[-6.2239e+00, -7.0451e+00, -6.0891e+00, ..., -6.1363e+00,
-7.4710e+00, -2.7651e+00],
[-7.1115e-02, -9.3360e-01, 2.4532e-01, ..., 4.1547e-01,
7.0489e-01, 1.7664e-01]],
[[-4.5522e-01, -7.9053e-02, 3.8452e-01, ..., -1.1282e+01,
-8.8551e+00, -2.8246e+00],
[-3.6540e-01, -4.2291e-01, 5.3293e-01, ..., -8.9112e+00,
-7.6302e+00, -4.4156e+00],
[-1.2434e+00, -7.9774e-01, -1.4475e-01, ..., -2.7826e+00,
-2.7772e+00, -1.9421e+00],
...,
[-8.8819e-01, -2.8204e-03, -4.7186e-02, ..., 1.4145e-02,
-4.4820e-02, 3.0066e-01],
[-1.2261e+00, 1.2937e-01, -2.6486e-02, ..., -5.3785e-02,
-7.1783e-02, 3.0098e-01],
[-1.1802e+00, -6.9732e-02, 1.0099e-01, ..., -4.1905e-03,
-1.1076e-01, -3.5670e-01]],
...,
[[ 1.7842e-01, -5.1135e-01, -3.1839e-01, ..., -5.3408e+00,
-5.5335e+00, -5.3909e+00],
[-6.8782e-01, -5.7444e-01, -5.2651e-01, ..., -3.8583e+00,
-4.4217e+00, -4.9720e+00],
[-2.2830e-01, -1.9741e-01, -6.8741e-01, ..., -2.3196e+00,
-2.3348e+00, -2.8226e+00],
...,
[-9.5626e-01, -1.3700e+00, -1.4521e+00, ..., -3.6522e-01,
-5.8041e-01, -1.1984e+00],
[-3.1889e+00, -3.0086e+00, -2.0590e+00, ..., -1.9149e+00,
-2.0379e+00, -2.0864e+00],
[-5.1650e-01, -1.6622e+00, -1.1294e+00, ..., -8.6311e-01,
-9.3030e-01, -1.3580e+00]],
[[-1.1922e-01, 7.3238e-02, 3.9619e-01, ..., -2.4918e+00,
-1.6622e+00, -8.8466e-01],
[-3.0001e-01, -1.7978e-01, 2.1741e-02, ..., -2.4826e-01,
-1.9278e+00, -4.1796e-01],
[-1.0796e+00, -6.0827e-01, -6.5013e-01, ..., -1.0750e+00,
-1.0768e+00, 4.7621e-01],
...,
[-3.6203e-02, -1.2756e+00, -5.8612e-01, ..., 2.7075e-01,
1.8695e-01, -7.4003e-01],
[-5.3131e-01, -1.8541e+00, -1.8495e-01, ..., 4.5097e-02,
3.2484e-02, -9.5830e-01],
[ 4.2146e-01, -1.7533e-01, 1.2181e-01, ..., 2.7645e-01,
3.4346e-01, 3.3856e-01]],
[[-6.1066e-01, 4.2503e-01, 5.6237e-01, ..., -7.5609e+00,
-7.4398e+00, -4.4382e+00],
[-8.0589e-01, 2.6058e-01, -2.1751e-01, ..., -1.9299e+00,
-2.9238e+00, -2.5314e+00],
[-1.1401e+00, -3.0085e-01, -6.2321e-01, ..., 1.8451e+00,
1.0285e+00, 2.1044e-01],
...,
[-1.2245e+00, -8.8291e-01, -3.3764e-02, ..., -4.5959e-01,
-2.3066e-02, 8.3183e-01],
[-1.9137e+00, -1.9202e+00, -1.0337e+00, ..., -8.6308e-01,
-4.0907e-01, 4.3226e-01],
[-9.6252e-01, -9.8647e-01, -2.7017e-01, ..., -3.3269e-01,
-1.3164e-01, 1.1467e+00]]],
[[[-1.9770e-01, -2.6124e-01, 1.9438e-01, ..., 5.6003e-02,
4.8137e-01, 1.5374e-01],
[-9.2213e-02, -1.4759e+00, -8.4938e-01, ..., -7.6677e-01,
-4.7997e-01, -3.1732e-01],
[-1.3304e-01, -1.2951e+00, -6.4845e-01, ..., -6.5299e-01,
-4.6608e-01, -6.9097e-01],
...,
[-1.4284e-01, -1.9045e+00, -1.4615e+00, ..., -1.3399e+00,
-1.2189e+00, -1.9401e+00],
[-3.8748e-01, -1.4770e+00, -7.4911e-01, ..., -3.1517e-01,
1.9421e-01, -6.5277e-01],
[-2.4362e-01, -1.3230e+00, -1.1539e+00, ..., -7.9870e-01,
-1.1398e+00, -7.8346e-01]],
[[-1.9113e-01, -1.4763e+00, -6.1186e-01, ..., -6.5318e-01,
-1.3445e+00, -2.8109e-01],
[-1.9396e+00, -3.9366e+00, -2.4631e+00, ..., -2.3884e+00,
-3.3541e+00, -1.1784e+00],
[-1.8432e+00, -3.3420e+00, -2.5294e+00, ..., -3.0076e+00,
-3.3078e+00, -1.2754e+00],
...,
[-3.6938e+00, -5.1988e+00, -5.0769e+00, ..., -6.2953e+00,
-6.5158e+00, -3.0829e+00],
[-5.4225e+00, -6.5877e+00, -5.7368e+00, ..., -5.1380e+00,
-5.6790e+00, -2.2416e+00],
[ 5.6922e-01, -3.9980e-01, 5.4618e-01, ..., -1.8791e-02,
1.1421e-01, -1.5432e-01]],
[[-5.2808e-01, -1.4856e-01, 3.5500e-01, ..., 4.2445e-01,
1.2100e-01, 9.1882e-01],
[-3.5915e-01, -4.9622e-01, 4.3872e-01, ..., 3.2756e-01,
2.7030e-01, 4.9375e-01],
[-8.6550e-01, -2.7508e-01, 1.3009e-01, ..., 7.0756e-02,
1.9058e-01, 3.8613e-01],
...,
[-1.7266e+00, -3.8014e-01, -6.1317e-02, ..., 3.3810e-01,
2.9143e-01, -3.8282e-02],
[-1.4434e+00, -2.7629e-01, -1.1429e-01, ..., 1.1199e-01,
1.4831e-01, -2.0345e-01],
[-5.8354e-01, -7.4815e-02, 3.7617e-03, ..., -1.1456e+00,
-6.1805e-01, -7.4048e-01]],
...,
[[ 1.1508e-01, -6.1033e-01, -4.2573e-01, ..., -6.3772e-02,
-2.4670e-01, -7.5125e-01],
[-8.1776e-01, -5.8873e-01, -5.4692e-01, ..., -2.7762e-01,
-9.9845e-01, -7.7476e-01],
[-5.1591e-01, -1.4244e-01, -1.3017e-02, ..., 2.0972e-01,
-1.2644e-01, -5.0886e-01],
...,
[-1.0231e+00, -8.7942e-01, -2.7928e-01, ..., -4.0992e-01,
-7.3822e-01, -1.3104e+00],
[-2.5536e+00, -2.0213e+00, -1.3797e+00, ..., -1.4081e+00,
-1.7878e+00, -1.7761e+00],
[-5.5878e-01, -1.1178e+00, -1.7331e-01, ..., -9.8298e-01,
-1.0975e+00, -6.6515e-01]],
[[-1.5746e-01, 6.6429e-02, 4.2815e-01, ..., 1.2914e-01,
5.4885e-01, 1.5083e-01],
[-3.9953e-01, -2.4326e-01, 3.5547e-02, ..., -3.0763e-02,
1.3810e-01, -4.2850e-02],
[-6.7682e-01, -5.4376e-01, 1.0222e-02, ..., -1.5120e-01,
-1.8810e-01, -5.6758e-02],
...,
[-1.4009e-01, -8.7854e-01, 3.0164e-01, ..., 6.2028e-01,
5.3036e-01, -3.0889e-01],
[-7.4891e-01, -2.0006e+00, -1.7680e-01, ..., 5.9142e-01,
7.0179e-01, -2.6558e-02],
[ 7.8897e-01, 3.2981e-02, 3.7172e-01, ..., 4.6965e-01,
3.2780e-01, 2.7120e-02]],
[[-6.6176e-01, 4.3700e-01, 5.5408e-01, ..., 4.7957e-01,
4.0042e-01, 6.0777e-01],
[-8.6285e-01, 2.9233e-01, -2.1436e-01, ..., -3.2920e-01,
-1.7076e-01, 6.6794e-01],
[-5.0639e-01, 2.9376e-01, 2.1751e-02, ..., -2.8305e-02,
1.2977e-01, 6.2596e-01],
...,
[-1.2433e+00, -5.9544e-01, -1.5624e-01, ..., -3.3769e-01,
-1.4773e-01, 1.3616e+00],
[-1.3500e+00, -8.0887e-01, -1.1644e-01, ..., -4.5768e-01,
-1.3768e-01, 1.2301e+00],
[-6.3305e-01, -4.9696e-01, 1.6136e-01, ..., 1.2019e-01,
2.4370e-01, 1.0559e+00]]],
[[[-3.3458e-01, -8.0505e-01, -1.5196e+00, ..., -5.5953e-01,
-1.2642e-01, -8.6612e-01],
[ 2.4766e-02, -1.3220e+00, -2.6435e-02, ..., -1.0124e+00,
-8.4182e-01, -1.2421e+00],
[-2.8541e-02, -1.2728e+00, 1.7172e-01, ..., -7.9313e-01,
-8.1454e-01, -1.7539e+00],
...,
[ 6.9324e-02, -5.9306e-01, -2.1807e-01, ..., -8.4284e-01,
-1.8581e+00, -1.3422e+00],
[-3.4686e-01, -4.7645e-01, 1.0216e-02, ..., -1.9285e-01,
1.7198e-01, -1.8857e-01],
[-6.0944e-01, -1.3284e+00, -1.1105e+00, ..., -1.3467e+00,
-1.2159e+00, -5.1019e-01]],
[[-1.9796e+00, -5.9323e+00, -9.5601e+00, ..., -3.9964e+00,
-4.9863e+00, -2.0504e+00],
[-3.9984e+00, -6.0740e+00, -5.9433e+00, ..., -5.5037e+00,
-6.7635e+00, -2.9414e+00],
[-3.4859e+00, -5.4998e+00, -4.7032e+00, ..., -5.1693e+00,
-5.7579e+00, -2.7941e+00],
...,
[-1.7785e+00, -2.2606e+00, -1.8864e+00, ..., -2.4299e+00,
-3.0580e+00, -1.1631e+00],
[-2.9001e+00, -3.1132e+00, -2.1280e+00, ..., -2.5467e+00,
-3.5798e+00, -1.0336e+00],
[ 3.5818e-01, -2.4041e-01, 3.4795e-01, ..., 3.3866e-01,
4.0004e-01, 2.0308e-01]],
[[-5.1169e-01, -1.6197e-01, -2.6429e+00, ..., 8.0925e-01,
5.6561e-01, 1.4947e+00],
[-6.5033e-01, 1.8661e-01, -2.3109e+00, ..., 4.6853e-01,
2.6598e-01, 4.2418e-01],
[-1.2152e+00, 5.0365e-01, 2.7978e-01, ..., 1.8927e-01,
1.7394e-01, 3.4007e-01],
...,
[-6.6167e-01, 6.0332e-02, 2.7753e-01, ..., 8.2482e-01,
8.4415e-01, 5.6637e-01],
[-4.6732e-01, 9.4262e-02, 1.9702e-01, ..., 3.7540e-01,
2.6745e-01, 5.5904e-01],
[-1.6996e-01, -2.2305e-01, 1.5071e-01, ..., 1.0788e-02,
8.3845e-02, 3.4483e-01]],
...,
[[ 9.1552e-01, 1.1031e+00, 1.1364e+00, ..., 3.9605e-01,
2.4225e-01, -9.9760e-01],
[-9.7140e-01, -7.2920e-01, 4.9460e-01, ..., -4.0090e-01,
-1.1149e+00, -1.2986e+00],
[-4.2277e-01, -1.1283e+00, -1.5163e+00, ..., -3.0930e-01,
-5.5420e-01, -1.1405e+00],
...,
[-6.5521e-01, -5.0143e-01, -3.6707e-01, ..., -5.7307e+00,
-2.3304e+00, -6.4515e-01],
[-2.2344e+00, -1.8417e+00, -1.3306e+00, ..., -2.6409e+00,
-1.3187e+00, -1.0869e+00],
[-5.9883e-01, -8.0997e-01, -4.1068e-01, ..., -9.4250e-02,
-3.1213e-01, -3.9809e-01]],
[[ 1.2692e-01, 4.4804e-01, -2.3322e-01, ..., 2.5131e-01,
6.7595e-01, -3.1471e-01],
[ 2.0334e-01, -3.8578e-01, -6.8772e-01, ..., 1.9372e-01,
3.8711e-01, -7.1095e-01],
[-1.1382e-01, -5.3334e-01, 2.0381e-01, ..., 2.4796e-01,
1.6518e-01, -6.3459e-01],
...,
[-2.1514e-01, -4.8003e-01, 2.6275e-01, ..., -7.4118e-01,
-1.9450e-01, -1.5544e-01],
[-5.2059e-01, -9.2621e-01, 4.0690e-02, ..., 2.2273e-01,
-6.7817e-02, -1.7907e-01],
[ 3.4591e-01, -4.6278e-02, -1.8251e-01, ..., -6.4942e-02,
-1.1010e-01, 1.7775e-01]],
[[-8.3453e-01, -4.6026e-01, -2.2627e-01, ..., 2.8751e-01,
4.0956e-01, 9.4140e-01],
[-1.1889e+00, -1.1581e+00, 2.4809e-01, ..., -6.5338e-01,
-2.9997e-01, 9.4937e-01],
[-9.0172e-01, -9.3965e-01, -1.5057e-01, ..., -3.6566e-01,
-7.7228e-02, 7.9000e-01],
...,
[-6.5319e-01, -2.7049e-01, -1.6707e-01, ..., 2.0343e+00,
1.2055e+00, 1.1518e+00],
[-1.1312e+00, -6.0950e-01, -3.3973e-01, ..., 4.3707e-01,
-7.3253e-02, 1.8380e-01],
[-3.5464e-01, -4.4765e-02, -1.6464e-01, ..., 6.8278e-02,
-1.5340e-01, 3.5822e-01]]],
[[[-2.1371e-01, -2.8833e-01, 1.6461e-01, ..., 2.6767e-02,
4.9199e-01, 1.6319e-01],
[-1.2827e-01, -1.5598e+00, -9.0742e-01, ..., -7.2992e-01,
-4.3692e-01, -3.3753e-01],
[-1.4968e-01, -1.3612e+00, -9.8185e-01, ..., -4.8436e-01,
-3.8284e-01, -6.8072e-01],
...,
[-2.5717e-01, -1.7286e+00, -1.2580e+00, ..., -1.1276e+00,
-9.6515e-01, -2.0489e+00],
[-1.1255e+00, -1.7710e+00, -8.2221e-01, ..., -8.2395e-01,
-4.7076e-01, -1.7876e+00],
[-1.2981e+00, -2.1643e+00, -1.5830e+00, ..., -1.5047e+00,
-1.5727e+00, -9.6630e-01]],
[[-2.0753e-01, -1.4738e+00, -6.7562e-01, ..., -6.4803e-01,
-1.3503e+00, -2.5840e-01],
[-2.0860e+00, -4.1098e+00, -2.5906e+00, ..., -2.4126e+00,
-3.3768e+00, -1.1526e+00],
[-2.4525e+00, -5.2494e+00, -5.4410e+00, ..., -2.5630e+00,
-2.9291e+00, -1.2149e+00],
...,
[-3.7792e+00, -5.5611e+00, -5.2304e+00, ..., -5.9475e+00,
-6.2027e+00, -2.9364e+00],
[-6.1551e+00, -6.9681e+00, -6.0322e+00, ..., -6.1920e+00,
-7.3485e+00, -2.6817e+00],
[-4.3934e-02, -9.1614e-01, 2.6260e-01, ..., 3.4173e-01,
6.8064e-01, 1.9071e-01]],
[[-5.4323e-01, -1.7637e-01, 3.6604e-01, ..., 3.4495e-01,
5.0637e-02, 8.9837e-01],
[-3.9687e-01, -4.9276e-01, 4.1508e-01, ..., 3.9659e-01,
2.3588e-01, 5.0010e-01],
[-9.1168e-01, -2.8903e-01, 7.7657e-02, ..., 9.3925e-02,
2.0401e-01, 4.1483e-01],
...,
[-1.4078e+00, 1.4159e-04, 4.7425e-02, ..., 2.7814e-01,
1.4624e-03, 2.9671e-01],
[-1.2072e+00, 1.1665e-01, -1.7399e-02, ..., 2.5784e-01,
-2.6915e-02, 2.8766e-01],
[-1.1546e+00, -7.6459e-02, 9.4689e-02, ..., -1.0828e-02,
-4.5927e-02, -3.4602e-01]],
...,
[[ 6.6535e-02, -5.6856e-01, -3.7124e-01, ..., -5.1919e-02,
-2.2925e-01, -7.3677e-01],
[-8.5836e-01, -5.9272e-01, -5.3259e-01, ..., -2.7592e-01,
-9.8080e-01, -7.5023e-01],
[-4.6841e-01, 5.9581e-02, 3.8301e-01, ..., 9.0734e-02,
-1.7891e-01, -5.2706e-01],
...,
[-6.1076e-01, -6.0045e-01, -3.4421e-01, ..., -1.2677e+00,
-5.8786e-01, -1.1737e+00],
[-3.1415e+00, -2.8809e+00, -1.9778e+00, ..., -1.9635e+00,
-2.0244e+00, -2.0614e+00],
[-5.4846e-01, -1.6451e+00, -1.0944e+00, ..., -8.5044e-01,
-9.4951e-01, -1.3485e+00]],
[[-1.8491e-01, 3.5240e-02, 4.2488e-01, ..., 8.8355e-02,
5.4902e-01, 1.8627e-01],
[-4.4514e-01, -2.5597e-01, 2.1991e-02, ..., -5.6361e-02,
1.2580e-01, -5.4889e-02],
[-5.3053e-01, -1.5578e-01, 1.3617e-01, ..., 4.7239e-02,
4.6810e-02, -1.4104e-03],
...,
[-1.2349e-01, -1.0292e+00, 5.5098e-02, ..., 3.1468e-01,
1.6461e-01, -7.0432e-01],
[-5.1370e-01, -1.8286e+00, -1.7974e-01, ..., -1.8834e-01,
8.9388e-03, -9.1122e-01],
[ 4.2878e-01, -1.3902e-01, 1.6611e-01, ..., 2.0086e-01,
3.0557e-01, 3.2395e-01]],
[[-6.7963e-01, 4.0264e-01, 5.6159e-01, ..., 4.7925e-01,
3.7081e-01, 5.9965e-01],
[-8.6672e-01, 2.3744e-01, -2.5085e-01, ..., -2.7161e-01,
-1.7978e-01, 6.6084e-01],
[-5.0109e-01, 2.8238e-01, -9.1676e-02, ..., 8.1849e-02,
6.4036e-02, 6.2204e-01],
...,
[-1.1183e+00, -9.2991e-01, -4.3934e-01, ..., 3.4005e-02,
-2.4894e-02, 8.3320e-01],
[-1.8998e+00, -1.8514e+00, -9.9031e-01, ..., -6.0096e-01,
-4.9025e-01, 4.3950e-01],
[-9.5285e-01, -9.7530e-01, -2.5859e-01, ..., -2.2607e-01,
-1.4519e-01, 1.1294e+00]]]], device='cuda:0'),
tensor([[[[-7.0762e-01, -1.4700e+00, -1.4455e+00, ..., -6.6473e-02,
-2.0377e-01, -1.0312e+00],
[-3.2748e-01, -2.4524e+00, -2.2522e+00, ..., 6.5760e-01,
8.8650e-02, -8.5423e-01],
[-1.2569e-01, -1.9368e+00, -1.7652e+00, ..., 1.3157e+00,
1.8299e+00, -9.1385e-01],
...,
[-7.7779e-02, -1.0892e+00, -3.7041e-01, ..., 1.8343e-02,
-4.1601e-01, -9.8944e-01],
[-1.2161e-01, -9.4726e-01, -3.7998e-02, ..., 8.8111e-02,
6.4052e-02, -1.0128e+00],
[-1.3018e-01, -1.1034e+00, -8.6375e-01, ..., -5.7861e-02,
-1.0781e+00, -6.3447e-01]],
[[-2.7479e+00, -5.3013e+00, -4.9234e+00, ..., -5.8367e+00,
-7.7429e+00, -4.2543e+00],
[-5.3798e+00, -8.2271e+00, -7.6267e+00, ..., -9.7687e+00,
-1.0814e+01, -5.9660e+00],
[-4.1063e+00, -5.9494e+00, -5.9120e+00, ..., -2.9865e+00,
-5.7910e+00, -3.6021e+00],
...,
[-2.1163e+00, -3.3524e+00, -2.6379e+00, ..., -4.1440e+00,
-4.3166e+00, -1.5407e+00],
[-4.5324e+00, -5.6195e+00, -4.9518e+00, ..., -7.1927e+00,
-6.9857e+00, -2.4468e+00],
[ 6.7610e-01, -3.0745e-01, 5.2537e-01, ..., -3.7828e-01,
2.3761e-01, -1.2608e-02]],
[[-6.4005e-01, 4.6078e-01, 8.2071e-01, ..., -3.0655e+00,
-3.0589e-01, 2.2832e+00],
[-9.5088e-01, 3.0950e-02, 4.0657e-01, ..., -3.4345e+00,
-3.3317e+00, 1.9223e-01],
[-1.7811e+00, -2.1575e-01, -4.3628e-02, ..., -3.5649e+00,
-3.6985e+00, -3.7637e-01],
...,
[-9.5227e-01, 4.0774e-02, 3.5829e-01, ..., -1.0255e+00,
1.5948e-01, 3.4606e-01],
[-9.8458e-01, 1.2551e-01, 5.6102e-01, ..., -1.1845e+00,
-1.1943e-03, 3.4446e-01],
[-4.8514e-01, -1.1098e-02, 1.3004e-01, ..., -1.9426e-01,
2.2247e-01, -1.0174e-02]],
...,
[[ 1.2519e+00, -3.6768e-01, -2.4915e-02, ..., -4.3537e-01,
-1.9801e+00, -2.8640e+00],
[-7.2772e-01, -8.5743e-01, -4.2098e-01, ..., -5.6003e+00,
-5.8360e+00, -3.7641e+00],
[-7.3603e-01, -9.1160e-01, -3.4256e-01, ..., -6.8406e+00,
-6.3418e+00, -3.7628e+00],
...,
[-4.7654e-01, -1.2161e-01, 7.3034e-01, ..., -4.4049e+00,
-6.3846e-01, -5.8325e-01],
[-2.0608e+00, -1.4528e+00, -1.2563e+00, ..., -4.5761e+00,
-1.8544e+00, -1.1642e+00],
[-3.3238e-01, -7.8846e-01, -2.1998e-01, ..., -2.8799e+00,
-9.8070e-01, -6.6452e-01]],
[[ 6.9223e-02, 1.4747e-02, 4.1670e-01, ..., 4.5140e-01,
6.5451e-01, -4.4945e-01],
[ 2.6209e-01, -7.5191e-01, 9.6427e-03, ..., -9.7817e-01,
4.5082e-01, -6.0857e-01],
[ 1.7682e-01, -6.0217e-01, 3.5006e-01, ..., -7.3619e-01,
1.2582e+00, 5.9548e-01],
...,
[-4.3970e-01, -6.0416e-01, -2.8195e-01, ..., 1.1066e+00,
1.6908e-01, -1.7248e-01],
[-7.5481e-01, -1.5294e+00, -3.9185e-01, ..., 1.0634e-01,
-1.6437e-02, -7.1426e-01],
[ 5.0556e-01, 5.2363e-02, 1.8067e-01, ..., 1.7228e+00,
4.5561e-01, 3.3505e-01]],
[[-1.5695e+00, -8.1205e-01, -4.7420e-01, ..., -3.9825e+00,
-3.0503e-01, 1.3319e+00],
[-2.1647e+00, -8.0337e-01, -6.5043e-01, ..., -5.8629e+00,
-6.6008e-01, 2.0752e+00],
[-1.5346e+00, -6.2631e-01, -3.2337e-01, ..., -5.5367e+00,
-1.2707e+00, 1.7570e+00],
...,
[-4.3957e-01, 9.8080e-02, -1.7838e-01, ..., 4.6976e-01,
8.4524e-01, 7.2147e-01],
[-8.8381e-01, -2.6104e-01, -8.2545e-02, ..., 6.6756e-02,
5.3091e-01, 4.5413e-01],
[-4.0715e-01, -3.1159e-01, -1.7708e-01, ..., -3.2682e-01,
-5.0543e-02, 6.0053e-01]]],
[[[-2.8039e-01, -2.4509e+00, -2.6106e+00, ..., -3.1346e-01,
-2.8400e+00, -2.4484e+00],
[ 9.1767e-01, -5.2793e+00, -3.8979e+00, ..., -5.8579e+00,
-8.4609e+00, -6.4906e-02],
[-4.7227e-01, -6.8083e+00, -4.0157e+00, ..., -8.8343e+00,
-5.8687e+00, 1.9037e+00],
...,
[-6.1064e-01, -2.1359e+00, -1.9952e+00, ..., -2.6554e+00,
-4.2920e+00, -3.1980e+00],
[-1.2496e+00, -2.0564e+00, -1.3495e+00, ..., 8.3016e-01,
1.2545e-01, -1.7534e+00],
[-9.7942e-01, -1.9859e+00, -1.4948e+00, ..., 1.6843e-01,
-2.6181e-01, -1.6680e+00]],
[[-5.7635e+00, -9.2571e+00, -1.0797e+01, ..., -1.0212e+01,
-1.3806e+01, -5.0340e+00],
[-1.0897e+01, -1.5010e+01, -1.6982e+01, ..., -1.5811e+01,
-1.5966e+01, -3.6981e+00],
[-7.2127e+00, -8.5131e+00, -9.2362e+00, ..., -1.4369e+01,
-1.1810e+01, 2.7064e-01],
...,
[-4.7676e+00, -6.7073e+00, -5.9663e+00, ..., -1.3357e+01,
-1.5331e+01, -4.1280e+00],
[-6.8192e+00, -8.0167e+00, -6.9259e+00, ..., -4.9291e+00,
-8.0718e+00, -4.0193e+00],
[-2.0150e-01, -9.8636e-01, 2.0152e-01, ..., 2.8999e-01,
8.5626e-01, -1.9596e-01]],
[[-8.5237e-01, 3.4374e-01, 8.9300e-01, ..., -9.4056e-01,
-1.1316e+00, 1.6351e+00],
[-1.8249e+00, -2.2462e+00, -5.5513e-01, ..., -4.2091e+00,
-9.1133e+00, -3.0948e+00],
[-3.1027e+00, -8.1886e+00, -9.3615e+00, ..., -5.7380e+00,
-1.5370e+01, -8.5125e+00],
...,
[-1.8543e+00, -1.0952e-01, 4.2934e-02, ..., -7.5439e+00,
-1.2367e+01, -5.7104e+00],
[-1.5586e+00, -5.1734e-02, -7.4329e-02, ..., 8.9389e-01,
-4.4991e+00, -2.5015e+00],
[-1.5178e+00, -6.6560e-01, -3.5713e-01, ..., 2.0578e+00,
-2.1051e-01, -8.6813e-01]],
...,
[[ 1.2545e+00, -3.1827e-01, -3.6500e-02, ..., 1.0628e+00,
5.6841e-01, -6.4534e-01],
[ 6.3725e-01, 8.9691e-01, -7.3464e-01, ..., 2.4326e+00,
-1.0207e-01, -6.4664e+00],
[ 4.6958e-01, -8.2316e-01, -4.4884e+00, ..., -2.1510e-01,
-6.5059e+00, -1.3473e+01],
...,
[-6.5977e-01, -1.2685e+00, -9.7872e-01, ..., 3.4892e-01,
-2.0428e+00, -1.0486e+01],
[-3.0657e+00, -3.3323e+00, -2.0684e+00, ..., -8.4146e-01,
-2.0159e+00, -9.7032e+00],
[-7.0702e-01, -1.8732e+00, -1.1564e+00, ..., -2.3429e+00,
-3.1540e+00, -3.7815e+00]],
[[ 4.2799e-01, -7.4049e-02, 5.7004e-02, ..., 1.3326e+00,
-9.2588e-03, -3.3266e+00],
[ 1.3778e+00, -1.3569e+00, -1.8914e+00, ..., -1.7171e+00,
-3.7020e+00, -2.0796e+00],
[ 7.0017e-01, -3.0712e+00, -4.3898e+00, ..., -5.9087e+00,
-7.3433e+00, 3.1537e-02],
...,
[ 1.7313e-01, -8.6843e-01, 3.9489e-01, ..., -2.3477e+00,
-3.6402e+00, -3.1805e+00],
[-1.0962e-01, -1.3766e+00, 9.4519e-02, ..., -3.2979e-01,
-1.4636e+00, -3.1375e+00],
[ 6.1688e-01, 9.3029e-02, 2.4014e-01, ..., 7.8148e-01,
-3.5521e-02, -2.3798e+00]],
[[-1.9387e+00, -1.2711e+00, -2.4199e-02, ..., -4.7136e-01,
-1.2231e+00, -5.2145e-01],
[-3.1312e+00, -1.9615e+00, -2.8106e+00, ..., -4.4341e+00,
-7.9375e+00, -3.8291e+00],
[-3.9274e+00, -3.8463e+00, -5.7137e+00, ..., -8.1906e+00,
-1.2566e+01, -7.4484e+00],
...,
[-2.1673e+00, -5.5621e-01, 1.5329e-01, ..., -2.8787e+00,
-8.4914e+00, -6.1979e+00],
[-2.7429e+00, -1.4181e+00, -3.9278e-01, ..., -1.7294e+00,
-8.5443e-01, -6.1041e-01],
[-1.2644e+00, -2.5211e-01, 3.5966e-01, ..., -6.2952e-01,
8.5097e-01, 1.6942e+00]]],
[[[-8.0122e-01, -1.4243e+00, -1.4129e+00, ..., 6.8391e-03,
6.3364e-01, 5.4271e-01],
[-3.6093e-01, -2.3070e+00, -2.1679e+00, ..., -4.7625e-01,
-2.4133e-01, 9.0650e-02],
[-3.9151e-01, -1.9639e+00, -1.8213e+00, ..., -5.1838e-01,
-1.1829e-01, -9.4371e-02],
...,
[-5.1606e-01, -2.0132e+00, -1.8706e+00, ..., -3.2642e-01,
-7.3265e-02, -1.3962e-01],
[-1.2043e+00, -1.8441e+00, -1.1165e+00, ..., 2.7756e-02,
2.8817e-01, -2.7106e-02],
[-1.0870e+00, -1.9702e+00, -1.4258e+00, ..., -1.0580e+00,
-1.0949e+00, -3.9997e-01]],
[[-2.4843e+00, -4.9147e+00, -4.3664e+00, ..., -4.0581e-01,
-9.8236e-01, -1.8187e-02],
[-5.1528e+00, -7.7468e+00, -6.6050e+00, ..., -1.7959e+00,
-2.6002e+00, -6.5228e-01],
[-4.3383e+00, -6.0783e+00, -5.5644e+00, ..., -1.6875e+00,
-1.5313e+00, -3.9706e-01],
...,
[-4.3342e+00, -6.0825e+00, -5.4855e+00, ..., -1.3008e+00,
-1.5421e+00, -3.5634e-01],
[-6.6170e+00, -7.7577e+00, -6.5162e+00, ..., -2.4670e+00,
-3.3273e+00, -9.5572e-01],
[-8.5522e-02, -1.0772e+00, 4.2150e-01, ..., 3.1175e-01,
3.5162e-01, 8.4795e-02]],
[[-5.9719e-01, 7.3154e-01, 1.0492e+00, ..., 8.1285e-01,
3.1938e-01, 1.0937e+00],
[-9.7930e-01, 6.9866e-02, 3.2686e-01, ..., 3.8694e-01,
2.6181e-02, 4.6977e-01],
[-1.6906e+00, -1.2466e-01, -1.2600e-01, ..., 7.1925e-02,
7.5061e-02, 4.0108e-01],
...,
[-1.7643e+00, -1.5876e-01, -8.3812e-02, ..., 8.7940e-02,
-1.9492e-02, 4.1016e-01],
[-1.5116e+00, -1.9871e-02, 1.1301e-03, ..., 1.3797e-01,
8.2842e-02, 5.9781e-01],
[-1.4171e+00, -6.1162e-01, -2.7466e-01, ..., -8.9576e-02,
2.6412e-02, 3.3633e-01]],
...,
[[ 1.2409e+00, -3.2482e-01, -2.2771e-01, ..., -8.8379e-02,
-3.7130e-01, -9.1988e-01],
[-6.7470e-01, -1.0038e+00, -8.2269e-01, ..., -7.1322e-01,
-1.2425e+00, -8.0679e-01],
[-6.6147e-01, -1.0462e+00, -7.1571e-01, ..., -9.7773e-02,
-1.8942e-01, -3.7363e-01],
...,
[-8.1696e-01, -1.1624e+00, -7.5750e-01, ..., -1.0320e-01,
-2.0451e-01, -3.6302e-01],
[-3.1690e+00, -3.0634e+00, -1.9939e+00, ..., -8.2382e-01,
-8.8745e-01, -8.0214e-01],
[-6.8449e-01, -1.7547e+00, -1.0334e+00, ..., -2.3904e-02,
-1.5205e-01, -3.3834e-01]],
[[ 8.2859e-02, 8.0669e-02, 5.7301e-01, ..., 3.2535e-01,
6.9352e-01, 2.9667e-01],
[ 1.1864e-01, -7.4002e-01, 1.2788e-01, ..., -2.8346e-02,
1.0344e-01, -4.5954e-02],
[-2.2259e-03, -6.9294e-01, 3.4697e-01, ..., -5.6616e-01,
-7.6714e-02, 2.0661e-01],
...,
[-6.3237e-02, -8.8814e-01, 2.5897e-01, ..., -1.5900e-01,
-1.1732e-01, 1.4103e-01],
[-4.6599e-01, -1.6245e+00, -3.6422e-02, ..., -3.9449e-01,
-3.2602e-01, -6.0296e-02],
[ 5.4021e-01, 1.0074e-01, 1.9669e-01, ..., -2.6239e-01,
-2.4379e-01, 3.2047e-01]],
[[-1.5228e+00, -6.0504e-01, 3.4896e-03, ..., 5.3379e-01,
3.5969e-01, 5.4221e-01],
[-2.1342e+00, -3.9637e-01, -2.6802e-02, ..., -7.9101e-02,
-1.7054e-01, 4.1558e-01],
[-1.7850e+00, -3.8306e-01, 2.1464e-01, ..., 7.6981e-02,
-1.7061e-01, 1.0536e-01],
...,
[-1.7700e+00, -5.2832e-01, 2.6125e-01, ..., 1.9908e-02,
-5.9661e-02, 7.9027e-02],
[-2.3862e+00, -1.3704e+00, -2.2047e-01, ..., -1.1720e-01,
-2.7425e-01, -2.1764e-01],
[-1.1161e+00, -4.2047e-01, 3.2539e-01, ..., -1.2706e-01,
-2.8990e-01, 1.1809e-01]]],
[[[-1.1542e-01, -5.1810e-01, -2.3470e-01, ..., -1.7524e-01,
3.4762e-01, 4.6761e-03],
[ 3.0720e-01, -1.1535e+00, -6.7866e-01, ..., -4.1823e-01,
-2.6041e-01, -1.2938e-01],
[ 2.3232e-01, -7.9903e-01, -3.7944e-01, ..., -1.9896e-01,
-2.1780e-01, -5.2218e-01],
...,
[-2.6665e-01, -1.6137e+00, -1.0190e+00, ..., -9.6530e-01,
-9.2810e-01, -2.0085e+00],
[-1.1133e+00, -1.7105e+00, -8.1555e-01, ..., -5.2834e-01,
-4.4721e-01, -1.7760e+00],
[-1.3004e+00, -2.1637e+00, -1.6171e+00, ..., -1.4541e+00,
-1.5769e+00, -9.6650e-01]],
[[-1.4728e+00, -3.1431e+00, -2.7217e+00, ..., -2.5510e+00,
-3.3503e+00, -1.2664e+00],
[-3.0359e+00, -4.9037e+00, -3.7077e+00, ..., -3.7103e+00,
-4.9266e+00, -2.1614e+00],
[-2.0738e+00, -2.9918e+00, -2.3472e+00, ..., -2.4527e+00,
-2.7119e+00, -1.2553e+00],
...,
[-3.6952e+00, -5.6258e+00, -5.5554e+00, ..., -5.3648e+00,
-6.1132e+00, -2.8446e+00],
[-6.2538e+00, -7.0812e+00, -6.0975e+00, ..., -6.0808e+00,
-7.4009e+00, -2.6536e+00],
[-4.2735e-02, -8.9975e-01, 2.6189e-01, ..., 3.8469e-01,
6.6786e-01, 1.8622e-01]],
[[-4.2121e-01, 2.2163e-01, 8.1364e-01, ..., 7.8406e-01,
5.3110e-01, 1.3711e+00],
[-2.8587e-01, -5.8791e-02, 5.3486e-01, ..., 5.0075e-01,
4.2104e-01, 2.6657e-01],
[-7.1834e-01, 7.3728e-02, 2.5960e-01, ..., 2.2979e-01,
3.4904e-01, 4.1548e-01],
...,
[-1.3978e+00, 3.6837e-01, 8.6045e-01, ..., 6.5545e-02,
-9.1922e-04, 2.9704e-01],
[-1.2270e+00, 1.3770e-01, -6.5007e-03, ..., -2.6812e-02,
2.3320e-02, 2.8922e-01],
[-1.1971e+00, -6.5638e-02, 9.6427e-02, ..., 4.3828e-03,
-1.4530e-02, -3.3193e-01]],
...,
[[ 9.7708e-01, 3.0680e-01, 2.9956e-01, ..., 3.6795e-01,
1.2192e-01, -9.6763e-01],
[-1.0158e+00, -8.2610e-01, -6.5576e-01, ..., -7.4626e-01,
-1.6462e+00, -1.1940e+00],
[-5.0293e-01, -3.0191e-01, -2.6711e-01, ..., -3.2711e-01,
-4.8859e-01, -6.1636e-01],
...,
[-6.0460e-01, -8.3800e-01, -1.0516e+00, ..., -3.4941e-01,
-5.5609e-01, -1.1630e+00],
[-3.1146e+00, -2.9362e+00, -2.0480e+00, ..., -1.9110e+00,
-2.0572e+00, -2.0598e+00],
[-5.1095e-01, -1.6176e+00, -1.0828e+00, ..., -8.3722e-01,
-9.4180e-01, -1.3440e+00]],
[[ 8.7232e-02, 3.2020e-01, 7.2512e-01, ..., 5.9229e-01,
1.0604e+00, 4.4646e-01],
[ 1.0485e-01, -3.5671e-01, 2.2095e-01, ..., 3.8322e-01,
6.0409e-01, -1.9952e-01],
[-1.9481e-01, -5.3865e-01, 2.1716e-01, ..., 3.1717e-01,
3.2035e-01, 2.3704e-02],
...,
[-1.2768e-01, -8.3957e-01, 3.9241e-01, ..., 2.3819e-01,
1.4474e-01, -6.9486e-01],
[-5.4094e-01, -1.8821e+00, -2.3362e-01, ..., 1.3025e-02,
-5.9604e-02, -9.0572e-01],
[ 4.4651e-01, -1.7354e-01, 1.2716e-01, ..., 2.5257e-01,
2.7581e-01, 3.3379e-01]],
[[-9.3552e-01, -2.5702e-01, 1.1318e-01, ..., 1.5270e-01,
4.2970e-02, 7.6742e-01],
[-1.1180e+00, -2.9566e-01, -3.3479e-01, ..., -4.0270e-01,
-1.0230e-01, 1.2367e+00],
[-6.9174e-01, -2.6019e-01, -2.6070e-01, ..., -2.3998e-01,
-1.5533e-01, 6.8110e-01],
...,
[-1.1350e+00, -1.1559e+00, -8.5925e-01, ..., -4.0378e-01,
-6.6572e-02, 8.3169e-01],
[-1.8297e+00, -1.8380e+00, -1.0108e+00, ..., -8.1508e-01,
-5.0568e-01, 4.2555e-01],
[-9.5106e-01, -9.6601e-01, -2.3533e-01, ..., -2.6983e-01,
-1.6404e-01, 1.1327e+00]]],
[[[ 1.5925e-02, -2.5552e-01, 2.1593e-03, ..., -1.2085e+00,
-6.3810e-01, -9.4239e-01],
[ 3.1854e-01, -1.4100e+00, -1.1597e+00, ..., -1.6497e+00,
-1.3131e+00, -1.2288e+00],
[-1.8462e+00, -3.7628e+00, -2.9915e+00, ..., -1.4334e+00,
-1.2936e+00, -1.7913e+00],
...,
[ 7.4136e-02, -5.5211e-01, -2.6097e-01, ..., -1.4640e+00,
-1.2007e+00, -1.6582e+00],
[-4.8591e-01, -5.4966e-01, 1.4794e-01, ..., -7.7501e-01,
-5.4778e-01, -1.4423e+00],
[-6.0788e-01, -1.2818e+00, -1.0622e+00, ..., -1.2740e+00,
-1.4883e+00, -9.0614e-01]],
[[-6.6945e-01, -2.1104e+00, -1.7160e+00, ..., -4.2052e+00,
-4.8313e+00, -1.7926e+00],
[-3.6246e+00, -6.4770e+00, -6.3303e+00, ..., -6.3728e+00,
-7.0133e+00, -2.8070e+00],
[-6.8551e+00, -9.0263e+00, -1.0270e+01, ..., -5.8991e+00,
-6.1103e+00, -2.6698e+00],
...,
[-1.7335e+00, -2.3530e+00, -2.1487e+00, ..., -5.3050e+00,
-5.4378e+00, -2.2843e+00],
[-3.6482e+00, -4.2265e+00, -3.4036e+00, ..., -6.2268e+00,
-7.2841e+00, -2.7915e+00],
[ 2.9113e-01, -3.0699e-01, 3.6983e-01, ..., 5.5555e-01,
6.5919e-01, 5.9631e-02]],
[[-2.8585e-01, 7.8887e-02, 7.7986e-01, ..., 9.9079e-01,
6.9293e-01, 1.4926e+00],
[ 4.5430e-02, -6.7079e-01, 5.1627e-01, ..., 3.3525e-01,
3.9998e-01, 3.1116e-01],
[-4.1863e-01, -7.7484e-01, 1.0067e-01, ..., -1.3104e-01,
2.0838e-01, 1.1710e-01],
...,
[-1.8175e-01, -1.5392e-01, 2.7416e-01, ..., -1.8468e-01,
9.9233e-02, 2.1686e-02],
[-1.6042e-01, 1.6048e-02, 1.9903e-01, ..., -6.2488e-02,
1.2530e-01, 1.5513e-01],
[-9.8957e-02, -4.3839e-01, 1.1396e-01, ..., -3.9265e-01,
-6.8548e-02, -2.6796e-01]],
...,
[[ 4.6552e-01, -3.7058e-01, -7.1015e-02, ..., -1.1956e-01,
-2.7652e-01, -1.4764e+00],
[-6.7981e-01, -6.7215e-01, -8.7565e-01, ..., -7.9870e-01,
-1.5727e+00, -1.7220e+00],
[ 6.2456e-01, 7.8921e-01, -3.4546e-01, ..., -7.3241e-01,
-9.5782e-01, -1.2630e+00],
...,
[-3.8564e-01, 1.8323e-01, 4.8341e-02, ..., -7.4351e-01,
-8.9697e-01, -1.1846e+00],
[-2.2199e+00, -1.6356e+00, -1.4281e+00, ..., -1.8581e+00,
-1.9383e+00, -1.7378e+00],
[-4.7500e-01, -7.2125e-01, -4.2487e-01, ..., -6.8185e-01,
-7.2407e-01, -8.4680e-01]],
[[ 7.4374e-02, 2.5615e-01, 6.9880e-01, ..., 3.7605e-01,
6.9315e-01, -1.8804e-02],
[ 3.3506e-01, 4.9897e-01, -1.8704e-01, ..., 2.5182e-01,
3.1674e-01, -2.8406e-01],
[-9.9064e-01, -1.3770e+00, 2.4891e-01, ..., 4.0044e-01,
9.6036e-02, -5.4074e-01],
...,
[-1.3463e-01, 4.6588e-02, 4.8692e-01, ..., 1.6935e-01,
1.3206e-01, -3.2884e-01],
[-4.5949e-01, -6.0803e-01, 7.8191e-02, ..., -4.4544e-02,
-1.2996e-01, -6.0200e-01],
[ 5.2486e-01, 3.8959e-01, 7.5966e-02, ..., 2.2364e-01,
1.3859e-01, 3.2190e-01]],
[[-9.7600e-01, 3.0358e-01, 3.9780e-01, ..., 1.4031e-01,
1.6879e-01, 8.1824e-01],
[-1.1240e+00, 3.1926e-02, -9.2328e-01, ..., 2.2992e-02,
1.1830e-01, 1.3291e+00],
[-2.4768e+00, -3.2808e+00, -3.4654e+00, ..., 2.3490e-01,
3.4681e-01, 1.1384e+00],
...,
[-9.7209e-01, 1.7835e-01, -1.1360e-01, ..., 2.4831e-01,
4.0733e-01, 1.2282e+00],
[-1.6025e+00, -2.4799e-01, -8.7814e-02, ..., -1.2856e-01,
1.5428e-01, 7.8268e-01],
[-6.5186e-01, 3.7126e-02, -2.2737e-01, ..., 2.6067e-01,
1.2360e-01, 1.0462e+00]]]], device='cuda:0'),
tensor([[[[ 3.2115e-02, -2.2025e-01, -1.5894e-02, ..., -1.1179e+00,
-6.8909e-01, -1.0918e+00],
[ 2.0931e-01, -1.1629e+00, -7.2959e-01, ..., -1.4511e+00,
-1.4788e+00, -1.3190e+00],
[ 5.3269e-02, -6.8560e-01, -5.2265e-01, ..., -1.0889e+00,
-1.2804e+00, -1.8324e+00],
...,
[ 3.9465e-02, -3.5038e-01, -5.6627e-01, ..., -1.3783e+00,
-1.1631e+00, -1.6440e+00],
[-3.8492e-01, -5.9751e-01, -3.4703e-01, ..., -7.6088e-01,
-5.7289e-01, -1.4300e+00],
[-5.5771e-01, -1.1959e+00, -9.9615e-01, ..., -1.2976e+00,
-1.5330e+00, -9.3272e-01]],
[[-7.0310e-01, -2.2293e+00, -1.8150e+00, ..., -4.4314e+00,
-5.2054e+00, -2.0839e+00],
[-2.4988e+00, -4.5299e+00, -3.2545e+00, ..., -6.4308e+00,
-7.0545e+00, -3.0705e+00],
[-2.0835e+00, -3.3024e+00, -2.6893e+00, ..., -5.6444e+00,
-5.8545e+00, -3.1041e+00],
...,
[-1.9629e+00, -3.3334e+00, -2.9509e+00, ..., -5.2572e+00,
-5.4046e+00, -2.3866e+00],
[-3.5635e+00, -4.6254e+00, -3.6403e+00, ..., -6.2759e+00,
-7.2028e+00, -2.8252e+00],
[ 2.6938e-01, -2.5440e-01, 4.0099e-01, ..., 5.5478e-01,
6.6749e-01, 5.0584e-02]],
[[-3.1290e-01, -8.5142e-03, 7.4425e-01, ..., 9.8461e-01,
7.2384e-01, 1.4807e+00],
[ 3.6209e-02, -4.8353e-01, 5.9604e-01, ..., 3.2259e-01,
2.0962e-01, 7.9737e-02],
[-3.4215e-01, -2.1565e-01, 1.6148e-01, ..., 1.0927e-01,
-9.7797e-02, -4.1677e-01],
...,
[-2.0746e+00, -1.3770e+00, 3.5630e-01, ..., -1.2707e-01,
5.6397e-02, 5.9674e-02],
[-3.1287e-01, -5.7813e-01, 3.8201e-01, ..., -7.2445e-02,
7.2235e-02, 2.3618e-01],
[-3.5135e-02, -4.2732e-01, 1.3538e-01, ..., -3.8742e-01,
-1.9917e-01, -2.6990e-01]],
...,
[[ 4.1061e-01, -4.1871e-01, -6.7576e-02, ..., -7.2347e-02,
-3.0853e-01, -1.4944e+00],
[-7.1203e-01, -5.5690e-01, -6.1538e-01, ..., -6.1408e-01,
-1.4005e+00, -1.9263e+00],
[-4.2736e-01, -3.0229e-01, -3.0056e-01, ..., -5.7569e-01,
-1.0081e+00, -1.7201e+00],
...,
[-6.2830e-01, -1.5639e+00, -7.8449e-01, ..., -6.7367e-01,
-8.4913e-01, -1.1677e+00],
[-2.8528e+00, -2.1105e+00, -1.3992e+00, ..., -1.8533e+00,
-1.9185e+00, -1.7476e+00],
[-4.6562e-01, -6.4513e-01, -3.9696e-01, ..., -6.9556e-01,
-7.5539e-01, -9.2961e-01]],
[[ 9.8295e-02, 2.4795e-01, 6.8221e-01, ..., 5.2256e-01,
8.2498e-01, -1.8710e-01],
[-2.1655e-02, 2.0200e-01, 4.2585e-01, ..., 4.8010e-01,
4.6471e-01, -4.6387e-01],
[-2.3571e-01, -7.8996e-02, 3.1108e-01, ..., 6.5943e-01,
3.6670e-01, -8.0011e-01],
...,
[-6.9830e-01, 1.2038e-01, 5.5754e-01, ..., 3.2629e-01,
1.9194e-01, -3.3716e-01],
[-7.0648e-01, -4.0549e-01, -1.7442e-01, ..., -4.7204e-02,
-9.0848e-02, -5.9419e-01],
[ 5.3347e-01, 4.5959e-01, 1.0699e-01, ..., 2.0044e-01,
1.5507e-01, 3.2091e-01]],
[[-9.9179e-01, 2.7695e-01, 4.2371e-01, ..., 1.0397e-01,
1.9160e-01, 8.4561e-01],
[-1.2609e+00, 3.0411e-01, -3.2144e-01, ..., -1.7814e-01,
-4.9364e-02, 1.1528e+00],
[-1.1942e+00, 4.0368e-01, -1.3728e-01, ..., 8.6716e-02,
3.8548e-01, 1.0520e+00],
...,
[-2.0106e+00, 5.7087e-01, 3.3947e-01, ..., 2.0310e-01,
3.8997e-01, 1.2671e+00],
[-1.7320e+00, 2.4346e-01, -6.1040e-02, ..., -1.4475e-01,
1.3248e-01, 8.1917e-01],
[-6.4987e-01, 1.1018e-01, -1.9601e-01, ..., 1.9304e-01,
8.2974e-02, 1.0412e+00]]],
[[[-7.0090e-01, -4.3356e+00, -5.5263e+00, ..., -1.0119e-01,
5.5032e-01, 4.0995e-01],
[-5.3903e-01, -9.3392e+00, -8.3662e+00, ..., -5.4339e-01,
-2.8060e-01, 4.2023e-02],
[ 4.0770e-01, -8.0527e+00, -9.9172e+00, ..., -8.0214e-01,
-2.2404e-01, -2.0597e-01],
...,
[-5.3220e-01, -1.9720e+00, -2.4530e+00, ..., -1.0700e+00,
-3.9073e-01, -2.0143e-01],
[-1.1294e+00, -1.8188e+00, -3.5973e-02, ..., -5.6396e-01,
-2.2127e-01, -7.0963e-02],
[-9.0231e-01, -1.7822e+00, -3.3956e-01, ..., -1.0413e+00,
-1.1128e+00, -3.8389e-01]],
[[-6.6182e+00, -1.3055e+01, -1.1853e+01, ..., -7.0200e-01,
-1.3284e+00, -1.6515e-01],
[-9.8715e+00, -1.5518e+01, -1.3307e+01, ..., -2.1248e+00,
-2.9847e+00, -8.1572e-01],
[-1.0583e+01, -1.5561e+01, -1.5086e+01, ..., -3.4631e+00,
-1.8737e+00, -5.0895e-01],
...,
[-4.5559e+00, -6.4629e+00, -1.0592e+01, ..., -3.0514e+00,
-2.7436e+00, -4.9795e-01],
[-6.5714e+00, -7.6778e+00, -7.1695e+00, ..., -3.3516e+00,
-3.5631e+00, -1.0964e+00],
[-1.2739e-01, -9.2877e-01, 2.7194e-01, ..., 3.7578e-01,
3.7855e-01, 7.5967e-02]],
[[-1.3292e+00, -2.2917e+00, -8.3937e-01, ..., 9.0944e-01,
3.8764e-01, 1.1789e+00],
[-2.2603e+00, -7.2400e+00, -1.0650e+01, ..., 4.8383e-01,
8.4004e-02, 4.8786e-01],
[-1.4144e+00, -1.0210e+01, -1.7142e+01, ..., 3.1877e-01,
1.3369e-01, 3.8559e-01],
...,
[-1.8107e+00, -5.4961e-02, -4.8004e-01, ..., -1.3244e+00,
1.1845e+00, 4.5476e-01],
[-1.5523e+00, 8.4900e-02, 2.7670e-01, ..., 2.5798e-01,
8.0360e-01, 6.2167e-01],
[-1.4891e+00, -5.3941e-01, 1.4556e+00, ..., -1.6484e-01,
8.0574e-02, 3.9354e-01]],
...,
[[ 1.6605e+00, 6.7557e-01, -4.5596e-02, ..., -7.2729e-02,
-3.3632e-01, -9.7493e-01],
[ 1.0873e+00, 1.5497e+00, -4.3519e+00, ..., -7.0194e-01,
-1.2947e+00, -8.9228e-01],
[ 1.8988e-01, -2.7971e+00, -1.0767e+01, ..., 6.6732e-03,
-2.5258e-01, -4.6279e-01],
...,
[-7.2786e-01, -1.2060e+00, -3.9084e-01, ..., -3.7716e+00,
-9.8528e-01, -4.1221e-01],
[-2.9152e+00, -3.0427e+00, -9.1474e-01, ..., -1.6874e+00,
-4.6789e-01, -8.2754e-01],
[-6.9712e-01, -1.8392e+00, -2.1551e+00, ..., -5.5574e-02,
-1.5511e-01, -3.5413e-01]],
[[ 6.7617e-01, -2.4104e-01, -2.8205e+00, ..., 3.6982e-01,
6.9387e-01, 3.0959e-01],
[ 1.2351e-01, -4.1597e+00, -8.0517e+00, ..., 1.9385e-02,
9.0211e-02, -5.5152e-02],
[ 5.1755e-02, -7.5324e+00, -1.2202e+01, ..., -1.5817e+00,
-2.3385e-01, 1.4980e-01],
...,
[ 1.1369e-01, -7.6059e-01, -4.6214e-01, ..., 2.1494e+00,
4.8906e-02, 1.4612e-01],
[-1.7765e-01, -1.3719e+00, -9.2892e-01, ..., -8.8751e-01,
-6.4028e-01, -1.1275e-01],
[ 6.3770e-01, 1.4900e-01, 4.9550e-01, ..., -1.9930e-01,
-2.4489e-01, 2.9011e-01]],
[[-1.9534e+00, -2.2022e+00, -7.7305e-01, ..., 5.8815e-01,
3.6725e-01, 6.2696e-01],
[-4.2241e+00, -5.3127e+00, -5.6256e+00, ..., 5.9506e-03,
-1.9873e-01, 5.0767e-01],
[-3.9927e+00, -4.7383e+00, -5.1753e+00, ..., 3.2263e-01,
-2.3478e-01, 1.8984e-01],
...,
[-1.9318e+00, -5.2559e-01, -2.2256e+00, ..., 6.0761e-01,
7.2950e-01, 1.7921e-01],
[-2.4877e+00, -1.3202e+00, -1.4953e+00, ..., 1.2881e+00,
-4.8208e-04, -1.1687e-01],
[-1.1656e+00, -2.3616e-01, -5.6441e-02, ..., -1.0099e-01,
-2.5947e-01, 1.7672e-01]]],
[[[-2.2887e-01, -6.1765e-01, -2.9955e-01, ..., -3.4084e-01,
1.2767e-01, -3.2867e-01],
[-4.6418e+00, -6.6795e+00, -5.0311e+00, ..., -8.5063e-01,
-6.1446e-01, -6.9325e-01],
[-3.1406e+00, -3.9453e+00, -5.6381e+00, ..., -1.0142e+00,
-6.4036e-01, -1.1178e+00],
...,
[-4.9592e-01, -1.6872e+00, -2.5357e+00, ..., -8.0891e+00,
-1.8177e+01, -8.4076e+00],
[-3.5177e-01, -1.2930e+00, -1.5690e+00, ..., -9.1082e+00,
-1.5528e+01, -7.5628e+00],
[-3.3580e-01, -1.2040e+00, -8.9522e-01, ..., -3.0429e+00,
-4.3925e+00, -1.9686e+00]],
[[-1.0442e+00, -2.6901e+00, -2.1342e+00, ..., -2.1677e+00,
-2.8828e+00, -9.9749e-01],
[-1.6082e+01, -2.1446e+01, -1.6322e+01, ..., -3.7855e+00,
-4.9395e+00, -1.8712e+00],
[-1.5575e+01, -1.9623e+01, -1.7946e+01, ..., -4.3572e+00,
-4.6305e+00, -1.8800e+00],
...,
[-4.5255e+00, -7.0063e+00, -7.8414e+00, ..., -5.6983e-01,
-5.0723e+00, 5.3679e+00],
[-6.3805e+00, -8.2403e+00, -7.0102e+00, ..., -4.3982e+00,
-1.0907e+01, 6.1964e+00],
[ 2.9559e-01, -9.3311e-01, 3.0996e-01, ..., -9.2429e-01,
-4.5796e+00, -1.3786e+00]],
[[-5.0331e-01, 1.6487e-01, 7.8012e-01, ..., 7.1735e-01,
5.0543e-01, 1.1929e+00],
[ 1.1582e+00, 2.4859e+00, 2.8944e+00, ..., 3.6403e-01,
4.3741e-01, 4.3105e-01],
[-2.5139e+00, 2.9757e+00, 4.6733e+00, ..., 1.2408e-01,
1.8652e-01, 1.9893e-01],
...,
[-1.4757e+00, -5.3726e-01, -1.2002e+00, ..., -2.2773e+01,
-2.5416e+01, -4.6551e+00],
[-1.7945e+00, 8.1737e-02, -8.5777e-01, ..., -2.0387e+01,
-2.1747e+01, -7.0062e+00],
[-1.1173e+00, 2.7225e-03, -6.7678e-01, ..., -1.5174e+01,
-6.9581e+00, -5.2756e+00]],
...,
[[ 1.6500e-01, -2.2623e-01, -1.1361e-01, ..., 7.9934e-02,
-2.5896e-02, -8.0501e-01],
[-1.2981e+00, -1.4227e+00, -1.4939e+00, ..., -2.3526e-01,
-1.0719e+00, -9.8955e-01],
[ 2.6024e+00, -3.0168e+00, -2.0012e+00, ..., -1.2574e-01,
-3.0392e-01, -6.9979e-01],
...,
[-8.9555e-01, -9.1400e-01, -3.8652e-01, ..., -4.3072e+00,
-7.4581e+00, -1.4827e+01],
[-2.2254e+00, -2.0376e+00, -1.4735e+00, ..., -8.2580e+00,
-7.7844e+00, -1.1983e+01],
[-5.6596e-01, -1.1825e+00, -3.8168e-01, ..., -2.6550e+00,
-3.8843e+00, -6.1355e+00]],
[[-8.6124e-02, 4.1323e-02, 2.8830e-01, ..., 6.7625e-02,
5.0699e-01, -1.9815e-02],
[-1.2458e-02, -2.0485e+00, -1.6793e+00, ..., 1.2820e-01,
3.0993e-01, -2.5173e-01],
[ 2.6387e+00, 1.9558e+00, 2.0457e+00, ..., -2.4025e-02,
-5.2132e-02, -3.0843e-01],
...,
[-2.2742e-01, -6.8712e-01, -4.1367e-01, ..., -8.4491e+00,
-7.5401e+00, -1.3523e+00],
[-5.4165e-01, -1.3766e+00, -2.6262e-01, ..., -5.4492e+00,
-6.5754e+00, 2.9895e+00],
[ 8.2336e-01, 2.5465e-01, 4.6597e-02, ..., 1.4883e+00,
1.1371e-01, 5.6946e+00]],
[[-5.7481e-01, 3.0465e-01, 5.5718e-01, ..., 5.3335e-01,
4.3097e-01, 8.5974e-01],
[-3.8365e+00, -2.5575e+00, -1.9446e+00, ..., -2.0663e-01,
-4.7725e-02, 9.7847e-01],
[-1.1681e+01, -1.0499e+01, -6.4071e+00, ..., 1.8338e-02,
2.6259e-01, 8.6271e-01],
...,
[-2.1629e+00, -1.4439e+00, -7.6140e-01, ..., -9.3089e+00,
-1.9552e+01, -1.5939e+01],
[-1.6943e+00, -1.4885e+00, -4.7845e-01, ..., -7.2919e+00,
-1.6170e+01, -1.4118e+01],
[-8.6891e-01, -4.9152e-01, 8.6883e-01, ..., -2.5727e+00,
-5.9530e+00, -4.0268e+00]]],
[[[-7.0057e-01, -1.4709e+00, -1.4464e+00, ..., -1.2464e+00,
-8.7435e-01, -1.6803e+00],
[-3.0042e-01, -2.4322e+00, -2.2403e+00, ..., -1.7307e+00,
-1.8517e+00, -1.8882e+00],
[-1.1348e-01, -1.9088e+00, -1.6998e+00, ..., -1.8353e+00,
-2.1034e+00, -2.3077e+00],
...,
[-2.4442e-01, -1.2996e+00, -6.1100e-01, ..., -1.1596e+00,
-7.4690e-01, -8.7352e-01],
[-5.9544e-01, -1.3295e+00, -3.8102e-01, ..., -5.3254e-01,
-2.8501e-01, -9.6979e-01],
[-5.5791e-01, -1.3767e+00, -1.0233e+00, ..., -6.7338e-01,
-1.1717e+00, -5.8941e-01]],
[[-2.6772e+00, -5.0743e+00, -4.6079e+00, ..., -4.8034e+00,
-6.0751e+00, -2.4809e+00],
[-5.3031e+00, -8.0861e+00, -7.4127e+00, ..., -7.6993e+00,
-9.1356e+00, -4.3277e+00],
[-4.0099e+00, -5.7813e+00, -5.6866e+00, ..., -9.2082e+00,
-8.3278e+00, -4.5512e+00],
...,
[-2.0875e+00, -3.4222e+00, -2.8439e+00, ..., -5.1789e+00,
-4.1650e+00, -1.4458e+00],
[-4.9338e+00, -6.0972e+00, -5.0675e+00, ..., -5.9598e+00,
-6.3480e+00, -2.2733e+00],
[ 4.8080e-01, -6.9564e-01, 4.4633e-01, ..., -4.7496e-01,
5.0772e-01, 1.9484e-02]],
[[-6.7155e-01, 4.2438e-01, 8.4484e-01, ..., 8.1180e-01,
6.3976e-01, 1.8935e+00],
[-9.3908e-01, -8.1971e-04, 4.1446e-01, ..., 5.2583e-01,
-2.3938e-02, 5.1165e-02],
[-1.7419e+00, -2.6723e-01, -4.2022e-02, ..., 2.1952e-01,
2.0482e-01, -1.3587e-02],
...,
[-1.0649e+00, -1.8316e-01, 1.1527e-02, ..., 1.0817e-01,
1.8188e-01, 3.7086e-01],
[-1.0459e+00, -9.4570e-02, 7.1784e-02, ..., -1.4597e+00,
-8.9809e-02, 3.9216e-01],
[-4.7823e-01, -6.8847e-03, -2.9476e-02, ..., -6.8962e-01,
1.2591e-01, 3.7103e-02]],
...,
[[ 1.2049e+00, -3.7586e-01, -6.7110e-02, ..., 2.3308e-01,
-1.0146e-01, -2.0919e+00],
[-7.3341e-01, -8.9874e-01, -4.6800e-01, ..., -4.9097e-01,
-1.4484e+00, -2.0296e+00],
[-7.2553e-01, -8.4053e-01, -2.9401e-01, ..., -3.9599e-01,
-5.6069e-01, -1.2030e+00],
...,
[-5.9164e-01, -6.6367e-02, 1.5821e-01, ..., -1.4053e+00,
-1.9994e-01, -5.3476e-01],
[-2.4819e+00, -1.6331e+00, -9.1867e-01, ..., -3.8815e+00,
-1.3008e+00, -1.0495e+00],
[-3.8507e-01, -1.1270e+00, -2.2340e-01, ..., -2.6890e+00,
-4.2120e-01, -5.7617e-01]],
[[ 1.4357e-02, -1.5948e-02, 4.4938e-01, ..., 4.8360e-01,
9.1736e-01, -6.5942e-02],
[ 2.0127e-01, -7.9560e-01, -4.3815e-02, ..., 4.5521e-01,
4.1387e-01, -7.8584e-01],
[ 1.3133e-01, -6.6644e-01, 2.6739e-01, ..., -7.0946e-02,
-2.2335e-01, -9.6315e-01],
...,
[-5.5632e-01, -6.2588e-01, 8.9399e-02, ..., 2.8918e-01,
-2.1211e-01, -8.0302e-02],
[-1.0299e+00, -1.7442e+00, -3.8130e-01, ..., 8.6438e-01,
-3.8414e-01, -6.1447e-01],
[ 4.6784e-01, 1.0530e-01, 2.4269e-01, ..., 5.4113e-01,
5.3115e-02, 3.6316e-01]],
[[-1.5461e+00, -8.0147e-01, -4.1966e-01, ..., -4.2053e-01,
-2.9156e-01, 8.9021e-01],
[-2.1123e+00, -7.6747e-01, -5.5348e-01, ..., -8.0439e-01,
-2.3962e-01, 1.6721e+00],
[-1.4632e+00, -5.8421e-01, -2.1744e-01, ..., -1.0424e+00,
-5.3222e-01, 1.4334e+00],
...,
[-4.9319e-01, 1.4386e-01, 9.2548e-02, ..., -1.7112e-01,
4.4145e-01, 7.2099e-01],
[-1.1083e+00, -5.1491e-01, -1.9067e-01, ..., 1.2602e-01,
3.4883e-01, 4.4138e-01],
[-5.6819e-01, -4.8902e-01, -2.3884e-01, ..., 1.5363e-01,
-1.3878e-02, 5.9975e-01]]],
[[[-1.6430e+00, -1.3545e+00, -8.4898e-01, ..., -1.0225e+00,
-5.4594e-01, -1.4574e+00],
[ 1.7067e-01, -7.8262e-01, -1.0866e+00, ..., -1.2005e+00,
-1.3224e+00, -1.5011e+00],
[ 1.0262e+00, 3.6565e-01, -1.2592e+00, ..., -9.1938e-01,
-1.1216e+00, -1.8405e+00],
...,
[ 5.7850e-02, -5.5866e-01, 4.7013e-01, ..., -5.1567e-01,
-1.0378e+00, -1.9567e+00],
[-5.0691e-01, -6.4237e-01, -1.3733e+00, ..., -1.3382e-01,
-5.8745e-01, -1.7447e+00],
[-6.2487e-01, -1.2556e+00, -2.4205e+00, ..., -1.2956e+00,
-1.5529e+00, -7.8399e-01]],
[[-4.8520e+00, -7.0958e+00, -4.2054e+00, ..., -3.9810e+00,
-5.4960e+00, -1.8413e+00],
[-5.2815e+00, -8.0565e+00, -7.6324e+00, ..., -5.9959e+00,
-8.1756e+00, -3.4447e+00],
[-5.1243e+00, -7.1653e+00, -7.7285e+00, ..., -5.0634e+00,
-6.2524e+00, -2.5795e+00],
...,
[-1.7368e+00, -2.4121e+00, -6.8038e+00, ..., -5.7019e+00,
-6.3796e+00, -2.4854e+00],
[-3.6002e+00, -4.1076e+00, -8.9232e+00, ..., -7.1120e+00,
-7.9996e+00, -2.5446e+00],
[ 2.8688e-01, -2.8601e-01, -1.0857e+00, ..., 4.2504e-01,
4.5557e-01, 2.2091e-01]],
[[-4.1089e+00, -4.3371e+00, -2.1197e+00, ..., 6.4603e-01,
2.7545e-01, 1.5352e+00],
[-4.2138e+00, -3.6566e+00, -2.1547e+00, ..., 5.8378e-01,
-4.3367e-02, 1.0513e-01],
[-3.8728e+00, -3.9740e+00, -2.2471e+00, ..., 3.7874e-02,
2.5515e-02, -1.2206e-01],
...,
[-2.2796e-01, -1.7854e-01, 1.2632e+00, ..., 2.8660e-01,
-4.2640e-02, -1.3375e-01],
[-1.8127e-01, -6.5700e-02, 1.6529e+00, ..., 3.5171e-01,
-4.3947e-02, -1.8951e-01],
[-1.0133e-01, -3.7794e-01, 7.7462e-01, ..., 3.0391e-01,
1.9354e-01, -5.9109e-01]],
...,
[[-1.2440e+00, -2.0033e+00, -1.3913e+00, ..., 3.9657e-01,
3.4257e-01, -1.3116e+00],
[-1.2713e-01, -2.6261e+00, -3.5837e+00, ..., -2.0844e-01,
-7.9067e-01, -1.4127e+00],
[-1.4275e+00, -5.5922e+00, -4.6647e+00, ..., -1.4279e-01,
-1.1377e-01, -9.8609e-01],
...,
[-4.2263e-01, -1.0221e-01, 3.3945e-01, ..., -5.0581e-01,
-1.1268e-01, -9.5225e-01],
[-2.1653e+00, -1.5436e+00, -1.6191e-01, ..., -2.5089e+00,
-1.5839e+00, -1.8283e+00],
[-4.4779e-01, -6.8997e-01, -1.6496e-01, ..., -2.7192e-01,
-5.2999e-01, -1.1938e+00]],
[[-2.9054e+00, 2.6566e-01, 5.0342e-01, ..., -9.9511e-02,
3.8481e-01, -3.5511e-01],
[-2.6219e+00, 5.4333e-01, 1.3133e+00, ..., 1.9527e-01,
9.7373e-02, -8.3446e-01],
[-4.3527e+00, 4.7948e-01, 1.6142e+00, ..., 2.4625e-01,
1.6602e-01, -4.6506e-01],
...,
[-1.2338e-01, 5.1401e-02, 1.2650e+00, ..., -6.3261e-02,
1.6456e-01, -5.0124e-01],
[-4.5710e-01, -6.5035e-01, 5.7004e-01, ..., -2.8215e-01,
-2.6339e-01, -9.0865e-01],
[ 5.2189e-01, 3.8501e-01, 4.8484e-01, ..., 3.0867e-01,
3.3627e-01, 3.1496e-01]],
[[-3.5782e+00, -3.7080e+00, -3.6143e-01, ..., -6.7485e-02,
-8.6240e-02, 5.7765e-01],
[-3.8195e+00, -4.9774e+00, -1.8268e-01, ..., -8.8152e-01,
-4.1104e-01, 9.1079e-01],
[-6.7754e+00, -4.1759e+00, 7.0246e-01, ..., -5.9857e-01,
-5.1996e-01, 6.3420e-01],
...,
[-9.5645e-01, 1.6216e-01, -4.4526e-01, ..., -2.1685e-01,
-5.1065e-01, 6.1662e-01],
[-1.5845e+00, -3.5837e-01, -7.7338e-01, ..., -2.1177e-01,
-6.8513e-01, 3.8732e-01],
[-6.3968e-01, -1.1809e-03, -6.1010e-01, ..., -1.6192e-01,
-4.2973e-01, 1.0133e+00]]]], device='cuda:0'),
tensor([[[[-1.9053e-01, -2.3497e-01, 1.9194e-01, ..., 6.7653e-02,
5.7930e-01, 2.5544e-01],
[-8.1457e-02, -1.4122e+00, -7.9465e-01, ..., -7.5490e-01,
-4.5310e-01, -2.5083e-01],
[-3.5175e-02, -1.1972e+00, -6.4134e-01, ..., -5.0624e-01,
-4.3664e-01, -6.7689e-01],
...,
[-2.5583e-01, -1.7212e+00, -1.1672e+00, ..., -9.5593e-01,
-9.6072e-01, -2.0317e+00],
[-1.1098e+00, -1.7532e+00, -8.3562e-01, ..., -5.3396e-01,
-4.5962e-01, -1.7793e+00],
[-1.2990e+00, -2.1790e+00, -1.6240e+00, ..., -1.4230e+00,
-1.5563e+00, -9.5604e-01]],
[[-1.4221e-01, -1.4430e+00, -6.9290e-01, ..., -5.9709e-01,
-1.3157e+00, -1.8261e-01],
[-1.8523e+00, -3.7860e+00, -2.4319e+00, ..., -2.4994e+00,
-3.5846e+00, -1.2228e+00],
[-1.8553e+00, -3.2121e+00, -2.5846e+00, ..., -2.6361e+00,
-3.0259e+00, -1.2615e+00],
...,
[-3.7329e+00, -5.5389e+00, -5.2216e+00, ..., -5.3964e+00,
-6.1222e+00, -2.8995e+00],
[-6.1474e+00, -7.0099e+00, -6.1307e+00, ..., -6.1686e+00,
-7.5343e+00, -2.7423e+00],
[-4.3983e-02, -9.0703e-01, 2.9416e-01, ..., 4.0501e-01,
6.8699e-01, 1.8613e-01]],
[[-4.4077e-01, -7.8401e-02, 3.4528e-01, ..., 3.9580e-01,
6.1563e-02, 9.5371e-01],
[-3.9287e-01, -4.7550e-01, 4.5047e-01, ..., 4.3417e-01,
2.6106e-01, 4.8616e-01],
[-7.5384e-01, -2.3216e-01, 1.2149e-01, ..., 6.4237e-02,
1.7349e-01, 4.0984e-01],
...,
[-1.4128e+00, -1.7759e-02, 1.9522e-01, ..., 7.2261e-02,
-2.1532e-04, 2.8534e-01],
[-1.2044e+00, 9.9113e-02, -6.0704e-02, ..., 5.2218e-04,
1.6452e-02, 2.9165e-01],
[-1.1281e+00, -5.6244e-02, 4.8567e-02, ..., -3.1763e-02,
-4.7804e-02, -3.4174e-01]],
...,
[[ 1.9494e-01, -5.0492e-01, -3.1367e-01, ..., -8.8904e-02,
-3.0687e-01, -8.2121e-01],
[-6.4265e-01, -5.8646e-01, -5.0745e-01, ..., -3.4883e-01,
-1.1698e+00, -8.3869e-01],
[-3.5947e-01, -1.9409e-01, -8.6577e-02, ..., 8.2481e-02,
-2.1749e-01, -5.3892e-01],
...,
[-5.8962e-01, -5.4667e-01, -4.8137e-01, ..., -3.7173e-01,
-5.8051e-01, -1.1958e+00],
[-3.1353e+00, -2.8770e+00, -2.0276e+00, ..., -1.8997e+00,
-2.0093e+00, -2.0465e+00],
[-5.5088e-01, -1.6252e+00, -1.0856e+00, ..., -8.4001e-01,
-9.6398e-01, -1.3702e+00]],
[[-7.5193e-02, 8.6443e-02, 4.0274e-01, ..., 1.1243e-01,
6.3090e-01, 2.4074e-01],
[-3.0436e-01, -2.1079e-01, 6.1645e-02, ..., -8.9627e-02,
1.1448e-01, -1.0546e-01],
[-5.5486e-01, -4.5065e-01, -7.3465e-03, ..., 2.5193e-02,
1.0347e-02, -5.3607e-02],
...,
[-1.1687e-01, -9.8672e-01, 2.0099e-01, ..., 2.6253e-01,
1.5444e-01, -7.3001e-01],
[-5.3423e-01, -1.8072e+00, -1.5360e-01, ..., 4.3802e-02,
-4.5026e-02, -9.8900e-01],
[ 4.2031e-01, -1.4741e-01, 1.3696e-01, ..., 2.7324e-01,
3.0387e-01, 3.0281e-01]],
[[-5.7489e-01, 4.0235e-01, 5.2679e-01, ..., 5.2377e-01,
4.0240e-01, 6.5114e-01],
[-7.7060e-01, 2.3004e-01, -3.1097e-01, ..., -2.8042e-01,
-1.5829e-01, 7.0938e-01],
[-4.7140e-01, 3.1163e-01, 1.7646e-02, ..., 5.6553e-02,
9.5404e-02, 6.0596e-01],
...,
[-1.1182e+00, -9.3022e-01, -4.5895e-01, ..., -4.1955e-01,
-6.0423e-02, 8.2506e-01],
[-1.8873e+00, -1.8428e+00, -9.5519e-01, ..., -7.4776e-01,
-4.3180e-01, 4.3619e-01],
[-9.6104e-01, -9.7407e-01, -2.9242e-01, ..., -2.9868e-01,
-1.5708e-01, 1.1177e+00]]],
[[[ 1.8847e-02, -1.9548e-01, -1.6047e-02, ..., -1.1871e+00,
-5.5965e-01, -8.4525e-01],
[ 3.3125e-01, -9.1245e-01, -6.9599e-01, ..., -1.6006e+00,
-1.2798e+00, -1.1458e+00],
[ 2.6869e-01, -5.6262e-01, -4.9428e-01, ..., -1.3755e+00,
-1.3999e+00, -1.8104e+00],
...,
[-6.5320e-03, -5.7743e-01, -4.1631e-01, ..., -1.4088e+00,
-1.1675e+00, -1.6623e+00],
[-5.9723e-01, -6.6781e-01, -6.1267e-02, ..., -7.9871e-01,
-5.8563e-01, -1.4080e+00],
[-7.2454e-01, -1.3427e+00, -1.1749e+00, ..., -1.0508e+00,
-1.3151e+00, -8.0036e-01]],
[[-6.9125e-01, -2.2153e+00, -1.8708e+00, ..., -3.9717e+00,
-4.6739e+00, -1.6945e+00],
[-1.9457e+00, -3.7579e+00, -2.9236e+00, ..., -6.1447e+00,
-6.9625e+00, -2.7839e+00],
[-1.5427e+00, -2.5967e+00, -2.5497e+00, ..., -5.7493e+00,
-6.0512e+00, -2.6970e+00],
...,
[-1.7322e+00, -2.4011e+00, -2.4540e+00, ..., -5.2970e+00,
-5.4994e+00, -2.3297e+00],
[-3.7998e+00, -4.3818e+00, -3.6491e+00, ..., -5.9456e+00,
-6.9223e+00, -2.7150e+00],
[ 2.8484e-01, -3.0245e-01, 3.9281e-01, ..., 5.4499e-01,
5.8428e-01, 4.5822e-02]],
[[-2.2622e-01, 1.3912e-01, 7.2801e-01, ..., 1.0155e+00,
7.0663e-01, 1.5258e+00],
[ 3.7772e-02, -4.3119e-01, 5.4150e-01, ..., 3.1530e-01,
3.6205e-01, 2.9785e-01],
[-1.4350e-01, -2.1562e-01, 1.8916e-01, ..., -1.3315e-01,
1.7297e-01, 1.5119e-01],
...,
[-2.2613e-01, -2.3148e-01, 1.0541e-01, ..., -1.2576e-01,
1.0994e-01, 7.7163e-02],
[-2.6058e-01, -1.6580e-01, 1.5114e-01, ..., -6.0156e-02,
1.1389e-01, 1.3609e-01],
[-1.5657e-01, -5.0068e-01, 9.7867e-02, ..., -4.1416e-01,
-7.0099e-02, -3.1563e-01]],
...,
[[ 5.8000e-01, -3.5076e-01, -1.3657e-02, ..., -1.1752e-01,
-3.2413e-01, -1.5157e+00],
[-5.5321e-01, -5.4121e-01, -5.2006e-01, ..., -7.9059e-01,
-1.6664e+00, -1.7685e+00],
[-3.7051e-01, -2.1982e-01, -2.5428e-01, ..., -6.4498e-01,
-8.7685e-01, -1.2112e+00],
...,
[-4.2952e-01, -1.6729e-01, -1.9355e-01, ..., -7.0068e-01,
-9.2765e-01, -1.2047e+00],
[-2.3098e+00, -1.7023e+00, -1.2641e+00, ..., -1.7056e+00,
-1.8684e+00, -1.6885e+00],
[-4.6577e-01, -7.8578e-01, -3.9762e-01, ..., -6.9899e-01,
-7.5497e-01, -7.9337e-01]],
[[ 1.7656e-01, 2.5540e-01, 6.2776e-01, ..., 4.3935e-01,
7.5629e-01, 4.1974e-02],
[ 1.1085e-01, 2.6712e-01, 4.8761e-01, ..., 3.2171e-01,
3.9018e-01, -2.8259e-01],
[-1.2044e-01, 8.1755e-02, 3.1378e-01, ..., 3.9895e-01,
1.5976e-01, -5.4096e-01],
...,
[-1.9812e-01, 6.0385e-02, 3.4156e-01, ..., 2.9208e-01,
1.1676e-01, -3.5406e-01],
[-5.4838e-01, -6.8443e-01, -1.7400e-02, ..., 7.2059e-02,
-5.0630e-02, -5.7757e-01],
[ 5.1144e-01, 3.8889e-01, 1.0141e-01, ..., 3.8011e-01,
2.1939e-01, 3.2397e-01]],
[[-7.9330e-01, 2.6271e-01, 3.1927e-01, ..., 1.8720e-01,
2.1961e-01, 8.9862e-01],
[-1.0732e+00, 2.2545e-01, -3.6293e-01, ..., -5.6348e-02,
1.7715e-01, 1.4143e+00],
[-9.2794e-01, 3.2637e-01, -1.0755e-01, ..., 1.8224e-01,
3.9049e-01, 1.1716e+00],
...,
[-9.8895e-01, 2.0741e-01, -8.2261e-02, ..., 2.3261e-01,
4.2818e-01, 1.2009e+00],
[-1.6839e+00, -3.6606e-01, -3.3999e-01, ..., -6.2317e-02,
2.4637e-01, 9.0384e-01],
[-7.2965e-01, -6.4833e-02, -2.8222e-01, ..., 3.1048e-01,
1.7679e-01, 1.0571e+00]]],
[[[-1.9387e-01, -2.4827e-01, 2.0693e-01, ..., 4.1320e-02,
4.8052e-01, 1.2929e-01],
[-8.5843e-02, -1.5420e+00, -8.7838e-01, ..., -7.4638e-01,
-4.5622e-01, -3.4945e-01],
[-1.4013e-01, -1.3418e+00, -6.2950e-01, ..., -4.8713e-01,
-3.8895e-01, -6.7768e-01],
...,
[-2.6546e-01, -1.7193e+00, -1.2543e+00, ..., -9.5696e-01,
-1.0148e+00, -2.0572e+00],
[-1.0926e+00, -1.7539e+00, -8.1109e-01, ..., -5.2800e-01,
-4.7920e-01, -1.8023e+00],
[-1.2818e+00, -2.1525e+00, -1.6242e+00, ..., -1.4001e+00,
-1.5409e+00, -9.5997e-01]],
[[-8.9686e-02, -1.4381e+00, -5.7334e-01, ..., -5.6271e-01,
-1.3427e+00, -2.6105e-01],
[-2.0506e+00, -4.0911e+00, -2.4517e+00, ..., -2.3342e+00,
-3.2874e+00, -1.1207e+00],
[-1.8670e+00, -3.3439e+00, -2.4800e+00, ..., -2.5555e+00,
-2.9558e+00, -1.2258e+00],
...,
[-3.7391e+00, -5.6076e+00, -5.2874e+00, ..., -5.5111e+00,
-6.2573e+00, -2.9141e+00],
[-6.1185e+00, -6.9472e+00, -6.0379e+00, ..., -6.0824e+00,
-7.3902e+00, -2.6992e+00],
[-4.9732e-02, -9.1485e-01, 2.7952e-01, ..., 4.1136e-01,
7.1445e-01, 1.8928e-01]],
[[-5.2490e-01, -1.2898e-01, 4.0007e-01, ..., 3.4780e-01,
6.7825e-02, 8.7168e-01],
[-4.3125e-01, -5.4499e-01, 4.5495e-01, ..., 3.6109e-01,
2.3104e-01, 4.7850e-01],
[-8.7591e-01, -3.2274e-01, 1.4100e-01, ..., 6.6444e-02,
1.6249e-01, 3.7984e-01],
...,
[-1.3835e+00, -5.2005e-02, -3.8379e-02, ..., -6.5619e-02,
-6.0508e-02, 2.2979e-01],
[-1.2232e+00, 9.6951e-02, -5.3286e-02, ..., -1.7767e-02,
2.2506e-02, 2.6767e-01],
[-1.1314e+00, -4.4577e-02, 5.1006e-02, ..., 3.0958e-02,
-1.7409e-02, -3.3427e-01]],
...,
[[ 1.1370e-01, -5.5438e-01, -3.5750e-01, ..., -6.3866e-02,
-2.3894e-01, -7.1103e-01],
[-8.1349e-01, -6.6490e-01, -5.8250e-01, ..., -2.8192e-01,
-9.4055e-01, -7.0953e-01],
[-4.6684e-01, -1.0354e-01, -4.4980e-02, ..., 9.3631e-02,
-1.9441e-01, -5.0939e-01],
...,
[-6.0048e-01, -5.4255e-01, -2.6187e-01, ..., -5.0212e-01,
-6.7358e-01, -1.2059e+00],
[-3.1018e+00, -2.8378e+00, -2.0272e+00, ..., -1.8816e+00,
-2.0381e+00, -2.0828e+00],
[-5.3009e-01, -1.6569e+00, -1.1432e+00, ..., -8.2442e-01,
-9.0057e-01, -1.3347e+00]],
[[-1.1828e-01, 7.0806e-02, 4.4518e-01, ..., 1.0121e-01,
5.2097e-01, 1.5276e-01],
[-4.1698e-01, -2.1644e-01, 1.3645e-03, ..., -8.9232e-02,
1.2381e-01, -5.8956e-02],
[-6.7974e-01, -5.4182e-01, 7.0242e-03, ..., 7.5071e-03,
3.0568e-02, -3.0017e-02],
...,
[-1.1747e-01, -9.9886e-01, 9.8468e-02, ..., 1.4937e-01,
4.3196e-02, -7.3245e-01],
[-5.3919e-01, -1.8075e+00, -9.6295e-02, ..., 1.0533e-02,
-4.3640e-02, -9.1266e-01],
[ 4.2381e-01, -1.5078e-01, 1.5189e-01, ..., 2.6474e-01,
3.1304e-01, 3.3773e-01]],
[[-6.3508e-01, 4.0468e-01, 5.6689e-01, ..., 4.8578e-01,
3.7696e-01, 5.9423e-01],
[-8.3448e-01, 2.9297e-01, -2.8658e-01, ..., -2.7281e-01,
-1.8259e-01, 6.2211e-01],
[-5.1104e-01, 3.3624e-01, 3.3098e-02, ..., 7.4826e-02,
1.0424e-01, 5.9536e-01],
...,
[-1.1202e+00, -9.4615e-01, -4.7098e-01, ..., -2.6953e-01,
-1.8119e-02, 8.8398e-01],
[-1.8764e+00, -1.9072e+00, -9.9565e-01, ..., -8.1862e-01,
-4.9394e-01, 4.4391e-01],
[-9.2546e-01, -9.9707e-01, -2.6096e-01, ..., -2.8291e-01,
-1.3828e-01, 1.1157e+00]]],
[[[-7.8268e-01, -1.3916e+00, -1.3962e+00, ..., -2.3406e-01,
1.9983e-01, -4.0921e-01],
[-3.3139e-01, -2.3040e+00, -2.1752e+00, ..., -1.7780e+00,
-6.5982e-01, -6.1969e-01],
[-4.1456e-01, -2.0512e+00, -2.1452e+00, ..., -1.6245e+00,
-5.6738e-01, -1.0837e+00],
...,
[-4.8353e-01, -1.9430e+00, -1.6898e+00, ..., -5.2089e-01,
3.3449e-01, -1.7860e-01],
[-1.2179e+00, -1.9182e+00, -1.0952e+00, ..., 3.7270e-01,
6.8898e-01, -9.2631e-01],
[-1.0829e+00, -2.0166e+00, -1.4438e+00, ..., -9.1527e-01,
-1.1280e+00, -6.4587e-01]],
[[-2.4692e+00, -4.9014e+00, -4.2687e+00, ..., -2.1815e+00,
-3.5194e+00, -1.3222e+00],
[-5.2417e+00, -7.8465e+00, -6.5312e+00, ..., -5.6507e+00,
-6.3666e+00, -2.1707e+00],
[-5.2064e+00, -7.8312e+00, -7.4442e+00, ..., -4.7414e+00,
-5.0932e+00, -1.8979e+00],
...,
[-4.3695e+00, -6.2625e+00, -5.4726e+00, ..., -2.5230e+00,
-3.1357e+00, -1.5664e+00],
[-6.6227e+00, -7.9097e+00, -6.6943e+00, ..., -4.0144e+00,
-5.5151e+00, -2.4105e+00],
[-3.9391e-02, -1.0626e+00, 3.6400e-01, ..., 3.9828e-01,
5.6537e-01, 1.5464e-01]],
[[-5.8436e-01, 6.9159e-01, 1.0556e+00, ..., 8.3371e-01,
5.6394e-01, 1.3415e+00],
[-9.3144e-01, 7.9797e-02, 3.3321e-01, ..., 6.8028e-01,
5.9318e-01, 2.0793e-01],
[-1.7201e+00, -7.0221e-02, -1.2257e-01, ..., -3.7306e-01,
6.8521e-01, 1.2133e-01],
...,
[-1.7460e+00, 5.1758e-02, -8.3384e-02, ..., 4.7434e-01,
5.7208e-01, -2.0865e-01],
[-1.5201e+00, -3.2475e-02, -3.8838e-02, ..., 5.6352e-01,
1.6517e-01, -3.1297e-01],
[-1.4331e+00, -6.1326e-01, -3.5144e-01, ..., -1.5602e-02,
1.7293e-01, -1.2723e-01]],
...,
[[ 1.1818e+00, -3.6424e-01, -2.3014e-01, ..., 3.5650e-01,
2.2082e-01, -7.6640e-01],
[-6.6806e-01, -9.9700e-01, -8.3573e-01, ..., -6.6575e-01,
-1.0477e+00, -9.6350e-01],
[-6.3735e-01, -1.2251e+00, -1.0905e+00, ..., -1.0194e+00,
-8.4872e-01, -6.8610e-01],
...,
[-8.8928e-01, -1.3212e+00, -1.0480e+00, ..., 8.1131e-01,
-2.0382e-01, -1.7733e+00],
[-3.1695e+00, -2.9883e+00, -2.0106e+00, ..., -1.6665e+00,
-1.8796e+00, -1.4278e+00],
[-6.7010e-01, -1.7160e+00, -1.0222e+00, ..., -1.7407e-01,
-2.7472e-01, -4.7682e-01]],
[[ 1.0564e-01, 8.5682e-02, 5.9997e-01, ..., 3.3523e-01,
8.1457e-01, 5.5973e-02],
[ 1.4035e-01, -7.2458e-01, 9.2032e-02, ..., -2.1225e-01,
2.5253e-01, -2.9334e-01],
[ 2.8102e-01, -4.3217e-01, -2.5629e-02, ..., 3.0876e-01,
2.6751e-01, -1.1531e-01],
...,
[-1.6720e-02, -7.9202e-01, 2.4723e-01, ..., 3.7863e-01,
1.2279e+00, 1.2070e+00],
[-4.1683e-01, -1.6164e+00, -1.8192e-02, ..., 1.2292e-01,
2.8279e-01, -5.9362e-01],
[ 5.8141e-01, 9.0729e-02, 2.0426e-01, ..., 2.2927e-01,
2.7535e-01, 3.6000e-01]],
[[-1.5403e+00, -6.3826e-01, -2.1996e-03, ..., 3.4297e-01,
2.4419e-01, 8.3340e-01],
[-2.0612e+00, -3.7275e-01, -1.8261e-02, ..., -1.0135e+00,
1.2675e-01, 1.1867e+00],
[-1.7114e+00, -5.1591e-01, -1.0244e-01, ..., -1.1275e+00,
4.6099e-01, 9.8819e-01],
...,
[-1.7953e+00, -4.2920e-01, 4.9548e-01, ..., -9.6101e-01,
-1.0618e+00, 4.4651e-01],
[-2.3440e+00, -1.2951e+00, -2.3215e-01, ..., 2.6593e-01,
8.7478e-01, 1.2240e+00],
[-1.1212e+00, -4.3093e-01, 3.0293e-01, ..., -1.8732e-01,
-2.1110e-01, 7.1222e-01]]],
[[[-1.9238e-01, -2.5680e-01, 1.7485e-01, ..., 5.2118e-02,
4.6740e-01, 1.2907e-01],
[-8.8299e-02, -1.4987e+00, -8.5888e-01, ..., -7.0859e-01,
-4.1082e-01, -3.2328e-01],
[-9.8581e-02, -1.3221e+00, -6.6668e-01, ..., -4.7936e-01,
-3.8720e-01, -6.7702e-01],
...,
[-1.4250e-01, -1.5055e+00, -1.2600e+00, ..., -1.0284e+00,
-1.0403e+00, -2.0344e+00],
[-8.0371e-01, -1.4889e+00, -8.4923e-01, ..., -5.2499e-01,
-4.5264e-01, -1.7478e+00],
[-1.0030e+00, -1.8582e+00, -1.5104e+00, ..., -1.5657e+00,
-1.7684e+00, -1.0574e+00]],
[[-1.9232e-01, -1.4042e+00, -5.6258e-01, ..., -6.0404e-01,
-1.3094e+00, -2.4914e-01],
[-1.9973e+00, -3.9852e+00, -2.3570e+00, ..., -2.3009e+00,
-3.2515e+00, -1.1026e+00],
[-1.8672e+00, -3.4229e+00, -2.5283e+00, ..., -2.5119e+00,
-2.9082e+00, -1.2128e+00],
...,
[-3.6886e+00, -5.4412e+00, -5.3429e+00, ..., -5.6357e+00,
-6.3160e+00, -2.8881e+00],
[-5.8905e+00, -6.7008e+00, -6.1193e+00, ..., -6.4676e+00,
-7.7723e+00, -2.8701e+00],
[ 3.7435e-02, -5.6930e-01, 3.1456e-01, ..., 3.9529e-01,
7.1639e-01, 1.6222e-01]],
[[-5.3664e-01, -1.9194e-01, 3.3435e-01, ..., 3.2675e-01,
3.7862e-02, 8.6271e-01],
[-3.3627e-01, -4.3061e-01, 5.1144e-01, ..., 4.0180e-01,
2.9648e-01, 4.9384e-01],
[-8.5495e-01, -2.3857e-01, 2.0306e-01, ..., 7.4967e-02,
1.6013e-01, 4.0513e-01],
...,
[-1.3469e+00, 9.2465e-02, 2.3484e-02, ..., 3.6684e-03,
5.6043e-04, 2.3009e-01],
[-1.2542e+00, 1.1296e-01, -6.6961e-02, ..., -5.5319e-02,
-6.2340e-02, 2.0977e-01],
[-1.1022e+00, -3.0337e-02, 3.3909e-02, ..., 4.5872e-03,
-8.4695e-02, -4.2545e-01]],
...,
[[ 5.3231e-02, -6.2172e-01, -4.0730e-01, ..., -8.7585e-02,
-2.4299e-01, -7.2628e-01],
[-8.4020e-01, -6.2870e-01, -5.8102e-01, ..., -2.6625e-01,
-9.5748e-01, -7.2788e-01],
[-5.1430e-01, -1.4096e-01, -3.7626e-02, ..., 6.1146e-02,
-2.1044e-01, -5.0192e-01],
...,
[-4.2593e-01, -6.8616e-01, -3.5058e-01, ..., -2.9252e-01,
-5.9564e-01, -1.3223e+00],
[-2.6966e+00, -2.8181e+00, -1.9307e+00, ..., -2.0454e+00,
-2.1965e+00, -2.1794e+00],
[-5.1688e-01, -1.5892e+00, -1.0017e+00, ..., -9.6620e-01,
-1.0442e+00, -1.4804e+00]],
[[-1.8432e-01, 4.1106e-02, 4.1437e-01, ..., 7.1688e-02,
5.2533e-01, 1.6083e-01],
[-4.3105e-01, -2.2952e-01, 1.8701e-03, ..., -8.1118e-02,
1.3328e-01, -4.4085e-02],
[-6.7199e-01, -5.7590e-01, -6.6934e-02, ..., 2.1718e-02,
2.3363e-02, -3.4475e-02],
...,
[ 6.8444e-02, -8.8062e-01, 8.0894e-02, ..., 2.1936e-01,
8.6379e-02, -7.5765e-01],
[-3.1744e-01, -1.6771e+00, -1.9205e-01, ..., -3.2767e-03,
-1.2056e-01, -1.0343e+00],
[ 5.3595e-01, -9.1129e-02, 1.7651e-01, ..., 1.9649e-01,
1.5767e-01, 2.3033e-01]],
[[-7.0953e-01, 4.0167e-01, 5.8565e-01, ..., 4.8284e-01,
3.8865e-01, 5.9520e-01],
[-8.5787e-01, 2.8027e-01, -2.3034e-01, ..., -2.7905e-01,
-1.7540e-01, 6.3882e-01],
[-4.5702e-01, 3.8461e-01, 3.1770e-02, ..., 6.0733e-02,
9.4897e-02, 5.8425e-01],
...,
[-1.0968e+00, -8.8527e-01, -4.0203e-01, ..., -4.8993e-01,
-8.9817e-02, 8.4654e-01],
[-1.7438e+00, -1.6145e+00, -8.8475e-01, ..., -8.5465e-01,
-4.3701e-01, 3.9270e-01],
[-9.2599e-01, -8.3056e-01, -1.9814e-01, ..., -3.2402e-01,
-2.3749e-01, 1.1043e+00]]]], device='cuda:0'),
tensor([[[[ 2.8280e-02, -2.4115e-01, -4.5476e-02, ..., -1.1999e+00,
-5.8178e-01, -9.1955e-01],
[ 3.1117e-01, -9.4757e-01, -7.6587e-01, ..., -1.3406e+00,
-1.2009e+00, -2.6720e+00],
[ 2.3623e-01, -5.9568e-01, -5.5805e-01, ..., -8.8880e-01,
-3.0730e+00, -4.0264e+00],
...,
[ 4.7891e-01, -5.3054e-02, -1.7058e-01, ..., -1.3998e+00,
-1.1549e+00, -1.7331e+00],
[-5.4782e-01, -4.4972e-01, 8.6542e-01, ..., -8.3276e-01,
-6.1220e-01, -1.4526e+00],
[-7.4405e-01, -1.3428e+00, -6.2896e-01, ..., -1.1192e+00,
-1.2933e+00, -7.9601e-01]],
[[-7.5487e-01, -2.1820e+00, -1.9551e+00, ..., -3.9822e+00,
-4.8494e+00, -1.7411e+00],
[-2.0374e+00, -3.7979e+00, -3.1591e+00, ..., -8.5392e+00,
-1.1922e+01, -5.5904e+00],
[-1.6465e+00, -2.6449e+00, -2.6417e+00, ..., -1.1089e+01,
-1.4543e+01, -7.0145e+00],
...,
[-1.9279e+00, -2.9090e+00, -5.7772e+00, ..., -5.3111e+00,
-5.6329e+00, -2.4257e+00],
[-3.9408e+00, -4.6661e+00, -4.3122e+00, ..., -6.0680e+00,
-6.8860e+00, -2.6682e+00],
[ 2.9729e-01, -3.3586e-01, 2.4825e-01, ..., 4.4122e-01,
6.1929e-01, 8.9986e-02]],
[[-2.1298e-01, 1.8117e-01, 7.3402e-01, ..., 1.0158e+00,
7.1090e-01, 1.5784e+00],
[ 3.8991e-02, -3.6054e-01, 6.0856e-01, ..., 1.2777e-01,
-3.6249e-01, -1.2399e-01],
[-1.6372e-01, -2.3699e-01, 2.3068e-01, ..., -5.4622e-01,
-1.0280e+00, -5.4438e-01],
...,
[ 5.6553e-01, -7.6049e-01, -2.3410e+00, ..., -1.5299e-01,
1.1638e-01, -7.7043e-02],
[-7.7789e-02, 6.7947e-02, -4.0126e-01, ..., -4.7878e-02,
1.4359e-01, 1.0408e-01],
[-1.0757e-01, -4.1935e-01, 9.4329e-01, ..., -3.8904e-01,
-7.4292e-02, -3.1728e-01]],
...,
[[ 6.3451e-01, -2.2182e-01, 8.6785e-02, ..., -1.1429e-01,
-3.0240e-01, -1.5843e+00],
[-5.2295e-01, -5.2714e-01, -4.9255e-01, ..., -8.3363e-01,
-1.6653e+00, -1.8365e+00],
[-3.6617e-01, -1.8378e-01, -2.0580e-01, ..., 4.0860e-01,
2.9802e-01, -2.5706e+00],
...,
[-6.9521e-01, -3.5668e-01, -7.2681e-01, ..., -7.7622e-01,
-9.9790e-01, -1.2757e+00],
[-2.5578e+00, -2.0906e+00, -1.4168e+00, ..., -1.7204e+00,
-1.8355e+00, -1.7182e+00],
[-4.7235e-01, -8.2671e-01, -6.2989e-01, ..., -7.3537e-01,
-8.2257e-01, -8.4655e-01]],
[[ 1.8351e-01, 2.9756e-01, 6.2535e-01, ..., 4.7899e-01,
8.2393e-01, 3.1748e-02],
[ 1.5127e-01, 2.5574e-01, 4.2490e-01, ..., 4.7512e-01,
6.7692e-01, -2.2089e+00],
[-1.2825e-01, 6.5603e-02, 2.7772e-01, ..., 1.4396e+00,
-6.3725e-02, -3.3315e+00],
...,
[ 9.4200e-02, -1.4783e-01, -1.4763e+00, ..., 3.5108e-01,
2.4742e-01, -3.6189e-01],
[-4.9832e-01, -6.2175e-01, -6.8431e-01, ..., 3.4917e-02,
1.3057e-02, -4.9543e-01],
[ 5.2495e-01, 3.6912e-01, 2.2950e-01, ..., 2.7825e-01,
2.3431e-01, 3.4667e-01]],
[[-8.3622e-01, 2.2833e-01, 3.0207e-01, ..., 1.6104e-01,
2.1142e-01, 8.8212e-01],
[-1.1214e+00, 1.8901e-01, -4.3776e-01, ..., -3.4117e-02,
2.9467e-01, 1.7065e+00],
[-9.5078e-01, 3.2149e-01, -1.2594e-01, ..., -2.5097e-01,
-1.2206e+00, 4.1375e-01],
...,
[-1.5460e+00, -1.5924e-02, -1.1209e-01, ..., 1.6763e-01,
4.2227e-01, 1.3375e+00],
[-1.7974e+00, -2.4171e-01, -3.3272e-01, ..., -1.3326e-01,
1.6410e-01, 9.2089e-01],
[-7.3548e-01, -1.3131e-01, -6.9438e-01, ..., 3.0851e-01,
2.6006e-01, 1.1479e+00]]],
[[[-7.0941e-01, -1.4382e+00, -1.4330e+00, ..., -1.2505e+00,
-8.5930e-01, -1.6916e+00],
[-2.9797e-01, -2.4447e+00, -2.2553e+00, ..., -1.8499e+00,
-1.8944e+00, -1.9224e+00],
[-1.2241e-01, -1.8944e+00, -1.6951e+00, ..., -1.3131e+00,
-1.4380e+00, -2.1016e+00],
...,
[-6.2998e-02, -1.1185e+00, -6.0818e-01, ..., -5.6460e-01,
-5.1487e-01, -9.0867e-01],
[-1.1603e-01, -9.8875e-01, -3.0601e-01, ..., -2.2882e-01,
-1.1144e-01, -9.7120e-01],
[-1.3469e-01, -1.0963e+00, -7.5166e-01, ..., -8.8183e-01,
-1.1155e+00, -6.2348e-01]],
[[-2.6537e+00, -5.0873e+00, -4.6134e+00, ..., -4.7528e+00,
-5.9959e+00, -2.4150e+00],
[-5.2950e+00, -8.0691e+00, -7.4032e+00, ..., -7.9820e+00,
-9.2154e+00, -4.3228e+00],
[-4.0140e+00, -5.7978e+00, -5.7423e+00, ..., -6.8181e+00,
-6.7444e+00, -3.1877e+00],
...,
[-2.0205e+00, -3.1877e+00, -2.9111e+00, ..., -2.9806e+00,
-3.4872e+00, -1.4992e+00],
[-4.4538e+00, -5.4383e+00, -4.9075e+00, ..., -5.0388e+00,
-6.2537e+00, -2.3893e+00],
[ 6.8317e-01, -3.1692e-01, 5.5145e-01, ..., 4.9292e-01,
5.5908e-01, -1.1955e-02]],
[[-6.6892e-01, 4.5348e-01, 8.5801e-01, ..., 8.2204e-01,
6.3535e-01, 1.8991e+00],
[-9.8967e-01, -1.0254e-02, 4.2970e-01, ..., 5.9034e-01,
-1.5915e-02, 1.0494e-01],
[-1.7356e+00, -2.3512e-01, -3.4587e-02, ..., 2.2969e-01,
8.6384e-02, -1.4221e-01],
...,
[-9.3634e-01, -9.6439e-02, 5.9949e-02, ..., -9.4453e-02,
8.2580e-02, 3.3321e-01],
[-9.5776e-01, 3.5507e-02, 5.9551e-02, ..., -4.0100e-02,
8.2271e-02, 3.9076e-01],
[-4.4012e-01, -2.0269e-02, 9.1352e-02, ..., -1.8651e-02,
8.6502e-02, 4.1223e-02]],
...,
[[ 1.2089e+00, -3.7657e-01, -5.0229e-02, ..., 2.2683e-01,
-1.0526e-01, -2.1056e+00],
[-7.5178e-01, -8.9558e-01, -4.8385e-01, ..., -5.1027e-01,
-1.4664e+00, -2.0454e+00],
[-7.2490e-01, -8.9919e-01, -3.0844e-01, ..., -2.6816e-01,
-5.9994e-01, -1.3795e+00],
...,
[-4.1782e-01, -1.2230e-01, 1.0836e-01, ..., 9.4872e-02,
-1.3668e-01, -5.6889e-01],
[-2.0135e+00, -1.3226e+00, -8.1876e-01, ..., -9.3553e-01,
-1.1379e+00, -1.1306e+00],
[-3.3088e-01, -7.5030e-01, -1.1763e-01, ..., -1.6478e-01,
-2.9988e-01, -6.1817e-01]],
[[ 3.4531e-02, 1.1906e-02, 4.6335e-01, ..., 4.8758e-01,
9.2576e-01, -3.4715e-02],
[ 1.7887e-01, -8.0912e-01, -3.3312e-02, ..., 2.0265e-01,
3.1352e-01, -8.0651e-01],
[ 9.5550e-02, -6.4388e-01, 2.8277e-01, ..., 3.1919e-01,
4.1637e-01, -3.7435e-01],
...,
[-3.8366e-01, -5.3257e-01, 6.4882e-02, ..., 1.2902e-01,
4.6515e-02, -1.3234e-01],
[-7.2863e-01, -1.4759e+00, -2.6932e-01, ..., -3.1727e-01,
-3.5030e-01, -6.9826e-01],
[ 5.0865e-01, 5.6743e-02, 2.2730e-01, ..., 2.2423e-01,
1.4093e-01, 3.2921e-01]],
[[-1.5563e+00, -7.9905e-01, -4.3496e-01, ..., -4.1990e-01,
-2.9041e-01, 8.7963e-01],
[-2.1293e+00, -7.5782e-01, -5.8808e-01, ..., -7.7491e-01,
-2.1852e-01, 1.6540e+00],
[-1.4862e+00, -5.8582e-01, -2.2542e-01, ..., -4.3970e-01,
-9.2772e-02, 1.4165e+00],
...,
[-4.4079e-01, 1.6274e-01, 1.2022e-01, ..., 1.5203e-01,
2.2197e-01, 7.1694e-01],
[-9.2191e-01, -3.0223e-01, -1.6602e-01, ..., -7.5861e-02,
9.5928e-02, 4.0915e-01],
[-4.2002e-01, -3.2453e-01, -1.9614e-01, ..., -2.3743e-01,
-2.4279e-01, 5.8007e-01]]],
[[[-2.3698e-01, -3.6337e-01, 1.0181e-01, ..., -2.3396e-02,
4.1549e-01, 6.6571e-02],
[-1.2538e-01, -1.6137e+00, -9.2844e-01, ..., -7.9439e-01,
-5.0395e-01, -4.1956e-01],
[-2.2558e-01, -1.8815e+00, -4.8657e-01, ..., -5.4261e-01,
-4.6946e-01, -8.1723e-01],
...,
[-1.4279e-01, -2.4550e+00, -8.3963e+00, ..., -8.2695e-01,
-2.7525e+00, -5.0617e-01],
[-3.4010e-01, -5.2986e-01, -3.8888e+00, ..., 1.7307e+00,
1.8031e+00, 3.0804e-01],
[-2.7676e-01, -8.3052e-01, -1.1644e+00, ..., -3.3344e-01,
3.3302e+00, -1.4022e-01]],
[[-3.3009e-01, -1.6695e+00, -8.7729e-01, ..., -9.5442e-01,
-1.6188e+00, -3.9627e-01],
[-2.2280e+00, -4.2745e+00, -2.7625e+00, ..., -2.6761e+00,
-3.6811e+00, -1.2843e+00],
[-4.9344e+00, -7.4049e+00, -9.3120e+00, ..., -2.9430e+00,
-3.3305e+00, -1.4352e+00],
...,
[-3.6599e+00, -8.5315e+00, -9.6801e+00, ..., -1.8173e+01,
-1.2373e+01, -5.2651e+00],
[-6.0418e+00, -1.0002e+01, -1.1007e+01, ..., -1.1593e+01,
-8.3728e+00, -7.9986e+00],
[ 4.9443e-01, 2.7798e-01, 6.0777e-01, ..., 8.1493e+00,
4.4949e+00, -6.5904e-01]],
[[-4.9466e-01, -7.9888e-02, 4.5644e-01, ..., 4.5447e-01,
1.4815e-01, 9.4773e-01],
[-4.3140e-01, -4.2817e-01, 4.8037e-01, ..., 4.0548e-01,
3.0641e-01, 5.0507e-01],
[-1.2149e+00, -2.6702e-01, 1.6160e-01, ..., 7.6135e-02,
1.6362e-01, 3.8900e-01],
...,
[-2.2561e+00, -1.2654e+00, -5.3861e+00, ..., -1.0620e+01,
-5.9578e+00, -3.5106e+00],
[-9.2171e-01, 2.6478e-01, -8.8312e+00, ..., 9.6865e-02,
-2.5545e+00, -2.3811e+00],
[-6.5187e-01, 3.8140e-01, -8.1520e+00, ..., 1.8447e+00,
-1.1534e+00, -1.3528e+00]],
...,
[[ 1.1307e-01, -4.9516e-01, -2.9820e-01, ..., -1.9127e-02,
-1.6702e-01, -7.1730e-01],
[-8.2324e-01, -5.7601e-01, -5.1502e-01, ..., -2.6431e-01,
-9.8397e-01, -7.8427e-01],
[-3.2055e-01, 1.7128e-01, 2.9458e-01, ..., 4.2231e-02,
-1.7232e-01, -5.3704e-01],
...,
[-3.1805e-02, 6.6275e-01, -2.7277e+00, ..., -7.1388e+00,
-5.0411e+00, -7.3093e+00],
[-3.1017e+00, -2.3031e+00, -5.3431e+00, ..., -8.9555e+00,
-6.0976e+00, -9.3927e+00],
[-5.6946e-01, -2.9440e-01, 1.6874e+00, ..., -1.6004e+00,
-6.9392e+00, -7.8313e+00]],
[[-1.5979e-01, 3.1726e-02, 3.9569e-01, ..., 9.0593e-02,
5.2736e-01, 1.2634e-01],
[-3.9302e-01, -2.8073e-01, 2.0025e-02, ..., -3.2568e-02,
1.7619e-01, -1.1882e-01],
[ 8.4745e-02, -2.6353e-01, -4.4492e-01, ..., 6.6893e-02,
5.2092e-02, -1.0735e-01],
...,
[-3.8224e-01, -1.4256e+00, -6.6341e+00, ..., -3.3609e+00,
-2.5500e+00, 1.1608e-02],
[-5.4942e-01, -2.1721e+00, -6.8732e+00, ..., -2.7473e+00,
-1.8028e+00, -2.4805e-01],
[ 7.7598e-01, -4.8547e-01, -2.7948e+00, ..., -1.4070e+00,
2.0505e+00, 1.0117e+00]],
[[-6.2252e-01, 3.9926e-01, 5.5027e-01, ..., 4.8782e-01,
3.8703e-01, 6.6100e-01],
[-8.5060e-01, 1.8990e-01, -2.4086e-01, ..., -2.2672e-01,
-1.2305e-01, 7.2519e-01],
[-8.0120e-01, 1.5288e-01, -7.2028e-02, ..., 1.2570e-01,
1.2961e-01, 6.5235e-01],
...,
[-1.2838e+00, -2.7595e+00, -5.9491e+00, ..., -1.2940e+01,
-8.3891e+00, 1.5540e+00],
[-1.6728e+00, -1.8351e+00, -1.1179e-01, ..., -1.2314e+01,
-9.3296e+00, 1.1230e+00],
[-7.3859e-01, -1.3645e+00, 1.3764e+00, ..., -6.6169e+00,
-6.0903e+00, 2.1809e+00]]],
[[[-3.1209e-01, -9.8446e-01, -9.0019e-01, ..., -2.6573e-01,
7.3468e-01, 4.9959e-01],
[-7.7686e-01, -2.4019e+00, -1.7083e+00, ..., -1.6230e-01,
5.2968e-01, 1.1775e-01],
[-1.3502e+00, -6.4545e-01, 1.1094e+00, ..., 3.0347e-01,
6.7549e-01, -9.1895e-02],
...,
[-5.8496e-02, -5.2817e-01, -2.7349e-01, ..., -3.8206e-01,
-1.2968e-01, -1.6116e-01],
[ 3.1370e-01, -6.9905e-01, -3.3600e-01, ..., -1.9869e-03,
2.8159e-01, -5.5634e-02],
[-1.4972e-01, -1.0909e+00, -9.6537e-01, ..., -1.2112e+00,
-1.2901e+00, -4.7420e-01]],
[[-2.0567e+00, -4.1309e+00, -3.7539e+00, ..., -1.3738e+00,
-2.0602e+00, -4.2099e-02],
[-6.6762e+00, -1.1729e+01, -1.1495e+01, ..., -2.8939e+00,
-4.0244e+00, -6.4849e-01],
[-3.3315e+00, -6.6663e+00, -8.5752e+00, ..., -2.7102e+00,
-3.5270e+00, -3.8707e-01],
...,
[-2.8466e+00, -4.3125e+00, -4.8699e+00, ..., -1.3799e+00,
-1.6889e+00, -4.4207e-01],
[-4.9966e+00, -6.0479e+00, -5.6313e+00, ..., -2.7749e+00,
-3.6963e+00, -1.0723e+00],
[ 7.2337e-01, -1.3047e-01, 5.3730e-01, ..., 3.6233e-01,
3.5886e-01, 3.9247e-02]],
[[-5.9305e-01, 3.6882e-01, 8.1920e-01, ..., 1.1749e+00,
5.5968e-01, 1.0328e+00],
[-8.2134e-01, 6.2654e-01, 1.2141e+00, ..., 4.8741e-01,
7.4469e-01, 4.6831e-01],
[-3.3900e+00, -2.9681e+00, -2.5388e+00, ..., -1.5429e+00,
7.2836e-01, 4.2957e-01],
...,
[-9.3888e-01, 4.1596e-01, -9.0774e-03, ..., -2.2232e-01,
-1.3671e-01, 4.4548e-01],
[-1.3797e+00, 3.3171e-01, -3.3657e-02, ..., 3.6780e-02,
5.1970e-02, 6.0600e-01],
[-5.5619e-01, 3.2984e-01, 1.5892e-01, ..., -1.3179e-01,
5.7453e-02, 3.9972e-01]],
...,
[[ 6.3861e-01, 5.0585e-02, 2.5477e-01, ..., 1.2533e-01,
-1.4386e-01, -8.6589e-01],
[-7.1933e-01, -4.8572e-01, -4.6010e-02, ..., -1.8080e+00,
-1.8157e+00, -7.9156e-01],
[ 5.1586e-01, -1.3967e+00, -1.8969e+00, ..., -3.6359e+00,
-1.6673e+00, -3.8872e-01],
...,
[ 3.4412e-02, -3.3154e-01, -8.3020e-02, ..., -2.9681e-01,
-3.7602e-01, -5.0953e-01],
[-2.3754e+00, -1.9678e+00, -1.4144e+00, ..., -1.0908e+00,
-1.1708e+00, -9.3884e-01],
[-4.5351e-01, -1.0157e+00, -4.0506e-01, ..., -6.9414e-02,
-1.7565e-01, -3.8167e-01]],
[[ 1.5287e-02, -2.3805e-01, -4.2724e-02, ..., -5.5489e-02,
2.4855e-01, 2.7233e-01],
[ 1.8834e-01, -6.2706e-01, -1.1269e-01, ..., 4.1233e-01,
3.3336e-01, -3.4206e-02],
[-1.0084e+00, -1.2768e+00, 5.1521e-01, ..., 5.4652e-01,
5.5235e-01, 1.7076e-01],
...,
[ 5.7235e-01, -3.5756e-01, 5.8004e-01, ..., -5.7477e-02,
-3.3590e-01, -4.2496e-02],
[-1.8346e-01, -1.8640e+00, -2.8430e-01, ..., -4.9495e-01,
-4.7675e-01, -1.7390e-01],
[ 7.3936e-01, -1.4751e-01, 2.5285e-01, ..., -3.2651e-01,
-3.0851e-01, 2.7761e-01]],
[[-6.6018e-01, -1.4867e-01, 4.5566e-02, ..., 7.5395e-01,
3.8580e-01, 5.0769e-01],
[-1.5107e+00, -1.2542e+00, -1.3638e+00, ..., -2.6751e-01,
-9.8244e-02, 3.8634e-01],
[-2.6220e+00, -4.8776e+00, -4.7837e+00, ..., 2.0954e-01,
3.0558e-01, 7.9930e-02],
...,
[-1.1298e+00, -1.5753e+00, -8.8451e-01, ..., 7.9632e-02,
-1.2777e-01, 1.0194e-02],
[-4.9344e-01, -8.8248e-01, -3.7870e-01, ..., -1.1922e-01,
-2.7698e-01, -2.3065e-01],
[-4.5340e-01, -9.1636e-01, -4.1205e-01, ..., -1.7069e-01,
-3.0399e-01, 1.3553e-01]]],
[[[-2.2967e+00, -7.9551e-01, 2.8195e-01, ..., -9.9693e-01,
-5.1246e-01, -1.4698e+00],
[-1.7806e+00, -1.4725e+00, -1.5642e+00, ..., -1.2158e+00,
-1.2759e+00, -1.5144e+00],
[-1.1858e-01, -8.6772e-01, -2.0478e+00, ..., -9.1278e-01,
-1.0868e+00, -1.8279e+00],
...,
[-1.8107e-01, -1.3055e+00, -4.9481e-01, ..., -9.1712e-01,
-1.0638e+00, -1.9451e+00],
[-2.2747e-01, -9.8638e-01, -1.2511e-01, ..., -5.1064e-01,
-5.9235e-01, -1.7248e+00],
[-2.4605e-01, -1.4123e+00, -1.0769e+00, ..., -1.1768e+00,
-1.5734e+00, -7.9004e-01]],
[[-3.2974e+00, -5.5513e+00, -2.7326e+00, ..., -3.9311e+00,
-5.4911e+00, -1.8344e+00],
[-7.0567e+00, -1.1245e+01, -4.6444e+00, ..., -6.0886e+00,
-8.2172e+00, -3.4246e+00],
[-4.4490e+00, -1.0676e+01, -3.7073e+00, ..., -5.9733e+00,
-6.3082e+00, -2.5663e+00],
...,
[-1.7161e+00, -2.8229e+00, -1.6422e+00, ..., -5.0435e+00,
-6.3545e+00, -2.4639e+00],
[-3.5997e+00, -4.3412e+00, -2.7791e+00, ..., -6.4743e+00,
-7.9495e+00, -2.5147e+00],
[ 7.0807e-01, -5.4980e-01, 4.7441e-01, ..., 3.9266e-01,
4.5657e-01, 2.5152e-01]],
[[-4.1324e+00, -2.7140e+00, 9.1964e-02, ..., 6.1859e-01,
2.1862e-01, 1.5124e+00],
[-1.0020e+01, -8.9870e+00, -4.5601e+00, ..., 5.9895e-01,
-9.0840e-02, 1.0747e-01],
[-1.1453e+01, -6.7480e+00, -5.7748e+00, ..., 3.6713e-01,
-5.1097e-02, -1.0984e-01],
...,
[-1.4491e+00, 3.3115e-01, 1.8578e-01, ..., 2.7552e-04,
-1.0721e-01, -1.9584e-01],
[-1.3935e+00, 3.7829e-01, 2.2321e-01, ..., -5.9021e-02,
-6.2551e-03, -2.2485e-01],
[-8.7706e-01, 1.0068e-01, -6.0798e-02, ..., 6.6327e-02,
2.2868e-01, -5.7381e-01]],
...,
[[ 4.4013e-02, -1.0512e+00, -1.7829e+00, ..., 4.3552e-01,
3.5987e-01, -1.3220e+00],
[-4.9934e+00, -5.9795e+00, -5.1549e+00, ..., -1.4676e-01,
-7.5817e-01, -1.3914e+00],
[-4.6809e+00, -4.7919e+00, -3.3023e+00, ..., -3.4887e-03,
-9.9368e-02, -9.6444e-01],
...,
[-9.6516e-01, -6.4840e-01, -1.3645e-01, ..., -1.0922e-01,
-9.9047e-02, -9.5829e-01],
[-2.2993e+00, -1.5687e+00, -8.3209e-01, ..., -1.5544e+00,
-1.5823e+00, -1.8676e+00],
[-6.0431e-01, -9.6720e-01, -1.1312e-01, ..., -3.9370e-01,
-5.4676e-01, -1.2024e+00]],
[[-2.4506e+00, -1.4480e+00, 9.3262e-01, ..., -7.5774e-02,
3.7084e-01, -3.5540e-01],
[-6.5705e+00, -5.6039e+00, -2.4947e+00, ..., 9.7345e-02,
1.0098e-01, -8.2752e-01],
[-7.0046e+00, -4.4606e+00, -1.9412e+00, ..., -2.0133e-01,
1.6134e-01, -4.4002e-01],
...,
[-5.4445e-01, -1.0816e+00, -8.8285e-02, ..., 2.5106e-01,
1.1709e-01, -4.8320e-01],
[-9.3598e-01, -1.8686e+00, -3.8438e-01, ..., -1.6817e-01,
-2.7959e-01, -8.9176e-01],
[ 2.9856e-01, -4.7327e-01, -2.3015e-01, ..., 3.1046e-01,
3.1979e-01, 3.0729e-01]],
[[-3.7055e+00, -4.4300e+00, -2.3489e+00, ..., -1.3927e-01,
-1.1049e-01, 5.7481e-01],
[-4.8638e+00, -9.1046e+00, -6.8042e+00, ..., -9.2862e-01,
-4.3173e-01, 9.1678e-01],
[-4.2041e+00, -9.6779e+00, -7.2400e+00, ..., -6.5637e-01,
-5.7950e-01, 6.2084e-01],
...,
[ 9.0497e-02, -4.2152e-02, 6.0793e-02, ..., -6.0812e-01,
-5.9147e-01, 6.1285e-01],
[-2.7821e-01, -3.4711e-01, -1.1468e-01, ..., -7.6931e-01,
-7.7497e-01, 3.7451e-01],
[ 7.9191e-02, -4.2108e-01, -9.8309e-02, ..., -4.6959e-01,
-4.6170e-01, 1.0092e+00]]]], device='cuda:0'),
tensor([[[[-6.9818e-01, -1.4120e+00, -1.4129e+00, ..., -3.1478e-02,
6.2897e-01, 5.6862e-01],
[-2.4203e-01, -2.4125e+00, -2.2017e+00, ..., -4.9676e-01,
-2.4618e-01, 1.4010e-01],
[-3.8318e-01, -2.1329e+00, -1.9067e+00, ..., -9.0033e-01,
-5.0877e-01, -7.1081e-01],
...,
[-4.7803e-01, -1.9077e+00, -1.8329e+00, ..., -3.5212e-01,
-6.7416e-02, -1.1144e-01],
[-9.8107e-01, -1.7144e+00, -1.1359e+00, ..., 1.8838e-02,
2.7060e-01, -1.0703e-02],
[-8.2397e-01, -1.6642e+00, -1.2444e+00, ..., -1.0508e+00,
-1.0813e+00, -3.7444e-01]],
[[-2.3852e+00, -4.8411e+00, -4.2073e+00, ..., -4.7494e-01,
-9.9228e-01, 1.9049e-02],
[-5.4269e+00, -8.1810e+00, -6.6115e+00, ..., -1.8260e+00,
-2.6802e+00, -6.7300e-01],
[-4.3930e+00, -6.3664e+00, -5.5524e+00, ..., -4.3911e+00,
-5.0537e+00, -2.2233e+00],
...,
[-4.3541e+00, -6.1908e+00, -5.6081e+00, ..., -1.3373e+00,
-1.5386e+00, -3.4490e-01],
[-6.3696e+00, -7.3819e+00, -6.3958e+00, ..., -2.4655e+00,
-3.3575e+00, -9.4130e-01],
[-1.8597e-02, -8.3702e-01, 3.8558e-01, ..., 3.2641e-01,
3.3283e-01, 7.9706e-02]],
[[-6.1951e-01, 5.8323e-01, 1.1379e+00, ..., 8.2230e-01,
2.7859e-01, 1.0955e+00],
[-9.0223e-01, -7.4191e-02, 4.3113e-01, ..., 4.2499e-01,
4.3502e-02, 4.8231e-01],
[-1.6235e+00, -1.7368e-01, -6.7289e-02, ..., 4.2816e-01,
5.7265e-01, 7.8474e-01],
...,
[-1.7659e+00, -7.2543e-02, -1.4391e-01, ..., 5.1705e-02,
9.9575e-02, 4.2202e-01],
[-1.5852e+00, 5.8234e-02, 6.6648e-03, ..., 1.2585e-01,
1.3337e-01, 5.9986e-01],
[-1.4495e+00, -5.2170e-01, -2.9635e-01, ..., -9.6083e-02,
8.0860e-02, 3.5021e-01]],
...,
[[ 1.0435e+00, -4.0368e-01, -2.3375e-01, ..., -1.2025e-01,
-3.9345e-01, -9.3121e-01],
[-8.0512e-01, -9.4709e-01, -8.4329e-01, ..., -6.8192e-01,
-1.3085e+00, -8.4087e-01],
[-6.4905e-01, -8.9392e-01, -7.1427e-01, ..., -2.5757e-01,
-5.8764e-01, -8.3438e-01],
...,
[-6.9862e-01, -1.0863e+00, -7.4912e-01, ..., -1.3098e-01,
-1.7909e-01, -3.5652e-01],
[-2.7849e+00, -2.8166e+00, -1.8273e+00, ..., -8.1972e-01,
-8.5702e-01, -7.8022e-01],
[-6.6331e-01, -1.6958e+00, -1.0292e+00, ..., -2.8957e-02,
-1.3964e-01, -3.2197e-01]],
[[ 1.4071e-02, -1.3361e-02, 6.0140e-01, ..., 3.0850e-01,
6.8580e-01, 3.2116e-01],
[-3.3349e-02, -7.2409e-01, 1.1828e-01, ..., -2.2395e-02,
1.1070e-01, -3.9549e-02],
[-9.2667e-02, -7.7035e-01, 2.7795e-01, ..., 4.4929e-02,
-6.9669e-02, -3.4419e-01],
...,
[ 1.3677e-01, -7.4068e-01, 3.0542e-01, ..., -1.8818e-01,
-1.1452e-01, 1.7119e-01],
[-1.9123e-01, -1.4110e+00, 4.9129e-02, ..., -4.2236e-01,
-3.4336e-01, -5.5810e-02],
[ 6.6467e-01, 1.4821e-01, 2.7300e-01, ..., -2.7551e-01,
-2.1953e-01, 3.0868e-01]],
[[-1.6147e+00, -5.8417e-01, 7.3173e-02, ..., 5.6064e-01,
3.5008e-01, 5.5080e-01],
[-2.1176e+00, -2.9695e-01, 5.3493e-02, ..., -3.4520e-02,
-1.5115e-01, 4.2415e-01],
[-1.7779e+00, -3.6072e-01, 2.7162e-01, ..., -4.3750e-01,
-8.1124e-01, -4.2652e-01],
...,
[-1.7670e+00, -4.0605e-01, 2.5536e-01, ..., 8.0417e-02,
-1.0317e-01, 1.0254e-01],
[-2.2503e+00, -1.0984e+00, -1.9629e-01, ..., -1.1714e-01,
-2.7406e-01, -1.9078e-01],
[-1.0656e+00, -2.0069e-01, 4.2713e-01, ..., -1.2795e-01,
-2.7960e-01, 1.1683e-01]]],
[[[ 1.4742e-02, -2.0653e-01, -1.7911e-02, ..., -9.5369e-01,
-5.6052e-01, -1.4369e+00],
[ 3.1240e-01, -1.0948e+00, -7.5779e-01, ..., -1.1436e+00,
-1.3888e+00, -1.5080e+00],
[ 2.1018e-01, -7.3158e-01, -5.6169e-01, ..., -8.9995e-01,
-1.1055e+00, -1.8283e+00],
...,
[ 4.2753e-02, -5.7488e-01, -4.3625e-01, ..., -9.6863e-01,
-1.0237e+00, -1.9242e+00],
[-4.4212e-01, -5.5425e-01, -2.8893e-02, ..., -4.5642e-01,
-5.6597e-01, -1.7121e+00],
[-5.9422e-01, -1.2740e+00, -1.0261e+00, ..., -1.1514e+00,
-1.5653e+00, -7.7141e-01]],
[[-7.0905e-01, -2.1850e+00, -1.7466e+00, ..., -4.0412e+00,
-5.4636e+00, -1.8590e+00],
[-2.2946e+00, -4.2572e+00, -3.0949e+00, ..., -6.0072e+00,
-8.0777e+00, -3.4308e+00],
[-1.6763e+00, -3.0164e+00, -2.5873e+00, ..., -5.0870e+00,
-6.2762e+00, -2.6699e+00],
...,
[-1.8068e+00, -2.4029e+00, -2.4541e+00, ..., -4.9958e+00,
-6.2754e+00, -2.4917e+00],
[-3.4758e+00, -4.1133e+00, -3.4074e+00, ..., -6.4190e+00,
-7.9029e+00, -2.5732e+00],
[ 3.1329e-01, -3.1260e-01, 3.7010e-01, ..., 4.4211e-01,
4.5179e-01, 1.7932e-01]],
[[-3.1072e-01, 3.4147e-02, 7.5752e-01, ..., 5.9841e-01,
2.7471e-01, 1.5784e+00],
[ 4.6094e-02, -4.4951e-01, 6.5539e-01, ..., 5.4178e-01,
-9.9024e-02, 1.4132e-01],
[-2.8986e-01, -2.4874e-01, 2.6377e-01, ..., 1.1194e-02,
-8.5181e-02, -1.6694e-01],
...,
[-2.4817e-01, -1.1616e-01, 1.0954e-01, ..., -1.7236e-02,
-9.2129e-02, -1.4620e-01],
[-1.5620e-01, -2.1515e-02, 1.5319e-01, ..., -1.8322e-01,
-6.1744e-02, -1.8536e-01],
[-1.0802e-01, -4.3300e-01, 8.0463e-02, ..., 5.3421e-02,
1.6455e-01, -6.0069e-01]],
...,
[[ 4.1893e-01, -4.4482e-01, -4.1038e-02, ..., 5.0144e-01,
3.8076e-01, -1.3668e+00],
[-7.2179e-01, -5.7381e-01, -6.0392e-01, ..., -1.2713e-01,
-7.5138e-01, -1.4953e+00],
[-4.4651e-01, -2.0567e-01, -2.8711e-01, ..., -4.2862e-02,
-4.5988e-02, -9.9637e-01],
...,
[-4.0350e-01, -1.5445e-01, -1.5311e-01, ..., -7.4088e-02,
-1.2180e-01, -9.8479e-01],
[-2.0681e+00, -1.5091e+00, -1.1193e+00, ..., -1.6337e+00,
-1.5833e+00, -1.8226e+00],
[-5.0694e-01, -6.5930e-01, -4.2709e-01, ..., -3.9365e-01,
-5.6675e-01, -1.1925e+00]],
[[ 9.5252e-02, 2.5474e-01, 6.8432e-01, ..., -9.7873e-02,
4.0506e-01, -3.6034e-01],
[ 5.7494e-03, 2.5616e-01, 3.9254e-01, ..., 2.2951e-01,
9.2672e-02, -8.5702e-01],
[-2.3719e-01, -1.6755e-02, 1.8306e-01, ..., 3.1138e-01,
1.4116e-01, -4.7715e-01],
...,
[-1.0584e-01, -3.1351e-02, 3.2042e-01, ..., 1.0842e-01,
1.9068e-01, -5.0310e-01],
[-3.5123e-01, -5.8059e-01, 3.9075e-03, ..., -1.8388e-01,
-2.5569e-01, -9.2083e-01],
[ 5.4085e-01, 4.1106e-01, 5.7447e-02, ..., 3.0639e-01,
3.0731e-01, 3.1039e-01]],
[[-9.6749e-01, 2.7094e-01, 3.7580e-01, ..., -1.1465e-01,
-6.8808e-02, 5.9566e-01],
[-1.1388e+00, 2.7383e-01, -3.7859e-01, ..., -9.6499e-01,
-4.3539e-01, 9.0014e-01],
[-8.6690e-01, 3.9409e-01, -1.2840e-01, ..., -6.3465e-01,
-5.3612e-01, 6.2141e-01],
...,
[-9.8235e-01, 2.4279e-01, -1.0225e-01, ..., -6.2249e-01,
-4.5635e-01, 6.2834e-01],
[-1.5004e+00, -2.3872e-01, -2.9133e-01, ..., -7.5569e-01,
-6.6092e-01, 4.1258e-01],
[-5.8142e-01, -5.3853e-03, -2.5747e-01, ..., -5.3060e-01,
-4.5423e-01, 1.0167e+00]]],
[[[-3.8827e-01, -1.1190e+00, -9.9862e-01, ..., -3.6735e-01,
3.6521e-01, 4.3365e-01],
[ 1.0515e-01, -1.8458e+00, -1.4220e+00, ..., -8.4295e-01,
-5.9138e-01, 7.1984e-02],
[-3.7902e+00, -4.0831e+00, -2.4565e+00, ..., -1.7020e+00,
-7.8237e-01, -1.8485e-01],
...,
[-3.3853e+00, -5.7662e+00, -5.9800e+00, ..., -4.5710e-01,
-1.9204e-01, -2.2112e-01],
[-4.8439e-01, -3.3184e+00, -1.9348e+00, ..., -3.8415e-02,
2.0758e-01, -6.9106e-02],
[ 1.1368e-01, -1.4407e+00, -1.5389e+00, ..., -1.0521e+00,
-1.1280e+00, -3.8062e-01]],
[[-2.2126e+00, -4.0332e+00, -3.7398e+00, ..., -1.5729e+00,
-1.7124e+00, -1.4208e-01],
[-3.9102e+00, -6.0624e+00, -5.5866e+00, ..., -4.0780e+00,
-4.3793e+00, -8.1710e-01],
[-7.6088e+00, -1.3811e+01, -1.0867e+01, ..., -5.3619e+00,
-4.7142e+00, -5.3431e-01],
...,
[-1.1009e+01, -1.1376e+01, -1.1067e+01, ..., -1.7044e+00,
-1.8893e+00, -5.0613e-01],
[-6.5228e+00, -9.8189e+00, -7.6964e+00, ..., -2.8562e+00,
-3.6570e+00, -1.0999e+00],
[ 7.6846e-01, -6.9943e-02, 7.9651e-01, ..., 3.8934e-01,
3.7874e-01, 7.3469e-02]],
[[-5.6331e-01, 4.0399e-01, 6.9171e-01, ..., 1.0291e+00,
5.1682e-01, 1.1946e+00],
[-8.7716e-01, 1.9315e-01, 5.5566e-01, ..., 1.3015e-01,
2.3477e-01, 4.8710e-01],
[-1.8456e+00, -9.9274e-01, 4.4903e-01, ..., -1.9618e+00,
-7.8652e-02, 4.5567e-01],
...,
[-1.0927e+01, -1.0270e+01, -4.7685e+00, ..., 1.1789e-01,
1.2368e-01, 4.6449e-01],
[-3.2098e+00, -1.9260e+00, 2.3868e+00, ..., 1.4815e-01,
1.8461e-01, 6.3751e-01],
[-6.9981e-02, 5.0787e-02, 8.6133e-01, ..., -1.4738e-01,
9.4564e-02, 3.8219e-01]],
...,
[[ 7.5513e-01, 1.2623e-01, 3.6285e-01, ..., 2.9998e-02,
-2.7539e-01, -9.4567e-01],
[-6.2567e-01, -2.6611e-01, -1.9999e-03, ..., -9.7128e-01,
-1.2611e+00, -8.8821e-01],
[-3.2585e-02, 1.0039e+00, 9.7948e-01, ..., -3.2572e+00,
-1.4436e+00, -4.4022e-01],
...,
[ 2.6257e+00, -2.3100e+00, -4.7654e+00, ..., -1.9069e-01,
-2.5881e-01, -4.0688e-01],
[-2.3681e+00, -5.0346e+00, -7.0401e+00, ..., -8.6655e-01,
-9.1823e-01, -8.3079e-01],
[-2.5813e+00, -3.4355e+00, -2.6728e+00, ..., -4.2667e-02,
-1.6821e-01, -3.7124e-01]],
[[-3.4134e-02, -3.2011e-01, -1.2138e-01, ..., -2.0508e-01,
4.6970e-01, 3.3523e-01],
[ 1.2466e-01, -1.0075e+00, -9.8846e-02, ..., -6.3778e-01,
-7.5729e-01, -3.5865e-02],
[-7.0724e-01, -1.9153e+00, -5.9369e-01, ..., -1.2338e+00,
-1.0544e+00, 1.5274e-01],
...,
[-2.6048e+00, -3.2012e+00, -1.5898e+00, ..., -1.7190e-01,
-1.6093e-01, 1.1366e-01],
[-1.5977e+00, -4.1541e+00, -2.9947e-01, ..., -3.9131e-01,
-4.2378e-01, -1.1903e-01],
[ 1.1546e+00, -1.8003e+00, -1.6180e+00, ..., -1.8693e-01,
-2.4923e-01, 2.9342e-01]],
[[-6.1854e-01, -2.4870e-01, -1.3442e-01, ..., 7.0745e-01,
4.7538e-01, 6.2870e-01],
[-1.1549e+00, -1.0249e+00, -1.0097e+00, ..., 1.2765e-01,
4.6942e-01, 5.3450e-01],
[-2.0676e+00, -1.8000e+00, -1.1306e+00, ..., -6.5552e-02,
7.4596e-01, 1.5783e-01],
...,
[-4.9869e+00, -7.6672e+00, -1.1621e+01, ..., 1.0227e-01,
-5.5734e-02, 1.7512e-01],
[-1.4118e+00, -1.8938e+00, -4.3830e+00, ..., -1.9230e-02,
-2.3381e-01, -1.3327e-01],
[-3.1754e-01, -1.5371e-01, 2.8198e-01, ..., -1.1579e-01,
-2.8361e-01, 1.8423e-01]]],
[[[-3.3609e-01, -4.7519e-01, -1.4271e-01, ..., -9.7491e-01,
-5.0617e-01, -1.4235e+00],
[-1.9440e-02, -1.4312e+00, -7.1937e-01, ..., -1.1433e+00,
-1.2606e+00, -1.4695e+00],
[-4.6023e-01, -1.5356e+00, -4.6155e-01, ..., -9.1657e-01,
-1.0828e+00, -1.8109e+00],
...,
[-2.9178e-01, -1.2964e+00, -4.8789e-01, ..., -9.1777e-01,
-1.0312e+00, -1.9199e+00],
[-5.3118e-01, -1.1288e+00, -1.0142e-01, ..., -4.4405e-01,
-5.3659e-01, -1.6923e+00],
[-6.2236e-01, -1.6067e+00, -1.2580e+00, ..., -1.1137e+00,
-1.5411e+00, -7.7326e-01]],
[[-6.7488e-01, -1.9385e+00, -7.7638e-01, ..., -3.9243e+00,
-5.4394e+00, -1.8366e+00],
[-2.1920e+00, -3.9590e+00, -1.9865e+00, ..., -5.8869e+00,
-8.1109e+00, -3.4374e+00],
[-2.5883e+00, -3.1983e+00, -1.9741e+00, ..., -5.0335e+00,
-6.2403e+00, -2.5797e+00],
...,
[-1.6866e+00, -2.8126e+00, -1.5163e+00, ..., -4.9948e+00,
-6.2844e+00, -2.4539e+00],
[-3.7718e+00, -4.6255e+00, -2.9105e+00, ..., -6.3400e+00,
-7.8575e+00, -2.4992e+00],
[ 5.4237e-01, -7.4225e-01, 3.9104e-01, ..., 4.4355e-01,
4.6711e-01, 2.2846e-01]],
[[-3.7644e-01, 7.3700e-01, 1.0123e+00, ..., 6.3982e-01,
2.8046e-01, 1.5483e+00],
[-7.8620e-01, 2.3104e-01, 5.0887e-01, ..., 6.1776e-01,
-4.3992e-02, 1.6952e-01],
[-1.3662e+00, 3.6844e-01, 3.1517e-01, ..., 2.9399e-04,
-1.8311e-02, -1.1266e-01],
...,
[-1.4375e+00, 3.1488e-01, 1.8148e-01, ..., 3.2037e-05,
-8.9075e-02, -1.4954e-01],
[-1.4076e+00, 2.9330e-01, 2.2822e-01, ..., -9.0877e-02,
4.7241e-03, -1.7203e-01],
[-9.0627e-01, 6.6730e-02, -1.5367e-01, ..., 7.7967e-02,
1.8791e-01, -5.8156e-01]],
...,
[[ 3.9260e-01, -1.4204e-01, -1.2570e-01, ..., 4.2458e-01,
3.1906e-01, -1.3055e+00],
[-1.4243e+00, -1.0834e+00, -6.3056e-01, ..., -1.3703e-01,
-7.8986e-01, -1.3650e+00],
[-7.3723e-01, -6.6119e-01, -2.4788e-01, ..., -1.3159e-01,
-1.0528e-01, -9.3416e-01],
...,
[-9.6295e-01, -5.6981e-01, -1.2931e-01, ..., -1.1110e-01,
-1.2064e-01, -9.4604e-01],
[-2.5865e+00, -1.7964e+00, -8.7895e-01, ..., -1.4966e+00,
-1.5712e+00, -1.8079e+00],
[-6.2869e-01, -1.1876e+00, -7.9199e-02, ..., -3.5854e-01,
-5.0648e-01, -1.1631e+00]],
[[ 1.9271e-02, 2.2816e-01, 4.8223e-01, ..., -1.2136e-01,
3.6892e-01, -3.5517e-01],
[-1.6779e-01, -6.9066e-01, -8.0199e-02, ..., 2.1463e-01,
1.3147e-01, -8.1741e-01],
[-6.4929e-01, -1.2074e+00, -2.3758e-01, ..., 2.5561e-01,
1.2299e-01, -4.5985e-01],
...,
[-5.8486e-01, -1.1460e+00, -1.4442e-01, ..., 2.5239e-01,
1.2263e-01, -5.0104e-01],
[-1.0530e+00, -1.9685e+00, -4.5302e-01, ..., -1.1467e-01,
-2.6109e-01, -8.8436e-01],
[ 2.6380e-01, -4.3358e-01, -2.3860e-01, ..., 3.6415e-01,
3.3297e-01, 2.9892e-01]],
[[-3.6821e-01, 1.1960e-01, 5.3010e-01, ..., -9.7721e-02,
-1.0590e-01, 5.8706e-01],
[-3.8299e-01, -3.8927e-02, 1.5810e-02, ..., -9.2298e-01,
-4.1481e-01, 9.1148e-01],
[-6.8647e-04, 1.2393e-01, 1.8662e-01, ..., -5.8229e-01,
-5.3191e-01, 6.3510e-01],
...,
[ 3.5621e-02, -8.3028e-02, 5.1251e-02, ..., -5.6169e-01,
-5.3682e-01, 5.9877e-01],
[-4.2506e-01, -5.4866e-01, -1.8682e-01, ..., -6.9496e-01,
-6.8470e-01, 3.9963e-01],
[ 2.1203e-02, -5.2079e-01, -2.1000e-01, ..., -4.8366e-01,
-4.4683e-01, 9.9276e-01]]],
[[[-3.4936e-01, -1.0117e+00, -8.7972e-01, ..., -1.1358e-02,
6.4274e-01, 5.4284e-01],
[ 1.3471e-01, -1.7595e+00, -1.2905e+00, ..., -4.9715e-01,
-2.5575e-01, 1.1552e-01],
[ 1.9212e-01, -1.3120e+00, -9.6821e-01, ..., -3.3704e-01,
-1.5787e-01, -9.4594e-02],
...,
[-3.5610e+00, -3.2123e+00, -1.7386e+00, ..., -1.0025e+00,
-1.2252e-01, -1.3184e-01],
[-3.1498e+00, -1.1155e+00, -6.3121e-01, ..., 1.3347e-02,
2.6330e-01, -9.6067e-03],
[-4.1893e-01, -2.2564e-01, -1.3110e+00, ..., -1.0548e+00,
-1.1076e+00, -3.9905e-01]],
[[-2.1456e+00, -4.1059e+00, -3.8338e+00, ..., -4.3895e-01,
-1.0484e+00, -1.5210e-02],
[-3.7693e+00, -6.0097e+00, -5.4148e+00, ..., -1.8520e+00,
-2.7375e+00, -6.9685e-01],
[-3.1040e+00, -4.6911e+00, -4.6245e+00, ..., -1.3567e+00,
-1.5397e+00, -3.9752e-01],
...,
[-6.2274e+00, -7.2179e+00, -8.5531e+00, ..., -1.6979e+00,
-1.6346e+00, -3.8108e-01],
[-6.3912e+00, -7.9201e+00, -8.2868e+00, ..., -2.4766e+00,
-3.3961e+00, -9.8502e-01],
[ 5.8436e-01, -9.8922e-01, -2.2082e-01, ..., 3.5075e-01,
3.5961e-01, 7.9237e-02]],
[[-5.6398e-01, 4.3923e-01, 7.5637e-01, ..., 8.4298e-01,
3.4104e-01, 1.1191e+00],
[-9.0302e-01, 1.3809e-01, 5.7795e-01, ..., 4.2038e-01,
3.8903e-02, 4.9104e-01],
[-1.2650e+00, 4.9722e-02, -8.1309e-03, ..., 7.7670e-02,
8.8293e-02, 4.2629e-01],
...,
[-3.2920e+00, -5.8657e+00, -3.7423e+00, ..., 7.9026e-01,
1.0215e-01, 4.3015e-01],
[-8.3626e-01, -3.9005e+00, -2.8541e+00, ..., 1.3359e-01,
1.4992e-01, 6.3180e-01],
[ 9.0247e-01, -2.9474e-01, -1.2200e+00, ..., -1.2588e-01,
6.6413e-02, 3.6313e-01]],
...,
[[ 7.3090e-01, 1.3935e-01, 3.3975e-01, ..., -1.0213e-01,
-3.9421e-01, -9.3806e-01],
[-6.5622e-01, -2.8468e-01, -1.1527e-02, ..., -7.0073e-01,
-1.2975e+00, -8.3969e-01],
[-4.8144e-01, -4.0428e-01, -2.3770e-02, ..., -1.1607e-01,
-2.1433e-01, -3.9954e-01],
...,
[ 1.3053e+00, -1.4419e+00, -5.5081e+00, ..., -3.5621e-01,
-1.7522e-01, -3.6696e-01],
[-1.6358e+00, -4.6760e+00, -6.1732e+00, ..., -8.1496e-01,
-8.8175e-01, -8.0557e-01],
[-5.6502e-01, -3.7195e+00, -2.1093e+00, ..., -4.3997e-02,
-1.4471e-01, -3.3405e-01]],
[[ 8.9217e-03, -2.5686e-01, -8.9352e-02, ..., 3.4466e-01,
7.1645e-01, 2.9100e-01],
[ 1.8843e-01, -8.8834e-01, -2.9148e-02, ..., -2.3823e-02,
1.1473e-01, -7.9621e-02],
[-1.2586e-01, -8.9981e-01, 7.5836e-02, ..., -1.5365e-01,
-9.6628e-02, 1.8324e-01],
...,
[-2.6289e+00, 3.4897e-01, 1.3640e+00, ..., -7.8542e-01,
-1.3542e-01, 1.5155e-01],
[-2.7761e+00, -1.9635e-02, 6.7953e-01, ..., -3.8812e-01,
-3.2160e-01, -6.6762e-02],
[ 3.3075e-01, 2.1476e+00, 4.3215e-01, ..., -2.6058e-01,
-2.3739e-01, 3.0436e-01]],
[[-5.8242e-01, -1.4518e-01, -1.0523e-02, ..., 5.6293e-01,
3.8461e-01, 5.6311e-01],
[-1.0712e+00, -8.9254e-01, -9.5208e-01, ..., -7.8758e-02,
-1.3538e-01, 4.1181e-01],
[-6.6060e-01, -7.8248e-01, -5.8932e-01, ..., 3.5811e-02,
-1.2905e-01, 9.4720e-02],
...,
[-1.0640e+01, -4.0260e+00, 1.2759e+00, ..., 3.0389e-01,
-2.6266e-02, 1.0844e-01],
[-7.3815e+00, -2.7657e+00, 1.9276e+00, ..., -5.5488e-02,
-2.4829e-01, -2.1091e-01],
[-2.1014e+00, -7.2115e-01, 1.0943e+00, ..., -1.1017e-01,
-2.6763e-01, 1.1420e-01]]]], device='cuda:0'),
tensor([[[[-1.3589e-01, -5.2527e-01, -2.3285e-01, ..., 3.1605e-01,
3.3221e-01, 1.4027e-02],
[ 2.7629e-01, -1.1161e+00, -6.8619e-01, ..., -8.9480e-01,
-3.7139e-01, -1.2156e-01],
[ 2.0832e-01, -7.6712e-01, -3.7101e-01, ..., -1.2350e+00,
-2.6284e-01, -5.3244e-01],
...,
[-2.7119e-01, -1.7358e+00, -1.2741e+00, ..., -1.3019e+00,
-9.9564e-01, -2.0502e+00],
[-1.1284e+00, -1.7622e+00, -8.0642e-01, ..., -5.2597e-01,
-4.4195e-01, -1.7970e+00],
[-1.3110e+00, -2.1916e+00, -1.6246e+00, ..., -1.4167e+00,
-1.5312e+00, -9.8839e-01]],
[[-1.4599e+00, -3.0164e+00, -2.5989e+00, ..., -4.2996e+00,
-3.4072e+00, -1.2481e+00],
[-2.9983e+00, -4.7281e+00, -3.5725e+00, ..., -5.3546e+00,
-4.9726e+00, -2.1228e+00],
[-2.0188e+00, -2.9257e+00, -2.2685e+00, ..., -3.5323e+00,
-2.6894e+00, -1.2285e+00],
...,
[-3.7494e+00, -5.4878e+00, -5.1765e+00, ..., -5.7543e+00,
-6.1913e+00, -2.9367e+00],
[-6.2051e+00, -7.0328e+00, -6.1272e+00, ..., -6.0284e+00,
-7.3851e+00, -2.6902e+00],
[-3.4270e-02, -9.0321e-01, 2.9347e-01, ..., 4.1286e-01,
7.0076e-01, 1.9279e-01]],
[[-4.0742e-01, 2.4209e-01, 8.2015e-01, ..., 8.2587e-01,
6.0408e-01, 1.3689e+00],
[-2.8908e-01, -3.8276e-02, 5.6149e-01, ..., 1.0591e+00,
4.4071e-01, 2.6841e-01],
[-6.9040e-01, 3.8746e-02, 2.8335e-01, ..., 7.8264e-01,
3.5606e-01, 4.3160e-01],
...,
[-1.3767e+00, -3.5439e-02, -3.0916e-02, ..., 3.0441e-01,
-9.5668e-03, 2.7751e-01],
[-1.1600e+00, 1.0066e-01, -4.7621e-02, ..., -4.3296e-02,
9.3510e-03, 2.9498e-01],
[-1.1418e+00, -6.7215e-02, 9.1020e-02, ..., -5.1501e-03,
-3.4412e-02, -3.5459e-01]],
...,
[[ 9.5441e-01, 2.7532e-01, 2.4686e-01, ..., -1.0578e+00,
-2.1031e-03, -9.7351e-01],
[-9.8190e-01, -8.6113e-01, -6.6370e-01, ..., -3.0327e+00,
-1.6884e+00, -1.2031e+00],
[-4.8265e-01, -3.4289e-01, -2.9013e-01, ..., -1.3277e+00,
-4.8908e-01, -6.3857e-01],
...,
[-6.1409e-01, -5.7293e-01, -3.1523e-01, ..., -2.6377e-01,
-5.7361e-01, -1.1718e+00],
[-3.1665e+00, -2.8971e+00, -2.0529e+00, ..., -1.8961e+00,
-1.9938e+00, -2.0646e+00],
[-5.3901e-01, -1.6275e+00, -1.1081e+00, ..., -8.5153e-01,
-9.2155e-01, -1.3060e+00]],
[[ 7.2978e-02, 3.0914e-01, 6.9030e-01, ..., 5.4126e-01,
1.1472e+00, 4.3607e-01],
[ 7.6097e-02, -3.7080e-01, 2.0889e-01, ..., 3.0977e-01,
5.5205e-01, -2.2155e-01],
[-2.1305e-01, -5.1217e-01, 1.7394e-01, ..., -1.7941e-01,
3.1466e-01, 1.7660e-02],
...,
[-1.5889e-01, -9.8283e-01, 1.1142e-01, ..., -3.6456e-02,
1.1410e-01, -7.3109e-01],
[-5.4399e-01, -1.8280e+00, -1.7314e-01, ..., 2.2887e-02,
-2.9687e-02, -9.2205e-01],
[ 4.2040e-01, -1.3522e-01, 1.2374e-01, ..., 2.8397e-01,
3.3186e-01, 3.5282e-01]],
[[-9.1598e-01, -2.3726e-01, 1.3318e-01, ..., 1.0029e-01,
1.0380e-01, 7.6986e-01],
[-1.1215e+00, -2.8151e-01, -3.1495e-01, ..., 4.1520e-01,
3.1924e-02, 1.2330e+00],
[-6.9337e-01, -1.9177e-01, -2.1286e-01, ..., 7.9041e-01,
-7.3119e-02, 6.7150e-01],
...,
[-1.1866e+00, -9.3677e-01, -4.2950e-01, ..., 3.2807e-02,
-7.3268e-02, 8.0614e-01],
[-1.9090e+00, -1.8377e+00, -9.9173e-01, ..., -7.8615e-01,
-4.7596e-01, 4.2074e-01],
[-9.7027e-01, -9.7641e-01, -2.8244e-01, ..., -2.8530e-01,
-1.5325e-01, 1.1381e+00]]],
[[[-2.2389e-01, -2.8022e-01, 2.0057e-01, ..., 2.2429e-02,
4.8382e-01, 1.3062e-01],
[-1.0787e-01, -1.4520e+00, -1.3635e+00, ..., -7.6670e-01,
-4.8718e-01, -3.7674e-01],
[-3.4201e-01, -3.3241e+00, -2.9632e+00, ..., -4.9790e-01,
-3.8466e-01, -7.2758e-01],
...,
[-6.8275e-02, -1.7233e+00, -1.5113e+00, ..., -1.3761e+00,
-1.3178e+00, -2.0140e+00],
[-1.2200e-01, -1.2794e+00, -7.8771e-01, ..., -9.3468e-02,
-9.9272e-02, -1.5054e+00],
[-6.2727e-02, -1.1320e+00, -1.0566e+00, ..., -1.9896e+00,
-1.5775e+00, -9.0508e-01]],
[[-2.5669e-01, -1.6599e+00, -3.0118e+00, ..., -6.8551e-01,
-1.4174e+00, -3.0618e-01],
[-2.1405e+00, -4.6337e+00, -6.5370e+00, ..., -2.5342e+00,
-3.4508e+00, -1.2042e+00],
[-4.4874e+00, -7.3747e+00, -1.1589e+01, ..., -2.7324e+00,
-3.1317e+00, -1.3386e+00],
...,
[-3.6926e+00, -5.1022e+00, -5.3399e+00, ..., -7.9461e+00,
-6.7992e+00, -2.9367e+00],
[-5.3074e+00, -6.2150e+00, -5.7744e+00, ..., -8.4268e+00,
-8.4798e+00, -2.9414e+00],
[ 5.4662e-01, -2.6634e-01, 5.5338e-01, ..., -3.3801e-01,
4.7064e-01, 5.9591e-02]],
[[-5.2432e-01, -1.4245e-01, -5.2860e-03, ..., 4.0072e-01,
9.8415e-02, 9.2207e-01],
[-3.6292e-01, -4.7565e-01, -2.0689e+00, ..., 3.7694e-01,
2.3793e-01, 4.8985e-01],
[-1.2247e+00, -8.0605e-01, -3.1270e+00, ..., 7.5010e-02,
1.9580e-01, 4.2278e-01],
...,
[-1.8761e+00, -3.4941e-01, -1.1137e-01, ..., 3.1375e-01,
-7.0654e-02, -1.9301e-01],
[-1.5912e+00, -2.5034e-01, -1.6024e-01, ..., 6.9599e-01,
2.8655e-01, -2.5572e-01],
[-6.7460e-01, -9.5524e-02, 2.4135e-02, ..., -1.4096e+00,
7.1935e-03, -4.9270e-01]],
...,
[[ 7.9853e-02, -5.8162e-01, -1.0037e-01, ..., -4.7713e-02,
-2.1421e-01, -7.3613e-01],
[-8.5895e-01, -5.8398e-01, 8.8484e-01, ..., -2.9782e-01,
-9.8830e-01, -7.5570e-01],
[-1.2738e-01, 7.2105e-01, 1.6519e+00, ..., 9.4936e-02,
-1.7457e-01, -5.4152e-01],
...,
[-9.0497e-01, -1.0285e+00, -3.3156e-01, ..., -4.7968e-01,
-6.0774e-01, -1.2551e+00],
[-2.3036e+00, -1.9512e+00, -1.3523e+00, ..., -3.8832e+00,
-2.3243e+00, -1.7226e+00],
[-5.2567e-01, -9.9060e-01, -1.6558e-01, ..., -3.2086e+00,
-1.5028e+00, -5.9368e-01]],
[[-1.7767e-01, 7.4708e-02, 1.0365e+00, ..., 1.0609e-01,
5.5153e-01, 1.5234e-01],
[-4.1880e-01, -2.0051e-01, 3.8330e-01, ..., -7.9957e-02,
9.1233e-02, -9.7799e-02],
[-2.8280e-02, -6.3318e-01, -5.3693e-01, ..., 9.4947e-03,
4.5580e-02, -6.6853e-02],
...,
[ 1.4636e-01, -6.5564e-01, 3.1296e-01, ..., -3.1083e-01,
2.1193e-02, -3.5252e-01],
[-5.2082e-01, -1.7814e+00, -9.5329e-02, ..., 2.1007e-01,
-4.5426e-01, -6.6528e-01],
[ 7.9269e-01, 7.9081e-02, 3.7851e-01, ..., -4.7054e-02,
3.9518e-01, 2.7141e-01]],
[[-6.6576e-01, 4.0552e-01, 3.8631e-01, ..., 4.9213e-01,
3.8148e-01, 6.2019e-01],
[-8.4970e-01, 2.0425e-01, -1.2421e+00, ..., -2.4018e-01,
-1.4954e-01, 6.8589e-01],
[-8.1452e-01, -5.1277e-01, -1.7608e+00, ..., 7.7142e-02,
1.0239e-01, 6.5279e-01],
...,
[-1.3181e+00, -5.5204e-01, -1.2439e-01, ..., -3.1767e-01,
1.5735e-02, 1.3986e+00],
[-1.2286e+00, -6.6254e-01, -1.2194e-01, ..., -1.4219e+00,
-4.2373e-02, 1.1769e+00],
[-5.5081e-01, -3.6689e-01, 1.6628e-01, ..., -4.8857e-01,
-1.2799e-01, 1.2761e+00]]],
[[[-1.9894e-01, -2.8318e-01, 1.9866e-01, ..., 5.0800e-02,
4.9864e-01, 1.6688e-01],
[-1.1567e-01, -1.5353e+00, -8.6716e-01, ..., -6.9776e-01,
-4.3092e-01, -3.0542e-01],
[-1.2020e-01, -1.3185e+00, -6.5913e-01, ..., -4.9895e-01,
-3.8724e-01, -6.9320e-01],
...,
[-2.5662e-01, -1.7168e+00, -1.2582e+00, ..., -9.5664e-01,
-9.5551e-01, -2.0193e+00],
[-1.0748e+00, -1.7053e+00, -7.9100e-01, ..., -5.5890e-01,
-4.8032e-01, -1.7813e+00],
[-1.2641e+00, -2.1313e+00, -1.5997e+00, ..., -1.4299e+00,
-1.5479e+00, -9.7786e-01]],
[[-2.0153e-01, -1.3960e+00, -5.6375e-01, ..., -6.0552e-01,
-1.3335e+00, -2.2983e-01],
[-2.0220e+00, -3.9932e+00, -2.3830e+00, ..., -2.3816e+00,
-3.3707e+00, -1.1471e+00],
[-1.8825e+00, -3.3703e+00, -2.5472e+00, ..., -2.5143e+00,
-3.0163e+00, -1.2745e+00],
...,
[-3.7344e+00, -5.5004e+00, -5.1287e+00, ..., -5.4541e+00,
-6.1154e+00, -2.8875e+00],
[-6.1688e+00, -7.0628e+00, -6.1268e+00, ..., -6.1378e+00,
-7.4389e+00, -2.7448e+00],
[-1.2257e-02, -8.7500e-01, 2.6195e-01, ..., 4.1456e-01,
7.0471e-01, 1.7379e-01]],
[[-5.1067e-01, -1.7273e-01, 3.3092e-01, ..., 3.4535e-01,
5.3505e-02, 8.7723e-01],
[-3.3614e-01, -4.5055e-01, 4.9937e-01, ..., 3.6312e-01,
2.6648e-01, 4.9837e-01],
[-8.3531e-01, -2.2140e-01, 2.2245e-01, ..., 3.3353e-02,
1.1029e-01, 4.2951e-01],
...,
[-1.3954e+00, 3.7643e-02, 3.9363e-02, ..., 5.5963e-02,
4.2408e-02, 3.2998e-01],
[-1.2617e+00, 1.3894e-01, 7.1073e-03, ..., -3.6945e-02,
-1.3487e-03, 3.1959e-01],
[-1.1981e+00, -9.8073e-02, 7.7581e-02, ..., 1.9017e-03,
-8.7356e-02, -3.3874e-01]],
...,
[[ 8.9752e-02, -6.0190e-01, -4.0704e-01, ..., -5.8255e-02,
-2.5918e-01, -7.4287e-01],
[-8.3660e-01, -6.4906e-01, -5.5236e-01, ..., -2.8686e-01,
-1.0308e+00, -7.3806e-01],
[-5.0305e-01, -1.4716e-01, -4.0745e-02, ..., 1.0264e-01,
-1.8312e-01, -4.8759e-01],
...,
[-6.0111e-01, -6.0334e-01, -3.4605e-01, ..., -3.3170e-01,
-5.8354e-01, -1.1984e+00],
[-3.0889e+00, -2.8872e+00, -2.0357e+00, ..., -1.8750e+00,
-2.0507e+00, -2.0783e+00],
[-5.3140e-01, -1.6332e+00, -1.0879e+00, ..., -8.3159e-01,
-9.2831e-01, -1.3485e+00]],
[[-1.7550e-01, 4.9808e-02, 4.4368e-01, ..., 1.0418e-01,
5.5925e-01, 1.6719e-01],
[-4.2342e-01, -2.6857e-01, -7.0215e-03, ..., -5.0189e-02,
1.4170e-01, -9.3840e-02],
[-6.4853e-01, -5.6618e-01, -5.1967e-02, ..., 1.0921e-02,
4.5027e-02, -6.5491e-02],
...,
[-1.2142e-01, -1.0252e+00, 2.3541e-02, ..., 2.7635e-01,
1.5908e-01, -7.3437e-01],
[-5.1805e-01, -1.8361e+00, -2.1258e-01, ..., 3.3764e-02,
-3.3436e-02, -9.5446e-01],
[ 4.5201e-01, -1.1831e-01, 1.1598e-01, ..., 2.8035e-01,
3.3181e-01, 3.2034e-01]],
[[-6.8260e-01, 4.2300e-01, 5.5434e-01, ..., 4.9125e-01,
3.9494e-01, 6.0867e-01],
[-8.6892e-01, 3.0382e-01, -2.4644e-01, ..., -2.7924e-01,
-1.5533e-01, 6.5628e-01],
[-4.7089e-01, 3.5812e-01, 1.5805e-02, ..., 5.2453e-02,
1.0418e-01, 6.0234e-01],
...,
[-1.0568e+00, -9.0912e-01, -4.3954e-01, ..., -4.0221e-01,
-3.2601e-02, 8.2624e-01],
[-1.8031e+00, -1.7898e+00, -9.4703e-01, ..., -8.0741e-01,
-4.3555e-01, 4.2701e-01],
[-9.7010e-01, -9.6100e-01, -2.5512e-01, ..., -3.1372e-01,
-1.3989e-01, 1.1215e+00]]],
[[[-7.4042e-01, -1.3999e+00, -1.3760e+00, ..., -2.2146e-01,
2.5528e-01, -3.7985e-01],
[-2.9998e-01, -2.4011e+00, -2.1406e+00, ..., -5.6738e-01,
-3.6219e-01, -5.6562e-01],
[-4.1720e-01, -2.1179e+00, -1.8505e+00, ..., -3.4099e-01,
-3.3625e-01, -1.0467e+00],
...,
[-4.8990e-01, -1.9563e+00, -1.8992e+00, ..., -1.6369e-01,
-1.0104e-01, -1.2714e+00],
[-9.3947e-01, -1.6874e+00, -1.2000e+00, ..., 1.7717e-02,
1.1573e-01, -9.5277e-01],
[-7.9573e-01, -1.6712e+00, -1.2673e+00, ..., -8.9716e-01,
-1.0953e+00, -6.5335e-01]],
[[-2.4001e+00, -4.8187e+00, -4.1605e+00, ..., -2.1717e+00,
-3.4738e+00, -1.2559e+00],
[-5.2778e+00, -7.9911e+00, -6.4936e+00, ..., -3.4441e+00,
-5.1654e+00, -2.1064e+00],
[-4.3053e+00, -6.2133e+00, -5.4240e+00, ..., -2.9913e+00,
-3.8486e+00, -1.7435e+00],
...,
[-4.4061e+00, -6.1684e+00, -5.5375e+00, ..., -3.1119e+00,
-3.9083e+00, -1.6621e+00],
[-6.4210e+00, -7.4716e+00, -6.3356e+00, ..., -3.9269e+00,
-5.4045e+00, -1.9854e+00],
[-3.5637e-03, -7.9963e-01, 4.1876e-01, ..., 4.3223e-01,
6.0437e-01, 1.6276e-01]],
[[-5.8528e-01, 5.8330e-01, 1.0778e+00, ..., 8.2584e-01,
5.5220e-01, 1.3712e+00],
[-9.2506e-01, -6.9939e-02, 4.1453e-01, ..., 4.3507e-01,
3.5899e-01, 2.4292e-01],
[-1.6612e+00, -1.5316e-01, -5.5012e-02, ..., 7.2608e-02,
2.7465e-01, 1.1496e-01],
...,
[-1.7094e+00, -6.2431e-02, -1.2094e-01, ..., 4.3047e-01,
1.4552e-02, -1.5254e-02],
[-1.5007e+00, 1.1934e-01, 2.8190e-02, ..., 1.3667e-01,
2.3813e-01, 1.4921e-01],
[-1.4210e+00, -4.9544e-01, -3.0380e-01, ..., 2.3523e-02,
1.9524e-01, -1.2195e-01]],
...,
[[ 1.0618e+00, -3.8411e-01, -1.9551e-01, ..., 3.5402e-01,
2.0068e-01, -7.9897e-01],
[-8.0113e-01, -9.7390e-01, -8.0966e-01, ..., -3.1048e-01,
-1.0879e+00, -9.8681e-01],
[-7.0012e-01, -9.9422e-01, -7.2342e-01, ..., -1.0842e-01,
-2.5396e-01, -6.2926e-01],
...,
[-7.4273e-01, -1.1524e+00, -7.9194e-01, ..., -3.7478e-01,
-4.2724e-01, -1.0167e+00],
[-2.7806e+00, -2.8605e+00, -1.8470e+00, ..., -1.0925e+00,
-1.1582e+00, -1.1182e+00],
[-6.6474e-01, -1.6755e+00, -1.0414e+00, ..., -1.5364e-01,
-2.7178e-01, -4.7292e-01]],
[[ 9.6970e-03, -1.3317e-02, 5.9614e-01, ..., 3.1654e-01,
8.2838e-01, 9.2177e-02],
[-3.6406e-02, -7.5319e-01, 1.3861e-01, ..., 4.2426e-01,
6.6543e-01, -2.6707e-01],
[-1.2494e-01, -8.4342e-01, 2.8301e-01, ..., 3.9586e-01,
4.2157e-01, -9.7293e-02],
...,
[ 6.3472e-02, -8.2481e-01, 2.4332e-01, ..., 5.7558e-01,
4.6935e-01, -4.7073e-01],
[-1.8771e-01, -1.4312e+00, -1.7265e-02, ..., 2.1805e-02,
8.4034e-02, -4.8207e-01],
[ 6.7876e-01, 1.2447e-01, 2.3152e-01, ..., 2.3006e-01,
3.0858e-01, 3.7265e-01]],
[[-1.5886e+00, -6.3508e-01, 7.4175e-03, ..., 3.1684e-01,
2.3968e-01, 8.5765e-01],
[-2.0679e+00, -3.4857e-01, -5.6719e-02, ..., -4.3034e-01,
-1.3816e-01, 1.2102e+00],
[-1.7299e+00, -3.8189e-01, 2.0512e-01, ..., -8.1336e-02,
-8.7800e-04, 9.6951e-01],
...,
[-1.7448e+00, -5.4376e-01, 2.2409e-01, ..., -2.0977e-01,
1.2792e-01, 1.0359e+00],
[-2.1781e+00, -1.1622e+00, -1.8967e-01, ..., -2.2976e-01,
-5.4113e-02, 6.3502e-01],
[-1.0403e+00, -2.0928e-01, 3.9687e-01, ..., -2.2654e-01,
-2.3242e-01, 7.0770e-01]]],
[[[-2.3142e-01, -6.1584e-01, -2.6924e-01, ..., -3.1014e-01,
1.1229e-01, -3.3994e-01],
[-7.8038e-02, -1.7187e+00, -4.0122e-01, ..., -8.4292e-01,
-5.9084e-01, -7.0999e-01],
[-1.5390e-02, -1.4813e+00, -5.1633e-01, ..., -2.0193e+00,
-9.1733e-01, -1.1587e+00],
...,
[-5.0621e-01, -1.8476e+00, -4.6956e-01, ..., -4.3477e+00,
-8.6851e-01, -2.7154e+00],
[-1.5953e+00, -2.4348e+00, -1.3713e+00, ..., 7.8951e-01,
-2.9785e+00, -2.2072e+00],
[-1.4026e+00, -2.6321e+00, -2.0835e+00, ..., -3.0215e+00,
-3.3785e+00, -1.2425e+00]],
[[-1.0388e+00, -2.6805e+00, -2.1551e+00, ..., -2.1614e+00,
-2.9115e+00, -1.0174e+00],
[-2.8056e+00, -5.1079e+00, -7.4598e+00, ..., -3.7242e+00,
-4.8751e+00, -1.8546e+00],
[-2.4666e+00, -4.3122e+00, -1.0281e+01, ..., -9.3485e+00,
-7.8917e+00, -2.2878e+00],
...,
[-4.3763e+00, -6.7843e+00, -5.8023e+00, ..., -1.1911e+01,
-1.2644e+01, -7.5162e+00],
[-6.8006e+00, -8.1414e+00, -7.1294e+00, ..., -9.9189e+00,
-9.0792e+00, -2.9529e+00],
[-2.1430e-01, -1.0304e+00, 3.3743e-02, ..., -2.2087e+00,
-1.1266e-03, 3.0476e-01]],
[[-4.9180e-01, 1.7789e-01, 7.6197e-01, ..., 7.5321e-01,
4.5259e-01, 1.1708e+00],
[-5.5353e-01, -2.4756e-01, -6.4210e-02, ..., 4.2790e-01,
4.3147e-01, 4.3177e-01],
[-1.0582e+00, -1.6137e-01, -5.0417e-01, ..., 9.7662e-01,
9.3888e-01, 2.8376e-01],
...,
[-1.5449e+00, 1.0165e+00, 8.2383e-01, ..., -5.1413e+00,
3.6282e+00, 5.1343e-01],
[-1.2440e+00, 1.0287e-01, -9.4843e-02, ..., -2.7882e+00,
2.9632e+00, 1.4469e-01],
[-1.2753e+00, -2.4452e-01, -1.4574e-01, ..., 3.8973e-01,
1.1244e+00, -5.7297e-01]],
...,
[[ 1.9673e-01, -1.9179e-01, -9.0436e-02, ..., 9.0968e-02,
-4.2841e-02, -8.0438e-01],
[-7.9394e-01, -4.7609e-01, -5.0674e-01, ..., -2.1037e-01,
-1.0312e+00, -9.7453e-01],
[-5.1232e-01, -2.0042e-01, 1.7151e+00, ..., -7.5590e-01,
-5.7056e-01, -8.2201e-01],
...,
[-5.9119e-01, -1.7314e+00, -2.0671e+00, ..., -2.0684e+01,
-1.6819e+01, -7.7743e+00],
[-3.5818e+00, -3.8055e+00, -2.7122e+00, ..., -1.4942e+01,
-5.0003e+00, -2.4268e+00],
[-4.6973e-01, -1.7423e+00, -1.1307e+00, ..., -4.5827e+00,
-1.1197e+00, -1.6633e+00]],
[[-6.8993e-02, 6.2485e-02, 2.9849e-01, ..., 5.3463e-02,
4.9764e-01, -4.5468e-02],
[-2.7323e-01, -2.6730e-01, 3.3500e-01, ..., 1.2467e-01,
2.9579e-01, -2.3391e-01],
[-4.4020e-01, -5.2352e-01, 1.6512e-02, ..., -2.2853e+00,
-1.8675e+00, -6.6039e-01],
...,
[-1.0326e-01, -7.4306e-01, -2.2050e-01, ..., 1.1638e+00,
-3.1423e+00, -6.4516e-01],
[-3.3888e-01, -1.8105e+00, -1.2049e-02, ..., 5.6620e+00,
-1.8630e+00, -9.1643e-01],
[ 4.9726e-01, -1.7168e-01, 1.3947e-01, ..., -1.5683e-02,
-6.4081e-01, 5.6983e-01]],
[[-5.5225e-01, 2.9896e-01, 5.9393e-01, ..., 5.0915e-01,
4.2700e-01, 8.4520e-01],
[-9.1627e-01, 3.4360e-02, -4.0426e-01, ..., -2.0905e-01,
-8.2456e-02, 9.4845e-01],
[-5.4684e-01, 1.4206e-01, -1.7332e+00, ..., -4.3498e-01,
7.0111e-01, 8.3015e-01],
...,
[-1.5768e+00, -1.6726e+00, -5.5546e-01, ..., 1.5806e+00,
5.4156e+00, 5.6948e+00],
[-2.6777e+00, -2.3414e+00, -1.5141e+00, ..., -4.2530e-01,
-2.1347e-01, 3.9724e-01],
[-1.2788e+00, -9.8940e-01, -3.9541e-01, ..., 1.7324e+00,
-9.2925e-02, 1.3504e+00]]]], device='cuda:0'),
tensor([[[[-8.1039e-01, -1.5404e+00, -1.5830e+00, ..., 1.7900e+00,
-3.7228e-01, -4.4323e-02],
[-4.3041e-01, -2.5872e+00, -2.4504e+00, ..., 2.3933e-01,
-9.4837e-01, -5.7139e-01],
[-5.8796e+00, -1.1394e+01, -1.2090e+01, ..., -3.2941e-01,
-3.7886e-01, -4.7392e-01],
...,
[-8.1495e+00, -1.9323e+01, -6.9338e+00, ..., -1.9701e+00,
-1.6611e+00, -3.2781e-01],
[-7.5859e+00, -1.6083e+01, -4.0505e+00, ..., -2.3222e+00,
1.1168e-01, -2.6855e-01],
[-3.4055e+00, -5.1054e+00, 4.2765e-01, ..., -4.3777e+00,
-4.6569e+00, -4.6597e-01]],
[[-2.6136e+00, -5.2676e+00, -4.5193e+00, ..., -4.3847e+00,
-1.5415e+00, -4.2489e-01],
[-5.8660e+00, -8.7934e+00, -7.2770e+00, ..., -9.1758e+00,
-3.5814e+00, -1.2400e+00],
[-1.9387e+01, -2.8896e+01, -3.0271e+01, ..., -8.3091e+00,
-2.8059e+00, -1.1854e+00],
...,
[-1.8233e+01, -2.0217e+01, 1.7060e+00, ..., -1.8695e+00,
-5.7637e+00, -9.8376e-01],
[-1.9258e+01, -1.8089e+01, -2.0268e+00, ..., -7.9390e+00,
-1.3464e+01, -2.2136e+00],
[-3.0786e+00, -4.4494e+00, -1.7351e+00, ..., 1.2991e+00,
-2.9887e+00, -1.3753e+00]],
[[-6.6188e-01, 5.2604e-01, 1.0895e+00, ..., 2.9563e+00,
5.2719e-01, 5.8309e-01],
[-8.6428e-01, -2.6579e-02, 2.4750e-01, ..., 2.8710e+00,
-3.3893e-01, 9.0666e-02],
[-2.3527e-01, 3.2297e+00, 4.5180e+00, ..., 2.7830e+00,
-9.3158e-02, 4.7920e-01],
...,
[-3.4721e+01, -2.8659e+01, -8.8492e+00, ..., -6.0745e+00,
-7.7852e+00, -2.3076e+00],
[-2.6069e+01, -2.1131e+01, -5.6407e+00, ..., -4.8385e+00,
-5.1303e+00, -9.3294e-01],
[-9.3494e+00, -5.8117e+00, -3.7107e+00, ..., -4.6492e+00,
-3.8445e+00, -3.4122e+00]],
...,
[[ 1.2986e+00, -4.5917e-01, -2.5639e-01, ..., -2.0681e+00,
-6.4322e-01, -1.2013e+00],
[-7.3726e-01, -1.0616e+00, -8.7650e-01, ..., -3.1002e+00,
-5.5801e-01, -8.1182e-01],
[-1.4033e+00, -2.4500e+00, -1.5626e+00, ..., -3.0899e+00,
6.2013e-03, -5.7645e-01],
...,
[ 2.5871e+00, -6.8278e+00, -1.1934e+01, ..., -1.8482e+00,
-4.2573e+00, -2.4815e+00],
[-4.9011e+00, -1.3744e+01, -1.7040e+01, ..., -5.5040e+00,
-6.3167e+00, -2.0938e+00],
[ 1.0567e+00, -8.7606e+00, -1.2774e+01, ..., -9.9353e-01,
-3.4112e+00, -2.0229e+00]],
[[ 4.9182e-02, -3.8671e-02, 6.8891e-01, ..., 1.7983e+00,
3.2186e-01, -1.5073e-01],
[ 3.1515e-02, -7.9697e-01, 1.5454e-01, ..., 9.2226e-01,
-2.6507e-01, -6.0115e-01],
[ 1.7374e+00, 3.7895e-01, 1.3641e+00, ..., -1.6325e-01,
3.7325e-02, -4.7965e-02],
...,
[-1.2760e+01, -4.5335e+00, 1.5928e+00, ..., -4.4726e+00,
-3.3230e+00, 1.8625e-01],
[-9.0027e+00, -7.0214e+00, -4.3432e-01, ..., -5.0706e+00,
-4.1535e+00, -1.3490e+00],
[-4.3590e+00, -3.4346e-01, 4.2129e+00, ..., -2.0859e+00,
-1.6231e+00, 1.3459e-01]],
[[-1.9349e+00, -9.1130e-01, -1.9485e-01, ..., 1.0037e-01,
1.1960e-01, -3.1179e-02],
[-2.5258e+00, -5.4934e-01, -2.2382e-01, ..., 5.7756e-01,
-2.2696e-01, -3.2804e-01],
[-5.3067e+00, -6.1695e+00, -7.1037e+00, ..., 1.3389e+00,
2.4263e-01, 3.7912e-02],
...,
[-1.0397e+01, -1.5633e+01, -4.4678e+00, ..., -7.2579e-01,
-2.6347e+00, -3.3962e+00],
[-7.7802e+00, -1.1198e+01, -3.6110e+00, ..., -5.8143e-01,
-3.6821e+00, -2.5254e+00],
[-3.2610e+00, -5.2805e+00, -3.9512e+00, ..., -2.8919e-01,
-3.1002e+00, -2.7090e+00]]],
[[[-4.3630e-01, -1.2562e+00, -9.8376e-01, ..., 2.7517e-01,
6.2351e-02, -4.2235e-01],
[-7.7713e-01, -2.9978e+00, 7.5380e-01, ..., 9.4436e-01,
-6.0512e-01, -7.5136e-01],
[-3.5751e+00, -3.9243e+00, -7.6064e-01, ..., -5.9044e-01,
-6.5998e-01, -1.2198e+00],
...,
[-2.6313e+00, 1.3978e+00, 2.1123e+00, ..., -5.8650e+00,
-1.6367e+00, -2.8181e+00],
[-1.3019e+00, -2.3365e+00, -2.3306e-01, ..., -2.2180e+00,
-3.4415e-01, -2.5938e+00],
[-1.5249e+00, -2.8679e+00, -2.0826e+00, ..., -2.5444e+00,
-2.1582e+00, -1.4418e+00]],
[[-6.7919e+00, -1.0860e+01, -6.1869e+00, ..., -5.0659e+00,
-3.3006e+00, -1.2006e+00],
[-9.6893e+00, -1.4253e+01, -1.3944e+01, ..., -7.7112e+00,
-5.1904e+00, -1.9475e+00],
[-6.9982e+00, -1.0889e+01, -1.1043e+01, ..., -8.9403e+00,
-4.8864e+00, -2.0702e+00],
...,
[-4.7843e+00, -8.9073e+00, -9.2428e+00, ..., -9.3449e+00,
-8.7777e+00, -5.2223e+00],
[-6.4352e+00, -8.9415e+00, -7.6806e+00, ..., -9.5975e+00,
-1.0666e+01, -4.0956e+00],
[-3.2751e-01, -1.2131e+00, -2.5762e-02, ..., -1.6240e+00,
-5.4113e-01, 6.9780e-02]],
[[-2.0366e+00, -7.7299e-01, 1.3445e+00, ..., 1.8777e+00,
5.9397e-01, 1.2666e+00],
[-8.9591e+00, -8.4154e+00, -1.5497e+00, ..., 2.0938e+00,
4.3223e-01, 3.7336e-01],
[-1.1539e+01, -1.2255e+01, -1.0892e+01, ..., 1.5518e+00,
1.8658e-01, 1.5589e-01],
...,
[-1.0906e+00, 2.4349e+00, -5.7887e+00, ..., -3.3784e+00,
-5.7957e+00, -4.1304e-01],
[-8.0982e-01, 2.0171e+00, 2.1190e-01, ..., 6.6517e-01,
-2.2015e+00, -1.8917e-01],
[-1.3456e+00, -1.1005e-01, -1.0739e-01, ..., -9.5897e-02,
-3.4248e-01, -6.2565e-01]],
...,
[[ 3.5246e-01, 7.4387e-01, 9.7334e-01, ..., 2.5541e-02,
2.1082e-02, -8.3069e-01],
[-2.1241e+00, -2.8915e+00, -3.8131e+00, ..., -3.0562e+00,
-1.0874e+00, -1.0393e+00],
[-5.9218e+00, -9.4285e+00, -6.9290e+00, ..., -3.8654e+00,
-3.5891e-01, -7.0475e-01],
...,
[-6.0177e-01, -6.3456e+00, -3.1706e+00, ..., -1.8984e+00,
-4.3440e+00, -3.8119e+00],
[-6.2479e+00, -7.9321e+00, -5.4808e+00, ..., -5.4793e+00,
-5.7479e+00, -4.1644e+00],
[-4.7213e-01, -1.9969e+00, -1.8590e+00, ..., -1.7302e+00,
-1.2618e+00, -1.7466e+00]],
[[-2.5331e-01, 4.6464e-01, -2.1359e+00, ..., -1.0413e+00,
5.5701e-01, -6.2443e-02],
[-1.8960e+00, -1.4678e+00, -2.9053e-01, ..., -5.9638e-01,
3.6251e-01, -2.8442e-01],
[-6.8200e+00, -4.6897e+00, -9.0124e-02, ..., -7.4319e-01,
-1.3286e-01, -4.0527e-01],
...,
[-7.9562e-01, 2.9613e-01, -7.0074e-01, ..., -2.6227e+00,
-3.7540e-02, -7.6041e-01],
[-6.4996e-01, -2.2129e+00, -8.7753e-01, ..., 9.9716e-02,
1.8385e+00, -7.2539e-01],
[ 4.6293e-01, -1.1125e-01, 4.0736e-01, ..., -9.4729e-01,
4.9885e-01, 6.9385e-01]],
[[-3.2313e+00, -8.1358e-01, 3.1030e-01, ..., 7.5581e-01,
4.3654e-01, 9.1085e-01],
[-1.0695e+01, -6.4142e+00, -5.6480e+00, ..., -1.2719e-01,
-9.4696e-02, 1.0005e+00],
[-1.2653e+01, -1.0022e+01, -8.2201e+00, ..., 9.4559e-02,
2.2878e-01, 9.0091e-01],
...,
[-5.0857e+00, -5.5645e+00, -1.0207e+00, ..., -8.2452e+00,
-1.5016e+00, 2.9638e+00],
[-3.2488e+00, -2.6029e+00, -1.0361e+00, ..., -3.4043e+00,
-1.0707e+00, 1.5812e+00],
[-1.3462e+00, -1.2410e+00, -4.3460e-01, ..., 7.6562e-01,
3.3595e-01, 1.7537e+00]]],
[[[ 2.9252e-02, -2.1585e-01, -4.2732e-02, ..., -9.6718e-01,
-5.3119e-01, -1.4141e+00],
[ 2.8916e-01, -9.1900e-01, -6.9407e-01, ..., -1.2024e+00,
-1.3417e+00, -1.4819e+00],
[ 2.3578e-01, -4.1711e-01, -4.8950e-01, ..., -9.1638e-01,
-1.0863e+00, -1.8060e+00],
...,
[-2.5655e-02, -6.2767e-01, -4.3321e-01, ..., -1.6132e+00,
-4.0497e-01, -1.6248e+00],
[-6.6526e-01, -6.9499e-01, -5.6775e-02, ..., -8.4122e-01,
-1.6776e-01, -1.6180e+00],
[-7.8184e-01, -1.3554e+00, -1.2041e+00, ..., -1.2963e+00,
-1.5839e+00, -8.5047e-01]],
[[-7.8683e-01, -2.1527e+00, -1.8443e+00, ..., -4.0026e+00,
-5.4870e+00, -1.8436e+00],
[-2.0256e+00, -3.7281e+00, -2.9951e+00, ..., -6.0357e+00,
-8.1980e+00, -3.4445e+00],
[-1.6628e+00, -3.8261e+00, -5.5383e+00, ..., -5.2703e+00,
-6.2656e+00, -2.5672e+00],
...,
[-1.7977e+00, -2.4621e+00, -2.4602e+00, ..., -6.8547e+00,
-7.0442e+00, -3.1790e+00],
[-3.8449e+00, -4.4514e+00, -3.7032e+00, ..., -6.6227e+00,
-8.2278e+00, -2.6862e+00],
[ 2.6830e-01, -3.2794e-01, 3.7319e-01, ..., 4.6532e-01,
4.1703e-01, 2.0940e-01]],
[[-2.3537e-01, 1.4865e-01, 7.0935e-01, ..., 6.7998e-01,
2.8067e-01, 1.5550e+00],
[ 2.3170e-02, -3.5685e-01, 5.4666e-01, ..., 5.7758e-01,
-8.5914e-02, 1.0715e-01],
[-1.2037e-01, -3.5340e-01, -2.5306e-02, ..., 7.7313e-02,
4.1960e-02, -1.2573e-01],
...,
[-2.4738e-01, -2.0141e-01, 1.3628e-01, ..., -7.3502e-01,
-1.1004e+00, 5.4494e-02],
[-2.6196e-01, -1.8982e-01, 1.4332e-01, ..., 6.8502e-01,
-3.3963e-01, -1.1145e-01],
[-1.4283e-01, -4.8399e-01, 6.7588e-02, ..., 2.0453e-01,
2.1022e-02, -5.9145e-01]],
...,
[[ 6.2149e-01, -2.3587e-01, -8.8004e-03, ..., 4.4818e-01,
3.5603e-01, -1.3175e+00],
[-5.1506e-01, -5.7201e-01, -5.3463e-01, ..., -1.8197e-01,
-8.0876e-01, -1.4157e+00],
[-4.0561e-01, -1.9360e-01, -7.7009e-02, ..., -9.6573e-02,
-1.3582e-01, -9.6522e-01],
...,
[-4.4717e-01, -1.6204e-01, -1.8744e-01, ..., -9.8066e-01,
-1.7407e+00, -1.6563e+00],
[-2.3707e+00, -1.7450e+00, -1.2543e+00, ..., -3.0491e+00,
-2.7353e+00, -2.0745e+00],
[-4.9251e-01, -8.3363e-01, -4.0717e-01, ..., -8.7690e-01,
-6.6195e-01, -1.0820e+00]],
[[ 1.4118e-01, 2.3695e-01, 6.4746e-01, ..., -1.1951e-01,
3.6024e-01, -3.4730e-01],
[ 1.1338e-01, 2.1700e-01, 4.8115e-01, ..., 1.4624e-01,
4.3320e-03, -8.2561e-01],
[-1.1688e-01, 2.3914e-01, 6.8530e-01, ..., 1.9497e-01,
7.8803e-02, -4.5283e-01],
...,
[-2.0947e-01, 2.5377e-02, 3.1021e-01, ..., -6.4092e-01,
3.2890e-01, -5.5209e-01],
[-5.6570e-01, -6.9318e-01, -2.4857e-02, ..., -2.5281e-01,
5.2777e-01, -9.0024e-01],
[ 5.0919e-01, 3.9938e-01, 8.0284e-02, ..., -3.3608e-01,
1.8674e-01, 2.2169e-01]],
[[-8.3961e-01, 2.4013e-01, 3.4088e-01, ..., -5.9213e-02,
-7.1501e-02, 5.9873e-01],
[-1.1504e+00, 1.7194e-01, -3.4492e-01, ..., -8.9878e-01,
-4.5567e-01, 9.0577e-01],
[-9.4464e-01, 3.5673e-01, -1.2243e-01, ..., -4.8760e-01,
-5.3218e-01, 6.4230e-01],
...,
[-9.6671e-01, 1.9704e-01, -1.1068e-01, ..., -4.3045e+00,
-8.0686e-01, 9.6580e-01],
[-1.7177e+00, -4.1542e-01, -3.4012e-01, ..., -1.6756e+00,
-5.6827e-01, 6.3181e-01],
[-7.4295e-01, -1.0479e-01, -2.9823e-01, ..., -9.8291e-02,
-1.0828e-01, 1.1090e+00]]],
[[[-1.6922e-01, -2.2146e-01, 1.6049e-01, ..., 3.1055e-02,
2.9308e-01, 2.4267e-01],
[-6.7593e-02, -1.2928e+00, -1.0374e+00, ..., -1.3702e+00,
-2.8150e-01, -2.9804e-01],
[-3.5454e-02, -9.7917e-01, -1.0713e+00, ..., -9.0868e-01,
-7.1200e-01, -6.6618e-01],
...,
[-1.7124e-01, -1.8817e+00, -1.3964e+00, ..., -1.3321e+00,
-1.1554e+00, -1.9312e+00],
[-4.6752e-01, -1.5296e+00, -7.3304e-01, ..., -6.2933e-01,
-4.0612e-01, -1.4090e+00],
[-2.9316e-01, -1.3713e+00, -1.1475e+00, ..., -1.0829e+00,
-1.1721e+00, -7.7816e-01]],
[[-1.8968e-01, -1.6489e+00, -1.0990e+00, ..., -3.7493e+00,
-2.3526e+00, -1.6410e-01],
[-1.9042e+00, -4.2511e+00, -3.1249e+00, ..., -6.3039e+00,
-4.8943e+00, -1.2229e+00],
[-1.8295e+00, -3.6956e+00, -3.1142e+00, ..., -4.9066e+00,
-3.8923e+00, -1.2615e+00],
...,
[-3.6494e+00, -5.1126e+00, -5.1393e+00, ..., -6.2500e+00,
-6.0210e+00, -2.8626e+00],
[-5.4585e+00, -6.5727e+00, -5.6044e+00, ..., -5.8765e+00,
-6.6995e+00, -2.7123e+00],
[ 5.1851e-01, -5.3744e-01, 5.0027e-01, ..., 4.7879e-01,
6.3160e-01, 1.0462e-01]],
[[-4.7413e-01, -1.0617e-01, 3.4129e-01, ..., 2.7741e-01,
-5.5931e-02, 9.5957e-01],
[-3.0999e-01, -3.8402e-01, 3.8023e-01, ..., 1.8871e+00,
2.6964e-01, 4.5242e-01],
[-7.5470e-01, -5.2702e-02, -1.0515e-01, ..., 4.1240e-01,
6.2657e-01, 4.0978e-01],
...,
[-1.7567e+00, -4.3706e-01, 3.2754e-02, ..., 6.2336e-02,
-5.4557e-02, -1.4133e-01],
[-1.4392e+00, -3.7831e-01, -8.7982e-02, ..., -4.7601e-02,
-1.5434e-01, -2.4997e-01],
[-5.7655e-01, -1.0042e-01, -1.8983e-04, ..., -9.6337e-02,
-8.0315e-02, -4.7351e-01]],
...,
[[ 1.7234e-01, -5.1809e-01, -3.8276e-01, ..., -2.1994e+00,
-2.2806e-01, -7.9755e-01],
[-6.9828e-01, -4.8111e-01, -4.1574e-01, ..., -3.4841e+00,
-1.8834e+00, -8.0024e-01],
[-3.8698e-01, -7.0000e-03, -6.8665e-02, ..., -2.9683e+00,
-8.7924e-01, -5.3253e-01],
...,
[-1.0178e+00, -8.9726e-01, -3.4463e-01, ..., -5.4554e-01,
-6.1835e-01, -1.2202e+00],
[-2.6236e+00, -2.0449e+00, -1.3705e+00, ..., -1.3473e+00,
-1.5398e+00, -1.5923e+00],
[-5.9909e-01, -1.2124e+00, -2.2826e-01, ..., -1.5412e-01,
-3.8513e-01, -4.9336e-01]],
[[-1.3432e-01, 1.2802e-01, 4.9187e-01, ..., 1.2320e+00,
3.0288e-02, 2.2840e-01],
[-3.1666e-01, -2.6855e-02, 1.7601e-01, ..., 1.2362e-01,
1.5273e-01, -1.3505e-01],
[-5.5718e-01, -1.9163e-01, 1.0696e-02, ..., -3.5973e-01,
-5.9125e-02, -3.8207e-02],
...,
[-1.7228e-01, -8.2389e-01, 3.7442e-01, ..., 6.0383e-01,
5.1176e-01, -3.1562e-01],
[-8.1065e-01, -1.9522e+00, -1.2601e-01, ..., 1.7737e-01,
1.4059e-01, -5.1177e-01],
[ 7.2719e-01, 2.6431e-02, 3.9112e-01, ..., 4.4100e-01,
4.9967e-01, 3.1814e-01]],
[[-6.0956e-01, 4.1910e-01, 5.4103e-01, ..., -1.1735e+00,
2.9806e-01, 6.4668e-01],
[-8.2832e-01, 1.9255e-01, -4.2140e-01, ..., 6.5124e-01,
-8.4146e-02, 6.8170e-01],
[-4.8000e-01, 2.1838e-01, -5.7233e-02, ..., 1.4556e+00,
7.0611e-01, 5.8771e-01],
...,
[-1.3107e+00, -6.4652e-01, -2.5181e-01, ..., -1.0867e-01,
-2.2934e-02, 1.3708e+00],
[-1.4650e+00, -9.1017e-01, -2.0182e-01, ..., 1.2120e-01,
6.0655e-02, 1.2026e+00],
[-6.4586e-01, -4.9264e-01, 1.4904e-01, ..., 3.0837e-01,
1.4426e-01, 1.2768e+00]]],
[[[-3.4428e-01, -5.9864e-01, -2.3622e-01, ..., 6.0543e-01,
-4.8682e-01, -1.3589e+00],
[-1.5498e+00, -2.3236e+00, -1.3134e+00, ..., 4.9294e-01,
-1.5178e+00, -1.1627e+00],
[-8.3882e-01, -9.9799e-01, -1.1542e+00, ..., -2.7130e+00,
-2.6484e+00, -1.8578e+00],
...,
[-2.0717e-01, -1.2523e+00, -6.1182e-01, ..., -1.1100e+00,
-1.1025e+00, -2.0003e+00],
[-3.7714e-01, -1.0390e+00, -1.7469e-01, ..., -1.5092e+00,
-6.8859e-01, -1.8139e+00],
[-4.7770e-01, -1.4841e+00, -1.2219e+00, ..., -1.3121e+00,
-1.6428e+00, -8.7236e-01]],
[[-8.5875e-01, -2.1819e+00, -1.2363e+00, ..., -1.1031e+01,
-9.4798e+00, -4.4616e+00],
[-6.3590e+00, -8.4121e+00, -4.6081e+00, ..., -1.0113e+01,
-9.4287e+00, -6.0088e+00],
[-1.0284e+01, -1.3379e+01, -8.9932e+00, ..., -8.7934e+00,
-6.9863e+00, -5.7854e+00],
...,
[-1.8649e+00, -2.9425e+00, -1.9167e+00, ..., -5.7432e+00,
-6.7127e+00, -2.5171e+00],
[-3.9529e+00, -4.7840e+00, -3.2480e+00, ..., -7.5702e+00,
-8.2320e+00, -2.5109e+00],
[ 6.4352e-01, -5.6349e-01, 4.5034e-01, ..., 2.5489e-01,
4.0053e-01, 2.6388e-01]],
[[-4.1564e-01, 7.2063e-01, 1.0763e+00, ..., 8.5792e-01,
2.0356e+00, 2.3671e+00],
[-6.7253e-01, 5.7806e-01, 8.3208e-01, ..., -7.1940e+00,
-1.0829e+00, 1.2515e+00],
[-2.7499e+00, 1.2764e+00, 2.3248e+00, ..., -8.8985e+00,
-4.5768e+00, 8.5396e-01],
...,
[-1.3495e+00, 3.0199e-01, 2.5755e-01, ..., 1.1391e+00,
-2.6723e-02, -1.7435e-01],
[-1.3634e+00, 3.0890e-01, 2.4286e-01, ..., 4.6302e-01,
-2.4689e-02, -2.1715e-01],
[-8.3052e-01, 4.7929e-02, -1.1604e-01, ..., 3.7049e-02,
1.6485e-01, -6.3882e-01]],
...,
[[ 6.1718e-01, -2.7331e-03, -9.5537e-02, ..., -2.3717e+00,
-8.3789e-01, -1.4956e+00],
[-9.7031e-01, -8.0470e-01, -9.2856e-01, ..., -5.7210e+00,
-4.5131e+00, -2.8472e+00],
[ 1.9649e-01, -2.1204e+00, -1.7799e+00, ..., -4.9127e+00,
-4.3625e+00, -2.7303e+00],
...,
[-9.0391e-01, -7.1977e-01, -1.9673e-01, ..., -4.5012e+00,
-2.3197e-01, -9.3385e-01],
[-2.4624e+00, -1.8317e+00, -9.5721e-01, ..., -2.5543e+00,
-1.6617e+00, -1.9317e+00],
[-6.0556e-01, -1.1785e+00, -1.1921e-01, ..., -3.8784e-01,
-5.4349e-01, -1.2345e+00]],
[[ 5.4338e-02, 3.0659e-01, 5.0569e-01, ..., 9.6913e-01,
-9.7305e-01, -1.5603e+00],
[-1.9524e-01, -1.6529e+00, -1.1216e+00, ..., -2.9085e+00,
-2.5334e+00, -1.5802e+00],
[ 6.2478e-01, -7.3783e-01, -9.9716e-01, ..., -4.7282e+00,
-2.3098e+00, -1.2531e+00],
...,
[-5.0552e-01, -1.0221e+00, -1.0889e-01, ..., 1.0320e+00,
1.3102e-01, -4.3697e-01],
[-9.6452e-01, -1.9217e+00, -4.6879e-01, ..., -3.9721e-01,
-2.9273e-01, -8.7174e-01],
[ 3.6211e-01, -3.1343e-01, -1.7604e-01, ..., 1.9762e-01,
2.7445e-01, 3.0712e-01]],
[[-4.1864e-01, 1.3788e-01, 5.6821e-01, ..., -4.1326e+00,
-1.7621e+00, 4.1288e-01],
[-1.2306e+00, -2.8953e-01, 7.5791e-02, ..., -8.1953e+00,
-2.2409e+00, 1.0388e+00],
[-5.0563e+00, -3.5925e+00, -9.2228e-01, ..., -8.2612e+00,
-1.2777e+00, 1.4250e+00],
...,
[-5.7627e-02, 3.8852e-02, 8.4831e-02, ..., 1.3428e+00,
-3.6469e-01, 6.4671e-01],
[-4.5577e-01, -4.2914e-01, -8.8391e-02, ..., 5.1095e-01,
-7.0844e-01, 3.8848e-01],
[-5.9169e-02, -4.6611e-01, -1.5992e-01, ..., -4.8850e-01,
-5.0991e-01, 1.0453e+00]]]], device='cuda:0'),
tensor([[[[-7.7095e-01, -1.3724e+00, -1.4007e+00, ..., -6.2807e-04,
6.3446e-01, 5.6481e-01],
[-3.6036e-01, -2.2614e+00, -2.1587e+00, ..., -4.7155e-01,
-2.4991e-01, 1.2596e-01],
[-3.8462e-01, -1.9398e+00, -1.8575e+00, ..., -4.0467e-01,
-1.2817e-01, -7.0804e-02],
...,
[-5.4383e-01, -2.0638e+00, -1.8549e+00, ..., -3.3475e-01,
-5.9933e-02, -1.2507e-01],
[-1.2241e+00, -1.9018e+00, -1.1373e+00, ..., 6.9796e-03,
2.8325e-01, -1.9280e-02],
[-1.0944e+00, -2.0011e+00, -1.4390e+00, ..., -1.0429e+00,
-1.0594e+00, -3.6544e-01]],
[[-2.5247e+00, -4.9589e+00, -4.3662e+00, ..., -4.0330e-01,
-9.8809e-01, 1.8211e-03],
[-5.1650e+00, -7.8222e+00, -6.5638e+00, ..., -1.7864e+00,
-2.6406e+00, -6.7279e-01],
[-4.2974e+00, -6.1085e+00, -5.4660e+00, ..., -1.5005e+00,
-1.4850e+00, -3.7067e-01],
...,
[-4.3596e+00, -6.0510e+00, -5.3877e+00, ..., -1.2826e+00,
-1.5348e+00, -3.4058e-01],
[-6.6850e+00, -7.8603e+00, -6.5712e+00, ..., -2.4695e+00,
-3.2971e+00, -9.3668e-01],
[-7.4013e-02, -1.1002e+00, 4.1228e-01, ..., 3.3500e-01,
3.4715e-01, 8.6670e-02]],
[[-5.5571e-01, 6.9919e-01, 1.0264e+00, ..., 8.0170e-01,
2.8270e-01, 1.0946e+00],
[-9.5344e-01, 9.0425e-02, 3.4255e-01, ..., 3.9857e-01,
2.8034e-02, 4.6907e-01],
[-1.6716e+00, -5.9933e-02, -8.2516e-02, ..., 1.0438e-01,
1.0477e-01, 4.0924e-01],
...,
[-1.8171e+00, -1.9926e-01, -9.7920e-02, ..., 5.3422e-02,
6.9414e-02, 4.2612e-01],
[-1.5016e+00, -6.5595e-03, -1.8943e-02, ..., 1.2186e-01,
1.3578e-01, 6.1425e-01],
[-1.3977e+00, -5.9957e-01, -2.5139e-01, ..., -8.9543e-02,
8.6400e-02, 3.6886e-01]],
...,
[[ 1.2073e+00, -3.1070e-01, -2.1016e-01, ..., -9.6142e-02,
-3.8857e-01, -9.4374e-01],
[-6.3949e-01, -9.6435e-01, -8.0986e-01, ..., -6.8916e-01,
-1.2838e+00, -8.3876e-01],
[-6.5922e-01, -1.0488e+00, -7.3077e-01, ..., -6.3937e-02,
-1.9796e-01, -3.9135e-01],
...,
[-8.5410e-01, -1.1490e+00, -7.7635e-01, ..., -1.2520e-01,
-1.9786e-01, -3.4869e-01],
[-3.2254e+00, -3.0952e+00, -2.0522e+00, ..., -8.2118e-01,
-8.6831e-01, -7.7711e-01],
[-6.6552e-01, -1.6879e+00, -9.8529e-01, ..., -1.3153e-02,
-1.2903e-01, -3.1667e-01]],
[[ 1.1773e-01, 8.9370e-02, 5.8187e-01, ..., 3.2169e-01,
6.9334e-01, 3.1653e-01],
[ 1.5434e-01, -6.8592e-01, 1.3143e-01, ..., -3.3421e-02,
1.0681e-01, -4.1190e-02],
[ 5.8360e-02, -6.7525e-01, 3.0833e-01, ..., -3.5234e-01,
-7.5871e-02, 2.0450e-01],
...,
[-1.2734e-01, -8.7385e-01, 2.8328e-01, ..., -1.6845e-01,
-9.5325e-02, 1.6631e-01],
[-5.0327e-01, -1.6523e+00, -5.9123e-02, ..., -4.2728e-01,
-3.3285e-01, -4.1978e-02],
[ 5.2400e-01, 8.0910e-02, 1.8193e-01, ..., -2.5182e-01,
-2.0928e-01, 3.3243e-01]],
[[-1.5121e+00, -6.2250e-01, 7.5322e-03, ..., 5.5065e-01,
3.6104e-01, 5.4671e-01],
[-2.0857e+00, -3.8804e-01, -2.2119e-02, ..., -6.7229e-02,
-1.3997e-01, 4.1363e-01],
[-1.7660e+00, -3.7348e-01, 2.4718e-01, ..., 8.4330e-02,
-1.5493e-01, 8.8851e-02],
...,
[-1.7742e+00, -5.2487e-01, 2.5388e-01, ..., 6.2264e-02,
-8.3100e-02, 1.0226e-01],
[-2.3440e+00, -1.3755e+00, -2.1564e-01, ..., -9.3851e-02,
-2.8936e-01, -1.8867e-01],
[-1.0801e+00, -4.3432e-01, 3.4962e-01, ..., -1.0530e-01,
-2.8264e-01, 1.2774e-01]]],
[[[-7.0959e-01, -1.4506e+00, -1.4314e+00, ..., -1.2755e+00,
-8.7797e-01, -1.6690e+00],
[-3.2173e-01, -2.4325e+00, -2.1073e+00, ..., -1.7253e+00,
-1.8903e+00, -1.9191e+00],
[-1.2087e-01, -1.8953e+00, -1.2315e+00, ..., -1.4427e+00,
-1.4178e+00, -2.1071e+00],
...,
[-6.6823e-02, -7.4429e-01, -2.0144e-01, ..., -7.4464e-02,
-3.2419e-03, -4.2785e-01],
[-6.3439e-01, -6.7073e-01, -3.4018e-01, ..., 1.3224e-01,
2.7409e-01, -1.6067e-01],
[-7.7881e-01, -1.5919e+00, -1.2297e+00, ..., -1.0356e+00,
-1.0917e+00, -4.5523e-01]],
[[-2.6569e+00, -5.0605e+00, -4.5589e+00, ..., -4.8048e+00,
-6.0167e+00, -2.4309e+00],
[-5.3525e+00, -8.1603e+00, -8.2288e+00, ..., -7.7869e+00,
-9.2491e+00, -4.3719e+00],
[-4.0357e+00, -5.8118e+00, -7.9046e+00, ..., -6.4966e+00,
-6.4770e+00, -3.2207e+00],
...,
[-1.8430e+00, -2.4272e+00, -2.5459e+00, ..., -1.8095e+00,
-2.2373e+00, -8.9523e-01],
[-3.0586e+00, -3.6154e+00, -3.1318e+00, ..., -2.1136e+00,
-3.1700e+00, -9.3697e-01],
[ 2.8834e-01, -4.7717e-01, 2.1088e-01, ..., 2.8307e-01,
4.2079e-01, 2.2794e-01]],
[[-6.4465e-01, 4.5506e-01, 8.6764e-01, ..., 8.4849e-01,
6.2714e-01, 1.9019e+00],
[-9.5446e-01, 3.0230e-02, 3.8761e-01, ..., 5.8742e-01,
-3.7506e-03, 1.0214e-01],
[-1.7694e+00, -2.2609e-01, 2.3279e-02, ..., 9.0638e-02,
2.2511e-02, -1.0123e-01],
...,
[-7.7314e-01, -1.5764e-02, 1.7390e-01, ..., 1.8876e-01,
2.4213e-01, 4.1777e-01],
[-5.0677e-01, 7.8260e-03, 9.0001e-02, ..., 1.6030e-01,
2.2209e-01, 5.6187e-01],
[-2.2319e-01, -3.4461e-01, 1.4815e-01, ..., -1.7243e-02,
1.1978e-01, 3.3084e-01]],
...,
[[ 1.2214e+00, -3.7877e-01, -5.5081e-02, ..., 2.3151e-01,
-1.0035e-01, -2.1071e+00],
[-7.3231e-01, -9.0215e-01, -5.4227e-01, ..., -5.6529e-01,
-1.4982e+00, -2.0449e+00],
[-6.9106e-01, -8.2784e-01, 5.7763e-02, ..., -2.8117e-01,
-5.9087e-01, -1.3652e+00],
...,
[-8.1846e-01, -5.0977e-01, -3.7149e-01, ..., -2.9504e-01,
-4.7025e-01, -5.4884e-01],
[-2.5395e+00, -2.0134e+00, -9.5023e-01, ..., -1.2892e+00,
-1.3186e+00, -1.0605e+00],
[-6.9466e-01, -9.5798e-01, -5.0834e-01, ..., -2.7118e-01,
-3.5733e-01, -3.7203e-01]],
[[ 4.3263e-02, 4.7829e-03, 4.4848e-01, ..., 4.3719e-01,
9.3177e-01, -1.9034e-02],
[ 2.2550e-01, -7.8035e-01, -1.5914e-02, ..., 4.3007e-01,
3.8310e-01, -8.3378e-01],
[ 1.6333e-01, -5.9912e-01, 8.3656e-01, ..., 2.9835e-01,
5.4284e-01, -3.6401e-01],
...,
[-4.0384e-01, -5.9453e-01, 3.9863e-01, ..., 3.6003e-01,
3.2486e-01, 7.2668e-02],
[-6.8812e-01, -9.9326e-01, -2.2855e-01, ..., 2.5703e-02,
-1.8676e-02, -1.2554e-01],
[ 3.0249e-01, -6.3499e-02, -5.1584e-01, ..., -1.3407e-01,
-7.7507e-02, 1.9442e-01]],
[[-1.5368e+00, -7.6905e-01, -4.3427e-01, ..., -4.4952e-01,
-2.8841e-01, 8.7107e-01],
[-2.1048e+00, -7.2800e-01, -5.1798e-01, ..., -7.9274e-01,
-2.0886e-01, 1.6516e+00],
[-1.4835e+00, -5.3734e-01, -4.7023e-01, ..., -2.3789e-01,
-1.1905e-01, 1.3957e+00],
...,
[-6.9264e-01, -2.1262e-01, -6.1775e-02, ..., -1.4026e-01,
8.4187e-03, 6.3478e-01],
[-1.2848e+00, -6.8539e-01, -7.4659e-01, ..., -3.2992e-01,
-2.1725e-01, 1.7127e-01],
[-4.1591e-01, -1.1373e-01, -3.1327e-01, ..., -2.2509e-01,
-2.0296e-01, 3.3375e-01]]],
[[[ 8.9122e-03, -3.3915e-01, 2.6184e-01, ..., -1.2931e+00,
-7.2520e-01, -1.2055e+00],
[ 2.4740e-01, -1.0196e+00, -8.3458e-01, ..., -2.4569e+00,
-1.6735e+00, -1.5113e+00],
[ 2.3086e-01, -6.4649e-01, -1.1414e+00, ..., -1.5573e+00,
-2.0527e+00, -4.5709e+00],
...,
[ 3.4953e-02, -6.2105e-01, 9.0511e-01, ..., -1.2924e+00,
-1.4630e+00, -1.9343e+00],
[-5.4189e-01, -6.9821e-01, 6.8331e-01, ..., -4.2176e-01,
-5.9936e-01, -1.7720e+00],
[-6.3550e-01, -1.2624e+00, -1.0038e+00, ..., -1.3405e+00,
-1.4940e+00, -9.1654e-01]],
[[-8.7566e-01, -2.3817e+00, -4.9711e+00, ..., -4.3953e+00,
-5.2273e+00, -1.9972e+00],
[-2.2073e+00, -4.0228e+00, -9.7626e+00, ..., -9.3176e+00,
-8.4315e+00, -2.9973e+00],
[-1.8487e+00, -2.9110e+00, -8.2058e+00, ..., -1.8831e+01,
-2.0497e+01, -8.8410e+00],
...,
[-1.9415e+00, -2.6934e+00, -2.5294e+00, ..., -5.6412e+00,
-7.0168e+00, -3.3384e+00],
[-3.9115e+00, -4.5346e+00, -4.1960e+00, ..., -6.9179e+00,
-7.9092e+00, -3.1850e+00],
[ 3.4600e-01, -2.4335e-01, 4.0381e-01, ..., 4.7587e-01,
6.4247e-01, 7.6342e-02]],
[[-1.9134e-01, 2.0251e-01, 3.2180e-01, ..., 9.3169e-01,
7.5485e-01, 1.6056e+00],
[ 5.8947e-02, -3.0002e-01, -2.0388e-01, ..., 5.9959e-01,
3.0635e-01, 2.5819e-01],
[-2.1101e-01, -1.7686e-01, 3.3905e-02, ..., 2.0790e+00,
1.1978e+00, 2.1561e-01],
...,
[-3.3719e-01, -1.7220e-01, 1.7078e+00, ..., -1.2353e+00,
-2.3358e-01, -4.6425e-01],
[-2.5770e-01, -1.1246e-01, 2.1679e+00, ..., 9.9589e-02,
-1.2372e-02, -6.8161e-02],
[-4.7742e-02, -3.2684e-01, 2.5121e-01, ..., -4.2826e-01,
-1.0858e-01, -4.2387e-01]],
...,
[[ 6.4409e-01, -3.6946e-02, 5.4215e-02, ..., -9.4919e-02,
-2.6874e-01, -1.5708e+00],
[-4.8865e-01, -4.2726e-01, 6.2217e-01, ..., -8.7694e-01,
-1.7202e+00, -1.7669e+00],
[-3.0476e-01, -1.6698e-01, 2.1586e+00, ..., -2.5708e+00,
-1.6140e+00, -1.6772e+00],
...,
[-4.3205e-01, -1.1936e-01, 6.7475e-01, ..., -8.0999e-01,
-2.1664e+00, -3.0963e+00],
[-2.2900e+00, -1.6551e+00, -2.3257e+00, ..., -3.2396e+00,
-3.1023e+00, -2.4182e+00],
[-4.7308e-01, -7.6996e-01, -4.9594e-01, ..., -8.6902e-01,
-9.2331e-01, -9.5683e-01]],
[[ 1.7655e-01, 2.7203e-01, 7.5919e-01, ..., 5.0546e-01,
8.5094e-01, -8.0292e-02],
[ 1.6313e-01, 1.7848e-01, 7.2701e-01, ..., -1.5827e+00,
-5.3580e-01, -4.0815e-01],
[-1.1226e-01, 2.2093e-02, -8.2525e-02, ..., -3.2919e-01,
-4.5113e-01, -3.9623e+00],
...,
[-1.4132e-01, -7.4717e-03, 2.5541e-01, ..., -4.1812e-01,
-3.7395e-02, -1.4001e-01],
[-5.4289e-01, -8.0465e-01, 4.5268e-01, ..., -3.2344e-01,
-8.5476e-01, -1.1251e+00],
[ 5.7296e-01, 3.7741e-01, 2.5510e-01, ..., 2.4138e-01,
1.8451e-01, 3.2550e-01]],
[[-8.0484e-01, 1.8904e-01, 2.0156e-01, ..., -1.1588e-03,
2.0599e-01, 8.8776e-01],
[-1.1291e+00, 7.6990e-02, -1.5287e+00, ..., -2.4799e-01,
1.7466e-01, 1.4700e+00],
[-9.1016e-01, 2.2506e-01, -2.8577e+00, ..., -3.1492e+00,
-5.8299e-01, 1.5337e+00],
...,
[-9.2658e-01, 1.1736e-01, -8.8924e-01, ..., 1.3765e+00,
3.7840e-01, 1.2572e+00],
[-1.6270e+00, -4.1502e-01, -9.5504e-01, ..., 5.1850e-01,
8.9658e-01, 1.4104e+00],
[-7.1828e-01, -1.2491e-01, -2.8681e-01, ..., 3.0114e-01,
1.9790e-01, 1.2248e+00]]],
[[[ 1.8141e-02, -2.3638e-01, -3.4731e-02, ..., -9.1846e-01,
-5.3276e-01, -1.4393e+00],
[ 3.1551e-01, -9.2516e-01, -7.0628e-01, ..., -1.1194e+00,
-1.3042e+00, -1.4872e+00],
[ 2.5898e-01, -5.6254e-01, -4.9239e-01, ..., -8.8577e-01,
-1.0642e+00, -1.8033e+00],
...,
[-3.0360e-02, -6.1487e-01, -4.6346e-01, ..., -1.1087e+00,
-1.0578e+00, -1.9151e+00],
[-6.6088e-01, -6.8260e-01, -5.4819e-02, ..., -6.9156e-01,
-5.6393e-01, -1.6905e+00],
[-7.8100e-01, -1.3440e+00, -1.1916e+00, ..., -1.2715e+00,
-1.5259e+00, -7.6982e-01]],
[[-7.4063e-01, -2.1508e+00, -1.8224e+00, ..., -3.9548e+00,
-5.4805e+00, -1.8692e+00],
[-2.0058e+00, -3.7566e+00, -3.0018e+00, ..., -5.8913e+00,
-8.0597e+00, -3.4102e+00],
[-1.5775e+00, -2.5311e+00, -2.5296e+00, ..., -5.0087e+00,
-6.1230e+00, -2.5350e+00],
...,
[-1.7893e+00, -2.6178e+00, -2.5677e+00, ..., -5.6257e+00,
-6.2480e+00, -2.5180e+00],
[-3.8251e+00, -4.3960e+00, -3.7062e+00, ..., -6.8198e+00,
-7.8737e+00, -2.5728e+00],
[ 2.4094e-01, -3.2189e-01, 3.4321e-01, ..., 3.0544e-01,
4.8030e-01, 1.9871e-01]],
[[-2.2956e-01, 1.3779e-01, 7.2367e-01, ..., 6.5465e-01,
2.7310e-01, 1.5017e+00],
[ 6.8123e-02, -3.7663e-01, 5.5851e-01, ..., 5.7075e-01,
-2.8178e-02, 1.3365e-01],
[-1.2080e-01, -2.1296e-01, 1.9640e-01, ..., 2.1331e-02,
-1.8764e-02, -1.2331e-01],
...,
[-2.6469e-01, -3.4023e-01, 1.0627e-01, ..., 3.4429e-01,
-1.1962e-01, -1.8580e-01],
[-2.6794e-01, -3.0515e-01, 1.3818e-02, ..., 1.0465e-01,
-3.4511e-02, -2.0662e-01],
[-1.0024e-01, -4.8545e-01, 1.0404e-02, ..., 9.2742e-02,
2.1455e-01, -5.7363e-01]],
...,
[[ 5.8182e-01, -2.8424e-01, -1.6647e-02, ..., 4.4288e-01,
3.5236e-01, -1.2934e+00],
[-5.2367e-01, -5.4900e-01, -5.3691e-01, ..., -1.5703e-01,
-7.6502e-01, -1.3938e+00],
[-3.7055e-01, -2.0798e-01, -2.2177e-01, ..., -1.3475e-01,
-9.9099e-02, -9.4336e-01],
...,
[-3.8767e-01, -4.3001e-02, -1.4451e-01, ..., -2.9161e-01,
-1.0557e-01, -9.4395e-01],
[-2.3123e+00, -1.7078e+00, -1.2769e+00, ..., -1.5689e+00,
-1.5895e+00, -1.8154e+00],
[-4.8587e-01, -8.8643e-01, -4.4134e-01, ..., -2.8026e-01,
-5.5197e-01, -1.1815e+00]],
[[ 1.4352e-01, 2.5431e-01, 6.3344e-01, ..., -9.3901e-02,
3.4965e-01, -3.9217e-01],
[ 1.4234e-01, 2.5974e-01, 4.7577e-01, ..., 2.1311e-01,
7.7634e-02, -8.4427e-01],
[-1.0863e-01, 1.0137e-01, 3.0323e-01, ..., 2.9617e-01,
1.5981e-01, -4.9844e-01],
...,
[-2.0439e-01, 9.5364e-02, 3.5575e-01, ..., 1.2984e-01,
1.5954e-01, -4.9248e-01],
[-5.9306e-01, -6.3936e-01, 1.0131e-01, ..., -3.0344e-01,
-3.1485e-01, -8.9457e-01],
[ 4.9040e-01, 4.3451e-01, 1.4849e-01, ..., 2.5951e-01,
3.1696e-01, 2.9401e-01]],
[[-8.6226e-01, 2.6260e-01, 3.5544e-01, ..., -5.1441e-02,
-4.5663e-02, 5.6376e-01],
[-1.1290e+00, 2.2709e-01, -3.3862e-01, ..., -9.2221e-01,
-4.0524e-01, 8.6152e-01],
[-9.5008e-01, 3.5048e-01, -1.2314e-01, ..., -5.6891e-01,
-4.9252e-01, 6.1658e-01],
...,
[-1.0136e+00, 1.5980e-01, -9.3216e-02, ..., -2.5765e-01,
-5.0322e-01, 6.0505e-01],
[-1.7570e+00, -4.3478e-01, -3.0266e-01, ..., -2.6947e-01,
-6.9767e-01, 3.9463e-01],
[-7.8669e-01, -9.1863e-02, -3.0787e-01, ..., -1.9296e-01,
-4.3903e-01, 1.0125e+00]]],
[[[-1.9391e-01, -2.7832e-01, 1.8398e-01, ..., -3.0020e-02,
4.7889e-01, 1.5291e-01],
[-1.2296e-01, -1.5385e+00, -8.8282e-01, ..., -8.3990e-01,
-5.0950e-01, -3.4689e-01],
[-1.1834e-01, -1.3278e+00, -3.5970e-01, ..., -5.5922e-01,
-4.2618e-01, -6.7809e-01],
...,
[-1.8232e-01, -1.9110e+00, -1.4007e+00, ..., -5.3702e-01,
-1.1139e+00, -1.9891e+00],
[-3.7151e-01, -1.4876e+00, -7.2903e-01, ..., -2.0306e-01,
-4.1863e-01, -1.4512e+00],
[-1.6607e-01, -1.3429e+00, -1.1777e+00, ..., -1.1624e+00,
-1.1867e+00, -7.7590e-01]],
[[-1.9275e-01, -1.4165e+00, -5.8584e-01, ..., -7.1943e-01,
-1.4052e+00, -2.6700e-01],
[-2.0620e+00, -4.0360e+00, -2.4811e+00, ..., -2.6953e+00,
-3.5233e+00, -1.2128e+00],
[-1.9239e+00, -3.4688e+00, -4.9937e+00, ..., -2.9093e+00,
-3.0334e+00, -1.2860e+00],
...,
[-3.6813e+00, -5.1524e+00, -5.1481e+00, ..., -6.4989e+00,
-6.2604e+00, -2.9110e+00],
[-5.4784e+00, -6.5968e+00, -5.8067e+00, ..., -7.1430e+00,
-6.8112e+00, -2.7739e+00],
[ 5.6933e-01, -4.8824e-01, 5.2521e-01, ..., -3.0504e-01,
6.5985e-01, 1.2262e-01]],
[[-4.9791e-01, -1.3657e-01, 3.8038e-01, ..., 3.6413e-01,
9.5253e-02, 9.0645e-01],
[-3.5360e-01, -4.1831e-01, 5.0090e-01, ..., 3.3165e-01,
2.7561e-01, 4.7496e-01],
[-8.8895e-01, -2.2939e-01, 2.3437e-01, ..., 7.0311e-02,
2.3307e-01, 4.3312e-01],
...,
[-1.7190e+00, -4.3067e-01, 3.1038e-02, ..., 1.1269e-01,
2.7919e-02, -1.9061e-01],
[-1.4144e+00, -3.1653e-01, -7.2779e-02, ..., -2.9956e-01,
-9.1122e-02, -3.0292e-01],
[-6.1211e-01, -8.0564e-02, 3.2372e-02, ..., 7.3464e-02,
-6.3157e-02, -4.9052e-01]],
...,
[[ 8.4412e-02, -5.7251e-01, -3.7160e-01, ..., -7.2626e-02,
-2.3958e-01, -7.2508e-01],
[-8.4844e-01, -6.2374e-01, -5.6887e-01, ..., -3.5101e-01,
-1.0419e+00, -7.6940e-01],
[-4.8618e-01, -1.0477e-01, 8.3590e-02, ..., 2.2491e-02,
-1.9209e-01, -5.2864e-01],
...,
[-1.0452e+00, -9.0448e-01, -3.4762e-01, ..., -1.2819e+00,
-6.3672e-01, -1.2762e+00],
[-2.5614e+00, -2.0115e+00, -1.4354e+00, ..., -2.6938e+00,
-1.6210e+00, -1.6163e+00],
[-5.5914e-01, -1.0958e+00, -1.9069e-01, ..., -1.3659e+00,
-4.3378e-01, -4.9866e-01]],
[[-1.7699e-01, 4.1811e-02, 4.3925e-01, ..., -4.2815e-03,
5.3167e-01, 1.6072e-01],
[-4.1443e-01, -2.5921e-01, -1.7414e-02, ..., -1.3724e-01,
7.6524e-02, -8.8996e-02],
[-6.3433e-01, -5.2873e-01, 5.7812e-01, ..., 1.1820e-01,
4.0428e-02, -3.8109e-02],
...,
[-1.8815e-01, -8.8035e-01, 2.7307e-01, ..., 8.4247e-01,
5.2985e-01, -3.4952e-01],
[-7.5140e-01, -1.9434e+00, -1.6214e-01, ..., 7.7205e-01,
1.3617e-01, -5.3941e-01],
[ 7.3092e-01, -1.0727e-02, 3.2365e-01, ..., 1.2024e+00,
4.8836e-01, 2.9886e-01]],
[[-6.7752e-01, 4.0738e-01, 5.6845e-01, ..., 5.1874e-01,
3.9329e-01, 6.0941e-01],
[-8.4876e-01, 2.8288e-01, -2.1092e-01, ..., -2.4034e-01,
-1.3067e-01, 6.6876e-01],
[-5.0493e-01, 3.0223e-01, 4.3982e-02, ..., 5.1465e-02,
1.1442e-01, 6.3383e-01],
...,
[-1.2825e+00, -6.7657e-01, -3.1522e-01, ..., -6.5556e-02,
-2.5342e-02, 1.4052e+00],
[-1.3671e+00, -8.3910e-01, -2.1057e-01, ..., 2.0870e-01,
8.7538e-02, 1.1973e+00],
[-6.0321e-01, -4.5657e-01, 1.3794e-01, ..., 6.5248e-01,
2.0805e-01, 1.2858e+00]]]], device='cuda:0'),
tensor([[[[ 3.5151e-01, -8.6261e-01, -1.5420e+00, ..., -9.3042e-01,
-5.0509e-01, -1.4347e+00],
[-5.8512e-01, -4.1110e+00, -3.9546e+00, ..., -1.2409e+00,
-1.3375e+00, -1.4804e+00],
[-4.0188e+00, -3.5047e+00, -1.5207e-01, ..., -9.6699e-01,
-1.0529e+00, -1.8236e+00],
...,
[ 3.8548e-02, -6.0480e-01, -4.2440e-01, ..., -9.7666e-01,
2.0271e-02, -1.5100e+00],
[-5.7563e-01, -6.5364e-01, -1.7619e-02, ..., -2.5710e+00,
4.6037e-01, -4.0628e-01],
[-7.1915e-01, -1.3047e+00, -1.1941e+00, ..., -5.0832e+00,
-3.6025e+00, -1.4149e+00]],
[[-2.9033e+00, -5.5060e+00, -7.1259e+00, ..., -3.9613e+00,
-5.4814e+00, -1.8846e+00],
[-1.0617e+01, -1.4873e+01, -1.4630e+01, ..., -6.0822e+00,
-8.2176e+00, -3.4697e+00],
[-8.7388e+00, -1.1197e+01, -1.0599e+01, ..., -5.3749e+00,
-6.5738e+00, -2.5435e+00],
...,
[-1.7821e+00, -2.4163e+00, -2.4980e+00, ..., -8.4726e+00,
-7.1365e+00, -3.6640e+00],
[-3.7850e+00, -4.3941e+00, -3.7918e+00, ..., -6.7109e+00,
-6.9517e+00, -1.4753e+00],
[ 3.1688e-01, -2.6118e-01, 3.8857e-01, ..., 9.4760e-02,
9.5936e-01, -2.8342e-01]],
[[-4.3253e-01, 1.5467e-01, 8.9656e-01, ..., 6.0612e-01,
2.3150e-01, 1.5426e+00],
[-2.3316e+00, -2.9052e+00, -9.5202e-01, ..., 5.8679e-01,
-1.2122e-01, 1.1931e-01],
[-4.6978e+00, -1.0009e+01, -9.9457e+00, ..., 1.9380e-01,
1.1430e-01, -9.9715e-02],
...,
[-2.3775e-01, -1.8726e-01, 1.2905e-01, ..., -3.7987e+00,
-7.4427e+00, -3.4211e+00],
[-2.3275e-01, -1.1400e-01, 1.6194e-01, ..., -5.2336e+00,
-6.1701e+00, -4.8903e+00],
[-1.1325e-01, -4.5498e-01, 1.0941e-01, ..., -2.0475e+00,
-1.9016e+00, -3.2986e+00]],
...,
[[ 5.9938e-01, -6.7149e-02, 1.9825e-01, ..., 4.6946e-01,
3.5152e-01, -1.3338e+00],
[ 8.8222e-01, 4.0092e-01, 6.4366e-01, ..., -1.5552e-01,
-7.8160e-01, -1.4172e+00],
[ 1.1292e+00, -2.9161e+00, -6.1324e+00, ..., 7.0468e-05,
-5.3338e-03, -9.9833e-01],
...,
[-4.1153e-01, -1.5989e-01, -1.6468e-01, ..., -6.0143e+00,
-4.8534e+00, -4.7662e+00],
[-2.2538e+00, -1.6987e+00, -1.2542e+00, ..., -6.7377e+00,
-5.7419e+00, -5.3502e+00],
[-4.6756e-01, -7.6609e-01, -3.6610e-01, ..., -2.5747e+00,
-3.4370e+00, -4.1225e+00]],
[[ 3.6681e-01, 2.4683e-01, 1.2999e-01, ..., -5.5784e-02,
4.2415e-01, -3.6867e-01],
[ 9.8718e-01, -1.9862e+00, -2.5366e+00, ..., 1.2397e-01,
4.8497e-02, -8.6379e-01],
[-1.6978e+00, -4.6813e+00, -4.1956e+00, ..., 1.5280e-01,
1.8737e-02, -4.6735e-01],
...,
[-1.3429e-01, -2.0318e-02, 3.1453e-01, ..., -2.0257e+00,
6.1214e-01, 3.5684e-01],
[-5.0051e-01, -7.4333e-01, -3.7851e-02, ..., -4.5050e+00,
-1.2827e+00, -1.6457e-01],
[ 5.5012e-01, 4.0983e-01, 6.4375e-02, ..., -3.0325e+00,
-6.0053e-01, -1.3690e-02]],
[[-9.5013e-01, -4.4812e-02, 2.5628e-01, ..., -1.1967e-01,
-7.7112e-02, 6.0420e-01],
[-2.8224e+00, -2.6428e+00, -2.9360e+00, ..., -9.4552e-01,
-3.8466e-01, 9.0708e-01],
[-7.0445e+00, -7.9442e+00, -8.6210e+00, ..., -6.7531e-01,
-5.1522e-01, 6.2423e-01],
...,
[-9.3267e-01, 1.9489e-01, -9.7993e-02, ..., -1.1258e+01,
-4.8295e+00, 2.4335e+00],
[-1.6375e+00, -3.8096e-01, -3.2526e-01, ..., -6.8721e+00,
-5.2398e+00, 2.0432e+00],
[-7.1293e-01, -9.7600e-02, -2.9073e-01, ..., -3.0344e+00,
-3.7316e+00, 1.1419e+00]]],
[[[-7.6401e-01, -1.4481e+00, -1.4308e+00, ..., -1.2518e+00,
-8.5125e-01, -1.6500e+00],
[-3.7413e-01, -2.4134e+00, -2.2430e+00, ..., -1.7725e+00,
-1.8885e+00, -1.9165e+00],
[-1.3436e-01, -1.8741e+00, -1.6877e+00, ..., -1.7297e+00,
-1.6912e+00, -2.1085e+00],
...,
[-2.1858e-01, -1.3068e+00, -6.6058e-01, ..., -6.0936e-01,
-4.9989e-01, -9.0780e-01],
[-5.8976e-01, -1.3187e+00, -3.6228e-01, ..., -2.7302e-01,
-9.9260e-02, -9.8322e-01],
[-5.4615e-01, -1.3873e+00, -1.0555e+00, ..., -8.5646e-01,
-1.0469e+00, -5.9446e-01]],
[[-2.7300e+00, -5.1373e+00, -4.6337e+00, ..., -4.7153e+00,
-5.9618e+00, -2.3854e+00],
[-5.3157e+00, -8.1380e+00, -7.5161e+00, ..., -7.7660e+00,
-9.2661e+00, -4.3682e+00],
[-4.0028e+00, -5.7883e+00, -5.7575e+00, ..., -7.9424e+00,
-7.1628e+00, -3.2079e+00],
...,
[-2.1337e+00, -3.4251e+00, -2.8101e+00, ..., -3.1139e+00,
-3.4518e+00, -1.4871e+00],
[-4.9341e+00, -6.1964e+00, -5.0430e+00, ..., -5.0547e+00,
-6.1742e+00, -2.3520e+00],
[ 5.0992e-01, -6.3761e-01, 3.9967e-01, ..., 4.7379e-01,
5.4741e-01, 2.6100e-03]],
[[-6.5131e-01, 5.3458e-01, 8.7516e-01, ..., 8.3881e-01,
6.1635e-01, 1.9202e+00],
[-9.9501e-01, 5.3726e-02, 3.9035e-01, ..., 5.6820e-01,
-1.9370e-02, 1.1406e-01],
[-1.7533e+00, -1.8221e-01, -5.0045e-02, ..., 1.5646e-01,
9.4714e-02, -9.7829e-02],
...,
[-1.0819e+00, -1.4436e-01, 5.8688e-02, ..., 2.4644e-02,
8.8199e-02, 3.5977e-01],
[-1.0608e+00, -6.3970e-02, 1.0155e-01, ..., -1.9026e-02,
8.4706e-02, 4.0269e-01],
[-5.0236e-01, -4.0893e-02, -1.4666e-02, ..., -3.7236e-02,
7.2842e-02, 4.9707e-02]],
...,
[[ 1.2813e+00, -3.1909e-01, -1.6469e-02, ..., 2.2982e-01,
-1.1186e-01, -2.1311e+00],
[-6.8019e-01, -9.3725e-01, -4.8478e-01, ..., -5.6413e-01,
-1.5123e+00, -2.0742e+00],
[-7.0041e-01, -9.1064e-01, -3.0947e-01, ..., -1.3090e-01,
-6.0596e-01, -1.3882e+00],
...,
[-5.5773e-01, -1.0880e-01, 1.1488e-01, ..., 6.2642e-02,
-1.6114e-01, -5.5545e-01],
[-2.4997e+00, -1.7003e+00, -9.0009e-01, ..., -9.4932e-01,
-1.1122e+00, -1.0682e+00],
[-4.0444e-01, -1.1199e+00, -2.0137e-01, ..., -1.8946e-01,
-2.9914e-01, -5.9414e-01]],
[[ 8.8907e-02, 5.0416e-02, 4.7170e-01, ..., 4.7626e-01,
9.4916e-01, 2.6442e-03],
[ 3.2504e-01, -7.5690e-01, -2.5836e-02, ..., 4.2144e-01,
3.6574e-01, -8.5940e-01],
[ 2.2327e-01, -6.0057e-01, 2.9469e-01, ..., -7.8631e-01,
-2.4239e-01, -3.8550e-01],
...,
[-5.3787e-01, -6.7004e-01, 6.6846e-02, ..., 7.6986e-02,
4.3706e-02, -1.1402e-01],
[-1.0043e+00, -1.7452e+00, -4.5132e-01, ..., -3.2408e-01,
-3.3830e-01, -6.7299e-01],
[ 4.9819e-01, 1.1556e-01, 1.8668e-01, ..., 2.2584e-01,
1.6496e-01, 3.2859e-01]],
[[-1.5292e+00, -7.6950e-01, -4.1566e-01, ..., -4.5908e-01,
-2.9313e-01, 8.6856e-01],
[-2.1839e+00, -7.8961e-01, -5.8227e-01, ..., -8.0133e-01,
-2.0127e-01, 1.6539e+00],
[-1.5170e+00, -5.6354e-01, -2.3612e-01, ..., -5.0327e-01,
-3.4048e-02, 1.4049e+00],
...,
[-4.7672e-01, 1.4084e-01, 1.1387e-01, ..., 1.4805e-01,
2.2106e-01, 6.9650e-01],
[-1.1108e+00, -4.7042e-01, -1.8429e-01, ..., -8.4245e-02,
1.2642e-01, 4.5524e-01],
[-5.5100e-01, -4.6730e-01, -2.6433e-01, ..., -1.6503e-01,
-1.9764e-01, 5.8682e-01]]],
[[[-3.2654e-01, -5.0662e-01, -1.8934e-01, ..., -9.7487e-01,
-5.4167e-01, -1.4736e+00],
[-1.3020e-03, -1.3891e+00, -6.5657e-01, ..., -1.2054e+00,
-1.3390e+00, -1.5217e+00],
[-9.0520e-02, -1.2423e+00, -1.0975e+00, ..., -9.2602e-01,
-1.1188e+00, -1.8621e+00],
...,
[ 2.4480e-01, -2.8101e+00, -9.0460e-01, ..., -2.8165e+00,
-3.4031e+00, -8.5113e-01],
[-3.2320e-01, -1.2212e+00, -4.3272e-01, ..., -7.8231e-01,
-1.4204e+00, -1.7030e+00],
[-5.8368e-01, -1.5863e+00, -1.2474e+00, ..., -4.6018e-01,
-1.3410e+00, -1.2937e+00]],
[[-7.3554e-01, -2.0600e+00, -9.2178e-01, ..., -3.9384e+00,
-5.4690e+00, -1.8627e+00],
[-2.2794e+00, -4.1223e+00, -2.6059e+00, ..., -5.9494e+00,
-8.0966e+00, -3.4360e+00],
[-2.0657e+00, -4.6104e+00, -3.0617e+00, ..., -5.0929e+00,
-6.2572e+00, -2.5845e+00],
...,
[-1.8152e+00, -4.7847e+00, -2.3348e+00, ..., -9.5807e+00,
-1.1287e+01, -5.4238e+00],
[-3.6860e+00, -4.3896e+00, -2.6139e+00, ..., -8.1249e+00,
-9.4075e+00, -3.9637e+00],
[ 5.3297e-01, -7.5464e-01, 4.3935e-01, ..., -2.9212e-01,
-2.6981e-01, -6.0693e-01]],
[[-3.9202e-01, 7.3386e-01, 1.0432e+00, ..., 6.1629e-01,
2.5405e-01, 1.5277e+00],
[-7.5149e-01, 2.4539e-01, 5.5726e-01, ..., 5.7601e-01,
-8.2816e-02, 8.3568e-02],
[-1.3116e+00, 1.3130e-01, -3.0394e-01, ..., 3.6377e-02,
-3.0004e-02, -1.7231e-01],
...,
[-3.9145e+00, 1.5153e-01, 6.5588e-01, ..., -7.5818e+00,
-2.2923e+00, -2.9176e+00],
[-1.3491e+00, 4.4017e-01, 4.5791e-01, ..., -1.9768e+00,
-6.3047e-01, 2.1943e-01],
[-8.6501e-01, 8.7697e-02, -1.4348e-01, ..., 7.9628e-01,
-1.6232e-01, -1.7708e-01]],
...,
[[ 4.4879e-01, -1.0386e-01, -1.2203e-01, ..., 4.5743e-01,
3.6407e-01, -1.3223e+00],
[-1.3501e+00, -1.0529e+00, -6.5218e-01, ..., -1.4276e-01,
-7.6591e-01, -1.4235e+00],
[-7.3027e-01, -3.5204e-01, 2.7708e-01, ..., -7.9281e-02,
-8.8325e-02, -9.6915e-01],
...,
[-1.0913e+00, -2.5458e+00, -3.4895e+00, ..., -2.0931e+00,
-3.7410e+00, -6.0355e+00],
[-3.0212e+00, -2.5293e+00, -1.5580e+00, ..., -3.3950e+00,
-5.9943e+00, -6.4223e+00],
[-6.6938e-01, -1.1988e+00, -7.5907e-02, ..., -3.1697e+00,
-2.9395e+00, -2.1709e+00]],
[[ 3.8694e-02, 2.4535e-01, 4.5162e-01, ..., -8.3099e-02,
4.2206e-01, -3.6459e-01],
[-9.9049e-02, -6.3184e-01, 6.1005e-02, ..., 1.8283e-01,
1.1297e-01, -8.2758e-01],
[-4.8887e-01, -7.1647e-01, -1.3389e-01, ..., 2.8492e-01,
1.5560e-01, -4.6535e-01],
...,
[-1.5484e+00, -3.2313e+00, -1.9310e-01, ..., -1.3332e+00,
-2.2506e+00, -1.4033e+00],
[-9.8592e-01, -2.6644e+00, -8.7286e-01, ..., -2.0354e+00,
-9.5454e-01, -2.4825e-02],
[ 2.4791e-01, -4.4361e-01, -2.0988e-01, ..., 7.2540e-01,
1.0154e-01, -4.8720e-01]],
[[-3.5989e-01, 1.6181e-01, 5.6414e-01, ..., -1.1426e-01,
-1.0574e-01, 5.7126e-01],
[-4.1945e-01, -2.9867e-02, 8.6484e-03, ..., -9.4223e-01,
-4.3255e-01, 9.0983e-01],
[ 1.8276e-02, -6.1722e-02, -6.2053e-01, ..., -6.0489e-01,
-5.0079e-01, 6.4294e-01],
...,
[ 7.9604e-01, -1.2191e-01, -6.5836e-01, ..., -3.2025e+00,
-6.0273e+00, -1.0941e+00],
[-2.3339e-01, -6.4079e-02, 1.9511e-01, ..., -2.6291e+00,
-1.5331e+00, 7.5731e-01],
[ 9.8419e-03, -5.1413e-01, -1.8072e-01, ..., -7.2610e-01,
1.4111e+00, 2.3459e+00]]],
[[[-3.2099e-01, -1.0285e+00, -8.4206e-01, ..., -5.9765e-02,
5.6889e-01, 4.7789e-01],
[ 1.5747e-01, -1.6090e+00, -1.6329e+00, ..., -4.6953e-01,
-2.2143e-01, 1.0884e-01],
[ 1.3267e-01, -1.0842e+00, -1.7816e+00, ..., -5.9078e-01,
-1.9247e-01, -8.0002e-01],
...,
[ 1.1771e-01, -2.7310e+00, -1.6871e+00, ..., -3.8189e-01,
3.2377e-01, -4.1651e-01],
[ 6.8215e-01, -1.1155e+00, -6.3519e-01, ..., 7.4822e-02,
3.5924e-01, 6.2462e-02],
[-1.8238e-01, -1.0840e+00, -9.7790e-01, ..., -1.2076e+00,
-1.2941e+00, -4.8047e-01]],
[[-2.1423e+00, -4.1296e+00, -3.7099e+00, ..., -6.0407e-01,
-1.1045e+00, -7.2663e-02],
[-3.9942e+00, -8.5533e+00, -8.5205e+00, ..., -1.8580e+00,
-2.6465e+00, -6.7128e-01],
[-3.2308e+00, -7.8688e+00, -8.8485e+00, ..., -4.9436e+00,
-5.4585e+00, -1.5292e+00],
...,
[-3.2874e+00, -6.6890e+00, -5.9652e+00, ..., -1.4422e+00,
-1.6265e+00, -1.1954e+00],
[-5.2543e+00, -7.1177e+00, -6.1565e+00, ..., -2.8610e+00,
-3.7301e+00, -9.9785e-01],
[ 6.7960e-01, -1.8474e-01, 4.7859e-01, ..., 3.8606e-01,
3.7149e-01, 5.3845e-02]],
[[-6.0317e-01, 3.9914e-01, 8.4154e-01, ..., 8.1512e-01,
3.3505e-01, 1.0663e+00],
[-9.5539e-01, 3.0805e-02, 3.5827e-01, ..., 4.4331e-01,
1.0829e-01, 4.6344e-01],
[-1.2986e+00, -4.1314e-01, -1.3546e+00, ..., 3.2398e-01,
2.5620e-01, 3.6409e-01],
...,
[-3.3705e+00, 1.3099e+00, -5.9751e-02, ..., 7.4604e-01,
1.4296e-02, 2.0579e-01],
[-1.9324e+00, 6.6648e-01, 9.0674e-02, ..., 2.1150e-01,
2.4253e-01, 6.8585e-01],
[-5.9177e-01, 3.1754e-01, 1.4585e-01, ..., -1.2391e-01,
8.4054e-02, 3.8439e-01]],
...,
[[ 6.8236e-01, 9.3168e-02, 3.0082e-01, ..., -7.0042e-02,
-3.1555e-01, -8.6953e-01],
[-8.0177e-01, -4.6728e-01, -3.4679e-01, ..., -6.2420e-01,
-1.2237e+00, -7.9820e-01],
[-5.3993e-01, 5.4047e-01, 1.6413e+00, ..., -3.7830e-01,
-6.4061e-01, -7.0794e-01],
...,
[ 3.1725e-01, -2.1886e+00, -2.3439e+00, ..., -6.8642e-01,
-1.2937e+00, -2.0506e+00],
[-3.1923e+00, -3.1718e+00, -1.9411e+00, ..., -9.8129e-01,
-1.1106e+00, -1.0353e+00],
[-4.1828e-01, -1.0111e+00, -3.4621e-01, ..., -4.6025e-02,
-1.8326e-01, -4.0348e-01]],
[[ 1.7674e-02, -2.5083e-01, -6.2144e-02, ..., 2.9584e-01,
6.4746e-01, 2.9264e-01],
[ 9.6891e-02, -6.2723e-01, 1.1942e-01, ..., 8.3057e-03,
1.2809e-01, -3.0809e-02],
[-2.0166e-01, -8.2284e-01, 5.3472e-02, ..., 1.6255e-03,
2.6620e-02, -7.0125e-01],
...,
[-8.6558e-01, -1.6006e+00, 9.5240e-01, ..., -3.0459e-01,
-7.5593e-01, -4.4553e-01],
[-5.5694e-01, -2.7525e+00, -5.7897e-01, ..., -4.3120e-01,
-4.1794e-01, -1.9665e-01],
[ 7.1219e-01, -1.4670e-01, 2.2481e-01, ..., -3.0066e-01,
-2.9955e-01, 2.8370e-01]],
[[-6.5255e-01, -1.5872e-01, 1.3985e-02, ..., 5.3284e-01,
3.2009e-01, 5.3849e-01],
[-1.0885e+00, -8.9177e-01, -1.2933e+00, ..., -3.7506e-02,
-1.3437e-01, 4.0301e-01],
[-7.1360e-01, -1.3262e+00, -2.2918e+00, ..., -7.4171e-01,
-1.0453e+00, -7.9196e-01],
...,
[ 2.8056e-02, -2.0034e+00, -1.5619e+00, ..., 5.8510e-02,
6.2612e-01, 1.0613e+00],
[ 2.3643e-01, -2.6331e-01, 1.4249e-01, ..., -1.9682e-01,
-3.3145e-01, -1.8560e-01],
[-4.5988e-01, -9.0587e-01, -3.5232e-01, ..., -1.8821e-01,
-3.7065e-01, 1.0383e-01]]],
[[[ 3.0695e-02, -2.6006e-01, -1.9218e-02, ..., -9.7957e-01,
-5.3725e-01, -1.4478e+00],
[ 2.8583e-01, -9.4022e-01, -7.2356e-01, ..., -1.1512e+00,
-1.3145e+00, -1.4931e+00],
[ 2.1656e-01, -6.0314e-01, -5.5376e-01, ..., -9.2855e-01,
-1.1052e+00, -1.8201e+00],
...,
[ 6.4960e-02, -5.1401e-01, -4.1447e-01, ..., -1.0894e+00,
-1.0478e+00, -1.8740e+00],
[-6.0761e-01, -6.2025e-01, -2.7619e-02, ..., -4.3863e-01,
-5.6441e-01, -1.7012e+00],
[-7.5577e-01, -1.3140e+00, -1.1992e+00, ..., -1.1242e+00,
-1.5401e+00, -7.4693e-01]],
[[-7.8516e-01, -2.1074e+00, -1.7990e+00, ..., -4.0254e+00,
-5.4667e+00, -1.8774e+00],
[-2.0562e+00, -3.9657e+00, -3.0851e+00, ..., -5.8384e+00,
-8.0050e+00, -3.3950e+00],
[-1.6037e+00, -2.7587e+00, -2.6561e+00, ..., -4.9947e+00,
-6.1432e+00, -2.6175e+00],
...,
[-1.8096e+00, -2.3768e+00, -2.4774e+00, ..., -5.3360e+00,
-6.5591e+00, -2.5998e+00],
[-3.7898e+00, -4.3427e+00, -3.6034e+00, ..., -6.4028e+00,
-7.8834e+00, -2.5275e+00],
[ 2.9166e-01, -2.8655e-01, 3.5228e-01, ..., 4.3716e-01,
4.7605e-01, 2.2670e-01]],
[[-2.2465e-01, 1.5682e-01, 7.5751e-01, ..., 6.1142e-01,
2.5196e-01, 1.5277e+00],
[ 3.8019e-02, -4.4268e-01, 5.7306e-01, ..., 5.9057e-01,
-1.0025e-01, 6.2881e-02],
[-1.3483e-01, -3.4861e-01, 2.2466e-01, ..., 6.6536e-02,
-5.0278e-02, -1.8684e-01],
...,
[-2.0438e-01, -1.8320e-01, 1.4917e-01, ..., -1.7036e-01,
-4.4009e-01, -2.5644e-01],
[-2.4798e-01, -1.9207e-01, 2.0025e-01, ..., -1.0114e-01,
-3.8585e-02, -2.0904e-01],
[-1.0997e-01, -5.2151e-01, 9.8110e-02, ..., 7.6189e-02,
2.2070e-01, -5.6557e-01]],
...,
[[ 5.9308e-01, -2.7857e-01, -2.8209e-03, ..., 4.8942e-01,
4.2070e-01, -1.3052e+00],
[-4.6840e-01, -5.4326e-01, -5.6646e-01, ..., -1.3113e-01,
-7.4817e-01, -1.3989e+00],
[-3.1624e-01, -9.7156e-02, -1.9651e-01, ..., -6.8015e-02,
-1.0586e-01, -9.7396e-01],
...,
[-4.1644e-01, -1.7600e-01, -1.7911e-01, ..., -5.6019e-01,
-3.3199e-01, -8.6875e-01],
[-2.3210e+00, -1.6960e+00, -1.2572e+00, ..., -1.5526e+00,
-1.5873e+00, -1.8226e+00],
[-4.8462e-01, -8.5564e-01, -4.3621e-01, ..., -3.5599e-01,
-5.2355e-01, -1.1700e+00]],
[[ 1.5990e-01, 2.3579e-01, 6.0159e-01, ..., -9.1864e-02,
4.2876e-01, -3.6416e-01],
[ 1.4424e-01, 2.5153e-01, 3.7527e-01, ..., 2.2040e-01,
1.4628e-01, -8.2483e-01],
[-1.1586e-01, 1.5438e-01, 2.8415e-01, ..., 2.3826e-01,
1.8653e-01, -4.4858e-01],
...,
[-1.4683e-01, 3.9642e-02, 2.9277e-01, ..., -1.0348e-01,
7.2960e-02, -4.6724e-01],
[-5.3024e-01, -6.9502e-01, -4.2546e-02, ..., -1.3783e-01,
-2.9464e-01, -9.2993e-01],
[ 5.3274e-01, 4.2663e-01, 5.8762e-02, ..., 3.3161e-01,
3.1126e-01, 3.0099e-01]],
[[-8.2286e-01, 2.7811e-01, 3.2287e-01, ..., -1.0906e-01,
-1.1524e-01, 5.8492e-01],
[-1.1472e+00, 2.3797e-01, -3.5447e-01, ..., -9.5360e-01,
-4.5484e-01, 8.9428e-01],
[-1.0023e+00, 3.3663e-01, -1.4172e-01, ..., -5.8821e-01,
-5.0821e-01, 6.5680e-01],
...,
[-9.9224e-01, 2.2329e-01, -1.1115e-01, ..., 2.6442e-01,
-1.3337e-01, 8.6599e-01],
[-1.7422e+00, -3.8916e-01, -3.8102e-01, ..., -6.9638e-01,
-7.0408e-01, 3.8876e-01],
[-7.8361e-01, -9.4614e-02, -3.0333e-01, ..., -4.7769e-01,
-4.4085e-01, 9.8607e-01]]]], device='cuda:0'),
tensor([[[[-2.0127e-01, -2.8299e-01, 2.0813e-01, ..., 5.7513e-02,
4.9676e-01, 1.7828e-01],
[-1.3219e-01, -1.5340e+00, -8.6825e-01, ..., -7.1464e-01,
-4.2499e-01, -2.9470e-01],
[-1.3897e-01, -1.3522e+00, -6.9125e-01, ..., -4.8032e-01,
-3.8481e-01, -6.5689e-01],
...,
[-2.3088e-01, -1.6990e+00, -9.8835e-01, ..., -1.0091e+00,
-1.0065e+00, -2.0004e+00],
[-9.5835e-01, -1.6279e+00, -6.6878e-01, ..., -5.3249e-01,
-4.2084e-01, -1.7503e+00],
[-1.1697e+00, -2.0601e+00, -1.5368e+00, ..., -1.4449e+00,
-1.5720e+00, -9.8156e-01]],
[[-1.7968e-01, -1.3907e+00, -5.8642e-01, ..., -5.7953e-01,
-1.3353e+00, -2.2377e-01],
[-2.0243e+00, -4.0048e+00, -2.3890e+00, ..., -2.3182e+00,
-3.2918e+00, -1.1039e+00],
[-1.8925e+00, -3.3647e+00, -2.5140e+00, ..., -2.4931e+00,
-2.8840e+00, -1.2011e+00],
...,
[-3.7422e+00, -5.5068e+00, -5.1972e+00, ..., -5.6993e+00,
-6.3584e+00, -3.0933e+00],
[-6.1491e+00, -7.0220e+00, -6.0421e+00, ..., -6.0652e+00,
-7.4202e+00, -2.7686e+00],
[ 3.3160e-02, -8.1130e-01, 2.2038e-01, ..., 4.3803e-01,
6.7086e-01, 1.8135e-01]],
[[-5.0347e-01, -1.5433e-01, 3.5677e-01, ..., 3.3009e-01,
3.2704e-02, 8.6805e-01],
[-3.6023e-01, -5.0066e-01, 4.5959e-01, ..., 3.5066e-01,
2.6456e-01, 4.7670e-01],
[-8.6737e-01, -2.6214e-01, 1.8617e-01, ..., 4.3096e-02,
1.7071e-01, 3.9462e-01],
...,
[-1.4074e+00, 3.6879e-02, 4.7663e-01, ..., -3.7609e-02,
6.6584e-02, 3.3990e-01],
[-1.2537e+00, 1.5720e-01, 1.3852e-01, ..., -1.1247e-01,
-9.6477e-02, 3.7293e-01],
[-1.1989e+00, -5.9417e-02, 2.6082e-01, ..., -2.6466e-02,
-1.7978e-01, -2.9843e-01]],
...,
[[ 7.1876e-02, -6.1503e-01, -3.9488e-01, ..., -9.0254e-02,
-2.5771e-01, -7.4662e-01],
[-8.1663e-01, -6.1583e-01, -5.4335e-01, ..., -2.9890e-01,
-9.8670e-01, -7.3496e-01],
[-5.0616e-01, -1.1212e-01, -3.8176e-02, ..., 8.9110e-02,
-1.9476e-01, -5.2275e-01],
...,
[-5.9944e-01, -5.7421e-01, -3.3329e-01, ..., -2.9257e-01,
-6.5788e-01, -1.2414e+00],
[-2.9553e+00, -2.7977e+00, -1.9804e+00, ..., -1.8007e+00,
-2.0401e+00, -2.0600e+00],
[-5.3103e-01, -1.5735e+00, -1.1603e+00, ..., -9.4180e-01,
-9.9351e-01, -1.3813e+00]],
[[-1.7031e-01, 3.3917e-02, 4.3084e-01, ..., 1.5178e-01,
5.5548e-01, 1.8948e-01],
[-4.3477e-01, -2.4344e-01, -1.2298e-02, ..., -4.3694e-02,
1.2884e-01, -2.5134e-02],
[-6.7181e-01, -5.9109e-01, -5.7240e-02, ..., 2.8738e-02,
2.8019e-02, -6.6951e-03],
...,
[-1.1951e-01, -1.0320e+00, 3.2693e-01, ..., 2.9115e-01,
1.0407e-01, -7.9609e-01],
[-5.0166e-01, -1.8481e+00, -8.6046e-02, ..., 1.2814e-01,
6.8550e-02, -9.8321e-01],
[ 4.8146e-01, -1.3300e-01, 1.5203e-01, ..., 3.0962e-01,
3.2806e-01, 2.2734e-01]],
[[-6.8295e-01, 4.0425e-01, 5.4653e-01, ..., 4.6402e-01,
3.6546e-01, 5.7995e-01],
[-8.6432e-01, 2.8835e-01, -2.6249e-01, ..., -2.5212e-01,
-1.5836e-01, 6.4988e-01],
[-4.9031e-01, 3.3280e-01, 3.2098e-02, ..., 8.7272e-02,
8.1751e-02, 5.9199e-01],
...,
[-1.0404e+00, -9.2637e-01, -8.6257e-01, ..., -4.0722e-01,
1.4137e-02, 8.5865e-01],
[-1.6645e+00, -1.6803e+00, -1.1096e+00, ..., -8.9195e-01,
-3.2699e-01, 4.7559e-01],
[-9.3838e-01, -8.6429e-01, -2.8588e-01, ..., -3.7654e-01,
-1.0061e-01, 1.1085e+00]]],
[[[-4.6323e-01, -6.4509e+00, -4.9040e+00, ..., -3.0431e-01,
6.3851e-01, -9.4548e-01],
[-5.5724e-01, -6.8931e+00, -5.4233e+00, ..., -1.9599e+00,
-9.1243e-01, -1.3130e+00],
[ 5.6101e-01, -3.3956e+00, -6.4887e+00, ..., 7.5681e-01,
-1.3752e+00, -1.8674e+00],
...,
[ 1.4034e-01, -7.2259e-01, -3.0845e-01, ..., -8.3100e-01,
-3.5402e-02, -5.8521e-01],
[-2.4077e-01, -5.2580e-01, -3.6911e-02, ..., -2.7747e-01,
8.2491e-01, -2.9134e-01],
[-5.2509e-01, -1.2774e+00, -1.0844e+00, ..., -1.7674e+00,
-1.3467e+00, -5.3906e-01]],
[[-4.7250e+00, -9.5098e+00, -1.0096e+01, ..., -5.0467e+00,
-7.0811e+00, -2.0577e+00],
[-8.0872e+00, -1.2315e+01, -1.2828e+01, ..., -1.1056e+01,
-1.0070e+01, -3.1490e+00],
[-8.0151e+00, -1.5782e+01, -2.2496e+01, ..., -8.2295e+00,
-8.0300e+00, -2.8572e+00],
...,
[-2.0607e+00, -2.7339e+00, -2.3769e+00, ..., -8.3666e+00,
-4.4217e+00, -1.1753e+00],
[-3.0968e+00, -3.3729e+00, -2.4478e+00, ..., -5.4361e+00,
-4.7971e+00, -1.1954e+00],
[ 4.1889e-01, -1.4141e-01, 4.3609e-01, ..., -2.3548e+00,
-4.3631e-01, 1.9885e-01]],
[[-1.4592e+00, -5.2764e+00, -6.2331e+00, ..., 7.3012e-01,
1.4064e+00, 1.5885e+00],
[-1.1109e+00, -7.5301e+00, -1.7099e+01, ..., -1.6538e+00,
8.9632e-01, 5.1535e-01],
[-1.3696e+00, -7.9322e+00, -1.6939e+01, ..., -3.0435e-02,
1.3689e+00, 2.3875e-01],
...,
[-8.2395e-01, 1.4587e-01, 3.1980e-01, ..., 2.7175e+00,
9.5967e-01, 3.8312e-01],
[-5.6328e-01, 1.4224e-01, 2.3303e-01, ..., -5.7198e-01,
7.3323e-01, 5.2662e-01],
[-1.4283e-01, -1.4371e-01, 1.2544e-01, ..., -2.3953e+00,
6.3771e-01, 3.3724e-01]],
...,
[[ 2.5876e+00, 4.6745e+00, 1.6902e+00, ..., -3.1697e+00,
-1.4602e+00, -1.1267e+00],
[ 6.6761e-01, -3.2662e-02, -7.8630e+00, ..., -7.7397e+00,
-3.9589e+00, -1.4526e+00],
[ 6.4584e-01, -1.3850e-01, -6.1159e+00, ..., -5.8834e+00,
-2.4408e+00, -1.2101e+00],
...,
[-6.4097e-01, -5.1084e-01, -3.2788e-01, ..., -3.9686e+00,
-8.6545e-01, -6.3377e-01],
[-2.1993e+00, -1.7772e+00, -1.2848e+00, ..., -7.0967e+00,
-3.1985e+00, -1.1363e+00],
[-6.3953e-01, -8.3969e-01, -4.1148e-01, ..., -4.4936e+00,
-2.3648e+00, -4.2311e-01]],
[[ 6.4745e-01, -1.0435e+00, -2.4390e+00, ..., 1.2629e+00,
7.8928e-01, -4.2140e-01],
[ 5.1332e-01, -3.8381e+00, -7.4347e+00, ..., -2.8988e+00,
-4.2362e-02, -7.0407e-01],
[ 5.1793e-01, -2.8386e+00, -1.0056e+01, ..., 1.2363e-01,
-3.4194e-01, -6.3818e-01],
...,
[-1.5895e-01, -5.9512e-01, 2.2830e-01, ..., 7.5832e-01,
-3.2684e-01, -1.8660e-02],
[-4.7626e-01, -1.0812e+00, 1.7956e-02, ..., 4.8660e-01,
-3.8805e-02, -2.5311e-01],
[ 4.2726e-01, -6.1078e-02, -1.3621e-01, ..., -1.8828e+00,
-1.3715e-01, 1.8675e-01]],
[[-1.6829e+00, -5.6242e+00, -9.1673e+00, ..., 4.1754e-01,
1.2809e+00, 8.5325e-01],
[-3.0158e+00, -7.2648e+00, -9.9568e+00, ..., -8.3372e-01,
9.6343e-01, 9.1854e-01],
[-1.9358e+00, -2.8256e+00, -5.5230e+00, ..., -9.7467e-01,
8.8940e-01, 7.8326e-01],
...,
[-6.3124e-01, -3.4751e-01, -2.1467e-01, ..., 1.7694e+00,
1.3385e-01, 6.6052e-01],
[-1.0946e+00, -6.7581e-01, -3.6657e-01, ..., 8.0801e-01,
-2.7832e-01, 2.6396e-01],
[-3.9291e-01, -1.5939e-01, -1.5309e-01, ..., -6.7094e-02,
-1.2352e-01, 4.5625e-01]]],
[[[-6.6121e-01, -1.4251e+00, -1.7207e+00, ..., -1.5935e+00,
-1.0015e+00, -1.7603e+00],
[-2.9417e-01, -2.5868e+00, -3.3881e+00, ..., -1.4195e+00,
-2.0286e+00, -1.9608e+00],
[-2.0568e-01, -2.0430e+00, -2.1186e+00, ..., -1.5074e+00,
-1.5791e+00, -2.1535e+00],
...,
[-1.3466e+00, -4.4637e+00, -3.8741e+00, ..., -7.4779e-01,
-3.5644e-01, -8.7715e-01],
[ 5.5184e-02, -3.5238e+00, -5.2642e+00, ..., -4.2495e-01,
4.3126e-01, -9.5382e-01],
[ 7.5345e-01, -9.4282e-01, -3.6756e+00, ..., -8.5924e-01,
-1.0305e+00, -7.1951e-01]],
[[-2.5259e+00, -5.1291e+00, -8.6700e+00, ..., -5.2323e+00,
-6.3506e+00, -2.5551e+00],
[-5.5264e+00, -8.4941e+00, -9.8705e+00, ..., -8.1789e+00,
-9.4701e+00, -4.4069e+00],
[-4.0986e+00, -6.1519e+00, -8.9655e+00, ..., -7.8233e+00,
-6.8729e+00, -3.2530e+00],
...,
[-5.3013e+00, -1.0559e+01, -6.7893e+00, ..., -5.2504e+00,
-4.4461e+00, -2.1961e+00],
[-6.8417e+00, -1.0857e+01, -6.7516e+00, ..., -6.0324e+00,
-6.4463e+00, -2.5823e+00],
[ 6.6150e-01, -3.2486e-01, -9.1816e-01, ..., -1.4275e-03,
3.7713e-01, -9.6052e-02]],
[[-7.2151e-01, 2.7662e-01, 3.9263e-01, ..., 7.3540e-01,
6.5824e-01, 1.9078e+00],
[-9.3452e-01, -2.1004e-01, -1.1331e+00, ..., 3.6376e-01,
-6.2460e-03, 8.2321e-02],
[-1.7310e+00, -2.6794e-01, -1.4225e+00, ..., 9.3550e-02,
-7.4663e-03, -9.1328e-02],
...,
[-5.4195e+00, -1.2817e+01, -1.0224e+01, ..., -8.3921e-01,
-2.8276e-01, 2.4723e-01],
[-5.3007e+00, -9.7165e+00, -8.4427e+00, ..., 2.3711e-01,
-4.5268e-01, 2.0861e-01],
[-6.9102e-01, -2.8572e+00, -6.1905e+00, ..., 1.9738e-01,
-1.0289e-01, 3.8086e-02]],
...,
[[ 1.0509e+00, -4.4825e-01, 6.4651e-01, ..., 5.4052e-01,
-3.6574e-02, -2.1107e+00],
[-1.0073e+00, -7.9182e-01, 1.1041e+00, ..., -7.8627e-01,
-1.5214e+00, -2.0651e+00],
[-8.1487e-01, -6.8744e-01, 8.1616e-02, ..., -1.0724e+00,
-7.0581e-01, -1.4311e+00],
...,
[-1.0015e+00, -3.2044e+00, -5.4495e+00, ..., -9.2287e-01,
-6.1309e-01, -8.5718e-01],
[-1.3012e+00, -1.9178e+00, -5.2626e+00, ..., -1.8787e+00,
-2.0998e+00, -1.5597e+00],
[ 4.2224e-01, 4.5023e-01, -2.6268e+00, ..., -5.3852e-01,
-5.9266e-01, -6.9955e-01]],
[[-9.4395e-02, -3.3860e-02, 8.5684e-01, ..., 2.4558e-01,
8.1560e-01, -3.1845e-03],
[-5.5227e-02, -8.4676e-01, -8.2156e-01, ..., 7.8522e-01,
3.0278e-01, -8.2509e-01],
[-1.1954e-01, -7.8443e-01, -8.5846e-01, ..., -7.8879e-02,
4.4546e-01, -4.0159e-01],
...,
[-1.2931e+00, -6.3070e+00, -5.9369e+00, ..., -8.8268e-01,
-5.1867e-01, 1.3669e-02],
[-2.2072e+00, -5.0473e+00, -5.3262e+00, ..., -6.0375e-01,
-5.5444e-02, -6.1908e-01],
[ 2.9680e-01, -8.1665e-01, -1.3925e+00, ..., 1.4982e-01,
1.7636e-01, 2.7885e-01]],
[[-1.7010e+00, -9.0242e-01, -1.3011e+00, ..., -8.2434e-01,
-2.8074e-01, 8.7258e-01],
[-2.1743e+00, -9.0369e-01, -2.5321e+00, ..., -1.2696e+00,
-1.4005e-01, 1.6596e+00],
[-1.5176e+00, -8.8569e-01, -1.4669e+00, ..., -1.8349e-01,
-5.7133e-02, 1.4105e+00],
...,
[-7.5595e-01, -3.5580e+00, -9.0493e+00, ..., -1.3329e+00,
-6.0578e-01, 9.6947e-01],
[-3.7484e-01, -2.1657e+00, -6.8919e+00, ..., -6.7179e-01,
-3.0335e-03, 8.6184e-01],
[-4.8587e-01, -6.6948e-02, -8.5403e-02, ..., 3.4039e-01,
9.3096e-02, 8.3677e-01]]],
[[[-2.2035e-01, -2.6091e-01, 1.9028e-01, ..., 4.2831e-02,
4.8883e-01, 1.4872e-01],
[-1.2118e-01, -1.5336e+00, -8.8078e-01, ..., -7.6497e-01,
-4.6584e-01, -3.0267e-01],
[-1.2406e-01, -1.3411e+00, -6.7433e-01, ..., -4.5275e-01,
-4.0472e-01, -6.8548e-01],
...,
[-2.6225e-01, -1.7067e+00, -5.0940e-01, ..., -9.3966e-01,
-9.4892e-01, -2.0196e+00],
[-1.0790e+00, -1.7472e+00, -3.5728e-01, ..., -5.1792e-01,
-4.5036e-01, -1.7814e+00],
[-1.2681e+00, -2.1515e+00, -1.4769e+00, ..., -1.4220e+00,
-1.5425e+00, -9.6257e-01]],
[[-2.0555e-01, -1.4416e+00, -5.6974e-01, ..., -6.8411e-01,
-1.3685e+00, -2.5838e-01],
[-2.0224e+00, -4.0150e+00, -2.3514e+00, ..., -2.4275e+00,
-3.3805e+00, -1.1509e+00],
[-1.8744e+00, -3.4161e+00, -2.5477e+00, ..., -2.5677e+00,
-2.9627e+00, -1.2730e+00],
...,
[-3.7028e+00, -5.4704e+00, -5.1508e+00, ..., -5.4073e+00,
-6.1614e+00, -2.8843e+00],
[-6.1515e+00, -7.0140e+00, -5.7906e+00, ..., -6.1220e+00,
-7.4020e+00, -2.6950e+00],
[-2.4814e-02, -8.9696e-01, 3.2981e-01, ..., 4.1128e-01,
7.0921e-01, 2.0218e-01]],
[[-5.3271e-01, -1.8747e-01, 3.4838e-01, ..., 3.8283e-01,
6.0136e-02, 8.5983e-01],
[-3.8605e-01, -4.8392e-01, 4.5422e-01, ..., 3.3880e-01,
3.0241e-01, 5.1536e-01],
[-8.6355e-01, -2.5249e-01, 1.5938e-01, ..., 5.3797e-03,
1.5209e-01, 4.2695e-01],
...,
[-1.3933e+00, 4.3633e-02, 1.0273e+00, ..., 5.4947e-02,
8.6943e-03, 3.1548e-01],
[-1.2479e+00, 1.5418e-01, 4.5495e-01, ..., -2.6782e-02,
2.0205e-02, 3.4834e-01],
[-1.1734e+00, -5.9861e-02, 3.2080e-01, ..., -1.1972e-03,
-3.6993e-02, -2.7756e-01]],
...,
[[ 4.4382e-02, -6.3826e-01, -3.8785e-01, ..., -7.4103e-02,
-2.4472e-01, -7.1491e-01],
[-8.3569e-01, -6.1831e-01, -5.8314e-01, ..., -2.3912e-01,
-1.0025e+00, -7.5692e-01],
[-5.0317e-01, -1.3951e-01, -1.0692e-02, ..., 6.7926e-02,
-2.2166e-01, -5.4404e-01],
...,
[-6.1457e-01, -6.2287e-01, -2.7581e-01, ..., -3.7944e-01,
-5.8017e-01, -1.1661e+00],
[-3.1303e+00, -2.8997e+00, -2.0646e+00, ..., -1.9010e+00,
-2.0308e+00, -2.0661e+00],
[-5.2880e-01, -1.6179e+00, -1.1935e+00, ..., -8.3620e-01,
-9.4736e-01, -1.3545e+00]],
[[-1.9216e-01, 2.4438e-02, 4.3522e-01, ..., 7.8661e-02,
5.0728e-01, 1.4333e-01],
[-4.5471e-01, -2.7339e-01, 7.2392e-03, ..., -8.8669e-02,
1.0644e-01, -7.0635e-02],
[-6.6051e-01, -5.8075e-01, -4.9035e-02, ..., 1.1863e-01,
5.8078e-02, -6.6825e-02],
...,
[-1.3730e-01, -1.0136e+00, 7.8064e-01, ..., 2.3055e-01,
1.1382e-01, -7.2019e-01],
[-5.2991e-01, -1.8577e+00, 2.3526e-01, ..., 2.4923e-02,
-5.1100e-02, -9.1522e-01],
[ 4.2913e-01, -1.5481e-01, 2.8808e-01, ..., 2.5464e-01,
3.1168e-01, 3.4814e-01]],
[[-6.8339e-01, 3.9221e-01, 5.4369e-01, ..., 5.0077e-01,
3.8064e-01, 6.0916e-01],
[-8.2848e-01, 2.9297e-01, -2.1360e-01, ..., -2.3973e-01,
-1.6334e-01, 6.5762e-01],
[-4.7048e-01, 3.4749e-01, 6.6358e-02, ..., 3.8500e-02,
8.8498e-02, 6.1583e-01],
...,
[-1.0732e+00, -9.2775e-01, -7.5483e-01, ..., -3.8831e-01,
-6.7767e-02, 8.0934e-01],
[-1.8191e+00, -1.8351e+00, -1.1432e+00, ..., -7.9943e-01,
-4.7576e-01, 4.1026e-01],
[-9.3125e-01, -9.6043e-01, -3.4020e-01, ..., -3.0272e-01,
-1.5973e-01, 1.1205e+00]]],
[[[-7.4017e-01, -1.4719e+00, -1.4671e+00, ..., -1.2503e+00,
-8.8469e-01, -1.7142e+00],
[-3.4362e-01, -2.4616e+00, -2.3328e+00, ..., -1.7909e+00,
-1.9120e+00, -1.9413e+00],
[-1.4570e-01, -1.9088e+00, -1.3520e+00, ..., -1.3130e+00,
-1.4597e+00, -2.1273e+00],
...,
[-3.7244e+00, -3.9914e+00, 2.5781e-01, ..., -6.4183e-01,
-4.6027e-01, -9.6493e-01],
[-1.5277e+00, -5.1874e+00, -3.5534e+00, ..., -2.6423e-01,
-6.5425e-02, -1.0055e+00],
[ 2.3963e-01, -3.5363e+00, -4.4116e+00, ..., -8.5406e-01,
-1.0285e+00, -6.0066e-01]],
[[-2.7198e+00, -5.1870e+00, -4.5986e+00, ..., -4.7627e+00,
-6.0331e+00, -2.4312e+00],
[-5.4109e+00, -8.2524e+00, -7.5938e+00, ..., -7.8036e+00,
-9.3559e+00, -4.4229e+00],
[-4.1349e+00, -6.1506e+00, -8.6815e+00, ..., -6.1020e+00,
-6.5906e+00, -3.2690e+00],
...,
[-4.0900e+00, -8.3814e+00, -4.8684e+00, ..., -3.2272e+00,
-3.5905e+00, -1.5574e+00],
[-1.0048e+01, -1.6115e+01, -8.2755e+00, ..., -5.2001e+00,
-6.0963e+00, -2.3618e+00],
[ 1.1205e+00, -2.6867e+00, -1.5726e-01, ..., 4.6903e-01,
6.4263e-01, 5.3868e-02]],
[[-6.6960e-01, 4.6221e-01, 8.6235e-01, ..., 8.3707e-01,
6.1775e-01, 1.9257e+00],
[-9.4609e-01, 7.0207e-02, 3.7567e-01, ..., 5.8873e-01,
5.3880e-03, 1.1033e-01],
[-1.7491e+00, -1.7015e-01, -1.4525e-01, ..., 1.6827e-02,
-1.0170e-02, -9.0602e-02],
...,
[-7.2483e+00, -9.9010e+00, -8.5736e+00, ..., 1.6531e-02,
6.7724e-02, 3.4516e-01],
[-8.5999e+00, -1.1153e+01, -1.1569e+01, ..., 9.5411e-02,
1.1246e-01, 3.3880e-01],
[-6.0058e+00, -9.2833e+00, -1.1394e+01, ..., 6.1271e-03,
1.0254e-01, 1.8494e-02]],
...,
[[ 1.2735e+00, -3.2997e-01, -5.5338e-02, ..., 2.2472e-01,
-8.8952e-02, -2.1481e+00],
[-7.2449e-01, -9.5347e-01, -5.4004e-01, ..., -5.6885e-01,
-1.5104e+00, -2.1070e+00],
[-7.1014e-01, -9.4110e-01, -2.2396e-01, ..., -3.2300e-01,
-5.7834e-01, -1.4064e+00],
...,
[-1.2609e+00, -4.2975e+00, -8.0843e+00, ..., 7.0275e-02,
-1.2157e-01, -5.2011e-01],
[-3.3582e+00, -6.0872e+00, -8.3970e+00, ..., -9.9360e-01,
-1.1156e+00, -1.0668e+00],
[ 1.8903e+00, 4.1229e-01, -1.7769e+00, ..., -2.9382e-01,
-2.9915e-01, -5.7436e-01]],
[[ 6.1831e-02, 1.1473e-02, 4.8155e-01, ..., 5.3507e-01,
9.6202e-01, -6.8913e-03],
[ 2.5506e-01, -8.0024e-01, -6.9544e-02, ..., 4.7257e-01,
4.0370e-01, -8.5562e-01],
[ 2.1991e-01, -6.3452e-01, 9.0985e-01, ..., 6.7592e-01,
5.3192e-01, -3.7500e-01],
...,
[-3.7719e+00, -5.6206e+00, -4.6135e+00, ..., -2.7226e-02,
9.1222e-02, -1.5841e-01],
[-4.5031e+00, -8.4226e+00, -8.1366e+00, ..., -3.4371e-01,
-2.7366e-01, -6.2998e-01],
[-1.7140e+00, -3.5914e+00, -3.1317e+00, ..., 2.7534e-01,
2.1530e-01, 3.7460e-01]],
[[-1.5915e+00, -8.7774e-01, -4.7810e-01, ..., -4.7108e-01,
-2.9939e-01, 8.6515e-01],
[-2.1745e+00, -7.8368e-01, -5.7551e-01, ..., -8.1688e-01,
-2.1032e-01, 1.6614e+00],
[-1.5336e+00, -5.7044e-01, -2.9278e-01, ..., -4.1509e-01,
-1.5243e-01, 1.3956e+00],
...,
[-1.1212e+00, -5.8915e+00, -8.4767e+00, ..., 1.7571e-01,
1.9204e-01, 7.4941e-01],
[ 1.3290e+00, -3.1291e+00, -6.5135e+00, ..., -7.8021e-02,
2.6361e-02, 4.7680e-01],
[ 9.3658e-01, -6.8678e-01, -2.7377e+00, ..., -1.4812e-01,
-2.0802e-01, 6.2028e-01]]]], device='cuda:0'),
tensor([[[[-1.3478e-01, -5.0520e-01, -2.1037e-01, ..., -1.4478e-01,
2.8665e-01, -7.0290e-02],
[ 2.5688e-01, -1.2153e+00, -7.1103e-01, ..., -4.2726e-01,
-2.5420e-01, -1.9391e-01],
[ 1.1965e-01, -8.9062e-01, -3.4718e-01, ..., -1.8227e-01,
-1.9797e-01, -5.4377e-01],
...,
[-2.4038e-01, -1.6727e+00, -7.9933e-01, ..., -9.4518e-01,
-9.8404e-01, -2.0556e+00],
[-1.1156e+00, -1.7116e+00, -7.5150e-01, ..., -5.3357e-01,
-4.6228e-01, -1.7915e+00],
[-1.3060e+00, -2.1451e+00, -1.6147e+00, ..., -1.4404e+00,
-1.5630e+00, -9.7725e-01]],
[[-1.4210e+00, -3.0319e+00, -2.4906e+00, ..., -2.5909e+00,
-3.3320e+00, -1.3080e+00],
[-3.1391e+00, -5.0372e+00, -3.6449e+00, ..., -3.6300e+00,
-4.6705e+00, -2.0181e+00],
[-2.0439e+00, -3.0976e+00, -2.3379e+00, ..., -2.4161e+00,
-2.7050e+00, -1.2560e+00],
...,
[-3.7429e+00, -5.4394e+00, -5.1645e+00, ..., -5.4428e+00,
-6.1842e+00, -2.9203e+00],
[-6.1966e+00, -6.9861e+00, -6.1225e+00, ..., -6.1002e+00,
-7.4148e+00, -2.7139e+00],
[-4.1477e-02, -9.0125e-01, 2.5643e-01, ..., 4.1244e-01,
7.0364e-01, 1.8951e-01]],
[[-4.4823e-01, 2.1489e-01, 8.3769e-01, ..., 8.2048e-01,
5.7243e-01, 1.3581e+00],
[-3.5714e-01, -3.9557e-02, 5.8161e-01, ..., 4.4667e-01,
4.1285e-01, 3.3341e-01],
[-8.0381e-01, 4.8435e-02, 2.7302e-01, ..., 2.0906e-01,
3.5373e-01, 4.2585e-01],
...,
[-1.3974e+00, 7.0783e-03, 8.6154e-01, ..., 2.5028e-02,
-2.6188e-02, 3.1770e-01],
[-1.1899e+00, 8.1848e-02, 1.1125e-01, ..., -3.6471e-02,
-2.8408e-02, 3.2217e-01],
[-1.1533e+00, -1.3393e-01, 8.2057e-02, ..., 1.0195e-02,
-6.1092e-02, -3.3420e-01]],
...,
[[ 9.0171e-01, 2.4889e-01, 2.3000e-01, ..., 3.3175e-01,
1.7175e-01, -8.8962e-01],
[-1.1448e+00, -8.6277e-01, -6.9576e-01, ..., -7.0754e-01,
-1.4644e+00, -1.1064e+00],
[-6.1173e-01, -2.6805e-01, -2.3374e-01, ..., -2.9822e-01,
-4.9385e-01, -6.0865e-01],
...,
[-5.6265e-01, -6.4174e-01, -7.2175e-01, ..., -3.7797e-01,
-5.7231e-01, -1.1887e+00],
[-3.1010e+00, -2.9252e+00, -2.2950e+00, ..., -1.9267e+00,
-2.0444e+00, -2.0571e+00],
[-5.2498e-01, -1.6371e+00, -1.1078e+00, ..., -8.7686e-01,
-9.5521e-01, -1.3520e+00]],
[[ 4.3223e-02, 2.8971e-01, 7.0483e-01, ..., 6.1597e-01,
9.5881e-01, 3.6024e-01],
[ 5.5907e-03, -4.0111e-01, 1.5901e-01, ..., 3.5981e-01,
5.8495e-01, -1.7877e-01],
[-2.9804e-01, -5.5593e-01, 2.0984e-01, ..., 3.2258e-01,
3.2454e-01, 1.5783e-02],
...,
[-7.2409e-02, -9.8471e-01, 2.3203e-01, ..., 2.7247e-01,
1.6999e-01, -7.2991e-01],
[-4.9270e-01, -1.7949e+00, -1.1589e-01, ..., 7.7376e-03,
-3.3723e-02, -9.3419e-01],
[ 4.3949e-01, -1.2334e-01, 1.1362e-01, ..., 2.4610e-01,
3.0605e-01, 3.2466e-01]],
[[-9.6138e-01, -2.3847e-01, 1.5968e-01, ..., 1.3307e-01,
5.2139e-02, 7.2711e-01],
[-1.1169e+00, -2.6440e-01, -3.1849e-01, ..., -3.4729e-01,
-1.0110e-01, 1.1458e+00],
[-7.1643e-01, -2.2272e-01, -2.4250e-01, ..., -2.0764e-01,
-1.1756e-01, 6.8200e-01],
...,
[-1.1621e+00, -9.2591e-01, -6.9873e-01, ..., -4.1942e-01,
-2.0741e-02, 8.2850e-01],
[-1.9130e+00, -1.8331e+00, -1.0306e+00, ..., -8.4288e-01,
-4.5994e-01, 4.2618e-01],
[-9.7608e-01, -9.7119e-01, -2.6299e-01, ..., -3.0673e-01,
-1.5232e-01, 1.1184e+00]]],
[[[-2.0184e-01, -2.5126e-01, 2.1246e-01, ..., 4.5520e-02,
5.0054e-01, 1.6564e-01],
[-1.0240e-01, -1.5227e+00, -8.7489e-01, ..., -7.1443e-01,
-4.4452e-01, -3.0597e-01],
[-9.5551e-02, -1.4549e+00, -1.0216e+00, ..., -4.8350e-01,
-3.9987e-01, -6.6967e-01],
...,
[-7.0401e-02, -1.7089e+00, -1.8395e+00, ..., -1.0368e+00,
-1.1183e+00, -1.9427e+00],
[-1.3131e-01, -1.3073e+00, -1.5018e+00, ..., -4.1522e-01,
-3.4168e-01, -1.4098e+00],
[-4.5221e-02, -1.0762e+00, -1.1264e+00, ..., -9.8853e-01,
-1.3217e+00, -8.8324e-01]],
[[-1.8218e-01, -1.4429e+00, -6.0253e-01, ..., -5.8296e-01,
-1.3296e+00, -2.1869e-01],
[-2.0331e+00, -4.0892e+00, -2.4598e+00, ..., -2.3199e+00,
-3.3083e+00, -1.1204e+00],
[-2.7270e+00, -6.0098e+00, -6.0284e+00, ..., -2.5107e+00,
-2.9142e+00, -1.2326e+00],
...,
[-3.5909e+00, -5.7820e+00, -6.3319e+00, ..., -5.4523e+00,
-5.9209e+00, -2.8457e+00],
[-5.1536e+00, -6.3712e+00, -3.2913e+00, ..., -5.7795e+00,
-6.9985e+00, -2.8312e+00],
[ 5.8020e-01, -2.3779e-01, -1.8063e-01, ..., 5.5339e-01,
6.8282e-01, 8.7445e-02]],
[[-5.3975e-01, -1.7049e-01, 3.4616e-01, ..., 3.3802e-01,
3.5552e-02, 8.6495e-01],
[-3.9515e-01, -4.5810e-01, 5.0517e-01, ..., 3.7860e-01,
2.6443e-01, 4.9329e-01],
[-8.9221e-01, -1.5891e-01, 5.6902e-01, ..., 5.5009e-02,
1.8249e-01, 4.0681e-01],
...,
[-1.7691e+00, -3.5557e-01, -3.8796e-02, ..., -4.1291e-02,
-6.5449e-02, -1.9092e-01],
[-1.4628e+00, -5.0887e-01, -8.4829e-01, ..., -1.8587e-01,
-1.7727e-01, -2.8322e-01],
[-5.8952e-01, 2.0942e-01, -7.0702e-01, ..., -1.0019e-01,
-9.2693e-02, -4.8326e-01]],
...,
[[ 7.1437e-02, -6.1490e-01, -4.0954e-01, ..., -9.2918e-02,
-2.5417e-01, -7.2712e-01],
[-8.4201e-01, -6.3349e-01, -5.7593e-01, ..., -2.7805e-01,
-9.6743e-01, -7.1756e-01],
[-4.6810e-01, -1.2074e-01, -2.9872e-01, ..., 1.0390e-01,
-1.8513e-01, -5.2168e-01],
...,
[-8.7530e-01, -8.1289e-01, -1.4377e-01, ..., -3.4070e-01,
-5.8332e-01, -1.2145e+00],
[-2.2639e+00, -1.4019e+00, -4.9454e-01, ..., -1.4745e+00,
-1.6497e+00, -1.6434e+00],
[-5.2027e-01, -9.6636e-01, -1.5832e+00, ..., -1.8933e-01,
-3.6117e-01, -5.2098e-01]],
[[-1.8240e-01, 3.4672e-02, 4.3264e-01, ..., 1.0699e-01,
5.5376e-01, 1.7974e-01],
[-4.2888e-01, -2.5833e-01, -8.0487e-03, ..., -6.3381e-02,
1.2041e-01, -4.7470e-02],
[-4.0960e-01, -1.0134e-01, 9.7871e-02, ..., 3.7110e-02,
3.2585e-02, -3.2123e-02],
...,
[ 5.7047e-02, -5.2579e-01, 4.9225e-01, ..., 6.1938e-01,
4.9674e-01, -3.3086e-01],
[-5.5495e-01, -1.7798e+00, -4.0237e-01, ..., 1.2686e-01,
8.4523e-02, -6.0598e-01],
[ 7.4081e-01, 9.7461e-02, 1.1470e-01, ..., 5.2355e-01,
4.5275e-01, 2.8341e-01]],
[[-6.8575e-01, 4.2446e-01, 5.7743e-01, ..., 4.8238e-01,
3.7369e-01, 5.9500e-01],
[-8.3563e-01, 3.1018e-01, -2.3645e-01, ..., -2.5465e-01,
-1.6096e-01, 6.4336e-01],
[-6.3428e-01, 1.6051e-01, -4.1531e-01, ..., 7.4454e-02,
9.4504e-02, 5.9832e-01],
...,
[-1.2455e+00, -5.7676e-01, -3.0892e-01, ..., -2.1005e-01,
-1.8334e-02, 1.3879e+00],
[-1.2018e+00, -1.0116e+00, -6.2775e-01, ..., -2.0169e-01,
5.3028e-02, 1.1668e+00],
[-5.3836e-01, -5.6140e-01, 1.1765e+00, ..., 7.5179e-03,
6.8217e-02, 1.2332e+00]]],
[[[-3.2112e-01, -4.7273e-01, -1.3331e-01, ..., -9.6632e-01,
-4.7461e-01, -1.3920e+00],
[-2.0771e-03, -1.4114e+00, -6.8635e-01, ..., -1.1762e+00,
-1.3136e+00, -1.4717e+00],
[-1.3964e-01, -1.3236e+00, -5.0664e-01, ..., -9.6213e-01,
-1.1017e+00, -1.8832e+00],
...,
[-2.9829e-01, -1.3090e+00, -4.6920e-01, ..., -1.1879e+00,
-1.0378e+00, -1.9124e+00],
[-4.7334e-01, -1.0908e+00, -8.3508e-02, ..., -4.5596e-01,
-5.8449e-01, -1.7273e+00],
[-4.9934e-01, -1.5434e+00, -1.2133e+00, ..., -9.7427e-01,
-1.3477e+00, -6.6984e-01]],
[[-6.8352e-01, -2.0171e+00, -8.4412e-01, ..., -3.9974e+00,
-5.4976e+00, -1.8299e+00],
[-2.2257e+00, -3.9999e+00, -2.0893e+00, ..., -5.9530e+00,
-8.2279e+00, -3.5356e+00],
[-1.8531e+00, -2.9519e+00, -1.6066e+00, ..., -5.1069e+00,
-6.3359e+00, -2.6779e+00],
...,
[-1.6531e+00, -2.7594e+00, -1.4968e+00, ..., -5.2072e+00,
-6.2609e+00, -2.4966e+00],
[-3.6685e+00, -4.5000e+00, -2.8342e+00, ..., -6.1965e+00,
-7.5675e+00, -2.4188e+00],
[ 5.8278e-01, -6.7527e-01, 4.5985e-01, ..., 3.9113e-01,
4.5524e-01, 2.3909e-01]],
[[-4.0205e-01, 7.2395e-01, 1.0295e+00, ..., 6.7329e-01,
3.1375e-01, 1.6133e+00],
[-8.0123e-01, 2.2049e-01, 5.1890e-01, ..., 5.5339e-01,
-1.9300e-01, 1.7918e-01],
[-1.2981e+00, 2.8793e-01, 1.7163e-01, ..., -1.9030e-02,
-1.9891e-01, -2.1351e-01],
...,
[-1.4252e+00, 3.0194e-01, 1.5745e-01, ..., 2.4384e-01,
-5.5601e-02, -1.3107e-01],
[-1.4013e+00, 3.2690e-01, 2.2720e-01, ..., -2.4928e-02,
-5.4354e-03, -2.0819e-01],
[-8.9810e-01, 8.3134e-02, -1.0631e-01, ..., 9.6070e-02,
2.1106e-01, -5.9536e-01]],
...,
[[ 3.9444e-01, -1.2212e-01, -1.1186e-01, ..., 4.7116e-01,
2.4848e-01, -1.3691e+00],
[-1.3976e+00, -1.0604e+00, -6.3525e-01, ..., -7.8039e-02,
-8.0354e-01, -1.4504e+00],
[-7.5676e-01, -4.8271e-01, -1.3293e-01, ..., -9.3291e-02,
-1.1911e-01, -9.8837e-01],
...,
[-9.6852e-01, -5.9785e-01, -1.1256e-01, ..., -1.4742e-01,
-1.2683e-01, -9.4606e-01],
[-2.5304e+00, -1.7477e+00, -8.7382e-01, ..., -1.3839e+00,
-1.4177e+00, -1.7327e+00],
[-6.4656e-01, -1.0996e+00, -3.3351e-02, ..., -3.8830e-01,
-5.0469e-01, -1.1262e+00]],
[[ 2.4731e-02, 2.3623e-01, 4.8019e-01, ..., -6.5701e-02,
4.3260e-01, -4.5297e-01],
[-1.8064e-01, -6.5565e-01, -2.2891e-02, ..., 2.6003e-01,
1.8918e-01, -8.5138e-01],
[-5.9327e-01, -1.0508e+00, -1.4162e-01, ..., 3.0278e-01,
2.5160e-01, -5.0707e-01],
...,
[-6.1307e-01, -1.1251e+00, -8.6946e-02, ..., -7.1683e-02,
2.0120e-01, -4.7938e-01],
[-1.0834e+00, -1.9618e+00, -3.9391e-01, ..., -1.1051e-01,
-1.8576e-01, -8.3393e-01],
[ 2.4213e-01, -4.6437e-01, -2.1276e-01, ..., 3.4370e-01,
3.6278e-01, 3.0690e-01]],
[[-3.6922e-01, 1.1134e-01, 5.2920e-01, ..., -1.1143e-03,
-3.5928e-02, 5.8899e-01],
[-4.0165e-01, -2.7964e-02, 6.7957e-03, ..., -9.6676e-01,
-3.7163e-01, 9.4339e-01],
[-1.1739e-02, -1.1609e-03, 6.5963e-02, ..., -6.8812e-01,
-4.6643e-01, 6.6244e-01],
...,
[ 2.8415e-02, -8.9669e-02, 6.7568e-02, ..., -4.9524e-01,
-5.3562e-01, 6.1420e-01],
[-3.5506e-01, -4.7231e-01, -1.2891e-01, ..., -6.4399e-01,
-6.4905e-01, 4.2921e-01],
[ 5.5977e-02, -4.9059e-01, -1.6037e-01, ..., -4.3863e-01,
-3.7014e-01, 1.0277e+00]]],
[[[-4.9566e-01, -1.5105e+00, -1.0174e+00, ..., -2.6897e+00,
8.5118e-01, 1.4598e-01],
[-2.3222e-01, -2.5586e+00, -1.8130e+00, ..., -5.1580e+00,
-1.5748e+00, -1.0837e-01],
[-1.3211e-01, -1.4785e+00, -1.0585e+00, ..., -3.0716e+00,
-1.4443e+00, -5.0743e-01],
...,
[-5.2650e+00, -1.1630e+00, -8.2994e+00, ..., -1.0866e+00,
-4.2603e-01, -4.9498e-01],
[-9.1547e-01, -6.2184e+00, -6.2578e+00, ..., -3.9463e-01,
1.0652e-02, -2.6674e-01],
[-1.0036e+00, -5.5912e+00, -8.9382e+00, ..., -1.0271e+00,
-1.1808e+00, -4.6820e-01]],
[[-2.7118e+00, -4.2970e+00, -3.1251e+00, ..., -8.4260e+00,
-8.8861e+00, -5.3167e-01],
[-4.9680e+00, -7.8153e+00, -7.3220e+00, ..., -8.0312e+00,
-1.1212e+01, -1.3168e+00],
[-3.9994e+00, -5.3923e+00, -6.0492e+00, ..., -6.2771e+00,
-8.1758e+00, -9.5754e-01],
...,
[-6.5673e+00, -6.2034e+00, 8.2361e-01, ..., -3.2134e+00,
-2.6943e+00, -1.0191e+00],
[-8.5103e+00, -1.5006e+01, -1.8033e+01, ..., -3.7401e+00,
-4.2958e+00, -1.4651e+00],
[-2.0508e+00, -3.1696e+00, -5.6274e+00, ..., 4.4501e-01,
4.7560e-01, 1.0462e-01]],
[[-5.8912e-01, 9.6197e-02, 7.4996e-01, ..., 3.2216e+00,
4.5516e+00, 1.3713e+00],
[-5.7027e-01, 1.6608e-01, 8.9569e-01, ..., -4.3964e+00,
3.8063e+00, 4.6609e-01],
[-1.2806e+00, -6.0547e-01, 8.0730e-02, ..., -4.7702e+00,
3.0396e+00, 4.5653e-01],
...,
[-1.2562e+01, -1.8449e+01, -2.7351e+01, ..., 4.0303e-01,
1.7795e-01, 4.0954e-01],
[-2.4558e+01, -2.2203e+01, -2.3547e+01, ..., 1.8370e-01,
1.6170e-01, 5.4147e-01],
[-1.3343e+01, -1.5607e+01, -1.3684e+01, ..., -9.9611e-02,
2.1058e-02, 2.5391e-01]],
...,
[[ 8.2879e-01, 1.9470e-01, 5.9230e-02, ..., -5.5124e+00,
-2.8881e+00, -1.0295e+00],
[-4.5976e-01, -3.1339e-01, -5.0619e-01, ..., -8.5042e+00,
-4.2394e+00, -1.1271e+00],
[ 2.0269e-01, -1.2107e-01, -3.9415e-01, ..., -6.6783e+00,
-2.6436e+00, -6.3934e-01],
...,
[ 2.7469e+00, -6.2175e+00, -9.7421e+00, ..., -6.5934e-01,
-3.6133e-01, -6.4083e-01],
[-3.6139e+00, -1.4161e+01, -1.3705e+01, ..., -9.9245e-01,
-1.0559e+00, -1.0098e+00],
[-7.0282e-01, -6.0162e+00, -9.0586e-01, ..., -2.5542e-01,
-2.8093e-01, -4.5418e-01]],
[[ 2.0099e-02, -5.9385e-01, -3.0034e-01, ..., 3.1405e+00,
2.7116e+00, 2.8860e-01],
[ 3.3047e-01, -7.9695e-01, -4.4710e-01, ..., 2.8382e+00,
6.6419e-01, -1.7208e-01],
[ 5.0982e-02, -3.2847e-01, 2.8276e-01, ..., 2.2011e+00,
-1.9499e-01, -9.8441e-02],
...,
[-3.5616e+00, -8.3400e-01, -4.7979e+00, ..., -3.0445e-01,
-8.7549e-02, -1.9355e-01],
[-8.9100e+00, -1.1312e+01, -1.0663e+01, ..., -5.9226e-01,
-3.8481e-01, -3.7771e-01],
[-4.3113e+00, -6.0124e+00, -9.0797e+00, ..., -9.2440e-02,
-1.4263e-01, 2.1394e-01]],
[[-6.9262e-01, -6.4751e-01, -4.6232e-01, ..., 1.3264e-02,
1.3716e+00, 8.0104e-01],
[-1.5502e+00, -1.6164e+00, -1.4874e+00, ..., 2.9405e+00,
3.2898e+00, 7.8246e-01],
[-1.6207e+00, -1.9672e+00, -1.3490e+00, ..., 1.8104e+00,
3.0430e+00, 3.6663e-01],
...,
[-4.9727e+00, -9.5832e+00, -1.4450e+01, ..., 1.0783e+00,
1.9997e-02, 3.4649e-01],
[-1.2206e+00, -4.5847e+00, -8.3833e+00, ..., 2.4168e-01,
-1.1359e-01, 7.4138e-02],
[-2.0295e+00, -1.0318e+00, -4.9619e+00, ..., -1.3568e-01,
-2.7091e-01, 3.2272e-01]]],
[[[ 2.4179e-02, -2.4490e-01, -4.9669e-02, ..., -9.7737e-01,
-5.1229e-01, -1.3786e+00],
[ 2.9388e-01, -9.1786e-01, -7.1706e-01, ..., -2.1259e+00,
-1.8876e+00, -1.7977e+00],
[ 2.3073e-01, -5.6882e-01, -4.9809e-01, ..., -2.4561e+00,
-3.8327e+00, -2.8887e+00],
...,
[-7.4508e-03, -5.9501e-01, -4.1605e-01, ..., -9.9274e-01,
-1.2135e+00, -1.9061e+00],
[-6.4470e-01, -6.9879e-01, -5.7164e-02, ..., -4.7226e-01,
-7.0118e-01, -1.6884e+00],
[-7.3425e-01, -1.3399e+00, -1.1796e+00, ..., -9.3815e-01,
-1.3195e+00, -6.4759e-01]],
[[-7.4739e-01, -2.1171e+00, -1.8338e+00, ..., -3.8427e+00,
-5.4343e+00, -1.7833e+00],
[-1.9873e+00, -3.7203e+00, -2.9871e+00, ..., -1.1011e+01,
-1.3747e+01, -5.9487e+00],
[-1.5990e+00, -2.5939e+00, -2.5482e+00, ..., -1.1794e+01,
-1.4269e+01, -6.8921e+00],
...,
[-1.7622e+00, -2.4047e+00, -2.4094e+00, ..., -5.5836e+00,
-6.6206e+00, -2.5144e+00],
[-3.7972e+00, -4.4335e+00, -3.6301e+00, ..., -6.1216e+00,
-7.5703e+00, -2.4009e+00],
[ 2.8865e-01, -3.1593e-01, 3.9161e-01, ..., 4.1326e-01,
4.7059e-01, 2.2006e-01]],
[[-2.2053e-01, 1.5575e-01, 7.1854e-01, ..., 7.0703e-01,
2.7823e-01, 1.5799e+00],
[ 4.1315e-02, -3.6040e-01, 5.9059e-01, ..., 5.1616e-01,
-1.8540e-01, 2.7495e-02],
[-1.2667e-01, -2.2710e-01, 2.1505e-01, ..., -2.2591e+00,
-1.0902e+00, -4.2271e-02],
...,
[-2.5910e-01, -1.5575e-01, 1.2848e-01, ..., -7.7781e-01,
6.0989e-02, -1.6373e-01],
[-2.6590e-01, -1.1225e-01, 1.8229e-01, ..., -2.4089e-01,
5.2613e-02, -1.8589e-01],
[-1.5573e-01, -4.8718e-01, 1.2951e-01, ..., 1.0941e-01,
1.7172e-01, -6.5820e-01]],
...,
[[ 5.9895e-01, -2.4097e-01, 2.4495e-02, ..., 3.8614e-01,
2.7758e-01, -1.3496e+00],
[-4.9289e-01, -5.1814e-01, -5.4024e-01, ..., -2.0720e-01,
-8.2667e-01, -1.3546e+00],
[-3.7124e-01, -1.9320e-01, -2.3805e-01, ..., 1.9679e-01,
-1.5802e-01, -2.4528e+00],
...,
[-4.5340e-01, -1.5625e-01, -1.4178e-01, ..., -9.9930e-01,
-4.6484e-01, -9.4273e-01],
[-2.3541e+00, -1.7263e+00, -1.2334e+00, ..., -1.3398e+00,
-1.2730e+00, -1.7226e+00],
[-4.6513e-01, -7.9364e-01, -3.8590e-01, ..., -3.6948e-01,
-5.1473e-01, -1.1271e+00]],
[[ 1.6224e-01, 2.5577e-01, 6.1764e-01, ..., -1.3025e-01,
3.8720e-01, -3.6354e-01],
[ 1.5235e-01, 2.5170e-01, 4.5135e-01, ..., -3.6121e-01,
-9.5790e-01, -2.2665e+00],
[-1.1608e-01, 1.0387e-01, 2.8655e-01, ..., -6.6890e-01,
-1.0865e+00, -2.5010e+00],
...,
[-1.9183e-01, -3.4021e-02, 3.2265e-01, ..., 7.2300e-01,
1.1232e-01, -5.1147e-01],
[-5.4301e-01, -7.3195e-01, -4.0355e-02, ..., -2.7324e-02,
-1.8586e-01, -8.0197e-01],
[ 5.2572e-01, 3.9127e-01, 7.2770e-02, ..., 3.8231e-01,
3.7092e-01, 3.1544e-01]],
[[-8.5359e-01, 2.4900e-01, 3.1447e-01, ..., -2.8988e-02,
-7.5152e-02, 5.9852e-01],
[-1.1151e+00, 1.9041e-01, -3.6481e-01, ..., -1.0028e+00,
-5.0238e-01, 4.8224e-01],
[-9.4675e-01, 3.3161e-01, -1.0813e-01, ..., -3.2192e+00,
-3.0217e+00, -1.0785e+00],
...,
[-9.0359e-01, 1.9113e-01, -9.7950e-02, ..., -2.3615e-01,
-2.3877e-01, 6.3130e-01],
[-1.6600e+00, -3.9020e-01, -3.6351e-01, ..., -3.7186e-01,
-5.5349e-01, 4.5761e-01],
[-7.0618e-01, -8.1639e-02, -3.0113e-01, ..., -4.2528e-01,
-4.0595e-01, 1.0077e+00]]]], device='cuda:0'),
tensor([[[[-1.8422e-01, -2.4205e-01, 1.8714e-01, ..., 5.7385e-02,
5.3128e-01, 1.8377e-01],
[-9.7653e-02, -1.5357e+00, -9.1270e-01, ..., -7.2647e-01,
-4.2818e-01, -3.1858e-01],
[-1.5554e-01, -1.3605e+00, -6.2199e-01, ..., -5.0231e-01,
-3.7582e-01, -6.5150e-01],
...,
[-6.4344e-02, -1.5312e+00, -1.2443e+00, ..., -9.1909e-01,
-9.4735e-01, -2.0166e+00],
[-1.0659e+00, -1.6988e+00, -7.9947e-01, ..., -5.3498e-01,
-4.3476e-01, -1.7646e+00],
[-1.2888e+00, -2.1614e+00, -1.6155e+00, ..., -1.4145e+00,
-1.5418e+00, -9.6973e-01]],
[[-1.2564e-01, -1.4901e+00, -6.7550e-01, ..., -5.7175e-01,
-1.3236e+00, -1.9960e-01],
[-2.0589e+00, -4.1059e+00, -2.4790e+00, ..., -2.3551e+00,
-3.3700e+00, -1.1222e+00],
[-1.8809e+00, -3.4129e+00, -2.5636e+00, ..., -2.5423e+00,
-2.9282e+00, -1.1923e+00],
...,
[-3.8050e+00, -5.4425e+00, -5.2250e+00, ..., -5.3704e+00,
-6.1369e+00, -2.9070e+00],
[-6.1907e+00, -7.0245e+00, -6.0501e+00, ..., -6.0202e+00,
-7.3529e+00, -2.6598e+00],
[-1.9596e-02, -8.7205e-01, 2.8584e-01, ..., 4.2236e-01,
7.1006e-01, 2.1699e-01]],
[[-5.4723e-01, -1.8664e-01, 3.9044e-01, ..., 3.5942e-01,
4.3494e-02, 8.6445e-01],
[-4.0240e-01, -5.7376e-01, 4.3087e-01, ..., 3.8294e-01,
2.3522e-01, 4.9612e-01],
[-8.6228e-01, -2.9436e-01, 1.0060e-01, ..., 9.7168e-02,
1.8896e-01, 4.2036e-01],
...,
[-1.0427e+00, 1.9985e-01, 3.7733e-02, ..., 7.7470e-02,
1.4695e-02, 2.7945e-01],
[-1.2049e+00, 1.0010e-01, -1.9806e-02, ..., 4.4347e-03,
-2.4312e-03, 2.9846e-01],
[-1.1399e+00, -8.0606e-02, 7.4784e-02, ..., 1.7384e-02,
-1.6538e-02, -3.4011e-01]],
...,
[[ 6.9495e-02, -6.1499e-01, -3.9493e-01, ..., -6.2589e-02,
-2.5170e-01, -7.3768e-01],
[-8.4594e-01, -5.9560e-01, -5.2201e-01, ..., -2.8722e-01,
-9.9050e-01, -7.2856e-01],
[-4.6220e-01, -7.3800e-02, -5.2124e-02, ..., 8.6101e-02,
-1.8156e-01, -5.1948e-01],
...,
[-7.8501e-01, -9.1454e-01, -7.3792e-01, ..., -3.4640e-01,
-5.8317e-01, -1.1808e+00],
[-3.1107e+00, -2.8849e+00, -2.0449e+00, ..., -1.8989e+00,
-2.0129e+00, -2.0512e+00],
[-5.5840e-01, -1.6109e+00, -1.1011e+00, ..., -8.7554e-01,
-9.3391e-01, -1.3147e+00]],
[[-1.6139e-01, 4.8922e-02, 4.2608e-01, ..., 1.0210e-01,
5.5796e-01, 1.8278e-01],
[-4.3058e-01, -2.0632e-01, 2.7206e-03, ..., -7.8355e-02,
1.3804e-01, -9.5972e-02],
[-6.7897e-01, -5.3128e-01, 7.8038e-02, ..., 2.2540e-02,
2.7676e-02, -1.1054e-02],
...,
[ 2.0144e-02, -1.0034e+00, -7.8742e-02, ..., 2.7718e-01,
1.6954e-01, -7.0532e-01],
[-5.0396e-01, -1.7990e+00, -1.8821e-01, ..., 6.9172e-03,
-3.6359e-02, -9.3485e-01],
[ 4.3003e-01, -1.5871e-01, 1.1099e-01, ..., 2.7500e-01,
3.1087e-01, 3.2690e-01]],
[[-6.9384e-01, 4.2475e-01, 5.8669e-01, ..., 4.8603e-01,
3.7506e-01, 5.9858e-01],
[-8.4987e-01, 3.0497e-01, -2.0883e-01, ..., -2.6812e-01,
-1.5881e-01, 6.3013e-01],
[-5.0531e-01, 2.9325e-01, -2.2929e-04, ..., 7.1621e-02,
6.7527e-02, 5.7135e-01],
...,
[-1.1132e+00, -7.0574e-01, -1.6268e-01, ..., -4.1863e-01,
-6.4878e-02, 8.1035e-01],
[-1.8319e+00, -1.8164e+00, -9.7412e-01, ..., -8.2230e-01,
-4.7666e-01, 4.1663e-01],
[-9.4329e-01, -9.5863e-01, -2.7254e-01, ..., -2.7902e-01,
-1.4366e-01, 1.1278e+00]]],
[[[ 5.5875e-01, -1.1016e+00, -1.4740e+00, ..., -9.7126e-01,
-5.3883e-01, -1.4711e+00],
[ 2.7335e-01, -2.7849e+00, -2.7665e-01, ..., -1.2020e+00,
-1.3422e+00, -1.5244e+00],
[-5.4529e-01, -3.5938e+00, -1.5418e+00, ..., -9.4264e-01,
-1.1081e+00, -1.8600e+00],
...,
[-1.0725e-01, -3.6608e-01, -4.0000e-01, ..., -3.6747e+00,
-9.8775e-01, -1.9009e+00],
[-4.4799e-01, -1.6277e-01, -4.5535e-01, ..., -3.8684e+00,
6.8818e-01, -1.3774e+00],
[-2.7868e-01, -2.8595e-01, -4.5621e-01, ..., -1.3433e+00,
-1.8681e+00, -7.4694e-01]],
[[-3.2124e+00, -4.9239e+00, -5.1843e+00, ..., -3.9371e+00,
-5.5147e+00, -1.8355e+00],
[-4.0496e+00, -7.2144e+00, -6.3584e+00, ..., -6.0476e+00,
-8.2706e+00, -3.4800e+00],
[-4.5539e+00, -6.9931e+00, -5.9659e+00, ..., -5.5138e+00,
-6.3653e+00, -2.6194e+00],
...,
[-1.2229e+00, -2.8452e+00, -3.5027e+00, ..., -1.0927e+01,
-1.1654e+01, -3.2231e+00],
[-3.7510e+00, -5.1256e+00, -4.9221e+00, ..., -8.6701e+00,
-1.2526e+01, -3.1006e+00],
[ 4.0131e-01, -9.2769e-01, -1.5960e-01, ..., -1.4858e+00,
-1.5651e+00, 3.1268e-01]],
[[-3.5982e-01, -2.3496e-01, -1.2459e+00, ..., 6.3248e-01,
2.3789e-01, 1.5215e+00],
[-9.2747e-01, -3.6461e+00, -5.3916e+00, ..., 5.9594e-01,
-9.6512e-02, 6.3987e-02],
[-1.8845e+00, -9.7616e+00, -1.3328e+01, ..., 1.0946e-01,
-2.9093e-02, -1.3788e-01],
...,
[-1.5051e+00, -1.1765e+00, -5.7471e-01, ..., -7.1188e+00,
-1.2949e+00, 1.6559e-01],
[-5.6339e-01, -1.2996e+00, -9.3382e-01, ..., -7.7612e+00,
-5.5683e-01, -9.4874e-02],
[-6.4306e-01, -4.2068e-01, -5.2102e-01, ..., -5.9498e+00,
7.3960e-01, -4.3729e-01]],
...,
[[ 6.6569e-01, 8.8465e-01, 1.4471e+00, ..., 4.6353e-01,
3.5165e-01, -1.3336e+00],
[ 3.7610e-01, -2.4849e-01, -3.8559e+00, ..., -1.7877e-01,
-8.0487e-01, -1.4284e+00],
[-6.6084e-01, -2.4816e+00, -5.9433e+00, ..., -1.2617e-02,
-1.1011e-01, -9.6333e-01],
...,
[ 3.2906e-01, -1.4441e+00, -2.3531e+00, ..., -1.0595e+01,
-5.0654e+00, -1.3058e+00],
[-1.4116e+00, -2.1585e+00, -1.8642e+00, ..., -8.5828e+00,
-8.3816e+00, -2.5699e+00],
[-3.9245e-01, -1.4867e+00, -1.1640e+00, ..., -7.6804e+00,
-5.1401e+00, -8.9168e-01]],
[[ 8.2225e-01, 3.4492e-01, -1.9541e-01, ..., -4.2162e-02,
4.2224e-01, -3.6658e-01],
[ 4.0140e-01, -1.0537e+00, -2.8283e+00, ..., 2.0689e-01,
9.6714e-02, -8.7431e-01],
[-6.3286e-01, -4.1499e+00, -6.6258e+00, ..., 5.0199e-02,
1.6416e-01, -4.8723e-01],
...,
[-3.3661e-01, 9.8468e-02, 7.0631e-01, ..., -3.7895e+00,
-1.4704e+00, -1.5548e+00],
[-9.8100e-01, -8.0306e-01, -5.8493e-01, ..., -2.0913e+00,
1.1395e+00, -8.5705e-01],
[-8.6832e-02, 3.9603e-01, 3.1108e-01, ..., 1.9072e+00,
1.7895e+00, 5.2679e-01]],
[[-1.0237e+00, -7.1388e-01, -2.3661e+00, ..., -1.1620e-01,
-8.2326e-02, 5.6146e-01],
[-2.4383e+00, -2.9821e+00, -6.9144e+00, ..., -9.1333e-01,
-3.9551e-01, 9.0238e-01],
[-1.6659e+00, -3.0804e+00, -7.6517e+00, ..., -5.1769e-01,
-5.1357e-01, 6.5479e-01],
...,
[-1.7949e+00, -4.6219e-01, -1.8412e+00, ..., -8.2028e-01,
-6.5210e-01, 1.1711e+00],
[-1.3538e+00, -1.1241e-02, -2.9682e-01, ..., -3.6768e+00,
-8.4031e-01, 8.4352e-01],
[-1.0578e+00, -6.1977e-01, 3.4285e-02, ..., -6.1417e-01,
8.4960e-01, 1.3148e+00]]],
[[[-2.1744e-01, -2.9362e-01, 1.8299e-01, ..., 3.9678e-02,
4.7752e-01, 1.4352e-01],
[-1.2511e-01, -1.5407e+00, -8.6372e-01, ..., -7.2695e-01,
-4.4992e-01, -3.1315e-01],
[-1.3033e-01, -1.2445e+00, -9.4115e-01, ..., -5.0219e-01,
-4.5077e-01, -7.2289e-01],
...,
[-2.5414e-01, -1.7329e+00, -1.2191e+00, ..., -1.2685e+00,
-1.0341e+00, -2.0832e+00],
[-1.0959e+00, -1.7545e+00, -8.0162e-01, ..., -5.3634e-01,
-4.9581e-01, -1.8275e+00],
[-1.2675e+00, -2.1762e+00, -1.6215e+00, ..., -1.3994e+00,
-1.5194e+00, -9.7840e-01]],
[[-2.4405e-01, -1.4888e+00, -6.6496e-01, ..., -6.8079e-01,
-1.4060e+00, -2.6659e-01],
[-2.0178e+00, -4.0539e+00, -2.4334e+00, ..., -2.3625e+00,
-3.4071e+00, -1.1636e+00],
[-1.9167e+00, -4.5883e+00, -4.5114e+00, ..., -2.6195e+00,
-3.0498e+00, -1.2906e+00],
...,
[-3.7152e+00, -5.5017e+00, -5.3384e+00, ..., -5.6830e+00,
-6.2044e+00, -2.9187e+00],
[-6.1945e+00, -7.0388e+00, -6.0422e+00, ..., -6.0616e+00,
-7.3841e+00, -2.7517e+00],
[-2.1030e-02, -8.7445e-01, 2.5968e-01, ..., 4.3729e-01,
7.2749e-01, 1.7608e-01]],
[[-5.2620e-01, -1.5135e-01, 3.5863e-01, ..., 3.4216e-01,
6.7996e-02, 8.9851e-01],
[-3.9617e-01, -4.5000e-01, 4.6329e-01, ..., 3.7320e-01,
2.7418e-01, 4.8500e-01],
[-8.9786e-01, -3.6995e-01, -1.2807e-01, ..., 2.9332e-02,
1.4749e-01, 3.9878e-01],
...,
[-1.4000e+00, 2.1858e-02, 3.3253e-01, ..., 4.6097e-01,
1.4970e-03, 2.9386e-01],
[-1.2268e+00, 1.4971e-01, -1.4699e-03, ..., -1.2369e-02,
1.4918e-03, 2.8033e-01],
[-1.1982e+00, -4.6599e-02, 1.2421e-01, ..., 3.3471e-03,
-7.0892e-02, -3.9158e-01]],
...,
[[ 4.3334e-02, -5.9126e-01, -3.7579e-01, ..., -5.6026e-02,
-2.2563e-01, -7.2581e-01],
[-8.3192e-01, -5.7771e-01, -5.3903e-01, ..., -2.5288e-01,
-9.7407e-01, -7.4548e-01],
[-5.5675e-01, -1.5101e-01, -1.9732e-02, ..., 8.5047e-02,
-1.8738e-01, -5.2125e-01],
...,
[-6.3972e-01, -6.2702e-01, -5.1401e-01, ..., -2.9093e-01,
-5.7700e-01, -1.2329e+00],
[-3.1266e+00, -2.9043e+00, -1.9868e+00, ..., -1.8477e+00,
-2.0060e+00, -2.0919e+00],
[-5.2925e-01, -1.6108e+00, -1.0788e+00, ..., -8.3287e-01,
-8.9024e-01, -1.2969e+00]],
[[-1.8837e-01, 2.9101e-02, 4.3482e-01, ..., 1.0030e-01,
5.3513e-01, 1.5567e-01],
[-4.4880e-01, -2.7021e-01, 2.9788e-02, ..., -2.3534e-02,
1.4962e-01, -6.2611e-02],
[-6.7349e-01, -3.4714e-01, 3.3509e-01, ..., 3.5541e-02,
2.5707e-02, -6.0819e-02],
...,
[-1.3034e-01, -1.0295e+00, 1.7585e-01, ..., 7.4217e-02,
1.3346e-01, -7.2551e-01],
[-5.3097e-01, -1.8764e+00, -2.3979e-01, ..., 4.7528e-02,
-9.1510e-03, -9.2957e-01],
[ 4.4275e-01, -1.5847e-01, 1.4210e-01, ..., 3.2455e-01,
3.7796e-01, 3.4941e-01]],
[[-6.8572e-01, 3.8687e-01, 5.3919e-01, ..., 4.7199e-01,
3.7732e-01, 6.1231e-01],
[-8.4900e-01, 2.7617e-01, -2.1568e-01, ..., -2.6883e-01,
-1.6599e-01, 6.6450e-01],
[-4.8507e-01, 2.9176e-01, -8.1333e-02, ..., 6.3194e-02,
8.5525e-02, 5.9105e-01],
...,
[-1.0971e+00, -9.3985e-01, -7.2435e-01, ..., -9.0755e-02,
-7.3476e-02, 8.1287e-01],
[-1.8004e+00, -1.8113e+00, -1.0457e+00, ..., -8.2008e-01,
-4.2036e-01, 4.6402e-01],
[-9.4079e-01, -9.4059e-01, -2.6584e-01, ..., -2.8576e-01,
-1.0632e-01, 1.1453e+00]]],
[[[-3.8205e-01, -5.3230e-01, -1.8645e+00, ..., -5.3845e-01,
-8.6973e-02, -8.1693e-01],
[-7.2195e-02, -1.1438e+00, -1.1044e+00, ..., -1.1679e+00,
-8.9115e-01, -1.2131e+00],
[-1.7472e-02, -9.7156e-01, 1.3774e+00, ..., -1.3319e+00,
-9.0195e-01, -1.7790e+00],
...,
[ 7.8256e-02, -5.8731e-01, -2.5364e-01, ..., 9.0531e-01,
-2.7054e-01, -5.0781e-01],
[-3.5721e-01, -4.8285e-01, 4.1598e-02, ..., -4.2997e-01,
-1.0676e-01, -2.4279e-01],
[-6.1385e-01, -1.3256e+00, -1.1035e+00, ..., -1.0314e+00,
-1.1683e+00, -5.1293e-01]],
[[-2.0369e+00, -5.8168e+00, -1.1685e+01, ..., -4.2576e+00,
-5.0188e+00, -2.0197e+00],
[-3.7972e+00, -7.2813e+00, -1.3470e+01, ..., -5.9582e+00,
-6.9993e+00, -3.0220e+00],
[-3.5450e+00, -5.2661e+00, -5.1297e+00, ..., -6.4251e+00,
-6.0994e+00, -2.8605e+00],
...,
[-1.8451e+00, -2.3538e+00, -1.9779e+00, ..., -4.2593e+00,
-2.7973e+00, -9.8830e-01],
[-2.9140e+00, -3.1295e+00, -2.1026e+00, ..., -3.9540e+00,
-3.4833e+00, -1.0838e+00],
[ 3.5825e-01, -2.4952e-01, 3.3917e-01, ..., 3.3374e-01,
4.7792e-01, 2.0594e-01]],
[[-4.0640e-01, 2.5174e-01, -2.3595e+00, ..., 9.8326e-01,
5.5478e-01, 1.5271e+00],
[-6.0203e-01, 2.2335e-01, -4.0680e+00, ..., 5.6241e-01,
2.3387e-01, 4.2116e-01],
[-1.1518e+00, 8.1576e-01, -1.2331e+00, ..., 2.8436e-01,
1.1410e-01, 2.9770e-01],
...,
[-6.7659e-01, 4.1070e-02, 2.3465e-01, ..., 6.2892e-01,
1.0927e+00, 4.0489e-01],
[-4.7374e-01, 6.8898e-02, 2.1404e-01, ..., 8.4935e-01,
5.4108e-01, 5.9513e-01],
[-1.5609e-01, -2.3226e-01, 1.3750e-01, ..., -4.6290e-03,
9.6466e-02, 3.6367e-01]],
...,
[[ 1.1019e+00, 9.2835e-01, 7.1404e-01, ..., 2.5579e-01,
2.0887e-01, -1.0405e+00],
[-6.0869e-01, 5.5580e-02, 2.8688e+00, ..., -6.0386e-01,
-1.2487e+00, -1.3747e+00],
[-1.7820e-01, -5.8881e-01, 1.3969e+00, ..., -6.4416e-02,
-5.4131e-01, -1.1315e+00],
...,
[-6.6094e-01, -5.2950e-01, -3.5420e-01, ..., -5.1084e+00,
-1.3031e+00, -5.9105e-01],
[-2.2667e+00, -1.7987e+00, -1.2700e+00, ..., -2.8398e+00,
-1.0837e+00, -1.1040e+00],
[-6.3252e-01, -8.8230e-01, -4.6586e-01, ..., -3.0441e-01,
-3.5276e-01, -3.9691e-01]],
[[ 1.8387e-01, 5.4752e-01, -6.2970e-01, ..., 2.6332e-01,
7.2848e-01, -2.8364e-01],
[ 4.1255e-01, -4.2656e-01, -6.1226e-01, ..., 8.4986e-02,
3.7696e-01, -7.7769e-01],
[ 4.3238e-02, -2.9314e-01, -5.3990e-01, ..., -5.2118e-01,
1.2050e-02, -6.7573e-01],
...,
[-2.0872e-01, -4.5965e-01, 2.4269e-01, ..., 2.0455e+00,
1.5628e-01, 3.5844e-03],
[-5.2229e-01, -9.4554e-01, 4.9607e-02, ..., -5.4491e-01,
-3.1474e-01, -2.1174e-01],
[ 3.3885e-01, -3.5995e-02, -1.8691e-01, ..., -1.0298e-01,
-6.7750e-02, 1.5314e-01]],
[[-6.5971e-01, -5.5143e-01, -1.2359e+00, ..., 3.4465e-01,
3.9318e-01, 9.3911e-01],
[-1.2296e+00, -1.3778e+00, -1.8922e+00, ..., -4.5847e-01,
-2.6875e-01, 9.6847e-01],
[-9.5005e-01, -1.4012e+00, -5.4968e-01, ..., -4.4108e-01,
-6.3543e-02, 7.3855e-01],
...,
[-6.9246e-01, -2.5052e-01, -1.2829e-01, ..., 1.4466e-01,
2.8966e-01, 6.2897e-01],
[-1.1553e+00, -5.8957e-01, -3.3738e-01, ..., 8.5263e-01,
1.2135e-01, 2.1983e-01],
[-3.8405e-01, -5.7913e-02, -1.8644e-01, ..., -2.2765e-01,
-1.9902e-01, 3.5297e-01]]],
[[[-2.0263e-02, -4.1386e-01, -2.4735e-01, ..., -1.0758e+00,
-5.1126e-01, -1.5454e+00],
[ 1.9802e-01, -1.0982e+00, -6.6477e-01, ..., -1.4081e+00,
-1.3721e+00, -1.5819e+00],
[ 2.3103e-01, -7.1613e-01, 3.0162e-01, ..., -1.1649e+00,
-1.1552e+00, -1.8923e+00],
...,
[ 7.5751e-02, 5.6655e-01, 1.5264e+00, ..., -1.6326e+00,
-4.9599e+00, -4.1892e+00],
[-4.6228e-01, -2.3466e-01, 2.3071e+00, ..., -3.0875e-02,
-1.7642e+00, -2.1623e+00],
[-6.0309e-01, -1.1269e+00, 1.5150e-01, ..., -1.0490e+00,
-1.4967e+00, -1.0904e+00]],
[[-1.0545e+00, -2.6484e+00, -2.6356e+00, ..., -3.9155e+00,
-5.5323e+00, -1.7651e+00],
[-2.4451e+00, -4.2570e+00, -4.9650e+00, ..., -6.2382e+00,
-8.5877e+00, -3.4548e+00],
[-2.1044e+00, -3.1486e+00, -9.5483e+00, ..., -5.1491e+00,
-6.5564e+00, -2.4363e+00],
...,
[-2.1211e+00, -3.3861e+00, -7.2387e+00, ..., -1.0530e+01,
-1.4221e+01, -5.0126e+00],
[-4.0516e+00, -5.0205e+00, -3.9690e+00, ..., -7.0969e+00,
-1.0275e+01, -3.5926e+00],
[ 4.1729e-01, -3.2162e-01, 3.7146e-01, ..., 1.2425e-02,
2.6110e-02, -1.2288e-01]],
[[-1.6954e-01, 2.6251e-01, 7.1204e-01, ..., 5.8623e-01,
1.1227e-01, 1.5592e+00],
[ 1.2595e-03, -2.8537e-01, 1.3111e-01, ..., 5.3718e-01,
-2.8486e-01, -2.2956e-02],
[-3.1693e-01, -2.1306e-01, -3.2719e-01, ..., -1.3149e-02,
-1.4915e-01, -1.7914e-01],
...,
[-4.5537e-01, 1.1508e+00, -5.5870e+00, ..., -4.4130e+00,
-4.5151e+00, 6.8581e-01],
[-3.2756e-01, 7.7775e-01, -4.7946e-01, ..., -8.8689e-01,
-4.2757e-01, -2.4033e-02],
[ 4.2711e-03, 5.7173e-02, 1.3547e+00, ..., 3.8824e-01,
2.8918e-01, -7.6614e-01]],
...,
[[ 7.2083e-01, 7.6206e-02, 2.9727e-01, ..., 5.3019e-01,
4.9353e-01, -1.3559e+00],
[-4.4329e-01, -3.7403e-01, -3.2062e-01, ..., -2.4844e-01,
-7.8848e-01, -1.4542e+00],
[-2.5677e-01, -1.0951e-01, 5.8821e-01, ..., -9.4291e-02,
-5.1321e-02, -9.6807e-01],
...,
[-4.6288e-01, -5.2174e-02, 1.2852e+00, ..., -3.0064e-02,
-1.2804e+00, -4.8573e+00],
[-2.2848e+00, -2.2332e+00, -1.3051e+00, ..., -1.7404e+00,
-2.4326e+00, -5.1582e+00],
[-4.6664e-01, -1.2021e+00, -1.7956e+00, ..., -1.1797e+00,
-1.6503e+00, -2.3041e+00]],
[[ 2.1419e-01, 2.7837e-01, 5.3434e-01, ..., 3.8647e-02,
5.0142e-01, -1.6395e-01],
[ 1.8270e-01, 1.5940e-01, 3.5123e-01, ..., 2.1654e-01,
1.1110e-01, -7.7314e-01],
[-8.1066e-02, 3.2651e-02, 1.5835e+00, ..., 3.3963e-01,
1.8143e-01, -2.6700e-01],
...,
[-1.2825e-01, 4.9179e-01, -7.3516e-01, ..., -2.5987e+00,
-3.5320e+00, -2.0112e+00],
[-5.7496e-01, -5.7772e-01, -1.6600e-01, ..., -8.9080e-02,
-1.9222e+00, -2.0960e+00],
[ 6.1619e-01, 4.7314e-01, 9.3425e-01, ..., 3.8510e-01,
-5.4871e-01, -3.8570e-01]],
[[-7.4509e-01, 1.6489e-01, 2.7893e-01, ..., -3.3874e-01,
-2.4199e-01, 5.5494e-01],
[-1.1238e+00, 3.7649e-02, -3.4471e-01, ..., -1.0360e+00,
-4.6174e-01, 1.0221e+00],
[-9.6849e-01, 1.8327e-01, -2.1511e-01, ..., -7.8029e-01,
-7.5473e-01, 6.6419e-01],
...,
[-9.2554e-01, -5.3985e-01, 3.3232e-01, ..., 2.5194e-01,
-1.7732e+00, -2.3814e+00],
[-1.5511e+00, -1.0139e+00, -3.9607e-01, ..., -1.2481e+00,
-1.3969e+00, -3.0830e-01],
[-7.2064e-01, -4.0099e-01, -1.5732e-01, ..., -3.8852e-01,
7.9667e-02, 1.6944e+00]]]], device='cuda:0'),
tensor([[[[-3.0867e-01, -5.0221e-01, -1.4357e-01, ..., -9.6993e-01,
-5.0428e-01, -1.4258e+00],
[-8.3401e-03, -1.3634e+00, -6.8684e-01, ..., -1.1511e+00,
-1.3083e+00, -1.5016e+00],
[-2.5509e-01, -1.3104e+00, -5.0238e-01, ..., -9.6253e-01,
-1.1047e+00, -1.8258e+00],
...,
[-2.8570e-01, -1.3226e+00, -4.4590e-01, ..., -1.0080e+00,
-9.9391e-01, -1.9233e+00],
[-4.3377e-01, -1.0820e+00, -8.1549e-02, ..., -4.2893e-01,
-5.5006e-01, -1.7091e+00],
[-5.6911e-01, -1.5895e+00, -1.2623e+00, ..., -1.1158e+00,
-1.5207e+00, -7.6318e-01]],
[[-6.5330e-01, -1.8876e+00, -7.8823e-01, ..., -3.9665e+00,
-5.5095e+00, -1.8715e+00],
[-2.1702e+00, -3.8592e+00, -2.0422e+00, ..., -5.8999e+00,
-8.1121e+00, -3.4512e+00],
[-2.0465e+00, -2.9971e+00, -1.7186e+00, ..., -5.3652e+00,
-6.2172e+00, -2.5957e+00],
...,
[-1.6279e+00, -2.6896e+00, -1.4784e+00, ..., -5.0578e+00,
-6.2729e+00, -2.4766e+00],
[-3.7330e+00, -4.5442e+00, -2.9048e+00, ..., -6.4559e+00,
-7.8923e+00, -2.5227e+00],
[ 5.8773e-01, -6.7585e-01, 4.1004e-01, ..., 4.3166e-01,
4.7079e-01, 2.4370e-01]],
[[-3.3209e-01, 7.4912e-01, 1.0037e+00, ..., 6.4004e-01,
2.8371e-01, 1.5256e+00],
[-7.2581e-01, 2.6413e-01, 5.2424e-01, ..., 6.0705e-01,
-7.3370e-02, 1.1510e-01],
[-1.2970e+00, 3.6915e-01, 2.5380e-01, ..., 8.3939e-02,
3.1987e-02, -1.0621e-01],
...,
[-1.3995e+00, 3.0110e-01, 1.6060e-01, ..., -4.7111e-02,
-1.1085e-01, -1.6280e-01],
[-1.4039e+00, 3.2062e-01, 2.1667e-01, ..., -1.1049e-01,
-2.8133e-02, -1.7654e-01],
[-9.1628e-01, 5.0892e-02, -1.2874e-01, ..., 6.4030e-02,
2.3219e-01, -5.1645e-01]],
...,
[[ 4.4531e-01, -7.7786e-02, -1.0896e-01, ..., 4.6602e-01,
3.5653e-01, -1.3072e+00],
[-1.2870e+00, -1.0541e+00, -6.1871e-01, ..., -1.8685e-01,
-8.2142e-01, -1.3874e+00],
[-6.0678e-01, -4.8990e-01, -9.7694e-02, ..., -1.7260e-02,
-1.2585e-01, -9.5791e-01],
...,
[-9.5932e-01, -5.8237e-01, -8.8634e-02, ..., -9.9104e-02,
-9.0510e-02, -9.0334e-01],
[-2.5354e+00, -1.7356e+00, -8.9325e-01, ..., -1.5563e+00,
-1.5561e+00, -1.7827e+00],
[-6.3904e-01, -1.0868e+00, -7.5508e-02, ..., -3.3989e-01,
-5.5386e-01, -1.2138e+00]],
[[ 5.9897e-02, 2.5211e-01, 4.6963e-01, ..., -9.8229e-02,
4.0150e-01, -3.7863e-01],
[-1.2037e-01, -6.8808e-01, -5.1764e-02, ..., 1.9502e-01,
1.2043e-01, -8.6867e-01],
[-5.5201e-01, -1.1680e+00, -2.4420e-01, ..., -2.8227e-02,
1.8106e-01, -4.7047e-01],
...,
[-5.8248e-01, -1.1016e+00, -1.1657e-01, ..., 1.3819e-01,
1.4875e-01, -4.8180e-01],
[-1.0655e+00, -1.9822e+00, -4.5194e-01, ..., -1.3409e-01,
-2.8075e-01, -9.1856e-01],
[ 2.3973e-01, -5.0497e-01, -2.4789e-01, ..., 3.3697e-01,
3.5783e-01, 2.9523e-01]],
[[-3.3279e-01, 1.2933e-01, 5.0562e-01, ..., -7.8904e-02,
-6.1442e-02, 5.9015e-01],
[-3.7268e-01, -2.4270e-02, 1.3830e-02, ..., -9.1104e-01,
-3.9469e-01, 8.9341e-01],
[-1.5180e-02, 1.0902e-01, 8.4933e-02, ..., -4.9616e-01,
-4.9206e-01, 6.3483e-01],
...,
[ 2.9029e-02, -8.3179e-02, 5.4272e-02, ..., -3.4921e-01,
-5.0957e-01, 5.8946e-01],
[-3.5955e-01, -4.9330e-01, -1.5822e-01, ..., -6.7668e-01,
-6.6484e-01, 3.8224e-01],
[ 1.8507e-02, -5.3714e-01, -1.6668e-01, ..., -4.8617e-01,
-4.3123e-01, 9.8873e-01]]],
[[[-7.3368e-01, -1.4615e+00, -1.4357e+00, ..., -1.2608e+00,
-8.4615e-01, -1.6412e+00],
[-3.3176e-01, -2.4284e+00, -2.2422e+00, ..., -1.7341e+00,
-1.8382e+00, -1.9094e+00],
[-1.0365e-01, -1.9062e+00, -1.6921e+00, ..., -1.3722e+00,
-1.4011e+00, -2.0733e+00],
...,
[-8.3253e-02, -7.5228e-01, -2.6146e-01, ..., -7.0982e-02,
-3.9195e-02, -4.2783e-01],
[-7.1833e-01, -7.1174e-01, 1.8184e-02, ..., 1.6167e-01,
2.5977e-01, -1.6289e-01],
[-8.6198e-01, -1.5407e+00, -1.2700e+00, ..., -1.0634e+00,
-1.1216e+00, -4.3385e-01]],
[[-2.6893e+00, -5.1013e+00, -4.5927e+00, ..., -4.7989e+00,
-5.9750e+00, -2.3799e+00],
[-5.2763e+00, -8.0297e+00, -7.4028e+00, ..., -7.7319e+00,
-9.2411e+00, -4.3536e+00],
[-4.0374e+00, -5.7435e+00, -5.6698e+00, ..., -6.1893e+00,
-6.4514e+00, -3.2125e+00],
...,
[-1.8370e+00, -2.3540e+00, -1.6314e+00, ..., -1.8256e+00,
-2.2583e+00, -9.2386e-01],
[-3.1798e+00, -3.5785e+00, -2.1305e+00, ..., -2.3021e+00,
-3.2603e+00, -9.3321e-01],
[ 2.3691e-01, -5.3678e-01, 2.5856e-01, ..., 3.2766e-01,
4.2680e-01, 2.1071e-01]],
[[-6.5349e-01, 4.5671e-01, 8.5747e-01, ..., 8.2582e-01,
6.4534e-01, 1.9244e+00],
[-9.6001e-01, 2.1790e-02, 3.8904e-01, ..., 5.6276e-01,
2.1808e-02, 1.6230e-01],
[-1.7724e+00, -2.2909e-01, -4.4910e-02, ..., 7.2785e-02,
-3.6136e-02, -1.1126e-01],
...,
[-7.6210e-01, 3.8079e-03, 3.1317e-01, ..., 1.9141e-01,
2.6709e-01, 4.4172e-01],
[-5.4449e-01, -7.3765e-02, 3.0492e-01, ..., 1.8705e-01,
1.7501e-01, 5.5147e-01],
[-2.0695e-01, -3.6800e-01, 9.6659e-02, ..., 1.7430e-02,
1.1505e-01, 3.5434e-01]],
...,
[[ 1.2296e+00, -3.6242e-01, -6.5958e-02, ..., 2.5040e-01,
-1.2611e-01, -2.1423e+00],
[-7.4904e-01, -9.1261e-01, -4.8777e-01, ..., -5.0672e-01,
-1.5359e+00, -2.1004e+00],
[-7.3352e-01, -9.1048e-01, -2.9734e-01, ..., -2.8926e-01,
-5.8976e-01, -1.3708e+00],
...,
[-8.2148e-01, -4.9776e-01, -3.6123e-01, ..., -3.3070e-01,
-5.2298e-01, -5.5484e-01],
[-2.6119e+00, -2.0070e+00, -1.4160e+00, ..., -1.3087e+00,
-1.3213e+00, -1.0488e+00],
[-6.6882e-01, -1.0491e+00, -4.7650e-01, ..., -2.1384e-01,
-3.3422e-01, -4.2017e-01]],
[[ 3.3261e-02, -5.4105e-03, 4.3968e-01, ..., 4.5392e-01,
9.2653e-01, -2.1523e-02],
[ 2.0793e-01, -8.0211e-01, -3.8840e-02, ..., 4.5374e-01,
4.2279e-01, -8.2418e-01],
[ 1.3298e-01, -6.7590e-01, 2.7761e-01, ..., 4.3814e-01,
5.0949e-01, -3.6135e-01],
...,
[-4.1807e-01, -5.8442e-01, 2.1146e-01, ..., 3.7610e-01,
3.4216e-01, 4.8264e-02],
[-7.4601e-01, -1.0822e+00, 2.0026e-02, ..., 5.0391e-02,
-7.0469e-02, -1.8190e-01],
[ 3.0714e-01, -6.3364e-02, -1.9393e-01, ..., -9.9671e-02,
-9.9661e-02, 1.7146e-01]],
[[-1.5328e+00, -8.0002e-01, -4.2628e-01, ..., -4.5670e-01,
-2.9257e-01, 8.8018e-01],
[-2.1121e+00, -7.7341e-01, -5.5018e-01, ..., -8.1646e-01,
-1.9111e-01, 1.6863e+00],
[-1.4378e+00, -5.4212e-01, -2.1706e-01, ..., -3.6966e-01,
-1.3695e-01, 1.4174e+00],
...,
[-7.2188e-01, -2.4456e-01, -1.7675e-01, ..., -1.2412e-01,
1.7874e-02, 6.0721e-01],
[-1.3638e+00, -7.5189e-01, -3.8876e-01, ..., -3.0732e-01,
-2.0107e-01, 1.4276e-01],
[-4.8195e-01, -1.3835e-01, -1.7732e-01, ..., -2.2254e-01,
-2.2657e-01, 3.3199e-01]]],
[[[-7.1799e-01, -1.4735e+00, -1.4311e+00, ..., -1.2464e+00,
-8.5621e-01, -1.6687e+00],
[-3.0544e-01, -2.4394e+00, -2.2443e+00, ..., -1.7598e+00,
-1.8740e+00, -1.9289e+00],
[-1.1745e-01, -1.8795e+00, -1.6962e+00, ..., -1.5718e+00,
-1.4107e+00, -2.1170e+00],
...,
[ 9.4505e-02, -5.1351e-01, -1.9416e-01, ..., -2.1163e-01,
-2.3690e-02, -4.1719e-01],
[-3.6902e-01, -4.6860e-01, 7.3818e-02, ..., 1.4968e-01,
2.9268e-01, -1.4208e-01],
[-6.2700e-01, -1.3405e+00, -1.0702e+00, ..., -1.0503e+00,
-1.1196e+00, -4.6415e-01]],
[[-2.6997e+00, -5.0862e+00, -4.5940e+00, ..., -4.7213e+00,
-5.9755e+00, -2.4048e+00],
[-5.3597e+00, -8.1142e+00, -7.4747e+00, ..., -7.7451e+00,
-9.2264e+00, -4.3745e+00],
[-3.9882e+00, -5.8069e+00, -5.7191e+00, ..., -7.1026e+00,
-6.4767e+00, -3.2171e+00],
...,
[-1.6846e+00, -2.1209e+00, -1.7676e+00, ..., -1.8974e+00,
-2.2440e+00, -9.0349e-01],
[-2.8122e+00, -2.9819e+00, -1.9222e+00, ..., -2.1946e+00,
-3.2227e+00, -9.4245e-01],
[ 3.5719e-01, -2.8852e-01, 2.7990e-01, ..., 2.9108e-01,
4.3422e-01, 2.0838e-01]],
[[-6.5813e-01, 4.5382e-01, 8.6557e-01, ..., 8.1956e-01,
6.0942e-01, 1.9081e+00],
[-9.3635e-01, 3.5334e-02, 4.1856e-01, ..., 5.7191e-01,
-2.1020e-02, 1.2645e-01],
[-1.7032e+00, -2.2389e-01, -3.8279e-02, ..., 1.4967e-01,
3.2643e-02, -8.4549e-02],
...,
[-6.1839e-01, 8.4201e-02, 2.4429e-01, ..., 1.7842e-01,
2.5772e-01, 4.4850e-01],
[-4.4978e-01, 6.1887e-02, 1.8293e-01, ..., 1.8359e-01,
2.1798e-01, 5.6482e-01],
[-1.5131e-01, -2.6155e-01, 1.4674e-01, ..., 1.1095e-02,
9.9814e-02, 3.3360e-01]],
...,
[[ 1.2361e+00, -3.5637e-01, -4.4367e-02, ..., 2.2562e-01,
-1.0592e-01, -2.1192e+00],
[-7.1199e-01, -9.0170e-01, -5.1474e-01, ..., -5.5578e-01,
-1.4925e+00, -2.0533e+00],
[-6.6634e-01, -8.4227e-01, -2.8538e-01, ..., -2.3027e-01,
-5.8971e-01, -1.3809e+00],
...,
[-6.4878e-01, -4.9428e-01, -3.2770e-01, ..., -3.0232e-01,
-4.5559e-01, -5.3916e-01],
[-2.2905e+00, -1.8138e+00, -1.2920e+00, ..., -1.2948e+00,
-1.3350e+00, -1.0768e+00],
[-5.9958e-01, -8.4472e-01, -4.6939e-01, ..., -2.4716e-01,
-3.4760e-01, -3.7766e-01]],
[[ 4.7000e-02, 1.2684e-02, 4.5024e-01, ..., 4.5790e-01,
9.2850e-01, -1.1236e-02],
[ 2.6537e-01, -7.6782e-01, -4.6816e-02, ..., 4.0499e-01,
3.8120e-01, -8.7494e-01],
[ 1.7313e-01, -5.9771e-01, 3.0630e-01, ..., -1.1311e-01,
5.2870e-01, -3.9115e-01],
...,
[-2.1125e-01, -4.3064e-01, 2.3670e-01, ..., 2.1739e-01,
3.2997e-01, 7.2465e-02],
[-5.2521e-01, -9.2487e-01, 6.3269e-02, ..., 5.9879e-03,
-1.4828e-02, -1.2791e-01],
[ 3.2257e-01, -2.9053e-02, -1.7677e-01, ..., -1.2770e-01,
-8.5076e-02, 1.8848e-01]],
[[-1.5436e+00, -7.8675e-01, -4.2212e-01, ..., -4.5766e-01,
-2.7059e-01, 8.6861e-01],
[-2.1167e+00, -7.7780e-01, -5.6184e-01, ..., -8.0138e-01,
-1.7487e-01, 1.6514e+00],
[-1.4707e+00, -5.5884e-01, -2.3234e-01, ..., -6.9225e-02,
-1.3717e-01, 1.3842e+00],
...,
[-6.5794e-01, -2.0530e-01, -9.9251e-02, ..., -6.8609e-02,
-7.6661e-03, 6.0764e-01],
[-1.1463e+00, -5.7596e-01, -3.5823e-01, ..., -3.1663e-01,
-1.9731e-01, 1.7183e-01],
[-3.7783e-01, -2.1423e-02, -1.7502e-01, ..., -2.3746e-01,
-2.0869e-01, 3.3787e-01]]],
[[[-3.4967e-01, -1.6894e+00, -2.5240e+00, ..., 2.1848e-01,
5.6022e-01, 4.7937e-01],
[-2.3798e+00, -4.0798e+00, -3.2609e+00, ..., 6.1262e-01,
-2.5792e-01, 8.6467e-02],
[-4.5552e+00, -1.0768e+00, -7.0538e-01, ..., -3.5295e-01,
-1.6847e-01, -1.4280e-01],
...,
[-4.9952e+00, -9.2763e+00, -1.0776e+01, ..., -4.0541e-01,
-1.5802e-01, -1.8703e-01],
[-5.9867e-01, -1.0396e+01, -1.0111e+01, ..., -1.4723e-02,
2.3271e-01, -6.4249e-02],
[ 1.3104e+00, -4.2855e+00, -3.8252e+00, ..., -1.0432e+00,
-1.1085e+00, -3.8490e-01]],
[[-2.1381e+00, -5.7793e+00, -4.4512e+00, ..., -2.5327e+00,
-1.1852e+00, -8.9360e-02],
[-9.5710e+00, -1.2574e+01, -8.1915e+00, ..., -3.4099e+00,
-2.8474e+00, -7.5873e-01],
[-8.6197e+00, -1.4751e+01, -9.5597e+00, ..., -5.1450e+00,
-1.6756e+00, -4.7534e-01],
...,
[-1.5674e+01, -1.9515e+01, -1.1849e+01, ..., -1.5126e+00,
-1.7598e+00, -4.3763e-01],
[-1.4156e+01, -1.6770e+01, -9.4664e+00, ..., -2.7215e+00,
-3.5774e+00, -1.0416e+00],
[-6.6314e-01, -2.9930e+00, -3.0984e-01, ..., 3.9065e-01,
3.7183e-01, 8.2713e-02]],
[[-6.2021e-01, -4.7157e-01, -4.5332e-01, ..., 1.0676e+00,
3.9996e-01, 1.1540e+00],
[-1.6705e+00, -1.3715e+00, -2.2233e+00, ..., 4.2502e-01,
1.0258e-01, 4.7936e-01],
[-9.0005e+00, -6.7132e+00, -3.5396e+00, ..., 7.3681e-01,
1.3273e-01, 4.5189e-01],
...,
[-2.3150e+01, -2.4112e+01, -1.6402e+01, ..., 7.3011e-02,
7.2171e-02, 4.2679e-01],
[-1.7009e+01, -1.2235e+01, -8.9605e+00, ..., 1.4369e-01,
1.6984e-01, 6.2285e-01],
[-4.9600e+00, -2.1395e-02, 1.7516e+00, ..., -1.1224e-01,
1.1066e-01, 3.8633e-01]],
...,
[[ 6.5378e-01, 8.0644e-01, 1.0639e+00, ..., -2.0613e-01,
-3.4937e-01, -9.3274e-01],
[-2.9487e-01, 7.8893e-01, -2.1292e-01, ..., -2.5645e+00,
-1.3285e+00, -8.6386e-01],
[ 5.6398e-01, -2.9436e+00, -3.9086e+00, ..., -2.8975e+00,
-2.2548e-01, -4.2562e-01],
...,
[ 1.0732e-01, -4.5903e+00, -6.8469e+00, ..., -1.4608e-01,
-2.1659e-01, -3.9627e-01],
[ 2.0790e+00, -2.6697e+00, -6.1713e+00, ..., -8.4129e-01,
-8.9641e-01, -8.2346e-01],
[-8.0307e-02, -2.7026e+00, -4.7838e+00, ..., -1.8170e-02,
-1.5192e-01, -3.5485e-01]],
[[-1.4503e-02, -5.4412e-01, -7.7131e-01, ..., -2.3080e-01,
6.8739e-01, 3.1619e-01],
[-9.9138e-01, -3.2180e+00, -1.9462e+00, ..., 8.8333e-01,
1.2084e-01, -3.9853e-02],
[-5.4115e+00, -3.2070e+00, 1.1507e+00, ..., -9.9142e-01,
-1.1095e-01, 1.8548e-01],
...,
[-9.8349e+00, -1.1056e+01, -9.3556e+00, ..., -1.6211e-01,
-1.7781e-01, 1.4195e-01],
[-6.6893e+00, -7.9051e+00, -4.7713e+00, ..., -4.3232e-01,
-4.0998e-01, -8.8192e-02],
[ 1.5531e-01, -1.7848e+00, 4.3808e-01, ..., -2.2737e-01,
-2.2931e-01, 3.1301e-01]],
[[-6.6216e-01, -4.3244e-01, -9.9124e-01, ..., 1.1468e+00,
3.5706e-01, 5.9650e-01],
[-1.5926e+00, -1.5383e+00, -2.2833e+00, ..., 4.2475e-01,
-1.4136e-01, 4.8223e-01],
[-4.9231e+00, -7.6007e+00, -3.0366e+00, ..., 8.4316e-01,
-1.7212e-01, 1.3492e-01],
...,
[-5.1709e+00, -8.8707e+00, -7.5324e+00, ..., 1.0520e-01,
-7.4873e-02, 1.5697e-01],
[-2.0403e+00, -6.6807e+00, -7.6351e+00, ..., -5.5368e-02,
-2.7029e-01, -1.5320e-01],
[ 9.6275e-02, -7.7825e-01, -1.9435e+00, ..., -1.3741e-01,
-2.8698e-01, 1.7221e-01]]],
[[[ 3.9174e-02, -2.3860e-01, -3.3173e-02, ..., -9.2371e-01,
-5.5858e-01, -1.4659e+00],
[ 3.2495e-01, -9.6544e-01, -7.5747e-01, ..., -1.0841e+00,
-1.3714e+00, -1.5458e+00],
[ 2.2608e-01, -5.6924e-01, -5.1221e-01, ..., -9.2612e-01,
-1.1148e+00, -1.8615e+00],
...,
[-7.6037e-03, -6.2987e-01, -4.2536e-01, ..., -9.3357e-01,
-1.0386e+00, -1.9282e+00],
[-5.9609e-01, -6.7252e-01, -4.6230e-02, ..., -4.5830e-01,
-5.6089e-01, -1.7182e+00],
[-7.0351e-01, -1.3458e+00, -1.1788e+00, ..., -1.1447e+00,
-1.5592e+00, -7.8161e-01]],
[[-7.2401e-01, -2.1051e+00, -1.8625e+00, ..., -4.1198e+00,
-5.5851e+00, -1.9282e+00],
[-2.0313e+00, -3.7551e+00, -3.0387e+00, ..., -6.0696e+00,
-8.2031e+00, -3.5414e+00],
[-1.6230e+00, -2.5990e+00, -2.5604e+00, ..., -5.0816e+00,
-6.2327e+00, -2.6303e+00],
...,
[-1.7841e+00, -2.4258e+00, -2.4215e+00, ..., -5.0259e+00,
-6.3231e+00, -2.4932e+00],
[-3.7963e+00, -4.4151e+00, -3.6974e+00, ..., -6.4635e+00,
-7.9712e+00, -2.5669e+00],
[ 3.0344e-01, -3.1263e-01, 3.9597e-01, ..., 4.1792e-01,
4.5465e-01, 2.1189e-01]],
[[-2.2896e-01, 1.4531e-01, 6.9781e-01, ..., 5.9172e-01,
3.2751e-01, 1.5392e+00],
[ 5.2076e-02, -3.3257e-01, 5.8026e-01, ..., 5.3278e-01,
-2.7105e-02, 1.5328e-01],
[-1.5086e-01, -2.1923e-01, 2.7168e-01, ..., -5.6849e-02,
-7.1557e-03, -1.1431e-01],
...,
[-2.3691e-01, -2.0257e-01, 1.2970e-01, ..., -4.1352e-02,
-4.9372e-02, -1.2499e-01],
[-2.1224e-01, -1.3512e-01, 1.5783e-01, ..., -1.5633e-01,
-2.6718e-02, -1.7394e-01],
[-1.4198e-01, -4.7575e-01, 1.0257e-01, ..., 5.2900e-02,
1.9378e-01, -5.6526e-01]],
...,
[[ 5.8865e-01, -2.5535e-01, 2.3408e-02, ..., 5.0257e-01,
3.2269e-01, -1.3698e+00],
[-5.3544e-01, -5.5438e-01, -5.2711e-01, ..., -1.2850e-01,
-7.7896e-01, -1.4809e+00],
[-3.7675e-01, -2.0480e-01, -2.2897e-01, ..., -1.4596e-01,
-1.2667e-01, -1.0048e+00],
...,
[-4.0906e-01, -1.3769e-01, -2.1263e-01, ..., -1.1647e-01,
-1.3041e-01, -9.5838e-01],
[-2.3013e+00, -1.6881e+00, -1.2806e+00, ..., -1.5467e+00,
-1.5842e+00, -1.8243e+00],
[-4.8715e-01, -8.0464e-01, -4.1331e-01, ..., -3.8788e-01,
-5.4859e-01, -1.1910e+00]],
[[ 1.6221e-01, 2.6899e-01, 6.1656e-01, ..., -9.2977e-02,
4.1540e-01, -4.1762e-01],
[ 1.3173e-01, 2.6231e-01, 4.1242e-01, ..., 2.2220e-01,
1.3649e-01, -9.0026e-01],
[-1.4871e-01, 9.6069e-02, 2.8079e-01, ..., 2.5500e-01,
1.5990e-01, -4.8706e-01],
...,
[-1.7594e-01, -3.1416e-02, 3.2114e-01, ..., 2.6193e-01,
1.6510e-01, -5.1621e-01],
[-5.1649e-01, -7.1020e-01, -5.5260e-02, ..., -1.4328e-01,
-2.8230e-01, -9.3779e-01],
[ 5.0749e-01, 3.6455e-01, 3.9266e-02, ..., 3.2443e-01,
3.1779e-01, 2.7688e-01]],
[[-8.5882e-01, 2.5404e-01, 3.3879e-01, ..., -9.1842e-02,
-1.9530e-02, 5.9702e-01],
[-1.1335e+00, 2.2432e-01, -4.0212e-01, ..., -9.6861e-01,
-3.4236e-01, 9.2583e-01],
[-9.5809e-01, 3.9531e-01, -1.2795e-01, ..., -6.5348e-01,
-4.5564e-01, 6.6706e-01],
...,
[-9.4930e-01, 2.0161e-01, -1.2937e-01, ..., -5.1251e-01,
-4.9272e-01, 6.2594e-01],
[-1.6485e+00, -3.5085e-01, -3.4724e-01, ..., -6.8085e-01,
-6.7588e-01, 3.9586e-01],
[-7.0451e-01, -8.3836e-02, -3.1476e-01, ..., -5.0487e-01,
-4.3258e-01, 9.9587e-01]]]], device='cuda:0'),
tensor([[[[ 3.5529e-02, -2.3059e-01, -2.8127e-02, ..., -9.9344e-01,
-5.4267e-01, -1.4567e+00],
[ 3.0695e-01, -9.2325e-01, -6.9172e-01, ..., -1.1610e+00,
-1.3037e+00, -1.4899e+00],
[ 2.1905e-01, -5.8122e-01, -5.0743e-01, ..., -8.9228e-01,
-1.0798e+00, -1.8337e+00],
...,
[-1.6096e-02, -6.1340e-01, -3.8890e-01, ..., -1.2410e+00,
-1.1702e+00, -1.9202e+00],
[-6.4263e-01, -6.9509e-01, -2.0970e-02, ..., -5.4226e-01,
-5.6917e-01, -1.6852e+00],
[-7.6481e-01, -1.3796e+00, -1.1999e+00, ..., -1.1564e+00,
-1.5526e+00, -7.6148e-01]],
[[-7.4705e-01, -2.1432e+00, -1.8270e+00, ..., -4.0004e+00,
-5.4921e+00, -1.8664e+00],
[-2.0747e+00, -3.7977e+00, -3.0024e+00, ..., -5.9712e+00,
-8.0771e+00, -3.4343e+00],
[-1.6413e+00, -2.6510e+00, -2.5990e+00, ..., -5.0228e+00,
-6.1892e+00, -2.5653e+00],
...,
[-1.7696e+00, -2.5083e+00, -2.5040e+00, ..., -5.6450e+00,
-6.6189e+00, -2.4833e+00],
[-3.7797e+00, -4.4232e+00, -3.6174e+00, ..., -6.4930e+00,
-8.0052e+00, -2.5500e+00],
[ 2.8060e-01, -3.4751e-01, 3.7445e-01, ..., 3.9911e-01,
4.3859e-01, 2.1783e-01]],
[[-2.1249e-01, 1.7520e-01, 7.3371e-01, ..., 6.3752e-01,
2.4893e-01, 1.5011e+00],
[ 6.1639e-02, -3.6243e-01, 5.9187e-01, ..., 6.3225e-01,
-4.7508e-02, 9.6606e-02],
[-1.6125e-01, -2.2202e-01, 2.4073e-01, ..., 7.8942e-02,
-3.5789e-03, -1.2438e-01],
...,
[-2.3828e-01, -2.1522e-01, 1.0244e-01, ..., -6.7841e-01,
-7.9203e-04, -1.5980e-01],
[-2.3016e-01, -1.2825e-01, 1.5215e-01, ..., -2.2554e-01,
2.5469e-03, -1.7608e-01],
[-1.7848e-01, -4.3522e-01, 1.1195e-01, ..., 3.0470e-02,
1.7602e-01, -5.9415e-01]],
...,
[[ 6.1150e-01, -2.6621e-01, -1.5998e-02, ..., 4.1898e-01,
3.5061e-01, -1.3005e+00],
[-5.0689e-01, -5.6441e-01, -5.3527e-01, ..., -1.6604e-01,
-7.6992e-01, -1.3845e+00],
[-3.5184e-01, -2.3425e-01, -2.5324e-01, ..., -1.0689e-01,
-7.4798e-02, -9.5132e-01],
...,
[-4.4829e-01, -1.0062e-01, -1.0067e-01, ..., -1.4103e+00,
-4.1946e-01, -9.5228e-01],
[-2.3775e+00, -1.7369e+00, -1.2386e+00, ..., -1.5427e+00,
-1.5485e+00, -1.8246e+00],
[-5.0003e-01, -8.2373e-01, -4.5407e-01, ..., -3.4994e-01,
-5.1390e-01, -1.1967e+00]],
[[ 1.6473e-01, 2.5689e-01, 6.3388e-01, ..., -1.3916e-01,
3.6686e-01, -3.9005e-01],
[ 1.3805e-01, 2.2142e-01, 4.4158e-01, ..., 1.6831e-01,
9.9208e-02, -8.3072e-01],
[-1.1780e-01, 7.8848e-02, 2.7642e-01, ..., 2.7824e-01,
1.8201e-01, -4.4967e-01],
...,
[-1.8599e-01, -7.1996e-03, 3.3390e-01, ..., 2.9879e-01,
5.7028e-02, -5.1567e-01],
[-5.3235e-01, -7.1554e-01, -2.6998e-02, ..., 2.8374e-02,
-2.2184e-01, -8.9348e-01],
[ 5.1428e-01, 3.8253e-01, 5.8090e-02, ..., 3.1728e-01,
3.1771e-01, 3.0505e-01]],
[[-8.2615e-01, 2.7400e-01, 3.2715e-01, ..., -6.3516e-02,
-7.9556e-02, 5.6572e-01],
[-1.1298e+00, 2.5511e-01, -3.5763e-01, ..., -9.1179e-01,
-4.2271e-01, 8.8975e-01],
[-9.7965e-01, 3.9064e-01, -1.1660e-01, ..., -5.6803e-01,
-5.2841e-01, 6.1322e-01],
...,
[-9.3973e-01, 1.7462e-01, -1.5349e-01, ..., 2.6726e-01,
-1.5775e-01, 6.1226e-01],
[-1.6625e+00, -4.3344e-01, -4.1782e-01, ..., -4.4295e-01,
-5.7454e-01, 4.2255e-01],
[-7.0956e-01, -9.6915e-02, -3.5656e-01, ..., -4.6822e-01,
-4.5370e-01, 1.0038e+00]]],
[[[-3.2250e-01, -5.1869e-01, 2.4428e-01, ..., -7.9621e-01,
-6.5715e-01, 1.3767e-01],
[-2.9328e-01, -1.8420e+00, -9.3023e-01, ..., -1.1993e+00,
-9.3917e-01, -4.2996e-01],
[-1.3197e-01, -1.6632e+00, -9.3244e-01, ..., 2.5713e-01,
-3.3391e-01, -8.0197e-01],
...,
[-8.2739e-01, -2.2679e-01, -1.7697e+00, ..., -1.2915e+00,
-1.0589e+00, -2.0554e+00],
[-1.0128e-01, -1.5324e+00, -1.0489e+00, ..., -5.9145e-01,
-4.5484e-01, -1.7476e+00],
[-1.3100e+00, -2.1268e+00, -1.5642e+00, ..., -1.5130e+00,
-1.6516e+00, -1.0046e+00]],
[[-7.0989e-01, -2.0379e+00, -1.1727e+00, ..., -6.8168e+00,
-5.0968e+00, -2.4850e+00],
[-2.1582e+00, -4.8008e+00, -3.4896e+00, ..., -1.0910e+01,
-1.0010e+01, -6.0315e+00],
[-2.2286e+00, -4.1947e+00, -4.0813e+00, ..., -1.0825e+01,
-9.2079e+00, -7.0342e+00],
...,
[-3.1263e+00, -5.9641e+00, -6.3828e+00, ..., -5.7966e+00,
-6.2978e+00, -2.9374e+00],
[-6.1280e+00, -8.1464e+00, -7.0577e+00, ..., -6.1104e+00,
-7.4833e+00, -2.7734e+00],
[-5.4735e-02, -9.2506e-01, 2.0076e-01, ..., 3.7150e-01,
6.9155e-01, 1.5451e-01]],
[[-4.3651e-01, -3.5718e-02, 2.3739e-01, ..., 1.5285e-01,
7.5547e-01, 1.4805e+00],
[-4.6703e-01, -6.5466e-01, 4.7493e-01, ..., -2.3316e+00,
7.9800e-01, 2.2959e+00],
[-8.6025e-01, -1.5776e+00, -7.4934e-01, ..., -7.6454e+00,
-3.5503e+00, 3.0239e+00],
...,
[-2.3347e+00, -1.8014e+00, -2.4877e+00, ..., 1.2741e-01,
-1.7188e-02, 2.5634e-01],
[-1.7732e+00, -3.1288e-01, -3.9336e-01, ..., -4.2935e-02,
-4.9064e-02, 2.4560e-01],
[-1.1540e+00, 2.1626e-02, 1.0447e-01, ..., -2.8277e-02,
-1.0197e-01, -4.6311e-01]],
...,
[[ 2.8727e-01, -5.1168e-01, -3.0007e-01, ..., 3.0360e-01,
-3.5120e-01, -8.8046e-01],
[-3.2922e-01, -5.1287e-01, -4.9517e-01, ..., -1.7075e+00,
-3.8232e+00, -2.3914e+00],
[-1.5041e-02, 1.6332e-01, -1.5611e-01, ..., -8.1501e+00,
-5.8955e+00, -2.3892e+00],
...,
[ 1.9215e+00, -1.4248e+00, -6.0514e-01, ..., -2.7454e-01,
-6.2670e-01, -1.3292e+00],
[-4.8951e+00, -5.1322e+00, -3.2715e+00, ..., -1.8716e+00,
-2.0703e+00, -2.1348e+00],
[-5.7797e-01, -1.8646e+00, -1.4111e+00, ..., -9.3482e-01,
-1.0102e+00, -1.3722e+00]],
[[-3.7688e-02, -6.4365e-02, 1.0029e-01, ..., 2.5912e-01,
-1.2978e+00, -1.2572e+00],
[-1.6459e-01, -8.7091e-02, -4.4612e-01, ..., -4.9577e-01,
8.7242e-02, -1.1621e+00],
[-4.0272e-01, -2.5917e-01, -2.9514e-01, ..., -2.1748e+00,
4.4814e-01, -3.4223e-01],
...,
[-4.2411e-01, 1.0579e+00, 9.1008e-01, ..., 6.8024e-02,
8.3353e-02, -7.6806e-01],
[-8.9462e-01, -3.2184e+00, -1.0314e+00, ..., 8.8093e-02,
-2.5661e-02, -9.4838e-01],
[ 4.5728e-01, -8.2799e-02, 1.9611e-01, ..., 2.5871e-01,
2.2610e-01, 2.7557e-01]],
[[-4.2012e-01, 3.1886e-01, 2.4978e-01, ..., -2.1377e-01,
-3.7705e-01, 5.2743e-01],
[-5.8050e-01, 1.9256e-03, -9.6540e-01, ..., -6.1627e+00,
-1.9769e+00, 5.0695e-01],
[-3.3487e-01, -6.1966e-03, -7.6243e-01, ..., -1.0336e+01,
-2.0011e+00, 7.5344e-01],
...,
[-2.4545e+00, -1.2758e+00, -8.5943e-01, ..., -2.1483e-01,
-8.4339e-02, 8.5772e-01],
[-7.7999e-01, -7.4331e-01, 3.5681e-01, ..., -8.4617e-01,
-4.3493e-01, 4.6799e-01],
[-9.6765e-01, -9.7183e-01, -1.3263e-01, ..., -3.3029e-01,
-1.7284e-01, 1.1576e+00]]],
[[[-7.5707e-01, -1.4496e+00, -1.4106e+00, ..., -8.4733e-02,
5.3890e-01, 4.5859e-01],
[-3.3720e-01, -3.0604e+00, -2.1831e+00, ..., -5.7699e-01,
-2.9104e-01, 7.7974e-02],
[-2.3672e+00, -6.1482e+00, -2.1143e+00, ..., -1.4848e+00,
-7.8566e-01, -7.3382e-02],
...,
[-5.0289e-01, -1.9210e+00, -1.8906e+00, ..., -5.3613e-01,
-4.5813e-01, -1.1373e-01],
[-1.0243e+00, -1.7485e+00, -1.2054e+00, ..., 6.6946e-02,
3.1088e-01, -1.5135e-03],
[-8.3458e-01, -1.6904e+00, -1.2753e+00, ..., -1.1937e+00,
-1.2817e+00, -4.7917e-01]],
[[-2.4004e+00, -4.8685e+00, -4.2044e+00, ..., -6.4865e-01,
-1.1624e+00, -1.1463e-01],
[-6.6597e+00, -9.5480e+00, -1.0562e+01, ..., -2.1389e+00,
-2.6974e+00, -6.3857e-01],
[-9.2466e+00, -9.4477e+00, -1.3856e+01, ..., -7.7160e+00,
-6.4974e+00, -1.8790e+00],
...,
[-4.4298e+00, -6.1927e+00, -5.5907e+00, ..., -2.0757e+00,
-2.1244e+00, -5.0478e-01],
[-6.4961e+00, -7.5534e+00, -6.4321e+00, ..., -2.8432e+00,
-3.7795e+00, -1.1215e+00],
[-5.3323e-02, -8.4543e-01, 3.7166e-01, ..., 4.2814e-01,
3.9205e-01, 5.2749e-02]],
[[-5.7192e-01, 6.0560e-01, 1.1107e+00, ..., 8.2028e-01,
3.5046e-01, 1.0703e+00],
[-1.1107e+00, -4.6145e-01, 4.0179e-01, ..., 4.1945e-01,
1.0159e-01, 4.6679e-01],
[-2.9937e+00, -1.2874e+00, -4.6542e-01, ..., 6.3110e-01,
7.3333e-01, 9.6040e-01],
...,
[-1.7492e+00, -8.9303e-02, -1.2068e-01, ..., -2.1177e-01,
9.0809e-01, 3.8626e-02],
[-1.5480e+00, 8.9776e-02, 1.6934e-02, ..., 9.6691e-02,
1.2237e-01, 6.1144e-01],
[-1.4587e+00, -5.0712e-01, -2.3771e-01, ..., -1.2102e-01,
6.9918e-02, 4.0929e-01]],
...,
[[ 1.1383e+00, -4.0717e-01, -2.4507e-01, ..., -6.8985e-02,
-2.9467e-01, -8.5396e-01],
[-6.7344e-01, -8.1490e-01, -4.0049e-01, ..., -5.5458e-01,
-1.1736e+00, -7.6849e-01],
[ 4.3161e-01, -8.9715e-02, -2.2850e+00, ..., 2.7773e-01,
1.5633e-01, -2.2624e-01],
...,
[-7.4124e-01, -1.2121e+00, -7.4154e-01, ..., -1.3191e+00,
-1.8408e+00, -1.0686e+00],
[-2.8355e+00, -2.9453e+00, -1.8509e+00, ..., -1.0239e+00,
-1.0816e+00, -8.9551e-01],
[-6.5765e-01, -1.7006e+00, -1.0178e+00, ..., -4.3548e-02,
-1.6224e-01, -4.0223e-01]],
[[ 2.7021e-02, -1.0317e-02, 6.1713e-01, ..., 2.9656e-01,
6.3410e-01, 2.6740e-01],
[ 2.7174e-01, -5.7509e-01, 4.5583e-01, ..., -2.0570e-01,
2.5138e-02, -1.9564e-02],
[-3.2205e-01, -3.2954e+00, 2.3239e-02, ..., -9.2886e-01,
-1.3728e+00, -7.9319e-01],
...,
[ 7.4005e-02, -8.0182e-01, 2.9401e-01, ..., -9.6918e-01,
-9.5910e-01, 1.8260e-01],
[-2.1861e-01, -1.4484e+00, 1.2274e-02, ..., -4.8788e-01,
-5.2320e-01, -2.1240e-01],
[ 6.7071e-01, 1.4746e-01, 2.7754e-01, ..., -2.8995e-01,
-3.0065e-01, 2.7887e-01]],
[[-1.6649e+00, -6.3150e-01, -1.5237e-02, ..., 5.1842e-01,
3.2693e-01, 5.3454e-01],
[-2.3329e+00, -7.5875e-01, 2.3433e-02, ..., -1.3072e-02,
-1.4075e-01, 3.9951e-01],
[-3.0151e+00, -1.9401e+00, -2.9511e+00, ..., 2.3727e-02,
6.5958e-02, -8.1686e-02],
...,
[-1.8299e+00, -5.0542e-01, 2.3830e-01, ..., 1.0932e+00,
2.4739e-01, 6.6317e-01],
[-2.2566e+00, -1.1660e+00, -2.3380e-01, ..., 1.8357e-02,
-1.8758e-01, -1.6688e-01],
[-1.0829e+00, -2.0541e-01, 4.3428e-01, ..., -1.8475e-01,
-3.5478e-01, 9.3711e-02]]],
[[[-7.1678e-01, -1.4789e+00, -1.4586e+00, ..., -1.2415e+00,
-8.6240e-01, -1.6672e+00],
[-3.1299e-01, -2.4634e+00, -2.2832e+00, ..., -1.7285e+00,
-1.8970e+00, -1.9017e+00],
[-1.2607e-01, -1.9062e+00, -1.6979e+00, ..., -1.2490e+00,
-1.4674e+00, -2.1186e+00],
...,
[ 5.5382e-02, -5.6105e-01, -2.2571e-01, ..., -9.8028e-02,
-2.2501e-02, -4.2147e-01],
[-3.8924e-01, -4.6972e-01, 3.8095e-02, ..., 1.3451e-01,
2.7553e-01, -1.4793e-01],
[-6.3907e-01, -1.3429e+00, -1.1118e+00, ..., -1.0760e+00,
-1.1438e+00, -4.9196e-01]],
[[-2.6874e+00, -5.0944e+00, -4.6150e+00, ..., -4.7193e+00,
-5.9757e+00, -2.3914e+00],
[-5.3774e+00, -8.1435e+00, -7.4682e+00, ..., -7.7520e+00,
-9.1975e+00, -4.3374e+00],
[-4.0040e+00, -5.7705e+00, -5.7015e+00, ..., -6.0742e+00,
-6.4508e+00, -3.2222e+00],
...,
[-1.7415e+00, -2.1354e+00, -1.6718e+00, ..., -1.8318e+00,
-2.2548e+00, -9.0644e-01],
[-2.8326e+00, -2.9782e+00, -1.9546e+00, ..., -2.1990e+00,
-3.2703e+00, -9.5712e-01],
[ 3.3094e-01, -2.9651e-01, 3.0974e-01, ..., 2.9551e-01,
4.4647e-01, 2.1250e-01]],
[[-6.6827e-01, 4.4587e-01, 8.4270e-01, ..., 8.2310e-01,
6.2718e-01, 1.9001e+00],
[-9.5313e-01, 3.4918e-02, 4.1890e-01, ..., 5.5712e-01,
1.5955e-02, 9.9944e-02],
[-1.7491e+00, -2.2630e-01, -2.1842e-02, ..., 5.9557e-03,
2.9752e-02, -1.1758e-01],
...,
[-6.3943e-01, 5.6343e-02, 2.2976e-01, ..., 1.8679e-01,
2.6261e-01, 4.3573e-01],
[-4.5026e-01, 7.3011e-02, 1.9917e-01, ..., 1.7965e-01,
2.2024e-01, 5.7056e-01],
[-1.5856e-01, -2.6365e-01, 1.3584e-01, ..., -1.9437e-02,
8.5289e-02, 3.4587e-01]],
...,
[[ 1.2231e+00, -3.7320e-01, -4.7145e-02, ..., 2.2239e-01,
-1.1109e-01, -2.1152e+00],
[-7.3157e-01, -8.7444e-01, -5.0156e-01, ..., -5.1734e-01,
-1.4941e+00, -2.0740e+00],
[-6.9406e-01, -8.1426e-01, -2.6861e-01, ..., -2.8998e-01,
-5.9729e-01, -1.4131e+00],
...,
[-6.9173e-01, -5.0785e-01, -3.3460e-01, ..., -3.3063e-01,
-4.9508e-01, -5.6288e-01],
[-2.2778e+00, -1.7923e+00, -1.3254e+00, ..., -1.3103e+00,
-1.3598e+00, -1.0827e+00],
[-6.2437e-01, -8.6341e-01, -4.4372e-01, ..., -2.3498e-01,
-3.1391e-01, -3.5493e-01]],
[[ 5.6960e-02, 3.0636e-02, 4.4716e-01, ..., 4.7036e-01,
9.3200e-01, -3.1089e-02],
[ 2.4137e-01, -8.0604e-01, -6.5378e-02, ..., 4.3207e-01,
3.7344e-01, -8.4137e-01],
[ 1.4276e-01, -6.3343e-01, 3.0560e-01, ..., 6.4686e-01,
4.9811e-01, -3.9661e-01],
...,
[-2.5858e-01, -4.5730e-01, 2.3901e-01, ..., 3.4683e-01,
3.2988e-01, 6.6652e-02],
[-5.5192e-01, -9.2521e-01, 3.8038e-02, ..., 1.5936e-02,
-2.5517e-02, -1.2775e-01],
[ 3.1536e-01, -3.7511e-02, -2.1157e-01, ..., -1.4415e-01,
-8.4851e-02, 1.9706e-01]],
[[-1.5586e+00, -8.0312e-01, -4.2961e-01, ..., -4.2135e-01,
-2.8066e-01, 8.6501e-01],
[-2.1118e+00, -7.7895e-01, -5.6301e-01, ..., -7.7324e-01,
-1.8906e-01, 1.6494e+00],
[-1.4830e+00, -5.6974e-01, -2.1377e-01, ..., -3.5501e-01,
-8.8637e-02, 1.3994e+00],
...,
[-6.9023e-01, -2.2244e-01, -1.3670e-01, ..., -1.5542e-01,
-1.3237e-02, 6.0954e-01],
[-1.1560e+00, -5.5146e-01, -3.1756e-01, ..., -3.2106e-01,
-2.1179e-01, 1.5661e-01],
[-3.8468e-01, -1.3226e-02, -1.6183e-01, ..., -2.4510e-01,
-2.0587e-01, 3.3916e-01]]],
[[[-5.7297e-01, -1.4539e+00, -1.4627e+00, ..., -1.3972e+00,
-8.8919e-01, -1.7312e+00],
[-1.7899e-01, -2.5499e+00, -2.1894e+00, ..., -1.7480e+00,
-1.8708e+00, -1.9607e+00],
[-2.1512e-01, -2.1071e+00, -1.5323e+00, ..., -1.4856e+00,
-1.4128e+00, -2.1251e+00],
...,
[-7.5177e-02, -1.2203e+00, -2.4630e+00, ..., -6.0142e-01,
-6.1362e-01, -9.4830e-01],
[-1.1663e-01, -4.3057e-01, -1.3481e+00, ..., -4.9550e-01,
-4.1309e-01, -1.0040e+00],
[-1.5740e-01, -3.4870e-01, -3.6435e-01, ..., -8.9691e-01,
-1.1035e+00, -6.1599e-01]],
[[-2.6496e+00, -5.0087e+00, -4.7459e+00, ..., -5.4402e+00,
-6.0565e+00, -2.4843e+00],
[-5.5993e+00, -8.2241e+00, -7.8096e+00, ..., -8.3001e+00,
-9.1479e+00, -4.3690e+00],
[-4.0873e+00, -5.7796e+00, -5.9827e+00, ..., -6.5803e+00,
-6.4934e+00, -3.2451e+00],
...,
[-2.0014e+00, -4.4009e+00, -4.5221e+00, ..., -3.3237e+00,
-3.8460e+00, -1.4921e+00],
[-4.5257e+00, -6.8595e+00, -6.8091e+00, ..., -5.3568e+00,
-6.5920e+00, -2.3779e+00],
[ 6.9953e-01, 7.0307e-01, 1.3201e+00, ..., 4.6812e-01,
5.7002e-01, 5.9112e-03]],
[[-7.4623e-01, 3.1056e-01, 9.0032e-01, ..., 7.9475e-01,
6.1012e-01, 1.8589e+00],
[-9.3134e-01, -1.9882e-01, 4.7685e-01, ..., 6.2338e-01,
6.0266e-03, 9.5991e-02],
[-1.6500e+00, -4.9059e-01, 4.7005e-02, ..., 9.8040e-02,
-9.4662e-03, -1.0513e-01],
...,
[-9.1059e-01, -4.5747e-02, -2.3655e+00, ..., -1.3982e+00,
4.1692e-01, 3.5425e-01],
[-9.7666e-01, 2.7219e-01, -3.5875e+00, ..., -6.8665e-01,
2.1151e-01, 3.9399e-01],
[-4.5801e-01, 5.5057e-01, -2.9325e+00, ..., -6.8567e-02,
1.0259e-01, 5.8124e-02]],
...,
[[ 1.0724e+00, -4.4802e-01, -9.7108e-02, ..., 1.7246e-01,
-3.7047e-02, -2.0833e+00],
[-8.5920e-01, -8.2142e-01, -4.0926e-01, ..., -8.8956e-01,
-1.4232e+00, -2.0445e+00],
[-7.6195e-01, -6.8656e-01, -1.5435e-01, ..., -6.3737e-01,
-5.8752e-01, -1.3714e+00],
...,
[-4.5378e-01, 1.6869e-02, -7.7433e-01, ..., -1.8536e+00,
-1.0649e+00, -5.0734e-01],
[-2.0066e+00, -9.9320e-01, -8.6418e-01, ..., -1.4526e+00,
-1.3290e+00, -1.0779e+00],
[-2.9865e-01, -2.3160e-01, 7.3558e-01, ..., -1.8544e-01,
-2.9407e-01, -6.2051e-01]],
[[-5.2889e-02, -5.3730e-02, 4.5807e-01, ..., 9.2617e-02,
9.1657e-01, -3.1433e-02],
[ 3.6924e-02, -7.8720e-01, 8.5763e-02, ..., 2.3108e-01,
3.8070e-01, -8.2123e-01],
[-1.2614e-01, -7.2255e-01, 5.1430e-01, ..., 3.6063e-01,
5.4870e-01, -3.8795e-01],
...,
[-3.7820e-01, -4.5171e-01, -1.6038e+00, ..., 3.0616e-02,
-1.1056e-01, -1.3883e-01],
[-7.6045e-01, -1.2793e+00, -1.8489e+00, ..., -3.3408e-01,
-4.0262e-01, -6.8286e-01],
[ 5.0334e-01, 1.1822e-01, -3.0369e-01, ..., 1.5988e-01,
1.4462e-01, 3.5706e-01]],
[[-1.6756e+00, -7.5941e-01, -3.7692e-01, ..., -4.6002e-01,
-3.2046e-01, 8.4469e-01],
[-2.2551e+00, -7.1505e-01, -5.8732e-01, ..., -7.5737e-01,
-2.0599e-01, 1.6481e+00],
[-1.6290e+00, -6.1488e-01, -4.0923e-01, ..., -2.7541e-01,
-1.0887e-01, 1.4080e+00],
...,
[-4.4415e-01, -2.7067e-01, -2.3254e-01, ..., 1.3191e-01,
8.2050e-01, 7.7282e-01],
[-9.0920e-01, -4.5403e-01, 7.8886e-01, ..., 5.4349e-01,
4.7921e-01, 4.1403e-01],
[-4.3257e-01, -5.6886e-01, 1.2224e+00, ..., -1.6080e-01,
-2.2734e-01, 5.8807e-01]]]], device='cuda:0'),
tensor([[[[-3.2585e-01, -5.2304e-01, -1.9838e-01, ..., -9.8368e-01,
-5.4687e-01, -1.5028e+00],
[-1.8269e-01, -2.4355e+00, -1.2568e+00, ..., -1.2506e+00,
-1.3235e+00, -1.5496e+00],
[-6.5447e-01, -3.6275e+00, -1.8542e+00, ..., -9.9420e-01,
-1.1003e+00, -1.8819e+00],
...,
[ 5.5158e-01, -1.5236e+00, -1.1046e+00, ..., -1.3293e-01,
-1.8031e+00, -2.9527e+00],
[-3.1110e-01, -1.0334e+00, -1.6545e-01, ..., -6.8732e-01,
-1.2256e+00, -2.1389e+00],
[-2.5269e-01, -1.4254e+00, -1.0815e+00, ..., -1.2059e+00,
-1.5999e+00, -8.3015e-01]],
[[-7.8408e-01, -2.2976e+00, -1.0745e+00, ..., -3.8642e+00,
-5.5217e+00, -1.8380e+00],
[-5.5477e+00, -6.3124e+00, -3.6762e+00, ..., -6.0165e+00,
-8.3326e+00, -3.4111e+00],
[-9.0946e+00, -1.0724e+01, -6.9945e+00, ..., -5.1157e+00,
-6.3768e+00, -2.4971e+00],
...,
[-2.0119e+00, -3.7760e+00, -2.4127e+00, ..., -7.0233e+00,
-9.1643e+00, -3.4598e+00],
[-3.6735e+00, -4.3838e+00, -2.9363e+00, ..., -7.0809e+00,
-8.4740e+00, -2.6647e+00],
[ 6.5487e-01, -5.8634e-01, 5.0345e-01, ..., 3.6130e-01,
4.0989e-01, 2.6484e-01]],
[[-4.7322e-01, 6.2482e-01, 1.1170e+00, ..., 6.5334e-01,
1.6973e-01, 1.4945e+00],
[-8.5883e-01, -5.2905e-02, 5.8395e-01, ..., 5.9371e-01,
-1.4158e-01, 2.7464e-02],
[-1.6604e+00, 9.0781e-01, 2.0119e+00, ..., 1.4991e-02,
-3.4528e-02, -1.3646e-01],
...,
[-1.4907e+00, -3.9565e-01, 1.8073e-01, ..., -2.2028e+00,
-1.8442e+00, 6.0583e-01],
[-1.3393e+00, 2.7210e-01, 1.5363e-01, ..., -7.2828e-01,
-4.6616e-01, -2.8270e-01],
[-8.4904e-01, 1.0026e-01, -9.6034e-02, ..., 5.5449e-02,
2.1436e-01, -5.8260e-01]],
...,
[[ 4.0863e-01, -2.1274e-01, -1.0937e-01, ..., 4.3573e-01,
3.9255e-01, -1.3050e+00],
[-1.6983e+00, -1.2058e+00, -7.6049e-01, ..., -2.1806e-01,
-7.7947e-01, -1.3837e+00],
[-2.2454e-01, -1.2189e+00, -6.5639e-01, ..., -9.2611e-02,
-4.7735e-02, -9.3078e-01],
...,
[-1.9681e+00, -1.2099e+00, -2.3090e-01, ..., -6.2880e+00,
-4.7876e+00, -1.8236e+00],
[-2.3946e+00, -1.6861e+00, -8.9376e-01, ..., -3.4425e+00,
-1.8402e+00, -1.5802e+00],
[-5.9881e-01, -1.0452e+00, -1.6497e-01, ..., -4.0312e-01,
-5.4683e-01, -1.2475e+00]],
[[ 5.2161e-03, 3.1230e-01, 5.5921e-01, ..., -3.4960e-02,
3.9249e-01, -3.1235e-01],
[-2.7178e-01, -1.7507e+00, -1.2174e+00, ..., 2.1935e-01,
4.1326e-02, -8.0433e-01],
[ 3.6532e-01, -1.4029e+00, -2.3718e+00, ..., 3.0009e-01,
1.4262e-01, -4.0096e-01],
...,
[-1.8898e-01, -1.2593e+00, -6.2514e-01, ..., 1.5548e+00,
7.2820e-01, -1.1062e+00],
[-9.1872e-01, -1.7399e+00, -3.3078e-01, ..., -6.1152e-01,
-7.9913e-01, -1.1779e+00],
[ 2.7684e-01, -3.9912e-01, -1.9570e-01, ..., 3.1032e-01,
3.0627e-01, 3.1060e-01]],
[[-5.1511e-01, 1.1087e-01, 6.0569e-01, ..., -1.5023e-01,
-1.2850e-01, 5.4953e-01],
[-6.6748e-01, -1.5948e-01, 1.9058e-01, ..., -9.4219e-01,
-4.5180e-01, 9.1414e-01],
[-1.7417e+00, -1.2409e+00, -3.5415e-01, ..., -6.0120e-01,
-5.9060e-01, 6.2639e-01],
...,
[ 7.7508e-01, 7.7539e-01, 3.9247e-01, ..., -1.7517e+00,
2.2139e+00, 1.9814e+00],
[-4.2456e-01, -3.9376e-01, -6.0556e-02, ..., 1.3726e-01,
3.7409e-01, 8.2710e-01],
[ 7.1562e-02, -3.8337e-01, -9.9230e-02, ..., -4.8531e-01,
-4.8012e-01, 1.0242e+00]]],
[[[-7.3286e-01, -1.4713e+00, -1.4319e+00, ..., -8.7339e-01,
-8.5836e-01, -1.6865e+00],
[-2.9323e-01, -2.4562e+00, -2.3438e+00, ..., -7.4320e-01,
-1.4568e+00, -1.7067e+00],
[-1.0403e-01, -1.9097e+00, -1.6840e+00, ..., 2.0150e-01,
-3.3773e-01, -1.6707e+00],
...,
[ 6.5762e-02, -5.2659e-01, 5.0357e-01, ..., -8.5143e-02,
-6.5562e-02, -4.3761e-01],
[-3.3044e-01, -4.3569e-01, 2.4896e-01, ..., 1.5319e-01,
2.6431e-01, -1.6797e-01],
[-6.2190e-01, -1.3275e+00, -1.0901e+00, ..., -1.0681e+00,
-1.1763e+00, -4.9596e-01]],
[[-2.7193e+00, -5.1347e+00, -4.5710e+00, ..., -5.0672e+00,
-6.3693e+00, -2.7917e+00],
[-5.4775e+00, -8.3742e+00, -7.6578e+00, ..., -7.5102e+00,
-9.6050e+00, -5.0860e+00],
[-4.0824e+00, -5.9326e+00, -6.0896e+00, ..., -5.8360e+00,
-6.9556e+00, -4.0285e+00],
...,
[-1.7117e+00, -2.1079e+00, -1.6883e+00, ..., -1.8595e+00,
-2.3011e+00, -9.3242e-01],
[-2.8040e+00, -2.9308e+00, -2.2768e+00, ..., -2.2527e+00,
-3.3986e+00, -1.0265e+00],
[ 3.3065e-01, -2.6304e-01, 3.1239e-01, ..., 3.2607e-01,
4.6609e-01, 1.9938e-01]],
[[-6.7948e-01, 4.6728e-01, 8.9884e-01, ..., 8.7078e-01,
7.4302e-01, 1.9846e+00],
[-1.0127e+00, -4.1765e-02, 4.3467e-01, ..., -5.2752e-01,
-6.4123e-02, 2.5408e-01],
[-1.7693e+00, -2.7674e-01, -5.3877e-02, ..., -1.6896e+00,
-7.5546e-01, 5.7405e-02],
...,
[-6.3617e-01, 8.5125e-02, 9.2608e-01, ..., 1.6584e-01,
2.5373e-01, 4.3311e-01],
[-4.1679e-01, 1.0188e-01, 6.2722e-01, ..., 1.7476e-01,
2.0531e-01, 5.5956e-01],
[-1.5927e-01, -2.3634e-01, 1.4731e-01, ..., 1.1733e-02,
9.6007e-02, 3.5963e-01]],
...,
[[ 1.2382e+00, -3.6024e-01, -6.4722e-02, ..., 7.9190e-01,
2.0063e-01, -1.9871e+00],
[-7.0630e-01, -8.5978e-01, -5.3032e-01, ..., -8.1301e-01,
-1.8075e+00, -2.2494e+00],
[-6.2882e-01, -7.1197e-01, -2.6346e-01, ..., -2.0693e+00,
-2.0091e+00, -2.0055e+00],
...,
[-6.3339e-01, -5.3040e-01, -5.2886e-01, ..., -3.5539e-01,
-5.0567e-01, -5.5615e-01],
[-2.2183e+00, -1.7897e+00, -1.9329e+00, ..., -1.3200e+00,
-1.3754e+00, -1.1078e+00],
[-5.8851e-01, -7.9861e-01, -4.2188e-01, ..., -2.3008e-01,
-3.3568e-01, -4.0305e-01]],
[[ 2.5908e-02, 5.0395e-03, 4.4999e-01, ..., 7.0179e-01,
5.2721e-01, -3.6966e-01],
[ 2.2593e-01, -7.8094e-01, -1.1932e-01, ..., 1.1807e+00,
6.2389e-01, -1.0425e+00],
[ 1.5867e-01, -5.5300e-01, 4.1613e-01, ..., 1.2590e+00,
2.0701e+00, -6.9001e-02],
...,
[-2.2368e-01, -4.6229e-01, 5.2478e-01, ..., 3.7330e-01,
3.2926e-01, 3.1408e-02],
[-5.0185e-01, -9.0083e-01, 4.0037e-01, ..., 2.5803e-02,
-2.2348e-02, -1.8234e-01],
[ 3.3127e-01, -3.0530e-02, -1.9550e-01, ..., -1.1458e-01,
-9.7840e-02, 1.6190e-01]],
[[-1.5423e+00, -7.9248e-01, -4.3672e-01, ..., -3.1473e-01,
3.3754e-04, 9.3275e-01],
[-2.1208e+00, -7.2599e-01, -5.3879e-01, ..., -1.9475e+00,
-3.6112e-02, 1.8353e+00],
[-1.5384e+00, -5.9096e-01, -1.6470e-01, ..., -2.7784e+00,
-1.1940e-01, 1.6047e+00],
...,
[-6.5002e-01, -2.5382e-01, -5.4177e-01, ..., -1.2191e-01,
3.4735e-03, 5.8074e-01],
[-1.1004e+00, -5.8947e-01, -4.8184e-01, ..., -3.2547e-01,
-1.7781e-01, 1.3826e-01],
[-3.3208e-01, -7.6301e-03, -1.6207e-01, ..., -2.4260e-01,
-2.2938e-01, 3.1959e-01]]],
[[[ 3.1843e-02, -2.2998e-01, -3.5623e-02, ..., -9.6783e-01,
-5.9618e-01, -1.4826e+00],
[ 2.9728e-01, -9.1901e-01, -6.6661e-01, ..., -1.1040e+00,
-1.3188e+00, -1.5180e+00],
[ 2.3970e-01, -5.7639e-01, -4.9390e-01, ..., -8.7538e-01,
-1.1178e+00, -1.8440e+00],
...,
[ 8.1810e-01, 1.4513e-01, -1.4303e-01, ..., -8.7614e-01,
-1.0336e+00, -1.9311e+00],
[-6.9670e-01, -7.4191e-01, -8.0308e-02, ..., -4.6870e-01,
-5.3935e-01, -1.7046e+00],
[-7.9350e-01, -1.3789e+00, -1.2086e+00, ..., -1.1275e+00,
-1.5381e+00, -7.8743e-01]],
[[-7.1403e-01, -2.1250e+00, -1.8720e+00, ..., -4.0078e+00,
-5.4604e+00, -1.8281e+00],
[-1.9680e+00, -3.6710e+00, -2.9689e+00, ..., -5.9616e+00,
-8.0718e+00, -3.4438e+00],
[-1.5394e+00, -2.5744e+00, -2.5142e+00, ..., -4.9889e+00,
-6.1216e+00, -2.6256e+00],
...,
[-1.7562e+00, -3.1693e+00, -3.4506e+00, ..., -4.9772e+00,
-6.2924e+00, -2.5426e+00],
[-3.8544e+00, -4.5116e+00, -3.6776e+00, ..., -6.4124e+00,
-7.9732e+00, -2.5848e+00],
[ 2.4619e-01, -3.6518e-01, 3.6717e-01, ..., 4.4707e-01,
4.6876e-01, 1.9804e-01]],
[[-2.1505e-01, 1.5807e-01, 7.0444e-01, ..., 6.3721e-01,
2.8728e-01, 1.5203e+00],
[ 2.5093e-02, -3.7128e-01, 5.2103e-01, ..., 5.9640e-01,
2.6772e-03, 1.3715e-01],
[-1.4633e-01, -2.1693e-01, 2.2145e-01, ..., 1.6603e-02,
-6.1236e-02, -1.4362e-01],
...,
[ 4.4726e-01, 3.8681e-03, 6.5924e-02, ..., -1.8498e-02,
-1.0729e-01, -1.8319e-01],
[-2.9102e-01, -2.1617e-01, 1.5483e-01, ..., -1.3498e-01,
-4.5312e-02, -2.0368e-01],
[-1.5001e-01, -4.9830e-01, 9.2717e-02, ..., 6.5035e-02,
1.6701e-01, -5.6532e-01]],
...,
[[ 5.9444e-01, -2.3561e-01, -1.2837e-02, ..., 4.3367e-01,
3.4670e-01, -1.3247e+00],
[-5.0579e-01, -5.3596e-01, -4.7339e-01, ..., -8.2057e-02,
-7.6124e-01, -1.4138e+00],
[-3.8671e-01, -2.0659e-01, -2.4658e-01, ..., -6.8708e-02,
-8.2600e-02, -1.0090e+00],
...,
[-1.4135e+00, -1.9097e+00, -1.7637e+00, ..., -1.1545e-01,
-1.0228e-01, -9.4723e-01],
[-2.3894e+00, -1.7431e+00, -1.2297e+00, ..., -1.5459e+00,
-1.6011e+00, -1.8298e+00],
[-4.8328e-01, -8.4941e-01, -4.1311e-01, ..., -3.7503e-01,
-5.4390e-01, -1.1745e+00]],
[[ 1.6819e-01, 2.6901e-01, 6.3412e-01, ..., -1.0971e-01,
3.1785e-01, -3.8664e-01],
[ 1.4312e-01, 2.5208e-01, 4.7274e-01, ..., 2.4108e-01,
1.2926e-01, -8.4881e-01],
[-1.2482e-01, 9.3476e-02, 2.7742e-01, ..., 3.1567e-01,
1.9349e-01, -4.9778e-01],
...,
[-2.8749e-01, -6.3728e-02, 1.6793e-01, ..., 3.3456e-01,
2.1816e-01, -5.3386e-01],
[-5.9245e-01, -7.3142e-01, -5.0370e-02, ..., -1.3914e-01,
-2.0004e-01, -9.3031e-01],
[ 5.0696e-01, 3.8768e-01, 7.5664e-02, ..., 3.6574e-01,
3.6329e-01, 2.7371e-01]],
[[-8.3694e-01, 2.2973e-01, 3.4845e-01, ..., -6.1346e-02,
-8.6079e-02, 5.5887e-01],
[-1.1157e+00, 1.7574e-01, -3.5432e-01, ..., -9.2947e-01,
-3.8556e-01, 9.1975e-01],
[-9.3170e-01, 3.3329e-01, -1.4612e-01, ..., -6.1358e-01,
-4.9198e-01, 6.3815e-01],
...,
[-2.2072e-01, 1.3806e+00, 1.3582e+00, ..., -5.8230e-01,
-4.8576e-01, 6.2869e-01],
[-1.7409e+00, -4.5483e-01, -3.8704e-01, ..., -7.3175e-01,
-6.3833e-01, 4.1281e-01],
[-7.5724e-01, -1.2336e-01, -3.1085e-01, ..., -5.0311e-01,
-3.8180e-01, 1.0059e+00]]],
[[[-6.8902e-01, -1.4285e+00, -1.4258e+00, ..., -1.2486e+00,
-8.6791e-01, -1.6540e+00],
[-2.9174e-01, -2.4486e+00, -2.2494e+00, ..., -1.7349e+00,
-1.8708e+00, -1.9154e+00],
[-1.1479e-01, -1.9067e+00, -1.6864e+00, ..., -1.2417e+00,
-1.4056e+00, -2.0875e+00],
...,
[ 6.5259e-02, -5.4412e-01, -2.3500e-01, ..., -1.0680e-01,
-2.4761e-02, -4.1935e-01],
[-4.0039e-01, -4.6913e-01, 3.0391e-02, ..., 1.3696e-01,
3.0374e-01, -1.4675e-01],
[-6.5395e-01, -1.3483e+00, -1.1142e+00, ..., -1.0894e+00,
-1.1760e+00, -4.9711e-01]],
[[-2.6600e+00, -5.0792e+00, -4.6032e+00, ..., -4.8077e+00,
-6.0164e+00, -2.4251e+00],
[-5.3054e+00, -8.0694e+00, -7.4250e+00, ..., -7.7181e+00,
-9.2252e+00, -4.3649e+00],
[-4.0009e+00, -5.7979e+00, -5.7632e+00, ..., -5.9280e+00,
-6.4222e+00, -3.2502e+00],
...,
[-1.7423e+00, -2.2262e+00, -1.7703e+00, ..., -1.7908e+00,
-2.2354e+00, -8.9311e-01],
[-2.8286e+00, -3.0063e+00, -1.9489e+00, ..., -2.1999e+00,
-3.3532e+00, -9.8938e-01],
[ 3.3263e-01, -2.8736e-01, 3.2906e-01, ..., 2.9525e-01,
4.4695e-01, 2.0738e-01]],
[[-6.6447e-01, 4.6510e-01, 8.7894e-01, ..., 8.0583e-01,
6.0586e-01, 1.8972e+00],
[-9.9286e-01, -1.2070e-02, 4.1595e-01, ..., 5.5654e-01,
-1.6609e-02, 1.4519e-01],
[-1.7682e+00, -2.7106e-01, -6.8253e-02, ..., 4.2557e-02,
-1.9660e-02, -9.6559e-02],
...,
[-6.4152e-01, 4.3546e-02, 2.3293e-01, ..., 1.6314e-01,
2.3523e-01, 3.9289e-01],
[-4.4792e-01, 4.7468e-02, 2.0678e-01, ..., 1.7562e-01,
2.1368e-01, 5.5604e-01],
[-1.5106e-01, -2.5953e-01, 1.2617e-01, ..., -6.6703e-03,
9.3812e-02, 3.3898e-01]],
...,
[[ 1.2135e+00, -3.6019e-01, -6.1199e-02, ..., 2.4226e-01,
-1.1025e-01, -2.1275e+00],
[-7.4062e-01, -8.8122e-01, -5.1799e-01, ..., -4.9973e-01,
-1.4923e+00, -2.0868e+00],
[-7.1105e-01, -8.8035e-01, -3.0793e-01, ..., -3.1980e-01,
-6.0234e-01, -1.3757e+00],
...,
[-6.4848e-01, -5.1951e-01, -3.9020e-01, ..., -3.1824e-01,
-4.8504e-01, -5.2140e-01],
[-2.3262e+00, -1.8092e+00, -1.2978e+00, ..., -1.3245e+00,
-1.3656e+00, -1.0832e+00],
[-6.3036e-01, -8.6501e-01, -4.4958e-01, ..., -2.4936e-01,
-3.2725e-01, -3.8225e-01]],
[[ 3.8249e-02, 2.2868e-02, 4.5888e-01, ..., 4.6424e-01,
9.0666e-01, -3.2583e-02],
[ 1.9330e-01, -7.8986e-01, -2.8753e-02, ..., 4.4595e-01,
3.8573e-01, -8.2767e-01],
[ 9.8972e-02, -6.3949e-01, 3.0504e-01, ..., 6.7268e-01,
5.4148e-01, -3.6093e-01],
...,
[-2.6430e-01, -4.4040e-01, 2.4199e-01, ..., 3.2920e-01,
3.1318e-01, 8.7513e-02],
[-5.5943e-01, -9.0586e-01, 5.5845e-02, ..., 5.1503e-03,
-3.7401e-02, -1.5828e-01],
[ 3.1171e-01, -2.1629e-02, -2.0716e-01, ..., -1.4516e-01,
-1.1659e-01, 1.7689e-01]],
[[-1.5501e+00, -8.0836e-01, -4.1025e-01, ..., -4.5024e-01,
-2.8304e-01, 8.8127e-01],
[-2.1315e+00, -7.8023e-01, -5.5620e-01, ..., -8.1335e-01,
-1.9715e-01, 1.6697e+00],
[-1.4932e+00, -5.8111e-01, -2.0111e-01, ..., -4.1755e-01,
-1.3951e-01, 1.4090e+00],
...,
[-6.7732e-01, -2.1901e-01, -8.5203e-02, ..., -1.4078e-01,
-5.7848e-03, 6.2713e-01],
[-1.2035e+00, -5.6485e-01, -3.2234e-01, ..., -3.2694e-01,
-2.0522e-01, 1.5176e-01],
[-3.8852e-01, -2.1424e-02, -1.7037e-01, ..., -2.2869e-01,
-2.4016e-01, 3.1828e-01]]],
[[[ 8.0973e-03, -2.3641e-01, -2.2153e-02, ..., -9.9189e-01,
-5.5044e-01, -1.4342e+00],
[ 2.9145e-01, -9.7969e-01, -7.0143e-01, ..., -1.1757e+00,
-1.3211e+00, -1.4895e+00],
[ 2.2941e-01, -6.1024e-01, -5.1926e-01, ..., -8.9664e-01,
-1.0731e+00, -1.8325e+00],
...,
[-1.5218e-02, -6.4671e-01, -4.6154e-01, ..., -1.0908e+00,
-1.3242e+00, -1.9204e+00],
[-6.3801e-01, -7.2563e-01, -8.2813e-02, ..., -4.8010e-01,
-8.0868e-01, -1.7046e+00],
[-7.6182e-01, -1.3960e+00, -1.2213e+00, ..., -1.1493e+00,
-1.5529e+00, -7.5668e-01]],
[[-7.6135e-01, -2.1708e+00, -1.8840e+00, ..., -4.0185e+00,
-5.4602e+00, -1.8398e+00],
[-2.2121e+00, -3.8366e+00, -3.0894e+00, ..., -6.0739e+00,
-8.1139e+00, -3.4172e+00],
[-1.7990e+00, -2.8213e+00, -2.7531e+00, ..., -5.1187e+00,
-6.2548e+00, -2.6022e+00],
...,
[-1.7838e+00, -2.5046e+00, -2.5127e+00, ..., -5.4869e+00,
-6.6488e+00, -2.4636e+00],
[-3.8509e+00, -4.4909e+00, -3.6877e+00, ..., -6.4847e+00,
-7.9987e+00, -2.5346e+00],
[ 2.6859e-01, -3.7014e-01, 3.3550e-01, ..., 4.4173e-01,
4.7160e-01, 1.9653e-01]],
[[-2.1961e-01, 1.5993e-01, 7.2544e-01, ..., 7.0125e-01,
2.6026e-01, 1.5210e+00],
[ 4.1648e-02, -3.3992e-01, 6.0025e-01, ..., 6.5845e-01,
-2.6254e-02, 9.6002e-02],
[-1.9669e-01, -2.8890e-01, 2.7975e-01, ..., 2.3289e-02,
2.4007e-02, -1.5530e-01],
...,
[-2.7631e-01, -2.4634e-01, 9.3904e-02, ..., -9.8789e-01,
1.1186e-01, -1.6698e-01],
[-2.7811e-01, -2.2446e-01, 8.9092e-02, ..., -3.3533e-01,
6.5449e-02, -2.2618e-01],
[-1.4273e-01, -4.6807e-01, 9.8608e-02, ..., 9.3203e-02,
2.0634e-01, -5.9832e-01]],
...,
[[ 5.8257e-01, -2.7876e-01, -1.1580e-02, ..., 4.2012e-01,
3.2158e-01, -1.3287e+00],
[-4.9740e-01, -5.5352e-01, -5.4303e-01, ..., -1.9192e-01,
-7.7173e-01, -1.3891e+00],
[-2.5897e-01, -2.1924e-01, -2.9710e-01, ..., -1.3176e-01,
-1.1041e-01, -9.4297e-01],
...,
[-4.2140e-01, -1.0688e-01, -1.0070e-01, ..., -1.2272e+00,
-4.2761e-01, -9.7340e-01],
[-2.3426e+00, -1.6963e+00, -1.2588e+00, ..., -1.4368e+00,
-1.3432e+00, -1.8148e+00],
[-5.0256e-01, -8.3989e-01, -4.5208e-01, ..., -3.6988e-01,
-5.2212e-01, -1.1933e+00]],
[[ 1.5623e-01, 2.7445e-01, 6.3961e-01, ..., -1.8237e-01,
3.8895e-01, -3.5947e-01],
[ 1.2834e-01, 2.2449e-01, 4.2138e-01, ..., 1.4294e-01,
4.1812e-02, -8.2712e-01],
[-1.2393e-01, 1.3820e-01, 2.0094e-01, ..., 2.5988e-01,
9.8437e-02, -4.7283e-01],
...,
[-2.0884e-01, 3.4025e-02, 3.6276e-01, ..., 1.3469e+00,
5.0719e-03, -5.1220e-01],
[-5.8207e-01, -7.2163e-01, -1.9664e-02, ..., -1.8324e-01,
-4.3205e-01, -9.0607e-01],
[ 4.9850e-01, 3.6023e-01, 4.1005e-02, ..., 3.3625e-01,
3.2305e-01, 2.8948e-01]],
[[-8.3087e-01, 2.6553e-01, 2.9899e-01, ..., -6.6104e-02,
-9.3605e-02, 5.7754e-01],
[-1.1496e+00, 2.4066e-01, -3.8899e-01, ..., -8.4921e-01,
-4.4901e-01, 8.7888e-01],
[-9.8352e-01, 4.1495e-01, -8.2335e-02, ..., -5.1967e-01,
-5.1331e-01, 6.3956e-01],
...,
[-9.7636e-01, 1.6310e-01, -1.6307e-01, ..., 1.8072e-01,
-1.3509e-01, 6.3694e-01],
[-1.7083e+00, -4.3863e-01, -4.3345e-01, ..., -1.5092e-01,
-4.9242e-01, 4.0944e-01],
[-7.5090e-01, -1.0647e-01, -3.2568e-01, ..., -4.8721e-01,
-4.2837e-01, 9.7988e-01]]]], device='cuda:0'),
tensor([[[[-6.1728e-01, -1.4151e+00, -1.4676e+00, ..., -1.2576e+00,
-8.7402e-01, -1.7025e+00],
[-2.3096e-01, -2.5684e+00, -2.3471e+00, ..., -1.7459e+00,
-1.8382e+00, -1.9339e+00],
[-1.7136e-01, -1.9801e+00, -1.6977e+00, ..., -1.2285e+00,
-1.3683e+00, -2.1003e+00],
...,
[ 7.6830e-02, -5.3892e-01, -2.0249e-01, ..., -1.1321e-01,
-3.5204e-02, -4.3571e-01],
[-3.1473e-01, -4.4597e-01, 3.2973e-02, ..., 1.4197e-01,
3.0618e-01, -1.3682e-01],
[-6.1065e-01, -1.3296e+00, -1.1013e+00, ..., -1.0773e+00,
-1.1689e+00, -4.9086e-01]],
[[-2.6310e+00, -4.9882e+00, -4.4204e+00, ..., -4.8940e+00,
-6.0079e+00, -2.4613e+00],
[-5.5760e+00, -8.4038e+00, -7.4551e+00, ..., -7.7405e+00,
-9.0842e+00, -4.3267e+00],
[-3.9369e+00, -5.8707e+00, -5.7008e+00, ..., -5.9413e+00,
-6.4155e+00, -3.2209e+00],
...,
[-1.7291e+00, -2.1622e+00, -1.7511e+00, ..., -1.8416e+00,
-2.2729e+00, -9.1828e-01],
[-2.7670e+00, -2.9310e+00, -1.9298e+00, ..., -2.2140e+00,
-3.3452e+00, -9.9589e-01],
[ 3.3722e-01, -2.4466e-01, 3.1928e-01, ..., 3.0573e-01,
4.6088e-01, 2.1157e-01]],
[[-6.8483e-01, 3.2254e-01, 8.6126e-01, ..., 8.4868e-01,
6.4265e-01, 1.8530e+00],
[-9.3441e-01, -1.2423e-01, 5.4545e-01, ..., 4.9802e-01,
2.4156e-02, 1.2022e-01],
[-1.6916e+00, -3.6535e-01, 2.1624e-02, ..., 3.7995e-02,
3.0321e-02, -5.5905e-02],
...,
[-6.4119e-01, 6.9436e-02, 2.7017e-01, ..., 1.6088e-01,
2.3005e-01, 4.5003e-01],
[-4.5542e-01, 9.1368e-02, 2.3757e-01, ..., 1.7005e-01,
2.2280e-01, 5.7919e-01],
[-1.6446e-01, -2.5370e-01, 1.5801e-01, ..., -1.5066e-03,
1.0081e-01, 3.5305e-01]],
...,
[[ 1.0503e+00, -4.8781e-01, -6.6360e-02, ..., 2.2636e-01,
-5.3754e-02, -2.0603e+00],
[-8.9081e-01, -7.9077e-01, -5.3178e-01, ..., -5.1268e-01,
-1.4261e+00, -2.0411e+00],
[-7.2385e-01, -6.7184e-01, -3.2384e-01, ..., -3.1838e-01,
-6.1061e-01, -1.3819e+00],
...,
[-6.6044e-01, -5.2267e-01, -3.2242e-01, ..., -3.2436e-01,
-4.7922e-01, -5.4784e-01],
[-2.2379e+00, -1.7678e+00, -1.2920e+00, ..., -1.3019e+00,
-1.3719e+00, -1.1000e+00],
[-6.0537e-01, -8.1371e-01, -4.5948e-01, ..., -2.5023e-01,
-3.5591e-01, -3.9678e-01]],
[[-5.9073e-02, -9.8059e-02, 4.3311e-01, ..., 5.0586e-01,
8.9907e-01, -4.3928e-02],
[ 4.5539e-02, -8.5336e-01, -1.3963e-01, ..., 4.1723e-01,
3.5263e-01, -8.4858e-01],
[ 4.2727e-03, -6.7000e-01, 2.6744e-01, ..., 6.9065e-01,
5.6359e-01, -4.0612e-01],
...,
[-2.3823e-01, -4.5962e-01, 2.2767e-01, ..., 3.2572e-01,
3.1291e-01, 3.2468e-02],
[-5.1183e-01, -9.0321e-01, 5.8574e-02, ..., 8.4122e-03,
-1.9646e-02, -1.4881e-01],
[ 3.0924e-01, -4.0728e-02, -1.9708e-01, ..., -1.3885e-01,
-1.0266e-01, 1.6631e-01]],
[[-1.5967e+00, -7.7771e-01, -4.5495e-01, ..., -4.3866e-01,
-2.9255e-01, 8.3482e-01],
[-2.0534e+00, -6.7311e-01, -5.9023e-01, ..., -7.4239e-01,
-1.6000e-01, 1.6585e+00],
[-1.5256e+00, -6.2093e-01, -2.2974e-01, ..., -3.5790e-01,
-7.2280e-02, 1.4251e+00],
...,
[-6.3353e-01, -1.8774e-01, -1.2400e-01, ..., -1.1146e-01,
1.8812e-02, 5.9170e-01],
[-1.0944e+00, -5.3216e-01, -3.4051e-01, ..., -3.3922e-01,
-1.9452e-01, 1.4981e-01],
[-3.4100e-01, 1.8739e-02, -1.5898e-01, ..., -2.4111e-01,
-2.2920e-01, 3.1120e-01]]],
[[[-3.2676e-01, -9.6827e-01, -5.1801e-01, ..., -5.6875e-01,
-9.4580e-02, -8.2825e-01],
[ 2.4391e-02, -1.9941e+00, -1.2203e+00, ..., -9.6902e-01,
-8.0562e-01, -1.2109e+00],
[-2.9227e-02, -1.6081e+00, -9.8229e-01, ..., -7.5388e-01,
-7.9944e-01, -1.7467e+00],
...,
[ 9.7610e-02, -5.2684e-01, -2.3868e-01, ..., -1.0189e-01,
-3.6268e-02, -4.3458e-01],
[-3.7170e-01, -4.3860e-01, 3.2719e-02, ..., 1.4010e-01,
2.9793e-01, -1.4708e-01],
[-6.3087e-01, -1.2671e+00, -1.0861e+00, ..., -1.0705e+00,
-1.1607e+00, -4.9265e-01]],
[[-1.9145e+00, -3.8298e+00, -3.9364e+00, ..., -3.9709e+00,
-4.8977e+00, -1.9686e+00],
[-3.9146e+00, -6.2269e+00, -5.2586e+00, ..., -5.4499e+00,
-6.7398e+00, -2.8711e+00],
[-3.4293e+00, -5.2142e+00, -4.7729e+00, ..., -5.1490e+00,
-5.7362e+00, -2.7598e+00],
...,
[-1.6772e+00, -2.0861e+00, -1.7238e+00, ..., -1.8216e+00,
-2.2211e+00, -8.9345e-01],
[-2.7435e+00, -2.8840e+00, -1.9914e+00, ..., -2.2225e+00,
-3.3266e+00, -9.8081e-01],
[ 3.4862e-01, -2.3766e-01, 3.1083e-01, ..., 2.8407e-01,
4.2096e-01, 1.9398e-01]],
[[-5.1904e-01, 4.0551e-01, 8.5203e-01, ..., 7.8335e-01,
5.7421e-01, 1.4603e+00],
[-6.8311e-01, -3.5113e-02, 5.8511e-01, ..., 3.7967e-01,
2.7240e-01, 4.1395e-01],
[-1.1948e+00, 2.6613e-02, 2.6561e-01, ..., 1.6552e-01,
1.9434e-01, 3.1620e-01],
...,
[-5.9122e-01, 7.3122e-02, 2.4328e-01, ..., 1.6049e-01,
2.2713e-01, 4.3028e-01],
[-4.0972e-01, 1.1212e-01, 2.0931e-01, ..., 1.8078e-01,
2.2681e-01, 5.8339e-01],
[-1.5918e-01, -2.4195e-01, 1.1732e-01, ..., -2.2788e-02,
9.7739e-02, 3.4062e-01]],
...,
[[ 9.4684e-01, 2.6213e-01, 3.9445e-01, ..., 4.4853e-01,
2.3337e-01, -9.8334e-01],
[-1.0184e+00, -6.1611e-01, -3.0224e-01, ..., -3.1949e-01,
-1.1195e+00, -1.3024e+00],
[-4.3488e-01, -3.2384e-01, -3.7116e-01, ..., -2.9196e-01,
-5.3939e-01, -1.1739e+00],
...,
[-6.4155e-01, -5.4582e-01, -3.4426e-01, ..., -3.5139e-01,
-5.0956e-01, -5.5544e-01],
[-2.1755e+00, -1.7325e+00, -1.2924e+00, ..., -1.3116e+00,
-1.3556e+00, -1.0918e+00],
[-5.7537e-01, -8.3570e-01, -4.2916e-01, ..., -2.6536e-01,
-3.5802e-01, -3.9391e-01]],
[[ 1.2492e-01, 2.0431e-01, 4.1924e-01, ..., 2.1354e-01,
6.6281e-01, -3.0549e-01],
[ 1.8967e-01, -6.4608e-01, -3.5822e-02, ..., 1.7276e-01,
3.8152e-01, -7.4578e-01],
[-1.6602e-01, -8.0834e-01, 1.4056e-01, ..., 2.3976e-01,
1.6708e-01, -6.5248e-01],
...,
[-1.9250e-01, -4.5409e-01, 1.9905e-01, ..., 3.2330e-01,
3.3473e-01, 5.1350e-02],
[-4.9023e-01, -9.2104e-01, 4.9714e-02, ..., -1.5409e-02,
-6.2853e-03, -1.4141e-01],
[ 3.8464e-01, 8.7109e-03, -1.8120e-01, ..., -1.4309e-01,
-9.2685e-02, 1.7486e-01]],
[[-7.9907e-01, -6.2153e-02, 2.6541e-01, ..., 3.2779e-01,
4.4731e-01, 9.8265e-01],
[-1.1815e+00, -5.3722e-01, -7.1368e-01, ..., -6.3165e-01,
-2.3578e-01, 1.0095e+00],
[-8.7468e-01, -5.8511e-01, -4.6026e-01, ..., -3.8594e-01,
-8.2516e-02, 8.0747e-01],
...,
[-6.2245e-01, -1.9365e-01, -1.4164e-01, ..., -1.6068e-01,
3.1984e-03, 6.1445e-01],
[-1.1468e+00, -5.7513e-01, -3.1451e-01, ..., -3.4880e-01,
-1.8742e-01, 1.6815e-01],
[-3.6527e-01, -3.1317e-02, -1.5316e-01, ..., -2.4963e-01,
-2.2203e-01, 3.3093e-01]]],
[[[-2.9457e-01, -4.5747e-01, -1.3932e-01, ..., -9.2617e-01,
-4.8708e-01, -1.4422e+00],
[ 1.6610e-03, -1.6150e+00, -7.7827e-01, ..., -1.1563e+00,
-1.3034e+00, -1.5139e+00],
[-1.4794e-01, -1.5109e+00, -5.3728e-01, ..., -9.1767e-01,
-1.0872e+00, -1.8142e+00],
...,
[-2.6481e-01, -1.2427e+00, -4.8446e-01, ..., -1.0714e+00,
-1.0010e+00, -1.9365e+00],
[-4.0092e-01, -1.0426e+00, -1.2736e-01, ..., -5.5734e-01,
-5.3602e-01, -1.7131e+00],
[-3.7478e-01, -1.4676e+00, -1.0908e+00, ..., -1.1297e+00,
-1.5339e+00, -7.7580e-01]],
[[-6.0720e-01, -1.9999e+00, -6.9223e-01, ..., -4.0909e+00,
-5.5338e+00, -1.8840e+00],
[-2.4400e+00, -4.4485e+00, -2.0782e+00, ..., -5.9119e+00,
-8.0626e+00, -3.4144e+00],
[-1.7489e+00, -3.2287e+00, -1.5441e+00, ..., -5.0019e+00,
-6.1957e+00, -2.5984e+00],
...,
[-1.6233e+00, -2.6712e+00, -1.5968e+00, ..., -5.2154e+00,
-6.3431e+00, -2.4719e+00],
[-3.4796e+00, -4.1795e+00, -2.6700e+00, ..., -6.5566e+00,
-7.9406e+00, -2.5295e+00],
[ 6.0571e-01, -6.2806e-01, 4.5069e-01, ..., 4.1786e-01,
4.7145e-01, 2.3194e-01]],
[[-4.8354e-01, 6.3751e-01, 1.0659e+00, ..., 6.2328e-01,
3.3532e-01, 1.5567e+00],
[-8.1782e-01, 1.7796e-01, 5.7212e-01, ..., 4.7388e-01,
-1.0533e-01, 8.2958e-02],
[-1.3925e+00, 2.6181e-01, 1.8754e-01, ..., -2.5587e-02,
-4.9513e-02, -1.6351e-01],
...,
[-1.3290e+00, 1.5020e-01, 6.2780e-02, ..., 4.9228e-01,
-1.0701e-01, -1.5803e-01],
[-1.2853e+00, 2.2328e-01, 1.5460e-01, ..., -8.5912e-02,
1.1665e-02, -1.5674e-01],
[-8.5986e-01, 6.3791e-02, -1.0452e-01, ..., 4.7721e-02,
2.3078e-01, -5.4289e-01]],
...,
[[ 2.5388e-01, -2.5674e-01, -1.5322e-01, ..., 5.5348e-01,
4.2631e-01, -1.3045e+00],
[-1.6615e+00, -1.1539e+00, -7.1457e-01, ..., -2.1043e-01,
-7.5325e-01, -1.4107e+00],
[-8.6128e-01, -4.2519e-01, -9.4800e-02, ..., -1.1294e-01,
-9.8217e-02, -9.7270e-01],
...,
[-8.8805e-01, -6.8432e-01, -1.3638e-01, ..., -3.1993e-01,
-1.1145e-01, -9.1749e-01],
[-2.3986e+00, -1.6319e+00, -7.9729e-01, ..., -1.5126e+00,
-1.5765e+00, -1.8150e+00],
[-6.1871e-01, -1.0311e+00, -8.2445e-02, ..., -3.7230e-01,
-5.4957e-01, -1.1938e+00]],
[[-1.8346e-02, 2.4929e-01, 5.3269e-01, ..., -6.1796e-02,
4.4779e-01, -3.6661e-01],
[-3.1447e-01, -7.0931e-01, -1.2390e-01, ..., 2.2498e-01,
1.4562e-01, -8.7903e-01],
[-6.4096e-01, -1.1010e+00, -1.8139e-01, ..., 2.7909e-01,
1.9583e-01, -4.6447e-01],
...,
[-5.2511e-01, -9.3736e-01, -6.6598e-02, ..., 1.4916e-01,
1.1450e-01, -5.0650e-01],
[-9.7906e-01, -1.8012e+00, -3.5090e-01, ..., -2.8879e-01,
-2.9191e-01, -9.0872e-01],
[ 2.3618e-01, -4.6795e-01, -2.2099e-01, ..., 3.1270e-01,
3.3380e-01, 3.1901e-01]],
[[-5.0118e-01, 1.2089e-01, 5.7702e-01, ..., -1.2846e-01,
-4.2781e-02, 5.9359e-01],
[-4.1962e-01, 3.0340e-02, 5.2454e-03, ..., -9.6861e-01,
-3.9009e-01, 8.8093e-01],
[ 3.7182e-02, 2.9047e-02, 4.8919e-02, ..., -6.3557e-01,
-5.0125e-01, 6.5238e-01],
...,
[-1.3365e-01, -5.3707e-02, 9.8773e-02, ..., -1.4506e-01,
-5.4693e-01, 5.9146e-01],
[-4.3557e-01, -4.1697e-01, -1.1918e-01, ..., -6.0155e-01,
-7.0334e-01, 3.7130e-01],
[ 8.8190e-02, -4.3193e-01, -1.1434e-01, ..., -4.7281e-01,
-4.4079e-01, 1.0011e+00]]],
[[[-2.2145e-01, -6.0166e-01, -1.4365e+00, ..., -1.1172e-01,
3.2330e-01, -1.8382e-02],
[-1.0611e-01, -1.0349e+00, -2.0736e-01, ..., -8.1937e-01,
-5.7282e-01, -5.3678e-01],
[-8.5339e-02, -6.7996e-01, 2.2571e+00, ..., -5.5371e-01,
-4.6710e-01, -9.1331e-01],
...,
[-2.0762e-01, -1.9687e+00, -6.5150e+00, ..., -5.7609e+00,
-3.0356e+00, -2.6967e+00],
[-5.4024e-01, -1.7165e+00, -3.4378e+00, ..., -4.1576e+00,
-2.0980e+00, -2.2640e+00],
[-3.0989e-01, -9.8279e-01, -1.3940e+00, ..., -1.2070e+00,
-9.4492e-01, -1.1205e+00]],
[[-5.4584e-01, -4.6756e+00, -9.3525e+00, ..., -1.2410e+00,
-1.9287e+00, -5.7588e-01],
[-2.3586e+00, -5.1993e+00, -1.0670e+01, ..., -3.0109e+00,
-4.0442e+00, -1.4777e+00],
[-2.1369e+00, -3.4004e+00, -6.6962e+00, ..., -3.1148e+00,
-3.6051e+00, -1.5658e+00],
...,
[-3.9935e+00, -7.5053e+00, -7.6407e+00, ..., -1.1667e+01,
-8.8398e+00, -5.0592e+00],
[-5.9110e+00, -9.6757e+00, -1.0282e+01, ..., -9.2818e+00,
-1.0529e+01, -4.8198e+00],
[ 4.2486e-01, -3.6326e-01, 1.4385e-01, ..., 8.5886e-01,
-9.9054e-01, -5.0299e-01]],
[[-4.8662e-01, -7.3029e-01, -4.5179e+00, ..., 5.2764e-01,
2.4596e-01, 1.0368e+00],
[-4.1004e-01, -5.4527e-02, -3.4072e+00, ..., 4.0871e-01,
3.2790e-01, 4.9300e-01],
[-9.4613e-01, 9.5928e-01, 8.7619e-01, ..., 1.1058e-01,
1.8243e-01, 3.8531e-01],
...,
[-1.8812e+00, -9.3110e-01, -3.4395e+00, ..., -1.0351e+01,
-3.1908e+00, -6.4015e-02],
[-1.5916e+00, 5.1907e-01, -5.8267e+00, ..., -6.2378e+00,
-2.3974e+00, 2.3696e-01],
[-8.3329e-01, 4.0267e-01, -3.5313e+00, ..., -5.0945e-01,
-4.0022e-01, -9.4647e-02]],
...,
[[ 1.2635e-01, 6.6562e-02, -1.1911e+00, ..., 1.6994e-02,
-1.2514e-01, -7.2888e-01],
[-8.0429e-01, 5.4496e-01, 2.4788e+00, ..., -2.4065e-01,
-1.0235e+00, -8.4616e-01],
[-5.2900e-01, -5.2223e-01, 1.2897e+00, ..., 3.3039e-02,
-1.9091e-01, -5.9179e-01],
...,
[-1.0114e+00, 5.5926e-01, 1.1587e+00, ..., -9.8853e+00,
-6.6792e+00, -2.7202e+00],
[-2.6659e+00, -2.0207e+00, -3.3092e+00, ..., -8.1085e+00,
-7.0509e+00, -2.7622e+00],
[-6.1322e-01, -4.8536e-01, 1.7542e+00, ..., -6.7348e+00,
-3.4224e+00, -5.5358e-01]],
[[-1.3613e-01, 2.4588e-01, -1.5414e+00, ..., 7.9097e-02,
5.2825e-01, 9.5685e-02],
[-3.4560e-01, -4.8841e-01, 6.1504e-01, ..., -1.0347e-02,
2.0358e-01, -1.5420e-01],
[-5.7440e-01, -7.0558e-01, 1.1207e+00, ..., 8.3248e-02,
8.0121e-02, -1.2533e-01],
...,
[-1.7212e-03, -6.8054e-01, -2.9636e+00, ..., -8.9739e+00,
-5.1072e+00, -2.5442e+00],
[-6.5990e-01, -1.7969e+00, -2.7812e+00, ..., -5.5527e+00,
-1.1848e+00, -1.9856e+00],
[ 7.8468e-01, -1.0339e-01, -4.4872e-01, ..., 4.7448e+00,
4.5828e+00, 9.4763e-01]],
[[-5.8583e-01, -8.7444e-02, -1.3845e+00, ..., 5.0795e-01,
4.0044e-01, 7.2547e-01],
[-8.6458e-01, -5.0544e-01, -2.7300e+00, ..., -2.4427e-01,
-1.1421e-01, 8.0109e-01],
[-5.1283e-01, -1.0433e-01, -1.3428e+00, ..., 9.5282e-02,
1.7744e-01, 7.1739e-01],
...,
[-1.5367e+00, -2.8349e+00, -6.4507e+00, ..., -7.1909e+00,
6.2423e-01, 2.5869e+00],
[-1.7008e+00, -2.7727e+00, -2.6462e+00, ..., -7.2864e+00,
5.8711e-01, 2.5604e+00],
[-8.1724e-01, -1.1631e+00, 2.4858e-02, ..., -8.6444e-01,
1.0123e+00, 2.1027e+00]]],
[[[ 3.3391e-02, -2.6754e-01, -6.6614e-02, ..., -9.7527e-01,
-5.4481e-01, -1.4054e+00],
[ 2.9533e-01, -9.3278e-01, -7.0139e-01, ..., -1.1716e+00,
-1.3754e+00, -1.4872e+00],
[ 2.3460e-01, -5.8984e-01, -5.0883e-01, ..., -1.0207e+00,
-1.2132e+00, -1.9802e+00],
...,
[ 4.2299e-02, -9.8897e-02, -1.3898e+00, ..., -1.5553e+00,
-1.0269e+00, -1.9794e+00],
[-6.0347e-01, -9.8331e-03, 2.3706e-01, ..., -4.6689e-01,
-5.6183e-01, -1.7704e+00],
[-7.2961e-01, -1.1541e+00, -8.8115e-01, ..., -1.0043e+00,
-1.3797e+00, -6.9774e-01]],
[[-7.6620e-01, -2.1946e+00, -1.9454e+00, ..., -3.8519e+00,
-5.4105e+00, -1.7832e+00],
[-2.0054e+00, -3.6852e+00, -3.0833e+00, ..., -5.9393e+00,
-8.2407e+00, -3.4506e+00],
[-1.6332e+00, -2.5962e+00, -2.6617e+00, ..., -6.2182e+00,
-6.4685e+00, -2.7055e+00],
...,
[-1.8486e+00, -3.3122e+00, -3.7912e+00, ..., -5.5396e+00,
-6.5025e+00, -2.5416e+00],
[-3.8944e+00, -4.5709e+00, -3.7164e+00, ..., -6.1448e+00,
-7.6560e+00, -2.4640e+00],
[ 2.9816e-01, -3.9897e-01, 3.0754e-01, ..., 4.0076e-01,
4.3218e-01, 2.2597e-01]],
[[-2.0828e-01, 1.8256e-01, 7.4076e-01, ..., 6.7594e-01,
2.6095e-01, 1.6450e+00],
[ 3.0409e-02, -3.3860e-01, 5.4447e-01, ..., 5.9882e-01,
-1.9187e-01, 1.2575e-01],
[-1.5943e-01, -2.2577e-01, 1.4524e-01, ..., 3.7248e-01,
-6.2906e-02, -8.7246e-02],
...,
[-2.7275e-01, -1.2100e+00, -4.7075e+00, ..., 3.4670e-02,
-2.6459e-02, -1.1457e-01],
[-2.3078e-01, 5.8105e-01, -4.9976e-01, ..., -1.3226e-01,
-3.0563e-02, -2.1617e-01],
[-1.2413e-01, -1.6130e-01, 7.8220e-02, ..., 7.2244e-02,
1.0746e-01, -6.2018e-01]],
...,
[[ 6.3046e-01, -1.9044e-01, 2.5053e-02, ..., 4.4826e-01,
3.2925e-01, -1.4131e+00],
[-4.7709e-01, -4.9504e-01, -4.7730e-01, ..., -1.9038e-01,
-8.0379e-01, -1.4723e+00],
[-3.5130e-01, -2.0035e-01, -2.2510e-01, ..., 2.3087e-01,
4.8672e-04, -1.0051e+00],
...,
[-3.8543e-01, 8.7208e-01, 4.8552e-01, ..., 9.6766e-03,
-1.8169e-01, -9.7264e-01],
[-2.3509e+00, -1.7338e+00, -1.0650e+00, ..., -1.3736e+00,
-1.4705e+00, -1.7764e+00],
[-4.8981e-01, -1.0472e+00, -6.4902e-01, ..., -3.8126e-01,
-4.9938e-01, -1.1698e+00]],
[[ 1.7252e-01, 2.6854e-01, 6.0800e-01, ..., -7.3827e-02,
4.8102e-01, -2.9983e-01],
[ 1.6185e-01, 2.3741e-01, 4.7798e-01, ..., 2.0608e-01,
1.3955e-01, -7.8297e-01],
[-1.0580e-01, 9.6512e-02, 3.0190e-01, ..., -2.2947e-01,
1.6280e-01, -4.6729e-01],
...,
[-1.8562e-01, -3.2754e-01, -4.3146e-01, ..., -6.2126e-01,
4.4260e-02, -5.4772e-01],
[-5.6067e-01, -5.8427e-01, 5.1358e-02, ..., 2.0742e-02,
-9.7993e-02, -8.7196e-01],
[ 5.2891e-01, 4.7508e-01, 1.4820e-01, ..., 4.0549e-01,
3.6302e-01, 2.8686e-01]],
[[-8.2722e-01, 2.4362e-01, 3.2828e-01, ..., -7.3865e-02,
-8.8751e-02, 6.2546e-01],
[-1.1188e+00, 1.5180e-01, -3.7207e-01, ..., -9.7116e-01,
-4.3005e-01, 9.1195e-01],
[-9.5407e-01, 2.9480e-01, -1.2506e-01, ..., -2.2421e-01,
-6.1238e-01, 5.3419e-01],
...,
[-9.7499e-01, -5.7127e-01, -1.8815e+00, ..., -5.7510e-01,
-4.8553e-01, 7.0974e-01],
[-1.7079e+00, -7.8913e-01, -1.2023e-01, ..., -7.4812e-01,
-6.4358e-01, 4.8194e-01],
[-7.4496e-01, -9.0400e-02, 1.0841e-01, ..., -4.7613e-01,
-3.9906e-01, 1.0619e+00]]]], device='cuda:0'),
tensor([[[[ 5.2083e-02, -2.2782e-01, -1.2751e-02, ..., -1.1768e+00,
-6.1891e-01, -9.3905e-01],
[ 3.1392e-01, -1.0913e+00, -7.6039e-01, ..., -1.6012e+00,
-1.2690e+00, -1.2173e+00],
[-5.6276e-03, -9.1370e-01, -5.7519e-01, ..., -2.1605e+00,
-1.7191e+00, -1.8502e+00],
...,
[ 5.4551e-02, -5.5160e-01, -4.1887e-01, ..., -2.0669e+00,
-1.6096e+00, -1.6743e+00],
[-4.9681e-01, -6.1341e-01, -7.3568e-02, ..., -7.9293e-01,
-5.4973e-01, -1.4284e+00],
[-6.2560e-01, -1.2392e+00, -1.0314e+00, ..., -1.3037e+00,
-1.4952e+00, -9.2189e-01]],
[[-6.6856e-01, -2.1187e+00, -1.6972e+00, ..., -4.1648e+00,
-4.8646e+00, -1.8340e+00],
[-2.2247e+00, -4.1634e+00, -3.0093e+00, ..., -6.1196e+00,
-6.9231e+00, -2.7809e+00],
[-1.9598e+00, -2.9591e+00, -2.4886e+00, ..., -7.5384e+00,
-7.9992e+00, -3.0110e+00],
...,
[-1.7080e+00, -2.3622e+00, -2.4614e+00, ..., -4.9648e+00,
-5.4226e+00, -2.3705e+00],
[-3.5744e+00, -4.0906e+00, -3.3928e+00, ..., -6.2559e+00,
-7.3137e+00, -2.8270e+00],
[ 2.9312e-01, -2.7912e-01, 3.7195e-01, ..., 5.9098e-01,
6.4804e-01, 4.9382e-02]],
[[-2.4789e-01, 7.0680e-02, 7.7077e-01, ..., 9.4397e-01,
6.9853e-01, 1.5101e+00],
[ 1.1532e-01, -4.0680e-01, 6.0952e-01, ..., 2.5237e-01,
3.0557e-01, 2.7217e-01],
[-2.1007e-01, -1.9648e-01, 2.5582e-01, ..., -7.5553e-02,
3.7970e-01, 4.1645e-01],
...,
[-2.2342e-01, -2.1401e-01, 1.2400e-01, ..., -7.9221e-01,
6.7804e-02, 2.2318e-02],
[-1.9464e-01, -1.1281e-01, 1.4787e-01, ..., -1.2959e-01,
6.7693e-02, 1.2493e-01],
[-8.2348e-02, -4.2760e-01, 1.1850e-01, ..., -3.9416e-01,
-1.0477e-01, -2.9241e-01]],
...,
[[ 4.5628e-01, -3.9108e-01, -5.2218e-02, ..., -1.2279e-01,
-2.3054e-01, -1.4096e+00],
[-7.2349e-01, -5.9833e-01, -6.3123e-01, ..., -7.9384e-01,
-1.5328e+00, -1.6550e+00],
[-5.8190e-01, -3.2756e-01, -2.7894e-01, ..., -5.5739e-01,
-7.0608e-01, -1.3799e+00],
...,
[-3.7954e-01, -1.2904e-01, -1.6897e-01, ..., -1.0373e+00,
-8.9455e-01, -1.2044e+00],
[-2.1354e+00, -1.5281e+00, -1.1240e+00, ..., -1.8926e+00,
-1.9386e+00, -1.7492e+00],
[-4.5902e-01, -7.0936e-01, -3.9099e-01, ..., -7.4743e-01,
-7.9095e-01, -8.6530e-01]],
[[ 9.9637e-02, 2.5210e-01, 7.0493e-01, ..., 4.3139e-01,
7.6373e-01, -9.4185e-03],
[ 3.9985e-02, 2.6334e-01, 4.3104e-01, ..., 3.0252e-01,
4.3489e-01, -2.8844e-01],
[-3.0390e-01, -1.2553e-01, 2.1053e-01, ..., -1.0965e-01,
-2.7164e-01, -8.1157e-01],
...,
[-1.2091e-01, 6.5325e-02, 3.3043e-01, ..., -1.1195e-01,
-3.4569e-01, -3.3811e-01],
[-4.4803e-01, -6.0371e-01, 4.8296e-02, ..., -8.4790e-02,
-1.1997e-01, -6.1356e-01],
[ 5.4013e-01, 4.1886e-01, 1.0677e-01, ..., 2.2289e-01,
1.5344e-01, 3.0044e-01]],
[[-9.4759e-01, 2.9533e-01, 3.9681e-01, ..., 1.1372e-01,
1.9032e-01, 8.2052e-01],
[-1.2039e+00, 2.3927e-01, -3.8679e-01, ..., -6.7724e-02,
1.9155e-01, 1.3897e+00],
[-1.1756e+00, 2.5297e-01, -1.6222e-01, ..., 4.2875e-02,
1.7074e-01, 8.6182e-01],
...,
[-9.8310e-01, 1.9893e-01, -8.1852e-02, ..., 1.0171e+00,
6.9643e-01, 1.2409e+00],
[-1.6396e+00, -3.3664e-01, -3.2905e-01, ..., -1.5641e-01,
1.9709e-01, 8.3370e-01],
[-6.6530e-01, 2.3553e-02, -2.3244e-01, ..., 1.7874e-01,
1.3642e-01, 1.0695e+00]]],
[[[-3.5008e-01, -9.7284e-01, -5.8633e-01, ..., -5.8085e-01,
-6.9298e-02, -7.7070e-01],
[ 7.9883e-03, -1.8318e+00, -8.5143e-01, ..., -1.0117e+00,
-8.2997e-01, -1.1817e+00],
[ 2.4815e-02, -1.4031e+00, -2.2996e-01, ..., -4.2340e-01,
-8.3484e-01, -1.7457e+00],
...,
[ 9.4530e-02, -5.2495e-01, -2.8029e-01, ..., -9.3016e-02,
-4.0257e-02, -4.1318e-01],
[-3.6615e-01, -4.2078e-01, 4.8046e-03, ..., 1.4506e-01,
2.9707e-01, -1.4986e-01],
[-6.3462e-01, -1.3304e+00, -1.1046e+00, ..., -1.1048e+00,
-1.1805e+00, -4.9680e-01]],
[[-2.0038e+00, -3.9059e+00, -4.8946e+00, ..., -3.9833e+00,
-4.9091e+00, -1.9966e+00],
[-3.7742e+00, -5.9679e+00, -7.3188e+00, ..., -6.4846e+00,
-6.7706e+00, -2.9072e+00],
[-3.4763e+00, -5.2998e+00, -6.1938e+00, ..., -6.5031e+00,
-5.8248e+00, -2.7867e+00],
...,
[-1.7034e+00, -2.1558e+00, -1.7559e+00, ..., -1.8106e+00,
-2.2396e+00, -9.1144e-01],
[-2.7809e+00, -2.9621e+00, -1.9954e+00, ..., -2.2407e+00,
-3.3297e+00, -9.7698e-01],
[ 3.4152e-01, -2.5300e-01, 3.2855e-01, ..., 2.7637e-01,
4.1856e-01, 1.8868e-01]],
[[-4.6450e-01, 4.7148e-01, 5.2100e-01, ..., 8.2934e-01,
5.6353e-01, 1.4342e+00],
[-6.5384e-01, 2.3429e-02, 9.5799e-01, ..., 6.1771e-01,
2.5938e-01, 3.3890e-01],
[-1.1349e+00, 9.7992e-02, 1.9525e-01, ..., 8.5805e-01,
1.5951e-01, 3.1145e-01],
...,
[-6.0700e-01, 3.8308e-02, 2.6121e-01, ..., 1.6151e-01,
2.5275e-01, 4.7810e-01],
[-3.8304e-01, 7.3413e-02, 1.8249e-01, ..., 1.5126e-01,
1.8501e-01, 6.0039e-01],
[-1.6047e-01, -2.5925e-01, 1.0449e-01, ..., -2.7267e-02,
5.4953e-02, 3.5219e-01]],
...,
[[ 1.0776e+00, 3.7627e-01, 1.1082e+00, ..., 4.6395e-01,
2.4788e-01, -1.0033e+00],
[-6.7690e-01, -5.3446e-01, -4.4155e-01, ..., -8.6049e-02,
-1.2119e+00, -1.3431e+00],
[-2.2532e-01, -3.6891e-01, 1.4527e-01, ..., -7.9068e-01,
-5.6225e-01, -1.1713e+00],
...,
[-6.2660e-01, -5.3181e-01, -3.6994e-01, ..., -2.9653e-01,
-5.0585e-01, -5.7083e-01],
[-2.2363e+00, -1.7776e+00, -1.3102e+00, ..., -1.3292e+00,
-1.3713e+00, -1.0874e+00],
[-6.0168e-01, -8.0895e-01, -4.2780e-01, ..., -2.5811e-01,
-3.5817e-01, -4.0512e-01]],
[[ 1.7882e-01, 1.9863e-01, 2.8494e-01, ..., 1.9434e-01,
7.2244e-01, -2.5336e-01],
[ 3.7129e-01, -6.4167e-01, -6.4767e-02, ..., -5.7578e-01,
3.6168e-01, -7.5726e-01],
[-3.0513e-04, -7.1490e-01, 3.5096e-01, ..., -2.4000e-01,
1.3835e-01, -6.9511e-01],
...,
[-2.1017e-01, -4.4921e-01, 2.2023e-01, ..., 3.5499e-01,
3.1362e-01, 7.7699e-02],
[-4.9248e-01, -8.3304e-01, 5.3013e-02, ..., 5.7442e-03,
-3.1814e-02, -1.6597e-01],
[ 3.4442e-01, 2.7008e-02, -1.8598e-01, ..., -1.5247e-01,
-1.2192e-01, 1.7185e-01]],
[[-6.5349e-01, -4.1718e-02, -5.3975e-01, ..., 3.3049e-01,
4.0186e-01, 9.5740e-01],
[-1.1633e+00, -6.5589e-01, -1.1644e+00, ..., -7.0804e-01,
-2.3146e-01, 1.0141e+00],
[-8.8385e-01, -6.3140e-01, -8.2279e-01, ..., -5.1276e-01,
-7.3841e-02, 7.6791e-01],
...,
[-6.7021e-01, -1.7347e-01, -1.1352e-01, ..., -1.4796e-01,
1.4772e-02, 6.1715e-01],
[-1.1592e+00, -5.4841e-01, -2.8283e-01, ..., -3.3338e-01,
-1.8854e-01, 1.4369e-01],
[-3.4075e-01, -3.4086e-02, -1.8013e-01, ..., -2.4835e-01,
-2.3170e-01, 3.1764e-01]]],
[[[-7.5935e-01, -1.4230e+00, -1.3970e+00, ..., -5.5967e-02,
5.4642e-01, 4.6494e-01],
[-1.9225e+00, -6.7853e+00, -5.5542e+00, ..., -4.8666e-01,
-2.2106e-01, 9.7560e-02],
[-3.4018e+00, -5.7114e+00, -2.2801e+00, ..., -1.0306e+00,
-8.2005e-01, -4.1030e-01],
...,
[-5.0125e-01, -1.9275e+00, -1.8794e+00, ..., -3.7369e-01,
-3.3302e-01, -2.3575e-01],
[-9.8573e-01, -1.7163e+00, -1.2039e+00, ..., 4.4752e-02,
3.1555e-01, 8.1623e-04],
[-8.2378e-01, -1.6639e+00, -1.2715e+00, ..., -1.2010e+00,
-1.2736e+00, -4.8514e-01]],
[[-2.3984e+00, -4.8228e+00, -4.1831e+00, ..., -6.0199e-01,
-1.1139e+00, -7.6303e-02],
[-1.2069e+01, -1.5593e+01, -1.6859e+01, ..., -1.7916e+00,
-2.5881e+00, -6.1589e-01],
[-1.0499e+01, -9.1254e+00, -8.1513e+00, ..., -3.5039e+00,
-3.0316e+00, -9.6673e-01],
...,
[-4.4211e+00, -6.1484e+00, -5.5586e+00, ..., -1.7385e+00,
-1.9079e+00, -4.7359e-01],
[-6.4119e+00, -7.4558e+00, -6.3528e+00, ..., -2.8518e+00,
-3.7585e+00, -1.0888e+00],
[-3.4298e-02, -8.2220e-01, 3.7137e-01, ..., 3.8635e-01,
3.6708e-01, 5.6273e-02]],
[[-5.9501e-01, 5.9991e-01, 1.1075e+00, ..., 7.9260e-01,
3.2533e-01, 1.0468e+00],
[-3.1901e+00, -3.0847e+00, -1.0328e+00, ..., 4.1952e-01,
9.9755e-02, 4.7060e-01],
[-3.6560e+00, -5.1885e+00, -5.7712e+00, ..., 1.2661e-01,
1.1973e-01, 3.8962e-01],
...,
[-1.7395e+00, -6.7634e-02, -8.9526e-02, ..., -4.0999e-01,
-7.2243e-02, 3.9097e-01],
[-1.5500e+00, 6.0592e-02, 3.4561e-02, ..., 1.1597e-01,
1.2142e-01, 6.3001e-01],
[-1.4347e+00, -5.4312e-01, -2.9377e-01, ..., -1.2827e-01,
8.8124e-02, 3.9959e-01]],
...,
[[ 1.0683e+00, -4.3268e-01, -2.0876e-01, ..., -6.5523e-02,
-2.9227e-01, -8.5127e-01],
[-6.0779e-01, -1.2636e+00, -1.4574e+00, ..., -6.1638e-01,
-1.1506e+00, -7.5987e-01],
[ 1.5371e+00, -1.6864e+00, -3.4373e+00, ..., -4.5846e-01,
-3.3582e-01, -4.3481e-01],
...,
[-7.3079e-01, -1.1662e+00, -7.5923e-01, ..., -4.9688e-01,
-2.5856e-01, -2.6627e-01],
[-2.7660e+00, -2.8618e+00, -1.8599e+00, ..., -9.8459e-01,
-1.0113e+00, -8.8734e-01],
[-6.7205e-01, -1.6901e+00, -1.0219e+00, ..., -3.8351e-02,
-1.6607e-01, -4.0307e-01]],
[[ 1.4358e-02, 1.3217e-02, 6.3648e-01, ..., 2.9457e-01,
6.0745e-01, 2.5170e-01],
[ 9.3299e-01, -2.7475e+00, -2.5700e+00, ..., 1.9517e-02,
1.4438e-01, -2.7462e-02],
[-1.0143e+00, -4.3821e+00, -2.0046e+00, ..., -5.4010e-01,
-1.1245e+00, -4.3810e-01],
...,
[ 8.7730e-02, -7.8199e-01, 2.8189e-01, ..., -6.5321e-01,
-6.4313e-01, 5.7773e-02],
[-2.2205e-01, -1.4363e+00, 2.6983e-02, ..., -4.9277e-01,
-4.4925e-01, -1.6721e-01],
[ 6.5281e-01, 1.3119e-01, 2.7802e-01, ..., -3.0839e-01,
-2.9435e-01, 2.8754e-01]],
[[-1.6206e+00, -6.4489e-01, 2.2980e-02, ..., 5.0543e-01,
3.1782e-01, 5.1857e-01],
[-3.1616e+00, -2.0465e+00, -7.7663e-01, ..., -3.2091e-02,
-1.5251e-01, 3.7561e-01],
[-6.5195e+00, -5.6787e+00, -3.3026e+00, ..., -5.5383e-01,
-5.8400e-01, -8.0823e-03],
...,
[-1.7708e+00, -4.8328e-01, 2.5054e-01, ..., 9.5169e-01,
2.6585e-01, 2.2609e-01],
[-2.2254e+00, -1.1676e+00, -1.7051e-01, ..., -7.4235e-02,
-2.9169e-01, -2.5029e-01],
[-1.0781e+00, -2.1219e-01, 4.6881e-01, ..., -1.7853e-01,
-3.6053e-01, 9.6490e-02]]],
[[[-2.1281e-01, -2.6864e-01, 9.2646e-02, ..., -3.9886e-04,
4.3675e-01, 1.0527e-01],
[-8.2493e-02, -1.9092e+00, -2.8328e+00, ..., -7.4681e-01,
-4.5219e-01, -3.6633e-01],
[-3.5155e-03, -1.7311e+00, -3.7740e+00, ..., -5.4500e-01,
-4.2715e-01, -7.2408e-01],
...,
[-1.6301e-01, -1.5362e+00, -7.8742e-01, ..., -1.4305e+00,
-9.6275e-01, -2.0676e+00],
[-8.9349e-01, -1.5030e+00, -6.9545e-01, ..., -7.1899e-01,
-4.8444e-01, -1.8144e+00],
[-1.0499e+00, -1.9117e+00, -1.5579e+00, ..., -1.7436e+00,
-1.6879e+00, -1.0702e+00]],
[[-2.7820e-01, -1.6755e+00, -7.5148e-01, ..., -7.5005e-01,
-1.4264e+00, -3.0579e-01],
[-2.3135e+00, -6.7915e+00, -5.1752e+00, ..., -2.4474e+00,
-3.4065e+00, -1.1741e+00],
[-2.1375e+00, -6.1460e+00, -5.4630e+00, ..., -2.7142e+00,
-3.0899e+00, -1.2887e+00],
...,
[-3.8487e+00, -5.5285e+00, -5.7223e+00, ..., -6.1341e+00,
-6.2222e+00, -2.8888e+00],
[-5.9551e+00, -6.7554e+00, -6.4916e+00, ..., -6.8634e+00,
-7.7830e+00, -2.8480e+00],
[ 4.0385e-02, -5.9900e-01, 2.6964e-01, ..., 3.0142e-01,
6.7099e-01, 1.5190e-01]],
[[-5.1685e-01, -1.7036e-01, 2.9555e-01, ..., 3.5765e-01,
8.9433e-02, 8.9173e-01],
[-3.3466e-01, -1.0419e+00, -3.9593e-01, ..., 4.0827e-01,
2.9442e-01, 5.0182e-01],
[-8.2753e-01, -1.8913e+00, -5.1164e+00, ..., 6.2584e-02,
1.9254e-01, 4.1864e-01],
...,
[-1.4071e+00, 5.6059e-02, 9.0446e-01, ..., 3.3972e-01,
1.1130e-02, 2.7532e-01],
[-1.1919e+00, 1.1078e-01, -1.6617e-02, ..., 6.8514e-02,
-3.6752e-02, 2.7153e-01],
[-1.0857e+00, -1.0524e-01, -3.0482e-02, ..., 9.9508e-02,
-7.2909e-02, -4.0132e-01]],
...,
[[ 8.2602e-02, -5.7845e-01, -3.6582e-01, ..., -7.5916e-02,
-2.1590e-01, -7.1056e-01],
[-7.9591e-01, -2.0032e-01, -1.2066e-01, ..., -2.6365e-01,
-9.6645e-01, -7.4116e-01],
[-4.2258e-01, 9.9517e-01, 7.7576e-02, ..., 7.4118e-02,
-1.7612e-01, -5.2427e-01],
...,
[-4.2597e-01, -7.3933e-01, -1.1789e+00, ..., -5.5621e-01,
-5.7689e-01, -1.2206e+00],
[-2.6288e+00, -2.8285e+00, -2.1377e+00, ..., -2.2243e+00,
-2.1800e+00, -2.1603e+00],
[-5.1846e-01, -1.6470e+00, -9.3204e-01, ..., -9.8037e-01,
-9.7542e-01, -1.3924e+00]],
[[-1.8216e-01, 9.3322e-02, 4.4641e-01, ..., 7.8944e-02,
5.0339e-01, 1.5234e-01],
[-3.2876e-01, 6.4042e-02, -4.5140e-01, ..., -5.6641e-03,
1.3067e-01, -6.7678e-02],
[-5.2797e-01, -5.3943e-01, -2.3090e+00, ..., 1.1400e-02,
2.9998e-02, -5.9048e-02],
...,
[ 9.8628e-02, -9.4432e-01, 6.8655e-01, ..., -1.4170e-01,
1.3892e-01, -7.6652e-01],
[-2.7890e-01, -1.6316e+00, -9.0879e-02, ..., -3.5218e-02,
-1.3137e-01, -1.0260e+00],
[ 5.5680e-01, -4.0800e-02, 1.7571e-01, ..., 2.7920e-01,
3.1556e-01, 3.0118e-01]],
[[-6.6676e-01, 4.0772e-01, 5.6377e-01, ..., 4.9730e-01,
4.0744e-01, 6.1466e-01],
[-8.6621e-01, 3.2949e-02, -7.7887e-01, ..., -2.6716e-01,
-1.3983e-01, 6.6430e-01],
[-5.8772e-01, -7.1482e-01, -1.9415e+00, ..., 9.7904e-02,
1.0983e-01, 6.1362e-01],
...,
[-1.1893e+00, -8.4858e-01, -8.2520e-01, ..., -9.7352e-02,
-4.1082e-02, 8.2160e-01],
[-1.8697e+00, -1.6416e+00, -9.0189e-01, ..., -7.7430e-01,
-4.3712e-01, 4.1088e-01],
[-9.8808e-01, -8.7075e-01, -3.3607e-01, ..., -1.6681e-01,
-1.6357e-01, 1.1453e+00]]],
[[[ 2.2956e-02, -3.0405e-01, -1.4340e-01, ..., -1.3660e+00,
-5.6195e-01, -1.5087e+00],
[ 2.6007e-01, -5.3803e-01, -2.1907e+00, ..., -1.3093e+00,
-1.2651e+00, -1.5368e+00],
[ 2.4214e-01, -1.0886e+00, -6.2312e+00, ..., -3.7301e-01,
-9.6928e-01, -1.8580e+00],
...,
[ 1.1732e-01, -5.8723e-01, 2.9715e-02, ..., -7.5684e+00,
-3.0538e+00, -1.2598e+00],
[-4.7660e-01, -6.9148e-01, -8.1602e-02, ..., -5.0173e+00,
-7.3958e-01, -1.7987e+00],
[-6.3918e-01, -1.2754e+00, -1.1283e+00, ..., -2.2092e+00,
-2.3226e+00, -1.8273e+00]],
[[-8.9010e-01, -2.4212e+00, -2.2676e+00, ..., -5.9692e+00,
-5.7923e+00, -1.8294e+00],
[-2.1902e+00, -8.4928e+00, -1.2167e+01, ..., -7.9491e+00,
-8.7550e+00, -3.4637e+00],
[-1.7646e+00, -6.5239e+00, -8.5636e+00, ..., -6.8898e+00,
-6.8241e+00, -2.4956e+00],
...,
[-1.8332e+00, -2.5547e+00, -2.8656e+00, ..., -1.2292e+01,
-1.1200e+01, -6.7569e+00],
[-3.8677e+00, -4.4785e+00, -3.8715e+00, ..., -7.4423e+00,
-1.0385e+01, -4.3928e+00],
[ 3.3567e-01, -2.2126e-01, 4.6578e-01, ..., -3.6472e-01,
-9.9574e-01, -1.8182e-01]],
[[-1.9543e-01, 1.8967e-01, 7.1783e-01, ..., 9.7920e-01,
2.4328e-01, 1.5349e+00],
[ 5.2084e-02, -6.5481e-01, -4.3537e-01, ..., 1.2423e+00,
-3.0491e-02, 6.5553e-02],
[-1.9434e-01, -1.9218e+00, -6.4795e+00, ..., -1.1018e-01,
1.5514e-01, -1.5868e-01],
...,
[-2.9552e-01, -1.4295e-01, 1.7042e+00, ..., -1.1984e+01,
-6.2423e+00, 1.7178e+00],
[-2.0498e-01, -4.9795e-02, 1.4931e-01, ..., -4.7489e+00,
-4.2038e+00, 1.2320e+00],
[-6.2290e-02, -3.8477e-01, 1.1480e-01, ..., 1.1682e-01,
-1.1915e+00, -1.2584e-01]],
...,
[[ 6.4830e-01, -1.0270e-01, 1.3560e-01, ..., 1.2347e+00,
4.5439e-01, -1.3409e+00],
[-4.7863e-01, -1.1956e-01, 2.8326e-01, ..., -9.4035e-01,
-8.6138e-01, -1.4107e+00],
[-3.4705e-01, 2.1803e+00, 3.9889e+00, ..., -2.1206e+00,
-2.3643e-01, -9.5783e-01],
...,
[-4.0047e-01, -1.2649e-01, -1.2272e+00, ..., -5.0674e+00,
-7.6348e+00, -4.7642e+00],
[-2.2173e+00, -1.5732e+00, -1.1725e+00, ..., -7.5332e+00,
-9.9465e+00, -5.3762e+00],
[-4.4322e-01, -7.0893e-01, -3.1948e-01, ..., -4.1511e+00,
-3.2176e+00, -1.4310e+00]],
[[ 1.8151e-01, 2.5904e-01, 5.8186e-01, ..., -1.7452e+00,
2.5257e-01, -2.8815e-01],
[ 1.7143e-01, 1.0100e+00, 1.0102e+00, ..., -5.4899e-01,
-4.0377e-02, -8.0465e-01],
[-1.0998e-01, 1.5532e-01, -7.5915e-01, ..., 7.1973e-01,
4.8477e-02, -3.8626e-01],
...,
[-1.0319e-01, -1.7129e-02, 8.3489e-01, ..., -5.1942e+00,
-2.6325e+00, -1.0854e+00],
[-4.4564e-01, -8.0532e-01, -5.7871e-02, ..., -1.0509e+00,
1.5309e+00, -7.5040e-01],
[ 5.4588e-01, 3.6615e-01, 8.0266e-02, ..., 1.2366e+00,
1.0143e+00, -1.3288e-01]],
[[-8.1171e-01, 2.3904e-01, 3.0745e-01, ..., 5.4270e-01,
-7.4438e-02, 5.2784e-01],
[-1.1230e+00, -1.4928e-01, -1.3662e+00, ..., -4.9894e-02,
-3.3441e-01, 9.3149e-01],
[-9.2361e-01, -1.4729e+00, -4.5061e+00, ..., 4.0900e-02,
-5.3931e-01, 6.4252e-01],
...,
[-9.2477e-01, 1.5199e-01, -1.0577e+00, ..., -9.0321e+00,
2.3493e+00, 2.9572e+00],
[-1.6083e+00, -3.7986e-01, -3.0837e-01, ..., -7.6662e+00,
9.3502e-01, 2.6364e+00],
[-7.2170e-01, -1.3962e-01, -2.3362e-01, ..., -8.3014e-01,
1.6990e+00, 2.2615e+00]]]], device='cuda:0'),
tensor([[[[-1.3089e-01, -5.0116e-01, -1.9822e-01, ..., -9.7813e-02,
2.7733e-01, -8.2664e-02],
[ 2.7618e-01, -1.1871e+00, -7.1195e-01, ..., -4.7792e-01,
-2.6020e-01, -1.8594e-01],
[ 1.4094e-01, -8.8191e-01, -3.7532e-01, ..., -2.6986e-01,
-2.1429e-01, -5.4495e-01],
...,
[-1.6485e-01, -1.4651e+00, -1.0916e+00, ..., -1.0109e+00,
-9.6421e-01, -2.0426e+00],
[-8.5295e-01, -1.5123e+00, -7.9282e-01, ..., -5.4581e-01,
-4.1873e-01, -1.7895e+00],
[-1.0522e+00, -1.8697e+00, -1.4961e+00, ..., -1.5951e+00,
-1.7147e+00, -1.0724e+00]],
[[-1.4398e+00, -3.0160e+00, -2.4792e+00, ..., -2.8767e+00,
-3.3730e+00, -1.3239e+00],
[-3.1371e+00, -4.9285e+00, -3.5668e+00, ..., -3.9262e+00,
-4.7182e+00, -2.0649e+00],
[-2.1062e+00, -3.0551e+00, -2.3045e+00, ..., -2.5040e+00,
-2.7138e+00, -1.2602e+00],
...,
[-3.7305e+00, -5.3982e+00, -5.3409e+00, ..., -5.5523e+00,
-6.1884e+00, -2.9309e+00],
[-5.9054e+00, -6.6001e+00, -6.0192e+00, ..., -6.5582e+00,
-7.8796e+00, -2.9055e+00],
[ 2.4254e-02, -5.9791e-01, 3.1457e-01, ..., 4.2524e-01,
6.9130e-01, 1.4519e-01]],
[[-4.5173e-01, 1.9165e-01, 8.3046e-01, ..., 9.4904e-01,
5.6633e-01, 1.3546e+00],
[-3.2157e-01, -2.6673e-02, 5.8229e-01, ..., 5.7346e-01,
4.4508e-01, 3.0929e-01],
[-7.7468e-01, 6.5827e-02, 2.8636e-01, ..., 1.7303e-01,
3.4305e-01, 4.1441e-01],
...,
[-1.3295e+00, 2.6051e-02, 4.4622e-01, ..., 1.3138e-02,
7.8813e-03, 2.8898e-01],
[-1.2525e+00, 1.3686e-01, -2.4301e-02, ..., -2.7144e-02,
-2.3599e-02, 3.3733e-01],
[-1.1096e+00, -3.5530e-02, 2.1718e-02, ..., -1.7005e-02,
-1.7140e-01, -2.9742e-01]],
...,
[[ 9.2868e-01, 2.5323e-01, 2.2220e-01, ..., 1.3332e-01,
1.4159e-01, -9.1042e-01],
[-1.1353e+00, -9.1222e-01, -7.3828e-01, ..., -9.2286e-01,
-1.4972e+00, -1.1174e+00],
[-5.6772e-01, -3.0123e-01, -2.4417e-01, ..., -3.2734e-01,
-5.0733e-01, -6.0869e-01],
...,
[-4.5685e-01, -8.1511e-01, -7.5623e-01, ..., -3.6105e-01,
-5.9817e-01, -1.2043e+00],
[-2.7248e+00, -2.8184e+00, -1.9053e+00, ..., -1.9961e+00,
-2.2287e+00, -2.1508e+00],
[-5.4210e-01, -1.6601e+00, -1.0618e+00, ..., -9.0012e-01,
-9.5893e-01, -1.4390e+00]],
[[ 5.0334e-02, 3.0402e-01, 7.3317e-01, ..., 7.1619e-01,
9.6376e-01, 3.6108e-01],
[ 2.2611e-02, -4.4395e-01, 1.4472e-01, ..., 4.2824e-01,
6.0661e-01, -1.7675e-01],
[-2.6518e-01, -6.0667e-01, 1.3753e-01, ..., 2.6790e-01,
3.2431e-01, 3.2955e-03],
...,
[ 8.9945e-02, -8.8931e-01, 2.5671e-01, ..., 2.5583e-01,
1.2082e-01, -7.8845e-01],
[-2.9859e-01, -1.6814e+00, -1.5490e-01, ..., 3.0915e-02,
-9.5576e-02, -1.0684e+00],
[ 5.2515e-01, -9.3217e-02, 1.7473e-01, ..., 2.7035e-01,
2.8712e-01, 2.8298e-01]],
[[-9.7897e-01, -2.7254e-01, 1.3405e-01, ..., 1.7537e-01,
5.9147e-02, 7.2583e-01],
[-1.1220e+00, -2.5964e-01, -3.0257e-01, ..., -1.5962e-01,
-9.5138e-02, 1.1597e+00],
[-7.2011e-01, -2.1841e-01, -2.1972e-01, ..., -1.1443e-01,
-1.1399e-01, 6.7213e-01],
...,
[-1.1270e+00, -8.6624e-01, -6.3320e-01, ..., -3.6254e-01,
-4.8012e-04, 8.3703e-01],
[-1.7839e+00, -1.6442e+00, -8.5718e-01, ..., -8.1403e-01,
-4.3258e-01, 4.0028e-01],
[-9.1714e-01, -8.7915e-01, -2.1892e-01, ..., -3.8355e-01,
-1.8721e-01, 1.1283e+00]]],
[[[-7.3213e-01, -1.4455e+00, -1.4607e+00, ..., -5.0816e-01,
-9.6682e-01, -1.7075e+00],
[-3.6799e-01, -2.4544e+00, -2.2855e+00, ..., 1.0274e-01,
-1.7259e+00, -1.9629e+00],
[-1.5171e-01, -1.9337e+00, -1.7498e+00, ..., 2.1305e-01,
-1.1222e+00, -2.1396e+00],
...,
[-3.5739e-02, -7.2985e-01, -6.6272e-01, ..., 6.6141e-01,
-2.8227e-01, -4.6891e-01],
[-6.7686e-01, -6.9825e-01, -1.8276e-01, ..., 3.5867e-01,
2.1317e-01, -2.0364e-01],
[-8.7501e-01, -1.5604e+00, -1.2374e+00, ..., -1.4120e+00,
-1.0835e+00, -4.5084e-01]],
[[-2.7104e+00, -5.1633e+00, -4.6781e+00, ..., -7.5864e+00,
-6.7991e+00, -2.4126e+00],
[-5.4306e+00, -8.2665e+00, -7.6317e+00, ..., -9.5112e+00,
-1.0186e+01, -4.4023e+00],
[-4.1165e+00, -5.9460e+00, -5.8213e+00, ..., -1.2909e+01,
-8.0441e+00, -3.2942e+00],
...,
[-1.8820e+00, -2.6353e+00, -3.5080e+00, ..., -4.4591e+00,
-2.5322e+00, -1.0375e+00],
[-3.2445e+00, -3.6682e+00, -4.0282e+00, ..., -3.1782e+00,
-3.2536e+00, -9.6470e-01],
[ 2.3145e-01, -5.2147e-01, 4.6926e-01, ..., -4.4245e-01,
4.5586e-01, 2.2998e-01]],
[[-6.6204e-01, 4.6632e-01, 8.4695e-01, ..., 1.5219e+00,
8.7989e-01, 1.9218e+00],
[-9.8462e-01, 7.8405e-02, 4.1565e-01, ..., -5.5841e-01,
1.6606e-01, 1.1140e-01],
[-1.8031e+00, -1.9810e-01, -4.1117e-02, ..., -1.3234e+00,
1.7662e-01, -1.0183e-01],
...,
[-7.9431e-01, 1.5109e-02, 1.0537e-01, ..., 6.8061e-01,
2.7729e-01, 4.1896e-01],
[-5.3392e-01, -9.1747e-02, 1.8834e-01, ..., 4.6898e-01,
2.2149e-01, 5.6748e-01],
[-2.1160e-01, -2.9162e-01, 2.0078e-01, ..., 2.9757e-01,
1.1317e-01, 3.5272e-01]],
...,
[[ 1.2747e+00, -3.5155e-01, -1.2070e-03, ..., -1.7778e+00,
-2.0055e-01, -2.1192e+00],
[-7.5410e-01, -9.5761e-01, -5.2421e-01, ..., -4.6292e+00,
-2.1424e+00, -2.0876e+00],
[-7.3978e-01, -9.1109e-01, -2.9145e-01, ..., -5.8154e+00,
-1.5640e+00, -1.4254e+00],
...,
[-7.7877e-01, -5.3040e-01, 1.6562e-01, ..., -2.9128e+00,
-3.7966e-01, -5.5811e-01],
[-2.6115e+00, -2.0186e+00, -1.4858e+00, ..., -3.4796e+00,
-1.2969e+00, -1.0506e+00],
[-7.0720e-01, -1.0827e+00, 1.2451e-01, ..., -1.8344e+00,
-3.8933e-01, -3.9017e-01]],
[[ 5.1166e-02, 2.8660e-02, 5.3366e-01, ..., 7.9100e-01,
6.7224e-01, -5.3797e-03],
[ 2.3121e-01, -8.0228e-01, -9.3582e-03, ..., 3.3364e+00,
4.1233e-01, -8.6810e-01],
[ 1.5867e-01, -6.3573e-01, 2.9819e-01, ..., 1.3645e+00,
2.2937e-01, -4.0271e-01],
...,
[-3.5292e-01, -5.1320e-01, -5.1457e-01, ..., 5.3885e-01,
1.4653e-01, 5.7133e-02],
[-7.0966e-01, -1.0617e+00, -5.9825e-01, ..., 2.7809e-01,
-4.2609e-02, -1.3558e-01],
[ 3.0229e-01, -7.9717e-02, -7.6454e-01, ..., 2.6675e-02,
-5.8032e-02, 1.9212e-01]],
[[-1.6023e+00, -8.6309e-01, -4.8354e-01, ..., 4.8856e-01,
1.0904e-01, 8.6607e-01],
[-2.2130e+00, -8.4794e-01, -6.3502e-01, ..., -3.3733e-01,
3.1214e-01, 1.6392e+00],
[-1.5643e+00, -6.1578e-01, -2.7719e-01, ..., -1.3938e+00,
2.1637e-01, 1.3893e+00],
...,
[-7.1434e-01, -2.7096e-01, -1.0669e+00, ..., 1.0336e+00,
2.7727e-01, 6.5867e-01],
[-1.3251e+00, -8.1927e-01, -7.6839e-01, ..., 3.4171e-01,
-2.0031e-01, 2.2106e-01],
[-4.4308e-01, -2.1268e-01, -4.5352e-01, ..., 5.2632e-01,
-1.7087e-01, 3.7867e-01]]],
[[[-6.1941e-01, -1.4341e+00, -1.4255e+00, ..., -1.2470e+00,
-9.1009e-01, -1.7538e+00],
[-2.4308e-01, -2.5780e+00, -2.2909e+00, ..., -1.7142e+00,
-1.8676e+00, -1.9642e+00],
[-1.3873e-01, -2.0349e+00, -1.7093e+00, ..., -1.2661e+00,
-1.4058e+00, -2.1107e+00],
...,
[ 7.4766e-02, -5.4189e-01, -1.7179e-01, ..., -1.7578e-01,
-2.0539e-02, -3.8918e-01],
[-3.8393e-01, -4.6917e-01, 8.4706e-02, ..., 1.6051e-01,
3.0528e-01, -1.4205e-01],
[-6.4506e-01, -1.3432e+00, -1.0838e+00, ..., -1.1004e+00,
-1.1684e+00, -4.9870e-01]],
[[-2.5381e+00, -4.8777e+00, -4.5125e+00, ..., -4.8807e+00,
-6.0562e+00, -2.4996e+00],
[-5.4853e+00, -8.2446e+00, -7.3747e+00, ..., -7.6608e+00,
-9.0944e+00, -4.3453e+00],
[-3.9943e+00, -5.7656e+00, -5.6066e+00, ..., -5.9802e+00,
-6.4203e+00, -3.2244e+00],
...,
[-1.7038e+00, -2.1669e+00, -1.8223e+00, ..., -1.8809e+00,
-2.2526e+00, -9.2528e-01],
[-2.7514e+00, -2.9076e+00, -1.8437e+00, ..., -2.1994e+00,
-3.2884e+00, -9.6271e-01],
[ 3.2221e-01, -3.0902e-01, 3.2481e-01, ..., 2.7704e-01,
4.2602e-01, 1.9848e-01]],
[[-7.0507e-01, 2.6382e-01, 8.6064e-01, ..., 7.7838e-01,
6.3368e-01, 1.8647e+00],
[-9.2568e-01, -2.0709e-01, 4.5135e-01, ..., 5.3640e-01,
1.7390e-02, 1.3137e-01],
[-1.6924e+00, -3.7139e-01, 4.0598e-03, ..., 6.2207e-02,
-5.8473e-03, -9.4659e-02],
...,
[-6.4873e-01, 1.1011e-01, 2.5437e-01, ..., 1.5010e-01,
2.8286e-01, 4.4582e-01],
[-4.5434e-01, 5.9301e-02, 2.5050e-01, ..., 1.6690e-01,
1.9364e-01, 6.0179e-01],
[-1.5326e-01, -2.5368e-01, 1.5209e-01, ..., -1.2099e-02,
7.8419e-02, 3.6806e-01]],
...,
[[ 1.0655e+00, -4.5177e-01, 2.0896e-02, ..., 2.3518e-01,
-4.8744e-02, -2.0751e+00],
[-9.6024e-01, -8.8419e-01, -5.1025e-01, ..., -4.9711e-01,
-1.4096e+00, -2.0460e+00],
[-8.0780e-01, -7.7721e-01, -2.8482e-01, ..., -3.6273e-01,
-5.8604e-01, -1.3717e+00],
...,
[-6.5795e-01, -5.2475e-01, -3.4994e-01, ..., -2.9317e-01,
-4.8787e-01, -5.4273e-01],
[-2.2998e+00, -1.7763e+00, -1.2287e+00, ..., -1.3212e+00,
-1.3644e+00, -1.0818e+00],
[-6.1201e-01, -8.5349e-01, -5.2510e-01, ..., -2.4852e-01,
-3.5474e-01, -4.0069e-01]],
[[-9.6405e-02, -6.2021e-02, 5.0303e-01, ..., 4.8263e-01,
8.9729e-01, -6.0085e-02],
[-3.0289e-02, -8.3434e-01, -6.7764e-02, ..., 4.5962e-01,
3.8063e-01, -8.4446e-01],
[-9.4547e-02, -7.5566e-01, 2.7537e-01, ..., 6.2966e-01,
5.5366e-01, -4.0977e-01],
...,
[-2.2870e-01, -4.4986e-01, 2.5580e-01, ..., 2.7784e-01,
3.4700e-01, 5.5440e-02],
[-5.2826e-01, -9.1585e-01, 8.6321e-02, ..., 2.6272e-02,
-3.5438e-03, -1.5658e-01],
[ 3.0574e-01, -5.0267e-02, -1.8690e-01, ..., -1.6491e-01,
-1.1308e-01, 1.6551e-01]],
[[-1.6647e+00, -8.2437e-01, -5.1994e-01, ..., -4.6565e-01,
-2.8967e-01, 8.3999e-01],
[-2.1393e+00, -7.7489e-01, -6.8027e-01, ..., -7.9397e-01,
-1.6969e-01, 1.6425e+00],
[-1.4995e+00, -6.6553e-01, -3.1131e-01, ..., -4.1408e-01,
-1.1453e-01, 1.4051e+00],
...,
[-6.5211e-01, -2.1357e-01, -1.1431e-01, ..., -4.8759e-02,
5.1920e-03, 6.0302e-01],
[-1.1500e+00, -5.5821e-01, -3.6136e-01, ..., -3.5322e-01,
-1.8863e-01, 1.4202e-01],
[-3.6423e-01, -8.7558e-03, -1.5649e-01, ..., -2.5578e-01,
-2.1451e-01, 3.1818e-01]]],
[[[-7.2909e-01, -1.4734e+00, -1.4798e+00, ..., -1.2460e+00,
-8.3021e-01, -1.6861e+00],
[-3.4310e-01, -2.4877e+00, -2.3391e+00, ..., -1.7744e+00,
-1.9252e+00, -1.9630e+00],
[-1.2736e-01, -1.9435e+00, -1.3703e+00, ..., -1.4547e+00,
-1.4416e+00, -2.1249e+00],
...,
[ 1.0050e-01, -6.2630e-01, -2.4417e-01, ..., -3.9752e-01,
-1.8665e-01, -4.9100e-01],
[-2.9050e-01, -4.8253e-01, -3.3481e-02, ..., 5.1892e-02,
2.4382e-01, -2.0939e-01],
[-5.7156e-01, -1.3393e+00, -1.1218e+00, ..., -1.0849e+00,
-1.1934e+00, -5.2342e-01]],
[[-2.6931e+00, -5.1596e+00, -4.6133e+00, ..., -4.6986e+00,
-5.9924e+00, -2.3677e+00],
[-5.5042e+00, -8.3170e+00, -7.5749e+00, ..., -7.8137e+00,
-9.3855e+00, -4.3906e+00],
[-4.1615e+00, -5.9999e+00, -8.2925e+00, ..., -6.6719e+00,
-6.5860e+00, -3.2578e+00],
...,
[-1.8064e+00, -2.3021e+00, -1.9417e+00, ..., -2.1688e+00,
-2.6089e+00, -1.0161e+00],
[-2.9085e+00, -3.1593e+00, -2.1287e+00, ..., -2.4569e+00,
-3.5621e+00, -1.1013e+00],
[ 3.5473e-01, -2.0235e-01, 3.4645e-01, ..., 3.4890e-01,
4.7937e-01, 2.0175e-01]],
[[-6.8159e-01, 4.7263e-01, 8.7265e-01, ..., 8.3091e-01,
6.1246e-01, 1.9171e+00],
[-9.6549e-01, 6.9312e-02, 3.8192e-01, ..., 5.4949e-01,
-1.7227e-02, 1.2712e-01],
[-1.7569e+00, -1.8020e-01, -2.4767e-01, ..., -1.1494e-03,
1.1839e-02, -6.7410e-02],
...,
[-6.9610e-01, 1.1982e-01, 3.0560e-01, ..., 1.2543e-01,
2.8609e-01, 4.2222e-01],
[-4.5373e-01, 7.6568e-02, 2.0838e-01, ..., 1.3482e-01,
2.1172e-01, 5.4673e-01],
[-1.5310e-01, -2.3688e-01, 1.4509e-01, ..., 6.6626e-03,
1.1695e-01, 3.3411e-01]],
...,
[[ 1.2675e+00, -3.5451e-01, -6.5559e-02, ..., 2.5059e-01,
-8.1075e-02, -2.1589e+00],
[-7.4840e-01, -9.4366e-01, -5.6010e-01, ..., -5.8656e-01,
-1.5303e+00, -2.1214e+00],
[-6.7959e-01, -9.0585e-01, -4.0446e-01, ..., -3.8556e-01,
-6.1573e-01, -1.4216e+00],
...,
[-6.4753e-01, -5.1684e-01, -3.1107e-01, ..., -7.5831e-01,
-5.4387e-01, -5.7077e-01],
[-2.2000e+00, -1.7778e+00, -1.2970e+00, ..., -1.3089e+00,
-1.3938e+00, -1.1135e+00],
[-5.8920e-01, -7.7081e-01, -4.3635e-01, ..., -2.6746e-01,
-3.6178e-01, -4.0295e-01]],
[[ 2.7401e-02, -4.0713e-03, 4.8268e-01, ..., 5.2581e-01,
9.5974e-01, 3.2853e-02],
[ 2.4262e-01, -8.0982e-01, -3.9037e-02, ..., 4.6905e-01,
3.6927e-01, -8.5622e-01],
[ 2.1280e-01, -6.2589e-01, 5.6819e-01, ..., -4.5072e-03,
5.4331e-01, -3.5691e-01],
...,
[-2.0014e-01, -4.7925e-01, 2.5696e-01, ..., 1.6110e-01,
2.0865e-01, 4.0957e-02],
[-4.8392e-01, -8.9792e-01, 4.0908e-03, ..., -8.7704e-02,
-9.3404e-02, -2.0147e-01],
[ 3.6754e-01, -2.9724e-02, -2.0762e-01, ..., -1.2216e-01,
-9.2904e-02, 1.6809e-01]],
[[-1.6121e+00, -8.6602e-01, -4.5041e-01, ..., -4.8228e-01,
-3.0285e-01, 9.0677e-01],
[-2.2071e+00, -8.0792e-01, -5.8249e-01, ..., -8.3313e-01,
-2.2421e-01, 1.6624e+00],
[-1.5595e+00, -5.8238e-01, -2.2733e-01, ..., -3.7902e-01,
-1.8121e-01, 1.4035e+00],
...,
[-6.8662e-01, -2.4895e-01, -1.6269e-01, ..., 2.7699e-01,
1.4558e-01, 6.3042e-01],
[-1.0729e+00, -5.0222e-01, -3.2139e-01, ..., -2.3895e-01,
-1.7397e-01, 2.1058e-01],
[-3.5190e-01, -5.7676e-02, -1.6171e-01, ..., -2.6147e-01,
-2.4460e-01, 3.6795e-01]]],
[[[ 4.9135e-02, -2.2745e-01, -6.7465e-02, ..., -1.1637e+00,
-6.5168e-01, -9.6904e-01],
[ 3.2922e-01, -9.4214e-01, -7.6807e-01, ..., -1.6260e+00,
-1.2865e+00, -1.2497e+00],
[ 2.2665e-01, -6.3389e-01, -5.2292e-01, ..., -2.2747e+00,
-1.6941e+00, -1.9777e+00],
...,
[-2.4796e-02, -6.0200e-01, -3.8056e-01, ..., -1.4719e+00,
-1.4203e+00, -1.6683e+00],
[-6.2916e-01, -6.9743e-01, -1.4781e-02, ..., -7.3718e-01,
-5.6744e-01, -1.4205e+00],
[-7.3486e-01, -1.3708e+00, -1.2072e+00, ..., -1.2799e+00,
-1.4845e+00, -9.0281e-01]],
[[-7.3150e-01, -2.3455e+00, -2.1000e+00, ..., -4.1131e+00,
-4.8322e+00, -1.8497e+00],
[-1.9908e+00, -3.8900e+00, -3.0996e+00, ..., -6.1924e+00,
-6.8880e+00, -2.7596e+00],
[-1.5602e+00, -2.5881e+00, -2.5032e+00, ..., -8.8906e+00,
-8.8733e+00, -3.4008e+00],
...,
[-1.7669e+00, -2.5006e+00, -2.4485e+00, ..., -5.6213e+00,
-5.6456e+00, -2.3289e+00],
[-3.8010e+00, -4.4554e+00, -3.7606e+00, ..., -6.2923e+00,
-7.2153e+00, -2.7802e+00],
[ 2.7850e-01, -3.2928e-01, 3.5693e-01, ..., 5.2758e-01,
6.4367e-01, 5.0733e-02]],
[[-2.2729e-01, 1.0185e-01, 7.3047e-01, ..., 9.9867e-01,
7.0553e-01, 1.5230e+00],
[ 3.4648e-02, -4.4866e-01, 4.3988e-01, ..., 3.2899e-01,
2.8081e-01, 2.6934e-01],
[-1.3337e-01, -2.1931e-01, 9.0449e-03, ..., 4.1765e-01,
6.1703e-01, 4.7701e-01],
...,
[-3.0242e-01, -2.5161e-01, 8.2443e-02, ..., -4.0967e-01,
2.2431e-01, 2.7823e-02],
[-2.7713e-01, -2.0199e-01, 1.6756e-01, ..., -3.2169e-02,
9.4161e-02, 1.5845e-01],
[-1.5153e-01, -5.2648e-01, 7.9154e-02, ..., -3.4691e-01,
-8.0915e-02, -2.2178e-01]],
...,
[[ 5.9561e-01, -2.4172e-01, -2.3691e-02, ..., -8.7488e-02,
-2.9502e-01, -1.4542e+00],
[-5.0643e-01, -4.3081e-01, -4.1026e-01, ..., -7.7888e-01,
-1.5330e+00, -1.6628e+00],
[-3.7862e-01, -1.4132e-01, -2.7653e-01, ..., -2.4597e-01,
-5.3868e-01, -1.0846e+00],
...,
[-3.9643e-01, -1.6767e-01, -2.0821e-01, ..., -1.4208e+00,
-9.3392e-01, -1.1762e+00],
[-2.3248e+00, -1.7067e+00, -1.2521e+00, ..., -1.8767e+00,
-1.9078e+00, -1.7468e+00],
[-4.8763e-01, -8.3145e-01, -4.2898e-01, ..., -7.3767e-01,
-7.7542e-01, -8.7473e-01]],
[[ 1.7920e-01, 2.5627e-01, 6.3653e-01, ..., 4.3275e-01,
7.5194e-01, -4.4830e-02],
[ 1.3293e-01, 2.9501e-01, 5.5294e-01, ..., 2.4692e-01,
3.9867e-01, -2.7681e-01],
[-1.3650e-01, 7.4563e-02, 3.9284e-01, ..., -3.8082e-01,
-2.5237e-01, -7.9114e-01],
...,
[-1.9750e-01, 1.5385e-02, 3.0371e-01, ..., -4.7234e-02,
-1.3045e-01, -3.2663e-01],
[-5.4389e-01, -6.6592e-01, -4.3085e-02, ..., -8.2972e-02,
-1.4987e-01, -5.9548e-01],
[ 4.8620e-01, 4.0110e-01, 4.4748e-02, ..., 2.0196e-01,
1.2561e-01, 3.1610e-01]],
[[-8.3742e-01, 2.5965e-01, 3.8653e-01, ..., 1.1157e-01,
2.0874e-01, 8.2999e-01],
[-1.1217e+00, 1.4967e-01, -3.7534e-01, ..., -4.8753e-02,
1.1685e-01, 1.3439e+00],
[-9.5890e-01, 2.4039e-01, -1.7070e-01, ..., -3.1377e-02,
1.6836e-01, 1.1172e+00],
...,
[-9.4201e-01, 1.6201e-01, -1.0012e-01, ..., 6.9721e-01,
5.8883e-01, 1.2415e+00],
[-1.6904e+00, -3.4251e-01, -3.3632e-01, ..., -1.5345e-01,
1.5677e-01, 8.0795e-01],
[-7.4463e-01, -7.5752e-02, -3.2992e-01, ..., 2.3634e-01,
1.3823e-01, 1.0655e+00]]]], device='cuda:0'),
tensor([[[[-1.0845e+00, -5.1274e+00, -5.7201e+00, ..., -4.8383e-01,
-6.1012e-02, -8.0936e-01],
[-6.3378e-01, -5.3146e+00, -3.5132e+00, ..., -1.0795e+00,
-8.8471e-01, -1.2189e+00],
[-2.6978e-02, -3.8530e+00, -5.5652e+00, ..., -9.4001e-01,
-8.8392e-01, -1.8035e+00],
...,
[ 1.1339e-01, -6.0489e-01, -2.4202e-01, ..., -1.5036e-01,
-9.6497e-02, -4.9331e-01],
[-3.1333e-01, -4.9972e-01, 1.3104e-02, ..., 1.2453e-01,
2.5710e-01, -2.2271e-01],
[-5.9204e-01, -1.2868e+00, -1.0486e+00, ..., -1.0343e+00,
-1.1634e+00, -5.0946e-01]],
[[-7.4928e+00, -1.3156e+01, -1.2171e+01, ..., -4.1069e+00,
-4.9910e+00, -1.9944e+00],
[-6.1160e+00, -8.9186e+00, -1.2669e+01, ..., -5.6746e+00,
-6.9925e+00, -3.0206e+00],
[-6.5531e+00, -1.1266e+01, -1.5965e+01, ..., -5.3211e+00,
-5.9729e+00, -2.8906e+00],
...,
[-1.8210e+00, -2.3725e+00, -1.9567e+00, ..., -2.0664e+00,
-2.4090e+00, -9.8355e-01],
[-2.8459e+00, -3.0614e+00, -2.0943e+00, ..., -2.3596e+00,
-3.4961e+00, -1.0799e+00],
[ 3.4473e-01, -2.3680e-01, 3.7029e-01, ..., 3.3876e-01,
4.7644e-01, 1.9936e-01]],
[[-4.0538e+00, -7.1794e+00, -9.8527e+00, ..., 9.2139e-01,
5.6157e-01, 1.5202e+00],
[-1.8446e+00, -7.1100e+00, -1.3599e+01, ..., 5.4430e-01,
2.6977e-01, 4.2770e-01],
[-1.1223e+00, -6.6509e+00, -1.3463e+01, ..., 1.9286e-01,
2.1218e-01, 3.6464e-01],
...,
[-6.8769e-01, 9.8659e-02, 2.3335e-01, ..., 1.7360e-01,
2.8006e-01, 4.1403e-01],
[-4.9694e-01, 9.6439e-02, 2.0745e-01, ..., 1.9666e-01,
2.3154e-01, 5.6081e-01],
[-1.3367e-01, -2.1748e-01, 1.3788e-01, ..., 1.1838e-02,
1.0584e-01, 3.3895e-01]],
...,
[[ 1.4549e+00, 5.5025e-01, -3.1679e+00, ..., 3.2516e-01,
2.1705e-01, -1.0485e+00],
[ 2.2231e+00, -6.1248e-01, -6.2644e+00, ..., -5.6766e-01,
-1.2546e+00, -1.3694e+00],
[-8.1767e-01, -2.9252e+00, -4.8018e+00, ..., -3.5066e-01,
-5.8470e-01, -1.1473e+00],
...,
[-6.6419e-01, -5.1144e-01, -3.5093e-01, ..., -2.9727e-01,
-5.1268e-01, -5.7862e-01],
[-2.2423e+00, -1.7658e+00, -1.2870e+00, ..., -1.3161e+00,
-1.3830e+00, -1.1053e+00],
[-6.2405e-01, -8.3050e-01, -4.2094e-01, ..., -2.7239e-01,
-3.3742e-01, -3.9323e-01]],
[[-6.8479e-01, -2.5138e+00, -6.1469e+00, ..., 2.8888e-01,
7.3278e-01, -2.6690e-01],
[-7.0911e-01, -3.9524e+00, -4.5906e+00, ..., 2.1115e-01,
3.6046e-01, -7.8664e-01],
[-3.5082e-01, -4.0737e+00, -6.5133e+00, ..., 1.9911e-01,
1.5636e-01, -6.8049e-01],
...,
[-1.9254e-01, -4.9619e-01, 2.2029e-01, ..., 2.5427e-01,
3.1174e-01, 2.2764e-02],
[-5.0853e-01, -9.6793e-01, 4.6946e-02, ..., 3.3484e-02,
-2.3820e-02, -1.9488e-01],
[ 3.4138e-01, -2.6430e-02, -1.4408e-01, ..., -6.2748e-02,
-6.9058e-02, 1.5371e-01]],
[[-1.6806e+00, -5.3479e+00, -9.3932e+00, ..., 3.0687e-01,
3.9130e-01, 9.6319e-01],
[-4.1815e+00, -4.9938e+00, -6.5697e+00, ..., -5.7571e-01,
-2.8893e-01, 9.7525e-01],
[-2.5946e+00, -2.5101e+00, -4.4401e+00, ..., -3.3485e-01,
-6.0811e-02, 7.5827e-01],
...,
[-6.4992e-01, -2.6474e-01, -1.5207e-01, ..., -1.5366e-01,
-1.9851e-02, 6.2785e-01],
[-1.1335e+00, -5.9344e-01, -3.2080e-01, ..., -3.2128e-01,
-1.4811e-01, 1.9866e-01],
[-3.6905e-01, -3.1714e-02, -1.4181e-01, ..., -2.0797e-01,
-2.0424e-01, 3.5487e-01]]],
[[[-3.2977e-01, -8.8602e-01, -4.0066e-01, ..., -5.7809e-01,
-2.9223e-02, -7.1757e-01],
[ 5.2986e-02, -1.6815e+00, -1.1113e+00, ..., -1.0177e+00,
-8.3088e-01, -1.1409e+00],
[ 3.4177e-02, -1.4166e+00, -9.7857e-01, ..., -7.9733e-01,
-8.2707e-01, -1.7400e+00],
...,
[-7.9086e-02, -7.2287e-01, -2.3475e-01, ..., -5.7037e-01,
-1.3885e-01, -4.1680e-01],
[-7.7531e-01, -7.4772e-01, -6.8132e-03, ..., 1.3366e-01,
4.0739e-01, -2.0150e-01],
[-9.6697e-01, -1.5880e+00, -1.2911e+00, ..., -6.8319e-01,
-1.1175e+00, -4.8575e-01]],
[[-1.9498e+00, -3.8708e+00, -3.8674e+00, ..., -4.0041e+00,
-4.9465e+00, -2.0101e+00],
[-3.7221e+00, -5.9249e+00, -5.1887e+00, ..., -5.5141e+00,
-6.8980e+00, -2.9708e+00],
[-3.4585e+00, -5.1819e+00, -4.9754e+00, ..., -5.1898e+00,
-5.8539e+00, -2.7992e+00],
...,
[-1.8748e+00, -2.3983e+00, -1.7375e+00, ..., -3.7438e+00,
-3.2371e+00, -1.0201e+00],
[-3.2308e+00, -3.6915e+00, -2.0920e+00, ..., -2.8643e+00,
-3.5088e+00, -1.0365e+00],
[ 2.1935e-01, -5.1481e-01, 2.6537e-01, ..., -4.9398e-01,
1.6300e-01, 1.7210e-01]],
[[-4.7968e-01, 5.4832e-01, 9.6191e-01, ..., 8.1334e-01,
5.5344e-01, 1.4695e+00],
[-6.4935e-01, 5.1841e-02, 5.5460e-01, ..., 3.4167e-01,
2.2244e-01, 3.8858e-01],
[-1.1467e+00, 1.2124e-01, 1.5131e-01, ..., 1.0187e-01,
1.4984e-01, 3.2781e-01],
...,
[-7.6385e-01, 4.7049e-02, 2.1010e-01, ..., 3.7984e-01,
3.4564e-01, 4.7379e-01],
[-5.5159e-01, 3.4160e-03, 3.6334e-01, ..., 1.6344e-01,
3.1559e-01, 6.0395e-01],
[-2.4085e-01, -4.2088e-01, 2.4402e-02, ..., -5.2543e-01,
1.8530e-01, 3.8740e-01]],
...,
[[ 1.0241e+00, 3.2448e-01, 2.1775e-01, ..., 5.0366e-01,
2.4773e-01, -1.0391e+00],
[-6.0334e-01, -4.9522e-01, -4.6451e-01, ..., -3.9113e-01,
-1.2628e+00, -1.3853e+00],
[-2.3070e-01, -4.5553e-01, -3.7585e-01, ..., -3.2275e-01,
-5.7406e-01, -1.1684e+00],
...,
[-8.2584e-01, -5.1757e-01, -3.9200e-01, ..., -6.9978e-01,
-5.3522e-01, -5.2308e-01],
[-2.6676e+00, -2.1282e+00, -1.3634e+00, ..., -1.7970e+00,
-1.7426e+00, -1.0508e+00],
[-6.9407e-01, -1.0846e+00, -5.0357e-01, ..., -9.6610e-01,
-8.6497e-01, -4.2694e-01]],
[[ 1.9694e-01, 1.2127e-01, 2.3079e-01, ..., 2.2601e-01,
7.2557e-01, -2.0870e-01],
[ 3.8282e-01, -7.2379e-01, -1.5019e-01, ..., 2.0738e-01,
3.8833e-01, -7.7678e-01],
[ 2.0089e-02, -7.0495e-01, 1.3518e-01, ..., 2.8169e-01,
1.4915e-01, -6.8463e-01],
...,
[-4.2947e-01, -5.7702e-01, 1.9894e-01, ..., -2.8507e-01,
-7.2021e-02, -4.1141e-02],
[-7.6011e-01, -1.1106e+00, -1.2671e-02, ..., 6.4659e-01,
2.3915e-01, -1.4435e-01],
[ 2.9906e-01, -2.8250e-02, -1.0100e-01, ..., -4.3112e-01,
-2.6952e-01, 1.6331e-01]],
[[-5.5224e-01, -1.2434e-02, 3.2166e-01, ..., 3.1408e-01,
4.0606e-01, 9.8052e-01],
[-1.1273e+00, -6.6270e-01, -4.5028e-01, ..., -6.3599e-01,
-2.1661e-01, 1.0524e+00],
[-8.7284e-01, -6.7107e-01, -3.3795e-01, ..., -3.5504e-01,
-8.2827e-02, 7.9652e-01],
...,
[-6.6689e-01, -2.9114e-01, -1.1383e-01, ..., -4.3001e-01,
2.9382e-02, 5.9838e-01],
[-1.3693e+00, -8.4555e-01, -4.3121e-01, ..., -8.3198e-01,
-4.9598e-01, 1.1910e-01],
[-5.5058e-01, -2.5967e-01, -2.6723e-01, ..., -1.9892e-01,
-1.0206e-01, 3.8288e-01]]],
[[[-7.6998e-01, -1.5047e+00, -1.4358e+00, ..., -1.2438e+00,
-8.5808e-01, -1.6896e+00],
[-3.7478e-01, -2.4615e+00, -1.7834e+00, ..., -1.7277e+00,
-1.8737e+00, -1.9210e+00],
[-1.5053e-01, -1.9208e+00, -1.3768e+00, ..., -1.2702e+00,
-1.4401e+00, -2.1359e+00],
...,
[ 1.0356e-01, -5.1340e-01, -2.1194e-01, ..., -3.5841e-01,
-3.4975e-01, -5.1943e-01],
[-3.4402e-01, -4.2320e-01, 3.6727e-02, ..., 1.3569e-01,
2.6071e-01, -1.6235e-01],
[-6.3570e-01, -1.2838e+00, -1.0747e+00, ..., -1.0846e+00,
-1.1788e+00, -4.9360e-01]],
[[-2.7206e+00, -5.1719e+00, -4.7978e+00, ..., -4.7986e+00,
-6.0822e+00, -2.4462e+00],
[-5.3091e+00, -8.1236e+00, -9.9272e+00, ..., -7.7090e+00,
-9.2396e+00, -4.3654e+00],
[-4.0839e+00, -5.8311e+00, -7.3173e+00, ..., -6.0388e+00,
-6.5217e+00, -3.2569e+00],
...,
[-1.7572e+00, -2.1284e+00, -1.7908e+00, ..., -2.5298e+00,
-2.7425e+00, -1.0743e+00],
[-2.8104e+00, -2.9610e+00, -1.9702e+00, ..., -2.2395e+00,
-3.3793e+00, -1.0103e+00],
[ 3.5463e-01, -2.2783e-01, 3.1224e-01, ..., 2.9553e-01,
4.3864e-01, 2.0655e-01]],
[[-6.6305e-01, 5.2729e-01, 8.4906e-01, ..., 8.2245e-01,
6.0320e-01, 1.9093e+00],
[-9.9072e-01, 8.6881e-02, 3.5289e-01, ..., 5.5289e-01,
-1.1703e-02, 1.1815e-01],
[-1.7759e+00, -2.0107e-01, -2.7665e-01, ..., 7.8863e-03,
-1.8832e-02, -1.1220e-01],
...,
[-6.1947e-01, 1.1005e-01, 2.7126e-01, ..., 5.3155e-02,
3.7306e-01, 3.2421e-01],
[-4.4949e-01, 8.0863e-02, 1.9243e-01, ..., 1.3171e-01,
2.0472e-01, 5.6766e-01],
[-1.3418e-01, -2.4945e-01, 1.2625e-01, ..., -1.9898e-02,
8.6003e-02, 3.4866e-01]],
...,
[[ 1.2846e+00, -3.4976e-01, -8.9320e-02, ..., 2.3380e-01,
-9.6909e-02, -2.1361e+00],
[-7.1461e-01, -9.3100e-01, -4.0408e-01, ..., -5.3096e-01,
-1.4953e+00, -2.0754e+00],
[-6.9718e-01, -8.7937e-01, 5.0170e-01, ..., -3.3328e-01,
-5.9363e-01, -1.3883e+00],
...,
[-6.2637e-01, -5.2165e-01, -3.1177e-01, ..., -1.1695e+00,
-9.1187e-01, -5.3248e-01],
[-2.2222e+00, -1.7984e+00, -1.3289e+00, ..., -1.3213e+00,
-1.3585e+00, -1.0994e+00],
[-5.8340e-01, -8.1536e-01, -4.3096e-01, ..., -2.7080e-01,
-3.5721e-01, -4.0206e-01]],
[[ 4.8551e-02, 2.2181e-02, 4.5149e-01, ..., 5.0236e-01,
9.4268e-01, -4.7630e-02],
[ 2.7842e-01, -8.0505e-01, 4.5297e-01, ..., 4.7249e-01,
4.1940e-01, -8.4652e-01],
[ 1.7500e-01, -5.9017e-01, 5.1937e-01, ..., 6.1362e-01,
5.5085e-01, -3.9548e-01],
...,
[-1.8231e-01, -4.5235e-01, 2.3697e-01, ..., 1.7968e-01,
1.0987e-01, 1.1961e-02],
[-4.9865e-01, -9.3315e-01, 5.4726e-02, ..., 3.9971e-02,
-3.6382e-02, -1.7181e-01],
[ 3.5440e-01, -5.9276e-03, -1.8103e-01, ..., -1.3270e-01,
-1.0979e-01, 1.6651e-01]],
[[-1.5401e+00, -7.8044e-01, -4.0276e-01, ..., -4.3665e-01,
-2.8570e-01, 8.6796e-01],
[-2.1789e+00, -8.1295e-01, -6.4947e-01, ..., -7.8260e-01,
-1.9792e-01, 1.6443e+00],
[-1.5184e+00, -5.9306e-01, -1.0005e+00, ..., -3.8688e-01,
-1.0161e-01, 1.3882e+00],
...,
[-6.5118e-01, -2.4399e-01, -1.3872e-01, ..., 7.0224e-01,
4.4507e-01, 8.4623e-01],
[-1.1553e+00, -6.0190e-01, -3.1096e-01, ..., -3.0812e-01,
-2.0355e-01, 1.4982e-01],
[-3.8382e-01, -4.8436e-02, -1.4736e-01, ..., -2.3497e-01,
-2.3249e-01, 3.2939e-01]]],
[[[-1.8739e-01, -2.5908e-01, 4.2665e-01, ..., 6.1534e-02,
4.9192e-01, 1.5822e-01],
[-1.0819e-01, -1.5555e+00, -1.8254e-01, ..., -7.3086e-01,
-4.4968e-01, -3.3987e-01],
[-1.2283e-01, -1.3510e+00, 2.4622e-01, ..., -4.9309e-01,
-4.2240e-01, -6.8793e-01],
...,
[-2.2262e-01, -1.7224e+00, -1.2992e+00, ..., -9.6733e-01,
-9.5917e-01, -2.0457e+00],
[-1.1099e+00, -1.7564e+00, -8.0161e-01, ..., -5.1789e-01,
-3.9039e-01, -1.7697e+00],
[-1.3067e+00, -2.1968e+00, -1.6313e+00, ..., -1.4568e+00,
-1.5639e+00, -9.7638e-01]],
[[-1.9928e-01, -1.4369e+00, -2.6616e+00, ..., -5.7431e-01,
-1.3245e+00, -2.4387e-01],
[-2.0401e+00, -4.1303e+00, -3.8166e+00, ..., -2.3819e+00,
-3.3694e+00, -1.1369e+00],
[-1.8519e+00, -3.4191e+00, -3.5105e+00, ..., -2.6630e+00,
-2.9748e+00, -1.2402e+00],
...,
[-3.7396e+00, -5.6163e+00, -5.4333e+00, ..., -5.4922e+00,
-6.2157e+00, -3.0157e+00],
[-6.1367e+00, -7.0031e+00, -6.0739e+00, ..., -6.1844e+00,
-7.5072e+00, -2.7009e+00],
[-5.4349e-02, -9.0938e-01, 2.8976e-01, ..., 3.6636e-01,
6.0962e-01, 1.9773e-01]],
[[-5.3854e-01, -1.9097e-01, 2.5028e-01, ..., 3.4596e-01,
6.8471e-02, 8.9330e-01],
[-3.5839e-01, -4.7158e-01, 9.7995e-01, ..., 3.8309e-01,
2.8692e-01, 5.0555e-01],
[-8.8862e-01, -2.6477e-01, 9.9137e-01, ..., 4.8885e-02,
1.9981e-01, 4.3301e-01],
...,
[-1.4034e+00, -5.0212e-02, -6.3705e-02, ..., 4.3562e-02,
6.7582e-02, 2.8076e-01],
[-1.2160e+00, 1.1171e-01, -6.7537e-02, ..., -4.2466e-02,
2.3903e-02, 3.3576e-01],
[-1.1448e+00, -5.4627e-02, 5.0200e-02, ..., -1.0873e-01,
-6.9285e-02, -3.1405e-01]],
...,
[[ 4.0649e-02, -6.2223e-01, 3.4963e-02, ..., -7.7865e-02,
-2.4761e-01, -7.4572e-01],
[-8.5562e-01, -5.8423e-01, -1.5682e-01, ..., -2.7299e-01,
-9.8896e-01, -7.6144e-01],
[-5.0275e-01, -5.9664e-02, -2.5237e-03, ..., 8.7158e-02,
-2.1737e-01, -5.1108e-01],
...,
[-5.9994e-01, -5.3399e-01, -2.6217e-01, ..., -3.8574e-01,
-6.0586e-01, -1.1844e+00],
[-3.1372e+00, -2.8613e+00, -1.9717e+00, ..., -1.9300e+00,
-2.0456e+00, -2.0536e+00],
[-5.5081e-01, -1.6699e+00, -1.1849e+00, ..., -9.0517e-01,
-9.7732e-01, -1.3569e+00]],
[[-1.8702e-01, 4.5076e-02, 7.9826e-01, ..., 7.4666e-02,
5.0539e-01, 1.5220e-01],
[-4.4511e-01, -2.5585e-01, 4.6703e-01, ..., -6.8319e-02,
8.9594e-02, -8.0103e-02],
[-6.7504e-01, -5.2958e-01, 6.3994e-01, ..., 5.4072e-02,
2.6341e-02, -6.0581e-02],
...,
[-9.4278e-02, -9.6805e-01, 5.9513e-02, ..., 2.1348e-01,
1.4512e-01, -7.2772e-01],
[-5.3350e-01, -1.7938e+00, -1.1828e-01, ..., 5.2480e-02,
-4.5316e-02, -9.3928e-01],
[ 4.1935e-01, -1.5601e-01, 1.5412e-01, ..., 2.9096e-01,
3.1989e-01, 3.2350e-01]],
[[-6.9500e-01, 4.0607e-01, 3.6532e-01, ..., 4.8904e-01,
3.9562e-01, 5.9781e-01],
[-8.4606e-01, 3.0928e-01, -5.8619e-01, ..., -2.8786e-01,
-1.6714e-01, 6.2441e-01],
[-5.0416e-01, 2.9864e-01, -6.2759e-02, ..., 7.6219e-02,
1.1945e-01, 5.7646e-01],
...,
[-1.1332e+00, -9.2882e-01, -4.3312e-01, ..., -4.0694e-01,
-6.1566e-02, 8.4937e-01],
[-1.9088e+00, -1.8959e+00, -1.0358e+00, ..., -7.7401e-01,
-3.8286e-01, 4.7739e-01],
[-9.5442e-01, -1.0191e+00, -3.1675e-01, ..., -3.2592e-01,
-1.3906e-01, 1.1544e+00]]],
[[[-1.3838e-01, -5.1766e-01, -2.0619e-01, ..., -2.5233e-01,
2.6010e-01, -1.1225e-01],
[ 2.6748e-01, -1.2460e+00, -7.2772e-01, ..., -4.0724e-01,
-2.3276e-01, -2.1050e-01],
[ 1.1051e-01, -9.3413e-01, -3.9385e-01, ..., -2.3597e-01,
-2.0330e-01, -5.3672e-01],
...,
[-1.6375e-01, -1.3355e+00, -1.4345e+00, ..., -1.2859e+00,
-1.0737e+00, -2.0584e+00],
[-8.2987e-01, -1.3072e+00, -8.3673e-01, ..., -9.0713e-01,
-5.0567e-01, -1.8193e+00],
[-9.9332e-01, -1.7446e+00, -1.5445e+00, ..., -1.9250e+00,
-1.6776e+00, -1.0812e+00]],
[[-1.3974e+00, -3.0034e+00, -2.5008e+00, ..., -2.6794e+00,
-3.3895e+00, -1.3364e+00],
[-3.1372e+00, -4.9743e+00, -3.5304e+00, ..., -3.6550e+00,
-4.6124e+00, -1.9385e+00],
[-2.0767e+00, -3.1003e+00, -2.1905e+00, ..., -2.5437e+00,
-2.6665e+00, -1.2119e+00],
...,
[-3.6987e+00, -5.6239e+00, -4.9880e+00, ..., -7.0587e+00,
-6.2157e+00, -2.9635e+00],
[-5.8765e+00, -6.5408e+00, -5.7878e+00, ..., -6.7383e+00,
-7.7890e+00, -2.8100e+00],
[ 4.6300e-02, -5.0818e-01, 4.4051e-01, ..., 4.7320e-01,
7.0105e-01, 1.7792e-01]],
[[-4.4874e-01, 1.9939e-01, 8.5609e-01, ..., 7.9955e-01,
6.1380e-01, 1.3625e+00],
[-3.3264e-01, -1.1143e-01, 5.8864e-01, ..., 4.9437e-01,
4.2501e-01, 3.6249e-01],
[-7.9814e-01, 1.8379e-03, 3.0006e-01, ..., 2.8789e-01,
3.5223e-01, 3.8255e-01],
...,
[-1.3497e+00, 3.1393e-01, -1.9558e-02, ..., -2.3862e-01,
6.9019e-02, 2.7288e-01],
[-1.2415e+00, 3.6475e-01, -3.1344e-01, ..., 9.8444e-02,
1.2964e-02, 3.3235e-01],
[-1.0946e+00, 1.4305e-01, -1.3060e-01, ..., 3.6650e-01,
-8.2217e-02, -3.4357e-01]],
...,
[[ 8.7402e-01, 1.9820e-01, 2.1847e-01, ..., 3.5694e-01,
1.5640e-01, -8.6263e-01],
[-1.1890e+00, -8.7044e-01, -6.8753e-01, ..., -6.9798e-01,
-1.4165e+00, -1.0865e+00],
[-6.0464e-01, -2.8634e-01, -2.7512e-01, ..., -3.4919e-01,
-4.9510e-01, -6.1324e-01],
...,
[-4.3325e-01, -6.3675e-01, -5.2897e-01, ..., -9.7456e-01,
-7.6506e-01, -1.2181e+00],
[-2.6411e+00, -2.7597e+00, -2.1640e+00, ..., -2.6728e+00,
-2.1609e+00, -2.1357e+00],
[-5.0179e-01, -1.6336e+00, -7.6168e-01, ..., -1.4051e+00,
-9.6168e-01, -1.3726e+00]],
[[ 4.7180e-02, 2.9501e-01, 7.3240e-01, ..., 4.7606e-01,
9.4172e-01, 3.1454e-01],
[-3.2954e-02, -4.0885e-01, 1.6203e-01, ..., 4.0556e-01,
5.8403e-01, -1.2286e-01],
[-3.2588e-01, -6.0967e-01, 1.5076e-01, ..., 3.2744e-01,
3.1368e-01, 1.9214e-02],
...,
[ 3.4028e-02, -6.2426e-01, 3.3937e-01, ..., -3.4117e-01,
-5.0235e-03, -7.8119e-01],
[-3.4689e-01, -1.4889e+00, -3.6554e-02, ..., -4.5057e-01,
-1.9564e-01, -1.0321e+00],
[ 5.7576e-01, 6.9631e-02, 1.5353e-01, ..., -1.8479e-01,
2.6579e-01, 2.9261e-01]],
[[-9.9964e-01, -2.4180e-01, 1.5349e-01, ..., 1.5075e-01,
6.3088e-02, 6.9856e-01],
[-1.1356e+00, -2.0993e-01, -3.0187e-01, ..., -3.4634e-01,
-8.5627e-02, 1.1238e+00],
[-7.3210e-01, -1.9590e-01, -2.0821e-01, ..., -1.5109e-01,
-1.0882e-01, 6.9350e-01],
...,
[-1.0997e+00, -1.0313e+00, -3.1068e-01, ..., -4.3906e-02,
1.0869e-01, 7.8792e-01],
[-1.7045e+00, -1.6348e+00, -3.7460e-01, ..., -4.2209e-01,
-3.4743e-01, 3.8847e-01],
[-9.0849e-01, -8.5499e-01, -2.0365e-03, ..., 1.3853e-01,
-1.8063e-01, 1.1137e+00]]]], device='cuda:0'),
tensor([[[[-3.6379e-01, -6.0644e-01, -2.6374e+00, ..., -5.6510e-01,
-8.4074e-02, -8.3006e-01],
[-6.0842e-03, -9.6917e-01, -1.6034e+00, ..., -1.9119e+00,
-1.2224e+00, -1.2385e+00],
[ 6.4714e-03, -7.1553e-01, -9.6607e-01, ..., -1.5429e+00,
-1.0351e+00, -1.7729e+00],
...,
[-4.8525e-02, -8.2332e-01, 1.6674e-01, ..., -4.3945e-01,
-8.7557e-02, -4.9351e-01],
[-6.9380e-01, -7.3068e-01, -7.1481e-01, ..., -1.2979e-01,
2.4783e-01, -1.9685e-01],
[-8.9061e-01, -1.5944e+00, -1.0116e+00, ..., -1.5602e+00,
-1.1074e+00, -4.6528e-01]],
[[-2.0047e+00, -5.0310e+00, -5.1061e+00, ..., -3.9865e+00,
-4.9944e+00, -1.9834e+00],
[-3.7969e+00, -7.3276e+00, -6.8258e+00, ..., -8.1181e+00,
-8.0385e+00, -2.9871e+00],
[-3.5250e+00, -6.2202e+00, -7.2794e+00, ..., -7.1839e+00,
-7.4217e+00, -2.8089e+00],
...,
[-1.9021e+00, -2.5704e+00, -4.4751e+00, ..., -3.1452e+00,
-2.4755e+00, -1.0298e+00],
[-3.2121e+00, -3.7389e+00, -5.6517e+00, ..., -3.2910e+00,
-3.2688e+00, -9.8754e-01],
[ 2.3638e-01, -5.6540e-01, -5.2907e-01, ..., -4.4781e-01,
4.6290e-01, 2.1319e-01]],
[[-4.4686e-01, 6.0902e-01, -6.9528e-01, ..., 8.3530e-01,
5.3857e-01, 1.5153e+00],
[-6.1633e-01, 8.8025e-01, -2.1075e+00, ..., 7.1959e-01,
4.2599e-01, 3.6813e-01],
[-1.1388e+00, 6.2211e-01, -2.0696e+00, ..., 2.2339e-01,
4.3205e-01, 2.6867e-01],
...,
[-8.1256e-01, 3.4130e-02, 3.4228e-01, ..., 4.1051e-01,
2.4811e-01, 4.1708e-01],
[-5.4245e-01, -8.0881e-02, 8.2103e-01, ..., 5.2929e-01,
1.7379e-01, 5.4479e-01],
[-2.2759e-01, -3.0944e-01, 6.2414e-01, ..., 3.3204e-01,
7.2563e-02, 3.3057e-01]],
...,
[[ 1.0618e+00, 1.3428e+00, 5.2763e-01, ..., 4.4402e-01,
2.1738e-01, -1.0416e+00],
[-7.2777e-01, -2.8805e-01, -6.4828e-01, ..., -5.0824e-01,
-1.1127e+00, -1.3780e+00],
[-2.3112e-01, -7.3369e-02, 5.9851e-01, ..., -1.3725e+00,
-9.7350e-01, -1.1839e+00],
...,
[-7.8375e-01, -5.3534e-01, 3.9804e-03, ..., -9.7325e-01,
-5.2764e-01, -5.7798e-01],
[-2.6016e+00, -2.0010e+00, -1.3772e-01, ..., -2.5605e+00,
-1.3450e+00, -1.0722e+00],
[-7.2798e-01, -1.0718e+00, -5.7732e-01, ..., -8.6729e-01,
-3.6841e-01, -3.7937e-01]],
[[ 1.7849e-01, 8.1722e-01, -1.1798e-01, ..., 2.2816e-01,
7.2745e-01, -2.6815e-01],
[ 3.6611e-01, 5.3904e-02, -1.3283e+00, ..., -1.0959e+00,
-3.9327e-01, -7.6693e-01],
[ 3.1085e-02, -1.8524e-01, -6.4073e-01, ..., -1.7527e-01,
-4.1982e-01, -6.7399e-01],
...,
[-3.9580e-01, -6.1847e-01, 1.1268e+00, ..., -3.0724e-02,
3.1819e-01, 2.4646e-02],
[-7.2699e-01, -1.0857e+00, -5.7574e-01, ..., 1.1989e-01,
-6.7449e-03, -1.4328e-01],
[ 3.2531e-01, -7.6121e-02, -5.3420e-01, ..., -1.2255e-01,
-9.1421e-02, 1.7433e-01]],
[[-6.9411e-01, -7.0157e-01, -8.6826e-01, ..., 2.9436e-01,
3.9958e-01, 9.7580e-01],
[-1.1822e+00, -1.0677e+00, 2.1745e-01, ..., -1.1493e+00,
-1.3277e-01, 1.0067e+00],
[-8.7197e-01, -9.9785e-01, -4.4901e-01, ..., -1.4152e+00,
2.8289e-02, 7.8036e-01],
...,
[-7.0649e-01, -2.9356e-01, -7.4253e-01, ..., -2.3497e-01,
-8.5855e-03, 6.3470e-01],
[-1.3176e+00, -8.3201e-01, -1.8134e+00, ..., -3.0387e-01,
-1.7256e-01, 2.0905e-01],
[-4.5998e-01, -2.0333e-01, -2.9150e-01, ..., 3.7000e-01,
-1.5969e-01, 3.6759e-01]]],
[[[-7.0617e-01, -1.4623e+00, -1.4535e+00, ..., -1.2581e+00,
-8.4151e-01, -1.6443e+00],
[-3.2306e-01, -2.4137e+00, -2.2263e+00, ..., -1.7114e+00,
-1.8541e+00, -1.8945e+00],
[-1.2063e-01, -1.8973e+00, -1.6573e+00, ..., -1.3235e+00,
-1.3910e+00, -2.0921e+00],
...,
[-7.2531e-02, -1.0922e+00, -6.4013e-01, ..., -5.6459e-01,
-4.5475e-01, -8.8961e-01],
[-1.4725e-01, -9.8143e-01, -4.3315e-01, ..., -2.6235e-01,
-9.1008e-02, -9.3412e-01],
[-1.3907e-01, -1.0587e+00, -9.5039e-01, ..., -8.9063e-01,
-1.0940e+00, -6.0620e-01]],
[[-2.7102e+00, -5.1250e+00, -4.6397e+00, ..., -4.7807e+00,
-6.0211e+00, -2.4109e+00],
[-5.2713e+00, -8.0966e+00, -7.5102e+00, ..., -7.6990e+00,
-9.2587e+00, -4.3869e+00],
[-4.0084e+00, -5.7742e+00, -6.3048e+00, ..., -6.2775e+00,
-6.4432e+00, -3.1972e+00],
...,
[-2.0011e+00, -3.2087e+00, -3.3041e+00, ..., -2.9049e+00,
-3.3746e+00, -1.4328e+00],
[-4.3826e+00, -5.3553e+00, -5.3252e+00, ..., -5.0601e+00,
-6.2122e+00, -2.3409e+00],
[ 6.6128e-01, -2.9916e-01, 4.0813e-01, ..., 4.8620e-01,
5.4087e-01, -2.9039e-03]],
[[-6.5715e-01, 4.6212e-01, 8.4629e-01, ..., 8.1925e-01,
6.2935e-01, 1.9030e+00],
[-9.4617e-01, 3.6807e-02, 4.0514e-01, ..., 5.1654e-01,
-5.6608e-03, 1.1999e-01],
[-1.7595e+00, -2.3720e-01, -8.9710e-02, ..., 6.0657e-02,
-4.9317e-03, -9.2675e-02],
...,
[-9.2459e-01, -9.5705e-02, 6.3564e-02, ..., 4.7289e-02,
6.0471e-02, 3.7207e-01],
[-9.4348e-01, 5.0522e-02, -1.8288e-01, ..., 3.4614e-03,
1.2642e-01, 3.9410e-01],
[-4.3650e-01, -1.3315e-02, 6.1704e-02, ..., -4.4734e-02,
1.1106e-01, 6.3742e-02]],
...,
[[ 1.2313e+00, -3.5821e-01, -6.1420e-02, ..., 2.1229e-01,
-1.0459e-01, -2.1151e+00],
[-7.2487e-01, -8.6683e-01, -4.6466e-01, ..., -5.3071e-01,
-1.4861e+00, -2.0579e+00],
[-7.4809e-01, -8.4862e-01, -1.8599e-01, ..., -3.5090e-01,
-5.9012e-01, -1.3658e+00],
...,
[-4.1932e-01, -1.0560e-01, 1.7194e-01, ..., 4.5605e-02,
-1.2765e-01, -5.1759e-01],
[-2.0042e+00, -1.3195e+00, -4.1410e-01, ..., -9.5935e-01,
-1.1454e+00, -1.1062e+00],
[-2.9379e-01, -7.4875e-01, 8.2817e-02, ..., -1.3434e-01,
-3.0768e-01, -5.9269e-01]],
[[ 4.3391e-02, -7.6438e-04, 4.2354e-01, ..., 4.6451e-01,
9.3518e-01, -2.8475e-02],
[ 2.5293e-01, -7.6628e-01, -1.4133e-02, ..., 4.4120e-01,
4.0406e-01, -8.4035e-01],
[ 1.5585e-01, -6.2062e-01, 4.3064e-01, ..., 4.0453e-01,
5.5716e-01, -3.5642e-01],
...,
[-3.9655e-01, -5.2620e-01, 1.1273e-01, ..., 1.1895e-01,
5.5719e-02, -9.0588e-02],
[-7.2758e-01, -1.4537e+00, -2.5758e-01, ..., -3.6447e-01,
-3.9013e-01, -6.5504e-01],
[ 5.0602e-01, 1.0320e-01, 1.0396e-01, ..., 2.0245e-01,
1.4657e-01, 3.6477e-01]],
[[-1.5234e+00, -7.7084e-01, -4.0389e-01, ..., -4.2418e-01,
-2.8295e-01, 8.6944e-01],
[-2.1204e+00, -7.9524e-01, -5.4961e-01, ..., -7.5863e-01,
-1.8393e-01, 1.6577e+00],
[-1.4545e+00, -5.7244e-01, -2.7450e-01, ..., -3.4428e-01,
-1.3088e-01, 1.3959e+00],
...,
[-4.3732e-01, 1.5130e-01, 6.2057e-02, ..., 1.6934e-01,
2.1076e-01, 6.9237e-01],
[-9.0679e-01, -2.8812e-01, -4.2523e-01, ..., -5.3550e-02,
4.8031e-02, 4.1521e-01],
[-4.0884e-01, -2.9502e-01, -6.3996e-01, ..., -2.3171e-01,
-2.3789e-01, 5.9295e-01]]],
[[[-4.4273e-01, -3.0428e-01, -3.5494e-01, ..., -9.5509e-01,
-5.4429e-01, -1.4435e+00],
[ 3.8421e-01, -1.2690e+00, -9.0709e-01, ..., -1.1335e+00,
-1.3384e+00, -1.4966e+00],
[-1.0674e-01, -6.9276e-01, -5.9810e-01, ..., -9.0093e-01,
-1.0946e+00, -1.8184e+00],
...,
[-3.0774e-01, -1.2631e+00, -5.8810e-01, ..., -9.0063e-01,
-1.0380e+00, -1.9290e+00],
[-4.0753e-01, -1.0807e+00, -9.0103e-02, ..., -4.5593e-01,
-5.4925e-01, -1.7218e+00],
[-3.8455e-01, -1.4669e+00, -1.0958e+00, ..., -1.1430e+00,
-1.5351e+00, -7.5890e-01]],
[[-1.5457e+00, -3.1717e+00, -2.4958e+00, ..., -3.9288e+00,
-5.4807e+00, -1.8467e+00],
[-3.7292e+00, -5.0011e+00, -4.4566e+00, ..., -5.8340e+00,
-8.0418e+00, -3.4066e+00],
[-4.2729e+00, -3.3180e+00, -4.1055e+00, ..., -4.9710e+00,
-6.1415e+00, -2.5651e+00],
...,
[-1.7220e+00, -2.7210e+00, -1.4729e+00, ..., -5.0901e+00,
-6.3398e+00, -2.5153e+00],
[-3.4647e+00, -4.1795e+00, -2.5974e+00, ..., -6.4892e+00,
-7.9785e+00, -2.5690e+00],
[ 5.9668e-01, -6.6128e-01, 4.4681e-01, ..., 4.3849e-01,
4.7355e-01, 2.3499e-01]],
[[-1.2146e+00, 4.8877e-01, 1.1829e+00, ..., 6.4289e-01,
2.6108e-01, 1.5161e+00],
[-2.6236e+00, -5.6183e-01, 1.6602e-02, ..., 5.7932e-01,
-5.3000e-02, 8.9925e-02],
[-2.6371e+00, -1.3481e+00, -1.9609e-01, ..., 1.2454e-03,
-3.5204e-03, -1.9182e-01],
...,
[-1.3731e+00, 2.7329e-01, 1.6213e-01, ..., -1.5883e-02,
-9.4398e-02, -1.8644e-01],
[-1.3685e+00, 3.1977e-01, 2.3150e-01, ..., -9.2436e-02,
-9.0318e-03, -2.1125e-01],
[-8.4431e-01, 5.0596e-02, -6.3496e-02, ..., 4.0978e-02,
2.3490e-01, -5.2766e-01]],
...,
[[ 2.2387e-01, -7.0172e-01, -2.4566e-01, ..., 4.4880e-01,
3.4630e-01, -1.3208e+00],
[-7.3312e-01, -1.4346e+00, -1.2204e+00, ..., -1.5019e-01,
-7.4806e-01, -1.3702e+00],
[-1.2393e-01, -1.6462e+00, -1.2939e+00, ..., -1.0363e-01,
-8.2631e-02, -9.4804e-01],
...,
[-1.0134e+00, -6.3431e-01, -7.0972e-02, ..., -1.3720e-01,
-1.2256e-01, -9.3140e-01],
[-2.3927e+00, -1.5675e+00, -7.7062e-01, ..., -1.5437e+00,
-1.5953e+00, -1.8327e+00],
[-6.1961e-01, -1.0128e+00, -8.2556e-02, ..., -3.5484e-01,
-5.4414e-01, -1.1945e+00]],
[[-9.9396e-02, 1.5047e+00, 9.5625e-01, ..., -9.6728e-02,
3.7587e-01, -3.6619e-01],
[-1.2302e+00, 1.5809e-01, 1.0602e+00, ..., 2.6116e-01,
1.0107e-01, -8.2848e-01],
[-1.3871e+00, 7.7989e-02, 1.0674e+00, ..., 2.9625e-01,
1.7872e-01, -4.5227e-01],
...,
[-5.6625e-01, -1.0642e+00, -2.0466e-01, ..., 2.4279e-01,
1.2513e-01, -5.3516e-01],
[-1.0086e+00, -1.8555e+00, -2.9997e-01, ..., -1.2651e-01,
-2.9637e-01, -9.7386e-01],
[ 2.3099e-01, -4.9091e-01, -2.2422e-01, ..., 3.4421e-01,
3.1970e-01, 2.7087e-01]],
[[-1.7450e+00, -1.0253e+00, 8.5607e-02, ..., -6.2899e-02,
-8.6925e-02, 5.7690e-01],
[-3.7790e+00, -1.6532e+00, -1.0848e-01, ..., -9.0021e-01,
-4.3024e-01, 8.7065e-01],
[-5.3543e+00, -1.6415e+00, 9.2946e-01, ..., -5.7116e-01,
-5.2571e-01, 6.3306e-01],
...,
[-7.7867e-02, 1.7159e-03, 1.0059e-01, ..., -5.2255e-01,
-4.9060e-01, 6.1647e-01],
[-3.7342e-01, -3.8518e-01, -1.4088e-01, ..., -6.4116e-01,
-6.6797e-01, 3.9461e-01],
[ 8.6656e-02, -3.9575e-01, -8.5805e-02, ..., -4.7998e-01,
-4.3879e-01, 9.9368e-01]]],
[[[-3.3399e-01, -4.8747e-01, -1.2007e-01, ..., -9.3224e-01,
-5.1610e-01, -1.4097e+00],
[-8.9605e-01, -2.1944e+00, -1.2666e+00, ..., -1.1764e+00,
-1.3653e+00, -1.4680e+00],
[-1.5905e+00, -1.2984e+00, 2.8734e-01, ..., -1.0756e+00,
-1.2202e+00, -1.8408e+00],
...,
[-2.8604e-01, -8.7257e-01, -2.2833e-01, ..., -7.8839e-01,
-1.0614e+00, -1.9598e+00],
[-4.9399e-01, -8.8723e-01, 3.6602e-01, ..., -4.7918e-01,
-5.9131e-01, -1.7244e+00],
[-4.5739e-01, -1.5550e+00, -1.2207e+00, ..., -9.6085e-01,
-1.3580e+00, -6.5462e-01]],
[[-7.1332e-01, -1.9656e+00, -8.4575e-01, ..., -3.8441e+00,
-5.3897e+00, -1.8061e+00],
[-4.6645e+00, -6.6346e+00, -3.7349e+00, ..., -5.9290e+00,
-8.1218e+00, -3.4685e+00],
[-4.5938e+00, -5.6123e+00, -3.5963e+00, ..., -5.6607e+00,
-6.5745e+00, -2.7040e+00],
...,
[-1.7049e+00, -2.7900e+00, -1.0996e+00, ..., -4.9399e+00,
-6.3464e+00, -2.6036e+00],
[-3.6670e+00, -4.8389e+00, -3.1799e+00, ..., -6.1477e+00,
-7.6069e+00, -2.4409e+00],
[ 5.7575e-01, -6.8926e-01, 5.0256e-01, ..., 4.0647e-01,
4.5398e-01, 2.4320e-01]],
[[-3.7398e-01, 7.2235e-01, 1.0160e+00, ..., 7.1375e-01,
3.1830e-01, 1.6085e+00],
[-6.4955e-01, 7.1389e-01, 1.0033e+00, ..., 5.8832e-01,
-1.1595e-01, 1.0359e-01],
[-3.1177e+00, -1.2148e+00, -7.5562e-01, ..., 1.5077e-01,
-3.7606e-02, -1.8318e-01],
...,
[-1.4239e+00, 5.0874e-01, -1.7812e+00, ..., 9.6069e-02,
-2.2755e-01, -6.5897e-02],
[-1.3955e+00, 7.5986e-01, 3.1243e-01, ..., -8.8326e-02,
-1.1211e-02, -2.0099e-01],
[-9.0969e-01, 5.9596e-02, -1.1069e-01, ..., 9.8381e-02,
2.2708e-01, -5.8822e-01]],
...,
[[ 3.9441e-01, -1.3028e-01, -1.2150e-01, ..., 4.5009e-01,
3.0934e-01, -1.3869e+00],
[-1.3115e+00, -1.0007e+00, -5.7524e-01, ..., -1.8589e-01,
-8.2146e-01, -1.4692e+00],
[-1.7511e-01, -6.6015e-01, -7.2746e-01, ..., -4.1359e-02,
-3.1279e-02, -9.7617e-01],
...,
[-9.7780e-01, -3.6233e-01, 3.7457e-01, ..., -7.0464e-01,
-5.3647e-01, -1.1037e+00],
[-2.5383e+00, -1.9902e+00, -1.3609e+00, ..., -1.3785e+00,
-1.4415e+00, -1.7556e+00],
[-6.4868e-01, -1.1295e+00, -8.9758e-02, ..., -3.2996e-01,
-4.9578e-01, -1.1560e+00]],
[[ 3.0164e-02, 2.4091e-01, 4.8386e-01, ..., -8.3504e-02,
4.6786e-01, -3.5112e-01],
[-2.8015e-01, -9.6375e-01, -2.5543e-01, ..., 1.8080e-01,
1.4849e-01, -8.1612e-01],
[-1.7045e+00, -1.4772e+00, 1.2443e-01, ..., 7.9641e-02,
-7.2020e-03, -5.1079e-01],
...,
[-5.9665e-01, -1.3243e+00, -6.8061e-01, ..., 2.7958e-02,
-1.5553e-01, -5.9678e-01],
[-1.0345e+00, -1.8544e+00, -2.4930e-01, ..., -8.2964e-02,
-1.8586e-01, -8.5847e-01],
[ 2.3796e-01, -4.3489e-01, -2.3870e-01, ..., 3.8645e-01,
3.6480e-01, 3.1346e-01]],
[[-3.6732e-01, 9.2656e-02, 5.2076e-01, ..., -4.2239e-02,
-4.8319e-02, 6.0610e-01],
[-6.9199e-01, -3.7369e-01, -2.3172e-01, ..., -9.6164e-01,
-3.8801e-01, 9.3731e-01],
[-1.7671e+00, -1.9409e+00, -9.2326e-01, ..., -6.6797e-01,
-4.7838e-01, 6.5572e-01],
...,
[-9.8056e-04, -6.3838e-01, -4.7759e-01, ..., 9.9898e-03,
-1.2264e-01, 7.8094e-01],
[-3.8953e-01, -3.9470e-01, 1.3839e-01, ..., -6.7951e-01,
-6.8232e-01, 4.2479e-01],
[ 6.1522e-02, -5.0636e-01, -1.2924e-01, ..., -4.3473e-01,
-4.0362e-01, 1.0120e+00]]],
[[[-7.6737e-01, -1.5883e+00, -1.5857e+00, ..., -2.7996e+00,
-1.1571e+00, -4.8786e-01],
[-5.0898e-01, -2.5886e+00, -2.5145e+00, ..., -4.2914e+00,
-2.7296e+00, -2.1483e+00],
[-3.3650e-01, -2.0627e+00, -1.9772e+00, ..., -1.2919e+00,
-2.0126e+00, -3.2656e+00],
...,
[ 1.4143e-01, -6.8539e-01, 1.6893e+00, ..., -2.2410e-01,
-1.1133e+00, -1.8508e+00],
[-1.9756e-01, -5.8459e-01, -3.3874e-02, ..., -8.3055e-01,
-5.2992e-01, -4.6751e-01],
[-4.6577e-01, -1.2354e+00, -1.0807e+00, ..., -1.0583e+00,
-1.1998e+00, -5.5784e-01]],
[[-2.9245e+00, -5.3051e+00, -4.6859e+00, ..., -8.2771e+00,
-8.0980e+00, -3.7697e+00],
[-5.9348e+00, -8.7299e+00, -7.9877e+00, ..., -1.4049e+01,
-1.3642e+01, -6.2385e+00],
[-4.5934e+00, -6.5013e+00, -6.2562e+00, ..., -1.2168e+01,
-8.4420e+00, -3.7677e+00],
...,
[-2.2558e+00, -3.3272e+00, -3.0104e+00, ..., -4.4837e+00,
-4.6619e+00, -1.5257e+00],
[-3.1847e+00, -3.5929e+00, -2.7973e+00, ..., -4.5846e+00,
-4.7161e+00, -1.3471e+00],
[ 4.5660e-01, -5.9656e-02, 5.4006e-01, ..., 4.1632e-01,
5.2066e-01, 2.1158e-01]],
[[-6.5019e-01, 5.2129e-01, 9.3006e-01, ..., -4.2385e+00,
-2.6114e-01, 3.0350e+00],
[-8.7407e-01, 9.2657e-02, 4.2206e-01, ..., -3.0887e+00,
-2.8778e+00, 1.0970e+00],
[-1.7401e+00, -2.0389e-01, -7.3483e-02, ..., -8.2966e-01,
-1.0477e+00, 4.3389e-01],
...,
[-9.2877e-01, 6.5331e-01, 1.7745e+00, ..., -1.3934e+00,
1.0564e-01, 1.0799e+00],
[-5.9188e-01, 2.0485e-01, 2.8462e-01, ..., -6.4265e-01,
4.0034e-01, 5.5715e-01],
[-1.3352e-01, -6.7467e-02, 1.0961e-01, ..., 1.0378e-02,
1.5707e-01, 3.1636e-01]],
...,
[[ 1.4991e+00, -2.9694e-01, -7.3318e-02, ..., 7.0462e-01,
-2.4736e+00, -3.4948e+00],
[-6.5862e-01, -1.1097e+00, -7.2654e-01, ..., -4.1148e+00,
-5.4630e+00, -3.9477e+00],
[-6.2391e-01, -9.7635e-01, -4.2379e-01, ..., -8.0327e+00,
-4.0859e+00, -1.9340e+00],
...,
[-6.6379e-01, -1.2158e+00, -2.8436e+00, ..., -6.5807e+00,
-4.4610e+00, -8.9205e-01],
[-2.1788e+00, -1.7233e+00, -1.2695e+00, ..., -2.3357e+00,
-1.7383e+00, -1.1376e+00],
[-6.6128e-01, -8.1104e-01, -3.4494e-01, ..., -2.9071e-01,
-4.5719e-01, -4.6873e-01]],
[[ 1.0109e-01, 6.1556e-03, 6.3670e-01, ..., -2.0916e+00,
3.3489e-01, 5.3363e-01],
[ 3.1709e-01, -8.8822e-01, -6.0664e-02, ..., -1.1182e+00,
-9.4248e-02, -1.0583e+00],
[ 3.4551e-01, -6.8720e-01, 2.1940e-01, ..., 2.6830e+00,
8.3661e-01, -9.5843e-01],
...,
[-1.2928e-01, -4.5539e-01, 9.0552e-01, ..., 2.8363e+00,
7.0172e-03, -8.3115e-01],
[-4.5543e-01, -1.1468e+00, 6.7164e-03, ..., -7.2791e-01,
-8.9143e-01, -5.1545e-01],
[ 4.6827e-01, -5.3599e-02, -9.1601e-02, ..., -2.5667e-02,
-1.6178e-03, 1.8697e-01]],
[[-1.8409e+00, -9.9750e-01, -6.2237e-01, ..., -3.9892e+00,
2.9886e-02, 8.6501e-01],
[-2.5370e+00, -8.0396e-01, -6.8874e-01, ..., -5.3343e+00,
1.9479e+00, 3.0210e+00],
[-1.9728e+00, -7.1371e-01, -3.2977e-01, ..., -2.1238e+00,
1.5615e+00, 2.6473e+00],
...,
[-6.5838e-01, -5.9365e-01, 2.3979e-01, ..., -5.1959e-01,
8.7236e-01, 1.5768e+00],
[-1.0760e+00, -7.1800e-01, -3.7660e-01, ..., 1.1130e+00,
-1.9584e-01, 2.5753e-01],
[-3.5394e-01, -1.7891e-01, -1.4597e-01, ..., -2.0982e-01,
-1.8511e-01, 5.0424e-01]]]], device='cuda:0'),
tensor([[[[-1.8068e-01, -2.1603e-01, 1.5790e-01, ..., 5.9702e-02,
5.0381e-01, 1.5910e-01],
[-9.5321e-02, -1.5041e+00, -1.6281e+00, ..., -7.3218e-01,
-4.2842e-01, -2.9345e-01],
[-1.4239e-01, -1.3798e+00, -2.6609e+00, ..., -4.9433e-01,
-3.9201e-01, -6.7034e-01],
...,
[-1.7674e-01, -1.8944e+00, -1.4378e+00, ..., -1.0808e+00,
-1.1342e+00, -1.9507e+00],
[-4.4864e-01, -1.4892e+00, -7.3962e-01, ..., -4.4046e-01,
-3.9308e-01, -1.3876e+00],
[-2.4219e-01, -1.3011e+00, -1.1455e+00, ..., -8.5429e-01,
-1.1226e+00, -7.7766e-01]],
[[-1.6513e-01, -1.5441e+00, -7.2411e-01, ..., -5.4804e-01,
-1.3594e+00, -2.4581e-01],
[-2.0747e+00, -4.8151e+00, -4.4198e+00, ..., -2.3187e+00,
-3.2905e+00, -1.1131e+00],
[-1.8755e+00, -5.0090e+00, -5.8739e+00, ..., -2.4827e+00,
-2.8975e+00, -1.2113e+00],
...,
[-3.6822e+00, -5.1886e+00, -5.1354e+00, ..., -5.4921e+00,
-5.9573e+00, -2.8518e+00],
[-5.4074e+00, -6.6704e+00, -5.8705e+00, ..., -5.4380e+00,
-6.5271e+00, -2.7313e+00],
[ 5.4019e-01, -4.8432e-01, 5.3489e-01, ..., 5.3872e-01,
6.9153e-01, 1.1330e-01]],
[[-5.3395e-01, -1.6199e-01, 3.6939e-01, ..., 3.3787e-01,
5.7011e-02, 8.9456e-01],
[-4.3042e-01, -6.1261e-01, -3.9916e-01, ..., 3.5772e-01,
2.3229e-01, 4.8198e-01],
[-9.0597e-01, -4.4340e-01, -1.9553e+00, ..., 4.9371e-02,
1.4373e-01, 3.8026e-01],
...,
[-1.7532e+00, -4.1216e-01, -8.9021e-02, ..., -1.0917e-02,
-3.3546e-02, -1.6985e-01],
[-1.4430e+00, -3.3090e-01, -1.1766e-01, ..., -1.5958e-01,
-1.3802e-01, -2.5819e-01],
[-5.8276e-01, -9.8744e-02, 1.6897e-02, ..., -1.0416e-01,
-8.7298e-02, -4.9707e-01]],
...,
[[ 7.2476e-02, -5.9445e-01, -3.7033e-01, ..., -7.7104e-02,
-2.5331e-01, -7.4011e-01],
[-8.3643e-01, -4.3408e-01, -2.7547e-01, ..., -2.8910e-01,
-9.6887e-01, -7.3756e-01],
[-4.9094e-01, 3.9990e-01, 5.5190e-01, ..., 8.4461e-02,
-1.7238e-01, -5.2382e-01],
...,
[-1.0169e+00, -8.8282e-01, -2.9230e-01, ..., -3.3210e-01,
-5.9359e-01, -1.2308e+00],
[-2.5893e+00, -2.0404e+00, -1.3888e+00, ..., -1.3301e+00,
-1.4829e+00, -1.5671e+00],
[-5.9568e-01, -1.1792e+00, -1.6769e-01, ..., -2.0589e-01,
-3.9148e-01, -4.8141e-01]],
[[-1.5914e-01, 7.7823e-02, 4.3432e-01, ..., 1.1371e-01,
5.3535e-01, 1.5931e-01],
[-4.4366e-01, -1.2812e-01, 2.9305e-02, ..., -6.6368e-02,
1.0531e-01, -3.6827e-02],
[-7.0303e-01, -2.8703e-01, -7.8132e-01, ..., 4.2278e-02,
2.2940e-02, -2.7062e-04],
...,
[-1.6013e-01, -8.6875e-01, 2.8838e-01, ..., 6.1467e-01,
5.3597e-01, -3.2896e-01],
[-7.7555e-01, -1.8953e+00, -1.4580e-01, ..., 2.0617e-01,
1.7706e-01, -4.9818e-01],
[ 7.3543e-01, 8.9617e-02, 3.9894e-01, ..., 5.8085e-01,
5.3266e-01, 3.0362e-01]],
[[-6.8601e-01, 3.9340e-01, 5.6999e-01, ..., 4.8318e-01,
3.7560e-01, 5.9606e-01],
[-8.5875e-01, 1.3528e-01, -7.5862e-01, ..., -2.8763e-01,
-2.1531e-01, 6.4932e-01],
[-5.3880e-01, -2.8137e-01, -1.7149e+00, ..., 6.8163e-02,
3.0583e-02, 6.1014e-01],
...,
[-1.2626e+00, -6.0443e-01, -1.1704e-01, ..., -1.9233e-01,
-1.9176e-02, 1.4003e+00],
[-1.4475e+00, -8.4652e-01, -1.4272e-01, ..., -1.9291e-01,
8.6766e-02, 1.2194e+00],
[-6.6541e-01, -4.9798e-01, 1.4174e-01, ..., 5.0840e-02,
1.5386e-01, 1.2725e+00]]],
[[[-2.3063e-01, -3.0394e-01, 1.6847e-01, ..., 4.8887e-02,
5.0009e-01, 1.4504e-01],
[-1.5874e-01, -1.5543e+00, -8.8963e-01, ..., -7.4404e-01,
-4.6007e-01, -3.3146e-01],
[-1.3222e-01, -1.3656e+00, -6.9908e-01, ..., -5.3447e-01,
-4.4414e-01, -6.9485e-01],
...,
[-7.9500e-02, -1.7582e+00, -1.4945e+00, ..., -2.3314e+00,
-8.9928e-01, -1.9354e+00],
[-1.5661e-01, -1.2832e+00, -7.7793e-01, ..., 7.2782e-01,
1.7120e+00, 3.3024e-01],
[-8.4825e-02, -1.1405e+00, -1.0600e+00, ..., -1.6931e+00,
-1.4937e+00, -3.6539e-01]],
[[-2.5469e-01, -1.4658e+00, -6.4753e-01, ..., -6.4859e-01,
-1.3994e+00, -2.5883e-01],
[-2.0800e+00, -4.1007e+00, -2.5129e+00, ..., -2.4288e+00,
-3.4020e+00, -1.1549e+00],
[-1.9513e+00, -3.4980e+00, -2.6694e+00, ..., -4.5178e+00,
-3.1663e+00, -1.2801e+00],
...,
[-3.6704e+00, -5.0206e+00, -5.2710e+00, ..., -1.0506e+01,
-9.7642e+00, -4.2055e+00],
[-5.2274e+00, -6.0847e+00, -5.5752e+00, ..., -8.3790e+00,
-9.6189e+00, -2.9599e+00],
[ 5.6274e-01, -2.6655e-01, 5.3985e-01, ..., -1.5595e+00,
-1.9381e+00, -1.3594e+00]],
[[-5.4508e-01, -1.4440e-01, 3.7298e-01, ..., 3.8695e-01,
7.2630e-02, 8.9452e-01],
[-3.8132e-01, -4.3708e-01, 4.9563e-01, ..., 3.9627e-01,
2.8155e-01, 5.0196e-01],
[-8.8882e-01, -2.4902e-01, 2.1067e-01, ..., 6.1621e-01,
2.1236e-01, 4.1927e-01],
...,
[-1.7950e+00, -3.8443e-01, -1.4907e-01, ..., -1.2882e+00,
-7.3846e-03, 4.4909e-02],
[-1.4926e+00, -3.2447e-01, -1.9034e-01, ..., -7.6592e+00,
-2.7771e+00, -9.8256e-01],
[-5.6455e-01, -9.8707e-02, 1.2872e-03, ..., -5.5266e+00,
-2.0372e+00, -1.4330e+00]],
...,
[[ 7.4706e-02, -5.5681e-01, -3.6709e-01, ..., -6.6148e-02,
-2.3625e-01, -7.2767e-01],
[-8.2730e-01, -6.1063e-01, -5.4699e-01, ..., -2.7946e-01,
-9.9746e-01, -7.4824e-01],
[-4.9298e-01, -9.9000e-02, -6.2860e-03, ..., 1.6963e-01,
-1.9876e-01, -5.2699e-01],
...,
[-8.9404e-01, -9.5471e-01, -3.0417e-01, ..., -1.7374e+00,
-1.7448e+00, -1.5228e+00],
[-2.3055e+00, -1.9060e+00, -1.3167e+00, ..., -6.3851e+00,
-5.8570e+00, -3.5727e+00],
[-5.4242e-01, -1.0278e+00, -2.2057e-01, ..., -7.0027e+00,
-5.5718e+00, -3.0023e+00]],
[[-2.0168e-01, 1.7259e-02, 4.3451e-01, ..., 8.8443e-02,
5.3914e-01, 1.6622e-01],
[-4.2978e-01, -2.8276e-01, -5.8514e-03, ..., -9.2646e-02,
8.2594e-02, -8.1683e-02],
[-6.4937e-01, -5.5202e-01, -1.9904e-02, ..., -6.2224e-01,
-3.4087e-02, -6.8003e-02],
...,
[ 6.1167e-02, -6.9599e-01, 3.5186e-01, ..., 6.4524e-01,
7.2490e-01, -1.1065e+00],
[-5.8862e-01, -1.7834e+00, -8.1797e-02, ..., -6.5483e-01,
1.0127e+00, 4.5596e-01],
[ 7.4531e-01, 6.9137e-02, 3.7640e-01, ..., -3.8358e+00,
-1.8271e+00, -4.6624e-04]],
[[-6.8328e-01, 3.9885e-01, 5.6293e-01, ..., 4.8587e-01,
3.8821e-01, 6.2217e-01],
[-8.5895e-01, 2.6558e-01, -2.0817e-01, ..., -2.7839e-01,
-1.6519e-01, 6.6373e-01],
[-5.0492e-01, 3.0053e-01, 4.9503e-02, ..., 9.2017e-02,
7.8730e-02, 6.0858e-01],
...,
[-1.2983e+00, -5.7680e-01, -1.1268e-01, ..., -1.3883e+00,
-8.3285e-01, 2.0586e+00],
[-1.2830e+00, -7.1188e-01, -1.0195e-01, ..., -3.0164e+00,
-2.4046e+00, 1.3814e+00],
[-5.6551e-01, -3.8958e-01, 1.8201e-01, ..., -1.8311e+00,
-2.4928e+00, 1.1072e+00]]],
[[[-1.8263e-01, -2.7518e-01, -2.2953e-01, ..., 2.1357e-01,
5.3251e-01, 2.0536e-01],
[-3.7247e-02, -1.3771e+00, -4.5944e-01, ..., -9.8767e-01,
-4.9741e-01, -2.8816e-01],
[ 1.8998e-02, -1.1326e+00, -3.8436e-01, ..., -7.6278e-01,
-4.3493e-01, -7.2373e-01],
...,
[-1.9245e-01, -2.1385e+00, -1.6464e+00, ..., -6.1543e-02,
-6.4783e-01, -1.7792e+00],
[-4.5026e-01, -1.6279e+00, -3.8991e-01, ..., 6.1806e-01,
-3.7924e-01, -1.2224e+00],
[-3.0748e-01, -1.3514e+00, -1.1783e+00, ..., -1.2096e-01,
-1.0571e+00, -6.3287e-01]],
[[-2.2595e-01, -1.4619e+00, -1.9785e+00, ..., -1.0883e+00,
-1.3865e+00, -2.2437e-01],
[-1.9587e+00, -3.8841e+00, -2.0228e+00, ..., -2.7440e+00,
-3.6366e+00, -1.2843e+00],
[-1.8397e+00, -3.2696e+00, -2.7059e+00, ..., -2.9763e+00,
-3.1226e+00, -1.2974e+00],
...,
[-3.8873e+00, -5.5106e+00, -4.7538e+00, ..., -5.0496e+00,
-6.7792e+00, -3.5241e+00],
[-5.6338e+00, -7.0235e+00, -5.9260e+00, ..., -5.7497e+00,
-7.1185e+00, -3.0367e+00],
[ 4.9164e-01, -5.4185e-01, 5.2039e-01, ..., 4.1619e-01,
-7.4992e-02, -1.8579e-01]],
[[-4.3059e-01, -1.0349e-01, 1.4647e-01, ..., 4.4344e-01,
6.8174e-02, 9.6940e-01],
[-2.4970e-01, -3.5530e-01, 9.2551e-01, ..., 6.3743e-01,
2.7275e-01, 4.8431e-01],
[-7.1606e-01, -1.8689e-01, 8.2917e-01, ..., 3.0938e-01,
1.5553e-01, 3.6617e-01],
...,
[-1.1765e+00, -5.4782e-01, -9.7680e-01, ..., -2.7659e+00,
7.8207e-02, 9.3776e-02],
[-1.4102e+00, 2.2909e-01, 3.2400e-02, ..., -3.5845e+00,
-1.3642e+00, -1.8615e-01],
[-6.2144e-01, -9.7587e-02, -5.8351e-03, ..., -8.2448e-01,
-1.1794e+00, -4.4807e-01]],
...,
[[ 2.0140e-01, -4.9961e-01, -1.0136e-01, ..., -4.4947e-01,
-2.8122e-01, -8.1104e-01],
[-6.6299e-01, -6.1911e-01, -1.6374e-01, ..., -3.7303e-01,
-1.1240e+00, -8.2543e-01],
[-3.3030e-01, -1.7336e-01, -3.6660e-01, ..., 1.0013e-01,
-1.9689e-01, -5.6202e-01],
...,
[-1.3772e+00, -1.4788e+00, -4.2970e-01, ..., -1.9497e+00,
-1.8388e+00, -1.5005e+00],
[-2.7075e+00, -2.4521e+00, -2.1115e+00, ..., -3.6697e+00,
-3.2414e+00, -1.9267e+00],
[-5.8644e-01, -1.2488e+00, -1.8292e-01, ..., -2.0226e+00,
-1.7405e+00, -1.1656e+00]],
[[-1.1314e-01, 9.5581e-02, 2.5635e-01, ..., 1.5944e-01,
6.0081e-01, 1.9762e-01],
[-2.5948e-01, -1.7544e-01, -3.6273e-01, ..., -1.2430e-01,
1.3908e-01, -1.1569e-01],
[-4.9178e-01, -4.0803e-01, -9.5036e-02, ..., -2.2711e-01,
3.6451e-02, -6.5348e-02],
...,
[ 1.6552e-01, -9.0270e-01, -1.1725e+00, ..., 8.2361e-01,
7.5950e-01, -4.4184e-01],
[-7.6596e-01, -1.8543e+00, -2.9545e-01, ..., -6.6301e-01,
6.8931e-02, -6.4247e-01],
[ 7.2640e-01, 6.1672e-02, 3.7103e-01, ..., 5.8984e-01,
8.9848e-01, 3.3419e-01]],
[[-6.0105e-01, 4.1458e-01, 4.4554e-02, ..., 5.1809e-01,
4.0609e-01, 6.8135e-01],
[-8.0066e-01, 2.5706e-01, -4.4714e-01, ..., -1.2036e-01,
-1.6347e-01, 7.3617e-01],
[-4.7096e-01, 3.5124e-01, -2.1215e-01, ..., 2.0601e-01,
6.4379e-02, 5.9378e-01],
...,
[-1.7559e+00, -1.3769e+00, 6.3210e-01, ..., -2.0015e+00,
2.7889e-02, 1.4821e+00],
[-1.5867e+00, -1.0997e+00, 4.4686e-01, ..., -2.5325e+00,
3.6718e-01, 1.4371e+00],
[-6.8285e-01, -5.0815e-01, 1.1676e-01, ..., -2.4466e+00,
5.0929e-01, 1.5138e+00]]],
[[[-1.8922e-01, -5.6572e-01, -2.0866e-01, ..., 5.3882e-01,
7.6663e-01, -1.0557e-01],
[ 2.4266e-01, -1.3110e+00, -2.7638e-01, ..., -2.1155e+00,
-1.0154e-01, -2.3207e-01],
[ 1.0794e-01, -1.0117e+00, -1.1762e-01, ..., -1.3220e+00,
-4.5561e-01, -5.8521e-01],
...,
[-3.1110e-01, -1.8042e+00, -1.2912e+00, ..., -1.0561e+00,
-1.0308e+00, -2.0816e+00],
[-1.2376e+00, -1.8449e+00, -8.8228e-01, ..., -5.5156e-01,
-4.8731e-01, -1.8537e+00],
[-1.3428e+00, -2.2873e+00, -1.7007e+00, ..., -1.4918e+00,
-1.5874e+00, -9.9813e-01]],
[[-1.5748e+00, -3.2312e+00, -4.1474e+00, ..., -3.2275e+00,
-4.2380e+00, -1.3735e+00],
[-3.2968e+00, -5.1997e+00, -5.9713e+00, ..., -4.4419e+00,
-5.6974e+00, -2.0998e+00],
[-2.2184e+00, -3.2994e+00, -4.7423e+00, ..., -3.6273e+00,
-3.1597e+00, -1.3053e+00],
...,
[-3.8318e+00, -5.6303e+00, -5.3359e+00, ..., -5.5913e+00,
-6.3287e+00, -2.9616e+00],
[-6.2495e+00, -7.1513e+00, -6.1464e+00, ..., -6.2597e+00,
-7.6179e+00, -2.7365e+00],
[-8.8288e-02, -9.3795e-01, 2.3341e-01, ..., 3.4938e-01,
6.4170e-01, 2.0631e-01]],
[[-4.6795e-01, 2.3949e-01, 4.8097e-01, ..., 7.8190e-01,
8.8620e-01, 1.3683e+00],
[-4.3610e-01, -5.8514e-02, 1.2953e+00, ..., 6.6608e-01,
8.2349e-01, 3.3280e-01],
[-8.8596e-01, 5.7302e-02, 5.5048e-01, ..., 7.1079e-01,
5.5484e-01, 4.3745e-01],
...,
[-1.4165e+00, -3.0431e-02, 1.4271e-01, ..., 5.1112e-02,
7.6255e-03, 2.8556e-01],
[-1.1688e+00, 1.0970e-01, -9.8443e-02, ..., -5.3086e-02,
8.6508e-03, 2.8676e-01],
[-1.1876e+00, -6.9940e-02, 5.2755e-02, ..., -6.9712e-02,
-4.0519e-02, -3.5296e-01]],
...,
[[ 8.9676e-01, 2.2068e-01, 1.4204e+00, ..., -3.1996e+00,
-5.4241e-01, -9.1105e-01],
[-1.1531e+00, -8.5468e-01, 1.3991e-01, ..., -4.1325e+00,
-2.4460e+00, -1.1120e+00],
[-6.1418e-01, -2.8571e-01, 2.8996e-01, ..., -2.8524e+00,
-8.7442e-01, -6.3298e-01],
...,
[-6.3513e-01, -6.1928e-01, -4.8839e-01, ..., -3.7039e-01,
-5.8879e-01, -1.1934e+00],
[-3.2258e+00, -3.0249e+00, -2.0847e+00, ..., -1.9232e+00,
-2.0596e+00, -2.0980e+00],
[-5.6087e-01, -1.6666e+00, -1.1167e+00, ..., -8.7115e-01,
-9.4197e-01, -1.3646e+00]],
[[ 4.3227e-02, 2.4715e-01, 9.7587e-01, ..., 4.4068e-01,
1.0767e+00, 3.3518e-01],
[ 1.8599e-02, -4.7429e-01, 6.6089e-01, ..., -5.3579e-01,
5.6513e-01, -2.1062e-01],
[-2.8446e-01, -6.5268e-01, 6.8517e-01, ..., 2.5193e-01,
1.1158e-01, -3.1905e-02],
...,
[-1.5837e-01, -1.0264e+00, 2.1387e-01, ..., 2.6719e-01,
1.3743e-01, -7.1870e-01],
[-5.5587e-01, -1.8293e+00, -1.4962e-01, ..., 9.4427e-02,
-8.2933e-03, -9.5335e-01],
[ 4.2379e-01, -1.7538e-01, 1.3378e-01, ..., 2.6269e-01,
3.1854e-01, 3.4606e-01]],
[[-9.6450e-01, -2.4510e-01, -5.7903e-01, ..., -8.2020e-01,
1.2478e-01, 7.4031e-01],
[-1.1003e+00, -2.5890e-01, -8.7418e-01, ..., 3.4266e-02,
2.3857e-01, 1.1526e+00],
[-7.0280e-01, -2.5796e-01, -1.1916e+00, ..., 6.3958e-01,
2.6607e-01, 6.8082e-01],
...,
[-1.2483e+00, -1.0202e+00, -5.6128e-01, ..., -4.1369e-01,
-7.2314e-02, 8.2842e-01],
[-2.0175e+00, -1.9382e+00, -1.0254e+00, ..., -7.9087e-01,
-4.8922e-01, 4.5483e-01],
[-9.8648e-01, -9.9521e-01, -2.8579e-01, ..., -2.8261e-01,
-1.6074e-01, 1.1811e+00]]],
[[[-2.8650e-01, -4.4834e-01, -9.6112e-02, ..., -9.4066e-01,
-5.0617e-01, -1.4295e+00],
[ 3.0405e-03, -1.6175e+00, -7.5568e-01, ..., -1.1564e+00,
-1.3085e+00, -1.4869e+00],
[-7.2994e-01, -2.2226e+00, -1.3526e+00, ..., -1.3056e+00,
-1.0983e+00, -1.8203e+00],
...,
[-2.4635e-01, -1.2785e+00, -4.8523e-01, ..., -9.1543e-01,
-1.0445e+00, -1.9341e+00],
[-3.1328e-01, -9.8330e-01, -6.2142e-02, ..., -4.2187e-01,
-5.2553e-01, -1.7057e+00],
[-3.3422e-01, -1.4756e+00, -1.1019e+00, ..., -1.1608e+00,
-1.5590e+00, -7.8509e-01]],
[[-5.7783e-01, -1.9511e+00, -6.6062e-01, ..., -3.9479e+00,
-5.4751e+00, -1.8443e+00],
[-2.4309e+00, -4.4088e+00, -2.0446e+00, ..., -5.9010e+00,
-8.0381e+00, -3.3966e+00],
[-6.4421e+00, -7.3891e+00, -4.0677e+00, ..., -6.5526e+00,
-6.1905e+00, -2.5721e+00],
...,
[-1.6501e+00, -2.6958e+00, -1.4961e+00, ..., -5.0809e+00,
-6.3338e+00, -2.4799e+00],
[-3.4935e+00, -4.2592e+00, -2.6313e+00, ..., -6.4094e+00,
-7.9405e+00, -2.5577e+00],
[ 6.3120e-01, -6.3459e-01, 4.5190e-01, ..., 4.2547e-01,
4.8149e-01, 2.0781e-01]],
[[-4.5799e-01, 6.2882e-01, 1.0384e+00, ..., 6.3840e-01,
2.6193e-01, 1.5214e+00],
[-8.0469e-01, 1.9993e-01, 5.3439e-01, ..., 5.9997e-01,
-8.9433e-02, 1.0483e-01],
[-1.5738e+00, 9.3987e-02, 1.2709e-01, ..., 2.6588e-01,
-4.8547e-03, -1.1714e-01],
...,
[-1.4147e+00, 3.0900e-01, 1.4157e-01, ..., -1.4195e-02,
-7.1477e-02, -1.5169e-01],
[-1.3911e+00, 3.6168e-01, 2.4056e-01, ..., -6.8480e-02,
2.0886e-02, -1.8341e-01],
[-8.7846e-01, 6.6536e-02, -9.0449e-02, ..., 5.8068e-02,
2.0408e-01, -5.6158e-01]],
...,
[[ 2.6124e-01, -2.3464e-01, -1.3982e-01, ..., 4.2900e-01,
3.4357e-01, -1.2942e+00],
[-1.6540e+00, -1.1662e+00, -7.4285e-01, ..., -1.4682e-01,
-7.7376e-01, -1.3929e+00],
[-1.0061e+00, -9.9985e-01, -4.0948e-01, ..., 1.0975e-01,
-9.6129e-02, -9.6741e-01],
...,
[-9.4069e-01, -5.7173e-01, -1.0252e-01, ..., -1.7216e-01,
-1.2334e-01, -9.3276e-01],
[-2.3343e+00, -1.5670e+00, -7.7041e-01, ..., -1.5474e+00,
-1.5853e+00, -1.8145e+00],
[-6.5813e-01, -1.0338e+00, -1.1318e-01, ..., -3.9975e-01,
-5.3679e-01, -1.1832e+00]],
[[-4.2213e-03, 2.6150e-01, 5.6361e-01, ..., -7.8934e-02,
3.6503e-01, -3.9114e-01],
[-2.9827e-01, -7.1441e-01, -1.2404e-01, ..., 1.9866e-01,
5.7684e-02, -8.6189e-01],
[-7.0279e-01, -1.8874e+00, -1.6170e+00, ..., -8.9721e-01,
1.3767e-01, -4.5719e-01],
...,
[-5.7576e-01, -1.1140e+00, -1.1881e-01, ..., 2.2934e-01,
1.0590e-01, -4.9996e-01],
[-9.9910e-01, -1.8862e+00, -3.5953e-01, ..., -1.3509e-01,
-2.6998e-01, -9.1265e-01],
[ 2.2930e-01, -5.1869e-01, -2.5109e-01, ..., 3.1857e-01,
3.1226e-01, 2.7796e-01]],
[[-5.0047e-01, 9.4523e-02, 5.6729e-01, ..., -6.5795e-02,
-4.5152e-02, 5.9244e-01],
[-4.2146e-01, -4.3332e-03, -2.3430e-02, ..., -9.4092e-01,
-4.2165e-01, 8.8622e-01],
[-3.6228e-01, -2.0485e-01, 1.3468e-01, ..., -4.1923e-01,
-5.4373e-01, 6.0858e-01],
...,
[ 2.6068e-02, -5.5064e-02, 7.0091e-02, ..., -5.1017e-01,
-5.3357e-01, 6.1281e-01],
[-3.0412e-01, -3.6720e-01, -1.1999e-01, ..., -6.8795e-01,
-6.8749e-01, 4.0131e-01],
[ 7.3667e-02, -4.6519e-01, -1.2990e-01, ..., -5.0383e-01,
-4.5022e-01, 9.9126e-01]]]], device='cuda:0'),
tensor([[[[-7.4039e-01, -1.4187e+00, -1.3818e+00, ..., -2.0935e-01,
1.7644e-01, -4.3038e-01],
[-2.9405e-01, -2.4305e+00, -2.1632e+00, ..., -5.9701e-01,
-3.6474e-01, -5.9791e-01],
[-3.9779e-01, -2.1140e+00, -1.8317e+00, ..., -3.5352e-01,
-3.5495e-01, -1.0592e+00],
...,
[-4.8478e-01, -1.9297e+00, -1.8591e+00, ..., -2.5893e-01,
-1.6563e-01, -1.1534e+00],
[-9.6480e-01, -1.6889e+00, -1.1945e+00, ..., 1.4918e-02,
1.0163e-01, -9.5262e-01],
[-7.7435e-01, -1.5657e+00, -1.2077e+00, ..., -9.2363e-01,
-1.1399e+00, -6.5621e-01]],
[[-2.3517e+00, -4.7401e+00, -4.1404e+00, ..., -2.2488e+00,
-3.5153e+00, -1.3622e+00],
[-5.3600e+00, -8.0880e+00, -6.5686e+00, ..., -3.4933e+00,
-5.1511e+00, -2.1440e+00],
[-4.3493e+00, -6.2218e+00, -5.5825e+00, ..., -3.0454e+00,
-3.8488e+00, -1.7453e+00],
...,
[-4.3920e+00, -6.1035e+00, -5.5082e+00, ..., -3.0880e+00,
-3.9378e+00, -1.7310e+00],
[-6.3867e+00, -7.4412e+00, -6.3197e+00, ..., -3.9322e+00,
-5.3924e+00, -1.9255e+00],
[-1.1080e-02, -8.2128e-01, 3.7123e-01, ..., 4.1799e-01,
5.9617e-01, 1.7763e-01]],
[[-6.1545e-01, 5.8790e-01, 1.0893e+00, ..., 8.1890e-01,
5.3538e-01, 1.3352e+00],
[-9.1943e-01, -4.4557e-02, 4.4105e-01, ..., 4.1022e-01,
2.9697e-01, 1.6124e-01],
[-1.6533e+00, -1.3809e-01, -3.1832e-02, ..., 8.4950e-02,
1.8797e-01, 3.7877e-02],
...,
[-1.7363e+00, -7.1336e-02, -9.9063e-02, ..., 3.1735e-01,
1.4732e-01, 9.6234e-02],
[-1.5452e+00, 8.5484e-02, 6.3282e-02, ..., 6.7978e-02,
2.4105e-01, 1.5365e-01],
[-1.4151e+00, -5.3565e-01, -2.7642e-01, ..., -1.5447e-02,
1.6902e-01, -1.4101e-01]],
...,
[[ 1.0267e+00, -4.4287e-01, -1.9544e-01, ..., 3.7040e-01,
1.9965e-01, -7.6513e-01],
[-8.2579e-01, -9.7196e-01, -8.1471e-01, ..., -2.6260e-01,
-1.0448e+00, -9.8055e-01],
[-6.7849e-01, -9.4842e-01, -7.1581e-01, ..., -1.4017e-01,
-2.0714e-01, -6.3766e-01],
...,
[-7.1984e-01, -1.1217e+00, -7.5982e-01, ..., -2.9313e-01,
-3.8285e-01, -8.0935e-01],
[-2.6950e+00, -2.7668e+00, -1.8225e+00, ..., -1.0967e+00,
-1.1402e+00, -1.1106e+00],
[-6.3921e-01, -1.6490e+00, -9.6983e-01, ..., -1.9499e-01,
-2.6099e-01, -4.8085e-01]],
[[-2.6580e-03, -4.1779e-03, 5.9551e-01, ..., 3.1573e-01,
8.0075e-01, 5.4580e-02],
[-3.3996e-02, -7.5271e-01, 1.2227e-01, ..., 3.6619e-01,
6.7398e-01, -2.5339e-01],
[-1.3555e-01, -8.3313e-01, 2.9735e-01, ..., 3.9732e-01,
4.3170e-01, -1.3233e-01],
...,
[ 5.1456e-02, -7.8186e-01, 2.9936e-01, ..., 4.9936e-01,
4.7036e-01, -3.0187e-01],
[-2.7370e-01, -1.4851e+00, 3.1078e-02, ..., 2.9384e-02,
1.2596e-01, -4.4217e-01],
[ 6.8827e-01, 1.9816e-01, 3.0756e-01, ..., 2.1642e-01,
2.9681e-01, 3.6597e-01]],
[[-1.5990e+00, -5.9929e-01, 2.3084e-02, ..., 3.1175e-01,
2.2221e-01, 8.6225e-01],
[-2.0829e+00, -3.1815e-01, -6.1245e-02, ..., -4.7370e-01,
-1.5482e-01, 1.2616e+00],
[-1.7424e+00, -3.6142e-01, 1.9842e-01, ..., -1.2775e-01,
-1.5524e-02, 1.0170e+00],
...,
[-1.7463e+00, -4.8905e-01, 2.4593e-01, ..., -1.3482e-01,
1.3756e-01, 1.0953e+00],
[-2.1484e+00, -1.0749e+00, -1.0714e-01, ..., -2.5281e-01,
-1.0154e-01, 6.2329e-01],
[-1.0597e+00, -1.6150e-01, 4.9851e-01, ..., -2.4522e-01,
-2.3940e-01, 7.1190e-01]]],
[[[-1.9577e-01, -2.6461e-01, 1.8190e-01, ..., 2.1551e-02,
4.9486e-01, 1.4903e-01],
[-1.0815e-01, -1.5447e+00, -9.1036e-01, ..., -7.0148e-01,
-4.4294e-01, -3.4247e-01],
[-1.3735e-01, -1.3523e+00, -6.6659e-01, ..., -5.0280e-01,
-4.3212e-01, -6.8945e-01],
...,
[-2.5434e-01, -1.4734e+00, -1.1193e+00, ..., -9.5414e-01,
-9.9175e-01, -2.0386e+00],
[-1.0846e+00, -1.6106e+00, -4.9438e-01, ..., -5.4033e-01,
-4.7759e-01, -1.7894e+00],
[-1.2444e+00, -2.1517e+00, -1.6432e+00, ..., -1.4349e+00,
-1.5643e+00, -9.7725e-01]],
[[-1.9033e-01, -1.4676e+00, -6.6098e-01, ..., -6.4512e-01,
-1.4056e+00, -2.3199e-01],
[-2.0732e+00, -4.1175e+00, -2.5978e+00, ..., -2.3356e+00,
-3.3880e+00, -1.1554e+00],
[-1.9322e+00, -3.5145e+00, -2.6816e+00, ..., -2.5263e+00,
-2.9561e+00, -1.2280e+00],
...,
[-3.7305e+00, -5.4527e+00, -5.0617e+00, ..., -5.4118e+00,
-6.1464e+00, -2.9203e+00],
[-6.0971e+00, -7.1082e+00, -6.2137e+00, ..., -6.1064e+00,
-7.4463e+00, -2.7490e+00],
[-3.6409e-02, -9.0719e-01, 2.6464e-01, ..., 4.1888e-01,
7.0878e-01, 1.7839e-01]],
[[-5.3163e-01, -1.3881e-01, 3.8967e-01, ..., 3.6403e-01,
4.3444e-02, 9.0204e-01],
[-4.2437e-01, -4.9312e-01, 4.9635e-01, ..., 3.9950e-01,
2.9717e-01, 4.9288e-01],
[-9.0585e-01, -3.0184e-01, 1.5945e-01, ..., -4.5113e-03,
1.8443e-01, 4.0536e-01],
...,
[-1.4015e+00, 3.4592e-01, -1.1501e+00, ..., 4.1529e-02,
-2.7625e-02, 3.0176e-01],
[-1.2534e+00, 5.5442e-01, -6.6345e-02, ..., -1.6141e-02,
-3.1635e-04, 3.2120e-01],
[-1.1616e+00, -2.4224e-02, 1.0806e-01, ..., -9.3810e-03,
-8.4944e-02, -3.7241e-01]],
...,
[[ 7.6229e-02, -5.9878e-01, -3.8410e-01, ..., -5.8068e-02,
-2.4562e-01, -7.4516e-01],
[-8.3438e-01, -5.9877e-01, -5.5317e-01, ..., -2.9320e-01,
-1.0040e+00, -7.6256e-01],
[-4.7891e-01, -4.6504e-02, -2.4071e-03, ..., 7.3226e-02,
-1.9313e-01, -5.2829e-01],
...,
[-6.3026e-01, -3.8606e-01, 8.3350e-01, ..., -3.7844e-01,
-6.0134e-01, -1.2623e+00],
[-3.1256e+00, -3.1677e+00, -2.3600e+00, ..., -1.8980e+00,
-2.0324e+00, -2.0829e+00],
[-5.4143e-01, -1.6028e+00, -1.0865e+00, ..., -8.5737e-01,
-9.3542e-01, -1.3417e+00]],
[[-1.6991e-01, 4.0227e-02, 4.0953e-01, ..., 5.3163e-02,
5.0927e-01, 1.7834e-01],
[-4.3254e-01, -2.2437e-01, 1.5407e-02, ..., -2.0333e-02,
1.4017e-01, -8.1789e-02],
[-6.8056e-01, -5.1001e-01, 4.5855e-02, ..., 2.8450e-02,
1.0862e-02, -2.9498e-02],
...,
[-1.3156e-01, -7.9773e-01, -2.5343e-02, ..., 2.9199e-01,
1.5282e-01, -7.5534e-01],
[-5.3416e-01, -1.6796e+00, -1.4536e-01, ..., 4.3766e-02,
-1.9722e-02, -9.4640e-01],
[ 4.4805e-01, -1.5367e-01, 8.5596e-02, ..., 2.7649e-01,
3.1342e-01, 3.2941e-01]],
[[-6.7300e-01, 4.0620e-01, 5.7320e-01, ..., 4.7956e-01,
3.7668e-01, 6.2479e-01],
[-8.6933e-01, 2.8324e-01, -1.9859e-01, ..., -2.7730e-01,
-1.8470e-01, 6.5647e-01],
[-5.2481e-01, 3.2751e-01, 7.0717e-02, ..., 7.0175e-02,
8.1628e-02, 6.0491e-01],
...,
[-1.0805e+00, -1.2512e+00, -3.2116e-01, ..., -4.2374e-01,
-4.4228e-02, 8.2748e-01],
[-1.8075e+00, -1.9556e+00, -8.1315e-01, ..., -8.4488e-01,
-4.6166e-01, 4.3135e-01],
[-9.1117e-01, -9.3420e-01, -2.8339e-01, ..., -3.1838e-01,
-1.4571e-01, 1.1343e+00]]],
[[[ 1.3241e-02, -3.3655e-01, -1.0418e-01, ..., -1.0369e+00,
-5.3914e-01, -1.5343e+00],
[ 2.7158e-01, -1.1779e+00, -8.7423e-01, ..., -1.6104e+00,
-1.3295e+00, -1.5697e+00],
[ 2.5432e-01, -1.1412e+00, -1.2288e+00, ..., -1.1059e+00,
-1.2484e+00, -1.9024e+00],
...,
[-3.5657e+00, -2.7943e+00, 1.4752e+00, ..., 2.5047e+00,
-2.0556e+00, -2.0296e+00],
[-3.5033e+00, -2.1041e+00, 7.2147e-01, ..., -1.8579e-01,
-2.0296e+00, -1.8590e+00],
[-1.1798e+00, -2.2054e+00, -1.1316e+00, ..., -1.2494e+00,
-1.6211e+00, -9.0044e-01]],
[[-8.5515e-01, -2.4130e+00, -2.2547e+00, ..., -3.8964e+00,
-5.5121e+00, -1.8163e+00],
[-2.5152e+00, -4.5591e+00, -4.2684e+00, ..., -7.4189e+00,
-8.4088e+00, -3.4473e+00],
[-2.7042e+00, -4.2464e+00, -4.8472e+00, ..., -1.0462e+01,
-7.3332e+00, -2.4978e+00],
...,
[-5.0861e+00, -4.6524e+00, -3.0936e+00, ..., -5.0772e+00,
-6.6198e+00, -2.4705e+00],
[-4.6798e+00, -6.4784e+00, -6.1977e+00, ..., -6.8761e+00,
-8.6189e+00, -2.4819e+00],
[ 5.1471e-01, -1.2319e+00, -2.2976e-01, ..., 3.1455e-01,
3.7632e-01, 2.7343e-01]],
[[-2.4623e-01, 1.4210e-01, 8.3250e-01, ..., 6.2134e-01,
1.6529e-01, 1.4902e+00],
[ 4.5383e-02, -4.0169e-01, 4.9654e-01, ..., 7.1926e-01,
-1.4280e-01, -1.0144e-02],
[-3.7616e-01, -2.8496e-01, 2.0908e-01, ..., 2.1340e+00,
2.3320e-02, -1.9774e-01],
...,
[-3.8376e+00, -6.6479e+00, -4.7315e+00, ..., -4.9007e+00,
8.9202e-01, -1.7107e-01],
[-7.0207e-01, -2.6270e+00, -4.6649e+00, ..., -1.7366e+00,
8.4732e-01, -2.0177e-01],
[ 8.7985e-01, 5.0786e-02, -1.1108e+00, ..., 5.5473e-02,
2.0148e-01, -6.3880e-01]],
...,
[[ 4.9647e-01, -2.4249e-01, 1.1288e-01, ..., 4.3223e-01,
3.9063e-01, -1.3460e+00],
[-7.0218e-01, -4.9333e-01, -4.8696e-01, ..., -2.5210e-01,
-7.4550e-01, -1.3890e+00],
[-3.7543e-01, -1.0140e-01, 2.4254e-01, ..., -8.5513e-01,
-8.7000e-02, -9.2090e-01],
...,
[ 1.0670e+00, -8.3205e-01, -6.8136e+00, ..., -6.0492e+00,
-2.0210e+00, -9.6690e-01],
[-1.0628e+00, -5.5830e+00, -8.1919e+00, ..., -4.0689e+00,
-2.2918e+00, -1.9351e+00],
[-1.5775e-01, -2.3972e+00, -1.4733e+00, ..., -4.4140e-01,
-5.3146e-01, -1.2678e+00]],
[[ 1.2578e-01, 2.8559e-01, 6.7728e-01, ..., -8.3569e-02,
4.0970e-01, -2.7697e-01],
[ 3.6047e-02, 2.0673e-01, 5.5560e-01, ..., -8.7016e-01,
8.0681e-02, -8.1546e-01],
[-4.2701e-02, 2.8878e-03, 8.4010e-02, ..., -2.0371e+00,
-6.8401e-01, -4.1448e-01],
...,
[-1.9133e+00, -1.1560e-01, 2.2880e+00, ..., 4.0752e+00,
-1.1008e+00, -4.6355e-01],
[-2.6412e+00, -1.8584e-01, 9.6358e-01, ..., 1.8993e-01,
-2.0510e+00, -8.5624e-01],
[-4.9719e-02, 9.3732e-01, 6.1027e-01, ..., 2.4231e-01,
2.7321e-01, 3.4261e-01]],
[[-9.0255e-01, 2.3577e-01, 3.3902e-01, ..., -2.2949e-01,
-1.8719e-01, 5.5134e-01],
[-1.1861e+00, 1.8750e-01, -4.2451e-01, ..., -1.0640e+00,
-4.2179e-01, 9.5390e-01],
[-9.8955e-01, -4.7702e-02, -8.2370e-01, ..., -1.4070e+00,
-6.9372e-01, 6.2616e-01],
...,
[-9.1165e+00, -4.6276e+00, -7.8908e-01, ..., 1.9379e+00,
7.5376e-01, 6.1612e-01],
[-5.3579e+00, -2.2874e+00, 2.6650e+00, ..., 1.2149e+00,
-8.7330e-01, 3.3679e-01],
[-8.5495e-01, -1.8166e-01, 9.7614e-01, ..., -5.0493e-01,
-5.2345e-01, 1.0546e+00]]],
[[[-2.0926e-01, -2.8427e-01, 2.6601e-01, ..., 9.0933e-02,
5.2168e-01, 2.1645e-01],
[-8.2567e-02, -1.3837e+00, -7.1139e-01, ..., -5.9155e-01,
-4.9883e-01, -3.0107e-01],
[-2.1586e-02, -1.1445e+00, -5.7628e-01, ..., -5.6614e-01,
-4.4320e-01, -7.2353e-01],
...,
[-2.8784e-01, -8.6634e-01, -3.1124e+00, ..., -9.7004e-01,
-9.9199e-01, -2.0500e+00],
[-1.1162e+00, -9.3681e-01, -2.1618e+00, ..., -5.4893e-01,
-4.7329e-01, -1.7941e+00],
[-1.2773e+00, -1.8854e+00, -1.4696e+00, ..., -1.4454e+00,
-1.5499e+00, -9.7540e-01]],
[[-2.0818e-01, -1.3530e+00, -2.0198e+00, ..., -1.1512e+00,
-1.4147e+00, -2.4855e-01],
[-1.9355e+00, -3.7877e+00, -4.6290e+00, ..., -3.5840e+00,
-3.6461e+00, -1.2794e+00],
[-1.8639e+00, -3.2463e+00, -4.3237e+00, ..., -4.0836e+00,
-3.1379e+00, -1.3237e+00],
...,
[-3.7740e+00, -8.8376e+00, -8.8471e+00, ..., -5.4445e+00,
-6.1986e+00, -2.9186e+00],
[-6.2270e+00, -8.4459e+00, -9.7623e+00, ..., -6.0685e+00,
-7.3471e+00, -2.6706e+00],
[-4.2740e-02, -1.1334e+00, -2.8986e-01, ..., 3.8733e-01,
6.8068e-01, 1.9379e-01]],
[[-4.3381e-01, -6.8159e-02, 2.9760e-01, ..., 5.5970e-01,
8.3314e-02, 9.8113e-01],
[-3.0360e-01, -3.6625e-01, 3.8499e-01, ..., 8.9759e-01,
2.8617e-01, 4.7821e-01],
[-7.3834e-01, -1.9004e-01, 5.1708e-01, ..., 5.5597e-01,
1.9368e-01, 4.0376e-01],
...,
[-1.3890e+00, 1.0675e+00, -7.7230e+00, ..., 5.5318e-02,
-4.3843e-03, 2.9694e-01],
[-1.2142e+00, 1.5242e+00, -6.1277e+00, ..., 4.3652e-03,
-1.0239e-02, 3.1247e-01],
[-1.1933e+00, 1.5795e+00, -1.5755e+00, ..., 1.1118e-02,
-5.5174e-02, -3.5770e-01]],
...,
[[ 2.6149e-01, -4.6894e-01, -2.9547e-01, ..., -2.4213e-02,
-2.7543e-01, -8.1774e-01],
[-6.4703e-01, -6.2691e-01, -3.5970e-01, ..., -5.6139e-01,
-1.1671e+00, -8.5058e-01],
[-3.5349e-01, -1.6169e-01, 2.9207e-01, ..., -3.0602e-01,
-1.9324e-01, -5.3343e-01],
...,
[-5.9940e-01, -2.3501e-01, -1.4837e+00, ..., -3.6053e-01,
-5.8961e-01, -1.1997e+00],
[-3.1135e+00, -1.7224e+00, -4.4748e-01, ..., -1.9206e+00,
-2.0495e+00, -2.0715e+00],
[-5.2440e-01, -1.5599e+00, -8.6860e-01, ..., -8.5173e-01,
-9.3659e-01, -1.3359e+00]],
[[-9.5766e-02, 1.0796e-01, 6.4080e-01, ..., -5.9986e-02,
6.0858e-01, 2.1360e-01],
[-2.5863e-01, -2.3594e-01, 2.3992e-01, ..., 1.6073e-02,
1.3291e-01, -1.0772e-01],
[-5.1444e-01, -4.3517e-01, 8.1185e-02, ..., 2.8420e-01,
3.6654e-02, -6.4481e-02],
...,
[-1.3336e-01, -1.2402e+00, -2.3088e+00, ..., 2.7661e-01,
1.4796e-01, -7.3033e-01],
[-5.2750e-01, -1.8427e+00, -2.0823e+00, ..., 1.7268e-02,
-3.1335e-02, -9.1549e-01],
[ 4.6217e-01, 4.1997e-02, -1.0148e+00, ..., 2.6462e-01,
3.0211e-01, 3.4046e-01]],
[[-5.6576e-01, 4.2145e-01, 6.6240e-01, ..., 5.3476e-01,
4.0952e-01, 6.6107e-01],
[-7.7908e-01, 2.2640e-01, -3.4214e-01, ..., -1.9556e-01,
-1.3026e-01, 7.3037e-01],
[-4.6290e-01, 3.1595e-01, -1.5678e-01, ..., 4.3478e-01,
9.9461e-02, 6.0068e-01],
...,
[-1.1419e+00, -2.6881e+00, -6.5539e-01, ..., -4.3351e-01,
-8.0001e-02, 8.0413e-01],
[-1.8430e+00, -2.8416e+00, -3.9225e-01, ..., -8.2508e-01,
-4.8913e-01, 4.2076e-01],
[-9.5938e-01, -1.4496e+00, 3.4501e-01, ..., -2.7988e-01,
-1.4373e-01, 1.1557e+00]]],
[[[-2.0083e-01, -3.1611e-01, 1.3682e-01, ..., 6.1770e-03,
4.1486e-01, 9.0582e-02],
[-1.2125e-01, -1.5971e+00, -8.7595e-01, ..., -7.9803e-01,
-5.5622e-01, -4.1878e-01],
[-9.7702e-02, -1.3196e+00, -5.5952e-01, ..., -5.3253e-01,
-4.6885e-01, -7.6380e-01],
...,
[-2.7178e-01, -1.7033e+00, -1.1968e+00, ..., -1.6474e+00,
-1.0141e+00, -2.0888e+00],
[-1.1755e+00, -1.7784e+00, -8.3774e-01, ..., -5.7852e-01,
-4.9992e-01, -1.8197e+00],
[-1.3099e+00, -2.2269e+00, -1.6662e+00, ..., -1.5019e+00,
-1.6074e+00, -9.9124e-01]],
[[-2.5856e-01, -1.5804e+00, -8.3249e-01, ..., -7.1729e-01,
-1.4955e+00, -3.5681e-01],
[-2.1946e+00, -4.3246e+00, -3.2069e+00, ..., -2.6169e+00,
-3.5224e+00, -1.2632e+00],
[-2.0140e+00, -3.8766e+00, -4.6000e+00, ..., -2.8123e+00,
-3.2399e+00, -1.3776e+00],
...,
[-3.8479e+00, -5.7140e+00, -5.2923e+00, ..., -5.6186e+00,
-6.2704e+00, -2.9271e+00],
[-6.1531e+00, -7.0284e+00, -6.1139e+00, ..., -6.2564e+00,
-7.5425e+00, -2.7382e+00],
[-7.0441e-02, -9.0817e-01, 2.2174e-01, ..., 3.8526e-01,
6.8814e-01, 2.1099e-01]],
[[-5.1904e-01, -1.2352e-01, 4.1143e-01, ..., 4.0323e-01,
1.4381e-01, 9.7125e-01],
[-3.8540e-01, -4.5329e-01, 4.7764e-01, ..., 3.5344e-01,
2.2348e-01, 4.7088e-01],
[-9.0732e-01, -1.8572e-01, -1.2690e-01, ..., 7.4194e-02,
1.4451e-01, 3.5340e-01],
...,
[-1.4476e+00, 3.5148e-02, 4.3746e-01, ..., 6.1796e-02,
2.7852e-03, 2.6720e-01],
[-1.2566e+00, 1.4786e-01, -3.9413e-02, ..., -4.8132e-02,
1.5113e-02, 2.7833e-01],
[-1.1966e+00, -5.8063e-02, 3.4014e-02, ..., -2.0457e-02,
-7.1850e-02, -3.6007e-01]],
...,
[[ 8.1109e-02, -5.3686e-01, -3.3366e-01, ..., -3.4871e-02,
-1.9016e-01, -7.2282e-01],
[-8.8541e-01, -6.1143e-01, -4.6128e-01, ..., -2.8049e-01,
-9.7609e-01, -7.7101e-01],
[-4.9416e-01, -6.6700e-02, 3.7740e-01, ..., 1.2574e-01,
-1.8743e-01, -5.6789e-01],
...,
[-5.7200e-01, -6.6226e-01, -7.8339e-01, ..., -3.9290e-01,
-5.6363e-01, -1.1994e+00],
[-3.1719e+00, -3.0441e+00, -2.0942e+00, ..., -1.9353e+00,
-2.0595e+00, -2.1166e+00],
[-5.4469e-01, -1.7257e+00, -1.1236e+00, ..., -8.2464e-01,
-9.2575e-01, -1.3819e+00]],
[[-1.6992e-01, 3.1148e-02, 4.2136e-01, ..., 1.0785e-01,
5.3409e-01, 1.4647e-01],
[-4.2644e-01, -2.6765e-01, 1.3735e-01, ..., -4.6481e-02,
1.2474e-01, -1.0513e-01],
[-6.5063e-01, -5.2749e-01, 5.4824e-01, ..., 4.2280e-02,
6.5671e-02, -5.6616e-02],
...,
[-1.0579e-01, -1.0818e+00, 1.9108e-01, ..., -3.6112e-01,
1.4428e-01, -7.0890e-01],
[-4.8163e-01, -1.8492e+00, -1.6477e-01, ..., 3.4700e-02,
-2.5872e-02, -9.3997e-01],
[ 4.6689e-01, -1.5186e-01, 1.0630e-01, ..., 2.6438e-01,
3.3137e-01, 3.6142e-01]],
[[-6.7593e-01, 4.0587e-01, 5.8048e-01, ..., 4.7544e-01,
3.9671e-01, 6.2256e-01],
[-8.5687e-01, 2.2740e-01, -2.5318e-01, ..., -2.5195e-01,
-1.0997e-01, 7.0006e-01],
[-4.6344e-01, 2.8143e-01, -2.0377e-01, ..., 9.2247e-02,
9.2876e-02, 6.1654e-01],
...,
[-1.1409e+00, -9.5263e-01, -5.7526e-01, ..., -3.5122e-02,
-9.2055e-02, 8.4690e-01],
[-1.9795e+00, -1.9000e+00, -1.0295e+00, ..., -8.3441e-01,
-5.1594e-01, 4.2492e-01],
[-9.8432e-01, -1.0057e+00, -2.5929e-01, ..., -3.1189e-01,
-1.5767e-01, 1.1729e+00]]]], device='cuda:0'),
tensor([[[[-7.1971e-01, -1.4022e+00, -1.4015e+00, ..., 3.5354e-03,
6.4663e-01, 5.5988e-01],
[-2.8767e-01, -2.4250e+00, -2.1413e+00, ..., -4.8991e-01,
-2.3855e-01, 1.2585e-01],
[-3.9796e-01, -2.1106e+00, -1.8389e+00, ..., -3.2410e-01,
-9.6301e-02, -9.1377e-02],
...,
[-4.7831e-01, -1.8556e+00, -1.8311e+00, ..., -3.3414e-01,
-6.6575e-02, -1.3161e-01],
[-9.4687e-01, -1.6388e+00, -1.1525e+00, ..., 1.7234e-02,
2.7102e-01, -2.2743e-02],
[-7.9757e-01, -1.6489e+00, -1.2753e+00, ..., -1.0500e+00,
-1.0999e+00, -3.9622e-01]],
[[-2.3898e+00, -4.8131e+00, -4.1943e+00, ..., -4.1613e-01,
-9.8046e-01, 4.7452e-03],
[-5.3520e+00, -8.0607e+00, -6.5488e+00, ..., -1.7661e+00,
-2.6324e+00, -6.6162e-01],
[-4.2978e+00, -6.2229e+00, -5.4491e+00, ..., -1.2592e+00,
-1.4833e+00, -3.7132e-01],
...,
[-4.3942e+00, -6.0846e+00, -5.5502e+00, ..., -1.2848e+00,
-1.5782e+00, -3.7358e-01],
[-6.3334e+00, -7.3952e+00, -6.2906e+00, ..., -2.4602e+00,
-3.3468e+00, -9.5359e-01],
[-6.6419e-03, -8.1971e-01, 3.7264e-01, ..., 3.3571e-01,
3.5047e-01, 8.2822e-02]],
[[-5.8609e-01, 5.8721e-01, 1.0851e+00, ..., 8.1622e-01,
3.0733e-01, 1.0887e+00],
[-9.2390e-01, -7.4763e-02, 3.7384e-01, ..., 3.9796e-01,
4.5891e-02, 4.9112e-01],
[-1.6314e+00, -1.6019e-01, -8.2422e-02, ..., 3.8242e-02,
7.6681e-02, 4.0424e-01],
...,
[-1.6978e+00, -1.5930e-01, -1.3732e-01, ..., 4.1533e-02,
7.7194e-02, 4.3854e-01],
[-1.5280e+00, 7.3063e-02, 3.7578e-02, ..., 1.1268e-01,
1.0048e-01, 6.0149e-01],
[-1.4115e+00, -5.2892e-01, -2.3800e-01, ..., -8.6985e-02,
5.7795e-02, 3.5003e-01]],
...,
[[ 1.0486e+00, -4.0722e-01, -2.0503e-01, ..., -1.1544e-01,
-4.0033e-01, -9.4154e-01],
[-8.3900e-01, -9.6543e-01, -7.9558e-01, ..., -6.8886e-01,
-1.2670e+00, -8.2086e-01],
[-7.1284e-01, -9.6989e-01, -6.9650e-01, ..., -1.2226e-01,
-1.7826e-01, -3.6542e-01],
...,
[-7.2093e-01, -1.2006e+00, -7.5025e-01, ..., -1.0908e-01,
-1.7359e-01, -3.5890e-01],
[-2.7256e+00, -2.8524e+00, -1.8529e+00, ..., -8.1971e-01,
-8.7651e-01, -7.8697e-01],
[-6.7271e-01, -1.6587e+00, -1.0031e+00, ..., -4.7603e-04,
-1.3435e-01, -3.2820e-01]],
[[ 2.6359e-02, 8.7677e-03, 6.0845e-01, ..., 3.2548e-01,
6.9367e-01, 2.9016e-01],
[-2.6602e-02, -7.2311e-01, 1.6537e-01, ..., -4.2675e-02,
1.1773e-01, -5.0714e-02],
[-1.5335e-01, -8.1843e-01, 3.1408e-01, ..., -1.7527e-01,
-6.9024e-02, 1.9007e-01],
...,
[ 1.1673e-01, -7.5069e-01, 2.7089e-01, ..., -1.4767e-01,
-1.0566e-01, 1.6638e-01],
[-2.3573e-01, -1.4247e+00, -2.4956e-03, ..., -4.0328e-01,
-3.1065e-01, -4.6695e-02],
[ 6.7163e-01, 1.2797e-01, 2.2843e-01, ..., -2.4024e-01,
-2.2127e-01, 3.2201e-01]],
[[-1.6062e+00, -6.0778e-01, 6.0195e-02, ..., 5.7271e-01,
3.7852e-01, 5.5187e-01],
[-2.0876e+00, -3.7940e-01, -7.6835e-02, ..., -7.3383e-02,
-1.4717e-01, 4.0189e-01],
[-1.7201e+00, -4.0803e-01, 1.4649e-01, ..., 3.0319e-02,
-1.5469e-01, 9.6361e-02],
...,
[-1.7682e+00, -4.8920e-01, 2.8922e-01, ..., 3.1893e-02,
-1.2095e-01, 1.0556e-01],
[-2.1609e+00, -1.1372e+00, -1.7820e-01, ..., -1.2474e-01,
-2.8679e-01, -1.9580e-01],
[-1.0335e+00, -1.7696e-01, 4.4108e-01, ..., -1.0232e-01,
-2.7891e-01, 1.1801e-01]]],
[[[-3.1277e-01, -4.5004e-01, -9.0415e-02, ..., -9.5903e-01,
-5.1017e-01, -1.4141e+00],
[-4.2887e-04, -1.6138e+00, -7.3373e-01, ..., -1.1462e+00,
-1.3052e+00, -1.4797e+00],
[-5.0172e-01, -1.7024e+00, -6.4360e-01, ..., -9.1762e-01,
-1.0868e+00, -1.8052e+00],
...,
[-2.6955e-01, -1.2811e+00, -4.9223e-01, ..., -9.0256e-01,
-1.0311e+00, -1.9096e+00],
[-3.1230e-01, -1.0122e+00, -6.0544e-02, ..., -4.0993e-01,
-5.2097e-01, -1.6897e+00],
[-2.3078e-01, -1.4391e+00, -1.0405e+00, ..., -1.1202e+00,
-1.5197e+00, -7.6231e-01]],
[[-5.9546e-01, -1.9919e+00, -7.0917e-01, ..., -4.0107e+00,
-5.4136e+00, -1.8530e+00],
[-2.4149e+00, -4.3843e+00, -2.0493e+00, ..., -5.9171e+00,
-8.0349e+00, -3.4231e+00],
[-2.8534e+00, -3.7435e+00, -1.8965e+00, ..., -5.0444e+00,
-6.2183e+00, -2.6157e+00],
...,
[-1.6585e+00, -2.8092e+00, -1.5362e+00, ..., -4.9883e+00,
-6.2853e+00, -2.4821e+00],
[-3.3781e+00, -4.2203e+00, -2.7067e+00, ..., -6.3186e+00,
-7.8295e+00, -2.5231e+00],
[ 6.4078e-01, -6.3780e-01, 4.9068e-01, ..., 4.2762e-01,
4.7359e-01, 2.2799e-01]],
[[-4.6955e-01, 6.5108e-01, 1.0540e+00, ..., 5.8298e-01,
2.9954e-01, 1.5483e+00],
[-8.5199e-01, 1.7320e-01, 5.5148e-01, ..., 5.0557e-01,
-9.2968e-02, 1.1570e-01],
[-1.3732e+00, 3.1555e-01, 2.1668e-01, ..., -3.4021e-03,
-5.0137e-02, -1.2540e-01],
...,
[-1.4165e+00, 2.8918e-01, 1.5703e-01, ..., -1.7689e-02,
-6.6403e-02, -1.6703e-01],
[-1.3935e+00, 3.2422e-01, 1.6413e-01, ..., -8.9682e-02,
1.5347e-03, -1.8958e-01],
[-8.8462e-01, 1.0334e-01, -8.9325e-02, ..., 4.4658e-02,
1.9987e-01, -5.8570e-01]],
...,
[[ 2.2008e-01, -2.6093e-01, -1.5971e-01, ..., 4.9240e-01,
4.0363e-01, -1.3065e+00],
[-1.6670e+00, -1.1420e+00, -6.9618e-01, ..., -1.6732e-01,
-7.7680e-01, -1.4075e+00],
[-9.5105e-01, -4.9860e-01, -1.7583e-01, ..., -9.8651e-02,
-1.1560e-01, -9.6161e-01],
...,
[-9.4423e-01, -5.4344e-01, -1.2025e-01, ..., -1.1453e-01,
-1.1481e-01, -9.4236e-01],
[-2.3304e+00, -1.5554e+00, -7.9243e-01, ..., -1.5004e+00,
-1.5326e+00, -1.7885e+00],
[-6.2119e-01, -9.8939e-01, -9.4143e-02, ..., -3.9381e-01,
-5.0756e-01, -1.1545e+00]],
[[-2.3767e-02, 2.6468e-01, 5.5331e-01, ..., -7.9204e-02,
4.0899e-01, -3.6433e-01],
[-3.2685e-01, -7.1342e-01, -7.1939e-02, ..., 2.3810e-01,
1.1898e-01, -8.3373e-01],
[-8.9552e-01, -1.4624e+00, -4.4443e-01, ..., 2.7533e-01,
1.5886e-01, -4.6022e-01],
...,
[-5.6495e-01, -1.1256e+00, -1.7443e-01, ..., 2.7168e-01,
1.5149e-01, -4.9659e-01],
[-9.3333e-01, -1.7922e+00, -3.4013e-01, ..., -1.0277e-01,
-2.3847e-01, -9.0025e-01],
[ 2.2747e-01, -4.9598e-01, -2.1746e-01, ..., 3.6084e-01,
3.3714e-01, 2.9138e-01]],
[[-5.0045e-01, 1.0663e-01, 5.4186e-01, ..., -1.4184e-01,
-7.1365e-02, 5.9054e-01],
[-3.9767e-01, 3.8322e-02, -2.9965e-02, ..., -9.2906e-01,
-4.1077e-01, 9.1032e-01],
[-3.8197e-02, -1.4476e-02, 5.9378e-02, ..., -5.9107e-01,
-5.0674e-01, 6.5193e-01],
...,
[ 3.9758e-02, -7.9014e-02, 7.9307e-02, ..., -5.3125e-01,
-5.0737e-01, 6.2354e-01],
[-2.7375e-01, -3.7997e-01, -7.2106e-02, ..., -6.7489e-01,
-6.6396e-01, 4.1570e-01],
[ 1.4585e-01, -4.0566e-01, -1.5425e-01, ..., -4.7526e-01,
-4.4071e-01, 9.9111e-01]]],
[[[-7.6129e-01, -1.4859e+00, -1.5877e+00, ..., -1.1677e+00,
-1.1656e+00, -1.9170e+00],
[-4.5076e-01, -2.5378e+00, -2.5075e+00, ..., -2.1968e+00,
-1.4766e+00, -1.5173e+00],
[-2.8821e-01, -2.0702e+00, -1.9741e+00, ..., -1.5976e+00,
-3.7742e-01, -1.6672e+00],
...,
[ 7.1278e-01, -5.1029e+00, -7.8848e+00, ..., 2.0549e-01,
-1.5333e-01, -9.7698e-01],
[ 9.3433e-01, -4.0612e+00, -5.7964e+00, ..., -3.4895e-01,
1.9370e+00, 7.1245e-01],
[ 5.3954e-01, -7.5253e-01, -2.5972e+00, ..., -9.1417e+00,
-4.4292e+00, -1.2532e+00]],
[[-2.7917e+00, -5.3193e+00, -4.7315e+00, ..., -7.5409e+00,
-7.3103e+00, -3.6992e+00],
[-5.7985e+00, -8.6417e+00, -7.8175e+00, ..., -1.2115e+01,
-1.1967e+01, -6.3598e+00],
[-4.5172e+00, -6.4142e+00, -6.1378e+00, ..., -1.0519e+01,
-1.0560e+01, -5.3044e+00],
...,
[-6.7400e+00, -1.1498e+01, -1.0096e+01, ..., -1.0032e+01,
-9.0367e+00, -4.0927e+00],
[-6.8294e+00, -9.1373e+00, -1.3813e+01, ..., -1.2711e+01,
-1.0679e+01, -5.7094e+00],
[ 9.5359e-01, 3.2565e-01, -3.5737e+00, ..., 2.9989e+00,
5.6796e-01, -7.3381e-01]],
[[-7.3578e-01, 4.4329e-01, 8.9083e-01, ..., 6.2324e-01,
8.2937e-01, 2.1501e+00],
[-9.0957e-01, 1.3706e-01, 3.9925e-01, ..., -2.5688e+00,
-1.5121e-01, 3.9288e-01],
[-1.8073e+00, -1.3875e-01, -1.0619e-01, ..., -4.1849e+00,
-1.5535e+00, 4.2843e-01],
...,
[-1.0684e+00, -4.1272e+00, -6.2497e+00, ..., -2.2282e+00,
1.4709e+00, 9.0531e-01],
[-1.0780e+00, -4.4753e+00, -8.4965e+00, ..., -7.4327e+00,
-5.5211e-01, 1.5748e+00],
[ 6.3328e-01, -3.3103e+00, -1.1678e+01, ..., -1.0660e+01,
-5.7108e+00, 5.1374e-01]],
...,
[[ 1.3766e+00, -3.5937e-01, -7.6543e-02, ..., 1.7581e+00,
7.1436e-01, -1.8853e+00],
[-7.8288e-01, -1.0553e+00, -6.6408e-01, ..., -1.3129e+00,
-3.0882e+00, -2.8741e+00],
[-7.3245e-01, -9.6075e-01, -4.0111e-01, ..., -4.2536e+00,
-5.2645e+00, -3.3007e+00],
...,
[-3.2505e-01, -6.5946e-01, -3.6424e+00, ..., -5.3631e+00,
-1.9967e+00, -5.8771e-01],
[-5.1241e-01, -6.5484e-02, -8.1216e+00, ..., -1.8063e+01,
-1.0909e+01, -4.2367e+00],
[-1.3154e-01, -1.5899e+00, -3.1373e+00, ..., -7.0464e+00,
-7.7941e+00, -4.0705e+00]],
[[ 8.8723e-03, 3.1931e-02, 6.3106e-01, ..., -5.8011e-02,
5.6220e-02, -5.6039e-01],
[ 1.9632e-01, -8.4256e-01, -3.1773e-02, ..., -6.2521e-01,
2.4955e-01, -1.0499e+00],
[ 2.0362e-01, -7.1833e-01, 2.4707e-01, ..., -3.7048e-01,
2.3047e+00, 1.2548e-01],
...,
[ 4.1281e-01, -1.5858e+00, -6.6212e+00, ..., -1.3550e+00,
-2.0813e+00, -1.9270e+00],
[-1.2871e-01, -2.7094e+00, -5.0989e+00, ..., -1.3060e+00,
-8.2592e-01, -1.7831e+00],
[ 4.1094e-01, -2.1069e-01, -4.4709e+00, ..., -4.6915e+00,
-4.8338e-01, 4.8765e-01]],
[[-1.8393e+00, -1.0032e+00, -5.6830e-01, ..., -1.2722e-01,
5.2524e-01, 1.1785e+00],
[-2.4648e+00, -8.6006e-01, -6.1684e-01, ..., -3.2471e+00,
1.0770e+00, 2.2636e+00],
[-1.8383e+00, -7.2363e-01, -2.9208e-01, ..., -7.6486e+00,
2.2254e-01, 1.8382e+00],
...,
[-1.2344e+00, -2.2365e+00, -3.4793e+00, ..., -2.8029e+00,
8.6031e-01, 1.0339e+00],
[-1.7715e+00, -3.2068e+00, -3.6274e+00, ..., -1.2007e+01,
-2.7807e+00, 5.9987e-02],
[-7.1587e-01, 3.3805e-02, 1.0675e-01, ..., -8.3718e+00,
-2.2103e+00, 4.8094e-01]]],
[[[-3.2318e-01, -4.6291e-01, -1.4449e-01, ..., -9.6311e-01,
-4.3269e-01, -1.3607e+00],
[-3.1813e-02, -1.3401e+00, -6.7547e-01, ..., -1.3198e+00,
-1.3008e+00, -1.4379e+00],
[-1.1876e-01, -1.2474e+00, -4.8693e-01, ..., -1.4493e+00,
-1.9843e+00, -2.5932e+00],
...,
[-3.1375e-01, -1.2967e+00, -4.8443e-01, ..., -1.2239e+00,
-1.2081e+00, -1.9743e+00],
[-5.0783e-01, -1.0664e+00, -7.9267e-02, ..., -3.8164e-01,
-5.0513e-01, -1.7367e+00],
[-6.2286e-01, -1.6122e+00, -1.2704e+00, ..., -9.6684e-01,
-1.3369e+00, -6.4930e-01]],
[[-7.3987e-01, -1.9714e+00, -8.3079e-01, ..., -3.8770e+00,
-5.4818e+00, -1.7830e+00],
[-2.1231e+00, -3.8510e+00, -2.0416e+00, ..., -6.3004e+00,
-8.3573e+00, -3.4807e+00],
[-1.6814e+00, -2.8057e+00, -1.5236e+00, ..., -7.2523e+00,
-8.7204e+00, -3.5586e+00],
...,
[-1.6448e+00, -2.6979e+00, -1.4562e+00, ..., -4.9058e+00,
-6.5818e+00, -2.7326e+00],
[-3.7744e+00, -4.5966e+00, -2.8635e+00, ..., -6.1340e+00,
-7.6165e+00, -2.3760e+00],
[ 5.4526e-01, -7.1114e-01, 3.8891e-01, ..., 3.9827e-01,
4.1202e-01, 3.0253e-01]],
[[-3.6395e-01, 7.3990e-01, 1.0043e+00, ..., 7.0371e-01,
2.8878e-01, 1.5995e+00],
[-7.5673e-01, 2.5020e-01, 5.4506e-01, ..., 6.4733e-01,
-1.2049e-01, 1.2690e-01],
[-1.2659e+00, 2.8779e-01, 1.6184e-01, ..., 4.6740e-01,
3.5052e-01, 1.4864e-01],
...,
[-1.4363e+00, 2.9102e-01, 1.4819e-01, ..., -2.6641e-01,
-2.3609e-02, -1.5484e-01],
[-1.3818e+00, 3.2377e-01, 2.4288e-01, ..., -2.7062e-01,
-1.0825e-01, -2.3739e-01],
[-8.8827e-01, 6.7878e-02, -1.6628e-01, ..., 9.0563e-02,
2.5420e-01, -5.2836e-01]],
...,
[[ 4.4463e-01, -7.2925e-02, -1.2672e-01, ..., 3.8842e-01,
2.6243e-01, -1.3179e+00],
[-1.3391e+00, -1.0506e+00, -6.3804e-01, ..., -2.3812e-01,
-8.9280e-01, -1.3977e+00],
[-7.4513e-01, -4.8135e-01, -9.9933e-02, ..., -2.2985e-01,
-3.4703e-01, -1.2316e+00],
...,
[-9.8594e-01, -6.0733e-01, -1.1381e-01, ..., 2.1953e-01,
-2.4248e-02, -9.4392e-01],
[-2.5495e+00, -1.7762e+00, -8.5005e-01, ..., -1.7267e+00,
-1.5650e+00, -1.7025e+00],
[-6.7154e-01, -1.1935e+00, -7.6377e-02, ..., -3.4855e-01,
-5.3905e-01, -1.1652e+00]],
[[ 2.0364e-02, 2.0862e-01, 4.7280e-01, ..., -9.3816e-02,
4.1135e-01, -3.7461e-01],
[-1.4300e-01, -6.9877e-01, -3.1038e-02, ..., 2.6541e-02,
1.1931e-02, -8.9939e-01],
[-5.2705e-01, -1.0046e+00, -1.5512e-01, ..., 3.9965e-01,
-2.0553e-01, -1.0084e+00],
...,
[-6.4000e-01, -1.1296e+00, -9.2092e-02, ..., 1.2791e-01,
1.8724e-01, -2.9306e-01],
[-1.1328e+00, -2.0130e+00, -4.4219e-01, ..., -2.5433e-01,
-3.3937e-01, -9.0249e-01],
[ 2.4381e-01, -4.6712e-01, -2.3655e-01, ..., 3.7580e-01,
3.8398e-01, 3.4050e-01]],
[[-3.5500e-01, 1.2150e-01, 5.2945e-01, ..., -1.9828e-02,
-2.8855e-02, 6.3267e-01],
[-4.2095e-01, -5.5638e-02, 1.9205e-02, ..., -9.8329e-01,
-4.0020e-01, 9.4298e-01],
[-1.3057e-02, 2.8843e-04, 5.2856e-02, ..., -1.3412e+00,
-1.1136e+00, 3.1327e-01],
...,
[-1.4840e-02, -7.1305e-02, 9.3563e-02, ..., -2.9204e-01,
-6.7206e-01, 4.8141e-01],
[-3.8378e-01, -4.9760e-01, -1.4162e-01, ..., -3.1542e-01,
-4.0596e-01, 5.5986e-01],
[-8.7135e-03, -5.5961e-01, -1.7483e-01, ..., -4.2546e-01,
-4.3273e-01, 1.0019e+00]]],
[[[-7.4612e-01, -1.4574e+00, -1.5217e+00, ..., -2.4745e-01,
1.7921e-01, -4.5352e-01],
[-3.3292e-01, -2.4627e+00, -2.2856e+00, ..., -6.3163e-01,
-4.8368e-01, -6.4061e-01],
[-4.3947e-01, -2.1628e+00, -3.2724e+00, ..., -3.8971e-01,
-3.8743e-01, -1.1040e+00],
...,
[ 4.2540e-01, -4.0792e+00, -5.5891e+00, ..., 2.0383e+00,
-8.4614e-01, -4.3630e+00],
[ 3.8386e-01, -1.2583e+00, -5.3288e+00, ..., 1.3826e+00,
1.3624e+00, -2.3757e+00],
[ 4.5432e-01, 5.5750e-01, -1.4259e+00, ..., -8.6094e-01,
-1.0125e+00, -6.4625e-01]],
[[-2.4861e+00, -5.0763e+00, -4.4349e+00, ..., -2.5000e+00,
-3.8310e+00, -1.4451e+00],
[-5.6669e+00, -8.5273e+00, -1.4470e+01, ..., -3.7315e+00,
-5.4516e+00, -2.3118e+00],
[-4.5933e+00, -6.5371e+00, -1.1664e+01, ..., -3.3521e+00,
-4.0826e+00, -1.9070e+00],
...,
[-8.2357e+00, -1.3379e+01, -1.6283e+01, ..., -3.7847e+00,
-7.1965e+00, -4.3243e+00],
[-7.2941e+00, -1.2624e+01, -1.4734e+01, ..., -4.4140e+00,
-6.4160e+00, -3.7411e+00],
[ 2.1696e-01, -3.5930e-01, -9.3491e-01, ..., 4.7219e-01,
6.8326e-01, 2.0048e-01]],
[[-6.6112e-01, 5.5339e-01, 1.1312e+00, ..., 8.0632e-01,
4.9465e-01, 1.3526e+00],
[-8.8462e-01, -9.9472e-02, -1.3722e+00, ..., 3.6974e-01,
2.3418e-01, 1.5171e-01],
[-1.5999e+00, -2.0143e-01, -3.5777e+00, ..., 1.0504e-01,
2.0919e-01, 7.5041e-02],
...,
[-2.0325e+00, -9.4925e+00, -9.9613e+00, ..., -2.5192e+00,
-3.7385e+00, -1.8966e+00],
[-9.8806e-01, -6.6372e+00, -7.2743e+00, ..., 1.1567e+00,
-3.9439e-01, -8.1656e-01],
[ 1.4183e-01, -1.7800e+00, -2.1691e+00, ..., 5.6801e-02,
2.3915e-01, -7.2878e-02]],
...,
[[ 1.1477e+00, -4.2731e-01, -2.5253e-01, ..., 4.2012e-01,
2.8042e-01, -8.2064e-01],
[-7.6807e-01, -9.7222e-01, -4.2418e-01, ..., -3.0972e-01,
-1.0003e+00, -9.7026e-01],
[-6.3439e-01, -9.5067e-01, 2.1598e+00, ..., -7.7231e-02,
-2.0670e-01, -6.0487e-01],
...,
[-3.4067e-01, -2.2767e+00, -7.2898e+00, ..., -7.2363e-01,
-1.1101e+00, -3.7183e+00],
[-1.5213e+00, 5.7847e-01, -2.8432e+00, ..., -2.8208e+00,
-3.3377e+00, -2.8144e+00],
[-1.1910e+00, -9.2330e-01, -1.7462e-01, ..., -1.0756e-01,
-2.6071e-01, -5.0878e-01]],
[[ 2.6601e-02, -8.1191e-03, 6.0704e-01, ..., 3.3951e-01,
8.0569e-01, 1.5238e-02],
[-3.5813e-03, -7.1428e-01, 1.0048e+00, ..., 4.0658e-01,
5.8093e-01, -3.4878e-01],
[-6.2067e-02, -7.5362e-01, 2.0018e-01, ..., 3.8691e-01,
4.1380e-01, -1.8022e-01],
...,
[-3.9706e-02, -4.9146e+00, -6.8064e+00, ..., -3.2297e-01,
-1.3375e+00, -1.3088e+00],
[-1.8882e-01, -3.5722e+00, -2.7568e+00, ..., 6.4355e-01,
-5.0513e-02, -1.7465e+00],
[ 1.0830e+00, 1.9808e-01, -6.8768e-01, ..., 2.9454e-01,
4.1232e-01, 3.6710e-01]],
[[-1.7541e+00, -7.1620e-01, 1.2884e-02, ..., 3.4043e-01,
2.9094e-01, 8.6694e-01],
[-2.3110e+00, -4.3241e-01, -5.6616e-01, ..., -4.1856e-01,
-1.1677e-01, 1.1554e+00],
[-1.9604e+00, -5.2762e-01, -2.8717e+00, ..., -6.5868e-02,
5.3303e-02, 9.1791e-01],
...,
[-2.6429e+00, -2.0984e-01, -3.8048e+00, ..., 1.4881e+00,
1.5953e+00, 1.4651e+00],
[-3.2008e+00, -1.1047e+00, -2.6306e+00, ..., 9.9851e-02,
1.4471e+00, 2.2030e+00],
[-1.5680e+00, 7.4889e-01, 1.8374e+00, ..., -2.0121e-01,
-1.8115e-01, 7.5039e-01]]]], device='cuda:0'),
tensor([[[[-2.1764e-01, -2.7250e-01, 1.7756e-01, ..., 3.9310e-02,
4.8279e-01, 1.6065e-01],
[-1.3874e-01, -1.5294e+00, -8.7134e-01, ..., -7.0744e-01,
-4.2070e-01, -3.1921e-01],
[-1.5249e-01, -1.3525e+00, -6.4802e-01, ..., -4.8585e-01,
-3.8418e-01, -6.9310e-01],
...,
[-2.7315e-01, -1.6994e+00, -1.2444e+00, ..., -9.4774e-01,
-9.4793e-01, -2.0269e+00],
[-1.1329e+00, -1.7738e+00, -8.1535e-01, ..., -5.3142e-01,
-4.7016e-01, -1.7875e+00],
[-1.3008e+00, -2.1819e+00, -1.6232e+00, ..., -1.4531e+00,
-1.5709e+00, -9.6601e-01]],
[[-2.1322e-01, -1.4509e+00, -6.4054e-01, ..., -6.5987e-01,
-1.3065e+00, -2.2790e-01],
[-2.0560e+00, -3.9956e+00, -2.4475e+00, ..., -2.3816e+00,
-3.3214e+00, -1.1578e+00],
[-1.8818e+00, -3.3436e+00, -2.5208e+00, ..., -2.6026e+00,
-3.0332e+00, -1.2811e+00],
...,
[-3.6796e+00, -5.4911e+00, -5.1359e+00, ..., -5.5005e+00,
-6.1541e+00, -2.9402e+00],
[-6.1851e+00, -7.0261e+00, -6.0679e+00, ..., -6.1894e+00,
-7.5331e+00, -2.7669e+00],
[-5.0481e-02, -9.2699e-01, 2.6926e-01, ..., 4.0545e-01,
6.7517e-01, 1.7861e-01]],
[[-5.2897e-01, -1.6461e-01, 3.5426e-01, ..., 3.6292e-01,
5.1889e-02, 8.8290e-01],
[-3.8055e-01, -4.5866e-01, 4.7913e-01, ..., 4.0403e-01,
2.7569e-01, 5.1859e-01],
[-8.8630e-01, -2.6237e-01, 1.5464e-01, ..., 4.1291e-02,
1.3959e-01, 4.2356e-01],
...,
[-1.3879e+00, -2.0406e-03, 1.5361e-02, ..., 4.0375e-02,
5.7263e-02, 2.9777e-01],
[-1.2243e+00, 1.3235e-01, -2.4352e-02, ..., -7.5642e-02,
3.0534e-03, 3.1909e-01],
[-1.1742e+00, -3.1981e-02, 1.0279e-01, ..., -3.2901e-02,
-1.1146e-01, -3.3148e-01]],
...,
[[ 9.0776e-02, -5.7864e-01, -3.7501e-01, ..., -6.1567e-02,
-2.4011e-01, -7.3157e-01],
[-8.4312e-01, -6.2413e-01, -5.4180e-01, ..., -2.7216e-01,
-9.9851e-01, -7.2192e-01],
[-5.1838e-01, -1.2495e-01, -4.3000e-02, ..., 8.3853e-02,
-1.8487e-01, -5.2807e-01],
...,
[-6.3286e-01, -5.9364e-01, -3.3270e-01, ..., -3.7082e-01,
-5.9012e-01, -1.1955e+00],
[-3.1838e+00, -2.9496e+00, -2.0254e+00, ..., -1.9081e+00,
-2.0507e+00, -2.0754e+00],
[-5.0827e-01, -1.6321e+00, -1.1117e+00, ..., -8.6734e-01,
-9.4657e-01, -1.3778e+00]],
[[-1.8527e-01, 3.0865e-02, 4.0312e-01, ..., 7.9220e-02,
5.2483e-01, 1.8105e-01],
[-4.3830e-01, -2.5331e-01, -2.9072e-05, ..., -6.0186e-02,
1.3350e-01, -7.6295e-02],
[-6.8707e-01, -5.7193e-01, -8.9555e-03, ..., 3.5707e-02,
2.2537e-02, -4.9165e-02],
...,
[-1.3690e-01, -1.0026e+00, 8.5867e-02, ..., 2.5028e-01,
1.7476e-01, -6.9687e-01],
[-5.4927e-01, -1.8629e+00, -1.9329e-01, ..., 7.3228e-02,
-5.0080e-03, -9.5957e-01],
[ 4.1756e-01, -1.6473e-01, 1.3217e-01, ..., 2.5892e-01,
3.3156e-01, 3.1619e-01]],
[[-6.7155e-01, 4.0895e-01, 5.6559e-01, ..., 4.9758e-01,
3.9760e-01, 6.1411e-01],
[-8.7308e-01, 2.5305e-01, -2.5420e-01, ..., -2.6175e-01,
-1.6057e-01, 6.7713e-01],
[-5.0065e-01, 3.0811e-01, 1.8522e-02, ..., 6.0630e-02,
4.6827e-02, 6.2724e-01],
...,
[-1.1315e+00, -9.7782e-01, -4.7693e-01, ..., -3.7528e-01,
-5.7345e-03, 8.5645e-01],
[-1.8532e+00, -1.8928e+00, -1.0258e+00, ..., -7.9843e-01,
-4.2147e-01, 4.6149e-01],
[-9.2440e-01, -9.8059e-01, -2.6427e-01, ..., -3.3171e-01,
-1.4819e-01, 1.1280e+00]]],
[[[-8.2590e-02, -5.0051e+00, -8.1293e+00, ..., -3.6581e+00,
-8.5477e-01, 1.7905e+00],
[ 1.7412e+00, -1.9269e-01, -5.6255e+00, ..., -4.7863e+00,
-2.8035e+00, -8.3580e-01],
[ 1.5393e+00, 1.1451e+00, -3.3018e+00, ..., -2.1473e+00,
-1.7032e+00, -1.4058e+00],
...,
[ 4.6290e-01, -2.3986e+00, -4.6937e+00, ..., -1.3215e+00,
-1.1775e+00, -2.3270e+00],
[-9.0821e-01, -1.5914e+00, -2.0286e+00, ..., -7.1620e-01,
-6.4450e-01, -1.9959e+00],
[-1.2098e+00, -2.0844e+00, -1.5480e+00, ..., -1.5655e+00,
-1.6848e+00, -1.0800e+00]],
[[-8.1609e+00, -1.7938e+01, -1.7375e+01, ..., -1.3884e+01,
-7.9498e+00, -4.2216e+00],
[-5.9028e+00, -1.3824e+01, -1.3946e+01, ..., -7.3577e+00,
-6.0308e+00, -3.7011e+00],
[-2.5331e+00, -5.6019e+00, -1.0476e+01, ..., -2.8195e+00,
-3.6341e+00, -2.4108e+00],
...,
[-5.9850e+00, -1.1353e+01, -1.4866e+01, ..., -5.9379e+00,
-6.7685e+00, -3.0809e+00],
[-6.4048e+00, -7.8130e+00, -8.5905e+00, ..., -6.7548e+00,
-8.0720e+00, -2.8458e+00],
[-2.8794e-01, -9.1436e-01, 2.3611e-01, ..., 3.4973e-01,
5.9415e-01, 2.8336e-01]],
[[-2.1525e+00, -1.1738e+01, -1.6821e+01, ..., -7.5306e+00,
5.2785e-01, 3.4536e+00],
[-8.5890e-01, -1.0922e+01, -1.6479e+01, ..., -6.3326e+00,
-4.7607e+00, 1.9267e+00],
[ 4.8777e-01, -2.5817e+00, -1.1330e+01, ..., -1.8006e+00,
-1.8145e+00, 1.5802e+00],
...,
[-6.3020e-01, -4.9150e+00, -1.3176e+01, ..., -2.6014e-02,
-1.5560e-01, 1.7685e-01],
[-3.6821e-01, -2.5954e+00, -5.5379e+00, ..., -7.4409e-02,
-6.1800e-02, 2.1435e-01],
[-4.9837e-01, 7.5543e-02, -3.6244e-01, ..., -7.0735e-02,
-5.0651e-02, -4.6162e-01]],
...,
[[ 1.6223e+00, -1.6006e+00, -6.2732e+00, ..., -6.4348e+00,
-7.9516e+00, -5.1205e+00],
[ 4.9412e-01, 3.1638e+00, -5.8150e-01, ..., -8.5901e+00,
-8.6766e+00, -4.1553e+00],
[-1.8323e+00, -3.2558e-01, -1.3295e+00, ..., -7.3129e+00,
-3.9341e+00, -1.7322e+00],
...,
[-3.0574e-01, -6.5356e-01, -2.6798e+00, ..., -3.3088e-01,
-5.5565e-01, -1.2096e+00],
[-3.2854e+00, -2.1274e+00, -2.9846e+00, ..., -2.0326e+00,
-2.0946e+00, -2.2198e+00],
[-9.2587e-01, -2.6068e+00, -3.1744e+00, ..., -8.4061e-01,
-9.3926e-01, -1.4852e+00]],
[[ 7.3574e-01, -3.4900e+00, -7.3488e+00, ..., -4.9657e+00,
-1.1451e+00, 5.8764e-01],
[ 1.5778e-01, -2.9001e+00, -6.0241e+00, ..., 3.4765e-01,
1.3621e+00, -8.4027e-01],
[ 4.2974e-01, -1.2672e+00, -4.1461e+00, ..., 9.3323e-01,
-6.3508e-01, -1.0478e+00],
...,
[ 8.8118e-01, -1.8199e+00, -4.2673e+00, ..., 1.8685e-01,
1.1475e-01, -7.4181e-01],
[ 2.7878e-01, -2.9901e+00, -3.7767e+00, ..., 6.8225e-02,
-5.1061e-02, -9.5482e-01],
[ 9.4397e-01, 2.4182e-01, -5.1763e-01, ..., 3.5957e-01,
4.0485e-01, 4.6675e-01]],
[[-1.5073e+00, -2.3181e+00, -1.1671e+01, ..., -6.4295e+00,
1.5474e-01, 1.6415e+00],
[-2.6385e+00, 8.0642e-01, -2.9165e+00, ..., -4.4489e+00,
1.8273e+00, 2.8232e+00],
[-2.2783e+00, -1.7799e-01, 6.3989e-01, ..., -2.0967e+00,
1.1095e+00, 1.4402e+00],
...,
[-2.1963e+00, -1.2668e+00, -1.7722e+00, ..., -5.1796e-01,
-1.6727e-01, 8.2909e-01],
[-2.9917e+00, -2.8853e+00, -2.5712e+00, ..., -9.7201e-01,
-6.2991e-01, 4.7563e-01],
[-1.5461e+00, -1.3941e+00, -1.0933e-01, ..., -2.7042e-01,
-1.9203e-01, 1.3305e+00]]],
[[[-3.7892e-01, -2.7938e+00, -4.5072e+00, ..., -7.6368e-02,
5.3536e-01, 4.8268e-01],
[-2.9927e-01, -5.7238e+00, -3.8884e+00, ..., -5.9902e-01,
-2.9905e-01, 4.4536e-02],
[-2.6046e-01, -5.5139e+00, -5.5950e+00, ..., -4.4806e-01,
-2.3876e-01, -5.6664e-02],
...,
[-4.9488e-01, -1.9329e+00, -1.7526e+00, ..., -3.5335e+00,
-2.2785e+00, -3.1361e-01],
[-1.0160e+00, -1.7273e+00, -1.1848e+00, ..., -1.2778e+00,
-1.3557e+00, -3.3875e-01],
[-8.3661e-01, -1.6720e+00, -1.2491e+00, ..., -1.1683e+00,
-1.3774e+00, -5.6419e-01]],
[[-5.9879e+00, -1.1623e+01, -1.1895e+01, ..., -6.5503e-01,
-1.1113e+00, -4.7867e-02],
[-7.9248e+00, -1.0774e+01, -1.0591e+01, ..., -2.1730e+00,
-2.8815e+00, -7.4339e-01],
[-6.8674e+00, -9.2923e+00, -1.0366e+01, ..., -1.8998e+00,
-1.9110e+00, -5.7956e-01],
...,
[-4.3993e+00, -6.1175e+00, -5.5987e+00, ..., -3.5579e+00,
-1.9185e+00, 2.5072e-01],
[-6.3884e+00, -7.4594e+00, -6.3382e+00, ..., -3.7482e+00,
-4.2664e+00, -1.0068e+00],
[-4.9744e-02, -8.7009e-01, 4.0329e-01, ..., 3.6854e-01,
3.2088e-01, 1.3231e-02]],
[[-7.1296e-01, -2.9094e-01, -3.4518e+00, ..., 8.0181e-01,
3.1363e-01, 1.0561e+00],
[-1.2042e+00, -3.0348e+00, -7.7424e+00, ..., 5.9014e-01,
2.2123e-01, 5.3072e-01],
[-1.0495e+00, -7.1418e+00, -1.1476e+01, ..., 2.4103e-01,
3.0350e-01, 5.6356e-01],
...,
[-1.7089e+00, -5.5752e-02, 8.9981e-02, ..., -5.1450e+00,
-1.2520e+00, -1.7038e+00],
[-1.5381e+00, 7.4190e-02, 6.3556e-02, ..., -1.0204e+00,
1.2126e+00, 8.0877e-02],
[-1.4262e+00, -5.0085e-01, -2.3906e-01, ..., -2.2602e-01,
-2.1899e-02, 2.9231e-01]],
...,
[[ 1.6771e+00, 9.2784e-01, -2.1780e-01, ..., -7.7616e-02,
-3.2969e-01, -8.6567e-01],
[ 4.3476e-01, -1.0778e+00, -6.8326e+00, ..., -6.5369e-01,
-1.2739e+00, -8.1719e-01],
[-4.1590e-01, -2.6280e+00, -7.6589e+00, ..., -2.8225e-01,
-3.3186e-01, -4.8233e-01],
...,
[-7.7160e-01, -1.2113e+00, -9.8905e-01, ..., -2.5533e+00,
-4.6723e+00, -3.2544e+00],
[-2.8251e+00, -2.8698e+00, -1.8626e+00, ..., -2.3765e+00,
-2.7488e+00, -1.6416e+00],
[-6.7536e-01, -1.6783e+00, -9.9823e-01, ..., -1.8461e-01,
-2.6637e-01, -4.0443e-01]],
[[ 8.2575e-01, -5.9115e-01, -4.1641e+00, ..., 2.5742e-01,
5.9355e-01, 2.6320e-01],
[ 5.7115e-01, -2.3238e+00, -6.2394e+00, ..., 9.5671e-02,
1.1269e-01, -6.9066e-02],
[ 1.1933e-01, -3.6098e+00, -7.0169e+00, ..., 3.0217e-02,
-1.9319e-03, 2.3562e-01],
...,
[ 4.1718e-02, -7.7594e-01, 3.1014e-01, ..., -1.0856e+00,
5.5880e-01, 5.7910e-01],
[-2.6138e-01, -1.4571e+00, 3.0559e-02, ..., -1.7111e+00,
-7.3298e-01, 2.8103e-01],
[ 6.6354e-01, 1.5233e-01, 3.0128e-01, ..., -3.7334e-01,
-5.1261e-01, 1.2650e-01]],
[[-1.8412e+00, -2.2455e+00, -2.9660e+00, ..., 5.1954e-01,
3.2669e-01, 5.1502e-01],
[-3.5380e+00, -3.0704e+00, -4.2899e+00, ..., -1.5296e-01,
-2.2210e-01, 3.6576e-01],
[-3.2705e+00, -1.9137e+00, -4.1354e+00, ..., -1.8285e-01,
-3.4480e-01, -4.8369e-02],
...,
[-1.7802e+00, -5.2418e-01, 2.5445e-01, ..., -1.9280e+00,
-3.2286e+00, 6.4299e-01],
[-2.2244e+00, -1.1564e+00, -1.9003e-01, ..., 1.2960e+00,
-2.3961e-01, 3.7461e-01],
[-1.0552e+00, -2.0275e-01, 4.4516e-01, ..., 5.7885e-02,
-1.8608e-01, 1.7179e-01]]],
[[[-1.1556e+00, -2.4681e+00, -1.5101e+00, ..., -2.6719e-01,
2.5409e-01, -3.9154e-01],
[-2.2809e+00, -2.8250e+00, -1.5967e+00, ..., -9.8082e-01,
-5.9850e-01, -8.1467e-01],
[-2.6378e+00, -7.9381e-01, -1.1784e+00, ..., -2.2315e+00,
-4.8742e-01, -1.0278e+00],
...,
[-5.0778e-01, -1.9828e+00, -1.9111e+00, ..., -3.6211e-01,
-7.8649e-01, -1.2407e+00],
[-1.0237e+00, -1.7536e+00, -1.2118e+00, ..., 3.1750e-04,
3.3124e-02, -1.0180e+00],
[-8.3773e-01, -1.6778e+00, -1.2737e+00, ..., -9.0469e-01,
-1.1252e+00, -6.6328e-01]],
[[-4.3154e+00, -6.0435e+00, -3.7283e+00, ..., -2.3199e+00,
-3.5982e+00, -1.3222e+00],
[-6.9381e+00, -1.0104e+01, -5.9619e+00, ..., -5.2861e+00,
-7.1910e+00, -2.9996e+00],
[-4.4929e+00, -6.4193e+00, -4.8499e+00, ..., -6.7810e+00,
-7.3622e+00, -2.4152e+00],
...,
[-4.4202e+00, -6.1389e+00, -5.5387e+00, ..., -3.4191e+00,
-4.2602e+00, -1.7993e+00],
[-6.4576e+00, -7.5403e+00, -6.4865e+00, ..., -3.9795e+00,
-5.5435e+00, -1.9936e+00],
[-4.5429e-02, -8.4256e-01, 3.6737e-01, ..., 4.3474e-01,
6.2059e-01, 1.7184e-01]],
[[-5.7764e-01, 5.0617e-01, 8.3061e-01, ..., 8.3542e-01,
5.7143e-01, 1.3421e+00],
[-1.8737e+00, -1.5800e+00, -1.4482e+00, ..., 3.0444e-01,
2.1151e-01, 1.6770e-01],
[-7.4487e+00, -6.6267e+00, -5.3230e+00, ..., -1.0698e+00,
-4.4720e-01, 2.3347e-01],
...,
[-1.7304e+00, -1.1183e-01, -1.5525e-01, ..., -9.2767e-01,
1.5658e-01, 6.6703e-02],
[-1.5093e+00, 6.3353e-02, 2.8283e-02, ..., -3.5899e-02,
1.7940e-01, 1.4776e-01],
[-1.4353e+00, -5.7051e-01, -2.9775e-01, ..., 1.3151e-02,
1.9189e-01, -1.4414e-01]],
...,
[[ 1.5231e+00, 3.6893e-01, -8.8497e-02, ..., 3.5703e-01,
2.4268e-01, -7.8187e-01],
[-1.1633e-01, -2.8679e+00, -4.3178e+00, ..., -6.8392e-02,
-6.9160e-01, -6.6131e-01],
[-1.9150e+00, -5.8600e+00, -6.0599e+00, ..., -7.1407e-03,
-7.3574e-01, -1.6114e+00],
...,
[-7.3221e-01, -1.1702e+00, -7.9010e-01, ..., -1.0454e+00,
-6.3682e-01, -6.2955e-01],
[-2.8042e+00, -2.9003e+00, -1.9056e+00, ..., -1.2491e+00,
-1.1716e+00, -1.1240e+00],
[-6.5760e-01, -1.6925e+00, -1.0057e+00, ..., -1.9928e-01,
-2.3431e-01, -4.9799e-01]],
[[ 2.8421e-01, -5.3186e-01, -5.9394e-03, ..., 3.0517e-01,
8.2774e-01, 5.1807e-02],
[-1.1901e+00, -2.6316e+00, -1.7772e+00, ..., 4.4273e-01,
4.2643e-01, -9.8195e-01],
[-3.9888e+00, -4.6461e+00, -4.8879e+00, ..., -1.5278e+00,
-4.3839e-01, -2.1924e-01],
...,
[ 5.0702e-02, -8.0608e-01, 2.8025e-01, ..., 5.5545e-01,
9.0285e-02, -2.9734e-01],
[-2.3343e-01, -1.4656e+00, 4.3644e-03, ..., -7.0811e-02,
8.2377e-02, -4.9898e-01],
[ 6.6819e-01, 1.5137e-01, 2.7098e-01, ..., 2.4405e-01,
3.0577e-01, 3.5689e-01]],
[[-1.9742e+00, -8.7056e-01, -2.0907e-01, ..., 3.2898e-01,
2.2391e-01, 8.3743e-01],
[-3.9829e+00, -1.8466e+00, -2.1736e+00, ..., -2.4929e-01,
-5.3182e-02, 1.0501e+00],
[-2.5983e+00, -4.5066e+00, -3.8020e+00, ..., -1.5565e+00,
-1.7958e+00, 2.8875e-01],
...,
[-1.8502e+00, -5.0779e-01, 2.4276e-01, ..., 2.5719e-01,
5.9939e-01, 1.0273e+00],
[-2.2902e+00, -1.1771e+00, -1.7250e-01, ..., -6.3974e-02,
7.5483e-03, 6.2279e-01],
[-1.1071e+00, -2.2477e-01, 4.3666e-01, ..., -2.3885e-01,
-2.5177e-01, 7.2450e-01]]],
[[[-2.2795e-01, -2.9183e-01, 1.7401e-01, ..., -2.6409e-02,
2.7776e-01, -9.9696e-02],
[-1.2948e-01, -1.5471e+00, -8.8829e-01, ..., -2.0416e+00,
-9.9747e-01, -3.3135e-01],
[-1.2698e-01, -1.3675e+00, -6.5979e-01, ..., -4.5899e+00,
-1.6554e+00, -6.3369e-01],
...,
[-1.9839e+00, -7.6562e-01, 5.5184e-01, ..., -8.8695e-01,
-1.0447e+00, -2.0754e+00],
[-1.8140e+00, -3.2088e+00, 5.5770e-01, ..., -5.3353e-01,
-4.9112e-01, -1.8156e+00],
[-1.7700e+00, -1.1133e+00, 5.3472e-01, ..., -1.4400e+00,
-1.5823e+00, -9.8945e-01]],
[[-2.2673e-01, -1.5078e+00, -6.9122e-01, ..., -6.8616e+00,
-5.5540e+00, -1.6976e+00],
[-2.0566e+00, -4.0708e+00, -2.4581e+00, ..., -5.5608e+00,
-6.2494e+00, -3.0814e+00],
[-1.9265e+00, -3.4350e+00, -2.6186e+00, ..., -5.3805e+00,
-6.2788e+00, -3.2990e+00],
...,
[-8.0433e+00, -1.0890e+01, -1.1388e+01, ..., -5.6532e+00,
-6.3254e+00, -2.9010e+00],
[-1.0235e+01, -1.0713e+01, -6.7013e+00, ..., -6.3243e+00,
-7.5477e+00, -2.7163e+00],
[ 1.9420e-01, -9.5762e-01, 3.6631e-01, ..., 3.2691e-01,
6.8282e-01, 2.0069e-01]],
[[-5.1294e-01, -1.4367e-01, 3.7092e-01, ..., 9.2315e-01,
1.5138e+00, 1.2614e+00],
[-3.9859e-01, -4.5137e-01, 4.7996e-01, ..., -4.2181e+00,
1.8721e+00, 1.5410e+00],
[-8.9592e-01, -2.1418e-01, 1.9976e-01, ..., -7.1423e+00,
-4.2353e-01, 1.4108e+00],
...,
[-6.3085e+00, -7.6583e+00, -7.6812e+00, ..., -2.0155e-01,
3.3580e-02, 2.9781e-01],
[-4.6621e+00, -3.7253e+00, -1.6472e+00, ..., -4.5640e-02,
6.5388e-03, 2.7870e-01],
[-7.5526e-01, 1.6733e-02, 6.7017e-01, ..., -4.3626e-03,
-5.0601e-02, -3.7769e-01]],
...,
[[ 8.2030e-02, -5.7497e-01, -3.4859e-01, ..., -3.0971e+00,
-6.7838e-01, -6.3693e-01],
[-8.5659e-01, -6.2407e-01, -5.2142e-01, ..., -4.9064e+00,
-2.3966e+00, -1.1115e+00],
[-5.3005e-01, -1.3692e-01, -4.3523e-02, ..., -4.9916e+00,
-3.7366e+00, -1.0311e+00],
...,
[-2.5710e+00, -6.8056e+00, -6.9620e+00, ..., -6.3642e-01,
-6.6968e-01, -1.2176e+00],
[-3.0458e+00, -2.8376e+00, -4.1855e+00, ..., -1.8660e+00,
-2.0137e+00, -2.0659e+00],
[ 1.5494e+00, -2.4043e+00, -5.3515e+00, ..., -8.8554e-01,
-9.3717e-01, -1.3573e+00]],
[[-1.8545e-01, 2.5467e-02, 4.2995e-01, ..., -5.8871e-01,
-9.0512e-01, -8.0694e-01],
[-4.5651e-01, -2.8225e-01, 1.0000e-02, ..., -1.9027e+00,
-5.1357e-01, -4.9834e-01],
[-6.7059e-01, -5.8060e-01, 1.0605e-02, ..., -2.3046e+00,
-8.5232e-01, -6.9000e-01],
...,
[-3.9998e+00, -4.1087e+00, -3.6407e+00, ..., 3.6492e-01,
1.2431e-01, -7.2636e-01],
[-2.6063e+00, -3.1462e+00, -8.0079e-01, ..., 2.0532e-03,
-7.2104e-02, -9.5683e-01],
[-1.3170e+00, -2.4316e-01, 1.8531e+00, ..., 2.8368e-01,
2.8022e-01, 3.2593e-01]],
[[-6.5123e-01, 3.9779e-01, 5.3567e-01, ..., -2.0505e+00,
-2.9705e-02, 6.1225e-01],
[-8.4585e-01, 2.8115e-01, -2.2002e-01, ..., -3.5312e+00,
-6.8796e-01, 9.0347e-01],
[-4.8825e-01, 3.1119e-01, 5.3997e-02, ..., -3.1180e+00,
9.5425e-01, 8.4063e-01],
...,
[-3.5046e+00, -8.3412e+00, -6.1776e+00, ..., -1.2033e-01,
5.1118e-02, 8.3443e-01],
[-2.5301e+00, -7.6021e+00, -6.2631e+00, ..., -6.4679e-01,
-4.3930e-01, 4.4334e-01],
[-1.9837e+00, -3.3437e+00, -8.9359e-01, ..., -2.4482e-01,
-1.5236e-01, 1.1473e+00]]]], device='cuda:0'),
tensor([[[[-1.5505e-01, -5.2389e-01, 3.4637e-02, ..., -4.2337e-01,
7.3790e-01, -7.2168e-02],
[ 3.0019e-01, -1.2080e+00, -3.4273e-01, ..., -1.1052e+00,
-8.4362e-01, -2.1272e-01],
[ 2.4141e-01, -8.4301e-01, -2.2813e-01, ..., -2.3022e+00,
-1.1815e+00, -6.0557e-01],
...,
[-3.5121e-01, -1.1829e+00, -1.1597e+00, ..., -1.0755e+00,
-1.0601e+00, -2.1697e+00],
[-1.2973e+00, -1.4547e+00, -1.6694e-01, ..., -5.9119e-01,
-5.1853e-01, -1.8799e+00],
[-1.3413e+00, -2.1514e+00, -1.4654e+00, ..., -1.5019e+00,
-1.6014e+00, -1.0134e+00]],
[[-1.6770e+00, -3.3105e+00, -4.6388e+00, ..., -6.8260e+00,
-4.7415e+00, -1.3695e+00],
[-3.2395e+00, -5.1398e+00, -4.0294e+00, ..., -9.4273e+00,
-6.3692e+00, -2.2654e+00],
[-2.2272e+00, -3.2819e+00, -2.7851e+00, ..., -5.3234e+00,
-3.1653e+00, -1.3748e+00],
...,
[-3.9197e+00, -7.1954e+00, -1.0316e+01, ..., -5.6956e+00,
-6.4418e+00, -2.9843e+00],
[-6.3207e+00, -7.4566e+00, -6.2588e+00, ..., -6.3217e+00,
-7.6307e+00, -2.7950e+00],
[-1.0192e-01, -1.1278e+00, -2.1721e-01, ..., 3.9115e-01,
6.5889e-01, 1.9770e-01]],
[[-4.4883e-01, 3.2169e-01, 7.2612e-01, ..., -7.0143e-01,
1.0958e+00, 1.3674e+00],
[-4.3831e-01, 8.0201e-02, 1.3569e+00, ..., 1.8532e-01,
1.3800e+00, 3.1153e-01],
[-8.1620e-01, 9.4895e-02, 1.0573e+00, ..., 8.9022e-01,
7.9132e-01, 4.1836e-01],
...,
[-1.4613e+00, 7.9789e-01, -4.9394e+00, ..., 3.7227e-03,
-1.6853e-02, 2.6157e-01],
[-1.2379e+00, 1.5299e+00, -1.1614e+00, ..., -5.2603e-02,
-3.5504e-02, 2.3893e-01],
[-1.2489e+00, 4.4607e-01, 1.1332e+00, ..., -1.6564e-02,
-1.0496e-01, -3.9580e-01]],
...,
[[ 9.5189e-01, 2.6357e-01, 1.1053e+00, ..., -2.5983e+00,
-1.2398e+00, -9.9109e-01],
[-9.4340e-01, -8.5639e-01, -4.2542e-01, ..., -6.0311e+00,
-2.9425e+00, -1.2311e+00],
[-4.8247e-01, -3.3877e-01, -7.6247e-01, ..., -3.6267e+00,
-4.7428e-01, -6.7837e-01],
...,
[-6.2509e-01, 2.9284e-01, 1.7580e+00, ..., -3.6642e-01,
-6.0826e-01, -1.2175e+00],
[-3.2007e+00, -3.3180e+00, -1.8924e+00, ..., -1.9157e+00,
-2.0534e+00, -2.1019e+00],
[-5.4043e-01, -2.1454e+00, -2.1072e+00, ..., -8.7200e-01,
-8.9507e-01, -1.3695e+00]],
[[ 8.6660e-02, 2.4429e-01, 7.6511e-01, ..., 1.3046e-01,
1.5331e+00, 3.7167e-01],
[ 1.4246e-01, -4.4122e-01, 3.6756e-01, ..., 1.6101e-02,
4.7256e-01, -1.9395e-01],
[-1.5308e-01, -5.4484e-01, 4.9328e-01, ..., -1.9177e-01,
-3.9865e-01, -1.6887e-02],
...,
[-1.5555e-01, -1.1131e+00, -1.2558e+00, ..., 2.5822e-01,
1.2108e-01, -7.6270e-01],
[-5.3804e-01, -1.7508e+00, -1.1233e+00, ..., 5.4193e-02,
9.9810e-03, -9.5566e-01],
[ 4.5946e-01, 2.1068e-01, 1.1037e-01, ..., 3.2039e-01,
3.6438e-01, 3.8033e-01]],
[[-8.5701e-01, -2.4438e-01, -4.8667e-01, ..., -1.2069e-01,
-9.5220e-03, 7.9168e-01],
[-1.0798e+00, -3.5995e-01, -1.1700e+00, ..., 2.2922e+00,
8.6297e-01, 1.2547e+00],
[-7.0917e-01, -2.7852e-01, -6.8390e-01, ..., 2.1741e+00,
2.5771e-01, 7.1942e-01],
...,
[-1.2499e+00, -2.6021e+00, -1.2240e+00, ..., -4.9306e-01,
-9.3655e-02, 8.4525e-01],
[-2.0542e+00, -2.8794e+00, -1.8091e+00, ..., -9.1399e-01,
-4.9199e-01, 4.9471e-01],
[-1.0289e+00, -1.3730e+00, -3.0196e-01, ..., -2.9902e-01,
-1.5179e-01, 1.2260e+00]]],
[[[ 1.2932e-02, -2.5125e-01, -6.5050e-02, ..., -6.1801e-01,
3.7881e-01, -4.5332e-01],
[ 3.3628e-01, -1.1456e+00, -1.0307e+00, ..., -1.8776e+00,
-1.4399e-01, -2.8751e-01],
[ 3.5836e-01, -9.5531e-01, -1.2846e+00, ..., -3.1459e+00,
-1.0372e+00, -2.0035e-01],
...,
[-5.5006e-03, -6.0544e-01, -4.2156e-01, ..., -1.9361e+00,
-2.1028e+00, -1.7496e+00],
[-6.7260e-01, -7.0148e-01, -5.4689e-02, ..., -6.7553e-01,
-9.9319e-01, -1.5296e+00],
[-7.9744e-01, -1.3479e+00, -1.1739e+00, ..., -1.0317e+00,
-1.7101e+00, -9.4183e-01]],
[[-8.5520e-01, -2.2499e+00, -2.0385e+00, ..., -2.9371e+00,
-3.6515e+00, -1.4618e+00],
[-2.0952e+00, -4.0297e+00, -3.6118e+00, ..., -5.5417e+00,
-5.5052e+00, -1.8049e+00],
[-1.8838e+00, -3.1141e+00, -2.8884e+00, ..., -5.5812e+00,
-4.2456e+00, -3.3059e-01],
...,
[-1.7792e+00, -2.4369e+00, -2.5024e+00, ..., -6.1488e+00,
-6.8025e+00, -2.4307e+00],
[-3.8131e+00, -4.4202e+00, -3.6669e+00, ..., -7.0827e+00,
-8.1126e+00, -2.8047e+00],
[ 2.4690e-01, -3.5071e-01, 3.6770e-01, ..., 4.4716e-02,
1.8777e-01, 6.4037e-02]],
[[-2.0964e-01, 1.8138e-01, 7.5648e-01, ..., 1.0573e+00,
6.5336e-01, 1.9080e+00],
[ 9.9920e-02, -6.0374e-01, 1.7615e-01, ..., -8.3588e-01,
-1.0596e+00, -6.9591e-04],
[-4.0698e-02, -1.1877e+00, -1.9281e+00, ..., -1.1467e+00,
-1.7611e+00, -1.2875e+00],
...,
[-2.7013e-01, -2.3767e-01, 1.1358e-01, ..., -9.5307e-01,
3.8552e-01, 7.3186e-02],
[-2.7923e-01, -2.3759e-01, 7.7716e-02, ..., -8.6500e-01,
2.1706e-01, 1.8562e-01],
[-1.1497e-01, -4.8309e-01, 1.0250e-01, ..., -1.1792e+00,
-2.8494e-02, -2.7043e-01]],
...,
[[ 6.4514e-01, -2.6950e-01, 7.0475e-03, ..., -1.2328e-01,
-1.6468e-01, -1.3284e+00],
[-4.5656e-01, -4.0441e-01, -4.0383e-01, ..., -1.7060e+00,
-1.7756e+00, -1.9123e+00],
[-3.0267e-01, 1.8595e-02, -2.4360e-01, ..., -1.3413e+00,
-1.5611e+00, -1.8365e+00],
...,
[-4.3292e-01, -1.1156e-01, -1.6119e-01, ..., -9.4240e-01,
-1.2318e+00, -1.2523e+00],
[-2.3509e+00, -1.7041e+00, -1.2323e+00, ..., -2.2198e+00,
-2.2497e+00, -1.7470e+00],
[-4.8677e-01, -8.5218e-01, -3.9403e-01, ..., -8.5844e-01,
-8.0864e-01, -8.4070e-01]],
[[ 1.8217e-01, 2.9379e-01, 6.4342e-01, ..., 4.4330e-01,
1.5555e+00, 3.7083e-01],
[ 2.5652e-01, 3.1623e-01, 6.0680e-01, ..., -1.1934e+00,
1.0368e-01, -3.1221e-02],
[ 1.0583e-02, -3.5250e-02, -1.9082e-02, ..., -2.2594e+00,
-1.3923e+00, 1.9664e-01],
...,
[-2.0399e-01, 2.2256e-02, 3.5220e-01, ..., -1.7139e-01,
-3.6240e-01, -3.4797e-01],
[-5.8154e-01, -7.1596e-01, -1.7565e-02, ..., 1.1460e-01,
-2.0203e-01, -6.6096e-01],
[ 5.0660e-01, 3.8634e-01, 8.7162e-02, ..., 5.5053e-01,
5.6945e-02, 2.5414e-01]],
[[-8.5490e-01, 2.2487e-01, 3.0887e-01, ..., -2.8601e-01,
-2.2248e-01, 5.5915e-01],
[-1.1819e+00, 9.9614e-02, -9.9524e-01, ..., -9.0354e-01,
-1.5192e+00, 6.3444e-01],
[-1.0221e+00, 1.2606e-01, -1.9429e+00, ..., -1.2432e+00,
-2.0320e+00, 1.2769e-01],
...,
[-9.4571e-01, 1.7760e-01, -7.8304e-02, ..., 4.9659e-01,
9.6881e-01, 1.2895e+00],
[-1.7209e+00, -4.4599e-01, -3.7676e-01, ..., 8.0671e-01,
8.9957e-01, 9.0359e-01],
[-7.4728e-01, -1.0668e-01, -3.4076e-01, ..., 9.2959e-01,
5.5558e-01, 1.0756e+00]]],
[[[-3.2916e-01, -4.8877e-01, -1.2794e-01, ..., -9.5490e-01,
-5.3857e-01, -1.4644e+00],
[-1.4397e-02, -1.4080e+00, -6.7759e-01, ..., -1.1629e+00,
-1.2860e+00, -1.4840e+00],
[-3.1580e-01, -1.4625e+00, -6.3388e-01, ..., -9.1012e-01,
-1.0452e+00, -1.8027e+00],
...,
[-2.9812e-01, -1.3091e+00, -4.4454e-01, ..., -1.2306e+00,
-1.2121e+00, -1.9361e+00],
[-5.3788e-01, -1.1509e+00, -9.5330e-02, ..., -5.3131e-01,
-5.8664e-01, -1.7268e+00],
[-6.0553e-01, -1.6069e+00, -1.2362e+00, ..., -1.1305e+00,
-1.5688e+00, -7.8400e-01]],
[[-6.8084e-01, -1.9670e+00, -8.4260e-01, ..., -3.9408e+00,
-5.4510e+00, -1.8327e+00],
[-2.1697e+00, -3.9309e+00, -2.0845e+00, ..., -5.9464e+00,
-8.0661e+00, -3.3922e+00],
[-2.7834e+00, -3.6063e+00, -2.0085e+00, ..., -5.0335e+00,
-6.1452e+00, -2.5282e+00],
...,
[-1.6591e+00, -2.7878e+00, -1.5689e+00, ..., -5.4812e+00,
-6.4997e+00, -2.4324e+00],
[-3.7806e+00, -4.6033e+00, -2.8555e+00, ..., -6.5836e+00,
-8.0424e+00, -2.5725e+00],
[ 5.4940e-01, -7.2714e-01, 4.0669e-01, ..., 4.3713e-01,
4.2558e-01, 2.0164e-01]],
[[-3.7776e-01, 7.5219e-01, 1.0350e+00, ..., 6.6717e-01,
2.5681e-01, 1.4986e+00],
[-7.9636e-01, 2.4770e-01, 5.4286e-01, ..., 5.5701e-01,
-6.2980e-02, 7.7821e-02],
[-1.2538e+00, 4.2616e-01, 2.1630e-01, ..., 2.3829e-02,
-3.0263e-03, -1.3821e-01],
...,
[-1.4335e+00, 2.7579e-01, 1.3594e-01, ..., -4.5660e-01,
-2.4664e-02, -1.3926e-01],
[-1.3976e+00, 2.8882e-01, 1.6499e-01, ..., -2.2182e-01,
1.7926e-02, -1.9805e-01],
[-8.8310e-01, 9.2946e-02, -1.2865e-01, ..., 7.1026e-02,
1.5453e-01, -5.6546e-01]],
...,
[[ 4.2454e-01, -1.0445e-01, -1.3035e-01, ..., 4.3928e-01,
3.6712e-01, -1.2878e+00],
[-1.3908e+00, -1.0782e+00, -6.2513e-01, ..., -1.5687e-01,
-7.4710e-01, -1.3829e+00],
[-8.5547e-01, -6.6566e-01, -2.0636e-01, ..., -7.3140e-02,
-9.7575e-02, -9.4209e-01],
...,
[-9.7022e-01, -5.7321e-01, -9.5452e-02, ..., -8.2362e-01,
-2.7360e-01, -8.9286e-01],
[-2.5839e+00, -1.7819e+00, -8.5573e-01, ..., -1.4831e+00,
-1.5895e+00, -1.8493e+00],
[-6.3894e-01, -1.1748e+00, -9.1392e-02, ..., -3.2985e-01,
-5.3632e-01, -1.2360e+00]],
[[ 2.7011e-02, 2.4291e-01, 4.8788e-01, ..., -7.7788e-02,
3.7343e-01, -3.9042e-01],
[-1.4961e-01, -6.7521e-01, -2.1350e-02, ..., 2.0154e-01,
7.1605e-02, -8.6235e-01],
[-4.8791e-01, -1.1775e+00, -4.3177e-01, ..., 2.8200e-01,
1.5034e-01, -5.0404e-01],
...,
[-6.0327e-01, -1.1194e+00, -7.4571e-02, ..., 1.8715e-01,
-8.3410e-02, -5.2483e-01],
[-1.0900e+00, -2.0014e+00, -4.5466e-01, ..., -2.7566e-01,
-3.5021e-01, -9.4975e-01],
[ 2.5572e-01, -4.5854e-01, -2.3600e-01, ..., 3.4966e-01,
3.0188e-01, 2.7724e-01]],
[[-3.6287e-01, 1.2862e-01, 5.3811e-01, ..., -5.9739e-02,
-5.6044e-02, 5.5264e-01],
[-3.7770e-01, -4.5385e-02, 2.2905e-04, ..., -8.8727e-01,
-3.9709e-01, 8.7606e-01],
[-2.3148e-03, -2.9289e-02, 5.5280e-02, ..., -5.4963e-01,
-4.8712e-01, 6.1986e-01],
...,
[ 1.9528e-02, -7.9396e-02, 3.8417e-02, ..., 7.0270e-02,
-2.4368e-01, 6.4519e-01],
[-4.1106e-01, -5.3651e-01, -1.8418e-01, ..., -5.5786e-01,
-6.5567e-01, 4.3453e-01],
[ 2.2049e-02, -5.1230e-01, -1.7172e-01, ..., -5.3704e-01,
-4.7235e-01, 1.0272e+00]]],
[[[-7.3043e-01, -1.4686e+00, -1.4222e+00, ..., -1.2549e+00,
-8.5615e-01, -1.6583e+00],
[-3.5373e-01, -2.4471e+00, -2.2587e+00, ..., -1.7342e+00,
-1.8800e+00, -1.9039e+00],
[-1.3138e-01, -1.8954e+00, -1.6901e+00, ..., -1.2528e+00,
-1.4243e+00, -2.1064e+00],
...,
[ 7.6098e-02, -5.5721e-01, -2.0909e-01, ..., -8.6251e-02,
-1.1732e-02, -4.2480e-01],
[-3.6176e-01, -4.5792e-01, 1.9568e-02, ..., 1.3494e-01,
3.0121e-01, -1.3702e-01],
[-6.2995e-01, -1.3319e+00, -1.1038e+00, ..., -1.0903e+00,
-1.1837e+00, -5.0149e-01]],
[[-2.6244e+00, -5.0230e+00, -4.5700e+00, ..., -4.8100e+00,
-6.0297e+00, -2.4034e+00],
[-5.3093e+00, -8.1086e+00, -7.4423e+00, ..., -7.7606e+00,
-9.2570e+00, -4.3653e+00],
[-4.0274e+00, -5.8166e+00, -5.7127e+00, ..., -6.0293e+00,
-6.4976e+00, -3.2326e+00],
...,
[-1.7253e+00, -2.1556e+00, -1.6847e+00, ..., -1.8372e+00,
-2.2704e+00, -8.9731e-01],
[-2.7746e+00, -2.9575e+00, -1.9454e+00, ..., -2.2100e+00,
-3.3744e+00, -1.0023e+00],
[ 3.2462e-01, -2.7665e-01, 3.1305e-01, ..., 2.9212e-01,
4.4460e-01, 2.0371e-01]],
[[-6.2776e-01, 4.8474e-01, 8.7899e-01, ..., 8.4658e-01,
6.1631e-01, 1.9048e+00],
[-9.5840e-01, 1.6282e-02, 4.1800e-01, ..., 6.0871e-01,
2.8210e-02, 1.3194e-01],
[-1.7793e+00, -2.2048e-01, -2.2519e-02, ..., 1.3590e-02,
3.8799e-02, -7.7216e-02],
...,
[-6.4805e-01, 8.9061e-02, 2.5801e-01, ..., 1.7953e-01,
2.5739e-01, 4.5848e-01],
[-4.4075e-01, 9.3448e-02, 2.3312e-01, ..., 1.7153e-01,
2.3292e-01, 5.8659e-01],
[-1.6391e-01, -2.6762e-01, 1.4606e-01, ..., -6.5884e-03,
9.3647e-02, 3.4787e-01]],
...,
[[ 1.2529e+00, -3.5406e-01, -7.2389e-02, ..., 2.1510e-01,
-1.1007e-01, -2.1265e+00],
[-7.2300e-01, -9.0876e-01, -5.0198e-01, ..., -5.4476e-01,
-1.5030e+00, -2.0606e+00],
[-7.0239e-01, -8.6573e-01, -2.7788e-01, ..., -3.3262e-01,
-6.0033e-01, -1.3842e+00],
...,
[-6.5915e-01, -5.0430e-01, -3.3461e-01, ..., -3.2570e-01,
-4.7671e-01, -5.4130e-01],
[-2.2533e+00, -1.7997e+00, -1.3338e+00, ..., -1.3310e+00,
-1.3741e+00, -1.0985e+00],
[-6.2324e-01, -8.3091e-01, -4.1939e-01, ..., -2.6058e-01,
-3.5846e-01, -3.9492e-01]],
[[ 5.5115e-02, 4.2577e-02, 4.6660e-01, ..., 4.3501e-01,
9.2414e-01, -1.0590e-02],
[ 2.2992e-01, -8.2020e-01, -6.2715e-02, ..., 4.5800e-01,
3.8137e-01, -8.3645e-01],
[ 1.3708e-01, -6.2085e-01, 2.9142e-01, ..., 6.6562e-01,
5.4144e-01, -3.6772e-01],
...,
[-2.3826e-01, -4.8217e-01, 2.1407e-01, ..., 3.5137e-01,
3.3203e-01, 4.0865e-02],
[-5.2034e-01, -8.9301e-01, 4.8001e-02, ..., 1.6122e-02,
-1.3750e-02, -1.6816e-01],
[ 3.1237e-01, -3.0283e-02, -1.9843e-01, ..., -1.4361e-01,
-1.0646e-01, 1.6557e-01]],
[[-1.5384e+00, -7.8037e-01, -4.2194e-01, ..., -4.3607e-01,
-2.8251e-01, 8.7982e-01],
[-2.1199e+00, -7.6542e-01, -5.5693e-01, ..., -7.9259e-01,
-2.0782e-01, 1.6604e+00],
[-1.4875e+00, -5.4351e-01, -1.9885e-01, ..., -3.4894e-01,
-1.2606e-01, 1.3814e+00],
...,
[-6.3489e-01, -2.1455e-01, -1.1124e-01, ..., -1.4275e-01,
2.7331e-02, 6.0894e-01],
[-1.1336e+00, -5.5550e-01, -3.2700e-01, ..., -3.4386e-01,
-1.9096e-01, 1.4451e-01],
[-3.5857e-01, 2.8394e-03, -1.7293e-01, ..., -2.4268e-01,
-2.2936e-01, 3.0612e-01]]],
[[[-3.3399e-01, -9.8981e-01, -8.2813e-01, ..., 6.7580e-03,
6.3006e-01, 5.3740e-01],
[ 8.0104e-02, -1.7331e+00, -1.2459e+00, ..., -4.7020e-01,
-2.4092e-01, 9.3938e-02],
[ 1.6637e-01, -1.2957e+00, -9.1676e-01, ..., -2.9806e-01,
-1.2720e-01, -1.0557e-01],
...,
[ 1.5407e-01, -1.4044e+00, -9.7778e-01, ..., -3.3108e-01,
-5.8261e-02, -1.2134e-01],
[-1.7503e-01, -1.0763e+00, -4.4482e-01, ..., 9.5643e-03,
2.7753e-01, 4.5836e-03],
[-3.9008e-01, -1.2855e+00, -1.1219e+00, ..., -1.0057e+00,
-1.0845e+00, -3.7064e-01]],
[[-2.1911e+00, -4.1202e+00, -3.8318e+00, ..., -4.2168e-01,
-1.0431e+00, -2.5401e-02],
[-3.7439e+00, -6.0239e+00, -5.4772e+00, ..., -1.7587e+00,
-2.6282e+00, -6.4789e-01],
[-3.1239e+00, -4.7434e+00, -4.6390e+00, ..., -1.3037e+00,
-1.4929e+00, -3.5824e-01],
...,
[-3.1578e+00, -4.7017e+00, -4.6078e+00, ..., -1.2826e+00,
-1.5529e+00, -3.3615e-01],
[-5.4843e+00, -6.6698e+00, -6.0591e+00, ..., -2.3989e+00,
-3.3518e+00, -9.4741e-01],
[ 6.8195e-01, -2.8366e-01, 4.9746e-01, ..., 3.3411e-01,
3.1893e-01, 8.3145e-02]],
[[-5.8632e-01, 3.9775e-01, 7.6261e-01, ..., 8.1175e-01,
2.8638e-01, 1.0789e+00],
[-8.5357e-01, 3.3374e-02, 5.4856e-01, ..., 3.9817e-01,
3.9906e-02, 4.6711e-01],
[-1.2062e+00, -2.8658e-02, -8.2676e-03, ..., 4.0068e-02,
8.4960e-02, 4.0042e-01],
...,
[-1.4423e+00, -6.4736e-02, -4.1114e-02, ..., 4.2037e-02,
7.1672e-02, 4.1072e-01],
[-1.3650e+00, 1.3709e-01, -5.5806e-02, ..., 1.4054e-01,
1.5107e-01, 5.9734e-01],
[-5.9528e-01, 2.5937e-01, 1.3393e-01, ..., -1.0412e-01,
4.2890e-02, 3.4131e-01]],
...,
[[ 7.6188e-01, 1.7680e-01, 3.7682e-01, ..., -1.0096e-01,
-3.8409e-01, -9.2061e-01],
[-6.3726e-01, -3.3698e-01, -3.1665e-02, ..., -6.8497e-01,
-1.2486e+00, -8.1328e-01],
[-4.4991e-01, -4.1333e-01, -4.8398e-02, ..., -1.0505e-01,
-1.8131e-01, -3.8042e-01],
...,
[-8.6865e-01, -5.5897e-01, -4.1204e-02, ..., -1.0702e-01,
-1.7675e-01, -3.5220e-01],
[-2.4652e+00, -1.9402e+00, -1.3682e+00, ..., -7.7774e-01,
-8.8261e-01, -8.0588e-01],
[-4.6526e-01, -1.1655e+00, -3.1788e-01, ..., 5.5974e-03,
-1.3224e-01, -3.3303e-01]],
[[ 3.3608e-02, -2.4160e-01, -6.1536e-02, ..., 3.4407e-01,
6.7765e-01, 2.8164e-01],
[ 1.6426e-01, -8.0466e-01, 2.9049e-02, ..., -5.6024e-03,
1.0279e-01, -6.9855e-02],
[-9.1554e-02, -8.3865e-01, 1.1516e-01, ..., -1.3162e-01,
-9.8294e-02, 1.7877e-01],
...,
[-1.6613e-01, -1.1179e+00, -2.0967e-02, ..., -1.6049e-01,
-1.3376e-01, 1.5499e-01],
[-6.8662e-01, -2.1445e+00, -3.8365e-01, ..., -3.7838e-01,
-3.2914e-01, -4.5605e-02],
[ 7.4066e-01, -1.2151e-01, 2.4164e-01, ..., -2.2326e-01,
-2.4856e-01, 2.9901e-01]],
[[-6.0704e-01, -1.8566e-01, 2.4253e-03, ..., 5.4592e-01,
3.6098e-01, 5.4602e-01],
[-1.1241e+00, -8.2834e-01, -9.0864e-01, ..., -6.6720e-02,
-1.6894e-01, 4.0268e-01],
[-7.0468e-01, -7.0998e-01, -5.3190e-01, ..., 5.4419e-02,
-1.6544e-01, 1.1754e-01],
...,
[-4.9631e-01, -6.9600e-01, -4.3712e-01, ..., 7.3639e-02,
-1.1028e-01, 1.0124e-01],
[-9.8926e-01, -1.1686e+00, -6.2130e-01, ..., -7.3391e-02,
-3.0507e-01, -2.1525e-01],
[-6.1111e-01, -9.8338e-01, -4.1585e-01, ..., -9.1107e-02,
-2.7815e-01, 1.0489e-01]]]], device='cuda:0'),
tensor([[[[-7.4095e-01, -1.4836e+00, -1.4443e+00, ..., -1.2467e+00,
-8.8616e-01, -1.6816e+00],
[-3.4546e-01, -2.4340e+00, -2.2576e+00, ..., -1.7176e+00,
-1.8777e+00, -1.9180e+00],
[-1.2260e-01, -1.9045e+00, -1.6938e+00, ..., -1.3005e+00,
-1.4705e+00, -2.1862e+00],
...,
[ 6.1335e-02, -5.6876e-01, -2.2359e-01, ..., -7.0776e-02,
-2.6238e-02, -4.0739e-01],
[-3.7849e-01, -4.8235e-01, 3.3922e-02, ..., 1.4950e-01,
2.7156e-01, -1.7383e-01],
[-6.4564e-01, -1.3484e+00, -1.1062e+00, ..., -1.0986e+00,
-1.1956e+00, -4.9975e-01]],
[[-2.6872e+00, -5.0703e+00, -4.5715e+00, ..., -4.8383e+00,
-6.0433e+00, -2.4354e+00],
[-5.2858e+00, -8.0381e+00, -7.4271e+00, ..., -7.7143e+00,
-9.2219e+00, -4.3696e+00],
[-4.0071e+00, -5.7327e+00, -5.6464e+00, ..., -6.2845e+00,
-6.8350e+00, -3.4135e+00],
...,
[-1.7113e+00, -2.1423e+00, -1.6754e+00, ..., -1.7568e+00,
-2.2186e+00, -9.2302e-01],
[-2.8175e+00, -3.0142e+00, -1.9661e+00, ..., -2.2282e+00,
-3.3473e+00, -9.9665e-01],
[ 3.3199e-01, -2.9773e-01, 3.1965e-01, ..., 3.1406e-01,
4.2769e-01, 1.9996e-01]],
[[-6.4797e-01, 4.9470e-01, 8.7017e-01, ..., 8.2659e-01,
6.1761e-01, 1.8881e+00],
[-9.6708e-01, 6.0702e-02, 4.1488e-01, ..., 5.6144e-01,
-1.5695e-02, 1.3332e-01],
[-1.7723e+00, -2.3651e-01, -3.6126e-02, ..., 8.9822e-02,
1.2044e-02, -1.0238e-01],
...,
[-6.3787e-01, 9.2617e-02, 2.6551e-01, ..., 1.9143e-01,
2.5133e-01, 4.6124e-01],
[-4.5457e-01, 7.7625e-02, 1.9960e-01, ..., 1.6056e-01,
1.4755e-01, 5.6909e-01],
[-1.8552e-01, -2.5448e-01, 1.4634e-01, ..., -2.5502e-02,
9.7822e-02, 3.5287e-01]],
...,
[[ 1.2518e+00, -3.7174e-01, -7.8894e-02, ..., 2.2775e-01,
-1.2324e-01, -2.1360e+00],
[-7.2332e-01, -9.1856e-01, -5.1593e-01, ..., -4.9927e-01,
-1.4724e+00, -2.0668e+00],
[-7.0671e-01, -8.9384e-01, -2.7915e-01, ..., -3.4318e-01,
-5.8702e-01, -1.3588e+00],
...,
[-6.5947e-01, -5.0041e-01, -3.3849e-01, ..., -3.1111e-01,
-5.1397e-01, -5.6990e-01],
[-2.2946e+00, -1.8224e+00, -1.3363e+00, ..., -1.3976e+00,
-1.3668e+00, -1.1111e+00],
[-6.0322e-01, -8.3254e-01, -4.5046e-01, ..., -2.4163e-01,
-3.4081e-01, -3.9910e-01]],
[[ 5.2045e-02, 2.5624e-02, 4.4361e-01, ..., 4.6632e-01,
9.0766e-01, -8.7032e-02],
[ 2.4386e-01, -8.3189e-01, -7.1471e-02, ..., 4.7297e-01,
4.0321e-01, -8.4330e-01],
[ 1.4402e-01, -6.4596e-01, 2.6940e-01, ..., 6.6098e-01,
4.9535e-01, -5.1985e-01],
...,
[-2.5461e-01, -4.8576e-01, 2.2708e-01, ..., 3.7964e-01,
3.4695e-01, 1.0130e-01],
[-5.3723e-01, -9.3056e-01, 3.7199e-02, ..., 3.4224e-02,
-7.7996e-02, -2.1098e-01],
[ 3.0228e-01, -3.4590e-02, -1.8389e-01, ..., -1.6613e-01,
-1.0599e-01, 1.6307e-01]],
[[-1.5288e+00, -7.9033e-01, -4.2412e-01, ..., -4.2790e-01,
-2.5605e-01, 8.5192e-01],
[-2.1032e+00, -7.8300e-01, -5.5063e-01, ..., -7.9591e-01,
-1.8199e-01, 1.6354e+00],
[-1.4659e+00, -5.5653e-01, -2.1736e-01, ..., -4.5550e-01,
-1.3412e-01, 1.3423e+00],
...,
[-6.5943e-01, -2.3366e-01, -1.2513e-01, ..., -1.1558e-01,
5.0618e-02, 6.0537e-01],
[-1.1481e+00, -5.7072e-01, -3.3087e-01, ..., -3.1584e-01,
-1.8014e-01, 1.5084e-01],
[-3.6748e-01, -9.8343e-03, -1.7043e-01, ..., -2.5971e-01,
-2.5636e-01, 3.1114e-01]]],
[[[-3.4180e-01, -4.8826e-01, -1.4552e-01, ..., -9.1698e-01,
-5.7184e-01, -1.5064e+00],
[-1.5333e-05, -1.5215e+00, -6.8540e-01, ..., -1.0924e+00,
-1.3236e+00, -1.5650e+00],
[-6.3518e-01, -2.1060e+00, -6.4885e-01, ..., -9.1298e-01,
-1.0805e+00, -1.8569e+00],
...,
[ 6.7405e-01, -6.3396e-01, -8.8725e-01, ..., -9.5716e-01,
-1.0014e+00, -2.8378e+00],
[-3.4832e-01, -9.4806e-01, 9.8839e-03, ..., 2.4788e-01,
-9.1795e-01, -2.3962e+00],
[-4.6010e-01, -1.5612e+00, -1.2472e+00, ..., -1.1856e+00,
-1.6633e+00, -7.6925e-01]],
[[-6.8267e-01, -2.0348e+00, -8.7021e-01, ..., -4.0864e+00,
-5.5203e+00, -1.8971e+00],
[-2.5375e+00, -4.5789e+00, -3.0783e+00, ..., -6.0293e+00,
-8.2209e+00, -3.5419e+00],
[-2.6006e+00, -4.1994e+00, -4.4778e+00, ..., -5.0691e+00,
-6.2515e+00, -2.6451e+00],
...,
[-1.7291e+00, -3.0651e+00, -1.9822e+00, ..., -8.6592e+00,
-9.0258e+00, -3.1236e+00],
[-3.7428e+00, -4.6059e+00, -2.9458e+00, ..., -7.0965e+00,
-8.4775e+00, -2.7029e+00],
[ 6.1475e-01, -6.2573e-01, 4.4551e-01, ..., 2.6725e-01,
3.7600e-01, 2.1602e-01]],
[[-3.9561e-01, 7.1704e-01, 1.0308e+00, ..., 5.9281e-01,
3.2537e-01, 1.5566e+00],
[-8.1804e-01, 1.3763e-01, 5.2083e-01, ..., 5.2112e-01,
-5.6330e-02, 1.3302e-01],
[-1.6630e+00, -3.3921e-01, -5.1562e-02, ..., -7.1082e-02,
-1.1015e-01, -1.5595e-01],
...,
[-1.4109e+00, -1.6027e-01, -5.7112e-01, ..., -3.6324e+00,
-2.6071e+00, 4.5998e-01],
[-1.3713e+00, 4.0966e-01, 1.9519e-01, ..., -5.8449e-01,
-3.0604e-01, -1.5427e-01],
[-8.8896e-01, 5.1958e-02, -1.2064e-01, ..., -2.5344e-01,
2.6459e-01, -5.3063e-01]],
...,
[[ 4.1013e-01, -1.1971e-01, -9.9577e-02, ..., 5.2840e-01,
3.4293e-01, -1.3728e+00],
[-1.3691e+00, -1.0978e+00, -7.3680e-01, ..., -1.1890e-01,
-7.3752e-01, -1.4405e+00],
[-4.2212e-01, -4.6674e-01, -6.8571e-01, ..., -1.2311e-01,
-9.6626e-02, -9.8486e-01],
...,
[-1.7369e+00, -1.1501e+00, 6.5100e-02, ..., -3.6918e+00,
-3.8454e+00, -1.6715e+00],
[-2.5289e+00, -1.8989e+00, -1.1395e+00, ..., -4.6586e+00,
-2.7046e+00, -1.6378e+00],
[-6.4198e-01, -1.0866e+00, -1.0160e-01, ..., -7.1686e-01,
-5.1521e-01, -1.1547e+00]],
[[ 2.4997e-02, 2.5668e-01, 5.0253e-01, ..., -8.0344e-02,
4.2400e-01, -3.9258e-01],
[-5.3120e-02, -5.1594e-01, -1.2978e-02, ..., 2.7894e-01,
1.9355e-01, -9.0529e-01],
[-6.3555e-01, -7.8096e-01, 8.0242e-01, ..., 3.1522e-01,
2.0849e-01, -5.1220e-01],
...,
[-3.5472e-01, -1.4700e+00, -5.7346e-01, ..., 8.6900e-02,
8.3042e-01, -7.2099e-01],
[-1.0301e+00, -1.9093e+00, -4.0895e-01, ..., -2.3171e-01,
-5.5421e-01, -1.3935e+00],
[ 2.4848e-01, -5.0311e-01, -2.5637e-01, ..., 1.1928e-01,
2.0357e-01, 2.7835e-01]],
[[-3.8109e-01, 1.1964e-01, 5.3845e-01, ..., -1.3815e-01,
-4.4520e-02, 6.1603e-01],
[-4.0741e-01, -8.4002e-02, -1.3833e-01, ..., -9.5951e-01,
-3.1168e-01, 9.4508e-01],
[-3.2429e-01, -1.0481e+00, -1.4414e+00, ..., -6.7338e-01,
-4.5614e-01, 6.4067e-01],
...,
[ 9.1315e-01, 1.2507e+00, 1.3761e+00, ..., -3.5469e+00,
1.5303e+00, 2.0559e+00],
[-2.9312e-01, -3.6039e-01, 1.1444e-01, ..., -3.4189e-01,
6.7992e-01, 1.1729e+00],
[ 3.3507e-02, -5.1034e-01, -1.7513e-01, ..., -1.5141e-01,
-2.3483e-01, 9.8745e-01]]],
[[[-1.6871e-01, -5.7186e-01, -1.9754e-01, ..., -9.8109e-02,
1.2500e+00, 6.1529e-01],
[ 2.7566e-01, -1.1944e+00, -6.9589e-01, ..., -5.9505e+00,
-3.6174e+00, -1.8376e+00],
[ 2.0341e-01, -8.3057e-01, -3.9122e-01, ..., -6.8399e+00,
-4.9195e+00, -1.7199e+00],
...,
[-3.2199e-01, -1.7485e+00, -1.3313e+00, ..., -1.2381e+00,
-1.0435e+00, -2.0947e+00],
[-1.1963e+00, -1.8206e+00, -8.5082e-01, ..., -5.4856e-01,
-4.6669e-01, -1.8263e+00],
[-1.3321e+00, -2.2450e+00, -1.6657e+00, ..., -1.4702e+00,
-1.5728e+00, -9.8290e-01]],
[[-1.5498e+00, -3.1405e+00, -2.7489e+00, ..., -1.2459e+00,
-2.9082e+00, -6.7954e-01],
[-3.1129e+00, -4.8861e+00, -3.7504e+00, ..., -9.8749e+00,
-8.4485e+00, -2.2193e+00],
[-2.1292e+00, -3.0905e+00, -2.4851e+00, ..., -8.0258e+00,
-5.5557e+00, -1.3080e+00],
...,
[-3.8075e+00, -5.6058e+00, -5.2208e+00, ..., -5.9072e+00,
-6.2921e+00, -2.9127e+00],
[-6.2330e+00, -7.1027e+00, -6.1306e+00, ..., -6.1704e+00,
-7.4635e+00, -2.6834e+00],
[-7.1543e-02, -9.3525e-01, 2.3540e-01, ..., 3.8106e-01,
6.7720e-01, 2.2043e-01]],
[[-4.0417e-01, 2.8503e-01, 8.0711e-01, ..., -9.8888e+00,
-3.9566e+00, 1.1257e+00],
[-3.5389e-01, -4.0139e-02, 6.4398e-01, ..., -1.1492e+01,
-6.2319e+00, -2.3945e+00],
[-7.4761e-01, 8.1105e-02, 3.5891e-01, ..., -1.1169e+01,
-4.9743e+00, -3.0700e+00],
...,
[-1.4294e+00, -2.6487e-03, -8.6054e-03, ..., 1.3079e-01,
9.6091e-03, 3.0605e-01],
[-1.2200e+00, 1.3330e-01, -3.1897e-02, ..., -1.4720e-02,
4.8918e-02, 3.0889e-01],
[-1.1984e+00, -7.7054e-02, 5.2991e-02, ..., -2.4458e-02,
-7.8675e-02, -3.7542e-01]],
...,
[[ 9.6922e-01, 2.8958e-01, 2.8860e-01, ..., -4.7163e+00,
-5.2530e+00, -4.8701e+00],
[-9.5268e-01, -8.6273e-01, -6.9569e-01, ..., -6.5291e+00,
-8.3062e+00, -5.5989e+00],
[-4.4883e-01, -3.5700e-01, -2.9712e-01, ..., -2.8255e+00,
-3.2374e+00, -4.1670e+00],
...,
[-6.1807e-01, -6.6614e-01, -3.1782e-01, ..., -4.2905e-01,
-5.5488e-01, -1.1868e+00],
[-3.1994e+00, -2.9904e+00, -2.0205e+00, ..., -1.9039e+00,
-2.0417e+00, -2.0910e+00],
[-5.4606e-01, -1.6621e+00, -1.0739e+00, ..., -8.0532e-01,
-9.1032e-01, -1.3493e+00]],
[[ 7.7732e-02, 2.8639e-01, 6.9278e-01, ..., -5.5930e+00,
-1.5387e+00, -3.2096e-01],
[ 1.0343e-01, -4.3147e-01, 1.8192e-01, ..., -7.5350e+00,
-3.8610e+00, -2.0801e+00],
[-2.0136e-01, -5.7056e-01, 2.0745e-01, ..., -6.1800e+00,
-2.5486e+00, -3.7870e-01],
...,
[-1.5522e-01, -1.0698e+00, 4.4069e-02, ..., 1.5258e-01,
8.5969e-02, -7.2443e-01],
[-5.2398e-01, -1.8524e+00, -1.8134e-01, ..., 5.9259e-02,
-1.3013e-02, -9.1225e-01],
[ 4.2232e-01, -1.5722e-01, 1.4834e-01, ..., 3.2621e-01,
3.6144e-01, 3.6135e-01]],
[[-8.8567e-01, -2.2201e-01, 2.4424e-02, ..., -8.7462e+00,
-6.3372e+00, -3.0952e-01],
[-1.0919e+00, -2.8488e-01, -4.1400e-01, ..., -8.0924e+00,
-6.4195e+00, 1.1854e+00],
[-6.8151e-01, -2.3226e-01, -2.6550e-01, ..., -3.9893e+00,
-4.6924e+00, 1.1078e+00],
...,
[-1.1670e+00, -9.9161e-01, -4.6620e-01, ..., -1.0103e-01,
-4.3753e-02, 8.2359e-01],
[-1.9334e+00, -1.9079e+00, -1.0200e+00, ..., -8.4890e-01,
-4.7638e-01, 4.5567e-01],
[-9.7016e-01, -9.8855e-01, -2.8416e-01, ..., -2.8766e-01,
-1.2889e-01, 1.1815e+00]]],
[[[ 3.8920e-02, -2.5550e-01, -9.1937e-03, ..., -1.2095e+00,
-6.6061e-01, -9.5559e-01],
[ 3.1479e-01, -1.1507e+00, -8.3524e-01, ..., -1.9902e+00,
-1.7295e+00, -1.5213e+00],
[ 2.2910e-01, -7.7230e-01, -6.5982e-01, ..., -2.1639e+00,
-1.6457e+00, -8.2985e-01],
...,
[ 2.5014e-02, 3.0888e-01, -1.5552e+00, ..., -4.4184e+00,
-4.4727e+00, -2.0476e+00],
[-5.1926e-01, 1.0774e-01, 4.8750e-01, ..., -1.3842e+00,
-1.8527e+00, -2.1099e+00],
[-6.1001e-01, -9.3838e-01, -5.9659e-01, ..., -1.2347e+00,
-1.4931e+00, -9.0831e-01]],
[[-7.7720e-01, -2.2456e+00, -1.8304e+00, ..., -4.2285e+00,
-4.8630e+00, -1.7990e+00],
[-2.5783e+00, -4.5282e+00, -3.4099e+00, ..., -7.3024e+00,
-8.2468e+00, -3.3979e+00],
[-2.2491e+00, -3.6767e+00, -3.2126e+00, ..., -5.5466e+00,
-4.9108e+00, -1.6965e+00],
...,
[-1.8056e+00, -4.1217e+00, -5.7464e+00, ..., -6.9469e+00,
-6.4599e+00, -2.3604e+00],
[-3.5823e+00, -3.9692e+00, -3.7046e+00, ..., -6.9941e+00,
-7.7946e+00, -3.0919e+00],
[ 2.7508e-01, -4.1472e-01, 4.9124e-01, ..., 5.4404e-01,
6.7337e-01, 6.3837e-02]],
[[-2.6872e-01, 5.5482e-02, 8.3370e-01, ..., 9.8710e-01,
6.9864e-01, 1.4899e+00],
[ 7.5329e-03, -5.6396e-01, 5.9191e-01, ..., 3.2560e-01,
6.1874e-01, 7.6417e-01],
[-4.4467e-01, -5.6480e-01, 1.5896e-01, ..., -8.1436e-01,
-4.0572e-01, 1.9309e-01],
...,
[-2.3942e-01, 2.6738e-01, -6.9178e+00, ..., -3.4740e+00,
7.8699e-02, -1.1234e+00],
[-2.0431e-01, 6.5490e-01, -3.1830e+00, ..., -3.0585e-01,
7.5488e-01, -3.5305e-01],
[-1.0198e-01, 2.7081e-01, -5.3504e-01, ..., -3.9793e-01,
-9.6571e-02, -2.6606e-01]],
...,
[[ 4.4401e-01, -4.1239e-01, -6.2832e-02, ..., -1.2695e-01,
-2.6980e-01, -1.4817e+00],
[-7.4414e-01, -5.0623e-01, -6.7839e-01, ..., -7.7552e-01,
-1.4375e+00, -1.7193e+00],
[-4.7114e-01, 1.2510e-03, -3.0499e-01, ..., -1.9864e-01,
-2.4446e-01, -1.6002e+00],
...,
[-4.2468e-01, 2.7293e-01, 5.6686e-01, ..., -5.4117e-01,
-2.7431e+00, -4.1314e+00],
[-2.1826e+00, -1.3580e+00, -1.5631e-01, ..., -3.6906e+00,
-3.4913e+00, -2.4502e+00],
[-4.9836e-01, -1.2957e+00, -7.3865e-01, ..., -7.0791e-01,
-7.2751e-01, -8.5270e-01]],
[[ 1.1868e-01, 2.7621e-01, 6.6452e-01, ..., 3.7497e-01,
6.5903e-01, -1.1016e-02],
[ 2.8655e-02, 2.5181e-01, 3.5939e-01, ..., 2.4120e-01,
3.1543e-01, -3.3707e-01],
[-2.8041e-01, -1.4257e-01, 6.6564e-02, ..., 8.5194e-04,
-1.2050e-01, 7.2932e-01],
...,
[-1.5686e-01, 9.9902e-01, -6.3161e-01, ..., -2.4464e+00,
-5.2698e-01, 8.0055e-01],
[-4.6283e-01, -1.4673e-01, -2.2971e-01, ..., -1.3881e+00,
-9.9691e-01, -9.6563e-01],
[ 5.2624e-01, 7.2992e-01, 6.3107e-02, ..., 2.3318e-01,
1.5182e-01, 3.1049e-01]],
[[-9.9591e-01, 2.4426e-01, 3.7112e-01, ..., 1.4676e-01,
1.8121e-01, 8.1366e-01],
[-1.2972e+00, 1.0999e-01, -5.0330e-01, ..., -1.8495e-01,
-6.4815e-02, 1.1162e+00],
[-9.9149e-01, 3.6598e-01, -1.4811e-01, ..., -6.8783e-01,
-8.1989e-01, 6.4796e-01],
...,
[-9.8871e-01, -1.9458e-01, 5.7071e-01, ..., -3.8264e-01,
-2.2594e+00, 8.3270e-01],
[-1.6034e+00, -7.9660e-01, 4.3341e-01, ..., 1.7886e+00,
1.3674e+00, 1.6486e+00],
[-6.4975e-01, -2.2960e-01, 2.1965e-01, ..., 2.4660e-01,
1.4816e-01, 1.0616e+00]]],
[[[-7.3796e-01, -1.4458e+00, -1.4253e+00, ..., 1.9056e-01,
5.6353e-01, 4.9642e-01],
[-2.8833e-01, -2.4589e+00, -1.8965e+00, ..., -1.7251e+00,
-2.1293e-01, 1.1447e-01],
[-3.9254e-01, -2.1486e+00, -1.5394e+00, ..., -1.9328e+00,
-1.6005e-01, -1.0325e-01],
...,
[-4.9013e-01, -1.9442e+00, -1.8836e+00, ..., -8.4596e-01,
-1.4951e+00, 1.9839e-02],
[-1.0309e+00, -1.7630e+00, -1.2149e+00, ..., -9.5113e-01,
-2.7565e-01, 2.4159e-01],
[-8.4698e-01, -1.7007e+00, -1.3197e+00, ..., -6.3285e-03,
-1.0319e+00, -6.7061e-02]],
[[-2.4068e+00, -4.8249e+00, -5.1244e+00, ..., -2.1244e+00,
-1.1420e+00, -7.3981e-02],
[-5.4349e+00, -8.2379e+00, -7.9781e+00, ..., -4.0742e+00,
-2.6592e+00, -6.5554e-01],
[-4.3884e+00, -6.4057e+00, -6.9916e+00, ..., -4.2116e+00,
-1.5923e+00, -4.0890e-01],
...,
[-4.4024e+00, -6.2061e+00, -5.7027e+00, ..., -3.2415e+00,
-4.4539e+00, -4.9741e+00],
[-6.4038e+00, -7.4604e+00, -6.3646e+00, ..., -4.8508e+00,
-5.7073e+00, -2.5649e+00],
[-6.5850e-02, -8.5756e-01, 4.0264e-01, ..., 1.0948e+00,
2.8424e-01, 3.2735e-02]],
[[-6.2136e-01, 5.9499e-01, 8.9118e-01, ..., 7.1049e-01,
3.2473e-01, 1.0705e+00],
[-9.1387e-01, -8.2972e-02, 3.3404e-01, ..., 1.1597e+00,
1.0125e-01, 4.8818e-01],
[-1.6300e+00, -1.9234e-01, 6.7364e-02, ..., 2.0647e+00,
1.2748e-01, 4.3792e-01],
...,
[-1.7436e+00, -8.1493e-02, -1.7691e-01, ..., -1.2062e-01,
-6.4292e+00, -4.1066e+00],
[-1.5376e+00, 7.6099e-02, 2.9319e-02, ..., -1.5069e+00,
-6.6863e+00, -4.5919e+00],
[-1.4066e+00, -4.9286e-01, -2.3824e-01, ..., -2.5000e+00,
-4.1345e+00, -2.7903e+00]],
...,
[[ 1.0511e+00, -4.5197e-01, 1.3917e-01, ..., -2.9370e+00,
-3.2145e-01, -8.6841e-01],
[-8.2817e-01, -9.7534e-01, -4.8056e-01, ..., -4.5187e+00,
-1.2122e+00, -7.8583e-01],
[-6.7502e-01, -9.0694e-01, -4.5442e-01, ..., -3.0259e+00,
-2.0112e-01, -3.9410e-01],
...,
[-7.5098e-01, -1.1279e+00, -6.7119e-01, ..., -3.5895e-01,
-3.2718e+00, -6.3039e+00],
[-2.8236e+00, -2.8890e+00, -1.8109e+00, ..., -2.9774e+00,
-5.1665e+00, -5.5090e+00],
[-6.6461e-01, -1.7181e+00, -1.0270e+00, ..., -1.2318e+00,
-1.6679e+00, -2.6422e+00]],
[[ 1.1311e-03, -3.4821e-02, 7.9486e-01, ..., 1.4775e+00,
6.3963e-01, 2.8916e-01],
[-2.9679e-02, -7.4685e-01, 2.6470e-01, ..., -1.1378e-02,
1.6432e-01, -4.0869e-02],
[-9.1106e-02, -7.8561e-01, 4.8341e-01, ..., -2.1810e+00,
-1.0143e-01, 1.6268e-01],
...,
[ 4.6021e-02, -7.8138e-01, 3.7322e-01, ..., -1.8469e+00,
-4.5071e+00, -1.5448e+00],
[-2.6604e-01, -1.4364e+00, 1.2366e-01, ..., -2.3290e+00,
-3.6811e+00, -1.4728e+00],
[ 6.3479e-01, 1.3646e-01, 2.8207e-01, ..., -4.5258e-01,
-1.6860e+00, 3.2458e-01]],
[[-1.6367e+00, -6.1066e-01, -2.9827e-01, ..., 3.6544e-01,
3.1707e-01, 5.1777e-01],
[-2.1063e+00, -3.1315e-01, -5.7598e-01, ..., -1.4612e-01,
-9.8287e-02, 4.0131e-01],
[-1.7622e+00, -4.1721e-01, -1.2140e-01, ..., -2.3488e-01,
-1.3844e-01, 8.3417e-02],
...,
[-1.7891e+00, -5.5884e-01, 1.9513e-01, ..., -3.1520e+00,
-3.2899e+00, -4.4135e+00],
[-2.2745e+00, -1.2512e+00, -2.9529e-01, ..., -2.1183e+00,
-3.0336e+00, -4.2273e+00],
[-1.0723e+00, -2.4196e-01, 3.7183e-01, ..., 4.2310e-01,
-1.4739e+00, -2.3420e+00]]]], device='cuda:0'),
tensor([[[[-1.9591e-01, -2.6392e-01, 1.8531e-01, ..., 1.1108e-01,
5.4153e-01, 1.6068e-01],
[-8.8010e-02, -1.5423e+00, -8.9786e-01, ..., -3.1558e-01,
-3.0156e-01, -3.2408e-01],
[-7.6515e-02, -1.3308e+00, -7.0285e-01, ..., 2.0449e-01,
-2.5976e-01, -6.9947e-01],
...,
[ 1.5446e-01, -1.7796e+00, -1.1868e+00, ..., -1.1020e-01,
-2.5773e-01, -1.1852e+00],
[-2.8728e-01, -1.1944e+00, -5.5059e-01, ..., 1.9107e-02,
9.7964e-02, -8.0796e-01],
[-2.6117e-01, -1.3073e+00, -1.2045e+00, ..., 6.2174e-02,
-2.6170e-01, -3.9204e-01]],
[[-2.0484e-01, -1.4700e+00, -6.1136e-01, ..., -6.0389e-01,
-1.3469e+00, -2.2427e-01],
[-2.0813e+00, -4.1348e+00, -2.4722e+00, ..., -2.3283e+00,
-2.9896e+00, -1.0973e+00],
[-1.9127e+00, -3.4588e+00, -2.6008e+00, ..., -3.2609e+00,
-3.2730e+00, -1.2968e+00],
...,
[-3.5386e+00, -5.1194e+00, -5.5921e+00, ..., -4.4616e+00,
-4.7252e+00, -2.2126e+00],
[-5.4554e+00, -6.5802e+00, -5.8438e+00, ..., -5.0659e+00,
-6.9458e+00, -2.9181e+00],
[ 4.9336e-01, -5.4727e-01, 4.9207e-01, ..., 9.5695e-01,
1.0580e+00, 9.4514e-02]],
[[-5.5279e-01, -1.7903e-01, 3.6665e-01, ..., 3.8883e-01,
3.5311e-02, 8.6980e-01],
[-3.7003e-01, -4.5432e-01, 5.0481e-01, ..., 5.2767e-01,
2.1743e-01, 4.6049e-01],
[-8.6905e-01, -2.6343e-01, 2.3366e-01, ..., -3.3650e-01,
1.1991e-01, 3.5100e-01],
...,
[-2.1419e+00, -6.0383e-01, 4.1120e-01, ..., -1.8383e+00,
-1.2528e+00, -6.8516e-01],
[-1.3456e+00, -4.0855e-01, 1.3714e-01, ..., -7.8743e-01,
-2.6065e-01, -2.6546e-01],
[-5.2776e-01, -4.4389e-02, -1.8496e-02, ..., 6.9352e-01,
3.5633e-01, -1.6675e-01]],
...,
[[ 5.3810e-02, -6.4040e-01, -3.9815e-01, ..., -2.7905e-02,
-1.9316e-01, -7.2432e-01],
[-8.3054e-01, -6.2135e-01, -5.7493e-01, ..., -1.3660e-01,
-8.0419e-01, -7.3770e-01],
[-4.5554e-01, -8.8540e-02, -4.2616e-02, ..., -1.3978e-02,
-3.6699e-02, -5.2833e-01],
...,
[-1.2920e-01, 2.1157e-01, 3.0093e-01, ..., -1.6137e+00,
-7.5591e-01, -1.1313e+00],
[-2.7520e+00, -2.3731e+00, -2.0289e+00, ..., -1.1573e+00,
-1.4634e+00, -1.7855e+00],
[-5.9191e-01, -1.2900e+00, -2.8063e-01, ..., 7.6562e-01,
-5.9961e-01, -7.6276e-01]],
[[-1.8594e-01, 1.9705e-02, 4.0840e-01, ..., 3.7773e-02,
4.9097e-01, 1.5573e-01],
[-4.2044e-01, -2.2412e-01, -2.5150e-02, ..., 8.1393e-02,
3.1527e-01, -4.7343e-02],
[-6.1761e-01, -4.9512e-01, -4.3938e-02, ..., 1.6811e-01,
9.6385e-02, -1.3691e-02],
...,
[-1.0835e-01, -8.3278e-01, 4.8411e-01, ..., 8.6373e-03,
-2.7520e-02, -3.6948e-01],
[-5.4317e-01, -1.6938e+00, -1.6007e-01, ..., -1.3826e+00,
-6.0532e-01, -2.4612e-01],
[ 7.8529e-01, 9.7203e-02, 3.0192e-01, ..., 6.1352e-01,
1.0624e+00, 8.4614e-01]],
[[-6.8205e-01, 4.1991e-01, 5.7116e-01, ..., 4.4438e-01,
3.4745e-01, 5.9182e-01],
[-8.4388e-01, 3.3940e-01, -2.1910e-01, ..., -4.3022e-01,
-2.7325e-01, 6.1463e-01],
[-5.0018e-01, 3.8681e-01, 6.2506e-02, ..., -6.7299e-01,
-4.8964e-02, 6.1881e-01],
...,
[-1.3701e+00, -9.2283e-01, -3.5990e-01, ..., -1.0583e+00,
-1.3666e+00, 1.1520e+00],
[-1.3785e+00, -7.2524e-01, -1.5944e-02, ..., -1.8675e+00,
-1.6804e+00, 7.4287e-01],
[-6.9533e-01, -5.3888e-01, 1.3859e-01, ..., -1.7333e+00,
-6.3791e-01, 1.2594e+00]]],
[[[-7.0960e-01, -1.4354e+00, -1.4306e+00, ..., -1.3158e+00,
-8.6694e-01, -1.6439e+00],
[-3.2033e-01, -2.4020e+00, -2.2334e+00, ..., -1.7460e+00,
-1.8666e+00, -1.8896e+00],
[-1.1506e-01, -1.8950e+00, -1.7297e+00, ..., -1.2767e+00,
-1.4586e+00, -2.1002e+00],
...,
[ 6.4057e-02, -5.6091e-01, -2.0204e-01, ..., -7.3625e-02,
-6.1912e-02, -4.2726e-01],
[-3.4168e-01, -4.5196e-01, 1.8501e-02, ..., 1.7801e-01,
2.7031e-01, -1.4426e-01],
[-6.0432e-01, -1.3404e+00, -1.1154e+00, ..., -1.0581e+00,
-1.2025e+00, -5.1593e-01]],
[[-2.7009e+00, -5.0965e+00, -4.5761e+00, ..., -4.8495e+00,
-6.0287e+00, -2.3777e+00],
[-5.3111e+00, -8.1077e+00, -7.4002e+00, ..., -7.8118e+00,
-9.2864e+00, -4.3583e+00],
[-4.0649e+00, -5.8865e+00, -5.8776e+00, ..., -6.0093e+00,
-6.4744e+00, -3.2031e+00],
...,
[-1.7037e+00, -2.1057e+00, -1.6846e+00, ..., -1.8551e+00,
-2.2090e+00, -8.7241e-01],
[-2.7412e+00, -2.9758e+00, -1.9627e+00, ..., -2.1593e+00,
-3.4152e+00, -1.0459e+00],
[ 3.1151e-01, -2.3847e-01, 3.3835e-01, ..., 3.3727e-01,
4.6626e-01, 1.7902e-01]],
[[-6.3878e-01, 4.8369e-01, 8.6161e-01, ..., 8.4380e-01,
6.4373e-01, 1.9064e+00],
[-9.5734e-01, 6.4951e-02, 4.3443e-01, ..., 5.3234e-01,
7.2732e-02, 1.1654e-01],
[-1.7226e+00, -1.7256e-01, 1.0819e-02, ..., -4.4587e-02,
6.8866e-02, -7.0279e-02],
...,
[-6.6236e-01, 8.1177e-02, 2.5396e-01, ..., 2.0781e-01,
2.4596e-01, 4.1401e-01],
[-4.3227e-01, 1.1940e-01, 2.3255e-01, ..., 1.7648e-01,
1.6523e-01, 5.1740e-01],
[-1.8174e-01, -2.8333e-01, 1.5034e-01, ..., 9.7688e-03,
5.3820e-02, 3.1368e-01]],
...,
[[ 1.2341e+00, -3.5513e-01, -5.0532e-02, ..., 2.1859e-01,
-1.1657e-01, -2.1110e+00],
[-6.9831e-01, -9.0347e-01, -5.0009e-01, ..., -5.3847e-01,
-1.5184e+00, -2.0602e+00],
[-6.8969e-01, -9.0474e-01, -3.0956e-01, ..., -3.0525e-01,
-6.2389e-01, -1.3911e+00],
...,
[-6.5818e-01, -4.9772e-01, -3.2329e-01, ..., -2.9467e-01,
-5.1022e-01, -5.6455e-01],
[-2.2002e+00, -1.8004e+00, -1.3276e+00, ..., -1.3040e+00,
-1.3747e+00, -1.1010e+00],
[-5.9630e-01, -7.5529e-01, -4.1723e-01, ..., -2.3794e-01,
-3.3841e-01, -3.9524e-01]],
[[ 7.4638e-02, 2.4708e-02, 4.3073e-01, ..., 4.2084e-01,
9.0351e-01, 2.4839e-04],
[ 2.7601e-01, -7.5633e-01, -2.6408e-02, ..., 4.6703e-01,
3.7171e-01, -8.5925e-01],
[ 2.4238e-01, -5.9574e-01, 2.7281e-01, ..., 6.9504e-01,
5.0661e-01, -3.9358e-01],
...,
[-2.4615e-01, -4.7215e-01, 2.5756e-01, ..., 3.5890e-01,
3.0777e-01, 5.5402e-02],
[-5.1848e-01, -8.6840e-01, 6.7193e-02, ..., 3.3060e-02,
-6.1725e-02, -2.0066e-01],
[ 3.0935e-01, -1.8798e-02, -1.8500e-01, ..., -1.1823e-01,
-1.4711e-01, 1.5119e-01]],
[[-1.5178e+00, -7.7361e-01, -4.1272e-01, ..., -4.0694e-01,
-2.8824e-01, 8.7037e-01],
[-2.1038e+00, -7.4148e-01, -5.6033e-01, ..., -7.2358e-01,
-1.6214e-01, 1.6661e+00],
[-1.4500e+00, -5.1204e-01, -2.4497e-01, ..., -3.4762e-01,
-9.7447e-02, 1.4006e+00],
...,
[-6.6813e-01, -2.0472e-01, -1.1528e-01, ..., -1.3482e-01,
-2.5760e-03, 6.0084e-01],
[-1.0847e+00, -4.8324e-01, -3.0037e-01, ..., -3.0417e-01,
-1.6247e-01, 1.3000e-01],
[-3.2438e-01, 3.0090e-02, -1.5607e-01, ..., -2.0521e-01,
-2.2292e-01, 2.7535e-01]]],
[[[-6.1090e-01, -1.4214e+00, -1.4503e+00, ..., -1.2303e+00,
-8.7366e-01, -1.7221e+00],
[-2.5737e-01, -2.6021e+00, -2.2816e+00, ..., -1.7147e+00,
-1.8381e+00, -1.9143e+00],
[-1.7757e-01, -2.0593e+00, -1.7169e+00, ..., -1.3674e+00,
-1.4095e+00, -2.1002e+00],
...,
[-7.4680e-02, -1.1222e+00, -6.0009e-01, ..., -5.4872e-01,
-4.5977e-01, -9.0136e-01],
[-1.4889e-01, -9.8808e-01, -2.7283e-01, ..., -2.3564e-01,
-7.4895e-02, -9.9186e-01],
[-1.6162e-01, -1.1096e+00, -7.8124e-01, ..., -8.6877e-01,
-1.0882e+00, -6.3180e-01]],
[[-2.4992e+00, -4.9299e+00, -4.7382e+00, ..., -4.8032e+00,
-6.0132e+00, -2.4667e+00],
[-5.5167e+00, -8.2396e+00, -7.4342e+00, ..., -7.6376e+00,
-9.0882e+00, -4.3251e+00],
[-4.0254e+00, -5.7528e+00, -5.6092e+00, ..., -6.6177e+00,
-6.4564e+00, -3.1827e+00],
...,
[-2.0284e+00, -3.1947e+00, -3.0247e+00, ..., -2.9311e+00,
-3.3796e+00, -1.4496e+00],
[-4.5411e+00, -5.4961e+00, -5.1483e+00, ..., -5.0633e+00,
-6.2344e+00, -2.3719e+00],
[ 6.8714e-01, -3.0449e-01, 5.8482e-01, ..., 4.9154e-01,
5.6374e-01, 4.4696e-04]],
[[-7.1822e-01, 3.3646e-01, 9.0560e-01, ..., 7.7023e-01,
6.1068e-01, 1.8590e+00],
[-9.7786e-01, -2.6559e-01, 4.2695e-01, ..., 5.3461e-01,
4.7392e-02, 1.3067e-01],
[-1.7165e+00, -4.2450e-01, 2.0486e-02, ..., 1.6504e-01,
9.6172e-03, -7.6632e-02],
...,
[-9.2361e-01, -1.0396e-01, 7.1588e-02, ..., 1.8366e-02,
7.3372e-02, 3.5735e-01],
[-9.5496e-01, 1.7635e-02, 9.3028e-02, ..., 1.5770e-02,
9.4725e-02, 4.2702e-01],
[-4.6764e-01, -5.2848e-02, 1.0417e-01, ..., -9.7146e-03,
9.4624e-02, 9.9132e-02]],
...,
[[ 1.0675e+00, -4.2918e-01, 6.0084e-02, ..., 2.3681e-01,
-4.7147e-02, -2.0771e+00],
[-9.5083e-01, -8.4837e-01, -4.1066e-01, ..., -5.0732e-01,
-1.4385e+00, -2.0452e+00],
[-7.9449e-01, -7.8786e-01, -2.7684e-01, ..., -1.7801e-01,
-6.0556e-01, -1.4016e+00],
...,
[-4.1545e-01, -1.2827e-01, 6.8991e-02, ..., 7.6079e-02,
-1.3212e-01, -5.2677e-01],
[-2.0618e+00, -1.3865e+00, -7.8544e-01, ..., -9.6783e-01,
-1.1455e+00, -1.0950e+00],
[-3.1872e-01, -7.8853e-01, -7.3796e-02, ..., -1.3574e-01,
-2.9527e-01, -6.1058e-01]],
[[-6.7147e-02, -2.3357e-02, 5.3320e-01, ..., 4.7089e-01,
8.8849e-01, -2.9342e-02],
[-5.9569e-02, -8.5431e-01, -2.0958e-03, ..., 4.4531e-01,
3.8216e-01, -7.8066e-01],
[-1.7063e-01, -7.5146e-01, 3.3255e-01, ..., 2.0077e-01,
5.3149e-01, -3.6681e-01],
...,
[-3.9329e-01, -5.1274e-01, 7.7759e-02, ..., 9.6279e-02,
6.9217e-02, -1.0123e-01],
[-7.4965e-01, -1.4863e+00, -2.7183e-01, ..., -3.7740e-01,
-3.4472e-01, -6.8878e-01],
[ 5.1179e-01, 7.5081e-02, 1.9518e-01, ..., 1.9371e-01,
1.6158e-01, 3.5684e-01]],
[[-1.6813e+00, -7.7433e-01, -4.6199e-01, ..., -4.8004e-01,
-3.1861e-01, 8.6536e-01],
[-2.2474e+00, -7.3469e-01, -7.2756e-01, ..., -8.0139e-01,
-2.1384e-01, 1.6481e+00],
[-1.5819e+00, -6.7092e-01, -3.3954e-01, ..., -1.5540e-01,
-1.2686e-01, 1.4224e+00],
...,
[-4.6303e-01, 1.4185e-01, 8.2194e-02, ..., 1.1995e-01,
1.9540e-01, 7.0858e-01],
[-9.4847e-01, -2.8707e-01, -1.9851e-01, ..., -1.1421e-01,
8.0210e-02, 4.2200e-01],
[-4.4299e-01, -3.2476e-01, -2.5413e-01, ..., -2.2380e-01,
-2.2365e-01, 6.0179e-01]]],
[[[-7.3585e-01, -1.4386e+00, -1.4307e+00, ..., -1.2335e+00,
-8.3594e-01, -1.6434e+00],
[-3.7584e-01, -2.3520e+00, -2.2342e+00, ..., -1.7366e+00,
-1.8634e+00, -1.8947e+00],
[-1.5982e-01, -1.8707e+00, -1.6900e+00, ..., -1.2226e+00,
-1.3921e+00, -2.0680e+00],
...,
[-2.0782e-01, -1.2766e+00, -5.1131e-01, ..., -5.3075e-01,
-4.4414e-01, -8.9961e-01],
[-5.5523e-01, -1.2824e+00, -1.8380e-01, ..., -1.9769e-01,
-7.9637e-02, -1.0077e+00],
[-5.5540e-01, -1.3507e+00, -7.6831e-01, ..., -8.7362e-01,
-1.0468e+00, -5.9333e-01]],
[[-2.7117e+00, -5.0950e+00, -4.5867e+00, ..., -4.7131e+00,
-5.9722e+00, -2.4016e+00],
[-5.2533e+00, -8.0563e+00, -7.4314e+00, ..., -7.6984e+00,
-9.1565e+00, -4.3512e+00],
[-3.9934e+00, -5.7542e+00, -5.7136e+00, ..., -5.9636e+00,
-6.4375e+00, -3.1924e+00],
...,
[-2.0743e+00, -3.3214e+00, -3.6458e+00, ..., -2.8707e+00,
-3.3412e+00, -1.4349e+00],
[-4.9620e+00, -6.1333e+00, -5.6866e+00, ..., -4.9205e+00,
-6.0067e+00, -2.2671e+00],
[ 5.1638e-01, -5.8115e-01, 5.3998e-01, ..., 4.5142e-01,
5.5360e-01, 2.0919e-02]],
[[-6.3920e-01, 4.9423e-01, 8.7014e-01, ..., 8.3645e-01,
6.1446e-01, 1.8900e+00],
[-9.3846e-01, 6.4374e-02, 4.2799e-01, ..., 5.5858e-01,
3.0371e-02, 1.2159e-01],
[-1.7653e+00, -2.0514e-01, -5.0806e-02, ..., 2.4519e-02,
6.3305e-02, -8.0885e-02],
...,
[-1.0486e+00, -1.3769e-01, 5.4689e-02, ..., 3.3638e-02,
7.8058e-02, 3.3609e-01],
[-1.0352e+00, -5.9356e-02, 5.5689e-02, ..., 1.2616e-02,
8.8756e-02, 3.9911e-01],
[-4.8949e-01, -5.7375e-02, 2.9789e-01, ..., -1.5301e-02,
1.0545e-01, 4.7939e-02]],
...,
[[ 1.2938e+00, -3.2256e-01, -6.3313e-02, ..., 2.1651e-01,
-1.0466e-01, -2.1204e+00],
[-6.8519e-01, -9.0869e-01, -4.8989e-01, ..., -5.5926e-01,
-1.4863e+00, -2.0469e+00],
[-7.2078e-01, -9.0926e-01, -2.8150e-01, ..., -3.1513e-01,
-6.0068e-01, -1.3807e+00],
...,
[-5.6690e-01, -1.0768e-01, 8.3773e-02, ..., 1.0272e-01,
-1.2044e-01, -5.3618e-01],
[-2.4655e+00, -1.6830e+00, -5.1128e-01, ..., -9.4101e-01,
-1.0601e+00, -1.0553e+00],
[-3.7512e-01, -1.0720e+00, -7.4463e-02, ..., -1.6356e-01,
-2.9744e-01, -5.7899e-01]],
[[ 6.3116e-02, 3.9579e-02, 4.3282e-01, ..., 4.8159e-01,
9.3584e-01, -4.0462e-02],
[ 3.0281e-01, -7.3994e-01, -4.5118e-02, ..., 4.3455e-01,
3.7403e-01, -8.2135e-01],
[ 2.2184e-01, -5.8212e-01, 2.8699e-01, ..., 6.6511e-01,
5.4161e-01, -3.6449e-01],
...,
[-5.4128e-01, -6.1767e-01, 2.3503e-01, ..., 8.5796e-02,
4.4354e-02, -7.3610e-02],
[-1.0060e+00, -1.7369e+00, -2.7893e-01, ..., -3.4311e-01,
-3.2611e-01, -6.4490e-01],
[ 4.9228e-01, 1.1708e-01, 3.7291e-01, ..., 1.6761e-01,
1.7025e-01, 3.6216e-01]],
[[-1.4973e+00, -7.5309e-01, -4.0946e-01, ..., -4.2724e-01,
-2.7324e-01, 8.7673e-01],
[-2.1567e+00, -7.8000e-01, -5.4588e-01, ..., -7.6414e-01,
-1.9483e-01, 1.6409e+00],
[-1.5367e+00, -5.6786e-01, -2.2391e-01, ..., -3.6553e-01,
-9.8298e-02, 1.4122e+00],
...,
[-4.7933e-01, 1.5760e-01, 7.8755e-02, ..., 1.3142e-01,
1.4863e-01, 7.4292e-01],
[-1.1054e+00, -4.5441e-01, -4.7182e-01, ..., -1.0086e-01,
2.2275e-02, 4.3263e-01],
[-5.6473e-01, -4.5374e-01, -4.2702e-01, ..., -2.0222e-01,
-2.4310e-01, 5.9990e-01]]],
[[[-9.1395e-01, -1.6155e+00, -7.6884e+00, ..., -1.9051e+00,
-2.0244e+00, 2.7723e-01],
[-5.4668e-01, -1.2678e+00, -5.0492e+00, ..., -8.0612e+00,
-4.0670e+00, -2.3494e+00],
[-2.7623e-01, -2.7209e+00, -8.9960e+00, ..., 9.0243e-01,
1.3957e+00, -1.5912e+00],
...,
[ 5.8136e-02, -2.3731e+00, -4.7501e+00, ..., -1.9658e+00,
-8.7250e-01, -1.3898e+00],
[-9.5206e-01, -1.5795e+00, -2.6409e+00, ..., -5.3212e-01,
-9.8288e-02, -4.0874e-01],
[-7.0448e-01, -1.2076e+00, -9.7265e-01, ..., -1.0334e+00,
-1.1841e+00, -4.8010e-01]],
[[-2.7755e+00, -1.3603e+01, -1.6584e+01, ..., -1.3443e+01,
-4.5067e+00, -1.4384e+00],
[-6.4153e+00, -1.9765e+01, -2.4336e+01, ..., -1.3412e+01,
-5.1775e+00, -3.4560e+00],
[-6.6091e+00, -1.7425e+01, -1.6228e+01, ..., -6.7165e+00,
-4.7325e+00, -1.8875e+00],
...,
[-7.0563e+00, -1.0326e+01, -1.5223e+01, ..., -5.1846e+00,
-4.5688e+00, -1.9891e+00],
[-6.7755e+00, -8.7534e+00, -1.0626e+01, ..., -4.2209e+00,
-4.7234e+00, -1.6773e+00],
[-2.6916e-01, -8.2807e-01, -5.3650e-01, ..., 4.8147e-01,
4.7194e-01, 8.1674e-02]],
[[-7.3155e-01, -2.9016e+00, -1.3228e+01, ..., -1.3312e+01,
-5.8973e+00, 9.9514e-01],
[-7.0207e-01, -2.5995e+00, -1.7356e+01, ..., -6.3310e+00,
-5.4906e+00, 9.2995e-01],
[-1.7818e+00, -3.4989e+00, -1.6912e+01, ..., 1.1810e+00,
5.7285e-01, 2.6944e+00],
...,
[-4.5313e-01, -3.1569e+00, -1.0288e+01, ..., 1.2715e-01,
-1.1171e+00, 1.0368e+00],
[-7.3513e-01, -1.5351e+00, -2.8543e+00, ..., 1.5280e-01,
1.7802e-01, 5.4551e-01],
[-8.4924e-01, -6.9400e-01, -5.8960e-01, ..., -4.2268e-02,
7.0603e-02, 1.7780e-01]],
...,
[[ 1.4711e+00, 6.0552e-01, -1.9159e+00, ..., -1.0010e+01,
-6.2879e+00, -4.3239e+00],
[-7.3446e-01, 1.4522e+00, 2.8175e+00, ..., -5.3305e+00,
-9.7480e+00, -5.9106e+00],
[-5.9896e-01, 3.1419e-02, 4.1961e+00, ..., -8.3185e+00,
-6.4927e+00, -2.7114e+00],
...,
[-1.0904e+00, -2.5507e+00, -4.5650e+00, ..., -3.9473e+00,
-2.7980e+00, -1.7552e+00],
[-3.1065e+00, -2.4345e+00, -1.6937e+00, ..., -1.0423e+00,
-1.1251e+00, -1.0642e+00],
[-1.3714e+00, -3.0366e+00, -2.3856e+00, ..., -2.8092e-01,
-3.5627e-01, -4.9632e-01]],
[[-8.6516e-03, -4.6814e-01, -3.2659e+00, ..., -6.3276e+00,
-5.1545e+00, -2.1932e+00],
[ 8.2722e-02, -1.2068e+00, -4.0925e+00, ..., -7.6552e+00,
-1.6963e+00, -1.0756e+00],
[ 3.7042e-01, -1.7557e+00, -6.1665e+00, ..., -9.2127e-01,
1.4298e+00, -5.3639e-01],
...,
[ 1.6544e+00, -2.0623e+00, -6.9448e+00, ..., -1.2869e+00,
-5.5324e-01, -6.4551e-01],
[ 7.9521e-01, -1.7257e+00, -2.8421e+00, ..., -5.3220e-01,
-4.2969e-01, -4.8016e-01],
[ 1.0886e+00, 4.4970e-01, -6.5229e-01, ..., 3.2850e-03,
-9.8872e-02, 1.9213e-01]],
[[-2.2469e+00, -2.0352e+00, -3.8929e+00, ..., -1.1681e+01,
-3.1337e+00, 2.5342e-01],
[-2.8513e+00, -1.5122e+00, 5.0197e-01, ..., -1.4473e+01,
-4.4376e+00, 1.2661e+00],
[-2.4199e+00, -4.2762e+00, -3.1217e+00, ..., -7.0954e+00,
-1.3404e+00, 6.0313e-01],
...,
[-3.6094e+00, -3.3982e-01, 7.0122e-01, ..., 3.4285e+00,
1.4153e+00, 1.1048e+00],
[-4.0575e+00, -1.8883e+00, -2.5941e+00, ..., 3.5811e-01,
-8.5124e-02, 2.0465e-01],
[-1.8559e+00, 1.8950e-01, 7.2984e-01, ..., -1.2438e-02,
-2.8202e-01, 3.9111e-01]]]], device='cuda:0'),
tensor([[[[-1.7099e+00, -2.2310e+00, -2.3153e+00, ..., -7.1530e-01,
2.2671e-01, 2.7315e-01],
[-1.3058e+00, -2.0892e+00, -3.1091e+00, ..., -1.5915e+00,
-4.1050e-01, -3.9150e-02],
[ 1.1113e+00, 4.3587e-02, -3.2967e+00, ..., -5.1609e-01,
1.2433e-01, -3.4261e-01],
...,
[-6.2074e-01, -2.1853e+00, -2.1474e+00, ..., -2.2913e+00,
-2.9158e-01, -3.8152e-01],
[-1.3014e+00, -2.0521e+00, -1.3626e+00, ..., -4.2297e-01,
1.4393e-01, -1.9083e-01],
[-9.5393e-01, -1.9488e+00, -1.4710e+00, ..., -1.0497e+00,
-1.1407e+00, -4.3119e-01]],
[[-6.1343e+00, -9.1376e+00, -6.5657e+00, ..., -4.0025e+00,
-2.4535e+00, -3.1764e-01],
[-1.0636e+01, -1.3804e+01, -9.8424e+00, ..., -9.0065e+00,
-6.4345e+00, -1.0574e+00],
[-8.9289e+00, -1.3423e+01, -1.4385e+01, ..., -1.0372e+01,
-7.7185e+00, -7.8008e-01],
...,
[-4.8998e+00, -6.8645e+00, -6.0377e+00, ..., -3.2816e+00,
-2.2426e+00, -7.2940e-01],
[-6.7414e+00, -8.0383e+00, -6.9449e+00, ..., -3.0807e+00,
-3.9538e+00, -1.2822e+00],
[-2.1887e-01, -1.0101e+00, 1.5342e-01, ..., 4.0506e-01,
4.4391e-01, 8.9217e-02]],
[[-5.8458e-01, 7.8309e-01, 1.3956e+00, ..., 1.8676e+00,
6.7264e-01, 1.2981e+00],
[-3.0742e+00, -6.4030e-01, 1.0333e+00, ..., 4.1819e+00,
1.8158e+00, 4.8938e-01],
[-6.4436e+00, -3.6604e+00, -9.5530e-01, ..., 3.4569e+00,
3.4968e+00, 4.6299e-01],
...,
[-1.8341e+00, -1.3686e-01, -1.3914e-01, ..., 1.0256e+00,
1.6931e-01, 4.5169e-01],
[-1.5242e+00, -3.3253e-02, -3.3172e-02, ..., 2.0892e-01,
2.1026e-01, 6.2603e-01],
[-1.5410e+00, -6.7087e-01, -4.3151e-01, ..., -1.4943e-01,
7.4047e-02, 3.4311e-01]],
...,
[[ 1.4601e+00, -3.9583e-02, -1.9030e-01, ..., 5.6986e-01,
-3.4297e-01, -9.8235e-01],
[ 6.2865e-01, -1.1494e-01, -1.2669e+00, ..., -2.3123e+00,
-2.1666e+00, -9.7545e-01],
[-1.4412e+00, -3.7483e+00, -2.9011e+00, ..., -5.2483e+00,
-2.0132e+00, -5.3510e-01],
...,
[-6.5861e-01, -1.2238e+00, -7.8073e-01, ..., -1.8225e+00,
-3.1903e-01, -5.3164e-01],
[-3.0277e+00, -3.3125e+00, -2.1402e+00, ..., -8.7647e-01,
-1.0032e+00, -9.0337e-01],
[-7.0720e-01, -1.8839e+00, -1.1798e+00, ..., -1.1680e-01,
-2.2903e-01, -3.8169e-01]],
[[ 4.8812e-02, -6.2580e-01, -6.5812e-01, ..., -1.0497e+00,
1.1188e-01, 3.3093e-01],
[-2.1078e-01, -6.2314e-01, -5.9192e-01, ..., -9.7546e-01,
-8.9321e-01, -9.7575e-02],
[-1.1170e+00, -5.2465e-01, 7.6829e-01, ..., 1.0247e+00,
-4.2075e-01, 1.6974e-03],
...,
[ 1.6964e-01, -9.4248e-01, 2.5653e-01, ..., -1.3590e+00,
-1.6560e-01, -2.2472e-02],
[-5.4652e-02, -1.3502e+00, 1.0517e-01, ..., -7.6541e-01,
-4.3922e-01, -2.7045e-01],
[ 6.9207e-01, 1.9908e-01, 2.7049e-01, ..., -1.6588e-01,
-2.1135e-01, 2.5646e-01]],
[[-2.0177e+00, -9.7798e-01, -9.4924e-02, ..., 5.2962e-01,
2.6501e-01, 7.1982e-01],
[-3.7575e+00, -1.7518e+00, -2.9934e-01, ..., -7.9093e-01,
-4.8895e-01, 6.5752e-01],
[-6.6865e+00, -4.5405e+00, -1.1771e+00, ..., 4.2560e-01,
-3.5776e-02, 2.6833e-01],
...,
[-2.1762e+00, -5.4335e-01, 1.1367e-01, ..., 1.4840e+00,
-5.2586e-02, 2.6254e-01],
[-2.8100e+00, -1.3957e+00, -3.5629e-01, ..., 4.2379e-02,
-1.8854e-01, -1.4699e-02],
[-1.2920e+00, -2.0295e-01, 3.6214e-01, ..., -1.1802e-01,
-2.9038e-01, 2.7547e-01]]],
[[[-8.2364e-01, -1.4423e+00, -1.4110e+00, ..., -1.1710e-02,
6.3346e-01, 5.6002e-01],
[-3.9486e-01, -2.1745e+00, -2.3784e+00, ..., -4.9420e-01,
-2.1510e-01, 1.1738e-01],
[-3.7664e-01, -1.5808e+00, -2.4933e+00, ..., -3.3801e-01,
-1.1390e-01, -8.1972e-02],
...,
[-8.6606e-01, -2.1376e+00, -1.0833e+00, ..., -3.2253e-01,
-7.2216e-02, -1.2461e-01],
[-1.2201e+00, -1.7687e+00, -1.0854e+00, ..., 1.0207e-02,
2.7031e-01, -2.7155e-02],
[-1.0760e+00, -1.9972e+00, -1.4499e+00, ..., -1.0345e+00,
-1.0633e+00, -3.8072e-01]],
[[-2.5837e+00, -4.9694e+00, -4.3765e+00, ..., -4.1932e-01,
-9.8484e-01, 7.1074e-03],
[-5.1532e+00, -8.4403e+00, -7.6912e+00, ..., -1.7683e+00,
-2.6158e+00, -6.4760e-01],
[-4.3263e+00, -7.3937e+00, -6.8219e+00, ..., -1.2882e+00,
-1.4781e+00, -3.7187e-01],
...,
[-4.5802e+00, -6.4899e+00, -6.1654e+00, ..., -1.2914e+00,
-1.5588e+00, -3.5720e-01],
[-6.8890e+00, -7.6684e+00, -6.4678e+00, ..., -2.4662e+00,
-3.3189e+00, -9.5639e-01],
[-6.9136e-02, -1.0728e+00, 4.2920e-01, ..., 3.4982e-01,
3.6065e-01, 9.1646e-02]],
[[-5.8228e-01, 7.3170e-01, 1.0414e+00, ..., 8.1287e-01,
2.9053e-01, 1.0897e+00],
[-9.9529e-01, 8.1142e-02, 3.3090e-01, ..., 4.2079e-01,
2.3303e-02, 4.5350e-01],
[-1.6855e+00, -2.1648e-01, -5.7533e-01, ..., 6.1991e-02,
1.1115e-01, 4.0958e-01],
...,
[-2.4750e+00, 1.2017e-01, 2.2785e-01, ..., 2.3947e-02,
7.4711e-02, 4.2803e-01],
[-6.9620e-01, 6.2115e-01, 7.0733e-01, ..., 1.1334e-01,
1.1724e-01, 6.0443e-01],
[-1.4147e+00, -5.7281e-01, -2.4866e-01, ..., -8.1470e-02,
8.1820e-02, 3.6133e-01]],
...,
[[ 1.2686e+00, -3.1320e-01, -2.1719e-01, ..., -9.9309e-02,
-3.7240e-01, -9.2543e-01],
[-6.1986e-01, -9.8653e-01, -7.8535e-01, ..., -6.8422e-01,
-1.2679e+00, -8.1619e-01],
[-6.6490e-01, -7.1848e-01, -7.5632e-02, ..., -9.4302e-02,
-1.7468e-01, -3.8094e-01],
...,
[-4.2362e-01, -2.1758e+00, -3.0150e+00, ..., -1.2224e-01,
-1.8433e-01, -3.6546e-01],
[-3.9127e+00, -4.3862e+00, -3.3655e+00, ..., -8.2957e-01,
-8.6231e-01, -7.8487e-01],
[-6.7747e-01, -1.7263e+00, -1.0099e+00, ..., -1.7090e-02,
-1.0822e-01, -3.1254e-01]],
[[ 1.1845e-01, 8.4119e-02, 5.8297e-01, ..., 3.3539e-01,
6.7487e-01, 3.1798e-01],
[ 1.3603e-01, -5.8549e-01, 1.7790e-01, ..., -2.1403e-02,
1.1656e-01, -4.2831e-02],
[ 1.9273e-02, -3.0185e-01, 5.6092e-02, ..., -1.5013e-01,
-7.3151e-02, 2.1295e-01],
...,
[-1.3689e+00, -2.5104e+00, -3.2374e-02, ..., -1.5499e-01,
-1.2081e-01, 1.5693e-01],
[-5.5685e-01, -2.2604e+00, -8.0434e-01, ..., -4.0795e-01,
-3.3106e-01, -5.8205e-02],
[ 5.6209e-01, 5.9268e-02, 1.8304e-01, ..., -2.3098e-01,
-2.0874e-01, 3.3322e-01]],
[[-1.5243e+00, -6.2794e-01, -3.4122e-02, ..., 5.6136e-01,
3.4486e-01, 5.4384e-01],
[-2.1293e+00, -3.9758e-01, -1.4325e-01, ..., -4.5797e-02,
-1.4463e-01, 4.1584e-01],
[-1.7614e+00, -5.4344e-01, -3.9441e-01, ..., 6.2567e-02,
-1.7693e-01, 9.2195e-02],
...,
[-1.4019e+00, -1.2275e+00, -1.2561e+00, ..., 8.2102e-02,
-7.9759e-02, 9.7373e-02],
[-2.5268e+00, -8.7156e-01, 3.8897e-01, ..., -1.1706e-01,
-2.7404e-01, -2.0208e-01],
[-1.0677e+00, -4.1869e-01, 3.2558e-01, ..., -1.0710e-01,
-2.7105e-01, 1.3056e-01]]],
[[[-7.1942e-01, -1.3914e+00, -1.3891e+00, ..., -1.0092e-02,
5.7785e-01, 4.7754e-01],
[-2.6871e-01, -2.4277e+00, -2.1464e+00, ..., -4.5281e-01,
-2.3723e-01, 1.0613e-01],
[-3.8552e-01, -2.1313e+00, -1.7656e+00, ..., -3.3971e-01,
-1.5477e-01, -1.1976e-01],
...,
[-4.6588e-01, -1.9332e+00, -1.8533e+00, ..., -3.2420e-01,
-7.1553e-02, -9.4193e-02],
[-9.2118e-01, -1.6656e+00, -1.1761e+00, ..., 6.8041e-02,
3.3616e-01, 5.2997e-03],
[-7.8603e-01, -1.6401e+00, -1.2637e+00, ..., -1.1904e+00,
-1.2827e+00, -5.0217e-01]],
[[-2.3275e+00, -4.7550e+00, -4.1098e+00, ..., -5.7271e-01,
-1.0755e+00, -2.2371e-02],
[-5.3193e+00, -8.0621e+00, -6.5111e+00, ..., -1.7493e+00,
-2.6325e+00, -6.6526e-01],
[-4.2852e+00, -6.2441e+00, -5.9059e+00, ..., -1.2789e+00,
-1.5443e+00, -3.8847e-01],
...,
[-4.3565e+00, -6.0955e+00, -5.5070e+00, ..., -1.2794e+00,
-1.5324e+00, -3.4825e-01],
[-6.3631e+00, -7.4121e+00, -6.3250e+00, ..., -2.7554e+00,
-3.7133e+00, -1.0769e+00],
[ 2.9839e-03, -7.9739e-01, 4.1629e-01, ..., 3.8042e-01,
3.8196e-01, 5.1958e-02]],
[[-6.1299e-01, 5.8038e-01, 1.0973e+00, ..., 7.6248e-01,
2.9655e-01, 1.0263e+00],
[-9.2317e-01, -4.4279e-02, 4.0286e-01, ..., 3.5828e-01,
5.6725e-02, 4.6600e-01],
[-1.6356e+00, -1.3510e-01, -4.7845e-02, ..., 2.9812e-02,
3.5675e-02, 3.7116e-01],
...,
[-1.7373e+00, -2.7263e-02, -8.4103e-02, ..., 5.1020e-02,
6.3801e-02, 4.3024e-01],
[-1.5570e+00, 1.1540e-01, 9.3754e-02, ..., 1.2596e-01,
8.5965e-02, 6.1589e-01],
[-1.4361e+00, -5.1967e-01, -2.4803e-01, ..., -9.4230e-02,
6.1855e-02, 3.9599e-01]],
...,
[[ 1.0061e+00, -4.2379e-01, -2.1480e-01, ..., -9.3735e-02,
-3.4816e-01, -8.6943e-01],
[-8.4989e-01, -9.8034e-01, -8.2585e-01, ..., -5.9208e-01,
-1.1563e+00, -7.9645e-01],
[-6.9199e-01, -9.6554e-01, -6.5247e-01, ..., -8.5772e-02,
-1.9444e-01, -4.0911e-01],
...,
[-7.5355e-01, -1.1723e+00, -7.5282e-01, ..., -8.3313e-02,
-1.9808e-01, -3.8317e-01],
[-2.7477e+00, -2.7980e+00, -1.8453e+00, ..., -9.5339e-01,
-1.0298e+00, -8.8494e-01],
[-6.7597e-01, -1.6510e+00, -9.9572e-01, ..., -1.5354e-02,
-1.6178e-01, -3.9226e-01]],
[[-4.5352e-03, -1.3503e-03, 5.8658e-01, ..., 2.7231e-01,
6.0579e-01, 2.4545e-01],
[-5.6956e-02, -7.4086e-01, 1.2290e-01, ..., -1.3323e-02,
9.4228e-02, -3.2058e-02],
[-1.2837e-01, -8.5169e-01, 4.4326e-01, ..., -1.4972e-01,
-8.3737e-02, 1.6897e-01],
...,
[ 7.2373e-02, -8.0550e-01, 2.6489e-01, ..., -1.4222e-01,
-1.2615e-01, 1.6657e-01],
[-2.3416e-01, -1.4416e+00, 8.9741e-03, ..., -4.7347e-01,
-4.1411e-01, -1.5766e-01],
[ 6.5228e-01, 1.1999e-01, 2.7547e-01, ..., -2.9584e-01,
-2.9071e-01, 2.8429e-01]],
[[-1.5865e+00, -6.0115e-01, 3.1915e-02, ..., 4.8087e-01,
2.9445e-01, 5.1899e-01],
[-2.0640e+00, -3.1864e-01, -1.2483e-02, ..., -4.3470e-02,
-1.9843e-01, 3.5239e-01],
[-1.6987e+00, -3.6876e-01, 1.5975e-01, ..., 5.3968e-02,
-1.7703e-01, 4.5894e-02],
...,
[-1.6952e+00, -4.8491e-01, 2.1268e-01, ..., 5.7488e-02,
-8.9156e-02, 8.0669e-02],
[-2.1494e+00, -1.1161e+00, -1.9012e-01, ..., -1.3878e-01,
-2.8222e-01, -2.5448e-01],
[-1.0495e+00, -1.9521e-01, 4.5648e-01, ..., -1.9856e-01,
-3.4872e-01, 8.8728e-02]]],
[[[ 4.6098e-02, -2.6295e-01, -7.3610e-02, ..., -9.6621e-01,
-5.2821e-01, -1.4422e+00],
[ 2.5779e-01, -9.6882e-01, -7.2258e-01, ..., -1.1852e+00,
-1.3253e+00, -1.4800e+00],
[ 2.3870e-01, -5.8317e-01, -5.6227e-01, ..., -9.2534e-01,
-1.1025e+00, -1.8257e+00],
...,
[ 3.5180e-02, -6.0007e-01, -3.6640e-01, ..., -8.9764e-01,
-1.0094e+00, -1.9276e+00],
[-6.4002e-01, -6.9118e-01, -3.2923e-03, ..., -4.4331e-01,
-5.5178e-01, -1.7126e+00],
[-7.6198e-01, -1.3551e+00, -1.1659e+00, ..., -1.1378e+00,
-1.5179e+00, -7.6448e-01]],
[[-9.2028e-01, -2.2745e+00, -1.9884e+00, ..., -3.9375e+00,
-5.4903e+00, -1.8513e+00],
[-2.2030e+00, -3.9170e+00, -3.2151e+00, ..., -5.9580e+00,
-8.1173e+00, -3.4106e+00],
[-1.7085e+00, -2.6644e+00, -2.7389e+00, ..., -5.2540e+00,
-6.1889e+00, -2.5539e+00],
...,
[-1.7220e+00, -2.4345e+00, -2.4900e+00, ..., -5.1087e+00,
-6.3017e+00, -2.4707e+00],
[-3.7875e+00, -4.4450e+00, -3.7853e+00, ..., -6.4926e+00,
-7.8882e+00, -2.5066e+00],
[ 2.6284e-01, -3.1932e-01, 3.6798e-01, ..., 4.1876e-01,
4.6848e-01, 2.3381e-01]],
[[-2.3366e-01, 1.8329e-01, 7.3136e-01, ..., 6.1561e-01,
2.5474e-01, 1.5274e+00],
[ 1.3135e-02, -3.8611e-01, 6.1892e-01, ..., 5.8258e-01,
-6.6458e-02, 1.1429e-01],
[-1.5097e-01, -2.9787e-01, 2.2790e-01, ..., 1.1147e-01,
-8.1026e-03, -1.2975e-01],
...,
[-2.8293e-01, -2.0475e-01, 1.6261e-01, ..., 3.3038e-02,
-1.0255e-01, -1.7591e-01],
[-2.6872e-01, -2.1249e-01, 1.9024e-01, ..., -4.4175e-02,
-7.3741e-03, -1.7476e-01],
[-1.2387e-01, -5.2229e-01, 1.0519e-01, ..., 9.5220e-02,
2.4219e-01, -5.2138e-01]],
...,
[[ 6.1992e-01, -2.4685e-01, 2.5186e-03, ..., 4.4114e-01,
3.3306e-01, -1.3199e+00],
[-3.8993e-01, -4.8844e-01, -5.5449e-01, ..., -1.3051e-01,
-7.8709e-01, -1.4029e+00],
[-2.4128e-01, -2.2479e-01, -2.6884e-01, ..., -1.0585e-01,
-1.2241e-01, -9.6982e-01],
...,
[-3.7901e-01, -1.5077e-01, -1.6222e-01, ..., -1.0226e-01,
-1.0667e-01, -9.3393e-01],
[-2.2944e+00, -1.7261e+00, -1.3060e+00, ..., -1.5982e+00,
-1.5515e+00, -1.8112e+00],
[-4.6058e-01, -7.9814e-01, -3.8108e-01, ..., -3.7332e-01,
-5.5414e-01, -1.1899e+00]],
[[ 1.8487e-01, 2.1792e-01, 5.6620e-01, ..., -7.0788e-02,
3.8046e-01, -3.7670e-01],
[ 1.4995e-01, 2.1468e-01, 3.6564e-01, ..., 1.5738e-01,
9.2632e-02, -8.3057e-01],
[-5.6704e-02, 2.4240e-01, 2.5883e-01, ..., 9.4559e-02,
1.5619e-01, -4.5931e-01],
...,
[-1.4550e-01, 2.6238e-02, 2.7531e-01, ..., 1.4075e-01,
1.4361e-01, -5.0064e-01],
[-5.3472e-01, -6.6965e-01, -4.0717e-02, ..., -2.4327e-01,
-2.8205e-01, -9.1088e-01],
[ 5.0408e-01, 4.2539e-01, 8.9566e-02, ..., 3.0306e-01,
3.3311e-01, 2.8480e-01]],
[[-8.1740e-01, 2.3334e-01, 2.9689e-01, ..., -8.8307e-02,
-8.3496e-02, 5.7977e-01],
[-1.1132e+00, 1.8857e-01, -3.5225e-01, ..., -9.2839e-01,
-4.3300e-01, 8.9782e-01],
[-1.0208e+00, 2.9658e-01, -9.2320e-02, ..., -5.4978e-01,
-5.2876e-01, 6.2886e-01],
...,
[-9.4428e-01, 1.7501e-01, -1.6865e-01, ..., -5.3122e-01,
-5.2623e-01, 6.1142e-01],
[-1.6886e+00, -3.9727e-01, -3.9212e-01, ..., -7.2980e-01,
-7.0359e-01, 3.8352e-01],
[-7.3190e-01, -9.1120e-02, -3.1161e-01, ..., -4.6469e-01,
-4.3395e-01, 9.9510e-01]]],
[[[-7.2772e-01, -1.3996e+00, -1.4335e+00, ..., -2.0282e-01,
2.5944e-01, -3.7829e-01],
[-2.7978e-01, -2.4253e+00, -2.2004e+00, ..., -5.9750e-01,
-3.9001e-01, -5.6996e-01],
[-3.8201e-01, -2.1088e+00, -1.8222e+00, ..., -3.5354e-01,
-3.4297e-01, -1.0400e+00],
...,
[-4.8708e-01, -1.9646e+00, -1.7752e+00, ..., -1.1490e-01,
6.3032e-02, -1.0700e+00],
[-9.6421e-01, -1.7559e+00, -1.2105e+00, ..., -2.7676e-03,
1.2316e-01, -9.9490e-01],
[-7.9116e-01, -1.6877e+00, -1.2586e+00, ..., -8.9944e-01,
-1.0753e+00, -6.4076e-01]],
[[-2.3418e+00, -4.7926e+00, -4.1853e+00, ..., -2.1095e+00,
-3.4436e+00, -1.2556e+00],
[-5.3591e+00, -8.1183e+00, -6.5108e+00, ..., -3.3964e+00,
-5.0844e+00, -2.0681e+00],
[-4.3204e+00, -6.2354e+00, -5.8802e+00, ..., -2.9938e+00,
-3.8116e+00, -1.7129e+00],
...,
[-4.3789e+00, -6.0909e+00, -5.6849e+00, ..., -2.8640e+00,
-3.5680e+00, -1.3378e+00],
[-6.3542e+00, -7.4187e+00, -6.3390e+00, ..., -3.8440e+00,
-5.3312e+00, -1.9255e+00],
[-9.0353e-03, -8.0857e-01, 3.9950e-01, ..., 4.1536e-01,
6.1662e-01, 1.8626e-01]],
[[-5.9108e-01, 5.9567e-01, 1.1100e+00, ..., 8.1777e-01,
5.7786e-01, 1.3613e+00],
[-9.0084e-01, 3.3105e-03, 4.0788e-01, ..., 4.2019e-01,
3.7587e-01, 2.9205e-01],
[-1.6630e+00, -1.2227e-01, -4.4456e-02, ..., 1.0570e-01,
3.1564e-01, 1.1703e-01],
...,
[-1.7117e+00, -5.1245e-02, 2.4624e-01, ..., 3.9782e-01,
1.5919e-01, -1.8165e-01],
[-1.5405e+00, 1.1678e-01, -6.8869e-03, ..., 9.0588e-02,
2.2918e-01, 1.5557e-01],
[-1.4121e+00, -5.1222e-01, -2.8046e-01, ..., 3.3573e-02,
2.0227e-01, -1.2159e-01]],
...,
[[ 1.0298e+00, -4.1602e-01, -2.2604e-01, ..., 3.3220e-01,
1.7378e-01, -8.0322e-01],
[-8.2149e-01, -9.5937e-01, -8.5114e-01, ..., -3.4938e-01,
-1.0397e+00, -9.6826e-01],
[-6.7761e-01, -9.4979e-01, -6.9707e-01, ..., -9.3551e-02,
-2.3176e-01, -6.0296e-01],
...,
[-7.4633e-01, -1.1144e+00, -9.6878e-01, ..., -3.4946e-01,
-4.2915e-01, -7.6944e-01],
[-2.7665e+00, -2.8326e+00, -1.8435e+00, ..., -1.0906e+00,
-1.1285e+00, -1.1005e+00],
[-6.6479e-01, -1.6244e+00, -9.8828e-01, ..., -1.4356e-01,
-2.4398e-01, -4.7529e-01]],
[[ 9.4811e-03, 3.1263e-02, 6.0372e-01, ..., 3.2475e-01,
8.1574e-01, 3.6896e-02],
[-1.0553e-02, -6.8760e-01, 1.3407e-01, ..., 3.7354e-01,
6.3768e-01, -2.8023e-01],
[-9.2770e-02, -7.9586e-01, 3.8304e-01, ..., 3.8838e-01,
4.5532e-01, -1.0226e-01],
...,
[ 2.6952e-02, -7.8776e-01, 3.9593e-01, ..., 6.5504e-01,
6.3021e-01, -5.8573e-01],
[-2.4824e-01, -1.4411e+00, 2.5578e-02, ..., 1.9129e-02,
1.1665e-01, -4.5644e-01],
[ 6.6011e-01, 1.4033e-01, 2.6694e-01, ..., 2.2301e-01,
3.0879e-01, 3.8150e-01]],
[[-1.6065e+00, -6.3616e-01, 7.5292e-02, ..., 3.4806e-01,
2.8189e-01, 8.4744e-01],
[-2.0855e+00, -3.4837e-01, -5.4583e-05, ..., -4.3427e-01,
-1.4703e-01, 1.1737e+00],
[-1.7354e+00, -3.5666e-01, 2.4956e-01, ..., -7.1784e-02,
1.7516e-02, 9.7195e-01],
...,
[-1.7645e+00, -5.3675e-01, 9.7547e-02, ..., -2.4349e-01,
2.0176e-01, 1.1062e+00],
[-2.2000e+00, -1.1443e+00, -1.9225e-01, ..., -2.9384e-01,
-1.3128e-01, 6.2603e-01],
[-1.0538e+00, -1.8790e-01, 4.2345e-01, ..., -2.1282e-01,
-2.5589e-01, 7.4206e-01]]]], device='cuda:0'),
tensor([[[[ 1.3349e-02, -2.3350e-01, -4.1818e-03, ..., -9.2866e-01,
-5.4868e-01, -1.4358e+00],
[ 3.1634e-01, -9.3732e-01, -6.9696e-01, ..., -1.1178e+00,
-1.3289e+00, -1.4943e+00],
[ 2.5180e-01, -5.8532e-01, -4.8051e-01, ..., -9.1144e-01,
-1.0856e+00, -1.8427e+00],
...,
[-3.0541e-02, -6.1730e-01, -3.9900e-01, ..., -8.7771e-01,
-1.0434e+00, -1.9126e+00],
[-6.6254e-01, -7.0337e-01, -6.8280e-02, ..., -4.3299e-01,
-5.3717e-01, -1.6853e+00],
[-7.6926e-01, -1.3431e+00, -1.1909e+00, ..., -1.1316e+00,
-1.5438e+00, -7.7893e-01]],
[[-7.4804e-01, -2.1565e+00, -1.7949e+00, ..., -3.8974e+00,
-5.4146e+00, -1.8268e+00],
[-2.0482e+00, -3.8000e+00, -3.0206e+00, ..., -5.8807e+00,
-8.0424e+00, -3.3982e+00],
[-1.5772e+00, -2.6073e+00, -2.5598e+00, ..., -5.1028e+00,
-6.1787e+00, -2.5872e+00],
...,
[-1.7903e+00, -2.4316e+00, -2.4007e+00, ..., -4.9862e+00,
-6.2602e+00, -2.4933e+00],
[-3.8023e+00, -4.4281e+00, -3.6405e+00, ..., -6.3573e+00,
-7.8739e+00, -2.5264e+00],
[ 2.6169e-01, -3.3651e-01, 3.6888e-01, ..., 4.2217e-01,
4.8346e-01, 2.0801e-01]],
[[-2.2900e-01, 1.4506e-01, 7.3498e-01, ..., 6.3923e-01,
2.9880e-01, 1.5611e+00],
[ 8.6031e-02, -3.8800e-01, 5.7646e-01, ..., 5.5497e-01,
-3.3776e-02, 1.4292e-01],
[-1.3192e-01, -2.1867e-01, 2.1266e-01, ..., -2.2941e-02,
1.0350e-02, -1.4727e-01],
...,
[-2.7270e-01, -1.9921e-01, 1.3867e-01, ..., -2.1870e-03,
-8.4359e-02, -1.5099e-01],
[-2.8256e-01, -1.9756e-01, 1.7375e-01, ..., -9.5231e-02,
1.3306e-03, -1.7044e-01],
[-1.3532e-01, -4.9056e-01, 8.9311e-02, ..., 7.6697e-02,
1.9608e-01, -5.8629e-01]],
...,
[[ 5.6941e-01, -3.1473e-01, -3.3514e-02, ..., 4.7578e-01,
3.3323e-01, -1.3659e+00],
[-5.4388e-01, -5.8253e-01, -5.6245e-01, ..., -1.2202e-01,
-7.7056e-01, -1.4265e+00],
[-3.7456e-01, -2.1612e-01, -2.3377e-01, ..., -9.2474e-02,
-1.0130e-01, -9.8100e-01],
...,
[-4.5312e-01, -1.5505e-01, -1.8111e-01, ..., -8.7658e-02,
-9.6325e-02, -9.5416e-01],
[-2.3710e+00, -1.7379e+00, -1.2472e+00, ..., -1.5214e+00,
-1.5653e+00, -1.8337e+00],
[-4.8369e-01, -8.2529e-01, -4.1731e-01, ..., -4.0074e-01,
-5.3723e-01, -1.1592e+00]],
[[ 1.3179e-01, 2.4280e-01, 6.6420e-01, ..., -8.9519e-02,
4.1576e-01, -3.3893e-01],
[ 1.3383e-01, 2.7636e-01, 4.6907e-01, ..., 2.1822e-01,
1.1713e-01, -8.0707e-01],
[-1.2404e-01, 8.9827e-02, 3.0670e-01, ..., 2.4245e-01,
1.6886e-01, -4.7586e-01],
...,
[-2.2957e-01, -1.4243e-02, 3.2379e-01, ..., 2.4185e-01,
1.2655e-01, -4.9961e-01],
[-5.7325e-01, -7.1638e-01, -4.0329e-02, ..., -1.5584e-01,
-2.5308e-01, -8.9010e-01],
[ 5.1680e-01, 3.9278e-01, 7.9935e-02, ..., 3.3845e-01,
3.1051e-01, 2.8887e-01]],
[[-8.6736e-01, 2.7184e-01, 3.3304e-01, ..., -9.4520e-02,
-6.8589e-02, 5.9869e-01],
[-1.1380e+00, 2.4110e-01, -3.4858e-01, ..., -9.3163e-01,
-3.8937e-01, 8.9896e-01],
[-9.3625e-01, 3.6784e-01, -1.3133e-01, ..., -6.3244e-01,
-4.7564e-01, 6.5076e-01],
...,
[-9.4630e-01, 1.9581e-01, -1.0995e-01, ..., -5.7820e-01,
-5.4074e-01, 6.1231e-01],
[-1.7026e+00, -4.0432e-01, -3.3335e-01, ..., -7.0938e-01,
-6.9246e-01, 3.9333e-01],
[-7.2120e-01, -8.3202e-02, -2.8744e-01, ..., -4.7717e-01,
-4.4837e-01, 9.7583e-01]]],
[[[-1.9166e-01, -2.8540e-01, 1.2348e-01, ..., 8.7861e-01,
7.7797e-01, 2.9246e-01],
[-2.7143e-02, -1.9391e+00, -1.3581e+00, ..., -1.5242e-01,
-4.0836e-01, -9.5031e-02],
[-5.5703e-01, -2.5312e+00, -1.5638e+00, ..., -8.7316e-01,
-3.8089e-01, -6.4427e-01],
...,
[-2.4311e-01, -1.7069e+00, -1.3209e+00, ..., -1.1425e+00,
-1.2208e+00, -2.0583e+00],
[-1.1388e+00, -1.7322e+00, -8.1227e-01, ..., -8.2708e-01,
-1.0230e+00, -1.8184e+00],
[-1.2979e+00, -2.1503e+00, -1.6276e+00, ..., -1.6490e+00,
-1.6190e+00, -9.7603e-01]],
[[-2.6720e-01, -1.5003e+00, -7.8433e-01, ..., -2.3670e+00,
-2.3080e+00, -1.0355e+00],
[-2.9915e+00, -4.8598e+00, -4.0178e+00, ..., -4.3292e+00,
-4.3452e+00, -2.1047e+00],
[-5.5642e+00, -6.4739e+00, -7.3632e+00, ..., -3.9057e+00,
-3.8037e+00, -1.6176e+00],
...,
[-3.7329e+00, -5.5046e+00, -5.1963e+00, ..., -6.9882e+00,
-7.3769e+00, -2.9414e+00],
[-6.1792e+00, -7.0093e+00, -6.1057e+00, ..., -7.4724e+00,
-8.0416e+00, -2.7185e+00],
[-7.1644e-02, -9.0909e-01, 2.6547e-01, ..., 2.3397e-01,
5.6448e-01, 1.8598e-01]],
[[-4.7602e-01, -8.0364e-02, 3.8560e-01, ..., -9.0632e-01,
-6.6835e-01, 7.9130e-01],
[-3.3310e-01, -3.7243e-01, 6.7236e-01, ..., -9.6318e-01,
-4.4396e-01, 2.3964e-01],
[-1.8112e+00, -1.6007e+00, -8.9572e-01, ..., 1.2194e-01,
-1.5520e-01, 3.5989e-01],
...,
[-1.4119e+00, 2.5198e-02, 2.0057e-02, ..., -2.6004e+00,
9.4008e-01, 3.0810e-01],
[-1.2685e+00, 1.6960e-01, -2.8497e-02, ..., -7.2240e-01,
4.3464e-01, 2.8455e-01],
[-1.1867e+00, -6.2413e-02, 9.0694e-02, ..., -2.0353e-01,
-1.1065e-02, -3.5858e-01]],
...,
[[ 1.9385e-01, -4.8012e-01, -3.7234e-01, ..., -1.2932e+00,
-4.4029e-01, -5.2907e-01],
[-5.9825e-01, -4.5618e-01, -3.9999e-01, ..., -1.2992e+00,
-1.4032e+00, -1.1469e+00],
[ 6.0535e-01, 3.7070e-01, 5.1116e-02, ..., -9.6021e-01,
-1.0956e+00, -1.1202e+00],
...,
[-6.1972e-01, -6.3384e-01, -3.1660e-01, ..., -4.9974e+00,
-2.8686e+00, -1.1904e+00],
[-3.1020e+00, -2.9833e+00, -2.0563e+00, ..., -4.3352e+00,
-2.4370e+00, -2.0828e+00],
[-5.2308e-01, -1.6847e+00, -1.0798e+00, ..., -9.6466e-01,
-9.5547e-01, -1.3582e+00]],
[[-1.3378e-01, 8.3423e-02, 3.8906e-01, ..., -5.3784e-01,
3.3837e-01, -1.1471e-02],
[-1.0941e-01, -7.6331e-02, -1.2904e-01, ..., 1.0870e-01,
-8.1421e-02, -2.4254e-01],
[-8.8590e-02, -1.2323e+00, -4.9304e-01, ..., -4.0234e-01,
-1.3971e-01, 4.4449e-02],
...,
[-1.1946e-01, -1.0489e+00, 3.3649e-02, ..., 9.0240e-01,
1.5793e-01, -7.1276e-01],
[-5.3792e-01, -1.8597e+00, -1.7151e-01, ..., -2.4567e-01,
-4.6646e-01, -9.0606e-01],
[ 4.3703e-01, -1.3549e-01, 1.2114e-01, ..., -1.3175e-01,
1.6145e-01, 3.1449e-01]],
[[-6.1586e-01, 4.1717e-01, 5.6172e-01, ..., -1.0031e+00,
-2.6504e-01, 4.8367e-01],
[-9.0009e-01, 1.5043e-01, -2.5906e-01, ..., -1.5724e+00,
-1.0840e+00, 8.0244e-01],
[-1.6117e+00, -1.9507e+00, -1.5877e+00, ..., -7.1348e-01,
-4.8075e-01, 9.5655e-01],
...,
[-1.1514e+00, -9.6653e-01, -4.7662e-01, ..., 1.3475e+00,
8.2878e-01, 8.5685e-01],
[-1.9218e+00, -1.8869e+00, -1.0046e+00, ..., 7.2666e-01,
-2.0252e-02, 4.6457e-01],
[-9.6235e-01, -9.7649e-01, -2.5352e-01, ..., -2.7839e-02,
-1.9118e-01, 1.1471e+00]]],
[[[-7.1829e-01, -1.4521e+00, -1.4044e+00, ..., -9.8101e-03,
6.2612e-01, 5.5539e-01],
[-2.9537e-01, -2.4422e+00, -2.1714e+00, ..., -4.8021e-01,
-2.5558e-01, 1.2516e-01],
[-3.9559e-01, -2.1562e+00, -1.8853e+00, ..., -3.0695e-01,
-1.5237e-01, -1.1771e-01],
...,
[-4.6653e-01, -1.9005e+00, -1.8182e+00, ..., -3.3527e-01,
-6.5663e-02, -1.3257e-01],
[-9.7636e-01, -1.7248e+00, -1.1784e+00, ..., 2.2139e-02,
2.7049e-01, -2.1506e-02],
[-8.1851e-01, -1.6710e+00, -1.2562e+00, ..., -1.0344e+00,
-1.0563e+00, -3.7115e-01]],
[[-2.4553e+00, -4.9263e+00, -4.1737e+00, ..., -4.3351e-01,
-1.0405e+00, -2.4803e-02],
[-5.4038e+00, -8.2001e+00, -6.5234e+00, ..., -1.7970e+00,
-2.7083e+00, -7.0193e-01],
[-4.3349e+00, -6.2889e+00, -5.4768e+00, ..., -1.3159e+00,
-1.5141e+00, -3.9977e-01],
...,
[-4.3215e+00, -6.0646e+00, -5.4814e+00, ..., -1.2891e+00,
-1.5401e+00, -3.6314e-01],
[-6.4107e+00, -7.3993e+00, -6.3037e+00, ..., -2.4721e+00,
-3.2930e+00, -9.4353e-01],
[-3.0514e-02, -8.4920e-01, 4.1239e-01, ..., 3.5699e-01,
3.6505e-01, 9.1810e-02]],
[[-5.7692e-01, 6.1055e-01, 1.1435e+00, ..., 7.9809e-01,
2.8093e-01, 1.1012e+00],
[-8.6740e-01, -8.6188e-02, 4.5922e-01, ..., 3.9378e-01,
2.1901e-02, 4.8114e-01],
[-1.6550e+00, -1.6561e-01, -5.5344e-02, ..., 2.7285e-02,
8.0235e-02, 3.8474e-01],
...,
[-1.7205e+00, -5.0241e-02, -1.0979e-01, ..., 5.6345e-02,
5.1240e-02, 4.0029e-01],
[-1.5449e+00, 8.6778e-02, 4.0917e-02, ..., 1.3197e-01,
1.1912e-01, 5.8746e-01],
[-1.4224e+00, -5.0013e-01, -2.4275e-01, ..., -8.3240e-02,
9.4997e-02, 3.6080e-01]],
...,
[[ 1.0830e+00, -4.0529e-01, -1.9375e-01, ..., -9.8734e-02,
-4.0991e-01, -9.5040e-01],
[-7.7983e-01, -9.1019e-01, -8.2888e-01, ..., -6.7430e-01,
-1.2735e+00, -8.4360e-01],
[-6.2585e-01, -9.0074e-01, -7.2268e-01, ..., -1.1288e-01,
-2.1354e-01, -4.1021e-01],
...,
[-7.6181e-01, -1.1778e+00, -7.5871e-01, ..., -1.1229e-01,
-1.8420e-01, -3.5154e-01],
[-2.7930e+00, -2.8332e+00, -1.8477e+00, ..., -8.2757e-01,
-8.5496e-01, -7.7626e-01],
[-6.4688e-01, -1.6591e+00, -9.9855e-01, ..., 1.3310e-02,
-1.1104e-01, -3.1433e-01]],
[[ 3.1396e-02, -1.8804e-02, 5.9012e-01, ..., 3.1267e-01,
6.9932e-01, 3.0337e-01],
[ 5.8990e-02, -7.2714e-01, 1.1678e-01, ..., -3.5743e-02,
1.0129e-01, -5.4677e-02],
[-1.1589e-02, -7.7921e-01, 2.8977e-01, ..., -1.6752e-01,
-1.0934e-01, 1.8020e-01],
...,
[ 5.2306e-02, -7.7473e-01, 2.7806e-01, ..., -1.4217e-01,
-9.9718e-02, 1.6020e-01],
[-2.7982e-01, -1.4784e+00, 1.9680e-02, ..., -3.8840e-01,
-3.4050e-01, -5.5929e-02],
[ 6.4391e-01, 1.2276e-01, 2.8589e-01, ..., -2.3680e-01,
-1.9991e-01, 3.3078e-01]],
[[-1.6167e+00, -5.6083e-01, 2.7074e-02, ..., 5.5094e-01,
3.6041e-01, 5.5048e-01],
[-2.1022e+00, -2.3108e-01, -3.2623e-03, ..., -5.6363e-02,
-1.8395e-01, 4.0968e-01],
[-1.7665e+00, -3.4747e-01, 2.6409e-01, ..., 3.1181e-02,
-1.9313e-01, 1.0124e-01],
...,
[-1.7220e+00, -5.1460e-01, 2.0654e-01, ..., 6.9471e-02,
-8.7015e-02, 1.0068e-01],
[-2.1664e+00, -1.1620e+00, -2.2154e-01, ..., -8.9943e-02,
-2.8064e-01, -1.9466e-01],
[-1.0405e+00, -1.9736e-01, 4.4573e-01, ..., -1.0982e-01,
-2.6812e-01, 1.3714e-01]]],
[[[ 3.5062e-02, -2.6765e-01, -6.7753e-02, ..., -1.1983e+00,
-6.6178e-01, -1.0672e+00],
[ 3.0245e-01, -9.7273e-01, -7.5446e-01, ..., -1.8886e+00,
-2.3948e+00, -2.2071e+00],
[ 2.5035e-01, -5.8158e-01, -5.1570e-01, ..., -3.7176e+00,
-5.2356e+00, -2.2065e+00],
...,
[ 5.7409e-02, 5.7441e-01, 1.1926e+00, ..., -1.5796e+00,
-9.9870e-01, -1.8529e+00],
[-6.2869e-01, -2.0299e-01, 9.5543e-01, ..., -8.4867e-01,
-5.8134e-01, -1.4649e+00],
[-7.3665e-01, -1.2978e+00, -1.1454e+00, ..., -1.3188e+00,
-1.4956e+00, -9.0317e-01]],
[[-7.5415e-01, -2.1871e+00, -1.9672e+00, ..., -4.1376e+00,
-4.9593e+00, -1.8782e+00],
[-2.0936e+00, -3.8160e+00, -3.1159e+00, ..., -8.3743e+00,
-9.7502e+00, -3.8653e+00],
[-1.6834e+00, -2.6801e+00, -2.6776e+00, ..., -1.6472e+01,
-1.4932e+01, -5.0024e+00],
...,
[-1.8305e+00, -2.4386e+00, -3.3569e+00, ..., -5.7026e+00,
-5.6924e+00, -2.5288e+00],
[-3.8685e+00, -4.8460e+00, -4.0925e+00, ..., -6.4166e+00,
-7.3790e+00, -2.8160e+00],
[ 2.7921e-01, -2.9694e-01, 3.9710e-01, ..., 5.2832e-01,
6.2760e-01, 5.7353e-02]],
[[-2.0788e-01, 1.7322e-01, 7.3703e-01, ..., 9.6490e-01,
7.2998e-01, 1.5487e+00],
[ 9.1522e-02, -3.3646e-01, 5.6768e-01, ..., -1.2260e-01,
-1.8485e-01, 4.7152e-01],
[-1.5674e-01, -1.9339e-01, 2.2976e-01, ..., -4.0688e+00,
-3.7369e+00, -1.2106e-01],
...,
[-2.6665e-01, 2.0075e+00, -3.5737e+00, ..., 3.1792e-01,
3.1641e-02, -2.9533e-01],
[-2.4291e-01, 1.1545e+00, 3.0632e-01, ..., -1.6391e-01,
1.0166e-01, 1.1883e-01],
[-7.0852e-02, -4.4462e-01, 9.6078e-02, ..., -4.4377e-01,
-1.2116e-01, -3.1069e-01]],
...,
[[ 6.0287e-01, -2.0251e-01, 5.9270e-02, ..., -1.0416e-01,
-2.7491e-01, -1.5145e+00],
[-5.2071e-01, -5.2784e-01, -5.1928e-01, ..., -7.4929e-01,
-1.1155e+00, -1.3882e+00],
[-3.3776e-01, -1.9383e-01, -2.0226e-01, ..., 1.1760e+00,
6.4495e-01, -2.2370e+00],
...,
[-4.0952e-01, -7.9441e-01, -8.4302e-01, ..., -1.1843e+00,
-1.3235e+00, -1.4175e+00],
[-2.2891e+00, -2.7052e+00, -2.7756e+00, ..., -1.9468e+00,
-1.9590e+00, -1.8143e+00],
[-4.5592e-01, -7.8024e-01, -3.5181e-01, ..., -7.5687e-01,
-8.3934e-01, -9.2632e-01]],
[[ 1.6448e-01, 2.6652e-01, 6.3438e-01, ..., 4.7203e-01,
7.9298e-01, -4.6921e-02],
[ 1.5265e-01, 2.5826e-01, 4.3834e-01, ..., 4.2827e-01,
8.5927e-02, -9.6821e-01],
[-1.2256e-01, 7.4897e-02, 2.9719e-01, ..., -6.7000e-01,
-4.2985e+00, -2.6952e+00],
...,
[-1.2399e-01, -2.9796e-02, -1.2926e+00, ..., 2.1930e-01,
2.8822e-01, -7.4639e-01],
[-5.4194e-01, -2.7606e-01, -1.1546e-01, ..., -4.7139e-02,
-9.7443e-02, -5.8027e-01],
[ 5.2820e-01, 4.1765e-01, 1.2642e-01, ..., 2.2911e-01,
1.6279e-01, 3.2348e-01]],
[[-8.4358e-01, 2.4811e-01, 3.0756e-01, ..., 7.7389e-02,
2.1054e-01, 8.5087e-01],
[-1.1257e+00, 1.9356e-01, -3.7274e-01, ..., -4.1813e-01,
-6.4057e-02, 1.1706e+00],
[-9.1616e-01, 3.3497e-01, -1.0527e-01, ..., -1.4390e+00,
-2.4526e+00, -8.1727e-01],
...,
[-1.0199e+00, -1.0299e+00, 1.3533e+00, ..., 6.2693e-01,
6.0625e-01, 1.4872e+00],
[-1.7457e+00, -7.4378e-01, 4.4496e-01, ..., -4.4464e-02,
1.3507e-01, 8.6013e-01],
[-7.3300e-01, -1.1080e-01, -2.6288e-01, ..., 2.4069e-01,
1.5530e-01, 1.1356e+00]]],
[[[-6.0289e-01, 4.9074e-01, -1.3072e+00, ..., -9.8460e+00,
-8.1595e+00, -4.1411e+00],
[ 1.4628e+00, 1.1744e+00, -2.4959e+00, ..., -1.1156e+01,
-1.2169e+01, -8.6252e+00],
[ 1.7968e-01, -8.0946e-02, -1.5193e+00, ..., -1.3709e+01,
-1.0384e+01, -5.3535e+00],
...,
[-3.7086e-01, -7.8224e-01, 1.1436e+00, ..., -9.5068e-01,
-1.1096e+00, -3.0964e+00],
[-1.3904e-01, -2.9554e-01, 1.3178e+00, ..., -6.6614e-01,
-7.2566e-01, -1.9795e+00],
[-7.7616e-01, -1.5456e+00, -4.9791e-01, ..., -1.3655e+00,
-1.7753e+00, -1.0460e+00]],
[[-3.1786e+00, -6.8886e+00, -2.3886e+00, ..., -1.6456e+01,
-2.2855e+01, 1.6409e+00],
[-2.9267e+00, -7.8467e+00, -4.1734e+00, ..., -2.1328e+01,
-2.9473e+01, 8.0221e-01],
[-3.1484e+00, -5.4130e+00, -2.6892e+00, ..., -1.2471e+01,
-8.3570e+00, 5.3283e+00],
...,
[-2.4798e+00, -3.7371e+00, -4.7860e+00, ..., -8.3262e+00,
-8.8581e+00, -3.9504e+00],
[-4.0214e+00, -5.2460e+00, -7.0261e+00, ..., -6.6546e+00,
-8.2726e+00, -2.3954e+00],
[ 3.0174e-01, -4.1725e-01, 1.5248e+00, ..., 2.2170e-01,
2.8725e-01, 3.1375e-01]],
[[-3.4740e+00, -5.1104e+00, -1.0899e+00, ..., -2.4231e+01,
-2.2842e+01, -1.2086e-01],
[-3.7879e+00, -2.7071e+00, -8.3621e-01, ..., -3.5757e+01,
-3.2052e+01, -5.9430e+00],
[-5.9607e+00, -2.7468e+00, -2.6434e+00, ..., -3.7579e+01,
-1.5066e+01, -7.8062e+00],
...,
[-6.2351e-01, -1.5682e-01, 2.8191e+00, ..., 9.5004e-01,
4.2539e-01, -2.8163e+00],
[-6.3891e-01, -2.0544e-01, 2.4720e+00, ..., -1.0541e-01,
-9.1876e-03, -1.8663e-01],
[-1.0626e+00, -1.1001e+00, 9.5294e-01, ..., -5.4432e-02,
1.2174e-01, -6.9555e-01]],
...,
[[-1.0392e-01, -7.8683e-01, -1.7704e-02, ..., -2.0433e+00,
-6.9537e+00, -1.3579e+01],
[-6.2032e-01, -9.6776e-01, -1.4470e+00, ..., -2.5850e+00,
-1.2403e+01, -2.1862e+01],
[-1.1532e-01, -5.1387e-01, -6.7129e-01, ..., -5.7004e+00,
-1.5498e+01, -2.2038e+01],
...,
[-7.7438e-01, -7.9131e-01, 7.7102e-01, ..., -6.1061e+00,
-6.0440e+00, -3.3294e+00],
[-2.1635e+00, -1.6479e+00, -1.5386e+00, ..., -1.7078e+00,
-1.8989e+00, -2.0996e+00],
[-9.4399e-01, -1.1802e+00, 7.6249e-02, ..., -5.5091e-01,
-5.2943e-01, -1.3059e+00]],
[[-2.3385e+00, 1.4013e+00, 1.3350e+00, ..., -8.9445e+00,
-3.2261e+00, 3.4762e+00],
[-1.2370e+00, -1.2920e+00, 2.6809e-01, ..., -8.1148e+00,
-1.6742e+00, 2.3246e+00],
[-5.6903e-01, -9.0578e-01, 1.6088e+00, ..., 3.4568e-01,
-2.7084e+00, 4.5348e+00],
...,
[ 2.0788e-01, 4.1318e-01, 3.2019e+00, ..., -9.0201e-01,
-2.4319e+00, -2.3947e+00],
[ 1.1938e-01, -3.9027e-01, 1.7656e+00, ..., -1.1508e-01,
-1.9544e-01, -7.3077e-01],
[ 4.2839e-01, -8.0435e-02, 3.2799e-01, ..., 2.2651e-01,
1.6580e-01, 4.0089e-01]],
[[-2.1546e+00, -4.5910e+00, -1.7823e+00, ..., -1.3520e+01,
-2.0075e+01, -9.4562e+00],
[-3.3647e+00, -7.2295e+00, -1.1347e+00, ..., -1.9777e+01,
-2.3010e+01, -1.3009e+01],
[-5.0680e+00, -1.0193e+01, -1.6246e+00, ..., -1.7755e+01,
-2.1543e+01, -5.8480e+00],
...,
[-1.9619e+00, 1.3530e-01, -1.6480e+00, ..., 2.5527e+00,
1.2065e+00, 3.1459e+00],
[-2.1590e+00, -2.4344e-01, -6.5998e-01, ..., -9.3510e-01,
-1.0529e+00, 2.9487e-01],
[-1.3638e+00, -1.3206e-01, -4.4924e-01, ..., -4.9614e-01,
-6.0167e-01, 1.0878e+00]]]], device='cuda:0'),
tensor([[[[-9.6871e-01, -3.2106e+00, -3.1893e+00, ..., -7.6686e-02,
3.9093e-01, 7.7850e-02],
[-2.5835e-01, -3.6794e+00, -5.7190e+00, ..., -8.1818e-01,
-6.0247e-01, -4.5279e-01],
[ 8.3849e-01, -2.9271e-01, -2.0935e+00, ..., -7.6464e-01,
-5.7679e-01, -8.7639e-01],
...,
[-4.0024e-01, -6.1029e-01, -4.2952e-01, ..., -2.4704e+00,
-1.5687e+00, -2.1973e+00],
[-1.3779e+00, -8.6029e-01, 9.9856e-01, ..., -7.2726e-01,
-5.6479e-01, -1.9225e+00],
[-1.4034e+00, -2.3850e+00, -1.7607e+00, ..., -1.5352e+00,
-1.6547e+00, -1.0350e+00]],
[[-3.0477e+00, -6.1454e+00, -7.7451e+00, ..., -1.4076e+00,
-1.8505e+00, -4.8466e-01],
[-4.4411e+00, -1.0440e+01, -1.3295e+01, ..., -3.3014e+00,
-4.2019e+00, -1.5134e+00],
[-2.0548e+00, -6.6869e+00, -1.1944e+01, ..., -3.5012e+00,
-3.9186e+00, -1.5972e+00],
...,
[-4.0153e+00, -5.3777e+00, -4.0553e+00, ..., -6.0799e+00,
-6.5601e+00, -3.0065e+00],
[-6.4182e+00, -7.5178e+00, -6.5353e+00, ..., -6.4755e+00,
-7.7895e+00, -2.7867e+00],
[-1.2259e-01, -9.9301e-01, 1.9504e-01, ..., 3.8067e-01,
6.3024e-01, 2.1536e-01]],
[[-2.4775e+00, -6.5054e+00, -9.7178e+00, ..., 6.9661e-01,
2.2396e-01, 1.0930e+00],
[-1.8556e+00, -6.9822e+00, -1.0511e+01, ..., 6.7117e-01,
2.9956e-01, 4.5114e-01],
[-4.6201e-01, -2.2308e+00, -7.7257e+00, ..., 1.9995e-01,
2.0079e-01, 3.1059e-01],
...,
[-1.4637e+00, -9.2631e-01, -5.9398e+00, ..., 7.6388e-01,
9.3159e-02, 2.4490e-01],
[-1.2041e+00, 1.9541e+00, 3.0306e-01, ..., 5.1468e-02,
-8.3022e-02, 1.9472e-01],
[-1.2579e+00, -1.4858e-01, -4.4057e-04, ..., -3.7757e-02,
-6.9373e-02, -4.4669e-01]],
...,
[[ 6.6745e-01, -1.4562e+00, -4.5494e+00, ..., 4.2215e-02,
-1.9344e-01, -7.8929e-01],
[ 2.3944e-01, -2.3560e-01, -4.9153e+00, ..., -2.9898e-01,
-1.1224e+00, -9.1080e-01],
[ 4.7656e-01, 2.4472e+00, 4.2606e-01, ..., 1.1379e-01,
-2.2424e-01, -6.0597e-01],
...,
[-6.3530e-01, 1.2417e+00, 4.3956e+00, ..., -1.0695e+00,
-4.9425e-01, -1.2407e+00],
[-3.3311e+00, -4.5990e+00, -5.3207e+00, ..., -2.0301e+00,
-2.0995e+00, -2.1794e+00],
[-5.5706e-01, -1.7350e+00, -1.1007e+00, ..., -8.2858e-01,
-9.0779e-01, -1.4162e+00]],
[[-7.1152e-01, -3.8798e+00, -5.9361e+00, ..., 1.0971e-01,
6.0610e-01, 1.4400e-01],
[-8.4376e-01, -1.5570e+00, -3.7208e+00, ..., 9.4907e-02,
1.2139e-01, -1.7994e-01],
[-4.6637e-01, -5.7480e-01, -5.6527e-01, ..., 1.7621e-01,
-2.1158e-01, -1.6333e-01],
...,
[-1.5086e-01, -1.6316e+00, -6.4541e-01, ..., -8.1227e-01,
-4.5442e-01, -7.6414e-01],
[-5.0306e-01, -1.5722e+00, 5.6595e-01, ..., -2.3215e-01,
-2.4929e-02, -9.7798e-01],
[ 4.4637e-01, -1.2998e-01, 1.7599e-01, ..., 3.1883e-01,
3.5280e-01, 3.8791e-01]],
[[-1.3864e+00, -3.9073e+00, -6.9756e+00, ..., 4.5186e-01,
4.1975e-01, 7.5406e-01],
[-1.4002e+00, -3.2313e+00, -6.1059e+00, ..., -2.7597e-01,
-1.4611e-01, 8.3194e-01],
[-8.9685e-01, -9.0988e-01, -2.9936e+00, ..., -1.5220e-01,
-6.8586e-03, 6.9308e-01],
...,
[-1.3556e+00, -1.8481e+00, -1.3986e+00, ..., 9.8442e-02,
-5.3240e-02, 8.3071e-01],
[-2.1918e+00, -2.5917e+00, -2.7804e-02, ..., -1.0706e+00,
-5.3652e-01, 4.5653e-01],
[-1.0612e+00, -1.0199e+00, -3.0440e-01, ..., -2.7281e-01,
-1.5356e-01, 1.2446e+00]]],
[[[-7.3097e-01, -1.4263e+00, -1.3955e+00, ..., 3.8377e-01,
6.4117e-01, 5.3646e-01],
[-2.9081e-01, -2.4502e+00, -2.1579e+00, ..., -2.7919e-01,
-2.2549e-01, 1.1377e-01],
[-4.0347e-01, -2.1311e+00, -1.8511e+00, ..., -1.0386e+00,
-1.3935e-01, -1.0436e-01],
...,
[-4.8852e-01, -1.5812e+00, -2.2314e+00, ..., -3.1789e-01,
-7.7168e-02, -1.2698e-01],
[-9.4505e-01, -9.8414e-01, -2.6481e-01, ..., 4.5071e-02,
2.6260e-01, -3.4042e-02],
[-7.8692e-01, -1.2661e+00, -3.0068e-01, ..., -1.0254e+00,
-1.0839e+00, -3.8521e-01]],
[[-2.3726e+00, -4.8313e+00, -4.2614e+00, ..., -2.3805e+00,
-1.0681e+00, -1.3012e-02],
[-5.3262e+00, -8.0442e+00, -6.4937e+00, ..., -4.3634e+00,
-2.6518e+00, -6.6098e-01],
[-4.3295e+00, -6.2234e+00, -5.3960e+00, ..., -4.9584e+00,
-1.4765e+00, -3.9330e-01],
...,
[-4.3661e+00, -6.3448e+00, -2.0096e+00, ..., -1.3114e+00,
-1.5358e+00, -3.6534e-01],
[-6.3266e+00, -7.4110e+00, -4.5053e+00, ..., -2.3997e+00,
-3.3147e+00, -9.5646e-01],
[ 7.8244e-03, -7.6152e-01, 1.1601e+00, ..., 3.4015e-01,
3.8082e-01, 8.4192e-02]],
[[-6.1066e-01, 5.5406e-01, 1.0807e+00, ..., 5.2297e-01,
3.0061e-01, 1.0959e+00],
[-9.1601e-01, -5.2966e-02, 3.3866e-01, ..., -4.1138e-01,
5.7145e-02, 4.8508e-01],
[-1.6273e+00, -1.3188e-01, -4.9818e-02, ..., 6.1515e-01,
7.5836e-02, 4.3208e-01],
...,
[-1.7157e+00, 1.5526e+00, -4.3066e+00, ..., 2.8454e-02,
8.7174e-02, 4.3967e-01],
[-1.5387e+00, 8.9790e-01, -2.7443e+00, ..., 1.3496e-01,
1.0630e-01, 6.1889e-01],
[-1.4308e+00, 3.2565e-01, -1.1193e+00, ..., -6.7519e-02,
5.7068e-02, 3.7668e-01]],
...,
[[ 1.0196e+00, -4.0950e-01, -1.7472e-01, ..., -9.0822e-01,
-3.4222e-01, -9.2796e-01],
[-8.3904e-01, -9.8470e-01, -8.2765e-01, ..., -2.6629e+00,
-1.2560e+00, -8.3452e-01],
[-6.9337e-01, -9.6580e-01, -7.1084e-01, ..., -3.4960e+00,
-1.9455e-01, -4.0405e-01],
...,
[-7.6128e-01, -1.5153e+00, -3.7958e+00, ..., -9.2836e-02,
-1.9130e-01, -3.8104e-01],
[-2.7503e+00, -2.6451e+00, -2.5589e+00, ..., -8.3401e-01,
-9.0778e-01, -8.0736e-01],
[-6.6207e-01, -1.6372e+00, -2.9912e-01, ..., -4.3384e-02,
-1.0268e-01, -3.2062e-01]],
[[ 8.0989e-03, 8.5599e-03, 6.2426e-01, ..., 4.9132e-01,
6.8329e-01, 2.7894e-01],
[-4.8136e-02, -7.5054e-01, 1.3914e-01, ..., 1.1801e+00,
1.5295e-01, -8.1590e-02],
[-1.2029e-01, -8.4960e-01, 2.8962e-01, ..., 7.2329e-02,
-7.4945e-02, 1.5659e-01],
...,
[ 3.1367e-02, -1.0248e+00, -2.1475e+00, ..., -1.3835e-01,
-9.7124e-02, 1.3533e-01],
[-2.4384e-01, -1.7089e+00, -1.6467e+00, ..., -3.7876e-01,
-2.8724e-01, -6.4584e-02],
[ 7.1499e-01, 1.0566e-01, -2.7566e-01, ..., -2.4454e-01,
-2.0390e-01, 3.0247e-01]],
[[-1.6121e+00, -6.3792e-01, 2.9313e-02, ..., 6.7640e-01,
3.4390e-01, 5.5417e-01],
[-2.0717e+00, -3.6027e-01, -3.6349e-02, ..., -7.2505e-01,
-1.5208e-01, 3.9718e-01],
[-1.6936e+00, -4.0033e-01, 1.8035e-01, ..., 1.3216e-01,
-1.3556e-01, 7.8630e-02],
...,
[-1.7067e+00, -1.2203e+00, 1.0750e+00, ..., 4.1688e-02,
-6.4414e-02, 9.0244e-02],
[-2.1208e+00, -1.3168e+00, 1.2213e+00, ..., -1.3318e-01,
-2.3240e-01, -2.2085e-01],
[-1.0225e+00, -3.5937e-01, 1.4071e+00, ..., -1.1888e-01,
-2.4882e-01, 1.0090e-01]]],
[[[-7.3128e-01, -1.4603e+00, -1.4352e+00, ..., -1.3605e+00,
-1.1910e+00, -1.6852e+00],
[-3.3117e-01, -2.4733e+00, -2.2783e+00, ..., -1.2183e+00,
-2.3247e+00, -1.9146e+00],
[-1.1963e-01, -1.9438e+00, -1.7299e+00, ..., -2.3129e+00,
-2.1768e+00, -2.1121e+00],
...,
[-7.0917e-02, -7.5264e-01, 2.1813e-01, ..., -6.6635e-01,
-1.7719e-02, -4.4475e-01],
[-7.4103e-01, -7.2590e-01, 2.9329e-01, ..., -3.9666e-01,
2.6983e-01, -1.6757e-01],
[-9.1679e-01, -1.5817e+00, -1.2783e+00, ..., -1.2621e+00,
-1.0679e+00, -4.4351e-01]],
[[-2.6782e+00, -5.1059e+00, -4.6350e+00, ..., -5.9296e+00,
-8.0387e+00, -2.4229e+00],
[-5.3917e+00, -8.1571e+00, -7.4748e+00, ..., -7.6399e+00,
-1.1641e+01, -4.3827e+00],
[-4.1122e+00, -5.8583e+00, -5.7574e+00, ..., -6.4822e+00,
-1.0429e+01, -3.2783e+00],
...,
[-1.9227e+00, -2.4844e+00, -1.9951e+00, ..., -2.9851e+00,
-2.3549e+00, -1.0078e+00],
[-3.2396e+00, -3.6370e+00, -1.9601e+00, ..., -2.3694e+00,
-3.1349e+00, -9.1106e-01],
[ 2.3045e-01, -5.3192e-01, 2.9471e-01, ..., 2.0493e-01,
4.3647e-01, 2.3400e-01]],
[[-6.5533e-01, 4.9893e-01, 8.6518e-01, ..., 1.4008e-01,
8.4291e-01, 1.8938e+00],
[-9.8476e-01, 6.7327e-02, 4.0846e-01, ..., -2.7424e+00,
2.7116e-01, 8.6482e-02],
[-1.7953e+00, -2.2664e-01, -2.7063e-02, ..., -4.4896e+00,
5.5421e-01, -1.4940e-01],
...,
[-7.7298e-01, -8.7266e-03, 8.3812e-01, ..., 6.4170e-01,
2.3939e-01, 4.2094e-01],
[-5.5239e-01, -7.6054e-02, 6.6952e-01, ..., 3.1594e-01,
2.1156e-01, 5.8109e-01],
[-2.1991e-01, -3.3989e-01, 3.0843e-01, ..., 5.6638e-02,
1.1051e-01, 3.6244e-01]],
...,
[[ 1.2365e+00, -3.8085e-01, -4.8556e-02, ..., -1.0908e+00,
-3.1244e-02, -2.1283e+00],
[-7.5389e-01, -9.6636e-01, -5.0665e-01, ..., -1.2411e+00,
-2.0938e+00, -2.0797e+00],
[-7.1711e-01, -8.8193e-01, -3.1667e-01, ..., -5.3566e+00,
-3.9505e+00, -1.3899e+00],
...,
[-8.2217e-01, -5.1495e-01, -1.5761e-01, ..., -6.8599e-01,
-5.0607e-01, -5.4546e-01],
[-2.6623e+00, -2.0854e+00, -1.4717e+00, ..., -1.0412e+00,
-1.3138e+00, -1.0303e+00],
[-7.2776e-01, -1.1249e+00, -6.2709e-01, ..., -1.2877e-01,
-3.6488e-01, -3.7734e-01]],
[[ 5.2335e-02, 2.6838e-02, 4.5684e-01, ..., 1.6527e-01,
2.2905e-01, -2.6076e-02],
[ 2.3949e-01, -8.0120e-01, -6.3355e-02, ..., -7.7299e-01,
-4.7212e-01, -8.2315e-01],
[ 1.5717e-01, -6.6232e-01, 2.8741e-01, ..., -1.6508e+00,
-1.0197e+00, -3.8292e-01],
...,
[-4.0315e-01, -5.7098e-01, 9.2100e-01, ..., 1.5446e-01,
3.2568e-01, 3.9326e-02],
[-7.5672e-01, -1.0964e+00, 4.0998e-01, ..., -1.8530e-01,
1.9013e-02, -1.1159e-01],
[ 3.2574e-01, -5.3047e-02, -4.3002e-02, ..., -3.5687e-01,
-4.7498e-02, 1.9804e-01]],
[[-1.5716e+00, -8.2786e-01, -4.4117e-01, ..., -1.3218e+00,
-7.0013e-02, 8.6334e-01],
[-2.1685e+00, -7.9176e-01, -6.0005e-01, ..., -4.1769e+00,
-6.3392e-01, 1.6445e+00],
[-1.5100e+00, -5.4632e-01, -2.4059e-01, ..., -3.0502e+00,
1.8107e-02, 1.3945e+00],
...,
[-7.2021e-01, -2.6043e-01, -3.3350e-01, ..., 6.6917e-01,
-1.6980e-02, 6.3820e-01],
[-1.3561e+00, -7.7137e-01, -4.3581e-01, ..., 2.9595e-01,
-2.2097e-01, 1.8887e-01],
[-4.7079e-01, -1.5811e-01, -2.7146e-01, ..., 2.9650e-02,
-1.7123e-01, 3.7008e-01]]],
[[[-3.3057e-01, -9.5415e-01, -8.7335e-01, ..., -5.7850e-01,
-4.3794e-02, -7.6608e-01],
[ 2.6266e-02, -1.7519e+00, -9.5688e-01, ..., -1.4130e+00,
-8.6750e-01, -1.1906e+00],
[ 3.1738e-02, -1.3495e+00, -4.3793e-01, ..., -1.3640e+00,
-8.5135e-01, -1.7506e+00],
...,
[-6.9594e-02, -7.3653e-01, -2.2744e-01, ..., -9.6053e-02,
-3.7364e-02, -4.2595e-01],
[-6.9845e-01, -6.8929e-01, -4.2352e-02, ..., 1.3824e-01,
2.7759e-01, -1.4461e-01],
[-8.9403e-01, -1.5712e+00, -1.2281e+00, ..., -1.0321e+00,
-1.0982e+00, -4.6023e-01]],
[[-1.9692e+00, -3.8637e+00, -4.9798e+00, ..., -3.9186e+00,
-4.9322e+00, -1.9837e+00],
[-3.7324e+00, -5.9235e+00, -6.5230e+00, ..., -6.3351e+00,
-6.8887e+00, -2.9884e+00],
[-3.4601e+00, -5.1723e+00, -5.8843e+00, ..., -6.8991e+00,
-5.8755e+00, -2.8099e+00],
...,
[-1.8198e+00, -2.3624e+00, -1.8428e+00, ..., -1.8817e+00,
-2.2787e+00, -9.2227e-01],
[-3.1933e+00, -3.5917e+00, -2.3064e+00, ..., -2.1295e+00,
-3.1926e+00, -9.3926e-01],
[ 2.4687e-01, -5.1438e-01, 2.5560e-01, ..., 2.5932e-01,
4.4388e-01, 2.1928e-01]],
[[-4.1865e-01, 4.8265e-01, 4.8771e-01, ..., 8.3632e-01,
5.6139e-01, 1.4533e+00],
[-5.6341e-01, 8.7799e-02, 4.5775e-01, ..., 4.8259e-01,
2.9549e-01, 3.7049e-01],
[-1.0702e+00, 2.0075e-01, 1.9024e-01, ..., 4.6991e-01,
1.7984e-01, 3.2896e-01],
...,
[-7.5747e-01, 2.1931e-03, 2.6185e-01, ..., 1.9453e-01,
2.8685e-01, 4.2260e-01],
[-5.5454e-01, -8.4956e-02, 2.2986e-01, ..., 2.2107e-01,
2.5071e-01, 5.6845e-01],
[-2.3892e-01, -3.5029e-01, 1.1258e-01, ..., -2.5941e-02,
7.7510e-02, 3.3522e-01]],
...,
[[ 1.1075e+00, 3.7333e-01, 9.9599e-01, ..., 4.3786e-01,
2.1104e-01, -1.0287e+00],
[-6.5541e-01, -5.7945e-01, -1.6495e-01, ..., -4.0908e-01,
-1.2518e+00, -1.3450e+00],
[-1.6749e-01, -4.2243e-01, 1.2930e-01, ..., -9.0425e-01,
-5.8711e-01, -1.1392e+00],
...,
[-8.0132e-01, -5.3420e-01, -3.5453e-01, ..., -3.2956e-01,
-4.9782e-01, -5.4820e-01],
[-2.6369e+00, -2.0865e+00, -1.2823e+00, ..., -1.2752e+00,
-1.3181e+00, -1.0601e+00],
[-6.8243e-01, -1.0714e+00, -4.8958e-01, ..., -2.7996e-01,
-3.4895e-01, -3.5974e-01]],
[[ 2.0800e-01, 2.3589e-01, 3.1701e-01, ..., 2.2186e-01,
7.4954e-01, -2.5951e-01],
[ 4.1091e-01, -6.3058e-01, -1.2415e-01, ..., -8.3973e-01,
3.8518e-01, -7.9264e-01],
[ 4.1420e-02, -7.4578e-01, 8.5119e-02, ..., -7.3381e-01,
1.2961e-01, -6.6621e-01],
...,
[-4.0936e-01, -5.7518e-01, 2.3277e-01, ..., 3.6088e-01,
3.3313e-01, 5.8659e-02],
[-7.3863e-01, -1.0748e+00, -5.9739e-02, ..., 6.3314e-02,
1.0937e-02, -1.1878e-01],
[ 3.2158e-01, -4.7504e-02, -1.6234e-01, ..., -1.1764e-01,
-7.6755e-02, 1.8282e-01]],
[[-6.7120e-01, -7.5226e-02, -2.9186e-01, ..., 3.3266e-01,
4.1570e-01, 9.6753e-01],
[-1.1538e+00, -6.5804e-01, -1.0126e+00, ..., -6.9820e-01,
-2.4059e-01, 1.0017e+00],
[-8.5632e-01, -5.9572e-01, -6.9992e-01, ..., -5.6983e-01,
-4.4365e-02, 7.8566e-01],
...,
[-6.9282e-01, -2.1576e-01, -1.2763e-01, ..., -1.4725e-01,
6.7368e-03, 6.2134e-01],
[-1.3095e+00, -7.2037e-01, -4.9901e-01, ..., -3.0858e-01,
-1.7931e-01, 1.6834e-01],
[-4.6850e-01, -1.4487e-01, -3.0318e-01, ..., -1.8086e-01,
-1.6914e-01, 3.3398e-01]]],
[[[-3.6042e-01, -1.0306e+00, -8.8203e-01, ..., -7.8153e-03,
5.9279e-01, 5.2497e-01],
[ 1.3698e-01, -1.7567e+00, -1.2871e+00, ..., -5.0933e-01,
-2.2992e-01, 1.0998e-01],
[ 1.8720e-01, -1.3119e+00, -9.7270e-01, ..., -3.2798e-01,
-1.1272e-01, -9.4611e-02],
...,
[ 6.6349e-01, -4.4633e+00, -2.3932e+00, ..., -6.6505e-01,
-2.2329e-01, -1.3802e-01],
[ 1.7349e+00, -2.3591e+00, -7.0342e-01, ..., -7.3049e-02,
2.4441e-01, -3.4494e-02],
[ 4.2418e-01, -1.5826e+00, -1.7696e+00, ..., -1.0361e+00,
-1.0973e+00, -3.9582e-01]],
[[-2.1214e+00, -4.0415e+00, -3.8093e+00, ..., -3.9887e-01,
-1.0291e+00, -1.6712e-02],
[-3.7398e+00, -5.9559e+00, -5.4034e+00, ..., -1.7606e+00,
-2.6744e+00, -6.8246e-01],
[-3.0888e+00, -4.6857e+00, -4.6178e+00, ..., -1.2727e+00,
-1.5292e+00, -3.7558e-01],
...,
[-5.9303e+00, -1.2261e+01, -8.4488e+00, ..., -1.8531e+00,
-1.6710e+00, -3.7540e-01],
[-3.8680e+00, -8.4243e+00, -7.3776e+00, ..., -2.3740e+00,
-3.2876e+00, -9.1773e-01],
[ 7.0637e-01, -1.1084e+00, -4.6558e-01, ..., 3.3324e-01,
3.2480e-01, 1.0337e-01]],
[[-5.6549e-01, 4.0616e-01, 7.1841e-01, ..., 7.9979e-01,
2.7062e-01, 1.0706e+00],
[-8.6963e-01, 1.2595e-01, 5.7438e-01, ..., 4.0182e-01,
-1.6250e-02, 4.6298e-01],
[-1.2457e+00, 4.2829e-02, 2.1766e-02, ..., 4.7895e-02,
4.5578e-02, 3.8242e-01],
...,
[-9.8475e+00, -7.6505e+00, -5.0439e+00, ..., 9.2339e-02,
9.3076e-02, 3.8277e-01],
[-5.9310e+00, -2.9083e+00, -3.2627e+00, ..., 8.4411e-02,
7.3946e-02, 5.6721e-01],
[-1.0380e+00, 2.5218e-01, 5.2072e-01, ..., -8.6903e-02,
1.2745e-02, 3.1965e-01]],
...,
[[ 7.4000e-01, 1.5156e-01, 3.3576e-01, ..., -1.0620e-01,
-3.7574e-01, -9.2416e-01],
[-6.6413e-01, -2.9525e-01, 1.1561e-02, ..., -6.8036e-01,
-1.2175e+00, -8.2581e-01],
[-4.7282e-01, -3.6757e-01, -4.6282e-03, ..., -9.7157e-02,
-1.9088e-01, -3.9698e-01],
...,
[-2.2973e-01, -3.7104e+00, -4.3499e+00, ..., -1.1067e+00,
-1.3388e-01, -3.7878e-01],
[-1.8613e+00, -2.8735e+00, -3.3574e+00, ..., -8.3060e-01,
-8.4830e-01, -8.0297e-01],
[-1.5256e+00, -2.4913e+00, -1.1240e+00, ..., -1.1241e-02,
-1.1590e-01, -3.5010e-01]],
[[ 3.2344e-03, -2.5021e-01, -7.9395e-02, ..., 3.2907e-01,
6.3138e-01, 2.8563e-01],
[ 1.8426e-01, -8.7172e-01, -5.4456e-02, ..., -4.1264e-02,
6.7218e-02, -4.1219e-02],
[-1.2195e-01, -8.6599e-01, 6.1509e-02, ..., -1.6518e-01,
-7.8729e-02, 2.0757e-01],
...,
[-2.1814e+00, -2.3908e+00, -1.4921e+00, ..., 1.8424e-01,
-2.1582e-01, 1.1604e-01],
[-1.5582e+00, -4.2623e+00, 1.7804e-01, ..., -4.9275e-01,
-3.5066e-01, -5.1636e-02],
[ 4.8328e-01, -1.4409e+00, -1.9082e-01, ..., -2.4204e-01,
-2.7169e-01, 2.7444e-01]],
[[-5.9764e-01, -1.4242e-01, -2.4777e-02, ..., 5.7056e-01,
3.3993e-01, 5.2743e-01],
[-1.0564e+00, -8.1040e-01, -9.6231e-01, ..., -5.3559e-02,
-1.9885e-01, 4.0313e-01],
[-6.5821e-01, -7.1228e-01, -5.9129e-01, ..., 7.5238e-02,
-2.3964e-01, 1.1142e-01],
...,
[-8.6207e-02, -4.5898e+00, -4.8163e+00, ..., 8.5211e-01,
1.0277e-01, 8.6396e-02],
[ 1.5764e+00, -2.4148e-01, -2.8744e+00, ..., 2.2297e-02,
-2.8157e-01, -2.3167e-01],
[ 7.8797e-01, 1.1624e+00, 1.7271e-01, ..., -3.4062e-02,
-2.4682e-01, 7.9131e-02]]]], device='cuda:0'),
tensor([[[[ 1.5565e-02, -2.8418e-01, -6.1011e-02, ..., -1.0344e+00,
-1.9540e+00, 6.7581e-01],
[ 3.1437e-01, -9.3126e-01, -7.2798e-01, ..., -3.2835e+00,
-4.8666e+00, 4.7852e-02],
[ 2.6042e-01, -5.5236e-01, -5.0873e-01, ..., -4.9301e+00,
-3.7716e+00, -6.3301e-01],
...,
[ 4.1672e-02, -5.9223e-01, -3.7616e-01, ..., -1.2676e+00,
-1.0272e+00, -1.7976e+00],
[-6.0518e-01, -6.6854e-01, -4.8173e-02, ..., -7.7829e-01,
-5.3837e-01, -1.4781e+00],
[-7.9268e-01, -1.3139e+00, -1.1846e+00, ..., -1.3518e+00,
-1.4936e+00, -9.0398e-01]],
[[-8.1066e-01, -2.2235e+00, -2.0000e+00, ..., -9.9048e+00,
-8.0097e+00, -3.0335e+00],
[-2.0876e+00, -3.8204e+00, -3.1419e+00, ..., -1.3720e+01,
-1.2611e+01, -5.5518e+00],
[-1.6299e+00, -2.6238e+00, -2.6419e+00, ..., -1.0603e+01,
-8.0925e+00, -2.3923e+00],
...,
[-1.8295e+00, -2.4469e+00, -2.6480e+00, ..., -5.4597e+00,
-5.6890e+00, -2.5409e+00],
[-3.9056e+00, -4.4862e+00, -3.8209e+00, ..., -6.4195e+00,
-7.3549e+00, -2.7992e+00],
[ 2.7569e-01, -2.7352e-01, 3.8174e-01, ..., 4.9207e-01,
6.3817e-01, 8.0111e-02]],
[[-1.8660e-01, 1.8450e-01, 7.7949e-01, ..., 3.2391e-01,
-1.2941e+00, 1.3372e+00],
[ 1.0756e-01, -3.5612e-01, 5.8372e-01, ..., -2.2105e+00,
-7.8204e+00, -1.7679e+00],
[-1.4029e-01, -2.1533e-01, 2.0625e-01, ..., -6.6497e+00,
-1.4124e+01, -5.7634e+00],
...,
[-2.8701e-01, -1.7366e-01, 2.2800e-01, ..., 7.5605e-03,
2.8787e-02, -1.4991e-01],
[-2.4051e-01, -1.5375e-01, 1.6597e-01, ..., -1.4812e-01,
9.3047e-02, 1.1464e-01],
[-8.6689e-02, -4.5651e-01, 7.1125e-02, ..., -4.0841e-01,
-1.0834e-01, -3.1876e-01]],
...,
[[ 6.3891e-01, -2.0869e-01, 3.5250e-02, ..., 1.7865e+00,
1.6041e+00, -3.0334e+00],
[-4.8899e-01, -5.3486e-01, -5.2343e-01, ..., -1.0416e+00,
-2.6614e+00, -7.9974e+00],
[-3.4381e-01, -2.2528e-01, -2.1194e-01, ..., -5.5569e+00,
-6.2349e+00, -1.2844e+01],
...,
[-4.0221e-01, -1.5970e-01, -2.6704e-01, ..., -1.0492e+00,
-1.1544e+00, -1.3903e+00],
[-2.3142e+00, -1.7100e+00, -1.2328e+00, ..., -1.9367e+00,
-1.9664e+00, -1.8006e+00],
[-4.8350e-01, -8.2911e-01, -3.6733e-01, ..., -7.7582e-01,
-8.2002e-01, -9.1414e-01]],
[[ 1.6997e-01, 2.8416e-01, 6.3902e-01, ..., 6.5215e-01,
7.8599e-01, -4.6050e-01],
[ 1.8177e-01, 2.6365e-01, 4.6063e-01, ..., -1.0614e+00,
-2.9252e+00, -1.4931e+00],
[-1.0588e-01, 8.2278e-02, 3.1002e-01, ..., -6.8201e+00,
-7.0149e+00, -1.9945e-01],
...,
[-1.5449e-01, -2.4283e-02, 3.7352e-01, ..., 4.4675e-01,
2.4448e-01, -5.4718e-01],
[-5.5019e-01, -7.6626e-01, -4.2425e-02, ..., -8.3222e-02,
-1.0653e-01, -5.5866e-01],
[ 5.4468e-01, 4.1603e-01, 1.0448e-01, ..., 1.8092e-01,
1.5863e-01, 3.2796e-01]],
[[-8.4380e-01, 2.6611e-01, 3.2245e-01, ..., -3.3808e-01,
-8.2053e-01, -5.0713e-01],
[-1.1420e+00, 2.0587e-01, -3.4606e-01, ..., -4.2238e+00,
-3.0952e+00, -1.2729e+00],
[-9.5531e-01, 3.3270e-01, -8.3722e-02, ..., -8.4918e+00,
-7.3685e+00, -3.3136e+00],
...,
[-9.3586e-01, 1.6710e-01, -1.5164e-01, ..., 2.5868e-01,
6.6653e-01, 1.5081e+00],
[-1.6813e+00, -4.2189e-01, -3.1896e-01, ..., -1.1503e-01,
1.5127e-01, 9.0895e-01],
[-7.4511e-01, -1.4331e-01, -2.9695e-01, ..., 2.5146e-01,
1.5270e-01, 1.1459e+00]]],
[[[-1.9103e-01, -2.7221e-01, 1.4317e-01, ..., 9.1518e-02,
5.7883e-01, 2.4974e-01],
[-7.6830e-02, -1.4149e+00, -8.9819e-01, ..., -7.4956e-01,
-4.7102e-01, -2.7819e-01],
[-2.2745e-02, -1.1664e+00, -6.6982e-01, ..., -5.4614e-01,
-4.2948e-01, -6.9089e-01],
...,
[-1.9600e-01, -1.9057e+00, -9.1897e-01, ..., -6.9888e-01,
-1.0827e+00, -1.9523e+00],
[-4.6610e-01, -1.5567e+00, -6.1727e-01, ..., -3.3663e-01,
-3.8281e-01, -1.4296e+00],
[-2.5396e-01, -1.3820e+00, -1.1884e+00, ..., -8.1276e-01,
-1.1759e+00, -7.7572e-01]],
[[-2.2293e-01, -1.4362e+00, -6.8276e-01, ..., -5.4107e-01,
-1.3202e+00, -1.7725e-01],
[-1.9524e+00, -3.8855e+00, -2.5089e+00, ..., -2.4564e+00,
-3.5434e+00, -1.2403e+00],
[-1.8129e+00, -3.2426e+00, -2.5986e+00, ..., -2.6133e+00,
-3.0100e+00, -1.2700e+00],
...,
[-3.6696e+00, -5.2138e+00, -4.8462e+00, ..., -6.2698e+00,
-5.9894e+00, -2.8365e+00],
[-5.5068e+00, -6.6502e+00, -6.0082e+00, ..., -6.5016e+00,
-6.7148e+00, -2.7003e+00],
[ 5.2474e-01, -5.3262e-01, 5.1186e-01, ..., -6.5651e-02,
6.4817e-01, 1.0840e-01]],
[[-4.6447e-01, -1.2511e-01, 3.4071e-01, ..., 2.6488e-01,
6.7496e-02, 9.6080e-01],
[-3.0714e-01, -4.3917e-01, 4.4067e-01, ..., 3.9639e-01,
2.4357e-01, 4.8638e-01],
[-7.1713e-01, -2.1856e-01, 1.7122e-01, ..., 1.1391e-01,
1.6700e-01, 4.1491e-01],
...,
[-1.7028e+00, -3.9812e-01, 8.0138e-01, ..., 3.1747e-01,
-5.5116e-02, -2.1390e-01],
[-1.4358e+00, -3.0215e-01, 4.2669e-01, ..., -2.9155e-01,
-1.1519e-01, -2.9824e-01],
[-6.0931e-01, -3.6945e-02, 1.8285e-03, ..., -2.4095e-01,
-4.2607e-02, -4.6379e-01]],
...,
[[ 1.7602e-01, -5.3492e-01, -3.6847e-01, ..., -1.6333e-01,
-2.9886e-01, -8.0615e-01],
[-7.0481e-01, -6.1165e-01, -5.0737e-01, ..., -3.5043e-01,
-1.1586e+00, -8.2034e-01],
[-3.8034e-01, -1.4627e-01, -5.3783e-03, ..., 8.1368e-02,
-1.8278e-01, -5.3062e-01],
...,
[-1.0240e+00, -8.5583e-01, -5.2628e-01, ..., -1.3600e+00,
-6.3441e-01, -1.2053e+00],
[-2.6147e+00, -2.0420e+00, -1.8322e+00, ..., -2.7084e+00,
-1.5715e+00, -1.5710e+00],
[-5.6384e-01, -1.1917e+00, -2.2324e-01, ..., -1.3438e+00,
-4.2126e-01, -5.0400e-01]],
[[-1.3351e-01, 6.3475e-02, 3.9904e-01, ..., 2.1015e-01,
6.2516e-01, 2.2040e-01],
[-3.2482e-01, -2.0377e-01, 1.7739e-02, ..., -3.9338e-02,
1.2865e-01, -1.1930e-01],
[-5.4013e-01, -4.3150e-01, -1.3319e-02, ..., -1.6421e-02,
3.4290e-02, -4.2084e-02],
...,
[-1.8854e-01, -8.0587e-01, 4.5355e-01, ..., 5.6985e-01,
5.2293e-01, -2.9857e-01],
[-8.0344e-01, -1.9618e+00, 1.7022e-02, ..., 1.8241e-01,
1.3533e-01, -5.3542e-01],
[ 7.2552e-01, 1.5262e-02, 3.3907e-01, ..., 1.2134e+00,
5.0258e-01, 2.9659e-01]],
[[-6.0966e-01, 4.3060e-01, 5.4954e-01, ..., 5.4461e-01,
3.9347e-01, 6.4576e-01],
[-8.2437e-01, 2.7148e-01, -2.2269e-01, ..., -2.6903e-01,
-1.6930e-01, 6.9075e-01],
[-4.5682e-01, 3.3220e-01, 3.8069e-02, ..., 6.1030e-02,
7.5910e-02, 5.9693e-01],
...,
[-1.3265e+00, -7.6176e-01, -5.3697e-01, ..., -1.8885e-01,
-4.1422e-02, 1.4075e+00],
[-1.4300e+00, -9.3943e-01, -6.3004e-01, ..., 8.8022e-02,
4.7576e-02, 1.1947e+00],
[-6.5033e-01, -4.8051e-01, 1.4312e-01, ..., 3.7230e-01,
1.8904e-01, 1.2529e+00]]],
[[[-7.0057e-01, -1.3620e+00, -1.4538e+00, ..., -1.2577e+00,
-8.4104e-01, -1.6430e+00],
[-3.4996e-01, -1.7680e+00, -3.1679e+00, ..., -1.7620e+00,
-1.8858e+00, -1.9163e+00],
[-1.2612e-01, -1.5840e+00, -4.4938e+00, ..., -1.2690e+00,
-1.3935e+00, -2.1315e+00],
...,
[-4.9862e-01, -8.5505e-01, -2.5701e-01, ..., -1.2489e+00,
-4.8500e-01, -9.5196e-01],
[-1.0103e+00, -6.6909e-01, 4.9738e-01, ..., 2.8076e-02,
1.0484e-01, -3.3366e-01],
[-5.1256e-01, -1.9328e+00, -1.3014e+00, ..., -1.0324e+00,
-1.1132e+00, -4.7530e-01]],
[[-2.6725e+00, -5.4168e+00, -5.8086e+00, ..., -4.7812e+00,
-5.9995e+00, -2.4164e+00],
[-5.3686e+00, -1.1973e+01, -1.4402e+01, ..., -7.7559e+00,
-9.2739e+00, -4.3949e+00],
[-4.0985e+00, -8.7397e+00, -1.0705e+01, ..., -6.0654e+00,
-6.5806e+00, -3.2986e+00],
...,
[-2.0922e+00, -2.6290e+00, -2.2084e+00, ..., -2.6617e+00,
-2.4902e+00, -1.3429e+00],
[-3.3275e+00, -4.2100e+00, -2.0663e+00, ..., -2.3326e+00,
-3.3900e+00, -9.9349e-01],
[ 6.6810e-01, -1.0172e+00, -5.4699e-01, ..., 3.0461e-01,
4.3460e-01, 2.1941e-01]],
[[-6.4791e-01, 4.2883e-01, 8.0966e-01, ..., 8.2335e-01,
5.9808e-01, 1.9166e+00],
[-9.6295e-01, -7.4265e-02, -5.3208e-01, ..., 5.6877e-01,
-3.0937e-02, 1.6884e-01],
[-1.7465e+00, -6.7906e-01, -3.3257e+00, ..., 2.6583e-02,
-6.4874e-02, -8.1083e-02],
...,
[-8.5234e-01, -4.1844e-01, -1.5687e-01, ..., 1.0047e+00,
8.5265e-01, 2.1192e-01],
[-6.2068e-01, -9.5926e-01, -1.0612e+00, ..., 9.0374e-02,
1.3618e-01, 5.5063e-01],
[-5.7502e-01, -8.5022e-01, -2.4286e-01, ..., -1.7911e-02,
7.4257e-02, 3.4880e-01]],
...,
[[ 1.2438e+00, -3.6350e-01, 3.9444e-02, ..., 2.3313e-01,
-8.6936e-02, -2.1226e+00],
[-7.4453e-01, -7.6913e-01, 3.5678e-01, ..., -5.4217e-01,
-1.5023e+00, -2.0969e+00],
[-7.1127e-01, 4.2122e-01, 1.9859e+00, ..., -3.3627e-01,
-6.1997e-01, -1.3987e+00],
...,
[-5.9164e-01, -5.9761e-01, -4.1151e-01, ..., -3.1283e+00,
-2.7110e+00, -1.5266e+00],
[-3.2987e+00, -3.2010e+00, -1.7048e+00, ..., -1.9357e+00,
-1.6624e+00, -1.0507e+00],
[-9.0730e-01, -1.8058e+00, -1.3620e+00, ..., -2.8851e-01,
-3.4274e-01, -3.8829e-01]],
[[ 5.3692e-02, 2.8553e-02, 6.2499e-01, ..., 5.0304e-01,
9.3164e-01, -2.1649e-02],
[ 2.5320e-01, -2.9536e-02, 3.0259e-01, ..., 4.6350e-01,
3.8360e-01, -8.2255e-01],
[ 1.9750e-01, -1.7324e-01, -4.2696e-01, ..., 6.4938e-01,
5.3079e-01, -4.1229e-01],
...,
[-6.0957e-01, -6.3559e-01, -1.1784e-01, ..., -5.9341e-01,
9.2707e-01, -3.1105e-01],
[-1.1090e+00, -1.5737e+00, -1.1942e+00, ..., -4.7311e-01,
-1.9631e-01, -4.0444e-01],
[ 3.0699e-01, -3.0612e-01, -8.0469e-01, ..., -1.1399e-01,
-6.8310e-02, 1.6653e-01]],
[[-1.5551e+00, -8.0307e-01, -4.9738e-01, ..., -4.5129e-01,
-2.7598e-01, 8.5587e-01],
[-2.1758e+00, -1.0062e+00, -1.7674e+00, ..., -8.0957e-01,
-1.9934e-01, 1.6494e+00],
[-1.5009e+00, -1.9277e+00, -3.9833e+00, ..., -4.0812e-01,
-1.1791e-01, 1.4071e+00],
...,
[-9.8468e-01, -8.3956e-01, -1.2605e+00, ..., -3.3351e-01,
2.8566e-01, 1.4483e+00],
[-1.2369e+00, -5.8763e-01, -1.5688e+00, ..., -1.7691e-01,
1.0578e-01, 2.3343e-01],
[-1.2211e-01, 2.8815e-02, -8.2443e-01, ..., -2.6316e-01,
-1.8907e-01, 3.5449e-01]]],
[[[-7.5436e-01, -1.4202e+00, -1.3864e+00, ..., -3.3777e-03,
6.4556e-01, 5.5164e-01],
[-2.8781e-01, -2.4091e+00, -2.1536e+00, ..., -4.8538e-01,
-2.1183e-01, 1.1625e-01],
[-4.4723e-01, -2.3945e+00, -2.6037e+00, ..., -3.1665e-01,
-1.0245e-01, -8.8175e-02],
...,
[-4.7155e-01, -1.9090e+00, -1.8534e+00, ..., -2.7932e-01,
-8.0024e-02, -1.7055e-01],
[-9.5882e-01, -1.6608e+00, -1.2137e+00, ..., 2.6012e-02,
2.7369e-01, -3.6166e-02],
[-7.7841e-01, -1.5477e+00, -1.2106e+00, ..., -1.0900e+00,
-1.0816e+00, -3.9575e-01]],
[[-2.3823e+00, -4.7856e+00, -4.1382e+00, ..., -4.3618e-01,
-1.0095e+00, 2.2887e-02],
[-5.3318e+00, -8.0657e+00, -6.5180e+00, ..., -1.8125e+00,
-2.7210e+00, -6.9118e-01],
[-5.3782e+00, -8.4851e+00, -8.8303e+00, ..., -1.2538e+00,
-1.5060e+00, -3.6791e-01],
...,
[-4.3727e+00, -6.0866e+00, -5.4646e+00, ..., -1.2589e+00,
-1.5279e+00, -4.1155e-01],
[-6.3880e+00, -7.4609e+00, -6.3390e+00, ..., -2.4219e+00,
-3.3312e+00, -9.4974e-01],
[ 3.2503e-03, -7.9215e-01, 3.7693e-01, ..., 2.9655e-01,
3.4815e-01, 7.8055e-02]],
[[-5.8883e-01, 6.0190e-01, 1.0982e+00, ..., 8.0633e-01,
2.8067e-01, 1.0896e+00],
[-9.2853e-01, -6.0171e-02, 4.0222e-01, ..., 4.2925e-01,
6.0172e-02, 4.7717e-01],
[-1.5116e+00, 3.2035e-01, 7.1342e-01, ..., 5.7933e-02,
1.0437e-01, 4.2510e-01],
...,
[-1.7151e+00, -6.0842e-02, -7.7320e-02, ..., -2.7508e-02,
1.8899e-02, 4.3956e-01],
[-1.5097e+00, 1.0003e-01, 7.5661e-02, ..., 5.8979e-02,
2.7083e-02, 6.1670e-01],
[-1.4291e+00, -5.4384e-01, -2.9205e-01, ..., -6.8331e-02,
1.9542e-02, 3.3555e-01]],
...,
[[ 1.0423e+00, -4.1929e-01, -2.0245e-01, ..., -1.1532e-01,
-3.7937e-01, -9.4956e-01],
[-8.2092e-01, -9.7735e-01, -8.1160e-01, ..., -6.9288e-01,
-1.2873e+00, -8.4410e-01],
[-5.8783e-01, -8.9690e-01, -6.6976e-01, ..., -1.0739e-01,
-1.6046e-01, -3.8463e-01],
...,
[-7.2307e-01, -1.1285e+00, -7.6629e-01, ..., -1.4858e-01,
-1.7251e-01, -3.6677e-01],
[-2.6850e+00, -2.7601e+00, -1.7982e+00, ..., -8.5136e-01,
-8.6570e-01, -7.7904e-01],
[-6.2266e-01, -1.6248e+00, -9.5792e-01, ..., -4.1486e-02,
-1.8225e-01, -3.3469e-01]],
[[ 8.3400e-03, 1.7179e-03, 5.8007e-01, ..., 3.1611e-01,
6.8150e-01, 2.9946e-01],
[-2.7848e-02, -7.4472e-01, 1.1558e-01, ..., -4.0678e-03,
1.1770e-01, -6.0608e-02],
[ 1.5332e-01, -5.1815e-01, 3.0405e-01, ..., -1.4517e-01,
-6.4777e-02, 2.0471e-01],
...,
[ 4.3412e-02, -8.1266e-01, 3.2518e-01, ..., -1.6084e-01,
-8.9801e-02, 1.2970e-01],
[-2.5034e-01, -1.4619e+00, 2.9870e-02, ..., -4.4811e-01,
-3.3853e-01, -6.5145e-02],
[ 7.0718e-01, 2.0533e-01, 3.1834e-01, ..., -2.8437e-01,
-2.2910e-01, 2.9433e-01]],
[[-1.5948e+00, -6.0946e-01, 3.7863e-02, ..., 5.5189e-01,
3.8052e-01, 5.7034e-01],
[-2.0536e+00, -3.4927e-01, -4.1341e-02, ..., -6.8106e-02,
-1.5129e-01, 4.2616e-01],
[-1.8241e+00, -7.7574e-01, -3.9363e-01, ..., 5.0042e-02,
-1.5255e-01, 1.0700e-01],
...,
[-1.7336e+00, -4.8903e-01, 2.6488e-01, ..., 3.1317e-03,
-7.1216e-02, 8.1913e-02],
[-2.1486e+00, -1.0789e+00, -1.0784e-01, ..., -1.6919e-01,
-2.5580e-01, -1.9793e-01],
[-1.0817e+00, -1.8131e-01, 4.9440e-01, ..., -1.4710e-01,
-2.7694e-01, 1.1568e-01]]],
[[[-7.3887e-01, -1.4387e+00, -1.4150e+00, ..., -1.2487e+00,
-8.4513e-01, -1.6383e+00],
[-3.6282e-01, -2.4093e+00, -2.2513e+00, ..., -1.7438e+00,
-1.8763e+00, -1.9333e+00],
[-1.4146e-01, -1.8923e+00, -1.6926e+00, ..., -1.2948e+00,
-1.3785e+00, -2.0993e+00],
...,
[-9.7287e-02, -1.1232e+00, -6.0582e-01, ..., -5.2184e-01,
-4.5412e-01, -9.0628e-01],
[-1.6253e-01, -1.0309e+00, -3.1352e-01, ..., -1.8757e-01,
-4.6136e-02, -9.6122e-01],
[-1.4563e-01, -1.1413e+00, -8.4962e-01, ..., -8.9693e-01,
-1.1339e+00, -6.3668e-01]],
[[-2.6919e+00, -5.1123e+00, -4.6121e+00, ..., -4.7538e+00,
-5.9634e+00, -2.3866e+00],
[-5.3164e+00, -8.1347e+00, -7.4631e+00, ..., -7.7717e+00,
-9.2840e+00, -4.3786e+00],
[-3.9816e+00, -5.7488e+00, -5.6588e+00, ..., -6.0878e+00,
-6.4055e+00, -3.2281e+00],
...,
[-1.9730e+00, -3.1433e+00, -3.0644e+00, ..., -2.9666e+00,
-3.4668e+00, -1.5314e+00],
[-4.5082e+00, -5.5228e+00, -5.0051e+00, ..., -4.9415e+00,
-6.1689e+00, -2.3596e+00],
[ 6.8650e-01, -3.7190e-01, 5.0360e-01, ..., 4.7354e-01,
5.5857e-01, -5.2319e-03]],
[[-6.2656e-01, 5.0125e-01, 8.6960e-01, ..., 8.4750e-01,
6.4780e-01, 1.9239e+00],
[-9.6793e-01, 4.7157e-02, 4.1287e-01, ..., 5.2604e-01,
-1.1738e-02, 1.2120e-01],
[-1.7815e+00, -1.9160e-01, -5.8997e-02, ..., 1.8857e-02,
3.2653e-03, -8.9581e-02],
...,
[-9.2977e-01, -1.3557e-01, 8.6696e-02, ..., -1.9185e-02,
9.6421e-02, 3.9147e-01],
[-9.8154e-01, 3.8677e-02, 1.6322e-02, ..., 9.2106e-03,
5.9019e-02, 3.6625e-01],
[-4.5118e-01, -9.9023e-03, 1.5642e-01, ..., -2.1952e-02,
7.6426e-02, 3.2515e-03]],
...,
[[ 1.2569e+00, -3.3126e-01, -5.2381e-02, ..., 2.2823e-01,
-1.1212e-01, -2.1228e+00],
[-7.1533e-01, -9.0440e-01, -4.7369e-01, ..., -5.3606e-01,
-1.5094e+00, -2.0697e+00],
[-7.0646e-01, -8.7745e-01, -2.8841e-01, ..., -2.9902e-01,
-5.7924e-01, -1.3594e+00],
...,
[-4.6258e-01, -1.2291e-01, 1.0912e-01, ..., 1.2607e-01,
-1.1429e-01, -5.6673e-01],
[-2.0747e+00, -1.3851e+00, -7.0458e-01, ..., -9.3762e-01,
-1.1349e+00, -1.1154e+00],
[-3.0993e-01, -7.9254e-01, -1.1143e-01, ..., -2.0255e-01,
-3.4281e-01, -6.1989e-01]],
[[ 7.6614e-02, 3.7574e-02, 4.5136e-01, ..., 4.6495e-01,
9.3979e-01, -8.4289e-03],
[ 2.8928e-01, -7.4506e-01, -4.4803e-02, ..., 4.1866e-01,
3.5603e-01, -8.8261e-01],
[ 1.8106e-01, -5.9760e-01, 3.0549e-01, ..., 4.4945e-01,
5.6633e-01, -3.7889e-01],
...,
[-4.0894e-01, -5.4080e-01, 1.3813e-01, ..., 1.3161e-01,
4.6693e-02, -1.2674e-01],
[-7.8161e-01, -1.5594e+00, -2.6968e-01, ..., -2.9701e-01,
-2.9185e-01, -6.9075e-01],
[ 5.0664e-01, 1.4334e-02, 1.5027e-01, ..., 1.9166e-01,
1.5936e-01, 3.2406e-01]],
[[-1.5210e+00, -7.6278e-01, -4.0911e-01, ..., -4.4139e-01,
-2.8267e-01, 8.6182e-01],
[-2.1460e+00, -7.7585e-01, -5.3863e-01, ..., -7.7113e-01,
-1.8072e-01, 1.6610e+00],
[-1.4949e+00, -5.5120e-01, -2.1100e-01, ..., -3.6025e-01,
-1.1476e-01, 1.3961e+00],
...,
[-4.5121e-01, 1.2948e-01, 2.3705e-02, ..., 1.2949e-01,
2.0702e-01, 7.2447e-01],
[-9.2618e-01, -3.1890e-01, -2.8575e-01, ..., -1.2173e-01,
6.4057e-02, 4.6279e-01],
[-4.2473e-01, -3.2020e-01, -2.7334e-01, ..., -2.5608e-01,
-2.6759e-01, 5.7685e-01]]]], device='cuda:0'),
tensor([[[[-3.4900e-01, -8.4022e-01, -1.3696e+00, ..., 2.3053e-01,
6.4866e-01, -7.4418e-01],
[ 1.9731e-02, -1.4881e+00, -1.4893e+00, ..., 1.3002e-01,
-1.5717e-01, -1.1192e+00],
[ 4.2349e-02, -1.3177e+00, -1.1980e+00, ..., 1.9995e-01,
-4.1485e-01, -1.7037e+00],
...,
[-5.4897e-02, -8.2104e-01, -2.6119e-01, ..., -1.2407e+00,
-2.9103e-01, -6.4828e-01],
[-7.4621e-01, -7.2959e-01, 7.9341e-02, ..., 1.3426e-01,
2.4032e-01, -3.8086e-01],
[-9.1031e-01, -1.5809e+00, -1.4771e+00, ..., -1.0527e+00,
-1.1237e+00, -4.6805e-01]],
[[-2.0224e+00, -4.5018e+00, -4.4394e+00, ..., -4.4505e+00,
-6.1142e+00, -1.9558e+00],
[-3.8022e+00, -6.2288e+00, -5.9788e+00, ..., -5.6544e+00,
-8.4727e+00, -2.9821e+00],
[-3.4820e+00, -5.7569e+00, -6.3209e+00, ..., -6.3591e+00,
-7.2975e+00, -2.8195e+00],
...,
[-1.9445e+00, -2.6514e+00, -1.9266e+00, ..., -1.2140e+00,
-2.3838e+00, -1.4095e+00],
[-3.2798e+00, -3.7461e+00, -2.5130e+00, ..., -2.2697e+00,
-3.5794e+00, -1.1819e+00],
[ 2.6451e-01, -5.4169e-01, 1.8435e-01, ..., 2.9685e-01,
4.2389e-01, 2.0131e-01]],
[[-4.9255e-01, 4.8696e-01, 7.9399e-02, ..., 3.8652e-01,
1.1091e+00, 1.4685e+00],
[-6.8191e-01, 1.2333e-01, -1.1519e+00, ..., -2.5463e+00,
8.6353e-01, 3.8088e-01],
[-1.1808e+00, 1.8045e-01, -1.2223e+00, ..., -3.7779e+00,
5.1886e-01, 2.9660e-01],
...,
[-7.3495e-01, -2.0796e-01, -1.9174e-02, ..., -5.4805e-02,
-1.0592e+00, 4.1289e-01],
[-5.6291e-01, -2.1549e-01, 2.8786e-01, ..., 2.4972e-01,
-4.1280e-01, 6.6909e-01],
[-2.3134e-01, -4.9185e-01, -5.4934e-02, ..., -1.6884e-03,
1.1513e-01, 3.2578e-01]],
...,
[[ 1.0692e+00, 4.6812e-01, -1.9852e-03, ..., -1.6948e-01,
-3.9119e-01, -1.0385e+00],
[-7.4006e-01, -3.2887e-01, -2.1987e-01, ..., -2.8031e+00,
-2.8104e+00, -1.3749e+00],
[-2.3284e-01, -2.4716e-01, 3.6902e-01, ..., -3.7388e+00,
-2.4147e+00, -1.1358e+00],
...,
[-7.9396e-01, -4.3548e-01, -2.5675e-01, ..., -3.7087e-01,
-1.1709e+00, -9.3065e-01],
[-2.6671e+00, -2.0848e+00, -1.5266e+00, ..., -1.8324e+00,
-1.5617e+00, -9.9045e-01],
[-6.4808e-01, -1.1245e+00, -5.9337e-01, ..., -2.4427e-01,
-3.5679e-01, -4.0241e-01]],
[[ 1.6892e-01, 4.0881e-01, 2.0534e-03, ..., 1.0367e+00,
1.1599e+00, -2.3132e-01],
[ 3.3197e-01, -4.8881e-01, -5.2314e-01, ..., 3.5874e-01,
6.5397e-01, -7.5957e-01],
[ 3.2222e-02, -5.4903e-01, -1.6879e-01, ..., -3.9082e-01,
-3.1156e-01, -6.7652e-01],
...,
[-3.7109e-01, -6.4514e-01, 2.3196e-02, ..., 1.3409e+00,
1.9193e+00, 4.9096e-01],
[-7.6864e-01, -1.0384e+00, -6.6762e-02, ..., -1.6456e-01,
-4.8274e-03, -2.4552e-01],
[ 3.2824e-01, 2.1896e-02, -2.2609e-01, ..., -1.2453e-01,
-9.6679e-02, 1.7822e-01]],
[[-6.8954e-01, -3.5644e-01, 3.9367e-01, ..., -3.3135e-01,
6.6147e-01, 9.8403e-01],
[-1.1601e+00, -9.2400e-01, 2.3762e-01, ..., -2.0961e+00,
-1.0596e-01, 1.0008e+00],
[-8.7927e-01, -8.8722e-01, -5.2320e-02, ..., -1.7842e+00,
1.6893e-01, 7.2870e-01],
...,
[-7.0284e-01, -2.7120e-01, -3.9361e-01, ..., -1.0547e+00,
2.9347e-01, 1.0003e+00],
[-1.3728e+00, -7.5035e-01, -5.9757e-01, ..., 5.5222e-01,
4.7879e-01, 4.6656e-01],
[-5.2236e-01, -1.2149e-01, -2.2525e-01, ..., -2.1361e-01,
-2.2828e-01, 3.2505e-01]]],
[[[-7.2175e-01, -1.4059e+00, -1.4192e+00, ..., -1.9175e-02,
5.9274e-01, 5.3636e-01],
[-2.8604e-01, -2.4010e+00, -2.1883e+00, ..., -2.8696e-01,
-2.5965e-01, 1.1325e-01],
[-3.9380e-01, -2.0870e+00, -1.8664e+00, ..., 1.4622e-02,
-1.7408e-01, -1.1066e-01],
...,
[-4.2092e-01, -1.6064e+00, -1.6632e+00, ..., -3.2747e-01,
-9.0124e-02, -1.3906e-01],
[-9.2870e-01, -1.6661e+00, -1.1470e+00, ..., 9.4090e-03,
2.5368e-01, -2.2587e-02],
[-7.7077e-01, -1.6461e+00, -1.2450e+00, ..., -1.0100e+00,
-1.0711e+00, -3.7871e-01]],
[[-2.3708e+00, -4.8152e+00, -4.1372e+00, ..., -8.2414e-01,
-1.0167e+00, -1.1226e-02],
[-5.3681e+00, -8.1456e+00, -6.5628e+00, ..., -2.7186e+00,
-2.6700e+00, -7.0138e-01],
[-4.3449e+00, -6.3015e+00, -5.5038e+00, ..., -2.6322e+00,
-1.5751e+00, -4.2740e-01],
...,
[-4.3914e+00, -6.3046e+00, -5.8025e+00, ..., -1.3359e+00,
-1.5891e+00, -3.8054e-01],
[-6.3803e+00, -7.4589e+00, -6.4173e+00, ..., -2.4480e+00,
-3.3439e+00, -9.6372e-01],
[-1.2309e-02, -7.7613e-01, 3.8806e-01, ..., 3.4703e-01,
3.6310e-01, 8.2291e-02]],
[[-5.8081e-01, 6.0806e-01, 1.0985e+00, ..., 8.6509e-01,
3.3462e-01, 1.0983e+00],
[-8.7615e-01, -4.1462e-02, 4.2349e-01, ..., 5.3541e-01,
5.7065e-02, 4.7192e-01],
[-1.6326e+00, -9.3290e-02, -5.0318e-02, ..., 2.5935e-01,
9.9777e-02, 4.2361e-01],
...,
[-1.6220e+00, 5.0100e-01, -1.0834e-01, ..., 5.4958e-02,
7.2004e-02, 4.3567e-01],
[-1.5212e+00, 1.2878e-01, 9.2649e-02, ..., 1.4742e-01,
1.2405e-01, 6.2234e-01],
[-1.4382e+00, -5.2369e-01, -2.6431e-01, ..., -7.8673e-02,
5.4490e-02, 3.4956e-01]],
...,
[[ 1.0638e+00, -4.3582e-01, -1.9917e-01, ..., 3.6324e-02,
-3.9063e-01, -9.3231e-01],
[-8.3295e-01, -9.7660e-01, -8.0717e-01, ..., -9.0183e-01,
-1.3034e+00, -8.3096e-01],
[-6.6197e-01, -9.8314e-01, -7.0723e-01, ..., -7.7012e-01,
-2.1861e-01, -3.8484e-01],
...,
[-8.4323e-01, -1.5170e+00, -1.2092e+00, ..., -1.1659e-01,
-1.8579e-01, -3.6855e-01],
[-2.7135e+00, -2.8436e+00, -1.8453e+00, ..., -8.1050e-01,
-8.7332e-01, -7.9730e-01],
[-6.5927e-01, -1.6344e+00, -9.9115e-01, ..., 9.7527e-03,
-1.2894e-01, -3.2453e-01]],
[[ 9.4707e-04, -8.0414e-03, 5.8924e-01, ..., 5.6160e-02,
6.8594e-01, 3.0433e-01],
[ 1.0357e-02, -7.1009e-01, 9.3004e-02, ..., -2.0441e-01,
1.2508e-01, -6.2855e-02],
[-2.7451e-02, -7.9488e-01, 2.7243e-01, ..., -8.1979e-02,
-1.0907e-01, 1.7379e-01],
...,
[ 6.5292e-02, -6.8680e-01, 2.0230e-01, ..., -1.3855e-01,
-1.1335e-01, 1.4551e-01],
[-2.2660e-01, -1.4026e+00, 1.0905e-04, ..., -3.7034e-01,
-3.0861e-01, -4.7040e-02],
[ 6.6628e-01, 1.4027e-01, 2.6900e-01, ..., -2.1253e-01,
-2.2182e-01, 3.1236e-01]],
[[-1.6227e+00, -5.9654e-01, 1.4987e-02, ..., 6.1406e-01,
3.7720e-01, 5.6142e-01],
[-2.0863e+00, -3.0516e-01, -6.3518e-02, ..., 1.6263e-02,
-1.2726e-01, 4.1406e-01],
[-1.7053e+00, -3.4451e-01, 1.7575e-01, ..., 3.0567e-01,
-1.5904e-01, 8.4346e-02],
...,
[-1.7825e+00, -6.6979e-01, 2.3620e-01, ..., 5.4153e-02,
-6.4328e-02, 9.3341e-02],
[-2.1615e+00, -1.0916e+00, -1.7214e-01, ..., -9.5993e-02,
-2.4573e-01, -2.1273e-01],
[-1.0589e+00, -1.7637e-01, 4.4311e-01, ..., -9.5690e-02,
-2.6321e-01, 1.1429e-01]]],
[[[ 2.5860e-02, -2.4115e-01, -3.0394e-02, ..., -1.6775e+00,
-9.8879e-01, -1.0501e+00],
[ 3.0748e-01, -1.1262e+00, -7.6168e-01, ..., -3.2500e+00,
-1.5301e+00, 3.9127e-01],
[ 1.1137e-01, -1.1117e+00, -6.4459e-01, ..., -3.4156e+00,
-2.2672e-01, 8.7769e-01],
...,
[ 3.1769e-01, -5.7707e-01, -6.1629e-01, ..., -1.0296e+00,
-1.3607e+00, -1.7178e+00],
[-5.2451e-01, -6.5023e-01, -9.6827e-02, ..., -1.2445e+00,
-7.3662e-01, -1.4503e+00],
[-6.0727e-01, -1.2087e+00, -9.8011e-01, ..., -1.5201e+00,
-1.5454e+00, -8.8636e-01]],
[[-7.1643e-01, -2.1895e+00, -1.8151e+00, ..., -5.3577e+00,
-6.2615e+00, -2.3827e+00],
[-2.2888e+00, -4.2137e+00, -3.0882e+00, ..., -8.8215e+00,
-8.1699e+00, -2.2806e+00],
[-3.3129e+00, -4.4181e+00, -3.6039e+00, ..., -6.0867e+00,
-6.1069e+00, -1.1213e+00],
...,
[-1.5491e+00, -2.4776e+00, -2.5970e+00, ..., -6.5657e+00,
-6.0767e+00, -2.4159e+00],
[-3.5954e+00, -4.0814e+00, -3.3956e+00, ..., -7.1349e+00,
-7.5854e+00, -2.8042e+00],
[ 2.8350e-01, -2.8553e-01, 4.1262e-01, ..., 4.5952e-01,
5.9076e-01, 4.9236e-02]],
[[-2.8962e-01, 5.0893e-02, 7.9537e-01, ..., 8.7576e-01,
8.1449e-01, 1.6865e+00],
[ 3.1852e-02, -4.6407e-01, 6.4403e-01, ..., -4.8073e-01,
-7.9951e-02, 5.4906e-01],
[-4.9586e-01, -2.9022e-01, 2.9502e-01, ..., -3.1308e+00,
-2.8002e+00, -1.5458e+00],
...,
[ 3.0495e-01, -6.8223e-01, 4.4227e-01, ..., -6.0904e-01,
3.9076e-01, -1.3664e-01],
[-1.9847e-01, -1.3434e-01, 1.2357e-01, ..., -8.9012e-02,
2.4676e-01, 1.0695e-01],
[-7.5409e-02, -3.9199e-01, 1.0674e-01, ..., -3.6635e-01,
-8.9059e-02, -3.0458e-01]],
...,
[[ 4.4164e-01, -3.8082e-01, -2.2855e-02, ..., 1.4671e-02,
2.0702e-04, -1.2289e+00],
[-6.9305e-01, -5.6602e-01, -6.1126e-01, ..., -2.9840e-01,
-2.2300e+00, -2.6685e+00],
[-7.7930e-01, -9.6888e-02, -5.6116e-02, ..., -2.5185e+00,
-5.1908e+00, -4.7110e+00],
...,
[-1.1039e+00, -8.4973e-01, -1.8468e-01, ..., -3.2562e+00,
-1.4751e+00, -1.2580e+00],
[-2.1665e+00, -1.5225e+00, -1.1479e+00, ..., -2.8317e+00,
-1.9723e+00, -1.7750e+00],
[-4.4390e-01, -6.7470e-01, -3.7027e-01, ..., -7.2372e-01,
-7.5916e-01, -9.2680e-01]],
[[ 1.1148e-01, 2.6683e-01, 6.8178e-01, ..., 3.4658e-01,
3.9117e-01, -3.0579e-01],
[ 2.0742e-02, 2.6764e-01, 4.3763e-01, ..., -8.4756e-01,
-8.2567e-01, -1.9268e-01],
[-1.3287e-01, -5.2473e-01, -1.5878e-01, ..., -2.4065e+00,
-2.0364e+00, -7.6587e-01],
...,
[-3.3607e-01, -1.9025e-01, 1.0134e-01, ..., 8.2634e-01,
1.3347e-01, -3.7052e-01],
[-4.8141e-01, -6.3386e-01, 5.6299e-02, ..., -3.1868e-01,
-2.5562e-01, -5.9903e-01],
[ 5.2873e-01, 4.3391e-01, 1.6749e-01, ..., -1.1817e-01,
6.9553e-02, 3.1474e-01]],
[[-9.5880e-01, 2.6399e-01, 3.7257e-01, ..., 1.6696e-01,
2.9862e-01, 8.9834e-01],
[-1.1692e+00, 2.4875e-01, -3.6854e-01, ..., -7.8349e-01,
-1.1761e+00, 9.0039e-02],
[-1.7938e+00, -1.2509e-01, -1.8047e-01, ..., -1.4435e+00,
-2.6671e+00, -1.7377e+00],
...,
[-1.0127e+00, 5.8071e-01, 2.7136e-01, ..., 3.0225e-01,
7.2832e-01, 1.2988e+00],
[-1.6195e+00, -3.3635e-01, -2.7559e-01, ..., 2.6361e-01,
4.3337e-01, 8.2855e-01],
[-6.5373e-01, 2.9369e-02, -1.9289e-01, ..., 4.1408e-01,
1.9647e-01, 1.0719e+00]]],
[[[-3.2747e-01, -4.8547e-01, -1.3165e-01, ..., -9.6561e-01,
-5.1665e-01, -1.4297e+00],
[-1.0606e-02, -1.4096e+00, -6.9809e-01, ..., -1.1618e+00,
-1.3093e+00, -1.4822e+00],
[-2.2780e-01, -1.4029e+00, -5.5157e-01, ..., -9.3241e-01,
-1.0979e+00, -1.8316e+00],
...,
[-3.0369e-01, -1.3334e+00, -5.3792e-01, ..., -9.4620e-01,
-1.0055e+00, -1.9202e+00],
[-4.6692e-01, -1.0611e+00, -9.0468e-02, ..., -4.8594e-01,
-5.3693e-01, -1.6955e+00],
[-5.4537e-01, -1.5764e+00, -1.2698e+00, ..., -1.1263e+00,
-1.5228e+00, -7.5789e-01]],
[[-6.5115e-01, -1.9355e+00, -8.0057e-01, ..., -4.0102e+00,
-5.4941e+00, -1.8932e+00],
[-2.1972e+00, -3.9871e+00, -2.0391e+00, ..., -5.8517e+00,
-7.9727e+00, -3.3986e+00],
[-2.0131e+00, -3.1064e+00, -1.7847e+00, ..., -4.9593e+00,
-6.1211e+00, -2.5804e+00],
...,
[-1.7588e+00, -2.9081e+00, -1.5880e+00, ..., -5.0189e+00,
-6.3057e+00, -2.5000e+00],
[-3.7263e+00, -4.7209e+00, -2.9629e+00, ..., -6.4482e+00,
-7.9427e+00, -2.5549e+00],
[ 5.7857e-01, -6.4342e-01, 3.8061e-01, ..., 4.0859e-01,
4.5856e-01, 2.1556e-01]],
[[-3.7475e-01, 7.2392e-01, 1.0089e+00, ..., 6.0792e-01,
2.4389e-01, 1.5314e+00],
[-7.6221e-01, 2.3547e-01, 5.4524e-01, ..., 5.8945e-01,
-1.1253e-01, 8.4595e-02],
[-1.3413e+00, 3.1600e-01, 2.3759e-01, ..., 5.6177e-02,
-1.9449e-02, -1.6957e-01],
...,
[-1.4159e+00, 2.5937e-01, 2.0006e-01, ..., -2.3889e-02,
-1.1958e-01, -1.5309e-01],
[-1.4074e+00, 2.6979e-01, 2.3817e-01, ..., -1.2172e-01,
-2.3998e-02, -1.6863e-01],
[-8.8344e-01, -6.5202e-03, -1.8594e-01, ..., 8.2235e-02,
2.1446e-01, -5.6372e-01]],
...,
[[ 3.8793e-01, -1.4120e-01, -1.3298e-01, ..., 4.8859e-01,
3.9965e-01, -1.3142e+00],
[-1.3904e+00, -1.0488e+00, -6.0239e-01, ..., -1.4918e-01,
-7.5351e-01, -1.4164e+00],
[-7.4336e-01, -5.1126e-01, -1.4219e-01, ..., -7.8414e-02,
-1.0141e-01, -9.7867e-01],
...,
[-9.2669e-01, -6.0393e-01, -1.3053e-01, ..., -1.1210e-01,
-1.3589e-01, -9.3808e-01],
[-2.4321e+00, -1.7016e+00, -8.7023e-01, ..., -1.5429e+00,
-1.5613e+00, -1.8063e+00],
[-5.8967e-01, -1.1666e+00, -1.2395e-01, ..., -3.8207e-01,
-5.5412e-01, -1.1879e+00]],
[[ 2.8909e-02, 2.4105e-01, 4.7357e-01, ..., -8.5199e-02,
4.2964e-01, -3.6670e-01],
[-1.6740e-01, -6.6979e-01, -4.8222e-02, ..., 2.1639e-01,
1.5065e-01, -8.2785e-01],
[-6.0313e-01, -1.0762e+00, -2.1933e-01, ..., 2.4814e-01,
1.8202e-01, -4.3777e-01],
...,
[-5.8267e-01, -1.1785e+00, -1.8688e-01, ..., 2.2869e-01,
1.5571e-01, -5.0811e-01],
[-1.0107e+00, -1.9059e+00, -4.7414e-01, ..., -1.8998e-01,
-2.6674e-01, -9.1647e-01],
[ 2.5319e-01, -3.9109e-01, -2.4434e-01, ..., 3.1820e-01,
3.3376e-01, 2.9675e-01]],
[[-3.6415e-01, 1.3425e-01, 5.4233e-01, ..., -1.1280e-01,
-9.4404e-02, 5.7412e-01],
[-3.4816e-01, 3.2156e-03, 1.9931e-02, ..., -9.7829e-01,
-4.5355e-01, 8.7958e-01],
[ 5.9538e-02, 3.8682e-02, 6.9735e-02, ..., -6.0324e-01,
-5.2872e-01, 6.3966e-01],
...,
[ 2.2103e-02, -3.6260e-02, 5.9960e-02, ..., -5.6028e-01,
-4.8892e-01, 6.1634e-01],
[-3.8090e-01, -4.8756e-01, -1.0100e-01, ..., -7.2575e-01,
-6.7001e-01, 3.9364e-01],
[-2.3914e-02, -5.6273e-01, -1.9195e-01, ..., -5.0406e-01,
-4.3477e-01, 9.8967e-01]]],
[[[-7.1643e-01, -4.6786e+00, -6.7042e+00, ..., -4.8224e-01,
1.5203e-01, 8.0209e-01],
[-4.1728e-01, -5.5179e+00, -5.5857e+00, ..., -7.8222e-01,
-5.3317e-01, 1.4009e-01],
[-5.5908e-01, -3.5918e+00, -6.7311e+00, ..., -3.7797e-01,
-2.9135e-01, 4.8206e-01],
...,
[-5.2075e-01, -2.0151e+00, -1.9124e+00, ..., -1.1972e+00,
-8.0003e-01, -2.9114e-01],
[-1.0936e+00, -1.8260e+00, -1.2557e+00, ..., -1.5091e+00,
1.8491e-01, -4.6311e-01],
[-8.8687e-01, -1.7644e+00, -1.3251e+00, ..., -1.9324e+00,
-1.4890e+00, -9.3384e-01]],
[[-4.1344e+00, -1.0712e+01, -8.4330e+00, ..., -1.9047e+00,
-2.8726e+00, -8.5266e-01],
[-6.3248e+00, -9.1898e+00, -9.6406e+00, ..., -2.7954e+00,
-3.8409e+00, -2.2059e+00],
[-6.7762e+00, -1.3622e+01, -2.0092e+01, ..., -2.2540e+00,
-2.1222e+00, -1.0431e+00],
...,
[-4.4888e+00, -6.2759e+00, -5.7132e+00, ..., -3.7912e+00,
-2.3440e+00, -8.8182e-01],
[-6.4933e+00, -7.5934e+00, -6.4768e+00, ..., -2.4222e+00,
-3.7851e+00, -1.4462e+00],
[-9.4997e-02, -8.8352e-01, 3.5153e-01, ..., -8.7853e-01,
-5.6912e-01, -2.4518e-01]],
[[-7.4346e-01, -1.3502e+00, -5.6720e+00, ..., 7.3405e-01,
-1.3770e-01, -3.3203e-02],
[-1.6838e+00, -2.3942e+00, -1.3706e+01, ..., -1.1121e-02,
-5.0451e-01, -9.9554e-01],
[-1.2906e+00, -2.8080e+00, -1.7189e+01, ..., -1.9176e+00,
-1.0899e+00, -1.2261e+00],
...,
[-1.7448e+00, -1.1540e-01, 9.7289e-02, ..., -5.7638e+00,
-3.3744e+00, 4.4072e-01],
[-1.5081e+00, 4.1826e-02, 6.4883e-03, ..., -1.4160e+00,
-1.1296e+00, 6.8239e-01],
[-1.4340e+00, -5.6012e-01, -3.3401e-01, ..., -3.6455e-01,
5.8071e-02, 6.2325e-01]],
...,
[[ 1.4575e+00, 2.1331e+00, -8.6260e-01, ..., -3.2317e-01,
-1.1166e+00, -1.9967e+00],
[ 1.6564e-02, -9.0368e-01, -7.9679e+00, ..., -1.7168e+00,
-2.3746e+00, -2.5458e+00],
[-5.0847e-01, -1.7257e+00, -6.1675e+00, ..., -1.2136e+00,
-1.8423e+00, -2.7274e+00],
...,
[-7.3881e-01, -1.1498e+00, -8.7519e-01, ..., -7.0627e+00,
-4.6960e+00, -1.9717e+00],
[-2.8583e+00, -2.9564e+00, -1.9296e+00, ..., -6.0925e+00,
-4.3772e+00, -1.9508e+00],
[-6.9055e-01, -1.7333e+00, -1.0329e+00, ..., -4.1162e+00,
-2.0018e+00, -3.6434e-01]],
[[ 4.2716e-01, -2.3982e+00, -6.1840e+00, ..., 1.0389e-01,
9.3533e-02, 2.7628e-01],
[-7.2333e-02, -5.1198e+00, -7.9054e+00, ..., -2.2799e-01,
-3.6396e-01, -4.2532e-01],
[ 7.3000e-02, -3.6759e+00, -8.8893e+00, ..., -1.4776e+00,
-1.3201e+00, -2.2657e-01],
...,
[ 6.9615e-02, -8.0274e-01, 4.0222e-01, ..., -2.1690e+00,
-1.3068e+00, -4.3553e-01],
[-2.2574e-01, -1.4320e+00, 7.9685e-02, ..., -5.9740e-01,
1.0997e+00, 2.4318e-01],
[ 6.4221e-01, 1.3972e-01, 2.8035e-01, ..., 3.0905e-01,
1.6227e-01, 1.9406e-01]],
[[-1.8237e+00, -4.7693e+00, -6.3708e+00, ..., -6.0336e-01,
-1.0297e+00, -2.1143e-01],
[-2.9842e+00, -4.1604e+00, -5.9676e+00, ..., -1.1707e+00,
-1.4551e+00, -2.5731e-01],
[-2.4899e+00, -1.7349e+00, -4.2035e+00, ..., -7.8864e-01,
-1.6857e+00, -9.8672e-01],
...,
[-1.9077e+00, -5.2613e-01, 1.1757e-01, ..., -4.6834e+00,
-4.4435e-01, 7.0247e-01],
[-2.4039e+00, -1.2592e+00, -2.3319e-01, ..., -1.8829e+00,
-2.4849e-01, 3.4373e-01],
[-1.1327e+00, -2.4615e-01, 4.1997e-01, ..., 1.0339e+00,
8.0822e-01, 5.3754e-01]]]], device='cuda:0'),
tensor([[[[-7.4379e-01, -1.3968e+00, -1.3748e+00, ..., 1.3394e-02,
6.4279e-01, 5.6043e-01],
[-2.8248e-01, -2.4027e+00, -2.1462e+00, ..., -4.6240e-01,
-2.3503e-01, 1.2617e-01],
[-3.8403e-01, -2.1446e+00, -1.8449e+00, ..., -3.1712e-01,
-1.2980e-01, -8.8771e-02],
...,
[-4.9394e-01, -1.9497e+00, -1.8638e+00, ..., -2.9399e-01,
-7.2561e-02, -1.3728e-01],
[-9.6555e-01, -1.7290e+00, -1.1298e+00, ..., 2.6346e-02,
2.9261e-01, -2.3909e-02],
[-8.1946e-01, -1.6846e+00, -1.2567e+00, ..., -1.0411e+00,
-1.0961e+00, -4.0009e-01]],
[[-2.3919e+00, -4.8587e+00, -4.1531e+00, ..., -4.2097e-01,
-1.0011e+00, 7.6932e-03],
[-5.3275e+00, -8.0822e+00, -6.5268e+00, ..., -1.7731e+00,
-2.6346e+00, -6.5880e-01],
[-4.3112e+00, -6.2194e+00, -5.3864e+00, ..., -1.3076e+00,
-1.4897e+00, -3.6742e-01],
...,
[-4.3625e+00, -6.1395e+00, -5.5119e+00, ..., -1.2809e+00,
-1.5437e+00, -3.7407e-01],
[-6.3946e+00, -7.4490e+00, -6.4085e+00, ..., -2.4663e+00,
-3.3116e+00, -9.5622e-01],
[ 4.7686e-03, -8.2033e-01, 3.9609e-01, ..., 3.2759e-01,
3.6303e-01, 8.6009e-02]],
[[-5.7702e-01, 6.0557e-01, 1.1118e+00, ..., 8.2430e-01,
2.8559e-01, 1.0921e+00],
[-9.3811e-01, -5.0658e-02, 4.3382e-01, ..., 4.0935e-01,
2.2520e-02, 4.7136e-01],
[-1.6684e+00, -1.2844e-01, -5.4531e-02, ..., 4.6830e-02,
9.4648e-02, 4.0879e-01],
...,
[-1.6994e+00, -8.9714e-02, -1.6992e-01, ..., 5.8398e-02,
6.2616e-02, 4.1428e-01],
[-1.5317e+00, 8.1648e-02, 3.8695e-02, ..., 1.1743e-01,
1.0894e-01, 5.9651e-01],
[-1.4350e+00, -5.2791e-01, -2.7841e-01, ..., -8.5300e-02,
5.6867e-02, 3.6645e-01]],
...,
[[ 1.0411e+00, -4.0006e-01, -1.9430e-01, ..., -1.0218e-01,
-3.8019e-01, -9.3550e-01],
[-7.8572e-01, -9.4610e-01, -8.3296e-01, ..., -6.8952e-01,
-1.2651e+00, -8.2759e-01],
[-6.7012e-01, -9.9587e-01, -7.6020e-01, ..., -1.0961e-01,
-1.9224e-01, -3.9173e-01],
...,
[-7.5375e-01, -1.1538e+00, -7.6742e-01, ..., -1.1632e-01,
-1.8071e-01, -3.7732e-01],
[-2.7730e+00, -2.8158e+00, -1.8603e+00, ..., -8.5612e-01,
-8.9133e-01, -7.9177e-01],
[-6.6834e-01, -1.6603e+00, -1.0090e+00, ..., -3.6417e-02,
-1.3113e-01, -3.2502e-01]],
[[ 2.9144e-02, -8.6519e-03, 5.6377e-01, ..., 3.3636e-01,
6.8288e-01, 3.1742e-01],
[ 1.8595e-02, -7.2109e-01, 1.2168e-01, ..., 5.6783e-03,
1.2580e-01, -5.3750e-02],
[-9.9624e-02, -8.1113e-01, 2.7793e-01, ..., -1.2314e-01,
-7.4015e-02, 1.7839e-01],
...,
[ 6.6600e-02, -7.7934e-01, 2.6227e-01, ..., -1.3886e-01,
-8.0103e-02, 1.5077e-01],
[-2.2269e-01, -1.4582e+00, 6.1489e-03, ..., -4.1085e-01,
-2.9440e-01, -6.3694e-02],
[ 6.6674e-01, 1.4144e-01, 2.7251e-01, ..., -2.5013e-01,
-2.1592e-01, 3.1072e-01]],
[[-1.5540e+00, -6.0370e-01, 3.4071e-02, ..., 5.4157e-01,
3.4095e-01, 5.4116e-01],
[-2.0454e+00, -3.4114e-01, -1.7629e-02, ..., -5.3105e-02,
-1.2634e-01, 4.2007e-01],
[-1.7213e+00, -3.9614e-01, 2.2934e-01, ..., 6.3282e-02,
-1.5251e-01, 8.8239e-02],
...,
[-1.7857e+00, -4.7741e-01, 2.7506e-01, ..., 1.1590e-02,
-7.7722e-02, 9.0045e-02],
[-2.2005e+00, -1.1223e+00, -1.3590e-01, ..., -1.5804e-01,
-2.4514e-01, -1.9348e-01],
[-1.0678e+00, -2.2904e-01, 4.4302e-01, ..., -1.0063e-01,
-2.5960e-01, 1.1026e-01]]],
[[[-7.5624e-01, -1.4509e+00, -1.4250e+00, ..., -1.2069e+00,
-8.2127e-01, -1.6240e+00],
[-3.6719e-01, -2.4295e+00, -2.2544e+00, ..., -1.5577e+00,
-1.8598e+00, -1.8848e+00],
[-1.1973e-01, -1.8928e+00, -1.6836e+00, ..., -1.2177e+00,
-1.3866e+00, -2.0800e+00],
...,
[ 7.9356e-02, -5.1666e-01, -2.0236e-01, ..., -8.4662e-02,
-3.0661e-02, -4.3296e-01],
[-3.7172e-01, -4.6720e-01, 2.7451e-02, ..., 1.3634e-01,
2.7165e-01, -1.5522e-01],
[-6.4014e-01, -1.3463e+00, -1.1008e+00, ..., -1.0386e+00,
-1.1501e+00, -4.9002e-01]],
[[-2.6837e+00, -5.1771e+00, -4.6666e+00, ..., -5.7696e+00,
-5.9541e+00, -2.3848e+00],
[-5.3297e+00, -8.1638e+00, -7.5040e+00, ..., -1.0068e+01,
-9.2038e+00, -4.3621e+00],
[-4.0046e+00, -5.7749e+00, -5.7081e+00, ..., -8.5402e+00,
-6.4439e+00, -3.2276e+00],
...,
[-1.6965e+00, -2.1468e+00, -1.6761e+00, ..., -1.8103e+00,
-2.2137e+00, -8.8372e-01],
[-2.7861e+00, -3.0001e+00, -1.9477e+00, ..., -2.1615e+00,
-3.2837e+00, -9.6405e-01],
[ 3.2344e-01, -2.9057e-01, 3.2975e-01, ..., 3.0182e-01,
4.3745e-01, 2.0309e-01]],
[[-6.3866e-01, 5.2584e-01, 8.5776e-01, ..., 1.2977e+00,
6.0509e-01, 1.9127e+00],
[-9.8884e-01, 7.1917e-02, 4.0657e-01, ..., 1.2747e+00,
-3.6196e-03, 1.2763e-01],
[-1.8083e+00, -1.9606e-01, -6.4381e-02, ..., 3.2495e-01,
2.4032e-02, -8.0565e-02],
...,
[-6.4599e-01, 8.0192e-02, 2.5596e-01, ..., 1.7397e-01,
2.5456e-01, 4.0374e-01],
[-4.4013e-01, 7.6832e-02, 1.9227e-01, ..., 1.7178e-01,
2.2933e-01, 5.7055e-01],
[-1.3912e-01, -2.5656e-01, 1.1853e-01, ..., -1.8494e-03,
9.8540e-02, 3.3275e-01]],
...,
[[ 1.2507e+00, -3.3763e-01, -3.4811e-02, ..., 2.3615e-01,
-1.0711e-01, -2.1293e+00],
[-7.1937e-01, -9.2612e-01, -4.8399e-01, ..., -7.6886e-01,
-1.5242e+00, -2.0671e+00],
[-6.9547e-01, -8.6687e-01, -2.9696e-01, ..., -1.3086e+00,
-6.0217e-01, -1.3643e+00],
...,
[-6.5559e-01, -4.5687e-01, -3.1591e-01, ..., -3.1047e-01,
-4.8606e-01, -5.4691e-01],
[-2.2881e+00, -1.8248e+00, -1.3719e+00, ..., -1.2816e+00,
-1.3410e+00, -1.0948e+00],
[-6.1361e-01, -8.3193e-01, -4.2650e-01, ..., -2.4272e-01,
-3.3558e-01, -3.8522e-01]],
[[ 8.0112e-02, 3.5389e-02, 4.4607e-01, ..., 2.0317e-01,
9.3283e-01, -1.9885e-02],
[ 3.0832e-01, -7.3336e-01, -3.4233e-02, ..., -7.2961e-02,
3.9490e-01, -8.2205e-01],
[ 1.9459e-01, -5.8693e-01, 3.0847e-01, ..., 5.9792e-01,
5.3373e-01, -3.6293e-01],
...,
[-2.3606e-01, -4.0927e-01, 2.6354e-01, ..., 3.4770e-01,
3.1301e-01, 6.6262e-02],
[-5.3210e-01, -8.7968e-01, 7.0170e-02, ..., 1.7581e-02,
-1.8606e-02, -1.3979e-01],
[ 3.0879e-01, -2.2821e-02, -2.0338e-01, ..., -1.1316e-01,
-9.6751e-02, 1.7755e-01]],
[[-1.5213e+00, -7.6758e-01, -4.1239e-01, ..., -3.7274e-01,
-2.8552e-01, 8.8207e-01],
[-2.1649e+00, -7.8197e-01, -5.5325e-01, ..., -6.7698e-01,
-2.1122e-01, 1.6599e+00],
[-1.4726e+00, -5.3508e-01, -2.0256e-01, ..., -2.8241e-02,
-1.1656e-01, 1.4028e+00],
...,
[-6.6164e-01, -1.7698e-01, -9.0010e-02, ..., -1.3241e-01,
1.1609e-02, 6.2067e-01],
[-1.1692e+00, -5.5770e-01, -3.2225e-01, ..., -3.2445e-01,
-1.9202e-01, 1.6753e-01],
[-3.7269e-01, 4.1113e-03, -1.6197e-01, ..., -2.1810e-01,
-2.1418e-01, 3.1676e-01]]],
[[[-7.5071e-01, -1.4605e+00, -1.4947e+00, ..., -2.8961e-01,
1.2029e-01, -4.5821e-01],
[-2.1468e-01, -2.7588e+00, -4.2882e+00, ..., -6.2082e-01,
-5.1840e-01, -7.0880e-01],
[ 9.6360e-02, -3.7640e+00, -6.4001e+00, ..., -2.6410e-01,
-1.2971e+00, -2.9498e+00],
...,
[-5.1816e-01, -1.9928e+00, -1.9439e+00, ..., 1.9130e+00,
5.4629e-01, -3.8579e+00],
[-1.0965e+00, -1.8562e+00, -1.2507e+00, ..., 4.6446e-01,
1.0489e+00, -1.9142e+00],
[-8.6926e-01, -1.7553e+00, -1.3188e+00, ..., -9.4608e-01,
-1.2187e+00, -7.5473e-01]],
[[-2.4684e+00, -4.9636e+00, -4.2643e+00, ..., -2.5540e+00,
-3.7635e+00, -1.4236e+00],
[-7.0639e+00, -1.3529e+01, -1.1377e+01, ..., -3.7822e+00,
-5.4353e+00, -2.2506e+00],
[-8.2009e+00, -1.4006e+01, -1.0302e+01, ..., -5.9143e+00,
-8.4063e+00, -4.1538e+00],
...,
[-4.5476e+00, -6.4042e+00, -5.7348e+00, ..., -3.6560e+00,
-6.0712e+00, -4.5382e+00],
[-6.5296e+00, -7.6288e+00, -6.6226e+00, ..., -4.8201e+00,
-5.9836e+00, -2.4110e+00],
[-1.0016e-01, -8.9464e-01, 3.1571e-01, ..., 5.3419e-01,
6.7253e-01, 1.2802e-01]],
[[-6.2497e-01, 5.8696e-01, 1.1336e+00, ..., 7.6028e-01,
5.3942e-01, 1.2968e+00],
[-1.0530e+00, -4.4681e-01, 1.6121e-01, ..., 3.1692e-01,
2.7516e-01, 1.6521e-01],
[-2.1208e+00, -2.7908e+00, -3.9769e+00, ..., -7.9733e-02,
3.3624e-01, 5.0471e-01],
...,
[-1.7886e+00, -5.3163e-02, -8.3211e-02, ..., 1.9255e-01,
-3.4103e+00, -2.8024e+00],
[-1.5921e+00, 5.6187e-02, -1.2052e-02, ..., 1.3725e+00,
1.6264e-01, -1.2893e+00],
[-1.4903e+00, -5.5748e-01, -2.9607e-01, ..., 6.4389e-02,
2.3110e-01, -1.3141e-01]],
...,
[[ 1.1580e+00, -4.2010e-01, -2.5342e-01, ..., 4.2433e-01,
3.0882e-01, -7.4871e-01],
[-7.3825e-01, -6.2225e-01, -4.3816e-01, ..., -2.4995e-01,
-9.2789e-01, -9.2418e-01],
[ 6.0317e-02, 1.6814e+00, 4.0806e-01, ..., 3.2649e-02,
-1.5452e-01, -9.0030e-01],
...,
[-7.1125e-01, -1.1946e+00, -7.4881e-01, ..., -1.3458e+00,
-1.2417e+00, -3.7034e+00],
[-2.8688e+00, -3.0442e+00, -1.9475e+00, ..., -2.5529e+00,
-3.3019e+00, -2.4417e+00],
[-6.5829e-01, -1.7619e+00, -1.0652e+00, ..., -8.6341e-02,
-2.1532e-01, -5.2342e-01]],
[[ 1.8326e-02, -2.9278e-02, 6.0699e-01, ..., 2.9541e-01,
7.5168e-01, -1.9585e-02],
[ 1.7310e-01, 1.1954e-02, -7.2450e-01, ..., 3.9521e-01,
6.1375e-01, -3.8319e-01],
[ 7.0497e-01, -4.6515e-01, -2.2205e+00, ..., 7.2276e-01,
5.7854e-01, -1.4210e+00],
...,
[ 1.6146e-01, -8.4972e-01, 2.8508e-01, ..., 2.1368e-01,
-4.5856e-01, -2.6847e+00],
[-1.0846e-01, -1.4136e+00, 7.6381e-02, ..., 5.6780e-01,
3.1626e-01, -2.5270e+00],
[ 6.7060e-01, 1.6253e-01, 3.1613e-01, ..., 2.9415e-01,
3.3770e-01, 3.3719e-01]],
[[-1.7211e+00, -6.8251e-01, -1.2455e-02, ..., 2.7761e-01,
2.3293e-01, 7.8967e-01],
[-2.1939e+00, -4.9073e-01, -1.5333e-01, ..., -3.7188e-01,
-6.1014e-02, 1.1232e+00],
[-2.4522e+00, -2.9700e+00, -2.0520e+00, ..., -1.3710e-01,
-3.5219e-01, 3.3415e-01],
...,
[-1.8605e+00, -4.5446e-01, 2.1455e-01, ..., -3.8454e-01,
1.9351e+00, 2.2060e+00],
[-2.4651e+00, -1.2733e+00, -2.5161e-01, ..., -9.5691e-01,
4.4289e-01, 2.0484e+00],
[-1.1617e+00, -2.4113e-01, 4.1324e-01, ..., -2.5233e-01,
-2.4236e-01, 6.9292e-01]]],
[[[ 3.2336e-02, -2.3756e-01, -7.3237e-02, ..., -9.7528e-01,
-5.3883e-01, -1.4963e+00],
[ 3.1706e-01, -9.6941e-01, -7.2886e-01, ..., -1.1883e+00,
-1.3208e+00, -1.5205e+00],
[ 2.1911e-01, -6.2388e-01, -4.8390e-01, ..., -1.3392e+00,
-1.0915e+00, -1.8470e+00],
...,
[-1.6962e-02, -5.5133e-01, -6.1506e-04, ..., -1.1842e+00,
-1.0842e+00, -1.9637e+00],
[-6.5767e-01, -7.0326e-01, 1.3663e-01, ..., -4.6673e-01,
-6.0664e-01, -1.7240e+00],
[-7.7831e-01, -1.3631e+00, -1.1783e+00, ..., -1.1450e+00,
-1.5835e+00, -8.0984e-01]],
[[-7.3316e-01, -2.2881e+00, -2.0308e+00, ..., -3.9793e+00,
-5.5643e+00, -1.8745e+00],
[-2.0902e+00, -3.9013e+00, -3.2580e+00, ..., -5.9919e+00,
-8.2065e+00, -3.4604e+00],
[-1.6761e+00, -2.6778e+00, -2.7943e+00, ..., -6.2959e+00,
-6.2831e+00, -2.5891e+00],
...,
[-1.8456e+00, -2.5864e+00, -2.6567e+00, ..., -5.2131e+00,
-6.3960e+00, -2.4473e+00],
[-3.9222e+00, -4.5773e+00, -3.8469e+00, ..., -6.3555e+00,
-7.9059e+00, -2.5362e+00],
[ 2.7768e-01, -3.1873e-01, 4.0440e-01, ..., 4.0430e-01,
4.4900e-01, 2.6336e-01]],
[[-2.5810e-01, 1.9567e-01, 7.9778e-01, ..., 6.3437e-01,
2.5657e-01, 1.4973e+00],
[-1.7482e-02, -4.4627e-01, 5.5200e-01, ..., 5.6938e-01,
-7.7425e-02, 6.8347e-02],
[-1.8973e-01, -2.6303e-01, 1.3394e-01, ..., 1.0408e-01,
-6.3808e-02, -1.6720e-01],
...,
[-2.9142e-01, 4.7842e-02, -8.1005e-02, ..., -5.7991e-02,
-1.0745e-01, -2.2531e-01],
[-2.4024e-01, -9.1913e-02, 2.8886e-01, ..., -1.0755e-01,
-9.7514e-03, -2.7616e-01],
[-1.3850e-01, -4.2926e-01, 1.0383e-01, ..., 4.1874e-02,
1.8097e-01, -6.5332e-01]],
...,
[[ 7.0324e-01, -1.5653e-01, 9.3703e-02, ..., 4.2723e-01,
3.5547e-01, -1.3114e+00],
[-4.9232e-01, -5.3136e-01, -4.9579e-01, ..., -1.4526e-01,
-7.6064e-01, -1.4081e+00],
[-2.9711e-01, -1.8228e-01, -2.1578e-01, ..., -2.0124e-01,
-1.1418e-01, -9.5536e-01],
...,
[-4.5162e-01, -2.4179e-01, -7.2600e-02, ..., -1.8634e-02,
-7.8026e-02, -9.6927e-01],
[-2.3930e+00, -1.7997e+00, -1.4450e+00, ..., -1.5470e+00,
-1.5940e+00, -1.8840e+00],
[-4.8974e-01, -8.3809e-01, -3.5864e-01, ..., -3.8768e-01,
-4.9269e-01, -1.1937e+00]],
[[ 2.1513e-01, 3.2421e-01, 6.3471e-01, ..., -3.4663e-02,
3.7291e-01, -4.0779e-01],
[ 1.5790e-01, 2.8809e-01, 4.7143e-01, ..., 2.3729e-01,
9.4296e-02, -8.5383e-01],
[-1.3523e-01, 9.4282e-02, 3.7827e-01, ..., -7.7226e-01,
2.0079e-01, -4.6395e-01],
...,
[-1.9507e-01, 8.3066e-02, 2.7053e-01, ..., 1.6037e-02,
1.3241e-01, -4.7296e-01],
[-5.6363e-01, -7.5247e-01, 5.3991e-02, ..., -1.4226e-01,
-2.5641e-01, -8.6871e-01],
[ 5.2505e-01, 3.8483e-01, 1.1079e-01, ..., 3.4108e-01,
3.2395e-01, 3.3187e-01]],
[[-8.0351e-01, 2.1190e-01, 3.4040e-01, ..., -7.2981e-02,
-6.7538e-02, 5.4163e-01],
[-1.1092e+00, 1.2789e-01, -4.1535e-01, ..., -9.3072e-01,
-4.0627e-01, 9.0307e-01],
[-9.7834e-01, 2.4256e-01, -8.5083e-02, ..., -5.1293e-01,
-5.2826e-01, 6.3508e-01],
...,
[-9.7022e-01, -4.7278e-02, -2.0965e-01, ..., -2.8262e-01,
-5.9261e-01, 5.9299e-01],
[-1.6778e+00, -4.3350e-01, -2.9563e-01, ..., -7.6898e-01,
-7.3574e-01, 4.1869e-01],
[-7.2015e-01, -1.1572e-01, -2.7368e-01, ..., -4.8368e-01,
-4.8339e-01, 1.0479e+00]]],
[[[-7.5919e-01, -1.5324e+00, -1.6118e+00, ..., -1.2644e+01,
-4.4697e+00, -3.2103e+00],
[-4.9435e-01, -2.5434e+00, -2.5063e+00, ..., -2.3187e+01,
-9.5065e+00, -2.2166e+00],
[-3.8087e-01, -2.1071e+00, -2.0361e+00, ..., -1.0306e+01,
1.0776e+01, 1.3615e+00],
...,
[ 1.1603e-01, -8.9648e-01, -3.9251e-01, ..., -8.7419e-01,
-3.6831e-01, -7.6747e-01],
[-2.2755e-01, -6.8194e-01, -9.7638e-02, ..., 7.3281e-02,
1.7724e-01, -4.0930e-01],
[-4.4306e-01, -1.2808e+00, -1.0987e+00, ..., -9.9692e-01,
-1.1966e+00, -5.7923e-01]],
[[-2.9639e+00, -5.4245e+00, -4.7304e+00, ..., -3.0387e+01,
-1.5768e+01, -3.5951e+00],
[-5.8440e+00, -8.6602e+00, -8.0235e+00, ..., -1.7103e+01,
-1.2684e+01, -1.1196e+01],
[-4.6325e+00, -6.5144e+00, -6.2386e+00, ..., 1.6302e+00,
2.4786e+00, -9.9316e+00],
...,
[-2.3087e+00, -3.1304e+00, -2.7904e+00, ..., -3.6866e+00,
-3.4890e+00, -1.4304e+00],
[-3.3411e+00, -3.7607e+00, -2.7675e+00, ..., -2.9362e+00,
-4.0454e+00, -1.3804e+00],
[ 4.8649e-01, -1.2295e-01, 5.2985e-01, ..., 4.6620e-01,
5.5938e-01, 2.1028e-01]],
[[-7.1872e-01, 4.6815e-01, 9.3501e-01, ..., -2.5266e+01,
-1.0831e+01, -2.6439e+00],
[-9.0581e-01, 1.1843e-01, 4.1470e-01, ..., -2.3084e+01,
-7.2731e+00, -4.0636e+00],
[-1.8046e+00, -2.2789e-01, -1.7144e-01, ..., -7.3447e+00,
-7.0486e+00, -2.4043e+00],
...,
[-1.0000e+00, 2.0120e-01, 3.3950e-01, ..., 8.0971e-01,
3.5058e-01, 3.9088e-01],
[-7.0562e-01, 2.1592e-01, 3.2479e-01, ..., 2.4566e-01,
3.0696e-01, 4.7455e-01],
[-1.7283e-01, -3.2469e-03, 1.4817e-01, ..., -7.3722e-04,
1.4575e-01, 2.4905e-01]],
...,
[[ 1.5154e+00, -3.2396e-01, -7.2383e-02, ..., -8.0839e+00,
-1.0725e+01, -1.1562e+01],
[-6.9258e-01, -1.0871e+00, -7.1287e-01, ..., -1.3645e+01,
-9.8106e+00, -1.9957e+01],
[-7.4506e-01, -1.0511e+00, -4.1282e-01, ..., -1.2787e+01,
-1.2838e+01, -2.2667e+01],
...,
[-6.6555e-01, -5.3335e-01, -2.8239e-01, ..., -1.3859e+00,
-6.0395e-01, -7.2527e-01],
[-2.1759e+00, -1.7298e+00, -1.2267e+00, ..., -1.3127e+00,
-1.4382e+00, -1.2080e+00],
[-6.3137e-01, -8.4692e-01, -3.6588e-01, ..., -3.5622e-01,
-4.7994e-01, -4.4282e-01]],
[[ 5.6886e-02, 2.4837e-02, 6.8497e-01, ..., 5.5982e+00,
4.0405e+00, -1.6445e+00],
[ 2.3604e-01, -8.5145e-01, 5.6502e-02, ..., 1.1389e+01,
2.7354e+00, -1.5093e+00],
[ 2.4118e-01, -7.5170e-01, 2.7044e-01, ..., 1.2272e+01,
2.8413e+00, -5.2351e+00],
...,
[-1.1002e-01, -7.4451e-01, 1.9179e-01, ..., -2.3702e-01,
1.0271e-01, -9.9595e-02],
[-5.0856e-01, -1.2903e+00, -3.7142e-02, ..., 2.8973e-02,
-3.5094e-02, -2.9304e-01],
[ 5.0708e-01, -1.0996e-01, -1.0403e-01, ..., 1.8329e-02,
2.2255e-02, 1.6931e-01]],
[[-1.9155e+00, -1.1168e+00, -6.2656e-01, ..., -1.8572e+01,
-8.8278e+00, 4.7977e+00],
[-2.6035e+00, -9.1729e-01, -7.0990e-01, ..., -6.6149e+00,
-1.2818e+00, 3.3674e+00],
[-2.0273e+00, -8.4716e-01, -3.9740e-01, ..., -1.1140e+01,
-1.2765e+01, -7.2801e+00],
...,
[-6.4136e-01, -4.4332e-01, -3.2392e-01, ..., 1.4726e-01,
-4.2637e-02, 7.0253e-01],
[-1.0210e+00, -7.3385e-01, -4.4827e-01, ..., -3.9316e-01,
-2.4153e-01, 3.5501e-01],
[-3.3498e-01, -2.3641e-01, -1.4044e-01, ..., -1.8708e-01,
-1.6650e-01, 5.2146e-01]]]], device='cuda:0'),
tensor([[[[-3.1397e-01, -1.0888e+00, -1.4470e+00, ..., -2.4526e-02,
6.0123e-01, 5.5143e-01],
[ 2.0942e-01, -2.0070e+00, -2.1699e+00, ..., -4.3073e-01,
-2.6478e-01, 1.1628e-01],
[ 2.1604e-01, -1.0992e+00, -1.6003e+00, ..., -3.2225e-01,
-1.2984e-01, -9.7250e-02],
...,
[-8.4470e-01, -2.1243e+00, -5.0817e-01, ..., -5.8539e-01,
-1.8694e-01, -1.5034e-01],
[ 1.6159e-01, -2.3102e+00, -1.3264e+00, ..., -3.8865e-01,
1.6836e-01, -6.2062e-02],
[ 1.0578e-01, -1.6818e+00, -1.3931e+00, ..., -1.1321e+00,
-1.0874e+00, -3.8284e-01]],
[[-2.1825e+00, -5.3260e+00, -5.1551e+00, ..., -5.4123e-01,
-1.0414e+00, -1.5504e-02],
[-4.0399e+00, -6.8815e+00, -5.1792e+00, ..., -1.8907e+00,
-2.7268e+00, -6.9844e-01],
[-3.1801e+00, -5.2816e+00, -5.9692e+00, ..., -1.2905e+00,
-1.5222e+00, -3.9036e-01],
...,
[-3.2800e+00, -5.2642e+00, -3.7690e+00, ..., -1.9351e+00,
-1.8039e+00, -4.2821e-01],
[-5.3288e+00, -6.9906e+00, -5.9109e+00, ..., -2.7036e+00,
-3.3200e+00, -9.8772e-01],
[ 7.7824e-01, -1.7104e-01, 3.7109e-01, ..., 2.2527e-01,
3.5571e-01, 6.4004e-02]],
[[-5.8906e-01, 9.2720e-02, 3.4921e-01, ..., 8.0976e-01,
2.9455e-01, 1.0944e+00],
[-8.7248e-01, -2.6647e-01, -1.3380e+00, ..., 4.3996e-01,
5.3708e-02, 4.7464e-01],
[-1.2725e+00, 2.1919e-01, -1.9448e+00, ..., 7.3991e-02,
1.1594e-01, 4.1473e-01],
...,
[-2.6537e+00, -1.3672e+00, -1.7116e+00, ..., -8.6051e-01,
3.4298e-02, 4.9983e-01],
[-2.6987e+00, -3.0089e-01, -4.6495e-01, ..., 5.3698e-02,
3.3841e-02, 6.2934e-01],
[-1.0632e+00, 5.9516e-01, 9.4203e-01, ..., -6.6266e-02,
4.5611e-02, 3.6704e-01]],
...,
[[ 6.6023e-01, 4.0930e-01, 5.9151e-01, ..., -1.0986e-01,
-3.9967e-01, -9.3796e-01],
[-8.1071e-01, -1.2142e-02, 3.2810e-01, ..., -6.6461e-01,
-1.3024e+00, -8.4563e-01],
[-5.1042e-01, -2.8218e-01, -2.0693e-01, ..., -9.0422e-02,
-1.9634e-01, -3.8324e-01],
...,
[-3.1930e-02, -1.0428e+00, -8.9428e-01, ..., -7.9922e-01,
-5.7655e-01, -2.8296e-01],
[-1.5980e+00, -1.3261e+00, -1.3134e+00, ..., -1.1770e+00,
-8.7680e-01, -7.0663e-01],
[ 6.4112e-02, -1.0475e-01, -1.6393e-01, ..., -1.1434e-01,
-1.3377e-01, -2.8246e-01]],
[[ 2.6106e-02, -4.8386e-02, 2.3128e-01, ..., 3.2006e-01,
6.7312e-01, 3.0683e-01],
[ 1.3627e-01, -8.8209e-01, -1.5878e-01, ..., 3.1147e-02,
8.8124e-02, -5.4073e-02],
[-1.4632e-01, -4.8352e-01, 1.7022e-01, ..., -1.1821e-01,
-8.7225e-02, 1.7931e-01],
...,
[-1.4843e+00, -2.4659e+00, -1.7766e-01, ..., -1.8511e-01,
-1.5726e-01, 2.1513e-01],
[-1.0457e+00, -3.3277e+00, -1.0168e+00, ..., -3.8450e-01,
-1.8992e-01, -3.9521e-02],
[ 5.2692e-01, -7.2604e-01, 3.1460e-01, ..., -3.5414e-01,
-2.3496e-01, 3.0590e-01]],
[[-6.6586e-01, -4.7286e-01, -7.7910e-01, ..., 5.4592e-01,
3.6805e-01, 5.4269e-01],
[-1.0506e+00, -1.1901e+00, -1.8635e+00, ..., -6.4365e-02,
-1.3515e-01, 4.2503e-01],
[-6.9520e-01, -8.9742e-01, -5.3337e-01, ..., 5.5525e-02,
-1.5320e-01, 9.5449e-02],
...,
[ 5.6098e-02, -1.5817e+00, -1.6658e+00, ..., -1.5268e-01,
3.3696e-01, 2.7412e-01],
[ 3.5800e-01, -1.1112e+00, -2.0090e+00, ..., -8.8201e-03,
-2.6191e-02, -8.6498e-02],
[-6.9607e-02, -8.2637e-01, -1.1746e+00, ..., -9.2716e-03,
-2.0338e-01, 1.4860e-01]]],
[[[-1.9787e-01, -7.9680e-01, 9.0765e-02, ..., -1.0820e-02,
4.3952e-01, 1.0957e-01],
[-4.2149e+00, -7.0904e+00, -2.3477e+00, ..., -7.4455e-01,
-4.6646e-01, -3.9772e-01],
[-8.3776e+00, -8.6697e+00, -1.7841e+00, ..., -5.3585e-01,
-4.4534e-01, -7.7104e-01],
...,
[-1.8505e-01, -1.5413e+00, -1.1905e+00, ..., -1.6291e+00,
-1.4239e+00, -2.1033e+00],
[-9.1206e-01, -1.5615e+00, -8.2335e-01, ..., -1.2334e+00,
-5.9347e-01, -1.8640e+00],
[-1.0536e+00, -1.9308e+00, -1.5477e+00, ..., -1.4474e+00,
-2.1070e+00, -1.1206e+00]],
[[-1.8117e+00, -3.1969e+00, -3.6728e+00, ..., -8.0808e-01,
-1.4680e+00, -3.3162e-01],
[-8.7688e+00, -1.1307e+01, -1.4493e+01, ..., -2.5181e+00,
-3.5125e+00, -1.2143e+00],
[-8.1860e+00, -9.9633e+00, -1.0780e+01, ..., -2.7654e+00,
-3.1939e+00, -1.3329e+00],
...,
[-3.8168e+00, -5.6095e+00, -5.5832e+00, ..., -9.0936e+00,
-7.5028e+00, -3.0618e+00],
[-5.9244e+00, -6.6621e+00, -6.0310e+00, ..., -1.0111e+01,
-9.0968e+00, -2.9315e+00],
[-1.8366e-02, -6.3901e-01, 2.4914e-01, ..., -1.3472e+00,
9.0931e-02, 1.9026e-01]],
[[-7.4630e-01, -5.5115e-01, 1.0663e-01, ..., 3.8030e-01,
1.1246e-01, 9.3052e-01],
[-5.3271e+00, -6.5918e+00, -9.3696e-01, ..., 4.3307e-01,
3.0302e-01, 4.9734e-01],
[-1.3229e+01, -2.0372e+01, -1.0810e+01, ..., 7.9230e-02,
1.7968e-01, 4.1524e-01],
...,
[-1.4222e+00, 8.9391e-02, 2.9994e-01, ..., 7.4762e-01,
6.9223e-02, 1.8555e-01],
[-1.2892e+00, 1.2694e-01, -8.0546e-02, ..., 6.1184e-01,
4.6234e-01, 1.9781e-01],
[-1.1216e+00, -2.7291e-02, 4.9441e-02, ..., -7.1184e-01,
1.6318e-01, -3.7820e-01]],
...,
[[ 2.0171e-01, -3.4233e-01, -1.2851e-01, ..., -4.0491e-02,
-1.9790e-01, -7.5257e-01],
[ 1.0147e+00, -6.1607e-01, -2.5791e+00, ..., -2.8947e-01,
-1.0121e+00, -7.7633e-01],
[ 2.8118e+00, -3.2239e+00, -1.0165e+01, ..., 7.5016e-02,
-1.7474e-01, -5.2654e-01],
...,
[-4.5186e-01, -7.9829e-01, -5.8294e-01, ..., -1.4241e+00,
-8.8503e-01, -1.3092e+00],
[-2.7460e+00, -2.9478e+00, -1.9504e+00, ..., -4.3051e+00,
-2.9375e+00, -2.2299e+00],
[-5.4754e-01, -1.6917e+00, -1.0884e+00, ..., -3.5425e+00,
-1.6810e+00, -1.4616e+00]],
[[ 1.1610e-01, 1.8379e-01, 1.5841e-01, ..., 9.1751e-02,
5.2754e-01, 1.5738e-01],
[-8.0132e-01, -2.9997e+00, -1.5763e+00, ..., -1.4914e-02,
1.4983e-01, -9.8249e-02],
[-5.3025e+00, -7.7811e+00, -2.0114e+00, ..., 1.5900e-02,
1.1990e-02, -7.0706e-02],
...,
[ 7.0899e-02, -9.4375e-01, 2.6712e-01, ..., -4.9596e-01,
-4.6905e-01, -7.8886e-01],
[-2.9885e-01, -1.6711e+00, -9.4094e-02, ..., 4.5874e-01,
-4.8316e-01, -1.0677e+00],
[ 5.2247e-01, -9.4047e-02, 1.7140e-01, ..., 3.7988e-01,
-2.3836e-01, 3.2208e-01]],
[[-5.3046e-01, 5.6012e-01, 7.8146e-01, ..., 4.9470e-01,
3.9228e-01, 6.4123e-01],
[-3.2528e+00, -6.1255e+00, -4.3472e+00, ..., -2.6477e-01,
-1.4079e-01, 7.1202e-01],
[-4.4538e+00, -1.4848e+01, -1.5047e+01, ..., 8.0392e-02,
1.0976e-01, 6.2278e-01],
...,
[-1.1644e+00, -9.3885e-01, -5.2873e-01, ..., -8.5149e-01,
-6.6889e-02, 8.1355e-01],
[-1.8890e+00, -1.7468e+00, -9.5451e-01, ..., -1.1566e+00,
-5.6047e-01, 4.1441e-01],
[-9.4799e-01, -8.6058e-01, -2.0589e-01, ..., 7.1456e-02,
-3.4579e-01, 1.2035e+00]]],
[[[-1.6939e+00, -4.6844e+00, -3.4370e+00, ..., -5.5020e-01,
-6.2743e-02, -7.9307e-01],
[-6.6968e-01, -3.8875e+00, -4.0735e-01, ..., -9.7882e-01,
-8.0262e-01, -1.1943e+00],
[ 2.7314e-01, -1.0266e+00, -8.6877e-03, ..., -7.8181e-01,
-7.7816e-01, -1.7149e+00],
...,
[ 9.3504e-02, -5.0790e-01, 2.0383e-01, ..., -6.2414e-02,
-2.1119e-02, -3.9538e-01],
[-3.6341e-01, -4.1829e-01, 8.2789e-02, ..., 1.0983e-01,
2.7778e-01, -1.3331e-01],
[-6.3370e-01, -1.2940e+00, -1.1091e+00, ..., -1.0789e+00,
-1.1610e+00, -5.0147e-01]],
[[-4.2790e+00, -8.2684e+00, -6.7549e+00, ..., -3.9347e+00,
-4.8950e+00, -2.0011e+00],
[-5.2929e+00, -8.3493e+00, -7.0225e+00, ..., -5.3992e+00,
-6.6937e+00, -2.8619e+00],
[-4.4606e+00, -6.3361e+00, -7.1027e+00, ..., -5.1893e+00,
-5.7286e+00, -2.7626e+00],
...,
[-1.6970e+00, -2.1294e+00, -1.5758e+00, ..., -1.7872e+00,
-2.1083e+00, -7.9022e-01],
[-2.6963e+00, -2.9337e+00, -2.1567e+00, ..., -2.1675e+00,
-3.2894e+00, -9.3632e-01],
[ 2.9585e-01, -2.6613e-01, 3.2414e-01, ..., 2.5267e-01,
3.9428e-01, 1.6317e-01]],
[[-1.3381e+00, -4.2305e+00, -6.2201e+00, ..., 8.1798e-01,
5.6409e-01, 1.4374e+00],
[-6.0392e-01, -4.1979e+00, -7.2677e+00, ..., 4.2003e-01,
2.9591e-01, 4.1498e-01],
[-1.0557e+00, -3.4803e+00, -2.9456e+00, ..., 1.7962e-01,
2.4975e-01, 4.0136e-01],
...,
[-6.1295e-01, 1.2728e-01, 5.7294e-01, ..., 1.5620e-01,
2.4870e-01, 4.4053e-01],
[-3.8650e-01, 9.1905e-02, 7.2306e-01, ..., 1.4099e-01,
1.8337e-01, 5.7878e-01],
[-1.3395e-01, -3.2639e-01, 1.3805e-01, ..., 3.3917e-03,
7.6554e-02, 2.9843e-01]],
...,
[[ 1.8749e+00, 1.7223e+00, -2.1104e+00, ..., 4.4821e-01,
2.5883e-01, -9.6724e-01],
[-1.3642e+00, -3.8389e+00, -6.4151e+00, ..., -3.6324e-01,
-1.1597e+00, -1.2744e+00],
[-3.5303e-01, -2.3734e+00, -3.6231e+00, ..., -2.9136e-01,
-5.5070e-01, -1.1458e+00],
...,
[-6.3625e-01, -5.5837e-01, -5.2920e-01, ..., -2.8966e-01,
-4.7124e-01, -5.6932e-01],
[-2.2038e+00, -1.7738e+00, -1.4077e+00, ..., -1.2782e+00,
-1.3560e+00, -1.1196e+00],
[-6.0491e-01, -7.8991e-01, -4.1887e-01, ..., -2.6095e-01,
-4.0906e-01, -4.0990e-01]],
[[-5.1938e-01, -2.9982e+00, -4.2424e+00, ..., 2.2122e-01,
6.9418e-01, -2.8337e-01],
[-5.4809e-01, -4.3295e+00, -4.2330e+00, ..., 1.6993e-01,
3.5931e-01, -7.6833e-01],
[ 2.5761e-01, -1.7087e+00, -2.8506e+00, ..., 2.5414e-01,
1.7408e-01, -6.5291e-01],
...,
[-2.1797e-01, -4.1442e-01, 2.9048e-01, ..., 3.5688e-01,
2.8576e-01, 1.0150e-01],
[-4.9546e-01, -7.9309e-01, 2.3306e-01, ..., -3.6532e-02,
-1.1101e-01, -1.3644e-01],
[ 3.0836e-01, 2.3175e-02, -1.7519e-01, ..., -1.6186e-01,
-1.5865e-01, 1.7144e-01]],
[[-1.7503e+00, -3.6077e+00, -2.3445e+00, ..., 3.2043e-01,
4.2882e-01, 9.5356e-01],
[-1.9758e+00, -2.9082e+00, -3.2774e+00, ..., -6.0364e-01,
-2.4732e-01, 9.6122e-01],
[-1.8958e+00, -1.9478e+00, -3.9506e+00, ..., -3.2322e-01,
-5.9075e-02, 7.8699e-01],
...,
[-7.3173e-01, -2.9102e-01, -6.8869e-01, ..., -1.3689e-01,
9.6818e-03, 6.4237e-01],
[-1.1804e+00, -5.0537e-01, -5.3519e-01, ..., -3.2933e-01,
-2.3209e-01, 1.9098e-01],
[-3.6172e-01, 3.1786e-02, -1.4566e-01, ..., -2.0997e-01,
-2.4048e-01, 3.0524e-01]]],
[[[-7.3445e-01, -1.4152e+00, -1.3991e+00, ..., -2.2638e-01,
1.7927e-01, -4.4956e-01],
[-2.9509e-01, -2.4145e+00, -2.1610e+00, ..., -5.4913e-01,
-3.8466e-01, -5.9206e-01],
[-4.0076e-01, -2.1176e+00, -1.8903e+00, ..., -3.5169e-01,
-3.9234e-01, -1.0635e+00],
...,
[-4.6878e-01, -1.8992e+00, -1.8368e+00, ..., -9.7303e-02,
1.4750e-01, -1.1558e+00],
[-9.2587e-01, -1.6393e+00, -1.1823e+00, ..., 4.1362e-02,
1.2494e-01, -9.7495e-01],
[-7.4055e-01, -1.5915e+00, -1.2104e+00, ..., -1.0356e+00,
-1.3046e+00, -7.5649e-01]],
[[-2.4195e+00, -4.8235e+00, -4.2231e+00, ..., -2.3066e+00,
-3.5205e+00, -1.2994e+00],
[-5.3077e+00, -7.9901e+00, -6.5265e+00, ..., -3.3578e+00,
-5.0018e+00, -2.0577e+00],
[-4.2923e+00, -6.2011e+00, -5.4221e+00, ..., -3.0175e+00,
-3.8648e+00, -1.7522e+00],
...,
[-4.3535e+00, -6.0644e+00, -5.4726e+00, ..., -3.1088e+00,
-4.0135e+00, -2.0924e+00],
[-6.1781e+00, -7.3799e+00, -6.2270e+00, ..., -4.1940e+00,
-5.8364e+00, -2.0763e+00],
[ 1.9693e-03, -7.9594e-01, 3.9671e-01, ..., 4.5255e-01,
6.0441e-01, 1.2892e-01]],
[[-6.2160e-01, 5.8724e-01, 1.1031e+00, ..., 8.0570e-01,
5.4441e-01, 1.3059e+00],
[-9.3676e-01, -7.0218e-02, 3.9276e-01, ..., 3.6632e-01,
3.6111e-01, 2.1873e-01],
[-1.6295e+00, -1.6519e-01, -7.1874e-02, ..., 1.1044e-02,
2.2732e-01, 9.1034e-02],
...,
[-1.7101e+00, -2.7647e-02, -8.3839e-02, ..., 7.4504e-01,
2.8060e-01, -4.0621e-01],
[-1.5648e+00, 6.3753e-02, 3.3858e-02, ..., 9.5446e-02,
2.2726e-01, 1.4382e-01],
[-1.3840e+00, -4.5645e-01, -2.6309e-01, ..., -1.5180e-02,
2.0019e-01, -1.2721e-01]],
...,
[[ 1.0695e+00, -3.8892e-01, -2.1227e-01, ..., 3.5881e-01,
2.5867e-01, -7.3840e-01],
[-8.2380e-01, -9.6564e-01, -8.1372e-01, ..., -2.3593e-01,
-9.7705e-01, -9.3537e-01],
[-6.9933e-01, -9.7150e-01, -7.1882e-01, ..., -1.3945e-01,
-2.1590e-01, -6.2125e-01],
...,
[-7.4814e-01, -1.1540e+00, -7.5394e-01, ..., -6.1453e-01,
-9.2958e-01, -1.1648e+00],
[-2.7281e+00, -2.7992e+00, -1.8727e+00, ..., -1.1829e+00,
-1.2714e+00, -1.2196e+00],
[-6.7075e-01, -1.5909e+00, -9.1097e-01, ..., -1.5373e-01,
-2.2882e-01, -5.2900e-01]],
[[ 7.6836e-03, -3.3665e-03, 5.8402e-01, ..., 2.9176e-01,
7.5264e-01, 5.5836e-02],
[-3.5905e-02, -7.4586e-01, 1.5610e-01, ..., 4.4349e-01,
6.7039e-01, -2.2820e-01],
[-1.3401e-01, -8.1345e-01, 3.0532e-01, ..., 4.1587e-01,
3.8762e-01, -1.2373e-01],
...,
[ 4.0156e-02, -8.3305e-01, 2.8564e-01, ..., 5.9865e-01,
4.8193e-01, -7.1463e-01],
[-2.3755e-01, -1.4797e+00, 2.9188e-04, ..., -8.7205e-03,
3.4087e-02, -5.5129e-01],
[ 7.3030e-01, 1.7660e-01, 3.0700e-01, ..., 2.1708e-01,
2.5000e-01, 3.4249e-01]],
[[-1.5956e+00, -6.0133e-01, 7.0815e-02, ..., 2.5815e-01,
1.8751e-01, 8.0211e-01],
[-2.1007e+00, -3.5871e-01, -7.7966e-03, ..., -4.1358e-01,
-1.1685e-01, 1.1580e+00],
[-1.7441e+00, -4.0124e-01, 2.0417e-01, ..., -7.9591e-02,
3.8289e-02, 9.4032e-01],
...,
[-1.7026e+00, -4.8279e-01, 2.2951e-01, ..., -1.6030e-01,
4.1075e-01, 1.4937e+00],
[-2.1050e+00, -1.0273e+00, -1.5363e-01, ..., -2.5507e-01,
-1.1512e-01, 5.5708e-01],
[-9.8875e-01, -1.2760e-01, 4.5583e-01, ..., -2.8857e-01,
-3.1073e-01, 6.8061e-01]]],
[[[-1.9655e-01, -3.0255e-01, 1.2018e-01, ..., 6.7886e-03,
4.5995e-01, 1.0263e-01],
[-1.3377e-01, -1.3688e+00, -8.0575e-01, ..., -7.8316e-01,
-4.8354e-01, -3.8833e-01],
[ 8.8477e-02, -1.4261e+00, -2.7174e+00, ..., -5.3265e-01,
-4.5573e-01, -7.7732e-01],
...,
[-2.9796e-01, -1.6777e+00, 2.0922e-01, ..., -7.0305e-01,
-8.7668e-01, -2.1287e+00],
[-1.2323e+00, -1.8098e+00, -4.4572e-01, ..., -8.9840e-01,
-6.6996e-01, -1.8082e+00],
[-1.3417e+00, -2.2657e+00, -1.5820e+00, ..., -1.4849e+00,
-1.6014e+00, -1.0042e+00]],
[[-3.1550e-01, -1.6657e+00, -9.3418e-01, ..., -8.4209e-01,
-1.5228e+00, -3.3905e-01],
[-2.2216e+00, -6.0074e+00, -6.8577e+00, ..., -2.5858e+00,
-3.5942e+00, -1.2478e+00],
[-3.9218e+00, -9.6856e+00, -1.1623e+01, ..., -2.8257e+00,
-3.2227e+00, -1.3623e+00],
...,
[-3.8451e+00, -5.6821e+00, -9.5407e+00, ..., -6.7294e+00,
-6.9359e+00, -2.9501e+00],
[-6.2610e+00, -7.1694e+00, -6.4148e+00, ..., -6.6141e+00,
-7.5079e+00, -2.7366e+00],
[-8.1638e-02, -9.4527e-01, 3.6612e-02, ..., 3.7903e-01,
6.7804e-01, 2.1983e-01]],
[[-5.3895e-01, -1.2473e-01, 4.2251e-01, ..., 4.4522e-01,
1.4886e-01, 9.4483e-01],
[-4.0650e-01, -6.3807e-01, -2.0670e-01, ..., 3.7790e-01,
3.1376e-01, 4.9490e-01],
[-1.0663e+00, -1.3174e+00, -1.7237e+00, ..., 7.0220e-02,
1.9533e-01, 4.0027e-01],
...,
[-1.4422e+00, 1.5093e-01, 1.7031e+00, ..., -7.6012e-02,
6.6897e-01, 2.7743e-01],
[-1.2279e+00, 1.2617e-01, 1.7205e+00, ..., 2.5805e-01,
2.5700e-01, 2.8656e-01],
[-1.2008e+00, -5.5581e-02, 1.1886e+00, ..., -1.8288e-02,
-7.6195e-02, -4.1290e-01]],
...,
[[ 9.4710e-02, -5.3890e-01, -3.3430e-01, ..., -1.5409e-02,
-1.6562e-01, -7.2244e-01],
[-8.2052e-01, -5.4620e-01, -3.4232e-01, ..., -2.7567e-01,
-1.0089e+00, -7.9219e-01],
[-4.8439e-01, 5.5326e-01, 1.4453e+00, ..., 6.8299e-02,
-2.0946e-01, -5.4822e-01],
...,
[-6.0030e-01, -7.5988e-01, -2.0142e-01, ..., -4.4613e+00,
-1.7217e+00, -1.2180e+00],
[-3.2413e+00, -3.0610e+00, -2.3554e+00, ..., -2.8132e+00,
-1.7445e+00, -2.0755e+00],
[-5.4488e-01, -1.6720e+00, -2.0849e+00, ..., -8.7225e-01,
-9.2452e-01, -1.3505e+00]],
[[-1.5686e-01, 2.4581e-02, 3.7074e-01, ..., 8.4173e-02,
5.5781e-01, 1.5895e-01],
[-4.4178e-01, 1.4111e-03, 5.4617e-01, ..., -5.7285e-02,
1.4715e-01, -9.8037e-02],
[-3.5210e-01, -2.8229e-02, -1.3545e+00, ..., 4.6888e-02,
6.0848e-02, -8.2360e-02],
...,
[-1.4782e-01, -1.0131e+00, -2.1329e-01, ..., 4.3456e-01,
7.7489e-02, -7.4111e-01],
[-5.3364e-01, -1.8679e+00, 7.9763e-02, ..., -5.3415e-01,
-1.0610e-01, -9.2155e-01],
[ 4.5662e-01, -1.5044e-01, 8.4752e-01, ..., 2.8295e-01,
3.4341e-01, 3.7693e-01]],
[[-6.3351e-01, 4.0840e-01, 5.6269e-01, ..., 4.7043e-01,
3.7575e-01, 6.3949e-01],
[-8.9742e-01, 4.0669e-01, -8.5040e-02, ..., -2.9871e-01,
-1.4042e-01, 7.2097e-01],
[-3.5152e-01, -1.5983e-01, -2.3684e+00, ..., 7.5625e-02,
1.3171e-01, 6.3932e-01],
...,
[-1.1816e+00, -1.0012e+00, -2.1868e+00, ..., 3.1745e-01,
4.5226e-01, 8.1503e-01],
[-1.9880e+00, -1.9183e+00, -2.5287e+00, ..., 4.5787e-02,
-1.9377e-01, 4.4618e-01],
[-9.6870e-01, -9.8922e-01, -1.1476e+00, ..., -2.9882e-01,
-1.7142e-01, 1.1840e+00]]]], device='cuda:0'),
tensor([[[[-2.2621e-01, -8.4946e-01, -7.4775e-01, ..., -5.5310e-01,
-1.5985e-02, -7.3225e-01],
[-9.0351e-02, -1.7011e+00, -9.1885e-01, ..., -1.0118e+00,
-8.3998e-01, -1.1616e+00],
[ 1.7432e-01, -1.2570e+00, -9.7126e-01, ..., -7.5853e-01,
-8.2328e-01, -1.7398e+00],
...,
[ 8.9460e-02, -4.2257e-01, -5.8753e-03, ..., -8.2991e-02,
-6.4228e-03, -4.3312e-01],
[-3.6443e-01, -4.4183e-01, 4.5916e-02, ..., 1.3982e-01,
3.0274e-01, -1.4034e-01],
[-6.5128e-01, -1.3057e+00, -1.1061e+00, ..., -1.0825e+00,
-1.1847e+00, -4.8210e-01]],
[[-2.8176e+00, -4.7703e+00, -4.5211e+00, ..., -3.8553e+00,
-4.9424e+00, -1.9475e+00],
[-3.4065e+00, -5.7061e+00, -5.2592e+00, ..., -5.5071e+00,
-6.9468e+00, -3.0192e+00],
[-3.4062e+00, -5.2773e+00, -5.0982e+00, ..., -5.1848e+00,
-5.7737e+00, -2.8184e+00],
...,
[-1.7317e+00, -2.1991e+00, -1.8312e+00, ..., -1.8215e+00,
-2.3031e+00, -8.8382e-01],
[-2.8265e+00, -3.0313e+00, -2.0849e+00, ..., -2.1820e+00,
-3.3021e+00, -9.3206e-01],
[ 3.4448e-01, -2.5234e-01, 3.2615e-01, ..., 2.7934e-01,
4.2965e-01, 2.1446e-01]],
[[-4.1913e-01, 3.2331e-01, 4.6777e-01, ..., 8.1893e-01,
5.3812e-01, 1.5039e+00],
[-7.2412e-01, 3.8984e-01, 6.5101e-01, ..., 4.1246e-01,
2.0959e-01, 3.9600e-01],
[-1.0709e+00, 7.7521e-02, 1.1765e-02, ..., 1.4084e-01,
1.3569e-01, 3.5517e-01],
...,
[-6.1232e-01, 3.4588e-01, 2.2645e-01, ..., 1.6643e-01,
2.6440e-01, 4.1297e-01],
[-4.3545e-01, 1.8414e-02, 1.8403e-01, ..., 1.5820e-01,
2.5128e-01, 5.5575e-01],
[-1.0915e-01, -2.4470e-01, 1.3416e-01, ..., -2.6220e-02,
1.2910e-01, 3.6075e-01]],
...,
[[ 2.8545e-01, -1.3487e-01, 2.8202e-01, ..., 4.6046e-01,
2.1709e-01, -1.0547e+00],
[ 2.4892e-02, -2.1148e-02, 6.7891e-03, ..., -4.0573e-01,
-1.2601e+00, -1.3631e+00],
[-4.0017e-01, -6.0679e-01, -4.0821e-01, ..., -2.8072e-01,
-5.5323e-01, -1.1810e+00],
...,
[-6.4615e-01, -6.9852e-01, -5.3202e-01, ..., -3.1362e-01,
-4.9429e-01, -5.7430e-01],
[-2.2650e+00, -1.7978e+00, -1.3179e+00, ..., -1.3351e+00,
-1.3759e+00, -1.1111e+00],
[-5.8209e-01, -7.8753e-01, -4.4447e-01, ..., -2.7400e-01,
-3.8029e-01, -4.0326e-01]],
[[ 8.7772e-02, -1.5256e-01, -2.1616e-01, ..., 2.6252e-01,
7.7132e-01, -2.2590e-01],
[ 6.3836e-02, -5.1403e-01, 3.3616e-01, ..., 1.7804e-01,
3.6375e-01, -8.0344e-01],
[ 8.5931e-02, -6.8228e-01, 9.5172e-02, ..., 2.2941e-01,
1.3489e-01, -6.6702e-01],
...,
[-2.1438e-01, -3.5069e-01, 2.0478e-01, ..., 3.5097e-01,
3.0561e-01, 4.9475e-02],
[-5.3635e-01, -8.9750e-01, 3.1294e-02, ..., 5.6732e-03,
-3.0390e-02, -1.7007e-01],
[ 3.3019e-01, 2.4043e-05, -2.0153e-01, ..., -1.4178e-01,
-1.1003e-01, 1.6384e-01]],
[[-3.5212e-01, -1.8574e-02, 8.2875e-02, ..., 3.3500e-01,
4.4176e-01, 9.8729e-01],
[-1.2207e+00, -9.0267e-01, -8.6256e-01, ..., -6.3321e-01,
-2.4982e-01, 9.8865e-01],
[-7.0131e-01, -4.2947e-01, -2.4765e-01, ..., -3.6385e-01,
-7.1540e-02, 7.4737e-01],
...,
[-6.8473e-01, -2.0471e-01, 2.8652e-01, ..., -1.2295e-01,
-8.9641e-04, 6.2437e-01],
[-1.1831e+00, -5.4903e-01, -3.1022e-01, ..., -3.4574e-01,
-2.3470e-01, 1.3793e-01],
[-3.7552e-01, -4.6852e-03, -1.4766e-01, ..., -2.2450e-01,
-2.0910e-01, 2.9661e-01]]],
[[[ 4.0634e-02, -2.4955e-01, -3.5807e-02, ..., -9.8260e-01,
-5.3325e-01, -1.4543e+00],
[ 3.3785e-01, -9.1968e-01, -7.1056e-01, ..., -1.1756e+00,
-1.3093e+00, -1.4736e+00],
[ 2.6732e-01, -5.4630e-01, -5.2795e-01, ..., -1.2034e+00,
-1.0885e+00, -1.8126e+00],
...,
[ 2.5648e-02, -5.5533e-01, -3.3094e-01, ..., -1.0505e+00,
-1.0220e+00, -1.9140e+00],
[-6.1536e-01, -6.6947e-01, 5.5988e-02, ..., -4.8068e-01,
-5.4741e-01, -1.7046e+00],
[-7.5910e-01, -1.3472e+00, -1.2072e+00, ..., -1.1665e+00,
-1.5235e+00, -7.6542e-01]],
[[-7.6878e-01, -2.1603e+00, -1.8627e+00, ..., -3.9690e+00,
-5.4787e+00, -1.8528e+00],
[-2.0061e+00, -3.7444e+00, -3.0023e+00, ..., -5.8727e+00,
-8.0637e+00, -3.3794e+00],
[-1.6106e+00, -2.5712e+00, -2.5658e+00, ..., -6.3977e+00,
-6.1660e+00, -2.5580e+00],
...,
[-1.8034e+00, -2.4089e+00, -2.6065e+00, ..., -5.1144e+00,
-6.3067e+00, -2.4919e+00],
[-3.8222e+00, -4.4536e+00, -4.0235e+00, ..., -6.4441e+00,
-7.8516e+00, -2.4990e+00],
[ 2.9166e-01, -3.1696e-01, 3.8202e-01, ..., 4.1530e-01,
4.8369e-01, 2.4296e-01]],
[[-2.0916e-01, 1.6431e-01, 7.3813e-01, ..., 6.2043e-01,
2.5808e-01, 1.5169e+00],
[ 9.7037e-02, -3.4243e-01, 5.9137e-01, ..., 5.4863e-01,
-6.8070e-02, 4.1773e-02],
[-1.3945e-01, -2.3230e-01, 2.4213e-01, ..., 1.2886e-01,
-2.5788e-02, -1.6174e-01],
...,
[-2.5905e-01, -2.0039e-01, 7.8538e-01, ..., 3.8839e-02,
-1.1555e-01, -1.8911e-01],
[-2.4595e-01, -1.4215e-01, 7.4658e-01, ..., -4.1514e-02,
-6.2282e-03, -2.3575e-01],
[-1.3405e-01, -5.0733e-01, 7.8039e-02, ..., 1.0532e-01,
2.5807e-01, -5.3445e-01]],
...,
[[ 5.9387e-01, -2.8041e-01, 2.5826e-03, ..., 4.4248e-01,
4.0664e-01, -1.2860e+00],
[-5.0657e-01, -5.6463e-01, -5.3547e-01, ..., -1.5854e-01,
-7.6046e-01, -1.3916e+00],
[-3.8267e-01, -2.1249e-01, -2.3518e-01, ..., 6.1128e-02,
-8.5473e-02, -9.4966e-01],
...,
[-4.4024e-01, -1.6124e-01, -2.8008e-01, ..., -1.6958e-02,
-1.0160e-01, -9.2034e-01],
[-2.3355e+00, -1.7139e+00, -1.7427e+00, ..., -1.5618e+00,
-1.5992e+00, -1.8410e+00],
[-4.9567e-01, -8.2241e-01, -3.7770e-01, ..., -3.8118e-01,
-5.4061e-01, -1.1622e+00]],
[[ 1.6217e-01, 2.5906e-01, 6.2649e-01, ..., -7.7758e-02,
4.0631e-01, -3.4200e-01],
[ 1.4980e-01, 2.5933e-01, 4.4961e-01, ..., 1.8663e-01,
1.0458e-01, -8.0515e-01],
[-1.1467e-01, 8.3915e-02, 2.6190e-01, ..., -7.7963e-01,
1.6925e-01, -4.4231e-01],
...,
[-1.8030e-01, 8.0649e-03, 3.1463e-01, ..., 1.8291e-01,
1.5492e-01, -4.8491e-01],
[-5.4756e-01, -7.1163e-01, 1.0725e-01, ..., -1.9750e-01,
-2.9223e-01, -8.9478e-01],
[ 5.2592e-01, 4.0271e-01, 6.4096e-02, ..., 3.1444e-01,
3.2218e-01, 2.8859e-01]],
[[-8.2750e-01, 2.6259e-01, 3.1351e-01, ..., -1.0737e-01,
-1.0319e-01, 6.0168e-01],
[-1.1435e+00, 2.1594e-01, -3.7045e-01, ..., -9.5872e-01,
-4.4885e-01, 9.0490e-01],
[-9.4867e-01, 3.5488e-01, -1.1131e-01, ..., -4.7066e-01,
-5.4015e-01, 6.3055e-01],
...,
[-9.5173e-01, 2.0487e-01, -7.9092e-01, ..., -4.3814e-01,
-5.3946e-01, 6.2037e-01],
[-1.6936e+00, -3.8089e-01, -5.8694e-01, ..., -7.1653e-01,
-7.4272e-01, 3.9044e-01],
[-7.5438e-01, -1.0531e-01, -2.9855e-01, ..., -4.6074e-01,
-4.2267e-01, 9.9420e-01]]],
[[[-2.2110e-01, -3.6771e-01, 6.1895e-02, ..., -5.5385e-02,
3.6742e-01, 6.4927e-03],
[-1.6081e-01, -1.6025e+00, -9.3843e-01, ..., -1.6636e+00,
-7.6736e-01, -5.2154e-01],
[-1.2401e-01, -1.3878e+00, -2.3212e-01, ..., -1.0116e+00,
-3.0276e-01, -8.3889e-01],
...,
[ 1.5961e-01, -4.5607e+00, -5.0490e+00, ..., -3.8649e+00,
-1.1442e+00, -2.5678e+00],
[ 6.0702e-01, -1.5593e+00, -1.4630e+00, ..., -6.5054e-01,
-6.2685e-01, -1.9194e+00],
[-2.3256e-01, -1.2666e+00, -1.7248e+00, ..., -1.5183e+00,
-1.6236e+00, -1.0233e+00]],
[[-4.2946e-01, -1.7698e+00, -1.0495e+00, ..., -1.0109e+00,
-1.6483e+00, -4.4150e-01],
[-2.2421e+00, -4.3104e+00, -2.8514e+00, ..., -5.3887e+00,
-5.9210e+00, -1.3986e+00],
[-2.1142e+00, -3.6753e+00, -7.1059e+00, ..., -5.1956e+00,
-6.9886e+00, -1.4950e+00],
...,
[-8.4307e+00, -1.3368e+01, -1.6917e+01, ..., -6.7354e+00,
-7.4921e+00, -3.1312e+00],
[-6.0358e+00, -7.9558e+00, -8.8790e+00, ..., -6.5682e+00,
-7.9286e+00, -2.8044e+00],
[-2.3611e-01, -8.7159e-01, -9.4166e-01, ..., 3.8271e-01,
6.6725e-01, 2.1683e-01]],
[[-4.9016e-01, -1.1146e-02, 5.0874e-01, ..., 4.6903e-01,
2.2316e-01, 1.0018e+00],
[-4.3229e-01, -4.2578e-01, 4.8697e-01, ..., 6.6074e-01,
5.6669e-01, 4.9359e-01],
[-9.4238e-01, -2.8820e-01, 1.1339e-01, ..., -3.6653e-01,
1.2354e+00, 4.0092e-01],
...,
[-2.9003e+00, -1.1736e+01, -8.9410e+00, ..., 1.4878e+00,
-4.4134e-01, 9.8599e-01],
[-2.2953e-03, -4.4615e+00, -1.4171e-01, ..., -1.8774e-01,
-1.7816e-02, 2.3431e-01],
[ 7.8486e-01, -3.4286e-01, 1.1949e+00, ..., -5.3345e-02,
-8.2022e-02, -4.1624e-01]],
...,
[[ 1.3192e-01, -4.2411e-01, -2.4962e-01, ..., 1.2585e-02,
-1.4863e-01, -7.2426e-01],
[-7.5977e-01, -5.0049e-01, -4.8034e-01, ..., 5.2204e-01,
-7.2609e-01, -8.1337e-01],
[-4.8559e-01, -1.2635e-01, 4.8866e-02, ..., -8.6839e-01,
-8.1797e-01, -5.4123e-01],
...,
[ 1.4998e+00, -1.1079e+00, -5.9329e+00, ..., -4.1789e+00,
-3.0565e+00, -1.4124e+00],
[-2.3749e+00, -2.0245e+00, -7.1586e+00, ..., -2.1260e+00,
-2.1449e+00, -2.1142e+00],
[-1.8890e+00, -3.5717e+00, -4.6577e+00, ..., -8.4808e-01,
-9.2810e-01, -1.4082e+00]],
[[-1.4973e-01, 2.9291e-02, 3.7267e-01, ..., 7.4662e-02,
5.0706e-01, 1.2168e-01],
[-4.1023e-01, -2.3405e-01, 6.8217e-02, ..., -1.6285e+00,
-1.4807e+00, -1.7937e-01],
[-6.3776e-01, -5.1391e-01, 8.2944e-01, ..., 1.1588e-01,
-1.1811e+00, -1.0191e-01],
...,
[-1.3137e+00, -5.1838e+00, -5.2506e+00, ..., -2.3192e+00,
9.4205e-01, -7.9941e-01],
[-2.5889e-01, -3.8516e+00, -1.9934e+00, ..., -3.3034e-01,
-2.0041e-01, -1.0094e+00],
[ 9.6279e-01, -4.9405e-01, -1.6891e+00, ..., 3.1897e-01,
3.5384e-01, 3.7346e-01]],
[[-5.9387e-01, 3.4720e-01, 5.3637e-01, ..., 4.7991e-01,
4.1213e-01, 7.0845e-01],
[-8.9826e-01, 1.4150e-01, -2.0948e-01, ..., -1.8441e-01,
-7.5263e-02, 7.7363e-01],
[-5.4656e-01, 2.1235e-01, 1.1362e-02, ..., -2.1907e-01,
7.5448e-01, 6.6694e-01],
...,
[-2.6458e+00, -4.5723e+00, -8.1223e+00, ..., 1.3189e-01,
1.2871e+00, 1.6067e+00],
[-3.1095e+00, -2.5022e+00, -3.3367e+00, ..., -6.7497e-01,
-3.9636e-01, 4.7627e-01],
[-1.4883e+00, 1.8882e-01, 1.1177e+00, ..., -2.9417e-01,
-1.6030e-01, 1.2210e+00]]],
[[[-7.4201e-01, -1.4382e+00, -1.4620e+00, ..., -4.1713e-01,
5.7884e-01, 4.8094e-01],
[-3.0118e-01, -2.4592e+00, -2.1498e+00, ..., -1.0730e+00,
-7.6028e-01, 5.7589e-02],
[-4.3475e-01, -2.3657e+00, -2.6699e+00, ..., -1.2194e+00,
-4.3025e-01, -1.8320e-01],
...,
[-5.0139e-01, -1.0941e+00, -1.0518e-01, ..., -3.8545e-01,
-9.3786e-02, -1.4489e-01],
[-1.0004e+00, -1.4949e+00, -4.9284e-01, ..., -1.3402e-02,
2.3234e-01, -4.8319e-02],
[-8.1690e-01, -1.7155e+00, -1.2849e+00, ..., -1.0350e+00,
-1.0983e+00, -3.8921e-01]],
[[-2.4226e+00, -4.9530e+00, -4.2474e+00, ..., -1.7645e+00,
-1.1705e+00, -7.1549e-02],
[-5.5229e+00, -8.2967e+00, -7.0875e+00, ..., -5.4189e+00,
-4.1645e+00, -8.4934e-01],
[-5.5279e+00, -1.0408e+01, -1.3805e+01, ..., -5.0657e+00,
-4.1091e+00, -5.5040e-01],
...,
[-4.4584e+00, -6.7571e+00, -5.3571e+00, ..., -1.3932e+00,
-1.7423e+00, -4.0916e-01],
[-6.4766e+00, -7.7688e+00, -6.6545e+00, ..., -2.5817e+00,
-3.4668e+00, -9.8571e-01],
[-4.3841e-02, -8.2447e-01, 3.3803e-01, ..., 3.3689e-01,
3.3057e-01, 8.8538e-02]],
[[-6.3222e-01, 5.6733e-01, 1.1192e+00, ..., 9.2973e-01,
4.0831e-01, 1.1430e+00],
[-9.0662e-01, -1.0109e-01, 3.1267e-01, ..., 6.0162e-01,
1.4200e-01, 4.7510e-01],
[-2.1015e+00, -1.0800e+00, -1.0084e+00, ..., 5.3732e-01,
4.2727e-01, 4.7293e-01],
...,
[-1.7565e+00, 1.7344e+00, -3.3936e+00, ..., 5.2506e-02,
5.9928e-02, 4.4034e-01],
[-1.5773e+00, 7.7810e-01, 4.5603e-01, ..., 1.1861e-01,
1.3456e-01, 5.8483e-01],
[-1.4687e+00, -5.3732e-01, -2.8446e-01, ..., -1.1137e-01,
9.3122e-02, 3.4015e-01]],
...,
[[ 1.0681e+00, -4.5425e-01, -2.2711e-01, ..., -2.8928e-02,
-3.7083e-01, -9.3463e-01],
[-8.2765e-01, -9.7938e-01, -8.3970e-01, ..., -9.7294e-01,
-1.2709e+00, -8.9092e-01],
[-5.7984e-01, -6.6484e-01, 2.8644e-02, ..., -1.2220e+00,
-4.9563e-01, -4.8194e-01],
...,
[-7.3173e-01, -1.9949e+00, -5.8387e-01, ..., -1.2084e-01,
-1.8507e-01, -3.8948e-01],
[-2.7891e+00, -3.2592e+00, -2.6332e+00, ..., -8.2615e-01,
-8.9203e-01, -8.3742e-01],
[-6.5710e-01, -1.7181e+00, -1.0554e+00, ..., -1.4157e-02,
-1.8880e-01, -3.5912e-01]],
[[-2.2148e-03, -2.3802e-02, 5.9347e-01, ..., -6.7636e-01,
6.9111e-01, 2.9196e-01],
[-2.2178e-02, -7.2133e-01, 1.3145e-01, ..., -1.2207e+00,
-9.3930e-01, -1.1123e-01],
[-1.3746e-01, -3.0512e-01, 4.5877e-01, ..., -9.5268e-01,
-1.1767e+00, 1.1281e-01],
...,
[ 9.5932e-02, -4.8503e-01, -1.0341e+00, ..., -1.4715e-01,
-1.4895e-01, 1.6029e-01],
[-1.8001e-01, -1.0985e+00, 1.9287e-01, ..., -4.4022e-01,
-3.9035e-01, -5.1981e-02],
[ 6.5354e-01, 1.2098e-01, 2.4950e-01, ..., -2.5480e-01,
-2.5453e-01, 3.2733e-01]],
[[-1.6992e+00, -6.6250e-01, 1.3242e-02, ..., 6.5257e-01,
3.5077e-01, 5.9071e-01],
[-2.2023e+00, -3.3369e-01, -6.4712e-03, ..., -6.8971e-01,
-4.0773e-02, 4.6222e-01],
[-1.7251e+00, -4.9041e-01, 2.4432e-01, ..., -1.0544e+00,
-1.8078e-01, 1.0506e-01],
...,
[-1.8045e+00, -1.4633e+00, 4.1915e-01, ..., 1.2142e-01,
-1.1635e-01, 1.3710e-01],
[-2.3020e+00, -1.6766e+00, -2.7958e-01, ..., -6.2322e-02,
-3.3854e-01, -1.6261e-01],
[-1.1247e+00, -1.8706e-01, 4.1695e-01, ..., -9.7279e-02,
-2.9589e-01, 1.5763e-01]]],
[[[-7.2437e-01, -1.0829e+00, -6.1111e+00, ..., -3.2618e-01,
3.8064e-01, 4.3051e-01],
[-2.6036e-01, -1.7481e+00, -4.8981e+00, ..., -6.0409e-01,
-1.8193e-01, 2.4921e-01],
[-3.2957e-01, -1.5123e+00, -3.6701e+00, ..., -4.0146e-01,
-1.2910e-01, 3.4339e-03],
...,
[-4.7499e-01, -1.8785e+00, -1.7274e+00, ..., -2.7125e+00,
1.2961e+00, 2.3956e-01],
[-9.7107e-01, -1.7133e+00, -1.2318e+00, ..., 1.3290e+00,
2.3750e+00, -4.6307e-01],
[-7.7900e-01, -1.6370e+00, -1.2450e+00, ..., -2.4923e-01,
-1.9151e+00, -1.5702e+00]],
[[-2.4034e+00, -6.9918e+00, -6.4042e+00, ..., -1.5349e+00,
-1.6802e+00, -2.1346e-01],
[-5.5225e+00, -1.0109e+01, -7.4107e+00, ..., -2.7867e+00,
-3.0909e+00, -7.2074e-01],
[-4.7884e+00, -7.8135e+00, -5.5668e+00, ..., -1.6250e+00,
-1.6999e+00, -3.3975e-01],
...,
[-4.4351e+00, -6.2007e+00, -5.6310e+00, ..., -6.8134e+00,
-4.7519e+00, -2.5196e+00],
[-6.3098e+00, -7.4736e+00, -6.4279e+00, ..., -2.1499e+00,
-4.2886e+00, -1.6977e+00],
[-2.0869e-02, -8.2349e-01, 3.6362e-01, ..., -5.4377e-01,
-2.4167e-01, -1.3235e-01]],
[[-6.1140e-01, 3.3397e-01, -7.8780e+00, ..., 9.9428e-01,
3.8789e-01, 1.0646e+00],
[-9.1197e-01, 6.8282e-01, -1.1437e+01, ..., 3.3379e-01,
1.5807e-01, 4.1363e-01],
[-1.6268e+00, 6.7260e-01, -9.0943e+00, ..., -5.6937e-01,
8.3273e-02, 4.2470e-01],
...,
[-1.7305e+00, 1.0902e-01, 5.3053e-03, ..., -3.9480e+00,
-2.6636e+00, 8.0965e-01],
[-1.5370e+00, 7.7576e-02, 7.4548e-02, ..., -1.7113e+00,
-8.6368e-01, 1.9139e+00],
[-1.4330e+00, -5.6972e-01, -3.3639e-01, ..., -8.9735e-01,
-2.4525e-01, 1.4630e+00]],
...,
[[ 1.0415e+00, 7.1991e-01, -1.5139e+00, ..., -1.2284e-01,
-5.0790e-01, -9.4579e-01],
[-8.3971e-01, -3.2700e-01, -3.0106e-01, ..., -5.8729e-01,
-1.4447e+00, -8.6934e-01],
[-6.2281e-01, -6.4946e-01, -2.0738e-01, ..., -6.7285e-01,
-6.7904e-01, -6.0238e-01],
...,
[-7.3493e-01, -1.3674e+00, -1.2100e+00, ..., -5.8880e+00,
-6.1636e+00, -3.1325e+00],
[-2.7508e+00, -2.8266e+00, -1.8784e+00, ..., -6.3053e+00,
-5.4744e+00, -2.4760e+00],
[-6.6067e-01, -1.6556e+00, -9.7889e-01, ..., -3.6620e+00,
-2.1360e+00, -7.2997e-01]],
[[ 1.6175e-03, 5.4239e-01, -3.9162e+00, ..., 2.8673e-01,
4.7663e-01, 2.3677e-01],
[-1.8941e-02, -8.8779e-02, -5.3006e+00, ..., 1.2093e-01,
-1.7004e-01, -3.0947e-02],
[ 2.3524e-04, -2.0302e-01, -3.3650e+00, ..., 4.9186e-02,
-6.8583e-01, -1.6974e-01],
...,
[ 5.3906e-02, -7.6725e-01, 2.7910e-01, ..., -4.5101e+00,
-1.4120e+00, 2.0213e-01],
[-2.0560e-01, -1.4239e+00, -5.0966e-03, ..., 4.6695e-01,
1.6322e+00, -2.8100e-01],
[ 6.9613e-01, 1.9216e-01, 3.0292e-01, ..., 1.0383e+00,
-6.5709e-01, -8.0120e-01]],
[[-1.6378e+00, -1.4924e+00, 3.7368e-01, ..., 3.2886e-01,
9.7512e-02, 4.5517e-01],
[-2.1185e+00, -1.5711e+00, 2.5899e+00, ..., -2.1468e-01,
-5.0966e-01, 2.8777e-01],
[-1.7715e+00, -1.5902e+00, 2.4671e+00, ..., 8.8833e-02,
-5.4693e-01, -1.8546e-01],
...,
[-1.7927e+00, -4.8874e-01, 4.1047e-01, ..., -7.7696e+00,
-1.7998e+00, 4.8490e-01],
[-2.2663e+00, -1.1090e+00, -1.3228e-01, ..., -2.8128e+00,
2.4001e-01, 7.5418e-02],
[-1.1072e+00, -1.7573e-01, 4.2767e-01, ..., 1.9387e+00,
9.8979e-01, 1.4747e-01]]]], device='cuda:0'),
tensor([[[[-7.5766e-01, -1.3773e+00, -1.4005e+00, ..., -2.2220e-01,
2.5470e-01, -3.6516e-01],
[-3.2201e-01, -2.3069e+00, -2.1700e+00, ..., -5.7422e-01,
-3.8243e-01, -5.3063e-01],
[-3.7596e-01, -1.9782e+00, -1.8535e+00, ..., -3.6826e-01,
-3.5223e-01, -1.0168e+00],
...,
[-5.5768e-01, -2.0476e+00, -1.8693e+00, ..., -1.1439e-01,
-3.5768e-01, -1.3397e+00],
[-1.2144e+00, -1.8519e+00, -1.1267e+00, ..., -8.1220e-03,
8.1885e-02, -9.5343e-01],
[-1.0950e+00, -1.9911e+00, -1.4416e+00, ..., -9.1772e-01,
-1.1521e+00, -6.5696e-01]],
[[-2.4815e+00, -4.8777e+00, -4.2871e+00, ..., -2.1241e+00,
-3.4984e+00, -1.2631e+00],
[-5.2066e+00, -7.8130e+00, -6.5095e+00, ..., -3.3532e+00,
-5.1340e+00, -2.0847e+00],
[-4.3574e+00, -6.1205e+00, -5.4676e+00, ..., -2.9864e+00,
-3.7981e+00, -1.7082e+00],
...,
[-4.3374e+00, -6.0974e+00, -5.4611e+00, ..., -3.1667e+00,
-4.1045e+00, -2.0000e+00],
[-6.6668e+00, -7.8203e+00, -6.5386e+00, ..., -3.9251e+00,
-5.4866e+00, -1.9737e+00],
[-6.5102e-02, -1.0731e+00, 4.2410e-01, ..., 4.0820e-01,
5.5494e-01, 1.6858e-01]],
[[-5.9175e-01, 6.6867e-01, 1.0348e+00, ..., 8.2375e-01,
5.7621e-01, 1.3735e+00],
[-9.4635e-01, 5.1703e-02, 3.5660e-01, ..., 4.2217e-01,
3.6467e-01, 2.4020e-01],
[-1.6595e+00, -7.9282e-02, -5.2582e-02, ..., 7.2102e-02,
2.9821e-01, 1.0953e-01],
...,
[-1.6869e+00, -2.2929e-01, -8.9995e-02, ..., 1.0627e-01,
-1.5045e-02, 2.5262e-01],
[-1.4636e+00, -2.2913e-02, -2.6838e-03, ..., -3.0229e-02,
1.8490e-01, 1.8964e-01],
[-1.4065e+00, -5.7161e-01, -2.5243e-01, ..., -4.5844e-02,
1.6467e-01, -1.1890e-01]],
...,
[[ 1.1789e+00, -3.6632e-01, -2.2235e-01, ..., 3.4940e-01,
1.8479e-01, -7.9483e-01],
[-7.1199e-01, -9.7844e-01, -8.1062e-01, ..., -3.5221e-01,
-1.0785e+00, -9.7115e-01],
[-6.6513e-01, -1.0281e+00, -7.2959e-01, ..., -1.5343e-01,
-2.3654e-01, -6.0661e-01],
...,
[-8.3033e-01, -1.1296e+00, -7.6728e-01, ..., -4.3265e-01,
-6.3514e-01, -1.1728e+00],
[-3.1958e+00, -3.0602e+00, -2.0116e+00, ..., -1.1139e+00,
-1.1327e+00, -1.1237e+00],
[-6.6615e-01, -1.7252e+00, -1.0201e+00, ..., -1.7811e-01,
-2.7533e-01, -5.0797e-01]],
[[ 8.7390e-02, 6.3237e-02, 6.0093e-01, ..., 3.3455e-01,
8.4702e-01, 7.5447e-02],
[ 8.4533e-02, -7.4795e-01, 1.0473e-01, ..., 4.0667e-01,
6.6970e-01, -2.7124e-01],
[ 1.0070e-02, -7.2391e-01, 2.6686e-01, ..., 3.0605e-01,
4.4001e-01, -1.0066e-01],
...,
[-1.3553e-01, -8.9221e-01, 2.7516e-01, ..., 5.4641e-01,
-5.9840e-02, -5.2549e-01],
[-4.8817e-01, -1.6282e+00, -4.2527e-02, ..., 1.3107e-01,
1.2692e-01, -5.0437e-01],
[ 5.5066e-01, 9.2529e-02, 1.9994e-01, ..., 2.6552e-01,
3.0946e-01, 3.4234e-01]],
[[-1.5483e+00, -6.2086e-01, 2.8925e-03, ..., 3.3901e-01,
2.4927e-01, 8.3508e-01],
[-2.0709e+00, -3.4403e-01, -2.3028e-02, ..., -4.4894e-01,
-1.2271e-01, 1.1801e+00],
[-1.7303e+00, -3.6193e-01, 2.2310e-01, ..., -7.4597e-02,
1.8814e-02, 9.5831e-01],
...,
[-1.8023e+00, -5.5235e-01, 2.6611e-01, ..., 4.9803e-02,
3.5377e-01, 1.0095e+00],
[-2.3414e+00, -1.3852e+00, -2.1115e-01, ..., -2.6668e-01,
-4.4637e-02, 6.0772e-01],
[-1.0854e+00, -4.3044e-01, 3.4814e-01, ..., -2.5261e-01,
-2.5573e-01, 7.2053e-01]]],
[[[-7.8761e-01, -1.5090e+00, -1.5169e+00, ..., -1.1009e+00,
-4.4435e-01, 7.6901e-01],
[-3.7384e-01, -2.0609e+00, -4.1794e+00, ..., -5.3244e+00,
-3.5509e+00, 2.7021e+00],
[-4.1770e-01, -1.6879e+00, -8.2611e+00, ..., -3.3467e+00,
-2.6395e+00, 2.0859e+00],
...,
[-5.4825e-01, -1.9627e+00, 1.4401e-01, ..., -4.1999e-01,
-2.4801e-01, -2.7507e-01],
[-1.1602e+00, -1.9253e+00, -1.0542e+00, ..., -2.2052e-01,
2.6248e-01, -6.4139e-02],
[-9.0813e-01, -1.8450e+00, -1.3748e+00, ..., -1.2142e+00,
-1.3033e+00, -4.9588e-01]],
[[-2.4618e+00, -4.9728e+00, -4.2514e+00, ..., -5.5559e+00,
-6.8104e+00, -2.6155e+00],
[-5.6207e+00, -1.3050e+01, -1.2588e+01, ..., -1.1675e+01,
-1.0613e+01, -4.0017e+00],
[-4.5353e+00, -1.3008e+01, -1.1440e+01, ..., -1.1986e+01,
-1.1692e+01, -3.3177e+00],
...,
[-4.6365e+00, -6.6571e+00, -6.2390e+00, ..., -2.3531e+00,
-2.1290e+00, -6.7516e-01],
[-6.6717e+00, -7.8411e+00, -7.0105e+00, ..., -3.3193e+00,
-4.0556e+00, -1.2523e+00],
[-1.2823e-01, -9.3023e-01, 2.6302e-01, ..., 3.9047e-01,
4.0075e-01, 4.2546e-02]],
[[-6.5204e-01, 5.3851e-01, 1.1050e+00, ..., 8.2425e-01,
6.1177e-01, 1.9737e+00],
[-8.4086e-01, -9.3991e-01, -2.0844e+00, ..., -1.9759e+00,
-4.4165e+00, -1.3985e+00],
[-1.6561e+00, -1.4040e+00, -7.1963e+00, ..., -5.7730e+00,
-1.0119e+01, -3.5001e+00],
...,
[-1.8153e+00, 3.3081e-01, 1.2631e+00, ..., 2.4809e-01,
3.2482e-02, 4.3066e-01],
[-1.5364e+00, 2.5193e-02, 1.0531e+00, ..., 1.5406e-01,
1.8608e-01, 6.4716e-01],
[-1.5113e+00, -6.5069e-01, -3.2511e-01, ..., -1.5662e-01,
6.4861e-02, 3.8934e-01]],
...,
[[ 1.1372e+00, -4.9207e-01, -2.5535e-01, ..., 1.7641e-01,
-6.4056e-01, -1.8504e+00],
[-8.0303e-01, -7.9641e-01, -1.3862e-01, ..., -4.6928e-01,
-5.1708e+00, -6.5278e+00],
[-6.4934e-01, 2.8432e-01, 5.4749e-01, ..., -2.1579e+00,
-8.0853e+00, -9.4648e+00],
...,
[-7.0329e-01, -1.7880e+00, -2.0712e+00, ..., -7.7708e-01,
-5.7707e-01, -6.4080e-01],
[-2.9444e+00, -3.1530e+00, -3.0680e+00, ..., -9.6720e-01,
-1.0622e+00, -9.2825e-01],
[-6.6030e-01, -1.7934e+00, -1.1363e+00, ..., -2.9454e-02,
-2.2540e-01, -4.4029e-01]],
[[-1.9913e-02, -7.1885e-02, 6.2938e-01, ..., 4.2995e-01,
-1.7381e-01, -7.5804e-01],
[-1.8034e-02, -2.1193e-01, -2.8566e-01, ..., -2.4046e+00,
-3.8127e+00, -1.2083e+00],
[-4.1295e-02, -1.7456e-01, -2.7546e+00, ..., -2.7072e+00,
-5.9166e+00, -6.1710e-01],
...,
[ 1.1926e-01, -6.8479e-01, 7.0917e-01, ..., -6.5742e-02,
-3.0159e-01, -6.9791e-03],
[-1.1685e-01, -1.3616e+00, 5.1283e-01, ..., -5.6987e-01,
-4.8589e-01, -2.3830e-01],
[ 6.5897e-01, 1.6052e-01, 2.7376e-01, ..., -2.7033e-01,
-3.1052e-01, 2.5341e-01]],
[[-1.8048e+00, -7.6732e-01, -8.7478e-02, ..., -1.0695e-01,
-9.1748e-01, -8.6248e-01],
[-2.3250e+00, -2.7026e-01, -5.6224e-01, ..., -2.4015e+00,
-4.9264e+00, -5.2184e+00],
[-1.9244e+00, -1.6134e+00, -3.1109e+00, ..., -2.7678e+00,
-6.1961e+00, -9.2288e+00],
...,
[-1.9862e+00, -7.1379e-01, -1.9902e-01, ..., 6.9319e-01,
1.9103e-01, 3.5789e-01],
[-2.5595e+00, -1.3213e+00, -8.1087e-01, ..., 2.0927e-01,
-2.5760e-01, -1.6105e-01],
[-1.2234e+00, -2.6246e-01, 3.5967e-01, ..., -1.2184e-01,
-3.6867e-01, 1.6681e-01]]],
[[[-1.9522e-01, -3.0197e-01, 1.3897e-01, ..., 1.2568e-03,
4.9155e-01, 1.3663e-01],
[-1.0950e-01, -1.2458e+00, -1.3846e+00, ..., -7.4943e-01,
-4.3820e-01, -3.7763e-01],
[-8.4164e-02, -1.0071e+00, -2.0075e+00, ..., -5.1373e-01,
-3.9932e-01, -7.5545e-01],
...,
[-1.3633e-01, -1.3615e+00, -3.6482e+00, ..., -1.2844e+00,
-1.2984e+00, -2.0106e+00],
[-4.7430e-01, -9.5856e-01, -1.0435e+00, ..., -2.9944e-01,
-5.3347e-01, -1.4858e+00],
[-2.8530e-01, -8.2721e-01, 2.5649e-01, ..., -8.9409e-01,
-1.2077e+00, -7.9800e-01]],
[[-3.1469e-01, -1.5287e+00, -8.0805e-01, ..., -7.9680e-01,
-1.5322e+00, -3.2423e-01],
[-2.2040e+00, -6.1334e+00, -5.0877e+00, ..., -2.5205e+00,
-3.5835e+00, -1.2115e+00],
[-2.1490e+00, -7.2517e+00, -7.5668e+00, ..., -2.7630e+00,
-3.1831e+00, -1.3344e+00],
...,
[-3.7176e+00, -7.2095e+00, -8.7949e+00, ..., -6.2087e+00,
-6.3224e+00, -2.9480e+00],
[-5.6294e+00, -6.9217e+00, -9.2691e+00, ..., -6.7300e+00,
-6.9527e+00, -2.7946e+00],
[ 4.8910e-01, -5.2678e-01, 1.0635e+00, ..., 2.7564e-01,
6.5724e-01, 1.0868e-01]],
[[-5.1965e-01, -1.3176e-01, 3.9893e-01, ..., 3.9200e-01,
9.0359e-02, 9.0122e-01],
[-3.7877e-01, -5.0041e-01, 4.2388e-01, ..., 3.9180e-01,
2.8501e-01, 5.1946e-01],
[-8.8014e-01, -7.2906e-01, -1.2197e+00, ..., 7.7248e-02,
1.9185e-01, 4.4021e-01],
...,
[-1.8309e+00, 3.6865e-01, -4.4322e+00, ..., -3.0774e-01,
2.3888e-03, -1.3290e-01],
[-1.5215e+00, -2.5469e-02, -5.2581e+00, ..., 5.3879e-02,
-1.5081e-01, -2.5260e-01],
[-6.2680e-01, 8.2509e-01, -1.4354e+00, ..., -5.7683e-02,
-1.0732e-01, -5.0544e-01]],
...,
[[ 1.0042e-01, -5.2433e-01, -3.5725e-01, ..., -2.9496e-02,
-2.0898e-01, -7.3595e-01],
[-8.5730e-01, -5.5899e-01, -3.6833e-01, ..., -2.5236e-01,
-9.9358e-01, -7.7216e-01],
[-4.9642e-01, 7.2623e-01, 8.5580e-01, ..., 8.2885e-02,
-2.0091e-01, -5.4604e-01],
...,
[-9.9604e-01, -6.4393e-01, -1.5702e+00, ..., -5.7950e-01,
-7.1910e-01, -1.2781e+00],
[-2.6218e+00, -1.6616e+00, -8.8689e-01, ..., -1.8027e+00,
-1.6103e+00, -1.6474e+00],
[-5.4651e-01, -1.3424e+00, 1.3380e+00, ..., -9.2082e-01,
-3.7878e-01, -5.3392e-01]],
[[-1.7783e-01, 3.9073e-02, 4.2462e-01, ..., 9.6230e-02,
5.5688e-01, 1.5833e-01],
[-4.0162e-01, 1.8703e-01, 6.1114e-02, ..., -3.3650e-02,
1.5594e-01, -1.0830e-01],
[-5.6134e-01, 5.0504e-02, -9.3467e-01, ..., 4.8447e-02,
5.7456e-02, -7.8670e-02],
...,
[-4.3809e-02, -2.0996e-01, -1.4584e+00, ..., 3.8220e-01,
5.1826e-01, -3.1593e-01],
[-7.1931e-01, -1.7130e+00, -2.0175e+00, ..., -8.2823e-02,
1.4663e-01, -5.2509e-01],
[ 7.5388e-01, 4.1947e-01, 2.4796e-02, ..., 6.2489e-01,
5.2399e-01, 3.2766e-01]],
[[-6.6030e-01, 4.0179e-01, 5.9224e-01, ..., 4.8819e-01,
3.8532e-01, 6.2357e-01],
[-8.4811e-01, 2.5271e-01, -4.3215e-01, ..., -2.4344e-01,
-1.0959e-01, 6.8777e-01],
[-4.8666e-01, -4.8122e-01, -1.6075e+00, ..., 8.8793e-02,
1.3906e-01, 6.3008e-01],
...,
[-1.3239e+00, -1.5228e+00, -1.0836e+00, ..., -1.9471e-01,
2.5145e-02, 1.4103e+00],
[-1.5466e+00, -1.4572e+00, 3.1761e-01, ..., -2.1370e-01,
1.3462e-01, 1.2172e+00],
[-7.3722e-01, -9.8386e-01, 4.8974e-01, ..., -1.2620e-01,
1.7134e-01, 1.3346e+00]]],
[[[-1.6261e-01, -2.5672e-01, 1.6309e-01, ..., -1.1595e-01,
4.9149e-01, 2.6121e-01],
[ 1.3151e-01, -2.0508e+00, -1.6454e+00, ..., -8.0498e-01,
-5.0482e-01, -2.7086e-01],
[-1.8460e-01, -2.7750e+00, -6.2582e-01, ..., -6.4469e-01,
-4.4305e-01, -7.4651e-01],
...,
[-1.7076e-01, -2.2810e+00, -4.9886e+00, ..., -9.1399e-01,
-1.2086e+00, -1.9469e+00],
[-4.5802e-01, -1.0922e+00, -2.3961e+00, ..., -7.8599e-02,
-2.7442e-01, -1.4030e+00],
[-2.9388e-01, -1.1502e+00, -1.7231e+00, ..., -9.7120e-01,
-1.3716e+00, -7.8842e-01]],
[[-1.9911e-01, -1.4097e+00, -6.6879e-01, ..., -1.0452e+00,
-1.6190e+00, -3.6783e-01],
[-3.9461e+00, -6.3445e+00, -5.3331e+00, ..., -3.0582e+00,
-3.9024e+00, -1.3327e+00],
[-3.7842e+00, -6.2544e+00, -5.4503e+00, ..., -2.8439e+00,
-3.1105e+00, -1.3049e+00],
...,
[-3.6340e+00, -6.6908e+00, -6.0914e+00, ..., -5.6638e+00,
-6.8131e+00, -2.8750e+00],
[-5.4890e+00, -7.5711e+00, -8.1285e+00, ..., -5.8120e+00,
-7.4796e+00, -2.7221e+00],
[ 5.1370e-01, -2.3214e-01, 2.3669e-01, ..., 3.2149e-01,
2.3836e-01, 9.6110e-02]],
[[-4.4837e-01, -8.1139e-02, 3.6465e-01, ..., 4.6516e-01,
2.0302e-01, 1.0639e+00],
[-4.0860e-01, -3.8013e-01, 1.0005e+00, ..., 3.7628e-01,
2.7025e-01, 6.1403e-01],
[-1.4954e+00, -2.2682e+00, -6.4459e-01, ..., -1.9809e-01,
-7.9461e-02, 4.5600e-01],
...,
[-1.7811e+00, 3.7884e-01, -5.2131e+00, ..., -4.9704e-01,
3.3117e-01, -1.5050e-01],
[-1.4426e+00, 4.3778e-01, -5.7408e+00, ..., -9.8328e-01,
2.0836e-01, -2.3859e-01],
[-5.9624e-01, 6.8338e-01, -3.3481e+00, ..., -1.3711e+00,
3.9541e-02, -4.8551e-01]],
...,
[[ 1.7758e-01, -5.1453e-01, -3.6098e-01, ..., -8.8954e-02,
-3.9743e-01, -8.1514e-01],
[-4.8846e-01, -3.0285e-01, 5.5662e-02, ..., -4.4886e-01,
-1.3878e+00, -9.2321e-01],
[ 6.8299e-01, 6.8507e-01, -9.1774e-01, ..., -1.0933e-01,
-2.8520e-01, -5.2885e-01],
...,
[-9.8865e-01, -5.5776e-01, -2.6811e+00, ..., -1.1016e+00,
-9.3695e-01, -1.2093e+00],
[-2.6152e+00, -2.1589e+00, -3.4420e+00, ..., -2.1640e+00,
-1.7853e+00, -1.5905e+00],
[-5.9239e-01, -8.2065e-01, 1.0615e+00, ..., -4.7440e-01,
-5.8991e-01, -5.0860e-01]],
[[-1.2526e-01, 6.9499e-02, 4.0417e-01, ..., 8.2372e-03,
6.7907e-01, 2.0418e-01],
[ 1.8435e-01, -7.3124e-02, -3.5756e-01, ..., -4.0841e-01,
1.2420e-01, -1.0215e-01],
[-3.4096e-01, -8.1241e-01, -6.7855e-01, ..., -2.3661e-01,
-5.5790e-02, -8.3807e-02],
...,
[-1.5945e-01, -7.0819e-01, -2.8908e+00, ..., 8.7708e-01,
3.8844e-01, -3.0391e-01],
[-8.1074e-01, -1.6447e+00, -2.7042e+00, ..., 9.5097e-01,
3.1707e-01, -5.1535e-01],
[ 7.2773e-01, -6.1036e-03, -1.0321e+00, ..., 1.1620e+00,
6.1033e-01, 3.1150e-01]],
[[-6.2980e-01, 4.1470e-01, 5.5802e-01, ..., 5.6494e-01,
3.4831e-01, 6.3138e-01],
[-9.1280e-01, 9.6054e-02, -2.8152e-01, ..., -3.4776e-01,
-9.3092e-02, 7.0691e-01],
[-1.4877e+00, -1.9272e+00, -3.7126e+00, ..., 5.4160e-03,
2.4987e-01, 7.2689e-01],
...,
[-1.3215e+00, -2.5106e+00, -1.6121e+00, ..., -2.1633e-01,
1.3485e-01, 1.4079e+00],
[-1.4761e+00, -1.8365e+00, 9.8407e-01, ..., 2.0985e-01,
3.1121e-01, 1.2102e+00],
[-6.6323e-01, -1.2363e+00, 3.4562e-01, ..., 6.1370e-01,
3.7428e-01, 1.2802e+00]]],
[[[-1.2532e-01, -5.0498e-01, -2.1204e-01, ..., -1.3061e-01,
2.9983e-01, -7.1496e-02],
[ 2.7147e-01, -1.2167e+00, -7.1481e-01, ..., -4.3756e-01,
-2.4780e-01, -1.7837e-01],
[ 1.2181e-01, -9.3898e-01, -3.9840e-01, ..., -1.9944e-01,
-1.8795e-01, -5.1818e-01],
...,
[-2.4917e-01, -1.6453e+00, -9.8729e-01, ..., -9.7023e-01,
-9.5944e-01, -2.0266e+00],
[-1.0694e+00, -1.6311e+00, -6.4730e-01, ..., -5.4088e-01,
-4.5671e-01, -1.7874e+00],
[-1.2781e+00, -2.0832e+00, -1.5974e+00, ..., -1.4346e+00,
-1.5544e+00, -9.6955e-01]],
[[-1.4418e+00, -3.0821e+00, -2.5402e+00, ..., -2.6420e+00,
-3.3587e+00, -1.3313e+00],
[-3.1420e+00, -5.0028e+00, -3.6023e+00, ..., -3.7305e+00,
-4.7061e+00, -2.0387e+00],
[-2.1182e+00, -3.1055e+00, -2.3006e+00, ..., -2.3907e+00,
-2.6778e+00, -1.2209e+00],
...,
[-3.7414e+00, -5.5070e+00, -5.1244e+00, ..., -5.5049e+00,
-6.1868e+00, -2.9306e+00],
[-6.2447e+00, -7.0971e+00, -6.2333e+00, ..., -6.1247e+00,
-7.4397e+00, -2.7348e+00],
[-2.5967e-03, -8.5387e-01, 2.5410e-01, ..., 4.0587e-01,
6.8505e-01, 1.8542e-01]],
[[-4.5849e-01, 1.9134e-01, 8.4094e-01, ..., 8.0765e-01,
5.6533e-01, 1.3552e+00],
[-3.5707e-01, -8.2292e-02, 5.6591e-01, ..., 4.6327e-01,
4.0593e-01, 2.9579e-01],
[-7.9322e-01, 4.4637e-02, 2.7159e-01, ..., 2.0021e-01,
3.3734e-01, 4.0289e-01],
...,
[-1.4175e+00, 1.3732e-01, 1.3155e-01, ..., 7.1610e-02,
5.4399e-02, 3.1626e-01],
[-1.2402e+00, 1.3661e-01, 2.6402e-01, ..., -4.9922e-02,
-1.1933e-02, 3.0838e-01],
[-1.1905e+00, -8.5017e-02, 8.6289e-02, ..., 3.9529e-03,
-8.0042e-02, -3.3031e-01]],
...,
[[ 8.7882e-01, 2.3106e-01, 2.2957e-01, ..., 2.9988e-01,
1.8394e-01, -8.8860e-01],
[-1.1689e+00, -8.4137e-01, -6.8358e-01, ..., -7.8457e-01,
-1.4773e+00, -1.0975e+00],
[-6.2636e-01, -2.8543e-01, -2.6590e-01, ..., -2.9821e-01,
-5.0766e-01, -6.0355e-01],
...,
[-6.1933e-01, -7.0831e-01, -2.2389e-01, ..., -3.3348e-01,
-5.8741e-01, -1.1954e+00],
[-3.0506e+00, -2.9071e+00, -2.2623e+00, ..., -1.8786e+00,
-2.0382e+00, -2.0527e+00],
[-5.0961e-01, -1.6083e+00, -1.0707e+00, ..., -8.5869e-01,
-9.6237e-01, -1.3728e+00]],
[[ 5.7643e-02, 2.9161e-01, 7.0100e-01, ..., 6.4341e-01,
9.7842e-01, 3.6546e-01],
[-2.6440e-02, -4.0081e-01, 1.7428e-01, ..., 4.2355e-01,
5.9652e-01, -1.5454e-01],
[-2.9534e-01, -6.0478e-01, 1.6511e-01, ..., 3.2705e-01,
3.2710e-01, 1.9405e-02],
...,
[-1.0607e-01, -9.5920e-01, 2.4270e-01, ..., 2.6043e-01,
1.5048e-01, -7.4589e-01],
[-5.1559e-01, -1.8084e+00, -1.1596e-01, ..., 5.7908e-02,
-1.3670e-02, -9.5496e-01],
[ 4.5900e-01, -9.6931e-02, 1.2691e-01, ..., 2.6828e-01,
3.1067e-01, 2.8753e-01]],
[[-9.8384e-01, -2.5004e-01, 1.4484e-01, ..., 1.1932e-01,
4.4122e-02, 7.3928e-01],
[-1.1441e+00, -2.1475e-01, -3.1475e-01, ..., -3.3515e-01,
-1.1045e-01, 1.1820e+00],
[-7.5280e-01, -2.0511e-01, -2.2677e-01, ..., -1.9206e-01,
-1.1099e-01, 6.8730e-01],
...,
[-1.1094e+00, -9.8460e-01, -7.5286e-01, ..., -3.8095e-01,
-2.7327e-02, 8.3228e-01],
[-1.8213e+00, -1.7903e+00, -1.0380e+00, ..., -8.0609e-01,
-4.2912e-01, 4.3713e-01],
[-9.7893e-01, -9.4189e-01, -2.1971e-01, ..., -3.1234e-01,
-1.3585e-01, 1.1244e+00]]]], device='cuda:0'),
tensor([[[[-2.1579e-01, -2.8077e-01, 1.9402e-01, ..., 7.8142e-02,
5.1358e-01, 1.5014e-01],
[-1.3557e-01, -1.5224e+00, -8.7669e-01, ..., -7.1272e-01,
-4.0062e-01, -3.1462e-01],
[-1.3215e-01, -1.3407e+00, -6.9071e-01, ..., -4.9712e-01,
-4.1073e-01, -6.8510e-01],
...,
[ 2.1980e-03, -1.7104e+00, -1.4541e+00, ..., -1.1500e+00,
-1.1221e+00, -1.9635e+00],
[-6.2058e-02, -1.2247e+00, -7.3789e-01, ..., -4.6470e-01,
-3.6408e-01, -1.4457e+00],
[-2.9691e-02, -1.0938e+00, -1.0530e+00, ..., -9.9345e-01,
-1.3045e+00, -8.6793e-01]],
[[-2.0476e-01, -1.4064e+00, -5.7100e-01, ..., -5.7605e-01,
-1.2942e+00, -2.2708e-01],
[-2.0018e+00, -3.9248e+00, -2.3949e+00, ..., -2.2562e+00,
-3.2729e+00, -1.1072e+00],
[-1.9003e+00, -3.3739e+00, -2.5196e+00, ..., -2.4898e+00,
-2.8742e+00, -1.1909e+00],
...,
[-3.6447e+00, -5.0238e+00, -5.3026e+00, ..., -5.7614e+00,
-6.0085e+00, -2.8760e+00],
[-5.1231e+00, -6.0545e+00, -5.5873e+00, ..., -6.1615e+00,
-7.0847e+00, -2.8748e+00],
[ 6.0846e-01, -2.2891e-01, 5.1798e-01, ..., 4.4205e-01,
6.8194e-01, 8.4011e-02]],
[[-5.2035e-01, -1.6144e-01, 3.4131e-01, ..., 3.3371e-01,
2.0156e-02, 8.7049e-01],
[-3.6787e-01, -4.6191e-01, 4.7611e-01, ..., 3.7331e-01,
2.5641e-01, 5.1275e-01],
[-8.5812e-01, -2.4090e-01, 1.6293e-01, ..., 2.8730e-02,
1.3574e-01, 4.0447e-01],
...,
[-1.8084e+00, -3.7988e-01, -6.1541e-02, ..., 2.7789e-02,
-8.7792e-02, -1.9400e-01],
[-1.5177e+00, -3.5561e-01, -1.6091e-01, ..., -6.7525e-02,
-1.8040e-01, -2.7492e-01],
[-5.4913e-01, -1.1745e-01, -5.8091e-03, ..., -8.3729e-02,
-6.4520e-02, -4.4720e-01]],
...,
[[ 7.5662e-02, -5.8831e-01, -3.9428e-01, ..., -7.5164e-02,
-2.2977e-01, -7.4018e-01],
[-8.3367e-01, -6.1250e-01, -5.5416e-01, ..., -2.5543e-01,
-9.5344e-01, -7.2852e-01],
[-5.3130e-01, -1.4280e-01, -2.7175e-02, ..., 6.8664e-02,
-1.8651e-01, -5.2236e-01],
...,
[-8.7350e-01, -9.5094e-01, -3.2701e-01, ..., -3.2019e-01,
-5.7647e-01, -1.2104e+00],
[-2.2391e+00, -1.8284e+00, -1.3428e+00, ..., -1.5091e+00,
-1.6567e+00, -1.6488e+00],
[-5.1706e-01, -9.8204e-01, -2.3107e-01, ..., -3.5911e-01,
-3.6104e-01, -5.3173e-01]],
[[-1.9561e-01, 2.9764e-02, 4.2261e-01, ..., 9.9693e-02,
5.2095e-01, 1.7553e-01],
[-4.3882e-01, -2.6808e-01, -8.1743e-04, ..., -6.3362e-02,
1.4129e-01, -2.5257e-02],
[-6.7345e-01, -5.8581e-01, -5.7016e-02, ..., 1.6869e-02,
1.8476e-02, -1.3733e-02],
...,
[ 1.8471e-01, -6.6746e-01, 2.7491e-01, ..., 4.1158e-01,
5.1508e-01, -3.2038e-01],
[-4.9928e-01, -1.7389e+00, -9.4530e-02, ..., 2.6622e-02,
4.4387e-02, -6.3664e-01],
[ 7.8398e-01, 9.7138e-02, 3.9106e-01, ..., 5.1621e-01,
4.5976e-01, 2.9931e-01]],
[[-6.8442e-01, 3.9980e-01, 5.4283e-01, ..., 4.6666e-01,
3.4784e-01, 6.0473e-01],
[-8.7299e-01, 2.6011e-01, -2.6191e-01, ..., -2.8532e-01,
-1.9508e-01, 6.4955e-01],
[-4.9225e-01, 3.2209e-01, 2.9391e-02, ..., 5.3954e-02,
3.6292e-02, 6.0621e-01],
...,
[-1.2474e+00, -4.4672e-01, -1.1438e-01, ..., -2.0591e-01,
-2.3624e-02, 1.3747e+00],
[-1.2516e+00, -6.1821e-01, -6.6928e-02, ..., -1.3360e-01,
6.5959e-02, 1.1502e+00],
[-5.6395e-01, -4.3836e-01, 1.7743e-01, ..., 7.3612e-02,
7.0071e-02, 1.2423e+00]]],
[[[-1.9662e-01, -2.6598e-01, 1.8459e-01, ..., 4.8426e-02,
5.0318e-01, 1.6957e-01],
[-9.3785e-02, -1.4843e+00, -8.6925e-01, ..., -7.2760e-01,
-4.3933e-01, -2.7554e-01],
[-2.4277e-01, -1.5169e+00, -1.7766e+00, ..., -5.0919e-01,
-3.9619e-01, -6.8043e-01],
...,
[-2.0696e-01, -1.6792e+00, -1.2789e+00, ..., -2.4661e+00,
-2.0896e+00, -1.5269e+00],
[-1.0749e+00, -1.6868e+00, -7.9373e-01, ..., -4.0543e-01,
-1.1434e+00, -1.9431e+00],
[-1.2746e+00, -2.1072e+00, -1.6004e+00, ..., -1.4401e+00,
-1.5495e+00, -9.6950e-01]],
[[-2.2870e-01, -1.4304e+00, -6.0833e-01, ..., -6.4581e-01,
-1.3723e+00, -2.4190e-01],
[-1.9511e+00, -3.9779e+00, -2.4019e+00, ..., -2.3416e+00,
-3.3420e+00, -1.1289e+00],
[-4.4724e+00, -1.0473e+01, -1.2091e+01, ..., -2.5433e+00,
-2.9575e+00, -1.2445e+00],
...,
[-3.7058e+00, -5.4490e+00, -5.1526e+00, ..., -8.0138e+00,
-9.0840e+00, -3.9160e+00],
[-6.1895e+00, -7.0027e+00, -6.1241e+00, ..., -7.3628e+00,
-8.7080e+00, -3.3929e+00],
[-2.8488e-02, -8.5776e-01, 2.6637e-01, ..., 4.0425e-01,
6.8084e-01, 1.8832e-01]],
[[-5.1269e-01, -1.6383e-01, 3.4356e-01, ..., 3.2870e-01,
5.1064e-02, 8.8260e-01],
[-2.9975e-01, -4.4233e-01, 4.9000e-01, ..., 3.5494e-01,
2.8318e-01, 5.0169e-01],
[-8.2466e-01, -1.0379e-01, 2.6293e-01, ..., 2.9979e-02,
1.3660e-01, 4.1596e-01],
...,
[-1.3816e+00, 5.4459e-02, 2.8927e-02, ..., -3.5604e+00,
-1.7119e+00, -2.6536e-01],
[-1.2284e+00, 1.4417e-01, -1.3892e-02, ..., -8.1265e-01,
5.5941e-02, 2.6643e-01],
[-1.1771e+00, -7.4550e-02, 1.0722e-01, ..., 1.1422e-03,
-3.6318e-02, -3.1369e-01]],
...,
[[ 7.8183e-02, -6.0944e-01, -3.8074e-01, ..., -7.5099e-02,
-2.5107e-01, -7.3826e-01],
[-8.0763e-01, -6.2235e-01, -5.5930e-01, ..., -2.5684e-01,
-9.9723e-01, -7.4880e-01],
[-6.2802e-01, -4.8085e-01, -6.1657e-01, ..., 8.5007e-02,
-1.9522e-01, -5.1371e-01],
...,
[-5.8645e-01, -6.6591e-01, -3.5855e-01, ..., -7.6077e-01,
-2.6550e+00, -2.8113e+00],
[-3.0524e+00, -2.9155e+00, -2.0367e+00, ..., -3.9706e+00,
-3.2321e+00, -2.4080e+00],
[-5.0726e-01, -1.6404e+00, -1.0928e+00, ..., -8.7234e-01,
-9.7924e-01, -1.3471e+00]],
[[-1.7121e-01, 3.7585e-02, 4.2003e-01, ..., 1.0407e-01,
5.4422e-01, 1.6193e-01],
[-3.8574e-01, -2.1769e-01, 1.1294e-02, ..., -4.1833e-02,
1.2626e-01, -5.2058e-02],
[ 1.6340e-01, 9.9692e-01, 4.4574e-01, ..., 4.4335e-02,
2.7126e-02, -4.0701e-02],
...,
[-7.3064e-02, -1.0271e+00, 2.6689e-02, ..., 1.0132e+00,
1.8541e+00, 1.2085e+00],
[-4.8522e-01, -1.8631e+00, -2.1894e-01, ..., -1.3305e+00,
-5.3772e-01, -7.4328e-01],
[ 4.6842e-01, -1.2275e-01, 1.1048e-01, ..., 2.4703e-01,
2.9744e-01, 3.1288e-01]],
[[-6.7209e-01, 4.0621e-01, 5.5847e-01, ..., 4.9395e-01,
3.9347e-01, 6.0068e-01],
[-8.4065e-01, 3.0807e-01, -2.3294e-01, ..., -2.8145e-01,
-1.7143e-01, 6.4938e-01],
[-1.1661e+00, -1.0196e+00, -1.1313e+00, ..., 5.0989e-02,
9.4229e-02, 6.0326e-01],
...,
[-1.1121e+00, -9.0232e-01, -4.7752e-01, ..., -6.7422e-01,
-1.2152e+00, 1.1562e+00],
[-1.8394e+00, -1.7944e+00, -9.8846e-01, ..., 1.9027e+00,
5.7216e-01, 1.0258e+00],
[-9.6073e-01, -9.8293e-01, -2.5123e-01, ..., -2.9029e-01,
-1.4337e-01, 1.1232e+00]]],
[[[-7.1540e-01, -1.4734e+00, -1.4495e+00, ..., -1.2731e+00,
-8.4132e-01, -1.6530e+00],
[-3.2258e-01, -2.4257e+00, -2.2694e+00, ..., -2.1073e+00,
-2.0094e+00, -1.9215e+00],
[-1.0428e-01, -1.8929e+00, -1.7562e+00, ..., -1.2027e+00,
-1.3232e+00, -2.1274e+00],
...,
[ 8.8304e-02, -5.2765e-01, -2.0675e-01, ..., -8.7019e-02,
-7.9658e-02, -4.5653e-01],
[-3.7420e-01, -4.5432e-01, 2.2171e-02, ..., 1.4043e-01,
2.7260e-01, -1.6852e-01],
[-6.1425e-01, -1.2655e+00, -1.0608e+00, ..., -1.1029e+00,
-1.1641e+00, -5.1772e-01]],
[[-2.6848e+00, -5.0997e+00, -4.5694e+00, ..., -4.7735e+00,
-6.0364e+00, -2.4220e+00],
[-5.2939e+00, -8.1647e+00, -7.5066e+00, ..., -9.0242e+00,
-9.9684e+00, -4.3696e+00],
[-4.1057e+00, -5.8916e+00, -5.7177e+00, ..., -6.6330e+00,
-7.6144e+00, -3.2675e+00],
...,
[-1.7177e+00, -2.1780e+00, -1.8013e+00, ..., -1.8378e+00,
-2.2565e+00, -9.5450e-01],
[-2.7877e+00, -2.9281e+00, -1.9285e+00, ..., -2.2399e+00,
-3.3247e+00, -9.9573e-01],
[ 3.4579e-01, -3.0027e-01, 2.9248e-01, ..., 3.0882e-01,
4.5728e-01, 1.8855e-01]],
[[-6.6800e-01, 4.5785e-01, 8.5950e-01, ..., 8.4060e-01,
6.2424e-01, 1.9331e+00],
[-9.6246e-01, 3.0709e-02, 4.0404e-01, ..., 6.6396e-01,
5.5979e-02, 1.3030e-01],
[-1.7788e+00, -2.3305e-01, -6.2884e-02, ..., 1.6851e-01,
2.7891e-01, -9.3155e-02],
...,
[-6.3775e-01, 8.5732e-02, 2.4689e-01, ..., 1.8276e-01,
2.1921e-01, 4.3088e-01],
[-4.5751e-01, 4.4847e-02, 2.1694e-01, ..., 1.5367e-01,
1.6720e-01, 5.4333e-01],
[-1.0582e-01, -2.3342e-01, 1.2838e-01, ..., -1.9311e-02,
5.6136e-02, 3.2559e-01]],
...,
[[ 1.2330e+00, -3.6696e-01, -6.0031e-02, ..., 2.3307e-01,
-7.0435e-02, -2.1223e+00],
[-7.4126e-01, -8.9082e-01, -5.0547e-01, ..., -3.6978e-01,
-1.3640e+00, -2.0768e+00],
[-7.1716e-01, -8.5198e-01, -2.8591e-01, ..., -6.9520e-01,
-6.7371e-01, -1.3945e+00],
...,
[-6.5497e-01, -5.1672e-01, -3.1921e-01, ..., -3.6058e-01,
-4.9699e-01, -5.8466e-01],
[-2.2692e+00, -1.7320e+00, -1.2658e+00, ..., -1.3379e+00,
-1.3455e+00, -1.0769e+00],
[-5.7496e-01, -8.7125e-01, -4.6813e-01, ..., -2.7210e-01,
-3.4531e-01, -3.7009e-01]],
[[ 4.0795e-02, -2.8647e-03, 4.3775e-01, ..., 4.7189e-01,
9.5019e-01, -6.9602e-03],
[ 2.3015e-01, -7.8000e-01, -3.9253e-02, ..., -3.6284e-01,
-1.1721e-01, -8.1076e-01],
[ 1.5764e-01, -6.5544e-01, 2.5349e-01, ..., 7.2952e-01,
2.2343e-01, -3.5575e-01],
...,
[-2.1662e-01, -4.6885e-01, 2.1862e-01, ..., 3.7016e-01,
3.3266e-01, 5.3218e-02],
[-5.2516e-01, -9.4197e-01, 3.8256e-02, ..., -1.0372e-02,
-5.5284e-03, -1.5447e-01],
[ 3.3600e-01, -6.8434e-04, -1.5838e-01, ..., -1.5422e-01,
-1.0043e-01, 1.8019e-01]],
[[-1.5440e+00, -8.0510e-01, -4.2245e-01, ..., -4.5331e-01,
-3.0358e-01, 8.6568e-01],
[-2.1219e+00, -7.8303e-01, -5.6727e-01, ..., -7.1396e-01,
-9.8344e-02, 1.6455e+00],
[-1.4550e+00, -5.5977e-01, -2.3020e-01, ..., -3.4178e-01,
3.1510e-02, 1.3941e+00],
...,
[-6.4352e-01, -2.1049e-01, -1.3025e-01, ..., -1.5281e-01,
2.2930e-02, 6.0987e-01],
[-1.1843e+00, -6.0895e-01, -3.4107e-01, ..., -3.5579e-01,
-1.7737e-01, 1.5674e-01],
[-3.7975e-01, -2.1427e-02, -1.3775e-01, ..., -2.4150e-01,
-1.9713e-01, 3.3159e-01]]],
[[[-3.0273e-01, -1.0277e+00, -8.1348e-01, ..., -3.9621e-02,
5.6555e-01, 4.9988e-01],
[ 2.0115e-01, -1.8971e+00, -1.2769e+00, ..., -6.6718e-01,
-2.8280e-01, 9.9997e-02],
[ 1.8518e-01, -1.4349e+00, -9.2175e-01, ..., -8.4709e-02,
2.8478e-02, -2.9415e-02],
...,
[ 2.6436e-01, -1.1024e+00, -1.0185e+00, ..., 8.5258e-01,
8.1134e-01, -7.2731e-01],
[ 7.6943e-02, -8.8633e-01, -4.2973e-01, ..., 2.0605e-01,
-9.4604e-03, -4.2772e-01],
[-1.4770e-01, -1.0851e+00, -9.4565e-01, ..., -1.2014e+00,
-1.2517e+00, -4.8112e-01]],
[[-2.0130e+00, -4.0595e+00, -3.7128e+00, ..., -5.5283e-01,
-1.1008e+00, -5.1051e-02],
[-3.9311e+00, -6.3579e+00, -5.5039e+00, ..., -2.3729e+00,
-2.8750e+00, -6.6034e-01],
[-3.1867e+00, -4.9994e+00, -5.0142e+00, ..., -2.0848e+00,
-1.8307e+00, -4.0698e-01],
...,
[-3.1853e+00, -4.5878e+00, -4.7413e+00, ..., -7.3594e-01,
-2.2583e+00, -2.5093e+00],
[-5.0687e+00, -6.0738e+00, -5.6718e+00, ..., -2.8223e+00,
-3.7301e+00, -1.0612e+00],
[ 7.3209e-01, -1.1552e-01, 5.1849e-01, ..., 3.7262e-01,
3.7680e-01, 5.9355e-02]],
[[-5.8819e-01, 3.7848e-01, 8.0957e-01, ..., 7.6881e-01,
2.8548e-01, 1.0360e+00],
[-9.0576e-01, 6.4774e-02, 6.3129e-01, ..., 7.1285e-01,
2.3776e-01, 4.7344e-01],
[-1.2828e+00, 4.7943e-03, 1.1396e-02, ..., 4.4632e-01,
4.5820e-01, 4.8384e-01],
...,
[-9.3964e-01, 2.2068e-01, -1.4523e-02, ..., 1.2988e-01,
-2.9592e-01, -7.0912e-01],
[-1.2789e+00, 1.7491e-01, -1.2757e-01, ..., -1.0693e-01,
-4.8013e-02, 6.3660e-01],
[-5.3622e-01, 3.1613e-01, 1.2607e-01, ..., -1.1164e-01,
7.3266e-02, 3.8945e-01]],
...,
[[ 6.3808e-01, 6.2994e-02, 2.3034e-01, ..., -8.8523e-02,
-3.3458e-01, -8.7030e-01],
[-8.5190e-01, -3.6830e-01, -9.9163e-02, ..., -6.4634e-01,
-1.2391e+00, -8.0525e-01],
[-5.0406e-01, -3.7770e-01, -3.8054e-02, ..., -3.2776e-01,
-2.8326e-01, -3.8476e-01],
...,
[-8.3200e-01, -8.9843e-01, -4.5336e-01, ..., -2.7518e+00,
-4.4228e+00, -3.6280e+00],
[-2.1372e+00, -1.7056e+00, -1.2326e+00, ..., -2.3175e+00,
-1.7903e+00, -1.0672e+00],
[-4.3094e-01, -9.9229e-01, -4.0977e-01, ..., -2.2036e-02,
-1.6313e-01, -3.9383e-01]],
[[ 9.5412e-03, -2.2957e-01, -6.2235e-03, ..., 2.9403e-01,
6.2154e-01, 2.6340e-01],
[ 1.1034e-01, -8.8296e-01, -5.5029e-02, ..., -4.6983e-02,
7.0889e-02, -5.1384e-02],
[-1.7720e-01, -9.6430e-01, 8.0271e-02, ..., 1.5116e-01,
1.0318e-02, 1.9118e-01],
...,
[ 9.5595e-02, -9.1719e-01, -1.3005e-01, ..., -6.9137e-01,
-1.0330e+00, -9.9043e-01],
[-5.1142e-01, -1.8985e+00, -2.0690e-01, ..., -6.3898e-01,
-1.1579e+00, -7.5131e-01],
[ 7.4258e-01, -1.3768e-01, 2.5448e-01, ..., -3.1842e-01,
-2.8632e-01, 3.0496e-01]],
[[-6.5992e-01, -1.3835e-01, 3.0634e-02, ..., 5.1475e-01,
3.2636e-01, 5.0311e-01],
[-1.0258e+00, -7.3967e-01, -9.3908e-01, ..., -1.5264e-01,
-1.9086e-01, 3.7430e-01],
[-6.2275e-01, -6.6201e-01, -5.8845e-01, ..., -4.3061e-01,
-3.1862e-01, 4.6544e-02],
...,
[-7.3113e-01, -7.5872e-01, -3.9565e-01, ..., -1.5209e-01,
8.5606e-01, 1.0094e+00],
[-8.5175e-01, -1.0793e+00, -6.2742e-01, ..., 5.6452e-01,
3.9404e-01, -2.7584e-02],
[-4.7379e-01, -8.8421e-01, -4.0403e-01, ..., -1.9072e-01,
-3.4995e-01, 1.0082e-01]]],
[[[ 3.2631e-02, -2.3561e-01, -3.8533e-02, ..., -9.5370e-01,
-4.9686e-01, -1.3849e+00],
[ 3.1202e-01, -9.2040e-01, -7.0265e-01, ..., -1.1243e+00,
-1.3519e+00, -1.4597e+00],
[ 2.5339e-01, -5.5687e-01, -4.8601e-01, ..., -1.0119e+00,
-1.4198e+00, -2.3197e+00],
...,
[-3.0343e-02, -6.3699e-01, -1.8751e-01, ..., -9.5465e-01,
-1.0331e+00, -1.9123e+00],
[-6.7908e-01, -7.2831e-01, -4.0336e-02, ..., -4.4353e-01,
-6.1451e-01, -1.6990e+00],
[-7.7170e-01, -1.3828e+00, -1.2067e+00, ..., -9.5514e-01,
-1.3965e+00, -6.4763e-01]],
[[-7.5427e-01, -2.1448e+00, -1.8781e+00, ..., -3.9513e+00,
-5.4677e+00, -1.8048e+00],
[-1.9773e+00, -3.6790e+00, -2.9602e+00, ..., -6.0159e+00,
-8.2925e+00, -3.5304e+00],
[-1.5471e+00, -2.5524e+00, -2.5048e+00, ..., -6.0404e+00,
-8.3193e+00, -3.1272e+00],
...,
[-1.8049e+00, -2.4589e+00, -2.5362e+00, ..., -4.9847e+00,
-6.3139e+00, -2.4670e+00],
[-3.8671e+00, -4.5083e+00, -3.7006e+00, ..., -6.1187e+00,
-7.5332e+00, -2.4193e+00],
[ 2.6074e-01, -3.4725e-01, 3.8468e-01, ..., 4.0665e-01,
4.5889e-01, 2.4736e-01]],
[[-2.1430e-01, 1.7611e-01, 7.3023e-01, ..., 6.6180e-01,
3.0521e-01, 1.6133e+00],
[ 5.0485e-02, -3.8147e-01, 5.5908e-01, ..., 5.1803e-01,
-1.2858e-01, 1.4269e-01],
[-1.1876e-01, -2.0821e-01, 2.2789e-01, ..., -1.6254e-01,
-7.2533e-04, -5.9938e-02],
...,
[-2.5180e-01, -1.8782e-01, 5.9957e-01, ..., 7.8050e-02,
-1.0325e-01, -1.8554e-01],
[-2.7532e-01, -1.9238e-01, 1.4264e-01, ..., -8.3647e-02,
-5.3744e-04, -2.0663e-01],
[-1.4259e-01, -5.1202e-01, 1.0133e-01, ..., 7.0969e-02,
2.2561e-01, -5.5949e-01]],
...,
[[ 6.0825e-01, -2.4546e-01, -6.4292e-03, ..., 4.5346e-01,
2.6697e-01, -1.3667e+00],
[-4.9469e-01, -5.3920e-01, -4.9784e-01, ..., -1.5654e-01,
-8.2395e-01, -1.4556e+00],
[-3.7572e-01, -2.1887e-01, -2.5169e-01, ..., 3.7830e-02,
-2.9551e-02, -1.1613e+00],
...,
[-4.4470e-01, -1.3723e-01, -3.9268e-01, ..., -1.2167e-02,
-5.8362e-02, -9.1637e-01],
[-2.3755e+00, -1.7356e+00, -1.2672e+00, ..., -1.3876e+00,
-1.3868e+00, -1.7193e+00],
[-4.9168e-01, -8.3434e-01, -3.9678e-01, ..., -3.3860e-01,
-5.1649e-01, -1.1878e+00]],
[[ 1.7034e-01, 2.6481e-01, 6.3530e-01, ..., -8.7993e-02,
4.3483e-01, -3.7238e-01],
[ 1.5336e-01, 2.5032e-01, 4.6918e-01, ..., 2.5296e-01,
1.2968e-01, -8.5305e-01],
[-1.1212e-01, 1.0004e-01, 2.9078e-01, ..., 5.1525e-01,
2.5146e-01, -1.1351e+00],
...,
[-2.1696e-01, -2.1282e-03, 4.2516e-01, ..., 1.8744e-01,
1.2690e-01, -4.8252e-01],
[-5.7840e-01, -7.0934e-01, -4.7253e-02, ..., -4.2267e-02,
-1.7307e-01, -8.4175e-01],
[ 4.9972e-01, 3.9763e-01, 6.8755e-02, ..., 3.8703e-01,
3.7934e-01, 3.2321e-01]],
[[-8.5787e-01, 2.3677e-01, 3.3660e-01, ..., -3.9240e-02,
-2.4212e-02, 6.3553e-01],
[-1.1137e+00, 1.8861e-01, -3.6098e-01, ..., -9.1110e-01,
-3.7368e-01, 9.3544e-01],
[-9.4197e-01, 3.5937e-01, -1.2109e-01, ..., -7.6366e-01,
-1.0170e+00, 1.6964e-01],
...,
[-9.6734e-01, 1.8189e-01, -1.8161e-01, ..., -3.7287e-01,
-5.2341e-01, 5.8473e-01],
[-1.7126e+00, -3.8918e-01, -3.4057e-01, ..., -6.5401e-01,
-6.3965e-01, 4.1275e-01],
[-7.5553e-01, -8.7811e-02, -3.0095e-01, ..., -4.2472e-01,
-3.9531e-01, 9.9286e-01]]]], device='cuda:0'),
tensor([[[[-2.0775e-01, -2.8579e-01, 1.8601e-01, ..., 2.1624e-02,
5.0021e-01, 1.5379e-01],
[-1.4786e-01, -1.5603e+00, -9.1559e-01, ..., -7.6974e-01,
-4.0223e-01, -3.0941e-01],
[-1.4444e-01, -1.3616e+00, -7.5276e-01, ..., -5.4063e-01,
-4.1494e-01, -7.0290e-01],
...,
[-2.6894e-01, -1.7203e+00, -1.1842e+00, ..., -1.0043e+00,
-9.7100e-01, -2.0353e+00],
[-1.1043e+00, -1.7243e+00, -7.7422e-01, ..., -5.2872e-01,
-4.6682e-01, -1.8148e+00],
[-1.2962e+00, -2.1573e+00, -1.5829e+00, ..., -1.4322e+00,
-1.5496e+00, -9.8293e-01]],
[[-1.7639e-01, -1.4060e+00, -5.9462e-01, ..., -7.3224e-01,
-1.4241e+00, -2.8225e-01],
[-2.0719e+00, -4.0878e+00, -2.4718e+00, ..., -2.3576e+00,
-3.3999e+00, -1.1095e+00],
[-1.9619e+00, -3.5455e+00, -2.5935e+00, ..., -2.6008e+00,
-2.9765e+00, -1.2090e+00],
...,
[-3.7140e+00, -5.4893e+00, -5.1778e+00, ..., -5.4851e+00,
-6.1871e+00, -2.9226e+00],
[-6.2047e+00, -7.0873e+00, -6.1488e+00, ..., -6.1480e+00,
-7.4590e+00, -2.7083e+00],
[-2.8360e-02, -8.8960e-01, 2.8452e-01, ..., 4.0735e-01,
6.9680e-01, 1.9696e-01]],
[[-5.2758e-01, -1.6739e-01, 3.9985e-01, ..., 3.7808e-01,
1.1041e-01, 8.9306e-01],
[-3.3564e-01, -4.9523e-01, 4.0636e-01, ..., 2.8894e-01,
2.4482e-01, 5.1522e-01],
[-8.7992e-01, -2.6380e-01, 1.4207e-01, ..., -2.5414e-02,
9.5726e-02, 3.9609e-01],
...,
[-1.3838e+00, 3.7195e-02, 1.8123e-01, ..., 2.8303e-02,
-1.3245e-02, 2.9071e-01],
[-1.2210e+00, 1.5413e-01, -3.4314e-02, ..., -2.7718e-02,
-8.6986e-03, 2.8450e-01],
[-1.1984e+00, -7.1084e-02, 5.7155e-02, ..., 6.8409e-03,
-1.7415e-02, -3.1414e-01]],
...,
[[ 7.3244e-02, -5.9767e-01, -3.7448e-01, ..., -8.0828e-02,
-2.0950e-01, -7.0158e-01],
[-8.5317e-01, -6.2555e-01, -5.8495e-01, ..., -2.5518e-01,
-9.9933e-01, -7.5719e-01],
[-4.8651e-01, -8.6875e-02, -1.1036e-02, ..., 5.8834e-02,
-2.1262e-01, -5.4839e-01],
...,
[-5.8935e-01, -5.8737e-01, -4.6311e-01, ..., -3.5748e-01,
-5.8803e-01, -1.1915e+00],
[-3.1289e+00, -2.8961e+00, -2.0093e+00, ..., -1.9422e+00,
-2.0521e+00, -2.0754e+00],
[-5.3263e-01, -1.6320e+00, -1.0758e+00, ..., -8.3112e-01,
-9.2540e-01, -1.3386e+00]],
[[-1.9487e-01, 4.0209e-02, 4.5418e-01, ..., 8.4083e-02,
5.3610e-01, 1.6555e-01],
[-4.5092e-01, -2.8880e-01, -2.4027e-02, ..., -8.1229e-02,
1.6434e-01, -3.8399e-02],
[-6.5738e-01, -5.5441e-01, -4.1679e-02, ..., 1.4712e-02,
1.7899e-02, -4.7229e-02],
...,
[-1.1463e-01, -1.0083e+00, 1.6641e-01, ..., 2.4019e-01,
1.2354e-01, -7.3145e-01],
[-5.1942e-01, -1.8422e+00, -1.9595e-01, ..., 1.7847e-02,
-6.9742e-02, -9.7299e-01],
[ 4.3782e-01, -1.3560e-01, 1.4899e-01, ..., 2.6421e-01,
3.0977e-01, 3.3771e-01]],
[[-6.6484e-01, 4.2671e-01, 5.4151e-01, ..., 4.7718e-01,
3.4738e-01, 5.9646e-01],
[-8.9731e-01, 3.0877e-01, -2.3559e-01, ..., -2.8850e-01,
-2.0808e-01, 6.4521e-01],
[-5.1248e-01, 3.0922e-01, 3.7945e-02, ..., 5.0426e-02,
7.7113e-02, 6.2762e-01],
...,
[-1.1096e+00, -9.6643e-01, -5.1393e-01, ..., -3.5020e-01,
-6.7518e-02, 8.2006e-01],
[-1.8191e+00, -1.8386e+00, -9.6580e-01, ..., -8.0114e-01,
-4.8470e-01, 4.1501e-01],
[-9.5754e-01, -9.8240e-01, -2.6187e-01, ..., -2.7810e-01,
-1.5789e-01, 1.1365e+00]]],
[[[-7.1076e-01, -1.4480e+00, -1.4491e+00, ..., -3.1150e+00,
-1.5750e+00, -1.6616e+00],
[-3.2452e-01, -2.3943e+00, -2.2471e+00, ..., -3.3686e+00,
-2.4715e+00, -1.9768e+00],
[-1.4786e-01, -1.8546e+00, -1.7023e+00, ..., -8.9893e-01,
-6.6060e-01, -2.1717e+00],
...,
[-8.1381e-02, -1.0404e+00, -8.5227e-01, ..., -1.0643e-01,
-5.3633e-01, -9.1998e-01],
[-1.2937e-01, -9.1998e-01, 8.8738e-02, ..., -2.1601e-01,
-1.5362e-01, -9.9729e-01],
[-1.3821e-01, -1.0350e+00, -4.0746e-01, ..., -1.1742e+00,
-1.1129e+00, -6.0989e-01]],
[[-2.7974e+00, -5.1884e+00, -4.6707e+00, ..., -7.6411e+00,
-8.0899e+00, -2.4167e+00],
[-5.4020e+00, -8.1996e+00, -7.5728e+00, ..., -1.1515e+01,
-1.2318e+01, -4.5399e+00],
[-4.0318e+00, -5.8639e+00, -5.8098e+00, ..., -6.5640e+00,
-9.5195e+00, -3.5683e+00],
...,
[-2.0167e+00, -3.8576e+00, -3.9517e+00, ..., -4.1106e+00,
-3.6632e+00, -1.5012e+00],
[-4.5545e+00, -5.6140e+00, -5.1974e+00, ..., -5.9692e+00,
-6.2696e+00, -2.3766e+00],
[ 7.0209e-01, -3.9984e-01, 3.0268e-01, ..., -3.9603e-03,
5.5846e-01, 2.9091e-02]],
[[-6.4068e-01, 5.2873e-01, 8.9415e-01, ..., 6.6621e-01,
8.3901e-01, 1.9142e+00],
[-9.8932e-01, 2.3898e-02, 4.3625e-01, ..., -4.7297e-01,
4.6909e-01, 1.4507e-01],
[-1.7661e+00, -2.5201e-01, -4.1555e-02, ..., -4.4058e+00,
4.3348e-01, -3.2801e-02],
...,
[-9.4236e-01, 1.3525e-02, -8.5870e-01, ..., 1.4884e-01,
1.1694e-01, 3.2898e-01],
[-9.9015e-01, 2.2648e-01, -5.4328e-01, ..., 3.5465e-01,
1.3258e-01, 3.5759e-01],
[-4.7410e-01, 2.3870e-01, 1.4776e-01, ..., 2.2401e-01,
1.2604e-01, 3.0997e-02]],
...,
[[ 1.3163e+00, -3.4415e-01, -4.8203e-02, ..., 5.0826e-01,
1.1400e-01, -2.1143e+00],
[-6.3663e-01, -8.7454e-01, -5.2691e-01, ..., -1.1533e+00,
-1.2605e+00, -2.0993e+00],
[-6.7211e-01, -8.8519e-01, -3.2972e-01, ..., -3.6418e+00,
-2.8055e+00, -1.4224e+00],
...,
[-4.3955e-01, -8.0363e-03, 1.8473e-01, ..., -7.1126e-01,
-2.0877e-01, -5.4700e-01],
[-2.0564e+00, -1.2684e+00, -5.7346e-01, ..., -2.0708e+00,
-1.1130e+00, -1.0953e+00],
[-2.9181e-01, -1.0428e+00, -2.8612e-01, ..., -9.5700e-01,
-2.6574e-01, -6.0956e-01]],
[[ 1.1786e-01, 3.4605e-02, 4.4566e-01, ..., -1.5031e+00,
-7.4990e-01, -2.6853e-02],
[ 3.3765e-01, -7.5765e-01, -4.7421e-02, ..., -1.1809e+00,
-1.5506e+00, -1.0802e+00],
[ 2.2773e-01, -5.5821e-01, 2.9691e-01, ..., -1.0705e+00,
-5.2976e-01, -4.4686e-01],
...,
[-3.9358e-01, -5.4306e-01, -1.9542e-01, ..., 5.2110e-01,
6.5851e-02, -1.0575e-01],
[-7.4740e-01, -1.6468e+00, -4.3483e-01, ..., -3.5086e-01,
-3.9243e-01, -6.7675e-01],
[ 5.2730e-01, 3.4471e-02, 2.3524e-01, ..., 1.6231e-01,
1.4675e-01, 3.6222e-01]],
[[-1.5333e+00, -7.6787e-01, -4.2346e-01, ..., -7.9172e-01,
-1.1541e-01, 8.7435e-01],
[-2.1573e+00, -7.4353e-01, -5.7013e-01, ..., -1.9798e+00,
-4.2485e-01, 1.6392e+00],
[-1.5525e+00, -5.3669e-01, -2.3750e-01, ..., -2.8810e+00,
-7.5378e-01, 1.3482e+00],
...,
[-4.5856e-01, -5.3605e-02, -2.4107e-02, ..., 3.4832e-01,
3.0466e-01, 7.1962e-01],
[-9.1458e-01, -5.5053e-01, -1.5357e-01, ..., 3.9773e-01,
1.1803e-01, 4.3897e-01],
[-4.4105e-01, -3.8121e-01, 1.1762e-01, ..., 3.9383e-01,
-2.2605e-01, 6.0404e-01]]],
[[[-2.1562e-01, -2.7556e-01, 2.0540e-01, ..., 1.2804e-02,
5.0136e-01, 1.5255e-01],
[-1.2277e-01, -1.5164e+00, -8.6965e-01, ..., -7.5155e-01,
-3.9809e-01, -3.2153e-01],
[-1.2272e-01, -1.3368e+00, -6.7793e-01, ..., -5.3900e-01,
-3.9093e-01, -6.8812e-01],
...,
[-1.2162e-01, -1.2580e+00, -1.1240e+00, ..., -1.3499e+00,
-9.3226e-01, -2.0454e+00],
[-7.4890e-01, -1.3102e+00, -6.0473e-01, ..., -4.2305e-01,
-4.8992e-01, -1.8104e+00],
[-9.3757e-01, -1.7769e+00, -1.4682e+00, ..., -1.5642e+00,
-1.7567e+00, -1.0636e+00]],
[[-1.9343e-01, -1.4093e+00, -5.4198e-01, ..., -6.6734e-01,
-1.3495e+00, -2.3535e-01],
[-2.0165e+00, -4.0140e+00, -2.4317e+00, ..., -2.4092e+00,
-3.3130e+00, -1.1170e+00],
[-1.8686e+00, -3.3764e+00, -2.5190e+00, ..., -2.7018e+00,
-2.9585e+00, -1.2178e+00],
...,
[-3.7199e+00, -5.2559e+00, -5.0841e+00, ..., -7.1854e+00,
-7.5971e+00, -2.9036e+00],
[-5.8639e+00, -6.6573e+00, -6.1545e+00, ..., -7.4567e+00,
-8.5238e+00, -2.7991e+00],
[ 8.5577e-02, -4.9088e-01, 3.1758e-01, ..., -3.3859e-01,
5.8168e-01, 1.7885e-01]],
[[-5.0609e-01, -1.6123e-01, 3.4967e-01, ..., 3.4219e-01,
4.3743e-02, 8.7823e-01],
[-3.4224e-01, -4.3984e-01, 4.8070e-01, ..., 4.1829e-01,
2.7760e-01, 5.0901e-01],
[-8.7660e-01, -2.3206e-01, 1.8238e-01, ..., 7.3629e-02,
1.5410e-01, 4.0298e-01],
...,
[-1.3500e+00, 3.5913e-01, -5.3969e-01, ..., -9.8785e-01,
3.2414e-01, 2.6554e-01],
[-1.2494e+00, 4.6479e-01, -4.8072e-02, ..., -9.4754e-01,
6.1750e-01, 2.9176e-01],
[-1.1171e+00, -3.0925e-02, 5.9872e-02, ..., -1.0266e+00,
3.0079e-01, -3.2136e-01]],
...,
[[ 8.0470e-02, -5.7525e-01, -3.6059e-01, ..., -7.4754e-02,
-2.5084e-01, -7.4820e-01],
[-8.4864e-01, -6.4312e-01, -5.8540e-01, ..., -2.7769e-01,
-9.8656e-01, -7.2887e-01],
[-5.1948e-01, -1.2309e-01, -1.5822e-02, ..., 1.4144e-03,
-1.9659e-01, -5.0640e-01],
...,
[-4.3162e-01, -6.5462e-01, 4.3032e-01, ..., -1.3102e+00,
-1.3209e+00, -1.2223e+00],
[-2.5526e+00, -2.9771e+00, -2.1715e+00, ..., -5.4623e+00,
-3.5532e+00, -2.1261e+00],
[-5.0629e-01, -1.5703e+00, -1.0201e+00, ..., -3.0282e+00,
-1.5019e+00, -1.3828e+00]],
[[-1.8270e-01, 4.1921e-02, 4.4781e-01, ..., 2.4100e-02,
5.3406e-01, 1.7909e-01],
[-4.2355e-01, -2.5213e-01, -6.0700e-04, ..., -1.0299e-01,
1.6361e-01, -6.7304e-02],
[-6.6499e-01, -5.7676e-01, -1.4613e-02, ..., 2.7588e-02,
1.5808e-02, -3.7856e-02],
...,
[ 9.8424e-02, -6.9983e-01, 1.2159e-01, ..., -1.5837e+00,
-1.2099e+00, -7.6671e-01],
[-2.6231e-01, -1.5314e+00, -1.2870e-01, ..., -6.9205e-01,
-8.3467e-01, -1.0069e+00],
[ 5.9334e-01, -6.8716e-02, 1.6598e-01, ..., 3.9053e-01,
2.6789e-01, 2.8222e-01]],
[[-6.7895e-01, 3.9440e-01, 5.5517e-01, ..., 4.7971e-01,
3.7905e-01, 6.0409e-01],
[-8.4571e-01, 2.8951e-01, -2.1856e-01, ..., -2.6515e-01,
-1.4397e-01, 6.4513e-01],
[-4.7581e-01, 3.2210e-01, 3.5528e-02, ..., 8.6704e-02,
8.2728e-02, 5.8783e-01],
...,
[-1.0718e+00, -9.4338e-01, -2.4552e-01, ..., -9.6995e-01,
-5.8716e-01, 7.9889e-01],
[-1.6712e+00, -1.5608e+00, -6.8449e-01, ..., -1.0994e+00,
-5.2090e-01, 3.7853e-01],
[-9.1761e-01, -7.9369e-01, -1.8057e-01, ..., -3.4244e-01,
1.5020e-01, 1.1212e+00]]],
[[[-2.2996e-01, -2.7530e-01, 1.9422e-01, ..., 5.8250e-02,
5.1352e-01, 1.7827e-01],
[-1.4066e-01, -1.5234e+00, -8.8889e-01, ..., -7.3475e-01,
-4.2397e-01, -2.9359e-01],
[-1.3681e-01, -1.3251e+00, -6.5818e-01, ..., -4.9137e-01,
-4.1191e-01, -6.7710e-01],
...,
[-2.6946e-01, -1.7406e+00, -1.2646e+00, ..., -9.4594e-01,
-9.5954e-01, -2.0397e+00],
[-1.1162e+00, -1.7438e+00, -7.9197e-01, ..., -5.2659e-01,
-4.8130e-01, -1.8117e+00],
[-1.2818e+00, -2.1616e+00, -1.6057e+00, ..., -1.3898e+00,
-1.5248e+00, -9.6972e-01]],
[[-2.0696e-01, -1.3931e+00, -5.8666e-01, ..., -5.7229e-01,
-1.3314e+00, -2.1800e-01],
[-2.0475e+00, -3.9682e+00, -2.3753e+00, ..., -2.3477e+00,
-3.3553e+00, -1.1150e+00],
[-1.8742e+00, -3.3673e+00, -2.4734e+00, ..., -2.5777e+00,
-2.9788e+00, -1.2445e+00],
...,
[-3.7060e+00, -5.4461e+00, -5.1321e+00, ..., -5.3889e+00,
-6.1344e+00, -2.9147e+00],
[-6.2306e+00, -7.0562e+00, -6.1225e+00, ..., -6.0816e+00,
-7.3727e+00, -2.7305e+00],
[-2.1366e-02, -8.6987e-01, 2.7295e-01, ..., 4.0350e-01,
7.0843e-01, 1.7323e-01]],
[[-5.1903e-01, -1.6092e-01, 3.3118e-01, ..., 3.5838e-01,
5.2328e-02, 8.8525e-01],
[-3.7613e-01, -4.4290e-01, 4.6649e-01, ..., 3.8659e-01,
2.8662e-01, 4.9257e-01],
[-8.6266e-01, -2.4012e-01, 1.9144e-01, ..., 3.4008e-02,
1.8928e-01, 4.0226e-01],
...,
[-1.3701e+00, -7.1839e-02, -4.8289e-02, ..., 7.7463e-02,
-1.6579e-02, 2.7536e-01],
[-1.1828e+00, 1.0987e-01, -2.3107e-02, ..., -9.4269e-04,
-8.1262e-03, 2.9723e-01],
[-1.1630e+00, -6.7266e-02, 8.9380e-02, ..., 1.4445e-02,
-4.4771e-02, -3.5904e-01]],
...,
[[ 7.5458e-02, -5.8561e-01, -3.9141e-01, ..., -7.3999e-02,
-2.4722e-01, -7.4739e-01],
[-8.4820e-01, -6.2705e-01, -5.3571e-01, ..., -2.7270e-01,
-9.9860e-01, -7.4152e-01],
[-5.1704e-01, -1.3788e-01, -3.8905e-02, ..., 9.7007e-02,
-2.1075e-01, -5.2192e-01],
...,
[-6.3882e-01, -6.0755e-01, -3.5165e-01, ..., -3.7346e-01,
-5.9051e-01, -1.1963e+00],
[-3.1103e+00, -2.8874e+00, -2.0228e+00, ..., -1.8734e+00,
-1.9945e+00, -2.0576e+00],
[-5.1434e-01, -1.6184e+00, -1.1059e+00, ..., -8.6886e-01,
-9.1985e-01, -1.3195e+00]],
[[-1.9637e-01, 3.3707e-02, 4.4059e-01, ..., 9.2418e-02,
5.4201e-01, 1.8617e-01],
[-4.4719e-01, -2.8847e-01, 2.0200e-03, ..., -6.7367e-02,
1.3071e-01, -4.9175e-02],
[-6.8452e-01, -5.8399e-01, -2.6832e-02, ..., 3.6048e-02,
1.1719e-02, -4.2563e-02],
...,
[-1.4000e-01, -9.9770e-01, 3.6897e-02, ..., 2.7500e-01,
1.6088e-01, -7.3527e-01],
[-5.4692e-01, -1.8418e+00, -1.9440e-01, ..., 1.9635e-02,
-1.9463e-02, -9.5598e-01],
[ 4.3278e-01, -1.4595e-01, 1.3807e-01, ..., 2.7663e-01,
3.2446e-01, 3.3695e-01]],
[[-6.7545e-01, 3.9189e-01, 5.5287e-01, ..., 4.8126e-01,
3.7583e-01, 5.9513e-01],
[-8.5926e-01, 2.6141e-01, -2.3587e-01, ..., -2.7970e-01,
-1.6725e-01, 6.3974e-01],
[-4.8445e-01, 3.4003e-01, 4.7758e-02, ..., 8.0957e-02,
1.0516e-01, 5.7930e-01],
...,
[-1.1630e+00, -9.3477e-01, -4.5325e-01, ..., -4.0670e-01,
-4.8763e-02, 8.2397e-01],
[-1.8585e+00, -1.8214e+00, -9.6429e-01, ..., -7.7879e-01,
-4.4189e-01, 4.2835e-01],
[-9.5710e-01, -9.6863e-01, -2.4326e-01, ..., -2.7327e-01,
-1.4371e-01, 1.1238e+00]]],
[[[-6.9380e-01, -1.4450e+00, -1.4863e+00, ..., -1.2330e+00,
-8.7777e-01, -1.6553e+00],
[-2.8945e-01, -2.4144e+00, -2.2780e+00, ..., -1.6191e+00,
-1.8770e+00, -1.8867e+00],
[-1.1405e-01, -1.8804e+00, -1.7272e+00, ..., -1.1579e+00,
-1.4494e+00, -2.0932e+00],
...,
[ 7.3928e-02, -5.1471e-01, -1.8283e-01, ..., -8.9023e-02,
-1.6080e-02, -4.1181e-01],
[-4.1876e-01, -4.6267e-01, 4.1957e-02, ..., 1.7273e-01,
3.0877e-01, -1.3988e-01],
[-6.7368e-01, -1.3316e+00, -1.0853e+00, ..., -1.0867e+00,
-1.1500e+00, -4.8619e-01]],
[[-2.7035e+00, -5.1991e+00, -4.7841e+00, ..., -5.0819e+00,
-6.0276e+00, -2.4273e+00],
[-5.2642e+00, -8.0803e+00, -7.5480e+00, ..., -8.1470e+00,
-9.1665e+00, -4.3342e+00],
[-4.0156e+00, -5.7212e+00, -5.6934e+00, ..., -7.1212e+00,
-6.4633e+00, -3.2562e+00],
...,
[-1.6972e+00, -2.1101e+00, -1.7632e+00, ..., -1.7994e+00,
-2.2187e+00, -8.9864e-01],
[-2.7948e+00, -2.9905e+00, -2.0008e+00, ..., -2.2057e+00,
-3.3093e+00, -9.8706e-01],
[ 3.1338e-01, -2.9120e-01, 3.1663e-01, ..., 2.7076e-01,
4.3228e-01, 2.0239e-01]],
[[-6.9792e-01, 4.4143e-01, 8.5426e-01, ..., 9.0960e-01,
6.6911e-01, 1.8962e+00],
[-1.0082e+00, -2.2916e-02, 4.0283e-01, ..., 6.2184e-01,
1.8395e-02, 1.0679e-01],
[-1.7467e+00, -2.9906e-01, -8.1715e-02, ..., 1.0218e-01,
7.5635e-03, -9.4554e-02],
...,
[-6.0213e-01, 8.4548e-02, 2.6704e-01, ..., 1.6832e-01,
2.4991e-01, 4.2646e-01],
[-4.3346e-01, 7.6705e-02, 2.0183e-01, ..., 1.6293e-01,
2.2361e-01, 5.7255e-01],
[-1.6092e-01, -2.9376e-01, 1.4344e-01, ..., -1.6174e-02,
6.6175e-02, 3.4111e-01]],
...,
[[ 1.2359e+00, -3.2833e-01, -6.5651e-02, ..., 3.0519e-01,
-8.1421e-02, -2.0903e+00],
[-7.0452e-01, -8.4059e-01, -4.6031e-01, ..., -5.5203e-01,
-1.4742e+00, -2.0815e+00],
[-7.1171e-01, -9.0814e-01, -3.0616e-01, ..., -6.0678e-01,
-6.1138e-01, -1.4028e+00],
...,
[-6.3299e-01, -5.0499e-01, -3.1099e-01, ..., -3.1214e-01,
-4.8239e-01, -5.4464e-01],
[-2.2807e+00, -1.8162e+00, -1.3155e+00, ..., -1.3079e+00,
-1.3641e+00, -1.0882e+00],
[-5.9919e-01, -8.3832e-01, -4.0271e-01, ..., -2.6497e-01,
-3.7183e-01, -3.9244e-01]],
[[ 4.3358e-02, 3.5430e-02, 4.1544e-01, ..., 3.9124e-01,
8.7607e-01, -2.7035e-02],
[ 1.9703e-01, -7.5556e-01, -3.3363e-02, ..., 4.4116e-01,
3.5334e-01, -8.3919e-01],
[ 8.1116e-02, -5.8614e-01, 3.1616e-01, ..., 6.2451e-01,
5.2684e-01, -3.9222e-01],
...,
[-2.0677e-01, -4.3836e-01, 2.5109e-01, ..., 3.4311e-01,
3.2714e-01, 7.1819e-02],
[-5.3158e-01, -8.7218e-01, 8.7588e-02, ..., 7.7687e-03,
-2.0783e-02, -1.4320e-01],
[ 3.1899e-01, -7.0741e-03, -1.5805e-01, ..., -1.7617e-01,
-1.1888e-01, 1.8387e-01]],
[[-1.5507e+00, -7.8416e-01, -4.0379e-01, ..., -4.2089e-01,
-2.6287e-01, 8.6286e-01],
[-2.1854e+00, -7.7229e-01, -5.5252e-01, ..., -7.9374e-01,
-1.7485e-01, 1.6590e+00],
[-1.5723e+00, -6.1481e-01, -2.2968e-01, ..., -3.7342e-01,
-1.3488e-01, 1.3917e+00],
...,
[-6.7761e-01, -2.2648e-01, -1.1335e-01, ..., -1.3926e-01,
4.8792e-04, 6.2671e-01],
[-1.2037e+00, -6.0073e-01, -3.2830e-01, ..., -3.4326e-01,
-2.1163e-01, 1.5505e-01],
[-3.8371e-01, -4.5969e-02, -1.6584e-01, ..., -2.6263e-01,
-2.3048e-01, 3.2738e-01]]]], device='cuda:0')],
7: [tensor([[[[ 6.5244e-01, 1.0337e+00, 7.8490e-01, ..., 3.8490e-01,
3.6473e-01, 1.8225e-01],
[-2.9436e-01, -1.5815e+00, -6.9339e-01, ..., 3.5254e-01,
3.8386e-01, -1.1195e+00],
[ 4.9726e-02, -8.3528e-01, -2.7141e-01, ..., -4.4964e-01,
9.5015e-01, 6.4806e-01],
...,
[ 1.3266e-01, -9.5129e-01, -8.1797e-01, ..., -9.9611e-02,
9.3283e-01, 5.1691e-01],
[-4.6874e-01, -1.7431e+00, -6.7924e-01, ..., 2.3004e-02,
1.3473e+00, -1.0076e-01],
[ 3.7870e-01, -5.6290e-01, -8.1998e-01, ..., -2.5071e-01,
3.5808e-01, 6.1060e-01]],
[[-3.3626e+00, -1.4456e+00, -3.2710e+00, ..., -1.7619e+00,
-2.2646e+00, -1.1142e+00],
[-4.4647e+00, -1.6229e+00, -2.5381e+00, ..., -4.2724e-01,
-1.2151e+00, -3.1036e-01],
[-4.5090e+00, -2.6029e+00, -3.0872e+00, ..., -1.4741e+00,
-1.4344e+00, -7.6109e-01],
...,
[-2.7366e+00, -1.7908e+00, -2.7808e+00, ..., -7.3538e-01,
1.9061e-01, 4.9494e-01],
[-3.3692e+00, -1.3271e+00, -2.2154e+00, ..., -4.0916e-01,
-3.3205e-01, 2.5408e-01],
[-1.1634e+00, -9.9744e-01, -1.7685e+00, ..., -3.7273e-01,
-6.8524e-01, -5.4943e-01]],
[[ 2.1328e+00, 3.0413e+00, 2.5152e+00, ..., 2.8815e+00,
1.4780e+00, 2.9038e+00],
[ 3.8395e-01, 1.2147e+00, 1.7227e+00, ..., 1.9650e+00,
8.8519e-02, 2.5291e+00],
[ 8.2278e-01, 1.5663e+00, 1.1588e+00, ..., 1.1993e+00,
6.7729e-01, 1.8665e+00],
...,
[ 9.8486e-01, 1.1761e+00, 4.5012e-01, ..., 1.0530e+00,
1.5699e-01, 1.4513e+00],
[ 5.7358e-01, 1.6871e+00, 8.4240e-01, ..., 1.1594e+00,
6.1606e-01, 1.5916e+00],
[ 1.4082e+00, 1.3732e+00, 1.4971e+00, ..., 1.4479e+00,
1.0875e+00, 2.0607e+00]],
...,
[[-1.7774e+00, -4.8704e+00, -4.3630e+00, ..., -3.4013e+00,
-2.5324e+00, -3.9246e+00],
[-2.1803e+00, -4.4147e+00, -4.5927e+00, ..., -5.5028e+00,
-3.9459e+00, -4.4519e+00],
[-9.0773e-01, -3.5381e+00, -3.8600e+00, ..., -2.7218e+00,
-2.3949e+00, -1.6059e+00],
...,
[-1.4470e+00, -3.6590e+00, -4.2848e+00, ..., -3.1634e+00,
-2.7399e+00, -1.5562e+00],
[-1.2508e+00, -4.0779e+00, -5.1658e+00, ..., -5.3101e+00,
-3.7244e+00, -3.7298e+00],
[-1.3770e+00, -4.4730e+00, -3.9998e+00, ..., -3.5514e+00,
-2.1796e+00, -2.7941e+00]],
[[ 7.4177e-01, 7.3674e-01, -1.6052e-01, ..., 8.9535e-01,
1.0569e+00, 1.2320e+00],
[ 1.3192e+00, 1.1641e+00, 5.6396e-01, ..., 1.6182e+00,
2.3836e+00, 2.5721e+00],
[ 7.7195e-01, 5.7637e-01, 1.2456e-01, ..., 1.0693e+00,
1.6668e+00, 1.9190e+00],
...,
[ 8.7707e-01, 1.0324e-02, -5.9606e-02, ..., 1.0894e+00,
1.9016e+00, 1.6782e+00],
[ 1.5479e+00, 1.3828e+00, 1.1087e+00, ..., 1.3132e+00,
1.5143e+00, 2.4612e+00],
[ 6.8106e-01, 7.9381e-01, 9.4387e-01, ..., 1.0810e+00,
1.3636e+00, 1.3764e+00]],
[[-3.0347e-01, -2.4584e-01, -1.0399e+00, ..., -2.1649e-01,
-1.9107e-01, 8.4133e-01],
[-1.3340e+00, -1.6275e+00, -2.2895e+00, ..., 1.5088e-01,
-5.5585e-02, 3.7891e-01],
[-1.7557e+00, -9.9734e-01, -1.3426e+00, ..., 2.2332e-01,
1.5688e-01, 2.2352e-01],
...,
[-1.7545e+00, -1.3871e+00, -1.8188e+00, ..., -2.5067e-01,
-1.6184e-01, -3.1588e-01],
[-1.5167e+00, -1.4532e+00, -1.5222e+00, ..., -2.1091e-01,
-2.6382e-01, 3.1199e-01],
[-1.5976e-01, -7.4476e-02, -6.0418e-01, ..., 2.3054e-01,
1.7393e-01, 3.1196e-01]]],
[[[ 1.0791e+00, 1.4461e+00, 1.1870e+00, ..., 2.5676e-01,
3.5329e-01, 2.3298e-01],
[-9.4623e-01, -1.4356e+00, -9.1655e-01, ..., 3.6740e-01,
2.9798e-01, -1.1927e+00],
[-4.7981e-01, -5.8096e-01, -4.8301e-01, ..., -3.5748e-01,
9.9306e-01, 6.0265e-01],
...,
[-1.8384e-01, -3.4513e-01, 5.2061e-02, ..., 3.6962e-01,
9.5365e-01, 4.5139e-01],
[-4.2791e-01, -7.6499e-01, -1.5441e-01, ..., 2.1738e-01,
1.3339e+00, -1.7822e-01],
[-1.4137e-01, -1.6778e-01, -7.1438e-01, ..., -2.0384e-01,
3.3627e-01, 5.9407e-01]],
[[-3.2077e+00, -1.9635e+00, -2.1841e+00, ..., -1.8092e+00,
-2.2850e+00, -1.1189e+00],
[-2.7930e+00, -3.2632e+00, -1.7403e+00, ..., -4.3129e-01,
-1.3306e+00, -3.2854e-01],
[-1.9247e+00, -2.2518e+00, -1.4845e+00, ..., -1.2869e+00,
-1.4798e+00, -7.3734e-01],
...,
[-5.9298e-03, -8.5881e-01, -3.4526e-01, ..., -1.1801e+00,
-3.9392e-01, 5.4242e-01],
[-1.3074e+00, -8.3887e-01, -3.8001e-01, ..., -5.7663e-01,
-4.6281e-01, 3.0428e-01],
[-4.6469e-01, -9.4798e-01, -7.8129e-01, ..., -3.1412e-01,
-7.4681e-01, -5.5956e-01]],
[[ 2.3505e+00, 2.3299e+00, 1.7150e+00, ..., 2.9436e+00,
1.4215e+00, 2.9375e+00],
[ 1.5382e-01, 2.2253e-01, 1.3335e+00, ..., 1.9205e+00,
-1.2522e-02, 2.5326e+00],
[-1.3206e-01, 3.1964e-01, 1.0561e+00, ..., 1.2653e+00,
6.1341e-01, 1.8416e+00],
...,
[-5.8968e-01, -4.1076e-02, 2.2886e-01, ..., -1.2381e+00,
-1.5184e-01, 1.4147e+00],
[-1.0072e+00, 5.5622e-01, 6.1329e-01, ..., 1.1797e+00,
7.0286e-01, 1.5044e+00],
[ 6.7992e-01, 5.1618e-01, 1.1239e+00, ..., 1.4543e+00,
1.0669e+00, 1.9917e+00]],
...,
[[-2.8870e+00, -8.4471e+00, -6.6205e+00, ..., -3.5528e+00,
-2.4588e+00, -4.0116e+00],
[-3.7358e+00, -1.0399e+01, -6.4146e+00, ..., -5.8660e+00,
-3.8588e+00, -4.3844e+00],
[-2.4429e+00, -8.2297e+00, -4.9681e+00, ..., -3.1353e+00,
-2.3303e+00, -1.5955e+00],
...,
[-3.6786e+00, -9.0301e+00, -5.1895e+00, ..., -3.6501e+00,
-2.0269e+00, -1.7995e+00],
[-4.0612e+00, -9.3774e+00, -5.1408e+00, ..., -4.5975e+00,
-3.8965e+00, -4.0403e+00],
[-3.1961e+00, -7.6495e+00, -3.0567e+00, ..., -4.0756e+00,
-2.4734e+00, -2.8924e+00]],
[[-4.3396e-01, 7.3663e-01, 9.2848e-01, ..., 1.3342e+00,
1.0783e+00, 1.2418e+00],
[ 8.1543e-01, 7.3161e-01, 1.3390e+00, ..., 1.9944e+00,
2.3465e+00, 2.5931e+00],
[ 7.3541e-01, 4.2950e-01, 1.4387e+00, ..., 1.5680e+00,
1.6191e+00, 1.9821e+00],
...,
[ 1.1918e+00, 3.9737e-01, 1.2886e+00, ..., 6.7311e-01,
1.7944e+00, 1.6619e+00],
[ 1.7010e+00, 1.5438e+00, 1.6183e+00, ..., 1.4618e+00,
1.4384e+00, 2.4712e+00],
[ 7.6679e-01, 6.5596e-01, 1.2256e+00, ..., 1.2554e+00,
1.3070e+00, 1.3563e+00]],
[[-1.9044e-01, -4.0084e-01, -3.4405e-01, ..., -1.4919e-01,
-1.4878e-01, 8.8958e-01],
[-6.6701e-01, -2.5781e+00, -1.1697e+00, ..., 2.1158e-01,
-1.0468e-01, 3.8296e-01],
[-1.5968e+00, -1.6962e+00, -7.2062e-01, ..., 3.7131e-01,
1.5181e-01, 2.6148e-01],
...,
[-1.6761e+00, -2.5299e+00, -5.5579e-01, ..., -1.4465e+00,
-8.5480e-01, -3.5370e-01],
[-1.4071e+00, -2.4987e+00, -7.2877e-01, ..., -4.7403e-01,
-2.2175e-01, 3.5275e-01],
[-4.3572e-01, -1.0881e+00, -5.7501e-01, ..., 2.6358e-01,
1.9800e-01, 3.3445e-01]]],
[[[ 6.8672e-01, 1.0296e+00, 6.6042e-01, ..., 1.0127e+00,
9.3394e-01, 3.4222e-01],
[-3.4772e-01, -1.5072e+00, -7.2936e-01, ..., -3.2051e-02,
1.2012e-02, -6.2263e-01],
[ 1.2304e-01, -8.5613e-01, -5.2565e-01, ..., -6.0317e-01,
-2.9658e-01, -5.0785e-01],
...,
[-8.1835e-01, 3.7812e-01, -7.3131e-01, ..., 1.1471e+00,
2.4233e+00, 4.3980e-01],
[-7.4034e-01, -7.7614e-01, 1.3280e+00, ..., -2.6656e+00,
3.8836e-02, -4.8612e-01],
[ 1.3786e-01, -9.9471e-01, 3.7560e-01, ..., 8.5601e-02,
-1.2334e-01, 5.9964e-01]],
[[-3.2511e+00, -1.5752e+00, -3.0412e+00, ..., -1.4949e+00,
-2.9350e+00, -1.0226e+00],
[-4.4961e+00, -1.8723e+00, -2.3265e+00, ..., -2.2107e+00,
-3.0988e+00, -2.6222e-01],
[-4.4163e+00, -2.5544e+00, -3.0382e+00, ..., -5.4136e+00,
-3.6294e+00, -1.2711e+00],
...,
[-4.4500e+00, -2.5212e+00, 2.1611e+00, ..., -1.6418e+00,
-2.3946e+00, -4.4872e-01],
[-4.0872e+00, -2.9130e+00, -3.0150e+00, ..., -3.9193e+00,
-2.4188e+00, -8.4906e-01],
[-1.4686e+00, -1.6059e+00, -2.0227e+00, ..., -1.0066e+00,
-6.1834e-01, -2.6998e-01]],
[[ 1.8354e+00, 2.8963e+00, 2.3749e+00, ..., 2.3360e+00,
8.4519e-01, 2.0466e+00],
[ 2.6926e-01, 1.2077e+00, 1.5587e+00, ..., 1.2548e+00,
2.7132e-02, 2.3287e+00],
[ 7.8717e-01, 1.4980e+00, 8.4976e-01, ..., 1.1403e+00,
7.5292e-01, 2.2511e+00],
...,
[ 2.8263e+00, 4.4612e-01, -6.2713e+00, ..., -3.3591e+00,
-6.3362e+00, -3.6509e+00],
[ 1.6376e+00, 3.3367e+00, -1.8262e+00, ..., -3.0069e+00,
-1.9911e+00, -4.1999e-01],
[ 1.6084e+00, 2.9137e+00, 2.4804e+00, ..., 1.6530e+00,
1.3779e+00, 1.5699e+00]],
...,
[[-1.5509e+00, -4.7322e+00, -4.0396e+00, ..., -4.5746e+00,
-3.2501e+00, -3.9367e+00],
[-1.9717e+00, -4.2401e+00, -4.4724e+00, ..., -5.2066e+00,
-3.8717e+00, -4.3549e+00],
[-1.0876e+00, -3.5839e+00, -3.9717e+00, ..., -6.9632e+00,
-4.2575e+00, -2.9259e+00],
...,
[-1.5393e+00, -5.4288e+00, -4.8208e+00, ..., -1.0600e+01,
-6.9517e+00, -7.1763e+00],
[-1.8770e+00, -5.6575e+00, -5.0294e+00, ..., -7.9799e+00,
-5.8245e+00, -5.7772e+00],
[-1.6880e+00, -4.7595e+00, -4.1796e+00, ..., -5.8950e+00,
-4.5675e+00, -5.4325e+00]],
[[ 6.9108e-01, 5.4610e-01, -1.5049e-01, ..., 1.3295e+00,
8.5130e-01, 1.6788e+00],
[ 1.1873e+00, 1.0697e+00, 5.6709e-01, ..., 2.3438e+00,
1.9119e+00, 2.9205e+00],
[ 5.6301e-01, 5.8202e-01, 5.8347e-02, ..., 2.3386e-01,
-5.1171e-01, 1.0649e+00],
...,
[-1.1124e-01, 1.0146e+00, 3.3396e+00, ..., -5.1097e-01,
1.1634e+00, 1.6080e+00],
[ 1.0364e+00, -4.7427e-02, 2.0822e+00, ..., 1.1421e+00,
-3.3113e-01, 2.2624e+00],
[ 6.6338e-01, 4.2181e-01, 1.0646e+00, ..., 2.3209e+00,
1.7580e+00, 1.5957e+00]],
[[-3.0677e-01, -1.8447e-01, -9.1103e-01, ..., 5.8432e-01,
2.3149e-01, 1.1261e+00],
[-1.4115e+00, -1.4263e+00, -2.1000e+00, ..., -4.0606e-01,
7.2608e-02, 5.3057e-01],
[-1.7146e+00, -7.8239e-01, -1.0747e+00, ..., -2.4545e-01,
-9.1589e-03, -1.3258e+00],
...,
[-2.0126e+00, -1.3663e+00, -8.2054e-01, ..., -2.7027e+00,
-2.6781e-01, -1.1422e+00],
[-1.4725e+00, -4.3572e-01, -1.2042e+00, ..., -3.2289e+00,
-2.0511e+00, -9.4009e-01],
[-1.1958e-01, -3.1268e-01, -7.0932e-01, ..., -1.3839e+00,
-7.8221e-01, 4.9547e-01]]],
[[[ 9.9411e-01, 1.3402e+00, 1.1000e+00, ..., -3.1253e+00,
1.1840e+00, 4.0289e-01],
[-7.3712e-01, -1.9986e+00, -7.5046e-01, ..., -1.6436e+00,
4.3458e-01, -1.3381e+00],
[-3.8937e-01, -1.0836e+00, -2.4151e-01, ..., -2.0759e+00,
-1.7471e+00, 9.6480e-01],
...,
[ 1.8569e-01, 1.6497e+00, -1.1369e+00, ..., 1.5251e+00,
-5.6312e-01, -1.3288e+00],
[-2.2395e-01, -7.5882e-01, 1.2318e+00, ..., 2.5275e+00,
1.9567e+00, -2.4020e-01],
[ 9.3580e-01, -3.6262e-01, -5.9210e-02, ..., -7.8869e-01,
-1.2115e+00, -3.0451e-01]],
[[-3.0017e+00, -1.6318e-01, -1.5379e+00, ..., -3.0109e+00,
-4.8884e+00, -1.2333e+00],
[-3.8168e+00, -2.3282e+00, -2.4419e+00, ..., -8.9243e+00,
-9.4200e+00, -1.5480e+00],
[-2.6688e+00, -2.1785e+00, -3.2280e+00, ..., -1.2519e+01,
-6.0821e+00, -7.5868e-01],
...,
[-1.3147e+00, -2.8902e+00, -1.2833e+00, ..., -3.0816e+00,
-5.3434e+00, -9.2279e-01],
[-2.7176e+00, -1.8060e-03, 1.5009e-01, ..., -2.8200e+00,
-5.9129e+00, -2.7225e+00],
[-1.8534e+00, -1.6461e+00, -1.0711e+00, ..., -7.0873e+00,
-5.9211e+00, -2.6495e+00]],
[[ 2.6404e+00, 2.8696e+00, 2.4711e+00, ..., 8.5560e-02,
3.7085e-01, 1.4219e+00],
[ 9.2547e-01, 5.4632e-01, 2.3908e+00, ..., -3.7397e+00,
-2.0553e+00, -9.7566e-03],
[ 1.2701e+00, 1.1104e+00, 2.1028e+00, ..., -2.8524e+00,
1.5533e+00, 2.1402e+00],
...,
[ 1.5895e+00, 8.6339e-01, -4.9570e+00, ..., -1.2518e+00,
-5.2751e+00, -8.3081e-01],
[ 1.2423e+00, 3.6091e+00, 2.2450e+00, ..., -1.0211e+00,
-5.4513e+00, -2.5857e+00],
[ 2.4599e+00, 2.0587e+00, 2.4293e+00, ..., -1.6951e+00,
-2.3171e+00, 4.9400e-01]],
...,
[[-2.8540e+00, -7.2024e+00, -5.6083e+00, ..., 1.6645e+00,
-1.2989e+01, -9.4372e+00],
[-3.3505e+00, -7.6951e+00, -5.5417e+00, ..., -8.3830e+00,
-2.4456e+01, -1.1029e+01],
[-2.1647e+00, -5.8628e+00, -4.0506e+00, ..., -1.8982e+01,
-1.1425e+01, -7.9265e+00],
...,
[-2.3185e+00, -5.1127e+00, -3.7318e+00, ..., -1.1974e+00,
2.2514e+00, -1.9779e+00],
[-2.6562e+00, -5.2396e+00, -5.1321e+00, ..., -6.0589e+00,
-4.1538e+00, -1.2988e+01],
[-2.6640e+00, -5.7703e+00, -2.4808e+00, ..., -4.7253e+00,
-2.4591e+00, -5.1731e+00]],
[[ 1.2853e-01, 9.5400e-01, 4.8558e-01, ..., -3.4887e+00,
-1.5969e+00, 2.3219e+00],
[ 1.4787e+00, 9.5846e-01, 4.4270e-01, ..., -2.6763e+00,
-1.9997e+00, 3.2822e+00],
[ 1.4565e+00, 7.3066e-01, 1.2041e-01, ..., -2.1784e+00,
1.3511e+00, 2.9094e+00],
...,
[ 1.4735e+00, -1.3786e+00, 2.7444e+00, ..., -2.6231e+00,
-1.3565e+00, 5.2742e-01],
[ 2.4592e+00, 1.4405e+00, 2.4277e+00, ..., 3.1278e-01,
8.7196e-01, 8.7803e-01],
[ 1.1301e+00, 7.4195e-01, 1.3370e+00, ..., -9.6763e-01,
-2.3773e-01, 6.3616e-01]],
[[-7.1255e-01, -6.3473e-02, -2.3385e-01, ..., 1.0526e+00,
-1.2209e-01, 5.1671e-01],
[-1.6448e+00, -1.7674e+00, -1.4056e+00, ..., 9.7484e-01,
-5.6521e+00, -3.4394e+00],
[-2.4329e+00, -1.1424e+00, -3.4959e-01, ..., -4.0322e+00,
-8.1407e+00, -1.2351e+00],
...,
[-2.0829e+00, 1.8281e-01, -7.4235e-01, ..., -3.1474e+00,
-4.1758e-01, 4.9747e-01],
[-1.4999e+00, -8.2091e-01, -1.4406e-02, ..., 3.3957e-01,
1.1925e-01, -1.2397e+00],
[-1.7801e-01, 2.6541e-01, -3.3861e-01, ..., -1.8997e+00,
-3.9692e+00, -2.0814e+00]]],
[[[ 8.2975e-01, 1.1835e+00, 1.1126e+00, ..., 3.2493e-01,
1.1653e-01, -4.4425e-01],
[-5.5360e-01, -1.6976e+00, 5.6198e-01, ..., 1.4029e+00,
1.3168e+00, -3.1496e-01],
[ 8.1116e-02, -8.0339e-01, 4.6457e-01, ..., 4.3066e-01,
8.2994e-01, 9.1808e-01],
...,
[ 3.1264e-01, -6.3082e-01, -7.0367e-01, ..., -1.1604e-01,
4.7418e-01, 6.3373e-01],
[-6.2190e-01, -1.9081e+00, -1.6106e+00, ..., -3.9556e-01,
-2.2181e-01, -2.0417e-01],
[-3.7552e-01, -1.7054e+00, -1.5986e+00, ..., -1.1414e+00,
-1.5362e+00, -2.7242e-01]],
[[-3.3704e+00, -5.1139e-01, -7.6626e-01, ..., -2.8076e+00,
-3.3814e+00, -8.2316e-01],
[-2.9403e+00, -2.1550e+00, -2.3074e+00, ..., -2.0073e+00,
-1.7189e+00, -1.4619e-01],
[-1.7800e+00, -2.2986e+00, -3.3504e+00, ..., -2.6448e+00,
-1.4058e+00, -4.8396e-01],
...,
[-1.5273e+00, -1.1847e+00, -1.9948e+00, ..., -3.0159e+00,
-2.3095e+00, -4.6702e-01],
[-3.1499e+00, -1.7406e+00, -3.0430e+00, ..., -2.4034e+00,
-2.1487e+00, -1.1440e+00],
[-2.2515e+00, -1.9165e+00, -2.6691e+00, ..., -1.9554e+00,
-1.8774e+00, -1.0087e+00]],
[[ 2.1488e+00, 2.7542e+00, 3.5120e-01, ..., 2.2602e+00,
4.6799e-01, 2.2259e+00],
[ 2.9501e-01, 9.5373e-01, 3.7280e-01, ..., 9.3377e-01,
-3.7819e-01, 2.1732e+00],
[ 9.4801e-01, 1.2166e+00, 2.6371e+00, ..., 5.4721e-01,
8.3625e-01, 2.3267e+00],
...,
[ 1.2907e+00, 1.4790e+00, 1.7238e+00, ..., -8.3755e-02,
1.5026e+00, 2.2829e+00],
[ 6.7493e-01, 1.8340e+00, 9.4209e-01, ..., 4.8268e-01,
1.0836e+00, 1.7108e+00],
[ 1.6032e+00, 1.6793e+00, 1.8278e+00, ..., 1.3385e+00,
1.7810e+00, 1.7576e+00]],
...,
[[-1.8892e+00, -6.1691e+00, -5.1542e+00, ..., -1.1648e+00,
-2.1069e+00, -3.7718e+00],
[-2.4300e+00, -7.3787e+00, -6.1350e+00, ..., -5.1116e+00,
-4.0372e+00, -4.9254e+00],
[-1.4701e+00, -4.1035e+00, -2.5854e+00, ..., -2.0678e+00,
-2.5051e+00, -2.0414e+00],
...,
[-2.4200e+00, -5.1610e+00, -4.1650e+00, ..., -4.0887e+00,
-3.8042e+00, -4.6804e+00],
[-2.7992e+00, -5.7223e+00, -5.4074e+00, ..., -7.0858e+00,
-6.1820e+00, -6.2457e+00],
[-1.4837e+00, -4.1561e+00, -2.7726e+00, ..., -3.3063e+00,
-2.7626e+00, -3.2537e+00]],
[[-3.4051e-01, 3.9147e-01, 1.8363e+00, ..., 3.3279e-01,
9.1618e-01, 1.5881e+00],
[ 1.1207e+00, -6.0755e-01, 6.0813e-01, ..., 8.2766e-01,
1.7041e+00, 2.4936e+00],
[ 9.8314e-01, -5.6199e-01, -1.6079e+00, ..., 9.3160e-01,
1.4595e+00, 1.7948e+00],
...,
[ 1.3219e+00, 4.3151e-01, -9.9869e-02, ..., 9.7647e-01,
1.5318e+00, 1.6627e+00],
[ 1.8332e+00, 1.7035e+00, 9.3084e-01, ..., 1.6507e+00,
1.4422e+00, 1.9915e+00],
[ 4.6870e-01, 7.1938e-01, 6.6692e-01, ..., 5.3569e-01,
6.9896e-01, 1.6270e+00]],
[[-6.5339e-01, -2.3596e-01, -1.1892e+00, ..., -1.8118e-01,
-3.7907e-01, 6.0930e-01],
[-1.2408e+00, -1.0189e+00, -1.4872e+00, ..., -1.6909e-01,
3.2060e-01, 7.8841e-01],
[-1.7358e+00, -4.0658e-01, 1.1502e+00, ..., -1.1176e+00,
5.3196e-01, -2.5879e-02],
...,
[-1.9300e+00, -1.6898e+00, -1.3536e+00, ..., -1.5811e+00,
-1.5631e+00, -6.4517e-01],
[-1.6306e+00, -2.1746e+00, -1.9998e+00, ..., -4.8557e-01,
-8.0098e-01, -5.7452e-01],
[-8.3123e-01, -9.5338e-01, -1.2095e+00, ..., -5.9424e-01,
-8.2919e-01, -9.0757e-02]]]], device='cuda:0'),
tensor([[[[ 8.4802e-01, 1.4942e+00, 1.1939e+00, ..., 1.0963e+00,
9.4303e-01, 9.7394e-01],
[-1.0111e+00, -2.0064e+00, -1.1810e+00, ..., -6.8862e-01,
-7.1996e-01, -1.6413e+00],
[-2.0301e-01, -7.3160e-01, 1.4087e-01, ..., 7.9890e-01,
1.0410e+00, 8.5060e-01],
...,
[-1.3765e-02, -1.1636e+00, -9.1611e-01, ..., -2.2421e-01,
-4.7772e-03, 2.3523e-01],
[-6.3568e-01, -1.8852e+00, -1.3644e+00, ..., -5.4978e-01,
-3.2142e-01, -2.1351e-01],
[-3.4118e-01, -1.7405e+00, -1.8751e+00, ..., -1.2240e+00,
-1.5554e+00, -2.7911e-01]],
[[-3.8170e+00, -7.6791e-01, -1.5419e+00, ..., -1.5016e+00,
-2.5690e+00, -1.2637e+00],
[-4.0565e+00, -2.8660e+00, -1.8303e+00, ..., -2.5316e+00,
-3.3972e+00, -9.9165e-01],
[-2.4225e+00, -1.6900e+00, -1.6022e+00, ..., -2.2447e+00,
-2.8860e+00, -1.1326e+00],
...,
[-1.6107e+00, -1.6843e+00, -2.0647e+00, ..., -1.5976e+00,
-1.9821e+00, -7.8941e-01],
[-3.0939e+00, -1.7006e+00, -2.8549e+00, ..., -1.8119e+00,
-2.4596e+00, -1.3064e+00],
[-2.4265e+00, -2.0598e+00, -2.7886e+00, ..., -1.9311e+00,
-1.9907e+00, -1.0003e+00]],
[[ 2.9207e+00, 4.2901e+00, 3.3769e+00, ..., 3.1065e+00,
8.6119e-01, 3.0743e+00],
[ 2.2929e-01, 1.1570e+00, 2.2882e+00, ..., 1.8690e+00,
-6.1713e-01, 2.8626e+00],
[ 9.0284e-01, 1.4797e+00, 1.5739e+00, ..., -8.1304e-01,
8.4566e-02, 2.0410e+00],
...,
[ 9.6213e-01, 1.1206e+00, 1.5923e+00, ..., 8.5293e-02,
1.0147e+00, 2.3107e+00],
[ 7.2198e-01, 1.8504e+00, 1.4612e+00, ..., -1.6889e-01,
9.9383e-01, 1.8513e+00],
[ 1.6813e+00, 1.6368e+00, 1.8633e+00, ..., 1.2189e+00,
1.8084e+00, 1.7937e+00]],
...,
[[-1.8899e+00, -7.3378e+00, -5.5049e+00, ..., -4.2334e+00,
-3.2583e+00, -5.6814e+00],
[-2.4181e+00, -7.0551e+00, -4.2825e+00, ..., -3.1798e+00,
-2.3151e+00, -4.3607e+00],
[-9.3304e-01, -4.5154e+00, -3.6885e+00, ..., -4.1699e+00,
-2.4695e+00, -2.3156e+00],
...,
[-2.5662e+00, -5.3922e+00, -4.3953e+00, ..., -6.4612e+00,
-4.6780e+00, -4.6635e+00],
[-2.7558e+00, -5.8688e+00, -5.6294e+00, ..., -7.2162e+00,
-5.7567e+00, -6.1281e+00],
[-1.5003e+00, -4.1843e+00, -2.8400e+00, ..., -3.4910e+00,
-2.6909e+00, -3.1663e+00]],
[[-2.0433e-01, 3.2352e-01, 6.1150e-01, ..., 8.2884e-01,
8.6552e-01, 1.3785e+00],
[ 1.6224e+00, 1.0810e+00, 1.8635e+00, ..., 2.0276e+00,
2.6948e+00, 3.0472e+00],
[ 1.1951e+00, 3.0723e-01, 8.7205e-01, ..., 4.1421e-01,
1.6910e+00, 2.3782e+00],
...,
[ 1.3194e+00, 1.9892e-01, -5.5305e-02, ..., 1.9797e+00,
1.1085e+00, 1.5748e+00],
[ 1.8804e+00, 1.6425e+00, 8.6302e-01, ..., 1.7410e+00,
1.2231e+00, 1.9759e+00],
[ 4.0292e-01, 7.4973e-01, 6.0473e-01, ..., 4.5477e-01,
6.4514e-01, 1.6624e+00]],
[[-5.5732e-01, 3.0398e-01, 1.3651e-01, ..., 1.9183e-01,
3.9628e-01, 1.3411e+00],
[-1.3475e+00, -1.7718e+00, -1.6315e+00, ..., -1.0332e+00,
-3.9923e-01, 1.9038e-01],
[-2.2009e+00, -7.4927e-01, 3.5145e-01, ..., 1.2674e+00,
6.9255e-01, 8.0709e-01],
...,
[-1.9374e+00, -1.8656e+00, -1.3413e+00, ..., -1.0279e+00,
-8.2773e-01, -8.7522e-01],
[-1.5888e+00, -2.0807e+00, -1.7161e+00, ..., -7.8557e-01,
-1.3691e+00, -7.2323e-01],
[-8.6286e-01, -9.1085e-01, -1.2503e+00, ..., -5.1658e-01,
-9.1650e-01, -8.7069e-02]]],
[[[ 1.0355e+00, 1.3355e+00, 1.2351e+00, ..., 1.5455e+00,
1.1621e+00, 5.1499e-01],
[-4.4072e-01, -2.1985e+00, -1.2004e+00, ..., -8.9692e-02,
-1.5907e-02, -1.6250e+00],
[-2.7591e-01, -9.7763e-01, -5.0041e-01, ..., -6.2371e-01,
1.8870e-01, 6.6898e-01],
...,
[ 4.7546e-01, -1.2463e-01, -1.2071e-01, ..., 4.1215e-01,
1.0315e+00, 8.2521e-01],
[-4.1601e-02, -5.9811e-01, 1.6284e-01, ..., 9.4512e-01,
1.5414e+00, 1.0816e-01],
[ 8.0733e-01, -2.0413e-01, -1.0983e-01, ..., 3.6656e-01,
2.7434e-01, 7.0051e-01]],
[[-2.6290e+00, 1.0464e-01, -1.4437e+00, ..., -2.0663e+00,
-2.4733e+00, -6.9326e-01],
[-3.8719e+00, -2.7798e+00, -2.6352e+00, ..., -4.0434e+00,
-4.2087e+00, -1.7045e+00],
[-3.0425e+00, -2.4892e+00, -2.8596e+00, ..., -3.1237e+00,
-2.8669e+00, -1.4228e+00],
...,
[-8.7830e-01, -3.0154e-01, -7.4417e-01, ..., -4.8227e-01,
7.8380e-02, 8.0593e-01],
[-2.4443e+00, -1.9427e-01, -3.0031e-01, ..., -6.3524e-02,
-7.5974e-02, 6.8142e-01],
[-1.5178e+00, -1.2263e+00, -9.2664e-01, ..., -4.8592e-01,
-6.3174e-01, -1.8392e-01]],
[[ 2.2656e+00, 2.8254e+00, 2.4786e+00, ..., 1.9232e+00,
-1.0379e-01, 2.6283e+00],
[ 8.7162e-01, 6.4438e-01, 2.3257e+00, ..., 1.7840e+00,
2.6777e-01, 2.7868e+00],
[ 1.3539e+00, 1.5100e+00, 1.9158e+00, ..., 1.4477e+00,
1.7364e+00, 2.9328e+00],
...,
[ 9.1085e-01, 8.8905e-01, 1.6300e+00, ..., 9.3114e-01,
5.7850e-01, 1.8592e+00],
[ 1.0725e+00, 2.1960e+00, 1.9718e+00, ..., 1.0441e+00,
7.7685e-01, 1.7443e+00],
[ 2.3831e+00, 1.8412e+00, 2.4174e+00, ..., 1.9696e+00,
1.1084e+00, 1.7664e+00]],
...,
[[-2.2342e+00, -6.3704e+00, -5.4433e+00, ..., -5.3028e+00,
-5.3792e+00, -5.7816e+00],
[-2.8449e+00, -6.9195e+00, -5.0157e+00, ..., -7.1078e+00,
-5.5055e+00, -5.3027e+00],
[-1.9311e+00, -5.3683e+00, -3.9922e+00, ..., -4.0888e+00,
-4.4378e+00, -3.3917e+00],
...,
[-2.4269e+00, -4.8785e+00, -2.7139e+00, ..., -3.1891e+00,
-2.7965e+00, -3.0295e+00],
[-2.4734e+00, -5.5272e+00, -4.0461e+00, ..., -4.3997e+00,
-3.2353e+00, -4.9146e+00],
[-2.5553e+00, -5.8154e+00, -2.8136e+00, ..., -2.9686e+00,
-1.6538e+00, -3.3247e+00]],
[[ 4.9247e-01, 1.1573e+00, 4.3180e-01, ..., -5.7136e-02,
1.7117e+00, 1.6724e+00],
[ 1.2039e+00, 6.9125e-01, 2.8528e-01, ..., -3.7838e-01,
1.3859e+00, 2.8197e+00],
[ 1.2701e+00, 7.1502e-01, 3.6669e-01, ..., 1.2697e+00,
1.3539e+00, 2.0825e+00],
...,
[ 1.5542e+00, 4.7691e-01, 1.1688e+00, ..., 1.8906e+00,
1.9457e+00, 1.4311e+00],
[ 2.0326e+00, 1.1817e+00, 1.7526e+00, ..., 1.9181e+00,
1.7018e+00, 2.5657e+00],
[ 1.1944e+00, 9.1062e-01, 1.3184e+00, ..., 1.3216e+00,
1.5029e+00, 1.5478e+00]],
[[-5.8022e-01, 1.2113e-01, -1.7152e-01, ..., 4.7118e-01,
4.7482e-03, 9.9232e-01],
[-1.3721e+00, -1.8554e+00, -1.6548e+00, ..., -2.3433e+00,
-1.3114e+00, -1.7863e-01],
[-2.2096e+00, -1.0362e+00, -3.7680e-01, ..., -1.1040e+00,
-2.9804e-01, -6.5006e-02],
...,
[-2.0057e+00, -1.4539e+00, -6.6784e-01, ..., -1.8366e-01,
-4.3699e-01, -4.1779e-01],
[-1.7489e+00, -1.2492e+00, -5.9064e-01, ..., -3.5924e-01,
-4.0436e-01, 4.0888e-01],
[-2.3056e-01, 4.1691e-01, -1.8385e-01, ..., 1.9008e-01,
1.3719e-02, 6.0862e-01]]],
[[[ 9.6298e-01, 1.1623e+00, 9.6457e-01, ..., 1.3384e+00,
1.1292e+00, 1.0396e+00],
[-9.3348e-01, -2.1117e+00, -1.1136e+00, ..., -7.0006e-01,
-6.3104e-01, -1.5741e+00],
[-8.7425e-02, -7.0035e-01, 1.5017e+00, ..., -8.0214e-01,
5.5906e-01, 7.1372e-01],
...,
[-2.6911e-02, -1.1223e+00, -9.2467e-01, ..., -9.6710e-01,
-3.4034e+00, -4.8045e-01],
[-6.4650e-01, -1.8712e+00, -1.1978e+00, ..., -3.8271e+00,
-2.9119e-01, 1.2005e+00],
[-3.1510e-01, -1.7121e+00, -1.6744e+00, ..., -2.3210e+00,
-2.1922e+00, -8.3713e-01]],
[[-3.7648e+00, -3.7367e-01, -1.2755e+00, ..., -1.4274e+00,
-2.2761e+00, -1.0164e+00],
[-4.0369e+00, -1.7585e+00, -7.4719e-01, ..., -2.0974e+00,
-3.1957e+00, -8.1247e-01],
[-2.6864e+00, -2.5998e+00, -5.2444e+00, ..., -1.9446e+00,
-2.5179e+00, -9.5683e-01],
...,
[-1.5698e+00, -1.6370e+00, -2.3497e+00, ..., -7.7981e+00,
-1.1810e+01, -6.1630e+00],
[-2.9915e+00, -1.6980e+00, -2.5769e+00, ..., -1.2400e+01,
-1.2757e+01, -6.1982e+00],
[-2.2972e+00, -2.0275e+00, -2.7727e+00, ..., -8.4209e+00,
-6.7754e+00, -3.2552e+00]],
[[ 3.5674e+00, 3.9442e+00, 7.5293e-01, ..., 2.6124e+00,
1.0890e+00, 2.9389e+00],
[ 1.1884e+00, 1.0682e+00, 5.3130e-01, ..., 1.1902e+00,
-9.0523e-01, 2.6329e+00],
[ 1.7320e+00, 2.1115e+00, 1.8870e+00, ..., 2.2002e+00,
2.7322e-01, 1.6635e+00],
...,
[ 9.4844e-01, 1.0875e+00, 1.2823e+00, ..., -4.0814e+00,
-1.1495e+01, -4.1270e+00],
[ 7.0755e-01, 1.8745e+00, 1.2883e+00, ..., -8.7999e+00,
-1.2976e+01, -7.0105e+00],
[ 1.7421e+00, 1.6624e+00, 1.8798e+00, ..., -1.8181e+00,
-2.3082e+00, -1.0879e+00]],
...,
[[-2.6183e+00, -7.5941e+00, -5.8316e+00, ..., -5.0626e+00,
-3.2523e+00, -5.4581e+00],
[-3.0566e+00, -6.6276e+00, -5.6870e+00, ..., -4.3175e+00,
-2.5279e+00, -4.1175e+00],
[-1.0654e+00, -4.1093e+00, -4.0844e+00, ..., -3.1805e+00,
-3.1418e+00, -3.1979e+00],
...,
[-2.6198e+00, -5.6182e+00, -4.2149e+00, ..., -1.3412e+01,
-9.8768e+00, -1.3226e+01],
[-2.8061e+00, -5.8742e+00, -5.4045e+00, ..., -1.1310e+01,
-1.4039e+01, -1.4748e+01],
[-1.4777e+00, -4.1127e+00, -2.7127e+00, ..., -5.2527e+00,
-5.0838e+00, -5.9649e+00]],
[[-1.6459e-01, 9.7304e-01, 4.0188e-01, ..., 9.7027e-01,
9.3423e-01, 1.4486e+00],
[ 1.8045e+00, 1.6998e+00, 1.7437e+00, ..., 2.1490e+00,
2.4512e+00, 2.9832e+00],
[ 1.2402e+00, -9.5540e-01, -5.5123e-01, ..., 2.1777e+00,
2.0877e+00, 2.3219e+00],
...,
[ 1.3464e+00, 2.5323e-01, 2.4011e-02, ..., -8.1803e+00,
-4.5300e+00, -1.5301e+00],
[ 1.9181e+00, 1.7382e+00, 1.1812e+00, ..., -8.7176e-02,
-1.1969e+00, 7.8334e-02],
[ 4.6908e-01, 8.1845e-01, 6.8759e-01, ..., -3.9131e-01,
-6.3964e-01, 1.9100e+00]],
[[-5.5458e-01, -6.4097e-01, 3.4328e-01, ..., 6.3078e-01,
5.8553e-01, 1.3764e+00],
[-1.2975e+00, -1.8660e+00, -1.9429e+00, ..., -1.1971e+00,
-5.8861e-01, 1.8081e-01],
[-2.3035e+00, -2.0860e-01, 6.3709e-03, ..., 8.0907e-02,
-1.5953e-01, 5.8895e-01],
...,
[-1.9765e+00, -1.8084e+00, -1.3705e+00, ..., -5.9309e+00,
-3.4480e+00, 1.9138e+00],
[-1.7236e+00, -2.0082e+00, -1.6134e+00, ..., -7.1071e+00,
-5.7231e+00, -1.7612e+00],
[-8.7322e-01, -8.2490e-01, -1.1486e+00, ..., -3.0907e+00,
-2.7672e+00, -5.2276e-01]]],
[[[ 5.6531e-01, 8.4821e-01, 5.7111e-01, ..., 3.4754e-01,
4.4890e-01, 3.9248e-02],
[-2.7672e-01, -8.9620e-01, -4.8631e-01, ..., 9.6549e-02,
4.6220e-02, -3.2975e-01],
[-2.0105e-01, -5.6303e-02, 2.2391e-02, ..., -6.0822e-02,
8.4422e-02, 6.1769e-01],
...,
[ 5.6415e-01, -9.7989e-02, -2.7492e-01, ..., 3.1253e-01,
8.4363e-01, 6.8775e-01],
[ 2.9407e-02, -4.5577e-01, 1.2968e-01, ..., 9.6556e-01,
1.5504e+00, 1.0827e-01],
[ 8.0394e-01, -9.4382e-02, -4.8604e-02, ..., 5.0139e-01,
2.9159e-01, 6.9968e-01]],
[[-2.3826e+00, -6.0608e-01, -1.8540e+00, ..., -1.6064e+00,
-2.1858e+00, -1.4659e+00],
[-2.5085e+00, -7.4452e-01, -1.6626e+00, ..., -1.3378e+00,
-2.3741e+00, -1.6180e+00],
[-2.4979e+00, -1.9306e+00, -2.4366e+00, ..., -1.9577e+00,
-2.3194e+00, -1.1259e+00],
...,
[-3.8761e-01, -6.3219e-01, -1.2475e+00, ..., -5.4094e-01,
-1.1044e-01, 7.7123e-01],
[-2.0323e+00, -3.4621e-01, -3.4221e-01, ..., 2.6230e-01,
1.4539e-01, 7.4738e-01],
[-1.1960e+00, -1.0956e+00, -9.1000e-01, ..., -3.6544e-01,
-4.9226e-01, -1.7220e-01]],
[[ 1.9450e+00, 2.5401e+00, 1.5877e+00, ..., 1.5734e+00,
7.9500e-01, 2.0335e+00],
[-2.5757e-01, 3.2000e-01, 5.6999e-01, ..., 7.5089e-01,
3.2497e-01, 2.2023e+00],
[ 2.0240e-02, 4.9549e-01, 5.9991e-01, ..., -8.0924e-02,
7.9484e-01, 1.9919e+00],
...,
[ 4.6486e-01, 2.7456e-01, 9.7020e-01, ..., 3.8834e-01,
7.2819e-02, 1.5953e+00],
[ 6.9330e-01, 1.9081e+00, 1.7902e+00, ..., 1.1739e+00,
7.7311e-01, 1.7232e+00],
[ 2.0302e+00, 1.9177e+00, 2.3796e+00, ..., 2.1558e+00,
1.0971e+00, 1.7681e+00]],
...,
[[-2.0597e+00, -4.9994e+00, -3.2366e+00, ..., -2.0932e+00,
-1.7020e+00, -2.9940e+00],
[-2.5207e+00, -5.3997e+00, -3.2111e+00, ..., -2.4666e+00,
-2.2218e+00, -2.8578e+00],
[-1.5769e+00, -5.5256e+00, -3.4775e+00, ..., -2.4645e+00,
-2.5302e+00, -2.2858e+00],
...,
[-2.1726e+00, -3.6503e+00, -1.9090e+00, ..., -2.3792e+00,
-2.0647e+00, -2.6867e+00],
[-2.0844e+00, -4.5755e+00, -3.7000e+00, ..., -4.3926e+00,
-3.1589e+00, -5.0254e+00],
[-2.1432e+00, -5.1600e+00, -2.7766e+00, ..., -3.1538e+00,
-1.5407e+00, -3.4701e+00]],
[[ 9.4083e-01, 1.6546e+00, 8.1194e-01, ..., 1.1311e+00,
1.5491e+00, 1.3863e+00],
[ 1.1836e+00, 1.0159e+00, 9.0459e-01, ..., 8.8171e-01,
7.8549e-01, 1.8814e+00],
[ 6.1853e-01, -3.6789e-01, 2.5108e-02, ..., 3.8190e-01,
9.1490e-02, 1.3638e+00],
...,
[ 1.3975e+00, 5.9535e-02, 5.3944e-01, ..., 1.6043e+00,
1.7639e+00, 1.4202e+00],
[ 1.9215e+00, 9.7036e-01, 1.6352e+00, ..., 2.1032e+00,
1.7356e+00, 2.4921e+00],
[ 1.2595e+00, 6.8921e-01, 1.2160e+00, ..., 1.3748e+00,
1.5915e+00, 1.5237e+00]],
[[-9.5921e-02, -6.4900e-01, -1.1819e+00, ..., -5.4820e-01,
-8.8580e-01, -1.2227e-01],
[-8.7746e-01, -2.6472e+00, -2.4522e+00, ..., -1.2097e+00,
-1.1289e+00, -6.8493e-01],
[-1.4841e+00, -2.2169e+00, -7.9216e-01, ..., -1.7325e-01,
-3.8673e-01, -2.8494e-01],
...,
[-1.5243e+00, -1.3579e+00, -9.6503e-01, ..., -5.4648e-01,
-4.8093e-01, -4.7601e-01],
[-1.5568e+00, -1.1208e+00, -6.6902e-01, ..., -4.8432e-01,
-4.3812e-01, 3.7522e-01],
[-3.5666e-02, 5.0842e-01, -1.8964e-01, ..., 5.5722e-02,
3.2147e-02, 6.3896e-01]]],
[[[ 5.6737e-01, 7.7471e-01, 4.6528e-01, ..., 3.1280e-01,
3.7338e-01, -3.9356e-02],
[-2.9321e-01, -8.9599e-01, -3.8136e-01, ..., 6.1693e-02,
5.9976e-02, -3.0914e-01],
[ 1.4226e-01, 6.8669e-01, 9.5530e-01, ..., 1.7772e-01,
2.1017e-01, 7.3868e-01],
...,
[ 2.0091e-01, -5.3417e-01, -4.0914e-01, ..., -5.7464e+00,
-3.6331e+00, 1.4084e-01],
[-4.0809e-01, -2.0083e+00, -1.7444e+00, ..., -4.2871e+00,
5.2949e-01, 2.6111e+00],
[ 4.6430e-01, -8.3335e-01, -9.5835e-01, ..., -2.7151e+00,
-1.2062e+00, 9.7199e-01]],
[[-2.6698e+00, -7.6339e-01, -1.9863e+00, ..., -1.7164e+00,
-2.3641e+00, -1.5390e+00],
[-2.5587e+00, -6.8840e-01, -1.6549e+00, ..., -1.4191e+00,
-2.4534e+00, -1.6887e+00],
[-3.0046e+00, -3.1101e+00, -3.8385e+00, ..., -1.9165e+00,
-2.2936e+00, -1.1533e+00],
...,
[-8.0708e-01, -6.1476e-01, -2.3450e+00, ..., -8.6943e+00,
-8.2099e+00, -4.1846e+00],
[-2.2761e+00, -4.9102e-02, -1.2229e+00, ..., -4.8739e+00,
-3.9008e+00, -1.8791e+00],
[-1.0457e+00, -1.2019e+00, -1.2643e+00, ..., -5.1909e+00,
-5.4482e+00, -9.6003e-01]],
[[ 2.0667e+00, 2.6184e+00, 1.7155e+00, ..., 1.6244e+00,
7.8078e-01, 2.1260e+00],
[-1.0714e-01, 4.0716e-01, 5.6890e-01, ..., 7.5165e-01,
3.6915e-01, 2.2987e+00],
[ 1.4993e-01, 1.4047e-01, 1.3852e+00, ..., 1.2223e-01,
7.2427e-01, 2.0306e+00],
...,
[ 1.3498e+00, 1.1929e+00, 2.0450e+00, ..., -3.5591e+00,
-3.3983e+00, 2.4283e+00],
[ 9.1472e-01, 2.1829e+00, 2.0850e+00, ..., -3.6598e+00,
-5.2225e+00, -1.1195e+00],
[ 1.8637e+00, 1.1145e+00, 1.6068e+00, ..., -3.5721e+00,
-4.5588e+00, -5.5401e-01]],
...,
[[-2.1292e+00, -5.1087e+00, -3.1895e+00, ..., -2.1888e+00,
-1.7107e+00, -3.0690e+00],
[-2.6082e+00, -5.7018e+00, -3.2082e+00, ..., -2.7214e+00,
-2.4029e+00, -3.1745e+00],
[-1.8119e+00, -6.6871e+00, -5.3840e+00, ..., -2.0137e+00,
-2.2090e+00, -2.2045e+00],
...,
[-1.9790e+00, -5.3823e+00, -3.2748e+00, ..., -2.6100e-01,
5.9273e-01, -9.7409e-01],
[-2.8362e+00, -6.9044e+00, -4.1702e+00, ..., 8.5642e-01,
-2.0888e+00, -6.2555e+00],
[-3.0734e+00, -7.2494e+00, -3.6232e+00, ..., -3.1093e-01,
-2.7276e+00, -1.1515e+00]],
[[ 7.8491e-01, 1.5808e+00, 8.2685e-01, ..., 1.0643e+00,
1.4548e+00, 1.4472e+00],
[ 1.1877e+00, 1.0109e+00, 7.4711e-01, ..., 8.8973e-01,
8.3673e-01, 1.9964e+00],
[ 3.4845e-01, -5.6862e-01, -1.3100e+00, ..., 6.6739e-01,
1.2878e-01, 1.2958e+00],
...,
[ 1.6504e+00, -9.1716e-03, -4.4639e-01, ..., -7.5990e+00,
-6.3026e+00, 9.1156e-02],
[ 2.2823e+00, 1.4937e+00, 1.1312e+00, ..., -3.5723e+00,
-2.4622e+00, 7.1437e-01],
[ 1.0333e+00, 5.9130e-01, 4.2861e-01, ..., -1.9456e+00,
-2.2042e+00, -3.9789e-01]],
[[-1.4838e-01, -7.4270e-01, -1.2263e+00, ..., -5.8277e-01,
-8.8229e-01, -8.3380e-02],
[-8.7818e-01, -2.5003e+00, -2.6076e+00, ..., -1.1347e+00,
-9.0789e-01, -5.4931e-01],
[-1.8843e+00, -1.8100e+00, -5.8391e-01, ..., 1.1029e-01,
-1.5044e-01, -2.0086e-01],
...,
[-2.3625e+00, -2.0450e+00, -8.4260e-01, ..., -2.5038e+00,
-5.7971e-01, -5.7325e-01],
[-2.5187e+00, -2.2872e+00, -1.2129e+00, ..., 2.6529e+00,
2.1184e+00, 2.1239e+00],
[-4.5710e-01, -2.1296e-01, -3.4318e-01, ..., 2.0486e-01,
-1.6483e+00, -7.5599e-01]]]], device='cuda:0'),
tensor([[[[ 8.4876e-01, 1.5312e+00, 1.2033e+00, ..., 3.4160e-01,
-3.2960e-01, 1.3647e-01],
[-9.1891e-01, -1.8130e+00, -1.2115e+00, ..., 7.7155e-01,
-9.3636e-01, -2.7428e+00],
[-6.9764e-02, -8.4874e-01, -9.6267e-03, ..., -1.9713e-01,
1.7641e+00, 2.0945e+00],
...,
[ 1.6282e+00, -4.3950e+00, 1.6274e+00, ..., 4.0912e-01,
-4.1941e-01, 5.1405e-01],
[ 8.8959e-01, -6.6590e+00, -1.1025e+00, ..., -3.2601e-01,
-6.5580e-01, 2.7737e-01],
[ 6.0882e-01, -1.1151e-01, -3.9983e+00, ..., -6.1722e-01,
-1.4818e+00, 1.6735e-02]],
[[-3.8862e+00, -8.5074e-01, -1.7146e+00, ..., -3.1968e+00,
-3.3366e+00, -1.6103e+00],
[-3.9522e+00, -2.8860e+00, -2.3927e+00, ..., -4.2894e+00,
-4.8176e+00, -1.7572e+00],
[-2.3547e+00, -2.3306e+00, -2.0879e+00, ..., -1.7501e+00,
-7.9084e-01, -1.4015e+00],
...,
[-5.5813e+00, 5.7534e-01, -1.2094e+00, ..., -1.3743e+00,
-1.9711e+00, -1.6288e+00],
[-9.8132e+00, -3.1582e+00, -2.4002e+00, ..., -2.4838e+00,
-2.0525e+00, -1.4042e+00],
[-7.2744e+00, -6.6733e+00, -6.2285e+00, ..., -2.1468e+00,
-2.1664e+00, -1.3675e+00]],
[[ 2.8250e+00, 4.2046e+00, 3.5083e+00, ..., 4.2615e+00,
2.5128e+00, 3.8581e+00],
[ 6.3179e-02, 1.1799e+00, 2.4294e+00, ..., 6.9922e-01,
-1.8741e+00, 9.5092e-01],
[ 7.5317e-01, 1.6018e+00, 1.7402e+00, ..., -1.3005e+00,
-2.4660e+00, -2.5309e-01],
...,
[-8.7388e+00, -8.4648e+00, -9.1685e-01, ..., -3.5563e-01,
-1.3574e-01, 1.4796e+00],
[-1.8513e+00, -7.6757e+00, -2.6088e+00, ..., -5.5854e-01,
1.3848e-01, 1.3389e+00],
[ 3.2848e+00, -1.3907e+00, -2.4837e+00, ..., 6.1119e-01,
1.0360e+00, 1.5515e+00]],
...,
[[-1.7669e+00, -7.0564e+00, -5.4421e+00, ..., -4.9720e+00,
-3.7381e+00, -5.7742e+00],
[-2.4168e+00, -6.9777e+00, -4.6753e+00, ..., -6.6529e+00,
-4.8041e+00, -4.9452e+00],
[-8.8687e-01, -4.4302e+00, -4.3093e+00, ..., -8.2522e+00,
-9.9863e+00, -1.1276e+01],
...,
[-4.1538e+00, -8.8215e+00, -1.2697e+01, ..., -1.6621e+00,
-5.8741e-01, -3.8473e+00],
[-4.2693e+00, -9.2107e+00, -9.6532e+00, ..., -3.2696e+00,
-1.3379e+00, -4.9544e+00],
[-1.9972e+00, -4.9269e+00, -3.0075e+00, ..., -1.2061e+00,
-5.4991e-01, -2.0734e+00]],
[[-2.8079e-01, 2.2629e-01, 5.6171e-01, ..., 8.8974e-01,
9.2887e-01, 1.7752e+00],
[ 1.6550e+00, 9.5254e-01, 1.8945e+00, ..., -2.6847e+00,
-1.0812e+00, 2.2606e+00],
[ 1.0636e+00, 1.9261e-02, 7.1074e-01, ..., 6.7359e-01,
1.0441e+00, -2.0545e-01],
...,
[ 3.0411e+00, -3.9964e-01, -4.5748e+00, ..., 1.0587e+00,
7.1825e-01, 9.3969e-01],
[ 1.6528e+00, 3.3914e+00, -8.2386e-01, ..., 7.9167e-01,
1.2544e+00, 1.2519e+00],
[-2.4167e+00, 3.7187e-01, 1.6443e-01, ..., 2.3838e-01,
4.7023e-01, 1.0937e+00]],
[[-5.9797e-01, 2.7684e-01, 5.7910e-02, ..., 3.2411e-01,
3.1221e-01, 8.4366e-01],
[-1.2924e+00, -1.6983e+00, -1.6568e+00, ..., -7.7990e-01,
-1.4254e+00, -9.6674e-03],
[-2.1387e+00, -1.0348e+00, 5.3893e-01, ..., 1.7406e+00,
2.0987e+00, 1.4992e+00],
...,
[-4.9559e+00, -1.2332e+00, -3.3654e+00, ..., -1.2864e+00,
-9.1515e-01, -8.1389e-01],
[-3.3291e+00, -2.6901e+00, -4.4585e+00, ..., -2.4310e+00,
-1.2255e+00, -3.9971e-01],
[-7.1919e-01, -2.3644e+00, -2.5961e+00, ..., -1.2827e+00,
-1.4584e+00, -2.6081e-01]]],
[[[ 9.1714e-01, 1.5926e+00, 1.0879e+00, ..., 8.4652e-01,
1.0044e+00, 4.4294e-01],
[-9.3979e-01, -1.5677e+00, -7.4568e-01, ..., -1.9728e-01,
-1.3968e-01, -3.5967e-01],
[-2.7219e-01, -5.5596e-01, -3.0665e-01, ..., -8.2583e-01,
-1.1420e-01, 6.0427e-01],
...,
[ 4.4097e-01, 1.2839e-01, 3.6040e-01, ..., -4.5642e-02,
-6.5274e-02, 3.5441e-01],
[-5.5415e-02, -9.0386e-01, 4.5259e-01, ..., -3.6117e-02,
1.1328e-01, -8.1649e-02],
[ 8.7460e-01, -3.9884e-01, -1.2701e-01, ..., -3.5095e-01,
-9.5634e-01, 1.5885e-01]],
[[-3.8191e+00, -1.4365e+00, -1.9261e+00, ..., -1.4702e+00,
-2.0576e+00, -1.8147e+00],
[-3.9057e+00, -3.2706e+00, -2.5579e+00, ..., -1.0439e+00,
-2.2200e+00, -2.1408e+00],
[-2.6582e+00, -2.6441e+00, -3.2691e+00, ..., -1.3991e+00,
-1.7106e+00, -1.4016e+00],
...,
[-1.6651e+00, -1.4730e+00, -2.0317e+00, ..., -6.7853e-01,
-9.7091e-01, -8.7012e-01],
[-3.2916e+00, -1.2359e-01, -6.0443e-01, ..., -5.9254e-01,
-1.1090e+00, -1.2448e+00],
[-2.3371e+00, -1.5732e+00, -1.1179e+00, ..., -6.9741e-01,
-7.8659e-01, -5.4045e-01]],
[[ 2.6140e+00, 3.7731e+00, 2.9350e+00, ..., 1.6990e+00,
8.1865e-01, 2.5802e+00],
[-1.6287e-01, 8.5651e-01, 1.9549e+00, ..., 1.3339e+00,
4.8970e-01, 2.6313e+00],
[ 2.5018e-01, 1.8894e+00, 2.1591e+00, ..., 7.8526e-01,
1.0065e+00, 1.8630e+00],
...,
[-3.5318e-01, -1.9570e-01, 1.4065e+00, ..., -2.3017e-01,
5.5465e-01, 1.4829e+00],
[ 7.5171e-01, 2.6078e+00, 1.9618e+00, ..., -1.6248e-01,
7.7755e-01, 1.1458e+00],
[ 1.7570e+00, 1.3628e+00, 1.7887e+00, ..., 5.3996e-01,
1.2855e+00, 1.3373e+00]],
...,
[[-1.4721e+00, -6.7259e+00, -5.5777e+00, ..., -5.1835e+00,
-3.6586e+00, -4.3166e+00],
[-2.0738e+00, -6.0034e+00, -4.6598e+00, ..., -4.6603e+00,
-3.8412e+00, -3.2907e+00],
[-4.4885e-01, -2.8220e+00, -3.6346e+00, ..., -1.7341e+00,
-2.0892e+00, -1.3409e+00],
...,
[-1.6503e+00, -3.6599e+00, -1.1910e+00, ..., -4.1346e+00,
-3.1508e+00, -2.6983e+00],
[-1.8080e+00, -4.4164e+00, -3.9741e+00, ..., -5.2521e+00,
-4.2654e+00, -4.3052e+00],
[-2.6872e+00, -5.5871e+00, -2.8284e+00, ..., -3.4636e+00,
-2.4871e+00, -2.3862e+00]],
[[-6.4459e-01, -2.2417e-01, 3.2624e-01, ..., 1.4516e+00,
1.6845e+00, 1.2035e+00],
[ 1.6375e+00, 6.2513e-01, 1.6945e+00, ..., 1.5454e+00,
1.1076e+00, 1.8660e+00],
[ 9.6986e-01, -3.3414e-01, 1.5742e-01, ..., 1.7474e+00,
7.8606e-01, 1.3205e+00],
...,
[ 7.6344e-01, -1.6274e+00, 9.3190e-01, ..., 1.4481e+00,
6.7623e-01, 9.5498e-01],
[ 1.9707e+00, 1.4302e+00, 2.0249e+00, ..., 1.3500e+00,
7.2438e-01, 1.4421e+00],
[ 9.8572e-01, 6.0768e-01, 1.2254e+00, ..., 7.5272e-01,
6.7743e-01, 1.4225e+00]],
[[-4.4394e-01, 5.4085e-01, 1.7353e-01, ..., -3.7891e-01,
-6.3729e-01, -8.7442e-02],
[-1.0853e+00, -1.2436e+00, -1.1171e+00, ..., -1.2756e+00,
-1.3562e+00, -1.1066e+00],
[-1.7189e+00, -6.6273e-01, 1.9887e-02, ..., -1.1754e+00,
-1.0855e+00, -2.1149e-01],
...,
[-1.6411e+00, -2.0763e+00, -1.7997e+00, ..., -9.1216e-01,
-1.5397e+00, -8.5013e-01],
[-2.0078e+00, -1.0320e+00, -7.3228e-01, ..., -7.6737e-01,
-1.6496e+00, -7.2203e-01],
[-5.2181e-01, 2.3484e-01, -2.4461e-01, ..., -2.4082e-01,
-8.3630e-01, -3.3916e-02]]],
[[[ 9.0040e-01, 1.5477e+00, 1.3662e+00, ..., -7.0231e-01,
6.4999e-01, 8.0542e-01],
[-9.3779e-01, -1.7195e+00, -1.1834e+00, ..., -1.7179e+00,
-2.3367e+00, -1.4948e+00],
[ 1.1274e-01, -7.2560e-01, -1.7085e+00, ..., 4.6800e+00,
2.1617e+00, 1.1680e+00],
...,
[-2.8865e-03, -1.0761e+00, -8.2051e-01, ..., -8.4235e+00,
-3.0587e+00, 1.0351e-02],
[-6.1131e-01, -1.7520e+00, -1.2623e+00, ..., -8.3161e+00,
-5.4037e+00, -5.1522e+00],
[-2.0450e-01, -1.6712e+00, -1.7494e+00, ..., -2.9846e+00,
-2.2731e+00, -1.6641e+00]],
[[-3.6576e+00, -7.6721e-01, -1.8683e+00, ..., -2.0140e+00,
-2.3929e+00, -1.2817e+00],
[-4.0607e+00, -3.9655e+00, -5.6194e+00, ..., -4.2730e+00,
-3.6902e+00, -1.1958e+00],
[-2.7863e+00, -5.3172e+00, -1.0080e+01, ..., -2.5810e+00,
-3.5104e+00, -1.0327e+00],
...,
[-1.4554e+00, -1.5709e+00, -2.1573e+00, ..., -8.4664e+00,
-7.5442e+00, -2.1063e+00],
[-2.8864e+00, -1.5235e+00, -2.5184e+00, ..., -1.2056e+01,
-1.3110e+01, -4.5632e+00],
[-2.2336e+00, -1.9250e+00, -2.5861e+00, ..., -7.0175e+00,
-8.1559e+00, -3.4243e+00]],
[[ 2.8676e+00, 4.4243e+00, 3.2651e+00, ..., 5.1505e+00,
1.4672e+00, 2.7105e+00],
[-1.0859e-03, 1.0604e+00, 1.4003e+00, ..., 1.4581e+00,
1.1271e+00, 2.3556e+00],
[ 5.7298e-01, 1.9851e+00, 3.2213e+00, ..., -3.7996e+00,
-1.4902e+00, 9.2343e-01],
...,
[ 1.0202e+00, 1.0177e+00, 1.2543e+00, ..., -1.0823e+01,
-6.2139e+00, -3.6747e+00],
[ 7.6311e-01, 1.9278e+00, 1.4123e+00, ..., -8.2823e+00,
-8.1447e+00, -4.4186e+00],
[ 1.8259e+00, 1.5813e+00, 1.8653e+00, ..., -1.3095e+00,
-4.2688e+00, -2.4190e+00]],
...,
[[-1.7103e+00, -7.1402e+00, -5.5485e+00, ..., -6.3219e+00,
-3.9198e+00, -6.3182e+00],
[-2.2668e+00, -7.0067e+00, -5.5622e+00, ..., -7.1008e+00,
-2.6132e+00, -5.4102e+00],
[-9.2978e-01, -5.1080e+00, -6.0487e+00, ..., -2.4151e+01,
-1.3438e+01, -6.2118e+00],
...,
[-2.6875e+00, -5.5629e+00, -4.0637e+00, ..., -2.4610e+00,
-6.0692e+00, -8.4221e+00],
[-3.0172e+00, -6.2712e+00, -5.2318e+00, ..., -1.2729e+00,
2.8189e-01, 8.4700e-01],
[-1.6970e+00, -4.5060e+00, -2.7317e+00, ..., -6.4075e-01,
1.3141e+00, 5.8780e-01]],
[[-1.2752e-01, 2.6441e-01, 3.4734e-01, ..., 7.2107e-01,
1.1901e+00, 1.6628e+00],
[ 1.5434e+00, 2.4494e-01, 5.4111e-01, ..., -2.1318e-01,
3.0189e+00, 3.0901e+00],
[ 7.4773e-01, -1.4116e+00, -5.1597e+00, ..., -4.1466e+00,
6.7717e-01, 2.1666e+00],
...,
[ 1.3982e+00, 2.5263e-01, -7.6547e-03, ..., 6.0158e-01,
-3.5254e+00, -1.4328e+00],
[ 2.0293e+00, 1.8232e+00, 1.1215e+00, ..., -7.7165e-01,
-4.8363e+00, -2.2220e+00],
[ 5.2772e-01, 9.0936e-01, 7.1836e-01, ..., -2.8882e+00,
-2.9774e+00, -1.5662e+00]],
[[-5.0530e-01, 2.5927e-01, 2.3753e-02, ..., -5.8507e-01,
-1.6893e-01, 1.2383e+00],
[-1.2670e+00, -2.4640e+00, -2.8908e+00, ..., -1.6231e+00,
-2.6900e+00, -9.2203e-02],
[-2.1849e+00, -2.1225e+00, -1.3611e+00, ..., 3.5047e+00,
8.6132e-01, 5.1025e-01],
...,
[-2.0882e+00, -1.9407e+00, -1.2397e+00, ..., 7.0679e-01,
1.5799e+00, 1.3559e+00],
[-1.7691e+00, -2.0652e+00, -1.5571e+00, ..., -8.3838e+00,
-4.6804e+00, 4.6281e-01],
[-8.7078e-01, -9.0810e-01, -1.1737e+00, ..., -5.0544e+00,
-7.3912e-01, 1.7387e+00]]],
[[[ 9.7953e-01, 1.6014e+00, 1.3481e+00, ..., -6.6894e-01,
5.4940e-01, 4.3590e-01],
[-8.0175e-01, -1.9871e+00, -7.1696e-01, ..., -7.0092e-01,
-7.3306e-01, -4.2437e-01],
[ 2.4331e-02, -6.3575e-01, -9.4508e-01, ..., -5.4188e-01,
-1.3212e+00, 5.8140e-01],
...,
[ 4.6272e-01, -2.8625e-01, -1.5520e+00, ..., 1.3444e-01,
-1.6882e-01, 3.5600e-01],
[-1.6051e-01, -9.8279e-01, -6.3699e-01, ..., 2.9654e-01,
1.7178e-02, -1.1862e-01],
[ 7.4216e-01, -3.0298e-01, -4.2622e-01, ..., -3.3505e-01,
-9.7767e-01, 1.4005e-01]],
[[-4.2391e+00, -1.1979e+00, -2.4845e+00, ..., -1.5245e+00,
-2.1540e+00, -1.8368e+00],
[-4.1147e+00, -2.8810e+00, -3.5969e+00, ..., -1.7565e+00,
-2.5677e+00, -2.1749e+00],
[-2.6125e+00, -1.9633e+00, -5.6903e+00, ..., -4.1176e+00,
-2.3616e+00, -1.3797e+00],
...,
[-8.6139e-01, -6.6133e-01, -2.7490e+00, ..., -3.3436e-01,
-9.0424e-01, -8.1445e-01],
[-2.8403e+00, -4.7666e-01, -1.4368e+00, ..., -1.6770e-01,
-1.0551e+00, -1.2042e+00],
[-2.0532e+00, -1.4362e+00, -1.6270e+00, ..., -7.3595e-01,
-6.8672e-01, -4.7594e-01]],
[[ 3.0430e+00, 4.1873e+00, 3.3518e+00, ..., 2.7023e+00,
6.7520e-01, 2.6349e+00],
[ 3.1319e-01, 8.1500e-01, 2.6394e+00, ..., 2.5068e+00,
3.7724e-01, 2.7101e+00],
[ 7.0516e-01, 1.3098e+00, 2.1564e+00, ..., 3.3474e-01,
1.1123e+00, 1.9718e+00],
...,
[ 2.3651e-01, 9.7953e-01, 2.6998e+00, ..., -1.6601e+00,
5.9456e-01, 1.5424e+00],
[ 2.7684e-01, 2.1298e+00, 1.9633e+00, ..., -2.9265e-01,
8.5708e-01, 1.1684e+00],
[ 1.3463e+00, 1.3779e+00, 1.6660e+00, ..., 2.2433e-01,
1.3167e+00, 1.3491e+00]],
...,
[[-2.0707e+00, -7.3829e+00, -6.0297e+00, ..., -3.9191e+00,
-2.9890e+00, -4.5027e+00],
[-2.6988e+00, -6.5025e+00, -4.8510e+00, ..., -1.0781e+01,
-4.9530e+00, -3.4904e+00],
[-1.4362e-01, -2.7635e+00, -3.5592e+00, ..., -1.2887e+01,
-5.2090e+00, -1.4031e+00],
...,
[-9.2758e-01, -2.2903e+00, -1.8751e+00, ..., -8.9650e+00,
-3.7652e+00, -2.8256e+00],
[-1.2784e+00, -3.7130e+00, -3.7740e+00, ..., -6.9659e+00,
-4.4730e+00, -4.3602e+00],
[-2.1733e+00, -4.9530e+00, -3.2459e+00, ..., -4.9711e+00,
-2.5784e+00, -2.4411e+00]],
[[-7.3556e-01, -2.1736e-02, -3.4193e-01, ..., 2.7117e+00,
1.9457e+00, 1.2186e+00],
[ 1.7697e+00, 1.0503e+00, 4.0163e-01, ..., 3.2168e+00,
1.8352e+00, 1.9263e+00],
[ 1.2004e+00, -1.4645e-01, -1.3035e+00, ..., 1.7913e+00,
2.2709e+00, 1.3425e+00],
...,
[ 1.2835e+00, -1.4741e-01, -2.7551e-02, ..., 2.8968e+00,
8.6955e-01, 1.0126e+00],
[ 1.8790e+00, 8.9682e-01, 9.7259e-01, ..., 2.4038e+00,
7.2011e-01, 1.5105e+00],
[ 1.0224e+00, 3.8500e-01, 9.7093e-01, ..., 1.1263e+00,
6.8483e-01, 1.4458e+00]],
[[-4.9914e-01, 6.4628e-01, -8.4005e-02, ..., -1.0691e+00,
-4.5575e-01, -4.1772e-02],
[-1.0129e+00, -1.1435e+00, -2.4675e+00, ..., -2.0274e+00,
-1.1041e+00, -1.1367e+00],
[-2.0002e+00, -8.4579e-01, -1.6641e+00, ..., -3.5917e+00,
-1.9251e+00, -2.8979e-01],
...,
[-1.7001e+00, -1.4832e+00, -1.4184e+00, ..., -2.1747e+00,
-1.3625e+00, -8.2572e-01],
[-1.7263e+00, -1.1037e+00, -9.1065e-01, ..., -8.0112e-01,
-1.6017e+00, -6.9745e-01],
[-3.4000e-01, 4.0755e-01, -2.4685e-01, ..., -3.7186e-03,
-8.1613e-01, -1.7141e-02]]],
[[[ 5.8955e-01, 8.5924e-01, 5.6558e-01, ..., 3.2611e-01,
4.2149e-01, 3.9617e-02],
[-2.4375e-01, -8.6534e-01, -4.6902e-01, ..., 2.9280e-01,
1.0032e-01, -3.6278e-01],
[-1.8269e-01, -6.8746e-02, 6.4651e-02, ..., -1.2460e-01,
2.5171e-02, 7.0244e-01],
...,
[ 4.6782e-01, -1.3636e-01, -1.8150e-02, ..., 2.9841e-01,
8.3655e-01, 7.0393e-01],
[-1.0098e-02, -3.8174e-01, 9.2903e-02, ..., 9.4775e-01,
1.6208e+00, 1.0771e-01],
[ 7.9431e-01, -8.0310e-02, -9.5454e-02, ..., 4.6909e-01,
2.9262e-01, 7.3302e-01]],
[[-2.3023e+00, -6.3458e-01, -1.9455e+00, ..., -1.5587e+00,
-2.2359e+00, -1.5426e+00],
[-2.3953e+00, -7.3131e-01, -1.9067e+00, ..., -1.1468e+00,
-2.4850e+00, -1.7677e+00],
[-2.3648e+00, -1.6738e+00, -2.6216e+00, ..., -1.9942e+00,
-2.3688e+00, -1.1822e+00],
...,
[-4.4718e-01, -8.5460e-01, -9.6627e-01, ..., -5.4754e-01,
-1.4234e-01, 7.8294e-01],
[-1.9369e+00, -2.9726e-01, -4.1284e-01, ..., 3.7126e-02,
6.2355e-02, 7.5943e-01],
[-1.2000e+00, -1.0754e+00, -1.0084e+00, ..., -5.1619e-01,
-4.6314e-01, -1.3317e-01]],
[[ 1.8712e+00, 2.4966e+00, 1.6365e+00, ..., 1.6092e+00,
8.4518e-01, 2.1341e+00],
[-2.5542e-01, 3.6730e-01, 5.5118e-01, ..., 8.5807e-01,
3.3913e-01, 2.2482e+00],
[ 4.7866e-02, 3.8220e-01, 7.6386e-01, ..., -1.8538e-01,
7.7290e-01, 1.9736e+00],
...,
[ 4.8086e-01, 3.7664e-01, 1.5091e+00, ..., 2.3780e-01,
8.6001e-02, 1.6366e+00],
[ 7.3349e-01, 1.9851e+00, 1.7964e+00, ..., 9.4586e-01,
7.6149e-01, 1.7171e+00],
[ 2.0376e+00, 1.9597e+00, 2.4530e+00, ..., 2.1828e+00,
1.0686e+00, 1.7882e+00]],
...,
[[-1.9025e+00, -4.7564e+00, -3.1442e+00, ..., -1.8937e+00,
-1.6438e+00, -3.0926e+00],
[-2.3707e+00, -5.1298e+00, -2.9494e+00, ..., -2.3708e+00,
-2.3464e+00, -3.0023e+00],
[-1.5571e+00, -5.4366e+00, -3.3338e+00, ..., -2.2254e+00,
-2.5126e+00, -2.3615e+00],
...,
[-2.1275e+00, -3.8956e+00, -2.3729e+00, ..., -3.5949e+00,
-2.1501e+00, -2.7263e+00],
[-2.0357e+00, -4.7980e+00, -3.9560e+00, ..., -4.3417e+00,
-3.2445e+00, -5.0350e+00],
[-2.1052e+00, -5.2166e+00, -2.8061e+00, ..., -2.7359e+00,
-1.5558e+00, -3.4873e+00]],
[[ 9.2640e-01, 1.5869e+00, 7.9180e-01, ..., 1.0743e+00,
1.5337e+00, 1.4225e+00],
[ 1.1446e+00, 1.0005e+00, 8.8962e-01, ..., 9.4968e-01,
7.5062e-01, 1.9604e+00],
[ 6.1484e-01, -2.1169e-01, 9.4833e-02, ..., 4.0947e-01,
1.2464e-01, 1.3985e+00],
...,
[ 1.3401e+00, 3.4154e-03, 5.0771e-01, ..., 1.7179e+00,
1.8726e+00, 1.4656e+00],
[ 1.9428e+00, 1.0243e+00, 1.8727e+00, ..., 1.9773e+00,
1.7266e+00, 2.5168e+00],
[ 1.2731e+00, 6.9366e-01, 1.1572e+00, ..., 1.3926e+00,
1.6172e+00, 1.5333e+00]],
[[-8.5513e-02, -6.2557e-01, -1.1837e+00, ..., -4.8391e-01,
-8.5581e-01, -9.0331e-02],
[-8.2729e-01, -2.5494e+00, -2.5802e+00, ..., -9.7821e-01,
-1.0427e+00, -7.0761e-01],
[-1.4388e+00, -2.0245e+00, -1.1233e+00, ..., -2.8194e-01,
-3.9827e-01, -2.5634e-01],
...,
[-1.5372e+00, -1.4578e+00, -5.7231e-01, ..., -1.2380e-01,
-4.1795e-01, -4.4114e-01],
[-1.5720e+00, -1.0647e+00, -6.6037e-01, ..., -5.0882e-01,
-4.3806e-01, 3.9884e-01],
[-4.4719e-02, 5.1738e-01, -2.1754e-01, ..., -1.0110e-01,
2.9256e-02, 6.4917e-01]]]], device='cuda:0'),
tensor([[[[ 8.5375e-01, 1.5117e+00, 1.1986e+00, ..., 1.3561e+00,
1.0818e+00, 9.4117e-01],
[-1.0302e+00, -2.0122e+00, -1.1584e+00, ..., -5.5612e-01,
-5.9744e-01, -1.5210e+00],
[-1.3672e-01, -7.2506e-01, 3.4249e-02, ..., -6.0704e-01,
6.3629e-01, 8.0519e-01],
...,
[ 4.8638e-01, -1.6075e-01, 5.3725e-01, ..., 1.8773e-01,
2.2903e-01, 6.5561e-01],
[ 3.6432e-01, -6.4072e-01, -2.4672e-01, ..., 5.5706e-01,
6.3999e-01, -5.5071e-02],
[ 5.8905e-01, 3.2006e-01, -3.0790e-01, ..., 3.4916e-01,
-5.8625e-02, 6.5542e-01]],
[[-3.8030e+00, -7.8181e-01, -1.5830e+00, ..., -1.6508e+00,
-2.5632e+00, -1.2538e+00],
[-4.0471e+00, -2.8521e+00, -1.7924e+00, ..., -2.0196e+00,
-3.2482e+00, -9.3142e-01],
[-2.5411e+00, -1.7252e+00, -1.5800e+00, ..., -1.7360e+00,
-2.6831e+00, -1.0271e+00],
...,
[-9.5747e-01, -1.0948e+00, -1.5271e+00, ..., -1.3764e+00,
-1.2977e+00, -1.5301e-01],
[-1.3489e+00, -4.0964e-01, -9.3139e-01, ..., -4.9049e-01,
-1.1255e+00, -9.3140e-02],
[ 4.7016e-01, -1.9304e-01, -6.3370e-01, ..., 3.7629e-01,
7.8269e-02, 3.8794e-01]],
[[ 2.9332e+00, 4.2844e+00, 3.3822e+00, ..., 2.7960e+00,
9.4318e-01, 3.0131e+00],
[ 2.4065e-01, 1.1051e+00, 2.2777e+00, ..., 1.4229e+00,
-1.0227e+00, 2.7432e+00],
[ 9.9088e-01, 1.5613e+00, 1.6997e+00, ..., 1.5273e+00,
2.2001e-01, 2.0511e+00],
...,
[ 5.1283e-01, 4.4073e-01, 1.1499e+00, ..., -2.4009e-01,
2.9019e-01, 1.4765e+00],
[ 2.0690e-01, 1.0451e+00, 1.3854e+00, ..., -1.8269e-01,
1.8004e-01, 1.1802e+00],
[ 1.3611e+00, 8.4420e-01, 1.8422e+00, ..., 6.3519e-01,
4.2609e-01, 1.1431e+00]],
...,
[[-1.9152e+00, -7.4150e+00, -5.5421e+00, ..., -3.9932e+00,
-3.2398e+00, -5.5559e+00],
[-2.4409e+00, -7.1090e+00, -4.3775e+00, ..., -2.9035e+00,
-2.7815e+00, -4.4777e+00],
[-1.0822e+00, -4.7277e+00, -3.8340e+00, ..., -2.2526e+00,
-3.2840e+00, -2.3992e+00],
...,
[-2.2861e+00, -5.6397e+00, -4.3270e+00, ..., -4.6240e+00,
-3.4992e+00, -3.4607e+00],
[-2.4563e+00, -6.6427e+00, -5.1731e+00, ..., -5.9318e+00,
-4.2671e+00, -4.8216e+00],
[-2.3820e+00, -6.6167e+00, -4.2522e+00, ..., -4.2476e+00,
-3.2075e+00, -4.1337e+00]],
[[-2.2111e-01, 3.0102e-01, 5.8644e-01, ..., 5.1855e-01,
7.8256e-01, 1.3945e+00],
[ 1.6544e+00, 1.0790e+00, 1.8517e+00, ..., 1.8361e+00,
2.4687e+00, 3.0174e+00],
[ 1.1812e+00, 2.4748e-01, 8.3017e-01, ..., 1.1038e+00,
2.1095e+00, 2.3759e+00],
...,
[ 1.1368e+00, -8.5203e-03, -2.1394e-01, ..., 1.4685e+00,
7.9795e-01, 1.2785e+00],
[ 1.6649e+00, 8.6024e-01, 7.4726e-01, ..., 1.3646e+00,
6.3450e-01, 2.0754e+00],
[ 1.1622e+00, 8.0443e-01, 1.0712e+00, ..., 1.3498e+00,
1.2464e+00, 1.3911e+00]],
[[-5.6738e-01, 2.9277e-01, 1.3357e-01, ..., 4.1768e-01,
5.0389e-01, 1.3566e+00],
[-1.3528e+00, -1.7909e+00, -1.5729e+00, ..., -1.0012e+00,
-5.3151e-01, 2.8059e-01],
[-2.2205e+00, -7.4675e-01, 3.5889e-01, ..., -4.8575e-01,
3.0766e-01, 7.6270e-01],
...,
[-1.5407e+00, -1.6963e+00, -1.0023e+00, ..., -1.0491e+00,
-1.0516e+00, -7.3564e-01],
[-1.2664e+00, -1.6319e+00, -1.1074e+00, ..., -4.4425e-01,
-1.2067e+00, -3.1497e-02],
[ 1.2508e-01, 7.3583e-02, 3.4999e-02, ..., 6.2753e-02,
-4.5039e-02, 7.7455e-01]]],
[[[ 1.0609e+00, 1.7394e+00, 1.1236e+00, ..., 7.9172e-01,
9.8338e-01, 4.4201e-01],
[-9.1331e-01, -2.0819e+00, -8.7767e-01, ..., -2.3707e-01,
-1.7458e-01, -4.1352e-01],
[-2.6598e-01, -1.1909e+00, -8.3557e-04, ..., -6.8134e-01,
-1.2086e-01, 6.3310e-01],
...,
[ 4.5802e-01, -3.3136e-01, -1.8651e-01, ..., -2.3751e-02,
-6.1161e-02, 4.4498e-01],
[-1.4226e-01, -9.9396e-01, -8.1456e-02, ..., -6.6785e-02,
1.1178e-01, -6.5510e-02],
[ 7.5569e-01, -2.8728e-01, -2.4738e-01, ..., -3.6164e-01,
-8.7771e-01, 1.3023e-01]],
[[-4.2743e+00, -1.7605e+00, -2.4146e+00, ..., -1.4846e+00,
-2.1439e+00, -1.8522e+00],
[-4.2275e+00, -3.6634e+00, -2.6976e+00, ..., -1.0486e+00,
-2.2136e+00, -2.1307e+00],
[-2.9245e+00, -2.0863e+00, -6.1650e-01, ..., -1.3764e+00,
-1.6595e+00, -1.3837e+00],
...,
[-8.1146e-01, -5.0840e-01, -1.2404e+00, ..., -7.7231e-01,
-6.3510e-01, -7.9436e-01],
[-2.7790e+00, -4.2326e-01, -9.0678e-01, ..., -6.2149e-01,
-9.0001e-01, -1.1490e+00],
[-2.0112e+00, -1.4934e+00, -1.4975e+00, ..., -6.6937e-01,
-6.2965e-01, -4.8431e-01]],
[[ 3.0649e+00, 4.2248e+00, 3.3675e+00, ..., 1.7721e+00,
8.9098e-01, 2.6090e+00],
[ 6.9164e-01, 1.2689e+00, 2.3374e+00, ..., 1.4466e+00,
7.3408e-01, 2.6773e+00],
[ 1.3338e+00, 1.1917e+00, 6.0082e-01, ..., 3.7981e-01,
1.0259e+00, 1.8593e+00],
...,
[ 3.0237e-01, 8.9262e-01, 1.0656e+00, ..., -2.1844e-01,
4.2426e-01, 1.3970e+00],
[ 3.6081e-01, 2.0179e+00, 1.4219e+00, ..., -2.3200e-01,
6.5964e-01, 1.0890e+00],
[ 1.4669e+00, 1.2728e+00, 1.7231e+00, ..., 5.7126e-01,
1.2840e+00, 1.3129e+00]],
...,
[[-2.0693e+00, -7.6020e+00, -6.1779e+00, ..., -5.0816e+00,
-3.6846e+00, -4.3720e+00],
[-2.8070e+00, -6.9073e+00, -4.6420e+00, ..., -5.0243e+00,
-3.8788e+00, -3.3205e+00],
[-2.9919e-01, -3.0272e+00, -3.0158e+00, ..., -3.2633e+00,
-2.4304e+00, -1.4238e+00],
...,
[-9.5361e-01, -2.2369e+00, -1.6100e+00, ..., -4.8566e+00,
-2.9567e+00, -2.7092e+00],
[-1.3202e+00, -3.7190e+00, -3.3364e+00, ..., -5.1395e+00,
-4.0007e+00, -4.1771e+00],
[-2.2182e+00, -4.9594e+00, -2.8903e+00, ..., -3.4003e+00,
-2.3950e+00, -2.2507e+00]],
[[-8.6036e-01, -3.0558e-01, 1.6467e-01, ..., 1.4226e+00,
1.6692e+00, 1.2030e+00],
[ 1.6062e+00, 4.8580e-01, 5.8026e-01, ..., 1.5219e+00,
1.2386e+00, 1.9038e+00],
[ 9.8297e-01, 3.8749e-01, 1.1173e+00, ..., 1.3266e+00,
5.2786e-01, 1.2902e+00],
...,
[ 1.3117e+00, -1.1752e-01, 9.3852e-01, ..., 1.6532e+00,
7.3389e-01, 8.4237e-01],
[ 1.9041e+00, 9.0289e-01, 1.6915e+00, ..., 1.2907e+00,
7.6186e-01, 1.4015e+00],
[ 1.0391e+00, 3.7182e-01, 9.0057e-01, ..., 7.5137e-01,
6.4414e-01, 1.3996e+00]],
[[-5.9885e-01, 6.4913e-01, 3.7516e-01, ..., -3.6124e-01,
-6.5550e-01, -7.3979e-02],
[-1.2518e+00, -1.2520e+00, -1.0487e+00, ..., -1.1330e+00,
-1.4191e+00, -1.1255e+00],
[-2.3028e+00, -6.9249e-01, 1.1548e+00, ..., -5.7867e-01,
-1.0784e+00, -2.6233e-01],
...,
[-1.7575e+00, -1.4643e+00, -1.0572e+00, ..., -5.5899e-01,
-1.1583e+00, -8.7860e-01],
[-1.7313e+00, -1.1039e+00, -7.9004e-01, ..., -8.2597e-01,
-1.6250e+00, -7.3623e-01],
[-3.4959e-01, 3.5120e-01, -2.8325e-01, ..., -2.9418e-01,
-8.9239e-01, -5.4089e-03]]],
[[[ 1.0746e+00, 1.0539e+00, 6.3371e-01, ..., 1.1505e+00,
9.2833e-01, 9.0476e-01],
[-2.9286e-01, -1.8283e+00, -5.4624e-01, ..., -3.7746e-01,
-9.1308e-01, -1.7130e+00],
[ 1.3185e-01, -4.0309e-01, 6.6382e-02, ..., 1.3278e+00,
1.1582e+00, 8.0922e-01],
...,
[-3.0978e-02, -1.1687e+00, -1.3490e+00, ..., -2.6449e-01,
1.1443e-02, 2.3878e-01],
[-6.0885e-01, -1.7810e+00, -1.6496e+00, ..., -5.3258e-01,
-3.1261e-01, -1.6237e-01],
[-3.5699e-01, -1.7110e+00, -1.7641e+00, ..., -1.1511e+00,
-1.5541e+00, -2.5928e-01]],
[[-3.4355e+00, -1.9053e-01, 5.0310e-01, ..., -1.4110e+00,
-2.2697e+00, -9.5485e-01],
[-3.4633e+00, -7.7227e-01, 9.7863e-01, ..., -2.7067e+00,
-3.3394e+00, -7.6761e-01],
[-1.9857e+00, -6.7898e-01, 6.4604e-01, ..., -2.5889e+00,
-2.8898e+00, -9.7374e-01],
...,
[-1.6107e+00, -1.6447e+00, -2.5273e+00, ..., -1.9188e+00,
-2.1460e+00, -6.9342e-01],
[-3.0344e+00, -1.6194e+00, -2.9440e+00, ..., -2.2430e+00,
-2.4654e+00, -1.2678e+00],
[-2.3153e+00, -1.9389e+00, -2.8753e+00, ..., -2.0027e+00,
-1.9485e+00, -9.9954e-01]],
[[ 3.6024e+00, 3.7214e+00, 1.1010e+00, ..., 2.9985e+00,
1.0093e+00, 2.7954e+00],
[ 1.4994e+00, 5.0297e-01, 5.5294e-01, ..., 1.5669e+00,
-2.1938e-01, 2.6813e+00],
[ 2.2074e+00, 1.2363e+00, -1.7545e-01, ..., -1.3144e-01,
4.1779e-01, 2.1100e+00],
...,
[ 9.5557e-01, 1.1270e+00, 2.8764e+00, ..., 1.5082e-01,
1.0963e+00, 2.3474e+00],
[ 6.9000e-01, 1.8307e+00, 2.2761e+00, ..., 3.0773e-02,
1.1008e+00, 1.8508e+00],
[ 1.6457e+00, 1.5956e+00, 2.2210e+00, ..., 1.3402e+00,
1.8452e+00, 1.8026e+00]],
...,
[[-2.8601e+00, -8.6149e+00, -4.5983e+00, ..., -3.9529e+00,
-3.0092e+00, -5.2945e+00],
[-3.2600e+00, -9.3756e+00, -3.3605e+00, ..., -2.9364e+00,
-2.0574e+00, -3.9900e+00],
[-1.5561e+00, -5.0481e+00, -2.1868e+00, ..., -3.4765e+00,
-3.1780e+00, -2.7661e+00],
...,
[-2.5318e+00, -5.2857e+00, -4.4874e+00, ..., -5.9550e+00,
-4.6918e+00, -4.6331e+00],
[-2.6532e+00, -5.6325e+00, -5.6802e+00, ..., -6.7089e+00,
-5.7421e+00, -6.2244e+00],
[-1.4124e+00, -3.9608e+00, -3.0246e+00, ..., -3.3501e+00,
-2.7369e+00, -3.2895e+00]],
[[-3.2816e-01, 6.5962e-01, 1.4398e+00, ..., 7.7778e-01,
9.7957e-01, 1.4997e+00],
[ 1.7687e+00, 1.0410e+00, 1.0930e+00, ..., 1.4778e+00,
2.5120e+00, 3.1038e+00],
[ 1.1535e+00, 7.9095e-01, 4.5847e-01, ..., -1.1450e-01,
1.6002e+00, 2.6014e+00],
...,
[ 1.2788e+00, 2.1786e-01, -6.6766e-01, ..., 1.7939e+00,
1.0913e+00, 1.6679e+00],
[ 1.8741e+00, 1.6250e+00, 4.3252e-01, ..., 1.6213e+00,
1.2548e+00, 1.9864e+00],
[ 4.5142e-01, 7.9236e-01, 3.7551e-01, ..., 5.1642e-01,
7.2348e-01, 1.6743e+00]],
[[-8.7159e-01, -2.9603e-01, 3.0254e-01, ..., 3.3157e-01,
3.5877e-01, 1.3411e+00],
[-1.5448e+00, -1.9467e+00, -2.2319e-01, ..., -5.8526e-01,
-6.2423e-01, 2.4046e-01],
[-1.8155e+00, -1.2956e+00, 2.5588e-01, ..., 1.5979e+00,
2.0638e-01, 8.5033e-01],
...,
[-1.9164e+00, -1.8186e+00, -1.3776e+00, ..., -1.3270e+00,
-1.0682e+00, -8.3320e-01],
[-1.5804e+00, -1.9350e+00, -1.9664e+00, ..., -9.2130e-01,
-1.3752e+00, -7.2505e-01],
[-8.5349e-01, -8.0400e-01, -1.2020e+00, ..., -5.5420e-01,
-8.2785e-01, -9.8017e-02]]],
[[[ 6.3901e-01, 1.2325e+00, 4.5086e-01, ..., 8.2635e-01,
1.0128e+00, 4.6701e-01],
[-1.1560e+00, -2.2705e+00, -3.9563e-01, ..., -2.6621e-01,
-1.6459e-01, -3.5651e-01],
[ 2.6459e-01, -2.5938e+00, -1.6411e+00, ..., -6.1797e-01,
-1.6971e-01, 6.3095e-01],
...,
[ 3.6390e-01, -3.3103e-01, 2.7078e-01, ..., -9.5305e-01,
-5.1586e-01, 6.7438e-01],
[-1.8829e-01, -1.0885e+00, -1.8226e-01, ..., 3.1413e-01,
1.8898e-02, -5.8577e-02],
[ 7.4176e-01, -2.9132e-01, -2.1557e-01, ..., -2.8776e-01,
-8.5345e-01, 1.8680e-01]],
[[-4.9305e+00, -2.6877e+00, -4.9011e+00, ..., -1.4543e+00,
-2.0731e+00, -1.8380e+00],
[-3.8458e+00, -7.6470e-01, -6.6425e-01, ..., -1.0643e+00,
-2.1347e+00, -2.1526e+00],
[-3.1030e+00, 2.3313e-01, -1.3625e+00, ..., -1.3597e+00,
-1.6268e+00, -1.4312e+00],
...,
[-7.7107e-01, -6.3724e-01, -1.1026e+00, ..., -1.6792e+00,
-1.2912e+00, -5.8403e-01],
[-2.8294e+00, -3.8193e-01, -7.6130e-01, ..., -8.3911e-01,
-1.2500e+00, -1.1118e+00],
[-2.1106e+00, -1.5143e+00, -1.3562e+00, ..., -6.2824e-01,
-7.2361e-01, -5.5392e-01]],
[[ 3.5670e+00, 4.2474e+00, 3.1152e+00, ..., 1.7246e+00,
8.5084e-01, 2.5657e+00],
[ 4.1215e-01, -1.8423e-01, -7.2615e-01, ..., 1.4998e+00,
6.9086e-01, 2.6188e+00],
[ 2.7045e-01, -4.4779e-01, -8.3710e-01, ..., 5.9944e-01,
1.0215e+00, 1.8258e+00],
...,
[ 2.6625e-01, 9.6364e-01, 1.3600e+00, ..., -1.8518e+00,
5.6624e-01, 1.2233e+00],
[ 3.0721e-01, 2.0216e+00, 1.5043e+00, ..., 6.8649e-02,
7.6421e-01, 1.0024e+00],
[ 1.4384e+00, 1.3845e+00, 1.7161e+00, ..., 6.0366e-01,
1.3638e+00, 1.3909e+00]],
...,
[[-2.6022e+00, -7.3092e+00, -4.8710e+00, ..., -4.9684e+00,
-3.6906e+00, -4.3531e+00],
[-3.6416e+00, -7.7929e+00, -6.1332e+00, ..., -4.8233e+00,
-3.8232e+00, -3.3027e+00],
[-2.2762e+00, -3.7204e+00, -2.7214e+00, ..., -3.1643e+00,
-2.3947e+00, -1.4340e+00],
...,
[-9.7052e-01, -2.2376e+00, -1.6250e+00, ..., -5.2569e+00,
-4.8571e+00, -3.6350e+00],
[-1.3051e+00, -3.7923e+00, -3.3764e+00, ..., -7.7472e+00,
-5.7831e+00, -4.9009e+00],
[-2.2079e+00, -4.9290e+00, -2.9985e+00, ..., -3.4083e+00,
-2.4695e+00, -2.4431e+00]],
[[-1.2671e+00, -2.1215e+00, -3.0760e+00, ..., 1.4473e+00,
1.6543e+00, 1.1755e+00],
[ 2.4638e+00, 1.8792e+00, 7.5279e-01, ..., 1.6009e+00,
1.1776e+00, 1.8493e+00],
[ 1.0947e+00, 1.1195e+00, 1.1646e+00, ..., 1.4156e+00,
5.8763e-01, 1.2735e+00],
...,
[ 1.3288e+00, -1.6411e-01, 9.5376e-01, ..., 1.2381e+00,
1.5045e+00, 1.5621e+00],
[ 1.8980e+00, 9.4553e-01, 1.7879e+00, ..., 1.9491e+00,
1.1905e+00, 1.5346e+00],
[ 1.0102e+00, 3.5358e-01, 9.3240e-01, ..., 7.6866e-01,
6.5842e-01, 1.4315e+00]],
[[-1.1039e+00, 2.7031e-01, -3.0453e-01, ..., -3.2087e-01,
-6.1224e-01, -7.1621e-02],
[-1.5109e+00, 2.3502e-01, -8.4955e-02, ..., -1.1778e+00,
-1.4127e+00, -1.1367e+00],
[-1.5581e+00, 4.7041e-01, -1.6462e-01, ..., -6.9730e-01,
-1.0662e+00, -2.4712e-01],
...,
[-1.7279e+00, -1.5238e+00, -7.6752e-01, ..., -2.2254e+00,
-2.1837e+00, -9.2995e-01],
[-1.7510e+00, -1.1136e+00, -9.4001e-01, ..., -2.5900e-01,
-1.2487e+00, -6.5279e-01],
[-3.8519e-01, 3.7541e-01, -3.3804e-01, ..., -2.5528e-01,
-7.7938e-01, 9.3033e-03]]],
[[[ 7.3176e-01, 1.1118e+00, 7.7326e-01, ..., 1.5070e-01,
3.1913e-01, 2.1151e-01],
[-2.8589e-01, -1.3199e+00, -5.1456e-01, ..., 5.4089e-01,
3.8240e-01, -1.1433e+00],
[ 1.9561e-01, -7.1871e-01, -3.6823e-01, ..., -1.1108e-01,
1.0961e+00, 6.2811e-01],
...,
[ 1.8004e-01, -1.0045e+00, -1.4686e+00, ..., 2.3632e-02,
9.5753e-01, 5.4524e-01],
[-3.7812e-01, -1.7694e+00, -1.7831e+00, ..., 2.8112e-01,
1.3854e+00, -1.7396e-01],
[ 3.8754e-01, -4.8656e-01, -9.8331e-01, ..., -2.3247e-01,
4.2635e-01, 5.8615e-01]],
[[-2.8570e+00, -1.5729e+00, -3.0781e+00, ..., -1.8385e+00,
-2.1700e+00, -1.0757e+00],
[-4.0640e+00, -1.8584e+00, -2.3623e+00, ..., -9.7295e-01,
-1.3203e+00, -2.7368e-01],
[-4.0167e+00, -2.5837e+00, -3.0656e+00, ..., -1.8304e+00,
-1.4050e+00, -7.4453e-01],
...,
[-2.5554e+00, -1.5800e+00, -4.5569e+00, ..., 1.1144e-01,
-1.1561e-02, 4.8242e-01],
[-3.4875e+00, -1.2713e+00, -3.8533e+00, ..., -9.8228e-03,
-4.0613e-01, 2.7332e-01],
[-1.1336e+00, -7.7514e-01, -2.6352e+00, ..., -3.3513e-01,
-7.1865e-01, -5.5248e-01]],
[[ 1.5472e+00, 2.7454e+00, 2.4040e+00, ..., 3.0667e+00,
1.4293e+00, 2.8592e+00],
[ 4.2061e-02, 1.2020e+00, 1.5570e+00, ..., 1.6080e+00,
-2.2223e-02, 2.5013e+00],
[ 5.8477e-01, 1.4280e+00, 9.0273e-01, ..., 9.7547e-01,
7.0134e-01, 1.8356e+00],
...,
[ 9.4477e-01, 1.1825e+00, 1.9243e+00, ..., 4.3292e-01,
-6.5120e-03, 1.3345e+00],
[ 6.8072e-01, 1.8052e+00, 1.9061e+00, ..., 8.5732e-01,
6.0930e-01, 1.4703e+00],
[ 1.5152e+00, 1.4802e+00, 2.3116e+00, ..., 1.4133e+00,
1.1165e+00, 1.9644e+00]],
...,
[[-1.1907e+00, -4.3275e+00, -3.8780e+00, ..., -2.7909e+00,
-2.3643e+00, -3.9792e+00],
[-1.5351e+00, -3.5236e+00, -4.1401e+00, ..., -5.4614e+00,
-3.9822e+00, -4.4348e+00],
[-8.8834e-01, -3.0857e+00, -3.7473e+00, ..., -3.2971e+00,
-2.4132e+00, -1.6157e+00],
...,
[-1.5881e+00, -3.9292e+00, -4.4744e+00, ..., -4.4796e+00,
-2.2321e+00, -1.5038e+00],
[-1.4407e+00, -4.2498e+00, -5.2678e+00, ..., -5.3851e+00,
-3.1752e+00, -3.6491e+00],
[-1.5771e+00, -4.7443e+00, -4.2791e+00, ..., -3.6553e+00,
-2.1750e+00, -2.7805e+00]],
[[ 8.0254e-01, 4.7286e-01, -9.2408e-02, ..., 1.3846e+00,
1.1664e+00, 1.2412e+00],
[ 1.1818e+00, 8.8890e-01, 5.7982e-01, ..., 1.8820e+00,
2.4130e+00, 2.5748e+00],
[ 5.6498e-01, 4.5866e-01, 6.5915e-02, ..., 1.4699e+00,
1.7542e+00, 1.9301e+00],
...,
[ 9.2178e-01, -2.1299e-02, -1.2409e+00, ..., 1.9559e+00,
1.7387e+00, 1.5951e+00],
[ 1.5460e+00, 1.3895e+00, 4.3648e-01, ..., 1.4705e+00,
1.4483e+00, 2.4343e+00],
[ 7.6631e-01, 8.8532e-01, 2.5150e-01, ..., 1.0877e+00,
1.3042e+00, 1.3709e+00]],
[[-2.7163e-01, -1.2671e-01, -9.7898e-01, ..., -2.0264e-01,
-9.3786e-02, 8.7827e-01],
[-1.3527e+00, -1.4146e+00, -1.9716e+00, ..., 2.7424e-01,
-4.1255e-02, 3.8913e-01],
[-1.5782e+00, -8.3675e-01, -1.0010e+00, ..., 7.3154e-02,
2.0383e-01, 2.2937e-01],
...,
[-1.6782e+00, -1.5982e+00, -2.3382e+00, ..., -5.8779e-01,
-4.4631e-01, -3.0436e-01],
[-1.3773e+00, -1.6279e+00, -2.2729e+00, ..., -5.5717e-01,
-5.8605e-01, 3.1054e-01],
[-1.8490e-01, 1.4455e-02, -7.9094e-01, ..., 2.3730e-01,
1.1076e-01, 3.5331e-01]]]], device='cuda:0'),
tensor([[[[ 7.2465e-01, 1.1111e+00, 5.9070e-01, ..., 7.4815e-01,
1.4002e+00, 5.7624e-01],
[-3.0407e-01, -1.3645e+00, -6.5075e-01, ..., 2.9323e-01,
1.1086e+00, -4.9910e-01],
[ 1.2336e-01, -9.8136e-01, -7.6666e-01, ..., 1.1869e-01,
5.2147e-02, 7.5990e-01],
...,
[ 1.1149e-01, -1.1517e+00, -9.7658e-01, ..., -9.7127e-01,
4.1771e-01, 6.3871e-01],
[-4.4229e-01, -1.8537e+00, -9.1297e-01, ..., 1.1278e-01,
8.7587e-01, 1.4970e-01],
[ 3.8411e-01, -5.0859e-01, -7.7309e-01, ..., -1.2463e-01,
-3.5503e-01, 5.1097e-01]],
[[-3.0273e+00, -1.5789e+00, -3.3575e+00, ..., -2.0910e+00,
-1.1611e+00, -6.9681e-02],
[-4.3345e+00, -1.9822e+00, -3.0229e+00, ..., -2.5734e+00,
-5.1310e-01, -2.8243e-01],
[-4.2120e+00, -3.1162e+00, -3.5507e+00, ..., -3.9163e+00,
-3.8452e-01, -2.5752e-01],
...,
[-2.8683e+00, -1.7157e+00, -3.0363e+00, ..., -1.8363e+00,
-1.4237e+00, 3.7689e-01],
[-3.6857e+00, -1.3665e+00, -2.3502e+00, ..., -3.7007e-01,
-8.6518e-01, 2.3949e-01],
[-1.2972e+00, -8.7250e-01, -1.6791e+00, ..., -7.5916e-01,
-3.3598e-01, 8.1777e-02]],
[[ 1.7296e+00, 2.8088e+00, 2.3383e+00, ..., 2.1854e+00,
4.8806e-01, 1.2920e+00],
[ 1.8981e-01, 1.4962e+00, 1.7395e+00, ..., 4.5805e-01,
-1.4326e+00, 3.5328e-01],
[ 6.5748e-01, 1.7940e+00, 6.1714e-01, ..., -1.3145e+00,
-1.6965e+00, 3.5506e-01],
...,
[ 9.2074e-01, 1.3044e+00, 9.4061e-01, ..., -1.2697e+00,
-2.9192e-02, 1.1895e+00],
[ 6.8927e-01, 1.8752e+00, 9.6615e-01, ..., -8.9301e-01,
3.5109e-01, 1.0637e+00],
[ 1.5457e+00, 1.6392e+00, 1.6072e+00, ..., 1.3320e-01,
5.9585e-01, 1.2902e+00]],
...,
[[-1.4496e+00, -4.6660e+00, -4.0899e+00, ..., -5.0587e+00,
-3.4535e+00, -4.0539e+00],
[-1.8189e+00, -3.8429e+00, -4.6731e+00, ..., -6.6217e+00,
-4.9600e+00, -4.8068e+00],
[-9.0798e-01, -2.8665e+00, -4.5867e+00, ..., -6.1654e+00,
-4.1082e+00, -3.5735e+00],
...,
[-1.5671e+00, -3.9844e+00, -4.6732e+00, ..., -7.9422e+00,
-2.5927e+00, -3.1610e+00],
[-1.4640e+00, -4.5470e+00, -5.4204e+00, ..., -9.5190e+00,
-3.7613e+00, -4.8728e+00],
[-1.5969e+00, -4.8960e+00, -4.3012e+00, ..., -6.2085e+00,
-2.0714e+00, -3.1974e+00]],
[[ 8.0090e-01, 2.9404e-01, -2.3664e-01, ..., 8.5147e-01,
1.3682e+00, 1.4759e+00],
[ 1.2179e+00, 7.4931e-01, -2.0414e-02, ..., 2.2588e+00,
2.0853e+00, 1.4646e+00],
[ 6.9652e-01, 3.3284e-01, 8.6880e-01, ..., 2.1099e+00,
1.5808e+00, 7.8702e-01],
...,
[ 8.5925e-01, 2.5959e-02, -5.6943e-01, ..., 3.7377e+00,
1.4829e+00, 1.7814e+00],
[ 1.5406e+00, 1.3846e+00, 1.2101e+00, ..., 3.8228e+00,
1.3459e+00, 2.4528e+00],
[ 7.2169e-01, 9.1576e-01, 9.4681e-01, ..., 1.8870e+00,
1.1695e+00, 1.3722e+00]],
[[-2.8051e-01, -2.7264e-01, -9.3419e-01, ..., 6.0718e-01,
1.2494e+00, 1.0203e+00],
[-1.4309e+00, -1.5310e+00, -1.9992e+00, ..., -5.6490e-01,
1.1816e+00, -1.2520e+00],
[-1.7316e+00, -1.2311e+00, -1.2100e+00, ..., -7.8530e-01,
1.0685e+00, -1.7646e+00],
...,
[-1.7867e+00, -1.5514e+00, -1.8459e+00, ..., -2.4525e+00,
-9.1870e-01, -6.6626e-01],
[-1.4562e+00, -1.6044e+00, -1.6896e+00, ..., -1.5016e+00,
-8.3436e-01, 2.2329e-02],
[-1.9000e-01, 4.3676e-02, -5.8994e-01, ..., -1.4405e+00,
-6.7300e-01, 2.1786e-01]]],
[[[ 9.2300e-01, 1.5659e+00, 1.3430e+00, ..., 1.3410e+00,
6.8998e-01, 3.3963e-01],
[-9.0888e-01, -1.9902e+00, -1.1109e+00, ..., 2.2871e+00,
6.3634e-01, -4.5013e-01],
[ 4.8402e-02, -6.4907e-01, -6.8085e-01, ..., -2.0108e-01,
-7.4485e-02, 5.6888e-01],
...,
[ 4.9915e-01, 1.1909e+00, -3.2562e+00, ..., 7.4343e-01,
5.7087e-01, 1.7486e-01],
[-2.5371e-01, -2.7965e-01, 5.8951e-01, ..., 5.9146e-01,
1.1841e+00, 1.5541e-01],
[ 6.3351e-01, -5.9298e-01, 2.2440e-01, ..., -6.9222e-01,
-3.1713e-01, 4.1056e-01]],
[[-4.1812e+00, -1.1415e+00, -2.6395e+00, ..., -2.7440e+00,
-2.3101e+00, -1.8438e+00],
[-4.0673e+00, -2.6116e+00, -2.7848e+00, ..., -2.1739e+00,
-2.5848e+00, -2.1450e+00],
[-2.6627e+00, -1.9702e+00, -4.1050e+00, ..., -2.7223e+00,
-2.3939e+00, -1.4533e+00],
...,
[-5.2599e-01, -9.9949e-01, 3.5749e-02, ..., 4.9131e-01,
1.9175e-01, 1.4228e-01],
[-3.5136e+00, -1.5422e+00, -3.2158e+00, ..., 1.5842e+00,
9.0813e-01, 2.5587e-01],
[-2.3205e+00, -1.8932e+00, -1.3241e+00, ..., -6.6319e-01,
-6.5929e-01, -8.8042e-01]],
[[ 3.0692e+00, 4.1225e+00, 3.1710e+00, ..., 2.0323e-01,
9.0623e-01, 2.6464e+00],
[ 3.9508e-01, 8.5179e-01, 2.7721e+00, ..., -2.0750e+00,
-3.9864e-01, 2.3621e+00],
[ 9.0513e-01, 1.1282e+00, 1.6604e+00, ..., -3.0891e+00,
-2.1080e-03, 1.6170e+00],
...,
[ 7.6822e-01, -2.8392e+00, -4.7230e+00, ..., -1.5122e+00,
-9.2095e-01, -1.7104e-01],
[ 6.0435e-01, 3.3137e+00, 9.4947e-01, ..., -1.5925e+00,
5.1245e-05, -1.7186e-02],
[ 1.3647e+00, 2.2320e+00, 2.3189e+00, ..., -9.9990e-01,
5.4280e-01, 5.7956e-01]],
...,
[[-2.0649e+00, -7.3870e+00, -5.8741e+00, ..., -3.9437e+00,
-2.2351e+00, -3.8028e+00],
[-2.7910e+00, -6.4797e+00, -4.5163e+00, ..., -6.6278e+00,
-5.2306e+00, -3.3443e+00],
[-2.8005e-01, -2.8948e+00, -3.4349e+00, ..., -3.3854e+00,
-2.8739e+00, -1.4986e+00],
...,
[-1.5965e+00, -3.8196e+00, -3.1247e+00, ..., -3.6250e+00,
-2.8353e+00, -6.8881e-01],
[-1.1912e+00, -4.3420e+00, -3.4978e+00, ..., -4.3501e+00,
-6.1642e+00, -3.7841e+00],
[-2.1432e+00, -5.0200e+00, -3.6762e+00, ..., -2.8762e+00,
-3.6543e+00, -2.0208e+00]],
[[-7.1851e-01, -9.0126e-02, -3.2864e-01, ..., 4.0968e-02,
1.7062e+00, 1.4497e+00],
[ 1.8067e+00, 1.0785e+00, 8.4678e-01, ..., -8.9444e-01,
9.5175e-01, 1.8698e+00],
[ 1.2111e+00, -4.1990e-02, 4.9645e-02, ..., -2.3118e+00,
2.3206e-01, 1.5187e+00],
...,
[ 6.1762e-01, 5.6177e-01, 2.5257e+00, ..., 8.2871e-01,
6.7311e-02, 9.7850e-02],
[ 1.5171e+00, -8.0074e-01, 8.5351e-01, ..., 1.4658e+00,
-3.6863e-01, 6.2617e-01],
[ 1.2126e+00, 8.3171e-01, 1.6025e+00, ..., 8.5570e-01,
2.2793e-01, 9.8286e-01]],
[[-5.2422e-01, 5.8442e-01, -1.4601e-01, ..., -1.0304e+00,
-1.2267e+00, 1.0913e-01],
[-1.1794e+00, -1.1510e+00, -2.0220e+00, ..., 1.1125e+00,
-1.1936e+00, -1.1545e+00],
[-2.0799e+00, -8.2469e-01, -1.1719e+00, ..., 2.7037e-02,
-2.2268e+00, -5.3118e-01],
...,
[-1.0830e+00, -9.3026e-01, -1.0200e+00, ..., 3.3365e-01,
-2.0908e+00, -1.4013e+00],
[-1.4946e+00, -1.9393e-01, -9.2286e-01, ..., -7.7523e-02,
-2.1257e+00, -1.5353e+00],
[-1.2917e-01, 6.6101e-01, -6.2776e-01, ..., -3.7422e-01,
-1.1746e+00, -5.3372e-01]]],
[[[ 6.8285e-01, 1.3127e+00, 7.8807e-01, ..., 4.3890e-01,
4.0221e-01, 2.6375e-01],
[-9.3574e-01, -2.5662e+00, -2.0819e-01, ..., 3.5523e-01,
2.3859e-01, -1.2099e+00],
[-1.4053e+00, -2.5799e+00, -1.5489e+00, ..., -6.5615e-01,
8.4448e-01, 5.6918e-01],
...,
[ 7.0442e-01, -3.5010e-01, 7.5757e-01, ..., -1.0590e+00,
-3.4391e+00, 4.0488e-01],
[ 6.9280e-02, -1.4638e+00, -7.4599e-01, ..., -2.3159e+00,
-3.0412e-01, 1.1371e+00],
[ 2.0253e-01, -5.9078e-01, -7.0886e-01, ..., 1.0007e+00,
-1.8224e+00, -3.1901e-02]],
[[-3.6766e+00, -3.2642e+00, -5.3147e+00, ..., -1.6271e+00,
-2.2330e+00, -1.1100e+00],
[-3.9483e+00, -4.2920e+00, -5.6014e+00, ..., -4.0806e-01,
-1.4630e+00, -3.3351e-01],
[-2.1543e-01, -1.3485e+00, -4.1376e-01, ..., -1.2097e+00,
-1.5455e+00, -7.8105e-01],
...,
[-2.3649e+00, -2.9223e-01, 1.8636e-01, ..., -4.8945e+00,
-5.2031e+00, -2.5029e+00],
[-3.6085e+00, -1.6783e+00, -1.7536e+00, ..., -1.1819e+00,
-4.9340e+00, -3.9360e+00],
[-1.4067e+00, -1.1284e+00, -1.9107e+00, ..., -6.0418e+00,
-9.4086e+00, -2.5715e+00]],
[[ 2.5171e+00, 3.0708e+00, 3.1965e+00, ..., 2.7634e+00,
1.4096e+00, 2.9381e+00],
[ 4.1485e-01, 4.9197e-01, 6.2562e-01, ..., 1.8435e+00,
-1.0619e-01, 2.5244e+00],
[-1.7813e-01, -8.6143e-02, -6.2187e-01, ..., 2.1162e+00,
8.8290e-01, 1.9236e+00],
...,
[ 9.2924e-01, 4.7365e-01, -5.9143e-01, ..., -1.5103e+00,
-1.2471e+00, 2.9111e+00],
[ 9.3817e-01, 1.3232e+00, 2.9236e-01, ..., -3.8566e+00,
-1.0653e+00, 1.7214e+00],
[ 1.5320e+00, 1.7467e+00, 1.6662e+00, ..., -5.4369e+00,
-2.8353e+00, 1.3389e+00]],
...,
[[-1.6924e+00, -4.6207e+00, -4.2102e+00, ..., -3.2310e+00,
-2.4253e+00, -4.0276e+00],
[-1.3245e+00, -3.2482e+00, -3.4446e+00, ..., -5.0939e+00,
-3.8140e+00, -4.3640e+00],
[ 1.3176e-01, -3.5852e+00, -2.2307e+00, ..., -2.2773e+00,
-2.0452e+00, -1.6691e+00],
...,
[-1.7352e+00, -4.0033e+00, -3.7711e+00, ..., 1.3221e+00,
1.6752e+00, -5.0880e+00],
[-1.5250e+00, -4.5471e+00, -5.0722e+00, ..., 1.3406e+00,
-3.3743e+00, -1.3404e+01],
[-1.3876e+00, -4.5409e+00, -4.1734e+00, ..., -3.2193e+00,
-6.4333e+00, -4.5626e+00]],
[[ 1.7017e-02, -1.1676e+00, -1.5738e+00, ..., 9.0348e-01,
1.0719e+00, 1.2348e+00],
[ 6.1694e-01, -1.3680e+00, -3.7626e+00, ..., 1.4322e+00,
2.2597e+00, 2.5799e+00],
[ 8.8439e-01, 1.8617e+00, 1.4941e+00, ..., 1.3665e+00,
1.8815e+00, 2.0108e+00],
...,
[ 1.0102e+00, 2.2959e-01, 1.8708e-02, ..., 1.5593e-02,
-2.1246e+00, 8.4384e-01],
[ 1.3307e+00, 1.1232e+00, 6.6697e-01, ..., 1.6576e-01,
-1.7882e+00, 8.0669e-01],
[ 7.0301e-01, 9.0854e-01, 1.0583e+00, ..., -6.1262e-01,
-1.5974e+00, 3.4578e-01]],
[[-4.9098e-01, 6.9956e-02, -2.5382e-01, ..., -1.3918e-01,
-1.1549e-01, 8.9974e-01],
[-1.5986e+00, -2.2868e+00, -8.6292e-01, ..., 3.4949e-02,
-2.0842e-01, 3.4479e-01],
[-6.8434e-02, -1.0851e+00, 9.2032e-01, ..., -1.3929e-01,
9.8612e-02, 2.3453e-01],
...,
[-1.5530e+00, -1.2622e+00, -8.9952e-01, ..., -2.2488e+00,
7.2334e-01, -2.1270e-02],
[-1.1839e+00, -1.5982e+00, -1.6464e+00, ..., 3.0000e-01,
1.3483e+00, 2.5446e-02],
[-2.6487e-01, -1.1651e-01, -5.7927e-01, ..., -3.7398e-02,
-4.1266e+00, -3.4196e+00]]],
[[[ 9.6859e-01, 1.5854e+00, 1.0640e+00, ..., 5.4977e-01,
3.2734e-01, -9.1796e-02],
[-8.3063e-01, -1.8655e+00, -1.0126e+00, ..., 3.6886e-01,
5.7600e-01, -6.9398e-01],
[ 9.2873e-03, -7.0196e-01, 6.1658e-01, ..., -6.5566e-01,
2.2453e-01, 7.2727e-01],
...,
[ 4.2111e-01, -2.3304e-01, -3.6548e-01, ..., 2.6515e-01,
9.2925e-01, 4.3747e-01],
[-2.2180e-01, -1.0741e+00, -3.4080e-01, ..., -5.1517e-02,
7.7333e-01, 8.6964e-02],
[ 7.3906e-01, -2.8041e-01, -2.9474e-01, ..., -4.8866e-01,
-4.0295e-01, 2.8321e-01]],
[[-4.1704e+00, -1.3529e+00, -2.1960e+00, ..., -3.0058e+00,
-3.5082e+00, -8.7612e-01],
[-4.0374e+00, -3.0775e+00, -2.2924e+00, ..., -2.1408e+00,
-3.2360e+00, -6.7920e-01],
[-2.6711e+00, -2.1681e+00, -1.9556e+00, ..., -3.0554e+00,
-3.2832e+00, -6.7406e-01],
...,
[-8.2559e-01, -5.3291e-01, -1.5676e+00, ..., -2.6523e+00,
-1.8775e+00, 6.8580e-02],
[-2.9157e+00, -4.2080e-01, -1.0553e+00, ..., -1.8389e+00,
-1.6606e+00, -5.0016e-02],
[-2.0590e+00, -1.4923e+00, -1.7758e+00, ..., -9.3914e-01,
-9.2426e-01, 5.9635e-02]],
[[ 3.0155e+00, 4.0866e+00, 3.2742e+00, ..., 2.3992e+00,
7.6312e-01, 2.4101e+00],
[ 4.0446e-01, 9.1738e-01, 2.3448e+00, ..., 1.9857e+00,
1.2683e-01, 2.4087e+00],
[ 1.1418e+00, 1.7223e+00, 1.3895e+00, ..., 8.5230e-01,
8.6297e-01, 2.2322e+00],
...,
[ 3.1572e-01, 9.4573e-01, 1.4182e+00, ..., 5.1786e-01,
-6.6287e-02, 1.4634e+00],
[ 3.7144e-01, 2.1430e+00, 1.6756e+00, ..., 2.0826e-01,
3.1983e-01, 1.2437e+00],
[ 1.4683e+00, 1.3943e+00, 1.7641e+00, ..., 1.1260e+00,
8.6828e-01, 1.3014e+00]],
...,
[[-2.0215e+00, -7.2799e+00, -5.9750e+00, ..., -4.1455e+00,
-3.0649e+00, -3.9561e+00],
[-2.7026e+00, -6.6488e+00, -4.9800e+00, ..., -5.6308e+00,
-4.7277e+00, -4.6346e+00],
[-3.1885e-01, -3.0935e+00, -3.1115e+00, ..., -4.4711e+00,
-3.3846e+00, -1.9691e+00],
...,
[-9.7125e-01, -2.3493e+00, -1.7091e+00, ..., -5.5796e+00,
-3.7717e+00, -2.6424e+00],
[-1.3001e+00, -3.7384e+00, -3.4464e+00, ..., -6.5380e+00,
-4.5274e+00, -4.6090e+00],
[-2.2275e+00, -5.0332e+00, -3.0493e+00, ..., -4.5230e+00,
-2.8689e+00, -3.2288e+00]],
[[-7.2614e-01, -1.3789e-01, 6.3071e-02, ..., 9.8840e-01,
1.0923e+00, 1.7677e+00],
[ 1.7320e+00, 1.0273e+00, 1.5474e+00, ..., 1.5410e+00,
1.5646e+00, 2.9044e+00],
[ 1.0127e+00, -1.2364e-01, 3.4663e-01, ..., 1.3080e+00,
8.9925e-01, 2.2329e+00],
...,
[ 1.2963e+00, -1.1477e-01, 6.3450e-01, ..., 1.5858e+00,
1.2650e+00, 1.6281e+00],
[ 1.8866e+00, 9.7833e-01, 1.6025e+00, ..., 1.5605e+00,
1.1673e+00, 2.3008e+00],
[ 1.0391e+00, 4.0595e-01, 8.8287e-01, ..., 1.0215e+00,
1.2190e+00, 1.5665e+00]],
[[-5.4192e-01, 5.5782e-01, 1.1138e-01, ..., -3.0123e-01,
-5.4724e-01, 9.4052e-01],
[-1.1531e+00, -1.2406e+00, -1.1313e+00, ..., -2.9927e-01,
-1.4354e-01, 1.0388e+00],
[-2.2183e+00, -6.7429e-01, 5.1850e-01, ..., 1.7236e-01,
-7.9602e-02, 5.3991e-01],
...,
[-1.7527e+00, -1.5088e+00, -1.2772e+00, ..., -6.4624e-01,
-1.0324e+00, -3.4638e-01],
[-1.7794e+00, -1.0972e+00, -1.0691e+00, ..., -8.2176e-01,
-1.1383e+00, 1.9690e-01],
[-3.6616e-01, 4.0146e-01, -4.3851e-01, ..., 1.4132e-01,
-2.3763e-01, 5.9963e-01]]],
[[[ 7.1547e-01, 1.0271e+00, 6.6550e-01, ..., 5.7587e-01,
6.3480e-01, 9.0991e-01],
[-3.3007e-01, -1.2051e+00, -5.7238e-01, ..., -6.7082e-01,
-1.9652e+00, -1.8750e+00],
[ 1.2486e-01, -7.8264e-01, -5.8415e-01, ..., -1.0111e+00,
-6.6553e-01, -5.2292e-01],
...,
[ 2.0082e-02, -1.2620e+00, -1.4218e+00, ..., 8.5554e-03,
3.6651e-01, 6.4728e-01],
[-4.8002e-01, -1.8040e+00, -1.0165e+00, ..., 4.7948e-01,
8.5647e-01, 1.7924e-01],
[ 3.5621e-01, -5.1638e-01, -7.4776e-01, ..., -6.5367e-02,
-2.9715e-01, 4.8092e-01]],
[[-3.1518e+00, -1.5386e+00, -3.0455e+00, ..., -1.8660e+00,
-1.1944e+00, -1.6780e+00],
[-4.4243e+00, -2.2309e+00, -3.6517e+00, ..., -3.1718e-01,
-1.6252e+00, -3.5244e+00],
[-4.4255e+00, -3.6889e+00, -4.9602e+00, ..., 3.8706e-01,
-1.0728e+00, -3.0027e+00],
...,
[-2.8917e+00, -1.9399e+00, -3.1975e+00, ..., -1.0420e+00,
-1.4307e+00, 4.0818e-01],
[-3.6122e+00, -1.3590e+00, -2.3703e+00, ..., -2.0010e-01,
-9.9158e-01, 2.6126e-01],
[-1.2390e+00, -8.3792e-01, -1.6765e+00, ..., -8.3166e-01,
-5.6396e-01, -1.9618e-02]],
[[ 1.7880e+00, 2.8342e+00, 2.3735e+00, ..., 2.5415e+00,
3.1885e-01, -5.0703e-02],
[ 2.3809e-01, 1.1208e+00, 1.4595e+00, ..., -1.7595e+00,
-2.1322e+00, -1.9035e+00],
[ 7.8487e-01, 2.3300e+00, 1.9906e+00, ..., -6.2666e+00,
-1.5982e+00, -1.2323e+00],
...,
[ 1.0569e+00, 1.3157e+00, 6.5061e-01, ..., -2.9511e-01,
-9.5532e-02, 1.1172e+00],
[ 6.9545e-01, 1.8907e+00, 6.1189e-01, ..., 1.0462e-01,
3.0912e-01, 1.0045e+00],
[ 1.5572e+00, 1.6038e+00, 1.6016e+00, ..., 6.8576e-01,
5.8593e-01, 1.1958e+00]],
...,
[[-1.4974e+00, -4.6805e+00, -3.9303e+00, ..., -5.5417e+00,
-1.7209e+00, -4.9674e+00],
[-1.8867e+00, -4.0399e+00, -5.0144e+00, ..., -7.5167e+00,
-4.1192e+00, -6.9002e+00],
[-1.0018e+00, -3.4692e+00, -4.5497e+00, ..., -7.7510e+00,
-8.6168e+00, -1.0491e+01],
...,
[-1.6140e+00, -4.2771e+00, -4.9467e+00, ..., -4.5267e+00,
-2.6215e+00, -3.0798e+00],
[-1.5249e+00, -4.5714e+00, -5.4961e+00, ..., -6.4894e+00,
-3.7394e+00, -4.6797e+00],
[-1.5931e+00, -4.8379e+00, -4.1884e+00, ..., -4.6206e+00,
-2.0380e+00, -3.0117e+00]],
[[ 7.4998e-01, 5.3604e-01, -1.6653e-01, ..., 1.8076e-01,
1.0322e+00, 5.1143e-01],
[ 1.2116e+00, 5.8653e-01, 5.9804e-01, ..., 1.3263e+00,
3.9170e-01, 1.8882e+00],
[ 5.8631e-01, -2.8453e-01, -1.5711e+00, ..., 2.9642e+00,
-1.0305e+00, -8.8282e-01],
...,
[ 8.0345e-01, 2.8349e-03, -8.7779e-01, ..., 2.7064e+00,
1.5597e+00, 1.8371e+00],
[ 1.5326e+00, 1.4238e+00, 1.1100e+00, ..., 2.7258e+00,
1.4224e+00, 2.4609e+00],
[ 6.7704e-01, 9.3226e-01, 9.7797e-01, ..., 1.2156e+00,
1.1880e+00, 1.3409e+00]],
[[-2.8609e-01, -2.0090e-01, -9.2303e-01, ..., 1.4321e+00,
4.2501e-01, -1.2701e-01],
[-1.4227e+00, -1.9251e+00, -1.7320e+00, ..., 2.1979e+00,
7.0130e-01, -4.5371e+00],
[-1.7336e+00, -1.4429e+00, -1.4832e+00, ..., 2.6626e+00,
-4.8377e-01, -5.3854e+00],
...,
[-1.7165e+00, -1.6639e+00, -2.1862e+00, ..., -1.5047e+00,
-9.3121e-01, -5.3477e-01],
[-1.4036e+00, -1.6078e+00, -1.8359e+00, ..., -9.6316e-01,
-8.8898e-01, 8.6730e-02],
[-1.8679e-01, 8.2871e-02, -5.5563e-01, ..., -1.1473e+00,
-7.8031e-01, 1.0591e-01]]]], device='cuda:0'),
tensor([[[[-2.0273e-01, -2.0245e-01, -8.9399e-01, ..., 8.8046e-02,
-3.7330e-03, -5.7279e-01],
[-1.6459e+00, -6.5280e-01, 4.0474e-01, ..., 5.2780e-01,
1.2475e+00, -1.6811e-01],
[-1.2854e-01, -2.8734e+00, -9.6607e-01, ..., -6.9288e-02,
7.0027e-01, 8.8520e-01],
...,
[ 3.0035e-01, -1.0421e+00, 1.1594e+00, ..., -1.2858e-01,
4.7995e-01, 5.9656e-01],
[-6.4250e-01, -2.2140e+00, -1.0021e+00, ..., -6.2631e-01,
-3.1910e-01, -1.4883e-01],
[-3.2193e-01, -1.7939e+00, -1.6640e+00, ..., -1.1700e+00,
-1.5479e+00, -2.7502e-01]],
[[-5.8746e-01, -1.2178e+00, 6.7220e-02, ..., -2.5156e+00,
-3.4688e+00, -9.0843e-01],
[-5.4498e-02, -1.4463e-04, 2.0023e+00, ..., -1.4364e+00,
-2.0333e+00, -1.4233e-01],
[-2.2343e+00, -2.0448e+00, -3.9253e-01, ..., -2.3240e+00,
-1.8109e+00, -4.8788e-01],
...,
[-1.4473e+00, -1.5939e+00, -3.3494e+00, ..., -1.4492e+00,
-1.6927e+00, -6.6998e-01],
[-3.0577e+00, -2.1320e+00, -3.1042e+00, ..., -2.3397e+00,
-2.4196e+00, -1.2693e+00],
[-2.3340e+00, -2.0199e+00, -2.7554e+00, ..., -1.9152e+00,
-1.8462e+00, -9.6934e-01]],
[[-5.5527e-01, 4.0090e-01, -7.6277e-01, ..., 2.3287e+00,
6.6819e-01, 2.2366e+00],
[-3.5492e+00, -2.5540e+00, -2.2121e+00, ..., 1.6952e+00,
5.3200e-02, 2.1791e+00],
[-2.3411e+00, -5.5755e+00, -5.9861e+00, ..., 9.1155e-01,
7.3079e-01, 2.2948e+00],
...,
[ 1.1802e+00, 2.1675e+00, 1.7639e+00, ..., -4.9088e-02,
1.0101e+00, 2.3249e+00],
[ 6.6200e-01, 1.9877e+00, 2.3804e+00, ..., 9.8110e-02,
1.0782e+00, 1.8560e+00],
[ 1.6817e+00, 1.6499e+00, 2.0066e+00, ..., 1.2917e+00,
1.7738e+00, 1.7473e+00]],
...,
[[-1.3555e+00, -6.6810e+00, -8.2274e-01, ..., -3.1226e+00,
-2.7642e+00, -3.6924e+00],
[-2.6678e+00, -7.0320e+00, -9.9020e-02, ..., -5.1681e+00,
-4.5387e+00, -5.0876e+00],
[-1.9008e+00, -5.4992e+00, -4.2062e+00, ..., -2.4425e+00,
-2.2916e+00, -2.1817e+00],
...,
[-2.2954e+00, -5.4438e+00, -4.6677e+00, ..., -7.8081e+00,
-5.2830e+00, -4.7260e+00],
[-2.8259e+00, -6.2710e+00, -5.4539e+00, ..., -6.2693e+00,
-5.5860e+00, -6.2369e+00],
[-1.5366e+00, -4.1942e+00, -2.7553e+00, ..., -3.2966e+00,
-2.7058e+00, -3.2856e+00]],
[[-5.1928e-01, -3.2007e+00, -2.7085e+00, ..., 6.2062e-01,
8.5582e-01, 1.6483e+00],
[ 7.8451e-01, -1.1987e+00, -1.3855e+00, ..., 1.1922e+00,
1.6994e+00, 2.5112e+00],
[ 9.3822e-01, 1.1111e+00, 2.2896e+00, ..., 8.9801e-01,
1.2052e+00, 1.8543e+00],
...,
[ 1.2949e+00, -2.9678e-01, -1.2514e+00, ..., 2.0517e+00,
1.1212e+00, 1.4864e+00],
[ 1.9050e+00, 1.3306e+00, 1.1212e+00, ..., 1.4390e+00,
1.2590e+00, 1.9457e+00],
[ 4.4672e-01, 8.0029e-01, 6.9397e-01, ..., 5.5441e-01,
7.3694e-01, 1.6629e+00]],
[[ 2.8436e-01, -1.3292e+00, -1.6511e+00, ..., -3.8480e-01,
-4.8726e-01, 5.2077e-01],
[-1.4360e+00, -2.0219e+00, 3.8672e-01, ..., -9.6907e-02,
5.1046e-01, 8.6642e-01],
[-1.9793e+00, -1.0635e+00, -2.1867e-02, ..., 2.4421e-01,
4.8513e-01, 7.7412e-03],
...,
[-1.8389e+00, -1.5645e+00, -1.8702e-01, ..., -2.3050e-01,
-6.0466e-01, -6.7331e-01],
[-1.6049e+00, -2.2502e+00, -1.3484e+00, ..., -1.0978e+00,
-1.3947e+00, -7.1152e-01],
[-8.4428e-01, -8.9578e-01, -1.1904e+00, ..., -6.2418e-01,
-8.4905e-01, -7.6720e-02]]],
[[[ 1.0595e+00, 2.0390e+00, 8.1649e-01, ..., 1.4303e+00,
1.1009e+00, 9.4878e-01],
[-1.2216e+00, -2.4220e+00, -1.6265e+00, ..., -3.1710e-01,
-5.1727e-01, -1.5567e+00],
[-2.5142e+00, -3.7357e+00, -1.5909e+00, ..., -3.2841e-02,
7.2407e-01, 8.5269e-01],
...,
[ 8.9233e-03, -1.1889e+00, -9.4710e-01, ..., -6.0085e+00,
-6.1571e+00, -1.5017e+00],
[-5.5340e-01, -1.7987e+00, -1.2678e+00, ..., 4.9509e-01,
-4.1724e-01, -3.6427e-01],
[-3.0436e-01, -1.6908e+00, -1.6419e+00, ..., -1.1830e+00,
-1.6219e+00, -2.9336e-01]],
[[-4.6403e+00, -2.9677e+00, -4.3678e+00, ..., -1.5470e+00,
-2.4923e+00, -1.2108e+00],
[-4.4486e+00, -6.6913e+00, -8.4473e+00, ..., -2.2008e+00,
-3.0463e+00, -9.0318e-01],
[-1.9364e+00, -1.7697e+00, -2.7138e+00, ..., -2.2293e+00,
-2.8741e+00, -1.0178e+00],
...,
[-1.5070e+00, -1.5549e+00, -2.4451e+00, ..., -1.5457e+01,
-1.4005e+01, -7.1846e+00],
[-2.9588e+00, -1.5646e+00, -2.8252e+00, ..., -5.3061e+00,
-4.7485e+00, -3.0401e+00],
[-2.2849e+00, -1.9933e+00, -2.7298e+00, ..., -2.0441e+00,
-1.9339e+00, -9.8755e-01]],
[[ 2.9414e+00, 4.7113e+00, 4.1296e+00, ..., 2.9300e+00,
1.0376e+00, 3.0006e+00],
[ 1.0318e+00, 2.5233e+00, 3.2255e+00, ..., 1.3494e+00,
-7.8008e-01, 2.7355e+00],
[ 4.5473e-01, 1.6326e+00, 2.8531e-01, ..., 3.5617e-01,
-4.4139e-01, 2.0746e+00],
...,
[ 9.7302e-01, 1.0345e+00, 1.2359e+00, ..., -7.2544e+00,
-8.9077e+00, -1.4237e+00],
[ 6.6044e-01, 1.8412e+00, 1.2054e+00, ..., 2.8202e+00,
1.4438e+00, 1.3800e+00],
[ 1.6099e+00, 1.6798e+00, 1.8580e+00, ..., 1.3249e+00,
1.8144e+00, 1.7634e+00]],
...,
[[-2.0382e+00, -8.3342e+00, -7.1745e+00, ..., -4.0358e+00,
-3.2319e+00, -5.4966e+00],
[-2.3920e+00, -7.8071e+00, -5.8866e+00, ..., -3.6454e+00,
-2.7233e+00, -4.3572e+00],
[-1.1279e-01, -3.4640e+00, -3.9603e+00, ..., -3.6850e+00,
-2.9583e+00, -2.4371e+00],
...,
[-2.5050e+00, -5.0047e+00, -4.0570e+00, ..., -8.0961e+00,
-9.6421e+00, -1.0362e+01],
[-2.6963e+00, -5.6005e+00, -5.2942e+00, ..., -1.1086e+01,
-1.0185e+01, -1.1984e+01],
[-1.4503e+00, -4.0526e+00, -2.8100e+00, ..., -3.2800e+00,
-2.7548e+00, -3.3106e+00]],
[[-3.5930e-01, -8.4359e-01, -4.6605e-01, ..., 6.3039e-01,
8.3041e-01, 1.3927e+00],
[ 3.7100e-01, -3.0889e+00, -6.1368e+00, ..., 1.8380e+00,
2.5544e+00, 3.0346e+00],
[ 1.2227e+00, 2.5216e+00, 1.7418e+00, ..., 8.5510e-01,
1.4425e+00, 2.3627e+00],
...,
[ 1.3067e+00, 3.0879e-01, -5.0856e-02, ..., -1.0677e+00,
-5.0767e+00, -1.1893e+00],
[ 1.8639e+00, 1.6521e+00, 1.0195e+00, ..., 3.5948e+00,
3.5241e+00, 3.5449e+00],
[ 4.0245e-01, 7.1577e-01, 7.2552e-01, ..., 5.5711e-01,
7.4774e-01, 1.6607e+00]],
[[-5.9542e-01, 2.5058e-01, 9.4119e-02, ..., 5.9391e-01,
5.1111e-01, 1.3542e+00],
[-1.3787e+00, -2.8901e+00, -2.4589e+00, ..., -6.4409e-01,
-4.3792e-01, 2.5526e-01],
[-1.0690e+00, -1.2340e+00, -6.6040e-01, ..., -8.9017e-02,
3.4343e-02, 7.6405e-01],
...,
[-1.8895e+00, -1.8435e+00, -1.3690e+00, ..., -7.3146e+00,
-7.8790e+00, -4.5842e+00],
[-1.4862e+00, -2.0439e+00, -1.7546e+00, ..., -2.4873e+00,
-3.2760e+00, -7.7419e-01],
[-7.8868e-01, -9.0463e-01, -1.1831e+00, ..., -6.0611e-01,
-8.6770e-01, -8.1939e-02]]],
[[[ 5.8851e-01, 8.2435e-01, 5.5362e-01, ..., 2.6710e-01,
3.8219e-01, 4.7111e-02],
[-2.4280e-01, -8.6477e-01, -4.5559e-01, ..., -9.2521e-01,
8.8979e-02, -2.8167e-01],
[-2.2169e-01, -9.3045e-02, 1.4763e-01, ..., -1.6132e-01,
-3.4775e-01, 6.3823e-01],
...,
[ 9.0453e-01, 4.9927e-01, -1.3562e+00, ..., 6.7482e-01,
9.7624e-01, 7.0702e-01],
[-2.3294e-01, -4.4208e-01, -5.5029e-01, ..., 9.1142e-01,
1.5205e+00, 1.0322e-01],
[ 1.0546e+00, 2.5589e-01, -8.6017e-01, ..., 3.3301e-01,
2.7934e-01, 7.1323e-01]],
[[-2.4505e+00, -5.8566e-01, -1.8489e+00, ..., -1.6280e+00,
-2.1778e+00, -1.4545e+00],
[-2.5114e+00, -5.1774e-01, -1.4084e+00, ..., -1.3465e+00,
-2.4161e+00, -1.5382e+00],
[-2.4742e+00, -1.4198e+00, -1.6203e+00, ..., -2.2021e+00,
-2.5258e+00, -1.0416e+00],
...,
[-1.2537e+00, 7.3760e-01, 6.6667e-01, ..., -6.2727e-01,
-2.0293e-01, 6.7570e-01],
[-2.7153e+00, 1.2195e+00, 1.8257e+00, ..., 6.8836e-02,
-8.0429e-02, 6.2984e-01],
[-1.8103e+00, -1.9071e+00, -1.7649e+00, ..., -4.2695e-01,
-5.2015e-01, -1.6622e-01]],
[[ 1.9975e+00, 2.5389e+00, 1.5849e+00, ..., 1.3778e+00,
6.9479e-01, 2.0012e+00],
[-1.9776e-01, 3.5427e-01, 5.5186e-01, ..., 2.0775e+00,
2.3780e-01, 2.1423e+00],
[ 3.6359e-02, 2.9695e-01, 2.1002e-01, ..., 3.6617e-01,
1.2410e+00, 1.9457e+00],
...,
[ 5.6025e-01, -4.6862e+00, -1.6892e+00, ..., 4.0793e-01,
4.5376e-02, 1.5788e+00],
[ 8.5711e-01, -1.1789e+00, -3.2738e-01, ..., 1.0180e+00,
6.9861e-01, 1.6735e+00],
[ 2.7337e+00, 1.0089e+00, 5.8935e-01, ..., 1.9508e+00,
1.0834e+00, 1.7499e+00]],
...,
[[-2.1024e+00, -5.0393e+00, -3.1709e+00, ..., -2.1350e+00,
-1.7240e+00, -2.9557e+00],
[-2.5601e+00, -5.4360e+00, -3.0423e+00, ..., -2.5233e+00,
-3.0358e+00, -2.8731e+00],
[-1.5723e+00, -5.4765e+00, -3.2541e+00, ..., -6.8513e+00,
-4.0375e+00, -2.1682e+00],
...,
[-2.6945e+00, -4.8230e+00, -4.4448e+00, ..., -2.1424e+00,
-2.0735e+00, -2.7741e+00],
[-2.2608e+00, -5.2903e+00, -4.5776e+00, ..., -4.3329e+00,
-3.0161e+00, -4.8246e+00],
[-2.2847e+00, -5.0911e+00, -2.3641e+00, ..., -2.9480e+00,
-1.5299e+00, -3.2601e+00]],
[[ 9.3598e-01, 1.6550e+00, 8.1194e-01, ..., 1.1224e+00,
1.5792e+00, 1.3748e+00],
[ 1.1985e+00, 1.0655e+00, 9.2341e-01, ..., 1.5101e+00,
1.0080e+00, 1.9117e+00],
[ 6.2988e-01, -2.9022e-02, 3.2620e-01, ..., 3.5219e-01,
1.0505e+00, 1.4309e+00],
...,
[ 1.4755e+00, -1.0014e-01, -1.3456e+00, ..., 1.6195e+00,
1.7755e+00, 1.4491e+00],
[ 2.3171e+00, 2.1241e+00, 1.2831e+00, ..., 1.9465e+00,
1.6922e+00, 2.5475e+00],
[ 6.9195e-01, 1.6077e-01, 5.5957e-01, ..., 1.3031e+00,
1.5009e+00, 1.5411e+00]],
[[-9.8885e-02, -6.5257e-01, -1.1763e+00, ..., -7.8410e-01,
-8.6965e-01, -1.1389e-01],
[-8.6190e-01, -2.5651e+00, -2.5259e+00, ..., -2.2429e+00,
-8.2476e-01, -6.8657e-01],
[-1.5094e+00, -1.9278e+00, -9.6518e-01, ..., -1.4312e+00,
-5.0133e-01, -2.8815e-01],
...,
[-1.7104e+00, 5.4678e-01, 8.8940e-02, ..., -5.9019e-01,
-3.8395e-01, -4.3411e-01],
[-1.8453e+00, -5.7192e-01, -8.7370e-01, ..., -3.7739e-01,
-4.1630e-01, 3.8820e-01],
[-8.1461e-02, 4.4427e-01, -4.8299e-02, ..., 1.6580e-01,
3.2417e-02, 6.0361e-01]]],
[[[ 7.5581e-01, 1.1048e+00, 7.3852e-01, ..., 4.0320e-01,
3.5564e-01, 2.0370e-01],
[-3.1450e-01, -1.3596e+00, -5.2063e-01, ..., 3.6448e-01,
3.6667e-01, -1.1646e+00],
[ 1.9401e-01, -7.4769e-01, -2.4699e-01, ..., -4.4226e-01,
1.0161e+00, 6.4890e-01],
...,
[ 1.0614e-01, -1.1551e+00, -8.5848e-01, ..., 1.3394e-01,
9.5354e-01, 5.1962e-01],
[-4.4000e-01, -1.8417e+00, -8.6860e-01, ..., -4.6693e-02,
1.4151e+00, -1.3340e-01],
[ 4.1332e-01, -4.9069e-01, -7.3469e-01, ..., -2.5935e-01,
3.8828e-01, 5.7843e-01]],
[[-2.7473e+00, -1.5555e+00, -3.0891e+00, ..., -1.7586e+00,
-2.2380e+00, -1.0777e+00],
[-4.0618e+00, -1.9253e+00, -2.3560e+00, ..., -4.0634e-01,
-1.2261e+00, -2.5400e-01],
[-4.1252e+00, -2.6113e+00, -3.2664e+00, ..., -1.4384e+00,
-1.4738e+00, -7.5328e-01],
...,
[-2.7746e+00, -1.7662e+00, -2.8319e+00, ..., 2.4786e-01,
9.0725e-02, 5.0584e-01],
[-3.6350e+00, -1.3622e+00, -2.3746e+00, ..., -2.3331e-01,
-3.2628e-01, 2.3946e-01],
[-1.2177e+00, -8.6742e-01, -1.6907e+00, ..., -3.3627e-01,
-4.7974e-01, -4.8549e-01]],
[[ 1.4892e+00, 2.7019e+00, 2.3310e+00, ..., 2.8377e+00,
1.4331e+00, 2.8767e+00],
[-5.1756e-03, 1.2340e+00, 1.5452e+00, ..., 1.9145e+00,
5.8817e-03, 2.5260e+00],
[ 6.4797e-01, 1.5051e+00, 9.9682e-01, ..., 1.1317e+00,
6.5719e-01, 1.8665e+00],
...,
[ 9.5087e-01, 1.1221e+00, 5.7942e-01, ..., 6.8376e-01,
-6.3714e-02, 1.2672e+00],
[ 7.0160e-01, 1.7837e+00, 8.3839e-01, ..., 7.9137e-01,
4.7319e-01, 1.3572e+00],
[ 1.5845e+00, 1.5434e+00, 1.6084e+00, ..., 1.4866e+00,
1.0251e+00, 1.9069e+00]],
...,
[[-1.1339e+00, -4.3545e+00, -3.8446e+00, ..., -3.4265e+00,
-2.5088e+00, -3.9246e+00],
[-1.4919e+00, -3.6263e+00, -4.1777e+00, ..., -5.4771e+00,
-3.9424e+00, -4.3819e+00],
[-9.8008e-01, -3.2327e+00, -3.8883e+00, ..., -2.6837e+00,
-2.3796e+00, -1.6094e+00],
...,
[-1.5536e+00, -3.8325e+00, -4.5328e+00, ..., -5.1522e+00,
-2.5018e+00, -1.4918e+00],
[-1.4043e+00, -4.3417e+00, -5.2131e+00, ..., -5.9397e+00,
-3.2655e+00, -3.5261e+00],
[-1.5972e+00, -4.8171e+00, -4.1262e+00, ..., -3.4895e+00,
-2.1610e+00, -2.6714e+00]],
[[ 8.4094e-01, 4.2052e-01, -1.1903e-01, ..., 9.1153e-01,
1.0780e+00, 1.2342e+00],
[ 1.1885e+00, 8.4540e-01, 5.3039e-01, ..., 1.6324e+00,
2.3880e+00, 2.5702e+00],
[ 5.5712e-01, 5.1013e-01, 2.4560e-01, ..., 1.0598e+00,
1.6389e+00, 1.9241e+00],
...,
[ 8.4582e-01, -6.0109e-02, -4.5528e-01, ..., 1.7722e+00,
1.7760e+00, 1.5878e+00],
[ 1.5286e+00, 1.4376e+00, 1.1212e+00, ..., 1.3146e+00,
1.4414e+00, 2.3676e+00],
[ 6.5940e-01, 9.2643e-01, 1.0358e+00, ..., 1.1258e+00,
1.3843e+00, 1.3516e+00]],
[[-2.5464e-01, -1.7559e-01, -9.9772e-01, ..., -2.0072e-01,
-1.8092e-01, 8.6839e-01],
[-1.3508e+00, -1.5224e+00, -1.9563e+00, ..., 1.5577e-01,
-7.8124e-02, 3.7010e-01],
[-1.5248e+00, -8.5707e-01, -1.0676e+00, ..., 2.5085e-01,
1.5462e-01, 2.1837e-01],
...,
[-1.8074e+00, -1.6321e+00, -1.6819e+00, ..., 4.5828e-02,
-2.4355e-01, -2.8659e-01],
[-1.5041e+00, -1.6606e+00, -1.6415e+00, ..., -2.3881e-01,
-4.8049e-01, 3.0235e-01],
[-1.4981e-01, 6.5739e-02, -4.8551e-01, ..., 2.1648e-01,
1.7667e-01, 3.2246e-01]]],
[[[ 5.7782e-01, 8.9731e-01, 5.7209e-01, ..., 4.9024e-01,
4.8333e-01, 7.0736e-02],
[-3.3876e-01, -7.7181e-01, -5.1966e-01, ..., 2.5522e-01,
1.5127e-01, -3.4605e-01],
[-2.5012e-01, -9.6421e-02, 2.7701e-01, ..., -2.3635e-01,
8.3711e-02, 8.0204e-01],
...,
[ 4.2960e-01, -2.5815e-01, 3.3216e-01, ..., 3.9316e-01,
8.6038e-01, 6.2773e-01],
[-6.9204e-02, -7.4231e-01, 2.9518e-02, ..., 8.8806e-01,
1.5514e+00, 1.2264e-01],
[ 8.0828e-01, -1.8666e-01, -1.3540e-01, ..., 4.0061e-01,
2.7856e-01, 7.4674e-01]],
[[-2.1877e+00, -7.4013e-01, -1.8372e+00, ..., -1.7308e+00,
-2.4374e+00, -1.7110e+00],
[-2.4902e+00, -7.3633e-01, -1.4002e+00, ..., -1.2447e+00,
-2.5778e+00, -1.8913e+00],
[-2.3090e+00, -1.2309e+00, -1.6759e+00, ..., -1.7778e+00,
-2.0564e+00, -1.0983e+00],
...,
[-4.9845e-01, -6.7942e-01, -1.0189e+00, ..., -7.8716e-01,
-3.2157e-01, 7.0950e-01],
[-2.6012e+00, -3.7108e-01, -3.3136e-01, ..., -4.5766e-02,
-5.0060e-02, 6.7423e-01],
[-1.5284e+00, -1.2530e+00, -9.2567e-01, ..., -4.6666e-01,
-6.0924e-01, -1.6561e-01]],
[[ 1.6874e+00, 2.3723e+00, 1.4951e+00, ..., 1.5882e+00,
7.2747e-01, 2.2193e+00],
[-3.6218e-01, 4.5545e-01, 6.1237e-01, ..., 9.3157e-01,
3.5659e-01, 2.4166e+00],
[ 5.9351e-02, 3.3525e-01, 1.3872e-01, ..., -2.0741e-01,
8.0322e-01, 2.0088e+00],
...,
[ 6.6332e-01, 4.8676e-01, 1.2460e+00, ..., 3.5931e-01,
1.5076e-01, 1.6624e+00],
[ 1.0613e+00, 2.1895e+00, 2.0856e+00, ..., 9.9536e-01,
7.5169e-01, 1.7154e+00],
[ 2.4103e+00, 1.8254e+00, 2.3826e+00, ..., 1.9548e+00,
1.0186e+00, 1.7118e+00]],
...,
[[-1.7416e+00, -4.7247e+00, -2.9836e+00, ..., -2.1558e+00,
-1.7156e+00, -3.3049e+00],
[-2.1034e+00, -4.9889e+00, -2.5725e+00, ..., -2.1971e+00,
-2.1882e+00, -3.0699e+00],
[-1.4073e+00, -5.1814e+00, -2.9295e+00, ..., -2.4071e+00,
-2.5118e+00, -2.2218e+00],
...,
[-2.5115e+00, -4.5207e+00, -2.2451e+00, ..., -2.7486e+00,
-2.0829e+00, -2.6538e+00],
[-2.4919e+00, -5.5236e+00, -3.7321e+00, ..., -4.4044e+00,
-3.1847e+00, -4.9100e+00],
[-2.6369e+00, -5.9583e+00, -2.8827e+00, ..., -2.9873e+00,
-1.6785e+00, -3.3384e+00]],
[[ 9.6447e-01, 1.4993e+00, 7.8563e-01, ..., 1.0243e+00,
1.4560e+00, 1.3449e+00],
[ 1.0723e+00, 9.1266e-01, 9.5219e-01, ..., 9.4582e-01,
8.5217e-01, 1.9915e+00],
[ 6.5450e-01, 4.1515e-02, 1.5726e-01, ..., 5.0035e-01,
2.1182e-01, 1.3936e+00],
...,
[ 1.4818e+00, -1.6021e-01, 2.4214e-01, ..., 1.4288e+00,
1.7602e+00, 1.5284e+00],
[ 1.9504e+00, 1.1427e+00, 1.8556e+00, ..., 1.9342e+00,
1.7035e+00, 2.5704e+00],
[ 1.1612e+00, 8.7938e-01, 1.3519e+00, ..., 1.3453e+00,
1.4894e+00, 1.5171e+00]],
[[-4.2577e-02, -6.6772e-01, -1.1677e+00, ..., -4.5839e-01,
-8.7810e-01, -1.0827e-01],
[-8.6681e-01, -2.6320e+00, -2.5236e+00, ..., -1.1693e+00,
-1.0823e+00, -7.8786e-01],
[-1.3754e+00, -1.6075e+00, -1.1860e+00, ..., -2.0172e-01,
-3.6913e-01, -1.9406e-01],
...,
[-1.5953e+00, -1.6073e+00, -3.4916e-01, ..., -4.3922e-01,
-4.7142e-01, -4.7923e-01],
[-1.7068e+00, -1.3184e+00, -6.1475e-01, ..., -3.7821e-01,
-3.4664e-01, 4.1764e-01],
[-2.1847e-01, 4.5580e-01, -2.0143e-01, ..., 2.3505e-01,
4.5247e-02, 6.2448e-01]]]], device='cuda:0'),
tensor([[[[ 6.7081e-01, 7.8475e-01, 3.8686e-01, ..., 4.9590e-01,
8.9973e-01, 4.6932e-01],
[-1.0551e+00, -8.4854e-01, -8.5644e-02, ..., 7.6423e-01,
-3.9200e-01, -3.6765e-01],
[-3.4538e-01, 7.0290e-01, 4.3005e-01, ..., -3.9982e-02,
-4.4488e-01, 6.2057e-01],
...,
[ 1.4826e-01, -5.2963e-01, -3.2680e-01, ..., -1.4298e-01,
-7.1661e-02, 4.1602e-01],
[-8.8990e-02, -1.1399e+00, -3.5290e-01, ..., -1.6000e-01,
4.8250e-02, -1.4732e-01],
[ 5.9481e-01, -2.0228e-01, -7.1275e-01, ..., -3.8907e-01,
-8.8671e-01, 1.5556e-01]],
[[-3.8874e+00, -1.9187e+00, -2.7577e+00, ..., -1.8358e+00,
-2.0769e+00, -1.8201e+00],
[-3.5876e+00, -2.3266e+00, -5.6983e-01, ..., -1.9469e+00,
-2.6225e+00, -2.1751e+00],
[-2.5376e+00, -1.7319e+00, -1.3186e+00, ..., -2.7840e+00,
-2.1310e+00, -1.4408e+00],
...,
[-2.2859e-01, -4.4491e-01, -6.6105e-01, ..., -1.0299e+00,
-9.9492e-01, -9.2497e-01],
[-1.5906e+00, -1.8921e-01, -3.1574e-01, ..., -9.4346e-01,
-1.2018e+00, -1.2724e+00],
[-8.3119e-02, -4.7020e-01, -6.9217e-01, ..., -7.1250e-01,
-7.4518e-01, -5.5064e-01]],
[[ 2.3558e+00, 3.0448e+00, 2.9479e+00, ..., 2.0887e+00,
7.7186e-01, 2.5662e+00],
[ 1.5615e-01, 7.8664e-01, 2.4479e+00, ..., 7.3396e-01,
9.4212e-02, 2.6410e+00],
[-5.5091e-01, -1.2102e-01, 9.7115e-01, ..., -5.3990e-01,
3.3104e-01, 1.8515e+00],
...,
[ 4.2068e-01, 4.3879e-01, 1.0019e+00, ..., -7.1189e-01,
5.5685e-01, 1.5180e+00],
[ 1.9291e-01, 1.7965e+00, 1.8232e+00, ..., -6.2654e-01,
8.0254e-01, 1.2151e+00],
[ 1.7421e+00, 1.0778e+00, 1.9276e+00, ..., 4.3173e-01,
1.3546e+00, 1.3855e+00]],
...,
[[-2.2981e+00, -7.1361e+00, -4.6935e+00, ..., -4.4303e+00,
-3.6339e+00, -4.3324e+00],
[-2.5469e+00, -8.6455e+00, -5.9656e+00, ..., -7.3279e+00,
-4.9086e+00, -3.2158e+00],
[-2.4095e+00, -6.3398e+00, -4.8091e+00, ..., -8.0364e+00,
-3.7084e+00, -1.3725e+00],
...,
[-3.2155e+00, -6.7097e+00, -3.4716e+00, ..., -5.1582e+00,
-3.1965e+00, -2.8655e+00],
[-3.4702e+00, -7.6808e+00, -4.5337e+00, ..., -6.0586e+00,
-4.2395e+00, -4.3515e+00],
[-2.8776e+00, -7.2837e+00, -3.7007e+00, ..., -3.6822e+00,
-2.4959e+00, -2.4658e+00]],
[[-7.1886e-01, 3.5826e-02, 3.3764e-01, ..., 1.8098e+00,
1.9840e+00, 1.1879e+00],
[ 9.5446e-01, 6.6692e-01, 1.4578e+00, ..., 1.7870e+00,
1.4234e+00, 1.8537e+00],
[ 1.3958e+00, -1.4821e-01, -2.4349e-01, ..., 1.4441e+00,
1.0205e+00, 1.2885e+00],
...,
[ 1.3509e+00, 4.7402e-02, 3.9248e-01, ..., 1.9582e+00,
7.5971e-01, 8.9771e-01],
[ 1.8230e+00, 1.3278e+00, 1.3922e+00, ..., 1.6193e+00,
7.4469e-01, 1.4558e+00],
[ 1.0736e+00, 9.5936e-01, 1.3254e+00, ..., 8.4941e-01,
7.0203e-01, 1.4499e+00]],
[[-7.1450e-01, -5.4729e-01, -6.1307e-01, ..., -2.7387e-01,
-4.4118e-01, -6.2672e-02],
[-1.4831e+00, -5.3758e-01, -5.1407e-01, ..., -7.0667e-01,
-1.9695e+00, -1.1238e+00],
[-2.3508e+00, 1.4248e+00, 4.4459e-01, ..., -1.2804e+00,
-2.3436e+00, -2.3028e-01],
...,
[-1.9956e+00, -2.0279e+00, -7.6328e-01, ..., -1.0092e+00,
-1.4891e+00, -8.5465e-01],
[-1.8935e+00, -2.0845e+00, -9.2354e-01, ..., -1.2726e+00,
-1.7208e+00, -7.7081e-01],
[ 3.8006e-02, 2.0279e-01, -2.3155e-01, ..., -4.7584e-01,
-7.8908e-01, 6.8084e-03]]],
[[[ 6.1000e-01, 6.9065e-01, 2.5961e-01, ..., 6.7166e-01,
9.3103e-01, 2.2414e-03],
[-6.2513e-01, -1.1896e+00, -9.3524e-01, ..., 1.0566e+00,
2.0791e+00, -1.7082e+00],
[-1.3239e+00, -2.3285e+00, -2.1261e+00, ..., -3.5386e+00,
-8.6522e-01, -4.5839e-01],
...,
[-1.6377e+00, -3.3826e+00, -4.7551e+00, ..., 2.5022e+00,
1.8562e-01, -8.5279e-01],
[-2.1307e+00, -5.4428e+00, -6.3152e+00, ..., -2.9333e+00,
1.7590e+00, -5.4478e-02],
[ 7.4748e-01, -3.2508e+00, -2.6995e+00, ..., -3.5807e+00,
2.2679e+00, 1.3473e+00]],
[[-3.8830e+00, -1.4868e+00, -4.1934e+00, ..., -2.2724e+00,
-2.6445e+00, -8.0328e-01],
[-7.0545e+00, -3.8597e+00, -5.0437e+00, ..., -1.2447e+00,
-4.7553e+00, -2.4604e+00],
[-9.3119e+00, -8.8140e+00, -9.2763e+00, ..., -6.2597e+00,
-8.3814e+00, -5.0745e+00],
...,
[-3.2286e+00, 8.6157e-01, -3.8978e+00, ..., -9.7151e+00,
-6.7525e+00, -2.3140e+00],
[-1.7141e+00, -7.5348e-01, -7.0847e+00, ..., -8.3285e+00,
-6.2404e+00, -1.4478e+00],
[-7.6469e-01, -2.3744e+00, -4.4646e+00, ..., -6.8089e+00,
-7.8717e+00, -1.5493e+00]],
[[ 2.4793e+00, 2.9214e+00, 2.8687e+00, ..., 2.1950e-01,
7.6307e-01, 3.5131e+00],
[ 5.2790e-01, 2.0691e+00, 4.3283e+00, ..., -1.3696e+00,
-2.3339e+00, 2.6556e+00],
[ 3.4926e+00, 2.5625e+00, 4.0171e+00, ..., -4.5148e+00,
-1.6123e+00, 4.4589e+00],
...,
[-1.0274e+00, -2.3461e+00, -5.1092e+00, ..., -1.1228e+01,
-1.8790e+00, 2.7178e+00],
[-4.5543e+00, -2.7038e+00, -3.8255e+00, ..., -8.8044e+00,
-1.7985e+00, 2.0360e+00],
[-3.6558e+00, -2.6168e+00, -1.3195e+00, ..., -9.3657e+00,
-1.3854e-02, 9.9658e-01]],
...,
[[-1.7901e+00, -4.9844e+00, -4.1327e+00, ..., -3.9193e+00,
-1.8080e+00, -4.2665e+00],
[-3.6986e+00, -7.5694e+00, -5.7789e+00, ..., -1.1303e+01,
-7.9762e+00, -8.6128e+00],
[-1.6874e-01, -3.9417e+00, -4.5742e+00, ..., 2.3080e+00,
-1.1817e-01, -8.5596e+00],
...,
[-2.1543e+00, -6.9958e+00, -3.5283e+00, ..., -8.4919e+00,
-7.0532e+00, -4.2407e+00],
[-1.0380e+00, -7.9316e+00, -9.5498e+00, ..., -5.4976e-01,
-4.4993e+00, -5.4850e+00],
[-1.6173e+00, -7.1180e+00, -5.1198e+00, ..., 1.7904e-01,
-8.7610e+00, -5.8776e+00]],
[[ 1.2014e-01, 6.0009e-01, -9.5848e-01, ..., -3.1176e+00,
3.2705e-01, 1.0183e+00],
[ 6.8700e-01, 6.8884e-01, -1.2908e+00, ..., -1.4913e+00,
-3.3531e-01, 2.3412e+00],
[-3.9331e+00, -5.9389e+00, -8.2175e+00, ..., -4.1201e+00,
-4.7108e+00, 7.9409e-01],
...,
[-1.1663e+00, -2.3693e+00, 6.9664e-01, ..., -4.1501e+00,
7.7187e-02, 2.5337e+00],
[ 5.7348e-01, -3.6404e-01, -1.5920e-01, ..., -6.4677e+00,
-2.9814e+00, 3.0691e+00],
[-8.0841e-02, -5.8247e-02, -6.6595e-02, ..., -2.2514e+00,
-2.1453e+00, 1.5680e+00]],
[[-6.7011e-01, -3.1228e-01, -1.7259e+00, ..., 3.4851e-01,
2.0047e-01, 3.4934e-01],
[-1.3684e+00, -1.0194e-01, -2.5799e+00, ..., 1.1944e+00,
1.4361e+00, 1.6280e-01],
[-3.0528e+00, -2.2224e-01, -1.1727e+00, ..., -3.1462e+00,
2.7108e-01, -3.7718e-01],
...,
[-2.2370e+00, 1.9155e+00, 9.5487e-01, ..., -1.6425e+00,
-3.2393e+00, -3.0530e+00],
[-3.7432e-01, 5.7388e-01, -5.2673e+00, ..., 6.1725e-02,
-8.3388e-01, -1.5116e-01],
[-2.1988e-01, -2.2265e+00, -6.6412e+00, ..., 2.5780e+00,
8.3825e-02, -3.8967e-01]]],
[[[ 7.3932e-01, 1.0989e+00, 7.3199e-01, ..., 4.6116e-01,
3.9691e-01, 2.0023e-01],
[-3.1914e-01, -1.3790e+00, -4.5755e-01, ..., 2.9364e-01,
3.1724e-01, -1.1220e+00],
[ 1.7983e-01, -7.7097e-01, -3.4424e-01, ..., -5.3548e-01,
9.4688e-01, 7.0809e-01],
...,
[ 8.5718e-02, -1.1887e+00, -1.0846e-01, ..., -1.2238e-01,
8.7361e-01, 5.0830e-01],
[-4.7347e-01, -1.9564e+00, -8.8406e-01, ..., -2.2937e-02,
1.3751e+00, -7.9147e-02],
[ 3.9940e-01, -5.3436e-01, -7.0201e-01, ..., -3.1837e-01,
3.6155e-01, 6.0418e-01]],
[[-2.9508e+00, -1.5681e+00, -3.0644e+00, ..., -1.8657e+00,
-2.4966e+00, -1.3393e+00],
[-4.2341e+00, -1.8097e+00, -2.2543e+00, ..., -3.8035e-01,
-1.2349e+00, -3.7520e-01],
[-4.2451e+00, -2.5891e+00, -2.9984e+00, ..., -1.5649e+00,
-1.3339e+00, -8.5539e-01],
...,
[-2.8440e+00, -1.8518e+00, -2.7172e+00, ..., -1.2305e-01,
-1.1748e-01, 4.3270e-01],
[-3.7640e+00, -1.4514e+00, -2.5118e+00, ..., -3.3131e-01,
-5.2648e-01, 1.8365e-01],
[-1.2972e+00, -8.9040e-01, -1.6941e+00, ..., -4.7364e-01,
-7.6359e-01, -5.0842e-01]],
[[ 1.5697e+00, 2.7488e+00, 2.3369e+00, ..., 2.9856e+00,
1.4461e+00, 3.0633e+00],
[ 6.4280e-02, 1.1951e+00, 1.5078e+00, ..., 2.0750e+00,
1.1225e-01, 2.7184e+00],
[ 6.7711e-01, 1.3786e+00, 8.1135e-01, ..., 1.2507e+00,
7.4902e-01, 1.8535e+00],
...,
[ 9.5934e-01, 1.4915e+00, 1.0969e+00, ..., 7.5060e-01,
1.9932e-01, 1.5082e+00],
[ 6.4153e-01, 1.8903e+00, 1.1263e+00, ..., 1.0941e+00,
6.0867e-01, 1.5112e+00],
[ 1.5324e+00, 1.6021e+00, 1.6502e+00, ..., 1.4522e+00,
1.0411e+00, 1.9656e+00]],
...,
[[-1.3116e+00, -4.5545e+00, -3.8615e+00, ..., -3.6383e+00,
-2.7195e+00, -4.2124e+00],
[-1.6510e+00, -3.7399e+00, -4.2095e+00, ..., -5.5749e+00,
-4.0326e+00, -4.7482e+00],
[-9.8052e-01, -3.1981e+00, -3.7120e+00, ..., -2.8149e+00,
-2.5132e+00, -1.5913e+00],
...,
[-1.6061e+00, -4.2009e+00, -4.6366e+00, ..., -4.2297e+00,
-2.5041e+00, -1.4382e+00],
[-1.4958e+00, -4.6748e+00, -5.3240e+00, ..., -4.6381e+00,
-3.3956e+00, -3.5069e+00],
[-1.6679e+00, -5.0878e+00, -4.2839e+00, ..., -3.3827e+00,
-2.1039e+00, -2.5356e+00]],
[[ 7.9896e-01, 4.3952e-01, -7.1990e-02, ..., 8.3045e-01,
9.7916e-01, 1.1963e+00],
[ 1.1830e+00, 9.3866e-01, 5.6830e-01, ..., 1.6329e+00,
2.4702e+00, 2.6660e+00],
[ 5.3751e-01, 5.5340e-01, 6.4828e-02, ..., 1.0831e+00,
1.6319e+00, 1.8817e+00],
...,
[ 8.0472e-01, -1.5823e-01, -3.3269e-01, ..., 1.3460e+00,
1.8581e+00, 1.7264e+00],
[ 1.4940e+00, 1.4070e+00, 1.2421e+00, ..., 1.2814e+00,
1.5272e+00, 2.4343e+00],
[ 6.6784e-01, 9.0208e-01, 1.0549e+00, ..., 1.0310e+00,
1.3557e+00, 1.3433e+00]],
[[-2.8305e-01, -1.5463e-01, -1.0124e+00, ..., -2.2350e-01,
-2.0119e-01, 8.7804e-01],
[-1.3862e+00, -1.4089e+00, -1.9192e+00, ..., 7.1365e-02,
-8.3315e-02, 4.3225e-01],
[-1.5995e+00, -8.5811e-01, -9.5481e-01, ..., 2.6658e-01,
1.9029e-01, 2.1602e-01],
...,
[-1.7231e+00, -1.5974e+00, -1.2604e+00, ..., -8.3913e-02,
-3.0015e-01, -3.1515e-01],
[-1.4526e+00, -1.6594e+00, -1.6358e+00, ..., -4.3723e-01,
-4.1075e-01, 3.3929e-01],
[-1.5883e-01, 7.4013e-02, -4.5157e-01, ..., 2.1962e-01,
1.6518e-01, 3.2216e-01]]],
[[[ 5.5971e-01, -8.5578e-01, 4.9432e-01, ..., 3.0200e-01,
3.7049e-01, 1.9149e-01],
[-1.2402e+00, -2.5657e+00, -3.1828e+00, ..., 4.8650e-02,
-1.6179e-01, -1.5576e+00],
[-2.0998e+00, -3.9985e+00, -3.0334e+00, ..., 1.4742e-01,
1.4074e+00, 1.0605e-01],
...,
[ 8.9991e-01, -7.6422e-01, 1.0005e+00, ..., 2.7706e-01,
7.8207e-01, 7.2813e-01],
[ 3.7248e-01, -8.8525e-01, 5.5228e-01, ..., 3.2921e-01,
1.1077e+00, 2.2960e-02],
[ 6.6214e-01, 8.5401e-01, -1.0398e+00, ..., -1.6460e-01,
5.9577e-01, 6.4992e-01]],
[[-2.9509e+00, 2.7647e-01, -2.1783e-01, ..., -1.8604e+00,
-2.3671e+00, -1.3016e+00],
[-2.1385e-01, -2.3922e+00, -3.9683e+00, ..., -1.1923e+00,
-2.1741e+00, -7.7564e-01],
[-1.4437e+00, -4.5545e+00, -3.8956e+00, ..., -1.6777e+00,
-3.6183e+00, -2.5789e+00],
...,
[ 3.5358e-02, 1.5341e+00, 1.0301e+00, ..., -1.1715e+00,
-2.0490e+00, -9.8445e-01],
[-9.6506e-01, 2.0795e+00, 2.1779e+00, ..., -5.4561e-01,
-1.1430e+00, -3.2152e-02],
[-1.0546e+00, -1.4930e+00, -4.5451e-01, ..., -4.8684e-01,
-8.0849e-01, -4.6273e-01]],
[[ 1.2058e+00, 9.5685e-01, -7.1588e-01, ..., 3.1184e+00,
1.4289e+00, 2.9958e+00],
[-3.6886e+00, -3.9861e+00, -3.6558e+00, ..., 2.2471e+00,
2.3515e-01, 3.1764e+00],
[-4.7414e+00, -3.3668e+00, -1.7825e+00, ..., -2.0412e+00,
-1.9467e-02, 3.1440e+00],
...,
[-1.5050e+00, -2.6874e+00, -1.0171e+00, ..., -5.0047e-01,
-1.5593e+00, 1.6310e+00],
[-7.2438e-01, -1.7320e+00, -7.1714e-01, ..., 9.6560e-01,
8.2395e-01, 2.0042e+00],
[ 1.5547e+00, -5.0948e-01, -2.0442e-01, ..., 1.5372e+00,
1.1330e+00, 2.0267e+00]],
...,
[[-8.3459e-01, -4.0512e+00, -2.9860e+00, ..., -3.7565e+00,
-2.5512e+00, -4.1142e+00],
[-2.0436e+00, -7.1214e+00, -4.4103e+00, ..., -5.1623e+00,
-3.4953e+00, -4.5235e+00],
[-1.0017e+00, -1.0086e+01, -2.4310e+00, ..., -3.8317e+00,
-2.0970e+00, -1.3413e+00],
...,
[-1.9631e+00, -4.6699e+00, -3.2882e+00, ..., -3.5562e+00,
-2.3969e+00, -2.4395e+00],
[-2.4532e+00, -6.1745e+00, -4.8177e+00, ..., -4.7699e+00,
-3.0325e+00, -3.4790e+00],
[-2.6403e+00, -6.2657e+00, -3.0795e+00, ..., -3.5402e+00,
-2.2760e+00, -2.7709e+00]],
[[-8.4103e-01, 1.4451e+00, -1.8867e-02, ..., 1.2593e+00,
1.0306e+00, 1.2302e+00],
[ 8.7517e-01, 1.0103e+00, 7.2025e-01, ..., 1.3706e+00,
2.0537e+00, 2.9217e+00],
[ 1.3577e+00, -1.3019e+00, -1.1611e+00, ..., -1.0578e+00,
-7.6856e-01, 1.3020e+00],
...,
[ 1.6775e+00, 3.9381e-01, -5.0266e-02, ..., 7.7803e-01,
1.3392e+00, 1.6387e+00],
[ 2.5390e+00, 2.1807e+00, 9.7767e-01, ..., 1.0392e+00,
1.1747e+00, 2.5295e+00],
[ 3.7763e-01, 5.8306e-01, 4.4611e-01, ..., 1.1339e+00,
1.4040e+00, 1.3199e+00]],
[[-6.6150e-01, -5.9904e-01, -3.6547e-01, ..., -1.1133e-01,
-1.9992e-01, 8.6775e-01],
[ 7.5292e-01, -4.0333e-02, -2.6086e+00, ..., -1.5414e-04,
8.8309e-02, 1.9245e-01],
[ 2.7010e-01, -2.0955e+00, -4.2424e+00, ..., 7.8846e-01,
1.2465e+00, 2.3419e-02],
...,
[-2.1726e+00, -4.3380e-01, -8.0433e-01, ..., -7.7833e-01,
-1.8374e+00, -1.5217e-01],
[-1.9215e+00, -1.1010e+00, -3.5223e-01, ..., -9.9389e-01,
-9.0588e-01, 6.7414e-01],
[-1.6364e-01, -8.4139e-01, -6.8219e-01, ..., -5.8283e-02,
1.6693e-01, 4.7460e-01]]],
[[[ 8.4658e-01, 1.4871e+00, 1.2087e+00, ..., 1.4211e+00,
1.0930e+00, 9.5714e-01],
[-9.9760e-01, -1.9391e+00, -1.1308e+00, ..., -3.5670e-01,
-5.4437e-01, -1.6022e+00],
[-1.1143e-01, -8.7009e-01, -3.5331e-01, ..., -5.9572e-02,
7.1254e-01, 7.6236e-01],
...,
[ 4.6155e-01, -5.5178e-01, -5.8705e-01, ..., 4.7621e-01,
2.0980e-01, 6.0985e-01],
[ 3.3074e-01, -1.8353e+00, -1.2430e+00, ..., 7.1802e-01,
6.5523e-01, -1.3377e-02],
[ 5.1420e-01, 4.1260e-01, 3.3739e-01, ..., 1.8052e-01,
-1.7843e-01, 6.3373e-01]],
[[-3.7161e+00, -7.5545e-01, -1.6240e+00, ..., -1.5696e+00,
-2.5286e+00, -1.2438e+00],
[-3.9500e+00, -3.1071e+00, -3.0748e+00, ..., -2.2685e+00,
-3.1303e+00, -9.4144e-01],
[-2.4492e+00, -2.5203e+00, -4.2055e+00, ..., -2.2801e+00,
-2.7424e+00, -1.0521e+00],
...,
[-9.9156e-01, -2.6857e+00, -3.4138e+00, ..., -1.5872e+00,
-1.5699e+00, -2.3143e-01],
[-1.8397e+00, -2.6860e-01, -7.3566e-01, ..., -6.7977e-01,
-1.2928e+00, -1.5219e-01],
[ 2.8915e-01, 3.8288e-02, -1.2976e+00, ..., 2.2068e-01,
7.2517e-03, 4.3408e-01]],
[[ 2.9163e+00, 4.2598e+00, 3.3701e+00, ..., 2.8916e+00,
1.0102e+00, 3.0541e+00],
[ 3.1741e-01, 1.0376e+00, 2.0281e+00, ..., 1.2870e+00,
-8.2272e-01, 2.7716e+00],
[ 1.1158e+00, 2.1020e+00, 2.4214e+00, ..., 4.2411e-01,
-2.8349e-01, 2.1132e+00],
...,
[ 4.0894e-01, 1.8929e+00, 1.3763e+00, ..., -3.1241e-01,
3.0080e-01, 1.5591e+00],
[ 1.0358e-01, 2.6546e+00, -3.2726e-01, ..., -1.2358e-02,
1.5623e-01, 1.1693e+00],
[ 1.3586e+00, 1.4437e+00, -1.1514e+00, ..., 7.5475e-01,
3.9120e-01, 1.1148e+00]],
...,
[[-1.9319e+00, -7.2818e+00, -5.4917e+00, ..., -4.0604e+00,
-3.2953e+00, -5.5773e+00],
[-2.4924e+00, -7.0802e+00, -4.9604e+00, ..., -3.5945e+00,
-2.7384e+00, -4.3870e+00],
[-1.1469e+00, -5.0031e+00, -5.0248e+00, ..., -2.7722e+00,
-2.9552e+00, -2.4104e+00],
...,
[-2.5967e+00, -5.7653e+00, -4.4428e+00, ..., -3.6303e+00,
-3.4475e+00, -3.4672e+00],
[-2.8311e+00, -7.7348e+00, -6.2727e+00, ..., -4.9469e+00,
-4.3725e+00, -4.8420e+00],
[-2.8428e+00, -7.8983e+00, -3.7443e+00, ..., -3.9363e+00,
-3.2114e+00, -4.0770e+00]],
[[-1.6150e-01, 3.0232e-01, 5.1158e-01, ..., 6.1186e-01,
8.3147e-01, 1.4251e+00],
[ 1.6472e+00, 7.1648e-01, 1.9300e+00, ..., 1.8314e+00,
2.5490e+00, 3.0493e+00],
[ 1.1781e+00, -7.1351e-01, -6.1857e-01, ..., 1.0258e+00,
1.5277e+00, 2.4098e+00],
...,
[ 1.0614e+00, -1.1864e+00, -1.3373e+00, ..., 9.1413e-01,
7.3831e-01, 1.3037e+00],
[ 1.5570e+00, 4.2514e-01, 7.7377e-01, ..., 1.2049e+00,
5.8874e-01, 2.0772e+00],
[ 1.1129e+00, 5.3588e-01, 6.8279e-01, ..., 1.3599e+00,
1.2271e+00, 1.3953e+00]],
[[-5.6854e-01, 2.6609e-01, 1.2438e-01, ..., 5.7738e-01,
5.2434e-01, 1.3621e+00],
[-1.3738e+00, -2.1884e+00, -1.7462e+00, ..., -6.3152e-01,
-4.0978e-01, 2.3399e-01],
[-2.2148e+00, -1.5369e+00, -2.2994e-01, ..., 2.3489e-01,
2.6198e-01, 7.5846e-01],
...,
[-1.5589e+00, -2.6335e+00, -1.7134e+00, ..., -7.9980e-01,
-1.1065e+00, -7.1207e-01],
[-1.2942e+00, -2.3794e+00, -1.3828e+00, ..., -7.7166e-01,
-1.2200e+00, 3.4009e-02],
[ 9.1475e-02, 4.3591e-01, -7.2654e-01, ..., 1.8995e-01,
-1.1149e-01, 7.7889e-01]]]], device='cuda:0'),
tensor([[[[ 2.8271e-01, 1.2022e+00, 1.1827e+00, ..., -3.2934e-01,
-8.8438e-03, 4.2379e-02],
[-8.7159e-01, -2.9250e+00, -6.5512e-01, ..., -5.8028e-01,
-6.9306e-01, -1.5314e+00],
[-3.4159e-01, -1.4854e+00, -1.4150e+00, ..., 1.4693e+00,
2.1546e+00, 6.5468e-01],
...,
[-5.1787e-01, -4.6876e-01, -8.7335e-01, ..., 2.0267e-02,
8.1652e-01, 5.1011e-01],
[-1.0860e+00, -1.9866e+00, -1.0444e+00, ..., 2.0698e-01,
1.3910e+00, -1.3023e-01],
[ 4.6251e-02, -1.8714e-01, 5.3454e-01, ..., -4.2868e-01,
4.0685e-01, 6.7207e-01]],
[[-2.0222e+00, -2.1062e+00, -4.1476e+00, ..., -1.4704e+00,
-2.1751e+00, -1.0578e+00],
[-2.5187e+00, -1.7051e+00, -2.1005e+00, ..., -1.8783e+00,
-1.9404e+00, -2.3515e-01],
[-8.1541e-01, -2.1422e+00, -2.6486e+00, ..., -1.5628e+00,
-2.2072e+00, -1.2142e+00],
...,
[-3.0188e+00, -9.6546e-01, -3.7815e-01, ..., -3.8770e-02,
7.1783e-02, 4.9593e-01],
[-3.8180e+00, -2.8673e-01, 2.8047e-02, ..., 2.3034e-01,
-6.1051e-01, 2.4740e-01],
[-1.5977e+00, -1.1230e+00, -1.6834e+00, ..., -3.0358e-01,
-7.7032e-01, -5.3673e-01]],
[[ 1.8139e+00, 1.9271e+00, 1.8030e+00, ..., 3.4715e+00,
1.5096e+00, 2.6714e+00],
[-3.1328e-01, -1.0576e+00, 6.6350e-01, ..., 2.1189e+00,
5.4417e-01, 2.6191e+00],
[-1.8458e-01, -8.4184e-01, -4.1263e-01, ..., -2.4616e+00,
-7.6951e-01, 2.2353e+00],
...,
[ 2.3444e+00, 8.2979e-01, -3.9468e+00, ..., -1.6536e-01,
2.4027e-02, 1.5197e+00],
[ 1.5273e+00, 1.1828e+00, -2.8221e+00, ..., 8.3529e-01,
6.5572e-01, 1.5847e+00],
[ 1.9639e+00, 1.5043e+00, -1.0418e+00, ..., 9.4350e-01,
1.0615e+00, 2.0380e+00]],
...,
[[-1.7957e+00, -4.0515e+00, -3.4119e+00, ..., -3.6150e+00,
-2.3496e+00, -4.0247e+00],
[-2.1224e+00, -2.2967e+00, -1.5952e+00, ..., -4.5856e+00,
-2.3771e+00, -4.2025e+00],
[-6.9848e-02, -2.0491e-01, 7.5324e-01, ..., -5.5301e+00,
-4.1058e+00, -2.2903e+00],
...,
[-1.4738e+00, -3.9337e+00, -3.4855e+00, ..., -6.5414e+00,
-3.6995e+00, -1.6470e+00],
[-1.2408e+00, -4.1636e+00, -3.2125e+00, ..., -6.6743e+00,
-3.3946e+00, -3.7419e+00],
[-1.4798e+00, -4.5750e+00, -2.6733e+00, ..., -5.7433e+00,
-2.1737e+00, -2.9077e+00]],
[[ 7.3234e-01, -7.5585e-01, -1.2756e+00, ..., 1.2085e+00,
1.3327e+00, 1.4511e+00],
[ 1.0596e+00, 7.0428e-01, -3.2652e-01, ..., 9.1944e-01,
2.9858e+00, 3.1216e+00],
[ 1.0016e-01, -2.0680e-02, -6.6926e-01, ..., -1.9050e+00,
9.5662e-01, 2.4604e+00],
...,
[ 4.7870e-01, 1.0465e+00, 6.5410e-01, ..., 2.1447e+00,
1.8453e+00, 1.7282e+00],
[ 1.1828e+00, 2.0154e+00, 2.0592e+00, ..., 1.3591e+00,
1.4542e+00, 2.4839e+00],
[ 4.1207e-01, 4.4460e-01, 1.2685e+00, ..., 1.0681e+00,
1.3893e+00, 1.3224e+00]],
[[-8.8588e-01, -1.1119e-01, -2.3240e-01, ..., -5.1120e-01,
-5.9016e-01, 7.4857e-01],
[-1.6943e+00, -1.6248e+00, -1.6968e-01, ..., -2.5565e-01,
-7.0386e-01, 3.0891e-01],
[-1.1548e+00, -2.0054e+00, -7.8876e-01, ..., 2.2421e+00,
6.5789e-01, 5.6062e-01],
...,
[-1.8653e+00, -8.6499e-01, -2.6288e-01, ..., -9.8349e-01,
2.8399e-02, -2.9807e-01],
[-1.7366e+00, -1.1862e+00, -1.4775e+00, ..., -4.8748e-01,
-4.2674e-01, 3.4376e-01],
[-2.3203e-01, 1.9318e-01, -5.7677e-01, ..., 5.7040e-01,
1.8444e-01, 3.7758e-01]]],
[[[-5.9997e+00, -5.9819e+00, -4.0524e+00, ..., 1.0975e+00,
6.2066e-01, 1.2965e+00],
[-7.2293e+00, -1.5714e+01, -1.3189e+01, ..., -4.1010e-01,
-2.5139e+00, -1.8631e+00],
[-3.0873e+00, -7.8189e+00, -7.7170e+00, ..., -2.8284e+00,
-1.3005e+00, 7.4781e-01],
...,
[ 9.4438e-01, -1.0632e+00, -8.6863e-01, ..., -4.5126e+00,
-1.8650e+00, -2.1475e+00],
[ 9.3096e-02, -3.8200e+00, -7.5170e+00, ..., 6.9332e-01,
-2.9199e-01, 1.0831e+00],
[-5.0058e-01, 1.1477e-01, -1.8782e+00, ..., -2.1500e+00,
-2.2657e+00, -6.1585e-01]],
[[-1.2104e+00, -6.4627e+00, -1.2396e+01, ..., -3.0283e+00,
-1.4143e+00, -9.8756e-01],
[-6.9417e+00, -1.1842e+01, -1.3573e+01, ..., -8.8268e+00,
-4.5869e+00, -9.4378e-01],
[-5.2358e+00, -7.4426e+00, -1.5293e+01, ..., -1.2534e+01,
-4.2347e+00, -1.3133e+00],
...,
[ 7.5679e-01, 3.7650e-01, -2.0785e+00, ..., -1.0871e+01,
-1.3062e+01, -7.1366e+00],
[-6.8358e-01, -5.7833e+00, -1.1225e+01, ..., -4.1083e+00,
-3.2259e+00, -2.9078e+00],
[-2.4333e-01, -2.4052e+00, -4.8996e+00, ..., -3.5714e+00,
-3.7335e+00, -1.9333e+00]],
[[-8.4364e-01, -2.3612e+00, -3.8457e+00, ..., 2.6876e+00,
1.4152e+00, 3.1331e+00],
[-4.4143e+00, -3.7036e+00, -7.4914e+00, ..., -1.5431e+00,
-4.4323e+00, 2.7318e+00],
[-2.7625e+00, -2.0764e+00, -1.5383e+00, ..., 1.2846e+00,
-4.4211e+00, 2.2845e+00],
...,
[-1.7127e+00, -1.7187e+00, 1.6983e-01, ..., -8.6167e+00,
-8.5906e+00, -6.3578e+00],
[-2.8438e+00, -5.1070e+00, -7.2926e+00, ..., 9.9141e-01,
1.9303e+00, 1.2857e+00],
[-9.0491e-01, -2.3911e+00, -2.0087e+00, ..., 3.9091e-01,
1.2740e+00, 1.8275e+00]],
...,
[[-6.8607e-01, -1.1804e+01, -7.5436e+00, ..., -1.5734e+01,
-7.7670e+00, -6.0469e+00],
[ 1.1116e-01, -1.1519e+01, -4.1580e+00, ..., -2.8894e+01,
-1.3822e+01, -4.2020e+00],
[-8.3114e-01, -1.1322e+01, -1.8519e+01, ..., -3.5763e+01,
-1.7052e+01, -2.3660e+00],
...,
[-1.4895e+00, -4.3225e+00, -5.3117e+00, ..., -1.1238e+01,
-9.7588e+00, -8.6166e+00],
[-3.0805e+00, -2.7702e+00, -3.4224e-02, ..., -9.9508e+00,
-8.0716e+00, -8.2963e+00],
[ 3.0088e-01, -2.8849e+00, -1.5238e+00, ..., -2.5468e+00,
-1.6462e+00, -2.2350e+00]],
[[ 2.1368e+00, 1.0185e+00, -1.9910e-01, ..., 2.7959e+00,
4.6381e+00, 1.4276e+00],
[-1.6133e+00, -7.9072e+00, -5.7868e+00, ..., 3.8418e+00,
7.2274e+00, 3.1173e+00],
[ 2.5918e+00, -1.1381e+00, -9.0485e+00, ..., 3.6279e+00,
6.1947e+00, 2.5316e+00],
...,
[ 1.6594e+00, 1.3073e+00, 1.1567e+00, ..., -2.3813e+00,
-4.1315e+00, 2.4792e-01],
[-8.2513e-03, -1.0148e+00, -9.7514e-01, ..., 4.2487e+00,
2.4730e+00, 2.9041e+00],
[-1.1191e+00, -2.0916e+00, -1.5037e+00, ..., -2.8268e-02,
3.1920e-01, 1.4212e+00]],
[[ 4.1109e-01, -1.6668e+00, -1.5798e+00, ..., 9.3178e-01,
1.4302e+00, 1.5120e+00],
[-6.2872e+00, -1.1708e+01, -1.5532e+01, ..., -2.5095e+00,
-2.3658e+00, 1.0279e-02],
[-6.2479e+00, -8.9345e+00, -8.8744e+00, ..., -8.1870e+00,
-2.4604e+00, 6.3712e-01],
...,
[-1.3262e+00, 3.2834e-02, 1.3484e-01, ..., -6.4870e+00,
-8.4434e+00, -4.0844e+00],
[-4.1905e-01, -5.9966e+00, -8.6026e+00, ..., -1.0699e+00,
-1.6632e+00, -9.9718e-01],
[-1.6191e+00, -3.0481e+00, -2.7621e+00, ..., -1.7148e+00,
-1.5110e+00, -1.3982e+00]]],
[[[ 5.6483e-01, 8.2867e-01, 5.6224e-01, ..., 3.4648e-01,
4.6930e-01, 3.7874e-02],
[-2.8126e-01, -9.5776e-01, -6.2087e-01, ..., 9.9122e-02,
2.3268e-01, -1.9747e-01],
[-2.2949e-01, -4.5489e-02, 3.2842e-01, ..., -1.9682e-01,
-1.3126e-02, 6.8165e-01],
...,
[ 4.5532e-01, -3.2627e-01, -4.3792e-01, ..., 4.5621e-01,
8.9416e-01, 6.7309e-01],
[-2.1994e-05, -5.9626e-01, 1.1568e-01, ..., 8.7496e-01,
1.5599e+00, 9.2512e-02],
[ 8.5891e-01, -2.0918e-01, -9.1547e-02, ..., 3.9627e-01,
2.8997e-01, 7.1499e-01]],
[[-2.3768e+00, -6.2204e-01, -1.8848e+00, ..., -1.5761e+00,
-2.1741e+00, -1.4535e+00],
[-2.5250e+00, -7.2849e-01, -1.5602e+00, ..., -1.1985e+00,
-2.1391e+00, -1.4622e+00],
[-2.4136e+00, -1.7132e+00, -2.2530e+00, ..., -2.0135e+00,
-2.1441e+00, -1.0451e+00],
...,
[-6.6239e-01, -1.0100e+00, -1.6158e+00, ..., -7.1366e-01,
-1.9732e-01, 7.1792e-01],
[-2.4763e+00, -8.2107e-02, -3.1043e-01, ..., -4.5229e-02,
-4.2109e-02, 6.8181e-01],
[-1.6009e+00, -1.2391e+00, -8.8021e-01, ..., -4.4551e-01,
-5.9519e-01, -1.6894e-01]],
[[ 1.9404e+00, 2.5647e+00, 1.5989e+00, ..., 1.5385e+00,
7.2621e-01, 1.9834e+00],
[-2.4443e-01, 3.7148e-01, 5.6812e-01, ..., 7.1857e-01,
2.6016e-01, 2.1405e+00],
[ 4.2112e-02, 3.0295e-01, 3.1103e-01, ..., -1.5044e-01,
6.6493e-01, 1.9206e+00],
...,
[ 7.1280e-01, 1.9584e-01, 1.4345e+00, ..., 4.1223e-01,
8.4814e-02, 1.5862e+00],
[ 1.0350e+00, 2.2043e+00, 1.8693e+00, ..., 1.0223e+00,
7.4561e-01, 1.6647e+00],
[ 2.3706e+00, 1.8149e+00, 2.3223e+00, ..., 1.9691e+00,
1.0428e+00, 1.7259e+00]],
...,
[[-2.0356e+00, -4.9491e+00, -3.2169e+00, ..., -1.9831e+00,
-1.6676e+00, -2.9041e+00],
[-2.5132e+00, -5.3868e+00, -2.9382e+00, ..., -2.3302e+00,
-2.1651e+00, -2.9023e+00],
[-1.5764e+00, -5.5453e+00, -3.4200e+00, ..., -2.1517e+00,
-2.3393e+00, -2.1350e+00],
...,
[-2.4998e+00, -4.2928e+00, -2.5891e+00, ..., -2.8579e+00,
-2.2400e+00, -2.7012e+00],
[-2.4751e+00, -5.2791e+00, -3.9768e+00, ..., -4.2953e+00,
-3.2087e+00, -4.8674e+00],
[-2.5949e+00, -5.7538e+00, -2.8287e+00, ..., -2.9370e+00,
-1.6840e+00, -3.3072e+00]],
[[ 9.3062e-01, 1.6402e+00, 7.8257e-01, ..., 1.1357e+00,
1.4420e+00, 1.3318e+00],
[ 1.1707e+00, 9.9494e-01, 9.1461e-01, ..., 1.0153e+00,
8.6188e-01, 1.8426e+00],
[ 6.3082e-01, -1.9696e-01, 2.7096e-01, ..., 5.2474e-01,
9.6343e-02, 1.2842e+00],
...,
[ 1.4771e+00, -1.8526e-02, 4.9203e-01, ..., 1.5066e+00,
1.8323e+00, 1.4998e+00],
[ 2.0526e+00, 1.2030e+00, 1.8003e+00, ..., 1.9344e+00,
1.7066e+00, 2.5738e+00],
[ 1.1274e+00, 8.3186e-01, 1.3132e+00, ..., 1.3226e+00,
1.4886e+00, 1.5288e+00]],
[[-7.1958e-02, -6.5888e-01, -1.1610e+00, ..., -5.4039e-01,
-8.8248e-01, -1.6600e-01],
[-8.3229e-01, -2.6516e+00, -2.5812e+00, ..., -1.1422e+00,
-1.0315e+00, -6.7815e-01],
[-1.4647e+00, -2.0690e+00, -1.0290e+00, ..., -3.1419e-01,
-5.3340e-01, -2.9873e-01],
...,
[-1.7188e+00, -1.6349e+00, -7.6271e-01, ..., -3.8949e-01,
-4.2007e-01, -4.1756e-01],
[-1.7159e+00, -1.1593e+00, -5.9476e-01, ..., -3.6902e-01,
-3.9550e-01, 4.2928e-01],
[-2.5889e-01, 3.8343e-01, -2.0559e-01, ..., 1.9570e-01,
2.3184e-02, 6.2279e-01]]],
[[[ 1.1315e-01, -9.2361e-01, -3.8826e+00, ..., 7.9361e-01,
7.6778e-01, -3.3321e-01],
[-2.5105e+00, -8.8759e+00, -1.3274e+01, ..., 4.9409e-01,
1.7140e+00, 3.2269e-01],
[-3.3627e+00, -8.8705e+00, -1.3497e+01, ..., -4.0765e-01,
6.6498e-01, 1.0609e+00],
...,
[ 4.6668e-01, -1.8201e-01, 5.2697e-01, ..., 1.1755e+00,
3.8170e+00, 4.7510e+00],
[-6.0695e-01, -1.8700e+00, -7.6613e-01, ..., -4.5523e+00,
-4.1368e+00, -8.6872e+00],
[ 2.6715e-02, -1.4463e+00, -1.7591e+00, ..., -1.9392e+00,
-1.0203e+00, -3.4464e-01]],
[[-3.0505e-01, -7.1049e+00, -4.4488e+00, ..., -2.7829e+00,
-4.2539e+00, -1.5583e+00],
[-1.9539e+00, -1.3218e+01, -1.5187e+01, ..., -1.7402e+00,
-2.5010e+00, -3.9266e-03],
[-2.3093e+00, -7.6524e+00, -1.0662e+01, ..., -2.7861e+00,
-1.8913e+00, -2.6399e-01],
...,
[-1.0098e+00, -2.3916e+00, -4.2929e+00, ..., -7.0380e+00,
-5.9923e+00, -2.5746e+00],
[-2.5421e+00, -2.0745e+00, -2.4222e+00, ..., -1.7416e+01,
-1.9273e+01, -8.0764e+00],
[-1.9015e+00, -1.9630e+00, -2.3952e+00, ..., -6.5699e+00,
-5.7498e+00, -3.5031e+00]],
[[-3.2927e-01, -5.1556e+00, -5.9885e+00, ..., 1.3702e+00,
1.2487e+00, 2.9406e+00],
[ 1.3024e+00, -1.0366e+01, -1.5506e+01, ..., -7.2313e-02,
-4.8701e-01, 1.5845e+00],
[-5.8527e-01, -5.6729e+00, -7.3913e+00, ..., -7.9841e-01,
3.5786e-01, 1.4624e+00],
...,
[ 1.5671e+00, 2.1192e+00, 1.8618e+00, ..., -1.9816e+01,
-2.6338e+01, -1.4001e+01],
[ 8.1785e-01, 2.3803e+00, 2.7206e+00, ..., -2.9326e+00,
-1.1927e+01, -6.0609e+00],
[ 1.9394e+00, 1.4603e+00, 1.9183e+00, ..., 4.2506e+00,
3.6381e+00, 2.1672e+00]],
...,
[[-2.4123e+00, -2.0536e+00, -6.1721e+00, ..., 2.9087e-01,
-3.0001e+00, -4.1381e+00],
[-2.9824e+00, -3.8593e+00, -7.1770e+00, ..., -1.4659e+00,
-6.0104e+00, -6.4621e+00],
[-2.9830e+00, -6.0250e+00, -7.0072e+00, ..., -1.1951e+00,
-1.7288e+00, -2.2616e+00],
...,
[-2.5793e+00, -6.0431e+00, -4.4752e+00, ..., -1.7298e+01,
-2.1612e+01, -2.7019e+01],
[-3.3612e+00, -7.1745e+00, -5.7506e+00, ..., -1.0826e+01,
-1.0011e+01, -1.3107e+01],
[-2.0757e+00, -5.4497e+00, -2.7644e+00, ..., -8.3819e+00,
-8.9985e+00, -1.0901e+01]],
[[ 1.8114e-01, -1.6754e+00, -1.5306e+00, ..., -1.4897e+00,
-4.7160e-01, 1.7561e+00],
[ 1.2826e+00, 2.0134e+00, -2.3465e-01, ..., -2.3101e-01,
1.1778e+00, 2.6275e+00],
[-1.5614e-01, -3.2337e-01, -3.9102e+00, ..., -9.2470e-01,
1.3212e+00, 1.8549e+00],
...,
[ 1.4744e+00, -8.4440e-01, -1.3282e+00, ..., -3.2323e+00,
-3.4971e+00, -3.5986e+00],
[ 2.1727e+00, 1.7897e+00, 1.4650e+00, ..., -3.6400e+00,
-2.3156e+00, 4.9074e-01],
[ 6.8038e-01, 8.7383e-01, 7.2980e-01, ..., 2.5889e+00,
3.9828e+00, 3.5833e+00]],
[[ 3.8669e-01, 5.4208e-01, 4.2946e+00, ..., 2.2207e-01,
1.2733e-01, 8.8032e-01],
[-1.2820e+00, -3.5696e+00, -4.2172e+00, ..., 1.1547e+00,
4.1401e-01, 1.1686e+00],
[-4.6131e+00, -7.0985e+00, -8.8975e+00, ..., -8.4239e-03,
4.3746e-02, -3.9205e-02],
...,
[-2.0204e+00, -1.2870e+00, -1.4725e-01, ..., -4.0981e+00,
-3.6082e+00, 6.1079e-01],
[-1.9479e+00, -2.3215e+00, -1.2908e+00, ..., -3.9487e+00,
-8.1551e+00, -4.8819e+00],
[-7.7833e-01, -9.0505e-01, -1.0141e+00, ..., -3.4768e+00,
-3.5183e+00, -1.6129e+00]]],
[[[ 7.8572e-01, 1.0816e+00, 6.9609e-01, ..., 6.5081e-01,
1.0137e+00, 4.7694e-01],
[-1.2554e+00, -1.1611e+00, -1.6322e-01, ..., -4.2891e-01,
-1.3562e-01, -2.6860e-01],
[-2.0196e-01, -9.7831e-01, -6.0647e-01, ..., -1.0597e+00,
-1.3512e-01, 6.4970e-01],
...,
[-8.5610e-01, -7.8813e-01, 1.0976e-01, ..., 4.9105e-01,
-8.8959e-02, 3.9443e-01],
[-3.0913e-01, -1.5753e+00, 1.6809e-01, ..., 1.8354e-01,
2.4975e-01, -1.4034e-02],
[ 1.1129e+00, -1.3543e-02, -6.8109e-01, ..., -3.2452e-01,
-8.2911e-01, 1.2025e-01]],
[[-4.0748e+00, -1.8922e+00, -3.0980e+00, ..., -1.5039e+00,
-1.8923e+00, -1.6900e+00],
[-3.1908e+00, -1.3913e+00, -7.8627e-01, ..., -1.5622e+00,
-1.9624e+00, -2.1297e+00],
[-2.0106e+00, -1.1194e+00, -2.1969e+00, ..., -1.7915e+00,
-1.5034e+00, -1.3849e+00],
...,
[-2.6065e+00, -4.1097e-01, -1.4208e+00, ..., -1.8159e+00,
-7.9016e-01, -8.5733e-01],
[-2.5280e+00, 4.7372e-03, -1.1797e+00, ..., -1.3692e+00,
-7.0277e-01, -1.1184e+00],
[-2.9643e-02, -8.9104e-01, -1.1477e+00, ..., -8.4088e-01,
-7.1489e-01, -5.2243e-01]],
[[ 2.8892e+00, 3.2381e+00, 3.4794e+00, ..., 1.3809e+00,
8.2592e-01, 2.5057e+00],
[ 8.7759e-01, 9.7701e-01, 3.1427e+00, ..., 1.1348e+00,
6.2291e-01, 2.5043e+00],
[ 1.1175e+00, 5.3374e-01, 2.1253e+00, ..., 2.3456e-01,
9.8030e-01, 1.7660e+00],
...,
[-2.0287e-01, -4.6007e-01, 7.2804e-01, ..., -3.3035e-01,
-1.8996e-01, 1.3686e+00],
[-3.9147e-01, 2.8817e-01, 1.5356e+00, ..., -3.2259e-01,
5.1913e-01, 1.0638e+00],
[ 1.2941e+00, -1.0269e-01, 1.7789e+00, ..., 3.1886e-01,
1.1007e+00, 1.2981e+00]],
...,
[[-3.0331e+00, -8.0757e+00, -5.1703e+00, ..., -5.4198e+00,
-3.4055e+00, -4.0495e+00],
[-3.4430e+00, -9.1160e+00, -6.3971e+00, ..., -5.7087e+00,
-3.5528e+00, -3.0047e+00],
[-2.3096e+00, -6.1829e+00, -3.3549e+00, ..., -5.6474e+00,
-2.2381e+00, -1.3375e+00],
...,
[-2.3667e+00, -3.2634e+00, -4.0576e+00, ..., -4.7992e+00,
-3.6027e+00, -2.5897e+00],
[-2.0730e+00, -4.4317e+00, -4.7401e+00, ..., -5.0050e+00,
-4.0846e+00, -4.1837e+00],
[-2.2337e+00, -5.3973e+00, -3.2533e+00, ..., -3.9145e+00,
-2.5678e+00, -2.2846e+00]],
[[-7.4830e-01, -7.3141e-02, -1.4695e-01, ..., 1.8268e+00,
1.6940e+00, 1.2139e+00],
[ 1.5524e+00, 1.0859e+00, 6.9142e-01, ..., 1.9502e+00,
1.1299e+00, 1.7546e+00],
[ 1.2123e+00, 3.2580e-01, -2.3764e-01, ..., 2.3703e+00,
5.8438e-01, 1.2460e+00],
...,
[ 8.7699e-01, -3.8259e-01, 1.4595e-01, ..., 1.3734e+00,
1.1309e+00, 8.3688e-01],
[ 1.4015e+00, 1.3180e+00, 7.9024e-01, ..., 1.5329e+00,
1.0180e+00, 1.3345e+00],
[ 6.2030e-01, 6.8969e-01, 8.7316e-01, ..., 9.2477e-01,
7.0389e-01, 1.3510e+00]],
[[-7.7227e-01, -7.0978e-01, -1.0588e+00, ..., -2.9135e-01,
-5.3069e-01, -1.9740e-02],
[-1.5402e+00, -1.4290e+00, -1.2342e+00, ..., -1.3073e+00,
-1.3391e+00, -1.1047e+00],
[-2.2304e+00, -7.5745e-01, -3.6159e-01, ..., -7.8527e-01,
-1.0893e+00, -2.5769e-01],
...,
[-3.3488e+00, -5.5818e-01, -7.1578e-01, ..., -1.3294e+00,
-1.8159e+00, -9.1363e-01],
[-2.7228e+00, -8.0552e-01, -8.5112e-01, ..., -1.8135e+00,
-1.5872e+00, -7.1832e-01],
[ 2.6291e-01, 1.4679e-01, -2.3809e-01, ..., -1.4468e-01,
-7.4615e-01, -2.2627e-02]]]], device='cuda:0'),
tensor([[[[ 6.5432e-01, 1.4092e+00, 3.7426e-01, ..., 2.0369e+00,
1.8549e-01, 4.7120e-01],
[-4.6215e-01, -6.1791e-01, -2.8940e+00, ..., 2.8276e+00,
-7.4205e-01, -4.3394e+00],
[ 2.5347e-01, -1.7693e+00, -4.8021e+00, ..., 4.5504e+00,
5.8131e+00, 2.0325e+00],
...,
[-1.6993e+00, -1.1717e+00, 2.2452e+00, ..., 5.6498e-01,
7.1193e-01, 4.7202e-02],
[-1.0557e+00, -1.7126e+00, 1.3412e+00, ..., -1.8045e-01,
1.9710e+00, -1.4111e-01],
[ 6.2972e-01, -4.8526e-01, -1.1215e+00, ..., -7.6993e-01,
5.1211e-01, 5.8334e-01]],
[[-3.8005e+00, -2.5020e+00, -6.0620e+00, ..., -2.4012e+00,
-3.1301e+00, -9.1564e-01],
[-4.2373e+00, -3.1570e+00, -8.6132e+00, ..., -1.4872e+01,
-9.3700e+00, -1.9963e+00],
[-4.2224e+00, -5.1569e+00, -7.2129e+00, ..., -7.5628e+00,
-5.4164e+00, -2.5567e+00],
...,
[-3.1651e+00, -5.1519e+00, -6.5643e+00, ..., -5.7800e+00,
-9.2289e-01, 1.9711e-01],
[-4.2672e+00, -1.2432e+00, -2.6009e+00, ..., -1.5458e+00,
-3.7320e-01, -3.5027e-01],
[-8.2038e-01, -8.5545e-01, -2.2780e+00, ..., -7.6534e-01,
-1.4412e+00, -1.1089e+00]],
[[ 2.3626e+00, 2.8534e+00, 2.1495e+00, ..., -2.0541e+00,
8.0079e-01, 2.6523e+00],
[ 8.6106e-01, 2.0034e+00, 2.7497e+00, ..., -7.6645e+00,
-4.2487e+00, 5.1582e+00],
[ 9.6707e-01, 2.2415e+00, 1.3602e+00, ..., -6.1646e+00,
-3.7280e+00, 3.5940e+00],
...,
[ 7.3305e-01, 3.2554e+00, 7.5198e-01, ..., 7.9299e-01,
2.0775e-02, 1.0412e+00],
[-2.7680e-01, 2.2851e+00, 3.9181e+00, ..., 2.2696e+00,
4.5164e-01, 1.6718e+00],
[ 2.0609e+00, 1.3967e+00, 1.9287e+00, ..., 9.3260e-01,
8.5599e-01, 2.2273e+00]],
...,
[[-1.7308e+00, -5.2846e+00, -5.5359e+00, ..., -2.1363e+01,
-1.1120e+01, -4.9594e+00],
[-2.3838e+00, -6.4629e+00, -6.3456e+00, ..., -1.9627e+01,
-7.0504e+00, -4.1619e+00],
[-1.4490e+00, -5.2361e+00, -7.1664e+00, ..., -8.3744e+00,
-7.9023e+00, -8.8555e+00],
...,
[-3.2128e+00, -6.3512e+00, -8.1288e+00, ..., -8.5920e+00,
-9.6672e+00, -5.7488e+00],
[-2.5269e+00, -7.4871e+00, -7.8154e+00, ..., -6.0886e+00,
-5.9048e+00, -3.9014e+00],
[-2.2662e+00, -6.0169e+00, -4.2253e+00, ..., -6.3232e+00,
-2.8625e+00, -2.9377e+00]],
[[ 2.7631e-01, -5.4975e-01, -1.9530e+00, ..., -2.9246e+00,
2.3045e+00, 1.4028e+00],
[ 1.2879e+00, -5.9209e-01, -4.6079e+00, ..., -1.0320e+01,
6.7186e-01, 3.5277e+00],
[ 5.6631e-01, -5.2769e-01, 2.7915e-01, ..., -8.1835e+00,
-2.5552e+00, 1.5286e+00],
...,
[ 7.3105e-01, -3.3812e+00, -3.2404e+00, ..., 2.6295e-01,
4.1922e+00, 2.0432e+00],
[ 1.7893e+00, 7.5179e-01, 8.8168e-02, ..., 2.1629e+00,
2.0138e+00, 2.5371e+00],
[ 7.6615e-01, 1.0485e+00, 6.0131e-01, ..., 1.4459e+00,
1.2792e+00, 1.2364e+00]],
[[-5.6551e-01, -1.2161e+00, -1.9417e+00, ..., -5.6819e-01,
-3.1455e+00, 4.5978e-01],
[-1.4877e+00, -2.5097e+00, -3.6713e+00, ..., -1.4477e+01,
-1.0068e+01, -2.4084e+00],
[-2.2057e+00, -2.6482e+00, -2.3846e+00, ..., -2.9892e+00,
1.1655e+00, 2.5569e+00],
...,
[-2.5802e+00, -4.4490e-03, 1.6552e-01, ..., -1.8888e+00,
-1.8100e-01, 4.9056e-01],
[-2.2853e+00, -1.7294e+00, -6.7741e-01, ..., -2.1296e-01,
-4.2785e-02, 6.1180e-01],
[-1.0420e-01, -3.0684e-02, -4.5409e-01, ..., 9.9355e-01,
2.2164e-01, 3.0328e-01]]],
[[[ 8.0818e-01, 1.0804e+00, 5.7636e-01, ..., 5.9006e-01,
9.8396e-01, 4.7596e-01],
[-1.4371e+00, -1.2767e+00, -1.5633e-01, ..., -9.5917e-01,
-3.5596e-01, -3.9002e-01],
[ 2.3580e-01, -1.4030e+00, -8.9460e-01, ..., -6.1962e-02,
-2.8589e-01, 6.0592e-01],
...,
[ 1.9358e-01, -8.2293e-01, 7.2007e-02, ..., -1.3290e-01,
-5.9414e-02, 4.2230e-01],
[ 2.4951e-01, -5.5025e-01, -7.8277e-02, ..., -2.4504e-01,
1.6576e-02, -6.2725e-02],
[ 4.7337e-01, -2.1389e-01, -7.5589e-01, ..., -4.5693e-01,
-9.5749e-01, 1.0030e-01]],
[[-4.1119e+00, -2.0581e+00, -3.0425e+00, ..., -1.7394e+00,
-2.2362e+00, -1.8677e+00],
[-3.5239e+00, -1.5981e+00, -7.9390e-01, ..., -1.8067e+00,
-2.4222e+00, -2.1621e+00],
[-3.9095e+00, -2.0251e+00, -1.2427e+00, ..., -1.4416e+00,
-1.7692e+00, -1.4330e+00],
...,
[-2.9369e+00, -3.0184e+00, -2.1443e+00, ..., -6.5434e-01,
-5.9929e-01, -8.2936e-01],
[-1.2554e+00, 4.1407e-03, -4.2950e-01, ..., -8.2620e-01,
-8.6425e-01, -1.1921e+00],
[ 2.0396e-01, -6.8421e-01, -8.7292e-01, ..., -7.0883e-01,
-6.0332e-01, -5.2591e-01]],
[[ 2.7951e+00, 3.4315e+00, 3.2848e+00, ..., 1.3098e+00,
7.2732e-01, 2.6028e+00],
[ 7.2280e-01, 1.1000e+00, 2.4622e+00, ..., 1.9886e+00,
6.3855e-01, 2.6792e+00],
[ 7.2994e-01, 7.0096e-01, 2.4635e+00, ..., 5.8205e-01,
1.0858e+00, 1.8590e+00],
...,
[-7.9834e-01, -3.6022e-01, 2.1645e+00, ..., -8.4021e-01,
4.5531e-01, 1.5020e+00],
[ 5.1560e-01, 1.9586e+00, 1.5773e+00, ..., -7.5032e-01,
6.9019e-01, 1.1445e+00],
[ 1.5615e+00, 1.1669e+00, 1.8565e+00, ..., 3.6423e-01,
1.2173e+00, 1.3530e+00]],
...,
[[-2.9975e+00, -8.4124e+00, -5.3454e+00, ..., -5.7565e+00,
-3.9039e+00, -4.5006e+00],
[-3.3834e+00, -9.3374e+00, -6.0842e+00, ..., -5.1917e+00,
-4.0716e+00, -3.4364e+00],
[-1.8226e+00, -5.7694e+00, -3.1889e+00, ..., -5.8154e+00,
-2.9856e+00, -1.4344e+00],
...,
[-2.3394e+00, -5.4148e+00, -4.1884e+00, ..., -6.6589e+00,
-3.6547e+00, -2.9238e+00],
[-2.7696e+00, -7.4715e+00, -5.6679e+00, ..., -5.8656e+00,
-3.9068e+00, -4.3446e+00],
[-2.3718e+00, -6.5674e+00, -3.5843e+00, ..., -3.5612e+00,
-2.1983e+00, -2.3526e+00]],
[[-7.8062e-01, -1.7332e-01, 2.8123e-01, ..., 1.8271e+00,
1.6576e+00, 1.1648e+00],
[ 1.4591e+00, 1.1765e+00, 1.2428e+00, ..., 2.5647e+00,
1.4349e+00, 1.8976e+00],
[-1.7287e-01, 9.5765e-02, 1.2569e+00, ..., 1.8576e+00,
1.1104e+00, 1.3172e+00],
...,
[ 3.2653e-01, -7.7673e-01, 1.0010e+00, ..., 2.1798e+00,
7.3793e-01, 8.7992e-01],
[ 1.8489e+00, 1.3476e+00, 1.6606e+00, ..., 1.7708e+00,
7.8247e-01, 1.3899e+00],
[ 1.1304e+00, 8.0649e-01, 1.1577e+00, ..., 8.8518e-01,
7.3400e-01, 1.4110e+00]],
[[-7.5800e-01, -7.3521e-01, -5.8109e-01, ..., -4.8510e-01,
-6.6759e-01, -5.2462e-02],
[-1.1919e+00, -1.3055e+00, -7.9110e-01, ..., -2.1836e+00,
-1.3717e+00, -1.1253e+00],
[-1.6226e+00, 4.4741e-01, -8.4064e-02, ..., -1.1811e+00,
-1.1837e+00, -2.3716e-01],
...,
[-2.9871e+00, -2.7460e+00, -8.3028e-01, ..., -4.0465e-01,
-9.8996e-01, -9.2154e-01],
[-2.2879e+00, -1.1839e+00, -1.9613e-01, ..., -1.3235e+00,
-1.4923e+00, -6.8825e-01],
[ 1.1789e-01, 9.5197e-02, -2.3095e-01, ..., -5.6757e-01,
-7.9211e-01, -4.1972e-03]]],
[[[ 9.8395e-01, 1.5549e+00, 1.0351e+00, ..., 4.8560e-01,
3.0504e-01, -2.2467e-01],
[-8.9672e-01, -2.0398e+00, -1.0500e+00, ..., 3.0656e-01,
7.3036e-01, -5.0251e-01],
[ 5.8054e-03, -8.2304e-01, -1.5643e-01, ..., -6.0418e-01,
2.3962e-01, 7.5840e-01],
...,
[ 4.0476e-01, -6.9525e-01, 8.6531e-01, ..., 1.4151e+00,
-2.4998e+00, -9.4696e-01],
[-2.1233e-01, -1.2971e+00, -3.0780e-01, ..., 5.8617e-01,
1.2663e+00, 8.9398e-02],
[ 7.3946e-01, -3.3950e-01, -3.2945e-01, ..., -4.7805e-01,
-1.5689e-01, 4.4224e-01]],
[[-4.2726e+00, -1.3728e+00, -2.4382e+00, ..., -2.8680e+00,
-3.5442e+00, -8.9027e-01],
[-4.0919e+00, -2.8734e+00, -2.4331e+00, ..., -2.0595e+00,
-3.0423e+00, -5.4833e-01],
[-2.6444e+00, -1.8008e+00, -2.9180e+00, ..., -3.1209e+00,
-3.2755e+00, -6.0397e-01],
...,
[-7.9837e-01, -9.6372e-01, -3.1944e+00, ..., -4.0378e+00,
-3.0884e+00, -6.2947e-01],
[-2.7685e+00, -6.3644e-01, -1.3734e+00, ..., -1.9244e+00,
-1.5977e+00, -2.6082e-01],
[-2.0513e+00, -1.4634e+00, -1.8122e+00, ..., -9.4607e-01,
-8.1245e-01, 1.3189e-01]],
[[ 3.0798e+00, 4.1258e+00, 3.3825e+00, ..., 2.2708e+00,
8.3668e-01, 2.3387e+00],
[ 4.2651e-01, 8.2899e-01, 2.4074e+00, ..., 1.8043e+00,
2.3913e-01, 2.3467e+00],
[ 8.9686e-01, 1.2903e+00, 1.9211e+00, ..., 7.4736e-01,
8.0263e-01, 2.0956e+00],
...,
[ 2.5913e-01, 1.3176e+00, 2.4472e+00, ..., -2.0209e+00,
-3.0975e+00, -4.0532e-01],
[ 2.8763e-01, 1.9274e+00, 2.3087e+00, ..., 8.3442e-01,
6.7614e-01, 1.2300e+00],
[ 1.3933e+00, 1.3452e+00, 1.7521e+00, ..., 1.1750e+00,
9.3933e-01, 1.4611e+00]],
...,
[[-2.1812e+00, -7.5210e+00, -5.9629e+00, ..., -3.9443e+00,
-2.9671e+00, -3.6148e+00],
[-2.8143e+00, -6.6244e+00, -4.6814e+00, ..., -5.5872e+00,
-4.7283e+00, -4.4918e+00],
[-2.4412e-01, -2.8547e+00, -3.0633e+00, ..., -4.7751e+00,
-3.3663e+00, -1.7538e+00],
...,
[-9.1706e-01, -2.6085e+00, -1.9092e+00, ..., -6.4219e+00,
-4.4753e+00, -4.0979e+00],
[-1.2974e+00, -3.9110e+00, -3.4754e+00, ..., -7.1947e+00,
-5.0864e+00, -4.8417e+00],
[-2.1733e+00, -4.8983e+00, -3.0637e+00, ..., -4.4658e+00,
-2.7228e+00, -3.4955e+00]],
[[-7.6391e-01, -1.4860e-01, -1.2801e-01, ..., 1.0100e+00,
9.8884e-01, 1.7961e+00],
[ 1.7978e+00, 1.1600e+00, 1.3423e+00, ..., 1.5886e+00,
1.5194e+00, 2.8771e+00],
[ 1.2130e+00, 2.8113e-01, 3.6884e-01, ..., 1.3778e+00,
8.5800e-01, 2.1595e+00],
...,
[ 1.2899e+00, -8.2589e-01, -1.0115e+00, ..., 1.4797e+00,
1.1241e+00, 1.2960e+00],
[ 1.9045e+00, 8.0145e-01, 1.8408e+00, ..., 1.9909e+00,
9.9928e-01, 2.3941e+00],
[ 1.0172e+00, 3.6508e-01, 8.4749e-01, ..., 9.7091e-01,
1.2145e+00, 1.5219e+00]],
[[-5.1096e-01, 5.4673e-01, 2.5122e-02, ..., -2.3469e-01,
-5.9279e-01, 8.4630e-01],
[-1.1177e+00, -1.2578e+00, -1.6697e+00, ..., -1.4404e-01,
1.1060e-02, 1.0983e+00],
[-2.0444e+00, -5.9143e-01, -5.5065e-01, ..., 3.8085e-01,
1.5875e-01, 4.5549e-01],
...,
[-1.7711e+00, -1.3515e+00, -2.6792e-01, ..., -1.0009e+00,
-1.4204e+00, -2.0198e+00],
[-1.7813e+00, -1.0651e+00, -7.7522e-01, ..., -1.2640e+00,
-1.0940e+00, 1.0595e-01],
[-3.5451e-01, 3.4406e-01, -4.4591e-01, ..., 1.7434e-01,
-1.3715e-01, 7.2037e-01]]],
[[[ 9.5836e-01, 1.5751e+00, 1.1452e+00, ..., 7.7107e-01,
9.7373e-01, 4.3499e-01],
[-8.7866e-01, -2.0441e+00, -8.7306e-01, ..., -2.8585e-01,
-1.6651e-01, -3.5431e-01],
[ 5.7346e-02, -5.5182e-01, 4.3740e-02, ..., -7.4068e-01,
-8.0154e-02, 6.3495e-01],
...,
[ 1.0297e-01, -2.3384e-01, 1.0768e+00, ..., 3.6167e-02,
-3.2220e-02, 4.0303e-01],
[-1.9319e-01, -1.0767e+00, 6.1031e-02, ..., 6.1444e-03,
8.4159e-02, -7.9599e-02],
[ 7.4520e-01, -2.9167e-01, -1.7480e-01, ..., -2.5361e-01,
-9.2336e-01, 1.7645e-01]],
[[-4.2398e+00, -1.2592e+00, -2.1492e+00, ..., -1.4993e+00,
-1.9991e+00, -1.7103e+00],
[-4.1247e+00, -2.8735e+00, -2.4215e+00, ..., -1.3180e+00,
-2.0923e+00, -2.0078e+00],
[-2.6131e+00, -2.2956e+00, -4.1106e+00, ..., -1.3612e+00,
-1.7509e+00, -1.3548e+00],
...,
[-1.0687e+00, -3.5344e-01, -3.9749e-02, ..., -3.4019e-01,
-8.1153e-01, -9.0714e-01],
[-2.9855e+00, -4.0679e-01, -6.9973e-01, ..., -6.7066e-01,
-1.0027e+00, -1.2833e+00],
[-2.0485e+00, -1.4607e+00, -1.3619e+00, ..., -6.0646e-01,
-6.8022e-01, -5.1080e-01]],
[[ 3.0460e+00, 4.1069e+00, 3.2272e+00, ..., 1.5388e+00,
8.8301e-01, 2.5516e+00],
[ 3.9001e-01, 7.8887e-01, 2.2256e+00, ..., 1.0370e+00,
7.3034e-01, 2.6191e+00],
[ 9.0371e-01, 1.2168e+00, 1.6424e+00, ..., 4.7025e-01,
8.5003e-01, 1.8505e+00],
...,
[ 1.8002e-01, 1.5211e+00, 1.3735e+00, ..., -4.6344e-01,
5.2128e-01, 1.5062e+00],
[ 3.8125e-01, 2.0770e+00, 1.3670e+00, ..., -1.9684e-01,
7.8350e-01, 1.2025e+00],
[ 1.4737e+00, 1.3374e+00, 1.6127e+00, ..., 5.8038e-01,
1.3135e+00, 1.3957e+00]],
...,
[[-2.1629e+00, -7.4933e+00, -5.9461e+00, ..., -5.0218e+00,
-3.6234e+00, -4.2297e+00],
[-2.8005e+00, -6.5848e+00, -4.5786e+00, ..., -5.2546e+00,
-3.6571e+00, -3.1692e+00],
[-2.5270e-01, -2.9271e+00, -3.9359e+00, ..., -3.6438e+00,
-2.5296e+00, -1.4182e+00],
...,
[-1.0192e+00, -2.1260e+00, -2.5074e+00, ..., -5.1855e+00,
-3.0726e+00, -2.8615e+00],
[-1.2950e+00, -3.7282e+00, -3.3571e+00, ..., -5.0512e+00,
-4.1003e+00, -4.3419e+00],
[-2.2645e+00, -5.0356e+00, -2.9319e+00, ..., -3.3620e+00,
-2.4584e+00, -2.4033e+00]],
[[-7.3960e-01, -7.8709e-02, 1.7557e-01, ..., 1.4451e+00,
1.7010e+00, 1.2052e+00],
[ 1.7925e+00, 1.0578e+00, 1.2382e+00, ..., 1.6743e+00,
1.2078e+00, 1.8417e+00],
[ 1.2538e+00, -2.4636e-01, 1.5714e-01, ..., 1.8521e+00,
4.8257e-01, 1.2590e+00],
...,
[ 1.3256e+00, 3.0310e-01, 1.3866e+00, ..., 1.6814e+00,
7.0417e-01, 8.9668e-01],
[ 1.8835e+00, 1.0433e+00, 1.8075e+00, ..., 1.3095e+00,
7.3709e-01, 1.4428e+00],
[ 1.0146e+00, 4.0639e-01, 9.4191e-01, ..., 7.9610e-01,
6.9578e-01, 1.4493e+00]],
[[-5.0669e-01, 6.0073e-01, 1.6297e-01, ..., -4.1245e-01,
-6.2437e-01, -7.3590e-02],
[-1.1234e+00, -1.1737e+00, -1.5519e+00, ..., -1.1062e+00,
-1.4383e+00, -1.1306e+00],
[-2.0688e+00, -9.7964e-01, -2.4498e-01, ..., -1.1492e+00,
-1.0433e+00, -2.7419e-01],
...,
[-1.7472e+00, -1.0999e+00, -7.8179e-01, ..., -4.9979e-01,
-1.3604e+00, -8.9692e-01],
[-1.8393e+00, -1.0252e+00, -6.7419e-01, ..., -8.5202e-01,
-1.5432e+00, -7.5727e-01],
[-3.6054e-01, 3.5273e-01, -2.4915e-01, ..., -2.4035e-01,
-7.8111e-01, -2.3964e-02]]],
[[[ 6.7339e-01, 1.0541e+00, 6.2412e-01, ..., -1.9594e+00,
3.6629e-01, 6.6767e-02],
[-4.9165e-01, -1.4813e+00, -5.5339e-01, ..., -2.6638e+00,
-8.4591e-01, 1.1585e-01],
[-1.2921e-01, -9.4937e-01, -5.7024e-01, ..., -3.4383e+00,
-4.3528e+00, -3.3248e+00],
...,
[ 4.1471e-01, -1.2207e+00, -3.9369e+00, ..., -6.2213e+00,
2.1893e-01, 4.0842e-01],
[-2.6819e-01, -2.2444e+00, -9.7077e-01, ..., -2.0806e+00,
-1.4539e+00, -7.1284e-01],
[ 2.2044e-01, -5.5695e-02, -4.3013e-01, ..., 1.3450e+00,
5.6430e-01, 7.3442e-01]],
[[-3.5778e+00, -1.5845e+00, -2.9270e+00, ..., -3.1504e+00,
-6.7438e+00, -4.2635e+00],
[-4.5912e+00, -1.8263e+00, -2.0801e+00, ..., -1.0870e+00,
-7.0392e+00, -3.3041e+00],
[-4.4847e+00, -2.2492e+00, -2.6667e+00, ..., -5.0789e+00,
-8.4335e+00, -2.8716e+00],
...,
[-3.8292e+00, -8.9827e+00, -1.5698e+01, ..., -5.4486e+00,
-9.7442e+00, -3.6953e+00],
[-6.5907e+00, -6.5658e+00, -8.1792e+00, ..., -5.2057e+00,
-7.3619e+00, -2.3284e+00],
[-1.6101e+00, -1.2450e+00, -2.4755e+00, ..., -1.0108e+00,
-1.7235e+00, -5.9650e-01]],
[[ 2.0868e+00, 2.9908e+00, 2.4575e+00, ..., 3.2786e+00,
3.8133e-01, 2.8736e+00],
[ 5.2966e-01, 1.3451e+00, 1.8030e+00, ..., -1.3512e+00,
-6.3134e+00, -9.6142e-01],
[ 9.4684e-01, 1.3951e+00, 8.4296e-01, ..., -5.5285e+00,
-9.0480e+00, -7.7033e+00],
...,
[-1.7539e+00, -7.2361e+00, -7.5195e+00, ..., -4.1820e+00,
-4.4592e+00, -7.5349e+00],
[-1.3303e+00, -2.0415e+00, 5.9976e-01, ..., -1.6994e+00,
-2.8876e+00, -3.8143e+00],
[ 2.4398e+00, 2.1967e+00, 1.9036e+00, ..., 2.0354e+00,
1.2043e+00, 6.1926e-01]],
...,
[[-1.6955e+00, -5.1113e+00, -4.0498e+00, ..., -4.4430e+00,
-3.1382e+00, -4.8515e+00],
[-2.1567e+00, -4.9031e+00, -4.8331e+00, ..., -1.0383e+01,
-5.8434e+00, -6.5655e+00],
[-1.3630e+00, -3.8994e+00, -3.5205e+00, ..., -1.9402e+01,
-3.9553e+00, -2.0914e+00],
...,
[-3.1777e+00, -3.6632e+00, -1.1371e+01, ..., -5.0629e+00,
-1.3424e+01, -7.9968e+00],
[-3.3412e+00, -6.8663e+00, -7.6384e+00, ..., -5.8789e+00,
-7.5811e+00, -8.5193e+00],
[-1.2709e+00, -5.5242e+00, -5.1749e+00, ..., -5.3298e+00,
-4.1253e+00, -6.3004e+00]],
[[ 4.8438e-01, 4.9733e-01, 1.8896e-02, ..., -2.0326e+00,
-4.1305e+00, -1.1592e-01],
[ 1.2403e+00, 1.0115e+00, 6.4245e-01, ..., -4.0701e-01,
-5.5028e+00, 4.2931e-01],
[ 6.6012e-01, 6.9954e-01, -5.0647e-02, ..., -1.5258e+00,
-6.5463e+00, -5.8990e-01],
...,
[-1.6212e+00, -3.2894e+00, -5.9218e+00, ..., 7.7801e-01,
-2.8708e+00, 1.6477e+00],
[-1.8589e-01, -3.6924e-01, 1.5921e+00, ..., 2.5512e+00,
3.8333e-01, 2.3133e+00],
[ 5.4244e-01, 9.0667e-01, 1.3443e+00, ..., 1.1939e+00,
1.7908e+00, 2.0092e+00]],
[[-3.9746e-01, -1.1241e-01, -7.9573e-01, ..., -2.2854e+00,
3.2690e-01, 8.2536e-01],
[-1.5718e+00, -1.1492e+00, -1.7462e+00, ..., -1.6288e+00,
6.9805e-01, 4.4920e+00],
[-1.6783e+00, -4.5780e-01, -8.6836e-01, ..., -4.6256e+00,
-2.8369e+00, 6.4114e+00],
...,
[-1.4588e+00, -1.6337e+00, -4.4760e+00, ..., -4.2770e+00,
-6.4236e+00, -4.1729e-01],
[-2.3364e+00, -4.9093e+00, -4.6164e+00, ..., -2.3597e+00,
-5.7340e+00, -1.1900e+00],
[-4.6066e-01, -3.0980e-01, -5.5850e-02, ..., -4.9504e-01,
-9.7668e-01, 1.3206e-01]]]], device='cuda:0'),
tensor([[[[ 9.1376e-01, 1.6332e+00, 1.0736e+00, ..., 1.6843e+00,
9.1081e-01, -4.6995e-01],
[-1.0366e+00, -2.0237e+00, -9.8774e-01, ..., -4.0832e-01,
1.5669e+00, -2.7137e-02],
[-3.9256e-01, -2.0840e-01, 3.7272e-01, ..., -1.7895e+00,
-6.5106e-01, 6.3794e-01],
...,
[ 1.6241e+00, -8.2057e-01, -3.4590e-01, ..., 1.0788e-01,
-1.0556e+00, 7.7137e-01],
[ 6.3558e-02, -1.4863e-01, 1.5777e-01, ..., 2.8153e-01,
-2.0943e+00, 5.2422e-02],
[ 7.9389e-01, -2.4989e-01, 5.4803e-02, ..., 9.4186e-01,
4.1994e-01, 2.4967e-02]],
[[-3.9508e+00, -1.3792e+00, -2.0015e+00, ..., -3.4528e+00,
-4.5956e+00, -9.1100e-01],
[-4.5357e+00, -3.5442e+00, -3.1564e+00, ..., -1.4959e+00,
-3.4355e+00, -1.0589e+00],
[-2.8714e+00, -9.4342e-01, -1.9545e+00, ..., -3.5785e+00,
-4.8467e+00, -2.7055e+00],
...,
[-3.6210e-01, 2.0981e-01, -2.0672e+00, ..., -2.2549e+00,
-1.9094e+00, -1.6506e+00],
[-3.1193e+00, 2.0166e-01, -5.2414e-01, ..., -2.2331e+00,
-3.1254e+00, -1.8605e+00],
[-2.4021e+00, -1.6651e+00, -1.5297e+00, ..., -2.6195e+00,
-3.9746e+00, -1.7663e+00]],
[[ 2.6632e+00, 3.8730e+00, 3.1866e+00, ..., 6.5010e-01,
1.7854e-01, 2.8238e+00],
[ 1.1983e-01, 8.3712e-01, 2.0833e+00, ..., -3.2322e-01,
-1.0116e+00, 2.0924e+00],
[ 7.6653e-01, 7.8218e-01, 1.0923e+00, ..., -1.9125e+00,
-8.3520e-01, 2.0977e+00],
...,
[-2.0549e-01, -2.4538e+00, -1.8840e+00, ..., 9.2616e-01,
1.0088e+00, -1.7130e+00],
[ 9.9659e-01, 2.3865e+00, 9.1986e-01, ..., -1.0567e+00,
-4.6020e+00, -1.2815e+00],
[ 1.7182e+00, 1.6025e+00, 1.9466e+00, ..., -1.1700e+00,
-6.4428e+00, -2.1699e+00]],
...,
[[-1.6791e+00, -6.9330e+00, -5.6234e+00, ..., -2.8285e+00,
-2.5428e+00, -3.4054e+00],
[-2.4401e+00, -5.9644e+00, -3.6789e+00, ..., -1.6003e+00,
-3.7917e+00, -6.7066e+00],
[-1.2053e+00, -2.9655e+00, -2.7077e+00, ..., 5.8827e-01,
8.8688e-01, -1.9679e+00],
...,
[-1.3300e+00, -3.6345e+00, -4.2441e+00, ..., -1.0343e+00,
4.0105e-01, -1.5488e+00],
[-1.2539e+00, -4.9117e+00, -3.9685e+00, ..., -4.1490e+00,
-4.4986e-01, -4.6443e+00],
[-2.3362e+00, -5.2721e+00, -3.0665e+00, ..., -4.2323e+00,
-2.5304e+00, -5.2241e+00]],
[[-7.1230e-01, -2.0694e-01, 1.0535e-01, ..., -1.1335e+00,
1.0662e-01, 2.1491e+00],
[ 1.4185e+00, 8.2746e-01, 1.1229e+00, ..., 2.1765e-01,
-2.7400e-01, 2.3356e+00],
[ 3.5829e-01, 2.7817e-01, -4.1299e-02, ..., -6.3513e-01,
-1.9243e+00, 6.5083e-01],
...,
[ 7.0181e-01, 7.4947e-01, 2.3920e+00, ..., -2.9585e+00,
-4.1859e+00, -1.1911e+00],
[ 1.7428e+00, 8.7539e-01, 1.8797e+00, ..., -2.2642e+00,
-6.9054e-01, 1.0023e+00],
[ 1.0533e+00, 4.9924e-01, 1.0760e+00, ..., -7.1345e-01,
6.7586e-01, 5.2940e-01]],
[[-4.0872e-01, 6.7447e-01, 1.7433e-01, ..., 3.2919e-01,
-2.3965e-01, 9.7351e-01],
[-1.0311e+00, -9.4872e-01, -1.3490e+00, ..., 1.7612e-01,
1.0455e+00, 1.1151e+00],
[-2.0051e+00, -4.7418e-01, -6.5202e-01, ..., -1.9533e+00,
-7.5699e-01, -2.5953e-01],
...,
[-6.0828e-01, -1.0673e+00, -1.0149e+00, ..., -3.7232e+00,
-1.5038e+00, -1.8042e-01],
[-1.5556e+00, -6.8305e-01, -7.7405e-01, ..., -2.6814e+00,
-1.0396e+00, 1.1804e+00],
[-4.6165e-01, 3.3209e-01, -3.7688e-01, ..., -1.1695e+00,
-1.8743e+00, -1.1797e+00]]],
[[[ 6.0665e-01, 8.5859e-01, 5.3819e-01, ..., 3.6359e-01,
4.6694e-01, 3.1968e-02],
[-2.4630e-01, -8.1188e-01, -1.9914e-01, ..., 8.8856e-02,
6.7463e-02, -3.6362e-01],
[-2.2897e-01, 2.1451e-01, 2.5420e-01, ..., -3.5981e-01,
7.3244e-02, 6.5571e-01],
...,
[ 4.2879e-01, -5.0973e-01, 1.1104e+00, ..., 3.8407e-01,
8.2953e-01, 6.4665e-01],
[-6.5175e-02, -7.8714e-01, 1.3602e-01, ..., 8.8386e-01,
1.5559e+00, 1.0760e-01],
[ 8.2512e-01, -2.1239e-01, -9.2493e-02, ..., 3.7755e-01,
2.8570e-01, 7.2159e-01]],
[[-2.3098e+00, -6.0001e-01, -1.9078e+00, ..., -1.6950e+00,
-2.2990e+00, -1.5539e+00],
[-2.4332e+00, -7.4441e-01, -2.4445e+00, ..., -1.3894e+00,
-2.5125e+00, -1.7334e+00],
[-2.4561e+00, -1.8176e+00, -4.0694e+00, ..., -2.1124e+00,
-2.3988e+00, -1.1755e+00],
...,
[-6.6469e-01, -8.2306e-01, -1.6243e+00, ..., -6.7651e-01,
-2.8154e-01, 7.1197e-01],
[-2.5052e+00, -5.0161e-01, -3.0928e-01, ..., -4.2435e-02,
-4.6845e-02, 6.8372e-01],
[-1.5277e+00, -1.1877e+00, -9.3901e-01, ..., -4.5466e-01,
-5.8865e-01, -1.7071e-01]],
[[ 1.8595e+00, 2.5011e+00, 1.6202e+00, ..., 1.5607e+00,
7.9838e-01, 2.0850e+00],
[-3.1381e-01, 2.6683e-01, 4.6949e-01, ..., 6.6974e-01,
3.0153e-01, 2.2689e+00],
[ 5.1643e-03, 2.7921e-01, 9.9654e-01, ..., 3.3021e-02,
7.5435e-01, 1.9784e+00],
...,
[ 7.5814e-01, 8.5969e-01, 2.6862e+00, ..., 4.4377e-01,
1.5256e-01, 1.6512e+00],
[ 1.0864e+00, 2.1938e+00, 2.3651e+00, ..., 1.0315e+00,
7.7168e-01, 1.6797e+00],
[ 2.3819e+00, 1.8315e+00, 2.4004e+00, ..., 1.9823e+00,
1.0357e+00, 1.7009e+00]],
...,
[[-2.0104e+00, -4.9702e+00, -3.1121e+00, ..., -1.9297e+00,
-1.6214e+00, -3.0987e+00],
[-2.4487e+00, -5.3831e+00, -3.0213e+00, ..., -2.3674e+00,
-2.2015e+00, -3.0072e+00],
[-1.5749e+00, -5.5278e+00, -3.8063e+00, ..., -2.6976e+00,
-2.5115e+00, -2.2851e+00],
...,
[-2.5271e+00, -4.8511e+00, -2.9843e+00, ..., -2.8669e+00,
-2.1906e+00, -2.6747e+00],
[-2.4831e+00, -5.6927e+00, -4.1091e+00, ..., -4.4075e+00,
-3.1924e+00, -4.8903e+00],
[-2.5997e+00, -5.8762e+00, -2.8611e+00, ..., -3.0125e+00,
-1.6958e+00, -3.3123e+00]],
[[ 9.4118e-01, 1.6233e+00, 7.8733e-01, ..., 1.0982e+00,
1.5175e+00, 1.3898e+00],
[ 1.1344e+00, 9.6817e-01, 5.8462e-01, ..., 8.6549e-01,
7.9075e-01, 1.9724e+00],
[ 6.2220e-01, -1.8739e-01, -5.9978e-01, ..., 7.3763e-01,
4.9544e-02, 1.3911e+00],
...,
[ 1.4882e+00, -1.8898e-01, 1.3374e-01, ..., 1.4929e+00,
1.7985e+00, 1.5451e+00],
[ 2.0530e+00, 1.2053e+00, 2.1521e+00, ..., 1.9397e+00,
1.7255e+00, 2.5771e+00],
[ 1.2170e+00, 9.2387e-01, 1.3054e+00, ..., 1.3398e+00,
1.4930e+00, 1.5212e+00]],
[[-5.2476e-02, -6.4457e-01, -1.1925e+00, ..., -4.8229e-01,
-8.6700e-01, -1.0860e-01],
[-8.0072e-01, -2.6060e+00, -2.9815e+00, ..., -1.0618e+00,
-1.0992e+00, -7.0403e-01],
[-1.4577e+00, -2.2081e+00, -1.4965e+00, ..., -1.7455e-01,
-4.3754e-01, -2.6514e-01],
...,
[-1.6904e+00, -1.5048e+00, 4.8840e-01, ..., -3.2062e-01,
-4.1697e-01, -4.3264e-01],
[-1.7212e+00, -1.2761e+00, -1.2005e-01, ..., -3.6064e-01,
-3.5507e-01, 4.4219e-01],
[-2.4340e-01, 4.4395e-01, -1.8355e-01, ..., 2.1372e-01,
3.1052e-02, 6.2948e-01]]],
[[[ 1.8795e-02, 2.2303e+00, 9.3129e-01, ..., 4.6884e-01,
7.7521e-01, 3.6936e-01],
[-8.5581e+00, -6.4713e+00, -6.9324e+00, ..., -1.6067e+00,
-4.2339e-01, -8.7153e-01],
[-8.9144e+00, -1.5812e+01, -1.3763e+01, ..., -1.4534e+00,
-1.0535e+00, 3.9426e-01],
...,
[ 2.7857e-02, -2.0818e-01, 1.6125e+00, ..., 1.6657e-01,
-2.6883e-01, 1.4679e-01],
[-6.6020e-02, -9.7793e-01, -4.2893e-01, ..., 1.7075e-01,
-4.0551e-02, -2.2038e-01],
[ 7.8064e-01, -4.5592e-01, -2.9813e-01, ..., -2.8090e-01,
-9.7669e-01, 3.5998e-02]],
[[-9.5342e+00, 3.3399e-01, -2.2202e-01, ..., -1.7779e+00,
-2.6642e+00, -1.5253e+00],
[-2.7102e+00, -2.9157e+00, -7.5008e+00, ..., -1.5739e+00,
-2.6027e+00, -2.0145e+00],
[-1.0059e+01, -8.3489e+00, -2.9164e+00, ..., -3.4024e+00,
-2.0022e+00, -1.1494e+00],
...,
[-1.9017e+00, -1.2423e+00, -1.3142e-01, ..., -4.7441e-01,
-7.3970e-01, -3.5295e-01],
[-3.2542e+00, -8.9090e-01, -1.1405e+00, ..., -5.3285e-01,
-9.6239e-01, -7.1674e-01],
[-2.6200e+00, -2.0674e+00, -1.2094e+00, ..., -5.9247e-01,
-4.9929e-01, -2.7523e-01]],
[[-2.3863e+00, -5.6509e+00, -8.4856e+00, ..., 1.2426e+00,
2.3551e-01, 2.6328e+00],
[-3.2611e+00, 1.8537e-01, -1.3979e+00, ..., 2.6479e+00,
3.8486e-01, 2.9121e+00],
[-5.2094e+00, -4.0959e+00, -9.3822e-01, ..., 1.5816e+00,
1.4665e+00, 2.6221e+00],
...,
[-3.9920e-01, 1.8944e+00, 1.8492e+00, ..., -3.3335e-01,
7.8482e-01, 1.9830e+00],
[-6.7808e-02, 2.1228e+00, 1.7849e+00, ..., 1.3231e-01,
1.1520e+00, 1.6306e+00],
[ 1.2281e+00, 1.2746e+00, 1.6650e+00, ..., 7.3253e-01,
1.3966e+00, 1.5444e+00]],
...,
[[-5.9848e+00, -1.4138e+01, -1.3315e+01, ..., -4.9621e+00,
-4.6219e+00, -5.2992e+00],
[-2.6863e+00, -2.2409e+01, -2.7484e+01, ..., -6.6059e+00,
-6.6367e+00, -5.1929e+00],
[ 6.6955e+00, -8.3936e+00, -1.3899e+01, ..., -9.0946e+00,
-5.7423e+00, -2.4668e+00],
...,
[-1.2244e+00, -3.9182e+00, -5.7894e+00, ..., -7.6333e+00,
-5.8256e+00, -3.8951e+00],
[-1.1564e+00, -3.7635e+00, -3.6956e+00, ..., -5.6264e+00,
-5.2443e+00, -4.9579e+00],
[-2.0293e+00, -5.2090e+00, -3.5439e+00, ..., -3.9624e+00,
-3.0920e+00, -2.9759e+00]],
[[-6.5150e+00, -3.4109e+00, -5.5218e-01, ..., 9.7530e-01,
1.3657e+00, 1.4696e+00],
[ 2.1207e+00, -1.9815e+00, -3.1214e+00, ..., 3.2978e+00,
1.4582e+00, 2.1450e+00],
[-4.3147e+00, -7.9349e+00, -8.6600e+00, ..., 2.5795e+00,
1.9345e+00, 1.5964e+00],
...,
[ 1.1214e+00, 8.1775e-01, 1.7749e+00, ..., 2.6700e+00,
1.1073e+00, 1.5363e+00],
[ 1.7350e+00, 4.0511e-01, 1.7164e+00, ..., 1.5809e+00,
9.3532e-01, 1.9859e+00],
[ 9.1243e-01, -1.7735e-02, 9.8293e-01, ..., 8.7410e-01,
8.1986e-01, 1.5938e+00]],
[[-2.7475e-01, 2.6903e-01, 1.5925e+00, ..., -4.5864e-01,
-5.2226e-01, 3.6599e-01],
[-9.7802e-01, -2.8416e+00, -4.0482e+00, ..., -8.7655e-01,
-8.6955e-01, -9.1458e-01],
[-3.1744e+00, -8.8695e+00, -1.1084e+01, ..., -2.1650e+00,
-1.2504e+00, -3.1992e-01],
...,
[-1.0142e+00, -1.3790e+00, -1.6314e+00, ..., 2.7880e-01,
-9.6337e-01, -7.7568e-01],
[-1.5952e+00, -7.7216e-01, -9.3384e-01, ..., -4.6167e-01,
-1.3812e+00, -4.5050e-01],
[-2.0578e-01, 2.7933e-01, -2.2086e-01, ..., -2.4776e-01,
-6.7463e-01, 1.2849e-01]]],
[[[ 8.2333e-01, 1.0377e+00, 5.7382e-01, ..., 6.3529e-01,
9.9424e-01, 4.6984e-01],
[-1.2232e+00, -1.1254e+00, -7.1575e-02, ..., -4.8161e-01,
-3.0619e-01, -3.4840e-01],
[-2.4342e-01, -1.0418e+00, -1.6338e-01, ..., -2.0219e+00,
-3.7319e-01, 4.8165e-01],
...,
[-2.9843e-01, -5.1916e-01, 6.8330e-01, ..., -4.4203e-01,
-7.1371e-03, 3.5213e-01],
[ 5.7803e-01, -1.5203e-01, 2.4817e-01, ..., -1.1904e-01,
1.0570e-01, -6.4806e-02],
[ 4.6925e-01, -2.5131e-01, -7.5127e-01, ..., -4.0244e-01,
-8.8389e-01, 1.1601e-01]],
[[-4.0905e+00, -2.1237e+00, -3.0363e+00, ..., -1.5846e+00,
-2.1087e+00, -1.8279e+00],
[-3.1923e+00, -1.8507e+00, -1.0101e+00, ..., -1.4242e+00,
-2.3036e+00, -2.1406e+00],
[-2.0477e+00, -1.5943e+00, -1.4502e+00, ..., -1.2936e+00,
-1.9257e+00, -1.4760e+00],
...,
[-1.9644e+00, -2.0532e+00, -2.4777e+00, ..., -1.5620e+00,
-8.9504e-01, -9.5988e-01],
[-1.6280e+00, -3.7419e-01, -8.5574e-01, ..., -7.7502e-01,
-7.4306e-01, -1.1385e+00],
[ 1.3947e-01, -8.0500e-01, -8.9858e-01, ..., -6.7621e-01,
-6.4096e-01, -5.1756e-01]],
[[ 2.7614e+00, 3.3732e+00, 3.2248e+00, ..., 1.5119e+00,
8.3947e-01, 2.5842e+00],
[ 7.8060e-01, 1.1500e+00, 2.7333e+00, ..., 1.1839e+00,
4.7154e-01, 2.6388e+00],
[ 1.1481e+00, 7.2958e-01, 2.0374e+00, ..., 2.3091e+00,
1.1570e+00, 1.7965e+00],
...,
[-2.3932e+00, -2.7446e+00, -2.0209e+00, ..., -6.1825e-01,
6.3181e-01, 1.4239e+00],
[ 5.4818e-01, 1.8255e+00, 1.7593e+00, ..., -4.0480e-01,
7.7060e-01, 1.1264e+00],
[ 1.7030e+00, 1.2386e+00, 1.9618e+00, ..., 3.2757e-01,
1.2180e+00, 1.3483e+00]],
...,
[[-2.9034e+00, -8.1912e+00, -5.2563e+00, ..., -5.6898e+00,
-3.6376e+00, -4.3631e+00],
[-3.3705e+00, -9.4039e+00, -6.3506e+00, ..., -6.3013e+00,
-3.9614e+00, -3.3312e+00],
[-2.2716e+00, -6.3908e+00, -3.6612e+00, ..., -5.1393e+00,
-2.6042e+00, -1.5706e+00],
...,
[-2.8167e+00, -6.2961e+00, -4.2971e+00, ..., -4.7886e+00,
-2.8624e+00, -2.7680e+00],
[-3.0553e+00, -6.3676e+00, -3.7922e+00, ..., -6.0491e+00,
-4.2042e+00, -4.2682e+00],
[-2.3834e+00, -6.6711e+00, -3.4220e+00, ..., -3.5390e+00,
-2.3261e+00, -2.3564e+00]],
[[-8.2738e-01, -1.5944e-01, 3.0526e-01, ..., 1.9180e+00,
1.7019e+00, 1.1775e+00],
[ 1.4293e+00, 9.6234e-01, 1.3019e+00, ..., 2.0623e+00,
1.0546e+00, 1.8536e+00],
[ 1.1536e+00, 2.8467e-01, 3.6993e-01, ..., 3.0106e+00,
1.0168e+00, 1.2339e+00],
...,
[ 7.7594e-01, -8.5919e-01, -1.0163e+00, ..., 1.6785e+00,
9.1975e-01, 9.6063e-01],
[ 1.6956e+00, 1.1228e+00, 1.0826e+00, ..., 1.8587e+00,
8.1339e-01, 1.4381e+00],
[ 1.2049e+00, 9.5665e-01, 1.2074e+00, ..., 8.5575e-01,
6.6818e-01, 1.3929e+00]],
[[-7.4846e-01, -7.6522e-01, -5.5936e-01, ..., -3.1102e-01,
-6.0841e-01, -4.9063e-02],
[-1.4113e+00, -1.5095e+00, -6.1382e-01, ..., -1.7614e+00,
-1.3993e+00, -1.0930e+00],
[-2.2100e+00, -6.8123e-01, 3.4397e-01, ..., -2.1345e+00,
-1.1207e+00, -1.8568e-01],
...,
[-1.7020e+00, -2.6145e+00, -2.3167e+00, ..., -1.4551e+00,
-1.3467e+00, -8.2308e-01],
[-2.2270e+00, -1.4110e+00, -8.4288e-01, ..., -1.0678e+00,
-1.2343e+00, -6.1816e-01],
[ 4.5606e-02, 3.7392e-02, -2.4225e-01, ..., -4.3064e-01,
-7.9896e-01, -5.6354e-03]]],
[[[ 8.4903e-01, 1.5223e+00, 1.2013e+00, ..., 1.4245e+00,
1.0885e+00, 9.7499e-01],
[-1.0170e+00, -1.9454e+00, -1.0894e+00, ..., -3.3953e-01,
-5.9283e-01, -1.6147e+00],
[-1.6368e-01, -8.0620e-01, -4.1814e-02, ..., 3.7396e-02,
7.0711e-01, 9.0577e-01],
...,
[ 5.3978e-01, -2.6677e-01, 1.3705e-01, ..., 2.1928e-01,
1.7219e-01, 6.6930e-01],
[ 3.7718e-01, -7.4550e-01, 1.6864e-01, ..., 6.7612e-01,
5.7280e-01, -7.7218e-02],
[ 6.1621e-01, 2.1106e-01, 2.0063e-01, ..., 3.8868e-01,
-3.7986e-02, 6.5878e-01]],
[[-3.8017e+00, -8.5265e-01, -1.6324e+00, ..., -1.6033e+00,
-2.4629e+00, -1.1867e+00],
[-3.9578e+00, -2.7978e+00, -1.8743e+00, ..., -2.1938e+00,
-3.0746e+00, -9.3209e-01],
[-2.4380e+00, -1.7403e+00, -1.6318e+00, ..., -2.7417e+00,
-2.2808e+00, -8.6065e-01],
...,
[-1.0486e+00, -1.2767e+00, -1.7432e+00, ..., -2.0649e+00,
-1.1342e+00, -1.4267e-01],
[-1.2990e+00, -3.9946e-01, -6.4076e-01, ..., -5.0106e-01,
-1.0783e+00, -3.7078e-02],
[ 5.5705e-01, -2.5778e-01, -1.2844e-01, ..., 3.4869e-01,
1.0792e-01, 4.5091e-01]],
[[ 2.9002e+00, 4.2293e+00, 3.3866e+00, ..., 2.8715e+00,
1.0274e+00, 2.9767e+00],
[ 1.8711e-01, 1.1554e+00, 2.3309e+00, ..., 1.3139e+00,
-8.0828e-01, 2.6987e+00],
[ 9.3440e-01, 1.5573e+00, 1.7427e+00, ..., 7.6074e-01,
4.3388e-01, 2.0428e+00],
...,
[ 4.9409e-01, 6.8302e-01, 7.6730e-01, ..., 5.4812e-01,
3.7724e-01, 1.4824e+00],
[ 1.7147e-01, 1.4716e+00, 5.6492e-01, ..., -1.6662e-01,
7.2843e-02, 1.1304e+00],
[ 1.3471e+00, 1.1549e+00, 6.9404e-01, ..., 2.3860e-01,
4.0302e-01, 1.1386e+00]],
...,
[[-1.8370e+00, -7.2400e+00, -5.4545e+00, ..., -4.0642e+00,
-3.2795e+00, -5.5264e+00],
[-2.3768e+00, -7.0034e+00, -4.3279e+00, ..., -3.6433e+00,
-2.7035e+00, -4.2845e+00],
[-1.0621e+00, -4.6310e+00, -3.6338e+00, ..., -2.7375e+00,
-2.6708e+00, -2.5252e+00],
...,
[-2.2831e+00, -5.4071e+00, -3.9711e+00, ..., -3.9246e+00,
-3.7674e+00, -3.4967e+00],
[-2.3982e+00, -6.5634e+00, -4.9508e+00, ..., -6.3294e+00,
-4.4122e+00, -4.8360e+00],
[-2.3732e+00, -6.6181e+00, -4.0504e+00, ..., -4.8321e+00,
-3.1890e+00, -4.1527e+00]],
[[-2.2878e-01, 2.6757e-01, 5.6197e-01, ..., 6.5870e-01,
8.6374e-01, 1.4153e+00],
[ 1.6661e+00, 1.1160e+00, 1.8104e+00, ..., 1.8933e+00,
2.5360e+00, 3.0237e+00],
[ 1.1048e+00, 2.5780e-01, 8.5473e-01, ..., 9.3680e-01,
1.4439e+00, 2.4308e+00],
...,
[ 1.0855e+00, -1.6580e-01, 2.1382e-01, ..., 1.6102e+00,
1.0144e+00, 1.2795e+00],
[ 1.6779e+00, 6.7433e-01, 1.1891e+00, ..., 1.9934e+00,
6.8542e-01, 2.0509e+00],
[ 1.1644e+00, 7.2025e-01, 1.7747e+00, ..., 1.5160e+00,
1.2703e+00, 1.3808e+00]],
[[-5.7884e-01, 2.8445e-01, 1.5884e-01, ..., 5.7383e-01,
5.2394e-01, 1.3564e+00],
[-1.3674e+00, -1.7423e+00, -1.5545e+00, ..., -6.1956e-01,
-4.1165e-01, 2.0239e-01],
[-2.1443e+00, -8.2644e-01, 2.9229e-01, ..., 8.5501e-01,
2.1443e-01, 5.2784e-01],
...,
[-1.5384e+00, -1.8613e+00, -1.2007e+00, ..., -8.3775e-01,
-8.6276e-01, -7.2458e-01],
[-1.2908e+00, -1.8023e+00, -1.0123e+00, ..., -4.8747e-01,
-1.0542e+00, -6.2758e-02],
[ 1.7463e-01, 1.0325e-01, 1.9180e-02, ..., -3.5719e-01,
-8.3342e-04, 7.8348e-01]]]], device='cuda:0'),
tensor([[[[ 1.3243e+00, 6.0777e-01, 9.1652e-01, ..., -2.1760e+00,
1.1328e+00, 1.0573e+00],
[-8.1968e-01, -3.2552e+00, 7.2781e-01, ..., -6.3145e+00,
-9.5420e-01, -4.4789e-01],
[-8.1987e-01, -2.2362e+00, 9.0524e-02, ..., -5.7162e+00,
2.1082e-01, -7.3455e-01],
...,
[-5.5518e+00, -5.6912e+00, -2.5594e-01, ..., 1.9941e+00,
-4.0353e+00, 5.9967e-01],
[-4.9314e+00, -4.8647e+00, -2.3858e+00, ..., 5.3480e-01,
-1.4122e+00, 8.0384e-02],
[-1.1198e+00, -4.4839e+00, 2.0186e-01, ..., 5.2834e+00,
1.2939e-01, 9.8277e-01]],
[[-3.1288e+00, -1.9557e+00, -2.1437e+00, ..., -4.3426e+00,
-4.9348e+00, -3.1876e+00],
[-3.5237e+00, -4.1032e+00, -2.6187e+00, ..., -1.1883e+01,
-1.0784e+01, -4.6459e+00],
[-3.7591e+00, -4.6961e+00, -5.3714e-01, ..., -1.4311e+01,
-1.1916e+01, -2.5448e+00],
...,
[ 1.5862e-01, -2.1482e+00, -1.9338e+01, ..., -1.7658e+01,
-2.8552e+00, 8.4284e-01],
[-2.6312e+00, -2.8878e+00, -2.9912e+01, ..., -1.3378e+01,
9.6080e-01, 4.6778e-01],
[-8.6745e+00, -8.8354e+00, -1.0045e+01, ..., -1.3253e+01,
-8.1856e-02, -6.7069e-01]],
[[ 3.2608e+00, 2.4352e+00, -2.0675e+00, ..., -3.6509e+00,
-1.6913e+00, 9.2264e-01],
[ 1.6092e+00, 9.8681e-02, -9.7256e+00, ..., -9.8249e+00,
-4.5027e+00, -1.1758e+00],
[ 2.8875e+00, 1.1959e-02, -5.8834e+00, ..., -1.1379e+01,
5.8055e-01, 1.8752e+00],
...,
[-5.3320e+00, -3.8213e+00, -1.5717e+01, ..., 2.5025e+00,
-6.3267e+00, 1.9805e+00],
[-8.1265e+00, -8.4185e+00, -1.7116e+01, ..., 6.2523e+00,
-3.1212e+00, 2.0967e+00],
[-1.2602e+01, -7.6152e+00, -5.9653e+00, ..., 2.3297e+00,
-3.2295e+00, 2.0453e+00]],
...,
[[-2.9885e+00, -9.0436e+00, -5.2175e+00, ..., 1.7400e+00,
-5.0778e+00, -1.2172e+01],
[-3.3291e+00, -1.0688e+01, -7.2548e+00, ..., 1.3021e+00,
-1.3646e+01, -1.4625e+01],
[-2.2978e+00, -7.4056e+00, -5.2382e+00, ..., 9.8945e-01,
-1.6798e+01, -1.0627e+01],
...,
[-8.5179e+00, -1.6759e+01, -1.2938e+00, ..., -4.9267e+01,
-2.7015e+01, -2.2900e+00],
[-2.0411e+00, -1.9803e+01, -8.9953e+00, ..., -3.2758e+01,
-1.9769e+01, -4.7830e+00],
[-1.1042e+00, -1.6830e+01, -1.1589e+01, ..., -3.2903e+01,
-1.2600e+01, -3.7220e+00]],
[[-6.5632e-01, -8.1776e-01, -2.2171e+00, ..., -3.7590e+00,
-4.3320e+00, 4.9339e-01],
[ 8.2285e-01, -1.2103e+00, -6.6427e-01, ..., -5.8036e+00,
-5.1822e+00, 1.9543e+00],
[-9.0011e-03, -2.2817e-01, 1.3491e+00, ..., -9.1998e+00,
-4.2303e+00, 3.4970e+00],
...,
[-5.9200e+00, -6.4958e+00, -5.0097e+00, ..., -9.8092e+00,
1.0900e+01, 1.7158e+00],
[-6.5324e-01, -2.0660e+00, -2.6124e+00, ..., -1.0272e+01,
5.3545e+00, 2.8039e+00],
[ 1.1690e+00, -2.7150e+00, -4.6865e+00, ..., -4.6420e+00,
2.8732e+00, 1.3420e+00]],
[[-1.3359e+00, -1.3187e+00, 1.0125e+00, ..., 1.5991e+00,
1.7834e+00, 1.5838e+00],
[-2.2051e+00, -2.9295e+00, 8.7044e-01, ..., 6.9906e-01,
-1.3261e+00, -3.2679e+00],
[-2.4304e+00, -2.2939e+00, 2.5840e+00, ..., 5.8472e-01,
-3.0034e+00, -2.0498e+00],
...,
[-1.3495e+00, -9.1452e+00, -9.1056e+00, ..., -6.4974e+00,
-1.0053e+01, -4.0489e-01],
[-4.4368e-01, -3.6656e+00, -8.9733e+00, ..., -1.2208e+00,
-5.3729e+00, 7.2334e-01],
[-5.6138e-01, -3.4022e+00, -8.3763e+00, ..., 1.2518e+00,
-5.2133e+00, 5.9443e-01]]],
[[[ 1.0586e+00, 1.3632e+00, 1.2277e+00, ..., 1.0967e+00,
9.3813e-01, 4.5752e-01],
[-6.3225e-01, -2.5172e+00, -1.1932e+00, ..., 8.9223e-01,
3.6331e-02, -1.4123e+00],
[-4.8644e-01, -1.1846e+00, -6.2739e-01, ..., 1.1313e-02,
2.3041e-01, 6.6788e-01],
...,
[ 5.4353e-01, 1.1879e-02, -2.1196e-01, ..., 4.1064e-01,
1.0555e+00, 8.7749e-01],
[-3.5580e-02, -4.0822e-01, 1.8756e-01, ..., 9.1858e-01,
1.5710e+00, 7.7304e-02],
[ 7.8308e-01, -6.1370e-02, -5.8943e-02, ..., 3.7448e-01,
2.8389e-01, 7.1614e-01]],
[[-2.9472e+00, 1.5570e-02, -1.5102e+00, ..., -1.7442e+00,
-2.5281e+00, -7.0207e-01],
[-4.2688e+00, -2.7923e+00, -2.6440e+00, ..., -3.9629e+00,
-3.9694e+00, -1.5710e+00],
[-3.1458e+00, -2.4722e+00, -2.8731e+00, ..., -4.3380e+00,
-2.8494e+00, -1.3791e+00],
...,
[-5.0619e-01, -3.8866e-01, -1.0598e+00, ..., -3.7200e-01,
2.4693e-01, 8.7136e-01],
[-1.9913e+00, -4.4927e-01, -2.1811e-01, ..., 1.6228e-01,
1.0463e-01, 7.5601e-01],
[-1.2083e+00, -1.1740e+00, -9.6139e-01, ..., -2.8572e-01,
-4.4851e-01, -1.3845e-01]],
[[ 2.5308e+00, 2.9912e+00, 2.4852e+00, ..., 2.0083e+00,
4.0917e-01, 2.6090e+00],
[ 8.4678e-01, 6.2510e-01, 2.3922e+00, ..., 1.2313e+00,
3.9895e-01, 2.7410e+00],
[ 1.1475e+00, 1.4617e+00, 2.1087e+00, ..., 5.9156e-01,
1.9174e+00, 2.8948e+00],
...,
[ 5.5514e-01, 7.4351e-01, 1.5021e+00, ..., 9.4342e-01,
5.2515e-01, 1.8508e+00],
[ 7.5299e-01, 2.0406e+00, 1.8489e+00, ..., 1.1958e+00,
7.5180e-01, 1.6856e+00],
[ 2.0934e+00, 2.0179e+00, 2.4146e+00, ..., 2.0939e+00,
1.0814e+00, 1.7876e+00]],
...,
[[-2.8266e+00, -7.1802e+00, -6.0297e+00, ..., -2.8747e+00,
-3.8944e+00, -5.4064e+00],
[-3.4596e+00, -7.7939e+00, -5.5582e+00, ..., -6.8914e+00,
-5.0700e+00, -5.1333e+00],
[-2.2625e+00, -6.4043e+00, -4.4057e+00, ..., -4.2185e+00,
-4.0991e+00, -3.5496e+00],
...,
[-2.1699e+00, -4.3715e+00, -2.3792e+00, ..., -3.4804e+00,
-2.9166e+00, -3.0877e+00],
[-2.0271e+00, -4.8410e+00, -3.7829e+00, ..., -4.5516e+00,
-3.2638e+00, -5.0048e+00],
[-2.1594e+00, -5.2810e+00, -2.7995e+00, ..., -3.0389e+00,
-1.6488e+00, -3.4877e+00]],
[[ 3.2110e-01, 1.1607e+00, 3.2739e-01, ..., -6.5375e-02,
1.5528e+00, 1.6866e+00],
[ 1.2901e+00, 8.9368e-01, 2.6590e-01, ..., -2.8850e-01,
1.3855e+00, 2.8049e+00],
[ 1.3386e+00, 8.0122e-01, 4.7553e-01, ..., 4.2228e-01,
1.3428e+00, 2.0402e+00],
...,
[ 1.3899e+00, 4.1621e-01, 1.0883e+00, ..., 1.9614e+00,
2.0239e+00, 1.3923e+00],
[ 1.9032e+00, 9.4039e-01, 1.7336e+00, ..., 2.0220e+00,
1.6898e+00, 2.5044e+00],
[ 1.2791e+00, 6.8855e-01, 1.2058e+00, ..., 1.4791e+00,
1.5924e+00, 1.5252e+00]],
[[-6.1984e-01, 4.5019e-02, -2.3897e-01, ..., 4.0982e-01,
4.3922e-02, 9.2122e-01],
[-1.4968e+00, -2.1904e+00, -1.7158e+00, ..., 1.0897e-01,
-9.6432e-01, -6.6844e-02],
[-2.3358e+00, -1.3834e+00, -4.9464e-01, ..., -1.3132e+00,
-4.8426e-01, -9.8531e-02],
...,
[-1.7240e+00, -1.2957e+00, -7.7189e-01, ..., -2.3147e-01,
-4.3843e-01, -4.2517e-01],
[-1.5537e+00, -1.1852e+00, -6.1370e-01, ..., -4.9222e-01,
-4.5997e-01, 3.8223e-01],
[-3.1780e-02, 5.4220e-01, -2.1954e-01, ..., 1.0252e-01,
3.7047e-02, 6.6493e-01]]],
[[[ 8.5826e-01, 1.4848e+00, 1.1711e+00, ..., 1.4172e+00,
1.0868e+00, 9.6052e-01],
[-1.0063e+00, -2.0770e+00, -1.3013e+00, ..., -3.6834e-01,
-5.8502e-01, -1.5973e+00],
[-6.2380e-02, -4.9768e-01, 3.1279e-01, ..., -8.6760e-02,
7.0955e-01, 8.6644e-01],
...,
[ 1.6482e-02, -3.1936e-01, 1.3611e+00, ..., 5.4913e-01,
1.1952e-01, 6.1762e-01],
[-1.8263e+00, -4.3401e+00, -3.1612e+00, ..., 7.0379e-01,
6.2458e-01, -4.7896e-02],
[ 3.2079e-01, -2.0823e+00, -2.2915e+00, ..., 3.1834e-01,
-9.1497e-02, 6.5030e-01]],
[[-3.8178e+00, -7.9522e-01, -1.6243e+00, ..., -1.5872e+00,
-2.5387e+00, -1.2853e+00],
[-4.0069e+00, -2.8413e+00, -1.8214e+00, ..., -2.1574e+00,
-3.1997e+00, -1.0452e+00],
[-2.7383e+00, -1.8924e+00, -1.8845e+00, ..., -2.4003e+00,
-2.6905e+00, -1.0910e+00],
...,
[-1.1102e+00, -1.0868e+00, -2.2783e+00, ..., -1.6315e+00,
-1.5175e+00, -2.0305e-01],
[-9.2804e-01, -1.5790e+00, -3.4000e+00, ..., -5.5073e-01,
-1.2240e+00, -1.2866e-01],
[-3.9783e-01, -2.8889e+00, -3.7085e+00, ..., 3.6647e-01,
2.3261e-02, 4.2223e-01]],
[[ 2.9374e+00, 4.2740e+00, 3.3976e+00, ..., 2.8527e+00,
1.0160e+00, 3.0563e+00],
[ 2.9283e-01, 1.0853e+00, 2.2474e+00, ..., 1.2479e+00,
-8.4109e-01, 2.7703e+00],
[ 1.0641e+00, 1.7667e+00, 2.1802e+00, ..., 6.7870e-01,
-1.2617e-01, 2.0595e+00],
...,
[ 5.8090e-01, -1.3088e-02, -2.8311e-02, ..., -1.1684e-01,
3.4937e-01, 1.5408e+00],
[-2.5986e-01, -4.6858e-01, -6.7345e-01, ..., -3.1871e-02,
2.2038e-01, 1.2201e+00],
[-2.2071e-01, -1.1438e+00, -7.5275e-01, ..., 8.2246e-01,
4.3302e-01, 1.1536e+00]],
...,
[[-1.9377e+00, -7.3637e+00, -5.4754e+00, ..., -3.9790e+00,
-3.2282e+00, -5.5415e+00],
[-2.4509e+00, -7.0437e+00, -4.2170e+00, ..., -3.5073e+00,
-2.6939e+00, -4.3200e+00],
[-1.1745e+00, -5.0132e+00, -4.5412e+00, ..., -3.0282e+00,
-2.9291e+00, -2.3224e+00],
...,
[-2.9473e+00, -5.8657e+00, -5.8524e+00, ..., -3.2891e+00,
-3.5537e+00, -3.4985e+00],
[-3.4555e+00, -5.2636e+00, -2.4487e+00, ..., -4.4815e+00,
-4.3792e+00, -4.8982e+00],
[-3.7598e+00, -5.5171e+00, 5.0927e-01, ..., -3.3571e+00,
-3.2237e+00, -4.1595e+00]],
[[-2.0201e-01, 3.3804e-01, 5.3569e-01, ..., 5.7814e-01,
8.0335e-01, 1.4006e+00],
[ 1.6151e+00, 1.0928e+00, 1.8362e+00, ..., 1.8287e+00,
2.4902e+00, 3.0431e+00],
[ 1.1037e+00, 2.7566e-01, 7.6742e-01, ..., 1.1963e+00,
1.6079e+00, 2.2948e+00],
...,
[ 1.6950e+00, -2.5544e-01, -2.4402e+00, ..., 1.0632e+00,
7.9497e-01, 1.3310e+00],
[ 2.5758e+00, 2.6702e+00, 1.3613e+00, ..., 1.1071e+00,
5.4581e-01, 2.0692e+00],
[ 1.0690e-01, 1.4865e-01, 2.0419e-01, ..., 1.2295e+00,
1.2076e+00, 1.4120e+00]],
[[-5.6577e-01, 2.8592e-01, 1.0768e-01, ..., 5.8046e-01,
5.1601e-01, 1.3472e+00],
[-1.3486e+00, -1.8569e+00, -1.6878e+00, ..., -6.2945e-01,
-3.7045e-01, 2.5165e-01],
[-2.2785e+00, -6.6835e-01, 2.4178e-01, ..., 3.7723e-01,
2.9921e-01, 7.7701e-01],
...,
[-2.1526e+00, -2.8811e-01, 1.9511e-01, ..., -6.6903e-01,
-1.1630e+00, -7.2132e-01],
[-1.5068e+00, -1.4985e+00, -7.4314e-01, ..., -5.5420e-01,
-1.2666e+00, -2.0295e-02],
[-2.5172e-01, -4.7225e+00, -2.9010e+00, ..., 2.1045e-01,
-1.0356e-01, 7.9837e-01]]],
[[[ 7.2470e-01, 1.0788e+00, 7.3474e-01, ..., 4.0157e-01,
3.2326e-01, 2.0006e-01],
[-3.1104e-01, -1.4750e+00, -5.6839e-01, ..., 3.8563e-01,
3.6569e-01, -1.1708e+00],
[ 1.3734e-01, -8.7307e-01, -2.7800e-01, ..., -6.0638e-01,
1.0200e+00, 6.1201e-01],
...,
[ 3.0447e-02, -1.1669e+00, -9.7222e-01, ..., 1.4326e-01,
9.4335e-01, 5.1720e-01],
[-4.4734e-01, -1.7937e+00, -8.4140e-01, ..., 4.7955e-02,
1.4176e+00, -1.3627e-01],
[ 3.3731e-01, -5.1133e-01, -7.8868e-01, ..., -3.7433e-01,
3.8574e-01, 5.8463e-01]],
[[-3.0495e+00, -1.4939e+00, -3.1193e+00, ..., -1.7751e+00,
-2.2264e+00, -1.0750e+00],
[-4.4365e+00, -1.8665e+00, -2.4163e+00, ..., -4.8567e-01,
-1.2714e+00, -2.9005e-01],
[-4.4070e+00, -2.6640e+00, -3.2478e+00, ..., -1.7237e+00,
-1.4045e+00, -7.2300e-01],
...,
[-2.8153e+00, -1.8611e+00, -2.3166e+00, ..., 1.4281e-01,
5.9524e-02, 4.9543e-01],
[-3.5511e+00, -1.3663e+00, -2.2385e+00, ..., -6.9341e-02,
-3.6612e-01, 2.5417e-01],
[-1.2698e+00, -9.0674e-01, -1.8460e+00, ..., -4.7810e-01,
-7.5781e-01, -5.5083e-01]],
[[ 1.7177e+00, 2.8131e+00, 2.3984e+00, ..., 2.8426e+00,
1.4283e+00, 2.8983e+00],
[ 1.5711e-01, 1.2110e+00, 1.5926e+00, ..., 1.9267e+00,
4.4331e-03, 2.4941e+00],
[ 7.1986e-01, 1.4614e+00, 9.5575e-01, ..., 9.9853e-01,
5.5459e-01, 1.7994e+00],
...,
[ 9.5872e-01, 1.2281e+00, 1.0075e+00, ..., 2.9239e-01,
1.4997e-01, 1.4397e+00],
[ 6.2012e-01, 1.7594e+00, 9.5165e-01, ..., 8.6730e-01,
6.2650e-01, 1.5347e+00],
[ 1.5319e+00, 1.5656e+00, 1.5490e+00, ..., 1.2974e+00,
1.0883e+00, 1.9884e+00]],
...,
[[-1.4575e+00, -4.7045e+00, -4.0065e+00, ..., -3.4469e+00,
-2.4531e+00, -3.9095e+00],
[-1.8328e+00, -3.9154e+00, -4.3122e+00, ..., -5.5303e+00,
-3.9227e+00, -4.3444e+00],
[-9.6601e-01, -3.2550e+00, -3.8057e+00, ..., -2.9197e+00,
-2.3753e+00, -1.4799e+00],
...,
[-1.5522e+00, -4.0135e+00, -4.5531e+00, ..., -4.8272e+00,
-2.3966e+00, -1.5473e+00],
[-1.4394e+00, -4.4239e+00, -5.7151e+00, ..., -6.0657e+00,
-3.4578e+00, -3.7436e+00],
[-1.4865e+00, -4.6804e+00, -4.2102e+00, ..., -4.0446e+00,
-2.2452e+00, -2.7748e+00]],
[[ 8.0879e-01, 5.0983e-01, -1.2759e-01, ..., 8.7305e-01,
1.1125e+00, 1.2377e+00],
[ 1.1908e+00, 9.9475e-01, 5.3415e-01, ..., 1.5518e+00,
2.4089e+00, 2.5750e+00],
[ 5.8013e-01, 5.7046e-01, -5.6599e-02, ..., 9.4602e-01,
1.6936e+00, 1.8985e+00],
...,
[ 8.1886e-01, 4.0998e-03, -2.6514e-01, ..., 1.5347e+00,
1.9121e+00, 1.6470e+00],
[ 1.4976e+00, 1.3960e+00, 1.0311e+00, ..., 1.3123e+00,
1.4999e+00, 2.4384e+00],
[ 6.1752e-01, 8.6458e-01, 9.5910e-01, ..., 1.0095e+00,
1.2940e+00, 1.3539e+00]],
[[-2.6680e-01, -2.0139e-01, -1.0223e+00, ..., -2.3608e-01,
-1.7551e-01, 8.5653e-01],
[-1.3912e+00, -1.5002e+00, -2.0618e+00, ..., 8.1315e-02,
-6.0908e-02, 3.5550e-01],
[-1.6636e+00, -9.1482e-01, -1.2523e+00, ..., 1.4262e-01,
2.2095e-01, 2.4732e-01],
...,
[-1.7136e+00, -1.5866e+00, -1.4739e+00, ..., -7.3215e-01,
-3.3918e-01, -3.1762e-01],
[-1.4281e+00, -1.5665e+00, -1.5220e+00, ..., -2.6281e-01,
-4.4320e-01, 3.1732e-01],
[-1.8192e-01, 5.5386e-02, -5.2528e-01, ..., 3.1149e-01,
1.4418e-01, 3.3471e-01]]],
[[[ 1.0024e+00, 2.0961e+00, -3.2989e+00, ..., -1.9474e+00,
-3.3969e-01, 1.1672e-01],
[-7.0685e-01, -1.8963e+00, -6.7742e+00, ..., -4.0645e+00,
-6.1668e+00, -5.2195e+00],
[-2.1359e-02, -2.0527e+00, 5.8203e-02, ..., -2.9393e+00,
-4.7788e+00, -9.4703e+00],
...,
[-8.6526e-01, -1.2421e+00, -7.4259e+00, ..., -9.3446e-01,
1.4004e+00, 3.4136e-01],
[-1.8238e+00, 1.5820e+00, -1.0747e+01, ..., -4.2603e+00,
1.0871e+00, 2.5508e-01],
[-1.2476e+00, 4.0491e+00, -6.7383e+00, ..., 1.7658e+00,
4.8423e-01, 6.5252e-01]],
[[-3.5448e+00, -2.1905e+00, -6.2569e+00, ..., -2.6148e+00,
-2.0122e+00, -4.5326e-01],
[-4.5137e+00, -2.9922e+00, -9.8317e+00, ..., -3.5835e+00,
-6.6216e+00, -7.8534e+00],
[-2.8937e+00, -7.1639e+00, -1.9620e+01, ..., -1.7452e+01,
-1.8053e+01, -7.9692e+00],
...,
[-8.5533e+00, -1.1462e+01, 5.4631e+00, ..., -7.4373e+00,
-6.1449e+00, 3.5033e-01],
[-8.3324e+00, -1.0876e+01, 1.0093e+00, ..., -7.7873e+00,
-4.5373e+00, 8.0846e-01],
[-3.3652e+00, -2.8110e+00, -2.7961e-01, ..., -7.5412e+00,
-2.7382e+00, 3.6636e-02]],
[[ 2.9664e+00, 5.3203e+00, -2.9694e+00, ..., -3.9642e+00,
-3.9427e+00, -3.1261e+00],
[-6.6198e-03, 3.4223e+00, -9.5258e+00, ..., -1.2354e+01,
-1.3532e+01, -7.7929e+00],
[ 4.7957e-01, 5.1640e+00, -1.0771e+01, ..., -1.3419e+01,
-7.2805e+00, -5.5709e+00],
...,
[ 3.6598e+00, -7.6864e+00, -1.1886e+01, ..., -4.6718e+00,
-4.9640e-01, -5.1238e-01],
[ 1.6993e+00, -1.3144e+01, -1.2424e+01, ..., -5.2562e+00,
1.7856e+00, -6.1777e-01],
[ 1.0903e+00, -1.3423e+01, -8.8114e+00, ..., -2.1454e+00,
6.9464e-01, 2.6947e-01]],
...,
[[-1.7907e+00, -7.2483e+00, -8.3329e+00, ..., -7.1104e+00,
2.8345e+00, -4.6103e+00],
[-2.4204e+00, -7.8528e+00, -1.4628e+01, ..., -1.0702e+01,
1.6303e+00, -4.3902e+00],
[-8.4679e-01, -5.7689e+00, -1.5388e+01, ..., -3.1908e+01,
9.7001e-01, 3.2356e+00],
...,
[-3.8650e+00, -1.4086e+01, -1.3066e+01, ..., -2.7655e+00,
-1.5404e+01, -9.0464e+00],
[-3.8398e+00, -1.3268e+01, -1.8039e+01, ..., -4.3440e+00,
-1.1592e+01, -1.0112e+01],
[-3.5795e+00, -8.8517e+00, -1.3908e+01, ..., -1.3105e+01,
-8.5190e+00, -6.7002e+00]],
[[ 8.3033e-02, -4.4756e+00, -4.6530e+00, ..., -1.4176e+00,
-3.3447e+00, -1.9770e+00],
[ 1.5129e+00, -4.5079e+00, -3.9684e+00, ..., -6.7163e+00,
-3.5340e+00, -2.9681e+00],
[ 9.0885e-01, -5.4100e+00, 9.0219e-01, ..., -1.6190e+01,
-1.1225e+01, -5.6404e+00],
...,
[-3.9179e+00, -5.9362e+00, 1.7130e+00, ..., -4.5065e+00,
-2.8503e-01, 3.6752e+00],
[-2.7136e+00, -3.7203e+00, -4.9802e-01, ..., -5.3048e+00,
8.8349e-01, 3.3318e+00],
[-9.3513e-02, 5.7321e-01, -2.0957e+00, ..., -3.2000e+00,
1.6462e+00, 1.9057e+00]],
[[-4.2983e-01, -1.3966e+00, -9.3179e-01, ..., -5.8691e+00,
-1.5275e+00, 2.9187e+00],
[-1.2519e+00, -3.3192e+00, -2.0575e+00, ..., -7.6169e+00,
-2.7045e+00, 1.5620e+00],
[-1.8725e+00, -3.7036e+00, -4.4644e+00, ..., -5.7855e+00,
-1.1321e+01, -7.4734e-01],
...,
[-4.0254e+00, -1.1685e+00, 6.7340e+00, ..., -1.2608e+00,
-2.0577e+00, -1.3339e+00],
[-3.6847e+00, 2.3038e+00, 5.7521e+00, ..., -2.6302e+00,
-2.9756e+00, -9.6683e-01],
[-7.4464e-01, 3.3339e+00, 9.4649e-01, ..., 5.3169e-01,
-1.5663e+00, -7.7703e-01]]]], device='cuda:0'),
tensor([[[[ 1.0780e+00, 1.4663e+00, 1.1862e+00, ..., 3.1345e-01,
3.4412e-01, 2.3642e-01],
[-8.5146e-01, -1.4070e+00, -9.1063e-01, ..., 4.2233e-01,
3.3019e-01, -1.1706e+00],
[-3.3676e-01, -7.6650e-01, -5.3492e-01, ..., -3.6657e-01,
1.0110e+00, 6.4022e-01],
...,
[ 4.5299e-01, -7.4548e-02, 6.0901e-01, ..., 3.9021e-01,
1.0397e+00, 5.4390e-01],
[-3.8360e-01, -8.7748e-01, -4.5587e-01, ..., 5.7025e-01,
1.5716e+00, -1.6384e-01],
[-1.0623e-01, -1.6431e-01, -7.4715e-01, ..., -6.5061e-02,
3.7775e-01, 6.2420e-01]],
[[-3.2099e+00, -1.9790e+00, -2.2773e+00, ..., -1.8160e+00,
-2.2383e+00, -1.1136e+00],
[-2.8016e+00, -3.2514e+00, -1.5577e+00, ..., -4.6356e-01,
-1.3499e+00, -3.3012e-01],
[-1.7706e+00, -2.3432e+00, -1.7172e+00, ..., -1.2327e+00,
-1.4635e+00, -7.4544e-01],
...,
[ 1.4056e-01, -8.2730e-01, -7.9156e-01, ..., -7.7361e-01,
3.5124e-01, 5.4228e-01],
[-1.2751e+00, -8.3554e-01, -8.9754e-01, ..., -6.6982e-01,
-2.5294e-01, 2.5467e-01],
[-4.1760e-01, -9.3965e-01, -8.2922e-01, ..., -5.4490e-01,
-7.2704e-01, -5.5670e-01]],
[[ 2.3427e+00, 2.3643e+00, 1.8312e+00, ..., 2.8471e+00,
1.3913e+00, 2.9320e+00],
[ 1.4059e-01, 1.0697e-01, 1.3895e+00, ..., 1.8483e+00,
-3.9114e-02, 2.5057e+00],
[-2.6358e-01, 2.7558e-01, 1.1001e+00, ..., 1.2168e+00,
6.5224e-01, 1.8598e+00],
...,
[-1.7732e-01, 1.0187e+00, 1.1076e+00, ..., 6.4606e-01,
-4.0364e-03, 1.4123e+00],
[-9.7116e-01, 6.9011e-01, 5.9680e-01, ..., 7.7982e-01,
6.1391e-01, 1.5230e+00],
[ 6.9343e-01, 5.4360e-01, 1.1198e+00, ..., 1.3633e+00,
1.1337e+00, 2.0341e+00]],
...,
[[-2.9088e+00, -8.4388e+00, -6.6632e+00, ..., -3.3384e+00,
-2.4495e+00, -3.9617e+00],
[-3.7669e+00, -1.0271e+01, -6.1798e+00, ..., -5.6345e+00,
-3.9083e+00, -4.3860e+00],
[-2.2022e+00, -7.3377e+00, -4.2329e+00, ..., -2.9881e+00,
-2.3491e+00, -1.6052e+00],
...,
[-3.1438e+00, -8.2898e+00, -5.6740e+00, ..., -2.9987e+00,
-2.5431e+00, -1.4950e+00],
[-4.0321e+00, -9.1394e+00, -5.0628e+00, ..., -4.5328e+00,
-3.5766e+00, -3.6989e+00],
[-3.2010e+00, -7.5962e+00, -2.9747e+00, ..., -3.9598e+00,
-2.3469e+00, -2.7777e+00]],
[[-4.4463e-01, 7.2785e-01, 8.6835e-01, ..., 1.2405e+00,
1.1049e+00, 1.2526e+00],
[ 7.2730e-01, 6.9954e-01, 1.4278e+00, ..., 1.8824e+00,
2.3374e+00, 2.5536e+00],
[ 3.7710e-01, 3.3974e-01, 1.6674e+00, ..., 1.5151e+00,
1.6421e+00, 1.9382e+00],
...,
[ 1.4387e+00, 4.9198e-01, 1.4648e+00, ..., 1.1484e+00,
2.2096e+00, 1.6329e+00],
[ 1.7048e+00, 1.5245e+00, 1.4463e+00, ..., 1.4335e+00,
1.5877e+00, 2.4324e+00],
[ 7.5195e-01, 6.6208e-01, 1.2566e+00, ..., 1.1987e+00,
1.2973e+00, 1.3695e+00]],
[[-2.0606e-01, -4.4882e-01, -3.9123e-01, ..., -1.1600e-01,
-1.0700e-01, 8.8536e-01],
[-6.4229e-01, -2.6616e+00, -1.1833e+00, ..., 1.9864e-01,
-8.2548e-02, 3.5601e-01],
[-1.9048e+00, -1.7276e+00, -6.9013e-01, ..., 3.5420e-01,
1.6866e-01, 2.3367e-01],
...,
[-1.8777e+00, -2.4827e+00, -3.5853e-01, ..., -6.0543e-01,
-2.2351e-01, -3.0451e-01],
[-1.3894e+00, -2.4732e+00, -8.3098e-01, ..., -1.2437e+00,
-3.6451e-01, 2.8611e-01],
[-3.8943e-01, -1.0706e+00, -5.7900e-01, ..., 3.2390e-01,
1.8768e-01, 3.4296e-01]]],
[[[ 8.3452e-01, 1.4835e+00, 1.2961e+00, ..., -1.5605e+00,
-1.8183e-01, 4.4099e-01],
[-8.4752e-01, -2.1852e+00, -1.2314e+00, ..., 2.2271e+00,
-5.5723e-01, -2.2286e+00],
[-7.8947e-02, -4.0105e-01, 4.3831e-01, ..., 1.3905e+00,
2.0769e+00, 9.4376e-01],
...,
[ 3.7312e-01, -2.0399e+00, -3.3205e+00, ..., -1.6590e+00,
-5.2766e-02, 6.7599e-01],
[-4.3020e-01, -3.3459e+00, -1.1499e+00, ..., 2.0784e+00,
3.0190e-01, -6.4041e-02],
[ 2.8048e-01, -9.8113e-01, -7.5003e-01, ..., 2.0698e+00,
-2.1195e-01, 6.1547e-01]],
[[-3.7565e+00, -4.5744e-01, -1.5005e+00, ..., -2.8407e+00,
-2.7921e+00, -9.2984e-01],
[-4.1915e+00, -2.7631e+00, -2.0530e+00, ..., -1.9344e+00,
-4.2623e+00, -8.1586e-01],
[-2.4990e+00, -1.8555e+00, -3.5859e+00, ..., -4.3632e+00,
-3.9331e+00, -7.2004e-01],
...,
[-1.7479e+00, -7.4609e+00, -6.7334e+00, ..., -2.3019e+00,
-1.6153e+00, 3.6154e-02],
[-2.6122e+00, -2.7056e+00, 8.9250e-01, ..., -2.7753e+00,
-9.9964e-01, 1.4700e-01],
[-1.4099e-01, -1.1109e+00, 1.0385e+00, ..., -2.9908e+00,
-1.7683e-01, 4.7930e-01]],
[[ 2.9979e+00, 4.2751e+00, 3.3665e+00, ..., 4.5509e+00,
2.1558e+00, 2.9304e+00],
[ 2.7366e-01, 1.0432e+00, 2.2315e+00, ..., -6.1855e-01,
-5.8941e-01, 2.6018e+00],
[ 9.4415e-01, 1.4345e+00, 1.2554e+00, ..., -3.3426e+00,
6.0091e-02, 1.9897e+00],
...,
[ 6.3413e-01, 3.1841e+00, 1.4744e+00, ..., 1.0590e+00,
3.7842e-01, 1.3852e+00],
[ 7.4707e-01, 3.3827e+00, -5.8968e+00, ..., -2.6354e+00,
1.1115e-01, 1.0922e+00],
[ 2.5392e+00, 5.0794e-01, -5.5613e+00, ..., -9.0912e-01,
-8.2531e-01, 1.0878e+00]],
...,
[[-1.9478e+00, -7.0942e+00, -5.2767e+00, ..., -2.4236e+00,
-1.9024e+00, -5.4318e+00],
[-2.5986e+00, -6.7109e+00, -3.7551e+00, ..., -9.9150e+00,
-4.5511e+00, -5.2592e+00],
[-1.0390e+00, -4.7120e+00, -4.3593e+00, ..., -6.2376e+00,
-7.6764e+00, -5.3056e+00],
...,
[-2.8060e+00, -6.0589e+00, -7.0969e+00, ..., -1.1182e+01,
-4.3761e+00, -3.5275e+00],
[-2.3405e+00, -9.4118e+00, -9.1711e+00, ..., -1.8787e+01,
-6.9407e+00, -4.6855e+00],
[-2.7316e+00, -8.5255e+00, -2.8853e+00, ..., -1.6417e+01,
-5.2788e+00, -4.0194e+00]],
[[-5.3599e-02, 4.8143e-01, 5.6631e-01, ..., 1.8722e+00,
1.8099e+00, 2.0620e+00],
[ 1.6275e+00, 1.1839e+00, 1.8557e+00, ..., 4.4309e-02,
2.5872e+00, 3.8325e+00],
[ 1.1530e+00, -1.3706e-01, 7.5722e-01, ..., -3.1614e+00,
4.9959e-01, 2.9852e+00],
...,
[ 5.4522e-01, -5.7838e+00, -3.7519e+00, ..., 3.7222e+00,
1.6450e+00, 1.3264e+00],
[ 1.0855e+00, -8.3784e-01, 3.3879e-01, ..., 3.3251e+00,
2.4254e+00, 2.1289e+00],
[ 8.6686e-01, 5.3038e-01, 1.6203e+00, ..., 2.3755e-01,
2.1249e+00, 1.3694e+00]],
[[-5.8441e-01, 3.3259e-01, 1.8266e-01, ..., -8.1987e-01,
5.5970e-02, 1.4848e+00],
[-1.3787e+00, -1.7897e+00, -1.7123e+00, ..., 1.6096e+00,
-6.6875e-01, 1.8294e-02],
[-2.0885e+00, -1.0676e+00, -1.8424e-01, ..., 5.4063e-01,
-3.2498e-01, -1.0568e-02],
...,
[-2.1521e+00, -4.2834e+00, -3.1747e+00, ..., -2.9459e+00,
-5.6920e-01, -6.5167e-01],
[-1.9008e+00, -3.7497e+00, 6.5907e-01, ..., -3.5666e+00,
-1.2392e+00, 2.0462e-02],
[ 4.5150e-01, -2.8087e-01, 3.6178e-01, ..., -3.3011e+00,
-1.7838e+00, 7.6674e-01]]],
[[[ 8.3997e-01, 1.2076e+00, 5.3804e-01, ..., 1.9641e-01,
5.8604e-02, -5.0831e-01],
[-6.0825e-01, -6.5319e-01, 1.0047e+00, ..., 4.7439e-01,
1.2402e+00, -2.5372e-01],
[-6.5378e-02, -3.6791e-01, 1.8713e+00, ..., -9.0012e-02,
6.8550e-01, 9.3026e-01],
...,
[ 2.5705e-01, -7.9156e-01, 1.0156e-02, ..., -1.0447e-01,
7.5092e-01, 5.9240e-01],
[-6.3199e-01, -2.1470e+00, -9.7468e-01, ..., -2.2353e-02,
-1.4022e-01, -1.2448e-01],
[-3.1666e-01, -1.8464e+00, -1.5995e+00, ..., -1.0819e+00,
-1.5508e+00, -2.8021e-01]],
[[-3.4353e+00, -4.7311e-01, -2.6960e+00, ..., -2.4525e+00,
-3.3474e+00, -8.2123e-01],
[-3.1560e+00, -1.5662e+00, -2.7206e+00, ..., -1.3733e+00,
-2.0552e+00, -1.1167e-01],
[-2.0500e+00, -6.2713e-01, -9.0503e-01, ..., -1.9785e+00,
-1.6902e+00, -4.4804e-01],
...,
[-1.5912e+00, -1.4477e+00, -1.5085e+00, ..., -3.1836e+00,
-1.2773e+00, -4.8852e-01],
[-3.1190e+00, -1.8610e+00, -2.6728e+00, ..., -1.7161e+00,
-2.0462e+00, -1.2250e+00],
[-2.2890e+00, -1.9955e+00, -2.8496e+00, ..., -1.9631e+00,
-1.9290e+00, -9.8846e-01]],
[[ 2.2249e+00, 2.7568e+00, 3.5958e-03, ..., 2.3035e+00,
5.5806e-01, 2.2359e+00],
[ 4.4596e-01, 1.0530e+00, -3.5936e+00, ..., 1.7950e+00,
-5.6053e-02, 2.1605e+00],
[ 1.0535e+00, 1.4260e+00, -2.5879e+00, ..., 1.1805e+00,
8.5925e-01, 2.3240e+00],
...,
[ 1.1941e+00, 1.7257e+00, 1.4060e+00, ..., 2.1968e-01,
1.1435e+00, 2.3017e+00],
[ 6.9009e-01, 1.9309e+00, 1.4946e+00, ..., 4.5144e-01,
8.9740e-01, 1.8127e+00],
[ 1.6101e+00, 1.7189e+00, 2.0540e+00, ..., 9.8726e-01,
1.8073e+00, 1.7610e+00]],
...,
[[-2.0429e+00, -6.3887e+00, -5.0242e+00, ..., -3.2118e+00,
-2.7819e+00, -3.7391e+00],
[-2.5951e+00, -7.6876e+00, -5.8001e+00, ..., -4.7944e+00,
-4.4935e+00, -4.9737e+00],
[-1.4592e+00, -4.1644e+00, -2.3686e+00, ..., -2.3766e+00,
-2.4667e+00, -2.0536e+00],
...,
[-2.3418e+00, -5.1710e+00, -4.2108e+00, ..., -5.2804e+00,
-5.1213e+00, -4.6668e+00],
[-2.7513e+00, -5.7508e+00, -5.3161e+00, ..., -7.0270e+00,
-6.7063e+00, -6.2179e+00],
[-1.4856e+00, -4.1708e+00, -2.6691e+00, ..., -4.2732e+00,
-2.7535e+00, -3.2726e+00]],
[[-3.7712e-01, 4.7175e-01, 9.4566e-02, ..., 5.9794e-01,
9.0301e-01, 1.6274e+00],
[ 1.1203e+00, 2.5302e-01, 6.7626e-01, ..., 1.2266e+00,
1.7240e+00, 2.5177e+00],
[ 1.0237e+00, 3.6632e-01, 1.2189e+00, ..., 1.3009e+00,
1.3873e+00, 1.8511e+00],
...,
[ 1.3335e+00, 2.6168e-01, 2.8382e-01, ..., 1.7757e+00,
1.6985e+00, 1.5764e+00],
[ 1.8630e+00, 1.5713e+00, 9.8506e-01, ..., 2.3345e+00,
1.3448e+00, 1.9559e+00],
[ 4.5484e-01, 6.5347e-01, 5.0827e-01, ..., 7.2820e-01,
7.2873e-01, 1.6596e+00]],
[[-6.2606e-01, -3.2924e-01, -2.4660e-01, ..., -4.2724e-01,
-5.0507e-01, 5.8572e-01],
[-1.2894e+00, -9.9680e-01, -1.1379e+00, ..., -2.5881e-01,
3.8939e-01, 8.2294e-01],
[-1.8536e+00, -1.4603e-01, -2.4089e-01, ..., 2.6991e-01,
4.4751e-01, 3.2530e-03],
...,
[-1.9618e+00, -1.8617e+00, -9.5694e-01, ..., -2.8452e+00,
-1.3575e+00, -5.2203e-01],
[-1.5972e+00, -2.2171e+00, -1.6965e+00, ..., -5.4511e-01,
-9.2816e-01, -7.0501e-01],
[-8.0830e-01, -9.8248e-01, -1.2480e+00, ..., -3.6768e-01,
-8.6892e-01, -9.5037e-02]]],
[[[ 6.9575e-01, 1.0459e+00, 9.8739e-01, ..., 9.4580e-02,
3.5381e-01, 2.0894e-01],
[-3.5197e-01, -1.4802e+00, -3.3943e-01, ..., -6.9068e-02,
4.2123e-01, -1.2092e+00],
[ 1.1487e-01, -8.4216e-01, -1.2286e+00, ..., -1.0414e+00,
1.0880e+00, 6.7718e-01],
...,
[ 3.1767e-02, -1.8184e+00, 1.3793e+00, ..., -3.0547e-01,
8.0539e-01, 4.8046e-01],
[-5.1228e-01, -2.7121e+00, -1.6606e+00, ..., 7.1796e-02,
1.3950e+00, -1.5442e-01],
[ 4.1787e-01, -9.7329e-01, -7.0891e-01, ..., -3.8739e-01,
4.2069e-01, 6.7917e-01]],
[[-3.1579e+00, -1.5394e+00, -3.7907e+00, ..., -1.8290e+00,
-1.9938e+00, -8.4007e-01],
[-4.4408e+00, -1.8845e+00, -3.5690e+00, ..., -7.0070e-01,
-1.1817e+00, -1.7384e-01],
[-4.4344e+00, -2.6525e+00, -5.3955e+00, ..., -1.5598e+00,
-1.3744e+00, -7.2795e-01],
...,
[-2.6828e+00, -2.6888e+00, -4.3731e+00, ..., -6.1133e-01,
2.4260e-01, 4.5297e-01],
[-3.5962e+00, -1.8018e+00, -3.5889e+00, ..., 3.9818e-01,
-4.0205e-01, 1.8344e-01],
[-1.1802e+00, -1.3369e+00, -2.6478e+00, ..., -4.1263e-01,
-7.2630e-01, -5.7179e-01]],
[[ 1.7790e+00, 2.8465e+00, 2.3976e+00, ..., 2.4280e+00,
1.3790e+00, 2.7616e+00],
[ 2.0802e-01, 1.2763e+00, 2.4742e+00, ..., 2.1346e+00,
-1.7949e-02, 2.3780e+00],
[ 7.7277e-01, 1.6033e+00, 1.8627e+00, ..., 1.3064e+00,
5.1287e-01, 1.8436e+00],
...,
[ 1.0185e+00, 3.6013e+00, 1.7744e+00, ..., 6.2645e-01,
-3.2322e-01, 1.4026e+00],
[ 6.6945e-01, 2.8656e+00, 3.1034e+00, ..., 7.1114e-01,
6.2415e-01, 1.5364e+00],
[ 1.6067e+00, 1.7109e+00, 3.0004e+00, ..., 9.9993e-01,
1.1173e+00, 2.0472e+00]],
...,
[[-1.4661e+00, -4.6914e+00, -4.0292e+00, ..., -3.4102e+00,
-2.3596e+00, -3.7524e+00],
[-1.9138e+00, -4.1495e+00, -4.6813e+00, ..., -5.7040e+00,
-3.7559e+00, -4.1183e+00],
[-1.0908e+00, -3.4993e+00, -4.0295e+00, ..., -5.5250e+00,
-2.3207e+00, -1.5714e+00],
...,
[-1.5961e+00, -4.5863e+00, -5.4097e+00, ..., -7.4124e+00,
-3.7544e+00, -1.3605e+00],
[-1.5228e+00, -5.2286e+00, -5.7729e+00, ..., -7.9982e+00,
-3.5014e+00, -3.5797e+00],
[-1.6891e+00, -5.3977e+00, -4.2042e+00, ..., -5.2876e+00,
-2.1328e+00, -2.7997e+00]],
[[ 7.2790e-01, 5.0858e-01, -8.6582e-01, ..., 8.3237e-01,
1.1947e+00, 1.2998e+00],
[ 1.1981e+00, 9.6119e-01, -9.0893e-01, ..., 2.3156e+00,
2.3383e+00, 2.5147e+00],
[ 5.6274e-01, 4.8297e-01, -1.2461e+00, ..., 2.3246e+00,
1.5745e+00, 1.9202e+00],
...,
[ 7.9498e-01, -1.3724e+00, -5.1976e-01, ..., 2.7570e+00,
2.1005e+00, 1.6341e+00],
[ 1.5225e+00, 8.2130e-01, -6.4615e-01, ..., 2.0476e+00,
1.4254e+00, 2.4445e+00],
[ 6.3739e-01, 3.9628e-01, -7.4156e-01, ..., 1.1416e+00,
1.3402e+00, 1.3701e+00]],
[[-2.9232e-01, -1.9106e-01, -1.6206e+00, ..., -3.2375e-01,
-1.3931e-01, 8.8818e-01],
[-1.4237e+00, -1.3995e+00, -3.1378e+00, ..., 3.1299e-01,
-1.1851e-01, 3.4506e-01],
[-1.7253e+00, -8.5951e-01, -3.1946e+00, ..., 3.2847e-01,
1.4254e-01, 2.0483e-01],
...,
[-1.7666e+00, -1.4488e+00, -1.4046e+00, ..., -1.6202e+00,
-7.6235e-02, -3.5933e-01],
[-1.4662e+00, -1.7262e+00, -7.3215e-01, ..., -7.1069e-01,
-3.7266e-01, 2.6728e-01],
[-1.3203e-01, 1.4174e-01, -4.6836e-01, ..., 4.9766e-01,
1.5611e-01, 3.6479e-01]]],
[[[ 1.0625e+00, 1.4204e+00, 1.2298e+00, ..., 3.2192e-01,
3.7398e-01, 1.9615e-01],
[-7.9345e-01, -1.7137e+00, -1.0731e+00, ..., 3.0360e-01,
3.3299e-01, -1.1711e+00],
[-4.8686e-01, -7.8017e-01, -6.6389e-01, ..., -4.2033e-01,
1.0182e+00, 6.8367e-01],
...,
[-1.0405e-01, -4.5897e-01, 1.0208e-01, ..., -1.5999e-01,
1.0006e+00, 6.2807e-01],
[-3.6356e-01, -7.8548e-01, -1.8942e-01, ..., 1.3989e-01,
1.3435e+00, -8.6001e-02],
[-1.2742e-01, -1.4329e-01, -7.1066e-01, ..., -1.9096e-01,
3.7492e-01, 5.9527e-01]],
[[-3.1666e+00, -1.9694e+00, -2.1568e+00, ..., -1.9529e+00,
-2.4851e+00, -1.3243e+00],
[-3.0354e+00, -3.0736e+00, -1.1488e+00, ..., -3.1676e-01,
-1.2440e+00, -3.3625e-01],
[-2.2337e+00, -2.3609e+00, -1.4780e+00, ..., -1.4173e+00,
-1.4465e+00, -8.6730e-01],
...,
[ 1.8020e-01, -8.6388e-01, -4.6589e-01, ..., -3.7131e-01,
-1.4531e-01, 4.4932e-01],
[-1.2588e+00, -8.2999e-01, -3.7838e-01, ..., -1.9301e-01,
-5.0405e-01, 1.8781e-01],
[-4.2898e-01, -9.7504e-01, -7.8851e-01, ..., -4.0756e-01,
-7.1423e-01, -5.0541e-01]],
[[ 2.3612e+00, 2.3296e+00, 1.7672e+00, ..., 3.0814e+00,
1.4271e+00, 3.0323e+00],
[-2.1055e-02, -1.8102e-01, 1.3703e+00, ..., 2.0918e+00,
8.3109e-02, 2.7154e+00],
[-7.9686e-01, -5.0568e-01, 7.0435e-01, ..., 1.2832e+00,
6.7901e-01, 1.8830e+00],
...,
[-5.4135e-01, -1.6412e-01, 3.3310e-01, ..., 8.3700e-01,
1.8676e-01, 1.4883e+00],
[-9.8882e-01, 5.6013e-01, 6.5847e-01, ..., 1.0303e+00,
5.1683e-01, 1.4411e+00],
[ 6.5832e-01, 5.2486e-01, 1.1226e+00, ..., 1.4136e+00,
9.8484e-01, 1.9346e+00]],
...,
[[-2.9256e+00, -8.3668e+00, -6.6551e+00, ..., -3.8583e+00,
-2.7952e+00, -4.1987e+00],
[-3.7794e+00, -1.0155e+01, -6.2261e+00, ..., -6.0303e+00,
-4.1035e+00, -4.7732e+00],
[-1.9076e+00, -6.2993e+00, -3.9044e+00, ..., -3.3475e+00,
-2.5668e+00, -1.6609e+00],
...,
[-3.6697e+00, -8.5177e+00, -4.8227e+00, ..., -3.5402e+00,
-2.2847e+00, -1.3693e+00],
[-4.0975e+00, -9.2810e+00, -4.8853e+00, ..., -4.8134e+00,
-3.2386e+00, -3.4067e+00],
[-3.2880e+00, -7.7587e+00, -3.0038e+00, ..., -3.5999e+00,
-2.0470e+00, -2.5026e+00]],
[[-4.1364e-01, 7.2575e-01, 8.1921e-01, ..., 1.2439e+00,
9.8894e-01, 1.1955e+00],
[ 6.2324e-01, 6.9127e-01, 1.4425e+00, ..., 2.0675e+00,
2.5001e+00, 2.6955e+00],
[ 8.7988e-03, -5.3396e-01, 1.1117e+00, ..., 1.5808e+00,
1.6323e+00, 1.9102e+00],
...,
[ 1.2117e+00, 3.2440e-01, 1.0991e+00, ..., 1.6746e+00,
1.8161e+00, 1.7023e+00],
[ 1.6747e+00, 1.5244e+00, 1.6193e+00, ..., 1.6869e+00,
1.5241e+00, 2.3944e+00],
[ 7.3787e-01, 6.2395e-01, 1.2602e+00, ..., 1.1382e+00,
1.3669e+00, 1.3215e+00]],
[[-2.5178e-01, -4.0785e-01, -3.1356e-01, ..., -1.5277e-01,
-2.1553e-01, 8.6205e-01],
[-5.8218e-01, -2.3402e+00, -1.1274e+00, ..., 1.6250e-01,
-7.7403e-02, 4.6117e-01],
[-1.8165e+00, -1.0599e+00, -9.4436e-01, ..., 4.9551e-01,
3.0576e-01, 2.1502e-01],
...,
[-1.5391e+00, -2.4235e+00, -6.2517e-01, ..., -2.0226e-01,
-3.8441e-01, -2.9541e-01],
[-1.3307e+00, -2.5453e+00, -7.4750e-01, ..., -5.4886e-01,
-4.1091e-01, 3.3068e-01],
[-3.8470e-01, -1.1208e+00, -5.2929e-01, ..., 3.3869e-02,
1.6802e-01, 3.3398e-01]]]], device='cuda:0'),
tensor([[[[ 5.6044e-01, 7.5290e-01, 5.1325e-01, ..., -5.2616e-01,
2.4044e-01, -1.7553e-01],
[-9.4826e-03, -8.4227e-01, -2.1015e-01, ..., -1.9737e+00,
-3.4307e-01, -1.3253e-01],
[ 5.5056e-02, -1.2883e+00, -2.3706e+00, ..., -3.2137e+00,
-4.4975e+00, -2.4364e+00],
...,
[ 2.0461e-01, -2.6401e-01, 1.7444e-01, ..., -1.0051e+00,
-1.6729e-01, 9.5726e-01],
[-4.6984e-01, -1.4263e+00, 1.7913e-02, ..., 2.2610e+00,
-8.5851e-01, 1.6259e-01],
[ 2.4620e-01, -7.3178e-01, 1.5853e-01, ..., 2.2073e+00,
-1.2054e-01, 2.5794e-01]],
[[-2.9704e+00, -9.8014e-01, -2.8927e+00, ..., -2.0958e+00,
-2.6183e+00, -1.4517e+00],
[-3.8227e+00, -3.1871e+00, -5.8399e+00, ..., -1.8987e+00,
-1.8435e+00, -1.0225e+00],
[-6.6030e+00, -7.7541e+00, -9.4043e+00, ..., -1.0236e+01,
-5.5188e+00, -1.2336e+00],
...,
[-1.5242e+00, -4.1811e-01, 5.2269e-01, ..., -4.6734e+00,
-4.3126e+00, 2.5221e-02],
[-2.5143e+00, -4.5072e-01, 3.9988e-02, ..., -8.1413e+00,
-2.9317e+00, 1.8823e-01],
[-1.4156e+00, -1.4835e+00, -4.2561e-01, ..., -6.3685e+00,
-2.5881e+00, -9.6214e-01]],
[[ 2.1996e+00, 2.6675e+00, 2.0008e+00, ..., 2.4658e+00,
9.3343e-01, 2.1434e+00],
[ 1.2883e-01, 5.6416e-01, 1.9072e+00, ..., 1.9471e+00,
2.1769e+00, 2.0611e+00],
[ 6.4548e-01, 1.5420e+00, 5.6373e+00, ..., -2.1113e+00,
4.6402e+00, 5.8805e+00],
...,
[ 1.8701e+00, 1.5501e+00, -1.7143e+00, ..., -1.4673e+00,
-1.0379e+00, 3.9517e-01],
[ 1.5825e+00, 2.5398e+00, -2.2561e+00, ..., -2.9458e+00,
-6.4796e-01, 1.8848e-01],
[ 2.7015e+00, 2.2706e+00, -2.6121e+00, ..., 4.0838e-01,
-2.6134e-01, 8.0778e-01]],
...,
[[-2.1232e+00, -5.2681e+00, -4.0197e+00, ..., -3.4674e+00,
-1.9077e+00, -3.1889e+00],
[-2.6918e+00, -6.6958e+00, -4.8195e+00, ..., -1.1258e+01,
-4.5471e+00, -5.2218e+00],
[-2.2831e+00, -5.8607e+00, -4.0246e+00, ..., -9.7261e+00,
-3.8899e+00, -4.5562e+00],
...,
[-2.0582e+00, -6.1802e+00, -2.9203e+00, ..., -6.6572e+00,
-9.9617e+00, -7.2841e+00],
[-2.8803e+00, -7.6046e+00, -3.6419e+00, ..., -2.0805e+01,
-8.1655e+00, -7.9236e+00],
[-3.0649e+00, -6.7010e+00, -2.7606e+00, ..., -1.5591e+01,
-4.2310e+00, -4.0439e+00]],
[[ 6.0065e-01, 1.4507e+00, -6.8295e-01, ..., 2.2457e+00,
1.4349e+00, 1.5694e+00],
[ 5.0735e-01, 6.0358e-01, -1.6238e+00, ..., 3.5290e+00,
1.4882e+00, 2.1151e+00],
[-1.3980e+00, -5.3331e+00, -6.1131e+00, ..., -5.7304e+00,
-3.0171e+00, 3.1390e+00],
...,
[ 1.0757e+00, 3.8089e-01, 9.5164e-02, ..., -2.5381e+00,
1.4395e+00, 2.6608e+00],
[ 1.9622e+00, 1.4246e+00, 6.7912e-01, ..., -2.7681e+00,
3.6776e+00, 3.0280e+00],
[ 8.9637e-01, 9.5816e-01, 2.1174e+00, ..., -1.3015e+00,
2.6728e+00, 1.3545e+00]],
[[-1.8897e-01, -7.1138e-01, -2.2217e+00, ..., -1.2671e-01,
-9.3886e-01, -6.4507e-02],
[-1.1558e+00, -2.1935e+00, -4.3469e+00, ..., -1.4547e+00,
-1.3673e+00, -6.9117e-01],
[-2.8462e+00, -2.1997e+00, -2.2933e+00, ..., -2.2461e+00,
-3.9606e+00, -1.9348e+00],
...,
[-2.3702e+00, -1.8423e+00, 1.1375e+00, ..., -1.5072e+00,
-4.1392e+00, -2.6171e+00],
[-2.4756e+00, -2.3738e+00, 5.1732e-02, ..., -2.5565e+00,
-4.2264e+00, -9.0720e-01],
[-7.7980e-02, -9.4960e-02, -1.1994e+00, ..., -4.3245e+00,
-3.1535e+00, -2.6399e-01]]],
[[[ 3.9197e-01, 8.3578e-01, 3.9089e-01, ..., 9.9380e-01,
8.5858e-01, 1.4740e-01],
[-7.5294e-02, -3.0184e-01, -4.9278e-02, ..., 1.6669e-02,
8.3955e-02, -3.1532e-01],
[ 7.8896e-02, 4.8826e-01, 5.0074e-02, ..., -2.6350e-01,
9.7035e-02, 6.9557e-01],
...,
[-2.2149e-02, -9.7237e-01, -4.3509e-01, ..., 2.2795e-02,
7.4910e-01, 9.3088e-01],
[-3.8810e-01, -1.9106e+00, -8.3500e-01, ..., -1.9595e-01,
7.5138e-01, 2.0750e-01],
[ 4.3044e-01, -7.7987e-01, -1.1413e+00, ..., -7.7721e-01,
-6.8472e-01, 2.2089e-01]],
[[-1.6350e+00, 5.6079e-01, 4.5127e-01, ..., -1.3012e+00,
-2.3760e+00, -1.7138e+00],
[-1.3033e+00, 1.0590e+00, 1.2868e+00, ..., -2.1366e+00,
-3.1214e+00, -2.0811e+00],
[-2.0455e+00, -1.7781e+00, -2.6157e+00, ..., -1.4982e+00,
-2.2234e+00, -1.2452e+00],
...,
[-1.0734e+00, -6.7044e-01, -1.1081e+00, ..., -1.8252e+00,
-1.4090e+00, 1.6175e-02],
[-2.3051e+00, -8.4819e-02, -8.5794e-01, ..., -1.4161e+00,
-1.8421e+00, -3.9350e-01],
[-1.0910e+00, -1.4017e+00, -1.8004e+00, ..., -2.1239e+00,
-1.9217e+00, -7.3104e-01]],
[[ 1.9470e+00, 1.5611e+00, -1.9050e-01, ..., 1.3959e+00,
6.9814e-01, 2.1876e+00],
[ 5.7717e-01, -3.2001e-01, -5.2537e-01, ..., 5.3581e-01,
3.7066e-01, 2.5300e+00],
[ 8.1118e-01, 3.5760e-01, -3.5528e-02, ..., 1.8139e-01,
8.0586e-01, 2.0673e+00],
...,
[ 1.1314e+00, 1.1665e+00, 1.4982e+00, ..., 3.2503e-01,
-3.2567e-01, 1.4116e+00],
[ 7.5584e-01, 1.9660e+00, 1.4147e+00, ..., -1.0574e-01,
-5.8030e-01, 9.9470e-01],
[ 1.7571e+00, 1.0148e+00, 1.5357e+00, ..., 6.1904e-01,
3.7500e-01, 1.3333e+00]],
...,
[[-1.6671e+00, -3.9533e+00, -1.4726e+00, ..., -1.9248e+00,
-2.0160e+00, -3.3167e+00],
[-1.9316e+00, -4.5847e+00, -1.5551e+00, ..., -9.7665e-01,
-1.3596e+00, -2.8857e+00],
[-1.3365e+00, -4.8624e+00, -2.0944e+00, ..., -2.0950e+00,
-2.3125e+00, -2.0254e+00],
...,
[-2.0117e+00, -5.2832e+00, -3.2329e+00, ..., -2.7662e+00,
-2.5808e+00, -2.7707e+00],
[-2.9492e+00, -7.4979e+00, -4.3650e+00, ..., -3.7700e+00,
-2.9307e+00, -3.9955e+00],
[-3.0865e+00, -7.6888e+00, -3.5830e+00, ..., -2.5496e+00,
-1.1605e+00, -2.4692e+00]],
[[ 1.0213e+00, 1.8318e+00, 4.5827e-01, ..., 9.8266e-01,
1.2925e+00, 1.3967e+00],
[ 1.3348e+00, 1.1651e+00, 8.1532e-01, ..., 3.3947e-01,
6.3856e-01, 1.9959e+00],
[ 4.9848e-01, -2.7619e-01, -2.2395e-01, ..., 5.0881e-01,
1.2040e-01, 1.3008e+00],
...,
[ 1.6785e+00, 2.5960e-01, 7.1852e-01, ..., 1.0206e+00,
1.5619e+00, 1.8579e+00],
[ 2.2483e+00, 1.6439e+00, 1.3318e+00, ..., 1.0226e+00,
1.1111e+00, 2.1300e+00],
[ 9.8227e-01, 4.6619e-01, 5.5297e-01, ..., -1.8456e-02,
6.4196e-01, 1.1045e+00]],
[[-3.0115e-01, -5.7740e-01, -6.2217e-01, ..., 1.4593e-01,
-7.3060e-01, -7.6853e-02],
[-8.4228e-01, -1.9970e+00, -1.0902e+00, ..., -1.6702e+00,
-1.3711e+00, -8.5201e-01],
[-1.0588e+00, -1.0711e+00, -5.4661e-01, ..., 1.1687e-01,
-1.3047e-01, -2.4226e-01],
...,
[-2.6762e+00, -2.0665e+00, -6.9527e-01, ..., -1.5447e-01,
-1.3821e-01, 2.3686e-01],
[-2.4548e+00, -2.2729e+00, -9.1527e-01, ..., -2.6949e-01,
-4.4288e-01, 4.9599e-01],
[-4.1787e-01, -4.0127e-01, -5.1499e-01, ..., -2.3654e-01,
-6.1592e-01, 2.3585e-01]]],
[[[ 1.2277e+00, 1.6543e+00, 3.9842e-01, ..., 4.0758e-01,
2.7277e-01, -1.7815e-01],
[-1.1576e+00, -1.6829e+00, -1.5012e+00, ..., 8.3967e-02,
-2.4263e-01, -1.0942e+00],
[-2.0378e+00, -2.8223e+00, -1.1906e+00, ..., 8.3319e-01,
-2.1617e-01, -5.0736e-01],
...,
[ 4.7740e-01, -2.4793e-01, -2.6218e-01, ..., 9.7209e-02,
1.2776e+00, 1.1723e+00],
[-1.5367e-01, -9.7676e-01, -2.1215e-01, ..., -1.0026e+00,
1.9228e-01, 1.2504e-01],
[ 7.2398e-01, -2.1640e-01, -2.3114e-01, ..., -2.6038e-01,
-3.1416e-01, 4.1032e-01]],
[[-5.5611e+00, -2.6285e+00, -2.7961e+00, ..., -2.8862e+00,
-3.4492e+00, -8.4278e-01],
[-7.2581e+00, -5.0509e+00, -2.6278e+00, ..., -3.2512e+00,
-3.6058e+00, -7.7294e-01],
[-5.2958e+00, -8.7391e-01, 1.0702e+00, ..., -6.1550e+00,
-6.2665e+00, -2.1088e+00],
...,
[-9.9064e-01, -5.9760e-01, -1.3995e+00, ..., -1.9698e+00,
-1.5788e+00, -1.3770e-01],
[-3.0266e+00, -5.2915e-01, -1.0382e+00, ..., -4.0126e+00,
-2.4529e+00, -5.4468e-01],
[-2.0435e+00, -1.3425e+00, -1.6258e+00, ..., -1.4502e+00,
-9.7470e-01, -4.8702e-02]],
[[ 3.1542e+00, 4.3821e+00, 4.0361e+00, ..., 2.4612e+00,
7.7359e-01, 2.2601e+00],
[ 7.3118e-01, 5.0572e-01, 1.2659e+00, ..., 3.7578e+00,
1.3043e+00, 3.0381e+00],
[ 7.9724e-01, -8.6074e-01, -6.8901e-01, ..., -7.8749e-02,
8.9696e-02, 3.0787e+00],
...,
[ 2.9449e-01, 9.5363e-01, 1.4205e+00, ..., -2.2005e+00,
-1.4330e+00, 3.4905e-01],
[ 3.9863e-01, 2.1597e+00, 1.7058e+00, ..., -6.0190e-01,
1.0191e-01, 1.1031e+00],
[ 1.4253e+00, 1.3835e+00, 1.7507e+00, ..., 1.5802e+00,
1.2160e+00, 1.4107e+00]],
...,
[[-2.3848e+00, -7.8979e+00, -5.1357e+00, ..., -4.2421e+00,
-3.0108e+00, -3.7780e+00],
[-3.9330e+00, -8.1727e+00, -6.8405e+00, ..., -7.1433e+00,
-5.7730e+00, -4.9770e+00],
[-2.3606e+00, -5.2909e+00, -6.1548e+00, ..., -6.2886e+00,
-4.2621e+00, -2.6856e+00],
...,
[-9.6166e-01, -2.3038e+00, -1.8459e+00, ..., -6.2748e+00,
-3.9883e+00, -2.6030e+00],
[-1.2730e+00, -3.7238e+00, -3.4107e+00, ..., -6.5157e+00,
-5.2800e+00, -3.9344e+00],
[-2.2271e+00, -5.0976e+00, -3.0899e+00, ..., -4.7326e+00,
-3.5359e+00, -3.7731e+00]],
[[-1.2311e+00, -3.7774e-01, -9.3010e-01, ..., 1.1079e+00,
1.0689e+00, 1.7850e+00],
[ 2.9669e-01, -1.1926e+00, -7.2109e-01, ..., 1.7103e+00,
1.4606e+00, 2.9820e+00],
[ 4.6401e-01, 1.0733e+00, 1.9807e+00, ..., -2.6856e+00,
-3.8699e+00, 4.7828e-01],
...,
[ 1.2700e+00, -1.0176e-01, 7.8301e-01, ..., 1.9809e-01,
8.5325e-01, 1.4059e+00],
[ 1.8234e+00, 9.7198e-01, 1.6405e+00, ..., 1.0126e+00,
2.4079e-01, 2.1962e+00],
[ 1.0417e+00, 4.6796e-01, 9.1614e-01, ..., 1.0576e+00,
1.2727e+00, 1.5952e+00]],
[[-7.5343e-01, 8.0018e-01, -2.1954e-01, ..., -3.3082e-01,
-6.1810e-01, 8.7130e-01],
[-1.9368e+00, -3.0056e-01, -2.0443e+00, ..., -2.5606e-01,
-4.8019e-01, 7.1589e-01],
[-2.8211e+00, 7.2972e-01, 1.3546e+00, ..., 1.1495e-01,
-4.4626e-01, -8.1157e-01],
...,
[-1.7879e+00, -1.4353e+00, -1.2069e+00, ..., -2.5613e+00,
-1.7608e+00, -6.6807e-01],
[-1.7693e+00, -1.0678e+00, -9.4269e-01, ..., -2.0954e+00,
-1.7247e+00, -8.5798e-02],
[-3.6541e-01, 4.2440e-01, -3.0261e-01, ..., -1.1694e-01,
6.7246e-02, 7.6563e-01]]],
[[[ 3.6122e-02, -5.6952e-01, -1.8756e+00, ..., 1.5583e+00,
1.1698e+00, 4.3714e-01],
[-4.2527e+00, -6.6440e-01, 2.8287e+00, ..., -4.1167e-01,
8.5047e-01, 1.0896e-01],
[-9.6431e+00, -1.4205e+01, -1.6234e+01, ..., -2.5854e-01,
-1.4621e-01, 7.6688e-01],
...,
[ 1.2321e+00, 6.4494e-01, 4.5069e-01, ..., -3.2836e+01,
-2.6926e+01, -1.5501e+01],
[-6.6939e-04, -1.3709e+00, -1.5888e-01, ..., -1.5634e+01,
-1.2818e+01, -9.6143e+00],
[ 1.6038e-01, -8.2738e-01, 2.2422e-01, ..., -4.1591e+00,
2.7039e+00, -1.0580e+00]],
[[-1.3118e+01, -1.3114e+01, -8.3568e+00, ..., -1.4137e+00,
-3.0035e+00, -1.4491e+00],
[-5.1050e+00, -7.2068e+00, -4.8272e+00, ..., 3.0023e-01,
-1.9424e+00, -9.8694e-01],
[-8.7810e+00, -1.1116e+01, -1.6148e+01, ..., -8.9640e-01,
-1.8236e+00, -5.2120e-01],
...,
[ 3.4236e-01, 1.3992e+00, 1.0358e-01, ..., -1.9667e+01,
-2.4254e+01, -1.7615e+01],
[-2.0557e+00, 5.1808e-01, 5.6300e-01, ..., -1.4309e+01,
-1.8460e+01, -1.3294e+01],
[-1.9939e+00, -8.4244e-01, -3.4042e-01, ..., -1.6076e+01,
-1.6985e+01, -1.1053e+01]],
[[ 1.2563e+00, -5.8550e-01, 1.3623e+00, ..., 9.4652e-01,
-3.4637e-01, 2.2946e+00],
[-3.4602e+00, -9.3325e+00, -1.4778e+01, ..., -5.1567e-02,
-1.7203e+00, 1.8340e+00],
[-5.3307e+00, -8.8447e+00, -1.5865e+01, ..., 1.9869e-01,
6.3253e-03, 1.6986e+00],
...,
[ 8.7399e-01, 9.1869e-01, -7.4488e-01, ..., -8.8453e+00,
-7.4152e+00, -1.4150e+01],
[ 9.0692e-01, 1.9002e+00, -2.3536e+00, ..., -4.9591e+00,
-7.6983e+00, -9.6855e+00],
[ 2.0694e+00, 1.0436e+00, -2.1478e+00, ..., -2.1996e+00,
-3.6757e+00, -3.3356e+00]],
...,
[[-3.2821e+00, -1.0618e+01, -8.4634e+00, ..., -2.8614e+00,
-2.1480e+00, -2.9099e+00],
[-5.9458e+00, -2.3555e+01, -2.3151e+01, ..., -1.5484e+00,
-2.2345e+00, -3.7581e+00],
[ 3.2960e-01, -3.0111e+00, -4.9674e+00, ..., 8.8616e-01,
-1.2109e-01, -9.7369e-01],
...,
[-1.4283e+00, -3.7957e+00, -8.6335e-01, ..., -6.1372e+00,
-9.9053e+00, -4.9660e+00],
[-2.5043e+00, -5.6003e+00, -2.1594e+00, ..., -6.5147e+00,
-1.1638e+01, -1.8975e+00],
[-2.8179e+00, -5.4719e+00, -1.6185e+00, ..., -2.5030e+00,
-1.1004e+01, -9.1837e-01]],
[[-1.0080e+01, -1.2410e+01, -8.6883e+00, ..., -7.2359e-03,
1.6506e-01, 1.1992e+00],
[ 3.4387e+00, -1.1632e+00, -4.2121e+00, ..., 6.0880e-01,
6.4007e-01, 1.6184e+00],
[ 1.3803e+00, 9.0841e-01, 2.2328e+00, ..., -2.4299e-02,
4.3411e-01, 1.4264e+00],
...,
[ 1.3749e+00, -2.1383e-01, 9.9928e-02, ..., -5.2941e+00,
-4.6282e+00, -4.8451e+00],
[ 1.7923e+00, 1.1666e+00, 1.4992e+00, ..., -1.3734e+01,
-1.4409e+01, -1.0417e+01],
[ 8.2408e-01, 6.4625e-01, 9.9053e-01, ..., -1.3972e+01,
-1.7868e+01, -1.1330e+01]],
[[-1.8657e+00, -9.9908e-01, -1.5798e+00, ..., 5.3704e-01,
1.9007e-03, 9.5252e-01],
[-1.3071e-01, 2.2479e-02, 2.1740e+00, ..., -6.0761e-01,
-3.8748e-01, 1.0450e+00],
[-5.4286e+00, -9.6391e+00, -6.5760e+00, ..., -1.1642e+00,
-1.1667e+00, 4.9389e-01],
...,
[-1.7569e+00, -1.0622e+00, 3.2821e-01, ..., -1.4684e+01,
-1.9249e+01, -1.3975e+01],
[-1.8004e+00, -1.3774e+00, 1.2115e-01, ..., -8.3364e+00,
-1.0117e+01, -8.6862e+00],
[-5.6655e-01, 1.8933e-01, -1.0752e+00, ..., -4.3891e+00,
-5.0978e+00, -3.4530e+00]]],
[[[ 6.4455e-01, 1.0283e+00, 7.2089e-01, ..., 9.4887e-01,
8.6531e-01, 3.2884e-01],
[-2.6985e-01, -1.5032e+00, -7.4710e-01, ..., 8.5687e-02,
2.9951e-01, -4.9514e-01],
[ 1.1272e-01, -4.8593e-01, -5.0896e-01, ..., -4.8385e-01,
-8.3913e-02, 4.8054e-01],
...,
[ 6.7482e-01, -4.1461e-01, -9.8340e-02, ..., 4.7845e-02,
4.8213e-01, -6.4892e-02],
[-4.8474e-01, -1.8678e+00, -1.1193e+00, ..., 4.2545e-01,
1.4433e+00, 4.2656e-01],
[ 3.3260e-01, -6.0104e-01, -1.2161e+00, ..., -4.6779e-02,
-3.2563e-01, 4.9664e-01]],
[[-3.4443e+00, -1.4269e+00, -3.2521e+00, ..., -1.5870e+00,
-2.9061e+00, -9.7671e-01],
[-4.3710e+00, -1.5543e+00, -2.4147e+00, ..., -1.5814e+00,
-2.6350e+00, -3.4788e-01],
[-3.5892e+00, -2.5691e+00, -2.8302e+00, ..., -2.9570e+00,
-2.5864e+00, -5.3743e-01],
...,
[-2.9320e+00, -1.9168e+00, -2.0516e+00, ..., -1.7200e+00,
-1.2487e+00, 5.0790e-01],
[-3.4431e+00, -1.4002e+00, -2.3216e+00, ..., -2.5304e-01,
-6.3405e-01, 3.8745e-01],
[-1.1998e+00, -9.4737e-01, -2.1755e+00, ..., -5.7731e-01,
-4.3197e-01, 5.0275e-02]],
[[ 2.2162e+00, 3.0737e+00, 2.4686e+00, ..., 2.2757e+00,
8.7339e-01, 1.9714e+00],
[ 4.9297e-01, 1.1924e+00, 1.7728e+00, ..., 1.1090e+00,
-7.2147e-02, 1.9163e+00],
[ 9.8330e-01, 1.2039e+00, 1.3086e+00, ..., 9.0661e-01,
1.2969e-01, 1.4174e+00],
...,
[ 1.1013e+00, 1.8175e+00, 1.0083e+00, ..., -6.7048e-02,
-3.7086e-01, 5.2933e-01],
[ 6.6215e-01, 1.7071e+00, 7.8415e-01, ..., 3.2082e-01,
6.9385e-01, 1.2691e+00],
[ 1.4805e+00, 1.4131e+00, 1.2755e+00, ..., 7.2303e-01,
6.0527e-01, 1.2877e+00]],
...,
[[-1.8843e+00, -5.0434e+00, -4.3727e+00, ..., -4.6568e+00,
-3.3449e+00, -3.8943e+00],
[-2.2174e+00, -4.2273e+00, -4.1464e+00, ..., -5.5574e+00,
-3.7090e+00, -3.8992e+00],
[-4.6346e-01, -2.0955e+00, -2.2205e+00, ..., -5.2929e+00,
-3.1534e+00, -1.6387e+00],
...,
[-1.3129e+00, -3.9803e+00, -5.0168e+00, ..., -5.2456e+00,
-2.8968e+00, -2.8315e+00],
[-1.3371e+00, -4.4807e+00, -5.2282e+00, ..., -6.4238e+00,
-3.7691e+00, -5.2360e+00],
[-1.4285e+00, -4.6480e+00, -3.9889e+00, ..., -4.5968e+00,
-1.9503e+00, -3.2062e+00]],
[[ 6.8475e-01, 7.4412e-01, -1.7140e-01, ..., 1.3528e+00,
9.1134e-01, 1.6656e+00],
[ 1.2343e+00, 1.1108e+00, 5.6650e-01, ..., 2.3914e+00,
1.8414e+00, 2.6707e+00],
[ 5.1162e-01, -6.6953e-01, -1.0276e-01, ..., 2.2455e+00,
1.0482e+00, 1.6231e+00],
...,
[ 5.6684e-01, 1.5786e-01, -2.4067e-02, ..., 2.5241e+00,
1.4104e+00, 1.6174e+00],
[ 1.5101e+00, 1.3674e+00, 8.0104e-01, ..., 3.0546e+00,
1.4501e+00, 2.3883e+00],
[ 6.4581e-01, 8.1764e-01, 6.2772e-01, ..., 1.3308e+00,
1.1988e+00, 1.3780e+00]],
[[-2.9398e-01, -2.4883e-01, -1.0314e+00, ..., 5.4095e-01,
1.7202e-01, 1.0704e+00],
[-1.2521e+00, -1.4144e+00, -2.4071e+00, ..., -3.2771e-01,
-9.0645e-02, 6.2133e-01],
[-1.6905e+00, -9.0501e-01, -1.0115e+00, ..., -9.5663e-02,
-5.9942e-02, -1.6060e-01],
...,
[-1.7862e+00, -1.1420e+00, -7.8772e-01, ..., -1.8306e+00,
-9.7129e-01, -1.5148e+00],
[-1.4436e+00, -1.7781e+00, -1.7143e+00, ..., -1.1923e+00,
-9.8361e-01, -9.8865e-02],
[-1.5251e-01, -1.6171e-01, -9.0314e-01, ..., -1.1859e+00,
-7.3356e-01, 2.6880e-01]]]], device='cuda:0'),
tensor([[[[ 8.1757e-01, 9.4160e-01, 6.0212e-01, ..., 6.9345e-01,
5.2498e-02, -4.8523e-01],
[-5.9682e-01, -1.0711e+00, 1.0096e-01, ..., 9.1199e-01,
1.4618e+00, -3.0180e-01],
[-6.6348e-03, -1.0941e-01, -8.1142e-03, ..., 1.1736e-01,
7.9590e-01, 9.1715e-01],
...,
[ 1.7573e-01, -9.2019e-01, -7.8595e-01, ..., 3.7302e-01,
5.0877e-01, 5.8163e-01],
[-6.5300e-01, -2.1160e+00, -1.7976e+00, ..., -5.0337e-01,
-2.4506e-01, -1.5907e-01],
[-1.5328e-01, -1.6111e+00, -1.2741e+00, ..., -1.1541e+00,
-1.5337e+00, -3.1727e-01]],
[[-3.3884e+00, -1.2345e+00, -1.3793e+00, ..., -3.0923e+00,
-3.3693e+00, -7.5133e-01],
[-2.9987e+00, -2.0403e+00, -1.2502e+00, ..., -2.1735e+00,
-1.8472e+00, -7.2706e-02],
[-1.9083e+00, -1.8780e+00, -3.1876e+00, ..., -2.4727e+00,
-1.5566e+00, -4.7910e-01],
...,
[-1.4233e+00, -1.0084e+00, -2.0833e+00, ..., -2.2743e+00,
-1.7884e+00, -4.8244e-01],
[-3.3403e+00, -1.0413e+00, -1.6829e+00, ..., -1.9556e+00,
-2.1861e+00, -1.0574e+00],
[-2.4187e+00, -1.5095e+00, -1.5118e+00, ..., -2.0944e+00,
-2.0243e+00, -9.7695e-01]],
[[ 2.2103e+00, 2.5947e+00, 1.9654e+00, ..., 2.5703e+00,
3.6354e-01, 2.1583e+00],
[ 3.8140e-01, 4.7373e-01, 1.8058e+00, ..., 1.5323e+00,
-1.9674e-02, 2.1206e+00],
[ 1.0329e+00, 9.6430e-01, 1.9556e+00, ..., 5.5184e-01,
7.3851e-01, 2.2664e+00],
...,
[ 1.0759e+00, 1.7705e+00, 1.8900e+00, ..., 9.2981e-01,
1.3778e+00, 2.0929e+00],
[ 5.2758e-01, 2.2544e+00, 1.1405e+00, ..., 6.9842e-01,
1.1184e+00, 1.7103e+00],
[ 1.6932e+00, 1.8179e+00, 1.3813e+00, ..., 1.6269e+00,
1.7426e+00, 1.6869e+00]],
...,
[[-1.9352e+00, -6.3522e+00, -4.4728e+00, ..., -2.3382e+00,
-2.9225e+00, -3.7448e+00],
[-2.5537e+00, -7.5975e+00, -4.8233e+00, ..., -4.6646e+00,
-4.4270e+00, -4.9204e+00],
[-1.4293e+00, -4.3511e+00, -2.7931e+00, ..., -2.3843e+00,
-2.5766e+00, -1.9962e+00],
...,
[-2.4120e+00, -5.2207e+00, -3.9205e+00, ..., -3.3852e+00,
-4.7675e+00, -4.9161e+00],
[-3.0418e+00, -5.9999e+00, -4.5476e+00, ..., -5.3035e+00,
-5.8740e+00, -5.9933e+00],
[-1.9097e+00, -4.6678e+00, -2.3903e+00, ..., -2.0435e+00,
-2.7731e+00, -3.1134e+00]],
[[-4.0234e-01, 2.8078e-01, 1.0194e+00, ..., -2.2122e-01,
9.5689e-01, 1.6296e+00],
[ 1.1100e+00, 3.0676e-01, 5.3014e-01, ..., 5.7629e-01,
1.8268e+00, 2.5089e+00],
[ 1.0278e+00, -1.1067e-01, -1.1801e-02, ..., 7.6286e-01,
1.3226e+00, 1.7704e+00],
...,
[ 1.4334e+00, 3.0170e-01, -1.6609e-01, ..., 1.8403e+00,
1.3575e+00, 1.6287e+00],
[ 1.9786e+00, 2.1194e+00, 1.6749e+00, ..., 2.0260e+00,
1.2846e+00, 1.9949e+00],
[ 4.0384e-01, 8.1262e-01, 1.0216e+00, ..., 8.6094e-01,
6.7930e-01, 1.6299e+00]],
[[-6.4242e-01, -2.6202e-01, -4.8572e-01, ..., -2.0316e-01,
-4.5345e-01, 5.7753e-01],
[-1.2289e+00, -1.2390e+00, -5.3518e-01, ..., -1.8364e-01,
4.4135e-01, 7.8825e-01],
[-1.8726e+00, -3.2422e-01, 2.7267e-01, ..., -1.7614e-01,
5.1587e-01, -1.5490e-02],
...,
[-1.9728e+00, -2.1854e+00, -1.7002e+00, ..., -5.4337e-01,
-9.2169e-01, -5.7011e-01],
[-1.6135e+00, -2.4237e+00, -1.8288e+00, ..., -5.9388e-01,
-1.1872e+00, -5.9788e-01],
[-8.3902e-01, -8.0659e-01, -8.6178e-01, ..., -8.2352e-01,
-9.7773e-01, -1.1436e-01]]],
[[[ 1.3134e+00, 1.5433e+00, 7.5361e-01, ..., -7.8599e-01,
5.3992e-01, -1.5690e-01],
[-5.7844e-01, -2.4805e+00, -1.9591e+00, ..., -1.5845e+00,
-2.0264e+00, -2.8102e+00],
[-5.3722e-01, -3.2079e+00, -3.6245e+00, ..., -1.8272e+00,
-2.6559e+00, -2.3475e+00],
...,
[ 2.2053e-01, -1.4218e+00, 1.6426e-01, ..., -1.2202e-02,
-2.2109e-01, 2.8938e-01],
[-1.3116e-01, -1.2189e+00, -1.9196e+00, ..., 2.6792e-02,
2.0750e-03, -1.5118e-01],
[ 7.3741e-01, -3.2678e-01, -8.9012e-01, ..., -3.4965e-01,
-1.0089e+00, 9.9370e-02]],
[[-5.0623e+00, -2.7212e+00, -2.5654e+00, ..., 1.0416e+00,
-2.3270e+00, -2.0337e+00],
[-5.9084e+00, -7.3431e+00, -3.5342e+00, ..., 5.1668e-01,
-4.5474e+00, -2.7243e+00],
[-5.6323e+00, -1.0433e+01, 1.2124e+00, ..., -1.5647e-01,
-2.0775e+00, -1.5570e+00],
...,
[-1.3111e+00, -3.0387e+00, -9.0347e+00, ..., -6.6420e-01,
-7.6656e-01, -5.8964e-01],
[-3.0454e+00, -1.5657e+00, -4.0231e+00, ..., -7.0203e-01,
-9.8296e-01, -1.0620e+00],
[-2.2695e+00, -1.5851e+00, -2.2475e+00, ..., -6.2099e-01,
-5.8277e-01, -4.2410e-01]],
[[ 3.2078e+00, 4.5508e+00, 4.2510e+00, ..., -3.8962e-01,
-1.7026e+00, -1.7368e+00],
[ 7.2607e-01, 1.0016e+00, 3.3115e+00, ..., -5.7874e+00,
-5.5290e+00, -2.1244e+00],
[ 1.7631e+00, -2.6086e-01, 6.4695e-01, ..., -3.4341e+00,
-1.3016e+00, -1.2550e+00],
...,
[-2.8410e-01, 2.1442e+00, 4.3478e+00, ..., -2.6190e-01,
7.3173e-01, 1.7869e+00],
[ 1.4217e-01, 1.4199e+00, 3.3168e+00, ..., -5.7999e-02,
1.0296e+00, 1.3994e+00],
[ 1.2728e+00, 1.3552e+00, 1.2489e+00, ..., 6.3149e-01,
1.4328e+00, 1.4704e+00]],
...,
[[-2.0691e+00, -8.1824e+00, -6.5849e+00, ..., -3.2439e+00,
-3.2398e+00, -3.2182e+00],
[-3.1521e+00, -8.4773e+00, -7.8766e+00, ..., -7.1652e+00,
-7.7268e+00, -8.5408e-01],
[-6.2346e-01, -7.9393e+00, -9.3750e+00, ..., -1.1631e+01,
-1.3349e+01, -7.7583e-01],
...,
[-9.5998e-01, -3.4345e+00, -5.9147e+00, ..., -4.6722e+00,
-3.7921e+00, -3.1655e+00],
[-1.2335e+00, -4.4445e+00, -5.3878e+00, ..., -5.5725e+00,
-4.7170e+00, -4.6254e+00],
[-2.1587e+00, -5.0687e+00, -3.4640e+00, ..., -3.6441e+00,
-2.7730e+00, -2.6698e+00]],
[[-1.1790e+00, -1.6488e-01, -2.9564e-02, ..., 1.8374e+00,
1.1089e+00, -4.1971e-01],
[ 9.9796e-02, -5.0531e+00, -1.7231e+00, ..., -3.3272e-01,
-9.1043e-01, 5.0381e-01],
[-2.9582e+00, -6.2635e+00, 1.0627e+00, ..., -9.1408e-01,
-2.6833e+00, 3.0815e-02],
...,
[ 1.1155e+00, -2.4059e+00, -4.9979e+00, ..., 1.6284e+00,
8.8697e-01, 1.1767e+00],
[ 1.8036e+00, 6.3672e-01, 9.3842e-01, ..., 1.4329e+00,
7.9787e-01, 1.7071e+00],
[ 9.9767e-01, 2.7123e-01, 5.4894e-01, ..., 8.6457e-01,
7.0392e-01, 1.5432e+00]],
[[-9.4186e-01, 9.1369e-01, 1.5166e-01, ..., 1.1282e-01,
-6.3209e-01, 2.6185e-01],
[-2.2835e+00, -7.4293e-01, -1.4746e+00, ..., 5.8099e-01,
-4.2928e+00, -2.6660e+00],
[-4.5274e+00, -1.3881e+00, 3.7077e+00, ..., 3.9403e-01,
-4.4500e+00, -3.4340e+00],
...,
[-1.4598e+00, -1.6362e+00, 4.9974e-01, ..., -7.0177e-01,
-1.3166e+00, -8.2802e-01],
[-1.6250e+00, -1.5815e+00, -1.3748e+00, ..., -5.6535e-01,
-1.3698e+00, -6.2510e-01],
[-3.1223e-01, 4.1406e-01, -7.7356e-01, ..., -2.3210e-01,
-7.4867e-01, 4.3108e-02]]],
[[[ 9.2682e-01, 1.5736e+00, 1.3691e+00, ..., 3.4030e+00,
1.7447e+00, -8.1736e-02],
[-7.7997e-01, -1.6984e+00, -5.4399e-01, ..., -3.4645e+00,
2.7115e+00, 1.1800e+00],
[-3.2115e-01, -6.8711e-01, -9.4054e-01, ..., -6.7612e+00,
-2.6028e+00, -4.6268e-01],
...,
[ 7.2334e-01, -1.3283e+00, -3.7501e-01, ..., -7.0230e-02,
-1.5208e-01, 3.1938e-01],
[-3.2857e-01, -6.6055e-01, -7.9743e-02, ..., -6.7513e-02,
6.8854e-03, -1.1895e-01],
[ 5.6683e-01, -2.4003e-01, -1.4852e-01, ..., -3.6743e-01,
-1.0493e+00, 1.1513e-01]],
[[-3.7838e+00, -1.1374e+00, -2.3854e+00, ..., -4.2069e+00,
-5.1975e+00, -1.9513e+00],
[-3.9797e+00, -3.1046e+00, -2.9779e+00, ..., -2.3529e+00,
-3.0426e+00, -2.5546e+00],
[-2.7705e+00, -1.8959e+00, -4.2917e+00, ..., -6.6840e+00,
-7.1124e+00, -5.5107e+00],
...,
[-3.3278e+00, -3.4433e+00, -4.6081e+00, ..., -7.6771e-01,
-9.3967e-01, -8.6936e-01],
[-4.1834e+00, -3.3105e+00, -2.1694e+00, ..., -7.1994e-01,
-1.0226e+00, -1.2586e+00],
[-3.3800e+00, -2.5364e+00, -1.8167e+00, ..., -7.0983e-01,
-6.6208e-01, -4.8036e-01]],
[[ 2.6909e+00, 3.8261e+00, 3.0377e+00, ..., -3.9360e+00,
-9.4683e-01, 1.0034e+00],
[ 1.3041e-02, 8.7626e-01, 2.6479e+00, ..., -7.0067e+00,
-1.7353e+00, 9.2637e-01],
[ 3.3858e-01, 1.9628e+00, 3.1355e+00, ..., -8.4589e+00,
-7.3973e-01, 8.4690e-01],
...,
[-4.3408e+00, -3.0793e+00, -2.8247e-01, ..., -3.0553e-01,
6.7694e-01, 1.6317e+00],
[-5.7542e-01, 2.7719e+00, 2.3900e+00, ..., -2.3238e-01,
8.7130e-01, 1.2123e+00],
[ 1.6083e+00, 1.8679e+00, 1.7886e+00, ..., 4.9340e-01,
1.2918e+00, 1.3780e+00]],
...,
[[-1.5812e+00, -6.7105e+00, -5.5772e+00, ..., -4.7021e+00,
-3.9395e+00, -5.3208e+00],
[-2.2559e+00, -5.9815e+00, -4.6859e+00, ..., -4.9136e+00,
-9.0499e+00, -1.1014e+01],
[-8.2403e-01, -3.3442e+00, -3.7918e+00, ..., -1.4428e+00,
-7.2290e+00, -5.8401e+00],
...,
[-8.6389e-01, -4.8577e+00, -6.4547e+00, ..., -4.3682e+00,
-3.5201e+00, -2.9953e+00],
[-1.1038e+00, -5.0593e+00, -5.4048e+00, ..., -5.3640e+00,
-4.4984e+00, -4.4517e+00],
[-2.3837e+00, -5.1427e+00, -3.5600e+00, ..., -3.5100e+00,
-2.5917e+00, -2.4573e+00]],
[[-5.4250e-01, -8.3375e-02, -3.5967e-01, ..., -4.3905e+00,
-5.5894e-01, 1.2860e+00],
[ 1.6719e+00, 7.6997e-01, 3.6812e-01, ..., -4.2595e+00,
-1.8964e+00, 1.8592e+00],
[ 1.0518e+00, 3.3056e-01, -5.3304e-01, ..., -4.2743e+00,
-3.3062e+00, 1.3986e+00],
...,
[ 1.8811e-01, -1.9639e+00, -3.5131e-01, ..., 1.4676e+00,
7.3250e-01, 1.0224e+00],
[ 1.5630e+00, 1.6628e-01, 3.0333e+00, ..., 1.3858e+00,
7.2855e-01, 1.5217e+00],
[ 9.0833e-01, 5.5477e-01, 1.4757e+00, ..., 7.7897e-01,
6.7044e-01, 1.4585e+00]],
[[-4.6026e-01, 5.9079e-01, -1.7116e-01, ..., 3.3006e+00,
-1.1917e-01, -2.5355e+00],
[-1.0594e+00, -1.2356e+00, -2.0665e+00, ..., 1.5044e+00,
1.5310e+00, -4.1444e+00],
[-1.7439e+00, -6.9829e-01, -1.1736e+00, ..., -1.9340e+00,
-1.5894e+00, -6.1878e+00],
...,
[-1.1509e+00, -3.2379e+00, -5.3173e+00, ..., -9.2423e-01,
-1.4609e+00, -9.0086e-01],
[-1.8633e+00, -1.4087e+00, -1.1530e+00, ..., -7.6339e-01,
-1.4902e+00, -7.2963e-01],
[-8.9116e-01, 6.6337e-01, -1.1215e-01, ..., -2.5694e-01,
-8.1429e-01, -2.7488e-02]]],
[[[ 6.5660e-01, 1.0334e+00, 7.4095e-01, ..., 4.4421e-01,
3.7899e-01, 2.1936e-01],
[-1.5147e-01, -1.1561e+00, -6.0473e-01, ..., 3.3337e-01,
3.4258e-01, -1.1877e+00],
[ 1.3477e-02, -7.1637e-01, -2.9778e-01, ..., -4.5709e-01,
1.0286e+00, 6.2198e-01],
...,
[ 1.3571e+00, 1.0626e-01, 7.3995e-02, ..., 1.6585e+00,
1.4067e+00, 2.6918e-01],
[-6.3468e-01, -1.2448e+00, -2.4383e-01, ..., 2.6116e-01,
7.3992e-01, -4.2805e-01],
[ 2.6851e-01, -6.4324e-01, -8.6932e-01, ..., -6.9422e-02,
6.3301e-01, 1.0196e+00]],
[[-3.5200e+00, -1.4922e+00, -3.2871e+00, ..., -1.7073e+00,
-2.1561e+00, -1.0775e+00],
[-4.7031e+00, -2.5527e+00, -2.7906e+00, ..., -4.9395e-01,
-1.2398e+00, -2.9150e-01],
[-5.1879e+00, -6.2455e+00, -7.2130e+00, ..., -1.5454e+00,
-1.4310e+00, -7.4272e-01],
...,
[-2.7977e+00, -2.9139e+00, -5.5733e+00, ..., -6.1433e-01,
-1.9865e+00, -3.1271e-01],
[-3.5962e+00, -1.4546e+00, -2.4681e+00, ..., -3.2852e+00,
-2.9177e+00, -2.1051e+00],
[-1.3253e+00, -1.0816e+00, -1.7836e+00, ..., -6.6287e-01,
-9.4118e-01, -1.5389e+00]],
[[ 2.2045e+00, 3.1308e+00, 2.5425e+00, ..., 2.8613e+00,
1.4164e+00, 2.9043e+00],
[ 7.0119e-01, 1.3535e+00, 2.3020e+00, ..., 1.9232e+00,
1.8135e-02, 2.4867e+00],
[ 2.2786e+00, 2.6058e+00, 2.7406e+00, ..., 1.2556e+00,
5.9898e-01, 1.7762e+00],
...,
[ 1.7262e+00, 1.8774e+00, 1.5451e+00, ..., -1.0970e+00,
-2.4286e+00, -2.0295e-01],
[ 9.5692e-01, 2.6957e+00, 2.7975e+00, ..., -1.2572e+00,
-2.3810e+00, 1.4015e-02],
[ 1.5154e+00, 1.5935e+00, 1.5079e+00, ..., 1.3252e+00,
1.7309e+00, 1.9194e+00]],
...,
[[-1.8950e+00, -5.1522e+00, -4.4495e+00, ..., -3.3131e+00,
-2.4633e+00, -3.9483e+00],
[-2.6037e+00, -5.9719e+00, -5.7356e+00, ..., -5.3499e+00,
-3.8556e+00, -4.3636e+00],
[-1.0434e+00, -3.9175e+00, -4.4148e+00, ..., -2.6571e+00,
-2.4944e+00, -1.6224e+00],
...,
[-1.8102e+00, -3.6480e+00, -4.5879e+00, ..., -5.7136e+00,
-1.6594e+00, -5.2773e+00],
[-1.2531e+00, -4.9504e+00, -5.6619e+00, ..., -6.8422e+00,
-3.5998e+00, -4.8177e+00],
[-1.4274e+00, -4.8223e+00, -4.3342e+00, ..., -4.2708e+00,
-3.4163e+00, -4.3162e+00]],
[[ 6.7047e-01, 6.8098e-01, -1.3304e-01, ..., 8.5887e-01,
1.0822e+00, 1.2268e+00],
[ 8.4905e-01, 4.7905e-01, 5.5391e-01, ..., 1.5601e+00,
2.3959e+00, 2.5621e+00],
[-2.1215e-01, -2.1641e+00, -2.0761e+00, ..., 9.9216e-01,
1.6436e+00, 1.9304e+00],
...,
[-6.9001e-02, -8.5248e-01, -9.5977e-01, ..., -8.8655e-01,
7.4189e-01, 9.4573e-01],
[ 1.5813e+00, 1.7915e+00, 1.9562e+00, ..., -2.5963e-01,
2.1835e-01, 1.3705e+00],
[ 6.9031e-01, 8.5216e-01, 1.0120e+00, ..., 1.8593e+00,
1.1789e+00, 1.4127e+00]],
[[-3.1642e-01, -2.6298e-01, -1.0699e+00, ..., -1.9318e-01,
-1.3274e-01, 8.6041e-01],
[-1.4832e+00, -1.3726e+00, -1.6257e+00, ..., 8.0882e-02,
-8.0707e-02, 3.5948e-01],
[-2.3524e+00, -8.1888e-01, -5.0117e-01, ..., 2.0075e-01,
1.9210e-01, 2.6380e-01],
...,
[-1.5131e+00, -8.8550e-01, -2.0555e+00, ..., -7.6298e-01,
2.4670e-01, 8.5093e-02],
[-1.5120e+00, -1.6351e+00, -1.5457e+00, ..., -2.1952e+00,
-2.7426e+00, -9.1472e-01],
[-1.8668e-01, -8.1795e-02, -5.6047e-01, ..., 3.9933e-01,
-1.1618e-01, 1.0011e+00]]],
[[[ 7.0134e-01, 1.0349e+00, 6.8320e-01, ..., 4.0219e-01,
3.2567e-01, 1.8010e-01],
[-3.2542e-01, -1.4529e+00, -6.1234e-01, ..., 2.6985e-01,
3.0662e-01, -1.3259e+00],
[ 1.2913e-01, -8.6735e-01, -3.4092e-01, ..., -2.4022e-01,
6.9941e-01, 5.0600e-01],
...,
[ 6.3896e-02, -1.6967e+00, 6.0850e-01, ..., 5.4067e-02,
7.5239e-01, 4.8439e-01],
[-4.3386e-01, -2.3780e+00, 3.0783e-01, ..., -6.7347e-02,
1.3736e+00, -1.4911e-01],
[ 3.8618e-01, -9.1372e-01, 1.0562e-01, ..., -2.2141e-01,
3.8898e-01, 6.0859e-01]],
[[-3.1169e+00, -1.4411e+00, -3.0331e+00, ..., -1.6996e+00,
-2.0773e+00, -9.3368e-01],
[-4.4748e+00, -1.8162e+00, -2.3644e+00, ..., -2.7052e-01,
-1.2049e+00, -2.5552e-01],
[-4.4375e+00, -2.5516e+00, -3.0625e+00, ..., -1.6237e+00,
-1.7427e+00, -7.2159e-01],
...,
[-2.7381e+00, -2.2517e+00, -2.2490e+00, ..., -1.2102e-02,
1.7952e-01, 4.6650e-01],
[-3.6314e+00, -1.7278e+00, -1.9379e+00, ..., -1.4093e-01,
-4.1799e-01, 2.4730e-01],
[-1.2341e+00, -1.3284e+00, -1.8696e+00, ..., -3.7693e-01,
-6.7698e-01, -5.4517e-01]],
[[ 1.7938e+00, 2.8065e+00, 2.3514e+00, ..., 2.7343e+00,
1.4164e+00, 2.8259e+00],
[ 1.9980e-01, 1.1810e+00, 1.5198e+00, ..., 1.9998e+00,
6.6024e-02, 2.3750e+00],
[ 7.2549e-01, 1.4238e+00, 8.9151e-01, ..., 1.5937e-01,
1.2396e+00, 1.9858e+00],
...,
[ 1.0972e+00, 2.1198e+00, -5.5181e-01, ..., 1.3477e+00,
-1.7719e-01, 1.4451e+00],
[ 7.2373e-01, 2.5072e+00, 2.5343e-01, ..., 7.3417e-01,
6.5020e-01, 1.5435e+00],
[ 1.5921e+00, 1.9645e+00, 1.7569e+00, ..., 1.4693e+00,
1.1041e+00, 2.0091e+00]],
...,
[[-1.5201e+00, -4.7438e+00, -3.9329e+00, ..., -3.2319e+00,
-2.2867e+00, -3.7440e+00],
[-1.9271e+00, -4.0665e+00, -4.3747e+00, ..., -4.8746e+00,
-3.4622e+00, -4.2325e+00],
[-1.0176e+00, -3.3003e+00, -3.8578e+00, ..., -1.4295e+00,
-9.9474e-01, -1.2565e+00],
...,
[-1.6231e+00, -4.1519e+00, -4.3062e+00, ..., -5.1363e+00,
-4.1454e+00, -1.5906e+00],
[-1.5869e+00, -4.7296e+00, -5.2973e+00, ..., -6.2928e+00,
-3.3823e+00, -3.6820e+00],
[-1.6518e+00, -5.0902e+00, -3.6787e+00, ..., -3.6016e+00,
-2.1616e+00, -2.7903e+00]],
[[ 7.5725e-01, 5.6767e-01, -8.2255e-02, ..., 8.7024e-01,
1.1349e+00, 1.2903e+00],
[ 1.1902e+00, 9.9351e-01, 5.2748e-01, ..., 1.4321e+00,
2.2415e+00, 2.5558e+00],
[ 5.8497e-01, 5.6448e-01, 6.7581e-02, ..., 3.0094e-01,
1.8598e+00, 2.2443e+00],
...,
[ 8.6432e-01, -5.4568e-01, 8.6746e-01, ..., 2.0935e+00,
1.9254e+00, 1.6480e+00],
[ 1.5363e+00, 9.5726e-01, 1.1937e+00, ..., 1.4316e+00,
1.5417e+00, 2.4632e+00],
[ 6.7642e-01, 5.3714e-01, 1.0289e+00, ..., 1.0691e+00,
1.3676e+00, 1.3703e+00]],
[[-2.6710e-01, -1.9784e-01, -1.0113e+00, ..., -2.1337e-01,
-1.5334e-01, 8.5162e-01],
[-1.3989e+00, -1.4159e+00, -2.0795e+00, ..., 1.2405e-01,
-2.1154e-01, 2.4868e-01],
[-1.7095e+00, -8.2133e-01, -1.0484e+00, ..., 5.1432e-01,
-8.1375e-02, 2.7586e-01],
...,
[-1.7444e+00, -1.9646e+00, -1.2363e+00, ..., 7.3533e-02,
2.5755e-02, -3.0999e-01],
[-1.4121e+00, -1.8756e+00, -6.1585e-01, ..., -1.4063e-01,
-4.5858e-01, 3.2311e-01],
[-1.6328e-01, -1.0065e-01, -1.2247e-01, ..., 2.0388e-01,
1.7002e-01, 3.4144e-01]]]], device='cuda:0'),
tensor([[[[ 8.2137e-01, 1.0420e+00, 5.8371e-01, ..., 6.2684e-01,
9.4888e-01, 4.3104e-01],
[-1.0989e+00, -9.9763e-01, 7.8479e-03, ..., -1.5244e-01,
-6.7398e-02, -3.0046e-01],
[-1.6093e-01, -9.6999e-01, -2.0043e-01, ..., -7.8279e-01,
-9.3461e-02, 7.5151e-01],
...,
[-1.0871e+00, 9.9233e-02, 4.7488e-01, ..., 1.6921e-01,
1.1442e+00, 5.5880e-01],
[ 4.9193e-01, -5.0648e-01, 1.5048e-01, ..., -1.4000e-01,
5.6152e-01, 2.3402e-01],
[ 4.9944e-01, -2.5321e-01, -8.0440e-01, ..., -3.0179e-01,
-8.7165e-01, 1.6773e-01]],
[[-4.1831e+00, -2.0723e+00, -3.0753e+00, ..., -1.5002e+00,
-1.9197e+00, -1.6896e+00],
[-3.4945e+00, -2.4998e+00, -1.6853e+00, ..., -1.4918e+00,
-2.0478e+00, -2.0364e+00],
[-2.5235e+00, -3.7563e+00, -3.2347e+00, ..., -1.1196e+00,
-1.8041e+00, -1.3748e+00],
...,
[-1.7375e+00, -4.0235e+00, -3.6990e+00, ..., -5.3643e-01,
-3.1941e-01, -9.1093e-01],
[-1.6699e+00, -9.9522e-01, -1.1182e+00, ..., -7.0952e-01,
-6.4011e-01, -7.7965e-01],
[ 9.3018e-02, -8.1146e-01, -8.8077e-01, ..., -5.7568e-01,
-5.9724e-01, -5.3427e-01]],
[[ 2.8514e+00, 3.4122e+00, 3.2543e+00, ..., 1.5298e+00,
9.2067e-01, 2.5350e+00],
[ 8.4631e-01, 1.1428e+00, 2.7950e+00, ..., 1.2475e+00,
6.9798e-01, 2.5638e+00],
[ 1.3090e+00, 1.1176e+00, 2.5508e+00, ..., 2.0100e-01,
7.9063e-01, 1.8128e+00],
...,
[-2.5215e+00, -2.0733e+00, -2.7905e+00, ..., -1.0742e-02,
4.5911e-01, 1.4548e+00],
[ 2.3593e-01, 1.8373e+00, 1.8277e+00, ..., -5.8361e-01,
1.0316e+00, 1.4739e+00],
[ 1.6143e+00, 1.2279e+00, 1.9142e+00, ..., 4.6164e-01,
1.3907e+00, 1.4307e+00]],
...,
[[-3.0402e+00, -8.2892e+00, -5.3567e+00, ..., -5.4781e+00,
-3.3823e+00, -4.0536e+00],
[-3.4445e+00, -9.7300e+00, -7.0840e+00, ..., -5.7085e+00,
-3.6544e+00, -3.0916e+00],
[-2.3618e+00, -7.1359e+00, -3.8491e+00, ..., -4.8027e+00,
-2.7134e+00, -1.4645e+00],
...,
[-2.0768e+00, -6.8869e+00, -4.0200e+00, ..., -5.9751e+00,
-3.3224e+00, -2.1399e+00],
[-2.8984e+00, -6.4568e+00, -4.2570e+00, ..., -5.8927e+00,
-4.0236e+00, -4.2528e+00],
[-2.4287e+00, -6.7281e+00, -3.5111e+00, ..., -3.7401e+00,
-2.4648e+00, -2.4924e+00]],
[[-8.0138e-01, -1.8159e-01, 2.3618e-01, ..., 1.9640e+00,
1.7071e+00, 1.2296e+00],
[ 1.4602e+00, 9.6121e-01, 1.5139e+00, ..., 2.0127e+00,
1.1309e+00, 1.7929e+00],
[ 9.0508e-01, -8.4577e-01, -1.5982e+00, ..., 1.7759e+00,
3.0393e-01, 1.0612e+00],
...,
[ 1.0567e+00, -1.9054e+00, -1.2439e+00, ..., 2.0181e+00,
5.0397e-01, 9.1563e-01],
[ 1.8439e+00, 9.3138e-01, 1.1684e+00, ..., 1.6461e+00,
7.2046e-01, 1.6219e+00],
[ 1.1667e+00, 8.3482e-01, 1.2324e+00, ..., 8.2684e-01,
7.0930e-01, 1.4610e+00]],
[[-7.8379e-01, -7.3400e-01, -5.8377e-01, ..., -2.7054e-01,
-5.5590e-01, -3.5900e-02],
[-1.5632e+00, -1.5870e+00, -2.6930e-01, ..., -1.4663e+00,
-1.3435e+00, -1.0464e+00],
[-2.6173e+00, -1.3526e+00, 3.1774e-01, ..., -8.6124e-01,
-1.0886e+00, -3.3724e-01],
...,
[-1.6258e+00, -3.5743e+00, -2.2998e+00, ..., -5.2299e-01,
-8.5931e-01, -9.0742e-01],
[-1.9806e+00, -1.7079e+00, -9.2481e-01, ..., -1.1589e+00,
-1.4178e+00, -5.3275e-01],
[ 5.2315e-02, 7.9432e-02, -2.2905e-01, ..., -3.8671e-01,
-7.4620e-01, 4.0978e-02]]],
[[[ 5.7365e-01, 8.3271e-01, 5.3645e-01, ..., 3.4402e-01,
4.2429e-01, 2.2133e-02],
[-2.4989e-01, -7.4272e-01, -3.3163e-01, ..., 1.6814e-01,
7.8223e-02, -2.7555e-01],
[-1.4862e-01, 1.9290e-01, 2.1570e-01, ..., -1.1758e-01,
-8.5714e-03, 6.2613e-01],
...,
[ 4.8204e-01, -2.1574e-01, 1.6535e-01, ..., 7.3171e-01,
7.0752e-01, 6.7125e-01],
[-9.6884e-03, -5.2278e-01, 1.8039e-01, ..., 1.2601e+00,
1.4675e+00, 5.3339e-02],
[ 8.4603e-01, -9.2581e-02, 5.8815e-02, ..., 1.1878e+00,
5.0517e-01, 7.2028e-01]],
[[-2.3951e+00, -6.3434e-01, -1.9050e+00, ..., -1.6683e+00,
-2.2569e+00, -1.5610e+00],
[-2.4551e+00, -7.6048e-01, -1.6972e+00, ..., -1.3135e+00,
-2.3604e+00, -1.7265e+00],
[-2.3413e+00, -1.8088e+00, -1.8666e+00, ..., -2.0332e+00,
-2.3520e+00, -1.1320e+00],
...,
[-4.4282e-01, -7.9341e-01, -8.3658e-01, ..., -4.7717e-01,
-2.4042e-01, 8.5389e-01],
[-1.9827e+00, -4.3555e-01, 5.2987e-01, ..., -1.1691e+00,
4.0393e-02, 7.9220e-01],
[-1.1859e+00, -1.0304e+00, -8.7946e-01, ..., -1.4539e+00,
-3.7960e-01, -1.4203e-01]],
[[ 1.9532e+00, 2.5301e+00, 1.6066e+00, ..., 1.6153e+00,
7.6666e-01, 2.1262e+00],
[-2.1627e-01, 4.0513e-01, 7.3954e-01, ..., 7.9301e-01,
3.2408e-01, 2.2745e+00],
[ 5.1961e-02, 9.0279e-01, 7.6478e-01, ..., -6.3556e-02,
7.5074e-01, 1.9852e+00],
...,
[ 4.4618e-01, 4.9901e-01, 1.9295e+00, ..., -6.2511e-02,
-4.3122e-01, 1.5900e+00],
[ 7.2917e-01, 1.9425e+00, 2.0608e+00, ..., -1.6275e-01,
3.7553e-01, 1.7467e+00],
[ 2.0419e+00, 1.9358e+00, 2.7150e+00, ..., 1.6676e+00,
1.0549e+00, 1.8681e+00]],
...,
[[-2.0036e+00, -4.8666e+00, -3.1291e+00, ..., -2.0519e+00,
-1.6496e+00, -3.1050e+00],
[-2.4360e+00, -5.2185e+00, -2.9984e+00, ..., -2.4076e+00,
-2.2729e+00, -3.0903e+00],
[-1.5691e+00, -5.5080e+00, -3.1440e+00, ..., -2.3861e+00,
-2.5431e+00, -2.2527e+00],
...,
[-2.0864e+00, -3.7522e+00, -2.0543e+00, ..., -5.9204e+00,
-2.5904e+00, -2.4180e+00],
[-2.0154e+00, -4.5979e+00, -4.0295e+00, ..., -6.5271e+00,
-3.2331e+00, -4.8307e+00],
[-2.1562e+00, -5.1408e+00, -2.6598e+00, ..., -2.4675e+00,
-1.3380e+00, -3.4536e+00]],
[[ 9.1344e-01, 1.6232e+00, 7.9581e-01, ..., 1.0717e+00,
1.5220e+00, 1.3885e+00],
[ 1.1815e+00, 1.1315e+00, 1.2958e+00, ..., 9.6016e-01,
8.3214e-01, 1.8877e+00],
[ 6.2535e-01, -1.8153e-01, 2.8862e-01, ..., 4.0728e-01,
8.8926e-02, 1.3846e+00],
...,
[ 1.3290e+00, -1.3250e-01, -2.2646e-01, ..., 1.8310e+00,
2.3705e+00, 1.4109e+00],
[ 1.9471e+00, 1.0199e+00, 1.9391e+00, ..., 1.0316e+00,
2.1996e+00, 2.5105e+00],
[ 1.2564e+00, 6.9411e-01, 1.3187e+00, ..., 6.8039e-01,
1.8992e+00, 1.5258e+00]],
[[-1.0642e-01, -6.5938e-01, -1.1548e+00, ..., -5.5466e-01,
-8.7159e-01, -1.1878e-01],
[-8.6033e-01, -2.7876e+00, -2.1364e+00, ..., -1.1897e+00,
-1.0354e+00, -7.1441e-01],
[-1.4305e+00, -2.3371e+00, -9.3814e-01, ..., -2.8317e-01,
-4.5438e-01, -2.4885e-01],
...,
[-1.5586e+00, -1.4735e+00, -2.2613e-01, ..., -4.9849e-01,
-2.5534e-01, -4.3913e-01],
[-1.5898e+00, -1.1847e+00, -4.8203e-01, ..., -1.8320e+00,
-6.9049e-01, 3.1271e-01],
[-3.2093e-02, 5.2905e-01, 9.9700e-03, ..., -7.8207e-01,
-7.4421e-02, 5.9320e-01]]],
[[[ 8.5369e-01, 1.5095e+00, 1.2126e+00, ..., 1.4131e+00,
1.0929e+00, 9.8103e-01],
[-1.0160e+00, -2.0753e+00, -9.7818e-01, ..., -4.1552e-01,
-5.9866e-01, -1.5642e+00],
[-2.1471e-01, -7.3536e-01, 1.0146e-01, ..., -4.8774e-01,
6.9652e-01, 7.8044e-01],
...,
[-1.6776e-03, -1.1622e+00, -9.6938e-01, ..., 9.1311e-01,
6.3267e-01, 4.1383e-01],
[-6.2144e-01, -1.8364e+00, -1.2154e+00, ..., -4.5197e-01,
-2.8189e-01, -1.7296e-01],
[-3.3419e-01, -1.6554e+00, -1.6496e+00, ..., -1.1869e+00,
-1.4636e+00, -2.4988e-01]],
[[-3.7968e+00, -7.7634e-01, -1.5827e+00, ..., -1.5558e+00,
-2.4905e+00, -1.2620e+00],
[-4.0565e+00, -2.8684e+00, -2.2328e+00, ..., -2.2080e+00,
-3.1391e+00, -9.7634e-01],
[-2.3861e+00, -1.7140e+00, -2.8659e+00, ..., -2.0760e+00,
-2.8413e+00, -1.0906e+00],
...,
[-1.5864e+00, -1.6690e+00, -2.4028e+00, ..., -1.3808e+00,
-1.9560e+00, -6.9900e-01],
[-3.1112e+00, -1.6396e+00, -2.5812e+00, ..., -2.2568e+00,
-2.2996e+00, -1.1742e+00],
[-2.3174e+00, -1.9390e+00, -2.5887e+00, ..., -1.8809e+00,
-1.7628e+00, -9.1084e-01]],
[[ 2.9329e+00, 4.2817e+00, 3.3839e+00, ..., 2.8647e+00,
1.0388e+00, 3.0614e+00],
[ 2.5400e-01, 1.0327e+00, 2.2298e+00, ..., 9.9628e-01,
-1.0057e+00, 2.7719e+00],
[ 1.0231e+00, 1.4961e+00, 2.3887e+00, ..., 1.1948e+00,
-2.8842e-01, 1.9833e+00],
...,
[ 9.9703e-01, 1.0731e+00, 1.2726e+00, ..., 6.1624e-01,
1.6062e+00, 2.2400e+00],
[ 7.1753e-01, 1.8480e+00, 1.2486e+00, ..., 2.2601e-01,
1.1198e+00, 1.7895e+00],
[ 1.6342e+00, 1.6481e+00, 1.7732e+00, ..., 1.2924e+00,
1.7937e+00, 1.7509e+00]],
...,
[[-1.9451e+00, -7.3996e+00, -5.5773e+00, ..., -4.0005e+00,
-3.2120e+00, -5.5457e+00],
[-2.4475e+00, -7.0268e+00, -4.3916e+00, ..., -3.6717e+00,
-2.6396e+00, -4.3549e+00],
[-9.4740e-01, -4.4624e+00, -4.0338e+00, ..., -2.4590e+00,
-2.9264e+00, -2.4173e+00],
...,
[-2.6339e+00, -5.3804e+00, -4.1309e+00, ..., -4.8427e+00,
-4.9361e+00, -5.7187e+00],
[-2.7672e+00, -5.7282e+00, -5.3026e+00, ..., -6.2152e+00,
-5.7083e+00, -6.1037e+00],
[-1.4558e+00, -4.1461e+00, -2.9062e+00, ..., -3.2505e+00,
-2.7832e+00, -3.2715e+00]],
[[-1.8881e-01, 3.3192e-01, 5.2208e-01, ..., 6.5596e-01,
8.2927e-01, 1.3892e+00],
[ 1.6431e+00, 1.0969e+00, 1.4178e+00, ..., 1.5734e+00,
2.4374e+00, 3.0198e+00],
[ 1.2123e+00, 2.2002e-01, 3.3422e-03, ..., 1.8357e+00,
1.3864e+00, 2.2755e+00],
...,
[ 1.3546e+00, 2.9123e-01, -3.9535e-02, ..., 1.0934e+00,
1.4460e+00, 1.8955e+00],
[ 1.8380e+00, 1.7190e+00, 1.1096e+00, ..., 1.3881e+00,
1.2463e+00, 1.9627e+00],
[ 4.4665e-01, 7.8255e-01, 7.4340e-01, ..., 5.5199e-01,
7.2700e-01, 1.6664e+00]],
[[-5.6247e-01, 3.1234e-01, 1.1352e-01, ..., 5.8383e-01,
5.3726e-01, 1.3632e+00],
[-1.3836e+00, -1.8639e+00, -1.9706e+00, ..., -9.5385e-01,
-5.1054e-01, 2.2815e-01],
[-2.2811e+00, -8.6416e-01, 5.1000e-02, ..., 1.5328e-01,
1.6710e-02, 7.1028e-01],
...,
[-1.9188e+00, -1.8311e+00, -1.3400e+00, ..., -6.5376e-01,
-6.0139e-01, -1.4921e-01],
[-1.5741e+00, -2.0784e+00, -1.6446e+00, ..., -1.0238e+00,
-1.2897e+00, -6.6913e-01],
[-7.9771e-01, -8.4789e-01, -1.1711e+00, ..., -6.0087e-01,
-8.1162e-01, -6.0703e-02]]],
[[[ 8.6434e-01, 1.5075e+00, 1.1864e+00, ..., 1.4284e+00,
1.1086e+00, 1.0618e+00],
[-1.0261e+00, -2.0824e+00, -1.1846e+00, ..., -3.7872e-01,
-5.3666e-01, -1.6164e+00],
[-1.7448e-01, -8.0298e-01, 7.4175e-02, ..., -2.8008e-01,
6.4030e-01, 8.3060e-01],
...,
[-1.8022e-02, -1.1179e+00, -1.2020e+00, ..., -5.3038e-01,
-3.5796e-02, 2.4708e-01],
[-5.9166e-01, -1.8091e+00, -1.3100e+00, ..., -5.7460e-01,
-3.3795e-01, -1.8903e-01],
[-3.9147e-01, -1.6216e+00, -1.6600e+00, ..., -1.1799e+00,
-1.4769e+00, -2.5414e-01]],
[[-3.7452e+00, -7.7318e-01, -1.6059e+00, ..., -1.4719e+00,
-2.5348e+00, -1.3572e+00],
[-3.9654e+00, -2.8533e+00, -1.8300e+00, ..., -2.0432e+00,
-3.2252e+00, -1.0175e+00],
[-2.3361e+00, -1.7731e+00, -1.8501e+00, ..., -2.0274e+00,
-2.7446e+00, -1.0211e+00],
...,
[-1.5568e+00, -1.5836e+00, -2.6166e+00, ..., -2.0566e+00,
-1.9817e+00, -7.0853e-01],
[-2.9456e+00, -1.5863e+00, -2.7958e+00, ..., -2.4871e+00,
-2.5033e+00, -1.2863e+00],
[-2.2760e+00, -1.9212e+00, -2.6759e+00, ..., -1.8950e+00,
-1.8139e+00, -9.6750e-01]],
[[ 2.9299e+00, 4.2679e+00, 3.3571e+00, ..., 2.7867e+00,
8.7267e-01, 3.0209e+00],
[ 2.9906e-01, 1.0766e+00, 2.2650e+00, ..., 1.0662e+00,
-1.0377e+00, 2.6694e+00],
[ 1.0828e+00, 1.5221e+00, 1.5993e+00, ..., 1.8494e-01,
-4.0526e-01, 1.9523e+00],
...,
[ 9.1376e-01, 1.0279e+00, 1.4737e+00, ..., 2.9727e-01,
1.1760e+00, 2.3157e+00],
[ 6.8872e-01, 1.6784e+00, 1.0934e+00, ..., 1.3555e-01,
1.0652e+00, 1.8652e+00],
[ 1.6479e+00, 1.6219e+00, 1.8297e+00, ..., 1.3204e+00,
1.8684e+00, 1.8137e+00]],
...,
[[-1.9432e+00, -7.3710e+00, -5.5173e+00, ..., -3.8433e+00,
-3.2494e+00, -5.7839e+00],
[-2.4635e+00, -7.0320e+00, -4.2762e+00, ..., -3.4698e+00,
-2.7763e+00, -4.5379e+00],
[-9.9271e-01, -4.5208e+00, -3.6483e+00, ..., -2.9426e+00,
-3.0177e+00, -2.4871e+00],
...,
[-2.5486e+00, -5.2420e+00, -4.4372e+00, ..., -6.1811e+00,
-4.8357e+00, -4.7056e+00],
[-2.6424e+00, -5.6099e+00, -5.5590e+00, ..., -6.2277e+00,
-5.7211e+00, -6.1622e+00],
[-1.3605e+00, -3.9417e+00, -2.8397e+00, ..., -3.2225e+00,
-2.6978e+00, -3.2560e+00]],
[[-1.8536e-01, 2.9023e-01, 5.0457e-01, ..., 6.3315e-01,
7.4896e-01, 1.3170e+00],
[ 1.6530e+00, 1.0908e+00, 1.7850e+00, ..., 1.8516e+00,
2.5269e+00, 2.9690e+00],
[ 1.2148e+00, 2.4497e-01, 7.6913e-01, ..., 9.6590e-01,
1.4269e+00, 2.3300e+00],
...,
[ 1.2906e+00, 2.7630e-01, -3.5661e-01, ..., 1.3214e+00,
1.1656e+00, 1.6464e+00],
[ 1.8560e+00, 1.6435e+00, 9.2224e-01, ..., 1.4079e+00,
1.2321e+00, 1.9602e+00],
[ 4.7756e-01, 7.6717e-01, 7.0547e-01, ..., 5.3470e-01,
7.4915e-01, 1.6918e+00]],
[[-5.6489e-01, 2.8902e-01, 1.1872e-01, ..., 6.3485e-01,
5.7250e-01, 1.3674e+00],
[-1.3785e+00, -1.8693e+00, -1.6358e+00, ..., -5.6726e-01,
-4.6581e-01, 1.6536e-01],
[-2.2560e+00, -8.9728e-01, 2.6033e-01, ..., -7.5602e-02,
4.7598e-02, 7.0325e-01],
...,
[-1.9160e+00, -1.7036e+00, -1.3709e+00, ..., -4.1554e-01,
-7.5887e-01, -8.5130e-01],
[-1.6111e+00, -2.0035e+00, -1.7120e+00, ..., -1.1521e+00,
-1.3748e+00, -7.3033e-01],
[-8.7235e-01, -8.2284e-01, -1.1532e+00, ..., -5.8621e-01,
-7.8953e-01, -8.6916e-02]]],
[[[ 7.1258e-01, 1.0523e+00, 7.4023e-01, ..., 9.9056e-01,
9.2145e-01, 3.4657e-01],
[-3.4502e-01, -1.4120e+00, -5.5634e-01, ..., 2.0398e-02,
4.3626e-02, -8.1539e-01],
[ 1.4118e-01, -8.3813e-01, -3.7047e-01, ..., -8.4146e-01,
-1.6905e-01, -9.5765e-01],
...,
[ 5.6838e-02, -1.4821e+00, 8.9372e-01, ..., 1.1264e+00,
4.9716e-01, 1.4532e-01],
[-4.0473e-01, -2.0116e+00, -2.6912e-01, ..., 3.9342e-01,
1.5980e+00, 4.6041e-01],
[ 4.1473e-01, -7.0212e-01, -1.0236e+00, ..., 4.8501e-02,
-3.0769e-01, 5.3016e-01]],
[[-3.0954e+00, -1.5816e+00, -3.0545e+00, ..., -1.5489e+00,
-2.8095e+00, -9.0478e-01],
[-4.2697e+00, -1.7683e+00, -2.1041e+00, ..., -3.1134e+00,
-3.4986e+00, -1.4130e-01],
[-4.2191e+00, -2.4550e+00, -3.0202e+00, ..., -5.6695e+00,
-5.3029e+00, -1.2784e+00],
...,
[-2.7829e+00, -2.4501e+00, -2.5892e+00, ..., -1.1255e+00,
-9.4724e-01, 4.5140e-01],
[-3.5337e+00, -1.6275e+00, -2.7251e+00, ..., 3.6232e-02,
-4.7310e-01, 2.7631e-01],
[-1.2200e+00, -1.1211e+00, -2.1385e+00, ..., -5.6243e-01,
-4.3497e-01, 4.0809e-02]],
[[ 1.7735e+00, 2.7423e+00, 2.3760e+00, ..., 2.2456e+00,
7.6770e-01, 1.8350e+00],
[ 2.6202e-01, 1.1104e+00, 1.4940e+00, ..., 1.0325e+00,
-1.4145e-01, 2.5444e+00],
[ 7.6524e-01, 1.1955e+00, 4.4162e-01, ..., 1.6730e+00,
4.0783e-01, 2.0934e+00],
...,
[ 1.0549e+00, 2.3167e+00, 4.3554e-01, ..., 2.5376e-01,
-3.2045e-01, 1.1354e-01],
[ 7.1078e-01, 2.1837e+00, 1.7216e+00, ..., 2.7594e-01,
5.8724e-01, 1.2214e+00],
[ 1.5958e+00, 1.5447e+00, 1.8729e+00, ..., 8.0911e-01,
6.0371e-01, 1.2978e+00]],
...,
[[-1.4753e+00, -4.6554e+00, -3.8061e+00, ..., -4.6236e+00,
-3.3382e+00, -3.8819e+00],
[-1.8673e+00, -3.8166e+00, -4.0097e+00, ..., -6.3676e+00,
-4.9214e+00, -4.7436e+00],
[-9.9777e-01, -3.1118e+00, -3.4689e+00, ..., -6.6191e+00,
-3.9715e+00, -2.5353e+00],
...,
[-1.5932e+00, -4.1599e+00, -5.1465e+00, ..., -5.3198e+00,
-2.9923e+00, -2.8308e+00],
[-1.5184e+00, -4.8525e+00, -5.7152e+00, ..., -6.3955e+00,
-3.7981e+00, -4.9247e+00],
[-1.6527e+00, -4.9336e+00, -3.7847e+00, ..., -4.5024e+00,
-2.0043e+00, -3.2799e+00]],
[[ 7.5265e-01, 4.5596e-01, -1.6318e-01, ..., 1.3488e+00,
9.0118e-01, 1.6666e+00],
[ 1.2522e+00, 9.6394e-01, 4.9090e-01, ..., 2.2802e+00,
1.6981e+00, 2.9310e+00],
[ 6.7014e-01, 6.0500e-01, -9.2498e-02, ..., -1.4658e-01,
-1.6323e+00, 8.2279e-01],
...,
[ 8.4925e-01, -6.9155e-01, 1.4860e-01, ..., 2.0931e+00,
1.2309e+00, 1.4844e+00],
[ 1.5730e+00, 1.2065e+00, 4.3279e-01, ..., 3.0505e+00,
1.2715e+00, 2.3166e+00],
[ 6.5018e-01, 8.4591e-01, 7.5542e-01, ..., 1.3394e+00,
1.2183e+00, 1.3813e+00]],
[[-2.8090e-01, -1.2691e-01, -8.4758e-01, ..., 6.0060e-01,
2.8028e-01, 1.1100e+00],
[-1.4453e+00, -1.3310e+00, -1.8126e+00, ..., -5.1996e-01,
3.4917e-01, 3.4992e-01],
[-1.7557e+00, -9.7515e-01, -7.8140e-01, ..., -8.6134e-01,
2.7380e-01, -8.4224e-01],
...,
[-1.7507e+00, -1.7302e+00, -9.1977e-01, ..., -1.7599e+00,
-8.7172e-01, -1.6737e+00],
[-1.4003e+00, -1.8059e+00, -1.0209e+00, ..., -9.1425e-01,
-9.2469e-01, -8.2699e-02],
[-1.4556e-01, -8.7378e-02, -8.0902e-01, ..., -1.0412e+00,
-7.2460e-01, 2.6548e-01]]]], device='cuda:0'),
tensor([[[[ 6.8401e-01, 1.0494e+00, 7.0433e-01, ..., -4.4422e-02,
2.6418e-01, 4.1013e-02],
[-3.5382e-01, -1.4517e+00, -4.9823e-01, ..., -3.3687e-01,
-6.9257e-01, -1.5777e+00],
[ 1.2068e-01, -8.4859e-01, -4.5989e-01, ..., 4.3346e-01,
2.8552e-01, 9.8133e-01],
...,
[-3.5813e-01, 1.5326e-01, -9.1133e-01, ..., -4.5233e+00,
-2.1130e+00, -1.2283e-01],
[-8.7806e-01, -2.6761e+00, 3.7370e-01, ..., 9.9690e-01,
1.2023e+00, -5.5864e-01],
[ 3.8192e-01, -1.5420e+00, -5.2404e-01, ..., -4.3705e-01,
3.8080e-01, 6.4573e-01]],
[[-3.1767e+00, -1.6296e+00, -3.1556e+00, ..., -1.6388e+00,
-2.1814e+00, -1.0955e+00],
[-4.3797e+00, -1.7714e+00, -2.3041e+00, ..., -2.3792e+00,
-2.4161e+00, -5.5182e-02],
[-4.2235e+00, -2.4720e+00, -3.0831e+00, ..., -2.7813e+00,
-3.2678e+00, -8.6324e-01],
...,
[-3.2589e+00, -4.5717e+00, -4.3641e+00, ..., -9.1847e+00,
-2.7707e+00, 5.9344e-01],
[-3.7149e+00, -3.2587e+00, -6.5124e+00, ..., -3.1324e-01,
-1.3421e-01, 4.5901e-01],
[-1.1718e+00, -1.8238e+00, -4.3311e+00, ..., -2.6219e-01,
-7.3698e-01, -6.0059e-01]],
[[ 1.8181e+00, 2.8801e+00, 2.4239e+00, ..., 2.9804e+00,
1.3560e+00, 2.7629e+00],
[ 3.2730e-01, 1.2559e+00, 1.6849e+00, ..., 2.2109e+00,
8.3561e-01, 2.2010e+00],
[ 8.4210e-01, 1.2939e+00, 9.6673e-01, ..., -2.3817e-01,
1.3608e+00, 1.5963e+00],
...,
[ 2.5506e+00, 1.5146e+00, -7.3735e+00, ..., -5.6856e+00,
-5.8921e-01, 1.4067e+00],
[ 9.4551e-01, 4.1694e+00, 3.5795e-01, ..., 2.8177e+00,
5.8471e-01, 9.1944e-01],
[ 1.5215e+00, 2.2860e+00, 3.9876e+00, ..., 7.4947e-01,
1.1025e+00, 2.0833e+00]],
...,
[[-1.4915e+00, -4.6464e+00, -3.8707e+00, ..., -2.8322e+00,
-2.0617e+00, -4.1636e+00],
[-1.9227e+00, -4.0211e+00, -4.3361e+00, ..., -3.5724e+00,
-2.5635e+00, -4.9220e+00],
[-9.8626e-01, -3.1380e+00, -3.3462e+00, ..., -2.7016e+00,
-4.3199e+00, -3.8697e+00],
...,
[-1.8836e+00, -4.4352e+00, -4.8178e+00, ..., -1.1311e+01,
-7.9137e+00, -5.6545e+00],
[-1.9181e+00, -4.0081e+00, -6.6150e+00, ..., -1.0932e+01,
-1.0761e+01, -7.6809e+00],
[-1.7182e+00, -4.9165e+00, -4.2877e+00, ..., -6.1634e+00,
-2.2905e+00, -2.8422e+00]],
[[ 7.0712e-01, 4.3073e-01, -1.2389e-01, ..., 1.2529e+00,
1.4122e+00, 1.4776e+00],
[ 1.1842e+00, 9.8588e-01, 5.2466e-01, ..., 1.0862e+00,
2.9732e+00, 2.9589e+00],
[ 5.7248e-01, 5.7039e-01, 1.0899e-01, ..., -3.1525e+00,
1.2951e+00, 2.4398e+00],
...,
[ 1.5800e-01, -1.3048e+00, 3.2136e+00, ..., -2.4898e+00,
2.4935e+00, 3.2270e+00],
[ 1.3416e+00, -3.3724e-01, -4.2983e-01, ..., 3.8226e+00,
2.8624e+00, 2.9688e+00],
[ 6.6010e-01, -1.7264e-01, -7.4857e-01, ..., 1.2875e+00,
1.3435e+00, 1.3442e+00]],
[[-3.4152e-01, -1.8215e-01, -9.2917e-01, ..., -3.5806e-01,
-3.7012e-01, 8.0757e-01],
[-1.4012e+00, -1.3456e+00, -1.7317e+00, ..., -5.6859e-01,
-1.3213e+00, 2.7314e-01],
[-1.6696e+00, -7.6671e-01, -8.8509e-01, ..., 1.6518e+00,
-2.9897e-01, 6.7054e-01],
...,
[-1.7452e+00, -8.7845e-01, -2.3865e+00, ..., -7.0303e+00,
-3.6757e+00, -9.9101e-01],
[-1.4475e+00, -1.1343e+00, -1.8666e+00, ..., -5.4730e-01,
1.7558e-01, 5.9491e-01],
[-7.4894e-02, -3.5533e-01, -6.7619e-01, ..., 7.4996e-01,
1.6477e-01, 3.3631e-01]]],
[[[ 9.6036e-01, 1.5258e+00, 1.0602e+00, ..., 5.4078e-01,
3.0127e-01, -1.6003e-01],
[-8.9475e-01, -1.8230e+00, -1.0854e+00, ..., 3.4017e-01,
6.4855e-01, -6.6301e-01],
[ 5.4880e-02, -7.8169e-01, -2.4734e-01, ..., -4.2353e-01,
4.1498e-01, 6.8126e-01],
...,
[ 4.5513e-01, -2.6355e-01, -4.1721e-01, ..., 3.7457e-01,
8.6091e-01, 6.6732e-01],
[-1.8223e-01, -1.0728e+00, -4.3083e-01, ..., -7.2133e-02,
6.7063e-01, 1.7728e-02],
[ 7.6302e-01, -3.3174e-01, -3.4043e-01, ..., -4.6811e-01,
-4.1724e-01, 3.0180e-01]],
[[-4.3074e+00, -1.3056e+00, -2.0742e+00, ..., -2.9561e+00,
-3.4853e+00, -8.4294e-01],
[-4.1909e+00, -3.5253e+00, -3.2343e+00, ..., -2.0567e+00,
-3.1851e+00, -5.6807e-01],
[-2.8462e+00, -3.5502e+00, -4.9699e+00, ..., -2.4467e+00,
-3.3545e+00, -7.2887e-01],
...,
[-8.0441e-01, -6.1822e-01, -1.6717e+00, ..., -2.5719e+00,
-2.0197e+00, -3.5023e-02],
[-2.8527e+00, -4.7672e-01, -1.2088e+00, ..., -1.6426e+00,
-1.6722e+00, -2.4124e-02],
[-2.0524e+00, -1.5058e+00, -1.8349e+00, ..., -9.5372e-01,
-1.0760e+00, 7.5807e-03]],
[[ 3.0983e+00, 4.1070e+00, 3.2215e+00, ..., 2.2975e+00,
7.6090e-01, 2.3387e+00],
[ 4.1379e-01, 7.8867e-01, 2.2310e+00, ..., 1.8977e+00,
1.6814e-01, 2.3629e+00],
[ 9.4217e-01, 2.1112e+00, 2.5212e+00, ..., 9.5489e-01,
6.6552e-01, 2.1824e+00],
...,
[ 3.2402e-01, 9.9148e-01, 1.3943e+00, ..., 1.2855e+00,
6.1359e-01, 1.6840e+00],
[ 3.9252e-01, 2.1288e+00, 1.6630e+00, ..., 7.6670e-02,
1.8196e-01, 1.1519e+00],
[ 1.4496e+00, 1.3353e+00, 1.7646e+00, ..., 1.0645e+00,
8.3100e-01, 1.3237e+00]],
...,
[[-2.2167e+00, -7.5750e+00, -6.0144e+00, ..., -4.1837e+00,
-3.1070e+00, -3.7928e+00],
[-2.8794e+00, -6.8706e+00, -5.5416e+00, ..., -5.5915e+00,
-4.8133e+00, -4.5340e+00],
[-3.0213e-01, -3.4656e+00, -3.8597e+00, ..., -4.6207e+00,
-3.4763e+00, -1.8353e+00],
...,
[-9.6083e-01, -2.3806e+00, -1.8576e+00, ..., -6.1751e+00,
-4.2699e+00, -3.2690e+00],
[-1.3225e+00, -3.8058e+00, -3.5055e+00, ..., -6.4360e+00,
-4.6265e+00, -4.6589e+00],
[-2.2525e+00, -5.0366e+00, -3.0874e+00, ..., -4.4960e+00,
-2.7310e+00, -3.2219e+00]],
[[-7.5410e-01, -9.3307e-02, 7.4344e-02, ..., 1.0003e+00,
1.0629e+00, 1.7868e+00],
[ 1.8085e+00, 6.3015e-01, 1.6240e+00, ..., 1.5423e+00,
1.5525e+00, 2.9100e+00],
[ 1.1059e+00, -9.3228e-01, -1.4931e+00, ..., 1.0353e+00,
6.3792e-01, 2.2381e+00],
...,
[ 1.3030e+00, -1.3189e-01, 5.9976e-01, ..., 2.0253e+00,
1.2795e+00, 1.7515e+00],
[ 1.9042e+00, 9.3775e-01, 1.5397e+00, ..., 1.6247e+00,
1.0670e+00, 2.3049e+00],
[ 1.0495e+00, 4.3182e-01, 8.4491e-01, ..., 9.4526e-01,
1.1336e+00, 1.5542e+00]],
[[-5.2286e-01, 5.5141e-01, 5.1069e-02, ..., -2.8896e-01,
-5.9276e-01, 9.0191e-01],
[-1.1590e+00, -1.5655e+00, -1.2111e+00, ..., -2.5094e-01,
-1.0638e-01, 1.0717e+00],
[-2.1885e+00, -1.1506e+00, -3.5590e-01, ..., 1.9511e-01,
9.7849e-03, 5.3135e-01],
...,
[-1.7361e+00, -1.4978e+00, -1.3082e+00, ..., -2.8934e-01,
-6.0155e-01, -1.4599e-01],
[-1.7791e+00, -1.1215e+00, -1.1198e+00, ..., -7.8286e-01,
-9.6192e-01, 3.1439e-01],
[-3.7473e-01, 3.8061e-01, -4.6737e-01, ..., 1.0539e-01,
-2.6752e-01, 5.8796e-01]]],
[[[ 8.9274e-01, 1.5527e+00, 1.4950e+00, ..., -5.3994e-01,
6.2213e-01, 8.0228e-01],
[-7.1966e-01, -1.5635e+00, -1.2988e+00, ..., -1.1361e+00,
-1.9559e+00, -1.8884e+00],
[ 1.4722e-01, -1.1185e+00, -1.0055e+00, ..., 4.8995e+00,
2.3874e+00, 1.4939e+00],
...,
[ 7.5593e-02, -5.0759e-01, 2.4921e-01, ..., -1.8241e+00,
2.4564e+00, 2.4474e+00],
[-6.2867e-01, -1.8512e+00, -2.5986e-01, ..., -6.1978e+00,
-3.6105e+00, 6.4875e-01],
[-5.7875e-02, -1.9018e+00, -8.4954e-01, ..., -3.0946e+00,
-4.0302e+00, 2.6565e-01]],
[[-3.7500e+00, -7.4111e-01, -2.6676e+00, ..., -1.1970e+00,
-2.3187e+00, -1.4433e+00],
[-4.3091e+00, -5.3731e+00, -5.9091e+00, ..., -3.1301e+00,
-3.2527e+00, -1.6258e+00],
[-2.9274e+00, -7.2970e+00, -8.9365e+00, ..., -2.4167e+00,
-3.4075e+00, -1.0360e+00],
...,
[-1.2253e+00, -1.4159e+00, -1.8527e+00, ..., -7.2824e+00,
-4.7121e+00, -1.3129e+00],
[-3.2658e+00, -1.4459e+00, -1.9630e+00, ..., -1.1533e+01,
-5.9185e+00, -2.0405e+00],
[-2.4509e+00, -1.8003e+00, -2.0236e+00, ..., -8.3234e+00,
-6.3145e+00, -3.0360e+00]],
[[ 2.9378e+00, 4.4263e+00, 3.1479e+00, ..., 4.4852e+00,
1.3753e+00, 2.9458e+00],
[-3.6467e-02, 4.6982e-01, 2.4930e+00, ..., 1.3805e+00,
1.3677e+00, 2.7509e+00],
[ 6.2619e-01, 1.9654e+00, 3.5990e+00, ..., -3.9316e+00,
-1.1368e+00, 1.3791e+00],
...,
[ 1.4275e+00, 2.0393e+00, 3.6681e-02, ..., -7.8108e+00,
-4.9046e+00, -2.4413e+00],
[ 7.4534e-01, 2.9510e+00, 1.4580e+00, ..., -7.2783e+00,
-2.9936e+00, -8.7986e-01],
[ 1.7701e+00, 1.8937e+00, 2.0701e+00, ..., -2.3980e+00,
-1.1929e+00, 1.3205e+00]],
...,
[[-1.8696e+00, -7.2489e+00, -5.6083e+00, ..., -5.8119e+00,
-3.1923e+00, -5.7998e+00],
[-2.4197e+00, -7.1878e+00, -6.4711e+00, ..., -7.0675e+00,
-3.1187e+00, -4.5653e+00],
[-1.0340e+00, -5.5708e+00, -5.9982e+00, ..., -2.3325e+01,
-1.3354e+01, -5.5314e+00],
...,
[-2.7264e+00, -5.4591e+00, -3.0032e+00, ..., -2.5128e+00,
-1.7467e+01, -9.8645e+00],
[-3.3219e+00, -6.5717e+00, -4.7979e+00, ..., -2.4179e+00,
-3.3297e+00, -7.9975e+00],
[-2.2293e+00, -5.2193e+00, -2.3231e+00, ..., 4.5474e-01,
8.3874e-01, -2.0418e+00]],
[[-9.8019e-02, 3.2320e-01, -2.3718e-02, ..., 9.6646e-01,
7.8068e-01, 1.5626e+00],
[ 1.5543e+00, -9.8731e-02, 2.0748e-01, ..., 1.7651e-01,
2.5586e+00, 3.2052e+00],
[ 8.8041e-01, -3.4818e+00, -5.4726e+00, ..., -3.9831e+00,
3.2647e-02, 2.2616e+00],
...,
[ 1.4468e+00, -3.5389e-02, 4.4883e-01, ..., -4.6077e+00,
-3.2397e+00, 2.3241e+00],
[ 2.1402e+00, 2.0593e+00, 2.0611e+00, ..., -3.2781e+00,
-3.6415e+00, 3.1309e-01],
[ 4.5068e-01, 9.6651e-01, 1.2814e+00, ..., -1.9925e+00,
-1.7683e+00, -2.6687e-01]],
[[-5.2062e-01, 2.6051e-01, -2.9959e-01, ..., 1.3404e-03,
1.9906e-01, 1.2551e+00],
[-1.3513e+00, -2.7724e+00, -2.3151e+00, ..., -7.1703e-01,
-3.1190e+00, 9.3832e-02],
[-2.6522e+00, -2.3605e+00, -1.3338e+00, ..., 3.2358e+00,
1.6183e-02, 9.2863e-01],
...,
[-1.8871e+00, -1.9144e+00, -1.3064e+00, ..., 3.7271e+00,
5.5850e-01, -1.9302e+00],
[-1.6435e+00, -2.2135e+00, -1.8731e+00, ..., -1.7753e+00,
-3.2653e-01, -3.6118e-01],
[-8.5682e-01, -8.1212e-01, -1.1383e+00, ..., -3.2154e+00,
-5.9732e-01, -6.7343e-01]]],
[[[ 1.0589e+00, 1.1757e+00, 4.3956e-01, ..., 9.2399e-01,
4.6691e-01, 2.4060e-01],
[-5.4482e-01, -2.8220e+00, -2.0663e+00, ..., 1.3890e+00,
1.4854e-01, -1.4835e+00],
[-4.7367e-01, -1.5505e+00, 2.0308e-01, ..., -7.1598e-01,
1.5513e-01, 7.2209e-01],
...,
[-2.0969e-01, 7.8941e-01, -2.3149e+00, ..., 3.9441e-01,
1.0095e+00, 7.8037e-01],
[ 2.9596e-02, 6.8246e-01, -1.2146e+00, ..., 9.4010e-01,
1.5936e+00, 1.2502e-01],
[ 8.4984e-01, 3.6900e-01, 1.6301e+00, ..., 3.8044e-01,
2.7521e-01, 6.6170e-01]],
[[-2.7505e+00, -1.0411e+00, -3.5503e+00, ..., -2.4270e+00,
-2.2912e+00, -6.6620e-01],
[-3.8330e+00, -4.4332e+00, -3.0617e+00, ..., -2.6198e+00,
-4.3985e+00, -1.5828e+00],
[-2.9705e+00, -3.1960e+00, -1.3166e+00, ..., -3.1199e+00,
-4.2371e+00, -1.4034e+00],
...,
[-7.2660e-01, 7.5544e-01, 1.5761e+00, ..., -5.0313e-01,
4.5714e-02, 8.6480e-01],
[-2.5320e+00, -3.0724e-01, 7.6971e-01, ..., -2.2207e-02,
1.0129e-01, 7.5402e-01],
[-2.0854e+00, -2.7389e+00, -4.8158e+00, ..., -5.6037e-01,
-5.3326e-01, -1.4738e-01]],
[[ 2.4166e+00, 3.8715e+00, 1.4554e+00, ..., 1.8925e+00,
7.8181e-01, 2.5979e+00],
[ 9.5072e-01, 1.7843e+00, -1.2507e+00, ..., 1.2507e+00,
-2.3918e-01, 2.5773e+00],
[ 1.6450e+00, 2.6590e+00, -2.6996e+00, ..., 2.8748e-01,
1.1766e+00, 2.6578e+00],
...,
[ 2.0382e+00, -2.2580e+00, -1.7192e+00, ..., 9.7939e-01,
5.4499e-01, 1.8174e+00],
[ 1.8075e+00, 1.5324e+00, -1.2130e+00, ..., 1.0313e+00,
7.2382e-01, 1.6504e+00],
[ 2.6133e+00, 3.0879e+00, 2.4181e+00, ..., 1.9753e+00,
1.0379e+00, 1.6946e+00]],
...,
[[-2.5464e+00, -6.9985e+00, -7.6548e+00, ..., -2.4120e+00,
-4.5570e+00, -5.3115e+00],
[-3.2920e+00, -8.2446e+00, -8.7554e+00, ..., -3.8115e+00,
-7.7553e+00, -5.4644e+00],
[-2.2297e+00, -6.2547e+00, -6.5654e+00, ..., -1.4663e+00,
-6.9101e+00, -4.0124e+00],
...,
[-2.1077e+00, -4.0234e+00, -2.3721e+00, ..., -3.2215e+00,
-2.7132e+00, -2.8691e+00],
[-2.7881e+00, -5.5283e+00, -3.9708e+00, ..., -4.5631e+00,
-3.1012e+00, -4.7827e+00],
[-2.9312e+00, -6.4948e+00, -3.2636e+00, ..., -3.0351e+00,
-1.5967e+00, -3.2005e+00]],
[[ 2.9717e-01, -3.7794e-01, -2.7541e+00, ..., 6.6021e-01,
2.0573e+00, 1.8636e+00],
[ 1.1702e+00, -8.8012e-01, -1.0977e+00, ..., -1.6864e-02,
1.6780e+00, 3.0167e+00],
[ 1.2269e+00, -2.3868e-01, 8.2743e-01, ..., 3.1355e-01,
1.5159e+00, 2.4764e+00],
...,
[ 1.4366e+00, 1.9054e+00, 1.0837e+00, ..., 1.9184e+00,
2.0079e+00, 1.4500e+00],
[ 1.5326e+00, 1.0069e+00, 2.4404e+00, ..., 1.9586e+00,
1.8396e+00, 2.5582e+00],
[ 1.0050e+00, -6.5581e-01, -5.5343e-01, ..., 1.3016e+00,
1.5178e+00, 1.5204e+00]],
[[-6.1629e-01, -7.8290e-01, -7.3284e-01, ..., 1.6587e-01,
-4.7881e-01, 9.3285e-01],
[-1.4306e+00, -3.2340e+00, -2.2411e+00, ..., 7.6249e-01,
-9.4976e-01, 1.5535e-02],
[-2.3578e+00, -1.7194e+00, -1.4347e-01, ..., 3.7303e-01,
-1.3613e+00, -1.4963e-01],
...,
[-1.5347e+00, -1.9664e+00, -4.3490e-01, ..., -1.9362e-01,
-4.4592e-01, -3.9811e-01],
[-1.5933e+00, -9.8110e-01, -8.2404e-01, ..., -3.8272e-01,
-3.7392e-01, 4.2349e-01],
[-2.1481e-02, 8.0992e-01, -7.0348e-01, ..., 1.7741e-01,
-1.1738e-03, 5.8943e-01]]],
[[[ 8.1482e-01, 1.4339e+00, 1.3397e+00, ..., 1.3910e+00,
1.0872e+00, 9.5741e-01],
[-9.9293e-01, -2.0097e+00, -1.2266e+00, ..., -5.5501e-01,
-5.6182e-01, -1.6121e+00],
[-1.8797e-01, -7.5110e-01, -3.9321e-02, ..., -1.3178e-01,
7.0497e-01, 8.6953e-01],
...,
[ 7.9337e-02, -1.1124e+00, -1.1480e+00, ..., -7.4247e-02,
-4.5315e-02, 1.8998e-01],
[-6.3786e-01, -1.8829e+00, -1.2528e+00, ..., -5.3977e-01,
-2.9441e-01, -1.2103e-01],
[-3.1233e-01, -1.7294e+00, -1.6958e+00, ..., -1.1853e+00,
-1.5867e+00, -2.7926e-01]],
[[-3.9226e+00, -7.5668e-01, -2.2132e+00, ..., -1.5865e+00,
-2.4912e+00, -1.2305e+00],
[-4.1222e+00, -2.7350e+00, -2.7865e+00, ..., -2.2178e+00,
-3.1157e+00, -9.5082e-01],
[-2.3693e+00, -1.5263e+00, -2.8744e+00, ..., -2.3307e+00,
-2.8261e+00, -1.0348e+00],
...,
[-1.4422e+00, -1.5417e+00, -2.9218e+00, ..., -2.0191e+00,
-2.1566e+00, -6.6158e-01],
[-3.0162e+00, -1.6323e+00, -2.7994e+00, ..., -2.1328e+00,
-2.3283e+00, -1.2948e+00],
[-2.2999e+00, -2.0258e+00, -2.7918e+00, ..., -2.0256e+00,
-2.0089e+00, -1.0736e+00]],
[[ 2.9943e+00, 4.2577e+00, 3.2709e+00, ..., 2.8802e+00,
1.0743e+00, 3.0599e+00],
[ 3.3419e-01, 1.0959e+00, 2.7714e+00, ..., 1.1111e+00,
-8.1362e-01, 2.7862e+00],
[ 1.0016e+00, 1.4341e+00, 1.8288e+00, ..., 7.6553e-01,
-3.0311e-01, 2.1003e+00],
...,
[ 8.8982e-01, 9.6420e-01, 9.6735e-01, ..., 4.3326e-01,
1.2757e+00, 2.2466e+00],
[ 6.0103e-01, 1.7399e+00, 1.2127e+00, ..., -8.3876e-02,
1.0637e+00, 1.8607e+00],
[ 1.6445e+00, 1.6128e+00, 1.8507e+00, ..., 1.2728e+00,
1.8034e+00, 1.8164e+00]],
...,
[[-1.9049e+00, -7.2065e+00, -5.4297e+00, ..., -4.0175e+00,
-3.2216e+00, -5.5384e+00],
[-2.5051e+00, -6.9329e+00, -4.2405e+00, ..., -3.7303e+00,
-2.7356e+00, -4.3433e+00],
[-9.9900e-01, -4.5379e+00, -3.6214e+00, ..., -2.6391e+00,
-3.0083e+00, -2.5146e+00],
...,
[-2.5787e+00, -5.2716e+00, -4.5531e+00, ..., -5.8632e+00,
-5.4766e+00, -5.1151e+00],
[-2.6734e+00, -5.5915e+00, -5.2754e+00, ..., -7.2314e+00,
-5.6385e+00, -6.2867e+00],
[-1.4353e+00, -3.9729e+00, -2.6957e+00, ..., -3.1855e+00,
-2.5739e+00, -3.1772e+00]],
[[-2.4174e-01, 3.3074e-01, 4.1699e-02, ..., 6.4232e-01,
8.6171e-01, 1.3946e+00],
[ 1.6228e+00, 1.1369e+00, 1.0058e+00, ..., 1.7310e+00,
2.5532e+00, 3.0453e+00],
[ 1.2204e+00, 3.2519e-01, -2.1225e-01, ..., 1.4229e+00,
1.5017e+00, 2.3809e+00],
...,
[ 1.2793e+00, 1.9593e-01, -4.5724e-01, ..., 1.5920e+00,
1.1520e+00, 1.6469e+00],
[ 1.8628e+00, 1.6411e+00, 1.0061e+00, ..., 1.5153e+00,
1.3054e+00, 1.9322e+00],
[ 4.2477e-01, 7.2612e-01, 6.7067e-01, ..., 5.0508e-01,
6.6411e-01, 1.6525e+00]],
[[-6.1412e-01, 2.8228e-01, -1.9822e-01, ..., 5.8456e-01,
5.1846e-01, 1.3488e+00],
[-1.4198e+00, -1.7505e+00, -2.0359e+00, ..., -7.9607e-01,
-4.2550e-01, 2.2112e-01],
[-2.2240e+00, -8.3405e-01, -1.1140e+00, ..., 6.1058e-01,
1.0331e-01, 7.6198e-01],
...,
[-1.8420e+00, -1.7961e+00, -1.7048e+00, ..., -6.7193e-01,
-5.9779e-01, -6.6108e-01],
[-1.5623e+00, -2.0565e+00, -1.6693e+00, ..., -5.7842e-01,
-1.2683e+00, -7.1554e-01],
[-8.1113e-01, -8.9572e-01, -1.2071e+00, ..., -6.2724e-01,
-9.1544e-01, -1.2887e-01]]]], device='cuda:0'),
tensor([[[[ 8.3082e-01, 1.4349e+00, 1.1496e+00, ..., 9.2046e-01,
1.0584e+00, 9.6502e-01],
[-1.0023e+00, -1.8470e+00, -1.2056e+00, ..., -6.6825e-01,
-5.7440e-01, -1.6311e+00],
[-2.3892e-01, -7.3136e-01, 5.6519e-02, ..., -3.2758e-01,
8.1137e-01, 8.5816e-01],
...,
[ 2.7858e-02, -1.0914e+00, -9.8826e-01, ..., 4.8034e-01,
-4.6308e-01, 2.1725e-01],
[-6.5355e-01, -1.8967e+00, -1.2745e+00, ..., -3.9455e-01,
-5.5812e-01, -1.6025e-01],
[-3.6803e-01, -1.7563e+00, -1.6792e+00, ..., -1.0236e+00,
-1.4490e+00, -2.8834e-01]],
[[-3.7880e+00, -6.9342e-01, -1.4310e+00, ..., -1.4884e+00,
-2.5348e+00, -1.2399e+00],
[-3.9497e+00, -2.7957e+00, -1.6213e+00, ..., -2.2439e+00,
-3.1834e+00, -9.8590e-01],
[-2.4387e+00, -1.6106e+00, -1.5048e+00, ..., -2.3757e+00,
-2.8626e+00, -1.0885e+00],
...,
[-1.5997e+00, -1.6756e+00, -2.5229e+00, ..., -3.6653e+00,
-2.6472e+00, -7.0171e-01],
[-3.1185e+00, -1.7157e+00, -2.7355e+00, ..., -5.2559e+00,
-2.4690e+00, -1.2040e+00],
[-2.3533e+00, -1.9315e+00, -2.6541e+00, ..., -3.8294e+00,
-1.8273e+00, -9.8354e-01]],
[[ 3.0016e+00, 4.2435e+00, 3.3510e+00, ..., 3.2811e+00,
1.0486e+00, 3.0737e+00],
[ 3.5779e-01, 1.0726e+00, 2.1653e+00, ..., 1.5143e+00,
-8.3974e-01, 2.7760e+00],
[ 1.0545e+00, 1.4428e+00, 1.5828e+00, ..., 4.1207e-01,
-9.4208e-02, 2.0682e+00],
...,
[ 1.0401e+00, 1.1012e+00, 1.1846e+00, ..., -3.4312e-01,
2.8047e-01, 2.3255e+00],
[ 6.8213e-01, 1.8511e+00, 1.1772e+00, ..., -3.3878e-01,
2.8192e-01, 1.8220e+00],
[ 1.6494e+00, 1.6084e+00, 1.8059e+00, ..., 1.2423e+00,
1.2800e+00, 1.7642e+00]],
...,
[[-1.8503e+00, -7.2218e+00, -5.1170e+00, ..., -4.2575e+00,
-3.2635e+00, -5.6649e+00],
[-2.4541e+00, -7.1251e+00, -3.9263e+00, ..., -5.1863e+00,
-2.8288e+00, -4.4439e+00],
[-1.0063e+00, -4.5364e+00, -3.4915e+00, ..., -5.8525e+00,
-3.0587e+00, -2.5693e+00],
...,
[-2.4931e+00, -5.3176e+00, -3.9742e+00, ..., -1.3175e+01,
-7.5721e+00, -4.5915e+00],
[-2.6685e+00, -5.6768e+00, -5.2372e+00, ..., -1.1072e+01,
-7.9389e+00, -6.1590e+00],
[-1.4624e+00, -4.0596e+00, -2.7719e+00, ..., -4.0015e+00,
-4.0547e+00, -3.2472e+00]],
[[-2.5213e-01, 3.2371e-01, 5.8382e-01, ..., 1.1337e+00,
8.4654e-01, 1.4291e+00],
[ 1.5859e+00, 1.0710e+00, 1.8563e+00, ..., 2.6382e+00,
2.5159e+00, 3.0897e+00],
[ 1.1901e+00, 3.1900e-01, 7.3664e-01, ..., 2.6928e+00,
1.6185e+00, 2.3670e+00],
...,
[ 1.2818e+00, 2.2028e-01, -4.6180e-02, ..., 1.3552e+00,
2.5614e+00, 1.6246e+00],
[ 1.8600e+00, 1.6444e+00, 1.1059e+00, ..., 5.9028e-01,
2.4652e+00, 1.9898e+00],
[ 4.4923e-01, 8.0474e-01, 7.0183e-01, ..., 1.3854e-01,
1.2835e+00, 1.6470e+00]],
[[-5.6437e-01, 2.3323e-01, 1.4729e-01, ..., 4.9981e-01,
5.2788e-01, 1.3661e+00],
[-1.3408e+00, -1.8338e+00, -1.5453e+00, ..., -3.4878e-01,
-4.0971e-01, 2.4762e-01],
[-2.2450e+00, -8.2285e-01, 2.0199e-01, ..., 6.6730e-01,
2.1052e-01, 7.9418e-01],
...,
[-1.8895e+00, -1.8935e+00, -1.5135e+00, ..., -1.8807e+00,
-2.2677e+00, -8.6281e-01],
[-1.6108e+00, -2.1301e+00, -1.7609e+00, ..., -3.2600e+00,
-2.8224e+00, -6.9145e-01],
[-8.5361e-01, -8.5725e-01, -1.1836e+00, ..., -1.9661e+00,
-1.5588e+00, -8.3567e-02]]],
[[[ 1.0796e+00, 1.3979e+00, 1.4323e+00, ..., 1.4638e+00,
9.9809e-01, 4.0646e-01],
[-6.3085e-01, -2.4839e+00, -1.2839e+00, ..., -2.2231e-01,
1.5160e-01, -1.3991e+00],
[-4.4891e-01, -1.0973e+00, -7.4502e-01, ..., -3.3119e-01,
2.5660e-01, 6.4153e-01],
...,
[ 4.6507e-01, 4.2554e-01, 1.5801e+00, ..., 4.2287e-01,
1.0615e+00, 7.9125e-01],
[-2.1235e-03, -5.2422e-01, 9.9398e-01, ..., 9.4448e-01,
1.5766e+00, 1.2110e-01],
[ 8.8991e-01, -2.8412e-01, -2.7203e-02, ..., 3.5508e-01,
2.7771e-01, 7.7417e-01]],
[[-2.9233e+00, 6.3309e-03, -1.3168e+00, ..., -2.2406e+00,
-2.5091e+00, -6.3207e-01],
[-4.2263e+00, -2.7821e+00, -2.5186e+00, ..., -4.2548e+00,
-3.8726e+00, -1.5083e+00],
[-3.1320e+00, -2.4012e+00, -2.5925e+00, ..., -3.3409e+00,
-2.7641e+00, -1.4131e+00],
...,
[-9.2153e-01, 4.2617e-02, 1.5913e+00, ..., -5.9356e-01,
6.7930e-02, 8.1724e-01],
[-2.5158e+00, -1.9583e-01, 2.6912e-01, ..., -5.8643e-02,
-5.2148e-02, 6.5749e-01],
[-1.4699e+00, -1.5314e+00, -1.1240e+00, ..., -4.8165e-01,
-5.8422e-01, -1.8106e-01]],
[[ 2.5109e+00, 2.9321e+00, 2.7985e+00, ..., 1.5866e+00,
2.6524e-01, 2.5767e+00],
[ 8.3994e-01, 5.7812e-01, 2.9361e+00, ..., 1.1214e+00,
3.5114e-01, 2.6892e+00],
[ 1.1652e+00, 1.4731e+00, 2.6237e+00, ..., 2.3632e-01,
1.6036e+00, 2.9023e+00],
...,
[ 1.1931e+00, 1.3612e+00, -1.0144e-01, ..., 9.7434e-01,
5.9930e-01, 1.8844e+00],
[ 1.0280e+00, 2.4844e+00, 1.8818e+00, ..., 1.0404e+00,
7.0204e-01, 1.6883e+00],
[ 2.3891e+00, 1.9707e+00, 2.6343e+00, ..., 1.9815e+00,
1.0259e+00, 1.6927e+00]],
...,
[[-2.8109e+00, -7.1518e+00, -5.9266e+00, ..., -6.0221e+00,
-5.1889e+00, -5.3534e+00],
[-3.4380e+00, -7.6538e+00, -5.5629e+00, ..., -7.1783e+00,
-5.5923e+00, -5.2181e+00],
[-2.2046e+00, -6.1413e+00, -4.2132e+00, ..., -5.6854e+00,
-4.6118e+00, -3.4627e+00],
...,
[-2.4643e+00, -4.9193e+00, -2.4986e+00, ..., -3.2096e+00,
-2.8581e+00, -3.0595e+00],
[-2.4552e+00, -5.2532e+00, -3.3596e+00, ..., -4.4631e+00,
-3.2264e+00, -4.9358e+00],
[-2.5739e+00, -5.7539e+00, -2.4332e+00, ..., -2.9511e+00,
-1.6377e+00, -3.3834e+00]],
[[ 3.3427e-01, 1.1620e+00, 3.3139e-01, ..., 5.4050e-01,
1.7160e+00, 1.7237e+00],
[ 1.3219e+00, 8.6926e-01, 1.4660e-01, ..., 6.8114e-01,
1.5553e+00, 2.7777e+00],
[ 1.3457e+00, 7.4773e-01, 6.1147e-01, ..., 1.9119e+00,
1.3938e+00, 2.0281e+00],
...,
[ 1.7144e+00, 6.5829e-01, 2.2331e+00, ..., 1.9080e+00,
2.0098e+00, 1.4765e+00],
[ 2.0132e+00, 1.0091e+00, 1.6529e+00, ..., 1.9300e+00,
1.6762e+00, 2.5551e+00],
[ 1.1820e+00, 7.8242e-01, 1.2898e+00, ..., 1.3996e+00,
1.5191e+00, 1.5056e+00]],
[[-6.2337e-01, 7.8725e-02, -3.9464e-01, ..., 5.5020e-01,
-7.1583e-02, 9.0892e-01],
[-1.5180e+00, -2.1159e+00, -1.8331e+00, ..., -1.9694e+00,
-9.3230e-01, -5.3583e-02],
[-2.3866e+00, -1.1687e+00, -6.0808e-01, ..., -1.4529e+00,
-2.6855e-01, -1.2354e-01],
...,
[-1.8661e+00, -9.2113e-01, -6.8964e-01, ..., -1.9442e-01,
-4.0990e-01, -4.3854e-01],
[-1.5533e+00, -1.1832e+00, -2.8637e-01, ..., -3.7965e-01,
-3.4852e-01, 4.4344e-01],
[-1.1537e-01, 3.9353e-01, -1.3371e-01, ..., 1.8491e-01,
3.5441e-02, 6.4237e-01]]],
[[[ 6.9120e-01, 1.0354e+00, 7.1201e-01, ..., 3.7507e-01,
3.5934e-01, 1.9933e-01],
[-3.7074e-01, -1.5166e+00, -6.2303e-01, ..., 3.7272e-01,
3.5847e-01, -1.1504e+00],
[ 7.4679e-02, -7.1107e-01, 8.6376e-02, ..., -4.1943e-01,
1.0619e+00, 6.5897e-01],
...,
[-8.9910e-03, -1.5640e+00, -6.9970e-01, ..., -5.9191e-01,
6.9458e-01, 4.9836e-01],
[-4.4372e-01, -1.9407e+00, -1.1531e+00, ..., -2.1753e-01,
1.1473e+00, -1.4198e-01],
[ 3.1489e-01, -5.7171e-01, -7.5726e-01, ..., -2.7708e-01,
3.7096e-01, 6.0930e-01]],
[[-3.1594e+00, -1.5581e+00, -3.0403e+00, ..., -1.7283e+00,
-2.1869e+00, -1.0257e+00],
[-4.4734e+00, -1.8707e+00, -2.2855e+00, ..., -6.2830e-01,
-1.2987e+00, -2.5620e-01],
[-4.3860e+00, -2.7759e+00, -3.7913e+00, ..., -1.5722e+00,
-1.4205e+00, -7.1383e-01],
...,
[-2.8648e+00, -2.2414e+00, -3.7625e+00, ..., -1.6777e+00,
3.7828e-01, 4.8901e-01],
[-3.5829e+00, -1.5015e+00, -2.6451e+00, ..., -4.3496e-01,
-3.0833e-01, 2.5717e-01],
[-1.3360e+00, -8.6914e-01, -1.6412e+00, ..., -4.2266e-01,
-7.1018e-01, -5.6771e-01]],
[[ 1.7963e+00, 2.8351e+00, 2.3640e+00, ..., 2.7658e+00,
1.4150e+00, 2.8513e+00],
[ 2.2621e-01, 1.1783e+00, 1.4288e+00, ..., 1.8051e+00,
9.2462e-03, 2.4917e+00],
[ 7.7046e-01, 1.4909e+00, 8.6943e-01, ..., 9.4387e-01,
6.4902e-01, 1.8244e+00],
...,
[ 9.7898e-01, 1.3710e+00, 2.6486e+00, ..., 7.1542e-01,
1.8571e-01, 1.4782e+00],
[ 6.5328e-01, 1.7554e+00, 1.1219e+00, ..., 9.5909e-01,
5.5545e-01, 1.5570e+00],
[ 1.5452e+00, 1.5483e+00, 1.6880e+00, ..., 1.5014e+00,
1.1296e+00, 2.0404e+00]],
...,
[[-1.4869e+00, -4.6977e+00, -3.9542e+00, ..., -3.1993e+00,
-2.5150e+00, -3.9629e+00],
[-1.8514e+00, -3.9114e+00, -4.2206e+00, ..., -5.2540e+00,
-3.9073e+00, -4.4026e+00],
[-9.0976e-01, -3.0568e+00, -3.8718e+00, ..., -2.4287e+00,
-2.3595e+00, -1.6411e+00],
...,
[-1.5756e+00, -4.6226e+00, -4.5704e+00, ..., -4.5045e+00,
-3.7628e+00, -1.5839e+00],
[-1.4802e+00, -4.6379e+00, -5.4679e+00, ..., -7.0511e+00,
-4.4842e+00, -3.7562e+00],
[-1.4809e+00, -4.6617e+00, -4.0792e+00, ..., -3.5191e+00,
-2.2008e+00, -2.7984e+00]],
[[ 7.2854e-01, 4.7936e-01, -6.4853e-02, ..., 8.8980e-01,
1.1094e+00, 1.2750e+00],
[ 1.1745e+00, 9.7275e-01, 5.7723e-01, ..., 1.4613e+00,
2.3504e+00, 2.5790e+00],
[ 5.8333e-01, 5.6444e-01, 1.1148e-01, ..., 8.4702e-01,
1.6352e+00, 1.9372e+00],
...,
[ 8.0114e-01, -3.9612e-01, -9.4356e-01, ..., 1.5594e+00,
1.9779e+00, 1.6951e+00],
[ 1.5440e+00, 1.3738e+00, 9.4345e-01, ..., 1.6537e+00,
1.4281e+00, 2.4543e+00],
[ 5.9813e-01, 9.4525e-01, 1.0529e+00, ..., 1.0878e+00,
1.3483e+00, 1.3680e+00]],
[[-2.8204e-01, -2.2426e-01, -9.5620e-01, ..., -1.3139e-01,
-1.1103e-01, 8.6996e-01],
[-1.4035e+00, -1.4505e+00, -2.0094e+00, ..., 1.8322e-01,
-6.7013e-02, 3.5804e-01],
[-1.6763e+00, -8.3310e-01, -7.1291e-01, ..., 1.7080e-01,
2.1974e-01, 2.3534e-01],
...,
[-1.6861e+00, -1.5345e+00, -1.0587e+00, ..., -1.0493e+00,
4.3768e-02, -3.3119e-01],
[-1.4384e+00, -1.6172e+00, -1.4862e+00, ..., 3.8610e-03,
-1.6897e-01, 2.8690e-01],
[-2.3279e-01, 6.2288e-02, -4.5649e-01, ..., 1.7768e-01,
1.4990e-01, 3.2172e-01]]],
[[[ 8.7184e-01, 1.5246e+00, 1.1352e+00, ..., -5.4969e-01,
2.8057e-01, 3.6184e-01],
[-8.4055e-01, -1.6628e+00, -6.4049e-01, ..., 5.8484e-01,
-3.5673e-01, -2.1046e-01],
[ 4.1488e-01, -1.1976e-01, 6.3043e-02, ..., 6.2716e-01,
1.8422e+00, 8.9631e-01],
...,
[ 4.9514e-01, -7.0392e-01, 7.0604e-01, ..., 1.2443e-01,
-1.0585e-01, 3.4327e-01],
[-4.5063e-02, -1.2400e+00, 5.7201e-02, ..., -6.4152e-03,
4.8746e-02, -1.2865e-01],
[ 8.4555e-01, -3.6341e-01, -1.8336e-01, ..., -3.3955e-01,
-9.4473e-01, 1.7152e-01]],
[[-4.0081e+00, -1.2661e+00, -1.8692e+00, ..., -2.7565e+00,
-3.0429e+00, -1.9607e+00],
[-4.1392e+00, -3.2049e+00, -3.7978e+00, ..., -3.3372e+00,
-3.3927e+00, -2.2757e+00],
[-2.7054e+00, -2.9636e+00, -5.9546e+00, ..., -1.4190e+00,
-2.6658e+00, -1.9641e+00],
...,
[-8.5028e-01, -4.8695e-01, -1.5857e+00, ..., -4.9551e-01,
-9.8168e-01, -8.4597e-01],
[-3.2306e+00, -1.4010e-01, -3.6271e-01, ..., -6.3243e-01,
-1.1263e+00, -1.2327e+00],
[-2.3485e+00, -1.5752e+00, -1.2501e+00, ..., -6.5575e-01,
-7.4214e-01, -5.1515e-01]],
[[ 2.7408e+00, 3.7627e+00, 2.8839e+00, ..., 2.8828e+00,
1.4613e+00, 2.1320e+00],
[ 2.4707e-02, 5.8309e-01, 1.5485e+00, ..., -1.5326e+00,
3.2784e-02, 2.0842e+00],
[ 8.4352e-01, 1.7028e+00, 3.5112e+00, ..., -4.2140e+00,
-1.0974e+00, 1.2656e+00],
...,
[ 5.2484e-01, 1.2040e+00, 1.9757e+00, ..., -3.7050e-01,
6.6605e-01, 1.5555e+00],
[ 6.1073e-01, 2.3154e+00, 2.0295e+00, ..., -1.3196e-01,
8.8739e-01, 1.2001e+00],
[ 1.6815e+00, 1.3182e+00, 1.7788e+00, ..., 5.8352e-01,
1.3353e+00, 1.3693e+00]],
...,
[[-1.5958e+00, -6.7090e+00, -5.5921e+00, ..., -5.5469e+00,
-3.6754e+00, -4.5671e+00],
[-2.2756e+00, -5.8448e+00, -4.9034e+00, ..., -6.4706e+00,
-4.4778e+00, -4.4544e+00],
[-2.9528e-01, -2.9049e+00, -3.2163e+00, ..., -7.9257e+00,
-5.9929e+00, -4.0381e+00],
...,
[-1.1890e+00, -2.3941e+00, -1.3350e+00, ..., -5.9603e+00,
-3.6249e+00, -2.7668e+00],
[-1.5957e+00, -4.1929e+00, -3.3186e+00, ..., -5.2140e+00,
-4.3570e+00, -4.2918e+00],
[-2.5800e+00, -5.4446e+00, -2.9819e+00, ..., -3.4343e+00,
-2.5912e+00, -2.4341e+00]],
[[-6.1075e-01, -9.6214e-02, 2.9245e-01, ..., 1.4549e+00,
2.2107e+00, 1.2261e+00],
[ 1.6603e+00, 6.7550e-01, 9.0759e-01, ..., -2.0858e-01,
1.8146e+00, 1.9280e+00],
[ 7.9511e-01, -2.9621e-01, -2.4624e+00, ..., -9.6065e-01,
-3.0364e-01, 1.6604e+00],
...,
[ 1.3349e+00, -5.1535e-01, -5.3248e-01, ..., 2.0940e+00,
7.8226e-01, 9.9771e-01],
[ 2.0326e+00, 1.2860e+00, 2.3143e+00, ..., 1.3173e+00,
7.0189e-01, 1.5086e+00],
[ 9.9144e-01, 5.4004e-01, 1.1518e+00, ..., 7.7168e-01,
6.7218e-01, 1.4406e+00]],
[[-5.1370e-01, 5.9713e-01, 1.7807e-01, ..., -5.5362e-01,
-1.0035e+00, 4.0870e-02],
[-1.1826e+00, -1.2247e+00, -1.6510e+00, ..., 1.0128e-01,
-2.0720e+00, -1.7918e+00],
[-2.2293e+00, -7.5739e-01, -2.8596e-01, ..., 2.0995e+00,
3.5996e-01, -1.4331e+00],
...,
[-1.7636e+00, -1.6811e+00, -2.9829e-01, ..., -2.3923e-02,
-1.4232e+00, -8.2853e-01],
[-1.9243e+00, -1.0241e+00, -4.7067e-01, ..., -7.5780e-01,
-1.6675e+00, -6.8173e-01],
[-4.9831e-01, 2.7676e-01, -2.6829e-01, ..., -2.3878e-01,
-8.0761e-01, -2.8749e-03]]],
[[[ 5.6852e-01, 8.5605e-01, 5.5570e-01, ..., 3.6474e-01,
4.3058e-01, 1.7065e-02],
[-2.6615e-01, -8.9365e-01, -5.3730e-01, ..., 1.3547e-01,
2.1084e-01, -2.8946e-01],
[-1.9061e-01, 9.0055e-03, 5.7203e-01, ..., 5.0371e-02,
2.7343e-01, 7.2114e-01],
...,
[ 6.7762e-02, -8.5807e-01, 1.1999e+00, ..., 5.8717e-01,
1.1037e+00, 8.9908e-01],
[-3.8115e-01, -2.1675e+00, -8.1607e-01, ..., -3.5732e-01,
8.6924e-01, 1.8278e-01],
[ 4.2532e-01, -1.1047e+00, -1.7183e+00, ..., 8.3937e-01,
-4.3352e-01, 1.8444e-01]],
[[-2.4484e+00, -6.8106e-01, -1.9175e+00, ..., -1.6813e+00,
-2.2854e+00, -1.4921e+00],
[-2.5110e+00, -7.1474e-01, -1.5085e+00, ..., -1.2217e+00,
-2.3342e+00, -1.5760e+00],
[-2.4598e+00, -1.4684e+00, -2.1597e+00, ..., -1.7656e+00,
-2.0659e+00, -1.1138e+00],
...,
[-1.0447e+00, -5.6337e-01, -1.3061e+00, ..., -1.8543e+00,
-1.2558e+00, 9.0772e-02],
[-2.2234e+00, -2.9877e-01, -2.0133e+00, ..., -2.1437e+00,
-1.7569e+00, -2.5720e-01],
[-1.0552e+00, -1.5370e+00, -2.3254e+00, ..., -3.0024e+00,
-1.7724e+00, -6.1504e-01]],
[[ 1.9154e+00, 2.5655e+00, 1.6432e+00, ..., 1.5458e+00,
7.6017e-01, 2.0532e+00],
[-2.3792e-01, 3.8336e-01, 6.3546e-01, ..., 8.0708e-01,
2.9835e-01, 2.1839e+00],
[-2.6642e-02, 3.1228e-01, 3.9528e-01, ..., -1.1472e-01,
7.3758e-01, 1.8797e+00],
...,
[ 1.1198e+00, 1.4895e+00, 1.4102e+00, ..., 9.3009e-02,
-6.8169e-01, 1.3240e+00],
[ 8.3705e-01, 2.3125e+00, 2.5168e+00, ..., -7.0535e-01,
-7.0558e-01, 9.6155e-01],
[ 1.8186e+00, 1.0570e+00, 2.3996e+00, ..., 1.6845e-01,
-1.1035e-01, 1.2496e+00]],
...,
[[-2.0074e+00, -4.9091e+00, -3.2025e+00, ..., -2.0366e+00,
-1.6745e+00, -2.9286e+00],
[-2.4609e+00, -5.3096e+00, -3.0009e+00, ..., -2.3174e+00,
-2.1963e+00, -2.8241e+00],
[-1.5507e+00, -5.3860e+00, -3.0640e+00, ..., -2.1387e+00,
-2.2092e+00, -1.9192e+00],
...,
[-2.0696e+00, -5.4898e+00, -3.1412e+00, ..., -2.5816e+00,
-2.3433e+00, -3.0238e+00],
[-2.8769e+00, -7.3441e+00, -3.8288e+00, ..., -2.9499e+00,
-3.3958e+00, -4.4939e+00],
[-2.9891e+00, -7.3021e+00, -3.1656e+00, ..., -4.7730e+00,
-2.0685e+00, -2.7489e+00]],
[[ 9.0433e-01, 1.6035e+00, 8.0486e-01, ..., 1.0570e+00,
1.4558e+00, 1.3950e+00],
[ 1.1394e+00, 9.7070e-01, 9.0486e-01, ..., 9.7145e-01,
8.4975e-01, 1.9109e+00],
[ 6.5859e-01, 2.0478e-01, 6.6309e-02, ..., 4.9267e-01,
1.4117e-01, 1.2350e+00],
...,
[ 1.6462e+00, 9.6029e-02, 8.9297e-01, ..., -6.0877e-01,
1.8704e+00, 1.8612e+00],
[ 2.2734e+00, 1.4106e+00, 1.1530e+00, ..., -1.1252e+00,
1.5082e+00, 2.2624e+00],
[ 1.0206e+00, 3.8421e-01, 4.8440e-01, ..., -1.1787e+00,
1.2410e+00, 1.1905e+00]],
[[-9.4534e-02, -6.3714e-01, -1.2098e+00, ..., -5.4213e-01,
-9.0185e-01, -1.2215e-01],
[-8.8226e-01, -2.6153e+00, -2.6008e+00, ..., -1.1367e+00,
-1.0719e+00, -5.8897e-01],
[-1.4465e+00, -1.5284e+00, -9.9208e-01, ..., -4.6496e-02,
-2.7503e-01, -2.3527e-01],
...,
[-2.6111e+00, -1.6894e+00, -6.9343e-01, ..., -4.0806e-01,
-9.8113e-01, 9.6951e-02],
[-2.5019e+00, -2.0886e+00, -9.8376e-01, ..., -3.1883e-01,
-1.3375e+00, 2.8227e-01],
[-4.4672e-01, -3.5769e-01, -6.7419e-01, ..., 2.1582e-02,
-1.4969e+00, -9.1102e-03]]]], device='cuda:0'),
tensor([[[[ 9.4269e-01, 1.5572e+00, 1.0325e+00, ..., 5.0319e-01,
3.2140e-01, -1.5724e-01],
[-8.0933e-01, -1.6116e+00, -6.4528e-01, ..., 2.7586e-01,
6.5547e-01, -5.8384e-01],
[ 3.7669e-02, -5.4897e-01, -2.8984e-01, ..., -7.0538e-01,
-4.6800e-01, 3.0257e-01],
...,
[ 5.6643e-01, -4.5650e-01, -6.0337e-01, ..., 1.0494e+00,
1.1507e+00, 3.2548e-01],
[ 1.2016e-02, -1.0720e+00, -2.2700e-01, ..., -5.2855e-02,
9.2598e-01, 2.3869e-01],
[ 9.1434e-01, -3.2364e-01, -3.0758e-01, ..., -4.5151e-01,
-3.9333e-01, 2.8414e-01]],
[[-3.8426e+00, -1.2900e+00, -1.9792e+00, ..., -2.9175e+00,
-3.4759e+00, -8.4431e-01],
[-3.9354e+00, -3.3118e+00, -3.2642e+00, ..., -2.0663e+00,
-3.0966e+00, -5.3979e-01],
[-2.4361e+00, -2.4135e+00, -4.4780e+00, ..., -4.3030e+00,
-3.4442e+00, -4.2083e-01],
...,
[-7.0999e-01, -3.7081e-01, -1.8762e+00, ..., -2.8374e+00,
-1.9957e+00, -1.7913e-01],
[-3.0858e+00, 1.2065e-01, -8.6547e-01, ..., -1.6947e+00,
-1.7185e+00, -1.2914e-01],
[-2.1043e+00, -1.3752e+00, -1.6725e+00, ..., -8.8938e-01,
-9.0979e-01, 6.3011e-02]],
[[ 2.7146e+00, 3.8038e+00, 3.1185e+00, ..., 2.2575e+00,
7.5798e-01, 2.3273e+00],
[ 7.5064e-02, 8.3707e-01, 2.1423e+00, ..., 1.8915e+00,
1.1704e-01, 2.3486e+00],
[ 6.2200e-01, 1.9184e+00, 2.5528e+00, ..., 1.1486e+00,
1.0523e+00, 2.5957e+00],
...,
[ 5.6547e-01, 8.3596e-01, 1.3566e+00, ..., 8.1057e-01,
7.8577e-02, 1.4999e+00],
[ 6.7621e-01, 2.3609e+00, 1.7885e+00, ..., 1.0849e-01,
3.2983e-01, 1.2681e+00],
[ 1.7039e+00, 1.3742e+00, 1.8471e+00, ..., 1.0365e+00,
7.9409e-01, 1.2794e+00]],
...,
[[-1.6465e+00, -6.8308e+00, -5.4804e+00, ..., -4.1202e+00,
-3.1136e+00, -3.7649e+00],
[-2.2240e+00, -6.1209e+00, -4.9127e+00, ..., -5.7048e+00,
-4.8419e+00, -4.5184e+00],
[-1.7559e-01, -2.7165e+00, -3.4651e+00, ..., -5.7602e+00,
-4.3812e+00, -2.7569e+00],
...,
[-1.0946e+00, -2.0793e+00, -1.5974e+00, ..., -6.3501e+00,
-3.8580e+00, -2.9168e+00],
[-1.5568e+00, -3.7865e+00, -3.4190e+00, ..., -6.5011e+00,
-4.7671e+00, -4.7583e+00],
[-2.5830e+00, -5.4164e+00, -2.8126e+00, ..., -4.4734e+00,
-2.8480e+00, -3.2125e+00]],
[[-6.0361e-01, -9.1159e-02, 1.4411e-01, ..., 9.5207e-01,
1.0573e+00, 1.7763e+00],
[ 1.6314e+00, 5.8082e-01, 1.3682e+00, ..., 1.5173e+00,
1.5453e+00, 2.8750e+00],
[ 1.1082e+00, -5.8781e-01, -6.9523e-01, ..., 1.0080e+00,
9.2640e-01, 2.3402e+00],
...,
[ 1.3295e+00, -1.6073e-01, 5.0903e-01, ..., 1.5919e+00,
1.3558e+00, 1.5999e+00],
[ 2.0534e+00, 1.3982e+00, 1.6819e+00, ..., 1.7837e+00,
1.1609e+00, 2.3664e+00],
[ 1.0594e+00, 6.1430e-01, 1.0329e+00, ..., 1.0355e+00,
1.1784e+00, 1.5459e+00]],
[[-4.5203e-01, 5.4001e-01, 1.1273e-01, ..., -2.6360e-01,
-5.2484e-01, 9.0337e-01],
[-1.0399e+00, -1.3728e+00, -1.4465e+00, ..., -2.5072e-01,
-5.2519e-02, 1.0972e+00],
[-1.8963e+00, -1.0309e+00, -5.5830e-01, ..., 1.6670e-01,
1.8257e-01, 1.0939e-01],
...,
[-1.7882e+00, -1.9554e+00, -1.6172e+00, ..., -7.3195e-01,
-6.6745e-01, -6.7915e-01],
[-1.8941e+00, -1.0914e+00, -1.0468e+00, ..., -7.8993e-01,
-1.1570e+00, 9.9212e-02],
[-4.1941e-01, 2.4048e-01, -4.5548e-01, ..., 1.1172e-01,
-2.0985e-01, 6.2061e-01]]],
[[[ 9.6108e-01, 1.6191e+00, 1.2063e+00, ..., 6.9332e-01,
9.8322e-01, 4.7947e-01],
[-8.4100e-01, -1.9865e+00, -5.4807e-01, ..., -1.7843e-01,
-2.4605e-01, -4.0208e-01],
[-8.5325e-02, -6.5584e-01, -1.2839e-01, ..., -6.9692e-01,
-1.4384e-01, 6.0692e-01],
...,
[ 4.4750e-01, -2.9211e-01, -1.7595e-01, ..., 3.6430e-01,
-5.0401e-02, 4.5648e-01],
[-1.7254e-01, -1.0492e+00, -1.5756e-01, ..., 2.9726e-02,
3.0818e-01, -1.6767e-02],
[ 7.6038e-01, -2.9100e-01, -2.6410e-01, ..., -3.5104e-01,
-8.1551e-01, 2.2747e-01]],
[[-4.1680e+00, -1.3657e+00, -2.2262e+00, ..., -1.4248e+00,
-2.1463e+00, -1.8651e+00],
[-4.1693e+00, -3.1857e+00, -1.6701e+00, ..., -1.1276e+00,
-2.3046e+00, -2.1789e+00],
[-2.9074e+00, -2.3264e+00, -1.2154e+00, ..., -1.2678e+00,
-1.7953e+00, -1.4177e+00],
...,
[-8.1339e-01, -6.6602e-01, -1.3166e+00, ..., -6.6746e-01,
-6.0071e-01, -6.9491e-01],
[-2.8811e+00, -4.3215e-01, -8.4830e-01, ..., -6.5975e-01,
-9.4325e-01, -1.1513e+00],
[-2.0917e+00, -1.5204e+00, -1.5127e+00, ..., -6.6360e-01,
-6.9016e-01, -5.1176e-01]],
[[ 3.0694e+00, 4.1440e+00, 3.2960e+00, ..., 1.7626e+00,
8.7822e-01, 2.5757e+00],
[ 6.3101e-01, 9.3899e-01, 2.3660e+00, ..., 1.2933e+00,
7.2028e-01, 2.6673e+00],
[ 1.4813e+00, 1.6171e+00, 1.0383e+00, ..., 5.4306e-01,
9.2862e-01, 1.7836e+00],
...,
[ 3.0375e-01, 9.4841e-01, 1.1424e+00, ..., -4.2428e-02,
3.4789e-01, 1.2589e+00],
[ 3.9890e-01, 2.1068e+00, 1.5044e+00, ..., -2.1824e-01,
6.9981e-01, 1.1029e+00],
[ 1.4775e+00, 1.3612e+00, 1.7061e+00, ..., 5.7158e-01,
1.3256e+00, 1.3816e+00]],
...,
[[-2.1034e+00, -7.5106e+00, -6.0230e+00, ..., -4.9600e+00,
-3.6175e+00, -4.4003e+00],
[-2.8020e+00, -6.9538e+00, -4.6384e+00, ..., -5.1619e+00,
-3.9160e+00, -3.2642e+00],
[-3.4091e-01, -3.4786e+00, -3.0044e+00, ..., -3.5096e+00,
-2.7605e+00, -1.4214e+00],
...,
[-9.4394e-01, -2.2658e+00, -1.7438e+00, ..., -4.2947e+00,
-3.2164e+00, -2.4574e+00],
[-1.3351e+00, -3.7529e+00, -3.4260e+00, ..., -5.1142e+00,
-4.0531e+00, -4.1938e+00],
[-2.2623e+00, -5.0331e+00, -2.9671e+00, ..., -3.4099e+00,
-2.4645e+00, -2.3695e+00]],
[[-7.8075e-01, -1.7289e-01, 1.1746e-01, ..., 1.5470e+00,
1.7062e+00, 1.1690e+00],
[ 1.6573e+00, 1.0976e+00, 1.2767e+00, ..., 1.4508e+00,
1.2311e+00, 1.8852e+00],
[ 9.0715e-01, 4.3423e-01, 9.3894e-01, ..., 1.4818e+00,
3.9016e-01, 1.2835e+00],
...,
[ 1.2971e+00, -1.5885e-01, 8.5752e-01, ..., 1.5632e+00,
7.0976e-01, 8.4460e-01],
[ 1.9058e+00, 9.9640e-01, 1.7473e+00, ..., 1.2592e+00,
6.4678e-01, 1.3576e+00],
[ 1.0280e+00, 4.0583e-01, 9.3840e-01, ..., 7.5191e-01,
6.5412e-01, 1.4329e+00]],
[[-5.6846e-01, 5.7874e-01, 9.6578e-02, ..., -3.5261e-01,
-6.1765e-01, -5.4321e-02],
[-1.3202e+00, -1.2635e+00, -6.1172e-01, ..., -1.0801e+00,
-1.5365e+00, -1.1482e+00],
[-2.3778e+00, -7.1132e-01, 1.1679e+00, ..., -7.2216e-01,
-1.2624e+00, -1.6898e-01],
...,
[-1.7670e+00, -1.5388e+00, -1.0182e+00, ..., -4.1034e-01,
-1.1153e+00, -8.4122e-01],
[-1.7751e+00, -1.1495e+00, -8.5261e-01, ..., -8.3492e-01,
-1.5959e+00, -7.2068e-01],
[-3.8181e-01, 3.6934e-01, -3.2630e-01, ..., -2.9271e-01,
-8.0297e-01, 1.9026e-02]]],
[[[ 8.6036e-01, 1.5381e+00, 1.2377e+00, ..., 9.2823e-01,
8.9965e-01, 9.4156e-01],
[-1.0019e+00, -2.0141e+00, -1.1752e+00, ..., -4.4044e-01,
-7.7140e-01, -1.6720e+00],
[-1.2294e-01, -6.6214e-01, 2.2764e-02, ..., 1.0920e+00,
4.4412e-01, 6.2007e-01],
...,
[ 4.6324e-01, -2.9841e-01, 4.8425e-02, ..., 6.6866e-01,
1.8860e-01, 6.4941e-01],
[ 3.6425e-01, -9.5069e-01, -8.5395e-01, ..., 8.3121e-01,
5.5844e-01, -1.1343e-01],
[ 5.6946e-01, -1.0386e-02, -7.2425e-01, ..., 3.9226e-01,
-8.2269e-02, 6.3844e-01]],
[[-3.6642e+00, -7.3605e-01, -1.5502e+00, ..., -1.4292e+00,
-2.4992e+00, -1.2596e+00],
[-3.9468e+00, -2.8558e+00, -1.8705e+00, ..., -2.1015e+00,
-3.1837e+00, -9.2310e-01],
[-2.4518e+00, -1.6738e+00, -1.6101e+00, ..., -2.5701e+00,
-2.6340e+00, -1.0005e+00],
...,
[-1.0222e+00, -1.3969e+00, -4.3109e+00, ..., -1.3247e+00,
-1.0804e+00, -1.2778e-01],
[-1.2954e+00, -7.4587e-01, -2.9900e+00, ..., -4.6012e-01,
-1.0575e+00, -7.2166e-02],
[ 4.6388e-01, -3.6935e-01, -1.2706e+00, ..., 1.7986e-01,
1.2169e-01, 4.6023e-01]],
[[ 2.8727e+00, 4.2496e+00, 3.3445e+00, ..., 3.2263e+00,
9.6426e-01, 2.9807e+00],
[ 1.8058e-01, 1.1094e+00, 2.2983e+00, ..., 1.7650e+00,
-6.7323e-01, 2.7475e+00],
[ 9.8265e-01, 1.6271e+00, 1.7730e+00, ..., 5.4294e-01,
7.4877e-01, 2.3659e+00],
...,
[ 4.9605e-01, 6.8929e-01, 2.7853e+00, ..., -5.1105e-01,
8.7895e-02, 1.3610e+00],
[ 2.0764e-01, 1.5091e+00, 2.6501e+00, ..., -2.7586e-01,
-3.2707e-02, 1.0562e+00],
[ 1.3627e+00, 1.4505e+00, 1.6678e+00, ..., 1.0119e-01,
3.7062e-01, 1.1388e+00]],
...,
[[-1.8273e+00, -7.2477e+00, -5.4721e+00, ..., -3.6875e+00,
-3.0174e+00, -5.4733e+00],
[-2.3775e+00, -6.9385e+00, -4.2229e+00, ..., -3.4212e+00,
-2.3311e+00, -4.2581e+00],
[-1.0983e+00, -4.7639e+00, -3.9186e+00, ..., -2.6976e+00,
-3.1335e+00, -2.4742e+00],
...,
[-2.2848e+00, -5.7218e+00, -4.5734e+00, ..., -3.1393e+00,
-3.4113e+00, -3.3390e+00],
[-2.4428e+00, -6.9193e+00, -5.8870e+00, ..., -4.8087e+00,
-4.2132e+00, -4.6113e+00],
[-2.3523e+00, -6.6376e+00, -4.9554e+00, ..., -4.6151e+00,
-3.1665e+00, -4.0775e+00]],
[[-1.5405e-01, 3.0266e-01, 5.5467e-01, ..., 8.7561e-01,
8.4722e-01, 1.3813e+00],
[ 1.6213e+00, 1.0580e+00, 1.8077e+00, ..., 1.9061e+00,
2.6130e+00, 3.0304e+00],
[ 1.1651e+00, 2.9438e-01, 8.3358e-01, ..., 2.8232e-01,
2.1162e+00, 2.6073e+00],
...,
[ 1.0963e+00, -3.3036e-01, -2.2160e+00, ..., 1.3755e+00,
8.0119e-01, 1.1348e+00],
[ 1.6884e+00, 5.0557e-01, -1.2896e+00, ..., 1.6281e+00,
5.8640e-01, 2.0091e+00],
[ 1.1713e+00, 7.9881e-01, 4.3989e-01, ..., 1.4397e+00,
1.2406e+00, 1.3456e+00]],
[[-5.3546e-01, 3.0791e-01, 1.4824e-01, ..., 2.8014e-01,
3.7482e-01, 1.3457e+00],
[-1.3585e+00, -1.8468e+00, -1.6509e+00, ..., -6.5273e-01,
-5.5053e-01, 2.2307e-01],
[-2.1698e+00, -7.4457e-01, 3.5127e-01, ..., 1.0295e+00,
-7.9072e-02, 7.1237e-01],
...,
[-1.5592e+00, -1.9716e+00, -2.2915e+00, ..., -9.7843e-01,
-9.6912e-01, -6.7599e-01],
[-1.2595e+00, -1.9313e+00, -1.7044e+00, ..., -7.7744e-01,
-1.2987e+00, -5.7205e-02],
[ 1.0828e-01, 4.2705e-02, -3.9718e-01, ..., -5.7565e-02,
-6.5123e-02, 7.8694e-01]]],
[[[ 9.6838e-01, 1.6326e+00, 1.0973e+00, ..., 4.9225e-01,
2.6500e-01, -2.0259e-01],
[-8.1411e-01, -2.0326e+00, -9.2696e-01, ..., 2.7254e-01,
7.5269e-01, -5.1318e-01],
[ 6.4587e-02, -7.7570e-01, -1.3094e-01, ..., -8.3158e-01,
-4.3749e-02, 4.9149e-01],
...,
[ 4.0299e-01, -5.6223e-01, 1.1466e+00, ..., -3.3848e+00,
-2.6520e-01, -6.1729e-01],
[-2.2780e-01, -1.2420e+00, -3.4570e-01, ..., -1.7671e+00,
-8.1226e-01, -3.6661e+00],
[ 7.2300e-01, -3.5634e-01, -4.5862e-01, ..., 1.5614e-01,
2.6032e+00, -1.1300e-02]],
[[-4.1848e+00, -1.2059e+00, -2.2361e+00, ..., -2.8166e+00,
-3.4354e+00, -7.6839e-01],
[-4.0770e+00, -2.8573e+00, -2.6316e+00, ..., -2.1663e+00,
-3.1093e+00, -4.1358e-01],
[-2.6198e+00, -1.9211e+00, -4.2396e+00, ..., -4.3522e+00,
-3.2710e+00, -4.7086e-01],
...,
[-9.0869e-01, -1.5346e+00, -3.2402e+00, ..., -2.4842e+00,
-2.5841e+00, -1.2809e+00],
[-2.9632e+00, -9.2745e-01, -2.0340e+00, ..., 3.4015e-01,
-3.2691e+00, -1.2117e+00],
[-2.0773e+00, -1.5523e+00, -2.0267e+00, ..., 1.2047e+00,
-2.7181e+00, -1.9815e+00]],
[[ 2.9977e+00, 4.2057e+00, 3.4164e+00, ..., 2.2783e+00,
7.5749e-01, 2.2480e+00],
[ 2.9262e-01, 9.0013e-01, 2.4498e+00, ..., 1.8531e+00,
8.5241e-02, 2.2618e+00],
[ 7.4130e-01, 1.3185e+00, 1.9759e+00, ..., 7.2280e-01,
8.5461e-01, 2.1452e+00],
...,
[ 1.0058e-01, 1.3612e+00, 2.2694e+00, ..., -3.2384e+00,
9.2014e-01, 8.9518e-02],
[ 2.3122e-01, 2.0309e+00, 2.5863e+00, ..., -5.5517e-01,
-1.1566e+00, -2.3501e+00],
[ 1.3508e+00, 1.3787e+00, 1.8445e+00, ..., -6.4658e-01,
-1.1459e+00, -2.5927e+00]],
...,
[[-2.0277e+00, -7.3148e+00, -5.8673e+00, ..., -3.9707e+00,
-2.9961e+00, -3.5699e+00],
[-2.7071e+00, -6.4449e+00, -4.6065e+00, ..., -5.7386e+00,
-4.7840e+00, -4.2453e+00],
[-2.0444e-01, -2.7114e+00, -3.1703e+00, ..., -5.6040e+00,
-3.8926e+00, -1.8878e+00],
...,
[-8.9490e-01, -2.5161e+00, -3.0025e+00, ..., -6.2295e+00,
-3.0461e+00, 1.3127e+00],
[-1.2322e+00, -3.8193e+00, -3.4806e+00, ..., -4.2077e+00,
-2.5001e+00, 1.1982e+00],
[-2.1530e+00, -4.9327e+00, -2.9857e+00, ..., -3.9414e+00,
-5.0401e+00, -3.3634e+00]],
[[-6.9678e-01, -9.1482e-02, -2.7483e-02, ..., 1.0605e+00,
1.0299e+00, 1.7903e+00],
[ 1.8007e+00, 1.1015e+00, 1.0785e+00, ..., 1.5996e+00,
1.5284e+00, 2.8070e+00],
[ 1.1952e+00, 2.0726e-01, -1.1820e+00, ..., 1.0811e+00,
8.4496e-01, 1.9284e+00],
...,
[ 1.2614e+00, -8.4534e-01, -3.8516e-01, ..., -5.2413e-02,
-2.3688e+00, -1.0105e+00],
[ 1.8436e+00, 7.6383e-01, 1.6359e+00, ..., -3.0950e+00,
-3.3266e+00, 1.0821e+00],
[ 1.0311e+00, 3.6782e-01, 9.0190e-01, ..., 7.0799e-01,
-7.1393e-01, 8.1869e-01]],
[[-4.6883e-01, 6.5441e-01, 1.8409e-02, ..., -1.9000e-01,
-5.5161e-01, 8.8415e-01],
[-1.0051e+00, -1.0887e+00, -1.9877e+00, ..., -6.2269e-02,
7.6800e-02, 1.0686e+00],
[-1.9546e+00, -5.1302e-01, -1.6818e+00, ..., 2.5244e-01,
2.8635e-01, 4.1356e-02],
...,
[-1.7088e+00, -1.4342e+00, 2.8591e-01, ..., -5.1003e+00,
-4.7624e+00, -1.3952e+00],
[-1.7202e+00, -1.0789e+00, -8.8430e-01, ..., -4.2806e+00,
-4.8448e+00, 1.1972e+00],
[-3.2156e-01, 4.3779e-01, -5.3268e-01, ..., -1.2451e+00,
5.3946e-01, 3.1173e-01]]],
[[[ 8.4853e-01, 1.5169e+00, 1.2073e+00, ..., 1.4244e+00,
1.1179e+00, 1.0181e+00],
[-9.6752e-01, -2.0496e+00, -1.0641e+00, ..., -4.1012e-01,
-8.7716e-01, -1.7700e+00],
[-1.7610e-01, -7.4249e-01, 1.4189e-01, ..., -2.0430e+00,
-3.2066e-01, 2.7829e-02],
...,
[-1.4496e+00, -8.9844e-01, -4.9123e-02, ..., -6.2673e-01,
-2.3462e+00, -1.6316e+00],
[-2.8836e+00, -3.6521e+00, -3.6967e-01, ..., 1.6052e+00,
2.0100e+00, 1.3511e+00],
[-2.1656e-01, -5.4482e-02, 1.0891e+00, ..., -1.2617e+00,
-1.6161e+00, -2.9884e-01]],
[[-3.8053e+00, -7.3160e-01, -1.5767e+00, ..., -1.5345e+00,
-2.4686e+00, -1.2226e+00],
[-4.1642e+00, -2.9211e+00, -2.1606e+00, ..., -2.0349e+00,
-2.8184e+00, -8.6880e-01],
[-2.5382e+00, -1.6900e+00, -2.6536e+00, ..., -3.1950e+00,
-4.9998e+00, -2.8657e+00],
...,
[-1.7495e+00, -2.0768e+00, -1.7700e+00, ..., -1.1680e+01,
-1.0808e+01, -5.1318e+00],
[-3.0574e+00, -2.1931e+00, -1.2276e+00, ..., -2.7995e+00,
-2.5562e+00, -2.0230e+00],
[-2.1884e+00, -3.3896e+00, -2.0621e+00, ..., -2.1470e+00,
-2.0763e+00, -1.0749e+00]],
[[ 2.9350e+00, 4.2906e+00, 3.3826e+00, ..., 2.9017e+00,
1.0692e+00, 3.0864e+00],
[ 2.0008e-01, 1.0321e+00, 2.2360e+00, ..., 1.0333e+00,
-9.4544e-01, 2.8602e+00],
[ 9.5160e-01, 1.4115e+00, 1.6526e+00, ..., 2.2381e+00,
-5.8321e-02, 2.9943e+00],
...,
[ 9.8458e-01, -3.6752e-02, -5.9220e-01, ..., -3.0394e+00,
-6.5616e+00, -2.1350e+00],
[-1.1482e+00, -1.6816e+00, -3.3496e+00, ..., 7.6993e-01,
1.8789e+00, 1.9280e+00],
[-5.3978e-02, -6.3914e-01, -9.2735e-01, ..., 1.2835e+00,
1.7252e+00, 1.7270e+00]],
...,
[[-1.9117e+00, -7.3402e+00, -5.5578e+00, ..., -4.0817e+00,
-3.2528e+00, -5.6914e+00],
[-2.4750e+00, -6.9532e+00, -4.3301e+00, ..., -4.1464e+00,
-2.6592e+00, -4.3486e+00],
[-9.2828e-01, -4.2518e+00, -3.8711e+00, ..., -2.0766e+00,
-2.3495e+00, -9.5759e-01],
...,
[-9.9402e-01, -3.1918e+00, -5.2455e+00, ..., -4.5361e+00,
-4.4082e+00, -7.8198e+00],
[-2.5335e-02, -1.4938e+00, -3.1779e+00, ..., -8.5815e+00,
-7.5872e+00, -7.8620e+00],
[-4.3206e-01, -2.0033e+00, -3.4614e-01, ..., -3.1943e+00,
-2.6868e+00, -3.2536e+00]],
[[-1.7694e-01, 3.3103e-01, 5.6361e-01, ..., 6.5766e-01,
8.8013e-01, 1.3927e+00],
[ 1.6719e+00, 1.0670e+00, 1.5394e+00, ..., 1.6229e+00,
2.3114e+00, 3.1912e+00],
[ 1.2183e+00, 2.8703e-01, -5.2323e-01, ..., 1.1251e+00,
-5.5610e-02, 2.2005e+00],
...,
[ 7.9688e-01, -9.3926e-01, -1.7038e+00, ..., 8.2558e-01,
1.0417e+00, 1.0110e+00],
[ 8.0682e-01, 1.1206e+00, -2.9885e-01, ..., 2.1214e+00,
1.2525e+00, 2.4144e+00],
[ 8.5292e-01, 7.4176e-01, 2.5515e-01, ..., 5.4121e-01,
6.4079e-01, 1.6234e+00]],
[[-5.7350e-01, 3.3694e-01, 1.4362e-01, ..., 6.0676e-01,
5.6057e-01, 1.3685e+00],
[-1.3567e+00, -1.7406e+00, -1.9109e+00, ..., -7.1775e-01,
-4.6365e-01, 4.7900e-02],
[-2.3203e+00, -7.8170e-01, -9.0653e-01, ..., -5.7850e-01,
5.4816e-01, 8.6420e-01],
...,
[-1.8120e+00, -4.7316e-01, -2.5567e-02, ..., -5.4508e+00,
-6.2100e+00, -5.0541e+00],
[-3.4752e+00, -2.3929e+00, -1.3677e+00, ..., -6.6667e-01,
-7.1303e-01, -1.1269e+00],
[-1.2958e+00, -1.8370e+00, -2.3388e-01, ..., -8.6002e-01,
-9.1422e-01, -8.1282e-02]]]], device='cuda:0'),
tensor([[[[ 7.5571e-01, 1.4258e+00, 1.4463e+00, ..., 3.2614e-01,
3.6530e-01, -2.0633e-02],
[-3.0993e-01, -1.2723e+00, -8.3528e-01, ..., 1.4600e-01,
4.6052e-02, -3.8178e-01],
[-1.5832e-01, -1.8553e+00, -2.5492e+00, ..., 6.9681e-02,
1.7652e-01, 6.8617e-01],
...,
[-6.3086e-02, -9.3551e-01, 3.5255e-01, ..., 3.0321e+00,
1.3786e+00, 9.8267e-01],
[-4.8321e-01, -2.1755e+00, -1.5736e+00, ..., 1.0934e+00,
1.3866e+00, 1.5225e-01],
[ 4.8228e-01, -8.6056e-01, -1.9827e+00, ..., -1.3321e+00,
-3.0353e-01, -2.8011e-01]],
[[-2.9901e+00, -2.5857e+00, -4.9167e+00, ..., -1.7233e+00,
-2.3761e+00, -1.5468e+00],
[-3.2304e+00, -4.1813e+00, -6.6683e+00, ..., -1.3711e+00,
-2.5988e+00, -1.7339e+00],
[-3.2937e+00, -2.2842e+00, -2.9109e-01, ..., -1.9076e+00,
-2.3755e+00, -1.1995e+00],
...,
[-9.6644e-01, -6.6614e-01, -3.7446e+00, ..., -2.5102e+00,
-1.3123e+00, 2.4228e-01],
[-2.3705e+00, -9.7409e-02, -2.9810e+00, ..., -2.4146e+00,
-3.6135e+00, -2.5094e-01],
[-1.0594e+00, -1.1614e+00, -2.1877e+00, ..., -3.7174e+00,
-5.1344e+00, -1.0736e+00]],
[[ 2.0482e+00, 2.3935e+00, 1.5401e+00, ..., 1.6793e+00,
8.1226e-01, 2.1402e+00],
[ 3.1713e-01, 1.2893e+00, 2.9544e+00, ..., 8.5614e-01,
3.7890e-01, 2.3066e+00],
[ 1.0307e+00, -1.4226e-01, -2.1193e+00, ..., 6.9780e-02,
7.9271e-01, 2.0524e+00],
...,
[ 1.0737e+00, 1.7407e+00, 3.2625e+00, ..., -2.3554e+00,
-8.2778e-01, 1.3264e+00],
[ 7.5399e-01, 2.1560e+00, 4.2620e+00, ..., -3.9547e+00,
-1.6383e+00, 1.0106e+00],
[ 1.8958e+00, 1.0607e+00, 2.6947e+00, ..., -3.4424e+00,
8.3375e-02, 8.5056e-01]],
...,
[[-2.2473e+00, -6.2940e+00, -5.3889e+00, ..., -2.1385e+00,
-1.7552e+00, -3.1522e+00],
[-2.8050e+00, -6.2550e+00, -2.3471e+00, ..., -2.6606e+00,
-2.4416e+00, -3.1819e+00],
[-1.4637e+00, -5.5663e+00, -2.6909e+00, ..., -2.0547e+00,
-2.2699e+00, -2.1866e+00],
...,
[-2.0188e+00, -5.1307e+00, -3.4013e+00, ..., -1.2392e+01,
-6.4379e+00, -4.0815e+00],
[-2.8559e+00, -7.0441e+00, -4.2842e+00, ..., -9.9461e+00,
-9.9346e+00, -6.8900e+00],
[-3.0556e+00, -7.2370e+00, -3.5700e+00, ..., -1.9604e+00,
-5.1513e+00, -4.2793e+00]],
[[ 6.6809e-01, 6.4269e-01, 9.3433e-02, ..., 1.0889e+00,
1.4710e+00, 1.4485e+00],
[ 5.2651e-01, -1.4042e+00, -4.8542e+00, ..., 9.2867e-01,
7.9919e-01, 2.0432e+00],
[ 1.7677e-01, -2.5624e-01, 1.2864e-01, ..., 5.3518e-01,
4.8758e-02, 1.3094e+00],
...,
[ 1.5983e+00, -4.8060e-02, -1.9570e+00, ..., -1.8029e+00,
2.0352e+00, 2.1247e+00],
[ 2.2370e+00, 1.4649e+00, -1.0740e+00, ..., -3.4014e+00,
1.3405e-01, 3.1731e+00],
[ 1.0470e+00, 6.3067e-01, -8.3587e-02, ..., -6.7403e-01,
-1.0646e-01, 1.8575e+00]],
[[-2.4508e-01, -1.0148e+00, -1.1478e+00, ..., -5.5162e-01,
-8.9186e-01, -7.2359e-02],
[-1.9296e+00, -3.8871e+00, -2.8529e+00, ..., -1.1043e+00,
-1.0007e+00, -5.8129e-01],
[-2.1295e+00, -3.1927e+00, -3.8848e-02, ..., 4.9877e-03,
-1.7077e-01, -2.0113e-01],
...,
[-2.5110e+00, -1.8153e+00, -1.8301e+00, ..., 1.4165e-01,
-1.3764e+00, -1.1322e-01],
[-2.5567e+00, -2.2645e+00, -1.8115e+00, ..., -3.2937e-01,
-2.0649e+00, -1.0054e+00],
[-4.4890e-01, -1.7854e-01, -5.6181e-01, ..., 7.2191e-01,
-3.6771e+00, -2.0825e+00]]],
[[[ 8.4901e-01, 7.1494e-01, 2.2643e-01, ..., -3.7676e-01,
-3.4638e-01, -6.9692e-01],
[ 7.0666e-01, -8.6296e-01, -6.6674e-01, ..., 2.7495e-01,
2.4661e-01, -6.7325e-01],
[ 6.0075e-01, -2.4900e-01, -5.9664e-01, ..., 1.0322e+00,
1.8235e+00, 9.2128e-01],
...,
[ 1.7096e-01, -1.0871e+00, -1.0563e+00, ..., -4.6206e-01,
3.7259e-01, 5.7069e-01],
[-6.6044e-01, -2.4106e+00, -2.1454e+00, ..., 8.6436e-01,
-3.6793e-01, -1.3649e-01],
[-1.2927e-01, -1.7993e+00, -9.4229e-01, ..., -7.8528e-01,
-1.3877e+00, -2.8779e-01]],
[[-2.4106e+00, 1.4574e+00, 9.2449e-01, ..., -3.0973e+00,
-3.2067e+00, -8.5587e-01],
[-2.3585e+00, -2.4796e-01, -1.1958e-01, ..., -3.7655e+00,
-2.7675e+00, 7.2176e-02],
[-1.6380e+00, 8.1950e-01, 1.5487e+00, ..., -1.1497e+00,
-2.2774e+00, -5.6058e-01],
...,
[-1.4394e+00, -1.9013e+00, -4.2057e+00, ..., -1.6864e+00,
-1.7319e+00, -6.0260e-01],
[-3.3695e+00, -1.7981e+00, -2.6804e+00, ..., -3.6510e+00,
-2.7327e+00, -1.2430e+00],
[-2.3301e+00, -1.5500e+00, -6.0189e-01, ..., -3.9834e+00,
-2.0166e+00, -1.0426e+00]],
[[ 1.6918e+00, 4.0215e-01, 2.6226e-01, ..., 2.6720e+00,
1.0059e+00, 2.2597e+00],
[-1.8244e-02, -3.1487e+00, -1.2246e+00, ..., 1.5897e+00,
5.9616e-01, 2.7469e+00],
[ 9.3164e-01, -2.0828e+00, -2.0872e-01, ..., 1.6022e-02,
-2.3139e-01, 2.5168e+00],
...,
[ 1.1690e+00, 1.6259e+00, 2.8934e+00, ..., 1.4114e+00,
1.8638e+00, 2.3420e+00],
[ 5.3356e-01, 2.6610e+00, 1.7581e+00, ..., -1.5890e-01,
1.1612e+00, 1.8851e+00],
[ 1.6797e+00, 2.1924e+00, 1.2518e+00, ..., 9.9101e-01,
1.8052e+00, 1.7996e+00]],
...,
[[-2.3354e+00, -5.3429e+00, -3.5632e+00, ..., -4.3437e+00,
-3.2666e+00, -4.1634e+00],
[-3.3872e+00, -6.3170e+00, -2.7038e+00, ..., -5.2137e+00,
-4.0889e+00, -5.5627e+00],
[-1.9330e+00, -3.2817e+00, -1.2518e+00, ..., -4.4226e+00,
-5.3200e+00, -4.2657e+00],
...,
[-2.4058e+00, -5.6936e+00, -4.4940e+00, ..., -4.5713e+00,
-4.8049e+00, -4.7270e+00],
[-2.9842e+00, -6.1348e+00, -4.8003e+00, ..., -9.7407e+00,
-6.7432e+00, -6.2310e+00],
[-1.9014e+00, -4.7248e+00, -3.3459e+00, ..., -3.6914e+00,
-2.1233e+00, -3.2003e+00]],
[[ 5.6520e-01, 1.6669e+00, 9.3536e-01, ..., 6.5033e-01,
1.2630e+00, 1.8504e+00],
[ 1.3571e+00, 4.1027e-01, -1.3847e+00, ..., -6.9511e-02,
1.9378e+00, 3.0920e+00],
[ 1.5817e+00, 1.3372e+00, 2.0630e-01, ..., -7.1166e-01,
1.9888e-01, 2.3740e+00],
...,
[ 1.4146e+00, -3.3560e-01, -1.2937e+00, ..., 2.3710e+00,
1.8072e+00, 1.5732e+00],
[ 1.9664e+00, 1.5819e+00, 7.9928e-01, ..., 4.8715e-01,
1.6680e+00, 2.0084e+00],
[ 3.8639e-01, 8.3844e-01, 9.6759e-01, ..., -5.8137e-01,
1.1844e+00, 1.7232e+00]],
[[-6.9329e-01, 7.8326e-01, 1.0919e-01, ..., -9.9865e-01,
-8.5210e-01, 4.0325e-01],
[-1.2989e+00, -1.6985e-01, -1.2963e+00, ..., -7.5762e-01,
-4.9098e-02, 8.8296e-01],
[-1.7514e+00, -7.0434e-01, -2.3679e-01, ..., 1.9392e+00,
1.0609e+00, 4.5862e-01],
...,
[-1.9911e+00, -2.3980e+00, -1.4206e+00, ..., -8.1239e-01,
-7.5926e-01, -5.6309e-01],
[-1.6649e+00, -2.3181e+00, -2.4923e+00, ..., -1.1368e+00,
-2.0463e+00, -6.7331e-01],
[-8.4667e-01, -9.8914e-01, -1.4863e+00, ..., -2.6792e+00,
-1.7155e+00, -9.2149e-02]]],
[[[ 9.3937e-01, 1.5069e+00, 1.2957e+00, ..., 1.4325e+00,
1.0819e+00, 9.7838e-01],
[-9.9851e-01, -2.3683e+00, -1.2474e+00, ..., -3.0821e-01,
-5.7886e-01, -1.5179e+00],
[-7.0711e-02, -1.0270e+00, 1.1981e-01, ..., -1.3528e-01,
6.8239e-01, 7.3479e-01],
...,
[ 5.1233e-01, -3.6948e-01, 1.2350e-01, ..., 5.6563e-01,
2.2124e-01, 6.6454e-01],
[ 4.4751e-01, -1.2681e+00, -5.4238e-01, ..., 6.6086e-01,
6.2116e-01, -3.8156e-02],
[ 5.7857e-01, 5.0252e-01, 4.5994e-01, ..., 2.6887e-01,
-4.8333e-02, 6.4254e-01]],
[[-4.1584e+00, -6.4772e-01, -1.7550e+00, ..., -1.5411e+00,
-2.3710e+00, -1.0914e+00],
[-4.1557e+00, -2.5149e+00, -1.9486e+00, ..., -2.2744e+00,
-3.0781e+00, -8.3572e-01],
[-2.5965e+00, -1.4547e+00, -1.5874e+00, ..., -2.6459e+00,
-2.8017e+00, -1.0078e+00],
...,
[-9.6995e-01, -1.7320e+00, -1.9767e+00, ..., -1.4807e+00,
-1.3030e+00, -1.4174e-01],
[-1.3537e+00, -6.0284e-01, -9.8303e-01, ..., -7.6180e-01,
-1.1754e+00, -7.4280e-02],
[ 5.9066e-01, 1.9109e-01, -6.9557e-01, ..., 3.2070e-01,
1.2732e-01, 4.5423e-01]],
[[ 3.3148e+00, 4.4786e+00, 3.6018e+00, ..., 2.7573e+00,
1.0614e+00, 2.8943e+00],
[ 8.1355e-01, 1.0285e+00, 2.4371e+00, ..., 1.2593e+00,
-7.9005e-01, 2.6569e+00],
[ 1.6578e+00, 1.4370e+00, 1.8944e+00, ..., 5.4451e-01,
3.6815e-02, 2.1286e+00],
...,
[ 4.7778e-01, 1.3072e+00, 1.0408e+00, ..., -1.1382e-01,
3.2182e-01, 1.4894e+00],
[ 2.4842e-01, 1.6960e+00, 1.1103e-01, ..., -2.6236e-01,
9.7358e-02, 1.1388e+00],
[ 1.4029e+00, 1.1332e+00, -6.2142e-01, ..., 7.7404e-01,
4.4063e-01, 1.1494e+00]],
...,
[[-2.7332e+00, -8.0667e+00, -5.9058e+00, ..., -4.0180e+00,
-3.1886e+00, -5.4255e+00],
[-3.3162e+00, -7.6000e+00, -4.5026e+00, ..., -3.5431e+00,
-2.7163e+00, -4.1980e+00],
[-1.4130e+00, -5.0944e+00, -3.8952e+00, ..., -3.1065e+00,
-3.0471e+00, -2.3778e+00],
...,
[-2.2543e+00, -5.4884e+00, -3.9564e+00, ..., -3.5538e+00,
-3.4793e+00, -3.5739e+00],
[-2.4290e+00, -6.5463e+00, -4.9042e+00, ..., -5.0753e+00,
-4.2611e+00, -4.8185e+00],
[-2.3379e+00, -6.6355e+00, -3.9277e+00, ..., -4.1146e+00,
-3.1417e+00, -4.0886e+00]],
[[-3.0296e-01, 4.6090e-01, 4.1671e-01, ..., 6.3302e-01,
8.6537e-01, 1.4233e+00],
[ 1.8816e+00, 1.3770e+00, 1.6052e+00, ..., 1.8789e+00,
2.5359e+00, 2.9944e+00],
[ 1.3295e+00, 4.6990e-01, 6.4081e-01, ..., 1.0474e+00,
1.6785e+00, 2.4154e+00],
...,
[ 1.1265e+00, -5.6650e-01, -3.2291e-01, ..., 1.0805e+00,
8.5762e-01, 1.2855e+00],
[ 1.6086e+00, 5.3098e-01, 1.2782e+00, ..., 1.0252e+00,
6.0439e-01, 2.0620e+00],
[ 1.2209e+00, 6.6160e-01, 7.3040e-01, ..., 1.1953e+00,
1.2605e+00, 1.4007e+00]],
[[-6.2262e-01, 3.3256e-01, 4.8972e-02, ..., 5.7063e-01,
5.2209e-01, 1.3566e+00],
[-1.5777e+00, -1.8976e+00, -1.7771e+00, ..., -6.3323e-01,
-4.3666e-01, 2.2307e-01],
[-2.4306e+00, -9.1125e-01, 3.1839e-01, ..., 3.7148e-01,
2.6493e-01, 7.0812e-01],
...,
[-1.5453e+00, -1.9260e+00, -1.3000e+00, ..., -6.4905e-01,
-8.6202e-01, -6.8839e-01],
[-1.2095e+00, -2.0706e+00, -1.3128e+00, ..., -6.5789e-01,
-1.1737e+00, -4.3650e-02],
[ 1.7371e-01, 2.4847e-01, -1.0375e-01, ..., 2.9322e-01,
-1.4725e-02, 7.6654e-01]]],
[[[ 5.9172e-01, 8.6446e-01, 5.5722e-01, ..., 3.3901e-01,
4.2874e-01, 2.5482e-02],
[-2.7459e-01, -8.7121e-01, -4.8677e-01, ..., 1.5692e-01,
6.8858e-02, -3.0647e-01],
[-1.9421e-01, -3.5865e-02, 3.5811e-01, ..., -1.6991e-01,
2.8808e-02, 6.3633e-01],
...,
[ 4.4595e-01, -2.3559e-01, -1.0381e-01, ..., 2.5291e-01,
8.6049e-01, 6.4797e-01],
[-4.5741e-02, -6.6969e-01, 1.2199e-01, ..., 9.2700e-01,
1.5847e+00, 1.1656e-01],
[ 8.1794e-01, -2.4175e-01, -1.4519e-01, ..., 4.0426e-01,
3.0365e-01, 7.1145e-01]],
[[-2.3699e+00, -5.6468e-01, -1.8203e+00, ..., -1.6005e+00,
-2.1966e+00, -1.5026e+00],
[-2.4913e+00, -5.4575e-01, -1.4426e+00, ..., -1.3262e+00,
-2.4042e+00, -1.7004e+00],
[-2.3791e+00, -1.5885e+00, -1.8694e+00, ..., -2.1393e+00,
-2.4252e+00, -1.1433e+00],
...,
[-6.3959e-01, -7.6062e-01, -1.1457e+00, ..., -5.0070e-01,
-1.6794e-01, 7.4693e-01],
[-2.5902e+00, -2.4695e-01, -2.7400e-01, ..., 1.2958e-02,
-2.2695e-02, 6.6351e-01],
[-1.6154e+00, -1.3214e+00, -9.3319e-01, ..., -4.7110e-01,
-6.3837e-01, -1.6804e-01]],
[[ 1.9367e+00, 2.5431e+00, 1.6002e+00, ..., 1.5790e+00,
7.7613e-01, 2.0721e+00],
[-2.5654e-01, 3.7880e-01, 6.0620e-01, ..., 8.1591e-01,
3.4115e-01, 2.2450e+00],
[ 4.0935e-03, 3.1517e-01, 4.1415e-01, ..., -1.6363e-01,
8.1613e-01, 2.0045e+00],
...,
[ 7.4102e-01, 4.3746e-01, 1.4166e+00, ..., 3.8761e-01,
1.2442e-01, 1.6006e+00],
[ 1.1045e+00, 2.2793e+00, 2.0797e+00, ..., 9.9423e-01,
7.0581e-01, 1.6869e+00],
[ 2.4406e+00, 1.8355e+00, 2.4170e+00, ..., 1.9812e+00,
1.0369e+00, 1.7143e+00]],
...,
[[-2.0405e+00, -4.9844e+00, -3.1689e+00, ..., -2.0588e+00,
-1.6564e+00, -3.0255e+00],
[-2.4718e+00, -5.2539e+00, -2.9905e+00, ..., -2.4346e+00,
-2.2072e+00, -2.9473e+00],
[-1.5103e+00, -5.3863e+00, -3.3471e+00, ..., -2.5097e+00,
-2.5061e+00, -2.2105e+00],
...,
[-2.6024e+00, -4.4601e+00, -2.4273e+00, ..., -3.6011e+00,
-2.0815e+00, -2.6457e+00],
[-2.5944e+00, -5.6627e+00, -4.1267e+00, ..., -4.4224e+00,
-3.2270e+00, -4.8944e+00],
[-2.6950e+00, -6.0171e+00, -2.9054e+00, ..., -2.9855e+00,
-1.6752e+00, -3.2977e+00]],
[[ 9.4728e-01, 1.6436e+00, 8.1059e-01, ..., 1.1323e+00,
1.5526e+00, 1.4050e+00],
[ 1.1786e+00, 1.0828e+00, 9.4219e-01, ..., 9.1358e-01,
7.9748e-01, 1.9088e+00],
[ 6.5845e-01, -9.3631e-02, 3.5314e-01, ..., 3.4178e-01,
1.1064e-01, 1.4191e+00],
...,
[ 1.5110e+00, 3.3666e-02, 7.8267e-01, ..., 1.4268e+00,
1.7711e+00, 1.4553e+00],
[ 2.0558e+00, 1.1919e+00, 1.7556e+00, ..., 1.9357e+00,
1.6825e+00, 2.5314e+00],
[ 1.1673e+00, 9.1618e-01, 1.3970e+00, ..., 1.3224e+00,
1.4701e+00, 1.5285e+00]],
[[-8.1679e-02, -6.2991e-01, -1.1848e+00, ..., -5.4566e-01,
-8.7555e-01, -1.1429e-01],
[-8.7269e-01, -2.5439e+00, -2.5737e+00, ..., -1.1992e+00,
-1.0948e+00, -6.8742e-01],
[-1.4584e+00, -1.9518e+00, -9.5190e-01, ..., -2.5765e-01,
-4.5467e-01, -2.6150e-01],
...,
[-1.7170e+00, -1.5891e+00, -6.0425e-01, ..., -1.0530e-01,
-4.3884e-01, -4.6121e-01],
[-1.7443e+00, -1.2713e+00, -5.9624e-01, ..., -4.0547e-01,
-4.0546e-01, 4.0488e-01],
[-2.9436e-01, 3.8420e-01, -2.3129e-01, ..., 2.0934e-01,
6.9419e-03, 6.2725e-01]]],
[[[ 9.6884e-01, 1.6248e+00, 1.0153e+00, ..., 5.8117e-01,
3.6417e-01, -1.3414e-01],
[-8.9885e-01, -1.9526e+00, -1.0536e+00, ..., 3.7632e-01,
6.2819e-01, -6.3981e-01],
[ 2.6930e-02, -8.4452e-01, -1.1896e-02, ..., -9.9093e-01,
8.0211e-02, 5.9771e-01],
...,
[ 4.6653e-01, -3.0037e-01, -3.2910e-01, ..., 8.0412e-01,
4.1386e-01, 3.7025e-01],
[-2.0781e-01, -1.0421e+00, -3.2054e-01, ..., 3.6310e-01,
9.7908e-01, 1.5459e-01],
[ 7.3352e-01, -2.7263e-01, -2.5179e-01, ..., -5.6903e-01,
-4.2410e-01, 2.9299e-01]],
[[-4.2162e+00, -1.3741e+00, -2.4979e+00, ..., -2.9117e+00,
-3.5687e+00, -8.9158e-01],
[-4.0172e+00, -2.8635e+00, -2.6104e+00, ..., -1.8269e+00,
-3.4566e+00, -6.2739e-01],
[-2.5796e+00, -1.7917e+00, -1.7995e+00, ..., -3.5330e+00,
-4.0681e+00, -5.8996e-01],
...,
[-8.7289e-01, -6.2291e-01, -1.5433e+00, ..., -1.7272e+00,
-1.6373e+00, 2.0688e-01],
[-2.9210e+00, -3.6977e-01, -1.1091e+00, ..., -1.3108e+00,
-1.4863e+00, -3.2845e-02],
[-2.0858e+00, -1.4061e+00, -1.7135e+00, ..., -1.2336e+00,
-1.0381e+00, 3.0745e-02]],
[[ 2.9818e+00, 4.1364e+00, 3.5562e+00, ..., 2.3208e+00,
7.5966e-01, 2.3901e+00],
[ 3.2639e-01, 9.5350e-01, 2.7067e+00, ..., 1.9557e+00,
-6.0869e-02, 2.4745e+00],
[ 8.1486e-01, 1.3573e+00, 1.7988e+00, ..., 1.5704e+00,
1.1973e+00, 2.5639e+00],
...,
[ 2.9825e-01, 1.0084e+00, 1.4769e+00, ..., 7.7738e-02,
-3.5888e-01, 1.1386e+00],
[ 3.2346e-01, 2.2091e+00, 1.6651e+00, ..., -3.0426e-03,
3.1945e-01, 1.2328e+00],
[ 1.3810e+00, 1.4052e+00, 1.7285e+00, ..., 9.1813e-01,
8.6775e-01, 1.4098e+00]],
...,
[[-1.9807e+00, -7.3158e+00, -6.0003e+00, ..., -4.1611e+00,
-3.2104e+00, -3.9507e+00],
[-2.6659e+00, -6.4388e+00, -4.6528e+00, ..., -4.9624e+00,
-4.8158e+00, -4.7177e+00],
[-2.1686e-01, -2.6315e+00, -3.3712e+00, ..., -3.3631e+00,
-4.2862e+00, -2.4828e+00],
...,
[-9.7511e-01, -2.3229e+00, -1.8887e+00, ..., -5.2647e+00,
-3.0702e+00, -1.7824e+00],
[-1.3170e+00, -3.8247e+00, -3.5965e+00, ..., -6.1453e+00,
-4.2586e+00, -4.3305e+00],
[-2.2848e+00, -5.1224e+00, -3.1813e+00, ..., -4.0172e+00,
-2.4167e+00, -3.0452e+00]],
[[-7.8174e-01, -2.0952e-01, -2.1656e-01, ..., 1.0138e+00,
1.0472e+00, 1.7726e+00],
[ 1.7561e+00, 1.0333e+00, 1.5625e+00, ..., 1.1751e+00,
1.3546e+00, 2.9207e+00],
[ 1.1993e+00, 2.3713e-01, 9.4154e-01, ..., 1.5011e+00,
9.5846e-01, 2.3032e+00],
...,
[ 1.2758e+00, -1.2090e-01, 7.1020e-01, ..., 1.4208e+00,
9.3901e-01, 1.4577e+00],
[ 1.8628e+00, 1.0101e+00, 1.6591e+00, ..., 1.6556e+00,
1.1482e+00, 2.2524e+00],
[ 1.0148e+00, 4.2290e-01, 8.7109e-01, ..., 7.2418e-01,
1.1855e+00, 1.5726e+00]],
[[-5.0325e-01, 5.7742e-01, -4.3153e-01, ..., -2.2457e-01,
-5.5960e-01, 9.0791e-01],
[-1.0660e+00, -1.1241e+00, -1.8437e+00, ..., -3.7899e-01,
-1.7531e-01, 1.0462e+00],
[-2.0229e+00, -5.4782e-01, -8.4832e-02, ..., -1.1316e-01,
9.8242e-02, 2.1779e-01],
...,
[-1.7470e+00, -1.5311e+00, -1.2739e+00, ..., -8.6828e-02,
-8.3038e-01, -2.3060e-01],
[-1.7435e+00, -1.1289e+00, -1.1175e+00, ..., -2.5749e-01,
-1.1259e+00, 1.5148e-01],
[-3.7684e-01, 3.7353e-01, -4.3030e-01, ..., 2.0618e-02,
-3.0750e-01, 5.6928e-01]]]], device='cuda:0'),
tensor([[[[ 9.8920e-01, 1.6184e+00, 1.1580e+00, ..., 8.0824e-01,
1.0044e+00, 4.0508e-01],
[-8.6247e-01, -1.9153e+00, -6.4824e-01, ..., -3.5465e-01,
-1.7655e-01, -2.9539e-01],
[ 7.6681e-02, -7.8201e-01, -5.1889e-01, ..., -1.2477e+00,
-2.3490e-01, 5.8008e-01],
...,
[ 4.1249e-01, -2.9253e-01, -1.1254e-01, ..., -1.3777e-01,
-9.9465e-02, 4.3577e-01],
[-2.4446e-01, -1.1248e+00, -3.6022e-02, ..., -4.5940e-02,
1.4547e-01, -8.9837e-02],
[ 7.3713e-01, -3.2842e-01, -1.6340e-01, ..., -3.5310e-01,
-8.7623e-01, 2.0042e-01]],
[[-4.0944e+00, -1.2666e+00, -2.1247e+00, ..., -1.4154e+00,
-2.0801e+00, -1.8470e+00],
[-3.9648e+00, -3.0063e+00, -2.5566e+00, ..., -1.0345e+00,
-2.2528e+00, -2.1093e+00],
[-2.5767e+00, -2.5693e+00, -3.7913e+00, ..., -1.0038e+00,
-1.7601e+00, -1.4706e+00],
...,
[-8.9981e-01, -7.0767e-01, -1.1796e+00, ..., -2.0100e-01,
-7.2816e-01, -8.0217e-01],
[-2.8965e+00, -3.6873e-01, -7.9110e-01, ..., -5.9452e-01,
-1.0547e+00, -1.1351e+00],
[-2.0925e+00, -1.4813e+00, -1.3766e+00, ..., -6.3880e-01,
-6.3672e-01, -4.5478e-01]],
[[ 2.9724e+00, 4.0797e+00, 3.2588e+00, ..., 1.7340e+00,
8.4473e-01, 2.6183e+00],
[ 3.4384e-01, 8.5806e-01, 2.3115e+00, ..., 1.0716e+00,
5.4872e-01, 2.7298e+00],
[ 9.1387e-01, 1.7224e+00, 2.4210e+00, ..., 1.4071e+00,
6.8841e-01, 1.8142e+00],
...,
[ 3.1728e-01, 1.0146e+00, 1.1591e+00, ..., -5.8386e-01,
3.5974e-01, 1.3087e+00],
[ 3.6201e-01, 2.1779e+00, 1.5162e+00, ..., -1.4939e-01,
6.8807e-01, 1.0685e+00],
[ 1.4239e+00, 1.3500e+00, 1.6675e+00, ..., 6.1375e-01,
1.2588e+00, 1.3423e+00]],
...,
[[-2.0708e+00, -7.3951e+00, -5.9249e+00, ..., -4.9855e+00,
-3.6575e+00, -4.3783e+00],
[-2.7147e+00, -6.5803e+00, -5.0778e+00, ..., -5.2917e+00,
-3.6125e+00, -3.3160e+00],
[-2.6226e-01, -2.9553e+00, -3.3773e+00, ..., -3.4963e+00,
-2.5327e+00, -1.3414e+00],
...,
[-9.6367e-01, -2.2789e+00, -1.8700e+00, ..., -7.1329e+00,
-3.8791e+00, -2.4702e+00],
[-1.3130e+00, -3.7068e+00, -3.5382e+00, ..., -5.1556e+00,
-4.2141e+00, -4.0214e+00],
[-2.2274e+00, -5.0065e+00, -3.0986e+00, ..., -3.3859e+00,
-2.4836e+00, -2.2852e+00]],
[[-7.1382e-01, -1.0582e-01, 2.1890e-01, ..., 1.4645e+00,
1.7106e+00, 1.2094e+00],
[ 1.7676e+00, 8.2768e-01, 1.4709e+00, ..., 1.3535e+00,
1.1415e+00, 1.8774e+00],
[ 1.1709e+00, -4.2912e-01, -1.8368e-01, ..., 2.7093e+00,
5.8914e-01, 1.3134e+00],
...,
[ 1.2788e+00, -1.8149e-01, 1.0692e+00, ..., 2.1336e+00,
7.1773e-01, 8.9951e-01],
[ 1.9048e+00, 1.0106e+00, 1.8485e+00, ..., 1.2938e+00,
6.4963e-01, 1.4436e+00],
[ 1.0121e+00, 3.9929e-01, 9.5839e-01, ..., 7.2166e-01,
6.8238e-01, 1.4020e+00]],
[[-4.8715e-01, 5.8591e-01, 1.2376e-01, ..., -3.5368e-01,
-5.8392e-01, -9.5181e-02],
[-1.0677e+00, -1.2944e+00, -1.1579e+00, ..., -1.3438e+00,
-1.3642e+00, -1.0681e+00],
[-2.0177e+00, -1.0941e+00, -2.7911e-01, ..., -6.9213e-01,
-8.8338e-01, -1.4670e-01],
...,
[-1.7994e+00, -1.5104e+00, -1.0269e+00, ..., -9.0975e-02,
-1.2734e+00, -8.0513e-01],
[-1.8435e+00, -1.1004e+00, -8.1045e-01, ..., -7.0457e-01,
-1.7740e+00, -6.6597e-01],
[-3.8771e-01, 3.5648e-01, -2.7163e-01, ..., -2.8612e-01,
-7.9957e-01, -6.4923e-03]]],
[[[ 6.8651e-01, 1.0826e+00, 6.9168e-01, ..., 7.4353e-01,
2.4210e-01, -6.3705e-01],
[-2.5448e-01, -1.0471e+00, -4.9713e-01, ..., -1.7511e+00,
-2.4297e+00, -1.0361e+00],
[ 2.8553e-02, -9.4171e-01, -4.9979e-01, ..., -1.3273e+00,
-4.2891e+00, -1.9847e+00],
...,
[ 5.4067e-02, -1.1466e+00, -1.0774e+00, ..., 1.2681e-01,
4.4290e-01, 6.4093e-01],
[-4.5439e-01, -1.7918e+00, -8.3824e-01, ..., 4.7851e-01,
8.2653e-01, 1.7876e-01],
[ 2.9029e-01, -5.3650e-01, -7.7454e-01, ..., -4.7565e-02,
-3.3498e-01, 4.5795e-01]],
[[-2.8852e+00, -1.6514e+00, -3.2412e+00, ..., -4.8381e+00,
-7.1196e+00, -3.2283e+00],
[-4.0041e+00, -2.4194e+00, -2.5706e+00, ..., -2.3022e+00,
-1.3579e+00, -9.9704e-02],
[-4.1833e+00, -3.6944e+00, -3.3409e+00, ..., -2.1978e+00,
-3.2676e+00, -3.0674e+00],
...,
[-2.8692e+00, -1.8708e+00, -2.9484e+00, ..., -1.0641e+00,
-1.4077e+00, 3.9919e-01],
[-3.5696e+00, -1.5047e+00, -2.1585e+00, ..., -2.6624e-01,
-9.3739e-01, 2.5797e-01],
[-1.2389e+00, -8.1651e-01, -1.5378e+00, ..., -7.9161e-01,
-5.1029e-01, 1.6950e-02]],
[[ 1.6255e+00, 2.7049e+00, 2.3851e+00, ..., 1.4460e+00,
6.0741e-01, 2.0789e+00],
[ 1.2661e-01, 1.1649e+00, 1.8031e+00, ..., -6.4758e-01,
-2.6288e+00, -1.4538e+00],
[ 6.8010e-01, 1.3137e+00, 5.2601e-01, ..., -1.6711e+00,
-5.5356e+00, -3.8899e+00],
...,
[ 9.3788e-01, 1.2711e+00, 2.6516e-01, ..., -4.1897e-01,
-1.2059e-01, 1.1408e+00],
[ 6.5922e-01, 1.7877e+00, 7.6415e-01, ..., 3.3279e-02,
3.2639e-01, 1.0328e+00],
[ 1.5032e+00, 1.4910e+00, 1.5596e+00, ..., 6.5335e-01,
5.6317e-01, 1.1854e+00]],
...,
[[-1.3585e+00, -4.5218e+00, -3.6940e+00, ..., -3.8788e+00,
-3.2149e+00, -5.4485e+00],
[-1.7156e+00, -3.8258e+00, -3.9700e+00, ..., -5.0900e+00,
-1.1650e+00, -1.0143e+01],
[-9.2061e-01, -2.8756e+00, -2.7570e+00, ..., -4.1377e+00,
1.4606e+00, -2.2637e+00],
...,
[-1.5383e+00, -4.0418e+00, -4.6415e+00, ..., -4.4408e+00,
-2.5721e+00, -3.0101e+00],
[-1.4016e+00, -4.4693e+00, -5.2502e+00, ..., -6.4530e+00,
-3.6818e+00, -4.6460e+00],
[-1.4409e+00, -4.6029e+00, -4.0812e+00, ..., -4.7291e+00,
-2.0596e+00, -2.9990e+00]],
[[ 8.0729e-01, 3.6505e-01, -1.6226e-01, ..., -3.0654e+00,
-6.0357e+00, -1.7997e+00],
[ 1.0486e+00, 8.4451e-01, 7.9624e-01, ..., 2.0247e+00,
-6.9859e-01, -3.3983e-01],
[ 3.4099e-01, -4.0723e-01, -8.5704e-01, ..., -8.0371e-02,
1.3854e+00, 1.6813e-01],
...,
[ 8.2366e-01, -1.4098e-02, -1.2979e-01, ..., 2.7236e+00,
1.5043e+00, 1.7428e+00],
[ 1.5105e+00, 1.3508e+00, 1.1647e+00, ..., 2.7776e+00,
1.4061e+00, 2.4040e+00],
[ 7.0534e-01, 9.7465e-01, 9.9279e-01, ..., 1.2573e+00,
1.1718e+00, 1.3329e+00]],
[[-2.1928e-01, -1.9172e-01, -8.6746e-01, ..., -1.5592e+00,
-1.5080e-02, 2.0378e-01],
[-1.3220e+00, -1.3788e+00, -1.2853e+00, ..., -2.7963e+00,
-7.5674e-01, 3.3337e+00],
[-1.7048e+00, -1.1764e+00, -5.1036e-01, ..., -4.1391e+00,
-2.6924e+00, 1.1404e+00],
...,
[-1.6834e+00, -1.6354e+00, -1.8441e+00, ..., -1.3087e+00,
-9.3299e-01, -6.3672e-01],
[-1.3952e+00, -1.6793e+00, -1.6805e+00, ..., -1.1061e+00,
-9.0309e-01, 5.0530e-02],
[-2.3414e-01, 1.3978e-02, -5.3472e-01, ..., -1.2785e+00,
-7.8801e-01, 1.1736e-01]]],
[[[ 7.2729e-01, 1.0818e+00, 7.0932e-01, ..., 1.0488e+00,
9.6321e-01, 3.8844e-01],
[-3.6213e-01, -1.3408e+00, -4.9655e-01, ..., 6.9212e-02,
1.0105e-01, -8.5983e-01],
[ 1.3156e-01, -6.6173e-01, 1.3571e-01, ..., 8.0744e-01,
4.5615e-01, -6.0207e-02],
...,
[-2.5540e-01, -1.4552e+00, 7.8811e-01, ..., -1.0687e-01,
3.9736e-01, 7.0107e-01],
[-4.5180e-01, -2.1094e+00, -8.1364e-01, ..., 7.4568e-01,
9.1366e-01, 1.5936e-01],
[ 3.5342e-01, -5.6249e-01, -7.8183e-01, ..., -7.8163e-03,
-3.6123e-01, 4.7703e-01]],
[[-3.0476e+00, -1.4868e+00, -3.0263e+00, ..., -1.4863e+00,
-2.9236e+00, -1.0115e+00],
[-4.3551e+00, -1.8373e+00, -2.3861e+00, ..., -2.5033e+00,
-3.7307e+00, -9.4422e-01],
[-4.2956e+00, -2.7022e+00, -3.7965e+00, ..., -3.8938e+00,
-5.4045e+00, -1.5512e+00],
...,
[-3.0095e+00, -3.1733e+00, -5.0500e+00, ..., -1.4966e+00,
-1.3854e+00, 3.5879e-01],
[-3.6701e+00, -1.9327e+00, -2.7526e+00, ..., -7.8430e-02,
-8.7938e-01, 2.7042e-01],
[-1.2460e+00, -8.6970e-01, -1.6589e+00, ..., -6.1378e-01,
-4.9612e-01, 4.2288e-03]],
[[ 1.7701e+00, 2.7759e+00, 2.3588e+00, ..., 2.2434e+00,
8.4493e-01, 2.0159e+00],
[ 2.4895e-01, 1.1751e+00, 1.5917e+00, ..., 2.1519e+00,
6.9373e-01, 2.8890e+00],
[ 1.1958e+00, 1.6553e+00, 1.5331e+00, ..., -3.9592e-01,
-1.1498e+00, 8.4804e-01],
...,
[ 1.0657e+00, 2.5120e+00, 1.6794e+00, ..., -2.9522e-01,
-2.4565e-02, 1.1147e+00],
[ 7.1778e-01, 1.7727e+00, 1.7901e+00, ..., 2.9705e-01,
4.1807e-01, 1.0710e+00],
[ 1.5642e+00, 1.5913e+00, 1.5861e+00, ..., 8.3619e-01,
6.2423e-01, 1.2807e+00]],
...,
[[-1.4728e+00, -4.7249e+00, -3.8906e+00, ..., -4.6218e+00,
-3.4261e+00, -4.0192e+00],
[-1.8777e+00, -3.9656e+00, -4.3427e+00, ..., -5.3588e+00,
-4.2140e+00, -4.2285e+00],
[-1.0850e+00, -3.0030e+00, -3.0888e+00, ..., -4.8371e+00,
-4.1878e+00, -2.1930e+00],
...,
[-1.7786e+00, -4.3595e+00, -5.6418e+00, ..., -3.6338e+00,
-2.4441e+00, -3.1551e+00],
[-1.5327e+00, -4.9124e+00, -5.7302e+00, ..., -6.5002e+00,
-3.8240e+00, -4.8595e+00],
[-1.6266e+00, -4.8850e+00, -4.2332e+00, ..., -4.4356e+00,
-1.9716e+00, -3.1960e+00]],
[[ 7.7871e-01, 5.1808e-01, -1.4449e-01, ..., 1.2720e+00,
8.8469e-01, 1.6853e+00],
[ 1.1810e+00, 9.4222e-01, 4.2196e-01, ..., 2.1984e+00,
1.4603e+00, 2.7888e+00],
[ 6.0047e-01, 5.8478e-01, -3.6148e-01, ..., -7.0905e-01,
-2.8572e+00, 4.4315e-01],
...,
[ 8.0898e-01, -8.4164e-01, -2.0554e+00, ..., 2.6602e+00,
1.5313e+00, 1.7854e+00],
[ 1.5525e+00, 1.2063e+00, 1.1987e+00, ..., 2.8677e+00,
1.3082e+00, 2.4517e+00],
[ 6.7300e-01, 9.1998e-01, 1.0180e+00, ..., 1.3013e+00,
1.1671e+00, 1.3726e+00]],
[[-2.5897e-01, -1.4837e-01, -8.7827e-01, ..., 6.1192e-01,
2.6903e-01, 1.1458e+00],
[-1.4492e+00, -1.3592e+00, -1.9170e+00, ..., -3.3540e-01,
-5.0506e-02, 4.1075e-01],
[-1.8590e+00, -6.9806e-01, -5.5658e-01, ..., -7.7725e-01,
-3.5506e-01, 4.8254e-01],
...,
[-1.7204e+00, -1.9303e+00, -9.0746e-01, ..., -1.9072e+00,
-9.5665e-01, -6.7374e-01],
[-1.4242e+00, -1.9682e+00, -1.7536e+00, ..., -7.9952e-01,
-8.2349e-01, 1.1977e-01],
[-1.8053e-01, 8.0266e-02, -5.8106e-01, ..., -1.0719e+00,
-8.0162e-01, 2.5786e-01]]],
[[[-1.3109e+00, -1.7164e-01, -8.1223e-01, ..., 8.0096e-01,
9.9075e-01, 4.6001e-01],
[-2.3265e+00, 6.2293e-01, -7.1210e-01, ..., -3.1840e-01,
-2.0745e-01, -3.5217e-01],
[-1.4453e+00, -1.3399e+00, -1.1964e-01, ..., -6.8860e-01,
-2.7381e-01, 5.9790e-01],
...,
[ 3.1016e-01, 6.8094e-01, 6.0411e-01, ..., 2.3814e+00,
1.2012e+00, 1.7637e-01],
[-3.2371e-01, -8.3998e-01, 3.8397e-01, ..., 1.9388e+00,
1.1105e+00, -1.8791e-01],
[ 7.2513e-01, -3.5128e-01, -3.4225e-01, ..., -3.2631e+00,
-1.5613e+00, -6.2227e-01]],
[[ 8.3069e-01, -1.2570e-01, -1.6294e+00, ..., -1.4999e+00,
-2.1449e+00, -1.8249e+00],
[ 3.7266e-01, -2.0685e+00, -2.7145e+00, ..., -1.1101e+00,
-2.2045e+00, -2.1380e+00],
[-3.1652e+00, -2.1238e-01, -1.6088e+00, ..., -1.3663e+00,
-1.7950e+00, -1.4577e+00],
...,
[-1.0603e+00, -7.5253e-01, -8.4375e-01, ..., -3.3082e+00,
-4.7576e+00, -3.9238e+00],
[-3.0277e+00, -7.1427e-01, -1.0047e+00, ..., -6.3231e+00,
-6.5120e+00, -3.2145e+00],
[-2.0955e+00, -1.6501e+00, -1.6643e+00, ..., -4.1118e+00,
-2.4889e+00, -1.5777e+00]],
[[ 5.6214e-01, 4.8845e-01, 2.9140e-01, ..., 1.7488e+00,
8.1642e-01, 2.6018e+00],
[-4.2091e+00, -8.4739e-01, -4.0894e-01, ..., 1.4564e+00,
6.5138e-01, 2.6346e+00],
[-6.3600e+00, -7.4080e-01, -1.2993e+00, ..., 5.6568e-01,
1.0341e+00, 1.8655e+00],
...,
[ 3.0263e-01, 1.2103e+00, 7.0176e-01, ..., -3.0615e+00,
-5.1071e+00, 3.9173e-01],
[ 2.2661e-01, 2.3995e+00, 1.9445e+00, ..., -7.5130e+00,
-6.1219e+00, -4.3455e-01],
[ 1.3776e+00, 1.4455e+00, 1.8163e+00, ..., -2.8620e+00,
5.1282e-02, 1.2148e+00]],
...,
[[-8.2510e-01, -5.8598e+00, -1.2453e+00, ..., -5.1648e+00,
-3.7974e+00, -4.3894e+00],
[-2.8305e+00, -1.3036e+01, 3.0319e-01, ..., -5.2168e+00,
-3.9685e+00, -3.3494e+00],
[-1.4012e+00, -1.1992e+01, -3.0413e+00, ..., -3.5371e+00,
-2.6105e+00, -1.4033e+00],
...,
[-9.2278e-01, -2.2844e+00, -1.8748e+00, ..., -7.4411e+00,
-3.7174e+00, -7.1317e+00],
[-1.2876e+00, -3.5714e+00, -3.5798e+00, ..., -1.3072e+01,
-1.1547e+01, -9.2787e+00],
[-2.2150e+00, -4.9889e+00, -2.9689e+00, ..., -5.4830e+00,
-5.8358e+00, -4.6810e+00]],
[[ 1.0954e+00, 1.4399e+00, 1.5281e+00, ..., 1.4390e+00,
1.6673e+00, 1.1962e+00],
[ 5.9076e-02, -2.1429e+00, -1.5667e-01, ..., 1.6227e+00,
1.2251e+00, 1.8607e+00],
[-2.4366e-01, -2.1365e+00, -3.0533e-01, ..., 1.4153e+00,
6.0059e-01, 1.3159e+00],
...,
[ 1.1533e+00, -8.4291e-02, 1.1651e+00, ..., -2.7313e+00,
3.2898e-01, -1.0588e+00],
[ 1.8089e+00, 8.8978e-01, 1.4032e+00, ..., -1.5135e+00,
3.8902e-01, -4.1882e-01],
[ 1.0220e+00, 3.9786e-01, 1.0898e+00, ..., -7.8659e-01,
-1.3002e-02, 1.5338e+00]],
[[ 1.4249e+00, -1.1279e+00, -1.6084e+00, ..., -3.5116e-01,
-6.6103e-01, -2.8213e-02],
[ 1.3293e+00, -4.4226e+00, -4.6521e+00, ..., -1.2195e+00,
-1.4692e+00, -1.0779e+00],
[-6.0044e-01, -2.6118e+00, -3.4966e+00, ..., -6.6329e-01,
-1.2344e+00, -2.2413e-01],
...,
[-1.6845e+00, -9.6168e-01, -7.4515e-01, ..., 4.9064e-01,
2.2720e+00, -7.9270e-01],
[-1.6732e+00, -1.0157e+00, -7.7873e-01, ..., -1.5568e+00,
-3.8058e+00, -4.8505e+00],
[-2.9556e-01, 3.8554e-01, -4.6826e-01, ..., -2.4933e+00,
-3.1619e+00, -5.9035e-01]]],
[[[ 9.2415e-01, 1.5098e+00, 1.2994e+00, ..., 1.3481e+00,
1.0243e+00, 9.6576e-01],
[-9.9005e-01, -2.3137e+00, -1.1999e+00, ..., -3.4487e-01,
-5.4417e-01, -1.5106e+00],
[-1.6167e-01, -1.1478e+00, 2.9259e-02, ..., -7.1087e-02,
7.6557e-01, 8.6479e-01],
...,
[-2.4588e-02, -1.1571e+00, -9.8396e-01, ..., -4.1575e-01,
6.8326e-02, 2.8646e-01],
[-6.0328e-01, -1.8771e+00, -1.2880e+00, ..., -5.4401e-01,
-3.4210e-01, -1.5833e-01],
[-3.5673e-01, -1.7033e+00, -1.6321e+00, ..., -1.1672e+00,
-1.5950e+00, -2.8649e-01]],
[[-4.2215e+00, -6.0390e-01, -1.6968e+00, ..., -1.5139e+00,
-2.3130e+00, -1.0363e+00],
[-4.2046e+00, -2.4655e+00, -1.8796e+00, ..., -2.2735e+00,
-3.0240e+00, -8.1507e-01],
[-2.6178e+00, -1.5315e+00, -1.7028e+00, ..., -2.5276e+00,
-2.6769e+00, -9.6594e-01],
...,
[-1.6162e+00, -1.6393e+00, -2.3882e+00, ..., -2.3746e+00,
-1.9092e+00, -7.1484e-01],
[-3.0551e+00, -1.6821e+00, -2.7606e+00, ..., -2.2469e+00,
-2.2346e+00, -1.2655e+00],
[-2.2625e+00, -1.9133e+00, -2.6641e+00, ..., -2.0122e+00,
-1.9525e+00, -1.0305e+00]],
[[ 3.3380e+00, 4.4936e+00, 3.5910e+00, ..., 2.7739e+00,
9.7960e-01, 2.8402e+00],
[ 8.1770e-01, 1.0327e+00, 2.4997e+00, ..., 1.2060e+00,
-8.8469e-01, 2.5554e+00],
[ 1.6284e+00, 1.3998e+00, 1.8344e+00, ..., 3.0551e-01,
-2.4380e-01, 1.9950e+00],
...,
[ 9.9967e-01, 1.0297e+00, 1.1663e+00, ..., 5.1224e-01,
1.0988e+00, 2.3155e+00],
[ 6.6667e-01, 1.7524e+00, 1.1448e+00, ..., 1.3088e-01,
1.0170e+00, 1.7874e+00],
[ 1.6058e+00, 1.6338e+00, 1.8467e+00, ..., 1.2605e+00,
1.7675e+00, 1.7882e+00]],
...,
[[-2.7250e+00, -8.1004e+00, -5.9268e+00, ..., -3.9265e+00,
-3.0620e+00, -5.2693e+00],
[-3.3159e+00, -7.7025e+00, -4.5880e+00, ..., -3.5879e+00,
-2.6024e+00, -4.0379e+00],
[-1.2978e+00, -4.8435e+00, -3.6448e+00, ..., -2.9079e+00,
-2.8825e+00, -2.3550e+00],
...,
[-2.5149e+00, -5.3574e+00, -4.1346e+00, ..., -4.4542e+00,
-4.8239e+00, -4.7596e+00],
[-2.6985e+00, -5.6842e+00, -5.3308e+00, ..., -6.2914e+00,
-5.7941e+00, -6.2770e+00],
[-1.4400e+00, -4.0185e+00, -2.7718e+00, ..., -3.3037e+00,
-2.6272e+00, -3.1825e+00]],
[[-3.3973e-01, 4.8546e-01, 5.2883e-01, ..., 6.8178e-01,
9.1106e-01, 1.4619e+00],
[ 1.8940e+00, 1.3110e+00, 1.5924e+00, ..., 1.8621e+00,
2.5053e+00, 2.9473e+00],
[ 1.3930e+00, 4.5552e-01, 7.0614e-01, ..., 9.9230e-01,
1.6250e+00, 2.3012e+00],
...,
[ 1.2963e+00, 2.3074e-01, 2.9522e-02, ..., 1.3549e+00,
1.1704e+00, 1.6273e+00],
[ 1.8615e+00, 1.6430e+00, 1.0252e+00, ..., 1.6169e+00,
1.2743e+00, 1.9142e+00],
[ 4.8136e-01, 7.6670e-01, 6.9394e-01, ..., 4.8266e-01,
6.6699e-01, 1.6163e+00]],
[[-6.5265e-01, 3.3314e-01, -4.0913e-02, ..., 5.0421e-01,
5.1730e-01, 1.3690e+00],
[-1.5945e+00, -1.7899e+00, -1.7506e+00, ..., -6.7838e-01,
-4.1668e-01, 2.6222e-01],
[-2.5101e+00, -9.8124e-01, 3.7105e-01, ..., 1.5031e-01,
8.2628e-02, 7.3097e-01],
...,
[-1.9093e+00, -1.8497e+00, -1.4734e+00, ..., -9.8066e-01,
-8.3968e-01, -8.2716e-01],
[-1.5542e+00, -1.9967e+00, -1.7932e+00, ..., -7.7973e-01,
-1.0942e+00, -7.4131e-01],
[-7.9894e-01, -8.4472e-01, -1.1694e+00, ..., -6.4051e-01,
-9.1399e-01, -1.1966e-01]]]], device='cuda:0'),
tensor([[[[ 8.5045e-01, 8.8857e-01, 5.5589e-01, ..., 1.8319e-01,
8.6809e-02, -4.2628e-01],
[-5.6468e-01, -1.0651e+00, -5.0274e-01, ..., 8.3616e-01,
1.2506e+00, -2.8986e-01],
[ 3.7116e-02, -2.0862e-01, -6.6538e-01, ..., 2.0090e-01,
8.0674e-01, 9.3269e-01],
...,
[ 2.7966e-01, -7.3232e-01, -5.5883e-01, ..., -8.4998e-01,
7.2304e-01, 5.6321e-01],
[-5.5653e-01, -1.7467e+00, -1.2637e+00, ..., -1.4165e+00,
-3.4325e-01, 3.2031e-02],
[-2.6863e-01, -1.6471e+00, -1.6198e+00, ..., -5.9802e-01,
-1.7128e+00, -2.2574e-01]],
[[-3.3377e+00, -1.3685e+00, -2.6495e+00, ..., -2.7932e+00,
-3.2189e+00, -7.7564e-01],
[-2.9517e+00, -2.1716e+00, -2.1929e+00, ..., -1.5504e+00,
-1.9758e+00, -1.7610e-01],
[-1.8334e+00, -1.5046e+00, -2.6980e+00, ..., -1.8875e+00,
-1.5379e+00, -5.1432e-01],
...,
[-1.4786e+00, -1.3075e+00, -1.9421e+00, ..., -4.5034e+00,
-4.4795e+00, -6.4985e-01],
[-2.9570e+00, -1.5830e+00, -2.7817e+00, ..., -4.8805e+00,
-3.7242e+00, -8.9549e-01],
[-2.1941e+00, -2.0089e+00, -2.7681e+00, ..., -2.9597e+00,
-2.0407e+00, -9.6068e-01]],
[[ 2.2359e+00, 2.6056e+00, 2.5553e+00, ..., 1.8261e+00,
3.8126e-01, 2.1390e+00],
[ 4.2686e-01, 3.7507e-01, 2.6445e+00, ..., 1.7275e+00,
-2.0684e-01, 2.0871e+00],
[ 1.0140e+00, 9.1017e-01, 2.4099e+00, ..., 9.0175e-01,
8.2269e-01, 2.2887e+00],
...,
[ 1.2680e+00, 1.4025e+00, 1.6682e+00, ..., -1.8781e+00,
8.5404e-01, 1.0660e+00],
[ 7.3094e-01, 1.8301e+00, 1.1918e+00, ..., -2.2986e+00,
1.2908e+00, 1.1951e+00],
[ 1.6027e+00, 1.6836e+00, 1.8339e+00, ..., 1.1862e+00,
1.6063e+00, 1.3941e+00]],
...,
[[-1.9548e+00, -6.2996e+00, -4.5985e+00, ..., -3.5155e+00,
-2.6733e+00, -3.8184e+00],
[-2.5604e+00, -7.5337e+00, -4.9085e+00, ..., -4.5395e+00,
-4.3029e+00, -4.9711e+00],
[-1.5376e+00, -4.4258e+00, -2.7176e+00, ..., -2.7912e+00,
-2.2616e+00, -2.0481e+00],
...,
[-2.2123e+00, -4.6886e+00, -3.6545e+00, ..., -3.7557e+00,
-7.3583e+00, -6.6222e+00],
[-2.5966e+00, -5.4979e+00, -5.2882e+00, ..., -6.5433e+00,
-6.4224e+00, -7.2128e+00],
[-1.4413e+00, -4.0782e+00, -2.8670e+00, ..., -4.2914e+00,
-3.7101e+00, -4.0790e+00]],
[[-3.5864e-01, 2.6798e-01, 1.4683e-01, ..., 8.1511e-01,
9.1706e-01, 1.5967e+00],
[ 1.1134e+00, 4.3216e-01, 1.1009e-01, ..., 1.3995e+00,
1.7524e+00, 2.4554e+00],
[ 1.0350e+00, 7.9702e-02, -2.4707e-01, ..., 1.3146e+00,
1.2726e+00, 1.7443e+00],
...,
[ 1.1465e+00, 3.4783e-01, 7.3415e-02, ..., 9.1729e-01,
7.4979e-01, 2.2307e+00],
[ 1.8182e+00, 1.6485e+00, 1.0436e+00, ..., 5.8341e-01,
8.6173e-01, 2.3173e+00],
[ 4.5116e-01, 6.6243e-01, 6.8486e-01, ..., 8.2952e-01,
1.1062e+00, 1.7678e+00]],
[[-6.3096e-01, -2.5473e-01, -7.8791e-01, ..., -5.9324e-01,
-4.4990e-01, 6.1959e-01],
[-1.2082e+00, -1.1759e+00, -1.1379e+00, ..., -2.1249e-01,
3.9238e-01, 8.1889e-01],
[-1.8182e+00, -1.9473e-01, -1.5765e-01, ..., 4.1078e-01,
4.1629e-01, -4.8072e-02],
...,
[-1.8440e+00, -1.7706e+00, -1.0686e+00, ..., -1.2450e+00,
-2.2349e+00, -1.1535e+00],
[-1.4530e+00, -1.9628e+00, -1.6842e+00, ..., -2.9477e+00,
-2.2976e+00, -6.5045e-01],
[-7.2250e-01, -8.6700e-01, -1.1824e+00, ..., -2.3478e-01,
-2.1062e-01, 1.6397e-01]]],
[[[ 9.5628e-01, 1.6116e+00, 1.1266e+00, ..., 8.0981e-01,
1.0202e+00, 4.5655e-01],
[-8.1223e-01, -1.5517e+00, -5.2396e-01, ..., -3.0023e-01,
-1.4852e-01, -3.2242e-01],
[-1.7502e-01, -4.7399e-01, -1.2034e-01, ..., -6.3441e-01,
-1.6230e-01, 6.3107e-01],
...,
[ 4.0238e-01, -2.4125e-01, -6.4046e-02, ..., 4.1946e-01,
-1.4676e-01, 3.4835e-01],
[-2.2599e-01, -1.1058e+00, 1.3103e-03, ..., -5.0960e-03,
1.2209e-01, -9.3372e-02],
[ 7.4251e-01, -3.1063e-01, -2.1921e-01, ..., -3.3632e-01,
-9.0333e-01, 1.5299e-01]],
[[-4.0465e+00, -1.2627e+00, -2.1155e+00, ..., -1.3869e+00,
-2.0220e+00, -1.8012e+00],
[-3.9317e+00, -2.8666e+00, -2.5012e+00, ..., -1.0493e+00,
-2.1303e+00, -2.1592e+00],
[-2.8329e+00, -2.5175e+00, -3.8513e+00, ..., -1.4267e+00,
-1.7160e+00, -1.4031e+00],
...,
[-9.4938e-01, -6.6707e-01, -1.1532e+00, ..., -1.2478e+00,
-1.0132e+00, -8.2609e-01],
[-2.9452e+00, -4.1035e-01, -7.0187e-01, ..., -6.3627e-01,
-1.1046e+00, -1.2410e+00],
[-2.0779e+00, -1.5159e+00, -1.4943e+00, ..., -6.9485e-01,
-7.0667e-01, -5.3379e-01]],
[[ 2.8671e+00, 4.0086e+00, 3.2068e+00, ..., 1.6761e+00,
8.0277e-01, 2.5541e+00],
[ 3.2747e-01, 1.0825e+00, 2.6291e+00, ..., 1.4601e+00,
6.7058e-01, 2.6012e+00],
[ 1.2799e+00, 1.7029e+00, 2.2773e+00, ..., 4.2870e-01,
9.6527e-01, 1.7942e+00],
...,
[ 2.8894e-01, 9.7473e-01, 9.9486e-01, ..., 4.0504e-01,
5.9189e-01, 1.2424e+00],
[ 3.8890e-01, 2.0912e+00, 1.3970e+00, ..., -1.9762e-01,
7.6109e-01, 1.1891e+00],
[ 1.4603e+00, 1.4059e+00, 1.6253e+00, ..., 5.5819e-01,
1.3432e+00, 1.3954e+00]],
...,
[[-1.8513e+00, -7.0745e+00, -5.7115e+00, ..., -4.9352e+00,
-3.5307e+00, -4.2077e+00],
[-2.7332e+00, -6.7683e+00, -4.9646e+00, ..., -4.8677e+00,
-3.6143e+00, -3.1233e+00],
[-6.7725e-01, -2.5584e+00, -2.7554e+00, ..., -3.5037e+00,
-2.2647e+00, -1.2855e+00],
...,
[-9.2777e-01, -2.2846e+00, -1.9216e+00, ..., -6.0155e+00,
-4.8066e+00, -3.6036e+00],
[-1.2730e+00, -3.7011e+00, -3.5272e+00, ..., -5.1530e+00,
-4.1923e+00, -4.3632e+00],
[-2.2089e+00, -4.9522e+00, -2.9655e+00, ..., -3.3460e+00,
-2.4561e+00, -2.4336e+00]],
[[-7.3855e-01, -1.4885e-01, 2.2099e-01, ..., 1.4719e+00,
1.7007e+00, 1.1912e+00],
[ 1.7479e+00, 8.9310e-01, 1.5087e+00, ..., 1.5743e+00,
1.1609e+00, 1.8131e+00],
[ 8.3635e-01, -6.0543e-01, -7.8143e-01, ..., 1.2168e+00,
5.4054e-01, 1.2831e+00],
...,
[ 1.2810e+00, -1.6603e-01, 1.0078e+00, ..., 2.0715e+00,
1.0871e+00, 1.1161e+00],
[ 1.8969e+00, 1.0525e+00, 1.8494e+00, ..., 1.2553e+00,
7.3144e-01, 1.4153e+00],
[ 1.0025e+00, 3.7255e-01, 9.1284e-01, ..., 7.0774e-01,
6.6067e-01, 1.4269e+00]],
[[-5.2804e-01, 5.8148e-01, 1.2514e-01, ..., -3.3758e-01,
-5.9325e-01, -6.8857e-02],
[-9.9697e-01, -9.9383e-01, -1.2634e+00, ..., -1.2858e+00,
-1.4349e+00, -1.1216e+00],
[-2.2047e+00, -4.8280e-01, -3.6281e-01, ..., -7.4300e-01,
-1.1298e+00, -2.0760e-01],
...,
[-1.8255e+00, -1.4511e+00, -9.6743e-01, ..., -9.0223e-01,
-9.7666e-01, -6.0870e-01],
[-1.8520e+00, -1.1000e+00, -7.9268e-01, ..., -7.2536e-01,
-1.6299e+00, -7.8671e-01],
[-3.8891e-01, 3.4710e-01, -3.4149e-01, ..., -2.8281e-01,
-7.9883e-01, 1.5982e-03]]],
[[[ 6.1079e-01, 1.0970e+00, 8.8944e-01, ..., -3.0164e+00,
5.0355e-01, 6.9053e-01],
[-9.4667e-01, -1.8126e+00, 1.4907e-01, ..., -3.8509e+00,
-1.2843e+00, 9.8956e-01],
[ 3.7168e-01, -1.0497e+00, 7.7868e-02, ..., -8.4348e-01,
-2.3463e+00, -3.4453e-01],
...,
[ 4.9132e-01, 5.7586e-01, 1.1919e+00, ..., 7.4262e-01,
7.1401e-01, 7.9889e-01],
[-5.9804e-02, 4.7236e-02, 1.0243e+00, ..., 1.9028e+00,
1.4333e+00, 7.4867e-02],
[ 7.7706e-01, 8.2402e-02, 1.3801e+00, ..., 7.1547e-01,
4.6119e-01, 7.3303e-01]],
[[-3.3640e+00, -7.9447e-01, -2.0194e+00, ..., -3.9622e+00,
-5.0674e+00, -2.5555e+00],
[-2.8361e+00, 1.2678e-01, 4.4539e-01, ..., -4.3996e+00,
-5.0111e+00, -1.3043e+00],
[-3.8556e-01, 1.2062e+00, 5.5211e-01, ..., -3.7290e+00,
-1.3079e+00, 2.7805e-02],
...,
[-6.1786e-01, 6.7645e-01, 1.0855e+00, ..., -1.0398e+00,
-6.7537e-01, 8.3186e-01],
[-2.0742e+00, 5.2094e-01, 1.7186e+00, ..., -1.2901e+00,
-6.6361e-01, 7.6152e-01],
[-1.3952e+00, -1.3539e+00, -1.5946e+00, ..., -1.8886e+00,
-8.6145e-01, -1.8614e-01]],
[[ 2.8352e+00, 2.6399e+00, 2.1597e+00, ..., -3.0843e+00,
-3.2477e+00, 4.6096e-01],
[ 4.3863e-01, -8.0700e-01, -2.2191e-01, ..., -2.9204e+00,
-3.3307e+00, -6.3910e-01],
[ 4.8842e-01, -1.1450e-01, 2.1783e-01, ..., -7.6276e-01,
-3.1011e+00, -1.7548e+00],
...,
[ 7.3800e-01, 9.7128e-01, -1.4319e+00, ..., -4.5605e-01,
-2.3444e-01, 1.8387e+00],
[ 8.3928e-01, 2.2516e+00, -4.0345e-01, ..., -2.1558e+00,
-1.2735e-01, 1.7393e+00],
[ 2.1547e+00, 2.4255e+00, 2.7799e+00, ..., -1.7681e+00,
8.4652e-01, 1.8783e+00]],
...,
[[-2.8720e+00, -6.2790e+00, -4.1230e+00, ..., 6.1865e-01,
-1.1412e+00, -3.4452e+00],
[-3.4614e+00, -7.0765e+00, -4.3804e+00, ..., 3.0647e-01,
-1.2044e+00, -4.3322e+00],
[-2.0699e+00, -4.6264e+00, -9.3340e-01, ..., -9.4891e-01,
7.8969e-01, -1.4842e+00],
...,
[-2.0912e+00, -4.3833e+00, -1.4297e+00, ..., -3.0814e+00,
-3.4926e+00, -2.9773e+00],
[-2.0197e+00, -4.6867e+00, -2.6457e+00, ..., -9.6854e+00,
-4.0624e+00, -5.0449e+00],
[-2.2032e+00, -5.2606e+00, -2.0688e+00, ..., -4.2589e+00,
-2.0876e+00, -3.4757e+00]],
[[-4.3366e-01, -7.4795e-02, -4.7569e-01, ..., -1.2537e+00,
-4.4943e+00, -6.5255e-01],
[ 1.7537e+00, 1.2704e+00, 6.7538e-01, ..., -2.3109e+00,
-3.1884e+00, -1.3298e-01],
[ 1.5748e+00, 1.3348e+00, 8.5272e-01, ..., -4.1015e+00,
-2.0342e+00, -3.3695e-02],
...,
[ 1.4549e+00, 8.4770e-01, 1.2050e+00, ..., 8.9887e-01,
1.8970e+00, 1.5082e+00],
[ 1.9428e+00, 1.4688e+00, 2.9537e+00, ..., 1.4211e+00,
1.9526e+00, 2.5886e+00],
[ 1.2305e+00, 4.9214e-01, 1.0973e+00, ..., 8.9175e-01,
1.5778e+00, 1.5881e+00]],
[[-1.1070e+00, -7.8721e-02, -3.0334e-01, ..., -5.7653e-03,
5.1659e-01, -1.2339e-01],
[-1.7063e+00, -4.6255e-01, -6.6272e-01, ..., -1.9602e+00,
-9.2065e-01, 6.5673e-01],
[-1.1733e+00, 6.0213e-02, -1.1616e-01, ..., -3.0440e+00,
-9.2652e-01, 5.1621e-01],
...,
[-1.7880e+00, -7.1105e-01, -1.3784e-01, ..., 1.0384e+00,
-1.5614e+00, -6.0845e-01],
[-1.5451e+00, -7.2501e-01, -8.8864e-01, ..., 5.7675e-01,
-2.4091e+00, 3.6661e-01],
[-5.7804e-02, 5.9109e-01, 3.0422e-01, ..., -1.6467e+00,
-6.8618e-01, 6.3743e-01]]],
[[[ 9.7630e-01, 1.5963e+00, 1.1594e+00, ..., 7.0026e-01,
1.0757e+00, 4.7971e-01],
[-8.6638e-01, -2.0167e+00, -9.0864e-01, ..., -5.8163e-01,
-1.5044e-01, -2.9880e-01],
[-4.2341e-02, -8.1520e-01, -2.9429e-01, ..., -8.1373e-01,
-1.9551e-01, 5.8431e-01],
...,
[ 4.5013e-01, -2.9701e-01, -3.1309e-01, ..., -3.0468e-02,
-6.2464e-02, 3.9025e-01],
[-2.2207e-01, -1.0802e+00, 7.9357e-02, ..., -4.4375e-02,
1.0689e-01, -7.6315e-02],
[ 7.2885e-01, -3.4512e-01, 1.9426e-01, ..., -3.1011e-01,
-8.4603e-01, 1.9320e-01]],
[[-4.3145e+00, -1.2948e+00, -2.3451e+00, ..., -1.3861e+00,
-1.9158e+00, -1.6946e+00],
[-4.1490e+00, -2.8149e+00, -2.2596e+00, ..., -1.9101e+00,
-1.8243e+00, -1.9909e+00],
[-2.7332e+00, -1.8213e+00, -2.7421e+00, ..., -1.8302e+00,
-1.3323e+00, -1.2924e+00],
...,
[-8.2585e-01, -6.5092e-01, -1.2071e+00, ..., -5.3026e-01,
-8.6154e-01, -8.0107e-01],
[-2.9235e+00, -5.3729e-01, -6.9911e-01, ..., -5.4518e-01,
-1.0579e+00, -1.1371e+00],
[-2.1080e+00, -1.5457e+00, -1.3505e+00, ..., -6.1429e-01,
-7.0763e-01, -4.7871e-01]],
[[ 3.0233e+00, 4.1089e+00, 3.4471e+00, ..., 1.3522e+00,
7.9667e-01, 2.4604e+00],
[ 3.8256e-01, 8.2286e-01, 2.5875e+00, ..., 1.4868e-01,
5.3436e-01, 2.4748e+00],
[ 8.7678e-01, 1.2058e+00, 1.9084e+00, ..., -7.6018e-01,
8.6649e-01, 1.6528e+00],
...,
[ 2.9517e-01, 1.0687e+00, 1.9069e+00, ..., -2.5250e-01,
4.5607e-01, 1.4104e+00],
[ 3.4622e-01, 2.2397e+00, 1.8174e+00, ..., -1.7281e-01,
7.2262e-01, 1.1161e+00],
[ 1.4147e+00, 1.4211e+00, 2.1797e+00, ..., 5.8949e-01,
1.2584e+00, 1.2973e+00]],
...,
[[-2.1279e+00, -7.5029e+00, -6.1230e+00, ..., -5.4843e+00,
-3.4433e+00, -4.0418e+00],
[-2.7462e+00, -6.6257e+00, -4.8497e+00, ..., -6.8650e+00,
-3.4251e+00, -3.0216e+00],
[-1.7571e-01, -2.7069e+00, -3.0393e+00, ..., -6.0926e+00,
-1.9544e+00, -1.2406e+00],
...,
[-9.6301e-01, -2.3737e+00, -2.2068e+00, ..., -3.9302e+00,
-3.0511e+00, -2.6145e+00],
[-1.3241e+00, -3.8339e+00, -3.8312e+00, ..., -5.0357e+00,
-4.0512e+00, -4.1591e+00],
[-2.2554e+00, -5.0999e+00, -3.1847e+00, ..., -3.3587e+00,
-2.4323e+00, -2.3717e+00]],
[[-7.7401e-01, -1.1757e-01, -2.1507e-02, ..., 2.2840e+00,
1.6673e+00, 1.1341e+00],
[ 1.7698e+00, 1.1001e+00, 1.2224e+00, ..., 2.2674e+00,
1.2099e+00, 1.7388e+00],
[ 1.1728e+00, 2.1234e-01, 1.1151e-01, ..., 2.3453e+00,
6.4628e-01, 1.2086e+00],
...,
[ 1.2798e+00, -1.7863e-01, 4.6375e-01, ..., 1.4479e+00,
6.3436e-01, 8.7261e-01],
[ 1.8729e+00, 9.0873e-01, 1.4419e+00, ..., 1.2845e+00,
7.0265e-01, 1.4192e+00],
[ 1.0184e+00, 3.8011e-01, 8.8022e-01, ..., 7.7083e-01,
6.8034e-01, 1.4101e+00]],
[[-5.2959e-01, 6.1586e-01, -6.2949e-02, ..., -2.0118e-01,
-5.2041e-01, -5.9774e-02],
[-1.0679e+00, -1.0827e+00, -1.7112e+00, ..., -1.7332e+00,
-1.3080e+00, -1.1650e+00],
[-2.0667e+00, -5.2974e-01, -5.2009e-01, ..., -1.8363e+00,
-9.9418e-01, -3.8165e-01],
...,
[-1.7493e+00, -1.5344e+00, -1.1110e+00, ..., -8.4139e-01,
-1.5366e+00, -8.5507e-01],
[-1.7917e+00, -1.1257e+00, -8.0234e-01, ..., -7.9948e-01,
-1.7071e+00, -6.9646e-01],
[-3.9005e-01, 4.1351e-01, 5.8272e-03, ..., -2.5213e-01,
-8.1922e-01, 5.5079e-03]]],
[[[ 8.3167e-01, 1.4547e+00, 1.1870e+00, ..., 1.3819e+00,
1.0644e+00, 9.4864e-01],
[-9.9217e-01, -2.0909e+00, -1.1712e+00, ..., -3.9263e-01,
-5.7149e-01, -1.6045e+00],
[-1.5357e-01, -7.6111e-01, 1.9835e-01, ..., -8.7667e-01,
5.2906e-01, 8.5781e-01],
...,
[-2.3805e-02, -1.2380e+00, -1.4557e+00, ..., -4.1380e-01,
-3.8639e-02, 2.2012e-01],
[-6.1635e-01, -1.9110e+00, -1.5084e+00, ..., -5.5189e-01,
-3.1586e-01, -1.5400e-01],
[-9.7921e-02, -1.7868e+00, -1.2190e+00, ..., -1.0662e+00,
-1.6008e+00, -2.9809e-01]],
[[-3.8028e+00, -7.3547e-01, -1.5493e+00, ..., -1.5652e+00,
-2.4616e+00, -1.2143e+00],
[-4.0115e+00, -2.7995e+00, -1.8057e+00, ..., -2.1759e+00,
-3.2291e+00, -9.6650e-01],
[-2.3140e+00, -1.5758e+00, -1.6785e+00, ..., -2.1325e+00,
-2.8444e+00, -1.0742e+00],
...,
[-1.5435e+00, -1.4276e+00, -2.5301e+00, ..., -1.9239e+00,
-2.1970e+00, -7.4980e-01],
[-3.4019e+00, -1.3994e+00, -2.1713e+00, ..., -2.1573e+00,
-2.4750e+00, -1.2910e+00],
[-2.3818e+00, -1.8540e+00, -2.1112e+00, ..., -1.9089e+00,
-2.0855e+00, -1.0154e+00]],
[[ 2.9291e+00, 4.2844e+00, 3.3145e+00, ..., 2.8507e+00,
1.0296e+00, 3.0526e+00],
[ 2.7665e-01, 1.1029e+00, 2.1925e+00, ..., 9.4359e-01,
-1.0374e+00, 2.7614e+00],
[ 1.0040e+00, 1.4681e+00, 1.5325e+00, ..., 9.2185e-01,
-2.0338e-01, 2.0635e+00],
...,
[ 9.1088e-01, 1.0048e+00, 2.1850e+00, ..., 6.0952e-01,
1.0422e+00, 2.3080e+00],
[ 4.9781e-01, 2.0353e+00, 2.1358e+00, ..., 3.4316e-01,
1.0593e+00, 1.8376e+00],
[ 1.6871e+00, 1.7114e+00, 2.3626e+00, ..., 1.5537e+00,
1.8348e+00, 1.7685e+00]],
...,
[[-1.9156e+00, -7.2568e+00, -5.3846e+00, ..., -3.9689e+00,
-3.2268e+00, -5.5560e+00],
[-2.4439e+00, -6.9576e+00, -4.1285e+00, ..., -3.5338e+00,
-2.6622e+00, -4.3720e+00],
[-9.7865e-01, -4.6045e+00, -3.6537e+00, ..., -2.5131e+00,
-3.0999e+00, -2.4851e+00],
...,
[-2.6567e+00, -5.4133e+00, -4.2987e+00, ..., -4.3482e+00,
-4.5926e+00, -4.5675e+00],
[-2.9591e+00, -5.9885e+00, -5.4994e+00, ..., -6.4176e+00,
-5.8669e+00, -6.1241e+00],
[-1.8989e+00, -4.6566e+00, -2.5858e+00, ..., -2.7594e+00,
-2.8608e+00, -3.2032e+00]],
[[-1.8033e-01, 3.5451e-01, 5.4312e-01, ..., 6.4673e-01,
8.4273e-01, 1.4132e+00],
[ 1.6236e+00, 1.1076e+00, 1.8501e+00, ..., 1.5336e+00,
2.3718e+00, 3.0332e+00],
[ 1.2120e+00, 2.8649e-01, 7.1311e-01, ..., 1.8795e+00,
1.5389e+00, 2.3571e+00],
...,
[ 1.3943e+00, 3.0600e-01, -2.2224e-01, ..., 1.4104e+00,
1.0642e+00, 1.6410e+00],
[ 1.9173e+00, 1.9110e+00, 1.3457e+00, ..., 1.7197e+00,
1.2622e+00, 1.9736e+00],
[ 3.8288e-01, 8.3634e-01, 1.3122e+00, ..., 9.5701e-01,
7.0058e-01, 1.6594e+00]],
[[-5.7300e-01, 2.7392e-01, 8.6441e-02, ..., 5.9297e-01,
5.2722e-01, 1.3619e+00],
[-1.4102e+00, -1.8445e+00, -1.6211e+00, ..., -9.0288e-01,
-5.2840e-01, 2.4777e-01],
[-2.2273e+00, -8.7665e-01, 2.2891e-01, ..., 1.9398e-02,
4.9452e-02, 7.8830e-01],
...,
[-1.9812e+00, -2.1530e+00, -1.8970e+00, ..., -8.5809e-01,
-1.1844e+00, -7.7730e-01],
[-1.5313e+00, -2.2615e+00, -2.0044e+00, ..., -9.4304e-01,
-1.4079e+00, -6.8227e-01],
[-7.8193e-01, -8.9349e-01, -7.3784e-01, ..., -6.5528e-01,
-9.3705e-01, -1.0107e-01]]]], device='cuda:0'),
tensor([[[[ 1.2494e+00, 1.0125e+00, -4.4752e+00, ..., 3.1410e+00,
7.9543e-01, -5.1433e-01],
[-2.0273e-01, -1.4436e+00, -6.4742e+00, ..., 5.6700e-02,
9.2872e-01, -9.4203e-02],
[-6.3866e-01, -1.6472e+00, -5.7949e+00, ..., 2.2167e-01,
7.2252e-01, 7.9656e-01],
...,
[ 3.9077e-01, -5.1101e-01, -8.0292e-01, ..., 5.6965e-01,
2.0820e+00, 9.8693e-01],
[-6.1678e-01, -1.7137e+00, -1.7230e+00, ..., -2.2677e+00,
-1.8623e+00, -5.5830e-01],
[-2.3933e-01, -1.7259e+00, -1.9861e+00, ..., -6.2343e-01,
-1.4160e+00, -6.3923e-01]],
[[-3.0457e+00, -2.0339e-02, 1.6420e+00, ..., -5.0491e+00,
-4.2954e+00, -1.1964e+00],
[-3.8280e+00, -2.4937e+00, -5.8301e-01, ..., -6.6952e+00,
-2.2187e+00, -1.8036e-01],
[-1.0755e+00, -3.4835e+00, -2.2416e+00, ..., -3.5796e+00,
-2.0080e+00, -5.6340e-01],
...,
[-1.2959e+00, -1.1940e+00, -7.9055e-01, ..., -7.3178e+00,
-5.1640e+00, -2.2379e+00],
[-2.8172e+00, -1.5488e+00, -2.5458e+00, ..., -5.4418e+00,
-4.1161e+00, -1.3526e+00],
[-2.2245e+00, -1.9172e+00, -3.0031e+00, ..., -2.2563e+00,
-2.0674e+00, -9.9534e-01]],
[[ 1.9721e+00, -1.0648e+00, -1.6105e+00, ..., -4.7656e-01,
-6.3420e-01, 2.3968e+00],
[ 1.2963e-01, -9.3592e+00, -1.6343e+00, ..., 3.2701e-01,
-1.7922e+00, 2.4240e+00],
[ 2.0028e-01, -8.1929e+00, -2.2336e+00, ..., 2.1119e+00,
-1.0164e-01, 2.2596e+00],
...,
[ 1.3549e+00, 1.4162e+00, 2.0987e+00, ..., -1.0568e+01,
-5.2430e+00, 3.4323e-01],
[ 7.8081e-01, 1.9220e+00, 1.8939e+00, ..., -4.5857e+00,
-3.8909e-01, 1.9178e+00],
[ 1.8009e+00, 1.5906e+00, 1.8702e+00, ..., 1.1755e+00,
1.7916e+00, 1.3766e+00]],
...,
[[-3.3245e+00, -6.1105e+00, -3.6232e+00, ..., -1.6244e+01,
-9.7012e+00, -3.8297e+00],
[-5.7593e+00, -8.0650e+00, -6.1825e+00, ..., -1.4679e+01,
-8.4671e+00, -5.4717e+00],
[-2.5353e+00, -4.6277e+00, -2.0886e+00, ..., -7.4659e+00,
-4.2945e+00, -2.3303e+00],
...,
[-2.2814e+00, -5.2555e+00, -3.6318e+00, ..., -1.1405e+01,
-1.1551e+01, -6.6491e+00],
[-2.9202e+00, -6.2194e+00, -5.3244e+00, ..., -8.0855e+00,
-7.8095e+00, -7.4420e+00],
[-1.6983e+00, -4.4587e+00, -2.8778e+00, ..., -6.2671e+00,
-5.5174e+00, -4.8149e+00]],
[[ 5.3886e-01, -2.4132e+00, -8.9578e-01, ..., -2.7392e+00,
1.2456e+00, 1.4956e+00],
[ 2.1670e+00, 6.0012e-01, -6.0464e-01, ..., -2.2273e+00,
2.0091e+00, 2.6419e+00],
[ 1.4501e+00, 1.7414e-01, -3.9317e+00, ..., 5.6576e-01,
1.5314e+00, 1.9468e+00],
...,
[ 1.3425e+00, 3.3105e-01, 3.1131e-02, ..., -1.5617e+00,
-1.0048e+00, 8.5262e-01],
[ 1.9994e+00, 1.7707e+00, 5.9417e-01, ..., -4.4275e-01,
4.8717e-01, 1.8665e+00],
[ 5.2103e-01, 8.9727e-01, 3.9221e-01, ..., 1.1389e+00,
1.0977e+00, 1.8236e+00]],
[[-1.7082e+00, 1.7823e+00, 1.3214e+00, ..., 6.7182e-01,
-6.9312e-01, 4.6937e-01],
[-2.2327e+00, 7.3315e-02, 7.4214e-01, ..., -1.6864e+00,
-2.4317e+00, 8.8119e-01],
[-2.4221e+00, -2.1258e+00, -2.1632e+00, ..., 7.4913e-01,
-1.2517e+00, 3.0265e-02],
...,
[-1.9583e+00, -1.8883e+00, -2.7612e-01, ..., -2.1669e+00,
-2.1041e+00, -1.7212e+00],
[-1.6767e+00, -2.1288e+00, -1.7131e+00, ..., -2.4771e+00,
-2.6185e+00, -2.4872e-01],
[-8.3032e-01, -8.9247e-01, -1.4291e+00, ..., -1.3096e-01,
-2.8494e-01, 3.1829e-01]]],
[[[-5.9208e+00, -1.2597e+01, -1.8572e+01, ..., 5.8608e-01,
9.2485e-01, -3.5789e-01],
[-6.9956e+00, -1.3250e+01, -1.9594e+01, ..., 5.4197e-01,
1.4384e+00, 5.4697e-02],
[-3.2066e+00, -2.3704e+00, -1.0514e+01, ..., 3.6634e-01,
1.1893e+00, 1.0660e+00],
...,
[ 8.0277e-01, 8.4901e-02, 7.2871e-02, ..., 2.9869e+00,
-7.6059e-01, -3.3395e+00],
[-5.5359e-02, 3.0442e-01, 3.9466e-01, ..., -3.8293e-01,
2.2607e+00, 4.4061e+00],
[-1.8958e-01, 5.2895e-02, 1.1406e-01, ..., -1.7241e+00,
-5.1244e+00, -6.3148e+00]],
[[-1.2588e+00, -1.1220e+01, -1.6314e+01, ..., -2.7685e+00,
-4.3240e+00, -1.8055e+00],
[-2.4288e+00, -1.2576e+01, -1.7201e+01, ..., -2.4111e+00,
-2.4521e+00, -2.3568e-01],
[-2.0433e+00, -1.3870e+01, -1.4766e+01, ..., -2.5433e+00,
-1.5529e+00, -8.2656e-01],
...,
[ 9.1562e-02, -8.9243e-01, -6.6480e-01, ..., -1.1304e+01,
-1.1894e+01, -6.6943e+00],
[-1.5879e+00, 1.0736e-01, 1.1330e+00, ..., -7.1526e+00,
-3.9028e+00, -2.6492e+00],
[-1.2191e+00, -4.1951e-01, -6.6475e-01, ..., -1.5163e+01,
-1.5727e+01, -6.6085e+00]],
[[ 2.5171e-01, -7.9913e+00, -1.3248e+01, ..., 1.7440e+00,
1.3593e+00, 2.8946e+00],
[-1.1139e+00, -9.1382e+00, -9.3666e+00, ..., 1.4381e-01,
-2.3603e-01, 2.3390e+00],
[-8.5095e-01, -9.1171e-01, -5.8166e+00, ..., -3.7416e-01,
6.8103e-01, 1.3971e+00],
...,
[ 6.5434e-01, 1.3131e-01, -1.1029e+00, ..., -4.9326e+00,
-7.1002e+00, -7.4813e+00],
[-2.8735e-01, 6.1597e-01, -6.6334e-01, ..., -1.5623e+01,
-1.6690e+01, -9.4867e+00],
[ 1.1077e+00, 4.1916e-01, 2.0591e-01, ..., -6.5136e+00,
-1.4267e+01, -6.6636e+00]],
...,
[[-5.7876e+00, -8.1524e+00, -9.2897e+00, ..., -3.9639e-01,
-2.7581e+00, -4.4296e+00],
[-6.6200e+00, -1.2171e+01, -9.5919e+00, ..., -1.9570e+00,
-4.2895e+00, -6.2014e+00],
[-2.9019e+00, -1.7025e+01, -1.9315e+00, ..., -1.5338e+00,
-1.8112e+00, -2.6762e+00],
...,
[-2.7746e+00, -5.1389e+00, -5.2771e-01, ..., -1.0855e+01,
-4.8541e+00, -5.8420e+00],
[-3.8819e+00, -7.6906e+00, -2.1384e+00, ..., -1.2650e+01,
-1.5735e+01, -2.6936e+01],
[-3.0235e+00, -6.4529e+00, -1.6236e+00, ..., -2.4624e+00,
-4.5620e+00, -1.0384e+01]],
[[ 2.9517e+00, 3.3934e+00, -4.2206e-01, ..., -1.4803e+00,
-4.3075e-01, 1.4553e+00],
[-1.6531e+00, -3.7279e+00, -1.3541e+01, ..., -1.2408e-01,
1.4940e+00, 2.7468e+00],
[-8.2377e+00, -2.3916e+01, -2.7167e+01, ..., -5.8474e-01,
1.3937e+00, 1.8083e+00],
...,
[ 1.3674e+00, 1.0783e-01, -1.0426e-01, ..., -1.9482e+01,
-1.7034e+01, -5.9813e+00],
[ 1.9341e+00, 1.7507e+00, 2.1299e+00, ..., -4.3500e+00,
-2.1908e+00, -3.4105e+00],
[ 7.3351e-01, 1.1394e+00, 1.3573e+00, ..., -2.7842e+00,
-1.9309e+00, -1.4572e+00]],
[[-4.9510e-01, -3.7870e+00, -2.2871e+00, ..., 2.5965e-01,
-3.8065e-02, 4.9892e-01],
[-3.7001e+00, -1.7752e+01, -1.5236e+01, ..., 5.5648e-01,
3.0405e-01, 9.8150e-01],
[-1.6754e+00, -1.3322e+01, -1.2544e+01, ..., 4.4672e-01,
9.9473e-01, 3.6925e-01],
...,
[-1.2166e+00, -2.2464e+00, -3.8615e-01, ..., -4.6133e+00,
-2.6658e+00, -5.1456e-01],
[-1.6346e+00, -2.8269e+00, -2.1975e-01, ..., -8.3897e+00,
-4.0187e+00, 3.8448e-01],
[-6.8892e-01, -1.4396e-01, -4.7851e-01, ..., -4.2054e+00,
-6.5604e+00, -7.2742e+00]]],
[[[ 1.0636e+00, 1.3611e+00, 1.1817e+00, ..., 1.3048e+00,
1.1065e+00, 4.3870e-01],
[-6.6719e-01, -2.4301e+00, -1.2530e+00, ..., 9.0657e-02,
4.4629e-02, -1.3679e+00],
[-5.0461e-01, -1.1362e+00, -5.6252e-01, ..., -5.2997e-01,
2.1840e-01, 6.0698e-01],
...,
[ 5.0430e-01, -8.1856e-02, -1.5904e-02, ..., 4.9155e-01,
1.0198e+00, 7.7349e-01],
[-1.8572e-02, -6.1067e-01, 6.8708e-03, ..., 8.1929e-01,
1.5667e+00, 1.2177e-01],
[ 8.1123e-01, -2.3190e-01, -1.0615e-01, ..., 3.5485e-01,
2.8755e-01, 7.2143e-01]],
[[-2.9468e+00, 7.3934e-02, -9.9864e-01, ..., -1.5242e+00,
-2.5632e+00, -7.1041e-01],
[-4.2601e+00, -2.7189e+00, -2.4488e+00, ..., -3.1456e+00,
-4.0546e+00, -1.5556e+00],
[-3.2162e+00, -2.5110e+00, -3.0166e+00, ..., -3.2013e+00,
-2.8949e+00, -1.3944e+00],
...,
[-7.5130e-01, -2.6356e-01, -1.1163e+00, ..., -1.0889e-01,
-2.1516e-02, 7.7861e-01],
[-2.4370e+00, -2.8761e-01, -5.5137e-01, ..., 1.3093e-01,
-5.5392e-02, 6.5270e-01],
[-1.5291e+00, -1.1933e+00, -8.7290e-01, ..., -4.8601e-01,
-6.1626e-01, -1.8222e-01]],
[[ 2.5429e+00, 3.0102e+00, 2.4666e+00, ..., 1.9744e+00,
2.3974e-01, 2.5102e+00],
[ 8.3549e-01, 6.3315e-01, 2.5691e+00, ..., 1.4101e+00,
2.6030e-01, 2.7174e+00],
[ 1.1369e+00, 1.4640e+00, 2.0211e+00, ..., 1.0603e+00,
1.7436e+00, 2.8737e+00],
...,
[ 8.9049e-01, 9.4621e-01, 1.8690e+00, ..., 7.4460e-01,
6.3184e-01, 1.8959e+00],
[ 1.0302e+00, 2.2152e+00, 1.9194e+00, ..., 1.0497e+00,
7.2999e-01, 1.6815e+00],
[ 2.3535e+00, 1.8451e+00, 2.4150e+00, ..., 1.9217e+00,
1.0117e+00, 1.7061e+00]],
...,
[[-2.8040e+00, -7.2189e+00, -6.0352e+00, ..., -3.3401e+00,
-4.1601e+00, -5.4630e+00],
[-3.4734e+00, -7.8281e+00, -5.5545e+00, ..., -3.9645e+00,
-4.5659e+00, -5.2300e+00],
[-2.2909e+00, -6.3630e+00, -4.5034e+00, ..., -2.7612e+00,
-3.7802e+00, -3.4432e+00],
...,
[-2.4298e+00, -4.9322e+00, -3.1098e+00, ..., -4.9315e+00,
-2.8693e+00, -3.0059e+00],
[-2.4320e+00, -5.3983e+00, -4.0949e+00, ..., -4.8664e+00,
-3.2423e+00, -4.9319e+00],
[-2.5206e+00, -5.7269e+00, -2.8585e+00, ..., -2.9654e+00,
-1.6416e+00, -3.3068e+00]],
[[ 3.0028e-01, 1.1582e+00, 4.3425e-01, ..., 4.2287e-01,
1.4289e+00, 1.6767e+00],
[ 1.3101e+00, 9.2895e-01, -2.0252e-02, ..., -1.4522e-01,
1.1569e+00, 2.7924e+00],
[ 1.3688e+00, 9.0080e-01, 6.0572e-01, ..., 7.4162e-01,
1.2016e+00, 2.0254e+00],
...,
[ 1.6197e+00, 5.8883e-01, 1.1636e+00, ..., 2.4112e+00,
2.0019e+00, 1.4865e+00],
[ 2.0304e+00, 1.1181e+00, 1.7298e+00, ..., 1.8731e+00,
1.7294e+00, 2.5781e+00],
[ 1.2032e+00, 9.2507e-01, 1.3626e+00, ..., 1.2986e+00,
1.5054e+00, 1.5307e+00]],
[[-6.1337e-01, 5.7686e-02, -1.3815e-01, ..., 3.6654e-01,
3.6632e-02, 9.0495e-01],
[-1.5129e+00, -2.1599e+00, -1.8224e+00, ..., -1.0801e+00,
-8.6786e-01, -8.8444e-02],
[-2.3612e+00, -1.4251e+00, -4.7617e-01, ..., -2.9220e-01,
-1.0531e-01, -1.4541e-01],
...,
[-1.9045e+00, -1.2993e+00, -8.4863e-01, ..., -3.2755e-01,
-4.4423e-01, -4.1804e-01],
[-1.6849e+00, -1.2351e+00, -7.5334e-01, ..., -9.8415e-02,
-3.3976e-01, 4.6680e-01],
[-2.3170e-01, 4.3356e-01, -1.7704e-01, ..., 1.8897e-01,
3.9969e-02, 6.3423e-01]]],
[[[ 9.1657e-01, 1.5760e+00, 1.0424e+00, ..., 1.3003e+00,
1.0726e+00, 9.9998e-01],
[-9.3740e-01, -2.1217e+00, -1.2779e+00, ..., -8.0521e-01,
-7.7929e-01, -1.5253e+00],
[-4.4267e-02, -1.1496e+00, -1.2645e-01, ..., -1.3660e+00,
5.1074e-01, 8.4850e-01],
...,
[-3.3187e-02, -1.1897e+00, -1.2742e+00, ..., -6.0241e-01,
1.0508e-01, 5.4043e-01],
[-5.9073e-01, -1.8363e+00, -1.3144e+00, ..., -6.0660e-01,
-2.4303e-01, 1.2031e-01],
[-3.6783e-01, -1.7385e+00, -1.6216e+00, ..., -3.2537e-01,
-1.6356e+00, -3.4158e-01]],
[[-4.1658e+00, -3.4974e-01, -1.5652e+00, ..., -1.6556e+00,
-2.3572e+00, -1.0339e+00],
[-4.0555e+00, -2.3584e+00, -2.6880e+00, ..., -2.4556e+00,
-3.0424e+00, -7.5323e-01],
[-2.3817e+00, -2.0295e+00, -3.1499e+00, ..., -3.1578e+00,
-2.5646e+00, -8.9639e-01],
...,
[-1.5241e+00, -1.6320e+00, -3.2803e+00, ..., -3.3384e+00,
-4.6139e+00, -1.6951e+00],
[-2.9163e+00, -1.6569e+00, -2.9731e+00, ..., -4.1579e+00,
-4.5131e+00, -1.7980e+00],
[-2.2817e+00, -1.9946e+00, -2.7360e+00, ..., -3.1042e+00,
-2.5580e+00, -1.2937e+00]],
[[ 3.3463e+00, 4.5702e+00, 3.6543e+00, ..., 2.6530e+00,
7.9688e-01, 2.7837e+00],
[ 8.0510e-01, 1.2018e+00, 1.9708e+00, ..., 1.1375e+00,
-1.2137e+00, 2.4602e+00],
[ 1.5659e+00, 1.8800e+00, 2.0438e+00, ..., 1.6820e+00,
-3.0464e-01, 1.8841e+00],
...,
[ 9.4685e-01, 1.1227e+00, 1.5426e+00, ..., -4.8219e+00,
-2.9991e-01, 1.9413e-01],
[ 6.7254e-01, 1.8699e+00, 1.1751e+00, ..., -6.0029e+00,
-1.3367e+00, 1.1065e+00],
[ 1.6471e+00, 1.6673e+00, 1.8434e+00, ..., 8.4138e-01,
1.6688e+00, 1.3120e+00]],
...,
[[-2.6693e+00, -7.8620e+00, -5.5257e+00, ..., -4.6132e+00,
-3.2897e+00, -5.2461e+00],
[-3.1986e+00, -7.4049e+00, -5.3228e+00, ..., -3.4850e+00,
-2.7043e+00, -4.0646e+00],
[-1.1979e+00, -4.7604e+00, -4.8722e+00, ..., -4.4551e+00,
-3.7721e+00, -2.4845e+00],
...,
[-2.5883e+00, -5.3135e+00, -4.7275e+00, ..., -3.3284e+00,
-7.3121e+00, -6.8198e+00],
[-2.7853e+00, -5.7814e+00, -5.4425e+00, ..., -8.2708e+00,
-7.6642e+00, -7.1363e+00],
[-1.4640e+00, -3.9937e+00, -2.6868e+00, ..., -4.8791e+00,
-4.0646e+00, -4.9025e+00]],
[[-2.8178e-01, 6.0327e-01, 5.6141e-01, ..., 6.4551e-01,
9.0253e-01, 1.4024e+00],
[ 1.8835e+00, 1.0914e+00, 1.5347e+00, ..., 1.7795e+00,
2.5353e+00, 2.9327e+00],
[ 1.3502e+00, -4.9279e-01, -5.3336e-01, ..., 1.8149e+00,
2.2192e+00, 2.3331e+00],
...,
[ 1.3544e+00, 2.6993e-01, -7.3216e-01, ..., 1.4373e+00,
4.5137e-01, 1.6237e+00],
[ 1.9061e+00, 1.6656e+00, 1.0158e+00, ..., 5.7275e-01,
-1.3061e-02, 1.9759e+00],
[ 4.7425e-01, 7.6291e-01, 6.8258e-01, ..., 6.5663e-01,
1.1252e+00, 1.8817e+00]],
[[-6.1652e-01, 2.8113e-01, -3.2987e-01, ..., 4.3258e-01,
5.6697e-01, 1.3917e+00],
[-1.5462e+00, -2.0581e+00, -2.4325e+00, ..., -8.3270e-01,
-5.3293e-01, 2.9089e-01],
[-2.3482e+00, -1.4658e+00, 5.0684e-01, ..., -6.0614e-01,
4.4645e-01, 7.2296e-01],
...,
[-1.9530e+00, -1.8557e+00, -1.8422e+00, ..., -6.2135e-01,
-1.4093e-01, -1.9931e+00],
[-1.5997e+00, -2.0646e+00, -1.9198e+00, ..., -3.1355e+00,
-2.9974e+00, -1.5745e+00],
[-8.5699e-01, -9.1393e-01, -1.1679e+00, ..., -1.0219e+00,
-3.8673e-01, 3.0795e-01]]],
[[[ 1.1109e+00, 1.2613e+00, 1.1816e+00, ..., 1.1662e+00,
1.8275e+00, 5.8886e-01],
[-3.2942e-01, -2.5951e+00, -1.0458e+00, ..., -1.3652e+00,
8.0612e-01, -1.0212e+00],
[-5.2572e-01, -1.9234e+00, -9.8035e-01, ..., -1.0081e+00,
1.0776e+00, 8.6870e-01],
...,
[ 5.3289e-01, 3.5105e-01, 3.1273e-01, ..., 6.6543e-01,
9.8292e-01, 7.8370e-01],
[ 1.0884e-02, -8.7364e-03, 3.7445e-01, ..., 9.6623e-01,
1.7218e+00, 1.7676e-01],
[ 8.1203e-01, -5.1157e-02, 5.0673e-02, ..., 5.5550e-01,
2.3487e-01, 6.5273e-01]],
[[-2.8541e+00, -4.4978e-01, -2.0610e+00, ..., -2.9715e+00,
-3.8236e+00, -1.1715e+00],
[-3.9697e+00, -3.0485e+00, -2.4447e+00, ..., -5.7561e+00,
-6.8956e+00, -2.0293e+00],
[-2.7073e+00, -1.2930e+00, 2.2823e-01, ..., -3.4947e+00,
-4.4245e+00, -1.4020e+00],
...,
[-5.2623e-01, 5.9106e-01, 1.1068e-01, ..., -1.1139e+00,
2.8066e-01, 8.2465e-01],
[-2.0109e+00, 4.6734e-01, 9.0680e-01, ..., -1.2510e-01,
3.0400e-01, 7.8943e-01],
[-1.2381e+00, -1.0757e+00, -4.2330e-01, ..., -4.5367e-01,
-5.7823e-01, -2.4081e-01]],
[[ 2.4211e+00, 2.9142e+00, 2.5885e+00, ..., 5.5700e-01,
1.6351e-01, 1.9006e+00],
[ 1.0675e+00, 4.5544e-01, 1.7356e+00, ..., -1.2409e+00,
-3.9387e-01, 2.0083e+00],
[ 2.0872e+00, 7.6900e-01, 4.5386e-01, ..., -1.5335e+00,
1.5843e+00, 2.0790e+00],
...,
[ 5.7106e-01, 1.2856e+00, 1.0230e+00, ..., 8.9031e-01,
2.1754e-01, 1.7514e+00],
[ 7.7122e-01, 2.3244e+00, 1.7281e+00, ..., 1.0202e+00,
6.2893e-01, 1.7718e+00],
[ 2.1033e+00, 2.2344e+00, 2.8712e+00, ..., 1.9074e+00,
9.2702e-01, 1.8055e+00]],
...,
[[-2.4572e+00, -6.8858e+00, -5.6501e+00, ..., -2.6680e+00,
-7.2573e+00, -6.4813e+00],
[-3.3329e+00, -7.8022e+00, -4.8685e+00, ..., 3.2012e-01,
-6.9435e+00, -5.7261e+00],
[-2.2509e+00, -5.5795e+00, -2.9538e+00, ..., 4.1075e-01,
-3.3394e+00, -3.9881e+00],
...,
[-2.1949e+00, -4.4077e+00, -2.2114e+00, ..., -3.0984e+00,
-2.8528e+00, -2.6854e+00],
[-2.0691e+00, -5.0379e+00, -3.5103e+00, ..., -4.4318e+00,
-3.3737e+00, -5.0074e+00],
[-2.1979e+00, -5.3729e+00, -2.2699e+00, ..., -3.1913e+00,
-1.5913e+00, -3.3235e+00]],
[[ 2.1559e-01, 7.5192e-01, -2.1749e-01, ..., -1.7855e+00,
-6.5108e-02, 2.0370e+00],
[ 1.0605e+00, 1.7593e-01, -5.3581e-01, ..., -1.6465e+00,
-9.3281e-01, 3.0814e+00],
[ 1.0258e+00, 1.2939e+00, 1.3615e+00, ..., 2.4681e-02,
1.6141e-01, 2.3162e+00],
...,
[ 1.4313e+00, 6.3733e-01, -4.0443e-02, ..., 1.7965e+00,
2.1295e+00, 1.3262e+00],
[ 1.9387e+00, 1.2492e+00, 2.2611e+00, ..., 2.0860e+00,
1.8003e+00, 2.3812e+00],
[ 1.2618e+00, 7.0191e-01, 1.2678e+00, ..., 1.3342e+00,
1.4745e+00, 1.4880e+00]],
[[-8.2226e-01, 4.8117e-02, -1.4087e-02, ..., 1.4083e+00,
7.0757e-02, 1.1225e+00],
[-1.8132e+00, -2.1897e+00, -1.7332e+00, ..., -4.0565e-01,
-1.7235e+00, -6.6405e-01],
[-2.6944e+00, -1.4018e+00, 3.5406e-01, ..., -7.0317e-01,
9.6500e-02, -2.8331e-01],
...,
[-1.7622e+00, -1.0968e+00, -8.4853e-01, ..., -6.6301e-01,
-5.2513e-01, -4.2076e-01],
[-1.5374e+00, -7.0750e-01, -1.9273e-01, ..., -8.0006e-01,
-1.9399e-01, 4.5365e-01],
[-4.6984e-02, 7.2336e-01, 1.0193e-01, ..., 3.8534e-01,
7.0749e-03, 6.0022e-01]]]], device='cuda:0'),
tensor([[[[ 7.6193e-01, 1.1728e+00, 5.4353e-01, ..., 1.7650e-01,
2.5349e-01, -1.3798e-01],
[-6.0481e-01, -1.2592e+00, -1.2876e+00, ..., 2.8502e-01,
5.3887e-02, -4.2500e-01],
[-8.9966e-01, -9.6230e-01, 3.8047e-02, ..., 7.3756e-02,
1.3830e-01, 7.7820e-01],
...,
[-8.0293e-02, -1.2926e+00, -2.4372e+00, ..., -4.8555e+00,
1.0427e+00, 9.0331e-01],
[-6.4336e-01, -2.3867e+00, -1.0325e+00, ..., -5.6398e+00,
-1.2780e+00, -6.4633e-02],
[ 2.6411e-01, -2.7326e+00, -1.2188e+00, ..., -4.4563e+00,
4.5611e-01, 6.6496e-01]],
[[-3.3620e+00, -1.6993e+00, -3.3023e+00, ..., -2.0385e+00,
-2.4794e+00, -1.4701e+00],
[-4.1249e+00, -3.5211e+00, -5.4873e+00, ..., -1.6128e+00,
-2.7034e+00, -1.6988e+00],
[-3.4988e+00, -1.0505e+00, -2.0291e+00, ..., -2.2275e+00,
-2.4459e+00, -1.1177e+00],
...,
[-1.2390e+00, -2.3606e+00, -7.0634e+00, ..., -3.6973e+00,
-3.5482e+00, -2.6054e+00],
[-2.2093e+00, -2.7311e-01, -3.3503e+00, ..., -7.9840e+00,
-6.2330e+00, -3.7779e+00],
[-1.1024e+00, -1.6050e+00, -7.6395e-01, ..., -1.0716e+01,
-8.0911e+00, -2.0913e+00]],
[[ 2.4161e+00, 2.7866e+00, 2.2306e+00, ..., 1.7212e+00,
9.1579e-01, 2.1904e+00],
[ 9.7080e-01, 1.4095e+00, 2.6156e+00, ..., 1.0001e+00,
5.0453e-01, 2.3355e+00],
[ 1.3200e+00, 2.3072e-01, 6.8512e-01, ..., -4.2854e-01,
8.7349e-01, 2.0942e+00],
...,
[ 5.6949e-01, 2.3788e+00, 1.0814e+00, ..., -5.3523e+00,
-4.5771e+00, 9.3842e-01],
[ 5.1417e-01, 3.5612e+00, -3.1975e-01, ..., -7.5551e+00,
-5.3543e+00, 1.8656e+00],
[ 1.6062e+00, 2.6913e+00, -1.2757e+00, ..., -6.9644e+00,
-3.0232e+00, 2.9821e-01]],
...,
[[-2.7162e+00, -6.1724e+00, -3.9355e+00, ..., -2.4511e+00,
-1.9272e+00, -3.2260e+00],
[-3.2599e+00, -5.4773e+00, -1.3372e+00, ..., -3.3541e+00,
-2.9293e+00, -3.5190e+00],
[-1.8795e+00, -3.7566e+00, -2.8235e+00, ..., -4.3016e+00,
-2.4670e+00, -2.3934e+00],
...,
[-1.5639e+00, -5.2763e+00, -6.5367e+00, ..., -4.1816e-01,
-4.5308e+00, -9.4191e+00],
[-2.0589e+00, -7.0046e+00, -4.4051e+00, ..., 1.5862e+00,
-4.1046e+00, -1.0798e+01],
[-2.6005e+00, -6.2733e+00, -2.3573e+00, ..., 6.0542e-01,
-5.9860e+00, -7.5607e+00]],
[[ 4.9670e-01, 9.6137e-01, 2.6598e-01, ..., 1.0197e+00,
1.4477e+00, 1.5673e+00],
[ 4.3736e-01, -1.3012e+00, -2.6467e+00, ..., 1.2097e+00,
8.0571e-01, 2.1807e+00],
[ 1.1921e+00, 6.0230e-01, -5.3293e-01, ..., 8.0360e-01,
1.6475e-01, 1.3805e+00],
...,
[ 1.2521e+00, -1.1751e+00, 1.2823e-01, ..., -3.7767e+00,
-2.6507e+00, 4.1241e-01],
[ 2.0576e+00, 4.8965e-02, -2.2720e+00, ..., -3.6173e+00,
-3.0698e+00, 1.3275e+00],
[ 1.0196e+00, 4.5832e-01, 2.9082e+00, ..., -2.2093e+00,
-3.4691e+00, 8.4138e-01]],
[[-1.1272e-01, -3.3272e-01, -1.2225e+00, ..., -6.7830e-01,
-9.0165e-01, -3.3448e-02],
[-1.9785e+00, -1.9700e+00, -2.0700e+00, ..., -5.6505e-01,
-7.2055e-01, -4.4817e-01],
[-3.0514e+00, 5.1050e-01, 6.6187e-01, ..., -5.7605e-01,
-1.7539e-01, -9.1745e-02],
...,
[-2.2183e+00, -3.2123e+00, -3.0950e+00, ..., -5.0327e+00,
-1.0734e+00, -1.2702e+00],
[-2.3014e+00, -1.7923e+00, -2.0687e+00, ..., -2.7549e+00,
6.0505e-02, -8.0080e-01],
[-3.5864e-01, 2.2749e-02, -2.0068e+00, ..., -7.2873e-01,
-4.2404e-01, -9.9984e-01]]],
[[[-7.7590e-02, 2.8686e-01, 2.6411e-01, ..., 5.6476e-01,
8.7966e-01, -2.6197e-01],
[ 5.1025e-02, 6.1060e-01, 2.3197e-01, ..., 2.4771e-02,
7.5807e-01, 5.8260e-01],
[-8.5845e-01, 1.0401e+00, 6.8629e-01, ..., -1.7091e+00,
-8.4810e-01, 1.1992e+00],
...,
[-2.9100e+00, 2.1996e+00, 2.2750e+00, ..., 4.4564e-01,
6.2780e-01, 5.6549e-01],
[ 2.6020e-01, -2.4499e+00, 1.5179e+00, ..., 1.0318e+00,
1.5207e+00, 1.1932e-01],
[ 2.1895e+00, -7.2701e-01, -8.8512e-01, ..., 4.0366e-01,
3.2208e-01, 7.5919e-01]],
[[ 9.2932e-01, 9.1328e-01, 3.5239e-01, ..., -1.2589e+00,
-3.2260e+00, -1.4301e+00],
[ 8.8915e-01, 3.4627e-01, -1.2250e-03, ..., -2.6944e-01,
-1.6511e+00, -2.6126e+00],
[ 2.8150e-01, 8.2499e-01, 4.1282e-01, ..., -2.7193e+00,
-2.0493e+00, -1.8353e+00],
...,
[ 7.7538e-01, 1.3635e+00, -5.1819e+00, ..., -9.2972e-01,
-2.7698e-01, 8.3818e-01],
[-3.6243e+00, -4.0594e-01, -6.3502e+00, ..., 2.0607e-01,
4.2224e-02, 6.8578e-01],
[-2.0353e+00, -1.9711e+00, -3.4394e+00, ..., -5.3812e-01,
-7.1121e-01, -2.3379e-01]],
[[ 9.3963e-01, 2.4277e-01, -3.9567e-01, ..., 1.1398e-01,
-1.4271e-01, 2.4859e+00],
[-1.2160e-01, -2.7923e-01, -1.1527e+00, ..., -1.0798e+00,
-1.3444e+00, 1.5264e+00],
[-1.4627e+00, -8.5476e-01, -1.5803e+00, ..., -2.7002e+00,
-3.7651e+00, -1.5241e+00],
...,
[-3.9941e+00, 2.4478e+00, 5.9979e-01, ..., 1.1624e+00,
2.4771e-01, 9.9917e-01],
[-3.9764e+00, 5.6706e-01, 1.4540e+00, ..., 1.2577e+00,
8.3094e-01, 1.7597e+00],
[ 1.8895e+00, 3.6370e-01, 5.8421e-01, ..., 1.9546e+00,
1.0182e+00, 1.7823e+00]],
...,
[[ 9.9488e-02, -8.1683e-01, 3.5322e-01, ..., -1.2254e+00,
-8.3022e-01, -1.5107e+00],
[-6.5535e-01, -4.0098e+00, 5.9440e-01, ..., -2.8116e+00,
-1.4940e+00, -3.7886e+00],
[-1.1725e+00, -5.2658e+00, -1.4222e+00, ..., -1.1625e+00,
1.1459e-01, -2.7211e+00],
...,
[-1.6302e+00, -6.7984e+00, -6.0489e+00, ..., -4.8872e+00,
-5.4561e+00, -5.5201e+00],
[-1.4046e+00, -7.8704e+00, -6.3549e+00, ..., -4.6406e+00,
-3.5483e+00, -4.9589e+00],
[-2.1087e+00, -5.6806e+00, -2.9837e+00, ..., -2.8897e+00,
-1.6034e+00, -3.3754e+00]],
[[ 7.2278e-01, 1.0382e+00, 7.7302e-01, ..., 4.4965e-02,
-1.2960e-01, 1.0720e+00],
[ 2.9964e-01, -5.6920e-01, 5.6551e-01, ..., 8.6600e-01,
-1.4382e+00, 2.7864e-01],
[ 9.6285e-01, 1.4400e-01, 8.7486e-01, ..., -4.7049e-01,
-2.7968e+00, -1.3541e+00],
...,
[ 1.3426e+00, -1.9662e+00, -3.7707e+00, ..., 2.4359e+00,
2.8051e+00, 1.9060e+00],
[ 2.3609e+00, 2.4503e+00, -1.8599e+00, ..., 1.9622e+00,
1.6302e+00, 2.5965e+00],
[ 5.3804e-01, 2.3474e-01, 2.7159e-01, ..., 1.2547e+00,
1.4915e+00, 1.5319e+00]],
[[-1.5668e-01, -6.9865e-01, -2.1020e-01, ..., -2.0644e-01,
-3.8372e-01, -7.0945e-01],
[ 4.8656e-01, -1.3931e+00, -1.1127e+00, ..., -4.3138e-01,
1.1327e+00, 7.2805e-02],
[-7.1476e-01, -1.0142e+00, -9.0743e-01, ..., -1.8846e+00,
-1.1455e-01, 3.1329e-02],
...,
[ 2.7374e-01, 8.3452e-01, -1.8539e+00, ..., -4.6532e-01,
-1.9473e-01, -2.8925e-01],
[-6.5437e-01, -1.1102e+00, -2.4399e+00, ..., 1.1331e-03,
-1.6079e-01, 4.8535e-01],
[-3.5825e-01, 1.1408e+00, 1.5101e-01, ..., 2.0770e-01,
1.5277e-02, 6.3924e-01]]],
[[[ 1.0108e+00, 1.7661e+00, 1.0319e+00, ..., 5.4874e-01,
3.3652e-01, -1.9255e-01],
[-8.8000e-01, -1.7629e+00, -8.7748e-01, ..., 2.8143e-01,
6.7174e-01, -5.9008e-01],
[-4.3163e-03, -8.9648e-01, 5.7895e-01, ..., -5.9288e-01,
8.2783e-02, 5.6547e-01],
...,
[ 7.4778e-01, -1.8168e+00, -1.4897e+00, ..., 1.2726e-01,
7.7936e-01, 6.1287e-01],
[-1.4384e-01, -3.1884e-01, 4.8297e-01, ..., 1.3488e-01,
7.7475e-01, -6.8363e-03],
[ 8.2916e-01, -4.1744e-01, -2.8765e-01, ..., -4.3641e-01,
-4.4334e-01, 3.5816e-01]],
[[-3.8435e+00, -1.7438e+00, -2.1528e+00, ..., -2.9931e+00,
-3.6640e+00, -9.4440e-01],
[-4.0874e+00, -4.2290e+00, -3.6383e+00, ..., -2.0605e+00,
-3.1473e+00, -5.4643e-01],
[-2.2327e+00, -1.2441e+00, -1.8025e+00, ..., -3.3052e+00,
-3.2635e+00, -7.4794e-01],
...,
[-4.0732e+00, -4.2346e+00, -1.5886e+00, ..., -3.0333e+00,
-2.5413e+00, -5.1690e-02],
[-3.3517e+00, -3.2752e-01, -1.2713e+00, ..., -1.5665e+00,
-1.6829e+00, -8.7441e-02],
[-2.3467e+00, -1.7550e+00, -1.7647e+00, ..., -9.5628e-01,
-1.1088e+00, -6.9875e-02]],
[[ 2.6736e+00, 3.8903e+00, 3.2522e+00, ..., 2.3069e+00,
7.9285e-01, 2.3706e+00],
[ 2.5134e-01, 1.1785e+00, 2.4112e+00, ..., 1.9606e+00,
2.0086e-01, 2.4366e+00],
[ 7.9643e-01, 1.4395e+00, 8.5697e-01, ..., 1.2533e+00,
1.2065e+00, 2.3274e+00],
...,
[-2.6204e-02, -3.0397e+00, -2.9781e+00, ..., 6.9806e-01,
9.5138e-01, 1.9963e+00],
[ 1.1343e+00, 3.3134e+00, 2.4169e+00, ..., 6.8503e-02,
2.4734e-01, 1.1406e+00],
[ 1.7564e+00, 1.6547e+00, 2.1301e+00, ..., 1.0506e+00,
9.0973e-01, 1.3174e+00]],
...,
[[-1.6489e+00, -7.2115e+00, -5.9813e+00, ..., -4.1349e+00,
-3.1521e+00, -3.8415e+00],
[-2.1653e+00, -6.2362e+00, -4.5695e+00, ..., -5.6163e+00,
-4.8137e+00, -4.6638e+00],
[-3.6653e-01, -2.3696e+00, -3.7415e+00, ..., -4.7131e+00,
-3.4361e+00, -2.0047e+00],
...,
[-2.2333e+00, -4.3192e+00, -4.5096e+00, ..., -5.1078e+00,
-3.9151e+00, -3.5901e+00],
[-1.6175e+00, -4.9749e+00, -4.7806e+00, ..., -6.5860e+00,
-4.7148e+00, -4.7540e+00],
[-2.5145e+00, -5.2700e+00, -2.8091e+00, ..., -4.4860e+00,
-2.7524e+00, -3.2847e+00]],
[[-7.3446e-01, -4.0784e-01, 1.3882e-01, ..., 9.0003e-01,
9.6495e-01, 1.7749e+00],
[ 1.4026e+00, 1.7721e-01, 8.8277e-01, ..., 1.5634e+00,
1.5686e+00, 2.9543e+00],
[ 1.1103e+00, 6.4534e-01, -1.9580e-01, ..., 1.1114e+00,
7.0279e-01, 2.2352e+00],
...,
[ 6.6605e-01, -7.8377e-01, -3.8955e-02, ..., 1.7577e+00,
1.4897e+00, 1.9870e+00],
[ 2.2400e+00, 1.7762e+00, 2.0535e+00, ..., 1.6711e+00,
9.5220e-01, 2.2854e+00],
[ 1.0024e+00, 6.7748e-01, 1.2813e+00, ..., 1.0553e+00,
1.1063e+00, 1.5822e+00]],
[[-4.4691e-01, 5.9136e-01, 1.8148e-01, ..., -2.4878e-01,
-5.9766e-01, 8.8198e-01],
[-1.3618e+00, -1.4809e+00, -1.1824e+00, ..., -2.3637e-01,
-5.5681e-02, 1.1109e+00],
[-1.7501e+00, -2.6881e-01, 4.0587e-01, ..., 6.7077e-01,
1.7918e-01, 4.0619e-01],
...,
[-2.5913e+00, -1.9935e+00, -1.5697e+00, ..., -7.3951e-01,
-8.3732e-01, -1.1239e-02],
[-2.2478e+00, -6.0267e-01, -7.9807e-01, ..., -6.1903e-01,
-8.6112e-01, 2.9397e-01],
[-4.6776e-01, 2.2906e-01, -4.9215e-01, ..., 1.2868e-01,
-2.6011e-01, 5.8774e-01]]],
[[[ 9.8131e-01, 1.6225e+00, 1.1459e+00, ..., 7.1126e-01,
9.2826e-01, 4.1884e-01],
[-8.2506e-01, -1.8971e+00, -6.8065e-01, ..., -3.0498e-01,
-4.0837e-02, -2.7900e-01],
[-1.1604e-02, -8.2890e-01, -1.4436e-01, ..., -8.3404e-01,
-1.4852e-01, 6.5081e-01],
...,
[ 2.0735e-01, -3.4307e-01, 5.0936e-01, ..., 5.4794e-01,
4.8407e-01, 1.9494e-02],
[-2.9563e-01, -1.1625e+00, 1.1712e-01, ..., -2.1722e-02,
7.9208e-01, 3.0367e-01],
[ 7.4291e-01, -2.8307e-01, -1.9260e-01, ..., -2.7951e-01,
-7.7167e-01, 2.0280e-01]],
[[-4.1486e+00, -1.2689e+00, -2.2014e+00, ..., -1.4124e+00,
-1.8781e+00, -1.6185e+00],
[-3.8939e+00, -2.8026e+00, -2.4828e+00, ..., -1.1892e+00,
-2.0093e+00, -2.0178e+00],
[-2.5823e+00, -1.9780e+00, -2.8875e+00, ..., -1.5032e+00,
-1.6361e+00, -1.3643e+00],
...,
[-1.0429e+00, -1.1387e+00, -1.8103e+00, ..., -1.2813e+00,
-8.3926e-01, -9.3332e-01],
[-2.9214e+00, -5.6079e-01, -4.8772e-01, ..., -6.1213e-01,
-5.6770e-01, -9.0165e-01],
[-2.0902e+00, -1.5091e+00, -1.4541e+00, ..., -5.4193e-01,
-5.6061e-01, -5.0874e-01]],
[[ 2.9360e+00, 4.0709e+00, 3.2579e+00, ..., 1.6333e+00,
9.4164e-01, 2.5056e+00],
[ 2.7680e-01, 7.9868e-01, 2.2340e+00, ..., 1.0786e+00,
7.5786e-01, 2.5091e+00],
[ 7.9547e-01, 1.4212e+00, 2.0001e+00, ..., 6.0093e-01,
1.0561e+00, 1.8331e+00],
...,
[ 4.3449e-01, 1.5834e+00, 1.7726e+00, ..., 3.0882e-01,
-2.2360e-01, 1.0224e+00],
[ 2.6445e-01, 2.2587e+00, 2.3313e+00, ..., 1.0056e-01,
1.0631e+00, 1.5371e+00],
[ 1.4508e+00, 1.3622e+00, 1.7167e+00, ..., 6.2135e-01,
1.4172e+00, 1.4914e+00]],
...,
[[-2.0803e+00, -7.3431e+00, -5.8568e+00, ..., -4.8063e+00,
-3.4288e+00, -3.9513e+00],
[-2.7183e+00, -6.5044e+00, -4.7245e+00, ..., -5.3871e+00,
-3.7838e+00, -3.0526e+00],
[-1.6217e-01, -2.6453e+00, -3.2133e+00, ..., -3.4589e+00,
-2.4538e+00, -1.3794e+00],
...,
[-1.0260e+00, -2.4090e+00, -1.7588e+00, ..., -4.0574e+00,
-2.7554e+00, -2.7221e+00],
[-1.3313e+00, -3.7677e+00, -3.3685e+00, ..., -4.9670e+00,
-4.2093e+00, -4.6097e+00],
[-2.2225e+00, -4.9838e+00, -3.0706e+00, ..., -3.3284e+00,
-2.4249e+00, -2.4890e+00]],
[[-7.5803e-01, -5.8953e-02, 1.9486e-01, ..., 1.4602e+00,
1.7324e+00, 1.2764e+00],
[ 1.7639e+00, 1.0106e+00, 1.3788e+00, ..., 1.5917e+00,
1.1066e+00, 1.7978e+00],
[ 1.1716e+00, 1.3891e-02, 3.0986e-01, ..., 1.9748e+00,
5.8205e-01, 1.2431e+00],
...,
[ 1.2110e+00, -7.5020e-01, -3.1366e-01, ..., 1.1233e+00,
6.8804e-01, 8.1141e-01],
[ 1.9013e+00, 1.0187e+00, 2.3210e+00, ..., 1.3979e+00,
7.3701e-01, 1.6356e+00],
[ 1.0322e+00, 3.8997e-01, 9.2790e-01, ..., 7.6398e-01,
6.8781e-01, 1.4619e+00]],
[[-4.9759e-01, 6.1358e-01, 1.1403e-01, ..., -3.5428e-01,
-5.8107e-01, -5.0031e-02],
[-1.0152e+00, -1.1710e+00, -1.4395e+00, ..., -1.0636e+00,
-1.3260e+00, -1.0615e+00],
[-1.9726e+00, -6.8545e-01, -1.2851e-01, ..., -5.5648e-01,
-1.1204e+00, -2.7778e-01],
...,
[-1.6939e+00, -9.9812e-01, -7.3372e-01, ..., -8.9841e-01,
-1.0923e+00, -1.5346e+00],
[-1.7475e+00, -9.9035e-01, -6.3346e-01, ..., -7.9959e-01,
-1.3790e+00, -6.0333e-01],
[-3.8119e-01, 3.7899e-01, -2.8830e-01, ..., -2.1473e-01,
-7.0387e-01, 3.3308e-02]]],
[[[ 7.4284e-01, 1.0965e+00, 7.2829e-01, ..., 4.7581e-01,
3.9828e-01, 2.1769e-01],
[-2.8832e-01, -1.3151e+00, -5.6623e-01, ..., 3.3833e-01,
3.4567e-01, -1.1668e+00],
[ 1.5737e-01, -7.5890e-01, -3.4010e-01, ..., -4.9055e-01,
1.0028e+00, 7.3029e-01],
...,
[ 1.3406e-01, -1.1745e+00, -6.0014e-01, ..., -1.6662e-01,
9.5802e-01, 5.0077e-01],
[-4.2088e-01, -1.7451e+00, -7.4183e-01, ..., 3.9294e-02,
1.4066e+00, -9.6841e-02],
[ 3.8726e-01, -5.4750e-01, -6.7647e-01, ..., -3.4009e-01,
3.7925e-01, 6.0437e-01]],
[[-2.9353e+00, -1.5137e+00, -3.0584e+00, ..., -1.9041e+00,
-2.4791e+00, -1.3297e+00],
[-4.1931e+00, -1.8294e+00, -2.3688e+00, ..., -4.8116e-01,
-1.4016e+00, -4.3511e-01],
[-4.1974e+00, -2.4773e+00, -3.0234e+00, ..., -1.5305e+00,
-1.4425e+00, -8.6455e-01],
...,
[-2.6860e+00, -1.8214e+00, -2.7101e+00, ..., -2.9574e-01,
-2.2589e-02, 4.6742e-01],
[-3.5340e+00, -1.2836e+00, -2.4061e+00, ..., -3.5733e-01,
-5.0624e-01, 1.7742e-01],
[-1.2763e+00, -8.8500e-01, -1.7169e+00, ..., -4.4549e-01,
-7.4647e-01, -5.3580e-01]],
[[ 1.5724e+00, 2.7542e+00, 2.4126e+00, ..., 2.8516e+00,
1.4283e+00, 3.0613e+00],
[ 1.0373e-01, 1.2132e+00, 1.5833e+00, ..., 1.9348e+00,
4.3182e-02, 2.6961e+00],
[ 6.6800e-01, 1.4672e+00, 9.0902e-01, ..., 1.1539e+00,
6.6065e-01, 1.8589e+00],
...,
[ 9.0102e-01, 1.2574e+00, 8.8057e-01, ..., 7.3138e-01,
3.2267e-02, 1.3957e+00],
[ 6.3729e-01, 1.9023e+00, 9.1422e-01, ..., 1.0430e+00,
5.3342e-01, 1.4086e+00],
[ 1.5360e+00, 1.5796e+00, 1.6122e+00, ..., 1.4003e+00,
1.0562e+00, 1.9255e+00]],
...,
[[-1.2511e+00, -4.4264e+00, -3.9268e+00, ..., -3.3834e+00,
-2.8082e+00, -4.2949e+00],
[-1.6481e+00, -3.6650e+00, -4.2414e+00, ..., -5.2330e+00,
-4.1600e+00, -4.8185e+00],
[-9.4968e-01, -3.1292e+00, -3.8360e+00, ..., -2.6169e+00,
-2.5939e+00, -1.7073e+00],
...,
[-1.4712e+00, -3.9557e+00, -4.2472e+00, ..., -3.7626e+00,
-2.4581e+00, -1.3717e+00],
[-1.4724e+00, -4.4858e+00, -5.2891e+00, ..., -4.4234e+00,
-3.2641e+00, -3.4141e+00],
[-1.6019e+00, -4.8568e+00, -4.1727e+00, ..., -3.3362e+00,
-2.0613e+00, -2.5267e+00]],
[[ 7.9659e-01, 5.2293e-01, -1.2306e-01, ..., 7.8480e-01,
1.0085e+00, 1.1985e+00],
[ 1.1972e+00, 9.5764e-01, 5.2340e-01, ..., 1.5766e+00,
2.4286e+00, 2.6410e+00],
[ 6.0971e-01, 5.5196e-01, 1.2825e-01, ..., 1.0151e+00,
1.6299e+00, 1.8757e+00],
...,
[ 8.4505e-01, -6.1175e-02, -9.7201e-02, ..., 1.1727e+00,
1.8228e+00, 1.6596e+00],
[ 1.5432e+00, 1.4264e+00, 1.1461e+00, ..., 1.2501e+00,
1.4834e+00, 2.4269e+00],
[ 6.6902e-01, 9.0451e-01, 1.0073e+00, ..., 1.0216e+00,
1.3261e+00, 1.3302e+00]],
[[-2.7897e-01, -1.7961e-01, -1.0063e+00, ..., -1.6776e-01,
-1.4757e-01, 8.6685e-01],
[-1.3714e+00, -1.4233e+00, -2.0815e+00, ..., 1.3572e-01,
-3.2506e-02, 4.2528e-01],
[-1.6293e+00, -7.1187e-01, -1.0606e+00, ..., 2.7136e-01,
2.4414e-01, 1.9450e-01],
...,
[-1.7468e+00, -1.6563e+00, -1.6746e+00, ..., -1.0734e-01,
-2.4004e-01, -2.8848e-01],
[-1.4353e+00, -1.5944e+00, -1.6978e+00, ..., -5.0218e-01,
-4.6235e-01, 3.3527e-01],
[-2.0042e-01, 4.0176e-02, -5.3328e-01, ..., 2.0627e-01,
1.3736e-01, 3.2796e-01]]]], device='cuda:0'),
tensor([[[[ 9.0811e-01, 9.4569e-01, 5.8798e-02, ..., -1.9468e-01,
4.3484e-01, 4.6097e-01],
[-1.3728e+00, -2.2178e+00, -1.4886e+00, ..., -2.9699e+00,
-1.0024e+00, -6.2098e-01],
[ 1.2032e-01, -9.3880e-02, 1.8990e+00, ..., 1.3752e+00,
-2.6371e+00, 2.3014e-01],
...,
[-1.1984e-01, -2.9773e+00, -6.0073e+00, ..., 2.2150e+00,
-2.5591e-01, -2.4403e-01],
[-9.0740e-02, -3.1772e-02, 9.0551e-01, ..., 6.4164e-01,
1.0838e+00, -1.6206e-01],
[ 4.7196e-01, -5.3266e-01, -3.7241e-01, ..., -9.2467e-02,
-5.6189e-01, 9.9090e-02]],
[[-5.0381e+00, -1.9679e+00, -2.2492e+00, ..., -2.0841e+00,
-2.3853e+00, -1.7622e+00],
[-4.7928e+00, -3.5715e+00, -2.6211e+00, ..., -2.7462e+00,
-2.6376e+00, -2.2373e+00],
[-1.3534e+00, 2.0575e-01, -4.1975e-01, ..., -4.3126e+00,
-2.0183e+00, -9.2117e-01],
...,
[-1.2906e+01, -1.2381e+01, -8.7082e+00, ..., -4.3227e+00,
-6.6562e-01, -7.4005e-01],
[-3.9128e+00, -5.1076e+00, -5.5212e+00, ..., -5.1596e+00,
-3.4611e-01, -1.0166e+00],
[-1.3918e-01, -1.6920e+00, -1.4353e+00, ..., -1.8304e+00,
-3.5039e-01, -4.0578e-01]],
[[ 2.9777e+00, 3.6299e+00, 3.6500e+00, ..., 1.8269e+00,
2.2157e-01, 2.6998e+00],
[ 1.5943e+00, 1.8927e+00, 2.5613e+00, ..., 3.3522e+00,
1.5920e+00, 2.8181e+00],
[-5.1012e-01, -1.2079e+00, -3.6739e-01, ..., 6.5251e-01,
2.0949e+00, 2.6514e+00],
...,
[-4.7549e-01, -4.6279e+00, -9.1710e+00, ..., 2.2613e+00,
-1.5649e+00, 6.5400e-01],
[ 1.7979e+00, 2.5682e+00, 8.6705e-01, ..., -9.4287e-01,
-3.5502e-01, 1.3446e+00],
[ 1.9184e+00, 1.9635e+00, 2.9886e+00, ..., 1.2094e+00,
6.3954e-01, 1.4192e+00]],
...,
[[-3.8800e+00, -9.5637e+00, -5.5680e+00, ..., -3.6776e+00,
-4.4466e+00, -4.9032e+00],
[-2.8214e+00, -8.7706e+00, -5.3283e+00, ..., -5.6286e+00,
-6.0435e+00, -4.1614e+00],
[-2.2932e+00, -6.8796e+00, -6.6239e+00, ..., -1.9529e+01,
-1.1171e+01, -2.5209e+00],
...,
[-4.8779e+00, -1.0269e+01, -6.1546e+00, ..., -1.9487e+01,
-1.6475e+01, -9.5962e+00],
[-4.6132e+00, -1.1270e+01, -5.8555e+00, ..., -1.5983e+01,
-9.0165e+00, -4.4277e+00],
[-2.3205e+00, -5.8987e+00, -3.3468e+00, ..., -5.9193e+00,
-6.3242e+00, -2.5970e+00]],
[[-8.9139e-01, 2.3238e-01, 5.4263e-01, ..., 1.8623e+00,
1.5133e+00, 1.2830e+00],
[-8.5876e-01, -9.2737e-01, 1.1112e+00, ..., 2.3975e+00,
2.5661e+00, 1.9586e+00],
[ 3.9065e-01, -7.7318e-01, -1.5246e+00, ..., 4.1639e-01,
3.1248e+00, 1.8058e+00],
...,
[-2.9915e+00, -6.5659e-01, 1.3386e+00, ..., 7.0228e-01,
3.9041e+00, 2.8392e+00],
[ 2.2814e+00, 1.8716e+00, 1.5110e+00, ..., -1.4808e+00,
2.2705e+00, 1.7244e+00],
[ 1.2851e+00, 6.6922e-01, 1.6545e+00, ..., 2.2213e+00,
1.9443e+00, 1.5125e+00]],
[[-5.8708e-01, -6.6081e-01, -6.6845e-01, ..., -6.3051e-01,
-7.3770e-01, 1.2685e-01],
[-1.9308e+00, -1.6590e+00, -1.0545e+00, ..., -3.8820e+00,
-1.0029e+00, -1.0185e+00],
[-1.4231e+00, 2.8356e-01, 9.1288e-01, ..., -9.8803e-01,
-2.5444e+00, -7.0441e-01],
...,
[-4.4598e+00, -6.0491e+00, -4.7619e+00, ..., -2.5586e+00,
-5.9270e+00, -1.1945e-01],
[-3.7123e+00, -2.4061e+00, -1.8236e+00, ..., -5.6511e+00,
-4.2865e+00, -5.3374e-01],
[ 1.4845e-01, -2.6376e-02, -5.3967e-01, ..., -1.7277e+00,
-1.2148e+00, 4.3222e-02]]],
[[[ 6.8712e-01, 1.0434e+00, 5.9902e-01, ..., 4.0479e-01,
4.8135e-01, 3.6824e-01],
[-4.5267e-01, -1.4881e+00, -5.1888e-01, ..., -2.6257e-01,
-2.7477e-01, -1.4374e+00],
[ 1.0707e-02, -9.8479e-01, -4.4684e-01, ..., -3.8916e+00,
4.1054e-01, 3.7632e-01],
...,
[-3.6041e+00, -9.2489e-01, -4.1799e+00, ..., -1.5640e+00,
2.4864e+00, -1.7288e-02],
[-1.2777e+00, 3.8364e-01, -2.7558e+00, ..., -6.5324e-01,
2.1304e+00, -4.7951e-01],
[-1.7036e+00, 2.0431e+00, 5.6626e-03, ..., 8.0415e-02,
-5.2900e-01, 5.8316e-01]],
[[-3.5602e+00, -1.6744e+00, -3.1694e+00, ..., -1.4883e+00,
-1.7920e+00, -8.9672e-01],
[-4.5749e+00, -1.8788e+00, -2.3280e+00, ..., -5.9536e-01,
-1.5986e+00, -3.3878e-01],
[-4.4676e+00, -2.5450e+00, -2.8921e+00, ..., -1.6533e+00,
-2.1485e+00, -8.9606e-01],
...,
[-2.2227e+00, -6.1709e+00, -4.7914e+00, ..., -1.2736e+01,
-1.0560e+01, 9.4297e-01],
[-4.8623e-01, -5.1437e+00, -7.1084e+00, ..., -5.8790e+00,
-4.8304e+00, 6.2389e-01],
[ 1.5687e-01, -2.3799e+00, -5.3190e+00, ..., -1.3724e+00,
-1.5674e+00, -8.8110e-01]],
[[ 2.0259e+00, 3.0239e+00, 2.5186e+00, ..., 2.4444e+00,
1.3356e+00, 2.9368e+00],
[ 4.8248e-01, 1.4012e+00, 1.8692e+00, ..., 1.7531e+00,
-4.8166e-01, 2.4514e+00],
[ 9.9735e-01, 1.4762e+00, 1.0312e+00, ..., 3.7360e+00,
8.6972e-01, 1.8345e+00],
...,
[-4.3234e-01, -3.0095e+00, -5.9072e+00, ..., -1.3852e+01,
-2.4220e-01, -9.2833e-01],
[ 9.4723e-01, -3.1810e+00, -8.2055e+00, ..., -5.4948e+00,
2.7461e-01, 5.4378e-01],
[-6.2530e-01, -2.6327e+00, -5.6611e+00, ..., 1.1545e+00,
3.2569e-01, 1.7128e+00]],
...,
[[-1.6363e+00, -5.0265e+00, -4.1954e+00, ..., -3.2607e+00,
-2.2657e+00, -3.9923e+00],
[-2.0989e+00, -4.8024e+00, -5.0148e+00, ..., -5.3609e+00,
-3.9246e+00, -3.9794e+00],
[-1.2575e+00, -3.9585e+00, -4.2149e+00, ..., -5.1098e+00,
-4.1992e+00, -2.0020e+00],
...,
[-2.1794e+00, -2.6440e+00, 1.6514e-01, ..., -1.0790e+01,
-1.7385e+01, -1.1609e+01],
[-1.7524e+00, -5.0499e+00, 8.7036e-02, ..., -9.8603e+00,
-7.6487e+00, -8.5830e+00],
[-3.4650e-01, -8.9164e+00, -3.4557e+00, ..., -6.4076e+00,
-5.1271e+00, -4.8408e+00]],
[[ 5.1119e-01, 4.5451e-01, -9.5039e-02, ..., 8.2468e-01,
1.2170e+00, 1.2360e+00],
[ 1.2683e+00, 1.0263e+00, 6.7812e-01, ..., 1.7755e+00,
2.1192e+00, 2.5688e+00],
[ 6.5936e-01, 6.0696e-01, 2.6483e-01, ..., 1.3156e+00,
2.2581e+00, 2.0423e+00],
...,
[-2.0161e+00, -3.1167e+00, -2.0631e+00, ..., -6.8540e-01,
-1.1632e+00, 3.6758e+00],
[-3.6162e+00, -7.0616e+00, -3.2461e+00, ..., 7.2271e-01,
6.2708e-01, 2.9279e+00],
[ 2.0470e-01, -1.8806e+00, -3.3416e-01, ..., 1.3459e+00,
1.6561e+00, 1.2166e+00]],
[[-3.8926e-01, -1.4106e-01, -9.6297e-01, ..., -1.8655e-01,
-6.8017e-02, 9.0756e-01],
[-1.5214e+00, -1.2518e+00, -1.8643e+00, ..., -7.1145e-03,
-4.9542e-01, 2.1711e-01],
[-1.8950e+00, -6.8354e-01, -7.7556e-01, ..., -2.3756e+00,
4.1974e-01, 1.8855e-01],
...,
[-1.9316e+00, -8.3508e+00, 7.3741e-01, ..., -5.0436e-01,
-2.6723e+00, -3.9531e+00],
[-1.2374e+00, -6.4963e+00, -1.2627e+00, ..., -2.4147e+00,
-1.2660e+00, -6.3153e-01],
[-7.4963e-01, -1.6353e+00, -1.3274e+00, ..., 1.2177e-01,
5.9424e-01, 6.2506e-01]]],
[[[ 9.4914e-01, 1.5763e+00, 1.0829e+00, ..., 4.7392e-01,
2.7762e-01, -1.6487e-01],
[-8.6580e-01, -1.9942e+00, -8.8161e-01, ..., 4.7727e-01,
3.8385e-01, -9.1719e-01],
[ 1.0724e-01, -7.5608e-01, -5.1770e-01, ..., -2.1090e+00,
-1.5928e+00, -7.9081e-01],
...,
[ 4.8106e-01, -1.3540e-01, -6.7594e-01, ..., -7.0807e-01,
-1.5383e+00, 7.4850e-03],
[-1.1460e-01, -1.0416e+00, -4.7684e-01, ..., 1.4502e+00,
3.7938e-01, -5.7130e-01],
[ 7.4784e-01, -3.2858e-01, -2.8899e-01, ..., -1.9732e-01,
3.7328e-01, 8.8142e-01]],
[[-4.2701e+00, -1.2933e+00, -2.2578e+00, ..., -2.7765e+00,
-3.4294e+00, -8.0392e-01],
[-4.1202e+00, -2.8574e+00, -2.7380e+00, ..., -2.3627e+00,
-3.4273e+00, -5.1955e-01],
[-2.6692e+00, -2.4507e+00, -5.2351e+00, ..., -5.1116e+00,
-6.1142e+00, -1.9173e+00],
...,
[-9.5065e-01, -5.1814e-01, -2.0894e+00, ..., -4.5581e-01,
-1.5723e+00, -6.0321e-01],
[-2.9214e+00, -4.2160e-01, -1.3479e+00, ..., -1.5743e+00,
-1.9958e+00, -7.3317e-02],
[-2.1281e+00, -1.5424e+00, -1.8272e+00, ..., -1.3547e+00,
-7.1109e-01, 1.4884e-01]],
[[ 3.1118e+00, 4.2579e+00, 3.3308e+00, ..., 2.1866e+00,
5.9959e-01, 2.2952e+00],
[ 3.9273e-01, 8.5974e-01, 2.2430e+00, ..., 2.2970e+00,
9.1360e-03, 2.7867e+00],
[ 8.7304e-01, 1.2902e+00, 1.8728e+00, ..., 3.4471e+00,
1.4502e+00, 2.6275e+00],
...,
[ 9.3215e-02, 8.3002e-01, 1.1973e+00, ..., -2.2356e+00,
-2.9395e+00, -1.6456e+00],
[ 2.1685e-01, 2.0554e+00, 1.6711e+00, ..., 8.2128e-01,
-1.1036e-01, 1.2342e-01],
[ 1.3743e+00, 1.3603e+00, 1.7541e+00, ..., 1.2867e+00,
1.4581e+00, 1.5574e+00]],
...,
[[-2.0850e+00, -7.3921e+00, -5.8215e+00, ..., -4.1157e+00,
-3.0716e+00, -3.7814e+00],
[-2.8060e+00, -6.4916e+00, -4.7681e+00, ..., -4.7976e+00,
-4.7357e+00, -5.3376e+00],
[-2.4426e-01, -2.8858e+00, -4.0063e+00, ..., -2.1752e+00,
-1.3059e+00, -2.5881e+00],
...,
[-9.6682e-01, -2.1322e+00, -2.1146e+00, ..., -4.0764e+00,
-3.2928e+00, -2.9204e+00],
[-1.2938e+00, -3.6290e+00, -3.3876e+00, ..., -6.7618e+00,
-4.4193e+00, -4.4041e+00],
[-2.2182e+00, -5.0568e+00, -3.1001e+00, ..., -4.1543e+00,
-2.6180e+00, -3.7697e+00]],
[[-7.4878e-01, -1.8270e-01, -2.0832e-02, ..., 1.0201e+00,
1.0141e+00, 1.7647e+00],
[ 1.8355e+00, 1.0497e+00, 1.2211e+00, ..., 1.3991e+00,
7.6421e-01, 2.5735e+00],
[ 1.2217e+00, -4.3444e-01, 1.5223e-01, ..., -2.1600e-01,
-3.3564e+00, 8.1632e-01],
...,
[ 1.2817e+00, -1.5233e-01, 3.7761e-01, ..., 2.9045e+00,
2.1708e+00, 1.7069e+00],
[ 1.8681e+00, 8.3359e-01, 1.5306e+00, ..., 1.1406e+00,
3.9281e-01, 2.2411e+00],
[ 1.0211e+00, 2.4448e-01, 8.2062e-01, ..., 1.0006e+00,
1.1794e+00, 1.4314e+00]],
[[-4.9751e-01, 6.1445e-01, 1.0397e-01, ..., -3.8276e-01,
-5.8861e-01, 8.9879e-01],
[-1.1085e+00, -1.2215e+00, -1.6891e+00, ..., -4.8501e-01,
4.5762e-02, 5.0880e-01],
[-2.0254e+00, -1.1286e+00, -1.0543e+00, ..., -1.3743e+00,
-1.1873e+00, -2.9699e-02],
...,
[-1.7348e+00, -1.2543e+00, -1.2849e+00, ..., -8.8378e-01,
-2.7495e+00, -6.4318e-01],
[-1.6451e+00, -9.6456e-01, -1.1754e+00, ..., -1.9262e-01,
-1.1739e+00, -5.7619e-01],
[-3.3422e-01, 3.5000e-01, -4.4871e-01, ..., -1.0260e-01,
-1.9652e-01, 6.0490e-01]]],
[[[-1.3402e+00, -9.0557e-01, -3.6823e+00, ..., 2.7433e-01,
2.4782e-01, -1.6230e-01],
[-1.3242e+00, -9.4837e-02, -2.7411e+00, ..., 1.5466e-01,
5.1532e-02, -3.7605e-01],
[-3.1629e+00, 1.5000e+00, 7.4859e-01, ..., -1.3496e-01,
5.7852e-02, 7.6092e-01],
...,
[-1.2261e+00, -2.2657e-01, -2.1857e+00, ..., -8.7087e-01,
1.4256e+00, 4.5546e-01],
[-4.5720e-01, -4.8223e-02, -3.7502e-01, ..., 7.0349e-01,
1.3165e+00, 1.2824e-01],
[ 8.2291e-01, -5.3446e-01, 5.3543e-01, ..., 3.7457e-01,
3.4849e-01, 7.7892e-01]],
[[ 8.8555e-01, -2.8881e-01, -3.2768e+00, ..., -1.9868e+00,
-2.5165e+00, -1.4561e+00],
[ 1.5647e+00, -2.7943e+00, -2.0937e+00, ..., -1.4700e+00,
-2.5431e+00, -1.5925e+00],
[ 6.1684e-01, -1.9686e-01, 4.2773e-01, ..., -2.3462e+00,
-2.4091e+00, -9.7118e-01],
...,
[ 9.2739e-02, 2.2868e-02, -2.6128e-01, ..., -5.2318e+00,
3.3129e-01, 8.8927e-01],
[-2.9335e+00, -1.7966e+00, -6.2050e+00, ..., 1.1663e-01,
4.5187e-01, 6.6291e-01],
[-2.2545e+00, -2.4740e+00, -1.7078e+00, ..., -6.4641e-01,
-9.0370e-01, -3.8630e-01]],
[[-4.8073e-01, 2.0340e-02, -1.4657e+00, ..., 1.8453e+00,
8.9536e-01, 2.1759e+00],
[-4.6802e+00, -2.1787e+00, -5.3848e-01, ..., 9.5331e-01,
5.0891e-01, 2.3522e+00],
[-6.0651e+00, -4.3818e+00, -2.9685e-01, ..., -4.8153e-02,
7.6226e-01, 2.0433e+00],
...,
[ 2.0126e+00, -1.6255e+00, -6.1790e+00, ..., 6.9280e-01,
2.7202e-01, 1.1768e+00],
[ 1.2020e+00, 2.3868e+00, -3.7606e+00, ..., 1.1616e+00,
3.0703e-01, 1.8383e+00],
[ 2.5309e+00, 2.9119e+00, 3.8675e+00, ..., 1.9097e+00,
1.0534e+00, 1.8364e+00]],
...,
[[-4.4436e+00, -7.5213e+00, -7.9906e-01, ..., -2.4369e+00,
-1.9945e+00, -3.1827e+00],
[-5.1964e+00, -1.2243e+01, -1.5396e-01, ..., -3.2045e+00,
-2.9913e+00, -3.5929e+00],
[-1.4432e+00, -7.2230e+00, -5.0685e+00, ..., -2.5752e+00,
-2.8858e+00, -2.6338e+00],
...,
[-2.2394e+00, -3.2234e+00, -1.6489e+00, ..., -7.8360e+00,
-5.5211e+00, -4.1568e+00],
[-2.4389e+00, -5.8664e+00, -3.8545e+00, ..., -7.7005e+00,
-6.3659e+00, -5.0086e+00],
[-2.5911e+00, -5.7641e+00, -3.2830e+00, ..., -4.0526e+00,
-1.8022e+00, -3.4945e+00]],
[[ 7.6665e-01, 1.7402e+00, -4.2088e-01, ..., 1.0095e+00,
1.3851e+00, 1.5668e+00],
[-2.7403e+00, -6.7197e+00, -4.9410e+00, ..., 9.7375e-01,
9.2013e-01, 2.1870e+00],
[ 1.1035e+00, 3.3323e-01, -8.1992e-01, ..., 5.5783e-01,
3.1207e-01, 1.4526e+00],
...,
[ 9.6927e-01, 7.5090e-01, 1.9979e+00, ..., 2.1159e+00,
3.6301e+00, 1.7433e+00],
[ 1.5958e+00, -1.7647e+00, -9.1018e-01, ..., 3.6325e+00,
1.8458e+00, 2.6741e+00],
[ 1.3249e+00, 5.8122e-01, 9.9448e-01, ..., 1.1588e+00,
1.4440e+00, 1.5041e+00]],
[[ 9.7818e-01, -2.5995e+00, -3.9758e+00, ..., -5.7576e-01,
-9.0794e-01, -3.3490e-02],
[ 1.6231e+00, -6.3515e+00, -5.0645e+00, ..., -8.7798e-01,
-6.5629e-01, -3.8869e-01],
[-3.2968e+00, -1.7904e+00, 1.7030e+00, ..., -6.0607e-02,
-1.8430e-01, -8.4553e-02],
...,
[-9.8226e-01, -2.3376e+00, -2.5841e+00, ..., -2.3903e+00,
-6.8777e-03, -3.1565e-01],
[-1.6931e+00, 1.1876e-01, -8.3623e-01, ..., -7.9448e-02,
7.3186e-02, 5.2338e-01],
[-1.2055e-01, 3.5970e-01, -9.8957e-01, ..., 2.9203e-01,
3.1152e-02, 6.1446e-01]]],
[[[ 9.7586e-01, 1.5827e+00, 1.1520e+00, ..., 6.7861e-01,
9.1127e-01, 4.4712e-01],
[-7.5706e-01, -1.9417e+00, -7.4829e-01, ..., -1.5708e+00,
-4.7991e-01, -3.6250e-01],
[ 4.1984e-02, -6.6615e-01, -3.5879e-01, ..., -4.4469e-01,
-6.7541e-01, 6.2073e-01],
...,
[ 4.0020e-01, -5.5549e-01, 1.1406e+00, ..., -1.5820e-01,
-1.1640e-01, 3.3500e-01],
[-2.2658e-01, -1.1230e+00, 9.9572e-02, ..., -5.2049e-02,
9.0084e-02, -1.0035e-01],
[ 7.1856e-01, -2.6275e-01, -1.8171e-01, ..., -3.3230e-01,
-9.4522e-01, 1.8635e-01]],
[[-4.2244e+00, -1.2266e+00, -2.1932e+00, ..., -1.4724e+00,
-2.2320e+00, -1.8567e+00],
[-3.9740e+00, -2.8298e+00, -3.0960e+00, ..., -1.3347e+00,
-2.3450e+00, -2.1337e+00],
[-2.5760e+00, -2.2578e+00, -4.9988e+00, ..., -1.9504e+00,
-2.1485e+00, -1.4129e+00],
...,
[-8.8045e-01, -1.0053e+00, -2.4512e+00, ..., -5.1803e-01,
-9.4112e-01, -8.6625e-01],
[-2.8849e+00, -6.3665e-01, -5.6312e-01, ..., -6.6287e-01,
-1.0568e+00, -1.2108e+00],
[-2.0095e+00, -1.4380e+00, -1.3817e+00, ..., -6.8170e-01,
-6.5087e-01, -5.1300e-01]],
[[ 3.0242e+00, 4.1051e+00, 3.2715e+00, ..., 1.1868e+00,
5.7224e-01, 2.6217e+00],
[ 4.2486e-01, 7.8347e-01, 2.2936e+00, ..., 2.3656e+00,
5.3978e-01, 2.6947e+00],
[ 8.6515e-01, 1.2304e+00, 2.0952e+00, ..., 7.2501e-01,
1.0067e+00, 1.8697e+00],
...,
[ 1.4212e-01, 1.4391e+00, 2.6548e+00, ..., -4.1393e-01,
5.5406e-01, 1.5021e+00],
[ 2.2464e-01, 2.0282e+00, 2.3556e+00, ..., -1.5463e-01,
8.0854e-01, 1.1775e+00],
[ 1.3603e+00, 1.3376e+00, 1.6749e+00, ..., 6.0601e-01,
1.3452e+00, 1.4009e+00]],
...,
[[-2.0765e+00, -7.2592e+00, -5.7888e+00, ..., -5.2652e+00,
-3.7144e+00, -4.4195e+00],
[-2.7376e+00, -6.5010e+00, -4.7181e+00, ..., -4.8449e+00,
-3.9737e+00, -3.3589e+00],
[-1.8601e-01, -2.8645e+00, -3.8733e+00, ..., -8.8798e+00,
-3.5774e+00, -1.3736e+00],
...,
[-9.1566e-01, -2.5134e+00, -2.6796e+00, ..., -5.1991e+00,
-3.2922e+00, -2.6978e+00],
[-1.2237e+00, -3.8968e+00, -3.4929e+00, ..., -5.3395e+00,
-4.3064e+00, -4.2599e+00],
[-2.1230e+00, -4.8787e+00, -3.0896e+00, ..., -3.4774e+00,
-2.5261e+00, -2.3915e+00]],
[[-6.9556e-01, -1.7341e-02, 1.2933e-01, ..., 1.4864e+00,
1.6306e+00, 1.1726e+00],
[ 1.7806e+00, 1.0879e+00, 1.0219e+00, ..., 2.9979e+00,
1.4811e+00, 1.8851e+00],
[ 1.1735e+00, -2.4443e-01, -9.4720e-02, ..., 2.3543e+00,
1.0813e+00, 1.3203e+00],
...,
[ 1.2513e+00, -1.0335e+00, -8.0928e-01, ..., 1.6238e+00,
7.0553e-01, 9.6845e-01],
[ 1.8160e+00, 9.0366e-01, 2.1383e+00, ..., 1.2845e+00,
7.3191e-01, 1.4843e+00],
[ 1.0167e+00, 3.4956e-01, 9.5787e-01, ..., 7.2929e-01,
6.9912e-01, 1.4390e+00]],
[[-5.1784e-01, 5.9330e-01, 1.1394e-01, ..., -6.2016e-01,
-6.6036e-01, -7.2892e-02],
[-1.0105e+00, -1.1502e+00, -1.7990e+00, ..., -2.0445e+00,
-1.2420e+00, -1.1384e+00],
[-1.9549e+00, -1.0215e+00, -1.0250e+00, ..., -7.6935e-01,
-1.1144e+00, -2.4435e-01],
...,
[-1.7057e+00, -9.8652e-01, 5.9333e-01, ..., -5.3485e-01,
-1.5703e+00, -8.6125e-01],
[-1.7353e+00, -8.1253e-01, -8.0049e-02, ..., -7.8880e-01,
-1.6897e+00, -7.2507e-01],
[-3.0376e-01, 4.1637e-01, -2.6756e-01, ..., -2.8548e-01,
-7.8835e-01, -1.1482e-02]]]], device='cuda:0'),
tensor([[[[-2.8663e-01, 1.3049e+00, 3.4774e-01, ..., 3.7273e-01,
3.8809e-01, 3.1005e-01],
[-3.1083e+00, -5.0192e+00, -2.5818e+00, ..., 3.4078e-01,
2.9159e-01, -1.1855e+00],
[-1.1473e+00, -3.7972e+00, -3.6501e+00, ..., -8.2229e-01,
9.9874e-01, 5.8990e-01],
...,
[ 2.9919e-01, -5.0059e-01, 4.6699e-01, ..., -2.5238e+00,
5.7094e-01, 4.6352e-01],
[ 1.2011e-01, -8.8685e-01, 4.9517e-01, ..., -4.3332e+00,
1.8141e+00, -1.0085e-01],
[ 3.3908e-01, -2.0442e-02, -5.0966e-01, ..., -4.8150e+00,
2.0379e+00, 8.0140e-01]],
[[-7.6815e+00, -7.3754e+00, -6.5821e+00, ..., -1.6004e+00,
-2.0959e+00, -1.1248e+00],
[-5.4691e+00, 8.5187e-01, 3.6234e+00, ..., -3.1475e-01,
-1.4713e+00, -3.5878e-01],
[-2.5357e+00, 1.3054e+00, -3.7300e+00, ..., -1.0731e+00,
-1.4505e+00, -7.6620e-01],
...,
[-1.5534e+00, -7.2004e-01, -1.0548e+00, ..., -2.6899e+00,
-4.4208e+00, -9.1603e-01],
[-3.0269e+00, -5.8320e-01, 1.7964e-01, ..., -8.1196e+00,
-8.5566e+00, -7.8537e-01],
[-1.2032e+00, -1.1858e+00, -2.1030e+00, ..., -8.2101e+00,
-8.5794e+00, -8.3334e-01]],
[[ 3.4385e+00, 3.8138e+00, 2.0405e+00, ..., 2.5630e+00,
1.2344e+00, 2.9211e+00],
[-7.8234e-02, -3.6967e+00, -1.8479e+00, ..., 1.7307e+00,
-3.3876e-01, 2.4116e+00],
[-1.0434e+00, -6.2732e+00, -5.2697e+00, ..., 2.1578e+00,
8.6874e-01, 1.8674e+00],
...,
[ 4.5521e-01, 4.5820e-01, -6.5721e-01, ..., -6.3720e+00,
-1.9553e+00, 1.2392e+00],
[ 3.8827e-01, 1.0939e+00, -3.6309e-01, ..., -9.1263e+00,
-3.5596e+00, 8.9179e-01],
[ 1.3961e+00, 1.3946e+00, 9.0347e-01, ..., -6.8694e+00,
-2.8990e-01, 7.0170e-01]],
...,
[[-2.8665e+00, -6.0407e+00, -4.8127e+00, ..., -3.0591e+00,
-2.2037e+00, -4.0829e+00],
[-4.2927e+00, -8.0725e+00, -8.9478e+00, ..., -4.8093e+00,
-3.5935e+00, -4.4485e+00],
[-3.8151e+00, -5.4794e+00, -5.2018e+00, ..., -2.0792e+00,
-1.5945e+00, -1.4986e+00],
...,
[-1.1855e+00, -3.0466e+00, -1.6186e+00, ..., -2.7885e-01,
-6.6821e+00, -4.3036e+00],
[-1.1727e+00, -5.1395e+00, -3.4591e+00, ..., -6.5965e-01,
-1.1964e+01, -6.3841e+00],
[-1.3764e+00, -5.2922e+00, -3.2585e+00, ..., -3.3680e-01,
-9.6314e+00, -5.1844e+00]],
[[-2.6438e+00, -6.3630e+00, -3.9616e+00, ..., 8.3517e-01,
1.0465e+00, 1.2242e+00],
[ 3.7114e-01, -1.6501e-01, 3.4095e-01, ..., 1.4606e+00,
2.1576e+00, 2.4965e+00],
[-8.3539e-01, -2.4973e-02, -1.5174e+00, ..., 1.2297e+00,
1.7911e+00, 1.9422e+00],
...,
[ 8.5797e-01, -4.4522e-01, 5.0915e-01, ..., -2.7060e+00,
-1.9549e+00, 2.6436e+00],
[ 9.7634e-01, 7.7285e-01, 1.8383e+00, ..., -3.6821e+00,
-4.6337e+00, 3.5766e+00],
[ 7.0155e-01, -2.8929e-01, 1.1823e-01, ..., -2.0003e+00,
-3.5683e+00, 1.7982e+00]],
[[-2.4781e+00, -3.5978e-02, -9.5297e-01, ..., -2.2926e-01,
-4.3510e-02, 8.7020e-01],
[-3.8834e+00, 1.1594e+00, 3.6538e+00, ..., 1.1757e-01,
-1.7630e-01, 3.1671e-01],
[-1.1610e+00, 1.2526e+00, -4.8044e-01, ..., -3.3997e-01,
1.6399e-01, 2.2231e-01],
...,
[-1.5417e+00, -1.7804e+00, -9.7316e-01, ..., 7.2523e-01,
-9.1003e-01, -8.8400e-01],
[-1.4571e+00, -1.8526e+00, -2.2186e-01, ..., 6.1527e-01,
-1.0134e+00, -4.5479e-01],
[-6.0959e-02, 1.2824e-02, -5.9854e-01, ..., 4.1367e-02,
-1.3049e+00, -7.5228e-02]]],
[[[ 9.8576e-01, 1.3922e+00, 1.0880e+00, ..., 3.0151e-01,
4.0756e-01, 2.8615e-01],
[-1.0861e+00, -2.5130e+00, -1.6361e+00, ..., 2.8890e-01,
3.2885e-01, -1.2627e+00],
[-1.2399e+00, -6.1630e-01, 1.0327e+00, ..., -3.6360e-01,
1.1394e+00, 7.0609e-01],
...,
[-1.9142e-01, -1.0178e+00, 1.3079e+00, ..., 2.6777e-01,
9.0660e-01, 5.1561e-01],
[ 1.0355e+00, 3.2535e-01, 1.4583e+00, ..., 1.2184e-01,
1.4080e+00, -1.3908e-01],
[-6.8982e-02, -2.4525e-01, -9.1477e-01, ..., -2.1495e-01,
3.8331e-01, 6.1425e-01]],
[[-3.4346e+00, -1.9810e+00, -2.1436e+00, ..., -1.5767e+00,
-2.0321e+00, -1.0116e+00],
[-4.3934e+00, -4.1323e+00, -1.7330e+00, ..., -2.4005e-01,
-1.3410e+00, -3.3142e-01],
[-2.6768e+00, -1.5917e+00, -3.2460e+00, ..., -1.0931e+00,
-1.3715e+00, -8.4220e-01],
...,
[-3.7096e+00, -5.4646e+00, -4.0055e+00, ..., -2.3847e-01,
1.7601e-01, 4.2194e-01],
[-1.4292e+00, -1.0992e+00, -7.2377e-01, ..., 3.1449e-02,
-4.8232e-01, 2.2934e-01],
[-4.0956e-01, -9.1631e-01, -8.9628e-01, ..., -3.8511e-01,
-8.2201e-01, -6.1728e-01]],
[[ 2.2089e+00, 2.2464e+00, 1.8309e+00, ..., 2.8798e+00,
1.2861e+00, 2.8857e+00],
[-7.0213e-02, 4.5488e-01, 1.7100e+00, ..., 1.7592e+00,
-2.3460e-01, 2.3905e+00],
[-3.3907e+00, -3.1799e+00, -7.4118e-01, ..., 6.6742e-01,
4.7467e-01, 1.8362e+00],
...,
[-2.6069e+00, -2.0694e+00, -1.5276e+00, ..., 1.7054e+00,
-1.8258e-01, 1.5350e+00],
[-4.3481e-01, 1.2862e+00, 1.4340e+00, ..., 9.7118e-01,
7.4151e-01, 1.5915e+00],
[ 6.5588e-01, 4.3258e-01, 1.1445e+00, ..., 1.5660e+00,
1.1469e+00, 2.0748e+00]],
...,
[[-2.9497e+00, -8.4576e+00, -6.7707e+00, ..., -3.3497e+00,
-2.1934e+00, -3.9122e+00],
[-2.6788e+00, -8.5343e+00, -6.3969e+00, ..., -5.6638e+00,
-3.4396e+00, -4.1620e+00],
[-1.7048e+00, -6.7127e+00, -6.6892e+00, ..., -3.7975e+00,
-2.0398e+00, -1.5377e+00],
...,
[-3.1947e+00, -9.2333e+00, -7.7145e+00, ..., -5.0708e+00,
-4.1563e+00, -1.6028e+00],
[-3.9613e+00, -1.0593e+01, -7.1588e+00, ..., -6.2011e+00,
-3.5447e+00, -3.8617e+00],
[-3.1256e+00, -7.6961e+00, -3.1590e+00, ..., -3.5427e+00,
-2.1747e+00, -2.8722e+00]],
[[-7.9366e-01, 5.6879e-01, 9.4360e-01, ..., 1.3631e+00,
1.0660e+00, 1.2216e+00],
[-5.3745e-01, -6.9714e-01, 1.3769e+00, ..., 2.0254e+00,
2.2796e+00, 2.4797e+00],
[ 5.0926e-01, -4.2096e+00, -2.1269e+00, ..., 1.6617e+00,
1.6205e+00, 1.8629e+00],
...,
[ 1.4557e-02, -2.2061e+00, 5.6931e-01, ..., 2.4435e+00,
2.2469e+00, 1.7451e+00],
[ 2.2284e+00, 2.0104e+00, 2.8668e+00, ..., 1.7138e+00,
1.5290e+00, 2.4959e+00],
[ 7.9761e-01, 6.0733e-01, 1.1691e+00, ..., 1.2011e+00,
1.3566e+00, 1.3668e+00]],
[[-1.2401e-01, -1.2444e-01, -3.8727e-01, ..., -8.5581e-02,
-7.0634e-02, 8.8947e-01],
[-7.4284e-01, -2.1429e+00, -1.6048e+00, ..., 1.3717e-01,
-1.2059e-01, 3.0257e-01],
[-1.9908e+00, 9.2571e-01, 1.0177e+00, ..., 3.7661e-01,
2.1110e-01, 2.5428e-01],
...,
[-2.1467e+00, -3.5341e+00, -3.1798e+00, ..., 5.3017e-02,
-7.4593e-02, -3.2169e-01],
[-2.0542e+00, -2.4881e+00, -7.3666e-01, ..., -4.5783e-01,
-3.0887e-01, 3.3761e-01],
[-3.8103e-01, -1.0651e+00, -5.5103e-01, ..., 4.6143e-02,
1.4418e-01, 3.4024e-01]]],
[[[ 1.0891e+00, 1.7449e+00, 9.9888e-01, ..., -1.5120e-01,
3.5042e-01, 1.0816e+00],
[-8.3666e-01, -2.3823e+00, -2.5417e+00, ..., -2.4359e+00,
-1.4369e+00, -1.7149e+00],
[-2.6463e-01, -1.8543e+00, -1.6557e+00, ..., 7.3966e-02,
-4.5559e-01, 4.8553e-01],
...,
[-2.9825e-02, -1.1843e+00, -1.0731e+00, ..., -6.7628e-02,
-1.1866e+00, -4.5648e-02],
[-6.2457e-01, -1.8046e+00, -1.2347e+00, ..., -1.2242e+00,
-2.7273e-01, -4.9131e-02],
[-2.6156e-01, -1.7176e+00, -1.7225e+00, ..., -9.0267e-01,
-1.4009e+00, -2.7493e-01]],
[[-3.7814e+00, -2.0006e+00, -3.9316e+00, ..., -1.2466e+00,
-2.3570e+00, -1.2178e+00],
[-4.6307e+00, -5.6179e+00, -4.6832e+00, ..., -3.1048e+00,
-2.9617e+00, -9.4378e-01],
[-3.4525e+00, -3.9011e+00, -1.2704e+00, ..., -5.7159e+00,
-2.6711e+00, -1.0662e+00],
...,
[-1.5399e+00, -1.6865e+00, -2.6827e+00, ..., -5.6069e+00,
-5.6120e+00, -1.5293e+00],
[-2.9283e+00, -1.5860e+00, -2.7286e+00, ..., -4.4774e+00,
-2.7670e+00, -9.5293e-01],
[-2.3839e+00, -2.0292e+00, -2.6453e+00, ..., -2.0252e+00,
-1.7532e+00, -9.0337e-01]],
[[ 2.9545e+00, 4.3825e+00, 3.7423e+00, ..., 2.7965e+00,
6.1246e-01, 3.0882e+00],
[ 6.9813e-01, 2.1594e+00, 3.8054e+00, ..., 2.7980e+00,
-1.0311e+00, 2.6186e+00],
[ 1.6782e+00, 1.9019e+00, 1.3100e+00, ..., 1.3412e+00,
1.5810e+00, 1.9774e+00],
...,
[ 9.3606e-01, 1.0380e+00, 1.1512e+00, ..., -4.6377e+00,
1.8128e-01, 2.4459e+00],
[ 6.8734e-01, 1.8606e+00, 1.3723e+00, ..., -3.6739e-01,
2.0953e+00, 1.5049e+00],
[ 1.7110e+00, 1.5942e+00, 1.9432e+00, ..., 1.3867e+00,
1.4472e+00, 1.5970e+00]],
...,
[[-1.8415e+00, -7.6676e+00, -5.7122e+00, ..., -1.7620e+00,
-2.7126e+00, -5.7307e+00],
[-2.4339e+00, -7.3997e+00, -4.4115e+00, ..., -5.4564e+00,
-4.3217e+00, -4.2640e+00],
[-9.6124e-01, -5.2719e+00, -5.2414e+00, ..., -1.2674e+01,
-6.7744e+00, -2.6040e+00],
...,
[-2.6178e+00, -5.4748e+00, -4.4026e+00, ..., -8.6163e+00,
-8.4627e+00, -6.3140e+00],
[-2.8586e+00, -6.0147e+00, -5.3974e+00, ..., -6.2621e+00,
-6.3190e+00, -7.7566e+00],
[-1.5733e+00, -4.2698e+00, -2.7344e+00, ..., -4.6465e+00,
-3.6736e+00, -3.2259e+00]],
[[-2.7570e-01, -6.1793e-01, -4.6680e-01, ..., 1.3608e+00,
1.2307e+00, 1.3838e+00],
[ 1.1055e+00, -4.9132e-01, -1.3057e+00, ..., 1.6455e+00,
3.3221e+00, 2.9758e+00],
[ 3.4744e-01, 3.4584e-01, 4.7168e-01, ..., -1.2239e+00,
3.6414e+00, 2.4557e+00],
...,
[ 1.3810e+00, 2.5031e-01, -1.9200e-01, ..., -7.1357e-01,
-2.4122e-01, 2.4055e+00],
[ 1.9856e+00, 1.7311e+00, 1.0080e+00, ..., 1.1820e+00,
2.4418e+00, 2.6157e+00],
[ 4.3086e-01, 8.2226e-01, 7.1096e-01, ..., 1.0864e+00,
1.0717e+00, 1.7092e+00]],
[[-6.4937e-01, 9.6371e-02, -1.0560e+00, ..., -3.1649e-01,
3.1756e-01, 1.4076e+00],
[-1.7873e+00, -2.3963e+00, -1.9295e+00, ..., -2.9799e+00,
-5.3749e-01, 1.2234e-01],
[-3.0026e+00, -1.2543e+00, 1.0263e+00, ..., -1.0582e+00,
-4.3440e-01, 7.4408e-01],
...,
[-2.0037e+00, -1.8781e+00, -1.4880e+00, ..., -1.8803e+00,
-5.4933e+00, -2.0156e+00],
[-1.6659e+00, -1.9826e+00, -1.6190e+00, ..., -3.6265e+00,
-1.8126e+00, -4.3637e-01],
[-8.6636e-01, -8.7490e-01, -1.1409e+00, ..., 2.1111e-01,
-4.5379e-01, 7.5332e-02]]],
[[[ 1.3563e-01, 1.2250e+00, 9.0580e-01, ..., -8.6959e-01,
-5.3074e-01, 5.4316e-01],
[-2.1948e+00, -3.7384e+00, -2.9726e+00, ..., 5.2979e-01,
-2.4244e+00, -2.0177e+00],
[-5.7368e-01, -4.5133e+00, -4.1742e+00, ..., -1.5900e+00,
-7.0465e-01, 1.6724e-01],
...,
[ 4.6707e-02, -2.3729e+00, 1.1959e+00, ..., -2.6943e+00,
-1.6197e+00, -9.6574e-01],
[-4.8206e-01, -5.1755e+00, 2.8516e-01, ..., -5.8406e+00,
-8.9923e-01, 6.9014e-01],
[ 4.0485e-02, -3.1661e+00, 1.0972e+00, ..., -1.2376e+00,
-5.2505e+00, -1.2669e+00]],
[[-6.5905e+00, -3.6238e+00, -1.0909e+01, ..., -2.0748e+00,
-2.4110e+00, -1.8265e+00],
[-5.6207e+00, -1.1717e+00, -4.5915e+00, ..., -8.6776e-01,
-1.8176e+00, -8.3183e-01],
[-4.8095e+00, 1.5618e+00, -2.2372e+00, ..., -2.1002e+00,
1.6134e+00, -1.7933e+00],
...,
[-1.3350e+00, -8.6097e+00, -1.5207e+01, ..., -3.0606e+00,
-3.5852e+00, -4.0757e+00],
[-2.2160e+00, -5.2704e+00, -7.9387e+00, ..., -3.1331e+00,
-4.5999e-01, -1.5757e+00],
[-1.7515e+00, -3.9681e+00, -8.0774e+00, ..., -8.1498e+00,
-5.2303e+00, -2.3796e+00]],
[[ 3.7869e+00, 5.2285e+00, 6.2486e+00, ..., 3.5996e+00,
2.3803e+00, 3.0446e+00],
[-2.3896e-01, -2.4400e-02, -7.1764e-01, ..., 7.8378e-01,
-1.6019e+00, -1.8985e+00],
[-1.1318e+00, 1.6658e-02, 4.1696e-01, ..., -5.3926e+00,
-2.0325e+00, -2.5294e+00],
...,
[ 1.5054e+00, 5.2546e+00, -4.7847e+00, ..., -6.4819e+00,
-2.0594e+00, -2.3364e+00],
[ 5.8331e-01, 5.4987e+00, -9.5860e-01, ..., -9.6566e+00,
-2.0305e+00, -3.4917e+00],
[ 1.8083e+00, 3.4095e+00, 3.7345e+00, ..., -4.6533e+00,
-4.9969e+00, -3.5878e+00]],
...,
[[-3.2488e+00, -8.7450e+00, -5.9979e+00, ..., -1.4699e+00,
-1.0217e+00, -5.5457e+00],
[-4.9591e+00, -1.0779e+01, -9.0662e+00, ..., -6.2352e+00,
-4.2845e+00, -5.5783e+00],
[-4.4963e+00, -8.2184e+00, -6.4229e+00, ..., -4.6599e+00,
-1.0004e+01, -8.8305e+00],
...,
[-2.5029e+00, -8.1222e+00, -1.2068e+01, ..., -9.6903e+00,
-1.2370e+01, -1.8089e+00],
[-3.3652e+00, -7.4593e+00, -1.1110e+01, ..., -9.1259e+00,
-1.2641e+01, -1.0944e+01],
[-2.2504e+00, -5.9887e+00, -5.1786e+00, ..., -4.8506e+00,
-6.3024e+00, -7.3292e+00]],
[[-1.5205e+00, -2.9112e+00, -6.4797e+00, ..., 1.2263e+00,
6.1434e-01, 7.0471e-01],
[ 1.8502e+00, 2.0086e+00, 9.3549e-01, ..., -2.2877e+00,
-3.0944e-01, 1.0786e+00],
[-3.0907e-01, 1.8011e+00, 2.0136e+00, ..., -2.7883e+00,
-7.2423e-01, 7.6752e-01],
...,
[ 1.2098e+00, -4.1536e+00, 1.4216e-01, ..., -3.8878e+00,
-8.3877e+00, -4.6208e+00],
[ 2.2166e+00, -1.2011e+00, 2.7628e+00, ..., 3.0544e+00,
-1.4403e+00, -2.1436e+00],
[ 7.5641e-01, -1.9446e+00, -3.1868e+00, ..., 1.1062e+00,
8.7576e-01, -7.4293e-02]],
[[-1.4664e+00, -4.8967e-01, -1.5413e+00, ..., 1.9465e-01,
4.6394e-03, 8.3642e-01],
[-2.0520e+00, 3.7601e-01, -2.2848e+00, ..., -6.7511e-01,
-3.0843e+00, -5.9819e-01],
[-1.8230e+00, 1.6049e+00, -8.0635e-01, ..., 2.1498e+00,
1.6947e+00, 1.2954e+00],
...,
[-2.3154e+00, -3.9856e+00, -4.5679e+00, ..., -3.5979e+00,
-3.8912e+00, -3.7932e+00],
[-1.9193e+00, -3.1981e+00, -2.7790e+00, ..., -3.3141e+00,
-4.0879e+00, -1.5248e+00],
[-7.2078e-01, -1.2583e+00, 6.4870e-01, ..., -2.0853e+00,
-5.5834e+00, -3.1719e+00]]],
[[[ 5.5463e-01, 5.3820e-01, 2.2839e-01, ..., 2.4536e-03,
-1.8684e-02, -3.8413e-01],
[-4.9087e-01, -3.0743e-01, 6.9456e-03, ..., 2.9807e-01,
4.7572e-01, -1.8544e-01],
[ 1.2748e-01, -1.1861e-01, -5.0146e-01, ..., -3.0051e-01,
1.6777e-01, 8.9299e-01],
...,
[-1.3740e+01, -9.6233e+00, -6.3262e-01, ..., 1.1187e+00,
6.5479e-01, 4.6325e-01],
[-8.2570e+00, -1.6444e+01, -3.3815e+00, ..., 1.9536e+00,
1.7515e+00, 1.3707e-01],
[ 1.0675e-02, -1.2066e+01, -1.9583e+00, ..., 6.8384e-01,
5.6369e-01, 9.4185e-01]],
[[-3.9500e+00, -1.7214e+00, -2.9266e+00, ..., -2.5147e+00,
-3.1648e+00, -1.3944e+00],
[-3.0364e+00, -1.5345e+00, -2.3041e+00, ..., -1.3577e+00,
-2.3774e+00, -1.0858e+00],
[-3.1724e+00, -4.3344e+00, -4.6709e+00, ..., -2.0918e+00,
-2.2816e+00, -5.6416e-01],
...,
[-8.3926e+00, -6.7732e+00, -1.0017e+01, ..., -1.7201e+00,
-5.7688e-01, 6.0651e-01],
[-1.7570e+01, -2.5688e+01, -2.9513e+01, ..., 5.5366e-01,
-4.3571e-01, 4.7923e-01],
[-1.5991e+01, -2.7239e+01, -2.8504e+01, ..., -3.4755e-01,
-1.1066e+00, -5.1730e-01]],
[[ 2.5881e+00, 2.7944e+00, 2.4172e+00, ..., 2.0458e+00,
9.2274e-01, 2.1385e+00],
[ 6.5096e-01, 5.6317e-01, 1.2885e+00, ..., 7.5726e-01,
4.5493e-01, 2.2202e+00],
[ 7.2582e-01, 7.1744e-01, 1.9883e+00, ..., -5.9412e-01,
4.9198e-01, 1.8458e+00],
...,
[-2.3681e+01, -3.1397e-01, 1.9054e+00, ..., -2.3412e+00,
2.8192e-01, 1.8961e+00],
[-3.8455e+01, -2.0397e+01, -1.3109e+01, ..., -6.2027e-01,
7.5433e-01, 2.0034e+00],
[-1.4602e+01, -1.0724e+01, -3.5767e+00, ..., 4.5801e-01,
7.8700e-01, 1.9722e+00]],
...,
[[-2.4853e+00, -5.9031e+00, -3.9423e+00, ..., -2.5892e+00,
-2.1292e+00, -2.9932e+00],
[-2.9944e+00, -7.3314e+00, -5.4941e+00, ..., -4.7159e+00,
-3.6864e+00, -4.0322e+00],
[-1.7742e+00, -7.1482e+00, -4.8579e+00, ..., -4.3600e+00,
-2.7311e+00, -2.2171e+00],
...,
[-4.4534e+00, -3.8254e+01, -2.5402e+01, ..., -9.0348e+00,
-2.0846e+00, -2.2422e+00],
[-6.5251e+00, -3.2385e+01, -2.7788e+01, ..., -9.3346e+00,
-3.0825e+00, -4.8531e+00],
[-7.6771e+00, -2.7078e+01, -2.1050e+01, ..., -7.8132e+00,
-1.7036e+00, -3.6076e+00]],
[[-1.5004e-01, 8.5435e-01, 4.1823e-01, ..., 8.0632e-01,
1.0709e+00, 1.7155e+00],
[ 1.1432e+00, 9.3448e-01, 1.6880e+00, ..., 1.1906e+00,
1.2020e+00, 2.3174e+00],
[ 2.8153e-01, -7.8728e-01, -1.7813e+00, ..., 1.1362e+00,
4.8016e-01, 1.5054e+00],
...,
[ 5.1801e-01, -1.9966e+01, -3.2796e+01, ..., 3.3353e+00,
1.9348e+00, 1.7126e+00],
[ 1.7113e+00, -1.3714e+01, -2.1771e+01, ..., 3.1465e+00,
1.6453e+00, 2.7691e+00],
[ 2.6463e+00, -2.7923e+00, -8.2570e+00, ..., 1.6097e+00,
1.4956e+00, 1.4278e+00]],
[[-4.8892e-01, -7.2173e-01, -1.1316e+00, ..., -6.2697e-01,
-9.1558e-01, 2.3938e-01],
[-1.1233e+00, -2.0426e+00, -1.0410e+00, ..., -4.0997e-01,
-1.9757e-01, 3.2854e-01],
[-2.3391e+00, -2.0639e+00, -4.7801e-01, ..., -6.7244e-02,
-2.7320e-01, 7.1428e-02],
...,
[-9.5505e-01, -5.9393e+00, -2.0899e+00, ..., -3.8756e+00,
-1.3562e-01, -4.4673e-01],
[-4.0103e+00, -1.4653e+01, -1.6071e+01, ..., -2.4297e+00,
-2.4606e-02, 7.1355e-01],
[-4.2462e+00, -1.7141e+01, -6.6366e+00, ..., -4.9473e-01,
1.2700e-01, 6.4088e-01]]]], device='cuda:0'),
tensor([[[[ 8.5715e-01, 1.4959e+00, 1.2270e+00, ..., 1.4842e+00,
1.1273e+00, 1.0264e+00],
[-9.2686e-01, -2.1159e+00, -1.2018e+00, ..., -4.6437e-01,
-6.5319e-01, -1.7003e+00],
[-9.6728e-02, -6.6631e-01, 3.0262e-01, ..., -3.6260e-01,
6.7676e-01, 8.2224e-01],
...,
[ 6.1203e-01, -8.3652e-01, -1.6715e+00, ..., 1.4683e+00,
2.9658e-01, 6.2372e-01],
[ 2.6491e-01, -5.4615e-01, -1.3522e+00, ..., 1.0509e+00,
5.0045e-01, -8.8440e-02],
[ 4.3613e-01, 1.8753e-02, -1.1356e-01, ..., 1.9089e+00,
-3.0743e-01, 6.5825e-01]],
[[-3.7781e+00, -6.6486e-01, -1.5230e+00, ..., -1.5178e+00,
-2.4031e+00, -1.1395e+00],
[-4.1055e+00, -2.9071e+00, -1.9061e+00, ..., -2.3432e+00,
-3.2395e+00, -9.5974e-01],
[-2.4752e+00, -1.6956e+00, -2.0265e+00, ..., -1.7682e+00,
-2.9145e+00, -1.0912e+00],
...,
[-9.2903e-01, -2.1018e+00, 7.6352e-02, ..., -3.4172e+00,
-1.2907e+00, -1.3760e-02],
[-1.0504e+00, 5.9114e-02, 1.6516e+00, ..., -4.0802e+00,
-1.2474e+00, 3.3528e-02],
[ 6.1647e-01, 5.3375e-01, 7.6985e-01, ..., -4.2168e+00,
-3.4801e-01, 4.7875e-01]],
[[ 2.9492e+00, 4.3569e+00, 3.3938e+00, ..., 2.9193e+00,
1.0320e+00, 3.0983e+00],
[ 2.5637e-01, 1.0790e+00, 2.2394e+00, ..., 9.5004e-01,
-1.0540e+00, 2.7624e+00],
[ 1.0062e+00, 1.6054e+00, 1.6334e+00, ..., 1.2704e+00,
-4.8437e-01, 2.1864e+00],
...,
[ 9.8165e-01, 1.5604e+00, 4.5470e-01, ..., -1.9050e+00,
1.2462e-01, 1.3793e+00],
[ 8.1909e-01, 1.9189e+00, -3.3188e+00, ..., -3.1653e-01,
9.5439e-02, 1.0811e+00],
[ 1.9299e+00, 9.6829e-01, -3.0921e+00, ..., 2.4080e-01,
-1.1836e+00, 1.1144e+00]],
...,
[[-1.9046e+00, -7.2250e+00, -5.4328e+00, ..., -4.1199e+00,
-3.3064e+00, -5.6808e+00],
[-2.4431e+00, -6.8498e+00, -4.0462e+00, ..., -3.5915e+00,
-2.5482e+00, -4.3059e+00],
[-1.0664e+00, -4.6433e+00, -3.8231e+00, ..., -2.7480e+00,
-3.3142e+00, -2.3988e+00],
...,
[-2.2331e+00, -5.5877e+00, -2.4073e+00, ..., -5.0999e+00,
-5.2678e+00, -3.5661e+00],
[-2.2261e+00, -6.8553e+00, -2.4348e+00, ..., -8.8307e+00,
-7.9018e+00, -4.7718e+00],
[-2.2910e+00, -6.5070e+00, -2.3228e+00, ..., -9.7273e+00,
-6.5304e+00, -4.0623e+00]],
[[-1.2458e-01, 3.4014e-01, 5.1414e-01, ..., 6.3419e-01,
8.9384e-01, 1.4424e+00],
[ 1.6125e+00, 1.1433e+00, 1.8388e+00, ..., 1.5655e+00,
2.4400e+00, 3.0722e+00],
[ 1.1511e+00, 2.5998e-01, 1.4955e-01, ..., 1.7985e+00,
1.5137e+00, 2.4350e+00],
...,
[ 1.0580e+00, -1.4471e-01, 6.0625e-01, ..., 2.8519e-01,
1.1394e+00, 1.5457e+00],
[ 1.6511e+00, 9.7350e-01, -1.4671e-01, ..., -6.5990e-01,
1.9873e+00, 2.1557e+00],
[ 1.2041e+00, 9.4086e-01, 1.3888e+00, ..., -1.3082e+00,
2.4470e+00, 1.3762e+00]],
[[-5.7374e-01, 3.2049e-01, 1.5419e-01, ..., 6.1930e-01,
5.7582e-01, 1.3830e+00],
[-1.3700e+00, -1.8719e+00, -1.7367e+00, ..., -9.6745e-01,
-5.3262e-01, 1.5608e-01],
[-2.1335e+00, -7.7079e-01, -1.3335e-01, ..., 3.9221e-01,
1.6555e-01, 7.4116e-01],
...,
[-1.6150e+00, -2.8092e+00, 7.0830e-01, ..., -2.3992e+00,
-2.8823e+00, -6.0853e-01],
[-1.2009e+00, -1.5237e+00, 9.3171e-01, ..., -6.4474e-01,
-3.4353e+00, 8.6469e-02],
[ 2.8294e-01, 2.5574e-01, -5.6404e-01, ..., -6.1414e-01,
-2.2133e+00, 7.7702e-01]]],
[[[ 8.7261e-01, 1.0494e+00, -2.2116e-01, ..., 1.4918e+00,
1.3490e-02, -5.4318e-01],
[-4.4996e-01, -2.4891e+00, -2.4528e+00, ..., 2.8906e-01,
1.6618e+00, -2.2506e-01],
[-6.7042e-02, -2.5077e+00, 3.9992e-01, ..., -1.2640e+00,
8.1556e-01, 9.0997e-01],
...,
[ 6.1748e-01, 5.3513e-02, -4.3345e+00, ..., -3.6581e+00,
-1.7965e+00, 5.1771e-01],
[-3.9425e-01, -9.4209e-02, 1.2908e-02, ..., -6.9693e-01,
-5.4616e-01, -4.0218e-01],
[-1.9139e-01, -1.6600e+00, -2.5012e+00, ..., -1.6506e+00,
-1.4454e+00, -1.8609e-01]],
[[-4.1381e+00, -3.2874e+00, -6.2812e+00, ..., -3.8503e+00,
-3.9390e+00, -1.2190e+00],
[-3.6651e+00, -8.5063e+00, -1.0820e+01, ..., -3.5146e+00,
-2.7875e+00, -1.2028e-01],
[-3.1595e+00, -8.7777e+00, -5.8986e+00, ..., -3.1676e+00,
-3.4559e+00, -4.6883e-01],
...,
[-4.7764e+00, -7.2953e+00, -1.3532e+01, ..., -9.9351e+00,
-5.4171e+00, -3.4500e-01],
[-3.3597e+00, -2.8152e+00, -6.8795e+00, ..., -1.5226e+00,
-2.2910e+00, -7.3692e-01],
[-2.1326e+00, -1.9387e+00, -4.6029e+00, ..., -1.3212e+00,
-1.8794e+00, -9.6484e-01]],
[[ 2.2782e+00, 3.4655e+00, -7.6374e-02, ..., 1.3868e+00,
1.5252e+00, 2.9494e+00],
[ 4.3849e-01, 4.0886e+00, -3.7542e+00, ..., 1.0591e+00,
-6.4865e-02, 2.2420e+00],
[ 1.7059e+00, 4.4544e+00, -6.2949e+00, ..., 6.2680e-01,
1.6825e-01, 1.8625e+00],
...,
[ 2.9946e+00, -4.7159e-01, -3.0699e+00, ..., -1.1536e+01,
8.3871e-01, 2.0733e+00],
[ 1.1441e+00, 3.5385e+00, 5.6608e+00, ..., -2.2986e+00,
1.0879e+00, 1.0476e+00],
[ 1.8088e+00, 1.6581e+00, 2.9842e+00, ..., -6.8415e-01,
1.7826e+00, 1.7996e+00]],
...,
[[-1.6248e+00, -6.8564e+00, -7.7657e+00, ..., -2.9807e+00,
-3.9500e+00, -4.0687e+00],
[-2.4180e+00, -9.6742e+00, -1.4667e+01, ..., -1.0186e+00,
-7.7935e+00, -6.5640e+00],
[-1.4919e+00, -7.7654e+00, -1.0477e+01, ..., -2.1740e+00,
-6.5721e+00, -3.8864e+00],
...,
[-3.1698e+00, -8.5958e+00, -9.5989e+00, ..., -1.9240e+01,
-7.8328e+00, -7.7823e+00],
[-3.5934e+00, -1.0078e+01, -1.0499e+01, ..., -2.2414e+01,
-9.7267e+00, -8.8476e+00],
[-1.5956e+00, -4.3327e+00, -3.3464e+00, ..., -9.3658e+00,
-2.6263e+00, -3.1896e+00]],
[[-8.3383e-01, -2.6627e+00, -3.1740e+00, ..., -1.4898e+00,
1.2404e+00, 1.9663e+00],
[ 1.9821e-01, -5.6688e+00, -5.7058e+00, ..., -8.9115e-01,
2.2197e+00, 3.1150e+00],
[-7.4792e-01, -5.1442e+00, -4.9425e+00, ..., -2.9424e-01,
1.9373e+00, 2.3211e+00],
...,
[-3.0010e+00, -2.5959e+00, -2.6761e+00, ..., 1.0942e+00,
1.0307e+00, 2.7878e+00],
[ 1.1014e+00, 2.4244e+00, 1.9151e+00, ..., 6.0080e+00,
2.4537e+00, 2.4287e+00],
[ 5.4117e-01, 8.9808e-01, -1.3715e+00, ..., 2.2986e+00,
6.6502e-01, 1.6392e+00]],
[[-8.4638e-01, -2.2406e+00, -4.2202e+00, ..., 4.2417e-01,
-3.6280e-01, 5.4384e-01],
[-1.7376e+00, -3.6097e+00, -3.6065e+00, ..., -1.8965e-02,
5.3076e-01, 8.3873e-01],
[-2.6724e+00, -3.7210e+00, 1.1520e+00, ..., 4.5793e-01,
-1.5740e-02, -1.9245e-01],
...,
[-2.4231e-01, -9.7157e-01, -4.2634e+00, ..., -9.1786e+00,
-4.3587e+00, -1.3205e+00],
[-1.5230e+00, -3.0796e+00, -1.7865e+00, ..., -7.6056e+00,
-6.5801e-02, -2.4649e-01],
[-8.3691e-01, -8.3068e-01, -1.5200e+00, ..., -3.9824e+00,
-8.1748e-01, -5.9999e-02]]],
[[[ 5.8347e-01, 8.6590e-01, 5.5270e-01, ..., 3.4380e-01,
3.9952e-01, -2.6782e-02],
[-2.7083e-01, -7.6871e-01, -3.7107e-01, ..., 7.5349e-02,
8.3768e-02, -3.2999e-01],
[-1.3146e-01, 3.3253e-02, 4.3319e-01, ..., 4.8906e-02,
3.5392e-01, 8.2589e-01],
...,
[ 3.4541e-02, -9.4363e-01, -1.2049e+00, ..., -6.9502e-01,
7.3795e-01, 8.8443e-01],
[-4.3500e-01, -2.5418e+00, -1.9509e+00, ..., 7.0199e-01,
7.4446e-01, 1.5009e-01],
[ 4.3688e-01, -1.0307e+00, -1.2542e+00, ..., 3.2618e-01,
-6.0467e-01, 2.0042e-01]],
[[-2.5094e+00, -6.8098e-01, -1.9551e+00, ..., -1.7496e+00,
-2.3366e+00, -1.5603e+00],
[-2.6965e+00, -1.1053e+00, -2.1446e+00, ..., -1.4340e+00,
-2.4404e+00, -1.7750e+00],
[-2.8738e+00, -2.7321e+00, -2.7968e+00, ..., -1.6185e+00,
-2.0247e+00, -1.3170e+00],
...,
[-1.0196e+00, -1.0242e+00, -3.3151e+00, ..., -1.7877e+00,
-1.2979e+00, 1.1827e-01],
[-2.3137e+00, -5.0125e-01, -1.6471e+00, ..., -2.2822e+00,
-1.1582e+00, -2.0406e-01],
[-1.0725e+00, -1.2832e+00, -4.7706e-01, ..., -3.6168e+00,
-1.6159e+00, -6.3092e-01]],
[[ 1.9942e+00, 2.5920e+00, 1.6899e+00, ..., 1.5793e+00,
7.9879e-01, 2.1342e+00],
[-1.9093e-01, 4.0774e-01, 6.9197e-01, ..., 6.7667e-01,
3.1645e-01, 2.3603e+00],
[ 2.0634e-01, 1.0310e+00, 1.5846e+00, ..., -5.3998e-02,
7.4552e-01, 1.9563e+00],
...,
[ 1.1401e+00, 1.4486e+00, 2.4036e+00, ..., 9.9273e-01,
-1.4119e-01, 1.3228e+00],
[ 8.0207e-01, 2.6925e+00, 2.7675e+00, ..., -9.5337e-01,
-3.8034e-01, 9.9597e-01],
[ 1.8049e+00, 1.6724e+00, 9.2738e-01, ..., 9.8862e-01,
2.4030e-01, 1.3047e+00]],
...,
[[-2.0830e+00, -5.0752e+00, -3.3095e+00, ..., -2.1397e+00,
-1.6486e+00, -3.0143e+00],
[-2.5377e+00, -5.5593e+00, -3.6584e+00, ..., -2.6706e+00,
-2.0504e+00, -2.9915e+00],
[-1.5533e+00, -5.5339e+00, -2.9391e+00, ..., -2.4062e+00,
-2.1066e+00, -2.2127e+00],
...,
[-2.0823e+00, -5.6501e+00, -3.3294e+00, ..., -2.5902e+00,
-2.7228e+00, -2.9330e+00],
[-2.9181e+00, -7.1699e+00, -4.5260e+00, ..., -7.0406e+00,
-3.9688e+00, -4.4068e+00],
[-3.0635e+00, -7.4015e+00, -3.5516e+00, ..., -4.9690e+00,
-2.0133e+00, -2.7964e+00]],
[[ 9.1070e-01, 1.6129e+00, 8.1077e-01, ..., 1.0339e+00,
1.4469e+00, 1.4049e+00],
[ 1.1510e+00, 7.7100e-01, 8.4102e-01, ..., 7.0671e-01,
7.7112e-01, 1.9595e+00],
[ 4.1949e-01, -5.8652e-01, -4.8323e-01, ..., 9.3547e-01,
6.6152e-02, 1.1734e+00],
...,
[ 1.6453e+00, -3.5352e-01, -9.6043e-01, ..., 2.4429e+00,
1.9410e+00, 1.8223e+00],
[ 2.2759e+00, 1.3860e+00, 9.2876e-01, ..., 1.1711e+00,
1.9050e+00, 2.2714e+00],
[ 1.0273e+00, 4.9577e-01, 7.0741e-01, ..., -3.9899e-01,
1.1770e+00, 1.1521e+00]],
[[-9.1466e-02, -6.2492e-01, -1.2201e+00, ..., -5.7156e-01,
-9.0413e-01, -1.6120e-01],
[-9.0720e-01, -2.6498e+00, -2.3840e+00, ..., -1.1981e+00,
-9.4527e-01, -6.2159e-01],
[-1.7604e+00, -1.5387e+00, -4.7283e-01, ..., 1.2820e-01,
-2.6271e-02, -2.1952e-01],
...,
[-2.6107e+00, -2.2652e+00, -1.3970e+00, ..., -1.1121e+00,
-3.4069e-01, 1.8196e-01],
[-2.5443e+00, -2.5015e+00, -1.1876e+00, ..., -1.6428e+00,
-6.1044e-01, 4.2263e-01],
[-4.7219e-01, -3.3432e-01, -4.4103e-01, ..., -1.5490e+00,
-1.3856e+00, 8.7637e-02]]],
[[[ 7.8468e-01, 1.0963e+00, 5.6826e-01, ..., 5.0875e-01,
9.6034e-01, 3.4063e-01],
[-1.3363e+00, -1.2088e+00, -2.5040e-03, ..., -7.9635e-01,
-3.8750e-01, -1.8029e-01],
[-4.8093e-02, -1.1741e+00, -3.2777e-01, ..., 9.3290e-01,
6.6649e-01, 8.3963e-01],
...,
[-2.9229e-01, -9.5483e-01, -2.3111e-02, ..., -2.0350e-01,
-1.0966e-01, 3.7701e-01],
[-8.7157e-02, -9.0274e-01, -3.3478e-01, ..., -1.7775e-01,
8.4585e-02, -9.9650e-02],
[ 4.6613e-01, -2.1527e-01, -7.9440e-01, ..., -4.6909e-01,
-9.3196e-01, 1.5420e-01]],
[[-4.2046e+00, -2.1148e+00, -3.0579e+00, ..., -1.4708e+00,
-2.2167e+00, -1.8761e+00],
[-3.4437e+00, -1.6828e+00, -8.7274e-01, ..., -2.1206e+00,
-2.5312e+00, -2.2060e+00],
[-2.2226e+00, -1.4657e+00, -1.2562e+00, ..., 8.0191e-02,
-1.7060e+00, -1.5406e+00],
...,
[-8.4432e-01, -6.2472e-01, -2.0326e-01, ..., -1.0168e+00,
-9.7654e-01, -8.8662e-01],
[-1.0268e+00, -4.9870e-01, -4.1429e-01, ..., -8.8232e-01,
-1.0883e+00, -1.2281e+00],
[ 2.1404e-01, -6.6325e-01, -7.9801e-01, ..., -7.7073e-01,
-6.8803e-01, -5.2223e-01]],
[[ 2.8073e+00, 3.3958e+00, 3.2437e+00, ..., 1.8393e+00,
6.6188e-01, 2.4792e+00],
[ 7.6926e-01, 1.0990e+00, 2.6552e+00, ..., 1.7993e+00,
7.5532e-01, 2.3984e+00],
[ 1.1945e+00, 8.0253e-01, 1.9264e+00, ..., -1.2657e+00,
5.1551e-01, 1.2228e+00],
...,
[ 3.0151e-01, 1.1305e+00, 1.8416e+00, ..., -6.8052e-01,
5.5413e-01, 1.5000e+00],
[ 1.8615e-01, 1.5261e+00, 1.6043e+00, ..., -5.9926e-01,
7.6237e-01, 1.1381e+00],
[ 1.5378e+00, 1.1126e+00, 1.8901e+00, ..., 3.7347e-01,
1.2798e+00, 1.3574e+00]],
...,
[[-2.9865e+00, -8.2973e+00, -5.4374e+00, ..., -5.6589e+00,
-3.3195e+00, -4.2328e+00],
[-3.3258e+00, -9.2015e+00, -6.4824e+00, ..., -4.3318e+00,
-2.6115e+00, -3.3747e+00],
[-2.0967e+00, -5.9651e+00, -3.7030e+00, ..., -4.7515e+00,
-3.3754e+00, -2.1614e+00],
...,
[-2.9917e+00, -6.6055e+00, -3.6343e+00, ..., -4.9705e+00,
-3.1119e+00, -2.7465e+00],
[-3.1371e+00, -7.3030e+00, -4.5025e+00, ..., -6.0689e+00,
-4.2060e+00, -4.3018e+00],
[-2.3685e+00, -6.5721e+00, -3.4542e+00, ..., -3.7268e+00,
-2.4023e+00, -2.3745e+00]],
[[-7.6467e-01, -1.4878e-01, 2.4875e-01, ..., 1.8357e+00,
1.7636e+00, 1.2720e+00],
[ 1.5254e+00, 1.0677e+00, 1.2116e+00, ..., 1.5993e+00,
1.3056e+00, 1.8816e+00],
[ 8.8398e-01, 2.8008e-01, 6.0511e-01, ..., 7.6818e-01,
2.1600e-01, 1.2221e+00],
...,
[ 1.1990e+00, 4.9398e-01, 1.4929e+00, ..., 1.9379e+00,
6.9581e-01, 9.3187e-01],
[ 1.8402e+00, 1.1459e+00, 1.2499e+00, ..., 1.6946e+00,
7.2130e-01, 1.4271e+00],
[ 1.1200e+00, 8.9955e-01, 1.1757e+00, ..., 8.6379e-01,
6.8431e-01, 1.4140e+00]],
[[-7.8329e-01, -7.0644e-01, -6.2281e-01, ..., -6.3131e-01,
-6.2821e-01, -7.0731e-02],
[-1.5559e+00, -1.4241e+00, -6.2366e-01, ..., -1.9288e+00,
-1.6153e+00, -1.0136e+00],
[-2.3786e+00, -6.5172e-01, 6.0492e-01, ..., 5.6503e-01,
-8.4241e-01, -3.6521e-01],
...,
[-2.4186e+00, -1.5959e+00, -8.5205e-01, ..., -1.0265e+00,
-1.4983e+00, -8.6468e-01],
[-1.7773e+00, -1.7896e+00, -8.1309e-01, ..., -1.1351e+00,
-1.6292e+00, -7.3155e-01],
[ 9.3358e-02, 1.0885e-01, -2.0458e-01, ..., -5.1235e-01,
-7.9668e-01, -1.8484e-02]]],
[[[-8.1148e-02, -1.3095e+00, -2.3347e+00, ..., 8.4608e-01,
8.1738e-01, 5.8892e-01],
[-1.2707e+00, 2.3093e+00, 1.5764e+00, ..., -2.5897e+00,
-4.5612e+00, -5.4314e+00],
[-8.3467e+00, -9.5659e+00, -3.1551e+00, ..., -3.1086e+00,
-1.4451e+00, -1.9117e+00],
...,
[-5.5513e+00, 6.7320e-01, 1.0572e+00, ..., -1.8338e+01,
-1.0225e+01, -1.0986e+01],
[ 7.9969e-01, -4.2081e-01, -4.4366e-01, ..., -6.0360e+00,
-8.8580e+00, -5.3662e+00],
[ 5.9998e-01, 2.3743e-01, 4.2925e-01, ..., -3.5322e-01,
-3.0052e+00, 2.6924e+00]],
[[-8.6195e+00, -6.2669e+00, -4.7052e+00, ..., -1.1839e+00,
-1.4796e+00, -7.7272e-01],
[-6.1250e+00, -8.9459e+00, -8.6827e+00, ..., -7.0200e+00,
-5.5558e+00, -9.6642e-01],
[-1.0964e+01, -1.4441e+01, -1.0374e+01, ..., -1.6270e+01,
-1.0285e+01, -2.6194e+00],
...,
[-9.7225e+00, -1.0957e+01, -3.1749e+00, ..., -2.6362e+01,
-1.9789e+01, -6.3602e+00],
[-6.4537e+00, -6.7326e+00, -7.0997e+00, ..., -2.1301e+01,
-1.3783e+01, -4.3060e+00],
[-9.3592e-01, -1.0123e+00, -1.7025e+00, ..., -1.6299e+01,
-1.6646e+01, -1.0584e+01]],
[[ 2.2664e+00, 4.7165e+00, 5.1043e+00, ..., 3.1023e+00,
1.3752e+00, 3.0696e+00],
[-4.4595e+00, -6.4193e+00, -2.0171e+00, ..., 6.6582e+00,
3.6517e+00, 6.4429e+00],
[-6.6961e+00, -1.0637e+01, -8.5399e+00, ..., -1.4256e-01,
-6.6805e+00, -2.3719e+00],
...,
[-1.0724e+01, -8.6285e+00, -7.2456e+00, ..., -2.0482e+01,
-9.3801e+00, -1.3699e+01],
[ 3.9478e-01, 7.7414e-01, -1.9079e+00, ..., -1.0368e+01,
-1.6114e+01, -1.3189e+01],
[ 1.9404e+00, 2.5548e+00, 3.7772e+00, ..., -1.4872e+01,
-2.4570e+01, -1.4851e+01]],
...,
[[-2.0019e+00, -8.9383e+00, -6.4062e+00, ..., -3.8708e+00,
-2.5345e+00, -4.5240e+00],
[-4.6570e+00, -1.9411e+01, -1.3252e+01, ..., -1.0200e+01,
-7.1192e+00, -6.1127e+00],
[-1.7167e+00, -1.5158e+01, -1.4114e+01, ..., -1.3784e+01,
-1.5409e+01, -8.1804e+00],
...,
[-2.3812e+00, -1.4317e+01, -6.5325e+00, ..., -3.1932e+00,
-1.1532e+01, -6.0784e-01],
[-2.9833e+00, -7.1846e+00, -5.4079e+00, ..., -8.1922e+00,
-5.5733e+00, -1.2154e+01],
[-2.1498e+00, -6.8862e+00, -5.1465e+00, ..., -2.1981e+01,
-1.2275e+01, -2.9841e+01]],
[[-5.3809e+00, -1.4097e+00, 1.1926e+00, ..., 1.4232e+00,
1.4076e+00, 1.2466e+00],
[-4.0192e+00, -9.9004e+00, -1.0270e+01, ..., 2.1041e-01,
6.1273e-01, 6.2639e-01],
[ 3.4332e+00, -1.2876e+00, -3.4553e+00, ..., -9.2102e+00,
-4.9208e+00, 1.0285e+00],
...,
[ 7.5512e-01, -2.9083e+00, -2.2442e-01, ..., -1.2849e+01,
-2.1916e+01, -9.4601e+00],
[ 6.2727e-01, -5.1830e-02, -5.1931e-01, ..., -2.1423e+01,
-2.5231e+01, -1.1654e+01],
[ 1.5387e+00, 1.5407e+00, 1.7346e+00, ..., -6.7894e+00,
-3.1219e+00, -5.8209e+00]],
[[-5.6759e-01, -1.5062e+00, -1.3730e+00, ..., 1.7905e-01,
1.0491e-01, 1.0033e+00],
[ 2.5419e-01, 5.8096e-01, -9.9537e-01, ..., -4.3870e+00,
-3.7244e+00, -1.2448e+00],
[-2.7973e+00, -1.7351e+00, -4.9615e-01, ..., -8.3180e-01,
-1.5323e+00, -2.2881e-01],
...,
[-3.8538e+00, -7.7589e+00, -3.0921e+00, ..., -2.1737e+01,
-2.3216e+01, -8.2943e+00],
[-2.9201e+00, -2.5274e+00, -3.6621e+00, ..., -1.8508e+01,
-7.3360e+00, 2.5725e+00],
[-7.9315e-01, -3.7038e-01, -9.9220e-01, ..., -3.9469e+00,
1.9377e+00, 4.5424e+00]]]], device='cuda:0'),
tensor([[[[-2.1255e+00, -4.3856e+00, -6.4635e+00, ..., -5.5123e-01,
-5.8368e-01, -7.4420e-01],
[-8.4468e+00, -1.6649e+01, -1.8543e+01, ..., 1.6042e-02,
-8.9940e-01, -2.6919e+00],
[-7.2159e+00, -1.4141e+01, -1.5337e+01, ..., 5.0916e-01,
1.8626e+00, 2.6200e-01],
...,
[-2.6373e+00, -8.2029e+00, -8.6925e+00, ..., 1.8258e+00,
2.0686e+00, 2.6183e+00],
[ 1.7917e+00, 1.4135e+00, 1.8117e+00, ..., -3.6264e+00,
-2.3870e+00, 1.0612e-01],
[ 7.9365e-01, -4.4968e-01, -4.8220e-01, ..., 1.1143e+00,
9.7828e-01, -2.1302e+00]],
[[-3.3743e+00, -2.1650e+00, -1.5898e+01, ..., -3.0861e+00,
-3.1920e+00, -7.7138e-01],
[-6.9665e+00, -1.6810e+01, -2.8416e+01, ..., -7.1437e+00,
-7.2759e+00, -1.9895e+00],
[-8.8356e+00, -1.1264e+01, -7.6861e+00, ..., -5.2768e+00,
-7.0004e+00, -2.0470e+00],
...,
[-1.4915e+01, -2.8529e+01, -2.8353e+01, ..., -6.7010e+00,
-5.9937e+00, -4.2658e+00],
[-5.0322e+00, -2.0467e+00, -2.6803e+00, ..., -1.6132e+01,
-1.1170e+01, -4.0348e+00],
[-2.8917e+00, -2.2863e+00, -1.6310e+00, ..., -4.8871e+00,
-4.3760e+00, -1.9776e+00]],
[[-5.2343e+00, -5.8374e+00, -7.0010e+00, ..., 3.4783e+00,
1.2306e+00, 2.7214e+00],
[-6.2556e+00, -7.5026e+00, -3.2520e+00, ..., 3.6235e+00,
2.0569e+00, 5.1592e+00],
[-4.9849e+00, -2.9795e+00, -2.2323e+00, ..., -3.7233e+00,
-5.3590e+00, -3.9765e+00],
...,
[-1.0533e+01, -1.5564e+01, -1.9737e+01, ..., -5.6770e+00,
-6.8237e+00, -4.5626e+00],
[ 1.7177e+00, 4.9293e+00, 3.6542e+00, ..., -9.9164e+00,
-1.3252e+01, -7.9255e+00],
[ 1.3143e+00, 1.4342e+00, 1.7747e+00, ..., 1.9890e+00,
-1.3616e+00, -7.7601e-01]],
...,
[[-3.2357e+00, -1.1892e+01, -1.7484e+01, ..., -4.4151e+00,
-3.5212e+00, -3.6093e+00],
[-1.3078e+00, -2.0737e+01, -2.2731e+01, ..., -6.9428e+00,
-5.5663e+00, -5.7480e+00],
[ 2.5537e-01, -5.4149e+00, -5.6339e+00, ..., -9.8670e+00,
-1.2313e+01, -8.4013e+00],
...,
[-2.4925e+00, -1.3882e+01, -1.2214e+01, ..., -1.0544e+01,
-1.0526e+01, -1.4366e+01],
[-3.9602e+00, -9.4519e+00, -1.2663e+01, ..., -1.0414e+01,
-9.3729e+00, -1.2122e+01],
[-2.1129e+00, -5.4710e+00, -3.4764e+00, ..., -5.7219e+00,
-4.6099e+00, -6.0184e+00]],
[[-2.3455e+00, -2.3452e+00, 1.7549e+00, ..., 7.4291e-01,
8.1534e-01, 1.7774e+00],
[ 1.6629e+00, -3.4212e+00, -7.5226e+00, ..., -2.7690e+00,
-1.1291e+00, 2.6628e+00],
[-2.5087e+00, -1.4423e+01, -1.3620e+01, ..., -1.9211e+00,
-7.0609e+00, -2.5122e+00],
...,
[-1.8378e+00, -7.3804e+00, -1.0445e+01, ..., -9.5595e-01,
-3.8998e+00, -2.6199e+00],
[ 2.8313e+00, 3.5756e+00, 6.9633e+00, ..., -2.9914e-01,
-8.1527e-01, -1.0827e+00],
[ 9.2887e-01, -8.1743e-02, 9.4767e-01, ..., 1.5683e-01,
1.0997e+00, 1.6480e+00]],
[[-2.5813e-03, 3.4199e+00, -1.0674e+00, ..., -8.8841e-01,
-9.9485e-01, 3.0467e-01],
[-5.7655e-01, -4.9020e+00, -1.8373e+01, ..., -1.9536e+00,
-1.3866e+00, 3.8742e-02],
[-4.9954e+00, -1.0970e+01, -8.5426e+00, ..., 1.6184e+00,
-4.9885e-01, 1.9721e+00],
...,
[-2.3022e+00, -7.6450e+00, -1.3452e+01, ..., -5.0275e+00,
-1.5333e+00, -2.2469e-01],
[-2.6746e+00, 9.3026e-01, -3.4255e+00, ..., -8.7435e+00,
-7.1806e+00, -4.4371e+00],
[-2.4991e-01, 2.4646e-01, -3.8001e-01, ..., -2.9116e-01,
-1.6431e+00, -1.8326e+00]]],
[[[ 1.0497e+00, 1.2978e+00, 1.1903e+00, ..., 1.1954e+00,
9.7572e-01, 4.8037e-01],
[-6.0115e-01, -2.4239e+00, -1.2082e+00, ..., 2.9863e-01,
6.0409e-02, -1.4465e+00],
[-4.6494e-01, -1.1967e+00, -5.5833e-01, ..., -9.2731e-01,
-4.3540e-02, 6.0831e-01],
...,
[ 5.6538e-01, -9.2010e-02, 1.2283e-01, ..., 5.1169e-01,
1.0780e+00, 8.0044e-01],
[ 1.0850e-02, -8.2360e-01, -1.5286e-01, ..., 8.4964e-01,
1.5156e+00, 1.1860e-01],
[ 8.3146e-01, -1.9353e-01, -1.4410e-01, ..., 3.8132e-01,
2.8987e-01, 6.6148e-01]],
[[-2.8946e+00, -2.3400e-02, -1.3497e+00, ..., -1.6464e+00,
-2.3906e+00, -6.0989e-01],
[-4.0785e+00, -2.7443e+00, -2.5151e+00, ..., -2.5619e+00,
-3.8793e+00, -1.6301e+00],
[-3.0763e+00, -2.4809e+00, -2.7880e+00, ..., -3.8259e+00,
-3.0565e+00, -1.3447e+00],
...,
[-7.0596e-01, -3.6997e-01, -1.2965e+00, ..., -7.8882e-01,
1.6804e-01, 7.9405e-01],
[-2.6079e+00, -4.4447e-01, -6.1262e-01, ..., 1.5384e-01,
1.3476e-01, 6.9304e-01],
[-1.5192e+00, -1.2655e+00, -8.9045e-01, ..., -5.1959e-01,
-6.0712e-01, -1.6557e-01]],
[[ 2.5153e+00, 2.9628e+00, 2.4872e+00, ..., 2.2644e+00,
6.2527e-01, 2.5675e+00],
[ 8.4460e-01, 6.4120e-01, 2.3599e+00, ..., 1.5684e+00,
4.9035e-01, 2.5493e+00],
[ 1.1320e+00, 1.4304e+00, 1.9717e+00, ..., 1.4081e+00,
1.8990e+00, 2.8383e+00],
...,
[ 9.4601e-01, 1.0737e+00, 2.0626e+00, ..., 1.0298e+00,
6.1627e-01, 1.8196e+00],
[ 1.1044e+00, 2.2283e+00, 2.0541e+00, ..., 1.1882e+00,
7.4436e-01, 1.6377e+00],
[ 2.4248e+00, 1.8423e+00, 2.3040e+00, ..., 1.9712e+00,
1.0361e+00, 1.7061e+00]],
...,
[[-2.6976e+00, -6.9410e+00, -5.7752e+00, ..., -3.1355e+00,
-3.9381e+00, -5.3754e+00],
[-3.4052e+00, -7.7069e+00, -5.3639e+00, ..., -4.1947e+00,
-4.5882e+00, -5.1080e+00],
[-2.2516e+00, -6.2521e+00, -4.2710e+00, ..., -2.5967e+00,
-3.5683e+00, -3.3669e+00],
...,
[-2.5055e+00, -4.7817e+00, -2.3388e+00, ..., -2.3395e+00,
-2.8406e+00, -2.9237e+00],
[-2.4851e+00, -5.4302e+00, -3.7079e+00, ..., -4.7493e+00,
-3.5951e+00, -4.7945e+00],
[-2.6781e+00, -6.0102e+00, -2.8636e+00, ..., -2.9535e+00,
-1.6929e+00, -3.2153e+00]],
[[ 3.1639e-01, 1.1635e+00, 4.1108e-01, ..., 9.0616e-01,
1.5472e+00, 1.6988e+00],
[ 1.3081e+00, 8.5580e-01, 2.7151e-01, ..., 2.7309e-01,
1.1677e+00, 2.7110e+00],
[ 1.3583e+00, 7.4248e-01, 5.1786e-01, ..., 1.1950e+00,
1.1547e+00, 2.0317e+00],
...,
[ 1.5934e+00, 4.5432e-01, 6.6079e-01, ..., 1.8961e+00,
2.0119e+00, 1.4726e+00],
[ 1.9782e+00, 1.1613e+00, 1.8183e+00, ..., 1.9980e+00,
1.6962e+00, 2.5601e+00],
[ 1.1607e+00, 8.7660e-01, 1.3694e+00, ..., 1.2750e+00,
1.4822e+00, 1.5383e+00]],
[[-6.1698e-01, 1.6454e-02, -1.7825e-01, ..., 5.4333e-01,
-2.7519e-03, 9.7662e-01],
[-1.4385e+00, -2.1989e+00, -1.7736e+00, ..., -5.4491e-01,
-6.4513e-01, -5.2126e-02],
[-2.2631e+00, -1.4340e+00, -5.3988e-01, ..., -5.4094e-01,
-3.8662e-01, -1.1117e-01],
...,
[-1.9992e+00, -1.3854e+00, -5.4847e-01, ..., -4.0829e-01,
-3.9136e-01, -3.8179e-01],
[-1.7265e+00, -1.3862e+00, -7.8341e-01, ..., 2.9757e-02,
-2.1141e-01, 4.3996e-01],
[-2.4597e-01, 4.0297e-01, -2.0080e-01, ..., 1.7735e-01,
1.2271e-02, 6.0679e-01]]],
[[[ 9.0837e-01, 1.4001e+00, 9.5158e-01, ..., 3.8116e-01,
2.3774e-01, -2.2289e-01],
[-8.5114e-01, -2.2308e+00, -1.7260e+00, ..., -2.6742e-01,
1.1764e-01, -1.2540e+00],
[ 1.4434e-01, -6.3171e-01, 8.0239e-01, ..., -1.3017e+00,
-2.7445e-01, 1.9674e-01],
...,
[ 4.6018e-01, -2.2308e-01, -3.3501e-01, ..., -6.4532e-01,
1.0698e+00, 7.8890e-01],
[-2.0196e-01, -1.0390e+00, -3.3017e-01, ..., 8.7724e-02,
8.7821e-01, 1.0457e-01],
[ 7.4458e-01, -2.7872e-01, -3.4940e-01, ..., -4.5456e-01,
-3.5436e-01, 3.2617e-01]],
[[-4.3877e+00, -1.4079e+00, -3.7660e+00, ..., -3.0737e+00,
-3.4939e+00, -7.8682e-01],
[-4.0526e+00, -2.7458e+00, -2.9463e+00, ..., -3.2115e+00,
-2.6274e+00, -1.3254e+00],
[-2.4531e+00, -1.3698e+00, -3.4161e+00, ..., -4.9882e+00,
-1.6892e+00, -9.5603e-01],
...,
[-8.0785e-01, -5.7277e-01, -1.6174e+00, ..., -2.9607e+00,
-2.2281e+00, -2.1023e-01],
[-2.8290e+00, -3.7799e-01, -1.1005e+00, ..., -1.4493e+00,
-1.4332e+00, -6.3776e-02],
[-1.9848e+00, -1.4471e+00, -1.7752e+00, ..., -8.3674e-01,
-7.4519e-01, 5.1562e-02]],
[[ 3.0692e+00, 4.3227e+00, 3.0133e+00, ..., 2.3365e+00,
7.8896e-01, 2.4287e+00],
[ 4.5292e-01, 1.0315e+00, 1.0764e+00, ..., 2.4097e+00,
6.8421e-01, 1.3429e+00],
[ 8.7130e-01, 1.6622e+00, 3.0230e-01, ..., 3.4588e-01,
7.8182e-01, 4.1745e-01],
...,
[ 2.1532e-01, 8.9370e-01, 1.4187e+00, ..., 1.7444e-01,
-5.6848e-01, 1.0508e+00],
[ 3.3634e-01, 2.0754e+00, 1.6812e+00, ..., 6.1118e-02,
3.4304e-01, 1.2752e+00],
[ 1.4653e+00, 1.3349e+00, 1.7357e+00, ..., 1.0694e+00,
9.3706e-01, 1.3667e+00]],
...,
[[-2.1051e+00, -7.3534e+00, -6.3293e+00, ..., -4.1255e+00,
-2.9982e+00, -3.5315e+00],
[-2.7727e+00, -6.4489e+00, -6.1899e+00, ..., -6.2347e+00,
-4.3726e+00, -3.6615e+00],
[-2.5493e-01, -3.1499e+00, -4.6138e+00, ..., -5.0285e+00,
-5.5425e+00, -1.7884e+00],
...,
[-9.3963e-01, -2.2888e+00, -1.6675e+00, ..., -5.3790e+00,
-4.6034e+00, -3.6688e+00],
[-1.2943e+00, -3.6934e+00, -3.3923e+00, ..., -6.5295e+00,
-4.0284e+00, -4.4050e+00],
[-2.2232e+00, -4.9697e+00, -2.9647e+00, ..., -4.5673e+00,
-2.5439e+00, -3.4272e+00]],
[[-7.9559e-01, -3.6772e-01, -1.0697e+00, ..., 9.7786e-01,
1.0198e+00, 1.7816e+00],
[ 1.7953e+00, 1.0795e+00, 1.5220e+00, ..., 9.5531e-01,
7.0498e-02, 1.5940e+00],
[ 1.2667e+00, -6.3254e-03, -6.5721e-01, ..., -3.5617e-01,
6.9688e-02, 9.4779e-01],
...,
[ 1.2439e+00, -1.5995e-01, 6.0756e-01, ..., 1.1435e+00,
1.2222e+00, 1.5701e+00],
[ 1.8535e+00, 9.5587e-01, 1.5409e+00, ..., 1.4569e+00,
9.0629e-01, 2.1522e+00],
[ 1.0316e+00, 4.2005e-01, 8.7118e-01, ..., 9.1849e-01,
1.2285e+00, 1.5489e+00]],
[[-5.4411e-01, 2.9039e-01, -4.9592e-01, ..., -2.3322e-01,
-3.6731e-01, 8.1657e-01],
[-1.1369e+00, -1.5648e+00, -1.8962e+00, ..., 1.2708e-01,
-3.8409e-01, 5.7936e-01],
[-2.0116e+00, -9.2066e-01, -1.1110e+00, ..., 5.6576e-01,
-1.1277e+00, -9.3068e-03],
...,
[-1.6783e+00, -1.4592e+00, -1.2590e+00, ..., -8.9801e-01,
-1.9629e+00, -7.6163e-01],
[-1.7420e+00, -1.0936e+00, -1.0424e+00, ..., -6.9074e-01,
-1.1060e+00, 8.3929e-02],
[-3.2340e-01, 3.8687e-01, -4.2912e-01, ..., 9.3021e-02,
-2.0561e-01, 6.4050e-01]]],
[[[ 8.3421e-01, 1.0757e+00, 5.0659e-01, ..., 1.8531e-01,
9.7575e-01, 4.2620e-01],
[-1.3429e+00, -1.1244e+00, -2.0657e-01, ..., -1.3989e+00,
-2.1587e-01, -4.0111e-01],
[-1.7218e-01, -9.9853e-01, -4.6105e-01, ..., -2.9628e+00,
-4.4467e-01, 6.1484e-01],
...,
[-1.2037e+00, 9.2055e-01, 2.6245e-01, ..., 3.1067e-02,
-1.3606e-01, 2.8702e-01],
[-2.8676e+00, -1.8207e+00, -8.0296e-01, ..., -1.4688e-01,
3.4514e-02, -1.4727e-01],
[-1.9843e-01, -1.6225e+00, -3.8702e-01, ..., -4.0709e-01,
-1.0240e+00, 1.4929e-01]],
[[-4.1996e+00, -2.0345e+00, -2.9758e+00, ..., -1.7321e+00,
-2.2223e+00, -1.8489e+00],
[-3.2534e+00, -1.7394e+00, -8.8346e-01, ..., -2.4695e+00,
-2.2829e+00, -2.1549e+00],
[-3.1979e+00, -1.5995e+00, -7.3590e-01, ..., -3.1798e+00,
-1.8468e+00, -1.4204e+00],
...,
[-2.4161e+00, -8.2322e+00, -1.4078e+01, ..., -7.8463e-01,
-9.8808e-01, -8.6008e-01],
[-5.6263e+00, -7.2954e+00, -8.7947e+00, ..., -8.8437e-01,
-1.1853e+00, -1.2461e+00],
[-3.9925e+00, -3.9133e+00, -5.0751e+00, ..., -7.4609e-01,
-7.2909e-01, -5.3072e-01]],
[[ 2.8547e+00, 3.4887e+00, 3.3727e+00, ..., 1.4310e+00,
8.3888e-01, 2.6655e+00],
[ 7.3769e-01, 1.2333e+00, 2.7827e+00, ..., 9.5328e-01,
6.9424e-01, 2.7290e+00],
[ 1.0985e+00, 7.4898e-01, 2.1410e+00, ..., -6.3641e-01,
9.1756e-01, 1.9318e+00],
...,
[-2.1690e+00, -1.6439e+00, -8.3214e+00, ..., -4.7994e-01,
5.8366e-01, 1.5657e+00],
[-7.2154e+00, -3.1781e+00, -6.4093e+00, ..., -5.6035e-01,
8.9058e-01, 1.2641e+00],
[-3.2669e+00, -2.6634e+00, -5.7556e-01, ..., 4.3245e-01,
1.3548e+00, 1.4188e+00]],
...,
[[-3.1207e+00, -8.6581e+00, -5.2340e+00, ..., -6.2790e+00,
-3.9746e+00, -4.5275e+00],
[-3.5640e+00, -9.9148e+00, -6.3059e+00, ..., -8.7302e+00,
-4.1847e+00, -3.6026e+00],
[-2.4600e+00, -7.1757e+00, -3.7245e+00, ..., -1.1485e+01,
-2.7244e+00, -1.5297e+00],
...,
[ 8.2251e-01, -5.5381e+00, -6.9533e+00, ..., -5.6737e+00,
-3.7697e+00, -2.8345e+00],
[-5.6875e-01, -6.2635e+00, -1.0125e+01, ..., -6.3025e+00,
-4.6273e+00, -4.3919e+00],
[-1.4079e+00, -4.5323e+00, -6.6341e+00, ..., -3.8072e+00,
-2.6924e+00, -2.5114e+00]],
[[-8.8752e-01, -1.7355e-01, 3.1559e-01, ..., 2.0209e+00,
1.6270e+00, 1.2187e+00],
[ 1.5414e+00, 1.0473e+00, 1.2467e+00, ..., 2.8979e+00,
1.2125e+00, 1.9219e+00],
[ 9.2589e-01, 3.5410e-01, 6.2660e-01, ..., 4.3678e+00,
6.7936e-01, 1.3383e+00],
...,
[ 7.7217e-02, -4.1992e+00, -5.3018e+00, ..., 2.1910e+00,
7.6499e-01, 1.0239e+00],
[ 6.8012e-01, -1.4445e+00, -2.5739e+00, ..., 1.7234e+00,
6.9199e-01, 1.5572e+00],
[ 3.7852e-01, -8.8765e-02, -1.2058e+00, ..., 8.7231e-01,
7.0913e-01, 1.4942e+00]],
[[-7.2318e-01, -7.8502e-01, -5.7957e-01, ..., -2.7631e-01,
-6.9368e-01, -3.7491e-02],
[-1.3765e+00, -1.4827e+00, -7.8779e-01, ..., -1.8257e+00,
-1.3993e+00, -1.1156e+00],
[-1.9990e+00, -1.0758e-01, 1.7367e-01, ..., -4.2821e+00,
-1.0184e+00, -2.2979e-01],
...,
[ 1.1248e+00, -1.0595e+00, -3.9970e+00, ..., -3.8737e-01,
-1.4765e+00, -8.3126e-01],
[-1.4721e+00, -2.7470e+00, -5.5781e+00, ..., -9.3346e-01,
-1.6967e+00, -6.6758e-01],
[-2.4070e+00, -5.8840e-01, -3.9190e+00, ..., -4.0872e-01,
-7.8689e-01, 4.7225e-03]]],
[[[ 5.7652e-01, 9.6261e-01, 2.6115e-01, ..., 1.0045e+00,
7.6010e-01, 3.8779e-02],
[-2.4830e-01, -4.7882e-01, -7.6845e-02, ..., 1.6294e-01,
3.9074e-01, -3.9463e-01],
[-2.0641e-01, -2.0444e-01, 7.1986e-01, ..., 4.0101e-01,
2.6524e-01, 7.2549e-01],
...,
[ 5.5652e-01, -2.6575e-01, -4.1569e-01, ..., 3.5491e-01,
8.5103e-01, 6.1460e-01],
[-2.3841e-01, -2.9957e-01, 1.2271e+00, ..., 8.9444e-01,
1.6286e+00, 1.2024e-01],
[ 8.5449e-01, -5.3952e-01, 1.5482e+00, ..., 4.6220e-01,
3.2440e-01, 7.5882e-01]],
[[-2.3362e+00, -9.8584e-02, -1.1795e+00, ..., -1.3163e+00,
-2.8905e+00, -1.6614e+00],
[-2.4074e+00, -2.2790e-01, -1.3150e+00, ..., -1.7000e+00,
-3.4286e+00, -1.8826e+00],
[-2.5769e+00, -1.2193e+00, -8.5240e-01, ..., -2.2714e+00,
-2.3787e+00, -1.0288e+00],
...,
[ 2.9448e-01, 6.7742e-01, -7.0559e-01, ..., -7.1037e-01,
-2.8953e-01, 7.2388e-01],
[-2.1629e+00, 1.3464e+00, 8.6786e-01, ..., 5.7168e-02,
-9.9628e-03, 6.9135e-01],
[-1.6747e+00, -1.1858e+00, -3.4183e+00, ..., -4.3616e-01,
-6.5339e-01, -2.4212e-01]],
[[ 1.7165e+00, 2.5224e+00, 7.2538e-01, ..., 8.1896e-01,
4.5144e-01, 2.1426e+00],
[-2.0091e-01, 1.2472e+00, -1.9682e-01, ..., 1.7294e-01,
1.6090e-01, 2.3446e+00],
[ 6.7396e-01, 1.6487e+00, -7.1427e-01, ..., -8.4359e-01,
4.6123e-01, 1.9206e+00],
...,
[ 1.2528e+00, 3.5574e-01, -1.4568e-01, ..., 4.3173e-01,
1.7836e-01, 1.6456e+00],
[ 1.5007e+00, 2.3249e+00, -1.1869e+00, ..., 1.0254e+00,
8.3440e-01, 1.7525e+00],
[ 2.7326e+00, 2.4083e+00, -1.6664e+00, ..., 1.9690e+00,
1.0616e+00, 1.7893e+00]],
...,
[[-1.6297e+00, -4.7253e+00, -3.3335e+00, ..., -8.5043e-01,
-1.6282e+00, -3.3879e+00],
[-2.0698e+00, -5.0172e+00, -2.7201e+00, ..., -4.7875e-01,
-1.6358e+00, -3.2617e+00],
[-1.4333e+00, -4.6687e+00, -2.5089e+00, ..., -3.7242e+00,
-3.0179e+00, -2.3069e+00],
...,
[-2.3751e+00, -3.5109e+00, -1.5448e+00, ..., -2.5760e+00,
-2.0976e+00, -2.6337e+00],
[-2.4217e+00, -4.9157e+00, -3.6911e+00, ..., -4.2929e+00,
-3.1538e+00, -4.9257e+00],
[-2.5890e+00, -5.9056e+00, -1.4859e+00, ..., -2.9979e+00,
-1.6575e+00, -3.3516e+00]],
[[ 8.0295e-01, 1.3600e+00, 1.1468e+00, ..., 2.0503e-01,
1.1876e+00, 1.4931e+00],
[ 9.8272e-01, 6.2177e-01, 7.2317e-01, ..., -4.1145e-01,
6.4243e-01, 2.1553e+00],
[ 3.4316e-01, 2.5911e-01, 1.3820e+00, ..., 1.0121e-01,
4.8594e-01, 1.4971e+00],
...,
[ 1.6688e+00, -1.3843e+00, -3.0031e+00, ..., 1.5412e+00,
1.8451e+00, 1.5380e+00],
[ 2.0975e+00, 1.5144e+00, 1.3609e+00, ..., 1.9327e+00,
1.7568e+00, 2.6136e+00],
[ 1.1116e+00, -2.8058e-01, 1.4313e+00, ..., 1.3065e+00,
1.4865e+00, 1.5122e+00]],
[[-9.1182e-02, -8.6531e-01, -8.8101e-01, ..., -2.9384e-01,
-7.9769e-01, -1.8882e-02],
[-9.4825e-01, -2.8095e+00, -2.6722e+00, ..., -1.3567e+00,
-8.2636e-01, -6.9407e-01],
[-1.6747e+00, -1.4719e+00, -1.3409e+00, ..., -3.6786e-01,
-3.0418e-01, -1.5019e-01],
...,
[-1.2539e+00, -1.3968e+00, 2.1410e+00, ..., -3.8475e-01,
-4.5969e-01, -4.6161e-01],
[-1.4810e+00, -1.0652e+00, -8.8467e-02, ..., -3.1970e-01,
-3.4661e-01, 4.5663e-01],
[-1.9079e-01, 9.5981e-01, 4.6631e-01, ..., 2.2393e-01,
1.8894e-02, 5.9348e-01]]]], device='cuda:0'),
tensor([[[[ 7.0518e-01, 1.0691e+00, 6.7061e-01, ..., 8.5350e-01,
1.6793e+00, 1.2426e+00],
[-3.5966e-01, -1.4814e+00, -5.5858e-01, ..., -5.3507e+00,
-5.6513e+00, -3.6744e+00],
[ 9.1941e-02, -9.7295e-01, -6.1117e-01, ..., -2.8200e+00,
-6.7088e+00, -5.2300e+00],
...,
[-8.5203e-02, -1.2119e+00, -9.0506e-01, ..., -1.7639e+00,
2.0516e+00, -5.7892e-02],
[-4.7825e-01, -2.3641e+00, -2.7822e-02, ..., -1.3348e+00,
-3.9227e-01, -2.9441e+00],
[ 3.8369e-01, -7.9600e-01, 4.7195e-01, ..., 9.6201e-01,
7.9083e-01, 1.8595e+00]],
[[-3.2691e+00, -1.6406e+00, -3.0643e+00, ..., -5.3003e+00,
-6.9781e+00, -2.4626e+00],
[-4.4753e+00, -1.8749e+00, -2.2340e+00, ..., -6.2400e+00,
-8.2120e+00, -3.6075e+00],
[-4.4039e+00, -2.6284e+00, -2.9622e+00, ..., -3.8084e+00,
-7.9875e+00, -5.0945e+00],
...,
[-2.9635e+00, -9.8355e-01, -2.4022e+00, ..., -1.0874e+00,
-4.1768e+00, -9.6197e-01],
[-3.6597e+00, -6.5827e-01, -8.7660e-01, ..., -3.5222e+00,
-9.7613e+00, -3.4986e+00],
[-1.1624e+00, -7.7904e-01, -2.1145e+00, ..., -1.7664e+00,
-9.6987e-01, -8.8808e-01]],
[[ 1.8204e+00, 2.9206e+00, 2.4051e+00, ..., -2.5090e+00,
-6.7398e+00, -3.3666e+00],
[ 3.0552e-01, 1.3229e+00, 1.7098e+00, ..., -3.7106e+00,
-1.0684e+01, -7.1733e+00],
[ 9.0240e-01, 1.4221e+00, 9.0040e-01, ..., -1.6670e+00,
-6.3935e+00, -6.5290e+00],
...,
[ 1.2188e+00, 3.1351e+00, 8.6441e-01, ..., -2.2487e-01,
-1.7688e+00, -3.5868e+00],
[ 8.3377e-01, 3.3738e+00, 5.2877e-01, ..., -3.0140e+00,
-6.6425e+00, -6.1590e+00],
[ 1.7023e+00, 2.5423e+00, 1.2855e+00, ..., -3.4347e-01,
1.6648e+00, 1.0420e+00]],
...,
[[-1.5010e+00, -4.7970e+00, -3.9932e+00, ..., -5.9837e+00,
-5.9995e+00, -9.9315e+00],
[-1.9583e+00, -4.3462e+00, -4.5904e+00, ..., -9.0674e+00,
-6.2343e+00, -9.2047e+00],
[-1.1428e+00, -3.5207e+00, -3.6750e+00, ..., -1.2185e+01,
-2.4397e+00, -2.1349e+00],
...,
[-1.8304e+00, -5.1639e+00, -5.2268e+00, ..., -2.3670e-01,
-6.6465e+00, -7.8794e+00],
[-1.7358e+00, -5.1622e+00, -5.5664e+00, ..., -3.7005e+00,
-6.1997e+00, -8.2919e+00],
[-1.7463e+00, -5.3660e+00, -3.9168e+00, ..., -3.5969e+00,
-1.9796e+00, -5.6761e+00]],
[[ 6.7561e-01, 4.4479e-01, -9.5954e-02, ..., -2.4465e+00,
-8.0389e+00, -3.2938e+00],
[ 1.2126e+00, 9.6559e-01, 5.2211e-01, ..., 7.1675e-01,
-8.2809e-01, -7.9402e-01],
[ 5.4865e-01, 5.1268e-01, -5.6056e-02, ..., -2.8847e+00,
3.5959e-01, 1.0155e+00],
...,
[ 8.1162e-01, 7.6378e-01, -6.5153e-01, ..., -7.1671e-01,
-7.1968e-01, 1.2198e+00],
[ 1.6147e+00, 1.7270e+00, 1.9389e+00, ..., 9.7838e-01,
-1.4083e-01, 2.1384e+00],
[ 6.8111e-01, 2.1120e-01, 4.9614e-01, ..., 2.6826e-01,
1.4007e+00, 1.7303e+00]],
[[-3.2970e-01, -1.5214e-01, -9.1305e-01, ..., -1.0230e+00,
-4.6037e-01, 1.2635e+00],
[-1.4385e+00, -1.3380e+00, -1.8882e+00, ..., -5.5316e+00,
-4.5359e+00, 2.7279e+00],
[-1.7354e+00, -7.6814e-01, -8.5142e-01, ..., -5.4574e+00,
-8.4022e+00, -2.2807e+00],
...,
[-1.7879e+00, -1.0226e+00, -1.8397e+00, ..., -1.9787e+00,
-2.1466e+00, 2.0608e+00],
[-1.4527e+00, -1.1562e+00, -1.5211e+00, ..., -1.7845e-01,
-2.9023e+00, -3.5975e+00],
[-1.4189e-01, 5.0143e-01, 1.8018e-01, ..., -1.1298e+00,
-4.4341e-01, 8.0173e-02]]],
[[[ 5.9837e-01, 8.9801e-01, 6.5686e-01, ..., 3.3471e-01,
4.7778e-01, 1.6746e-01],
[-1.5972e-01, -2.3766e-01, 5.7470e-01, ..., 8.2366e-01,
2.2291e-01, -4.5694e-01],
[ 2.2284e-02, 3.3670e-01, 4.5660e-01, ..., 6.9794e-01,
-1.1409e-02, 3.6668e-01],
...,
[ 1.4412e+00, 4.9234e-01, 2.0499e-01, ..., -4.3159e+00,
-1.7417e+00, 1.3403e+00],
[ 1.0861e-02, -4.0272e-01, -7.9550e-01, ..., -4.9210e-01,
-3.7757e-01, 3.9255e-01],
[ 8.7712e-01, -6.0367e-01, -1.7570e+00, ..., 3.8978e-01,
1.0671e+00, 1.3864e+00]],
[[-2.2960e+00, -1.8267e-01, -7.0641e-01, ..., -2.6042e+00,
-2.8365e+00, -2.0537e+00],
[-2.1085e+00, 9.4437e-01, 1.6894e+00, ..., -2.9628e+00,
-3.5937e+00, -2.5208e+00],
[-3.0233e+00, -2.0491e+00, -2.7713e+00, ..., -1.5275e+00,
-1.8964e+00, -1.0727e+00],
...,
[-1.2388e+00, -1.3881e+00, -1.4509e+00, ..., -1.0348e+01,
-5.4470e+00, -8.2567e-01],
[-2.8235e+00, -7.4295e-01, -2.2569e+00, ..., -4.0600e+00,
-4.2619e+00, -2.4371e+00],
[-2.0787e+00, -2.4309e+00, -3.5309e+00, ..., -4.8393e+00,
-4.4430e+00, -3.3877e+00]],
[[ 1.7206e+00, 1.7668e+00, 4.7082e-01, ..., -4.0774e-01,
7.8826e-02, 2.0188e+00],
[-2.8299e-01, 2.0466e-01, 3.5979e-01, ..., -1.2151e+00,
-1.0971e+00, 1.9446e+00],
[ 1.1133e+00, 8.4524e-01, 8.2242e-01, ..., -8.1019e-01,
1.9592e-01, 1.7419e+00],
...,
[ 1.0290e+00, -8.0720e-01, -1.2269e+00, ..., -6.2164e+00,
-3.4980e+00, -4.7894e-02],
[ 1.3598e+00, 3.1208e+00, 3.8780e+00, ..., -4.1375e+00,
-5.7586e+00, -7.7921e-01],
[ 2.4916e+00, 1.9967e+00, 3.2903e+00, ..., -6.0304e+00,
-6.9311e+00, -2.3392e+00]],
...,
[[-1.7689e+00, -4.7947e+00, -2.7363e+00, ..., -1.1997e+00,
-4.0671e-01, -3.7472e+00],
[-2.1104e+00, -5.2420e+00, -2.9717e+00, ..., -2.2075e+00,
-3.1267e+00, -4.2451e+00],
[-1.7817e+00, -5.0470e+00, -2.4738e+00, ..., -5.6990e+00,
-5.4051e+00, -3.0970e+00],
...,
[-2.1005e+00, -4.0353e+00, -1.7521e+00, ..., -3.5173e+00,
-1.0113e+00, -5.3023e+00],
[-2.0669e+00, -5.7300e+00, -4.2414e+00, ..., -1.1226e+01,
-4.0556e+00, -8.9614e+00],
[-2.6502e+00, -5.7641e+00, -3.0674e+00, ..., -9.6790e+00,
-5.1773e+00, -7.8263e+00]],
[[ 7.1905e-01, 1.4018e+00, 4.8614e-01, ..., -3.7906e-01,
4.0833e-01, 1.1229e+00],
[ 1.2720e+00, 1.5630e+00, 1.6657e+00, ..., -1.7933e-01,
8.9675e-01, 2.0196e+00],
[ 2.2028e-01, -3.7394e-01, -1.6306e-01, ..., 5.4414e-01,
1.2550e-01, 1.2948e+00],
...,
[ 1.2632e+00, -6.7215e-01, -9.0701e-02, ..., -7.2715e+00,
-2.8540e+00, 8.7310e-02],
[ 2.0379e+00, 1.4663e-01, -8.8034e-01, ..., -4.8916e+00,
-2.7843e-01, 2.6583e+00],
[ 1.1264e+00, 6.9791e-01, -5.1042e-01, ..., -2.0387e+00,
1.2739e-02, 5.9780e-01]],
[[-2.1859e-01, -8.2263e-01, -1.4521e+00, ..., -1.0709e+00,
-5.9147e-01, -2.1704e-01],
[-1.0840e+00, -2.0136e+00, -2.0069e+00, ..., -8.9608e-01,
-2.3772e-01, -1.2233e+00],
[-1.5715e+00, -9.2413e-01, -5.6051e-01, ..., -1.2831e-01,
-1.1498e+00, -8.6000e-01],
...,
[-1.3548e+00, -1.4611e+00, -5.9662e-01, ..., -8.0715e+00,
-3.8842e+00, 7.2444e-01],
[-1.6388e+00, -1.0767e+00, -1.1370e-01, ..., -2.6413e+00,
-6.0396e-01, 2.9291e+00],
[-1.9893e-01, 1.6755e-01, -8.6419e-01, ..., -3.5147e+00,
-2.0365e+00, -8.7792e-02]]],
[[[ 1.0952e+00, 1.4560e+00, 1.2096e+00, ..., 2.5329e-01,
3.5277e-01, 1.9707e-01],
[-9.9044e-01, -1.5083e+00, -9.0523e-01, ..., 3.4559e-01,
3.5638e-01, -1.1221e+00],
[-4.5796e-01, -5.6349e-01, -2.0366e-01, ..., -3.3279e-01,
9.9687e-01, 6.2872e-01],
...,
[ 8.5284e-02, -2.7714e-01, 4.3032e-01, ..., -4.3022e-01,
6.5310e-01, 4.3377e-01],
[-4.7978e-01, -8.0638e-01, -1.9483e-01, ..., 3.9415e-01,
1.3585e+00, -1.3633e-01],
[-2.1254e-01, -2.2167e-01, -7.5062e-01, ..., -1.8713e-01,
4.0008e-01, 5.9176e-01]],
[[-3.1327e+00, -1.9801e+00, -2.1974e+00, ..., -1.8090e+00,
-2.2475e+00, -1.1345e+00],
[-2.8261e+00, -3.4149e+00, -1.6405e+00, ..., -3.5964e-01,
-1.2525e+00, -3.3039e-01],
[-2.0111e+00, -2.2736e+00, -1.6121e+00, ..., -1.2369e+00,
-1.5028e+00, -7.7188e-01],
...,
[-1.4209e-01, -1.1837e+00, -4.1867e-01, ..., -2.2344e+00,
-1.6227e-01, 5.1011e-01],
[-1.2419e+00, -8.3285e-01, -3.8812e-01, ..., -1.4945e-01,
-3.1918e-01, 2.7242e-01],
[-4.4242e-01, -9.8849e-01, -8.3969e-01, ..., -2.5441e-01,
-7.1392e-01, -5.4107e-01]],
[[ 2.3357e+00, 2.3428e+00, 1.7946e+00, ..., 2.9386e+00,
1.4267e+00, 2.9460e+00],
[ 2.1809e-01, 1.8859e-01, 1.4874e+00, ..., 1.9639e+00,
2.8093e-02, 2.5834e+00],
[ 4.2277e-02, 4.0778e-01, 1.2780e+00, ..., 1.2368e+00,
6.6908e-01, 1.8818e+00],
...,
[-3.3745e-01, 3.6246e-01, 1.0662e+00, ..., 9.8365e-02,
1.9649e-01, 1.2719e+00],
[-1.0237e+00, 4.4975e-01, 4.1735e-01, ..., 1.3536e+00,
4.8568e-01, 1.4634e+00],
[ 6.3825e-01, 4.7203e-01, 1.0416e+00, ..., 1.4798e+00,
1.0765e+00, 2.0107e+00]],
...,
[[-2.9034e+00, -8.4357e+00, -6.7054e+00, ..., -3.6117e+00,
-2.4701e+00, -3.9449e+00],
[-3.7731e+00, -1.0251e+01, -6.3342e+00, ..., -5.9256e+00,
-3.9090e+00, -4.4332e+00],
[-2.6252e+00, -8.2012e+00, -5.0928e+00, ..., -3.2402e+00,
-2.4506e+00, -1.6537e+00],
...,
[-3.4616e+00, -9.1855e+00, -5.7826e+00, ..., -3.9400e+00,
-3.0943e+00, -2.1475e+00],
[-4.1331e+00, -9.6096e+00, -5.2239e+00, ..., -6.3174e+00,
-4.6478e+00, -3.9815e+00],
[-3.2262e+00, -7.7413e+00, -3.0643e+00, ..., -3.5936e+00,
-2.1619e+00, -2.7959e+00]],
[[-3.6547e-01, 7.2198e-01, 9.0899e-01, ..., 1.3047e+00,
1.0814e+00, 1.2312e+00],
[ 8.3708e-01, 7.0815e-01, 1.3673e+00, ..., 1.9840e+00,
2.3828e+00, 2.5611e+00],
[ 7.2876e-01, 7.2892e-01, 1.4656e+00, ..., 1.5146e+00,
1.6416e+00, 1.9578e+00],
...,
[ 1.3116e+00, 6.0986e-01, 1.6527e+00, ..., 1.3099e+00,
2.0051e+00, 1.6753e+00],
[ 1.7005e+00, 1.5445e+00, 1.6111e+00, ..., 1.7595e+00,
1.4830e+00, 2.4525e+00],
[ 7.8190e-01, 6.5040e-01, 1.2040e+00, ..., 1.1958e+00,
1.3268e+00, 1.3766e+00]],
[[-2.1442e-01, -3.8442e-01, -3.4358e-01, ..., -1.8662e-01,
-1.5743e-01, 8.5813e-01],
[-7.3454e-01, -2.7108e+00, -1.1339e+00, ..., 2.0562e-01,
-5.9880e-02, 3.8418e-01],
[-1.5592e+00, -1.6326e+00, -3.9981e-01, ..., 3.5753e-01,
1.5933e-01, 2.3983e-01],
...,
[-1.7228e+00, -2.5535e+00, -6.9620e-01, ..., -1.7831e+00,
-6.0201e-01, -3.0465e-01],
[-1.4033e+00, -2.5965e+00, -7.0153e-01, ..., 2.8788e-02,
-8.3850e-02, 3.4739e-01],
[-4.2349e-01, -1.1952e+00, -6.3200e-01, ..., 1.0189e-01,
1.7643e-01, 3.3712e-01]]],
[[[ 8.1078e-01, 1.0323e+00, 5.4816e-01, ..., 7.2553e-01,
1.0158e+00, 4.5949e-01],
[-1.1225e+00, -1.0919e+00, 7.7253e-02, ..., -2.4017e-01,
-8.5586e-02, -3.1933e-01],
[-3.0557e-01, -1.1395e+00, -1.2190e-02, ..., -5.8540e-01,
-1.7403e-01, 6.4483e-01],
...,
[ 4.7455e-02, 2.7237e-01, 6.3170e-01, ..., -1.0218e-01,
-3.5692e-02, 3.8960e-01],
[ 4.0423e-01, -8.5082e-01, 1.2232e-01, ..., -4.8681e-02,
1.6427e-01, -6.4835e-02],
[ 5.2818e-01, -1.7310e-01, -7.4152e-01, ..., -5.3655e-01,
-9.7786e-01, 1.3147e-01]],
[[-4.0897e+00, -2.0905e+00, -3.0668e+00, ..., -1.5861e+00,
-2.0327e+00, -1.8002e+00],
[-3.4326e+00, -2.2432e+00, -1.2769e+00, ..., -1.4931e+00,
-2.0417e+00, -2.1225e+00],
[-2.6032e+00, -2.6077e+00, -2.6845e+00, ..., -1.4251e+00,
-1.5255e+00, -1.3799e+00],
...,
[-4.7846e-02, -2.3148e+00, -2.9864e+00, ..., -7.0050e-01,
-8.8702e-01, -8.5960e-01],
[-1.4918e+00, -1.6357e-01, -2.0317e-01, ..., -6.4856e-01,
-1.0338e+00, -1.2179e+00],
[ 5.7550e-02, -6.3712e-01, -8.0329e-01, ..., -7.7453e-01,
-6.7324e-01, -5.0067e-01]],
[[ 2.7504e+00, 3.4236e+00, 3.2752e+00, ..., 1.4876e+00,
8.2483e-01, 2.5746e+00],
[ 7.3352e-01, 1.0855e+00, 2.7617e+00, ..., 1.2528e+00,
6.8092e-01, 2.5889e+00],
[ 1.2273e+00, 1.0631e+00, 2.3882e+00, ..., -2.7609e-01,
9.7025e-01, 1.7855e+00],
...,
[-6.2827e-01, -2.2017e+00, -2.8460e+00, ..., -6.9742e-01,
5.9351e-01, 1.5418e+00],
[ 3.1831e-01, 1.4668e+00, 1.6460e+00, ..., -6.6308e-01,
7.2296e-01, 1.1575e+00],
[ 1.6323e+00, 1.1914e+00, 1.8626e+00, ..., 2.5449e-01,
1.2701e+00, 1.4014e+00]],
...,
[[-2.9061e+00, -8.1928e+00, -5.3717e+00, ..., -5.6889e+00,
-3.5293e+00, -4.2534e+00],
[-3.3332e+00, -9.4863e+00, -7.2057e+00, ..., -5.5430e+00,
-3.4966e+00, -3.2218e+00],
[-2.2851e+00, -6.2588e+00, -3.5329e+00, ..., -4.6623e+00,
-2.0230e+00, -1.2629e+00],
...,
[-2.2304e+00, -6.0736e+00, -2.7034e+00, ..., -5.2213e+00,
-3.2393e+00, -2.7977e+00],
[-2.7203e+00, -6.5171e+00, -4.2166e+00, ..., -5.7145e+00,
-4.4081e+00, -4.3586e+00],
[-2.4015e+00, -6.6769e+00, -3.6307e+00, ..., -3.4698e+00,
-2.4653e+00, -2.4062e+00]],
[[-7.6394e-01, -1.9398e-01, 1.9268e-01, ..., 1.8524e+00,
1.6652e+00, 1.1655e+00],
[ 1.4216e+00, 1.0096e+00, 1.2979e+00, ..., 1.9805e+00,
1.1795e+00, 1.8037e+00],
[ 8.4970e-01, -8.0099e-01, -1.4859e+00, ..., 1.6484e+00,
6.2395e-01, 1.2500e+00],
...,
[ 1.3778e+00, -6.0518e-01, -6.2677e-01, ..., 1.8999e+00,
6.7936e-01, 9.4740e-01],
[ 2.0201e+00, 1.3436e+00, 1.2780e+00, ..., 1.7571e+00,
6.7388e-01, 1.4310e+00],
[ 1.2014e+00, 9.1059e-01, 1.1703e+00, ..., 8.3475e-01,
6.3806e-01, 1.4145e+00]],
[[-7.4846e-01, -7.1936e-01, -6.5671e-01, ..., -3.2350e-01,
-5.8501e-01, -7.5583e-02],
[-1.4797e+00, -1.5688e+00, -3.8809e-01, ..., -1.6188e+00,
-1.3182e+00, -1.1506e+00],
[-2.2074e+00, -9.6842e-01, -5.9284e-02, ..., -8.5828e-01,
-1.0373e+00, -2.6774e-01],
...,
[-1.1321e+00, -1.9262e+00, -1.5695e+00, ..., -5.5958e-01,
-1.5067e+00, -8.4313e-01],
[-1.9483e+00, -1.5831e+00, -9.4531e-01, ..., -7.3708e-01,
-1.5389e+00, -7.4700e-01],
[-1.4969e-02, 1.2621e-01, -2.2389e-01, ..., -4.0420e-01,
-7.8214e-01, -2.5960e-02]]],
[[[ 9.5090e-01, 1.5134e+00, 1.2563e+00, ..., 6.0560e-01,
9.7098e-01, 4.7609e-01],
[-6.6167e-01, -2.0116e+00, -8.9848e-01, ..., -6.1246e-01,
-2.4720e-01, -5.4623e-01],
[ 1.3866e-01, -6.2022e-01, -9.0554e-01, ..., -5.2394e-02,
-4.2092e-01, 4.7626e-01],
...,
[ 3.2083e-01, -6.3414e-01, -2.3098e-01, ..., 2.3622e-01,
-1.5209e-01, 3.4648e-01],
[-1.6804e-01, -1.3473e+00, -3.6426e-01, ..., -4.1080e-02,
1.2101e-02, -1.4885e-01],
[ 7.9400e-01, -3.5453e-01, -2.2377e-01, ..., -3.5376e-01,
-9.6876e-01, 1.5420e-01]],
[[-4.2520e+00, -9.6351e-01, -2.2375e+00, ..., -1.6743e+00,
-2.2024e+00, -1.8276e+00],
[-4.0448e+00, -2.7341e+00, -3.6208e+00, ..., -1.2896e+00,
-2.5130e+00, -2.2375e+00],
[-2.5525e+00, -2.2648e+00, -6.2758e+00, ..., -1.3316e+00,
-1.6981e+00, -1.3296e+00],
...,
[-9.4706e-01, -1.3154e+00, -3.0250e+00, ..., -4.6301e-01,
-8.3651e-01, -7.5681e-01],
[-2.9222e+00, -8.1040e-01, -1.6706e+00, ..., -5.8395e-01,
-9.5281e-01, -1.1668e+00],
[-2.0954e+00, -1.5020e+00, -1.4383e+00, ..., -7.1153e-01,
-5.0025e-01, -3.7237e-01]],
[[ 3.1036e+00, 4.1354e+00, 3.3673e+00, ..., 1.3697e+00,
7.7551e-01, 2.6378e+00],
[ 5.1033e-01, 6.8542e-01, 2.3721e+00, ..., 2.0153e+00,
5.5993e-01, 2.7427e+00],
[ 9.0147e-01, 1.3081e+00, 2.8743e+00, ..., 5.6917e-01,
1.3060e+00, 2.1055e+00],
...,
[ 1.5646e-01, 7.2997e-01, 2.4811e+00, ..., -3.4890e-01,
6.5339e-01, 1.5831e+00],
[ 3.3531e-01, 1.8068e+00, 1.6866e+00, ..., -1.6840e-01,
8.3141e-01, 1.1891e+00],
[ 1.4030e+00, 1.3729e+00, 1.6873e+00, ..., 5.6212e-01,
1.2380e+00, 1.3488e+00]],
...,
[[-2.1343e+00, -7.1845e+00, -5.8539e+00, ..., -5.3957e+00,
-4.1251e+00, -4.6645e+00],
[-2.9267e+00, -6.4266e+00, -4.5295e+00, ..., -4.8669e+00,
-4.2482e+00, -3.6662e+00],
[-1.9725e-01, -2.9096e+00, -4.2417e+00, ..., -3.8402e+00,
-2.2481e+00, -1.4325e+00],
...,
[-1.0177e+00, -2.7673e+00, -2.6979e+00, ..., -5.1078e+00,
-3.5704e+00, -2.8911e+00],
[-1.2921e+00, -3.8004e+00, -4.0447e+00, ..., -5.2988e+00,
-4.3955e+00, -4.2097e+00],
[-2.2379e+00, -5.0296e+00, -3.0956e+00, ..., -3.5020e+00,
-2.5274e+00, -2.4391e+00]],
[[-6.2637e-01, 1.0125e-01, -1.0806e-01, ..., 1.4547e+00,
1.6693e+00, 1.2153e+00],
[ 1.7908e+00, 1.0797e+00, 6.5822e-01, ..., 1.9598e+00,
1.1945e+00, 1.8845e+00],
[ 1.2262e+00, -6.6576e-01, -1.4468e+00, ..., 1.0036e+00,
7.3469e-01, 1.5550e+00],
...,
[ 1.2686e+00, -7.5840e-01, -1.8346e+00, ..., 1.8595e+00,
7.1748e-01, 9.9492e-01],
[ 1.9142e+00, 7.0302e-01, 1.4474e+00, ..., 1.3120e+00,
6.9365e-01, 1.5575e+00],
[ 1.0280e+00, 3.6192e-01, 9.7364e-01, ..., 7.8121e-01,
7.7707e-01, 1.4605e+00]],
[[-4.9675e-01, 6.5650e-01, -7.8155e-02, ..., -4.7513e-01,
-6.4890e-01, 2.8156e-03],
[-1.0025e+00, -1.1317e+00, -2.3654e+00, ..., -1.4580e+00,
-1.4402e+00, -1.1199e+00],
[-1.9707e+00, -1.2866e+00, -1.2813e+00, ..., -1.5434e+00,
-1.9160e+00, -1.6065e-01],
...,
[-1.6878e+00, -1.7161e+00, 2.3463e-03, ..., -5.1666e-01,
-1.3821e+00, -8.7930e-01],
[-1.7608e+00, -1.3096e+00, -1.3237e+00, ..., -6.0220e-01,
-1.5482e+00, -6.8101e-01],
[-3.5540e-01, 3.4188e-01, -3.3167e-01, ..., -3.1980e-01,
-7.5192e-01, 1.3073e-02]]]], device='cuda:0'),
tensor([[[[ 1.0547e+00, 1.3597e+00, 1.2195e+00, ..., 1.2474e+00,
9.9568e-01, 4.6802e-01],
[-5.3775e-01, -2.3351e+00, -1.0637e+00, ..., 7.4418e-03,
-5.2271e-02, -1.4566e+00],
[-3.1366e-01, -1.1508e+00, -5.8047e-01, ..., -6.4648e-01,
9.7043e-02, 5.9275e-01],
...,
[ 5.7195e-01, 2.2740e-02, 2.0397e-01, ..., 4.2768e-01,
1.0250e+00, 8.0606e-01],
[-5.1782e-02, -7.0631e-01, -8.0761e-02, ..., 9.1096e-01,
1.5289e+00, 1.1849e-01],
[ 7.9648e-01, -1.8612e-01, -8.8465e-02, ..., 4.1312e-01,
2.7773e-01, 7.0587e-01]],
[[-2.8708e+00, -1.9517e-02, -1.6011e+00, ..., -1.5270e+00,
-2.3803e+00, -6.6897e-01],
[-4.0107e+00, -2.5873e+00, -2.6279e+00, ..., -2.5431e+00,
-3.9094e+00, -1.6253e+00],
[-2.9126e+00, -2.2280e+00, -2.8376e+00, ..., -3.1505e+00,
-2.9754e+00, -1.3923e+00],
...,
[-6.5118e-01, -2.8172e-01, -9.2281e-01, ..., -5.1625e-01,
5.8150e-02, 7.5618e-01],
[-2.5720e+00, -3.9163e-01, -6.1040e-01, ..., -5.2232e-02,
-6.8734e-02, 6.6920e-01],
[-1.5517e+00, -1.2061e+00, -9.3091e-01, ..., -4.8090e-01,
-6.1413e-01, -1.4174e-01]],
[[ 2.5247e+00, 2.8898e+00, 2.5319e+00, ..., 2.0927e+00,
5.5914e-01, 2.6058e+00],
[ 9.6386e-01, 5.8393e-01, 2.3961e+00, ..., 1.8813e+00,
4.8459e-01, 2.6707e+00],
[ 1.2581e+00, 1.3483e+00, 1.8674e+00, ..., 1.3496e+00,
1.8575e+00, 2.8794e+00],
...,
[ 8.1880e-01, 8.6802e-01, 2.0853e+00, ..., 9.7181e-01,
5.7501e-01, 1.8567e+00],
[ 1.0219e+00, 2.2093e+00, 1.9575e+00, ..., 1.0078e+00,
7.1817e-01, 1.6958e+00],
[ 2.4019e+00, 1.8631e+00, 2.3968e+00, ..., 1.9949e+00,
1.0031e+00, 1.6863e+00]],
...,
[[-2.6347e+00, -6.7084e+00, -5.6429e+00, ..., -3.9286e+00,
-4.2782e+00, -5.4801e+00],
[-3.2263e+00, -7.0980e+00, -5.2178e+00, ..., -4.6365e+00,
-5.0276e+00, -5.1843e+00],
[-2.1749e+00, -5.8559e+00, -4.2490e+00, ..., -3.0392e+00,
-3.7395e+00, -3.3492e+00],
...,
[-2.5088e+00, -5.1283e+00, -2.9604e+00, ..., -3.3126e+00,
-2.9306e+00, -3.0709e+00],
[-2.5019e+00, -5.5122e+00, -3.9164e+00, ..., -4.3813e+00,
-3.2662e+00, -4.8935e+00],
[-2.6214e+00, -5.9256e+00, -2.9349e+00, ..., -2.9411e+00,
-1.7040e+00, -3.3436e+00]],
[[ 3.5992e-01, 1.1172e+00, 3.3902e-01, ..., 7.6670e-01,
1.5415e+00, 1.6788e+00],
[ 1.3470e+00, 8.9494e-01, 3.0743e-01, ..., 3.7996e-01,
1.3139e+00, 2.7534e+00],
[ 1.3677e+00, 8.0007e-01, 5.7052e-01, ..., 9.5933e-01,
1.1756e+00, 2.0602e+00],
...,
[ 1.6196e+00, 4.6044e-01, 8.7708e-01, ..., 1.9564e+00,
1.9593e+00, 1.4189e+00],
[ 1.9849e+00, 1.1045e+00, 1.8647e+00, ..., 1.9394e+00,
1.6666e+00, 2.5485e+00],
[ 1.1741e+00, 8.7763e-01, 1.3394e+00, ..., 1.3545e+00,
1.5164e+00, 1.5329e+00]],
[[-6.6181e-01, 8.5006e-02, -2.0986e-01, ..., 2.9954e-01,
1.1493e-02, 9.3247e-01],
[-1.5741e+00, -1.9426e+00, -1.5923e+00, ..., -8.0318e-01,
-7.7043e-01, -9.3682e-02],
[-2.2111e+00, -1.3041e+00, -2.6454e-01, ..., -2.8376e-01,
-3.8013e-01, -1.2444e-01],
...,
[-1.9106e+00, -1.3415e+00, -5.0841e-01, ..., -1.5862e-01,
-4.1711e-01, -4.4623e-01],
[-1.6899e+00, -1.4071e+00, -8.1910e-01, ..., -3.6016e-01,
-3.9013e-01, 4.4395e-01],
[-2.7644e-01, 4.2414e-01, -1.9143e-01, ..., 2.3568e-01,
5.2383e-02, 6.6276e-01]]],
[[[ 7.1475e-01, 1.0559e+00, 6.6144e-01, ..., 3.8627e-01,
2.9153e-01, 1.8335e-01],
[-3.3285e-01, -1.4450e+00, -5.7422e-01, ..., 4.5148e-01,
4.2285e-01, -1.1067e+00],
[ 1.6679e-01, -4.7892e-01, 3.0804e-01, ..., -4.2652e-01,
1.0611e+00, 6.2470e-01],
...,
[ 8.0271e-02, -1.2408e+00, -7.6073e-01, ..., -7.2495e-01,
4.4113e-01, 3.9802e-01],
[-4.4295e-01, -1.8237e+00, -7.5311e-01, ..., 5.7576e-02,
1.2417e+00, -1.3706e-01],
[ 3.6158e-01, -5.4303e-01, -6.8540e-01, ..., -2.1656e-01,
4.1138e-01, 5.8218e-01]],
[[-3.1275e+00, -1.5043e+00, -3.0990e+00, ..., -1.8782e+00,
-2.2583e+00, -1.0430e+00],
[-4.4564e+00, -1.8721e+00, -2.3868e+00, ..., -4.5958e-01,
-1.2375e+00, -2.4780e-01],
[-4.3901e+00, -2.9786e+00, -4.2479e+00, ..., -1.4257e+00,
-1.5310e+00, -7.4326e-01],
...,
[-2.8621e+00, -1.9671e+00, -2.9188e+00, ..., -2.7127e+00,
-1.9513e+00, 5.8552e-02],
[-3.6268e+00, -1.3944e+00, -2.2758e+00, ..., -2.9322e-01,
-4.7664e-01, 2.2533e-01],
[-1.2332e+00, -8.4892e-01, -1.7109e+00, ..., -3.4542e-01,
-7.3979e-01, -5.6678e-01]],
[[ 1.7359e+00, 2.8122e+00, 2.3452e+00, ..., 2.7914e+00,
1.4984e+00, 2.8595e+00],
[ 1.3206e-01, 1.1966e+00, 1.5600e+00, ..., 1.8561e+00,
2.8134e-02, 2.4871e+00],
[ 8.3445e-01, 1.6218e+00, 1.2340e+00, ..., 1.0461e+00,
5.8117e-01, 1.7977e+00],
...,
[ 9.8750e-01, 1.2624e+00, 8.2224e-01, ..., -1.4808e+00,
-6.4254e-01, 1.6365e+00],
[ 6.6958e-01, 1.8204e+00, 8.6910e-01, ..., 1.1060e+00,
7.9527e-01, 1.6042e+00],
[ 1.5312e+00, 1.5435e+00, 1.6450e+00, ..., 1.4337e+00,
1.1261e+00, 2.0115e+00]],
...,
[[-1.4966e+00, -4.7370e+00, -4.0470e+00, ..., -3.3358e+00,
-2.5591e+00, -3.8701e+00],
[-1.8559e+00, -3.9456e+00, -4.4224e+00, ..., -5.2832e+00,
-4.0857e+00, -4.4345e+00],
[-1.0060e+00, -3.3754e+00, -4.5043e+00, ..., -2.4411e+00,
-2.5464e+00, -1.6573e+00],
...,
[-1.5725e+00, -4.1159e+00, -4.4258e+00, ..., -3.9128e+00,
-2.0130e+00, -1.5402e+00],
[-1.4569e+00, -4.3983e+00, -5.1962e+00, ..., -6.1386e+00,
-4.7343e+00, -4.5713e+00],
[-1.5625e+00, -4.7766e+00, -4.1026e+00, ..., -3.6436e+00,
-2.2296e+00, -2.6663e+00]],
[[ 7.8452e-01, 5.4233e-01, -1.1066e-01, ..., 8.3857e-01,
1.0976e+00, 1.2778e+00],
[ 1.1885e+00, 9.7493e-01, 5.2519e-01, ..., 1.5170e+00,
2.3811e+00, 2.5822e+00],
[ 5.0865e-01, 5.1427e-01, -2.0908e-01, ..., 1.0050e+00,
1.6144e+00, 1.9409e+00],
...,
[ 8.0449e-01, -3.9607e-02, 6.1553e-02, ..., -4.9498e-01,
1.2793e+00, 1.7680e+00],
[ 1.5100e+00, 1.4110e+00, 1.1851e+00, ..., 1.5477e+00,
1.5761e+00, 2.5047e+00],
[ 6.7749e-01, 8.8754e-01, 1.0341e+00, ..., 1.0677e+00,
1.3046e+00, 1.3923e+00]],
[[-2.7161e-01, -2.0167e-01, -1.0914e+00, ..., -1.4898e-01,
-1.7420e-01, 8.8712e-01],
[-1.4035e+00, -1.4542e+00, -2.1346e+00, ..., 2.8151e-01,
-1.5788e-02, 4.5733e-01],
[-1.7516e+00, -6.5864e-01, -6.5537e-01, ..., 2.5893e-01,
2.0603e-01, 2.4467e-01],
...,
[-1.7238e+00, -1.6909e+00, -1.8032e+00, ..., -2.1483e+00,
-1.4674e+00, -2.7827e-01],
[-1.4051e+00, -1.6055e+00, -1.5263e+00, ..., -4.5658e-01,
-2.1816e-01, 5.8414e-01],
[-1.5292e-01, 4.3860e-02, -4.8212e-01, ..., 2.5026e-01,
1.3746e-01, 2.8419e-01]]],
[[[ 9.6252e-01, 1.5996e+00, 1.1523e+00, ..., 5.4042e-01,
1.0195e+00, 4.4928e-01],
[-9.1973e-01, -1.9932e+00, -7.1892e-01, ..., -4.3454e-01,
-1.3892e-01, -3.1688e-01],
[-5.6280e-03, -6.9470e-01, -3.4335e-01, ..., 1.9210e-01,
-1.4703e-01, 6.1728e-01],
...,
[ 4.2087e-01, -3.2636e-01, 7.7577e-02, ..., 7.7133e-03,
-2.8507e-02, 4.3673e-01],
[-2.2032e-01, -1.1489e+00, -2.4763e-01, ..., -3.0912e-02,
1.0188e-01, -1.6124e-02],
[ 7.4407e-01, -3.4286e-01, -1.9602e-01, ..., -3.6824e-01,
-9.2329e-01, 6.8838e-02]],
[[-4.2684e+00, -1.3846e+00, -2.2322e+00, ..., -1.6406e+00,
-2.0383e+00, -1.7752e+00],
[-4.0616e+00, -2.8319e+00, -2.8335e+00, ..., -1.1410e+00,
-1.9533e+00, -2.0554e+00],
[-2.5885e+00, -2.1960e+00, -4.8925e+00, ..., -1.2555e+00,
-1.5045e+00, -1.3237e+00],
...,
[-8.3457e-01, -8.5860e-01, -1.4484e+00, ..., -6.9511e-01,
-8.4099e-01, -8.5539e-01],
[-2.9031e+00, -5.3282e-01, -1.0629e+00, ..., -7.1667e-01,
-9.7093e-01, -1.1807e+00],
[-2.0978e+00, -1.5353e+00, -1.4363e+00, ..., -7.8252e-01,
-7.0039e-01, -4.6723e-01]],
[[ 3.0341e+00, 4.1510e+00, 3.2836e+00, ..., 1.5883e+00,
7.6883e-01, 2.5329e+00],
[ 3.3893e-01, 8.8083e-01, 2.3031e+00, ..., 1.8374e+00,
6.7470e-01, 2.5549e+00],
[ 8.5434e-01, 1.2352e+00, 2.2772e+00, ..., -3.4029e-01,
9.3950e-01, 1.7180e+00],
...,
[ 2.8317e-01, 9.0200e-01, 1.7583e+00, ..., -3.8366e-01,
4.3307e-01, 1.4284e+00],
[ 3.6129e-01, 2.0955e+00, 1.5873e+00, ..., -2.9096e-01,
6.7797e-01, 1.0861e+00],
[ 1.4511e+00, 1.3402e+00, 1.6163e+00, ..., 5.2371e-01,
1.2657e+00, 1.2901e+00]],
...,
[[-2.0705e+00, -7.4572e+00, -5.9131e+00, ..., -3.6265e+00,
-3.5454e+00, -4.1708e+00],
[-2.6823e+00, -6.5085e+00, -4.8002e+00, ..., -4.2366e+00,
-3.5587e+00, -3.1528e+00],
[-1.8976e-01, -2.7678e+00, -3.9487e+00, ..., -5.0930e+00,
-2.2694e+00, -1.1272e+00],
...,
[-9.3721e-01, -2.4450e+00, -1.6150e+00, ..., -3.9851e+00,
-2.8926e+00, -2.8179e+00],
[-1.3437e+00, -3.7502e+00, -3.3527e+00, ..., -5.1259e+00,
-4.0344e+00, -4.3315e+00],
[-2.2586e+00, -5.0531e+00, -3.0515e+00, ..., -3.3469e+00,
-2.3812e+00, -2.2766e+00]],
[[-7.8916e-01, -1.8304e-01, 6.4639e-02, ..., 1.6423e+00,
1.6769e+00, 1.1822e+00],
[ 1.7987e+00, 1.0972e+00, 9.0313e-01, ..., 2.1463e+00,
1.2496e+00, 1.7947e+00],
[ 1.1926e+00, -2.3683e-01, -3.8595e-01, ..., 1.8352e+00,
5.9479e-01, 1.2632e+00],
...,
[ 1.2867e+00, -2.6570e-01, 1.0779e+00, ..., 1.3638e+00,
6.8697e-01, 8.3912e-01],
[ 1.8727e+00, 8.9904e-01, 1.6965e+00, ..., 1.3114e+00,
8.4127e-01, 1.3632e+00],
[ 1.0330e+00, 4.0356e-01, 9.9816e-01, ..., 7.1201e-01,
6.6255e-01, 1.4210e+00]],
[[-5.0078e-01, 5.7484e-01, 6.2216e-02, ..., -1.0746e-01,
-5.9436e-01, -6.1526e-02],
[-1.0922e+00, -1.1297e+00, -2.0138e+00, ..., -1.4060e+00,
-1.3758e+00, -1.1201e+00],
[-2.0540e+00, -1.0177e+00, -8.0990e-01, ..., -8.9235e-01,
-1.1453e+00, -2.6896e-01],
...,
[-1.7523e+00, -1.4975e+00, -5.7057e-01, ..., -1.0312e+00,
-1.3427e+00, -9.0919e-01],
[-1.7686e+00, -1.1238e+00, -8.8323e-01, ..., -8.8013e-01,
-1.5620e+00, -7.7147e-01],
[-3.6007e-01, 3.8926e-01, -3.1024e-01, ..., -3.1958e-01,
-9.4827e-01, -1.1848e-02]]],
[[[ 6.9753e-01, 1.0725e+00, 8.1491e-01, ..., 3.9347e-01,
3.4879e-01, 2.3880e-01],
[-4.1577e-01, -1.2477e+00, -2.6630e-01, ..., 7.6541e-02,
2.1260e-01, -1.2674e+00],
[-4.7316e-01, -9.8810e-01, -1.1046e+00, ..., -1.7994e+00,
5.4756e-01, 3.2524e-01],
...,
[ 1.9350e-02, -1.1733e+00, -8.0875e-01, ..., 8.0979e-02,
1.0638e+00, 4.9217e-01],
[-5.2789e-01, -1.8535e+00, -8.8973e-01, ..., 4.7641e-02,
1.4495e+00, -1.3630e-01],
[ 2.7770e-01, -6.3396e-01, -8.4134e-01, ..., -2.4750e-01,
3.9581e-01, 5.9069e-01]],
[[-3.3515e+00, -1.5712e+00, -3.2297e+00, ..., -1.7017e+00,
-2.1860e+00, -1.1109e+00],
[-4.3048e+00, -1.3264e+00, -1.6825e+00, ..., -4.0568e-02,
-1.4469e+00, -3.4767e-01],
[-4.3913e+00, -3.5711e+00, -3.0707e+00, ..., -1.6286e+00,
-1.6020e+00, -6.9432e-01],
...,
[-3.1050e+00, -1.9253e+00, -2.8697e+00, ..., -2.6351e-01,
3.6846e-01, 5.1540e-01],
[-3.7206e+00, -1.6008e+00, -2.6333e+00, ..., -1.6595e-01,
-1.4214e-01, 2.6498e-01],
[-1.3388e+00, -9.9868e-01, -1.9112e+00, ..., -3.9707e-01,
-6.6702e-01, -5.2640e-01]],
[[ 2.0350e+00, 2.9868e+00, 2.5817e+00, ..., 2.7892e+00,
1.3940e+00, 2.9237e+00],
[ 3.3682e-01, 1.4120e+00, 1.9156e+00, ..., 2.0875e+00,
-1.6859e-01, 2.4774e+00],
[ 4.7210e-01, 9.0312e-01, 1.4400e+00, ..., 2.4264e+00,
1.2236e+00, 1.9072e+00],
...,
[ 8.6662e-01, 1.2865e+00, 1.0008e+00, ..., 5.0010e-01,
8.0865e-02, 1.3562e+00],
[ 5.9541e-01, 1.7845e+00, 1.0040e+00, ..., 1.0792e+00,
5.9463e-01, 1.4564e+00],
[ 1.4516e+00, 1.5518e+00, 1.5938e+00, ..., 1.4471e+00,
1.0726e+00, 1.9481e+00]],
...,
[[-1.7005e+00, -4.8044e+00, -4.1200e+00, ..., -3.3903e+00,
-2.4334e+00, -4.0089e+00],
[-2.3098e+00, -4.4039e+00, -4.0040e+00, ..., -5.9494e+00,
-4.0428e+00, -4.4875e+00],
[-1.5226e-02, -2.2778e+00, -2.2301e+00, ..., -2.2465e+00,
-1.9724e+00, -1.7311e+00],
...,
[-1.5866e+00, -4.0973e+00, -4.8571e+00, ..., -3.6600e+00,
-2.4056e+00, -1.3748e+00],
[-1.3907e+00, -4.4143e+00, -5.7938e+00, ..., -6.0908e+00,
-3.6442e+00, -3.5695e+00],
[-1.3970e+00, -4.6118e+00, -4.2807e+00, ..., -3.6564e+00,
-2.2618e+00, -2.6887e+00]],
[[ 7.1631e-01, 6.4475e-01, -9.7673e-02, ..., 9.1277e-01,
1.1006e+00, 1.2080e+00],
[ 9.8439e-01, 1.0554e+00, 6.2585e-01, ..., 1.7721e+00,
2.1955e+00, 2.5536e+00],
[-4.0162e-01, -1.2782e+00, 4.2251e-01, ..., 1.9510e+00,
2.3360e+00, 2.1441e+00],
...,
[ 7.9589e-01, 8.0459e-03, -5.4505e-01, ..., 1.2232e+00,
1.9817e+00, 1.5468e+00],
[ 1.4559e+00, 1.3069e+00, 1.0411e+00, ..., 1.3222e+00,
1.4685e+00, 2.3897e+00],
[ 6.9032e-01, 8.8066e-01, 9.8639e-01, ..., 1.0753e+00,
1.2998e+00, 1.3581e+00]],
[[-3.1861e-01, -1.8833e-01, -9.6740e-01, ..., -2.4960e-01,
-1.5096e-01, 8.6844e-01],
[-1.0479e+00, -1.4675e+00, -2.1075e+00, ..., -2.6272e-01,
-3.7343e-01, 3.2847e-01],
[-1.6533e+00, -1.3948e+00, -9.5033e-01, ..., -4.5432e-01,
-1.9783e-02, 3.0934e-01],
...,
[-1.7688e+00, -1.4623e+00, -1.7363e+00, ..., -5.8348e-01,
-2.3009e-01, -3.6809e-01],
[-1.5034e+00, -1.5326e+00, -1.7954e+00, ..., -1.1317e-01,
-2.6501e-01, 2.9674e-01],
[-2.2614e-01, -1.7845e-02, -5.7953e-01, ..., 2.1907e-01,
1.4613e-01, 3.3942e-01]]],
[[[ 9.5812e-01, 1.5294e+00, 1.0513e+00, ..., 4.1777e-01,
3.1124e-01, -1.7393e-01],
[-9.5278e-01, -2.0567e+00, -9.9228e-01, ..., 1.6664e-01,
6.3596e-01, -5.6975e-01],
[-3.0513e-02, -8.0120e-01, -5.2764e-02, ..., -8.4927e-01,
-1.9562e-01, 3.6295e-01],
...,
[ 4.3644e-01, -2.6957e-01, -8.6705e-01, ..., 1.6391e+00,
-8.8867e-01, -2.1174e-01],
[-2.1114e-01, -1.1031e+00, -6.9602e-01, ..., 1.2013e+00,
-5.9041e-01, 2.2767e-02],
[ 7.4511e-01, -3.3155e-01, -4.6972e-01, ..., 7.4463e-01,
-5.6125e-01, 6.1828e-01]],
[[-4.3259e+00, -1.3697e+00, -2.0583e+00, ..., -2.8502e+00,
-3.3921e+00, -7.2289e-01],
[-4.2343e+00, -2.8446e+00, -2.0703e+00, ..., -2.5230e+00,
-2.9908e+00, -4.0940e-01],
[-2.6783e+00, -1.6258e+00, -2.4029e+00, ..., -4.0885e+00,
-3.0126e+00, -6.6427e-01],
...,
[-8.3131e-01, -5.5896e-01, -2.4781e+00, ..., -3.3135e-01,
8.8161e-01, -3.7266e-01],
[-2.9129e+00, -4.6810e-01, -1.7498e+00, ..., 5.8163e-01,
5.2827e-01, 1.0040e-02],
[-2.0729e+00, -1.5337e+00, -2.1757e+00, ..., 3.8098e-01,
1.1403e+00, 5.1387e-01]],
[[ 3.0747e+00, 4.0888e+00, 3.1835e+00, ..., 2.2849e+00,
7.4498e-01, 2.2751e+00],
[ 3.7478e-01, 7.5983e-01, 2.2575e+00, ..., 1.8048e+00,
-4.3536e-02, 2.1114e+00],
[ 8.5574e-01, 1.2440e+00, 1.6231e+00, ..., 5.9866e-01,
1.6187e-01, 1.6166e+00],
...,
[ 3.3562e-01, 9.6944e-01, 1.9988e+00, ..., -8.2395e+00,
-4.2734e+00, -9.8872e-01],
[ 3.4886e-01, 2.1414e+00, 2.0097e+00, ..., -5.6557e+00,
-2.6368e+00, -7.3050e-01],
[ 1.4139e+00, 1.3093e+00, 1.9797e+00, ..., -1.5698e+00,
-4.8669e-01, 4.0451e-01]],
...,
[[-2.1442e+00, -7.5544e+00, -6.0409e+00, ..., -3.9428e+00,
-2.9498e+00, -3.5190e+00],
[-2.7957e+00, -6.5208e+00, -4.6660e+00, ..., -5.7432e+00,
-4.7222e+00, -4.1977e+00],
[-2.1909e-01, -2.7429e+00, -2.9080e+00, ..., -5.2864e+00,
-3.0480e+00, -1.3130e+00],
...,
[-9.6102e-01, -2.2608e+00, -1.9619e+00, ..., -5.5272e+00,
-7.9609e+00, -8.4548e+00],
[-1.3362e+00, -3.7215e+00, -3.7116e+00, ..., -5.7279e+00,
-7.0394e+00, -5.3530e+00],
[-2.3068e+00, -5.0880e+00, -3.3186e+00, ..., -4.2047e+00,
-4.0830e+00, -2.8353e+00]],
[[-7.7802e-01, -1.3206e-01, 8.2078e-02, ..., 1.0342e+00,
1.0473e+00, 1.7855e+00],
[ 1.7727e+00, 1.1016e+00, 1.5312e+00, ..., 1.5097e+00,
1.6651e+00, 2.7930e+00],
[ 1.2524e+00, 2.7787e-01, 6.5100e-01, ..., 6.6656e-01,
7.4026e-01, 1.8453e+00],
...,
[ 1.2848e+00, -1.8520e-01, -3.4412e-02, ..., 1.6584e+00,
-3.8257e-01, -1.1763e+00],
[ 1.8679e+00, 9.3930e-01, 1.3049e+00, ..., 2.4816e+00,
-1.1064e+00, -5.2012e-01],
[ 1.0140e+00, 3.9774e-01, 5.0631e-01, ..., 2.0375e+00,
5.1160e-01, 1.0611e+00]],
[[-5.4037e-01, 5.6756e-01, 5.3019e-02, ..., -3.0484e-01,
-4.8778e-01, 9.1730e-01],
[-1.1874e+00, -1.1967e+00, -1.3440e+00, ..., -2.9049e-01,
2.5158e-01, 1.0155e+00],
[-2.1184e+00, -5.7826e-01, -1.1700e-01, ..., -1.1857e-01,
4.3022e-01, 1.5602e-01],
...,
[-1.7849e+00, -1.4637e+00, -1.6744e+00, ..., 2.2582e+00,
3.3772e-01, -5.3235e+00],
[-1.7696e+00, -1.1001e+00, -1.2785e+00, ..., 9.0593e-01,
-1.9659e+00, -4.3157e+00],
[-3.6354e-01, 3.5468e-01, -4.7742e-01, ..., -2.0989e-01,
-7.4623e-01, -8.9301e-01]]]], device='cuda:0'),
tensor([[[[ 9.6895e-01, 1.6089e+00, 1.2494e+00, ..., 4.9571e-01,
2.9295e-01, -1.6841e-01],
[-8.9153e-01, -1.9764e+00, -1.1857e+00, ..., 2.0635e-01,
6.2803e-01, -6.6061e-01],
[ 4.2713e-02, -1.0474e+00, -9.2643e-01, ..., -5.5110e-01,
-9.7758e-02, 6.0635e-01],
...,
[ 4.2299e-01, -3.8762e-01, -6.7078e-01, ..., -2.7966e-01,
1.3428e+00, 1.1001e+00],
[-1.5886e-01, -1.0136e+00, -3.0737e-01, ..., 3.7922e-02,
6.6418e-01, -2.7642e-02],
[ 7.7555e-01, -2.3494e-01, -3.0919e-01, ..., -3.8851e-01,
-3.3583e-01, 3.2737e-01]],
[[-4.1617e+00, -1.3280e+00, -2.8143e+00, ..., -2.7974e+00,
-3.3538e+00, -7.5530e-01],
[-3.9833e+00, -3.2217e+00, -4.1361e+00, ..., -2.1526e+00,
-3.0990e+00, -5.8796e-01],
[-2.6096e+00, -2.6544e+00, -4.4508e+00, ..., -4.3567e+00,
-3.4386e+00, -1.9376e-01],
...,
[-8.7201e-01, -5.6276e-01, -2.0690e+00, ..., -2.6505e+00,
-1.5194e+00, 1.9818e-01],
[-2.8115e+00, -1.8926e-01, -1.0282e+00, ..., -1.6136e+00,
-1.5155e+00, 1.2312e-01],
[-1.9912e+00, -1.3821e+00, -1.7690e+00, ..., -7.6862e-01,
-7.8893e-01, 1.1157e-01]],
[[ 3.0095e+00, 4.1773e+00, 3.3127e+00, ..., 2.2519e+00,
7.5456e-01, 2.3066e+00],
[ 3.3124e-01, 9.6146e-01, 2.8051e+00, ..., 1.7614e+00,
5.4623e-02, 2.1660e+00],
[ 9.0714e-01, 1.9113e+00, 2.2557e+00, ..., 4.3582e-01,
6.3139e-01, 2.3679e+00],
...,
[ 2.8407e-01, 9.3014e-01, 1.3645e+00, ..., 8.4514e-01,
3.8721e-01, 1.6404e+00],
[ 3.2536e-01, 2.0813e+00, 1.5499e+00, ..., -8.4710e-03,
9.1271e-02, 1.1180e+00],
[ 1.3993e+00, 1.3476e+00, 1.7003e+00, ..., 1.1044e+00,
8.2300e-01, 1.3460e+00]],
...,
[[-2.1062e+00, -7.4058e+00, -6.1166e+00, ..., -3.9984e+00,
-2.9636e+00, -3.5990e+00],
[-2.8104e+00, -6.6412e+00, -5.1491e+00, ..., -5.5441e+00,
-4.6145e+00, -4.3053e+00],
[-2.7066e-01, -3.0270e+00, -3.6213e+00, ..., -5.2456e+00,
-4.1084e+00, -2.1310e+00],
...,
[-9.6321e-01, -2.2043e+00, -1.8420e+00, ..., -5.2733e+00,
-4.9679e+00, -3.2822e+00],
[-1.3069e+00, -3.7049e+00, -3.3803e+00, ..., -6.5217e+00,
-4.5824e+00, -4.6361e+00],
[-2.2412e+00, -5.0699e+00, -3.0853e+00, ..., -4.5476e+00,
-2.7774e+00, -3.4705e+00]],
[[-7.2092e-01, -1.5832e-01, -3.8666e-01, ..., 1.0741e+00,
1.0676e+00, 1.8099e+00],
[ 1.8218e+00, 7.1499e-01, 9.8597e-01, ..., 1.5356e+00,
1.4816e+00, 2.8061e+00],
[ 1.1937e+00, -3.4192e-01, -1.0003e+00, ..., 1.2072e+00,
1.0284e+00, 2.3083e+00],
...,
[ 1.2997e+00, -4.9670e-02, 5.6790e-01, ..., 1.9655e+00,
1.2009e+00, 1.7899e+00],
[ 1.8777e+00, 1.0130e+00, 1.6551e+00, ..., 1.4849e+00,
9.4222e-01, 2.2611e+00],
[ 1.0066e+00, 3.3464e-01, 7.7315e-01, ..., 9.9351e-01,
1.2141e+00, 1.5568e+00]],
[[-5.1103e-01, 5.5892e-01, -1.3274e-01, ..., -2.5797e-01,
-5.5088e-01, 9.1869e-01],
[-1.0701e+00, -1.5375e+00, -2.3498e+00, ..., -1.9112e-01,
-1.3358e-02, 1.0389e+00],
[-2.0528e+00, -1.0678e+00, -9.5989e-01, ..., 2.4835e-01,
6.5751e-01, 2.6793e-01],
...,
[-1.7614e+00, -1.4947e+00, -1.3501e+00, ..., -5.1045e-01,
-1.3731e+00, -8.5946e-02],
[-1.7591e+00, -1.0628e+00, -1.0327e+00, ..., -8.0743e-01,
-9.4191e-01, 3.2155e-01],
[-3.3701e-01, 3.4684e-01, -4.3493e-01, ..., 1.8586e-01,
-1.1806e-01, 6.6575e-01]]],
[[[ 6.0049e-01, 8.8121e-01, 5.0812e-01, ..., 5.5264e-01,
5.9336e-01, 6.9892e-02],
[-2.6567e-01, -6.8744e-01, -4.7634e-01, ..., 7.7126e-01,
1.9400e-01, -3.6185e-01],
[-1.6043e-01, 1.7292e-01, 1.9380e-01, ..., -5.6784e-01,
6.2254e-02, 6.4353e-01],
...,
[ 3.9517e-01, -4.1476e-01, 4.7398e-01, ..., 6.9133e-01,
8.9377e-01, 6.6524e-01],
[-5.4525e-02, -7.5307e-01, -1.9726e-02, ..., 7.2207e-01,
1.4428e+00, 8.3950e-02],
[ 7.9555e-01, -1.9102e-01, -4.1950e-02, ..., 2.8186e-01,
3.0096e-01, 7.2533e-01]],
[[-2.2482e+00, -6.6266e-01, -1.8390e+00, ..., -1.7284e+00,
-2.5514e+00, -1.5680e+00],
[-2.3885e+00, -6.7160e-01, -1.4967e+00, ..., -2.2190e+00,
-2.5586e+00, -1.7554e+00],
[-2.3011e+00, -2.1925e+00, -3.1123e+00, ..., -2.2913e+00,
-1.8417e+00, -1.0962e+00],
...,
[-6.7457e-01, -8.5883e-01, -8.7254e-01, ..., -2.3479e+00,
4.3399e-01, 7.0256e-01],
[-2.5557e+00, -3.7177e-01, -5.1328e-01, ..., -8.4507e-02,
2.7516e-01, 6.5259e-01],
[-1.5583e+00, -1.2085e+00, -9.0588e-01, ..., -4.9696e-01,
-5.6801e-01, -1.5802e-01]],
[[ 1.6971e+00, 2.3408e+00, 1.6163e+00, ..., 1.3065e+00,
5.5222e-01, 2.1335e+00],
[-2.7873e-01, 4.4305e-01, 7.0701e-01, ..., 6.0630e-01,
1.9368e-01, 2.3669e+00],
[ 1.5294e-01, 3.5007e-01, 2.0790e-01, ..., 2.4558e-03,
8.4924e-01, 1.9789e+00],
...,
[ 7.4460e-01, 5.8371e-01, 1.6843e+00, ..., 2.9909e-01,
1.0727e-01, 1.5994e+00],
[ 1.1158e+00, 2.2592e+00, 2.0797e+00, ..., 1.3190e+00,
5.0900e-01, 1.6667e+00],
[ 2.4140e+00, 1.8479e+00, 2.4126e+00, ..., 1.6652e+00,
1.0369e+00, 1.7036e+00]],
...,
[[-1.7156e+00, -4.6055e+00, -2.9090e+00, ..., -1.9246e+00,
-1.4880e+00, -3.2372e+00],
[-2.1066e+00, -4.9351e+00, -2.7455e+00, ..., -3.6666e+00,
-2.0020e+00, -2.9887e+00],
[-1.3916e+00, -5.1824e+00, -3.7673e+00, ..., -2.9692e+00,
-2.5496e+00, -2.1076e+00],
...,
[-2.5774e+00, -4.5651e+00, -2.1469e+00, ..., -4.4666e+00,
-3.6747e+00, -2.7055e+00],
[-2.4962e+00, -5.5462e+00, -3.8848e+00, ..., -5.4953e+00,
-4.7721e+00, -4.9013e+00],
[-2.5995e+00, -5.9000e+00, -2.9290e+00, ..., -4.1394e+00,
-1.8551e+00, -3.3492e+00]],
[[ 8.8645e-01, 1.4934e+00, 8.3407e-01, ..., 8.2889e-01,
1.4081e+00, 1.3900e+00],
[ 1.0705e+00, 9.1863e-01, 8.8247e-01, ..., 2.7872e-01,
7.6406e-01, 1.9951e+00],
[ 6.5399e-01, -3.7890e-01, 3.8311e-01, ..., 1.1638e-01,
3.5766e-01, 1.4360e+00],
...,
[ 1.5113e+00, -2.5535e-02, 6.3755e-01, ..., 9.2635e-01,
2.1281e+00, 1.4206e+00],
[ 2.0222e+00, 1.1390e+00, 1.8821e+00, ..., 2.1681e+00,
1.6856e+00, 2.5383e+00],
[ 1.1779e+00, 9.1962e-01, 1.3422e+00, ..., 1.3351e+00,
1.4479e+00, 1.5297e+00]],
[[-8.8063e-02, -6.0581e-01, -1.1749e+00, ..., -4.5345e-01,
-8.4334e-01, -1.1309e-01],
[-8.8271e-01, -2.3683e+00, -2.4508e+00, ..., -1.7645e+00,
-1.3077e+00, -7.6589e-01],
[-1.3697e+00, -1.9940e+00, -7.0829e-01, ..., -1.2096e+00,
-4.5639e-01, -1.9047e-01],
...,
[-1.6930e+00, -1.6259e+00, 1.3051e-02, ..., -2.1500e+00,
-2.1243e-01, -4.5319e-01],
[-1.7530e+00, -1.3381e+00, -5.9888e-01, ..., 9.9333e-02,
-6.3992e-02, 3.9837e-01],
[-2.7472e-01, 4.3476e-01, -1.5354e-01, ..., 5.1906e-01,
1.8188e-02, 6.5873e-01]]],
[[[ 7.7671e-01, -6.4473e-01, -1.7802e+00, ..., 5.9375e-01,
5.9975e-01, 4.6235e-01],
[-1.5613e-01, 1.3119e+00, 2.1737e+00, ..., 3.0568e-02,
4.1336e-02, -1.5749e+00],
[-7.5929e-01, -1.1841e+00, -6.3586e-01, ..., -8.6455e-01,
7.7461e-01, 6.7299e-01],
...,
[-8.6037e-01, -3.4224e+00, -1.3694e+00, ..., 3.9040e-02,
5.1164e-01, 9.2725e-01],
[-1.3039e+00, -7.8007e-01, 1.2002e+00, ..., -2.5814e-03,
2.2365e+00, 7.0066e-02],
[-2.8702e-02, 1.6883e-01, -9.5452e-01, ..., -2.0141e+00,
-2.8108e+00, -1.0136e+00]],
[[-7.8892e+00, -5.2878e+00, -4.4811e+00, ..., -1.3888e+00,
-1.7202e+00, -9.7748e-01],
[-9.5814e-01, 2.7065e-01, -1.3882e+00, ..., -6.6795e-01,
-1.7223e+00, -4.7802e-01],
[ 1.3444e+00, -1.9388e-01, -3.9468e+00, ..., -9.7279e-01,
-1.7506e+00, -1.0611e+00],
...,
[-4.4147e+00, -6.6427e+00, -9.5682e+00, ..., -3.6966e+00,
-3.4163e+00, -2.1070e+00],
[-5.5375e+00, -7.1245e+00, -6.0010e+00, ..., -3.0655e+00,
-3.9826e+00, -3.4220e+00],
[-4.4738e+00, -5.3310e+00, -2.5850e+00, ..., -5.5056e+00,
-4.8836e+00, -2.5708e+00]],
[[-2.4572e-01, 2.4432e+00, 4.7897e+00, ..., 2.9511e+00,
1.3899e+00, 3.1015e+00],
[-3.4504e+00, -2.4307e+00, -2.0423e+00, ..., 2.0904e+00,
-3.6401e-02, 2.6169e+00],
[-2.9522e+00, -4.7970e+00, -5.0794e+00, ..., 2.1883e+00,
3.3066e-01, 1.9648e+00],
...,
[-3.7346e+00, -8.2079e+00, -6.6826e+00, ..., -2.5774e+00,
-3.6179e+00, -6.3040e-01],
[-5.1207e+00, -6.2459e+00, -4.8450e+00, ..., -6.7803e+00,
-6.0786e+00, -2.0171e+00],
[-7.3938e-01, -3.3641e+00, -9.4728e-01, ..., -2.0223e+00,
-1.1142e+00, 1.2601e+00]],
...,
[[-3.6061e-01, -3.4639e+00, -4.5507e+00, ..., -3.5722e+00,
-2.4980e+00, -4.3157e+00],
[-1.9549e+00, -9.1672e+00, -1.4280e+01, ..., -5.1978e+00,
-3.3342e+00, -4.0943e+00],
[-2.2790e+00, -6.4691e+00, -9.2711e+00, ..., -3.9206e+00,
-3.1155e+00, -1.8994e+00],
...,
[ 6.3856e-01, -1.4734e+00, -9.1536e+00, ..., -5.6404e+00,
-6.4092e+00, -7.4541e+00],
[-2.1970e+00, -8.0348e+00, -1.1505e+01, ..., -1.0431e+01,
-1.2223e+01, -1.1504e+01],
[-2.3926e+00, -6.3335e+00, -4.5208e+00, ..., -5.9045e+00,
-4.5367e+00, -6.1116e+00]],
[[-5.3409e+00, -4.2781e+00, -8.1312e-01, ..., 1.3456e+00,
1.2656e+00, 1.2180e+00],
[ 6.5320e-02, -1.1842e+00, 1.8252e-01, ..., 1.9983e+00,
2.3991e+00, 2.6527e+00],
[ 8.7551e-01, -1.9690e+00, -1.2600e+00, ..., 1.8601e+00,
1.5322e+00, 2.1007e+00],
...,
[-1.5352e+00, -9.7084e-01, -2.5674e+00, ..., -2.9992e+00,
1.5436e-01, 1.6640e+00],
[ 5.3254e-01, 1.1162e-01, -2.0612e+00, ..., -7.4493e-01,
-3.9064e-01, 1.3066e+00],
[-9.9652e-01, -1.6617e+00, 4.4697e-01, ..., -2.2881e+00,
-3.9548e-01, 1.3154e+00]],
[[-1.2689e+00, -1.0203e+00, -3.2970e+00, ..., 6.7800e-02,
3.2960e-02, 9.1394e-01],
[-1.3712e+00, 3.4383e+00, 2.0102e+00, ..., -1.3262e-01,
-3.8123e-01, 1.1864e-01],
[-2.1907e+00, 1.1147e+00, -9.9699e-01, ..., -2.7343e-01,
5.7964e-02, 3.7144e-01],
...,
[-1.4486e+00, 5.3975e-01, 2.6225e-01, ..., -1.7765e+00,
-1.9591e-01, 1.2837e+00],
[ 4.4808e-01, -3.6197e-01, -5.6522e-01, ..., -2.7759e+00,
-2.1387e+00, -1.1135e+00],
[-1.0661e+00, -3.0635e+00, -1.3438e+00, ..., -3.0114e+00,
-1.9419e+00, 6.4140e-01]]],
[[[ 5.9648e-01, 8.8134e-01, 5.9616e-01, ..., 6.1978e-01,
6.5447e-01, 1.0073e-01],
[-3.1673e-01, -7.1818e-01, -4.3044e-01, ..., 6.1745e-01,
1.7860e-01, -3.8652e-01],
[-6.8399e-02, 1.2991e-01, 9.8642e-02, ..., -1.0758e-01,
-4.1105e-02, 7.1643e-01],
...,
[-2.2809e-02, -1.0541e+00, -1.1618e+00, ..., 1.4665e-01,
5.7053e-01, 8.9118e-01],
[-3.7485e-01, -2.0912e+00, -1.2802e+00, ..., -7.5221e-02,
6.4739e-01, 1.1132e-01],
[ 4.8314e-01, -9.1758e-01, -8.3671e-01, ..., -2.6249e-01,
-5.8813e-01, 1.9915e-01]],
[[-2.2788e+00, -7.0267e-01, -9.4044e-01, ..., -1.2175e+00,
-2.4923e+00, -1.7000e+00],
[-2.4723e+00, -7.2824e-01, -1.1589e+00, ..., -9.6886e-01,
-2.7288e+00, -1.9040e+00],
[-2.5191e+00, -2.4029e+00, -2.4237e+00, ..., -2.0103e+00,
-2.1788e+00, -1.2018e+00],
...,
[-1.0515e+00, -8.6729e-01, -2.7118e+00, ..., -2.0691e+00,
-1.2538e+00, 7.7439e-02],
[-2.1761e+00, -5.5084e-02, -1.4697e+00, ..., -1.4858e+00,
-1.5457e+00, -2.5225e-01],
[-9.0191e-01, -1.0506e+00, -9.2948e-01, ..., -2.1237e+00,
-1.6781e+00, -6.1960e-01]],
[[ 1.6866e+00, 2.3230e+00, 1.5903e+00, ..., 1.5509e+00,
6.1651e-01, 2.0984e+00],
[-2.6798e-01, 5.2785e-01, 1.0457e+00, ..., 6.4809e-01,
1.9557e-01, 2.3400e+00],
[ 6.8711e-02, 4.3403e-01, 9.9618e-01, ..., -3.7093e-01,
7.3752e-01, 2.0253e+00],
...,
[ 1.0586e+00, 1.2951e+00, 1.9267e+00, ..., 7.2197e-01,
-4.4047e-01, 1.3722e+00],
[ 7.7582e-01, 2.3690e+00, 1.9767e+00, ..., -2.7513e-02,
-3.4239e-01, 1.0125e+00],
[ 1.7443e+00, 1.3815e+00, 4.2842e-01, ..., 1.1775e+00,
5.2890e-01, 1.3110e+00]],
...,
[[-1.6830e+00, -4.5135e+00, -3.1395e+00, ..., -8.2183e-01,
-1.6192e+00, -3.3042e+00],
[-2.1095e+00, -4.7373e+00, -2.6331e+00, ..., -2.2972e+00,
-2.7277e+00, -3.1231e+00],
[-1.3516e+00, -5.0937e+00, -3.5295e+00, ..., -2.0784e+00,
-2.0476e+00, -1.9001e+00],
...,
[-2.0902e+00, -5.2969e+00, -3.3291e+00, ..., -3.8236e+00,
-3.3283e+00, -2.9242e+00],
[-2.8824e+00, -7.1509e+00, -4.3262e+00, ..., -3.6564e+00,
-3.4622e+00, -4.3567e+00],
[-2.9993e+00, -7.2562e+00, -3.6535e+00, ..., -2.5153e+00,
-2.0033e+00, -2.7891e+00]],
[[ 8.7839e-01, 1.4747e+00, 1.1432e+00, ..., 6.5169e-01,
1.2870e+00, 1.4507e+00],
[ 1.1023e+00, 1.0075e+00, 6.5912e-01, ..., 9.6052e-01,
8.5771e-01, 2.0583e+00],
[ 5.9152e-01, -3.6571e-02, 2.6675e-01, ..., 3.1816e-01,
2.4617e-01, 1.3155e+00],
...,
[ 1.6532e+00, -2.1826e-01, -5.5159e-01, ..., 1.7127e+00,
1.5716e+00, 1.8294e+00],
[ 2.2542e+00, 1.5076e+00, 1.2201e+00, ..., 1.0512e+00,
1.3563e+00, 2.2548e+00],
[ 1.0318e+00, 6.1650e-01, 1.0414e+00, ..., 7.4278e-02,
7.2055e-01, 1.1610e+00]],
[[-1.1507e-01, -6.0616e-01, -1.0853e+00, ..., -3.1423e-01,
-1.2601e+00, -3.0023e-02],
[-9.2132e-01, -2.3077e+00, -2.1009e+00, ..., -7.6452e-01,
-1.4397e+00, -7.7224e-01],
[-1.3574e+00, -1.5876e+00, -1.4656e-01, ..., -7.4292e-01,
-4.8345e-01, -2.9070e-01],
...,
[-2.6086e+00, -2.1299e+00, -1.0334e+00, ..., -1.4498e-02,
-2.9803e-01, 1.5490e-01],
[-2.5119e+00, -2.3582e+00, -1.2853e+00, ..., -6.6899e-01,
-4.9135e-01, 3.7435e-01],
[-4.1473e-01, -2.4150e-01, -5.4692e-01, ..., -4.2202e-01,
-6.3224e-01, 1.0178e-01]]],
[[[ 6.8260e-01, 1.0174e+00, 6.6856e-01, ..., 9.7056e-01,
9.3013e-01, 3.4105e-01],
[-3.3749e-01, -1.4842e+00, -6.7884e-01, ..., 1.0070e-01,
4.7703e-01, -4.5078e-01],
[ 1.0981e-01, -8.4900e-01, -4.4446e-01, ..., -1.4144e-01,
2.0007e-01, 6.5391e-01],
...,
[-4.8782e-03, 3.5972e-01, -1.6247e-01, ..., 3.2006e-01,
8.9827e-01, 1.1359e+00],
[-2.9971e-01, -3.5720e-01, 3.4742e-01, ..., 4.5981e-01,
8.6891e-01, 1.4071e-01],
[ 2.6560e-01, -7.2278e-01, -2.5695e-01, ..., 4.4035e-02,
-3.3462e-01, 5.0343e-01]],
[[-3.1387e+00, -1.4788e+00, -2.8862e+00, ..., -1.5261e+00,
-2.9191e+00, -1.0484e+00],
[-4.4458e+00, -1.7821e+00, -2.1280e+00, ..., -1.2752e+00,
-2.6426e+00, -3.3891e-01],
[-4.4482e+00, -2.5776e+00, -2.8606e+00, ..., -1.7049e+00,
-2.7425e+00, -5.5864e-01],
...,
[-1.9665e+00, 1.2796e+00, 2.7870e+00, ..., -8.8263e-01,
-9.7801e-01, 5.3019e-01],
[-3.2559e+00, -7.3962e-01, -2.3616e+00, ..., -2.2748e-01,
-8.5219e-01, 2.9930e-01],
[-1.2080e+00, -1.1005e+00, -2.0471e+00, ..., -5.4565e-01,
-3.6012e-01, 6.6998e-02]],
[[ 1.7590e+00, 2.7921e+00, 2.3205e+00, ..., 2.2889e+00,
7.9589e-01, 1.9578e+00],
[ 1.9190e-01, 1.1618e+00, 1.3978e+00, ..., 1.1810e+00,
-1.6053e-01, 1.9360e+00],
[ 7.4855e-01, 1.4475e+00, 7.5216e-01, ..., 1.2065e+00,
3.8670e-01, 2.1720e+00],
...,
[ 1.7513e+00, 4.2698e-01, -2.3230e+00, ..., 1.6625e-01,
4.7629e-01, 1.3248e+00],
[ 1.3609e+00, 2.6704e+00, 1.0094e+00, ..., 7.7302e-02,
3.6562e-01, 1.1325e+00],
[ 1.6123e+00, 2.2319e+00, 2.3841e+00, ..., 7.8191e-01,
6.0106e-01, 1.3015e+00]],
...,
[[-1.4979e+00, -4.7182e+00, -3.9177e+00, ..., -4.5959e+00,
-3.3124e+00, -4.0320e+00],
[-1.8859e+00, -4.0360e+00, -4.2879e+00, ..., -5.4394e+00,
-3.6352e+00, -4.2920e+00],
[-9.9652e-01, -3.3254e+00, -3.7289e+00, ..., -5.2410e+00,
-3.7504e+00, -2.7627e+00],
...,
[-1.6578e+00, -3.4655e+00, -2.3836e+00, ..., -4.2997e+00,
-2.5790e+00, -3.4765e+00],
[-1.4271e+00, -4.4578e+00, -4.3865e+00, ..., -6.1255e+00,
-3.6393e+00, -4.8711e+00],
[-1.5975e+00, -4.3490e+00, -3.6778e+00, ..., -4.5288e+00,
-2.0586e+00, -3.2285e+00]],
[[ 7.5125e-01, 5.4030e-01, -9.6843e-02, ..., 1.3991e+00,
8.9460e-01, 1.6245e+00],
[ 1.1739e+00, 9.9436e-01, 5.0428e-01, ..., 2.5779e+00,
1.8520e+00, 2.6580e+00],
[ 5.8401e-01, 5.8133e-01, 6.3206e-03, ..., 2.4340e+00,
9.3762e-01, 1.9063e+00],
...,
[ 1.1156e+00, 1.0259e+00, 1.5990e+00, ..., 3.0627e+00,
1.6630e+00, 1.7024e+00],
[ 1.4864e+00, 7.5510e-01, 6.2988e-02, ..., 2.7645e+00,
1.3287e+00, 2.4500e+00],
[ 7.2521e-01, 8.7090e-01, 1.2576e+00, ..., 1.3401e+00,
1.1994e+00, 1.3740e+00]],
[[-2.9410e-01, -2.1942e-01, -9.6440e-01, ..., 6.0636e-01,
2.5022e-01, 1.0909e+00],
[-1.3961e+00, -1.4742e+00, -2.1085e+00, ..., -2.6360e-01,
-1.9335e-01, 6.3372e-01],
[-1.6731e+00, -8.7438e-01, -1.1335e+00, ..., -8.1990e-01,
-3.7865e-01, -2.7694e-01],
...,
[-1.4595e+00, -1.2711e+00, -3.6554e-01, ..., -1.3804e+00,
-9.5511e-01, -8.1539e-01],
[-1.1920e+00, -6.4688e-01, -7.3380e-01, ..., -1.0646e+00,
-8.7650e-01, 5.4546e-02],
[-4.5360e-02, -3.9937e-02, -8.8336e-01, ..., -1.0278e+00,
-6.8411e-01, 2.8121e-01]]]], device='cuda:0'),
tensor([[[[ 8.5119e-01, 1.4945e+00, 1.1783e+00, ..., 1.4113e+00,
1.0997e+00, 9.7748e-01],
[-1.0042e+00, -1.9956e+00, -1.0843e+00, ..., -3.9192e-01,
-5.5792e-01, -1.5655e+00],
[-1.0241e-01, -7.2647e-01, 3.0990e-01, ..., -9.8011e-02,
6.8935e-01, 7.8768e-01],
...,
[ 5.4535e-01, -2.3079e-01, 1.2270e-01, ..., 3.7110e-01,
1.2128e-01, 6.3544e-01],
[ 3.6769e-01, -1.0309e+00, -1.5411e-01, ..., 7.1195e-01,
6.7208e-01, -7.5415e-02],
[ 6.2157e-01, 3.5007e-01, 1.6530e-01, ..., 2.5151e-01,
-5.9912e-02, 6.5445e-01]],
[[-3.7090e+00, -7.1060e-01, -1.5924e+00, ..., -1.5472e+00,
-2.4793e+00, -1.2707e+00],
[-3.9484e+00, -2.8350e+00, -1.9548e+00, ..., -2.3070e+00,
-3.0903e+00, -9.4608e-01],
[-2.4594e+00, -1.7500e+00, -2.6125e+00, ..., -2.0491e+00,
-2.8160e+00, -1.0628e+00],
...,
[-9.0106e-01, -1.2244e+00, -1.6826e+00, ..., -1.7943e+00,
-1.1289e+00, -9.9606e-02],
[-1.2975e+00, -3.9702e-01, -3.8323e-01, ..., -4.9043e-01,
-1.1531e+00, -4.0922e-02],
[ 5.4929e-01, -8.6447e-02, 1.7318e-01, ..., 3.6209e-01,
1.1670e-01, 4.1574e-01]],
[[ 2.8956e+00, 4.2363e+00, 3.3675e+00, ..., 2.9630e+00,
1.1003e+00, 3.0918e+00],
[ 2.3636e-01, 1.1044e+00, 2.3071e+00, ..., 1.2124e+00,
-7.6221e-01, 2.8170e+00],
[ 1.0456e+00, 1.6485e+00, 1.7089e+00, ..., 5.9368e-01,
-3.4668e-01, 2.0017e+00],
...,
[ 4.8155e-01, 7.0993e-01, 6.8946e-01, ..., -2.5859e-02,
1.1804e-01, 1.5047e+00],
[ 1.9535e-01, 1.3806e+00, 6.1189e-01, ..., 5.1591e-02,
1.9243e-01, 1.1282e+00],
[ 1.4071e+00, 1.2079e+00, 4.4588e-01, ..., 5.0699e-01,
4.6807e-01, 1.1697e+00]],
...,
[[-1.8884e+00, -7.2912e+00, -5.3938e+00, ..., -4.0603e+00,
-3.2800e+00, -5.6104e+00],
[-2.4360e+00, -7.0636e+00, -4.2583e+00, ..., -3.6291e+00,
-2.7535e+00, -4.3965e+00],
[-1.1160e+00, -4.8265e+00, -4.0446e+00, ..., -3.3651e+00,
-3.1984e+00, -2.3767e+00],
...,
[-2.2189e+00, -5.6180e+00, -3.7050e+00, ..., -5.0925e+00,
-4.5789e+00, -3.4705e+00],
[-2.4032e+00, -6.5926e+00, -4.5291e+00, ..., -5.0586e+00,
-4.2220e+00, -4.8230e+00],
[-2.3734e+00, -6.5274e+00, -4.3344e+00, ..., -4.4922e+00,
-3.0909e+00, -4.1040e+00]],
[[-1.8886e-01, 3.0314e-01, 5.4854e-01, ..., 6.6173e-01,
8.6702e-01, 1.4000e+00],
[ 1.6329e+00, 1.0904e+00, 1.7723e+00, ..., 1.7066e+00,
2.5810e+00, 3.0450e+00],
[ 1.1474e+00, 2.3829e-01, 2.3634e-03, ..., 1.3725e+00,
1.4514e+00, 2.3074e+00],
...,
[ 1.0429e+00, -3.3911e-01, -1.3310e-01, ..., 1.6271e+00,
1.0947e+00, 1.2984e+00],
[ 1.6354e+00, 7.1996e-01, 1.2996e+00, ..., 1.3140e+00,
6.1340e-01, 2.0656e+00],
[ 1.1619e+00, 8.5009e-01, 1.5558e+00, ..., 1.3983e+00,
1.2977e+00, 1.4024e+00]],
[[-5.5243e-01, 2.6459e-01, 1.1452e-01, ..., 6.0205e-01,
5.4419e-01, 1.3540e+00],
[-1.3497e+00, -1.8429e+00, -1.6998e+00, ..., -7.8397e-01,
-4.7162e-01, 2.1097e-01],
[-2.1795e+00, -8.3985e-01, -6.5585e-01, ..., -3.2760e-02,
2.7301e-03, 7.0097e-01],
...,
[-1.4788e+00, -1.8280e+00, -7.6944e-01, ..., -4.4215e-01,
-8.9081e-01, -6.7972e-01],
[-1.2340e+00, -1.8867e+00, -1.0413e+00, ..., -6.2573e-01,
-1.1869e+00, -3.4659e-02],
[ 1.9807e-01, 1.7536e-01, -6.8554e-02, ..., 7.2746e-02,
-2.6913e-02, 7.5728e-01]]],
[[[ 9.2399e-01, 1.5578e+00, 1.5550e+00, ..., 1.3620e+00,
1.0844e+00, 1.0105e+00],
[-9.1362e-01, -2.4438e+00, -4.3105e-01, ..., -4.5213e-01,
-5.2063e-01, -1.5009e+00],
[-1.1695e-01, -9.7347e-01, 9.7927e-01, ..., -3.1724e-01,
7.7531e-01, 7.9840e-01],
...,
[ 5.3274e-02, -1.1147e+00, -9.7899e-01, ..., -2.2288e-01,
1.8648e-01, 2.5840e-01],
[-5.7837e-01, -1.7759e+00, -1.1751e+00, ..., -3.0255e-01,
-1.4285e-01, -1.4224e-01],
[-3.6164e-01, -1.6837e+00, -1.6376e+00, ..., -1.1331e+00,
-1.5421e+00, -2.7775e-01]],
[[-4.2301e+00, -1.5162e-01, -1.6122e+00, ..., -1.4288e+00,
-2.2412e+00, -9.4936e-01],
[-4.1255e+00, -1.9282e+00, -1.0641e+00, ..., -1.9470e+00,
-3.1720e+00, -8.2689e-01],
[-2.4270e+00, -1.1227e+00, -1.9233e+00, ..., -2.4553e+00,
-2.9128e+00, -9.9194e-01],
...,
[-1.4987e+00, -1.5476e+00, -2.4018e+00, ..., -3.5899e+00,
-1.7184e+00, -6.9966e-01],
[-2.9592e+00, -1.6424e+00, -2.6709e+00, ..., -2.2223e+00,
-1.9886e+00, -1.1920e+00],
[-2.3142e+00, -1.9494e+00, -2.6554e+00, ..., -1.9586e+00,
-1.9315e+00, -1.0072e+00]],
[[ 3.2590e+00, 4.5767e+00, 1.3813e+00, ..., 2.4441e+00,
8.8738e-01, 2.7989e+00],
[ 7.5314e-01, 1.2540e+00, -5.0584e-02, ..., 7.3186e-01,
-1.1258e+00, 2.5085e+00],
[ 1.4904e+00, 1.4336e+00, 1.6861e+00, ..., 1.0463e+00,
-2.3316e-01, 2.0119e+00],
...,
[ 9.6983e-01, 1.0509e+00, 1.2509e+00, ..., -1.3297e-01,
1.0848e+00, 2.3130e+00],
[ 6.5854e-01, 1.7862e+00, 1.1689e+00, ..., 3.6816e-01,
9.6835e-01, 1.8138e+00],
[ 1.6395e+00, 1.6035e+00, 1.8339e+00, ..., 1.2575e+00,
1.7749e+00, 1.7834e+00]],
...,
[[-2.5334e+00, -7.8345e+00, -5.5426e+00, ..., -4.0606e+00,
-3.0370e+00, -5.4136e+00],
[-3.0457e+00, -7.0324e+00, -3.7084e+00, ..., -3.5330e+00,
-2.5513e+00, -4.1799e+00],
[-1.1404e+00, -4.3504e+00, -3.4651e+00, ..., -1.9048e+00,
-2.7547e+00, -2.4036e+00],
...,
[-2.5238e+00, -5.1827e+00, -3.9690e+00, ..., -4.5495e+00,
-4.1957e+00, -4.5330e+00],
[-2.6015e+00, -5.5837e+00, -5.3550e+00, ..., -6.8956e+00,
-6.2980e+00, -6.1582e+00],
[-1.3859e+00, -3.9160e+00, -2.7866e+00, ..., -3.3286e+00,
-2.7254e+00, -3.1732e+00]],
[[-3.2237e-01, 6.9005e-01, 1.1229e-01, ..., 7.3414e-01,
8.6705e-01, 1.4401e+00],
[ 1.8306e+00, 1.3067e+00, 2.4734e+00, ..., 1.8537e+00,
2.3762e+00, 2.8772e+00],
[ 1.3882e+00, 3.2200e-01, 9.6953e-02, ..., 1.9037e+00,
1.7397e+00, 2.3142e+00],
...,
[ 1.2950e+00, 2.7110e-01, -2.6415e-02, ..., 9.5180e-01,
1.5869e+00, 1.6267e+00],
[ 1.8518e+00, 1.5848e+00, 1.0150e+00, ..., 1.6673e+00,
1.3688e+00, 1.9729e+00],
[ 4.4410e-01, 7.8040e-01, 7.1845e-01, ..., 4.9434e-01,
6.6487e-01, 1.6487e+00]],
[[-6.1068e-01, 3.9034e-01, -4.6046e-01, ..., 5.9799e-01,
5.9586e-01, 1.3589e+00],
[-1.4926e+00, -1.7486e+00, -2.7555e+00, ..., -7.8302e-01,
-4.4742e-01, 1.6917e-01],
[-2.4030e+00, -7.7633e-01, -1.4976e-01, ..., 4.9113e-01,
1.2100e-01, 6.8108e-01],
...,
[-1.8476e+00, -1.7796e+00, -1.4457e+00, ..., -2.7688e+00,
-1.1930e+00, -8.8250e-01],
[-1.5426e+00, -1.9736e+00, -1.6629e+00, ..., -3.4354e-01,
-8.4995e-01, -6.9381e-01],
[-8.5177e-01, -8.1217e-01, -1.1583e+00, ..., -5.4935e-01,
-8.5698e-01, -1.0433e-01]]],
[[[ 1.1127e+00, 1.6060e+00, 1.2684e+00, ..., 2.5744e-01,
3.5907e-01, 2.0284e-01],
[-1.1738e+00, -1.6245e+00, -1.0451e+00, ..., 4.1477e-01,
3.8433e-01, -1.1207e+00],
[-6.2580e-01, -7.1614e-01, -5.3110e-01, ..., -3.6043e-01,
9.9668e-01, 6.8479e-01],
...,
[ 6.0241e-01, 1.4430e+00, 3.6742e-01, ..., -6.8703e-01,
7.1806e-01, 5.2984e-01],
[ 5.0198e-03, -3.0033e-01, 1.9365e-01, ..., 2.2007e-01,
1.5224e+00, -9.3803e-02],
[-1.9225e-01, 7.8170e-02, -7.7195e-01, ..., -1.5315e-01,
3.7692e-01, 5.5742e-01]],
[[-3.4471e+00, -1.9009e+00, -2.2837e+00, ..., -1.8563e+00,
-2.3071e+00, -1.1425e+00],
[-3.2518e+00, -3.3124e+00, -1.6930e+00, ..., -3.6298e-01,
-1.2270e+00, -2.9002e-01],
[-1.9093e+00, -1.6985e+00, -1.1134e+00, ..., -1.1867e+00,
-1.3694e+00, -7.0906e-01],
...,
[-4.5878e-01, -2.4927e+00, -2.8545e+00, ..., -3.5196e+00,
-1.0329e-02, 4.6795e-01],
[-1.0521e+00, -1.3672e+00, -1.0366e+00, ..., -1.4709e+00,
-2.1489e-02, 2.2541e-01],
[-3.0497e-01, -9.5784e-01, -9.1071e-01, ..., -1.1506e+00,
-6.7452e-01, -5.3860e-01]],
[[ 2.5887e+00, 2.5644e+00, 1.8584e+00, ..., 2.9940e+00,
1.4298e+00, 2.9375e+00],
[ 1.8091e-01, -6.2105e-02, 1.4275e+00, ..., 1.9408e+00,
8.7006e-03, 2.5174e+00],
[-2.2392e-01, -1.5844e-01, 1.1069e+00, ..., 1.2458e+00,
6.4286e-01, 1.8264e+00],
...,
[-1.0986e+00, -3.5140e+00, -3.2449e-01, ..., -2.4418e-01,
-1.9402e-01, 1.3403e+00],
[-2.9894e-01, 9.6041e-01, 1.1428e+00, ..., 4.5713e-01,
-1.3346e-02, 1.4183e+00],
[ 8.4369e-01, 8.3475e-01, 1.0435e+00, ..., 1.0127e+00,
9.2289e-01, 1.9653e+00]],
...,
[[-3.4731e+00, -9.1114e+00, -7.2034e+00, ..., -3.6066e+00,
-2.4069e+00, -3.9415e+00],
[-4.6842e+00, -1.0638e+01, -6.4932e+00, ..., -6.0379e+00,
-3.8412e+00, -4.3881e+00],
[-3.0741e+00, -8.0123e+00, -5.1807e+00, ..., -3.3024e+00,
-2.3163e+00, -1.5922e+00],
...,
[-2.4290e+00, -4.8685e+00, -4.3553e+00, ..., -8.4959e+00,
-3.1259e+00, -1.4655e+00],
[-3.5066e+00, -7.5316e+00, -3.9821e+00, ..., -6.7204e+00,
-4.9031e+00, -3.5587e+00],
[-2.7897e+00, -7.3463e+00, -3.2579e+00, ..., -3.5008e+00,
-2.4801e+00, -2.6128e+00]],
[[-4.6342e-01, 8.6833e-01, 8.5547e-01, ..., 1.3641e+00,
1.0548e+00, 1.2426e+00],
[ 7.5058e-01, 8.3710e-01, 1.3082e+00, ..., 2.0440e+00,
2.3544e+00, 2.5665e+00],
[ 4.4389e-01, 4.9999e-01, 1.1317e+00, ..., 1.6633e+00,
1.6228e+00, 1.9227e+00],
...,
[ 4.1448e-01, -3.5215e-01, 3.5082e-01, ..., 5.0822e-01,
2.7516e+00, 1.5872e+00],
[ 6.7873e-01, 9.7364e-01, 1.8642e+00, ..., 2.2474e+00,
1.8783e+00, 2.3920e+00],
[ 7.8321e-01, 6.8428e-01, 1.2383e+00, ..., 1.1964e+00,
1.3813e+00, 1.3324e+00]],
[[-1.2221e-01, -2.7798e-01, -4.2032e-01, ..., -1.4443e-01,
-1.4189e-01, 8.8689e-01],
[-8.2840e-01, -3.0127e+00, -1.3890e+00, ..., 2.9179e-01,
-4.3492e-02, 3.7455e-01],
[-2.3654e+00, -2.6670e+00, -7.2166e-01, ..., 4.4718e-01,
1.4955e-01, 2.1268e-01],
...,
[-1.3167e+00, -2.8194e+00, -1.6294e+00, ..., -4.0086e+00,
-1.0906e+00, -2.8982e-01],
[-1.2810e+00, -2.8875e+00, -6.4957e-01, ..., -1.5995e+00,
-2.4140e-01, 2.9752e-01],
[-4.1027e-01, -1.1551e+00, -4.2957e-01, ..., -3.6893e-01,
2.2685e-01, 3.0525e-01]]],
[[[ 8.3976e-01, 1.2991e+00, 5.6701e-01, ..., -1.8119e+00,
-3.0910e+00, 8.9393e-03],
[-1.3783e+00, -2.0420e+00, -4.6809e-01, ..., -5.4610e+00,
-7.4762e+00, 7.5753e-01],
[-1.1344e-01, -7.0885e-01, 5.6840e-01, ..., -6.4054e+00,
-7.4264e+00, 1.5688e+00],
...,
[-1.4798e+01, -2.4793e+01, -2.4950e+01, ..., 1.0525e+00,
6.8388e-01, 6.4783e-01],
[-1.4267e+01, -2.5130e+01, -2.6384e+01, ..., 2.3765e-01,
1.9912e-01, 1.5784e-01],
[-4.3706e+00, -8.4576e+00, -8.6627e+00, ..., 2.6416e-01,
8.8947e-02, 1.5315e-01]],
[[-4.3588e+00, -1.9673e+00, -2.7532e+00, ..., -6.3139e+00,
-3.5048e+00, -5.5947e+00],
[-3.8457e+00, -1.5562e+00, 2.2854e-01, ..., -9.6968e+00,
-1.1365e+01, -1.2051e+01],
[-1.7778e+00, -4.7523e-01, -5.0818e-01, ..., -2.1008e+01,
-1.5966e+01, -9.7508e+00],
...,
[-2.0233e+01, -2.9352e+01, -3.9602e+01, ..., -1.2784e+00,
5.9192e-01, 6.6315e-01],
[-2.4315e+01, -3.5193e+01, -6.1545e+01, ..., -1.4074e+00,
4.0467e-01, 4.2735e-01],
[-1.1235e+01, -2.0115e+01, -3.7648e+01, ..., -1.0981e+00,
-1.4209e+00, -1.0080e+00]],
[[ 3.2077e+00, 3.7208e+00, 3.3207e+00, ..., -3.5643e+00,
-4.9491e+00, 4.3485e-02],
[ 1.3107e+00, 1.1995e+00, 2.1480e+00, ..., -1.0220e+01,
-1.2219e+01, -1.3955e+00],
[ 1.2243e+00, -2.0800e-01, -1.1714e-01, ..., -1.2986e+01,
-8.7202e+00, 2.4615e+00],
...,
[-6.9277e+00, -7.5591e+00, -1.6060e+01, ..., -9.0535e-01,
-8.1124e-01, 3.4865e-01],
[-7.9799e+00, -8.0545e+00, -1.2989e+01, ..., -1.3588e+00,
-7.3212e-01, 5.6680e-01],
[-1.8145e+00, -7.1059e+00, -7.3029e+00, ..., -8.5050e-03,
9.5672e-01, 1.5619e+00]],
...,
[[-3.4591e+00, -8.6832e+00, -4.8505e+00, ..., -2.6474e+00,
-3.5424e-01, -1.5242e+01],
[-3.2395e+00, -8.5214e+00, -4.8908e+00, ..., 8.9352e-01,
-1.4046e+00, -3.3561e+01],
[-2.1150e+00, -5.4873e+00, -3.0002e+00, ..., 2.9256e+00,
1.3292e+00, -2.1232e+01],
...,
[-1.4021e+01, -2.0465e+01, -9.6100e+00, ..., -4.8398e+00,
-1.7162e+00, -1.5202e+00],
[-1.1465e+01, -1.7186e+01, -5.4853e+00, ..., -4.4206e+00,
-2.4413e+00, -2.9353e+00],
[-3.7546e+00, -5.5488e+00, -3.5981e+00, ..., -3.3322e+00,
-2.0821e+00, -1.4317e+00]],
[[-6.8997e-01, -3.7539e-01, -2.6421e-01, ..., -3.3006e+00,
-5.1420e+00, -4.6733e+00],
[ 1.5415e+00, 1.0770e+00, 1.5906e+00, ..., -7.0371e+00,
-7.0840e+00, -7.5251e+00],
[ 1.4160e+00, -4.0994e-03, -9.8730e-01, ..., -6.2543e+00,
-8.7955e+00, -3.0269e+00],
...,
[-3.0198e+00, -1.3562e+01, -2.4436e+01, ..., 3.4175e-01,
1.6175e+00, 8.8930e-01],
[-7.9753e+00, -2.3059e+01, -3.5018e+01, ..., 1.1876e+00,
1.3426e+00, 1.5924e+00],
[-6.3890e+00, -1.6681e+01, -2.7251e+01, ..., 1.6931e-01,
1.1777e-01, 8.9730e-01]],
[[-8.3046e-01, -2.6061e-01, -5.0521e-01, ..., -5.1891e+00,
2.0340e+00, -3.9869e-01],
[-1.9878e+00, -1.1542e+00, -1.5372e-01, ..., -1.0616e+01,
2.1695e+00, -5.5849e+00],
[-2.0417e+00, -5.4305e-02, 8.9373e-01, ..., -1.0234e+01,
5.6394e-01, -4.2970e+00],
...,
[-4.5031e+00, -1.1774e+01, -1.2935e+01, ..., -1.6200e+00,
-4.0430e-01, -3.5135e-01],
[-8.3074e+00, -2.2168e+01, -2.6250e+01, ..., -1.1205e+00,
-1.2045e+00, -4.6517e-01],
[-4.6384e+00, -6.8947e+00, -8.3017e+00, ..., -5.0840e-01,
-1.3999e+00, -6.0417e-01]]],
[[[ 7.3736e-01, 1.0982e+00, 7.0625e-01, ..., 9.7671e-01,
9.2114e-01, 2.9145e-01],
[-3.1373e-01, -1.4323e+00, -6.9404e-01, ..., 2.3149e-01,
5.8014e-01, -5.6036e-01],
[ 1.0124e-01, -8.2481e-01, -3.5037e-01, ..., -2.7250e-01,
2.6320e-01, 8.7863e-01],
...,
[ 9.8641e-02, -9.3737e-01, -6.7845e-01, ..., 1.3926e+00,
-4.7694e-01, 1.0956e+00],
[-4.0472e-01, -1.7600e+00, -3.3345e-01, ..., 1.0762e+00,
9.2493e-01, -8.7117e-01],
[ 3.7612e-01, -6.7527e-01, 5.5544e-01, ..., -2.8871e-01,
2.1606e-01, 7.9147e-01]],
[[-3.0142e+00, -1.4674e+00, -2.9946e+00, ..., -1.6705e+00,
-2.9403e+00, -1.0297e+00],
[-4.3519e+00, -1.8366e+00, -2.2631e+00, ..., -1.3018e+00,
-2.6836e+00, -4.3118e-01],
[-4.3381e+00, -2.5999e+00, -3.0873e+00, ..., -2.2867e+00,
-2.4945e+00, -5.2116e-01],
...,
[-2.7495e+00, -2.0313e+00, -2.9394e+00, ..., -3.4755e-01,
6.0125e-01, 5.9773e-01],
[-3.5877e+00, -1.7274e+00, -2.6540e+00, ..., -6.0339e-01,
-1.4857e+00, -1.3684e-01],
[-1.2640e+00, -1.5792e+00, -2.2883e+00, ..., -9.0254e-01,
-4.8693e-01, -5.3718e-02]],
[[ 1.6962e+00, 2.8059e+00, 2.3811e+00, ..., 2.3970e+00,
8.0081e-01, 2.1047e+00],
[ 1.1585e-01, 1.1867e+00, 1.4888e+00, ..., 1.4424e+00,
-1.0974e-01, 2.0806e+00],
[ 6.1045e-01, 1.4754e+00, 8.1815e-01, ..., 8.1231e-02,
3.2211e-01, 1.8423e+00],
...,
[ 1.0255e+00, 2.4011e+00, 5.7727e-01, ..., -2.4205e+00,
-3.1523e-01, 2.4071e-01],
[ 6.7915e-01, 2.7979e+00, 6.2493e-01, ..., 4.0778e-01,
-2.0053e-01, -1.0063e-01],
[ 1.5581e+00, 2.4148e+00, 2.0280e+00, ..., 7.9135e-01,
7.5432e-01, 1.2938e+00]],
...,
[[-1.4375e+00, -4.6550e+00, -3.9587e+00, ..., -4.7919e+00,
-3.4403e+00, -4.2058e+00],
[-1.8091e+00, -3.8926e+00, -4.3049e+00, ..., -5.7238e+00,
-3.8754e+00, -4.5306e+00],
[-9.7801e-01, -3.2732e+00, -3.9423e+00, ..., -5.2951e+00,
-3.1886e+00, -2.2701e+00],
...,
[-1.5728e+00, -4.3012e+00, -6.2754e+00, ..., -4.1669e+00,
-3.3063e+00, -4.6561e+00],
[-1.5068e+00, -4.6523e+00, -5.4703e+00, ..., -6.4007e+00,
-4.1734e+00, -4.3537e+00],
[-1.6177e+00, -4.9136e+00, -3.7670e+00, ..., -4.5291e+00,
-2.0796e+00, -3.1246e+00]],
[[ 8.1375e-01, 5.7101e-01, -1.1847e-01, ..., 1.3831e+00,
9.9023e-01, 1.6876e+00],
[ 1.1720e+00, 9.8436e-01, 5.2000e-01, ..., 2.5588e+00,
1.9025e+00, 2.7846e+00],
[ 5.4890e-01, 5.6984e-01, 3.2353e-03, ..., 2.4097e+00,
1.5283e+00, 2.0161e+00],
...,
[ 8.8496e-01, -2.7091e-01, -3.3143e-01, ..., 3.7050e+00,
1.9123e+00, 1.0963e+00],
[ 1.5720e+00, 1.3024e+00, 1.4177e+00, ..., 2.7680e+00,
9.3534e-01, 2.6056e+00],
[ 6.7676e-01, 5.5022e-01, 7.2748e-01, ..., 1.5047e+00,
1.0958e+00, 1.2319e+00]],
[[-2.6053e-01, -1.7308e-01, -9.9199e-01, ..., 5.8577e-01,
2.0402e-01, 1.1135e+00],
[-1.3746e+00, -1.4624e+00, -2.2452e+00, ..., -2.3061e-01,
-1.0885e-01, 6.1377e-01],
[-1.6651e+00, -7.8666e-01, -1.1620e+00, ..., -3.0202e-01,
4.0250e-02, -1.6541e-03],
...,
[-1.7415e+00, -1.7705e+00, -2.2414e+00, ..., -1.6244e+00,
-4.0321e-01, -2.5931e+00],
[-1.3888e+00, -1.5331e+00, -2.1017e+00, ..., -8.5087e-01,
-6.5338e-01, -4.4775e-01],
[-1.6976e-01, 9.1999e-02, -3.4040e-01, ..., -1.3013e+00,
-8.2127e-01, 9.1366e-02]]]], device='cuda:0'),
tensor([[[[ 8.7136e-01, 1.5360e+00, 1.2080e+00, ..., 1.7104e+00,
2.3184e+00, 2.2923e+00],
[-9.5048e-01, -2.0783e+00, -1.1354e+00, ..., -2.6272e+00,
-3.0561e+00, -3.7162e+00],
[-1.3629e-01, -7.6996e-01, -1.9771e-01, ..., -3.8456e+00,
-1.6553e+00, -2.1099e+00],
...,
[ 5.4322e-01, -9.2357e-01, -2.7906e+00, ..., -4.7325e-01,
-1.1574e-02, 2.1610e-01],
[-1.8992e+00, -6.3178e-01, -3.1566e+00, ..., -5.9603e-01,
-3.2720e-01, -1.3659e-01],
[-4.4137e-01, 9.3256e-01, -4.9013e+00, ..., -1.1667e+00,
-1.5499e+00, -2.8529e-01]],
[[-3.8189e+00, -8.2737e-01, -1.7556e+00, ..., 1.5518e+00,
6.5259e-01, -8.9173e-01],
[-4.0522e+00, -3.0588e+00, -2.1427e+00, ..., -6.4017e-01,
-4.6484e+00, -2.6742e+00],
[-2.4434e+00, -1.9766e+00, -1.1458e+00, ..., -3.8353e+00,
-2.2632e+00, -3.1774e-01],
...,
[-3.5699e+00, -9.8676e-01, -4.2030e-02, ..., -2.2222e+00,
-2.2464e+00, -7.6192e-01],
[-2.1304e+00, -1.6844e-01, 2.3851e+00, ..., -2.4147e+00,
-2.3982e+00, -1.2387e+00],
[-3.2243e+00, -4.7734e+00, -4.6087e+00, ..., -2.0127e+00,
-1.9904e+00, -1.0077e+00]],
[[ 2.9505e+00, 4.3085e+00, 3.3799e+00, ..., -3.9150e+00,
-3.1402e+00, -1.1036e+00],
[ 3.2105e-01, 1.2531e+00, 2.3338e+00, ..., -2.3287e+00,
-3.0001e+00, -1.6647e+00],
[ 1.1171e+00, 1.6066e+00, 1.2706e+00, ..., -8.2454e-01,
4.9021e-01, -8.9964e-01],
...,
[ 8.6062e-01, -1.0373e+01, -2.0401e+00, ..., 2.0743e-01,
1.0965e+00, 2.3160e+00],
[ 1.2462e+00, -7.7912e+00, -3.3356e+00, ..., 5.0012e-02,
1.0262e+00, 1.8232e+00],
[ 3.2847e+00, -2.4001e+00, -8.1502e+00, ..., 1.2960e+00,
1.7709e+00, 1.7672e+00]],
...,
[[-1.9193e+00, -7.3341e+00, -5.4598e+00, ..., -4.5060e+00,
-6.6459e+00, -9.4824e+00],
[-2.4627e+00, -7.1369e+00, -4.4599e+00, ..., -1.0777e+01,
-9.8830e+00, -4.8363e+00],
[-9.4997e-01, -4.6886e+00, -4.0715e+00, ..., -3.0188e+00,
-1.0756e+00, 9.2160e-01],
...,
[-3.9675e+00, -6.0805e+00, -9.5842e+00, ..., -4.9235e+00,
-4.5081e+00, -4.5834e+00],
[-2.4997e+00, -6.6649e+00, -5.4088e+00, ..., -6.2251e+00,
-5.6912e+00, -6.1664e+00],
[-1.4067e+00, -4.2590e+00, -3.9475e+00, ..., -3.3447e+00,
-2.7513e+00, -3.2132e+00]],
[[-2.1982e-01, 2.6772e-01, 5.7691e-01, ..., -4.0179e-01,
1.7426e-01, -2.4705e-01],
[ 1.6433e+00, 1.0158e+00, 1.4813e+00, ..., -5.3913e-01,
-1.9482e+00, -5.6768e-01],
[ 1.1146e+00, 3.3835e-01, 7.2905e-01, ..., -1.8934e+00,
-1.9950e+00, -3.5336e-02],
...,
[ 1.3915e+00, 1.3835e+00, -2.8548e+00, ..., 1.1250e+00,
1.0849e+00, 1.6770e+00],
[ 1.9640e+00, 4.0904e+00, -1.7767e-01, ..., 1.4888e+00,
1.2893e+00, 1.9888e+00],
[-1.0863e+00, 2.1759e+00, 2.1751e+00, ..., 4.9557e-01,
6.8633e-01, 1.6472e+00]],
[[-5.9135e-01, 2.9240e-01, 8.0563e-02, ..., 1.3932e+00,
3.7172e-01, 3.5785e-01],
[-1.4043e+00, -2.0073e+00, -1.5354e+00, ..., 6.2178e-01,
-2.2975e+00, -2.9905e+00],
[-2.3231e+00, -1.1096e+00, 6.2403e-01, ..., -6.1633e+00,
-4.9956e+00, -2.1613e+00],
...,
[-2.9009e+00, -4.5010e-01, 1.1783e-01, ..., -1.0926e+00,
-1.1480e+00, -8.3183e-01],
[-2.2848e+00, -4.8493e+00, -2.0233e+00, ..., -1.1697e+00,
-1.3359e+00, -6.8015e-01],
[ 4.1087e-02, -2.4065e+00, -2.9664e+00, ..., -6.1237e-01,
-8.7075e-01, -1.1022e-01]]],
[[[ 5.1941e-01, 4.4770e-01, 3.5134e-01, ..., -7.0326e+00,
-1.1636e+01, -8.0936e+00],
[-4.8835e-02, -9.5792e-01, -4.0843e-01, ..., -1.1534e+01,
-1.4192e+01, -1.1275e+01],
[ 2.9317e-01, -9.6912e-02, -6.8795e-02, ..., -6.7296e+00,
-1.2797e+01, -9.3610e+00],
...,
[-1.7053e+00, -9.4444e-01, -8.3729e+00, ..., -1.7961e-01,
7.0580e-01, 1.0489e+00],
[ 1.9124e-01, -4.3359e+00, -4.1400e+00, ..., 7.6249e-01,
9.3132e-01, -1.8652e-01],
[ 4.3260e+00, -1.9679e+00, -2.7837e+00, ..., -1.1175e-01,
-2.0948e-01, 8.2312e-01]],
[[-3.8710e+00, -8.0680e-01, -2.3835e+00, ..., -2.2698e+00,
-7.2404e+00, -5.4120e+00],
[-2.1557e+00, 1.4520e-01, -1.0834e+00, ..., -9.8627e+00,
-1.5255e+01, -1.0464e+01],
[-9.2418e-01, 1.2536e+00, -3.4329e-01, ..., -5.1373e+00,
-1.8300e+01, -1.5721e+01],
...,
[-1.8079e+01, -1.8258e-01, -3.2393e+00, ..., -3.0442e-01,
-1.4639e+00, 4.7007e-01],
[-2.0619e+01, 7.0635e-01, -4.1798e+00, ..., -7.2988e-01,
-1.3915e+00, 4.0722e-01],
[-1.0701e+01, -1.2251e+00, -3.6011e+00, ..., -9.5400e-01,
-1.1097e+00, -6.8735e-01]],
[[ 2.6586e+00, 2.4603e+00, 2.1115e+00, ..., 4.8210e-01,
-4.0647e+00, -7.2437e+00],
[ 1.1344e+00, -5.8179e-01, 6.7372e-01, ..., -1.5653e+00,
-9.2312e+00, -1.1844e+01],
[ 1.1367e+00, -7.0158e-01, -6.9909e-01, ..., -2.1485e+00,
-1.2814e+01, -1.1379e+01],
...,
[-4.3193e+00, -1.4363e+01, -3.3596e+00, ..., -2.0132e+00,
-3.1389e+00, -2.0777e+00],
[-1.0332e+01, -1.6837e+01, -5.1085e+00, ..., 1.2062e+00,
6.6219e-01, 9.4218e-01],
[-8.7189e+00, -1.4919e+01, -2.3297e+00, ..., 1.3081e+00,
1.4239e+00, 2.0037e+00]],
...,
[[-1.5461e+00, -3.6165e+00, -2.6224e+00, ..., -1.3114e+01,
-8.5772e+00, -6.8148e-01],
[-2.9654e+00, -5.1914e+00, -2.8872e+00, ..., -1.6596e+01,
-8.4146e+00, 1.8142e+00],
[-1.3639e+00, -2.8967e+00, -1.8852e+00, ..., -1.2667e+01,
-5.8109e+00, 9.5708e-01],
...,
[-1.2849e+01, -1.8984e+01, -1.4218e+01, ..., -1.3723e+00,
-2.8334e+00, -4.2298e+00],
[-1.3324e+01, -2.1977e+01, -2.3329e+01, ..., -2.5191e+00,
-2.9414e+00, -5.3774e+00],
[-7.7623e+00, -1.8364e+01, -2.1052e+01, ..., -2.0761e+00,
-9.6008e-01, -3.8556e+00]],
[[ 1.1641e-02, 6.6898e-01, -2.2918e-01, ..., -1.1066e+00,
-1.5199e+00, -1.1492e+00],
[ 1.3340e+00, 7.4661e-01, -1.6353e-01, ..., -7.7589e+00,
-9.2399e+00, -6.6902e+00],
[ 1.3686e+00, 7.3687e-03, -1.1844e+00, ..., -8.3893e+00,
-1.0045e+01, -8.0363e+00],
...,
[-3.1567e+00, 1.0422e+00, -5.2375e+00, ..., 1.9390e+00,
1.7713e+00, 1.7750e+00],
[-1.8191e+00, 2.3796e+00, -4.5551e+00, ..., 1.2377e+00,
1.1111e+00, 2.9062e+00],
[ 1.1331e+00, -3.3076e-02, -4.0363e+00, ..., 1.4551e+00,
1.5807e+00, 1.4328e+00]],
[[-8.0713e-01, -2.6487e-01, -6.4169e-01, ..., -6.2717e-01,
-1.9779e+00, -3.3380e+00],
[-8.7019e-01, -3.0126e-01, -1.2028e+00, ..., -1.3846e+01,
-1.6831e+01, -5.7218e+00],
[-1.4905e+00, -2.3985e-01, -1.7657e-01, ..., -8.5964e+00,
-1.1362e+01, -4.6177e+00],
...,
[-5.7441e+00, 5.4080e+00, -3.0590e+00, ..., -3.6265e-01,
-4.9761e-01, -6.5809e-01],
[-5.7005e+00, 5.2430e+00, -5.9224e+00, ..., -6.3474e-01,
-8.0925e-01, -2.4097e-01],
[-6.6444e-01, 2.7987e+00, -1.0277e+01, ..., -4.3898e-01,
-5.4006e-01, -5.7241e-01]]],
[[[ 6.5245e-01, 1.0279e+00, 6.9450e-01, ..., 7.3792e-01,
2.1164e-01, -2.0267e-01],
[-6.1271e-01, -1.7205e+00, -3.0245e-01, ..., -9.1017e-01,
-5.2591e-01, -1.8079e+00],
[-7.3702e-01, -1.1448e+00, -9.6414e-01, ..., 3.0501e-01,
1.2129e-01, 3.2022e-01],
...,
[ 1.6117e+00, -1.9674e+00, -2.1716e+00, ..., 3.3112e-01,
8.1308e-01, 9.7347e-01],
[ 1.0099e-01, -1.0179e+00, -5.0190e-01, ..., -1.3466e+00,
3.4744e-01, 1.2860e-01],
[ 3.2214e-01, -6.4647e-01, -9.5254e-01, ..., -1.3779e-01,
-5.0333e-01, 4.7507e-01]],
[[-3.5510e+00, -1.4879e+00, -3.2448e+00, ..., -2.5054e+00,
-3.9234e+00, -1.3880e+00],
[-5.3542e+00, -1.8504e+00, -1.1689e+00, ..., -4.3113e+00,
-6.4074e+00, -2.0166e+00],
[-5.9385e+00, -6.5808e+00, -4.0187e+00, ..., -2.4477e+00,
-3.5032e+00, -7.8158e-01],
...,
[-5.1830e+00, -5.2348e+00, -7.4018e+00, ..., -2.3791e+00,
-3.2919e+00, -2.8195e-01],
[-4.0609e+00, -1.8403e+00, -3.6095e+00, ..., -2.6180e+00,
-1.5509e+00, 2.4212e-01],
[-1.2664e+00, -1.0774e+00, -1.8889e+00, ..., -8.9522e-01,
-5.3140e-01, -1.1834e-01]],
[[ 2.2490e+00, 3.1294e+00, 2.5156e+00, ..., 3.0210e+00,
1.4730e+00, 3.2027e+00],
[ 3.1814e-01, 1.6700e+00, 2.6719e+00, ..., 2.7594e+00,
4.3611e-01, 2.9882e+00],
[ 1.1720e+00, 5.3933e-01, 2.5190e+00, ..., -1.6804e+00,
-5.6956e+00, -3.6183e+00],
...,
[ 3.6935e-01, -9.1772e-01, -7.7979e-01, ..., -3.7666e+00,
-2.9497e-01, 9.3975e-01],
[ 1.3458e+00, 2.7911e+00, 2.2721e+00, ..., -1.6505e+00,
3.6517e-01, 7.7914e-01],
[ 1.5496e+00, 1.5516e+00, 1.4992e+00, ..., 6.8885e-01,
4.1542e-01, 1.0610e+00]],
...,
[[-1.9568e+00, -5.1347e+00, -4.3291e+00, ..., -5.3012e+00,
-4.3814e+00, -4.9243e+00],
[-2.9859e+00, -5.7112e+00, -5.0372e+00, ..., -4.4500e+00,
-3.3429e+00, -4.7037e+00],
[-1.1508e-01, -2.9985e+00, -3.1412e+00, ..., -6.8159e+00,
-4.9153e+00, -6.4995e+00],
...,
[-2.6290e+00, -5.0678e+00, -7.4383e+00, ..., -6.7814e+00,
-5.6224e+00, -4.8906e+00],
[-1.7823e+00, -5.9462e+00, -7.1133e+00, ..., -8.1901e+00,
-4.8133e+00, -6.1573e+00],
[-1.5420e+00, -4.9447e+00, -4.3606e+00, ..., -7.1239e+00,
-3.9629e+00, -4.0239e+00]],
[[ 6.3067e-01, 6.5527e-01, -1.6731e-01, ..., 8.3797e-01,
2.3091e-01, 1.8738e+00],
[ 3.5619e-01, 1.0785e+00, 9.9712e-01, ..., 2.0222e-01,
-2.8368e+00, 1.3950e+00],
[-2.4572e+00, -3.6020e+00, -6.9657e-01, ..., 6.2629e-01,
-1.1555e+00, -5.8039e-01],
...,
[-3.8530e-01, -1.5427e+00, -3.7149e+00, ..., 1.0237e+00,
1.1340e+00, 2.0148e+00],
[ 2.0291e+00, 2.0915e+00, 1.9186e+00, ..., 2.0068e+00,
1.6686e+00, 2.6197e+00],
[ 6.7715e-01, 8.7917e-01, 8.9317e-01, ..., 1.8451e+00,
1.3540e+00, 1.4273e+00]],
[[-3.1275e-01, -2.9274e-01, -1.0005e+00, ..., 5.1861e-01,
1.2055e-01, 5.0728e-01],
[-1.1240e+00, -1.5891e+00, -2.2395e+00, ..., -1.2090e+00,
-3.1123e-01, 9.7005e-01],
[-2.0461e+00, -1.3175e+00, -1.2896e+00, ..., 6.0621e-01,
-3.3772e-01, 2.5992e+00],
...,
[-1.9300e+00, -1.8796e+00, -4.9953e+00, ..., -1.3295e+00,
-2.6674e+00, -1.7768e+00],
[-1.9524e+00, -8.7658e-01, -2.1562e+00, ..., -2.5026e+00,
-1.6784e+00, -1.7964e-01],
[-1.5768e-01, -3.2597e-02, -6.3825e-01, ..., -1.1781e+00,
-2.7423e-01, 2.6556e-01]]],
[[[ 5.3833e-01, 8.9565e-01, 8.2057e-01, ..., -1.9020e+00,
1.6526e+00, 1.6317e-01],
[-6.6446e-01, -2.1142e+00, -5.3664e-01, ..., -3.0579e+00,
-1.1797e+00, -1.0195e+00],
[ 1.4701e-01, -7.7550e-01, -7.1195e-01, ..., -2.3752e+00,
-6.4983e+00, -2.3981e+00],
...,
[ 2.6371e-01, -1.5881e+00, 1.6645e+00, ..., -2.0880e-01,
4.4575e-01, 2.1403e-01],
[-4.7409e-03, -1.5905e+00, 9.2629e-01, ..., 9.4059e-01,
1.6771e+00, -1.3080e-01],
[ 4.2050e-01, -4.1415e-01, -3.9630e-01, ..., 1.0789e-01,
6.6018e-01, 6.6112e-01]],
[[-4.0548e+00, -1.7561e+00, -4.3021e+00, ..., -3.4883e+00,
-5.3085e+00, -2.2688e+00],
[-4.3059e+00, -2.6299e+00, -5.1898e+00, ..., -4.1160e+00,
-5.0745e+00, -1.9820e+00],
[-2.2375e+00, -1.4957e+00, -7.8444e+00, ..., -9.2628e+00,
-4.0657e+00, -2.5049e+00],
...,
[-2.2496e+00, -2.0872e+00, -1.0446e+00, ..., -3.0480e+00,
-2.0756e+00, 3.4815e-01],
[-3.0320e+00, -1.6213e+00, -1.6399e+00, ..., -1.9647e+00,
-1.0241e+00, 6.1176e-02],
[-1.0910e+00, -1.3169e+00, -2.0825e+00, ..., -9.2369e-01,
-1.0711e+00, -8.1148e-01]],
[[ 2.7066e+00, 3.3315e+00, 2.8564e+00, ..., 8.6209e-01,
-2.3766e+00, 1.6223e+00],
[ 1.5191e+00, 1.0612e+00, 2.9211e+00, ..., -1.5890e+00,
-5.0672e+00, 6.6583e-02],
[ 6.6918e-01, -2.6806e-01, 2.6900e+00, ..., -9.4353e+00,
-6.8017e+00, -2.0744e+00],
...,
[ 7.1395e-02, 9.8289e-01, -5.6904e-01, ..., 3.4784e-01,
8.0293e-01, 1.3897e+00],
[ 5.6165e-01, 1.8478e+00, 1.2165e+00, ..., 5.9958e-01,
7.2638e-01, 1.4179e+00],
[ 1.5401e+00, 1.5903e+00, 2.1389e+00, ..., 1.3216e+00,
1.0605e+00, 2.0977e+00]],
...,
[[-2.1743e+00, -5.4713e+00, -4.5260e+00, ..., -5.6135e+00,
-2.8005e+00, -4.1064e+00],
[-2.1747e+00, -3.7390e+00, -3.7150e+00, ..., -1.5205e+01,
-2.2083e+00, -5.5594e+00],
[-5.8822e-01, -1.3560e+00, -1.7489e+00, ..., -2.2197e+01,
1.4935e+00, -2.3082e+00],
...,
[-1.2363e+00, -2.0897e+00, -3.2585e+00, ..., -1.3279e+00,
-2.6790e+00, -4.7272e+00],
[-1.1507e+00, -4.1581e+00, -5.0935e+00, ..., -5.7904e+00,
-3.6654e+00, -4.8092e+00],
[-1.4113e+00, -4.7748e+00, -3.7255e+00, ..., -3.5207e+00,
-2.4649e+00, -3.1098e+00]],
[[ 2.8678e-01, 5.6679e-01, -1.4165e+00, ..., -2.3093e+00,
-3.1366e+00, 3.7159e-02],
[ 9.5770e-01, 7.9387e-01, -2.4342e+00, ..., 2.3276e+00,
-8.4220e-01, 8.1518e-01],
[ 9.8384e-01, 8.5354e-01, -4.7712e+00, ..., -6.3775e+00,
-3.5515e+00, -1.3184e+00],
...,
[ 1.0787e+00, -2.4053e-01, 1.3757e+00, ..., -5.0785e-01,
2.3854e+00, 2.5677e+00],
[ 1.5760e+00, 1.1177e+00, 1.3248e+00, ..., 6.3824e-02,
1.5830e+00, 2.5339e+00],
[ 8.0697e-01, 6.6087e-01, 7.2928e-01, ..., 9.8783e-01,
1.3875e+00, 1.3073e+00]],
[[-4.2700e-01, -1.6961e-01, -2.1071e+00, ..., -2.4710e+00,
7.0781e-01, 4.8586e-01],
[-1.5428e+00, -1.3666e+00, -3.8009e+00, ..., -4.5634e+00,
1.7496e+00, 2.0465e+00],
[-1.3809e+00, -3.3644e-01, -2.5839e+00, ..., -1.0367e+01,
-5.7025e-01, 5.2087e-01],
...,
[-2.2188e+00, -2.1120e+00, -1.5407e+00, ..., -6.3933e-01,
-1.3342e+00, 7.1634e-02],
[-1.7811e+00, -1.8673e+00, -4.8889e-01, ..., -9.7009e-01,
-6.8127e-01, 4.1859e-01],
[-2.9780e-01, -3.2514e-01, -3.0539e-01, ..., 9.9150e-02,
2.2579e-01, 3.7462e-01]]],
[[[ 8.5512e-01, 1.0690e+00, 5.7198e-01, ..., 4.4148e-01,
5.7456e-01, 1.5955e-01],
[-1.2458e+00, -1.1318e+00, -1.7628e-01, ..., -4.8380e-01,
-1.8461e+00, -8.4406e-01],
[ 3.1533e-01, -1.8427e+00, -1.2344e+00, ..., 3.4543e-01,
-1.2252e+00, 3.3154e-01],
...,
[-1.0422e+00, 5.8224e-01, -5.0680e-01, ..., -2.1471e-01,
-1.4220e-01, 3.5720e-01],
[ 7.2512e-01, -1.7018e+00, -6.2861e-01, ..., -1.4513e-01,
-8.2461e-03, -1.4909e-01],
[ 7.3029e-01, 3.5912e-01, -5.4931e-01, ..., -3.7133e-01,
-9.0677e-01, 1.4593e-01]],
[[-4.1081e+00, -1.9702e+00, -2.9686e+00, ..., -1.5884e+00,
-2.3368e+00, -1.7657e+00],
[-3.2606e+00, -1.9738e+00, -1.5499e+00, ..., -3.7474e+00,
-3.2775e+00, -2.5099e+00],
[-4.8118e+00, -2.6429e+00, -2.7531e+00, ..., -3.0116e+00,
-3.6736e+00, -2.2632e+00],
...,
[-1.2935e+00, -4.5985e+00, -4.7428e+00, ..., -9.1141e-01,
-9.2322e-01, -8.2343e-01],
[-2.2871e+00, -1.1305e+00, -1.1166e+00, ..., -8.5759e-01,
-1.0007e+00, -1.1896e+00],
[-4.7465e-01, -6.4746e-01, -7.9021e-01, ..., -6.6985e-01,
-6.1346e-01, -4.9644e-01]],
[[ 2.8505e+00, 3.4630e+00, 3.2676e+00, ..., 2.0338e+00,
7.1616e-01, 2.5701e+00],
[ 7.8425e-01, 1.1383e+00, 2.5073e+00, ..., 2.5204e+00,
1.6887e+00, 2.4425e+00],
[ 1.7044e+00, 2.6986e+00, 3.6284e+00, ..., -6.6665e-01,
1.4977e+00, 1.4406e+00],
...,
[-2.6735e+00, -4.7363e+00, -4.4955e+00, ..., -6.0705e-01,
7.1460e-01, 1.6009e+00],
[-2.4151e-01, -8.0066e-03, 8.5818e-01, ..., -6.0204e-01,
8.9845e-01, 1.2438e+00],
[ 1.9629e+00, 1.6540e+00, 1.8169e+00, ..., 3.7360e-01,
1.3487e+00, 1.3948e+00]],
...,
[[-3.1153e+00, -8.5620e+00, -5.2000e+00, ..., -6.0665e+00,
-3.7818e+00, -4.4932e+00],
[-3.5044e+00, -9.6120e+00, -6.6168e+00, ..., -5.8987e+00,
-4.7500e+00, -4.1638e+00],
[-2.9655e+00, -8.1648e+00, -5.0222e+00, ..., -6.1871e+00,
-8.5516e+00, -3.6643e+00],
...,
[-2.6408e+00, -6.4485e+00, -6.7816e+00, ..., -5.3624e+00,
-3.5404e+00, -2.9208e+00],
[-2.8475e+00, -6.6010e+00, -4.7217e+00, ..., -6.2997e+00,
-4.4321e+00, -4.3996e+00],
[-1.7465e+00, -6.1977e+00, -4.3760e+00, ..., -3.9015e+00,
-2.5797e+00, -2.4899e+00]],
[[-8.3936e-01, -1.0376e-01, 3.0311e-01, ..., 1.8950e+00,
1.8653e+00, 1.3906e+00],
[ 1.4910e+00, 9.1504e-01, 1.1634e+00, ..., -2.1829e-01,
1.9217e+00, 1.9506e+00],
[-8.4799e-01, 1.0717e-01, -4.4697e-01, ..., -2.2832e+00,
7.2036e-01, 1.9348e+00],
...,
[ 1.7159e+00, -1.1340e+00, -5.9795e-01, ..., 2.1105e+00,
7.6992e-01, 1.0533e+00],
[ 9.3047e-01, -1.2200e-01, 1.4379e+00, ..., 1.7001e+00,
7.4305e-01, 1.5456e+00],
[ 1.2613e+00, 8.5932e-01, 1.2354e+00, ..., 8.6486e-01,
6.9265e-01, 1.4578e+00]],
[[-7.2421e-01, -7.6315e-01, -5.7150e-01, ..., -6.1911e-01,
-8.6720e-01, 3.6184e-02],
[-1.4573e+00, -1.7693e+00, -6.3761e-01, ..., -1.8349e+00,
-3.2165e+00, -9.8443e-01],
[-1.6799e+00, -6.5910e-01, -2.7083e-01, ..., 4.1901e+00,
-3.8234e+00, -1.9487e-01],
...,
[-1.6165e+00, -2.4223e+00, -2.9883e+00, ..., -8.0620e-01,
-1.4278e+00, -8.5205e-01],
[-1.3836e+00, -2.2431e+00, -1.8505e+00, ..., -1.0583e+00,
-1.5970e+00, -7.1273e-01],
[-6.8184e-01, 1.5189e-01, 3.5219e-01, ..., -4.5253e-01,
-7.5295e-01, 2.3488e-02]]]], device='cuda:0'),
tensor([[[[-1.9538e+00, -7.2917e-01, -3.5530e-02, ..., 2.2684e-01,
1.9764e+00, 2.4562e-01],
[-3.0291e+00, -4.5903e+00, -3.6633e+00, ..., -4.0060e+00,
1.4466e+00, -5.5216e-01],
[-1.1274e+00, -2.8206e+00, -1.1729e+00, ..., -6.6780e-02,
1.6856e+00, 1.8969e+00],
...,
[ 2.8050e-01, -6.7893e-01, -4.7054e-01, ..., -3.7591e-01,
-2.7793e-01, -2.9322e-01],
[-6.2518e-01, -1.7913e+00, -1.1871e+00, ..., 1.0199e+00,
1.1728e+00, 5.9556e-01],
[-2.5672e-01, -1.7156e+00, -1.7193e+00, ..., -1.1577e+00,
-1.5542e+00, -2.3892e-01]],
[[-2.4011e+00, 2.0373e+00, -1.7298e+00, ..., -2.2307e+00,
-5.7952e+00, -1.8706e+00],
[-3.0765e+00, -7.7665e+00, -1.2279e+01, ..., -4.8891e+00,
-7.5531e+00, -2.0880e+00],
[-2.4637e+00, -1.9995e+00, -4.6009e+00, ..., -8.9050e-01,
-3.4667e+00, -8.4361e-01],
...,
[-1.2910e+00, -1.2771e+00, -1.8748e+00, ..., -9.1311e+00,
-5.9581e+00, -2.5470e+00],
[-2.8371e+00, -1.6140e+00, -2.7735e+00, ..., -2.8134e+00,
-1.6700e+00, -1.1105e+00],
[-2.2261e+00, -2.0540e+00, -2.8384e+00, ..., -1.9610e+00,
-1.9296e+00, -9.9003e-01]],
[[-2.9619e+00, -3.4829e+00, -4.3642e+00, ..., -1.1237e+00,
-9.6400e-01, 1.3188e+00],
[-8.7918e+00, -1.1184e+01, -6.6361e+00, ..., -5.0338e+00,
-2.5346e+00, 7.7560e-01],
[-6.6048e+00, -2.4859e+00, -4.0405e+00, ..., -4.5254e+00,
-8.9598e-01, -5.7304e-02],
...,
[ 1.1715e+00, 1.5404e+00, 1.7128e+00, ..., -2.5215e+00,
-3.7874e+00, -6.9013e-01],
[ 6.5696e-01, 1.9308e+00, 1.3338e+00, ..., 1.6168e+00,
1.5351e+00, 1.6790e+00],
[ 1.7114e+00, 1.6315e+00, 1.8920e+00, ..., 1.2988e+00,
1.7847e+00, 1.7997e+00]],
...,
[[-4.2941e-01, -5.8593e+00, -1.3463e+01, ..., -3.1728e+00,
-8.8547e+00, -7.1866e+00],
[-9.6784e-01, -1.1314e+01, -1.2648e+01, ..., 1.3462e+00,
-9.5506e+00, -5.5402e+00],
[ 3.3778e-02, -5.5652e+00, -2.0349e+00, ..., -2.8263e+00,
-6.8620e+00, -3.8214e+00],
...,
[-2.3209e+00, -5.2555e+00, -3.9151e+00, ..., -6.0781e+00,
-6.9487e+00, -6.8901e+00],
[-2.8797e+00, -6.0173e+00, -5.3752e+00, ..., -7.5521e+00,
-7.7193e+00, -8.7464e+00],
[-1.6271e+00, -4.2963e+00, -2.6971e+00, ..., -3.2423e+00,
-2.6068e+00, -3.1265e+00]],
[[-1.8493e+00, 1.9808e+00, -2.6282e+00, ..., -3.0877e+00,
-2.2622e+00, 1.6250e+00],
[-1.8349e+00, -1.6394e+00, -4.9201e+00, ..., -3.6689e+00,
-3.7763e+00, 2.2072e+00],
[-8.7551e-01, -4.6497e+00, -5.8511e+00, ..., 3.9031e-01,
-1.2533e+00, 1.1621e+00],
...,
[ 1.3607e+00, 3.1793e-01, 2.7713e-01, ..., 3.5026e-01,
-6.2224e-01, 6.2335e-01],
[ 1.9696e+00, 1.7299e+00, 1.0723e+00, ..., 2.4025e+00,
1.6724e+00, 2.1849e+00],
[ 4.9129e-01, 8.0268e-01, 6.5387e-01, ..., 5.2326e-01,
7.0762e-01, 1.6401e+00]],
[[-2.2368e-01, 1.1722e+00, -1.0191e+00, ..., 1.2697e+00,
7.2716e-01, 4.8945e-01],
[ 1.4345e+00, -5.0526e-01, -1.1057e+01, ..., -1.5132e+00,
-1.9020e+00, -1.3619e+00],
[-1.9419e-01, -1.2157e+00, -5.1882e+00, ..., 1.2623e+00,
1.3166e+00, 9.8603e-02],
...,
[-1.9079e+00, -1.8419e+00, -1.1296e+00, ..., -4.9389e+00,
-3.3346e+00, -2.2175e+00],
[-1.6563e+00, -2.0941e+00, -1.6351e+00, ..., -1.2987e+00,
-1.1087e+00, 1.4743e-01],
[-8.3039e-01, -9.4953e-01, -1.2066e+00, ..., -6.0232e-01,
-9.1776e-01, -9.1069e-02]]],
[[[ 9.5092e-01, 1.6183e+00, 1.0815e+00, ..., 1.6886e+00,
1.0392e+00, 9.2553e-01],
[-1.0025e+00, -2.6432e+00, -1.5003e+00, ..., -4.1772e-01,
-2.7953e-01, -1.5395e+00],
[-9.2013e-02, -1.3996e+00, 5.6132e-01, ..., -4.6179e-01,
1.1670e+00, 9.3265e-01],
...,
[ 5.0999e-01, -1.4511e-01, 3.1097e-01, ..., 5.5648e-01,
1.9224e-01, 6.2122e-01],
[ 3.3396e-01, -5.9754e-01, 2.8550e-01, ..., 7.9154e-01,
6.3640e-01, -9.4077e-02],
[ 5.7975e-01, 2.5650e-01, -5.9522e-02, ..., 3.6764e-01,
-9.4450e-02, 6.6702e-01]],
[[-4.2090e+00, -1.1368e+00, -2.7829e+00, ..., -2.1192e+00,
-2.5196e+00, -9.9548e-01],
[-4.2942e+00, -3.4175e+00, -4.1248e+00, ..., -2.4089e+00,
-4.0443e+00, -8.7132e-01],
[-2.8882e+00, -2.6083e+00, -3.7821e+00, ..., -2.1626e+00,
-3.4558e+00, -8.8737e-01],
...,
[-9.6964e-01, -1.1193e+00, -6.0611e-01, ..., -1.5698e+00,
-1.4740e+00, -2.0761e-01],
[-1.2693e+00, -2.8456e-01, 6.4909e-01, ..., -6.9665e-01,
-1.2937e+00, -1.1753e-01],
[ 4.5171e-01, -1.9181e-01, -3.8812e-01, ..., 3.0488e-01,
1.0881e-01, 5.0265e-01]],
[[ 3.3518e+00, 4.8785e+00, 3.1517e+00, ..., 2.0196e+00,
1.2373e+00, 2.9111e+00],
[ 9.3234e-01, 1.9443e+00, 3.3566e-01, ..., 3.2672e-01,
-7.5698e-01, 2.5028e+00],
[ 1.8794e+00, 2.0664e+00, -3.2394e+00, ..., 1.3174e-01,
-6.5353e-02, 1.8506e+00],
...,
[ 4.3658e-01, 4.4482e-01, 9.8331e-01, ..., -6.2375e-01,
2.7162e-01, 1.5590e+00],
[ 2.0275e-01, 1.1382e+00, 1.1173e+00, ..., -4.5454e-01,
1.9493e-01, 1.1736e+00],
[ 1.3671e+00, 8.7622e-01, 1.7016e+00, ..., 6.4358e-01,
4.0065e-01, 1.1236e+00]],
...,
[[-2.7237e+00, -8.1931e+00, -6.7408e+00, ..., -2.8883e+00,
-3.2545e+00, -5.4251e+00],
[-3.3060e+00, -7.7141e+00, -6.7463e+00, ..., -4.3249e-01,
-2.9034e+00, -4.3869e+00],
[-1.4257e+00, -5.2399e+00, -4.8149e+00, ..., -6.2587e-03,
-2.9039e+00, -2.8336e+00],
...,
[-2.2747e+00, -5.7111e+00, -3.7958e+00, ..., -3.9047e+00,
-3.4276e+00, -3.4612e+00],
[-2.4535e+00, -6.6376e+00, -4.9552e+00, ..., -5.1718e+00,
-4.3780e+00, -4.8647e+00],
[-2.3481e+00, -6.5005e+00, -3.9677e+00, ..., -4.3859e+00,
-3.2369e+00, -4.2397e+00]],
[[-3.6925e-01, -1.6398e-02, -8.9683e-01, ..., -1.8595e-01,
8.7264e-01, 1.6662e+00],
[ 1.8274e+00, 4.9787e-01, -6.6987e-01, ..., 4.2428e-01,
2.3341e+00, 3.1687e+00],
[ 1.2188e+00, 1.3812e-01, 2.3705e-02, ..., -3.1436e-01,
1.6650e+00, 2.5260e+00],
...,
[ 1.0885e+00, -8.0119e-02, 4.4239e-01, ..., 1.2296e+00,
7.2006e-01, 1.2416e+00],
[ 1.6894e+00, 8.3015e-01, 1.1570e+00, ..., 1.2817e+00,
4.9985e-01, 2.0573e+00],
[ 1.1970e+00, 7.5612e-01, 1.2006e+00, ..., 1.3909e+00,
1.2441e+00, 1.4115e+00]],
[[-6.9739e-01, -2.4892e-01, -8.7021e-02, ..., 9.0742e-01,
5.1944e-01, 1.4274e+00],
[-1.7024e+00, -2.4598e+00, -2.1794e+00, ..., 1.6377e-01,
-5.5033e-01, 2.4939e-01],
[-2.5496e+00, -1.2769e+00, 6.4262e-01, ..., 4.1342e-01,
1.8149e-01, 7.6159e-01],
...,
[-1.5243e+00, -1.7370e+00, -1.0082e+00, ..., -8.2226e-01,
-1.1359e+00, -7.7207e-01],
[-1.2579e+00, -1.5683e+00, -5.9287e-01, ..., -9.3043e-01,
-1.2711e+00, -1.5003e-02],
[ 1.1349e-01, 5.7420e-02, 1.5132e-01, ..., 2.2677e-01,
-4.4844e-02, 8.4036e-01]]],
[[[ 7.1318e-01, 1.0440e+00, 7.1609e-01, ..., 1.3378e+00,
-2.3659e-01, -2.8137e-01],
[-3.1914e-01, -1.4502e+00, -5.6083e-01, ..., -5.0119e-01,
-7.8823e-02, -1.3826e+00],
[ 1.4839e-01, -7.7146e-01, -2.8383e-01, ..., -2.5336e+00,
1.0020e+00, 5.6751e-01],
...,
[ 9.4508e-02, -1.6241e+00, -1.2880e+00, ..., -4.4529e-01,
8.9272e-01, 4.8408e-01],
[-4.6198e-01, -3.2380e+00, 7.0098e-01, ..., -4.7554e-02,
1.3732e+00, -1.3011e-01],
[ 3.9832e-01, -1.2621e+00, 2.0646e+00, ..., -2.8143e-01,
3.8693e-01, 5.9354e-01]],
[[-3.0884e+00, -1.4536e+00, -3.0292e+00, ..., -2.5046e+00,
-3.2955e+00, -1.1276e+00],
[-4.3892e+00, -1.7503e+00, -2.3002e+00, ..., -1.5826e+00,
-3.6301e+00, -3.7798e-01],
[-4.3273e+00, -2.4996e+00, -2.9175e+00, ..., -3.0760e+00,
-4.4694e+00, -7.2381e-01],
...,
[-2.6614e+00, -3.7622e+00, -5.4117e+00, ..., 3.0658e-01,
-9.1157e-02, 4.6996e-01],
[-3.4802e+00, -3.5276e+00, -4.9458e+00, ..., -3.5594e-01,
-4.4159e-01, 2.5540e-01],
[-1.2153e+00, -2.0611e+00, -3.4109e+00, ..., -3.8986e-01,
-7.4029e-01, -5.4569e-01]],
[[ 1.7720e+00, 2.8278e+00, 2.3598e+00, ..., 8.2550e-01,
1.3828e+00, 3.6540e+00],
[ 1.8082e-01, 1.2037e+00, 1.5037e+00, ..., 1.0424e-03,
-4.8844e-02, 2.3308e+00],
[ 7.1518e-01, 1.3995e+00, 8.1428e-01, ..., -1.5150e+00,
6.1555e-01, 1.5522e+00],
...,
[ 9.9498e-01, 3.4058e+00, -2.1290e+00, ..., -2.6704e-01,
1.4226e-01, 1.4324e+00],
[ 6.7398e-01, 3.9713e+00, 1.9558e-01, ..., 9.4123e-01,
6.1613e-01, 1.5204e+00],
[ 1.5197e+00, 2.9087e+00, 2.5090e+00, ..., 1.4645e+00,
1.0697e+00, 1.9772e+00]],
...,
[[-1.4936e+00, -4.7335e+00, -3.9765e+00, ..., -1.1374e+00,
-2.0557e+00, -3.9549e+00],
[-1.8545e+00, -3.9811e+00, -4.3033e+00, ..., 6.1029e-02,
-4.0358e+00, -5.8653e+00],
[-9.8052e-01, -3.3554e+00, -3.7700e+00, ..., 4.9565e-01,
-5.9457e+00, -4.9320e+00],
...,
[-1.5366e+00, -4.2145e+00, -7.7020e+00, ..., -8.1427e+00,
-2.3460e+00, -1.5378e+00],
[-1.4350e+00, -4.4471e+00, -6.1272e+00, ..., -5.5099e+00,
-3.3932e+00, -3.7213e+00],
[-1.6160e+00, -4.9948e+00, -4.3040e+00, ..., -3.5207e+00,
-2.2079e+00, -2.7874e+00]],
[[ 7.8643e-01, 5.7528e-01, -1.0522e-01, ..., -1.4186e+00,
3.2903e-01, 2.3001e+00],
[ 1.2214e+00, 1.0305e+00, 5.1518e-01, ..., -1.4465e+00,
2.8997e-01, 3.7649e+00],
[ 6.4434e-01, 6.0109e-01, 9.2717e-02, ..., -2.4309e+00,
-4.5826e-01, 3.2070e+00],
...,
[ 8.9880e-01, -1.4451e+00, -5.3658e-01, ..., 1.8015e+00,
1.8456e+00, 1.6558e+00],
[ 1.5384e+00, -1.1166e-01, 1.6757e+00, ..., 1.3302e+00,
1.5139e+00, 2.4620e+00],
[ 6.5564e-01, -2.5879e-01, 9.3724e-01, ..., 1.0708e+00,
1.3267e+00, 1.3568e+00]],
[[-2.7694e-01, -1.9056e-01, -1.0079e+00, ..., 5.7637e-01,
-6.6518e-01, 8.5416e-01],
[-1.3961e+00, -1.4631e+00, -2.0577e+00, ..., 1.3033e+00,
-6.0740e-01, 4.3741e-01],
[-1.6571e+00, -8.6224e-01, -9.5970e-01, ..., 2.9964e-02,
-6.6731e-01, 1.4963e-01],
...,
[-1.7365e+00, -1.5295e+00, -2.4941e+00, ..., -7.6855e-01,
-3.9424e-01, -3.1632e-01],
[-1.4363e+00, -1.9713e+00, -9.1097e-01, ..., -3.6594e-01,
-4.6582e-01, 3.3879e-01],
[-1.5227e-01, -3.9005e-01, 4.2481e-01, ..., 2.2880e-01,
1.8475e-01, 3.2766e-01]]],
[[[ 8.6364e-01, 1.5306e+00, 1.2626e+00, ..., 4.3631e-01,
6.4927e-01, -8.0690e-01],
[-1.0961e+00, -1.9023e+00, -1.3307e+00, ..., -3.3742e-01,
2.1215e+00, 2.5449e-01],
[-7.4350e-01, -1.2546e+00, -9.1603e-01, ..., -2.8450e+00,
-3.9535e+00, 3.5952e+00],
...,
[-1.4244e+00, 6.4489e-02, -3.3502e+00, ..., -1.0174e+00,
2.8535e-01, 2.0870e-01],
[-1.3375e+00, -4.1779e-01, 6.0743e-01, ..., -2.4040e-01,
-7.7396e-02, -1.2736e-01],
[-7.2464e-01, -2.3051e+00, 1.9544e-01, ..., -1.1906e+00,
-1.4281e+00, -2.0843e-01]],
[[-3.8201e+00, -5.9377e-01, -1.4657e+00, ..., -7.5412e+00,
-7.4011e+00, -3.1798e+00],
[-5.9328e+00, -3.6385e+00, -5.3269e+00, ..., -5.0077e-01,
-5.2667e+00, -4.3892e+00],
[-7.2474e+00, -5.6987e+00, -8.6506e+00, ..., -5.8329e+00,
-3.9302e+00, -4.2795e+00],
...,
[-2.4465e+00, -6.1509e+00, -6.0078e-01, ..., -4.6251e+00,
-9.5670e-01, -5.4281e-01],
[-3.8631e+00, -4.4302e+00, -3.9262e+00, ..., -2.2177e+00,
-1.7672e+00, -1.0877e+00],
[-2.8634e+00, -2.3390e+00, -2.3641e+00, ..., -1.8167e+00,
-1.6972e+00, -8.1386e-01]],
[[ 2.9895e+00, 4.4375e+00, 3.4151e+00, ..., 1.4123e+00,
7.3615e-01, 5.2229e+00],
[ 4.6742e-01, 1.7854e+00, 3.2662e+00, ..., -8.2004e+00,
-6.7820e+00, -5.5775e-01],
[ 2.3541e+00, 1.5975e+00, -5.2487e-01, ..., -6.1841e+00,
-6.7460e+00, -9.8328e-01],
...,
[ 3.0872e+00, -1.1320e+00, -8.7136e+00, ..., 1.4073e+00,
-3.7127e-03, 2.3241e+00],
[ 1.3618e+00, 4.1705e+00, -7.6243e-01, ..., 3.6413e-01,
6.8402e-01, 1.7986e+00],
[ 1.1131e+00, 2.8388e+00, 1.8225e+00, ..., 1.2879e+00,
1.7311e+00, 1.7703e+00]],
...,
[[-1.9125e+00, -7.3237e+00, -5.4763e+00, ..., -3.8251e+00,
-5.8374e+00, -5.3604e+00],
[-3.9679e+00, -8.3130e+00, -4.1268e+00, ..., -1.2459e+01,
-1.1392e+01, -1.3372e+01],
[-2.2118e+00, -7.7076e+00, -5.4075e+00, ..., -1.2501e+01,
-8.8454e-01, -1.1413e+01],
...,
[-2.9380e+00, -9.3362e+00, -4.8456e+00, ..., -8.6805e+00,
-9.7734e+00, -4.4062e+00],
[-3.7580e+00, -8.2541e+00, -7.8321e+00, ..., -8.8015e+00,
-7.6025e+00, -5.9934e+00],
[-1.5611e+00, -4.4017e+00, -5.5999e+00, ..., -3.1821e+00,
-2.6578e+00, -3.1588e+00]],
[[-1.2205e-01, 3.6356e-01, 5.8980e-01, ..., -6.1595e+00,
-2.4302e+00, 2.7154e+00],
[ 1.7288e+00, 9.4496e-01, 2.3002e-01, ..., -1.5378e+00,
-4.2538e+00, 1.5926e+00],
[-1.9007e+00, -3.7082e+00, -5.7464e+00, ..., -2.4103e+00,
-5.6282e+00, -2.7925e+00],
...,
[-7.5825e-01, -1.4466e-01, 4.2924e+00, ..., 2.7596e+00,
2.3027e+00, 1.7368e+00],
[ 9.7367e-01, -7.8444e-01, 1.2402e+00, ..., 2.6059e+00,
1.5467e+00, 2.0106e+00],
[ 3.1146e-01, 9.4423e-01, 8.5556e-01, ..., 5.4804e-01,
7.5202e-01, 1.6411e+00]],
[[-5.2898e-01, 3.5858e-01, 1.6389e-01, ..., 6.6711e-03,
8.7138e-01, -9.9779e-01],
[-1.0584e+00, -1.1214e+00, -1.8486e+00, ..., 4.6819e-01,
5.4822e+00, 1.2067e+00],
[-2.5276e+00, 1.6850e-01, -2.0582e+00, ..., -3.5479e+00,
2.0887e+00, 1.8082e+00],
...,
[-7.8499e-01, -1.7417e+00, -6.1423e-01, ..., -3.6520e+00,
-1.8683e+00, -7.9590e-01],
[-1.7114e+00, 8.3848e-02, -1.5343e+00, ..., -1.5633e-01,
-7.7144e-01, -6.2102e-01],
[-8.8019e-01, -7.1979e-01, -1.9363e+00, ..., -5.3546e-01,
-8.0255e-01, -7.0616e-02]]],
[[[ 1.2776e+00, 1.0313e+00, -3.0048e-01, ..., 5.0660e-01,
8.5423e-01, 3.6857e-01],
[-6.8342e-01, -1.9971e+00, -1.7880e+00, ..., -4.0872e-01,
-1.0386e-01, -1.3325e+00],
[-4.9142e-01, -3.3496e-01, -3.7826e-03, ..., 8.8353e-01,
2.2383e-01, 7.0126e-01],
...,
[ 5.0977e-01, -1.9674e-01, -1.8668e+00, ..., 3.1366e-01,
1.0140e+00, 7.9743e-01],
[-6.6340e-01, -2.0342e+00, -8.7970e-01, ..., 1.0167e+00,
1.6340e+00, 8.8502e-02],
[ 1.5685e+00, -8.1057e-01, -1.6496e+00, ..., 3.1575e-01,
2.6338e-01, 7.5484e-01]],
[[-2.3103e+00, 1.2397e+00, 7.7801e-01, ..., -1.5235e+00,
-2.4437e+00, -6.7483e-01],
[-3.3079e+00, -1.1161e-01, -5.5279e-01, ..., -2.2715e+00,
-3.7525e+00, -1.5871e+00],
[-2.5933e+00, 1.9494e-03, 5.0642e-01, ..., -2.7528e+00,
-2.9570e+00, -1.4344e+00],
...,
[ 6.5939e-02, 1.6551e+00, 4.3818e-01, ..., -5.6792e-01,
-3.2676e-02, 7.8460e-01],
[-2.2524e+00, 5.3344e-01, 3.8579e-01, ..., 5.6512e-02,
-3.0961e-02, 6.6753e-01],
[-1.3067e+00, -1.0765e+00, 5.0952e-01, ..., -6.2112e-01,
-6.0404e-01, -1.6516e-01]],
[[ 2.8947e+00, 1.9702e+00, -8.4866e-01, ..., 2.2896e+00,
4.2600e-01, 2.5600e+00],
[ 1.6295e+00, 2.6086e-01, -2.5637e+00, ..., 2.0748e+00,
9.6103e-01, 2.4870e+00],
[ 2.0978e+00, 1.2044e+00, -2.7636e+00, ..., 7.9528e-01,
1.4863e+00, 2.5982e+00],
...,
[-1.9634e+00, -2.1608e+00, 2.4638e-02, ..., 9.5445e-01,
6.6790e-01, 1.9057e+00],
[-1.7462e+00, -4.8671e+00, 1.6541e-01, ..., 1.0035e+00,
7.4096e-01, 1.7045e+00],
[ 1.1252e+00, -6.1791e+00, 8.4930e-02, ..., 1.9023e+00,
1.0288e+00, 1.7581e+00]],
...,
[[-2.8052e+00, -7.1878e+00, -4.1567e+00, ..., -4.2980e+00,
-4.1702e+00, -5.2316e+00],
[-3.3555e+00, -8.1871e+00, -4.9780e+00, ..., -4.9872e+00,
-5.6172e+00, -5.0504e+00],
[-2.1363e+00, -6.2161e+00, -3.9473e+00, ..., -5.2041e+00,
-5.2145e+00, -2.9516e+00],
...,
[-1.8929e+00, -4.8246e+00, -1.2314e+00, ..., -3.3399e+00,
-2.6152e+00, -2.9636e+00],
[-2.5863e+00, -4.5951e+00, -4.9470e+00, ..., -4.5560e+00,
-3.2539e+00, -4.8578e+00],
[-1.7696e+00, -4.5061e+00, -3.6511e+00, ..., -2.8130e+00,
-1.5050e+00, -3.2496e+00]],
[[ 4.7588e-01, 1.1937e+00, 7.0607e-01, ..., 1.4704e+00,
1.4412e+00, 1.7318e+00],
[ 1.2322e+00, 3.8091e-02, -2.5778e+00, ..., 9.8305e-01,
1.6726e+00, 2.7547e+00],
[ 1.3825e+00, 1.7638e+00, 4.9136e-01, ..., 3.8907e-01,
1.1128e+00, 2.0661e+00],
...,
[ 6.0618e-01, -1.3536e+00, -2.3652e-01, ..., 1.8751e+00,
2.1381e+00, 1.4921e+00],
[ 2.5111e+00, 6.2820e-01, -1.7951e+00, ..., 1.9121e+00,
1.6985e+00, 2.5764e+00],
[ 6.7317e-01, 2.1187e+00, 4.5553e-01, ..., 1.2998e+00,
1.5122e+00, 1.5179e+00]],
[[-6.5362e-01, -2.6001e-01, 1.4224e-01, ..., -3.5303e-02,
-2.6506e-01, 9.1074e-01],
[-1.3956e+00, -1.7519e+00, -1.1618e-01, ..., -9.8602e-01,
-5.0146e-01, 8.4677e-02],
[-2.2489e+00, -1.1426e+00, -1.1964e+00, ..., 3.4717e-01,
-5.6912e-01, -6.5196e-02],
...,
[-1.9981e+00, -2.0231e-01, -5.3433e-01, ..., -1.7111e-01,
-4.8176e-01, -4.5789e-01],
[-2.3122e+00, 4.0497e-01, 5.9633e-01, ..., -2.8649e-01,
-4.1053e-01, 4.2534e-01],
[-1.2621e-01, -5.5840e-01, -4.6449e-01, ..., 1.2079e-01,
-1.6364e-03, 6.2616e-01]]]], device='cuda:0'),
tensor([[[[ 9.6504e-01, 1.7728e+00, 1.6612e+00, ..., 3.7478e-01,
6.8847e-01, 2.5274e-01],
[-4.6512e-01, -2.0781e+00, -1.9315e+00, ..., -6.1395e-01,
3.2826e-01, -1.1632e+00],
[ 1.2326e+00, -5.4022e-01, -1.3530e+00, ..., -3.6829e+00,
-2.7713e+00, 5.5858e-02],
...,
[ 2.0824e+00, -2.6502e+00, -5.1827e+00, ..., 1.2252e+00,
1.2890e+00, 1.4774e-01],
[ 1.4525e+00, -3.1341e+00, -2.6816e+00, ..., -6.7478e-01,
1.9727e+00, -1.6411e-01],
[ 7.9701e-01, -2.7736e+00, 6.6004e-01, ..., 6.6922e-02,
-8.5792e-01, 8.3057e-02]],
[[-4.1684e+00, -8.0470e-01, -3.4115e+00, ..., -2.2577e+00,
-2.7247e+00, -1.5381e+00],
[-4.4715e+00, -3.4672e+00, -6.2038e+00, ..., -9.9163e-01,
-2.5892e+00, -1.9456e+00],
[-4.1075e+00, -4.9993e+00, -6.6978e+00, ..., -4.1574e+00,
-2.6527e+00, -1.0721e+00],
...,
[-5.7613e+00, -1.9184e+00, -5.4703e+00, ..., -4.3807e+00,
-1.5452e+00, -1.4245e-01],
[-4.0631e+00, -6.7869e+00, -1.9564e+01, ..., -4.0736e+00,
-3.4546e-01, -7.3063e-01],
[-2.6962e+00, -1.3243e+00, 1.2422e+00, ..., -6.9718e-01,
-7.1669e-01, -1.9992e-01]],
[[ 3.1247e+00, 4.4939e+00, 3.4873e+00, ..., 1.1345e+00,
6.9250e-01, 2.6350e+00],
[ 7.8905e-02, 8.7211e-01, 3.0708e+00, ..., 3.9252e+00,
1.8187e-01, 2.6288e+00],
[ 2.1785e-01, 1.9327e+00, 3.0006e+00, ..., 3.7121e+00,
3.7106e+00, 1.9033e+00],
...,
[ 1.8633e-01, -6.8669e+00, -7.9889e+00, ..., 5.7667e-01,
-2.6907e+00, 1.8248e+00],
[ 2.0136e+00, 2.2568e+00, -1.4909e+01, ..., -3.4119e+00,
2.3686e-01, 1.7915e+00],
[ 1.4563e+00, 5.9244e+00, 4.1680e-01, ..., 2.7957e-01,
6.8797e-01, 1.6085e+00]],
...,
[[-1.9226e+00, -7.0627e+00, -6.1881e+00, ..., -5.0269e+00,
-4.5888e+00, -5.4268e+00],
[-2.7971e+00, -6.2904e+00, -5.1735e+00, ..., -7.5648e+00,
-7.7769e+00, -6.2206e+00],
[-2.8666e-01, -4.5712e+00, -5.5357e+00, ..., -5.9567e+00,
-5.8637e+00, -5.0426e+00],
...,
[-3.4479e+00, -4.0334e+00, -9.2110e+00, ..., -2.1920e+01,
-1.2810e+01, -6.5166e+00],
[-3.2123e+00, -8.0701e+00, -1.2001e+01, ..., -1.7094e+01,
-9.4533e+00, -5.1036e+00],
[-2.2323e+00, -7.9884e+00, -9.0276e+00, ..., -1.0474e+01,
-7.1779e+00, -3.0757e+00]],
[[-4.6306e-01, -6.9694e-03, -9.3704e-01, ..., 8.7427e-01,
1.3443e+00, 1.5467e+00],
[ 1.9192e+00, 1.0086e-01, 1.6679e-01, ..., 2.9375e+00,
1.3204e+00, 2.1122e+00],
[ 6.5195e-01, -1.0012e+00, -2.5135e+00, ..., -4.0743e-01,
2.8422e+00, 2.0492e+00],
...,
[-1.0037e-01, 3.7139e+00, 4.2346e-01, ..., -7.6201e-01,
3.6634e+00, 1.6086e+00],
[-2.8232e-01, -4.0588e+00, -6.9388e-01, ..., 5.4006e-01,
2.3506e+00, 2.0386e+00],
[ 1.1789e+00, -2.1234e+00, 1.0697e+00, ..., 3.1761e+00,
1.3763e+00, 1.6518e+00]],
[[-4.2935e-01, 7.3965e-01, -1.7627e-01, ..., -2.9789e-01,
-5.1696e-01, 3.7847e-01],
[-9.6860e-01, -2.1683e+00, -3.1781e+00, ..., -7.8859e-01,
-1.0865e+00, -8.3600e-01],
[-2.4325e+00, -7.1217e-01, -1.7452e+00, ..., -3.8545e+00,
-3.3043e+00, 7.9787e-02],
...,
[-1.8297e+00, -4.6569e+00, -5.5931e+00, ..., 1.8568e+00,
-4.8863e+00, -3.0036e-01],
[ 2.0702e-01, -2.2419e+00, -6.2475e+00, ..., -8.1571e+00,
-2.1422e+00, -4.9252e-01],
[ 2.4970e-01, 1.7635e+00, 3.2143e+00, ..., -6.9174e-01,
-4.4313e-01, 1.2336e-01]]],
[[[ 8.8108e-01, 1.0960e+00, 6.3607e-01, ..., -4.3838e-01,
4.3960e-01, -5.3164e-01],
[-8.2009e-01, -1.3342e+00, -2.8491e-01, ..., 9.1264e-01,
1.5865e+00, -1.8258e-01],
[-1.3122e-01, -3.2242e-01, -4.4192e-03, ..., 2.2728e-01,
7.8552e-01, 8.7704e-01],
...,
[ 3.0530e-01, -6.3961e-01, -4.7644e-01, ..., -1.6865e+00,
-1.5331e+00, 1.3910e-01],
[-6.2835e-01, -1.8800e+00, -1.2092e+00, ..., -3.8561e-01,
-3.6490e-01, 1.4385e-01],
[-3.1841e-01, -1.7789e+00, -1.6431e+00, ..., -1.2086e+00,
-1.5479e+00, -2.9632e-01]],
[[-3.7391e+00, -1.3962e+00, -2.6892e+00, ..., -2.0525e+00,
-3.8242e+00, -9.1113e-01],
[-3.5357e+00, -2.0709e+00, -1.3563e+00, ..., -1.0742e+00,
-2.9940e+00, -1.7860e-01],
[-2.2107e+00, -1.3188e+00, -1.6430e+00, ..., -2.5053e+00,
-2.0746e+00, -2.6881e-01],
...,
[-1.4490e+00, -1.2766e+00, -1.9718e+00, ..., -5.8544e+00,
-4.2941e+00, -2.0451e+00],
[-3.0147e+00, -1.6411e+00, -2.8410e+00, ..., -3.2249e+00,
-2.4924e+00, -1.1853e+00],
[-2.3326e+00, -2.0423e+00, -2.7812e+00, ..., -2.0211e+00,
-1.9443e+00, -1.0391e+00]],
[[ 2.6122e+00, 2.8222e+00, 2.5120e+00, ..., 1.0982e+00,
8.3027e-01, 1.9652e+00],
[ 6.6583e-01, 2.0267e-01, 2.3386e+00, ..., -4.3357e-01,
-6.4964e-01, 1.8100e+00],
[ 1.3131e+00, 6.4041e-01, 1.7129e+00, ..., -1.2529e+00,
5.6857e-01, 2.2295e+00],
...,
[ 1.1737e+00, 1.4500e+00, 1.6060e+00, ..., -1.3836e+00,
-3.5673e-01, 1.8156e+00],
[ 6.2464e-01, 1.8273e+00, 1.1581e+00, ..., 1.2227e+00,
1.7494e+00, 1.8723e+00],
[ 1.5971e+00, 1.6409e+00, 1.8519e+00, ..., 1.3097e+00,
1.8122e+00, 1.7606e+00]],
...,
[[-2.6454e+00, -7.4022e+00, -5.2773e+00, ..., 1.5327e-01,
-1.6289e+00, -3.5636e+00],
[-3.1245e+00, -8.0745e+00, -5.3866e+00, ..., -3.9097e+00,
-4.5548e+00, -5.0287e+00],
[-1.6777e+00, -4.8890e+00, -2.7995e+00, ..., -2.7841e+00,
-1.7734e+00, -2.1853e+00],
...,
[-2.2240e+00, -4.9744e+00, -3.8287e+00, ..., -6.8024e+00,
-6.1805e+00, -4.8073e+00],
[-2.7249e+00, -5.7144e+00, -5.4266e+00, ..., -7.2650e+00,
-7.3685e+00, -7.9190e+00],
[-1.5109e+00, -4.0964e+00, -2.7987e+00, ..., -3.2444e+00,
-2.5940e+00, -3.1651e+00]],
[[-4.8937e-01, 3.4269e-01, 1.7050e-01, ..., 3.6918e-01,
4.0882e-01, 1.7316e+00],
[ 1.2049e+00, 6.3648e-01, 6.3293e-01, ..., 6.9009e-01,
9.4527e-01, 2.4110e+00],
[ 1.1944e+00, 1.7853e-01, 3.7010e-01, ..., 2.9110e-01,
9.9016e-01, 1.9000e+00],
...,
[ 1.2506e+00, 2.5824e-01, 1.1197e-01, ..., 3.7885e-01,
-6.2251e-01, 1.1756e+00],
[ 1.9131e+00, 1.6607e+00, 1.0528e+00, ..., 2.4289e+00,
1.7400e+00, 2.3173e+00],
[ 4.3819e-01, 7.5516e-01, 7.0139e-01, ..., 4.8962e-01,
6.6124e-01, 1.6376e+00]],
[[-6.3205e-01, -3.0932e-01, -5.3302e-01, ..., -1.8141e-01,
-1.3591e-01, 5.9380e-01],
[-1.3882e+00, -1.4190e+00, -8.5777e-01, ..., 5.5356e-01,
7.1739e-01, 7.0570e-01],
[-2.1675e+00, -2.7462e-01, 5.2565e-01, ..., -1.4173e+00,
-2.9348e-01, 5.9988e-03],
...,
[-1.8322e+00, -1.7951e+00, -1.2252e+00, ..., -3.2299e+00,
-2.9478e+00, -1.7660e+00],
[-1.5433e+00, -2.0407e+00, -1.7494e+00, ..., -1.8534e+00,
-1.5969e+00, -1.4218e-01],
[-8.4040e-01, -9.3905e-01, -1.1843e+00, ..., -7.1867e-01,
-9.3340e-01, -1.0889e-01]]],
[[[ 9.0920e-01, 1.4213e+00, 1.2529e+00, ..., -7.8027e-02,
6.1902e-02, -3.1537e-01],
[ 2.2738e-02, -1.0432e+00, -8.5349e-01, ..., -2.8166e-01,
4.9269e-01, -2.3182e-01],
[-1.4260e+00, -3.5583e+00, -4.4800e+00, ..., -4.6966e+00,
-2.1323e+00, 6.1954e-01],
...,
[-9.1835e-01, -2.4535e+00, 4.6786e-01, ..., 4.0526e-01,
8.8151e-01, 6.1391e-01],
[-7.3875e-01, -2.2910e+00, -9.4094e-01, ..., 9.2819e-01,
1.8302e+00, 1.0880e-01],
[ 1.1189e+00, -6.3056e-01, -7.5188e-01, ..., 3.6885e-01,
5.1460e-01, 8.7775e-01]],
[[-4.1305e+00, -3.9916e+00, -6.1052e+00, ..., -2.4346e+00,
-2.8765e+00, -1.4042e+00],
[-5.8044e+00, -6.9037e+00, -1.0222e+01, ..., -7.6023e-01,
-2.4221e+00, -1.3015e+00],
[-8.2402e+00, -7.1663e+00, -4.0871e+00, ..., -4.0011e+00,
-3.2034e+00, -8.4429e-01],
...,
[-3.1984e+00, -1.7628e+00, -1.1999e+00, ..., -1.6332e-01,
-1.7661e-01, 7.5598e-01],
[-4.3341e+00, 1.6233e+00, -3.7532e-01, ..., -2.6663e-02,
-1.7855e-01, 7.3134e-01],
[-1.9946e+00, -9.3409e-01, -1.5946e+00, ..., -6.1752e-01,
-9.4372e-01, -4.7293e-01]],
[[ 2.2786e+00, 2.5551e+00, 2.7093e+00, ..., 1.6600e+00,
8.5907e-01, 2.1258e+00],
[ 3.0241e-01, 2.2296e+00, 6.4547e+00, ..., 2.3936e+00,
2.4303e-01, 2.1734e+00],
[ 1.2152e+00, -7.3301e-01, -1.3853e+00, ..., 4.5660e+00,
3.4291e+00, 1.7071e+00],
...,
[ 1.3514e+00, 2.0622e+00, -5.6386e+00, ..., 2.7060e-01,
1.5934e-01, 1.6541e+00],
[ 1.2382e+00, -1.9176e+00, -5.1163e+00, ..., 1.0366e+00,
7.5749e-01, 1.8561e+00],
[ 8.7033e-01, -4.2219e+00, -1.8388e+00, ..., 1.9488e+00,
1.0366e+00, 1.9823e+00]],
...,
[[-2.4721e+00, -7.1238e+00, -6.7208e+00, ..., -2.6440e+00,
-2.0208e+00, -3.0264e+00],
[-3.8379e+00, -7.6231e+00, -4.2858e+00, ..., -6.2353e+00,
-4.6721e+00, -4.0353e+00],
[-2.0450e+00, -6.4812e+00, -7.9868e+00, ..., -6.2240e+00,
-5.1294e+00, -3.6495e+00],
...,
[-2.7678e+00, -5.8775e+00, -7.8111e+00, ..., -3.4487e+00,
-2.5037e+00, -2.2570e+00],
[-3.3449e+00, -7.1575e+00, -6.8252e+00, ..., -4.3916e+00,
-3.1041e+00, -4.7350e+00],
[-2.9180e+00, -6.1500e+00, -8.2289e+00, ..., -3.1199e+00,
-1.6961e+00, -3.5172e+00]],
[[ 7.5468e-02, 8.5649e-02, -1.3980e+00, ..., 7.8783e-01,
1.1601e+00, 1.6925e+00],
[-4.7413e-01, -4.3114e+00, -7.6320e+00, ..., 1.8184e+00,
8.1875e-01, 2.1254e+00],
[-3.7087e+00, -5.7372e+00, -2.0627e+00, ..., -1.1010e+00,
2.0527e+00, 1.6610e+00],
...,
[-2.2025e-01, -2.4515e+00, -1.4955e+00, ..., 1.5455e+00,
1.8743e+00, 1.5967e+00],
[ 1.9834e+00, 6.8887e-02, 5.9055e-01, ..., 1.8694e+00,
1.7736e+00, 2.7806e+00],
[ 1.4369e+00, -7.7596e-03, -1.0692e+00, ..., 1.3417e+00,
1.4915e+00, 1.4184e+00]],
[[-4.3634e-01, -7.9175e-01, -7.6274e-01, ..., -7.6657e-01,
-8.7400e-01, 1.6376e-01],
[-2.6314e+00, -3.0147e+00, -2.6998e+00, ..., -9.9542e-01,
-5.6162e-01, 2.4253e-01],
[-4.0877e+00, -1.9824e+00, -1.5315e+00, ..., -4.1355e+00,
-1.3053e+00, 5.0480e-01],
...,
[-2.5576e+00, -1.7286e+00, 6.9353e-01, ..., 6.1547e-02,
-2.4830e-01, -3.8958e-01],
[-2.0464e+00, 4.0826e-01, 2.7161e+00, ..., -1.5113e-01,
-2.2763e-01, 5.4285e-01],
[-1.0795e-01, 9.8160e-01, -1.0724e+00, ..., 2.9729e-01,
5.6763e-02, 5.6920e-01]]],
[[[ 4.4808e-01, 2.9384e-01, -2.5226e-01, ..., -5.8681e-01,
-1.3682e-01, 3.2237e-01],
[ 1.1269e+00, 1.6432e+00, 1.4702e+00, ..., -7.5765e-01,
5.0537e-01, -2.6320e+00],
[ 1.5776e-01, -5.2949e-01, -1.2005e+00, ..., 3.8024e+00,
4.0658e+00, 1.2695e+00],
...,
[ 5.5331e-01, -4.8418e+00, -2.7938e-01, ..., -8.9576e+00,
-7.8393e+00, 1.2188e+00],
[ 1.0186e-01, -4.4963e+00, 8.0036e-01, ..., -5.0306e+00,
-7.3144e+00, 5.4435e-03],
[ 8.6681e-01, -3.1138e+00, 1.3960e-01, ..., 1.5826e+00,
-1.9839e+00, 1.6437e+00]],
[[-2.3552e+00, -9.4501e-01, -1.6201e+00, ..., -2.1237e+00,
-1.4289e+00, 6.2569e-01],
[-1.5275e+00, 8.0274e-01, -1.3277e+00, ..., -1.5878e+00,
-4.8735e+00, -2.5340e+00],
[-7.9763e+00, -1.1828e+01, -1.5697e+01, ..., 2.2730e+00,
4.8971e-01, -8.5369e-01],
...,
[-6.7138e-01, -1.2003e+01, -1.9989e+01, ..., -3.2543e+01,
-2.4149e+01, -8.4273e+00],
[-2.8331e+00, -8.2211e+00, -1.2416e+01, ..., -3.3003e+01,
-2.4977e+01, -1.2231e+01],
[-2.2363e+00, -6.2340e+00, -4.3868e+00, ..., -3.1540e+01,
-2.9378e+01, -1.2716e+01]],
[[ 1.9093e+00, 1.5380e+00, 1.1927e+00, ..., -1.3924e+00,
2.3707e-01, 5.0793e-01],
[ 8.8154e-01, 1.7811e+00, 2.2045e+00, ..., -1.6417e+00,
-1.8288e+00, -9.8744e-01],
[-5.9946e-01, -3.0453e+00, -2.4769e+00, ..., -2.3104e+00,
-6.9992e-01, 1.0657e+00],
...,
[ 5.4429e-01, 5.3157e+00, -3.6256e-02, ..., -1.1571e+01,
-1.1857e+01, -8.8044e+00],
[ 1.0659e+00, 6.0936e+00, -1.6241e-01, ..., -1.0108e+01,
-1.4857e+01, -1.1960e+01],
[ 2.3633e+00, 4.9158e+00, 1.4775e+00, ..., -9.4414e+00,
-1.0342e+01, -1.1284e+01]],
...,
[[-1.3028e+00, -3.1962e+00, -1.0504e+00, ..., 2.0012e-01,
8.5199e-02, -1.3407e+00],
[-3.2509e+00, -8.4175e+00, -8.1129e+00, ..., -2.7382e+00,
-9.2924e-01, -8.0985e-01],
[-1.7968e+00, -7.7344e+00, -7.6461e+00, ..., -6.8154e+00,
-4.3250e+00, -4.2525e+00],
...,
[-2.2278e+00, -5.6729e+00, -1.0857e+01, ..., -6.0832e+00,
-4.3027e+00, -8.8774e+00],
[-2.2714e+00, -7.0701e+00, -1.1414e+01, ..., -9.3952e+00,
-7.8053e+00, -1.2694e+01],
[-2.6249e+00, -6.1456e+00, -5.2007e+00, ..., -1.6718e+01,
-1.1317e+01, -9.3983e+00]],
[[ 1.3504e-01, 3.9857e-01, 1.6433e-01, ..., 1.8290e+00,
8.6189e-01, 1.3211e+00],
[-1.7811e-01, -5.3360e-01, -1.2575e+00, ..., -2.9169e+00,
-2.1087e+00, 1.5000e+00],
[-3.6480e+00, -8.5091e+00, -1.3584e+01, ..., -1.4527e+00,
-1.4514e+00, -6.8494e-02],
...,
[ 1.4328e+00, -5.6011e+00, 4.6150e-01, ..., -1.4689e+01,
-1.6222e+01, -7.1837e+00],
[ 1.9450e+00, -2.8524e+00, 1.7709e+00, ..., -1.4285e+01,
-1.4532e+01, -8.2386e+00],
[ 1.0643e+00, -1.5902e+00, 1.9950e+00, ..., -5.4124e+00,
-7.2973e+00, -4.5927e+00]],
[[-4.3441e-01, -1.1296e+00, -1.1910e+00, ..., -3.8026e+00,
-1.7441e+00, 2.3816e-02],
[-9.1860e-01, -1.3389e+00, -6.7018e-01, ..., -3.3085e+00,
-1.1464e+00, 4.7171e-01],
[-1.7960e+00, -3.0122e+00, -3.1108e+00, ..., -3.2922e-01,
2.2189e+00, 1.5816e+00],
...,
[-1.4069e+00, -6.3635e+00, -2.8286e+00, ..., -2.3821e+01,
-8.8427e+00, 1.0980e+00],
[-1.4914e+00, -4.4878e+00, -1.5331e+00, ..., -1.7710e+01,
-1.3648e+01, 1.2749e+00],
[-2.5166e-01, -5.9411e-01, 7.3392e-01, ..., -7.0328e+00,
-3.8688e+00, 1.7451e+00]]],
[[[ 6.1937e-01, 7.2029e-01, -3.8979e-01, ..., 2.1564e-01,
4.7414e-01, 2.9794e-01],
[-1.7040e+00, -3.4168e+00, -5.7604e-01, ..., -1.8981e+00,
-2.6028e+00, -2.2044e+00],
[-2.8130e+00, -4.4349e+00, 1.1259e+00, ..., -4.0266e+00,
-1.8411e+00, 7.4488e-01],
...,
[-4.0637e+00, -5.0548e-01, 2.1883e+00, ..., -4.2975e-03,
6.4936e-01, 8.2739e-01],
[-1.0671e+00, -3.5506e+00, 1.3384e+00, ..., -1.0009e-01,
-1.1262e-01, -1.3356e-01],
[ 1.6109e+00, -1.3322e+00, -1.7256e+00, ..., -3.7043e-01,
-9.6084e-01, 5.7740e-02]],
[[-5.2431e+00, -3.7986e+00, -3.1055e+00, ..., -3.1305e+00,
-1.7712e+00, -1.6806e+00],
[-5.2681e+00, -7.1269e+00, -4.8314e+00, ..., -4.7950e+00,
-1.6644e+00, -2.0157e+00],
[-4.2830e+00, -7.7217e+00, -4.1027e+00, ..., -6.5617e+00,
2.0699e+00, -1.1096e+00],
...,
[-4.8368e+00, -5.7265e+00, -1.4303e+01, ..., -8.0155e-01,
-3.5431e-01, -3.6115e-01],
[-4.5150e+00, -2.6820e+00, -7.5486e+00, ..., -9.5205e-01,
-8.8420e-01, -9.7322e-01],
[-2.6553e+00, -3.6728e+00, -6.9703e+00, ..., -6.9171e-01,
-5.5871e-01, -3.9634e-01]],
[[ 2.7952e+00, 3.1298e+00, 3.8834e+00, ..., 1.8347e+00,
1.9875e+00, 2.8474e+00],
[ 8.4549e-01, 1.2905e+00, 3.3583e+00, ..., 3.4295e+00,
1.0693e+00, 1.4859e+00],
[-1.2522e+00, -1.7713e+00, -3.8801e+00, ..., 1.5630e+00,
-1.2322e+00, -9.9854e-01],
...,
[-1.5629e+01, -3.5093e+00, -6.1742e+00, ..., -3.4391e-01,
1.1948e+00, 2.1600e+00],
[-6.3151e+00, -3.0891e+00, -6.3623e+00, ..., -4.5536e-01,
1.0725e+00, 1.4572e+00],
[ 1.1797e+00, -1.8061e+00, -2.6163e+00, ..., 4.4080e-01,
1.4172e+00, 1.4450e+00]],
...,
[[-2.6720e+00, -8.2901e+00, -5.8044e+00, ..., -6.9733e+00,
-5.2346e+00, -4.4982e+00],
[-2.0646e+00, -6.9645e+00, -7.3057e+00, ..., -1.0433e+01,
-8.1377e+00, -3.4087e+00],
[-2.1248e+00, -5.5994e+00, -6.2249e+00, ..., -9.7154e+00,
-9.2743e+00, -6.8955e+00],
...,
[-2.9828e+00, -1.4260e+01, -6.7957e+00, ..., -5.5192e+00,
-3.7004e+00, -3.4118e+00],
[-2.5472e+00, -9.9241e+00, -9.6608e+00, ..., -6.4838e+00,
-4.6353e+00, -4.5927e+00],
[-2.2175e+00, -6.3409e+00, -4.5383e+00, ..., -4.1148e+00,
-2.7707e+00, -2.6761e+00]],
[[-1.8090e+00, -1.2752e+00, 1.5034e-01, ..., 1.7365e+00,
1.6309e+00, 1.0261e+00],
[-1.7091e+00, -7.7451e+00, -3.6681e+00, ..., -1.8332e+00,
-1.2725e-01, 1.4807e+00],
[-3.0196e-01, -3.2746e+00, -6.8762e+00, ..., -3.2405e-01,
2.0086e+00, 2.2451e+00],
...,
[ 1.0135e+00, -3.1296e+00, -5.5184e+00, ..., 2.2832e+00,
9.5769e-01, 1.2571e+00],
[ 2.9335e+00, 9.7571e-01, -1.2879e+00, ..., 1.8635e+00,
9.0873e-01, 1.7723e+00],
[-3.9551e-01, -6.4683e-01, -1.3319e+00, ..., 9.3353e-01,
7.3818e-01, 1.5430e+00]],
[[-9.8003e-01, -4.9411e-01, -2.2090e-01, ..., 1.7529e-01,
-6.2390e-01, -5.2190e-01],
[-2.4331e+00, -2.5846e+00, -3.3872e-03, ..., -3.2473e+00,
-3.8019e+00, -2.1850e+00],
[-8.7697e-01, -3.2231e+00, 3.3905e+00, ..., -5.9826e-01,
1.9602e+00, -1.1941e+00],
...,
[-6.3923e-01, -2.6836e+00, -5.9636e+00, ..., -6.3892e-01,
-1.1120e+00, -7.9251e-01],
[-2.3717e+00, -2.4714e+00, -2.1120e+00, ..., -8.6406e-01,
-1.3737e+00, -5.6532e-01],
[-5.9793e-01, 4.4780e-01, -2.0188e+00, ..., -3.9087e-01,
-7.1197e-01, 5.1210e-02]]]], device='cuda:0'),
tensor([[[[ 7.1901e-01, 1.0357e+00, 6.9581e-01, ..., 5.3624e-01,
1.9877e-01, 2.1341e-01],
[-3.3341e-01, -1.4511e+00, -5.4427e-01, ..., 2.4547e-01,
-3.8137e-01, -2.3643e+00],
[ 9.2059e-02, -8.7071e-01, -4.4875e-01, ..., -1.4211e+00,
1.0482e-01, 3.6205e-01],
...,
[ 4.2638e-01, -3.3229e+00, -4.6068e+00, ..., 1.9108e-01,
1.1558e+00, 9.5002e-01],
[-2.4026e+00, -2.1438e+00, -2.7971e+00, ..., 6.7222e-01,
1.2785e+00, -8.3092e-02],
[-7.6081e-01, 2.6116e-01, -3.7755e+00, ..., -2.8487e-01,
3.9307e-01, 6.1138e-01]],
[[-3.1765e+00, -1.5066e+00, -3.1342e+00, ..., -2.4106e+00,
-2.4264e+00, -1.1549e+00],
[-4.4237e+00, -1.8438e+00, -2.4378e+00, ..., -2.2328e+00,
-1.9774e+00, -7.1564e-01],
[-4.3087e+00, -2.4771e+00, -3.1055e+00, ..., -4.3146e+00,
-1.0627e+00, -1.6324e-02],
...,
[-2.7919e+00, -2.1211e+00, -2.5479e+00, ..., -7.9613e-01,
1.1429e-01, 4.3354e-01],
[-1.3696e+00, -9.7589e-01, 2.2283e-01, ..., 7.9118e-02,
-5.4772e-01, 1.1588e-01],
[-1.4159e+00, -2.8213e+00, -1.7342e-01, ..., -2.9712e-01,
-7.4812e-01, -5.6166e-01]],
[[ 1.8045e+00, 2.8812e+00, 2.4262e+00, ..., 2.9337e+00,
1.8370e+00, 3.5034e+00],
[ 2.7331e-01, 1.2647e+00, 1.6891e+00, ..., 1.8854e+00,
-1.4844e-01, 2.0531e+00],
[ 8.2148e-01, 1.3990e+00, 9.4809e-01, ..., 1.0692e+00,
-1.2402e+00, 5.2897e-01],
...,
[ 1.1213e+00, -7.6369e+00, -3.4891e+00, ..., -8.3274e-02,
9.2980e-01, 2.0122e+00],
[ 9.3882e-01, -5.1183e+00, -3.8894e+00, ..., 9.5921e-01,
5.5623e-01, 1.4498e+00],
[ 3.3305e+00, -2.1696e+00, -4.8676e+00, ..., 1.0992e+00,
1.0064e+00, 1.9479e+00]],
...,
[[-1.5016e+00, -4.7032e+00, -3.9308e+00, ..., -3.8771e+00,
-3.1733e+00, -3.8809e+00],
[-1.9358e+00, -4.1719e+00, -4.4431e+00, ..., -7.0433e+00,
-4.8177e+00, -3.5396e+00],
[-1.0960e+00, -3.4949e+00, -3.9292e+00, ..., -3.9650e+00,
-2.4824e+00, -1.2589e+00],
...,
[-2.4922e+00, -4.6732e+00, -7.2441e+00, ..., -4.8636e+00,
-3.4335e+00, -2.5012e+00],
[-1.3773e+00, -6.2281e+00, -7.2207e+00, ..., -7.4917e+00,
-3.6890e+00, -3.4622e+00],
[-1.0670e+00, -4.0085e+00, -5.7814e+00, ..., -5.3714e+00,
-1.9987e+00, -2.5435e+00]],
[[ 7.2317e-01, 5.5574e-01, -7.9905e-02, ..., 6.7709e-01,
1.0426e+00, 1.3254e+00],
[ 1.2057e+00, 9.7650e-01, 5.2590e-01, ..., 6.6805e-01,
1.3072e+00, 2.1467e+00],
[ 6.1175e-01, 5.5417e-01, 1.0280e-01, ..., -3.0838e-01,
2.7380e-01, 1.1730e+00],
...,
[-4.1619e-01, -2.7678e+00, -3.1769e+00, ..., 2.0509e+00,
2.2520e+00, 2.0445e+00],
[ 3.5651e-01, 1.3384e+00, -1.9944e+00, ..., 1.7729e+00,
1.5030e+00, 2.4525e+00],
[-1.1652e+00, 1.6757e+00, 2.2932e+00, ..., 9.9876e-01,
1.3932e+00, 1.3207e+00]],
[[-3.0375e-01, -1.8158e-01, -9.7580e-01, ..., -7.8555e-02,
8.3233e-02, 8.1826e-01],
[-1.4429e+00, -1.3650e+00, -1.9479e+00, ..., -3.9144e-02,
5.5276e-01, 3.2311e-02],
[-1.7747e+00, -7.4216e-01, -9.8172e-01, ..., -5.5684e-01,
8.9591e-01, 1.3720e+00],
...,
[-2.8580e+00, 3.0853e-01, -7.6414e-01, ..., -1.6801e+00,
-2.9486e-01, -5.8931e-02],
[-1.4239e+00, -2.2782e+00, -1.8528e+00, ..., -4.2723e-01,
-2.0843e-01, 3.5617e-01],
[ 4.2216e-01, -6.1668e-01, -1.4134e+00, ..., 4.2497e-01,
1.2306e-01, 3.4032e-01]]],
[[[ 9.2255e-01, 1.5898e+00, 1.1280e+00, ..., 8.6532e-01,
1.0137e+00, 4.4853e-01],
[-8.3704e-01, -1.8056e+00, -8.2935e-01, ..., -3.0760e-01,
-1.4202e-01, -3.2751e-01],
[ 7.6448e-02, -5.7123e-01, 4.9547e-02, ..., -5.5897e-01,
-1.7885e-01, 6.5960e-01],
...,
[ 5.2754e-01, -1.8333e-01, 1.1971e-01, ..., -3.3964e-02,
-6.6809e-02, 4.0318e-01],
[-1.2050e-01, -1.1940e+00, -5.4785e-02, ..., 3.1464e-02,
1.0385e-01, -8.9676e-02],
[ 8.4943e-01, -3.7874e-01, -1.2074e-01, ..., -3.1942e-01,
-9.1980e-01, 1.4841e-01]],
[[-3.7703e+00, -1.3589e+00, -2.0467e+00, ..., -1.4251e+00,
-2.0329e+00, -1.7826e+00],
[-3.8907e+00, -3.1012e+00, -2.0337e+00, ..., -1.2119e+00,
-2.1645e+00, -2.1233e+00],
[-2.4484e+00, -1.9040e+00, -2.3441e+00, ..., -1.3027e+00,
-1.7313e+00, -1.3981e+00],
...,
[-8.9849e-01, -3.3370e-01, -7.8189e-01, ..., -8.1232e-01,
-8.4170e-01, -8.9023e-01],
[-3.2134e+00, -1.1125e-01, -4.8921e-01, ..., -7.3317e-01,
-1.0763e+00, -1.2208e+00],
[-2.3660e+00, -1.5748e+00, -1.2012e+00, ..., -7.2056e-01,
-7.5767e-01, -5.3784e-01]],
[[ 2.6068e+00, 3.7414e+00, 2.9818e+00, ..., 1.6682e+00,
8.6393e-01, 2.5678e+00],
[-1.1854e-01, 7.5923e-01, 2.0982e+00, ..., 1.4400e+00,
7.5606e-01, 2.6266e+00],
[ 5.7936e-01, 1.3551e+00, 1.3869e+00, ..., 1.7219e-01,
8.6979e-01, 1.8053e+00],
...,
[ 7.2361e-01, 1.1177e+00, 1.2057e+00, ..., -2.6898e-01,
4.9433e-01, 1.5035e+00],
[ 7.1287e-01, 2.4315e+00, 1.6674e+00, ..., -1.8027e-01,
7.5527e-01, 1.1889e+00],
[ 1.7495e+00, 1.3397e+00, 1.8036e+00, ..., 5.5905e-01,
1.3328e+00, 1.3675e+00]],
...,
[[-1.5287e+00, -6.7279e+00, -5.7522e+00, ..., -5.0388e+00,
-3.5830e+00, -4.2411e+00],
[-2.2181e+00, -5.9575e+00, -4.3735e+00, ..., -4.8850e+00,
-3.5458e+00, -3.1810e+00],
[-1.9201e-01, -2.4564e+00, -2.9586e+00, ..., -3.1954e+00,
-2.2685e+00, -1.3632e+00],
...,
[-1.1425e+00, -2.6404e+00, -2.0619e+00, ..., -4.0813e+00,
-2.9474e+00, -2.8121e+00],
[-1.6450e+00, -4.0991e+00, -3.4641e+00, ..., -5.2385e+00,
-4.1975e+00, -4.4339e+00],
[-2.6402e+00, -5.4732e+00, -2.9052e+00, ..., -3.3896e+00,
-2.4588e+00, -2.4542e+00]],
[[-5.8838e-01, -2.1555e-01, 2.3608e-01, ..., 1.4386e+00,
1.6902e+00, 1.2066e+00],
[ 1.6744e+00, 8.1982e-01, 1.5951e+00, ..., 1.5056e+00,
1.1799e+00, 1.8443e+00],
[ 1.1021e+00, 7.1046e-02, 9.5655e-01, ..., 1.2358e+00,
4.3848e-01, 1.2436e+00],
...,
[ 1.4268e+00, 3.4754e-02, 1.1769e+00, ..., 1.3760e+00,
7.1666e-01, 8.7047e-01],
[ 2.0144e+00, 1.3121e+00, 1.9139e+00, ..., 1.2727e+00,
7.5017e-01, 1.4179e+00],
[ 9.8800e-01, 6.0647e-01, 1.1638e+00, ..., 7.2806e-01,
6.6081e-01, 1.4268e+00]],
[[-4.8371e-01, 5.7464e-01, 1.7121e-01, ..., -3.0997e-01,
-5.8455e-01, -7.5826e-02],
[-1.0770e+00, -1.1565e+00, -1.1534e+00, ..., -1.3468e+00,
-1.3651e+00, -1.1586e+00],
[-1.9718e+00, -6.4139e-01, -1.8342e-02, ..., -8.6933e-01,
-1.0046e+00, -2.7479e-01],
...,
[-1.9774e+00, -1.5872e+00, -8.2852e-01, ..., -9.9505e-01,
-1.3843e+00, -9.0455e-01],
[-1.9471e+00, -1.1167e+00, -8.4123e-01, ..., -7.7776e-01,
-1.6251e+00, -7.8150e-01],
[-5.2830e-01, 2.3264e-01, -2.5075e-01, ..., -2.8734e-01,
-8.3427e-01, -7.7639e-03]]],
[[[-2.4306e-02, 1.0340e+00, -7.9905e-01, ..., 1.5824e+00,
1.1649e+00, 1.1093e+00],
[-3.8074e+00, -4.5473e+00, -2.1079e+00, ..., -9.3001e-01,
-1.1995e+00, -1.8793e+00],
[-3.8343e-01, -3.8511e+00, -4.9964e+00, ..., 1.1470e-01,
-9.5603e-01, 3.6764e-01],
...,
[ 6.2083e-01, -2.9088e+00, -2.9646e+00, ..., 8.3552e-01,
3.4724e-01, 6.4877e-01],
[-1.3280e+00, -4.3268e+00, -2.5851e-01, ..., 8.5011e-01,
7.4533e-01, -2.6822e-02],
[-1.4809e-02, -1.0577e+00, -1.0843e+00, ..., 2.5296e-01,
-3.5258e-02, 6.1797e-01]],
[[-8.1450e+00, -3.2401e+00, -4.4526e+00, ..., -1.4547e+00,
-2.3704e+00, -1.0837e+00],
[-4.5937e+00, 1.7315e+00, -1.3641e+00, ..., -2.3383e+00,
-3.4779e+00, -9.3378e-01],
[ 5.5790e-02, -1.6844e+00, -8.6306e+00, ..., -5.6174e+00,
-5.1630e+00, -1.2028e+00],
...,
[-3.4159e+00, -7.9009e+00, -4.7507e+00, ..., -1.1174e+00,
-1.1024e+00, 1.9952e-02],
[-3.5034e+00, -3.0521e+00, 2.7515e+00, ..., -3.0231e-01,
-7.7498e-01, 1.8981e-01],
[-1.0413e+00, -1.8852e+00, 1.7796e+00, ..., 2.4037e-01,
1.5867e-01, 4.6935e-01]],
[[ 4.1011e+00, 2.9569e+00, 2.2185e+00, ..., 2.9817e+00,
1.0166e+00, 3.0751e+00],
[-1.8258e+00, -2.4823e+00, -2.7003e+00, ..., 2.2395e+00,
-1.0129e-01, 2.2172e+00],
[-5.8186e+00, -4.6663e+00, -2.5769e+00, ..., -1.0703e+00,
1.0955e+00, 2.2972e+00],
...,
[ 1.8414e+00, 2.8685e+00, -3.7464e+00, ..., 1.6073e-01,
5.3982e-01, 1.4003e+00],
[ 2.2099e+00, 1.7180e+00, -1.0390e+01, ..., -1.6407e-01,
-5.8598e-02, 1.0563e+00],
[ 2.7354e+00, -1.0953e+00, -9.2330e+00, ..., 7.0458e-01,
3.1191e-01, 1.1570e+00]],
...,
[[-3.4790e+00, -9.7046e+00, -5.3652e+00, ..., -4.2097e+00,
-3.3482e+00, -5.7621e+00],
[-3.4338e+00, -1.1379e+01, -9.9117e+00, ..., -4.3123e+00,
-2.7440e+00, -5.2322e+00],
[-2.3236e+00, -9.0571e+00, -7.9501e+00, ..., -3.7972e+00,
-2.4569e+00, -4.3099e+00],
...,
[-2.7120e+00, -7.0061e+00, -1.0337e+01, ..., -2.8424e+00,
-2.9032e+00, -3.4054e+00],
[-2.2928e+00, -1.0171e+01, -1.2315e+01, ..., -5.0938e+00,
-4.1527e+00, -4.6935e+00],
[-2.6455e+00, -9.0327e+00, -6.0910e+00, ..., -4.0886e+00,
-2.9991e+00, -3.8569e+00]],
[[-4.2829e+00, -2.6526e+00, -3.0020e+00, ..., 6.2143e-01,
8.8095e-01, 1.4295e+00],
[ 3.7541e-01, 2.3333e+00, 2.3203e+00, ..., 7.6450e-01,
2.5384e+00, 3.0190e+00],
[-6.7679e-01, -1.0245e-02, 4.2429e-01, ..., -1.9306e+00,
1.3186e+00, 3.0431e+00],
...,
[-8.3008e-01, -6.3160e+00, -3.1522e+00, ..., 9.7791e-01,
8.9400e-01, 1.4000e+00],
[-4.8129e-01, -1.6868e+00, 8.0611e-01, ..., 1.0838e+00,
8.5922e-01, 2.0999e+00],
[ 1.1351e-01, -1.4469e+00, -4.2100e-01, ..., 1.0618e+00,
1.2609e+00, 1.3061e+00]],
[[-2.1159e+00, -7.6300e-01, -1.8020e+00, ..., 6.7957e-01,
6.3082e-01, 1.4137e+00],
[-2.2986e+00, 5.2470e-01, 5.6063e-01, ..., -7.3286e-01,
-1.2904e+00, 6.3535e-02],
[ 6.0180e-01, 2.2332e+00, -2.9963e+00, ..., 1.3430e+00,
-9.9548e-01, 6.0445e-01],
...,
[-2.5149e+00, -3.9338e+00, -1.9146e+00, ..., -6.4251e-01,
-7.3302e-01, -4.6523e-01],
[-2.4784e+00, -2.6432e+00, 4.5057e+00, ..., -5.4704e-01,
-1.0614e+00, 6.4196e-02],
[ 2.7754e-01, -6.8007e-01, 3.4284e+00, ..., 2.7237e-01,
-1.6655e-01, 7.1387e-01]]],
[[[ 9.0135e-01, 1.3176e+00, 2.1187e+00, ..., 4.8683e-01,
9.7607e-01, 1.1373e+00],
[-1.5214e+00, -5.7046e+00, -8.1865e+00, ..., -2.4219e+00,
-1.1654e+00, -1.7894e+00],
[ 4.6678e-01, -2.8658e+00, -3.9345e+00, ..., 2.5760e+00,
1.2866e+00, 5.6096e-01],
...,
[-1.4856e-01, -1.0610e+00, 6.8993e-01, ..., -1.3475e-01,
2.1905e-01, 3.5010e-01],
[-6.4506e-01, -1.4923e+00, -1.6025e+00, ..., 2.4688e-01,
2.5721e-01, -6.7815e-02],
[ 2.0584e-01, -8.9916e-01, -6.8854e-01, ..., -1.4018e+00,
-1.4484e+00, -2.0471e-01]],
[[-3.8612e+00, -1.6426e+00, -5.8212e+00, ..., -1.2144e+00,
-2.3671e+00, -1.1904e+00],
[-3.6733e+00, -2.8003e+00, -5.4303e+00, ..., -4.5988e+00,
-3.7991e+00, -1.1206e+00],
[-3.6063e+00, -3.5126e+00, -4.3570e+00, ..., -2.7199e+00,
-2.9177e+00, -1.3116e+00],
...,
[-1.8908e+00, -1.8799e+00, -1.5587e+00, ..., -3.7255e+00,
-3.3597e+00, -9.4339e-01],
[-2.4721e+00, 1.0471e+00, -5.3435e-01, ..., -2.4364e+00,
-2.7003e+00, -1.2274e+00],
[-1.8981e+00, -2.4792e-01, -1.2810e+00, ..., -2.9207e+00,
-2.4144e+00, -1.1140e+00]],
[[ 2.2124e+00, 2.7277e+00, 1.0611e+00, ..., 3.4516e+00,
1.0819e+00, 3.1465e+00],
[-1.7219e+00, -5.0708e-01, -1.1814e+00, ..., 2.5454e+00,
-1.0093e-01, 2.8619e+00],
[-2.0152e+00, -1.8533e+00, -3.3269e+00, ..., -4.8895e-01,
7.8521e-01, 2.7665e+00],
...,
[ 1.3303e+00, 2.1956e+00, 1.2127e+00, ..., -1.7984e+00,
6.9239e-01, 2.4165e+00],
[ 1.2164e+00, 1.7154e+00, -8.0449e-01, ..., 2.7848e-01,
1.0331e+00, 1.7749e+00],
[ 2.1828e+00, -2.5280e-02, -6.6812e-01, ..., 9.6589e-01,
1.9533e+00, 1.9174e+00]],
...,
[[-1.6929e+00, -7.1360e+00, -8.3520e+00, ..., -4.3219e+00,
-3.5557e+00, -5.9838e+00],
[-3.6694e+00, -7.5888e+00, -5.6334e+00, ..., -6.3879e+00,
-2.4746e+00, -4.1559e+00],
[-2.5549e-01, -3.6426e+00, -1.8472e+00, ..., -7.4693e+00,
-3.1873e+00, -2.1452e+00],
...,
[-2.8570e+00, -6.4116e+00, -6.7812e+00, ..., -4.3246e+00,
-3.0533e+00, -3.8144e+00],
[-2.9575e+00, -5.2817e+00, -2.6121e+00, ..., -6.8601e+00,
-5.4831e+00, -5.7594e+00],
[-2.0975e+00, -4.8200e+00, -1.2318e+00, ..., -1.9213e+00,
-1.6258e+00, -2.6489e+00]],
[[-2.3895e-01, -3.5889e-01, -4.5062e+00, ..., 1.5240e+00,
1.0701e+00, 1.4405e+00],
[ 1.5820e+00, 1.8766e+00, 2.6953e+00, ..., 2.8989e+00,
2.7826e+00, 3.0822e+00],
[ 2.9743e-01, -3.8941e+00, -4.6134e+00, ..., 1.1708e+00,
7.6341e-01, 2.3127e+00],
...,
[ 1.1757e+00, 1.8609e+00, 2.5015e+00, ..., 4.3425e-01,
1.4846e+00, 1.9366e+00],
[ 2.2216e+00, 2.6511e+00, 1.1132e+00, ..., 1.2938e+00,
1.3153e+00, 2.0730e+00],
[ 5.2763e-01, 1.0472e+00, 6.6059e-01, ..., 2.8923e-01,
5.9080e-01, 1.6608e+00]],
[[-8.1971e-01, 4.3519e-01, -1.0104e+00, ..., 7.3439e-01,
4.0480e-01, 1.4025e+00],
[-1.0074e+00, 3.7044e-02, -4.2877e+00, ..., -1.2133e+00,
-4.5877e-01, 7.6703e-03],
[-3.1037e+00, -1.4390e+00, -5.6199e+00, ..., -5.7355e-01,
-1.7376e-01, 4.2809e-01],
...,
[-2.4714e+00, -2.7017e+00, -1.8569e+00, ..., -2.0792e+00,
-1.6201e+00, -6.0915e-01],
[-1.7567e+00, -1.2275e+00, -1.1979e+00, ..., 4.9518e-02,
-6.0726e-01, -3.0256e-01],
[-5.9665e-01, -4.3425e-01, -9.2030e-01, ..., -1.3649e+00,
-1.3142e+00, -1.7691e-01]]],
[[[ 7.0954e-01, 9.4069e-01, 4.8003e-01, ..., 5.8493e-01,
1.0309e+00, 4.7482e-01],
[-1.3110e+00, -1.1977e+00, -2.5073e-01, ..., -4.8342e-01,
-5.6134e-02, -5.1988e-01],
[-3.8435e-01, -9.4385e-01, -1.4623e-01, ..., -1.0892e+00,
-2.3163e-02, 3.3617e-01],
...,
[-2.4355e-01, -8.5109e-01, -3.3810e-01, ..., 2.7850e-01,
4.6074e-01, 5.4279e-01],
[ 9.0633e-02, -8.5760e-01, -3.1514e-01, ..., 5.5758e-02,
5.3806e-01, 8.8631e-02],
[ 5.2441e-01, -1.1883e-01, -8.2828e-01, ..., -3.6803e-01,
-7.3297e-01, 1.9154e-01]],
[[-4.2818e+00, -2.1314e+00, -3.1149e+00, ..., -1.4228e+00,
-1.5256e+00, -1.3929e+00],
[-3.4190e+00, -1.8024e+00, -9.0876e-01, ..., -1.6918e+00,
-1.2990e+00, -1.3336e+00],
[-2.1250e+00, -1.3980e+00, -1.3626e+00, ..., -1.0822e+00,
3.8934e-01, -8.4353e-02],
...,
[-1.4667e+00, -5.4204e-01, -1.0826e+00, ..., -7.9032e-01,
-1.7131e-01, -4.7715e-01],
[-1.4824e+00, -4.2110e-01, -6.1305e-01, ..., -6.0195e-01,
-5.1292e-01, -8.5191e-01],
[ 7.6101e-02, -8.3100e-01, -8.8949e-01, ..., -5.8726e-01,
-4.3419e-01, -5.0081e-01]],
[[ 2.8013e+00, 3.5110e+00, 3.4123e+00, ..., 1.2849e+00,
9.4933e-01, 2.0532e+00],
[ 7.3224e-01, 1.1737e+00, 2.8310e+00, ..., 7.2409e-01,
-6.2930e-02, 1.3662e+00],
[ 9.6519e-01, 5.6872e-01, 1.8432e+00, ..., 1.2909e-01,
-4.1749e-01, 2.2074e-01],
...,
[ 3.8896e-01, 5.7276e-01, 8.1719e-01, ..., -7.7291e-01,
4.4904e-01, 1.4156e+00],
[ 1.0700e-01, 1.2939e+00, 1.4067e+00, ..., -6.6281e-01,
8.4206e-01, 1.3048e+00],
[ 1.3897e+00, 8.9627e-01, 1.7764e+00, ..., 3.3156e-01,
1.4292e+00, 1.4958e+00]],
...,
[[-2.8411e+00, -8.2106e+00, -5.3755e+00, ..., -5.2612e+00,
-3.0947e+00, -2.9925e+00],
[-3.3396e+00, -9.4753e+00, -6.4378e+00, ..., -5.7568e+00,
-2.7536e+00, -1.2805e+00],
[-2.2268e+00, -6.4114e+00, -3.7926e+00, ..., -4.8262e+00,
-1.6403e+00, -7.5786e-01],
...,
[-2.5663e+00, -6.0037e+00, -3.8739e+00, ..., -5.7622e+00,
-2.7655e+00, -2.4113e+00],
[-2.8092e+00, -7.3913e+00, -4.8739e+00, ..., -6.1399e+00,
-3.9973e+00, -4.3634e+00],
[-2.3123e+00, -6.5309e+00, -3.6924e+00, ..., -3.6802e+00,
-2.3578e+00, -2.3981e+00]],
[[-9.0890e-01, -1.4688e-01, 1.7738e-01, ..., 1.6840e+00,
1.4797e+00, 9.5826e-01],
[ 1.4043e+00, 9.7036e-01, 1.2434e+00, ..., 1.0919e+00,
5.6795e-01, 1.4546e+00],
[ 1.2008e+00, 3.7893e-01, 3.6593e-01, ..., 1.0640e+00,
6.9902e-01, 7.2759e-01],
...,
[ 1.0187e+00, 2.3515e-01, 4.9122e-01, ..., 2.0967e+00,
8.5226e-01, 8.5973e-01],
[ 1.7152e+00, 1.2036e+00, 1.1877e+00, ..., 1.7219e+00,
8.1743e-01, 1.4755e+00],
[ 9.8733e-01, 7.2691e-01, 1.0702e+00, ..., 8.4820e-01,
6.8468e-01, 1.4571e+00]],
[[-7.3555e-01, -8.0493e-01, -7.4332e-01, ..., 3.8111e-02,
-3.1914e-01, -2.3058e-01],
[-1.4065e+00, -1.4623e+00, -8.8483e-01, ..., -1.8269e+00,
-1.2375e+00, -1.4323e+00],
[-2.2207e+00, -6.7964e-01, 1.3138e-01, ..., -1.4482e+00,
2.2464e-01, -5.2709e-01],
...,
[-2.1624e+00, -1.5224e+00, -5.8375e-01, ..., -1.2843e+00,
-1.1264e+00, -8.6030e-01],
[-1.5395e+00, -1.5617e+00, -7.9804e-01, ..., -1.2043e+00,
-1.3956e+00, -7.0574e-01],
[ 1.3859e-01, -5.0536e-02, -2.1593e-01, ..., -4.0964e-01,
-6.8436e-01, 3.0810e-02]]]], device='cuda:0'),
tensor([[[[ 9.8025e-01, 1.5773e+00, 1.4459e+00, ..., 8.0790e-01,
1.0091e+00, 4.3169e-01],
[-9.1375e-01, -2.1617e+00, -4.0767e-01, ..., -1.4003e-01,
-2.6820e-02, -3.3454e-01],
[ 5.4208e-03, -9.3841e-01, -5.0837e-01, ..., -5.2126e-01,
-1.0385e-01, 6.4317e-01],
...,
[ 4.6559e-01, -2.8907e-01, -2.3008e-01, ..., 5.9025e-02,
-5.8566e-02, 4.0969e-01],
[-1.7081e-01, -9.4868e-01, -8.1955e-03, ..., -3.1024e-03,
1.2704e-01, -9.5590e-02],
[ 7.2237e-01, -2.8021e-01, -2.0188e-01, ..., -3.3588e-01,
-9.2303e-01, 1.7139e-01]],
[[-4.2329e+00, -1.3166e+00, -2.3995e+00, ..., -1.4233e+00,
-2.0787e+00, -1.8343e+00],
[-4.0714e+00, -2.8468e+00, -1.6444e+00, ..., -9.0043e-01,
-2.0050e+00, -2.1168e+00],
[-2.6666e+00, -1.8222e+00, -3.1568e+00, ..., -1.3108e+00,
-1.4717e+00, -1.3915e+00],
...,
[-7.4996e-01, -5.8114e-01, -2.1509e+00, ..., -6.4877e-01,
-8.4509e-01, -8.4379e-01],
[-2.7202e+00, -4.3345e-01, -9.3906e-01, ..., -6.1866e-01,
-1.0458e+00, -1.1960e+00],
[-1.9822e+00, -1.4780e+00, -1.4117e+00, ..., -6.9049e-01,
-7.3229e-01, -5.2020e-01]],
[[ 3.0347e+00, 4.2787e+00, 2.8785e+00, ..., 1.7409e+00,
8.7618e-01, 2.6142e+00],
[ 3.6769e-01, 9.1860e-01, 2.8002e+00, ..., 1.4877e+00,
7.3446e-01, 2.6229e+00],
[ 8.8814e-01, 1.3005e+00, 2.5225e+00, ..., 5.0913e-01,
9.5889e-01, 1.8012e+00],
...,
[ 2.6623e-01, 8.9658e-01, 1.6158e+00, ..., -2.5311e-01,
5.4159e-01, 1.4937e+00],
[ 2.5422e-01, 2.0128e+00, 1.3362e+00, ..., -9.7632e-02,
7.5210e-01, 1.1499e+00],
[ 1.3674e+00, 1.3483e+00, 1.6891e+00, ..., 5.7311e-01,
1.2801e+00, 1.3508e+00]],
...,
[[-2.1793e+00, -7.5713e+00, -6.5379e+00, ..., -4.5862e+00,
-3.3918e+00, -4.2470e+00],
[-2.8534e+00, -6.6338e+00, -5.8357e+00, ..., -4.5377e+00,
-3.4919e+00, -3.3074e+00],
[-2.5975e-01, -2.8132e+00, -3.7615e+00, ..., -2.9394e+00,
-1.9987e+00, -1.3980e+00],
...,
[-9.3870e-01, -2.3083e+00, -2.2617e+00, ..., -4.3054e+00,
-3.1722e+00, -2.7162e+00],
[-1.2934e+00, -3.7237e+00, -4.0646e+00, ..., -5.1805e+00,
-4.3222e+00, -4.3074e+00],
[-2.1554e+00, -4.8814e+00, -3.1104e+00, ..., -3.2972e+00,
-2.5167e+00, -2.4005e+00]],
[[-7.4583e-01, -1.8256e-01, -1.1212e+00, ..., 1.4296e+00,
1.5624e+00, 1.1977e+00],
[ 1.8199e+00, 1.1204e+00, -1.7560e-01, ..., 1.5410e+00,
1.1441e+00, 1.8441e+00],
[ 1.2299e+00, 1.9600e-01, -7.3719e-01, ..., 1.2790e+00,
5.7779e-01, 1.2103e+00],
...,
[ 1.2876e+00, -1.3795e-01, 3.8959e-01, ..., 1.5714e+00,
7.1876e-01, 9.0120e-01],
[ 1.8830e+00, 9.1436e-01, 1.6935e+00, ..., 1.3172e+00,
7.2139e-01, 1.4252e+00],
[ 1.0255e+00, 3.7398e-01, 9.4185e-01, ..., 7.5407e-01,
6.5142e-01, 1.4148e+00]],
[[-5.0007e-01, 5.0967e-01, -8.7900e-01, ..., -2.9845e-01,
-5.8841e-01, -9.0294e-02],
[-1.1013e+00, -1.2456e+00, -2.1646e+00, ..., -1.1288e+00,
-1.2761e+00, -1.1441e+00],
[-2.0730e+00, -5.9915e-01, -1.7356e+00, ..., -7.5371e-01,
-9.5724e-01, -3.1613e-01],
...,
[-1.6878e+00, -1.4846e+00, -1.2865e+00, ..., -9.4493e-01,
-1.5147e+00, -8.6117e-01],
[-1.7098e+00, -1.1473e+00, -7.5647e-01, ..., -5.9016e-01,
-1.6320e+00, -7.2596e-01],
[-3.4061e-01, 3.6322e-01, -2.0730e-01, ..., -2.3981e-01,
-8.1955e-01, -9.1575e-03]]],
[[[ 6.4142e-01, 9.0569e-01, 4.6592e-01, ..., 8.8730e-01,
7.5563e-01, 5.1140e-01],
[-5.1280e-01, -1.4296e+00, -4.1429e-01, ..., 1.2539e-02,
8.8964e-02, -1.7160e+00],
[-6.9957e-02, -8.6322e-01, 1.9008e-01, ..., -9.3192e-01,
3.2647e-01, 7.8763e-01],
...,
[-1.1176e-01, -1.8630e+00, 1.7728e+00, ..., -9.8245e+00,
-7.8665e+00, -4.1985e+00],
[-3.2865e-01, -2.2130e+00, 7.6625e-02, ..., -3.1729e+00,
-4.0287e+00, 1.7677e-01],
[ 4.7614e-01, -6.2359e-01, -1.2351e+00, ..., -2.8723e+00,
-6.6201e+00, 3.2269e+00]],
[[-3.8177e+00, -1.6711e+00, -2.8923e+00, ..., -1.2975e+00,
-1.4971e+00, -6.4634e-01],
[-4.3907e+00, -1.8799e+00, -1.9203e+00, ..., -9.9488e-01,
-1.7514e+00, -3.7272e-01],
[-3.9938e+00, -2.6882e+00, -3.7022e+00, ..., -2.5383e+00,
-2.1026e+00, -1.2278e+00],
...,
[-2.6846e+00, -2.5508e+00, -5.3636e+00, ..., -1.6257e+01,
-1.8473e+01, -7.0953e+00],
[-3.3767e+00, -1.2949e+00, -2.4764e+00, ..., -1.5493e+01,
-1.0816e+01, -7.8813e+00],
[-8.3289e-01, -1.2306e+00, -2.2213e+00, ..., -1.6862e+01,
-2.0392e+01, -1.0384e+01]],
[[ 2.3582e+00, 3.0497e+00, 2.4639e+00, ..., 3.0355e+00,
1.3181e+00, 2.8295e+00],
[ 8.5476e-01, 1.2873e+00, 1.8060e+00, ..., 2.0508e+00,
-1.1584e-01, 2.4859e+00],
[ 1.3066e+00, 1.3050e+00, 8.1766e-01, ..., 2.1720e+00,
1.8543e+00, 2.1968e+00],
...,
[ 1.2793e+00, 2.9256e+00, 2.0999e+00, ..., 5.8030e-02,
-9.7448e+00, -7.0762e+00],
[ 9.8268e-01, 2.1419e+00, 3.6314e+00, ..., -3.6032e+00,
-6.8929e+00, -9.3301e+00],
[ 2.0193e+00, 1.2729e+00, 1.9849e+00, ..., -1.1319e+01,
-2.0699e+01, -1.7866e+01]],
...,
[[-1.7488e+00, -5.2526e+00, -4.0043e+00, ..., -3.4445e+00,
-2.5180e+00, -4.2431e+00],
[-2.3693e+00, -5.6769e+00, -4.9473e+00, ..., -4.5757e+00,
-3.1210e+00, -3.7794e+00],
[-1.3835e+00, -4.1637e+00, -4.1359e+00, ..., -3.4264e+00,
-3.5575e+00, -2.2688e+00],
...,
[-2.2099e+00, -5.6503e+00, -5.7774e+00, ..., 5.1970e+00,
1.2804e+00, 2.4012e-01],
[-2.2617e+00, -6.7028e+00, -6.0057e+00, ..., 2.5789e+00,
-2.3145e+00, -1.4376e+01],
[-2.0861e+00, -5.8654e+00, -4.0473e+00, ..., -6.0004e+00,
-1.2428e+01, -2.1970e+01]],
[[ 2.6725e-01, 4.5793e-01, 1.2631e-01, ..., 9.1450e-01,
1.4000e+00, 1.2870e+00],
[ 1.3050e+00, 9.7296e-01, 7.0876e-01, ..., 1.5303e+00,
2.4294e+00, 2.6799e+00],
[ 8.4034e-01, 3.4856e-01, 7.2963e-02, ..., 1.1749e+00,
1.7511e+00, 1.9130e+00],
...,
[ 1.0062e+00, -1.6223e+00, -2.4988e+00, ..., -2.7724e+01,
-1.3771e+01, -7.6015e+00],
[ 1.8552e+00, 1.4385e+00, 4.7952e-02, ..., -2.5061e+01,
-1.7597e+01, -6.3005e+00],
[ 8.1415e-01, 1.0072e+00, 8.6303e-01, ..., -7.2384e+00,
-5.6937e+00, -4.7180e+00]],
[[-5.3628e-01, -1.6755e-01, -7.5811e-01, ..., 8.7759e-02,
7.0624e-02, 1.0032e+00],
[-1.5767e+00, -1.1660e+00, -1.3637e+00, ..., -3.3895e-01,
-5.0788e-01, 1.2750e-01],
[-2.0369e+00, -1.1172e+00, -6.5580e-01, ..., 2.0028e-01,
1.9838e-03, -5.2240e-02],
...,
[-1.9682e+00, -1.3241e+00, 3.5397e-01, ..., -1.4784e+01,
-1.4051e+01, -2.1441e+00],
[-1.6832e+00, -1.7079e+00, -4.7894e-02, ..., -9.1860e+00,
-4.6862e+00, 2.7710e+00],
[-1.0300e-01, 8.3676e-03, -2.9504e-01, ..., -3.9599e+00,
-5.4668e+00, -8.4177e-01]]],
[[[ 1.2389e+00, 7.1671e-01, -4.9843e-01, ..., 1.4682e+00,
1.4416e+00, 1.4434e+00],
[-2.5117e+00, -5.3388e+00, -4.8676e+00, ..., -4.7135e-01,
-9.9214e-01, -2.1760e+00],
[-2.9623e+00, -4.0244e+00, -9.2529e+00, ..., -3.5024e+00,
-8.8113e-01, 8.0268e-01],
...,
[ 6.2624e-01, 2.1265e+00, -1.3497e+00, ..., 1.6792e+00,
1.4404e+00, 2.2772e-01],
[-6.2705e-01, -1.5435e+00, 4.3279e-01, ..., -1.0407e+00,
2.7390e-01, -9.0815e-03],
[-2.0507e-02, -1.4835e+00, -1.8591e+00, ..., -1.7879e+00,
-1.3230e+00, -1.1433e-01]],
[[-4.1769e+00, -8.8771e+00, -9.2634e+00, ..., -1.3691e+00,
-1.9778e+00, -1.0673e+00],
[-6.4204e+00, -1.0742e+01, -1.8512e+00, ..., -1.7551e+00,
-3.4694e+00, -1.2214e+00],
[-6.2454e+00, -1.0770e+01, -5.3845e+00, ..., -4.1579e+00,
-3.4909e+00, -1.4354e+00],
...,
[-2.0640e+00, -5.0312e+00, -5.8715e+00, ..., -6.5852e+00,
-1.7142e+00, -9.4939e-02],
[-2.6588e+00, -1.9496e+00, -2.0962e+00, ..., -4.3177e+00,
-1.6229e+00, -7.1185e-01],
[-1.8876e+00, -2.1565e+00, -3.0743e+00, ..., -2.3644e+00,
-1.6624e+00, -7.3461e-01]],
[[ 3.6957e+00, 6.0899e+00, 3.3418e+00, ..., 2.8139e+00,
1.1110e+00, 3.3500e+00],
[-5.3976e-01, 7.6144e-01, -7.2305e+00, ..., 2.7577e+00,
-1.1156e+00, 2.7617e+00],
[ 1.9425e+00, -2.1909e+00, -1.5881e+01, ..., 3.0122e+00,
2.6387e+00, 2.6292e+00],
...,
[ 3.2355e+00, 1.9079e+00, -4.2245e-02, ..., -5.3711e-01,
5.1969e-01, 1.3341e+00],
[ 7.0781e-01, 3.4024e+00, 3.8429e+00, ..., 9.9139e-01,
8.0666e-02, 1.6908e+00],
[ 1.9233e+00, 1.5069e+00, 1.7701e+00, ..., 4.6162e-01,
1.1401e+00, 1.6565e+00]],
...,
[[-2.3090e+00, -8.7878e+00, -6.8829e+00, ..., -5.3521e+00,
-3.8260e+00, -6.6182e+00],
[-1.5105e+00, -1.0402e+01, -1.0738e+01, ..., -4.8780e+00,
-3.3894e+00, -4.6020e+00],
[-1.6909e-01, -8.8440e+00, -9.2495e+00, ..., -4.1574e+00,
-3.5389e+00, -3.4536e+00],
...,
[-2.7559e+00, -9.8193e+00, -4.4280e+00, ..., -1.4150e+01,
-9.8443e+00, -6.6405e+00],
[-3.4358e+00, -8.1176e+00, -9.1991e+00, ..., -9.2321e+00,
-9.2542e+00, -5.7499e+00],
[-2.1101e+00, -5.5571e+00, -3.1247e+00, ..., -7.5105e+00,
-4.0665e+00, -3.0296e+00]],
[[-1.1890e+00, -5.3028e+00, -7.0282e+00, ..., 6.7393e-01,
1.1902e+00, 1.3875e+00],
[-3.3261e-01, -4.9236e+00, -5.6514e-01, ..., 2.6352e+00,
2.4030e+00, 3.1500e+00],
[-5.0275e+00, -7.9671e+00, -5.3033e+00, ..., 2.5250e-02,
3.4317e+00, 2.9989e+00],
...,
[-5.8244e-01, -3.1813e+00, -1.2097e+00, ..., -4.4037e-01,
3.3009e+00, 2.3396e+00],
[ 2.2888e+00, 2.0694e+00, 4.8974e-01, ..., 1.9418e+00,
1.9895e+00, 2.2213e+00],
[ 7.2607e-01, 9.0588e-01, 4.5900e-01, ..., 1.4632e+00,
7.9701e-01, 1.6341e+00]],
[[-2.3015e+00, -2.2033e+00, -1.9975e-02, ..., 4.7829e-01,
9.0132e-01, 1.5710e+00],
[-2.3016e+00, -6.5454e+00, -5.0213e-01, ..., -8.5172e-01,
-1.4257e+00, -2.3879e-01],
[ 7.6890e-02, -1.7247e+00, 2.2899e+00, ..., -2.6877e+00,
-8.7410e-01, 1.0514e+00],
...,
[-9.1552e-01, -1.2294e+00, -1.0671e-01, ..., -5.5497e+00,
-2.1424e+00, -8.0588e-01],
[-1.6449e+00, -2.1761e+00, -1.1863e+00, ..., -1.5193e+00,
-1.3273e+00, -2.2611e-01],
[-7.3859e-01, -8.8687e-01, -1.2954e+00, ..., 4.3910e-01,
-5.4754e-01, -1.0713e-01]]],
[[[ 8.4612e-01, 1.4729e+00, 1.2059e+00, ..., 1.4216e+00,
1.1061e+00, 9.9960e-01],
[-9.2793e-01, -1.7349e+00, -8.9542e-01, ..., -4.1578e-01,
-6.6614e-01, -1.6185e+00],
[-1.3854e-01, -7.1587e-01, -4.5516e-01, ..., -9.0260e-01,
8.2611e-01, 7.8396e-01],
...,
[-1.2286e-02, -1.1521e+00, -1.2977e+00, ..., -1.6722e+00,
-6.2167e-01, 2.2468e-02],
[-6.3168e-01, -1.8474e+00, -1.2428e+00, ..., -3.9146e-01,
-2.7564e-01, -2.3840e-01],
[-3.1488e-01, -1.7084e+00, -1.6560e+00, ..., -1.2269e+00,
-1.4971e+00, -2.6026e-01]],
[[-3.7339e+00, -7.4083e-01, -1.5986e+00, ..., -1.5146e+00,
-2.4263e+00, -1.2307e+00],
[-3.9523e+00, -2.9106e+00, -2.7177e+00, ..., -2.1432e+00,
-3.1649e+00, -9.2543e-01],
[-2.3010e+00, -2.1563e+00, -4.0364e+00, ..., -1.7210e+00,
-2.7991e+00, -1.0431e+00],
...,
[-1.5070e+00, -1.6253e+00, -2.8243e+00, ..., -4.5852e+00,
-2.8333e+00, -1.2044e-01],
[-3.0500e+00, -1.6109e+00, -2.7387e+00, ..., -2.0545e+00,
-2.0685e+00, -1.0926e+00],
[-2.3224e+00, -1.9987e+00, -2.7627e+00, ..., -2.0016e+00,
-1.9404e+00, -9.4438e-01]],
[[ 2.8471e+00, 4.1609e+00, 3.3361e+00, ..., 2.9321e+00,
1.1189e+00, 3.0670e+00],
[ 1.6572e-01, 1.0693e+00, 2.2108e+00, ..., 1.1000e+00,
-1.1603e+00, 2.7733e+00],
[ 9.7381e-01, 2.1275e+00, 2.7320e+00, ..., 1.8145e+00,
-3.8853e-01, 1.9051e+00],
...,
[ 9.3806e-01, 1.0348e+00, 1.1628e+00, ..., 9.0490e-01,
2.3698e+00, 1.5519e+00],
[ 6.5522e-01, 1.8523e+00, 1.2383e+00, ..., 6.6056e-01,
1.1906e+00, 1.7300e+00],
[ 1.6534e+00, 1.6333e+00, 1.8384e+00, ..., 1.2981e+00,
1.8534e+00, 1.7952e+00]],
...,
[[-1.8441e+00, -7.1130e+00, -5.4146e+00, ..., -4.1530e+00,
-3.2707e+00, -5.5419e+00],
[-2.4685e+00, -7.0034e+00, -4.8835e+00, ..., -3.6701e+00,
-2.5982e+00, -4.2773e+00],
[-9.5118e-01, -4.7122e+00, -3.8738e+00, ..., -2.3233e+00,
-3.2521e+00, -2.4837e+00],
...,
[-2.5754e+00, -5.3073e+00, -4.3423e+00, ..., -5.3626e+00,
-6.2940e+00, -7.6870e+00],
[-2.7904e+00, -5.7552e+00, -5.3457e+00, ..., -8.3976e+00,
-7.1389e+00, -6.4942e+00],
[-1.5259e+00, -4.1668e+00, -2.7968e+00, ..., -3.3277e+00,
-2.7224e+00, -3.2086e+00]],
[[-1.5293e-01, 3.5817e-01, 5.8209e-01, ..., 6.9633e-01,
8.6862e-01, 1.3778e+00],
[ 1.6292e+00, 6.9341e-01, 1.8097e+00, ..., 1.5441e+00,
2.3550e+00, 3.0251e+00],
[ 1.1700e+00, -3.6832e-01, -8.2221e-01, ..., 2.3396e+00,
1.4796e+00, 2.2728e+00],
...,
[ 1.2797e+00, 1.8868e-01, -3.1965e-01, ..., 1.2681e+00,
2.3426e+00, 2.2932e+00],
[ 1.9021e+00, 1.7088e+00, 1.0778e+00, ..., 2.0237e+00,
1.5136e+00, 2.0354e+00],
[ 4.4436e-01, 7.7934e-01, 6.8531e-01, ..., 4.9300e-01,
7.3387e-01, 1.6852e+00]],
[[-5.8538e-01, 3.0959e-01, 1.4418e-01, ..., 6.1238e-01,
5.4032e-01, 1.3438e+00],
[-1.3642e+00, -1.9361e+00, -1.4507e+00, ..., -1.0950e+00,
-6.5852e-01, 1.7569e-01],
[-2.2247e+00, -1.3467e+00, -2.2033e-01, ..., -3.6151e-01,
1.9402e-02, 6.9080e-01],
...,
[-1.9486e+00, -1.8729e+00, -1.5289e+00, ..., -2.2296e+00,
-7.9125e-01, -4.6300e-01],
[-1.5975e+00, -2.0763e+00, -1.6556e+00, ..., -2.3841e-01,
-6.4928e-01, -4.3469e-01],
[-8.2630e-01, -8.9575e-01, -1.1718e+00, ..., -5.9725e-01,
-8.7054e-01, -9.3480e-02]]],
[[[ 1.1077e+00, 1.4138e+00, 1.0599e+00, ..., 1.2634e+00,
1.0676e+00, 5.0003e-01],
[-5.5476e-01, -1.2695e+00, -9.5184e-01, ..., 3.4773e-02,
-9.6353e-02, -1.6554e+00],
[-3.2951e-01, -4.6692e-01, 1.4416e+00, ..., -2.0304e-01,
2.4808e-02, 6.7876e-01],
...,
[ 1.3113e+00, 2.1763e-01, -5.8703e-01, ..., 3.6222e-01,
1.0319e+00, 8.1622e-01],
[-1.0573e-01, -7.0011e-01, -1.9379e-02, ..., 9.2949e-01,
1.5493e+00, 1.2924e-01],
[ 8.6230e-01, -1.4711e-01, -1.4615e-01, ..., 3.8504e-01,
2.8481e-01, 7.5005e-01]],
[[-2.4298e+00, 8.7726e-01, -1.0784e-01, ..., -1.6720e+00,
-2.2799e+00, -6.1877e-01],
[-3.7794e+00, -7.8309e-01, 1.3912e+00, ..., -2.3699e+00,
-4.3013e+00, -1.7552e+00],
[-2.9951e+00, -1.4646e+00, -1.1470e+00, ..., -2.5225e+00,
-3.1858e+00, -1.4796e+00],
...,
[-2.0011e+00, -3.0097e+00, -3.8786e+00, ..., -5.9643e-01,
-1.4804e-01, 7.6690e-01],
[-2.6305e+00, -5.0628e-01, -2.9279e-01, ..., -2.2239e-02,
-1.7155e-01, 6.6488e-01],
[-1.4410e+00, -1.1698e+00, -8.4613e-01, ..., -4.3858e-01,
-5.8552e-01, -1.3981e-01]],
[[ 2.5090e+00, 2.0705e+00, 8.3717e-01, ..., 2.1243e+00,
5.0181e-01, 2.6496e+00],
[ 1.2432e+00, -1.0825e-01, -1.1858e+00, ..., 2.0992e+00,
4.8837e-01, 2.7857e+00],
[ 1.5799e+00, 1.9541e+00, 8.2679e-01, ..., 1.1632e+00,
1.7920e+00, 2.9840e+00],
...,
[ 1.4087e+00, 1.0717e+00, 1.8550e+00, ..., 9.2198e-01,
6.3522e-01, 1.9541e+00],
[ 1.1219e+00, 2.6154e+00, 2.7758e+00, ..., 9.6185e-01,
7.0531e-01, 1.7237e+00],
[ 2.3667e+00, 1.7879e+00, 2.3302e+00, ..., 1.9063e+00,
1.0090e+00, 1.7311e+00]],
...,
[[-2.4568e+00, -6.7309e+00, -4.5051e+00, ..., -6.6538e+00,
-4.5411e+00, -5.7748e+00],
[-2.9305e+00, -7.0426e+00, -5.3166e+00, ..., -5.6105e+00,
-5.2149e+00, -5.3094e+00],
[-2.0319e+00, -5.7480e+00, -4.5056e+00, ..., -5.4607e+00,
-4.1038e+00, -3.4923e+00],
...,
[-2.3945e+00, -4.7677e+00, -3.4847e+00, ..., -3.0840e+00,
-2.7741e+00, -3.0020e+00],
[-2.3659e+00, -5.5625e+00, -4.3597e+00, ..., -4.3194e+00,
-3.1336e+00, -4.9048e+00],
[-2.5770e+00, -5.7866e+00, -2.8029e+00, ..., -2.9111e+00,
-1.6022e+00, -3.3328e+00]],
[[ 4.4493e-01, 9.7877e-01, -2.6478e-01, ..., 8.7147e-01,
1.6082e+00, 1.7209e+00],
[ 1.1893e+00, 1.6873e+00, 8.8545e-01, ..., 7.8047e-01,
1.2260e+00, 2.7792e+00],
[ 1.3149e+00, 6.5960e-01, 3.2788e-01, ..., 1.3947e+00,
1.0584e+00, 2.0295e+00],
...,
[ 1.0762e+00, -6.6660e-01, -7.9556e-01, ..., 1.8802e+00,
1.9131e+00, 1.4779e+00],
[ 2.1534e+00, 1.6492e+00, 2.4786e+00, ..., 1.9313e+00,
1.6885e+00, 2.5476e+00],
[ 1.1253e+00, 8.3997e-01, 1.3339e+00, ..., 1.3202e+00,
1.5542e+00, 1.5416e+00]],
[[-5.8686e-01, 9.9653e-02, 7.1034e-01, ..., -7.4543e-03,
-2.5730e-02, 1.0162e+00],
[-1.3947e+00, -1.4817e+00, -8.8457e-01, ..., -1.8623e+00,
-8.2034e-01, -2.3937e-01],
[-2.1541e+00, -5.0413e-01, 7.7308e-03, ..., 2.2793e-01,
-4.0785e-01, -1.3278e-01],
...,
[-1.8706e+00, -1.0419e+00, -1.3909e+00, ..., -1.8754e-01,
-4.5006e-01, -4.4120e-01],
[-1.5553e+00, -1.3127e+00, -6.6945e-01, ..., -3.2313e-01,
-3.8665e-01, 4.2300e-01],
[-1.7050e-01, 4.0231e-01, -1.9267e-01, ..., 2.3226e-01,
5.4714e-02, 6.2415e-01]]]], device='cuda:0'),
tensor([[[[ 8.7692e-01, 1.4946e+00, 1.1873e+00, ..., -1.0941e+00,
6.6882e-01, 3.7527e-01],
[-9.2670e-01, -2.0499e+00, -1.2816e+00, ..., -1.3266e+00,
1.4973e-04, -1.0630e+00],
[ 2.2389e-01, -5.3699e-01, -3.9684e-01, ..., -1.3681e+00,
-3.8667e+00, 4.3743e-01],
...,
[-3.3268e-01, -6.0516e-01, -4.5272e-01, ..., -1.5209e+00,
-8.3605e-01, 6.6644e-01],
[ 4.4458e-01, -2.6077e+00, 4.8590e-01, ..., -2.2561e-01,
-5.8517e-01, -4.6938e-02],
[ 1.5440e+00, -9.0488e-01, -3.4529e-01, ..., -4.1809e-01,
-4.5672e-01, -2.5715e-01]],
[[-3.8138e+00, -7.1313e-01, -1.6274e+00, ..., -2.4770e+00,
-4.5918e+00, -2.5992e+00],
[-4.0632e+00, -3.0018e+00, -2.0787e+00, ..., -1.7893e+00,
-3.1525e+00, -2.5353e+00],
[-2.9195e+00, -4.3844e+00, -3.1551e+00, ..., -5.5138e+00,
-2.8035e+00, -2.7959e+00],
...,
[-1.3491e+00, -1.4494e+00, -3.6896e+00, ..., -1.2342e+00,
-2.4330e+00, -1.5918e+00],
[-1.9601e+00, 2.2154e+00, -2.7199e-01, ..., -7.2531e-01,
-2.9798e+00, -1.4057e+00],
[-2.4176e+00, -2.4346e-01, -2.2538e+00, ..., -9.4374e-01,
-2.6405e+00, -1.4490e+00]],
[[ 2.9766e+00, 4.3953e+00, 3.4684e+00, ..., 3.1658e+00,
1.8172e-01, 2.8101e+00],
[ 2.5377e-01, 1.1502e+00, 2.1928e+00, ..., -2.6142e+00,
-5.6688e+00, -8.2573e-01],
[ 9.6061e-01, 1.4937e+00, 1.7263e+00, ..., -6.9881e+00,
-5.8145e+00, -3.3926e+00],
...,
[-4.2839e+00, -3.4859e+00, -7.9565e-01, ..., -4.9061e-01,
-4.4365e-01, 8.1656e-01],
[-1.7830e+00, -4.4353e-01, 6.1035e-03, ..., -9.9598e-01,
-1.0197e+00, 2.2120e-01],
[ 1.4256e+00, 1.2202e-01, -6.0656e-01, ..., -7.1997e-01,
1.0154e-01, 5.5159e-01]],
...,
[[-1.8988e+00, -7.2339e+00, -5.3726e+00, ..., -2.5111e+00,
-1.6629e+00, -4.0341e+00],
[-2.4800e+00, -6.9607e+00, -4.2184e+00, ..., -8.1473e+00,
-2.2723e+00, -4.6777e+00],
[-1.0378e+00, -5.4335e+00, -5.6734e+00, ..., -1.4440e+01,
-5.8832e-01, -2.4026e+00],
...,
[-1.8589e+00, -5.1576e+00, -1.3160e+00, ..., 7.8660e-01,
3.6861e-01, -1.4669e+00],
[-1.8853e+00, -5.1583e+00, -3.1521e+00, ..., -4.5903e-02,
-1.2392e+00, -3.5207e-01],
[-1.6030e+00, -4.0570e+00, -2.1409e+00, ..., -3.8515e-03,
-3.2876e-02, 1.2355e-01]],
[[-1.6022e-01, 2.9056e-01, 4.7087e-01, ..., -9.5364e-01,
-1.5728e+00, 6.7669e-01],
[ 1.6486e+00, 1.1039e+00, 1.9114e+00, ..., 6.7296e-01,
-2.3978e+00, 3.0819e-02],
[ 6.9806e-01, -1.0348e-01, 8.4744e-01, ..., -2.8908e-01,
-3.8739e-01, -3.2177e+00],
...,
[ 1.3239e+00, 6.6483e-01, -1.9299e+00, ..., 4.6696e-01,
-7.8535e-02, 9.0823e-01],
[ 2.4282e+00, 2.0043e+00, -2.0403e-01, ..., 2.4108e-01,
1.5784e-01, 1.2141e+00],
[ 1.2057e-01, 1.1878e+00, 5.2790e-01, ..., 9.5902e-02,
-8.1208e-03, 1.0496e+00]],
[[-5.5451e-01, 3.0044e-01, 1.3160e-01, ..., -1.8952e+00,
9.6872e-01, 1.2124e+00],
[-1.3285e+00, -1.9185e+00, -1.7591e+00, ..., -2.0898e+00,
1.3239e+00, 2.2593e+00],
[-2.4381e+00, -1.4636e+00, 7.0902e-01, ..., -3.5370e+00,
2.7407e-01, 2.6312e+00],
...,
[-9.8177e-01, 3.1702e-02, -2.3382e+00, ..., -1.2328e+00,
-9.0896e-01, 8.6815e-01],
[-1.7499e+00, -1.1649e+00, -1.3451e+00, ..., -2.7527e-01,
-1.5079e+00, 1.7351e-01],
[ 1.1703e-01, -2.6425e-02, -6.3257e-01, ..., -8.3739e-01,
-1.9264e+00, -3.3847e-01]]],
[[[ 8.2032e-01, 1.0898e+00, -9.3796e-02, ..., 6.4291e-01,
9.4630e-01, 4.4623e-01],
[-1.3489e+00, -1.4741e+00, -7.0182e-01, ..., -3.2105e-01,
-5.4614e-02, -2.7469e-01],
[-3.8533e-01, -2.6211e+00, -1.5021e+00, ..., -7.1046e-01,
-4.3200e-01, -2.4560e-02],
...,
[-1.4210e+00, 9.0121e-01, 3.2079e-01, ..., 1.6878e-02,
1.3872e+00, 4.6614e-01],
[-6.1475e-01, 2.3692e-01, 8.9568e-01, ..., -1.1394e-01,
7.4403e-01, 3.3242e-01],
[ 1.2217e+00, -5.5219e-01, -4.8459e-01, ..., -3.0308e-01,
-9.0646e-01, 1.5086e-01]],
[[-4.2392e+00, -2.5834e+00, -5.0827e+00, ..., -1.5397e+00,
-1.9471e+00, -1.7025e+00],
[-3.4556e+00, -2.6513e+00, -2.2662e+00, ..., -2.0047e+00,
-1.8543e+00, -1.9741e+00],
[-3.1864e+00, -1.6057e+00, -4.5201e-01, ..., -1.9137e+00,
-1.6253e+00, -9.6334e-01],
...,
[ 1.4518e-01, -6.1899e-02, 2.9071e-01, ..., -1.0569e+00,
-8.2261e-01, -1.0875e+00],
[ 1.6366e-01, -8.5380e-01, -1.1364e+00, ..., -8.2405e-01,
-5.5685e-01, -6.7387e-01],
[-7.8409e-01, -1.3619e+00, -9.5974e-01, ..., -5.7198e-01,
-4.8916e-01, -4.7550e-01]],
[[ 2.8499e+00, 3.4991e+00, 3.5364e+00, ..., 1.4843e+00,
9.4961e-01, 2.5693e+00],
[ 8.2726e-01, 1.5749e+00, 1.7488e+00, ..., 1.1117e+00,
8.7352e-01, 2.6692e+00],
[ 1.5182e+00, 1.9778e+00, -5.5951e-01, ..., 5.9622e-01,
4.7592e-01, 1.3864e+00],
...,
[-1.0223e+00, -1.4941e+00, -2.5437e+00, ..., 3.5345e-01,
4.7932e-01, 1.5825e+00],
[-9.8234e-01, -1.2953e+00, -2.6489e+00, ..., -4.5003e-01,
1.2979e+00, 1.5246e+00],
[ 1.3703e+00, 7.2092e-02, 9.8507e-01, ..., 4.5267e-01,
1.4527e+00, 1.4573e+00]],
...,
[[-3.1817e+00, -8.6894e+00, -5.6183e+00, ..., -5.5711e+00,
-3.5667e+00, -4.1947e+00],
[-3.6564e+00, -1.0043e+01, -7.0217e+00, ..., -6.0062e+00,
-4.1273e+00, -2.9654e+00],
[-2.1051e+00, -7.1714e+00, -3.2479e+00, ..., -4.8060e+00,
-1.8925e+00, -1.3134e-01],
...,
[-9.2898e-01, -4.2863e+00, -7.3819e-01, ..., -4.5448e+00,
-3.4194e+00, -2.3348e+00],
[-2.1984e+00, -6.2945e+00, -3.7132e+00, ..., -6.3077e+00,
-4.1681e+00, -4.4070e+00],
[-1.7827e+00, -4.9392e+00, -2.6123e+00, ..., -3.8805e+00,
-2.5588e+00, -2.5547e+00]],
[[-9.7192e-01, -5.5740e-01, -1.1019e+00, ..., 1.8965e+00,
1.7012e+00, 1.2408e+00],
[ 1.2904e+00, 6.3467e-01, -4.1507e-01, ..., 1.8397e+00,
1.3399e+00, 1.6357e+00],
[ 7.8430e-01, 1.2108e+00, 1.8535e+00, ..., 9.7451e-01,
-3.2224e-01, 9.2047e-01],
...,
[-1.5904e-01, -7.9609e-01, 3.8111e-01, ..., 1.9120e+00,
5.4264e-01, 1.2143e+00],
[ 1.6813e+00, 8.2051e-01, 3.8241e-01, ..., 1.7101e+00,
8.8013e-01, 1.6727e+00],
[ 7.1832e-01, 2.1435e-01, 4.5206e-01, ..., 8.6287e-01,
7.6990e-01, 1.4994e+00]],
[[-8.1097e-01, -1.2136e+00, -1.1153e+00, ..., -2.0371e-01,
-5.1556e-01, -1.2302e-02],
[-1.6150e+00, -2.5421e+00, -1.2941e+00, ..., -1.2407e+00,
-9.1702e-01, -1.2614e+00],
[-1.4837e+00, -7.9567e-01, 7.4581e-01, ..., -8.4760e-01,
-1.0299e+00, -5.7632e-01],
...,
[-1.1408e+00, -2.3293e+00, -7.8020e-02, ..., -1.2693e+00,
-9.5799e-01, -1.0651e+00],
[-1.5716e+00, -1.3400e+00, -1.3353e+00, ..., -5.9855e-01,
-1.1739e+00, -5.2751e-01],
[ 8.4896e-02, -8.4944e-02, -4.7553e-01, ..., -3.6563e-01,
-7.3876e-01, 5.2549e-02]]],
[[[ 5.8138e-01, 8.4757e-01, 5.7019e-01, ..., 3.4769e-01,
4.4536e-01, 2.2618e-02],
[-2.6716e-01, -8.8276e-01, -5.1841e-01, ..., 1.2982e-01,
1.0488e-01, -2.9373e-01],
[-2.5381e-01, -1.0209e-01, 5.6233e-02, ..., -7.9438e-03,
-2.8185e-02, 6.3679e-01],
...,
[ 6.1882e-01, 6.7326e-01, 9.4993e-01, ..., 4.5964e-01,
8.9532e-01, 6.2777e-01],
[ 6.8948e-02, -8.1961e-02, 1.2649e+00, ..., 8.8860e-01,
1.5728e+00, 1.4667e-01],
[ 8.4097e-01, -2.4789e-01, -1.3010e-01, ..., 3.8287e-01,
2.7926e-01, 7.1588e-01]],
[[-2.3823e+00, -6.3725e-01, -2.0868e+00, ..., -1.6711e+00,
-2.2589e+00, -1.5354e+00],
[-2.4846e+00, -6.4746e-01, -1.8149e+00, ..., -1.3142e+00,
-2.4413e+00, -1.7598e+00],
[-2.5154e+00, -1.6465e+00, -2.9283e+00, ..., -2.1612e+00,
-2.4271e+00, -1.1307e+00],
...,
[-2.9518e-01, 1.2131e+00, 9.6220e-01, ..., -8.0404e-01,
-2.5517e-01, 7.2011e-01],
[-2.4428e+00, -8.4489e-02, -5.4672e-01, ..., -5.0756e-02,
1.4601e-02, 6.8430e-01],
[-1.5861e+00, -1.5285e+00, -1.2223e+00, ..., -4.5894e-01,
-5.4721e-01, -1.7235e-01]],
[[ 1.9715e+00, 2.5603e+00, 1.6907e+00, ..., 1.5111e+00,
7.6598e-01, 2.1152e+00],
[-2.0877e-01, 3.9076e-01, 6.4056e-01, ..., 7.2374e-01,
3.1709e-01, 2.2744e+00],
[ 5.4131e-02, 2.9782e-01, 8.1948e-01, ..., 3.6778e-02,
7.5840e-01, 1.9875e+00],
...,
[ 8.3479e-01, -1.2617e-01, -1.2058e+00, ..., 5.2114e-01,
1.1052e-01, 1.6048e+00],
[ 1.1519e+00, 2.6023e+00, 2.1793e+00, ..., 1.0251e+00,
7.5164e-01, 1.7070e+00],
[ 2.4375e+00, 1.9578e+00, 2.6711e+00, ..., 1.9730e+00,
1.0565e+00, 1.6859e+00]],
...,
[[-2.1095e+00, -5.0672e+00, -3.1348e+00, ..., -2.0119e+00,
-1.6457e+00, -3.0323e+00],
[-2.5525e+00, -5.5480e+00, -2.9977e+00, ..., -2.2864e+00,
-2.3619e+00, -3.0423e+00],
[-1.5979e+00, -5.4782e+00, -3.3701e+00, ..., -1.9440e+00,
-2.5250e+00, -2.3077e+00],
...,
[-2.5278e+00, -3.9888e+00, -1.4262e+00, ..., -2.7321e+00,
-2.0772e+00, -2.6203e+00],
[-2.4103e+00, -5.3087e+00, -3.3228e+00, ..., -4.4403e+00,
-3.1695e+00, -4.9422e+00],
[-2.5285e+00, -5.6645e+00, -2.3512e+00, ..., -3.0518e+00,
-1.6526e+00, -3.3453e+00]],
[[ 9.3582e-01, 1.6264e+00, 6.8281e-01, ..., 1.0630e+00,
1.5040e+00, 1.4278e+00],
[ 1.1755e+00, 9.1418e-01, 5.5987e-01, ..., 1.0159e+00,
8.1306e-01, 1.9270e+00],
[ 6.6219e-01, -1.3547e-01, -5.7153e-01, ..., 8.0897e-01,
1.0629e-01, 1.4033e+00],
...,
[ 1.6656e+00, 5.8856e-01, 1.5207e+00, ..., 1.5140e+00,
1.7981e+00, 1.4715e+00],
[ 2.0228e+00, 8.7550e-01, 1.1054e+00, ..., 1.9737e+00,
1.7442e+00, 2.5624e+00],
[ 1.1962e+00, 9.2824e-01, 1.6123e+00, ..., 1.3329e+00,
1.5081e+00, 1.5276e+00]],
[[-7.2086e-02, -6.6254e-01, -1.2332e+00, ..., -5.1831e-01,
-8.4779e-01, -7.2135e-02],
[-8.4637e-01, -2.6471e+00, -3.0509e+00, ..., -9.4385e-01,
-9.4058e-01, -7.0235e-01],
[-1.5494e+00, -1.9587e+00, -1.8050e+00, ..., -1.1647e-01,
-4.5943e-01, -2.8817e-01],
...,
[-1.5378e+00, -1.0907e+00, -1.7897e-01, ..., -4.5199e-01,
-4.6228e-01, -4.9146e-01],
[-1.6379e+00, -9.8132e-01, -1.1499e-01, ..., -4.0439e-01,
-3.7011e-01, 4.1823e-01],
[-2.5287e-01, 4.4836e-01, -1.6230e-01, ..., 2.0165e-01,
3.2978e-02, 6.3054e-01]]],
[[[ 6.3449e-01, 9.8547e-01, 7.4189e-01, ..., 4.3307e-01,
3.8592e-01, 2.2994e-01],
[-2.9008e-01, -1.3927e+00, -1.7430e-01, ..., 3.1143e-01,
3.2051e-01, -1.2387e+00],
[-3.1482e-01, -1.2936e+00, -4.2083e-02, ..., -4.0556e-01,
1.0326e+00, 6.7779e-01],
...,
[ 8.2737e-01, -9.6000e-01, 2.1595e-01, ..., 8.4682e-01,
4.4508e-01, 2.2200e-01],
[ 1.5146e-01, -5.3097e-01, 1.9712e+00, ..., 4.8854e-01,
1.8556e+00, 3.4795e-01],
[ 2.5148e-01, -8.6895e-01, -8.4149e-01, ..., 1.2830e-01,
4.5640e-01, 7.4425e-01]],
[[-3.5900e+00, -1.6369e+00, -3.2834e+00, ..., -1.6350e+00,
-2.0597e+00, -9.7714e-01],
[-4.4472e+00, -1.9956e+00, -2.9381e+00, ..., -4.7116e-01,
-1.2324e+00, -2.3609e-01],
[-3.3876e+00, -2.9157e+00, -4.0072e+00, ..., -1.4709e+00,
-1.4261e+00, -7.4742e-01],
...,
[-2.1084e+00, -3.6777e+00, -4.0827e+00, ..., -4.6833e+00,
-5.2241e+00, -2.4522e+00],
[-3.1247e+00, -1.1198e+00, -2.1032e+00, ..., -2.7714e+00,
-1.7639e+00, -3.9151e-01],
[-1.2802e+00, -1.2880e+00, -2.0470e+00, ..., -1.2091e+00,
-1.1664e+00, -7.3194e-01]],
[[ 2.2867e+00, 3.0965e+00, 2.5531e+00, ..., 2.8590e+00,
1.4223e+00, 2.8729e+00],
[ 7.2698e-01, 1.5228e+00, 1.7963e+00, ..., 1.9420e+00,
1.3186e-03, 2.4414e+00],
[ 6.6876e-01, 5.6173e-01, 1.0335e+00, ..., 1.2065e+00,
6.1142e-01, 1.8237e+00],
...,
[-1.2255e-01, -1.1001e+00, -5.5265e+00, ..., -2.1504e+00,
-5.4937e+00, -1.2234e+00],
[ 1.0650e+00, 3.4078e+00, 2.9298e+00, ..., -3.0130e-01,
2.9853e-01, 1.3172e+00],
[ 1.5626e+00, 1.6421e+00, 1.9233e+00, ..., 1.4011e+00,
1.3705e+00, 2.1959e+00]],
...,
[[-1.8611e+00, -5.0361e+00, -4.3122e+00, ..., -3.3848e+00,
-2.4160e+00, -3.8329e+00],
[-2.2425e+00, -4.1569e+00, -3.8240e+00, ..., -5.3971e+00,
-3.7387e+00, -4.1803e+00],
[-1.8348e-01, -1.8531e+00, -8.9236e-01, ..., -2.6763e+00,
-2.3873e+00, -1.6375e+00],
...,
[-1.9126e+00, -4.5638e+00, -3.3554e+00, ..., -6.5278e+00,
-2.8631e+00, -4.4705e+00],
[-1.7853e+00, -5.4166e+00, -6.1203e+00, ..., -4.8136e+00,
-4.7817e+00, -5.6347e+00],
[-1.5389e+00, -4.9972e+00, -3.9966e+00, ..., -4.1192e+00,
-3.6084e+00, -4.0066e+00]],
[[ 5.9458e-01, 6.0200e-01, -6.5049e-02, ..., 9.2474e-01,
1.1275e+00, 1.2657e+00],
[ 1.1207e+00, 1.0447e+00, 5.5148e-01, ..., 1.5835e+00,
2.3803e+00, 2.5538e+00],
[ 5.1871e-01, -7.8131e-01, -1.7926e+00, ..., 9.8272e-01,
1.6808e+00, 1.9358e+00],
...,
[ 1.3037e-01, -1.3768e+00, 8.3849e-01, ..., -7.4851e-01,
1.6358e+00, 5.9308e-01],
[ 1.5951e+00, 1.0145e+00, 1.2368e+00, ..., 7.8440e-01,
5.4020e-01, 2.5998e+00],
[ 7.2274e-01, 8.2423e-01, 1.0953e+00, ..., 1.1318e+00,
1.3646e+00, 1.3358e+00]],
[[-3.7225e-01, -2.3758e-01, -9.4000e-01, ..., -1.8404e-01,
-1.5550e-01, 8.7742e-01],
[-1.3155e+00, -1.0025e+00, -1.1401e+00, ..., 8.3669e-02,
-1.3222e-01, 3.2684e-01],
[-2.0157e+00, -8.6179e-01, -8.1904e-02, ..., 2.0551e-01,
1.6166e-01, 2.5350e-01],
...,
[-1.0209e+00, -1.2285e+00, -2.7080e+00, ..., -3.4794e+00,
-2.1432e+00, -1.1809e+00],
[-1.3554e+00, -1.0015e+00, -6.4570e-01, ..., -2.1804e+00,
-1.1477e+00, 7.4708e-01],
[-1.6878e-01, -1.3645e-01, -5.0326e-01, ..., 3.0851e-01,
1.3958e-01, 7.5791e-01]]],
[[[ 7.1715e-01, 1.0756e+00, 7.2523e-01, ..., 4.5477e-01,
3.8855e-01, 2.2042e-01],
[-2.8831e-01, -1.4409e+00, -5.1082e-01, ..., 2.7206e-01,
3.8414e-01, -1.1467e+00],
[ 1.4534e-01, -7.9447e-01, -2.9360e-01, ..., -5.5804e-01,
1.0755e+00, 6.4707e-01],
...,
[ 5.4175e-02, -1.4200e+00, -3.6123e-02, ..., -9.4222e-02,
9.0723e-01, 5.5808e-01],
[-4.2806e-01, -1.9559e+00, -9.7980e-01, ..., -1.1290e-01,
1.3818e+00, -8.9367e-02],
[ 3.2452e-01, -5.5305e-01, -6.9428e-01, ..., -3.2933e-01,
3.4738e-01, 5.8483e-01]],
[[-3.0699e+00, -1.5240e+00, -3.1410e+00, ..., -1.8496e+00,
-2.3588e+00, -1.2882e+00],
[-4.3873e+00, -1.7825e+00, -2.3508e+00, ..., -3.9396e-01,
-1.2143e+00, -3.2766e-01],
[-4.2599e+00, -2.4034e+00, -2.9037e+00, ..., -1.6475e+00,
-1.3075e+00, -7.7197e-01],
...,
[-2.8542e+00, -1.7986e+00, -2.3403e+00, ..., -4.3606e-01,
3.8242e-02, 4.4692e-01],
[-3.5942e+00, -1.4261e+00, -2.3887e+00, ..., -3.7204e-01,
-4.8464e-01, 1.5206e-01],
[-1.3081e+00, -8.8211e-01, -1.5800e+00, ..., -5.1341e-01,
-7.5475e-01, -5.5003e-01]],
[[ 1.6982e+00, 2.7980e+00, 2.3928e+00, ..., 3.0026e+00,
1.3935e+00, 3.0071e+00],
[ 2.0926e-01, 1.1769e+00, 1.5966e+00, ..., 2.0916e+00,
6.4425e-03, 2.5944e+00],
[ 7.9148e-01, 1.3680e+00, 8.6898e-01, ..., 1.1455e+00,
6.6808e-01, 1.7910e+00],
...,
[ 9.3700e-01, 1.4118e+00, 1.4091e+00, ..., 1.1260e+00,
5.7161e-02, 1.4675e+00],
[ 6.5957e-01, 1.7546e+00, 8.8823e-01, ..., 1.0335e+00,
5.1474e-01, 1.4430e+00],
[ 1.5739e+00, 1.5954e+00, 1.6200e+00, ..., 1.4025e+00,
9.8416e-01, 1.9743e+00]],
...,
[[-1.4559e+00, -4.6363e+00, -4.0259e+00, ..., -3.6401e+00,
-2.7572e+00, -4.2461e+00],
[-1.7978e+00, -3.7134e+00, -4.2354e+00, ..., -5.5041e+00,
-4.0183e+00, -4.7626e+00],
[-8.8640e-01, -2.9575e+00, -3.5023e+00, ..., -2.8882e+00,
-2.4340e+00, -1.4899e+00],
...,
[-1.5832e+00, -4.2237e+00, -4.6958e+00, ..., -3.9457e+00,
-3.3685e+00, -1.3747e+00],
[-1.5240e+00, -4.7232e+00, -5.2981e+00, ..., -4.9629e+00,
-3.4083e+00, -3.4279e+00],
[-1.5809e+00, -4.8650e+00, -4.1618e+00, ..., -3.4148e+00,
-2.0556e+00, -2.4715e+00]],
[[ 7.8102e-01, 5.2384e-01, -1.4610e-01, ..., 8.0272e-01,
1.0150e+00, 1.1836e+00],
[ 1.1891e+00, 9.8544e-01, 4.9412e-01, ..., 1.5968e+00,
2.5087e+00, 2.6380e+00],
[ 6.2523e-01, 6.2010e-01, 1.1112e-01, ..., 8.2693e-01,
1.6485e+00, 1.9481e+00],
...,
[ 8.2592e-01, -1.3240e-01, -3.6742e-01, ..., 1.5016e+00,
1.8749e+00, 1.7550e+00],
[ 1.5636e+00, 1.3636e+00, 1.1949e+00, ..., 1.2726e+00,
1.5157e+00, 2.3721e+00],
[ 6.6217e-01, 9.3083e-01, 1.0817e+00, ..., 9.8445e-01,
1.3530e+00, 1.3259e+00]],
[[-2.8280e-01, -1.7554e-01, -1.0198e+00, ..., -1.9430e-01,
-1.6181e-01, 8.3350e-01],
[-1.3711e+00, -1.3982e+00, -1.9988e+00, ..., 6.0638e-02,
-4.8424e-02, 3.8431e-01],
[-1.6277e+00, -7.5522e-01, -9.1686e-01, ..., 2.6045e-01,
3.5472e-01, 3.2897e-01],
...,
[-1.7453e+00, -1.5262e+00, -1.0972e+00, ..., -3.4311e-02,
-9.8125e-03, -3.1005e-01],
[-1.4050e+00, -1.6513e+00, -1.5569e+00, ..., -2.7424e-01,
-3.2221e-01, 3.1697e-01],
[-1.9024e-01, -6.9956e-03, -4.5411e-01, ..., 1.8984e-01,
1.4613e-01, 3.2725e-01]]]], device='cuda:0'),
tensor([[[[ 1.0697e+00, 1.3727e+00, 1.2462e+00, ..., 1.4954e+00,
1.0432e+00, 4.3872e-01],
[-6.4526e-01, -2.4705e+00, -1.6118e+00, ..., 4.0109e-02,
-2.6635e-02, -1.4302e+00],
[-4.9660e-01, -1.1802e+00, -9.6719e-01, ..., -2.9491e-01,
1.1644e-01, 5.7808e-01],
...,
[ 7.0942e-01, 5.7317e-01, 1.1039e+00, ..., 3.9729e-01,
1.0060e+00, 8.1142e-01],
[-7.1144e-02, -6.8376e-01, 1.7526e-01, ..., 9.5454e-01,
1.5372e+00, 1.0044e-01],
[ 8.3135e-01, -1.8065e-01, -7.7486e-02, ..., 3.6160e-01,
2.5765e-01, 7.5428e-01]],
[[-2.9311e+00, 3.2502e-02, -1.3571e+00, ..., -1.6122e+00,
-2.4727e+00, -6.6967e-01],
[-4.2479e+00, -2.7448e+00, -3.1223e+00, ..., -3.5977e+00,
-3.9712e+00, -1.6894e+00],
[-3.1177e+00, -2.4524e+00, -3.4148e+00, ..., -3.5261e+00,
-2.8401e+00, -1.4078e+00],
...,
[-6.4589e-01, -2.3154e-01, -8.1019e-01, ..., -5.1037e-01,
1.1704e-02, 7.8162e-01],
[-2.5721e+00, -1.7264e-01, -6.8234e-03, ..., -2.9864e-02,
-1.3245e-01, 6.4760e-01],
[-1.5093e+00, -1.1396e+00, -9.5411e-01, ..., -4.5670e-01,
-5.6699e-01, -1.5429e-01]],
[[ 2.5427e+00, 2.9591e+00, 2.7139e+00, ..., 1.8590e+00,
4.8924e-01, 2.5797e+00],
[ 8.7093e-01, 6.4244e-01, 2.8548e+00, ..., 1.4662e+00,
6.1693e-01, 2.6792e+00],
[ 1.1519e+00, 1.6590e+00, 2.9387e+00, ..., 9.2807e-01,
1.7617e+00, 2.9069e+00],
...,
[ 1.1924e+00, 1.4116e+00, 1.7162e+00, ..., 9.3249e-01,
5.5042e-01, 1.8984e+00],
[ 1.0837e+00, 2.3736e+00, 2.3159e+00, ..., 1.0362e+00,
7.5772e-01, 1.7282e+00],
[ 2.3593e+00, 1.8330e+00, 2.4056e+00, ..., 1.9658e+00,
1.0905e+00, 1.7598e+00]],
...,
[[-2.8183e+00, -7.1402e+00, -6.0228e+00, ..., -4.8455e+00,
-4.4413e+00, -5.4303e+00],
[-3.4378e+00, -7.6682e+00, -5.5369e+00, ..., -5.1909e+00,
-4.9779e+00, -5.1969e+00],
[-2.2438e+00, -6.2588e+00, -4.3077e+00, ..., -2.8480e+00,
-3.9426e+00, -3.4553e+00],
...,
[-2.5484e+00, -4.9744e+00, -2.5494e+00, ..., -3.1613e+00,
-2.7672e+00, -3.0013e+00],
[-2.3966e+00, -5.2551e+00, -3.7999e+00, ..., -4.3997e+00,
-3.1775e+00, -4.9301e+00],
[-2.5820e+00, -5.8477e+00, -2.8853e+00, ..., -2.9524e+00,
-1.6399e+00, -3.3861e+00]],
[[ 3.1425e-01, 1.1576e+00, 3.9509e-01, ..., 5.1034e-01,
1.6124e+00, 1.6829e+00],
[ 1.3203e+00, 9.3673e-01, -3.7239e-01, ..., 1.7560e-01,
1.4528e+00, 2.7244e+00],
[ 1.3419e+00, 8.2656e-01, -9.5831e-02, ..., 7.5892e-01,
1.1729e+00, 2.0477e+00],
...,
[ 1.6519e+00, 3.2178e-01, 9.1688e-01, ..., 1.8946e+00,
1.9505e+00, 1.4379e+00],
[ 2.0533e+00, 1.3398e+00, 2.0262e+00, ..., 1.9130e+00,
1.6667e+00, 2.5596e+00],
[ 1.1942e+00, 8.7666e-01, 1.2863e+00, ..., 1.3381e+00,
1.5354e+00, 1.5505e+00]],
[[-6.4131e-01, 4.9123e-02, -4.6394e-01, ..., 2.5737e-01,
1.4479e-02, 9.0983e-01],
[-1.5285e+00, -2.1588e+00, -1.7568e+00, ..., -1.4147e+00,
-7.7642e-01, -9.5178e-02],
[-2.3270e+00, -1.4560e+00, -7.0954e-01, ..., -4.2141e-01,
-3.3430e-01, -1.0491e-01],
...,
[-1.5785e+00, -1.3008e+00, -5.8856e-01, ..., -1.7547e-01,
-4.5976e-01, -4.2637e-01],
[-1.6449e+00, -1.3759e+00, -7.3429e-01, ..., -3.5412e-01,
-4.4175e-01, 3.9843e-01],
[-2.2483e-01, 4.6249e-01, -2.4462e-01, ..., 1.8517e-01,
3.3631e-02, 6.4858e-01]]],
[[[ 8.3129e-01, 1.0606e+00, 8.6387e-01, ..., 1.9074e+00,
-4.3253e-01, -8.3119e-01],
[-6.6041e-01, -1.1622e+00, -1.6300e-01, ..., 1.4173e+00,
2.3476e+00, -6.2980e-01],
[-4.7586e-02, -4.7673e-01, -1.9028e+00, ..., -2.3060e+00,
1.7348e+00, 6.7543e-01],
...,
[ 2.7806e-01, -6.0218e-01, -5.2889e-01, ..., 1.1668e+00,
9.3922e-01, 3.9654e-01],
[-5.9999e-01, -1.7789e+00, -1.2137e+00, ..., -1.6668e+00,
-7.2852e-01, 7.6329e-02],
[-2.7121e-01, -1.6505e+00, -1.6736e+00, ..., -9.8818e-01,
-1.7213e+00, -3.2386e-01]],
[[-3.5330e+00, -9.5120e-01, -2.0997e+00, ..., -6.1598e+00,
-4.6623e+00, -8.4443e-01],
[-3.1651e+00, -2.1134e+00, -3.6735e+00, ..., -3.8228e+00,
-4.6621e+00, -9.0429e-02],
[-1.9653e+00, -1.9419e+00, -4.5355e+00, ..., -4.1008e+00,
-5.4401e+00, -2.9443e-01],
...,
[-1.3803e+00, -1.2595e+00, -1.9439e+00, ..., -2.4683e+00,
-4.6083e+00, -1.4649e+00],
[-2.9453e+00, -1.6182e+00, -2.6165e+00, ..., -4.8238e+00,
-4.1370e+00, -1.2054e+00],
[-2.2448e+00, -1.9699e+00, -2.7012e+00, ..., -2.6113e+00,
-2.2058e+00, -9.2971e-01]],
[[ 2.2843e+00, 2.8380e+00, 1.0880e+00, ..., 5.3367e-01,
1.0102e+00, 2.8357e+00],
[ 4.4452e-01, 8.4718e-01, 1.5116e+00, ..., -3.1367e-02,
-4.6577e-01, 2.0268e+00],
[ 1.0571e+00, 9.9381e-01, 3.0305e+00, ..., -2.7500e+00,
7.5876e-01, 1.5797e+00],
...,
[ 1.2430e+00, 1.4475e+00, 1.6644e+00, ..., -3.9517e+00,
1.9909e-01, 1.7945e+00],
[ 7.4825e-01, 1.8475e+00, 1.2856e+00, ..., -1.9091e+00,
1.4747e+00, 1.9112e+00],
[ 1.7065e+00, 1.5910e+00, 1.8580e+00, ..., 1.5201e+00,
1.5869e+00, 1.4226e+00]],
...,
[[-1.9743e+00, -6.5001e+00, -4.8588e+00, ..., -6.4928e+00,
-4.8377e+00, -4.8333e+00],
[-2.4903e+00, -7.7140e+00, -4.9172e+00, ..., -4.1890e+00,
-8.9887e+00, -8.8217e+00],
[-1.3710e+00, -4.0411e+00, -1.5572e+00, ..., 5.3418e-01,
-8.0770e+00, -6.6806e+00],
...,
[-2.2988e+00, -5.1186e+00, -3.7714e+00, ..., -7.4667e+00,
-7.9727e+00, -5.5278e+00],
[-2.7979e+00, -5.6887e+00, -5.2377e+00, ..., -7.9468e+00,
-6.4198e+00, -6.9434e+00],
[-1.5170e+00, -4.1246e+00, -2.7716e+00, ..., -5.5939e+00,
-4.7067e+00, -4.2281e+00]],
[[-4.6246e-01, 2.7855e-01, 1.0162e+00, ..., -4.3430e+00,
1.7434e-01, 2.2843e+00],
[ 1.1356e+00, -5.5493e-02, -1.2215e+00, ..., -3.3785e+00,
5.7538e-01, 3.8370e+00],
[ 1.0003e+00, -1.4570e-01, -2.6751e+00, ..., -2.0490e+00,
1.1956e-01, 2.9047e+00],
...,
[ 1.3240e+00, 3.5691e-01, 1.1765e-01, ..., -8.6019e-02,
1.0189e-01, 1.7117e+00],
[ 1.9123e+00, 1.7196e+00, 1.0663e+00, ..., 9.1070e-02,
1.5755e+00, 2.5117e+00],
[ 4.5828e-01, 8.1457e-01, 7.1416e-01, ..., 1.1147e+00,
1.2341e+00, 1.7894e+00]],
[[-6.4046e-01, -1.6869e-01, -2.2139e+00, ..., 2.3783e-01,
-1.6558e+00, 4.0188e-01],
[-1.2111e+00, -1.0790e+00, -1.0612e+00, ..., 4.5991e-01,
4.5985e-01, 5.6165e-01],
[-1.8838e+00, -2.8745e-01, 5.6061e-01, ..., 4.0622e-01,
1.3201e+00, -9.6236e-01],
...,
[-1.8912e+00, -1.7291e+00, -1.1713e+00, ..., -9.4031e-01,
-2.9543e+00, -1.9644e+00],
[-1.6312e+00, -2.0077e+00, -1.6028e+00, ..., -3.0233e+00,
-2.9453e+00, -5.9325e-01],
[-8.2400e-01, -8.4461e-01, -1.1383e+00, ..., -1.9816e-01,
-2.4392e-02, 2.5111e-01]]],
[[[ 7.5387e-01, 1.1148e+00, -1.3260e-01, ..., 1.9208e-01,
6.1759e-01, 1.5954e-01],
[-1.6265e+00, -2.6680e+00, -1.1823e+00, ..., 1.3197e-01,
-1.2626e+00, -5.5810e-01],
[-1.3995e+00, -3.2971e+00, -2.6770e-01, ..., 1.6905e+00,
8.3855e-01, 5.4114e-01],
...,
[-2.8355e-01, -3.6886e-01, -7.3764e-02, ..., -1.3378e-01,
-1.9007e-01, 3.0183e-01],
[ 9.6545e-02, -4.1397e-01, 5.9449e-02, ..., -1.2042e-01,
2.4959e-02, -1.2541e-01],
[ 4.4548e-01, -3.2700e-01, -8.5899e-01, ..., -4.1918e-01,
-1.0002e+00, 1.5802e-01]],
[[-5.2976e+00, -2.6308e+00, -3.5807e+00, ..., -1.9273e+00,
-2.3959e+00, -1.7580e+00],
[-5.2484e+00, -4.4243e+00, -3.6360e+00, ..., -2.5907e+00,
-2.6579e+00, -2.1786e+00],
[-2.5312e+00, -1.3591e+00, -1.6387e+00, ..., -1.6505e+00,
-2.3123e+00, -1.3557e+00],
...,
[-2.0170e+00, -1.1693e+00, -6.4939e-01, ..., -8.4381e-01,
-1.0080e+00, -8.7367e-01],
[-1.3707e+00, 1.6063e-01, -1.3299e-01, ..., -9.1843e-01,
-1.1243e+00, -1.2454e+00],
[ 9.9376e-02, -7.9945e-01, -9.4503e-01, ..., -7.3449e-01,
-6.9604e-01, -4.7701e-01]],
[[ 2.6080e+00, 3.5808e+00, 3.6646e+00, ..., 2.4551e+00,
7.6652e-01, 2.5729e+00],
[ 1.0708e+00, 2.2329e+00, 2.8565e+00, ..., 1.0368e+00,
1.5495e+00, 2.9565e+00],
[-1.2188e+00, -6.5480e-01, -1.0504e+00, ..., -1.4000e+00,
7.9037e-01, 2.1143e+00],
...,
[ 4.7219e-01, 6.5149e-01, 1.3310e+00, ..., -6.6523e-01,
7.0297e-01, 1.6193e+00],
[ 5.8733e-01, 2.0674e+00, 1.6158e+00, ..., -5.6257e-01,
9.0617e-01, 1.2245e+00],
[ 1.8224e+00, 1.2879e+00, 1.9618e+00, ..., 3.0518e-01,
1.3030e+00, 1.3662e+00]],
...,
[[-3.3533e+00, -8.7998e+00, -5.3893e+00, ..., -6.1115e+00,
-3.7034e+00, -4.4681e+00],
[-3.7084e+00, -9.0840e+00, -5.5071e+00, ..., -5.1789e+00,
-4.1095e+00, -3.9761e+00],
[-2.8316e+00, -5.6648e+00, -5.4301e+00, ..., -6.9631e+00,
-7.4164e+00, -3.3422e+00],
...,
[-1.9266e+00, -5.7685e+00, -4.5827e+00, ..., -5.7515e+00,
-3.5472e+00, -2.9298e+00],
[-3.0604e+00, -7.6106e+00, -4.8718e+00, ..., -6.2436e+00,
-4.6477e+00, -4.4722e+00],
[-2.3951e+00, -6.4599e+00, -3.1764e+00, ..., -3.8318e+00,
-2.7581e+00, -2.5481e+00]],
[[-9.1819e-01, -7.0665e-01, -1.3803e-01, ..., 2.0987e+00,
1.8870e+00, 1.3640e+00],
[-3.6909e-01, -1.4240e+00, -2.2518e+00, ..., 2.0986e-01,
1.8553e+00, 2.1422e+00],
[-4.6255e-02, 4.4045e-01, -2.0885e-01, ..., -8.2918e-01,
7.9428e-01, 2.0547e+00],
...,
[ 6.9569e-01, 2.9942e-01, 1.8123e+00, ..., 2.0637e+00,
7.6879e-01, 1.0390e+00],
[ 1.9463e+00, 1.5134e+00, 1.2457e+00, ..., 1.6694e+00,
6.8313e-01, 1.5389e+00],
[ 1.1752e+00, 9.9991e-01, 1.2672e+00, ..., 9.0101e-01,
6.7594e-01, 1.4656e+00]],
[[-5.7068e-01, -5.0484e-01, -7.1964e-01, ..., -9.1139e-01,
-8.9689e-01, -5.4951e-03],
[-2.3091e+00, -2.2513e+00, -9.4706e-01, ..., -2.4058e+00,
-2.5721e+00, -8.5761e-01],
[-2.3771e+00, -4.6461e-01, 1.4751e-02, ..., 2.0885e+00,
-9.8793e-01, -4.5562e-01],
...,
[-3.0249e+00, -2.5494e+00, -6.3769e-01, ..., -7.1982e-01,
-1.5139e+00, -8.5976e-01],
[-2.3355e+00, -1.2620e+00, -5.2000e-01, ..., -9.0315e-01,
-1.5933e+00, -7.0532e-01],
[ 2.7056e-02, 9.4276e-02, -2.7700e-01, ..., -3.7962e-01,
-8.3133e-01, -5.5036e-03]]],
[[[ 5.6706e-01, 7.8712e-01, 4.8639e-01, ..., 3.2188e-01,
3.9682e-01, -9.4017e-03],
[-1.8138e-01, -2.6702e-01, 1.8790e-01, ..., 1.5195e-01,
1.0174e-01, -3.2161e-01],
[-6.8660e-01, -4.0712e-01, -6.5560e-01, ..., -1.8266e-01,
3.1032e-01, 6.3135e-01],
...,
[ 1.0357e-01, -1.2479e+00, -8.7215e-02, ..., -5.1789e-02,
3.5028e-01, 5.5342e-01],
[ 8.6339e-03, -6.9150e-01, -6.1200e-02, ..., 8.5010e-01,
1.5705e+00, 1.2926e-01],
[ 8.4693e-01, -2.0483e-01, -1.3502e-01, ..., 3.9674e-01,
2.8523e-01, 7.0448e-01]],
[[-2.5279e+00, -7.0074e-01, -2.0071e+00, ..., -1.6906e+00,
-2.2850e+00, -1.5117e+00],
[-2.9136e+00, -2.2842e+00, -4.0821e+00, ..., -1.4346e+00,
-2.4841e+00, -1.6671e+00],
[-5.7035e+00, -6.2675e+00, -8.4669e+00, ..., -1.2678e+00,
-2.1072e+00, -1.0882e+00],
...,
[-9.2019e-01, -1.7997e+00, -1.6162e+00, ..., -1.0422e+00,
-2.5576e-02, 9.1410e-01],
[-2.5478e+00, -3.4940e-01, -5.5055e-01, ..., -3.6158e-02,
1.1430e-02, 6.4503e-01],
[-1.6819e+00, -1.3941e+00, -8.9456e-01, ..., -5.6465e-01,
-7.1858e-01, -2.3515e-01]],
[[ 2.0359e+00, 2.6056e+00, 1.6983e+00, ..., 1.6508e+00,
8.1674e-01, 2.1157e+00],
[-1.7238e-01, 4.2476e-01, 7.0827e-01, ..., 6.8221e-01,
3.6542e-01, 2.2768e+00],
[ 4.8112e-01, 2.7663e+00, 4.0375e+00, ..., 1.5051e+00,
9.2810e-01, 1.8076e+00],
...,
[ 5.7001e-01, 1.3867e+00, 2.6085e+00, ..., 1.3803e+00,
2.1682e-01, 1.0518e+00],
[ 1.1535e+00, 2.2413e+00, 1.8700e+00, ..., 1.0770e+00,
7.6083e-01, 1.6841e+00],
[ 2.4523e+00, 1.9121e+00, 2.3511e+00, ..., 2.0354e+00,
1.0403e+00, 1.6738e+00]],
...,
[[-2.0750e+00, -5.0096e+00, -3.1850e+00, ..., -2.0250e+00,
-1.7117e+00, -3.1368e+00],
[-2.6890e+00, -6.0985e+00, -4.7285e+00, ..., -2.6890e+00,
-2.3587e+00, -3.2063e+00],
[-2.3969e+00, -5.5982e+00, -2.0663e+00, ..., -3.0649e+00,
-2.9456e+00, -2.8042e+00],
...,
[-2.6304e+00, -4.5964e+00, -4.0358e+00, ..., -6.8217e+00,
-6.0699e+00, -4.9392e+00],
[-2.5385e+00, -5.6264e+00, -3.9521e+00, ..., -4.4650e+00,
-3.2781e+00, -4.9350e+00],
[-2.7048e+00, -6.0071e+00, -2.7901e+00, ..., -3.0631e+00,
-1.7189e+00, -3.3435e+00]],
[[ 8.6897e-01, 1.6187e+00, 7.9806e-01, ..., 1.0867e+00,
1.4963e+00, 1.4506e+00],
[ 1.1855e+00, 6.2573e-01, 6.8364e-01, ..., 9.1201e-01,
8.1649e-01, 1.9899e+00],
[-5.9835e-01, -3.0941e+00, -4.4723e+00, ..., 1.2142e+00,
2.9929e-01, 1.3817e+00],
...,
[ 1.6956e+00, 1.2764e-01, 1.2626e+00, ..., 2.4355e+00,
2.2549e+00, 1.7552e+00],
[ 2.0419e+00, 1.1898e+00, 1.8799e+00, ..., 1.9639e+00,
1.7038e+00, 2.5736e+00],
[ 1.1458e+00, 8.4381e-01, 1.3438e+00, ..., 1.3187e+00,
1.4361e+00, 1.5339e+00]],
[[-1.2252e-01, -7.2792e-01, -1.2530e+00, ..., -5.5478e-01,
-8.3958e-01, -8.8963e-02],
[-7.8145e-01, -2.7778e+00, -2.2561e+00, ..., -1.1611e+00,
-9.5624e-01, -6.6155e-01],
[-2.1553e+00, -2.1617e+00, -2.2368e+00, ..., -6.1017e-01,
-6.8036e-01, -2.1314e-01],
...,
[-1.4094e+00, -1.9164e+00, -1.5502e+00, ..., -1.6326e-01,
4.0730e-01, -2.6738e-01],
[-1.6938e+00, -1.3165e+00, -5.5018e-01, ..., -3.7603e-01,
-2.9521e-01, 4.6463e-01],
[-2.7413e-01, 3.3480e-01, -2.1848e-01, ..., 1.9313e-01,
-8.4290e-04, 6.2017e-01]]],
[[[ 8.6271e-01, 1.5238e+00, 1.2570e+00, ..., 1.1396e-01,
1.5450e-01, -4.8842e-01],
[-1.0023e+00, -1.9671e+00, -1.1411e+00, ..., 6.3721e-01,
1.7117e+00, -3.1569e-01],
[-1.8971e-01, -8.1310e-01, 5.5661e-02, ..., -3.0006e+00,
-2.6316e+00, 9.3867e-01],
...,
[-3.3217e-01, 1.0476e-01, -3.2233e+00, ..., -2.0303e-01,
2.1073e-01, 2.6524e-01],
[-7.4779e-01, 3.9120e-01, -2.0973e+00, ..., -7.5434e-01,
-2.7957e-01, -1.6032e-01],
[-1.0861e-01, 6.7365e-01, -9.4940e-01, ..., -1.0989e+00,
-1.5141e+00, -2.4050e-01]],
[[-3.7542e+00, -7.7100e-01, -1.5875e+00, ..., -5.5274e+00,
-6.2982e+00, -2.2856e+00],
[-4.0252e+00, -2.9530e+00, -2.1699e+00, ..., -9.7248e-01,
-2.4543e+00, -2.4658e+00],
[-2.4102e+00, -2.1433e+00, -2.3233e+00, ..., -5.7499e+00,
-3.2483e+00, -1.1675e+00],
...,
[-5.6023e-01, -5.5121e-01, 3.4763e-01, ..., -1.7626e+00,
-2.6330e+00, -7.2895e-01],
[-1.6335e+00, 3.1982e-02, 9.4684e-01, ..., -2.3045e+00,
-2.7299e+00, -1.1661e+00],
[-1.5389e+00, 1.2479e-01, 8.1258e-01, ..., -2.1586e+00,
-2.1984e+00, -9.4896e-01]],
[[ 2.9355e+00, 4.3034e+00, 3.4174e+00, ..., 1.6433e+00,
1.2511e+00, 4.5498e+00],
[ 2.6089e-01, 1.1845e+00, 2.3835e+00, ..., -2.6523e+00,
-3.7499e+00, -9.7275e-02],
[ 9.8918e-01, 1.7472e+00, 1.9142e+00, ..., -2.8996e+00,
-3.2951e+00, -3.2230e-01],
...,
[ 2.1533e+00, -5.5556e+00, -4.6590e+00, ..., -7.2187e-02,
1.5773e+00, 2.4008e+00],
[ 1.4793e+00, -4.1288e+00, -4.0519e+00, ..., 7.3661e-03,
1.5147e+00, 1.7417e+00],
[ 2.6196e+00, -1.7760e+00, -2.0651e+00, ..., 1.1406e+00,
1.9972e+00, 1.7320e+00]],
...,
[[-1.8814e+00, -7.3206e+00, -5.5354e+00, ..., -3.8452e+00,
-2.5921e+00, -4.2440e+00],
[-2.4299e+00, -7.0008e+00, -4.4672e+00, ..., -1.0470e+01,
-9.3138e+00, -1.0686e+01],
[-9.1835e-01, -4.4612e+00, -4.2029e+00, ..., -4.8979e+00,
3.1621e-01, -4.3894e+00],
...,
[-2.5895e+00, -5.8459e+00, -5.1859e+00, ..., -2.3233e+00,
-4.2023e+00, -4.5014e+00],
[-3.0610e+00, -5.0951e+00, -2.4433e+00, ..., -3.8027e+00,
-5.1603e+00, -5.9915e+00],
[-1.7224e+00, -4.5365e+00, -9.7620e-01, ..., -1.9879e+00,
-2.3338e+00, -3.1234e+00]],
[[-1.7239e-01, 2.9112e-01, 5.2646e-01, ..., -4.8252e+00,
-2.4589e+00, 1.2919e+00],
[ 1.6292e+00, 9.8173e-01, 1.8409e+00, ..., 6.5046e-01,
-2.9323e+00, 1.4320e-01],
[ 1.1553e+00, 6.3479e-02, 3.4960e-01, ..., 8.4518e-01,
-1.4948e-01, -6.6571e-01],
...,
[ 1.4940e+00, -4.1067e-01, -6.3676e-02, ..., 9.6347e-01,
7.5088e-01, 1.8952e+00],
[ 1.5741e+00, 4.8618e-01, -1.5711e+00, ..., 1.2720e+00,
1.0194e+00, 2.1349e+00],
[ 6.2325e-01, 1.9194e+00, 1.3778e+00, ..., 5.1218e-01,
6.1764e-01, 1.7396e+00]],
[[-5.5922e-01, 2.8814e-01, 1.3606e-01, ..., -2.0690e-01,
4.8524e-01, 2.6632e-01],
[-1.3617e+00, -1.8998e+00, -1.7752e+00, ..., -1.8470e-02,
1.3454e+00, 1.6531e+00],
[-2.2397e+00, -1.0710e+00, 4.8052e-01, ..., -4.4749e+00,
-1.9799e+00, 1.3262e+00],
...,
[-1.6506e+00, -4.1673e-01, 1.3027e+00, ..., -7.5616e-01,
-1.3351e+00, -7.2372e-01],
[-1.3153e+00, -1.1056e+00, -1.2332e+00, ..., -1.1124e+00,
-1.2921e+00, -6.9063e-01],
[-4.2389e-01, -4.7338e-01, 4.9778e-01, ..., -7.3376e-01,
-9.8825e-01, -5.8805e-02]]]], device='cuda:0'),
tensor([[[[ 6.0830e-01, 7.9608e-01, 3.4332e-01, ..., -1.3115e+01,
-8.7838e+00, -2.0381e+00],
[-4.7210e-01, -1.0929e+00, -7.8901e-01, ..., -2.3097e+01,
-1.6600e+01, -8.1881e+00],
[-3.0312e-02, -8.6978e-01, -1.7932e+00, ..., -2.1204e+01,
-1.3148e+01, 1.8607e-01],
...,
[-1.3854e+00, -3.9131e+00, 2.4327e+00, ..., -4.1859e-01,
6.2766e-01, 1.0189e+00],
[-1.9832e+00, -2.4046e+00, -2.6735e-01, ..., 3.0373e-01,
5.9962e-01, -1.9588e-01],
[-9.0694e-01, 6.1872e-01, 9.9592e-01, ..., 3.8892e-01,
4.8476e-01, 9.7652e-01]],
[[-3.9591e+00, -1.6228e+00, -3.4299e+00, ..., -1.8782e+01,
-7.0816e+00, -3.1200e+00],
[-4.2337e+00, -2.0895e+00, -3.9193e+00, ..., -3.4638e+01,
-2.3262e+01, -6.4364e+00],
[-3.6832e+00, -3.3703e+00, -5.7005e+00, ..., -3.6563e+01,
-2.8831e+01, -8.8720e+00],
...,
[-8.6383e+00, -1.5877e+01, -5.9198e+00, ..., 5.6927e-01,
-7.6312e-01, 3.6241e-01],
[-8.8641e+00, -1.1604e+01, -2.7460e+00, ..., 6.5353e-02,
-1.3838e+00, 2.9941e-01],
[-4.4397e+00, -5.5021e+00, 7.8917e-01, ..., 1.1211e-01,
-1.1329e+00, -6.1474e-01]],
[[ 2.4894e+00, 3.0344e+00, 2.4858e+00, ..., -1.0827e+01,
-7.8913e+00, -3.0198e+00],
[ 1.0486e+00, 1.4682e+00, 2.1572e+00, ..., -1.2936e+01,
-1.7953e+01, -1.1357e+01],
[ 1.2311e+00, 1.7867e+00, 8.4884e-01, ..., -1.8125e+01,
-1.9934e+01, -7.6417e+00],
...,
[ 4.1697e+00, 1.0149e+00, -1.1890e+01, ..., -1.8617e+00,
-2.5326e+00, -2.2214e+00],
[ 3.9228e+00, 6.9076e-01, -1.7184e+01, ..., 1.6620e-01,
-8.0480e-03, 7.0134e-01],
[ 4.5737e+00, -2.4532e-01, -1.5081e+01, ..., 5.6133e-01,
1.3404e+00, 1.9455e+00]],
...,
[[-1.8093e+00, -5.0814e+00, -4.0636e+00, ..., -1.5947e+01,
4.3125e-01, -2.1417e+00],
[-2.5201e+00, -6.0513e+00, -5.9640e+00, ..., -5.2693e+00,
6.9141e+00, 1.3844e+00],
[-1.4920e+00, -4.6424e+00, -6.4086e+00, ..., 9.3324e-01,
3.8468e+00, -8.7458e+00],
...,
[-3.3755e+00, -1.1108e+01, -1.3662e+01, ..., -5.2018e-01,
-1.2260e+00, -3.4971e+00],
[-3.5833e+00, -1.1681e+01, -1.2066e+01, ..., -2.2053e+00,
-1.3337e+00, -4.1558e+00],
[-2.4426e+00, -1.0086e+01, -8.4278e+00, ..., -1.8946e+00,
-5.0564e-01, -3.4945e+00]],
[[ 1.9436e-01, 4.0400e-01, -5.3728e-02, ..., -4.0462e+00,
-2.0731e-01, -5.5963e-01],
[ 1.3560e+00, 5.4664e-01, -4.1218e-01, ..., -1.5144e+01,
-1.7897e+01, -1.1919e+01],
[ 1.0400e+00, -1.1709e+00, -1.9136e+00, ..., -1.5949e+01,
-1.1143e+01, -9.6594e+00],
...,
[-2.6517e+00, -1.8027e+00, 2.8994e+00, ..., 1.5209e+00,
1.0124e+00, 1.3037e+00],
[-1.9981e+00, 3.2383e-02, 3.3715e+00, ..., 1.1631e+00,
9.1460e-01, 2.7834e+00],
[-6.0226e-01, 1.2091e+00, 3.7534e+00, ..., 1.0936e+00,
1.4104e+00, 1.4396e+00]],
[[-6.3993e-01, -3.3992e-01, -7.8323e-01, ..., -3.6307e+00,
-3.2782e+00, -7.1996e-01],
[-1.5730e+00, -1.4985e+00, -1.5710e+00, ..., -2.3897e+01,
-1.0605e+01, -2.2004e+00],
[-2.0226e+00, -2.0456e+00, -9.7668e-01, ..., -1.3626e+01,
-3.7183e+00, 2.5457e+00],
...,
[-4.2567e+00, -5.8712e+00, 5.2899e+00, ..., -4.6633e-01,
-8.4707e-01, -1.9078e-01],
[-4.2782e+00, -5.1624e+00, 4.8899e+00, ..., -9.5436e-01,
-1.1507e+00, -1.1458e-01],
[-3.6283e-01, -7.0502e-01, -3.3314e-01, ..., -2.3417e-01,
-5.8565e-01, -2.7693e-01]]],
[[[ 5.9798e-01, 8.7366e-01, 5.7852e-01, ..., 4.2726e-01,
4.7096e-01, -4.0352e-05],
[-2.2703e-01, -7.2814e-01, -5.6515e-01, ..., -1.5171e-01,
1.9197e-01, -2.4606e-01],
[-1.1974e-01, 1.7982e-02, 4.1909e-01, ..., 1.0811e+00,
4.9617e-01, 6.4860e-01],
...,
[-8.7918e-02, -8.6036e-01, -6.2121e-01, ..., -1.1428e+00,
-1.6051e+00, 5.4499e-01],
[-4.9368e-01, -1.7866e+00, -9.0477e-01, ..., -3.2922e+00,
-3.4953e+00, 5.2842e-01],
[ 3.9749e-01, -6.8224e-01, -1.1140e+00, ..., 1.7150e+00,
6.5233e-01, 8.3443e-01]],
[[-2.2587e+00, -5.5416e-01, -1.8487e+00, ..., -1.6133e+00,
-2.3944e+00, -1.5650e+00],
[-2.2573e+00, -5.5126e-01, -1.5855e+00, ..., -1.1549e+00,
-2.1631e+00, -1.6251e+00],
[-2.1695e+00, -1.4016e+00, -1.6197e+00, ..., -1.3688e+00,
-1.7633e+00, -1.0781e+00],
...,
[-1.1613e+00, -9.5859e-01, -1.5874e+00, ..., -5.3712e+00,
-3.4159e+00, -9.0473e-01],
[-2.0029e+00, -4.1738e-01, -1.1327e+00, ..., -5.3471e+00,
-3.3275e+00, -1.1239e+00],
[-7.2213e-01, -1.3718e+00, -1.9512e+00, ..., -2.9664e+00,
-4.4109e+00, -2.3033e+00]],
[[ 1.8395e+00, 2.3963e+00, 1.6387e+00, ..., 1.5250e+00,
7.0124e-01, 2.0620e+00],
[-2.1580e-01, 3.9336e-01, 6.2355e-01, ..., 7.6808e-01,
3.6715e-01, 2.2537e+00],
[-7.8276e-03, 3.1524e-01, 3.6468e-01, ..., -5.3446e-01,
4.3835e-01, 1.9308e+00],
...,
[ 7.0194e-01, 9.5536e-01, 1.1783e+00, ..., -1.6382e+00,
-1.8161e+00, 9.2227e-01],
[ 5.4760e-01, 1.7269e+00, 1.2793e+00, ..., -4.4355e+00,
-2.3023e+00, 2.2521e+00],
[ 1.5997e+00, 9.7060e-01, 1.5434e+00, ..., -2.1464e+00,
-3.3429e+00, 9.3127e-01]],
...,
[[-1.9252e+00, -4.6878e+00, -2.8381e+00, ..., -1.7965e+00,
-1.5174e+00, -2.8152e+00],
[-2.3274e+00, -5.0466e+00, -2.6282e+00, ..., -1.2891e+00,
-1.5344e+00, -2.6772e+00],
[-1.4401e+00, -5.2459e+00, -2.8373e+00, ..., -2.5175e+00,
-2.7117e+00, -1.8391e+00],
...,
[-1.8288e+00, -5.1359e+00, -2.8197e+00, ..., -5.6266e+00,
4.8090e-01, -8.4476e-01],
[-2.3757e+00, -6.6381e+00, -4.2051e+00, ..., 2.7457e-01,
5.5646e-01, -5.2551e+00],
[-2.6332e+00, -6.9253e+00, -3.4601e+00, ..., -1.9255e+00,
-4.2375e+00, -8.4298e+00]],
[[ 9.3720e-01, 1.6128e+00, 7.9712e-01, ..., 1.1049e+00,
1.4577e+00, 1.3357e+00],
[ 1.1136e+00, 9.4287e-01, 8.7392e-01, ..., 1.0572e+00,
9.7749e-01, 1.8713e+00],
[ 6.4394e-01, 9.1430e-02, 2.7111e-01, ..., 4.2808e-01,
2.8712e-01, 1.2710e+00],
...,
[ 1.4083e+00, -1.9000e-01, 3.7541e-01, ..., -1.6027e+00,
-1.0262e+00, 8.4657e-01],
[ 2.0658e+00, 1.3336e+00, 1.1611e+00, ..., -3.7637e+00,
-1.3807e+00, 1.1957e+00],
[ 1.0207e+00, 3.6819e-01, 3.8597e-01, ..., -6.7548e-01,
-9.1179e-01, -3.9603e-01]],
[[-6.0091e-02, -6.0122e-01, -1.1113e+00, ..., -6.6147e-01,
-9.0735e-01, -1.3950e-01],
[-8.2092e-01, -2.5155e+00, -2.4983e+00, ..., -1.3843e+00,
-8.9773e-01, -4.5303e-01],
[-1.2974e+00, -1.5277e+00, -6.5154e-01, ..., 3.7143e-01,
-4.6945e-01, -2.2862e-01],
...,
[-2.3299e+00, -1.7136e+00, -8.5745e-01, ..., -3.9439e+00,
-8.9485e-01, 2.6578e-01],
[-2.3857e+00, -1.9841e+00, -9.1648e-01, ..., -2.1743e+00,
-2.9068e-01, 9.3232e-01],
[-2.9326e-01, -2.3475e-01, -4.6771e-01, ..., 2.9111e+00,
1.5678e-01, -4.8675e-01]]],
[[[ 7.3981e-01, 1.1023e+00, 7.2529e-01, ..., 3.7208e-01,
3.4293e-01, 1.8298e-01],
[-3.0018e-01, -1.3696e+00, -5.6175e-01, ..., 3.3404e-01,
3.8306e-01, -1.1093e+00],
[ 1.6454e-01, -8.2377e-01, -1.7704e-01, ..., -4.8957e-01,
1.0106e+00, 6.4774e-01],
...,
[ 8.4722e-02, -1.1320e+00, -7.7295e-01, ..., -2.3400e-02,
9.8339e-01, 5.4108e-01],
[-4.5370e-01, -1.8471e+00, -8.1711e-01, ..., 4.4004e-02,
1.3978e+00, -1.5110e-01],
[ 3.9253e-01, -5.7253e-01, -6.9077e-01, ..., -2.4696e-01,
4.0016e-01, 5.9832e-01]],
[[-2.9999e+00, -1.4741e+00, -3.0931e+00, ..., -1.7800e+00,
-2.2139e+00, -1.0328e+00],
[-4.3384e+00, -1.8047e+00, -2.3771e+00, ..., -4.1346e-01,
-1.1628e+00, -2.4948e-01],
[-4.3591e+00, -2.6210e+00, -3.2944e+00, ..., -1.5069e+00,
-1.3955e+00, -7.3718e-01],
...,
[-2.7248e+00, -1.7403e+00, -2.6405e+00, ..., -4.3282e-02,
4.8963e-02, 5.2668e-01],
[-3.6411e+00, -1.3478e+00, -2.3896e+00, ..., -3.0142e-01,
-3.7096e-01, 2.8389e-01],
[-1.3112e+00, -9.1202e-01, -1.7010e+00, ..., -3.6629e-01,
-6.9551e-01, -5.3293e-01]],
[[ 1.6495e+00, 2.7729e+00, 2.3927e+00, ..., 2.8530e+00,
1.3941e+00, 2.8341e+00],
[ 1.1769e-01, 1.2180e+00, 1.5671e+00, ..., 1.9121e+00,
1.2552e-02, 2.5021e+00],
[ 6.7191e-01, 1.4801e+00, 9.6117e-01, ..., 9.7935e-01,
6.2794e-01, 1.8219e+00],
...,
[ 9.6812e-01, 1.2390e+00, 6.2228e-01, ..., 6.8371e-01,
1.0112e-01, 1.4447e+00],
[ 6.4616e-01, 1.8674e+00, 9.5662e-01, ..., 1.0280e+00,
6.5252e-01, 1.5783e+00],
[ 1.5189e+00, 1.5837e+00, 1.6616e+00, ..., 1.4712e+00,
1.1242e+00, 2.0287e+00]],
...,
[[-1.3890e+00, -4.6264e+00, -3.9927e+00, ..., -3.4312e+00,
-2.4897e+00, -3.8478e+00],
[-1.7595e+00, -3.8014e+00, -4.3636e+00, ..., -5.5262e+00,
-3.8732e+00, -4.3451e+00],
[-9.5032e-01, -3.1567e+00, -3.8874e+00, ..., -2.7466e+00,
-2.4288e+00, -1.5996e+00],
...,
[-1.5757e+00, -3.9849e+00, -4.5379e+00, ..., -3.9818e+00,
-2.3114e+00, -1.5295e+00],
[-1.5124e+00, -4.5368e+00, -5.3453e+00, ..., -4.9714e+00,
-3.3583e+00, -3.6962e+00],
[-1.6528e+00, -4.9810e+00, -4.2042e+00, ..., -3.5530e+00,
-2.2193e+00, -2.7849e+00]],
[[ 8.1161e-01, 5.3152e-01, -1.1888e-01, ..., 9.0321e-01,
1.0907e+00, 1.2603e+00],
[ 1.1961e+00, 9.7207e-01, 5.1744e-01, ..., 1.6103e+00,
2.3941e+00, 2.5480e+00],
[ 5.7774e-01, 5.4624e-01, 4.7001e-02, ..., 9.6797e-01,
1.6592e+00, 1.8975e+00],
...,
[ 8.5036e-01, -5.5321e-03, -1.2972e-01, ..., 1.3175e+00,
1.8650e+00, 1.6291e+00],
[ 1.5172e+00, 1.4132e+00, 1.1259e+00, ..., 1.3186e+00,
1.5253e+00, 2.4532e+00],
[ 6.5606e-01, 9.1058e-01, 1.0493e+00, ..., 1.0779e+00,
1.3294e+00, 1.3790e+00]],
[[-2.6421e-01, -1.7451e-01, -1.0333e+00, ..., -2.2457e-01,
-1.7202e-01, 8.6268e-01],
[-1.3882e+00, -1.4500e+00, -2.1169e+00, ..., 1.1314e-01,
-5.3498e-02, 4.0006e-01],
[-1.6579e+00, -8.2369e-01, -1.2099e+00, ..., 2.0752e-01,
1.9490e-01, 2.2113e-01],
...,
[-1.7290e+00, -1.7210e+00, -1.6840e+00, ..., -2.1882e-01,
-3.7907e-01, -3.1552e-01],
[-1.4172e+00, -1.6376e+00, -1.6293e+00, ..., -4.1825e-01,
-4.9201e-01, 2.9739e-01],
[-1.7143e-01, 2.0898e-02, -4.9058e-01, ..., 2.1977e-01,
1.7783e-01, 3.3564e-01]]],
[[[ 1.0448e+00, 1.3251e+00, 1.4298e+00, ..., -1.1098e+00,
-1.3988e-01, -7.3489e-01],
[-5.4425e-01, -2.3610e+00, -1.1497e+00, ..., 2.1725e+00,
1.3847e+00, 4.0627e-01],
[-2.9772e-01, -1.1391e+00, -7.4391e-01, ..., 7.8243e-02,
1.9977e+00, 1.4838e+00],
...,
[ 5.1542e-01, 9.4342e-04, 7.1615e-02, ..., 2.5083e-01,
8.8276e-01, 7.3064e-01],
[ 5.8799e-02, -6.8967e-01, 6.5215e-02, ..., 8.9415e-01,
1.5742e+00, 1.2392e-01],
[ 8.6307e-01, -1.6779e-01, -1.5818e-01, ..., 3.8252e-01,
2.6803e-01, 6.9575e-01]],
[[-2.6948e+00, 2.0297e-01, -1.1243e+00, ..., -1.5584e+00,
-2.4131e+00, -7.4971e-01],
[-3.9760e+00, -2.8524e+00, -3.2685e+00, ..., 1.4787e+00,
-2.0569e-01, -5.1854e-01],
[-3.0566e+00, -2.7564e+00, -3.5143e+00, ..., -1.0651e+00,
-3.8238e-01, -1.3031e-01],
...,
[-7.8746e-01, 5.2367e-02, -5.1832e-01, ..., -7.7286e-01,
7.0494e-03, 8.5718e-01],
[-2.5572e+00, -2.4403e-01, -2.9724e-01, ..., -6.5342e-02,
-5.6890e-02, 6.8110e-01],
[-1.5412e+00, -1.3135e+00, -9.8759e-01, ..., -4.9260e-01,
-6.6296e-01, -1.7602e-01]],
[[ 2.3452e+00, 2.9056e+00, 2.2017e+00, ..., -5.0026e-01,
-3.8886e-01, 1.1291e+00],
[ 8.1525e-01, 7.5438e-01, 2.2368e+00, ..., -3.3767e+00,
-1.4641e+00, 6.9231e-01],
[ 1.3155e+00, 1.5950e+00, 2.3384e+00, ..., -5.6794e+00,
-4.2553e+00, -7.5963e-01],
...,
[ 8.4523e-01, 7.1360e-01, 1.4988e+00, ..., 1.1637e+00,
6.3722e-01, 1.8031e+00],
[ 1.0774e+00, 2.1123e+00, 1.9575e+00, ..., 9.6106e-01,
7.7314e-01, 1.6941e+00],
[ 2.4454e+00, 1.8860e+00, 2.3500e+00, ..., 1.9164e+00,
1.0152e+00, 1.7140e+00]],
...,
[[-2.4063e+00, -6.6791e+00, -5.8228e+00, ..., -7.1658e-01,
-2.2774e-01, -1.9665e-02],
[-3.0240e+00, -7.2678e+00, -5.0914e+00, ..., -8.6205e+00,
-4.8380e+00, -2.3345e+00],
[-2.0280e+00, -5.8010e+00, -3.7702e+00, ..., -1.0259e+01,
-8.7632e+00, -6.0582e+00],
...,
[-2.4648e+00, -4.3453e+00, -2.5217e+00, ..., -5.1701e+00,
-4.6163e+00, -3.9334e+00],
[-2.4581e+00, -5.0924e+00, -3.6742e+00, ..., -4.0819e+00,
-3.1957e+00, -4.9622e+00],
[-2.5997e+00, -5.8927e+00, -2.8619e+00, ..., -2.8429e+00,
-1.7104e+00, -3.3362e+00]],
[[ 4.3433e-01, 1.1730e+00, 6.8646e-01, ..., -1.1668e+00,
9.0000e-03, 1.1360e+00],
[ 1.2651e+00, 7.5070e-01, 1.1506e-01, ..., -2.7450e+00,
-1.5879e+00, 3.3427e-01],
[ 1.2800e+00, 5.6755e-01, 9.6118e-02, ..., -6.5433e-01,
-6.6764e-01, 3.8124e-01],
...,
[ 1.6574e+00, 6.8735e-01, 1.0519e+00, ..., 2.5955e+00,
2.3367e+00, 1.6409e+00],
[ 1.9705e+00, 1.1216e+00, 1.7189e+00, ..., 1.9366e+00,
1.6983e+00, 2.5653e+00],
[ 1.1559e+00, 7.7061e-01, 1.3139e+00, ..., 1.2969e+00,
1.4555e+00, 1.5637e+00]],
[[-5.5260e-01, 1.0256e-01, -1.9163e-01, ..., -1.2700e+00,
-1.1704e+00, -2.7512e-01],
[-1.3775e+00, -2.0021e+00, -1.7186e+00, ..., -5.1634e-02,
-2.7842e-01, 3.9953e-01],
[-2.1987e+00, -1.2303e+00, -6.4218e-01, ..., -4.3675e+00,
-2.4289e+00, -1.2059e+00],
...,
[-2.0776e+00, -1.1956e+00, -6.1290e-01, ..., -3.8869e-01,
-5.5830e-01, -3.5970e-01],
[-1.7242e+00, -1.2898e+00, -6.2251e-01, ..., -3.3763e-01,
-3.6926e-01, 4.4074e-01],
[-2.6067e-01, 3.8649e-01, -2.3364e-01, ..., 2.6714e-01,
1.1498e-02, 6.2185e-01]]],
[[[ 9.7756e-01, 1.5960e+00, 1.1318e+00, ..., 8.2440e-01,
1.0115e+00, 4.7236e-01],
[-8.8620e-01, -2.0632e+00, -8.9708e-01, ..., -3.0094e-01,
-1.7948e-01, -3.2960e-01],
[-1.6662e-02, -8.2878e-01, -2.0210e-02, ..., -6.3692e-01,
-1.1900e-01, 6.3761e-01],
...,
[ 4.6576e-01, -2.8089e-01, -3.0834e-01, ..., -3.5552e-02,
-7.7264e-02, 4.0178e-01],
[-1.8036e-01, -1.0527e+00, -1.0255e-01, ..., -7.9992e-04,
1.3223e-01, -8.9518e-02],
[ 7.7321e-01, -3.3044e-01, -2.1425e-01, ..., -3.2676e-01,
-8.9600e-01, 1.8263e-01]],
[[-4.2306e+00, -1.2448e+00, -2.1446e+00, ..., -1.3842e+00,
-1.9894e+00, -1.7656e+00],
[-4.0955e+00, -2.8038e+00, -2.1093e+00, ..., -1.1977e+00,
-2.1122e+00, -2.0954e+00],
[-2.6672e+00, -1.9566e+00, -2.2977e+00, ..., -1.5070e+00,
-1.6100e+00, -1.3742e+00],
...,
[-7.3014e-01, -6.0798e-01, -1.4655e+00, ..., -4.9110e-01,
-8.9594e-01, -8.5121e-01],
[-2.7535e+00, -3.7089e-01, -9.1842e-01, ..., -6.4835e-01,
-1.1157e+00, -1.2071e+00],
[-2.0050e+00, -1.4849e+00, -1.4495e+00, ..., -6.4665e-01,
-7.0850e-01, -4.9958e-01]],
[[ 3.0183e+00, 4.1133e+00, 3.2505e+00, ..., 1.6563e+00,
8.2339e-01, 2.5204e+00],
[ 3.8688e-01, 8.6852e-01, 2.3330e+00, ..., 1.3449e+00,
6.4380e-01, 2.5736e+00],
[ 8.8821e-01, 1.2513e+00, 1.7670e+00, ..., 4.3819e-01,
1.0484e+00, 1.8269e+00],
...,
[ 3.1538e-01, 9.5359e-01, 1.0565e+00, ..., -3.4047e-01,
5.3707e-01, 1.4929e+00],
[ 3.4960e-01, 2.0690e+00, 1.4934e+00, ..., -1.2643e-01,
8.1021e-01, 1.1715e+00],
[ 1.4292e+00, 1.3262e+00, 1.7048e+00, ..., 6.1039e-01,
1.3601e+00, 1.3739e+00]],
...,
[[-2.1655e+00, -7.5210e+00, -6.0139e+00, ..., -4.9576e+00,
-3.5216e+00, -4.2010e+00],
[-2.7696e+00, -6.6199e+00, -4.6128e+00, ..., -4.9302e+00,
-3.5950e+00, -3.0987e+00],
[-2.2477e-01, -2.8639e+00, -2.9253e+00, ..., -3.3867e+00,
-2.3256e+00, -1.3414e+00],
...,
[-9.6480e-01, -2.3245e+00, -1.8777e+00, ..., -4.7783e+00,
-2.9806e+00, -2.7057e+00],
[-1.3424e+00, -3.7378e+00, -3.4266e+00, ..., -5.0916e+00,
-4.1632e+00, -4.2847e+00],
[-2.2359e+00, -4.9858e+00, -3.0133e+00, ..., -3.3203e+00,
-2.4798e+00, -2.4031e+00]],
[[-7.2884e-01, -7.7729e-02, 2.1923e-01, ..., 1.4657e+00,
1.6927e+00, 1.1880e+00],
[ 1.7749e+00, 1.0726e+00, 1.5488e+00, ..., 1.5220e+00,
1.1754e+00, 1.8142e+00],
[ 1.1908e+00, 1.8397e-01, 5.7562e-01, ..., 1.4870e+00,
6.0738e-01, 1.2809e+00],
...,
[ 1.3114e+00, -1.5764e-01, 8.3096e-01, ..., 1.5730e+00,
6.6300e-01, 8.9866e-01],
[ 1.9090e+00, 9.6369e-01, 1.7153e+00, ..., 1.2611e+00,
7.1910e-01, 1.4372e+00],
[ 1.0323e+00, 3.9822e-01, 9.6232e-01, ..., 7.5576e-01,
6.7988e-01, 1.4409e+00]],
[[-4.9913e-01, 5.7760e-01, 8.0120e-02, ..., -3.1883e-01,
-5.6621e-01, -1.8830e-02],
[-1.0628e+00, -1.1670e+00, -1.3785e+00, ..., -1.2116e+00,
-1.4522e+00, -1.0937e+00],
[-1.9927e+00, -5.3407e-01, -1.2131e-01, ..., -6.0229e-01,
-1.1097e+00, -2.2619e-01],
...,
[-1.7249e+00, -1.4988e+00, -1.0498e+00, ..., -6.2826e-01,
-1.5104e+00, -8.5420e-01],
[-1.7672e+00, -1.1481e+00, -8.7622e-01, ..., -7.3403e-01,
-1.6922e+00, -7.4197e-01],
[-3.6048e-01, 3.4698e-01, -3.0289e-01, ..., -2.2110e-01,
-8.0433e-01, -8.8614e-04]]]], device='cuda:0'),
tensor([[[[ 8.2475e-01, 9.2120e-01, 7.4787e-01, ..., -1.3464e-01,
7.3052e-02, -4.5236e-01],
[-6.4891e-01, -1.2362e+00, -1.3801e-01, ..., 2.1936e-02,
1.1802e+00, -3.1040e-01],
[-2.6489e-02, -2.6916e-01, -1.9105e-01, ..., -3.4938e-01,
7.1628e-01, 9.7734e-01],
...,
[ 1.0087e-01, -5.6440e-01, -6.9653e-01, ..., 4.1221e-01,
9.6650e-01, 5.2600e-01],
[-7.0667e-01, -2.4014e+00, -9.6841e-01, ..., -7.8181e-02,
9.4860e-02, -1.1351e-01],
[-3.1539e-01, -1.0606e+00, 1.3905e+00, ..., -4.4211e-01,
-1.3601e+00, -2.8211e-01]],
[[-3.3262e+00, -1.3955e+00, -3.8798e+00, ..., -2.3564e+00,
-3.2660e+00, -7.9692e-01],
[-3.0123e+00, -2.1958e+00, -2.0339e+00, ..., -1.4455e+00,
-2.0804e+00, -1.4777e-01],
[-1.8961e+00, -1.5796e+00, -1.4510e+00, ..., -1.7193e+00,
-1.5699e+00, -5.0650e-01],
...,
[-1.3489e+00, -1.9564e+00, -8.5197e-01, ..., -2.4165e+00,
-2.8557e+00, -6.7488e-01],
[-3.4206e+00, -1.2438e+00, -1.4824e+00, ..., -2.9152e+00,
-3.1130e+00, -1.2516e+00],
[-2.7557e+00, -3.4577e+00, -3.8114e+00, ..., -2.4483e+00,
-2.2126e+00, -9.9544e-01]],
[[ 2.2435e+00, 2.7156e+00, 2.0699e+00, ..., 2.6147e+00,
5.0843e-01, 2.1984e+00],
[ 3.6372e-01, 3.0864e-01, 2.3223e+00, ..., 1.7361e+00,
-1.9310e-01, 2.1556e+00],
[ 1.0627e+00, 9.6157e-01, 1.9035e+00, ..., 1.4241e+00,
8.6414e-01, 2.2324e+00],
...,
[ 2.2707e+00, 1.7481e+00, -1.9627e+00, ..., -6.7472e-01,
1.0346e+00, 2.1165e+00],
[ 1.5774e+00, 2.3692e+00, -2.9500e+00, ..., -1.0792e+00,
1.1815e+00, 1.6333e+00],
[ 2.4495e+00, 2.6298e+00, -2.2676e+00, ..., 1.1523e+00,
1.8723e+00, 1.6611e+00]],
...,
[[-1.9947e+00, -6.4972e+00, -4.9430e+00, ..., -3.1770e+00,
-2.8009e+00, -3.8205e+00],
[-2.5717e+00, -7.5715e+00, -5.2345e+00, ..., -6.3466e+00,
-4.4414e+00, -4.9609e+00],
[-1.4792e+00, -4.2838e+00, -2.4801e+00, ..., -4.7499e+00,
-2.1375e+00, -1.9756e+00],
...,
[-2.4410e+00, -5.3251e+00, -3.1689e+00, ..., -1.2722e+00,
-4.3076e+00, -5.1965e+00],
[-2.8967e+00, -5.8177e+00, -4.4935e+00, ..., -4.2385e+00,
-5.1560e+00, -6.5139e+00],
[-1.8750e+00, -4.6991e+00, -2.7501e+00, ..., -2.3383e+00,
-2.4924e+00, -3.3085e+00]],
[[-3.6946e-01, 1.3239e-01, -1.0240e+00, ..., 1.3467e+00,
9.3994e-01, 1.5888e+00],
[ 1.1286e+00, 3.8199e-01, -4.8894e-01, ..., 2.0715e+00,
1.7470e+00, 2.4914e+00],
[ 1.0625e+00, -5.9453e-02, 2.0412e-01, ..., 2.0918e+00,
1.4719e+00, 1.7415e+00],
...,
[ 1.2866e+00, 5.6261e-01, 3.8895e-01, ..., 3.3264e-01,
1.0131e+00, 1.8748e+00],
[ 1.9046e+00, 1.9809e+00, 1.3010e+00, ..., 9.5152e-01,
1.1762e+00, 2.1147e+00],
[ 1.4051e-01, 2.8360e-01, 1.2202e+00, ..., 4.7410e-01,
7.5052e-01, 1.7411e+00]],
[[-6.2762e-01, -3.3164e-01, -1.4304e+00, ..., -2.5260e-01,
-4.4005e-01, 6.0659e-01],
[-1.2420e+00, -1.3609e+00, -1.4301e+00, ..., -1.6445e-01,
4.4013e-01, 7.9644e-01],
[-1.8840e+00, -2.2101e-01, -7.3323e-01, ..., -5.5796e-03,
5.0806e-01, 5.9696e-03],
...,
[-1.9197e+00, -2.4641e+00, -4.5182e-01, ..., 1.7833e-01,
-1.1168e+00, -1.0592e+00],
[-1.5595e+00, -2.0730e+00, -1.4647e+00, ..., -1.1650e+00,
-1.7585e+00, -8.8682e-01],
[-6.7085e-01, -3.6484e-01, -1.6730e+00, ..., -4.6628e-01,
-7.3864e-01, -2.4688e-03]]],
[[[ 5.4200e-01, 7.9820e-01, 4.9818e-01, ..., 1.7166e+00,
1.6972e+00, 5.4962e-01],
[-3.2572e-01, -6.8360e-01, -3.8330e-01, ..., -2.4572e+00,
2.5100e-01, 1.9332e-01],
[-1.9003e-01, -2.0983e-01, 2.9589e-01, ..., -1.2433e+00,
7.1199e-01, 1.0137e+00],
...,
[-2.5200e+00, 1.0425e+00, 4.2003e-01, ..., 8.2021e-01,
8.2618e-01, 6.5817e-01],
[-1.0783e+00, -5.3329e+00, 1.6054e+00, ..., 9.8461e-01,
1.5590e+00, 1.1740e-01],
[ 1.8958e+00, -2.5200e+00, -3.0516e+00, ..., 3.1073e-01,
2.3029e-01, 7.5916e-01]],
[[-2.2961e+00, -7.0017e-01, -1.7228e+00, ..., -1.6185e+00,
-3.9752e+00, -2.1496e+00],
[-2.4627e+00, -7.8627e-01, -7.9975e-01, ..., -2.3515e+00,
-4.2832e+00, -2.6976e+00],
[-2.3571e+00, -1.4018e+00, -2.9450e-01, ..., -6.9679e-01,
-2.5108e+00, -1.8874e+00],
...,
[-2.6913e-01, 2.7071e+00, 7.7593e-02, ..., -1.1566e+00,
-1.5736e-01, 6.9339e-01],
[-4.4891e+00, 1.9569e+00, 1.6468e+00, ..., 2.2066e-01,
2.2812e-03, 6.8194e-01],
[-3.4634e+00, -5.7028e+00, -6.2918e+00, ..., -4.9054e-01,
-6.1254e-01, -1.9371e-01]],
[[ 1.7714e+00, 2.4188e+00, 1.4333e+00, ..., -1.1853e+00,
-1.1136e+00, 1.2805e+00],
[-2.0206e-01, 4.9419e-01, 4.4888e-01, ..., -2.6170e+00,
-1.5860e+00, 1.8428e+00],
[ 5.1150e-01, 5.3918e-01, -9.2676e-02, ..., -2.3260e+00,
1.7913e-01, 1.4540e+00],
...,
[-7.7156e+00, -1.3406e+00, 5.1966e-01, ..., 3.3277e-01,
1.6997e-01, 1.6333e+00],
[-6.0633e+00, -3.9710e+00, -2.6701e+00, ..., 1.2284e+00,
7.3955e-01, 1.7406e+00],
[ 2.1579e+00, -1.1446e+00, -2.9644e+00, ..., 1.9545e+00,
1.0630e+00, 1.7781e+00]],
...,
[[-1.7910e+00, -4.6880e+00, -2.8029e+00, ..., -3.1577e+00,
-3.9073e+00, -3.6138e+00],
[-2.1195e+00, -4.9329e+00, -2.7136e+00, ..., 6.5126e-02,
-3.1520e+00, -3.7627e+00],
[-1.3270e+00, -4.6702e+00, -2.3324e+00, ..., -7.5757e-02,
-4.2381e+00, -3.7936e+00],
...,
[-1.5004e+00, -9.6251e+00, -5.2151e+00, ..., -2.7160e+00,
-2.1182e+00, -2.7680e+00],
[-2.6494e+00, -8.6421e+00, -1.0860e+01, ..., -4.8569e+00,
-3.4590e+00, -4.9443e+00],
[-2.5395e+00, -6.2692e+00, -6.2998e+00, ..., -3.2468e+00,
-1.6296e+00, -3.3646e+00]],
[[ 8.2949e-01, 1.4341e+00, 8.5847e-01, ..., -1.0907e+00,
3.0340e-01, 1.3075e+00],
[ 1.1054e+00, 1.0106e+00, 8.8592e-01, ..., -2.2419e+00,
-1.3823e+00, 1.8811e+00],
[ 5.8228e-01, -1.3080e-01, 1.0763e-01, ..., -1.1122e+00,
-1.2070e+00, 1.6234e+00],
...,
[ 1.1842e+00, -4.2432e+00, -4.5634e+00, ..., 1.6431e+00,
1.8869e+00, 1.5013e+00],
[ 3.5246e+00, 1.1683e-01, -1.3312e+00, ..., 2.0490e+00,
1.6874e+00, 2.5555e+00],
[-4.8264e-01, -1.1012e+00, -1.7183e+00, ..., 1.3203e+00,
1.5153e+00, 1.5231e+00]],
[[-1.0412e-01, -6.8256e-01, -1.0534e+00, ..., 1.5174e+00,
-5.9551e-01, -6.8891e-02],
[-1.0135e+00, -2.5606e+00, -2.1023e+00, ..., -1.4642e+00,
-1.2414e+00, -1.1728e+00],
[-1.6945e+00, -1.5663e+00, -2.8389e-01, ..., 1.6293e-03,
-1.3786e-01, -7.8195e-01],
...,
[-2.8694e+00, -2.7834e+00, -2.4197e+00, ..., -1.1578e+00,
-2.9021e-01, -4.2730e-01],
[-3.1697e+00, -1.8388e+00, -1.0658e+00, ..., -1.2030e-01,
-2.2296e-01, 4.1847e-01],
[ 2.5352e-01, 3.2251e-01, -1.0346e+00, ..., 2.3638e-01,
1.3969e-02, 6.2970e-01]]],
[[[ 9.3891e-01, 1.5923e+00, 1.1113e+00, ..., 7.6420e-01,
9.7628e-01, 4.4642e-01],
[-9.4647e-01, -1.9669e+00, -6.6672e-01, ..., -2.9231e-01,
-7.6998e-02, -2.9939e-01],
[-1.6183e-02, -6.2160e-01, 3.9435e-03, ..., -6.9613e-01,
-1.7782e-01, 7.1499e-01],
...,
[ 4.0989e-01, -1.3139e-01, -3.5394e-01, ..., 5.5752e-01,
6.0228e-01, 8.4633e-01],
[-1.6850e-01, -9.4145e-01, -1.5751e-01, ..., 1.0219e-02,
2.1505e-01, -9.3388e-02],
[ 7.2008e-01, -3.0158e-01, -1.3921e-01, ..., -2.6963e-01,
-8.2898e-01, 1.9770e-01]],
[[-4.3397e+00, -1.2828e+00, -2.1679e+00, ..., -1.3763e+00,
-1.8866e+00, -1.5777e+00],
[-4.2225e+00, -2.8344e+00, -2.4943e+00, ..., -1.0307e+00,
-1.8887e+00, -1.7795e+00],
[-2.7611e+00, -2.0541e+00, -4.2322e+00, ..., -1.5147e+00,
-1.5511e+00, -1.2410e+00],
...,
[-8.8053e-01, -4.8982e-01, -1.2889e+00, ..., -6.8282e-02,
-2.2964e-01, -7.2938e-01],
[-2.8417e+00, -2.8494e-01, -7.6207e-01, ..., -5.3719e-01,
-9.5643e-01, -1.1885e+00],
[-2.0211e+00, -1.4580e+00, -1.3681e+00, ..., -4.9404e-01,
-5.0230e-01, -4.8645e-01]],
[[ 3.0737e+00, 4.1302e+00, 3.2225e+00, ..., 1.6735e+00,
9.1342e-01, 2.3894e+00],
[ 4.1082e-01, 7.7331e-01, 2.2196e+00, ..., 1.2837e+00,
6.1547e-01, 2.3677e+00],
[ 8.4052e-01, 1.2263e+00, 1.9629e+00, ..., 2.6327e-01,
9.4441e-01, 1.6927e+00],
...,
[ 6.0196e-02, 7.8656e-01, 1.0991e+00, ..., 1.3867e-01,
9.3162e-01, 1.6975e+00],
[ 1.4898e-01, 2.0581e+00, 1.4369e+00, ..., -1.5804e-01,
8.6071e-01, 1.2670e+00],
[ 1.3601e+00, 1.3886e+00, 1.6600e+00, ..., 6.4487e-01,
1.4881e+00, 1.4611e+00]],
...,
[[-2.1184e+00, -7.5555e+00, -6.1073e+00, ..., -4.8462e+00,
-3.4752e+00, -3.7616e+00],
[-2.7518e+00, -6.5883e+00, -4.8773e+00, ..., -5.0246e+00,
-3.7633e+00, -2.8176e+00],
[-2.1839e-01, -2.8924e+00, -3.9118e+00, ..., -3.1034e+00,
-2.1291e+00, -1.0060e+00],
...,
[-9.3850e-01, -2.1672e+00, -1.5750e+00, ..., -4.2539e+00,
-3.2522e+00, -3.1949e+00],
[-1.2758e+00, -3.6222e+00, -3.1529e+00, ..., -5.1914e+00,
-4.0372e+00, -4.3417e+00],
[-2.1598e+00, -4.9788e+00, -2.9644e+00, ..., -3.3892e+00,
-2.4146e+00, -2.4757e+00]],
[[-8.0862e-01, -1.4940e-01, 1.8359e-01, ..., 1.5360e+00,
1.6766e+00, 1.1943e+00],
[ 1.7303e+00, 1.0314e+00, 1.3512e+00, ..., 1.6327e+00,
1.1423e+00, 1.7646e+00],
[ 1.2333e+00, -1.1411e-01, 2.7353e-03, ..., 1.2195e+00,
4.4418e-01, 1.0443e+00],
...,
[ 1.2392e+00, -6.8933e-02, 8.7103e-01, ..., 1.5236e+00,
8.6918e-01, 1.1509e+00],
[ 1.8078e+00, 8.6580e-01, 1.7760e+00, ..., 1.2027e+00,
6.9603e-01, 1.4610e+00],
[ 1.0083e+00, 2.7626e-01, 9.3773e-01, ..., 7.6831e-01,
7.3683e-01, 1.4908e+00]],
[[-4.8643e-01, 6.0772e-01, 1.0358e-01, ..., -3.0952e-01,
-6.1258e-01, -4.7901e-02],
[-1.1129e+00, -1.1034e+00, -1.6007e+00, ..., -1.0498e+00,
-1.4799e+00, -1.0579e+00],
[-2.1292e+00, -7.6094e-01, -5.5713e-01, ..., -5.5276e-01,
-1.1018e+00, -3.2477e-01],
...,
[-1.6860e+00, -1.3070e+00, -1.0042e+00, ..., -7.3091e-01,
-9.6359e-01, -8.5238e-01],
[-1.6388e+00, -9.3485e-01, -8.7049e-01, ..., -8.4806e-01,
-1.7025e+00, -7.6540e-01],
[-3.0284e-01, 4.0117e-01, -3.6880e-01, ..., -1.9180e-01,
-7.5934e-01, 6.2332e-02]]],
[[[ 6.8227e-01, 1.0422e+00, 5.5362e-01, ..., -1.3500e+00,
-7.9233e-01, -1.6833e+00],
[-4.4642e-01, -1.5574e+00, -6.2290e-01, ..., 1.2227e+00,
1.7611e+00, -3.5051e-01],
[ 1.8947e-01, -8.7191e-01, -2.5442e-01, ..., -3.2676e+00,
-4.6644e+00, -2.9511e+00],
...,
[-1.4877e+00, -8.9416e-01, -3.8611e-01, ..., -2.5564e+00,
-1.4838e+00, -3.1049e+00],
[-1.1723e+00, -1.2085e+00, 2.1761e-01, ..., 1.7595e+00,
2.5103e+00, 1.6065e+00],
[ 4.4545e-02, -8.3588e-02, 7.8556e-01, ..., -1.9827e-03,
-2.2637e-01, 4.6089e-01]],
[[-3.5104e+00, -1.6849e+00, -3.1127e+00, ..., -5.5576e+00,
-8.8364e+00, -3.3976e+00],
[-4.5734e+00, -1.8809e+00, -2.3054e+00, ..., -2.9999e+00,
-5.9700e+00, -1.6931e+00],
[-4.5471e+00, -3.2954e+00, -3.1698e+00, ..., -2.4205e+00,
-2.5961e+00, -1.9250e+00],
...,
[-3.2989e+00, -1.7408e+00, -1.1862e+00, ..., -3.5270e+00,
-1.0571e+01, -1.9407e+00],
[-4.5425e+00, -1.3170e+00, 1.5610e-01, ..., -1.3376e-01,
-8.5066e-01, -1.5518e-01],
[-1.8373e+00, -1.9331e+00, -1.3265e+00, ..., -9.1980e-01,
-7.5121e-01, -8.3735e-02]],
[[ 2.0032e+00, 3.0222e+00, 2.4327e+00, ..., 3.1213e+00,
-2.0282e-01, 3.9088e+00],
[ 4.1474e-01, 1.4337e+00, 1.7748e+00, ..., 1.1870e-01,
-7.6671e+00, -8.6059e+00],
[ 9.9690e-01, 1.4419e+00, 1.1809e+00, ..., -7.6916e-01,
-8.7863e+00, -6.4363e+00],
...,
[ 2.6943e+00, 1.0955e-01, -5.6977e+00, ..., -1.0944e+00,
-6.8201e+00, -5.1347e+00],
[ 1.2180e+00, 1.4602e+00, -8.0276e-01, ..., 1.5116e+00,
2.0976e+00, 1.4776e+00],
[ 2.0873e+00, 2.0239e+00, 1.0179e+00, ..., 8.1500e-01,
5.6292e-01, 1.1429e+00]],
...,
[[-1.6228e+00, -4.9833e+00, -4.1425e+00, ..., -3.6239e+00,
-3.1827e+00, -5.3777e+00],
[-2.0790e+00, -4.7084e+00, -4.9068e+00, ..., -8.0240e+00,
-8.9951e+00, -1.2749e+01],
[-1.2621e+00, -4.2633e+00, -4.4547e+00, ..., -1.0608e+01,
-5.2663e+00, -1.1299e+01],
...,
[-2.3252e+00, -4.6832e+00, -3.4146e+00, ..., -4.3956e+00,
-5.4764e+00, -6.2633e+00],
[-2.1844e+00, -5.3905e+00, -3.9187e+00, ..., -7.4545e+00,
-4.6710e+00, -8.1774e+00],
[-2.0380e+00, -5.5350e+00, -3.3950e+00, ..., -4.5333e+00,
-2.0469e+00, -3.0330e+00]],
[[ 5.5683e-01, 4.1531e-01, -1.2326e-01, ..., -2.5017e+00,
-7.0886e+00, -5.7817e-01],
[ 1.2720e+00, 1.0073e+00, 5.2018e-01, ..., -3.4629e-01,
-6.2066e+00, -8.9146e-01],
[ 3.9984e-01, 6.3155e-01, -8.5659e-02, ..., 3.6913e-01,
7.8435e-01, -4.0046e-01],
...,
[ 1.1352e-01, 1.4337e-01, 1.5012e+00, ..., 2.1481e+00,
-2.0311e-01, 1.8943e+00],
[ 1.1412e+00, 1.3885e+00, 2.5338e+00, ..., 3.4790e+00,
2.4495e+00, 3.3233e+00],
[ 3.9881e-01, 1.6136e-01, 5.9446e-01, ..., 1.2447e+00,
1.2462e+00, 1.3177e+00]],
[[-3.8348e-01, -1.5049e-01, -9.5381e-01, ..., -1.5790e+00,
-6.6319e-01, -9.3420e-01],
[-1.5159e+00, -1.3192e+00, -1.9744e+00, ..., -1.3701e+00,
2.0835e-01, 4.8439e+00],
[-2.0570e+00, -5.1287e-01, -4.3600e-01, ..., -1.7999e+00,
-9.2492e-02, 4.1763e+00],
...,
[-1.9848e+00, -1.3767e+00, -7.8509e-01, ..., -3.2121e+00,
-5.6792e+00, -3.5630e+00],
[-2.0508e+00, -1.2118e+00, -3.2855e-01, ..., -1.1503e+00,
-9.2677e-01, 5.8654e-02],
[-2.9100e-01, 1.4376e-01, -2.4688e-01, ..., -1.0653e+00,
-7.5745e-01, 3.6270e-02]]],
[[[-7.1578e-02, 7.8239e-02, 1.1081e+00, ..., -2.4527e-01,
-9.8903e-02, -7.9009e-01],
[ 5.3381e-02, -7.5743e-01, -2.5549e+00, ..., -6.1720e-02,
9.9503e-01, -9.4689e-01],
[-1.3593e-01, 4.3394e-02, -5.4040e-01, ..., 6.3337e-01,
1.7197e+00, 2.1122e+00],
...,
[ 4.9382e-01, 5.3109e-01, 2.6853e-01, ..., -8.5415e-03,
4.1201e-01, 6.3218e-01],
[-1.1679e-01, -5.4701e-01, 3.4653e-01, ..., -6.0689e-01,
-4.4210e-01, -2.1461e-01],
[-1.9345e-01, -9.4402e-01, -3.3924e-01, ..., -1.1649e+00,
-1.5620e+00, -3.1469e-01]],
[[-9.0584e-01, 1.8029e+00, 1.0425e+00, ..., -2.5643e+00,
-3.6782e+00, -8.8118e-01],
[-2.6312e-02, -1.1133e+00, -1.7064e+00, ..., -2.6708e+00,
-3.9841e+00, -1.2098e+00],
[ 5.3691e-01, -5.5451e-01, -6.1163e-01, ..., -8.9273e-01,
-2.2556e+00, -1.5974e+00],
...,
[-3.8909e-01, 1.6976e+00, 1.2218e+00, ..., -1.7715e+00,
-1.8257e+00, -6.3773e-01],
[-2.3917e+00, 3.7966e-01, -3.9188e-01, ..., -2.3709e+00,
-2.2799e+00, -1.2449e+00],
[-2.1546e+00, -1.5461e+00, -2.6858e+00, ..., -1.9927e+00,
-1.9088e+00, -1.0125e+00]],
[[ 1.3408e+00, -1.0023e-01, -8.4644e-01, ..., 2.6607e+00,
5.1522e-01, 2.3066e+00],
[-1.4787e+00, -3.4939e+00, -2.5340e+00, ..., 1.9426e+00,
-1.0876e-01, 2.3853e+00],
[-1.8533e+00, -1.0545e+00, -8.6764e-01, ..., -2.1526e-03,
-3.3525e-01, 1.0847e+00],
...,
[ 8.3054e-01, -6.3229e-01, -3.4304e-01, ..., 1.7386e-01,
1.0603e+00, 2.2529e+00],
[ 3.2624e-01, 8.7258e-01, 2.5223e-01, ..., 9.7894e-03,
8.6549e-01, 1.7322e+00],
[ 1.7284e+00, 1.8168e+00, 1.7525e+00, ..., 1.2746e+00,
1.7075e+00, 1.7269e+00]],
...,
[[-1.7609e+00, -5.3976e+00, -5.4014e+00, ..., -3.4263e+00,
-2.9004e+00, -3.7877e+00],
[-3.3677e+00, -1.0007e+01, -3.1689e+00, ..., -4.8368e+00,
-3.7822e+00, -4.7510e+00],
[-2.0518e+00, -7.8238e+00, -1.2629e+00, ..., -5.0694e+00,
-3.6690e+00, -3.2822e+00],
...,
[-2.3755e+00, -5.4733e+00, -3.0474e+00, ..., -5.6696e+00,
-4.4381e+00, -4.7817e+00],
[-2.7125e+00, -5.6380e+00, -4.7207e+00, ..., -6.2777e+00,
-5.6737e+00, -6.0755e+00],
[-1.3704e+00, -3.8050e+00, -2.2555e+00, ..., -3.3149e+00,
-2.7048e+00, -3.1300e+00]],
[[ 2.3640e-01, 1.7411e+00, 9.8023e-01, ..., 4.2129e-01,
8.3065e-01, 1.7819e+00],
[ 2.3170e-01, -2.0757e-02, -8.8241e-01, ..., -4.6229e-01,
6.9619e-01, 2.4747e+00],
[-2.6092e-01, -2.2080e+00, -1.1769e+00, ..., 6.1013e-01,
-2.5817e-01, 1.0377e+00],
...,
[ 1.4217e+00, 9.1488e-01, 1.3486e-01, ..., 1.3028e+00,
1.1615e+00, 1.4546e+00],
[ 2.0853e+00, 2.0314e+00, 1.2591e+00, ..., 1.5082e+00,
1.2770e+00, 1.8952e+00],
[ 5.0131e-01, 6.8678e-01, 4.6343e-01, ..., 5.1255e-01,
6.3959e-01, 1.5810e+00]],
[[-2.8141e-01, 2.3333e-01, 8.4753e-01, ..., -6.7360e-01,
-5.8405e-01, 4.1692e-01],
[ 8.3977e-01, -2.5588e+00, -3.4972e+00, ..., -6.3740e-01,
6.3808e-01, 9.6796e-01],
[-6.3180e-01, -2.4536e+00, -1.6355e+00, ..., 1.1807e+00,
1.6440e+00, 1.2096e+00],
...,
[-1.4341e+00, -8.7950e-01, -9.8419e-01, ..., -5.7830e-01,
-8.9309e-01, -6.0105e-01],
[-1.2357e+00, -1.4811e+00, -6.8230e-01, ..., -1.2268e+00,
-1.4336e+00, -7.4073e-01],
[-6.7927e-01, -6.1488e-01, -8.1650e-01, ..., -6.5198e-01,
-9.4066e-01, -1.2012e-01]]]], device='cuda:0'),
tensor([[[[ 8.0899e-01, 9.7635e-01, 5.5945e-01, ..., 7.2728e-01,
9.5604e-01, 4.2106e-01],
[-1.2906e+00, -1.1198e+00, -1.2181e-01, ..., -1.9634e-01,
2.8280e-02, -3.1831e-01],
[-1.9588e-01, -1.0583e+00, -7.8001e-02, ..., -9.2688e-01,
-2.4743e-01, 7.0446e-01],
...,
[ 3.9680e-01, -1.5959e-01, -1.1487e-01, ..., 5.3918e-01,
4.5986e-01, 4.9307e-01],
[-1.0983e-01, -9.2537e-01, -5.9064e-01, ..., 1.1929e-01,
7.8436e-01, 9.6070e-02],
[ 4.7235e-01, -2.3646e-01, -7.7774e-01, ..., -2.8801e-01,
-5.7994e-01, 3.8009e-01]],
[[-4.1412e+00, -2.2076e+00, -3.1064e+00, ..., -1.6166e+00,
-1.9884e+00, -1.6414e+00],
[-3.3239e+00, -1.6887e+00, -7.8315e-01, ..., -1.7112e+00,
-2.1264e+00, -2.0249e+00],
[-2.7045e+00, -1.8348e+00, -1.4976e+00, ..., -1.6422e+00,
-1.5398e+00, -1.3927e+00],
...,
[-5.0765e-01, -7.0692e-01, -8.6987e-01, ..., -8.1655e-01,
1.9320e-01, -2.1372e-01],
[-1.1068e+00, -4.5170e-01, -6.6384e-01, ..., -8.6842e-01,
-5.3039e-01, -1.0287e+00],
[ 1.4928e-01, -6.3991e-01, -7.8053e-01, ..., -6.2567e-01,
-6.1577e-01, -5.8247e-01]],
[[ 2.7733e+00, 3.4357e+00, 3.3274e+00, ..., 1.0822e+00,
9.8346e-01, 2.5004e+00],
[ 7.3044e-01, 1.1835e+00, 2.6566e+00, ..., 7.4112e-01,
8.1348e-01, 2.5362e+00],
[ 9.5647e-01, 5.1154e-01, 1.7838e+00, ..., 4.9116e-01,
1.2448e+00, 1.8835e+00],
...,
[ 3.5897e-01, 5.4739e-01, 1.5814e+00, ..., -4.2950e-01,
-2.1881e-02, 9.1415e-01],
[ 2.8115e-01, 1.6484e+00, 1.5665e+00, ..., -4.2367e-01,
7.3259e-01, 1.1996e+00],
[ 1.6524e+00, 1.1373e+00, 1.8492e+00, ..., 5.3788e-01,
1.4347e+00, 1.5651e+00]],
...,
[[-2.8725e+00, -8.1966e+00, -5.3230e+00, ..., -5.1380e+00,
-3.5054e+00, -4.0140e+00],
[-3.4911e+00, -9.5644e+00, -6.3433e+00, ..., -4.8485e+00,
-3.7813e+00, -3.1184e+00],
[-1.7844e+00, -6.0641e+00, -3.5700e+00, ..., -3.8789e+00,
-2.1879e+00, -9.6074e-01],
...,
[-2.5024e+00, -6.2034e+00, -3.1739e+00, ..., -5.1462e+00,
-2.7522e+00, -2.4641e+00],
[-3.1075e+00, -7.4120e+00, -4.5588e+00, ..., -5.7496e+00,
-4.2050e+00, -4.3695e+00],
[-2.4126e+00, -6.6273e+00, -3.4722e+00, ..., -3.7702e+00,
-2.5217e+00, -2.5188e+00]],
[[-8.8636e-01, -1.8012e-01, 2.2931e-01, ..., 1.6166e+00,
1.6306e+00, 1.3084e+00],
[ 1.4544e+00, 1.1137e+00, 1.2808e+00, ..., 1.8602e+00,
1.0916e+00, 1.8426e+00],
[ 8.3973e-01, 2.3410e-01, 4.3299e-01, ..., 1.6407e+00,
6.0355e-01, 1.1427e+00],
...,
[ 1.2113e+00, -1.5653e-01, 3.5106e-01, ..., 2.2298e+00,
8.4722e-01, 7.5076e-01],
[ 1.8507e+00, 1.1743e+00, 1.2580e+00, ..., 1.6185e+00,
6.4294e-01, 1.3405e+00],
[ 1.1523e+00, 9.1549e-01, 1.1738e+00, ..., 8.5181e-01,
6.9155e-01, 1.4601e+00]],
[[-7.5300e-01, -8.3241e-01, -6.0726e-01, ..., -1.7212e-01,
-5.8010e-01, -3.9016e-02],
[-1.2811e+00, -1.4062e+00, -6.7567e-01, ..., -1.1655e+00,
-1.2178e+00, -1.0868e+00],
[-2.2312e+00, -1.8488e-01, 9.0907e-02, ..., -9.6940e-01,
-8.7687e-01, -3.8091e-01],
...,
[-2.0332e+00, -2.1081e+00, -3.7303e-01, ..., -1.1673e+00,
-1.1541e+00, -1.0092e+00],
[-1.7970e+00, -1.8779e+00, -9.6484e-01, ..., -1.3510e+00,
-1.4323e+00, -8.8537e-01],
[ 7.3636e-02, 1.1533e-01, -1.9620e-01, ..., -3.8002e-01,
-6.6162e-01, 6.3707e-02]]],
[[[ 8.8303e-01, 1.5209e+00, 1.2258e+00, ..., 1.5122e+00,
1.1434e+00, 1.0512e+00],
[-9.5112e-01, -1.9638e+00, -1.2066e+00, ..., -4.7583e-01,
-7.1588e-01, -1.6721e+00],
[ 1.8055e-01, -4.6032e-01, 6.4091e-01, ..., -7.7457e-01,
7.9120e-01, 6.3625e-01],
...,
[-1.9725e-01, -1.0728e+00, 3.1856e-01, ..., -3.5141e+00,
-1.5640e+00, -3.3989e-01],
[-6.0422e-01, -1.8823e+00, -1.4694e+00, ..., -1.2412e+00,
-5.5087e-01, -2.8964e-01],
[-2.4781e-01, -1.6583e+00, -1.6475e+00, ..., -1.1647e+00,
-1.5266e+00, -2.7604e-01]],
[[-3.6273e+00, -7.7851e-01, -1.5324e+00, ..., -1.5716e+00,
-2.4968e+00, -1.2518e+00],
[-3.9281e+00, -3.0173e+00, -1.8511e+00, ..., -2.2091e+00,
-3.3127e+00, -9.7095e-01],
[-2.7039e+00, -3.2273e+00, -3.2800e+00, ..., -1.8121e+00,
-2.7265e+00, -1.0915e+00],
...,
[-1.7017e+00, -2.3327e+00, -2.4600e+00, ..., -6.9634e+00,
-4.1396e+00, -1.2282e+00],
[-3.0107e+00, -1.6855e+00, -2.9703e+00, ..., -2.8216e+00,
-1.7999e+00, -8.7200e-01],
[-2.2192e+00, -1.9700e+00, -2.7092e+00, ..., -1.9679e+00,
-1.9683e+00, -1.0080e+00]],
[[ 2.7764e+00, 4.2518e+00, 3.3886e+00, ..., 2.8528e+00,
1.0559e+00, 3.0844e+00],
[ 7.2132e-02, 1.1508e+00, 2.3202e+00, ..., 9.8725e-01,
-1.0960e+00, 2.7459e+00],
[ 1.0177e+00, 1.5705e+00, 1.7055e+00, ..., 1.9824e+00,
5.2158e-02, 1.7508e+00],
...,
[ 9.1347e-01, 2.1373e+00, 2.8802e+00, ..., -1.0939e-01,
1.4094e+00, 3.0075e+00],
[ 7.4081e-01, 1.8438e+00, 1.2785e+00, ..., 7.0084e-01,
9.6424e-01, 1.4335e+00],
[ 1.7048e+00, 1.6613e+00, 1.8811e+00, ..., 1.2624e+00,
1.7886e+00, 1.8016e+00]],
...,
[[-1.7627e+00, -7.1994e+00, -5.4230e+00, ..., -4.0262e+00,
-3.4250e+00, -5.7641e+00],
[-2.2763e+00, -6.9995e+00, -4.1484e+00, ..., -3.3748e+00,
-2.6888e+00, -4.4162e+00],
[-1.0476e+00, -5.2049e+00, -5.5205e+00, ..., -2.4550e+00,
-3.5126e+00, -3.0155e+00],
...,
[-2.8131e+00, -5.6150e+00, -5.6785e+00, ..., -6.9133e+00,
-6.1432e+00, -7.8157e+00],
[-2.7079e+00, -5.6004e+00, -5.4505e+00, ..., -1.1523e+01,
-9.4740e+00, -8.0362e+00],
[-1.4552e+00, -4.0791e+00, -2.8202e+00, ..., -3.2279e+00,
-2.7242e+00, -3.1653e+00]],
[[-1.4898e-01, 2.9933e-01, 6.0411e-01, ..., 5.4063e-01,
8.3888e-01, 1.4080e+00],
[ 1.5441e+00, 1.0135e+00, 1.9197e+00, ..., 1.5783e+00,
2.4386e+00, 3.0831e+00],
[ 6.6895e-01, -1.4673e-01, 7.1326e-01, ..., 1.9528e+00,
1.6580e+00, 2.3802e+00],
...,
[ 1.4610e+00, 4.4721e-01, 7.1709e-02, ..., -1.1496e+00,
2.0401e+00, 2.7036e+00],
[ 1.9080e+00, 1.7015e+00, 8.9104e-01, ..., 2.9457e+00,
2.0887e+00, 2.1917e+00],
[ 4.7919e-01, 7.8505e-01, 6.9538e-01, ..., 4.6085e-01,
6.4392e-01, 1.6509e+00]],
[[-5.2254e-01, 2.5929e-01, 1.7335e-01, ..., 6.1178e-01,
5.5748e-01, 1.3924e+00],
[-1.3018e+00, -1.8494e+00, -1.6576e+00, ..., -1.0247e+00,
-6.1299e-01, 1.9054e-01],
[-2.2998e+00, -8.6700e-01, 5.5680e-01, ..., -3.5711e-01,
-2.9702e-01, 6.4560e-01],
...,
[-1.7569e+00, -1.7261e+00, -1.3509e+00, ..., -5.3048e+00,
-2.8275e+00, -1.5019e-01],
[-1.6286e+00, -1.9273e+00, -1.6667e+00, ..., -2.4256e-01,
-2.3039e-01, -1.1974e-01],
[-7.8458e-01, -8.0211e-01, -1.1431e+00, ..., -5.6746e-01,
-8.7496e-01, -9.2352e-02]]],
[[[ 9.7728e-01, 1.6417e+00, 1.0962e+00, ..., 4.3145e-01,
7.9911e-01, 4.0595e-01],
[-6.8764e-01, -2.1558e+00, -1.3627e+00, ..., 1.6834e-02,
-1.5791e-01, -3.2172e-01],
[ 1.1567e-01, -9.8666e-01, 5.2612e-02, ..., 1.3570e-01,
-4.4647e-01, 4.9999e-01],
...,
[ 1.2597e+00, -5.0423e+00, -1.7882e+00, ..., 2.9044e-01,
9.6046e-01, 1.6899e-01],
[-2.2098e-01, -2.9708e+00, 9.4656e-01, ..., 7.7296e-01,
5.9177e-01, 1.2628e-01],
[ 1.2534e+00, -2.9154e+00, 1.5774e-02, ..., 3.7194e-01,
-2.0107e-01, 3.1641e-01]],
[[-4.2121e+00, -1.4964e+00, -2.7554e+00, ..., -1.9558e+00,
-2.1343e+00, -1.5074e+00],
[-4.5988e+00, -3.9870e+00, -2.0843e+00, ..., -1.1267e+00,
-1.3505e+00, -1.3988e+00],
[-5.1848e+00, -2.9354e+00, -5.5001e-02, ..., -1.4108e+00,
-5.9325e-01, -6.4317e-01],
...,
[-4.2416e+00, -2.8772e+00, -1.7606e+01, ..., 9.5199e-01,
4.8617e-01, -1.0291e-01],
[-2.8692e+00, -6.0342e-01, -1.2467e+01, ..., 1.0659e+00,
-1.7460e-01, -8.9530e-01],
[-3.2222e+00, -2.7966e+00, -7.6083e+00, ..., -1.2277e-01,
-5.4029e-01, -3.8069e-01]],
[[ 3.1566e+00, 4.5838e+00, 3.5020e+00, ..., 1.9950e+00,
7.6898e-01, 2.4558e+00],
[ 5.0997e-01, 1.1878e+00, 1.6610e+00, ..., 1.8458e+00,
5.5820e-01, 2.1491e+00],
[ 4.8831e-01, 2.1622e+00, -5.2817e-01, ..., 4.9979e-01,
2.2134e-02, 1.0050e+00],
...,
[-1.3843e+01, -5.8900e+00, -8.9336e+00, ..., -2.8938e+00,
3.9074e-02, 7.0405e-01],
[-5.8141e+00, -3.9386e-01, -9.1620e-01, ..., -1.5791e+00,
1.1431e-01, 7.4542e-01],
[-5.2425e-01, -2.1905e+00, -2.5645e-01, ..., 4.2546e-01,
1.1565e+00, 1.3633e+00]],
...,
[[-2.0393e+00, -7.4660e+00, -6.4957e+00, ..., -4.3513e+00,
-3.6673e+00, -4.0517e+00],
[-2.8409e+00, -7.0610e+00, -5.9623e+00, ..., -5.3320e+00,
-4.1793e+00, -3.3078e+00],
[-1.3283e+00, -4.9936e+00, -5.2437e+00, ..., -3.4103e+00,
-2.2119e+00, -4.9044e-01],
...,
[-1.1152e+00, -1.1553e+01, -1.3094e+01, ..., -8.0588e+00,
-2.9139e+00, -2.3214e+00],
[-2.2588e+00, -6.4416e+00, -1.0766e+01, ..., -8.5230e+00,
-4.1368e+00, -4.0381e+00],
[-1.8678e+00, -6.8634e+00, -7.6552e+00, ..., -4.9823e+00,
-2.0380e+00, -2.2855e+00]],
[[-7.2545e-01, -3.2443e-01, -7.0194e-01, ..., 1.6241e+00,
1.4500e+00, 1.2376e+00],
[ 1.5410e+00, 4.2257e-01, 9.6089e-01, ..., 1.6495e+00,
1.3543e+00, 1.6760e+00],
[ 3.6781e-01, -1.1951e+00, -2.5726e-01, ..., 6.7227e-01,
6.2984e-01, 1.0675e+00],
...,
[ 2.1205e-01, -3.0755e+00, -3.3317e+00, ..., 3.2862e+00,
5.3063e-01, 2.4053e-01],
[ 2.4589e+00, -3.7963e-01, -4.1663e+00, ..., 2.3645e+00,
7.9620e-01, 1.3484e+00],
[ 9.0006e-01, -3.4491e-02, -1.5198e+00, ..., 1.1725e+00,
6.0067e-01, 1.4474e+00]],
[[-5.3076e-01, 5.0944e-01, 2.4358e-01, ..., -3.8827e-01,
-5.6526e-01, 1.4680e-01],
[-1.3229e+00, -1.7207e+00, -1.4525e+00, ..., -1.0583e+00,
-9.0382e-01, -8.8052e-01],
[-2.3542e+00, 1.4726e-01, 1.0172e+00, ..., -7.4532e-01,
-9.1607e-01, -5.0686e-01],
...,
[-2.3380e-01, -1.2265e+00, -1.3066e+01, ..., -2.6486e+00,
-1.1868e+00, -1.4104e+00],
[-2.1590e+00, -1.9784e+00, -3.6489e+00, ..., -2.4066e+00,
-1.0955e+00, -7.8242e-01],
[-5.5249e-01, 9.0379e-02, -2.0557e+00, ..., -1.2527e+00,
-5.6470e-01, 2.9691e-01]]],
[[[ 8.1229e-01, 9.1387e-01, 4.8741e-01, ..., 5.4290e-01,
9.3497e-01, 4.1234e-01],
[-1.1213e+00, -9.7632e-01, 2.0516e-02, ..., 1.0318e+00,
-6.4500e-01, -5.7297e-01],
[-2.5269e-01, -8.0026e-01, -1.1957e-01, ..., 7.3155e-01,
3.8570e-01, -2.5222e-01],
...,
[ 5.0752e-01, -6.6892e-01, 2.3384e-01, ..., -1.2934e-01,
-9.3038e-02, 3.3041e-01],
[ 2.1798e-01, -1.2432e+00, 5.2543e-01, ..., -1.0968e-01,
1.9203e-02, -1.2432e-01],
[ 4.5959e-01, 2.4263e-01, -4.6412e-01, ..., -2.6014e-01,
-9.9591e-01, 1.6939e-01]],
[[-3.7695e+00, -1.9815e+00, -2.7900e+00, ..., -2.6239e+00,
-2.2754e+00, -1.7956e+00],
[-2.8278e+00, -1.8090e+00, -8.2066e-01, ..., -4.0137e+00,
-3.3736e+00, -2.1921e+00],
[-1.9231e+00, -1.5400e+00, -1.1806e+00, ..., -4.8754e+00,
-4.0588e+00, -2.0505e+00],
...,
[-1.8357e+00, -3.0989e+00, -3.6829e+00, ..., -5.5190e-01,
-9.9033e-01, -8.9281e-01],
[-3.6230e+00, -1.5991e+00, -1.0476e+00, ..., -7.8541e-01,
-1.1524e+00, -1.2865e+00],
[-6.7459e-01, -1.5752e-01, -3.4713e-01, ..., -6.4186e-01,
-7.3285e-01, -5.3524e-01]],
[[ 2.6226e+00, 3.2255e+00, 3.1073e+00, ..., 1.9307e+00,
2.3493e-01, 2.5708e+00],
[ 7.2471e-01, 1.2760e+00, 2.6252e+00, ..., 6.6668e-01,
8.7126e-01, 2.6884e+00],
[ 1.0129e+00, 7.5917e-01, 1.6399e+00, ..., -3.9562e+00,
3.2733e-01, 1.8535e+00],
...,
[-2.8503e+00, -1.6291e+00, 1.4956e+00, ..., -7.3212e-01,
6.4537e-01, 1.6264e+00],
[-6.0962e-01, 1.4285e+00, 2.4734e+00, ..., -5.8936e-01,
9.1260e-01, 1.3202e+00],
[ 1.8466e+00, 1.3610e+00, 1.7276e+00, ..., 3.7019e-01,
1.3688e+00, 1.4584e+00]],
...,
[[-2.5229e+00, -7.6310e+00, -4.9080e+00, ..., -3.0679e+00,
-3.3449e+00, -4.5906e+00],
[-3.0780e+00, -9.0870e+00, -6.3347e+00, ..., -4.8604e+00,
-4.4939e+00, -3.9416e+00],
[-2.1632e+00, -6.2472e+00, -3.6199e+00, ..., -6.1990e+00,
-2.3796e+00, -1.6655e+00],
...,
[-1.9239e+00, -4.7789e+00, -6.5088e+00, ..., -6.9309e+00,
-3.7720e+00, -2.8865e+00],
[-2.4202e+00, -6.9341e+00, -5.0368e+00, ..., -6.1838e+00,
-4.3705e+00, -4.3659e+00],
[-2.3306e+00, -7.2122e+00, -4.5996e+00, ..., -4.1112e+00,
-2.5991e+00, -2.4382e+00]],
[[-7.2411e-01, -1.2724e-01, 3.5170e-01, ..., 1.1333e+00,
1.7560e+00, 1.2225e+00],
[ 1.4059e+00, 8.3085e-01, 1.1513e+00, ..., -1.2148e+00,
1.5620e+00, 2.0358e+00],
[ 1.2101e+00, 4.5252e-01, 3.9446e-01, ..., -3.3863e+00,
-4.8895e-02, 1.3575e+00],
...,
[ 9.5402e-01, 3.1660e-02, -1.2957e+00, ..., 2.5004e+00,
7.3655e-01, 9.9493e-01],
[ 1.2849e+00, 7.1709e-01, 2.4046e+00, ..., 1.6761e+00,
7.2161e-01, 1.5357e+00],
[ 1.1649e+00, 1.1533e+00, 1.5144e+00, ..., 8.2940e-01,
7.1595e-01, 1.5015e+00]],
[[-7.8322e-01, -7.0829e-01, -5.2470e-01, ..., -1.5721e+00,
-7.9099e-01, -1.4361e-03],
[-1.3780e+00, -1.3926e+00, -6.0955e-01, ..., -8.8021e-01,
-2.3939e+00, -1.0592e+00],
[-2.0089e+00, -4.6526e-01, 1.7647e-01, ..., 4.1140e-01,
-2.6122e+00, -1.2374e+00],
...,
[-8.5658e-01, -2.7777e+00, -4.0850e+00, ..., -3.7533e-01,
-1.3977e+00, -8.4502e-01],
[-2.7403e+00, -3.2972e+00, -8.5087e-01, ..., -1.1150e+00,
-1.6122e+00, -6.9468e-01],
[-8.8811e-01, 4.7756e-01, 4.3455e-01, ..., -4.6797e-01,
-7.7894e-01, -1.1494e-03]]],
[[[ 5.7270e-01, 8.6174e-01, 2.1518e-01, ..., 5.3867e-01,
5.7753e-01, 1.2680e-02],
[-3.0133e-01, -4.7451e-01, -1.2350e-01, ..., 4.5609e-01,
7.9866e-02, -2.9376e-01],
[-1.3580e-01, 3.8295e-01, 6.3474e-01, ..., 5.3745e-01,
1.8804e-01, 7.9194e-01],
...,
[-2.3120e-01, -2.9096e-01, 3.1901e-01, ..., 1.1429e+00,
8.3721e-01, 7.8823e-01],
[-6.1975e-01, -1.5306e+00, 3.7754e-01, ..., -2.3871e+00,
6.2980e-01, 1.0063e-01],
[ 2.3668e-01, -1.3655e+00, -7.2754e-01, ..., -2.7833e-01,
9.3873e-01, 6.1527e-02]],
[[-2.4050e+00, 1.5605e-01, -6.3570e-01, ..., -3.9613e-01,
-2.6928e+00, -1.6709e+00],
[-2.4147e+00, 7.9059e-02, -8.8996e-01, ..., -1.8942e+00,
-3.2805e+00, -1.9347e+00],
[-2.4874e+00, -1.1918e+00, -1.4367e+00, ..., -3.1183e+00,
-1.7393e+00, -1.1283e+00],
...,
[-9.7605e-01, -1.7437e-02, 1.0099e+00, ..., -3.0075e+00,
-2.7367e+00, 6.5411e-02],
[-2.0399e+00, -1.0102e-01, -1.8166e+00, ..., -3.4121e+00,
-3.7582e+00, -6.4450e-02],
[-1.2126e+00, -1.4420e+00, -1.7632e+00, ..., -2.8849e+00,
-3.7717e+00, -7.3757e-01]],
[[ 1.8125e+00, 2.5626e+00, 7.7559e-01, ..., 1.4924e+00,
1.4842e-01, 2.1653e+00],
[-1.8313e-01, 1.5843e+00, 2.9929e-01, ..., 8.7569e-01,
-2.8270e-01, 2.4290e+00],
[ 2.3827e-01, 1.0292e+00, 3.7289e-01, ..., -2.4237e-01,
6.8750e-01, 2.0132e+00],
...,
[ 1.8626e+00, 1.4859e+00, -1.0685e-02, ..., -2.0128e+00,
3.8642e-01, 1.4991e+00],
[ 1.1928e+00, 2.7782e+00, 1.9091e+00, ..., -3.2715e+00,
-2.4895e-02, 8.5924e-01],
[ 1.7447e+00, 1.5233e+00, 2.3245e+00, ..., -1.5475e+00,
5.3271e-01, 4.0091e-01]],
...,
[[-1.7945e+00, -4.7539e+00, -3.4770e+00, ..., -2.2711e+00,
-1.4155e+00, -3.2795e+00],
[-2.2545e+00, -5.2590e+00, -3.8756e+00, ..., -6.1010e+00,
-2.1686e+00, -3.1635e+00],
[-1.3670e+00, -5.0849e+00, -4.0476e+00, ..., -4.8438e+00,
-3.4613e+00, -1.9900e+00],
...,
[-2.0619e+00, -5.6582e+00, -2.6821e+00, ..., -1.0207e+00,
-4.0581e+00, -4.6968e+00],
[-2.9324e+00, -7.2861e+00, -4.7696e+00, ..., 2.5603e-01,
-6.6716e+00, -7.4685e+00],
[-3.0673e+00, -6.9490e+00, -3.5725e+00, ..., -2.1065e+00,
-7.4058e+00, -4.9476e+00]],
[[ 8.1795e-01, 1.5820e+00, 9.8103e-01, ..., 3.5011e-01,
1.3854e+00, 1.4312e+00],
[ 1.1432e+00, 1.0123e+00, 8.5511e-01, ..., 2.2832e-01,
1.2013e+00, 2.0140e+00],
[ 7.2525e-01, 4.2320e-01, 8.5523e-01, ..., 1.0906e-01,
1.1741e+00, 1.3276e+00],
...,
[ 1.6094e+00, 5.2371e-01, 1.8381e+00, ..., -1.5982e+00,
1.2173e+00, 2.6830e+00],
[ 2.2442e+00, 8.7694e-01, 9.9055e-02, ..., -2.8330e+00,
9.8020e-01, 3.7041e+00],
[ 1.0348e+00, 6.1589e-01, 7.2412e-01, ..., -1.8423e+00,
-3.3769e-01, 1.9258e+00]],
[[-1.0312e-01, -6.7278e-01, -1.5327e+00, ..., -1.6189e+00,
-1.0879e+00, -7.3414e-02],
[-9.0075e-01, -2.1203e+00, -2.6257e+00, ..., -3.2462e+00,
-1.7788e+00, -6.8582e-01],
[-1.4401e+00, -1.3290e+00, -1.4583e+00, ..., -1.3451e+00,
-6.9786e-01, -1.3826e-01],
...,
[-2.2260e+00, -1.6893e+00, -5.5832e-01, ..., 1.2453e+00,
-1.3073e+00, -5.7317e-01],
[-2.1147e+00, -1.5708e+00, -1.3247e+00, ..., 5.7491e-01,
-1.0908e+00, -8.4569e-01],
[-2.2329e-01, -4.7747e-01, -1.0331e+00, ..., 7.4609e-01,
-2.1505e+00, -1.8809e+00]]]], device='cuda:0'),
tensor([[[[ 9.2487e-01, 1.5659e+00, 1.1607e+00, ..., 5.6992e-01,
3.2432e-01, -1.1899e-01],
[-7.9216e-01, -1.8315e+00, -1.0887e+00, ..., 3.3150e-01,
5.6958e-01, -7.4898e-01],
[-7.7221e-02, -6.9870e-01, 1.8569e-01, ..., -4.2657e-01,
1.7107e-01, 5.2833e-01],
...,
[ 4.3807e-01, -2.9621e-01, -3.5802e-01, ..., 1.2989e+00,
1.1631e+00, -9.5831e-01],
[-1.6038e-01, -1.0051e+00, -3.4941e-01, ..., -9.6063e-02,
1.0425e+00, 2.9483e-01],
[ 7.4530e-01, -2.4853e-01, -2.6476e-01, ..., -4.5801e-01,
-4.4499e-01, 2.7791e-01]],
[[-4.4095e+00, -1.2576e+00, -2.4400e+00, ..., -2.9348e+00,
-3.4989e+00, -8.3905e-01],
[-4.3626e+00, -3.1909e+00, -2.8187e+00, ..., -2.0680e+00,
-3.2991e+00, -6.4978e-01],
[-2.8986e+00, -2.3502e+00, -2.4909e+00, ..., -3.1493e+00,
-3.9983e+00, -9.5841e-01],
...,
[-7.4168e-01, -4.8549e-01, -1.6798e+00, ..., -4.0499e+00,
-3.6580e+00, -7.6551e-01],
[-2.7111e+00, -3.3189e-01, -1.0817e+00, ..., -1.8899e+00,
-2.0944e+00, -3.7163e-01],
[-1.9591e+00, -1.4922e+00, -1.8691e+00, ..., -9.9763e-01,
-1.0223e+00, 1.1498e-02]],
[[ 3.1534e+00, 4.1465e+00, 3.2571e+00, ..., 2.3454e+00,
7.2410e-01, 2.3784e+00],
[ 6.8483e-01, 1.0643e+00, 2.7254e+00, ..., 1.8961e+00,
1.4116e-01, 2.4142e+00],
[ 1.4010e+00, 1.7668e+00, 1.4918e+00, ..., 1.3557e+00,
9.8399e-01, 2.7058e+00],
...,
[ 2.2930e-01, 9.0240e-01, 1.4586e+00, ..., 1.4158e+00,
-4.3791e-01, 5.7778e-01],
[ 2.7234e-01, 1.9863e+00, 1.6183e+00, ..., 4.3245e-01,
5.5064e-01, 1.3278e+00],
[ 1.3626e+00, 1.3319e+00, 1.7779e+00, ..., 1.1038e+00,
8.0111e-01, 1.2778e+00]],
...,
[[-2.1632e+00, -7.4355e+00, -6.1143e+00, ..., -4.0499e+00,
-3.1582e+00, -3.9720e+00],
[-2.9074e+00, -6.8566e+00, -4.9604e+00, ..., -5.4533e+00,
-4.7869e+00, -4.6550e+00],
[-4.7791e-01, -3.2604e+00, -2.8197e+00, ..., -4.0438e+00,
-3.5438e+00, -2.2074e+00],
...,
[-9.3875e-01, -2.1623e+00, -1.5959e+00, ..., -5.9119e+00,
-4.5320e+00, -3.8227e+00],
[-1.2877e+00, -3.6311e+00, -3.3450e+00, ..., -6.6348e+00,
-5.0972e+00, -5.1621e+00],
[-2.1786e+00, -4.9656e+00, -3.0739e+00, ..., -4.4472e+00,
-2.8298e+00, -3.2408e+00]],
[[-7.5053e-01, -4.5538e-02, -7.6365e-02, ..., 9.3104e-01,
1.1163e+00, 1.7868e+00],
[ 1.6765e+00, 1.0530e+00, 9.3493e-01, ..., 1.3663e+00,
1.5617e+00, 2.9325e+00],
[ 1.0760e+00, -3.5754e-01, -5.0886e-02, ..., 1.0555e+00,
3.0458e-01, 2.1645e+00],
...,
[ 1.3150e+00, -8.4483e-02, 6.8533e-01, ..., 1.7145e+00,
1.7424e+00, 1.9237e+00],
[ 1.8636e+00, 8.9236e-01, 1.6318e+00, ..., 1.7896e+00,
1.3560e+00, 2.4863e+00],
[ 1.0296e+00, 2.9292e-01, 7.7885e-01, ..., 9.8887e-01,
1.1843e+00, 1.5743e+00]],
[[-5.5605e-01, 5.6656e-01, 1.6486e-02, ..., -2.3532e-01,
-5.5540e-01, 9.2452e-01],
[-1.2543e+00, -1.2021e+00, -1.2461e+00, ..., -2.5705e-01,
-1.4537e-01, 1.0250e+00],
[-2.1559e+00, -5.7407e-01, 5.2545e-01, ..., 2.0762e-01,
2.7108e-01, 3.8376e-01],
...,
[-1.6886e+00, -1.4754e+00, -1.2933e+00, ..., -1.0431e+00,
-1.3760e+00, -1.2550e+00],
[-1.6669e+00, -1.1082e+00, -1.0578e+00, ..., -1.0607e+00,
-1.4434e+00, 9.3085e-02],
[-3.0090e-01, 3.5720e-01, -4.2953e-01, ..., 8.1598e-02,
-2.3518e-01, 6.0920e-01]]],
[[[ 6.0050e-01, 8.7314e-01, 5.5463e-01, ..., -1.2742e-01,
2.1920e-01, 1.7228e-01],
[ 2.8700e-01, -1.2538e+00, -1.3079e+00, ..., 1.1719e+00,
7.7979e-01, -3.0236e-01],
[-8.1555e-01, -1.0093e+00, -1.4117e+00, ..., -4.8243e-01,
4.9903e-01, 6.3912e-01],
...,
[ 1.4451e+00, 5.2836e-01, -4.1527e-01, ..., 3.8903e-01,
8.5336e-01, 6.7342e-01],
[ 1.9827e-01, -3.2047e-01, 4.9008e-01, ..., 8.3121e-01,
1.5314e+00, 1.2576e-01],
[ 8.2397e-01, -2.3203e-01, -1.5617e-01, ..., 3.3250e-01,
2.7041e-01, 7.4589e-01]],
[[-2.3974e+00, -7.4538e-01, -1.8889e+00, ..., 3.6540e-01,
-1.3924e-01, 2.9614e-01],
[-4.9237e+00, -2.3110e+00, -1.4226e+00, ..., -3.1858e-02,
-2.3162e-01, -2.1415e-01],
[-8.0208e+00, -8.2465e+00, -4.5605e+00, ..., -1.9372e+00,
-4.9011e-01, 3.0289e-01],
...,
[-3.1595e-01, -5.7032e-01, -2.8961e+00, ..., 6.1380e-02,
-2.6575e-01, 6.6868e-01],
[-2.5566e+00, -2.7014e-01, -2.5015e-01, ..., 1.5545e-01,
-6.0942e-02, 6.3877e-01],
[-1.6363e+00, -1.3200e+00, -1.0193e+00, ..., -5.0425e-01,
-6.0525e-01, -2.1316e-01]],
[[ 1.8093e+00, 2.4289e+00, 1.6823e+00, ..., -4.2410e-01,
2.4207e-01, -7.1359e-02],
[ 3.0569e-01, 1.1303e+00, 1.4490e+00, ..., -5.5466e-01,
-1.2998e+00, -1.4623e+00],
[-1.4358e+00, -2.7897e+00, 1.0712e+00, ..., -2.9908e+00,
-2.6118e+00, -1.2930e+00],
...,
[ 7.9180e-01, -2.0110e-01, 1.6960e+00, ..., -1.0702e-01,
1.9890e-02, 1.5746e+00],
[ 1.0219e+00, 2.6239e+00, 2.3647e+00, ..., 9.4329e-01,
6.4411e-01, 1.6661e+00],
[ 2.4089e+00, 1.9038e+00, 2.4217e+00, ..., 1.9723e+00,
9.7732e-01, 1.6940e+00]],
...,
[[-1.8057e+00, -4.7375e+00, -3.1576e+00, ..., -4.2558e-01,
-3.6567e-01, 7.1915e-02],
[-3.5374e+00, -6.3482e+00, -2.9222e+00, ..., -3.7650e+00,
-3.7288e+00, -1.2159e+00],
[-1.8686e+00, -5.1759e+00, -2.1474e+00, ..., -4.5602e+00,
-5.1868e+00, -4.5074e+00],
...,
[-2.2193e+00, -5.3028e+00, -3.7411e+00, ..., -5.6227e+00,
-2.0923e+00, -2.7692e+00],
[-2.4955e+00, -5.4305e+00, -4.1329e+00, ..., -5.0112e+00,
-3.1228e+00, -4.9435e+00],
[-2.6426e+00, -5.9425e+00, -2.8911e+00, ..., -2.9954e+00,
-1.6216e+00, -3.3594e+00]],
[[ 8.7169e-01, 1.4638e+00, 8.1942e-01, ..., -2.0901e-01,
2.3124e-01, 7.0808e-01],
[ 6.0976e-01, 1.0901e+00, 8.4617e-01, ..., -1.9710e+00,
-2.0014e+00, -4.0729e-01],
[-4.9127e+00, -6.7136e+00, -4.9755e+00, ..., 9.4143e-01,
4.1687e-01, 3.8282e-01],
...,
[ 1.0835e+00, -2.4319e-01, -6.6950e-01, ..., 2.3397e+00,
1.7632e+00, 1.4484e+00],
[ 1.9654e+00, 9.5301e-01, 1.8337e+00, ..., 1.9603e+00,
1.6658e+00, 2.5476e+00],
[ 1.1630e+00, 8.2196e-01, 1.3103e+00, ..., 1.3447e+00,
1.4838e+00, 1.5123e+00]],
[[-1.0767e-01, -5.8438e-01, -1.2183e+00, ..., -6.3640e-01,
-6.6639e-01, 1.2377e-01],
[-8.6846e-01, -1.9112e+00, -2.0380e+00, ..., -9.4702e-01,
-1.3639e+00, -7.0660e-01],
[-2.4653e+00, -5.3411e-01, -2.0397e+00, ..., -1.7468e+00,
-2.3588e+00, -1.3590e+00],
...,
[-1.3593e+00, -7.9133e-01, -1.0968e+00, ..., -4.7501e-01,
-3.6953e-01, -4.5230e-01],
[-1.6688e+00, -1.5498e+00, -1.0530e+00, ..., -1.2194e-01,
-3.2868e-01, 4.4021e-01],
[-2.6269e-01, 3.9003e-01, -2.3181e-01, ..., 1.6556e-01,
4.6875e-02, 6.4836e-01]]],
[[[ 6.1694e-01, 8.9746e-01, 5.7294e-01, ..., 4.1872e-01,
4.8056e-01, 2.3849e-02],
[-2.8025e-01, -8.5513e-01, -4.9347e-01, ..., 2.6263e-01,
2.5278e-01, -2.8571e-01],
[-6.4846e-02, 1.7385e-01, 4.9257e-01, ..., -1.9230e-01,
-9.8258e-02, 6.5410e-01],
...,
[-1.2627e-01, -1.0233e+00, -6.9899e-01, ..., 3.1444e-01,
9.0020e-01, 9.5667e-01],
[-3.9217e-01, -1.9312e+00, -8.7471e-01, ..., -4.1780e-01,
8.5236e-01, 6.9736e-01],
[ 4.8861e-01, -7.5593e-01, -1.0947e+00, ..., -1.6749e+00,
-1.6027e+00, -5.7614e-01]],
[[-2.2603e+00, -6.0986e-01, -1.8729e+00, ..., -1.7498e+00,
-2.4047e+00, -1.6047e+00],
[-2.4538e+00, -6.4669e-01, -1.4291e+00, ..., -1.0189e+00,
-2.3645e+00, -1.6992e+00],
[-2.3180e+00, -1.6703e+00, -2.1756e+00, ..., -2.3735e+00,
-2.5707e+00, -1.1787e+00],
...,
[-1.0109e+00, -5.1971e-01, -1.7237e+00, ..., -2.0645e+00,
-1.7508e+00, -7.7032e-02],
[-2.2920e+00, 1.3544e-02, -9.8578e-01, ..., -8.0013e-01,
-1.2030e+00, -4.2340e-01],
[-1.0211e+00, -1.2047e+00, -1.7759e+00, ..., -2.8802e+00,
-3.5106e+00, -1.8087e+00]],
[[ 1.8279e+00, 2.4275e+00, 1.5482e+00, ..., 1.5650e+00,
7.7768e-01, 2.0940e+00],
[-2.9342e-01, 3.5667e-01, 6.2353e-01, ..., 8.5709e-01,
4.2811e-01, 2.2399e+00],
[ 3.6060e-03, 3.7645e-01, 3.7879e-01, ..., 1.4126e-01,
1.3806e+00, 2.3059e+00],
...,
[ 1.1083e+00, 1.1179e+00, 1.6046e+00, ..., 3.7199e-01,
-5.0653e-01, 1.0294e+00],
[ 6.9348e-01, 1.9678e+00, 1.4391e+00, ..., -2.7525e-01,
-6.1891e-01, 9.7499e-01],
[ 1.6906e+00, 9.5223e-01, 1.5466e+00, ..., -3.0690e-01,
1.9488e-01, 1.3211e+00]],
...,
[[-1.9140e+00, -4.7814e+00, -3.1232e+00, ..., -1.9774e+00,
-1.6382e+00, -3.0099e+00],
[-2.2821e+00, -4.9257e+00, -2.8172e+00, ..., -2.3969e+00,
-2.2485e+00, -3.1152e+00],
[-1.4851e+00, -5.2296e+00, -3.2506e+00, ..., -1.2593e+00,
-1.6950e+00, -1.7431e+00],
...,
[-2.1988e+00, -5.4381e+00, -4.2005e+00, ..., -2.8427e+00,
-2.8233e+00, -2.7684e+00],
[-2.8098e+00, -7.1671e+00, -4.3282e+00, ..., -3.0920e+00,
-3.1747e+00, -4.2474e+00],
[-2.9634e+00, -7.2894e+00, -3.6050e+00, ..., -2.0036e+00,
-6.5870e-01, -1.0304e+00]],
[[ 9.3467e-01, 1.5804e+00, 7.8324e-01, ..., 9.2414e-01,
1.4329e+00, 1.3992e+00],
[ 1.1261e+00, 1.0106e+00, 9.0938e-01, ..., 9.0746e-01,
8.1129e-01, 1.9878e+00],
[ 6.2828e-01, 9.6691e-02, 5.9364e-01, ..., 5.1475e-01,
3.5315e-01, 1.5607e+00],
...,
[ 1.8409e+00, 6.6580e-01, 1.2884e+00, ..., 4.1665e-01,
1.2435e+00, 1.9086e+00],
[ 2.2667e+00, 1.6432e+00, 1.3412e+00, ..., 1.1799e+00,
1.1122e+00, 1.7705e+00],
[ 9.8289e-01, 4.8522e-01, 5.6114e-01, ..., -6.1100e-01,
1.2625e-02, 1.0874e+00]],
[[-7.7536e-02, -6.1543e-01, -1.1553e+00, ..., -4.8272e-01,
-9.0615e-01, -1.1349e-01],
[-9.1286e-01, -2.5734e+00, -2.5222e+00, ..., -9.9536e-01,
-9.7727e-01, -6.1758e-01],
[-1.2811e+00, -1.5873e+00, -4.5957e-01, ..., 2.4940e-01,
-5.1303e-01, -1.4574e-01],
...,
[-2.7811e+00, -2.1465e+00, -8.8594e-01, ..., -1.8077e-01,
-1.9385e-01, 4.5495e-01],
[-2.5240e+00, -2.1782e+00, -9.4623e-01, ..., 4.6034e-01,
3.2096e-01, 6.0541e-01],
[-4.4989e-01, -2.7851e-01, -4.7741e-01, ..., -8.9936e-01,
-1.6717e+00, -1.2199e+00]]],
[[[ 1.0998e+00, 1.3327e+00, 1.1947e+00, ..., 2.1141e-01,
1.4085e-02, -5.4322e-01],
[-9.3389e-01, -9.9953e-01, -1.1603e+00, ..., 6.0947e-01,
1.2629e+00, -1.3701e-01],
[-4.3839e-01, -6.1912e-01, 1.0587e+00, ..., -1.5568e-01,
7.3121e-01, 8.6203e-01],
...,
[ 3.3896e-01, -6.5348e-01, -8.8673e-01, ..., -1.7262e+00,
-2.6896e-01, 3.2435e-01],
[-6.1360e-01, -1.7399e+00, -1.3613e+00, ..., 1.6964e+00,
1.1675e+00, 1.7452e-01],
[-3.4566e-01, -1.6921e+00, -1.6352e+00, ..., -8.5236e-01,
-1.1536e+00, -2.1860e-01]],
[[-2.9279e+00, 1.5275e+00, 9.0133e-01, ..., -2.6235e+00,
-3.4349e+00, -9.1905e-01],
[-3.2066e+00, -1.8499e+00, -4.1430e-01, ..., -1.3487e+00,
-1.8743e+00, -7.9654e-02],
[-2.5496e+00, -2.9742e+00, -4.5745e+00, ..., -1.9615e+00,
-1.7495e+00, -5.5355e-01],
...,
[-1.3849e+00, -1.2830e+00, -2.4502e+00, ..., -1.2442e+01,
-8.7155e+00, -3.3316e+00],
[-2.8705e+00, -1.6461e+00, -2.8523e+00, ..., -4.1261e+00,
-3.6864e+00, -1.4066e+00],
[-2.2429e+00, -1.9614e+00, -2.6590e+00, ..., -3.0504e+00,
-2.4203e+00, -1.0488e+00]],
[[ 3.1363e+00, 1.3534e+00, -1.2828e+00, ..., 2.3103e+00,
6.8905e-01, 2.2121e+00],
[ 1.5867e+00, -6.0121e-01, -2.1100e+00, ..., 1.9543e+00,
6.7242e-02, 2.1150e+00],
[ 1.3094e+00, 1.2036e+00, 9.0888e-01, ..., 1.0328e+00,
8.1800e-01, 2.1155e+00],
...,
[ 1.1888e+00, 1.4569e+00, 1.8914e+00, ..., -7.7204e+00,
-5.0409e+00, -1.3873e+00],
[ 7.0796e-01, 1.8394e+00, 1.1867e+00, ..., -9.1840e-01,
1.4789e+00, 1.2066e+00],
[ 1.6600e+00, 1.6314e+00, 1.9084e+00, ..., -2.6345e-01,
1.7309e+00, 1.7676e+00]],
...,
[[-2.6787e+00, -6.3274e+00, -3.0633e+00, ..., -3.0504e+00,
-2.7976e+00, -3.6665e+00],
[-3.1653e+00, -7.7464e+00, -3.0762e+00, ..., -4.7051e+00,
-4.5265e+00, -5.0868e+00],
[-1.5746e+00, -4.4070e+00, -2.0853e+00, ..., -2.6179e+00,
-2.5325e+00, -2.2767e+00],
...,
[-2.2914e+00, -5.1549e+00, -4.3165e+00, ..., -5.6802e+00,
-7.9468e+00, -7.6759e+00],
[-2.7388e+00, -5.8953e+00, -5.5060e+00, ..., -6.0285e+00,
-6.0788e+00, -8.7723e+00],
[-1.4943e+00, -4.1393e+00, -2.7451e+00, ..., -2.9803e+00,
-2.6182e+00, -3.3231e+00]],
[[-1.0612e-01, 2.7766e+00, 2.0807e+00, ..., 4.4860e-01,
8.5054e-01, 1.6061e+00],
[ 7.3543e-01, 1.3662e-01, 2.1206e+00, ..., 1.2784e+00,
1.7945e+00, 2.5029e+00],
[ 9.5727e-01, -1.2288e+00, -1.1687e+00, ..., 1.0579e+00,
1.2391e+00, 1.7910e+00],
...,
[ 1.2137e+00, 3.1091e-01, -4.9018e-01, ..., -2.9825e+00,
-2.3918e+00, 5.3498e-01],
[ 1.8911e+00, 1.6437e+00, 9.5698e-01, ..., 6.8612e-01,
1.5092e+00, 2.7380e+00],
[ 4.7544e-01, 7.7752e-01, 6.9599e-01, ..., 6.3610e-02,
6.6075e-01, 1.7423e+00]],
[[-6.4041e-01, -5.9228e-01, -4.3805e-01, ..., -3.3053e-01,
-4.6437e-01, 5.1493e-01],
[-1.3331e+00, -1.5255e+00, -2.4152e+00, ..., -1.4796e-01,
4.8200e-01, 8.5518e-01],
[-2.0735e+00, -3.8478e-01, 1.7655e-01, ..., 3.1767e-01,
3.0548e-01, -1.2292e-02],
...,
[-1.8074e+00, -1.8547e+00, -1.3443e+00, ..., -4.6268e+00,
-6.7290e+00, -2.9970e+00],
[-1.5454e+00, -2.0626e+00, -1.7776e+00, ..., -6.9753e-01,
-2.3894e+00, 1.2334e-01],
[-8.2401e-01, -8.8459e-01, -1.1672e+00, ..., -6.0241e-01,
-8.8127e-01, -2.1013e-02]]],
[[[ 5.8269e-01, 8.3650e-01, 5.5407e-01, ..., 3.1725e-01,
4.2433e-01, 2.9220e-02],
[-2.0537e-01, -5.6763e-01, -2.9928e-01, ..., 1.6860e-01,
5.2492e-02, -3.6288e-01],
[-3.2907e-01, -2.0758e-01, -1.2777e-01, ..., -7.3552e-02,
9.6813e-02, 6.9780e-01],
...,
[ 4.3866e-01, -3.8524e-01, -8.0217e-01, ..., 1.4793e+00,
8.5109e-01, 6.6533e-01],
[ 1.4145e-02, -6.6217e-01, -1.7901e-01, ..., 1.2052e+00,
1.5958e+00, 1.1276e-01],
[ 8.3761e-01, -2.0178e-01, -2.0346e-01, ..., 4.6409e-01,
2.8776e-01, 7.0933e-01]],
[[-2.4102e+00, -6.5065e-01, -1.9180e+00, ..., -1.6198e+00,
-2.2205e+00, -1.5159e+00],
[-2.5775e+00, -1.2328e+00, -1.6408e+00, ..., -1.3335e+00,
-2.4068e+00, -1.6831e+00],
[-2.6913e+00, -3.3847e+00, -3.2173e+00, ..., -2.0765e+00,
-2.3551e+00, -1.0895e+00],
...,
[-6.6623e-01, -1.0054e+00, -2.2258e+00, ..., -1.7708e+00,
-2.9778e-02, 7.2127e-01],
[-2.5660e+00, -2.5269e-01, -9.5715e-01, ..., -9.2865e-01,
2.4214e-01, 6.6112e-01],
[-1.6036e+00, -1.3733e+00, -1.0846e+00, ..., -6.3645e-01,
-6.7062e-01, -2.2127e-01]],
[[ 1.9449e+00, 2.5279e+00, 1.6308e+00, ..., 1.6390e+00,
7.5505e-01, 2.0728e+00],
[-1.1361e-01, 3.0116e-01, 7.6330e-01, ..., 8.1343e-01,
2.8877e-01, 2.2573e+00],
[ 9.1061e-01, 1.0211e+00, 9.1123e-01, ..., -6.9868e-02,
7.5869e-01, 2.0201e+00],
...,
[ 7.5446e-01, 2.8585e-01, 2.8492e+00, ..., -4.9738e-01,
-4.1665e-01, 1.5757e+00],
[ 1.1158e+00, 2.2540e+00, 1.8010e+00, ..., -1.3573e-01,
6.4464e-01, 1.6721e+00],
[ 2.4367e+00, 1.8783e+00, 2.3963e+00, ..., 1.6510e+00,
9.1977e-01, 1.7123e+00]],
...,
[[-2.0428e+00, -4.9152e+00, -3.1115e+00, ..., -2.0384e+00,
-1.6533e+00, -3.0996e+00],
[-2.5356e+00, -5.5902e+00, -2.9893e+00, ..., -2.5364e+00,
-2.2728e+00, -3.0217e+00],
[-1.2994e+00, -4.0545e+00, -1.9425e+00, ..., -2.4153e+00,
-2.5733e+00, -2.4003e+00],
...,
[-2.5870e+00, -4.4885e+00, -2.7185e+00, ..., -4.8734e+00,
-2.6277e+00, -2.6930e+00],
[-2.5551e+00, -5.4950e+00, -4.3306e+00, ..., -4.7241e+00,
-3.7014e+00, -4.9226e+00],
[-2.6541e+00, -5.9503e+00, -2.8761e+00, ..., -3.8867e+00,
-2.5347e+00, -3.2960e+00]],
[[ 9.1493e-01, 1.6126e+00, 7.8401e-01, ..., 1.1180e+00,
1.5399e+00, 1.3943e+00],
[ 9.7479e-01, 6.5048e-01, 9.1999e-01, ..., 8.8806e-01,
7.9551e-01, 1.9309e+00],
[-2.9288e-02, -2.1446e+00, -1.4624e+00, ..., 4.2958e-01,
1.3080e-01, 1.4074e+00],
...,
[ 1.5164e+00, 5.7786e-02, -8.1435e-02, ..., 8.9827e-01,
1.8572e+00, 1.5139e+00],
[ 2.0438e+00, 1.1921e+00, 1.7363e+00, ..., 1.3469e+00,
1.7773e+00, 2.5688e+00],
[ 1.1504e+00, 8.3587e-01, 1.3289e+00, ..., 1.4031e+00,
1.3635e+00, 1.5316e+00]],
[[-1.1750e-01, -6.6071e-01, -1.1617e+00, ..., -5.5661e-01,
-8.6572e-01, -1.0150e-01],
[-9.4742e-01, -2.5301e+00, -2.0378e+00, ..., -1.1705e+00,
-1.0772e+00, -6.8184e-01],
[-1.9275e+00, -1.6579e+00, -4.4882e-01, ..., -2.8600e-01,
-3.8413e-01, -2.2369e-01],
...,
[-1.7191e+00, -1.7022e+00, -5.6608e-01, ..., -1.4065e-01,
-4.9449e-01, -4.3203e-01],
[-1.7237e+00, -1.2591e+00, -8.2071e-01, ..., -1.0920e+00,
-2.5652e-01, 4.5251e-01],
[-2.7031e-01, 3.7895e-01, -2.1485e-01, ..., 7.4330e-01,
1.2981e-01, 6.2051e-01]]]], device='cuda:0'),
tensor([[[[ 8.5573e-01, 1.5123e+00, 1.1972e+00, ..., 2.5842e-01,
1.5788e+00, 1.0910e+00],
[-9.7565e-01, -1.9867e+00, -1.0422e+00, ..., -3.4674e+00,
4.2442e-01, 2.2988e-02],
[-1.0070e-01, -7.2561e-01, 8.6873e-02, ..., -1.9010e+00,
-2.7605e-01, 2.3998e+00],
...,
[ 4.7100e-01, 4.6531e-01, 1.3046e+00, ..., 9.8475e-01,
9.6753e-01, 3.9209e-01],
[ 3.4751e-01, -3.6168e-01, 1.3992e+00, ..., -1.4582e+00,
9.9374e-01, -1.2204e-01],
[ 5.7271e-01, 2.7244e-01, -1.8943e-01, ..., 1.5307e+00,
7.3180e-01, 7.5889e-01]],
[[-3.7888e+00, -6.9624e-01, -1.4862e+00, ..., -1.5702e+00,
-4.1421e+00, -2.3041e+00],
[-4.0125e+00, -2.8010e+00, -1.7325e+00, ..., -3.1532e+00,
-5.8897e+00, -3.3642e+00],
[-2.4906e+00, -1.8003e+00, -1.6733e+00, ..., -4.3705e+00,
-5.4614e+00, -1.5331e+00],
...,
[-8.7616e-01, -4.4774e-01, -3.1507e-01, ..., -4.2425e+00,
-1.8219e+00, -2.7044e-01],
[-1.3454e+00, -4.6273e-01, -2.0830e+00, ..., -3.5040e+00,
-2.0699e+00, -2.6515e-01],
[ 4.4040e-01, -4.8867e-01, -1.1297e+00, ..., -1.8972e+00,
-1.0099e+00, 3.1276e-01]],
[[ 2.9328e+00, 4.3027e+00, 3.3499e+00, ..., -1.8167e+00,
-1.5319e+00, 2.4854e+00],
[ 2.4912e-01, 1.1061e+00, 2.2364e+00, ..., -4.0594e+00,
-4.9161e+00, 1.3447e+00],
[ 1.0546e+00, 1.6332e+00, 1.7261e+00, ..., -3.3287e+00,
-2.3848e+00, 1.4502e+00],
...,
[ 8.2008e-01, 1.0066e+00, -7.2209e-01, ..., -2.8778e+00,
-1.0909e+00, 1.0417e+00],
[ 3.7435e-01, 1.9419e+00, 1.2297e+00, ..., -2.4798e+00,
4.1839e-01, 9.5029e-01],
[ 1.4803e+00, 1.1078e+00, 2.4865e+00, ..., -2.3775e+00,
-2.4660e-01, 9.2444e-01]],
...,
[[-1.8950e+00, -7.3075e+00, -5.3653e+00, ..., -7.0935e-01,
-9.7396e-01, -4.6793e+00],
[-2.4382e+00, -7.0206e+00, -4.2250e+00, ..., 1.3125e+00,
-7.9675e-01, -6.2741e+00],
[-1.0595e+00, -4.7105e+00, -3.7062e+00, ..., -2.3227e-01,
-1.5685e+00, -2.2181e+00],
...,
[-2.2422e+00, -5.0464e+00, -2.7987e+00, ..., -4.2542e+00,
-4.9150e+00, -4.9013e+00],
[-2.4404e+00, -6.7100e+00, -4.6706e+00, ..., -8.2621e-01,
-3.9643e+00, -5.8442e+00],
[-2.3248e+00, -6.6231e+00, -3.7352e+00, ..., -7.6304e+00,
-5.0527e+00, -4.7178e+00]],
[[-1.9817e-01, 2.8704e-01, 5.6188e-01, ..., -1.6740e+00,
-3.1248e+00, 6.8356e-01],
[ 1.6371e+00, 1.0998e+00, 1.8687e+00, ..., -6.2436e-01,
-3.3825e+00, 8.4082e-01],
[ 1.1479e+00, 2.0286e-01, 7.7906e-01, ..., -3.1780e-01,
-2.3496e+00, 4.8917e-01],
...,
[ 1.0631e+00, 2.8105e-01, 1.4370e+00, ..., -2.6038e+00,
5.6258e-01, 1.6306e+00],
[ 1.5969e+00, 3.9920e-01, 9.8917e-02, ..., -2.1934e+00,
3.0121e-01, 2.3961e+00],
[ 1.1899e+00, 7.1060e-01, 8.6235e-01, ..., -9.5544e-01,
8.3948e-01, 1.5609e+00]],
[[-5.6028e-01, 2.9557e-01, 1.4099e-01, ..., 3.1404e-01,
9.3886e-01, 1.3262e+00],
[-1.3783e+00, -1.8191e+00, -1.5878e+00, ..., -1.3212e+00,
-5.4336e-02, 8.5428e-01],
[-2.1791e+00, -7.8961e-01, 2.2985e-01, ..., -1.3744e+00,
-6.3334e-01, 9.3361e-01],
...,
[-1.5491e+00, -1.4125e+00, -1.2378e+00, ..., -1.1363e+00,
-2.3555e+00, -1.9941e+00],
[-1.2720e+00, -1.1837e+00, -4.6869e-01, ..., -2.7814e-01,
-1.2770e+00, -1.1540e+00],
[ 1.5455e-01, 7.0157e-02, -1.1163e-01, ..., 1.0116e+00,
-7.5190e-01, 4.1107e-01]]],
[[[ 5.3879e-01, 7.3661e-01, -1.0394e+00, ..., -1.0028e+00,
-1.5844e+00, -1.2112e+00],
[-1.5333e+00, -2.8147e+00, -1.1291e+00, ..., -2.6214e+00,
-1.7709e+00, -5.4710e-01],
[-1.0575e+00, -3.7873e+00, -3.2931e+00, ..., 7.2431e-01,
-2.9899e+00, -1.7544e+00],
...,
[ 5.1590e-01, -7.7323e-01, -1.5389e+00, ..., 2.0438e-01,
5.2906e-01, 5.4252e-01],
[-9.0026e-02, -1.0462e+00, -7.3028e-01, ..., -3.0765e-01,
-1.5731e+00, -2.3727e-01],
[ 7.6232e-01, -4.2108e-01, -4.2094e-01, ..., -3.5053e-01,
-3.1667e-01, -3.3028e+00]],
[[-6.8999e+00, -3.3758e+00, -7.8203e+00, ..., -5.4773e+00,
-3.9365e+00, -2.0589e+00],
[-6.8580e+00, -2.4756e+00, -4.1535e+00, ..., -1.9009e+00,
9.4300e-01, -1.4596e-01],
[-5.4274e+00, -1.2536e+00, -3.8288e+00, ..., -4.8038e-01,
-3.9942e+00, -5.0557e+00],
...,
[-1.1360e+00, -1.9944e+00, -3.6839e+00, ..., -1.1331e+00,
9.9959e-02, -9.4256e-01],
[-3.1931e+00, -8.4482e-01, -1.6571e+00, ..., -2.8038e+00,
-1.9762e+00, -1.2016e+00],
[-2.5170e+00, -1.8026e+00, -1.6583e+00, ..., -5.8497e+00,
-4.6966e+00, -1.2597e+00]],
[[ 3.2318e+00, 4.7248e+00, 3.7600e+00, ..., 3.1737e+00,
1.3173e+00, 9.7549e-01],
[ 2.6331e-01, 5.8744e-01, -6.4070e-01, ..., 4.2042e-01,
-4.0258e+00, -2.9290e+00],
[-8.5265e-01, 1.1229e+00, 6.5833e-02, ..., -7.9127e+00,
-1.1441e+01, -5.7915e+00],
...,
[-1.9227e-01, 5.3877e-01, 3.9344e+00, ..., -3.4040e+00,
-3.2547e+00, -1.7763e-01],
[ 9.8024e-02, 2.2146e+00, 2.0439e+00, ..., -4.0056e+00,
-6.1876e+00, -9.7514e-01],
[ 1.3655e+00, 1.4157e+00, 1.7801e+00, ..., 1.8722e+00,
-2.9865e+00, -2.0990e+00]],
...,
[[-3.1019e+00, -8.0546e+00, -4.9544e+00, ..., -3.7926e+00,
-4.7489e+00, -2.8925e+00],
[-4.0956e+00, -9.6601e+00, -8.5955e+00, ..., -5.4252e+00,
-7.1443e+00, -8.2224e+00],
[-2.6265e+00, -5.1698e+00, -7.2794e+00, ..., -9.1273e+00,
-1.2353e+01, -1.1229e+01],
...,
[-8.0721e-01, -2.9434e+00, -2.7000e+00, ..., -3.1739e+00,
-8.2111e-01, -4.6128e+00],
[-1.1938e+00, -3.7595e+00, -3.6204e+00, ..., -7.5717e+00,
-5.5395e+00, -7.8427e+00],
[-2.1486e+00, -5.1874e+00, -3.3399e+00, ..., -5.0173e+00,
-3.5314e+00, -5.1212e+00]],
[[-1.2833e+00, -2.3491e+00, -4.9956e+00, ..., -3.6741e+00,
-3.3171e+00, -8.1208e-01],
[ 9.6310e-01, 2.2658e-01, -1.1243e+00, ..., -2.9583e-01,
-5.7257e-01, -3.6068e-01],
[ 6.6355e-01, 2.4701e+00, 1.9984e+00, ..., -7.0821e-01,
-9.5094e-01, -2.4040e+00],
...,
[ 1.1983e+00, -5.6247e-01, 2.5349e-01, ..., 1.8010e-01,
1.3932e-01, -1.1038e+00],
[ 1.7947e+00, 6.6376e-01, 1.6682e+00, ..., 2.2350e+00,
3.2766e+00, 7.4124e-01],
[ 9.6001e-01, 1.6053e-01, 8.8063e-01, ..., -3.0418e+00,
3.1986e-01, 1.0106e+00]],
[[-1.3779e+00, 3.3027e-01, -1.7176e+00, ..., -1.7619e-01,
-9.7486e-01, -8.3792e-02],
[-2.0702e+00, 9.2712e-02, -1.8977e+00, ..., -1.0313e+00,
3.8869e-01, 1.3570e+00],
[-2.2074e+00, 1.4409e+00, -1.0237e+00, ..., 2.5385e+00,
3.4921e-01, -4.0833e+00],
...,
[-1.5254e+00, -1.8753e+00, -1.1697e+00, ..., -1.8459e+00,
-1.2971e-01, 1.7053e-01],
[-1.5598e+00, -9.3591e-01, -1.2662e+00, ..., -1.1443e+00,
-5.5074e-01, -1.2513e+00],
[-2.9882e-01, 3.2628e-01, -4.1304e-01, ..., 1.1935e-01,
-2.1349e+00, -2.7034e+00]]],
[[[ 9.3505e-01, 1.5937e+00, 1.1347e+00, ..., 8.4833e-01,
1.0159e+00, 4.8894e-01],
[-8.5116e-01, -1.7367e+00, -7.7064e-01, ..., -2.7527e-01,
-2.1387e-01, -3.2972e-01],
[ 5.5188e-02, -3.5823e-01, 7.4347e-02, ..., -1.0511e+00,
-2.8796e-01, 6.3811e-01],
...,
[ 6.0823e-01, -7.4614e-02, -9.1277e-02, ..., -1.3743e-02,
2.8215e-03, 4.0790e-01],
[-1.0599e-01, -1.1312e+00, -1.5459e-01, ..., 1.8941e-02,
1.7240e-01, -7.5412e-02],
[ 8.2754e-01, -3.3526e-01, -2.2143e-01, ..., -3.0472e-01,
-9.1102e-01, 1.6762e-01]],
[[-3.7031e+00, -1.3386e+00, -1.9925e+00, ..., -1.4338e+00,
-2.0718e+00, -1.8040e+00],
[-3.8127e+00, -3.0682e+00, -2.1504e+00, ..., -1.0869e+00,
-2.2527e+00, -2.1137e+00],
[-2.3989e+00, -1.9260e+00, -2.9744e+00, ..., -1.3472e+00,
-1.7899e+00, -1.3992e+00],
...,
[-7.8416e-01, -4.5219e-01, -1.2117e+00, ..., -7.6604e-01,
-7.8245e-01, -8.2936e-01],
[-3.1110e+00, -1.8630e-01, -6.0270e-01, ..., -6.6887e-01,
-9.9547e-01, -1.1668e+00],
[-2.2619e+00, -1.5899e+00, -1.3804e+00, ..., -6.6032e-01,
-6.4530e-01, -4.6732e-01]],
[[ 2.5625e+00, 3.7236e+00, 2.9419e+00, ..., 1.6698e+00,
8.3153e-01, 2.5243e+00],
[-1.3198e-01, 7.8219e-01, 2.0793e+00, ..., 1.1380e+00,
5.1858e-01, 2.5754e+00],
[ 5.2515e-01, 1.4461e+00, 1.6404e+00, ..., 6.1582e-01,
8.3748e-01, 1.7905e+00],
...,
[ 4.5984e-01, 9.1446e-01, 1.2476e+00, ..., -2.4853e-01,
4.1439e-01, 1.4261e+00],
[ 6.2939e-01, 2.2798e+00, 1.6630e+00, ..., -1.2288e-01,
7.1923e-01, 1.1020e+00],
[ 1.7255e+00, 1.3358e+00, 1.7849e+00, ..., 5.9819e-01,
1.2720e+00, 1.3317e+00]],
...,
[[-1.4781e+00, -6.7118e+00, -5.5338e+00, ..., -4.9042e+00,
-3.6160e+00, -4.2896e+00],
[-2.0921e+00, -5.9665e+00, -4.2700e+00, ..., -4.5912e+00,
-3.5625e+00, -3.1416e+00],
[-2.6609e-01, -2.5082e+00, -3.2507e+00, ..., -2.3699e+00,
-2.1655e+00, -1.3381e+00],
...,
[-1.0447e+00, -2.5373e+00, -1.7870e+00, ..., -3.9720e+00,
-2.8597e+00, -2.6087e+00],
[-1.5675e+00, -3.8857e+00, -3.3176e+00, ..., -5.0918e+00,
-4.0659e+00, -4.1833e+00],
[-2.5611e+00, -5.3473e+00, -2.8615e+00, ..., -3.4058e+00,
-2.3599e+00, -2.3173e+00]],
[[-6.1506e-01, -1.9137e-01, 2.7793e-01, ..., 1.4717e+00,
1.6468e+00, 1.1584e+00],
[ 1.6407e+00, 8.1355e-01, 1.5184e+00, ..., 1.4964e+00,
1.0867e+00, 1.8168e+00],
[ 1.0963e+00, 2.2931e-02, 5.7730e-01, ..., 1.8114e+00,
6.0722e-01, 1.2777e+00],
...,
[ 1.2823e+00, -1.2512e-01, 1.0377e+00, ..., 1.3986e+00,
7.1041e-01, 8.3175e-01],
[ 1.9481e+00, 1.2597e+00, 1.8098e+00, ..., 1.3309e+00,
7.4587e-01, 1.4176e+00],
[ 1.0045e+00, 5.3365e-01, 1.1040e+00, ..., 7.5813e-01,
7.1740e-01, 1.4218e+00]],
[[-4.4992e-01, 5.3562e-01, 1.8755e-01, ..., -2.8967e-01,
-5.9542e-01, -6.6708e-02],
[-1.0558e+00, -1.2108e+00, -1.2330e+00, ..., -1.2530e+00,
-1.4332e+00, -1.1138e+00],
[-1.8278e+00, -6.2600e-01, -1.9360e-01, ..., -1.1241e+00,
-1.0470e+00, -2.3496e-01],
...,
[-1.7658e+00, -1.5216e+00, -8.7674e-01, ..., -1.0488e+00,
-1.4384e+00, -8.8132e-01],
[-1.8521e+00, -1.1273e+00, -8.3465e-01, ..., -8.3158e-01,
-1.6283e+00, -7.0823e-01],
[-4.6355e-01, 2.4062e-01, -2.9787e-01, ..., -2.9667e-01,
-8.3666e-01, -1.2005e-02]]],
[[[ 1.0456e+00, 1.2908e+00, 1.2913e+00, ..., 1.1825e+00,
1.0384e+00, 4.9215e-01],
[-5.2775e-01, -2.3960e+00, -1.2807e+00, ..., -2.1741e-01,
-1.7257e-01, -1.6247e+00],
[-2.8800e-01, -1.1060e+00, -7.1158e-01, ..., -4.9118e-01,
2.1676e-02, 6.6773e-01],
...,
[ 1.6764e-01, -3.1104e-01, 7.8777e-01, ..., 3.2198e-01,
1.0119e+00, 7.3966e-01],
[-9.8599e-02, -1.0150e+00, -2.0356e-01, ..., 8.9760e-01,
1.5929e+00, 1.2365e-01],
[ 8.1427e-01, -1.9505e-01, -1.0069e-01, ..., 3.8018e-01,
2.6462e-01, 7.0772e-01]],
[[-2.7355e+00, 1.3049e-01, -1.1529e+00, ..., -1.5772e+00,
-2.3184e+00, -6.2460e-01],
[-4.0398e+00, -2.7329e+00, -2.6216e+00, ..., -2.2946e+00,
-4.2200e+00, -1.6911e+00],
[-3.1345e+00, -2.5600e+00, -3.0020e+00, ..., -2.6961e+00,
-3.1017e+00, -1.4023e+00],
...,
[-1.0290e+00, -1.2588e+00, -1.4472e+00, ..., -2.3943e-01,
1.0128e-02, 7.7065e-01],
[-2.5862e+00, -6.7589e-01, -3.1354e-01, ..., 1.1885e-01,
-4.7434e-02, 6.7354e-01],
[-1.5356e+00, -1.1684e+00, -8.7954e-01, ..., -4.8263e-01,
-6.5805e-01, -1.7732e-01]],
[[ 2.3676e+00, 2.9248e+00, 2.4399e+00, ..., 1.5520e+00,
4.7207e-01, 2.6329e+00],
[ 8.1148e-01, 6.7459e-01, 2.3531e+00, ..., 1.4536e+00,
3.8107e-01, 2.7368e+00],
[ 1.3625e+00, 1.5940e+00, 1.9691e+00, ..., 1.2004e+00,
1.7753e+00, 2.8815e+00],
...,
[ 8.7285e-01, 2.0544e+00, 2.0086e+00, ..., 6.9715e-01,
5.6164e-01, 1.8535e+00],
[ 9.8807e-01, 2.2396e+00, 2.3829e+00, ..., 1.0643e+00,
8.3240e-01, 1.7576e+00],
[ 2.3687e+00, 1.8363e+00, 2.3681e+00, ..., 1.9451e+00,
1.0866e+00, 1.7740e+00]],
...,
[[-2.4622e+00, -6.6651e+00, -5.4985e+00, ..., -5.3630e+00,
-4.4892e+00, -5.7048e+00],
[-3.1122e+00, -7.3281e+00, -5.0469e+00, ..., -6.0338e+00,
-5.1505e+00, -5.2527e+00],
[-2.0568e+00, -5.8122e+00, -3.8883e+00, ..., -5.1014e+00,
-3.9823e+00, -3.3190e+00],
...,
[-2.7153e+00, -5.4890e+00, -3.7131e+00, ..., -5.1334e+00,
-2.7938e+00, -2.9034e+00],
[-2.5376e+00, -5.4575e+00, -4.0100e+00, ..., -4.5018e+00,
-3.2368e+00, -4.8780e+00],
[-2.6084e+00, -5.8380e+00, -2.8550e+00, ..., -2.9708e+00,
-1.6707e+00, -3.2906e+00]],
[[ 4.2698e-01, 1.0957e+00, 4.0258e-01, ..., 1.2520e+00,
1.5850e+00, 1.6928e+00],
[ 1.2836e+00, 7.9109e-01, -9.5725e-02, ..., 8.5637e-01,
1.2788e+00, 2.8039e+00],
[ 1.3053e+00, 7.2250e-01, 1.6670e-01, ..., 1.5408e+00,
1.1364e+00, 2.0571e+00],
...,
[ 1.7008e+00, 1.4761e-01, 2.5733e-01, ..., 2.2067e+00,
1.9380e+00, 1.4529e+00],
[ 2.0313e+00, 1.2600e+00, 2.2532e+00, ..., 1.9535e+00,
1.6848e+00, 2.5790e+00],
[ 1.1742e+00, 8.8848e-01, 1.3500e+00, ..., 1.2964e+00,
1.4823e+00, 1.5455e+00]],
[[-6.2021e-01, 7.9354e-02, -1.5458e-01, ..., 1.5008e-01,
-6.7300e-02, 1.0144e+00],
[-1.4531e+00, -2.0236e+00, -1.5808e+00, ..., -2.3077e+00,
-8.6649e-01, -1.4276e-01],
[-2.2689e+00, -1.2394e+00, -5.2607e-01, ..., -7.1119e-01,
-4.5240e-01, -8.3773e-02],
...,
[-1.6764e+00, -1.2055e+00, -1.1154e+00, ..., -4.0769e-01,
-4.9184e-01, -4.4408e-01],
[-1.5928e+00, -1.2653e+00, -7.9936e-01, ..., -2.0147e-01,
-3.7452e-01, 4.2993e-01],
[-2.4471e-01, 4.2436e-01, -2.0356e-01, ..., 2.1274e-01,
1.0054e-03, 5.9830e-01]]],
[[[ 7.4201e-01, 1.2610e+00, 7.4161e-01, ..., 8.8721e-01,
8.4099e-01, 2.5465e-01],
[-5.5783e-01, -1.3404e+00, -1.0227e+00, ..., 1.4167e-01,
4.9354e-01, -4.5717e-01],
[-5.8231e-01, -2.8411e+00, -2.7440e-01, ..., -4.5943e-02,
2.2127e-01, 3.4512e-01],
...,
[ 8.1639e-01, 1.1791e-01, 3.3228e-01, ..., -1.9409e-02,
2.3789e-01, 6.4166e-01],
[-2.5496e-01, -1.0229e+00, 1.8317e-03, ..., 6.2830e-01,
9.0762e-01, 1.3571e-01],
[ 2.2888e-01, -7.4040e-01, -8.5653e-01, ..., 1.4239e-02,
-3.0744e-01, 5.1345e-01]],
[[-3.7887e+00, -2.2320e+00, -3.3492e+00, ..., -1.5393e+00,
-2.9357e+00, -9.8129e-01],
[-5.7952e+00, -3.5643e+00, -4.0656e+00, ..., -1.2072e+00,
-2.5538e+00, -3.8246e-01],
[-7.8125e+00, -3.4616e+00, -1.4378e+00, ..., -2.6914e+00,
-3.1786e+00, -8.9630e-01],
...,
[-2.2830e+00, 2.6547e-01, 2.9907e-03, ..., -1.4883e+00,
-1.4481e+00, 3.6528e-01],
[-3.5312e+00, -1.4799e+00, -2.7227e+00, ..., -1.1130e-01,
-7.8462e-01, 3.2730e-01],
[-1.3478e+00, -1.2313e+00, -2.1943e+00, ..., -5.3534e-01,
-3.0230e-01, 1.4389e-01]],
[[ 1.9237e+00, 3.2258e+00, 2.8034e+00, ..., 2.2927e+00,
8.3753e-01, 2.0286e+00],
[ 6.4384e-01, 2.2826e+00, 2.5937e+00, ..., 1.3525e+00,
2.9341e-02, 1.9116e+00],
[ 1.0274e+00, -1.2481e+00, -2.1933e+00, ..., 1.3452e+00,
1.0079e+00, 2.7144e+00],
...,
[ 1.0320e+00, 5.3352e-01, -1.4029e+00, ..., 5.0953e-01,
6.3131e-01, 1.4061e+00],
[ 1.0310e+00, 2.2357e+00, 6.2788e-01, ..., 1.0940e-01,
4.3820e-01, 1.0813e+00],
[ 1.4781e+00, 1.7750e+00, 1.8304e+00, ..., 7.3753e-01,
5.6417e-01, 1.2998e+00]],
...,
[[-2.2167e+00, -5.8296e+00, -4.8997e+00, ..., -4.5471e+00,
-3.3008e+00, -3.8389e+00],
[-3.5817e+00, -6.2056e+00, -3.4697e+00, ..., -5.3335e+00,
-3.6896e+00, -4.1793e+00],
[-1.6802e+00, -5.4711e+00, -4.5150e+00, ..., -4.4962e+00,
-2.7022e+00, -2.0791e+00],
...,
[-1.7122e+00, -3.9884e+00, -3.7865e+00, ..., -5.2189e+00,
-4.0485e+00, -4.3393e+00],
[-1.4477e+00, -4.4044e+00, -4.8738e+00, ..., -6.4791e+00,
-3.8647e+00, -4.8825e+00],
[-1.4399e+00, -4.5952e+00, -4.0198e+00, ..., -4.4622e+00,
-2.0490e+00, -3.2444e+00]],
[[ 6.5673e-01, 8.8012e-01, -3.2879e-02, ..., 1.4110e+00,
8.9308e-01, 1.7053e+00],
[ 1.3947e-02, -4.2992e-01, -6.5360e-01, ..., 2.5701e+00,
1.8112e+00, 2.6193e+00],
[-2.2732e+00, -3.6538e+00, -2.9115e-01, ..., 1.9740e+00,
5.7522e-01, 1.7499e+00],
...,
[ 1.0783e+00, 3.0017e-01, 2.9484e-01, ..., 3.3792e+00,
2.1053e+00, 2.0152e+00],
[ 1.4701e+00, 1.1525e+00, 8.1608e-01, ..., 2.8207e+00,
1.3247e+00, 2.4346e+00],
[ 6.7263e-01, 8.3886e-01, 9.2766e-01, ..., 1.3142e+00,
1.2245e+00, 1.3889e+00]],
[[-1.9216e-01, -1.8030e-01, -9.5142e-01, ..., 5.2727e-01,
1.3331e-01, 1.0823e+00],
[-1.6123e+00, -2.4081e+00, -2.3100e+00, ..., -4.3328e-02,
-7.8789e-02, 6.4666e-01],
[-3.2329e+00, -8.2345e-01, 4.1116e-01, ..., -4.0445e-01,
-4.1636e-01, -5.9034e-01],
...,
[-1.4133e+00, -7.0056e-01, -6.1122e-01, ..., -2.2816e+00,
-1.1277e+00, -6.7058e-01],
[-1.2700e+00, -1.2517e+00, -1.5349e+00, ..., -9.8207e-01,
-8.0973e-01, 5.7463e-02],
[-2.0343e-01, -1.7416e-01, -8.5206e-01, ..., -1.0853e+00,
-7.1271e-01, 3.0182e-01]]]], device='cuda:0'),
tensor([[[[ 6.8053e-01, 1.0253e+00, 7.2469e-01, ..., 8.5602e-01,
1.0123e+00, 2.6137e-01],
[-3.4637e-01, -1.5892e+00, -8.1955e-01, ..., 2.8432e-02,
8.5689e-01, -9.6919e-03],
[ 6.5777e-02, -6.0319e-01, -3.8330e-01, ..., -1.2338e-01,
3.3598e-01, 6.8934e-01],
...,
[ 5.2432e-01, -1.2167e+00, 1.0645e-01, ..., -2.4739e-01,
4.5988e-01, 7.1055e-01],
[-6.8437e-02, -1.8101e+00, -3.1448e-01, ..., 4.4250e-01,
9.1830e-01, 1.6704e-01],
[ 5.3318e-01, 1.0042e-01, -5.5879e-01, ..., -2.3641e-02,
-3.8721e-01, 5.5909e-01]],
[[-3.4272e+00, -1.4132e+00, -3.2901e+00, ..., -1.8094e+00,
-2.9340e+00, -9.7307e-01],
[-4.3281e+00, -1.6566e+00, -2.6097e+00, ..., -1.2473e+00,
-1.0601e+00, 2.7320e-01],
[-3.6399e+00, -2.3588e+00, -3.0817e+00, ..., -2.4265e+00,
-9.9763e-01, -2.3024e-01],
...,
[-5.3014e-01, -1.8154e+00, -4.1616e+00, ..., -1.4628e+00,
-1.3571e+00, 3.8370e-01],
[-2.7474e+00, -2.0313e+00, -2.9760e+00, ..., -2.7979e-01,
-9.6273e-01, 1.5653e-01],
[-1.8380e+00, -1.1746e+00, -1.7817e+00, ..., -4.5254e-01,
-3.1477e-01, 1.9535e-02]],
[[ 2.2118e+00, 3.0674e+00, 2.4303e+00, ..., 2.4243e+00,
6.1391e-01, 1.4424e+00],
[ 1.4396e-01, 1.0804e+00, 1.6084e+00, ..., 1.1859e+00,
-7.5844e-01, 1.2414e+00],
[ 1.0801e-01, 1.2560e+00, 7.2806e-01, ..., 1.7076e-01,
-9.8474e-01, 8.1103e-01],
...,
[-1.1066e+00, -3.8681e-01, 3.9484e-01, ..., -9.3430e-01,
-1.8771e-01, 1.2112e+00],
[-6.8700e-01, 7.4559e-01, 9.0963e-01, ..., -2.0394e-01,
3.1608e-01, 1.1842e+00],
[ 1.5315e+00, 1.6868e+00, 1.3917e+00, ..., 5.6537e-01,
5.7703e-01, 1.3844e+00]],
...,
[[-1.9309e+00, -5.1358e+00, -4.2937e+00, ..., -4.8765e+00,
-3.4533e+00, -3.3626e+00],
[-2.0223e+00, -4.2868e+00, -4.3039e+00, ..., -6.2516e+00,
-4.0963e+00, -4.2750e+00],
[-4.6842e-01, -2.9020e+00, -3.5871e+00, ..., -5.9468e+00,
-3.5978e+00, -1.8697e+00],
...,
[-2.0156e-01, -1.0967e+00, -6.4365e+00, ..., -5.6455e+00,
-2.6597e+00, -3.1885e+00],
[-9.6604e-01, -3.6691e+00, -5.6944e+00, ..., -7.0660e+00,
-3.6592e+00, -4.9936e+00],
[-9.5602e-01, -4.3631e+00, -4.4673e+00, ..., -4.9344e+00,
-1.9338e+00, -3.3091e+00]],
[[ 6.5217e-01, 7.3252e-01, -2.6632e-01, ..., 1.0844e+00,
5.1091e-01, 1.3061e+00],
[ 1.2468e+00, 1.1209e+00, 3.1909e-01, ..., 2.6750e+00,
1.9737e+00, 2.2023e+00],
[ 8.6584e-01, 4.6970e-01, -1.9437e-01, ..., 2.1871e+00,
1.3335e+00, 1.3439e+00],
...,
[ 8.0935e-01, -4.3624e-01, -1.0150e+00, ..., 3.0123e+00,
1.4356e+00, 1.6598e+00],
[ 1.0328e+00, 4.9141e-01, 5.2350e-01, ..., 3.0793e+00,
1.2092e+00, 2.3466e+00],
[ 5.0664e-01, 7.7213e-01, 1.0291e+00, ..., 1.5589e+00,
1.2086e+00, 1.3747e+00]],
[[-2.8280e-01, -2.8544e-01, -1.1107e+00, ..., 4.9850e-01,
4.6279e-01, 1.1275e+00],
[-1.4085e+00, -1.6363e+00, -2.6904e+00, ..., -4.0337e-01,
4.2552e-01, 9.9262e-01],
[-1.6155e+00, -7.3603e-01, -1.9723e+00, ..., -4.8603e-01,
6.8629e-01, -2.2383e-01],
...,
[-1.1918e+00, -9.4109e-01, -2.1427e+00, ..., -1.8046e+00,
-8.8367e-01, -6.4402e-01],
[-1.1337e+00, -2.2205e+00, -2.0454e+00, ..., -1.2525e+00,
-8.7856e-01, 6.1102e-02],
[-5.5184e-01, -7.0607e-02, -2.1606e-01, ..., -1.1515e+00,
-6.8963e-01, 3.1776e-01]]],
[[[ 2.0721e-01, 1.4228e+00, -8.5364e-01, ..., 7.5823e-01,
9.9271e-01, 4.5359e-01],
[-3.6291e+00, -5.5818e+00, -2.9525e+00, ..., -5.4464e-01,
-4.0583e-01, -4.8131e-01],
[-2.8405e-01, -5.9675e+00, -5.2969e+00, ..., -3.4386e+00,
-5.8075e-01, 4.5971e-01],
...,
[ 3.1822e-01, -1.2485e+00, 2.4964e+00, ..., 2.8513e+00,
5.1774e-01, 3.7924e-01],
[-2.3156e-01, -2.8138e+00, 6.5240e-01, ..., -3.2632e+00,
-1.2057e+00, 2.7963e-01],
[ 7.4300e-01, -1.1395e+00, 6.8753e-01, ..., -9.8027e-01,
-2.0733e+00, -1.1546e-01]],
[[-7.0676e+00, -4.4960e+00, -5.8037e+00, ..., -1.6305e+00,
-2.1972e+00, -1.6962e+00],
[-6.7031e+00, 1.0531e+00, -1.7866e-02, ..., -1.0097e+00,
-2.2639e+00, -1.8120e+00],
[-6.0819e+00, -1.3947e+00, -2.5980e+00, ..., -1.9183e+00,
-2.1793e+00, -1.2155e+00],
...,
[-1.0708e+00, -5.3131e-02, -2.3561e+00, ..., -5.1772e+00,
-5.3740e+00, -1.5614e+00],
[-3.1115e+00, 1.8113e-01, -8.9572e-01, ..., -8.7365e+00,
-4.3221e+00, -1.3906e+00],
[-2.2816e+00, -1.6807e+00, -2.7808e+00, ..., -2.4718e+00,
-1.3854e+00, -6.9950e-01]],
[[ 3.9577e+00, 2.3522e+00, -7.1153e-01, ..., 1.7598e+00,
7.8053e-01, 2.5525e+00],
[-9.8707e-01, -1.8257e+00, -7.4843e-01, ..., 1.2753e+00,
2.4714e-01, 2.6858e+00],
[-6.7452e-01, -5.4821e+00, -1.6982e+00, ..., 3.1456e+00,
3.0833e-01, 1.9727e+00],
...,
[ 8.6960e-02, 2.8026e+00, -2.2816e-01, ..., -8.6934e+00,
-9.5361e-01, 9.3697e-01],
[ 2.4652e-01, 3.4017e+00, 5.0634e-01, ..., -7.0856e+00,
6.4570e-01, 1.2866e+00],
[ 1.4079e+00, 1.9740e+00, 2.2575e+00, ..., 3.6985e-02,
5.5360e-01, 9.6545e-01]],
...,
[[-3.6622e+00, -9.5149e+00, -7.3302e+00, ..., -5.2427e+00,
-4.2219e+00, -4.5568e+00],
[-5.7883e+00, -9.9769e+00, -7.1059e+00, ..., -6.3101e+00,
-5.0871e+00, -3.6784e+00],
[-4.6078e+00, -6.6911e+00, -5.2084e+00, ..., -4.8561e+00,
-4.3391e+00, -1.4133e+00],
...,
[-8.4516e-01, -2.3201e+00, -4.7558e+00, ..., -1.7971e+01,
-1.1522e+01, -4.7950e+00],
[-1.2344e+00, -4.1640e+00, -4.3341e+00, ..., -1.3106e+01,
-7.8862e+00, -5.9171e+00],
[-2.2296e+00, -5.3184e+00, -3.7022e+00, ..., -8.5869e+00,
-7.0366e+00, -5.0657e+00]],
[[-2.9880e+00, -4.8979e+00, -2.7324e+00, ..., 1.3989e+00,
1.5976e+00, 1.2183e+00],
[ 1.5206e+00, 1.6550e-01, 8.7544e-01, ..., 1.4705e+00,
1.0852e+00, 1.9562e+00],
[-1.2419e+00, -1.9611e+00, -9.9797e-01, ..., 2.8950e+00,
8.6619e-01, 1.4336e+00],
...,
[ 1.2051e+00, -1.7476e+00, -6.3310e-02, ..., -3.9802e-01,
-4.7306e-01, 1.6004e+00],
[ 1.8061e+00, -3.5346e-01, 2.4562e+00, ..., -1.8109e+00,
4.4128e-01, 2.0669e+00],
[ 9.6966e-01, -5.1352e-01, -1.9122e+00, ..., 1.9678e+00,
1.6349e+00, 1.4394e+00]],
[[-1.6748e+00, 1.2928e+00, -1.5781e+00, ..., -2.2671e-01,
-6.4844e-01, 3.4642e-02],
[-2.9472e+00, 1.2151e+00, 2.3846e+00, ..., -1.2661e+00,
-1.4307e+00, -9.8228e-01],
[-2.8170e+00, 1.0194e+00, -7.4100e-01, ..., -2.6444e+00,
-1.2912e+00, -1.8649e-01],
...,
[-1.7060e+00, -9.6866e-01, -8.0606e-01, ..., 4.8195e-01,
-2.5557e+00, -3.1693e+00],
[-1.6724e+00, -4.2997e-01, -9.7124e-01, ..., -5.5342e+00,
-3.3298e+00, -8.9286e-01],
[-2.9672e-01, 5.0183e-01, 4.7644e-01, ..., -3.2325e-01,
-1.0690e-01, 3.5312e-01]]],
[[[ 4.2997e-01, -1.3552e+00, -1.8485e+00, ..., -1.2260e-02,
-1.0414e-01, -4.2437e-01],
[-2.4489e+00, -6.1623e+00, -5.6053e+00, ..., 4.1672e-01,
6.8861e-01, -2.9161e-01],
[-1.2225e+00, 3.6110e+00, 5.0049e+00, ..., -1.5086e+00,
-1.7297e-01, 7.9349e-01],
...,
[ 1.2346e+00, 2.2397e-01, 4.5958e-01, ..., -1.8176e+01,
-1.1898e+01, -1.2047e+01],
[-1.1450e-01, -3.9116e-01, 5.0416e-01, ..., -9.1840e+00,
6.0026e-01, -6.1625e+00],
[ 7.5931e-01, -5.9277e-01, 2.7421e-01, ..., -4.7813e+00,
5.6005e+00, 8.7842e-01]],
[[-8.7292e+00, -3.9917e+00, -4.0511e+00, ..., -2.6812e+00,
-3.4376e+00, -1.3405e+00],
[-1.8120e+01, -1.6809e+01, -1.4350e+01, ..., -9.0187e-01,
-2.5162e+00, -9.7336e-01],
[-4.7646e+00, -4.4877e+00, -5.3514e+00, ..., -3.0899e+00,
-2.3475e+00, -5.1501e-01],
...,
[-2.9736e-02, 1.2361e+00, 1.7541e-01, ..., -1.1714e+01,
-2.0753e+01, -1.4671e+01],
[-1.3510e+00, 1.7575e+00, 1.6701e+00, ..., -1.6671e+01,
-1.9859e+01, -8.1192e+00],
[-1.2540e+00, -1.6420e-01, 7.0053e-02, ..., -1.5938e+01,
-1.9691e+01, -7.7540e+00]],
[[ 3.5557e+00, 5.5559e+00, 4.7448e+00, ..., 2.0839e+00,
9.2428e-01, 2.1649e+00],
[ 2.2031e+00, 6.3658e+00, 6.7215e+00, ..., 1.5141e+00,
4.2238e-01, 2.1322e+00],
[-4.7753e+00, -8.6467e+00, -9.1656e+00, ..., 2.4146e+00,
1.4692e+00, 1.9641e+00],
...,
[ 9.4554e-01, 6.4950e-02, -1.7474e+00, ..., -4.5475e+00,
-9.5716e+00, -8.8354e+00],
[ 4.9071e-01, 1.6016e+00, -1.9371e+00, ..., -9.5989e+00,
-7.2660e+00, -6.0645e+00],
[ 1.7962e+00, 5.7509e-01, -1.5603e+00, ..., -1.2014e+01,
-6.6150e+00, -4.2250e+00]],
...,
[[-6.8620e+00, -1.3516e+01, -1.0216e+01, ..., -2.6041e+00,
-2.2887e+00, -3.1207e+00],
[-3.6249e+00, -1.5415e+01, -1.1466e+01, ..., -5.3453e+00,
-4.1376e+00, -4.1854e+00],
[-8.0265e+00, -3.1475e+01, -2.6760e+01, ..., -3.6362e+00,
-2.4940e+00, -2.0228e+00],
...,
[-1.5964e+00, -4.2086e+00, -4.9037e-01, ..., -2.5985e+00,
-7.2763e+00, 1.0688e+00],
[-2.1410e+00, -5.0192e+00, -1.3345e+00, ..., -3.9243e+00,
-1.8835e+01, 6.1913e-01],
[-2.8824e+00, -5.5055e+00, -2.4578e+00, ..., -3.8186e-01,
-1.8532e+01, -5.5732e+00]],
[[-3.1905e+00, -3.3199e-03, -6.8087e-01, ..., 6.9408e-01,
9.7167e-01, 1.7927e+00],
[-1.0390e+01, -1.1638e+01, -4.9554e+00, ..., 1.2634e+00,
1.1459e+00, 2.3847e+00],
[-2.0418e-01, -4.9972e+00, -6.6111e+00, ..., 2.0175e+00,
1.1752e+00, 1.6884e+00],
...,
[ 1.1724e+00, -4.9568e-01, -5.6503e-02, ..., -1.3406e+01,
-1.6862e+01, -6.2157e+00],
[ 1.6937e+00, 9.9434e-01, 1.5209e+00, ..., -1.5940e+01,
-2.6176e+01, -9.0924e+00],
[ 9.7726e-01, 8.2238e-01, 9.4632e-01, ..., -4.5926e+00,
-1.4184e+01, -5.4335e+00]],
[[-4.6362e-01, -2.5992e+00, -2.8045e+00, ..., -5.5035e-01,
-9.1026e-01, 3.5642e-01],
[-2.3269e+00, -4.4487e+00, -3.2454e+00, ..., -4.3356e-01,
-1.8736e-01, 5.2337e-01],
[ 2.3401e+00, 4.4109e+00, 6.1905e+00, ..., 3.2851e-01,
-2.3550e-02, 2.5810e-01],
...,
[-1.5870e+00, -1.0943e+00, 4.0631e-01, ..., -3.1535e+00,
-8.9647e+00, -4.5434e+00],
[-1.8815e+00, -1.0141e+00, 1.0186e-01, ..., -6.5319e-01,
-7.2742e+00, -3.2295e+00],
[ 1.9644e-02, 5.0542e-01, -7.3984e-01, ..., 2.5353e+00,
-2.7221e+00, -5.3489e-01]]],
[[[ 8.5019e-01, 1.4458e+00, 1.1861e+00, ..., 1.4002e+00,
1.0797e+00, 9.1177e-01],
[-9.4951e-01, -2.0291e+00, -1.1657e+00, ..., -4.3999e-01,
-5.1073e-01, -1.4345e+00],
[-1.1676e-01, -7.9257e-01, 2.2721e-02, ..., -3.3509e-01,
3.2579e-01, 4.9754e-01],
...,
[ 4.1655e-01, -2.3553e-01, 3.4370e-01, ..., 8.3525e-01,
-1.7427e-01, 4.8939e-01],
[ 2.7196e-01, -8.2479e-01, -6.6059e-01, ..., 1.5493e+00,
1.1314e+00, 7.4545e-02],
[ 5.0308e-01, 2.6673e-01, -4.6697e-01, ..., -1.3195e-01,
4.1268e-02, 5.8195e-01]],
[[-3.7889e+00, -6.8861e-01, -1.5689e+00, ..., -1.6219e+00,
-2.5306e+00, -1.1958e+00],
[-3.9192e+00, -2.6948e+00, -1.8269e+00, ..., -1.6381e+00,
-2.5874e+00, -8.1416e-01],
[-2.4305e+00, -1.5972e+00, -1.5664e+00, ..., -2.6522e+00,
-3.2542e+00, -1.3482e+00],
...,
[-1.0448e+00, -1.1412e+00, -2.1209e+00, ..., -2.6084e+00,
-1.9438e+00, -3.5590e-01],
[-1.9088e+00, -3.2667e-01, -1.3468e+00, ..., -1.8700e+00,
-1.7703e+00, -3.3590e-01],
[ 2.6126e-01, 5.2865e-02, -1.8706e-01, ..., -1.6732e+00,
-9.9026e-01, 4.1806e-01]],
[[ 2.9517e+00, 4.2483e+00, 3.3969e+00, ..., 2.9227e+00,
9.6449e-01, 2.8892e+00],
[ 3.6842e-01, 1.0693e+00, 2.2645e+00, ..., 1.8762e+00,
-6.0223e-01, 2.6014e+00],
[ 1.1159e+00, 1.5596e+00, 1.7430e+00, ..., 1.7792e+00,
3.0294e-01, 2.1384e+00],
...,
[ 4.5104e-01, 4.8917e-01, 5.7421e-02, ..., -1.6639e+00,
6.2891e-01, 1.7060e+00],
[ 7.6762e-02, 1.2804e+00, 7.1570e-01, ..., -3.2216e+00,
-1.0128e+00, 9.6391e-01],
[ 1.3348e+00, 8.4174e-01, 1.7780e+00, ..., -3.4562e-01,
5.0757e-01, 9.3890e-01]],
...,
[[-1.9443e+00, -7.2362e+00, -5.3918e+00, ..., -4.0452e+00,
-3.1968e+00, -5.2604e+00],
[-2.5425e+00, -7.0289e+00, -4.2326e+00, ..., -3.7946e+00,
-2.6353e+00, -4.2611e+00],
[-1.1297e+00, -4.7476e+00, -3.6991e+00, ..., -2.1471e+00,
-1.0852e+00, -1.3311e+00],
...,
[-2.6596e+00, -5.8564e+00, -4.0338e+00, ..., -3.1720e+00,
-5.1440e+00, -4.1818e+00],
[-2.8710e+00, -7.3416e+00, -4.8568e+00, ..., -5.8024e+00,
-6.0545e+00, -5.5463e+00],
[-2.8707e+00, -7.5817e+00, -4.4402e+00, ..., -9.4085e-01,
-2.5483e+00, -4.4307e+00]],
[[-1.7914e-01, 3.6311e-01, 5.3041e-01, ..., 5.9439e-01,
7.9956e-01, 1.4203e+00],
[ 1.6840e+00, 1.1430e+00, 1.7647e+00, ..., 1.9322e+00,
2.2710e+00, 3.0407e+00],
[ 1.1869e+00, 2.9842e-01, 7.9581e-01, ..., 1.0510e+00,
5.4282e-01, 2.1620e+00],
...,
[ 1.0783e+00, 5.3315e-02, -1.0742e-02, ..., 3.3706e-01,
1.4081e+00, 1.7676e+00],
[ 1.5464e+00, 1.0167e+00, 6.7876e-01, ..., -1.6637e+00,
1.6744e-01, 2.2964e+00],
[ 1.0808e+00, 7.3972e-01, 6.8797e-01, ..., -2.3839e-01,
8.2830e-01, 1.5138e+00]],
[[-5.9582e-01, 2.9323e-01, 1.1576e-01, ..., 5.7233e-01,
4.9534e-01, 1.3719e+00],
[-1.3777e+00, -1.7916e+00, -1.6495e+00, ..., -7.5734e-01,
-4.8186e-01, 2.2644e-01],
[-2.2278e+00, -8.4118e-01, 3.3605e-01, ..., -3.7795e-01,
8.9769e-02, 7.0315e-01],
...,
[-1.5391e+00, -2.0341e+00, -1.4632e+00, ..., 2.1578e-01,
-7.6021e-01, -8.5654e-01],
[-1.3124e+00, -2.0322e+00, -9.2903e-01, ..., -6.0565e-01,
-1.9845e+00, -7.1558e-01],
[ 8.0561e-02, 3.7862e-02, -2.7321e-01, ..., -2.4286e-01,
-1.2592e+00, 4.4036e-01]]],
[[[ 1.9678e-01, 5.5525e-01, 4.6049e-02, ..., 3.9043e-01,
3.7555e-01, 2.3889e-01],
[-1.0595e+00, -8.8005e-01, -3.4431e-01, ..., 3.5468e-01,
2.9295e-01, -1.2008e+00],
[-3.9102e-01, -1.3174e+00, -5.8020e-01, ..., -5.0303e-01,
8.6879e-01, 6.4122e-01],
...,
[ 8.4654e-02, -9.8619e-01, -1.8964e+00, ..., 2.2175e-01,
1.1672e+00, 4.9379e-01],
[-5.3164e-01, -3.0084e+00, -1.9771e+00, ..., -4.3813e-02,
1.3205e+00, -1.4861e-01],
[ 2.8764e-01, -1.7420e+00, 7.2216e-01, ..., 2.9631e-02,
5.1628e-01, 5.9108e-01]],
[[ 6.8705e-01, -7.3766e-01, -1.7462e+00, ..., -1.7018e+00,
-2.2299e+00, -1.0768e+00],
[ 2.8919e-01, 2.1047e-02, -1.6513e+00, ..., -4.8560e-01,
-1.2923e+00, -2.8708e-01],
[-2.8028e-01, -1.7930e+00, -4.6559e+00, ..., -1.3434e+00,
-1.5867e+00, -7.8655e-01],
...,
[-3.0549e+00, -5.0166e+00, -9.1049e+00, ..., -7.9848e-01,
4.6707e-01, 4.3296e-01],
[-3.7068e+00, -3.7083e+00, -8.3166e+00, ..., -2.2009e+00,
-3.7657e-01, 2.3106e-01],
[-1.2893e+00, -2.2500e+00, -4.0658e+00, ..., -2.2957e+00,
-6.0965e-01, -5.7451e-01]],
[[-5.9250e-01, -3.7207e-01, 1.2180e+00, ..., 2.7667e+00,
1.4233e+00, 2.9301e+00],
[-2.1913e+00, -1.6957e+00, 2.2978e-01, ..., 1.9138e+00,
6.0597e-02, 2.5587e+00],
[-1.4295e+00, -4.0990e+00, -2.5952e+00, ..., 7.5324e-01,
3.0297e-01, 1.8658e+00],
...,
[ 9.8120e-01, 4.2958e+00, 6.4627e-01, ..., -9.8069e-01,
-8.2255e-03, 1.4635e+00],
[ 6.6474e-01, 4.3636e+00, -1.4390e+00, ..., -1.0691e+00,
5.6065e-01, 1.5282e+00],
[ 1.5236e+00, 2.2282e+00, -4.5574e+00, ..., -8.2568e-01,
8.4262e-01, 1.9885e+00]],
...,
[[ 3.7840e-01, 4.3145e-02, 9.0101e-01, ..., -3.3965e+00,
-2.5396e+00, -3.9848e+00],
[ 1.1563e+00, -3.3933e-01, -4.6076e-01, ..., -5.4240e+00,
-3.9205e+00, -4.3893e+00],
[ 3.7888e-01, 5.9325e-01, -2.4993e+00, ..., -3.1494e+00,
-2.5790e+00, -1.6953e+00],
...,
[-1.6168e+00, -4.8871e+00, -7.9253e+00, ..., -5.0750e+00,
-4.2364e+00, -1.6237e+00],
[-1.4772e+00, -4.7223e+00, -7.4595e+00, ..., -7.0493e+00,
-4.3037e+00, -3.8100e+00],
[-1.5272e+00, -5.0523e+00, -3.7249e+00, ..., -5.5709e+00,
-2.9060e+00, -2.8401e+00]],
[[-3.9641e-01, -1.9785e+00, -4.5611e-01, ..., 9.2035e-01,
1.0929e+00, 1.2484e+00],
[-1.2974e+00, -2.8389e+00, -1.4422e+00, ..., 1.5991e+00,
2.4123e+00, 2.6136e+00],
[-1.1442e+00, -2.2460e+00, -3.1650e+00, ..., 6.9448e-01,
1.4766e+00, 1.9943e+00],
...,
[ 7.2442e-01, -2.1792e+00, -5.2597e+00, ..., 7.0508e-01,
2.4309e+00, 1.6948e+00],
[ 1.4619e+00, -4.2184e-01, -3.6004e+00, ..., 1.5790e+00,
1.7452e+00, 2.4630e+00],
[ 6.7269e-01, -2.6015e-01, 2.8998e-02, ..., 1.0341e+00,
1.2883e+00, 1.3637e+00]],
[[-1.0463e-01, -6.3241e-01, -6.4700e-01, ..., -2.2597e-01,
-1.5277e-01, 8.8719e-01],
[-7.7652e-01, -4.9496e-01, -3.2329e-01, ..., 1.2804e-01,
-1.0806e-01, 3.6934e-01],
[-1.1798e+00, 1.7781e-01, -1.8210e-01, ..., -2.4785e-01,
6.0149e-03, 2.6034e-01],
...,
[-1.7338e+00, -2.7398e+00, -3.2142e+00, ..., -8.0488e-01,
-2.0373e-01, -3.1242e-01],
[-1.4735e+00, -2.8846e+00, -3.0483e+00, ..., -1.7626e+00,
-9.5179e-01, 3.4683e-01],
[-1.6491e-01, -3.1276e-01, 1.6132e-01, ..., -2.2405e+00,
3.8894e-02, 3.5102e-01]]]], device='cuda:0'),
tensor([[[[ 5.9704e-01, 8.5068e-01, 5.0793e-01, ..., 5.5815e-01,
4.9492e-01, 5.8908e-02],
[-2.8371e-01, -6.4687e-01, -3.2707e-01, ..., 3.0923e-01,
3.4395e-02, -4.3858e-01],
[-1.4274e-01, -1.5888e-02, 1.8376e-01, ..., -2.7102e-02,
7.3113e-02, 6.8244e-01],
...,
[ 4.7593e-01, -4.3941e-01, -2.6688e-02, ..., 3.5450e-01,
8.5889e-01, 6.8339e-01],
[ 7.5572e-03, -7.6032e-01, -1.6591e-01, ..., 9.1345e-01,
1.5443e+00, 8.5929e-02],
[ 8.1527e-01, -2.2430e-01, -1.7426e-01, ..., 3.5093e-01,
3.6175e-01, 7.5441e-01]],
[[-2.1348e+00, -6.5813e-01, -1.4895e+00, ..., -2.0625e+00,
-2.4134e+00, -1.6864e+00],
[-2.2430e+00, -5.6938e-01, -1.1441e+00, ..., -1.7908e+00,
-2.6360e+00, -1.9150e+00],
[-2.1043e+00, -1.3242e+00, -1.8273e+00, ..., -2.0063e+00,
-2.3595e+00, -1.2163e+00],
...,
[-7.1450e-01, -1.1414e+00, -1.8939e+00, ..., -6.5401e-01,
-2.0855e-01, 7.2753e-01],
[-2.6867e+00, -3.0607e-01, -6.4936e-01, ..., -3.4542e-02,
-7.7894e-02, 6.9297e-01],
[-1.6089e+00, -1.2599e+00, -8.6818e-01, ..., -3.9864e-01,
-5.5125e-01, -1.5227e-01]],
[[ 1.6838e+00, 2.2715e+00, 1.5792e+00, ..., 1.2175e+00,
7.2719e-01, 2.2129e+00],
[-2.5638e-01, 4.4686e-01, 6.9055e-01, ..., 4.7659e-01,
4.1046e-01, 2.4644e+00],
[ 1.3241e-01, 3.0584e-01, 8.7716e-01, ..., -3.5116e-01,
7.9413e-01, 2.0756e+00],
...,
[ 8.1086e-01, 2.7447e-01, 2.0762e+00, ..., 4.1372e-01,
8.9412e-02, 1.6375e+00],
[ 1.1398e+00, 2.1834e+00, 1.9277e+00, ..., 9.8326e-01,
6.7851e-01, 1.7050e+00],
[ 2.4326e+00, 1.8346e+00, 2.3492e+00, ..., 1.9055e+00,
1.0153e+00, 1.7586e+00]],
...,
[[-1.6746e+00, -4.4996e+00, -2.7408e+00, ..., -2.0426e+00,
-1.7325e+00, -3.4062e+00],
[-1.9986e+00, -4.7916e+00, -2.4313e+00, ..., -2.0920e+00,
-2.0881e+00, -3.1996e+00],
[-1.3841e+00, -5.0545e+00, -2.8979e+00, ..., -2.4527e+00,
-2.4730e+00, -2.2189e+00],
...,
[-2.6439e+00, -4.5218e+00, -2.4417e+00, ..., -3.3832e+00,
-2.0347e+00, -2.6375e+00],
[-2.5828e+00, -5.5629e+00, -3.8382e+00, ..., -4.3564e+00,
-3.1004e+00, -4.8133e+00],
[-2.6765e+00, -6.0008e+00, -2.8461e+00, ..., -2.8855e+00,
-1.5689e+00, -3.2869e+00]],
[[ 9.1003e-01, 1.4416e+00, 8.9943e-01, ..., 8.2035e-01,
1.5010e+00, 1.3792e+00],
[ 1.0709e+00, 9.1797e-01, 7.2338e-01, ..., 8.2903e-01,
8.8577e-01, 2.0429e+00],
[ 6.4738e-01, -2.4377e-02, -2.1981e-01, ..., 6.3790e-01,
1.4691e-01, 1.4478e+00],
...,
[ 1.5014e+00, -6.6178e-02, 2.1245e-01, ..., 1.4744e+00,
1.8745e+00, 1.4946e+00],
[ 2.0157e+00, 1.2763e+00, 1.8384e+00, ..., 1.9138e+00,
1.7236e+00, 2.5690e+00],
[ 1.1462e+00, 9.1297e-01, 1.3562e+00, ..., 1.3225e+00,
1.5216e+00, 1.5164e+00]],
[[-6.9583e-02, -6.4476e-01, -1.0554e+00, ..., -6.3595e-01,
-8.7340e-01, -7.4148e-02],
[-8.5112e-01, -2.4292e+00, -2.4382e+00, ..., -1.7411e+00,
-1.0827e+00, -7.7945e-01],
[-1.2949e+00, -1.5654e+00, -1.2313e+00, ..., -8.3336e-01,
-3.4492e-01, -2.0631e-01],
...,
[-1.7515e+00, -1.7916e+00, -3.3180e-01, ..., -1.7089e-01,
-4.5489e-01, -4.3316e-01],
[-1.7823e+00, -1.3705e+00, -7.2252e-01, ..., -3.8234e-01,
-4.1007e-01, 4.2153e-01],
[-2.9079e-01, 3.6360e-01, -2.1903e-01, ..., 2.1524e-01,
6.3301e-02, 6.4288e-01]]],
[[[ 7.1378e-01, 1.0568e+00, 6.7429e-01, ..., 1.0086e+00,
9.3217e-01, 2.8974e-01],
[-3.3215e-01, -1.4292e+00, -6.4654e-01, ..., 9.3683e-02,
4.7733e-01, -5.5541e-01],
[ 7.2838e-02, -8.2994e-01, -5.0986e-01, ..., -2.0319e-01,
2.4545e-01, 4.8388e-01],
...,
[ 1.0488e-01, -1.2669e+00, -9.3236e-01, ..., 1.3218e-01,
3.2791e-01, 6.1574e-01],
[-4.7565e-01, -1.8179e+00, -9.0542e-01, ..., 6.1069e-01,
8.4692e-01, 9.9336e-02],
[ 3.5723e-01, -5.0827e-01, -7.5953e-01, ..., -7.9953e-02,
-1.7512e-01, 5.3236e-01]],
[[-3.0448e+00, -1.4155e+00, -2.9937e+00, ..., -1.6309e+00,
-2.9570e+00, -9.9283e-01],
[-4.3682e+00, -1.7497e+00, -2.2845e+00, ..., -1.3212e+00,
-2.6816e+00, -5.4738e-01],
[-4.3616e+00, -2.5450e+00, -2.9548e+00, ..., -2.5203e+00,
-2.8073e+00, -9.3971e-01],
...,
[-2.7669e+00, -1.9234e+00, -3.0063e+00, ..., -1.1152e+00,
-1.3431e+00, 3.6974e-01],
[-3.6088e+00, -1.4016e+00, -2.2974e+00, ..., -3.1674e-01,
-1.0061e+00, 2.2219e-01],
[-1.2568e+00, -8.9944e-01, -1.6866e+00, ..., -7.7016e-01,
-4.3913e-01, 4.7989e-02]],
[[ 1.7735e+00, 2.7719e+00, 2.3198e+00, ..., 2.3424e+00,
8.0622e-01, 2.1016e+00],
[ 1.3223e-01, 1.1492e+00, 1.4217e+00, ..., 1.4811e+00,
5.0061e-02, 2.0701e+00],
[ 6.5160e-01, 1.3505e+00, 6.7400e-01, ..., 3.9545e-01,
6.9767e-01, 1.9341e+00],
...,
[ 9.9840e-01, 1.1533e+00, 9.2813e-01, ..., -3.0148e-01,
-1.0023e-03, 1.2021e+00],
[ 6.6273e-01, 1.8105e+00, 7.5993e-01, ..., 2.8716e-02,
3.2260e-01, 1.0527e+00],
[ 1.5367e+00, 1.5375e+00, 1.5147e+00, ..., 6.2013e-01,
5.9457e-01, 1.2597e+00]],
...,
[[-1.5718e+00, -4.8813e+00, -3.8621e+00, ..., -4.7784e+00,
-3.4733e+00, -4.1473e+00],
[-1.8856e+00, -4.1025e+00, -4.1375e+00, ..., -5.7674e+00,
-3.9070e+00, -4.4201e+00],
[-9.6823e-01, -3.2499e+00, -3.5544e+00, ..., -5.1399e+00,
-2.1893e+00, -1.0814e+00],
...,
[-1.4978e+00, -3.9334e+00, -4.1471e+00, ..., -4.6119e+00,
-2.4683e+00, -3.0390e+00],
[-1.4076e+00, -4.4048e+00, -5.2004e+00, ..., -6.4679e+00,
-3.6810e+00, -4.5107e+00],
[-1.5537e+00, -4.8323e+00, -4.1308e+00, ..., -4.6267e+00,
-2.0254e+00, -3.0356e+00]],
[[ 8.0018e-01, 5.9422e-01, -1.2942e-01, ..., 1.3835e+00,
9.7003e-01, 1.6963e+00],
[ 1.1751e+00, 9.4772e-01, 5.0844e-01, ..., 2.6379e+00,
1.8928e+00, 2.7277e+00],
[ 5.7361e-01, 5.1379e-01, 5.2419e-02, ..., 2.2567e+00,
1.0461e+00, 1.7257e+00],
...,
[ 8.1731e-01, -4.5569e-02, -3.9879e-01, ..., 2.7438e+00,
1.6106e+00, 1.7485e+00],
[ 1.5146e+00, 1.3592e+00, 1.0861e+00, ..., 2.7839e+00,
1.4252e+00, 2.4800e+00],
[ 6.6677e-01, 8.8175e-01, 9.3972e-01, ..., 1.2736e+00,
1.1610e+00, 1.3594e+00]],
[[-2.4393e-01, -2.5722e-01, -9.6481e-01, ..., 5.8607e-01,
1.8856e-01, 1.1036e+00],
[-1.3528e+00, -1.5381e+00, -2.1106e+00, ..., -1.6312e-01,
-7.5274e-02, 5.5544e-01],
[-1.6464e+00, -9.8066e-01, -1.1908e+00, ..., -1.9734e-01,
-2.9013e-02, -2.0052e-01],
...,
[-1.7469e+00, -1.6891e+00, -1.9197e+00, ..., -1.2209e+00,
-8.8282e-01, -6.0052e-01],
[-1.4659e+00, -1.6846e+00, -1.6671e+00, ..., -8.2267e-01,
-8.5818e-01, -5.2277e-02],
[-1.6739e-01, 2.8377e-02, -5.7840e-01, ..., -1.1517e+00,
-6.9600e-01, 1.0849e-01]]],
[[[ 5.9818e-01, 8.6217e-01, 6.0042e-01, ..., 3.3111e-01,
4.3983e-01, 4.8199e-02],
[-2.7111e-01, -8.1315e-01, -5.4646e-01, ..., 1.2053e-01,
8.1215e-02, -2.5454e-01],
[ 1.8441e-01, 3.0687e-01, 4.7950e-02, ..., -8.8500e-02,
3.2232e-02, 6.4526e-01],
...,
[ 3.7125e-01, -2.9176e-01, -1.6955e-01, ..., 5.9537e-02,
6.5815e-01, 7.4728e-01],
[-1.1013e-02, -6.0904e-01, 2.1180e-01, ..., 8.5381e-01,
1.4237e+00, 7.6570e-02],
[ 8.1171e-01, -2.4850e-01, -1.7246e-01, ..., 3.9034e-01,
2.7459e-01, 7.4901e-01]],
[[-2.3995e+00, -5.4460e-01, -1.8426e+00, ..., -1.6422e+00,
-2.2009e+00, -1.4774e+00],
[-2.4420e+00, -5.0625e-01, -1.4853e+00, ..., -1.2641e+00,
-2.2722e+00, -1.6076e+00],
[-3.0699e+00, -2.4018e+00, -1.6278e+00, ..., -1.9817e+00,
-2.2377e+00, -1.1115e+00],
...,
[-7.9754e-01, -6.8139e-01, -1.1576e+00, ..., -9.7537e-01,
-6.3287e-01, 6.3188e-01],
[-2.5223e+00, -2.0264e-01, -3.2370e-01, ..., 4.6068e-02,
7.4972e-02, 7.2385e-01],
[-1.6005e+00, -1.2457e+00, -9.2849e-01, ..., -4.9680e-01,
-5.3572e-01, -1.3139e-01]],
[[ 1.9348e+00, 2.4354e+00, 1.5748e+00, ..., 1.5831e+00,
7.4342e-01, 2.0432e+00],
[-2.0860e-01, 3.0500e-01, 5.2409e-01, ..., 7.7572e-01,
2.5293e-01, 2.1774e+00],
[ 3.9921e-01, 5.5442e-01, 9.8875e-01, ..., -5.8968e-02,
7.5568e-01, 1.9317e+00],
...,
[ 9.2695e-01, 9.9685e-01, 2.1264e+00, ..., 4.9250e-01,
4.6404e-01, 1.9076e+00],
[ 1.1117e+00, 2.1816e+00, 1.8559e+00, ..., 1.1882e+00,
8.6685e-01, 1.7199e+00],
[ 2.4345e+00, 1.8204e+00, 2.3314e+00, ..., 1.9860e+00,
1.0298e+00, 1.6920e+00]],
...,
[[-2.0861e+00, -4.9941e+00, -3.0568e+00, ..., -2.0696e+00,
-1.6491e+00, -2.9252e+00],
[-2.4891e+00, -5.1910e+00, -2.7906e+00, ..., -2.4132e+00,
-2.2235e+00, -2.9003e+00],
[-2.0550e+00, -6.6399e+00, -4.4076e+00, ..., -2.3614e+00,
-2.4423e+00, -2.2953e+00],
...,
[-2.3489e+00, -4.4875e+00, -2.9591e+00, ..., -3.7295e+00,
-2.5020e+00, -2.4633e+00],
[-2.5703e+00, -5.4628e+00, -3.8069e+00, ..., -4.5420e+00,
-3.5273e+00, -4.9655e+00],
[-2.6461e+00, -5.8812e+00, -2.6431e+00, ..., -3.0577e+00,
-1.6814e+00, -3.3571e+00]],
[[ 9.5542e-01, 1.6551e+00, 7.9682e-01, ..., 1.0962e+00,
1.5281e+00, 1.3651e+00],
[ 1.1631e+00, 1.0569e+00, 9.3815e-01, ..., 9.6629e-01,
8.4415e-01, 1.8925e+00],
[ 2.6302e-01, -5.7061e-01, 1.6509e-01, ..., 4.5258e-01,
1.4699e-01, 1.3645e+00],
...,
[ 1.7062e+00, 6.2297e-01, 1.7200e+00, ..., 1.6422e+00,
1.6116e+00, 1.5699e+00],
[ 2.0506e+00, 1.1814e+00, 1.7382e+00, ..., 2.0888e+00,
1.6974e+00, 2.5822e+00],
[ 1.1562e+00, 8.7844e-01, 1.3126e+00, ..., 1.3504e+00,
1.5169e+00, 1.5070e+00]],
[[-9.1153e-02, -6.3852e-01, -1.0877e+00, ..., -5.7362e-01,
-8.7551e-01, -1.2428e-01],
[-8.9947e-01, -2.5468e+00, -2.4727e+00, ..., -1.2106e+00,
-1.0499e+00, -6.4787e-01],
[-1.6569e+00, -1.6429e+00, -7.1464e-01, ..., -2.6462e-01,
-4.0939e-01, -3.0842e-01],
...,
[-1.7537e+00, -1.7895e+00, -9.6062e-01, ..., -8.2244e-01,
-7.0336e-01, -3.1116e-01],
[-1.7422e+00, -1.3542e+00, -5.6467e-01, ..., -3.4564e-01,
-2.8375e-01, 4.7544e-01],
[-2.9210e-01, 3.5207e-01, -2.5090e-01, ..., 2.1255e-01,
6.4976e-02, 6.3797e-01]]],
[[[ 9.3843e-01, 1.5910e+00, 1.0710e+00, ..., 3.2906e-01,
2.2456e-01, -2.5852e-01],
[-8.7546e-01, -1.6874e+00, -9.0619e-01, ..., 7.1614e-02,
3.0010e-01, -5.8161e-01],
[-1.7494e-01, -7.5697e-01, -6.2262e-01, ..., 6.5458e-01,
1.6350e+00, 7.3648e-01],
...,
[ 5.9697e-01, 3.7960e-01, 5.1705e-01, ..., 1.4879e+00,
2.4811e+00, 1.3348e+00],
[-1.7705e-01, -1.1019e+00, -3.6545e-02, ..., 6.9012e-01,
4.9332e-01, -9.0075e-01],
[ 8.7420e-01, -3.2413e-01, -3.8095e-01, ..., -1.4318e-01,
8.8439e-02, 6.3609e-01]],
[[-3.8265e+00, -1.3472e+00, -1.9367e+00, ..., -2.8579e+00,
-3.4600e+00, -8.8998e-01],
[-4.1734e+00, -3.3524e+00, -1.9121e+00, ..., -2.3442e+00,
-3.5234e+00, -6.1892e-01],
[-3.0576e+00, -3.4596e+00, -5.4328e+00, ..., -1.1926e+00,
-3.2063e+00, -1.0396e+00],
...,
[-9.4879e-01, -6.4474e-01, -1.5936e+00, ..., 9.9608e-01,
-6.8854e-01, 3.9410e-01],
[-3.1361e+00, 1.2563e-02, -7.7624e-01, ..., -4.0488e+00,
-5.4761e+00, -2.5181e+00],
[-2.2184e+00, -1.4725e+00, -1.6734e+00, ..., -1.3887e+00,
-1.4885e+00, -4.4851e-01]],
[[ 2.6851e+00, 3.7969e+00, 3.0735e+00, ..., 2.7246e+00,
7.1853e-01, 2.0746e+00],
[-1.1340e-01, 8.6374e-01, 2.3639e+00, ..., 1.5631e+00,
1.0605e-02, 2.0121e+00],
[ 8.1856e-01, 1.4595e+00, 1.1763e+00, ..., -5.6725e-01,
2.9516e-01, 1.1690e+00],
...,
[ 7.6328e-01, 1.3459e+00, 1.6121e+00, ..., -1.6102e+00,
-2.5099e+00, -9.1564e-01],
[ 6.0294e-01, 2.5455e+00, 2.2953e+00, ..., -1.6960e+00,
-2.8374e+00, -9.0734e-01],
[ 1.7725e+00, 1.3712e+00, 1.8410e+00, ..., 1.8886e+00,
1.8429e+00, 1.9597e+00]],
...,
[[-1.6071e+00, -6.8847e+00, -5.6251e+00, ..., -3.1313e+00,
-2.4600e+00, -3.7090e+00],
[-2.2694e+00, -6.2120e+00, -4.5333e+00, ..., -2.6919e+00,
-3.2237e+00, -4.8288e+00],
[-2.3999e-02, -1.9294e+00, -2.1619e+00, ..., -3.2706e+00,
-4.1941e+00, -3.7370e+00],
...,
[-1.1730e+00, -2.4329e+00, -1.7653e+00, ..., -3.6027e+00,
-2.6957e+00, -4.2093e+00],
[-1.5220e+00, -3.7040e+00, -3.3925e+00, ..., -5.3722e+00,
-3.7172e+00, -4.7348e+00],
[-2.5697e+00, -5.4471e+00, -2.8572e+00, ..., -3.8378e+00,
-3.0519e+00, -4.0779e+00]],
[[-6.3405e-01, -1.4918e-01, 1.8320e-01, ..., 4.3680e-01,
1.0490e+00, 1.7257e+00],
[ 1.5769e+00, 9.7504e-01, 1.7083e+00, ..., 3.1766e-01,
1.5079e+00, 3.0700e+00],
[ 4.4394e-01, -6.5150e-01, -9.2199e-01, ..., -1.9981e-01,
2.9119e-01, 2.2013e+00],
...,
[ 1.3091e+00, -3.3424e-01, 4.7501e-01, ..., 1.1579e+00,
9.1246e-01, 1.2777e+00],
[ 2.0206e+00, 1.5080e+00, 1.9567e+00, ..., 5.6139e-01,
-4.2445e-01, 1.2897e+00],
[ 1.0023e+00, 5.7019e-01, 1.0196e+00, ..., 1.1491e+00,
1.3388e+00, 1.8170e+00]],
[[-4.4271e-01, 5.5388e-01, 1.1577e-01, ..., -4.8323e-01,
-7.0773e-01, 8.1973e-01],
[-9.0365e-01, -9.3246e-01, -1.1067e+00, ..., -8.6843e-01,
-4.9027e-01, 1.0332e+00],
[-1.9765e+00, -7.6227e-01, -2.6976e-01, ..., 5.9606e-01,
2.0992e-01, 3.0507e-01],
...,
[-1.8078e+00, -1.3841e+00, -9.6124e-01, ..., -1.6638e-01,
1.4525e+00, 1.4057e+00],
[-1.9028e+00, -1.1080e+00, -1.0084e+00, ..., -8.5165e-01,
-2.2382e+00, -1.7539e+00],
[-4.8096e-01, 2.4237e-01, -4.1455e-01, ..., -3.6714e-01,
-6.9628e-01, 5.5974e-01]]],
[[[ 5.8698e-01, 8.6602e-01, 5.3966e-01, ..., 3.2733e-01,
4.1436e-01, 1.7766e-02],
[-2.7117e-01, -8.2527e-01, -3.7519e-01, ..., 1.7918e-01,
1.2698e-01, -2.7305e-01],
[-2.1955e-01, -1.4830e-01, 2.2683e-01, ..., -1.4273e-01,
-3.0563e-02, 6.5374e-01],
...,
[ 4.6009e-01, -8.8434e-02, -3.4036e-01, ..., 4.0928e-01,
8.2435e-01, 7.5621e-01],
[ 1.9564e-02, -4.2324e-01, 7.0018e-02, ..., 1.0348e+00,
1.6369e+00, 9.7946e-02],
[ 7.9434e-01, -3.7508e-02, -4.4322e-02, ..., 4.2249e-01,
3.3418e-01, 6.7022e-01]],
[[-2.3759e+00, -6.3424e-01, -1.8660e+00, ..., -1.6092e+00,
-2.2313e+00, -1.4903e+00],
[-2.5031e+00, -5.8820e-01, -1.5706e+00, ..., -1.2349e+00,
-2.3231e+00, -1.6352e+00],
[-2.5031e+00, -1.5483e+00, -2.1878e+00, ..., -2.0995e+00,
-2.3663e+00, -1.1288e+00],
...,
[-4.4806e-01, -5.8139e-01, -1.3466e+00, ..., -4.8556e-01,
4.0948e-02, 7.9341e-01],
[-1.9647e+00, -2.8813e-01, -4.7955e-01, ..., 3.3200e-01,
1.5154e-01, 8.3017e-01],
[-1.1687e+00, -1.1229e+00, -9.2492e-01, ..., -3.7952e-01,
-4.2833e-01, -1.6905e-01]],
[[ 1.9258e+00, 2.5424e+00, 1.6483e+00, ..., 1.5881e+00,
7.9943e-01, 2.0792e+00],
[-2.3481e-01, 4.1296e-01, 6.8604e-01, ..., 7.8823e-01,
3.7118e-01, 2.2072e+00],
[ 4.8753e-02, 2.9444e-01, 5.4963e-01, ..., -9.8606e-02,
7.7398e-01, 1.9780e+00],
...,
[ 4.6496e-01, 3.6297e-01, 9.7244e-01, ..., 3.0506e-01,
-4.7520e-02, 1.5396e+00],
[ 7.2722e-01, 1.9349e+00, 1.7105e+00, ..., 1.1420e+00,
7.6523e-01, 1.6892e+00],
[ 2.0745e+00, 1.9857e+00, 2.3802e+00, ..., 2.0262e+00,
1.2136e+00, 1.8976e+00]],
...,
[[-2.0000e+00, -4.9334e+00, -3.1955e+00, ..., -2.0550e+00,
-1.6649e+00, -2.9973e+00],
[-2.4724e+00, -5.3142e+00, -3.1331e+00, ..., -2.5331e+00,
-2.2889e+00, -3.0021e+00],
[-1.5267e+00, -5.3424e+00, -3.3193e+00, ..., -2.3880e+00,
-2.4353e+00, -2.2881e+00],
...,
[-2.1234e+00, -3.6332e+00, -2.2875e+00, ..., -2.5700e+00,
-1.8557e+00, -2.3619e+00],
[-2.0601e+00, -4.6583e+00, -3.6828e+00, ..., -4.2411e+00,
-2.9643e+00, -4.7054e+00],
[-2.1295e+00, -5.1705e+00, -2.6593e+00, ..., -2.8597e+00,
-1.3544e+00, -3.2073e+00]],
[[ 9.4902e-01, 1.6197e+00, 8.2974e-01, ..., 1.0968e+00,
1.5061e+00, 1.3979e+00],
[ 1.1820e+00, 1.0760e+00, 8.1631e-01, ..., 9.7374e-01,
8.2578e-01, 1.9092e+00],
[ 6.1410e-01, -8.9978e-02, -1.2298e-02, ..., 3.9865e-01,
1.2981e-01, 1.3815e+00],
...,
[ 1.3826e+00, 8.2881e-02, 3.7425e-01, ..., 1.5529e+00,
1.8124e+00, 1.4225e+00],
[ 1.9363e+00, 1.0251e+00, 1.6592e+00, ..., 2.0539e+00,
1.7840e+00, 2.4724e+00],
[ 1.2709e+00, 6.7279e-01, 1.2496e+00, ..., 1.3427e+00,
1.6256e+00, 1.5101e+00]],
[[-8.8983e-02, -6.2410e-01, -1.2049e+00, ..., -5.8697e-01,
-8.9932e-01, -1.3046e-01],
[-8.6714e-01, -2.4712e+00, -2.6407e+00, ..., -1.2002e+00,
-1.0451e+00, -6.9793e-01],
[-1.4938e+00, -1.8918e+00, -1.3936e+00, ..., -3.4511e-01,
-4.9614e-01, -2.8283e-01],
...,
[-1.5793e+00, -1.3083e+00, -8.6135e-01, ..., -4.9262e-01,
-4.4316e-01, -4.1325e-01],
[-1.5672e+00, -1.1158e+00, -7.4370e-01, ..., -3.6111e-01,
-5.1860e-01, 3.2954e-01],
[-3.0642e-02, 5.0269e-01, -1.9218e-01, ..., 1.9192e-02,
-4.9339e-02, 5.4819e-01]]]], device='cuda:0'),
tensor([[[[ 7.2636e-01, 1.1122e+00, 6.6131e-01, ..., 9.1952e-01,
3.0994e-01, 2.3203e-01],
[-3.2838e-01, -1.4461e+00, -6.2372e-01, ..., -5.9701e-01,
-6.0370e-01, -2.3147e+00],
[ 1.2633e-01, -8.6875e-01, -2.5264e-01, ..., -1.6716e+00,
-6.6257e-01, 2.1868e-01],
...,
[ 7.0921e-01, -7.2990e-01, -1.0760e+00, ..., -5.6582e-02,
7.1338e-01, 6.3815e-01],
[-4.0526e-01, -7.8937e-01, 7.8529e-01, ..., 3.0687e-01,
9.5653e-01, 2.5643e-01],
[ 2.9666e-01, -6.0093e-01, 1.6851e+00, ..., -1.0470e-01,
-3.0336e-01, 4.9777e-01]],
[[-3.0838e+00, -1.4952e+00, -3.1013e+00, ..., -3.1223e+00,
-3.5646e+00, -8.2880e-01],
[-4.3857e+00, -1.8200e+00, -2.3317e+00, ..., -5.5221e+00,
-4.3406e+00, -9.3643e-01],
[-4.3758e+00, -2.6609e+00, -3.8160e+00, ..., -6.4365e+00,
-1.2993e+00, 1.2926e+00],
...,
[-2.7800e+00, -2.1596e-01, -5.0124e+00, ..., -1.2266e+00,
-1.4706e+00, 4.2313e-01],
[-3.4736e+00, 6.1886e-01, -1.9910e+00, ..., -2.2906e-01,
-1.0386e+00, 2.2087e-01],
[-1.4156e+00, -9.8429e-01, -2.0495e+00, ..., -7.6319e-01,
-6.3305e-01, -6.9739e-02]],
[[ 1.7698e+00, 2.8866e+00, 2.4275e+00, ..., 2.2382e+00,
9.3526e-01, 2.5756e+00],
[ 1.9213e-01, 1.2760e+00, 1.6171e+00, ..., 1.6041e+00,
-2.0802e-01, 2.2825e+00],
[ 7.9562e-01, 1.4409e+00, 7.7241e-01, ..., 1.5146e+00,
-3.5501e+00, -6.6008e-01],
...,
[ 1.1492e+00, -2.0390e+00, -9.0074e+00, ..., -1.3013e-01,
3.4278e-02, 1.1488e+00],
[ 1.3415e+00, 2.3580e+00, -2.8849e+00, ..., 3.1175e-02,
3.4371e-01, 1.0987e+00],
[ 1.7060e+00, 2.6795e+00, 7.2499e-01, ..., 5.3426e-01,
5.9024e-01, 1.2057e+00]],
...,
[[-1.4081e+00, -4.6458e+00, -4.0175e+00, ..., -5.1614e+00,
-4.2323e+00, -4.6323e+00],
[-1.8508e+00, -4.0446e+00, -4.4515e+00, ..., -7.9176e+00,
-5.7368e+00, -4.7758e+00],
[-1.0641e+00, -3.3913e+00, -4.0208e+00, ..., -8.5218e+00,
-7.1110e+00, -6.9396e+00],
...,
[-2.6361e+00, -2.8293e+00, -2.7825e+00, ..., -4.2921e+00,
-2.4521e+00, -3.0474e+00],
[-1.3773e+00, -4.4375e+00, -3.5644e+00, ..., -6.9562e+00,
-3.5729e+00, -4.7198e+00],
[-1.6159e+00, -5.0340e+00, -3.6908e+00, ..., -5.1203e+00,
-1.9656e+00, -3.0268e+00]],
[[ 7.7633e-01, 5.2063e-01, -1.1033e-01, ..., 7.5278e-01,
5.8974e-01, 1.7327e+00],
[ 1.2257e+00, 9.5737e-01, 5.2326e-01, ..., 2.3203e-01,
-7.6764e-01, 1.3723e+00],
[ 5.9180e-01, 5.8473e-01, -3.0796e-01, ..., -1.7707e+00,
-1.8594e+00, 9.5486e-02],
...,
[-5.2514e-01, 1.4971e+00, 1.2135e+00, ..., 2.6483e+00,
1.5170e+00, 1.7709e+00],
[ 1.2742e+00, 5.0772e-01, 1.5006e+00, ..., 2.7270e+00,
1.4097e+00, 2.4766e+00],
[ 6.7904e-01, -2.4575e-02, 4.7328e-01, ..., 1.3115e+00,
1.1793e+00, 1.3183e+00]],
[[-2.9162e-01, -9.9589e-02, -9.4659e-01, ..., 3.0004e-01,
4.1574e-01, 9.7522e-01],
[-1.4141e+00, -1.3755e+00, -2.1554e+00, ..., -1.1387e+00,
5.0356e-02, -9.1478e-01],
[-1.7446e+00, -8.7764e-01, -1.4372e+00, ..., -2.0083e+00,
8.4905e-01, 1.2905e+00],
...,
[-7.8158e-01, -1.3127e+00, -1.2482e+00, ..., -1.6998e+00,
-1.0584e+00, -8.1407e-01],
[-1.0277e+00, -9.0733e-01, -9.7659e-01, ..., -1.4674e+00,
-9.0476e-01, 5.0687e-02],
[-2.3317e-01, -1.3115e-01, 1.2767e-02, ..., -1.3777e+00,
-7.6302e-01, 6.3139e-02]]],
[[[ 8.4513e-01, 1.5042e+00, 1.1862e+00, ..., 1.3832e+00,
1.0846e+00, 9.7020e-01],
[-1.0074e+00, -2.0510e+00, -1.1795e+00, ..., -5.4616e-01,
-6.3777e-01, -1.5659e+00],
[-1.2168e-01, -7.7372e-01, 2.8878e-02, ..., 3.1975e-01,
7.0174e-01, 7.7123e-01],
...,
[ 4.6896e-01, -2.1144e-01, 2.1730e-01, ..., 4.7613e-01,
2.2296e-01, 6.2631e-01],
[ 3.3820e-01, -6.4132e-01, -3.6839e-01, ..., 6.9818e-01,
6.8637e-01, -1.9843e-02],
[ 6.0090e-01, 2.4849e-01, -1.8700e-01, ..., 3.0844e-01,
-8.0668e-02, 6.2971e-01]],
[[-3.7527e+00, -7.1428e-01, -1.5999e+00, ..., -1.4060e+00,
-2.4798e+00, -1.2257e+00],
[-3.9924e+00, -2.7276e+00, -1.8712e+00, ..., -1.9691e+00,
-3.1221e+00, -8.9520e-01],
[-2.5130e+00, -1.6125e+00, -1.6170e+00, ..., -2.1040e+00,
-2.8306e+00, -1.0153e+00],
...,
[-1.0533e+00, -1.2426e+00, -1.9264e+00, ..., -1.2937e+00,
-1.3261e+00, -1.7848e-01],
[-1.3195e+00, -3.8818e-01, -7.5225e-01, ..., -5.7128e-01,
-1.0499e+00, -5.3092e-02],
[ 5.0239e-01, -1.8066e-01, -1.4767e-01, ..., 2.7250e-01,
8.4804e-02, 4.2786e-01]],
[[ 2.9450e+00, 4.2491e+00, 3.3971e+00, ..., 2.8599e+00,
9.3539e-01, 2.9624e+00],
[ 3.3240e-01, 1.0878e+00, 2.3001e+00, ..., 1.6587e+00,
-7.4368e-01, 2.7333e+00],
[ 1.1273e+00, 1.6045e+00, 1.7552e+00, ..., 7.8333e-01,
5.9556e-01, 2.1904e+00],
...,
[ 4.4060e-01, 4.9737e-01, 9.7351e-01, ..., -3.6681e-01,
2.8285e-01, 1.4670e+00],
[ 1.9293e-01, 1.1442e+00, 1.1822e+00, ..., -3.0177e-01,
1.9337e-01, 1.1361e+00],
[ 1.3558e+00, 9.4165e-01, 1.4106e+00, ..., 7.4324e-01,
4.2723e-01, 1.1551e+00]],
...,
[[-1.9390e+00, -7.3155e+00, -5.4590e+00, ..., -3.9806e+00,
-3.1898e+00, -5.4254e+00],
[-2.5115e+00, -6.9421e+00, -4.2271e+00, ..., -2.8348e+00,
-2.3445e+00, -4.2537e+00],
[-1.1328e+00, -4.6937e+00, -3.7100e+00, ..., -2.1864e+00,
-2.6489e+00, -2.2655e+00],
...,
[-2.2240e+00, -5.6057e+00, -3.9354e+00, ..., -3.4774e+00,
-3.1276e+00, -3.3884e+00],
[-2.4316e+00, -6.6765e+00, -4.9541e+00, ..., -4.9037e+00,
-4.1388e+00, -4.8471e+00],
[-2.3676e+00, -6.5983e+00, -4.1843e+00, ..., -4.2840e+00,
-3.1866e+00, -4.0862e+00]],
[[-1.8196e-01, 3.1470e-01, 5.2623e-01, ..., 6.7785e-01,
8.1533e-01, 1.3786e+00],
[ 1.6662e+00, 1.1306e+00, 1.8087e+00, ..., 1.8869e+00,
2.4947e+00, 3.0000e+00],
[ 1.1783e+00, 2.7879e-01, 7.8435e-01, ..., 9.2550e-01,
1.8969e+00, 2.4651e+00],
...,
[ 1.0430e+00, -1.1533e-01, -1.4686e-01, ..., 9.8150e-01,
7.2348e-01, 1.2621e+00],
[ 1.6727e+00, 8.5359e-01, 6.5933e-01, ..., 1.1656e+00,
6.3677e-01, 2.0349e+00],
[ 1.1584e+00, 7.9740e-01, 1.2239e+00, ..., 1.1730e+00,
1.2325e+00, 1.3889e+00]],
[[-5.7114e-01, 3.0446e-01, 9.3631e-02, ..., 4.8033e-01,
4.7791e-01, 1.3364e+00],
[-1.3992e+00, -1.7979e+00, -1.6689e+00, ..., -7.9933e-01,
-5.3521e-01, 2.2495e-01],
[-2.2341e+00, -7.8887e-01, 2.9585e-01, ..., 6.3669e-01,
1.7217e-01, 7.4301e-01],
...,
[-1.5735e+00, -1.7720e+00, -1.2184e+00, ..., -5.4355e-01,
-9.9788e-01, -7.2217e-01],
[-1.2659e+00, -1.6254e+00, -1.1418e+00, ..., -7.5826e-01,
-1.1960e+00, -6.9287e-02],
[ 1.6601e-01, 7.1164e-02, 4.7654e-02, ..., 2.1201e-01,
-6.3648e-02, 7.5474e-01]]],
[[[ 5.8657e-01, 7.7753e-01, 5.0680e-01, ..., 2.9072e-01,
3.2020e-01, -9.0239e-02],
[-3.1115e-01, -5.9465e-01, -4.5981e-01, ..., 1.1978e-01,
6.3009e-02, -3.2088e-01],
[-6.3927e-01, -1.1182e+00, -1.3494e+00, ..., 1.1342e-01,
1.6451e-01, 7.2906e-01],
...,
[ 7.6005e-01, -1.3206e+00, -1.6025e+00, ..., -3.6933e+00,
-4.7595e+00, -2.1004e+00],
[ 4.3393e-01, -1.2626e+00, -3.8166e+00, ..., -3.3312e+00,
-9.1670e+00, -5.4900e+00],
[-8.2964e-01, 1.1871e+00, -4.0430e+00, ..., 4.4744e-01,
1.8379e+00, 1.0862e+00]],
[[-2.6955e+00, -8.2908e-01, -2.1491e+00, ..., -1.8045e+00,
-2.3453e+00, -1.4784e+00],
[-3.1687e+00, -1.9669e+00, -3.5672e+00, ..., -1.3995e+00,
-2.4674e+00, -1.6698e+00],
[-4.3970e+00, -5.7908e+00, -7.4781e+00, ..., -2.0103e+00,
-2.3214e+00, -1.1068e+00],
...,
[-3.5044e-01, 1.5189e-01, 2.4077e+00, ..., -1.1837e+01,
-7.9790e+00, -3.2828e+00],
[-2.3623e+00, -2.3089e+00, 5.6827e-01, ..., -1.1603e+01,
-1.1148e+01, -2.9935e+00],
[-9.2125e-01, -6.4862e-01, 9.9163e-01, ..., -8.8936e+00,
-8.8662e+00, -3.6097e+00]],
[[ 2.0559e+00, 2.5995e+00, 1.8721e+00, ..., 1.7450e+00,
8.1038e-01, 2.1504e+00],
[ 2.6340e-01, 6.0965e-01, 1.2811e+00, ..., 8.4465e-01,
4.0852e-01, 2.3043e+00],
[ 1.3774e+00, 1.5560e+00, 2.1668e+00, ..., 1.1565e-01,
7.5618e-01, 2.0489e+00],
...,
[ 5.1159e-01, -5.4758e-02, 1.2309e+00, ..., -5.5814e+00,
-8.4635e+00, -1.7863e+00],
[ 1.4116e+00, -5.5921e+00, -6.4367e+00, ..., -9.8354e+00,
-1.3422e+01, -1.0724e+00],
[ 3.3356e+00, -5.9226e+00, -7.1174e+00, ..., -2.3323e+00,
-1.0166e+01, -2.4529e+00]],
...,
[[-2.0795e+00, -5.0327e+00, -3.1340e+00, ..., -2.2207e+00,
-1.8212e+00, -3.1211e+00],
[-2.7028e+00, -5.6098e+00, -2.9107e+00, ..., -2.9060e+00,
-2.7168e+00, -3.3091e+00],
[-3.9322e-01, -3.5354e+00, -2.0097e+00, ..., -2.3232e+00,
-2.4355e+00, -2.2251e+00],
...,
[-1.2584e+00, -6.8989e+00, -6.7692e+00, ..., -9.1543e+00,
2.4067e+00, -1.4797e+00],
[-3.0243e+00, -6.0330e+00, -7.7356e+00, ..., -5.4158e+00,
2.9434e+00, -3.4905e+00],
[-2.2600e+00, -5.8206e+00, -4.9384e+00, ..., -9.5591e-01,
2.2165e-01, -1.0670e+01]],
[[ 7.3388e-01, 1.5097e+00, 7.8799e-01, ..., 1.0805e+00,
1.4947e+00, 1.5058e+00],
[ 8.1269e-01, 5.5133e-01, 6.3219e-01, ..., 9.7717e-01,
8.6512e-01, 2.0417e+00],
[-1.0692e+00, -4.0896e+00, -6.1506e+00, ..., 5.5894e-01,
1.7179e-01, 1.3521e+00],
...,
[ 1.7984e+00, -5.8445e-01, 2.3370e+00, ..., -3.3592e+00,
-3.7571e+00, -1.9270e+00],
[ 1.0782e+00, 8.4159e-01, -1.4379e+00, ..., -3.5420e+00,
-4.2654e+00, -1.7409e+00],
[ 7.5502e-01, 2.3712e+00, 2.6123e-01, ..., -7.0434e-01,
-3.2224e+00, -1.9970e+00]],
[[-1.4556e-01, -6.9077e-01, -1.2199e+00, ..., -6.1002e-01,
-8.8974e-01, -6.8248e-02],
[-8.6459e-01, -2.0982e+00, -2.0206e+00, ..., -1.0949e+00,
-8.3146e-01, -5.1792e-01],
[-1.7362e+00, -1.9677e+00, -1.2602e+00, ..., -5.7541e-03,
-1.8739e-01, -1.4908e-01],
...,
[-3.1647e+00, -6.5018e-01, 1.4725e+00, ..., -6.7049e+00,
-8.1502e-01, 3.5258e+00],
[-2.2594e+00, -2.1525e+00, 2.2665e+00, ..., -8.5559e+00,
-1.7194e-01, 1.5545e+00],
[-2.2796e-01, -1.7232e+00, -1.9935e+00, ..., 2.4822e+00,
4.5472e+00, -5.0726e-01]]],
[[[ 1.0841e+00, 9.2148e-01, -3.4836e-01, ..., 9.4917e-01,
-7.6466e-03, 1.7349e-01],
[-1.3589e+00, -1.5798e+00, -1.9584e-01, ..., 1.3717e+00,
5.4407e-01, -5.4402e-01],
[-2.4666e+00, -3.4160e+00, -1.3521e+00, ..., -2.0870e+00,
-1.4162e-01, 5.1931e-01],
...,
[ 1.1074e+00, 1.7126e+00, 1.2041e+00, ..., -4.1159e-02,
1.8226e-01, 5.2637e-01],
[ 2.1653e-01, -2.1604e-01, 3.2688e-01, ..., -2.3547e-02,
3.3404e-01, -6.9586e-02],
[ 4.9907e-01, -5.4455e-02, -7.0722e-01, ..., -3.2442e-01,
-7.6479e-01, 2.1130e-01]],
[[-5.5043e+00, -3.9631e+00, -3.6411e+00, ..., -3.2365e+00,
-2.9192e+00, -1.7498e+00],
[-5.4700e+00, -5.8809e+00, -3.9786e+00, ..., -7.4739e-01,
-3.7784e+00, -2.4459e+00],
[-8.6609e-01, -2.3976e+00, -2.0428e+00, ..., -2.5769e+00,
-4.9437e+00, -1.9458e+00],
...,
[ 1.1762e+00, -2.5279e-01, 4.2368e-01, ..., -7.7271e-01,
-4.5017e-01, -4.4713e-01],
[-1.8356e+00, -2.3908e+00, -7.4200e-01, ..., -5.8783e-01,
-8.4765e-01, -9.6113e-01],
[ 1.2229e-01, -8.8416e-01, -9.2492e-01, ..., -6.5851e-01,
-6.7084e-01, -5.4622e-01]],
[[ 3.0826e+00, 4.2962e+00, 4.3668e+00, ..., 8.6308e-01,
1.2763e+00, 2.6191e+00],
[ 8.5433e-01, 1.6999e+00, 1.4434e+00, ..., -1.7491e+00,
-3.5570e-01, 2.0004e+00],
[-8.5553e-01, -9.0266e-01, -1.4073e+00, ..., -3.9326e+00,
-1.5570e-01, 1.1685e+00],
...,
[-2.1897e-01, -1.3630e+00, -9.7712e-01, ..., -1.1233e+00,
3.4250e-01, 1.3011e+00],
[-4.7636e-01, 4.1237e-01, 3.7754e-01, ..., -7.0525e-01,
7.2926e-01, 1.1894e+00],
[ 1.7198e+00, 1.5335e+00, 1.9196e+00, ..., 4.4856e-01,
1.5321e+00, 1.5760e+00]],
...,
[[-2.8185e+00, -8.7260e+00, -5.9954e+00, ..., -2.6960e+00,
-2.3830e+00, -3.7335e+00],
[-2.6067e+00, -9.1951e+00, -6.9460e+00, ..., -3.5263e+00,
-7.4237e+00, -4.4338e+00],
[-1.3983e+00, -5.5464e+00, -4.8087e+00, ..., -8.3467e-01,
-7.7660e+00, -3.6868e+00],
...,
[-2.0583e+00, -3.5401e+00, -2.5049e+00, ..., -3.3362e+00,
-2.2066e+00, -2.2301e+00],
[-2.7896e+00, -6.4240e+00, -4.0642e+00, ..., -5.0473e+00,
-3.7848e+00, -4.0466e+00],
[-2.3492e+00, -6.5742e+00, -3.8293e+00, ..., -3.4557e+00,
-2.2229e+00, -2.3769e+00]],
[[-1.4682e+00, -8.4063e-01, 2.7966e-01, ..., -1.2726e-01,
1.8681e+00, 1.7541e+00],
[-4.8592e-01, -3.2958e+00, -2.5210e+00, ..., -5.4151e-01,
6.1066e-01, 2.1392e+00],
[ 2.1098e+00, 2.7731e+00, -3.2709e-01, ..., -8.8246e-01,
-8.0163e-01, 2.1600e+00],
...,
[ 6.9677e-01, -5.2383e-01, -1.5502e-01, ..., 1.4828e+00,
7.2225e-01, 9.6921e-01],
[ 1.0652e+00, 9.8374e-02, 6.5472e-01, ..., 1.3616e+00,
5.7035e-01, 1.5527e+00],
[ 1.1005e+00, 9.3663e-01, 1.3610e+00, ..., 7.2189e-01,
6.3535e-01, 1.4801e+00]],
[[-7.7542e-01, -4.6729e-01, -4.9395e-01, ..., -7.0361e-02,
-1.4867e+00, 2.1984e-01],
[-1.5635e+00, -1.5551e+00, -5.1647e-01, ..., 2.3970e+00,
-1.2503e+00, -1.2855e+00],
[-5.9157e-01, 2.4610e-01, 3.8155e-01, ..., 1.2901e+00,
-2.8896e+00, -1.4815e+00],
...,
[-2.8516e-01, -4.5255e-01, -3.7463e-01, ..., -7.7640e-01,
-1.4884e+00, -7.7830e-01],
[-1.2454e+00, -2.5783e+00, -1.3519e+00, ..., -1.1717e+00,
-1.5972e+00, -7.4524e-01],
[-4.5035e-02, 2.1999e-01, -1.0300e-01, ..., -5.0889e-01,
-7.4919e-01, 1.0712e-01]]],
[[[ 1.3263e-02, 5.5298e-01, 6.9953e-01, ..., 2.7460e-01,
4.0031e-01, 2.4262e-01],
[-4.4825e+00, -5.2806e+00, -3.6452e+00, ..., -6.2798e-02,
1.7625e-01, -1.1898e+00],
[-7.0114e+00, -5.3063e+00, -3.5306e+00, ..., -8.4223e-01,
6.7484e-01, 7.1203e-01],
...,
[ 3.1660e-01, 2.9784e-01, 3.7879e-01, ..., 1.7206e-01,
8.5452e-01, 4.7134e-01],
[-4.6999e-01, -8.2982e-01, -2.0280e-01, ..., 2.2458e-01,
1.4703e+00, -1.1957e-01],
[-2.4628e-01, -2.4317e-01, -8.6484e-01, ..., -1.6818e-01,
4.3972e-01, 6.1389e-01]],
[[-8.9773e-02, 2.0779e-01, -2.0063e+00, ..., -1.7015e+00,
-2.1055e+00, -1.0517e+00],
[-1.0350e+00, -7.1706e+00, -5.1425e+00, ..., -3.8539e-01,
-1.4299e+00, -3.2600e-01],
[-3.9234e+00, -6.9247e+00, -6.0298e+00, ..., -1.1690e+00,
-1.4983e+00, -8.0608e-01],
...,
[ 1.7467e-01, -9.4792e-01, -7.7699e-01, ..., 5.4324e-01,
-9.1549e-02, 4.3813e-01],
[-9.8436e-01, -8.9599e-01, -5.3161e-01, ..., 2.2705e-01,
-4.1712e-01, 2.2968e-01],
[-3.6819e-01, -1.1258e+00, -9.7494e-01, ..., -2.3119e-01,
-7.5139e-01, -6.1004e-01]],
[[-1.4166e+00, -5.5142e-01, -1.7369e+00, ..., 2.8397e+00,
1.3152e+00, 2.9320e+00],
[-5.5631e+00, -4.8279e+00, -3.8345e+00, ..., 2.0740e+00,
-1.0016e-01, 2.5323e+00],
[-6.1475e+00, -3.2549e+00, -4.6455e+00, ..., 2.3165e+00,
9.8479e-01, 1.9326e+00],
...,
[-1.2079e-01, 4.6635e-01, 6.8091e-01, ..., 6.6705e-01,
1.7031e-01, 1.4841e+00],
[-8.4128e-01, 5.8883e-01, 4.7085e-01, ..., 8.7107e-01,
7.1610e-01, 1.5955e+00],
[ 7.1671e-01, 4.2599e-01, 9.9407e-01, ..., 1.5427e+00,
1.1485e+00, 2.0441e+00]],
...,
[[-4.5022e+00, -6.9515e+00, -2.5992e+00, ..., -3.5250e+00,
-2.3215e+00, -3.9188e+00],
[-7.5915e+00, -1.1332e+01, 7.7548e-01, ..., -5.7025e+00,
-3.7336e+00, -4.2507e+00],
[-6.3217e+00, -5.4840e+00, 1.4825e+00, ..., -3.5619e+00,
-2.5496e+00, -1.7035e+00],
...,
[-2.8657e+00, -8.0166e+00, -5.5310e+00, ..., -6.9807e+00,
-2.5344e+00, -1.4888e+00],
[-3.8457e+00, -9.2926e+00, -5.3298e+00, ..., -6.8864e+00,
-3.3387e+00, -3.7798e+00],
[-2.9208e+00, -7.4372e+00, -3.1335e+00, ..., -3.8450e+00,
-2.2075e+00, -2.7847e+00]],
[[ 1.5603e+00, 2.0115e+00, -1.8890e+00, ..., 1.3213e+00,
1.0450e+00, 1.2480e+00],
[ 1.6461e+00, 9.9473e-01, -1.5533e+00, ..., 2.2059e+00,
2.2988e+00, 2.5593e+00],
[-6.8211e-01, -8.3116e-01, -1.9223e+00, ..., 2.0331e+00,
2.1548e+00, 1.9499e+00],
...,
[ 9.9660e-01, 1.0694e-01, 1.5333e+00, ..., 2.5273e+00,
1.8714e+00, 1.6982e+00],
[ 1.7879e+00, 1.4766e+00, 1.6047e+00, ..., 1.7488e+00,
1.5729e+00, 2.4825e+00],
[ 8.1687e-01, 6.1138e-01, 1.1561e+00, ..., 1.2404e+00,
1.3351e+00, 1.3758e+00]],
[[ 9.4566e-02, 3.1923e-01, 1.5954e-01, ..., -1.6367e-01,
-1.1311e-01, 8.6680e-01],
[ 8.2287e-01, -6.7303e+00, -3.6173e+00, ..., -1.3233e-01,
-2.3992e-01, 2.9965e-01],
[-1.5587e+00, -1.0836e+01, -3.2700e+00, ..., -2.3672e-01,
2.0026e-01, 2.6964e-01],
...,
[-1.8049e+00, -2.4372e+00, -1.6696e-01, ..., -4.8054e-01,
-2.8556e-01, -3.5468e-01],
[-1.4537e+00, -2.3705e+00, -5.2596e-01, ..., -4.1505e-01,
-4.4306e-01, 3.3978e-01],
[-3.5373e-01, -1.1451e+00, -5.8203e-01, ..., 9.8687e-02,
1.4819e-01, 3.2043e-01]]]], device='cuda:0'),
tensor([[[[ 9.4637e-01, 1.5405e+00, 1.1546e+00, ..., 7.9108e-01,
1.0073e+00, 4.6616e-01],
[-8.9841e-01, -2.1128e+00, -9.8015e-01, ..., -3.2345e-01,
-4.4342e-01, -6.5313e-01],
[ 3.5028e-02, -3.7550e-01, 4.6756e-01, ..., -1.0927e+00,
-1.1985e+00, -6.5260e-01],
...,
[ 3.2454e-01, -2.2408e-01, -4.6247e-03, ..., -4.1366e-02,
-8.8584e-02, 3.5901e-01],
[-1.5242e-01, -9.9981e-01, -7.4699e-02, ..., 1.7227e-02,
9.1053e-02, -1.0049e-01],
[ 7.2157e-01, -2.6587e-01, -1.8506e-01, ..., -3.6312e-01,
-9.0628e-01, 1.6696e-01]],
[[-4.2966e+00, -1.2403e+00, -2.1678e+00, ..., -1.4274e+00,
-2.0792e+00, -1.8351e+00],
[-4.2084e+00, -2.7465e+00, -2.0220e+00, ..., -1.5706e+00,
-2.1671e+00, -2.1730e+00],
[-3.1206e+00, -2.4230e+00, -2.8588e+00, ..., -4.2350e+00,
-3.6609e+00, -1.9330e+00],
...,
[-9.2384e-01, -4.5850e-01, -9.7307e-01, ..., -6.3893e-01,
-1.0366e+00, -9.0045e-01],
[-2.8725e+00, -3.5068e-01, -7.6793e-01, ..., -6.7221e-01,
-1.1361e+00, -1.2199e+00],
[-2.0373e+00, -1.4991e+00, -1.4705e+00, ..., -6.8354e-01,
-7.2274e-01, -5.1392e-01]],
[[ 3.1041e+00, 4.0943e+00, 3.1951e+00, ..., 1.7027e+00,
8.2407e-01, 2.5732e+00],
[ 4.6121e-01, 7.4704e-01, 2.1981e+00, ..., 2.0759e+00,
1.1336e+00, 2.8934e+00],
[ 9.9473e-01, 1.4998e+00, 1.8597e+00, ..., 1.9485e+00,
1.8383e+00, 2.1577e+00],
...,
[ 1.5212e-01, 9.7581e-01, 1.2579e+00, ..., -2.0578e-01,
6.3499e-01, 1.4956e+00],
[ 2.8700e-01, 2.0541e+00, 1.3956e+00, ..., -1.4982e-01,
8.5524e-01, 1.1644e+00],
[ 1.4161e+00, 1.4076e+00, 1.6751e+00, ..., 5.7368e-01,
1.3491e+00, 1.3529e+00]],
...,
[[-2.1877e+00, -7.4688e+00, -6.0832e+00, ..., -5.0856e+00,
-3.6070e+00, -4.3448e+00],
[-2.8719e+00, -6.4973e+00, -4.5778e+00, ..., -6.6045e+00,
-5.0589e+00, -3.8270e+00],
[-5.8147e-01, -3.4470e+00, -3.8009e+00, ..., -4.2309e+00,
-2.9273e+00, -1.8550e+00],
...,
[-9.0159e-01, -2.1518e+00, -1.6296e+00, ..., -4.0660e+00,
-3.1557e+00, -2.6924e+00],
[-1.2698e+00, -3.6595e+00, -3.3355e+00, ..., -5.2540e+00,
-4.2022e+00, -4.2418e+00],
[-2.1697e+00, -4.9433e+00, -2.9604e+00, ..., -3.2871e+00,
-2.4932e+00, -2.3890e+00]],
[[-7.2627e-01, -8.4574e-02, 1.6515e-01, ..., 1.4451e+00,
1.6971e+00, 1.1975e+00],
[ 1.7858e+00, 1.1051e+00, 1.6130e+00, ..., 1.4305e+00,
1.1621e+00, 1.6572e+00],
[ 1.1688e+00, 6.6495e-02, 8.3351e-01, ..., -8.4908e-01,
-1.3137e+00, 8.6040e-01],
...,
[ 1.2994e+00, 2.3874e-02, 1.1582e+00, ..., 1.3971e+00,
6.7929e-01, 9.5274e-01],
[ 1.8754e+00, 9.4898e-01, 1.7683e+00, ..., 1.1850e+00,
7.5160e-01, 1.4764e+00],
[ 1.0224e+00, 3.1890e-01, 9.3221e-01, ..., 7.1538e-01,
7.0071e-01, 1.4382e+00]],
[[-5.5927e-01, 5.8637e-01, 1.3430e-01, ..., -3.4426e-01,
-6.1144e-01, -2.9942e-02],
[-1.1804e+00, -1.1284e+00, -1.2325e+00, ..., -1.1223e+00,
-1.7712e+00, -1.7403e+00],
[-2.1020e+00, -4.2740e-01, 3.7412e-01, ..., -1.0535e+00,
-1.6148e+00, -9.2282e-01],
...,
[-1.7805e+00, -1.4469e+00, -8.8328e-01, ..., -9.3866e-01,
-1.6127e+00, -8.4512e-01],
[-1.7222e+00, -1.0434e+00, -8.2314e-01, ..., -8.0689e-01,
-1.7237e+00, -7.0256e-01],
[-3.3635e-01, 3.6696e-01, -3.5044e-01, ..., -2.3598e-01,
-8.1514e-01, -9.2529e-03]]],
[[[ 6.7135e-01, 1.0602e+00, 7.5526e-01, ..., 3.9342e-01,
3.4927e-01, 2.4828e-01],
[-3.7013e-01, -1.5787e+00, -7.0914e-01, ..., 3.8192e-01,
2.9862e-01, -1.1910e+00],
[ 1.9583e-03, -6.6363e-01, 1.7784e-02, ..., -4.7965e-01,
1.0060e+00, 6.7588e-01],
...,
[ 1.8748e-01, -9.0973e-01, -9.0934e-02, ..., 1.8584e-01,
9.6430e-01, 5.6634e-01],
[-4.5939e-01, -1.6616e+00, -7.8939e-01, ..., -7.5159e-02,
1.3467e+00, -1.7041e-01],
[ 2.8821e-01, -5.4414e-01, -8.2976e-01, ..., -2.9576e-01,
3.8967e-01, 5.7000e-01]],
[[-3.4495e+00, -1.3524e+00, -3.2362e+00, ..., -1.6613e+00,
-2.2181e+00, -1.1235e+00],
[-4.5878e+00, -1.5558e+00, -2.4673e+00, ..., -3.7558e-01,
-1.3219e+00, -4.0153e-01],
[-4.6534e+00, -2.2518e+00, -2.5239e+00, ..., -1.4658e+00,
-1.2545e+00, -7.2379e-01],
...,
[-2.3692e+00, -2.0323e+00, -2.4088e+00, ..., -2.0502e-02,
2.4175e-01, 4.7257e-01],
[-3.3708e+00, -1.5273e+00, -2.2903e+00, ..., -3.1134e-02,
-3.8946e-01, 2.3712e-01],
[-1.1130e+00, -9.4475e-01, -1.7075e+00, ..., -3.2308e-01,
-6.9281e-01, -5.6465e-01]],
[[ 2.2536e+00, 3.1061e+00, 2.5502e+00, ..., 2.7844e+00,
1.3691e+00, 2.9192e+00],
[ 4.0692e-01, 1.1846e+00, 1.7309e+00, ..., 1.7580e+00,
-1.2541e-01, 2.4306e+00],
[ 9.1940e-01, 1.7912e+00, 1.2058e+00, ..., 9.1942e-01,
5.2758e-01, 1.6553e+00],
...,
[ 9.7773e-01, 1.6661e+00, 1.2710e+00, ..., 1.2140e+00,
-2.0001e-01, 1.3756e+00],
[ 5.2585e-01, 1.6378e+00, 7.6441e-01, ..., 7.1795e-01,
6.1823e-01, 1.4826e+00],
[ 1.4021e+00, 1.4853e+00, 1.4543e+00, ..., 1.4612e+00,
1.1489e+00, 2.0110e+00]],
...,
[[-1.9590e+00, -5.2332e+00, -4.3588e+00, ..., -3.2249e+00,
-2.2861e+00, -4.0139e+00],
[-2.3440e+00, -4.8419e+00, -4.5993e+00, ..., -5.3667e+00,
-3.6901e+00, -4.2800e+00],
[-9.3772e-01, -3.5288e+00, -3.0830e+00, ..., -2.6982e+00,
-2.1821e+00, -1.3747e+00],
...,
[-1.0950e+00, -3.4726e+00, -3.8621e+00, ..., -4.3565e+00,
-3.8996e+00, -1.6266e+00],
[-1.2570e+00, -4.1238e+00, -5.1199e+00, ..., -6.1920e+00,
-3.4572e+00, -3.6482e+00],
[-1.2835e+00, -4.3475e+00, -3.9065e+00, ..., -3.4295e+00,
-2.0482e+00, -2.7405e+00]],
[[ 6.9037e-01, 7.5843e-01, -1.4016e-01, ..., 9.3867e-01,
1.0726e+00, 1.2082e+00],
[ 1.3411e+00, 1.1630e+00, 6.0119e-01, ..., 1.6128e+00,
2.3361e+00, 2.5007e+00],
[ 6.4970e-01, 7.5510e-01, 2.0926e-01, ..., 9.9383e-01,
1.6067e+00, 1.7890e+00],
...,
[ 9.0778e-01, -1.1569e-01, 1.9569e-01, ..., 1.7893e+00,
1.8001e+00, 1.6456e+00],
[ 1.4588e+00, 1.2852e+00, 1.1488e+00, ..., 1.4939e+00,
1.3617e+00, 2.4413e+00],
[ 7.0951e-01, 8.0586e-01, 1.0227e+00, ..., 1.0659e+00,
1.3336e+00, 1.3809e+00]],
[[-2.6441e-01, -2.5119e-01, -1.0327e+00, ..., -1.8865e-01,
-1.2421e-01, 8.6811e-01],
[-1.3312e+00, -1.5977e+00, -2.3031e+00, ..., 1.4642e-01,
-3.2473e-02, 3.3690e-01],
[-1.3816e+00, -3.5549e-01, -1.0320e+00, ..., 1.5287e-01,
1.3948e-01, 2.1216e-01],
...,
[-1.5552e+00, -2.0404e+00, -1.4103e+00, ..., -1.4172e-01,
2.0938e-02, -2.5988e-01],
[-1.3434e+00, -1.5168e+00, -1.5621e+00, ..., -7.3167e-02,
-4.6182e-01, 3.0227e-01],
[-1.5227e-01, -9.4824e-02, -5.1811e-01, ..., 1.7029e-01,
1.1505e-01, 3.2043e-01]]],
[[[ 8.2557e-01, 1.0167e+00, 5.0485e-01, ..., -2.9452e-01,
3.2622e-01, 2.0509e-01],
[-1.3997e+00, -2.2244e+00, -7.3678e-01, ..., 2.0658e+00,
-2.4539e-01, -6.1151e-01],
[-1.5632e+00, -3.7665e+00, -7.1272e-01, ..., 9.8548e-01,
-2.2739e-01, 3.7443e-01],
...,
[-1.0927e+00, 1.0360e-01, -1.0859e+00, ..., 6.2468e-01,
-2.2506e-01, 3.8881e-01],
[-4.5256e+00, -3.2228e+00, 4.5244e-02, ..., -2.4380e-02,
-1.0020e-02, -1.7891e-01],
[ 4.3773e-01, -1.6101e+00, -1.7660e+00, ..., -4.1926e-01,
-9.7416e-01, 1.0142e-01]],
[[-3.7903e+00, -1.7431e+00, -2.5738e+00, ..., -3.4329e+00,
-2.7417e+00, -1.7740e+00],
[-3.5458e+00, -2.6849e+00, -1.3850e+00, ..., -1.5455e+00,
-2.6167e+00, -2.2978e+00],
[-4.7784e+00, -5.7488e+00, -4.9336e+00, ..., -3.1954e+00,
-2.5248e+00, -1.8955e+00],
...,
[-1.6524e+00, -3.9127e+00, -6.5175e+00, ..., -2.8348e-01,
-5.2298e-01, -6.5819e-01],
[-7.2792e+00, -8.1679e+00, -7.6764e+00, ..., -8.1780e-01,
-1.0252e+00, -1.1048e+00],
[-3.5871e+00, -3.6290e+00, -4.7305e+00, ..., -7.3638e-01,
-6.6259e-01, -4.5845e-01]],
[[ 2.6924e+00, 3.3092e+00, 3.1264e+00, ..., 2.7223e+00,
1.1820e+00, 2.3070e+00],
[ 7.7334e-01, 2.3709e+00, 2.9709e+00, ..., -1.1802e-01,
5.4251e-01, 2.0816e+00],
[ 8.4675e-01, 1.3474e+00, 8.0813e-01, ..., -3.4144e+00,
7.7624e-01, 1.2259e+00],
...,
[-3.8473e+00, -2.9792e+00, -5.8279e+00, ..., -2.6580e-01,
6.3055e-01, 1.4458e+00],
[-9.8642e+00, -6.3480e+00, -7.9334e+00, ..., -5.1102e-01,
1.0188e+00, 1.3837e+00],
[ 1.2795e+00, -7.5769e-01, -1.4488e+00, ..., 3.7797e-01,
1.4011e+00, 1.4514e+00]],
...,
[[-2.6964e+00, -7.8996e+00, -4.6316e+00, ..., -2.9085e+00,
-2.3914e+00, -4.7643e+00],
[-2.5900e+00, -9.1054e+00, -7.3274e+00, ..., -4.9579e+00,
-3.4953e+00, -4.9344e+00],
[-1.3957e+00, -5.1652e+00, -4.1156e+00, ..., -4.5643e+00,
-6.6699e+00, -4.6615e+00],
...,
[ 4.6884e-01, -7.8088e+00, -6.0099e+00, ..., -6.6948e+00,
-5.0095e+00, -4.0677e+00],
[-3.4284e+00, -1.0490e+01, -9.8942e+00, ..., -6.2947e+00,
-4.7336e+00, -4.4562e+00],
[-3.1058e+00, -6.8764e+00, -5.1963e+00, ..., -3.8898e+00,
-2.8266e+00, -2.6075e+00]],
[[-6.5373e-01, 1.0286e-02, 5.2639e-01, ..., 1.1925e+00,
1.9709e+00, 1.6013e+00],
[-2.0582e-02, -9.4436e-01, 5.2353e-01, ..., -3.7012e-01,
1.3474e+00, 2.0345e+00],
[-2.3228e+00, -6.1822e+00, -5.4561e+00, ..., -3.2443e+00,
8.9533e-01, 1.5649e+00],
...,
[-4.6359e-01, -4.3935e+00, -4.4897e+00, ..., 2.3082e+00,
1.1117e+00, 1.2149e+00],
[ 1.6680e+00, -6.6166e-01, -4.4700e-01, ..., 1.7151e+00,
7.1766e-01, 1.7253e+00],
[-1.2257e+00, -1.2856e+00, -1.5653e+00, ..., 8.6756e-01,
6.8608e-01, 1.5309e+00]],
[[-7.3007e-01, -6.7199e-01, -4.8410e-01, ..., -1.1992e+00,
-9.0722e-01, 1.4341e-01],
[-1.7337e+00, -1.4944e+00, -1.9065e-01, ..., -1.0689e-01,
-2.1629e+00, -1.4229e+00],
[-2.3677e+00, -1.8105e+00, 4.7484e-01, ..., 4.5322e-01,
-2.5815e+00, -1.3343e+00],
...,
[-2.9063e-01, 3.2625e-01, -3.6469e-01, ..., -5.7885e-01,
-8.5123e-01, -6.8060e-01],
[-2.2746e+00, -3.7701e+00, -2.4942e+00, ..., -7.1371e-01,
-1.5054e+00, -5.9016e-01],
[-8.9389e-01, -6.4514e-01, -1.4111e+00, ..., -3.5438e-01,
-7.6896e-01, 6.0562e-02]]],
[[[ 1.1104e+00, 1.4922e+00, 1.1953e+00, ..., 2.6714e-01,
3.3407e-01, 1.9547e-01],
[-9.6322e-01, -1.4990e+00, -9.5171e-01, ..., 4.1408e-01,
3.6179e-01, -1.1330e+00],
[-6.5273e-01, -5.5961e-01, -6.5387e-01, ..., -4.4884e-01,
8.3719e-01, 6.1654e-01],
...,
[-1.1094e-01, -1.6651e-01, 1.6350e-01, ..., 3.0684e-01,
7.4538e-01, 4.7068e-01],
[-4.1285e-01, -7.8741e-01, -2.1138e-01, ..., 2.6569e-01,
1.4075e+00, -1.1309e-01],
[-1.2231e-01, -1.7862e-01, -7.7701e-01, ..., -2.1055e-01,
3.3807e-01, 5.7994e-01]],
[[-3.1817e+00, -1.9974e+00, -2.1131e+00, ..., -1.8418e+00,
-2.3206e+00, -1.1528e+00],
[-2.8939e+00, -3.1475e+00, -1.4558e+00, ..., -3.7569e-01,
-1.3417e+00, -3.5436e-01],
[-1.8915e+00, -2.4557e+00, -2.0194e+00, ..., -1.1608e+00,
-1.6555e+00, -7.8025e-01],
...,
[-2.8214e-02, -6.9896e-01, -2.6807e-01, ..., 5.5990e-01,
3.6195e-02, 4.6812e-01],
[-1.3041e+00, -7.9275e-01, -3.4579e-01, ..., 1.6000e-01,
-4.4088e-01, 2.3000e-01],
[-4.8037e-01, -9.9500e-01, -7.8967e-01, ..., -3.1948e-01,
-7.1703e-01, -5.4889e-01]],
[[ 2.3328e+00, 2.3254e+00, 1.7780e+00, ..., 2.9320e+00,
1.4516e+00, 2.9726e+00],
[ 4.4723e-02, 1.6885e-02, 1.2206e+00, ..., 1.9141e+00,
8.6231e-02, 2.5985e+00],
[-6.4315e-01, -3.3989e-01, 9.2543e-01, ..., 9.0567e-01,
5.2165e-01, 1.9304e+00],
...,
[-6.2884e-01, -1.1628e-02, 6.5238e-01, ..., 7.5383e-01,
1.3613e-01, 1.4617e+00],
[-9.8613e-01, 5.7876e-01, 6.9064e-01, ..., 1.0805e+00,
6.6091e-01, 1.5490e+00],
[ 6.5951e-01, 5.2102e-01, 1.1071e+00, ..., 1.4707e+00,
1.0859e+00, 2.0338e+00]],
...,
[[-2.9636e+00, -8.5664e+00, -6.7757e+00, ..., -3.6282e+00,
-2.4461e+00, -3.9375e+00],
[-3.7656e+00, -1.0158e+01, -6.3327e+00, ..., -5.8780e+00,
-3.8639e+00, -4.3831e+00],
[-1.7176e+00, -6.5412e+00, -4.0419e+00, ..., -3.4927e+00,
-2.3350e+00, -1.6329e+00],
...,
[-3.4823e+00, -8.6717e+00, -5.3333e+00, ..., -5.6959e+00,
-3.0108e+00, -1.5231e+00],
[-4.1205e+00, -9.4252e+00, -5.2950e+00, ..., -6.0105e+00,
-3.3632e+00, -3.6929e+00],
[-3.2952e+00, -7.8460e+00, -3.0192e+00, ..., -3.8313e+00,
-2.1294e+00, -2.7027e+00]],
[[-4.1185e-01, 7.2540e-01, 9.4101e-01, ..., 1.3125e+00,
1.0674e+00, 1.2636e+00],
[ 7.8116e-01, 8.1929e-01, 1.4264e+00, ..., 1.9612e+00,
2.3560e+00, 2.5913e+00],
[ 4.4716e-01, 1.1201e-01, 1.3720e+00, ..., 1.3660e+00,
1.5674e+00, 1.9782e+00],
...,
[ 1.1798e+00, 3.5164e-01, 1.1318e+00, ..., 2.0771e+00,
1.7517e+00, 1.6901e+00],
[ 1.6928e+00, 1.5660e+00, 1.6317e+00, ..., 1.6296e+00,
1.5158e+00, 2.4300e+00],
[ 7.2197e-01, 6.0390e-01, 1.2669e+00, ..., 1.1948e+00,
1.3705e+00, 1.3803e+00]],
[[-1.9834e-01, -3.8184e-01, -3.9059e-01, ..., -1.6215e-01,
-1.5098e-01, 8.8103e-01],
[-4.9119e-01, -2.5543e+00, -1.2110e+00, ..., 2.6109e-01,
-7.3717e-02, 4.1087e-01],
[-1.4463e+00, -1.4687e+00, -1.0462e+00, ..., 1.9722e-01,
3.7685e-02, 2.8436e-01],
...,
[-1.7283e+00, -2.4371e+00, -1.2236e-01, ..., 3.5780e-02,
-1.7491e-01, -3.3924e-01],
[-1.3425e+00, -2.5121e+00, -6.3427e-01, ..., -3.0093e-01,
-4.4296e-01, 3.0792e-01],
[-4.0310e-01, -1.1220e+00, -6.4106e-01, ..., 7.2150e-02,
1.2495e-01, 2.9935e-01]]],
[[[ 7.8394e-01, 9.1138e-01, 4.5447e-01, ..., 7.0798e-01,
1.0289e+00, 4.6199e-01],
[-1.1673e+00, -9.8839e-01, -5.9554e-04, ..., -3.2308e-01,
-2.2002e-01, -3.2227e-01],
[-2.8001e-01, -8.5185e-01, -7.6359e-02, ..., -1.1811e+00,
-2.1378e-01, 6.1697e-01],
...,
[ 3.2043e-01, -1.1594e+00, 8.6395e-01, ..., 2.9596e-01,
3.9544e-01, 3.4889e-01],
[-1.6999e-01, -1.7200e+00, 2.3526e-03, ..., 3.8689e-02,
1.8210e-01, -8.4113e-02],
[ 4.9376e-01, 1.1809e+00, -5.1468e-01, ..., -4.2332e-01,
-9.8558e-01, 1.4877e-01]],
[[-3.9067e+00, -2.0480e+00, -2.9333e+00, ..., -1.6321e+00,
-2.1367e+00, -1.8780e+00],
[-2.9907e+00, -1.7397e+00, -8.1360e-01, ..., -1.6810e+00,
-2.1401e+00, -2.1623e+00],
[-1.9616e+00, -1.4469e+00, -1.3760e+00, ..., -1.5977e+00,
-1.6923e+00, -1.4299e+00],
...,
[-3.7581e+00, -1.5458e+00, -3.5169e+00, ..., -2.9064e+00,
-5.0519e-01, -8.7753e-01],
[-2.8542e+00, -2.3736e+00, -4.0158e+00, ..., -7.7899e-01,
-7.3914e-01, -1.2216e+00],
[-1.8944e+00, -3.1879e+00, -3.0211e+00, ..., -7.1933e-01,
-7.0733e-01, -5.1397e-01]],
[[ 2.6375e+00, 3.2599e+00, 3.1337e+00, ..., 1.5176e+00,
8.3018e-01, 2.6018e+00],
[ 7.2206e-01, 1.2519e+00, 2.4382e+00, ..., 1.0103e+00,
7.1793e-01, 2.6832e+00],
[ 1.0583e+00, 7.9944e-01, 1.5485e+00, ..., -1.3259e-01,
8.8713e-01, 1.8625e+00],
...,
[-5.2072e-01, -3.4462e+00, -1.7107e+00, ..., -3.3687e-01,
6.8377e-01, 1.4946e+00],
[-1.6607e+00, -5.5738e+00, -1.2224e+00, ..., -3.2542e-01,
5.0197e-01, 1.1801e+00],
[-3.4152e-01, -3.2294e+00, -5.3710e-01, ..., 3.3034e-01,
1.3273e+00, 1.3839e+00]],
...,
[[-2.5848e+00, -7.6852e+00, -4.8419e+00, ..., -5.7634e+00,
-3.7702e+00, -4.4510e+00],
[-3.1115e+00, -9.1994e+00, -6.2326e+00, ..., -5.7547e+00,
-3.8348e+00, -3.4264e+00],
[-2.1723e+00, -6.3176e+00, -3.6178e+00, ..., -5.1864e+00,
-2.4195e+00, -1.4693e+00],
...,
[ 2.1399e-01, 1.6804e-01, -3.9335e+00, ..., -6.3319e+00,
-3.4946e+00, -2.8250e+00],
[-9.1981e-01, -1.6162e+00, -7.5680e+00, ..., -6.8219e+00,
-6.1675e+00, -4.4291e+00],
[-1.6939e+00, -4.0596e+00, -5.4415e+00, ..., -3.7561e+00,
-2.5983e+00, -2.4623e+00]],
[[-7.4718e-01, -1.3382e-01, 3.0408e-01, ..., 1.9023e+00,
1.6393e+00, 1.1626e+00],
[ 1.4105e+00, 9.0737e-01, 1.2158e+00, ..., 1.8662e+00,
1.2608e+00, 1.8763e+00],
[ 1.1384e+00, 3.4045e-01, 4.5767e-01, ..., 2.2485e+00,
5.4401e-01, 1.3188e+00],
...,
[-1.1203e+00, -2.2925e-01, -1.1934e+00, ..., 7.3561e-01,
1.5829e+00, 9.8645e-01],
[-8.4699e-01, 5.2245e-01, -7.2957e-01, ..., 2.3909e+00,
1.0086e+00, 1.4820e+00],
[-6.4696e-01, -2.5726e-01, 3.3785e-01, ..., 8.6408e-01,
6.8263e-01, 1.4505e+00]],
[[-7.8550e-01, -7.5561e-01, -5.9593e-01, ..., -2.8658e-01,
-6.1112e-01, -9.2657e-02],
[-1.4038e+00, -1.4227e+00, -5.9212e-01, ..., -1.4546e+00,
-1.4163e+00, -1.1367e+00],
[-2.1211e+00, -6.7035e-01, 2.2518e-01, ..., -7.1445e-01,
-1.0761e+00, -2.5323e-01],
...,
[-2.6082e+00, -3.9601e-01, 1.0747e+00, ..., -3.8897e+00,
-1.7776e+00, -8.6462e-01],
[-1.6973e+00, -1.0341e+00, -4.8370e-01, ..., -6.6942e-01,
-1.2181e+00, -7.0521e-01],
[ 1.8339e-01, -7.3613e-01, -1.0183e+00, ..., -3.7088e-01,
-7.7807e-01, -1.3927e-02]]]], device='cuda:0'),
tensor([[[[ 1.0541e+00, 1.3127e+00, 1.2378e+00, ..., 1.2025e+00,
1.0847e+00, 5.1853e-01],
[-5.2485e-01, -2.3046e+00, -1.1405e+00, ..., 5.9986e-01,
-4.3052e-01, -1.6081e+00],
[-2.8108e-01, -1.0695e+00, -6.7185e-01, ..., -1.7292e-01,
1.3466e-01, 6.3461e-01],
...,
[ 4.6640e-01, -1.6861e-01, -2.7624e-01, ..., 4.2658e-01,
1.0639e+00, 8.0068e-01],
[-1.9438e-02, -7.5136e-01, 6.5354e-02, ..., 7.5564e-01,
1.4012e+00, 1.0419e-01],
[ 8.2610e-01, -2.0832e-01, -1.8317e-01, ..., 3.7789e-01,
3.0938e-01, 6.9578e-01]],
[[-2.6863e+00, 1.1131e-02, -1.3202e+00, ..., -1.5468e+00,
-2.6911e+00, -6.3867e-01],
[-3.8591e+00, -2.7137e+00, -2.3448e+00, ..., -6.3519e+00,
-4.3504e+00, -1.7308e+00],
[-2.9356e+00, -2.4836e+00, -2.8739e+00, ..., -6.0889e+00,
-2.6782e+00, -1.4801e+00],
...,
[-8.9051e-01, -3.5504e-01, -8.6399e-01, ..., -2.0195e+00,
6.4192e-01, 8.3588e-01],
[-2.6993e+00, -1.8481e-01, -3.4275e-01, ..., 1.1212e-02,
2.2059e-01, 6.7086e-01],
[-1.5664e+00, -1.2908e+00, -1.0355e+00, ..., -4.8933e-01,
-6.2516e-01, -1.5827e-01]],
[[ 2.3502e+00, 2.8614e+00, 2.5842e+00, ..., 1.6818e+00,
-3.2777e-01, 2.4893e+00],
[ 8.3498e-01, 6.5942e-01, 2.4408e+00, ..., 7.5413e-01,
-8.4209e-01, 2.8040e+00],
[ 1.3129e+00, 1.5152e+00, 1.9901e+00, ..., 1.6895e-01,
1.2825e+00, 2.9376e+00],
...,
[ 1.0418e+00, 8.8128e-01, 1.7216e+00, ..., 1.1551e+00,
6.6140e-01, 1.8377e+00],
[ 1.1770e+00, 2.2520e+00, 1.9934e+00, ..., 1.1812e+00,
5.2591e-01, 1.6377e+00],
[ 2.4181e+00, 1.8624e+00, 2.3664e+00, ..., 1.9778e+00,
9.9410e-01, 1.6654e+00]],
...,
[[-2.3516e+00, -6.4683e+00, -5.4906e+00, ..., -3.3409e+00,
-4.9927e+00, -5.8761e+00],
[-2.9451e+00, -7.0098e+00, -5.0964e+00, ..., -1.0521e+01,
-6.5709e+00, -5.2889e+00],
[-1.9967e+00, -5.5655e+00, -3.9799e+00, ..., -8.6877e+00,
-5.8886e+00, -3.3879e+00],
...,
[-2.4995e+00, -4.8365e+00, -2.4457e+00, ..., -3.5260e+00,
-4.0738e+00, -2.9836e+00],
[-2.5520e+00, -5.3982e+00, -3.8788e+00, ..., -6.3681e+00,
-4.9186e+00, -4.8546e+00],
[-2.6896e+00, -6.0431e+00, -2.8443e+00, ..., -2.9980e+00,
-1.6970e+00, -3.2757e+00]],
[[ 4.1377e-01, 1.1045e+00, 4.4467e-01, ..., -7.2582e-01,
1.7342e+00, 1.7008e+00],
[ 1.3155e+00, 8.3653e-01, 2.2510e-01, ..., -1.3885e+00,
1.8220e+00, 2.8257e+00],
[ 1.3075e+00, 7.3202e-01, 2.8198e-01, ..., -3.4748e-01,
1.5478e+00, 2.0408e+00],
...,
[ 1.6233e+00, 5.7353e-01, 1.2572e+00, ..., 1.6673e+00,
2.2996e+00, 1.4573e+00],
[ 2.0067e+00, 1.2611e+00, 1.7762e+00, ..., 2.3289e+00,
1.6334e+00, 2.5341e+00],
[ 1.1381e+00, 8.3937e-01, 1.3303e+00, ..., 1.3193e+00,
1.4737e+00, 1.5240e+00]],
[[-6.2193e-01, 8.1775e-02, -2.4970e-01, ..., 4.0988e-01,
-4.6366e-02, 9.9915e-01],
[-1.4205e+00, -1.8808e+00, -1.6169e+00, ..., -2.1486e-01,
-1.7683e+00, -2.8043e-01],
[-2.1740e+00, -1.2134e+00, -4.4113e-01, ..., -2.8272e+00,
-1.1306e+00, -7.8821e-02],
...,
[-2.0844e+00, -1.4367e+00, -6.1067e-01, ..., -1.5420e+00,
-5.2658e-01, -4.2120e-01],
[-1.7800e+00, -1.3301e+00, -6.9665e-01, ..., 2.3805e-01,
3.2052e-02, 4.4676e-01],
[-2.8446e-01, 3.7025e-01, -2.6086e-01, ..., 2.3127e-01,
2.8184e-02, 6.2817e-01]]],
[[[ 5.2846e-01, 8.1194e-01, 5.9004e-01, ..., 3.2809e-01,
3.9915e-01, -9.1903e-03],
[-2.8380e-01, -1.4960e+00, -1.5418e+00, ..., 1.3561e-02,
1.8587e-03, -3.6297e-01],
[-3.7398e-01, -8.7729e-02, -4.1675e-02, ..., 9.6145e-02,
2.4681e-01, 6.8532e-01],
...,
[-2.7788e-01, -8.0076e-01, -5.1695e-01, ..., -2.1894e-01,
4.2784e-01, 7.9374e-01],
[-6.6769e-01, -1.7326e+00, -1.0373e+00, ..., 1.7322e+00,
1.1392e+00, 1.5737e-01],
[ 3.1497e-01, -6.9502e-01, -1.1190e+00, ..., -1.8008e-01,
6.2815e-01, 3.6294e-01]],
[[-2.6474e+00, -1.4055e+00, -2.7258e+00, ..., -1.6970e+00,
-2.2545e+00, -1.4884e+00],
[-2.9487e+00, -1.2215e+00, -1.4248e+00, ..., -1.4933e+00,
-2.4295e+00, -1.6232e+00],
[-3.5823e+00, -2.0208e+00, -1.6374e+00, ..., -2.0637e+00,
-2.4210e+00, -1.2148e+00],
...,
[-1.1762e+00, -1.2754e+00, -1.8816e+00, ..., -1.7759e+00,
-1.1813e+00, 1.7858e-01],
[-2.1123e+00, -6.4547e-01, -1.1717e+00, ..., -2.1678e+00,
-1.4556e+00, 1.2589e-01],
[-8.8485e-01, -1.4018e+00, -1.8637e+00, ..., -2.1632e+00,
-1.6085e+00, -2.9892e-01]],
[[ 2.0435e+00, 2.8031e+00, 1.7175e+00, ..., 1.6581e+00,
7.9378e-01, 2.1039e+00],
[ 7.7236e-02, 1.0514e+00, 3.9832e-01, ..., 8.0606e-01,
3.2976e-01, 2.3237e+00],
[ 1.5536e+00, 1.4164e+00, -3.3827e+00, ..., 6.6761e-02,
7.9172e-01, 2.1111e+00],
...,
[ 4.7236e-01, 8.8417e-01, 1.6832e+00, ..., 1.7032e+00,
6.8958e-01, 1.6885e+00],
[ 4.7352e-01, 1.7090e+00, 1.4101e+00, ..., -9.2936e-01,
-7.2274e-01, 1.6120e+00],
[ 1.5736e+00, 9.8883e-01, 1.5950e+00, ..., -9.4592e-02,
4.0005e-01, 1.4195e+00]],
...,
[[-2.1048e+00, -5.1011e+00, -3.4160e+00, ..., -2.1856e+00,
-1.7713e+00, -3.1272e+00],
[-3.0283e+00, -5.5481e+00, -1.9274e+00, ..., -2.5609e+00,
-2.3802e+00, -3.1672e+00],
[-8.8333e-01, -6.4838e+00, -1.7331e+00, ..., -2.4083e+00,
-2.1867e+00, -2.1738e+00],
...,
[-1.6873e+00, -5.1659e+00, -3.4904e+00, ..., -2.4788e+00,
-2.0905e+00, -2.8969e+00],
[-2.1670e+00, -6.4893e+00, -4.3466e+00, ..., -8.0484e+00,
-4.1563e+00, -4.2731e+00],
[-2.5790e+00, -6.6544e+00, -3.5538e+00, ..., -1.8466e+00,
-2.6133e+00, -2.5034e+00]],
[[ 7.7356e-01, 1.0796e+00, -7.6513e-01, ..., 1.1270e+00,
1.5210e+00, 1.4190e+00],
[ 1.1258e+00, 1.3739e-01, -2.0524e-01, ..., 8.0310e-01,
8.2250e-01, 1.9963e+00],
[-2.5190e-01, -1.3890e+00, -1.5226e+00, ..., 6.9371e-01,
-3.8030e-02, 1.2900e+00],
...,
[ 1.2995e+00, -3.6267e-01, 2.4924e-01, ..., 1.2892e+00,
2.4384e+00, 1.8379e+00],
[ 2.0127e+00, 1.1866e+00, 1.1470e+00, ..., -9.8601e-01,
1.2345e+00, 2.6437e+00],
[ 1.0257e+00, 4.1748e-01, 4.3601e-01, ..., -2.1296e-01,
5.8870e-01, 1.4902e+00]],
[[-1.4275e-01, -1.0292e+00, -1.1220e+00, ..., -5.9807e-01,
-8.6743e-01, -1.0741e-01],
[-1.1284e+00, -3.0229e+00, -2.0497e+00, ..., -1.1842e+00,
-1.0072e+00, -6.2341e-01],
[-1.7177e+00, -8.5740e-01, -1.3957e-01, ..., 3.6322e-01,
-1.6247e-01, -2.2436e-01],
...,
[-2.1824e+00, -1.7205e+00, -8.9391e-01, ..., -1.3387e+00,
-8.3806e-01, 8.2683e-03],
[-2.3615e+00, -1.9120e+00, -9.8795e-01, ..., -6.4176e-02,
-1.1567e+00, 5.5147e-01],
[-3.3851e-01, -1.3991e-01, -4.7427e-01, ..., -4.9256e-01,
-1.7086e+00, -1.3062e-01]]],
[[[ 5.7422e-01, 8.5508e-01, 5.5449e-01, ..., 3.5288e-01,
4.3943e-01, 3.7949e-02],
[-2.4900e-01, -9.0060e-01, -5.3210e-01, ..., 9.8452e-02,
5.7856e-02, -2.9614e-01],
[-1.5460e-01, -9.2937e-02, 2.4935e-01, ..., -2.3375e-01,
2.9402e-02, 6.6322e-01],
...,
[ 4.3419e-01, -3.3233e-01, -2.5641e-01, ..., 4.0365e-01,
8.6072e-01, 6.6959e-01],
[-7.1316e-02, -6.7590e-01, 1.0441e-01, ..., 8.6995e-01,
1.5086e+00, 1.2476e-01],
[ 8.2649e-01, -2.1180e-01, -1.1038e-01, ..., 3.9412e-01,
2.7902e-01, 7.3186e-01]],
[[-2.3967e+00, -5.9181e-01, -1.8609e+00, ..., -1.6383e+00,
-2.2526e+00, -1.5807e+00],
[-2.5099e+00, -6.1743e-01, -1.5962e+00, ..., -1.2992e+00,
-2.4440e+00, -1.7911e+00],
[-2.4595e+00, -1.5134e+00, -1.7552e+00, ..., -1.9901e+00,
-2.3014e+00, -1.2044e+00],
...,
[-6.3480e-01, -8.6946e-01, -1.4613e+00, ..., -8.6441e-01,
-3.1676e-01, 6.7126e-01],
[-2.5759e+00, -1.8213e-01, -3.2992e-01, ..., -7.3306e-02,
-7.8929e-02, 6.6754e-01],
[-1.5404e+00, -1.2213e+00, -9.2897e-01, ..., -4.3881e-01,
-5.4521e-01, -1.4746e-01]],
[[ 1.9291e+00, 2.5264e+00, 1.6267e+00, ..., 1.5934e+00,
7.4331e-01, 2.1271e+00],
[-2.2750e-01, 3.5802e-01, 6.1557e-01, ..., 6.8330e-01,
2.7042e-01, 2.3064e+00],
[ 5.2599e-02, 2.5885e-01, 2.9200e-01, ..., -1.9242e-01,
7.2115e-01, 1.9374e+00],
...,
[ 7.2913e-01, 2.5113e-01, 1.2323e+00, ..., 4.2685e-01,
9.8363e-02, 1.6747e+00],
[ 1.0663e+00, 2.2002e+00, 1.9968e+00, ..., 9.5752e-01,
7.0217e-01, 1.7166e+00],
[ 2.3867e+00, 1.7939e+00, 2.4088e+00, ..., 1.9156e+00,
9.9714e-01, 1.7105e+00]],
...,
[[-1.9719e+00, -4.8758e+00, -3.2285e+00, ..., -2.0543e+00,
-1.6402e+00, -3.1015e+00],
[-2.4592e+00, -5.2450e+00, -3.0017e+00, ..., -2.3272e+00,
-2.1502e+00, -3.0533e+00],
[-1.5537e+00, -5.4273e+00, -3.3254e+00, ..., -2.3068e+00,
-2.4613e+00, -2.2257e+00],
...,
[-2.5833e+00, -4.4265e+00, -2.5379e+00, ..., -2.7249e+00,
-2.1599e+00, -2.7290e+00],
[-2.4676e+00, -5.3663e+00, -3.9965e+00, ..., -4.3355e+00,
-3.1223e+00, -4.8875e+00],
[-2.6039e+00, -5.8099e+00, -2.8430e+00, ..., -2.9736e+00,
-1.6251e+00, -3.3250e+00]],
[[ 9.1232e-01, 1.6296e+00, 8.1763e-01, ..., 1.0981e+00,
1.4905e+00, 1.3778e+00],
[ 1.1503e+00, 1.0379e+00, 9.0225e-01, ..., 9.3705e-01,
7.6679e-01, 1.9060e+00],
[ 6.1027e-01, -7.6412e-02, 9.7075e-02, ..., 5.6242e-01,
3.4116e-02, 1.2797e+00],
...,
[ 1.5031e+00, 1.2214e-03, 6.6007e-01, ..., 1.5049e+00,
1.8052e+00, 1.5057e+00],
[ 2.0473e+00, 1.1900e+00, 1.7827e+00, ..., 1.9699e+00,
1.7197e+00, 2.5512e+00],
[ 1.1687e+00, 9.0196e-01, 1.3485e+00, ..., 1.3131e+00,
1.5148e+00, 1.5228e+00]],
[[-9.2912e-02, -6.0969e-01, -1.1808e+00, ..., -5.3201e-01,
-8.4667e-01, -1.2751e-01],
[-8.7488e-01, -2.4896e+00, -2.5661e+00, ..., -1.2127e+00,
-1.0318e+00, -7.1420e-01],
[-1.4689e+00, -1.8370e+00, -1.1732e+00, ..., -2.4571e-01,
-4.0976e-01, -2.5707e-01],
...,
[-1.6927e+00, -1.6413e+00, -8.2154e-01, ..., -4.0647e-01,
-3.7635e-01, -4.4064e-01],
[-1.7415e+00, -1.2081e+00, -6.2007e-01, ..., -3.6717e-01,
-3.2028e-01, 4.2431e-01],
[-2.4559e-01, 4.5041e-01, -2.1076e-01, ..., 2.3046e-01,
5.3486e-02, 6.2522e-01]]],
[[[ 9.7028e-01, 1.5529e+00, 1.0652e+00, ..., 5.4059e-01,
3.2208e-01, -1.1516e-01],
[-8.1046e-01, -1.9403e+00, -9.3643e-01, ..., 4.0965e-01,
5.8408e-01, -6.5795e-01],
[ 5.0628e-02, -6.5596e-01, 9.5495e-02, ..., -6.7781e-01,
3.7075e-03, 8.0587e-01],
...,
[ 4.2383e-01, -3.7324e-01, -5.9303e-01, ..., 3.7953e-01,
6.3960e-01, 2.5620e-02],
[-2.1756e-01, -1.0050e+00, -3.5355e-01, ..., -2.3132e-02,
1.2505e+00, 2.1944e-01],
[ 7.2092e-01, -2.5113e-01, -2.8174e-01, ..., -4.2029e-01,
-3.6221e-01, 3.4965e-01]],
[[-4.1419e+00, -1.1777e+00, -2.1268e+00, ..., -2.9239e+00,
-3.4763e+00, -8.8715e-01],
[-3.8836e+00, -2.6927e+00, -2.3372e+00, ..., -2.0468e+00,
-3.2001e+00, -6.8344e-01],
[-2.5258e+00, -1.8672e+00, -3.3433e+00, ..., -3.6375e+00,
-3.4272e+00, -5.2356e-01],
...,
[-8.3078e-01, -5.7544e-01, -1.8561e+00, ..., -2.8116e+00,
-1.4118e+00, 1.8785e-01],
[-2.8534e+00, -4.3775e-01, -1.0177e+00, ..., -1.6437e+00,
-1.3356e+00, 1.5372e-02],
[-2.0326e+00, -1.3224e+00, -1.6364e+00, ..., -9.9209e-01,
-9.0906e-01, 2.5140e-02]],
[[ 2.9943e+00, 4.0430e+00, 3.3519e+00, ..., 2.3041e+00,
7.5223e-01, 2.4223e+00],
[ 3.7559e-01, 8.1039e-01, 2.3749e+00, ..., 1.8173e+00,
1.2326e-01, 2.4130e+00],
[ 8.8380e-01, 1.3206e+00, 2.1909e+00, ..., 1.7379e+00,
7.9406e-01, 2.2525e+00],
...,
[ 2.6443e-01, 8.3239e-01, 1.4405e+00, ..., -3.0227e-01,
-1.4945e+00, 5.1961e-01],
[ 3.4586e-01, 2.1202e+00, 1.6795e+00, ..., 1.6824e-01,
7.8013e-02, 1.1446e+00],
[ 1.3931e+00, 1.3460e+00, 1.7792e+00, ..., 1.0787e+00,
8.7402e-01, 1.3556e+00]],
...,
[[-2.0595e+00, -7.2444e+00, -5.7539e+00, ..., -4.2001e+00,
-3.0542e+00, -3.8769e+00],
[-2.7550e+00, -6.4803e+00, -4.5462e+00, ..., -6.0966e+00,
-4.7528e+00, -4.6015e+00],
[-2.5413e-01, -2.7769e+00, -3.3498e+00, ..., -5.6382e+00,
-3.5863e+00, -2.0309e+00],
...,
[-9.3019e-01, -2.4464e+00, -1.9430e+00, ..., -5.7542e+00,
-3.5431e+00, -1.9266e+00],
[-1.2750e+00, -3.7899e+00, -3.4379e+00, ..., -6.3215e+00,
-4.4500e+00, -4.2926e+00],
[-2.1887e+00, -5.0042e+00, -3.0867e+00, ..., -4.4078e+00,
-2.6613e+00, -3.1722e+00]],
[[-7.3333e-01, -3.8172e-02, 1.0514e-02, ..., 1.0002e+00,
1.0731e+00, 1.7579e+00],
[ 1.7928e+00, 1.1148e+00, 1.1980e+00, ..., 1.3804e+00,
1.5087e+00, 2.8789e+00],
[ 1.1965e+00, 7.7092e-02, -1.7516e-01, ..., 1.7385e+00,
9.2017e-01, 2.2251e+00],
...,
[ 1.2818e+00, -1.7934e-01, 3.3155e-01, ..., 1.1758e+00,
1.1371e+00, 1.3296e+00],
[ 1.8618e+00, 9.2082e-01, 1.5809e+00, ..., 1.6669e+00,
9.3375e-01, 2.1106e+00],
[ 1.0537e+00, 4.6316e-01, 8.6623e-01, ..., 9.6560e-01,
1.1694e+00, 1.5500e+00]],
[[-5.2184e-01, 5.7625e-01, 1.8263e-02, ..., -2.6804e-01,
-5.2709e-01, 9.4128e-01],
[-1.0299e+00, -1.1048e+00, -1.7057e+00, ..., -3.6399e-01,
-1.0759e-01, 1.0321e+00],
[-2.0058e+00, -6.9399e-01, -9.4080e-01, ..., 7.3037e-01,
2.6325e-01, 5.3026e-01],
...,
[-1.7307e+00, -1.4017e+00, -1.4962e+00, ..., -3.8848e-01,
-5.5219e-01, -9.9370e-01],
[-1.7180e+00, -1.0946e+00, -1.0967e+00, ..., -6.5264e-01,
-1.1030e+00, 1.5075e-01],
[-3.4402e-01, 4.4583e-01, -4.1116e-01, ..., 1.1548e-01,
-2.1687e-01, 6.0998e-01]]],
[[[ 5.7902e-01, 7.6396e-01, 5.3075e-01, ..., -1.3252e-01,
-1.5373e-01, -4.4215e-01],
[-6.3152e-01, -6.0780e-01, -1.9204e-01, ..., -3.7000e-01,
2.6024e-01, -1.6839e-01],
[ 1.0920e-01, -9.7982e-01, -3.6456e+00, ..., 1.2028e-01,
-1.7465e+00, -2.5704e-01],
...,
[-3.5609e-01, -1.0232e+00, 3.8500e-01, ..., -2.5137e+00,
3.7131e-01, -2.7744e+00],
[-1.3332e-01, -1.0707e+00, -5.5352e-01, ..., 2.5255e+00,
2.8356e+00, 4.4112e+00],
[ 8.9033e-01, -1.8720e-01, -6.7521e-01, ..., -4.4541e+00,
-8.9700e-01, 1.6617e+00]],
[[-4.0815e+00, -2.5005e+00, -5.5899e+00, ..., -2.8480e+00,
-3.3890e+00, -1.3239e+00],
[-3.1249e+00, -2.7986e+00, -8.4864e+00, ..., -1.6201e+00,
-2.5180e+00, -8.6880e-01],
[-2.9680e+00, -6.4486e+00, -1.1993e+01, ..., -5.9511e+00,
-4.2544e+00, -3.3382e-01],
...,
[-1.9998e+00, -3.6606e+00, -2.3835e+00, ..., -3.0229e+01,
-3.5387e+01, -1.2031e+01],
[-3.3567e+00, -9.8233e-01, -2.5400e+00, ..., -1.1919e+01,
-7.7770e+00, -2.7064e+00],
[-2.4953e+00, -2.1234e+00, -2.5544e+00, ..., -1.3504e+01,
-5.4581e+00, -1.0018e+00]],
[[ 2.6908e+00, 2.3426e+00, 1.4147e+00, ..., 1.7009e+00,
9.8593e-01, 2.1775e+00],
[ 7.1760e-01, -1.7455e-01, 1.6181e+00, ..., 3.1074e+00,
1.5307e+00, 1.6633e+00],
[ 6.8594e-01, -2.4814e-01, 8.3372e-01, ..., 3.4365e-02,
2.3539e+00, 3.4335e+00],
...,
[ 3.5317e-01, 3.3978e+00, 2.9113e+00, ..., -2.2811e+01,
-1.5519e+01, -1.2152e+01],
[ 9.2369e-01, 3.2334e+00, 3.9465e+00, ..., -1.5872e+01,
1.7299e-01, 7.0759e-01],
[ 2.3578e+00, 2.0952e+00, 1.5586e+00, ..., -1.0915e+01,
-9.8288e-01, 9.6418e-01]],
...,
[[-2.5040e+00, -6.1531e+00, -5.3141e+00, ..., -2.9713e+00,
-2.3406e+00, -3.0725e+00],
[-3.0088e+00, -7.7713e+00, -8.1669e+00, ..., -5.7389e+00,
-4.4220e+00, -5.2250e+00],
[-1.7690e+00, -6.2271e+00, -8.2048e+00, ..., -3.7156e+00,
-2.4878e+00, -4.2704e+00],
...,
[-2.6994e+00, -4.7456e+00, -5.2093e+00, ..., -7.5564e+00,
-1.9943e+01, -1.5866e+01],
[-2.2689e+00, -5.6374e+00, -5.4459e+00, ..., -2.2682e+01,
-1.8857e+01, -1.3008e+01],
[-2.5803e+00, -6.2007e+00, -3.3788e+00, ..., -1.1688e+01,
-9.1346e+00, -8.2684e+00]],
[[-2.4746e-01, 1.8780e-01, -2.7635e-01, ..., 6.2649e-01,
1.0367e+00, 1.7902e+00],
[ 1.1737e+00, -2.6398e-01, -2.4484e+00, ..., 1.1882e+00,
1.2072e+00, 2.2067e+00],
[ 4.6953e-01, -2.7580e+00, -6.3074e+00, ..., -3.0724e+00,
-3.8674e-01, 2.6919e+00],
...,
[ 1.5226e+00, -1.0416e+00, -2.5667e+00, ..., -1.0039e+01,
-1.1431e+01, -3.3229e+00],
[ 1.9754e+00, 6.6034e-01, 9.2336e-01, ..., -3.7545e+00,
-7.7479e-01, 5.9510e+00],
[ 1.0263e+00, 3.0768e-01, 1.1189e+00, ..., -3.5263e+00,
1.0552e+00, 2.1378e+00]],
[[-5.5010e-01, -9.7965e-01, -1.3818e+00, ..., -7.2788e-01,
-9.0443e-01, 3.4514e-01],
[-1.2759e+00, -3.5835e+00, -3.0070e+00, ..., -6.4939e-01,
-5.8183e-01, 6.3200e-01],
[-2.2227e+00, -4.4708e+00, -3.4114e+00, ..., -3.0559e-01,
-2.2632e+00, -5.2121e-01],
...,
[-1.0797e+00, -6.5133e-01, -1.5793e-01, ..., 9.4486e-01,
-6.6425e+00, -8.8732e+00],
[-1.3737e+00, -9.8435e-01, -1.1333e+00, ..., 2.4452e-01,
-1.7095e+00, -2.5934e+00],
[-2.2932e-01, 5.0975e-01, -6.1707e-01, ..., -4.7071e+00,
2.1078e-02, 1.9336e-01]]]], device='cuda:0'),
tensor([[[[ 9.6515e-01, 1.7449e+00, 1.2616e+00, ..., 1.8381e+00,
1.5110e+00, 4.4681e-01],
[ 1.2116e-01, -3.1076e+00, -5.0402e+00, ..., 1.5149e-02,
5.8165e-01, -3.2146e-02],
[-6.6934e-01, -3.9575e+00, -3.9843e+00, ..., 1.2710e+00,
-1.7549e-01, 3.2452e-01],
...,
[-5.9575e+00, -1.1198e+01, -6.1657e+00, ..., -1.6038e+00,
-2.5765e+00, -9.0013e-01],
[-7.1012e+00, -1.0743e+01, -2.3394e+00, ..., -6.5988e-01,
-9.4589e-01, 5.4520e-01],
[-1.8215e+00, -1.8201e+00, 1.4466e+00, ..., 9.4738e-01,
2.8381e-01, 3.8888e-01]],
[[-4.0524e+00, -9.0626e-01, -2.0714e+00, ..., -5.9956e+00,
-8.8815e-01, -8.6718e-02],
[-1.0967e+01, -1.0349e+01, -8.2357e+00, ..., -8.8340e+00,
-2.0434e-01, 4.6438e-02],
[-1.8261e+01, -2.7349e+01, -2.9256e+01, ..., -7.8153e+00,
-4.4193e-02, -6.2361e-01],
...,
[-5.8816e+00, -2.2844e+01, -1.8327e+01, ..., 1.3334e+00,
-1.1058e+00, -1.7681e-01],
[-4.4299e+00, -1.2003e+01, -8.9857e+00, ..., 2.2568e+00,
2.1039e-01, -2.0426e-01],
[ 6.5176e-01, -5.5102e+00, -6.7478e+00, ..., 2.4247e+00,
1.9898e+00, 8.1017e-01]],
[[ 3.0331e+00, 4.3718e+00, 3.4443e+00, ..., -1.0729e+00,
-8.1705e-01, 1.0821e+00],
[-6.9988e-01, 3.0400e+00, 4.7463e+00, ..., -8.5358e-01,
-2.7926e+00, 9.6126e-01],
[ 4.8422e+00, 8.4371e+00, 7.9301e+00, ..., 1.7715e+00,
-2.9779e+00, 7.8129e-01],
...,
[-2.1861e+01, -9.4728e+00, -3.4948e+00, ..., -6.3043e+00,
-1.6849e+00, 6.8016e-01],
[-1.0486e+01, -7.4148e+00, -3.3089e+00, ..., -3.1359e+00,
-9.0522e-01, -4.0671e-01],
[-2.9722e+00, -5.5176e+00, -3.9359e+00, ..., -1.8761e+00,
-1.2494e-01, 3.5412e-01]],
...,
[[-1.8871e+00, -7.0341e+00, -5.5891e+00, ..., -8.7048e+00,
-8.1856e+00, -2.6995e+00],
[-5.7866e+00, -1.4680e+01, -1.5462e+01, ..., -1.4369e+01,
-1.2165e+01, -3.1117e+00],
[ 1.8507e+00, -8.4458e+00, -1.4872e+01, ..., -1.6667e+01,
-1.4827e+01, -2.3608e+00],
...,
[-1.3881e+01, -3.5837e+01, -6.8777e+00, ..., -4.5777e+00,
-2.9205e+00, -1.5922e-01],
[-4.7178e+00, -1.6124e+01, -1.7989e+00, ..., -4.3208e+00,
-2.7498e+00, 3.9149e-01],
[-4.2073e+00, -9.4794e+00, -5.6253e+00, ..., -1.5923e+00,
-3.9198e+00, -5.1008e-01]],
[[-5.1100e-01, -2.8406e-02, 6.8042e-02, ..., -2.3108e+00,
2.9978e+00, 7.1377e-01],
[-1.9952e+00, -2.0810e+00, 7.4437e-02, ..., -4.6484e+00,
3.9256e+00, 8.6461e-01],
[-1.4352e+01, -2.3557e+01, -1.5606e+01, ..., -4.0615e+00,
4.0469e+00, 8.4966e-01],
...,
[-1.2838e+00, -1.3036e+01, -1.7707e+01, ..., 2.2533e+00,
-3.0424e+00, -6.9704e-01],
[-4.5247e+00, -1.1082e+01, -1.1171e+01, ..., 1.4516e-01,
-2.8251e+00, -6.0813e-01],
[ 1.8784e+00, 9.1464e-01, -2.2627e+00, ..., -2.0870e-01,
-8.3227e-01, 6.3188e-01]],
[[-4.0929e-01, 7.5927e-01, 2.5494e-01, ..., 1.0967e+00,
2.7801e-01, -5.8195e-01],
[-7.8802e-01, -2.8247e+00, -3.2844e+00, ..., -2.6561e+00,
-2.7148e+00, -1.2167e+00],
[-1.6204e+00, -2.1552e+00, -4.5198e-01, ..., -1.5047e+00,
-3.6378e+00, -8.0599e-01],
...,
[-6.8657e+00, -2.4185e+01, -1.4441e+01, ..., -3.5135e-01,
-3.3339e+00, -1.5354e+00],
[-5.0239e+00, -1.8097e+01, -2.9897e+00, ..., -1.3118e+00,
-3.3802e+00, -5.4059e-01],
[-1.5251e+00, -3.1360e+00, 2.3254e+00, ..., -7.8898e-01,
-1.7407e+00, 5.0479e-01]]],
[[[ 8.5327e-01, 2.0232e+00, 1.6864e-01, ..., -1.9929e+00,
-4.1218e-01, -5.5327e-01],
[-4.5771e+00, -5.8596e+00, 5.7591e-01, ..., 8.9482e-01,
5.8601e-01, -1.3916e-01],
[-4.6797e+00, -9.3856e+00, -6.8335e+00, ..., -4.4132e-01,
-8.8038e-01, 6.9848e-01],
...,
[-1.6919e+00, 1.3205e+00, 4.2552e+00, ..., 3.1596e+00,
-1.5428e+00, 2.6971e-01],
[-1.0703e+00, -9.0009e-01, 1.3550e+00, ..., 2.5422e+00,
2.1408e+00, 4.9270e-01],
[ 1.6056e+00, -5.9003e-01, -3.9754e-01, ..., -2.2849e+00,
-1.3133e+00, 2.5064e-01]],
[[-6.6656e+00, -5.4477e+00, -6.8969e+00, ..., -4.8218e+00,
-4.5089e+00, -1.3473e+00],
[-5.2968e+00, -4.8702e+00, -7.0362e+00, ..., -4.8398e+00,
-3.7979e+00, -7.5465e-01],
[-4.2764e+00, -7.5525e+00, -8.3960e+00, ..., -8.0393e+00,
-3.0229e+00, -4.3817e-01],
...,
[-1.7910e+00, -6.3031e+00, -1.5377e+00, ..., -9.0452e-01,
-4.2213e+00, -1.8079e+00],
[-6.2254e+00, -4.9856e+00, -1.5870e+00, ..., -6.9346e-01,
-4.2232e+00, -2.2325e+00],
[-2.8827e+00, -2.1696e+00, -2.6202e+00, ..., -5.7533e+00,
-5.4137e+00, -2.1865e+00]],
[[ 2.7905e-01, -1.9140e+00, 7.1762e-01, ..., 2.4427e+00,
8.0725e-01, 2.1749e+00],
[-2.9805e+00, -7.1079e+00, -3.6286e+00, ..., 1.2116e-01,
-7.5724e-01, 2.0384e+00],
[-2.5303e-01, -6.8506e+00, -5.7211e+00, ..., -3.3449e-01,
7.0587e-01, 1.9343e+00],
...,
[-2.5725e+00, -1.8168e+00, -1.0145e+01, ..., -3.6482e+00,
-4.0333e+00, 3.4299e+00],
[-2.6328e+00, 1.9511e+00, -2.1017e+00, ..., -4.8591e+00,
-4.3452e+00, 8.2506e-01],
[ 2.4201e+00, 2.5152e+00, 1.5673e+00, ..., -1.9663e+00,
-1.4265e+00, 2.0591e+00]],
...,
[[-2.9519e+00, -6.6633e+00, -4.3047e+00, ..., -3.8604e+00,
-3.5078e+00, -3.2758e+00],
[-3.3658e+00, -5.8938e+00, -7.4310e+00, ..., -1.5198e+01,
-9.4471e+00, -4.6062e+00],
[-3.8987e+00, -9.5938e-01, -2.0185e+00, ..., -1.5646e+01,
-9.2629e+00, -2.8523e+00],
...,
[-4.0362e+00, -9.7022e+00, -4.6461e+00, ..., -2.1162e+00,
1.8260e-01, -8.6531e+00],
[-3.8091e+00, -7.9705e+00, -3.7530e+00, ..., -1.6047e+01,
-8.2986e+00, -1.0726e+01],
[-2.4384e+00, -7.4448e+00, -3.6531e+00, ..., -1.0135e+01,
-3.5641e+00, -4.5823e+00]],
[[-3.7220e+00, -7.0901e+00, -5.4242e+00, ..., -4.5729e-01,
9.9004e-01, 1.8180e+00],
[ 1.5683e+00, -3.2989e+00, -1.1884e+01, ..., -1.3623e+00,
2.1000e+00, 2.4152e+00],
[ 3.1263e-01, 5.5365e-01, -6.8671e+00, ..., -3.6024e+00,
3.3549e+00, 1.7920e+00],
...,
[ 2.3709e+00, 1.3550e+00, 3.7263e+00, ..., -4.1783e+00,
3.1201e-01, 1.3174e+00],
[ 1.2648e+00, -1.3950e+00, 1.0930e+00, ..., -1.8010e+00,
4.3909e-01, 1.9607e+00],
[ 1.1111e+00, -4.8984e-01, -2.6976e-01, ..., -2.0860e+00,
-9.8529e-02, 1.1312e+00]],
[[-1.8655e+00, 8.3542e-01, -2.7766e+00, ..., -2.5274e+00,
-6.1293e-01, 5.1080e-01],
[-3.4643e+00, -8.1222e-01, 3.7741e+00, ..., -8.1522e-01,
-3.7973e-01, 8.1750e-01],
[-3.2651e+00, -7.4821e+00, -5.1856e-01, ..., -3.2419e+00,
-2.2959e+00, 9.1883e-02],
...,
[ 1.9885e+00, 1.0130e+00, -1.8135e+00, ..., -1.2858e+00,
9.5005e-02, 1.0254e+00],
[-7.0872e-01, -1.8782e+00, -1.4353e+00, ..., -2.1350e+00,
-1.7113e+00, -8.0345e-01],
[-7.9662e-01, 1.5507e-01, 6.0116e-02, ..., -3.9361e+00,
-2.1572e+00, 2.2567e-01]]],
[[[ 7.3956e-01, 1.0537e+00, 7.2786e-01, ..., 3.9926e-01,
3.0153e-01, 2.1496e-01],
[-2.9991e-01, -1.3146e+00, -5.0001e-01, ..., 4.3251e-01,
2.7603e-01, -1.1666e+00],
[ 3.1553e-01, -1.1101e+00, -4.7264e-01, ..., -3.7774e-01,
9.8947e-01, 6.0557e-01],
...,
[ 8.1524e-02, -1.1537e+00, -5.5719e-01, ..., 4.2718e-03,
4.5376e-01, 9.6708e-01],
[-4.2877e-01, -1.8545e+00, -7.2751e-01, ..., 7.2355e-01,
1.9527e+00, 2.8913e-01],
[ 3.9685e-01, -5.2205e-01, -7.5866e-01, ..., -2.9016e-01,
1.4779e-01, 4.8287e-01]],
[[-2.9058e+00, -1.4943e+00, -3.0466e+00, ..., -1.8455e+00,
-2.3170e+00, -1.1531e+00],
[-4.2595e+00, -2.5573e+00, -3.2358e+00, ..., -3.3202e-01,
-1.3108e+00, -3.3805e-01],
[-4.5723e+00, -5.6978e+00, -4.9979e+00, ..., -1.1798e+00,
-1.5046e+00, -7.4490e-01],
...,
[-2.8310e+00, -1.7475e+00, -2.4205e+00, ..., -9.0023e-01,
-1.9855e+00, -8.4135e-01],
[-3.6764e+00, -1.3247e+00, -2.1784e+00, ..., -8.6063e-01,
-1.5690e+00, -8.4258e-01],
[-1.2816e+00, -8.8013e-01, -1.6355e+00, ..., -2.0184e+00,
-1.4808e+00, -1.0290e+00]],
[[ 1.6216e+00, 2.7411e+00, 2.3345e+00, ..., 2.7983e+00,
1.4518e+00, 2.9477e+00],
[ 6.7018e-02, 1.1854e+00, 1.4333e+00, ..., 1.8210e+00,
4.0159e-02, 2.5447e+00],
[ 7.1578e-01, 2.0261e+00, 2.2363e+00, ..., 1.5056e+00,
8.2998e-01, 1.8941e+00],
...,
[ 1.0090e+00, 1.3296e+00, 4.7730e-01, ..., -1.0602e+00,
-1.4575e+00, 1.8348e+00],
[ 6.8678e-01, 1.8273e+00, 9.4236e-01, ..., -1.0669e+00,
-1.7967e+00, 7.1401e-01],
[ 1.5892e+00, 1.5683e+00, 1.6063e+00, ..., 9.1729e-01,
9.6524e-01, 2.1046e+00]],
...,
[[-1.2779e+00, -4.4480e+00, -3.8358e+00, ..., -3.4285e+00,
-2.4807e+00, -3.9130e+00],
[-1.6967e+00, -3.9479e+00, -4.7942e+00, ..., -5.3877e+00,
-3.9521e+00, -4.4031e+00],
[-1.0593e+00, -3.9884e+00, -3.8544e+00, ..., -2.1322e+00,
-2.3101e+00, -1.6401e+00],
...,
[-1.6329e+00, -4.1230e+00, -4.4203e+00, ..., -8.5787e-01,
-1.2808e+00, -2.8641e+00],
[-1.5122e+00, -4.5858e+00, -5.2607e+00, ..., -4.7119e+00,
-3.8423e+00, -4.4211e+00],
[-1.6293e+00, -4.9300e+00, -4.0926e+00, ..., -2.9947e+00,
-1.7588e+00, -2.8647e+00]],
[[ 8.1955e-01, 4.9044e-01, -8.1334e-02, ..., 8.8307e-01,
1.0836e+00, 1.2485e+00],
[ 1.0136e+00, 7.7409e-01, 6.1283e-01, ..., 1.5738e+00,
2.3705e+00, 2.5736e+00],
[-2.3439e-01, -1.1464e+00, -1.9865e+00, ..., 1.3342e+00,
1.7388e+00, 1.9522e+00],
...,
[ 8.3860e-01, 2.9172e-02, -9.1501e-03, ..., 3.5360e-01,
1.4747e+00, 1.4275e+00],
[ 1.5470e+00, 1.4560e+00, 1.1501e+00, ..., 3.0984e-01,
7.9805e-01, 1.9558e+00],
[ 6.5094e-01, 8.9891e-01, 1.0675e+00, ..., 1.5211e-02,
1.0092e+00, 1.2785e+00]],
[[-2.8064e-01, -2.0968e-01, -9.8638e-01, ..., -2.2135e-01,
-2.2458e-01, 8.5742e-01],
[-1.4260e+00, -1.4950e+00, -1.5269e+00, ..., 2.1141e-01,
-1.3881e-01, 3.7322e-01],
[-2.3072e+00, -1.0368e+00, -4.4385e-01, ..., 3.9834e-01,
8.1146e-02, 2.0012e-01],
...,
[-1.7299e+00, -1.5862e+00, -1.5904e+00, ..., -1.2582e+00,
6.0682e-02, 3.4283e-01],
[-1.4475e+00, -1.6054e+00, -1.4562e+00, ..., -9.5722e-01,
-1.0849e+00, 1.8559e-02],
[-1.6260e-01, 3.8147e-02, -4.8231e-01, ..., -6.1513e-01,
-4.4791e-01, 4.7180e-01]]],
[[[ 5.2181e-01, 8.7371e-01, 5.9418e-01, ..., 1.7898e+00,
2.6217e-01, 4.1873e-02],
[-3.3366e-01, -8.3168e-01, -1.5279e-03, ..., 1.0967e+00,
1.0656e+00, -6.4239e-01],
[-1.5745e-01, 1.1612e-01, 4.3708e-01, ..., 2.1134e-01,
3.5773e-01, 4.4947e-01],
...,
[-2.3599e-02, -9.0924e-01, -5.4230e-01, ..., 1.1964e+00,
7.9255e-01, 8.7644e-01],
[-4.0127e-01, -1.9587e+00, -1.5599e+00, ..., 4.7244e-03,
5.2794e-01, 1.0519e-01],
[ 5.1687e-01, -7.9241e-01, -1.6036e+00, ..., -3.7604e-01,
-7.1770e-01, 1.6726e-01]],
[[-2.3384e+00, -9.7478e-01, -1.9464e+00, ..., -2.4019e+00,
-2.9161e+00, -1.6186e+00],
[-2.5451e+00, -7.8854e-01, -2.4589e-01, ..., -4.9379e+00,
-3.4057e+00, -2.0247e+00],
[-2.7680e+00, -1.4290e+00, 1.5128e-01, ..., -5.3186e+00,
-4.4670e+00, -1.1632e+00],
...,
[-1.0547e+00, -7.8088e-01, -1.9634e+00, ..., -9.9777e-01,
-1.2467e+00, 9.2451e-02],
[-2.2826e+00, -1.1158e-02, -2.0860e+00, ..., -1.7456e+00,
-1.6761e+00, -2.5229e-01],
[-1.0038e+00, -1.2285e+00, -2.8640e+00, ..., -2.4133e+00,
-1.7562e+00, -5.9777e-01]],
[[ 1.7854e+00, 2.4708e+00, 1.4086e+00, ..., -7.8142e-01,
5.0892e-01, 2.6921e+00],
[-1.4802e-01, 4.8739e-01, 2.2614e-01, ..., -2.0476e+00,
-1.2909e+00, 2.0715e+00],
[ 6.1754e-01, 1.2364e-01, -1.4176e+00, ..., -4.0618e-01,
3.0257e-01, 1.5526e+00],
...,
[ 1.0511e+00, 1.1761e+00, 1.9644e+00, ..., -1.3396e+00,
-1.5531e-01, 1.4052e+00],
[ 7.4856e-01, 2.0932e+00, 2.0210e+00, ..., -9.6601e-01,
-4.6897e-01, 9.5850e-01],
[ 1.7587e+00, 1.0469e+00, 2.2243e+00, ..., 2.3714e-01,
3.0105e-01, 1.2699e+00]],
...,
[[-1.7497e+00, -4.7138e+00, -2.9682e+00, ..., -1.0145e+01,
-4.5360e+00, -3.4591e+00],
[-2.1213e+00, -4.9097e+00, -2.7269e+00, ..., -9.0868e+00,
-7.7517e+00, -4.3175e+00],
[-1.3130e+00, -5.0459e+00, -2.7351e+00, ..., -3.4301e+00,
-7.4105e+00, -3.8919e+00],
...,
[-2.0728e+00, -5.3824e+00, -3.3162e+00, ..., -9.4238e+00,
-2.7688e+00, -2.9141e+00],
[-2.9067e+00, -7.2153e+00, -4.6549e+00, ..., -8.7267e+00,
-3.9073e+00, -4.3504e+00],
[-3.0653e+00, -7.5260e+00, -3.7566e+00, ..., -4.7794e+00,
-2.4542e+00, -2.7751e+00]],
[[ 8.4678e-01, 1.1879e+00, 3.6757e-01, ..., -2.8232e+00,
2.0416e+00, 1.8070e+00],
[ 1.0434e+00, 1.0567e+00, 1.2062e+00, ..., -2.3966e+00,
1.2630e+00, 2.4961e+00],
[ 2.4501e-01, 5.6146e-01, 1.5327e+00, ..., -1.2753e-01,
-2.6709e-02, 2.1629e+00],
...,
[ 1.6103e+00, 1.1710e-02, 7.5396e-03, ..., 2.3863e+00,
1.8434e+00, 1.8507e+00],
[ 2.2612e+00, 1.6173e+00, 9.7877e-01, ..., 1.9340e+00,
1.5115e+00, 2.2516e+00],
[ 1.0124e+00, 4.4028e-01, -7.5440e-03, ..., 1.1028e+00,
7.4035e-01, 1.1791e+00]],
[[-1.4834e-01, -7.0447e-01, -1.1010e+00, ..., -5.8897e-01,
-1.3982e+00, 1.3071e-01],
[-1.1087e+00, -2.4973e+00, -1.6645e+00, ..., -2.2770e+00,
-1.4581e+00, -1.4481e+00],
[-1.5243e+00, -1.3329e+00, 4.6344e-01, ..., -1.5628e+00,
-2.6402e+00, -1.4128e+00],
...,
[-2.5694e+00, -2.0475e+00, -8.1959e-01, ..., -2.7306e+00,
-4.4087e-01, 1.5300e-01],
[-2.4609e+00, -2.2462e+00, -1.3430e+00, ..., -2.8862e+00,
-8.5839e-01, 3.8142e-01],
[-3.8395e-01, -3.1291e-01, -7.5208e-01, ..., -1.6329e+00,
-7.4198e-01, 9.4218e-02]]],
[[[ 4.5928e-01, 5.9262e-01, 8.9668e-01, ..., 2.2976e+00,
2.5604e+00, -6.8852e-02],
[-1.3431e+00, -3.1109e+00, -4.4290e+00, ..., -3.5922e+00,
1.6161e+00, 4.8112e-01],
[-3.0087e+00, 4.0298e-01, 1.6266e+00, ..., -1.8643e+00,
-1.3412e+00, 1.4570e+00],
...,
[-5.6868e-03, -5.2878e-01, 1.1081e-01, ..., 8.8160e-01,
-6.3484e-02, 4.4032e-01],
[-2.3490e-01, -8.5305e-01, -2.3290e-01, ..., 5.1759e-02,
2.0909e+00, -1.8758e-01],
[ 4.8097e-02, -1.4009e-01, -7.8889e-01, ..., -8.2891e-01,
4.2927e-02, 5.7544e-01]],
[[-4.1746e+00, -1.9159e+00, -1.9916e+00, ..., -2.5048e+00,
-5.2630e+00, -2.0983e+00],
[-8.1157e+00, -8.1555e+00, -2.9378e+00, ..., -4.7798e+00,
-3.8894e+00, -2.2461e+00],
[-7.5769e+00, -8.4726e+00, -7.0153e+00, ..., -4.5937e+00,
-4.9621e+00, -3.0849e+00],
...,
[-5.6337e-02, -8.5685e-01, -7.9616e-01, ..., -4.3874e+00,
-4.2247e-01, 3.7492e-01],
[-1.3035e+00, -6.9034e-01, -5.1026e-01, ..., -3.4332e+00,
-4.2514e-01, 1.1611e-01],
[-4.0781e-01, -8.7477e-01, -8.3028e-01, ..., -1.2719e+00,
-9.4293e-01, -7.4450e-01]],
[[ 2.0581e+00, 3.0625e+00, 1.9354e+00, ..., -1.4083e+00,
-2.8542e-01, 3.5180e+00],
[-1.5985e+00, 5.2545e-01, 4.4855e+00, ..., -3.2098e+00,
-1.8057e+00, 2.0231e+00],
[-6.1217e+00, -8.2149e+00, -2.7791e+00, ..., -2.8410e+00,
-2.3853e+00, 1.6934e+00],
...,
[-3.0567e-01, -2.6770e-02, 6.0317e-01, ..., -8.3426e-01,
-1.8656e+00, 1.4324e+00],
[-8.0629e-01, 6.9644e-01, 8.5427e-01, ..., -1.9526e+00,
-2.6486e-01, 1.5507e+00],
[ 7.1211e-01, 5.0269e-01, 1.1582e+00, ..., 1.3133e+00,
3.0412e-01, 2.0721e+00]],
...,
[[-3.0528e+00, -9.1491e+00, -7.2641e+00, ..., -7.9867e+00,
-6.2839e+00, -4.4565e+00],
[-2.7060e+00, -1.0638e+01, -7.5191e+00, ..., -2.4902e+00,
-5.2170e+00, -8.3082e+00],
[-1.9866e+00, -1.3302e+01, -1.3772e+01, ..., 1.1076e+00,
4.4231e-01, -4.6362e+00],
...,
[-3.2768e+00, -7.7294e+00, -4.7958e+00, ..., -1.4418e+01,
-8.3713e+00, -2.0127e+00],
[-3.6823e+00, -8.6173e+00, -5.1222e+00, ..., -1.0623e+01,
-7.5525e+00, -4.0582e+00],
[-3.0291e+00, -7.5633e+00, -3.2911e+00, ..., -6.2094e+00,
-5.7238e+00, -3.0434e+00]],
[[-1.1583e+00, 9.1650e-01, 8.5911e-01, ..., -4.3418e+00,
-2.5381e+00, 8.6195e-01],
[-4.8719e+00, -3.8385e+00, 9.5788e-01, ..., -5.5011e-01,
-1.1732e+00, 1.7944e+00],
[-1.1871e+00, -1.2966e+01, -8.2129e+00, ..., 3.7066e-01,
-8.9914e-01, 9.3569e-01],
...,
[ 1.2368e+00, 1.1432e-01, 1.0217e+00, ..., 7.2034e-01,
2.7110e+00, 1.7357e+00],
[ 1.7852e+00, 1.5689e+00, 1.6564e+00, ..., -4.5425e-01,
1.8592e+00, 2.4868e+00],
[ 7.9714e-01, 6.0598e-01, 1.2053e+00, ..., 1.5852e+00,
1.4701e+00, 1.3021e+00]],
[[ 1.6781e-01, -3.7304e-01, -7.1423e-01, ..., 7.5586e-01,
-6.5412e-01, -8.7933e-01],
[-6.1666e-01, -2.6488e+00, -3.9451e+00, ..., -1.3389e+00,
1.0514e+00, 9.2056e-01],
[-3.8294e+00, 1.9612e+00, 3.9721e-01, ..., -4.7414e+00,
-6.2280e-03, -2.3337e-01],
...,
[-1.6948e+00, -2.0917e+00, -8.9618e-01, ..., 1.1815e+00,
-4.8022e+00, -3.2487e-01],
[-1.4945e+00, -2.2159e+00, -8.3109e-01, ..., -2.6535e+00,
-1.5875e+00, 3.9831e-01],
[-3.3439e-01, -8.6337e-01, -6.0060e-01, ..., 2.1233e-01,
3.7609e-01, 3.8014e-01]]]], device='cuda:0'),
tensor([[[[ 9.2592e-01, 1.5417e+00, 1.0849e+00, ..., 8.0596e-01,
1.0140e+00, 4.8226e-01],
[-8.5351e-01, -1.7196e+00, -8.3851e-01, ..., -2.8443e-01,
-2.3253e-01, -3.5351e-01],
[ 5.4089e-02, -5.2943e-01, 5.1410e-03, ..., -6.9471e-01,
-1.6238e-01, 6.4593e-01],
...,
[ 5.3783e-01, -2.1186e-01, 1.0472e-01, ..., -3.9335e-02,
-4.6583e-02, 3.6124e-01],
[-7.5181e-02, -1.1467e+00, -1.2023e-01, ..., -3.6073e-02,
1.1403e-01, -1.1569e-01],
[ 8.1442e-01, -3.8316e-01, -2.1450e-01, ..., -3.1856e-01,
-9.0184e-01, 1.7992e-01]],
[[-3.8374e+00, -1.3744e+00, -1.8792e+00, ..., -1.3819e+00,
-2.0228e+00, -1.8214e+00],
[-3.8301e+00, -3.0431e+00, -1.8955e+00, ..., -1.1964e+00,
-2.2389e+00, -2.1569e+00],
[-2.3988e+00, -1.8580e+00, -2.2904e+00, ..., -1.4027e+00,
-1.7119e+00, -1.4370e+00],
...,
[-8.4582e-01, -4.0549e-01, -9.7993e-01, ..., -6.4749e-01,
-1.0068e+00, -9.2363e-01],
[-3.1081e+00, -1.6762e-01, -6.4166e-01, ..., -5.7776e-01,
-1.1904e+00, -1.2512e+00],
[-2.3336e+00, -1.6400e+00, -1.2799e+00, ..., -6.3267e-01,
-7.8092e-01, -5.5773e-01]],
[[ 2.6388e+00, 3.7332e+00, 2.9680e+00, ..., 1.6828e+00,
8.1761e-01, 2.5469e+00],
[-6.8063e-02, 8.1162e-01, 2.0110e+00, ..., 1.2594e+00,
5.8467e-01, 2.6256e+00],
[ 5.4876e-01, 1.3461e+00, 1.4074e+00, ..., 7.1564e-01,
1.0377e+00, 1.8421e+00],
...,
[ 5.7735e-01, 1.0786e+00, 1.1362e+00, ..., -2.0612e-01,
5.3031e-01, 1.5118e+00],
[ 6.7622e-01, 2.3157e+00, 1.6973e+00, ..., -9.6392e-02,
8.1299e-01, 1.1868e+00],
[ 1.7775e+00, 1.3076e+00, 1.7836e+00, ..., 6.2882e-01,
1.3329e+00, 1.3630e+00]],
...,
[[-1.5120e+00, -6.7090e+00, -5.4990e+00, ..., -4.9583e+00,
-3.5802e+00, -4.3365e+00],
[-2.1078e+00, -6.0148e+00, -4.2664e+00, ..., -4.9901e+00,
-3.6355e+00, -3.1824e+00],
[-1.9484e-01, -2.4804e+00, -2.7863e+00, ..., -2.6634e+00,
-2.2109e+00, -1.4238e+00],
...,
[-1.0464e+00, -2.5409e+00, -1.9087e+00, ..., -4.0337e+00,
-3.1196e+00, -2.7253e+00],
[-1.6226e+00, -4.0047e+00, -3.3149e+00, ..., -5.1810e+00,
-4.1956e+00, -4.2625e+00],
[-2.5737e+00, -5.3959e+00, -2.8855e+00, ..., -3.4275e+00,
-2.4519e+00, -2.3976e+00]],
[[-6.4652e-01, -1.5910e-01, 3.1490e-01, ..., 1.5523e+00,
1.6939e+00, 1.1677e+00],
[ 1.6404e+00, 8.8131e-01, 1.6841e+00, ..., 1.5870e+00,
1.1462e+00, 1.8411e+00],
[ 1.0419e+00, 2.1249e-01, 6.4362e-01, ..., 1.7537e+00,
6.4105e-01, 1.2873e+00],
...,
[ 1.3665e+00, -3.4000e-02, 1.1341e+00, ..., 1.4440e+00,
6.7683e-01, 9.0828e-01],
[ 1.9839e+00, 1.2461e+00, 1.8174e+00, ..., 1.3010e+00,
7.1539e-01, 1.4655e+00],
[ 1.0008e+00, 5.6945e-01, 1.1625e+00, ..., 7.3871e-01,
6.8923e-01, 1.4454e+00]],
[[-4.7920e-01, 5.1294e-01, 1.7698e-01, ..., -3.0634e-01,
-5.6960e-01, -5.3581e-02],
[-1.0541e+00, -1.1739e+00, -1.1430e+00, ..., -1.2505e+00,
-1.4575e+00, -1.1480e+00],
[-1.8909e+00, -6.0810e-01, -3.2814e-01, ..., -8.6996e-01,
-1.0548e+00, -2.4438e-01],
...,
[-1.8891e+00, -1.4875e+00, -7.9487e-01, ..., -8.9971e-01,
-1.5558e+00, -8.5654e-01],
[-1.8785e+00, -1.1421e+00, -8.0813e-01, ..., -7.5989e-01,
-1.7228e+00, -6.8762e-01],
[-5.4075e-01, 2.1063e-01, -2.5149e-01, ..., -2.5017e-01,
-8.0262e-01, -7.0561e-03]]],
[[[ 8.4712e-01, 1.5407e+00, 1.2565e+00, ..., 1.3750e+00,
1.0930e+00, 9.8328e-01],
[-9.9721e-01, -1.9650e+00, -1.3627e+00, ..., -3.4061e-01,
-5.5639e-01, -1.5685e+00],
[-2.0215e-01, -1.0546e+00, -3.5288e-01, ..., -2.6057e-01,
7.8252e-01, 8.4784e-01],
...,
[-9.7341e-02, -1.4958e+00, -1.4723e+00, ..., -2.7240e-02,
1.7454e-01, 2.1094e-01],
[-6.8397e-01, -2.0635e+00, -2.0478e+00, ..., -5.9002e-01,
-3.0023e-01, -1.5781e-01],
[-2.7740e-01, -1.4795e+00, -7.4885e-01, ..., -1.2423e+00,
-1.5247e+00, -2.8514e-01]],
[[-3.8034e+00, -8.2682e-01, -2.1301e+00, ..., -1.5585e+00,
-2.4256e+00, -1.1928e+00],
[-4.0482e+00, -3.1567e+00, -3.4636e+00, ..., -2.1114e+00,
-3.1844e+00, -9.0042e-01],
[-2.4459e+00, -2.4431e+00, -3.9806e+00, ..., -1.9603e+00,
-2.9215e+00, -1.0444e+00],
...,
[-1.5819e+00, -1.5553e+00, -2.5768e+00, ..., -1.9429e+00,
-1.9150e+00, -6.9333e-01],
[-3.2110e+00, -8.5752e-01, -1.0421e+00, ..., -2.4962e+00,
-2.5382e+00, -1.2425e+00],
[-2.3237e+00, -1.1926e+00, -5.9854e-01, ..., -2.0229e+00,
-2.0405e+00, -1.0013e+00]],
[[ 2.9327e+00, 4.3129e+00, 3.3352e+00, ..., 2.9436e+00,
1.0926e+00, 3.0004e+00],
[ 2.7515e-01, 1.2314e+00, 2.4076e+00, ..., 1.1551e+00,
-9.8319e-01, 2.7358e+00],
[ 9.7761e-01, 2.0466e+00, 2.2106e+00, ..., 6.2502e-01,
-3.4596e-01, 2.0360e+00],
...,
[ 8.7576e-01, 1.2688e+00, 1.5819e+00, ..., 6.0833e-01,
1.0950e+00, 2.3108e+00],
[ 4.5558e-01, 2.2115e+00, 1.1250e+00, ..., 1.1320e-01,
1.1087e+00, 1.8396e+00],
[ 1.5638e+00, 1.8479e+00, 5.5229e-01, ..., 1.3022e+00,
1.8581e+00, 1.7597e+00]],
...,
[[-1.9308e+00, -7.3387e+00, -5.6375e+00, ..., -4.0349e+00,
-3.2489e+00, -5.5312e+00],
[-2.4763e+00, -7.1448e+00, -5.0788e+00, ..., -3.4539e+00,
-2.7584e+00, -4.3200e+00],
[-9.6909e-01, -4.7480e+00, -5.0307e+00, ..., -1.7054e+00,
-3.0906e+00, -2.4604e+00],
...,
[-2.7063e+00, -5.5212e+00, -3.8723e+00, ..., -5.7537e+00,
-5.3969e+00, -4.6944e+00],
[-3.0360e+00, -6.1215e+00, -4.4895e+00, ..., -6.1747e+00,
-5.7559e+00, -6.1516e+00],
[-1.8032e+00, -4.5763e+00, -2.7585e+00, ..., -3.2500e+00,
-2.7308e+00, -3.2308e+00]],
[[-1.9890e-01, 2.9128e-01, 3.5159e-01, ..., 6.7243e-01,
8.7809e-01, 1.4044e+00],
[ 1.6472e+00, 5.9864e-01, 1.4335e+00, ..., 1.6529e+00,
2.4514e+00, 3.0383e+00],
[ 1.2265e+00, -6.1491e-01, -7.0470e-01, ..., 1.7215e+00,
1.7476e+00, 2.3725e+00],
...,
[ 1.3603e+00, 2.0662e-01, -6.9426e-01, ..., 1.2538e+00,
1.1180e+00, 1.6439e+00],
[ 1.9247e+00, 1.9596e+00, 1.5437e+00, ..., 1.3927e+00,
1.2302e+00, 2.0080e+00],
[ 4.1635e-01, 9.0891e-01, 1.2575e+00, ..., 5.9064e-01,
6.9116e-01, 1.6829e+00]],
[[-5.7260e-01, 2.5844e-01, -4.6201e-02, ..., 5.8606e-01,
5.2508e-01, 1.3593e+00],
[-1.3777e+00, -2.1801e+00, -2.0325e+00, ..., -8.9624e-01,
-5.6507e-01, 2.3390e-01],
[-2.3013e+00, -1.3828e+00, -2.7469e-01, ..., -1.2591e-01,
1.0679e-01, 7.8108e-01],
...,
[-1.9628e+00, -2.3158e+00, -1.4926e+00, ..., -2.2317e-01,
-6.5123e-01, -7.3395e-01],
[-1.5672e+00, -2.3713e+00, -2.0119e+00, ..., -1.1343e+00,
-1.4255e+00, -6.9698e-01],
[-8.6877e-01, -7.1550e-01, -9.1282e-01, ..., -6.2137e-01,
-8.8779e-01, -9.1840e-02]]],
[[[ 6.6276e-01, 1.1069e+00, -6.0651e-01, ..., 1.8018e-01,
6.6968e-01, 2.6319e-01],
[-4.1968e-01, -1.6435e+00, -2.4432e+00, ..., -2.7814e+00,
-3.3429e+00, -1.3773e+00],
[ 8.2849e-01, -1.4001e+00, -4.2528e+00, ..., 2.3352e+00,
-2.0187e+00, -5.5213e+00],
...,
[ 3.8246e-01, -4.9956e-01, 3.4400e+00, ..., -1.0300e+00,
1.2574e+00, 7.2956e-01],
[-4.3348e-01, -2.8272e+00, 5.8030e-02, ..., -2.9977e-01,
-1.1101e+00, -7.2201e-01],
[ 4.0853e-01, -1.2462e+00, -1.6185e+00, ..., 5.5994e-02,
-1.2762e-01, 5.9724e-01]],
[[-3.4472e+00, -2.6833e+00, -7.4301e+00, ..., -1.5282e+00,
-3.1750e+00, -1.2008e+00],
[-4.4346e+00, -4.1140e+00, -6.8929e+00, ..., -6.0682e+00,
-4.3704e+00, -5.6646e-02],
[-6.9944e+00, -9.2549e+00, -1.0533e+01, ..., -1.0271e+01,
-1.1968e+01, -1.8755e+00],
...,
[-3.6859e+00, -7.3169e+00, -6.5959e+00, ..., -9.6748e-01,
-1.2358e+00, 1.7366e-01],
[-3.7115e+00, -2.4059e+00, -5.1740e+00, ..., -2.6299e+00,
-2.2951e+00, -1.4009e-01],
[-1.1824e+00, -2.1594e+00, -3.5805e+00, ..., -1.5438e+00,
-8.5195e-01, -1.5055e-01]],
[[ 2.0025e+00, 2.5836e+00, 1.8027e+00, ..., 3.1752e+00,
8.5318e-01, 1.9232e+00],
[ 5.0316e-01, 1.8380e+00, 1.6048e+00, ..., 4.2445e+00,
1.5018e+00, 4.3181e+00],
[ 5.6522e-01, 3.1060e+00, 1.3577e+00, ..., -2.1812e+00,
8.7569e-01, 6.9560e+00],
...,
[ 2.3052e+00, 6.7240e+00, 3.0533e+00, ..., -3.7465e+00,
-3.7544e+00, -1.5786e+00],
[ 8.9312e-01, 3.7625e+00, 3.5507e+00, ..., 8.1233e-01,
-3.9559e-01, 2.9703e-01],
[ 1.7671e+00, 1.6279e+00, 2.8822e+00, ..., 1.7532e+00,
1.4128e+00, 1.4861e+00]],
...,
[[-1.5220e+00, -5.1944e+00, -5.8462e+00, ..., -5.7569e+00,
-3.6003e+00, -4.2303e+00],
[-2.0157e+00, -4.6996e+00, -7.9529e+00, ..., -7.5011e+00,
-5.6992e+00, -6.2652e+00],
[-2.5212e+00, -7.4776e+00, -1.3802e+01, ..., -1.4925e+01,
-8.9591e+00, -6.2133e+00],
...,
[-2.1620e+00, -6.8754e+00, -1.2535e+01, ..., -7.2756e+00,
-6.0539e+00, -4.6854e+00],
[-1.9598e+00, -6.1494e+00, -7.6667e+00, ..., -7.6701e+00,
-5.2654e+00, -5.3256e+00],
[-1.9383e+00, -5.8616e+00, -4.5452e+00, ..., -4.9869e+00,
-3.3174e+00, -4.9397e+00]],
[[ 5.5078e-01, -9.2633e-01, -2.9958e+00, ..., 1.7374e+00,
6.6488e-01, 1.6203e+00],
[ 1.2552e+00, -1.4836e+00, -4.2926e+00, ..., 9.8089e-01,
2.1929e+00, 3.1236e+00],
[-5.3932e-01, -2.0142e+00, 4.9115e-01, ..., -1.3769e+01,
-1.2761e+01, -1.8267e+00],
...,
[ 8.2555e-01, -1.8544e+00, 3.1238e+00, ..., 1.6837e+00,
3.2536e-01, 1.2081e+00],
[ 1.6686e+00, 7.0866e-02, -2.2591e+00, ..., 2.2488e+00,
7.2651e-01, 2.5439e+00],
[ 6.9655e-01, 6.0958e-02, -1.0227e-01, ..., 1.6330e+00,
1.6181e+00, 1.5039e+00]],
[[-3.9408e-01, -1.6033e+00, -1.9061e+00, ..., -3.2197e-01,
-1.8310e-01, 9.5295e-01],
[-1.4758e+00, -3.2071e+00, -3.8133e+00, ..., -2.8790e+00,
-7.0114e-01, 2.8136e-01],
[-2.2931e+00, -2.0830e+00, -4.4369e+00, ..., 2.0143e+00,
-3.5461e+00, -3.5781e+00],
...,
[-1.2729e+00, -2.8756e+00, -2.3765e+00, ..., -3.2189e+00,
-2.3020e+00, -1.7516e+00],
[-1.4910e+00, -1.6757e+00, 5.2980e-01, ..., -2.4289e+00,
-2.8851e+00, -7.1943e-01],
[-1.5490e-01, -2.9662e-01, -5.4250e-01, ..., -1.4319e+00,
-8.1743e-01, 2.0756e-01]]],
[[[ 7.2447e-01, 1.0674e+00, 7.1517e-01, ..., 4.0636e-01,
3.3422e-01, 1.8986e-01],
[-2.9276e-01, -1.4427e+00, -5.5360e-01, ..., 4.9252e-02,
3.9650e-01, -1.1280e+00],
[ 1.4846e-01, -8.2907e-01, -4.4970e-01, ..., -1.1290e+00,
1.0378e+00, 6.1578e-01],
...,
[ 1.5317e-01, -1.1157e+00, -2.7105e-01, ..., 2.1518e-01,
9.0348e-01, 5.5504e-01],
[-3.7347e-01, -1.8734e+00, -6.2895e-01, ..., 3.3923e-02,
1.4312e+00, -1.1678e-01],
[ 4.6459e-01, -5.1511e-01, -7.4237e-01, ..., -1.4053e-01,
3.7669e-01, 6.0153e-01]],
[[-3.0079e+00, -1.5091e+00, -3.1613e+00, ..., -1.8867e+00,
-2.1872e+00, -1.0489e+00],
[-4.3445e+00, -1.8351e+00, -2.4308e+00, ..., -5.3688e-01,
-1.1894e+00, -2.6001e-01],
[-4.3835e+00, -2.6538e+00, -3.0976e+00, ..., -1.5321e+00,
-1.4089e+00, -7.3736e-01],
...,
[-2.7199e+00, -1.5076e+00, -1.9843e+00, ..., -5.7356e-01,
1.9869e-01, 5.1108e-01],
[-3.5595e+00, -1.1786e+00, -1.9055e+00, ..., -1.0017e+00,
-9.2291e-02, 2.8313e-01],
[-1.1964e+00, -8.4829e-01, -1.7229e+00, ..., -6.7681e-01,
-6.5750e-01, -5.4445e-01]],
[[ 1.6896e+00, 2.7951e+00, 2.3469e+00, ..., 2.6743e+00,
1.4826e+00, 2.8749e+00],
[ 1.4339e-01, 1.1976e+00, 1.6131e+00, ..., 1.8052e+00,
5.5306e-02, 2.4921e+00],
[ 7.0841e-01, 1.4554e+00, 9.7375e-01, ..., 1.3323e+00,
5.9393e-01, 1.8614e+00],
...,
[ 9.0669e-01, 1.0076e+00, 2.3692e-01, ..., -9.5526e-01,
1.8837e-01, 1.3914e+00],
[ 7.3223e-01, 1.5840e+00, 5.7675e-01, ..., -5.6433e-01,
5.7679e-01, 1.5089e+00],
[ 1.5758e+00, 1.3800e+00, 1.4428e+00, ..., 5.0643e-01,
9.9306e-01, 1.9933e+00]],
...,
[[-1.4096e+00, -4.6026e+00, -3.9625e+00, ..., -3.3490e+00,
-2.4751e+00, -3.8763e+00],
[-1.7990e+00, -3.8743e+00, -4.2953e+00, ..., -5.6239e+00,
-3.9263e+00, -4.3944e+00],
[-9.6132e-01, -3.2617e+00, -3.8927e+00, ..., -4.0133e+00,
-2.4408e+00, -1.6434e+00],
...,
[-1.5025e+00, -3.7172e+00, -3.8368e+00, ..., -1.0908e+01,
-2.7113e+00, -1.4412e+00],
[-1.4173e+00, -4.2017e+00, -4.6696e+00, ..., -8.6938e+00,
-3.9751e+00, -3.6478e+00],
[-1.6252e+00, -4.7638e+00, -3.7964e+00, ..., -4.4187e+00,
-2.6888e+00, -2.7110e+00]],
[[ 8.0356e-01, 5.1124e-01, -1.3744e-01, ..., 8.2704e-01,
1.1022e+00, 1.2524e+00],
[ 1.1910e+00, 9.5890e-01, 4.3933e-01, ..., 1.7452e+00,
2.3912e+00, 2.5563e+00],
[ 5.7128e-01, 5.6843e-01, 5.7587e-02, ..., 1.9278e+00,
1.6721e+00, 1.9409e+00],
...,
[ 9.2748e-01, 2.0264e-03, -3.7043e-02, ..., 3.3641e+00,
2.3618e+00, 1.6423e+00],
[ 1.5131e+00, 1.4555e+00, 1.1545e+00, ..., 2.3448e+00,
1.8253e+00, 2.4356e+00],
[ 6.4558e-01, 8.1294e-01, 9.0565e-01, ..., 1.6248e+00,
1.3180e+00, 1.3516e+00]],
[[-2.6833e-01, -1.8797e-01, -1.0456e+00, ..., -2.3221e-01,
-1.7440e-01, 8.3534e-01],
[-1.3773e+00, -1.4997e+00, -2.1090e+00, ..., 1.4205e-01,
-4.3642e-02, 3.6525e-01],
[-1.6684e+00, -9.0635e-01, -1.1532e+00, ..., 3.4891e-01,
2.2478e-01, 2.3096e-01],
...,
[-1.8598e+00, -1.6198e+00, -1.2713e+00, ..., -1.0360e+00,
-5.1673e-01, -2.9423e-01],
[-1.5184e+00, -1.6389e+00, -1.2518e+00, ..., -1.7027e+00,
-3.7780e-01, 3.3097e-01],
[-1.0898e-01, -1.3895e-01, -5.7652e-01, ..., -6.9127e-01,
1.8266e-01, 3.1573e-01]]],
[[[ 5.9391e-01, 8.4321e-01, 7.3662e-01, ..., 2.8921e-01,
3.6483e-01, -1.0312e-05],
[-2.3224e-01, -7.0165e-01, -4.2459e-01, ..., 1.8679e-01,
6.0950e-03, -3.8136e-01],
[-1.6943e-01, 8.1742e-02, -9.4478e-01, ..., 3.8011e-01,
2.7657e-01, 7.1719e-01],
...,
[ 2.6931e-02, -1.0538e+00, -5.4283e-01, ..., 8.7475e-02,
1.0669e+00, 9.2791e-01],
[-4.0412e-01, -1.9323e+00, -8.7799e-01, ..., -6.3232e-02,
9.2887e-01, 1.5535e-01],
[ 4.1063e-01, -8.3509e-01, -1.1449e+00, ..., 1.4735e+00,
-3.0837e-01, 1.9872e-01]],
[[-2.4665e+00, -6.3961e-01, -2.3105e+00, ..., -1.6323e+00,
-2.3129e+00, -1.5343e+00],
[-2.5513e+00, -1.1327e+00, -3.5436e+00, ..., -1.1986e+00,
-2.4565e+00, -1.7066e+00],
[-2.4757e+00, -2.5557e+00, -5.5706e+00, ..., -1.6625e+00,
-2.3690e+00, -1.2490e+00],
...,
[-1.0778e+00, -8.2566e-01, -1.6425e+00, ..., -1.6746e+00,
-1.1372e+00, 1.3993e-01],
[-2.1476e+00, -4.0745e-02, -1.3344e+00, ..., -2.2408e+00,
-1.8251e+00, -2.1918e-01],
[-9.2855e-01, -1.1560e+00, -1.8965e+00, ..., -3.5151e+00,
-1.8550e+00, -5.6391e-01]],
[[ 1.9697e+00, 2.5401e+00, 1.6870e+00, ..., 1.7151e+00,
8.0952e-01, 2.0814e+00],
[-1.6837e-01, 3.0889e-01, 9.6225e-01, ..., 1.1512e+00,
4.1288e-01, 2.2569e+00],
[ 1.5179e-02, 1.1693e+00, 2.2835e+00, ..., 2.8858e-01,
8.1460e-01, 2.0220e+00],
...,
[ 1.1499e+00, 1.2812e+00, 1.5175e+00, ..., 2.4659e-02,
-4.4217e-01, 1.3352e+00],
[ 7.8709e-01, 2.0434e+00, 1.5722e+00, ..., -1.2368e+00,
-7.9592e-01, 9.6738e-01],
[ 1.7481e+00, 9.9532e-01, 1.6032e+00, ..., -1.3551e-01,
-1.4864e-01, 1.2448e+00]],
...,
[[-1.9925e+00, -4.8609e+00, -3.2613e+00, ..., -1.8611e+00,
-1.5736e+00, -3.0582e+00],
[-2.4851e+00, -5.3246e+00, -3.8445e+00, ..., -2.2542e+00,
-2.2035e+00, -3.0911e+00],
[-1.5275e+00, -5.4589e+00, -3.6105e+00, ..., -2.5203e+00,
-2.2020e+00, -2.0655e+00],
...,
[-2.0825e+00, -5.4404e+00, -2.8643e+00, ..., -1.6290e-01,
-1.7267e+00, -2.9131e+00],
[-2.9306e+00, -7.3088e+00, -4.3294e+00, ..., -2.9857e+00,
-3.7143e+00, -4.3400e+00],
[-3.0264e+00, -7.3118e+00, -3.4831e+00, ..., -5.6313e+00,
-2.1433e+00, -2.7564e+00]],
[[ 8.8743e-01, 1.6251e+00, 3.9200e-01, ..., 1.1063e+00,
1.4926e+00, 1.4169e+00],
[ 1.1657e+00, 6.2776e-01, -1.0386e-01, ..., 1.0504e+00,
8.4327e-01, 1.9747e+00],
[ 6.5615e-01, -9.8923e-01, -2.7513e+00, ..., 9.8697e-01,
1.1538e-01, 1.2977e+00],
...,
[ 1.6029e+00, 7.6506e-02, 4.7116e-01, ..., -8.6724e-01,
1.9520e+00, 1.8395e+00],
[ 2.2707e+00, 1.5481e+00, 1.1435e+00, ..., -1.3581e+00,
1.8750e+00, 2.2741e+00],
[ 1.0649e+00, 5.3505e-01, 5.7275e-01, ..., -2.1910e+00,
1.4466e+00, 1.1890e+00]],
[[-1.0693e-01, -6.0641e-01, -1.5146e+00, ..., -6.1365e-01,
-9.1528e-01, -1.1524e-01],
[-8.6135e-01, -2.9074e+00, -3.2386e+00, ..., -1.1868e+00,
-9.7666e-01, -6.1610e-01],
[-1.4616e+00, -2.1858e+00, -1.9631e+00, ..., 1.8323e-01,
-1.4807e-01, -1.9718e-01],
...,
[-2.6023e+00, -2.0517e+00, -8.9654e-01, ..., -2.4239e-02,
-5.4605e-01, 1.2577e-01],
[-2.5012e+00, -2.1795e+00, -9.5166e-01, ..., -1.4477e-01,
-1.1563e+00, 2.6077e-01],
[-4.5886e-01, -3.1351e-01, -5.3661e-01, ..., -1.7989e-01,
-1.5850e+00, -6.3055e-03]]]], device='cuda:0'),
tensor([[[[ 1.7032e-01, 2.6953e-02, -8.3268e-01, ..., 3.7675e-01,
3.8666e-01, 2.4327e-01],
[-1.6964e+00, -4.5999e+00, -8.8239e-01, ..., 3.2420e-01,
3.2365e-01, -1.2518e+00],
[-1.9851e+00, -2.9457e+00, -2.2782e+00, ..., 2.1817e-01,
9.8142e-01, 4.7343e-01],
...,
[ 9.0998e-03, -1.1938e+00, -6.2476e-01, ..., -2.4488e+00,
-4.4468e+00, 1.9466e-01],
[-4.6655e-01, -1.8256e+00, -7.4018e-01, ..., -1.9372e+00,
-3.3476e+00, -2.4027e+00],
[ 3.7442e-01, -5.4082e-01, -7.7884e-01, ..., 5.3141e-01,
-3.5776e-01, -6.6652e-01]],
[[-6.6548e+00, -4.3987e+00, -6.6112e+00, ..., -1.6717e+00,
-2.1840e+00, -1.1254e+00],
[-9.3471e+00, -3.3828e+00, -1.6174e+00, ..., -5.4396e-01,
-1.4469e+00, -3.5602e-01],
[-2.0166e+00, 7.9968e-01, -2.5402e+00, ..., -1.5248e+00,
-1.4893e+00, -8.2075e-01],
...,
[-2.9338e+00, -1.9319e+00, -2.5433e+00, ..., -5.8826e+00,
-4.6265e+00, -2.8899e+00],
[-3.6548e+00, -1.4577e+00, -2.2331e+00, ..., -5.1323e+00,
-5.9394e+00, -3.6454e+00],
[-1.2519e+00, -9.1910e-01, -1.6805e+00, ..., -1.7623e+00,
-5.4161e+00, -3.9326e+00]],
[[ 2.3706e+00, 3.9128e+00, 3.0213e+00, ..., 2.7807e+00,
1.2990e+00, 2.8889e+00],
[ 8.3053e-01, 1.7862e+00, -6.6182e-01, ..., 1.8357e+00,
-2.8284e-02, 2.5498e+00],
[-3.9588e+00, -2.4075e+00, -1.8140e+00, ..., 1.0169e+00,
1.0638e+00, 2.0168e+00],
...,
[ 9.8270e-01, 1.2391e+00, 5.5075e-01, ..., -3.1827e+00,
-5.2457e+00, -1.8557e+00],
[ 6.5391e-01, 1.7875e+00, 9.8190e-01, ..., -1.0021e+00,
-4.5873e+00, -1.7492e+00],
[ 1.5716e+00, 1.5374e+00, 1.6830e+00, ..., -6.0304e-02,
-2.3993e+00, -1.6626e+00]],
...,
[[-2.4978e+00, -4.5856e+00, -2.1652e+00, ..., -3.2107e+00,
-2.3190e+00, -4.0366e+00],
[-1.0197e+00, -5.0064e+00, -6.2494e+00, ..., -4.6249e+00,
-3.3541e+00, -4.4212e+00],
[-2.0746e+00, -6.8178e+00, -9.4748e+00, ..., -1.7609e+00,
-1.8058e+00, -1.5485e+00],
...,
[-1.5933e+00, -4.1093e+00, -4.4348e+00, ..., -7.0753e+00,
-9.6607e-02, -2.1018e+00],
[-1.4982e+00, -4.6439e+00, -5.3863e+00, ..., -7.2302e+00,
7.2881e-01, -1.1116e+00],
[-1.5862e+00, -4.8682e+00, -4.1838e+00, ..., -1.9832e+00,
-2.0078e-01, -1.0044e+00]],
[[-9.0546e-01, -1.8552e+00, -3.5804e+00, ..., 8.8912e-01,
1.0613e+00, 1.2120e+00],
[-3.5609e+00, -1.2809e+00, -6.9381e-02, ..., 1.5905e+00,
2.3359e+00, 2.5848e+00],
[-1.0788e+00, 5.7451e-01, 9.5363e-01, ..., 5.6774e-01,
1.6906e+00, 2.1520e+00],
...,
[ 7.7068e-01, -3.4599e-02, 2.2972e-02, ..., -6.5409e-01,
-9.3358e-01, -1.4398e+00],
[ 1.5084e+00, 1.3946e+00, 1.1525e+00, ..., -2.7978e+00,
1.1904e-01, -1.2573e+00],
[ 6.3700e-01, 9.0116e-01, 1.0693e+00, ..., -1.8088e+00,
-1.0141e-01, 4.2475e-02]],
[[-9.4854e-01, -6.9243e-01, -2.0392e+00, ..., -2.4641e-01,
-1.3061e-01, 8.5339e-01],
[-3.1672e+00, -1.9727e+00, -1.5625e+00, ..., 2.4269e-01,
-1.4800e-01, 2.9064e-01],
[ 1.1934e-01, 2.7035e+00, -1.8384e-01, ..., 6.6304e-01,
2.3999e-02, 3.0770e-01],
...,
[-1.7485e+00, -1.6363e+00, -1.4889e+00, ..., -6.6826e+00,
-1.8897e+00, 2.3770e+00],
[-1.4662e+00, -1.5912e+00, -1.4721e+00, ..., -7.8052e+00,
-4.7098e+00, 9.3679e-01],
[-1.5750e-01, 7.4272e-02, -4.4567e-01, ..., -2.6387e+00,
-1.9779e+00, 2.8491e-01]]],
[[[ 8.5406e-01, 1.5075e+00, 1.2026e+00, ..., 1.4322e+00,
1.1080e+00, 1.0050e+00],
[-1.0231e+00, -1.9672e+00, -1.1991e+00, ..., -4.4315e-01,
-6.2818e-01, -1.6851e+00],
[-1.4346e-01, -7.2310e-01, 2.0000e-01, ..., -1.1360e+00,
-2.5507e-01, 3.9497e-01],
...,
[ 4.6682e-01, -1.1864e-01, 2.7278e-01, ..., 4.7397e-01,
1.9597e-01, 6.5294e-01],
[ 3.1214e-01, -6.8377e-01, -1.4844e+00, ..., 7.9564e-01,
5.9457e-01, -7.4647e-02],
[ 5.1459e-01, 3.0172e-01, -1.1986e-01, ..., 3.7449e-01,
-8.4970e-02, 6.6386e-01]],
[[-3.6786e+00, -8.7939e-01, -1.6379e+00, ..., -1.5641e+00,
-2.4626e+00, -1.2308e+00],
[-3.8413e+00, -2.8426e+00, -1.9017e+00, ..., -1.4825e+00,
-3.1206e+00, -9.2315e-01],
[-2.4334e+00, -1.7777e+00, -2.1961e+00, ..., -2.7510e+00,
-3.0702e+00, -9.5707e-01],
...,
[-9.5574e-01, -9.8604e-01, -2.6084e+00, ..., -1.5095e+00,
-1.3525e+00, -2.2060e-01],
[-1.8217e+00, -2.4282e-01, -2.1305e+00, ..., -4.6773e-01,
-1.1713e+00, -1.3402e-01],
[ 2.7820e-01, 9.4613e-02, -2.8652e-01, ..., 2.5255e-01,
5.2373e-02, 4.2628e-01]],
[[ 2.8314e+00, 4.2032e+00, 3.3719e+00, ..., 2.9229e+00,
1.0473e+00, 3.0424e+00],
[ 1.1780e-01, 1.1157e+00, 2.2672e+00, ..., 1.7001e+00,
-9.3683e-01, 2.6389e+00],
[ 8.7402e-01, 1.6011e+00, 1.6139e+00, ..., 1.9948e+00,
9.1656e-01, 2.1427e+00],
...,
[ 4.0381e-01, 5.5614e-01, 1.4053e+00, ..., -3.6059e-01,
2.4641e-01, 1.4670e+00],
[ 3.7114e-02, 1.3358e+00, 1.8672e+00, ..., 6.3234e-02,
1.5003e-01, 1.1634e+00],
[ 1.3007e+00, 8.3223e-01, 2.0598e+00, ..., 3.6395e-01,
4.3273e-01, 1.1665e+00]],
...,
[[-1.7150e+00, -7.0302e+00, -5.3889e+00, ..., -4.0929e+00,
-3.2775e+00, -5.6259e+00],
[-2.3069e+00, -6.8643e+00, -4.1705e+00, ..., -3.3740e+00,
-2.6921e+00, -4.4184e+00],
[-1.0820e+00, -4.6432e+00, -3.8536e+00, ..., -1.8590e+00,
-2.4962e+00, -2.5641e+00],
...,
[-2.6324e+00, -5.9264e+00, -4.3959e+00, ..., -3.5390e+00,
-3.3107e+00, -3.5030e+00],
[-2.8859e+00, -7.3534e+00, -4.9144e+00, ..., -4.9564e+00,
-4.1071e+00, -4.7495e+00],
[-2.8418e+00, -7.5298e+00, -4.5748e+00, ..., -4.0964e+00,
-3.0409e+00, -4.0704e+00]],
[[-1.4989e-01, 2.2745e-01, 5.1248e-01, ..., 6.3573e-01,
8.5494e-01, 1.3775e+00],
[ 1.6062e+00, 1.0936e+00, 1.8368e+00, ..., 1.8699e+00,
2.2656e+00, 2.9736e+00],
[ 1.0417e+00, 2.0077e-01, 1.8670e-01, ..., 8.1041e-01,
2.1699e+00, 2.6288e+00],
...,
[ 1.0279e+00, 4.3974e-02, -1.0563e+00, ..., 1.0254e+00,
7.9672e-01, 1.2603e+00],
[ 1.5350e+00, 1.0177e+00, 3.9129e-01, ..., 1.4311e+00,
6.4305e-01, 2.0264e+00],
[ 1.1111e+00, 7.5422e-01, 1.1673e+00, ..., 1.2872e+00,
1.2703e+00, 1.3810e+00]],
[[-5.4438e-01, 2.4528e-01, 1.6001e-01, ..., 5.8322e-01,
5.3275e-01, 1.3583e+00],
[-1.3577e+00, -1.8293e+00, -1.6680e+00, ..., -9.0734e-01,
-7.9100e-01, 1.4379e-01],
[-2.0938e+00, -7.6432e-01, -3.1864e-01, ..., -6.1336e-01,
-4.4761e-01, 8.2688e-01],
...,
[-1.4791e+00, -1.9443e+00, -1.5028e+00, ..., -7.4083e-01,
-1.0040e+00, -7.6510e-01],
[-1.2347e+00, -1.8643e+00, -1.9540e+00, ..., -6.4765e-01,
-1.1705e+00, -9.3757e-02],
[ 7.9971e-02, 2.6965e-02, -1.3813e-01, ..., -1.3186e-01,
-7.4583e-02, 7.4449e-01]]],
[[[ 1.0569e+00, 1.3799e+00, 1.1450e+00, ..., 2.7445e-01,
3.8697e-01, 2.4169e-01],
[-7.9236e-01, -1.4975e+00, -7.8584e-01, ..., 2.8428e-01,
3.0621e-01, -1.2117e+00],
[-6.6806e-01, -9.3809e-01, -9.2074e-01, ..., -3.1188e-01,
1.0119e+00, 6.1206e-01],
...,
[-2.1035e+00, -9.3159e-02, 1.3158e-01, ..., -3.0876e-01,
4.6467e-01, -1.8718e-01],
[ 5.6998e-01, -5.7054e-01, 4.2259e-01, ..., -1.3571e+00,
-5.0638e-01, 4.0985e-01],
[-3.2200e-03, 5.4271e-01, 2.4958e-02, ..., 1.0492e-01,
-9.8232e-01, -1.0810e+00]],
[[-3.2599e+00, -2.1267e+00, -2.3709e+00, ..., -1.6969e+00,
-2.1414e+00, -1.0216e+00],
[-3.1567e+00, -3.8516e+00, -1.7598e+00, ..., -4.1253e-01,
-1.2858e+00, -2.9247e-01],
[-2.4609e+00, -2.9946e+00, -8.8421e-01, ..., -1.3307e+00,
-1.4550e+00, -7.5391e-01],
...,
[-1.7196e+00, -2.5806e+00, -3.8239e+00, ..., -3.7840e+00,
-1.9723e+00, -8.9001e-01],
[-2.3068e+00, -1.5232e+00, -1.1601e+00, ..., -3.5055e+00,
-3.3673e+00, -1.8757e+00],
[-7.2609e-01, -1.1957e+00, -1.1702e+00, ..., -3.3056e+00,
-4.1345e+00, -2.0285e+00]],
[[ 2.3592e+00, 2.3546e+00, 1.9618e+00, ..., 2.9302e+00,
1.3374e+00, 2.8763e+00],
[ 2.4565e-01, 2.0280e-01, 2.0668e+00, ..., 1.8908e+00,
-6.1400e-02, 2.4711e+00],
[ 3.7021e-01, 7.3688e-01, 1.1733e+00, ..., 1.0273e+00,
6.1926e-01, 1.8232e+00],
...,
[-3.3160e+00, -2.6594e+00, -5.3520e+00, ..., -1.5412e+00,
-1.0507e+00, -2.0775e+00],
[-1.3963e+00, -8.6668e-01, -1.0131e+00, ..., -3.9498e+00,
-4.1440e+00, -4.0333e+00],
[ 7.3004e-01, 9.0157e-01, 1.2334e+00, ..., -2.3878e+00,
-2.6916e+00, -1.1655e+00]],
...,
[[-2.8739e+00, -8.3519e+00, -6.8689e+00, ..., -3.4934e+00,
-2.3623e+00, -3.9249e+00],
[-3.7455e+00, -1.0261e+01, -6.3423e+00, ..., -5.8362e+00,
-3.6959e+00, -4.2624e+00],
[-2.6924e+00, -7.3608e+00, -3.4054e+00, ..., -3.1916e+00,
-2.2857e+00, -1.5949e+00],
...,
[-1.8680e+00, -8.6587e+00, -3.2675e+00, ..., 1.0585e-01,
-1.0354e+00, -1.9170e+00],
[-3.7607e+00, -8.5146e+00, -4.3899e+00, ..., -1.4733e+00,
-3.8193e+00, -9.5456e+00],
[-2.8238e+00, -7.3762e+00, -3.2147e+00, ..., -2.4744e+00,
-3.4135e+00, -5.6363e+00]],
[[-4.2106e-01, 6.3170e-01, 8.6282e-01, ..., 1.3719e+00,
1.0674e+00, 1.2538e+00],
[ 8.1303e-01, 6.2394e-01, 1.0964e+00, ..., 2.0284e+00,
2.3360e+00, 2.5559e+00],
[ 2.6867e-01, 1.2833e-01, 1.1963e+00, ..., 1.4555e+00,
1.6365e+00, 1.9546e+00],
...,
[ 1.4652e+00, -1.5506e-01, 7.6051e-02, ..., -3.1316e+00,
-3.3416e+00, -1.6145e-01],
[ 1.5320e+00, 8.4171e-01, 3.6230e-01, ..., -6.1186e-01,
-6.5409e-01, 1.1118e+00],
[ 8.3947e-01, 3.5623e-01, 1.1186e+00, ..., -1.0795e-01,
-7.8119e-01, 7.3367e-01]],
[[-2.7047e-01, -4.3834e-01, -3.4852e-01, ..., -1.2876e-01,
-1.1220e-01, 9.0163e-01],
[-7.8173e-01, -2.4701e+00, -1.0058e+00, ..., 1.5636e-01,
-8.4935e-02, 3.7019e-01],
[-1.7565e+00, -2.0909e+00, -5.9651e-01, ..., 4.2290e-01,
1.7237e-01, 2.4138e-01],
...,
[-1.8313e+00, -4.5854e+00, -2.1212e+00, ..., -1.2569e+00,
-6.0849e-02, 7.4854e-01],
[-1.9263e+00, -2.1550e+00, -1.1068e+00, ..., -1.1376e+00,
-6.1067e-01, -4.5917e-01],
[-7.2504e-01, -1.0984e+00, -6.3972e-01, ..., -7.2460e-01,
-1.8857e+00, -2.0535e+00]]],
[[[ 1.0107e+00, 1.2001e+00, 4.9605e-01, ..., 5.6404e-01,
9.2633e-01, 4.4767e-01],
[-1.2072e+00, -1.5746e+00, -7.2806e-01, ..., -3.5549e-01,
-4.9479e-01, -5.7891e-01],
[-6.0158e-01, -1.7040e+00, -1.8528e-01, ..., -1.3255e+00,
-1.0609e+00, -3.0864e-01],
...,
[-1.6200e+00, 1.6670e+00, 1.9909e+00, ..., 8.5099e-01,
4.9090e-01, -3.8433e-01],
[-2.4528e-01, -1.8340e+00, -6.2940e-01, ..., 5.4229e-01,
1.4506e+00, 3.8327e-01],
[ 3.0535e-01, 2.3211e-01, -5.7882e-01, ..., -3.4692e-01,
-7.0018e-01, 2.7975e-01]],
[[-4.1526e+00, -2.7304e+00, -3.7285e+00, ..., -1.5755e+00,
-1.9808e+00, -1.6669e+00],
[-3.7383e+00, -3.4767e+00, -3.6605e+00, ..., -2.2059e+00,
-2.0698e+00, -1.9635e+00],
[-3.0643e+00, -2.4584e+00, -7.2930e-01, ..., -3.5066e+00,
-3.3359e+00, -1.6018e+00],
...,
[-2.4971e-01, -1.6243e+00, -1.5410e+00, ..., -1.3042e+00,
-8.1439e-01, -4.6382e-01],
[-3.9823e+00, -4.0381e+00, -2.2322e+00, ..., -2.0624e-01,
-5.6520e-01, -1.5299e+00],
[-7.6215e-01, -9.5959e-01, -7.5546e-01, ..., -5.4443e-01,
-5.7007e-01, -4.7661e-01]],
[[ 2.8590e+00, 3.3310e+00, 3.1643e+00, ..., 1.4680e+00,
9.2295e-01, 2.5614e+00],
[ 1.0379e+00, 1.9414e+00, 3.5510e+00, ..., 1.3512e+00,
1.0814e+00, 2.9559e+00],
[ 1.5058e+00, 1.3193e+00, 6.6220e-01, ..., 1.3845e+00,
1.3760e+00, 1.3667e+00],
...,
[-4.2610e+00, -2.9067e+00, -4.9275e+00, ..., -1.6099e+00,
-1.1758e+00, 4.0847e-01],
[-1.4718e+00, -4.1923e-01, -8.5173e-02, ..., 3.6198e-01,
1.5426e+00, 1.4414e+00],
[ 1.7512e+00, 1.6713e+00, 2.0221e+00, ..., 4.9810e-01,
1.5318e+00, 1.5750e+00]],
...,
[[-3.0503e+00, -8.6327e+00, -5.8126e+00, ..., -5.6778e+00,
-3.6245e+00, -4.2253e+00],
[-3.4650e+00, -9.8946e+00, -5.6327e+00, ..., -6.1147e+00,
-4.4109e+00, -3.4460e+00],
[-2.1017e+00, -6.3378e+00, -3.1394e+00, ..., -2.9006e+00,
-1.5570e+00, -1.1437e+00],
...,
[-2.2277e+00, -1.1053e+01, -6.0596e+00, ..., -5.2198e+00,
-2.4825e+00, -3.2282e+00],
[-3.0168e+00, -8.4506e+00, -5.8009e+00, ..., -6.5733e+00,
-4.8060e+00, -4.7078e+00],
[-2.3392e+00, -6.9776e+00, -5.5437e+00, ..., -3.7476e+00,
-2.5318e+00, -2.5447e+00]],
[[-8.3096e-01, -2.1536e-01, 9.3726e-02, ..., 1.9540e+00,
1.6871e+00, 1.2512e+00],
[ 1.1592e+00, -1.8799e-01, -6.7708e-01, ..., 1.7119e+00,
9.7142e-01, 1.7665e+00],
[ 6.3737e-01, -4.6130e-01, 6.5430e-01, ..., 2.0295e-01,
-1.9402e+00, 3.5271e-01],
...,
[ 1.4874e+00, -2.2798e+00, -1.1209e+00, ..., 1.2841e+00,
1.1925e+00, 1.1719e+00],
[ 8.2520e-01, -1.1550e+00, -7.8138e-01, ..., 1.8077e+00,
8.0290e-01, 1.7963e+00],
[ 1.5206e+00, 1.6194e+00, 1.5952e+00, ..., 8.7315e-01,
7.0933e-01, 1.5352e+00]],
[[-8.2308e-01, -7.3397e-01, -4.7391e-01, ..., -2.9607e-01,
-5.8003e-01, 1.8624e-03],
[-1.7661e+00, -2.0519e+00, -8.0786e-01, ..., -1.4800e+00,
-1.6044e+00, -2.0322e+00],
[-2.0095e+00, -1.4359e+00, 4.3083e-01, ..., -1.4846e+00,
-1.8530e+00, -1.6266e+00],
...,
[-7.1453e-01, -2.7689e+00, -1.2123e+00, ..., -1.3649e+00,
-1.3885e+00, -1.7988e+00],
[-2.0789e+00, -3.2478e+00, -1.6739e+00, ..., -1.2017e+00,
-1.0279e+00, -6.9844e-01],
[-6.6652e-01, 4.2853e-01, 1.0336e-01, ..., -2.8128e-01,
-6.5141e-01, 6.4421e-02]]],
[[[ 7.1315e-01, 1.0567e+00, 7.0256e-01, ..., 3.4578e-01,
3.3025e-01, 2.1434e-01],
[-3.2858e-01, -1.4210e+00, -5.9580e-01, ..., 3.8444e-01,
3.9637e-01, -1.1747e+00],
[ 1.5870e-01, -7.9418e-01, -8.7347e-02, ..., -4.4869e-01,
1.0617e+00, 7.0907e-01],
...,
[-1.1089e-01, -4.5628e-01, -1.7083e-01, ..., -1.2067e+00,
2.1876e-01, 3.2255e-01],
[-4.9111e-01, -1.9090e+00, -6.7556e-01, ..., -1.7863e-01,
9.6754e-01, -8.0261e-03],
[ 4.1866e-01, -5.1540e-01, -7.3436e-01, ..., -3.0408e-01,
3.8782e-01, 5.7624e-01]],
[[-3.1055e+00, -1.5159e+00, -3.0744e+00, ..., -1.6892e+00,
-2.1084e+00, -1.0031e+00],
[-4.2928e+00, -1.8040e+00, -2.3212e+00, ..., -3.8954e-01,
-1.1804e+00, -2.5834e-01],
[-4.1897e+00, -2.3458e+00, -2.5397e+00, ..., -1.3931e+00,
-1.3003e+00, -6.9987e-01],
...,
[-3.5348e+00, -2.3622e+00, -3.5000e+00, ..., -5.3595e+00,
-3.6241e+00, -7.9916e-01],
[-3.7734e+00, -1.3680e+00, -2.2889e+00, ..., -7.4797e-01,
-4.8980e-01, 2.2647e-01],
[-1.2479e+00, -8.7499e-01, -1.7588e+00, ..., -3.9620e-01,
-7.3205e-01, -5.4232e-01]],
[[ 1.7544e+00, 2.7794e+00, 2.4087e+00, ..., 2.7819e+00,
1.2751e+00, 2.7965e+00],
[ 1.9650e-01, 1.1402e+00, 1.4784e+00, ..., 1.8013e+00,
-2.2305e-01, 2.3588e+00],
[ 7.6289e-01, 1.2029e+00, 7.0861e-01, ..., 1.1405e+00,
5.5644e-01, 1.7802e+00],
...,
[ 1.4851e+00, 2.7588e+00, 2.9105e+00, ..., -3.0030e+00,
-1.8939e+00, 1.3876e+00],
[ 6.8266e-01, 1.8575e+00, 9.4291e-01, ..., 1.1949e+00,
8.2620e-01, 1.7728e+00],
[ 1.5471e+00, 1.5403e+00, 1.6516e+00, ..., 1.4632e+00,
1.0983e+00, 1.9982e+00]],
...,
[[-1.4495e+00, -4.6487e+00, -3.9068e+00, ..., -3.2965e+00,
-2.2308e+00, -3.8008e+00],
[-1.8020e+00, -3.7447e+00, -4.0348e+00, ..., -5.4666e+00,
-3.7194e+00, -4.2067e+00],
[-9.7185e-01, -3.0221e+00, -3.2783e+00, ..., -2.5754e+00,
-2.2687e+00, -1.4541e+00],
...,
[-1.6391e+00, -5.2240e+00, -6.4422e+00, ..., -6.1318e+00,
-3.3762e+00, -1.3106e+00],
[-1.4108e+00, -4.3609e+00, -5.2660e+00, ..., -6.9874e+00,
-5.5588e+00, -4.9545e+00],
[-1.5263e+00, -4.7428e+00, -4.1139e+00, ..., -3.5285e+00,
-2.2371e+00, -2.7423e+00]],
[[ 7.5545e-01, 5.7518e-01, -1.1480e-01, ..., 9.0033e-01,
1.0813e+00, 1.2481e+00],
[ 1.2293e+00, 9.7958e-01, 5.3068e-01, ..., 1.5075e+00,
2.3153e+00, 2.5286e+00],
[ 6.3056e-01, 5.2417e-01, 1.0998e-01, ..., 1.0485e+00,
1.6359e+00, 1.8698e+00],
...,
[ 9.0856e-01, 8.2912e-01, 1.5764e+00, ..., -1.7594e+00,
4.8615e-01, 1.6723e+00],
[ 1.4603e+00, 1.4734e+00, 1.1530e+00, ..., 2.0174e+00,
1.7300e+00, 2.5352e+00],
[ 6.0492e-01, 9.2214e-01, 1.0195e+00, ..., 1.0711e+00,
1.3314e+00, 1.3607e+00]],
[[-2.7851e-01, -1.4904e-01, -9.9329e-01, ..., -2.3323e-01,
-1.2328e-01, 9.1295e-01],
[-1.4181e+00, -1.2906e+00, -2.0388e+00, ..., 1.2833e-01,
-3.1431e-02, 4.2444e-01],
[-1.7195e+00, -7.1441e-01, -5.8659e-01, ..., 2.0087e-01,
1.9286e-01, 2.4152e-01],
...,
[-2.0331e+00, -1.8942e+00, -2.0816e+00, ..., -3.6636e+00,
-2.5968e+00, -1.1405e+00],
[-1.5280e+00, -1.5462e+00, -1.5298e+00, ..., -9.5630e-01,
-5.9919e-01, 7.7135e-01],
[-9.7747e-02, 9.9089e-02, -4.4950e-01, ..., 1.8416e-01,
1.4239e-01, 3.4965e-01]]]], device='cuda:0'),
tensor([[[[ 5.8039e-01, 8.2211e-01, 5.7176e-01, ..., 3.0774e-01,
4.7365e-01, 5.5159e-02],
[-2.1186e-01, -8.5235e-01, -5.4950e-01, ..., 1.0425e-01,
1.3588e-01, -3.3301e-01],
[-1.5542e-01, -8.4377e-02, 1.8856e-01, ..., -3.4091e-01,
2.7153e-02, 6.4546e-01],
...,
[ 3.4381e-01, -2.9484e-01, 1.5484e+00, ..., 4.0884e-01,
7.4823e-01, 6.9183e-01],
[-6.4433e-02, -3.3433e-01, 1.2115e+00, ..., 9.4968e-01,
1.6050e+00, 1.8905e-01],
[ 8.4724e-01, -1.7141e-01, 6.4094e-01, ..., 3.4696e-01,
2.5066e-01, 7.1793e-01]],
[[-2.3939e+00, -5.7865e-01, -1.8828e+00, ..., -1.5581e+00,
-2.1838e+00, -1.4966e+00],
[-2.4013e+00, -5.1244e-01, -1.5103e+00, ..., -1.3667e+00,
-2.3815e+00, -1.6395e+00],
[-2.3170e+00, -1.3710e+00, -1.8436e+00, ..., -2.4249e+00,
-2.3279e+00, -1.0861e+00],
...,
[-7.3975e-01, -2.2313e-01, -2.4587e-01, ..., -7.7220e-01,
-4.3577e-01, 6.2818e-01],
[-2.4623e+00, 2.7699e-01, 6.5247e-01, ..., 2.0192e-01,
-2.0563e-01, 5.7480e-01],
[-1.4737e+00, -1.0915e+00, -8.9225e-01, ..., -5.3882e-01,
-8.4509e-01, -2.8106e-01]],
[[ 1.9658e+00, 2.4875e+00, 1.5949e+00, ..., 1.6080e+00,
7.4841e-01, 2.0526e+00],
[-2.0496e-01, 3.3175e-01, 5.2326e-01, ..., 7.5285e-01,
2.8315e-01, 2.1961e+00],
[ 1.7562e-02, 2.7586e-01, 1.9098e-01, ..., -3.0868e-01,
7.5248e-01, 1.9586e+00],
...,
[ 7.5686e-01, 9.9821e-01, 4.1339e-01, ..., 2.0175e-01,
2.0085e-02, 1.7378e+00],
[ 1.0268e+00, 2.3458e+00, 1.6704e+00, ..., 8.3727e-01,
5.9549e-01, 1.8099e+00],
[ 2.3120e+00, 1.9431e+00, 2.6155e+00, ..., 1.7994e+00,
1.0147e+00, 1.7951e+00]],
...,
[[-1.9878e+00, -4.8367e+00, -3.1034e+00, ..., -1.9401e+00,
-1.7299e+00, -3.0464e+00],
[-2.4464e+00, -5.2637e+00, -2.8945e+00, ..., -2.5209e+00,
-2.2848e+00, -2.9265e+00],
[-1.5362e+00, -5.4159e+00, -3.2151e+00, ..., -3.5638e+00,
-2.5026e+00, -2.2256e+00],
...,
[-2.5287e+00, -4.5016e+00, -2.2565e+00, ..., -2.7086e+00,
-1.9612e+00, -2.8609e+00],
[-2.4538e+00, -5.5850e+00, -2.9984e+00, ..., -4.2910e+00,
-3.0393e+00, -5.0908e+00],
[-2.5115e+00, -5.7879e+00, -2.2388e+00, ..., -2.7568e+00,
-1.4591e+00, -3.2228e+00]],
[[ 9.1190e-01, 1.6323e+00, 7.8526e-01, ..., 1.2907e+00,
1.5331e+00, 1.4022e+00],
[ 1.1941e+00, 1.0525e+00, 8.0467e-01, ..., 1.3837e+00,
7.9246e-01, 1.9086e+00],
[ 6.3450e-01, -5.6553e-02, -1.3224e-01, ..., 1.0915e+00,
1.0275e-01, 1.4127e+00],
...,
[ 1.5053e+00, 1.7037e-01, 9.4498e-01, ..., 1.4118e+00,
1.7083e+00, 1.4772e+00],
[ 2.1004e+00, 1.3204e+00, 2.4341e+00, ..., 1.9716e+00,
1.7418e+00, 2.4443e+00],
[ 1.2201e+00, 8.6256e-01, 1.2773e+00, ..., 1.1773e+00,
1.4153e+00, 1.5608e+00]],
[[-9.5446e-02, -6.2475e-01, -1.1307e+00, ..., -4.8283e-01,
-8.5683e-01, -8.3863e-02],
[-8.4930e-01, -2.5220e+00, -2.6504e+00, ..., -8.8860e-01,
-1.1211e+00, -7.1221e-01],
[-1.4401e+00, -1.9006e+00, -1.3891e+00, ..., -5.6549e-01,
-4.6171e-01, -2.7486e-01],
...,
[-1.7261e+00, -1.4612e+00, -1.7484e-01, ..., -3.5671e-01,
-2.9840e-01, -4.0133e-01],
[-1.7617e+00, -9.9525e-01, -1.0841e-01, ..., -1.9747e-01,
-2.4916e-01, 4.5433e-01],
[-2.2842e-01, 5.4315e-01, -1.2874e-01, ..., 2.2559e-01,
-3.9096e-02, 5.8984e-01]]],
[[[ 5.8142e-01, 1.5339e-01, 9.1507e-01, ..., 3.7477e+00,
2.0575e+00, 1.2468e-01],
[-8.1262e-01, -5.2296e+00, -5.7071e+00, ..., -6.6742e+00,
-4.4347e-01, -4.9211e-01],
[ 1.1958e-01, -2.1720e+00, -2.6279e+00, ..., -1.3107e+00,
1.4101e+00, 3.4754e-01],
...,
[ 2.9797e-01, -5.5217e-01, -3.3196e-01, ..., 2.6419e+00,
8.4415e-01, 6.0817e-01],
[-6.1656e-01, -1.8114e+00, -1.5499e+00, ..., 9.4495e-01,
1.1313e+00, -7.7972e-02],
[-1.7525e-01, -1.6713e+00, -2.1086e+00, ..., -3.2984e+00,
-5.6909e-01, -1.0331e-01]],
[[-4.2295e+00, 2.5255e-01, -2.4977e-01, ..., -3.2099e+00,
-4.5698e+00, -1.2529e+00],
[-4.1490e+00, -1.3471e+00, -3.9849e+00, ..., -5.3553e+00,
-5.7536e+00, -1.0052e+00],
[-2.1185e+00, -4.6738e-01, -1.9395e+00, ..., -3.1278e+00,
-4.9613e+00, -1.2172e+00],
...,
[-1.2750e+00, -1.2324e+00, -1.9814e+00, ..., -6.1366e+00,
-2.4653e+00, -1.8253e-01],
[-2.8851e+00, -1.5916e+00, -2.6769e+00, ..., -5.4724e+00,
-3.7325e+00, -1.3229e+00],
[-2.2148e+00, -1.9389e+00, -2.6479e+00, ..., -4.6158e+00,
-3.8680e+00, -1.2468e+00]],
[[ 1.7206e+00, 5.1254e-01, -1.7702e+00, ..., -2.2020e+00,
-1.1508e+00, 1.4501e+00],
[ 8.9992e-01, -2.4900e+00, -2.9019e+00, ..., -2.3503e+00,
2.7272e-01, 1.6689e+00],
[-8.8240e-01, -4.5319e+00, -1.4513e+00, ..., -2.2109e+00,
2.6399e+00, 1.1587e+00],
...,
[ 1.1545e+00, 1.4825e+00, 2.2953e+00, ..., -1.4036e+00,
-1.2050e+00, 2.0249e+00],
[ 7.9847e-01, 1.9546e+00, 2.2499e+00, ..., -1.5542e+00,
-1.2540e+00, 1.2126e+00],
[ 1.8579e+00, 1.5601e+00, 2.6567e+00, ..., -4.4026e-01,
1.0979e+00, 1.5600e+00]],
...,
[[-2.9773e+00, -5.7541e+00, -3.6542e+00, ..., -8.2545e+00,
-1.0385e+01, -7.8215e+00],
[-5.4101e+00, -9.5785e+00, -4.6329e+00, ..., -8.6645e-01,
-1.1645e+01, -8.5515e+00],
[-2.8888e+00, -5.0015e+00, -4.3987e+00, ..., -2.1192e+00,
-1.1084e+01, -4.3667e+00],
...,
[-2.4335e+00, -5.5786e+00, -4.0169e+00, ..., -1.9366e+01,
-1.3505e+01, -4.5013e+00],
[-3.0790e+00, -6.3021e+00, -5.2251e+00, ..., -1.1275e+01,
-1.2907e+01, -6.3646e+00],
[-1.7223e+00, -4.5281e+00, -2.6925e+00, ..., -2.2541e-01,
-2.6959e+00, -3.1671e+00]],
[[-7.6423e-01, -1.1357e-01, 7.0605e-01, ..., -4.8522e+00,
-3.9120e-01, 1.2673e+00],
[-6.1584e-02, -4.6790e-01, 8.1344e-01, ..., -3.8567e+00,
-4.4866e-01, 3.1129e+00],
[ 1.6312e+00, -9.5398e-01, -6.4819e+00, ..., -1.6457e-01,
-1.1578e+00, 2.2176e+00],
...,
[ 1.4068e+00, 2.4305e-01, -1.2942e-01, ..., -7.8628e-01,
3.8367e+00, 1.9960e+00],
[ 2.0473e+00, 1.8824e+00, 1.1634e+00, ..., -2.5743e+00,
1.5133e+00, 2.7083e+00],
[ 5.4302e-01, 9.6113e-01, 9.5356e-01, ..., -2.1413e+00,
-5.9946e-02, 1.9990e+00]],
[[-1.6050e+00, 6.7720e-01, -4.5838e-01, ..., 2.6648e+00,
9.3765e-01, -3.2166e-01],
[-1.5632e+00, 9.1847e-01, -1.3304e+00, ..., -2.5066e+00,
-3.4117e+00, -2.6777e+00],
[-2.9720e+00, -2.0475e+00, -2.5007e+00, ..., 2.3782e+00,
-9.7074e-01, -2.8650e+00],
...,
[-2.0524e+00, -1.8028e+00, -1.2936e+00, ..., -2.6766e+00,
-5.7197e+00, -2.3211e-01],
[-1.8600e+00, -2.0474e+00, -1.8051e+00, ..., 8.6684e-02,
-4.9248e+00, -1.3785e+00],
[-8.7965e-01, -8.5702e-01, -1.2201e+00, ..., -9.7472e-01,
-2.9374e+00, -1.4051e+00]]],
[[[ 9.2731e-01, 1.5453e+00, 1.6556e+00, ..., 6.0699e-01,
1.0059e+00, 9.8173e-01],
[-9.8275e-01, -2.5939e+00, -2.7093e+00, ..., 7.3692e-01,
-7.9653e-01, -1.5996e+00],
[-7.6264e-02, -3.0473e-01, 3.3774e-01, ..., -4.0152e-01,
5.2422e-01, 8.0760e-01],
...,
[-2.4042e-01, -1.0121e+00, -2.6064e+00, ..., 6.0995e-01,
1.5396e-01, 6.6866e-01],
[-9.2541e-01, -7.4442e-01, -2.7985e+00, ..., 1.1185e+00,
1.2315e+00, 5.8842e-02],
[ 6.7050e-01, -4.2703e-01, 2.8201e-01, ..., 2.8102e-01,
-1.9079e-01, 6.2368e-01]],
[[-4.1191e+00, -6.5403e-01, -2.3806e+00, ..., -1.5611e+00,
-1.8722e+00, -1.0732e+00],
[-4.2001e+00, -2.4463e+00, -1.7973e+00, ..., -2.0567e+00,
-2.9463e+00, -8.3951e-01],
[-2.6126e+00, -2.4321e-01, -1.6207e+00, ..., -2.4735e+00,
-2.7157e+00, -1.0765e+00],
...,
[-1.3485e-01, -5.5790e-01, -1.1483e+00, ..., -2.4897e-01,
-1.6344e+00, 7.7678e-02],
[-7.4252e-01, 1.2499e-01, 8.6887e-01, ..., 3.0589e-01,
-2.0270e+00, -7.2349e-01],
[-4.8875e-02, 6.7419e-01, 1.2136e+00, ..., -1.2500e+00,
-1.2944e+00, -9.2090e-02]],
[[ 3.4066e+00, 5.0095e+00, 3.3401e+00, ..., 3.0885e+00,
2.1724e+00, 2.9130e+00],
[ 9.9665e-01, 1.6821e+00, 1.0702e+00, ..., 4.7838e-01,
-1.2556e+00, 2.5484e+00],
[ 1.8972e+00, 2.3006e+00, -2.2450e+00, ..., 7.7496e-01,
-2.9501e-01, 1.9932e+00],
...,
[-3.8458e+00, -3.5521e-01, -3.7753e-01, ..., -3.2485e-01,
-6.4228e-01, 1.1772e+00],
[-3.9108e+00, -6.7592e-01, -3.4076e-01, ..., -3.9021e-01,
-1.0587e+00, 5.2123e-01],
[-9.0677e-01, -4.7761e+00, -1.2443e+00, ..., -1.4496e-02,
-1.8617e-01, 1.4315e+00]],
...,
[[-2.6985e+00, -7.9620e+00, -6.0514e+00, ..., -5.6266e+00,
-3.3394e+00, -5.3975e+00],
[-3.2494e+00, -7.4646e+00, -5.3948e+00, ..., -8.4267e+00,
-4.6325e+00, -4.1991e+00],
[-1.3179e+00, -5.2667e+00, -3.4437e+00, ..., -2.9832e+00,
-3.3277e+00, -2.3867e+00],
...,
[-5.8230e-01, -6.2091e+00, -3.0806e+00, ..., -4.6101e-01,
-6.0681e-01, -2.5655e+00],
[-7.4722e-01, -3.2245e+00, -3.0032e+00, ..., -3.1097e+00,
-2.7780e+00, -5.1623e+00],
[-1.3559e+00, -3.3814e+00, -4.9308e+00, ..., -3.8308e+00,
-1.9866e+00, -3.3793e+00]],
[[-2.9123e-01, 3.4170e-01, -1.4489e+00, ..., 1.3400e+00,
1.3051e+00, 1.4934e+00],
[ 1.8135e+00, 1.3398e+00, 1.6796e+00, ..., 1.1142e+00,
2.6876e+00, 2.9568e+00],
[ 1.2613e+00, 6.9977e-01, 1.2486e+00, ..., 1.9489e-01,
1.6834e+00, 2.3126e+00],
...,
[ 2.0539e-01, -1.8602e+00, -1.3728e+00, ..., 4.1840e-01,
6.7086e-01, 1.3041e+00],
[ 1.4655e+00, -1.7386e+00, -1.6968e+00, ..., 2.7064e-01,
3.8979e-01, 1.2725e+00],
[ 1.4440e+00, 1.4532e+00, 9.0900e-01, ..., 1.2279e-01,
1.0108e+00, 1.0800e+00]],
[[-6.4899e-01, -2.6999e-01, -1.9714e-01, ..., 1.2107e-01,
6.2192e-01, 1.3637e+00],
[-1.5674e+00, -2.4427e+00, -1.9624e+00, ..., -4.9491e-01,
-6.3697e-01, 1.8593e-01],
[-2.3667e+00, -6.6557e-01, 1.6922e-02, ..., -9.0415e-01,
6.2292e-02, 7.0727e-01],
...,
[-1.2097e-01, -1.5333e+00, -3.7317e+00, ..., -8.7705e-01,
-6.1383e-01, 2.6766e-01],
[-2.2539e+00, -1.3632e+00, -2.4547e-01, ..., -2.4296e-01,
-7.4261e-01, -8.3692e-02],
[-6.3514e-01, -3.4762e-01, 3.4543e-01, ..., -9.0339e-01,
-1.6552e+00, 3.5548e-01]]],
[[[ 5.9064e-01, 8.4265e-01, 6.8202e-01, ..., 3.5645e-01,
4.3584e-01, -5.6022e-03],
[-2.6626e-01, -8.9101e-01, -5.2909e-01, ..., 1.9025e-01,
1.3879e-01, -2.5774e-01],
[-1.8085e-01, -8.2944e-02, -3.7920e-01, ..., 2.6722e-02,
1.1877e-01, 6.5540e-01],
...,
[ 4.3143e-01, -8.0341e-01, 1.5423e+00, ..., 3.4908e-01,
8.6952e-01, 6.6249e-01],
[-7.8702e-02, -8.3271e-01, 1.9338e+00, ..., 9.4693e-01,
1.5639e+00, 9.5992e-02],
[ 8.0277e-01, -2.4199e-01, 1.2617e+00, ..., 3.8966e-01,
2.4891e-01, 7.5758e-01]],
[[-2.3401e+00, -6.0938e-01, -2.1769e+00, ..., -1.6320e+00,
-2.2160e+00, -1.5208e+00],
[-2.4798e+00, -5.7194e-01, -2.2523e+00, ..., -1.1605e+00,
-2.3126e+00, -1.7119e+00],
[-2.3948e+00, -1.4559e+00, -3.8066e+00, ..., -1.6893e+00,
-2.3794e+00, -1.1969e+00],
...,
[-5.7455e-01, -1.6868e+00, -2.7947e+00, ..., -7.3741e-01,
-2.3173e-01, 6.7637e-01],
[-2.5305e+00, -6.7379e-01, -6.5573e-01, ..., 3.6585e-02,
-8.3567e-02, 6.3261e-01],
[-1.5726e+00, -1.5746e+00, -8.1112e-01, ..., -4.5217e-01,
-6.0597e-01, -1.9240e-01]],
[[ 1.9442e+00, 2.5040e+00, 1.7952e+00, ..., 1.5776e+00,
7.3725e-01, 2.1055e+00],
[-2.2140e-01, 3.1158e-01, 1.1272e+00, ..., 7.3869e-01,
3.6123e-01, 2.3150e+00],
[ 2.6265e-02, 2.1651e-01, 1.5349e+00, ..., -1.6634e-01,
7.0281e-01, 1.9984e+00],
...,
[ 7.2762e-01, 1.7137e+00, -1.4421e+00, ..., 4.0182e-01,
1.4671e-01, 1.6383e+00],
[ 1.0650e+00, 2.9597e+00, 4.2281e-01, ..., 1.0125e+00,
7.3280e-01, 1.7179e+00],
[ 2.4097e+00, 2.1441e+00, 2.2639e+00, ..., 1.9531e+00,
1.0501e+00, 1.7250e+00]],
...,
[[-2.0400e+00, -4.9405e+00, -3.2853e+00, ..., -1.9339e+00,
-1.5877e+00, -2.9539e+00],
[-2.4807e+00, -5.2934e+00, -3.0763e+00, ..., -2.3007e+00,
-2.1072e+00, -2.9784e+00],
[-1.5687e+00, -5.4451e+00, -3.1436e+00, ..., -2.3269e+00,
-2.4036e+00, -2.2680e+00],
...,
[-2.5627e+00, -4.5888e+00, -3.3264e+00, ..., -2.7541e+00,
-2.2015e+00, -2.7678e+00],
[-2.5546e+00, -5.8834e+00, -3.7704e+00, ..., -4.4659e+00,
-3.2325e+00, -4.9797e+00],
[-2.6375e+00, -6.0589e+00, -2.2686e+00, ..., -2.9864e+00,
-1.6271e+00, -3.3772e+00]],
[[ 9.3832e-01, 1.6033e+00, 3.7894e-01, ..., 1.1528e+00,
1.4937e+00, 1.4172e+00],
[ 1.1794e+00, 1.0597e+00, 5.5570e-02, ..., 1.0610e+00,
8.0522e-01, 1.9131e+00],
[ 6.4383e-01, -4.1236e-02, -1.5137e+00, ..., 6.3828e-01,
6.0654e-03, 1.3863e+00],
...,
[ 1.5158e+00, -7.1676e-01, -1.3030e-04, ..., 1.5447e+00,
1.8055e+00, 1.4879e+00],
[ 2.0409e+00, 8.6072e-01, 2.0417e+00, ..., 1.9318e+00,
1.6544e+00, 2.5420e+00],
[ 1.1801e+00, 8.0328e-01, 1.9564e+00, ..., 1.3414e+00,
1.4995e+00, 1.5169e+00]],
[[-8.1120e-02, -6.3814e-01, -1.6224e+00, ..., -4.7062e-01,
-8.8643e-01, -1.2045e-01],
[-8.5132e-01, -2.5570e+00, -3.6388e+00, ..., -1.0753e+00,
-9.6479e-01, -6.3464e-01],
[-1.4255e+00, -1.9162e+00, -2.5698e+00, ..., -2.8290e-01,
-3.9016e-01, -2.7758e-01],
...,
[-1.6897e+00, -2.1699e+00, -2.5221e-01, ..., -4.4349e-01,
-4.2732e-01, -4.7873e-01],
[-1.7227e+00, -1.5563e+00, -6.3953e-02, ..., -3.3477e-01,
-3.7547e-01, 4.2304e-01],
[-2.7276e-01, 4.6258e-01, 2.1640e-01, ..., 2.1390e-01,
4.2219e-02, 6.3198e-01]]],
[[[ 8.7066e-01, 1.4806e+00, 1.2125e+00, ..., 1.5237e+00,
1.1495e+00, 1.0290e+00],
[-8.9879e-01, -1.6989e+00, -1.2321e+00, ..., -3.8011e-01,
-5.8570e-01, -1.6557e+00],
[-1.5754e-01, -7.2097e-01, -3.3303e-01, ..., -2.5734e-01,
6.9026e-01, 7.6634e-01],
...,
[-1.1997e+00, -2.5877e+00, -4.7583e+00, ..., 3.1907e-01,
1.1188e-01, 6.4630e-01],
[-1.0536e+00, -2.3032e+00, -4.5829e+00, ..., 8.9373e-01,
6.7004e-01, -5.1692e-02],
[-1.7170e+00, 1.2211e+00, 1.1110e+00, ..., 6.6655e-01,
-1.5376e-01, 6.2179e-01]],
[[-3.7636e+00, -6.2605e-01, -1.4825e+00, ..., -1.5525e+00,
-2.3924e+00, -1.1767e+00],
[-4.0460e+00, -3.3229e+00, -3.1205e+00, ..., -2.4612e+00,
-3.1581e+00, -9.3550e-01],
[-2.6399e+00, -3.3331e+00, -4.4304e+00, ..., -2.0641e+00,
-2.8414e+00, -1.0969e+00],
...,
[ 1.3827e+00, -2.4621e+00, -7.8941e+00, ..., -1.4721e+00,
-1.4807e+00, -9.6665e-02],
[-9.1161e-01, -1.5483e+00, -4.0852e+00, ..., -6.6433e-01,
-1.1573e+00, -5.4145e-02],
[-4.8700e-02, 1.9662e+00, -1.1738e+00, ..., -1.0313e-01,
-7.7861e-02, 3.9397e-01]],
[[ 2.9226e+00, 4.3011e+00, 3.4051e+00, ..., 2.9504e+00,
1.0683e+00, 3.0788e+00],
[ 2.0437e-01, 1.1907e+00, 2.1490e+00, ..., 1.2932e+00,
-8.0580e-01, 2.7605e+00],
[ 8.9142e-01, 2.4349e+00, 3.0439e+00, ..., 5.5584e-01,
-1.7537e-01, 2.0554e+00],
...,
[-2.0431e+00, -1.4311e+00, -4.3994e+00, ..., 2.8056e-01,
3.1927e-01, 1.5105e+00],
[-6.0941e+00, -2.0506e+00, -3.5516e+00, ..., -2.3255e-01,
2.3680e-01, 1.0863e+00],
[-4.7772e+00, -2.5366e+00, -1.4960e+00, ..., 7.0752e-01,
3.6536e-01, 1.1054e+00]],
...,
[[-1.8425e+00, -7.2109e+00, -5.3486e+00, ..., -4.1700e+00,
-3.3881e+00, -5.6836e+00],
[-2.4566e+00, -7.1305e+00, -4.9289e+00, ..., -3.4536e+00,
-2.7200e+00, -4.3348e+00],
[-1.0076e+00, -5.0312e+00, -4.6092e+00, ..., -3.6097e+00,
-3.0923e+00, -2.4249e+00],
...,
[-2.5653e+00, -9.4881e+00, -8.2690e-01, ..., -3.2335e+00,
-3.4680e+00, -3.4677e+00],
[-2.9904e+00, -1.2005e+01, -2.2337e+00, ..., -5.5586e+00,
-4.3400e+00, -4.7559e+00],
[-1.6523e+00, -8.5141e+00, -8.0426e+00, ..., -3.9114e+00,
-3.0832e+00, -4.0146e+00]],
[[-1.7625e-01, 3.8873e-01, 6.6470e-01, ..., 6.0731e-01,
9.2207e-01, 1.4046e+00],
[ 1.5789e+00, 6.0098e-01, 1.8666e+00, ..., 1.7520e+00,
2.6017e+00, 3.0622e+00],
[ 1.0755e+00, -7.6304e-01, -1.1205e+00, ..., 1.1017e+00,
1.3876e+00, 2.3517e+00],
...,
[ 1.7169e+00, 1.0622e+00, 1.0003e+00, ..., 1.5018e+00,
7.9520e-01, 1.3828e+00],
[-9.8531e-01, -4.9089e+00, -4.6471e-01, ..., 1.1074e+00,
7.0162e-01, 2.0794e+00],
[ 9.3252e-01, -3.6686e-01, -1.2584e+00, ..., 1.1805e+00,
1.2949e+00, 1.3309e+00]],
[[-5.4239e-01, 3.0864e-01, 1.6156e-01, ..., 6.3122e-01,
5.8542e-01, 1.3792e+00],
[-1.2722e+00, -2.1130e+00, -1.6173e+00, ..., -7.7235e-01,
-4.5118e-01, 1.6845e-01],
[-2.1275e+00, -1.3410e+00, -2.2723e-02, ..., 1.3407e-01,
4.4238e-02, 6.7191e-01],
...,
[ 7.1336e-01, -2.8314e+00, -5.7104e+00, ..., -7.8238e-01,
-1.0069e+00, -5.7413e-01],
[ 1.9697e-01, -7.2370e+00, -6.2237e+00, ..., -9.1605e-01,
-1.1647e+00, 1.3143e-01],
[-2.6463e+00, -3.6966e+00, -1.2590e+00, ..., -4.9033e-01,
-2.3870e-01, 7.7816e-01]]]], device='cuda:0'),
tensor([[[[ 1.0707e+00, 1.3432e+00, 1.2544e+00, ..., 1.1989e+00,
9.7513e-01, 4.4470e-01],
[-5.6706e-01, -2.3678e+00, -1.2491e+00, ..., -4.3551e-01,
-1.1948e-01, -1.3822e+00],
[-4.4095e-01, -1.0989e+00, -6.5243e-01, ..., -4.6740e-01,
-2.2959e-02, 6.0847e-01],
...,
[ 5.0465e-01, -8.1873e-02, 1.4725e+00, ..., 3.9499e-01,
1.0160e+00, 7.8295e-01],
[ 5.5397e-02, -7.2466e-01, 4.1083e-01, ..., 9.0782e-01,
1.5234e+00, 1.0207e-01],
[ 8.7372e-01, -2.4980e-01, -4.7969e-01, ..., 3.8442e-01,
2.5854e-01, 7.1871e-01]],
[[-2.8144e+00, 4.0910e-03, -1.3129e+00, ..., -1.3910e+00,
-2.3720e+00, -7.0125e-01],
[-4.0168e+00, -2.7718e+00, -2.8377e+00, ..., -2.5301e+00,
-4.0117e+00, -1.6207e+00],
[-3.0455e+00, -2.4788e+00, -3.5455e+00, ..., -2.8879e+00,
-3.0699e+00, -1.3388e+00],
...,
[-7.6093e-01, -7.6540e-01, -1.1758e+00, ..., -4.2571e-01,
1.5905e-02, 7.6105e-01],
[-2.4807e+00, -5.1660e-01, -6.4587e-01, ..., -2.0466e-02,
-9.1294e-02, 6.3331e-01],
[-1.4754e+00, -1.4503e+00, -1.3068e+00, ..., -4.4288e-01,
-5.8700e-01, -1.8247e-01]],
[[ 2.4580e+00, 2.9527e+00, 2.7917e+00, ..., 1.8160e+00,
6.5422e-01, 2.6334e+00],
[ 8.6564e-01, 6.5730e-01, 2.6957e+00, ..., 1.6151e+00,
5.5968e-01, 2.6785e+00],
[ 1.2431e+00, 1.4626e+00, 2.6535e+00, ..., 1.3935e+00,
1.7268e+00, 2.8508e+00],
...,
[ 8.9694e-01, 1.3747e+00, 1.6192e+00, ..., 9.2085e-01,
5.5063e-01, 1.8545e+00],
[ 1.0593e+00, 2.1723e+00, 2.8325e+00, ..., 1.0203e+00,
7.3757e-01, 1.7003e+00],
[ 2.3980e+00, 1.8612e+00, 2.4628e+00, ..., 1.9904e+00,
1.0713e+00, 1.7259e+00]],
...,
[[-2.6090e+00, -6.8587e+00, -5.7943e+00, ..., -4.4288e+00,
-4.4581e+00, -5.4092e+00],
[-3.3479e+00, -7.5335e+00, -5.3386e+00, ..., -4.9725e+00,
-5.1752e+00, -5.1595e+00],
[-2.2403e+00, -6.1473e+00, -4.4224e+00, ..., -3.7584e+00,
-3.9504e+00, -3.3926e+00],
...,
[-2.4513e+00, -4.7059e+00, -3.1977e+00, ..., -3.1103e+00,
-2.7492e+00, -3.1098e+00],
[-2.4407e+00, -5.4963e+00, -3.9657e+00, ..., -4.2992e+00,
-3.1466e+00, -4.9566e+00],
[-2.6044e+00, -5.7906e+00, -2.6135e+00, ..., -2.9046e+00,
-1.6381e+00, -3.3476e+00]],
[[ 3.4768e-01, 1.1417e+00, 4.3237e-01, ..., 1.1493e+00,
1.4930e+00, 1.6851e+00],
[ 1.2684e+00, 7.9923e-01, 2.0790e-02, ..., 7.2892e-01,
1.2716e+00, 2.7491e+00],
[ 1.3305e+00, 7.7645e-01, -1.3558e-01, ..., 1.5371e+00,
1.0927e+00, 2.0514e+00],
...,
[ 1.5450e+00, -1.5170e-02, 3.6361e-01, ..., 1.9062e+00,
1.8943e+00, 1.4119e+00],
[ 2.0089e+00, 1.1262e+00, 1.7606e+00, ..., 1.9261e+00,
1.6634e+00, 2.5491e+00],
[ 1.1582e+00, 8.2374e-01, 1.4244e+00, ..., 1.3314e+00,
1.4996e+00, 1.5482e+00]],
[[-6.0915e-01, 5.7569e-02, -3.4672e-01, ..., 4.8986e-01,
-9.7725e-02, 9.1526e-01],
[-1.4322e+00, -2.1672e+00, -2.0626e+00, ..., -9.0416e-01,
-7.8281e-01, -3.9909e-02],
[-2.2802e+00, -1.3071e+00, -1.2159e+00, ..., -1.1110e-01,
-4.2413e-01, -1.1468e-01],
...,
[-1.9783e+00, -1.2909e+00, 2.9282e-01, ..., -2.1916e-01,
-4.5422e-01, -5.0257e-01],
[-1.6744e+00, -1.2935e+00, -2.2264e-01, ..., -4.6133e-01,
-4.1438e-01, 4.0669e-01],
[-1.7425e-01, 3.9234e-01, -3.2915e-01, ..., 1.8033e-01,
1.5092e-02, 6.3400e-01]]],
[[[ 5.9753e-01, 8.7945e-01, 5.7619e-01, ..., 3.5732e-01,
4.4507e-01, 3.6509e-02],
[-1.3113e-01, -4.5179e-01, -5.2263e-01, ..., 1.2830e-01,
6.3546e-02, -3.5227e-01],
[-2.8223e-01, -4.0436e-01, -1.0796e-01, ..., 9.2453e-02,
1.7997e-01, 6.7643e-01],
...,
[-1.9665e-01, -8.0277e-01, -1.5917e-01, ..., -4.2400e-02,
7.2716e-01, 8.8447e-01],
[-7.6901e-01, -2.3829e+00, -2.2148e+00, ..., -1.9431e-01,
7.3354e-01, 8.8822e-02],
[ 5.3698e-01, -1.4913e-01, -1.5990e+00, ..., -6.2039e-01,
-5.6170e-01, 2.2289e-01]],
[[-2.3356e+00, -5.9380e-01, -1.8749e+00, ..., -1.6023e+00,
-2.2315e+00, -1.5197e+00],
[-2.7607e+00, -1.7486e+00, -1.4945e+00, ..., -1.3398e+00,
-2.3878e+00, -1.6826e+00],
[-3.0064e+00, -4.7622e+00, -5.0331e+00, ..., -1.8480e+00,
-2.2975e+00, -1.2475e+00],
...,
[-1.4818e+00, -1.8102e+00, -2.3010e+00, ..., -1.7885e+00,
-1.1354e+00, 1.3799e-01],
[-2.2078e+00, -1.0811e+00, 1.0751e+00, ..., -1.3018e+00,
-1.4556e+00, -2.1148e-01],
[-5.4583e-01, -1.0436e+00, -9.1434e-01, ..., -1.6605e+00,
-1.4678e+00, -5.1862e-01]],
[[ 1.9064e+00, 2.5116e+00, 1.6094e+00, ..., 1.6402e+00,
7.6815e-01, 2.0681e+00],
[-7.8897e-02, 4.5758e-01, 9.8476e-01, ..., 8.2874e-01,
2.9148e-01, 2.2213e+00],
[ 1.2643e+00, 1.4710e+00, 1.0706e+00, ..., 9.7619e-02,
7.9337e-01, 2.0161e+00],
...,
[ 9.4103e-01, 1.5042e+00, 1.9175e+00, ..., 1.2864e-01,
-5.0259e-01, 1.3278e+00],
[ 8.7800e-01, 1.6726e+00, 1.2109e+00, ..., 1.5451e-01,
-4.4083e-01, 1.0042e+00],
[ 1.9226e+00, -9.0419e-01, -5.4945e-01, ..., 1.0304e+00,
5.2945e-01, 1.3762e+00]],
...,
[[-2.0231e+00, -4.9111e+00, -3.2028e+00, ..., -2.0699e+00,
-1.7165e+00, -3.0581e+00],
[-2.7577e+00, -5.9531e+00, -3.5647e+00, ..., -2.4636e+00,
-2.2747e+00, -2.9675e+00],
[-9.9276e-01, -3.8882e+00, -2.4573e+00, ..., -2.0937e+00,
-2.2106e+00, -2.0952e+00],
...,
[-1.9211e+00, -5.1967e+00, -3.1680e+00, ..., -2.8664e+00,
-2.7101e+00, -2.8782e+00],
[-2.2995e+00, -6.4627e+00, -3.5522e+00, ..., -4.2311e+00,
-3.5092e+00, -4.3641e+00],
[-2.5315e+00, -6.6888e+00, -4.2480e+00, ..., -3.2469e+00,
-2.0638e+00, -2.8400e+00]],
[[ 9.4975e-01, 1.6278e+00, 7.7989e-01, ..., 1.1111e+00,
1.5360e+00, 1.3938e+00],
[ 1.0214e+00, 7.1119e-01, 7.7471e-01, ..., 9.2506e-01,
7.8665e-01, 1.9244e+00],
[-6.0640e-01, -2.9367e+00, -1.6209e+00, ..., 4.8658e-01,
2.3780e-02, 1.2798e+00],
...,
[ 1.2448e+00, -1.1089e+00, -1.0697e+00, ..., 1.0016e+00,
1.5592e+00, 1.7457e+00],
[ 1.9418e+00, 1.2788e+00, 2.8760e+00, ..., 1.1792e+00,
1.2069e+00, 2.2152e+00],
[ 1.0348e+00, 1.1830e-01, -7.8102e-01, ..., 1.7206e-01,
7.7241e-01, 1.2009e+00]],
[[-8.2355e-02, -5.8560e-01, -1.1848e+00, ..., -5.4146e-01,
-8.8254e-01, -1.0692e-01],
[-8.7938e-01, -2.3313e+00, -2.0167e+00, ..., -1.2191e+00,
-1.1272e+00, -6.9668e-01],
[-1.8725e+00, -1.9613e+00, -2.5174e-01, ..., -8.7635e-02,
-2.1590e-01, -2.6763e-01],
...,
[-2.5104e+00, -1.4618e+00, -7.0665e-01, ..., -3.5019e-01,
-2.7160e-01, 1.2564e-01],
[-2.5143e+00, -2.1266e+00, 4.0822e-01, ..., -4.4360e-01,
-5.3163e-01, 3.3225e-01],
[-7.8822e-02, -3.7733e-01, 4.9209e-01, ..., -6.7504e-02,
-5.0519e-01, 1.8263e-01]]],
[[[ 1.1010e+00, 1.4875e+00, 1.1929e+00, ..., 3.0980e-01,
3.7599e-01, 2.0062e-01],
[-9.4244e-01, -1.5073e+00, -9.0674e-01, ..., 3.6190e-01,
4.1176e-01, -1.0338e+00],
[-1.5360e-01, -8.1150e-01, -9.5367e-01, ..., -4.8233e-01,
1.0641e+00, 7.0747e-01],
...,
[ 7.8460e-01, 5.7498e-01, 6.3276e-01, ..., -7.5053e-01,
8.7675e-01, 5.9607e-01],
[-4.5942e-01, -8.6741e-01, -2.0825e-01, ..., 5.7479e-02,
1.1111e+00, -9.0727e-02],
[-1.8953e-01, -2.3900e-01, -7.9222e-01, ..., -2.3034e-01,
3.7459e-01, 5.8324e-01]],
[[-3.1872e+00, -1.9727e+00, -2.1886e+00, ..., -2.0117e+00,
-2.5356e+00, -1.4354e+00],
[-2.8803e+00, -3.2256e+00, -1.6047e+00, ..., -2.3190e-01,
-1.2320e+00, -4.9541e-01],
[-2.7711e+00, -2.4087e+00, -1.1480e+00, ..., -1.1484e+00,
-1.4015e+00, -8.3751e-01],
...,
[-2.8681e-02, -9.2221e-01, -6.7562e-01, ..., -1.9119e+00,
3.4177e-01, 4.7451e-01],
[-1.1947e+00, -8.5147e-01, -5.0565e-01, ..., -2.5715e-01,
-2.6502e-01, 1.9407e-01],
[-4.5983e-01, -1.0954e+00, -8.8168e-01, ..., -4.8192e-01,
-7.4111e-01, -5.2458e-01]],
[[ 2.3720e+00, 2.3450e+00, 1.7762e+00, ..., 3.0301e+00,
1.2469e+00, 3.0740e+00],
[ 1.8598e-01, 1.5961e-01, 1.4499e+00, ..., 1.9552e+00,
-2.1823e-01, 2.6063e+00],
[ 2.2761e-01, 6.6211e-01, 1.3974e+00, ..., 9.2744e-01,
2.1694e-01, 1.6478e+00],
...,
[-1.8088e-01, 4.6287e-01, 8.2036e-01, ..., 2.7349e-01,
2.9946e-01, 1.5071e+00],
[-1.0152e+00, 5.0927e-01, 5.0045e-01, ..., 1.1890e+00,
4.6290e-01, 1.4591e+00],
[ 6.6614e-01, 4.7688e-01, 1.0621e+00, ..., 1.4852e+00,
1.0096e+00, 1.9245e+00]],
...,
[[-2.9913e+00, -8.4822e+00, -6.6975e+00, ..., -3.7626e+00,
-2.5123e+00, -4.1226e+00],
[-3.8864e+00, -1.0228e+01, -6.3602e+00, ..., -6.0122e+00,
-3.8983e+00, -4.8319e+00],
[-2.8697e+00, -9.3280e+00, -5.8601e+00, ..., -3.3014e+00,
-2.2446e+00, -1.6146e+00],
...,
[-3.3379e+00, -8.8716e+00, -6.1106e+00, ..., -4.2995e+00,
-3.3535e+00, -1.4906e+00],
[-4.1574e+00, -9.5330e+00, -5.1942e+00, ..., -6.5223e+00,
-4.5219e+00, -3.5165e+00],
[-3.2649e+00, -7.7509e+00, -3.0241e+00, ..., -3.5617e+00,
-2.1002e+00, -2.5716e+00]],
[[-3.7649e-01, 7.5638e-01, 8.7503e-01, ..., 1.1298e+00,
8.4059e-01, 1.1575e+00],
[ 7.9385e-01, 7.8781e-01, 1.3475e+00, ..., 2.0141e+00,
2.4125e+00, 2.6101e+00],
[ 1.0589e-01, 2.5056e-01, 1.5619e+00, ..., 1.3916e+00,
1.3487e+00, 1.8327e+00],
...,
[ 1.5442e+00, 6.9030e-01, 1.7134e+00, ..., 1.6511e+00,
1.9589e+00, 1.6848e+00],
[ 1.7402e+00, 1.5377e+00, 1.5909e+00, ..., 1.9359e+00,
1.4600e+00, 2.4619e+00],
[ 7.5704e-01, 6.2783e-01, 1.2175e+00, ..., 1.1079e+00,
1.3412e+00, 1.3216e+00]],
[[-2.2722e-01, -3.6957e-01, -3.7011e-01, ..., -1.8257e-01,
-9.5470e-02, 8.6823e-01],
[-7.3054e-01, -2.7019e+00, -1.1356e+00, ..., 2.0751e-01,
1.0321e-01, 5.3626e-01],
[-1.4854e+00, -1.7259e+00, -6.2478e-01, ..., 2.9557e-01,
1.5095e-01, 2.4920e-01],
...,
[-2.0268e+00, -2.6563e+00, -6.4770e-01, ..., -1.4502e+00,
-1.8835e-01, -2.7556e-01],
[-1.3879e+00, -2.5926e+00, -7.3893e-01, ..., -1.4131e-01,
-2.0175e-01, 3.4456e-01],
[-4.0046e-01, -1.2344e+00, -6.2128e-01, ..., 2.5239e-02,
1.4676e-01, 3.3333e-01]]],
[[[ 8.7581e-01, 1.2967e+00, 6.9601e-01, ..., 1.3185e+00,
2.9725e+00, -1.2239e-01],
[-6.8985e-01, -7.1053e-01, -2.5595e-01, ..., -7.5412e+00,
1.8808e-01, -1.9474e+00],
[-1.1545e-02, -9.9422e-01, 6.3764e-01, ..., -5.9602e+00,
-8.8486e-02, -1.4470e+00],
...,
[-5.8705e+00, -1.0213e+01, -1.9305e+01, ..., 1.7489e-01,
-3.5741e-01, 3.1013e-01],
[-1.2407e+01, -1.5337e+01, -1.4104e+01, ..., 2.5448e-01,
2.4620e-01, -3.9143e-02],
[-4.0549e+00, -4.3483e+00, -2.9355e+00, ..., -8.2989e-02,
-7.3246e-01, 4.6588e-02]],
[[-4.0503e+00, -7.4666e-01, -1.8845e+00, ..., -9.1393e+00,
-8.3024e+00, -2.5369e+00],
[-2.7016e+00, -9.8173e-01, -8.4016e-01, ..., -2.1543e+01,
-1.5332e+01, -3.4630e+00],
[-1.3310e+00, 5.3253e-01, -4.2211e-01, ..., -2.1620e+01,
-1.0864e+01, -1.3586e+00],
...,
[-9.5655e+00, -1.9962e+01, -2.8299e+01, ..., -3.9741e+00,
1.6136e-03, -2.3306e-01],
[-2.3990e+01, -3.1434e+01, -3.4127e+01, ..., -1.5092e+00,
4.7456e-02, -5.3590e-01],
[-6.3799e+00, -1.0268e+01, -1.9927e+01, ..., -5.3991e-01,
-3.7518e-01, -2.7974e-01]],
[[ 2.9366e+00, 3.1593e+00, 3.4012e+00, ..., -5.4234e+00,
-1.3293e+00, -5.3452e-01],
[ 1.4701e+00, 8.5286e-01, 2.3773e+00, ..., -1.5760e+01,
-4.0501e+00, -2.6415e+00],
[ 4.3743e-01, -3.7700e-01, 2.2709e-01, ..., -9.8721e+00,
3.9346e+00, -1.5807e+00],
...,
[-9.5705e+00, -9.1739e+00, -9.6872e+00, ..., 1.1596e+00,
-4.4429e-01, 1.8271e+00],
[-1.8055e+01, -1.0382e+01, -9.8959e+00, ..., 1.2758e-01,
5.1009e-01, 1.4907e+00],
[-3.6506e+00, -5.2907e+00, -1.0749e+01, ..., 2.5167e-01,
1.1108e+00, 1.4634e+00]],
...,
[[-3.0556e+00, -7.6652e+00, -3.8097e+00, ..., -7.0738e+00,
-2.8581e+01, -1.9396e+01],
[-3.6418e+00, -8.9767e+00, -3.9110e+00, ..., -9.0242e+00,
-4.0584e+01, -2.2932e+01],
[-2.3600e+00, -5.4903e+00, -2.0075e+00, ..., 4.9308e-01,
-2.7182e+01, -2.1050e+01],
...,
[-3.2170e+00, -2.8637e+01, -1.7915e+01, ..., -7.9578e+00,
-8.9857e+00, -3.6292e+00],
[-1.4177e+00, -1.9863e+01, -8.0278e+00, ..., -6.4468e+00,
-6.5384e+00, -4.6554e+00],
[-1.5613e+00, -4.5919e+00, -1.3317e+00, ..., -5.0270e+00,
-3.3129e+00, -2.7453e+00]],
[[-6.3205e-01, 4.9107e-01, 2.4749e-01, ..., -1.1089e+01,
-1.3534e+00, 4.8303e+00],
[ 1.0425e+00, 4.1195e-01, 1.1977e+00, ..., -1.4431e+01,
-3.0872e+00, 6.5250e+00],
[ 1.6847e+00, 1.0019e+00, -1.9406e-01, ..., -1.3650e+01,
-3.6305e-01, 6.0446e+00],
...,
[ 4.5209e-01, -1.2406e+01, -1.9491e+01, ..., 1.7178e+00,
2.5453e+00, 1.5248e+00],
[ 1.3681e+00, -1.1160e+01, -2.2726e+01, ..., 2.5907e+00,
1.6894e+00, 1.9105e+00],
[ 1.9373e-03, -5.0398e+00, -9.2369e+00, ..., 1.1085e+00,
8.3874e-01, 1.4590e+00]],
[[-6.4972e-01, 2.3993e-02, -2.7838e-01, ..., 6.4189e+00,
-5.4434e-01, -1.5107e+00],
[-1.2882e+00, -2.2981e-01, -4.8633e-01, ..., 2.2815e+00,
-1.1045e+01, -9.2012e+00],
[-1.5313e+00, 7.6546e-01, 9.4956e-01, ..., 1.2365e+00,
-6.8609e+00, -8.3408e+00],
...,
[-2.9691e+00, -4.3088e+00, -4.0797e+00, ..., -2.4506e+00,
-2.5900e+00, -7.0427e-01],
[-4.8429e+00, -1.6905e+01, -1.8038e+01, ..., -1.5124e+00,
-1.3754e+00, -4.7847e-01],
[-2.4067e+00, -5.5865e+00, -3.3783e+00, ..., -1.8915e-01,
-5.6977e-01, 2.0218e-02]]],
[[[ 7.4021e-01, 1.1086e+00, 7.4758e-01, ..., -8.5728e-02,
-4.9946e-01, -9.5557e-02],
[-3.0236e-01, -1.3479e+00, -4.5752e-01, ..., -5.7907e-01,
-6.1265e-01, -2.1964e+00],
[ 2.6666e-01, -3.1008e-01, 5.7871e-01, ..., -1.6458e-01,
1.7129e+00, 1.5631e+00],
...,
[-3.9169e-02, -1.5110e+00, -1.0145e+00, ..., 7.8887e-01,
1.1904e+00, 4.2710e-01],
[-4.7204e-01, -1.8365e+00, -8.4209e-01, ..., -1.8175e-01,
1.2614e+00, -6.9016e-02],
[ 3.3057e-01, -5.4687e-01, -6.6844e-01, ..., -2.7160e-01,
3.2185e-01, 5.6454e-01]],
[[-2.9580e+00, -1.4856e+00, -3.0112e+00, ..., -2.5612e+00,
-2.7929e+00, -1.4706e+00],
[-4.2428e+00, -1.7964e+00, -2.3073e+00, ..., -2.2806e+00,
-3.4074e+00, -9.9656e-01],
[-4.5496e+00, -3.5206e+00, -4.9566e+00, ..., 2.5805e-01,
-3.3756e-01, 1.3256e-01],
...,
[-2.9500e+00, -2.3577e+00, -3.3780e+00, ..., -2.7157e-01,
-1.1956e+00, 3.1696e-01],
[-3.7056e+00, -1.4601e+00, -2.3595e+00, ..., -1.3399e+00,
-9.0648e-01, 1.5339e-01],
[-1.2889e+00, -8.6993e-01, -1.6900e+00, ..., -5.6440e-01,
-6.5680e-01, -4.9034e-01]],
[[ 1.6306e+00, 2.7865e+00, 2.4255e+00, ..., 3.2532e+00,
1.7632e+00, 3.5783e+00],
[ 1.3261e-01, 1.2449e+00, 1.5781e+00, ..., 2.7753e+00,
-2.7911e-02, 2.8538e+00],
[ 1.1291e+00, 1.6863e+00, 1.7679e+00, ..., 5.1321e-01,
-1.0386e+00, -6.5336e-03],
...,
[ 9.7597e-01, 1.2775e+00, 1.5999e+00, ..., -1.3089e+00,
-6.1113e-01, 1.2494e+00],
[ 7.1311e-01, 1.9395e+00, 9.1666e-01, ..., 6.9242e-01,
6.6121e-01, 1.3234e+00],
[ 1.5760e+00, 1.6319e+00, 1.6255e+00, ..., 1.4987e+00,
8.3102e-01, 1.8453e+00]],
...,
[[-1.3258e+00, -4.5137e+00, -3.8850e+00, ..., -4.0328e+00,
-3.0919e+00, -4.2241e+00],
[-1.6941e+00, -3.7812e+00, -4.2378e+00, ..., -4.9352e+00,
-3.1804e+00, -3.9266e+00],
[-1.1276e+00, -3.5369e+00, -4.2892e+00, ..., -2.8785e+00,
-3.8127e+00, -3.9558e+00],
...,
[-1.6670e+00, -4.3936e+00, -4.8193e+00, ..., -3.5624e+00,
-2.3389e+00, -1.6659e+00],
[-1.5453e+00, -4.6735e+00, -5.5078e+00, ..., -4.1843e+00,
-3.0779e+00, -3.7919e+00],
[-1.6211e+00, -4.9373e+00, -4.2530e+00, ..., -4.5128e+00,
-2.8750e+00, -2.7472e+00]],
[[ 7.9933e-01, 5.4157e-01, -3.9717e-02, ..., 8.5643e-01,
8.6848e-01, 1.3759e+00],
[ 1.1908e+00, 9.3531e-01, 5.3035e-01, ..., -2.4516e-01,
1.4920e+00, 2.7597e+00],
[ 4.4168e-01, 6.5613e-01, 8.8075e-02, ..., 7.6618e-01,
2.5280e-01, 8.0667e-01],
...,
[ 7.5979e-01, -2.0740e-01, -9.6186e-02, ..., 3.7876e-01,
1.3471e+00, 1.6482e+00],
[ 1.5322e+00, 1.4274e+00, 1.2017e+00, ..., 7.1616e-01,
1.4748e+00, 2.4098e+00],
[ 6.6376e-01, 9.4837e-01, 1.0732e+00, ..., 1.0439e+00,
1.4459e+00, 1.3128e+00]],
[[-2.6227e-01, -1.4485e-01, -9.9412e-01, ..., -3.7712e-01,
-5.6925e-01, 3.6595e-01],
[-1.3845e+00, -1.4098e+00, -1.9651e+00, ..., -6.3220e-01,
-3.3602e-01, -3.8153e-02],
[-1.6293e+00, -5.2860e-01, -2.3222e-01, ..., 4.8515e-01,
7.4784e-01, 8.0207e-01],
...,
[-1.7213e+00, -1.9652e+00, -2.0721e+00, ..., -8.0595e-01,
-1.5123e+00, -2.1979e-01],
[-1.4156e+00, -1.6869e+00, -1.5948e+00, ..., -9.3258e-01,
-3.8425e-01, 3.8756e-01],
[-1.8040e-01, 8.0312e-02, -4.7015e-01, ..., 3.8871e-01,
4.1002e-01, 3.8008e-01]]]], device='cuda:0'),
tensor([[[[ 5.6095e-01, 8.7513e-01, 5.8603e-01, ..., 3.4345e-01,
4.3428e-01, 4.2731e-02],
[-3.0684e-01, -8.7101e-01, -4.8648e-01, ..., 1.2915e-01,
3.5012e-02, -3.2378e-01],
[-2.4747e-01, -8.3963e-02, 2.3083e-01, ..., -4.2657e-02,
6.8085e-02, 6.9599e-01],
...,
[ 1.9831e+00, 4.7349e-02, 1.1166e+00, ..., -3.0310e-01,
4.8202e-01, 6.3531e-01],
[-4.1659e-02, -7.1537e-01, 1.0474e-01, ..., 8.8915e-01,
1.5789e+00, 1.2836e-01],
[ 8.1878e-01, -2.2607e-01, -1.5786e-01, ..., 4.1428e-01,
3.2706e-01, 6.7897e-01]],
[[-2.4360e+00, -6.5428e-01, -1.8592e+00, ..., -1.6220e+00,
-2.2188e+00, -1.5766e+00],
[-2.5681e+00, -6.5119e-01, -1.4425e+00, ..., -1.3268e+00,
-2.4329e+00, -1.7956e+00],
[-2.3601e+00, -1.2664e+00, -1.4727e+00, ..., -1.8566e+00,
-2.2646e+00, -1.1070e+00],
...,
[-1.3990e+00, -1.2054e+00, -1.5715e+00, ..., -3.9872e-01,
1.5517e-01, 7.6422e-01],
[-2.5544e+00, -2.0926e-01, 4.5224e-02, ..., -1.4043e-02,
3.6708e-02, 6.8697e-01],
[-1.5641e+00, -1.2359e+00, -9.4094e-01, ..., -4.4964e-01,
-6.5424e-01, -1.7080e-01]],
[[ 1.9299e+00, 2.5255e+00, 1.5741e+00, ..., 1.5872e+00,
7.3285e-01, 2.1170e+00],
[-2.5984e-01, 3.3563e-01, 5.4633e-01, ..., 8.0377e-01,
2.9131e-01, 2.3104e+00],
[ 8.9690e-02, 2.2372e-01, 7.5104e-02, ..., -1.0616e-03,
7.3593e-01, 1.9635e+00],
...,
[ 1.3783e+00, 1.3653e+00, 2.7939e+00, ..., 3.0418e-01,
-2.8085e-01, 1.5661e+00],
[ 1.0995e+00, 2.3884e+00, 2.3371e+00, ..., 1.0066e+00,
7.3304e-01, 1.6877e+00],
[ 2.4000e+00, 1.8486e+00, 2.3977e+00, ..., 1.9649e+00,
1.0414e+00, 1.6972e+00]],
...,
[[-2.0981e+00, -5.0776e+00, -3.2381e+00, ..., -2.0972e+00,
-1.6348e+00, -3.1017e+00],
[-2.4994e+00, -5.2648e+00, -2.8438e+00, ..., -2.4263e+00,
-2.1667e+00, -2.9268e+00],
[-1.5019e+00, -5.3164e+00, -3.0341e+00, ..., -2.3409e+00,
-2.4391e+00, -2.2323e+00],
...,
[-2.8963e+00, -4.8435e+00, -3.2770e+00, ..., -7.5882e+00,
-4.8553e+00, -2.8695e+00],
[-2.5534e+00, -5.6238e+00, -3.9883e+00, ..., -4.3810e+00,
-3.1862e+00, -4.8623e+00],
[-2.6574e+00, -5.9049e+00, -2.8461e+00, ..., -3.0011e+00,
-1.7043e+00, -3.2563e+00]],
[[ 9.5994e-01, 1.6439e+00, 7.7685e-01, ..., 1.1080e+00,
1.5636e+00, 1.3789e+00],
[ 1.1892e+00, 1.0269e+00, 8.8428e-01, ..., 9.0278e-01,
7.9267e-01, 1.8790e+00],
[ 7.2135e-01, 7.2498e-02, 3.1904e-01, ..., 4.0692e-01,
1.1159e-01, 1.3763e+00],
...,
[ 1.0764e+00, 5.1220e-02, 1.1313e+00, ..., 2.2799e+00,
1.8739e+00, 1.5410e+00],
[ 2.1004e+00, 1.3953e+00, 2.0260e+00, ..., 1.9750e+00,
1.7429e+00, 2.5575e+00],
[ 1.1686e+00, 9.1491e-01, 1.3385e+00, ..., 1.3084e+00,
1.4330e+00, 1.5353e+00]],
[[-1.0256e-01, -6.3890e-01, -1.1923e+00, ..., -5.4168e-01,
-8.5290e-01, -1.3172e-01],
[-9.3563e-01, -2.5993e+00, -2.4757e+00, ..., -1.2034e+00,
-1.0712e+00, -7.4669e-01],
[-1.5022e+00, -1.8518e+00, -8.1326e-01, ..., -2.8632e-01,
-4.0101e-01, -1.9853e-01],
...,
[-1.3236e+00, -1.4098e+00, -1.0222e+00, ..., 2.6412e-01,
1.0863e-01, -3.6248e-01],
[-1.7589e+00, -1.4884e+00, -5.7290e-01, ..., -3.7226e-01,
-3.6042e-01, 4.3184e-01],
[-2.6080e-01, 3.7004e-01, -1.9810e-01, ..., 2.3854e-01,
6.7717e-04, 6.2341e-01]]],
[[[ 8.5844e-01, 1.2390e+00, 1.7872e-01, ..., 2.1007e-01,
4.0613e-01, 2.8291e-01],
[-1.6412e+00, -2.1664e+00, -1.4596e+00, ..., 9.0288e-02,
2.0440e-01, -1.2634e+00],
[ 1.2425e-01, -2.6146e+00, -3.9937e+00, ..., -8.1101e-01,
9.5913e-01, 6.4167e-01],
...,
[ 6.3409e-01, 5.6365e-01, 7.9204e-01, ..., -3.3213e+00,
-7.9891e-01, 2.5705e-01],
[ 4.1179e-01, -7.7044e-01, 6.1278e-01, ..., -3.3548e+00,
1.5063e+00, 4.4530e-01],
[ 6.6226e-01, 1.1741e+00, 5.7765e-01, ..., -1.0348e+00,
1.9459e+00, 6.1603e-01]],
[[-4.5048e+00, -1.1258e+00, -3.7053e-01, ..., -1.7347e+00,
-2.0833e+00, -1.0444e+00],
[-3.1933e+00, 1.4130e+00, -2.0146e+00, ..., -7.5420e-01,
-1.4102e+00, -3.0469e-01],
[-1.9821e+00, -6.3858e-01, -2.8776e+00, ..., -1.6307e+00,
-1.4832e+00, -7.9153e-01],
...,
[ 5.1123e-01, 5.1038e-01, -3.9030e-01, ..., -6.6837e+00,
-2.8153e+00, -5.3120e-01],
[-4.2616e-01, 1.0216e+00, 4.3844e-01, ..., -1.2107e+00,
-3.0359e+00, -1.2887e+00],
[ 6.8273e-01, -9.2847e-02, -1.0545e+00, ..., -5.9125e+00,
-7.1186e+00, -2.0622e+00]],
[[ 2.8229e+00, 1.6993e+00, -1.1683e+00, ..., 2.3181e+00,
1.3504e+00, 2.9434e+00],
[-3.6805e-01, -3.2193e-01, -3.4810e+00, ..., 1.7741e+00,
-1.6838e-01, 2.5206e+00],
[-1.5315e+00, -1.8541e+00, -1.2842e+00, ..., 1.5906e+00,
7.1602e-01, 1.8591e+00],
...,
[-6.0539e-02, 1.8907e-01, -5.1791e-01, ..., -2.9458e+00,
-3.9730e+00, 1.1976e+00],
[-9.8991e-01, -8.4185e-01, -7.1432e-01, ..., -2.9564e+00,
-3.1788e+00, -3.1693e-01],
[ 5.5614e-01, 5.6840e-01, 3.5894e-01, ..., -3.9582e+00,
-2.9261e+00, -3.9393e-01]],
...,
[[-1.8960e+00, -5.6230e+00, -2.1196e+00, ..., -3.4806e+00,
-2.4416e+00, -3.9999e+00],
[-2.9560e+00, -5.9135e+00, -3.3118e+00, ..., -5.0324e+00,
-3.6687e+00, -4.2586e+00],
[-8.5148e-01, -4.2472e+00, -2.6590e+00, ..., -3.3016e+00,
-2.0817e+00, -1.6335e+00],
...,
[ 3.6331e-01, -1.5794e-01, -1.4431e+00, ..., -1.6969e-01,
-9.7675e-01, -2.1563e+00],
[ 6.8310e-01, 8.2455e-02, -1.5453e+00, ..., -4.3619e-03,
-7.0001e+00, -7.2790e+00],
[-7.8707e-02, -1.7688e+00, -2.7996e+00, ..., -3.6942e+00,
-1.0532e+01, -4.1164e+00]],
[[-7.2449e-01, -7.6484e-02, 5.7955e-01, ..., 8.1937e-01,
1.1013e+00, 1.2378e+00],
[ 1.5788e+00, 1.3950e+00, 1.1035e+00, ..., 1.8777e+00,
2.2732e+00, 2.5788e+00],
[ 6.7950e-01, 5.3296e-01, 7.1758e-01, ..., 1.4475e+00,
1.7233e+00, 1.9775e+00],
...,
[ 8.9465e-01, 1.1331e-01, -5.9516e-01, ..., -5.1064e+00,
-4.7157e-01, 1.3084e+00],
[ 7.8253e-01, 5.6442e-01, 2.8685e-01, ..., -4.4047e+00,
-7.4624e-01, 2.9828e+00],
[ 6.9882e-01, -1.7605e-02, 2.8386e-01, ..., -2.3222e+00,
-3.4525e+00, 1.4264e+00]],
[[-4.8225e-01, 4.0702e-01, -1.2787e-01, ..., -2.7439e-01,
-8.1535e-02, 8.9749e-01],
[-1.5460e+00, 6.6507e-01, -5.8179e-01, ..., 6.5596e-01,
-1.7163e-01, 3.1390e-01],
[-8.0482e-01, -1.9176e-01, -3.0900e+00, ..., 2.9742e-01,
1.8978e-01, 2.5748e-01],
...,
[-6.1617e-01, -2.9724e-01, -3.6820e-01, ..., -3.2013e+00,
-1.9890e+00, -1.6618e-01],
[-6.4178e-01, -1.0103e+00, -9.1894e-01, ..., -1.3495e-01,
7.1569e-01, 1.3798e+00],
[ 6.2396e-01, 5.4206e-01, -2.1389e-01, ..., -1.4684e-02,
-1.2779e+00, -7.2563e-01]]],
[[[ 5.9345e-01, 8.2669e-01, 5.3857e-01, ..., 3.2372e-01,
4.0456e-01, 2.7186e-03],
[-2.2594e-01, -6.9776e-01, -2.5329e-01, ..., 1.6491e-01,
1.4875e-01, -3.0358e-01],
[ 2.1171e-01, -4.1355e-01, 1.7902e-01, ..., -2.3339e-01,
7.8041e-03, 6.4434e-01],
...,
[ 3.2698e-01, -7.6896e-01, -3.6334e-01, ..., 2.2548e-01,
5.3833e-01, 6.1553e-01],
[-4.2783e-02, -7.9688e-01, -1.7842e-01, ..., 6.2639e-01,
1.3193e+00, 1.1435e-01],
[ 8.1663e-01, -1.9515e-01, -8.9544e-02, ..., 3.7485e-01,
3.3191e-01, 7.3335e-01]],
[[-2.4325e+00, -6.1451e-01, -1.9030e+00, ..., -1.6699e+00,
-2.3273e+00, -1.5423e+00],
[-2.5830e+00, -1.2562e+00, -2.3168e+00, ..., -1.2881e+00,
-2.4411e+00, -1.6868e+00],
[-3.4151e+00, -4.2622e+00, -4.4610e+00, ..., -2.0140e+00,
-2.2785e+00, -1.1126e+00],
...,
[-7.2476e-01, -1.1430e+00, -2.1518e+00, ..., -3.0791e+00,
1.6025e-01, 7.0257e-01],
[-2.5788e+00, -3.8004e-01, -4.8305e-01, ..., 3.2287e-02,
2.0716e-01, 6.4915e-01],
[-1.5742e+00, -1.2175e+00, -1.0225e+00, ..., -4.9540e-01,
-5.6581e-01, -1.5890e-01]],
[[ 1.9524e+00, 2.5472e+00, 1.6508e+00, ..., 1.6061e+00,
7.6809e-01, 2.1039e+00],
[-2.7063e-01, 3.3223e-01, 6.8944e-01, ..., 7.8826e-01,
3.2509e-01, 2.2264e+00],
[ 8.8448e-02, 9.5975e-01, 2.3514e+00, ..., -3.2623e-01,
7.3349e-01, 1.9497e+00],
...,
[ 6.6779e-01, 4.0586e-01, 2.2433e+00, ..., 8.4037e-02,
7.2549e-02, 1.5726e+00],
[ 1.0906e+00, 2.2011e+00, 2.1613e+00, ..., 1.0590e+00,
3.8920e-01, 1.6305e+00],
[ 2.3988e+00, 1.8514e+00, 2.4250e+00, ..., 1.9230e+00,
9.1990e-01, 1.6629e+00]],
...,
[[-2.0644e+00, -4.9646e+00, -3.0951e+00, ..., -2.0112e+00,
-1.6683e+00, -3.0513e+00],
[-2.5465e+00, -5.6161e+00, -3.5899e+00, ..., -2.4480e+00,
-2.3662e+00, -3.0395e+00],
[-1.5859e+00, -6.1665e+00, -3.5403e+00, ..., -2.5059e+00,
-2.5224e+00, -2.2593e+00],
...,
[-2.6221e+00, -4.7295e+00, -2.2899e+00, ..., -3.9479e+00,
-3.3522e+00, -2.5633e+00],
[-2.5328e+00, -5.6161e+00, -3.9854e+00, ..., -6.8070e+00,
-4.8355e+00, -4.7556e+00],
[-2.6393e+00, -5.9899e+00, -2.9182e+00, ..., -3.0534e+00,
-1.6562e+00, -3.2332e+00]],
[[ 8.9526e-01, 1.6105e+00, 8.1173e-01, ..., 1.0539e+00,
1.4945e+00, 1.4137e+00],
[ 1.1729e+00, 1.1641e+00, 1.1885e+00, ..., 9.0909e-01,
8.0610e-01, 1.9442e+00],
[-2.4409e-01, -1.5272e+00, -9.5693e-01, ..., 2.8403e-01,
7.0314e-02, 1.3750e+00],
...,
[ 1.5433e+00, -1.5214e-01, 5.0944e-01, ..., 1.8487e+00,
2.0522e+00, 1.4844e+00],
[ 2.0460e+00, 1.1492e+00, 1.9500e+00, ..., 2.4344e+00,
1.6732e+00, 2.5511e+00],
[ 1.2014e+00, 8.7559e-01, 1.3008e+00, ..., 1.3237e+00,
1.4825e+00, 1.4986e+00]],
[[-1.0173e-01, -6.4391e-01, -1.1541e+00, ..., -5.6091e-01,
-9.0061e-01, -9.3290e-02],
[-8.5432e-01, -2.3976e+00, -2.1923e+00, ..., -1.1394e+00,
-1.0566e+00, -6.3302e-01],
[-2.3936e+00, -1.9860e+00, -6.7060e-01, ..., -3.4100e-01,
-4.4318e-01, -2.1716e-01],
...,
[-1.6235e+00, -1.6880e+00, -6.0800e-01, ..., -1.2868e+00,
2.8071e-02, -3.8108e-01],
[-1.7136e+00, -1.3192e+00, -4.6904e-01, ..., 4.3898e-01,
7.0494e-02, 4.4752e-01],
[-2.7485e-01, 4.3585e-01, -2.1780e-01, ..., 2.3737e-01,
5.0790e-02, 6.1016e-01]]],
[[[ 9.3387e-01, 1.6655e+00, -7.4900e-01, ..., 4.4927e-01,
6.3955e-02, -4.5598e-01],
[-7.3696e-01, -2.5818e-01, 6.0481e-02, ..., 2.2951e-01,
1.3833e+00, -3.0031e-01],
[-1.5808e-01, -1.6783e+00, 3.9483e-01, ..., 4.2966e-01,
5.1001e-01, 8.9633e-01],
...,
[ 3.0652e-01, -6.7474e-01, -5.7149e-01, ..., 3.3456e+00,
1.8127e+00, 7.7095e-01],
[-6.1743e-01, -1.7986e+00, -1.2396e+00, ..., -2.6686e+00,
2.0097e-01, 6.7681e-02],
[-2.4757e-01, -1.6968e+00, -1.6595e+00, ..., -1.5287e+00,
-1.9082e+00, -3.9689e-01]],
[[-2.8494e+00, 2.1737e-01, -1.1102e+00, ..., -2.5596e+00,
-3.4169e+00, -8.9770e-01],
[-2.3450e+00, 2.2837e-01, -1.2521e+00, ..., -1.2922e+00,
-1.9528e+00, -1.3105e-01],
[-1.6607e+00, -1.7198e+00, -4.0484e+00, ..., -2.2424e+00,
-1.5338e+00, -4.7854e-01],
...,
[-1.3508e+00, -1.2264e+00, -1.9286e+00, ..., -5.6247e+00,
-3.0692e+00, -7.2140e-01],
[-2.9462e+00, -1.5432e+00, -2.7251e+00, ..., -8.5424e+00,
-3.2891e+00, -1.2176e+00],
[-2.2691e+00, -1.9199e+00, -2.6997e+00, ..., -3.5823e+00,
-2.0184e+00, -9.0085e-01]],
[[ 2.7568e+00, 1.9076e+00, -3.2968e-01, ..., 2.6073e+00,
5.1673e-01, 2.2797e+00],
[ 1.1648e+00, -2.0969e-01, -4.0924e+00, ..., 1.0888e+00,
-2.5121e-01, 2.1989e+00],
[ 1.4283e+00, 1.2785e+00, -4.9027e+00, ..., 1.0175e+00,
1.2653e+00, 2.2628e+00],
...,
[ 1.1501e+00, 1.4407e+00, 1.7463e+00, ..., -3.7438e+00,
-6.5187e-02, 1.8296e+00],
[ 6.9928e-01, 1.8959e+00, 1.3268e+00, ..., -9.9522e-01,
1.8640e+00, 1.7395e+00],
[ 1.6953e+00, 1.6334e+00, 1.8771e+00, ..., 1.1436e+00,
9.9251e-01, 1.3272e+00]],
...,
[[-1.7701e+00, -5.6532e+00, -2.8960e+00, ..., -4.9800e+00,
-2.7890e+00, -3.8796e+00],
[-2.1440e+00, -7.2058e+00, -3.8670e+00, ..., -8.7768e+00,
-4.7038e+00, -5.1695e+00],
[-1.2978e+00, -4.5828e+00, -4.6376e+00, ..., -5.3899e+00,
-2.9352e+00, -2.0921e+00],
...,
[-2.2800e+00, -5.0927e+00, -3.7732e+00, ..., -1.8512e+01,
-9.4521e+00, -5.6036e+00],
[-2.7960e+00, -5.8141e+00, -5.2895e+00, ..., -1.0136e+01,
-7.6597e+00, -7.3113e+00],
[-1.5448e+00, -4.1971e+00, -2.7707e+00, ..., -8.6964e+00,
-7.1140e+00, -5.0300e+00]],
[[-2.7345e-01, 8.3004e-01, -1.1421e-02, ..., 8.0056e-01,
8.9304e-01, 1.5701e+00],
[ 7.8872e-01, 2.0910e-01, 1.0009e+00, ..., 2.2489e+00,
1.8210e+00, 2.5302e+00],
[ 8.9724e-01, -1.9605e+00, 2.3198e+00, ..., 2.1169e+00,
1.9093e+00, 1.8120e+00],
...,
[ 1.2893e+00, 3.0637e-01, 1.2519e-01, ..., -1.9772e+00,
7.3996e-01, 1.7656e+00],
[ 1.9256e+00, 1.6981e+00, 1.0760e+00, ..., -1.9057e+00,
1.8060e+00, 2.2022e+00],
[ 4.7247e-01, 8.0901e-01, 7.0251e-01, ..., 2.2645e+00,
1.9081e+00, 1.7042e+00]],
[[-7.3985e-01, -1.1001e+00, 2.3208e-01, ..., -1.7861e-01,
-4.7019e-01, 5.8465e-01],
[-1.0157e+00, -1.3129e+00, -3.3638e+00, ..., -2.2462e+00,
3.3443e-01, 8.1446e-01],
[-1.4723e+00, -2.4772e-02, -2.4093e+00, ..., -5.9010e-01,
4.0568e-01, -3.8917e-03],
...,
[-1.8814e+00, -1.8306e+00, -1.1800e+00, ..., -1.0561e+00,
-1.9293e+00, -1.6221e+00],
[-1.6195e+00, -2.0518e+00, -1.6481e+00, ..., -6.6009e+00,
-1.9853e+00, -4.2028e-01],
[-8.3631e-01, -8.5311e-01, -1.1665e+00, ..., 3.4861e-01,
1.1609e-01, 3.1606e-01]]],
[[[ 6.2819e-01, 9.3375e-01, 9.1900e-01, ..., 7.5935e-01,
6.0141e-01, 5.3018e-01],
[-5.2147e-01, -1.0887e+00, -9.2629e-01, ..., -4.6508e-02,
-1.6082e-02, -1.5930e+00],
[ 5.6671e-02, -1.0384e+00, -4.0128e+00, ..., -5.1087e-01,
1.0707e+00, 6.7675e-01],
...,
[-1.9158e+00, 4.0348e-01, 3.1072e-01, ..., -5.4765e+00,
1.0094e+00, 1.5382e+00],
[-1.1237e+00, -2.1025e+00, 1.0181e+00, ..., 4.0143e-01,
-3.0651e-01, 1.0969e+00],
[ 2.4393e-01, -1.2412e+00, 1.0793e+00, ..., 3.0335e-01,
-1.4997e-01, -1.7379e+00]],
[[-3.7749e+00, -1.7910e+00, -4.9436e+00, ..., -1.3115e+00,
-1.7225e+00, -9.9953e-01],
[-4.5362e+00, -3.2906e+00, -8.7165e+00, ..., -9.6365e-01,
-1.8767e+00, -6.1443e-01],
[-4.4416e+00, -6.5615e+00, -1.1869e+01, ..., -1.6854e+00,
-1.7462e+00, -1.0803e+00],
...,
[-6.9967e+00, -6.4374e+00, -5.9923e-02, ..., -3.4517e+00,
-1.0158e+00, -4.4796e-01],
[-4.8550e+00, -3.3606e+00, -7.9462e-01, ..., 1.9455e+00,
4.6342e-01, 1.1212e+00],
[-1.4896e+00, -2.4847e+00, -3.8838e+00, ..., -1.4809e+00,
-5.1459e+00, -2.7371e+00]],
[[ 2.2486e+00, 2.9390e+00, 2.2336e+00, ..., 2.7044e+00,
1.2780e+00, 3.0719e+00],
[ 7.0727e-01, 9.5852e-01, 1.8457e+00, ..., 1.6827e+00,
-1.5731e-01, 2.5701e+00],
[ 9.8935e-01, 2.2194e+00, 1.5699e+00, ..., 1.1238e+00,
7.8416e-01, 1.9630e+00],
...,
[ 4.0557e+00, 5.1059e-01, -1.4682e+01, ..., -9.6834e+00,
-1.3583e+00, -1.2432e-01],
[ 2.1461e+00, 3.5172e+00, -3.5472e+00, ..., -1.1658e+00,
-1.2483e+00, 7.1221e-01],
[ 2.0615e+00, 3.0712e+00, 3.3588e+00, ..., 1.9025e+00,
-6.7916e-01, 1.0570e+00]],
...,
[[-1.7132e+00, -5.1570e+00, -4.1092e+00, ..., -3.1417e+00,
-2.2131e+00, -4.3046e+00],
[-2.3034e+00, -5.5078e+00, -7.3225e+00, ..., -4.3922e+00,
-3.0224e+00, -3.9002e+00],
[-1.2976e+00, -4.7913e+00, -7.7089e+00, ..., -2.6802e+00,
-2.3192e+00, -1.7994e+00],
...,
[-2.4979e+00, -7.2223e+00, -8.1756e+00, ..., -4.8501e+00,
-6.7536e+00, -2.3338e+00],
[-2.7756e+00, -6.6343e+00, -6.8975e+00, ..., -6.6037e+00,
-6.2171e+00, -7.3739e+00],
[-2.2148e+00, -6.3042e+00, -4.5290e+00, ..., -2.9355e+00,
-2.6192e+00, -2.4589e+00]],
[[ 3.3532e-01, 3.5324e-01, -8.8817e-01, ..., 8.4807e-01,
1.2459e+00, 1.2027e+00],
[ 1.2693e+00, -4.8469e-01, -2.6139e+00, ..., 1.4813e+00,
2.3407e+00, 2.6180e+00],
[ 6.2358e-01, -3.6720e+00, -8.3920e+00, ..., 1.1667e+00,
1.6784e+00, 2.0557e+00],
...,
[-1.1468e+00, 1.0271e-01, 3.6483e+00, ..., 1.4343e+00,
-2.4973e+00, -1.6734e+00],
[ 9.8510e-01, -5.4120e-01, 3.3591e+00, ..., 3.3671e+00,
2.1665e+00, 2.2947e+00],
[ 4.8338e-01, -1.7587e+00, -2.8576e+00, ..., -3.1756e-01,
-3.4659e-01, 1.7419e+00]],
[[-4.8031e-01, -2.8375e-01, -1.5915e+00, ..., -9.6116e-03,
6.6111e-02, 9.4308e-01],
[-1.5609e+00, -2.6584e+00, -2.8565e+00, ..., -1.5792e-01,
-3.8989e-01, 8.4781e-02],
[-2.0886e+00, -2.9413e+00, -3.5064e+00, ..., 3.9360e-01,
1.3173e-01, 3.3854e-01],
...,
[-3.0593e+00, -1.4010e+00, -3.8952e-01, ..., -1.7184e+00,
-3.3167e+00, -1.6075e+00],
[-2.2386e+00, -1.0017e+00, 3.9688e-01, ..., 8.7705e-01,
-1.7127e-01, 8.4744e-01],
[-1.3942e-01, -2.6795e-01, 7.5823e-01, ..., -3.8259e-01,
1.0718e-01, -9.5425e-01]]]], device='cuda:0'),
tensor([[[[ 1.0607e+00, 1.4449e+00, 1.1692e+00, ..., 2.6562e-01,
3.1758e-01, 2.3096e-01],
[-9.6841e-01, -1.4348e+00, -9.9110e-01, ..., 4.0976e-01,
3.0995e-01, -1.1908e+00],
[-2.8068e-01, -3.1665e-01, -6.1963e-01, ..., -4.0241e-01,
9.7220e-01, 6.6588e-01],
...,
[-2.2605e-01, -3.9046e-01, 3.9062e-02, ..., 2.3378e-02,
9.3486e-01, 4.9034e-01],
[-4.0956e-01, -8.4367e-01, -2.1583e-01, ..., 2.3636e-01,
1.3663e+00, -1.1957e-01],
[-1.7550e-01, -2.1303e-01, -7.3446e-01, ..., -1.7938e-01,
3.9378e-01, 6.3200e-01]],
[[-3.1731e+00, -1.9384e+00, -2.0983e+00, ..., -1.8154e+00,
-2.1755e+00, -1.0643e+00],
[-2.8211e+00, -3.3255e+00, -1.6676e+00, ..., -3.8226e-01,
-1.3961e+00, -3.0543e-01],
[-2.1006e+00, -2.8183e+00, -1.7369e+00, ..., -9.8752e-01,
-1.5275e+00, -7.4948e-01],
...,
[ 8.8526e-02, -8.7690e-01, -4.8088e-01, ..., -7.5866e-01,
2.7114e-01, 5.1084e-01],
[-1.1589e+00, -8.1330e-01, -4.4011e-01, ..., -2.7972e-01,
-3.6014e-01, 2.2358e-01],
[-3.7442e-01, -9.3865e-01, -7.9914e-01, ..., -2.7970e-01,
-7.1698e-01, -5.8485e-01]],
[[ 2.3341e+00, 2.3072e+00, 1.7433e+00, ..., 2.8568e+00,
1.3907e+00, 2.9183e+00],
[ 2.2366e-01, 1.0036e-01, 1.3054e+00, ..., 1.8449e+00,
-1.3790e-01, 2.5255e+00],
[-2.5135e-01, 2.6936e-01, 1.1949e+00, ..., 1.3323e+00,
7.2536e-01, 1.8755e+00],
...,
[-5.8684e-01, -1.6655e-01, 4.2811e-02, ..., 1.2485e+00,
1.3790e-02, 1.3971e+00],
[-9.7569e-01, 5.3734e-01, 4.8590e-01, ..., 1.2634e+00,
5.7122e-01, 1.5648e+00],
[ 6.9624e-01, 4.6852e-01, 1.0962e+00, ..., 1.5058e+00,
1.1092e+00, 2.0332e+00]],
...,
[[-2.8602e+00, -8.3549e+00, -6.5717e+00, ..., -3.5043e+00,
-2.4155e+00, -3.9638e+00],
[-3.5764e+00, -9.8657e+00, -6.2215e+00, ..., -5.5684e+00,
-3.8573e+00, -4.3762e+00],
[-2.1912e+00, -7.6638e+00, -4.9224e+00, ..., -2.3691e+00,
-2.2418e+00, -1.6474e+00],
...,
[-3.7082e+00, -8.7850e+00, -5.1255e+00, ..., -2.9801e+00,
-3.3222e+00, -1.5626e+00],
[-4.0538e+00, -9.3337e+00, -5.0094e+00, ..., -5.0717e+00,
-3.7333e+00, -3.7336e+00],
[-3.1688e+00, -7.5495e+00, -2.9070e+00, ..., -3.6535e+00,
-2.2001e+00, -2.7835e+00]],
[[-3.9890e-01, 7.3140e-01, 9.2984e-01, ..., 1.3144e+00,
1.1042e+00, 1.2806e+00],
[ 7.8792e-01, 7.6238e-01, 1.3494e+00, ..., 1.8205e+00,
2.2723e+00, 2.5877e+00],
[ 6.6078e-01, 4.9946e-01, 1.6434e+00, ..., 1.5624e+00,
1.6995e+00, 1.9590e+00],
...,
[ 1.1590e+00, 3.8470e-01, 1.1750e+00, ..., 1.6768e+00,
1.9967e+00, 1.6772e+00],
[ 1.7473e+00, 1.5412e+00, 1.5821e+00, ..., 1.6546e+00,
1.4929e+00, 2.4458e+00],
[ 7.8479e-01, 6.6224e-01, 1.2301e+00, ..., 1.2213e+00,
1.3246e+00, 1.3455e+00]],
[[-2.3038e-01, -3.9568e-01, -3.2678e-01, ..., -1.7443e-01,
-1.4083e-01, 8.8433e-01],
[-6.6053e-01, -2.4556e+00, -1.1388e+00, ..., 1.9746e-01,
-1.6587e-01, 3.8965e-01],
[-1.5974e+00, -1.3412e+00, -9.1342e-01, ..., 2.8871e-01,
1.2843e-01, 2.9569e-01],
...,
[-1.5925e+00, -2.4287e+00, -6.7715e-01, ..., -4.0999e-01,
-1.0547e-01, -3.1340e-01],
[-1.3993e+00, -2.5377e+00, -7.4838e-01, ..., -3.8229e-01,
-2.4366e-01, 3.1071e-01],
[-4.2326e-01, -1.1347e+00, -5.4298e-01, ..., 1.0604e-01,
1.9174e-01, 3.3124e-01]]],
[[[ 8.6660e-01, 1.4673e+00, 1.2068e+00, ..., 1.4208e+00,
1.0854e+00, 9.7821e-01],
[-9.4108e-01, -1.9672e+00, -1.1726e+00, ..., -2.9179e-01,
-5.9640e-01, -1.5654e+00],
[-1.9632e-01, -7.8873e-01, 1.3097e-02, ..., -2.9490e-01,
6.5582e-01, 8.5628e-01],
...,
[-1.0856e-01, -1.2897e+00, -1.0131e+00, ..., -5.4832e-01,
1.9694e-02, 2.6504e-01],
[-6.6664e-01, -2.0057e+00, -1.3349e+00, ..., -5.5770e-01,
-3.7921e-01, -1.3503e-01],
[-1.9669e-01, -1.7699e+00, -1.6613e+00, ..., -1.1345e+00,
-1.4662e+00, -2.9518e-01]],
[[-3.6961e+00, -6.9505e-01, -1.5628e+00, ..., -1.6373e+00,
-2.5679e+00, -1.3038e+00],
[-3.8393e+00, -2.7320e+00, -1.7944e+00, ..., -2.0959e+00,
-3.2010e+00, -1.0115e+00],
[-2.3610e+00, -1.7320e+00, -1.7659e+00, ..., -2.3691e+00,
-2.8955e+00, -1.0564e+00],
...,
[-1.6717e+00, -1.5225e+00, -2.6946e+00, ..., -1.9720e+00,
-2.1608e+00, -7.9078e-01],
[-3.4013e+00, -1.4336e+00, -2.9609e+00, ..., -2.3651e+00,
-2.3182e+00, -1.2398e+00],
[-2.3330e+00, -1.7937e+00, -2.4322e+00, ..., -1.9035e+00,
-1.9882e+00, -1.0424e+00]],
[[ 2.8954e+00, 4.2116e+00, 3.3871e+00, ..., 2.9240e+00,
1.0772e+00, 3.1064e+00],
[ 2.7744e-01, 1.0824e+00, 2.3055e+00, ..., 1.2755e+00,
-8.6900e-01, 2.7909e+00],
[ 1.0333e+00, 1.5196e+00, 1.6379e+00, ..., 1.0434e+00,
3.9889e-02, 2.1245e+00],
...,
[ 9.0957e-01, 1.0673e+00, 1.2896e+00, ..., 2.4156e-01,
1.1831e+00, 2.3880e+00],
[ 5.8024e-01, 1.8434e+00, 1.4222e+00, ..., 6.8809e-02,
1.0024e+00, 1.9130e+00],
[ 1.6509e+00, 1.5051e+00, 1.9053e+00, ..., 1.3221e+00,
1.7076e+00, 1.7259e+00]],
...,
[[-1.8772e+00, -7.2040e+00, -5.3799e+00, ..., -4.0702e+00,
-3.2752e+00, -5.6117e+00],
[-2.4795e+00, -7.0821e+00, -4.2329e+00, ..., -3.5563e+00,
-2.7480e+00, -4.3897e+00],
[-9.8454e-01, -4.5157e+00, -3.6522e+00, ..., -2.1695e+00,
-2.8241e+00, -2.4324e+00],
...,
[-2.7093e+00, -5.5377e+00, -4.5905e+00, ..., -5.4343e+00,
-4.4757e+00, -4.7186e+00],
[-3.0659e+00, -6.0437e+00, -5.3219e+00, ..., -6.3895e+00,
-5.7401e+00, -6.1594e+00],
[-1.8893e+00, -4.5924e+00, -2.6618e+00, ..., -3.3387e+00,
-2.9010e+00, -3.2240e+00]],
[[-1.7236e-01, 3.6570e-01, 5.7935e-01, ..., 6.2011e-01,
8.3068e-01, 1.4108e+00],
[ 1.6592e+00, 1.0981e+00, 1.8228e+00, ..., 1.7785e+00,
2.4946e+00, 3.0419e+00],
[ 1.1820e+00, 2.2916e-01, 7.8742e-01, ..., 1.5934e+00,
1.7305e+00, 2.3968e+00],
...,
[ 1.4269e+00, 2.8900e-01, -3.2392e-01, ..., 1.1571e+00,
1.1060e+00, 1.6640e+00],
[ 1.9724e+00, 1.9069e+00, 1.0220e+00, ..., 1.4915e+00,
1.3333e+00, 2.0265e+00],
[ 4.2554e-01, 9.3536e-01, 9.0968e-01, ..., 5.8366e-01,
6.1635e-01, 1.6482e+00]],
[[-5.7325e-01, 2.7955e-01, 1.2060e-01, ..., 5.7439e-01,
5.5553e-01, 1.3837e+00],
[-1.3023e+00, -1.7779e+00, -1.6227e+00, ..., -7.7061e-01,
-4.8571e-01, 2.6494e-01],
[-2.2356e+00, -9.2740e-01, 2.2849e-01, ..., 1.5866e-01,
9.0443e-02, 8.2396e-01],
...,
[-2.0502e+00, -2.0495e+00, -1.5670e+00, ..., -6.3418e-01,
-1.0677e+00, -9.0081e-01],
[-1.6458e+00, -2.2737e+00, -1.7340e+00, ..., -1.1635e+00,
-1.2751e+00, -6.6190e-01],
[-8.6283e-01, -9.3712e-01, -1.1334e+00, ..., -5.9224e-01,
-8.3754e-01, -8.1445e-02]]],
[[[ 8.5141e-01, 1.4917e+00, 1.1992e+00, ..., 1.4121e+00,
1.0931e+00, 9.9098e-01],
[-1.0028e+00, -2.0375e+00, -1.1773e+00, ..., -3.9523e-01,
-6.5709e-01, -1.5680e+00],
[-2.2152e-01, -8.2914e-01, 1.3767e-01, ..., -8.0550e-01,
5.8311e-01, 8.4955e-01],
...,
[-5.9722e-03, -1.1305e+00, -1.3445e+00, ..., -1.6202e-01,
-5.5000e-02, 2.5490e-01],
[-5.9062e-01, -1.8197e+00, -1.4537e+00, ..., -5.7769e-01,
-3.7188e-01, -1.5663e-01],
[-2.8409e-01, -1.6724e+00, -1.6309e+00, ..., -1.1778e+00,
-1.5483e+00, -3.0333e-01]],
[[-3.7722e+00, -7.5849e-01, -1.5944e+00, ..., -1.5537e+00,
-2.4295e+00, -1.1886e+00],
[-4.0198e+00, -2.8945e+00, -1.9194e+00, ..., -1.8712e+00,
-3.2056e+00, -8.8358e-01],
[-2.3458e+00, -1.7132e+00, -1.7048e+00, ..., -2.0977e+00,
-2.8377e+00, -1.0106e+00],
...,
[-1.5687e+00, -1.5514e+00, -2.5997e+00, ..., -1.9152e+00,
-1.7767e+00, -7.7297e-01],
[-3.0021e+00, -1.6374e+00, -2.9445e+00, ..., -1.9419e+00,
-2.3170e+00, -1.2418e+00],
[-2.2362e+00, -1.9499e+00, -2.8211e+00, ..., -2.0208e+00,
-2.0478e+00, -1.0376e+00]],
[[ 2.9164e+00, 4.2422e+00, 3.3709e+00, ..., 2.9097e+00,
1.0442e+00, 3.0035e+00],
[ 2.6967e-01, 1.1095e+00, 2.2809e+00, ..., 1.1516e+00,
-1.1955e+00, 2.7548e+00],
[ 9.3586e-01, 1.4941e+00, 1.5753e+00, ..., 8.1818e-01,
-1.6042e-01, 2.0612e+00],
...,
[ 9.2520e-01, 1.0039e+00, 1.5708e+00, ..., 8.8802e-01,
8.4636e-01, 2.3350e+00],
[ 6.3952e-01, 1.8152e+00, 1.0814e+00, ..., -2.5745e-01,
1.0339e+00, 1.8544e+00],
[ 1.6326e+00, 1.6857e+00, 1.8439e+00, ..., 1.3110e+00,
1.7745e+00, 1.7740e+00]],
...,
[[-1.9064e+00, -7.2621e+00, -5.4182e+00, ..., -4.0469e+00,
-3.2683e+00, -5.6185e+00],
[-2.4589e+00, -6.9995e+00, -4.1663e+00, ..., -3.3446e+00,
-2.8347e+00, -4.3749e+00],
[-9.5441e-01, -4.5384e+00, -3.6343e+00, ..., -1.6347e+00,
-3.2942e+00, -2.5123e+00],
...,
[-2.5888e+00, -5.2513e+00, -4.4526e+00, ..., -6.4807e+00,
-5.9452e+00, -4.6990e+00],
[-2.7514e+00, -5.6747e+00, -5.4752e+00, ..., -8.2755e+00,
-5.8318e+00, -6.1575e+00],
[-1.4818e+00, -4.1216e+00, -2.8054e+00, ..., -3.3533e+00,
-2.7257e+00, -3.1900e+00]],
[[-1.6830e-01, 3.1811e-01, 5.4404e-01, ..., 6.4791e-01,
8.6760e-01, 1.3753e+00],
[ 1.6085e+00, 1.0902e+00, 1.8279e+00, ..., 1.6851e+00,
2.3968e+00, 3.0168e+00],
[ 1.1844e+00, 3.1101e-01, 8.4176e-01, ..., 2.1776e+00,
1.9621e+00, 2.3819e+00],
...,
[ 1.2551e+00, 2.6602e-01, -5.4646e-01, ..., 2.2873e+00,
1.3371e+00, 1.6050e+00],
[ 1.8628e+00, 1.6457e+00, 9.3207e-01, ..., 1.7814e+00,
1.2822e+00, 1.9647e+00],
[ 4.4912e-01, 7.3681e-01, 6.4741e-01, ..., 5.0828e-01,
6.6843e-01, 1.6484e+00]],
[[-5.5360e-01, 3.0796e-01, 1.4320e-01, ..., 5.6688e-01,
5.3406e-01, 1.3595e+00],
[-1.3710e+00, -1.8282e+00, -1.6277e+00, ..., -1.0567e+00,
-6.8978e-01, 2.2057e-01],
[-2.2074e+00, -8.7024e-01, 3.2207e-01, ..., -4.7687e-01,
4.9480e-02, 7.8968e-01],
...,
[-1.9250e+00, -1.8234e+00, -1.2338e+00, ..., -9.7455e-01,
-8.4527e-01, -8.3285e-01],
[-1.5185e+00, -2.0504e+00, -1.8016e+00, ..., -6.9429e-01,
-1.0950e+00, -6.5356e-01],
[-7.8295e-01, -8.4849e-01, -1.2186e+00, ..., -5.5824e-01,
-8.8280e-01, -1.2780e-01]]],
[[[ 8.2059e-01, 2.0589e-01, -1.9031e-01, ..., -4.3980e-02,
6.9440e-01, 4.3445e-01],
[-1.7596e+00, -3.5679e+00, -2.3628e+00, ..., 1.7601e+00,
-5.4239e-01, -5.0050e-01],
[-1.2641e+00, -1.6554e+00, -2.1116e+00, ..., -1.9490e+00,
-7.2312e-01, 5.5679e-01],
...,
[-8.1060e-01, -1.0835e+00, -2.4415e+00, ..., -2.3532e-01,
-1.0236e-01, 3.1423e-01],
[-7.3558e+00, 1.2589e+00, -1.4437e+00, ..., -1.3064e-01,
5.7825e-02, -1.6590e-01],
[-3.7875e+00, -1.7931e+00, -6.8908e-01, ..., -3.7869e-01,
-9.4802e-01, 1.6539e-01]],
[[-3.8801e+00, -1.4205e+00, 1.4777e-01, ..., -3.3248e+00,
-2.0798e+00, -1.7519e+00],
[-3.7822e+00, -2.3389e+00, 1.0118e+00, ..., -3.5728e+00,
-2.5602e+00, -2.1143e+00],
[ 1.5874e+00, -1.7307e+00, -2.2516e-01, ..., -6.6190e+00,
-3.2788e+00, -1.3934e+00],
...,
[-2.5780e+00, -8.3466e+00, -7.0887e+00, ..., -8.8462e-01,
-8.7785e-01, -7.0420e-01],
[-3.5953e+00, -3.8779e+00, -2.6778e+00, ..., -8.8880e-01,
-1.0981e+00, -1.1246e+00],
[-6.2441e+00, -6.5614e+00, -5.4592e+00, ..., -6.9332e-01,
-7.0841e-01, -4.8600e-01]],
[[ 2.9454e+00, 3.0016e+00, 1.6672e+00, ..., 1.4236e+00,
1.3469e+00, 2.6481e+00],
[-2.1385e-01, -2.7850e+00, -2.8096e+00, ..., 2.6571e-01,
-5.7381e-01, 2.7553e+00],
[-2.9018e+00, -5.6590e+00, -3.4404e+00, ..., 1.7393e+00,
-3.0400e-01, 2.0471e+00],
...,
[-4.7826e+00, -3.7723e+00, -7.2046e+00, ..., -6.9836e-01,
5.9837e-01, 1.5597e+00],
[-1.0271e+01, -2.2793e+00, -3.8599e+00, ..., -5.0967e-01,
9.8483e-01, 1.2705e+00],
[-6.9198e+00, -5.6043e+00, -6.1835e+00, ..., 4.0448e-01,
1.3971e+00, 1.4121e+00]],
...,
[[-2.2206e+00, -6.8130e+00, -3.9562e+00, ..., -5.1193e+00,
-5.2639e+00, -4.5873e+00],
[-4.0389e+00, -7.4865e+00, -5.4496e+00, ..., -1.2253e+01,
-9.1785e+00, -3.7485e+00],
[-5.6063e+00, -1.1827e+01, -4.1692e+00, ..., -8.6931e+00,
-6.5045e+00, -1.6760e+00],
...,
[-5.5925e+00, -1.6216e+01, -2.0217e+00, ..., -5.3585e+00,
-3.5909e+00, -2.7637e+00],
[-1.1370e+00, -1.6365e+01, -3.1931e+00, ..., -6.2200e+00,
-4.6137e+00, -4.3399e+00],
[-1.1205e+00, -7.5874e+00, -7.2476e+00, ..., -3.8971e+00,
-2.7261e+00, -2.5589e+00]],
[[-1.5440e+00, -3.2813e-01, 1.0825e+00, ..., 7.4737e-01,
2.2529e+00, 1.2594e+00],
[-1.5193e+00, -3.7103e+00, -1.9344e+00, ..., -6.3441e-01,
2.1612e+00, 1.9436e+00],
[ 3.0974e+00, 1.1849e+00, -1.9982e+00, ..., -9.8086e-01,
1.2520e+00, 1.3472e+00],
...,
[-2.8621e+00, -6.3761e+00, -1.8709e+00, ..., 2.1186e+00,
7.2166e-01, 1.0729e+00],
[-8.9238e-01, -8.6598e+00, -3.4031e+00, ..., 1.7184e+00,
6.5540e-01, 1.6395e+00],
[ 1.7592e+00, -1.4577e+00, -2.1219e+00, ..., 8.9949e-01,
6.7861e-01, 1.5006e+00]],
[[-1.2326e+00, -7.0933e-01, -3.0517e-01, ..., -5.5842e-01,
-7.9629e-01, 2.0401e-02],
[-2.0283e+00, -2.6194e+00, 1.1040e+00, ..., -1.0295e+00,
-2.7600e+00, -1.1132e+00],
[ 9.4182e-01, -5.2996e-01, 1.2392e+00, ..., -2.7289e+00,
-2.7046e+00, -3.0426e-01],
...,
[ 3.1920e-01, -9.7277e+00, -5.2100e+00, ..., -7.6795e-01,
-1.6216e+00, -8.1518e-01],
[-3.5092e+00, -6.6173e+00, -2.9301e+00, ..., -8.6678e-01,
-1.6377e+00, -6.1432e-01],
[-2.7479e+00, -2.8544e+00, -2.9985e+00, ..., -3.1193e-01,
-7.6345e-01, 3.6457e-02]]],
[[[ 7.3784e-01, 1.0803e+00, 7.0486e-01, ..., 3.5511e-01,
3.3454e-01, 2.2455e-01],
[-3.0468e-01, -1.3834e+00, -5.2201e-01, ..., 4.1295e-01,
4.0861e-01, -1.1503e+00],
[ 1.3501e-01, -8.2142e-01, -2.0287e-01, ..., -7.8781e-01,
9.7207e-01, 4.3799e-01],
...,
[ 3.2279e-02, -1.3110e+00, -3.9321e-01, ..., 2.4880e-01,
7.4540e-01, 5.0043e-01],
[-4.4346e-01, -1.8251e+00, -9.1970e-01, ..., -7.6855e-02,
1.3663e+00, -1.3154e-01],
[ 3.2324e-01, -5.7139e-01, -9.2301e-01, ..., -2.4161e-01,
3.7580e-01, 5.8117e-01]],
[[-3.0657e+00, -1.4442e+00, -3.0511e+00, ..., -1.6896e+00,
-2.2453e+00, -1.1138e+00],
[-4.4069e+00, -1.7572e+00, -2.3373e+00, ..., -3.5471e-01,
-1.3552e+00, -3.3901e-01],
[-4.4332e+00, -2.5583e+00, -3.2234e+00, ..., -1.1013e+00,
-1.4882e+00, -8.6557e-01],
...,
[-2.8735e+00, -1.7485e+00, -1.7109e+00, ..., 1.4792e-01,
1.4174e-01, 4.5331e-01],
[-3.6741e+00, -1.3474e+00, -2.2242e+00, ..., -1.9795e-01,
-4.7298e-01, 2.6679e-01],
[-1.2582e+00, -8.6137e-01, -1.9555e+00, ..., -3.4501e-01,
-7.1689e-01, -5.4891e-01]],
[[ 1.7388e+00, 2.8375e+00, 2.3894e+00, ..., 2.7483e+00,
1.3737e+00, 2.9068e+00],
[ 1.8270e-01, 1.2407e+00, 1.5836e+00, ..., 1.6421e+00,
-1.2798e-01, 2.5101e+00],
[ 7.4472e-01, 1.5108e+00, 9.9375e-01, ..., 1.0080e+00,
7.4506e-01, 1.7819e+00],
...,
[ 9.5001e-01, 1.3023e+00, 1.4043e+00, ..., 9.4021e-01,
-2.9887e-01, 1.4695e+00],
[ 6.4587e-01, 1.8345e+00, 9.2613e-01, ..., 8.1459e-01,
6.7221e-01, 1.5716e+00],
[ 1.5353e+00, 1.5834e+00, 1.5871e+00, ..., 1.4964e+00,
1.1006e+00, 2.0259e+00]],
...,
[[-1.4371e+00, -4.6767e+00, -3.9880e+00, ..., -3.0160e+00,
-2.3449e+00, -4.0425e+00],
[-1.8342e+00, -3.9687e+00, -4.4772e+00, ..., -5.0655e+00,
-3.6287e+00, -4.4675e+00],
[-9.9086e-01, -3.2809e+00, -4.0066e+00, ..., -2.5450e+00,
-2.0492e+00, -1.8092e+00],
...,
[-1.6304e+00, -4.3126e+00, -4.3508e+00, ..., -5.6721e+00,
-4.7157e+00, -1.6660e+00],
[-1.5386e+00, -4.6965e+00, -5.6939e+00, ..., -5.3937e+00,
-3.3702e+00, -3.8075e+00],
[-1.5696e+00, -4.9028e+00, -4.2268e+00, ..., -3.3929e+00,
-2.1827e+00, -2.8189e+00]],
[[ 7.9939e-01, 5.9598e-01, -3.3154e-02, ..., 8.7697e-01,
1.0018e+00, 1.2456e+00],
[ 1.2190e+00, 9.8475e-01, 5.2963e-01, ..., 1.4986e+00,
2.2898e+00, 2.5815e+00],
[ 6.1972e-01, 6.0991e-01, -2.5999e-01, ..., 9.5835e-01,
1.4549e+00, 1.9634e+00],
...,
[ 8.4520e-01, -2.8558e-02, 6.8398e-02, ..., 2.2061e+00,
1.8569e+00, 1.6930e+00],
[ 1.5234e+00, 1.3695e+00, 1.0078e+00, ..., 1.2866e+00,
1.4692e+00, 2.4609e+00],
[ 6.6753e-01, 9.0332e-01, 8.9209e-01, ..., 1.0790e+00,
1.3401e+00, 1.3804e+00]],
[[-2.6070e-01, -1.3359e-01, -1.0513e+00, ..., -1.5942e-01,
-1.1461e-01, 8.6692e-01],
[-1.4102e+00, -1.3841e+00, -2.1733e+00, ..., 2.4787e-01,
-6.6807e-02, 3.7690e-01],
[-1.7521e+00, -7.4227e-01, -1.3455e+00, ..., -2.3776e-02,
-1.0413e-01, 1.6982e-01],
...,
[-1.7338e+00, -1.7002e+00, -1.2779e+00, ..., 3.0974e-01,
1.0921e-01, -3.2547e-01],
[-1.4204e+00, -1.6358e+00, -1.7728e+00, ..., -3.1400e-01,
-4.5310e-01, 3.2879e-01],
[-1.7826e-01, 4.8865e-02, -6.4311e-01, ..., 1.9456e-01,
1.3070e-01, 3.3992e-01]]]], device='cuda:0'),
tensor([[[[ 7.2646e-01, 1.0879e+00, 7.2707e-01, ..., 3.2088e-01,
3.3445e-01, 1.9631e-01],
[-3.0806e-01, -1.4437e+00, -5.6575e-01, ..., 3.8715e-01,
3.4899e-01, -1.1683e+00],
[ 1.5053e-01, -6.8116e-01, -2.0790e-01, ..., -4.0778e-01,
1.0494e+00, 6.5872e-01],
...,
[-1.9227e-01, -1.3741e+00, -4.5418e-01, ..., 3.8755e-01,
8.5006e-01, 4.3644e-01],
[-4.2343e-01, -1.7538e+00, -9.1127e-01, ..., 5.7516e-01,
1.5271e+00, -1.5239e-01],
[ 3.5838e-01, -5.7087e-01, -6.8568e-01, ..., -1.5206e-01,
3.5765e-01, 5.8706e-01]],
[[-3.0681e+00, -1.5402e+00, -3.1078e+00, ..., -1.7196e+00,
-2.1484e+00, -1.0114e+00],
[-4.3806e+00, -1.8508e+00, -2.4224e+00, ..., -4.9079e-01,
-1.2091e+00, -2.3973e-01],
[-4.2411e+00, -2.8465e+00, -3.3233e+00, ..., -1.4808e+00,
-1.4171e+00, -7.5656e-01],
...,
[-3.0498e+00, -2.2090e+00, -2.7350e+00, ..., -1.0122e+00,
-6.4473e-01, 4.8789e-01],
[-3.6689e+00, -1.5604e+00, -2.4752e+00, ..., -8.5006e-01,
-7.0107e-01, 2.1048e-01],
[-1.2747e+00, -9.4552e-01, -1.6926e+00, ..., -6.4697e-01,
-7.9973e-01, -5.8168e-01]],
[[ 1.7038e+00, 2.8148e+00, 2.3928e+00, ..., 2.8006e+00,
1.4200e+00, 2.8751e+00],
[ 1.5839e-01, 1.1960e+00, 1.5787e+00, ..., 1.9847e+00,
2.8603e-02, 2.4936e+00],
[ 7.0838e-01, 1.4070e+00, 1.0906e+00, ..., 1.2432e+00,
6.4800e-01, 1.8171e+00],
...,
[ 8.6795e-01, 1.2872e+00, 1.5714e+00, ..., -9.4751e-01,
-4.3064e-01, 1.2896e+00],
[ 6.8909e-01, 1.8652e+00, 7.8473e-01, ..., 7.7817e-02,
6.3884e-01, 1.3706e+00],
[ 1.5728e+00, 1.6121e+00, 1.6831e+00, ..., 1.4207e+00,
1.0604e+00, 1.9806e+00]],
...,
[[-1.4460e+00, -4.6818e+00, -4.0295e+00, ..., -3.4093e+00,
-2.4435e+00, -3.8538e+00],
[-1.7919e+00, -3.7853e+00, -4.3212e+00, ..., -5.5906e+00,
-3.8885e+00, -4.3089e+00],
[-8.9718e-01, -3.0489e+00, -3.8413e+00, ..., -2.8072e+00,
-2.4766e+00, -1.6487e+00],
...,
[-1.6395e+00, -4.2589e+00, -4.5198e+00, ..., -3.2489e+00,
-2.1303e+00, -2.2132e+00],
[-1.5120e+00, -4.6653e+00, -5.2988e+00, ..., -4.2897e+00,
-3.3237e+00, -4.2014e+00],
[-1.6242e+00, -4.8560e+00, -4.0275e+00, ..., -3.7452e+00,
-2.4947e+00, -2.8587e+00]],
[[ 7.8057e-01, 5.2644e-01, -1.3350e-01, ..., 9.5051e-01,
1.1373e+00, 1.2634e+00],
[ 1.1851e+00, 9.4460e-01, 4.8624e-01, ..., 1.5818e+00,
2.4211e+00, 2.5793e+00],
[ 5.5846e-01, 5.5054e-01, 4.2808e-01, ..., 1.0099e+00,
1.6824e+00, 1.9396e+00],
...,
[ 7.1791e-01, -1.8297e-01, -2.9627e-01, ..., 2.1701e-01,
1.5150e+00, 1.6770e+00],
[ 1.5416e+00, 1.3617e+00, 1.1853e+00, ..., 6.2029e-01,
1.3965e+00, 2.4182e+00],
[ 6.8225e-01, 8.5915e-01, 1.0498e+00, ..., 9.3524e-01,
1.2831e+00, 1.3523e+00]],
[[-2.8250e-01, -1.6157e-01, -1.0194e+00, ..., -2.8829e-01,
-1.6237e-01, 8.7152e-01],
[-1.4005e+00, -1.4065e+00, -2.0792e+00, ..., 1.2321e-01,
-5.9684e-02, 3.8268e-01],
[-1.6887e+00, -7.3629e-01, -5.4047e-01, ..., 2.1038e-01,
1.7962e-01, 2.5193e-01],
...,
[-1.8462e+00, -1.9778e+00, -1.5608e+00, ..., -6.9120e-01,
-9.3838e-01, -3.2136e-01],
[-1.3835e+00, -1.6857e+00, -1.6924e+00, ..., -1.0226e+00,
-5.2726e-01, 3.6983e-01],
[-1.6277e-01, 2.6295e-02, -5.4231e-01, ..., 4.3256e-01,
2.9568e-01, 3.7096e-01]]],
[[[ 6.6355e-01, 8.5099e-01, 3.3859e-01, ..., 1.3715e+00,
3.7601e-01, -1.8427e+00],
[-1.4067e-01, -2.9526e-01, -7.0084e-02, ..., -2.2070e+00,
2.5093e+00, 5.0359e-02],
[ 1.0952e-01, -7.4082e-02, 4.3566e-01, ..., -8.4783e+00,
-2.1997e+00, 3.0303e+00],
...,
[ 2.5598e-01, 1.9876e+00, 1.6938e+00, ..., 8.2495e-01,
7.0140e-01, 6.0538e-01],
[-1.3841e+00, -1.9637e+00, -2.9162e-01, ..., 1.1717e+00,
1.5706e+00, 1.2038e-01],
[ 8.1807e-01, 2.9488e-01, 5.7848e-01, ..., 4.3688e-01,
3.6792e-01, 6.3607e-01]],
[[-2.2586e+00, -5.0897e-01, -1.5638e+00, ..., -5.2864e+00,
-4.0156e+00, -1.7116e+00],
[-1.8135e+00, 6.1633e-01, -3.9727e-01, ..., -2.6954e+00,
-3.2472e+00, -4.5390e+00],
[-1.0461e+00, 1.1965e+00, 4.9251e-01, ..., -7.2941e+00,
-6.9286e+00, -4.6543e+00],
...,
[ 2.6084e-01, -4.8481e-01, 1.6205e-01, ..., -1.6415e+00,
1.2468e-01, 7.4965e-01],
[-8.1919e+00, -8.5581e+00, -4.8800e+00, ..., 1.1855e-01,
1.5631e-01, 7.5600e-01],
[-2.5353e+00, -2.2916e+00, -2.5859e+00, ..., -6.8314e-01,
-6.6556e-01, -1.3054e-01]],
[[ 1.5734e+00, 1.8613e+00, 1.5185e+00, ..., 2.2380e-01,
2.0805e+00, 4.1634e+00],
[-4.7935e-01, 2.1618e-01, 3.1509e-01, ..., -5.5121e+00,
-3.0601e+00, 2.4847e+00],
[-4.3810e-01, -5.8390e-01, -5.5426e-01, ..., -9.5243e+00,
-9.0772e+00, -1.7851e+00],
...,
[-3.2840e+00, -3.6253e+00, -2.5188e+00, ..., -9.7025e-02,
1.4781e-01, 1.5466e+00],
[-4.8378e+00, -4.5443e+00, -3.6988e+00, ..., 8.4679e-01,
6.3501e-01, 1.6769e+00],
[ 2.5573e+00, 2.7595e+00, 2.8489e+00, ..., 1.7903e+00,
1.0455e+00, 1.7757e+00]],
...,
[[-2.0654e+00, -5.0408e+00, -2.1378e+00, ..., -3.5357e+00,
-3.4693e+00, -3.5463e+00],
[-2.6879e+00, -5.5620e+00, -1.9497e+00, ..., -3.2913e+00,
-9.2708e+00, -1.0965e+01],
[-2.0080e+00, -4.4136e+00, -1.3040e+00, ..., -1.1088e+00,
-4.0156e+00, -1.2829e+01],
...,
[-1.6794e+00, -8.8396e+00, -7.6252e+00, ..., -3.2760e+00,
-2.4801e+00, -2.5698e+00],
[-1.7937e+00, -8.1332e+00, -5.9693e+00, ..., -4.5532e+00,
-3.6693e+00, -4.6825e+00],
[-2.4252e+00, -5.9352e+00, -3.1848e+00, ..., -3.1306e+00,
-1.5921e+00, -3.0995e+00]],
[[ 8.5019e-01, 1.1807e+00, 6.3958e-01, ..., -4.3092e+00,
-2.0781e+00, 1.3934e+00],
[ 1.5455e+00, 7.5805e-01, 1.0138e+00, ..., -2.9621e+00,
-6.9400e+00, -1.3642e+00],
[ 1.4639e+00, 6.4560e-01, 3.4270e-01, ..., -4.4331e-01,
-6.5976e+00, -3.5210e+00],
...,
[ 7.7349e-01, -1.5770e+00, -1.0981e+00, ..., 1.3403e+00,
2.0595e+00, 1.4725e+00],
[ 1.0719e+00, -1.6289e+00, -1.1072e+00, ..., 2.0884e+00,
1.8042e+00, 2.4986e+00],
[ 1.1989e+00, 6.2745e-01, 1.1704e+00, ..., 1.2150e+00,
1.5145e+00, 1.5515e+00]],
[[-6.4177e-02, -5.2238e-01, -1.2444e+00, ..., 4.0440e-01,
-1.9345e+00, -1.9287e+00],
[-8.2697e-01, -1.2382e+00, -2.0336e+00, ..., 1.2741e-01,
2.5928e+00, -1.7636e+00],
[-1.2406e+00, -3.4067e-01, -5.9001e-01, ..., -4.4614e+00,
1.5974e+00, 6.1729e-01],
...,
[-6.4862e-01, -9.4195e-02, 2.8646e-01, ..., -9.7320e-01,
-1.9921e-01, -3.4758e-01],
[-2.5968e+00, -4.1669e+00, -3.7439e+00, ..., -1.8960e-01,
-2.4495e-01, 3.9270e-01],
[-1.1468e+00, 8.3717e-01, -7.9915e-02, ..., 3.1544e-01,
-8.6950e-02, 5.8669e-01]]],
[[[ 9.2461e-01, 1.7694e+00, 9.0071e-01, ..., 6.4462e-01,
8.9935e-01, 4.4236e-01],
[-1.1039e+00, -2.4445e+00, -1.2974e+00, ..., -4.2728e-01,
-1.0263e+00, -7.7272e-01],
[-1.6142e-01, -4.1724e+00, -1.4269e+00, ..., -9.9369e-01,
-1.1112e+00, 8.5054e-02],
...,
[ 4.2261e-01, -2.1189e-01, -2.2312e-01, ..., -1.3026e+00,
6.9523e-01, 3.3824e-01],
[-1.9108e-01, -9.9553e-01, -1.3362e-01, ..., 1.0038e+00,
3.5451e-01, 4.2220e-01],
[ 7.2131e-01, -2.6940e-01, -1.6070e-01, ..., -2.8055e-01,
-8.8244e-01, 1.3332e-01]],
[[-4.4662e+00, -1.7509e+00, -3.3696e+00, ..., -1.4802e+00,
-2.0368e+00, -1.6593e+00],
[-5.2677e+00, -4.3085e+00, -6.6117e+00, ..., -1.8371e+00,
-2.0646e+00, -2.1326e+00],
[-4.8606e+00, -3.4122e+00, -5.1180e+00, ..., -3.7673e+00,
-2.8707e+00, -1.1948e+00],
...,
[-9.0970e-01, -7.6428e-01, -1.1901e+00, ..., -2.7565e-01,
-1.8275e+00, -2.4799e-01],
[-2.9336e+00, -5.6823e-01, -8.6594e-01, ..., -4.9791e-01,
-1.0567e+00, -1.1053e+00],
[-2.1085e+00, -1.4947e+00, -1.3864e+00, ..., -5.4569e-01,
-5.7409e-01, -5.3161e-01]],
[[ 3.0840e+00, 4.4134e+00, 3.0648e+00, ..., 1.6509e+00,
8.9945e-01, 2.5313e+00],
[ 6.0221e-01, 9.7420e-01, 2.4152e+00, ..., 2.5172e+00,
1.7328e+00, 2.7838e+00],
[ 1.1102e+00, -7.2932e-02, -1.3996e+00, ..., 1.2204e+00,
8.4652e-01, 9.0974e-01],
...,
[ 1.1021e-01, 9.6177e-01, 1.1664e+00, ..., -1.5201e+00,
-1.3974e+00, -3.0342e-01],
[ 2.6535e-01, 2.1355e+00, 1.5227e+00, ..., 6.0370e-01,
9.5222e-01, 1.3686e+00],
[ 1.3847e+00, 1.3877e+00, 1.6716e+00, ..., 8.3849e-01,
1.5199e+00, 1.5723e+00]],
...,
[[-2.2019e+00, -7.3737e+00, -5.6424e+00, ..., -5.0131e+00,
-3.6323e+00, -4.1729e+00],
[-3.3047e+00, -6.6886e+00, -4.4756e+00, ..., -5.6634e+00,
-4.6515e+00, -3.5489e+00],
[ 9.1790e-01, -1.8364e+00, -3.8181e+00, ..., -3.3496e+00,
-2.4802e+00, -7.7429e-01],
...,
[-9.1493e-01, -2.2472e+00, -1.7230e+00, ..., -3.8899e+00,
-4.3032e+00, -2.0453e+00],
[-1.2713e+00, -3.7112e+00, -3.4476e+00, ..., -5.8806e+00,
-5.0314e+00, -4.7985e+00],
[-2.1977e+00, -5.0460e+00, -3.1423e+00, ..., -3.4573e+00,
-2.6522e+00, -2.6350e+00]],
[[-7.6444e-01, -7.1965e-01, -5.0130e-01, ..., 1.5537e+00,
1.6595e+00, 1.2360e+00],
[ 7.2566e-01, 3.7897e-02, -1.5635e+00, ..., 1.2553e+00,
1.4100e+00, 1.8216e+00],
[-1.1589e+00, -7.4698e-01, -2.4924e-01, ..., -7.6826e-01,
-5.3827e-01, 1.0739e+00],
...,
[ 1.2163e+00, -1.8173e-01, 1.0765e+00, ..., 1.7426e+00,
9.9341e-01, 9.2601e-01],
[ 1.8216e+00, 8.3782e-01, 1.7960e+00, ..., 1.6689e+00,
1.1307e+00, 1.6140e+00],
[ 1.0314e+00, 3.4369e-01, 9.7417e-01, ..., 7.9232e-01,
7.4253e-01, 1.5515e+00]],
[[-3.5652e-01, 5.5223e-01, 1.7618e-01, ..., -4.4353e-01,
-6.3116e-01, -2.2492e-02],
[-1.1019e+00, -1.4972e+00, -2.3907e+00, ..., -1.4438e+00,
-1.9996e+00, -1.2818e+00],
[-1.7366e+00, -5.9382e-01, -2.2647e+00, ..., -8.0864e-01,
-2.0436e+00, -7.3329e-01],
...,
[-1.6211e+00, -1.4981e+00, -1.0544e+00, ..., -1.4492e+00,
-2.0446e+00, -1.1435e+00],
[-1.6547e+00, -1.0687e+00, -8.6505e-01, ..., -6.7160e-01,
-1.5871e+00, -5.5566e-01],
[-3.3331e-01, 4.4195e-01, -3.0573e-01, ..., -2.5669e-01,
-6.9645e-01, 1.6183e-01]]],
[[[ 8.5191e-01, 1.5037e+00, 1.1768e+00, ..., 1.4075e+00,
1.0711e+00, 9.7161e-01],
[-1.0426e+00, -2.0268e+00, -1.1663e+00, ..., -3.3413e-01,
-5.6321e-01, -1.6082e+00],
[-2.0060e-01, -7.7431e-01, 3.2530e-01, ..., 1.9741e-01,
9.7991e-01, 8.0633e-01],
...,
[-5.0037e-02, -1.2329e+00, -1.0828e+00, ..., -1.9391e-01,
-1.3588e-01, 1.5092e-01],
[-6.1375e-01, -1.8716e+00, -1.2466e+00, ..., -6.1836e-01,
-3.1301e-01, -1.5220e-01],
[-3.2159e-01, -1.7172e+00, -1.6481e+00, ..., -1.2016e+00,
-1.5737e+00, -3.1606e-01]],
[[-3.7670e+00, -7.7399e-01, -1.5626e+00, ..., -1.5731e+00,
-2.4501e+00, -1.2134e+00],
[-4.0447e+00, -2.8742e+00, -1.8067e+00, ..., -2.2401e+00,
-3.0740e+00, -9.1133e-01],
[-2.3838e+00, -1.6429e+00, -1.8616e+00, ..., -1.6208e+00,
-2.6248e+00, -1.0227e+00],
...,
[-1.6174e+00, -1.6066e+00, -2.5902e+00, ..., -2.0580e+00,
-2.1092e+00, -6.7112e-01],
[-3.0376e+00, -1.5827e+00, -2.7713e+00, ..., -2.4591e+00,
-2.4410e+00, -1.2199e+00],
[-2.3113e+00, -1.9026e+00, -2.6816e+00, ..., -2.0183e+00,
-2.0477e+00, -1.0268e+00]],
[[ 2.9177e+00, 4.2570e+00, 3.3652e+00, ..., 2.9339e+00,
1.0647e+00, 3.0288e+00],
[ 2.5862e-01, 1.0822e+00, 2.2556e+00, ..., 1.3372e+00,
-7.9338e-01, 2.7445e+00],
[ 1.0049e+00, 1.4803e+00, 1.5129e+00, ..., 8.1413e-01,
-4.4469e-01, 1.8675e+00],
...,
[ 9.6831e-01, 1.0397e+00, 1.1603e+00, ..., 7.3799e-01,
1.2070e+00, 2.3370e+00],
[ 6.9214e-01, 1.8012e+00, 1.1145e+00, ..., 4.8470e-02,
1.0811e+00, 1.8329e+00],
[ 1.6410e+00, 1.6247e+00, 1.8529e+00, ..., 1.2907e+00,
1.7551e+00, 1.7425e+00]],
...,
[[-1.9530e+00, -7.4100e+00, -5.4930e+00, ..., -4.1025e+00,
-3.2849e+00, -5.5025e+00],
[-2.4722e+00, -7.1016e+00, -4.3105e+00, ..., -3.7064e+00,
-2.7875e+00, -4.2820e+00],
[-1.0096e+00, -4.6076e+00, -3.8309e+00, ..., -2.9914e+00,
-3.3683e+00, -2.5233e+00],
...,
[-2.5073e+00, -5.3001e+00, -4.3899e+00, ..., -6.8597e+00,
-5.5269e+00, -4.6486e+00],
[-2.7294e+00, -5.6905e+00, -5.3235e+00, ..., -6.2674e+00,
-5.6956e+00, -6.2089e+00],
[-1.4864e+00, -4.0450e+00, -2.7832e+00, ..., -3.3032e+00,
-2.7266e+00, -3.2265e+00]],
[[-1.9716e-01, 3.1034e-01, 5.4383e-01, ..., 6.5756e-01,
8.6029e-01, 1.4124e+00],
[ 1.6368e+00, 1.0793e+00, 1.8167e+00, ..., 1.8937e+00,
2.5503e+00, 3.0386e+00],
[ 1.2337e+00, 3.1772e-01, 5.0839e-01, ..., 1.4762e+00,
1.4247e+00, 2.2476e+00],
...,
[ 1.2983e+00, 2.8500e-01, -1.2695e-01, ..., 1.4911e+00,
1.1805e+00, 1.6281e+00],
[ 1.9092e+00, 1.6749e+00, 1.0409e+00, ..., 1.4452e+00,
1.2957e+00, 1.9735e+00],
[ 4.5121e-01, 8.1787e-01, 7.0716e-01, ..., 5.4201e-01,
7.0276e-01, 1.6487e+00]],
[[-5.6059e-01, 3.0268e-01, 1.2387e-01, ..., 5.9087e-01,
5.2347e-01, 1.3518e+00],
[-1.4167e+00, -1.8503e+00, -1.6277e+00, ..., -6.2280e-01,
-4.1100e-01, 2.3122e-01],
[-2.2723e+00, -8.3798e-01, 6.7988e-02, ..., 1.8707e-01,
-1.0584e-02, 6.5835e-01],
...,
[-1.9669e+00, -1.8664e+00, -1.4212e+00, ..., -1.9470e-02,
-7.0536e-01, -7.8729e-01],
[-1.5727e+00, -2.0103e+00, -1.7252e+00, ..., -1.1971e+00,
-1.3729e+00, -6.8219e-01],
[-8.3813e-01, -8.5218e-01, -1.1772e+00, ..., -6.1375e-01,
-8.9086e-01, -1.0884e-01]]],
[[[ 8.8040e-01, 1.5325e+00, 1.1554e+00, ..., 1.2279e+00,
9.5286e-01, 1.0019e+00],
[-9.4304e-01, -2.2638e+00, -1.4028e+00, ..., 4.2581e-02,
-6.2460e-01, -1.5362e+00],
[ 5.1605e-02, -1.0569e+00, 2.2826e-01, ..., 6.2474e-02,
6.0487e-01, 7.6161e-01],
...,
[ 8.5556e-01, 4.8479e-02, -5.0951e-01, ..., 5.9995e-01,
4.4131e-01, 6.7081e-01],
[ 2.0099e-01, 3.5929e-01, -9.4733e-01, ..., -2.8402e-01,
4.6736e-01, 3.1833e-02],
[ 3.8090e-01, 2.0556e+00, -1.2595e+00, ..., 3.1253e-01,
-3.2513e-01, 6.3664e-01]],
[[-4.2483e+00, -6.9814e-01, -2.2390e+00, ..., -1.4660e+00,
-2.2308e+00, -1.0488e+00],
[-4.2304e+00, -2.3870e+00, -2.3368e+00, ..., -2.4397e+00,
-2.8765e+00, -8.0424e-01],
[-2.6417e+00, -1.4596e+00, -1.9985e+00, ..., -3.0886e+00,
-2.5104e+00, -9.9784e-01],
...,
[-1.3804e+00, -1.8237e+00, 1.4538e+00, ..., -1.3420e+00,
-2.2157e+00, -6.2179e-01],
[-1.8019e+00, -1.4672e+00, 1.8489e+00, ..., -2.2089e+00,
-1.6604e+00, -1.3882e-01],
[-2.1434e-01, -1.3139e+00, 1.4597e+00, ..., -1.0051e+00,
-3.9418e-01, 1.9947e-01]],
[[ 3.3287e+00, 4.4071e+00, 3.6312e+00, ..., 2.5153e+00,
1.2116e+00, 2.8681e+00],
[ 7.8598e-01, 9.8401e-01, 2.7219e+00, ..., 1.0796e+00,
-7.3964e-01, 2.5705e+00],
[ 1.5782e+00, 1.5000e+00, 2.0279e+00, ..., 1.2298e+00,
4.2368e-02, 2.0437e+00],
...,
[ 1.5751e+00, -2.0277e+00, -1.4187e+00, ..., -1.5243e+00,
6.8192e-01, 7.4292e-01],
[ 1.3140e+00, -3.5536e+00, -4.1507e+00, ..., -1.8026e+00,
3.6874e-01, 8.7548e-01],
[ 2.8130e+00, -2.6278e+00, -4.7207e+00, ..., 1.0312e+00,
5.7397e-01, 1.0053e+00]],
...,
[[-2.6368e+00, -7.7552e+00, -5.7154e+00, ..., -5.3379e+00,
-2.8248e+00, -5.3821e+00],
[-3.2725e+00, -7.1197e+00, -4.7782e+00, ..., -5.7987e+00,
-2.7725e+00, -4.1497e+00],
[-1.4234e+00, -4.6653e+00, -4.3126e+00, ..., -4.2203e+00,
-3.1832e+00, -2.4289e+00],
...,
[-2.4295e+00, -6.2558e+00, -5.1102e+00, ..., -2.4070e+00,
-4.0007e+00, -4.5360e+00],
[-2.4368e+00, -7.5384e+00, -5.6402e+00, ..., -5.0030e+00,
-4.3506e+00, -5.5627e+00],
[-2.1094e+00, -7.0960e+00, -3.9397e+00, ..., -5.4148e+00,
-4.0417e+00, -5.0147e+00]],
[[-3.0760e-01, 3.7689e-01, 1.7564e-01, ..., 1.2146e+00,
1.0155e+00, 1.4362e+00],
[ 1.9190e+00, 1.2595e+00, 1.2737e+00, ..., 1.7844e+00,
2.5921e+00, 2.9661e+00],
[ 1.3725e+00, 3.3784e-01, 7.8207e-01, ..., 1.1811e+00,
1.9375e+00, 2.3826e+00],
...,
[ 4.1207e-01, 1.0726e+00, 8.0959e-01, ..., 1.1879e+00,
8.6409e-01, 1.7484e+00],
[ 9.9494e-01, 1.8515e+00, 3.6653e-01, ..., 1.9823e-01,
7.8741e-02, 2.1312e+00],
[ 1.0398e+00, 2.3890e+00, 1.3169e+00, ..., 1.2766e+00,
1.3719e+00, 1.4549e+00]],
[[-7.0718e-01, 3.1083e-01, -3.1319e-01, ..., -2.7460e-01,
5.0420e-01, 1.3688e+00],
[-1.6941e+00, -1.7638e+00, -1.8138e+00, ..., -1.5267e+00,
-5.4211e-01, 2.1494e-01],
[-2.4868e+00, -9.6203e-01, -1.6399e-01, ..., -2.9553e-01,
2.0494e-01, 7.2526e-01],
...,
[-1.8638e+00, -1.8603e+00, 1.2317e+00, ..., -3.3630e-01,
-6.2998e-01, -1.4833e+00],
[-1.4652e+00, -2.1265e+00, 1.9736e+00, ..., -2.2048e+00,
-1.7871e+00, -2.2973e-01],
[ 3.3476e-01, -7.0047e-01, -2.8628e-02, ..., -2.1850e-01,
2.9128e-02, 9.1186e-01]]]], device='cuda:0'),
tensor([[[[ 5.4358e-01, 1.3720e+00, 1.2015e+00, ..., 4.0755e-01,
4.4708e-01, 2.9782e-01],
[-7.7691e-01, -3.2058e+00, -3.2996e+00, ..., 4.9340e-02,
2.6824e-01, -1.3388e+00],
[-8.7549e-01, 1.7132e+00, -8.2622e-01, ..., -4.0407e-01,
1.0325e+00, 6.6540e-01],
...,
[ 6.2323e-01, -2.2830e+00, -1.5134e+00, ..., 4.4740e-01,
-6.8588e-01, 3.4017e-01],
[ 5.9899e-01, -4.6549e-01, 1.8528e-01, ..., -6.1124e-01,
-4.0372e-01, -6.4080e-01],
[-1.3905e-01, -1.5500e-01, -9.1757e-01, ..., 2.1913e-01,
-1.1307e-01, 5.5564e-01]],
[[-3.8955e+00, -1.3344e+00, -2.2536e+00, ..., -1.6346e+00,
-1.9303e+00, -9.5622e-01],
[-7.9175e+00, -4.4582e+00, -2.4459e+00, ..., -7.2168e-01,
-1.3442e+00, -2.9677e-01],
[-4.4752e+00, -6.0183e+00, -6.8038e+00, ..., -1.7039e+00,
-1.5427e+00, -8.6416e-01],
...,
[-2.6350e+00, -3.4397e+00, -2.0249e+00, ..., -5.9280e+00,
-6.4120e+00, -1.9854e+00],
[-1.4032e+00, -4.9735e-01, -1.3125e-01, ..., -3.6682e+00,
-4.4922e+00, -5.0906e-01],
[-3.4601e-01, -1.0996e+00, -1.0543e+00, ..., -1.3322e+00,
-1.6506e+00, -1.0704e+00]],
[[ 2.4391e+00, 2.5411e+00, 1.3296e+00, ..., 2.9725e+00,
1.4028e+00, 2.9037e+00],
[ 3.9790e-03, 5.2947e-01, 3.4408e+00, ..., 1.5379e+00,
-8.2868e-03, 2.4955e+00],
[-1.5130e+00, -2.3887e+00, 1.4935e+00, ..., 6.0241e-01,
6.7139e-01, 1.9077e+00],
...,
[-1.6667e+00, 4.3106e-01, 1.8308e+00, ..., -7.4827e+00,
-1.9056e+00, 1.2027e+00],
[-5.5809e-01, 9.1978e-01, 7.4682e-01, ..., -3.0775e+00,
2.7333e-01, 1.9609e+00],
[ 8.1882e-01, 4.9542e-01, 1.0851e+00, ..., 1.7032e+00,
1.4276e+00, 1.7751e+00]],
...,
[[-4.0331e+00, -9.1194e+00, -6.8814e+00, ..., -3.5535e+00,
-2.4911e+00, -3.9912e+00],
[-4.3171e+00, -1.1789e+01, -7.3152e+00, ..., -6.0973e+00,
-3.6981e+00, -4.2020e+00],
[-2.5270e+00, -1.2567e+01, -1.1483e+01, ..., -4.7624e+00,
-2.4173e+00, -1.6967e+00],
...,
[-3.0325e+00, -8.8043e+00, -7.9670e+00, ..., -4.1078e+00,
-7.7274e+00, -7.1895e+00],
[-3.6476e+00, -1.0042e+01, -7.2440e+00, ..., -7.4309e+00,
-5.1938e+00, -6.9500e+00],
[-2.8905e+00, -7.4371e+00, -3.2209e+00, ..., -6.1736e+00,
-5.3239e+00, -5.6303e+00]],
[[-5.8609e-01, 1.3978e+00, 5.6166e-01, ..., 1.3370e+00,
1.1939e+00, 1.2480e+00],
[-2.3571e+00, -5.9767e-01, 2.4761e+00, ..., 1.8461e+00,
2.3805e+00, 2.5664e+00],
[-4.2322e+00, -9.0280e+00, -4.2763e+00, ..., 1.8543e+00,
1.6753e+00, 1.9561e+00],
...,
[ 2.8667e-01, -5.3384e-01, 2.7711e+00, ..., 5.2580e-01,
-6.7662e-01, 2.1204e+00],
[ 2.2470e+00, 2.2337e+00, 2.2713e+00, ..., -4.7098e-01,
3.4695e-01, 3.5009e+00],
[ 8.7024e-01, 5.3686e-01, 1.0925e+00, ..., 1.3385e+00,
1.4391e+00, 1.5570e+00]],
[[ 1.5448e-01, -3.4286e-01, -6.9840e-01, ..., -2.9181e-02,
-9.5482e-02, 8.8436e-01],
[-4.9522e-01, -3.4864e+00, -2.5822e+00, ..., -6.4206e-02,
-1.9600e-01, 2.6858e-01],
[-1.9489e+00, 2.2956e+00, -2.1084e+00, ..., 6.7008e-01,
1.7203e-01, 2.6325e-01],
...,
[-1.7052e+00, -4.0645e+00, -3.2072e+00, ..., -1.6464e+00,
-1.0122e+00, -2.3958e+00],
[-2.1773e+00, -2.2431e+00, -2.3576e-01, ..., -3.1936e+00,
-1.9901e+00, 2.1534e-01],
[-3.2785e-01, -1.0563e+00, -6.7044e-01, ..., -1.5129e-01,
7.7214e-01, 1.1503e+00]]],
[[[ 8.3557e-01, 1.4780e+00, 1.1877e+00, ..., 1.4111e+00,
1.1660e+00, 5.9563e-01],
[-9.6425e-01, -2.0008e+00, -1.1826e+00, ..., 1.6930e-01,
2.4285e-01, -1.6542e+00],
[-1.8606e-01, -7.0550e-01, 2.5528e-01, ..., -1.4287e+00,
1.3864e+00, 1.3510e+00],
...,
[-1.3163e-02, -2.3548e+00, -4.6575e-01, ..., -4.3517e-01,
-6.0363e-02, 2.3569e-01],
[-6.0555e-01, -2.4700e+00, -1.1136e+00, ..., -6.2648e-01,
-3.3929e-01, -1.7194e-01],
[-3.4722e-01, -1.9215e+00, -1.7684e+00, ..., -1.1487e+00,
-1.5452e+00, -2.9730e-01]],
[[-3.8070e+00, -7.9960e-01, -1.6329e+00, ..., -2.5466e+00,
-3.2596e+00, -1.4015e+00],
[-4.0501e+00, -2.8608e+00, -1.9432e+00, ..., -1.3320e+00,
-4.2676e+00, -1.8697e+00],
[-2.4832e+00, -1.6571e+00, -1.8260e+00, ..., -2.0013e+00,
-3.9168e+00, -2.0417e+00],
...,
[-1.5203e+00, -2.4055e+00, -4.6131e+00, ..., -2.2188e+00,
-2.1605e+00, -7.3225e-01],
[-2.9653e+00, -2.3319e+00, -4.4870e+00, ..., -2.4410e+00,
-2.2873e+00, -1.2252e+00],
[-2.1905e+00, -2.4885e+00, -3.5588e+00, ..., -2.0136e+00,
-1.9117e+00, -1.0182e+00]],
[[ 2.9337e+00, 4.1935e+00, 3.3830e+00, ..., 1.0084e+00,
7.4619e-01, 3.1911e+00],
[ 3.0065e-01, 9.5441e-01, 2.2477e+00, ..., -2.0546e+00,
-3.1098e+00, 2.7555e+00],
[ 1.1187e+00, 1.4908e+00, 1.8042e+00, ..., -3.1616e+00,
-2.6360e+00, 1.9051e+00],
...,
[ 1.1190e+00, 2.4823e+00, 1.0876e+00, ..., 1.6813e-01,
1.1323e+00, 2.3778e+00],
[ 7.7187e-01, 2.0786e+00, 2.2430e+00, ..., 2.4938e-02,
1.0272e+00, 1.8258e+00],
[ 1.6426e+00, 1.5484e+00, 2.4716e+00, ..., 1.3330e+00,
1.7441e+00, 1.7557e+00]],
...,
[[-1.8821e+00, -7.2232e+00, -5.4658e+00, ..., -3.1247e+00,
-2.2355e+00, -4.9694e+00],
[-2.4936e+00, -6.9136e+00, -4.1715e+00, ..., -4.6911e+00,
-2.4378e+00, -4.7568e+00],
[-1.0029e+00, -4.2092e+00, -3.5485e+00, ..., 3.2085e-01,
-1.2665e+00, -3.3677e+00],
...,
[-2.5782e+00, -5.3986e+00, -4.9144e+00, ..., -4.9901e+00,
-4.5790e+00, -4.7111e+00],
[-2.6303e+00, -6.0797e+00, -5.6713e+00, ..., -6.3261e+00,
-5.7502e+00, -6.2582e+00],
[-1.4012e+00, -4.0916e+00, -2.8169e+00, ..., -3.3601e+00,
-2.7750e+00, -3.2467e+00]],
[[-1.8489e-01, 3.4573e-01, 4.8128e-01, ..., -7.6581e-01,
5.2812e-01, 1.5346e+00],
[ 1.6889e+00, 1.0295e+00, 1.7732e+00, ..., 1.2165e-01,
1.0421e+00, 3.0191e+00],
[ 1.2694e+00, 2.2229e-01, 7.5406e-01, ..., -1.2691e+00,
-2.0658e+00, 1.9234e+00],
...,
[ 1.2919e+00, -7.0131e-01, 4.0599e-02, ..., 1.1470e+00,
1.1165e+00, 1.6520e+00],
[ 1.8379e+00, 9.9244e-01, -2.5266e-01, ..., 1.4737e+00,
1.3244e+00, 1.9819e+00],
[ 4.8879e-01, 5.0927e-01, 9.3643e-02, ..., 4.9573e-01,
6.6982e-01, 1.6203e+00]],
[[-5.9246e-01, 3.2656e-01, 1.3812e-01, ..., 1.1022e+00,
7.5207e-01, 1.2340e+00],
[-1.3899e+00, -1.7355e+00, -1.6278e+00, ..., 1.2025e+00,
8.8269e-01, 6.1685e-01],
[-2.3962e+00, -8.2623e-01, 5.7212e-01, ..., 1.9210e-01,
1.2017e+00, 1.2216e+00],
...,
[-1.9142e+00, -1.5632e+00, -9.0180e-01, ..., -1.0502e+00,
-1.1315e+00, -8.8039e-01],
[-1.5357e+00, -2.3738e+00, -1.2947e+00, ..., -1.1996e+00,
-1.2538e+00, -6.9089e-01],
[-7.7096e-01, -1.1050e+00, -1.6087e+00, ..., -5.9932e-01,
-8.1594e-01, -9.5935e-02]]],
[[[ 7.3544e-01, 1.0785e+00, 7.2868e-01, ..., 3.9501e-01,
2.9691e-01, 1.6909e-01],
[-3.0046e-01, -1.3449e+00, -6.0271e-01, ..., 4.2570e-01,
4.3132e-01, -1.1636e+00],
[ 1.3712e-01, -7.6992e-01, -3.9727e-01, ..., -4.8529e-01,
9.9210e-01, 7.0680e-01],
...,
[ 1.6344e+00, -1.6223e+00, -1.8306e+00, ..., -7.5616e-01,
6.0954e-01, 5.1585e-01],
[-2.1559e-01, -1.5145e+00, -4.4213e-01, ..., -3.4192e-02,
1.4134e+00, -7.7559e-02],
[ 3.9283e-01, -5.6823e-01, -6.9052e-01, ..., -2.6202e-01,
3.4422e-01, 5.8930e-01]],
[[-2.9694e+00, -1.4919e+00, -3.0955e+00, ..., -1.8566e+00,
-2.2047e+00, -1.0256e+00],
[-4.1905e+00, -1.8296e+00, -2.3916e+00, ..., -4.3787e-01,
-1.1969e+00, -3.1481e-01],
[-4.2916e+00, -2.6518e+00, -3.1392e+00, ..., -1.4982e+00,
-1.3180e+00, -7.4140e-01],
...,
[-5.7961e+00, -6.8337e+00, -8.1544e+00, ..., -5.3124e-01,
-4.4067e-02, 5.3312e-01],
[-4.1479e+00, -1.9912e+00, -2.8493e+00, ..., -4.5880e-01,
-4.3895e-01, 2.3254e-01],
[-1.2845e+00, -8.5509e-01, -1.6856e+00, ..., -3.3781e-01,
-7.1891e-01, -5.7001e-01]],
[[ 1.6595e+00, 2.7551e+00, 2.3950e+00, ..., 2.8309e+00,
1.4537e+00, 2.8976e+00],
[ 1.5998e-01, 1.1324e+00, 1.5282e+00, ..., 1.9013e+00,
-2.8022e-02, 2.4526e+00],
[ 6.7371e-01, 1.4702e+00, 8.9648e-01, ..., 1.1082e+00,
5.4398e-01, 1.8083e+00],
...,
[ 1.5450e+00, 2.5427e-01, -6.9194e-01, ..., 7.1281e-01,
1.9832e-01, 1.4651e+00],
[ 1.4531e+00, 3.3124e+00, 2.6973e+00, ..., 1.0041e+00,
5.4815e-01, 1.4998e+00],
[ 1.5434e+00, 1.5621e+00, 1.6328e+00, ..., 1.4564e+00,
1.0281e+00, 2.0041e+00]],
...,
[[-1.3430e+00, -4.5215e+00, -3.8301e+00, ..., -3.3909e+00,
-2.4547e+00, -3.8186e+00],
[-1.7067e+00, -3.8321e+00, -4.1831e+00, ..., -5.4566e+00,
-3.9659e+00, -4.3164e+00],
[-9.6076e-01, -3.3183e+00, -3.8498e+00, ..., -2.6334e+00,
-2.3485e+00, -1.5853e+00],
...,
[-2.4242e+00, -4.6928e+00, -5.9048e+00, ..., -6.7350e+00,
-5.1035e+00, -3.0016e+00],
[-1.8428e+00, -5.5269e+00, -7.1807e+00, ..., -4.2862e+00,
-3.2935e+00, -3.7469e+00],
[-1.6745e+00, -4.9686e+00, -4.1616e+00, ..., -3.5033e+00,
-2.1440e+00, -2.7678e+00]],
[[ 7.9358e-01, 5.3938e-01, -1.3389e-01, ..., 8.5110e-01,
1.1159e+00, 1.2917e+00],
[ 1.2030e+00, 9.6309e-01, 5.8573e-01, ..., 1.5802e+00,
2.3590e+00, 2.5574e+00],
[ 5.8475e-01, 5.2294e-01, 9.8084e-02, ..., 1.0684e+00,
1.6500e+00, 1.9029e+00],
...,
[-1.0400e+00, -1.8256e+00, -2.4615e+00, ..., 1.8557e+00,
2.0098e+00, 1.7872e+00],
[ 2.0868e+00, 2.7711e+00, 2.9993e+00, ..., 1.2569e+00,
1.4548e+00, 2.4113e+00],
[ 6.5826e-01, 8.9861e-01, 9.7804e-01, ..., 1.0580e+00,
1.3180e+00, 1.3484e+00]],
[[-2.6065e-01, -2.0109e-01, -9.5452e-01, ..., -2.0401e-01,
-1.3800e-01, 8.7650e-01],
[-1.3518e+00, -1.5071e+00, -2.0447e+00, ..., 2.0724e-01,
2.4805e-02, 3.8332e-01],
[-1.6149e+00, -8.8506e-01, -1.1284e+00, ..., 2.1146e-01,
2.0241e-01, 2.1438e-01],
...,
[-1.5917e+00, -1.9799e+00, -4.0571e+00, ..., -3.8910e-01,
-2.0545e-02, 4.4420e-02],
[-1.7160e+00, -1.9966e+00, -2.1390e+00, ..., -7.3991e-01,
-4.0416e-01, 3.2605e-01],
[-1.5370e-01, 5.5788e-02, -5.2540e-01, ..., 2.3035e-01,
1.3570e-01, 3.3621e-01]]],
[[[ 8.5726e-01, 1.4931e+00, 1.1979e+00, ..., 1.4148e+00,
1.0816e+00, 9.7151e-01],
[-9.8552e-01, -2.0368e+00, -1.1878e+00, ..., -2.7455e-01,
-5.3559e-01, -1.5408e+00],
[-1.5963e-01, -8.0352e-01, 1.1918e-01, ..., -5.1711e-02,
6.8461e-01, 8.5163e-01],
...,
[-4.0010e-02, -1.1222e+00, -1.1665e+00, ..., -5.1751e-01,
-9.0276e-03, 2.7304e-01],
[-6.6095e-01, -1.9113e+00, -1.3077e+00, ..., -5.8925e-01,
-3.3771e-01, -1.6964e-01],
[-3.1758e-01, -1.6826e+00, -1.6419e+00, ..., -1.1644e+00,
-1.5760e+00, -2.9330e-01]],
[[-3.7804e+00, -7.3553e-01, -1.6003e+00, ..., -1.5632e+00,
-2.4657e+00, -1.2301e+00],
[-3.9437e+00, -2.7296e+00, -1.9150e+00, ..., -2.1907e+00,
-3.1111e+00, -9.6751e-01],
[-2.2972e+00, -1.5906e+00, -1.8049e+00, ..., -2.2361e+00,
-2.8788e+00, -1.0341e+00],
...,
[-1.5142e+00, -1.5872e+00, -2.3754e+00, ..., -1.9087e+00,
-2.1576e+00, -7.2440e-01],
[-3.0905e+00, -1.6045e+00, -2.7804e+00, ..., -2.3906e+00,
-2.4347e+00, -1.2643e+00],
[-2.3515e+00, -1.9458e+00, -2.7323e+00, ..., -2.0195e+00,
-1.9943e+00, -1.0188e+00]],
[[ 2.9302e+00, 4.2145e+00, 3.4019e+00, ..., 2.9040e+00,
1.0331e+00, 3.0453e+00],
[ 3.1996e-01, 1.0380e+00, 2.2987e+00, ..., 1.3177e+00,
-8.1134e-01, 2.7704e+00],
[ 1.0771e+00, 1.4779e+00, 1.6112e+00, ..., 1.3838e-01,
-3.3076e-01, 2.0877e+00],
...,
[ 8.5621e-01, 1.0709e+00, 1.4182e+00, ..., 6.7361e-02,
1.0558e+00, 2.2987e+00],
[ 6.1687e-01, 1.7764e+00, 1.1196e+00, ..., 6.6325e-02,
1.0476e+00, 1.8076e+00],
[ 1.6281e+00, 1.6585e+00, 1.8317e+00, ..., 1.3082e+00,
1.8122e+00, 1.7847e+00]],
...,
[[-1.9352e+00, -7.2675e+00, -5.4123e+00, ..., -4.0769e+00,
-3.2043e+00, -5.4935e+00],
[-2.4995e+00, -6.8768e+00, -4.1590e+00, ..., -3.6894e+00,
-2.7452e+00, -4.3091e+00],
[-9.7927e-01, -4.4062e+00, -3.5498e+00, ..., -3.1574e+00,
-2.9638e+00, -2.4693e+00],
...,
[-2.5047e+00, -5.3518e+00, -4.3313e+00, ..., -5.9128e+00,
-4.4411e+00, -4.5734e+00],
[-2.7335e+00, -5.6924e+00, -5.4036e+00, ..., -6.2735e+00,
-5.7242e+00, -6.2204e+00],
[-1.4908e+00, -4.1254e+00, -2.7889e+00, ..., -3.2980e+00,
-2.7119e+00, -3.2294e+00]],
[[-1.8940e-01, 3.3218e-01, 5.3974e-01, ..., 6.6998e-01,
8.5414e-01, 1.3952e+00],
[ 1.6690e+00, 1.1105e+00, 1.7998e+00, ..., 1.8765e+00,
2.5475e+00, 3.0020e+00],
[ 1.2479e+00, 2.6057e-01, 7.5823e-01, ..., 9.4700e-01,
1.5042e+00, 2.3622e+00],
...,
[ 1.2803e+00, 1.9784e-01, -1.8929e-01, ..., 1.1459e+00,
1.0620e+00, 1.5922e+00],
[ 1.8655e+00, 1.6695e+00, 1.0004e+00, ..., 1.4050e+00,
1.2475e+00, 1.9249e+00],
[ 3.9952e-01, 7.7367e-01, 7.2666e-01, ..., 5.2487e-01,
7.0337e-01, 1.6718e+00]],
[[-5.6999e-01, 3.2048e-01, 1.2544e-01, ..., 5.7632e-01,
5.4443e-01, 1.3627e+00],
[-1.3978e+00, -1.7738e+00, -1.6339e+00, ..., -6.3687e-01,
-4.3835e-01, 2.4867e-01],
[-2.2486e+00, -8.9125e-01, 2.6011e-01, ..., 4.9097e-02,
5.6940e-03, 7.8621e-01],
...,
[-1.9367e+00, -1.9090e+00, -1.3491e+00, ..., -6.5410e-01,
-1.2130e+00, -8.5039e-01],
[-1.5685e+00, -2.1079e+00, -1.7564e+00, ..., -1.0647e+00,
-1.3996e+00, -7.0759e-01],
[-8.5191e-01, -8.5063e-01, -1.1815e+00, ..., -5.9293e-01,
-8.9071e-01, -9.4022e-02]]],
[[[ 7.2271e-01, 1.0764e+00, 7.4828e-01, ..., 3.4322e-01,
3.1438e-01, 2.1911e-01],
[-3.8255e-01, -1.4527e+00, -5.1281e-01, ..., 4.3049e-01,
3.2973e-01, -1.1848e+00],
[ 3.5608e-02, -7.4226e-01, -1.6000e-01, ..., -3.6352e-01,
1.0180e+00, 6.0624e-01],
...,
[ 8.3524e-02, -1.2018e+00, -7.0047e-01, ..., 1.1985e+00,
1.2747e+00, 4.8786e-01],
[-3.9785e-01, -1.8317e+00, -7.7408e-01, ..., -4.9244e-01,
1.0329e+00, -1.8374e-01],
[ 4.1334e-01, -5.1449e-01, -6.9475e-01, ..., -2.5287e-01,
2.1724e-01, 5.0464e-01]],
[[-3.1060e+00, -1.4887e+00, -3.0126e+00, ..., -1.7656e+00,
-2.2362e+00, -1.0875e+00],
[-4.4417e+00, -1.7188e+00, -2.1500e+00, ..., -5.2716e-01,
-1.2020e+00, -2.7306e-01],
[-4.2059e+00, -2.2447e+00, -3.2590e+00, ..., -1.5176e+00,
-1.4227e+00, -7.2593e-01],
...,
[-2.7790e+00, -1.8042e+00, -2.6566e+00, ..., -5.7626e-01,
-1.5996e+00, 2.8845e-01],
[-3.5664e+00, -1.1854e+00, -1.9947e+00, ..., -1.9931e+00,
-1.2224e+00, 3.0288e-01],
[-1.2399e+00, -7.9012e-01, -1.5817e+00, ..., -5.9058e-01,
-7.7763e-01, -5.8214e-01]],
[[ 1.7578e+00, 2.8540e+00, 2.3937e+00, ..., 2.8892e+00,
1.4313e+00, 2.9262e+00],
[ 2.2678e-01, 1.1200e+00, 1.5489e+00, ..., 1.8810e+00,
4.2244e-02, 2.5334e+00],
[ 7.8891e-01, 1.1644e+00, 6.4184e-01, ..., 1.0446e+00,
6.5041e-01, 1.8220e+00],
...,
[ 1.0515e+00, 1.3694e+00, 4.7536e-01, ..., -2.8691e+00,
-1.2313e+00, 1.2530e+00],
[ 6.9496e-01, 1.8648e+00, 6.3789e-01, ..., -6.4201e-01,
3.3863e-01, 1.3716e+00],
[ 1.5766e+00, 1.5520e+00, 1.5203e+00, ..., 1.6172e+00,
7.7756e-01, 1.8033e+00]],
...,
[[-1.4899e+00, -4.6817e+00, -3.9306e+00, ..., -3.4346e+00,
-2.4346e+00, -3.8854e+00],
[-1.8084e+00, -3.7340e+00, -4.1489e+00, ..., -5.5647e+00,
-3.8794e+00, -4.3091e+00],
[-8.9920e-01, -2.7914e+00, -3.4662e+00, ..., -2.6697e+00,
-2.4992e+00, -1.5999e+00],
...,
[-1.5543e+00, -4.0316e+00, -4.4364e+00, ..., -5.2309e+00,
-3.8863e+00, -2.1145e+00],
[-1.4871e+00, -4.5254e+00, -5.0488e+00, ..., -5.6729e+00,
-3.6246e+00, -4.4642e+00],
[-1.6170e+00, -4.8438e+00, -3.9533e+00, ..., -5.1836e+00,
-3.6980e+00, -3.4459e+00]],
[[ 7.4055e-01, 5.0086e-01, -1.4696e-01, ..., 9.5385e-01,
1.1304e+00, 1.2370e+00],
[ 1.1968e+00, 9.9768e-01, 5.0541e-01, ..., 1.6242e+00,
2.4337e+00, 2.5676e+00],
[ 7.0886e-01, 6.7670e-01, -5.4594e-03, ..., 9.9460e-01,
1.6571e+00, 1.9447e+00],
...,
[ 8.9596e-01, 6.1135e-02, -1.5850e-01, ..., 1.9178e-01,
1.1606e+00, 1.5329e+00],
[ 1.5621e+00, 1.5003e+00, 1.1889e+00, ..., 1.9441e-01,
1.4030e+00, 2.3972e+00],
[ 6.4409e-01, 8.6630e-01, 9.3620e-01, ..., 1.1668e+00,
1.4578e+00, 1.3515e+00]],
[[-2.9564e-01, -1.7165e-01, -9.4347e-01, ..., -2.2371e-01,
-1.9334e-01, 8.6512e-01],
[-1.4708e+00, -1.3198e+00, -1.8568e+00, ..., 1.7641e-01,
-1.1471e-01, 3.7127e-01],
[-1.7649e+00, -5.8824e-01, -9.6792e-01, ..., 3.3678e-01,
1.5681e-01, 2.3761e-01],
...,
[-1.7903e+00, -1.6882e+00, -1.8200e+00, ..., -1.0581e-01,
-1.5545e+00, -4.9015e-01],
[-1.4363e+00, -1.6371e+00, -1.6539e+00, ..., -1.1727e+00,
-9.0378e-01, 3.3924e-01],
[-1.3004e-01, 5.3624e-02, -6.1083e-01, ..., 5.3741e-01,
5.6109e-01, 4.9373e-01]]]], device='cuda:0'),
tensor([[[[ 9.1213e-01, 1.5401e+00, 1.2617e+00, ..., 1.4527e+00,
1.0904e+00, 9.7455e-01],
[-9.8971e-01, -2.2696e+00, -1.2573e+00, ..., -3.6084e-01,
-5.7112e-01, -1.5290e+00],
[-1.4113e-01, -1.0706e+00, -3.3240e-02, ..., -1.0616e-01,
7.6365e-01, 8.3328e-01],
...,
[-1.7360e-02, -1.1003e+00, -9.2049e-01, ..., -3.2442e-01,
-7.2498e-02, 2.2993e-01],
[-5.9558e-01, -1.8338e+00, -1.1991e+00, ..., -6.3050e-01,
-2.6165e-01, -1.4951e-01],
[-3.7478e-01, -1.7252e+00, -1.6256e+00, ..., -1.1217e+00,
-1.5330e+00, -2.8681e-01]],
[[-4.2499e+00, -6.0094e-01, -1.5371e+00, ..., -1.5283e+00,
-2.3230e+00, -1.0521e+00],
[-4.3073e+00, -2.4563e+00, -1.9362e+00, ..., -2.3654e+00,
-3.0794e+00, -8.4280e-01],
[-2.4769e+00, -1.3750e+00, -1.7409e+00, ..., -2.7071e+00,
-2.7923e+00, -9.7715e-01],
...,
[-1.5783e+00, -1.7229e+00, -2.4976e+00, ..., -1.9284e+00,
-2.0974e+00, -7.4806e-01],
[-2.9878e+00, -1.7239e+00, -2.7508e+00, ..., -2.2375e+00,
-2.2684e+00, -1.1887e+00],
[-2.2914e+00, -1.9719e+00, -2.7382e+00, ..., -1.9782e+00,
-1.9589e+00, -1.0412e+00]],
[[ 3.2904e+00, 4.4210e+00, 3.6997e+00, ..., 2.7700e+00,
1.0302e+00, 2.8821e+00],
[ 7.9877e-01, 1.0040e+00, 2.5991e+00, ..., 1.2052e+00,
-7.8339e-01, 2.6194e+00],
[ 1.5109e+00, 1.4024e+00, 1.8429e+00, ..., 4.9595e-01,
-6.2060e-02, 2.0501e+00],
...,
[ 8.9499e-01, 1.0767e+00, 1.2695e+00, ..., 2.9515e-01,
1.0916e+00, 2.3765e+00],
[ 6.2822e-01, 1.8041e+00, 1.1751e+00, ..., 8.3234e-02,
1.0520e+00, 1.8456e+00],
[ 1.6164e+00, 1.6456e+00, 1.8533e+00, ..., 1.3111e+00,
1.8059e+00, 1.7972e+00]],
...,
[[-2.6757e+00, -8.0300e+00, -5.7751e+00, ..., -4.1156e+00,
-3.2655e+00, -5.4274e+00],
[-3.1405e+00, -7.3826e+00, -4.4017e+00, ..., -3.6478e+00,
-2.7329e+00, -4.1894e+00],
[-1.0716e+00, -4.5085e+00, -3.4824e+00, ..., -2.9077e+00,
-3.0568e+00, -2.4993e+00],
...,
[-2.5746e+00, -5.4300e+00, -4.3967e+00, ..., -5.2343e+00,
-4.5074e+00, -4.6187e+00],
[-2.7270e+00, -5.7312e+00, -5.5000e+00, ..., -6.6313e+00,
-5.7811e+00, -6.2565e+00],
[-1.4352e+00, -3.9615e+00, -2.8157e+00, ..., -3.3158e+00,
-2.7139e+00, -3.2084e+00]],
[[-3.3949e-01, 4.8433e-01, 5.3801e-01, ..., 6.8279e-01,
8.8663e-01, 1.4473e+00],
[ 1.7661e+00, 1.3250e+00, 1.6717e+00, ..., 1.8566e+00,
2.5219e+00, 2.9908e+00],
[ 1.4243e+00, 4.5721e-01, 7.0786e-01, ..., 1.2154e+00,
1.7129e+00, 2.3626e+00],
...,
[ 1.3061e+00, 2.0569e-01, -8.7518e-02, ..., 1.2843e+00,
1.1429e+00, 1.5818e+00],
[ 1.8813e+00, 1.6432e+00, 1.0820e+00, ..., 1.4590e+00,
1.3170e+00, 1.9996e+00],
[ 4.7716e-01, 7.5575e-01, 6.9046e-01, ..., 5.1173e-01,
6.7461e-01, 1.6437e+00]],
[[-6.2256e-01, 3.7077e-01, -4.2716e-02, ..., 5.8091e-01,
5.4754e-01, 1.3485e+00],
[-1.6024e+00, -1.7089e+00, -1.8234e+00, ..., -6.8520e-01,
-4.1205e-01, 1.9979e-01],
[-2.4243e+00, -8.4596e-01, 3.4851e-01, ..., 3.4190e-01,
1.8588e-01, 7.5379e-01],
...,
[-1.8805e+00, -1.7652e+00, -1.4429e+00, ..., -8.1909e-01,
-1.1559e+00, -8.9978e-01],
[-1.5631e+00, -2.0180e+00, -1.7185e+00, ..., -8.1480e-01,
-1.2434e+00, -6.6848e-01],
[-8.5187e-01, -8.5202e-01, -1.1405e+00, ..., -5.7485e-01,
-8.2781e-01, -1.0563e-01]]],
[[[ 9.0332e-01, 1.1022e+00, 7.1284e-01, ..., 7.0170e-02,
6.2329e-04, -5.6054e-01],
[-7.8246e-01, -1.3716e+00, 2.5878e-01, ..., 5.4145e-01,
1.2665e+00, -1.6844e-01],
[-1.0211e-01, -3.3237e-01, 2.8360e-01, ..., -8.8789e-03,
7.1615e-01, 9.4263e-01],
...,
[ 2.4387e-01, -7.0715e-01, -5.5451e-01, ..., 1.8587e-01,
5.3209e-01, 5.8607e-01],
[-5.8868e-01, -1.8385e+00, -1.2521e+00, ..., -6.0925e-01,
-3.3087e-01, -1.6147e-01],
[-2.7283e-01, -1.6251e+00, -1.6541e+00, ..., -1.1581e+00,
-1.5361e+00, -2.7275e-01]],
[[-3.6678e+00, -1.1477e+00, -1.4937e+00, ..., -2.4742e+00,
-3.4310e+00, -8.6345e-01],
[-3.4755e+00, -1.9835e+00, -4.7436e-01, ..., -1.3216e+00,
-2.0207e+00, -1.2933e-01],
[-2.1362e+00, -1.4688e+00, -1.6607e+00, ..., -2.3017e+00,
-1.6043e+00, -3.8588e-01],
...,
[-1.5200e+00, -1.2430e+00, -1.9394e+00, ..., -1.7321e+00,
-1.8727e+00, -6.3731e-01],
[-3.0904e+00, -1.6111e+00, -2.7117e+00, ..., -2.3567e+00,
-2.4616e+00, -1.2723e+00],
[-2.2661e+00, -1.9583e+00, -2.6970e+00, ..., -1.9752e+00,
-1.9531e+00, -1.0176e+00]],
[[ 2.5596e+00, 2.9854e+00, 1.9990e+00, ..., 2.2804e+00,
6.0785e-01, 2.1477e+00],
[ 6.3660e-01, 4.5421e-01, 1.7174e+00, ..., 1.7122e+00,
4.9037e-03, 2.0893e+00],
[ 1.2880e+00, 7.9631e-01, 1.9663e+00, ..., 9.5028e-01,
8.4570e-01, 2.2423e+00],
...,
[ 1.2027e+00, 1.4314e+00, 1.6225e+00, ..., 6.2979e-01,
1.2365e+00, 2.3641e+00],
[ 7.1568e-01, 1.8335e+00, 1.1680e+00, ..., 1.0379e-01,
1.1149e+00, 1.8644e+00],
[ 1.6493e+00, 1.6603e+00, 1.8410e+00, ..., 1.3312e+00,
1.8260e+00, 1.8021e+00]],
...,
[[-2.5536e+00, -7.2181e+00, -5.3522e+00, ..., -3.0058e+00,
-2.6794e+00, -3.6366e+00],
[-3.0577e+00, -7.8585e+00, -5.2217e+00, ..., -5.1560e+00,
-4.4733e+00, -4.9582e+00],
[-1.6681e+00, -4.8050e+00, -2.7512e+00, ..., -2.6062e+00,
-2.2947e+00, -2.0969e+00],
...,
[-2.2610e+00, -4.9577e+00, -3.8809e+00, ..., -4.9068e+00,
-4.4920e+00, -4.7633e+00],
[-2.6067e+00, -5.6083e+00, -5.3035e+00, ..., -6.4063e+00,
-5.6707e+00, -6.2178e+00],
[-1.4153e+00, -4.1660e+00, -2.8792e+00, ..., -3.2241e+00,
-2.6827e+00, -3.2700e+00]],
[[-4.4987e-01, 4.0132e-01, 6.1430e-01, ..., 6.2575e-01,
8.3433e-01, 1.6353e+00],
[ 1.2069e+00, 5.3607e-01, 1.0576e+00, ..., 1.2281e+00,
1.6675e+00, 2.4816e+00],
[ 1.1840e+00, 9.2190e-02, 1.7676e-01, ..., 1.0040e+00,
1.3330e+00, 1.8337e+00],
...,
[ 1.2225e+00, 3.5884e-01, 5.2051e-02, ..., 1.4319e+00,
1.1361e+00, 1.5397e+00],
[ 1.8240e+00, 1.6232e+00, 1.0417e+00, ..., 1.4507e+00,
1.2573e+00, 1.9866e+00],
[ 4.4071e-01, 7.4938e-01, 7.3788e-01, ..., 5.3954e-01,
7.1929e-01, 1.6743e+00]],
[[-6.2973e-01, -2.8459e-01, -5.6803e-01, ..., -4.1928e-01,
-5.0705e-01, 5.1427e-01],
[-1.3719e+00, -1.3609e+00, -8.1893e-01, ..., -1.1655e-01,
4.5121e-01, 8.4334e-01],
[-2.1477e+00, -2.5813e-01, 4.9787e-01, ..., 2.4977e-01,
4.2528e-01, 4.7910e-02],
...,
[-1.8665e+00, -1.7158e+00, -1.1515e+00, ..., -5.4901e-01,
-7.8743e-01, -5.8129e-01],
[-1.5474e+00, -1.9497e+00, -1.6607e+00, ..., -9.3717e-01,
-1.3789e+00, -6.9564e-01],
[-7.3530e-01, -7.4079e-01, -1.1561e+00, ..., -6.1206e-01,
-8.4490e-01, -9.9913e-02]]],
[[[ 1.0765e+00, 1.5687e+00, 1.2731e+00, ..., 2.5288e-01,
4.0595e-01, 3.0427e-01],
[-1.1901e+00, -1.6594e+00, -9.9809e-01, ..., 3.4067e-01,
3.2060e-01, -1.1864e+00],
[-8.8658e-01, -1.0489e+00, -5.7805e-01, ..., -4.7501e-01,
1.0063e+00, 6.3000e-01],
...,
[-3.8985e-02, -2.0489e-01, 2.7643e-01, ..., 8.0022e-01,
1.1386e+00, 5.0388e-01],
[-3.6414e-01, -8.7945e-01, -2.5674e-01, ..., 6.1885e-01,
1.4171e+00, -1.3167e-01],
[-2.1402e-01, -2.0473e-01, -9.1997e-01, ..., -2.7040e-01,
3.0134e-01, 5.9882e-01]],
[[-3.5616e+00, -1.9777e+00, -2.2902e+00, ..., -1.6915e+00,
-2.1758e+00, -1.0611e+00],
[-3.3727e+00, -3.2659e+00, -1.7577e+00, ..., -2.6178e-01,
-1.3603e+00, -2.9118e-01],
[-2.8173e+00, -1.7936e+00, -1.6908e+00, ..., -1.1306e+00,
-1.4888e+00, -6.8872e-01],
...,
[-2.0017e-01, -4.6177e-01, -6.7822e-01, ..., -1.3671e+00,
4.9821e-01, 4.8958e-01],
[-9.5444e-01, -7.3010e-01, -5.8219e-01, ..., 2.3328e-01,
-7.8624e-02, 2.4672e-01],
[-3.6530e-01, -1.1483e+00, -1.0961e+00, ..., -1.9365e-01,
-7.1893e-01, -5.5979e-01]],
[[ 2.6305e+00, 2.6204e+00, 1.8185e+00, ..., 2.8031e+00,
1.2613e+00, 2.8156e+00],
[ 1.4212e-01, 2.1667e-03, 1.5077e+00, ..., 1.5700e+00,
-2.6619e-01, 2.3653e+00],
[-1.4759e-01, 2.5883e-01, 8.8162e-01, ..., 8.3427e-01,
4.2284e-01, 1.7730e+00],
...,
[-5.2884e-01, -3.3796e-01, 1.5631e-01, ..., 7.2283e-01,
2.5109e-01, 1.4332e+00],
[-9.0629e-01, 4.0570e-01, 6.0754e-01, ..., 1.2921e+00,
4.6613e-01, 1.5443e+00],
[ 7.0035e-01, 4.7766e-01, 9.7146e-01, ..., 1.0093e+00,
1.0733e+00, 2.0204e+00]],
...,
[[-3.4893e+00, -9.0985e+00, -7.3064e+00, ..., -3.2085e+00,
-2.1836e+00, -4.0779e+00],
[-4.6586e+00, -1.0952e+01, -6.8846e+00, ..., -5.5681e+00,
-3.6485e+00, -4.3620e+00],
[-3.5271e+00, -9.8653e+00, -5.4865e+00, ..., -3.2734e+00,
-2.0509e+00, -1.5280e+00],
...,
[-3.4019e+00, -8.1885e+00, -5.5145e+00, ..., -4.5732e+00,
-3.0687e+00, -1.5327e+00],
[-3.8057e+00, -8.9174e+00, -4.8333e+00, ..., -7.2579e+00,
-5.1457e+00, -3.7769e+00],
[-3.0079e+00, -7.4758e+00, -2.9457e+00, ..., -5.5455e+00,
-2.5117e+00, -2.7889e+00]],
[[-5.1545e-01, 8.5636e-01, 8.6002e-01, ..., 1.3677e+00,
1.0141e+00, 1.2244e+00],
[ 7.8590e-01, 8.9355e-01, 1.3748e+00, ..., 1.9690e+00,
2.2671e+00, 2.5339e+00],
[ 3.7940e-01, 9.9104e-01, 1.5219e+00, ..., 1.2837e+00,
1.4114e+00, 1.9129e+00],
...,
[ 1.2588e+00, 7.4252e-01, 1.2876e+00, ..., 1.4763e+00,
2.1246e+00, 1.6609e+00],
[ 1.7248e+00, 1.3380e+00, 1.5301e+00, ..., 2.1060e+00,
1.5085e+00, 2.4492e+00],
[ 7.9259e-01, 4.9988e-01, 1.0875e+00, ..., 1.3709e+00,
1.2835e+00, 1.3660e+00]],
[[-1.8237e-01, -3.4943e-01, -4.0432e-01, ..., -1.3054e-01,
-8.8304e-02, 8.9660e-01],
[-8.9267e-01, -3.0287e+00, -1.3856e+00, ..., 3.1339e-01,
-8.4230e-02, 3.6151e-01],
[-1.4980e+00, -2.2912e+00, -7.1994e-01, ..., 1.2992e-01,
4.3190e-02, 2.0510e-01],
...,
[-2.2745e+00, -1.8617e+00, -4.7136e-01, ..., -2.3296e+00,
-4.5016e-01, -3.3360e-01],
[-1.4297e+00, -2.4692e+00, -8.2603e-01, ..., -1.1988e-01,
-4.1286e-02, 3.1576e-01],
[-3.7498e-01, -1.2408e+00, -7.1363e-01, ..., 5.6057e-01,
1.9173e-01, 3.2302e-01]]],
[[[ 6.3828e-01, 1.2552e+00, -7.3002e-01, ..., 3.5316e-01,
2.1134e-01, -1.8280e-01],
[-9.6348e-02, 4.1936e-01, -2.9078e+00, ..., 1.7409e-01,
1.2498e-01, -3.0816e-01],
[-2.3389e-01, 9.0725e-01, -9.3889e-01, ..., 1.5618e-01,
1.6052e-01, 8.2153e-01],
...,
[-1.2172e-01, -6.5738e-01, -1.5737e+00, ..., -4.7796e+00,
-2.7156e-01, 2.8496e-01],
[-9.7888e-01, -1.4079e+00, -1.9821e+00, ..., -4.6451e+00,
-2.1907e-01, 3.3983e-01],
[-3.8693e-01, -1.0485e+00, -1.5153e+00, ..., 1.6154e+00,
2.3563e+00, 1.0458e+00]],
[[-1.9171e+00, -1.9862e-01, -2.2725e+00, ..., -2.1549e+00,
-2.5498e+00, -1.4334e+00],
[-1.3638e+00, -3.7425e-02, -1.7070e+00, ..., -1.3240e+00,
-2.5007e+00, -1.5590e+00],
[-2.5649e+00, -4.6860e+00, -1.0080e+01, ..., -1.8400e+00,
-2.3432e+00, -1.0279e+00],
...,
[-1.0472e+00, -1.5447e+00, 1.6442e+00, ..., -6.9365e+00,
-3.6329e+00, -1.8993e+00],
[-1.3911e+00, 6.1988e-01, 1.5480e-01, ..., -5.3617e+00,
-4.2220e+00, -4.4871e+00],
[-7.9915e-01, 1.9757e-01, 5.9037e-01, ..., -8.4519e+00,
-7.1611e+00, -4.2556e+00]],
[[ 2.7003e+00, 3.7545e-01, -1.4328e+00, ..., 1.5219e+00,
8.4314e-01, 2.1722e+00],
[ 1.5441e+00, -2.7210e+00, -5.1766e+00, ..., 4.2935e-01,
4.6710e-01, 2.3272e+00],
[ 1.7325e+00, -5.1569e-01, -1.0246e+01, ..., -1.0570e-01,
7.2685e-01, 2.0609e+00],
...,
[ 2.2638e+00, 2.0141e+00, -1.0889e+00, ..., -7.9170e+00,
-4.0668e+00, 1.0587e-01],
[ 2.2840e+00, 2.2872e+00, -2.1903e+00, ..., -1.2106e+01,
-7.3906e+00, 9.2399e-01],
[ 2.8879e+00, 7.9320e-01, -3.2087e+00, ..., -1.2876e+01,
-9.0914e+00, -1.1947e+00]],
...,
[[-2.3878e+00, -4.7867e+00, -1.9755e+00, ..., -2.4808e+00,
-1.9575e+00, -3.1461e+00],
[-2.6169e+00, -5.5484e+00, -4.1284e+00, ..., -3.9526e+00,
-3.0172e+00, -3.6492e+00],
[-1.6359e+00, -5.7202e+00, -9.7719e+00, ..., -2.9893e+00,
-2.4894e+00, -2.4114e+00],
...,
[-1.8356e+00, -7.0073e+00, -4.3369e+00, ..., -6.3558e-01,
-5.3232e+00, -1.0132e+01],
[-2.6501e+00, -6.5911e+00, -3.5370e+00, ..., -3.0533e+00,
-7.1332e+00, -9.9337e+00],
[-2.8205e+00, -5.6119e+00, -2.6288e+00, ..., -1.8019e+01,
-1.4515e+01, -6.6629e+00]],
[[ 6.4455e-01, -2.1513e-01, -4.9671e+00, ..., 1.0965e+00,
1.3398e+00, 1.5821e+00],
[ 1.0252e+00, 7.1700e-01, 4.6985e-01, ..., 1.2546e+00,
9.2862e-01, 2.1642e+00],
[ 1.0530e-03, -2.8487e+00, 1.6784e+00, ..., 9.5331e-01,
2.5453e-01, 1.3554e+00],
...,
[ 1.5774e+00, -8.6499e-01, -1.5085e-01, ..., -4.7986e+00,
-2.5851e+00, 7.9778e-01],
[ 2.1908e+00, 2.0211e-01, -2.1230e+00, ..., -9.4570e+00,
-1.0053e+00, 2.2953e+00],
[ 1.0616e+00, 1.6984e+00, 2.3546e+00, ..., -7.1165e+00,
-2.6910e+00, 5.8286e-01]],
[[-3.2659e-01, -1.6123e+00, 1.4097e+00, ..., -7.1630e-01,
-9.1042e-01, 1.4466e-02],
[-7.3965e-01, -4.5002e+00, -2.4127e+00, ..., -1.4069e+00,
-5.8352e-01, -2.8847e-01],
[-1.2460e+00, -1.4428e+00, -1.7655e+00, ..., 4.9476e-02,
-1.4946e-01, -6.6364e-02],
...,
[-2.4753e+00, -2.1564e+00, 1.8407e+00, ..., -7.8557e+00,
-1.4234e+00, -1.5998e+00],
[-1.8743e+00, -2.2570e+00, 5.7260e-01, ..., -1.8823e+00,
1.8451e+00, 4.9159e-01],
[-6.8241e-02, -2.6182e-01, -2.0504e+00, ..., 1.8153e+00,
-2.5127e-01, -1.4145e+00]]],
[[[ 7.4366e-01, 1.0846e+00, 7.5890e-01, ..., 5.0138e-01,
4.3527e-01, 2.5383e-01],
[-3.0146e-01, -1.3973e+00, -4.7649e-01, ..., -5.2757e-02,
1.6405e-01, -1.2822e+00],
[ 1.7198e-01, -7.4568e-01, -2.0681e-01, ..., -1.1955e+00,
9.8779e-01, 6.3053e-01],
...,
[-5.8642e-02, 3.6127e-01, -1.1375e+00, ..., -5.4992e-02,
7.9721e-01, 4.7153e-01],
[-3.5532e-01, -7.6351e-02, -1.2257e+00, ..., -1.1239e-01,
1.1736e+00, 1.8960e-02],
[ 1.9650e-01, 1.9683e-01, 1.2265e+00, ..., -3.3083e-01,
3.5797e-01, 6.3656e-01]],
[[-2.9723e+00, -1.6123e+00, -3.1673e+00, ..., -1.7838e+00,
-2.4198e+00, -1.3105e+00],
[-4.1894e+00, -1.8055e+00, -2.3341e+00, ..., -3.3537e-01,
-1.6735e+00, -5.3039e-01],
[-4.1462e+00, -2.7047e+00, -4.3644e+00, ..., -1.7778e+00,
-1.6914e+00, -9.1084e-01],
...,
[-1.4703e+00, 4.2541e-01, 8.7364e-01, ..., -6.9008e-01,
5.8439e-01, 7.6337e-01],
[-2.7688e+00, -5.2218e-01, 4.4230e-01, ..., -2.3739e-01,
-1.5858e-01, 1.7775e-01],
[-1.2142e+00, -7.4929e-01, -1.8020e+00, ..., -5.2551e-01,
-7.3949e-01, -5.5362e-01]],
[[ 1.6381e+00, 2.7862e+00, 2.4045e+00, ..., 3.0556e+00,
1.3635e+00, 3.0352e+00],
[ 1.7275e-01, 1.1710e+00, 1.6836e+00, ..., 2.1862e+00,
-3.3128e-01, 2.5927e+00],
[ 7.6182e-01, 1.4135e+00, 7.3585e-01, ..., 2.2916e+00,
4.2206e-01, 1.7048e+00],
...,
[ 1.9616e+00, -2.6812e+00, -2.5832e+00, ..., 1.5973e+00,
2.3788e-01, 1.0455e+00],
[ 1.4013e+00, -8.0965e-02, -3.0043e+00, ..., 5.1932e-01,
2.1416e-01, 1.5509e+00],
[ 2.1216e+00, 2.6174e+00, 1.6396e+00, ..., 1.3806e+00,
9.7710e-01, 2.0280e+00]],
...,
[[-1.2927e+00, -4.4346e+00, -3.8536e+00, ..., -3.6217e+00,
-2.5715e+00, -4.2763e+00],
[-1.7104e+00, -3.7500e+00, -4.2147e+00, ..., -5.6562e+00,
-4.0064e+00, -4.5416e+00],
[-9.8726e-01, -3.2456e+00, -4.0906e+00, ..., -2.1469e+00,
-2.3399e+00, -1.4699e+00],
...,
[-1.6737e+00, -3.4550e+00, -1.8161e+00, ..., -5.6451e+00,
-5.0828e+00, -3.2236e+00],
[-1.6683e+00, -4.2757e+00, -3.7203e+00, ..., -8.2906e+00,
-4.9900e+00, -3.5448e+00],
[-1.6888e+00, -4.2464e+00, -3.4646e+00, ..., -3.2975e+00,
-1.8859e+00, -2.5653e+00]],
[[ 7.8134e-01, 4.2518e-01, -1.6387e-01, ..., 8.2117e-01,
9.8951e-01, 1.1598e+00],
[ 1.2343e+00, 9.5291e-01, 5.0192e-01, ..., 1.6447e+00,
2.2587e+00, 2.6065e+00],
[ 5.8204e-01, 3.9436e-01, -2.9939e-01, ..., 1.7524e+00,
1.8239e+00, 1.7765e+00],
...,
[ 1.1259e+00, 4.7280e-01, -5.4793e-01, ..., 2.1702e+00,
2.9446e+00, 1.8889e+00],
[ 1.3238e+00, 1.4310e+00, 2.5008e+00, ..., 2.1849e+00,
1.5666e+00, 2.3797e+00],
[ 4.7472e-01, 6.2187e-02, 4.1068e-01, ..., 9.4804e-01,
1.3661e+00, 1.2968e+00]],
[[-2.8325e-01, -1.4043e-01, -9.3253e-01, ..., -1.9839e-01,
-1.3914e-01, 9.1779e-01],
[-1.3821e+00, -1.4085e+00, -1.7730e+00, ..., -2.7818e-01,
-2.2222e-01, 4.3044e-01],
[-1.6457e+00, -1.0216e+00, -1.2593e+00, ..., -5.1712e-01,
2.5856e-01, 3.4796e-01],
...,
[-1.3263e+00, -1.2634e+00, 7.0142e-01, ..., -4.8680e-01,
4.6268e-01, -3.0239e-01],
[-1.0884e+00, -6.8152e-01, -1.3829e-01, ..., 2.4586e-01,
5.4545e-02, 3.2672e-01],
[ 7.0631e-02, 5.1789e-01, 2.1023e-01, ..., 1.6655e-01,
1.2726e-01, 3.2082e-01]]]], device='cuda:0'),
tensor([[[[ 6.4497e-01, 1.0025e+00, 7.1696e-01, ..., 8.8679e-01,
9.0600e-01, 3.4978e-01],
[-2.6523e-01, -1.5105e+00, -7.2942e-01, ..., -1.4820e-01,
4.0643e-01, -5.8595e-01],
[ 2.3255e-02, -7.1134e-01, -1.9469e-01, ..., 1.5248e-01,
6.0553e-01, 5.2958e-01],
...,
[ 3.1756e-01, -8.2275e-01, -3.8740e-01, ..., -9.3269e-01,
-5.6158e-01, 3.9086e-01],
[-5.3719e-01, -1.8812e+00, -8.2782e-01, ..., 1.1683e+00,
7.3341e-01, 2.5336e-02],
[ 3.0264e-01, -6.2265e-01, -9.0038e-01, ..., 5.7718e-02,
-2.5432e-01, 5.1000e-01]],
[[-3.5468e+00, -1.5168e+00, -3.1892e+00, ..., -1.4226e+00,
-2.7939e+00, -9.2814e-01],
[-4.4839e+00, -1.5527e+00, -2.3585e+00, ..., -9.7937e-01,
-3.0585e+00, -8.2578e-01],
[-4.7377e+00, -2.8926e+00, -3.1937e+00, ..., -2.2828e+00,
-3.2744e+00, -1.5866e+00],
...,
[-2.6635e+00, -1.9664e+00, -2.7097e+00, ..., -5.1315e+00,
-2.4703e+00, 4.9016e-01],
[-3.6014e+00, -1.6052e+00, -2.3184e+00, ..., -7.8772e-01,
-1.0044e+00, 3.1883e-01],
[-1.3204e+00, -1.0223e+00, -1.8673e+00, ..., -5.6311e-01,
-4.0463e-01, 4.4884e-02]],
[[ 2.2125e+00, 3.0980e+00, 2.4904e+00, ..., 2.2075e+00,
7.3262e-01, 1.9618e+00],
[ 5.0568e-01, 1.2442e+00, 1.6483e+00, ..., 1.6851e+00,
8.7468e-02, 2.0210e+00],
[ 9.3963e-01, 1.7389e+00, 1.5055e+00, ..., 2.4350e-01,
-9.8525e-01, -6.5188e-01],
...,
[ 7.8069e-01, 1.2835e+00, 1.2311e+00, ..., -1.9357e+00,
2.0938e-01, 8.2142e-01],
[ 5.7252e-01, 1.7685e+00, 7.7490e-01, ..., 8.2059e-01,
5.3590e-01, 9.1662e-01],
[ 1.4562e+00, 1.4935e+00, 1.4439e+00, ..., 7.9864e-01,
6.2552e-01, 1.2823e+00]],
...,
[[-1.9375e+00, -5.1446e+00, -4.2857e+00, ..., -4.5911e+00,
-3.1878e+00, -3.9448e+00],
[-2.3806e+00, -4.7461e+00, -4.3964e+00, ..., -5.3995e+00,
-3.0545e+00, -3.6329e+00],
[-5.1329e-01, -2.9306e+00, -3.1226e+00, ..., -3.9642e+00,
-2.2501e+00, -1.4021e+00],
...,
[-1.3446e+00, -3.9749e+00, -4.3174e+00, ..., -4.6465e+00,
-4.3507e+00, -4.2955e+00],
[-1.3173e+00, -4.3053e+00, -5.3806e+00, ..., -7.5434e+00,
-5.4694e+00, -5.1623e+00],
[-1.4386e+00, -4.6413e+00, -4.2810e+00, ..., -4.4584e+00,
-1.9955e+00, -3.2031e+00]],
[[ 6.7277e-01, 7.0987e-01, -1.4310e-01, ..., 1.4535e+00,
9.4840e-01, 1.6894e+00],
[ 1.2118e+00, 1.1652e+00, 5.5693e-01, ..., 2.6587e+00,
1.3147e+00, 2.4582e+00],
[ 9.4318e-02, -1.4715e-01, 4.5371e-02, ..., 1.1925e+00,
-1.0559e+00, 5.7140e-01],
...,
[ 7.6412e-01, -1.7720e-01, -1.2234e-01, ..., 1.6403e+00,
1.7428e+00, 2.1128e+00],
[ 1.3962e+00, 1.2104e+00, 1.0473e+00, ..., 3.4450e+00,
1.4292e+00, 2.5521e+00],
[ 6.1708e-01, 8.5807e-01, 8.9432e-01, ..., 1.2977e+00,
1.1687e+00, 1.3672e+00]],
[[-2.9888e-01, -2.9754e-01, -1.0026e+00, ..., 5.9053e-01,
2.5612e-01, 1.1299e+00],
[-1.2592e+00, -1.6615e+00, -2.3438e+00, ..., -3.1450e-01,
-1.7995e-01, 5.5835e-01],
[-1.3161e+00, -8.5838e-01, -1.1022e+00, ..., -2.2091e-01,
-9.2187e-01, 3.6209e-01],
...,
[-1.7847e+00, -1.6964e+00, -1.6803e+00, ..., -3.0509e+00,
-2.2892e+00, -1.0915e+00],
[-1.4863e+00, -1.6314e+00, -1.6053e+00, ..., -9.0491e-02,
-6.5767e-01, 3.9932e-03],
[-1.9189e-01, -3.4970e-02, -6.3739e-01, ..., -1.0092e+00,
-6.4459e-01, 2.2730e-01]]],
[[[ 8.1152e-01, 1.2094e+00, 1.1012e-02, ..., -3.0906e-01,
-1.2164e-01, -4.5260e-01],
[-6.0737e-01, -5.5108e-01, 1.1179e+00, ..., -5.4659e-01,
1.0366e+00, -2.9985e-01],
[-2.9340e-02, -6.1262e-01, 1.0848e+00, ..., 1.1420e+00,
5.8624e-01, 9.4830e-01],
...,
[ 2.6895e-01, -7.5186e-01, -5.7519e-01, ..., 7.1279e-02,
4.8312e-01, 6.4648e-01],
[-5.6853e-01, -1.7554e+00, -1.2471e+00, ..., -6.3814e-01,
-3.6872e-01, -1.3722e-01],
[-3.6126e-01, -1.5376e+00, -1.6416e+00, ..., -1.1518e+00,
-1.5806e+00, -2.9004e-01]],
[[-3.4047e+00, -6.5866e-01, -2.4109e+00, ..., -2.4672e+00,
-3.2026e+00, -6.5681e-01],
[-2.9862e+00, -1.4053e+00, -3.0256e+00, ..., -1.2315e+00,
-2.0169e+00, -1.0757e-01],
[-1.9316e+00, -1.3478e+00, -1.6464e+00, ..., -2.3754e+00,
-1.5990e+00, -4.5033e-01],
...,
[-1.5364e+00, -1.1480e+00, -1.9853e+00, ..., -1.5822e+00,
-1.9215e+00, -6.9683e-01],
[-3.0677e+00, -1.5464e+00, -2.5901e+00, ..., -2.4060e+00,
-2.4040e+00, -1.2992e+00],
[-2.2237e+00, -1.9150e+00, -2.6735e+00, ..., -1.9142e+00,
-1.9489e+00, -1.0341e+00]],
[[ 2.2282e+00, 2.7800e+00, 4.5711e-01, ..., 2.3119e+00,
3.1775e-01, 2.0983e+00],
[ 3.9262e-01, 9.7564e-01, -2.0216e+00, ..., 3.0060e+00,
-3.9289e-02, 2.0179e+00],
[ 1.0071e+00, 1.2417e+00, -2.2201e+00, ..., 1.2960e+00,
1.0727e+00, 2.3027e+00],
...,
[ 1.2512e+00, 1.4818e+00, 1.5171e+00, ..., 1.1943e-01,
1.2335e+00, 2.3939e+00],
[ 7.3386e-01, 1.7720e+00, 1.1056e+00, ..., -2.8010e-02,
1.0456e+00, 1.8763e+00],
[ 1.6173e+00, 1.6494e+00, 1.7832e+00, ..., 1.2404e+00,
1.7574e+00, 1.7810e+00]],
...,
[[-1.9327e+00, -6.3567e+00, -5.0635e+00, ..., -2.8074e+00,
-2.7379e+00, -3.6008e+00],
[-2.5545e+00, -7.8531e+00, -5.6829e+00, ..., -5.4503e+00,
-4.6299e+00, -4.7402e+00],
[-1.4346e+00, -4.2508e+00, -2.9915e+00, ..., -7.8670e+00,
-3.5656e+00, -1.9844e+00],
...,
[-2.2553e+00, -4.7280e+00, -3.8761e+00, ..., -6.2605e+00,
-4.2992e+00, -4.8632e+00],
[-2.6029e+00, -5.5748e+00, -5.2670e+00, ..., -6.0626e+00,
-5.5688e+00, -6.3043e+00],
[-1.3748e+00, -4.0529e+00, -2.8477e+00, ..., -3.2587e+00,
-2.6346e+00, -3.1784e+00]],
[[-4.1115e-01, 1.7447e-01, 7.8449e-01, ..., 9.4302e-01,
9.7796e-01, 1.6526e+00],
[ 1.1248e+00, -1.6534e-01, 3.7819e-01, ..., 2.2270e+00,
1.9820e+00, 2.4612e+00],
[ 1.0628e+00, -1.1333e-01, 1.1340e+00, ..., 1.1686e+00,
1.7644e+00, 1.7768e+00],
...,
[ 1.2444e+00, 4.4736e-01, 8.4034e-02, ..., 1.4536e+00,
1.0825e+00, 1.4914e+00],
[ 1.7705e+00, 1.6669e+00, 1.0584e+00, ..., 1.4813e+00,
1.2624e+00, 1.9096e+00],
[ 4.4488e-01, 7.1012e-01, 7.1369e-01, ..., 5.0628e-01,
6.8377e-01, 1.6605e+00]],
[[-6.5352e-01, -5.6101e-01, -4.2046e-01, ..., -7.2683e-01,
-5.4344e-01, 6.3780e-01],
[-1.2583e+00, -1.1128e+00, -1.2122e+00, ..., -9.8356e-01,
5.9237e-01, 8.0280e-01],
[-1.9087e+00, -2.4446e-01, -1.1762e+00, ..., 7.9356e-01,
4.7357e-01, -4.8993e-02],
...,
[-1.9063e+00, -1.7247e+00, -1.1584e+00, ..., -4.8825e-01,
-8.3873e-01, -6.2050e-01],
[-1.5322e+00, -2.0257e+00, -1.5910e+00, ..., -1.2087e+00,
-1.3208e+00, -6.9035e-01],
[-7.6654e-01, -8.2845e-01, -1.1440e+00, ..., -6.1974e-01,
-9.0800e-01, -9.7200e-02]]],
[[[ 9.5773e-01, 1.5006e+00, 2.5643e-01, ..., 6.5015e-01,
9.0047e-01, 4.2509e-01],
[-1.2665e+00, -2.6069e+00, -1.3338e+00, ..., -3.1642e-01,
-2.0592e-01, -3.3587e-01],
[-3.7324e+00, -2.3136e+00, -4.2034e-01, ..., -1.6033e+00,
-1.2480e+00, -8.7007e-02],
...,
[ 4.2434e-01, -2.5835e-01, -1.3169e-01, ..., -1.3030e+00,
-7.5378e-01, 5.2060e-02],
[-1.6195e-01, -1.0229e+00, -8.9095e-02, ..., 2.3798e-01,
-8.6695e-02, -1.3232e-01],
[ 7.3687e-01, -3.0304e-01, -2.0054e-01, ..., -2.5650e-01,
-8.2382e-01, 1.8604e-01]],
[[-6.4591e+00, -2.6666e+00, -3.4706e+00, ..., -1.4495e+00,
-1.9287e+00, -1.6595e+00],
[-8.4569e+00, -5.5708e+00, -3.5570e+00, ..., -8.5787e-01,
-1.7227e+00, -1.9270e+00],
[-3.9812e+00, 9.7888e-01, 7.0927e-01, ..., -2.9358e+00,
-1.8103e+00, -1.0410e+00],
...,
[-8.7309e-01, -7.0976e-01, -1.2072e+00, ..., -3.7467e+00,
-2.9485e+00, -1.6954e+00],
[-2.8280e+00, -4.4444e-01, -8.9242e-01, ..., -9.8376e-01,
-1.0576e+00, -1.1323e+00],
[-2.0494e+00, -1.4767e+00, -1.3833e+00, ..., -4.2808e-01,
-5.1350e-01, -4.8464e-01]],
[[ 2.8259e+00, 5.2348e+00, 3.8055e+00, ..., 1.6912e+00,
9.5421e-01, 2.5684e+00],
[ 4.5008e-01, 1.3653e-01, 1.0243e+00, ..., 1.8024e+00,
1.0903e+00, 2.5292e+00],
[-2.4996e+00, -6.9812e-01, -1.4406e+00, ..., 1.1486e+00,
1.9356e+00, 2.2330e+00],
...,
[ 1.2975e-01, 9.1516e-01, 1.1655e+00, ..., -1.6653e+00,
2.1524e-01, 1.4044e+00],
[ 2.5582e-01, 2.0659e+00, 1.4873e+00, ..., 4.6113e-01,
1.1390e+00, 1.2178e+00],
[ 1.3967e+00, 1.3175e+00, 1.6783e+00, ..., 6.9620e-01,
1.4630e+00, 1.4681e+00]],
...,
[[-3.1249e+00, -8.0447e+00, -5.5167e+00, ..., -4.8612e+00,
-3.5291e+00, -4.0866e+00],
[-3.1635e+00, -9.9555e+00, -6.3344e+00, ..., -5.7461e+00,
-4.3598e+00, -3.2368e+00],
[-9.1514e-01, -7.4894e+00, -9.0007e+00, ..., -3.0060e+00,
-2.6277e+00, -8.4467e-01],
...,
[-9.3190e-01, -2.2267e+00, -1.7060e+00, ..., -5.7635e+00,
-5.2794e+00, -3.6525e+00],
[-1.3113e+00, -3.7070e+00, -3.3827e+00, ..., -6.8232e+00,
-6.0208e+00, -5.6124e+00],
[-2.2013e+00, -4.9375e+00, -3.0038e+00, ..., -3.4467e+00,
-2.5625e+00, -2.5956e+00]],
[[-1.5153e+00, -3.9982e-01, -8.8466e-01, ..., 1.4703e+00,
1.7106e+00, 1.2623e+00],
[-2.2012e+00, -2.6170e+00, -3.2687e+00, ..., 1.6728e+00,
1.3337e+00, 1.6093e+00],
[ 7.8539e-01, 2.0600e+00, 9.5410e-01, ..., 4.4880e-01,
1.0598e+00, 1.5446e+00],
...,
[ 1.2315e+00, -1.9374e-01, 1.0493e+00, ..., 8.0881e-01,
1.7730e-01, 1.0655e+00],
[ 1.8675e+00, 9.1152e-01, 1.7428e+00, ..., 2.2807e+00,
1.2787e+00, 1.9045e+00],
[ 1.0189e+00, 3.6233e-01, 9.3975e-01, ..., 8.0147e-01,
7.3143e-01, 1.5102e+00]],
[[-5.0046e-01, 1.0027e+00, -1.6451e-01, ..., -3.9432e-01,
-5.8227e-01, -3.1906e-02],
[-1.6150e+00, -1.7009e+00, -2.3157e+00, ..., -1.1343e+00,
-1.6185e+00, -1.0987e+00],
[-1.9635e+00, 9.4107e-01, 3.9443e-01, ..., -1.3857e+00,
-2.2726e+00, -8.7074e-01],
...,
[-1.6896e+00, -1.4454e+00, -9.9307e-01, ..., -2.4929e+00,
-3.1066e+00, -1.1980e+00],
[-1.6898e+00, -1.0716e+00, -8.0111e-01, ..., -1.2525e+00,
-1.7117e+00, -5.3579e-01],
[-3.2068e-01, 3.5642e-01, -2.8306e-01, ..., -2.0157e-01,
-7.0860e-01, 8.3829e-02]]],
[[[ 6.7483e-01, 8.2458e-01, 4.6223e-01, ..., 2.9304e-01,
3.6271e-01, -1.9745e-02],
[-1.7642e-01, -9.5753e-01, -1.2233e+00, ..., 2.3142e-01,
1.5824e-01, -3.3504e-01],
[ 5.3270e-02, -1.4016e+00, -2.4258e+00, ..., -3.0550e-01,
-1.7835e-03, 6.7266e-01],
...,
[ 4.8229e-01, -1.6085e+00, -1.8432e+00, ..., 3.9027e-01,
8.4689e-01, 6.8145e-01],
[ 6.6002e-02, -6.1765e-01, -5.3527e-01, ..., 9.6341e-01,
1.4539e+00, 1.2842e-01],
[ 8.8303e-01, -1.1113e-01, -1.9498e-01, ..., 7.1715e-01,
4.0899e-01, 6.7398e-01]],
[[-2.8499e+00, -1.1001e+00, -1.8826e+00, ..., -1.6896e+00,
-2.2256e+00, -1.4497e+00],
[-3.0761e+00, -1.5406e+00, -1.5109e+00, ..., -1.3950e+00,
-2.3420e+00, -1.6191e+00],
[-3.3404e+00, -2.5399e+00, 5.7353e-01, ..., -2.2890e+00,
-2.3679e+00, -1.0979e+00],
...,
[-5.6394e-01, -2.4387e+00, -4.4263e+00, ..., -1.7875e+00,
5.5636e-01, 7.4608e-01],
[-2.0540e+00, -7.4247e-01, -1.0168e+00, ..., 2.3386e-01,
2.6974e-01, 7.5736e-01],
[-1.0937e+00, -1.3090e+00, -1.3561e+00, ..., -4.6896e-01,
-5.3017e-01, -2.0807e-01]],
[[ 2.0505e+00, 2.3304e+00, 2.0423e+00, ..., 1.6166e+00,
7.9858e-01, 2.0568e+00],
[ 4.4954e-01, 3.9019e-01, 4.3800e-01, ..., 7.4624e-01,
3.4372e-01, 2.2052e+00],
[ 1.2445e+00, -1.0641e+00, 3.9201e-01, ..., -5.4891e-02,
8.4453e-01, 1.9905e+00],
...,
[ 4.4087e-01, 6.4809e-01, 4.0008e+00, ..., 5.9367e-01,
-3.0463e-01, 1.6234e+00],
[ 6.7245e-01, 1.7097e+00, 2.2933e+00, ..., 1.3967e+00,
6.3134e-01, 1.8043e+00],
[ 2.0159e+00, 2.0411e+00, 2.5355e+00, ..., 2.0902e+00,
1.0781e+00, 1.8543e+00]],
...,
[[-2.0317e+00, -5.4935e+00, -2.8193e+00, ..., -2.1238e+00,
-1.6945e+00, -2.9861e+00],
[-2.5608e+00, -4.9599e+00, -2.5628e+00, ..., -2.7976e+00,
-2.5221e+00, -3.0715e+00],
[-1.7813e+00, -5.8535e+00, -3.2809e+00, ..., -2.4180e+00,
-2.5539e+00, -2.3420e+00],
...,
[-2.0581e+00, -3.9599e+00, -2.4706e+00, ..., -3.1132e+00,
-4.1749e+00, -2.5722e+00],
[-1.9921e+00, -4.7929e+00, -4.3765e+00, ..., -5.3550e+00,
-4.3636e+00, -4.9851e+00],
[-2.1578e+00, -5.2997e+00, -2.6786e+00, ..., -3.5033e+00,
-1.5675e+00, -3.4662e+00]],
[[ 6.8743e-01, 1.0440e+00, 7.5189e-02, ..., 1.1053e+00,
1.5210e+00, 1.4367e+00],
[ 6.1770e-01, -7.0659e-01, 2.9524e-02, ..., 9.0309e-01,
8.1070e-01, 1.9457e+00],
[-1.7908e-01, 2.1129e-01, 1.7518e+00, ..., 7.2524e-01,
1.8585e-01, 1.3992e+00],
...,
[ 1.3064e+00, -1.1009e+00, -1.0302e+00, ..., 1.8185e+00,
2.3435e+00, 1.4665e+00],
[ 1.9010e+00, 8.6868e-01, 1.3385e+00, ..., 2.4218e+00,
1.8806e+00, 2.5676e+00],
[ 1.2743e+00, 6.2021e-01, 1.2539e+00, ..., 1.1857e+00,
1.5784e+00, 1.5515e+00]],
[[-2.7793e-01, -5.0615e-01, -1.1549e+00, ..., -6.0050e-01,
-9.0312e-01, -1.1433e-01],
[-1.4764e+00, -2.0275e+00, -1.8880e+00, ..., -1.0594e+00,
-9.8352e-01, -6.2383e-01],
[-2.4069e+00, -7.3980e-01, 1.1598e+00, ..., -9.7940e-03,
-3.6543e-01, -2.3261e-01],
...,
[-1.6182e+00, -2.3682e+00, -2.7014e-01, ..., -1.6562e+00,
-5.4296e-01, -4.1532e-01],
[-1.5550e+00, -1.4062e+00, -1.0196e+00, ..., 9.4400e-02,
-5.0308e-02, 4.2692e-01],
[ 7.2232e-02, 4.6381e-01, -4.3506e-01, ..., -1.1440e-01,
3.3378e-02, 5.8117e-01]]],
[[[ 9.5237e-01, 1.4246e+00, 9.5280e-01, ..., -1.5552e+00,
1.8576e-02, 2.6832e-01],
[-4.8782e-01, -1.8796e+00, -6.9910e-01, ..., 3.6017e-01,
-4.0650e-02, -1.4854e+00],
[-1.7478e-01, -4.0206e+00, -2.7950e+00, ..., -1.1219e-02,
1.7227e+00, 6.1405e-01],
...,
[-2.0083e-01, -2.1415e+00, 6.1666e-01, ..., -1.2862e+00,
-3.1154e+00, 1.2894e+00],
[-4.7401e-01, -2.3745e+00, -8.5906e-01, ..., -2.4765e-01,
-5.8225e-01, 2.6553e-01],
[ 3.7135e-01, -4.4692e-01, -8.3069e-01, ..., -6.2509e-01,
-2.4147e+00, -2.5084e-01]],
[[-3.7695e+00, -4.2250e+00, -4.9120e+00, ..., -2.6894e+00,
-1.8278e+00, -9.9586e-01],
[-4.8120e+00, -6.4235e+00, -7.1913e+00, ..., -2.9293e+00,
-1.9614e+00, -3.4863e-01],
[-5.5895e+00, -6.1959e+00, -1.6293e+00, ..., -4.6581e+00,
-1.8229e+00, -8.3788e-01],
...,
[-3.0764e+00, -2.4501e+00, -4.0717e+00, ..., -4.8827e+00,
-5.1275e+00, -5.1184e+00],
[-3.7409e+00, -2.0038e+00, -2.2331e+00, ..., -3.0341e+00,
-7.1540e+00, -4.5469e+00],
[-1.0511e+00, -8.8871e-01, -1.6978e+00, ..., -8.0347e+00,
-7.6824e+00, -2.7114e+00]],
[[ 1.9665e+00, 2.6026e+00, 2.4787e+00, ..., 4.0733e+00,
1.5959e+00, 3.0668e+00],
[ 1.1145e+00, 2.7315e+00, 2.3190e+00, ..., 1.0911e-01,
-1.5864e-01, 2.5274e+00],
[ 2.1580e+00, 1.2050e+00, -1.9547e+00, ..., -3.3907e-01,
-9.3339e-01, 1.9255e+00],
...,
[ 1.2237e+00, 1.7775e+00, 2.7573e+00, ..., -4.5521e+00,
-3.5973e+00, 3.5078e+00],
[ 8.7161e-01, 1.0733e+00, 1.5428e+00, ..., -5.7447e+00,
-4.0352e+00, 1.3611e+00],
[ 1.7727e+00, 1.5364e+00, 1.6678e+00, ..., -8.0223e+00,
-6.4592e+00, 3.5969e-02]],
...,
[[-1.6082e+00, -5.9105e+00, -5.8303e+00, ..., -9.0178e-01,
-2.7980e+00, -3.9934e+00],
[-2.1252e+00, -5.0948e+00, -4.7754e+00, ..., -8.0320e+00,
-7.7481e+00, -4.3090e+00],
[-8.3873e-01, -4.9690e+00, -5.2394e+00, ..., -9.7017e+00,
-9.5292e+00, -1.8913e+00],
...,
[-1.9168e+00, -5.0027e+00, -5.7140e+00, ..., 1.0466e+00,
-1.3143e-01, -1.1143e+01],
[-1.8251e+00, -5.5840e+00, -5.6267e+00, ..., -4.5385e+00,
-9.0923e+00, -1.5323e+01],
[-1.7407e+00, -5.3009e+00, -4.3175e+00, ..., -6.2626e+00,
-8.0560e+00, -8.2985e+00]],
[[-3.2709e-02, -5.4518e-01, -3.1628e-01, ..., 9.4521e-01,
1.9213e+00, 1.2976e+00],
[ 3.1365e-01, -3.5996e+00, -5.8255e+00, ..., -8.7430e-01,
3.5085e+00, 2.6705e+00],
[-3.3038e-01, -7.0310e-01, 2.8330e-01, ..., -4.4934e+00,
3.1961e+00, 2.1110e+00],
...,
[ 7.8842e-01, -9.4038e-01, -2.4142e+00, ..., -1.3874e+00,
-1.4085e+00, 9.1505e-01],
[ 1.5852e+00, 1.2790e+00, 1.4199e+00, ..., -1.8118e-01,
-3.0848e-01, 1.8396e+00],
[ 7.7189e-01, 1.0207e+00, 1.1150e+00, ..., -1.5005e+00,
-1.2947e+00, 5.1218e-01]],
[[-6.1616e-01, -5.9771e-01, -2.4460e-01, ..., -2.6706e+00,
-3.8547e-01, 9.0320e-01],
[-2.5363e+00, -2.8270e+00, -4.2337e-01, ..., 6.7442e-01,
-8.2977e-01, 2.0676e-01],
[-2.9570e+00, -2.8299e+00, 1.7474e+00, ..., -4.8911e-01,
-1.9480e+00, 2.8746e-01],
...,
[-1.8099e+00, -1.8131e+00, 8.9652e-02, ..., -2.6361e+00,
1.1610e+00, -2.0862e+00],
[-1.5524e+00, -1.9868e+00, -9.3213e-01, ..., -1.2147e+00,
1.4105e+00, -9.2032e-01],
[-1.2309e-01, 1.0625e-01, -3.9868e-01, ..., -4.5084e+00,
-3.9696e+00, -2.4792e+00]]]], device='cuda:0'),
tensor([[[[ 1.0799e+00, 1.3527e+00, 1.3649e+00, ..., 1.2263e+00,
9.7947e-01, 4.3498e-01],
[-5.7491e-01, -2.4912e+00, -1.6672e+00, ..., -2.1540e-01,
9.5186e-02, -1.3850e+00],
[-3.8360e-01, -1.1248e+00, -1.3238e+00, ..., -3.6722e-01,
-4.6981e-03, 6.1848e-01],
...,
[ 6.2923e-01, 5.8991e-03, 6.4535e-01, ..., 3.6461e-01,
9.7769e-01, 8.3957e-01],
[-4.8328e-02, -4.8975e-01, 2.6233e-01, ..., 9.1487e-01,
1.5971e+00, 1.3742e-01],
[ 7.8243e-01, -1.1840e-01, -1.2824e-01, ..., 5.2201e-01,
2.5344e-01, 7.8643e-01]],
[[-2.6662e+00, 1.3621e-01, -1.6403e+00, ..., -1.9111e+00,
-2.4245e+00, -6.7506e-01],
[-4.0823e+00, -2.8315e+00, -3.4812e+00, ..., -3.0720e+00,
-3.7841e+00, -1.5396e+00],
[-3.0480e+00, -2.4754e+00, -3.6196e+00, ..., -2.9323e+00,
-2.9712e+00, -1.3827e+00],
...,
[-5.3275e-01, -7.9031e-01, -1.0897e+00, ..., -2.5409e-01,
7.3335e-02, 7.7519e-01],
[-2.0407e+00, -6.4323e-01, -1.8508e-01, ..., 1.9673e-01,
-4.1010e-02, 6.9570e-01],
[-1.2483e+00, -1.1752e+00, -9.1893e-01, ..., -3.1828e-01,
-5.2617e-01, -1.6474e-01]],
[[ 2.3975e+00, 2.9089e+00, 2.9984e+00, ..., 1.6141e+00,
5.2161e-01, 2.6095e+00],
[ 8.0708e-01, 6.0543e-01, 3.1474e+00, ..., 1.1243e+00,
6.1718e-01, 2.6717e+00],
[ 1.2419e+00, 1.4276e+00, 2.8106e+00, ..., 5.1405e-01,
1.6833e+00, 2.8492e+00],
...,
[ 5.0770e-01, 8.8172e-01, 2.3783e+00, ..., 7.6966e-01,
4.4510e-01, 1.9450e+00],
[ 7.2644e-01, 1.9262e+00, 2.0833e+00, ..., 1.0028e+00,
6.9323e-01, 1.8457e+00],
[ 2.1000e+00, 1.9595e+00, 2.4430e+00, ..., 2.0056e+00,
9.6144e-01, 1.8265e+00]],
...,
[[-2.5640e+00, -6.8225e+00, -5.8736e+00, ..., -5.2464e+00,
-4.6510e+00, -5.3827e+00],
[-3.2130e+00, -7.3311e+00, -5.4653e+00, ..., -5.4951e+00,
-5.3238e+00, -5.2012e+00],
[-2.1512e+00, -5.9386e+00, -4.1090e+00, ..., -5.3360e+00,
-4.1814e+00, -3.4014e+00],
...,
[-2.1996e+00, -4.5162e+00, -2.3332e+00, ..., -4.3005e+00,
-2.7475e+00, -3.0317e+00],
[-2.0663e+00, -4.9150e+00, -3.7467e+00, ..., -4.4962e+00,
-3.1257e+00, -5.1177e+00],
[-2.1954e+00, -5.3288e+00, -2.7608e+00, ..., -3.1405e+00,
-1.5800e+00, -3.5661e+00]],
[[ 4.2977e-01, 1.1553e+00, 4.9576e-02, ..., 1.1568e+00,
1.5601e+00, 1.7028e+00],
[ 1.2882e+00, 8.1638e-01, -5.3031e-01, ..., 1.2289e+00,
1.4348e+00, 2.7729e+00],
[ 1.3480e+00, 7.3869e-01, -1.1983e-01, ..., 1.6115e+00,
1.2189e+00, 2.0310e+00],
...,
[ 1.3469e+00, 1.3928e-01, 2.8360e-01, ..., 2.0463e+00,
1.9570e+00, 1.3759e+00],
[ 1.9021e+00, 9.6448e-01, 1.9728e+00, ..., 1.9511e+00,
1.6465e+00, 2.4284e+00],
[ 1.2625e+00, 7.0744e-01, 1.2835e+00, ..., 1.3294e+00,
1.5517e+00, 1.5113e+00]],
[[-5.9283e-01, 9.8571e-02, -5.8308e-01, ..., 2.7155e-01,
-9.0429e-02, 9.2818e-01],
[-1.4385e+00, -2.1499e+00, -2.2922e+00, ..., -1.5341e+00,
-7.4112e-01, -4.0097e-02],
[-2.2901e+00, -1.2885e+00, -1.0103e+00, ..., -6.0773e-01,
-3.4253e-01, -1.4152e-01],
...,
[-1.6747e+00, -1.4828e+00, 1.3834e-01, ..., -3.2849e-01,
-3.5226e-01, -4.7942e-01],
[-1.5111e+00, -1.2770e+00, -3.7394e-01, ..., -3.6423e-01,
-3.4814e-01, 4.0270e-01],
[-2.2070e-02, 5.2750e-01, -2.0461e-01, ..., 1.0069e-01,
5.5151e-02, 6.7005e-01]]],
[[[ 8.8577e-01, 1.5298e+00, 1.2300e+00, ..., 1.7753e+00,
4.8248e-01, 9.5186e-01],
[-9.7874e-01, -2.0431e+00, -1.2029e+00, ..., 2.4894e+00,
-6.4328e-02, -1.8058e+00],
[-1.9957e-01, -7.7729e-01, 9.0181e-02, ..., -2.7366e+00,
1.2485e+00, 8.2651e-01],
...,
[-7.1223e-02, -1.5773e+00, -1.3511e+00, ..., 1.1901e+00,
3.4033e-01, 8.0234e-02],
[-6.8042e-01, -1.8326e+00, -7.4257e-01, ..., 1.0570e+00,
4.3389e-01, -2.4100e-01],
[-1.6925e-01, -1.1640e+00, 2.5397e-01, ..., 4.1531e-01,
-1.3052e+00, -2.9202e-01]],
[[-3.6731e+00, -6.9570e-01, -1.5037e+00, ..., -4.2459e+00,
-3.7116e+00, -1.3059e+00],
[-3.9509e+00, -2.9555e+00, -1.9063e+00, ..., -4.7446e+00,
-3.4693e+00, -1.0509e+00],
[-2.3937e+00, -1.8422e+00, -1.6519e+00, ..., -5.5877e+00,
-3.5414e+00, -1.2965e+00],
...,
[-1.5085e+00, -2.9821e-01, 3.8725e-01, ..., -5.8177e+00,
-3.5550e+00, -6.0461e-01],
[-3.3000e+00, 5.6999e-01, 2.8035e-02, ..., -5.3486e+00,
-2.5668e+00, -1.1346e+00],
[-2.3210e+00, -8.4930e-01, -1.6931e-01, ..., -3.8871e+00,
-2.0466e+00, -1.0535e+00]],
[[ 2.9057e+00, 4.3409e+00, 3.4086e+00, ..., -6.5402e-01,
1.5382e+00, 2.8079e+00],
[ 2.0843e-01, 1.1641e+00, 2.2665e+00, ..., -2.8775e+00,
-3.2695e+00, 2.2503e+00],
[ 9.6201e-01, 1.5826e+00, 1.6473e+00, ..., -1.8715e+00,
-1.2906e-01, 2.1026e+00],
...,
[ 9.5908e-01, 2.5168e+00, 1.1223e+00, ..., -2.8714e+00,
2.8218e-01, 2.2966e+00],
[ 5.7627e-01, 2.5400e+00, -5.1671e-01, ..., -5.8573e-01,
5.1721e-01, 1.8195e+00],
[ 1.7601e+00, 1.9886e+00, -1.1824e+00, ..., 7.4580e-01,
8.5097e-01, 1.7800e+00]],
...,
[[-1.8421e+00, -7.2517e+00, -5.4735e+00, ..., -4.4413e+00,
-1.7248e+00, -5.7513e+00],
[-2.3864e+00, -6.9537e+00, -4.1731e+00, ..., -8.3069e+00,
-3.0661e+00, -4.7158e+00],
[-9.0566e-01, -4.4251e+00, -3.6853e+00, ..., -4.3911e-01,
-1.6647e+00, -1.7064e+00],
...,
[-2.7404e+00, -5.4025e+00, -3.4992e+00, ..., -7.9995e+00,
-4.7069e+00, -4.5884e+00],
[-3.0384e+00, -6.3106e+00, -3.2181e+00, ..., -7.0121e+00,
-6.5207e+00, -6.5161e+00],
[-1.9216e+00, -4.9009e+00, -1.9679e+00, ..., -2.5764e+00,
-3.1691e+00, -3.2618e+00]],
[[-1.2749e-01, 3.1422e-01, 6.0075e-01, ..., -1.1029e+00,
1.2954e+00, 1.2786e+00],
[ 1.6088e+00, 1.0395e+00, 1.9095e+00, ..., -2.6763e+00,
2.6329e+00, 2.8895e+00],
[ 1.1424e+00, 2.4255e-01, 8.7846e-01, ..., -3.5059e+00,
7.6425e-01, 2.3507e+00],
...,
[ 1.4464e+00, 5.9641e-01, 3.9937e-01, ..., -2.6092e+00,
1.9656e+00, 1.8795e+00],
[ 1.9980e+00, 2.3327e+00, 1.0900e+00, ..., -9.1072e-01,
2.2307e+00, 2.1038e+00],
[ 4.3996e-01, 1.2063e+00, 1.9297e+00, ..., -4.8665e-01,
1.0127e+00, 1.6921e+00]],
[[-5.4141e-01, 2.9799e-01, 1.6357e-01, ..., -1.4974e-02,
5.6268e-01, 1.3437e+00],
[-1.3399e+00, -1.9072e+00, -1.7306e+00, ..., 4.8009e-01,
-1.0769e+00, -4.3734e-01],
[-2.1730e+00, -8.9776e-01, 2.7116e-01, ..., -5.7888e-01,
6.2626e-01, 4.5311e-01],
...,
[-2.0440e+00, -1.7530e+00, -8.0619e-01, ..., -1.3920e+00,
-3.7891e+00, -1.1767e+00],
[-1.6767e+00, -1.6734e+00, -1.1073e+00, ..., -8.9937e-01,
-2.6182e+00, -4.5399e-01],
[-8.4213e-01, -2.6393e-01, -1.3836e+00, ..., -1.1456e+00,
-1.6805e+00, -7.5602e-02]]],
[[[ 9.3148e-01, 1.4973e+00, 1.2046e+00, ..., 1.3871e+00,
1.0528e+00, 9.7374e-01],
[-9.6785e-01, -2.3664e+00, -1.2357e+00, ..., -4.5680e-01,
-5.3614e-01, -1.4611e+00],
[-1.2565e-01, -1.1365e+00, 7.4209e-02, ..., -2.6388e-01,
6.5166e-01, 8.2166e-01],
...,
[-4.5604e-02, -1.1002e+00, -8.7482e-01, ..., -2.1866e-01,
1.5946e-01, 2.6968e-01],
[-6.2103e-01, -2.0039e+00, -1.3598e+00, ..., -5.4341e-01,
-3.7671e-01, -1.2607e-01],
[-3.0288e-01, -1.7736e+00, -1.6667e+00, ..., -1.1549e+00,
-1.6052e+00, -2.7777e-01]],
[[-4.1720e+00, -5.3113e-01, -1.3243e+00, ..., -1.5514e+00,
-2.3722e+00, -1.0899e+00],
[-4.1772e+00, -2.3701e+00, -1.5991e+00, ..., -2.3859e+00,
-3.0279e+00, -7.9896e-01],
[-2.4853e+00, -1.4034e+00, -1.8909e+00, ..., -2.4409e+00,
-2.8002e+00, -9.5644e-01],
...,
[-1.6078e+00, -1.6579e+00, -2.5410e+00, ..., -2.0980e+00,
-2.0037e+00, -8.2840e-01],
[-3.1443e+00, -1.7654e+00, -2.9685e+00, ..., -2.2871e+00,
-2.3723e+00, -1.3317e+00],
[-2.3597e+00, -2.0478e+00, -2.9221e+00, ..., -2.0318e+00,
-1.9848e+00, -1.0625e+00]],
[[ 3.2973e+00, 4.4756e+00, 3.5548e+00, ..., 2.7846e+00,
1.1090e+00, 2.8749e+00],
[ 8.1284e-01, 1.0583e+00, 2.4249e+00, ..., 1.0669e+00,
-8.2243e-01, 2.5987e+00],
[ 1.6246e+00, 1.4079e+00, 1.7032e+00, ..., 5.8975e-01,
-1.8792e-01, 1.9924e+00],
...,
[ 8.9413e-01, 1.1294e+00, 1.3091e+00, ..., 5.4708e-01,
1.1424e+00, 2.3754e+00],
[ 6.0732e-01, 1.8314e+00, 1.2400e+00, ..., 9.5417e-03,
1.0693e+00, 1.9415e+00],
[ 1.6584e+00, 1.6895e+00, 1.8783e+00, ..., 1.2479e+00,
1.8196e+00, 1.8372e+00]],
...,
[[-2.7119e+00, -7.9955e+00, -5.8057e+00, ..., -4.0413e+00,
-3.1902e+00, -5.3474e+00],
[-3.2713e+00, -7.5398e+00, -4.4295e+00, ..., -3.7241e+00,
-2.7242e+00, -4.1926e+00],
[-1.2607e+00, -4.7895e+00, -3.6056e+00, ..., -2.7634e+00,
-2.9992e+00, -2.4914e+00],
...,
[-2.5780e+00, -5.4775e+00, -4.3411e+00, ..., -4.3706e+00,
-4.7125e+00, -4.7894e+00],
[-2.8185e+00, -5.8163e+00, -5.4267e+00, ..., -6.8002e+00,
-5.7670e+00, -6.3818e+00],
[-1.5276e+00, -4.1588e+00, -2.8397e+00, ..., -3.2835e+00,
-2.6375e+00, -3.2781e+00]],
[[-3.1576e-01, 4.9573e-01, 4.2557e-01, ..., 6.8602e-01,
8.5602e-01, 1.4266e+00],
[ 1.8847e+00, 1.3817e+00, 1.6114e+00, ..., 1.7590e+00,
2.5221e+00, 2.9512e+00],
[ 1.4027e+00, 4.4828e-01, 3.7223e-01, ..., 1.5848e+00,
1.7249e+00, 2.3601e+00],
...,
[ 1.3372e+00, 1.5343e-01, -2.0110e-01, ..., 1.4353e+00,
1.1322e+00, 1.5695e+00],
[ 1.9112e+00, 1.6797e+00, 1.0797e+00, ..., 1.5355e+00,
1.2494e+00, 1.9460e+00],
[ 4.4491e-01, 7.2885e-01, 5.8972e-01, ..., 5.1713e-01,
6.9181e-01, 1.6701e+00]],
[[-6.3987e-01, 3.5650e-01, -4.3623e-02, ..., 5.4898e-01,
5.3797e-01, 1.3581e+00],
[-1.5880e+00, -1.8084e+00, -1.9175e+00, ..., -8.7721e-01,
-4.3879e-01, 2.5055e-01],
[-2.4305e+00, -9.6705e-01, 9.7861e-02, ..., 2.7184e-01,
8.0864e-02, 7.7085e-01],
...,
[-1.9147e+00, -1.7931e+00, -1.5412e+00, ..., -1.1772e+00,
-7.7512e-01, -9.5060e-01],
[-1.5776e+00, -2.0992e+00, -1.8291e+00, ..., -6.5920e-01,
-1.1710e+00, -7.7000e-01],
[-8.3940e-01, -8.5596e-01, -1.2391e+00, ..., -6.3187e-01,
-8.5439e-01, -1.1871e-01]]],
[[[ 8.7918e-01, 1.5080e+00, 1.2871e+00, ..., 1.5092e+00,
1.1608e+00, 1.0490e+00],
[-9.2196e-01, -1.8423e+00, -8.1830e-01, ..., -5.1706e-01,
-7.3400e-01, -1.6412e+00],
[-1.9075e-01, -4.9286e-01, -7.2480e-01, ..., -1.0023e+00,
7.0585e-01, 7.5382e-01],
...,
[-3.3132e-02, -1.1011e+00, -9.6134e-01, ..., 6.0242e-01,
3.7963e-01, 3.7435e-01],
[-6.5420e-01, -1.8580e+00, -1.3867e+00, ..., 1.2578e-01,
-1.0047e-01, -1.4171e-01],
[-3.4053e-01, -1.6382e+00, -1.6459e+00, ..., -1.1093e+00,
-1.5015e+00, -2.6430e-01]],
[[-3.7423e+00, -6.6058e-01, -1.5960e+00, ..., -1.4607e+00,
-2.4171e+00, -1.2460e+00],
[-4.0342e+00, -3.1495e+00, -4.4783e+00, ..., -2.1632e+00,
-3.3628e+00, -9.8604e-01],
[-2.4435e+00, -2.7804e+00, -8.4473e+00, ..., -1.8883e+00,
-2.7475e+00, -1.1183e+00],
...,
[-1.5976e+00, -1.6536e+00, -2.7250e+00, ..., -2.9890e+00,
-1.9336e+00, -3.5651e-01],
[-3.0498e+00, -1.6406e+00, -2.8302e+00, ..., -2.1097e+00,
-2.0741e+00, -1.1315e+00],
[-2.1772e+00, -1.8239e+00, -2.6041e+00, ..., -1.9815e+00,
-1.9188e+00, -9.4787e-01]],
[[ 2.9253e+00, 4.3227e+00, 3.3991e+00, ..., 2.9299e+00,
1.0873e+00, 3.1246e+00],
[ 2.1305e-01, 9.1420e-01, 1.8723e+00, ..., 9.9984e-01,
-1.0734e+00, 2.8662e+00],
[ 8.6562e-01, 1.4684e+00, 2.7208e+00, ..., 1.8576e+00,
-9.5679e-02, 1.9659e+00],
...,
[ 1.0209e+00, 1.1069e+00, 1.5480e+00, ..., -6.8516e-01,
1.3707e+00, 1.7503e+00],
[ 7.5445e-01, 1.8000e+00, 1.2064e+00, ..., 2.8376e-01,
1.2289e+00, 1.7370e+00],
[ 1.6649e+00, 1.5597e+00, 1.8666e+00, ..., 1.3850e+00,
1.8069e+00, 1.7676e+00]],
...,
[[-1.8603e+00, -7.2270e+00, -5.3611e+00, ..., -4.0890e+00,
-3.3473e+00, -5.7212e+00],
[-2.4338e+00, -6.9615e+00, -4.8698e+00, ..., -3.5839e+00,
-2.4985e+00, -4.3333e+00],
[-8.8727e-01, -4.8276e+00, -6.1946e+00, ..., -2.7748e+00,
-2.9640e+00, -2.6731e+00],
...,
[-2.6201e+00, -5.3747e+00, -4.6063e+00, ..., -4.3688e+00,
-4.1653e+00, -6.0223e+00],
[-2.7815e+00, -5.6896e+00, -5.3857e+00, ..., -5.4453e+00,
-5.9299e+00, -6.2529e+00],
[-1.4893e+00, -4.1578e+00, -2.7987e+00, ..., -3.5104e+00,
-2.7942e+00, -3.2181e+00]],
[[-1.3229e-01, 3.4668e-01, 4.9923e-01, ..., 6.7571e-01,
8.9523e-01, 1.3794e+00],
[ 1.5953e+00, 8.4231e-01, 9.2573e-01, ..., 1.4341e+00,
2.3284e+00, 3.0495e+00],
[ 1.1205e+00, -1.3529e+00, -3.4717e+00, ..., 2.4435e+00,
1.7615e+00, 2.4042e+00],
...,
[ 1.3480e+00, 2.3108e-01, -4.8107e-01, ..., 5.3384e-01,
1.7420e+00, 1.8596e+00],
[ 1.9072e+00, 1.7389e+00, 1.0010e+00, ..., 1.5809e+00,
1.3276e+00, 2.0537e+00],
[ 5.4792e-01, 8.2982e-01, 6.7493e-01, ..., 5.5177e-01,
6.8910e-01, 1.6679e+00]],
[[-5.1761e-01, 3.1198e-01, 1.1392e-01, ..., 6.5764e-01,
5.9025e-01, 1.3730e+00],
[-1.2733e+00, -2.0536e+00, -2.6519e+00, ..., -1.1122e+00,
-7.3638e-01, 1.2548e-01],
[-2.1199e+00, -2.3502e+00, -8.3370e-01, ..., -1.4116e-02,
1.3757e-01, 6.0334e-01],
...,
[-2.0474e+00, -1.8019e+00, -1.6844e+00, ..., -9.9769e-01,
-9.0836e-01, -6.3340e-01],
[-1.7186e+00, -2.0299e+00, -1.7323e+00, ..., -5.6905e-01,
-7.9067e-01, -5.6629e-01],
[-8.6620e-01, -8.5028e-01, -1.1105e+00, ..., -3.3411e-01,
-7.0362e-01, -2.0367e-02]]],
[[[ 6.9318e-01, 9.8373e-01, 7.2873e-01, ..., 9.4710e-01,
9.0322e-01, 2.9556e-01],
[-3.3165e-01, -1.4543e+00, -5.4777e-01, ..., -3.4915e-01,
1.8080e-01, -6.3686e-01],
[ 1.5623e-01, -8.2609e-01, -4.7289e-01, ..., -6.9496e-03,
2.3398e-01, 4.9719e-01],
...,
[ 1.0913e-01, -1.0754e+00, -9.5872e-01, ..., -6.3619e-01,
7.5976e-01, 6.3780e-01],
[-3.4583e-01, -1.7440e+00, -6.9780e-01, ..., 7.3219e-01,
9.3700e-01, 1.1947e-01],
[ 3.9497e-01, -4.6671e-01, -7.5968e-01, ..., 1.9941e-02,
-3.0726e-01, 5.2837e-01]],
[[-3.0779e+00, -1.6054e+00, -3.1859e+00, ..., -1.6061e+00,
-2.9331e+00, -1.0052e+00],
[-4.2886e+00, -1.6374e+00, -1.8749e+00, ..., -1.1190e+00,
-2.8995e+00, -7.2557e-01],
[-4.3018e+00, -2.4157e+00, -2.4834e+00, ..., -2.9248e+00,
-3.1728e+00, -9.3789e-01],
...,
[-2.5174e+00, -1.7497e+00, -2.8374e+00, ..., -2.5698e+00,
-1.7910e+00, 4.4790e-01],
[-3.4967e+00, -1.3132e+00, -2.0767e+00, ..., 1.1294e-01,
-6.6205e-01, 2.7914e-01],
[-1.1598e+00, -7.7587e-01, -1.5784e+00, ..., -5.0807e-01,
-3.8718e-01, 2.4480e-02]],
[[ 1.7433e+00, 2.7820e+00, 2.2874e+00, ..., 2.2709e+00,
7.5778e-01, 1.9602e+00],
[ 2.2252e-01, 1.1075e+00, 1.2822e+00, ..., 2.0013e+00,
1.8798e-01, 2.1955e+00],
[ 7.5368e-01, 1.2210e+00, 3.7146e-01, ..., 3.9680e-02,
-6.4005e-01, 3.1985e-01],
...,
[ 7.8442e-01, 1.0571e+00, 4.2981e-01, ..., -1.6443e+00,
3.1154e-01, 1.2085e+00],
[ 6.3499e-01, 1.7381e+00, 8.4833e-01, ..., 6.9218e-01,
5.2213e-01, 9.3565e-01],
[ 1.5765e+00, 1.5293e+00, 1.5269e+00, ..., 7.1804e-01,
6.0130e-01, 1.3069e+00]],
...,
[[-1.4603e+00, -4.6172e+00, -3.7625e+00, ..., -4.7114e+00,
-3.2640e+00, -3.8551e+00],
[-1.8106e+00, -3.8687e+00, -3.7789e+00, ..., -5.8142e+00,
-3.7313e+00, -4.1383e+00],
[-9.7357e-01, -3.3541e+00, -3.2409e+00, ..., -5.2650e+00,
-3.4454e+00, -1.7160e+00],
...,
[-1.4519e+00, -3.6810e+00, -4.1824e+00, ..., -3.9805e+00,
-2.6244e+00, -3.3361e+00],
[-1.4574e+00, -4.2109e+00, -5.0233e+00, ..., -6.7775e+00,
-4.5150e+00, -5.3709e+00],
[-1.5487e+00, -4.6912e+00, -4.0238e+00, ..., -4.5646e+00,
-1.9963e+00, -3.2343e+00]],
[[ 7.6878e-01, 4.1997e-01, -2.8327e-01, ..., 1.3125e+00,
8.6625e-01, 1.6627e+00],
[ 1.2575e+00, 1.1006e+00, 6.8667e-01, ..., 2.5006e+00,
1.6117e+00, 2.5967e+00],
[ 6.0455e-01, 6.7558e-01, 1.6980e-01, ..., -1.3292e-01,
-1.5545e+00, 6.7920e-01],
...,
[ 8.4186e-01, -1.4249e-01, -3.5926e-01, ..., 1.9125e+00,
1.8035e+00, 1.9184e+00],
[ 1.4889e+00, 1.3626e+00, 1.0822e+00, ..., 3.1626e+00,
1.5650e+00, 2.5345e+00],
[ 7.1110e-01, 9.0074e-01, 9.7781e-01, ..., 1.3414e+00,
1.1813e+00, 1.3678e+00]],
[[-2.9629e-01, -2.2931e-01, -8.9104e-01, ..., 5.0214e-01,
2.1240e-01, 1.0741e+00],
[-1.4270e+00, -1.5605e+00, -1.7969e+00, ..., -4.3205e-01,
-1.5651e-01, 6.5496e-01],
[-1.6573e+00, -1.0156e+00, -9.6534e-01, ..., -4.2056e-01,
-4.9067e-01, 5.5960e-01],
...,
[-1.7438e+00, -1.6138e+00, -1.8391e+00, ..., -4.0898e+00,
-2.1738e+00, -1.1349e+00],
[-1.3783e+00, -1.6002e+00, -1.5366e+00, ..., -9.0963e-01,
-6.3499e-01, 7.4458e-02],
[-1.6620e-01, 3.9508e-02, -5.7492e-01, ..., -1.1682e+00,
-7.1120e-01, 2.2724e-01]]]], device='cuda:0'),
tensor([[[[ 3.5193e-01, 2.2564e-01, -2.4468e+00, ..., 1.6578e-01,
6.1969e-02, -4.8355e-01],
[-1.3977e+00, 9.5914e-02, -1.8491e+00, ..., 6.4644e-01,
1.3112e+00, -3.4219e-01],
[ 2.5680e-01, -2.9293e+00, -3.7325e+00, ..., 2.9396e-01,
7.2610e-01, 8.9927e-01],
...,
[ 2.7866e-01, -5.7310e-01, -4.3392e-01, ..., 3.8866e-01,
7.1033e-02, 7.9693e-02],
[-6.3659e-01, -1.8406e+00, -1.3825e+00, ..., -6.3577e-01,
-2.3692e-01, -1.4264e-01],
[-2.9676e-01, -1.7563e+00, -1.6619e+00, ..., -1.1348e+00,
-1.4802e+00, -2.5477e-01]],
[[-2.4440e-01, 1.6151e-01, 6.6191e-01, ..., -2.7919e+00,
-3.3741e+00, -8.5823e-01],
[-1.2431e+00, 1.4852e+00, -6.7857e-01, ..., -1.9646e+00,
-1.9475e+00, -1.7509e-01],
[-2.7155e+00, -3.0765e+00, -2.2072e+00, ..., -1.7658e+00,
-1.9801e+00, -6.7892e-01],
...,
[-1.4290e+00, -1.2558e+00, -1.9278e+00, ..., -2.5427e+00,
-2.2046e+00, -2.7103e-01],
[-3.0106e+00, -1.6090e+00, -2.7931e+00, ..., -1.9846e+00,
-2.1975e+00, -1.1445e+00],
[-2.3414e+00, -2.0080e+00, -2.6726e+00, ..., -1.9190e+00,
-1.8691e+00, -9.4258e-01]],
[[ 1.5335e+00, -8.9201e-01, -1.5749e+00, ..., 2.4086e+00,
5.9649e-01, 2.2824e+00],
[-1.9281e+00, -2.8291e+00, -1.4463e+00, ..., 1.8851e+00,
2.3023e-02, 2.3111e+00],
[-4.4710e-01, -6.5470e+00, -4.3643e+00, ..., 1.2113e+00,
9.8975e-01, 2.2372e+00],
...,
[ 1.1828e+00, 1.4757e+00, 1.8910e+00, ..., 2.1163e+00,
1.2057e+00, 1.6899e+00],
[ 6.8695e-01, 1.9055e+00, 1.3207e+00, ..., 6.3448e-02,
1.0734e+00, 1.7983e+00],
[ 1.7072e+00, 1.6104e+00, 1.8910e+00, ..., 1.2917e+00,
1.7446e+00, 1.7591e+00]],
...,
[[-8.5123e-01, -3.5968e+00, 8.3586e-03, ..., -2.0859e+00,
-2.8143e+00, -3.8885e+00],
[-4.2381e+00, -5.6805e+00, -2.3049e+00, ..., -4.3732e+00,
-4.4575e+00, -5.1803e+00],
[-1.2443e+00, -3.4201e+00, 4.7070e-02, ..., -1.0492e+00,
-2.4661e+00, -2.4382e+00],
...,
[-2.3712e+00, -5.4178e+00, -4.0101e+00, ..., -8.4352e+00,
-9.7800e+00, -8.1944e+00],
[-2.9164e+00, -6.0617e+00, -5.4616e+00, ..., -6.8915e+00,
-5.6524e+00, -6.2263e+00],
[-1.5993e+00, -4.3016e+00, -2.7843e+00, ..., -3.3198e+00,
-2.7679e+00, -3.2812e+00]],
[[-1.9799e-01, -1.5820e+00, -1.0323e-01, ..., 3.3744e-01,
9.5267e-01, 1.6149e+00],
[ 9.4238e-01, 1.0037e+00, -4.2864e-01, ..., 7.2877e-01,
1.8054e+00, 2.5849e+00],
[-1.0358e+00, -9.7799e-01, -2.1883e+00, ..., 8.6635e-01,
1.1465e+00, 1.8903e+00],
...,
[ 1.3254e+00, 2.8272e-01, 6.8892e-03, ..., 4.0952e+00,
2.9753e+00, 2.4040e+00],
[ 1.9525e+00, 1.7548e+00, 1.0038e+00, ..., 1.6907e+00,
1.3103e+00, 1.9853e+00],
[ 4.7862e-01, 8.2391e-01, 6.9184e-01, ..., 5.2573e-01,
6.9432e-01, 1.6233e+00]],
[[-1.3422e-01, -1.1034e-01, 3.8411e-01, ..., -3.2901e-01,
-4.3139e-01, 5.9169e-01],
[-1.7614e+00, -5.5052e-01, 3.0891e-01, ..., -4.3605e-01,
4.5169e-01, 8.1725e-01],
[-2.0303e+00, -9.8207e-01, -4.1850e+00, ..., -3.2594e-01,
3.7216e-01, -3.0010e-03],
...,
[-1.9155e+00, -1.8016e+00, -1.0678e+00, ..., -1.1304e+00,
-7.4934e-01, -5.2312e-02],
[-1.6715e+00, -2.0344e+00, -1.6955e+00, ..., -5.7175e-01,
-1.0476e+00, -6.5222e-01],
[-8.6209e-01, -8.9689e-01, -1.1841e+00, ..., -5.8793e-01,
-8.4065e-01, -7.0056e-02]]],
[[[ 8.6278e-01, 1.0472e+00, 6.6793e-01, ..., 1.0136e-01,
8.4541e-02, -3.6324e-01],
[-4.9248e-01, -9.5933e-01, -1.2154e-01, ..., 6.2682e-01,
1.2742e+00, -3.8279e-01],
[-2.7289e-02, -2.0740e-01, 4.9988e-02, ..., -1.5745e-02,
8.8348e-01, 9.4181e-01],
...,
[ 2.5433e-01, -6.3418e-01, -3.4519e-01, ..., -5.5041e-01,
1.0584e-01, 4.6363e-01],
[-7.0009e-01, -2.0049e+00, -1.4170e+00, ..., 1.1542e+00,
7.6823e-01, -6.3374e-02],
[-1.5736e-01, -1.8847e+00, -1.6858e+00, ..., -1.1798e+00,
-1.2224e+00, 1.6151e-02]],
[[-3.4050e+00, -1.4281e+00, -2.3878e+00, ..., -2.2460e+00,
-3.1209e+00, -7.5303e-01],
[-3.1378e+00, -2.4389e+00, -1.5216e+00, ..., -1.1123e+00,
-2.1903e+00, -2.5759e-01],
[-1.9242e+00, -1.8839e+00, -1.8736e+00, ..., -1.8604e+00,
-1.9262e+00, -5.7742e-01],
...,
[-1.5759e+00, -1.2664e+00, -2.1461e+00, ..., -1.8028e+00,
-1.8763e+00, -6.2442e-01],
[-3.5201e+00, -1.4578e+00, -2.9815e+00, ..., -1.0859e+00,
-2.3667e+00, -1.4119e+00],
[-2.2921e+00, -1.5442e+00, -2.2838e+00, ..., -3.3328e+00,
-2.8231e+00, -1.2133e+00]],
[[ 2.0732e+00, 2.5970e+00, 2.3108e+00, ..., 2.1270e+00,
4.4391e-01, 2.1715e+00],
[ 3.3973e-01, 5.2577e-01, 2.1071e+00, ..., 1.5695e+00,
-2.6056e-01, 2.1448e+00],
[ 1.0382e+00, 8.9042e-01, 1.7281e+00, ..., 9.2597e-01,
8.7473e-01, 2.3939e+00],
...,
[ 1.2794e+00, 1.7061e+00, 2.1127e+00, ..., 4.9412e-01,
1.4585e+00, 2.5625e+00],
[ 5.4937e-01, 2.1178e+00, 1.4177e+00, ..., -7.1067e-01,
8.5623e-01, 1.8247e+00],
[ 1.6739e+00, 1.6008e+00, 1.6025e+00, ..., -2.7103e-01,
1.8319e+00, 2.2256e+00]],
...,
[[-1.8990e+00, -6.3458e+00, -4.6576e+00, ..., -2.9113e+00,
-2.4912e+00, -3.7728e+00],
[-2.4593e+00, -7.4597e+00, -4.7855e+00, ..., -4.8382e+00,
-4.2541e+00, -4.7763e+00],
[-1.5040e+00, -4.5795e+00, -2.5606e+00, ..., -2.2928e+00,
-2.2954e+00, -2.0670e+00],
...,
[-2.3018e+00, -5.3911e+00, -4.3424e+00, ..., -2.7885e+00,
-2.6395e+00, -3.8769e+00],
[-2.9649e+00, -6.0166e+00, -5.1845e+00, ..., -5.2183e+00,
-5.2932e+00, -6.2544e+00],
[-1.8705e+00, -4.6003e+00, -2.7579e+00, ..., -4.9516e-01,
-4.4338e-01, -2.5709e+00]],
[[-3.8505e-01, 2.0475e-01, 3.4586e-01, ..., 7.0815e-01,
9.8931e-01, 1.5975e+00],
[ 1.0251e+00, 4.3324e-01, 8.9734e-01, ..., 1.2044e+00,
1.6696e+00, 2.4782e+00],
[ 9.9082e-01, 5.7108e-02, 2.9664e-01, ..., 9.2814e-01,
1.0493e+00, 1.8026e+00],
...,
[ 1.3555e+00, 3.0104e-01, 6.0912e-03, ..., 1.5021e+00,
1.9687e+00, 2.0278e+00],
[ 1.9030e+00, 1.8752e+00, 1.0359e+00, ..., 6.9058e-01,
9.4163e-01, 1.8288e+00],
[ 4.7563e-01, 1.0416e+00, 8.9839e-01, ..., -3.3791e-01,
1.8911e-01, 1.6496e+00]],
[[-6.4960e-01, -1.8759e-01, -4.2967e-01, ..., -3.9017e-01,
-4.0021e-01, 6.6905e-01],
[-1.1346e+00, -1.0017e+00, -5.9177e-01, ..., 3.4903e-02,
3.9016e-01, 7.6045e-01],
[-1.7502e+00, -1.7551e-01, 5.4085e-01, ..., 2.8012e-01,
3.6497e-01, -6.4438e-02],
...,
[-1.9663e+00, -2.0905e+00, -1.4215e+00, ..., -6.7188e-01,
-6.8302e-01, -3.4917e-01],
[-1.6354e+00, -2.2755e+00, -1.9495e+00, ..., 6.4087e-01,
-6.6067e-01, -7.6306e-01],
[-8.1173e-01, -8.3548e-01, -1.2076e+00, ..., -2.0552e+00,
-1.7864e+00, -4.6206e-01]]],
[[[ 8.4821e-01, 1.6984e+00, 1.1514e+00, ..., 1.4153e+00,
1.1199e+00, 9.7587e-01],
[-9.9596e-01, -2.0185e+00, -1.0592e+00, ..., -5.7674e-01,
-5.2315e-01, -1.5707e+00],
[-1.6075e-01, -9.3874e-01, -8.4245e-01, ..., -3.9694e-01,
6.2954e-01, 8.4561e-01],
...,
[ 8.0093e-03, -1.1485e+00, -8.8611e-01, ..., -1.2131e+00,
-2.8298e-01, 7.7397e-02],
[-5.8945e-01, -1.8208e+00, -1.2549e+00, ..., 2.4755e-01,
1.1957e-01, 1.2766e-01],
[-2.4985e-01, -1.6575e+00, -1.6648e+00, ..., -1.1632e+00,
-1.5431e+00, -3.1997e-01]],
[[-3.6113e+00, -1.1723e+00, -2.4994e+00, ..., -1.6128e+00,
-2.4516e+00, -1.2195e+00],
[-3.8853e+00, -3.4178e+00, -2.8785e+00, ..., -2.3786e+00,
-3.0692e+00, -9.1133e-01],
[-2.3121e+00, -2.2168e+00, -2.0890e+00, ..., -2.3050e+00,
-2.7579e+00, -1.0419e+00],
...,
[-1.5649e+00, -1.6487e+00, -2.3074e+00, ..., -3.9080e+00,
-3.9251e+00, -1.5891e+00],
[-2.9893e+00, -1.6369e+00, -2.8027e+00, ..., -2.4170e+00,
-2.2642e+00, -1.0958e+00],
[-2.2587e+00, -1.9820e+00, -2.7164e+00, ..., -2.0011e+00,
-2.0159e+00, -1.0438e+00]],
[[ 2.7957e+00, 4.3015e+00, 3.4629e+00, ..., 2.8116e+00,
1.0326e+00, 3.0387e+00],
[ 1.2869e-01, 1.6235e+00, 2.7392e+00, ..., 1.1507e+00,
-8.3515e-01, 2.7418e+00],
[ 9.6638e-01, 2.0372e+00, 1.2137e+00, ..., 7.9928e-01,
-3.2185e-01, 2.0318e+00],
...,
[ 1.0104e+00, 1.0190e+00, 1.2423e+00, ..., -2.4738e+00,
-2.9779e-01, 1.3738e+00],
[ 7.5251e-01, 1.7794e+00, 1.1925e+00, ..., 8.0250e-01,
1.6088e+00, 1.7850e+00],
[ 1.6968e+00, 1.6368e+00, 1.8719e+00, ..., 1.3058e+00,
1.7839e+00, 1.7688e+00]],
...,
[[-1.7356e+00, -7.1799e+00, -5.7197e+00, ..., -3.9568e+00,
-3.3321e+00, -5.6208e+00],
[-2.2918e+00, -6.9571e+00, -4.6881e+00, ..., -3.5588e+00,
-2.7886e+00, -4.4418e+00],
[-9.5749e-01, -4.4112e+00, -4.6217e+00, ..., -2.9305e+00,
-3.2027e+00, -2.4707e+00],
...,
[-2.5922e+00, -5.1475e+00, -4.0493e+00, ..., -5.0983e+00,
-5.0852e+00, -4.1828e+00],
[-2.6936e+00, -5.5384e+00, -5.2329e+00, ..., -7.0471e+00,
-6.7182e+00, -7.2583e+00],
[-1.4487e+00, -4.0256e+00, -2.7307e+00, ..., -3.1464e+00,
-2.6721e+00, -3.1712e+00]],
[[-1.4799e-01, -1.0903e-01, 5.3784e-01, ..., 5.5411e-01,
8.5304e-01, 1.3983e+00],
[ 1.5678e+00, 6.2093e-01, 5.8342e-01, ..., 1.7394e+00,
2.5611e+00, 3.0350e+00],
[ 1.1398e+00, 1.4706e-01, 1.5146e+00, ..., 1.5771e+00,
1.5436e+00, 2.3704e+00],
...,
[ 1.2640e+00, 2.8294e-01, -2.4992e-02, ..., -7.1763e-01,
-5.7698e-02, 1.6332e+00],
[ 1.8840e+00, 1.6874e+00, 9.7392e-01, ..., 1.5018e+00,
1.4723e+00, 2.0548e+00],
[ 4.3324e-01, 7.5647e-01, 6.9662e-01, ..., 4.7134e-01,
6.3901e-01, 1.6482e+00]],
[[-5.3052e-01, 2.8945e-02, 5.7579e-02, ..., 5.4386e-01,
5.5459e-01, 1.3576e+00],
[-1.3370e+00, -2.2015e+00, -1.4954e+00, ..., -8.7804e-01,
-4.4175e-01, 2.1538e-01],
[-2.1694e+00, -1.2997e+00, 1.8568e-01, ..., 1.3659e-01,
5.2695e-02, 7.5263e-01],
...,
[-1.9237e+00, -1.7918e+00, -1.2916e+00, ..., -3.0359e+00,
-2.8947e+00, -1.1966e+00],
[-1.5448e+00, -1.9331e+00, -1.6451e+00, ..., -8.7470e-01,
-8.1436e-01, 4.6892e-02],
[-7.4525e-01, -8.0774e-01, -1.1573e+00, ..., -6.1884e-01,
-8.6925e-01, -9.4762e-02]]],
[[[ 5.6976e-01, 7.6252e-01, 8.1469e-01, ..., 3.6261e-01,
4.2024e-01, 8.8122e-03],
[-2.9675e-01, -1.2099e+00, 3.0883e-01, ..., 2.0034e-01,
1.1785e-01, -3.3274e-01],
[-1.9148e-01, -6.4895e-01, 9.9080e-01, ..., -1.1001e-01,
2.7131e-02, 6.1635e-01],
...,
[ 4.4178e-01, -2.5277e-01, -7.8152e-01, ..., 3.7088e-01,
8.4987e-01, 7.0634e-01],
[-4.7808e-02, -6.7924e-01, -1.7060e-01, ..., 1.0172e+00,
1.5632e+00, 1.4324e-01],
[ 8.4392e-01, -2.0153e-01, -4.8342e-01, ..., 4.2684e-01,
3.6831e-01, 7.0919e-01]],
[[-2.4086e+00, -6.3214e-01, -2.4957e+00, ..., -1.6866e+00,
-2.2902e+00, -1.5229e+00],
[-2.5699e+00, -1.6122e+00, -3.6380e+00, ..., -1.2491e+00,
-2.3082e+00, -1.6533e+00],
[-2.3898e+00, -3.1768e+00, -5.3819e+00, ..., -1.9915e+00,
-2.2252e+00, -1.1034e+00],
...,
[-7.4597e-01, -4.6272e-01, -2.6834e+00, ..., -7.4510e-01,
-2.5196e-01, 7.2810e-01],
[-2.6598e+00, -1.8846e-01, -1.0514e+00, ..., 6.2269e-02,
-2.0277e-01, 6.2800e-01],
[-1.6215e+00, -1.3011e+00, -1.5256e+00, ..., -3.5875e-01,
-8.0350e-01, -2.3522e-01]],
[[ 1.9671e+00, 2.9684e+00, 1.1687e+00, ..., 1.5833e+00,
8.1814e-01, 2.1158e+00],
[-2.0193e-01, 1.6031e+00, 6.7536e-01, ..., 8.2731e-01,
3.7126e-01, 2.2407e+00],
[ 8.2723e-02, 1.5543e+00, 1.0394e-01, ..., -6.8585e-02,
7.3657e-01, 1.9691e+00],
...,
[ 8.0676e-01, 4.0690e-01, 2.0220e+00, ..., 3.0595e-01,
9.6159e-02, 1.6376e+00],
[ 1.1799e+00, 2.2707e+00, 2.4592e+00, ..., 9.0443e-01,
6.9741e-01, 1.8242e+00],
[ 2.4483e+00, 1.9004e+00, 2.4597e+00, ..., 1.8505e+00,
1.0624e+00, 1.8276e+00]],
...,
[[-2.0806e+00, -5.2528e+00, -4.2807e+00, ..., -2.0941e+00,
-1.6844e+00, -3.0489e+00],
[-2.4959e+00, -5.3699e+00, -5.2807e+00, ..., -2.4522e+00,
-2.3045e+00, -3.0294e+00],
[-1.5389e+00, -5.4476e+00, -5.1121e+00, ..., -2.3375e+00,
-2.4968e+00, -2.2823e+00],
...,
[-2.5591e+00, -4.0443e+00, -2.2499e+00, ..., -2.5334e+00,
-1.9287e+00, -2.7258e+00],
[-2.4810e+00, -5.4117e+00, -3.9338e+00, ..., -4.1659e+00,
-3.0678e+00, -4.9658e+00],
[-2.6585e+00, -5.9314e+00, -2.7340e+00, ..., -2.7199e+00,
-1.5917e+00, -3.3053e+00]],
[[ 9.3153e-01, 1.2864e+00, 8.9969e-01, ..., 1.0787e+00,
1.4921e+00, 1.3787e+00],
[ 1.1917e+00, 1.2048e-01, 6.6164e-01, ..., 8.8151e-01,
8.0120e-01, 1.8942e+00],
[ 6.7937e-01, -1.3711e+00, -4.8935e-02, ..., 3.9838e-01,
1.0375e-01, 1.3967e+00],
...,
[ 1.5453e+00, 3.1337e-02, -2.1809e-01, ..., 1.6498e+00,
1.8153e+00, 1.4677e+00],
[ 2.0640e+00, 1.2529e+00, 1.5469e+00, ..., 1.9118e+00,
1.6968e+00, 2.5413e+00],
[ 1.1608e+00, 8.6792e-01, 1.0881e+00, ..., 1.2419e+00,
1.4212e+00, 1.5779e+00]],
[[-9.3441e-02, -9.2984e-01, -1.5455e+00, ..., -5.5267e-01,
-9.0230e-01, -1.4684e-01],
[-9.0993e-01, -3.2224e+00, -3.2586e+00, ..., -1.1387e+00,
-1.0001e+00, -6.8732e-01],
[-1.4936e+00, -2.3299e+00, -2.0002e+00, ..., -1.4222e-01,
-3.0873e-01, -2.8761e-01],
...,
[-1.8441e+00, -1.5406e+00, -1.3815e+00, ..., -2.6276e-01,
-4.2150e-01, -3.8903e-01],
[-1.8222e+00, -1.2919e+00, -1.0280e+00, ..., -1.5279e-01,
-3.9320e-01, 4.4104e-01],
[-2.8119e-01, 3.8173e-01, -4.3214e-01, ..., 2.7833e-01,
-3.1528e-02, 6.2955e-01]]],
[[[ 1.0789e+00, 1.3815e+00, 1.1524e+00, ..., 8.7572e-01,
8.7896e-01, 4.1702e-01],
[-6.1041e-01, -2.5454e+00, -1.3442e+00, ..., -3.4829e-01,
-1.3123e-01, -1.2408e+00],
[-4.6988e-01, -1.2069e+00, -7.2255e-01, ..., -4.5811e-01,
-7.4942e-02, 5.0418e-01],
...,
[ 3.8736e-01, 2.9197e-01, 1.0838e+00, ..., 1.0870e-01,
8.0368e-01, 8.2166e-01],
[-1.0530e-01, -9.6961e-02, 1.0414e+00, ..., 9.6366e-01,
1.3792e+00, 9.7926e-02],
[ 7.5970e-01, 4.7142e-01, 9.1314e-01, ..., 1.1416e+00,
5.6838e-01, 6.6997e-01]],
[[-2.9088e+00, -2.9044e-02, -1.8175e+00, ..., -1.3523e+00,
-2.4462e+00, -7.7198e-01],
[-4.1622e+00, -2.8149e+00, -3.2090e+00, ..., -2.3282e+00,
-3.8733e+00, -1.5116e+00],
[-3.0106e+00, -2.4688e+00, -3.5166e+00, ..., -3.2925e+00,
-3.0615e+00, -1.2690e+00],
...,
[-7.7976e-01, -1.4449e-01, 1.3256e+00, ..., -7.5922e-01,
-4.4689e-01, 7.6347e-01],
[-2.3085e+00, -4.1815e-01, 1.4966e+00, ..., -1.3474e+00,
-1.2975e-01, 6.8566e-01],
[-1.4773e+00, -1.3280e+00, 1.4830e-01, ..., -1.4673e+00,
-3.3078e-01, -1.7165e-01]],
[[ 2.5182e+00, 2.9523e+00, 2.9470e+00, ..., 2.1032e+00,
8.0448e-01, 2.6552e+00],
[ 8.3723e-01, 5.4150e-01, 2.9059e+00, ..., 1.8483e+00,
6.4903e-01, 2.6484e+00],
[ 1.1307e+00, 1.3896e+00, 2.6961e+00, ..., 7.1340e-01,
1.7921e+00, 2.8117e+00],
...,
[ 1.0061e+00, 8.7011e-01, -1.5229e+00, ..., 1.2425e+00,
4.4609e-01, 1.7817e+00],
[ 9.9045e-01, 1.5949e+00, -2.8099e-01, ..., 8.8084e-01,
7.9453e-01, 1.6252e+00],
[ 2.2649e+00, 1.7855e+00, 1.5502e+00, ..., 1.3438e+00,
9.9833e-01, 1.7794e+00]],
...,
[[-2.7873e+00, -7.0797e+00, -6.0648e+00, ..., -4.4831e+00,
-4.2778e+00, -5.2831e+00],
[-3.4138e+00, -7.5537e+00, -5.6778e+00, ..., -5.7976e+00,
-5.3091e+00, -5.1750e+00],
[-2.2581e+00, -6.1790e+00, -4.4385e+00, ..., -4.9005e+00,
-3.9709e+00, -3.3128e+00],
...,
[-2.1124e+00, -4.6131e+00, -1.4230e+00, ..., -1.5274e+00,
-2.6853e+00, -3.0918e+00],
[-2.0131e+00, -4.7669e+00, -2.7035e+00, ..., -6.8591e+00,
-3.5212e+00, -5.0409e+00],
[-2.2726e+00, -5.0235e+00, -1.9328e+00, ..., -4.6099e+00,
-1.7879e+00, -3.4084e+00]],
[[ 3.3414e-01, 1.1409e+00, -2.8214e-02, ..., 1.6731e+00,
1.4419e+00, 1.6843e+00],
[ 1.3346e+00, 8.9135e-01, -9.8498e-02, ..., 1.4158e+00,
1.4247e+00, 2.7654e+00],
[ 1.3685e+00, 7.9779e-01, 1.6664e-01, ..., 1.2854e+00,
1.3035e+00, 2.1165e+00],
...,
[ 1.4389e+00, 7.3900e-01, 1.4817e+00, ..., 1.2153e+00,
2.0727e+00, 1.3935e+00],
[ 1.8687e+00, 1.2473e+00, 2.1122e+00, ..., 1.0110e+00,
2.1153e+00, 2.5293e+00],
[ 1.1616e+00, 8.6570e-01, 1.7450e+00, ..., 8.0388e-01,
1.8633e+00, 1.5774e+00]],
[[-6.3309e-01, 7.4810e-02, -6.2247e-01, ..., 4.1488e-01,
-1.2979e-01, 9.0420e-01],
[-1.5010e+00, -2.1666e+00, -2.2551e+00, ..., -6.2655e-01,
-7.0662e-01, 7.0713e-02],
[-2.2893e+00, -1.4410e+00, -1.0795e+00, ..., -3.6681e-01,
-4.9620e-01, -1.2171e-01],
...,
[-1.8845e+00, -1.2050e+00, 3.1603e-01, ..., -3.0879e-01,
-7.2371e-01, -7.3588e-01],
[-1.5759e+00, -8.9333e-01, 3.6734e-01, ..., -8.0949e-01,
-6.3911e-01, 3.4507e-01],
[-2.9995e-02, 7.8265e-01, 3.9689e-02, ..., -9.0658e-01,
-2.5016e-02, 6.8547e-01]]]], device='cuda:0'),
tensor([[[[ 1.2146e+00, 7.0761e-01, 1.3148e+00, ..., -2.0750e-01,
-1.5725e-01, -5.7070e-01],
[-7.8921e-01, -1.5643e-01, 3.5559e-02, ..., -4.6967e-01,
4.3066e-01, -4.6931e-01],
[-7.8214e-01, 4.2739e-01, 1.4466e+00, ..., 1.7326e+00,
1.1358e+00, 3.4369e-01],
...,
[ 2.0745e-01, -1.5328e+00, 3.1466e-02, ..., 6.0056e-01,
4.4680e-01, 6.1441e-01],
[-6.6619e-01, -2.8704e+00, -1.2483e+00, ..., -1.6694e-02,
-4.7713e-01, -1.4449e-01],
[-1.5624e-01, -1.6703e+00, -3.0771e-01, ..., -1.9954e-01,
-1.3684e+00, -3.0414e-01]],
[[-3.4876e+00, -1.0500e+00, 1.1214e+00, ..., -2.5895e+00,
-3.4844e+00, -9.1094e-01],
[-4.1312e+00, -4.1744e+00, -1.0320e+00, ..., -2.7709e+00,
-2.4105e+00, 1.7793e-02],
[-2.7780e+00, -1.5856e+00, 1.2248e+00, ..., -2.1906e+00,
-2.2738e+00, -6.5771e-01],
...,
[-1.4181e+00, -2.4218e+00, -2.2359e+00, ..., -2.4568e+00,
-2.0147e+00, -5.5023e-01],
[-3.3490e+00, -1.6462e+00, -1.9914e+00, ..., -3.8017e+00,
-2.1691e+00, -1.2194e+00],
[-2.3554e+00, -1.4993e+00, -1.1240e+00, ..., -3.4527e+00,
-1.8911e+00, -1.0521e+00]],
[[ 2.8530e+00, 1.4265e+00, -2.4632e+00, ..., 2.9482e+00,
6.4582e-01, 2.0802e+00],
[ 1.7393e+00, -1.5560e+00, -8.7545e+00, ..., 2.0821e+00,
8.4064e-01, 2.3742e+00],
[ 1.8509e+00, 8.8825e-01, -5.1676e+00, ..., -5.8647e-02,
1.1351e+00, 2.6114e+00],
...,
[ 1.1637e+00, 2.8679e+00, 1.8975e+00, ..., 9.0771e-01,
1.2380e+00, 2.3285e+00],
[ 6.0833e-01, 3.2773e+00, 1.2784e+00, ..., 5.7241e-01,
6.8789e-01, 1.8332e+00],
[ 1.7484e+00, 2.0472e+00, -3.1258e-01, ..., 1.2737e+00,
1.7101e+00, 1.7710e+00]],
...,
[[-2.1260e+00, -7.1543e+00, -3.7826e+00, ..., -3.5051e+00,
-2.9104e+00, -3.8695e+00],
[-2.5973e+00, -8.8963e+00, -3.5177e+00, ..., -4.1751e+00,
-4.2218e+00, -5.5418e+00],
[-1.2461e+00, -4.5819e+00, -2.2839e+00, ..., -5.0558e+00,
-5.6581e+00, -3.3964e+00],
...,
[-2.4878e+00, -5.4647e+00, -4.2978e+00, ..., -3.3812e+00,
-4.4755e+00, -4.8140e+00],
[-3.0581e+00, -6.2191e+00, -5.3758e+00, ..., -5.7078e+00,
-6.2772e+00, -6.1524e+00],
[-1.9516e+00, -4.9049e+00, -2.5624e+00, ..., -2.9197e+00,
-2.6854e+00, -3.0986e+00]],
[[-6.5014e-01, 8.0805e-01, 1.6362e+00, ..., 9.9259e-01,
1.0350e+00, 1.6239e+00],
[-1.1596e-01, -2.6379e-01, 9.3053e-01, ..., 8.5454e-01,
2.5391e+00, 2.8005e+00],
[ 5.8931e-01, 7.8491e-01, 1.3793e+00, ..., -1.0143e-01,
1.3088e+00, 2.3214e+00],
...,
[ 1.4847e+00, -4.2237e-01, -1.0758e+00, ..., 6.7060e-01,
2.0516e+00, 1.5791e+00],
[ 2.0175e+00, 1.6630e+00, 9.7313e-01, ..., 8.8669e-01,
2.1907e+00, 2.0013e+00],
[ 4.3511e-01, 7.6607e-01, 1.1250e+00, ..., 5.7181e-02,
1.2778e+00, 1.6480e+00]],
[[-9.9485e-01, -5.4347e-01, 1.6974e+00, ..., -8.1152e-01,
-8.2291e-01, 5.3301e-01],
[-1.4926e+00, -1.5245e+00, 1.1232e+00, ..., -1.1364e+00,
-1.9587e-01, 9.6712e-01],
[-1.9825e+00, -5.4070e-01, 2.5418e-01, ..., 1.9222e+00,
5.8538e-01, 7.9276e-02],
...,
[-2.0299e+00, -2.7838e+00, -1.1905e+00, ..., -9.3203e-01,
-6.5113e-01, -5.2655e-01],
[-1.6714e+00, -2.7405e+00, -1.5858e+00, ..., -1.2267e+00,
-1.5807e+00, -6.9006e-01],
[-8.3759e-01, -6.2844e-01, -1.1382e+00, ..., -1.5399e+00,
-1.3196e+00, -1.1120e-01]]],
[[[ 8.5865e-01, 1.4814e+00, 1.1638e+00, ..., 1.4201e+00,
1.1038e+00, 9.8641e-01],
[-1.0014e+00, -1.9351e+00, -1.1563e+00, ..., -4.1194e-01,
-5.6766e-01, -1.5909e+00],
[-1.4200e-01, -7.8095e-01, -9.2190e-02, ..., -4.7882e-01,
6.9601e-01, 7.9579e-01],
...,
[ 4.8444e-01, -2.6984e-01, 1.6525e-01, ..., 5.5736e-01,
1.8183e-01, 6.4775e-01],
[ 3.3683e-01, -9.6209e-01, -1.5627e-01, ..., 7.3763e-01,
6.2978e-01, -6.6858e-02],
[ 5.7702e-01, 2.6288e-01, -2.6912e-02, ..., 2.9105e-01,
-7.9309e-02, 6.6127e-01]],
[[-3.7473e+00, -7.6193e-01, -1.5806e+00, ..., -1.5009e+00,
-2.4883e+00, -1.2395e+00],
[-3.9138e+00, -2.8487e+00, -2.0235e+00, ..., -2.1123e+00,
-3.2430e+00, -9.4548e-01],
[-2.4857e+00, -1.9475e+00, -1.9189e+00, ..., -2.3872e+00,
-2.8906e+00, -1.0495e+00],
...,
[-1.0126e+00, -1.2167e+00, -1.8606e+00, ..., -1.5273e+00,
-1.3052e+00, -1.8992e-01],
[-1.4263e+00, -3.8440e-01, -5.7180e-01, ..., -6.2206e-01,
-1.1098e+00, -1.3151e-01],
[ 4.3867e-01, 8.2087e-02, 6.2637e-01, ..., 3.5325e-01,
6.2588e-02, 4.0370e-01]],
[[ 2.8938e+00, 4.2437e+00, 3.3815e+00, ..., 2.8491e+00,
1.0287e+00, 3.0311e+00],
[ 2.6622e-01, 1.1351e+00, 2.1736e+00, ..., 1.1191e+00,
-9.2141e-01, 2.7579e+00],
[ 1.0274e+00, 1.7839e+00, 1.9937e+00, ..., 1.0683e+00,
-3.8562e-02, 2.0958e+00],
...,
[ 5.0000e-01, 7.8779e-01, 1.0545e+00, ..., -3.6799e-01,
2.7504e-01, 1.5301e+00],
[ 2.5533e-01, 1.4212e+00, 1.2167e+00, ..., -9.0024e-02,
2.5965e-01, 1.2020e+00],
[ 1.3907e+00, 9.6126e-01, 7.0138e-01, ..., 7.1668e-01,
4.2712e-01, 1.1497e+00]],
...,
[[-1.8979e+00, -7.2614e+00, -5.3687e+00, ..., -3.9841e+00,
-3.2614e+00, -5.5852e+00],
[-2.4578e+00, -7.1122e+00, -4.4117e+00, ..., -3.4337e+00,
-2.7243e+00, -4.3954e+00],
[-1.1338e+00, -4.8861e+00, -3.8660e+00, ..., -1.9949e+00,
-2.7123e+00, -2.4599e+00],
...,
[-2.2690e+00, -5.6481e+00, -4.0297e+00, ..., -3.4495e+00,
-3.3929e+00, -3.3559e+00],
[-2.4150e+00, -6.6152e+00, -4.6213e+00, ..., -4.9317e+00,
-4.2834e+00, -4.6887e+00],
[-2.3523e+00, -6.7411e+00, -4.4116e+00, ..., -4.1845e+00,
-3.2918e+00, -4.1015e+00]],
[[-2.1763e-01, 3.1175e-01, 5.5925e-01, ..., 6.4509e-01,
8.1900e-01, 1.4056e+00],
[ 1.6403e+00, 9.8224e-01, 1.7980e+00, ..., 1.6091e+00,
2.4551e+00, 3.0582e+00],
[ 1.1261e+00, 5.9095e-02, 7.4667e-01, ..., 1.4723e+00,
1.6329e+00, 2.4031e+00],
...,
[ 1.0902e+00, -2.9087e-01, -4.1109e-01, ..., 8.9046e-01,
8.5958e-01, 1.2881e+00],
[ 1.6249e+00, 9.6294e-01, 1.2543e+00, ..., 1.0653e+00,
6.1214e-01, 2.0938e+00],
[ 1.1452e+00, 9.1186e-01, 1.5206e+00, ..., 1.2407e+00,
1.2038e+00, 1.3870e+00]],
[[-5.7316e-01, 2.7240e-01, 1.4747e-01, ..., 6.0536e-01,
5.1959e-01, 1.3594e+00],
[-1.3453e+00, -1.8886e+00, -1.5679e+00, ..., -8.8075e-01,
-5.2627e-01, 2.3404e-01],
[-2.1634e+00, -9.3227e-01, 4.4184e-01, ..., 1.7635e-01,
2.6391e-01, 7.5218e-01],
...,
[-1.5606e+00, -1.8272e+00, -1.1176e+00, ..., -6.7030e-01,
-1.1191e+00, -7.2405e-01],
[-1.2696e+00, -1.7218e+00, -1.2784e+00, ..., -6.8176e-01,
-1.2339e+00, 1.1469e-03],
[ 1.4047e-01, 1.3296e-01, -1.2715e-01, ..., 2.8809e-01,
-6.9290e-02, 7.6696e-01]]],
[[[ 3.8429e-01, 1.8493e+00, 1.9682e+00, ..., 1.4193e-01,
3.1333e-01, 1.7823e-01],
[-2.9945e+00, -3.5057e+00, -1.1836e+00, ..., 2.9405e-01,
4.0035e-01, -1.1812e+00],
[-1.9448e+00, -3.3594e+00, -1.4930e+00, ..., -4.1316e-01,
1.0368e+00, 6.5489e-01],
...,
[ 3.5018e-01, -9.0155e-02, -9.9241e-02, ..., -4.1789e-02,
9.0399e-01, 5.5027e-01],
[-4.7762e-01, -7.4105e-01, -1.7800e-01, ..., 2.5001e-01,
1.4264e+00, -1.1676e-01],
[-1.6301e-01, -3.5183e-01, -9.7843e-01, ..., -1.9223e-01,
4.0420e-01, 6.1311e-01]],
[[-2.1899e+00, -3.0920e+00, -2.5583e+00, ..., -1.7846e+00,
-2.2152e+00, -1.0652e+00],
[-4.8887e+00, -4.5770e+00, -2.5646e+00, ..., -4.6723e-01,
-1.2423e+00, -2.5183e-01],
[-5.7075e+00, -4.6559e+00, -3.2762e+00, ..., -1.2039e+00,
-1.4376e+00, -7.3410e-01],
...,
[ 2.6119e-01, -1.3075e+00, -8.1032e-01, ..., -2.8826e-01,
-4.6610e-02, 4.8817e-01],
[-1.0990e+00, -7.9279e-01, -5.7243e-01, ..., -4.4230e-02,
-4.0462e-01, 2.7449e-01],
[-4.6384e-01, -1.1594e+00, -1.0889e+00, ..., -2.7334e-01,
-7.2826e-01, -5.3217e-01]],
[[ 8.5208e-01, 6.9965e-01, 9.9852e-01, ..., 2.9410e+00,
1.4424e+00, 2.8671e+00],
[-1.3912e+00, -3.2833e+00, -2.9839e-01, ..., 1.8464e+00,
-2.3499e-02, 2.4876e+00],
[-2.7485e-01, -4.2242e+00, -1.2060e+00, ..., 1.2756e+00,
6.2330e-01, 1.7965e+00],
...,
[-5.6655e-01, -1.9682e-01, 3.6057e-01, ..., 7.8696e-01,
9.3914e-02, 1.4433e+00],
[-9.5681e-01, 4.8558e-01, 5.0323e-01, ..., 1.0850e+00,
5.9895e-01, 1.5471e+00],
[ 6.7844e-01, 2.9820e-01, 9.7224e-01, ..., 1.4738e+00,
1.0516e+00, 2.0223e+00]],
...,
[[-1.2076e+00, -4.7216e+00, -4.6449e+00, ..., -3.4208e+00,
-2.4318e+00, -3.9126e+00],
[-1.6154e+00, -1.7541e+00, -3.0010e+00, ..., -6.2859e+00,
-3.9288e+00, -4.3590e+00],
[-8.5280e-01, 1.1088e+00, -2.6126e+00, ..., -3.8127e+00,
-2.4020e+00, -1.5661e+00],
...,
[-3.2489e+00, -8.6561e+00, -5.0331e+00, ..., -3.2783e+00,
-2.3147e+00, -1.5208e+00],
[-3.9066e+00, -9.3676e+00, -5.2871e+00, ..., -4.8525e+00,
-3.3764e+00, -3.7449e+00],
[-3.1115e+00, -7.6281e+00, -3.1220e+00, ..., -3.5770e+00,
-2.1599e+00, -2.8079e+00]],
[[ 2.3566e-01, -1.8281e+00, -7.0196e-01, ..., 1.4977e+00,
1.1098e+00, 1.2457e+00],
[ 1.7441e-01, -1.5940e+00, -4.7719e-01, ..., 2.1342e+00,
2.3397e+00, 2.5765e+00],
[-7.6316e-01, -1.4914e+00, -3.1210e-01, ..., 2.0312e+00,
1.5935e+00, 1.9281e+00],
...,
[ 1.0835e+00, 7.6741e-02, 1.2497e+00, ..., 1.6043e+00,
1.8488e+00, 1.6563e+00],
[ 1.7323e+00, 1.4502e+00, 1.5727e+00, ..., 1.5957e+00,
1.5250e+00, 2.4539e+00],
[ 8.0512e-01, 5.5953e-01, 1.0669e+00, ..., 1.1857e+00,
1.3243e+00, 1.3510e+00]],
[[-1.2669e+00, 5.2903e-01, 3.3053e-01, ..., -9.7531e-02,
-1.7229e-01, 8.7344e-01],
[-4.4328e+00, -1.2629e+00, 2.6498e-01, ..., 2.6091e-01,
-6.7431e-02, 4.0055e-01],
[-4.8549e+00, -8.9690e-01, 1.9542e-01, ..., 5.9803e-01,
1.8769e-01, 2.2560e-01],
...,
[-1.4207e+00, -2.5743e+00, -6.6672e-01, ..., -3.6423e-01,
-3.7249e-01, -3.3133e-01],
[-1.4256e+00, -2.4436e+00, -5.8793e-01, ..., -5.9858e-01,
-4.4514e-01, 3.3271e-01],
[-3.8049e-01, -1.2447e+00, -6.8947e-01, ..., 7.2949e-02,
1.6817e-01, 3.5917e-01]]],
[[[ 1.2637e+00, 1.3373e+00, 7.3612e-01, ..., 3.5150e-01,
3.9028e-01, 2.4602e-01],
[-1.0901e+00, -2.4304e+00, -1.3603e+00, ..., 2.1055e-01,
2.7720e-01, -1.2705e+00],
[-1.7556e+00, -1.0848e+00, -1.9610e-01, ..., -6.2978e-01,
6.8588e-01, 5.1095e-01],
...,
[-2.4810e-01, 7.8167e-01, 5.1174e-01, ..., -9.1758e-01,
4.8025e-01, 8.3030e-01],
[-4.3394e-01, 2.0825e-01, 5.6932e-01, ..., 4.4537e-01,
1.3876e+00, -2.7582e-02],
[-2.9350e-01, -4.1205e-01, -3.7815e-01, ..., -1.7256e-01,
4.0792e-01, 6.2156e-01]],
[[-3.8173e+00, -2.3232e+00, -1.8927e+00, ..., -1.9583e+00,
-2.4518e+00, -1.2348e+00],
[-4.1990e+00, -4.8764e+00, -1.6837e+00, ..., -2.4246e-01,
-1.3687e+00, -4.5228e-01],
[ 1.0127e+00, 1.0532e+00, 3.7649e-01, ..., -1.6695e+00,
-1.6756e+00, -9.7980e-01],
...,
[ 8.7399e-01, 6.6596e-01, 2.2088e+00, ..., -1.6257e+00,
-8.7185e-01, -3.4871e-01],
[-9.1453e-01, -9.0552e-01, -8.8941e-01, ..., 1.6943e-01,
-3.2572e-01, 2.4043e-01],
[-5.2089e-01, -1.5606e+00, -1.3746e+00, ..., -3.9548e-01,
-7.1401e-01, -5.0860e-01]],
[[ 2.3409e+00, 2.7276e+00, 2.5338e+00, ..., 3.0435e+00,
1.3375e+00, 3.0281e+00],
[-1.6710e-01, -2.1887e-01, 2.9456e-01, ..., 2.2707e+00,
-5.6559e-03, 2.5291e+00],
[-3.8929e+00, -3.2196e+00, -1.9839e+00, ..., 1.3176e+00,
9.6369e-01, 1.9231e+00],
...,
[ 4.5341e-01, -8.7604e-01, -1.2758e+00, ..., -1.6048e-01,
6.5457e-01, 2.1043e+00],
[-2.4233e-01, 1.0199e+00, 1.9840e-01, ..., 1.3407e+00,
9.5212e-01, 1.6856e+00],
[ 8.7178e-01, 1.3471e+00, 1.9690e+00, ..., 1.3734e+00,
1.0196e+00, 1.9488e+00]],
...,
[[-3.3191e+00, -9.3099e+00, -7.4400e+00, ..., -3.8022e+00,
-2.6006e+00, -4.1966e+00],
[-2.7151e+00, -8.7140e+00, -6.3343e+00, ..., -6.0100e+00,
-3.8514e+00, -4.6356e+00],
[-3.1285e+00, -8.4812e+00, -7.5420e+00, ..., -2.5891e+00,
-1.4663e+00, -1.3199e+00],
...,
[-3.6907e+00, -7.8198e+00, -2.6917e+00, ..., -3.3743e+00,
-2.1448e+00, -2.0646e+00],
[-4.0978e+00, -8.9461e+00, -3.8482e+00, ..., -5.1523e+00,
-4.2723e+00, -4.4009e+00],
[-3.1742e+00, -7.2762e+00, -2.5897e+00, ..., -3.6695e+00,
-2.0906e+00, -2.5482e+00]],
[[-1.1887e+00, 2.8141e-01, 1.0078e+00, ..., 1.2086e+00,
1.0018e+00, 1.2332e+00],
[-1.2604e+00, -1.7152e+00, 9.5739e-01, ..., 2.1074e+00,
2.3342e+00, 2.6315e+00],
[ 1.7454e+00, 1.4192e+00, 1.1361e+00, ..., 1.4433e+00,
1.7826e+00, 2.0694e+00],
...,
[ 1.4042e+00, 1.1067e+00, 1.4866e+00, ..., 1.3630e+00,
1.0564e+00, 1.8439e+00],
[ 1.7081e+00, 9.7733e-01, 6.4745e-01, ..., 1.9315e+00,
1.6030e+00, 2.4707e+00],
[ 8.7413e-01, 4.6391e-01, 1.2661e+00, ..., 1.1674e+00,
1.3550e+00, 1.3309e+00]],
[[-1.6360e-01, -2.3555e-01, -6.0423e-01, ..., -1.8688e-01,
-1.6115e-01, 9.1193e-01],
[-1.1438e+00, -2.2719e+00, -7.6828e-01, ..., 1.2829e-01,
-1.3209e-01, 3.7272e-01],
[-4.2471e-01, 1.1306e+00, 9.9044e-01, ..., 4.3721e-01,
1.5113e-01, 2.3878e-01],
...,
[-1.1481e+00, -2.8684e+00, -3.8491e-01, ..., -9.4893e-01,
-8.9936e-01, -1.5441e-01],
[-1.0256e+00, -2.2377e+00, -7.3485e-01, ..., -6.3692e-01,
-3.1846e-01, 6.4314e-01],
[-2.5379e-01, -1.3232e+00, -7.7026e-01, ..., 6.0501e-02,
1.7972e-01, 3.3540e-01]]],
[[[ 8.7097e-01, 1.5786e+00, 1.3741e+00, ..., -2.1037e+00,
7.4974e-01, 1.6057e+00],
[-7.9380e-01, -2.2592e+00, -1.4248e+00, ..., -3.4831e+00,
-2.7466e+00, -2.8435e+00],
[-1.1447e-02, -5.4534e-01, 6.8618e-03, ..., 1.5922e+00,
-9.1070e-01, -2.2353e-01],
...,
[-2.1240e-01, -2.0088e+00, 1.6685e+00, ..., 2.7720e+00,
2.7258e-01, 1.8560e-01],
[-7.2404e-01, -2.2977e+00, -1.6253e+00, ..., -4.5768e+00,
-2.2782e+00, -2.7963e-01],
[-8.4717e-02, -1.6249e+00, -2.0593e+00, ..., -1.6099e+00,
-1.8262e+00, -4.9481e-01]],
[[-3.8066e+00, -5.9842e-01, -1.4843e+00, ..., -5.4018e+00,
-3.5188e+00, -3.0473e+00],
[-4.3592e+00, -3.0429e+00, -2.3914e+00, ..., -5.8541e+00,
-9.8821e+00, -5.8699e+00],
[-2.5545e+00, -2.3725e+00, -3.4648e+00, ..., -5.6197e+00,
-7.5088e+00, -2.3760e+00],
...,
[-1.2286e+00, -4.7797e+00, -1.0545e+01, ..., -8.4086e+00,
-9.5013e+00, -3.2882e+00],
[-2.9654e+00, -3.0289e+00, -5.2186e+00, ..., -9.1732e+00,
-4.7963e+00, -1.2841e+00],
[-2.1853e+00, -2.0020e+00, -3.0354e+00, ..., -3.0690e+00,
-2.0369e+00, -6.4357e-01]],
[[ 2.9003e+00, 4.4699e+00, 3.5559e+00, ..., -1.6939e+00,
-6.6162e-01, 1.9789e+00],
[ 2.5158e-03, 1.1876e+00, 2.3691e+00, ..., -6.7162e+00,
-5.3508e+00, 9.0185e-01],
[ 4.9775e-01, 1.0915e+00, 9.1892e-01, ..., -3.9773e+00,
-2.3953e+00, 1.7217e+00],
...,
[ 7.3535e-01, 3.6203e+00, 2.1516e+00, ..., -1.3125e+01,
-6.2209e+00, 1.8306e+00],
[ 5.2430e-01, 1.9024e+00, 4.4137e+00, ..., -2.3663e+00,
1.4081e+00, 2.2307e+00],
[ 1.9509e+00, 1.4904e+00, 1.9191e+00, ..., 1.3198e+00,
1.1063e+00, 1.2264e+00]],
...,
[[-1.8096e+00, -6.9977e+00, -5.5762e+00, ..., 2.1697e+00,
-9.3807e-01, -1.4500e+01],
[-2.4156e+00, -6.4583e+00, -3.7394e+00, ..., 1.9377e+00,
-6.6423e+00, -1.4186e+01],
[-7.8829e-01, -4.0700e+00, -4.2779e+00, ..., -1.2636e+01,
-1.1655e+01, -7.1861e+00],
...,
[-3.3317e+00, -7.8442e+00, -9.1700e+00, ..., -1.8241e+01,
-1.3504e+01, -8.6933e+00],
[-3.3493e+00, -7.5954e+00, -7.9593e+00, ..., -1.1277e+01,
-8.0390e+00, -8.2698e+00],
[-1.8971e+00, -5.0381e+00, -2.9471e+00, ..., -9.2699e+00,
-7.3586e+00, -5.2411e+00]],
[[-4.8054e-02, 3.5216e-01, 5.5089e-01, ..., -6.9305e-01,
-2.0260e+00, -2.5384e-01],
[ 1.5684e+00, 1.1942e+00, 1.8406e+00, ..., -3.1062e+00,
-3.3567e-01, 1.6876e+00],
[ 9.6892e-01, 1.1395e-01, 6.1901e-01, ..., -1.0805e+00,
1.8390e+00, 2.9761e+00],
...,
[ 1.5608e+00, -3.1156e+00, -4.6257e+00, ..., -1.8734e+00,
-1.7362e+00, 3.2867e-01],
[ 2.1124e+00, 6.7926e-01, 1.7476e-01, ..., -1.3349e+00,
1.6502e+00, 3.0116e+00],
[ 6.2936e-01, 9.8227e-01, 3.7506e-01, ..., 2.4630e+00,
1.5860e+00, 1.8540e+00]],
[[-5.3838e-01, 3.7506e-01, 2.3293e-01, ..., -2.4931e+00,
1.0967e+00, 1.7544e+00],
[-1.3092e+00, -1.7484e+00, -2.0045e+00, ..., -8.6615e-01,
3.7588e-01, -3.1926e+00],
[-1.9733e+00, -9.3680e-01, 1.9814e-01, ..., -5.2258e-01,
-2.0163e+00, -1.9708e+00],
...,
[-2.0792e+00, -1.3328e+00, -1.6734e-03, ..., -2.6760e+00,
-4.7848e+00, -3.4625e+00],
[-1.9227e+00, -3.1876e+00, -2.3561e+00, ..., -5.1076e+00,
-3.2045e+00, 9.5667e-02],
[-8.3682e-01, -8.7359e-01, -1.3108e+00, ..., -1.3288e-01,
-1.8331e-01, 3.8155e-01]]]], device='cuda:0'),
tensor([[[[ 6.3765e-01, 8.9224e-01, 6.6972e-01, ..., 3.5485e-01,
4.4081e-01, 3.8966e-02],
[-1.2869e-01, -9.5189e-01, -3.1112e-01, ..., 1.4653e-01,
1.3522e-01, -2.8213e-01],
[ 2.1218e-01, -3.2443e-01, -4.3668e-01, ..., 1.3572e-01,
2.3168e-01, 6.6450e-01],
...,
[ 6.4987e-03, -8.4221e-01, -7.0907e-01, ..., 4.1752e-01,
7.3394e-01, 8.5728e-01],
[-3.7426e-01, -1.8872e+00, -8.1481e-01, ..., 1.6374e-01,
6.7676e-01, 1.5587e-01],
[ 4.9606e-01, -7.6478e-01, -1.1411e+00, ..., -7.5259e-01,
-6.4032e-01, 1.3963e-01]],
[[-2.4957e+00, -1.0679e+00, -2.2091e+00, ..., -1.6526e+00,
-2.2971e+00, -1.5542e+00],
[-2.6589e+00, -1.3862e+00, -1.1907e+00, ..., -1.2256e+00,
-2.3326e+00, -1.6085e+00],
[-2.5128e+00, -2.1969e+00, 6.5161e-01, ..., -1.6958e+00,
-2.2234e+00, -1.1211e+00],
...,
[-1.0866e+00, -7.0942e-01, -1.7544e+00, ..., -1.4431e+00,
-1.2279e+00, 8.8514e-02],
[-2.2993e+00, -2.3989e-02, -7.2684e-01, ..., -9.4463e-01,
-1.5221e+00, -2.7784e-01],
[-1.0453e+00, -1.2438e+00, -1.6161e+00, ..., -1.8895e+00,
-1.7283e+00, -6.4151e-01]],
[[ 1.9444e+00, 2.3543e+00, 1.5619e+00, ..., 1.6116e+00,
7.5071e-01, 2.0953e+00],
[-2.5103e-01, -5.2784e-02, 8.7405e-01, ..., 8.4216e-01,
2.9037e-01, 2.2441e+00],
[ 4.4812e-01, -1.8893e-02, 8.1638e-03, ..., 9.6793e-02,
7.6800e-01, 1.9770e+00],
...,
[ 1.1094e+00, 1.0996e+00, 1.9630e+00, ..., 4.4338e-01,
-4.2480e-01, 1.3312e+00],
[ 7.7646e-01, 1.9800e+00, 2.1277e+00, ..., -1.1770e-01,
-5.3464e-01, 9.4692e-01],
[ 1.7950e+00, 1.0259e+00, 2.1212e+00, ..., 8.1257e-01,
4.5136e-01, 1.2282e+00]],
...,
[[-2.0234e+00, -5.0901e+00, -3.2570e+00, ..., -2.0243e+00,
-1.6575e+00, -3.0123e+00],
[-2.4656e+00, -5.6025e+00, -2.4867e+00, ..., -2.3941e+00,
-2.2300e+00, -2.9185e+00],
[-1.6138e+00, -5.8608e+00, -3.1044e+00, ..., -2.1538e+00,
-2.1988e+00, -1.8748e+00],
...,
[-2.0293e+00, -5.3069e+00, -3.1536e+00, ..., -3.2596e+00,
-2.8692e+00, -3.0032e+00],
[-2.8595e+00, -7.1688e+00, -4.8696e+00, ..., -6.1654e+00,
-3.7130e+00, -4.3891e+00],
[-3.0146e+00, -7.3950e+00, -3.5992e+00, ..., -3.7612e+00,
-2.1885e+00, -2.7147e+00]],
[[ 9.4311e-01, 1.3204e+00, 5.2420e-01, ..., 1.1083e+00,
1.4920e+00, 1.3483e+00],
[ 9.7439e-01, 3.2928e-01, 2.9103e-01, ..., 9.9791e-01,
8.5030e-01, 1.9189e+00],
[ 8.5047e-03, 2.0672e-01, 6.8647e-01, ..., 5.2949e-01,
8.0113e-02, 1.3150e+00],
...,
[ 1.6162e+00, 6.0598e-02, 1.2326e-01, ..., 2.0321e+00,
1.6168e+00, 1.8372e+00],
[ 2.2343e+00, 1.6539e+00, 1.0873e+00, ..., 2.0665e+00,
1.2797e+00, 2.2202e+00],
[ 9.9485e-01, 4.9717e-01, 8.2849e-01, ..., 4.0631e-01,
6.5800e-01, 1.1651e+00]],
[[-2.1543e-01, -6.4069e-01, -9.7612e-01, ..., -5.5367e-01,
-9.0442e-01, -1.1360e-01],
[-1.4463e+00, -2.6068e+00, -2.0109e+00, ..., -1.2099e+00,
-1.0979e+00, -6.1656e-01],
[-2.2056e+00, -1.7152e+00, 7.8182e-01, ..., -4.5163e-02,
-3.3311e-01, -2.2263e-01],
...,
[-2.6293e+00, -1.9696e+00, -1.3553e+00, ..., -8.7601e-01,
-2.3226e-01, 1.9507e-01],
[-2.5005e+00, -2.2050e+00, -1.1022e+00, ..., -3.2330e-01,
-4.4875e-01, 3.9108e-01],
[-3.8886e-01, -3.1100e-01, -4.4455e-01, ..., -5.0401e-01,
-5.6521e-01, 4.8225e-02]]],
[[[ 5.9410e-01, 8.4938e-01, 5.5744e-01, ..., 3.5200e-01,
4.1234e-01, -6.8231e-03],
[-2.5268e-01, -8.8121e-01, -5.4489e-01, ..., -2.9154e-01,
-1.6193e-02, -3.5430e-01],
[-1.1170e-01, 8.6058e-02, 6.4972e-01, ..., -1.3269e+00,
-5.9206e-03, 6.4389e-01],
...,
[-1.4955e-01, -8.6116e-01, -6.6232e-01, ..., 1.7139e+00,
1.9225e+00, 1.3075e+00],
[-5.3431e-01, -1.8013e+00, -9.0576e-01, ..., -4.2539e+00,
-4.5613e-01, 1.3542e+00],
[ 4.0923e-01, -6.8993e-01, -1.1140e+00, ..., -5.8347e+00,
-4.8367e+00, -2.0491e+00]],
[[-2.3673e+00, -6.5839e-01, -1.9127e+00, ..., -1.6916e+00,
-2.3109e+00, -1.5325e+00],
[-2.4449e+00, -7.5494e-01, -1.5932e+00, ..., -1.1392e+00,
-2.5788e+00, -1.6953e+00],
[-2.4125e+00, -1.5897e+00, -2.8328e+00, ..., -2.5393e+00,
-2.5310e+00, -1.2553e+00],
...,
[-1.2563e+00, -1.0521e+00, -1.5581e+00, ..., 6.7979e-02,
-1.8293e+00, -7.7561e-01],
[-2.0391e+00, -4.3627e-01, -1.1808e+00, ..., -4.6274e+00,
-5.2666e+00, -9.9570e-01],
[-7.5808e-01, -1.3637e+00, -1.9887e+00, ..., -8.7111e+00,
-8.0412e+00, -3.5199e+00]],
[[ 1.9078e+00, 2.4942e+00, 1.6779e+00, ..., 1.6234e+00,
7.9179e-01, 2.1015e+00],
[-1.9761e-01, 3.6392e-01, 6.5563e-01, ..., 1.6352e+00,
1.6722e-01, 2.3235e+00],
[ 1.3260e-03, 2.7817e-01, 1.9270e-01, ..., 1.3086e+00,
1.4703e+00, 2.1194e+00],
...,
[ 7.1989e-01, 9.4428e-01, 1.1625e+00, ..., -4.8691e+00,
-6.1918e+00, -9.5541e-01],
[ 6.4574e-01, 1.7183e+00, 1.2524e+00, ..., -3.5495e+00,
-4.8339e+00, -1.2753e+00],
[ 1.6267e+00, 9.8804e-01, 1.5150e+00, ..., -1.1990e+00,
-2.9571e+00, -9.4114e-02]],
...,
[[-1.9621e+00, -4.7836e+00, -3.1775e+00, ..., -2.1476e+00,
-1.7138e+00, -3.1008e+00],
[-2.4527e+00, -5.2507e+00, -2.9581e+00, ..., -2.4926e+00,
-2.9172e+00, -3.0855e+00],
[-1.6101e+00, -5.4168e+00, -3.3601e+00, ..., -1.7453e+00,
-2.7695e+00, -2.0465e+00],
...,
[-1.7706e+00, -5.0624e+00, -2.9030e+00, ..., -8.2582e+00,
-9.9970e+00, -5.1647e+00],
[-2.2768e+00, -6.4820e+00, -4.2240e+00, ..., -4.9934e+00,
-7.5728e+00, -5.6865e+00],
[-2.6029e+00, -6.7452e+00, -3.4409e+00, ..., -1.3138e-01,
2.6753e-02, -4.4873e-01]],
[[ 9.0007e-01, 1.6079e+00, 7.9319e-01, ..., 1.0865e+00,
1.5161e+00, 1.4197e+00],
[ 1.1589e+00, 9.5535e-01, 8.5411e-01, ..., 1.0152e+00,
7.5808e-01, 1.9818e+00],
[ 6.6702e-01, -4.4454e-02, -5.9623e-01, ..., 6.9979e-01,
9.7206e-01, 1.3403e+00],
...,
[ 1.4026e+00, -1.6326e-01, 4.1730e-01, ..., -1.2263e+00,
-1.9501e+00, 6.4433e-01],
[ 2.0806e+00, 1.3301e+00, 1.1315e+00, ..., -2.0480e+00,
-2.5787e+00, -2.9166e-01],
[ 1.0245e+00, 3.5369e-01, 3.3949e-01, ..., -1.8165e+00,
-1.9530e+00, -2.8785e-01]],
[[-1.1104e-01, -6.1920e-01, -1.1570e+00, ..., -5.7415e-01,
-9.0157e-01, -1.1710e-01],
[-8.6102e-01, -2.5495e+00, -2.4995e+00, ..., -1.7104e+00,
-1.0337e+00, -6.3927e-01],
[-1.4556e+00, -1.6445e+00, -1.2699e+00, ..., -1.3168e+00,
2.4981e-01, -2.0773e-01],
...,
[-2.3749e+00, -1.7183e+00, -7.8742e-01, ..., 2.1718e+00,
7.6008e-01, 1.2711e+00],
[-2.4102e+00, -1.9209e+00, -8.9493e-01, ..., 4.4830e-01,
-9.0694e-01, 1.9843e+00],
[-3.0420e-01, -2.0418e-01, -4.8237e-01, ..., -5.7166e+00,
-4.1353e+00, -4.6203e-01]]],
[[[ 5.6253e-01, 9.0899e-01, 5.6641e-01, ..., 6.3829e-01,
5.1958e-01, -8.9744e-03],
[-3.0419e-01, -5.8591e-01, 2.3430e-01, ..., 6.1093e-01,
6.1945e-02, -4.1188e-01],
[-2.2153e-01, -1.6515e-01, 1.2983e+00, ..., -6.8999e-01,
2.3917e-01, 7.9403e-01],
...,
[ 6.5607e-01, 1.0822e+00, -1.6359e+00, ..., -8.7930e-01,
1.5620e+00, 1.5225e+00],
[-2.6445e-01, -3.3489e-01, 1.0084e+00, ..., -3.1206e+00,
-2.4389e-01, 1.5985e-01],
[ 1.8775e-01, -9.5697e-01, -7.0012e-01, ..., -1.1665e+00,
-1.8914e-01, 6.7471e-01]],
[[-2.3710e+00, 3.6501e-01, 2.0750e-01, ..., -2.1136e+00,
-2.6874e+00, -1.6598e+00],
[-2.5915e+00, -5.8011e-01, -1.1880e+00, ..., -2.3040e+00,
-2.8319e+00, -1.8841e+00],
[-2.5212e+00, -1.6003e+00, -2.9861e+00, ..., -1.7661e+00,
-2.1003e+00, -1.2159e+00],
...,
[-3.2729e-01, -9.1979e-02, -5.6196e-01, ..., -1.7758e+00,
-1.8960e+00, -5.3497e-01],
[-1.8116e+00, -3.0750e-01, -3.4977e+00, ..., -4.0324e+00,
-3.2682e+00, -1.2593e+00],
[-1.1136e+00, -1.3917e+00, -2.4026e+00, ..., -3.8742e+00,
-3.0176e+00, -1.5983e+00]],
[[ 1.8194e+00, 2.4856e+00, 5.4865e-01, ..., 1.6889e+00,
4.7082e-01, 2.2202e+00],
[-1.6411e-01, 1.0257e+00, -7.8469e-02, ..., 1.0086e+00,
2.6690e-01, 2.4772e+00],
[ 2.0408e-01, 1.1161e+00, 1.0624e+00, ..., 9.2975e-01,
6.9159e-01, 2.1099e+00],
...,
[ 1.4483e+00, -7.6994e-01, -7.3809e-01, ..., -2.7452e+00,
-1.3112e+00, 1.0048e+00],
[ 1.2018e+00, 2.7351e+00, 2.6062e+00, ..., -4.1766e+00,
-1.6603e+00, 7.8646e-01],
[ 1.7937e+00, 1.5796e+00, 2.9389e+00, ..., -1.3434e+00,
-1.4806e-01, 1.4227e+00]],
...,
[[-1.7748e+00, -4.8419e+00, -2.7854e+00, ..., -2.7053e+00,
-1.9575e+00, -3.3906e+00],
[-2.1642e+00, -4.7533e+00, -2.9305e+00, ..., -3.5398e+00,
-2.1787e+00, -3.2083e+00],
[-1.2514e+00, -4.8165e+00, -2.8235e+00, ..., -2.8207e+00,
-2.8100e+00, -2.1454e+00],
...,
[-2.3026e+00, -4.5228e+00, -4.1641e+00, ..., -1.6373e-01,
-2.1861e+00, -4.2324e+00],
[-2.5740e+00, -7.0775e+00, -5.9449e+00, ..., 9.3545e-01,
-6.2061e-01, -5.0436e+00],
[-2.9954e+00, -7.0950e+00, -4.0202e+00, ..., 3.6319e-01,
3.0302e-01, -2.2676e+00]],
[[ 8.6851e-01, 1.8021e+00, 1.4324e+00, ..., 8.1466e-01,
1.5093e+00, 1.4316e+00],
[ 1.0593e+00, 6.7367e-01, 9.5215e-01, ..., 3.4457e-01,
8.3859e-01, 2.0799e+00],
[ 5.9613e-01, -1.6239e-01, -5.8852e-01, ..., 9.2090e-01,
4.2272e-01, 1.3628e+00],
...,
[ 1.3658e+00, 7.0376e-01, 1.2415e+00, ..., -5.5121e-01,
-3.0260e-01, 1.6126e+00],
[ 2.1654e+00, 7.1375e-01, -8.8656e-02, ..., -1.1853e+00,
-1.2970e+00, 1.7127e+00],
[ 1.0307e+00, 6.9118e-01, 2.2937e-01, ..., -9.8576e-01,
-4.2064e-01, 9.5315e-01]],
[[-9.5160e-02, -6.0796e-01, -1.2691e+00, ..., -5.5386e-01,
-1.0569e+00, -9.0611e-02],
[-9.0855e-01, -1.9334e+00, -2.7430e+00, ..., -2.3748e+00,
-1.3246e+00, -7.3610e-01],
[-1.4044e+00, -9.4895e-01, -3.0347e-02, ..., -7.9290e-01,
-4.7410e-02, -1.9718e-01],
...,
[-2.2235e+00, -1.6891e+00, -5.5500e-01, ..., 3.4774e-01,
4.5689e-01, 2.4679e-01],
[-2.1123e+00, -1.9316e+00, -1.1797e+00, ..., -1.3378e+00,
-8.4572e-01, -2.3879e-01],
[-2.4102e-01, -5.4122e-01, -1.1245e+00, ..., 6.8639e-02,
-4.7848e-01, -2.1426e-01]]],
[[[ 1.0525e+00, 2.0138e+00, 8.0490e-01, ..., 1.6256e+00,
1.4463e+00, 6.0392e-01],
[-6.3993e-01, -2.6112e+00, -2.5525e-01, ..., -1.7029e+00,
6.1564e-01, -1.2341e+00],
[-4.2127e-01, -1.8849e+00, -1.0133e+00, ..., 2.0891e-01,
7.8226e-01, 4.8041e-01],
...,
[ 4.4105e-01, -1.2335e-01, 4.4283e-01, ..., 3.8611e-01,
1.0233e+00, 8.0950e-01],
[-5.0479e-03, -7.4525e-01, -1.3144e-01, ..., 9.7700e-01,
1.6202e+00, 1.0836e-01],
[ 8.3929e-01, -2.2075e-01, -1.4763e-01, ..., 3.1965e-01,
3.2014e-01, 7.6977e-01]],
[[-2.8321e+00, 5.6612e-01, -1.7511e+00, ..., -2.8970e+00,
-3.8466e+00, -8.3417e-01],
[-3.8795e+00, -1.7753e+00, -1.9248e+00, ..., -6.6590e+00,
-6.7332e+00, -1.6463e+00],
[-2.8866e+00, -2.7475e+00, -2.4008e+00, ..., -8.6208e+00,
-3.9526e+00, -1.2259e+00],
...,
[-1.0554e+00, -8.2675e-01, -1.2225e+00, ..., -3.1185e-01,
-2.6351e-02, 8.0121e-01],
[-2.7005e+00, -2.4953e-01, -5.9113e-01, ..., 7.3299e-02,
-1.2648e-01, 6.6344e-01],
[-1.7051e+00, -1.3454e+00, -9.7262e-01, ..., -5.6544e-01,
-7.5079e-01, -2.4266e-01]],
[[ 2.4843e+00, 3.5652e+00, 1.3993e+00, ..., -3.1570e-01,
-9.5056e-01, 2.3546e+00],
[ 7.8438e-01, 1.9513e+00, 1.7584e+00, ..., -2.1912e+00,
-1.3328e+00, 2.1232e+00],
[ 1.2367e+00, 2.4832e+00, 2.1070e+00, ..., -1.9337e+00,
1.1850e+00, 2.5531e+00],
...,
[ 1.0129e+00, 8.5454e-01, 3.2250e+00, ..., 6.3870e-01,
6.5183e-01, 1.9267e+00],
[ 1.1595e+00, 2.2659e+00, 1.9565e+00, ..., 9.8239e-01,
7.2977e-01, 1.7711e+00],
[ 2.4279e+00, 1.9044e+00, 2.3573e+00, ..., 1.9175e+00,
1.0265e+00, 1.8135e+00]],
...,
[[-2.6893e+00, -7.0200e+00, -6.9386e+00, ..., -1.9509e+00,
-6.3256e+00, -5.6878e+00],
[-3.3594e+00, -7.9605e+00, -8.0680e+00, ..., -7.0586e+00,
-9.1154e+00, -5.9586e+00],
[-2.1860e+00, -6.1039e+00, -4.9696e+00, ..., -5.6958e+00,
-5.3125e+00, -4.0673e+00],
...,
[-2.3945e+00, -5.0156e+00, -2.9147e+00, ..., -4.4181e+00,
-2.6919e+00, -3.0081e+00],
[-2.5288e+00, -5.5020e+00, -3.9127e+00, ..., -4.3784e+00,
-3.2326e+00, -4.9409e+00],
[-2.6970e+00, -6.0257e+00, -2.7950e+00, ..., -2.8204e+00,
-1.5746e+00, -3.3545e+00]],
[[ 2.3718e-01, 1.1997e+00, 2.1871e-01, ..., -4.0714e+00,
5.3405e-02, 1.8432e+00],
[ 1.3208e+00, 5.5344e-01, 2.5061e-01, ..., -4.8454e+00,
-2.1889e-01, 2.9055e+00],
[ 1.3454e+00, 1.9864e-01, 5.9248e-01, ..., -1.6318e+00,
9.2048e-01, 2.2480e+00],
...,
[ 1.5744e+00, 3.9377e-01, 1.0762e+00, ..., 2.1002e+00,
2.0889e+00, 1.5070e+00],
[ 2.0271e+00, 1.2532e+00, 1.9102e+00, ..., 1.9166e+00,
1.7110e+00, 2.6168e+00],
[ 1.1326e+00, 7.7845e-01, 1.2805e+00, ..., 1.2667e+00,
1.4959e+00, 1.5297e+00]],
[[-6.1750e-01, -5.7320e-01, -9.7556e-01, ..., 1.2019e+00,
-1.9708e-01, 1.1556e+00],
[-1.4238e+00, -2.1217e+00, -1.5058e+00, ..., 2.5176e-01,
-2.3709e+00, -6.0086e-01],
[-2.3657e+00, -1.7655e+00, -2.6878e-01, ..., 5.1696e-01,
-1.4290e+00, -5.5696e-01],
...,
[-2.0602e+00, -1.5674e+00, 1.0385e-01, ..., -1.1128e-01,
-4.2667e-01, -4.0034e-01],
[-1.6963e+00, -1.2704e+00, -6.3178e-01, ..., -2.6270e-01,
-3.3232e-01, 4.6950e-01],
[-2.6771e-01, 3.4782e-01, -2.4019e-01, ..., 1.9877e-01,
1.7473e-02, 6.4624e-01]]],
[[[ 1.0889e+00, 1.5816e+00, 1.2710e+00, ..., 2.5774e-01,
3.2109e-01, 1.9072e-01],
[-6.4128e-01, -1.8373e+00, -8.5754e-01, ..., 8.2445e-02,
2.0045e-01, -1.1125e+00],
[-9.1446e-01, -1.3743e+00, -1.1711e+00, ..., -1.6475e+00,
7.8316e-01, 6.6809e-01],
...,
[-1.5705e-01, -4.3271e-01, 1.7576e-02, ..., 8.6105e-02,
8.9087e-01, 4.5917e-01],
[-4.5177e-01, -8.2607e-01, -1.8733e-01, ..., 2.2103e-01,
1.4041e+00, -1.4051e-01],
[-2.2732e-01, -2.6870e-01, -9.1603e-01, ..., -2.6375e-01,
3.1630e-01, 5.7826e-01]],
[[-3.4690e+00, -1.8091e+00, -2.1424e+00, ..., -1.8604e+00,
-2.2147e+00, -1.0692e+00],
[-4.2583e+00, -2.9458e+00, -1.1112e+00, ..., -2.4788e-01,
-1.3436e+00, -2.6312e-01],
[-4.1768e+00, -4.1360e+00, -2.8540e+00, ..., -1.4428e+00,
-1.5031e+00, -7.2956e-01],
...,
[-1.0410e-01, -1.0026e+00, -5.9409e-01, ..., 1.0901e-01,
-9.7032e-02, 4.7647e-01],
[-9.8936e-01, -8.7262e-01, -5.4658e-01, ..., 3.0596e-02,
-4.3153e-01, 2.4940e-01],
[-2.9883e-01, -1.0916e+00, -9.9653e-01, ..., -3.3692e-01,
-7.1506e-01, -5.4684e-01]],
[[ 2.6256e+00, 2.6013e+00, 1.8481e+00, ..., 2.9800e+00,
1.4225e+00, 2.8653e+00],
[ 3.8925e-02, 2.0326e-01, 1.7789e+00, ..., 1.8939e+00,
-3.4022e-01, 2.5146e+00],
[-1.5167e+00, -9.8997e-01, 9.0671e-01, ..., 2.5171e+00,
6.0306e-01, 1.8511e+00],
...,
[-5.5951e-01, 2.4361e-03, 2.3949e-01, ..., 8.3607e-01,
1.0305e-01, 1.4492e+00],
[-8.6116e-01, 4.8687e-01, 4.6962e-01, ..., 1.0709e+00,
6.2457e-01, 1.5261e+00],
[ 7.0134e-01, 4.0503e-01, 9.5734e-01, ..., 1.4551e+00,
1.1018e+00, 2.0153e+00]],
...,
[[-3.4637e+00, -9.0438e+00, -7.0922e+00, ..., -3.6299e+00,
-2.4586e+00, -3.8787e+00],
[-4.7050e+00, -1.1035e+01, -6.7366e+00, ..., -6.6133e+00,
-4.1707e+00, -4.3615e+00],
[-2.2568e+00, -7.1270e+00, -3.0987e+00, ..., -3.5125e+00,
-2.7108e+00, -1.5836e+00],
...,
[-3.4570e+00, -8.9517e+00, -5.4167e+00, ..., -4.4355e+00,
-2.4151e+00, -1.4839e+00],
[-3.9252e+00, -9.1591e+00, -5.0966e+00, ..., -5.9366e+00,
-3.4124e+00, -3.7693e+00],
[-3.0251e+00, -7.4491e+00, -2.9811e+00, ..., -3.8164e+00,
-2.1467e+00, -2.8018e+00]],
[[-4.7303e-01, 9.2766e-01, 9.5854e-01, ..., 1.3277e+00,
1.0834e+00, 1.2462e+00],
[ 6.9252e-01, 1.5199e+00, 1.8339e+00, ..., 2.1358e+00,
2.2283e+00, 2.5465e+00],
[-1.1341e+00, -2.4623e+00, -5.7576e-01, ..., 2.9636e+00,
2.1522e+00, 1.8802e+00],
...,
[ 1.2218e+00, 4.0420e-01, 1.3018e+00, ..., 1.8332e+00,
1.9148e+00, 1.6729e+00],
[ 1.7709e+00, 1.4904e+00, 1.5559e+00, ..., 1.6328e+00,
1.5288e+00, 2.4404e+00],
[ 8.0503e-01, 6.4087e-01, 1.1633e+00, ..., 1.1768e+00,
1.3519e+00, 1.3785e+00]],
[[-1.4835e-01, -3.0797e-01, -4.0121e-01, ..., -1.5881e-01,
-1.7708e-01, 8.5064e-01],
[-4.7063e-01, -2.3101e+00, -1.4062e+00, ..., -2.9517e-01,
-2.2384e-01, 3.7473e-01],
[-2.3144e+00, -1.1982e+00, -2.2866e+00, ..., -6.1577e-01,
4.4867e-01, 2.3118e-01],
...,
[-1.7720e+00, -2.4329e+00, -5.9841e-01, ..., -2.4822e-01,
-4.1068e-01, -3.3590e-01],
[-1.4114e+00, -2.4620e+00, -6.2576e-01, ..., -3.6433e-01,
-4.5322e-01, 3.2000e-01],
[-3.1288e-01, -1.2547e+00, -5.9320e-01, ..., 8.5034e-02,
1.2610e-01, 3.2492e-01]]]], device='cuda:0'),
tensor([[[[ 9.6923e-01, 1.5956e+00, 1.0707e+00, ..., 4.9938e-01,
3.0643e-01, -1.7510e-01],
[-9.0440e-01, -2.0059e+00, -9.2214e-01, ..., 3.7108e-01,
6.8527e-01, -6.0179e-01],
[ 1.2015e-02, -6.6602e-01, -9.0859e-02, ..., -5.2377e-01,
2.4171e-01, 7.5044e-01],
...,
[ 4.4320e-01, -2.7598e-01, -3.9678e-01, ..., 3.3510e-01,
5.6506e-01, 2.3287e-01],
[-1.9815e-01, -1.1122e+00, -3.5918e-01, ..., -1.8498e-02,
9.7075e-01, 1.6602e-01],
[ 7.5964e-01, -2.9716e-01, -2.9862e-01, ..., -5.4112e-01,
-4.6619e-01, 2.7256e-01]],
[[-4.1577e+00, -1.2772e+00, -2.2457e+00, ..., -2.9420e+00,
-3.3819e+00, -7.5435e-01],
[-4.0030e+00, -2.8062e+00, -2.4467e+00, ..., -1.9038e+00,
-2.9277e+00, -5.6805e-01],
[-2.6666e+00, -2.0747e+00, -3.5765e+00, ..., -2.8995e+00,
-2.8967e+00, -6.0394e-01],
...,
[-7.8934e-01, -5.5839e-01, -1.6721e+00, ..., -2.7290e+00,
-1.9188e+00, 2.3644e-02],
[-2.8428e+00, -3.7610e-01, -1.0815e+00, ..., -1.7086e+00,
-1.6114e+00, -5.0098e-02],
[-2.0204e+00, -1.4799e+00, -1.8121e+00, ..., -9.4606e-01,
-1.0096e+00, 4.8825e-03]],
[[ 3.0213e+00, 4.1386e+00, 3.3338e+00, ..., 2.2905e+00,
6.1381e-01, 2.2671e+00],
[ 3.5306e-01, 8.4222e-01, 2.4000e+00, ..., 1.7607e+00,
-1.1848e-01, 2.1515e+00],
[ 8.6266e-01, 1.3237e+00, 1.8493e+00, ..., 7.7973e-01,
5.9688e-01, 2.0315e+00],
...,
[ 3.1034e-01, 9.3071e-01, 1.4668e+00, ..., -1.8584e-01,
-8.0318e-01, 1.1031e+00],
[ 2.9842e-01, 1.9920e+00, 1.6463e+00, ..., 1.7300e-01,
2.4447e-01, 1.1519e+00],
[ 1.3702e+00, 1.2790e+00, 1.7257e+00, ..., 1.0981e+00,
8.2858e-01, 1.3081e+00]],
...,
[[-2.1028e+00, -7.4177e+00, -5.9782e+00, ..., -4.1936e+00,
-2.7968e+00, -3.5497e+00],
[-2.7546e+00, -6.5729e+00, -4.7647e+00, ..., -5.5191e+00,
-4.4276e+00, -4.3725e+00],
[-2.7431e-01, -2.8802e+00, -3.5038e+00, ..., -4.6586e+00,
-2.8965e+00, -1.7912e+00],
...,
[-9.5147e-01, -2.2614e+00, -1.7474e+00, ..., -5.1401e+00,
-3.4896e+00, -2.0227e+00],
[-1.3130e+00, -3.6367e+00, -3.5247e+00, ..., -6.2091e+00,
-4.6163e+00, -4.6096e+00],
[-2.2404e+00, -5.0121e+00, -3.1935e+00, ..., -4.4298e+00,
-2.7937e+00, -3.2156e+00]],
[[-7.4683e-01, -1.2799e-01, -1.0353e-02, ..., 9.1922e-01,
1.0426e+00, 1.7406e+00],
[ 1.8128e+00, 1.0822e+00, 1.2976e+00, ..., 1.5468e+00,
1.5461e+00, 2.8071e+00],
[ 1.2220e+00, -1.1460e-02, 4.3626e-01, ..., 1.3188e+00,
9.6588e-01, 2.1330e+00],
...,
[ 1.3141e+00, -1.6317e-01, 5.9123e-01, ..., 1.1621e+00,
9.4023e-01, 1.4236e+00],
[ 1.8710e+00, 9.4452e-01, 1.6033e+00, ..., 1.6088e+00,
9.7137e-01, 2.2553e+00],
[ 9.9677e-01, 3.5242e-01, 8.1134e-01, ..., 1.0108e+00,
1.1946e+00, 1.5621e+00]],
[[-5.1230e-01, 5.8213e-01, 4.3112e-02, ..., -3.5949e-01,
-5.3780e-01, 9.4471e-01],
[-1.1182e+00, -1.1699e+00, -1.5698e+00, ..., -3.6369e-01,
2.9052e-03, 1.1045e+00],
[-2.0717e+00, -7.2996e-01, -2.6099e-01, ..., 3.2907e-01,
-9.0829e-02, 5.3780e-01],
...,
[-1.7779e+00, -1.4395e+00, -1.2670e+00, ..., -4.1421e-01,
-9.6657e-01, -7.0336e-01],
[-1.7964e+00, -1.1079e+00, -1.0559e+00, ..., -7.8091e-01,
-1.1612e+00, 2.4642e-01],
[-3.4443e-01, 3.8170e-01, -4.2383e-01, ..., 4.3519e-02,
-2.9514e-01, 5.9929e-01]]],
[[[ 5.9663e-01, 8.7112e-01, 5.7822e-01, ..., 3.4334e-01,
4.2753e-01, 3.3947e-02],
[-2.9804e-01, -8.9569e-01, -5.0123e-01, ..., 2.8245e-01,
1.7084e-01, -3.1079e-01],
[-2.4132e-01, -1.5822e-01, 3.2064e-01, ..., -3.0477e-01,
7.5255e-03, 6.5561e-01],
...,
[ 1.9660e-01, 7.5325e-01, 6.5592e-01, ..., 4.0493e-01,
8.3716e-01, 6.4587e-01],
[-1.1042e-01, -1.1873e-01, 1.2248e+00, ..., 8.9548e-01,
1.5358e+00, 8.9372e-02],
[ 8.1813e-01, -4.5771e-01, 2.8831e-02, ..., 3.6642e-01,
3.1268e-01, 7.2017e-01]],
[[-2.4542e+00, -6.6932e-01, -1.9236e+00, ..., -1.6233e+00,
-2.2879e+00, -1.5508e+00],
[-2.5895e+00, -7.3469e-01, -1.5475e+00, ..., -1.2020e+00,
-2.4443e+00, -1.7127e+00],
[-2.4982e+00, -1.5031e+00, -2.1913e+00, ..., -2.1449e+00,
-2.3850e+00, -1.1206e+00],
...,
[-6.0188e-01, 3.2740e-01, 1.8584e+00, ..., -6.6426e-01,
-2.6050e-01, 6.9147e-01],
[-2.6507e+00, -5.2905e-01, 5.7977e-02, ..., 1.1110e-03,
-8.6339e-02, 6.6300e-01],
[-1.7702e+00, -1.6518e+00, -1.0358e+00, ..., -4.6445e-01,
-5.6699e-01, -1.4738e-01]],
[[ 1.9484e+00, 2.5312e+00, 1.6600e+00, ..., 1.5509e+00,
8.0824e-01, 2.1189e+00],
[-2.2388e-01, 3.6079e-01, 6.4408e-01, ..., 7.5230e-01,
3.2418e-01, 2.2444e+00],
[ 6.6928e-02, 2.7526e-01, 1.8462e-01, ..., -1.1379e-01,
7.9819e-01, 2.0034e+00],
...,
[ 1.2956e+00, 3.8138e-01, -1.1976e+00, ..., 3.9062e-01,
6.9304e-02, 1.6236e+00],
[ 1.2503e+00, 2.8159e+00, 1.2896e+00, ..., 9.9862e-01,
7.0454e-01, 1.6598e+00],
[ 2.4232e+00, 2.0930e+00, 2.5674e+00, ..., 1.9451e+00,
1.0186e+00, 1.6978e+00]],
...,
[[-2.0883e+00, -5.0418e+00, -3.2625e+00, ..., -1.9282e+00,
-1.6046e+00, -3.0709e+00],
[-2.5268e+00, -5.2763e+00, -3.0276e+00, ..., -2.3469e+00,
-2.3907e+00, -3.0350e+00],
[-1.5545e+00, -5.1916e+00, -3.1766e+00, ..., -2.1144e+00,
-2.5121e+00, -2.3044e+00],
...,
[-2.3856e+00, -4.2598e+00, -1.5784e+00, ..., -2.7294e+00,
-2.0170e+00, -2.6497e+00],
[-2.5013e+00, -5.6354e+00, -3.7766e+00, ..., -4.3475e+00,
-3.1199e+00, -4.8613e+00],
[-2.6221e+00, -5.6770e+00, -2.4674e+00, ..., -2.9722e+00,
-1.6376e+00, -3.3225e+00]],
[[ 9.2195e-01, 1.5878e+00, 7.7706e-01, ..., 1.0588e+00,
1.5010e+00, 1.4086e+00],
[ 1.2009e+00, 9.7683e-01, 8.2727e-01, ..., 9.2083e-01,
7.7498e-01, 1.9456e+00],
[ 6.9125e-01, -4.0344e-02, 1.7528e-02, ..., 5.0471e-01,
8.9230e-02, 1.4317e+00],
...,
[ 1.5414e+00, 6.0596e-01, 2.2593e+00, ..., 1.5047e+00,
1.7623e+00, 1.4363e+00],
[ 1.9551e+00, 8.2877e-01, 1.6126e+00, ..., 1.9327e+00,
1.6797e+00, 2.5404e+00],
[ 1.2161e+00, 1.0481e+00, 1.3998e+00, ..., 1.3098e+00,
1.5205e+00, 1.5434e+00]],
[[-1.1846e-01, -5.9971e-01, -1.1720e+00, ..., -5.2050e-01,
-8.6638e-01, -9.3827e-02],
[-9.5429e-01, -2.5159e+00, -2.4985e+00, ..., -9.7243e-01,
-1.0624e+00, -6.9130e-01],
[-1.5658e+00, -1.8798e+00, -1.1543e+00, ..., -2.7749e-01,
-4.9781e-01, -2.4864e-01],
...,
[-1.4981e+00, -1.2355e+00, -3.8480e-01, ..., -3.9735e-01,
-4.5916e-01, -4.8095e-01],
[-1.5830e+00, -7.4076e-01, -4.4698e-01, ..., -3.7194e-01,
-4.0816e-01, 3.9199e-01],
[-1.6600e-01, 2.7514e-01, -5.5165e-01, ..., 1.8176e-01,
3.4284e-02, 6.4263e-01]]],
[[[ 5.9223e-01, 9.6602e-01, 5.5819e-01, ..., 8.6920e-02,
2.7185e-01, 3.4649e-01],
[-4.4139e-01, -1.4747e+00, -4.6219e-01, ..., -2.0879e+00,
-3.5160e-01, -1.5220e+00],
[-8.8334e-02, -6.1580e-01, 6.6645e-02, ..., -9.3496e-01,
-1.1185e+00, -3.2340e-02],
...,
[ 1.5055e-01, -6.6918e-01, -8.6883e-01, ..., -2.1780e-01,
3.0109e+00, -1.6404e-01],
[ 3.7586e-01, -4.2297e-01, -2.9084e+00, ..., -2.9961e+00,
1.1812e+00, -3.7856e-01],
[-1.8719e-01, -5.0191e-03, -1.1396e+00, ..., -2.6585e-01,
-7.1090e-01, 4.4803e-01]],
[[-4.0724e+00, -1.6925e+00, -3.5203e+00, ..., -1.3830e+00,
-1.9642e+00, -9.1432e-01],
[-5.0540e+00, -1.9278e+00, -2.0328e+00, ..., -1.2944e+00,
-1.7878e+00, -3.5304e-01],
[-4.1916e+00, -2.5002e+00, -1.7199e+00, ..., -3.1055e+00,
-2.6030e+00, -9.2481e-01],
...,
[-2.3534e+00, -3.6296e+00, -5.8549e+00, ..., -1.0132e+01,
-1.0882e+01, 6.2533e-01],
[-8.2708e-01, -6.1373e+00, -1.1382e+01, ..., -6.0616e+00,
-5.7509e+00, 1.0372e+00],
[-2.3029e+00, -4.1623e+00, -6.1491e+00, ..., -9.3747e-01,
-1.3563e+00, -6.5755e-01]],
[[ 2.4939e+00, 3.3592e+00, 2.6263e+00, ..., 2.4828e+00,
1.0669e+00, 2.9368e+00],
[ 7.4407e-01, 1.6394e+00, 2.3146e+00, ..., 3.3566e+00,
-3.5322e-01, 2.4607e+00],
[ 1.5835e+00, 1.5259e+00, 5.1342e-01, ..., 3.6598e+00,
1.7316e+00, 1.8760e+00],
...,
[-6.3978e-01, -4.0548e+00, -9.1063e+00, ..., -1.0556e+01,
-5.5328e-01, -5.5073e-01],
[-5.6272e-01, -5.0343e+00, -8.4301e+00, ..., -4.8762e+00,
8.0428e-01, 9.9435e-01],
[-6.1320e-01, -2.1448e+00, -2.2321e+00, ..., 2.0413e+00,
2.2777e-01, 1.2826e+00]],
...,
[[-2.0836e+00, -5.6205e+00, -4.8110e+00, ..., -3.4620e+00,
-2.4010e+00, -4.0097e+00],
[-2.7223e+00, -5.8969e+00, -4.9343e+00, ..., -6.0900e+00,
-4.1441e+00, -4.1301e+00],
[-9.0338e-01, -2.9914e+00, -2.4010e+00, ..., -1.1501e+01,
-4.9755e+00, -2.4700e+00],
...,
[ 8.3768e-01, 8.8616e-01, -3.3794e+00, ..., -8.2637e+00,
-2.4038e+01, -1.1591e+01],
[-1.1396e+00, -2.0642e+00, -7.5738e+00, ..., -8.0588e+00,
-1.0430e+01, -8.9581e+00],
[-1.0725e+00, -4.4417e+00, -4.7780e+00, ..., -5.9127e+00,
-5.7052e+00, -5.8549e+00]],
[[ 3.2598e-01, 3.8689e-01, -2.8692e-01, ..., 9.7814e-01,
1.1748e+00, 1.2466e+00],
[ 1.3415e+00, 1.1541e+00, 3.6633e-01, ..., 2.9538e+00,
2.4531e+00, 2.5530e+00],
[ 4.3645e-01, -1.6157e-01, 2.2546e-01, ..., -1.5190e+00,
2.9878e+00, 2.4110e+00],
...,
[-1.4776e+00, -2.5052e+00, -2.4618e+00, ..., 3.9950e-01,
-1.8476e+00, 3.5254e+00],
[-1.0351e+00, -2.3190e+00, -1.6399e+00, ..., 6.5839e-01,
6.8703e-01, 3.0989e+00],
[-5.3989e-01, -1.6469e+00, 4.2974e-01, ..., 1.6902e+00,
2.4226e+00, 1.4091e+00]],
[[-4.5959e-01, -2.4825e-01, -9.5580e-01, ..., -4.8076e-01,
-1.7658e-01, 9.1527e-01],
[-1.4535e+00, -1.0293e+00, -1.8371e+00, ..., -1.5254e+00,
-1.3694e-01, 1.5339e-01],
[-2.2067e+00, -8.9868e-01, -4.0996e-01, ..., -3.6321e+00,
-2.3817e-01, 3.7841e-01],
...,
[-2.4299e+00, -6.5475e-01, 2.0213e+00, ..., 6.3085e-01,
-3.9785e+00, -3.4871e+00],
[-1.0398e+00, -2.1328e+00, -3.2657e+00, ..., -3.2889e+00,
-3.8785e+00, -4.5045e-01],
[-8.5152e-01, -2.7232e+00, -2.7913e+00, ..., 7.4161e-02,
6.2968e-01, 4.3877e-01]]],
[[[ 5.9879e-01, 8.4553e-01, 4.9650e-01, ..., 1.4533e-01,
3.1114e-01, 2.7459e-02],
[-2.9386e-01, -6.4775e-01, -2.3007e-01, ..., 3.4824e-01,
-1.7437e-01, -4.3460e-01],
[-1.7954e-01, 3.4555e-01, 3.5299e-01, ..., 5.4125e-01,
-1.5055e-01, 7.2327e-01],
...,
[-5.0143e-02, 4.5804e-01, -2.0900e+00, ..., 3.7701e-01,
8.9199e-01, 6.5586e-01],
[-5.2967e-01, 2.2804e-01, -1.2243e+00, ..., 9.4376e-01,
1.5764e+00, 1.3088e-01],
[ 7.2423e-01, 5.0978e-01, 1.6234e-01, ..., 3.9109e-01,
2.8036e-01, 6.9062e-01]],
[[-2.1550e+00, -6.0844e-01, -2.1688e+00, ..., -1.8653e+00,
-2.3624e+00, -1.6074e+00],
[-2.3475e+00, -3.8531e-01, -9.4825e-01, ..., -2.2121e+00,
-2.8256e+00, -1.8387e+00],
[-2.2573e+00, -1.1266e+00, -2.7563e+00, ..., -3.6738e+00,
-2.5708e+00, -1.1730e+00],
...,
[-1.0631e+00, -1.6315e+00, 1.0951e+00, ..., -7.3930e-01,
-2.3584e-01, 7.1462e-01],
[-2.9439e+00, -5.3314e-01, 2.1960e+00, ..., 4.1269e-03,
6.9365e-03, 6.7825e-01],
[-2.4678e+00, -2.2095e+00, 5.6976e-01, ..., -5.0402e-01,
-6.5810e-01, -2.0366e-01]],
[[ 1.6076e+00, 2.2200e+00, 1.3986e+00, ..., 2.1516e+00,
9.0359e-01, 2.2065e+00],
[-3.3657e-01, 7.2225e-01, 1.0637e+00, ..., 1.4750e+00,
4.0465e-01, 2.4387e+00],
[ 2.1766e-01, 1.1903e+00, 7.1092e-01, ..., -1.0004e-01,
4.9460e-01, 2.0525e+00],
...,
[ 2.8940e+00, -1.1440e+00, -4.5612e+00, ..., 4.3244e-01,
1.2324e-01, 1.6153e+00],
[ 2.0260e+00, 1.2135e+00, -1.8833e+00, ..., 9.9579e-01,
7.8537e-01, 1.7259e+00],
[ 3.2672e+00, 1.7311e+00, -1.1573e+00, ..., 1.9329e+00,
1.0818e+00, 1.7620e+00]],
...,
[[-1.5337e+00, -4.5245e+00, -3.3235e+00, ..., -1.8765e+00,
-1.7724e+00, -3.3928e+00],
[-1.8900e+00, -4.9009e+00, -3.3012e+00, ..., -3.7556e+00,
-3.3515e+00, -3.2795e+00],
[-1.3707e+00, -4.9747e+00, -4.6898e+00, ..., -7.1406e+00,
-4.6106e+00, -2.3210e+00],
...,
[-3.0132e+00, -5.3080e+00, -4.5201e+00, ..., -2.7035e+00,
-2.0385e+00, -2.6784e+00],
[-2.8587e+00, -6.2785e+00, -3.2211e+00, ..., -4.4403e+00,
-3.1567e+00, -4.8962e+00],
[-2.7709e+00, -6.4997e+00, -2.2859e+00, ..., -2.9951e+00,
-1.6294e+00, -3.2708e+00]],
[[ 8.8364e-01, 1.3116e+00, 5.7528e-01, ..., 1.2485e+00,
1.8694e+00, 1.4381e+00],
[ 1.0233e+00, 8.5739e-01, 6.4615e-01, ..., 1.1468e+00,
1.5301e+00, 2.0831e+00],
[ 5.6230e-01, -1.2334e-01, 6.2669e-01, ..., 2.1933e-01,
1.1271e+00, 1.4409e+00],
...,
[ 7.8872e-01, 1.0076e+00, 6.7362e-01, ..., 1.5549e+00,
1.8455e+00, 1.5031e+00],
[ 1.5527e+00, 2.0711e+00, 2.6783e+00, ..., 1.9704e+00,
1.7405e+00, 2.5609e+00],
[ 6.7743e-01, 5.8975e-01, 2.2573e+00, ..., 1.2789e+00,
1.4779e+00, 1.5585e+00]],
[[-8.2211e-02, -5.7108e-01, -1.6520e+00, ..., -8.5994e-01,
-8.0622e-01, -6.2069e-02],
[-9.0090e-01, -2.4843e+00, -2.7674e+00, ..., -1.1241e+00,
-7.5740e-01, -7.3842e-01],
[-1.4084e+00, -1.5811e+00, -2.0943e+00, ..., -8.4585e-01,
-7.7924e-01, -1.9906e-01],
...,
[-1.0056e+00, -2.5647e+00, 4.2784e-01, ..., -4.3397e-01,
-4.3122e-01, -4.4695e-01],
[-1.5543e+00, -1.3234e+00, -3.8138e-01, ..., -3.4133e-01,
-3.7557e-01, 4.1193e-01],
[ 8.0037e-02, 1.2822e+00, 1.8258e-01, ..., 2.0312e-01,
-2.9493e-03, 5.8755e-01]]],
[[[ 5.6496e-01, 7.7726e-01, 4.7955e-01, ..., 3.4251e-01,
3.9211e-01, 8.0933e-03],
[-2.4254e-01, -8.4111e-01, -7.7682e-01, ..., 1.0592e-01,
6.2559e-02, -3.6783e-01],
[-6.0405e-02, -3.7009e-02, 2.9899e-01, ..., -1.2039e-01,
1.3847e-01, 7.4677e-01],
...,
[ 3.4087e-01, 3.1775e-01, 4.9660e-01, ..., -1.4734e-01,
1.1644e+00, 6.4340e-01],
[-1.0898e-02, -6.5736e-01, -1.7125e-01, ..., 1.0313e+00,
1.4834e+00, 1.1905e-01],
[ 8.3632e-01, -2.0953e-01, -2.1432e-01, ..., 3.2241e-01,
2.9600e-01, 7.6972e-01]],
[[-2.6444e+00, -7.7792e-01, -2.1986e+00, ..., -1.7417e+00,
-2.2774e+00, -1.4286e+00],
[-2.6014e+00, -1.2354e+00, -1.8011e+00, ..., -1.4451e+00,
-2.4662e+00, -1.5947e+00],
[-2.8719e+00, -2.6919e+00, -1.7164e+00, ..., -2.0969e+00,
-2.1285e+00, -9.8153e-01],
...,
[-6.6019e-01, -7.4572e-01, -1.6540e+00, ..., -2.4337e+00,
9.5867e-01, 7.5467e-01],
[-2.4959e+00, -3.7443e-01, -8.7467e-01, ..., 2.1354e-01,
1.8031e-01, 6.8205e-01],
[-1.6891e+00, -1.4806e+00, -9.6141e-01, ..., -5.6853e-01,
-6.6588e-01, -2.4546e-01]],
[[ 2.0632e+00, 2.6608e+00, 1.8270e+00, ..., 1.6920e+00,
8.3469e-01, 2.0791e+00],
[-8.7661e-02, 4.4803e-01, 5.6176e-01, ..., 7.7949e-01,
2.8676e-01, 2.1850e+00],
[ 3.5674e-01, 9.6831e-01, 5.3793e-01, ..., -2.3476e-01,
7.0613e-01, 1.8633e+00],
...,
[ 7.5662e-01, 1.0262e+00, 3.1518e+00, ..., 1.5628e+00,
-7.2715e-01, 1.6834e+00],
[ 1.0594e+00, 2.1987e+00, 1.9053e+00, ..., 9.3804e-01,
5.6429e-01, 1.7500e+00],
[ 2.4119e+00, 1.9144e+00, 2.4056e+00, ..., 1.8747e+00,
1.0684e+00, 1.7718e+00]],
...,
[[-2.1403e+00, -5.0919e+00, -3.3970e+00, ..., -2.1864e+00,
-1.7749e+00, -3.0019e+00],
[-2.6336e+00, -5.7376e+00, -3.3410e+00, ..., -2.6661e+00,
-2.4496e+00, -3.0177e+00],
[-1.6877e+00, -5.7538e+00, -3.3215e+00, ..., -2.6212e+00,
-2.5885e+00, -2.4583e+00],
...,
[-2.1993e+00, -4.8145e+00, -2.5888e+00, ..., -5.7708e+00,
-7.1450e+00, -2.6515e+00],
[-2.3359e+00, -5.3714e+00, -4.0822e+00, ..., -6.4562e+00,
-4.5151e+00, -4.9481e+00],
[-2.5345e+00, -5.7761e+00, -2.7493e+00, ..., -3.4222e+00,
-1.7038e+00, -3.4255e+00]],
[[ 8.1296e-01, 1.5576e+00, 7.8046e-01, ..., 1.0847e+00,
1.4631e+00, 1.4363e+00],
[ 1.1337e+00, 7.3337e-01, 5.7380e-01, ..., 8.9948e-01,
8.3167e-01, 2.0167e+00],
[ 4.1815e-01, -3.5970e-01, -5.0143e-01, ..., 5.0563e-01,
1.6838e-01, 1.3492e+00],
...,
[ 1.5557e+00, 1.3032e-01, -1.6548e-01, ..., 3.5628e+00,
2.9659e+00, 1.5814e+00],
[ 2.0007e+00, 1.1699e+00, 1.8288e+00, ..., 2.5418e+00,
1.7262e+00, 2.6261e+00],
[ 1.1296e+00, 7.3696e-01, 1.3174e+00, ..., 1.2845e+00,
1.4962e+00, 1.5163e+00]],
[[-1.2738e-01, -6.9443e-01, -1.2152e+00, ..., -5.8628e-01,
-8.5731e-01, -8.4461e-02],
[-8.5224e-01, -2.8667e+00, -2.1985e+00, ..., -1.1238e+00,
-9.8676e-01, -5.7441e-01],
[-2.0180e+00, -2.0102e+00, -2.1783e-02, ..., -2.1250e-01,
-3.4227e-01, -1.6783e-01],
...,
[-1.5117e+00, -1.5617e+00, 6.0268e-02, ..., -2.1869e+00,
-2.4682e-01, -4.2265e-01],
[-1.6581e+00, -1.1929e+00, -9.4927e-01, ..., -4.5806e-02,
-9.5942e-02, 4.5165e-01],
[-2.1631e-01, 4.2369e-01, -2.2519e-01, ..., 2.6646e-01,
4.3136e-02, 6.1593e-01]]]], device='cuda:0'),
tensor([[[[ 9.7312e-01, 1.5717e+00, 1.2166e+00, ..., 7.9157e-01,
1.0003e+00, 4.3671e-01],
[-8.8944e-01, -1.9921e+00, -6.8917e-01, ..., -4.4380e-01,
-1.6655e-01, -2.9612e-01],
[ 1.6780e-02, -8.3496e-01, -7.6334e-03, ..., -8.8006e-01,
-1.6050e-01, 6.2210e-01],
...,
[ 4.8700e-01, -2.5097e-01, -3.3062e-01, ..., 9.2187e-02,
-5.5308e-02, 3.7087e-01],
[-1.8917e-01, -9.7705e-01, -6.5773e-02, ..., -4.1398e-02,
9.1169e-02, -1.0017e-01],
[ 7.3058e-01, -3.1761e-01, -1.7527e-01, ..., -3.3811e-01,
-9.1710e-01, 1.5915e-01]],
[[-4.2250e+00, -1.3253e+00, -2.4339e+00, ..., -1.5820e+00,
-2.1082e+00, -1.8627e+00],
[-3.9609e+00, -2.8037e+00, -2.9927e+00, ..., -1.3328e+00,
-2.1640e+00, -2.1537e+00],
[-2.6178e+00, -1.8487e+00, -4.0799e+00, ..., -1.6604e+00,
-1.7046e+00, -1.4390e+00],
...,
[-7.8641e-01, -6.8627e-01, -1.4644e+00, ..., -4.4450e-01,
-1.0612e+00, -9.2320e-01],
[-2.8074e+00, -3.9211e-01, -7.7712e-01, ..., -5.8139e-01,
-1.1334e+00, -1.2714e+00],
[-2.0492e+00, -1.4917e+00, -1.4247e+00, ..., -6.5696e-01,
-7.3789e-01, -5.4090e-01]],
[[ 3.0216e+00, 4.1089e+00, 3.2670e+00, ..., 1.5098e+00,
8.2450e-01, 2.5858e+00],
[ 3.4360e-01, 8.0260e-01, 2.5050e+00, ..., 1.2321e+00,
7.0803e-01, 2.6566e+00],
[ 8.8984e-01, 1.2864e+00, 2.3586e+00, ..., 4.2917e-01,
1.0381e+00, 1.8235e+00],
...,
[ 1.5674e-01, 8.5157e-01, 1.1123e+00, ..., -4.8813e-01,
5.3993e-01, 1.5335e+00],
[ 1.9788e-01, 2.0743e+00, 1.4907e+00, ..., -1.7205e-01,
7.7978e-01, 1.1794e+00],
[ 1.3786e+00, 1.3543e+00, 1.6985e+00, ..., 5.6593e-01,
1.2970e+00, 1.3580e+00]],
...,
[[-2.1157e+00, -7.4002e+00, -5.9443e+00, ..., -5.0260e+00,
-3.6691e+00, -4.3467e+00],
[-2.8227e+00, -6.6093e+00, -4.8109e+00, ..., -4.9630e+00,
-3.7099e+00, -3.3236e+00],
[-2.8208e-01, -2.7782e+00, -3.2395e+00, ..., -4.0015e+00,
-2.3819e+00, -1.3650e+00],
...,
[-9.2669e-01, -2.2371e+00, -1.6954e+00, ..., -5.1354e+00,
-3.0868e+00, -2.7120e+00],
[-1.3165e+00, -3.6820e+00, -3.3962e+00, ..., -5.0645e+00,
-4.1304e+00, -4.2666e+00],
[-2.2095e+00, -4.8955e+00, -2.9783e+00, ..., -3.3168e+00,
-2.4056e+00, -2.3718e+00]],
[[-7.5901e-01, -1.3278e-01, -2.2014e-01, ..., 1.3762e+00,
1.6701e+00, 1.1797e+00],
[ 1.8225e+00, 1.1310e+00, 6.7131e-01, ..., 1.6266e+00,
1.2041e+00, 1.8373e+00],
[ 1.2068e+00, 2.4338e-01, -8.3511e-01, ..., 1.7151e+00,
6.0570e-01, 1.2866e+00],
...,
[ 1.2176e+00, -2.0451e-01, 9.7177e-01, ..., 1.6002e+00,
6.7612e-01, 9.2102e-01],
[ 1.8593e+00, 9.5894e-01, 1.7865e+00, ..., 1.2904e+00,
7.2620e-01, 1.4539e+00],
[ 9.9636e-01, 3.3736e-01, 9.1210e-01, ..., 7.3304e-01,
6.7878e-01, 1.4287e+00]],
[[-5.2874e-01, 5.7559e-01, 3.4639e-03, ..., -3.7856e-01,
-6.4869e-01, -9.9261e-02],
[-1.0850e+00, -1.1825e+00, -2.1548e+00, ..., -1.2397e+00,
-1.4417e+00, -1.1186e+00],
[-2.0576e+00, -6.0802e-01, -1.4577e+00, ..., -5.9729e-01,
-1.1266e+00, -2.3015e-01],
...,
[-1.6130e+00, -1.4792e+00, -1.1462e+00, ..., -6.1483e-01,
-1.5331e+00, -8.3572e-01],
[-1.6788e+00, -1.0993e+00, -8.9964e-01, ..., -7.7783e-01,
-1.6548e+00, -6.9977e-01],
[-3.1971e-01, 3.3549e-01, -3.3509e-01, ..., -2.6866e-01,
-8.3935e-01, -1.1782e-02]]],
[[[ 1.0963e+00, 1.5987e+00, 1.3085e+00, ..., 2.9068e-01,
3.7872e-01, 2.6120e-01],
[-1.2051e+00, -1.6821e+00, -1.0509e+00, ..., 3.9654e-01,
3.3237e-01, -1.1592e+00],
[-2.4504e-01, -7.9679e-01, -1.0981e+00, ..., -3.1393e-01,
9.8160e-01, 6.4138e-01],
...,
[-2.1757e-01, -4.4793e-01, -3.8589e-02, ..., 7.0957e-02,
9.5365e-01, 5.0349e-01],
[-4.1771e-01, -7.4301e-01, -1.8397e-01, ..., 1.8956e-01,
1.3950e+00, -1.0651e-01],
[-3.1728e-01, -3.0851e-01, -8.7765e-01, ..., -2.2873e-01,
3.5357e-01, 5.7165e-01]],
[[-3.4913e+00, -1.9775e+00, -2.2141e+00, ..., -1.6230e+00,
-2.2266e+00, -1.1069e+00],
[-3.2026e+00, -3.1165e+00, -1.6769e+00, ..., -3.2870e-01,
-1.4221e+00, -3.6692e-01],
[-2.6107e+00, -2.0942e+00, -1.6550e+00, ..., -1.1652e+00,
-1.4989e+00, -7.3737e-01],
...,
[ 1.0814e-01, -1.0271e+00, -5.3391e-01, ..., -1.2509e-01,
-6.8987e-02, 5.0056e-01],
[-9.3614e-01, -8.7549e-01, -4.0644e-01, ..., -7.5273e-02,
-4.6338e-01, 2.4360e-01],
[-4.3109e-01, -1.1861e+00, -9.7082e-01, ..., -3.1336e-01,
-7.3888e-01, -5.3583e-01]],
[[ 2.6264e+00, 2.6105e+00, 1.8694e+00, ..., 2.7689e+00,
1.3298e+00, 2.8860e+00],
[ 2.8322e-02, -3.4991e-02, 1.2751e+00, ..., 1.6724e+00,
-1.2543e-01, 2.4745e+00],
[-6.2823e-01, 3.4109e-01, 1.3518e+00, ..., 1.1572e+00,
6.2147e-01, 1.8643e+00],
...,
[-5.5622e-01, -2.7097e-01, 2.4985e-01, ..., 7.7574e-01,
1.2186e-01, 1.4499e+00],
[-8.8398e-01, 3.7837e-01, 4.2035e-01, ..., 1.1152e+00,
6.1337e-01, 1.5004e+00],
[ 6.5655e-01, 4.1796e-01, 9.5201e-01, ..., 1.4889e+00,
1.0570e+00, 1.9796e+00]],
...,
[[-3.5045e+00, -9.1307e+00, -7.2796e+00, ..., -3.2290e+00,
-2.2361e+00, -3.9370e+00],
[-4.6841e+00, -1.0685e+01, -6.5840e+00, ..., -5.5923e+00,
-3.6464e+00, -4.2969e+00],
[-2.6555e+00, -8.4057e+00, -5.1002e+00, ..., -3.0483e+00,
-2.2336e+00, -1.5917e+00],
...,
[-3.6311e+00, -8.6909e+00, -4.8042e+00, ..., -4.0684e+00,
-2.3914e+00, -1.5444e+00],
[-4.0527e+00, -9.5139e+00, -5.0178e+00, ..., -4.9319e+00,
-3.3677e+00, -3.6722e+00],
[-3.0195e+00, -7.5444e+00, -3.0379e+00, ..., -3.6750e+00,
-2.1700e+00, -2.7152e+00]],
[[-5.2260e-01, 8.3365e-01, 8.2804e-01, ..., 1.3444e+00,
1.0265e+00, 1.2483e+00],
[ 5.7891e-01, 1.0644e+00, 1.3228e+00, ..., 1.9120e+00,
2.2798e+00, 2.5234e+00],
[ 4.7418e-03, 2.3737e-01, 2.1098e+00, ..., 1.5160e+00,
1.5924e+00, 1.9131e+00],
...,
[ 1.1624e+00, 2.7302e-01, 1.2229e+00, ..., 1.6227e+00,
1.8284e+00, 1.6782e+00],
[ 1.7336e+00, 1.4157e+00, 1.5700e+00, ..., 1.6115e+00,
1.5062e+00, 2.4259e+00],
[ 7.9967e-01, 5.7899e-01, 1.1045e+00, ..., 1.1816e+00,
1.3526e+00, 1.3594e+00]],
[[-1.7016e-01, -3.4241e-01, -3.8613e-01, ..., -8.5403e-02,
-1.2018e-01, 8.8563e-01],
[-7.1583e-01, -3.0828e+00, -1.4345e+00, ..., 2.7363e-01,
-9.4867e-02, 3.4218e-01],
[-1.9077e+00, -2.2987e+00, -1.0294e+00, ..., 3.4098e-01,
1.1688e-01, 2.2465e-01],
...,
[-1.6225e+00, -2.4413e+00, -6.2898e-01, ..., -8.6767e-02,
-3.3184e-01, -3.0614e-01],
[-1.3527e+00, -2.4728e+00, -6.3992e-01, ..., -5.5037e-01,
-4.5877e-01, 3.2480e-01],
[-3.9268e-01, -1.3237e+00, -7.0342e-01, ..., 5.9172e-02,
1.2702e-01, 3.2967e-01]]],
[[[ 9.1629e-01, 1.5766e+00, 1.3594e+00, ..., 1.3629e+00,
8.5038e-01, 2.6412e-01],
[-8.8290e-01, -2.2028e+00, -8.8070e-01, ..., -9.0804e-01,
-7.7994e-01, -1.5610e+00],
[-1.2162e-01, -6.6731e-01, 6.7710e-03, ..., -4.3573e+00,
2.0449e-02, 1.4393e+00],
...,
[ 3.3411e-01, -2.0569e+00, -1.8863e+00, ..., -1.5443e+00,
8.1423e-01, -1.1467e+00],
[-1.4710e+00, -3.1581e+00, -3.1819e-01, ..., -9.0658e+00,
-1.8598e-02, 9.9810e-02],
[ 5.5391e-01, -3.1639e+00, -1.0956e+00, ..., -7.7250e+00,
-5.1519e+00, 1.7555e+00]],
[[-3.7495e+00, -5.5918e-01, -1.3859e+00, ..., -3.1612e+00,
-4.6351e+00, -1.4490e+00],
[-4.4235e+00, -3.2329e+00, -3.2498e+00, ..., -3.7534e+00,
-6.8237e+00, -3.3346e+00],
[-2.7891e+00, -2.0177e+00, -3.9801e+00, ..., -3.0922e+00,
-7.2686e+00, -4.0393e+00],
...,
[-4.2756e+00, 3.3133e-01, -8.4427e-01, ..., -6.4768e+00,
-5.1342e+00, -1.4711e+00],
[-6.9433e+00, 8.6402e-01, 2.5194e-01, ..., -1.7379e+01,
-1.0479e+01, -6.9424e+00],
[-2.8594e+00, -3.1902e+00, -2.6230e+00, ..., -1.9605e+01,
-9.2944e+00, -6.7364e+00]],
[[ 3.0302e+00, 4.6010e+00, 3.5157e+00, ..., -2.8872e-01,
2.7056e-01, 2.9346e+00],
[ 2.0774e-01, 1.2436e+00, 1.9723e+00, ..., -3.9049e+00,
-3.8973e+00, 2.6214e+00],
[ 7.4021e-01, 1.7570e+00, 2.7091e+00, ..., -5.1888e+00,
-2.8875e+00, 1.9766e+00],
...,
[ 8.7753e-01, -2.2359e+00, -2.1150e+00, ..., -1.0348e+01,
-3.3654e+00, 1.6237e+00],
[-1.3655e+00, -5.0406e+00, -1.8230e+00, ..., -1.5429e+01,
-9.7799e+00, -2.8712e+00],
[ 1.0696e+00, -6.0642e+00, -3.3620e+00, ..., -1.4604e+01,
-9.0119e+00, -7.6914e-01]],
...,
[[-1.8968e+00, -7.3194e+00, -5.6414e+00, ..., -3.7690e+00,
-1.6235e+00, -5.3527e+00],
[-2.4527e+00, -6.7459e+00, -4.0676e+00, ..., -3.7428e+00,
-1.8202e+00, -8.2134e+00],
[-9.1041e-01, -4.4534e+00, -4.3571e+00, ..., 1.1905e+00,
-2.8906e+00, -9.6955e+00],
...,
[-2.9551e+00, -6.0170e+00, -3.1080e+00, ..., -1.5270e+00,
-6.7416e+00, -6.4567e+00],
[-4.3481e+00, -7.2877e+00, -6.7379e+00, ..., -2.3839e+00,
-6.2403e+00, -1.3651e+01],
[-4.3369e+00, -7.3633e+00, -7.1507e+00, ..., 6.5821e-01,
-6.2530e-02, -1.3770e+01]],
[[-4.9737e-02, 3.2737e-01, 5.9805e-01, ..., -2.1975e+00,
-3.5545e-01, 1.3574e+00],
[ 1.5415e+00, 1.0945e+00, 1.0509e+00, ..., -1.7313e+00,
-1.2888e+00, 2.7432e+00],
[ 1.0549e+00, 2.2075e-01, -8.2644e-01, ..., -3.0340e+00,
-5.2221e+00, 1.3353e+00],
...,
[-6.7695e-01, -2.0909e+00, -3.4725e-01, ..., -9.0590e+00,
-5.6637e+00, 5.2418e-01],
[ 7.8070e-01, -2.6268e-01, -1.4998e+00, ..., -7.4263e+00,
-9.0431e+00, -2.3167e+00],
[ 4.2202e-02, 1.1160e+00, -8.0054e-01, ..., -8.5970e+00,
-4.3998e+00, -2.7815e+00]],
[[-5.0887e-01, 3.7515e-01, 2.5931e-01, ..., 1.4621e+00,
3.2701e-01, 5.3700e-01],
[-1.3248e+00, -1.9294e+00, -2.8360e+00, ..., 2.9534e+00,
1.5169e+00, -6.3332e-01],
[-2.0630e+00, -7.3376e-01, -9.5508e-01, ..., 2.0880e+00,
9.3750e-01, -5.8042e-01],
...,
[-1.7224e+00, 1.1749e+00, -2.9514e-01, ..., 8.7869e-01,
3.8520e-01, -4.1209e+00],
[-4.5114e+00, 2.2835e-01, -6.2138e-01, ..., -2.4121e+00,
1.7620e+00, -2.0855e+00],
[-9.7654e-03, 5.2791e-01, -1.8995e+00, ..., -5.7079e+00,
1.1458e+00, 9.0321e-02]]],
[[[ 1.0903e+00, 1.4160e+00, 1.1978e+00, ..., 2.4526e-01,
3.4405e-01, 1.8852e-01],
[-9.5417e-01, -1.5173e+00, -8.7292e-01, ..., -1.3690e-02,
-8.0736e-02, -1.3571e+00],
[-2.9647e-01, -4.2887e-01, -3.2973e-01, ..., 6.1586e-01,
7.4232e-01, -4.7882e-01],
...,
[ 4.1350e-01, 5.0787e-01, 8.6474e-01, ..., 2.0231e-01,
1.5532e+00, 6.7873e-01],
[-4.2716e-01, -8.3898e-01, -1.7872e-01, ..., 1.6707e-01,
1.0555e+00, -2.8323e-01],
[-1.2281e-01, -1.6933e-01, -7.3760e-01, ..., -2.8137e-01,
3.0236e-01, 7.5953e-01]],
[[-3.0409e+00, -1.9569e+00, -2.1535e+00, ..., -1.9602e+00,
-2.5370e+00, -1.2992e+00],
[-2.7404e+00, -3.3489e+00, -1.5769e+00, ..., -1.1735e+00,
-1.7077e+00, -5.2047e-01],
[-1.9192e+00, -2.3255e+00, -1.8570e+00, ..., -3.5314e+00,
-4.0762e+00, -1.9388e+00],
...,
[-2.9541e-01, -1.4017e+00, -9.7816e-01, ..., 2.1920e-01,
-1.4574e-01, 2.2101e-01],
[-1.2800e+00, -8.1382e-01, -4.9953e-01, ..., -7.0690e-01,
-1.2027e+00, 4.4878e-02],
[-3.7926e-01, -9.6924e-01, -8.0165e-01, ..., -4.7506e-01,
-7.0144e-01, -5.4543e-01]],
[[ 2.2651e+00, 2.3152e+00, 1.7762e+00, ..., 3.1805e+00,
1.3794e+00, 2.9968e+00],
[ 1.7238e-01, 2.2599e-01, 1.5225e+00, ..., 3.0186e+00,
7.3289e-01, 3.2259e+00],
[ 2.7979e-01, 6.3708e-01, 1.6052e+00, ..., 1.0428e+00,
1.3686e+00, 2.6551e+00],
...,
[-5.6413e-02, 5.7919e-01, 9.3249e-01, ..., -2.2613e+00,
-1.9727e+00, 2.3156e-01],
[-9.9637e-01, 6.1405e-01, 6.4200e-01, ..., 8.3140e-01,
3.9722e-01, 1.7197e+00],
[ 6.2418e-01, 5.2608e-01, 1.0750e+00, ..., 1.5165e+00,
1.2470e+00, 2.2443e+00]],
...,
[[-2.7367e+00, -8.1558e+00, -6.5603e+00, ..., -3.9854e+00,
-2.7477e+00, -4.1501e+00],
[-3.6257e+00, -9.9135e+00, -6.1577e+00, ..., -7.1186e+00,
-5.1196e+00, -5.2792e+00],
[-2.7285e+00, -7.9026e+00, -4.8567e+00, ..., -5.4740e+00,
-4.7932e+00, -3.0643e+00],
...,
[-3.4714e+00, -8.2206e+00, -5.1208e+00, ..., -4.2553e+00,
-2.9204e+00, -1.9060e+00],
[-4.1327e+00, -9.2638e+00, -4.9908e+00, ..., -4.9915e+00,
-3.5511e+00, -3.4413e+00],
[-3.3226e+00, -7.7682e+00, -2.9990e+00, ..., -3.8138e+00,
-2.5785e+00, -3.0124e+00]],
[[-3.2950e-01, 7.0623e-01, 8.6096e-01, ..., 1.2171e+00,
9.6835e-01, 1.2239e+00],
[ 8.6975e-01, 7.6689e-01, 1.3461e+00, ..., 2.3552e+00,
2.7366e+00, 2.8170e+00],
[ 6.0068e-01, 7.4493e-01, 1.2889e+00, ..., -9.0901e-01,
-1.8367e-01, 1.9406e+00],
...,
[ 1.4403e+00, 7.5437e-01, 1.5448e+00, ..., 8.9510e-01,
1.2146e+00, 1.4107e+00],
[ 1.7075e+00, 1.6011e+00, 1.6519e+00, ..., 1.1707e+00,
1.0572e+00, 2.4452e+00],
[ 7.3892e-01, 6.6138e-01, 1.2760e+00, ..., 1.2851e+00,
1.4255e+00, 1.2965e+00]],
[[-2.5451e-01, -3.5162e-01, -3.2919e-01, ..., -2.9450e-01,
-2.9016e-01, 8.5993e-01],
[-6.7176e-01, -2.6236e+00, -1.0765e+00, ..., 2.1599e-01,
-1.8957e-01, 1.1088e-01],
[-1.7103e+00, -1.8418e+00, -4.8988e-01, ..., 6.0755e-01,
-4.4916e-01, -1.8789e-01],
...,
[-1.9608e+00, -2.6623e+00, -9.1234e-01, ..., -1.5819e+00,
-1.4339e+00, -7.1789e-01],
[-1.4097e+00, -2.5528e+00, -7.4743e-01, ..., -9.3019e-01,
-7.1991e-01, 3.1392e-01],
[-3.6799e-01, -1.1442e+00, -5.5443e-01, ..., -4.5424e-03,
1.9252e-01, 5.5966e-01]]],
[[[ 9.3845e-01, 2.1144e+00, 7.2924e-01, ..., 4.8908e-01,
2.2548e-01, -2.2484e-01],
[-8.0063e-01, -1.9398e+00, -1.5396e+00, ..., 3.6444e-01,
7.6497e-01, -6.5743e-01],
[ 1.3206e-01, -1.9510e+00, -4.8684e+00, ..., -3.3028e-01,
3.3720e-01, 6.8035e-01],
...,
[ 1.4450e+00, -8.5487e-01, -2.5283e-01, ..., -9.4359e-01,
-5.6817e+00, 2.2334e+00],
[-1.9128e-01, -3.4999e+00, 1.1105e+00, ..., 1.8692e+00,
-3.6675e-01, -4.6768e+00],
[ 1.6383e+00, -1.5337e+00, -4.5319e+00, ..., -8.0932e-01,
9.5408e-01, 9.0072e-01]],
[[-4.1981e+00, -2.9323e+00, -5.0568e+00, ..., -2.8690e+00,
-3.2889e+00, -6.9330e-01],
[-4.2328e+00, -5.0329e+00, -7.1550e+00, ..., -2.2362e+00,
-2.8596e+00, -3.5472e-01],
[-2.7596e+00, -4.9783e+00, -6.2560e+00, ..., -2.9354e+00,
-2.9490e+00, -6.1577e-01],
...,
[-2.6123e+00, 5.4153e-01, -1.4828e-01, ..., -6.8811e+00,
-3.1785e+00, 1.0846e+00],
[-2.8535e+00, 2.4413e+00, 3.1160e+00, ..., -7.5890e+00,
-8.6792e+00, -3.7341e+00],
[-2.3683e+00, -3.8481e+00, -1.7349e+00, ..., -2.6982e+00,
-3.1255e+00, -1.6022e+00]],
[[ 3.1090e+00, 4.1595e+00, 2.7871e+00, ..., 2.2113e+00,
5.8395e-01, 2.1704e+00],
[ 4.1399e-01, 1.4162e+00, 2.5866e+00, ..., 2.0963e+00,
-4.7052e-02, 2.1883e+00],
[ 9.0373e-01, 1.7948e+00, -8.8455e-01, ..., 9.1376e-01,
6.4263e-01, 1.9527e+00],
...,
[-3.4444e+00, -5.5520e+00, -8.5027e-01, ..., -8.4152e+00,
-1.0323e+01, -2.8002e+00],
[-3.8197e+00, -4.1795e+00, -1.0651e+00, ..., 4.8665e-01,
-6.2445e+00, -2.8723e+00],
[ 1.0177e+00, -5.5619e+00, -7.4029e+00, ..., 2.5079e+00,
1.0871e+00, 1.2655e+00]],
...,
[[-2.0239e+00, -7.1269e+00, -6.8089e+00, ..., -4.0548e+00,
-3.0547e+00, -3.7228e+00],
[-2.7572e+00, -6.6938e+00, -5.2063e+00, ..., -5.3092e+00,
-4.7356e+00, -4.5782e+00],
[-2.0763e-01, -2.5257e+00, -5.1408e+00, ..., -3.3911e+00,
-3.0566e+00, -1.7939e+00],
...,
[-2.0791e+00, -6.5126e+00, -6.6186e+00, ..., -7.6946e+00,
-8.3847e+00, -1.3958e+01],
[-1.9624e+00, -6.0984e+00, -7.4499e+00, ..., -9.9724e+00,
-5.1993e+00, -7.1609e+00],
[-2.3916e+00, -5.3753e+00, -6.1508e+00, ..., -4.3487e+00,
-4.6612e+00, -4.8117e+00]],
[[-6.5824e-01, -1.6414e+00, -1.2339e+00, ..., 9.6529e-01,
1.0950e+00, 1.7312e+00],
[ 1.7994e+00, -1.0739e+00, -5.4098e+00, ..., 1.4173e+00,
1.6130e+00, 2.7915e+00],
[ 1.1692e+00, -1.7758e+00, 1.3116e-01, ..., 1.0170e+00,
9.4647e-01, 2.0788e+00],
...,
[ 1.3259e+00, -2.4638e+00, -3.6359e+00, ..., 5.2191e+00,
8.7555e-01, -1.4164e+00],
[ 3.6531e+00, -2.6081e-01, -2.2264e+00, ..., -6.2835e-01,
1.1081e+00, 1.1266e+00],
[ 1.6122e-01, 2.2320e+00, 1.6727e+00, ..., 4.4938e-01,
1.1623e+00, 2.0157e+00]],
[[-5.1382e-01, -2.3978e-01, 4.6556e-01, ..., -2.7015e-01,
-5.7963e-01, 8.5457e-01],
[-1.1487e+00, -2.0295e+00, -1.3338e+00, ..., -2.4301e-01,
-6.4038e-02, 1.0443e+00],
[-2.0438e+00, -2.7701e+00, -1.2260e+00, ..., 3.7390e-01,
4.6181e-02, 3.9973e-01],
...,
[-2.2585e+00, 1.2002e+00, -2.7695e+00, ..., -1.8040e+00,
-3.6603e+00, -2.3182e+00],
[-3.8629e+00, -1.9254e+00, -1.2590e+00, ..., -1.5906e+00,
-2.1954e+00, -2.1831e+00],
[ 5.6734e-01, -3.2079e-01, -1.4626e+00, ..., -6.0481e-01,
-1.2734e+00, 6.8927e-01]]]], device='cuda:0'),
tensor([[[[ 5.8013e-01, 8.4122e-01, 5.2726e-01, ..., 3.1789e-01,
3.9520e-01, 2.3563e-02],
[-2.1939e-01, -8.1339e-01, -3.0977e-01, ..., 1.8750e-01,
7.3795e-02, -3.6248e-01],
[-7.3296e-02, 1.8691e-01, 3.7200e-01, ..., -2.3263e-01,
1.4774e-02, 6.6073e-01],
...,
[ 4.8286e-01, -3.5313e-01, -6.4789e-01, ..., 4.9191e-01,
8.3177e-01, 6.9428e-01],
[-6.7567e-02, -6.8910e-01, -3.5807e-02, ..., 1.0958e+00,
1.5453e+00, 1.0028e-01],
[ 7.9152e-01, -2.5142e-01, -1.4681e-01, ..., 3.4474e-01,
3.1994e-01, 7.7718e-01]],
[[-2.3814e+00, -6.4936e-01, -1.8808e+00, ..., -1.6716e+00,
-2.2542e+00, -1.5473e+00],
[-2.3918e+00, -5.9459e-01, -1.6738e+00, ..., -1.2326e+00,
-2.4653e+00, -1.7603e+00],
[-2.3973e+00, -2.2343e+00, -2.4545e+00, ..., -2.0405e+00,
-2.4489e+00, -1.2058e+00],
...,
[-5.8747e-01, -1.0655e+00, -2.1614e+00, ..., -1.7871e-01,
-2.9228e-01, 7.0036e-01],
[-2.6402e+00, -3.3969e-01, -9.4110e-01, ..., 5.0314e-01,
-1.5178e-01, 6.2417e-01],
[-1.6056e+00, -1.2407e+00, -9.8028e-01, ..., -5.3205e-01,
-5.5885e-01, -1.8294e-01]],
[[ 1.9123e+00, 2.4852e+00, 1.6282e+00, ..., 1.5912e+00,
7.6645e-01, 2.1078e+00],
[-2.2041e-01, 3.5116e-01, 6.3641e-01, ..., 8.4144e-01,
3.2722e-01, 2.3059e+00],
[ 9.3944e-02, 2.9932e-01, 1.0443e+00, ..., -9.6866e-02,
7.6539e-01, 2.0538e+00],
...,
[ 7.3633e-01, 1.2964e-01, 2.3329e+00, ..., -4.3232e-01,
7.0079e-02, 1.6536e+00],
[ 1.1062e+00, 2.2872e+00, 1.8947e+00, ..., 8.5290e-01,
6.2432e-01, 1.7088e+00],
[ 2.4138e+00, 1.8366e+00, 2.3772e+00, ..., 1.4973e+00,
1.0068e+00, 1.7451e+00]],
...,
[[-1.9342e+00, -4.7917e+00, -3.1111e+00, ..., -2.0880e+00,
-1.6718e+00, -3.1460e+00],
[-2.4031e+00, -5.1868e+00, -2.9064e+00, ..., -2.4257e+00,
-2.2647e+00, -3.0307e+00],
[-1.5710e+00, -5.5662e+00, -3.7291e+00, ..., -2.2383e+00,
-2.4069e+00, -2.2377e+00],
...,
[-2.5870e+00, -4.6671e+00, -3.1886e+00, ..., -5.4743e+00,
-2.0489e+00, -2.7041e+00],
[-2.5522e+00, -5.5582e+00, -4.2261e+00, ..., -5.4780e+00,
-3.0741e+00, -4.8712e+00],
[-2.6798e+00, -6.0639e+00, -2.9187e+00, ..., -4.4061e+00,
-1.5461e+00, -3.3490e+00]],
[[ 9.0728e-01, 1.5979e+00, 8.1913e-01, ..., 1.0897e+00,
1.5239e+00, 1.4042e+00],
[ 1.1758e+00, 1.0312e+00, 6.6047e-01, ..., 9.6352e-01,
7.6406e-01, 1.9245e+00],
[ 6.0239e-01, -2.4156e-01, 2.0527e-01, ..., 2.7666e-01,
1.1896e-02, 1.3808e+00],
...,
[ 1.5263e+00, -1.3671e-01, -3.6622e-01, ..., 1.8769e+00,
1.8133e+00, 1.5070e+00],
[ 1.9796e+00, 1.0905e+00, 1.8302e+00, ..., 1.9600e+00,
1.7004e+00, 2.5471e+00],
[ 1.1709e+00, 8.7227e-01, 1.3354e+00, ..., 1.2711e+00,
1.5258e+00, 1.5140e+00]],
[[-1.1801e-01, -6.3604e-01, -1.1944e+00, ..., -5.7026e-01,
-8.7495e-01, -1.1740e-01],
[-8.5281e-01, -2.5444e+00, -2.7542e+00, ..., -1.1370e+00,
-1.0440e+00, -7.0705e-01],
[-1.5152e+00, -2.1134e+00, -5.6629e-01, ..., -2.6274e-01,
-3.9397e-01, -1.3429e-01],
...,
[-1.6688e+00, -1.9954e+00, -9.4568e-01, ..., -7.9169e-01,
-3.8806e-01, -4.0316e-01],
[-1.7178e+00, -1.4030e+00, -1.0215e+00, ..., -5.0006e-01,
-3.8419e-01, 4.4564e-01],
[-2.8774e-01, 4.0913e-01, -2.2167e-01, ..., 4.1806e-01,
5.3402e-02, 6.5636e-01]]],
[[[ 1.6162e+00, 6.5790e-01, -3.4634e+00, ..., -1.9865e+00,
2.2808e+00, 1.7317e+00],
[-8.0210e-01, -6.2174e+00, -3.5226e-01, ..., -3.6917e+00,
-2.5119e+00, -9.4328e-01],
[-4.6260e-02, -1.1588e+00, -3.9124e+00, ..., 1.8400e-01,
-1.0847e+00, 9.0091e-01],
...,
[ 1.2038e-02, 1.0573e+00, -3.1384e+00, ..., 1.1883e+00,
1.0199e+00, 7.5680e-01],
[-2.7395e-01, -6.3325e-03, -2.0614e+00, ..., 1.0018e+00,
1.6844e+00, 1.0489e-01],
[ 1.4029e+00, 1.1196e+00, -1.6941e+00, ..., 4.1103e-01,
5.3857e-01, 8.6010e-01]],
[[-3.1168e+00, -7.6030e-01, -3.4845e+00, ..., -1.9777e+00,
-1.9983e+00, -4.4392e+00],
[-5.5961e+00, -7.6912e-01, 4.2594e-01, ..., -1.1623e+01,
-1.0238e+01, -9.5275e+00],
[-5.6432e+00, -7.3223e+00, -8.6493e+00, ..., -1.2443e+01,
-8.7222e+00, -4.5736e+00],
...,
[-2.2179e+00, 4.9308e-01, 1.5050e+00, ..., 6.7604e-02,
-1.1851e-02, 7.9806e-01],
[-3.0108e+00, -5.1784e-01, 2.3337e-01, ..., 5.8500e-01,
-1.2045e-01, 6.9867e-01],
[-2.2240e+00, -3.1519e+00, -3.1964e+00, ..., -4.6681e-01,
-9.3888e-01, -4.4047e-01]],
[[-1.2484e+00, -2.0692e+00, -2.6479e+00, ..., -5.8954e+00,
-3.4105e+00, 1.4455e-01],
[-3.6017e+00, -7.3313e+00, -3.0568e+00, ..., -1.0875e+01,
-7.4972e+00, -1.1675e+00],
[ 1.2491e+00, -5.5238e+00, -9.8503e+00, ..., -5.1933e+00,
-1.6627e+00, 2.3644e+00],
...,
[ 1.0533e-01, -8.6044e+00, -2.9855e+00, ..., 9.5313e-01,
6.2045e-01, 1.9068e+00],
[ 6.7203e-01, -3.6608e+00, -1.1017e+00, ..., 4.8064e-01,
8.1704e-01, 1.8771e+00],
[ 2.6366e+00, 1.0261e+00, -3.1594e-01, ..., 1.9409e+00,
9.8186e-01, 1.8974e+00]],
...,
[[-2.8058e+00, -6.3005e+00, -9.8694e+00, ..., 1.6874e+00,
-3.5723e+00, -2.0136e+01],
[-4.6205e+00, -5.7539e+00, -7.9186e+00, ..., -1.4172e+00,
-4.2313e+00, -1.8973e+01],
[-2.6609e+00, -5.1084e+00, -2.7828e+00, ..., -2.9008e+00,
-2.5284e+00, -6.0391e+00],
...,
[-2.6617e+00, -5.2478e+00, -5.7920e+00, ..., -4.3741e+00,
-2.8718e+00, -2.7864e+00],
[-2.3895e+00, -5.9966e+00, -6.2515e+00, ..., -6.9380e+00,
-3.3832e+00, -4.9694e+00],
[-2.4773e+00, -6.4260e+00, -3.6562e+00, ..., -3.1309e+00,
-1.8425e+00, -3.6354e+00]],
[[ 2.0349e+00, -1.3271e+00, -5.6759e+00, ..., -3.3625e+00,
-3.6207e+00, -1.8848e+00],
[ 2.5692e+00, 1.9254e+00, -1.3150e+00, ..., -5.4908e+00,
-3.3726e+00, -1.6115e+00],
[-1.1122e+00, 2.0643e+00, 1.5889e+00, ..., -4.1146e+00,
-8.5309e-01, 1.6745e+00],
...,
[ 7.0809e-01, 1.0042e+00, -2.0966e+00, ..., 3.0427e+00,
2.1759e+00, 1.5511e+00],
[ 1.6051e+00, 2.3589e+00, 2.3637e+00, ..., 2.3113e+00,
1.6910e+00, 2.7422e+00],
[-5.4105e-02, -4.3678e-01, 6.9885e-01, ..., 1.3344e+00,
1.4641e+00, 1.4677e+00]],
[[-2.3888e+00, 4.9760e-01, -2.9361e+00, ..., 2.5305e-01,
2.7504e+00, 1.4153e+00],
[-4.7773e+00, -4.3582e+00, -4.0006e+00, ..., -2.4985e+00,
-1.8326e+00, -7.0010e+00],
[-2.7223e+00, -2.2073e+00, -2.8282e+00, ..., -1.7288e+00,
-1.4483e+00, -1.5302e+00],
...,
[-2.3372e+00, 1.2030e+00, -2.7703e-01, ..., -3.2577e-01,
-4.3718e-01, -4.4253e-01],
[-1.8627e+00, -3.1142e-01, -2.2198e+00, ..., 3.7621e-01,
-1.1031e-01, 5.7464e-01],
[-3.8462e-02, 4.5301e-01, 4.6471e-01, ..., 3.1428e-01,
6.2284e-02, 6.5233e-01]]],
[[[ 4.7336e-01, 1.0752e+00, -5.6285e-02, ..., 5.8741e-01,
8.6471e-01, 3.8631e-01],
[-2.4940e+00, -2.7839e+00, -1.7568e+00, ..., -1.3447e-01,
-1.0922e-01, -4.6152e-01],
[ 1.7982e-01, -2.0921e+00, -3.6066e+00, ..., -2.4255e-01,
1.2450e-01, 7.1621e-01],
...,
[ 3.4253e-01, -2.1387e-01, 5.0766e-01, ..., 9.2672e-01,
4.0050e-01, 9.7159e-01],
[-2.3011e-01, -1.0983e+00, -1.3796e-01, ..., -2.3825e+00,
7.6099e-02, 3.3612e-01],
[ 7.3425e-01, -3.3121e-01, -1.9781e-01, ..., 6.2147e-01,
-6.9139e-01, 4.0335e-01]],
[[-5.8820e+00, -8.9591e-01, 2.2792e+00, ..., -1.5222e+00,
-2.0474e+00, -1.6958e+00],
[-4.7927e+00, 1.9701e+00, -1.0356e+00, ..., -1.4310e+00,
-2.3824e+00, -2.1498e+00],
[-3.9043e+00, 5.6356e-01, -1.9235e+00, ..., -8.7532e-01,
-1.5435e+00, -1.4756e+00],
...,
[-9.2646e-01, -8.9902e-01, -1.3502e+00, ..., -9.0296e-01,
-2.4781e+00, -1.3242e+00],
[-2.9519e+00, -5.2051e-01, -9.9989e-01, ..., -3.1826e+00,
-3.8646e+00, -1.7048e+00],
[-2.1408e+00, -1.5178e+00, -1.4343e+00, ..., -1.7175e+00,
-1.5082e+00, -1.1791e+00]],
[[ 3.7549e+00, 3.9778e-01, -1.3886e+00, ..., 1.9293e+00,
1.0402e+00, 2.6512e+00],
[-2.1131e-01, -5.1416e+00, -4.3995e+00, ..., 1.5382e+00,
8.5208e-01, 2.5608e+00],
[-1.4794e+00, -6.8303e+00, -1.6949e+00, ..., 8.3737e-02,
8.7986e-01, 1.7401e+00],
...,
[ 2.5659e-01, 1.0864e+00, 1.4876e+00, ..., -1.5597e+00,
-2.5487e+00, -1.6353e+00],
[ 3.3180e-01, 2.1889e+00, 1.8785e+00, ..., -5.0244e+00,
-3.4500e+00, -2.6298e+00],
[ 1.4272e+00, 1.3504e+00, 1.7313e+00, ..., 4.8001e-01,
9.2050e-01, 7.7331e-01]],
...,
[[-3.8784e+00, -9.9453e+00, -6.5774e+00, ..., -5.0825e+00,
-3.5041e+00, -4.0971e+00],
[-5.9198e+00, -1.0724e+01, -7.3505e+00, ..., -5.0534e+00,
-3.5353e+00, -2.9788e+00],
[-3.3814e+00, -5.6630e+00, -5.0322e+00, ..., -2.6811e+00,
-1.8262e+00, -1.2584e+00],
...,
[-1.0084e+00, -2.3719e+00, -1.7767e+00, ..., -7.3082e+00,
-8.3169e+00, -3.3512e+00],
[-1.3502e+00, -3.7961e+00, -3.4144e+00, ..., -5.2926e+00,
-7.7233e+00, -4.9503e+00],
[-2.2703e+00, -5.0867e+00, -3.0859e+00, ..., -3.3878e+00,
-3.3667e+00, -3.0733e+00]],
[[-2.7492e+00, -1.4097e+00, 2.4798e+00, ..., 1.6094e+00,
1.7792e+00, 1.2897e+00],
[ 1.5810e+00, 5.7846e-01, 2.4932e-01, ..., 1.1989e+00,
1.1214e+00, 1.8676e+00],
[ 4.7061e-01, -6.6712e-01, -2.0967e+00, ..., 1.0413e+00,
4.2088e-01, 1.1851e+00],
...,
[ 1.2365e+00, -4.2098e-01, 8.3079e-01, ..., -9.6958e-01,
-7.9191e-01, 7.6893e-01],
[ 1.8768e+00, 9.2840e-01, 1.8060e+00, ..., 2.3027e-01,
-1.0475e+00, 9.4255e-01],
[ 1.0333e+00, 4.0002e-01, 1.0219e+00, ..., 9.7591e-01,
7.5518e-01, 1.4067e+00]],
[[-1.4595e+00, 1.2912e+00, 1.0516e+00, ..., -3.9782e-01,
-5.8957e-01, -3.7303e-02],
[-2.1363e+00, 1.6199e+00, 1.9454e+00, ..., -1.0381e+00,
-1.3253e+00, -1.0682e+00],
[-2.5519e+00, 1.0427e+00, -1.7255e+00, ..., -2.2201e-01,
-7.5799e-01, -1.2508e-01],
...,
[-1.7062e+00, -1.5229e+00, -8.3134e-01, ..., -8.1936e-01,
-2.1778e+00, 5.5374e-01],
[-1.7485e+00, -1.1052e+00, -9.3532e-01, ..., -3.7081e+00,
-4.8389e+00, -1.2161e+00],
[-3.9084e-01, 3.9537e-01, -3.1476e-01, ..., -7.6657e-01,
-1.2167e+00, -1.5287e-01]]],
[[[ 8.1903e-01, 1.0517e+00, 6.0799e-01, ..., 1.0972e-01,
6.3922e-02, -1.0899e-01],
[-1.4495e+00, -2.2856e+00, -3.1914e+00, ..., -5.0413e-01,
5.9054e-01, -1.1167e+00],
[-3.2080e-01, -1.2342e+00, -4.1864e+00, ..., -8.8351e-01,
-1.7976e-02, 9.7698e-01],
...,
[ 4.4873e-01, -2.5942e-01, -4.8571e-01, ..., 6.1364e-02,
1.2997e+00, 1.1865e+00],
[-1.9525e-01, -1.0733e+00, -4.0971e-01, ..., -1.2654e-01,
7.4919e-01, 7.6709e-02],
[ 7.6884e-01, -2.7504e-01, -3.2883e-01, ..., -3.4254e-01,
-2.8448e-01, 2.8937e-01]],
[[-3.3599e+00, -4.2961e-01, -4.2535e-01, ..., -3.0709e+00,
-3.4518e+00, -8.9170e-01],
[-2.3718e+00, -6.2584e-01, -4.0450e+00, ..., -1.9173e+00,
-4.2923e+00, -1.5451e+00],
[ 5.3873e-02, -2.2122e+00, -3.3590e+00, ..., -1.8868e+00,
-3.6659e+00, -1.4510e+00],
...,
[-8.5334e-01, -5.6602e-01, -1.7521e+00, ..., -2.8356e+00,
-3.0890e+00, -3.5173e-01],
[-2.8880e+00, -3.6857e-01, -1.2143e+00, ..., -2.5970e+00,
-1.7691e+00, 3.5072e-02],
[-2.0324e+00, -1.4208e+00, -1.7689e+00, ..., -1.0616e+00,
-9.5131e-01, 2.3625e-02]],
[[ 2.5038e+00, 2.0131e+00, 1.1454e+00, ..., 2.3414e+00,
6.9135e-01, 2.5111e+00],
[-2.5879e+00, -3.3072e+00, -2.2050e+00, ..., 2.8122e+00,
-2.5457e-01, 2.5757e+00],
[-1.4320e+00, -2.5379e+00, -1.2804e+00, ..., -8.0201e-01,
-2.8884e+00, -5.4731e-01],
...,
[ 2.4682e-01, 9.3977e-01, 1.4083e+00, ..., -1.5857e+00,
-2.1163e-01, 1.4874e+00],
[ 3.4642e-01, 2.1190e+00, 1.7401e+00, ..., 1.1501e-01,
8.3050e-01, 1.1137e+00],
[ 1.4075e+00, 1.3273e+00, 1.7813e+00, ..., 1.1541e+00,
7.5313e-01, 1.2729e+00]],
...,
[[-9.4948e-01, -5.3127e+00, -5.3481e+00, ..., -3.8576e+00,
-2.7960e+00, -3.8354e+00],
[-2.8535e+00, -5.0864e+00, -4.7533e+00, ..., -4.4202e+00,
-3.2758e+00, -4.0371e+00],
[-5.5616e-01, -4.9741e+00, -1.1753e+00, ..., -3.4580e+00,
-2.7683e+00, -1.8934e+00],
...,
[-9.5456e-01, -2.2166e+00, -1.7224e+00, ..., -3.7036e+00,
-3.0816e+00, -3.6211e+00],
[-1.2825e+00, -3.5936e+00, -3.3134e+00, ..., -5.4174e+00,
-4.0894e+00, -4.9968e+00],
[-2.2272e+00, -5.0439e+00, -3.0790e+00, ..., -4.9218e+00,
-3.1919e+00, -3.3470e+00]],
[[-1.5157e+00, -5.7455e-01, 7.1690e-01, ..., 7.4508e-01,
9.1349e-01, 1.6889e+00],
[ 1.6769e+00, 2.7274e-01, -9.3008e-02, ..., 5.0389e-01,
-6.5336e-01, 1.8108e+00],
[ 8.7646e-01, 1.0209e+00, 3.6312e-01, ..., 6.0974e-01,
-2.3401e+00, -5.5646e-01],
...,
[ 1.3066e+00, -1.1582e-01, 6.1102e-01, ..., 1.5701e-02,
7.8238e-01, 2.0019e+00],
[ 1.8784e+00, 9.2822e-01, 1.5459e+00, ..., 9.0848e-01,
1.2492e+00, 2.4125e+00],
[ 1.0418e+00, 3.8415e-01, 8.6200e-01, ..., 1.0813e+00,
1.2583e+00, 1.5687e+00]],
[[-7.8554e-01, 9.2254e-01, 5.4174e-01, ..., -2.2908e-01,
-4.7027e-01, 8.7127e-01],
[-6.8332e-01, -9.7196e-02, -6.9358e-01, ..., -1.1280e+00,
-3.3712e-01, 8.7409e-01],
[ 2.9059e-01, -1.2842e+00, -1.2714e+00, ..., -2.8170e-01,
-4.7561e-01, 1.5563e+00],
...,
[-1.7403e+00, -1.4720e+00, -1.3721e+00, ..., -1.9180e+00,
-1.9973e+00, -8.8565e-01],
[-1.7286e+00, -1.0709e+00, -1.1300e+00, ..., -1.5286e+00,
-9.3715e-01, 4.4640e-01],
[-3.2188e-01, 3.8953e-01, -4.2968e-01, ..., 2.7150e-01,
4.1389e-02, 6.8110e-01]]],
[[[ 5.8497e-01, 8.0747e-01, 5.2460e-01, ..., 2.1667e+00,
7.3908e-01, -1.0662e+00],
[-2.5208e-01, -9.1431e-01, -5.0990e-01, ..., -1.4891e+00,
3.6029e+00, 8.9018e-01],
[-2.2333e-01, -1.2182e-01, 1.6675e-01, ..., -4.3026e+00,
8.8603e-01, 1.2540e+00],
...,
[-1.7253e+00, -3.3506e+00, -3.3078e+00, ..., 7.2948e-01,
8.1969e-01, 6.2822e-01],
[-2.0142e+00, -1.1882e+00, -7.2282e-01, ..., 1.0153e+00,
1.5683e+00, 1.2125e-01],
[-4.2817e-01, 3.4718e-01, 1.6541e+00, ..., 6.5036e-01,
2.3739e-01, 7.3454e-01]],
[[-2.4467e+00, -6.5083e-01, -1.9176e+00, ..., -3.1675e+00,
-4.0712e+00, -1.4868e+00],
[-2.4585e+00, -7.0044e-01, -1.5769e+00, ..., -3.8082e+00,
-4.1909e+00, -2.3133e+00],
[-2.4260e+00, -1.5059e+00, -1.6725e+00, ..., -4.2756e+00,
-5.3327e+00, -4.0936e+00],
...,
[-1.4486e+00, -6.2047e+00, -6.3689e+00, ..., -1.6891e+00,
-6.9341e-01, 6.6771e-01],
[ 2.0893e-02, 8.9250e-02, -8.8551e-01, ..., -3.0410e-01,
-6.7547e-02, 6.7432e-01],
[ 2.6462e-01, -6.0037e-01, -4.0865e+00, ..., -6.4498e-01,
-6.4806e-01, -2.0144e-01]],
[[ 1.9406e+00, 2.5371e+00, 1.6579e+00, ..., -7.3390e-01,
6.4764e-01, 3.0350e+00],
[-2.0877e-01, 4.0806e-01, 6.2281e-01, ..., -4.2617e+00,
-1.3921e+00, 1.0331e+00],
[ 9.2459e-02, 3.3862e-01, 2.5514e-01, ..., -7.7861e+00,
-2.0255e+00, -4.7168e-01],
...,
[-2.4417e+00, -1.8848e+00, -2.6542e+00, ..., -4.7051e-02,
3.4087e-01, 1.4954e+00],
[-1.8759e+00, -2.8900e-01, -2.7562e+00, ..., 7.3448e-01,
8.9012e-01, 1.6118e+00],
[-1.4477e+00, -1.4536e+00, -2.9047e+00, ..., 1.9537e+00,
1.0306e+00, 1.7448e+00]],
...,
[[-2.0369e+00, -4.9803e+00, -3.1488e+00, ..., -5.1762e+00,
-4.6014e+00, -3.2447e+00],
[-2.5376e+00, -5.4831e+00, -3.0522e+00, ..., -3.0990e+00,
-1.0369e+01, -9.4527e+00],
[-1.5896e+00, -5.4837e+00, -3.2646e+00, ..., 7.1008e-01,
-5.9498e+00, -1.2747e+01],
...,
[-2.3773e+00, -3.8927e+00, 4.4078e-01, ..., -3.4651e+00,
-2.1054e+00, -3.0642e+00],
[ 4.7751e-01, -2.2280e+00, -1.4547e+00, ..., -3.8332e+00,
-3.1547e+00, -4.9479e+00],
[-8.9270e-02, -3.5389e+00, -6.1594e+00, ..., -3.4750e+00,
-1.8252e+00, -3.2981e+00]],
[[ 8.7925e-01, 1.6001e+00, 8.2695e-01, ..., -6.2823e+00,
-2.7534e+00, 1.6772e+00],
[ 1.1504e+00, 9.1612e-01, 8.8403e-01, ..., -2.8384e+00,
-2.7799e+00, 2.2826e+00],
[ 6.1724e-01, -8.0718e-02, 3.2478e-01, ..., -1.4318e+00,
-2.1960e+00, 9.6504e-01],
...,
[-2.0508e+00, -2.8231e+00, -2.9396e+00, ..., 7.6818e-01,
1.8071e+00, 1.6482e+00],
[-2.0597e+00, -4.1476e+00, -2.0606e+00, ..., 1.9053e+00,
1.7470e+00, 2.6284e+00],
[ 9.1585e-01, 1.0276e+00, -7.3364e-01, ..., 1.3221e+00,
1.5227e+00, 1.5424e+00]],
[[-9.3381e-02, -6.7380e-01, -1.2066e+00, ..., 1.5467e-01,
-2.8064e+00, -1.3779e+00],
[-8.4100e-01, -2.5904e+00, -2.5266e+00, ..., 2.0775e-01,
1.4094e+00, -1.0755e+00],
[-1.5035e+00, -1.8813e+00, -8.6377e-01, ..., -1.7885e+00,
5.2926e-01, -1.8658e+00],
...,
[ 5.1142e-02, -5.8568e+00, -4.5632e+00, ..., -4.9215e-01,
-4.3930e-01, -5.0491e-01],
[-1.1103e+00, -3.4537e+00, 3.3043e-01, ..., -1.0816e+00,
-3.4067e-01, 4.3731e-01],
[ 9.7694e-01, 1.9846e+00, 1.1835e+00, ..., 3.2042e-01,
9.1449e-02, 6.4375e-01]]]], device='cuda:0'),
tensor([[[[ 1.0804e+00, 1.5370e+00, 1.5996e+00, ..., -4.8043e-01,
1.3537e+00, 8.5728e-01],
[-5.7137e-01, -2.3030e+00, 2.7894e-01, ..., -5.3109e-01,
-6.9154e-01, -1.3588e+00],
[-2.9003e-01, -1.3695e+00, -1.8097e+00, ..., -3.1729e-01,
-4.7792e-01, 8.8832e-01],
...,
[ 4.9674e-02, 1.1521e+00, -6.3578e-01, ..., 1.1685e-02,
9.0071e-01, 7.7618e-01],
[-2.5801e-01, -1.0576e-01, -8.3610e-03, ..., 9.0226e-01,
1.6023e+00, 1.1939e-01],
[ 7.0970e-01, 1.0276e-01, 6.2394e-01, ..., 3.2560e-01,
3.6803e-01, 8.3195e-01]],
[[-2.6699e+00, -3.8770e-02, -2.4125e+00, ..., -4.0792e-01,
-3.7989e+00, -8.8931e-01],
[-3.6522e+00, -2.9669e+00, -4.2664e+00, ..., -7.3503e+00,
-6.6295e+00, -1.6804e+00],
[-2.8738e+00, -2.8983e+00, -5.6230e+00, ..., -8.5748e+00,
-3.6042e+00, -1.0338e+00],
...,
[-1.8665e-01, 3.1178e-01, 1.7271e+00, ..., -9.6983e-01,
2.3874e-01, 7.6312e-01],
[-3.2015e+00, -2.9593e-01, -1.3335e-01, ..., 8.5707e-02,
-1.2395e-01, 6.7389e-01],
[-2.4660e+00, -2.5480e+00, -2.9263e+00, ..., -5.4672e-01,
-7.2935e-01, -2.9191e-01]],
[[ 2.3512e+00, 3.1870e+00, 6.9348e-01, ..., 1.7750e+00,
-5.2399e-01, 1.7942e+00],
[ 7.9591e-01, 1.4280e+00, 9.9807e-01, ..., -3.0472e-01,
-2.9371e+00, 1.4460e+00],
[ 1.3471e+00, 1.8668e+00, 3.9482e+00, ..., 2.3190e-01,
5.4015e-01, 2.4615e+00],
...,
[ 2.2223e+00, 8.5956e-01, -2.1418e+00, ..., 7.8700e-01,
1.9152e-01, 1.9342e+00],
[ 1.8077e+00, 2.6355e+00, -2.5767e+00, ..., 1.0279e+00,
7.0344e-01, 1.7833e+00],
[ 2.7758e+00, 2.9939e+00, 2.7567e+00, ..., 1.9209e+00,
9.6148e-01, 1.7904e+00]],
...,
[[-2.4246e+00, -6.9253e+00, -5.8074e+00, ..., -3.3476e+00,
-1.0039e+01, -7.2987e+00],
[-3.0225e+00, -7.6266e+00, -6.4138e+00, ..., -1.2694e+01,
-1.4456e+01, -6.9105e+00],
[-2.0007e+00, -5.5795e+00, -4.1053e+00, ..., -1.8773e+01,
-1.0859e+01, -5.0429e+00],
...,
[-2.3405e+00, -4.6137e+00, -1.5871e+00, ..., -6.7149e+00,
-4.8718e+00, -2.8372e+00],
[-2.5466e+00, -5.8120e+00, -3.1184e+00, ..., -4.3873e+00,
-3.1083e+00, -4.9001e+00],
[-2.9535e+00, -6.3332e+00, -2.7499e+00, ..., -3.0714e+00,
-1.5914e+00, -3.4021e+00]],
[[ 3.3773e-01, 8.1620e-01, 1.1460e+00, ..., -1.2361e+00,
3.2364e-01, 1.9624e+00],
[ 1.3621e+00, -2.8023e-01, -3.8751e-01, ..., -1.4973e+00,
2.6361e-01, 2.9785e+00],
[ 1.3310e+00, 1.2881e-01, -3.1648e+00, ..., -1.8473e+00,
2.3442e+00, 2.3792e+00],
...,
[ 1.6690e+00, 1.5044e+00, 1.9769e+00, ..., 2.6207e+00,
2.1817e+00, 1.5215e+00],
[ 1.5815e+00, 7.0163e-01, 2.7775e+00, ..., 1.9923e+00,
1.7210e+00, 2.6386e+00],
[ 8.9571e-01, -2.1737e-01, -1.3983e+00, ..., 1.3003e+00,
1.4794e+00, 1.4776e+00]],
[[-5.9546e-01, -1.9828e-01, -2.7933e+00, ..., 3.0194e-01,
-9.1489e-02, 1.2000e+00],
[-1.3943e+00, -1.7175e+00, -1.8841e+00, ..., -9.6669e-01,
-5.0322e+00, -1.2329e+00],
[-2.3399e+00, -7.5874e-01, 4.6694e-01, ..., -4.0043e+00,
-4.2337e+00, -3.0465e-01],
...,
[-1.2667e+00, -1.9628e+00, -6.8921e-01, ..., -1.0655e-02,
-5.5075e-01, -4.4475e-01],
[-1.4957e+00, -7.9052e-01, -6.8136e-01, ..., -2.4829e-01,
-2.8993e-01, 5.0233e-01],
[ 8.1655e-03, 8.8754e-01, -3.7283e-02, ..., 2.1436e-01,
1.2012e-02, 6.4016e-01]]],
[[[ 6.7877e-01, 1.0169e+00, 8.3536e-01, ..., 1.1117e+00,
1.8151e+00, 1.5503e+00],
[-2.7831e-01, -1.4590e+00, -2.7383e-01, ..., -1.4015e+00,
-2.2181e+00, -9.0059e-01],
[ 1.8819e-01, -1.0992e+00, -6.4559e-01, ..., -5.2988e-01,
-1.6915e+00, 1.3858e-01],
...,
[ 8.2459e-02, -1.0777e+00, -7.9826e-01, ..., -1.3289e+00,
5.5018e-01, 6.4314e-01],
[-4.5527e-01, -1.8578e+00, -8.7906e-01, ..., 1.0355e-01,
8.1471e-01, 1.4631e-01],
[ 4.3351e-01, -5.3512e-01, -6.9429e-01, ..., -2.1445e-01,
-2.7249e-01, 4.4284e-01]],
[[-2.9108e+00, -1.5442e+00, -3.3437e+00, ..., -1.1608e+00,
-3.6919e+00, -2.2624e+00],
[-4.6282e+00, -1.2967e+00, -1.2085e+00, ..., -2.3803e+00,
-3.9167e+00, -1.5304e+00],
[-5.0474e+00, -1.6939e+00, -6.0774e-01, ..., -8.5449e-01,
-1.8879e+00, -1.3064e+00],
...,
[-2.8107e+00, -1.7436e+00, -2.4740e+00, ..., -2.1193e+00,
-2.0186e+00, 2.2862e-01],
[-3.6779e+00, -1.2712e+00, -2.1378e+00, ..., -9.8304e-01,
-1.8041e+00, -7.8606e-02],
[-1.2796e+00, -8.6896e-01, -1.6248e+00, ..., -1.7290e+00,
-1.2679e+00, -9.3894e-02]],
[[ 1.7730e+00, 2.7560e+00, 2.4452e+00, ..., 9.4725e-01,
-5.6620e-01, 1.2138e+00],
[ 1.9010e-01, 1.1597e+00, 1.4268e+00, ..., -1.0170e+00,
-2.4370e+00, -9.0202e-02],
[ 4.6383e-01, 6.5454e-01, 4.7569e-01, ..., -1.0023e+00,
-5.0157e-01, -1.1209e-01],
...,
[ 1.0234e+00, 1.2870e+00, 3.5411e-01, ..., 1.4106e-01,
6.5931e-02, 6.7734e-01],
[ 6.7554e-01, 1.8916e+00, 7.0468e-01, ..., -1.0153e+00,
5.3996e-01, 6.4541e-01],
[ 1.5367e+00, 1.6248e+00, 1.4907e+00, ..., -8.8126e-01,
3.4738e-01, 1.1969e+00]],
...,
[[-1.3798e+00, -4.4889e+00, -3.1045e+00, ..., -5.5057e+00,
-3.4844e+00, -4.7913e+00],
[-1.8476e+00, -3.8474e+00, -3.2893e+00, ..., -7.1783e+00,
-9.2382e-01, -1.2648e+00],
[-1.2332e+00, -3.0641e+00, -2.3571e+00, ..., -6.5754e+00,
-5.5760e-01, 3.2960e-01],
...,
[-1.5802e+00, -4.0688e+00, -4.3761e+00, ..., -4.5913e-01,
-4.4216e+00, -4.6307e+00],
[-1.4722e+00, -4.4813e+00, -5.1599e+00, ..., -3.3344e+00,
-7.3024e+00, -6.2484e+00],
[-1.6370e+00, -4.9687e+00, -4.2090e+00, ..., -3.0684e+00,
-3.9628e+00, -3.7546e+00]],
[[ 7.3232e-01, 2.6482e-01, -6.0496e-01, ..., 1.0864e+00,
-8.4342e-01, -3.3662e-02],
[ 1.2930e+00, 1.0320e+00, 7.8563e-01, ..., 1.5587e+00,
4.0436e-01, 9.4698e-01],
[ 5.4766e-01, 1.2495e+00, 9.0145e-01, ..., -2.3952e-02,
1.1588e+00, 9.6700e-01],
...,
[ 8.7365e-01, 6.9895e-02, -4.7864e-02, ..., 4.4891e-01,
1.0924e+00, 2.1445e+00],
[ 1.5240e+00, 1.5039e+00, 1.1616e+00, ..., 1.7419e+00,
1.0018e+00, 2.8675e+00],
[ 6.2947e-01, 8.9748e-01, 9.0647e-01, ..., 8.0895e-01,
1.1654e+00, 1.6003e+00]],
[[-3.4990e-01, -1.3370e-01, -9.3222e-01, ..., -3.3490e-02,
3.4575e-01, 1.6589e+00],
[-1.4071e+00, -8.1035e-01, -1.7119e+00, ..., -5.3046e+00,
-2.1495e+00, 8.8240e-01],
[-1.8544e+00, 2.2032e-01, -1.0521e+00, ..., -5.2719e+00,
-3.2160e+00, 3.9479e-01],
...,
[-1.7663e+00, -1.6765e+00, -1.7329e+00, ..., 2.2038e-01,
-1.9757e+00, -1.1711e+00],
[-1.4887e+00, -1.6632e+00, -1.6958e+00, ..., 3.7191e-01,
-1.9423e+00, -8.7893e-01],
[-1.3369e-01, 3.4354e-02, -6.5491e-01, ..., -1.8247e+00,
-2.2294e+00, -4.0745e-01]]],
[[[ 1.1223e+00, 1.4752e+00, 1.2148e+00, ..., 2.9921e-01,
3.7623e-01, 2.1476e-01],
[-9.6224e-01, -1.5070e+00, -1.0173e+00, ..., 3.4079e-01,
3.8220e-01, -1.1710e+00],
[ 3.1233e-02, -1.0132e+00, -1.8689e+00, ..., -3.5022e-01,
1.0097e+00, 6.2621e-01],
...,
[ 5.4811e-01, 4.1173e-01, 1.0788e+00, ..., 4.0100e-01,
5.4529e-01, 5.0832e-01],
[-3.9457e-01, -8.9338e-01, -2.9297e-01, ..., 2.3325e-01,
1.3502e+00, -1.3920e-01],
[-8.7410e-02, -1.8009e-01, -6.8585e-01, ..., -2.5621e-01,
2.6559e-01, 6.1636e-01]],
[[-3.1070e+00, -2.0330e+00, -2.2435e+00, ..., -1.8085e+00,
-2.1051e+00, -9.9801e-01],
[-2.8739e+00, -3.2241e+00, -1.5446e+00, ..., -4.1867e-01,
-1.1528e+00, -2.1882e-01],
[-4.1760e+00, -3.8120e+00, -1.5437e+00, ..., -1.2681e+00,
-1.4295e+00, -7.4325e-01],
...,
[-3.1781e-01, -8.8146e-01, -5.4355e-01, ..., -1.4642e+00,
-7.5589e-01, 4.1238e-01],
[-1.2932e+00, -7.2522e-01, -3.5029e-01, ..., -5.1939e-01,
-3.7028e-01, 2.8532e-01],
[-4.6416e-01, -1.0221e+00, -8.3166e-01, ..., -2.8562e-01,
-6.5298e-01, -5.5577e-01]],
[[ 2.2907e+00, 2.3179e+00, 1.8072e+00, ..., 2.9326e+00,
1.4158e+00, 2.8135e+00],
[ 1.0878e-01, 1.9521e-01, 1.4109e+00, ..., 1.9169e+00,
-4.2002e-02, 2.4505e+00],
[ 2.4035e-01, 1.4831e+00, 2.2245e+00, ..., 1.2980e+00,
5.6165e-01, 1.8114e+00],
...,
[-3.2013e-01, 4.1707e-01, 8.5236e-01, ..., -1.3829e+00,
-3.3328e-01, 1.6531e+00],
[-1.0052e+00, 5.7238e-01, 4.8498e-01, ..., 1.3130e+00,
8.8493e-01, 1.5602e+00],
[ 6.6147e-01, 4.9948e-01, 1.0367e+00, ..., 1.2320e+00,
1.0356e+00, 2.1424e+00]],
...,
[[-2.8913e+00, -8.4432e+00, -6.7240e+00, ..., -3.5270e+00,
-2.4472e+00, -3.8607e+00],
[-3.7923e+00, -1.0188e+01, -6.2148e+00, ..., -5.8335e+00,
-3.8541e+00, -4.2884e+00],
[-2.5414e+00, -9.3269e+00, -6.3857e+00, ..., -3.2459e+00,
-2.3836e+00, -1.5884e+00],
...,
[-3.5917e+00, -8.5072e+00, -5.0236e+00, ..., -4.4983e+00,
-2.4626e+00, -2.0874e+00],
[-4.0837e+00, -9.4122e+00, -4.9064e+00, ..., -4.9261e+00,
-3.9891e+00, -4.4219e+00],
[-3.2722e+00, -7.9010e+00, -3.0359e+00, ..., -4.8280e+00,
-2.5045e+00, -2.8939e+00]],
[[-3.8968e-01, 7.1005e-01, 8.6342e-01, ..., 1.3044e+00,
1.1178e+00, 1.2441e+00],
[ 6.5926e-01, 8.0885e-01, 1.3544e+00, ..., 1.9949e+00,
2.3875e+00, 2.5473e+00],
[-1.0070e+00, -2.3013e-01, 2.3576e+00, ..., 1.5847e+00,
1.6303e+00, 1.9096e+00],
...,
[ 1.4415e+00, 7.9365e-01, 1.3784e+00, ..., 5.0603e-01,
1.5793e+00, 1.7526e+00],
[ 1.7061e+00, 1.5713e+00, 1.6933e+00, ..., 1.5762e+00,
1.4825e+00, 2.4832e+00],
[ 7.5953e-01, 5.6227e-01, 1.1327e+00, ..., 1.2677e+00,
1.3763e+00, 1.3498e+00]],
[[-1.9776e-01, -3.8665e-01, -3.6707e-01, ..., -1.4812e-01,
-1.3575e-01, 8.6191e-01],
[-4.7377e-01, -2.6888e+00, -1.2421e+00, ..., 1.7237e-01,
-4.0163e-02, 3.6241e-01],
[-1.6806e+00, -1.9405e+00, -1.1777e+00, ..., 3.6857e-01,
2.3161e-01, 2.1559e-01],
...,
[-1.9650e+00, -2.5567e+00, -8.6303e-01, ..., -1.6215e+00,
-1.2910e+00, -8.3846e-02],
[-1.3869e+00, -2.5438e+00, -8.2188e-01, ..., -7.7587e-01,
-1.6021e-01, 3.8976e-01],
[-4.0918e-01, -1.1173e+00, -6.3176e-01, ..., 3.4616e-01,
2.5826e-01, 2.9862e-01]]],
[[[ 8.4817e-01, 1.5136e+00, 1.2220e+00, ..., 1.3892e+00,
1.0898e+00, 9.7704e-01],
[-1.0201e+00, -2.0917e+00, -1.2154e+00, ..., -2.7657e-01,
-5.3308e-01, -1.5820e+00],
[-2.3123e-01, -7.8777e-01, 2.3957e-01, ..., -5.5976e-02,
7.5773e-01, 8.6497e-01],
...,
[-2.9382e-03, -1.1673e+00, -1.1461e+00, ..., -3.8776e-01,
-5.2636e-02, 2.2335e-01],
[-5.8637e-01, -1.8373e+00, -1.1746e+00, ..., -5.8496e-01,
-2.7416e-01, -1.4254e-01],
[-3.4756e-01, -1.7436e+00, -1.6313e+00, ..., -1.1915e+00,
-1.5727e+00, -2.8410e-01]],
[[-3.7309e+00, -8.0376e-01, -1.6517e+00, ..., -1.5412e+00,
-2.4790e+00, -1.2304e+00],
[-3.9488e+00, -2.8959e+00, -1.8652e+00, ..., -2.2756e+00,
-3.1460e+00, -9.4582e-01],
[-2.3380e+00, -1.7078e+00, -1.8058e+00, ..., -2.4985e+00,
-2.8174e+00, -1.0537e+00],
...,
[-1.6639e+00, -1.7003e+00, -2.5934e+00, ..., -2.0901e+00,
-2.1918e+00, -7.6458e-01],
[-3.0863e+00, -1.6410e+00, -2.7088e+00, ..., -2.4584e+00,
-2.4251e+00, -1.2340e+00],
[-2.3255e+00, -1.9327e+00, -2.6786e+00, ..., -2.0230e+00,
-1.9470e+00, -1.0177e+00]],
[[ 2.8722e+00, 4.2458e+00, 3.3769e+00, ..., 2.9240e+00,
1.0931e+00, 3.0538e+00],
[ 1.9766e-01, 1.0615e+00, 2.2990e+00, ..., 1.3517e+00,
-7.5488e-01, 2.7748e+00],
[ 9.6604e-01, 1.4769e+00, 1.5795e+00, ..., 3.3321e-01,
-1.1610e-01, 2.0663e+00],
...,
[ 9.7882e-01, 1.1063e+00, 1.1431e+00, ..., 2.3285e-01,
1.1448e+00, 2.4048e+00],
[ 6.6958e-01, 1.8335e+00, 1.2289e+00, ..., 5.9948e-02,
1.1273e+00, 1.8745e+00],
[ 1.6142e+00, 1.6189e+00, 1.9090e+00, ..., 1.3154e+00,
1.8392e+00, 1.8090e+00]],
...,
[[-1.8691e+00, -7.2978e+00, -5.5627e+00, ..., -4.0931e+00,
-3.3230e+00, -5.5754e+00],
[-2.3912e+00, -6.9825e+00, -4.2660e+00, ..., -3.7172e+00,
-2.8684e+00, -4.3811e+00],
[-9.6008e-01, -4.4585e+00, -3.7619e+00, ..., -2.8304e+00,
-3.0576e+00, -2.5323e+00],
...,
[-2.5957e+00, -5.3942e+00, -4.4305e+00, ..., -6.0400e+00,
-4.7302e+00, -4.7512e+00],
[-2.7546e+00, -5.7454e+00, -5.4830e+00, ..., -6.2817e+00,
-5.7680e+00, -6.3438e+00],
[-1.4700e+00, -4.0528e+00, -2.8304e+00, ..., -3.3366e+00,
-2.6889e+00, -3.2602e+00]],
[[-1.7974e-01, 2.9583e-01, 5.2237e-01, ..., 6.9545e-01,
8.7301e-01, 1.4038e+00],
[ 1.6283e+00, 1.0615e+00, 1.8620e+00, ..., 1.8251e+00,
2.5703e+00, 3.0418e+00],
[ 1.1952e+00, 2.6882e-01, 6.2002e-01, ..., 1.0378e+00,
1.6395e+00, 2.3845e+00],
...,
[ 1.2864e+00, 2.3931e-01, -7.8554e-02, ..., 1.1329e+00,
1.0781e+00, 1.6319e+00],
[ 1.8879e+00, 1.6697e+00, 1.0645e+00, ..., 1.4483e+00,
1.2944e+00, 1.9962e+00],
[ 4.6261e-01, 8.0804e-01, 7.1565e-01, ..., 5.0928e-01,
7.2195e-01, 1.6703e+00]],
[[-5.6474e-01, 2.8006e-01, 1.2673e-01, ..., 6.0618e-01,
5.3118e-01, 1.3605e+00],
[-1.3674e+00, -1.8528e+00, -1.6171e+00, ..., -5.9618e-01,
-4.5589e-01, 2.3142e-01],
[-2.2347e+00, -8.8369e-01, 2.1978e-01, ..., 1.6702e-01,
1.4927e-01, 7.9367e-01],
...,
[-1.9111e+00, -1.7996e+00, -1.5499e+00, ..., -4.6888e-01,
-9.1139e-01, -8.4145e-01],
[-1.5560e+00, -1.9733e+00, -1.7176e+00, ..., -1.1902e+00,
-1.2960e+00, -6.7901e-01],
[-8.3989e-01, -8.6073e-01, -1.1584e+00, ..., -6.0726e-01,
-8.2837e-01, -1.0624e-01]]],
[[[ 7.5620e-01, 9.0739e-01, 5.3257e-01, ..., 7.1572e-01,
9.9828e-01, 4.4452e-01],
[-1.0811e+00, -1.0476e+00, -1.8515e-02, ..., -2.3735e-01,
-1.4033e-01, -3.1669e-01],
[-1.8300e-01, -9.5757e-01, -1.7902e-01, ..., -7.2585e-01,
-1.5244e-01, 5.9580e-01],
...,
[ 5.0914e-01, -4.6805e-01, -1.6839e-01, ..., -2.2574e-01,
-1.0840e-01, 3.4233e-01],
[-3.1754e-02, -1.1280e+00, -4.0717e-01, ..., -1.5520e-01,
1.3709e-01, -1.2416e-01],
[ 5.8962e-01, -2.1665e-01, -6.9289e-01, ..., -4.8656e-01,
-9.7071e-01, 1.8014e-01]],
[[-3.8169e+00, -1.8439e+00, -2.9225e+00, ..., -1.5500e+00,
-2.0709e+00, -1.8341e+00],
[-2.8773e+00, -1.5367e+00, -1.0473e+00, ..., -1.5391e+00,
-2.1477e+00, -2.0824e+00],
[-1.8859e+00, -1.3248e+00, -1.4394e+00, ..., -1.4529e+00,
-1.8106e+00, -1.3919e+00],
...,
[-6.8980e-01, -5.8514e-01, -9.8787e-01, ..., -9.9010e-01,
-1.0344e+00, -8.8045e-01],
[-1.5126e+00, -5.2677e-02, -3.1263e-01, ..., -8.0763e-01,
-1.1695e+00, -1.2649e+00],
[-2.8157e-02, -4.8145e-01, -7.0393e-01, ..., -7.3628e-01,
-6.7315e-01, -4.8032e-01]],
[[ 2.6809e+00, 3.0522e+00, 3.0423e+00, ..., 1.4241e+00,
8.4087e-01, 2.5460e+00],
[ 8.0140e-01, 1.0683e+00, 2.4129e+00, ..., 1.1028e+00,
6.8734e-01, 2.6137e+00],
[ 1.0446e+00, 7.0490e-01, 1.5076e+00, ..., 1.3863e-02,
9.0645e-01, 1.8382e+00],
...,
[-1.2342e-03, 4.0605e-01, 1.3379e+00, ..., -6.6226e-01,
5.7383e-01, 1.4916e+00],
[ 2.0696e-01, 1.9307e+00, 1.8495e+00, ..., -6.1784e-01,
8.0130e-01, 1.1245e+00],
[ 1.7458e+00, 1.0853e+00, 1.9234e+00, ..., 3.2397e-01,
1.3158e+00, 1.3925e+00]],
...,
[[-2.5710e+00, -7.2690e+00, -4.6975e+00, ..., -5.6251e+00,
-3.6069e+00, -4.2631e+00],
[-3.1147e+00, -8.6852e+00, -5.9198e+00, ..., -5.4640e+00,
-3.7278e+00, -3.1781e+00],
[-2.0954e+00, -5.9216e+00, -3.2934e+00, ..., -4.1486e+00,
-2.2988e+00, -1.2449e+00],
...,
[-2.5964e+00, -5.3829e+00, -3.1127e+00, ..., -5.1568e+00,
-3.0953e+00, -2.6525e+00],
[-3.4824e+00, -7.8539e+00, -4.6799e+00, ..., -6.0468e+00,
-4.1836e+00, -4.1189e+00],
[-2.8430e+00, -7.1792e+00, -3.6550e+00, ..., -3.6868e+00,
-2.3739e+00, -2.2817e+00]],
[[-6.3769e-01, -2.6204e-02, 2.9669e-01, ..., 1.8668e+00,
1.6784e+00, 1.1842e+00],
[ 1.4480e+00, 9.3843e-01, 1.0336e+00, ..., 1.9052e+00,
1.1858e+00, 1.8356e+00],
[ 1.1365e+00, 2.6899e-01, 3.6381e-01, ..., 1.7116e+00,
4.7605e-01, 1.3062e+00],
...,
[ 9.8280e-01, -5.0742e-01, 3.5164e-01, ..., 1.9032e+00,
6.5831e-01, 9.8328e-01],
[ 1.8920e+00, 1.3663e+00, 1.4584e+00, ..., 1.6431e+00,
6.7615e-01, 1.4743e+00],
[ 1.1094e+00, 9.7119e-01, 1.2971e+00, ..., 8.7674e-01,
7.1737e-01, 1.4338e+00]],
[[-8.1569e-01, -7.0241e-01, -5.4604e-01, ..., -2.8708e-01,
-6.2409e-01, -8.5128e-02],
[-1.4585e+00, -1.2742e+00, -5.9789e-01, ..., -1.5145e+00,
-1.4606e+00, -1.0866e+00],
[-2.0911e+00, -7.7289e-01, 1.9830e-01, ..., -8.6608e-01,
-1.1197e+00, -2.5153e-01],
...,
[-1.8993e+00, -2.2527e+00, -8.0045e-01, ..., -1.0393e+00,
-1.6120e+00, -8.2506e-01],
[-2.0316e+00, -2.0010e+00, -8.6731e-01, ..., -1.1168e+00,
-1.6478e+00, -7.1088e-01],
[ 1.7667e-02, 1.7344e-01, -2.0509e-01, ..., -4.4056e-01,
-8.0262e-01, -4.1580e-02]]]], device='cuda:0'),
tensor([[[[ 8.6551e-01, 1.5220e+00, 1.2089e+00, ..., 1.3967e+00,
1.0704e+00, 9.2501e-01],
[-1.0322e+00, -2.0134e+00, -1.1390e+00, ..., -3.3336e-01,
-4.9573e-01, -1.4945e+00],
[-1.2186e-01, -5.0207e-01, 6.3457e-01, ..., -3.0507e-01,
-3.8612e-01, 2.4819e-01],
...,
[-4.3134e-02, -1.2223e+00, -1.1137e+00, ..., -3.7872e-01,
4.0177e-02, 3.3964e-01],
[-5.9974e-01, -1.8999e+00, -1.3200e+00, ..., -6.2405e-01,
-5.0325e-01, -2.7490e-01],
[-3.5450e-01, -1.7058e+00, -1.6385e+00, ..., -1.2074e+00,
-1.4454e+00, -3.0835e-01]],
[[-3.6500e+00, -7.9230e-01, -1.5710e+00, ..., -1.6444e+00,
-2.5088e+00, -1.2566e+00],
[-3.9042e+00, -2.9069e+00, -1.8185e+00, ..., -2.1795e+00,
-2.9577e+00, -9.1235e-01],
[-2.3840e+00, -2.1942e+00, -3.8051e+00, ..., -3.5050e+00,
-3.1468e+00, -1.1954e+00],
...,
[-1.6327e+00, -1.7442e+00, -2.6997e+00, ..., -2.0438e+00,
-2.0066e+00, -7.5900e-01],
[-3.0514e+00, -1.7057e+00, -2.8477e+00, ..., -2.4678e+00,
-2.4605e+00, -1.2722e+00],
[-2.2895e+00, -1.8904e+00, -2.6620e+00, ..., -1.9354e+00,
-1.9080e+00, -9.9749e-01]],
[[ 2.8321e+00, 4.2122e+00, 3.3536e+00, ..., 2.9288e+00,
1.0807e+00, 3.0315e+00],
[ 1.6163e-01, 1.1024e+00, 2.2992e+00, ..., 1.4393e+00,
-7.1301e-01, 2.7694e+00],
[ 9.3652e-01, 1.3124e+00, 1.1495e+00, ..., 2.7411e+00,
9.0577e-01, 3.1514e+00],
...,
[ 9.3027e-01, 1.1027e+00, 1.4226e+00, ..., 5.2068e-02,
1.0554e+00, 2.3532e+00],
[ 5.9805e-01, 1.7777e+00, 1.1350e+00, ..., 4.7149e-03,
9.6927e-01, 1.8271e+00],
[ 1.5987e+00, 1.6388e+00, 1.8430e+00, ..., 1.3863e+00,
1.8147e+00, 1.7372e+00]],
...,
[[-1.8455e+00, -7.3147e+00, -5.4689e+00, ..., -4.0621e+00,
-3.2777e+00, -5.5266e+00],
[-2.3602e+00, -7.0357e+00, -4.2228e+00, ..., -3.7307e+00,
-2.8702e+00, -4.5073e+00],
[-9.5897e-01, -4.7379e+00, -5.2653e+00, ..., -4.9284e+00,
-4.3698e+00, -2.7237e+00],
...,
[-2.5712e+00, -5.5033e+00, -4.3054e+00, ..., -5.0715e+00,
-4.6212e+00, -4.8028e+00],
[-2.7372e+00, -5.7780e+00, -5.3998e+00, ..., -6.3932e+00,
-5.6880e+00, -6.1093e+00],
[-1.4708e+00, -4.1026e+00, -2.8067e+00, ..., -3.3276e+00,
-2.7239e+00, -3.2586e+00]],
[[-1.7750e-01, 2.8546e-01, 5.6689e-01, ..., 6.0083e-01,
8.1227e-01, 1.4054e+00],
[ 1.6132e+00, 1.0436e+00, 1.8500e+00, ..., 1.8869e+00,
2.5818e+00, 3.0213e+00],
[ 1.1146e+00, -3.0626e-01, 6.2877e-01, ..., 7.0102e-01,
1.3986e+00, 2.3872e+00],
...,
[ 1.3234e+00, 2.6322e-01, -5.3295e-02, ..., 1.2061e+00,
1.1337e+00, 1.5450e+00],
[ 1.8622e+00, 1.5989e+00, 9.8400e-01, ..., 1.4333e+00,
1.2743e+00, 1.9663e+00],
[ 4.4450e-01, 7.7422e-01, 7.1903e-01, ..., 4.6510e-01,
6.6756e-01, 1.6502e+00]],
[[-5.4147e-01, 2.7316e-01, 1.4666e-01, ..., 5.7532e-01,
5.3309e-01, 1.3383e+00],
[-1.3397e+00, -1.8502e+00, -1.5710e+00, ..., -5.9539e-01,
-3.5333e-01, 2.5080e-01],
[-2.2045e+00, -1.1804e+00, 2.8680e-01, ..., -1.0762e+00,
-4.4661e-01, 5.1229e-01],
...,
[-1.8930e+00, -1.7827e+00, -1.6057e+00, ..., -1.0680e+00,
-1.0518e+00, -8.6371e-01],
[-1.5260e+00, -2.0603e+00, -1.8189e+00, ..., -1.2526e+00,
-1.4693e+00, -7.3556e-01],
[-8.3578e-01, -8.4832e-01, -1.1774e+00, ..., -6.8330e-01,
-8.4248e-01, -4.2842e-02]]],
[[[ 1.0657e+00, 1.5509e+00, 1.1492e+00, ..., 2.3756e-01,
3.3402e-01, 2.6198e-01],
[-1.0225e+00, -1.5205e+00, -8.8628e-01, ..., 4.3543e-01,
3.6242e-01, -1.1629e+00],
[-9.0018e-01, -9.2903e-01, -2.9639e-01, ..., -3.6348e-01,
1.1211e+00, 7.0697e-01],
...,
[ 1.2649e+00, -1.6839e+00, -1.2005e+00, ..., 6.5456e-01,
-1.1027e+00, 2.3792e-01],
[ 9.2780e-01, 9.0357e-01, 1.1606e+00, ..., 6.6863e-01,
-4.7787e-01, -7.3365e-01],
[-1.8122e-01, -2.5306e-01, -5.7661e-01, ..., -3.3913e-01,
-3.4321e-02, 3.3856e-01]],
[[-3.2956e+00, -2.0635e+00, -2.3737e+00, ..., -1.7440e+00,
-2.1607e+00, -1.0826e+00],
[-2.9835e+00, -4.0580e+00, -3.2265e+00, ..., -3.8697e-01,
-1.4037e+00, -3.7707e-01],
[-8.2059e-01, -9.3292e-01, -1.7695e+00, ..., -1.1038e+00,
-1.4144e+00, -7.6030e-01],
...,
[-4.8730e+00, -6.1838e+00, -3.1617e+00, ..., -1.6143e+00,
-4.2462e+00, -2.3484e+00],
[-2.0062e+00, -2.3705e+00, -2.1469e+00, ..., -3.2161e+00,
-5.0741e+00, -1.0279e+00],
[-4.3754e-01, -1.0740e+00, -9.5646e-01, ..., -1.6831e+00,
-2.2627e+00, -1.2333e+00]],
[[ 2.3951e+00, 2.3368e+00, 1.8961e+00, ..., 2.8436e+00,
1.3268e+00, 2.9278e+00],
[ 4.4306e-01, 1.5703e-01, 1.1393e+00, ..., 1.6343e+00,
-1.9074e-01, 2.4696e+00],
[ 1.9708e-01, 2.4311e-01, 3.6019e-01, ..., 1.0616e+00,
5.6126e-01, 1.8257e+00],
...,
[-8.7716e-01, -4.2624e+00, -4.2962e+00, ..., -2.4619e+00,
-4.4848e-01, 1.1574e+00],
[ 5.6533e-01, 2.2175e+00, 1.0515e+00, ..., -2.3286e+00,
-3.4487e-01, 1.5547e+00],
[ 8.3304e-01, 8.9953e-01, 1.4851e+00, ..., 1.8953e+00,
1.8050e+00, 1.6471e+00]],
...,
[[-3.1236e+00, -8.6478e+00, -6.7681e+00, ..., -3.1748e+00,
-2.2982e+00, -4.0441e+00],
[-3.4455e+00, -9.3933e+00, -5.3115e+00, ..., -5.5274e+00,
-3.6337e+00, -4.3885e+00],
[-1.5409e+00, -5.5836e+00, -2.4177e+00, ..., -2.7777e+00,
-2.1807e+00, -1.6870e+00],
...,
[-3.5543e+00, -6.8126e+00, -5.3147e+00, ..., -1.0604e+00,
-5.0992e+00, -6.5821e+00],
[-4.0606e+00, -8.8683e+00, -4.8593e+00, ..., -4.6250e+00,
-5.5834e+00, -6.4041e+00],
[-3.0424e+00, -7.2891e+00, -2.7602e+00, ..., -3.9306e+00,
-3.3515e+00, -5.2110e+00]],
[[-3.9790e-01, 7.2968e-01, 7.6454e-01, ..., 1.3488e+00,
1.0803e+00, 1.2894e+00],
[ 6.5533e-01, 1.7112e-01, 4.7456e-01, ..., 1.8608e+00,
2.2606e+00, 2.5618e+00],
[ 1.0368e+00, 9.9164e-01, -3.4097e-01, ..., 1.5529e+00,
1.5630e+00, 1.9353e+00],
...,
[ 5.2313e-01, 4.8450e-01, 3.8837e-01, ..., 4.6727e-01,
2.5564e-01, 1.1227e+00],
[ 2.2794e+00, 2.3901e+00, 2.4463e+00, ..., 3.4804e-01,
-1.4408e-01, 2.5198e+00],
[ 8.8494e-01, 7.8739e-01, 1.3586e+00, ..., 6.1431e-01,
1.5886e+00, 1.7795e+00]],
[[-1.6191e-01, -3.5259e-01, -2.3159e-01, ..., -6.0932e-02,
-5.1885e-02, 8.7354e-01],
[-8.6492e-01, -2.8591e+00, -9.8808e-01, ..., 3.2409e-01,
-6.9137e-02, 3.4544e-01],
[-1.6211e+00, -1.1789e+00, -4.6389e-01, ..., 3.1377e-01,
1.6572e-01, 2.8416e-01],
...,
[-2.6361e+00, -4.5091e+00, -2.9488e+00, ..., 5.2327e-01,
-6.6665e-01, -1.2874e+00],
[-2.1833e+00, -2.9434e+00, -1.3303e+00, ..., -2.2584e+00,
-3.9999e+00, -2.3912e-01],
[-4.2890e-01, -1.2122e+00, -6.6544e-01, ..., -3.7491e-01,
2.0392e-01, 8.7107e-01]]],
[[[ 1.0391e+00, 1.3062e+00, 1.4000e+00, ..., -5.3630e+00,
-2.5085e+00, 9.1073e-01],
[-4.7256e-01, -2.2785e+00, -1.4688e+00, ..., -3.2209e+00,
-2.8518e+00, -2.9365e+00],
[-2.8716e-01, -1.0595e+00, -1.0658e+00, ..., -3.0051e-01,
-8.1175e-01, -7.2468e-01],
...,
[ 4.9693e-01, -2.4494e-01, -5.4466e-01, ..., -2.2474e-01,
1.0201e+00, 8.2829e-01],
[ 1.5208e-02, -6.3056e-01, 6.0019e-03, ..., 6.7382e-01,
1.3300e+00, 1.3019e-01],
[ 8.2301e-01, -1.9879e-01, -1.3213e-01, ..., 3.6719e-01,
2.8528e-01, 7.5636e-01]],
[[-2.6524e+00, 1.9168e-01, -1.3101e+00, ..., -7.3813e+00,
-7.6712e+00, -3.6843e+00],
[-3.6903e+00, -2.7390e+00, -3.5881e+00, ..., -8.2640e+00,
-1.1435e+01, -5.4339e+00],
[-2.7948e+00, -2.4424e+00, -3.8449e+00, ..., -3.3162e+00,
-4.3399e+00, -2.5099e+00],
...,
[-7.4077e-01, -7.4972e-01, -1.7472e+00, ..., -2.7055e+00,
1.5624e-02, 7.1289e-01],
[-2.5620e+00, -2.5151e-01, -3.5539e-01, ..., -1.4071e-01,
-8.8052e-03, 6.5552e-01],
[-1.6852e+00, -1.3569e+00, -9.6033e-01, ..., -5.2032e-01,
-7.0989e-01, -2.2997e-01]],
[[ 2.3213e+00, 2.8599e+00, 2.3215e+00, ..., -2.6332e+00,
-2.6573e+00, -1.3952e+00],
[ 7.8605e-01, 6.5393e-01, 2.2853e+00, ..., -4.2776e+00,
-6.7539e+00, -4.3149e+00],
[ 1.2724e+00, 1.4908e+00, 2.6984e+00, ..., -5.3257e-01,
-2.1399e+00, -3.5561e+00],
...,
[ 8.9992e-01, 8.1804e-01, 2.5317e+00, ..., 1.0424e+00,
1.1219e+00, 1.8996e+00],
[ 1.1407e+00, 2.3333e+00, 2.0162e+00, ..., 1.1025e+00,
6.4235e-01, 1.7493e+00],
[ 2.4498e+00, 1.9187e+00, 2.3983e+00, ..., 1.9237e+00,
1.0225e+00, 1.7658e+00]],
...,
[[-2.4007e+00, -6.6031e+00, -5.6682e+00, ..., -8.2593e-01,
-2.6086e+00, -5.9109e+00],
[-3.0148e+00, -7.2987e+00, -5.0996e+00, ..., 2.1406e-01,
2.7385e-03, -1.4160e+00],
[-2.0389e+00, -5.7296e+00, -3.8078e+00, ..., -4.3602e+00,
-1.3390e+00, -2.2492e+00],
...,
[-2.4523e+00, -5.0333e+00, -3.1359e+00, ..., -2.6005e+00,
-3.1435e+00, -3.3442e+00],
[-2.5214e+00, -5.5706e+00, -4.0269e+00, ..., -6.3056e+00,
-4.5802e+00, -5.0453e+00],
[-2.6675e+00, -6.0446e+00, -2.9388e+00, ..., -3.0001e+00,
-1.7022e+00, -3.4336e+00]],
[[ 3.6866e-01, 1.1588e+00, 5.6603e-01, ..., 2.1967e+00,
-1.1212e+00, -2.2541e+00],
[ 1.2755e+00, 6.4774e-01, -6.3014e-01, ..., -4.3111e+00,
-4.5912e+00, -1.9631e+00],
[ 1.3278e+00, 7.0190e-01, -6.5076e-01, ..., -4.3094e+00,
-2.5949e+00, 3.1393e-01],
...,
[ 1.6099e+00, 4.4966e-01, 1.2407e+00, ..., 1.7776e+00,
2.2197e+00, 1.5598e+00],
[ 2.0088e+00, 1.1294e+00, 1.8229e+00, ..., 2.2398e+00,
1.6966e+00, 2.6500e+00],
[ 1.1583e+00, 8.2941e-01, 1.3446e+00, ..., 1.3321e+00,
1.5319e+00, 1.5390e+00]],
[[-6.1035e-01, 4.7985e-02, -4.8362e-01, ..., -3.8081e+00,
-1.2387e+00, 7.4035e-01],
[-1.3340e+00, -1.8732e+00, -1.5829e+00, ..., -6.3711e+00,
-8.2932e+00, -1.2747e+00],
[-2.2092e+00, -1.1065e+00, -1.4682e-01, ..., -3.1367e+00,
-3.3120e+00, 1.6243e+00],
...,
[-1.8864e+00, -1.5851e+00, -9.3097e-01, ..., -1.5642e+00,
-3.4041e-01, -5.2940e-01],
[-1.6400e+00, -1.2542e+00, -6.6805e-01, ..., 4.2846e-01,
8.1678e-02, 4.9792e-01],
[-2.7685e-01, 4.1533e-01, -1.9143e-01, ..., 2.5815e-01,
4.0768e-02, 6.2449e-01]]],
[[[ 6.3491e-01, 1.0599e+00, 7.9388e-01, ..., 7.9798e-01,
6.8303e-01, 1.3358e-01],
[-3.8223e-01, -1.6043e+00, -6.0651e-01, ..., 5.0015e-01,
6.2738e-01, -4.5454e-01],
[-1.0707e-01, -9.7236e-01, -1.7778e-01, ..., 2.7293e-01,
4.4981e-01, 1.6512e+00],
...,
[-1.0478e+00, 5.8880e-01, -1.6993e+00, ..., 4.4787e-01,
8.9998e-01, 1.0851e+00],
[-1.6588e+00, -1.0468e+00, -1.8614e+00, ..., -4.5239e+00,
-8.0744e-01, -1.3324e+00],
[-4.8988e-01, 5.2888e-01, 2.2511e-01, ..., 1.1518e+00,
4.7433e-01, 8.8130e-01]],
[[-3.4194e+00, -1.4833e+00, -3.3553e+00, ..., -1.8288e+00,
-3.3771e+00, -1.2845e+00],
[-4.2448e+00, -1.2714e+00, -2.2019e+00, ..., -1.5151e+00,
-3.5628e+00, -1.1396e+00],
[-4.2407e+00, -1.8196e+00, -2.3681e+00, ..., 5.7922e-02,
-2.2764e-01, -1.9057e-01],
...,
[-4.8889e+00, -4.2793e+00, 1.0586e+00, ..., 4.4425e-01,
8.6998e-01, 3.1119e-02],
[-5.5169e+00, -2.5112e+00, 1.3898e+00, ..., -4.2481e+00,
-3.9442e+00, -1.2622e+00],
[-2.3886e+00, -2.5972e+00, -1.9996e+00, ..., -1.5994e+00,
-1.0054e+00, -6.7316e-01]],
[[ 2.1909e+00, 3.0784e+00, 2.5160e+00, ..., 2.5285e+00,
1.1135e+00, 2.4511e+00],
[ 3.6136e-01, 9.1070e-01, 1.4164e+00, ..., 1.5806e+00,
6.8459e-02, 1.2322e+00],
[ 1.0029e+00, 9.8978e-01, 2.5673e-01, ..., -1.3110e+00,
-1.0692e+00, 4.9231e-01],
...,
[ 3.2392e+00, -6.6791e-01, -5.0858e+00, ..., -1.9639e+00,
-3.2152e+00, -3.0480e+00],
[ 1.9801e+00, 6.5684e-01, -2.5925e+00, ..., -4.5899e+00,
-3.7781e+00, -3.3977e+00],
[ 2.1431e+00, 2.4030e+00, 1.3000e-01, ..., 1.6134e+00,
1.6109e+00, 1.2912e+00]],
...,
[[-1.8361e+00, -5.0298e+00, -4.1866e+00, ..., -4.5117e+00,
-3.4826e+00, -3.9272e+00],
[-2.2970e+00, -4.4132e+00, -4.1810e+00, ..., -4.1453e+00,
-3.3216e+00, -4.1722e+00],
[-8.1313e-01, -2.6652e+00, -2.1505e+00, ..., -3.9366e+00,
-3.9173e+00, -3.8295e+00],
...,
[-1.4231e+00, -5.7560e+00, -5.2970e+00, ..., -8.6738e+00,
-9.2766e+00, -7.5329e+00],
[-1.2613e+00, -5.8941e+00, -5.1929e+00, ..., -7.6939e+00,
-7.8868e+00, -7.9272e+00],
[-1.6128e+00, -4.9015e+00, -3.3965e+00, ..., -5.0133e+00,
-4.2686e+00, -5.7173e+00]],
[[ 6.4490e-01, 5.3012e-01, -4.2914e-01, ..., 1.2295e+00,
7.6901e-01, 1.7680e+00],
[ 1.4763e+00, 1.2404e+00, 1.3631e-01, ..., 1.2760e+00,
1.7452e-01, 1.9975e+00],
[ 9.8092e-01, 7.8053e-01, -3.2458e-01, ..., 1.0531e+00,
9.0377e-01, 1.0813e+00],
...,
[-2.5803e-01, 6.5053e-01, 8.9608e-01, ..., -6.1317e-01,
-4.4028e-01, 1.2698e+00],
[ 5.0813e-01, 2.0437e+00, 3.0657e+00, ..., 5.0409e-01,
-8.6222e-01, 1.6925e+00],
[-1.3472e-01, -5.7155e-01, 1.1052e+00, ..., 2.2523e+00,
1.9289e+00, 1.8048e+00]],
[[-3.4549e-01, -2.2136e-01, -8.4988e-01, ..., 4.5283e-01,
6.1716e-02, 1.0059e+00],
[-1.4757e+00, -1.3379e+00, -1.8652e+00, ..., -1.3674e-01,
-7.0789e-01, 1.0681e+00],
[-1.9968e+00, -7.1877e-01, -6.8865e-01, ..., 8.9758e-01,
5.3702e-01, 1.4347e+00],
...,
[-2.5281e+00, -1.2882e+00, 1.2347e+00, ..., -3.2317e+00,
-1.6960e+00, 5.8640e-01],
[-2.4231e+00, -9.0292e-01, -1.6386e-02, ..., -4.2436e+00,
-4.4866e+00, -1.8182e+00],
[-3.9711e-01, 9.4110e-01, -4.0457e-01, ..., -1.2048e+00,
-9.6004e-01, -4.8255e-02]]],
[[[ 9.6723e-01, 1.5989e+00, 1.5392e+00, ..., 4.4595e-01,
1.4875e-01, 4.4949e-01],
[-8.8865e-01, -2.3697e+00, -1.6806e+00, ..., 8.3858e-01,
-8.0082e-01, -3.4585e-01],
[ 4.7487e-02, -7.5091e-01, 8.9224e-01, ..., 3.6225e-01,
-2.4022e-01, 6.9328e-01],
...,
[ 4.8739e-01, -4.2268e-01, -2.9746e-01, ..., 5.5003e-01,
-2.3620e+00, -8.5156e-01],
[-1.9112e-01, -1.0820e+00, -1.1190e-01, ..., -1.4891e-01,
-1.2806e+00, -1.1289e+00],
[ 7.6913e-01, -3.2188e-01, -1.8140e-01, ..., -4.0228e-01,
-3.1066e-01, -1.4861e-01]],
[[-4.2299e+00, -8.7929e-01, -2.7661e+00, ..., -1.5283e+00,
-1.9977e+00, -1.7151e+00],
[-4.1588e+00, -2.6744e+00, -2.3018e+00, ..., -3.7952e+00,
-3.3648e+00, -2.1203e+00],
[-2.7248e+00, -2.0762e+00, -1.4549e+00, ..., -4.8011e+00,
-1.8790e+00, -1.4434e+00],
...,
[-8.2810e-01, -9.1321e-01, -1.3049e+00, ..., 1.4008e-01,
-6.5880e-01, -5.7559e-01],
[-2.9178e+00, -5.1156e-01, -7.9666e-01, ..., 1.8746e+00,
-8.8344e-01, -2.9242e-02],
[-2.1339e+00, -1.5516e+00, -1.4582e+00, ..., 1.3786e+00,
6.4187e-02, 8.4885e-01]],
[[ 3.0490e+00, 4.4453e+00, 3.5177e+00, ..., 8.0675e-01,
9.2034e-01, 2.5846e+00],
[ 3.6511e-01, 1.1580e+00, 2.5633e+00, ..., -1.1692e+00,
-8.0689e-01, 2.5877e+00],
[ 8.9288e-01, 1.5768e+00, 1.1056e+00, ..., -4.8658e-01,
-8.9784e-01, 1.8930e+00],
...,
[ 2.8671e-01, 7.2117e-01, 1.4798e+00, ..., -5.4683e+00,
-4.2917e+00, -1.0350e-01],
[ 3.8650e-01, 2.2174e+00, 1.4390e+00, ..., -3.6401e+00,
-3.0904e+00, -2.1581e-01],
[ 1.4625e+00, 1.4364e+00, 1.6858e+00, ..., -1.4918e+00,
-1.1352e+00, 1.3561e-01]],
...,
[[-2.1357e+00, -7.5221e+00, -6.1977e+00, ..., -5.4090e+00,
-3.6256e+00, -4.2243e+00],
[-2.8178e+00, -6.4935e+00, -5.2635e+00, ..., -1.3235e+01,
-5.2579e+00, -3.3022e+00],
[-2.5947e-01, -2.9125e+00, -3.7799e+00, ..., -1.2076e+01,
-4.7715e+00, -1.2189e+00],
...,
[-9.4770e-01, -2.5231e+00, -2.1597e+00, ..., -5.3415e+00,
-2.8103e+00, -4.4447e+00],
[-1.2885e+00, -3.6744e+00, -3.5742e+00, ..., -3.9099e+00,
-3.8617e+00, -1.3164e+00],
[-2.2335e+00, -5.0147e+00, -3.0057e+00, ..., -2.0622e+00,
-3.3061e+00, 4.8589e-01]],
[[-7.3258e-01, -8.8457e-02, -2.6204e-01, ..., -1.1632e+00,
1.7495e+00, 1.2443e+00],
[ 1.8064e+00, 1.1118e+00, 1.1727e+00, ..., -3.0494e+00,
1.4541e+00, 1.8692e+00],
[ 1.2131e+00, -2.6591e-01, 1.6693e+00, ..., -3.4205e+00,
1.2453e+00, 1.1360e+00],
...,
[ 1.3180e+00, -2.9577e-01, 7.3841e-01, ..., 1.4332e+00,
6.5314e-01, -4.5407e-01],
[ 1.8861e+00, 9.4756e-01, 1.7755e+00, ..., 1.3452e+00,
-1.1956e+00, -1.6034e-02],
[ 1.0194e+00, 3.8198e-01, 9.6032e-01, ..., -1.0562e-01,
-1.5959e+00, 1.8633e-01]],
[[-5.0663e-01, 5.1202e-01, -3.3473e-01, ..., -2.2783e-01,
-6.0938e-01, -3.2836e-03],
[-1.1125e+00, -1.3725e+00, -2.1347e+00, ..., -3.2322e+00,
-3.1292e+00, -1.0475e+00],
[-2.0948e+00, -1.0247e+00, -4.4273e-02, ..., -5.2222e+00,
-2.0189e+00, -3.2961e-01],
...,
[-1.7418e+00, -1.5542e+00, -8.3095e-01, ..., 1.5832e+00,
1.5975e-01, -2.1722e+00],
[-1.7503e+00, -1.1482e+00, -9.4286e-01, ..., 2.2666e+00,
-1.3181e+00, -2.1021e+00],
[-3.6276e-01, 3.7366e-01, -3.3995e-01, ..., 3.9482e-01,
-3.4502e+00, -1.2410e+00]]]], device='cuda:0'),
tensor([[[[ 5.7503e-01, 8.5640e-01, 5.5499e-01, ..., 2.5948e-01,
5.1307e-01, 2.1788e-02],
[-2.7893e-01, -8.4740e-01, -4.7677e-01, ..., 6.6507e-01,
5.2605e-01, -1.8879e-01],
[-2.2563e-01, -2.2916e-03, 4.8053e-01, ..., -1.9091e-01,
3.9393e-01, 8.1077e-01],
...,
[ 9.0406e-01, 8.1777e-01, 1.4894e+00, ..., -1.1348e+00,
-1.3207e+00, -3.7752e-01],
[ 2.1283e-01, -1.3151e+00, -7.0645e-02, ..., -1.8451e+00,
-1.6164e+00, -1.0985e+00],
[ 6.8481e-01, -4.6586e-01, -1.0473e+00, ..., 3.2951e-01,
9.8337e-01, 6.0669e-01]],
[[-2.4543e+00, -6.1711e-01, -1.8594e+00, ..., -1.7941e+00,
-2.4556e+00, -1.5542e+00],
[-2.6628e+00, -6.1761e-01, -1.4695e+00, ..., -1.0875e+00,
-2.4973e+00, -1.5698e+00],
[-2.6538e+00, -1.3974e+00, -1.5471e+00, ..., -2.0837e+00,
-2.7462e+00, -1.1989e+00],
...,
[ 5.7629e-01, 1.8724e+00, 1.2633e-01, ..., -2.7286e+00,
-3.8080e+00, -1.3589e+00],
[-2.2661e+00, -8.8557e-01, -1.9885e+00, ..., -3.5639e+00,
-2.9840e+00, -8.4374e-01],
[-1.1094e+00, -1.6191e+00, -2.1269e+00, ..., -1.9316e+00,
-2.1709e+00, -4.4877e-01]],
[[ 1.9658e+00, 2.5612e+00, 1.6317e+00, ..., 1.3231e+00,
6.7120e-01, 2.1215e+00],
[-2.0037e-01, 4.3011e-01, 6.0913e-01, ..., -1.0061e-01,
-3.8257e-02, 2.2153e+00],
[-2.5583e-02, 2.4838e-01, 3.3532e-01, ..., -1.0700e+00,
3.2535e-01, 1.9329e+00],
...,
[ 4.6898e-01, -3.1961e-01, -5.6036e-01, ..., -1.4657e+00,
-1.4010e+00, 8.3639e-01],
[ 8.9159e-01, 7.3465e-01, -1.8114e-01, ..., -1.7396e+00,
-1.1527e+00, 8.2084e-01],
[ 1.9043e+00, 1.1135e+00, 1.8021e+00, ..., -3.9418e-01,
-1.4499e+00, 4.0308e-02]],
...,
[[-2.0870e+00, -5.0575e+00, -3.2702e+00, ..., -2.1953e+00,
-1.5330e+00, -2.8734e+00],
[-2.5747e+00, -5.3998e+00, -3.1741e+00, ..., -2.7796e+00,
-2.0676e+00, -2.7571e+00],
[-1.5001e+00, -5.2225e+00, -3.0857e+00, ..., -1.4939e+00,
-6.9256e-01, -1.8663e+00],
...,
[-1.9363e+00, -4.7531e+00, -2.2485e+00, ..., -4.4001e+00,
-9.5832e-02, 5.4916e-01],
[-2.5863e+00, -6.7626e+00, -4.2160e+00, ..., -2.0344e+00,
4.2581e-01, -1.0599e+00],
[-2.7582e+00, -7.2266e+00, -3.0428e+00, ..., -7.7986e-01,
5.6581e-01, -2.8548e+00]],
[[ 9.4031e-01, 1.6218e+00, 8.1262e-01, ..., 1.0249e+00,
1.6085e+00, 1.4402e+00],
[ 1.1972e+00, 1.0482e+00, 8.6288e-01, ..., 6.3747e-01,
4.4120e-01, 1.8666e+00],
[ 7.0359e-01, 2.3714e-01, 4.0623e-01, ..., -4.3239e-01,
-7.7166e-01, 1.3298e+00],
...,
[ 2.0596e+00, 2.4216e-01, -3.9417e-01, ..., -5.6058e-01,
4.8727e-02, 6.4849e-01],
[ 2.0529e+00, 1.0196e+00, 2.5013e-02, ..., -1.6595e+00,
-1.1384e+00, 1.3892e+00],
[ 9.6807e-01, 3.9169e-01, 7.2277e-01, ..., -2.8177e-01,
1.7583e-01, 6.2938e-01]],
[[-9.3521e-02, -6.0889e-01, -1.2081e+00, ..., -4.9619e-01,
-8.6844e-01, -8.6346e-02],
[-9.0092e-01, -2.4608e+00, -2.5948e+00, ..., -6.1239e-01,
-1.2287e+00, -7.9965e-01],
[-1.5386e+00, -1.3777e+00, -6.6462e-01, ..., -2.4861e-01,
-3.9309e-01, -2.6903e-01],
...,
[-2.0714e+00, -8.6337e-01, 2.9044e-01, ..., -2.1681e+00,
-1.9414e+00, 3.4408e-01],
[-2.1442e+00, -1.9902e+00, -1.0709e+00, ..., -3.0941e+00,
-5.6406e-01, 8.2880e-01],
[-4.5403e-01, -1.9720e-01, -5.9580e-01, ..., 8.9790e-01,
4.8460e-01, -7.6496e-02]]],
[[[ 8.4729e-01, 1.4955e+00, 1.1824e+00, ..., 1.3568e+00,
1.0515e+00, 9.6943e-01],
[-1.0114e+00, -1.9138e+00, -1.1336e+00, ..., -2.5042e-01,
-6.0100e-01, -1.6512e+00],
[-5.6984e-02, -6.1065e-01, 3.5230e-01, ..., -1.0588e-01,
8.4026e-01, 8.0227e-01],
...,
[-3.3233e-02, -1.1748e+00, -9.1746e-01, ..., -4.5398e-01,
-2.3016e-01, 5.8004e-01],
[-5.9997e-01, -1.8430e+00, -1.2363e+00, ..., -5.7234e-01,
-3.5617e-01, -1.2050e-01],
[-4.0407e-01, -1.7231e+00, -1.5905e+00, ..., -1.1177e+00,
-1.6059e+00, -3.0091e-01]],
[[-3.8107e+00, -8.0336e-01, -1.5986e+00, ..., -1.5529e+00,
-2.5138e+00, -1.2266e+00],
[-4.0256e+00, -2.8301e+00, -1.8673e+00, ..., -2.1952e+00,
-3.2457e+00, -9.3630e-01],
[-2.4230e+00, -1.8739e+00, -2.5000e+00, ..., -2.1365e+00,
-2.9406e+00, -1.0770e+00],
...,
[-1.6493e+00, -1.7599e+00, -2.3867e+00, ..., -3.0990e+00,
-2.2302e+00, -7.1969e-01],
[-3.0098e+00, -1.8158e+00, -2.8350e+00, ..., -2.3430e+00,
-2.0801e+00, -1.2261e+00],
[-2.2074e+00, -1.9234e+00, -2.6189e+00, ..., -1.9823e+00,
-1.6782e+00, -8.7420e-01]],
[[ 2.9144e+00, 4.2382e+00, 3.3788e+00, ..., 2.8516e+00,
1.0829e+00, 3.0578e+00],
[ 2.3144e-01, 1.1892e+00, 2.3528e+00, ..., 1.2693e+00,
-7.5511e-01, 2.7640e+00],
[ 1.0814e+00, 1.7384e+00, 2.1135e+00, ..., 6.2965e-01,
-2.9631e-02, 2.0687e+00],
...,
[ 9.7313e-01, 1.1333e+00, 1.1948e+00, ..., 1.2838e+00,
1.5909e+00, 2.5870e+00],
[ 6.1976e-01, 1.7814e+00, 1.1259e+00, ..., -3.9475e-02,
9.4595e-01, 1.7381e+00],
[ 1.5369e+00, 1.5973e+00, 1.8126e+00, ..., 1.2486e+00,
1.7028e+00, 1.6981e+00]],
...,
[[-1.8619e+00, -7.2551e+00, -5.4504e+00, ..., -3.9199e+00,
-3.2755e+00, -5.6024e+00],
[-2.4278e+00, -7.0794e+00, -4.3713e+00, ..., -3.5044e+00,
-2.8181e+00, -4.3385e+00],
[-9.9667e-01, -4.4741e+00, -3.7246e+00, ..., -3.1591e+00,
-3.1162e+00, -2.6215e+00],
...,
[-2.5901e+00, -5.3780e+00, -4.3846e+00, ..., -7.8420e+00,
-7.3697e+00, -7.4522e+00],
[-2.6574e+00, -5.6080e+00, -5.4810e+00, ..., -6.2644e+00,
-5.7090e+00, -6.1158e+00],
[-1.4425e+00, -3.9058e+00, -2.8109e+00, ..., -3.2941e+00,
-2.7836e+00, -3.1456e+00]],
[[-2.0768e-01, 3.0275e-01, 5.8489e-01, ..., 6.8021e-01,
8.5706e-01, 1.4359e+00],
[ 1.6252e+00, 1.1040e+00, 1.8403e+00, ..., 1.8523e+00,
2.5400e+00, 3.1036e+00],
[ 1.1787e+00, 3.4760e-01, 6.8830e-01, ..., 1.0858e+00,
1.4987e+00, 2.3691e+00],
...,
[ 1.3286e+00, 2.5614e-01, 9.6970e-02, ..., 2.8778e+00,
2.2892e+00, 2.1832e+00],
[ 1.8474e+00, 1.6149e+00, 1.0165e+00, ..., 1.5344e+00,
1.3033e+00, 1.8799e+00],
[ 5.1496e-01, 7.6485e-01, 6.5458e-01, ..., 5.2099e-01,
7.2041e-01, 1.6093e+00]],
[[-5.6787e-01, 2.8707e-01, 1.3374e-01, ..., 5.9488e-01,
4.9753e-01, 1.3638e+00],
[-1.3658e+00, -1.7400e+00, -1.5605e+00, ..., -4.9676e-01,
-4.5598e-01, 2.2628e-01],
[-2.2691e+00, -9.7626e-01, 1.2842e-01, ..., 2.4094e-01,
1.5272e-01, 7.3488e-01],
...,
[-1.9513e+00, -1.7840e+00, -1.3879e+00, ..., -2.0049e+00,
-1.4889e+00, -3.6318e-01],
[-1.5639e+00, -2.0124e+00, -1.7730e+00, ..., -1.1274e+00,
-1.2977e+00, -7.9608e-01],
[-8.0369e-01, -9.4151e-01, -1.1681e+00, ..., -5.6480e-01,
-8.9718e-01, -1.2848e-01]]],
[[[ 8.9905e-01, 1.4873e+00, 1.2130e+00, ..., 1.4396e+00,
1.0797e+00, 9.6011e-01],
[-1.0098e+00, -2.4055e+00, -1.2711e+00, ..., -3.8349e-01,
-6.1503e-01, -1.5005e+00],
[ 1.5897e-02, -9.1984e-01, 2.9012e-01, ..., -5.8985e-01,
6.3215e-01, 7.6003e-01],
...,
[ 4.6626e-01, -2.1586e-01, -8.8903e-04, ..., 3.5631e-01,
2.2104e-01, 6.7790e-01],
[ 3.7652e-01, -7.7837e-01, -8.3657e-01, ..., 7.7153e-01,
6.0670e-01, -4.6826e-02],
[ 6.1398e-01, 1.7961e-01, -5.5110e-01, ..., 4.1724e-01,
-9.6817e-02, 6.8778e-01]],
[[-4.2677e+00, -6.2746e-01, -1.6485e+00, ..., -1.4567e+00,
-2.3646e+00, -1.1209e+00],
[-4.2544e+00, -2.2816e+00, -1.7330e+00, ..., -2.1092e+00,
-3.1883e+00, -8.4994e-01],
[-2.6104e+00, -1.2978e+00, -1.7247e+00, ..., -2.6413e+00,
-2.9478e+00, -1.0549e+00],
...,
[-1.0151e+00, -1.1101e+00, -3.5230e+00, ..., -1.6564e+00,
-1.3248e+00, -1.6022e-01],
[-1.3886e+00, -4.4225e-01, -2.1081e+00, ..., -3.3467e-01,
-1.2320e+00, -1.7529e-01],
[ 4.9087e-01, -1.5114e-01, -1.2490e+00, ..., 4.2919e-01,
3.4030e-02, 3.8804e-01]],
[[ 3.3672e+00, 4.5010e+00, 3.5709e+00, ..., 2.7523e+00,
1.0865e+00, 2.9347e+00],
[ 8.1972e-01, 1.0271e+00, 2.2342e+00, ..., 1.1409e+00,
-9.5075e-01, 2.6756e+00],
[ 1.6469e+00, 1.4582e+00, 1.7468e+00, ..., 1.5483e+00,
-6.4560e-02, 2.1042e+00],
...,
[ 4.9852e-01, 5.5558e-01, 2.1798e+00, ..., -3.7661e-01,
2.3994e-01, 1.5396e+00],
[ 1.8078e-01, 1.1926e+00, 2.4890e+00, ..., -4.0040e-01,
1.8734e-01, 1.2539e+00],
[ 1.3926e+00, 1.0672e+00, 2.0915e+00, ..., 4.1023e-01,
3.9796e-01, 1.1497e+00]],
...,
[[-2.6716e+00, -7.9015e+00, -5.8621e+00, ..., -4.1147e+00,
-3.1381e+00, -5.3463e+00],
[-3.2417e+00, -7.1932e+00, -4.5375e+00, ..., -3.2941e+00,
-2.5563e+00, -4.1618e+00],
[-1.4444e+00, -4.9499e+00, -3.8676e+00, ..., -1.5827e+00,
-2.9227e+00, -2.4214e+00],
...,
[-2.2968e+00, -5.6320e+00, -4.2712e+00, ..., -4.6262e+00,
-3.3639e+00, -3.5326e+00],
[-2.4573e+00, -6.6289e+00, -5.3657e+00, ..., -6.1035e+00,
-4.1880e+00, -4.9600e+00],
[-2.3656e+00, -6.5527e+00, -4.5899e+00, ..., -5.1030e+00,
-3.1875e+00, -4.3122e+00]],
[[-3.0705e-01, 4.6899e-01, 4.4588e-01, ..., 7.4445e-01,
8.6786e-01, 1.4341e+00],
[ 1.9207e+00, 1.4855e+00, 1.7020e+00, ..., 1.7246e+00,
2.3662e+00, 2.9831e+00],
[ 1.3182e+00, 4.3890e-01, 1.7307e-01, ..., 1.6990e+00,
1.7587e+00, 2.3870e+00],
...,
[ 1.1120e+00, -1.2702e-01, -1.2468e+00, ..., 1.9592e+00,
7.8673e-01, 1.2318e+00],
[ 1.6634e+00, 7.6884e-01, -4.1476e-01, ..., 1.9685e+00,
5.1422e-01, 2.0088e+00],
[ 1.1739e+00, 8.4244e-01, 5.3402e-01, ..., 1.6938e+00,
1.2272e+00, 1.3846e+00]],
[[-6.9930e-01, 3.1099e-01, -6.2041e-02, ..., 6.4447e-01,
5.3600e-01, 1.3519e+00],
[-1.6586e+00, -1.8147e+00, -1.9375e+00, ..., -8.5794e-01,
-5.8464e-01, 2.2372e-01],
[-2.3961e+00, -8.5878e-01, -1.5787e-01, ..., -1.8366e-03,
1.2600e-01, 7.4613e-01],
...,
[-1.5950e+00, -1.6671e+00, -2.3281e+00, ..., -1.3022e+00,
-1.0065e+00, -7.6256e-01],
[-1.2971e+00, -1.6418e+00, -1.7876e+00, ..., -1.5190e+00,
-1.1824e+00, -9.2129e-02],
[ 1.3275e-01, 9.9090e-02, -3.7364e-01, ..., -2.1133e-01,
-4.6645e-02, 7.9094e-01]]],
[[[ 8.4015e-01, 1.4776e+00, 1.2051e+00, ..., 1.4356e+00,
1.1309e+00, 9.4612e-01],
[-9.7434e-01, -1.9690e+00, -1.1231e+00, ..., -3.1420e-01,
-5.0534e-01, -1.5230e+00],
[-1.4445e-01, -6.8228e-01, -2.1801e-02, ..., -1.3819e-01,
6.6753e-01, 7.7147e-01],
...,
[ 4.0856e-01, -1.0534e-01, 3.4484e-01, ..., 5.3292e-01,
2.3736e-01, 6.3810e-01],
[ 3.3371e-01, -1.1540e+00, -1.6813e+00, ..., 6.8015e-01,
5.9404e-01, -3.8096e-02],
[ 5.2808e-01, -4.3919e-02, -1.5290e-01, ..., 2.0452e-01,
-1.9415e-01, 6.1965e-01]],
[[-3.7140e+00, -8.2037e-01, -1.6143e+00, ..., -1.5366e+00,
-2.4717e+00, -1.2267e+00],
[-3.8843e+00, -2.8439e+00, -1.8453e+00, ..., -2.2567e+00,
-3.1093e+00, -8.8475e-01],
[-2.3714e+00, -1.7352e+00, -1.5850e+00, ..., -2.6383e+00,
-2.8530e+00, -1.0800e+00],
...,
[-1.0156e+00, -1.2884e+00, -3.1128e+00, ..., -1.5451e+00,
-1.5479e+00, -2.0785e-01],
[-1.8411e+00, -7.0153e-01, -1.4244e+00, ..., -7.6037e-01,
-1.2919e+00, -1.5919e-01],
[ 3.1389e-01, -1.9792e-01, -2.4643e-01, ..., 2.3142e-01,
-2.1857e-02, 4.0636e-01]],
[[ 2.8337e+00, 4.1956e+00, 3.3612e+00, ..., 2.9244e+00,
1.0832e+00, 3.0166e+00],
[ 1.6707e-01, 1.1254e+00, 2.3282e+00, ..., 1.3484e+00,
-7.1241e-01, 2.8190e+00],
[ 9.4386e-01, 1.5742e+00, 1.7497e+00, ..., 5.4789e-01,
7.0468e-03, 2.1661e+00],
...,
[ 4.1428e-01, 8.2811e-01, 1.0871e+00, ..., -3.8603e-01,
2.5299e-01, 1.5562e+00],
[ 7.8697e-02, 1.8391e+00, 1.3458e+00, ..., -2.1329e-01,
1.9762e-01, 1.2339e+00],
[ 1.3396e+00, 1.4987e+00, 7.0727e-01, ..., 8.1635e-01,
4.3957e-01, 1.1365e+00]],
...,
[[-1.7514e+00, -7.0640e+00, -5.4195e+00, ..., -4.0795e+00,
-3.3243e+00, -5.5398e+00],
[-2.3168e+00, -6.9266e+00, -4.1994e+00, ..., -3.6184e+00,
-2.8317e+00, -4.4050e+00],
[-1.0696e+00, -4.6267e+00, -3.6713e+00, ..., -3.2133e+00,
-3.1141e+00, -2.3887e+00],
...,
[-2.5964e+00, -6.0078e+00, -4.6422e+00, ..., -3.5755e+00,
-3.5461e+00, -3.4705e+00],
[-2.8021e+00, -7.1636e+00, -5.3194e+00, ..., -5.0366e+00,
-4.3882e+00, -4.9244e+00],
[-2.7724e+00, -7.3986e+00, -5.0423e+00, ..., -4.1466e+00,
-3.2877e+00, -4.1818e+00]],
[[-1.5842e-01, 2.6748e-01, 5.4505e-01, ..., 6.5039e-01,
8.5118e-01, 1.3991e+00],
[ 1.6118e+00, 1.0633e+00, 1.8276e+00, ..., 1.8697e+00,
2.5675e+00, 3.0655e+00],
[ 1.0677e+00, 2.2864e-01, 8.4008e-01, ..., 1.0646e+00,
1.6478e+00, 2.4170e+00],
...,
[ 1.0473e+00, -3.0327e-01, -1.0796e+00, ..., 8.4194e-01,
6.6036e-01, 1.2665e+00],
[ 1.5712e+00, 6.2796e-01, 5.0612e-01, ..., 9.4651e-01,
5.4933e-01, 2.0399e+00],
[ 1.1169e+00, 6.5061e-01, 8.2950e-01, ..., 1.2034e+00,
1.2286e+00, 1.4023e+00]],
[[-5.8660e-01, 2.4837e-01, 1.5668e-01, ..., 6.1339e-01,
5.2429e-01, 1.3514e+00],
[-1.3647e+00, -1.7978e+00, -1.5636e+00, ..., -6.4533e-01,
-4.5186e-01, 2.4561e-01],
[-2.1080e+00, -7.7963e-01, 3.2731e-01, ..., 2.8951e-01,
2.7812e-01, 7.4922e-01],
...,
[-1.5628e+00, -2.1243e+00, -1.2547e+00, ..., -7.2861e-01,
-1.1577e+00, -7.0106e-01],
[-1.2992e+00, -2.2047e+00, -1.7013e+00, ..., -7.7162e-01,
-1.3459e+00, 2.6589e-03],
[ 7.0819e-02, 5.1282e-02, -7.4458e-02, ..., 2.7946e-01,
-1.3198e-01, 7.9798e-01]]],
[[[ 1.1359e+00, 1.7568e+00, -1.7789e+00, ..., -6.3680e+00,
-5.8891e+00, -1.3356e+00],
[-5.3060e-01, 1.2237e-01, -2.9386e+00, ..., -5.8232e+00,
-3.2845e+00, -6.8109e-01],
[-3.4829e-01, -1.3551e+00, -2.9437e+00, ..., -1.2131e+00,
1.4526e+00, 1.8420e+00],
...,
[ 1.4313e+00, 3.7452e+00, -4.1148e-01, ..., -2.8837e+00,
-2.3698e+00, -3.7934e-01],
[ 3.2697e-01, 5.0435e-01, -8.9652e-01, ..., 1.6422e+00,
5.0233e-01, -1.8097e-01],
[ 7.5704e-01, 7.5006e-01, -2.5910e+00, ..., 2.2847e-01,
-7.4053e-01, 1.1110e-01]],
[[-2.5649e+00, -9.3491e-01, 5.7262e-02, ..., -8.5373e+00,
-4.3361e+00, -2.1849e+00],
[-2.6191e+00, -2.1163e+00, -3.9249e-01, ..., -1.0018e+01,
-7.1006e+00, -5.2574e+00],
[-4.6612e+00, -3.1557e+00, -2.2622e+00, ..., -1.5675e+01,
-1.1707e+01, -6.6096e+00],
...,
[-4.9667e+00, 1.0045e+00, 1.6800e+00, ..., -1.0754e+01,
-7.1239e+00, -2.6453e+00],
[-3.6618e+00, 2.5889e+00, 4.8113e+00, ..., -9.0611e-01,
-4.7483e-01, -9.1137e-01],
[-3.7938e+00, -3.2514e+00, -2.2855e+00, ..., -5.8200e-01,
-3.2622e-01, -1.6577e-01]],
[[ 3.1732e+00, 5.7526e-01, -9.3066e-01, ..., -1.3700e+00,
-2.6538e+00, 3.2680e-01],
[ 1.7113e+00, -3.4337e+00, -4.7774e+00, ..., -1.2510e+00,
-5.0974e+00, 1.0571e+00],
[ 1.3106e+00, -1.2278e+00, -6.2022e+00, ..., -8.2859e+00,
-4.9972e+00, -4.1051e-01],
...,
[-1.0352e+00, -1.0338e+01, -2.6291e+00, ..., -8.5149e+00,
-6.1300e+00, 7.4919e-01],
[-1.5687e+00, -3.8008e+00, -6.9644e-01, ..., 1.5240e+00,
1.1929e+00, 1.2214e+00],
[ 1.8425e+00, 7.0549e-01, -2.2030e+00, ..., 6.8915e-01,
1.2382e+00, 1.5362e+00]],
...,
[[-2.3939e+00, -5.7997e+00, -2.3512e+00, ..., -2.8391e+00,
2.2914e+00, -2.3167e+00],
[-3.0163e+00, -5.2288e+00, -1.2902e+00, ..., 1.2177e+00,
-2.3424e+00, -1.0787e+01],
[-1.6847e+00, -3.5491e+00, -4.8416e+00, ..., -2.6472e+00,
-1.0002e+01, -1.3681e+01],
...,
[-2.5504e+00, -7.8076e+00, -7.4813e+00, ..., -1.1557e+01,
-9.3418e+00, -5.8045e+00],
[-1.8873e+00, -8.0864e+00, -6.9509e+00, ..., -9.9650e+00,
-9.6283e+00, -8.6768e+00],
[-2.3436e+00, -6.3600e+00, -4.1486e+00, ..., -5.1785e+00,
-3.5488e+00, -3.0968e+00]],
[[ 9.4192e-01, 2.0257e+00, 1.0264e+00, ..., -5.6733e+00,
-3.3960e+00, -1.3129e+00],
[ 1.6213e+00, 1.4832e+00, -6.5253e-01, ..., -1.3280e+01,
-6.7110e+00, -3.5881e+00],
[-7.4100e-01, -2.9097e+00, 6.1499e-01, ..., -6.5375e+00,
-4.2636e+00, -2.9147e+00],
...,
[-3.1445e-01, 4.0678e-01, -2.3142e+00, ..., -3.4901e+00,
-1.5248e+00, 2.5084e-01],
[ 2.5552e+00, 1.9612e+00, 6.0763e-01, ..., 4.3863e+00,
3.0007e+00, 2.9603e+00],
[-5.8031e-02, -2.0590e-01, 3.7904e-01, ..., 1.5278e+00,
9.6542e-01, 1.5823e+00]],
[[-7.2975e-01, -6.7112e-01, 3.7492e-01, ..., -6.6302e+00,
-2.6801e+00, 1.1438e-01],
[-5.8154e-01, -3.9991e+00, -2.8112e+00, ..., -7.3989e+00,
7.3320e-01, 1.6225e+00],
[-2.2609e+00, -1.3766e+00, -1.8102e+00, ..., 3.9587e-01,
8.3146e-01, -1.1972e+00],
...,
[-1.4594e+00, 1.3397e+00, -1.4519e+00, ..., -9.0655e+00,
-6.9312e+00, -2.7479e+00],
[-2.1003e+00, 4.7753e-01, -1.2460e+00, ..., -4.1374e+00,
-2.8410e+00, 1.5065e-01],
[ 2.9296e-03, 1.1820e+00, -1.3581e-01, ..., -1.1358e+00,
-6.5718e-01, 4.1337e-02]]]], device='cuda:0'),
tensor([[[[ 3.1603e-01, 1.1249e+00, 1.4855e-01, ..., -2.1019e+00,
-1.1421e+00, -2.2626e-01],
[-1.4296e+00, -4.9337e-01, 1.5227e+00, ..., 2.3636e+00,
-2.3167e+00, -2.0291e+00],
[-2.2601e+00, -3.1722e+00, -9.2405e-01, ..., 3.3034e+00,
7.8261e-01, -7.2672e-01],
...,
[ 4.8628e-01, -2.4667e-01, -8.0689e-01, ..., -2.8823e-01,
2.4918e-01, 2.9852e-01],
[-6.5232e-02, -9.2537e-01, -5.8996e-01, ..., -2.5454e-01,
2.4024e-01, -1.5400e-01],
[ 7.9751e-01, -4.0856e-01, -3.3200e-01, ..., -4.4451e-01,
-8.4457e-01, 7.4791e-02]],
[[-6.8844e+00, -4.0540e+00, -3.1724e+00, ..., -4.6398e+00,
-3.3002e+00, -1.7932e+00],
[-2.8506e+00, -1.5875e+00, -4.4097e+00, ..., -7.5252e+00,
-4.9211e+00, -2.6494e+00],
[-4.8487e+00, -6.5395e+00, -8.8418e+00, ..., -9.3459e+00,
-6.1625e+00, -2.0913e+00],
...,
[-1.2279e+00, -8.0550e-01, -1.3003e+00, ..., -5.2539e+00,
-2.4673e-02, -2.9186e-01],
[-3.2397e+00, -7.6187e-01, -1.2503e+00, ..., -2.0852e+00,
-1.3534e-01, -8.0098e-01],
[-2.5831e+00, -1.8336e+00, -1.2903e+00, ..., -8.6982e-01,
-4.6974e-01, -3.1786e-01]],
[[ 3.1306e+00, 2.9016e+00, 3.3943e+00, ..., 6.2311e+00,
2.3049e+00, 2.7417e+00],
[-2.6720e+00, -2.3606e+00, 8.5510e-02, ..., -2.4581e+00,
2.7432e+00, 3.0105e+00],
[-2.9185e+00, -4.4625e+00, -2.8600e+00, ..., -6.8673e+00,
-4.3578e-01, 1.4748e+00],
...,
[-2.1638e-01, 9.8006e-01, 1.4595e+00, ..., 1.9290e+00,
3.0708e-01, 1.5654e+00],
[ 5.1421e-02, 2.2145e+00, 1.5257e+00, ..., 7.3092e-01,
1.9775e-01, 1.5047e+00],
[ 1.2945e+00, 1.3259e+00, 1.6243e+00, ..., 1.3002e-01,
1.4175e+00, 1.4966e+00]],
...,
[[-1.4508e+00, -6.4684e+00, -5.0639e+00, ..., -6.8713e+00,
-5.5577e+00, -5.0136e+00],
[-3.9245e+00, -9.4720e+00, -7.8109e+00, ..., -1.9206e+01,
-1.5923e+01, -7.0752e+00],
[-3.3381e+00, -2.4597e+00, -1.4687e+00, ..., -2.0447e+01,
-2.6200e+01, -1.0082e+01],
...,
[-7.9493e-01, -2.1002e+00, -2.0984e+00, ..., -1.2216e+01,
-8.9873e+00, -4.3420e+00],
[-1.2111e+00, -3.8641e+00, -3.7857e+00, ..., -7.8314e+00,
-7.8150e+00, -4.6110e+00],
[-2.1478e+00, -5.3137e+00, -3.3673e+00, ..., -5.9730e+00,
-3.0594e+00, -2.7202e+00]],
[[-4.1608e+00, -3.0274e+00, -8.0354e-02, ..., 1.3900e+00,
2.9447e+00, 1.8575e+00],
[ 2.4414e+00, -2.2079e+00, -1.4878e+00, ..., -3.5637e+00,
3.3654e+00, 3.2885e+00],
[ 8.7672e-01, -4.1026e+00, -1.0451e+01, ..., -6.9109e+00,
1.7653e+00, 4.2343e+00],
...,
[ 1.2324e+00, -2.8046e-01, 8.0129e-01, ..., 5.3451e-01,
3.1668e+00, 1.4592e+00],
[ 1.7556e+00, 5.8333e-01, 1.4745e+00, ..., 3.5188e+00,
1.9282e+00, 1.8507e+00],
[ 9.1606e-01, 7.9854e-02, 1.0153e+00, ..., 1.3434e+00,
7.2369e-01, 1.5372e+00]],
[[-1.3612e+00, 7.4142e-01, -3.8714e-01, ..., -3.5294e+00,
-2.0943e+00, 4.5832e-01],
[-1.8417e+00, 1.9897e+00, 2.2173e+00, ..., 2.7248e+00,
-5.2309e+00, -7.9891e-01],
[-2.0132e+00, -4.2820e-01, 1.6856e+00, ..., 6.2738e+00,
-7.4711e+00, -2.4337e+00],
...,
[-1.5831e+00, -1.3326e+00, -1.1562e+00, ..., -8.5505e+00,
-3.5283e+00, -8.7665e-01],
[-1.5420e+00, -7.8272e-01, -1.0910e+00, ..., -1.5220e+00,
-1.3414e+00, -5.4788e-01],
[-2.8197e-01, 3.2700e-01, -2.6834e-01, ..., 4.1809e-01,
-6.5744e-01, 8.3981e-02]]],
[[[ 9.6488e-01, 1.6751e+00, 1.2283e+00, ..., 8.1586e-01,
1.0038e+00, 4.8170e-01],
[-7.7282e-01, -1.7120e+00, -7.9498e-01, ..., -2.8798e-01,
-1.9957e-01, -3.5484e-01],
[-8.4620e-02, -7.3435e-01, -5.3170e-02, ..., -7.6260e-01,
-2.0145e-01, 6.2076e-01],
...,
[ 5.2816e-01, -6.4672e-02, 8.8373e-01, ..., 2.0171e-01,
5.6103e-02, 3.7984e-01],
[ 4.3432e-01, -1.1614e+00, -1.2774e-01, ..., -5.2436e-02,
7.6688e-02, -1.0613e-01],
[ 1.0145e+00, -1.7653e-01, 4.3764e-01, ..., -2.9071e-01,
-9.0594e-01, 1.6915e-01]],
[[-3.6298e+00, -1.7150e+00, -2.3239e+00, ..., -1.4074e+00,
-1.9913e+00, -1.7724e+00],
[-3.7942e+00, -3.7055e+00, -2.5850e+00, ..., -1.1910e+00,
-2.1786e+00, -2.1415e+00],
[-2.6189e+00, -1.9425e+00, -8.2493e-01, ..., -1.4450e+00,
-1.8892e+00, -1.4376e+00],
...,
[ 1.5310e-01, 8.6339e-01, -5.2496e-01, ..., -1.9506e+00,
-1.2338e+00, -8.2474e-01],
[-1.7692e+00, -6.0331e-01, -2.3775e+00, ..., -6.3438e-01,
-1.1446e+00, -1.2163e+00],
[-2.0443e+00, -9.4667e-01, -1.2793e+00, ..., -6.3452e-01,
-7.6057e-01, -5.1153e-01]],
[[ 2.5446e+00, 3.6961e+00, 3.0965e+00, ..., 1.7042e+00,
7.9499e-01, 2.5473e+00],
[ 1.0539e-01, 1.0767e+00, 2.4057e+00, ..., 1.3790e+00,
6.3372e-01, 2.5924e+00],
[ 1.1223e+00, 1.3373e+00, 8.6864e-01, ..., 3.2454e-01,
9.2708e-01, 1.7842e+00],
...,
[ 3.3629e-01, -3.2910e-01, -8.0686e-01, ..., 1.2092e-01,
5.6008e-01, 1.4056e+00],
[ 4.7544e-01, 5.1352e-01, -5.3480e-01, ..., -1.5974e-01,
8.1967e-01, 1.1515e+00],
[ 1.8460e+00, 1.2865e+00, 1.7786e+00, ..., 5.7924e-01,
1.2979e+00, 1.3166e+00]],
...,
[[-1.3597e+00, -6.6573e+00, -5.6969e+00, ..., -5.1297e+00,
-3.5885e+00, -4.2926e+00],
[-2.0575e+00, -6.2041e+00, -4.3162e+00, ..., -5.1302e+00,
-3.7409e+00, -3.1714e+00],
[-2.6474e-01, -2.6951e+00, -3.0387e+00, ..., -3.9175e+00,
-2.5927e+00, -1.3788e+00],
...,
[-2.5490e-01, -1.6248e+00, -1.1312e+00, ..., -6.6043e+00,
-5.0246e+00, -3.6295e+00],
[-1.7224e+00, -4.5215e+00, -3.6469e+00, ..., -5.1140e+00,
-4.2100e+00, -4.2620e+00],
[-1.9802e+00, -5.1443e+00, -3.3792e+00, ..., -3.4132e+00,
-2.4866e+00, -2.3947e+00]],
[[-6.3146e-01, -3.2311e-01, 1.1240e-01, ..., 1.5326e+00,
1.7536e+00, 1.1879e+00],
[ 1.4684e+00, 4.4393e-01, 8.7405e-01, ..., 1.5421e+00,
1.1840e+00, 1.8106e+00],
[ 7.9648e-01, -2.1221e-01, 1.1203e+00, ..., 1.4957e+00,
4.4543e-01, 1.2890e+00],
...,
[ 1.0189e+00, 3.0571e-01, 2.3689e-02, ..., 2.1520e+00,
1.0040e+00, 1.1957e+00],
[ 1.6990e+00, 4.1678e-01, 1.8477e-01, ..., 1.3322e+00,
7.0953e-01, 1.4681e+00],
[ 8.1433e-01, 6.2696e-01, 1.0905e+00, ..., 7.5270e-01,
6.6885e-01, 1.4433e+00]],
[[-5.1343e-01, 5.0170e-01, 3.0857e-01, ..., -3.3158e-01,
-5.8170e-01, -3.2283e-02],
[-1.2071e+00, -1.2903e+00, -1.1425e+00, ..., -1.2623e+00,
-1.4884e+00, -1.1155e+00],
[-2.1232e+00, -4.8555e-01, 8.5589e-01, ..., -8.8454e-01,
-1.2300e+00, -1.6822e-01],
...,
[-7.6933e-01, -3.7543e-01, -7.0640e-01, ..., -1.5495e+00,
-1.3676e+00, -4.5696e-01],
[-1.0981e+00, -1.1717e+00, -1.9833e+00, ..., -8.2781e-01,
-1.6733e+00, -6.8509e-01],
[-7.2348e-01, 4.5820e-01, 8.8707e-02, ..., -2.1756e-01,
-8.2978e-01, 9.0864e-03]]],
[[[ 9.6258e-01, 1.5724e+00, 1.1313e+00, ..., 7.7903e-01,
9.6546e-01, 4.1872e-01],
[-9.2927e-01, -1.9631e+00, -9.0873e-01, ..., -1.9114e-01,
-1.0400e-01, -2.3759e-01],
[ 9.8636e-03, -8.1485e-01, -2.0103e-01, ..., -7.8879e-01,
-1.9462e-01, 6.5707e-01],
...,
[ 3.7434e-01, -2.5919e-01, -1.1996e-01, ..., 2.3566e-02,
6.4456e-02, 4.6699e-01],
[-2.2978e-01, -1.0214e+00, -4.5270e-02, ..., 2.8830e-02,
1.7377e-01, -9.8529e-02],
[ 7.2134e-01, -3.1379e-01, -1.6496e-01, ..., -2.4182e-01,
-8.3665e-01, 1.9105e-01]],
[[-4.2673e+00, -1.2152e+00, -2.1525e+00, ..., -1.3916e+00,
-2.0499e+00, -1.7557e+00],
[-4.1360e+00, -2.7873e+00, -2.3126e+00, ..., -1.0363e+00,
-1.9697e+00, -2.0315e+00],
[-2.6791e+00, -2.0904e+00, -2.4900e+00, ..., -1.3506e+00,
-1.3676e+00, -1.2512e+00],
...,
[-8.7222e-01, -6.8527e-01, -1.1744e+00, ..., -5.2695e-01,
-7.5230e-01, -8.2546e-01],
[-2.8377e+00, -4.9513e-01, -8.8505e-01, ..., -5.6291e-01,
-9.3647e-01, -1.1580e+00],
[-2.0189e+00, -1.4581e+00, -1.3746e+00, ..., -4.9587e-01,
-5.4727e-01, -5.0248e-01]],
[[ 3.0858e+00, 4.1392e+00, 3.2297e+00, ..., 1.7205e+00,
9.4601e-01, 2.5907e+00],
[ 4.6908e-01, 8.9149e-01, 2.2736e+00, ..., 1.3185e+00,
6.8108e-01, 2.5212e+00],
[ 9.2579e-01, 1.4966e+00, 1.8378e+00, ..., 2.7862e-01,
9.5557e-01, 1.6984e+00],
...,
[ 2.3011e-01, 9.5055e-01, 1.1419e+00, ..., -3.2518e-01,
5.1934e-01, 1.5184e+00],
[ 3.0768e-01, 2.0908e+00, 1.4875e+00, ..., -1.4757e-01,
8.1692e-01, 1.2480e+00],
[ 1.4151e+00, 1.3280e+00, 1.7230e+00, ..., 6.4788e-01,
1.4130e+00, 1.4525e+00]],
...,
[[-2.2068e+00, -7.5771e+00, -6.0422e+00, ..., -4.6976e+00,
-3.2008e+00, -3.8731e+00],
[-2.8583e+00, -6.6974e+00, -4.8079e+00, ..., -4.9251e+00,
-3.4322e+00, -2.9633e+00],
[-2.8195e-01, -2.9946e+00, -3.2767e+00, ..., -3.3265e+00,
-2.0803e+00, -1.1638e+00],
...,
[-9.4435e-01, -2.3096e+00, -1.8221e+00, ..., -4.2948e+00,
-3.0930e+00, -2.6945e+00],
[-1.3058e+00, -3.7346e+00, -3.4818e+00, ..., -5.1992e+00,
-4.1245e+00, -4.5022e+00],
[-2.1770e+00, -4.9078e+00, -3.0109e+00, ..., -3.4206e+00,
-2.4240e+00, -2.5740e+00]],
[[-7.5121e-01, -8.8057e-02, 2.0504e-01, ..., 1.4890e+00,
1.6092e+00, 1.1837e+00],
[ 1.7862e+00, 1.0240e+00, 1.5990e+00, ..., 1.5176e+00,
1.1203e+00, 1.7969e+00],
[ 1.2253e+00, 3.4813e-02, 7.4383e-01, ..., 1.2846e+00,
6.1147e-01, 1.1808e+00],
...,
[ 1.2611e+00, -1.8083e-01, 1.0173e+00, ..., 1.3996e+00,
6.6081e-01, 8.3029e-01],
[ 1.8983e+00, 9.0119e-01, 1.7158e+00, ..., 1.2718e+00,
7.1595e-01, 1.4204e+00],
[ 1.0423e+00, 4.3511e-01, 9.7915e-01, ..., 7.9095e-01,
7.1553e-01, 1.4518e+00]],
[[-5.1040e-01, 5.6461e-01, 1.0811e-01, ..., -3.2313e-01,
-5.4436e-01, -3.8530e-02],
[-1.1511e+00, -1.2291e+00, -1.2560e+00, ..., -1.1509e+00,
-1.2962e+00, -1.0307e+00],
[-2.0943e+00, -7.9921e-01, 2.0185e-01, ..., -7.3278e-01,
-1.0378e+00, -3.1902e-01],
...,
[-1.7541e+00, -1.4525e+00, -1.0521e+00, ..., -8.0315e-01,
-1.4248e+00, -9.2952e-01],
[-1.7951e+00, -1.0696e+00, -8.1748e-01, ..., -8.5469e-01,
-1.6491e+00, -8.1635e-01],
[-3.5261e-01, 3.9069e-01, -2.5268e-01, ..., -2.0097e-01,
-7.4745e-01, 3.9328e-02]]],
[[[ 7.2908e-01, 1.0698e+00, 7.0769e-01, ..., 4.0328e-01,
3.7361e-01, 1.9288e-01],
[-3.4917e-01, -1.2501e+00, -4.8246e-01, ..., 3.0995e-01,
3.4499e-01, -1.1343e+00],
[ 1.7649e-01, -5.8975e-01, -2.8059e-01, ..., -6.0002e-01,
9.4497e-01, 6.9176e-01],
...,
[ 1.0064e-01, -1.0962e+00, -4.6244e-01, ..., 2.2711e-01,
9.2562e-01, 5.1652e-01],
[-3.9908e-01, -1.8218e+00, -1.2492e+00, ..., 2.8852e-01,
1.3326e+00, -1.1640e-01],
[ 4.4097e-01, -5.1890e-01, -1.2036e+00, ..., -2.8589e-01,
4.0447e-01, 5.9716e-01]],
[[-3.0796e+00, -1.3967e+00, -2.9699e+00, ..., -1.7138e+00,
-2.2361e+00, -1.0921e+00],
[-4.2683e+00, -1.5275e+00, -2.1981e+00, ..., -3.4628e-01,
-1.2930e+00, -2.9637e-01],
[-3.8441e+00, -1.9115e+00, -2.7707e+00, ..., -1.1152e+00,
-1.5885e+00, -7.9908e-01],
...,
[-2.5497e+00, -1.5351e+00, -7.2443e-01, ..., 1.5822e-01,
2.1389e-01, 4.7422e-01],
[-3.4854e+00, -1.2955e+00, -2.1274e+00, ..., 1.7518e-01,
-3.6897e-01, 2.5965e-01],
[-1.1432e+00, -8.2308e-01, -2.7087e+00, ..., -4.0194e-01,
-7.1942e-01, -5.8554e-01]],
[[ 1.7488e+00, 2.7489e+00, 2.4434e+00, ..., 2.8241e+00,
1.3945e+00, 2.9012e+00],
[ 3.4607e-01, 1.0250e+00, 1.6141e+00, ..., 1.7955e+00,
-7.0824e-02, 2.5369e+00],
[ 7.7054e-01, 9.8387e-01, 7.0315e-01, ..., 1.5513e+00,
5.6518e-01, 1.8685e+00],
...,
[ 9.3266e-01, 1.2389e+00, 8.1629e-01, ..., 1.1015e+00,
-7.6768e-02, 1.4635e+00],
[ 6.2817e-01, 1.8148e+00, 1.6187e+00, ..., 8.1519e-01,
6.7607e-01, 1.5903e+00],
[ 1.5386e+00, 1.5165e+00, 1.9614e+00, ..., 1.4805e+00,
1.1239e+00, 2.0285e+00]],
...,
[[-1.4320e+00, -4.5307e+00, -3.6734e+00, ..., -3.2967e+00,
-2.4030e+00, -3.9014e+00],
[-1.7313e+00, -3.5917e+00, -3.8363e+00, ..., -5.2476e+00,
-3.7960e+00, -4.3738e+00],
[-9.2118e-01, -2.8967e+00, -3.1844e+00, ..., -2.1752e+00,
-2.4807e+00, -1.6355e+00],
...,
[-1.4874e+00, -3.7177e+00, -3.7862e+00, ..., -5.2422e+00,
-3.9024e+00, -1.5513e+00],
[-1.4306e+00, -4.2912e+00, -5.1390e+00, ..., -6.7129e+00,
-3.4961e+00, -3.8361e+00],
[-1.5796e+00, -4.8163e+00, -4.5280e+00, ..., -3.7865e+00,
-2.1952e+00, -2.8366e+00]],
[[ 7.4529e-01, 5.9470e-01, -1.5385e-01, ..., 8.8375e-01,
1.0480e+00, 1.2454e+00],
[ 1.1823e+00, 1.0186e+00, 5.4122e-01, ..., 1.5120e+00,
2.3308e+00, 2.5738e+00],
[ 8.0220e-01, 7.0243e-01, 1.0568e-01, ..., 1.3108e+00,
1.6186e+00, 1.9106e+00],
...,
[ 9.0162e-01, -4.1368e-02, 4.9502e-01, ..., 2.3012e+00,
1.9599e+00, 1.6672e+00],
[ 1.4806e+00, 1.3239e+00, 5.9845e-01, ..., 1.7173e+00,
1.5277e+00, 2.4548e+00],
[ 7.1824e-01, 8.3327e-01, -1.8290e-01, ..., 1.0199e+00,
1.3342e+00, 1.3649e+00]],
[[-2.8698e-01, -1.2616e-01, -9.2461e-01, ..., -2.1652e-01,
-1.6556e-01, 8.7554e-01],
[-1.4474e+00, -1.1622e+00, -1.7241e+00, ..., 3.8637e-02,
-1.0987e-01, 3.9360e-01],
[-1.6641e+00, -4.9233e-01, -7.6938e-01, ..., -6.5919e-02,
4.4792e-02, 2.3841e-01],
...,
[-1.7605e+00, -1.5204e+00, -1.1554e+00, ..., -8.2597e-01,
-5.3598e-03, -3.5071e-01],
[-1.4490e+00, -1.5479e+00, -1.3718e+00, ..., -3.7506e-01,
-3.8321e-01, 3.3006e-01],
[-1.1940e-01, 5.2661e-02, -1.0395e+00, ..., 2.4530e-01,
1.5628e-01, 3.4840e-01]]],
[[[ 9.6941e-01, 1.5723e+00, 1.0501e+00, ..., 5.4593e-01,
3.3083e-01, -1.7131e-01],
[-9.3558e-01, -1.9825e+00, -8.6835e-01, ..., 2.9081e-01,
6.0003e-01, -5.8330e-01],
[-4.3041e-02, -7.8193e-01, -2.0636e-01, ..., -5.8438e-01,
2.2502e-01, 7.1538e-01],
...,
[ 4.6265e-01, -4.2954e-01, 2.6798e-01, ..., -1.2017e-01,
5.8789e-01, 3.8104e-03],
[-1.7279e-01, -1.1479e+00, -3.0613e-01, ..., -1.3802e-01,
1.0199e+00, 2.3202e-01],
[ 7.3360e-01, -2.6813e-01, -3.2950e-01, ..., -5.0164e-01,
-4.4184e-01, 2.9759e-01]],
[[-4.3155e+00, -1.3392e+00, -2.1251e+00, ..., -2.9674e+00,
-3.6266e+00, -9.6242e-01],
[-4.1695e+00, -2.9361e+00, -2.4950e+00, ..., -2.1496e+00,
-3.1973e+00, -6.0851e-01],
[-2.6788e+00, -2.1485e+00, -3.4685e+00, ..., -3.2810e+00,
-3.3888e+00, -5.9363e-01],
...,
[-8.6610e-01, -7.0666e-01, -1.7821e+00, ..., -4.6390e+00,
-3.0477e+00, -7.1049e-01],
[-2.8551e+00, -4.6111e-01, -1.0207e+00, ..., -2.0790e+00,
-1.7345e+00, -1.8280e-02],
[-2.0743e+00, -1.4143e+00, -1.7977e+00, ..., -1.0494e+00,
-1.0770e+00, 2.8459e-02]],
[[ 3.0936e+00, 4.1401e+00, 3.3110e+00, ..., 2.3474e+00,
8.0718e-01, 2.4126e+00],
[ 4.0362e-01, 8.7557e-01, 2.3838e+00, ..., 2.0205e+00,
3.0916e-01, 2.5314e+00],
[ 8.5035e-01, 1.3706e+00, 2.0766e+00, ..., 7.6824e-01,
9.0512e-01, 2.2348e+00],
...,
[ 3.9139e-01, 9.4806e-01, 1.2510e+00, ..., 1.7918e-01,
-1.1285e+00, 1.1332e+00],
[ 4.1042e-01, 2.0610e+00, 1.7520e+00, ..., 2.6777e-01,
3.3648e-01, 1.2525e+00],
[ 1.4519e+00, 1.3899e+00, 1.7324e+00, ..., 1.1374e+00,
8.2772e-01, 1.2702e+00]],
...,
[[-2.1491e+00, -7.5356e+00, -6.0289e+00, ..., -4.1567e+00,
-3.1820e+00, -3.9201e+00],
[-2.7720e+00, -6.6446e+00, -4.9562e+00, ..., -5.6998e+00,
-4.8330e+00, -4.7730e+00],
[-2.2842e-01, -2.8855e+00, -3.5639e+00, ..., -4.6491e+00,
-3.4945e+00, -2.1057e+00],
...,
[-9.8048e-01, -2.5219e+00, -1.9960e+00, ..., -5.7619e+00,
-5.0985e+00, -2.5957e+00],
[-1.3014e+00, -3.8498e+00, -3.4241e+00, ..., -6.5470e+00,
-4.9665e+00, -4.9102e+00],
[-2.2198e+00, -4.9830e+00, -3.0556e+00, ..., -4.5301e+00,
-2.7881e+00, -3.2966e+00]],
[[-7.8870e-01, -1.4545e-01, 8.3154e-02, ..., 9.9373e-01,
9.9142e-01, 1.7558e+00],
[ 1.7897e+00, 1.0145e+00, 1.3721e+00, ..., 1.5459e+00,
1.5645e+00, 2.9222e+00],
[ 1.2217e+00, -5.0998e-03, 3.2515e-01, ..., 1.2989e+00,
9.0731e-01, 2.2610e+00],
...,
[ 1.3099e+00, -3.2411e-01, 2.4719e-01, ..., 1.0660e+00,
6.6883e-01, 1.4273e+00],
[ 1.9172e+00, 1.0129e+00, 1.6411e+00, ..., 1.6210e+00,
1.0535e+00, 2.2873e+00],
[ 1.0230e+00, 4.3546e-01, 7.9493e-01, ..., 9.7619e-01,
1.1496e+00, 1.5655e+00]],
[[-5.2055e-01, 5.9786e-01, 5.0974e-02, ..., -2.7953e-01,
-5.9521e-01, 8.5440e-01],
[-1.1552e+00, -1.2014e+00, -1.4468e+00, ..., -3.2828e-01,
-1.3397e-01, 1.0646e+00],
[-2.0905e+00, -8.5280e-01, -2.0386e-01, ..., 2.8868e-01,
8.8318e-02, 5.4419e-01],
...,
[-1.7852e+00, -1.5449e+00, -6.7570e-01, ..., -9.9974e-01,
-1.1563e+00, -1.0036e+00],
[-1.7910e+00, -1.1079e+00, -9.7277e-01, ..., -7.9879e-01,
-1.0984e+00, 3.3811e-01],
[-3.8199e-01, 3.6555e-01, -4.3756e-01, ..., 6.6358e-02,
-2.8458e-01, 6.2691e-01]]]], device='cuda:0'),
tensor([[[[ 7.0881e-01, 1.0664e+00, 7.5445e-01, ..., 4.2418e-01,
3.4324e-01, 2.3005e-01],
[-2.8659e-01, -1.4723e+00, -5.7238e-01, ..., 4.0060e-01,
3.0581e-01, -1.1802e+00],
[ 1.1177e-01, -8.6299e-01, -3.0103e-01, ..., -4.9234e-01,
1.0069e+00, 5.8591e-01],
...,
[ 8.7049e-02, -1.1277e+00, -9.0389e-01, ..., -1.0673e-01,
9.4464e-01, 5.0988e-01],
[-4.4401e-01, -1.8183e+00, -8.2469e-01, ..., -4.2593e-02,
1.4160e+00, -1.3412e-01],
[ 3.6020e-01, -5.6975e-01, -6.7935e-01, ..., -2.6500e-01,
3.5182e-01, 5.4778e-01]],
[[-3.0685e+00, -1.4897e+00, -3.0851e+00, ..., -1.7629e+00,
-2.2991e+00, -1.1651e+00],
[-4.3705e+00, -1.7788e+00, -2.4026e+00, ..., -4.4114e-01,
-1.3666e+00, -3.6979e-01],
[-4.4064e+00, -2.6422e+00, -3.3776e+00, ..., -1.4265e+00,
-1.5657e+00, -7.5960e-01],
...,
[-2.7531e+00, -1.8571e+00, -2.9140e+00, ..., 2.8732e-01,
-2.0947e-02, 5.0685e-01],
[-3.6623e+00, -1.3795e+00, -2.3544e+00, ..., -2.0352e-01,
-3.6369e-01, 2.6043e-01],
[-1.3447e+00, -8.8473e-01, -1.6935e+00, ..., -3.4743e-01,
-6.8593e-01, -5.1622e-01]],
[[ 1.7584e+00, 2.8364e+00, 2.4237e+00, ..., 2.8232e+00,
1.4527e+00, 2.9803e+00],
[ 2.0971e-01, 1.1740e+00, 1.5463e+00, ..., 1.7840e+00,
-3.2081e-02, 2.5716e+00],
[ 7.3513e-01, 1.4709e+00, 8.7917e-01, ..., 1.0269e+00,
5.7655e-01, 1.8270e+00],
...,
[ 9.6299e-01, 1.3147e+00, 5.0897e-01, ..., 5.2639e-01,
1.2313e-01, 1.4302e+00],
[ 6.6324e-01, 1.9727e+00, 9.2642e-01, ..., 9.6808e-01,
6.7919e-01, 1.5079e+00],
[ 1.5510e+00, 1.6519e+00, 1.6953e+00, ..., 1.4683e+00,
1.1120e+00, 1.9846e+00]],
...,
[[-1.4878e+00, -4.6668e+00, -4.0020e+00, ..., -3.3141e+00,
-2.4251e+00, -4.0355e+00],
[-1.8799e+00, -3.9626e+00, -4.3210e+00, ..., -5.3081e+00,
-3.8164e+00, -4.4380e+00],
[-9.6353e-01, -3.1798e+00, -3.7715e+00, ..., -2.4332e+00,
-2.2788e+00, -1.6159e+00],
...,
[-1.5771e+00, -4.1553e+00, -4.4927e+00, ..., -5.0975e+00,
-2.4380e+00, -1.5807e+00],
[-1.5492e+00, -4.6028e+00, -5.2679e+00, ..., -5.5246e+00,
-3.3581e+00, -3.7355e+00],
[-1.6384e+00, -4.9535e+00, -4.1811e+00, ..., -3.5639e+00,
-2.1692e+00, -2.7454e+00]],
[[ 7.9106e-01, 5.6332e-01, -1.0138e-01, ..., 8.8476e-01,
1.0709e+00, 1.2355e+00],
[ 1.2029e+00, 9.7656e-01, 4.7487e-01, ..., 1.5695e+00,
2.3403e+00, 2.5909e+00],
[ 6.1324e-01, 5.7566e-01, 1.2717e-01, ..., 1.0051e+00,
1.5488e+00, 1.9712e+00],
...,
[ 8.3131e-01, -2.0776e-02, -5.1115e-02, ..., 1.5914e+00,
1.8376e+00, 1.6509e+00],
[ 1.5339e+00, 1.4507e+00, 1.2047e+00, ..., 1.3000e+00,
1.5290e+00, 2.4180e+00],
[ 6.5891e-01, 9.2774e-01, 1.0251e+00, ..., 1.0751e+00,
1.3849e+00, 1.3847e+00]],
[[-2.7139e-01, -1.8418e-01, -9.7480e-01, ..., -1.7568e-01,
-1.4942e-01, 8.7694e-01],
[-1.3533e+00, -1.4741e+00, -2.0539e+00, ..., 1.5550e-01,
-1.1023e-01, 3.7293e-01],
[-1.6921e+00, -9.0998e-01, -9.0520e-01, ..., 2.0714e-01,
1.1852e-01, 2.5514e-01],
...,
[-1.7077e+00, -1.7240e+00, -1.7296e+00, ..., -8.2128e-02,
-3.2940e-01, -3.1189e-01],
[-1.4136e+00, -1.6648e+00, -1.6507e+00, ..., -2.9742e-01,
-4.6722e-01, 3.0169e-01],
[-2.1322e-01, 3.4384e-02, -4.6032e-01, ..., 2.1606e-01,
1.0627e-01, 3.2466e-01]]],
[[[ 6.7101e-01, 1.1456e+00, 5.3471e-01, ..., 1.1196e-01,
3.9422e-01, -5.7154e-02],
[-7.4251e-01, -1.4380e+00, -5.6171e-01, ..., 6.7291e-01,
1.3297e-01, -2.0085e-01],
[-4.6302e-01, -1.7748e+00, -3.0989e-01, ..., 6.6878e-01,
4.5984e-01, 9.6888e-01],
...,
[ 2.0355e-01, -4.9255e-01, 5.1366e-01, ..., -8.1621e-02,
1.1496e+00, 4.6534e-01],
[-8.5184e-02, -6.4076e-01, -1.3479e-01, ..., 3.4716e-01,
1.2803e+00, 3.0017e-01],
[ 8.4043e-01, -1.7628e-01, -9.5894e-02, ..., -5.2779e-02,
-1.8142e-01, 5.4923e-01]],
[[-2.6517e+00, -1.0317e+00, -1.9764e+00, ..., -1.6425e+00,
-3.3995e+00, -2.0238e+00],
[-3.1223e+00, -8.1016e-01, -2.4426e+00, ..., -1.0761e+00,
-2.6663e+00, -1.9931e+00],
[-2.5453e+00, -3.2835e-02, -1.3310e+00, ..., -6.9917e-01,
-2.3005e+00, -1.7883e+00],
...,
[-8.6604e-01, -1.0974e+00, -8.9726e-01, ..., -5.5525e+00,
-3.4919e+00, 1.5618e-01],
[-2.4979e+00, -2.3294e-01, -7.9439e-01, ..., -3.8510e+00,
-1.2244e+00, 9.3053e-01],
[-1.5475e+00, -1.3415e+00, -1.0001e+00, ..., -1.5714e+00,
-8.1609e-01, -2.5486e-01]],
[[ 1.9035e+00, 2.4710e+00, 1.8448e+00, ..., -4.2339e-01,
-5.5847e-03, 1.9492e+00],
[ 4.9699e-01, 7.0628e-01, 8.8565e-01, ..., -1.8430e-01,
-3.8715e-01, 1.9082e+00],
[ 8.6221e-01, -2.7357e-01, -1.6052e+00, ..., -1.0965e+00,
-1.8752e-01, 1.7597e+00],
...,
[ 4.7205e-01, 7.9866e-01, 1.6550e+00, ..., -3.5452e+00,
-1.4374e+00, 5.5190e-01],
[ 1.1045e+00, 2.2462e+00, 1.7825e+00, ..., -1.8494e+00,
6.8131e-01, 1.3468e+00],
[ 2.3854e+00, 1.8982e+00, 2.4182e+00, ..., 1.6255e+00,
8.2727e-01, 1.3412e+00]],
...,
[[-2.0416e+00, -5.0235e+00, -3.3662e+00, ..., 3.9877e-01,
8.1060e-01, -1.2680e+00],
[-2.2629e+00, -4.0598e+00, -1.7190e+00, ..., -1.1092e+00,
3.4124e-01, -2.3632e+00],
[-4.8038e-02, -2.2802e+00, -2.4473e+00, ..., -1.7915e+00,
-1.0342e+00, -2.3716e+00],
...,
[-2.5719e+00, -4.3034e+00, -3.7694e+00, ..., -7.2207e+00,
-7.0708e+00, -5.4652e+00],
[-2.3307e+00, -5.2527e+00, -4.0811e+00, ..., -8.2466e+00,
-5.5048e+00, -7.5297e+00],
[-2.5612e+00, -5.7338e+00, -2.8267e+00, ..., -4.1116e+00,
-3.8053e+00, -5.0362e+00]],
[[ 8.3027e-01, 1.1792e+00, 5.8016e-01, ..., 6.1511e-01,
4.6520e-01, 1.0008e+00],
[ 5.9376e-01, 4.3523e-01, -1.8683e-02, ..., -6.9023e-01,
1.0782e-01, 1.6474e+00],
[-1.1412e-01, 1.0413e+00, -7.0907e-01, ..., -2.0134e-01,
8.8947e-02, 1.1830e+00],
...,
[ 1.6754e+00, 2.5721e-01, 3.8755e-01, ..., -2.5653e+00,
1.1346e+00, 1.7920e+00],
[ 2.0370e+00, 1.1468e+00, 1.7689e+00, ..., -3.6904e-01,
1.6986e+00, 2.9424e+00],
[ 1.1722e+00, 8.2037e-01, 1.3247e+00, ..., 1.3791e+00,
1.9457e+00, 1.6438e+00]],
[[-4.8437e-02, -3.5050e-01, -1.1332e+00, ..., -6.1122e-01,
-1.1576e+00, -4.6887e-02],
[-1.2913e+00, -2.2455e+00, -1.9622e+00, ..., -1.0791e+00,
-7.4384e-01, 1.7280e-01],
[-1.2146e+00, -3.1746e-01, -4.1346e-01, ..., -9.6504e-01,
-2.0710e-01, -2.1764e-01],
...,
[-1.4366e+00, -1.5968e+00, -9.6549e-01, ..., 5.2579e-02,
-2.8077e+00, -1.6017e+00],
[-1.7345e+00, -1.1761e+00, -8.1578e-01, ..., -2.2407e+00,
-2.2394e+00, 5.0857e-01],
[-1.9549e-01, 4.3683e-01, -2.3626e-01, ..., 1.0154e-01,
2.2505e-01, 7.8884e-01]]],
[[[ 9.2318e-01, 1.5477e+00, 1.1307e+00, ..., 8.1001e-01,
1.0380e+00, 4.7790e-01],
[-8.4036e-01, -2.0935e+00, -7.8560e-01, ..., -3.2249e-01,
-1.7816e-01, -3.3055e-01],
[ 8.9150e-02, -7.4893e-01, 2.0074e-01, ..., -8.6605e-01,
-1.5174e-01, 6.1566e-01],
...,
[ 4.1275e-01, -2.3009e-01, -1.9434e-01, ..., 8.0988e-02,
-1.0020e-01, 3.6239e-01],
[-2.1021e-01, -1.0460e+00, -9.7035e-02, ..., -2.0492e-02,
1.0305e-01, -1.1412e-01],
[ 7.3532e-01, -3.1465e-01, -1.9626e-01, ..., -3.3007e-01,
-8.7209e-01, 1.7093e-01]],
[[-4.3825e+00, -1.2252e+00, -2.1815e+00, ..., -1.3953e+00,
-2.0872e+00, -1.8638e+00],
[-4.1190e+00, -2.6862e+00, -2.3533e+00, ..., -1.1836e+00,
-2.1158e+00, -2.1278e+00],
[-2.4826e+00, -1.6361e+00, -2.8800e+00, ..., -1.5130e+00,
-1.5636e+00, -1.3713e+00],
...,
[-8.4312e-01, -6.8124e-01, -1.1469e+00, ..., -4.5213e-01,
-8.9795e-01, -8.8880e-01],
[-2.8441e+00, -4.0301e-01, -7.9996e-01, ..., -6.5742e-01,
-1.1433e+00, -1.1891e+00],
[-2.0755e+00, -1.4975e+00, -1.4192e+00, ..., -6.6978e-01,
-7.6284e-01, -5.2593e-01]],
[[ 3.0392e+00, 4.0830e+00, 3.1590e+00, ..., 1.7178e+00,
8.2531e-01, 2.5577e+00],
[ 3.9288e-01, 7.3722e-01, 2.2452e+00, ..., 1.2772e+00,
6.7181e-01, 2.6251e+00],
[ 8.3357e-01, 1.2351e+00, 1.7926e+00, ..., 3.8198e-01,
1.0145e+00, 1.8129e+00],
...,
[ 3.0153e-01, 9.6393e-01, 1.1853e+00, ..., -2.9208e-01,
5.0145e-01, 1.4596e+00],
[ 3.2468e-01, 2.1064e+00, 1.5229e+00, ..., -1.3590e-01,
7.4529e-01, 1.1388e+00],
[ 1.4442e+00, 1.3423e+00, 1.7219e+00, ..., 5.9834e-01,
1.2760e+00, 1.3116e+00]],
...,
[[-2.1592e+00, -7.3592e+00, -5.9956e+00, ..., -4.9484e+00,
-3.5670e+00, -4.2669e+00],
[-2.8066e+00, -6.4374e+00, -4.6094e+00, ..., -4.8443e+00,
-3.5960e+00, -3.1698e+00],
[-2.2771e-01, -2.7998e+00, -3.1083e+00, ..., -3.3773e+00,
-2.2499e+00, -1.3189e+00],
...,
[-9.6114e-01, -2.3014e+00, -1.7421e+00, ..., -5.0283e+00,
-3.1379e+00, -2.6940e+00],
[-1.3296e+00, -3.7459e+00, -3.5023e+00, ..., -5.2158e+00,
-4.2028e+00, -4.2298e+00],
[-2.2224e+00, -4.9738e+00, -3.0441e+00, ..., -3.3854e+00,
-2.4834e+00, -2.4015e+00]],
[[-7.4801e-01, 1.3127e-02, 1.8451e-01, ..., 1.4927e+00,
1.6388e+00, 1.1461e+00],
[ 1.7807e+00, 1.1603e+00, 1.3564e+00, ..., 1.4487e+00,
1.1882e+00, 1.8413e+00],
[ 1.2471e+00, 1.8340e-01, -3.7768e-02, ..., 1.5830e+00,
6.1183e-01, 1.2957e+00],
...,
[ 1.2570e+00, -1.5637e-01, 9.8507e-01, ..., 1.7384e+00,
6.8402e-01, 8.9941e-01],
[ 1.9031e+00, 9.6820e-01, 1.7478e+00, ..., 1.2618e+00,
7.0076e-01, 1.4419e+00],
[ 1.0310e+00, 3.8309e-01, 9.6316e-01, ..., 7.5165e-01,
6.6305e-01, 1.4149e+00]],
[[-5.1882e-01, 6.1150e-01, 6.9927e-02, ..., -3.0204e-01,
-5.8921e-01, -5.8104e-02],
[-1.0709e+00, -1.0718e+00, -1.4752e+00, ..., -1.1700e+00,
-1.4395e+00, -1.1132e+00],
[-2.0089e+00, -5.2634e-01, -7.2058e-01, ..., -4.3609e-01,
-1.0692e+00, -2.4926e-01],
...,
[-1.7452e+00, -1.4583e+00, -1.0426e+00, ..., -4.7124e-01,
-1.4525e+00, -8.7290e-01],
[-1.7730e+00, -1.0860e+00, -8.5445e-01, ..., -7.8615e-01,
-1.7319e+00, -6.9256e-01],
[-3.7778e-01, 3.7998e-01, -3.2053e-01, ..., -2.5836e-01,
-7.9896e-01, -1.3074e-03]]],
[[[ 7.0664e-01, 1.0503e+00, 6.6008e-01, ..., 8.9499e-01,
6.2546e-01, 2.4135e-01],
[-3.5767e-01, -1.4690e+00, -6.2525e-01, ..., -1.0539e+00,
-1.1075e+00, -1.0927e+00],
[ 4.9308e-02, -9.1188e-01, -4.9198e-01, ..., -1.5296e+00,
-1.6399e+00, 1.0037e+00],
...,
[-2.1788e+00, -3.1185e-01, -2.6739e+00, ..., 8.8258e-01,
2.2531e+00, 3.8930e-01],
[-2.1400e+00, -2.4570e+00, 3.1646e-01, ..., 8.3525e-01,
1.3677e+00, 3.5710e-01],
[ 5.6214e-02, -1.2300e+00, 4.6965e-01, ..., 8.6768e-02,
-2.7489e-01, 5.0485e-01]],
[[-3.3083e+00, -1.5800e+00, -3.0585e+00, ..., -2.0173e+00,
-2.7474e+00, -1.1900e+00],
[-4.5176e+00, -1.8515e+00, -2.2985e+00, ..., -1.8996e+00,
-2.6575e+00, -7.3515e-01],
[-4.4838e+00, -2.6064e+00, -3.0365e+00, ..., -1.8429e+00,
-7.5218e-01, 3.9247e-01],
...,
[-6.6833e+00, -9.9035e+00, -3.6450e+00, ..., -5.0875e+00,
-3.1382e+00, -1.2524e+00],
[-5.3601e+00, -4.9307e+00, -4.5842e+00, ..., -5.7309e-01,
-6.9136e-01, 3.4613e-01],
[-1.5985e+00, -1.3924e+00, -1.5245e+00, ..., -5.4784e-01,
-5.1241e-01, -1.4088e-02]],
[[ 1.8922e+00, 2.9442e+00, 2.4468e+00, ..., 2.3793e+00,
1.2988e+00, 2.4084e+00],
[ 3.3116e-01, 1.3529e+00, 1.7002e+00, ..., 2.3483e+00,
1.1981e+00, 1.9884e+00],
[ 7.8609e-01, 1.5184e+00, 9.3438e-01, ..., 6.2662e-01,
-7.3073e-01, -1.6945e+00],
...,
[ 4.0970e+00, 2.5625e+00, -5.2415e+00, ..., 1.2705e-01,
1.4631e-01, 7.9841e-01],
[ 1.1096e+00, 4.3876e+00, -5.4709e-01, ..., 7.6579e-01,
5.3092e-01, 1.2055e+00],
[ 1.3698e+00, 2.7521e+00, 1.9376e+00, ..., 7.9905e-01,
5.6309e-01, 1.2733e+00]],
...,
[[-1.5706e+00, -4.8039e+00, -4.0219e+00, ..., -4.7486e+00,
-3.0190e+00, -3.7382e+00],
[-1.9870e+00, -4.3205e+00, -4.6360e+00, ..., -6.6021e+00,
-3.2381e+00, -3.5923e+00],
[-1.0594e+00, -3.6112e+00, -4.0332e+00, ..., -9.8023e+00,
-5.0788e+00, -4.2111e+00],
...,
[-1.7661e+00, -6.7653e+00, -7.7156e+00, ..., -6.0731e+00,
-3.8419e+00, -5.3589e+00],
[-1.9115e+00, -5.6037e+00, -8.7449e+00, ..., -5.7164e+00,
-4.4347e+00, -5.2020e+00],
[-1.6107e+00, -4.6072e+00, -4.9900e+00, ..., -4.6543e+00,
-2.0570e+00, -3.1530e+00]],
[[ 6.9445e-01, 5.6033e-01, -7.7997e-02, ..., 1.2304e+00,
6.6526e-01, 1.6907e+00],
[ 1.1902e+00, 1.0038e+00, 5.4026e-01, ..., 2.1578e+00,
2.0704e-01, 1.9941e+00],
[ 5.9343e-01, 5.9411e-01, 6.8586e-02, ..., -5.9782e-01,
-1.9348e+00, 2.9681e-01],
...,
[-2.4902e+00, -2.6616e+00, 2.4561e+00, ..., 2.3473e+00,
3.0992e+00, 2.7236e+00],
[ 2.8698e-01, -2.9526e-01, 8.6581e-01, ..., 3.2257e+00,
1.5497e+00, 2.5463e+00],
[ 7.0299e-01, 1.1349e+00, 1.2314e+00, ..., 1.3732e+00,
1.2016e+00, 1.3508e+00]],
[[-3.0825e-01, -1.7889e-01, -9.4964e-01, ..., 8.5395e-01,
2.5025e-01, 7.1288e-01],
[-1.4357e+00, -1.3879e+00, -2.0831e+00, ..., -4.2206e-01,
-1.5257e+00, 4.2372e-01],
[-1.7654e+00, -7.9825e-01, -9.6742e-01, ..., -2.8085e-01,
-2.4225e+00, 1.3133e+00],
...,
[-2.1852e+00, -8.8031e-01, -2.5884e-01, ..., -3.8415e+00,
-2.1228e+00, -2.2274e+00],
[-2.2899e+00, -5.6061e-01, -2.7748e+00, ..., -6.4869e-01,
-7.9570e-01, -2.3198e-02],
[-1.2565e-01, -1.3324e-01, -1.0575e+00, ..., -8.2511e-01,
-7.0641e-01, 2.1175e-01]]],
[[[-7.2806e-01, -4.6641e-01, 8.2877e-01, ..., -3.5179e+00,
-1.5616e+00, -1.0551e+00],
[-4.5005e+00, -4.4192e+00, -3.7095e+00, ..., -1.3083e+01,
-7.3099e+00, -5.4498e+00],
[-2.6129e+00, -2.3337e+00, -5.0599e+00, ..., -1.6010e+01,
-9.2293e+00, -8.7834e+00],
...,
[-3.3275e-01, -2.7475e+00, 1.6265e+00, ..., -3.0252e+00,
-3.2480e+00, -4.3257e+00],
[-4.9202e-01, -2.6028e+00, 4.3149e+00, ..., 2.0386e+00,
2.7891e+00, 3.4627e-01],
[-7.8391e-02, -2.1364e+00, 3.2405e+00, ..., -3.7888e-01,
4.1689e-01, 6.5388e-01]],
[[ 1.0912e+00, -2.7881e+00, -1.1807e+00, ..., -6.5877e-01,
-4.6863e+00, -5.0028e+00],
[-2.9435e+00, -9.9906e+00, -3.8535e+00, ..., -8.2572e+00,
-1.1081e+01, -6.4319e+00],
[-2.9772e+00, -9.5906e+00, -6.6322e+00, ..., -1.6883e+01,
-2.0215e+01, -7.6264e+00],
...,
[-1.3873e+00, -9.3021e+00, -1.4495e+01, ..., -1.4320e+01,
-1.4217e+01, -6.0275e+00],
[-2.5369e+00, -6.3617e+00, -1.3894e+01, ..., -3.7969e-01,
-6.9291e-01, -8.7046e-02],
[-1.2644e+00, -3.7225e+00, -3.3995e+00, ..., -5.3111e-01,
-1.2270e+00, -9.8621e-01]],
[[-1.9619e+00, -1.7220e+00, -1.0484e+00, ..., -4.1616e+00,
-4.8523e+00, -2.4077e+00],
[-2.7166e+00, -4.6499e+00, -2.6281e+00, ..., -4.5986e+00,
-2.5734e+00, -4.3476e+00],
[-6.0252e-01, -2.9004e+00, -3.6621e+00, ..., -6.8748e+00,
-9.4990e+00, -8.9496e+00],
...,
[-3.6866e-01, 4.2887e+00, -7.0202e+00, ..., -3.9232e+00,
-6.3712e+00, -4.5443e+00],
[-3.2124e-01, 4.8055e+00, -8.5969e+00, ..., 2.8170e+00,
2.7199e+00, 1.6560e+00],
[ 8.9250e-01, 4.1364e+00, -5.1464e+00, ..., 1.7798e+00,
1.2867e+00, 2.2918e+00]],
...,
[[-2.2434e-01, -3.0149e+00, -1.2312e+00, ..., -7.8908e+00,
-7.1151e+00, -5.9913e+00],
[ 5.2083e-01, -4.5771e-01, 2.9309e+00, ..., -1.0649e+01,
-1.1918e+01, -8.0702e+00],
[ 8.1192e-01, 1.3238e+00, 2.2672e+00, ..., -5.8672e+00,
-4.5296e+00, -2.3549e+00],
...,
[-7.0134e-01, -3.3437e+00, -1.5522e+01, ..., -9.4305e+00,
-1.0131e+01, -7.3493e+00],
[-7.4395e-01, -4.1972e+00, -9.4562e+00, ..., -1.3334e+01,
-1.0886e+01, -1.0079e+01],
[-6.8823e-01, -2.3707e+00, -3.4217e+00, ..., -3.1520e+00,
-1.9681e+00, -2.9511e+00]],
[[ 2.1839e+00, -1.9878e+00, -1.7717e+00, ..., -2.7931e+00,
-9.4721e+00, -2.6077e+00],
[ 2.9509e-01, -3.6964e+00, -1.5414e+00, ..., 2.5086e+00,
-1.2306e+01, -6.8338e+00],
[-2.0469e+00, -4.5708e+00, -2.8954e+00, ..., -3.1553e+00,
-1.4827e+01, -3.7773e+00],
...,
[ 6.1390e-01, -6.5594e+00, -7.3269e-01, ..., -1.0423e+00,
-2.1006e+00, 1.5493e+00],
[ 1.3567e+00, -3.7034e+00, -3.1889e-01, ..., 5.1162e+00,
3.9301e+00, 4.0928e+00],
[ 7.8819e-01, -9.0619e-01, 3.3806e+00, ..., 1.1253e+00,
1.2930e+00, 1.2779e+00]],
[[ 9.2792e-02, -2.0251e+00, 6.0675e-01, ..., 3.1288e+00,
2.1833e+00, 2.4954e+00],
[-1.4871e+00, -9.0724e+00, -2.5487e-01, ..., -6.8037e-01,
-4.4179e+00, 2.3208e-02],
[-1.3262e+00, -6.8015e+00, 1.8177e+00, ..., -9.5777e+00,
-1.2003e+01, 3.1092e+00],
...,
[-1.6241e+00, -4.6018e+00, -2.6614e+00, ..., -2.6761e+00,
-7.4083e+00, -2.7403e+00],
[-1.6264e+00, -3.4400e+00, -1.9080e+00, ..., -1.0897e+00,
-1.6717e+00, 3.8174e-01],
[-7.5845e-01, -8.9759e-01, -7.5133e-01, ..., 2.2698e-02,
3.4023e-02, 3.3373e-01]]]], device='cuda:0'),
tensor([[[[ 3.7614e-01, 1.4435e-01, -1.1076e+00, ..., 7.1556e-01,
3.4302e-01, -1.2070e-01],
[ 7.8295e-01, -1.5940e+00, -4.2260e-01, ..., 8.4273e-01,
-6.1291e-02, -5.0161e-01],
[ 1.4676e+00, -1.0489e+00, 1.1982e+00, ..., 5.4800e-01,
1.5493e-01, 7.0082e-01],
...,
[-8.7087e-01, -5.7235e-01, -1.9019e+00, ..., 2.0543e-01,
1.2652e+00, 7.9050e-01],
[-5.5676e-01, -6.7546e-01, -7.3270e-01, ..., 9.9514e-01,
1.8193e+00, 1.0584e-02],
[ 7.8252e-01, -6.6961e-01, 5.3686e-01, ..., 1.6772e-01,
5.3128e-01, 7.9716e-01]],
[[-9.1737e-01, 4.0809e-01, 3.7168e-01, ..., -2.2118e+00,
-2.6808e+00, -1.5784e+00],
[ 2.6078e-01, 1.3088e+00, 6.7898e-01, ..., -2.1990e+00,
-2.5823e+00, -1.6699e+00],
[-1.9041e-01, 5.1225e-01, -4.9336e-01, ..., -3.2006e+00,
-2.0925e+00, -8.9905e-01],
...,
[-1.0441e-01, 4.6309e-01, 1.2862e+00, ..., -4.4423e+00,
-7.5610e-01, 1.1028e+00],
[-3.4460e+00, -3.6910e+00, -8.4201e+00, ..., 3.0734e-01,
1.4065e-01, 7.6794e-01],
[-2.4286e+00, -3.2052e+00, -2.4883e+00, ..., -6.0221e-01,
-7.8335e-01, -3.0830e-01]],
[[-1.5361e+00, -1.0761e+00, -5.9536e-01, ..., 2.0396e+00,
9.7581e-01, 2.2216e+00],
[-4.7321e+00, -2.5646e+00, 5.5555e-01, ..., 1.3772e+00,
7.0630e-01, 2.5076e+00],
[-2.3305e+00, -7.6879e+00, -5.8574e+00, ..., 1.1079e-01,
1.3434e+00, 2.1858e+00],
...,
[ 1.7455e+00, -1.2333e+00, -5.7155e+00, ..., 3.3269e-02,
6.6867e-01, 1.0584e+00],
[ 1.1822e+00, 1.4820e+00, -5.1540e+00, ..., 1.2751e+00,
4.4932e-01, 1.5924e+00],
[ 2.4315e+00, 2.6312e+00, 3.6871e+00, ..., 1.4457e+00,
1.0229e+00, 1.8475e+00]],
...,
[[-1.0989e+00, -5.2315e+00, -3.1840e+00, ..., -3.5715e+00,
-2.0410e+00, -3.4369e+00],
[-1.5427e+00, -3.0985e+00, -1.1031e+00, ..., -4.0778e+00,
-2.9290e+00, -3.4622e+00],
[-1.5775e+00, -4.7644e+00, -4.1916e+00, ..., -2.8414e+00,
-2.9396e+00, -2.0934e+00],
...,
[-2.2379e+00, -3.0401e+00, -1.9885e+00, ..., -7.4429e+00,
-4.6911e+00, -4.9378e+00],
[-2.3797e+00, -6.1568e+00, -5.3311e+00, ..., -6.3677e+00,
-5.3404e+00, -5.6398e+00],
[-2.6051e+00, -5.7924e+00, -4.0775e+00, ..., -4.8180e+00,
-1.9673e+00, -3.4246e+00]],
[[-1.1344e-01, -1.3405e+00, -1.7801e+00, ..., 9.8239e-01,
1.6180e+00, 1.5215e+00],
[-5.4775e-02, -3.5414e+00, -4.7097e+00, ..., 5.7945e-01,
1.3785e+00, 2.2672e+00],
[ 1.8754e+00, 1.6247e+00, 1.6185e+00, ..., -2.6331e-01,
8.3329e-01, 1.6222e+00],
...,
[ 1.0278e+00, 1.8322e-01, 2.3069e+00, ..., 1.1231e+00,
3.5128e+00, 2.1930e+00],
[ 1.4127e+00, -2.8213e+00, -1.8014e+00, ..., 2.9719e+00,
2.2564e+00, 2.7028e+00],
[ 1.2738e+00, 1.6072e-01, 1.2173e+00, ..., 1.2279e+00,
1.3595e+00, 1.4802e+00]],
[[ 3.4979e-01, 4.4309e-01, -3.1362e+00, ..., -5.5260e-01,
-8.2125e-01, 3.4566e-02],
[-1.8912e+00, -3.1832e-01, -1.4715e+00, ..., -7.8336e-01,
-8.1843e-01, -3.6144e-01],
[-1.9633e+00, -2.4784e+00, 1.3177e+00, ..., 3.5831e-01,
-1.8531e-01, 2.1608e-01],
...,
[-1.0115e+00, -2.8178e+00, -2.9148e+00, ..., -3.8352e+00,
-5.4038e-01, -4.2755e-01],
[-1.7509e+00, -2.2060e-01, -6.3717e-01, ..., 9.9302e-03,
3.5267e-02, 6.0238e-01],
[-9.3740e-02, 1.1444e-01, -8.8787e-01, ..., 6.5717e-01,
-4.3268e-02, 6.2884e-01]]],
[[[ 9.4992e-01, 1.5182e+00, 1.0984e+00, ..., 1.2117e-01,
7.1151e-01, 4.3747e-01],
[-8.8436e-01, -2.0149e+00, -7.7846e-01, ..., 4.4622e-01,
-8.4680e-01, -3.2785e-01],
[ 2.9878e-02, -8.3329e-01, 1.3052e-01, ..., -2.5178e-01,
-6.3768e-01, 6.1799e-01],
...,
[ 1.2823e-01, 3.3418e-01, 4.0521e-01, ..., 2.8987e-02,
-2.0172e-02, 4.2105e-01],
[-2.4300e-01, 5.7907e-01, 4.4981e-01, ..., -1.7071e-02,
1.5174e-01, -2.9277e-02],
[ 9.1801e-01, 1.1055e+00, 1.0874e-01, ..., -3.2812e-01,
-9.1552e-01, 1.5292e-01]],
[[-4.2070e+00, -1.1714e+00, -2.0503e+00, ..., -2.2539e+00,
-1.6993e+00, -1.8220e+00],
[-4.0680e+00, -2.7683e+00, -2.3546e+00, ..., -2.6203e+00,
-2.9639e+00, -2.1785e+00],
[-2.6628e+00, -1.7824e+00, -2.7631e+00, ..., -4.8758e+00,
-2.7667e+00, -1.4583e+00],
...,
[-1.5285e+00, -2.6574e+00, -5.4864e-01, ..., -6.3373e-01,
-7.6938e-01, -8.1451e-01],
[-2.6481e+00, -6.8026e-01, 1.7651e+00, ..., -6.2525e-01,
-9.7225e-01, -1.1985e+00],
[-2.3544e+00, -1.4067e+00, 1.2398e-01, ..., -6.9008e-01,
-6.7555e-01, -5.3340e-01]],
[[ 3.0629e+00, 4.0924e+00, 3.1240e+00, ..., 1.2765e+00,
1.1109e+00, 2.6066e+00],
[ 4.2185e-01, 8.2697e-01, 2.0594e+00, ..., 8.4622e-01,
5.9249e-02, 2.6636e+00],
[ 9.1099e-01, 1.2614e+00, 1.8484e+00, ..., -3.9561e-01,
-1.4569e+00, 1.8493e+00],
...,
[ 1.3651e+00, -1.4354e+00, -1.4177e+00, ..., -2.6466e-01,
5.4304e-01, 1.5068e+00],
[ 9.6647e-01, 2.3398e-01, -8.4278e-01, ..., -2.0564e-01,
7.7237e-01, 1.1782e+00],
[ 1.9679e+00, 6.1015e-01, -1.9124e-02, ..., 5.7524e-01,
1.3228e+00, 1.3694e+00]],
...,
[[-2.1277e+00, -7.4378e+00, -5.9012e+00, ..., -2.8258e+00,
-4.4184e+00, -4.3255e+00],
[-2.8327e+00, -6.5911e+00, -4.6328e+00, ..., -8.0197e+00,
-7.4601e+00, -3.3065e+00],
[-2.7743e-01, -2.8901e+00, -3.3937e+00, ..., -1.3642e+01,
-7.5603e+00, -1.3570e+00],
...,
[-1.1294e+00, -2.8317e+00, -5.2295e+00, ..., -3.8315e+00,
-2.9802e+00, -2.7477e+00],
[-1.4161e+00, -3.7238e+00, -4.5265e+00, ..., -5.0806e+00,
-4.1278e+00, -4.3887e+00],
[-2.2683e+00, -4.7296e+00, -2.1385e+00, ..., -3.3413e+00,
-2.3963e+00, -2.4199e+00]],
[[-7.0970e-01, -4.5513e-02, 2.2007e-01, ..., 1.4049e+00,
2.5706e+00, 1.1952e+00],
[ 1.7825e+00, 1.0807e+00, 1.1611e+00, ..., -2.2801e-01,
2.0943e+00, 1.8177e+00],
[ 1.2091e+00, 2.4673e-01, -6.2091e-01, ..., -1.4848e+00,
2.0646e+00, 1.2655e+00],
...,
[ 6.1803e-01, -1.4084e+00, 7.5075e-01, ..., 1.3797e+00,
6.7876e-01, 9.0063e-01],
[ 1.7436e+00, 1.2826e+00, 1.4729e+00, ..., 1.2841e+00,
7.2964e-01, 1.4082e+00],
[ 9.5426e-01, 1.2668e+00, 1.9268e+00, ..., 7.4192e-01,
6.8741e-01, 1.4274e+00]],
[[-5.1011e-01, 5.5664e-01, 1.0687e-01, ..., -5.2281e-01,
-2.8461e-01, -6.3825e-02],
[-1.1310e+00, -1.2113e+00, -1.6631e+00, ..., -1.4670e+00,
-9.8931e-01, -1.1272e+00],
[-2.0742e+00, -6.1799e-01, -1.2272e+00, ..., -2.0495e+00,
-1.4988e+00, -3.2015e-01],
...,
[-1.2684e+00, -6.1836e-01, 1.8754e+00, ..., -9.4495e-01,
-1.3726e+00, -8.6964e-01],
[-1.5023e+00, -8.9774e-01, 1.7981e-01, ..., -8.3341e-01,
-1.5272e+00, -7.5259e-01],
[-8.4925e-02, 4.8245e-01, 3.0865e-01, ..., -3.1518e-01,
-8.0290e-01, 8.9809e-04]]],
[[[ 8.7176e-01, 1.5297e+00, 1.2085e+00, ..., 2.1352e+00,
9.2698e-01, 6.8337e-01],
[-1.0117e+00, -1.9829e+00, -1.1670e+00, ..., -1.9006e-01,
-4.6355e-01, -2.0233e+00],
[-2.0509e-01, -7.9414e-01, 1.1848e-01, ..., -1.6924e+00,
4.6679e-01, 1.1901e+00],
...,
[-4.4879e-02, -1.9522e+00, -7.2438e-01, ..., -2.7775e-01,
-4.1603e-01, 2.0642e-01],
[-6.3971e-01, -2.5623e+00, -2.4497e-01, ..., -1.4122e-01,
-5.1084e-01, -9.5218e-02],
[-1.2308e-01, -1.9655e+00, -1.0889e-01, ..., -7.2846e-01,
-1.6332e+00, -2.9902e-01]],
[[-3.7438e+00, -7.9234e-01, -1.5791e+00, ..., -1.5071e+00,
-3.1774e+00, -1.7827e+00],
[-3.9780e+00, -2.8558e+00, -1.8198e+00, ..., -2.0209e+00,
-4.7806e+00, -1.3769e+00],
[-2.3973e+00, -1.7024e+00, -1.6993e+00, ..., -3.6840e+00,
-3.3446e+00, -8.1922e-01],
...,
[-1.5842e+00, -2.9756e+00, -3.6147e+00, ..., -4.7596e+00,
-2.6175e+00, -7.5143e-01],
[-3.4300e+00, -2.3881e+00, -3.3341e+00, ..., -4.0553e+00,
-2.3792e+00, -1.2883e+00],
[-2.4218e+00, -2.6922e+00, -3.0097e+00, ..., -2.7007e+00,
-2.1805e+00, -1.1850e+00]],
[[ 2.9114e+00, 4.2979e+00, 3.3979e+00, ..., 1.6950e-01,
1.3030e+00, 3.3479e+00],
[ 1.9300e-01, 1.1023e+00, 2.3031e+00, ..., -1.3234e+00,
-1.0513e+00, 1.9547e+00],
[ 9.6079e-01, 1.5154e+00, 1.6515e+00, ..., -2.0772e+00,
5.8190e-01, 1.5171e+00],
...,
[ 9.0308e-01, 2.2530e+00, 9.3968e-01, ..., -5.5637e-02,
5.3166e-01, 2.3607e+00],
[ 5.0349e-01, 2.9470e+00, 1.6387e+00, ..., 2.3387e-01,
7.4502e-01, 1.9415e+00],
[ 1.6540e+00, 2.1509e+00, 2.8408e+00, ..., 1.3331e+00,
1.6698e+00, 1.8262e+00]],
...,
[[-1.8952e+00, -7.3625e+00, -5.4799e+00, ..., -3.5547e+00,
-2.1122e+00, -5.6361e+00],
[-2.4232e+00, -7.0377e+00, -4.2805e+00, ..., -1.4288e+00,
-3.1228e+00, -5.7025e+00],
[-9.4070e-01, -4.4671e+00, -3.7245e+00, ..., 1.4290e+00,
-2.2419e+00, -5.0716e+00],
...,
[-2.6873e+00, -5.5369e+00, -5.5949e+00, ..., -8.5359e+00,
-6.0578e+00, -4.6428e+00],
[-3.0455e+00, -6.1985e+00, -5.5591e+00, ..., -7.4785e+00,
-7.3601e+00, -6.3345e+00],
[-1.9301e+00, -4.7859e+00, -2.7772e+00, ..., -2.7918e+00,
-3.4623e+00, -3.2390e+00]],
[[-1.9439e-01, 2.8597e-01, 5.7719e-01, ..., -1.9291e+00,
-2.3516e-01, 1.7836e+00],
[ 1.6457e+00, 1.0827e+00, 1.8377e+00, ..., -1.2842e+00,
4.8804e-01, 3.2700e+00],
[ 1.1825e+00, 2.7625e-01, 8.4668e-01, ..., -1.5364e+00,
-7.5776e-02, 2.6667e+00],
...,
[ 1.3907e+00, -5.8027e-01, -1.5719e-01, ..., 7.8543e-01,
2.2028e+00, 1.6237e+00],
[ 1.9885e+00, 1.2562e+00, 1.1402e+00, ..., 1.4824e+00,
1.9218e+00, 1.9792e+00],
[ 4.0162e-01, 2.8523e-01, 6.9942e-01, ..., 9.5530e-01,
9.5078e-01, 1.6899e+00]],
[[-5.2993e-01, 3.0551e-01, 1.2471e-01, ..., 1.1073e-01,
3.1556e-02, 1.5383e+00],
[-1.3373e+00, -1.8021e+00, -1.6672e+00, ..., 9.8232e-02,
-1.6584e+00, -6.9452e-01],
[-2.2410e+00, -8.4936e-01, 3.1098e-01, ..., -4.8481e-01,
-5.8604e-01, 5.9576e-01],
...,
[-2.0310e+00, -2.7989e+00, -1.8230e+00, ..., -2.4305e+00,
-2.1180e+00, -8.5478e-01],
[-1.6190e+00, -2.6067e+00, -1.2634e+00, ..., -2.0793e+00,
-1.7959e+00, -6.2956e-01],
[-8.5353e-01, -1.1440e+00, -5.6471e-01, ..., -1.1155e+00,
-8.3302e-01, -1.2923e-01]]],
[[[ 8.1532e-01, 1.2745e+00, 5.5242e-01, ..., 3.1698e-02,
4.8469e-02, -4.7213e-01],
[-6.5137e-01, -8.2713e-01, 1.2317e+00, ..., -9.7987e-01,
1.0353e+00, -3.2511e-01],
[-1.7738e-02, -3.9973e-01, 1.2370e+00, ..., 5.2312e-01,
5.1265e-01, 9.4700e-01],
...,
[ 1.6177e-01, -8.0954e-01, -7.1265e-01, ..., 3.0775e-01,
5.8237e-01, 5.7221e-01],
[-6.0922e-01, -1.9893e+00, -2.1046e+00, ..., -4.3012e-01,
-2.5085e-01, -9.8549e-02],
[-1.6962e-01, -1.7733e+00, -1.5454e+00, ..., -1.1127e+00,
-1.6383e+00, -3.2010e-01]],
[[-3.3537e+00, -2.6452e-02, -1.7368e+00, ..., -2.3655e+00,
-3.3708e+00, -7.3975e-01],
[-3.0300e+00, -1.1363e+00, -1.1933e+00, ..., -1.4025e+00,
-2.2566e+00, -1.6359e-01],
[-1.9058e+00, -8.8885e-01, -3.8218e-01, ..., -2.2075e+00,
-1.7955e+00, -5.0003e-01],
...,
[-1.4649e+00, -9.1172e-01, -2.4646e+00, ..., -1.5129e+00,
-1.9319e+00, -6.3625e-01],
[-3.3531e+00, -1.2639e+00, -2.4426e+00, ..., -2.1641e+00,
-2.4583e+00, -1.2478e+00],
[-2.3077e+00, -1.6217e+00, -1.6646e+00, ..., -2.0819e+00,
-2.0922e+00, -1.0548e+00]],
[[ 2.1995e+00, 2.7086e+00, -4.1944e-01, ..., 2.1424e+00,
1.3882e-01, 2.1901e+00],
[ 3.4354e-01, 8.0617e-01, -3.1242e+00, ..., 2.4146e+00,
-2.5281e-01, 2.1290e+00],
[ 9.5850e-01, 1.2063e+00, -1.9811e+00, ..., 7.5321e-01,
1.1795e+00, 2.3554e+00],
...,
[ 1.1161e+00, 1.5051e+00, 2.3569e+00, ..., 6.9355e-01,
1.2950e+00, 2.3590e+00],
[ 4.7133e-01, 2.1292e+00, 1.8783e+00, ..., 1.0613e-01,
1.0673e+00, 1.8295e+00],
[ 1.6061e+00, 1.8933e+00, 2.0007e+00, ..., 1.2862e+00,
1.8047e+00, 1.7668e+00]],
...,
[[-1.9280e+00, -6.4300e+00, -4.8874e+00, ..., -3.3468e+00,
-2.7765e+00, -3.7347e+00],
[-2.4881e+00, -7.4889e+00, -5.3270e+00, ..., -4.4123e+00,
-4.5277e+00, -4.8906e+00],
[-1.4370e+00, -4.1874e+00, -2.5989e+00, ..., -5.4962e+00,
-3.0633e+00, -2.0730e+00],
...,
[-2.3960e+00, -5.2863e+00, -4.3350e+00, ..., -4.5054e+00,
-4.5684e+00, -4.8003e+00],
[-2.9818e+00, -6.0048e+00, -4.9942e+00, ..., -6.4582e+00,
-5.8652e+00, -6.2757e+00],
[-1.8372e+00, -4.5400e+00, -2.6394e+00, ..., -3.0638e+00,
-2.7523e+00, -3.2011e+00]],
[[-3.6773e-01, 6.8561e-01, 6.3035e-01, ..., 6.0684e-01,
7.8325e-01, 1.6642e+00],
[ 1.1141e+00, 3.7712e-01, 1.0973e+00, ..., 1.8338e+00,
1.7630e+00, 2.5227e+00],
[ 1.0128e+00, 1.9481e-01, 1.6329e+00, ..., 9.9674e-01,
1.5980e+00, 1.8293e+00],
...,
[ 1.4071e+00, 3.8361e-01, -5.4325e-01, ..., 1.6783e+00,
1.1184e+00, 1.6049e+00],
[ 1.9150e+00, 1.8713e+00, 1.3263e+00, ..., 1.6877e+00,
1.3026e+00, 1.9780e+00],
[ 4.0612e-01, 9.0588e-01, 1.2089e+00, ..., 6.8129e-01,
6.8586e-01, 1.6514e+00]],
[[-6.5378e-01, -2.1514e-01, -1.0551e-01, ..., -7.6953e-01,
-5.1729e-01, 6.2551e-01],
[-1.2804e+00, -9.2274e-01, -1.2837e+00, ..., -9.9709e-01,
5.0602e-01, 8.2590e-01],
[-1.8080e+00, -1.8843e-01, -6.9178e-01, ..., 6.1491e-01,
5.6465e-01, 1.6886e-02],
...,
[-1.9561e+00, -2.0088e+00, -1.6885e+00, ..., -4.5609e-01,
-7.8195e-01, -5.6902e-01],
[-1.5837e+00, -2.2598e+00, -2.2575e+00, ..., -6.2574e-01,
-1.3119e+00, -6.9504e-01],
[-8.6403e-01, -8.5674e-01, -1.0238e+00, ..., -7.1580e-01,
-9.5457e-01, -1.3539e-01]]],
[[[ 7.9169e-01, 9.0562e-01, 4.7602e-01, ..., 6.9691e-01,
1.0325e+00, 4.8133e-01],
[-1.0877e+00, -9.8752e-01, -1.9745e-02, ..., -2.8009e-01,
-1.5765e-01, -3.2225e-01],
[-2.7376e-01, -8.5781e-01, -9.9809e-02, ..., -5.5274e-01,
-1.4142e-01, 6.3463e-01],
...,
[-3.0656e+00, 1.8448e+00, 7.8172e-02, ..., 7.4176e-01,
1.2757e-03, 4.2894e-01],
[-4.6204e+00, 1.3733e+00, 1.3484e+00, ..., 1.1676e-02,
-5.3218e-02, 1.0310e-02],
[ 1.1885e+00, -2.4429e+00, -4.8096e-01, ..., -4.1768e-01,
-1.0211e+00, 1.5005e-01]],
[[-3.7943e+00, -1.8888e+00, -2.8717e+00, ..., -1.4803e+00,
-1.9201e+00, -1.7079e+00],
[-3.0113e+00, -1.7009e+00, -8.3265e-01, ..., -1.4678e+00,
-1.9856e+00, -1.9835e+00],
[-2.0723e+00, -1.4859e+00, -1.4189e+00, ..., -1.2514e+00,
-1.5626e+00, -1.3335e+00],
...,
[-1.0386e+00, -2.3508e+00, -4.4434e+00, ..., -1.4760e+00,
-1.0846e+00, -7.3979e-01],
[-3.4124e+00, 2.0791e-01, -9.5603e-01, ..., -1.3347e+00,
-9.0878e-01, -1.1460e+00],
[-6.0730e+00, -5.0671e+00, -4.8965e+00, ..., -7.0452e-01,
-6.5585e-01, -4.6372e-01]],
[[ 2.6065e+00, 3.2336e+00, 3.0814e+00, ..., 1.4783e+00,
8.3336e-01, 2.4622e+00],
[ 7.0809e-01, 1.2214e+00, 2.3848e+00, ..., 1.2607e+00,
6.0679e-01, 2.4887e+00],
[ 1.0893e+00, 8.1105e-01, 1.5113e+00, ..., 2.1107e-01,
9.2484e-01, 1.7279e+00],
...,
[-6.3805e+00, -2.7262e+00, -2.3133e+00, ..., -1.6873e+00,
1.8474e-01, 1.1630e+00],
[-7.1297e+00, -2.1569e+00, -2.0531e+00, ..., -5.9253e-01,
6.1793e-01, 9.7082e-01],
[-3.5439e-01, -3.0880e+00, -2.1349e+00, ..., 2.8151e-01,
1.1855e+00, 1.3455e+00]],
...,
[[-2.5261e+00, -7.5471e+00, -4.8357e+00, ..., -5.6458e+00,
-3.5752e+00, -4.0183e+00],
[-3.1264e+00, -9.1168e+00, -6.2290e+00, ..., -5.6568e+00,
-3.6054e+00, -2.8929e+00],
[-2.2017e+00, -6.3074e+00, -3.6714e+00, ..., -4.2983e+00,
-2.2730e+00, -1.1801e+00],
...,
[-2.7069e+00, -1.8830e+01, -1.3335e+00, ..., -5.5187e+00,
-4.1050e+00, -2.4260e+00],
[-3.4055e+00, -1.4621e+01, -6.6855e+00, ..., -6.5565e+00,
-4.9548e+00, -4.1153e+00],
[-2.0532e+00, -6.9815e+00, -6.1822e+00, ..., -4.3147e+00,
-2.5476e+00, -2.1902e+00]],
[[-6.8694e-01, -7.6231e-02, 2.8429e-01, ..., 1.9516e+00,
1.6909e+00, 1.1543e+00],
[ 1.3923e+00, 8.4682e-01, 1.1335e+00, ..., 2.0059e+00,
1.1665e+00, 1.7765e+00],
[ 1.1287e+00, 3.0117e-01, 4.3502e-01, ..., 1.6592e+00,
5.8780e-01, 1.2904e+00],
...,
[ 1.2357e-01, -5.8864e+00, 4.3721e-01, ..., 1.6605e+00,
7.1994e-01, 1.1383e+00],
[ 1.8016e+00, -2.9613e+00, -1.5954e+00, ..., 1.6337e+00,
9.3072e-01, 1.4658e+00],
[ 1.7918e-01, -1.0116e-01, -2.1136e+00, ..., 9.9827e-01,
7.5125e-01, 1.4048e+00]],
[[-7.5118e-01, -7.2536e-01, -6.3085e-01, ..., -2.6379e-01,
-6.0131e-01, -4.8919e-02],
[-1.3832e+00, -1.4433e+00, -6.8039e-01, ..., -1.5429e+00,
-1.4682e+00, -1.0712e+00],
[-2.1709e+00, -6.7573e-01, 1.7570e-01, ..., -6.0675e-01,
-1.1284e+00, -1.8129e-01],
...,
[ 9.0844e-01, -6.2046e+00, -3.1503e+00, ..., -1.9945e+00,
-2.7895e+00, -8.9806e-01],
[-2.3912e+00, -5.0190e+00, -1.9609e+00, ..., -1.1398e+00,
-1.2749e+00, -6.4458e-01],
[-1.7998e+00, -1.7169e+00, -2.3887e+00, ..., -3.3474e-01,
-7.0956e-01, -4.3611e-02]]]], device='cuda:0'),
tensor([[[[ 6.8979e-01, 1.0521e+00, 6.6892e-01, ..., 6.6158e-01,
2.0382e+00, 5.1530e-01],
[-3.1587e-01, -1.4992e+00, -6.0692e-01, ..., 1.0469e-01,
-6.1504e-01, -2.0624e+00],
[ 1.3994e-01, -9.5414e-01, -5.1958e-01, ..., -3.5746e+00,
-6.4587e+00, -4.0945e-01],
...,
[-4.3151e-02, -1.4693e+00, -2.8537e+00, ..., 1.7670e-01,
4.9688e-01, 2.8682e-01],
[-4.9386e-01, -1.8878e+00, -2.2353e+00, ..., 2.6791e-01,
1.2046e+00, 2.5143e-01],
[ 4.7458e-01, -5.0239e-01, -1.4464e+00, ..., 1.9330e-02,
-3.0777e-01, 5.2325e-01]],
[[-3.2412e+00, -1.6565e+00, -3.1740e+00, ..., -2.4779e+00,
-1.3434e+00, 3.0873e-02],
[-4.4759e+00, -1.8816e+00, -2.3382e+00, ..., -4.0787e+00,
-2.2143e+00, -1.7519e+00],
[-4.3762e+00, -2.6083e+00, -3.1773e+00, ..., -6.3498e+00,
-5.5722e+00, -3.4889e+00],
...,
[-2.9279e+00, -2.1507e+00, -5.5040e+00, ..., -2.0705e+00,
-2.0233e+00, 2.7768e-01],
[-3.7281e+00, -1.3369e+00, -3.7162e+00, ..., -5.2368e-01,
-9.4401e-01, 3.1014e-01],
[-1.1694e+00, -8.4344e-01, -2.2960e+00, ..., -6.2209e-01,
-5.3206e-01, -2.6915e-03]],
[[ 1.7894e+00, 2.8887e+00, 2.4335e+00, ..., 7.9401e-01,
-1.3212e+00, 7.4051e-02],
[ 2.4612e-01, 1.2781e+00, 1.7026e+00, ..., -4.8067e+00,
-9.7683e+00, -5.1985e+00],
[ 7.7057e-01, 1.4813e+00, 7.9243e-01, ..., -6.2739e+00,
-9.7568e+00, -3.5854e+00],
...,
[ 1.0794e+00, 1.2284e+00, 3.0177e+00, ..., 7.7359e-03,
1.3345e-01, 7.6244e-01],
[ 6.8170e-01, 1.8422e+00, 2.0002e+00, ..., 1.7919e-01,
5.8232e-01, 1.1505e+00],
[ 1.5836e+00, 1.5674e+00, 1.6799e+00, ..., 7.0277e-01,
6.1540e-01, 1.3022e+00]],
...,
[[-1.4712e+00, -4.6585e+00, -4.0989e+00, ..., -4.3082e+00,
-2.6721e+00, -2.9325e+00],
[-1.9517e+00, -4.1103e+00, -4.5637e+00, ..., -9.7669e+00,
-3.9338e+00, -7.0493e+00],
[-1.0566e+00, -3.3739e+00, -3.9521e+00, ..., -1.0105e+01,
-2.6525e+00, -1.0167e+01],
...,
[-1.6858e+00, -4.4470e+00, -4.6400e+00, ..., -3.9441e+00,
-3.6675e+00, -3.3905e+00],
[-1.6141e+00, -4.7664e+00, -5.7061e+00, ..., -6.1447e+00,
-3.5746e+00, -5.1107e+00],
[-1.7217e+00, -5.1357e+00, -4.3417e+00, ..., -4.7677e+00,
-1.9225e+00, -3.2257e+00]],
[[ 7.0466e-01, 4.6718e-01, -1.8485e-01, ..., -2.1626e+00,
-1.2038e+00, 9.9012e-01],
[ 1.2127e+00, 1.0033e+00, 4.7444e-01, ..., -1.3696e+00,
-2.8422e+00, -1.4292e+00],
[ 5.3515e-01, 5.0374e-01, 6.1597e-02, ..., -5.9793e-01,
-1.2332e+00, -4.7145e+00],
...,
[ 7.7781e-01, -2.2357e-01, -2.3463e+00, ..., 2.7230e+00,
1.5593e+00, 2.1166e+00],
[ 1.4822e+00, 1.4768e+00, -2.9079e-01, ..., 2.8096e+00,
1.4173e+00, 2.5238e+00],
[ 6.2241e-01, 9.2920e-01, 5.6753e-01, ..., 1.4378e+00,
1.2269e+00, 1.3716e+00]],
[[-3.2329e-01, -1.3985e-01, -9.5645e-01, ..., -8.7113e-01,
2.0206e+00, 4.2939e-01],
[-1.4243e+00, -1.3435e+00, -2.0233e+00, ..., -3.4239e+00,
1.4412e+00, 5.6052e-01],
[-1.7300e+00, -9.0725e-01, -9.8375e-01, ..., -5.9640e+00,
-3.2766e-01, -3.4139e-02],
...,
[-1.7766e+00, -1.7160e+00, -2.2076e+00, ..., -1.8881e+00,
-1.2822e+00, -1.4714e+00],
[-1.4690e+00, -1.6041e+00, -2.5448e+00, ..., -1.7834e+00,
-8.7714e-01, -6.5501e-02],
[-9.1837e-02, 1.2261e-01, -1.0705e+00, ..., -1.3796e+00,
-7.2212e-01, 2.6533e-01]]],
[[[ 5.7401e-01, 8.5736e-01, 4.8014e-01, ..., 6.0688e-01,
4.6273e-01, 5.2380e-02],
[-3.0217e-01, -7.7561e-01, -3.8827e-01, ..., 3.8194e-01,
3.9805e-02, -4.1554e-01],
[-1.6962e-01, -3.2166e-02, 2.7465e-01, ..., -1.0327e-01,
2.1455e-01, 7.4078e-01],
...,
[ 6.5167e-02, -3.3283e-01, 2.0167e+00, ..., 1.1375e+00,
1.1603e+00, 9.0073e-01],
[-3.8385e-01, -1.9752e+00, -3.3805e-01, ..., -6.0260e-01,
8.3314e-01, 1.6063e-01],
[ 4.5785e-01, -9.9743e-01, -1.4447e+00, ..., 5.0051e-01,
-4.2197e-01, 2.2128e-01]],
[[-2.3140e+00, -7.3515e-01, -1.6189e+00, ..., -2.1334e+00,
-2.4660e+00, -1.6972e+00],
[-2.4976e+00, -6.8158e-01, -1.5771e+00, ..., -1.9031e+00,
-2.6566e+00, -1.9010e+00],
[-2.4297e+00, -1.3023e+00, -2.3382e+00, ..., -1.7006e+00,
-2.1852e+00, -1.2676e+00],
...,
[-1.0681e+00, -6.9108e-01, -1.6102e+00, ..., -1.4418e+00,
-1.0683e+00, 1.1740e-01],
[-2.2615e+00, -1.0768e-01, -1.5890e+00, ..., -2.5679e+00,
-1.5921e+00, -1.6737e-01],
[-1.0245e+00, -1.4035e+00, -2.1518e+00, ..., -3.2105e+00,
-1.7515e+00, -5.7955e-01]],
[[ 1.7563e+00, 2.3908e+00, 1.6846e+00, ..., 1.5481e+00,
6.5132e-01, 2.2151e+00],
[-2.6054e-01, 4.3447e-01, 7.6635e-01, ..., 3.9689e-01,
3.0843e-01, 2.4666e+00],
[ 1.2476e-01, 3.3217e-01, 1.1093e+00, ..., -7.8117e-02,
8.1021e-01, 2.1140e+00],
...,
[ 1.2263e+00, 1.9994e+00, 1.1933e+00, ..., 3.3222e-02,
-6.3118e-01, 1.3576e+00],
[ 8.0139e-01, 2.3366e+00, 2.7616e+00, ..., -1.3811e+00,
-6.1634e-01, 1.0211e+00],
[ 1.7875e+00, 8.6731e-01, 1.7303e+00, ..., -3.3225e-01,
-9.7587e-02, 1.2790e+00]],
...,
[[-1.7767e+00, -4.6868e+00, -3.0413e+00, ..., -1.2768e+00,
-1.7007e+00, -3.3796e+00],
[-2.1898e+00, -4.9890e+00, -2.5799e+00, ..., -2.3159e+00,
-2.0671e+00, -3.1698e+00],
[-1.3190e+00, -4.9996e+00, -2.8682e+00, ..., -2.4470e+00,
-2.1595e+00, -2.0790e+00],
...,
[-2.0895e+00, -5.4585e+00, -3.5408e+00, ..., -2.5932e+00,
-2.3382e+00, -2.9199e+00],
[-2.9132e+00, -7.4579e+00, -4.2996e+00, ..., -2.5588e+00,
-3.2327e+00, -4.3902e+00],
[-3.0366e+00, -7.4624e+00, -3.3160e+00, ..., -3.8676e+00,
-2.1940e+00, -2.7938e+00]],
[[ 8.9312e-01, 1.4808e+00, 8.6182e-01, ..., 9.0296e-01,
1.4503e+00, 1.3708e+00],
[ 1.0945e+00, 9.9759e-01, 7.0988e-01, ..., 7.0887e-01,
7.9130e-01, 2.0192e+00],
[ 6.6266e-01, 2.6984e-01, -4.6256e-01, ..., 8.0664e-01,
1.6740e-01, 1.3262e+00],
...,
[ 1.6486e+00, -1.1157e-01, 4.5327e-01, ..., -2.7334e-01,
1.7359e+00, 1.8150e+00],
[ 2.2544e+00, 1.3746e+00, 6.4324e-01, ..., -1.5205e+00,
1.5806e+00, 2.2821e+00],
[ 1.0177e+00, 4.0331e-01, 5.7083e-01, ..., -1.5961e+00,
1.1181e+00, 1.1724e+00]],
[[-9.4856e-02, -6.4602e-01, -1.2836e+00, ..., -3.3131e-01,
-9.0000e-01, -9.6847e-02],
[-9.0574e-01, -2.4157e+00, -2.8345e+00, ..., -1.4664e+00,
-1.1407e+00, -7.6867e-01],
[-1.2962e+00, -1.3034e+00, -1.1635e+00, ..., -1.9293e-01,
-1.1348e-01, -2.0660e-01],
...,
[-2.5807e+00, -1.3202e+00, -3.4967e-01, ..., -4.5529e-01,
-6.2790e-01, 1.0596e-01],
[-2.4742e+00, -2.2844e+00, -5.8594e-01, ..., -5.6163e-01,
-1.0429e+00, 3.3436e-01],
[-4.2322e-01, -3.3519e-01, -6.5124e-01, ..., 6.5528e-01,
-1.0633e+00, 4.8218e-02]]],
[[[ 1.1651e+00, 1.7112e+00, 6.8622e-01, ..., 1.4482e+00,
1.1245e+00, 9.9797e-01],
[-1.0908e+00, -2.1807e+00, -5.2944e-01, ..., -4.8473e-01,
-5.8719e-01, -1.5914e+00],
[-5.6991e-01, -2.2766e+00, -1.4015e+00, ..., -7.1752e-02,
8.7102e-01, 6.2487e-01],
...,
[-1.0288e-01, -9.7138e-01, -1.1997e+00, ..., -3.6406e-01,
9.2076e-01, 5.4848e-01],
[ 1.1438e-01, -1.7631e+00, -1.5554e+00, ..., -5.9265e-01,
2.1482e-02, 5.3562e-01],
[ 7.0964e-01, -3.8186e-01, -6.3217e-01, ..., -3.4208e-01,
-9.5757e-01, -1.1582e-01]],
[[-4.1532e+00, -2.5998e+00, -3.2519e+00, ..., -1.5450e+00,
-2.4214e+00, -1.2268e+00],
[-4.6442e+00, -5.3185e+00, -3.9671e+00, ..., -2.3407e+00,
-3.1112e+00, -9.6300e-01],
[-3.6152e+00, -3.3980e+00, -3.5622e-01, ..., -1.7573e+00,
-2.4662e+00, -1.0228e+00],
...,
[ 2.9414e-01, 1.3089e+00, 3.3875e-01, ..., -2.6236e+00,
-5.1078e+00, -2.3055e+00],
[ 6.4961e-02, 1.2677e+00, -1.1190e+00, ..., -2.1273e+00,
-2.1381e+00, -1.5858e+00],
[-3.7430e-01, -5.5338e-02, -1.2657e+00, ..., -1.7925e+00,
-1.7512e+00, -9.8833e-01]],
[[ 3.1307e+00, 4.5989e+00, 3.6457e+00, ..., 2.9206e+00,
1.0653e+00, 3.0752e+00],
[ 1.0945e+00, 2.2448e+00, 2.0499e+00, ..., 9.6993e-01,
-8.0260e-01, 2.7746e+00],
[ 1.9673e+00, 2.5359e+00, -2.5567e+00, ..., 6.1336e-01,
-1.4434e-01, 1.7701e+00],
...,
[ 1.4255e-01, -3.2072e-01, -6.0850e-02, ..., -1.7740e+00,
-3.8204e+00, -1.6594e+00],
[-1.0979e+00, 4.5935e-02, -2.4309e-01, ..., -7.7172e-01,
7.3727e-01, 1.3318e+00],
[-1.5691e-01, 1.5517e-01, 4.1490e-02, ..., 1.3990e+00,
2.0535e+00, 1.6954e+00]],
...,
[[-1.8906e+00, -7.8967e+00, -6.5769e+00, ..., -4.0861e+00,
-3.2919e+00, -5.6292e+00],
[-2.7029e+00, -8.1134e+00, -6.2231e+00, ..., -3.6642e+00,
-2.7217e+00, -4.3869e+00],
[-1.1551e+00, -6.3994e+00, -5.2499e+00, ..., -3.6237e+00,
-3.3153e+00, -3.0458e+00],
...,
[-1.9348e+00, -4.4836e+00, -2.3721e+00, ..., -2.6532e+00,
-3.6463e+00, -4.7617e+00],
[-2.7354e+00, -5.5528e+00, -2.0284e+00, ..., -5.0059e+00,
-4.5378e+00, -5.8702e+00],
[-1.8258e+00, -4.7665e+00, -1.0886e+00, ..., -3.0060e+00,
-2.6911e+00, -3.7625e+00]],
[[-5.1421e-01, -2.1670e-01, -6.2280e-01, ..., 6.3712e-01,
8.7014e-01, 1.3860e+00],
[ 8.8450e-01, -7.9640e-01, -6.6845e-01, ..., 1.6393e+00,
2.5500e+00, 3.0259e+00],
[ 3.0047e-01, 8.9609e-01, 1.6529e+00, ..., 1.5338e+00,
1.5890e+00, 2.3308e+00],
...,
[ 1.5914e+00, 1.4463e+00, 5.4651e-01, ..., 6.4821e-02,
-3.9052e-01, 5.6728e-01],
[ 2.0058e+00, 1.6751e+00, 8.7821e-01, ..., 1.0810e+00,
4.8578e-01, 2.0608e+00],
[ 4.9618e-01, -1.7474e-01, 2.4584e-01, ..., 6.5388e-01,
6.7457e-01, 1.6911e+00]],
[[-8.6195e-01, -3.5489e-01, -4.4077e-02, ..., 6.0530e-01,
5.7599e-01, 1.3545e+00],
[-2.0939e+00, -2.8724e+00, -8.2869e-01, ..., -8.5812e-01,
-4.2300e-01, 1.7628e-01],
[-2.7206e+00, -1.9322e+00, 9.2115e-01, ..., 4.6671e-01,
-9.1271e-02, 6.3395e-01],
...,
[-1.6776e+00, -5.0157e-01, -4.7333e-01, ..., -2.3913e+00,
-2.2768e+00, -1.4272e+00],
[-5.3223e-01, -7.3420e-01, -1.9227e+00, ..., -1.3838e+00,
-1.3527e+00, -5.6491e-01],
[-1.0340e-01, -1.2128e+00, -1.8950e+00, ..., -6.4741e-01,
-5.2427e-01, 2.9334e-01]]],
[[[ 9.6809e-01, 1.5755e+00, 1.1354e+00, ..., 7.9945e-01,
9.9374e-01, 4.4834e-01],
[-6.8788e-01, -1.4101e+00, -5.3136e-01, ..., -2.6756e-01,
-1.1182e-01, -3.6710e-01],
[ 1.0609e-02, -7.4002e-01, 2.5923e-01, ..., -1.0004e+00,
-2.6356e-01, 4.6968e-01],
...,
[ 4.2822e-01, -3.2925e-01, -1.0885e-01, ..., 4.5558e-02,
3.5164e-02, 4.9957e-01],
[-1.8737e-01, -1.1245e+00, -1.1729e-01, ..., -5.3930e-02,
7.5051e-02, -4.4929e-02],
[ 7.6280e-01, -2.8027e-01, -1.3892e-01, ..., -2.5748e-01,
-8.4799e-01, 1.4619e-01]],
[[-4.1846e+00, -1.2971e+00, -2.1440e+00, ..., -1.4310e+00,
-2.0625e+00, -1.8210e+00],
[-4.2009e+00, -3.6010e+00, -3.0610e+00, ..., -1.0586e+00,
-2.2109e+00, -2.1347e+00],
[-3.3746e+00, -4.2363e+00, -6.2002e+00, ..., -1.4213e+00,
-1.6362e+00, -1.4643e+00],
...,
[-8.8880e-01, -6.1596e-01, -1.2053e+00, ..., -6.7803e-01,
-6.9882e-01, -7.5335e-01],
[-2.9081e+00, -4.3029e-01, -8.6466e-01, ..., -6.2938e-01,
-9.7817e-01, -1.1819e+00],
[-1.9779e+00, -1.3790e+00, -1.3339e+00, ..., -5.6932e-01,
-6.6033e-01, -5.3396e-01]],
[[ 3.0051e+00, 4.0951e+00, 3.2196e+00, ..., 1.7426e+00,
8.2743e-01, 2.5811e+00],
[ 5.8339e-01, 1.2290e+00, 2.9355e+00, ..., 1.4550e+00,
6.8391e-01, 2.6346e+00],
[ 1.8167e+00, 2.4920e+00, 2.5009e+00, ..., 9.9355e-01,
1.3264e+00, 1.8746e+00],
...,
[ 3.2502e-01, 9.6727e-01, 1.1007e+00, ..., -1.2220e-01,
7.0705e-01, 1.5249e+00],
[ 3.2319e-01, 2.0031e+00, 1.4401e+00, ..., -2.5487e-01,
6.7250e-01, 1.1514e+00],
[ 1.3653e+00, 1.2654e+00, 1.6207e+00, ..., 6.3859e-01,
1.2808e+00, 1.3502e+00]],
...,
[[-2.0931e+00, -7.4350e+00, -5.9497e+00, ..., -5.0727e+00,
-3.6224e+00, -4.3548e+00],
[-3.1173e+00, -7.6703e+00, -6.1461e+00, ..., -5.0934e+00,
-3.7853e+00, -3.2915e+00],
[-5.4107e-01, -3.1833e+00, -4.3029e+00, ..., -5.2484e+00,
-3.5853e+00, -1.7527e+00],
...,
[-9.7626e-01, -2.2739e+00, -1.7650e+00, ..., -3.6161e+00,
-2.7571e+00, -2.7165e+00],
[-1.3409e+00, -3.6647e+00, -3.3390e+00, ..., -4.9076e+00,
-3.6505e+00, -4.1716e+00],
[-2.2603e+00, -5.0438e+00, -3.1494e+00, ..., -3.1892e+00,
-2.1563e+00, -2.2974e+00]],
[[-7.7213e-01, -1.3017e-01, 2.1184e-01, ..., 1.4932e+00,
1.7356e+00, 1.2009e+00],
[ 1.6210e+00, 6.6671e-01, 1.3422e+00, ..., 1.5799e+00,
1.1931e+00, 1.8488e+00],
[ 4.0786e-01, -1.8266e+00, -1.9682e+00, ..., 1.3194e+00,
6.7323e-01, 1.1248e+00],
...,
[ 1.3076e+00, -1.5837e-01, 9.5205e-01, ..., 1.5096e+00,
7.4785e-01, 1.0362e+00],
[ 1.8763e+00, 9.2862e-01, 1.7879e+00, ..., 1.2070e+00,
7.5799e-01, 1.3672e+00],
[ 1.0282e+00, 4.2139e-01, 9.6946e-01, ..., 6.9122e-01,
7.2129e-01, 1.4305e+00]],
[[-5.1281e-01, 5.5737e-01, 9.9576e-02, ..., -3.5617e-01,
-6.1814e-01, -7.6556e-02],
[-1.1779e+00, -1.2602e+00, -1.1733e+00, ..., -1.2662e+00,
-1.4547e+00, -1.1478e+00],
[-2.6623e+00, -1.2824e+00, -3.1348e-02, ..., -1.1767e+00,
-1.4858e+00, -4.8505e-01],
...,
[-1.7855e+00, -1.4794e+00, -9.7076e-01, ..., -1.2205e+00,
-1.3173e+00, -7.1158e-01],
[-1.7797e+00, -1.1462e+00, -8.2743e-01, ..., -1.0434e+00,
-1.5930e+00, -6.6495e-01],
[-3.3005e-01, 4.1637e-01, -2.4669e-01, ..., -3.4693e-01,
-8.1066e-01, 2.7891e-02]]],
[[[ 8.5537e-01, 1.4682e+00, 1.1903e+00, ..., 1.4323e+00,
1.0981e+00, 9.9651e-01],
[-1.0055e+00, -1.9645e+00, -1.1905e+00, ..., -5.1483e-01,
-6.6318e-01, -1.6195e+00],
[-2.0152e-01, -7.0069e-01, 6.7896e-02, ..., -4.9199e-01,
6.2098e-01, 7.7891e-01],
...,
[ 5.1360e-01, -2.8782e-01, 2.7519e-01, ..., 4.6557e-01,
2.2767e-01, 6.8892e-01],
[ 3.7523e-01, -7.7549e-01, 1.4571e-02, ..., 6.6950e-01,
6.3599e-01, -5.1852e-02],
[ 5.8887e-01, 2.9130e-01, 1.8179e-01, ..., 2.2904e-01,
-1.1505e-01, 6.2254e-01]],
[[-3.7588e+00, -8.1426e-01, -1.5797e+00, ..., -1.5907e+00,
-2.4981e+00, -1.2436e+00],
[-3.8911e+00, -2.8633e+00, -1.8069e+00, ..., -2.2796e+00,
-3.1730e+00, -9.3022e-01],
[-2.4346e+00, -1.7191e+00, -1.4823e+00, ..., -2.2118e+00,
-2.8764e+00, -1.0432e+00],
...,
[-9.7414e-01, -1.0307e+00, -1.3556e+00, ..., -1.7271e+00,
-1.2189e+00, -1.4682e-01],
[-1.3611e+00, -2.9243e-01, -2.3204e-01, ..., -7.2357e-01,
-1.1692e+00, -5.2137e-02],
[ 4.5423e-01, -1.6525e-01, -2.2402e-01, ..., 2.6725e-01,
1.1639e-01, 4.4676e-01]],
[[ 2.8645e+00, 4.2493e+00, 3.3743e+00, ..., 2.9174e+00,
1.0951e+00, 3.0499e+00],
[ 1.9052e-01, 1.1184e+00, 2.2234e+00, ..., 1.0385e+00,
-9.0700e-01, 2.7823e+00],
[ 9.3771e-01, 1.5975e+00, 1.7254e+00, ..., 9.7682e-01,
-1.2781e-01, 2.1058e+00],
...,
[ 4.6374e-01, 6.5883e-01, 5.5028e-01, ..., -6.6961e-02,
2.3073e-01, 1.4840e+00],
[ 1.9103e-01, 1.2470e+00, 5.0926e-01, ..., -3.0511e-01,
7.3453e-02, 1.1157e+00],
[ 1.3685e+00, 1.0115e+00, 9.3554e-01, ..., 7.9115e-01,
4.4864e-01, 1.1694e+00]],
...,
[[-1.7892e+00, -7.1658e+00, -5.4464e+00, ..., -4.0743e+00,
-3.3441e+00, -5.6456e+00],
[-2.3667e+00, -7.0086e+00, -4.2087e+00, ..., -3.5809e+00,
-2.7257e+00, -4.4137e+00],
[-1.0443e+00, -4.7069e+00, -3.7585e+00, ..., -2.3389e+00,
-2.8958e+00, -2.4749e+00],
...,
[-2.3101e+00, -5.6555e+00, -3.7739e+00, ..., -4.2332e+00,
-3.8434e+00, -3.5034e+00],
[-2.4933e+00, -6.7407e+00, -4.6589e+00, ..., -5.0364e+00,
-4.3061e+00, -4.8453e+00],
[-2.4378e+00, -6.6898e+00, -4.2149e+00, ..., -4.0902e+00,
-3.1129e+00, -4.0318e+00]],
[[-1.8922e-01, 2.9024e-01, 5.6315e-01, ..., 6.2930e-01,
8.1894e-01, 1.3892e+00],
[ 1.5760e+00, 1.0663e+00, 1.8649e+00, ..., 1.6953e+00,
2.5054e+00, 3.0691e+00],
[ 1.0929e+00, 2.6722e-01, 8.6081e-01, ..., 1.6186e+00,
1.5286e+00, 2.3849e+00],
...,
[ 1.1032e+00, -1.4313e-01, 1.4228e-01, ..., 1.1792e+00,
8.6989e-01, 1.2831e+00],
[ 1.6611e+00, 8.3053e-01, 1.2869e+00, ..., 9.6998e-01,
6.4981e-01, 2.0720e+00],
[ 1.1588e+00, 7.9491e-01, 1.4152e+00, ..., 1.1410e+00,
1.2469e+00, 1.3881e+00]],
[[-5.6755e-01, 2.6812e-01, 1.4176e-01, ..., 5.9531e-01,
5.2149e-01, 1.3536e+00],
[-1.3304e+00, -1.8221e+00, -1.5878e+00, ..., -9.3209e-01,
-5.0919e-01, 2.2126e-01],
[-2.1218e+00, -7.8261e-01, 3.7019e-01, ..., 1.3774e-01,
1.4976e-01, 7.7103e-01],
...,
[-1.5102e+00, -1.7666e+00, -9.3914e-01, ..., -3.7237e-01,
-8.2702e-01, -6.9520e-01],
[-1.2436e+00, -1.7466e+00, -1.0027e+00, ..., -7.3324e-01,
-1.2182e+00, -4.4892e-02],
[ 1.3152e-01, 9.2890e-02, 1.7519e-02, ..., 2.6419e-01,
-1.2016e-01, 7.5650e-01]]]], device='cuda:0'),
tensor([[[[ 8.7624e-01, 1.2131e+00, 4.4323e-01, ..., 1.2819e+00,
9.6715e-01, -6.3749e-01],
[-7.2354e-01, -1.7214e-01, -5.7061e-01, ..., -3.0529e+00,
1.5874e+00, -1.8653e-01],
[-2.2113e-01, 4.9498e-01, 3.7570e-02, ..., -3.5913e+00,
1.0123e+00, 1.2159e+00],
...,
[ 2.0362e-01, -4.9992e-01, -2.9385e-01, ..., 1.5545e+00,
1.8087e+00, 1.5558e+00],
[-5.7205e-01, -1.5941e+00, -7.4589e-01, ..., -1.5108e+00,
-2.0591e+00, -5.5873e-01],
[ 8.1664e-02, -1.5151e+00, -1.3132e+00, ..., -6.2116e-01,
-1.6280e+00, -3.0090e-01]],
[[-3.1741e+00, -7.0678e-01, 4.6488e-01, ..., -3.2756e+00,
-5.1942e+00, -1.2621e+00],
[-3.2208e+00, -1.6432e+00, 1.3670e+00, ..., -4.3415e+00,
-5.4691e+00, -7.3773e-01],
[-2.2346e+00, -1.3942e+00, 9.5913e-01, ..., -3.4771e+00,
-3.8875e+00, -5.0807e-01],
...,
[-1.1571e+00, 3.5006e-02, -4.3796e-01, ..., 8.3046e-02,
-2.7329e+00, -2.5204e+00],
[-3.0686e+00, -3.8307e-01, -2.2403e+00, ..., -6.9259e+00,
-6.6306e+00, -3.7824e+00],
[-2.2246e+00, -1.4119e+00, -2.4540e+00, ..., -2.7132e+00,
-2.6734e+00, -1.2761e+00]],
[[ 2.4649e+00, 1.9414e+00, 7.7772e-01, ..., -1.4286e+00,
-6.4111e-01, 2.1099e+00],
[ 8.5571e-01, -1.0124e+00, -6.2618e-01, ..., -1.8800e+00,
-1.3137e+00, 1.6746e+00],
[ 1.4525e+00, 6.0024e-01, -2.3462e-01, ..., -1.7422e+00,
6.7273e-01, 1.7702e+00],
...,
[ 1.2512e+00, 1.0436e+00, 7.4073e-01, ..., -4.3134e+00,
-2.6600e+00, 5.2667e-01],
[ 6.5595e-01, 1.6023e+00, 7.1954e-01, ..., -3.9198e+00,
-1.2028e+00, 1.5699e+00],
[ 1.7715e+00, 1.2364e+00, 1.6180e+00, ..., 1.7929e+00,
2.4014e+00, 1.8690e+00]],
...,
[[-2.0485e+00, -6.5957e+00, -3.3747e+00, ..., -3.2065e+00,
-4.0178e+00, -4.8848e+00],
[-2.4011e+00, -7.7081e+00, -3.0078e+00, ..., 8.1002e-01,
-5.0351e+00, -6.7538e+00],
[-1.2889e+00, -4.1957e+00, -1.2671e+00, ..., 1.3293e+00,
-4.2945e+00, -4.1343e+00],
...,
[-2.2998e+00, -4.9915e+00, -3.1986e+00, ..., -5.9288e+00,
-6.7397e+00, -7.8940e+00],
[-2.8575e+00, -5.6787e+00, -4.4804e+00, ..., -6.2506e+00,
-6.8159e+00, -6.2404e+00],
[-1.8866e+00, -4.4947e+00, -2.1974e+00, ..., -4.5783e+00,
-4.2276e+00, -4.7617e+00]],
[[-4.2398e-01, 8.8485e-01, 1.4522e+00, ..., -3.1602e+00,
-2.1681e-01, 2.4086e+00],
[ 9.5613e-01, 9.2012e-01, 1.1233e+00, ..., -2.9807e+00,
-9.0669e-01, 3.5253e+00],
[ 8.3218e-01, 3.8128e-01, 1.2594e+00, ..., -1.5656e+00,
-7.8837e-01, 2.2008e+00],
...,
[ 1.4212e+00, 6.3467e-01, 2.9587e-01, ..., -3.7007e-01,
-6.0278e-02, 5.1903e-01],
[ 2.1195e+00, 1.9514e+00, 6.7615e-01, ..., -1.5337e+00,
-8.6280e-01, 1.5322e+00],
[ 5.2958e-01, 9.3831e-01, 5.5741e-01, ..., 1.2055e+00,
1.0652e+00, 2.0776e+00]],
[[-7.0654e-01, -1.0375e-01, 5.6528e-01, ..., 1.4981e+00,
-7.9482e-01, 5.0472e-01],
[-1.3954e+00, -1.3634e+00, -1.8870e-01, ..., 9.3747e-02,
1.2739e-03, 1.4718e-01],
[-1.9543e+00, -3.7367e-01, 1.3284e-01, ..., 1.1909e-01,
5.2371e-01, -1.9710e-01],
...,
[-1.9934e+00, -1.8533e+00, -1.0787e+00, ..., -6.3309e-01,
-1.7695e-01, -4.7147e-01],
[-1.5954e+00, -1.7046e+00, -1.2850e+00, ..., -4.0227e+00,
-3.5886e+00, -1.9622e+00],
[-7.1349e-01, -7.1464e-01, -1.2143e+00, ..., -6.5646e-01,
-8.3000e-01, 6.5976e-01]]],
[[[ 9.6330e-01, 1.5896e+00, 1.1318e+00, ..., 4.5420e-01,
8.4819e-01, 4.7456e-01],
[-8.7207e-01, -2.0428e+00, -8.5927e-01, ..., 1.3027e-01,
-2.5733e-01, -3.8265e-01],
[ 4.0280e-02, -5.4864e-01, -6.7623e-02, ..., -9.5461e-02,
-4.1383e-01, 6.0332e-01],
...,
[ 4.0697e-01, 9.2643e-01, 2.3479e-01, ..., -7.8297e-02,
-1.1057e-01, 3.4976e-01],
[-4.0611e-01, -9.2962e-01, 6.6028e-01, ..., -3.4059e-02,
7.9950e-02, -9.6239e-02],
[ 7.2867e-01, -2.7721e-01, -1.2684e-01, ..., -3.5803e-01,
-9.6185e-01, 1.5017e-01]],
[[-4.2882e+00, -1.2213e+00, -2.1448e+00, ..., -1.8788e+00,
-2.1240e+00, -1.8030e+00],
[-4.1728e+00, -2.8091e+00, -2.3777e+00, ..., -1.7100e+00,
-2.3691e+00, -2.1381e+00],
[-2.6355e+00, -2.3438e+00, -3.8858e+00, ..., -2.3623e+00,
-1.9539e+00, -1.4196e+00],
...,
[-9.7142e-01, -1.4293e-01, -5.6092e-01, ..., -6.9143e-01,
-9.8521e-01, -9.0372e-01],
[-2.8821e+00, -8.2568e-02, 4.0366e-02, ..., -6.8456e-01,
-1.0846e+00, -1.2510e+00],
[-2.0207e+00, -1.3996e+00, -1.3459e+00, ..., -7.0931e-01,
-6.6731e-01, -4.9390e-01]],
[[ 3.0482e+00, 4.1738e+00, 3.2455e+00, ..., 1.9503e+00,
8.4651e-01, 2.5877e+00],
[ 3.8637e-01, 8.9831e-01, 2.2363e+00, ..., 1.5941e+00,
4.6346e-01, 2.6403e+00],
[ 8.1505e-01, 1.2922e+00, 1.8792e+00, ..., 4.4267e-01,
7.3382e-01, 1.8696e+00],
...,
[ 5.5098e-01, 9.6480e-01, -7.4903e-02, ..., -2.5405e-01,
6.1100e-01, 1.5334e+00],
[ 2.6492e-01, 2.4485e+00, 1.7483e+00, ..., -1.6318e-01,
8.2912e-01, 1.1532e+00],
[ 1.3734e+00, 1.4115e+00, 1.6801e+00, ..., 5.5410e-01,
1.3008e+00, 1.3543e+00]],
...,
[[-2.1201e+00, -7.4744e+00, -6.0100e+00, ..., -3.9668e+00,
-3.4423e+00, -4.4246e+00],
[-2.7164e+00, -6.5614e+00, -4.6759e+00, ..., -4.7849e+00,
-3.9608e+00, -3.3807e+00],
[-1.7982e-01, -2.9532e+00, -4.0351e+00, ..., -4.3968e+00,
-2.6437e+00, -1.4415e+00],
...,
[-1.0051e+00, -2.8937e+00, -2.0634e+00, ..., -4.1788e+00,
-3.2107e+00, -2.8025e+00],
[-1.3087e+00, -3.7292e+00, -3.5782e+00, ..., -5.2884e+00,
-4.2657e+00, -4.3181e+00],
[-2.1773e+00, -4.9756e+00, -3.0651e+00, ..., -3.4303e+00,
-2.4597e+00, -2.3784e+00]],
[[-7.4031e-01, -4.2242e-02, 2.1110e-01, ..., 1.5940e+00,
1.8165e+00, 1.1909e+00],
[ 1.7758e+00, 1.1297e+00, 1.3779e+00, ..., 1.3386e+00,
1.3990e+00, 1.8613e+00],
[ 1.1886e+00, -3.7903e-01, 2.6505e-01, ..., 6.7651e-01,
9.1784e-01, 1.2884e+00],
...,
[ 1.1035e+00, -6.4182e-01, 1.2022e+00, ..., 1.4280e+00,
7.1126e-01, 9.6603e-01],
[ 1.9399e+00, 1.2178e+00, 2.0172e+00, ..., 1.2925e+00,
7.3353e-01, 1.4715e+00],
[ 1.0435e+00, 4.5775e-01, 9.8648e-01, ..., 7.4940e-01,
7.1990e-01, 1.4361e+00]],
[[-5.0245e-01, 6.1401e-01, 7.7698e-02, ..., -9.0413e-01,
-5.8176e-01, -5.0745e-02],
[-1.0684e+00, -1.1267e+00, -1.6090e+00, ..., -1.4268e+00,
-1.2540e+00, -1.1388e+00],
[-2.0055e+00, -8.9117e-01, -2.2790e-01, ..., -1.5758e+00,
-1.6154e+00, -2.6859e-01],
...,
[-1.7006e+00, -7.0600e-01, -6.9748e-01, ..., -8.8244e-01,
-1.5005e+00, -8.7561e-01],
[-1.7340e+00, -8.5301e-01, -7.1816e-01, ..., -7.5220e-01,
-1.6038e+00, -7.0768e-01],
[-3.5165e-01, 4.1779e-01, -2.7127e-01, ..., -2.9130e-01,
-8.0421e-01, -2.4075e-02]]],
[[[ 6.6828e-01, 1.0791e+00, 7.3110e-01, ..., 5.1048e-01,
3.6743e-01, 1.3108e-01],
[-2.1916e-01, -1.2461e+00, -2.8989e-01, ..., 5.2109e-01,
1.3015e+00, 3.2377e-01],
[-5.7645e-03, -1.5027e+00, -1.4745e+00, ..., -1.9835e-01,
-8.3005e-01, -1.4865e+00],
...,
[ 5.6208e-01, -9.2570e-01, -8.7961e-01, ..., 1.0343e+00,
1.3691e+00, 1.0918e+00],
[-6.9069e-03, -1.1351e+00, -4.0108e-01, ..., 8.6077e-01,
1.1504e+00, 2.5716e-01],
[ 3.5193e-01, -6.0709e-01, -9.2964e-01, ..., -4.5277e-01,
-3.3329e-01, 5.4576e-01]],
[[-3.5130e+00, -1.4269e+00, -3.1311e+00, ..., -2.3990e+00,
-3.5814e+00, -1.8314e+00],
[-4.1652e+00, -1.3232e+00, -3.0308e+00, ..., -2.6225e-01,
-5.6819e-01, -9.9734e-01],
[-4.9312e+00, -3.2199e+00, -4.5153e+00, ..., 9.7529e-01,
-3.1331e+00, -2.5939e+00],
...,
[-2.2003e+00, -3.5151e+00, -4.3218e+00, ..., -1.9686e+00,
-1.9695e+00, -2.0976e-01],
[-3.0713e+00, -1.3596e+00, -2.1567e+00, ..., -2.2747e+00,
-1.4317e+00, 2.6580e-01],
[-1.2791e+00, -1.1042e+00, -1.8808e+00, ..., -1.9085e+00,
-5.2046e-01, -1.0103e-02]],
[[ 2.2288e+00, 3.0932e+00, 2.5649e+00, ..., 2.7783e+00,
1.2569e+00, 1.6529e+00],
[ 3.2418e-01, 1.1706e+00, 1.7011e+00, ..., 8.8122e-02,
-1.8317e+00, -4.1825e-01],
[ 5.8693e-01, 1.1061e+00, 2.7110e+00, ..., -1.2570e+00,
-2.5082e+00, -1.6876e+00],
...,
[-5.8134e-02, 4.5909e-01, 1.3590e+00, ..., -3.1506e-01,
7.0942e-01, 1.0729e+00],
[ 8.1143e-01, 2.3176e+00, 1.3685e+00, ..., -2.2342e+00,
3.1934e-01, 8.2734e-01],
[ 1.5115e+00, 1.5182e+00, 1.3950e+00, ..., -1.1693e-02,
4.9253e-01, 1.0857e+00]],
...,
[[-1.9111e+00, -5.2089e+00, -4.3520e+00, ..., -3.8838e+00,
-2.2402e+00, -3.1348e+00],
[-2.1697e+00, -4.4187e+00, -4.7480e+00, ..., -5.1947e+00,
-4.3764e+00, -5.2602e+00],
[ 1.1728e-01, -2.1080e+00, -3.1307e+00, ..., -4.6891e+00,
-5.8992e+00, -1.1272e+00],
...,
[-1.1680e+00, -2.2869e+00, -3.5469e+00, ..., -3.2144e+00,
-3.9562e+00, -3.7163e+00],
[-1.2738e+00, -4.6261e+00, -5.7798e+00, ..., -8.5226e+00,
-4.5761e+00, -5.3276e+00],
[-1.5347e+00, -4.9409e+00, -4.3261e+00, ..., -4.0800e+00,
-2.8387e+00, -3.5413e+00]],
[[ 6.4908e-01, 6.8497e-01, -1.1363e-01, ..., 3.2192e-01,
-4.7809e-01, 1.0292e+00],
[ 1.0632e+00, 9.0919e-01, 1.2973e-01, ..., 1.0451e+00,
1.0269e+00, 9.9329e-01],
[-1.0539e+00, -9.7095e-01, -1.1463e+00, ..., 1.3348e+00,
-1.7044e-03, 9.4005e-01],
...,
[ 2.1156e-02, -1.0396e+00, -7.4352e-01, ..., -1.7130e-01,
1.4557e+00, 2.0450e+00],
[ 1.2722e+00, 1.4486e+00, 1.4733e+00, ..., 5.4757e-01,
1.5375e+00, 2.6066e+00],
[ 6.6725e-01, 8.3786e-01, 8.7674e-01, ..., 1.0806e+00,
1.3248e+00, 1.4302e+00]],
[[-2.9116e-01, -2.3639e-01, -9.7928e-01, ..., 3.3137e-01,
-5.7525e-01, 7.5864e-01],
[-1.4054e+00, -1.5848e+00, -2.8956e+00, ..., 4.6025e-01,
7.8897e-02, 9.9418e-01],
[-2.3064e+00, -1.7948e+00, -1.7079e+00, ..., 1.2735e+00,
-5.5675e-01, -1.1861e+00],
...,
[-1.5235e+00, -2.7082e+00, -2.2025e+00, ..., 3.7620e-01,
-1.1000e+00, -1.2548e+00],
[-1.3022e+00, -1.4645e+00, -9.5143e-01, ..., -1.6638e+00,
-1.7303e+00, -3.0223e-01],
[-1.7470e-01, -9.2146e-02, -6.5972e-01, ..., -1.7375e+00,
-4.9925e-01, 1.2333e-01]]],
[[[ 1.0903e+00, 1.4985e+00, 1.2217e+00, ..., 2.1932e-01,
3.4119e-01, 2.2164e-01],
[-9.4426e-01, -1.4388e+00, -8.8172e-01, ..., 3.8077e-01,
3.8663e-01, -1.1708e+00],
[-5.3105e-01, -4.9395e-01, -4.9922e-01, ..., -2.9627e-01,
1.0655e+00, 6.6097e-01],
...,
[-1.7091e-01, -5.1622e-01, -4.5141e-02, ..., 1.3352e-01,
8.6803e-01, 5.2845e-01],
[-3.8221e-01, -7.1654e-01, -1.2695e-01, ..., 1.8627e-01,
1.3614e+00, -1.3255e-01],
[-1.0426e-01, -1.0681e-01, -6.0767e-01, ..., -1.3337e-01,
4.2466e-01, 5.9600e-01]],
[[-3.2027e+00, -2.0098e+00, -2.1718e+00, ..., -1.7360e+00,
-2.1243e+00, -1.0232e+00],
[-2.8983e+00, -3.3259e+00, -1.5490e+00, ..., -3.5437e-01,
-1.2257e+00, -2.4871e-01],
[-1.8587e+00, -2.4473e+00, -1.6382e+00, ..., -1.1849e+00,
-1.3780e+00, -7.1449e-01],
...,
[ 1.8911e-01, -7.5806e-01, -4.9529e-01, ..., -2.2652e-01,
6.8375e-02, 4.8164e-01],
[-1.1854e+00, -5.5533e-01, -3.8397e-01, ..., 4.1631e-02,
-4.3182e-01, 2.6161e-01],
[-2.7225e-01, -6.5512e-01, -6.9084e-01, ..., -2.1977e-01,
-7.1333e-01, -5.4389e-01]],
[[ 2.3602e+00, 2.3466e+00, 1.8016e+00, ..., 2.9123e+00,
1.2868e+00, 2.8176e+00],
[ 1.8822e-01, 1.3505e-01, 1.4485e+00, ..., 1.8615e+00,
-1.7640e-01, 2.4054e+00],
[-2.5486e-01, 7.9002e-02, 1.0807e+00, ..., 1.2156e+00,
6.0927e-01, 1.7917e+00],
...,
[-6.4045e-01, -3.2892e-01, 2.8408e-01, ..., 1.0919e+00,
-1.5019e-02, 1.4671e+00],
[-1.0540e+00, 5.4187e-01, 5.4881e-01, ..., 1.0825e+00,
6.3103e-01, 1.5562e+00],
[ 5.6209e-01, 4.3673e-01, 1.0656e+00, ..., 1.4193e+00,
1.1041e+00, 2.0118e+00]],
...,
[[-3.0084e+00, -8.5952e+00, -6.8251e+00, ..., -3.4661e+00,
-2.2350e+00, -3.8838e+00],
[-3.9073e+00, -1.0344e+01, -6.3872e+00, ..., -5.8944e+00,
-3.7226e+00, -4.2734e+00],
[-2.5225e+00, -7.8393e+00, -4.6017e+00, ..., -3.1985e+00,
-2.2664e+00, -1.4844e+00],
...,
[-3.7341e+00, -8.4280e+00, -4.6556e+00, ..., -3.7150e+00,
-3.2573e+00, -1.5537e+00],
[-4.0678e+00, -8.9056e+00, -4.6934e+00, ..., -5.2278e+00,
-3.3662e+00, -3.7181e+00],
[-3.1496e+00, -7.4889e+00, -2.9546e+00, ..., -3.7907e+00,
-2.1547e+00, -2.7637e+00]],
[[-4.1468e-01, 7.4320e-01, 9.0221e-01, ..., 1.3442e+00,
1.0860e+00, 1.2273e+00],
[ 7.8468e-01, 7.6291e-01, 1.3293e+00, ..., 1.9529e+00,
2.3254e+00, 2.5343e+00],
[ 7.1480e-01, 4.3055e-01, 1.3779e+00, ..., 1.5461e+00,
1.5894e+00, 1.8876e+00],
...,
[ 1.1884e+00, 2.3776e-01, 1.0578e+00, ..., 1.8033e+00,
1.7946e+00, 1.6446e+00],
[ 1.6745e+00, 1.5400e+00, 1.6103e+00, ..., 1.6207e+00,
1.4726e+00, 2.4354e+00],
[ 7.8563e-01, 6.8590e-01, 1.2124e+00, ..., 1.1685e+00,
1.3356e+00, 1.3801e+00]],
[[-2.0348e-01, -3.8447e-01, -3.7275e-01, ..., -1.8291e-01,
-1.1531e-01, 9.0760e-01],
[-6.9798e-01, -2.7080e+00, -1.1362e+00, ..., 1.9874e-01,
-3.5604e-02, 4.1638e-01],
[-1.7499e+00, -1.8201e+00, -6.0275e-01, ..., 3.8363e-01,
1.5342e-01, 2.2574e-01],
...,
[-1.5341e+00, -2.5030e+00, -6.2254e-01, ..., -1.5085e-01,
-8.1230e-02, -3.2448e-01],
[-1.3782e+00, -2.4809e+00, -5.9672e-01, ..., -4.4945e-01,
-4.4946e-01, 3.1751e-01],
[-3.7399e-01, -1.0310e+00, -5.7686e-01, ..., 9.7869e-02,
1.2006e-01, 3.3704e-01]]],
[[[ 7.3729e-01, 7.0790e-01, 1.4313e-01, ..., 6.5877e-01,
7.2822e-01, 5.4729e-01],
[-2.1601e+00, -2.7527e+00, -3.3685e+00, ..., 5.1960e-02,
2.5144e-01, -7.1849e-01],
[ 5.9015e-01, -1.6245e+00, -3.4075e+00, ..., -5.6550e-01,
2.6143e-01, 3.3728e-01],
...,
[ 4.6411e-01, -3.2989e-01, -5.0052e-01, ..., -7.1020e-01,
-1.3463e+00, 7.1371e-01],
[-1.6703e-01, -1.1701e+00, -9.8762e-01, ..., 3.7141e-01,
4.2782e-01, 1.1354e+00],
[ 7.6230e-01, -3.3247e-01, -7.1045e-01, ..., -4.9251e-01,
-1.1018e+00, 3.2585e-01]],
[[-4.5015e+00, -1.0716e+00, 1.8167e+00, ..., -3.5638e-01,
2.4344e-01, 7.4347e-02],
[-3.0824e+00, 8.7706e-02, 4.2121e-02, ..., 5.4808e-01,
-9.3626e-01, -6.3657e-01],
[-1.7281e+00, -2.1200e+00, -6.4030e-01, ..., -2.8030e-01,
-4.0768e-01, 8.7528e-02],
...,
[-9.8314e-01, -6.4378e-01, -5.9820e-01, ..., -5.3288e+00,
-3.4065e+00, -1.7947e+00],
[-3.0339e+00, -5.1185e-01, -1.4547e+00, ..., -4.3749e+00,
-4.6263e+00, -3.3122e+00],
[-2.1625e+00, -1.5226e+00, -2.1158e+00, ..., -7.0077e+00,
-5.3967e+00, -2.5951e+00]],
[[ 3.1242e+00, 5.5387e-02, -1.4783e+00, ..., -3.2834e-03,
-1.6268e-01, 1.5697e-01],
[ 1.0738e+00, -2.8462e+00, -1.2195e+00, ..., -1.0155e+00,
-8.0007e-01, -6.0318e-01],
[ 5.8570e-01, -3.2932e+00, -6.0525e-01, ..., -7.0343e-01,
-2.0606e-01, -5.6060e-02],
...,
[ 1.8628e-01, 1.0285e+00, 1.9539e+00, ..., -3.0378e+00,
-3.6987e+00, -8.6330e-01],
[ 3.5624e-01, 2.1868e+00, 2.0419e+00, ..., -4.5177e+00,
-3.5800e+00, -1.2646e+00],
[ 1.4012e+00, 1.3710e+00, 1.5870e+00, ..., -4.6210e+00,
-2.4646e+00, 5.8049e-03]],
...,
[[-3.3069e+00, -8.7807e+00, -4.6923e+00, ..., -1.9448e+00,
-9.3768e-01, -1.1531e+00],
[-4.6870e+00, -8.8355e+00, -5.0819e+00, ..., -1.8982e+00,
-2.3298e+00, -1.1542e-01],
[-1.0710e+00, -3.0605e+00, -1.9058e+00, ..., -2.4139e-01,
-1.1387e+00, 2.1238e-01],
...,
[-9.2856e-01, -2.2755e+00, -2.0232e+00, ..., -1.6978e+00,
-6.9293e-01, -2.4659e+00],
[-1.2705e+00, -3.7579e+00, -3.7170e+00, ..., -4.5312e+00,
-6.6676e+00, -8.5559e+00],
[-2.2483e+00, -5.1110e+00, -3.1737e+00, ..., -4.7483e+00,
-2.9542e+00, -4.3474e+00]],
[[-1.3888e+00, -1.5137e+00, 1.2954e+00, ..., 9.2652e-01,
8.5311e-01, 8.4991e-01],
[ 1.1513e+00, 1.6812e+00, 8.4133e-01, ..., 4.2110e-01,
-4.2908e-01, 4.6602e-01],
[ 7.3464e-01, -1.2801e+00, -3.8927e+00, ..., 7.3110e-01,
2.3206e-01, 6.4911e-01],
...,
[ 1.2681e+00, -2.4978e-01, -2.4147e-01, ..., -4.9351e+00,
-2.3784e+00, 1.9683e-01],
[ 1.8659e+00, 9.1536e-01, 1.2097e+00, ..., -2.3036e+00,
-1.0378e+00, 7.5248e-01],
[ 1.0238e+00, 3.6421e-01, 8.1887e-01, ..., -1.7319e+00,
-1.4070e+00, 5.6816e-01]],
[[-1.0637e+00, 2.5741e-01, 6.0774e-01, ..., 3.9337e-01,
-3.7449e-02, -4.0534e-01],
[-1.0418e+00, -1.5574e-01, 6.6142e-01, ..., -2.1073e-02,
-1.5746e+00, -1.2392e+00],
[-2.2439e+00, -5.7295e-01, -1.3648e+00, ..., -8.6170e-01,
-1.0316e+00, -2.4963e-01],
...,
[-1.7589e+00, -1.5030e+00, -1.0757e-01, ..., -2.4970e+00,
-8.4688e-01, 2.3362e-01],
[-1.7189e+00, -1.0850e+00, -8.1078e-01, ..., -1.0528e+00,
-7.3145e-01, -4.8101e-01],
[-3.3672e-01, 3.3952e-01, -4.5992e-01, ..., -1.8369e+00,
-3.5128e+00, -1.0325e+00]]]], device='cuda:0'),
tensor([[[[ 9.6867e-01, 1.5827e+00, 1.1199e+00, ..., 8.2190e-01,
1.0358e+00, 4.9011e-01],
[-8.6323e-01, -1.8892e+00, -7.5772e-01, ..., -2.9079e-01,
-1.5965e-01, -3.7205e-01],
[ 1.8432e-02, -7.8968e-01, 1.9254e-01, ..., -7.6868e-01,
-1.3192e-01, 6.3477e-01],
...,
[ 4.8065e-01, -2.4153e-01, -1.1768e-01, ..., 2.8238e-02,
-5.6979e-02, 3.9121e-01],
[-1.9811e-01, -1.0109e+00, -4.3572e-02, ..., -6.9135e-02,
1.1651e-01, -8.7911e-02],
[ 7.5248e-01, -2.8305e-01, -1.6340e-01, ..., -3.3230e-01,
-9.0545e-01, 1.6727e-01]],
[[-4.2495e+00, -1.2273e+00, -2.1205e+00, ..., -1.4143e+00,
-1.9808e+00, -1.7913e+00],
[-3.9668e+00, -2.7117e+00, -2.1579e+00, ..., -1.1509e+00,
-2.1915e+00, -2.1884e+00],
[-2.6372e+00, -1.8565e+00, -2.9844e+00, ..., -1.3766e+00,
-1.7266e+00, -1.4415e+00],
...,
[-7.9954e-01, -4.8662e-01, -1.1202e+00, ..., -5.6719e-01,
-8.1517e-01, -8.3969e-01],
[-2.8457e+00, -3.7677e-01, -7.9204e-01, ..., -6.3795e-01,
-1.0709e+00, -1.2303e+00],
[-2.0068e+00, -1.4575e+00, -1.3778e+00, ..., -6.5912e-01,
-7.4190e-01, -5.6179e-01]],
[[ 3.0254e+00, 4.0855e+00, 3.2411e+00, ..., 1.6866e+00,
8.0497e-01, 2.5522e+00],
[ 4.0160e-01, 8.4815e-01, 2.2754e+00, ..., 1.3674e+00,
6.3060e-01, 2.6205e+00],
[ 8.7783e-01, 1.2752e+00, 1.7668e+00, ..., 2.9044e-01,
9.3238e-01, 1.8627e+00],
...,
[ 3.3311e-01, 9.1405e-01, 1.0897e+00, ..., -2.7810e-01,
5.3244e-01, 1.4465e+00],
[ 3.6106e-01, 2.1232e+00, 1.5020e+00, ..., -1.8194e-01,
7.4065e-01, 1.1385e+00],
[ 1.4465e+00, 1.3759e+00, 1.7162e+00, ..., 6.0668e-01,
1.3088e+00, 1.3706e+00]],
...,
[[-2.0735e+00, -7.3921e+00, -5.8733e+00, ..., -4.9320e+00,
-3.5938e+00, -4.3343e+00],
[-2.7576e+00, -6.6143e+00, -4.6451e+00, ..., -4.8275e+00,
-3.6816e+00, -3.1816e+00],
[-2.3834e-01, -2.8107e+00, -3.1274e+00, ..., -3.0603e+00,
-2.2758e+00, -1.3771e+00],
...,
[-9.8444e-01, -2.3343e+00, -1.5830e+00, ..., -4.4094e+00,
-3.0877e+00, -2.7357e+00],
[-1.2943e+00, -3.7173e+00, -3.2822e+00, ..., -5.0392e+00,
-4.1715e+00, -4.3578e+00],
[-2.2182e+00, -5.0017e+00, -2.9821e+00, ..., -3.3682e+00,
-2.4095e+00, -2.4211e+00]],
[[-7.7292e-01, -1.1862e-01, 2.3299e-01, ..., 1.4672e+00,
1.7321e+00, 1.1817e+00],
[ 1.7685e+00, 1.0692e+00, 1.4832e+00, ..., 1.5217e+00,
1.1673e+00, 1.8186e+00],
[ 1.1831e+00, 2.2726e-01, -1.7199e-02, ..., 1.2851e+00,
5.4767e-01, 1.2881e+00],
...,
[ 1.3173e+00, -9.7757e-02, 1.0175e+00, ..., 1.5164e+00,
6.9964e-01, 8.9451e-01],
[ 1.9161e+00, 9.6718e-01, 1.7869e+00, ..., 1.3183e+00,
7.1802e-01, 1.4174e+00],
[ 1.0466e+00, 4.1186e-01, 9.9028e-01, ..., 7.5969e-01,
6.6456e-01, 1.4192e+00]],
[[-4.9523e-01, 5.8608e-01, 1.0460e-01, ..., -3.3352e-01,
-5.4717e-01, -3.0342e-02],
[-1.0284e+00, -1.0918e+00, -1.4245e+00, ..., -1.2000e+00,
-1.4306e+00, -1.1408e+00],
[-2.0459e+00, -5.6789e-01, -8.0789e-01, ..., -6.7279e-01,
-1.0745e+00, -2.6166e-01],
...,
[-1.7403e+00, -1.4761e+00, -9.9145e-01, ..., -5.8746e-01,
-1.3794e+00, -8.8182e-01],
[-1.7544e+00, -1.1092e+00, -8.3370e-01, ..., -8.2744e-01,
-1.6245e+00, -7.4158e-01],
[-3.5874e-01, 3.7603e-01, -2.9450e-01, ..., -3.0421e-01,
-8.3663e-01, -9.9074e-03]]],
[[[ 8.6293e-01, 1.4652e+00, 1.1739e+00, ..., 1.0101e+00,
7.8813e-01, 9.5722e-01],
[-9.9686e-01, -1.9035e+00, -1.1748e+00, ..., 1.5775e-01,
-6.8647e-01, -1.5721e+00],
[-2.4026e-01, -7.1988e-01, 1.1676e-01, ..., 2.1198e-01,
5.3508e-01, 8.7343e-01],
...,
[ 4.2973e-03, -1.1496e+00, -1.0628e+00, ..., -3.9650e-01,
-8.1500e-03, 2.1984e-01],
[-6.6864e-01, -1.8917e+00, -1.4841e+00, ..., -3.5963e-01,
-2.7872e-01, -1.4532e-01],
[-3.5532e-01, -1.7421e+00, -1.7831e+00, ..., -9.7043e-01,
-1.5482e+00, -2.8665e-01]],
[[-3.7247e+00, -8.1088e-01, -1.5748e+00, ..., -2.0454e+00,
-2.4427e+00, -1.2634e+00],
[-3.8434e+00, -2.8597e+00, -1.7856e+00, ..., -3.3763e+00,
-3.0469e+00, -9.6338e-01],
[-2.3352e+00, -1.7355e+00, -1.6252e+00, ..., -3.2001e+00,
-2.3672e+00, -1.0533e+00],
...,
[-1.7189e+00, -1.7379e+00, -2.4201e+00, ..., -2.1321e+00,
-2.1694e+00, -7.0550e-01],
[-3.1965e+00, -1.6806e+00, -3.0204e+00, ..., -2.0402e+00,
-2.4116e+00, -1.2497e+00],
[-2.3935e+00, -1.9669e+00, -2.8879e+00, ..., -1.8171e+00,
-1.9573e+00, -9.8540e-01]],
[[ 2.8508e+00, 4.2441e+00, 3.3481e+00, ..., 3.4076e+00,
1.1657e+00, 3.0869e+00],
[ 1.4530e-01, 1.1439e+00, 2.2546e+00, ..., 1.7693e+00,
-7.7511e-01, 2.8163e+00],
[ 8.6214e-01, 1.5149e+00, 1.5564e+00, ..., 1.4163e+00,
-2.1785e-01, 2.1033e+00],
...,
[ 1.0199e+00, 1.0725e+00, 1.6029e+00, ..., -8.8722e-01,
1.0525e+00, 2.2925e+00],
[ 7.0153e-01, 1.8425e+00, 1.6404e+00, ..., -5.4636e-01,
1.0934e+00, 1.8284e+00],
[ 1.6048e+00, 1.5969e+00, 2.2414e+00, ..., 1.0684e+00,
1.8181e+00, 1.7732e+00]],
...,
[[-1.7371e+00, -7.1050e+00, -5.3846e+00, ..., -5.0040e+00,
-3.3490e+00, -5.5591e+00],
[-2.3386e+00, -6.9665e+00, -4.1969e+00, ..., -5.9530e+00,
-3.8999e+00, -4.3904e+00],
[-9.0875e-01, -4.5020e+00, -3.6649e+00, ..., -6.6896e+00,
-5.3889e+00, -2.5096e+00],
...,
[-2.6106e+00, -5.3253e+00, -4.2426e+00, ..., -7.7722e+00,
-4.6122e+00, -4.5935e+00],
[-2.7653e+00, -5.7760e+00, -5.4004e+00, ..., -8.1446e+00,
-5.7139e+00, -6.1929e+00],
[-1.5166e+00, -4.1750e+00, -2.8483e+00, ..., -4.3660e+00,
-2.7383e+00, -3.2192e+00]],
[[-1.7832e-01, 2.8822e-01, 5.8110e-01, ..., 6.9591e-01,
1.2511e+00, 1.4045e+00],
[ 1.5580e+00, 1.0515e+00, 1.8956e+00, ..., 1.5468e+00,
3.1763e+00, 3.0557e+00],
[ 1.1123e+00, 2.2281e-01, 8.9136e-01, ..., 9.1094e-01,
2.4468e+00, 2.3844e+00],
...,
[ 1.2992e+00, 2.4805e-01, -3.7210e-02, ..., 2.3638e+00,
1.0700e+00, 1.6356e+00],
[ 1.8702e+00, 1.7323e+00, 1.0195e+00, ..., 2.3833e+00,
1.2665e+00, 1.9786e+00],
[ 4.1579e-01, 7.6853e-01, 5.3462e-01, ..., 1.1163e+00,
6.9788e-01, 1.6658e+00]],
[[-5.4417e-01, 2.6353e-01, 1.3534e-01, ..., -3.4551e-01,
4.7171e-01, 1.3573e+00],
[-1.3117e+00, -1.8099e+00, -1.5854e+00, ..., -1.4228e+00,
-4.5501e-01, 2.2797e-01],
[-2.1539e+00, -8.6267e-01, 3.2115e-01, ..., -3.8291e-01,
-8.4029e-02, 7.6619e-01],
...,
[-1.9679e+00, -1.8580e+00, -1.5116e+00, ..., -1.5489e+00,
-1.1634e+00, -8.6423e-01],
[-1.5913e+00, -2.1041e+00, -1.9090e+00, ..., -1.5059e+00,
-1.3699e+00, -7.0670e-01],
[-8.3554e-01, -9.4739e-01, -1.1971e+00, ..., -8.1917e-01,
-8.9329e-01, -1.2050e-01]]],
[[[ 1.1437e+00, 1.8396e+00, 9.1830e-01, ..., 3.0119e-01,
1.8703e-01, -1.8619e-01],
[-1.3854e+00, -2.2964e+00, -1.8259e+00, ..., 4.5959e-01,
3.5696e-01, -9.5708e-01],
[-1.2642e+00, -1.7049e+00, -1.6486e+00, ..., -4.2498e-01,
-5.4466e-01, -8.6409e-01],
...,
[ 3.6372e-01, -2.6514e-01, -8.0374e-01, ..., 6.0257e-01,
-6.4358e+00, -1.6974e+00],
[-2.1273e-01, -1.0626e+00, -6.5268e-01, ..., -3.4836e-02,
1.5360e+00, -3.6994e+00],
[ 7.1348e-01, -3.6559e-01, -3.4146e-01, ..., -1.2047e+00,
8.0483e-01, 1.0326e+00]],
[[-5.0256e+00, -2.3646e+00, -2.6197e+00, ..., -2.7375e+00,
-3.1367e+00, -6.0375e-01],
[-6.2070e+00, -5.3486e+00, -4.0061e+00, ..., -3.6115e+00,
-3.9609e+00, -1.5119e-01],
[-5.2056e+00, -3.1263e+00, -9.1359e-01, ..., -5.9255e+00,
-5.3656e+00, -1.5357e+00],
...,
[-1.0252e+00, -8.1820e-01, -2.0324e+00, ..., -1.0008e+01,
-6.5463e+00, -1.2105e+00],
[-3.0296e+00, -6.4062e-01, -1.4166e+00, ..., -6.9809e+00,
-7.6810e+00, -3.0852e+00],
[-2.1944e+00, -1.6693e+00, -1.7722e+00, ..., -2.6452e+00,
-1.7769e+00, -3.8918e-01]],
[[ 3.2423e+00, 4.4368e+00, 3.8769e+00, ..., 2.1942e+00,
6.0229e-01, 2.1748e+00],
[ 1.1115e+00, 1.6944e+00, 2.7266e+00, ..., 1.6004e+00,
-6.7867e-02, 2.7640e+00],
[ 1.1951e+00, 2.2297e-01, -2.2947e-01, ..., 3.3674e+00,
2.0884e+00, 2.3998e+00],
...,
[-1.9220e-02, 9.6679e-01, 2.0496e+00, ..., -3.7070e+00,
-9.5801e+00, -3.1505e+00],
[ 1.2446e-01, 2.1349e+00, 1.6290e+00, ..., 1.6848e+00,
-2.7346e+00, -1.8115e+00],
[ 1.2824e+00, 1.3302e+00, 1.7896e+00, ..., 1.9347e+00,
1.4895e+00, 1.4618e+00]],
...,
[[-2.3167e+00, -8.1169e+00, -6.2113e+00, ..., -3.9494e+00,
-2.8415e+00, -3.4634e+00],
[-3.3098e+00, -6.6779e+00, -4.5978e+00, ..., -6.4863e+00,
-6.0274e+00, -5.1301e+00],
[-2.5784e+00, -4.7974e+00, -2.7749e+00, ..., -6.9909e+00,
-3.0152e+00, -2.8015e+00],
...,
[-8.2832e-01, -2.1599e+00, -2.1124e+00, ..., -1.0633e+01,
-7.1118e+00, -9.5138e+00],
[-1.1659e+00, -3.6298e+00, -3.8882e+00, ..., -1.1044e+01,
-7.3759e+00, -7.2655e+00],
[-2.1296e+00, -5.0124e+00, -3.2461e+00, ..., -4.4272e+00,
-3.8223e+00, -5.2761e+00]],
[[-1.1278e+00, -1.3782e-01, -7.1797e-03, ..., 1.1055e+00,
1.1134e+00, 1.7635e+00],
[ 9.7128e-01, -2.0016e+00, -8.6351e-01, ..., 1.2928e+00,
1.1898e+00, 2.8267e+00],
[ 7.3287e-01, 1.8512e-01, 1.0997e+00, ..., -1.2773e+00,
-2.8968e+00, 2.6825e-01],
...,
[ 1.2267e+00, -2.1997e-01, -1.6156e-01, ..., 1.6932e+00,
2.6139e+00, 6.0426e-01],
[ 1.7909e+00, 7.1431e-01, 1.6511e+00, ..., -1.2693e+00,
5.0224e-01, 2.6583e+00],
[ 9.9717e-01, 1.8295e-01, 8.1291e-01, ..., 4.7596e-01,
1.2595e+00, 1.6406e+00]],
[[-5.5334e-01, 9.8593e-01, 2.2121e-02, ..., -3.7249e-01,
-4.7774e-01, 9.3384e-01],
[-1.9879e+00, -6.0157e-01, -1.9142e+00, ..., -3.0891e-01,
5.4138e-01, 5.5359e-01],
[-2.3939e+00, 6.8022e-01, 1.5378e+00, ..., 1.2171e-01,
-4.9193e-01, -2.3974e-01],
...,
[-1.6871e+00, -1.4031e+00, -1.0251e+00, ..., -2.5582e+00,
-1.8306e+00, -1.9768e+00],
[-1.6691e+00, -9.0990e-01, -9.8972e-01, ..., -2.3112e+00,
-2.9647e+00, -1.9337e+00],
[-2.5758e-01, 3.7426e-01, -4.1991e-01, ..., -2.7196e-01,
-5.9144e-01, 7.1428e-01]]],
[[[ 7.2801e-01, 1.1080e+00, 7.7337e-01, ..., 5.0117e-01,
3.9679e-01, 2.4984e-01],
[-3.1070e-01, -1.4213e+00, -4.9824e-01, ..., 2.7450e-01,
1.9209e-01, -1.1851e+00],
[ 1.6834e-01, -8.5133e-01, -3.5581e-01, ..., -2.2727e+00,
1.0093e+00, 5.2632e-01],
...,
[-2.2878e-01, -4.4598e-01, -4.9095e-03, ..., -6.4032e-01,
1.0120e+00, 4.7546e-01],
[-6.2705e-01, -1.8027e+00, -7.0241e-01, ..., 2.3346e-01,
1.4550e+00, -1.0760e-01],
[ 3.9198e-01, -7.1689e-01, -9.6451e-01, ..., -2.5105e-01,
4.0897e-01, 5.8569e-01]],
[[-2.9612e+00, -1.5076e+00, -3.2689e+00, ..., -1.7093e+00,
-2.0883e+00, -9.8764e-01],
[-4.1901e+00, -1.7989e+00, -2.3863e+00, ..., -4.3972e-01,
-1.3378e+00, -2.2541e-01],
[-3.9985e+00, -2.3912e+00, -3.3237e+00, ..., -1.4890e+00,
-1.8118e+00, -8.5316e-01],
...,
[-2.8534e+00, -1.0596e+00, -6.0849e-01, ..., -1.5564e+00,
6.2829e-01, 4.7089e-01],
[-3.7971e+00, -1.3575e+00, -2.6258e+00, ..., -9.7265e-02,
-2.8102e-01, 2.1452e-01],
[-1.2031e+00, -1.1500e+00, -2.6421e+00, ..., -4.0281e-01,
-7.8105e-01, -6.2099e-01]],
[[ 1.7704e+00, 2.7329e+00, 2.4526e+00, ..., 2.8360e+00,
1.4144e+00, 2.8777e+00],
[ 2.2333e-01, 1.1320e+00, 1.6778e+00, ..., 1.4538e+00,
-3.2886e-01, 2.4863e+00],
[ 7.5688e-01, 1.2486e+00, 8.2314e-01, ..., 3.5307e+00,
7.3122e-01, 1.8328e+00],
...,
[ 1.4777e+00, 1.0715e+00, -9.0015e-01, ..., 9.1868e-01,
-4.6882e-01, 1.4190e+00],
[ 6.4664e-01, 2.0174e+00, 1.0569e+00, ..., 1.0524e+00,
6.3354e-01, 1.5059e+00],
[ 1.6242e+00, 1.6968e+00, 1.8801e+00, ..., 1.2128e+00,
1.1122e+00, 2.0430e+00]],
...,
[[-1.3725e+00, -4.5411e+00, -3.8248e+00, ..., -3.2693e+00,
-2.4967e+00, -3.9032e+00],
[-1.7797e+00, -3.8225e+00, -4.0757e+00, ..., -6.0460e+00,
-3.9298e+00, -4.3125e+00],
[-1.1462e+00, -3.3741e+00, -3.7493e+00, ..., -3.0429e+00,
-2.5281e+00, -1.7254e+00],
...,
[-1.8503e+00, -4.5523e+00, -3.5249e+00, ..., -5.4332e+00,
-5.2554e+00, -1.4416e+00],
[-1.7739e+00, -5.0140e+00, -5.1272e+00, ..., -6.5227e+00,
-3.9017e+00, -3.6517e+00],
[-1.7872e+00, -5.1262e+00, -4.0562e+00, ..., -4.5893e+00,
-2.0495e+00, -2.7376e+00]],
[[ 7.9353e-01, 4.9962e-01, -7.8894e-02, ..., 8.4428e-01,
1.0760e+00, 1.2506e+00],
[ 1.2292e+00, 9.2395e-01, 6.0084e-01, ..., 1.1215e+00,
2.1615e+00, 2.5785e+00],
[ 6.8150e-01, 4.9780e-01, -2.1318e-01, ..., 2.6454e+00,
1.6688e+00, 1.9607e+00],
...,
[ 6.5585e-01, -2.8632e-01, -2.9796e-01, ..., 2.3318e+00,
2.4782e+00, 1.7214e+00],
[ 1.4525e+00, 1.1616e+00, 5.3169e-02, ..., 1.4819e+00,
1.5163e+00, 2.4744e+00],
[ 6.7668e-01, 8.3708e-01, 5.2847e-01, ..., 1.0974e+00,
1.3686e+00, 1.3750e+00]],
[[-3.0519e-01, -7.5627e-02, -8.5634e-01, ..., -9.1141e-02,
-9.8985e-02, 8.8290e-01],
[-1.4192e+00, -1.3500e+00, -1.7449e+00, ..., -2.6885e-01,
-2.3692e-01, 3.3108e-01],
[-1.7290e+00, -8.2924e-01, -1.0464e+00, ..., -3.1992e-01,
4.5571e-01, 1.6427e-01],
...,
[-1.7952e+00, -1.1351e+00, -1.1278e+00, ..., -1.9863e+00,
-2.4141e-01, -2.9352e-01],
[-1.4089e+00, -1.6701e+00, -1.4551e+00, ..., -2.8707e-01,
-2.0364e-01, 3.2064e-01],
[-1.4413e-01, -1.0043e-01, -9.2967e-01, ..., 4.0050e-01,
1.0891e-01, 2.5943e-01]]],
[[[ 9.1329e-01, 1.5930e+00, 1.8659e+00, ..., -6.8274e+00,
-1.5366e+00, 1.3800e+00],
[-7.3081e-01, -2.2459e+00, -1.5590e+00, ..., -1.7094e+01,
-1.1331e+01, 1.7387e+00],
[-3.8405e-02, -7.8338e-01, -1.0699e+00, ..., -1.5438e+01,
-1.3494e+01, -1.5526e-01],
...,
[ 2.3155e-02, -8.8136e-01, -1.4614e+00, ..., -7.2382e-01,
-7.6864e-02, 2.8866e-01],
[-5.9210e-01, -1.6403e+00, -1.1118e+00, ..., -2.7708e-02,
-1.0895e-02, -5.5816e-02],
[-3.4981e-02, -1.5347e+00, -1.6850e+00, ..., -1.1336e+00,
-1.2925e+00, -1.4364e-01]],
[[-3.6792e+00, -4.7320e-01, -2.0564e+00, ..., -3.7463e+00,
-5.4050e+00, -4.0365e+00],
[-4.3317e+00, -3.0884e+00, -3.7084e+00, ..., -1.7808e+01,
-7.5715e+00, -7.0009e+00],
[-2.5495e+00, -2.0364e+00, -5.1193e+00, ..., -1.5540e+01,
-2.3954e+01, -1.4854e+01],
...,
[-1.0247e+00, -1.6772e+00, -3.8949e+00, ..., -5.0847e+00,
-1.9737e+00, -1.3733e-01],
[-2.6158e+00, -1.5688e+00, -2.5908e+00, ..., -2.1184e+00,
-1.7637e+00, -7.3067e-01],
[-1.9874e+00, -1.9686e+00, -2.3948e+00, ..., -1.8360e+00,
-1.7419e+00, -7.2739e-01]],
[[ 2.9600e+00, 4.5198e+00, 3.8576e+00, ..., -9.1271e+00,
-6.7807e+00, -1.7080e+00],
[ 4.0155e-02, 1.1667e+00, 3.0755e+00, ..., -1.7021e+01,
-2.2280e+01, -8.2891e+00],
[ 6.0349e-01, 1.6309e+00, 3.2725e+00, ..., -1.8260e+01,
-3.0417e+01, -1.5929e+01],
...,
[ 8.9249e-01, 1.2029e+00, 1.9974e+00, ..., 5.2516e-01,
1.8891e+00, 1.5136e+00],
[ 6.8822e-01, 2.0823e+00, 1.5656e+00, ..., 4.6182e-01,
7.2433e-01, 1.6239e+00],
[ 1.9004e+00, 1.4728e+00, 1.9432e+00, ..., 1.2609e+00,
1.6828e+00, 1.6913e+00]],
...,
[[-1.8060e+00, -6.9690e+00, -5.9800e+00, ..., -2.8373e+01,
-7.4263e+00, -7.5400e+00],
[-2.4949e+00, -6.4150e+00, -4.2404e+00, ..., -4.3518e+01,
-4.5356e+00, -8.8676e+00],
[-7.7371e-01, -3.9824e+00, -3.7575e+00, ..., -2.6310e+01,
-1.9498e+00, -4.5924e+00],
...,
[-2.7931e+00, -6.1832e+00, -5.4574e+00, ..., -5.7026e+00,
-5.8045e+00, -6.0510e+00],
[-3.3420e+00, -7.0885e+00, -5.7435e+00, ..., -8.5482e+00,
-7.4915e+00, -5.9060e+00],
[-2.0844e+00, -5.3660e+00, -2.9046e+00, ..., -3.1926e+00,
-2.5558e+00, -2.9542e+00]],
[[-2.2578e-02, 3.2093e-01, -8.1073e-01, ..., -1.0527e+01,
-5.3094e+00, 2.3594e-01],
[ 1.5634e+00, 1.1140e+00, 1.4807e-02, ..., -1.8121e+01,
-9.9899e+00, -1.0343e+01],
[ 9.6741e-01, 1.5544e-01, -1.5044e+00, ..., -1.1989e+01,
-9.5281e-01, -1.5224e+01],
...,
[ 1.5079e+00, 1.7450e-01, -1.4455e+00, ..., 7.9893e-01,
2.2870e+00, 2.1878e+00],
[ 2.1569e+00, 1.9461e+00, 9.8464e-01, ..., 2.3050e+00,
1.3980e+00, 2.1597e+00],
[ 6.7856e-01, 9.7991e-01, 7.7065e-01, ..., 5.6039e-01,
6.9262e-01, 1.6376e+00]],
[[-5.0701e-01, 3.6465e-01, -1.2359e+00, ..., -4.1556e+00,
4.0346e+00, 4.4714e+00],
[-1.2849e+00, -1.8645e+00, -3.5848e+00, ..., -2.6586e+01,
-5.5046e+00, 1.6678e+00],
[-1.9418e+00, -9.7122e-01, -2.1745e+00, ..., -1.4011e+01,
-3.4889e+00, -4.1521e-02],
...,
[-2.1146e+00, -2.0763e+00, -1.7792e+00, ..., -1.8646e+00,
-5.9375e-01, -5.0468e-01],
[-1.9495e+00, -2.0566e+00, -1.4908e+00, ..., 3.7900e-01,
-5.2114e-01, -2.3414e-01],
[-7.7622e-01, -8.4097e-01, -1.0371e+00, ..., -5.6069e-01,
-9.3695e-01, -1.1248e-01]]]], device='cuda:0'),
tensor([[[[ 8.4385e-01, 8.9977e-01, 7.8955e-01, ..., 6.8692e-01,
1.0067e+00, 4.6815e-01],
[-1.1808e+00, -1.1000e+00, -3.8314e-01, ..., -6.6161e-02,
-1.4449e-01, -3.8534e-01],
[-1.3449e-01, -8.7713e-02, 1.8259e-01, ..., -5.9193e-01,
-8.3455e-02, 6.3079e-01],
...,
[ 1.1229e-01, 7.4568e-01, -9.2110e-01, ..., 6.5595e-01,
2.9643e-01, 4.8830e-01],
[-1.1636e+00, 3.6587e-01, -2.0993e-01, ..., 6.7843e-01,
3.3282e-01, 3.7689e-03],
[-4.6746e-01, 2.8261e-01, 1.1005e+00, ..., -2.5249e-01,
-8.6854e-01, 1.2979e-01]],
[[-3.9333e+00, -1.4162e+00, -1.3690e+00, ..., -1.5482e+00,
-2.0798e+00, -1.8169e+00],
[-2.8930e+00, -9.4870e-01, 1.4147e+00, ..., -1.4763e+00,
-2.1460e+00, -2.1171e+00],
[-1.8788e+00, -9.6441e-01, 2.3691e-01, ..., -1.5039e+00,
-1.6419e+00, -1.4161e+00],
...,
[ 1.2276e+00, -2.5962e-01, -1.8080e+00, ..., -4.3607e-01,
-1.3317e+00, -1.5414e+00],
[-3.5117e-01, -9.7746e-04, -1.0409e+00, ..., -8.1204e-01,
-1.7959e+00, -1.8686e+00],
[-5.5746e-01, -8.3389e-03, -3.3259e-01, ..., -1.2839e+00,
-9.7817e-01, -7.6023e-01]],
[[ 3.0304e+00, 2.9921e+00, 2.2526e+00, ..., 1.5569e+00,
8.7239e-01, 2.5502e+00],
[ 1.0764e+00, 8.1788e-02, 3.1557e-01, ..., 1.2745e+00,
7.0040e-01, 2.6094e+00],
[ 1.4612e+00, -9.7103e-01, -7.6936e-01, ..., 2.1065e-01,
1.0418e+00, 1.8165e+00],
...,
[-2.0432e+00, -1.2791e+00, -9.5156e-01, ..., -1.6502e+00,
3.2314e-01, 1.0695e+00],
[-3.4456e+00, -1.1526e-01, -8.0931e-01, ..., -2.0096e+00,
2.1034e-01, 1.0379e+00],
[-7.4581e-01, -4.2977e-01, 1.1655e-01, ..., 3.3679e-01,
1.4998e+00, 1.3920e+00]],
...,
[[-3.1739e+00, -8.4434e+00, -4.2695e+00, ..., -5.2477e+00,
-3.5682e+00, -4.3058e+00],
[-3.4765e+00, -1.0000e+01, -4.8931e+00, ..., -5.2867e+00,
-3.6909e+00, -3.2461e+00],
[-2.2744e+00, -6.3430e+00, -3.0273e+00, ..., -3.8766e+00,
-2.4237e+00, -1.3901e+00],
...,
[-2.6149e+00, -7.2702e+00, -1.7825e-01, ..., -1.9772e+00,
-2.3143e+00, -3.1558e+00],
[-3.4049e+00, -9.4135e+00, -1.9843e+00, ..., -5.5639e+00,
-3.8138e+00, -4.2953e+00],
[-2.0894e+00, -6.4228e+00, -3.0944e+00, ..., -3.5311e+00,
-2.2066e+00, -2.4587e+00]],
[[-7.2716e-01, -2.6658e-01, 8.1461e-02, ..., 1.8706e+00,
1.6819e+00, 1.1856e+00],
[ 1.5931e+00, 1.3700e+00, 2.1263e+00, ..., 1.8650e+00,
1.2249e+00, 1.8600e+00],
[ 1.1307e+00, 1.5694e-01, 3.3436e-01, ..., 1.5604e+00,
5.7903e-01, 1.2885e+00],
...,
[ 8.1680e-01, -7.3729e-01, 2.1046e-01, ..., 1.3852e+00,
9.1879e-01, 9.0759e-01],
[ 1.0023e+00, -1.8693e+00, -4.9677e-01, ..., 9.1175e-01,
7.5529e-01, 1.4049e+00],
[ 1.1853e+00, 2.6439e-01, -1.6747e-01, ..., 6.2454e-01,
7.3645e-01, 1.4875e+00]],
[[-9.9392e-01, -1.0177e+00, -3.6837e-01, ..., -1.9864e-01,
-5.9607e-01, -5.8378e-02],
[-1.5927e+00, -1.7873e+00, 9.5549e-02, ..., -1.3064e+00,
-1.3962e+00, -1.1259e+00],
[-2.1409e+00, -1.1642e+00, 9.8207e-01, ..., -5.5652e-01,
-1.1164e+00, -2.2552e-01],
...,
[-2.4411e-01, -3.5438e+00, -2.3628e+00, ..., -1.4280e+00,
-6.6328e-01, -7.5170e-01],
[-9.7800e-01, -4.0360e+00, -2.8995e+00, ..., -1.7915e+00,
-2.0984e+00, -8.6684e-01],
[-9.4648e-01, -5.7843e-01, -1.7744e-01, ..., -9.3137e-01,
-7.3820e-01, 1.6396e-01]]],
[[[ 3.6840e-01, 7.2487e-02, 2.4894e-01, ..., 3.0353e-01,
3.2963e-01, -4.9431e-02],
[-1.7188e+00, -3.2244e+00, 5.8491e-01, ..., 2.1350e-01,
9.5901e-02, -3.8404e-01],
[-4.2755e+00, -4.9073e+00, -4.5400e+00, ..., -1.4044e-01,
1.3852e-02, 7.0623e-01],
...,
[ 2.0483e-01, -2.7728e-01, 8.8329e-01, ..., -3.2098e-01,
-1.5449e-01, 5.4242e-01],
[-5.9188e-02, -4.6429e-01, -2.0012e-01, ..., 2.0010e+00,
2.4227e+00, 5.3563e-02],
[ 7.8836e-01, -1.0195e-01, -3.6111e-01, ..., 1.6449e+00,
1.2250e+00, 7.7007e-01]],
[[-3.7768e+00, -2.6543e+00, -6.2023e+00, ..., -1.7301e+00,
-2.2737e+00, -1.4627e+00],
[-5.3859e+00, -3.8596e+00, -6.1530e+00, ..., -1.3886e+00,
-2.4751e+00, -1.6478e+00],
[-2.6895e+00, -2.6104e+00, -3.4073e+00, ..., -2.2331e+00,
-2.4125e+00, -1.0785e+00],
...,
[-8.2110e-01, -9.6046e-01, -9.0465e-01, ..., -2.0361e+00,
-8.8860e-01, 1.0652e+00],
[-2.0920e+00, -5.8660e-01, -6.1041e-01, ..., 2.9335e-01,
-5.8915e-01, 9.4745e-01],
[-1.3875e+00, -1.4312e+00, -1.1312e+00, ..., -3.6071e-01,
-1.5602e+00, -1.7999e-01]],
[[ 2.3127e+00, 1.8944e+00, 1.0193e+00, ..., 1.7282e+00,
8.2656e-01, 2.1338e+00],
[-5.9309e-01, -1.1971e+00, -3.9775e+00, ..., 8.8284e-01,
4.2214e-01, 2.2877e+00],
[-2.0080e+00, 9.6581e-01, -3.8204e+00, ..., -1.3150e-01,
8.4410e-01, 2.1056e+00],
...,
[ 8.2513e-02, 1.4291e+00, 2.4406e+00, ..., 3.4809e-01,
6.9247e-01, 1.8169e+00],
[ 7.1600e-01, 1.9520e+00, 2.0996e+00, ..., -1.5733e+00,
-3.0947e-01, 2.0410e+00],
[ 2.0966e+00, 2.0599e+00, 2.7347e+00, ..., 7.7115e-01,
6.9184e-01, 2.1505e+00]],
...,
[[-1.5661e+00, -4.0569e+00, -2.4881e+00, ..., -2.1842e+00,
-1.7880e+00, -3.1833e+00],
[-6.1173e-01, -5.4773e+00, -5.8607e+00, ..., -2.7745e+00,
-2.6276e+00, -3.2976e+00],
[-3.4870e-01, -7.5639e+00, -6.5782e+00, ..., -2.5340e+00,
-2.7106e+00, -2.4825e+00],
...,
[-2.1262e+00, -4.0168e+00, -3.5320e+00, ..., -5.7794e+00,
-2.4933e+00, -2.4316e+00],
[-1.9720e+00, -4.7834e+00, -4.0594e+00, ..., -9.6383e+00,
-4.7511e+00, -4.9671e+00],
[-2.1243e+00, -5.2179e+00, -2.6852e+00, ..., -5.6136e+00,
-3.4953e+00, -3.1970e+00]],
[[-1.7906e-01, -1.7852e+00, -4.9106e+00, ..., 1.0888e+00,
1.5227e+00, 1.4921e+00],
[-8.5679e-01, -2.5115e+00, -5.7910e+00, ..., 9.5863e-01,
8.1060e-01, 2.0000e+00],
[ 1.2085e+00, 1.8527e+00, 4.7470e-01, ..., 4.2292e-01,
1.7869e-01, 1.4177e+00],
...,
[ 1.6356e+00, 3.7835e-01, -3.1863e-01, ..., 1.3224e+00,
2.8002e+00, 1.8160e+00],
[ 1.9230e+00, 1.0399e+00, 1.8171e+00, ..., -2.3007e-01,
1.9859e+00, 2.8625e+00],
[ 1.2035e+00, 5.5035e-01, 1.2808e+00, ..., 1.0474e-01,
1.1688e+00, 1.4895e+00]],
[[-4.1206e-01, -6.1092e-01, -1.9138e+00, ..., -5.7375e-01,
-8.8858e-01, -9.4433e-02],
[-1.1341e+00, -1.5447e+00, -2.9246e+00, ..., -1.1085e+00,
-9.5143e-01, -6.5436e-01],
[ 1.3725e-01, -6.1306e-01, -2.3306e+00, ..., -2.5181e-01,
-3.4704e-01, -1.8245e-01],
...,
[-1.2285e+00, -1.0898e+00, -5.4931e-01, ..., -5.2615e-01,
-7.7239e-01, -2.6630e-01],
[-1.5600e+00, -1.1218e+00, -9.8015e-01, ..., -3.5401e-01,
-1.1970e+00, 6.4692e-01],
[-4.3711e-02, 5.1822e-01, -1.0284e-01, ..., -6.3457e-01,
-1.1417e+00, 5.2388e-01]]],
[[[-1.5292e-01, 6.9261e-01, 1.2605e+00, ..., 6.0703e-02,
-8.0088e-03, -5.6007e-01],
[-1.5474e+00, -2.9190e+00, -3.6846e+00, ..., 5.2241e-01,
1.2646e+00, -2.7165e-01],
[ 1.8209e-01, -1.1294e+00, -8.1675e-01, ..., -2.1337e-02,
7.3482e-01, 9.2068e-01],
...,
[ 2.1883e-01, -5.5230e-01, 9.2349e-01, ..., 8.3383e-02,
5.5481e-01, 6.3424e-01],
[-6.3670e-01, -1.7011e+00, -1.8132e-01, ..., -6.5221e-01,
-4.5748e-01, -1.9320e-01],
[-3.4281e-01, -1.8343e+00, -1.9225e+00, ..., -1.1027e+00,
-1.5979e+00, -2.3026e-01]],
[[-1.3520e+00, 2.1537e+00, 1.6821e+00, ..., -2.4716e+00,
-3.3103e+00, -7.4779e-01],
[-4.3788e-01, -4.4146e-01, -5.3724e+00, ..., -1.4574e+00,
-2.1262e+00, -1.6633e-01],
[-1.5257e+00, 5.1461e-01, -3.3463e-01, ..., -2.3219e+00,
-1.8039e+00, -5.0412e-01],
...,
[-1.5739e+00, -3.9196e-01, -5.9178e-01, ..., -1.6780e+00,
-1.9079e+00, -6.7007e-01],
[-3.2284e+00, -1.2598e+00, -2.3304e+00, ..., -2.4912e+00,
-2.5167e+00, -1.3710e+00],
[-2.2368e+00, -2.0596e+00, -3.0927e+00, ..., -1.8487e+00,
-1.7534e+00, -8.5156e-01]],
[[ 7.0021e-01, -1.5381e+00, -3.0949e+00, ..., 2.2768e+00,
5.7868e-01, 2.1548e+00],
[-2.0893e+00, -5.2310e+00, -6.5642e+00, ..., 1.7395e+00,
-1.3472e-02, 2.1030e+00],
[-1.4956e+00, -1.1835e+00, -2.7068e+00, ..., 8.9729e-01,
8.5385e-01, 2.3378e+00],
...,
[ 1.4962e+00, 2.3418e+00, 1.1272e+00, ..., 2.4065e-01,
1.2158e+00, 2.2993e+00],
[ 7.4645e-01, 2.2888e+00, 2.3464e+00, ..., 3.3336e-02,
1.1388e+00, 1.8700e+00],
[ 1.5437e+00, 1.5890e+00, 2.1220e+00, ..., 1.2696e+00,
1.8342e+00, 1.7849e+00]],
...,
[[-1.8788e+00, -6.4037e+00, -7.9415e+00, ..., -3.1195e+00,
-2.7069e+00, -3.6803e+00],
[-3.5158e+00, -8.7241e+00, -6.2347e+00, ..., -5.1935e+00,
-4.5787e+00, -4.9997e+00],
[-1.7228e+00, -5.0538e+00, -1.7192e+00, ..., -2.5820e+00,
-2.3439e+00, -2.2087e+00],
...,
[-2.3431e+00, -5.1693e+00, -3.9311e+00, ..., -5.9080e+00,
-4.4016e+00, -4.7715e+00],
[-2.7754e+00, -5.7689e+00, -5.5023e+00, ..., -6.2101e+00,
-5.5032e+00, -6.0004e+00],
[-1.4582e+00, -4.0934e+00, -2.7275e+00, ..., -3.3129e+00,
-2.7712e+00, -3.1393e+00]],
[[-1.0005e+00, 1.6144e+00, 4.9041e-01, ..., 6.5008e-01,
9.6030e-01, 1.6720e+00],
[-6.7506e-01, -1.9421e-01, -1.1934e+00, ..., 1.1667e+00,
1.6303e+00, 2.4770e+00],
[ 5.9757e-01, -1.6160e+00, -3.5568e+00, ..., 1.0152e+00,
1.2217e+00, 1.8230e+00],
...,
[ 1.2594e+00, -1.1280e-01, 1.7421e-01, ..., 1.2504e+00,
1.0543e+00, 1.4887e+00],
[ 1.7871e+00, 1.6180e+00, 1.9653e-01, ..., 1.3547e+00,
1.2431e+00, 1.9402e+00],
[ 4.8385e-01, 6.8102e-01, 5.6086e-01, ..., 5.4491e-01,
7.5900e-01, 1.6955e+00]],
[[-1.2949e-01, 1.0258e+00, 4.6117e-01, ..., -4.1593e-01,
-4.8107e-01, 5.3791e-01],
[ 7.5245e-02, 9.1347e-01, -2.9861e+00, ..., -1.0485e-01,
4.8936e-01, 8.0386e-01],
[-1.8742e+00, 1.9938e-01, -4.1096e+00, ..., 3.6864e-01,
5.3624e-01, 5.0948e-03],
...,
[-1.9302e+00, -1.1628e+00, -8.3250e-01, ..., -5.2959e-01,
-8.7927e-01, -5.1276e-01],
[-1.5948e+00, -1.7872e+00, -9.5947e-01, ..., -1.2036e+00,
-1.5399e+00, -8.0030e-01],
[-8.0616e-01, -9.5180e-01, -1.1163e+00, ..., -5.4616e-01,
-9.2871e-01, -1.1341e-01]]],
[[[ 9.5692e-01, 1.5568e+00, 1.0677e+00, ..., 4.6600e-01,
3.0165e-01, -1.2270e-01],
[-8.8810e-01, -1.9699e+00, -1.0003e+00, ..., 3.2332e-01,
6.8043e-01, -5.8172e-01],
[ 1.6969e-02, -7.9301e-01, 1.4775e-02, ..., -5.9640e-01,
1.3603e-01, 6.8265e-01],
...,
[ 4.6160e-01, -2.8276e-01, -3.8960e-01, ..., 7.2000e-01,
1.3805e+00, -1.5114e+00],
[-1.8515e-01, -9.8261e-01, -3.4463e-01, ..., -2.4151e-01,
9.5100e-01, 3.2251e-01],
[ 7.3630e-01, -2.5918e-01, -2.8146e-01, ..., -4.4111e-01,
-4.0720e-01, 3.0291e-01]],
[[-4.1631e+00, -1.2676e+00, -2.1779e+00, ..., -2.7648e+00,
-3.2728e+00, -7.7896e-01],
[-4.0196e+00, -2.7716e+00, -2.2159e+00, ..., -1.8172e+00,
-3.0071e+00, -6.9849e-01],
[-2.6377e+00, -1.7752e+00, -2.5029e+00, ..., -2.7764e+00,
-3.0589e+00, -6.4092e-01],
...,
[-8.2748e-01, -5.6702e-01, -1.5708e+00, ..., -4.1060e+00,
-3.4720e+00, -9.1588e-01],
[-2.8499e+00, -3.2412e-01, -1.0294e+00, ..., -2.0823e+00,
-1.5848e+00, -3.9478e-02],
[-2.0598e+00, -1.4147e+00, -1.7056e+00, ..., -8.7533e-01,
-8.2201e-01, 9.4459e-02]],
[[ 3.0119e+00, 4.0286e+00, 3.2537e+00, ..., 2.1383e+00,
7.5663e-01, 2.3534e+00],
[ 3.7573e-01, 8.1136e-01, 2.3955e+00, ..., 1.6284e+00,
2.8741e-04, 2.1365e+00],
[ 8.8352e-01, 1.3212e+00, 1.6607e+00, ..., 5.3634e-01,
7.3380e-01, 2.0182e+00],
...,
[ 3.0195e-01, 1.0084e+00, 1.4797e+00, ..., 1.1593e+00,
-9.9687e-01, 1.4189e-01],
[ 3.0589e-01, 2.1719e+00, 1.6006e+00, ..., 3.4613e-01,
5.5753e-01, 1.2950e+00],
[ 1.4023e+00, 1.4505e+00, 1.7102e+00, ..., 1.0924e+00,
8.2133e-01, 1.3445e+00]],
...,
[[-2.0557e+00, -7.3298e+00, -5.9255e+00, ..., -3.9028e+00,
-2.7251e+00, -3.5747e+00],
[-2.7540e+00, -6.5335e+00, -4.6226e+00, ..., -5.3697e+00,
-4.4963e+00, -4.3595e+00],
[-3.0128e-01, -2.8115e+00, -2.9518e+00, ..., -4.5450e+00,
-3.0096e+00, -1.8471e+00],
...,
[-9.6661e-01, -2.3454e+00, -1.7776e+00, ..., -5.5944e+00,
-5.1438e+00, -3.7864e+00],
[-1.2908e+00, -3.8242e+00, -3.5154e+00, ..., -6.4796e+00,
-5.2042e+00, -5.5769e+00],
[-2.2170e+00, -5.0925e+00, -3.0283e+00, ..., -4.5206e+00,
-2.8490e+00, -3.3911e+00]],
[[-7.2940e-01, -1.1330e-01, 3.6085e-02, ..., 1.0376e+00,
1.0755e+00, 1.7958e+00],
[ 1.7946e+00, 1.0896e+00, 1.4682e+00, ..., 1.6414e+00,
1.4593e+00, 2.7634e+00],
[ 1.1991e+00, 2.5252e-01, 7.1668e-01, ..., 1.3269e+00,
9.2375e-01, 2.1352e+00],
...,
[ 1.3002e+00, -7.7181e-02, 6.6713e-01, ..., 6.8502e-01,
7.7334e-01, 1.5759e+00],
[ 1.8538e+00, 1.0167e+00, 1.6413e+00, ..., 1.7294e+00,
1.1856e+00, 2.3820e+00],
[ 1.0061e+00, 3.8238e-01, 8.3471e-01, ..., 9.8202e-01,
1.1939e+00, 1.5551e+00]],
[[-5.4699e-01, 5.6918e-01, 5.9632e-02, ..., -3.4555e-01,
-4.9075e-01, 9.5361e-01],
[-1.1383e+00, -1.1577e+00, -1.3660e+00, ..., -1.1142e-01,
2.4864e-02, 1.0607e+00],
[-2.0695e+00, -5.9051e-01, -2.3144e-01, ..., 2.3434e-01,
-1.5417e-01, 4.5977e-01],
...,
[-1.7359e+00, -1.5239e+00, -1.2991e+00, ..., -6.6285e-02,
-1.0364e+00, -1.4930e+00],
[-1.7303e+00, -1.0997e+00, -1.1168e+00, ..., -7.1313e-01,
-1.4201e+00, 6.9423e-02],
[-3.5679e-01, 3.5325e-01, -4.6738e-01, ..., 1.6129e-01,
-1.7938e-01, 6.6075e-01]]],
[[[ 5.8184e-01, 9.5310e-01, 4.2450e-01, ..., 3.0223e-01,
3.6094e-01, -3.8818e-02],
[-2.2328e-01, -1.3336e+00, -4.6450e-01, ..., 1.1587e-01,
1.1725e-02, -3.6117e-01],
[-6.4965e-01, -1.4055e+00, -1.0560e+00, ..., -1.5044e-01,
9.0815e-02, 7.1587e-01],
...,
[ 1.0650e-01, -3.7855e+00, 8.8775e-01, ..., 1.5711e+00,
1.3560e+00, 5.0158e-01],
[-2.6230e-02, -2.5335e+00, -6.2895e-01, ..., -1.9222e-01,
1.5040e+00, 1.9009e-01],
[ 8.3735e-01, -1.0930e+00, 1.4085e-02, ..., 1.1093e+00,
-2.0645e-01, 5.1817e-01]],
[[-2.8703e+00, -1.9673e+00, -3.0845e+00, ..., -1.7796e+00,
-2.3900e+00, -1.5310e+00],
[-3.9984e+00, -3.1166e+00, -3.7201e+00, ..., -1.4162e+00,
-2.5739e+00, -1.6995e+00],
[-4.9138e+00, -3.7213e+00, -1.9499e+00, ..., -2.0690e+00,
-2.3873e+00, -1.0838e+00],
...,
[-1.1134e+00, -2.7614e+00, -7.2778e+00, ..., -3.3476e+00,
-2.7312e+00, 4.8570e-01],
[-2.6735e+00, -9.0748e-01, -2.5512e+00, ..., -1.5780e+00,
-7.3271e-01, 9.4239e-01],
[-1.7220e+00, -2.3694e+00, -2.9853e+00, ..., -3.5048e-01,
-8.6559e-01, -3.2395e-01]],
[[ 2.0121e+00, 2.4144e+00, 1.4902e+00, ..., 1.6419e+00,
8.2773e-01, 2.1268e+00],
[-1.7520e-01, 9.5227e-01, 1.9792e+00, ..., 7.5970e-01,
3.6780e-01, 2.3042e+00],
[ 1.0231e+00, 6.2895e-01, 3.3062e-01, ..., -2.5094e-01,
7.5674e-01, 2.0100e+00],
...,
[ 3.9592e-01, 5.8726e+00, 1.5781e+00, ..., -4.1365e+00,
-1.3225e-01, 1.0141e+00],
[ 1.0986e+00, 3.6984e+00, 3.0974e+00, ..., -2.7066e+00,
7.8098e-01, 1.4972e+00],
[ 2.4280e+00, 1.5903e+00, 2.6741e+00, ..., 1.6901e+00,
5.7999e-01, 1.4074e+00]],
...,
[[-2.2120e+00, -5.4532e+00, -4.2147e+00, ..., -2.1004e+00,
-1.7113e+00, -3.1185e+00],
[-2.9546e+00, -6.5698e+00, -3.2205e+00, ..., -2.6221e+00,
-2.4500e+00, -3.2556e+00],
[-2.1100e+00, -5.1732e+00, -3.4555e+00, ..., -2.3476e+00,
-2.6504e+00, -2.5265e+00],
...,
[-2.8341e+00, -4.6457e+00, -7.2608e+00, ..., -9.4102e+00,
-6.5738e+00, -4.7224e+00],
[-2.4871e+00, -6.4088e+00, -6.1614e+00, ..., -9.0419e+00,
-5.6633e+00, -6.6141e+00],
[-2.6794e+00, -6.5707e+00, -3.8650e+00, ..., -7.3061e+00,
-5.1124e+00, -4.6023e+00]],
[[ 6.9428e-01, 1.0749e+00, 8.6619e-01, ..., 1.0264e+00,
1.4630e+00, 1.4707e+00],
[ 6.7218e-01, -9.7081e-02, -1.3752e+00, ..., 8.6737e-01,
8.2662e-01, 2.0499e+00],
[-8.2393e-01, -1.7132e+00, -5.4531e-01, ..., 3.4890e-01,
1.2022e-01, 1.4021e+00],
...,
[ 1.6428e+00, -2.6583e+00, -2.3509e-01, ..., 4.7419e-01,
1.4683e+00, 1.8874e+00],
[ 2.0167e+00, 2.5936e-01, 1.1161e+00, ..., 4.5604e-01,
1.9218e+00, 2.7878e+00],
[ 1.1095e+00, 1.1993e-01, -2.6481e-01, ..., 2.4108e+00,
1.6663e+00, 1.5804e+00]],
[[-1.7511e-01, -9.2863e-01, -1.1783e+00, ..., -5.9866e-01,
-8.7657e-01, -5.1487e-02],
[-1.3303e+00, -2.5455e+00, -2.3435e+00, ..., -1.1032e+00,
-8.9554e-01, -5.6231e-01],
[-2.3612e+00, -2.1753e+00, -8.0595e-01, ..., -1.7731e-01,
-3.2922e-01, -1.8683e-01],
...,
[-1.4115e+00, -1.2273e+00, -3.2545e+00, ..., -8.4564e-01,
-1.7426e+00, -1.3616e+00],
[-1.6109e+00, -1.4220e+00, -9.4504e-01, ..., -2.7723e+00,
-7.3917e-01, 6.8311e-01],
[-2.5050e-01, 2.7184e-01, 3.1760e-01, ..., 4.8838e-01,
6.0952e-01, 9.0555e-01]]]], device='cuda:0'),
tensor([[[[ 5.5242e-01, 3.1299e-01, -4.8775e-01, ..., 1.3588e-01,
9.0876e-02, -4.4965e-01],
[-7.3487e-01, 7.6536e-01, 1.4301e+00, ..., 5.0167e-01,
1.2071e+00, -3.4510e-01],
[ 1.7694e-01, -4.1051e-01, -4.9697e-02, ..., -2.5278e-01,
4.7563e-01, 9.2179e-01],
...,
[ 3.5526e-01, 3.1890e-01, 5.9606e-01, ..., -7.7467e-02,
3.1219e-01, 4.8401e-01],
[-7.3960e-01, -1.8983e+00, -1.0257e+00, ..., -6.1447e-01,
-2.3250e-01, -1.5234e-01],
[-2.8641e-01, -1.6663e+00, -1.6570e+00, ..., -1.1725e+00,
-1.5172e+00, -2.7260e-01]],
[[-3.8965e+00, -2.2720e+00, -1.8594e+00, ..., -2.4054e+00,
-3.2354e+00, -8.0960e-01],
[-1.0883e+00, 7.5601e-01, 3.3934e-01, ..., -1.3818e+00,
-2.0252e+00, -1.8432e-01],
[-2.1900e+00, -2.3582e+00, -2.4417e+00, ..., -2.5399e+00,
-1.9992e+00, -7.0138e-01],
...,
[-1.5772e+00, -1.4843e+00, -2.0739e+00, ..., -1.6956e+00,
-1.5359e+00, -3.9252e-01],
[-3.2359e+00, -1.7425e+00, -2.5777e+00, ..., -2.4630e+00,
-2.3854e+00, -1.1685e+00],
[-2.4113e+00, -1.9771e+00, -2.5642e+00, ..., -2.0176e+00,
-2.0122e+00, -1.0632e+00]],
[[ 2.8043e+00, 2.3538e+00, 1.8046e+00, ..., 2.3073e+00,
5.1206e-01, 2.2038e+00],
[ 7.8482e-01, 5.1324e-01, 1.0136e+00, ..., 1.7909e+00,
-1.3419e-01, 2.1789e+00],
[ 9.0604e-01, 7.5362e-02, 7.3880e-01, ..., 1.6264e+00,
1.3814e+00, 2.6720e+00],
...,
[ 1.7781e+00, 1.8762e+00, 1.6907e+00, ..., 7.4795e-01,
1.1612e+00, 2.1688e+00],
[ 8.0920e-01, 2.2207e+00, 2.0129e+00, ..., 7.9825e-02,
1.1451e+00, 1.8607e+00],
[ 1.6730e+00, 1.6462e+00, 1.8592e+00, ..., 1.3144e+00,
1.9056e+00, 1.8408e+00]],
...,
[[-1.9941e+00, -5.6373e+00, -2.0396e+00, ..., -3.0936e+00,
-2.7601e+00, -3.9002e+00],
[-2.7197e+00, -8.3989e+00, -5.6457e+00, ..., -5.0783e+00,
-4.3788e+00, -4.9797e+00],
[-1.7235e+00, -4.0881e+00, -2.0097e+00, ..., -3.7331e+00,
-3.6407e+00, -2.9237e+00],
...,
[-2.2470e+00, -4.8778e+00, -3.9499e+00, ..., -7.9978e+00,
-6.4769e+00, -5.4575e+00],
[-2.6698e+00, -5.6208e+00, -5.5540e+00, ..., -6.1113e+00,
-5.7488e+00, -6.2036e+00],
[-1.4356e+00, -4.1331e+00, -2.8709e+00, ..., -3.1898e+00,
-2.6430e+00, -3.2223e+00]],
[[-1.2337e+00, -8.7440e-01, 2.4295e-01, ..., 6.3715e-01,
9.6080e-01, 1.5859e+00],
[ 1.7322e+00, 6.9506e-01, 8.2369e-01, ..., 1.1415e+00,
1.6779e+00, 2.4719e+00],
[ 8.2737e-01, -3.3749e-01, -1.9126e-01, ..., 9.4183e-01,
1.1230e+00, 1.6620e+00],
...,
[ 1.2459e+00, 3.9099e-02, -3.6479e-01, ..., 2.0623e+00,
1.5454e+00, 1.6766e+00],
[ 1.8384e+00, 1.9735e+00, 1.4384e+00, ..., 1.3859e+00,
1.2543e+00, 2.0159e+00],
[ 3.8534e-01, 7.3601e-01, 7.0270e-01, ..., 5.3482e-01,
6.5744e-01, 1.6614e+00]],
[[-7.5111e-01, -1.3354e+00, -6.9483e-01, ..., -3.7191e-01,
-4.2825e-01, 5.7763e-01],
[-1.0681e+00, 1.1323e-01, 7.2179e-01, ..., -2.3247e-01,
4.1781e-01, 7.5019e-01],
[-1.4731e+00, -3.8682e-01, -8.8517e-02, ..., 3.1455e-02,
1.1283e-01, -4.9364e-01],
...,
[-1.8239e+00, -1.0917e+00, -4.6129e-01, ..., -1.6240e-01,
-4.8523e-01, -4.2655e-01],
[-1.5349e+00, -1.9891e+00, -1.7943e+00, ..., -1.1477e+00,
-1.3619e+00, -7.1004e-01],
[-8.3318e-01, -8.4174e-01, -1.1204e+00, ..., -6.1760e-01,
-8.5192e-01, -1.3780e-01]]],
[[[ 7.0791e-01, 1.0601e+00, 7.0963e-01, ..., 4.3546e-01,
3.7614e-01, 2.0467e-01],
[-3.0230e-01, -1.4216e+00, -4.7789e-01, ..., 3.9341e-01,
1.8523e-01, -1.2015e+00],
[ 1.0971e-01, -9.1978e-01, -2.4546e-01, ..., -2.0366e+00,
6.4085e-01, 6.7572e-01],
...,
[-9.9755e-03, -1.6608e+00, -8.3096e-01, ..., -1.3431e-01,
8.3600e-01, 5.0831e-01],
[-5.0142e-01, -2.0152e+00, -1.4847e-01, ..., -4.6719e-02,
1.3954e+00, -1.2409e-01],
[ 3.8069e-01, -7.9189e-01, -1.1067e+00, ..., -3.0540e-01,
3.7013e-01, 6.0227e-01]],
[[-3.1591e+00, -1.5691e+00, -3.1021e+00, ..., -1.6522e+00,
-2.1995e+00, -9.8806e-01],
[-4.4232e+00, -1.8005e+00, -2.3650e+00, ..., -3.0345e-02,
-1.4045e+00, -2.3695e-01],
[-4.4956e+00, -2.6524e+00, -3.1823e+00, ..., -1.1093e+00,
-1.4973e+00, -7.2215e-01],
...,
[-2.9998e+00, -2.0873e+00, -2.5587e+00, ..., -5.3936e-02,
2.1628e-01, 4.6833e-01],
[-3.6995e+00, -1.3930e+00, -2.3449e+00, ..., 2.0463e-01,
-3.9233e-01, 2.3218e-01],
[-1.2447e+00, -1.1228e+00, -2.2726e+00, ..., -4.2900e-01,
-8.0349e-01, -5.9204e-01]],
[[ 1.7807e+00, 2.8701e+00, 2.4032e+00, ..., 2.7921e+00,
1.3221e+00, 2.8054e+00],
[ 2.1972e-01, 1.2417e+00, 1.6063e+00, ..., 1.5808e+00,
-4.0788e-01, 2.4193e+00],
[ 7.2269e-01, 1.4184e+00, 1.0125e+00, ..., 2.6245e+00,
9.0766e-01, 1.8090e+00],
...,
[ 1.0415e+00, 2.1245e+00, 1.7021e+00, ..., 1.2665e+00,
-2.1968e-01, 1.4613e+00],
[ 7.0037e-01, 2.2802e+00, 2.5291e+00, ..., 6.6687e-01,
6.5020e-01, 1.5890e+00],
[ 1.5482e+00, 1.3707e+00, 2.0850e+00, ..., 1.4549e+00,
1.1192e+00, 2.0316e+00]],
...,
[[-1.4908e+00, -4.7142e+00, -3.9625e+00, ..., -3.2839e+00,
-2.3628e+00, -3.8776e+00],
[-1.9445e+00, -4.0558e+00, -4.4866e+00, ..., -5.3815e+00,
-3.9275e+00, -4.2788e+00],
[-9.8362e-01, -3.2161e+00, -3.8142e+00, ..., -1.6974e+00,
-2.4824e+00, -1.5725e+00],
...,
[-1.5657e+00, -4.3062e+00, -4.9619e+00, ..., -5.7824e+00,
-4.0062e+00, -1.5114e+00],
[-1.4769e+00, -4.9941e+00, -5.8145e+00, ..., -7.7496e+00,
-3.6903e+00, -3.7547e+00],
[-1.5925e+00, -4.9590e+00, -3.9837e+00, ..., -3.8994e+00,
-2.1999e+00, -2.7724e+00]],
[[ 7.5292e-01, 5.0875e-01, -5.4586e-02, ..., 8.6036e-01,
1.0609e+00, 1.2780e+00],
[ 1.2262e+00, 1.0090e+00, 5.1749e-01, ..., 1.4455e+00,
2.1581e+00, 2.5740e+00],
[ 5.4363e-01, 5.5974e-01, -3.7911e-03, ..., 2.5877e+00,
2.1763e+00, 1.9118e+00],
...,
[ 7.6747e-01, -2.4355e-01, -3.3072e-01, ..., 2.3596e+00,
2.0685e+00, 1.6915e+00],
[ 1.5057e+00, 1.2433e+00, -2.5478e-02, ..., 1.6878e+00,
1.5271e+00, 2.4596e+00],
[ 6.2649e-01, 8.7809e-01, 8.1439e-01, ..., 1.0268e+00,
1.3064e+00, 1.3516e+00]],
[[-2.7944e-01, -1.9745e-01, -1.0073e+00, ..., -1.8131e-01,
-1.7674e-01, 8.9481e-01],
[-1.3792e+00, -1.3795e+00, -2.0233e+00, ..., -2.1550e-01,
-3.1266e-01, 3.9184e-01],
[-1.7227e+00, -8.7633e-01, -1.0623e+00, ..., -1.1382e+00,
1.7733e-01, 2.1743e-01],
...,
[-1.7900e+00, -1.4032e+00, -1.3955e+00, ..., -3.3482e-01,
-4.1658e-02, -3.1478e-01],
[-1.4792e+00, -1.3977e+00, -5.7044e-01, ..., -4.9387e-02,
-2.9466e-01, 3.2403e-01],
[-1.5781e-01, 4.6176e-02, -4.8067e-01, ..., 2.5011e-01,
1.4014e-01, 3.1264e-01]]],
[[[ 5.6688e-01, 7.0952e-01, 4.5500e-01, ..., 1.0367e-01,
-1.6380e-01, -4.5383e-01],
[-2.8948e-01, -6.0846e-01, -1.6317e-01, ..., -1.4432e+00,
-2.7705e+00, -6.3448e-01],
[-1.8502e-01, -1.6901e-01, -5.9912e-01, ..., 2.1812e+00,
-4.7499e-01, -4.7277e-02],
...,
[ 1.3744e+00, -2.4059e+00, 1.4831e+00, ..., -1.3203e+00,
-2.4415e-01, -1.2427e-01],
[ 7.7932e-01, -5.5850e+00, 3.8384e-01, ..., -7.1094e-02,
8.8667e-01, -1.6224e-01],
[ 1.7111e+00, -1.1991e+00, -2.7152e+00, ..., 3.8001e-01,
2.4852e-01, 7.4615e-01]],
[[-2.8141e+00, -9.5946e-01, -2.2836e+00, ..., -2.2501e+00,
-2.5705e+00, -1.4128e+00],
[-2.4314e+00, -1.0776e+00, -4.1679e+00, ..., -4.3205e+00,
-3.8913e+00, -2.0186e+00],
[-2.3668e+00, -2.4551e+00, -6.7258e+00, ..., -4.4837e+00,
-5.1638e+00, -1.4182e+00],
...,
[-1.9104e+00, 2.6862e+00, 9.7290e-01, ..., -1.4973e+01,
-8.8921e+00, -2.1199e+00],
[-4.1970e+00, 9.0687e-01, -9.8321e-01, ..., -2.1599e+00,
-6.1646e-01, 5.3113e-01],
[-2.4111e+00, -5.5090e+00, -9.0338e+00, ..., -4.3790e-01,
-6.1989e-01, -2.7118e-01]],
[[ 2.1526e+00, 2.5664e+00, 1.9665e+00, ..., 2.2040e+00,
1.1657e+00, 2.1053e+00],
[ 1.0865e-01, 1.6168e-01, 2.4240e-01, ..., 1.0457e+00,
3.0246e+00, 2.2042e+00],
[ 3.4378e-01, 7.6791e-01, 1.2877e+00, ..., -4.4703e+00,
4.4258e-01, 2.2789e+00],
...,
[-6.5633e+00, -4.7929e+00, -1.5501e+00, ..., -9.4177e+00,
-8.0316e+00, -8.7968e-01],
[-3.5225e+00, -5.7501e+00, -4.6991e+00, ..., -5.3018e-01,
1.2579e-01, 1.7023e+00],
[ 3.2101e+00, -1.4782e-01, -2.5788e+00, ..., 2.0283e+00,
1.1006e+00, 1.8514e+00]],
...,
[[-2.0748e+00, -4.9071e+00, -3.0484e+00, ..., -2.9383e+00,
-2.1958e+00, -3.5729e+00],
[-2.6477e+00, -5.6854e+00, -4.1252e+00, ..., -2.4357e+00,
-3.1479e+00, -5.1734e+00],
[-1.5932e+00, -5.5324e+00, -4.4818e+00, ..., -5.5231e+00,
-1.2245e+01, -7.9789e+00],
...,
[-3.2758e+00, -5.2447e+00, -2.8848e+00, ..., -1.7362e+01,
-1.3938e+01, -8.8281e+00],
[-2.7352e+00, -6.5558e+00, -9.0932e+00, ..., -1.0043e+01,
-8.5167e+00, -8.4503e+00],
[-2.4221e+00, -5.6536e+00, -6.3821e+00, ..., -3.3772e+00,
-2.2751e+00, -3.7525e+00]],
[[ 6.5104e-01, 1.3929e+00, 6.5975e-01, ..., 6.7096e-01,
1.3111e+00, 1.5870e+00],
[ 1.2405e+00, 4.4774e-01, 1.1154e+00, ..., -2.3108e+00,
1.2102e+00, 2.5404e+00],
[ 6.5907e-01, -1.5494e+00, -3.6736e+00, ..., -5.1309e+00,
-2.2381e-01, 2.7203e+00],
...,
[ 2.9712e-01, -3.7597e+00, -4.6622e+00, ..., -4.4686e+00,
-4.0744e-01, 1.0717e+00],
[ 3.1231e+00, 9.5078e-01, -5.2696e-01, ..., 2.8998e+00,
2.2354e+00, 3.3008e+00],
[-1.7654e+00, -1.8363e+00, -2.5967e+00, ..., 1.4555e+00,
1.5307e+00, 1.5404e+00]],
[[-2.3982e-01, -7.1502e-01, -1.1251e+00, ..., -6.9229e-01,
-1.4464e+00, -2.2704e-01],
[-9.1441e-01, -2.8068e+00, -2.6150e+00, ..., -1.3390e+00,
-2.9395e+00, -1.7424e-01],
[-1.6085e+00, -2.9686e+00, -1.4594e+00, ..., 5.2677e+00,
-1.2540e+00, -1.1716e+00],
...,
[-2.2166e+00, 2.8743e-01, -3.4003e-01, ..., -9.2121e+00,
-3.9220e+00, -2.5227e+00],
[-2.0332e+00, -9.8783e-01, 5.4705e-01, ..., -2.1961e+00,
-3.3565e-02, 1.2648e+00],
[ 7.4305e-01, -4.8148e-01, -1.4398e+00, ..., 2.1866e-01,
2.0215e-01, 7.5961e-01]]],
[[[ 9.3785e-01, 1.5903e+00, 1.3365e+00, ..., -9.5747e-01,
5.2592e-01, 3.1634e-01],
[-7.8799e-01, -2.0209e+00, -8.3475e-01, ..., 3.3799e-03,
-1.1826e+00, -6.5085e-01],
[-3.8896e-01, -1.3998e+00, -1.3964e+00, ..., 1.8561e+00,
3.2370e-01, 5.9031e-01],
...,
[-2.9866e-01, 9.0592e-01, 2.3604e-01, ..., -5.6682e-02,
-1.3315e-01, 2.9348e-01],
[-8.1754e-01, -4.2047e-01, 1.3801e+00, ..., -5.5972e-02,
1.5393e-02, -1.3285e-01],
[ 6.4257e-01, -5.9397e-01, 1.9715e-01, ..., -3.4920e-01,
-9.9305e-01, 1.7575e-01]],
[[-4.2736e+00, -1.1662e+00, -3.2048e+00, ..., -1.5495e+00,
-2.2117e+00, -1.8732e+00],
[-4.7647e+00, -4.4589e+00, -5.3755e+00, ..., -2.2029e+00,
-2.5254e+00, -2.0161e+00],
[-4.2170e+00, -5.3465e+00, -7.6947e+00, ..., -2.4525e+00,
-1.6583e+00, -1.1764e+00],
...,
[-1.6633e+00, -1.7370e+00, 6.0114e-01, ..., -2.7040e-01,
-1.0346e+00, -8.6011e-01],
[-3.1489e+00, -7.2341e-01, -1.3548e+00, ..., -5.2891e-01,
-1.1000e+00, -1.1421e+00],
[-2.2251e+00, -1.6509e+00, -9.2310e-01, ..., -6.6825e-01,
-7.2347e-01, -4.5589e-01]],
[[ 3.1336e+00, 4.2533e+00, 3.1990e+00, ..., 3.0907e+00,
8.4726e-01, 2.4910e+00],
[ 5.7546e-01, 9.9684e-01, 3.0112e+00, ..., 1.4952e+00,
1.5425e+00, 2.7487e+00],
[ 1.3486e+00, 2.5800e+00, 3.0942e+00, ..., -9.3498e-01,
1.0548e+00, 2.4417e+00],
...,
[ 1.0510e+00, 1.3618e+00, -1.4139e+00, ..., -4.3621e-01,
6.0488e-01, 1.5069e+00],
[ 4.2622e-01, 3.4745e+00, 1.4671e+00, ..., -9.8730e-02,
9.0886e-01, 1.1912e+00],
[ 1.4157e+00, 1.7413e+00, 1.8174e+00, ..., 6.2941e-01,
1.3584e+00, 1.3791e+00]],
...,
[[-2.1230e+00, -7.3385e+00, -5.8845e+00, ..., -3.6487e+00,
-2.7283e+00, -4.4557e+00],
[-2.9705e+00, -6.6676e+00, -5.7897e+00, ..., -9.0194e+00,
-4.0124e+00, -3.8502e+00],
[-9.7044e-01, -2.5429e+00, -1.4777e+00, ..., -1.1723e+01,
-8.1413e+00, -2.0641e+00],
...,
[-1.1242e+00, -4.3024e+00, -3.1125e+00, ..., -4.9420e+00,
-3.4896e+00, -2.5863e+00],
[-1.3303e+00, -4.4625e+00, -4.6303e+00, ..., -5.2870e+00,
-4.4886e+00, -4.1349e+00],
[-2.1326e+00, -4.8937e+00, -3.1503e+00, ..., -3.5427e+00,
-2.6699e+00, -2.3444e+00]],
[[-6.9474e-01, -8.1350e-02, -6.6510e-01, ..., 2.8586e+00,
1.8864e+00, 1.2375e+00],
[ 1.5052e+00, 5.1788e-01, 3.6784e-01, ..., 5.8393e-02,
2.7802e+00, 1.9631e+00],
[ 6.3052e-01, -1.9587e+00, -3.1276e+00, ..., -1.7504e+00,
1.9844e+00, 1.8946e+00],
...,
[ 2.0764e-01, -1.2389e+00, 2.4927e+00, ..., 1.7279e+00,
6.6453e-01, 1.0668e+00],
[ 2.0283e+00, 4.4116e-01, 9.6622e-01, ..., 1.3606e+00,
6.8288e-01, 1.6228e+00],
[ 1.1230e+00, 7.5562e-01, 1.2030e+00, ..., 7.8833e-01,
6.6248e-01, 1.4682e+00]],
[[-5.0878e-01, 6.4134e-01, -2.7983e-01, ..., -1.2171e+00,
-5.5668e-01, -4.4228e-03],
[-1.0916e+00, -1.2355e+00, -1.7157e+00, ..., -2.3380e+00,
-2.1132e+00, -8.9255e-01],
[-2.4443e+00, -7.5159e-01, -1.1640e+00, ..., 6.7392e-01,
-2.4368e+00, 1.8346e-01],
...,
[-1.8046e+00, -7.2974e-01, -3.7187e-01, ..., -2.4448e-01,
-1.6650e+00, -7.8968e-01],
[-1.6298e+00, 2.4742e-02, -2.9578e-01, ..., -6.0772e-01,
-1.7877e+00, -6.3783e-01],
[-2.2674e-01, 5.9196e-01, 2.4516e-02, ..., -2.4037e-01,
-8.5033e-01, -9.8671e-03]]],
[[[ 1.0534e+00, 2.0458e+00, -1.2211e+00, ..., 8.3722e-01,
9.5740e-01, 3.6001e-01],
[-1.2596e+00, -4.0740e-01, -5.4052e+00, ..., 7.8209e-03,
1.5788e-01, -2.8761e-01],
[-8.0878e-01, 4.1966e-01, -4.7090e+00, ..., -4.4839e-01,
-1.8636e-01, 2.0201e-01],
...,
[ 4.2037e-01, -8.4972e-02, 2.0213e-01, ..., -1.1964e+00,
-2.1103e+00, 8.4089e-01],
[-2.7224e-01, -1.1880e+00, -1.8415e-01, ..., 1.7028e+00,
8.0048e-01, 1.1716e+00],
[ 7.7004e-01, -2.5713e-01, -2.7950e-01, ..., -7.8407e-01,
-3.1373e+00, -4.7693e-01]],
[[-3.9158e+00, -2.5005e+00, 2.4170e+00, ..., -1.3864e+00,
-1.6557e+00, -1.5104e+00],
[-4.7523e+00, -7.0581e+00, 2.4499e+00, ..., 4.9570e-01,
-1.1394e+00, -1.7262e+00],
[-4.3122e+00, -6.1488e+00, 1.6127e+00, ..., -1.3951e-01,
-1.7506e+00, -1.1650e+00],
...,
[-1.0178e+00, -1.0165e+00, -2.0212e+00, ..., -3.4264e+00,
-1.7827e+00, -2.5471e+00],
[-3.0164e+00, -5.9478e-01, -1.2667e+00, ..., -3.8675e+00,
-4.6004e+00, -4.6608e+00],
[-2.0570e+00, -1.3635e+00, -1.4683e+00, ..., -1.0410e+01,
-7.1565e+00, -3.2056e+00]],
[[ 3.4814e+00, -1.7634e-01, -7.5595e-01, ..., 1.3706e+00,
7.0557e-01, 2.0728e+00],
[ 1.5969e+00, -8.4435e+00, -3.1952e+00, ..., -1.5358e-01,
-1.5763e-01, 1.6211e+00],
[ 1.7390e+00, -7.6794e+00, -3.2067e+00, ..., -7.0486e-01,
1.2515e-01, 1.1138e+00],
...,
[ 3.9687e-01, 9.7419e-01, 1.5968e+00, ..., -3.4946e+00,
-2.8720e+00, 1.9859e+00],
[ 2.4722e-01, 2.1914e+00, 2.1350e+00, ..., -4.7273e+00,
-1.9059e+00, 3.0526e-01],
[ 1.3645e+00, 1.3576e+00, 1.6326e+00, ..., -7.6234e+00,
-2.7019e-01, 2.0450e+00]],
...,
[[-2.3173e+00, -8.0032e+00, -7.2185e+00, ..., -4.0319e+00,
-2.2698e+00, -2.8973e+00],
[-2.9402e+00, -8.6198e+00, -7.1038e+00, ..., -3.5834e+00,
-3.6793e+00, -2.4938e+00],
[-7.1951e-01, -4.4901e+00, -5.1397e+00, ..., -7.6712e-01,
-2.3356e+00, 5.8619e-02],
...,
[-1.0074e+00, -2.3409e+00, -1.6986e+00, ..., 9.9316e-01,
-5.8886e-01, -6.1449e+00],
[-1.2963e+00, -3.5796e+00, -3.3793e+00, ..., -5.9025e+00,
-1.1641e+01, -1.3814e+01],
[-2.2494e+00, -5.0562e+00, -3.0941e+00, ..., -4.7110e+00,
-3.7914e+00, -4.7342e+00]],
[[-1.2359e+00, -6.7004e-01, 1.9481e+00, ..., 8.0763e-01,
1.5500e+00, 1.2916e+00],
[ 4.5693e-01, -1.1043e+00, 1.7904e+00, ..., 1.2896e+00,
8.5934e-01, 1.3904e+00],
[ 3.5519e-01, -1.3232e-02, 1.8920e+00, ..., 8.9567e-01,
1.0623e-01, 1.0590e+00],
...,
[ 1.2382e+00, -5.1792e-01, 2.6776e-01, ..., -1.2170e+00,
-1.1886e+00, -7.1558e-01],
[ 1.8639e+00, 9.8032e-01, 1.6964e+00, ..., -1.7979e-01,
-9.3884e-01, -1.8920e+00],
[ 1.0177e+00, 4.2008e-01, 1.0043e+00, ..., -1.7137e+00,
-1.6140e+00, 7.9060e-01]],
[[-8.1501e-01, 1.2108e+00, 2.1325e+00, ..., -2.5835e-01,
-4.4836e-01, 1.7573e-01],
[-1.5210e+00, -2.1833e+00, 1.4599e+00, ..., 3.6014e-02,
-5.1089e-01, -7.9097e-01],
[-2.3373e+00, -6.8279e-01, 2.4885e+00, ..., -1.4090e-01,
-1.7702e+00, -5.0937e-01],
...,
[-1.8025e+00, -1.3030e+00, -1.1259e+00, ..., -7.2967e-01,
1.1764e+00, -6.4550e-01],
[-1.7790e+00, -1.1470e+00, -1.0107e+00, ..., 2.7525e+00,
1.9301e-01, -3.8189e+00],
[-3.5461e-01, 4.1423e-01, -3.4002e-01, ..., -5.1453e+00,
-7.3277e+00, -2.4019e+00]]]], device='cuda:0'),
tensor([[[[ 9.4467e-01, 1.5691e+00, 1.0502e+00, ..., 5.9880e-01,
3.4639e-01, -1.1794e-01],
[-8.3788e-01, -1.8228e+00, -9.4284e-01, ..., 3.4861e-01,
5.6588e-01, -6.7370e-01],
[ 2.6238e-02, -5.8400e-01, -7.4988e-02, ..., -7.7435e-01,
1.7180e-01, 6.8935e-01],
...,
[ 6.3166e-01, -3.4394e-01, -2.8383e-01, ..., 1.0960e+00,
-2.8435e-01, -1.1353e-01],
[-6.9682e-02, -1.1267e+00, -3.2894e-01, ..., 3.4424e-01,
1.2712e+00, 1.4158e-01],
[ 8.4615e-01, -3.3716e-01, -2.9106e-01, ..., -5.6460e-01,
-3.8341e-01, 2.2521e-01]],
[[-3.7687e+00, -1.2497e+00, -1.9316e+00, ..., -2.8767e+00,
-3.5196e+00, -9.2707e-01],
[-3.8927e+00, -3.0096e+00, -2.2137e+00, ..., -2.0545e+00,
-3.2430e+00, -6.4942e-01],
[-2.4322e+00, -1.6992e+00, -2.6202e+00, ..., -2.9973e+00,
-3.3266e+00, -7.0523e-01],
...,
[-8.3980e-01, -3.5131e-01, -1.5540e+00, ..., -1.7812e+00,
-1.3944e+00, 1.3896e-01],
[-3.0931e+00, -9.1388e-02, -8.1468e-01, ..., -9.9932e-01,
-1.3896e+00, -7.9304e-02],
[-2.2834e+00, -1.5937e+00, -1.6788e+00, ..., -1.0326e+00,
-1.1578e+00, -3.2985e-02]],
[[ 2.6723e+00, 3.7581e+00, 3.0762e+00, ..., 2.3244e+00,
7.6063e-01, 2.4168e+00],
[ 7.7844e-03, 7.5760e-01, 2.1997e+00, ..., 1.8257e+00,
1.3215e-01, 2.4530e+00],
[ 5.9480e-01, 1.3439e+00, 1.5746e+00, ..., 8.4617e-01,
8.8515e-01, 2.1887e+00],
...,
[ 6.4099e-01, 1.0946e+00, 1.4934e+00, ..., -6.3851e-01,
-1.0633e+00, 4.4667e-01],
[ 6.8744e-01, 2.3413e+00, 1.8936e+00, ..., 2.6016e-01,
5.2849e-01, 1.3297e+00],
[ 1.7730e+00, 1.3435e+00, 1.9284e+00, ..., 1.0041e+00,
8.5852e-01, 1.3332e+00]],
...,
[[-1.6338e+00, -6.8602e+00, -5.6179e+00, ..., -4.2055e+00,
-3.1361e+00, -3.9559e+00],
[-2.2412e+00, -6.0269e+00, -4.3508e+00, ..., -5.5511e+00,
-4.7494e+00, -4.6687e+00],
[-2.5557e-01, -2.5193e+00, -2.8697e+00, ..., -4.2510e+00,
-3.1696e+00, -2.0462e+00],
...,
[-1.1553e+00, -2.4679e+00, -1.7481e+00, ..., -4.7993e+00,
-3.1169e+00, -2.2680e+00],
[-1.6621e+00, -4.0362e+00, -3.4037e+00, ..., -6.2826e+00,
-4.4638e+00, -4.3832e+00],
[-2.6224e+00, -5.4440e+00, -2.8952e+00, ..., -4.3609e+00,
-2.7612e+00, -3.1919e+00]],
[[-6.1178e-01, -1.0930e-01, 1.8464e-01, ..., 1.0271e+00,
1.0417e+00, 1.7588e+00],
[ 1.6488e+00, 8.6997e-01, 1.5039e+00, ..., 1.4869e+00,
1.5005e+00, 2.9397e+00],
[ 1.0967e+00, 2.2120e-01, 4.1762e-01, ..., 1.3968e+00,
8.4745e-01, 2.2673e+00],
...,
[ 1.3930e+00, -1.4142e-02, 7.7266e-01, ..., 1.5004e+00,
1.0394e+00, 1.5659e+00],
[ 1.9970e+00, 1.2937e+00, 1.7139e+00, ..., 1.6705e+00,
8.9138e-01, 2.1164e+00],
[ 9.9710e-01, 5.7688e-01, 1.0271e+00, ..., 8.6646e-01,
1.0968e+00, 1.5935e+00]],
[[-4.7607e-01, 5.6541e-01, 9.4388e-02, ..., -2.0077e-01,
-5.5870e-01, 9.1681e-01],
[-1.0591e+00, -1.2060e+00, -1.3297e+00, ..., -2.8963e-01,
-1.6832e-01, 1.0497e+00],
[-1.9091e+00, -5.5537e-01, -4.1038e-01, ..., 1.6116e-01,
-1.1436e-01, 5.3350e-01],
...,
[-1.8841e+00, -1.5596e+00, -1.2444e+00, ..., -4.9606e-01,
-8.0690e-01, -7.0108e-01],
[-1.8391e+00, -1.1740e+00, -1.0599e+00, ..., -4.9685e-01,
-8.7605e-01, 2.2955e-01],
[-5.0331e-01, 2.0661e-01, -3.9632e-01, ..., 8.9669e-02,
-2.9716e-01, 6.0695e-01]]],
[[[ 1.1754e+00, 1.6714e+00, 5.3602e-01, ..., 5.1742e-01,
1.5864e-01, 6.6813e-01],
[-6.9874e-01, -2.5012e+00, -2.2041e+00, ..., -2.6976e+00,
-1.1400e+00, -6.1224e-01],
[-4.1639e-01, -3.5403e+00, -2.8352e+00, ..., -2.9339e-01,
3.9514e-02, -2.9908e-01],
...,
[-9.7304e-02, -1.8164e+00, 1.4097e+00, ..., 3.9778e-01,
-2.1108e-02, 4.2243e-01],
[-2.6536e-01, -1.6405e+00, -1.1270e+00, ..., 3.2904e-01,
5.1277e-01, 6.9688e-02],
[ 7.4774e-01, -7.1607e-01, -8.1436e-01, ..., -6.3381e-02,
-9.3026e-01, 1.3963e-01]],
[[-4.9860e+00, -3.3326e+00, -2.9843e+00, ..., -3.0548e+00,
-3.2323e+00, -2.1119e+00],
[-5.7950e+00, -7.6203e+00, -2.7072e+00, ..., 8.5479e-01,
-9.2463e-01, -1.3194e+00],
[-5.1043e+00, -9.2065e+00, 1.2134e+00, ..., 1.6587e+00,
-1.5154e+00, -1.2679e+00],
...,
[-1.6580e+00, -3.0965e+00, -7.4801e+00, ..., -2.1871e+00,
-1.7237e+00, -1.0905e+00],
[-3.0954e+00, -1.5249e+00, -4.2617e+00, ..., -1.2095e+00,
-8.2923e-01, -9.8642e-01],
[-2.2684e+00, -2.0000e+00, -2.4223e+00, ..., -5.9872e-01,
-4.3869e-01, -4.3022e-01]],
[[ 3.1341e+00, 4.3740e+00, 3.9972e+00, ..., 1.7352e+00,
-3.4896e-01, -1.0177e+00],
[ 2.8692e-01, 6.8965e-01, 2.2875e+00, ..., -1.8975e+00,
-1.6080e+00, -1.6163e+00],
[ 1.6037e+00, -6.9801e-01, 3.9325e-01, ..., -9.4256e+00,
-3.2971e+00, -1.8699e+00],
...,
[-5.5910e-01, 3.5614e+00, 2.0647e+00, ..., -1.4327e+00,
-9.2766e-02, 1.7681e+00],
[ 2.3857e-02, 2.0302e+00, 3.1689e+00, ..., 5.0666e-01,
1.4778e+00, 1.6374e+00],
[ 1.3172e+00, 1.0783e+00, 1.6973e+00, ..., 8.9391e-01,
1.4418e+00, 1.5885e+00]],
...,
[[-1.9507e+00, -7.8859e+00, -6.3726e+00, ..., -6.6575e+00,
-3.2464e+00, -2.8065e+00],
[-3.0418e+00, -9.1270e+00, -8.9494e+00, ..., -6.3129e+00,
-7.0183e+00, -4.5850e+00],
[-8.3507e-01, -8.7105e+00, -1.0603e+01, ..., -8.2866e+00,
-1.2697e+01, -4.2134e+00],
...,
[-1.4417e+00, -4.2634e+00, -5.9933e+00, ..., -5.8299e+00,
-4.3137e+00, -2.6957e+00],
[-1.2991e+00, -4.2757e+00, -5.0224e+00, ..., -5.2783e+00,
-4.9729e+00, -4.8756e+00],
[-2.1789e+00, -5.0650e+00, -3.0939e+00, ..., -3.7730e+00,
-3.0288e+00, -2.7120e+00]],
[[-1.1859e+00, -6.6602e-01, -1.3033e-01, ..., -9.7037e-01,
-2.3624e-02, 3.4088e-02],
[-3.7654e-02, -4.9798e+00, -7.3384e-01, ..., 5.0707e-01,
8.3623e-01, 2.2889e+00],
[-3.1593e+00, -6.5872e+00, 1.4927e+00, ..., 2.5688e-01,
-2.8528e+00, 6.3236e-01],
...,
[ 1.3530e+00, -1.4327e+00, -1.2307e+00, ..., 4.6906e-01,
7.5685e-01, 1.1201e+00],
[ 1.7779e+00, 2.4584e-01, 1.4948e-01, ..., 1.4894e+00,
1.0861e+00, 1.9029e+00],
[ 1.0098e+00, 2.0573e-01, 5.2297e-01, ..., 9.9654e-01,
8.6228e-01, 1.5990e+00]],
[[-8.0378e-01, 6.5550e-01, 3.2582e-01, ..., -6.4314e-01,
-1.3938e+00, -4.2290e-02],
[-2.2039e+00, -1.1648e+00, -1.9622e+00, ..., 1.7415e+00,
2.1637e-01, -5.2840e-01],
[-4.3618e+00, -1.4129e+00, 3.3793e+00, ..., 2.2503e+00,
-1.0468e+00, -4.2480e+00],
...,
[-1.5205e+00, -6.7712e-01, -2.4130e+00, ..., -1.4156e+00,
-2.8300e+00, -1.5685e+00],
[-1.5477e+00, -1.7290e+00, -6.6262e-01, ..., -1.4827e+00,
-1.7803e+00, -5.1150e-01],
[-2.7969e-01, 3.0187e-01, -5.2666e-01, ..., 4.8164e-03,
-5.7487e-01, 1.0616e-01]]],
[[[ 6.4430e-01, 9.8835e-01, 7.8209e-01, ..., 3.2614e-01,
3.9289e-01, -2.7009e-02],
[-2.0907e-01, -1.1153e+00, -5.3280e-01, ..., 1.5322e-01,
8.5951e-02, -3.2098e-01],
[-1.9625e-01, -1.2088e+00, -3.5350e-01, ..., 6.1993e-02,
1.7889e-01, 6.9786e-01],
...,
[-2.2024e-01, -2.4751e-01, -1.3115e+00, ..., -3.1944e-01,
8.1173e-01, 8.9674e-01],
[-1.1321e+00, -1.4131e+00, -8.0658e-01, ..., -8.0872e-04,
6.3633e-01, 1.6379e-01],
[ 2.9564e-01, -9.6600e-02, 4.4159e-02, ..., 2.1753e-01,
-5.7847e-01, 2.2826e-01]],
[[-2.8608e+00, -1.5203e+00, -2.5940e+00, ..., -1.6353e+00,
-2.2021e+00, -1.4778e+00],
[-3.2985e+00, -2.9837e+00, -2.8455e+00, ..., -1.3823e+00,
-2.4289e+00, -1.6846e+00],
[-3.6085e+00, -2.7812e+00, -1.2066e+00, ..., -1.9692e+00,
-2.3626e+00, -1.1795e+00],
...,
[-1.5137e+00, -1.1718e+00, 9.9066e-01, ..., -1.5804e+00,
-1.5160e+00, 2.8641e-02],
[-2.5408e+00, 6.0592e-01, 7.0546e-01, ..., -1.4150e+00,
-1.4939e+00, -2.9136e-01],
[-1.2452e+00, -8.1325e-01, -3.8165e-01, ..., -2.3862e+00,
-1.6135e+00, -6.7187e-01]],
[[ 2.0247e+00, 2.5961e+00, 2.0705e+00, ..., 1.6436e+00,
7.5647e-01, 2.1058e+00],
[ 3.5168e-01, 1.0774e+00, 1.6557e+00, ..., 7.9484e-01,
3.7153e-01, 2.3147e+00],
[ 1.4910e+00, 2.0449e-01, -6.1821e-01, ..., 6.9332e-02,
7.8285e-01, 2.0712e+00],
...,
[ 2.1854e+00, 7.4842e-01, -4.0575e+00, ..., 4.2761e-01,
-2.5214e-01, 1.2783e+00],
[ 1.6566e+00, 1.7171e+00, -4.2387e+00, ..., -1.0835e+00,
1.5974e-03, 9.3869e-01],
[ 2.6984e+00, 2.1461e+00, -4.2458e+00, ..., 1.1226e+00,
4.9481e-01, 1.2158e+00]],
...,
[[-2.0137e+00, -5.3927e+00, -4.0189e+00, ..., -2.1510e+00,
-1.7351e+00, -3.1103e+00],
[-2.6721e+00, -5.7231e+00, -2.8629e+00, ..., -2.6632e+00,
-2.4547e+00, -3.2050e+00],
[-1.5265e+00, -5.4782e+00, -4.2397e+00, ..., -2.0578e+00,
-2.2857e+00, -2.2174e+00],
...,
[-2.1996e+00, -5.7152e+00, -2.8840e+00, ..., -3.2814e+00,
-2.4057e+00, -3.0319e+00],
[-2.7569e+00, -7.3864e+00, -3.3724e+00, ..., -6.2077e+00,
-3.1400e+00, -4.4430e+00],
[-2.9677e+00, -7.3812e+00, -2.2778e+00, ..., -4.5812e+00,
-1.7889e+00, -2.8219e+00]],
[[ 6.7780e-01, 1.1527e+00, 2.5125e-01, ..., 1.1188e+00,
1.5399e+00, 1.4623e+00],
[ 1.9030e-01, -1.2059e+00, -8.0539e-01, ..., 9.1676e-01,
8.0587e-01, 1.9538e+00],
[-7.4536e-01, -8.7438e-01, 1.0245e+00, ..., 5.0446e-01,
1.0293e-01, 1.3080e+00],
...,
[ 1.0774e+00, 5.5385e-01, 6.3178e-01, ..., 2.1573e+00,
1.5167e+00, 1.9051e+00],
[ 1.6444e+00, 2.6029e+00, 1.4762e+00, ..., 1.9131e+00,
1.5660e+00, 2.3246e+00],
[ 7.1071e-01, 7.3429e-01, 2.9477e+00, ..., 2.8789e-01,
1.0150e+00, 1.1902e+00]],
[[-2.2629e-01, -6.2251e-01, -1.0493e+00, ..., -5.5236e-01,
-8.3034e-01, -8.9944e-02],
[-1.5956e+00, -2.9492e+00, -2.3054e+00, ..., -1.1093e+00,
-9.3501e-01, -6.5230e-01],
[-2.1596e+00, -1.7515e+00, 6.1321e-01, ..., -1.6227e-02,
-1.9647e-01, -2.5338e-01],
...,
[-2.3020e+00, -2.4397e+00, 8.3287e-01, ..., -1.7724e-01,
-4.4095e-01, 1.0844e-01],
[-2.3411e+00, -1.9854e+00, -7.6731e-01, ..., -2.3058e+00,
-5.3367e-01, 3.4883e-01],
[-1.0810e-01, 3.0218e-01, -1.7615e+00, ..., -1.8321e+00,
-7.0555e-01, 7.1577e-02]]],
[[[ 6.5925e-01, 1.3922e+00, 8.0785e-01, ..., 7.4005e-01,
7.3985e-01, 1.2742e-01],
[-6.4835e-01, -1.1610e+00, -1.0668e+00, ..., 5.1336e-01,
1.8715e-01, -1.4863e-01],
[-1.3602e+00, -8.3422e-01, -2.0729e-02, ..., 3.4806e-01,
1.7007e-01, 7.5421e-01],
...,
[ 2.8362e-01, -8.9018e-01, -1.5389e+00, ..., 2.7982e-01,
8.2315e-01, 8.7315e-01],
[-5.4734e-01, -7.3466e-03, -1.0472e+00, ..., -6.5249e-01,
8.0197e-01, 1.7791e-01],
[ 2.6670e-01, 1.3986e+00, -1.7573e+00, ..., -1.0118e+00,
-6.3389e-02, 2.4928e-01]],
[[-2.7880e+00, -9.6576e-01, -2.2908e+00, ..., -1.7235e+00,
-2.5243e+00, -1.8293e+00],
[-3.1360e+00, -2.1923e+00, -3.3873e+00, ..., -7.5569e-01,
-2.3152e+00, -2.1318e+00],
[-2.5258e+00, -4.1142e-02, -4.4541e-01, ..., -1.1598e+00,
-2.1590e+00, -1.4446e+00],
...,
[-1.4417e+00, -1.1059e+00, 2.5675e+00, ..., -2.3850e+00,
-2.1815e+00, -9.3975e-02],
[-1.6979e+00, -2.2600e-02, 1.4836e+00, ..., -2.4451e+00,
-2.8916e+00, -4.1905e-01],
[-8.6103e-01, 5.9513e-02, 1.6716e+00, ..., -2.6459e+00,
-2.8659e+00, -8.3150e-01]],
[[ 1.9455e+00, 2.5821e+00, 2.0645e+00, ..., 1.6525e+00,
7.9148e-01, 2.3790e+00],
[ 8.4437e-01, 5.1239e-01, 1.3858e+00, ..., 6.0355e-01,
1.6237e-01, 2.5400e+00],
[ 3.9949e-01, -1.7686e-01, -5.1151e-01, ..., -4.0216e-01,
5.2791e-01, 1.9388e+00],
...,
[ 2.3179e+00, -2.1016e+00, -4.9875e+00, ..., -1.1149e+00,
5.6723e-02, 1.2599e+00],
[ 1.8324e+00, -3.0061e+00, -4.6756e+00, ..., -1.5790e+00,
-9.1556e-01, 7.3289e-01],
[ 2.4676e+00, -2.9038e+00, -2.2225e+00, ..., -7.7636e-01,
1.2409e-03, 1.1144e+00]],
...,
[[-2.2030e+00, -5.3264e+00, -3.5778e+00, ..., -1.9761e+00,
-1.4170e+00, -3.2999e+00],
[-2.2329e+00, -3.9426e+00, -1.0546e+00, ..., -1.8298e+00,
-1.7217e+00, -3.4004e+00],
[-1.6417e+00, -3.1833e+00, -2.0534e+00, ..., -1.3978e+00,
-1.7738e+00, -2.1661e+00],
...,
[-2.3097e+00, -7.1827e+00, -5.9611e+00, ..., -6.1647e-01,
-2.2868e+00, -2.9320e+00],
[-2.9914e+00, -6.3584e+00, -4.9787e+00, ..., -8.6877e-01,
-3.2884e+00, -4.5967e+00],
[-3.0571e+00, -6.3648e+00, -2.5378e+00, ..., -8.6829e-01,
-1.3135e+00, -2.8175e+00]],
[[ 8.1012e-01, 9.2896e-01, 1.8203e-01, ..., 9.3350e-01,
1.2347e+00, 1.3046e+00],
[ 6.5440e-01, -4.7693e-01, -7.1969e-01, ..., 1.0142e+00,
9.5221e-01, 2.0361e+00],
[ 9.4772e-01, 1.6561e+00, 1.3407e+00, ..., 6.8310e-01,
3.5074e-01, 1.2617e+00],
...,
[ 9.6215e-01, -1.8913e-01, 9.2028e-01, ..., -4.2932e-01,
1.1745e+00, 2.1519e+00],
[ 1.8351e+00, 1.0576e+00, -1.2317e+00, ..., -3.5481e-01,
4.4430e-01, 2.5344e+00],
[ 9.7354e-01, 2.2193e+00, 1.3521e+00, ..., -5.4601e-01,
1.6841e-01, 1.2972e+00]],
[[ 8.1273e-02, -2.7319e-01, -1.1214e+00, ..., -3.9983e-01,
-7.4140e-01, -2.6862e-02],
[-1.3303e+00, -1.7680e+00, -2.0547e+00, ..., -1.0500e+00,
-7.9562e-01, -7.3071e-01],
[-1.8394e+00, 6.8505e-01, 3.1127e-01, ..., -3.3257e-02,
-3.7988e-01, -3.7248e-01],
...,
[-2.3710e+00, -1.5060e+00, 2.1381e+00, ..., 7.6939e-01,
-3.2050e-01, 9.1677e-02],
[-2.0010e+00, -1.7232e+00, 7.1447e-01, ..., 7.1408e-01,
-5.5228e-01, -2.3728e-02],
[ 1.2217e-01, -1.0510e+00, -7.2417e-01, ..., 2.8976e-01,
-8.8364e-01, -3.4694e-01]]],
[[[ 1.0625e+00, 1.3534e+00, 1.2423e+00, ..., 1.2215e+00,
9.8665e-01, 4.3415e-01],
[-6.1170e-01, -2.4615e+00, -1.1303e+00, ..., -4.0322e-01,
-7.9283e-03, -1.3948e+00],
[-4.4334e-01, -1.2160e+00, -5.5945e-01, ..., -6.9962e-01,
-3.4983e-02, 6.1634e-01],
...,
[ 4.5185e-01, 3.4984e-01, 1.3170e+00, ..., 3.9902e-01,
9.9572e-01, 7.9452e-01],
[-3.9607e-02, -6.5903e-01, 7.4585e-01, ..., 9.2706e-01,
1.5612e+00, 1.5045e-01],
[ 8.6200e-01, -2.9439e-01, -2.3658e-01, ..., 3.6340e-01,
2.5476e-01, 7.0597e-01]],
[[-2.9219e+00, 2.0920e-02, -1.4792e+00, ..., -1.4256e+00,
-2.4180e+00, -6.5235e-01],
[-4.1916e+00, -2.6602e+00, -2.5117e+00, ..., -2.3058e+00,
-3.9785e+00, -1.5573e+00],
[-3.1800e+00, -2.4709e+00, -2.6505e+00, ..., -2.8981e+00,
-3.0696e+00, -1.3487e+00],
...,
[-8.2006e-01, -1.0777e-01, 7.7240e-01, ..., -6.0417e-01,
-5.0304e-02, 7.6393e-01],
[-2.5478e+00, -2.7871e-01, -2.7834e-01, ..., 2.9043e-02,
-4.8945e-02, 6.7536e-01],
[-1.4692e+00, -1.3071e+00, -1.0866e+00, ..., -4.9125e-01,
-6.1780e-01, -2.0910e-01]],
[[ 2.5283e+00, 2.9364e+00, 2.5503e+00, ..., 2.0188e+00,
5.5894e-01, 2.5869e+00],
[ 8.6724e-01, 6.2767e-01, 2.3914e+00, ..., 1.8333e+00,
4.8659e-01, 2.5997e+00],
[ 1.1505e+00, 1.4168e+00, 2.1309e+00, ..., 1.2365e+00,
1.6843e+00, 2.7990e+00],
...,
[ 1.0053e+00, 9.6402e-01, 4.8147e-01, ..., 9.5885e-01,
5.7632e-01, 1.9226e+00],
[ 1.0975e+00, 2.3733e+00, 2.3240e+00, ..., 9.7855e-01,
6.9844e-01, 1.7506e+00],
[ 2.3847e+00, 1.7900e+00, 2.5080e+00, ..., 1.9065e+00,
1.0388e+00, 1.7587e+00]],
...,
[[-2.7636e+00, -7.0406e+00, -5.8834e+00, ..., -4.2831e+00,
-4.4246e+00, -5.3926e+00],
[-3.4617e+00, -7.6340e+00, -5.4516e+00, ..., -4.7955e+00,
-5.1858e+00, -5.2010e+00],
[-2.2694e+00, -6.1364e+00, -4.2524e+00, ..., -3.5240e+00,
-3.9340e+00, -3.2745e+00],
...,
[-2.4634e+00, -4.7857e+00, -2.1590e+00, ..., -3.1428e+00,
-2.7998e+00, -3.0137e+00],
[-2.4470e+00, -5.3576e+00, -3.3856e+00, ..., -4.4025e+00,
-3.1771e+00, -4.9804e+00],
[-2.6142e+00, -5.7910e+00, -2.5479e+00, ..., -2.9177e+00,
-1.5505e+00, -3.2974e+00]],
[[ 3.3031e-01, 1.1502e+00, 3.8801e-01, ..., 9.6269e-01,
1.4797e+00, 1.7113e+00],
[ 1.3379e+00, 9.2483e-01, 2.4947e-01, ..., 7.3309e-01,
1.3019e+00, 2.7579e+00],
[ 1.3383e+00, 7.8834e-01, 4.2447e-01, ..., 1.3024e+00,
1.1150e+00, 1.9999e+00],
...,
[ 1.5534e+00, 4.0268e-01, 1.5824e+00, ..., 1.9280e+00,
1.9901e+00, 1.4751e+00],
[ 2.0254e+00, 9.5720e-01, 1.4513e+00, ..., 1.9764e+00,
1.7629e+00, 2.5477e+00],
[ 1.1922e+00, 9.2278e-01, 1.5058e+00, ..., 1.2687e+00,
1.5038e+00, 1.5338e+00]],
[[-6.6437e-01, 6.0338e-02, -2.4514e-01, ..., 4.6848e-01,
-7.0430e-02, 9.5154e-01],
[-1.5364e+00, -2.0942e+00, -1.7075e+00, ..., -8.9596e-01,
-7.4086e-01, -1.3879e-02],
[-2.3514e+00, -1.3817e+00, -4.1290e-01, ..., -3.1131e-01,
-4.0830e-01, -1.4454e-01],
...,
[-1.9594e+00, -1.1239e+00, -5.0286e-01, ..., -1.7195e-01,
-4.0456e-01, -4.3316e-01],
[-1.6663e+00, -1.2690e+00, -3.0269e-01, ..., -3.1504e-01,
-3.1889e-01, 4.4153e-01],
[-1.8439e-01, 4.4963e-01, -1.9819e-01, ..., 2.0798e-01,
2.7925e-02, 5.9478e-01]]]], device='cuda:0'),
tensor([[[[ 5.8866e-01, 8.4631e-01, 5.3664e-01, ..., 3.5628e-01,
4.6497e-01, 2.6820e-02],
[-2.2979e-01, -8.3594e-01, -4.5666e-01, ..., 1.6205e-01,
1.5027e-01, -2.6451e-01],
[-1.1883e-01, 5.0161e-02, 3.3394e-01, ..., 7.1002e-02,
1.6515e-01, 6.7877e-01],
...,
[-1.7185e-01, -8.2782e-01, -5.7884e-01, ..., 2.2146e-01,
7.9328e-01, 9.1020e-01],
[-5.4760e-01, -1.7636e+00, -8.7963e-01, ..., 6.5847e-02,
7.3955e-01, 5.4350e-02],
[ 4.3846e-01, -6.3269e-01, -1.0542e+00, ..., -2.7561e-01,
-5.5527e-01, 2.5209e-01]],
[[-2.3295e+00, -6.0510e-01, -1.8537e+00, ..., -1.5981e+00,
-2.2222e+00, -1.5014e+00],
[-2.3711e+00, -5.2649e-01, -1.5014e+00, ..., -1.2832e+00,
-2.3455e+00, -1.6805e+00],
[-2.3851e+00, -1.4917e+00, -2.2171e+00, ..., -1.7719e+00,
-2.2799e+00, -1.1948e+00],
...,
[-1.2889e+00, -1.0445e+00, -1.5242e+00, ..., -1.5775e+00,
-1.0020e+00, 1.8350e-01],
[-2.0671e+00, -4.4765e-01, -1.0852e+00, ..., -1.4648e+00,
-1.3542e+00, -1.6182e-01],
[-6.1896e-01, -1.3824e+00, -1.9305e+00, ..., -2.3410e+00,
-1.3654e+00, -4.8614e-01]],
[[ 1.8846e+00, 2.4869e+00, 1.6273e+00, ..., 1.5649e+00,
7.3214e-01, 2.1094e+00],
[-2.4755e-01, 3.4979e-01, 6.1945e-01, ..., 7.6703e-01,
2.7847e-01, 2.2525e+00],
[-4.8127e-02, 2.1326e-01, 6.0843e-01, ..., 7.3322e-02,
8.0656e-01, 2.0657e+00],
...,
[ 5.5522e-01, 8.6110e-01, 1.1342e+00, ..., 3.2383e-01,
-4.0676e-01, 1.3014e+00],
[ 4.2006e-01, 1.6407e+00, 1.2529e+00, ..., -7.3935e-02,
-2.3339e-01, 9.7215e-01],
[ 1.5522e+00, 9.2972e-01, 1.5432e+00, ..., 1.2946e+00,
6.1340e-01, 1.3701e+00]],
...,
[[-1.9172e+00, -4.7669e+00, -3.0838e+00, ..., -2.0372e+00,
-1.6090e+00, -3.0168e+00],
[-2.3995e+00, -5.1940e+00, -2.9149e+00, ..., -2.3449e+00,
-2.1377e+00, -2.9101e+00],
[-1.5496e+00, -5.3233e+00, -3.0743e+00, ..., -1.9779e+00,
-2.0853e+00, -1.9401e+00],
...,
[-1.7762e+00, -5.0122e+00, -2.9072e+00, ..., -2.0918e+00,
-2.6697e+00, -2.8681e+00],
[-2.2005e+00, -6.3596e+00, -4.0970e+00, ..., -3.3849e+00,
-3.4250e+00, -4.3505e+00],
[-2.5587e+00, -6.7903e+00, -3.3917e+00, ..., -2.2999e+00,
-2.0125e+00, -2.9352e+00]],
[[ 9.2160e-01, 1.5926e+00, 7.8948e-01, ..., 1.0981e+00,
1.5232e+00, 1.3871e+00],
[ 1.1777e+00, 1.0534e+00, 8.3407e-01, ..., 9.1971e-01,
7.9649e-01, 1.8687e+00],
[ 6.4981e-01, 1.3022e-01, 1.2793e-01, ..., 5.0475e-01,
8.0795e-02, 1.3272e+00],
...,
[ 1.3523e+00, -2.1688e-01, 4.2171e-01, ..., 1.3267e+00,
1.6269e+00, 1.7345e+00],
[ 2.0329e+00, 1.2726e+00, 1.1296e+00, ..., 1.0632e+00,
1.3802e+00, 2.2394e+00],
[ 1.0273e+00, 3.1876e-01, 3.3389e-01, ..., 3.8667e-02,
8.8056e-01, 1.1960e+00]],
[[-1.0828e-01, -6.1369e-01, -1.1613e+00, ..., -5.2706e-01,
-8.6747e-01, -9.7836e-02],
[-8.3975e-01, -2.4935e+00, -2.5992e+00, ..., -1.1864e+00,
-1.0765e+00, -6.6535e-01],
[-1.3947e+00, -1.5850e+00, -1.0877e+00, ..., -6.4544e-03,
-3.0968e-01, -2.4963e-01],
...,
[-2.3288e+00, -1.6175e+00, -7.4863e-01, ..., -4.4283e-01,
-2.9441e-01, 1.1603e-01],
[-2.3795e+00, -1.8152e+00, -8.3867e-01, ..., -7.5360e-01,
-6.5389e-01, 3.0587e-01],
[-1.6934e-01, -1.8415e-01, -4.0466e-01, ..., -5.7522e-01,
-6.0931e-01, 2.0449e-01]]],
[[[ 5.7674e-01, 8.4818e-01, 5.2851e-01, ..., 3.6625e-01,
4.4346e-01, 7.6055e-03],
[ 2.0854e-01, -5.2993e-01, -1.5321e+00, ..., 1.8158e-01,
1.4848e-01, -2.7943e-01],
[-2.1536e-01, -1.0768e+00, -3.0018e+00, ..., -1.0624e-01,
2.2428e-02, 6.3627e-01],
...,
[ 4.2590e-01, -2.7225e-01, -1.4992e-01, ..., 1.4101e+00,
3.6987e+00, 2.3376e+00],
[-1.7861e-02, -6.6135e-01, 1.2116e-01, ..., -4.9161e+00,
-4.8090e-01, -3.6005e-01],
[ 8.3941e-01, -1.3847e-01, -1.1802e-01, ..., -1.6899e-01,
-3.4948e-01, 1.1193e+00]],
[[-2.4290e+00, -6.5193e-01, -1.9200e+00, ..., -1.5980e+00,
-2.2778e+00, -1.5624e+00],
[-4.0975e+00, -4.5505e+00, -3.2457e+00, ..., -1.2770e+00,
-2.3833e+00, -1.7101e+00],
[-5.9185e+00, -1.0945e+01, -1.1971e+01, ..., -1.9988e+00,
-2.3448e+00, -1.1289e+00],
...,
[-6.3764e-01, -8.8606e-01, -1.5330e+00, ..., 6.3510e-01,
-1.2713e-01, -5.1302e-01],
[-2.4396e+00, -1.8257e-01, -2.1894e-01, ..., -8.3806e+00,
-6.3060e+00, -2.5465e+00],
[-1.4444e+00, -1.2761e+00, -9.8293e-01, ..., -2.5650e+00,
-1.8253e+00, -1.1740e+00]],
[[ 1.9642e+00, 2.5311e+00, 1.6358e+00, ..., 1.5757e+00,
7.4128e-01, 2.0998e+00],
[ 1.7008e-01, 3.3618e-01, 1.6447e+00, ..., 8.1747e-01,
3.4090e-01, 2.2678e+00],
[ 1.9462e+00, 3.0301e+00, 1.0574e+00, ..., -5.1065e-02,
7.1879e-01, 1.9659e+00],
...,
[ 6.6226e-01, 3.5810e-01, 1.4125e+00, ..., -5.9092e+00,
-4.2629e+00, -8.0616e-01],
[ 1.0250e+00, 2.1777e+00, 2.0229e+00, ..., -6.5134e+00,
-4.3352e+00, -8.8545e-01],
[ 2.3543e+00, 1.8734e+00, 2.4029e+00, ..., 2.8778e+00,
1.1922e+00, 1.7519e+00]],
...,
[[-2.0055e+00, -4.9074e+00, -3.2099e+00, ..., -2.0306e+00,
-1.6436e+00, -3.0350e+00],
[-2.7266e+00, -6.7195e+00, -4.9042e+00, ..., -2.3577e+00,
-2.1730e+00, -3.0211e+00],
[-8.9268e-01, -4.1169e+00, -4.1800e+00, ..., -2.3214e+00,
-2.4261e+00, -2.1751e+00],
...,
[-2.4496e+00, -4.1534e+00, -2.8638e+00, ..., -1.1773e+01,
-1.2905e+01, -7.6734e+00],
[-2.3720e+00, -5.2206e+00, -3.9345e+00, ..., -9.3861e+00,
-9.7364e+00, -6.6233e+00],
[-2.5396e+00, -5.7327e+00, -2.8858e+00, ..., -6.4172e+00,
-5.9474e+00, -6.5714e+00]],
[[ 9.1736e-01, 1.6299e+00, 7.9433e-01, ..., 1.1013e+00,
1.4981e+00, 1.3998e+00],
[ 1.2217e-01, -9.1843e-01, 2.0377e-01, ..., 9.4742e-01,
8.1095e-01, 1.9240e+00],
[-2.1424e+00, -8.8862e+00, -1.0332e+01, ..., 4.1372e-01,
1.0399e-01, 1.3767e+00],
...,
[ 1.4705e+00, 7.5014e-02, 4.8846e-01, ..., -3.1121e+00,
-1.4718e+00, 1.5038e+00],
[ 2.0165e+00, 1.1556e+00, 1.7931e+00, ..., -2.7518e+00,
-2.8099e+00, 1.8902e+00],
[ 1.1754e+00, 8.2994e-01, 1.3224e+00, ..., 2.8034e+00,
2.4489e+00, 1.7786e+00]],
[[-9.9195e-02, -6.4384e-01, -1.1798e+00, ..., -5.2233e-01,
-8.8996e-01, -1.2153e-01],
[-1.0763e+00, -2.0573e+00, -2.0270e+00, ..., -1.2051e+00,
-1.0515e+00, -6.6115e-01],
[-2.3970e+00, -2.7630e+00, -6.2851e-01, ..., -2.8659e-01,
-4.3051e-01, -2.6301e-01],
...,
[-1.7197e+00, -1.5525e+00, -7.8477e-01, ..., -2.2566e+00,
-1.7661e-01, 9.0405e-01],
[-1.6988e+00, -1.1944e+00, -6.1331e-01, ..., -6.0475e+00,
-5.2177e+00, -1.8282e+00],
[-1.5034e-01, 4.9059e-01, -2.1095e-01, ..., -8.6867e-01,
-5.6023e-01, 1.0949e+00]]],
[[[ 8.7377e-01, 1.4916e+00, 1.2004e+00, ..., 1.0403e+00,
9.0172e-01, 8.4544e-01],
[-9.3673e-01, -1.9805e+00, -1.1727e+00, ..., -5.9193e-01,
-1.1107e+00, -1.7475e+00],
[-1.6435e-01, -7.7458e-01, 1.3043e-01, ..., 1.8080e+00,
1.3438e+00, 8.1204e-01],
...,
[-3.3324e-02, -1.1332e+00, -1.1839e+00, ..., -3.1621e-01,
-9.4835e-02, 2.2278e-01],
[-6.0993e-01, -1.9638e+00, -1.3655e+00, ..., -6.4760e-01,
-3.6816e-01, -1.6111e-01],
[-2.5652e-01, -1.6437e+00, -1.6870e+00, ..., -1.1607e+00,
-1.5710e+00, -3.2885e-01]],
[[-3.7168e+00, -7.3070e-01, -1.5843e+00, ..., -1.4718e+00,
-2.5920e+00, -1.2242e+00],
[-3.9185e+00, -2.8268e+00, -1.8529e+00, ..., -2.6806e+00,
-3.4594e+00, -9.5605e-01],
[-2.3926e+00, -1.7209e+00, -1.5444e+00, ..., -2.3707e+00,
-2.8427e+00, -9.9263e-01],
...,
[-1.5910e+00, -1.6374e+00, -2.8285e+00, ..., -1.8605e+00,
-2.0401e+00, -7.7427e-01],
[-3.1272e+00, -1.6042e+00, -2.8476e+00, ..., -2.3306e+00,
-2.2013e+00, -1.2103e+00],
[-2.2995e+00, -1.9666e+00, -2.6960e+00, ..., -1.9174e+00,
-1.8928e+00, -1.0134e+00]],
[[ 2.9095e+00, 4.2360e+00, 3.4191e+00, ..., 3.1414e+00,
1.0207e+00, 2.9864e+00],
[ 2.4530e-01, 1.0986e+00, 2.2887e+00, ..., 1.2975e+00,
-1.4928e-01, 2.8874e+00],
[ 9.9057e-01, 1.4669e+00, 1.5703e+00, ..., -8.2514e-01,
2.6316e-01, 2.3006e+00],
...,
[ 9.6205e-01, 1.0300e+00, 1.5134e+00, ..., 4.3347e-02,
9.1431e-01, 2.2107e+00],
[ 7.4265e-01, 1.8440e+00, 1.2544e+00, ..., 4.5308e-02,
9.1869e-01, 1.7535e+00],
[ 1.7159e+00, 1.6363e+00, 1.8472e+00, ..., 1.2604e+00,
1.6897e+00, 1.6952e+00]],
...,
[[-1.8821e+00, -7.2585e+00, -5.4154e+00, ..., -3.9506e+00,
-2.9562e+00, -5.5260e+00],
[-2.4559e+00, -7.0962e+00, -4.2399e+00, ..., -2.4946e+00,
-1.9897e+00, -4.3826e+00],
[-9.3006e-01, -4.5046e+00, -3.6291e+00, ..., -3.8835e+00,
-4.1933e+00, -3.1752e+00],
...,
[-2.5817e+00, -5.2679e+00, -4.4787e+00, ..., -6.0597e+00,
-4.5101e+00, -4.5688e+00],
[-2.7316e+00, -5.6091e+00, -5.4276e+00, ..., -6.2722e+00,
-5.6221e+00, -6.1295e+00],
[-1.4744e+00, -4.2198e+00, -2.9056e+00, ..., -3.3055e+00,
-2.7053e+00, -3.1655e+00]],
[[-1.8238e-01, 3.4563e-01, 5.7235e-01, ..., 8.8368e-01,
9.4044e-01, 1.4780e+00],
[ 1.6631e+00, 1.0843e+00, 1.7912e+00, ..., 1.4305e+00,
2.6390e+00, 3.1324e+00],
[ 1.1990e+00, 2.9384e-01, 8.5316e-01, ..., -2.0440e-01,
1.4520e+00, 2.6082e+00],
...,
[ 1.3338e+00, 2.5160e-01, -5.7759e-01, ..., 1.3424e+00,
1.1131e+00, 1.5366e+00],
[ 1.8960e+00, 1.7102e+00, 9.8041e-01, ..., 1.5057e+00,
1.3185e+00, 1.8970e+00],
[ 4.2801e-01, 7.6109e-01, 7.1746e-01, ..., 5.1909e-01,
6.3341e-01, 1.5902e+00]],
[[-5.6388e-01, 3.0461e-01, 1.5096e-01, ..., 2.1418e-01,
2.4499e-01, 1.3054e+00],
[-1.3252e+00, -1.7985e+00, -1.6738e+00, ..., -6.4442e-01,
-8.7189e-01, 2.1659e-01],
[-2.2403e+00, -8.9450e-01, 3.4409e-01, ..., 2.2117e+00,
1.5376e-01, 8.0677e-01],
...,
[-1.9713e+00, -1.7735e+00, -1.5334e+00, ..., -7.2874e-01,
-1.1259e+00, -8.9694e-01],
[-1.6395e+00, -2.0474e+00, -1.7237e+00, ..., -1.2061e+00,
-1.2997e+00, -6.5859e-01],
[-7.9875e-01, -8.0749e-01, -1.1528e+00, ..., -6.2701e-01,
-9.2946e-01, -1.1354e-01]]],
[[[ 7.9993e-01, 1.0659e+00, 5.8296e-01, ..., 4.7798e-01,
8.5572e-01, 3.6233e-01],
[-1.3294e+00, -1.2666e+00, -9.9581e-02, ..., -4.3740e-01,
-2.2423e-01, -3.4022e-01],
[-2.1739e-01, -5.6738e-01, -9.5937e-02, ..., 3.6954e-01,
5.0222e-01, 8.4835e-01],
...,
[ 1.3621e+00, -7.9090e-01, -3.0431e-01, ..., 3.7146e-01,
-1.3122e-01, -8.2780e-01],
[-1.5447e-01, -9.5833e-01, -2.1244e-01, ..., -7.4024e-01,
-5.5132e-01, -1.7297e-01],
[ 4.9950e-01, -2.3492e-01, -7.8700e-01, ..., 3.5195e-01,
-5.3857e-01, 2.6356e-01]],
[[-4.2229e+00, -2.0106e+00, -3.0905e+00, ..., -1.5781e+00,
-1.9851e+00, -1.6721e+00],
[-3.5723e+00, -1.8088e+00, -9.9905e-01, ..., -2.0327e+00,
-2.4319e+00, -2.1131e+00],
[-2.9693e+00, -2.2130e+00, -2.0967e+00, ..., -3.8825e-01,
-1.6037e+00, -1.4952e+00],
...,
[-1.0417e+00, -1.0273e+00, -1.3711e+00, ..., -3.2136e+00,
-6.0522e+00, -3.8457e+00],
[-1.1363e+00, -4.5748e-01, -1.5162e-01, ..., -2.8953e+00,
-2.4978e+00, -1.9684e+00],
[ 1.9650e-01, -6.6354e-01, -7.6979e-01, ..., -9.3523e-01,
-7.0721e-01, -4.3032e-01]],
[[ 2.8415e+00, 3.4774e+00, 3.3003e+00, ..., 1.8406e+00,
9.5265e-01, 2.5099e+00],
[ 7.9242e-01, 1.1452e+00, 2.7288e+00, ..., 1.6651e+00,
1.0364e+00, 2.5587e+00],
[ 1.1664e+00, 8.0560e-01, 2.4593e+00, ..., -1.1933e+00,
3.4779e-01, 1.6584e+00],
...,
[ 6.2649e-01, 4.0784e-01, 1.1595e+00, ..., -1.6048e+00,
-3.2789e+00, -1.3476e+00],
[ 4.2000e-01, 1.7672e+00, 1.9246e+00, ..., -8.3619e-01,
8.7843e-01, 1.3001e+00],
[ 1.6369e+00, 1.0992e+00, 1.8556e+00, ..., 1.1846e+00,
1.7497e+00, 1.1225e+00]],
...,
[[-3.1386e+00, -8.5630e+00, -5.5242e+00, ..., -5.7429e+00,
-3.5491e+00, -3.9994e+00],
[-3.5171e+00, -9.6528e+00, -6.5455e+00, ..., -5.2200e+00,
-3.6678e+00, -3.0601e+00],
[-2.7515e+00, -7.6774e+00, -4.9047e+00, ..., -4.6748e+00,
-3.0788e+00, -1.5510e+00],
...,
[-2.9617e+00, -5.5613e+00, -3.3373e+00, ..., -2.3741e+00,
-3.2833e+00, -4.3847e+00],
[-3.0209e+00, -7.3558e+00, -4.2161e+00, ..., -4.7296e+00,
-4.0974e+00, -3.9581e+00],
[-2.4342e+00, -6.7181e+00, -3.3351e+00, ..., -3.8112e+00,
-3.0861e+00, -3.7727e+00]],
[[-7.9756e-01, -9.8734e-02, 2.2260e-01, ..., 2.0646e+00,
1.8375e+00, 1.3127e+00],
[ 1.5103e+00, 1.0803e+00, 1.2771e+00, ..., 1.8637e+00,
1.4659e+00, 1.9209e+00],
[ 1.0539e+00, 1.5837e-02, 3.3894e-01, ..., 9.2035e-01,
2.9042e-01, 1.0636e+00],
...,
[ 1.0699e+00, 4.3250e-03, 3.2203e-01, ..., 3.5516e-01,
-2.8958e-01, -4.9430e-01],
[ 1.9378e+00, 1.3140e+00, 1.4572e+00, ..., 4.4200e-01,
1.1903e-01, 1.7294e+00],
[ 1.1524e+00, 8.7633e-01, 1.1410e+00, ..., 1.1700e+00,
1.0216e+00, 1.7580e+00]],
[[-7.3932e-01, -7.6783e-01, -6.2928e-01, ..., -4.3616e-01,
-6.7457e-01, -4.8946e-02],
[-1.4850e+00, -1.6408e+00, -7.2948e-01, ..., -1.3148e+00,
-1.4728e+00, -1.0417e+00],
[-2.3367e+00, -5.5415e-01, 4.7546e-01, ..., 7.8586e-01,
-9.5081e-01, -4.5547e-01],
...,
[-1.7630e+00, -1.5829e+00, -1.3943e+00, ..., 4.9259e-02,
-2.2302e+00, -4.4409e+00],
[-1.9060e+00, -2.0596e+00, -6.7114e-01, ..., -1.8699e+00,
-2.7870e+00, -1.0656e+00],
[ 1.1509e-01, 7.8040e-02, -2.0348e-01, ..., -6.1879e-01,
-5.5148e-01, 3.7443e-01]]],
[[[ 7.2974e-01, 1.0781e+00, 7.3634e-01, ..., 4.3459e-01,
3.6950e-01, 2.1882e-01],
[-3.0369e-01, -1.3995e+00, -5.2733e-01, ..., 4.1444e-01,
2.3200e-01, -1.1764e+00],
[ 1.5544e-01, -8.2002e-01, -3.8513e-01, ..., -4.2111e-01,
6.1796e-01, 3.2232e-01],
...,
[ 6.7677e-02, -1.0911e+00, 8.7250e-01, ..., -2.7011e-01,
1.1431e+00, 8.4665e-01],
[-4.6652e-01, -1.9678e+00, -7.1256e-01, ..., 3.9185e-02,
1.2880e+00, -8.7043e-02],
[ 3.9728e-01, -5.4030e-01, -6.9717e-01, ..., -2.8746e-01,
4.0855e-01, 6.2493e-01]],
[[-2.9879e+00, -1.5201e+00, -3.0938e+00, ..., -1.8044e+00,
-2.4950e+00, -1.3455e+00],
[-4.2882e+00, -1.8499e+00, -2.3656e+00, ..., -6.9751e-01,
-1.3364e+00, -7.2279e-01],
[-4.3339e+00, -2.6309e+00, -3.1126e+00, ..., -2.0029e+00,
-2.0340e+00, -1.1642e+00],
...,
[-2.8755e+00, -1.5669e+00, -1.7834e+00, ..., -9.5592e-01,
6.5844e-01, 5.0203e-01],
[-3.7015e+00, -1.3530e+00, -2.1349e+00, ..., -5.9516e-01,
-3.9369e-01, 1.4540e-01],
[-1.2304e+00, -8.2178e-01, -1.6998e+00, ..., -4.7738e-01,
-7.6395e-01, -6.3382e-01]],
[[ 1.6659e+00, 2.7652e+00, 2.4100e+00, ..., 2.8425e+00,
1.3478e+00, 3.0316e+00],
[ 1.5806e-01, 1.1794e+00, 1.6009e+00, ..., 1.9524e+00,
9.2281e-02, 2.7620e+00],
[ 7.1377e-01, 1.4441e+00, 8.9771e-01, ..., 1.3072e+00,
7.4642e-01, 1.3470e+00],
...,
[ 1.0439e+00, 1.8748e+00, 6.9068e-01, ..., 1.3280e+00,
4.8013e-01, 1.8627e+00],
[ 6.9845e-01, 1.8994e+00, 1.1790e+00, ..., 1.2535e+00,
5.1821e-01, 1.4947e+00],
[ 1.5433e+00, 1.5682e+00, 1.6166e+00, ..., 1.4073e+00,
1.0842e+00, 1.9832e+00]],
...,
[[-1.3689e+00, -4.5666e+00, -3.8992e+00, ..., -3.3318e+00,
-2.5899e+00, -4.2470e+00],
[-1.7571e+00, -3.8398e+00, -4.2894e+00, ..., -5.6201e+00,
-3.8151e+00, -4.4824e+00],
[-9.8136e-01, -3.2267e+00, -3.8149e+00, ..., -2.5689e+00,
-1.7369e+00, -9.4904e-01],
...,
[-1.6124e+00, -4.3780e+00, -4.3180e+00, ..., -3.4102e+00,
-4.0306e+00, -2.3661e+00],
[-1.5222e+00, -4.6613e+00, -5.0374e+00, ..., -5.3352e+00,
-4.0213e+00, -3.4501e+00],
[-1.6551e+00, -4.9371e+00, -4.1711e+00, ..., -3.4735e+00,
-2.0937e+00, -2.5092e+00]],
[[ 8.0445e-01, 5.1918e-01, -1.1211e-01, ..., 8.0186e-01,
9.5291e-01, 1.1924e+00],
[ 1.1908e+00, 9.5594e-01, 5.5048e-01, ..., 1.6058e+00,
2.5532e+00, 2.6072e+00],
[ 5.5682e-01, 5.3013e-01, 1.0679e-01, ..., 3.4807e-01,
5.1713e-01, 1.3479e+00],
...,
[ 8.0596e-01, -2.9740e-01, -1.7305e-01, ..., 1.5059e+00,
2.1736e+00, 1.7949e+00],
[ 1.5334e+00, 1.4809e+00, 1.4074e+00, ..., 1.2826e+00,
1.4605e+00, 2.4327e+00],
[ 6.7573e-01, 9.0896e-01, 1.0147e+00, ..., 1.0504e+00,
1.2961e+00, 1.3159e+00]],
[[-2.7046e-01, -1.6558e-01, -9.7769e-01, ..., -1.4153e-01,
-1.7380e-01, 8.7188e-01],
[-1.3739e+00, -1.4288e+00, -1.9661e+00, ..., 2.7554e-01,
1.5082e-01, 3.9614e-01],
[-1.6362e+00, -8.7534e-01, -1.0061e+00, ..., 6.3564e-02,
-8.0082e-02, 2.1720e-01],
...,
[-1.7261e+00, -1.4490e+00, -7.2810e-01, ..., -2.8215e-01,
-2.0431e-01, -8.5198e-02],
[-1.4083e+00, -1.6354e+00, -1.4591e+00, ..., -1.0009e-01,
-1.4920e-01, 3.1910e-01],
[-1.5329e-01, 3.7720e-02, -5.1293e-01, ..., 1.9685e-01,
1.9326e-01, 2.8359e-01]]]], device='cuda:0'),
tensor([[[[ 5.5905e-01, 7.9835e-01, 5.5584e-01, ..., 3.4630e-01,
4.4493e-01, 2.8278e-02],
[-2.7533e-01, -9.5871e-01, -6.0311e-01, ..., 1.9092e-01,
1.7526e-01, -1.6844e-01],
[ 1.0896e-01, -1.1008e-01, 1.6235e-01, ..., -1.2678e-01,
-6.5867e-02, 6.1782e-01],
...,
[ 3.8030e-01, -2.7437e-01, 1.0425e-01, ..., 2.6740e-01,
8.5582e-01, 6.5789e-01],
[-4.9748e-02, -7.3318e-01, -1.7258e-01, ..., 8.0462e-01,
1.5424e+00, 8.0043e-02],
[ 8.0865e-01, -2.1056e-01, -1.1525e-01, ..., 3.8962e-01,
3.0750e-01, 7.4213e-01]],
[[-2.5017e+00, -7.5215e-01, -1.9366e+00, ..., -1.6304e+00,
-2.3123e+00, -1.5707e+00],
[-2.5727e+00, -6.8708e-01, -1.3957e+00, ..., -1.0807e+00,
-2.2367e+00, -1.7175e+00],
[-3.5107e+00, -1.6657e+00, -9.5870e-01, ..., -1.8784e+00,
-2.2892e+00, -1.0888e+00],
...,
[-7.0597e-01, -8.3762e-01, -1.4079e+00, ..., -6.1597e-01,
-1.1136e-01, 7.5143e-01],
[-2.5975e+00, -4.0034e-01, -7.5122e-01, ..., 1.4275e-01,
-1.7549e-02, 6.9824e-01],
[-1.5548e+00, -1.1951e+00, -9.3487e-01, ..., -4.7517e-01,
-6.0906e-01, -1.7612e-01]],
[[ 1.9945e+00, 2.5645e+00, 1.5808e+00, ..., 1.4850e+00,
6.9780e-01, 2.1462e+00],
[-1.6081e-01, 3.5135e-01, 5.6369e-01, ..., 7.1098e-01,
2.3315e-01, 2.2295e+00],
[ 1.2214e-01, 3.8529e-01, 1.0808e+00, ..., -1.2564e-01,
6.1935e-01, 1.9243e+00],
...,
[ 6.5962e-01, 6.7543e-01, 2.4210e+00, ..., 6.1756e-01,
1.7594e-01, 1.6202e+00],
[ 1.1314e+00, 2.1775e+00, 1.8682e+00, ..., 8.1514e-01,
7.3106e-01, 1.7184e+00],
[ 2.3993e+00, 1.8311e+00, 2.3223e+00, ..., 1.9613e+00,
1.0531e+00, 1.7368e+00]],
...,
[[-2.0091e+00, -4.8988e+00, -3.2266e+00, ..., -2.0355e+00,
-1.4584e+00, -2.8710e+00],
[-2.4676e+00, -5.3354e+00, -2.9974e+00, ..., -2.4588e+00,
-2.1116e+00, -2.9851e+00],
[-2.0565e+00, -6.2842e+00, -3.1363e+00, ..., -2.2869e+00,
-2.2665e+00, -2.1497e+00],
...,
[-2.6082e+00, -4.5257e+00, -2.8601e+00, ..., -3.2221e+00,
-2.1376e+00, -2.7311e+00],
[-2.5300e+00, -5.5348e+00, -3.9355e+00, ..., -5.7062e+00,
-3.3067e+00, -4.9088e+00],
[-2.6749e+00, -6.0171e+00, -2.8685e+00, ..., -2.9429e+00,
-1.6556e+00, -3.4052e+00]],
[[ 8.6463e-01, 1.5927e+00, 7.6711e-01, ..., 9.9912e-01,
1.4089e+00, 1.3757e+00],
[ 1.1767e+00, 1.0516e+00, 9.4694e-01, ..., 1.0358e+00,
8.5598e-01, 1.8470e+00],
[ 5.1761e-01, 2.5814e-02, 3.9866e-01, ..., 3.8145e-01,
1.3386e-01, 1.3675e+00],
...,
[ 1.5081e+00, 2.5367e-02, 4.7331e-01, ..., 1.8351e+00,
1.8991e+00, 1.4700e+00],
[ 2.0143e+00, 1.1758e+00, 1.8435e+00, ..., 2.0020e+00,
1.6566e+00, 2.5534e+00],
[ 1.2069e+00, 9.1340e-01, 1.3484e+00, ..., 1.3192e+00,
1.4675e+00, 1.5169e+00]],
[[-1.3149e-01, -7.0589e-01, -1.1564e+00, ..., -5.9501e-01,
-8.3802e-01, -5.7401e-02],
[-8.8610e-01, -2.6226e+00, -2.5352e+00, ..., -1.1584e+00,
-9.6079e-01, -5.5014e-01],
[-1.6298e+00, -1.3483e+00, -8.6395e-01, ..., -3.6142e-01,
-5.5992e-01, -2.5692e-01],
...,
[-1.6265e+00, -1.8129e+00, -2.9376e-01, ..., -4.2595e-01,
-3.2766e-01, -4.7437e-01],
[-1.6997e+00, -1.3658e+00, -6.7970e-01, ..., 1.4175e-01,
-3.2764e-01, 3.9014e-01],
[-2.7254e-01, 4.5514e-01, -2.0913e-01, ..., 2.1656e-01,
3.5633e-02, 6.3892e-01]]],
[[[ 8.3526e-01, 1.5205e+00, 1.2097e+00, ..., -7.1243e-01,
-1.0383e+00, 2.0479e-01],
[-9.8610e-01, -1.9725e+00, -1.1353e+00, ..., 1.8694e+00,
-3.6212e-01, -2.3500e+00],
[-1.1248e-01, -6.8967e-01, 3.8057e-02, ..., 4.4241e-02,
1.7524e+00, 1.1149e+00],
...,
[ 5.2885e-01, -4.1011e-01, 1.0948e+00, ..., 3.1927e-01,
2.4670e-01, 6.8876e-01],
[ 2.7638e-01, -5.5431e-01, 9.4761e-01, ..., 1.3226e+00,
8.4465e-01, -4.2014e-02],
[ 5.5660e-01, 6.6209e-01, 3.3897e-01, ..., 1.3253e+00,
1.3478e-01, 6.4193e-01]],
[[-3.7732e+00, -7.9182e-01, -1.6304e+00, ..., -2.5832e+00,
-2.8881e+00, -1.0675e+00],
[-3.9916e+00, -2.7775e+00, -1.9377e+00, ..., -4.5906e-01,
-4.4724e+00, -9.5562e-01],
[-2.3787e+00, -1.5766e+00, -1.5963e+00, ..., -2.6257e+00,
-4.9631e+00, -7.1673e-01],
...,
[-1.0437e+00, -2.0973e-01, 2.6616e-01, ..., -2.3762e+00,
-2.0130e+00, -1.3194e-01],
[-1.3270e+00, 8.3531e-01, 2.0844e+00, ..., -3.0933e+00,
-1.2059e+00, -7.5450e-02],
[ 4.0234e-01, -2.9878e-01, -1.2809e-01, ..., -1.4480e+00,
-5.3438e-02, 4.0469e-01]],
[[ 2.8005e+00, 4.1699e+00, 3.3818e+00, ..., 3.1752e+00,
3.1023e+00, 3.1391e+00],
[ 8.1612e-02, 1.0269e+00, 2.2978e+00, ..., -7.9068e-01,
-1.1857e+00, 2.9356e+00],
[ 8.6901e-01, 1.5352e+00, 1.7343e+00, ..., -2.8639e+00,
-9.0772e-01, 1.8259e+00],
...,
[ 1.1561e+00, 1.0078e+00, -2.1547e+00, ..., -7.7506e-01,
-1.9397e-01, 1.2956e+00],
[ 5.6436e-01, 1.5147e+00, -1.9750e-01, ..., -4.6357e-01,
-5.3784e-01, 9.9254e-01],
[ 1.7287e+00, 1.5973e+00, 7.5703e-01, ..., 3.3236e-01,
2.2481e-01, 1.1039e+00]],
...,
[[-1.8138e+00, -7.1429e+00, -5.4317e+00, ..., -3.3634e+00,
-2.0387e+00, -4.9728e+00],
[-2.4003e+00, -6.8108e+00, -4.1713e+00, ..., -9.2428e+00,
-7.0782e+00, -4.6367e+00],
[-1.0464e+00, -4.5085e+00, -3.5883e+00, ..., -2.7389e+00,
-7.7047e+00, -3.6645e+00],
...,
[-2.3826e+00, -5.4685e+00, -1.8338e+00, ..., -3.9966e+00,
-4.7006e+00, -3.5106e+00],
[-2.4613e+00, -6.7267e+00, -2.8478e+00, ..., -8.5161e+00,
-5.0337e+00, -4.9164e+00],
[-2.4482e+00, -6.7427e+00, -3.3055e+00, ..., -7.2867e+00,
-3.6554e+00, -4.1777e+00]],
[[-1.7961e-01, 2.7357e-01, 5.4265e-01, ..., 8.3752e-01,
1.7158e+00, 2.0334e+00],
[ 1.6217e+00, 1.1193e+00, 1.8516e+00, ..., 4.2425e-03,
1.9234e+00, 3.6504e+00],
[ 1.1295e+00, 2.7158e-01, 8.2840e-01, ..., -1.4743e+00,
-7.0933e-01, 2.5780e+00],
...,
[ 1.0410e+00, 2.5199e-01, 1.0974e+00, ..., -7.2545e-01,
1.0270e+00, 1.4156e+00],
[ 1.5364e+00, 1.0199e+00, 2.1761e+00, ..., -1.0121e+00,
1.1684e+00, 2.1653e+00],
[ 9.9895e-01, 3.6364e-01, 1.2027e+00, ..., 2.6864e-01,
1.5922e+00, 1.4008e+00]],
[[-5.5389e-01, 3.2645e-01, 1.3637e-01, ..., -5.3229e-01,
-8.4902e-01, 1.5000e+00],
[-1.3523e+00, -1.6972e+00, -1.6295e+00, ..., 3.0832e+00,
-1.8923e-01, 3.9491e-01],
[-2.0966e+00, -6.8187e-01, 2.9142e-01, ..., 1.6274e+00,
3.7871e-02, 2.2715e-01],
...,
[-1.4358e+00, -1.8137e+00, -2.2074e-02, ..., -9.7239e-01,
-1.6867e+00, -8.1407e-01],
[-1.2239e+00, -1.6186e+00, 1.6385e-01, ..., -1.6428e+00,
-2.2032e+00, -2.7716e-06],
[ 3.3916e-01, 3.8640e-01, -2.2349e-01, ..., -9.2892e-01,
-8.4964e-01, 8.0007e-01]]],
[[[ 5.9543e-01, 8.3998e-01, 5.6256e-01, ..., 2.3550e-01,
3.8463e-01, 2.4480e-02],
[-2.0580e-01, -9.1931e-01, -4.8380e-01, ..., 1.4394e-01,
8.0491e-02, -2.8334e-01],
[-1.6915e-01, -1.0956e-01, 2.0105e-01, ..., -1.9211e-01,
-3.1748e-02, 6.3320e-01],
...,
[ 2.8422e-01, 9.1840e-01, 1.0430e+00, ..., 8.9337e-01,
1.1167e+00, 2.8905e-01],
[-1.0128e-02, 2.0084e-01, 1.3905e+00, ..., 1.0507e-01,
1.3144e+00, 7.9267e-02],
[ 7.8273e-01, -2.3635e-01, 1.3158e-01, ..., 7.4748e-01,
1.4296e-01, 8.5061e-01]],
[[-2.3660e+00, -5.8378e-01, -1.9273e+00, ..., -1.5904e+00,
-2.2446e+00, -1.5489e+00],
[-2.4757e+00, -7.0255e-01, -1.8470e+00, ..., -1.3127e+00,
-2.3387e+00, -1.7141e+00],
[-2.3959e+00, -1.5144e+00, -2.4876e+00, ..., -2.7317e+00,
-2.3930e+00, -1.1234e+00],
...,
[-6.3099e-01, -2.1430e-01, 1.9668e+00, ..., -9.6043e-01,
-1.0442e+00, 6.4902e-01],
[-2.0716e+00, -6.0509e-01, -4.3295e-02, ..., -7.4614e-01,
-1.0688e+00, 7.6405e-01],
[-1.2145e+00, -1.2639e+00, -9.2167e-01, ..., -1.5332e+00,
-1.0346e+00, -4.4762e-02]],
[[ 1.9476e+00, 2.5146e+00, 1.6534e+00, ..., 1.6429e+00,
7.8151e-01, 2.1279e+00],
[-1.9261e-01, 3.5451e-01, 6.0726e-01, ..., 6.6024e-01,
3.5548e-01, 2.2495e+00],
[ 5.1804e-02, 2.9969e-01, 7.2307e-01, ..., -7.2215e-01,
7.9168e-01, 1.9891e+00],
...,
[ 7.6617e-01, 3.4286e-01, -1.0701e+00, ..., -1.7308e+00,
-3.1202e-01, 1.5062e+00],
[ 6.8852e-01, 2.3660e+00, 1.4099e+00, ..., 1.2469e-01,
1.6743e+00, 1.7648e+00],
[ 1.9563e+00, 2.0638e+00, 2.5743e+00, ..., 1.3410e+00,
1.2832e+00, 1.5654e+00]],
...,
[[-1.9572e+00, -4.8047e+00, -3.1622e+00, ..., -1.7615e+00,
-1.5855e+00, -3.1109e+00],
[-2.4403e+00, -5.2238e+00, -2.9391e+00, ..., -2.8094e+00,
-2.2793e+00, -3.0715e+00],
[-1.5283e+00, -5.3591e+00, -3.3205e+00, ..., -3.9909e+00,
-2.5844e+00, -2.2780e+00],
...,
[-1.9483e+00, -3.5437e+00, -1.2452e+00, ..., -2.7090e+00,
-5.1533e+00, -2.9367e+00],
[-2.0127e+00, -4.5462e+00, -3.4016e+00, ..., -1.6206e+00,
-3.4563e+00, -5.5265e+00],
[-2.0668e+00, -4.8593e+00, -2.2647e+00, ..., -1.3854e+00,
-1.4649e+00, -4.0320e+00]],
[[ 9.2269e-01, 1.6358e+00, 7.8366e-01, ..., 1.3506e+00,
1.5512e+00, 1.3947e+00],
[ 1.1539e+00, 9.9514e-01, 6.6932e-01, ..., 1.6384e+00,
8.7859e-01, 1.9035e+00],
[ 6.3013e-01, -6.4929e-02, -4.1442e-01, ..., 1.2368e+00,
1.9987e-01, 1.3999e+00],
...,
[ 1.2891e+00, 5.7551e-01, 2.2815e+00, ..., 6.2654e-01,
1.5165e+00, 1.9136e+00],
[ 1.8928e+00, 7.2696e-01, 1.5358e+00, ..., 1.3809e+00,
1.4347e+00, 2.9748e+00],
[ 1.3260e+00, 8.1630e-01, 1.3351e+00, ..., 7.8236e-01,
1.6804e+00, 1.6309e+00]],
[[-8.9532e-02, -5.9841e-01, -1.1557e+00, ..., -5.0028e-01,
-8.5688e-01, -1.1999e-01],
[-8.5390e-01, -2.5499e+00, -2.8440e+00, ..., -8.9592e-01,
-9.9813e-01, -7.1439e-01],
[-1.4681e+00, -1.9131e+00, -1.5372e+00, ..., -4.3600e-01,
-5.1751e-01, -2.3896e-01],
...,
[-1.4772e+00, -1.1296e+00, -4.2737e-01, ..., -7.1478e-01,
-1.6371e+00, -8.2051e-01],
[-1.4695e+00, -6.6227e-01, -3.7290e-01, ..., -3.3897e-01,
-8.6160e-01, 2.5284e-01],
[ 9.7343e-02, 4.5953e-01, -3.8033e-01, ..., -3.7191e-01,
-3.9441e-01, 5.5187e-01]]],
[[[ 5.9180e-01, 8.6087e-01, 5.6462e-01, ..., 3.6411e-01,
4.3536e-01, 2.4957e-02],
[-2.6232e-01, -8.7359e-01, -5.1244e-01, ..., 1.6014e-01,
9.3209e-02, -3.3173e-01],
[-1.4074e-01, 1.8524e-01, 5.7941e-01, ..., -1.9595e-01,
-2.7180e-02, 6.5092e-01],
...,
[ 3.6713e-01, -4.3790e-01, 4.0609e-01, ..., 3.6663e-01,
8.6027e-01, 6.6104e-01],
[-3.0779e-02, -6.9030e-01, 1.3071e-01, ..., 9.3386e-01,
1.5637e+00, 9.6416e-02],
[ 8.1157e-01, -2.2730e-01, -1.1786e-01, ..., 4.0188e-01,
3.3407e-01, 6.9729e-01]],
[[-2.2981e+00, -6.1263e-01, -1.8325e+00, ..., -1.6668e+00,
-2.2728e+00, -1.5405e+00],
[-2.4057e+00, -6.6196e-01, -1.4482e+00, ..., -1.3190e+00,
-2.4101e+00, -1.7083e+00],
[-2.4284e+00, -1.9229e+00, -2.6384e+00, ..., -2.1039e+00,
-2.3183e+00, -1.1310e+00],
...,
[-7.5965e-01, -8.6498e-01, -9.2972e-01, ..., -4.9081e-01,
-1.6217e-01, 7.2362e-01],
[-2.6027e+00, -1.5518e-01, -3.0407e-01, ..., -2.4541e-02,
-2.6984e-02, 6.6555e-01],
[-1.5419e+00, -1.1845e+00, -9.7109e-01, ..., -4.6006e-01,
-5.3155e-01, -1.3180e-01]],
[[ 1.8468e+00, 2.4850e+00, 1.6194e+00, ..., 1.5979e+00,
7.8277e-01, 2.1089e+00],
[-2.9852e-01, 3.5592e-01, 6.1742e-01, ..., 8.3522e-01,
3.4948e-01, 2.2354e+00],
[ 9.6196e-03, 3.0400e-01, 1.8839e-01, ..., -1.6874e-01,
7.0890e-01, 1.9795e+00],
...,
[ 6.8656e-01, 7.0945e-01, 2.0764e+00, ..., 2.6782e-01,
3.9777e-02, 1.5677e+00],
[ 1.1511e+00, 2.2468e+00, 1.9947e+00, ..., 1.0108e+00,
6.6116e-01, 1.6302e+00],
[ 2.4030e+00, 1.8164e+00, 2.3598e+00, ..., 1.9800e+00,
1.0120e+00, 1.6814e+00]],
...,
[[-1.9576e+00, -4.8466e+00, -3.1133e+00, ..., -2.1012e+00,
-1.6886e+00, -3.0733e+00],
[-2.3784e+00, -5.2081e+00, -2.8967e+00, ..., -2.4595e+00,
-2.3012e+00, -2.9797e+00],
[-1.5927e+00, -5.6102e+00, -3.9273e+00, ..., -2.3956e+00,
-2.4685e+00, -2.2818e+00],
...,
[-2.6284e+00, -4.2930e+00, -2.2559e+00, ..., -4.2105e+00,
-2.1620e+00, -2.6323e+00],
[-2.5171e+00, -5.4164e+00, -3.7751e+00, ..., -4.3709e+00,
-3.1914e+00, -4.7962e+00],
[-2.6652e+00, -5.9672e+00, -2.8681e+00, ..., -2.9656e+00,
-1.6381e+00, -3.2778e+00]],
[[ 9.3661e-01, 1.6110e+00, 8.0550e-01, ..., 1.0715e+00,
1.5365e+00, 1.3963e+00],
[ 1.1436e+00, 9.9406e-01, 9.1516e-01, ..., 9.1996e-01,
7.8862e-01, 1.9287e+00],
[ 6.2739e-01, -7.1319e-02, 5.6439e-01, ..., 3.0138e-01,
5.6326e-02, 1.3924e+00],
...,
[ 1.5561e+00, 8.0178e-02, 1.2364e+00, ..., 1.5681e+00,
1.8388e+00, 1.4512e+00],
[ 2.0194e+00, 1.1647e+00, 1.7601e+00, ..., 1.9755e+00,
1.7145e+00, 2.5254e+00],
[ 1.1802e+00, 8.8015e-01, 1.3260e+00, ..., 1.3302e+00,
1.5102e+00, 1.5299e+00]],
[[-8.6301e-02, -6.2526e-01, -1.1581e+00, ..., -5.2507e-01,
-8.8819e-01, -1.0749e-01],
[-8.4570e-01, -2.5684e+00, -2.5208e+00, ..., -1.1561e+00,
-1.0619e+00, -7.1556e-01],
[-1.4351e+00, -1.8847e+00, -6.2074e-01, ..., -2.9538e-01,
-4.0650e-01, -2.8347e-01],
...,
[-1.6275e+00, -1.6707e+00, -6.4443e-01, ..., 2.4903e-02,
-3.9755e-01, -4.2508e-01],
[-1.7334e+00, -1.3034e+00, -6.2604e-01, ..., -4.0530e-01,
-4.0850e-01, 4.1546e-01],
[-2.6720e-01, 4.0326e-01, -2.0519e-01, ..., 1.7306e-01,
5.0264e-02, 6.4293e-01]]],
[[[ 8.0536e-01, 1.4811e+00, 1.1548e+00, ..., 1.2219e+00,
9.6154e-01, 9.6344e-01],
[-1.0054e+00, -1.9807e+00, -1.0623e+00, ..., 6.3997e-02,
-5.7588e-01, -1.5424e+00],
[-1.8426e-01, -7.7863e-01, 1.6510e-01, ..., 1.5857e-01,
7.7276e-01, 8.7502e-01],
...,
[ 5.3657e-02, -1.0879e+00, -1.7152e+00, ..., -3.3269e-01,
5.3467e-02, 2.4484e-01],
[-6.0378e-01, -1.8439e+00, -1.3473e+00, ..., -5.8130e-01,
-3.0341e-01, -1.3940e-01],
[-2.9970e-01, -1.7415e+00, -1.7288e+00, ..., -1.2141e+00,
-1.6002e+00, -2.8517e-01]],
[[-3.8202e+00, -7.3357e-01, -1.5502e+00, ..., -1.6112e+00,
-2.3991e+00, -1.1218e+00],
[-4.0630e+00, -2.7100e+00, -1.8260e+00, ..., -2.1275e+00,
-3.1036e+00, -8.4179e-01],
[-2.3498e+00, -1.4325e+00, -1.7115e+00, ..., -2.5887e+00,
-2.6913e+00, -9.6223e-01],
...,
[-1.5616e+00, -1.6300e+00, -2.7932e+00, ..., -1.8773e+00,
-2.2571e+00, -7.5996e-01],
[-3.0540e+00, -1.6577e+00, -2.7962e+00, ..., -2.3032e+00,
-2.4984e+00, -1.2966e+00],
[-2.2821e+00, -1.9988e+00, -2.9578e+00, ..., -1.9794e+00,
-1.9701e+00, -1.0262e+00]],
[[ 2.9279e+00, 4.1878e+00, 3.3329e+00, ..., 3.3327e+00,
1.1720e+00, 2.9298e+00],
[ 2.9521e-01, 1.0059e+00, 2.2209e+00, ..., 1.2410e+00,
-8.3038e-01, 2.6501e+00],
[ 1.1195e+00, 1.4204e+00, 1.5364e+00, ..., 7.5253e-01,
-9.2357e-02, 2.0485e+00],
...,
[ 1.0532e+00, 1.0778e+00, 2.3038e+00, ..., 1.3093e-01,
1.0837e+00, 2.3411e+00],
[ 7.1200e-01, 1.8562e+00, 1.9294e+00, ..., 6.4204e-02,
1.0700e+00, 1.8514e+00],
[ 1.6647e+00, 1.6904e+00, 2.0395e+00, ..., 1.3059e+00,
1.8162e+00, 1.7923e+00]],
...,
[[-1.8576e+00, -7.1971e+00, -5.3551e+00, ..., -3.9567e+00,
-3.1477e+00, -5.3674e+00],
[-2.4792e+00, -6.8233e+00, -4.1630e+00, ..., -4.7061e+00,
-2.9035e+00, -4.1263e+00],
[-1.0111e+00, -4.2824e+00, -3.4197e+00, ..., -3.2905e+00,
-3.3512e+00, -2.3948e+00],
...,
[-2.5569e+00, -5.1789e+00, -4.1540e+00, ..., -5.5611e+00,
-4.5166e+00, -4.6658e+00],
[-2.7034e+00, -5.6367e+00, -5.6095e+00, ..., -6.4043e+00,
-5.6745e+00, -6.2481e+00],
[-1.4977e+00, -4.1049e+00, -3.0111e+00, ..., -3.3279e+00,
-2.6342e+00, -3.2175e+00]],
[[-2.3179e-01, 3.0084e-01, 5.5359e-01, ..., 7.0763e-01,
9.7574e-01, 1.4248e+00],
[ 1.6712e+00, 1.1029e+00, 1.7669e+00, ..., 1.7388e+00,
2.7124e+00, 2.9944e+00],
[ 1.2571e+00, 3.6989e-01, 7.2649e-01, ..., 8.1427e-01,
2.0230e+00, 2.3259e+00],
...,
[ 1.2330e+00, 2.3015e-01, -5.5846e-01, ..., 1.3159e+00,
1.0560e+00, 1.6058e+00],
[ 1.8618e+00, 1.6464e+00, 6.0365e-01, ..., 1.4272e+00,
1.2576e+00, 1.9487e+00],
[ 4.5411e-01, 7.8622e-01, 5.4651e-01, ..., 5.2359e-01,
7.1965e-01, 1.6622e+00]],
[[-6.0578e-01, 3.2317e-01, 1.3031e-01, ..., 4.9680e-01,
5.1550e-01, 1.3747e+00],
[-1.4656e+00, -1.7182e+00, -1.5307e+00, ..., -7.2710e-01,
-4.4995e-01, 2.3053e-01],
[-2.3260e+00, -8.7132e-01, 3.1037e-01, ..., -3.1395e-03,
1.6209e-01, 7.2848e-01],
...,
[-1.8501e+00, -1.8388e+00, -1.4678e+00, ..., -8.9542e-01,
-1.1864e+00, -8.4317e-01],
[-1.4811e+00, -2.0189e+00, -1.6904e+00, ..., -9.7499e-01,
-1.3831e+00, -7.0701e-01],
[-7.6873e-01, -8.6193e-01, -1.1263e+00, ..., -6.2687e-01,
-8.8694e-01, -1.0854e-01]]]], device='cuda:0')],
10: [tensor([[[[-1.4307e-01, 1.2029e+00, 2.1854e-01, ..., -3.0721e+00,
-2.4462e+00, -1.1601e+00],
[ 5.0743e-01, 8.0299e-01, 8.3677e-01, ..., -1.5401e+00,
-2.1548e+00, -2.7240e-01],
[ 1.2135e-02, -4.4783e-01, -2.8447e-01, ..., -9.0429e-01,
-2.6326e+00, -1.1430e-01],
...,
[-7.7364e-01, -1.9374e+00, 1.9155e-01, ..., -2.2588e+00,
-4.1397e+00, 2.4722e-01],
[-2.0332e+00, -1.3596e+00, -7.0416e-01, ..., -3.3390e+00,
-3.7752e+00, 4.5873e-01],
[-9.8517e-01, -2.8462e+00, -3.0607e+00, ..., -3.7963e+00,
-3.2529e+00, -1.5080e+00]],
[[-1.4862e+00, -1.9687e+00, -3.4767e+00, ..., -6.3547e+00,
-7.3421e+00, -1.4736e+00],
[-3.9314e+00, -4.5182e+00, -4.0292e+00, ..., -5.9403e+00,
-8.3471e+00, -2.4595e+00],
[-4.0303e+00, -4.6464e+00, -4.3561e+00, ..., -4.9235e+00,
-7.8318e+00, -2.5174e+00],
...,
[-5.6034e+00, -5.4571e+00, -4.6420e+00, ..., -8.0229e+00,
-9.1631e+00, -1.9870e+00],
[-7.1511e+00, -6.9001e+00, -6.9053e+00, ..., -9.6214e+00,
-1.0795e+01, -4.4061e+00],
[-1.4976e+00, -2.9460e+00, -3.5019e+00, ..., -3.7481e+00,
-3.7657e+00, -8.1157e-01]],
[[ 1.8480e-01, 1.2504e+00, 1.6468e+00, ..., 1.5326e+00,
2.4491e+00, 1.0151e+00],
[ 1.9533e-01, -1.8327e-02, 6.4620e-01, ..., 1.5393e+00,
3.0631e+00, 1.2336e-01],
[ 3.3526e-01, 1.4657e+00, 1.5013e+00, ..., 1.9829e-01,
1.8925e+00, -3.5109e-01],
...,
[ 2.8824e-01, 6.8543e-01, 1.2403e+00, ..., 5.0093e-01,
9.4346e-01, -1.2991e+00],
[ 4.6494e-01, 3.5055e-01, 1.2215e+00, ..., -2.1847e-01,
8.4852e-01, -1.4319e+00],
[ 5.0980e-01, 7.9207e-01, 1.3712e+00, ..., -5.8096e-02,
-4.3800e-01, -1.8516e+00]],
...,
[[-9.6924e-01, -1.6393e+00, -2.3841e+00, ..., -5.1147e+00,
-5.2942e+00, -8.0556e-01],
[-2.9357e+00, -2.9359e+00, -1.8767e+00, ..., -3.6972e+00,
-4.8497e+00, -1.8292e+00],
[-2.3097e+00, -1.6599e+00, -9.1055e-01, ..., -4.5413e+00,
-5.1850e+00, -1.3376e+00],
...,
[-1.7323e+00, -1.3029e+00, -7.1122e-01, ..., -4.4483e+00,
-5.1317e+00, -1.9638e+00],
[-2.8660e+00, -2.6527e+00, -2.5103e+00, ..., -3.8028e+00,
-5.1506e+00, -1.1811e+00],
[-1.2077e+00, -1.6810e+00, -2.0501e+00, ..., -2.0157e+00,
-2.1668e+00, 2.2981e-01]],
[[-1.5982e+00, -2.0443e+00, -4.0827e+00, ..., -1.5109e+00,
-5.2348e-01, -5.7447e-01],
[-3.8438e+00, -3.7556e+00, -4.7678e+00, ..., -3.1977e+00,
-1.8287e+00, -3.8112e-01],
[-4.5010e+00, -4.2710e+00, -4.4021e+00, ..., -2.3404e+00,
-1.5639e+00, -4.4522e-01],
...,
[-4.8015e+00, -3.9709e+00, -4.3116e+00, ..., -4.2492e+00,
-1.9757e+00, -9.0618e-01],
[-3.9925e+00, -1.9220e+00, -2.8908e+00, ..., -1.6672e+00,
-5.1109e-01, 1.1622e+00],
[-3.0015e+00, -3.0170e+00, -2.5835e+00, ..., -3.2881e-01,
-7.8450e-01, 6.3465e-01]],
[[ 1.2607e+00, -5.3438e-02, 1.1574e+00, ..., 5.7924e-01,
8.0363e-01, -1.4421e+00],
[ 2.4663e+00, 1.4254e+00, 1.3423e+00, ..., 3.7540e+00,
3.0870e+00, 2.3130e+00],
[ 1.6555e+00, 5.7378e-01, 5.5169e-02, ..., 3.0748e+00,
1.8247e+00, 1.2280e+00],
...,
[ 2.3303e+00, 1.4232e+00, 1.1954e+00, ..., 2.6066e+00,
2.4040e+00, 7.7447e-01],
[ 2.6876e+00, 3.7446e+00, 2.4074e+00, ..., 2.7068e+00,
1.3679e+00, 1.9582e+00],
[ 1.7951e+00, 1.5774e+00, 3.5313e-01, ..., 4.0017e-01,
7.7485e-02, 8.5503e-01]]],
[[[ 7.7068e-01, 1.5231e+00, 6.5252e-01, ..., -3.7869e+00,
-2.5774e+00, -1.0130e+00],
[ 1.6373e+00, 1.1703e+00, 5.8255e-01, ..., -2.9749e+00,
-2.5793e+00, -2.2706e-01],
[ 7.3812e-01, 2.7880e-01, -7.3596e-03, ..., -2.5508e+00,
-2.9900e+00, -2.7595e-02],
...,
[ 8.2227e-01, -9.7802e-01, -6.9216e-01, ..., -4.1533e-01,
-3.6634e+00, 8.1515e-02],
[ 1.7048e-01, -3.2228e-01, -6.4297e-01, ..., -3.5573e+00,
-3.5767e+00, 3.2990e-01],
[-3.8396e-01, -3.4529e+00, -3.3581e+00, ..., -4.1066e+00,
-3.5230e+00, -1.5787e+00]],
[[ 3.7672e-01, -1.1134e+00, -2.6465e+00, ..., -7.5664e+00,
-7.4219e+00, -1.3316e+00],
[-1.3269e+00, -2.9435e+00, -4.1140e+00, ..., -7.9885e+00,
-8.5659e+00, -2.4290e+00],
[-2.2871e-02, -1.5939e+00, -2.7820e+00, ..., -6.7497e+00,
-7.8428e+00, -2.4083e+00],
...,
[-2.3725e+00, -3.8443e+00, -3.5847e+00, ..., -8.3248e+00,
-9.6766e+00, -2.3384e+00],
[-4.2379e+00, -5.8259e+00, -5.0334e+00, ..., -1.0118e+01,
-1.1063e+01, -4.4159e+00],
[-3.6361e-01, -3.3353e+00, -3.5570e+00, ..., -4.0254e+00,
-3.5596e+00, -8.0626e-01]],
[[ 4.5453e-01, 2.1387e+00, 1.8362e+00, ..., 7.8477e-01,
2.2088e+00, 9.6135e-01],
[-1.7320e-01, 9.8187e-01, 5.6281e-01, ..., 1.0158e+00,
2.7858e+00, 8.1203e-03],
[ 5.8966e-01, 3.3826e+00, 2.1642e+00, ..., -6.9860e-01,
2.0187e+00, -3.0497e-01],
...,
[ 1.0572e+00, 2.8660e+00, 1.2632e+00, ..., -4.4529e-01,
1.3482e+00, -1.4830e+00],
[ 5.3204e-01, 2.1382e+00, 8.7114e-01, ..., -1.1683e+00,
3.7239e-01, -1.7091e+00],
[-3.0793e-01, 1.8199e+00, 1.0787e+00, ..., -4.7905e-01,
-6.6192e-01, -1.9278e+00]],
...,
[[-3.2408e-01, -8.3228e-01, -1.5044e+00, ..., -5.4856e+00,
-5.4447e+00, -7.1853e-01],
[-2.3687e+00, -4.0747e+00, -2.3310e+00, ..., -4.5193e+00,
-5.3582e+00, -1.8547e+00],
[-8.9885e-01, -2.4487e+00, -1.3399e+00, ..., -5.2923e+00,
-5.8430e+00, -1.2806e+00],
...,
[-1.4845e+00, -3.0874e+00, -1.3042e+00, ..., -4.8132e+00,
-6.5136e+00, -1.6975e+00],
[-1.8819e+00, -3.6883e+00, -1.9028e+00, ..., -3.1942e+00,
-5.3175e+00, -9.6681e-01],
[-1.2346e+00, -3.7102e+00, -2.5782e+00, ..., -1.1860e+00,
-2.2957e+00, 3.0587e-01]],
[[-1.0162e+00, -8.0707e-01, -8.3853e-01, ..., -2.0777e+00,
-4.8904e-01, -6.1942e-01],
[-2.9302e+00, -2.8613e+00, -1.4499e+00, ..., -3.7759e+00,
-1.8170e+00, -3.9871e-01],
[-2.0272e+00, -3.6461e+00, -1.4725e+00, ..., -2.3422e+00,
-1.5280e+00, -3.8861e-01],
...,
[-3.1016e+00, -4.4001e+00, -2.4116e+00, ..., -3.6446e+00,
-2.2852e+00, -4.2860e-01],
[-3.0946e+00, -2.3283e+00, -8.7084e-01, ..., -1.6964e+00,
-6.9287e-01, 1.4822e+00],
[-3.2921e+00, -3.3896e+00, -1.3810e+00, ..., -1.7544e-01,
-9.6789e-01, 7.0665e-01]],
[[ 8.3423e-01, 7.9095e-01, 1.0116e+00, ..., -2.7261e-01,
6.1356e-01, -1.3884e+00],
[ 1.3348e+00, 1.7686e+00, 1.7170e+00, ..., 3.2413e+00,
3.0192e+00, 2.2828e+00],
[ 4.5094e-01, -5.9340e-02, 4.9544e-01, ..., 2.3415e+00,
1.6051e+00, 1.2708e+00],
...,
[ 1.8793e+00, 1.2826e+00, 2.0155e+00, ..., 3.0949e+00,
2.0418e+00, 5.3300e-01],
[ 3.0990e+00, 3.3892e+00, 2.1855e+00, ..., 8.0420e-01,
9.4622e-01, 1.9405e+00],
[ 1.9651e+00, 5.7928e-01, 7.1468e-01, ..., 8.3477e-01,
3.4984e-01, 9.3993e-01]]],
[[[ 3.4558e-01, 1.1945e+00, 2.2152e-01, ..., -7.4647e-01,
-1.6152e+00, -3.6453e-01],
[ 3.9534e-01, 9.1393e-01, 1.5074e+00, ..., 4.6238e-01,
9.4684e-01, 5.2449e-01],
[-5.2461e-01, -4.7594e-01, -2.6612e-02, ..., -1.0553e+00,
-4.0426e+00, -2.8078e+00],
...,
[-3.6945e-01, -9.2149e+00, -2.6647e+00, ..., -1.4931e-01,
2.9025e+00, 3.5160e+00],
[-1.6857e+00, -4.6023e+00, -2.7669e+00, ..., -2.8456e+00,
-5.0799e+00, -9.8235e-01],
[-6.0123e-01, -3.2764e+00, -4.4790e+00, ..., -4.6199e+00,
-3.4951e+00, -1.9757e+00]],
[[-1.1679e+00, -1.7985e+00, -2.8044e+00, ..., -4.5574e+00,
-6.0961e+00, -1.6291e+00],
[-4.1713e+00, -4.3728e+00, -3.2235e+00, ..., -7.0664e+00,
-7.0733e+00, -4.3673e+00],
[-4.1532e+00, -4.3999e+00, -3.7557e+00, ..., -9.8158e+00,
-8.4907e+00, -3.5995e+00],
...,
[-5.4191e+00, -1.1335e+01, -6.0546e+00, ..., 1.2626e-01,
-3.3428e+00, -2.5081e+00],
[-6.9217e+00, -7.9902e+00, -1.2407e+01, ..., -1.4424e+01,
-1.2430e+01, -6.6163e+00],
[-2.0957e+00, -2.7724e+00, -2.1356e+00, ..., -3.6505e+00,
-3.0575e+00, -7.2606e-01]],
[[-4.6951e-03, 1.2948e+00, 1.5965e+00, ..., 6.0905e-01,
2.5673e+00, 1.1166e+00],
[ 9.7231e-02, 6.1757e-02, 3.7771e-01, ..., -5.8198e-01,
1.4385e+00, -3.7724e-01],
[ 6.2333e-01, 1.3521e+00, 1.4387e+00, ..., -1.2039e+00,
3.5906e+00, 2.0491e+00],
...,
[-3.3282e-01, 2.7859e-01, -5.5391e+00, ..., -3.1375e+00,
-3.2302e-01, -1.8743e+00],
[-2.7363e-01, 7.4219e-01, -4.0706e-01, ..., -3.3239e+00,
-1.4861e+00, -1.5560e+00],
[ 2.7793e-01, 7.3258e-02, 1.3074e+00, ..., -1.0390e+00,
3.3443e-03, -1.4345e+00]],
...,
[[-7.0971e-01, -1.5980e+00, -1.9897e+00, ..., -2.9871e+00,
-4.6854e+00, -1.0440e+00],
[-2.8544e+00, -2.6164e+00, -1.3307e+00, ..., -3.7482e+00,
-4.7867e+00, -1.6915e+00],
[-2.1176e+00, -1.3044e+00, -1.1805e+00, ..., -5.4688e+00,
-7.0178e+00, -2.7142e+00],
...,
[-2.3397e+00, -4.6578e+00, 9.2660e-01, ..., -1.3582e+00,
-1.4362e+00, -2.5065e+00],
[-2.7562e+00, -4.6854e+00, -3.5740e+00, ..., -7.6193e+00,
-6.1591e+00, -1.9310e+00],
[-1.1267e+00, -1.1546e+00, -4.8476e-01, ..., 5.1391e-01,
-1.1653e+00, 3.5842e-01]],
[[-1.4225e+00, -2.3394e+00, -3.8443e+00, ..., -2.7415e+00,
-1.9276e+00, -4.6381e-01],
[-3.8973e+00, -3.3672e+00, -4.0052e+00, ..., -2.1030e+00,
-1.9473e+00, -2.0502e-01],
[-4.7410e+00, -3.7435e+00, -3.2310e+00, ..., -2.8097e+00,
-5.8309e+00, -2.0448e+00],
...,
[-6.5181e+00, -4.2806e+00, 4.3058e+00, ..., -2.0867e+00,
-1.1344e+00, 6.2934e-01],
[-4.2512e+00, -4.9024e+00, 8.7886e-01, ..., -1.9230e+00,
-3.4423e+00, 1.2496e+00],
[-2.6916e+00, -3.5010e+00, -1.0576e+00, ..., 2.4391e+00,
-3.8732e-01, 6.6566e-01]],
[[ 1.2201e+00, -1.7198e-02, 1.0646e+00, ..., -1.0595e+00,
1.4620e-01, -2.8066e-01],
[ 2.3846e+00, 1.2972e+00, 1.1870e+00, ..., 2.2356e+00,
3.2509e+00, 3.7355e+00],
[ 1.3909e+00, 7.2321e-01, 4.5439e-01, ..., 1.4739e+00,
-2.9975e-01, 7.7316e-01],
...,
[ 2.7005e+00, 1.9144e+00, 6.6792e+00, ..., -4.7947e+00,
2.2732e-01, 3.3211e+00],
[ 2.5808e+00, 2.6606e+00, 6.6969e+00, ..., 3.0619e+00,
1.6766e+00, 2.8927e+00],
[ 2.4611e+00, 1.3773e+00, -1.2120e+00, ..., -1.3256e+00,
-2.0438e+00, 1.1391e+00]]],
[[[ 3.9932e-01, 9.4595e-01, 2.0723e+00, ..., 2.3778e+00,
2.4336e+00, 8.8457e-01],
[-5.9019e-02, -5.2769e-01, 1.8658e+00, ..., 8.5792e+00,
1.8559e+00, -6.4165e-01],
[-2.1796e-01, -1.2191e-02, 1.4589e+00, ..., 2.5081e+00,
-3.6979e+00, -2.1764e+00],
...,
[-1.1628e+00, -5.6380e+00, 1.3048e+00, ..., -5.9882e+00,
-3.2941e+00, 3.6957e+00],
[-2.1824e+00, -2.9279e+00, -4.7864e-01, ..., -1.4045e+00,
5.1439e+00, 3.4044e+00],
[-2.4141e+00, -4.2912e+00, -4.1909e+00, ..., -2.6158e+00,
-2.8412e+00, -2.7331e+00]],
[[-1.5567e+00, -2.3457e+00, -3.4119e+00, ..., 3.6943e+00,
-8.3462e-01, -2.5025e+00],
[-5.3126e+00, -6.1367e+00, -5.4429e+00, ..., 2.5156e+00,
-8.0885e+00, -7.6481e+00],
[-3.8474e+00, -4.3792e+00, -4.7254e+00, ..., -8.9665e+00,
-1.2696e+01, -5.1116e+00],
...,
[-7.0379e+00, -1.2992e+01, -1.1836e+01, ..., -7.7303e+00,
-2.0046e+00, 3.2521e+00],
[-9.7619e+00, -9.5179e+00, -6.7391e+00, ..., -8.7655e-02,
-9.9806e-01, -2.1894e+00],
[-2.6199e+00, -5.8114e+00, -4.4596e+00, ..., -5.1918e+00,
-5.9501e+00, -4.4559e+00]],
[[ 7.6624e-02, 1.5673e+00, 7.6137e-01, ..., -6.5907e-01,
2.8544e+00, 6.1459e-01],
[ 9.8362e-02, 5.2924e-01, -8.6409e-01, ..., -3.4381e+00,
2.7756e+00, 1.7825e+00],
[ 7.0401e-01, 2.4604e+00, 1.5529e+00, ..., 6.0788e-01,
3.7190e+00, 1.2416e+00],
...,
[-2.3828e-01, -4.7490e-01, -1.0911e+01, ..., -4.2412e+00,
-2.7566e-01, -5.2728e-01],
[-1.1547e+00, -5.2964e-01, -9.4037e-01, ..., -4.5555e+00,
-2.6408e+00, -1.7678e-01],
[ 5.3463e-01, 1.6904e-02, -2.3861e-01, ..., -2.9691e+00,
-3.2648e+00, 6.5583e-02]],
...,
[[-9.0180e-01, -1.3530e+00, 2.8239e-01, ..., -1.2052e+00,
-4.9487e+00, 1.4020e-01],
[-4.0001e+00, -4.3519e+00, -1.0790e+00, ..., -4.9899e+00,
-6.7788e+00, -2.2065e+00],
[-2.2714e+00, -1.4087e+00, -3.2970e-01, ..., -8.4666e+00,
-6.5589e+00, -2.0750e+00],
...,
[-3.9155e+00, -7.3246e+00, -5.0845e+00, ..., -6.8729e+00,
-3.5687e+00, -3.6991e+00],
[-2.8920e+00, -5.3942e+00, -2.1615e+00, ..., -9.0702e-02,
-1.5124e+00, -3.4780e+00],
[-3.4872e+00, -4.1753e+00, -3.8056e+00, ..., -5.1677e+00,
-5.9803e+00, -3.8311e+00]],
[[-1.3369e+00, -3.5669e-01, -6.0701e-01, ..., 2.6966e+00,
-2.2728e+00, 1.3759e+00],
[-4.0284e+00, -1.1828e+00, -6.8972e-01, ..., 8.1045e-01,
-6.9446e+00, 9.0782e-02],
[-3.6419e+00, -1.2986e+00, -1.8137e+00, ..., -6.8738e+00,
-8.5915e+00, -4.9652e-01],
...,
[-5.9285e+00, -3.1170e+00, 2.8154e+00, ..., -3.7674e+00,
8.4524e-02, 2.2220e+00],
[-3.4331e+00, -1.1072e-01, 2.4772e+00, ..., 2.4571e+00,
4.0356e+00, 1.2972e+00],
[-3.2570e+00, -2.0632e+00, -1.1693e+00, ..., -5.6839e-01,
4.7939e-01, 2.1503e-02]],
[[ 1.7152e+00, 5.7508e-01, 8.9134e-01, ..., 1.1150e-01,
2.6364e+00, 1.5759e-01],
[ 3.3384e+00, 2.1722e+00, 2.4068e+00, ..., 6.0091e+00,
3.6293e+00, 3.6777e+00],
[ 2.0017e+00, 8.1204e-01, 7.2221e-01, ..., 8.3656e+00,
1.9485e+00, 1.1953e+00],
...,
[ 3.6468e+00, 2.6703e+00, 8.8898e+00, ..., -8.0868e+00,
-4.2047e+00, 1.9451e+00],
[ 4.9261e+00, 2.9938e+00, 4.1298e+00, ..., -5.1762e+00,
1.8349e+00, 4.6753e+00],
[ 2.8780e+00, 1.6860e+00, 7.3311e-01, ..., 2.9514e+00,
3.5168e+00, 1.7482e+00]]],
[[[ 1.3779e-01, 4.1049e-01, -4.1007e+00, ..., -1.3536e-01,
-3.1496e+00, -1.8644e+00],
[ 1.9014e-01, 6.4619e-01, -3.0153e+00, ..., 1.9724e+00,
-1.1034e+00, -1.0004e+00],
[-9.7698e-02, 1.1210e+00, -2.2769e-01, ..., -1.3607e+00,
-1.2950e+00, -1.2490e+00],
...,
[-1.0273e+00, -2.4249e+00, -6.9132e-02, ..., -1.4856e+00,
-1.2585e+00, -8.1933e-01],
[-1.2339e+00, -1.3301e+00, -2.5261e-01, ..., -3.4764e-01,
-8.8963e-01, -2.2250e+00],
[-2.6348e+00, -5.1856e+00, -4.3969e+00, ..., -5.0680e+00,
-4.2815e+00, -3.8808e+00]],
[[-1.6055e+00, -1.7831e+00, -8.1331e+00, ..., -2.2567e+00,
-7.4175e+00, -2.3155e+00],
[-4.5122e+00, -4.8154e+00, -5.2982e+00, ..., -5.4156e+00,
-7.9241e+00, -2.8965e+00],
[-3.5281e+00, -5.7874e+00, -5.8207e+00, ..., -1.2340e+01,
-1.1212e+01, -5.3083e+00],
...,
[-5.5654e+00, -5.1748e+00, -2.6871e+00, ..., -7.8221e+00,
-6.3186e+00, -3.4822e+00],
[-7.2580e+00, -8.0696e+00, -6.8709e+00, ..., -7.7878e+00,
-8.7505e+00, -6.6646e+00],
[-2.4912e+00, -4.2268e+00, -3.8915e+00, ..., -4.3029e+00,
-4.2279e+00, -2.5757e+00]],
[[ 5.9501e-01, 2.1567e+00, 1.1874e+00, ..., 3.0846e+00,
3.6882e+00, 1.1070e+00],
[-6.4140e-01, -6.1939e-01, 1.1400e-01, ..., 1.4531e+00,
2.9953e+00, 3.3977e-01],
[ 4.2455e-01, 6.8571e-01, 8.9102e-01, ..., -9.0233e-01,
-5.9940e-01, -8.0872e-01],
...,
[ 3.9499e-01, 8.4643e-01, -1.6980e-01, ..., 2.3196e-01,
9.2984e-02, -3.4320e-01],
[ 8.4996e-01, 6.1984e-01, 4.0305e-01, ..., 4.8048e-01,
-8.1535e-01, -2.2036e-01],
[ 1.3001e+00, 1.5232e+00, 1.2863e+00, ..., 1.2632e+00,
9.5802e-01, 2.7964e-01]],
...,
[[-6.8915e-01, -2.6738e+00, -4.6816e+00, ..., -5.3309e+00,
-5.5263e+00, -1.7937e+00],
[-1.9736e+00, -3.3625e+00, -3.5033e+00, ..., -4.2416e+00,
-5.2990e+00, -2.4227e+00],
[-1.5927e+00, -1.6959e+00, -3.6006e+00, ..., -6.1548e+00,
-7.1076e+00, -3.0292e+00],
...,
[-1.9549e+00, -1.5696e+00, 7.1971e-01, ..., -3.2929e+00,
-3.4403e+00, -1.1515e+00],
[-3.1051e+00, -3.1019e+00, -1.8247e+00, ..., -1.7089e+00,
-4.1863e+00, -2.5123e+00],
[-3.2651e+00, -3.5958e+00, -3.2020e+00, ..., -3.4423e+00,
-3.9391e+00, -1.7468e+00]],
[[-1.6095e+00, -2.9595e+00, -1.8178e-01, ..., -3.2521e+00,
-1.0342e+00, 7.3618e-01],
[-3.1653e+00, -2.9671e+00, -2.6938e+00, ..., -3.8386e+00,
-1.6785e+00, 8.3374e-02],
[-2.1941e+00, -1.0106e+00, -4.4102e+00, ..., -2.3162e+00,
-1.1817e-01, 4.5933e-01],
...,
[-3.2762e+00, -1.7310e+00, -2.1728e+00, ..., -2.1487e+00,
-2.2860e+00, -3.2336e-01],
[-3.7589e+00, -7.2436e-01, -1.9867e+00, ..., -1.5730e+00,
-3.0738e+00, -9.0521e-01],
[-4.3424e+00, -2.9577e+00, -2.5599e+00, ..., -1.7716e+00,
-2.9606e+00, -1.0255e+00]],
[[ 1.3935e+00, 1.1456e+00, 1.5242e+00, ..., 2.9389e+00,
1.5162e+00, -1.7841e-01],
[ 2.6479e+00, 2.0902e+00, 1.4705e+00, ..., 3.8891e+00,
3.5224e+00, 4.1992e+00],
[ 2.0014e+00, 1.6327e+00, 1.6020e+00, ..., 3.2570e+00,
3.1189e+00, 1.8082e+00],
...,
[ 2.1629e+00, 4.8502e-01, 3.8795e-01, ..., 2.8201e+00,
1.6377e+00, 1.7210e+00],
[ 3.6597e+00, 3.5514e+00, 2.4635e+00, ..., 3.0271e+00,
2.8983e+00, 2.6802e+00],
[ 1.8377e+00, 5.3586e-01, 2.6185e-02, ..., -1.3281e-01,
-3.7505e-01, -2.8967e-01]]]], device='cuda:0'),
tensor([[[[ 2.4658e-01, 1.4218e+00, 4.7647e-01, ..., -8.7971e-01,
-2.6467e-01, 6.3551e-01],
[ 4.0244e-01, 1.4223e-01, 6.2499e-01, ..., -3.6799e+00,
-1.9447e+00, 1.0376e+00],
[-5.3623e-01, 9.2121e-02, 6.5634e-01, ..., -2.7166e-01,
-1.4903e+00, 7.6227e-01],
...,
[-9.2174e-01, -1.8733e+00, -3.0967e-01, ..., -1.4358e+00,
-4.5933e-01, -3.7705e-01],
[-9.0210e-01, -8.4848e-01, 1.7928e-01, ..., -9.9840e-01,
-9.7441e-02, -1.5135e+00],
[-2.7298e+00, -5.2626e+00, -4.1791e+00, ..., -5.0038e+00,
-4.0499e+00, -3.8350e+00]],
[[-8.1393e-01, -2.3411e+00, -4.1620e+00, ..., -4.4644e+00,
-5.8021e+00, -4.9209e-01],
[-4.7208e+00, -6.1673e+00, -5.2226e+00, ..., -4.9896e+00,
-6.9652e+00, -1.5581e+00],
[-3.4995e+00, -4.0526e+00, -3.5056e+00, ..., -1.1096e+00,
-4.6469e+00, -1.3127e+00],
...,
[-5.8471e+00, -5.8748e+00, -3.9734e+00, ..., -6.7068e+00,
-4.8913e+00, -3.7070e+00],
[-6.9905e+00, -7.9696e+00, -5.9854e+00, ..., -6.8451e+00,
-7.9770e+00, -6.6083e+00],
[-2.5438e+00, -4.5780e+00, -4.3551e+00, ..., -3.7650e+00,
-4.2084e+00, -2.5674e+00]],
[[-3.0598e-01, 2.1538e+00, 1.1122e+00, ..., 9.3340e-01,
2.2638e+00, 9.3162e-01],
[-1.6214e-01, 1.1966e+00, 9.8080e-01, ..., 2.5647e+00,
1.8394e+00, -3.3775e-01],
[-9.6245e-02, 1.9826e+00, 1.6375e+00, ..., 1.8223e+00,
2.5895e+00, 4.4080e-01],
...,
[ 8.8718e-01, 1.3721e+00, 9.6082e-01, ..., -1.5567e-01,
5.4293e-01, 4.6678e-01],
[ 7.6113e-01, 6.4355e-01, 2.2120e-01, ..., -2.6276e-01,
3.4801e-01, 2.3473e-01],
[ 1.3357e+00, 1.6084e+00, 1.3277e+00, ..., 1.2290e+00,
1.2789e+00, 1.2975e-01]],
...,
[[-8.4536e-01, -1.9137e+00, -1.7613e+00, ..., -2.6290e+00,
-3.1278e+00, -5.9243e-01],
[-3.2505e+00, -4.4496e+00, -2.6272e+00, ..., -3.9937e+00,
-4.0586e+00, -2.9925e+00],
[-1.4644e+00, -2.1923e+00, -7.2936e-01, ..., -1.4247e+00,
-2.9882e+00, -1.6604e+00],
...,
[-2.7218e+00, -2.3977e+00, -1.6465e+00, ..., -1.6252e+00,
-3.7542e+00, -1.7550e+00],
[-2.9169e+00, -2.8833e+00, -1.3783e+00, ..., -1.5891e+00,
-4.2981e+00, -2.3122e+00],
[-3.5892e+00, -3.9548e+00, -3.1349e+00, ..., -3.3580e+00,
-3.7978e+00, -1.5709e+00]],
[[-1.7345e+00, -2.2142e+00, -2.8023e+00, ..., -1.5694e+00,
-1.4917e+00, -1.3303e+00],
[-4.4002e+00, -3.6870e+00, -2.7912e+00, ..., -1.8345e+00,
-1.8551e-01, -3.4244e-01],
[-3.3178e+00, -2.9436e+00, -2.8126e+00, ..., -2.6502e-01,
5.5718e-02, 1.3986e-01],
...,
[-4.0780e+00, -2.6800e+00, -4.1064e+00, ..., -1.7056e+00,
-2.7073e+00, -1.5133e+00],
[-3.6583e+00, -6.0232e-01, -2.2746e+00, ..., -7.0590e-01,
-2.6433e+00, -1.3237e+00],
[-4.4817e+00, -2.9769e+00, -2.5997e+00, ..., -2.0188e+00,
-3.1592e+00, -1.2897e+00]],
[[ 1.0945e+00, 6.1680e-01, 6.7531e-01, ..., -5.0952e-01,
1.1658e-01, -4.2912e-01],
[ 3.4533e+00, 3.3105e+00, 2.2463e+00, ..., 2.0871e+00,
1.7643e+00, 2.2600e+00],
[ 1.4375e+00, 5.8684e-01, 1.0464e-01, ..., 1.7663e+00,
2.7306e+00, 2.3646e+00],
...,
[ 2.8254e+00, 9.7070e-01, 1.6533e+00, ..., 2.1976e+00,
2.3268e+00, 1.1045e+00],
[ 3.7753e+00, 3.5145e+00, 2.0352e+00, ..., 3.0394e+00,
2.3508e+00, 2.0644e+00],
[ 1.9562e+00, 7.2400e-01, 2.5038e-01, ..., 2.5020e-01,
-3.6301e-01, -2.3060e-01]]],
[[[ 3.0767e-01, 1.3606e+00, 8.3976e-01, ..., 1.9655e+00,
3.1545e-01, 7.9222e-01],
[-4.9034e-02, -5.9141e-01, 9.2889e-01, ..., 2.2451e-01,
-1.0037e+00, -4.8675e-01],
[-4.0228e-01, -6.3930e-01, 6.0543e-01, ..., -1.2063e+00,
-3.7505e-01, -9.6857e-01],
...,
[-8.1669e-01, -2.1522e+00, -1.2931e-01, ..., -3.0065e+00,
-3.6085e+00, -8.0040e-01],
[-2.3041e+00, -1.7906e+00, 3.0041e-01, ..., -2.6490e+00,
-2.9064e+00, 2.9535e-02],
[-2.3472e+00, -4.0371e+00, -3.0570e+00, ..., -4.1981e+00,
-3.6089e+00, -1.6313e+00]],
[[-1.5776e+00, -1.8056e+00, -2.3320e+00, ..., -2.6277e-02,
-5.6027e+00, -1.0730e+00],
[-5.0833e+00, -6.6903e+00, -4.5220e+00, ..., -5.8035e+00,
-8.3627e+00, -4.0401e+00],
[-4.4612e+00, -5.1156e+00, -3.1918e+00, ..., -6.2923e+00,
-5.6233e+00, -4.1772e+00],
...,
[-7.2224e+00, -9.2597e+00, -7.4898e+00, ..., -8.2256e+00,
-9.3793e+00, -2.3141e+00],
[-9.7491e+00, -1.0083e+01, -8.2686e+00, ..., -1.0736e+01,
-1.1061e+01, -3.9562e+00],
[-2.5196e+00, -4.9736e+00, -4.2216e+00, ..., -4.2436e+00,
-4.2888e+00, -6.8866e-01]],
[[-3.5680e-01, 6.0408e-01, 6.2926e-01, ..., 1.6625e+00,
2.4054e+00, 1.4071e-01],
[-4.4280e-03, -3.9190e-02, -2.4866e-01, ..., 2.8191e-01,
2.1993e+00, -7.4793e-01],
[ 7.9142e-01, 2.3750e+00, 1.3029e+00, ..., 1.0935e+00,
1.6798e+00, 6.6498e-01],
...,
[ 5.4869e-01, 1.4329e+00, 4.3212e-01, ..., 1.3396e+00,
6.8599e-01, -9.9463e-01],
[-4.0069e-01, 8.5073e-01, 2.0492e-01, ..., 2.6925e-01,
6.3485e-01, -6.1203e-01],
[ 6.1439e-01, 1.1830e+00, 9.7382e-01, ..., -2.1590e-01,
-3.8107e-01, -1.8663e+00]],
...,
[[-9.2934e-01, -1.0349e+00, -6.0304e-01, ..., -2.1628e+00,
-1.7244e+00, 1.9410e-01],
[-4.0373e+00, -4.9815e+00, -2.3624e+00, ..., -4.6152e+00,
-5.1535e+00, -3.2777e+00],
[-2.3606e+00, -1.4208e+00, -6.4161e-01, ..., -1.9782e+00,
-3.4516e+00, -2.2588e+00],
...,
[-2.9597e+00, -4.0488e+00, -1.7952e+00, ..., -4.4623e+00,
-6.1013e+00, -2.9562e+00],
[-3.5534e+00, -5.4239e+00, -2.7068e+00, ..., -4.1332e+00,
-5.4056e+00, -1.3719e+00],
[-3.2828e+00, -3.7367e+00, -3.1735e+00, ..., -2.8210e+00,
-2.7234e+00, 4.0582e-01]],
[[-9.6605e-01, 1.7989e-01, -1.5368e+00, ..., -1.5188e+00,
-1.3680e-01, 3.2348e-01],
[-3.9779e+00, -1.2327e+00, -1.4391e+00, ..., -3.5509e+00,
-2.3019e+00, 4.2612e-01],
[-3.7156e+00, -1.4784e+00, -1.4073e+00, ..., -2.2828e+00,
-2.9147e+00, -5.3329e-01],
...,
[-4.5107e+00, -2.4611e+00, -3.1467e+00, ..., -3.1158e+00,
-2.5139e+00, -8.4720e-01],
[-3.6189e+00, -9.6483e-01, -2.0022e+00, ..., -1.9407e+00,
-1.1077e+00, 3.5849e-01],
[-3.7967e+00, -3.0276e+00, -3.0402e+00, ..., -1.8634e+00,
-1.3302e+00, 3.0977e-01]],
[[ 1.6672e+00, 6.7204e-01, 5.5087e-01, ..., 1.3420e+00,
1.7555e+00, 3.6591e-01],
[ 3.1716e+00, 2.1103e+00, 1.7524e+00, ..., 1.1450e+00,
2.4835e+00, 3.6803e+00],
[ 2.1436e+00, 3.2781e-01, 3.9893e-01, ..., 1.9797e+00,
2.9025e+00, 2.2799e+00],
...,
[ 3.0953e+00, 2.6005e+00, 2.3877e+00, ..., 1.7425e+00,
1.3947e+00, 1.7412e-01],
[ 4.7810e+00, 4.6805e+00, 4.0356e+00, ..., 2.9745e+00,
1.9197e+00, 1.6020e+00],
[ 2.8440e+00, 2.7687e+00, 1.9957e+00, ..., 5.7254e-01,
5.2793e-01, 1.0087e+00]]],
[[[-2.3336e-01, -8.1936e-01, -1.2560e+00, ..., -6.6420e-02,
-2.1152e-01, 9.3313e-01],
[ 3.3271e-01, -1.4179e+00, -1.7101e+00, ..., -8.4712e-01,
-2.2529e+00, 8.1617e-01],
[-8.4266e-01, 3.5365e-01, -4.0943e-01, ..., -1.7270e+00,
-1.3121e+00, 3.6609e-02],
...,
[-9.4333e-01, -1.9356e+00, -3.8511e-01, ..., -6.8304e+00,
-3.5629e-01, 3.5408e+00],
[-1.1363e+00, -1.0329e+00, -3.6974e-01, ..., -3.0675e-01,
2.4936e+00, 7.1424e-01],
[-2.7947e+00, -5.3016e+00, -4.6477e+00, ..., -6.1374e+00,
-7.2751e+00, -4.1087e+00]],
[[-1.9086e+00, -4.0942e+00, -5.2349e+00, ..., -4.1918e+00,
-4.8423e+00, -1.9198e-01],
[-5.5858e+00, -6.6011e+00, -7.6073e+00, ..., -9.1324e+00,
-8.7538e+00, -1.7210e+00],
[-4.4959e+00, -5.0864e+00, -7.7357e+00, ..., -3.6563e+00,
-5.9288e+00, -2.2627e+00],
...,
[-6.0323e+00, -6.0329e+00, -3.8694e+00, ..., -4.8661e-01,
-2.1818e+00, 2.9193e-01],
[-7.2621e+00, -8.1144e+00, -6.1521e+00, ..., -5.0573e+00,
-7.6755e+00, -7.4932e+00],
[-2.5887e+00, -4.6289e+00, -4.0731e+00, ..., -6.9626e+00,
-5.9725e+00, -2.8381e+00]],
[[-2.8690e-01, 2.1196e+00, 6.6782e-01, ..., 8.6097e-01,
1.7420e+00, 8.7398e-01],
[ 5.1358e-01, 1.3034e+00, -1.8648e+00, ..., 6.2367e-01,
7.1200e-01, -8.3106e-01],
[-1.1101e-02, 2.1113e+00, 2.7708e+00, ..., 8.8199e-01,
1.5747e+00, -3.0277e-02],
...,
[ 8.8333e-01, 1.2082e+00, 7.1940e-01, ..., 4.2247e+00,
1.6713e+00, -4.8405e-01],
[ 8.8113e-01, 7.6807e-01, 2.1983e-01, ..., -7.3810e-01,
-1.0901e+00, -2.1546e+00],
[ 1.4201e+00, 1.5781e+00, 9.1013e-01, ..., -1.1007e+00,
-6.7381e-01, -2.1450e+00]],
...,
[[-1.9583e+00, -3.2941e+00, -3.2569e+00, ..., -1.6448e+00,
-2.7556e+00, -2.8440e-01],
[-4.2396e+00, -4.5577e+00, -3.1371e+00, ..., -3.7954e+00,
-5.2473e+00, -2.7204e+00],
[-2.5077e+00, -3.2473e+00, -3.7027e+00, ..., -2.0534e+00,
-3.7725e+00, -2.2442e+00],
...,
[-2.7251e+00, -2.8085e+00, -7.6831e-01, ..., -2.8959e+00,
-4.2321e+00, -2.7751e+00],
[-2.9740e+00, -3.0273e+00, -1.7071e+00, ..., -2.3510e+00,
-1.9249e+00, -1.3534e+00],
[-3.6028e+00, -3.9973e+00, -3.4124e+00, ..., -9.6213e+00,
-7.4848e+00, -3.9996e+00]],
[[-2.0063e+00, -2.0762e+00, -1.1454e+00, ..., -1.5852e+00,
-1.0900e+00, -1.3399e+00],
[-5.6414e+00, -3.1317e+00, 9.0822e-01, ..., -3.2144e+00,
-1.8598e+00, -1.1226e+00],
[-4.8345e+00, -3.9439e+00, -3.9842e+00, ..., 6.3189e-01,
-6.7207e-01, -4.6041e-01],
...,
[-4.0604e+00, -2.5422e+00, -2.6974e+00, ..., -1.3889e+01,
-9.3671e+00, -9.7808e-01],
[-3.7447e+00, -8.2124e-01, -1.4828e+00, ..., -2.2765e+00,
-6.2884e+00, -4.9284e-01],
[-4.5488e+00, -3.0781e+00, -2.2248e+00, ..., 3.1206e-01,
-3.7928e+00, 4.2182e-01]],
[[ 1.7115e+00, 5.1733e-01, -5.1353e-01, ..., -5.6235e-01,
-2.7884e-01, -4.1478e-01],
[ 4.0430e+00, 2.6980e+00, 3.1351e+00, ..., 2.0713e+00,
1.8887e+00, 2.0103e+00],
[ 1.8908e+00, 9.6291e-01, 3.3308e+00, ..., 4.1681e-01,
8.8751e-01, 3.3011e-01],
...,
[ 3.0601e+00, 1.2436e+00, 1.3720e+00, ..., -1.4465e+01,
-9.2636e+00, 1.2272e+00],
[ 3.8016e+00, 3.6680e+00, 1.9951e+00, ..., -2.7270e-02,
6.9975e-01, 4.6249e+00],
[ 1.9311e+00, 7.4659e-01, -1.3478e-01, ..., -8.4923e-01,
-1.2707e+00, 6.6881e-01]]],
[[[-1.4283e-01, 7.6527e-01, -1.0303e-01, ..., -1.2379e+00,
-1.0229e+00, -7.7030e-01],
[ 8.0954e-01, 9.1820e-01, 5.9664e-01, ..., -5.0052e-01,
3.7440e-02, -1.0057e+00],
[ 1.3103e+00, 2.0289e-01, -9.6454e-01, ..., -5.6209e-01,
-4.2515e-01, -1.9771e+00],
...,
[-7.8265e-01, -4.2902e-01, 1.0584e+00, ..., -1.5807e+00,
-2.4953e+00, -1.6921e-02],
[-2.1886e+00, -9.3905e-01, 5.9806e-01, ..., -2.5400e+00,
-2.9844e+00, 8.2435e-02],
[-1.6574e+00, -3.0320e+00, -2.7822e+00, ..., -4.3253e+00,
-3.7965e+00, -1.7032e+00]],
[[-7.1880e-01, -1.4562e+00, -2.5875e+00, ..., -2.9096e+00,
-4.5290e+00, -2.3138e+00],
[-1.9070e+00, -3.8119e+00, -4.4451e+00, ..., -3.2432e+00,
-5.0148e+00, -4.5007e+00],
[-1.7094e+00, -3.5933e+00, -5.2923e+00, ..., -2.4205e+00,
-4.0145e+00, -3.8046e+00],
...,
[-5.7696e+00, -7.1613e+00, -6.3735e+00, ..., -7.7674e+00,
-8.6855e+00, -1.9502e+00],
[-8.9070e+00, -9.3988e+00, -8.4454e+00, ..., -1.0749e+01,
-1.1080e+01, -3.9168e+00],
[-1.9855e+00, -4.1591e+00, -3.8251e+00, ..., -4.6252e+00,
-4.5263e+00, -7.2582e-01]],
[[ 6.8593e-01, 1.2684e+00, 1.0997e+00, ..., 1.4774e+00,
1.7363e+00, 8.7488e-01],
[ 8.8301e-01, 1.5357e+00, 1.2158e+00, ..., 1.2943e+00,
2.1475e+00, 8.3478e-01],
[ 5.3158e-01, 3.0497e+00, 2.2045e+00, ..., 1.1454e+00,
2.4688e+00, 1.9236e+00],
...,
[ 4.1154e-02, 3.3930e-01, 1.3209e-01, ..., 1.5592e+00,
1.1976e+00, -3.7214e-01],
[-7.6964e-01, 5.2743e-01, -3.6902e-02, ..., 8.6902e-01,
1.1524e+00, -4.7088e-01],
[ 8.1995e-02, 5.6318e-01, 7.8964e-01, ..., 2.0727e-01,
-3.2350e-01, -1.8843e+00]],
...,
[[-9.8697e-01, -1.2368e+00, -1.9888e+00, ..., -2.4816e+00,
-3.1710e+00, -1.1919e+00],
[-2.1989e+00, -2.9509e+00, -2.3453e+00, ..., -2.8432e+00,
-4.2988e+00, -3.4407e+00],
[-7.6945e-01, -2.1254e+00, -2.1112e+00, ..., -2.0895e+00,
-4.1275e+00, -3.0870e+00],
...,
[-2.1750e+00, -3.6307e+00, -7.7835e-01, ..., -4.9850e+00,
-6.1489e+00, -2.4808e+00],
[-3.0416e+00, -5.4679e+00, -2.5284e+00, ..., -4.2069e+00,
-5.2713e+00, -1.2701e+00],
[-2.3262e+00, -3.1032e+00, -2.6616e+00, ..., -2.8850e+00,
-2.6186e+00, 3.2945e-01]],
[[-1.5206e+00, -2.3066e+00, -3.4649e+00, ..., -2.7904e+00,
-2.6854e+00, -5.4203e-01],
[-4.2797e+00, -4.8458e+00, -5.1866e+00, ..., -4.1734e+00,
-5.7159e+00, -1.8832e+00],
[-3.9749e+00, -5.7303e+00, -5.4230e+00, ..., -2.8472e+00,
-4.7481e+00, -2.1298e+00],
...,
[-4.2567e+00, -2.6205e+00, -2.9990e+00, ..., -4.6371e+00,
-3.4421e+00, -1.5109e+00],
[-3.3257e+00, -1.0177e+00, -1.6314e+00, ..., -2.4809e+00,
-1.5162e+00, 9.3576e-02],
[-3.0759e+00, -2.6844e+00, -2.8998e+00, ..., -1.9214e+00,
-1.2192e+00, 2.7416e-01]],
[[ 9.0928e-01, -1.4292e-01, 8.0558e-01, ..., 5.1520e-01,
1.3066e+00, 5.6296e-01],
[ 1.7400e+00, 9.7971e-01, 1.1994e+00, ..., 7.2708e-01,
1.8544e+00, 3.1749e+00],
[ 1.3758e+00, 4.8696e-01, 1.3240e+00, ..., 4.5631e-01,
9.6645e-01, 8.8055e-01],
...,
[ 3.5841e+00, 3.5375e+00, 2.6906e+00, ..., 3.1900e+00,
2.5379e+00, 4.9903e-01],
[ 3.9977e+00, 4.3581e+00, 3.5797e+00, ..., 2.9409e+00,
1.7580e+00, 1.3825e+00],
[ 2.6382e+00, 2.5853e+00, 1.8630e+00, ..., 3.7808e-01,
6.0977e-01, 1.0220e+00]]],
[[[-2.6231e-01, 3.9418e-01, -6.0144e-01, ..., -1.4675e+00,
-1.1579e+00, -9.3891e-01],
[ 8.1641e-01, 7.7651e-01, 2.1383e+00, ..., -9.0454e-01,
2.2775e-01, -1.1638e+00],
[ 2.6270e+00, 1.7048e+00, 1.5926e+00, ..., -4.0889e-01,
-1.2971e-01, -1.7327e+00],
...,
[-1.3207e+00, -1.1054e+00, -2.1877e-02, ..., -2.6784e+00,
-3.5762e+00, -1.2983e+00],
[-2.1732e+00, -1.1840e+00, -1.7791e+00, ..., -2.7141e-01,
1.6855e+00, 5.8771e-01],
[-2.1065e+00, -4.0803e+00, -3.2950e+00, ..., 3.1842e-02,
-8.3149e-01, -1.0644e-02]],
[[-9.2188e-01, -1.7022e+00, -2.6956e+00, ..., -3.1618e+00,
-4.8236e+00, -2.6424e+00],
[-1.9576e+00, -4.0069e+00, -5.4452e+00, ..., -4.5570e+00,
-5.1626e+00, -4.7905e+00],
[-2.4665e+00, -6.1279e+00, -5.6306e+00, ..., -3.6587e+00,
-4.2762e+00, -3.6811e+00],
...,
[-6.9057e+00, -7.4163e+00, -6.9675e+00, ..., -1.2711e+01,
-1.0719e+01, -1.8396e+00],
[-9.4136e+00, -9.5999e+00, -7.3643e+00, ..., 1.4008e+00,
1.0168e+00, 4.3334e-01],
[-2.3144e+00, -4.3225e+00, -3.3533e+00, ..., 1.7093e-01,
-1.8713e+00, -1.1673e+00]],
[[ 8.0463e-01, 1.3117e+00, 1.1881e+00, ..., 1.4530e+00,
1.9682e+00, 8.8573e-01],
[ 1.0870e+00, 1.9515e+00, 1.6872e+00, ..., 1.2458e+00,
2.3916e+00, 1.0243e+00],
[ 3.5430e-01, 2.7882e+00, 2.2404e-01, ..., 1.2189e+00,
2.2984e+00, 2.0928e+00],
...,
[-2.2707e-02, 2.9883e-01, -3.9964e-01, ..., -1.6723e+01,
-1.2245e+01, -2.1471e+00],
[ 5.5766e-02, -4.5098e-01, -9.9640e-01, ..., -9.5939e+00,
-6.9775e+00, -8.5818e-01],
[ 7.8139e-01, 2.9959e-01, -5.2694e-01, ..., -7.1927e+00,
-9.5490e+00, -6.0632e+00]],
...,
[[-1.1562e+00, -1.4140e+00, -2.0549e+00, ..., -2.7901e+00,
-3.4791e+00, -1.3005e+00],
[-2.3027e+00, -3.2269e+00, -2.5228e+00, ..., -3.6244e+00,
-4.6493e+00, -3.5792e+00],
[-6.2524e-01, -2.6887e+00, -3.4929e+00, ..., -1.9577e+00,
-3.9994e+00, -2.7926e+00],
...,
[-1.7194e+00, -8.6764e-01, -8.3042e-01, ..., -2.2865e+00,
-2.9067e+00, -3.5598e+00],
[-3.0089e+00, -3.0995e+00, -6.9063e-01, ..., 2.4073e+00,
1.8066e+00, -3.2343e-01],
[-2.4359e+00, -2.2853e+00, -1.3751e+00, ..., -6.3618e-01,
-1.6110e+00, -1.2369e+00]],
[[-1.7801e+00, -2.3588e+00, -3.5254e+00, ..., -3.0382e+00,
-2.8541e+00, -4.3023e-01],
[-4.4398e+00, -3.7321e+00, -3.7535e+00, ..., -4.1481e+00,
-5.5698e+00, -1.7023e+00],
[-4.0442e+00, -3.3369e+00, -2.1070e+00, ..., -1.8288e+00,
-3.8618e+00, -1.5156e+00],
...,
[-3.0913e+00, -1.7201e+00, -3.4653e+00, ..., 2.1982e+00,
2.2625e+00, 2.9679e+00],
[-3.3097e+00, 8.5705e-01, -4.7772e-01, ..., -6.4992e-01,
1.5896e+00, 1.6183e+00],
[-3.8207e+00, -1.4331e+00, -9.8866e-01, ..., 8.0536e-01,
1.0261e+00, 8.2949e-01]],
[[ 8.9879e-01, -1.0176e-01, 6.1067e-01, ..., 4.3061e-01,
1.4112e+00, 6.5806e-01],
[ 1.7980e+00, 1.1932e+00, 1.6053e+00, ..., 7.5452e-01,
2.1262e+00, 3.4174e+00],
[ 1.9744e+00, 1.6258e+00, 1.9039e+00, ..., 3.6978e-01,
1.1026e+00, 9.8321e-01],
...,
[ 3.2397e+00, 2.1934e+00, 1.2844e+00, ..., 3.3250e-01,
3.1031e+00, 4.0986e+00],
[ 4.8488e+00, 4.8708e+00, 2.4006e+00, ..., 1.2992e-01,
3.5731e+00, 3.9280e+00],
[ 2.4458e+00, 2.0741e+00, 4.4478e-01, ..., 8.6177e-01,
1.0089e+00, 2.0224e+00]]]], device='cuda:0'),
tensor([[[[ 3.7962e-02, 1.5019e+00, 1.1457e+00, ..., -1.9229e+00,
-6.6162e-01, -7.7862e-02],
[ 4.0194e-01, 9.2968e-01, 4.8735e-01, ..., -4.7780e+00,
-4.0364e+00, -2.0764e+00],
[-3.3861e-01, 8.8277e-01, -8.7891e-01, ..., 5.8955e-01,
2.2232e+00, 2.2142e+00],
...,
[-5.5780e+00, -3.3948e+00, -3.0964e+00, ..., -2.0094e-05,
1.4878e+00, 1.2083e+00],
[-5.4900e+00, 3.8786e-01, -1.7274e-01, ..., -2.3187e-01,
1.9015e+00, 1.1197e+00],
[-4.5478e+00, -6.2133e+00, -2.8800e+00, ..., -2.9997e+00,
-2.0176e+00, -2.3465e+00]],
[[-8.0081e-01, -2.1806e+00, -4.6244e+00, ..., -2.9609e+00,
-4.4067e+00, -4.5308e-01],
[-4.2775e+00, -5.7873e+00, -6.7513e+00, ..., -1.2131e+01,
-1.1102e+01, -4.7219e+00],
[-3.3544e+00, -4.3793e+00, -6.0100e+00, ..., 3.9355e+00,
3.4627e+00, 2.3172e+00],
...,
[-1.0078e+01, -1.5407e+00, 2.4533e+00, ..., -2.3302e+00,
-2.0285e+00, -9.1460e-01],
[-9.3716e+00, -8.6176e+00, 6.3452e-01, ..., -4.2351e+00,
-3.7663e+00, -3.6503e+00],
[-1.9716e+00, -6.7360e+00, -8.0192e+00, ..., -2.8371e+00,
-2.1986e+00, -1.7176e+00]],
[[-2.5979e-01, 2.3813e+00, 1.3339e+00, ..., 9.3590e-01,
1.9715e+00, 4.3144e-01],
[-2.0779e-01, 1.3661e+00, 8.7801e-01, ..., 4.1630e+00,
5.3295e+00, 2.6047e+00],
[ 6.2122e-02, 1.8823e+00, 8.9999e-01, ..., -6.3406e+00,
-4.2523e+00, -1.9423e+00],
...,
[-3.4492e-01, -3.1799e+00, 2.6836e+00, ..., -5.0786e-01,
-9.7991e-01, 9.4266e-01],
[ 1.1415e-01, -6.7009e+00, -1.4367e+00, ..., 1.7398e-02,
-8.6696e-01, 6.7509e-01],
[ 5.0123e-01, -4.5973e-01, -8.9644e-01, ..., 1.2779e+00,
8.5664e-02, 1.6541e-01]],
...,
[[-8.4333e-01, -1.6772e+00, -1.8324e+00, ..., -1.8706e+00,
-2.8954e+00, 3.3713e-01],
[-2.8523e+00, -3.7486e+00, -3.1034e+00, ..., -3.5893e+00,
-2.9840e+00, -1.7928e+00],
[-1.0664e+00, -2.2982e+00, -1.9778e+00, ..., 5.2233e+00,
4.9173e+00, 3.5221e+00],
...,
[-5.9163e+00, -1.7292e+00, -2.1536e+00, ..., -9.5286e-01,
-2.0827e+00, -2.5378e+00],
[-4.8594e+00, -2.5689e+00, -3.0238e+00, ..., -2.0361e+00,
-1.3359e+00, -1.7802e+00],
[-3.1172e+00, -6.4508e+00, -7.0512e+00, ..., -1.9143e+00,
-2.0603e+00, -1.6715e+00]],
[[-1.6563e+00, -2.1643e+00, -2.7933e+00, ..., -6.8071e-01,
-5.0190e-01, 1.5090e-02],
[-3.8281e+00, -3.5897e+00, -4.1947e+00, ..., -3.9409e+00,
-2.9951e+00, -4.1723e-01],
[-3.0373e+00, -3.2677e+00, -3.9848e+00, ..., 3.9826e+00,
1.9446e+00, 1.1625e+00],
...,
[-4.4862e-01, 2.6494e-05, -9.4918e+00, ..., 1.9399e-01,
-1.4209e+00, -2.3108e+00],
[-2.2131e+00, 7.5726e+00, -2.5135e+00, ..., -8.7872e-01,
-1.8095e+00, -2.1665e+00],
[-4.4714e+00, 4.6501e-01, -1.4906e+00, ..., -2.1552e+00,
-2.5879e+00, -1.2169e+00]],
[[ 1.0170e+00, 6.1703e-01, 1.0810e+00, ..., -8.5548e-01,
-2.2761e-01, -6.8240e-01],
[ 3.3696e+00, 3.5041e+00, 3.1495e+00, ..., 5.3975e+00,
4.4607e+00, 4.7310e+00],
[ 1.2913e+00, 1.4163e+00, 9.4337e-01, ..., -2.7488e-01,
4.0565e-01, 9.7070e-02],
...,
[ 3.4173e+00, -2.6833e+00, -8.2998e+00, ..., 3.8816e-01,
8.8490e-01, 1.5629e+00],
[ 2.0294e+00, 7.1666e+00, -2.9373e+00, ..., -3.1518e-01,
1.1904e+00, 1.5138e+00],
[-3.8270e-01, 3.9175e+00, 2.2052e+00, ..., -1.0489e+00,
-6.2960e-02, -1.7836e-01]]],
[[[ 5.0302e-02, 1.4050e+00, 9.8046e-01, ..., -1.8941e-02,
3.6525e-01, -1.8939e-01],
[ 4.8819e-01, 1.1475e+00, 2.1613e+00, ..., 1.8161e-01,
4.0358e-01, -5.5657e-01],
[-3.6578e-01, 6.8867e-01, -4.0251e-01, ..., -5.7073e-01,
-1.4764e-01, -1.1300e+00],
...,
[ 2.9492e-01, 3.8078e-01, 8.3186e-01, ..., -9.1555e-02,
5.0159e-01, -2.8489e-01],
[-1.8394e+00, 2.5660e-01, 4.0503e-01, ..., 1.9022e-01,
3.2519e-01, -1.1127e+00],
[-1.9086e+00, -3.2831e+00, -2.1296e+00, ..., -3.1748e+00,
-2.8592e+00, -2.7147e+00]],
[[-5.1540e-01, -1.6172e+00, -3.7662e+00, ..., -2.3998e+00,
-3.7044e+00, -2.2016e+00],
[-3.5925e+00, -4.0108e+00, -5.2659e+00, ..., -3.0159e+00,
-4.4748e+00, -3.6975e+00],
[-2.8693e+00, -4.1436e+00, -7.9242e+00, ..., -1.6817e+00,
-3.0479e+00, -3.0473e+00],
...,
[-5.5570e+00, -6.7120e+00, -6.9247e+00, ..., -3.0149e+00,
-3.1135e+00, -2.5559e+00],
[-8.3784e+00, -8.0408e+00, -6.7877e+00, ..., -4.7370e+00,
-5.0467e+00, -4.9146e+00],
[-2.2093e+00, -4.0475e+00, -2.7102e+00, ..., -3.0023e+00,
-2.7939e+00, -1.5944e+00]],
[[-2.4468e-01, 2.5012e+00, 1.6997e+00, ..., 1.5601e+00,
1.7188e+00, 8.7856e-01],
[-6.0588e-01, 6.1208e-01, 5.2654e-01, ..., 9.4126e-01,
1.7353e+00, 8.0194e-01],
[-4.6976e-01, 1.1747e+00, 1.0769e+00, ..., 2.4394e+00,
2.1592e+00, 1.3340e+00],
...,
[-9.7126e-01, 1.0192e+00, 8.7659e-01, ..., 1.6061e+00,
1.1787e+00, 6.9541e-01],
[-1.3136e+00, -5.0472e-01, 9.2419e-01, ..., 1.4480e+00,
9.1415e-01, 3.8994e-01],
[ 1.3791e-01, -5.4499e-01, -4.1569e-02, ..., 1.6812e+00,
1.5923e+00, 5.6077e-01]],
...,
[[-7.8570e-01, -1.7207e+00, -1.8313e+00, ..., -1.6231e+00,
-2.2627e+00, -4.5212e-01],
[-2.6260e+00, -3.5652e+00, -2.0129e+00, ..., -2.2950e+00,
-3.9443e+00, -3.0770e+00],
[-1.5944e+00, -2.6098e+00, -1.9866e+00, ..., -1.7636e+00,
-3.1085e+00, -2.0535e+00],
...,
[-1.5955e+00, -3.1410e+00, -8.6289e-01, ..., -1.5412e+00,
-2.0678e+00, -8.9397e-01],
[-2.5136e+00, -3.1287e+00, -1.2899e+00, ..., -1.4171e+00,
-2.7177e+00, -1.5190e+00],
[-2.2862e+00, -1.9993e+00, -1.5469e+00, ..., -1.9111e+00,
-2.1933e+00, -5.8741e-01]],
[[-1.4213e+00, -2.2019e+00, -2.2249e+00, ..., -6.9296e-01,
-1.4559e+00, -6.1639e-01],
[-3.3064e+00, -2.8454e+00, -2.7101e+00, ..., -1.0708e+00,
-3.7722e+00, -1.8488e+00],
[-2.3965e+00, -2.5641e+00, -4.1630e+00, ..., -4.6816e-01,
-3.2998e+00, -2.2977e+00],
...,
[-2.4538e+00, -2.8641e+00, -4.5218e+00, ..., -8.8533e-01,
-3.1544e+00, -2.0431e+00],
[-2.1671e+00, 5.9856e-01, -1.7276e+00, ..., -8.3865e-01,
-3.3079e+00, -1.9077e+00],
[-3.0123e+00, -1.6898e+00, -1.7423e+00, ..., -1.4479e+00,
-2.6811e+00, -1.2303e+00]],
[[ 7.6472e-01, 2.0304e-01, 8.5960e-01, ..., 2.6677e-01,
1.2733e+00, 6.1941e-01],
[ 3.3341e+00, 2.5370e+00, 3.1084e+00, ..., 5.4860e-01,
9.9964e-01, 2.4651e+00],
[ 1.2374e+00, 1.0541e+00, 2.0601e+00, ..., 1.4137e+00,
7.4342e-01, 6.8224e-01],
...,
[ 3.6701e+00, 3.1947e+00, 3.4684e+00, ..., 6.4211e-01,
1.3984e-01, 5.5511e-01],
[ 4.1070e+00, 4.1405e+00, 3.1005e+00, ..., 1.1428e+00,
8.0420e-01, 1.2063e+00],
[ 2.8030e+00, 2.8620e+00, 2.0247e+00, ..., -6.0638e-02,
-3.0899e-01, -1.3171e-01]]],
[[[ 8.4231e-03, 1.2111e+00, 1.3441e+00, ..., -3.7588e+00,
-1.3571e+00, 3.9155e-01],
[ 8.7052e-01, 1.9079e+00, 3.6380e+00, ..., -8.2488e+00,
-1.7507e+00, 1.8849e-01],
[ 1.3356e+00, 1.8850e+00, 1.0890e-01, ..., 1.4346e+00,
2.2910e+00, 7.9506e-01],
...,
[-8.4783e-01, -1.4877e+00, -4.6207e-02, ..., -3.9700e+00,
4.2982e-01, -6.7750e-01],
[-1.0413e+00, -9.7646e-01, -1.1293e-01, ..., -8.0044e+00,
-7.3338e+00, -8.1501e-01],
[-2.8511e+00, -5.4323e+00, -4.3715e+00, ..., -6.1960e+00,
-2.2298e+00, 2.3674e+00]],
[[-5.2146e-01, -1.4707e+00, -4.0014e+00, ..., -4.4094e+00,
-5.2141e+00, -5.4574e-01],
[-3.5334e+00, -6.6009e+00, -8.2687e+00, ..., -1.0486e+01,
-5.8535e+00, -1.5816e+00],
[-2.9401e+00, -5.8032e+00, -1.5213e+01, ..., 1.6426e+00,
-2.4266e+00, -1.5606e+00],
...,
[-6.2365e+00, -6.5375e+00, -4.6509e+00, ..., 5.0580e-02,
1.3633e+00, 1.2659e+00],
[-7.4729e+00, -8.3604e+00, -6.2326e+00, ..., -5.1195e+00,
-8.9742e+00, -2.0168e+00],
[-2.6499e+00, -4.8725e+00, -4.1936e+00, ..., -2.9881e+00,
-1.2610e+00, 1.8801e+00]],
[[-4.7111e-01, 2.0692e+00, 1.1612e+00, ..., 4.7542e-01,
2.4470e+00, 1.1371e+00],
[-2.4407e-01, 1.3212e+00, 1.0758e-02, ..., 7.1943e+00,
4.2371e+00, -2.5837e-01],
[ 2.0463e-01, 1.8422e+00, -4.0799e-01, ..., 1.0436e+00,
5.3868e+00, 2.5153e+00],
...,
[ 1.0619e+00, 1.5158e+00, 9.4683e-01, ..., -9.2767e+00,
-9.5425e+00, -1.5698e+00],
[ 7.7811e-01, 9.0146e-01, 5.5180e-01, ..., -4.7512e+00,
-9.1614e+00, -6.6333e+00],
[ 1.4055e+00, 1.7061e+00, 1.1722e+00, ..., -1.7193e+00,
-1.4777e+00, -1.3634e+00]],
...,
[[-5.4991e-01, -1.0265e+00, -2.4478e-01, ..., -1.5435e+00,
-2.6186e+00, -1.4312e+00],
[-2.7229e+00, -3.3915e+00, -2.7945e+00, ..., -6.8957e+00,
-5.3978e+00, -4.4503e+00],
[-2.9527e-01, -2.8781e+00, -6.2321e+00, ..., 2.4696e+00,
-1.2412e+00, -2.0864e+00],
...,
[-2.9973e+00, -3.1336e+00, -1.3200e+00, ..., 8.9930e-01,
-3.3183e+00, -1.6715e+00],
[-2.9747e+00, -3.1120e+00, -1.7569e+00, ..., -3.9248e+00,
-6.1278e+00, -3.2588e+00],
[-3.7698e+00, -4.1975e+00, -3.2645e+00, ..., -4.0308e+00,
-3.2559e+00, -1.7657e+00]],
[[-1.4942e+00, -1.7066e+00, -2.0004e+00, ..., -1.1714e-01,
-1.5810e+00, -7.2025e-01],
[-3.5618e+00, -3.0123e+00, -2.0214e+00, ..., -1.5678e+00,
2.6566e-01, -3.0959e-01],
[-2.9639e+00, -4.1438e+00, -8.5872e+00, ..., 4.4405e+00,
3.2652e-01, 7.4929e-01],
...,
[-3.9137e+00, -2.1094e+00, -1.7494e+00, ..., -1.5396e+00,
-1.0741e+00, 1.2175e+00],
[-3.8088e+00, -4.8492e-01, -1.9456e+00, ..., -4.3105e+00,
2.0685e-02, 2.4235e+00],
[-4.7345e+00, -2.9861e+00, -2.7332e+00, ..., -3.6390e+00,
1.8136e-01, 1.7924e+00]],
[[ 7.9117e-01, 1.6038e-01, 4.1504e-01, ..., -1.7320e+00,
-1.1548e+00, -1.1545e+00],
[ 2.8842e+00, 3.1614e+00, 3.6917e+00, ..., 4.1666e+00,
4.1927e+00, 1.9583e+00],
[ 1.1072e+00, 1.7828e+00, 3.1681e+00, ..., 2.6244e+00,
4.0161e+00, 1.9926e+00],
...,
[ 3.1207e+00, 1.5015e+00, 1.0659e+00, ..., -2.3674e+00,
-1.8901e+00, -5.8226e-01],
[ 4.0458e+00, 3.7738e+00, 2.1802e+00, ..., -7.7207e+00,
-6.7528e+00, -1.8202e+00],
[ 2.1153e+00, 8.2007e-01, 8.8389e-02, ..., -6.6619e+00,
-4.1832e+00, 1.3119e+00]]],
[[[-3.8900e-01, 1.0415e+00, 1.8573e+00, ..., -2.2453e+00,
-8.5881e-01, -1.1830e-01],
[ 1.1171e-02, 3.9406e-01, 3.1716e+00, ..., 1.7375e+00,
-7.4075e-01, -1.0445e+00],
[-1.1720e+00, 6.7849e-01, 1.7643e+00, ..., 1.0909e+00,
-2.2454e+00, -2.6043e+00],
...,
[-3.8023e-01, -1.7462e-01, 6.2736e-01, ..., -1.4965e+00,
-1.1299e+00, -5.1290e-01],
[-1.4365e+00, -2.7958e-01, 1.1888e-01, ..., 3.3841e-01,
-5.5852e-03, -1.2775e+00],
[-1.1873e+00, -2.2340e+00, -1.6027e+00, ..., -3.2426e+00,
-2.8640e+00, -2.7774e+00]],
[[-1.6023e+00, -1.9076e+00, -2.9791e+00, ..., -6.4459e+00,
-5.5657e+00, -2.0575e+00],
[-4.5850e+00, -4.9110e+00, -4.5217e+00, ..., -4.5707e+00,
-9.2147e+00, -4.1326e+00],
[-3.6783e+00, -5.2161e+00, -5.7763e+00, ..., -5.3780e+00,
-1.0305e+01, -3.8650e+00],
...,
[-4.4323e+00, -5.7145e+00, -4.7064e+00, ..., -6.3833e+00,
-3.4063e+00, -2.9911e+00],
[-7.6513e+00, -8.0811e+00, -6.6089e+00, ..., -4.1019e+00,
-5.0432e+00, -5.1712e+00],
[-1.3898e+00, -3.0844e+00, -2.2614e+00, ..., -2.7032e+00,
-2.9608e+00, -1.6332e+00]],
[[-2.4251e-01, 2.0695e+00, 1.7938e+00, ..., 1.8491e+00,
1.5357e+00, 8.9982e-01],
[-5.4180e-01, 1.8159e-01, 8.9696e-01, ..., 4.9451e+00,
1.4909e+00, 3.3412e-01],
[-1.3724e-01, 2.3339e+00, 1.4503e+00, ..., 5.4579e+00,
1.6468e+00, 9.9694e-01],
...,
[-7.3066e-01, 6.6891e-01, -4.4932e-01, ..., 1.9555e+00,
8.1796e-01, 9.1397e-01],
[-1.3383e+00, 3.4418e-01, 8.6802e-01, ..., 3.4424e-02,
1.1623e+00, 5.8989e-01],
[-4.6703e-01, -3.4215e-01, 1.7006e-01, ..., 4.8902e-01,
1.9392e+00, 5.5195e-01]],
...,
[[-1.4290e+00, -1.5898e+00, -1.3139e-01, ..., -3.3501e+00,
-4.5538e+00, -6.9204e-01],
[-3.5515e+00, -3.3473e+00, -1.3409e+00, ..., -3.0544e+00,
-5.5239e+00, -4.6010e+00],
[-2.6402e+00, -2.3129e+00, -1.3613e+00, ..., -8.4627e+00,
-3.5093e+00, -4.4720e+00],
...,
[-1.0880e+00, -2.8035e+00, -1.9672e+00, ..., -1.2954e+00,
-3.4488e+00, -1.5085e+00],
[-2.0991e+00, -4.3656e+00, -2.0218e+00, ..., -8.6710e-01,
-3.8054e+00, -1.7219e+00],
[-1.3098e+00, -1.5389e+00, -6.9430e-01, ..., -2.6537e+00,
-2.8876e+00, -5.8926e-01]],
[[-1.8588e+00, -1.3177e+00, -1.9114e+00, ..., 7.4612e-01,
-1.5324e+00, -8.2343e-01],
[-3.9914e+00, -1.4580e+00, -3.0197e+00, ..., -1.0400e+00,
-3.4002e+00, -2.0324e+00],
[-3.5898e+00, -2.7070e+00, -7.2793e+00, ..., -5.4083e+00,
-1.5989e+00, -2.2616e+00],
...,
[-2.7188e+00, -2.0501e+00, -5.4098e+00, ..., -2.4814e+00,
-3.1969e+00, -1.6400e+00],
[-1.9369e+00, 1.5848e-01, -2.4681e+00, ..., 6.2347e-02,
-2.7147e+00, -1.7116e+00],
[-2.0282e+00, -1.0817e+00, -1.2351e+00, ..., -8.7297e-01,
-2.2301e+00, -1.2137e+00]],
[[ 1.1479e+00, 2.9706e-01, 5.6344e-01, ..., 1.1866e+00,
9.1583e-01, 5.8220e-01],
[ 4.0114e+00, 2.7743e+00, 2.3100e+00, ..., 5.1976e+00,
2.0389e+00, 1.9855e+00],
[ 1.5395e+00, 7.8104e-01, 1.7764e+00, ..., 5.9653e+00,
7.0074e-01, 2.6847e-01],
...,
[ 2.9102e+00, 2.3675e+00, 2.8156e+00, ..., 1.3752e+00,
1.1966e+00, 3.9697e-01],
[ 3.7918e+00, 3.7680e+00, 2.8563e+00, ..., 2.1303e+00,
8.1427e-01, 1.1908e+00],
[ 2.3653e+00, 2.2608e+00, 2.0360e+00, ..., 1.6959e+00,
-2.6783e-01, -1.6350e-01]]],
[[[-8.3561e-02, 7.3236e-01, -2.2139e-01, ..., -1.2795e+00,
-8.4915e-01, -8.1933e-01],
[ 7.6665e-01, 8.7833e-01, 1.0116e+00, ..., 5.0498e-02,
3.3438e-01, -1.0820e+00],
[ 1.2350e+00, -4.3022e-02, 4.0727e-02, ..., -4.2378e-01,
-2.8079e-01, -1.9793e+00],
...,
[-5.5518e-01, -6.0083e-01, 6.6744e-01, ..., -1.5977e+00,
-2.7308e+00, -1.3701e-01],
[-2.0862e+00, -1.0499e+00, 9.3556e-01, ..., -2.2398e+00,
-3.0211e+00, 2.9203e-02],
[-1.6965e+00, -3.2231e+00, -2.9584e+00, ..., -3.9168e+00,
-3.7732e+00, -1.6576e+00]],
[[-6.7439e-01, -1.3854e+00, -2.5859e+00, ..., -2.8345e+00,
-4.3969e+00, -2.4202e+00],
[-1.8802e+00, -3.7847e+00, -4.2966e+00, ..., -2.7062e+00,
-4.7691e+00, -4.6264e+00],
[-1.7314e+00, -3.6353e+00, -4.1259e+00, ..., -2.3149e+00,
-3.9517e+00, -3.6950e+00],
...,
[-5.8268e+00, -7.8808e+00, -6.8805e+00, ..., -7.0191e+00,
-8.7575e+00, -1.9875e+00],
[-8.8437e+00, -9.6441e+00, -9.0914e+00, ..., -1.0823e+01,
-1.1217e+01, -3.9769e+00],
[-2.0208e+00, -4.2458e+00, -3.9536e+00, ..., -4.3769e+00,
-4.3427e+00, -6.6145e-01]],
[[ 6.5621e-01, 1.1985e+00, 1.1573e+00, ..., 1.6862e+00,
1.8915e+00, 8.0924e-01],
[ 8.0413e-01, 1.4571e+00, 1.5046e+00, ..., 1.2344e+00,
2.3903e+00, 9.2074e-01],
[ 4.8293e-01, 3.0951e+00, 1.7721e+00, ..., 9.6690e-01,
2.5262e+00, 2.0570e+00],
...,
[-6.6636e-02, 7.9160e-01, -1.5488e-01, ..., 1.5779e+00,
1.3983e+00, -2.9660e-01],
[-8.2358e-01, 6.9559e-01, -1.3167e-01, ..., 1.0738e+00,
1.4748e+00, -3.7868e-01],
[ 1.0346e-01, 6.4009e-01, 9.5174e-01, ..., 2.9685e-01,
-3.0030e-01, -1.9034e+00]],
...,
[[-9.2417e-01, -1.2329e+00, -1.9805e+00, ..., -2.4521e+00,
-3.2203e+00, -1.1232e+00],
[-2.1682e+00, -2.8632e+00, -2.1287e+00, ..., -2.3479e+00,
-4.2915e+00, -3.4762e+00],
[-8.1051e-01, -1.7686e+00, -1.5990e+00, ..., -2.1136e+00,
-3.9538e+00, -3.0536e+00],
...,
[-2.2730e+00, -4.0660e+00, -1.2444e+00, ..., -4.8417e+00,
-6.4165e+00, -2.6941e+00],
[-2.9928e+00, -5.4676e+00, -2.1247e+00, ..., -4.7984e+00,
-5.4589e+00, -1.3711e+00],
[-2.4614e+00, -3.3967e+00, -2.8586e+00, ..., -2.8040e+00,
-2.5895e+00, 4.4365e-01]],
[[-1.4926e+00, -2.3168e+00, -3.5935e+00, ..., -2.8574e+00,
-2.6783e+00, -5.3732e-01],
[-4.2319e+00, -4.6335e+00, -5.3595e+00, ..., -3.9536e+00,
-5.8669e+00, -1.9962e+00],
[-3.9679e+00, -5.4149e+00, -5.6766e+00, ..., -2.9497e+00,
-4.8320e+00, -2.1759e+00],
...,
[-4.3571e+00, -2.8949e+00, -3.1363e+00, ..., -3.9345e+00,
-3.1486e+00, -1.5013e+00],
[-3.3531e+00, -1.0135e+00, -1.2311e+00, ..., -2.3912e+00,
-1.5298e+00, 6.6107e-02],
[-3.0985e+00, -2.7884e+00, -2.7050e+00, ..., -1.7487e+00,
-1.3101e+00, 3.2107e-01]],
[[ 8.7773e-01, -1.4442e-01, 6.5774e-01, ..., 4.9142e-01,
1.3571e+00, 5.6427e-01],
[ 1.6639e+00, 8.0718e-01, 8.7722e-01, ..., 9.0139e-01,
1.8848e+00, 3.1741e+00],
[ 1.4305e+00, 3.3068e-01, 1.2951e+00, ..., 3.7768e-01,
1.0582e+00, 8.0220e-01],
...,
[ 3.6174e+00, 3.4411e+00, 2.9659e+00, ..., 3.6299e+00,
2.6685e+00, 5.3779e-01],
[ 3.9976e+00, 4.2487e+00, 3.7166e+00, ..., 2.6977e+00,
1.7019e+00, 1.4008e+00],
[ 2.7257e+00, 2.5607e+00, 2.0460e+00, ..., 4.9476e-01,
5.9893e-01, 1.0269e+00]]]], device='cuda:0'),
tensor([[[[ 2.7419e-01, 1.4196e+00, 3.7891e-01, ..., -9.6944e-01,
-1.9563e-01, 5.7336e-01],
[ 5.0952e-01, 2.3304e-01, 4.7536e-01, ..., -1.3246e+00,
-9.2720e-01, 9.4936e-01],
[ 1.4208e-01, 1.6885e-01, 1.5792e+00, ..., 9.0350e-01,
1.3050e-01, 2.7674e-01],
...,
[ 3.7628e-01, -7.7498e-01, 1.1768e+00, ..., -1.0649e+00,
-8.9010e-01, -1.0629e+00],
[-1.3278e+00, -1.1033e+00, -8.2885e-01, ..., -1.7142e+00,
-2.6824e+00, -1.3035e+00],
[-1.6004e-01, -2.3947e+00, -2.2290e+00, ..., -2.1105e+00,
-2.1982e+00, -9.7283e-01]],
[[-7.8393e-01, -2.3319e+00, -4.1350e+00, ..., -3.9520e+00,
-5.0534e+00, -3.7375e-01],
[-4.7204e+00, -6.1672e+00, -5.0708e+00, ..., -4.7282e+00,
-7.4447e+00, -1.7857e+00],
[-3.6764e+00, -5.2360e+00, -6.3949e+00, ..., -4.0275e+00,
-8.1632e+00, -3.1224e+00],
...,
[-4.3739e+00, -5.3799e+00, -2.8324e+00, ..., -4.1091e+00,
-4.7558e+00, -2.3083e+00],
[-7.3749e+00, -8.9029e+00, -6.9898e+00, ..., -7.0978e+00,
-9.0060e+00, -4.0409e+00],
[-9.9946e-01, -3.1435e+00, -1.9548e+00, ..., -2.5061e+00,
-2.7410e+00, -3.4366e-01]],
[[-3.1812e-01, 2.1810e+00, 1.1615e+00, ..., 9.5647e-01,
2.1985e+00, 9.6321e-01],
[-7.8531e-02, 1.0846e+00, 8.2475e-01, ..., 1.4950e+00,
1.1838e+00, -5.8063e-01],
[ 2.8408e-02, 1.9299e+00, 1.7725e+00, ..., 2.6314e+00,
1.8335e+00, -3.9632e-01],
...,
[ 8.7235e-02, 1.2165e+00, 8.2747e-01, ..., 1.5326e+00,
1.4253e+00, 5.6623e-01],
[ 2.8609e-01, 7.4091e-01, -1.0038e-01, ..., 1.2762e+00,
2.4243e+00, 5.9075e-01],
[-5.9023e-01, 5.4988e-01, -3.1962e-01, ..., 1.0849e+00,
1.1771e+00, -7.6221e-01]],
...,
[[-8.3446e-01, -1.9277e+00, -1.7917e+00, ..., -2.8195e+00,
-3.0723e+00, -4.9021e-01],
[-3.1857e+00, -4.5245e+00, -2.4379e+00, ..., -2.6496e+00,
-4.3252e+00, -2.8079e+00],
[-1.5282e+00, -2.6511e+00, -1.1535e+00, ..., -2.0256e+00,
-4.0590e+00, -2.3154e+00],
...,
[-9.5495e-01, -1.3792e+00, -4.0086e-02, ..., -1.0996e+00,
-3.8563e+00, -1.7439e+00],
[-2.8190e+00, -3.4543e+00, -2.3400e+00, ..., -3.0081e+00,
-5.3272e+00, -1.4447e+00],
[-9.1209e-01, -1.4668e+00, -5.6682e-02, ..., -9.7321e-01,
-1.3357e+00, 1.1679e+00]],
[[-1.7334e+00, -2.2728e+00, -2.9099e+00, ..., -1.7252e+00,
-1.6676e+00, -1.3360e+00],
[-4.3621e+00, -3.6191e+00, -2.7412e+00, ..., -2.1675e+00,
-1.7662e+00, -1.0963e+00],
[-2.7582e+00, -1.8476e+00, -1.7003e+00, ..., -4.5883e-02,
-1.5151e-01, -6.5680e-01],
...,
[-3.7435e+00, -3.2484e+00, -4.5320e+00, ..., -1.6764e+00,
-2.5297e+00, -1.2106e+00],
[-4.2867e+00, -1.4575e+00, -2.6189e+00, ..., -9.2719e-01,
-2.2131e+00, -5.5189e-01],
[-3.0140e+00, -1.5649e+00, -7.3354e-01, ..., -2.8172e-01,
-7.0594e-01, 3.2496e-02]],
[[ 1.0842e+00, 5.9115e-01, 6.3882e-01, ..., -6.5356e-01,
2.7564e-01, -3.9587e-01],
[ 3.6006e+00, 3.3610e+00, 2.0254e+00, ..., 1.7311e+00,
2.1596e+00, 2.2588e+00],
[ 1.7672e+00, 1.3208e+00, 1.2431e+00, ..., 3.7370e+00,
2.8545e+00, 9.9592e-01],
...,
[ 2.7039e+00, 1.0262e+00, 1.7530e+00, ..., 2.0532e+00,
1.5654e+00, 4.8655e-01],
[ 3.3747e+00, 4.1392e+00, 3.0939e+00, ..., 3.2501e+00,
9.2948e-01, 1.6871e+00],
[ 2.2206e+00, 1.7228e+00, 6.1868e-01, ..., 7.2041e-02,
-5.7303e-02, 1.0191e+00]]],
[[[ 5.6550e-02, 1.7387e+00, 1.1421e+00, ..., 2.3159e-02,
4.2564e-01, -2.5835e-01],
[ 4.0216e-01, -1.0460e+00, -1.6025e+00, ..., 4.5022e-01,
7.2458e-01, -6.4859e-01],
[-1.2950e+00, -3.8881e+00, -2.2496e+00, ..., -3.8520e-01,
1.3193e-01, -1.0119e+00],
...,
[-4.7066e-01, -4.5616e-01, 4.9815e-01, ..., -3.4149e-01,
3.1832e-01, -1.6703e-01],
[-1.4943e+00, -9.2726e-02, 2.5197e-01, ..., 2.4780e-01,
6.6779e-01, -9.7633e-01],
[-1.2668e+00, -2.4119e+00, -1.5992e+00, ..., -3.0962e+00,
-2.8423e+00, -2.7286e+00]],
[[-1.1872e+00, -2.1193e+00, -3.6923e+00, ..., -2.0032e+00,
-3.7740e+00, -2.3610e+00],
[-4.8702e+00, -7.7857e+00, -6.1557e+00, ..., -2.0997e+00,
-3.9996e+00, -3.6965e+00],
[-4.4447e+00, -7.5066e+00, -3.2067e+00, ..., -2.9884e+00,
-2.4494e+00, -2.9715e+00],
...,
[-4.6222e+00, -5.1276e+00, -4.3663e+00, ..., -2.9667e+00,
-2.7644e+00, -2.3156e+00],
[-7.6193e+00, -7.7956e+00, -5.9830e+00, ..., -4.6367e+00,
-5.1213e+00, -4.8463e+00],
[-1.4129e+00, -3.2494e+00, -1.8016e+00, ..., -2.9443e+00,
-2.7874e+00, -1.6425e+00]],
[[-2.5817e-01, 2.1962e+00, 1.1597e+00, ..., 1.7717e+00,
1.8249e+00, 8.8852e-01],
[-8.8895e-01, -2.5820e-01, 1.4768e+00, ..., 1.0494e+00,
1.6161e+00, 8.4706e-01],
[-5.1324e-01, 1.8702e+00, 1.7957e+00, ..., 1.4589e+00,
1.9516e+00, 1.4728e+00],
...,
[-5.8156e-01, 6.0876e-01, 1.3300e+00, ..., 1.0075e+00,
8.2695e-01, 8.0911e-01],
[-1.2839e+00, 1.4440e-02, 1.0887e+00, ..., 1.3251e+00,
8.5205e-01, 5.1363e-01],
[-4.2980e-01, -3.7710e-01, -1.1583e-01, ..., 1.6275e+00,
1.5724e+00, 4.8289e-01]],
...,
[[-9.2770e-01, -2.1800e+00, -2.0628e+00, ..., -1.5596e+00,
-2.0523e+00, -4.8194e-01],
[-3.7212e+00, -4.8066e+00, -4.2721e+00, ..., -2.6658e+00,
-3.4927e+00, -3.0427e+00],
[-3.6167e+00, -3.9427e+00, -1.2297e+00, ..., -2.1423e+00,
-2.8994e+00, -1.9506e+00],
...,
[-1.1456e+00, -2.5028e+00, -1.0721e+00, ..., -1.1672e+00,
-2.0468e+00, -1.1313e+00],
[-2.0757e+00, -3.8879e+00, -1.4930e+00, ..., -1.5344e+00,
-2.7787e+00, -1.6088e+00],
[-1.4485e+00, -1.4871e+00, -6.0823e-01, ..., -1.8473e+00,
-2.1996e+00, -5.7896e-01]],
[[-1.7942e+00, -1.8203e+00, -2.1452e+00, ..., -5.7218e-01,
-1.4564e+00, -6.8534e-01],
[-4.1952e+00, -3.2390e+00, -3.7538e+00, ..., -1.7057e+00,
-3.7394e+00, -1.8170e+00],
[-3.9798e+00, -1.9077e+00, -1.3263e+00, ..., -9.0842e-01,
-4.3808e+00, -2.4837e+00],
...,
[-2.8040e+00, -2.6715e+00, -3.1856e+00, ..., -1.4511e+00,
-2.9716e+00, -2.0387e+00],
[-1.9787e+00, -1.2214e-01, -1.8292e+00, ..., -3.8458e-01,
-2.9987e+00, -1.8484e+00],
[-2.1102e+00, -1.4322e+00, -1.7822e+00, ..., -1.6336e+00,
-2.8112e+00, -1.1997e+00]],
[[ 1.1146e+00, 2.8361e-01, 1.0444e+00, ..., 2.4620e-01,
1.3340e+00, 6.0527e-01],
[ 3.9990e+00, 3.4812e+00, 1.9051e+00, ..., 7.6440e-01,
1.2636e+00, 2.5489e+00],
[ 2.0145e+00, 5.6293e-01, -1.1533e+00, ..., 3.5753e-01,
1.0408e-01, 8.0984e-01],
...,
[ 2.9537e+00, 2.4891e+00, 2.3336e+00, ..., 1.1274e+00,
7.0465e-01, 5.2580e-01],
[ 3.7877e+00, 3.9545e+00, 2.7004e+00, ..., 9.1534e-01,
6.4415e-01, 1.0864e+00],
[ 2.4108e+00, 2.2704e+00, 1.5126e+00, ..., -1.8058e-01,
-3.8808e-01, -1.4906e-01]]],
[[[-3.0931e-01, -1.1548e+00, -2.2294e+00, ..., -8.5669e-01,
-3.2208e-01, 8.3602e-01],
[ 2.5264e-01, -4.2970e+00, -3.2123e+00, ..., -2.9478e+00,
-1.6905e+00, 1.1269e+00],
[-1.4118e+00, -5.7662e+00, -3.0661e+00, ..., 7.4654e-01,
4.4830e-01, 1.0769e+00],
...,
[-1.0291e+00, -1.8346e+00, -9.2885e-01, ..., -2.0048e+00,
-5.9096e-01, -7.0240e-01],
[-1.0697e+00, -7.2910e-01, -1.6243e-01, ..., -1.5488e+00,
-2.3748e-01, -1.6603e+00],
[-2.6818e+00, -4.9027e+00, -3.9410e+00, ..., -5.1786e+00,
-4.1954e+00, -3.8943e+00]],
[[-2.0150e+00, -4.3992e+00, -3.4000e+00, ..., -3.5591e+00,
-5.3556e+00, -2.6703e-01],
[-5.8815e+00, -7.4855e+00, -2.7135e+00, ..., -4.7935e+00,
-7.1176e+00, -1.3314e+00],
[-5.3032e+00, -8.5902e+00, -4.7507e+00, ..., -6.3177e-01,
-5.2880e+00, -1.6690e+00],
...,
[-5.8755e+00, -6.1573e+00, -3.5339e+00, ..., -6.2518e+00,
-5.2357e+00, -3.8170e+00],
[-7.0417e+00, -7.9060e+00, -5.7593e+00, ..., -7.3259e+00,
-8.2440e+00, -6.6823e+00],
[-2.4878e+00, -4.4710e+00, -4.1176e+00, ..., -3.9706e+00,
-4.2699e+00, -2.5811e+00]],
[[-5.4279e-02, 3.0947e+00, 1.7823e+00, ..., 1.0026e+00,
2.2223e+00, 8.6883e-01],
[-7.4440e-02, 2.1437e+00, 3.6779e-01, ..., 2.6231e+00,
2.1990e+00, -6.0179e-01],
[-1.7085e-02, 1.7045e+00, 2.2244e-01, ..., 1.2912e+00,
2.8954e+00, 4.2559e-01],
...,
[ 8.9434e-01, 1.2241e+00, 5.1125e-01, ..., 8.0010e-02,
7.1593e-01, 2.4678e-01],
[ 8.0949e-01, 5.9224e-01, 3.8312e-01, ..., -2.4980e-01,
4.3424e-01, 1.9544e-01],
[ 1.3136e+00, 1.3722e+00, 9.5330e-01, ..., 1.5040e+00,
1.2636e+00, 1.0353e-01]],
...,
[[-1.5337e+00, -3.7456e+00, -2.0495e+00, ..., -2.1092e+00,
-2.7614e+00, -6.0254e-01],
[-4.1191e+00, -5.7468e+00, -2.4104e+00, ..., -4.1076e+00,
-3.9350e+00, -3.1397e+00],
[-3.7086e+00, -5.5015e+00, -2.3273e+00, ..., -8.8248e-01,
-3.5325e+00, -1.7858e+00],
...,
[-2.7294e+00, -2.4515e+00, -2.4800e+00, ..., -2.0585e+00,
-3.7163e+00, -1.6788e+00],
[-2.9652e+00, -2.8398e+00, -2.3711e+00, ..., -1.8802e+00,
-4.4163e+00, -2.3589e+00],
[-3.4925e+00, -3.6690e+00, -2.9411e+00, ..., -3.5904e+00,
-3.7846e+00, -1.6495e+00]],
[[-2.1703e+00, -1.6597e+00, -1.6027e+00, ..., -9.9044e-01,
-6.4113e-01, -1.1558e+00],
[-5.5899e+00, -3.0391e+00, -2.3882e+00, ..., -2.2605e+00,
-1.3893e-01, -1.6366e-02],
[-5.2097e+00, -1.9325e+00, -3.9488e-01, ..., -4.4376e-01,
-8.2676e-01, 7.2263e-02],
...,
[-4.2105e+00, -2.1407e+00, -5.6855e+00, ..., -2.6858e+00,
-3.1016e+00, -1.6212e+00],
[-3.6725e+00, -1.7790e-01, -3.7739e+00, ..., -1.3726e+00,
-2.9635e+00, -1.2803e+00],
[-4.3688e+00, -2.5140e+00, -2.7393e+00, ..., -2.1632e+00,
-3.2443e+00, -1.2581e+00]],
[[ 1.6048e+00, 5.9428e-01, -3.3613e-01, ..., -3.9085e-01,
-7.0260e-02, -4.3493e-01],
[ 4.5933e+00, 3.2482e+00, 1.9164e-01, ..., 2.3164e+00,
2.6599e+00, 2.0158e+00],
[ 2.3287e+00, 7.6468e-01, -1.4843e+00, ..., 1.7075e+00,
4.2640e+00, 1.8987e+00],
...,
[ 2.9384e+00, 1.1428e+00, 2.0934e+00, ..., 2.3226e+00,
2.1084e+00, 1.3258e+00],
[ 3.6749e+00, 3.4276e+00, 2.1530e+00, ..., 2.9288e+00,
2.3068e+00, 2.1484e+00],
[ 1.9266e+00, 9.1770e-01, 9.7973e-02, ..., 1.3896e-01,
-2.9264e-01, -1.5246e-01]]],
[[[-1.1022e+00, -1.8996e+00, -3.1593e+00, ..., 4.6757e-02,
5.0671e-01, -1.8267e-01],
[-1.4528e+00, -1.3694e+00, 2.1000e+00, ..., 4.8589e-01,
7.8406e-01, -5.2036e-01],
[-4.2564e+00, -1.8944e+00, -9.6327e-01, ..., 8.4470e-03,
-3.3023e-02, -1.0727e+00],
...,
[-2.2027e-01, -8.2493e-03, 5.9669e-02, ..., -2.3808e+00,
-1.5420e+00, -3.9981e-01],
[-1.4760e+00, 1.3191e-01, 3.0667e-01, ..., -4.8218e-01,
-3.8012e-01, -1.4188e+00],
[-1.3340e+00, -2.4854e+00, -1.9818e+00, ..., -3.5489e+00,
-2.8554e+00, -2.6024e+00]],
[[-4.0095e+00, -4.4405e+00, -6.3202e+00, ..., -2.1264e+00,
-3.5672e+00, -2.2373e+00],
[-6.6642e+00, -3.5544e+00, 4.1493e-01, ..., -2.4015e+00,
-3.9567e+00, -3.5474e+00],
[-4.6185e+00, -1.0984e+00, -1.3575e+00, ..., -2.3677e+00,
-2.4964e+00, -2.6409e+00],
...,
[-4.5448e+00, -5.2550e+00, -4.7199e+00, ..., -7.3538e+00,
-3.5516e+00, -2.1725e+00],
[-7.6295e+00, -7.9971e+00, -6.6563e+00, ..., -4.8819e+00,
-5.4099e+00, -5.0368e+00],
[-1.5174e+00, -3.2756e+00, -2.0992e+00, ..., -3.4418e+00,
-2.8075e+00, -1.7053e+00]],
[[-1.3899e-01, 2.5535e+00, 2.8549e+00, ..., 1.6356e+00,
1.7964e+00, 9.0474e-01],
[ 9.1117e-01, -1.0258e+00, -1.3568e-02, ..., 9.2365e-01,
1.5496e+00, 8.4569e-01],
[ 2.5255e+00, -3.5889e-01, -3.4524e+00, ..., 1.6968e+00,
1.8604e+00, 1.3706e+00],
...,
[-6.9270e-01, -1.4526e-01, 8.3851e-01, ..., -1.1230e+00,
5.2731e-02, 2.3699e-01],
[-1.2440e+00, -2.3867e-01, 1.7555e-01, ..., 6.2812e-01,
-3.6597e-01, 5.5418e-01],
[-3.6939e-01, -2.7057e-01, 1.7342e-01, ..., 1.0424e+00,
1.6935e+00, 1.0846e+00]],
...,
[[-2.6419e+00, -2.9030e+00, -4.2863e+00, ..., -1.4746e+00,
-2.0172e+00, -4.3903e-01],
[-4.4340e+00, -1.3364e+00, 1.5800e+00, ..., -2.3183e+00,
-3.5135e+00, -2.9767e+00],
[-4.8567e+00, -2.3170e+00, -1.0572e+00, ..., -1.5721e+00,
-2.5926e+00, -2.0020e+00],
...,
[-8.3465e-01, -2.1236e+00, -8.2173e-01, ..., -3.7810e+00,
-3.2267e+00, -1.1951e+00],
[-2.0602e+00, -4.0026e+00, -1.4146e+00, ..., -2.0334e+00,
-4.2939e+00, -2.6304e+00],
[-1.4952e+00, -1.6412e+00, -8.9146e-01, ..., -2.6019e+00,
-2.6524e+00, -9.0199e-01]],
[[-1.9743e+00, -2.9631e+00, -5.4801e+00, ..., -5.4598e-01,
-1.4882e+00, -7.0060e-01],
[-3.2434e+00, 1.0176e+00, -1.3750e+00, ..., -1.1397e+00,
-3.6823e+00, -1.9335e+00],
[-3.4863e+00, 5.1314e-02, -1.7360e+00, ..., -1.9402e+00,
-4.4554e+00, -2.6483e+00],
...,
[-2.7593e+00, -3.1658e+00, -3.4118e+00, ..., -2.4257e+00,
-4.0968e+00, -3.8012e-01],
[-1.9492e+00, -2.0740e-01, -1.5430e+00, ..., -1.8323e-01,
-2.4094e+00, -8.9526e-01],
[-2.2142e+00, -1.6782e+00, -1.7726e+00, ..., -3.5587e-01,
-1.8177e+00, -6.8765e-01]],
[[ 2.3970e+00, -1.0509e+00, -4.0200e-01, ..., 3.0891e-01,
1.2943e+00, 6.1452e-01],
[ 5.0088e+00, 2.5075e+00, 2.2304e+00, ..., 9.1199e-01,
1.1277e+00, 2.5018e+00],
[ 1.3685e+00, -3.6578e-01, -3.5585e+00, ..., 5.2400e-01,
1.3133e-01, 6.3620e-01],
...,
[ 3.0799e+00, 2.3281e+00, 1.4845e+00, ..., 2.3083e+00,
6.7769e-01, 1.3539e+00],
[ 3.8227e+00, 4.0508e+00, 2.8762e+00, ..., 1.5899e+00,
1.4441e+00, 1.3072e+00],
[ 2.3668e+00, 2.2966e+00, 1.5075e+00, ..., -2.5444e-01,
-7.0523e-01, -4.6239e-01]]],
[[[ 3.2271e-01, 1.3673e+00, 1.2272e-01, ..., -2.6578e+00,
-2.4687e+00, -9.9450e-01],
[ 4.9833e-01, 1.0733e+00, 1.2458e+00, ..., -8.0449e-02,
-2.6226e+00, -2.1248e-01],
[-1.6144e-01, -1.4157e-01, 1.8035e-01, ..., 1.3300e-01,
-3.2166e+00, -1.0810e-01],
...,
[-5.2168e-01, -1.3752e+00, 4.2889e-01, ..., -3.1896e+00,
-3.6413e+00, 7.2522e-01],
[-2.0139e+00, -1.2476e+00, -1.1432e+00, ..., -3.7326e+00,
-3.5388e+00, 6.8099e-01],
[-1.0520e+00, -2.9119e+00, -3.1654e+00, ..., -3.7843e+00,
-3.1669e+00, -1.4130e+00]],
[[-8.5504e-01, -1.3546e+00, -3.2130e+00, ..., -5.7527e+00,
-7.1366e+00, -1.2917e+00],
[-3.5233e+00, -3.8471e+00, -3.4351e+00, ..., -6.7139e+00,
-8.8369e+00, -2.3923e+00],
[-3.4357e+00, -3.9620e+00, -3.7578e+00, ..., -5.9364e+00,
-8.5839e+00, -2.6549e+00],
...,
[-5.7145e+00, -5.9345e+00, -4.4338e+00, ..., -7.8381e+00,
-8.5099e+00, -1.7622e+00],
[-7.6292e+00, -7.4362e+00, -6.7505e+00, ..., -9.1211e+00,
-1.0801e+01, -4.2897e+00],
[-1.6860e+00, -3.0172e+00, -2.9341e+00, ..., -3.2237e+00,
-3.6758e+00, -7.5927e-01]],
[[-7.6958e-02, 1.2528e+00, 1.4932e+00, ..., 2.1764e+00,
2.3548e+00, 1.0231e+00],
[-5.1106e-02, -1.5725e-02, 7.9133e-01, ..., 3.0332e+00,
2.7245e+00, 7.7687e-02],
[ 6.5432e-01, 1.1683e+00, 1.5269e+00, ..., 1.6020e+00,
1.6903e+00, -3.1464e-01],
...,
[ 2.5117e-01, 1.0742e+00, 4.9615e-01, ..., 1.0961e-02,
1.3417e+00, -1.3528e+00],
[ 3.4512e-01, 4.9812e-01, 1.3183e-01, ..., -1.3542e-01,
1.3110e+00, -1.6137e+00],
[ 4.7952e-01, 8.9612e-01, 6.4459e-01, ..., -1.8443e-01,
-2.4202e-01, -1.9187e+00]],
...,
[[-4.4768e-01, -1.4712e+00, -1.7133e+00, ..., -4.5784e+00,
-5.4445e+00, -8.0183e-01],
[-2.4489e+00, -2.5554e+00, -1.3440e+00, ..., -3.3011e+00,
-5.1075e+00, -1.9449e+00],
[-1.7021e+00, -1.2776e+00, -9.3547e-01, ..., -3.5732e+00,
-5.4942e+00, -1.6151e+00],
...,
[-1.7698e+00, -1.2667e+00, -3.7280e-01, ..., -2.9555e+00,
-5.6055e+00, -1.4408e+00],
[-3.2974e+00, -2.6296e+00, -1.7507e+00, ..., -3.0099e+00,
-5.4434e+00, -8.5387e-01],
[-1.4002e+00, -1.3611e+00, -1.6750e+00, ..., -1.6272e+00,
-2.3264e+00, 4.4883e-01]],
[[-1.1358e+00, -2.5015e+00, -3.3861e+00, ..., -9.2865e-01,
-7.1912e-01, -5.7377e-01],
[-3.3448e+00, -3.5492e+00, -4.1752e+00, ..., -3.0436e+00,
-2.1922e+00, -3.3324e-01],
[-4.2452e+00, -3.8655e+00, -4.0008e+00, ..., -2.2844e+00,
-2.5646e+00, -4.2719e-01],
...,
[-4.6932e+00, -3.7372e+00, -5.5058e+00, ..., -3.1179e+00,
-1.9684e+00, -8.7038e-01],
[-4.0661e+00, -1.3610e+00, -3.3941e+00, ..., -1.9259e+00,
-5.4987e-01, 1.2349e+00],
[-3.0651e+00, -2.2099e+00, -1.8221e+00, ..., -3.8460e-01,
-8.1887e-01, 6.6624e-01]],
[[ 1.1500e+00, 3.6980e-02, 1.1031e+00, ..., 1.3430e+00,
6.4332e-01, -1.4265e+00],
[ 2.0548e+00, 1.0770e+00, 1.2735e+00, ..., 5.3437e+00,
2.9835e+00, 2.2764e+00],
[ 1.2964e+00, 8.3386e-01, 4.3619e-01, ..., 3.5292e+00,
1.1209e+00, 1.2448e+00],
...,
[ 2.6872e+00, 1.9179e+00, 1.6049e+00, ..., 2.4375e+00,
2.0811e+00, 6.1234e-01],
[ 2.8901e+00, 4.0116e+00, 2.3764e+00, ..., 2.1964e+00,
1.0397e+00, 1.8676e+00],
[ 1.9313e+00, 1.8802e+00, 1.6164e-01, ..., 5.0613e-01,
-1.7658e-01, 9.4848e-01]]]], device='cuda:0'),
tensor([[[[ 2.9538e-01, 1.3169e+00, 2.8915e-01, ..., -1.6048e+00,
-1.4379e+00, 4.1910e-01],
[ 3.3867e-01, 1.2500e+00, 9.4981e-01, ..., -2.5843e+00,
-5.7179e-01, 9.4558e-01],
[-3.1584e-01, 2.7802e-01, -2.5685e+00, ..., -4.1644e+00,
-4.0960e-01, 5.8282e-01],
...,
[-7.1437e-01, -1.8997e+00, -9.6198e-02, ..., -3.6387e+00,
-2.6880e+00, -1.8323e-01],
[-1.9755e+00, -1.8222e+00, -6.8687e-01, ..., -2.9910e+00,
-3.2824e+00, -1.9212e-01],
[-1.0492e+00, -3.2886e+00, -3.1415e+00, ..., -3.3679e+00,
-3.4783e+00, -1.2355e+00]],
[[-1.0306e+00, -1.5328e+00, -3.6909e+00, ..., -5.2402e+00,
-2.9938e+00, 2.5468e-01],
[-4.0145e+00, -4.0255e+00, -5.6158e+00, ..., -7.8159e+00,
-1.4184e+00, -2.1480e-01],
[-4.0353e+00, -4.0519e+00, -6.4695e+00, ..., -9.3970e+00,
-2.2471e+00, 1.3200e-01],
...,
[-5.9818e+00, -5.8968e+00, -4.7456e+00, ..., -1.1709e+01,
-6.8137e+00, -1.6854e+00],
[-7.7794e+00, -7.7129e+00, -7.2358e+00, ..., -1.0956e+01,
-8.4377e+00, -3.7686e+00],
[-1.7608e+00, -3.2215e+00, -3.4853e+00, ..., -4.5391e+00,
-3.5353e+00, -4.2896e-01]],
[[-5.6018e-02, 1.4352e+00, 1.3225e+00, ..., 8.9090e-01,
2.3771e+00, 2.8342e-01],
[ 1.4678e-02, -1.1735e-01, 4.4159e-01, ..., -3.7867e-01,
1.6021e-01, -9.0821e-01],
[ 7.8839e-01, 7.4974e-01, 1.4270e+00, ..., -1.4908e+00,
-7.4970e-01, 9.6480e-01],
...,
[ 3.4019e-01, 7.4165e-01, 1.0811e+00, ..., -1.2422e-01,
4.5834e-01, -1.7472e-02],
[ 3.9065e-01, 5.8649e-01, 8.6263e-01, ..., 2.6034e-01,
2.2159e-01, -4.8266e-01],
[ 4.5006e-01, 1.0582e+00, 1.4957e+00, ..., -1.9461e-01,
-3.5255e-02, -1.6455e+00]],
...,
[[-4.8674e-01, -1.4336e+00, -2.1109e+00, ..., -2.7066e+00,
-2.7905e+00, -1.7428e-01],
[-2.5472e+00, -2.6113e+00, -1.9943e+00, ..., -2.9437e+00,
-1.3875e+00, -1.7099e+00],
[-1.7190e+00, -1.5320e+00, -2.4900e+00, ..., -3.5796e+00,
-1.3696e+00, -1.9826e+00],
...,
[-1.8684e+00, -1.6403e+00, -1.2165e+00, ..., -2.5381e+00,
-5.8294e+00, -1.3646e+00],
[-3.1998e+00, -3.2318e+00, -2.0232e+00, ..., -1.0608e+00,
-5.1119e+00, -4.6368e-01],
[-1.4115e+00, -1.8718e+00, -2.0152e+00, ..., -8.7036e-01,
-2.6643e+00, 1.0630e+00]],
[[-1.2891e+00, -2.4995e+00, -3.5900e+00, ..., -2.7909e+00,
-1.6559e+00, 1.6447e-01],
[-3.4370e+00, -3.3453e+00, -5.5060e+00, ..., -2.7473e+00,
-2.7003e+00, -5.4550e-01],
[-4.3085e+00, -4.3108e+00, -3.8029e+00, ..., -2.6045e+00,
-2.1616e+00, -1.9572e+00],
...,
[-4.8037e+00, -3.6878e+00, -3.0461e+00, ..., -4.9220e+00,
-5.4884e+00, -1.2155e+00],
[-4.1979e+00, -1.9464e+00, -1.8130e+00, ..., -1.5456e+00,
-2.6504e+00, 2.6283e-01],
[-3.1788e+00, -3.1658e+00, -1.5958e+00, ..., 4.2608e-02,
-6.7807e-01, 5.0873e-01]],
[[ 1.2295e+00, -5.5739e-02, 1.4678e+00, ..., -1.0346e+00,
-7.9552e-01, 1.0231e-02],
[ 2.3499e+00, 1.2224e+00, 2.4692e+00, ..., 8.2297e-01,
-8.3105e-02, 1.7690e+00],
[ 1.4882e+00, 1.3162e+00, 1.5157e+00, ..., 2.4112e-01,
2.7081e-01, -7.1377e-01],
...,
[ 2.4736e+00, 1.7790e+00, 1.3012e+00, ..., -1.7169e-01,
8.9988e-01, 7.8733e-01],
[ 2.9216e+00, 3.9450e+00, 2.8321e+00, ..., 2.2131e-01,
5.7778e-01, 1.5427e+00],
[ 2.0127e+00, 1.6069e+00, 7.7570e-01, ..., 1.8500e-01,
-4.4939e-01, 7.3734e-01]]],
[[[-2.4543e-01, 1.0695e+00, 1.2515e+00, ..., -1.2800e+00,
-6.4326e-01, -6.8964e-01],
[-1.5045e-02, -4.6927e-01, 2.0305e+00, ..., 2.5655e+00,
2.1856e+00, -9.8525e-01],
[-1.2630e+00, 3.6879e-01, 1.2704e+00, ..., 2.3313e+00,
1.8050e+00, -9.9510e-01],
...,
[-8.1785e-01, -2.9725e+00, 1.2422e+00, ..., 1.0322e+00,
1.2942e+00, 1.5290e-01],
[-1.5759e+00, -1.7587e+00, -1.3521e+00, ..., 7.3310e-01,
4.6658e-01, 1.2328e-01],
[-5.5497e-01, -1.9362e+00, -2.4563e+00, ..., -1.0836e+00,
-6.3518e-02, -8.4821e-01]],
[[-1.6016e+00, -2.2169e+00, -3.6227e+00, ..., -2.1467e+00,
-2.9059e+00, -2.0543e+00],
[-4.8157e+00, -4.5660e+00, -3.1785e+00, ..., 1.7237e+00,
-2.1374e+00, -3.6607e+00],
[-3.6904e+00, -5.3606e+00, -6.1947e+00, ..., 1.3134e+00,
-1.7463e+00, -2.9230e+00],
...,
[-3.9952e+00, -7.5868e+00, -6.6054e+00, ..., 5.0776e-01,
-6.9781e-01, -1.1261e+00],
[-7.7155e+00, -7.9971e+00, -1.0690e+01, ..., -7.7241e-01,
2.4916e-01, -1.1118e+00],
[-1.4160e+00, -3.0825e+00, -1.1627e+00, ..., -2.9500e+00,
-7.5406e-01, -4.7128e-01]],
[[-2.4701e-01, 2.1184e+00, 1.7529e+00, ..., 3.7034e+00,
2.2331e+00, 8.4149e-01],
[-5.6534e-01, 7.1456e-02, -1.6779e-01, ..., 1.0528e+00,
2.5930e+00, 1.2599e+00],
[ 1.5543e-03, 2.4287e+00, 7.7482e-01, ..., -1.0763e+00,
2.1647e+00, 1.4408e+00],
...,
[-4.5295e-01, -6.5276e-01, -6.8629e+00, ..., 7.7447e-01,
-7.1245e-01, -5.1456e-01],
[-2.2001e+00, -3.2959e-02, 4.3714e-01, ..., -1.2689e+00,
5.8026e-01, 4.9199e-01],
[-1.0653e+00, -1.0227e+00, 3.9469e-01, ..., -1.3920e+00,
7.0515e-01, 1.0446e+00]],
...,
[[-1.3170e+00, -1.7881e+00, -1.3431e+00, ..., -2.6039e+00,
-2.9867e+00, -1.1418e+00],
[-3.6012e+00, -3.2784e+00, -1.9334e+00, ..., -1.2707e+00,
-2.6149e+00, -3.3696e+00],
[-2.6442e+00, -2.2496e+00, -2.6090e+00, ..., -2.9134e+00,
-4.4156e+00, -2.2201e+00],
...,
[-1.2477e+00, -3.9340e+00, -2.2848e+00, ..., -9.5679e-01,
-1.9711e+00, -1.8964e-01],
[-1.8868e+00, -5.9731e+00, -6.1177e+00, ..., 5.3169e-01,
-1.0017e+00, 5.6484e-01],
[-6.6273e-01, -3.9228e-01, 6.8539e-01, ..., -3.0070e-01,
-1.0417e+00, 8.4481e-01]],
[[-1.7799e+00, -1.4046e+00, -3.0110e+00, ..., -8.6157e-01,
4.4555e-01, -1.2332e-01],
[-4.0257e+00, -1.3934e+00, -2.3950e+00, ..., -1.7071e+00,
-1.8444e+00, -7.5333e-01],
[-3.8441e+00, -2.9439e+00, -5.4286e+00, ..., -3.6101e+00,
-3.6888e+00, -1.1640e+00],
...,
[-2.7489e+00, 6.0952e-01, 3.5688e+00, ..., -1.9273e+00,
-3.4154e+00, -2.1562e+00],
[-1.6185e+00, -1.5048e+00, -1.4628e-01, ..., 1.4572e+00,
-2.2377e+00, -1.3109e+00],
[-1.5769e+00, -1.2281e+00, -7.0878e-02, ..., 1.9029e+00,
-7.5832e-01, -1.4069e-01]],
[[ 1.2589e+00, 3.8009e-01, 7.7907e-01, ..., 2.9843e+00,
2.0664e+00, 8.3208e-01],
[ 3.9899e+00, 2.6176e+00, 1.9895e+00, ..., 1.7231e+00,
3.0081e+00, 2.3746e+00],
[ 1.4625e+00, 6.1367e-01, 1.5350e+00, ..., 8.1077e-01,
2.7403e+00, 9.6225e-01],
...,
[ 2.2356e+00, 3.4747e+00, 3.8819e+00, ..., -1.0100e+00,
-1.7275e+00, -1.6371e+00],
[ 3.9733e+00, 3.2143e+00, 5.4307e+00, ..., -2.3527e-01,
-2.5808e+00, -1.3264e+00],
[ 2.7396e+00, 1.9697e+00, 6.8714e-02, ..., 1.7628e+00,
-3.5015e-01, 5.5814e-02]]],
[[[-8.0632e-01, -4.5372e-01, -1.1235e+00, ..., -3.0581e+00,
-2.2228e+00, -9.3593e-01],
[-2.3860e+00, -1.4396e+00, -2.2293e+00, ..., -1.6131e+00,
-2.4474e+00, -2.6622e-01],
[ 2.5821e-02, 5.9382e-01, -1.0907e+00, ..., -7.5364e-01,
-2.8235e+00, -4.8468e-01],
...,
[-8.1745e-01, -1.4017e+00, 1.4672e+00, ..., -7.1894e+00,
2.0134e+00, 1.4966e+00],
[-1.8506e+00, -1.2729e+00, -2.2042e-01, ..., -1.5799e+00,
6.4467e+00, 2.8010e+00],
[-1.0310e+00, -3.0082e+00, -2.7880e+00, ..., 3.5297e+00,
1.6706e-01, -1.9357e+00]],
[[-5.7529e-01, -2.4090e+00, -2.8739e+00, ..., -6.2133e+00,
-6.9730e+00, -1.2779e+00],
[-3.5326e+00, -4.6113e+00, -8.4673e+00, ..., -6.9693e+00,
-9.1212e+00, -2.6385e+00],
[ 9.1064e-01, 5.3210e-01, 1.0342e+00, ..., -4.4099e+00,
-8.9222e+00, -2.8850e+00],
...,
[-5.8493e+00, -4.9257e+00, -2.0855e+00, ..., -7.1330e+00,
1.9681e-01, 4.1186e-01],
[-7.3655e+00, -7.9559e+00, -8.0573e+00, ..., -2.9107e-01,
2.3311e+00, -1.9222e+00],
[-1.7964e+00, -2.7127e+00, -3.2190e+00, ..., -3.7235e-02,
-3.5192e+00, -3.3258e+00]],
[[-4.9588e-02, 1.6307e+00, 2.4726e+00, ..., 1.3900e+00,
2.3461e+00, 9.6175e-01],
[ 1.0124e+00, 6.7718e-01, 2.8523e+00, ..., 1.4869e+00,
2.9944e+00, -8.7008e-02],
[-2.1629e-01, -1.5965e+00, -4.5537e+00, ..., 1.7010e+00,
2.3250e+00, -3.6287e-01],
...,
[-2.0190e-01, 5.0989e-02, 7.4352e-01, ..., -7.7186e+00,
-2.5943e+00, 1.5046e+00],
[ 5.9844e-02, 1.1463e-01, 2.6334e-01, ..., -5.7014e+00,
-3.3375e+00, 1.0467e+00],
[ 2.3458e-01, 8.2592e-01, 1.6712e+00, ..., -4.9423e+00,
-2.9418e+00, 5.6690e-01]],
...,
[[-8.8417e-01, -2.3500e+00, -2.9609e+00, ..., -4.9458e+00,
-5.0819e+00, -6.1078e-01],
[-3.7384e+00, -2.6241e+00, -3.1619e+00, ..., -3.4816e+00,
-5.1721e+00, -2.0064e+00],
[-1.6224e-02, 1.7213e-01, 2.7990e+00, ..., -4.4036e+00,
-4.7516e+00, -1.9094e+00],
...,
[-1.4116e+00, -4.1852e-01, 6.2984e-01, ..., -2.8039e+00,
-3.9732e+00, -6.2495e+00],
[-3.1760e+00, -3.0062e+00, -3.1212e+00, ..., -1.3793e+00,
-2.9778e+00, -4.4520e+00],
[-1.2805e+00, -1.0627e+00, -1.3044e+00, ..., -1.5079e-01,
-5.1710e+00, -2.7124e+00]],
[[-1.5412e+00, -2.3323e+00, -5.0841e+00, ..., -1.4017e+00,
-4.3240e-01, -6.3732e-01],
[-3.9433e+00, -3.1025e+00, -7.0594e+00, ..., -2.9223e+00,
-1.7746e+00, -3.5483e-01],
[-2.1079e+00, -3.9415e-02, 2.2270e+00, ..., -6.2016e-01,
-7.1146e-01, -3.0036e-01],
...,
[-4.3896e+00, -2.0045e+00, -2.6342e+00, ..., 1.2123e+00,
1.7560e+00, -5.9233e-01],
[-4.4005e+00, -1.7634e+00, -2.7835e+00, ..., 1.7720e+00,
2.9885e+00, -1.0310e+00],
[-3.1759e+00, -3.0265e+00, -2.5701e+00, ..., 2.3831e+00,
-2.2699e+00, -1.9997e+00]],
[[ 9.3190e-01, -8.0283e-01, 4.0619e-01, ..., 5.3888e-01,
8.2654e-01, -1.3991e+00],
[ 1.7445e+00, 2.6590e+00, 1.9680e+00, ..., 3.7939e+00,
3.1319e+00, 2.2040e+00],
[ 3.3819e-01, -2.4962e-02, -1.2974e+00, ..., 3.4634e+00,
1.8694e+00, 9.1922e-01],
...,
[ 2.6627e+00, 1.4152e+00, 8.1329e-01, ..., -7.0901e+00,
1.5757e+00, 2.7104e+00],
[ 2.7370e+00, 4.2192e+00, 2.9050e+00, ..., -2.8850e+00,
4.8065e+00, 3.2599e+00],
[ 1.8491e+00, 1.2504e+00, -6.4968e-02, ..., 3.6666e+00,
4.2594e+00, 4.1104e-01]]],
[[[-2.5894e-01, 1.4221e+00, 1.4754e+00, ..., -1.0795e+00,
-3.3476e+00, -8.6307e-01],
[ 3.6434e-01, -1.4834e-01, 8.6325e-01, ..., 3.5251e-01,
-2.2674e+00, -5.6729e-01],
[-5.4078e-01, -1.5982e+00, -2.3364e+00, ..., 2.3593e-01,
-2.2109e+00, -9.6722e-01],
...,
[-2.1665e-01, -2.7652e-01, 1.1987e+00, ..., -2.3246e+00,
-2.9240e+00, 2.6443e-01],
[-1.5910e+00, -1.5144e-01, 9.9582e-01, ..., -2.5042e+00,
-3.8653e+00, -7.5648e-01],
[-1.2544e+00, -2.4002e+00, -1.3184e+00, ..., -3.6751e+00,
-3.7922e+00, -1.6678e+00]],
[[-1.4475e+00, -2.4214e+00, -3.8320e+00, ..., -5.3558e+00,
-9.0471e+00, -1.8608e+00],
[-4.7566e+00, -6.4907e+00, -5.5959e+00, ..., -5.3557e+00,
-8.8463e+00, -3.3905e+00],
[-3.7420e+00, -6.3299e+00, -7.9076e+00, ..., -5.6597e+00,
-8.2942e+00, -3.3446e+00],
...,
[-4.6015e+00, -5.3664e+00, -5.7186e+00, ..., -6.3137e+00,
-8.2104e+00, -2.4059e+00],
[-7.8245e+00, -8.0893e+00, -7.5423e+00, ..., -9.1356e+00,
-1.1010e+01, -4.2834e+00],
[-1.4694e+00, -3.1553e+00, -2.3962e+00, ..., -3.8947e+00,
-4.1583e+00, -5.7975e-01]],
[[-2.2952e-01, 2.3187e+00, 1.3948e+00, ..., 2.1124e+00,
2.4695e+00, 6.8623e-01],
[-3.9514e-01, 1.0066e-01, -1.9028e-01, ..., 1.3585e+00,
2.1620e+00, -3.0135e-01],
[-1.2197e-01, 1.3838e+00, 1.4659e+00, ..., -4.5079e-03,
1.3010e+00, -4.5735e-01],
...,
[-7.8580e-01, 8.0123e-01, 1.5581e+00, ..., 9.3788e-01,
2.9479e-01, -4.2872e-01],
[-1.2346e+00, 1.5058e-01, 1.0561e+00, ..., -3.3464e-01,
1.6457e-01, -7.9343e-01],
[-3.2420e-01, -2.5119e-01, 2.3164e-01, ..., 5.3944e-01,
2.5822e-01, -1.8098e+00]],
...,
[[-1.2474e+00, -2.0503e+00, -6.2035e-01, ..., -3.5300e+00,
-5.4378e+00, -1.1960e+00],
[-3.2276e+00, -4.0050e+00, -2.0025e+00, ..., -2.9144e+00,
-5.6324e+00, -2.5729e+00],
[-2.5198e+00, -3.2304e+00, -2.2823e+00, ..., -3.2176e+00,
-6.5331e+00, -1.9565e+00],
...,
[-9.6303e-01, -2.3580e+00, -5.5951e-01, ..., -1.9573e+00,
-4.0477e+00, -2.1506e+00],
[-2.1560e+00, -3.8603e+00, -6.8776e-01, ..., -2.1814e+00,
-4.3850e+00, -4.1077e-01],
[-1.3955e+00, -1.4354e+00, 1.4295e-01, ..., -1.4768e+00,
-2.2039e+00, 9.2421e-01]],
[[-1.7809e+00, -1.5812e+00, -1.5420e+00, ..., -2.1044e+00,
-1.0025e+00, 8.9987e-02],
[-4.1868e+00, -2.6744e+00, -7.2750e-01, ..., -2.4003e+00,
-2.8850e+00, -1.8977e-01],
[-4.2580e+00, -2.9086e+00, -1.3983e+00, ..., -1.3583e+00,
-2.9120e+00, 6.8523e-02],
...,
[-2.9035e+00, -2.9040e+00, -3.1278e+00, ..., -2.2449e+00,
-1.4253e+00, 2.9064e-01],
[-2.0991e+00, -1.7056e-01, -1.3374e+00, ..., -6.7114e-01,
-1.2258e+00, 1.1908e+00],
[-2.1985e+00, -1.3921e+00, -4.4453e-01, ..., -7.2030e-01,
-1.1199e+00, 5.7283e-01]],
[[ 1.1717e+00, 5.1433e-01, 1.2547e+00, ..., 1.7474e+00,
1.6847e+00, 1.4073e-01],
[ 4.0582e+00, 3.7114e+00, 3.0762e+00, ..., 3.5447e+00,
3.0413e+00, 3.8089e+00],
[ 1.8747e+00, 1.4494e+00, -1.4785e-01, ..., 3.3011e+00,
2.0098e+00, 2.0979e+00],
...,
[ 3.0878e+00, 2.6062e+00, 2.6248e+00, ..., 2.3174e+00,
2.7295e+00, 1.6221e+00],
[ 3.7956e+00, 4.0435e+00, 2.9998e+00, ..., 2.7357e+00,
1.1101e+00, 2.5563e+00],
[ 2.3772e+00, 2.3166e+00, 1.6260e+00, ..., 4.2411e-01,
-1.3792e-01, 1.2666e+00]]],
[[[ 1.2766e-01, 1.2918e+00, 7.1412e-01, ..., -4.0771e+00,
-3.4237e-01, 1.2273e+00],
[ 7.2700e-01, 1.1811e+00, 2.5808e+00, ..., -6.2233e+00,
2.7564e-01, 1.4916e+00],
[-9.7290e-02, 2.9105e-01, -1.1977e+00, ..., -6.6185e+00,
4.0280e-01, -1.0885e+00],
...,
[-8.8722e-01, -1.9297e+00, 1.4335e+00, ..., -3.4002e+00,
-1.9057e+00, -1.3570e-01],
[-1.9386e+00, -2.0893e+00, -8.5535e-01, ..., -2.7033e+00,
-2.2116e+00, -5.5562e-02],
[-1.1620e+00, -3.4166e+00, -3.4070e+00, ..., -3.5324e+00,
-3.0677e+00, -1.2314e+00]],
[[-9.2728e-01, -1.6716e+00, -2.9702e+00, ..., -6.2977e+00,
-2.4467e+00, -1.4614e-01],
[-3.9540e+00, -5.6867e+00, -5.3136e+00, ..., -6.8240e+00,
4.9101e-01, -8.5607e-01],
[-4.6080e+00, -5.3342e+00, -1.0609e+01, ..., -5.5450e+00,
2.1810e+00, -1.9232e+00],
...,
[-6.1191e+00, -5.6870e+00, -3.3766e+00, ..., -7.3430e+00,
-6.2179e+00, -1.6080e+00],
[-7.7865e+00, -7.6701e+00, -6.6322e+00, ..., -8.4772e+00,
-8.1891e+00, -3.5847e+00],
[-1.8088e+00, -3.2612e+00, -3.0826e+00, ..., -4.0704e+00,
-3.4192e+00, -5.3770e-01]],
[[-1.0121e-01, 1.4636e+00, 1.8579e+00, ..., 1.2196e+00,
9.3319e-01, 8.0352e-01],
[ 1.5166e-01, 3.3821e-01, -4.7369e-01, ..., -8.8770e-02,
-2.3763e+00, -1.6831e+00],
[ 7.8093e-01, 6.2550e-01, 8.7017e-01, ..., -3.9029e+00,
-2.4512e+00, -9.2794e-01],
...,
[ 3.0618e-01, 8.6604e-01, 2.4231e-01, ..., -3.5708e-01,
5.9395e-01, -4.0129e-01],
[ 5.0325e-01, 8.8425e-01, 9.5258e-01, ..., -3.5602e-01,
7.1788e-01, -5.2339e-01],
[ 5.4536e-01, 1.0623e+00, 1.4788e+00, ..., -6.2474e-01,
-2.3188e-02, -1.6811e+00]],
...,
[[-5.1556e-01, -1.5680e+00, -2.0238e+00, ..., -3.0870e+00,
3.7217e-01, 1.2427e+00],
[-2.4033e+00, -3.0469e+00, -1.7838e+00, ..., -1.1751e+00,
9.7323e-01, 1.6035e-01],
[-1.4546e+00, -2.5447e+00, -5.0426e+00, ..., 1.2281e+00,
5.1631e-01, -1.1511e+00],
...,
[-2.2106e+00, -1.8433e+00, 3.4791e-01, ..., -3.5778e+00,
-4.7080e+00, -1.3169e+00],
[-3.2763e+00, -3.3855e+00, -1.6783e+00, ..., -2.9043e+00,
-4.4658e+00, -2.1842e-01],
[-1.5903e+00, -1.9289e+00, -1.7025e+00, ..., -1.6121e+00,
-2.0808e+00, 1.0331e+00]],
[[-1.2701e+00, -2.1743e+00, -3.4509e+00, ..., -1.5197e+00,
-1.7375e+00, -4.0925e-01],
[-3.6637e+00, -3.5632e+00, -3.7288e+00, ..., -3.4379e+00,
-5.1715e+00, -2.4827e+00],
[-4.7902e+00, -5.2893e+00, -6.7468e+00, ..., 1.6836e+00,
-7.6504e+00, -5.2955e+00],
...,
[-4.7867e+00, -3.2520e+00, -2.8324e+00, ..., -4.2831e+00,
-3.9442e+00, -1.0128e+00],
[-4.2915e+00, -2.0740e+00, -1.5405e+00, ..., -7.1610e-01,
-1.6901e+00, 4.5566e-01],
[-3.2560e+00, -3.1762e+00, -1.3797e+00, ..., 3.3115e-01,
-5.5812e-01, 5.6800e-01]],
[[ 1.1088e+00, -1.0691e-01, 1.2553e+00, ..., -1.8324e+00,
5.2009e-01, 7.5887e-01],
[ 2.3919e+00, 1.5352e+00, 2.8828e+00, ..., -2.0601e+00,
1.0473e+00, -1.2438e+00],
[ 1.3289e+00, 1.3132e+00, 3.2022e+00, ..., -1.0421e+00,
-2.1421e+00, -4.1076e+00],
...,
[ 2.4290e+00, 2.1443e+00, 9.2091e-01, ..., 1.1976e-01,
1.0785e+00, 7.1864e-01],
[ 3.0202e+00, 3.8186e+00, 2.6641e+00, ..., 4.7441e-01,
6.3410e-01, 1.4124e+00],
[ 2.0241e+00, 1.5602e+00, 2.5836e-01, ..., -4.1603e-01,
-3.7559e-01, 6.9634e-01]]]], device='cuda:0'),
tensor([[[[-3.7264e-01, -1.1352e+00, -4.5008e+00, ..., -2.3120e+00,
-3.7412e+00, -1.7842e+00],
[ 5.9633e-01, -3.7148e+00, -3.1149e+00, ..., -5.7071e-01,
-1.3263e+00, -7.5603e-01],
[-4.9009e-02, -1.8050e+00, 1.6011e+00, ..., -4.7655e-01,
-9.1552e-01, -1.0900e+00],
...,
[-1.0512e+00, -2.2173e+00, -2.7317e+00, ..., -1.9824e+00,
-4.0656e-02, -3.1739e-01],
[-9.0439e-01, -3.0550e-01, -1.1990e+00, ..., -1.3476e+00,
8.3994e-02, -1.5858e+00],
[-2.8495e+00, -4.9300e+00, -4.1895e+00, ..., -4.4099e+00,
-4.0328e+00, -3.7639e+00]],
[[-1.0639e+00, -2.4009e+00, -3.3943e+00, ..., -5.8110e+00,
-8.6611e+00, -2.3990e+00],
[-1.6975e+00, -3.5963e+00, -8.5839e-02, ..., -5.5709e+00,
-7.6139e+00, -2.9220e+00],
[-6.1156e+00, -9.3722e+00, -7.0299e+00, ..., -8.9000e+00,
-9.9480e+00, -5.2110e+00],
...,
[-5.5548e+00, -4.4653e+00, -5.6782e+00, ..., -5.0740e+00,
-4.5251e+00, -3.4164e+00],
[-7.5209e+00, -8.5215e+00, -6.5091e+00, ..., -7.5246e+00,
-8.2514e+00, -6.5383e+00],
[-2.5466e+00, -4.6117e+00, -4.5429e+00, ..., -3.9099e+00,
-4.0833e+00, -2.4818e+00]],
[[-5.4671e-01, 3.0894e-01, -2.1181e+00, ..., 2.0071e+00,
2.9897e+00, 1.0336e+00],
[-4.3332e+00, -3.7551e+00, -5.5004e+00, ..., 1.3158e+00,
2.9906e+00, 5.7586e-01],
[-2.6506e+00, -4.5007e+00, -7.8879e+00, ..., 2.5597e-03,
2.0179e-01, -4.6566e-01],
...,
[ 2.1516e-01, 1.0096e-03, -1.2105e-01, ..., -7.0007e-01,
-4.9682e-01, -4.3001e-02],
[ 5.8092e-01, 4.1771e-01, 1.5194e-01, ..., 4.4215e-01,
-1.4472e-01, 5.3414e-02],
[ 1.3006e+00, 1.9078e+00, 1.2205e+00, ..., 1.7912e+00,
1.0758e+00, 5.2878e-02]],
...,
[[-8.6990e-01, -2.4158e+00, -1.4911e+00, ..., -4.8862e+00,
-6.0478e+00, -1.9504e+00],
[-2.5058e+00, -5.4105e+00, -1.1145e+00, ..., -3.6671e+00,
-4.9785e+00, -2.4381e+00],
[-3.9019e+00, -3.5093e+00, -1.3664e-01, ..., -5.3045e+00,
-6.3794e+00, -2.8197e+00],
...,
[-1.9940e+00, -2.4927e+00, -2.1973e+00, ..., -1.8858e+00,
-2.7906e+00, -1.4208e+00],
[-3.1741e+00, -3.0941e+00, -1.6026e+00, ..., -2.8586e+00,
-4.4959e+00, -2.4218e+00],
[-3.6385e+00, -3.9515e+00, -3.2486e+00, ..., -3.0414e+00,
-3.5424e+00, -1.5633e+00]],
[[-2.1163e+00, -4.5429e+00, -5.8328e+00, ..., -2.2994e+00,
-1.1400e+00, 6.9501e-01],
[-2.6218e+00, -3.6680e+00, -4.3563e+00, ..., -2.9356e+00,
-2.3426e+00, -9.9741e-02],
[-1.3474e+00, 1.0569e+00, 2.9654e+00, ..., -2.0232e+00,
-1.2655e+00, 1.7492e-01],
...,
[-2.8018e+00, -2.6059e+00, -3.6038e+00, ..., -2.5612e-01,
-1.5472e+00, -7.8248e-01],
[-3.2246e+00, -1.0514e+00, -3.1605e+00, ..., -2.8907e-01,
-2.4930e+00, -1.1462e+00],
[-4.2205e+00, -2.8381e+00, -2.7383e+00, ..., -2.0195e+00,
-3.2104e+00, -1.2780e+00]],
[[ 1.5301e+00, -5.5388e-01, -2.5881e+00, ..., 1.0193e+00,
1.5769e+00, -1.4606e-01],
[ 3.5074e+00, -1.8599e+00, -2.8837e+00, ..., 3.0005e+00,
3.4050e+00, 4.0391e+00],
[ 3.7773e+00, 2.6710e+00, 3.3443e+00, ..., 2.7387e+00,
2.5305e+00, 1.7933e+00],
...,
[ 1.9491e+00, 7.1765e-01, 1.2528e+00, ..., 2.4910e+00,
2.8731e+00, 1.7184e+00],
[ 3.7981e+00, 3.9150e+00, 2.1758e+00, ..., 2.8056e+00,
1.8232e+00, 2.0600e+00],
[ 1.9676e+00, 1.0648e+00, -6.5377e-02, ..., -4.1263e-01,
-2.1189e-01, -9.9690e-02]]],
[[[ 4.0272e-01, 7.0925e-01, 9.2017e-01, ..., -5.2429e-01,
-7.0841e-02, 6.5516e-01],
[-2.0865e+00, -3.0128e+00, -4.1406e+00, ..., -9.9449e-01,
-1.2793e+00, 1.0408e+00],
[-3.4821e+00, -7.6387e-01, -3.9026e-01, ..., -1.2000e+00,
-2.4009e+00, 5.0936e-01],
...,
[-1.1657e+00, -2.2499e+00, 1.5472e-01, ..., -4.5219e+00,
-4.2059e+00, -4.3410e+00],
[-1.0453e+00, -9.3402e-01, -2.8125e-01, ..., -3.2486e+00,
-1.4205e+00, -1.5317e+00],
[-2.6390e+00, -5.1301e+00, -4.5121e+00, ..., -5.4168e+00,
-5.9321e+00, -5.0024e+00]],
[[-1.3942e+00, -1.7327e+00, -2.7805e+00, ..., -3.2899e+00,
-5.0976e+00, -3.7670e-01],
[-7.2887e+00, -8.5592e+00, -1.2571e+01, ..., -5.2242e+00,
-7.1200e+00, -1.6250e+00],
[-1.5258e+00, 7.5825e-01, 1.8055e+00, ..., -1.0528e+01,
-9.4051e+00, -2.3706e+00],
...,
[-5.7643e+00, -5.6707e+00, -3.5855e+00, ..., -2.1768e+01,
-1.7496e+01, -8.4113e+00],
[-6.9543e+00, -7.7382e+00, -6.3224e+00, ..., -6.9552e+00,
-7.6812e+00, -6.6463e+00],
[-2.5045e+00, -4.4113e+00, -4.1410e+00, ..., -5.2324e+00,
-5.5429e+00, -3.5372e+00]],
[[-1.2165e+00, 1.1572e+00, 1.3128e+00, ..., 8.5274e-01,
2.1642e+00, 9.9097e-01],
[-5.0857e-01, 1.2688e+00, 5.2769e+00, ..., 6.2788e-01,
1.1785e+00, -3.9435e-01],
[ 1.2481e+00, -1.6986e+00, -2.4227e+00, ..., -7.4922e-02,
1.7620e+00, 1.3297e-01],
...,
[ 9.6873e-01, 1.0229e+00, 1.2683e+00, ..., -1.8265e+00,
-3.7495e+00, -1.1950e+00],
[ 8.7105e-01, 6.6050e-01, 6.1317e-01, ..., 1.8832e+00,
1.3235e+00, -6.8214e-01],
[ 1.2883e+00, 1.6081e+00, 1.3209e+00, ..., 1.0398e+00,
1.7479e+00, 2.7847e-01]],
...,
[[-8.7338e-01, -2.2934e+00, -1.8980e+00, ..., -2.3983e+00,
-2.9426e+00, -5.1085e-01],
[-4.4208e+00, -4.7181e+00, -5.5606e+00, ..., -4.1473e+00,
-3.9005e+00, -2.5022e+00],
[-2.3641e+00, 2.6266e-01, 3.7659e+00, ..., -4.9906e+00,
-5.5882e+00, -1.5705e+00],
...,
[-2.8384e+00, -2.7000e+00, -1.0429e+00, ..., -1.0760e+01,
-8.9397e+00, -4.9904e+00],
[-2.9118e+00, -2.9696e+00, -1.9113e+00, ..., 5.1238e-01,
-2.9448e+00, -2.8020e+00],
[-3.4145e+00, -3.8400e+00, -3.3043e+00, ..., -4.2953e+00,
-5.1220e+00, -3.0411e+00]],
[[-1.5237e+00, -2.7894e+00, -3.9934e+00, ..., -1.6494e+00,
-1.4504e+00, -1.3354e+00],
[-3.4437e+00, -4.9641e+00, -1.0132e+01, ..., -3.4516e+00,
-1.8678e+00, -9.0046e-01],
[-5.4577e-01, 1.3727e+00, 2.9373e+00, ..., -2.0830e+00,
-2.4209e+00, -1.2031e+00],
...,
[-4.2222e+00, -2.2137e+00, -2.8485e+00, ..., -2.9573e+00,
-7.2398e+00, -1.5290e+00],
[-3.7377e+00, -8.8867e-01, -2.5488e+00, ..., -7.3336e-02,
-1.9400e+00, -1.8778e-01],
[-4.3916e+00, -3.0915e+00, -2.9266e+00, ..., -8.3913e-01,
-1.9468e+00, 5.7872e-02]],
[[ 1.3288e+00, -4.2882e-01, -1.5791e-01, ..., -5.6713e-01,
1.2269e-01, -4.2646e-01],
[ 3.4920e+00, 3.2008e+00, 3.1728e+00, ..., 1.6134e+00,
1.9266e+00, 2.5804e+00],
[-1.5299e-01, 1.9652e+00, 6.1558e-01, ..., 1.5530e+00,
1.7130e+00, 1.7227e+00],
...,
[ 2.7284e+00, 1.0093e+00, 1.2533e+00, ..., 9.0879e+00,
6.1484e+00, 3.9161e+00],
[ 3.5796e+00, 3.5399e+00, 2.2135e+00, ..., 2.3834e+00,
1.6355e+00, 2.9935e+00],
[ 1.9321e+00, 7.5787e-01, 2.4993e-02, ..., -5.3381e-02,
-2.0051e-02, -3.0413e-02]]],
[[[-1.9843e-01, 5.9657e-01, -1.8757e-01, ..., -1.9492e+00,
-1.5083e+00, -7.3043e-01],
[ 6.9910e-01, 3.4140e-01, 1.2775e-01, ..., 2.2162e-01,
-2.7660e-01, -1.1786e+00],
[ 1.0746e+00, -6.1337e-01, -8.2507e-02, ..., 1.3098e+00,
-1.5672e+00, -2.8987e+00],
...,
[-5.0547e+00, -3.9812e+00, 1.4397e+00, ..., -1.1903e+00,
-2.2257e+00, 2.2101e-01],
[-5.0675e+00, -3.2530e+00, 1.2292e+00, ..., -2.4556e+00,
-2.7682e+00, 2.5236e-01],
[-2.9444e+00, -4.8018e+00, -1.1409e+00, ..., -4.3842e+00,
-3.5061e+00, -1.4722e+00]],
[[-8.1035e-01, -1.5594e+00, -2.3887e+00, ..., -4.1500e+00,
-4.8930e+00, -2.0406e+00],
[-1.9446e+00, -3.6660e+00, -3.5540e+00, ..., -3.1745e+00,
-6.4722e+00, -4.6039e+00],
[-1.7852e+00, -3.4051e+00, -3.3673e+00, ..., -4.1051e+00,
-7.3720e+00, -4.4761e+00],
...,
[-9.6808e+00, -7.3152e+00, 3.7701e-01, ..., -8.6015e+00,
-8.5018e+00, -1.7916e+00],
[-1.0583e+01, -9.0937e+00, -1.2714e+00, ..., -1.0578e+01,
-1.0764e+01, -3.7872e+00],
[-2.6917e+00, -5.5781e+00, -4.7192e+00, ..., -4.1277e+00,
-4.1431e+00, -5.9926e-01]],
[[ 7.2497e-01, 1.2462e+00, 1.1326e+00, ..., 1.5724e+00,
1.8685e+00, 9.1404e-01],
[ 8.8423e-01, 1.4358e+00, 1.2756e+00, ..., 2.7674e+00,
2.2147e+00, 6.5068e-01],
[ 5.4073e-01, 3.1165e+00, 2.2720e+00, ..., 5.1208e+00,
3.0077e+00, 1.4558e+00],
...,
[ 2.8995e-01, -2.3840e+00, -5.5721e+00, ..., 1.3993e+00,
8.8398e-01, -3.6388e-02],
[-1.5356e+00, -4.5956e+00, -8.1082e+00, ..., -5.1677e-01,
7.1378e-01, -4.1758e-01],
[-1.0983e-01, -1.0393e+00, -3.3508e+00, ..., -2.8590e-01,
-5.2283e-01, -1.8935e+00]],
...,
[[-1.0875e+00, -1.4351e+00, -1.7277e+00, ..., -2.3857e+00,
-4.1478e+00, -1.1443e+00],
[-2.3937e+00, -3.3363e+00, -1.7570e+00, ..., -2.6217e+00,
-4.9859e+00, -4.1022e+00],
[-1.0606e+00, -2.2213e+00, -4.1471e-01, ..., -5.2063e+00,
-5.4124e+00, -4.7774e+00],
...,
[-4.4421e+00, -3.5184e+00, -4.4414e+00, ..., -4.7036e+00,
-6.0824e+00, -2.5309e+00],
[-4.0298e+00, -3.4661e+00, -2.9133e+00, ..., -3.8784e+00,
-5.2136e+00, -1.2256e+00],
[-4.0534e+00, -4.6783e+00, -4.5666e+00, ..., -2.7606e+00,
-2.5024e+00, 4.1934e-01]],
[[-1.6026e+00, -2.3113e+00, -3.3218e+00, ..., -2.3279e+00,
-2.3548e+00, -6.4596e-01],
[-4.4088e+00, -4.6880e+00, -4.4805e+00, ..., -2.0035e+00,
-5.2455e+00, -1.8935e+00],
[-4.0518e+00, -5.2126e+00, -3.9694e+00, ..., -1.9017e+00,
-3.0187e+00, -1.9517e+00],
...,
[-4.6026e+00, 1.4660e+00, -5.2286e+00, ..., -4.1804e+00,
-2.9342e+00, -1.1315e+00],
[-2.4125e+00, 4.3887e+00, 3.2805e-01, ..., -2.3936e+00,
-1.1557e+00, 3.9630e-01],
[-3.3673e+00, 2.7931e-02, 2.2006e-01, ..., -1.7161e+00,
-1.1368e+00, 3.3935e-01]],
[[ 9.1868e-01, -1.5694e-01, 6.4350e-01, ..., 3.9600e-01,
1.0359e+00, 4.3611e-01],
[ 1.7718e+00, 9.2113e-01, 4.9342e-01, ..., 1.6497e+00,
1.8846e+00, 2.5897e+00],
[ 1.3381e+00, 1.2965e-01, 2.1270e-01, ..., 4.4133e+00,
1.3933e+00, 1.0864e-01],
...,
[ 3.7368e+00, 3.8217e+00, -6.1386e-01, ..., 3.2614e+00,
2.8969e+00, 6.9740e-01],
[ 3.3424e+00, 5.3441e+00, 1.1399e-01, ..., 2.7910e+00,
1.9515e+00, 1.4883e+00],
[ 1.6850e+00, 3.2908e+00, 3.1420e+00, ..., 5.0979e-01,
4.9685e-01, 9.0753e-01]]],
[[[ 4.9394e-01, 1.4003e+00, 1.2968e-01, ..., -3.1313e+00,
-2.4315e+00, -1.0754e+00],
[ 5.8687e-01, 1.1157e+00, 1.0904e+00, ..., -1.6322e+00,
-2.2211e+00, -2.1617e-01],
[-1.5208e-01, -7.1313e-02, -8.3097e-02, ..., -1.1425e+00,
-2.7742e+00, -5.9225e-02],
...,
[-7.7715e-01, -1.9801e+00, 3.7935e-01, ..., -2.5814e+00,
-3.6167e+00, 7.4014e-01],
[-2.0977e+00, -1.8163e+00, -7.6917e-01, ..., -3.4624e+00,
-3.4698e+00, 8.6710e-01],
[-1.1260e+00, -3.3204e+00, -3.3215e+00, ..., -3.5368e+00,
-3.0662e+00, -1.1536e+00]],
[[-7.6764e-01, -1.3111e+00, -3.3759e+00, ..., -6.3714e+00,
-7.3221e+00, -1.3901e+00],
[-3.6186e+00, -4.0699e+00, -4.0733e+00, ..., -5.8619e+00,
-8.3213e+00, -2.4319e+00],
[-3.5794e+00, -4.1939e+00, -4.2023e+00, ..., -5.2502e+00,
-7.9030e+00, -2.5965e+00],
...,
[-5.9213e+00, -5.7857e+00, -4.4265e+00, ..., -6.6433e+00,
-8.1638e+00, -1.5764e+00],
[-7.7721e+00, -7.7742e+00, -7.3722e+00, ..., -9.3216e+00,
-1.0879e+01, -4.0667e+00],
[-1.7407e+00, -3.2796e+00, -3.7102e+00, ..., -3.5751e+00,
-3.7054e+00, -5.9876e-01]],
[[-1.0849e-01, 1.2068e+00, 1.6478e+00, ..., 1.5671e+00,
2.4213e+00, 9.6794e-01],
[-8.3452e-02, -1.2078e-02, 8.3599e-01, ..., 1.4792e+00,
2.9877e+00, -1.6316e-03],
[ 7.0543e-01, 1.2941e+00, 1.0592e+00, ..., 1.6493e-01,
1.9650e+00, -3.0454e-01],
...,
[ 2.6134e-01, 6.3662e-01, 1.4351e+00, ..., 2.3392e-01,
1.3424e+00, -1.2981e+00],
[ 4.2656e-01, 4.3646e-01, 1.3247e+00, ..., 8.6569e-02,
1.2438e+00, -1.5641e+00],
[ 5.7027e-01, 1.1123e+00, 1.7239e+00, ..., 3.4135e-01,
-3.4602e-01, -2.0494e+00]],
...,
[[-3.6870e-01, -1.4496e+00, -2.1435e+00, ..., -5.0948e+00,
-5.2211e+00, -7.4164e-01],
[-2.4291e+00, -2.6818e+00, -1.7046e+00, ..., -3.5651e+00,
-4.9330e+00, -1.8218e+00],
[-1.7252e+00, -1.1340e+00, -1.3091e+00, ..., -4.5983e+00,
-5.2553e+00, -1.4534e+00],
...,
[-1.8715e+00, -1.5252e+00, -1.0310e+00, ..., -3.2644e+00,
-4.9822e+00, -1.4031e+00],
[-3.3034e+00, -3.2425e+00, -2.8342e+00, ..., -3.8745e+00,
-5.3094e+00, -8.0082e-01],
[-1.5269e+00, -2.0035e+00, -2.4462e+00, ..., -1.9378e+00,
-1.9465e+00, 4.9671e-01]],
[[-1.0752e+00, -2.5446e+00, -3.9559e+00, ..., -1.4883e+00,
-4.9682e-01, -5.8781e-01],
[-3.3391e+00, -3.6129e+00, -4.6714e+00, ..., -3.2582e+00,
-1.7462e+00, -3.4073e-01],
[-4.3600e+00, -3.9983e+00, -4.3180e+00, ..., -2.5859e+00,
-1.7675e+00, -3.6991e-01],
...,
[-4.7292e+00, -3.5230e+00, -3.4795e+00, ..., -2.2949e+00,
-1.3339e+00, -9.4423e-01],
[-4.1869e+00, -1.6940e+00, -2.4367e+00, ..., -8.5254e-01,
-5.7188e-02, 1.2074e+00],
[-3.2728e+00, -3.1408e+00, -2.3567e+00, ..., 7.7800e-02,
-4.8802e-01, 7.4627e-01]],
[[ 1.1463e+00, 6.9091e-04, 1.1225e+00, ..., 5.5457e-01,
7.9950e-01, -1.4297e+00],
[ 2.0735e+00, 1.0741e+00, 1.4649e+00, ..., 3.7359e+00,
3.0651e+00, 2.2951e+00],
[ 1.3825e+00, 8.8182e-01, 6.9237e-01, ..., 3.1640e+00,
1.7428e+00, 1.3229e+00],
...,
[ 2.4488e+00, 1.5091e+00, 1.0187e+00, ..., 3.4193e+00,
2.3984e+00, 6.1290e-01],
[ 2.9080e+00, 3.8644e+00, 2.6534e+00, ..., 3.1145e+00,
1.0225e+00, 1.8571e+00],
[ 1.9703e+00, 1.7135e+00, 5.5883e-01, ..., -1.8738e-01,
-2.6951e-01, 8.5957e-01]]],
[[[ 3.1503e-01, 9.2352e-01, -3.9229e-01, ..., -1.1574e+00,
-7.1050e-01, -8.3096e-01],
[ 9.4192e-01, 4.8142e-01, -3.5839e-01, ..., -4.4992e-01,
1.9031e-01, -1.2257e+00],
[ 1.0445e+00, -5.9557e-01, -4.3327e-02, ..., -7.2069e-01,
-3.7230e-01, -2.0688e+00],
...,
[-2.2029e-01, -6.8358e-01, -5.3065e-01, ..., -1.2968e+00,
-2.5917e+00, -1.3645e-01],
[-2.2691e+00, -1.5333e+00, 7.5638e-01, ..., -2.4556e+00,
-2.8005e+00, 1.0743e-01],
[-2.3257e+00, -3.8935e+00, -2.8795e+00, ..., -4.1844e+00,
-3.5063e+00, -1.5620e+00]],
[[-3.9696e-01, -1.2669e+00, -2.4767e+00, ..., -2.8706e+00,
-4.6583e+00, -2.5299e+00],
[-1.7653e+00, -3.4776e+00, -3.7603e+00, ..., -3.1440e+00,
-5.0823e+00, -4.6899e+00],
[-1.7773e+00, -3.2159e+00, -3.1513e+00, ..., -2.6811e+00,
-4.0202e+00, -3.6293e+00],
...,
[-6.2486e+00, -8.0543e+00, -6.9867e+00, ..., -7.8016e+00,
-8.8894e+00, -2.0023e+00],
[-9.7055e+00, -1.0281e+01, -8.3342e+00, ..., -1.0571e+01,
-1.0716e+01, -3.7918e+00],
[-2.4641e+00, -4.8685e+00, -4.3147e+00, ..., -4.1902e+00,
-4.1841e+00, -6.5179e-01]],
[[ 5.4441e-01, 1.1182e+00, 1.0233e+00, ..., 1.5722e+00,
1.8238e+00, 8.5081e-01],
[ 5.8613e-01, 1.3210e+00, 1.2403e+00, ..., 1.0522e+00,
2.1717e+00, 8.8711e-01],
[ 6.1675e-01, 2.9286e+00, 2.1386e+00, ..., 1.1806e+00,
2.2057e+00, 1.8461e+00],
...,
[-1.3063e-01, 5.2135e-01, 7.4407e-01, ..., 1.4059e+00,
1.5342e+00, -1.9218e-01],
[-6.5647e-01, 6.4322e-01, -5.0648e-01, ..., 6.5736e-01,
1.1945e+00, -3.6299e-01],
[ 5.3595e-01, 1.3246e+00, 8.6448e-01, ..., -1.3982e-01,
-5.7475e-01, -1.9059e+00]],
...,
[[-6.6266e-01, -1.2656e+00, -1.9812e+00, ..., -2.5587e+00,
-3.3171e+00, -1.1197e+00],
[-2.0861e+00, -3.1623e+00, -2.3650e+00, ..., -2.7927e+00,
-4.4622e+00, -3.5786e+00],
[-8.7511e-01, -1.6609e+00, -3.7925e-01, ..., -2.4892e+00,
-4.1548e+00, -3.0369e+00],
...,
[-2.4012e+00, -4.1988e+00, -2.5204e+00, ..., -5.4470e+00,
-6.4180e+00, -2.5520e+00],
[-3.6739e+00, -5.8777e+00, -2.5571e+00, ..., -4.2457e+00,
-5.1985e+00, -1.2637e+00],
[-3.1577e+00, -3.7699e+00, -3.2064e+00, ..., -2.6968e+00,
-2.5907e+00, 3.7477e-01]],
[[-1.3126e+00, -2.6486e+00, -3.5951e+00, ..., -2.8374e+00,
-2.7797e+00, -6.1994e-01],
[-4.0452e+00, -5.0464e+00, -4.7851e+00, ..., -3.7493e+00,
-5.9042e+00, -2.0875e+00],
[-4.0440e+00, -5.1046e+00, -4.0942e+00, ..., -2.5054e+00,
-4.8412e+00, -2.1574e+00],
...,
[-3.9904e+00, -3.1701e+00, -3.8174e+00, ..., -3.9978e+00,
-3.0328e+00, -1.2828e+00],
[-3.4733e+00, -9.3163e-01, -1.9153e+00, ..., -2.4027e+00,
-1.4157e+00, 2.7404e-01],
[-3.6857e+00, -3.0562e+00, -2.9642e+00, ..., -1.6782e+00,
-1.2266e+00, 3.2906e-01]],
[[ 8.2873e-01, -7.4030e-02, 7.8515e-01, ..., 4.9216e-01,
1.5888e+00, 6.5201e-01],
[ 1.5327e+00, 4.5367e-01, 2.8029e-01, ..., 6.5230e-01,
1.8671e+00, 3.1706e+00],
[ 1.4488e+00, -1.9843e-01, -1.3218e-01, ..., 5.4106e-01,
8.8069e-01, 7.0883e-01],
...,
[ 3.9432e+00, 3.1572e+00, 2.6355e+00, ..., 3.0630e+00,
2.6695e+00, 6.7774e-01],
[ 4.6110e+00, 4.3915e+00, 3.2711e+00, ..., 2.6025e+00,
1.8605e+00, 1.5254e+00],
[ 2.7528e+00, 2.8682e+00, 2.1566e+00, ..., 5.1239e-01,
6.0043e-01, 1.0304e+00]]]], device='cuda:0'),
tensor([[[[ 4.9906e-01, 7.6499e-01, -7.7080e-01, ..., 3.0738e-01,
4.6914e-01, -1.8057e-01],
[ 3.2678e-01, 1.7831e-01, 7.0282e-01, ..., 2.8253e+00,
7.9382e-01, -8.4903e-01],
[ 1.9517e+00, 7.3011e-01, 2.8064e+00, ..., 2.1409e+00,
-6.3053e-01, -1.5715e+00],
...,
[-1.7579e-02, -1.0623e+00, 2.8742e-01, ..., 3.1874e-01,
6.5051e-02, -2.9080e-01],
[-1.8869e+00, -1.7048e+00, 1.0923e-02, ..., 3.0782e-01,
-2.1210e-01, -1.2148e+00],
[-1.2966e+00, -3.9929e+00, -3.2000e+00, ..., -2.5182e+00,
-3.0105e+00, -2.8198e+00]],
[[-4.4653e-01, -3.2616e+00, -6.3534e+00, ..., -1.6984e+00,
-3.8880e+00, -2.0204e+00],
[-4.7801e+00, -4.1961e+00, -4.8409e+00, ..., -4.6749e-01,
-5.6836e+00, -3.5164e+00],
[-1.8582e+00, -1.3093e+00, -3.1730e+00, ..., -1.5457e+00,
-4.8484e+00, -2.8123e+00],
...,
[-5.6973e+00, -7.4416e+00, -7.9372e+00, ..., -2.5373e+00,
-3.3176e+00, -2.5862e+00],
[-8.3540e+00, -9.7009e+00, -1.0037e+01, ..., -3.9582e+00,
-5.3153e+00, -5.1276e+00],
[-1.8065e+00, -4.6928e+00, -4.5829e+00, ..., -2.8994e+00,
-2.8171e+00, -1.6560e+00]],
[[ 7.2695e-01, 3.0385e+00, 1.8370e+00, ..., 2.2466e+00,
1.5779e+00, 8.3919e-01],
[ 5.5799e-01, 2.9574e+00, 1.9702e+00, ..., 2.5048e+00,
1.2975e+00, 5.3120e-01],
[-4.4268e-01, 1.6984e+00, 1.7949e+00, ..., 2.4290e+00,
1.6029e+00, 1.0489e+00],
...,
[ 1.0348e+00, 1.9712e+00, 1.4057e+00, ..., 7.6301e-01,
8.8150e-01, 5.6139e-01],
[ 5.9738e-01, 1.7644e+00, 6.3653e-01, ..., 9.1199e-01,
9.9679e-01, 3.1630e-01],
[ 3.7049e-01, 1.7765e+00, 8.8439e-01, ..., 1.3340e+00,
1.8172e+00, 5.5677e-01]],
...,
[[-1.2175e+00, -3.2507e+00, -2.1324e+00, ..., -1.9525e+00,
-2.4048e+00, -5.9612e-01],
[-3.0634e+00, -4.6480e+00, -1.6213e+00, ..., -3.3464e+00,
-3.3313e+00, -3.4728e+00],
[-2.2273e-01, -1.1178e+00, -7.8959e-01, ..., -4.4713e+00,
-2.3900e+00, -2.2371e+00],
...,
[-2.2999e+00, -3.4021e+00, -1.4663e+00, ..., -7.5222e-01,
-2.4924e+00, -8.0660e-01],
[-3.3392e+00, -5.1812e+00, -2.2698e+00, ..., -9.6320e-01,
-3.0962e+00, -1.6256e+00],
[-2.3273e+00, -3.6022e+00, -1.6529e+00, ..., -1.8625e+00,
-2.4701e+00, -5.9962e-01]],
[[-2.2514e+00, -3.2350e+00, -1.2028e+00, ..., 2.6062e-01,
-1.5157e+00, -8.1252e-01],
[-4.1452e+00, -4.2920e+00, -1.7377e+00, ..., -1.5455e+00,
-3.3060e+00, -2.0800e+00],
[-2.3332e+00, -2.4692e+00, -2.0344e+00, ..., -3.8243e+00,
-5.2260e+00, -2.7107e+00],
...,
[-4.2303e+00, -3.6436e+00, -1.4891e+00, ..., -5.0015e-01,
-3.5243e+00, -2.1297e+00],
[-4.4406e+00, -1.3592e+00, -2.2819e-01, ..., -2.6930e-01,
-3.2119e+00, -1.8477e+00],
[-4.1278e+00, -3.0760e+00, -1.3281e-01, ..., -7.0704e-01,
-2.5963e+00, -1.2294e+00]],
[[ 1.0845e+00, 1.3744e+00, 1.2931e+00, ..., 1.7591e+00,
1.1922e+00, 5.2131e-01],
[ 2.6219e+00, 2.6804e+00, 3.2061e+00, ..., 3.4247e+00,
1.0198e+00, 2.2144e+00],
[ 2.5847e+00, -8.2583e-02, 2.7333e+00, ..., 2.4112e+00,
-1.1830e+00, 6.9350e-01],
...,
[ 3.7353e+00, 3.4342e+00, 3.7802e+00, ..., 1.4986e+00,
3.6578e-01, 5.6446e-01],
[ 4.3072e+00, 4.4770e+00, 3.9121e+00, ..., 1.7787e+00,
7.4840e-01, 1.2066e+00],
[ 2.5072e+00, 1.8838e+00, 1.9143e+00, ..., 6.4833e-01,
-2.8542e-01, -1.3553e-01]]],
[[[ 2.1975e-01, 1.5408e+00, 1.9956e+00, ..., -5.0627e+00,
-3.1943e+00, -1.4423e+00],
[ 8.0500e-01, -1.7601e+00, -1.5884e+00, ..., 3.0541e-01,
2.3079e+00, 2.3661e+00],
[-4.0442e+00, -1.0156e+01, -7.8581e+00, ..., -6.5924e+00,
2.7837e+00, 2.4697e+00],
...,
[-4.0934e+00, -5.0288e+00, -7.8459e-02, ..., 2.3245e+00,
3.1061e-01, 2.5014e-01],
[-2.4573e+00, -1.5421e+00, -9.0632e-01, ..., -1.6460e+00,
9.5832e-01, -3.6337e-01],
[-2.3637e-01, -3.0081e-01, -1.4542e+00, ..., 1.7811e+00,
1.4632e+00, -2.0887e-01]],
[[-2.3373e+00, -4.1538e+00, -7.4469e+00, ..., -8.5469e+00,
-5.1113e+00, -6.6697e-01],
[-8.6176e+00, -8.0558e+00, -4.9439e+00, ..., 7.2085e-02,
4.8185e-01, -5.4993e-01],
[-1.2239e+01, -1.7939e+01, -1.1614e+01, ..., -1.0833e+01,
1.1605e+00, 5.5800e-01],
...,
[-5.0024e+00, -6.4516e-01, 1.2122e+00, ..., -9.0020e-01,
-7.0087e+00, -1.5372e+00],
[-1.7729e+00, 7.2055e-01, 1.1521e-01, ..., 5.6910e-01,
-3.6568e+00, -3.5335e+00],
[ 5.6447e-01, 8.1008e-01, -1.6781e+00, ..., 2.3611e+00,
1.4720e-01, -6.5845e-01]],
[[ 9.1692e-01, 1.9883e+00, 1.6891e+00, ..., 4.1136e+00,
4.3669e+00, 1.9227e+00],
[ 1.9825e-01, 4.8879e-01, 8.5194e-01, ..., -3.2222e+00,
1.3142e+00, 2.2610e+00],
[-8.6552e-03, 3.9224e+00, 5.6437e+00, ..., -1.0336e+01,
-3.9493e+00, 3.6505e+00],
...,
[ 4.1037e-01, -1.0567e+00, -5.6530e+00, ..., -5.1017e+00,
1.3672e-01, -9.4972e-01],
[-6.4131e-01, -3.7897e+00, -4.1775e+00, ..., -6.7926e+00,
3.8300e+00, 9.8004e-01],
[-4.7661e-01, -1.6408e+00, -2.4876e+00, ..., -2.4573e+00,
-7.2893e-01, -4.8736e-01]],
...,
[[-1.4919e+00, -1.5673e+00, -2.6512e+00, ..., -4.7494e+00,
-4.9260e+00, -1.5305e+00],
[-5.4082e+00, -5.5353e+00, -3.9964e+00, ..., 3.6454e+00,
6.5922e-01, -1.1415e+00],
[-7.5518e+00, -7.3134e+00, -7.3755e+00, ..., 1.5211e-01,
-1.4886e+00, -3.9679e+00],
...,
[-2.3728e+00, 9.7497e-01, -6.8735e-01, ..., -2.6139e-02,
-3.6281e+00, -4.0615e+00],
[-1.0326e+00, 3.2800e-01, -2.2081e+00, ..., -5.4129e+00,
-5.7761e+00, -2.4542e+00],
[ 1.1918e-01, -1.0836e+00, -2.7860e+00, ..., -2.1895e+00,
-4.6312e+00, 9.7972e-01]],
[[-1.5599e+00, -2.8724e-01, -3.1502e+00, ..., -8.5824e-01,
4.6313e-02, 1.0580e+00],
[-3.5328e+00, -1.7064e-01, -4.0670e+00, ..., 5.2524e+00,
4.6060e-01, 5.6296e-01],
[-7.1196e+00, -5.3126e+00, -8.5473e+00, ..., 5.7017e+00,
5.6401e-01, -1.1894e+00],
...,
[-5.6272e-01, 6.7653e-01, 2.4408e-02, ..., 3.4610e+00,
-1.0536e-01, -1.0021e-01],
[-1.5388e+00, -3.1890e+00, -6.7035e+00, ..., -3.9595e+00,
-1.3828e+00, 1.0151e+00],
[-5.2458e-01, -4.0111e+00, -4.5734e+00, ..., 2.0973e+00,
-2.3492e+00, 1.3422e+00]],
[[ 1.9897e+00, 7.0603e-01, 2.1749e+00, ..., 1.4790e-01,
1.8388e+00, -4.3343e-01],
[ 5.1078e+00, 2.7591e+00, 1.0259e+00, ..., -1.2147e+00,
2.5538e+00, 4.8909e+00],
[ 4.1869e+00, 2.1566e+00, 2.7421e+00, ..., -5.8895e+00,
1.6104e+00, 3.7441e+00],
...,
[ 9.1135e-01, 5.5714e-01, 3.0528e+00, ..., 2.7134e+00,
6.2724e+00, 1.7486e+00],
[ 6.8868e-01, 8.3732e-01, -4.0642e+00, ..., -7.4544e-01,
5.3599e+00, 3.5735e+00],
[ 5.7673e-01, -1.8177e-01, -2.2676e+00, ..., -1.4201e+00,
7.1078e-01, 1.9879e+00]]],
[[[ 3.9359e-01, 1.3463e+00, 8.6859e-02, ..., -3.1668e+00,
-2.4061e+00, -1.2727e+00],
[ 4.5224e-01, 1.0177e+00, 1.2746e+00, ..., -1.6985e+00,
-1.9291e+00, -2.5857e-01],
[-3.1786e-01, -3.8772e-01, -2.5738e-01, ..., -1.0017e+00,
-2.6947e+00, -3.4087e-01],
...,
[-5.3868e-01, -1.4559e+00, -6.5316e-01, ..., -1.7983e+00,
-3.7153e+00, 2.7548e-01],
[-2.0201e+00, -1.3742e+00, -1.0632e+00, ..., -3.0391e+00,
-3.5175e+00, 3.8449e-01],
[-1.1022e+00, -3.2246e+00, -3.2481e+00, ..., -3.5456e+00,
-3.1313e+00, -1.3733e+00]],
[[-9.1654e-01, -1.3986e+00, -3.4523e+00, ..., -6.6656e+00,
-7.7395e+00, -1.6863e+00],
[-3.6753e+00, -3.9601e+00, -3.3696e+00, ..., -6.0931e+00,
-8.4045e+00, -2.6708e+00],
[-3.7272e+00, -4.2528e+00, -3.9018e+00, ..., -5.2489e+00,
-7.8154e+00, -2.8893e+00],
...,
[-5.9884e+00, -6.0705e+00, -5.0182e+00, ..., -6.3839e+00,
-8.8028e+00, -2.1227e+00],
[-7.8460e+00, -7.9713e+00, -7.7889e+00, ..., -9.2040e+00,
-1.0911e+01, -4.3255e+00],
[-1.7709e+00, -3.3534e+00, -4.0312e+00, ..., -3.4374e+00,
-3.6903e+00, -7.2912e-01]],
[[-1.1228e-01, 1.2565e+00, 1.7189e+00, ..., 1.4463e+00,
2.5337e+00, 1.0214e+00],
[-1.2145e-01, -1.3645e-01, 6.7118e-01, ..., 1.4134e+00,
3.0311e+00, 8.7977e-02],
[ 6.9167e-01, 1.2013e+00, 1.2351e+00, ..., 2.6693e-01,
2.0851e+00, -2.5662e-01],
...,
[ 2.6350e-01, 7.5457e-01, 1.2981e+00, ..., 2.4199e-01,
1.5023e+00, -1.2025e+00],
[ 4.0531e-01, 5.4979e-01, 7.5284e-01, ..., 2.1319e-01,
1.5564e+00, -1.3485e+00],
[ 4.9005e-01, 1.2990e+00, 1.7726e+00, ..., 2.7881e-03,
-4.6109e-01, -1.8741e+00]],
...,
[[-4.6783e-01, -1.4925e+00, -2.1216e+00, ..., -5.1721e+00,
-5.4774e+00, -9.0841e-01],
[-2.5535e+00, -2.5795e+00, -1.3545e+00, ..., -3.7584e+00,
-4.9153e+00, -1.9689e+00],
[-1.9392e+00, -1.3723e+00, -1.1086e+00, ..., -4.4891e+00,
-5.4263e+00, -1.6044e+00],
...,
[-1.8442e+00, -1.4794e+00, -1.3516e+00, ..., -3.0514e+00,
-5.5004e+00, -1.5892e+00],
[-3.3655e+00, -3.2809e+00, -2.3198e+00, ..., -3.8925e+00,
-5.2211e+00, -8.9667e-01],
[-1.5163e+00, -2.1058e+00, -2.4841e+00, ..., -1.7514e+00,
-2.1218e+00, 4.0442e-01]],
[[-1.1510e+00, -2.4930e+00, -3.8538e+00, ..., -1.5856e+00,
-8.9605e-01, -7.1509e-01],
[-3.3873e+00, -3.4072e+00, -4.6087e+00, ..., -3.1503e+00,
-2.1569e+00, -7.0831e-01],
[-4.4446e+00, -3.9111e+00, -4.3267e+00, ..., -2.1329e+00,
-2.0473e+00, -8.5397e-01],
...,
[-4.7112e+00, -4.3636e+00, -4.2427e+00, ..., -2.8786e+00,
-1.8836e+00, -9.2228e-01],
[-4.0750e+00, -2.2575e+00, -2.5757e+00, ..., -1.3325e+00,
-4.1630e-01, 1.1697e+00],
[-3.1283e+00, -3.3476e+00, -2.5830e+00, ..., -5.0322e-01,
-8.5963e-01, 7.0194e-01]],
[[ 1.2094e+00, -2.9749e-02, 1.0796e+00, ..., 5.2406e-01,
9.8928e-01, -1.3503e+00],
[ 2.0852e+00, 1.0527e+00, 1.3491e+00, ..., 3.6663e+00,
3.1024e+00, 2.5451e+00],
[ 1.3537e+00, 8.3828e-01, 4.6789e-01, ..., 3.2443e+00,
1.7723e+00, 1.3467e+00],
...,
[ 2.5497e+00, 1.5750e+00, 1.0381e+00, ..., 3.5239e+00,
2.0864e+00, 6.6412e-01],
[ 2.9246e+00, 4.1979e+00, 2.3113e+00, ..., 2.4852e+00,
1.0219e+00, 1.8737e+00],
[ 2.0051e+00, 1.8053e+00, 6.8993e-01, ..., 3.2562e-01,
-1.9289e-02, 8.6638e-01]]],
[[[-3.8497e+00, -1.7731e+00, 1.2769e+00, ..., -2.8736e+00,
-2.2703e+00, -1.4944e+00],
[-1.5014e+00, 1.1006e-01, -1.8579e-01, ..., -4.4283e+00,
-2.7874e+00, -1.4184e-01],
[ 2.0856e-01, 5.7854e-01, -3.9971e+00, ..., -5.8479e-01,
-2.9829e+00, -1.8443e+00],
...,
[-2.2265e+00, 8.3945e-02, 1.1724e+00, ..., 4.1069e-01,
-2.9171e-02, 9.7697e-01],
[-1.6224e+00, -4.8265e-01, 1.6747e+00, ..., -2.9204e+00,
-2.7855e+00, 3.1496e-01],
[-7.2925e-01, -4.0324e+00, -7.7083e-01, ..., -3.7215e+00,
-2.9557e+00, -1.1828e+00]],
[[-3.1693e+00, -1.9107e+00, 2.6665e-01, ..., -8.4762e+00,
-8.6037e+00, -2.6411e+00],
[-8.9096e-01, 1.2732e+00, -1.4050e+00, ..., -6.2312e+00,
-6.2160e+00, -2.0002e+00],
[ 1.1217e+00, 4.6492e-01, -2.8172e+00, ..., -3.4808e+00,
-4.6810e+00, -1.9638e+00],
...,
[-5.6148e+00, -1.8470e+00, 3.2846e-01, ..., -5.1485e+00,
-7.0178e+00, -2.5616e+00],
[-6.1294e+00, -5.7863e+00, 8.2698e-01, ..., -9.5792e+00,
-1.1158e+01, -4.2131e+00],
[-6.2871e-01, -5.3425e+00, -3.5403e+00, ..., -3.7035e+00,
-3.4989e+00, -6.1817e-01]],
[[ 2.4606e+00, 2.1817e+00, 5.0084e-01, ..., 9.9866e-01,
2.5092e+00, 8.3315e-01],
[ 1.7181e+00, -1.8780e+00, -3.4886e+00, ..., 3.2303e+00,
4.3806e+00, 4.6165e-01],
[-1.3768e+00, -8.3029e-01, 1.2213e-01, ..., 2.9058e+00,
5.1733e+00, 2.7008e+00],
...,
[ 1.7759e+00, -3.0048e-02, 7.8504e-01, ..., -2.9869e+00,
-1.5088e+00, 3.2343e-02],
[ 7.9422e-01, -2.3093e+00, -1.1656e+00, ..., -2.1695e+00,
6.5546e-02, -1.3809e+00],
[-3.9303e-01, -5.4074e-01, -1.6631e+00, ..., -1.7660e-01,
-3.6893e-01, -2.2919e+00]],
...,
[[-3.3834e+00, 6.7461e-01, 9.8612e-01, ..., -5.5979e+00,
-5.4865e+00, -9.3386e-01],
[-1.3889e+00, 3.0634e-01, -1.8162e+00, ..., -4.4260e+00,
-4.7076e+00, -2.2824e+00],
[ 1.0060e+00, -9.1905e-01, -1.7847e+00, ..., -2.2751e+00,
-4.0061e+00, -2.3971e+00],
...,
[-2.5553e+00, -1.5593e+00, -1.2787e+00, ..., -1.5733e+00,
-3.6167e+00, -1.8141e+00],
[-2.3079e+00, -1.4317e+00, -7.9117e-01, ..., -2.6510e+00,
-4.6637e+00, -7.6594e-01],
[-1.8908e+00, -3.1750e+00, -1.4214e+00, ..., -1.0241e+00,
-1.5885e+00, 6.1025e-01]],
[[-1.9552e+00, 2.0462e-02, 1.2915e-01, ..., -2.2572e+00,
-1.5904e+00, -1.0104e+00],
[-3.9855e+00, -3.7992e+00, -5.9846e-01, ..., -4.2633e+00,
-1.6108e+00, 5.9268e-01],
[-1.1331e+00, -4.0520e+00, -3.3835e+00, ..., -3.4548e+00,
-5.4369e+00, -1.9044e+00],
...,
[-2.5198e+00, -1.6558e+00, -3.7504e+00, ..., -4.2738e-01,
-1.4253e+00, -2.0103e+00],
[-1.9519e+00, 3.4041e+00, 2.3779e-01, ..., -2.6356e-02,
-4.4657e-01, 8.8932e-01],
[-3.8648e+00, -1.2714e-01, 2.5016e-01, ..., 6.9194e-01,
-7.2743e-01, 5.6009e-01]],
[[ 6.1014e-01, 1.7199e+00, 9.1235e-01, ..., 6.6707e-01,
1.4985e+00, -8.0049e-01],
[-4.9422e-01, 1.7668e+00, -1.1537e+00, ..., 1.5631e+00,
2.5646e+00, 2.4955e+00],
[-1.3830e+00, -2.2474e+00, -5.0506e+00, ..., 2.2945e+00,
1.9928e+00, 3.1732e+00],
...,
[ 3.4882e+00, 5.4008e-01, -2.8323e+00, ..., 2.5631e+00,
2.9273e+00, 7.6940e-01],
[ 2.9053e+00, 4.0128e+00, -1.0693e+00, ..., 1.0724e+00,
3.8106e-01, 1.9367e+00],
[ 1.7442e+00, 1.8923e+00, 1.5051e+00, ..., 1.2232e-02,
-1.7247e-01, 1.2014e+00]]],
[[[ 1.7960e-01, 1.1919e+00, 8.1376e-01, ..., -4.8761e-01,
4.0670e-02, 6.4109e-01],
[ 5.9710e-01, 9.2433e-01, 1.3101e+00, ..., -9.2278e-01,
-1.1968e+00, 9.4004e-01],
[ 3.9656e-01, 1.5084e+00, -1.0171e+00, ..., -1.1681e+00,
-8.0366e-01, 4.9026e-01],
...,
[ 5.4553e-01, -6.2169e-01, -4.7425e+00, ..., -6.1551e-01,
-7.7122e-01, -1.1755e+00],
[-1.4635e+00, -1.3917e+00, -7.0257e+00, ..., -2.2218e+00,
-2.7877e+00, -1.3599e+00],
[-2.0837e-01, -2.6026e+00, -6.3187e+00, ..., -2.5883e+00,
-2.3951e+00, -8.5863e-01]],
[[-8.4325e-01, -2.3277e+00, -4.1508e+00, ..., -3.2232e+00,
-5.0357e+00, -4.2669e-01],
[-4.8270e+00, -6.3138e+00, -8.4188e+00, ..., -5.3890e+00,
-7.3440e+00, -1.7848e+00],
[-3.7688e+00, -5.5511e+00, -1.2457e+01, ..., -8.3298e+00,
-8.4981e+00, -2.9973e+00],
...,
[-5.1646e+00, -5.4502e+00, -1.1300e+01, ..., -4.6487e+00,
-5.3126e+00, -2.3235e+00],
[-9.2023e+00, -7.8398e+00, -7.9067e+00, ..., -8.3135e+00,
-9.1551e+00, -4.0011e+00],
[-1.2634e+00, -1.9515e+00, -3.3573e+00, ..., -3.0676e+00,
-2.7554e+00, -2.5538e-01]],
[[-3.8498e-01, 2.1383e+00, 1.4241e+00, ..., 8.9995e-01,
2.2106e+00, 9.4948e-01],
[-1.3795e-01, 1.3894e+00, 7.8374e-01, ..., 5.6528e-01,
1.2243e+00, -4.3764e-01],
[ 1.0496e-01, 1.8568e+00, 7.1255e-01, ..., -1.3067e-01,
1.5024e+00, -1.2663e-01],
...,
[ 2.3368e-01, -6.7110e-02, 1.7445e+00, ..., 1.3202e+00,
1.1207e+00, 5.9552e-01],
[ 1.2184e-01, -2.8124e-01, 2.7647e+00, ..., 1.9067e+00,
2.0794e+00, 7.0018e-01],
[-6.4427e-01, -2.4009e-01, 2.0639e+00, ..., 5.4896e-01,
8.6759e-01, -8.1220e-01]],
...,
[[-7.6584e-01, -1.7207e+00, -1.8065e+00, ..., -2.4062e+00,
-2.9229e+00, -4.8879e-01],
[-2.9642e+00, -4.0521e+00, -2.9752e+00, ..., -3.7396e+00,
-4.0209e+00, -2.5484e+00],
[-1.2401e+00, -2.7208e+00, -3.0248e+00, ..., -2.9939e+00,
-4.8006e+00, -1.5217e+00],
...,
[-8.8926e-01, -3.0058e+00, -4.5301e+00, ..., -1.8449e+00,
-3.8742e+00, -1.8480e+00],
[-2.8743e+00, -4.0709e+00, -4.4384e+00, ..., -2.9467e+00,
-4.6937e+00, -1.4320e+00],
[-5.1774e-01, -1.2641e+00, -1.1418e+00, ..., -5.4773e-01,
-1.1937e+00, 1.2685e+00]],
[[-1.6729e+00, -2.2201e+00, -2.7392e+00, ..., -1.7088e+00,
-1.5267e+00, -1.3346e+00],
[-4.0893e+00, -3.6927e+00, -4.2323e+00, ..., -3.4765e+00,
-2.0688e+00, -8.8166e-01],
[-2.4746e+00, -2.8514e+00, -5.6979e+00, ..., -1.8483e-01,
-7.9006e-01, -4.8153e-01],
...,
[-3.4071e+00, -5.2602e+00, -3.0650e+00, ..., -1.9161e+00,
-2.7779e+00, -1.1990e+00],
[-4.3747e+00, -2.5044e+00, -1.2421e+00, ..., -1.2398e+00,
-2.3636e+00, -6.1166e-01],
[-2.7012e+00, -2.3839e+00, -4.3864e-01, ..., -1.9362e-01,
-8.5047e-01, 6.7584e-02]],
[[ 1.0213e+00, 5.4177e-01, 7.0670e-01, ..., -5.6277e-01,
1.5575e-01, -4.3612e-01],
[ 3.5699e+00, 3.0926e+00, 3.6421e+00, ..., 1.7880e+00,
2.0788e+00, 2.5059e+00],
[ 1.5939e+00, 2.0880e+00, 3.7260e+00, ..., 2.0263e+00,
2.3607e+00, 1.7652e+00],
...,
[ 2.8589e+00, 2.3804e+00, 1.5816e+00, ..., 1.2874e+00,
1.1351e+00, 4.9824e-01],
[ 3.7117e+00, 4.0901e+00, -3.3625e-01, ..., 2.1779e+00,
1.2793e+00, 1.7667e+00],
[ 2.5401e+00, 1.5529e+00, 2.5967e-01, ..., 3.7291e-01,
1.3073e-01, 1.0402e+00]]]], device='cuda:0'),
tensor([[[[ 7.0226e-01, 2.7771e-01, -5.3094e-01, ..., -2.1645e+00,
-2.9054e+00, -1.7610e+00],
[-5.1723e-01, -2.0082e+00, 2.2106e+00, ..., -6.4133e+00,
-3.4083e+00, -4.4704e-01],
[-1.8536e+00, -3.0792e+00, 7.7267e-03, ..., 2.2639e-01,
-1.2612e+00, -3.3365e-03],
...,
[-1.4441e+00, -8.8679e+00, -3.5177e+00, ..., -2.2214e+00,
-4.4004e+00, 4.4686e-01],
[-2.5899e+00, -5.8975e+00, -2.0707e+00, ..., -2.8235e+00,
-3.7053e+00, 4.1461e-01],
[-1.2928e+00, -4.3376e+00, -2.8433e+00, ..., -3.5442e+00,
-3.0294e+00, -1.4060e+00]],
[[-1.3538e-01, -1.2090e+00, -1.9814e+00, ..., -7.2229e+00,
-9.1257e+00, -2.2596e+00],
[-9.8423e-01, -2.5513e+00, 1.8569e-01, ..., -7.1284e+00,
-8.2137e+00, -2.7394e+00],
[-7.9308e-01, -3.0058e+00, -5.6281e-01, ..., -2.8184e-01,
-4.5158e+00, -1.9615e+00],
...,
[-5.7111e+00, -8.5481e+00, -8.7505e-01, ..., -8.6604e+00,
-8.6604e+00, -2.3073e+00],
[-6.7486e+00, -7.2944e+00, -2.8063e+00, ..., -8.7335e+00,
-1.0845e+01, -4.6183e+00],
[-1.1526e+00, -2.4757e+00, -3.3613e+00, ..., -2.8160e+00,
-3.7109e+00, -6.9684e-01]],
[[-1.5283e-01, 1.2929e+00, 2.6125e+00, ..., 1.4077e+00,
2.5071e+00, 8.2787e-01],
[-7.3998e-01, -3.1566e+00, -1.2993e+00, ..., 6.0732e+00,
4.8180e+00, 1.8941e-01],
[ 1.6060e-01, -8.8880e-01, -1.6729e+00, ..., 4.1022e+00,
6.0238e+00, 1.8780e+00],
...,
[-3.1278e-01, 1.6616e+00, 3.0782e-01, ..., 9.1159e-01,
1.1655e+00, -6.5220e-01],
[ 4.7702e-01, 6.9088e-01, -1.7971e+00, ..., -5.0194e-01,
1.2212e+00, -1.2800e+00],
[ 3.9852e-01, 1.2404e+00, 1.7012e-01, ..., -1.1089e+00,
-4.4516e-01, -1.8794e+00]],
...,
[[-6.9459e-02, -5.4949e-01, -1.3437e+00, ..., -4.9749e+00,
-5.2340e+00, -1.5263e+00],
[-1.6904e+00, -9.3129e-02, 1.1623e+00, ..., -6.3638e+00,
-5.9328e+00, -2.9011e+00],
[-9.7432e-01, -4.0464e-01, 1.3716e+00, ..., -1.2464e+00,
-4.4630e+00, -3.1756e+00],
...,
[-2.6897e+00, -3.5676e+00, -4.7185e-01, ..., -4.0091e+00,
-6.3215e+00, -2.0124e+00],
[-3.1186e+00, -3.0251e+00, -1.0285e+00, ..., -2.9900e+00,
-5.4968e+00, -1.0395e+00],
[-1.0747e+00, -2.0255e+00, -2.6923e+00, ..., -2.3499e+00,
-2.3833e+00, 4.6219e-01]],
[[-9.6079e-01, -1.8452e+00, -4.3085e+00, ..., -1.6056e+00,
-1.4022e+00, -1.0525e+00],
[-2.1644e+00, -8.9277e-01, -1.9893e+00, ..., -5.6967e+00,
-1.2578e+00, -2.4373e-01],
[-2.9705e+00, -1.2989e+00, -7.4648e-01, ..., -3.8535e+00,
-3.0606e+00, 8.9772e-01],
...,
[-4.8550e+00, -1.8823e+00, -1.6232e+00, ..., -3.0020e+00,
-1.2312e+00, -6.5296e-01],
[-4.0893e+00, -3.3505e-01, 2.4842e+00, ..., -3.4983e-01,
2.2084e-01, 1.2392e+00],
[-2.9733e+00, -2.4980e+00, 1.1445e+00, ..., -5.3793e-03,
-5.1386e-01, 5.8125e-01]],
[[ 1.4962e+00, -6.6569e-01, 8.0807e-01, ..., 1.4258e+00,
1.0055e+00, -1.5406e+00],
[ 9.1767e-01, -1.1076e+00, -1.0046e-01, ..., 3.8312e+00,
3.8124e+00, 2.4614e+00],
[-1.6947e+00, -3.4511e+00, -1.8924e+00, ..., 2.0948e+00,
3.7220e+00, 2.6286e+00],
...,
[ 2.6126e+00, 5.3459e-01, -1.7391e+00, ..., 3.9701e+00,
2.7479e+00, 3.4330e-01],
[ 2.6211e+00, 1.7581e+00, 1.4012e+00, ..., 3.3227e+00,
1.0908e+00, 1.7527e+00],
[ 1.5366e+00, 3.8660e-01, 2.1312e+00, ..., 1.6562e+00,
2.6498e-01, 9.1816e-01]]],
[[[-2.9896e+00, -1.4038e+00, -1.3245e+00, ..., 4.1125e+00,
2.9966e+00, 1.4472e+00],
[-7.9486e+00, -6.6148e+00, -5.9905e+00, ..., 3.4641e+00,
-3.0973e-01, -6.8475e-01],
[-5.5592e+00, -9.2135e+00, -1.0512e+01, ..., 1.7928e+00,
-4.2790e+00, -2.6015e+00],
...,
[-1.3922e+00, 2.0348e-01, -3.0496e-01, ..., -2.6597e+00,
-1.6669e+00, -1.0472e+00],
[-1.4287e+00, -3.3146e+00, -6.3785e+00, ..., -2.4716e+00,
-1.1097e+00, 4.2609e-02],
[-2.4692e-01, -1.3565e+00, -1.6796e+00, ..., -5.9741e+00,
-4.3818e+00, -3.9129e+00]],
[[-1.3060e-02, -1.7111e+00, -1.2741e+00, ..., -4.3631e+00,
-7.3995e+00, -7.3049e-02],
[-5.7388e-01, -4.7087e+00, -5.2780e+00, ..., -8.9668e+00,
-1.9700e+01, -3.6918e+00],
[ 9.4432e-01, -3.7497e-01, -1.7298e+00, ..., -1.5028e+01,
-2.5664e+01, -4.7595e+00],
...,
[ 1.6079e+00, 3.2383e+00, 4.1947e+00, ..., -1.5356e+01,
-1.0100e+01, -8.0678e+00],
[-2.0811e-01, -2.5422e+00, -4.0165e+00, ..., -3.8320e+00,
-3.4003e+00, -4.0921e+00],
[ 9.8522e-01, -4.3716e-01, -8.4186e-01, ..., -6.0979e+00,
-5.3636e+00, -2.7276e+00]],
[[ 1.4648e+00, 3.4133e-01, -2.1794e+00, ..., 2.8570e+00,
-2.3128e-01, -7.6667e-02],
[ 9.3801e-01, 2.0637e+00, 1.6451e+00, ..., 4.7113e+00,
-2.0158e+00, -2.8632e+00],
[-1.6239e+00, 5.2081e-01, 1.0660e+00, ..., 7.4346e+00,
-6.0368e-01, -1.9166e+00],
...,
[ 2.5951e-01, -2.4669e+00, -4.5363e+00, ..., -3.2463e+00,
-1.6730e+00, -1.8630e+00],
[-1.2939e+00, -1.9559e+00, -4.5352e+00, ..., -1.2917e+00,
-2.4265e+00, -1.3550e+00],
[-4.7629e-01, 2.7559e-01, 8.1764e-01, ..., -3.4910e-02,
-6.3173e-01, -6.2790e-01]],
...,
[[ 5.8198e-02, 1.7901e+00, 1.8434e+00, ..., -3.9627e+00,
-1.0026e+00, -1.5320e+00],
[-4.0476e-01, -4.0873e+00, -4.9827e+00, ..., -1.0767e+01,
-4.3069e+00, -7.0521e+00],
[ 8.0476e-02, 6.5243e-01, -2.6251e+00, ..., -1.2668e+01,
-4.2716e+00, -8.3953e+00],
...,
[ 2.7361e+00, 4.1719e+00, 5.4568e+00, ..., -7.9223e+00,
-5.2290e+00, -2.7654e+00],
[-4.9422e-01, -2.5666e+00, -4.3157e+00, ..., -2.6134e+00,
-4.2342e+00, -1.9371e+00],
[ 9.8838e-01, -8.5057e-02, -2.0306e-01, ..., -5.4044e+00,
-5.1083e+00, -3.8511e+00]],
[[-4.4520e+00, -4.5408e+00, 2.8185e-01, ..., 5.4497e-01,
-2.7119e+00, -2.6268e+00],
[-5.8307e+00, -1.2116e+01, -9.3051e+00, ..., -7.4657e+00,
-3.1902e+00, -1.4445e+00],
[-1.1000e+00, -5.8067e+00, -1.0847e+01, ..., -1.1928e+01,
-5.8311e+00, -2.6522e+00],
...,
[ 1.0944e+00, 4.3815e+00, 7.0032e+00, ..., -3.8644e+00,
-5.2168e+00, -1.1493e+00],
[-2.7568e+00, -1.6407e+00, 1.4839e+00, ..., 4.0272e+00,
1.8162e+00, 1.1019e+00],
[-4.4218e-01, -1.8223e+00, -6.9081e-01, ..., -5.3618e-01,
-2.3095e+00, -5.3944e-01]],
[[-1.1272e+00, -7.9481e-01, -5.9023e-02, ..., 5.3297e+00,
1.1287e+00, -1.4822e+00],
[-4.7581e+00, -6.5734e+00, -6.3460e+00, ..., 1.0723e+01,
3.0383e+00, -3.6120e-02],
[-2.5284e+00, -4.7290e+00, -5.7681e+00, ..., 1.1188e+01,
7.3738e-01, -2.5609e-01],
...,
[-1.0135e+00, -2.5833e-02, 6.0886e-01, ..., 5.9506e+00,
3.5155e+00, 3.9109e+00],
[-1.1124e+00, -3.7387e+00, -7.3014e+00, ..., 2.3215e+00,
1.3810e+00, 2.0536e+00],
[-8.0421e-01, -1.0081e+00, -1.5125e+00, ..., -1.0808e+00,
-1.1296e+00, -9.7844e-01]]],
[[[-8.3063e-02, 7.2775e-01, -2.1353e-01, ..., -1.3461e+00,
-1.0069e+00, -7.6870e-01],
[ 7.3240e-01, 3.2660e-01, -2.0508e-01, ..., -4.8925e-01,
4.3903e-01, -7.6191e-01],
[ 1.1565e+00, -2.0481e-01, 5.0808e-01, ..., -8.6213e-01,
-5.5501e-01, -1.8863e+00],
...,
[-6.5351e-01, -1.0516e+00, 9.1192e-01, ..., -1.2599e+00,
-2.4575e+00, -3.8948e-02],
[-2.3789e+00, -1.8300e+00, 7.6411e-01, ..., -2.2121e+00,
-2.7432e+00, 2.0461e-01],
[-2.3257e+00, -4.0182e+00, -2.9435e+00, ..., -4.0746e+00,
-3.5306e+00, -1.5211e+00]],
[[-7.2039e-01, -1.4336e+00, -2.4399e+00, ..., -2.8755e+00,
-4.3956e+00, -2.1997e+00],
[-1.9145e+00, -3.7342e+00, -3.9215e+00, ..., -3.3558e+00,
-4.5160e+00, -4.1608e+00],
[-1.6570e+00, -3.5899e+00, -4.0034e+00, ..., -3.5631e+00,
-4.2747e+00, -3.5161e+00],
...,
[-6.6891e+00, -8.6581e+00, -7.3260e+00, ..., -7.7068e+00,
-8.6572e+00, -1.8954e+00],
[-9.5778e+00, -9.9202e+00, -8.0664e+00, ..., -1.0693e+01,
-1.0736e+01, -3.7821e+00],
[-2.5151e+00, -4.9513e+00, -4.1708e+00, ..., -4.0512e+00,
-4.1863e+00, -6.3543e-01]],
[[ 6.5859e-01, 1.1622e+00, 1.0049e+00, ..., 1.4448e+00,
1.8003e+00, 1.0227e+00],
[ 8.3312e-01, 1.5100e+00, 1.4413e+00, ..., 1.0546e+00,
1.9685e+00, 9.1799e-01],
[ 4.7399e-01, 3.2488e+00, 2.1922e+00, ..., 1.3373e+00,
2.1602e+00, 1.8115e+00],
...,
[ 1.0871e-02, 2.8787e-01, -9.7212e-01, ..., 1.7684e+00,
1.6626e+00, -1.3288e-01],
[-6.5678e-01, 5.8365e-01, -1.4556e-01, ..., 3.2987e-01,
1.0861e+00, -3.9623e-01],
[ 5.2056e-01, 1.0620e+00, 9.0870e-01, ..., -3.1313e-01,
-4.7676e-01, -1.9304e+00]],
...,
[[-9.8524e-01, -1.2700e+00, -1.8992e+00, ..., -2.5630e+00,
-3.2319e+00, -1.2419e+00],
[-2.2953e+00, -3.2664e+00, -2.4133e+00, ..., -3.0067e+00,
-3.9434e+00, -3.1984e+00],
[-8.4293e-01, -1.8068e+00, -4.7317e-01, ..., -2.4072e+00,
-4.2932e+00, -3.0000e+00],
...,
[-2.8773e+00, -4.2596e+00, -1.4622e+00, ..., -5.3632e+00,
-6.1537e+00, -2.5206e+00],
[-3.6309e+00, -5.9150e+00, -2.7350e+00, ..., -4.1014e+00,
-5.2360e+00, -1.1820e+00],
[-3.2256e+00, -3.7580e+00, -3.1452e+00, ..., -2.5966e+00,
-2.6205e+00, 4.6498e-01]],
[[-1.5231e+00, -2.2968e+00, -3.5251e+00, ..., -2.8146e+00,
-2.8022e+00, -6.3196e-01],
[-4.3591e+00, -4.8544e+00, -4.6780e+00, ..., -3.7574e+00,
-5.6440e+00, -1.9766e+00],
[-4.0283e+00, -5.3018e+00, -3.9024e+00, ..., -2.6462e+00,
-4.8093e+00, -2.0296e+00],
...,
[-4.3303e+00, -2.4978e+00, -4.1091e+00, ..., -3.7346e+00,
-2.8214e+00, -1.1874e+00],
[-3.5062e+00, -4.7639e-01, -1.6145e+00, ..., -2.2600e+00,
-1.2365e+00, 4.1816e-01],
[-3.7212e+00, -2.9594e+00, -2.8845e+00, ..., -1.7895e+00,
-1.2490e+00, 3.6741e-01]],
[[ 9.0515e-01, -1.7008e-01, 6.4186e-01, ..., 3.7684e-01,
1.3053e+00, 5.9630e-01],
[ 1.7631e+00, 8.1375e-01, 3.3478e-01, ..., 6.3459e-01,
1.6865e+00, 3.1237e+00],
[ 1.3745e+00, 6.5295e-03, 4.0840e-01, ..., 5.9362e-01,
7.9159e-01, 7.7204e-01],
...,
[ 3.9710e+00, 4.0975e+00, 3.1232e+00, ..., 3.0858e+00,
2.7714e+00, 6.3001e-01],
[ 4.4980e+00, 4.3509e+00, 3.5960e+00, ..., 2.5519e+00,
1.8608e+00, 1.4467e+00],
[ 2.8141e+00, 2.7178e+00, 2.0967e+00, ..., 5.8996e-01,
5.2348e-01, 1.0057e+00]]],
[[[ 2.3221e+00, 5.1836e-01, 2.3864e+00, ..., 1.4557e-01,
-2.3115e+00, -2.0676e+00],
[-1.3969e-01, -3.4868e+00, 1.1012e+00, ..., 2.9273e+00,
1.5084e+00, 5.3163e-01],
[-4.4411e+00, -7.3982e+00, -7.2073e+00, ..., 1.2276e+00,
2.0563e+00, 1.0949e+00],
...,
[-4.2954e-01, -2.4049e+00, -1.2133e+00, ..., 5.8384e+00,
5.5447e+00, 5.5605e+00],
[-7.9782e-01, -2.7256e-01, 4.0775e-01, ..., -4.5925e+00,
-7.4216e+00, -4.7460e+00],
[-2.6362e+00, -5.2470e+00, -4.3235e+00, ..., -6.3669e+00,
-8.1678e+00, -6.5374e+00]],
[[-7.3681e-01, -3.3365e+00, 1.7474e+00, ..., -1.2020e+00,
-5.8042e+00, -1.7907e+00],
[-2.5455e-01, -3.6617e+00, 1.9494e+00, ..., 1.1960e+00,
-3.4117e+00, -2.7627e+00],
[-9.7662e-01, -2.5906e+00, -5.6831e-01, ..., -1.6792e+00,
-4.6983e+00, -4.5786e+00],
...,
[-5.4565e+00, -5.6249e+00, -5.9769e+00, ..., -8.9174e+00,
-7.1809e+00, -1.5687e+00],
[-8.0360e+00, -9.3993e+00, -6.5808e+00, ..., -2.0162e+01,
-2.5173e+01, -1.4398e+01],
[-2.3447e+00, -4.9970e+00, -4.4673e+00, ..., -4.2399e+00,
-5.3973e+00, -2.1762e+00]],
[[-1.0813e+00, -2.6615e+00, -3.3309e+00, ..., 1.6608e+00,
4.2542e+00, 2.0275e+00],
[-2.1558e+00, -4.9168e+00, -8.6949e+00, ..., -1.3094e+00,
2.5126e+00, 1.7775e+00],
[ 1.1253e+00, 3.7259e-01, -3.2310e+00, ..., -3.7852e+00,
-2.2140e+00, -2.2007e+00],
...,
[-4.2638e-01, 6.2104e-01, 8.6460e-01, ..., -8.1544e+00,
-6.3139e+00, -1.6407e+00],
[ 1.4345e-01, -1.8103e-01, -4.5229e-01, ..., -5.6906e+00,
-7.3293e+00, -8.1699e+00],
[ 1.4804e+00, 1.7383e+00, 7.7048e-01, ..., 2.6437e+00,
2.5508e+00, -1.0785e+00]],
...,
[[ 2.4231e+00, 5.0133e+00, 5.8632e+00, ..., -2.0482e+00,
-5.6360e+00, -2.1021e+00],
[ 1.2391e+00, 2.2699e+00, 3.3724e+00, ..., -5.1439e-01,
-3.5237e+00, -2.1152e-01],
[-1.8837e+00, -1.9171e+00, -1.5749e+00, ..., -1.2100e+00,
-3.6093e+00, -8.7384e-01],
...,
[-1.4798e+00, -2.7989e+00, -2.9993e+00, ..., -8.3341e-01,
-1.3170e+00, -2.9952e-01],
[-2.9476e+00, -2.9347e+00, -4.5873e-01, ..., -1.3302e+01,
-1.5923e+01, -8.8970e+00],
[-3.8150e+00, -4.6446e+00, -3.5036e+00, ..., -1.5905e+00,
-2.6666e+00, -9.5358e-01]],
[[-8.0349e-01, 9.7079e-01, -1.3041e+00, ..., -3.1801e+00,
-1.7052e+00, 1.0329e+00],
[-1.4982e+00, 2.0003e-01, -3.6032e+00, ..., 6.9021e-01,
-2.1569e+00, 9.6043e-02],
[-7.4679e+00, -1.0527e+01, -1.4192e+01, ..., 1.3962e+00,
-1.5776e+00, 6.2700e-01],
...,
[-2.5129e+00, -1.8616e+00, -1.6080e+00, ..., 2.1216e+00,
3.6090e+00, 3.2907e+00],
[-3.2225e+00, 2.9679e-01, -1.4112e+00, ..., -3.7275e+00,
-4.5007e+00, 1.3147e+00],
[-4.6566e+00, -2.6590e+00, -2.1369e+00, ..., -4.5938e+00,
-6.0699e+00, -6.8937e-01]],
[[ 1.0737e+00, -1.5861e+00, -2.4340e-01, ..., 1.3995e+00,
2.7166e+00, -4.6720e-01],
[-6.1760e-01, -4.1295e+00, -4.6191e+00, ..., 1.4032e+00,
3.3853e+00, 4.6326e+00],
[-2.0167e+00, -5.7854e+00, -9.0012e+00, ..., 1.4859e+00,
2.2666e+00, 2.0872e+00],
...,
[ 3.0630e+00, 1.3851e+00, 3.6926e+00, ..., 6.8268e+00,
4.9638e+00, 3.6380e+00],
[ 4.9989e+00, 4.5000e+00, 2.4077e+00, ..., 1.3602e+01,
1.2521e+01, 6.9858e+00],
[ 2.5433e+00, 1.4738e+00, 7.4085e-01, ..., -3.2301e+00,
-4.2842e+00, -7.7781e-01]]],
[[[ 5.5546e-01, 4.7829e-01, 2.0208e-01, ..., 1.9811e-03,
2.9036e-01, -6.3489e-02],
[ 8.7628e-01, 1.6311e-01, 2.1611e+00, ..., 6.7307e-01,
6.2137e-01, -4.5948e-01],
[-8.2053e-01, -6.4469e-01, 2.0029e+00, ..., 2.7644e-01,
-1.3423e-01, -8.5348e-01],
...,
[-2.6346e+00, -6.3410e-01, 2.7410e+00, ..., -2.3421e-01,
3.1492e-01, -3.0748e-01],
[-4.0003e+00, 2.8509e-01, 1.7033e+00, ..., -8.9800e-01,
4.1910e-01, -9.0059e-01],
[-5.8346e-01, -1.3805e+00, -2.0940e+00, ..., -2.8280e+00,
-2.8912e+00, -2.6078e+00]],
[[-1.7420e+00, -3.8639e+00, -6.2125e+00, ..., -2.5066e+00,
-3.3875e+00, -2.0343e+00],
[-5.1267e+00, -5.6843e+00, -3.7303e+00, ..., -3.9159e+00,
-3.7910e+00, -3.5024e+00],
[-4.3190e+00, -5.5650e+00, -3.8499e+00, ..., -4.6504e+00,
-2.8383e+00, -2.7135e+00],
...,
[-6.6716e+00, -3.9560e+00, -3.8888e+00, ..., -2.8414e+00,
-3.2308e+00, -2.2426e+00],
[-7.4620e+00, -5.7091e+00, -6.3432e+00, ..., -4.9279e+00,
-4.2523e+00, -4.5535e+00],
[-8.7805e-01, -2.9709e+00, -3.4461e+00, ..., -3.4539e+00,
-2.5470e+00, -1.5960e+00]],
[[ 7.0659e-01, 2.7336e+00, 2.4464e+00, ..., 1.8236e+00,
1.5046e+00, 8.5540e-01],
[-1.9094e-02, 9.6945e-01, 8.0118e-01, ..., 1.3565e+00,
1.1834e+00, 8.4666e-01],
[ 8.7806e-01, 2.3448e+00, 3.3402e-01, ..., 1.2856e+00,
1.4620e+00, 1.4363e+00],
...,
[ 6.4522e-01, -3.9670e-01, 1.8346e+00, ..., 1.8930e+00,
3.1566e-01, 7.5736e-01],
[ 9.5584e-01, -1.8849e+00, 7.9623e-01, ..., 1.5293e+00,
1.0750e-01, 6.4940e-01],
[ 2.0414e-02, -6.1599e-01, 9.0569e-01, ..., 1.0303e+00,
1.2947e+00, 6.9726e-01]],
...,
[[-1.0645e+00, -2.7455e+00, -3.3736e+00, ..., -1.5625e+00,
-2.0720e+00, -4.2185e-01],
[-3.1087e+00, -3.3040e+00, 2.9993e-02, ..., -1.9443e+00,
-4.0908e+00, -2.9648e+00],
[-2.6056e+00, -2.5965e+00, -6.3819e-02, ..., -6.8127e-01,
-3.9398e+00, -1.7500e+00],
...,
[-1.6027e+00, -1.7352e-02, -1.9306e+00, ..., -3.3516e+00,
-1.5100e+00, -1.3487e+00],
[-2.5815e+00, -1.0490e+00, -3.0526e+00, ..., -1.7578e+00,
-2.3141e+00, -1.6392e+00],
[-6.0799e-01, -1.6472e+00, -2.2062e+00, ..., -1.3485e+00,
-2.5007e+00, -7.4819e-01]],
[[-2.5027e+00, -2.5345e+00, -3.9015e+00, ..., 8.6115e-02,
-1.2798e+00, -5.8820e-01],
[-4.3853e+00, -2.0464e+00, -3.2212e+00, ..., -5.7017e-01,
-3.8069e+00, -1.8257e+00],
[-4.1972e+00, -1.8800e+00, -2.4134e+00, ..., -1.7869e-01,
-4.4159e+00, -2.5143e+00],
...,
[-3.9167e+00, -2.1397e+00, -2.7145e+00, ..., -2.9999e+00,
-3.6215e+00, -1.7607e+00],
[-3.2269e+00, 6.8222e-01, -1.7732e+00, ..., -1.0117e+00,
-2.7968e+00, -1.4731e+00],
[-2.7589e+00, -1.3428e+00, -8.3203e-01, ..., -7.7754e-01,
-2.3324e+00, -9.6967e-01]],
[[ 1.6726e+00, 1.0793e+00, 1.6091e+00, ..., 5.2384e-01,
9.7738e-01, 6.3897e-01],
[ 3.3281e+00, 2.6198e+00, 2.5224e+00, ..., 1.1862e+00,
7.6924e-01, 2.4666e+00],
[ 1.4893e+00, 8.8455e-01, 2.2614e+00, ..., 5.7682e-01,
-1.7018e-01, 6.6918e-01],
...,
[ 1.7909e+00, 1.0118e+00, 2.8667e+00, ..., 1.2056e+00,
2.5683e-01, 5.5220e-01],
[ 1.0403e+00, 4.2160e+00, 3.2451e+00, ..., 1.3609e+00,
8.0678e-01, 1.0283e+00],
[ 1.0778e+00, 2.4234e+00, 1.3297e+00, ..., 8.6565e-01,
3.1754e-01, -1.3275e-01]]]], device='cuda:0'),
tensor([[[[ 6.1220e-01, 9.8290e-01, 5.1602e-01, ..., 9.5841e-01,
-7.3330e-01, -1.4813e+00],
[ 6.4199e-01, -7.2865e-01, -3.3296e-02, ..., -2.9142e+00,
-6.5344e+00, -1.8457e+00],
[ 1.2695e+00, -1.0460e+00, -5.1431e+00, ..., 1.3677e+00,
1.7160e+00, 2.1243e+00],
...,
[ 5.0570e-01, -1.1096e+00, -1.6032e+00, ..., -6.0871e+00,
-6.6470e+00, -1.7402e+00],
[-1.4790e+00, -4.2861e-01, -1.4068e+00, ..., -5.9104e+00,
-5.3747e+00, 4.9638e-01],
[-2.1331e+00, -3.8119e+00, -2.3592e+00, ..., -4.5417e+00,
-3.2335e+00, -8.2141e-01]],
[[-1.6283e+00, -3.9159e+00, -6.6479e+00, ..., -2.8874e-02,
-6.9981e+00, -3.4518e+00],
[-4.6259e+00, -5.9823e+00, -1.3151e+01, ..., -7.8210e+00,
-1.4467e+01, -1.9340e+00],
[-6.9442e+00, -1.1641e+01, -1.1912e+01, ..., -8.1710e-01,
-4.8573e-01, -5.5633e-01],
...,
[-4.5659e+00, -4.1058e+00, -8.6682e+00, ..., -1.0842e+01,
-1.3485e+01, -1.9081e+00],
[-8.9304e+00, -9.0942e+00, -3.9613e+00, ..., -9.2553e+00,
-1.0192e+01, -4.0304e+00],
[-2.1108e+00, -4.2056e+00, -3.5987e+00, ..., -3.4595e+00,
-2.9641e+00, -5.9360e-01]],
[[ 1.8483e-01, 1.5836e+00, -4.8731e-02, ..., 1.2127e-01,
2.3904e+00, 4.2705e-01],
[ 4.4628e-02, -9.0096e-01, -4.1539e-01, ..., 3.5535e+00,
6.7426e+00, 2.7574e+00],
[ 1.3939e+00, 1.0126e+00, 3.4963e+00, ..., -5.1030e-01,
6.8097e+00, 4.1056e+00],
...,
[-2.3021e-01, 4.4071e-01, 1.3732e+00, ..., 5.5794e+00,
-7.7562e-01, -4.9367e+00],
[ 2.9593e-01, -1.4803e+00, 5.2788e-01, ..., -1.0283e+00,
-1.7072e+00, -1.2258e+00],
[ 7.1168e-01, 9.9711e-03, -1.6586e+00, ..., -2.8880e+00,
-1.9642e+00, -1.7882e+00]],
...,
[[-5.8022e-01, -2.2513e+00, -3.0654e+00, ..., -2.1939e+00,
-2.6351e+00, -1.5760e+00],
[-2.3496e+00, -4.1504e+00, -6.4271e+00, ..., -1.1417e+01,
-1.0494e+01, -5.5484e+00],
[-2.2429e+00, -4.9488e+00, -7.8892e+00, ..., 2.0511e+00,
2.0560e+00, -2.9135e+00],
...,
[-4.6614e-01, -1.9256e+00, -4.9464e+00, ..., -7.9212e+00,
-5.0279e+00, -2.5158e+00],
[-3.2613e+00, -2.3312e+00, -1.1672e+00, ..., -4.5325e+00,
-4.4139e+00, -2.1134e+00],
[-2.4779e+00, -2.3670e+00, -2.2606e+00, ..., -2.6885e+00,
-2.8140e+00, 2.9801e-01]],
[[-1.8624e+00, -5.1349e-01, -1.3572e+00, ..., 8.7225e-01,
2.5343e+00, 1.0220e+00],
[-4.5008e+00, -1.1816e+00, -2.2675e+00, ..., -3.8302e+00,
-1.5674e-01, 2.5470e+00],
[-6.2640e+00, -4.5627e+00, -8.7963e-01, ..., 3.6440e+00,
-3.0769e+00, 4.2590e-01],
...,
[-2.7582e+00, -6.3346e+00, -6.2963e+00, ..., -8.7851e+00,
-1.9558e+00, 3.6370e-02],
[-3.0517e+00, 3.1370e-01, -2.2095e+00, ..., -4.5964e+00,
-2.4668e-01, 1.8130e+00],
[-3.2133e+00, -5.7607e-01, 5.5848e-02, ..., -3.6719e-02,
-3.9744e-01, 8.6996e-01]],
[[ 1.5938e+00, 3.2384e-01, 2.5722e+00, ..., 3.8880e+00,
2.8238e+00, -6.5697e-01],
[ 2.1995e+00, 1.9836e+00, 5.7104e+00, ..., 3.4946e+00,
2.6830e+00, 4.7858e+00],
[ 2.3957e+00, 2.9954e+00, 5.0442e+00, ..., 6.2019e-01,
3.3984e+00, 6.9262e+00],
...,
[ 2.7571e+00, 1.5173e+00, 6.5228e+00, ..., 5.9889e-01,
2.2257e+00, 2.9455e+00],
[ 4.0846e+00, 3.7430e+00, 1.5151e+00, ..., 1.9273e+00,
1.5554e+00, 1.6179e+00],
[ 1.9282e+00, 1.1938e+00, 5.8020e-01, ..., 2.1220e+00,
2.7308e-01, 7.2961e-01]]],
[[[ 8.9692e-01, 7.5278e-01, -7.6272e-01, ..., 5.1929e-02,
1.0390e-01, -1.6875e-01],
[ 1.4795e+00, 9.2570e-01, 9.2581e-01, ..., 5.8239e-01,
-3.2483e-02, -6.0169e-01],
[-1.4625e+00, -1.5513e+00, 1.2270e+00, ..., 7.2273e-01,
-4.3022e-01, -1.3312e+00],
...,
[-1.6302e-01, -2.7250e+00, -1.1893e+00, ..., -1.0134e+00,
-2.0327e-01, -3.0462e-01],
[-2.0037e+00, -1.0613e+00, -1.0659e-01, ..., -2.4090e-01,
5.6482e-02, -1.0903e+00],
[-1.1876e+00, -3.8048e+00, -3.2870e+00, ..., -2.7219e+00,
-2.8884e+00, -2.6378e+00]],
[[-1.5224e+00, -4.0652e+00, -6.6638e+00, ..., -3.8949e+00,
-4.1973e+00, -2.1995e+00],
[-2.9452e+00, -5.3760e+00, -6.7820e+00, ..., -3.5089e+00,
-5.3708e+00, -3.3019e+00],
[-4.1234e+00, -2.8631e+00, -3.7385e+00, ..., -2.4122e+00,
-4.1635e+00, -2.7426e+00],
...,
[-7.2165e+00, -1.0187e+01, -8.5925e+00, ..., -3.2866e+00,
-2.8534e+00, -2.5536e+00],
[-7.1341e+00, -8.3415e+00, -9.5999e+00, ..., -4.4665e+00,
-5.1436e+00, -4.8876e+00],
[-1.6473e+00, -4.0702e+00, -4.2953e+00, ..., -3.0599e+00,
-2.7607e+00, -1.5475e+00]],
[[ 6.7741e-01, 2.8349e+00, 1.6995e+00, ..., 1.9325e+00,
1.5450e+00, 8.8498e-01],
[-4.3301e-01, 1.5181e+00, 1.5128e+00, ..., 2.0222e+00,
1.2305e+00, 6.8323e-01],
[ 1.7397e+00, 4.3766e+00, 1.5462e+00, ..., 4.1117e+00,
1.8794e+00, 1.1393e+00],
...,
[-3.8448e-01, -8.1339e-02, 1.6223e+00, ..., 1.7376e-01,
2.6363e-01, 1.0402e+00],
[ 9.7491e-01, 2.4607e+00, -1.3495e-02, ..., 8.2385e-01,
9.2558e-01, 7.8363e-01],
[ 4.6571e-01, 6.6174e-01, 5.2574e-01, ..., 1.3162e+00,
1.5249e+00, 4.5840e-01]],
...,
[[-7.7105e-01, -3.1962e+00, -2.5195e+00, ..., -1.3753e+00,
-2.7773e+00, -4.1746e-01],
[-2.8133e+00, -4.4679e+00, -1.4838e+00, ..., -2.6512e+00,
-4.1329e+00, -3.3473e+00],
[-3.3738e+00, -3.4252e+00, -8.3517e-01, ..., -2.2850e+00,
-3.1972e+00, -2.4821e+00],
...,
[-3.4750e+00, -4.3438e+00, -1.9106e+00, ..., -8.6982e-01,
-2.7736e+00, -1.4402e+00],
[-2.0346e+00, -2.8713e+00, -2.0448e+00, ..., -1.2589e+00,
-3.1022e+00, -1.7467e+00],
[-1.9626e+00, -3.2069e+00, -1.5468e+00, ..., -1.9454e+00,
-2.3633e+00, -6.1013e-01]],
[[-2.5477e+00, -3.2026e+00, -1.6780e+00, ..., 4.1463e-01,
-1.4368e+00, -7.9189e-01],
[-3.9237e+00, -3.4121e+00, -2.5266e+00, ..., 5.6816e-01,
-3.6636e+00, -2.1376e+00],
[-4.8312e+00, -4.1259e+00, -6.5414e-01, ..., -8.7603e-01,
-2.8314e+00, -2.8284e+00],
...,
[-3.5811e+00, -4.8821e+00, -3.1258e+00, ..., -3.2059e-01,
-2.8317e+00, -1.8961e+00],
[-3.7076e+00, -9.3753e-01, -1.4614e-01, ..., 5.5248e-01,
-2.6919e+00, -1.9348e+00],
[-3.5499e+00, -2.3078e+00, 8.5821e-02, ..., -7.2408e-01,
-2.3675e+00, -1.3315e+00]],
[[ 1.6520e+00, 1.3042e+00, 1.5619e+00, ..., 1.1341e+00,
1.1012e+00, 6.4844e-01],
[ 2.6238e+00, 3.2818e+00, 3.6367e+00, ..., 2.0639e+00,
1.1537e+00, 2.2724e+00],
[ 3.1177e-01, 2.2048e-01, 2.1674e+00, ..., 2.6278e+00,
1.8089e-01, 6.1288e-01],
...,
[ 4.0416e+00, 2.7170e+00, 1.2963e+00, ..., 1.3923e+00,
1.0693e+00, 5.6558e-01],
[ 2.5490e+00, 3.9507e+00, 4.3877e+00, ..., 1.5925e+00,
6.0395e-01, 1.0659e+00],
[ 2.6129e+00, 2.1206e+00, 1.4969e+00, ..., 3.6120e-01,
-2.5159e-01, -9.1009e-02]]],
[[[-2.4492e-01, 1.0869e+00, 1.3981e+00, ..., -1.0240e+00,
-3.5459e+00, -1.1668e+00],
[ 4.7650e-02, 9.9530e-02, 2.8446e+00, ..., 7.0565e-01,
-2.0476e+00, -6.6660e-01],
[-1.2934e+00, -3.6790e-01, 1.9031e+00, ..., 7.6473e-01,
-1.9789e+00, -9.3655e-01],
...,
[-2.5927e-01, -2.0303e-01, -2.6946e-01, ..., -2.6744e+00,
-1.9061e+00, 2.6115e-01],
[-1.5636e+00, 6.5205e-01, 1.2764e+00, ..., -3.6321e+00,
-3.9183e+00, -3.5573e-01],
[-1.4776e+00, -2.4576e+00, -1.2591e+00, ..., -4.0175e+00,
-4.2299e+00, -1.9422e+00]],
[[-1.6389e+00, -2.4821e+00, -3.1807e+00, ..., -5.1166e+00,
-9.0164e+00, -2.2657e+00],
[-4.9338e+00, -5.0684e+00, -4.1984e+00, ..., -5.3259e+00,
-8.5904e+00, -3.4702e+00],
[-3.9094e+00, -4.1826e+00, -4.7224e+00, ..., -5.2390e+00,
-7.4929e+00, -3.0360e+00],
...,
[-4.8232e+00, -5.1119e+00, -4.9918e+00, ..., -1.0283e+01,
-1.2159e+01, -2.5599e+00],
[-7.7500e+00, -8.3150e+00, -7.7000e+00, ..., -7.9524e+00,
-9.4932e+00, -4.4182e+00],
[-1.4563e+00, -3.1107e+00, -2.5633e+00, ..., -4.2546e+00,
-4.4215e+00, -6.1904e-01]],
[[-2.4955e-01, 2.1455e+00, 1.3554e+00, ..., 2.0498e+00,
2.5298e+00, 7.9548e-01],
[-5.0869e-01, 6.3836e-02, 6.6347e-02, ..., 1.1726e+00,
2.3230e+00, 5.5012e-02],
[-4.0386e-02, 1.8619e+00, 1.1788e+00, ..., -3.2291e-01,
1.2159e+00, -3.2034e-01],
...,
[-5.2694e-01, -2.4369e-01, 4.1378e-01, ..., 4.2404e-01,
-2.9004e+00, -1.8458e+00],
[-1.0764e+00, -1.2292e-01, -2.4564e-01, ..., 1.0313e+00,
2.0198e-01, -5.5759e-01],
[-2.0307e-01, -9.3977e-02, -3.2942e-02, ..., 3.2714e-01,
-4.3852e-01, -1.7177e+00]],
...,
[[-1.3329e+00, -1.9537e+00, -4.8361e-01, ..., -3.4685e+00,
-5.7830e+00, -1.3952e+00],
[-3.6601e+00, -3.5316e+00, -1.4731e+00, ..., -2.3848e+00,
-5.7006e+00, -2.3093e+00],
[-2.7793e+00, -2.3839e+00, -1.2870e+00, ..., -2.6484e+00,
-6.4488e+00, -1.6670e+00],
...,
[-1.1875e+00, -2.6835e+00, -9.1516e-01, ..., -5.0204e+00,
-7.8242e+00, -2.7210e+00],
[-2.2609e+00, -3.6914e+00, -4.0917e-01, ..., -1.8128e+00,
-3.7936e+00, -2.1913e+00],
[-1.6150e+00, -1.7679e+00, -4.5594e-02, ..., -1.6973e+00,
-2.1233e+00, 7.3259e-01]],
[[-1.7864e+00, -1.4342e+00, -1.9167e+00, ..., -2.2947e+00,
-1.2001e+00, 3.1364e-01],
[-4.0791e+00, -1.5884e+00, -1.9955e+00, ..., -2.2564e+00,
-3.0183e+00, -1.8434e-02],
[-3.8779e+00, -2.7379e+00, -2.9151e+00, ..., -6.0751e-01,
-2.1164e+00, 3.9151e-01],
...,
[-2.8470e+00, -3.1716e+00, -2.5701e+00, ..., -4.2026e-01,
-1.6776e-01, -8.6751e-01],
[-1.8115e+00, 2.0585e-01, -1.4457e+00, ..., -4.8825e-01,
1.3660e-01, 1.2741e+00],
[-2.0000e+00, -1.0770e+00, -3.8324e-01, ..., -6.3687e-01,
-2.9709e-01, 8.7258e-01]],
[[ 1.2667e+00, 3.9459e-01, 8.9358e-01, ..., 1.5033e+00,
1.5407e+00, 1.2088e-01],
[ 3.9938e+00, 3.0761e+00, 2.8267e+00, ..., 3.5032e+00,
2.9474e+00, 3.8656e+00],
[ 1.5504e+00, 5.8505e-01, 9.8026e-01, ..., 3.3198e+00,
1.8805e+00, 2.1229e+00],
...,
[ 2.7791e+00, 2.2901e+00, 1.1627e+00, ..., 8.0962e+00,
4.0841e+00, 1.1605e+00],
[ 3.5334e+00, 4.3004e+00, 2.5592e+00, ..., 1.9257e+00,
7.5463e-01, 2.4907e+00],
[ 2.3588e+00, 2.5969e+00, 1.7764e+00, ..., 4.0270e-01,
1.1477e-01, 1.6042e+00]]],
[[[-2.7653e-01, 1.0555e+00, 6.5898e-01, ..., -4.0153e-01,
3.3850e-01, -2.0624e-01],
[-6.1774e-02, -7.7958e-02, 2.0932e+00, ..., -2.4652e-01,
1.6778e-01, -7.0479e-01],
[-1.3214e+00, 3.6627e-01, 2.7967e+00, ..., -7.1707e-01,
-7.6270e-01, -1.1981e+00],
...,
[-1.4322e-01, -1.0451e+00, -2.0462e-01, ..., -3.9277e-01,
6.9681e-01, -2.3781e-01],
[-1.7722e+00, -5.8120e-01, -2.2748e-01, ..., 2.9148e-01,
5.0659e-01, -1.1703e+00],
[-1.1721e+00, -2.4551e+00, -1.6102e+00, ..., -3.0908e+00,
-2.7978e+00, -2.7496e+00]],
[[-1.6256e+00, -2.3095e+00, -3.5985e+00, ..., -2.7711e+00,
-3.5396e+00, -2.3047e+00],
[-4.7884e+00, -4.6562e+00, -3.9668e+00, ..., -3.9014e+00,
-4.2315e+00, -3.6038e+00],
[-3.9652e+00, -5.3343e+00, -5.4182e+00, ..., -2.5054e+00,
-3.5013e+00, -3.0758e+00],
...,
[-5.0365e+00, -5.0657e+00, -3.4163e+00, ..., -2.7163e+00,
-2.9853e+00, -2.6287e+00],
[-7.8692e+00, -8.4399e+00, -6.1279e+00, ..., -5.1183e+00,
-5.2000e+00, -5.0140e+00],
[-1.4528e+00, -3.2179e+00, -1.8968e+00, ..., -2.9381e+00,
-2.7553e+00, -1.6096e+00]],
[[-2.2496e-01, 2.2833e+00, 1.5477e+00, ..., 1.6506e+00,
1.6914e+00, 8.7825e-01],
[-6.0012e-01, 3.3806e-01, 1.1972e+00, ..., 1.1984e+00,
1.1887e+00, 7.6513e-01],
[ 1.6326e-01, 2.2437e+00, 1.5662e+00, ..., 1.8950e+00,
1.7992e+00, 1.3947e+00],
...,
[-9.8696e-01, -1.5440e-01, 1.4074e+00, ..., 1.1422e+00,
1.1885e+00, 9.0186e-01],
[-9.4137e-01, -8.8384e-01, 2.4732e-01, ..., 1.6608e+00,
1.0461e+00, 5.2432e-01],
[-4.5832e-01, -6.2179e-01, -1.2242e-01, ..., 1.6564e+00,
1.5806e+00, 5.8635e-01]],
...,
[[-1.4011e+00, -2.1855e+00, -2.1580e+00, ..., -1.9569e+00,
-2.0672e+00, -5.0349e-01],
[-3.5865e+00, -3.7557e+00, -1.9551e+00, ..., -2.2988e+00,
-4.3087e+00, -3.0840e+00],
[-2.6365e+00, -2.5279e+00, -2.4472e+00, ..., -1.6491e+00,
-3.5306e+00, -2.0908e+00],
...,
[-8.5905e-01, -2.1259e+00, -2.9252e-01, ..., -1.7130e+00,
-2.3279e+00, -1.0824e+00],
[-2.3818e+00, -4.1020e+00, -2.0076e+00, ..., -1.8779e+00,
-2.8579e+00, -1.6660e+00],
[-1.2685e+00, -1.3928e+00, -6.4494e-01, ..., -1.8417e+00,
-2.1506e+00, -6.1473e-01]],
[[-1.8074e+00, -1.8888e+00, -3.0802e+00, ..., -1.0284e+00,
-1.4694e+00, -6.2691e-01],
[-4.0614e+00, -2.2542e+00, -3.0530e+00, ..., -1.3943e+00,
-3.7426e+00, -1.7240e+00],
[-3.8734e+00, -2.5367e+00, -3.4986e+00, ..., -6.2836e-01,
-3.9725e+00, -2.6518e+00],
...,
[-2.8395e+00, -2.6953e+00, -1.9006e+00, ..., -5.2646e-01,
-2.5335e+00, -2.0506e+00],
[-1.8771e+00, 2.3093e-01, -8.7704e-01, ..., -4.3189e-01,
-3.0759e+00, -2.0745e+00],
[-2.2144e+00, -1.4790e+00, -1.6051e+00, ..., -1.6138e+00,
-2.7106e+00, -1.3308e+00]],
[[ 1.2399e+00, 2.7258e-01, 9.1113e-01, ..., 2.4512e-01,
1.2085e+00, 6.5025e-01],
[ 3.9648e+00, 2.8258e+00, 2.2647e+00, ..., 7.0666e-01,
9.1760e-01, 2.4449e+00],
[ 1.5257e+00, 7.3580e-01, 1.8313e+00, ..., 5.9804e-01,
-1.3944e-02, 6.7804e-01],
...,
[ 3.2202e+00, 1.5151e+00, 2.1988e+00, ..., 1.3916e+00,
2.8445e-01, 4.4394e-01],
[ 3.7140e+00, 3.5518e+00, 2.5114e+00, ..., 1.0156e+00,
7.7334e-01, 1.1645e+00],
[ 2.2927e+00, 2.1733e+00, 1.5601e+00, ..., -4.9069e-02,
-2.2450e-01, -1.2597e-01]]],
[[[ 4.6726e-01, 1.0057e+00, 3.8465e-02, ..., -2.7978e+00,
-3.9136e+00, -2.8942e+00],
[ 5.5941e-01, 5.2031e-01, 1.3480e+00, ..., 5.7194e+00,
-1.4319e+00, 8.8864e-01],
[-2.2686e-01, -5.5660e-01, -5.5111e-02, ..., 7.1731e+00,
-6.8526e+00, 1.4598e+00],
...,
[-1.8942e+00, 4.7045e+00, 3.5764e+00, ..., 1.2407e+00,
-1.1425e+00, 4.6356e+00],
[-5.9762e-01, -1.7353e+00, -2.0019e+00, ..., -2.4246e+00,
-5.3331e+00, -1.9138e+00],
[-2.4023e+00, -4.1120e+00, -5.1440e+00, ..., -4.7931e+00,
-4.2742e+00, -2.1113e+00]],
[[-1.4009e+00, -2.3748e+00, -3.2989e+00, ..., -6.8754e+00,
-9.0636e+00, -1.0840e+00],
[-4.7248e+00, -4.8428e+00, -3.0688e+00, ..., -1.6472e+00,
-1.0087e+01, 1.3014e+00],
[-4.1507e+00, -5.9893e+00, -7.4821e+00, ..., -8.0630e-01,
-1.4263e+01, 1.2301e+00],
...,
[-1.4658e+00, -1.2075e+00, -9.2672e+00, ..., -4.8268e+00,
-7.6402e-01, -1.7121e+00],
[-8.2404e+00, -1.1317e+01, -1.0850e+01, ..., -1.4337e+01,
-1.4764e+01, -6.7357e+00],
[-7.5010e-01, -2.4576e+00, -3.9354e+00, ..., -3.3608e+00,
-3.4132e+00, -4.1028e-01]],
[[-8.9975e-03, 1.4874e+00, 1.6991e+00, ..., 2.7380e+00,
4.6256e+00, 2.6399e+00],
[ 1.2792e-01, 8.4985e-02, 4.2397e-01, ..., -2.8314e+00,
-5.9857e-01, 8.8742e-01],
[ 1.2206e+00, 1.5580e+00, 1.1156e+00, ..., -9.0854e+00,
-1.0125e+01, -7.1354e+00],
...,
[-5.0404e-02, -1.5707e+00, -1.7186e+00, ..., -4.3274e+00,
2.0528e+00, -6.4659e-01],
[-1.6436e+00, -3.3085e+00, -1.1492e+00, ..., -2.8469e+00,
-1.3946e+00, 1.3220e-01],
[-9.8279e-01, 1.1830e+00, 1.1672e+00, ..., -6.4828e-01,
-2.1399e-01, -1.3393e+00]],
...,
[[-7.9462e-01, -1.9030e+00, -2.2654e+00, ..., -3.6885e+00,
-4.4835e+00, -3.5558e+00],
[-3.0307e+00, -2.8112e+00, -1.2643e+00, ..., 2.1165e+00,
6.9144e-01, -1.9094e-01],
[-2.2606e+00, -1.9144e+00, -1.8876e+00, ..., -1.4092e+00,
2.2052e+00, 1.0853e+00],
...,
[ 8.5658e-01, -8.8679e-01, -8.0929e+00, ..., -5.0905e+00,
-4.7161e+00, -1.9172e+00],
[-1.8449e+00, -3.5855e+00, -2.3251e+00, ..., -7.6178e+00,
-8.6071e+00, -2.6963e+00],
[-3.3770e-01, -1.1192e+00, -1.3706e+00, ..., 5.1526e-01,
-1.4233e+00, -1.5822e-01]],
[[-1.6571e+00, -2.4559e+00, -3.7935e+00, ..., -3.3766e+00,
-4.2913e+00, 2.2244e-01],
[-4.4946e+00, -3.3633e+00, -4.0431e+00, ..., -2.8348e+00,
-2.1065e+00, -2.2277e+00],
[-3.8780e+00, -2.3153e+00, -2.1448e+00, ..., -3.8348e+00,
-3.5531e+00, 6.9502e-01],
...,
[-1.3641e+00, 1.3191e-01, -9.8036e+00, ..., -1.9724e+00,
-4.7304e+00, -3.2199e-01],
[-2.4496e+00, -1.6306e+00, -4.8109e+00, ..., -7.3197e-01,
-3.8759e+00, -2.2215e-01],
[-8.2508e-01, -1.4360e+00, -7.7663e-01, ..., 3.3546e+00,
5.3666e-02, 1.4018e+00]],
[[ 1.3605e+00, 7.9672e-02, 1.1538e+00, ..., 1.0453e+00,
1.9667e+00, 3.6488e-01],
[ 2.6366e+00, 1.2825e+00, 1.0053e+00, ..., 8.1754e+00,
4.0697e-01, 3.0759e+00],
[ 1.7260e+00, 9.2656e-01, 1.1992e+00, ..., 1.6016e+00,
-3.6429e+00, 2.5940e+00],
...,
[-2.7815e+00, 2.6247e+00, 8.5192e+00, ..., 1.6760e-01,
-4.7739e+00, 2.9202e+00],
[ 3.5526e+00, 5.7367e+00, 1.6157e+00, ..., 5.4498e+00,
3.4483e+00, 4.3864e+00],
[ 7.8424e-01, -7.7999e-01, 1.5650e-01, ..., -1.2136e+00,
-1.9579e+00, 9.1610e-01]]]], device='cuda:0'),
tensor([[[[ 2.7481e-01, 1.7512e+00, 1.6935e+00, ..., -9.8912e-01,
-3.2826e+00, -1.2702e+00],
[ 3.7197e-01, -1.5043e+00, 1.0637e+00, ..., 9.0083e-01,
2.0559e+00, 1.9786e+00],
[-1.4943e+00, -4.4421e+00, 1.2214e-01, ..., -1.9139e+00,
4.1805e-01, 1.0609e-01],
...,
[-1.9813e+00, -5.9887e-01, 2.9041e+00, ..., -3.0394e+00,
-5.2862e+00, 2.2292e-01],
[-2.9322e+00, -1.6203e+00, 6.6771e-01, ..., -3.5498e+00,
-7.0113e-01, 2.5652e+00],
[-1.6204e+00, -3.1402e+00, -1.7135e+00, ..., -4.9478e-01,
2.4783e+00, 5.7679e-01]],
[[-8.0041e-01, -1.6944e+00, -3.5779e+00, ..., -2.8095e+00,
-6.6251e+00, -1.5714e+00],
[-4.6898e+00, -4.8902e+00, -4.0364e+00, ..., 7.1787e-01,
2.2688e-01, -8.1521e-01],
[-4.6636e+00, -5.6547e+00, -4.4519e+00, ..., -3.6548e+00,
-3.1960e+00, -1.2014e+00],
...,
[-6.2993e+00, -7.6176e+00, -4.8091e+00, ..., -2.7864e+00,
-2.0650e+00, 1.6990e+00],
[-7.7258e+00, -6.7351e+00, -7.5974e+00, ..., -2.6595e+00,
-1.2743e+00, 6.6563e-01],
[-2.0793e+00, -3.4696e+00, -2.2328e+00, ..., -6.4590e-01,
-6.3352e-01, -1.3854e+00]],
[[-1.1737e-01, 2.2312e+00, 1.3954e+00, ..., 2.1964e+00,
4.3236e+00, 1.3718e+00],
[-6.4230e-01, 3.5602e-01, 1.0055e+00, ..., -3.9746e+00,
1.5474e+00, 7.0239e-01],
[-6.2089e-02, 1.4573e+00, 1.8641e+00, ..., -3.3985e+00,
-1.8066e+00, 1.6866e+00],
...,
[-1.4209e+00, -4.3291e+00, -4.9296e+00, ..., -1.2075e+00,
-4.5533e-01, 8.6751e-01],
[-2.3200e+00, 3.0166e-01, -1.0705e+00, ..., 1.8121e-01,
3.1069e-01, -2.0036e-01],
[-4.6974e-01, -5.3933e-01, 1.2237e-01, ..., -2.6226e-01,
-1.3280e+00, -1.7143e+00]],
...,
[[-9.8703e-01, -1.7865e+00, -2.4601e-01, ..., -2.2908e+00,
-4.4515e+00, -1.2772e+00],
[-3.4053e+00, -4.1762e+00, -1.5904e+00, ..., 1.1944e+00,
-3.7479e-01, -1.0376e+00],
[-2.6554e+00, -1.3536e+00, 3.9411e-01, ..., 1.3530e-01,
-2.0651e+00, -3.6008e+00],
...,
[-1.6049e+00, -4.9824e+00, -2.8225e+00, ..., -1.1064e+00,
-8.9798e-01, 3.1769e-02],
[-1.6982e+00, -4.9816e+00, -3.0418e+00, ..., -2.3144e+00,
-1.7677e+00, -1.5931e+00],
[-1.8032e+00, -1.6220e+00, -1.0122e-01, ..., 4.4467e-01,
6.4377e-02, -1.3307e+00]],
[[-1.5699e+00, -1.8600e+00, -1.4396e+00, ..., -3.1995e+00,
-2.1834e+00, 1.4972e+00],
[-4.2017e+00, -3.2751e+00, -2.7138e+00, ..., 1.1864e+00,
-1.8954e+00, 4.2846e-01],
[-2.6873e+00, -1.2764e+00, -3.2336e+00, ..., 1.0555e+00,
-3.8548e+00, -1.7231e+00],
...,
[-1.7721e+00, 4.0184e-03, -1.7920e+00, ..., -5.3072e+00,
-4.1265e+00, -1.1888e+00],
[-8.3099e-01, 1.3948e+00, 1.2115e-01, ..., -4.0169e+00,
-1.8088e+00, -1.2151e-01],
[-2.4277e+00, -1.3340e+00, 1.3581e-01, ..., -1.3395e+00,
3.8908e-01, -6.2353e-01]],
[[ 9.1107e-01, 2.0092e-01, 9.4071e-01, ..., 7.4616e-01,
2.0017e+00, 9.2732e-01],
[ 4.1683e+00, 2.1155e+00, 2.0022e+00, ..., -1.9033e+00,
9.1134e-01, 5.8616e+00],
[ 1.7001e+00, -1.2434e+00, -4.4246e-01, ..., -1.5958e+00,
1.2423e+00, 3.6147e+00],
...,
[ 3.5393e+00, 5.4941e+00, 4.1700e+00, ..., -3.4679e+00,
-4.3556e+00, -1.8334e+00],
[ 2.7516e+00, 2.5368e+00, 2.7096e+00, ..., -5.4829e+00,
-2.5107e+00, 1.3328e+00],
[ 2.5776e+00, 2.1263e+00, 1.1758e+00, ..., -1.5255e+00,
1.0439e+00, 1.7452e+00]]],
[[[-1.0529e-01, 5.9494e-01, -3.0034e-01, ..., -1.2300e+00,
-7.8585e-01, -7.9767e-01],
[ 7.4945e-01, 1.0365e+00, 1.5021e+00, ..., -5.5021e-01,
4.5016e-01, -1.0265e+00],
[ 1.3337e+00, 6.6411e-01, 1.0626e+00, ..., -5.5404e-01,
-4.5249e-01, -2.0763e+00],
...,
[-4.8766e-01, -5.5322e-01, -1.1239e+00, ..., -1.4541e+00,
-2.5160e+00, -1.1094e-01],
[-2.3415e+00, -1.3625e+00, 3.7917e-01, ..., -2.2756e+00,
-2.7675e+00, 1.4753e-01],
[-2.4522e+00, -3.9742e+00, -2.9971e+00, ..., -4.1171e+00,
-3.5388e+00, -1.5181e+00]],
[[-5.8802e-01, -1.2102e+00, -2.4571e+00, ..., -2.6581e+00,
-4.3678e+00, -2.4090e+00],
[-1.7546e+00, -3.5786e+00, -4.9235e+00, ..., -3.1701e+00,
-4.7519e+00, -4.5982e+00],
[-1.6749e+00, -3.4535e+00, -3.9917e+00, ..., -4.0223e+00,
-3.9606e+00, -3.7059e+00],
...,
[-6.7570e+00, -8.5337e+00, -7.9845e+00, ..., -7.6783e+00,
-8.8316e+00, -1.9348e+00],
[-9.8221e+00, -1.0310e+01, -8.6557e+00, ..., -1.0638e+01,
-1.0780e+01, -3.7662e+00],
[-2.5504e+00, -4.9644e+00, -4.6444e+00, ..., -4.1997e+00,
-4.2394e+00, -6.3689e-01]],
[[ 6.6266e-01, 1.1527e+00, 1.3399e+00, ..., 1.6200e+00,
1.8108e+00, 8.6983e-01],
[ 7.6863e-01, 1.4929e+00, 1.7033e+00, ..., 1.5131e+00,
2.1897e+00, 8.7997e-01],
[ 3.9650e-01, 3.0498e+00, 8.8760e-01, ..., 1.0901e+00,
2.2917e+00, 2.0470e+00],
...,
[-2.0523e-01, -2.1833e-01, 4.8055e-01, ..., 1.5330e+00,
1.6359e+00, -1.6342e-01],
[-7.4284e-01, 4.5061e-01, -1.0801e+00, ..., 5.9949e-01,
1.1810e+00, -3.4357e-01],
[ 6.7340e-01, 1.2447e+00, 8.1396e-01, ..., -1.7713e-01,
-5.3927e-01, -1.9068e+00]],
...,
[[-9.1396e-01, -1.0795e+00, -1.7906e+00, ..., -2.6784e+00,
-3.0732e+00, -1.1843e+00],
[-2.1198e+00, -2.7179e+00, -1.9019e+00, ..., -3.5388e+00,
-4.2099e+00, -3.4771e+00],
[-6.1164e-01, -1.1877e+00, -1.5629e+00, ..., -2.7425e+00,
-4.5018e+00, -3.1455e+00],
...,
[-2.6353e+00, -4.3199e+00, -4.2410e+00, ..., -5.7343e+00,
-6.4220e+00, -2.5803e+00],
[-3.6829e+00, -5.7665e+00, -3.3229e+00, ..., -4.4866e+00,
-5.3022e+00, -1.2706e+00],
[-3.3565e+00, -4.0834e+00, -3.9232e+00, ..., -2.7766e+00,
-2.6103e+00, 4.2593e-01]],
[[-1.4651e+00, -2.1883e+00, -3.4198e+00, ..., -2.7977e+00,
-2.8879e+00, -5.9727e-01],
[-4.2551e+00, -4.1399e+00, -4.8783e+00, ..., -4.0495e+00,
-6.0999e+00, -1.9631e+00],
[-3.8314e+00, -5.0949e+00, -5.0135e+00, ..., -2.3989e+00,
-5.1275e+00, -2.2461e+00],
...,
[-3.5637e+00, -3.4434e+00, -5.4908e+00, ..., -3.6815e+00,
-2.8272e+00, -1.2117e+00],
[-3.2784e+00, -6.0569e-01, -2.5654e+00, ..., -2.1631e+00,
-1.2398e+00, 3.6983e-01],
[-3.7330e+00, -2.9155e+00, -3.1249e+00, ..., -1.6436e+00,
-1.2144e+00, 3.5235e-01]],
[[ 8.4094e-01, -2.1553e-01, 5.2007e-01, ..., 3.9835e-01,
1.4322e+00, 6.0243e-01],
[ 1.6252e+00, 8.2423e-01, 1.1130e+00, ..., 3.1860e-01,
1.7985e+00, 3.2519e+00],
[ 1.3982e+00, 3.7196e-01, 1.2886e+00, ..., 1.1416e-01,
5.1883e-01, 8.7157e-01],
...,
[ 3.8194e+00, 3.8398e+00, 3.0963e+00, ..., 2.9439e+00,
2.6488e+00, 6.8184e-01],
[ 4.5861e+00, 4.9156e+00, 3.8204e+00, ..., 2.4783e+00,
1.8573e+00, 1.4745e+00],
[ 2.7776e+00, 2.9311e+00, 2.0107e+00, ..., 5.1814e-01,
5.5088e-01, 1.0230e+00]]],
[[[-1.0846e+01, -6.7166e+00, -1.6803e+00, ..., -2.3131e+00,
-4.3170e-01, -1.9069e-01],
[-9.2997e+00, -8.1981e+00, -5.6925e+00, ..., -1.9396e+00,
-2.0060e-01, -1.2286e+00],
[-8.4377e+00, -1.1225e+01, -8.6108e+00, ..., -1.8839e+00,
-2.6004e+00, -3.0247e+00],
...,
[-3.3587e-01, -9.2850e-01, -1.5156e+00, ..., -1.6681e+00,
-1.2184e+00, -1.6397e+00],
[-1.8645e+00, -7.9297e-03, -8.3568e-01, ..., -8.3717e-01,
-7.8686e-01, -2.2350e+00],
[-1.0371e+00, -1.6726e+00, -1.8015e+00, ..., -3.6508e+00,
-3.3935e+00, -3.2934e+00]],
[[-5.4528e+00, -4.0666e+00, -2.6164e+00, ..., -7.1293e+00,
-5.0752e+00, -2.3953e+00],
[ 1.0861e+00, 5.2340e-01, -2.1459e+00, ..., -8.0989e+00,
-8.3998e+00, -4.3796e+00],
[-2.0438e+00, -4.0033e+00, -6.4014e+00, ..., -6.7903e+00,
-1.0779e+01, -4.9869e+00],
...,
[-4.1379e+00, -3.6980e+00, -3.1099e+00, ..., -5.1603e+00,
-4.8202e+00, -4.7999e+00],
[-6.9823e+00, -7.6402e+00, -6.6115e+00, ..., -6.9626e+00,
-8.0380e+00, -6.3729e+00],
[-1.0959e+00, -3.1832e+00, -1.8308e+00, ..., -3.4608e+00,
-3.5897e+00, -1.9309e+00]],
[[ 2.6104e+00, 4.5386e+00, 3.5194e+00, ..., 1.1410e+00,
2.8399e+00, 6.9533e-01],
[-9.2467e+00, -8.9379e+00, -3.3604e+00, ..., 2.2548e+00,
1.4720e+00, 1.2951e-01],
[ 8.4778e-01, 3.7307e+00, -1.0081e+00, ..., 6.4344e+00,
2.0711e+00, 1.0549e+00],
...,
[-3.0682e+00, -1.0318e+00, 9.7328e-01, ..., -6.8908e-01,
1.2253e+00, 1.4618e+00],
[-2.0794e+00, -1.1647e+00, 1.0773e-01, ..., 1.0539e+00,
1.8475e+00, 9.4966e-01],
[-1.2120e+00, -6.8096e-01, 7.8535e-01, ..., 1.5804e+00,
1.7160e+00, 2.9097e-01]],
...,
[[-2.7533e+00, -1.2558e+00, 1.1995e+00, ..., -5.9573e+00,
-3.9388e+00, -2.9599e-01],
[ 7.6711e-01, 7.8637e-01, -9.5704e-02, ..., -4.8942e+00,
-5.8280e+00, -4.0787e+00],
[-1.8619e+00, -6.2844e+00, -4.4938e+00, ..., -8.2732e+00,
-6.5584e+00, -5.4369e+00],
...,
[ 6.2518e-01, -1.4500e+00, 1.8708e+00, ..., -4.1391e+00,
-5.7816e+00, -2.7368e+00],
[-1.8657e+00, -4.4238e+00, -2.7812e+00, ..., -3.2197e+00,
-5.5121e+00, -2.4236e+00],
[-1.0861e+00, -1.8096e+00, -5.4781e-01, ..., -2.4614e+00,
-3.0806e+00, -8.4454e-01]],
[[-6.8137e+00, -9.1201e+00, -8.8334e+00, ..., -8.2069e-01,
-1.8772e+00, -4.5690e-02],
[-2.9285e+00, -3.6253e+00, -3.8933e+00, ..., 1.1990e+00,
-3.6085e+00, -1.4198e+00],
[-7.7191e+00, -1.3838e+01, -1.9319e+01, ..., -1.0520e+00,
-1.5634e+00, -1.6004e+00],
...,
[-1.4777e+00, -1.8037e+00, -1.1277e+00, ..., -1.9622e+00,
-3.3805e+00, -1.3999e+00],
[-2.4572e-01, 6.8376e-01, -4.2128e-01, ..., -6.0319e-01,
-3.6569e+00, -1.3987e+00],
[-1.3872e+00, -1.3098e+00, -1.7903e+00, ..., -1.6196e+00,
-3.0115e+00, -1.1218e+00]],
[[ 9.3357e-01, 1.4551e+00, 1.1657e+00, ..., 1.0140e-01,
1.2223e+00, 8.1662e-01],
[-4.9009e+00, -5.8692e+00, -9.7260e+00, ..., 2.8539e+00,
1.7181e+00, 2.7756e+00],
[-6.7640e+00, -5.8037e+00, -1.0159e+01, ..., 5.8872e+00,
1.9062e+00, 3.2457e-01],
...,
[ 2.4497e+00, 9.5946e-01, 2.0143e+00, ..., 3.3593e+00,
2.3483e+00, 7.8604e-01],
[ 3.0636e+00, 2.8476e+00, 2.9121e+00, ..., 2.2299e+00,
1.1648e+00, 1.4562e+00],
[ 2.3288e+00, 2.6259e+00, 1.7681e+00, ..., -7.2597e-02,
-5.1705e-01, -1.0050e-01]]],
[[[ 7.6219e-01, 5.9207e-01, -5.1543e-01, ..., -3.2294e-01,
1.4978e-01, -1.6820e-01],
[ 1.1687e+00, 3.5630e-01, 1.2078e+00, ..., 9.1612e-02,
-5.4040e-01, -7.0106e-01],
[ 2.6633e-01, 9.7416e-02, 8.3344e-01, ..., 4.0784e-01,
-2.0287e+00, -1.7874e+00],
...,
[ 6.6435e-01, -8.7844e-02, 1.7282e+00, ..., -8.5750e-01,
-7.8581e-01, -6.8999e-01],
[-2.2966e+00, -2.4170e+00, -9.7168e-02, ..., -5.5015e-01,
-3.6831e-01, -1.3562e+00],
[-9.8686e-01, -3.5735e+00, -3.1040e+00, ..., -2.9014e+00,
-2.9143e+00, -2.6803e+00]],
[[-1.4583e+00, -4.0833e+00, -6.9592e+00, ..., -3.6409e+00,
-3.7535e+00, -2.1505e+00],
[-4.8868e+00, -6.3868e+00, -6.5326e+00, ..., -6.7534e+00,
-5.8430e+00, -3.4730e+00],
[-4.0443e+00, -5.6290e+00, -6.5815e+00, ..., -2.1971e+00,
-5.4930e+00, -3.5932e+00],
...,
[-5.4650e+00, -6.9892e+00, -5.9624e+00, ..., -4.9555e+00,
-4.0118e+00, -2.7607e+00],
[-6.8311e+00, -9.5024e+00, -1.0129e+01, ..., -4.7265e+00,
-5.0913e+00, -5.0035e+00],
[-1.8727e+00, -4.3972e+00, -4.3618e+00, ..., -3.0281e+00,
-2.7951e+00, -1.6064e+00]],
[[ 7.8422e-01, 2.9969e+00, 1.8348e+00, ..., 1.7197e+00,
1.4311e+00, 8.9524e-01],
[-1.0564e-01, 1.6042e+00, 1.5450e+00, ..., 1.5574e+00,
1.0565e+00, 6.6510e-01],
[ 7.7006e-01, 2.1985e+00, 9.8891e-01, ..., 2.4891e+00,
1.4822e+00, 1.3297e+00],
...,
[-1.2995e+00, -1.7578e+00, -3.0383e-01, ..., 2.4065e-01,
3.9190e-01, 5.2577e-01],
[ 6.1875e-01, 6.6827e-01, -1.0350e+00, ..., 7.0578e-01,
6.7688e-01, 5.4961e-01],
[ 3.6383e-01, 1.0026e+00, 4.1222e-01, ..., 1.1217e+00,
1.5893e+00, 6.4426e-01]],
...,
[[-8.9593e-01, -3.3193e+00, -2.6760e+00, ..., -1.5588e+00,
-2.5109e+00, -4.3871e-01],
[-2.7657e+00, -4.1672e+00, -1.6678e+00, ..., -1.6356e+00,
-5.2795e+00, -3.2852e+00],
[-2.1432e+00, -3.3975e+00, -2.0465e+00, ..., -1.8451e+00,
-3.0178e+00, -3.0223e+00],
...,
[-2.3378e+00, -3.4202e+00, -1.0601e+00, ..., -2.3729e+00,
-2.8323e+00, -1.3499e+00],
[-2.5399e+00, -4.9639e+00, -3.1202e+00, ..., -7.9008e-01,
-3.1662e+00, -1.9042e+00],
[-1.9299e+00, -2.7202e+00, -1.1657e+00, ..., -1.7693e+00,
-2.5641e+00, -6.8135e-01]],
[[-2.4821e+00, -3.1691e+00, -1.5518e+00, ..., 6.7641e-01,
-1.2613e+00, -5.6076e-01],
[-4.3100e+00, -3.6155e+00, -1.9505e+00, ..., -1.2013e-01,
-3.8364e+00, -2.1123e+00],
[-3.4320e+00, -2.5226e+00, -2.1113e+00, ..., 1.2704e+00,
-3.0290e+00, -2.3942e+00],
...,
[-2.5560e+00, -3.0961e+00, -1.4784e+00, ..., -1.4488e+00,
-3.4976e+00, -1.5199e+00],
[-3.0894e+00, -2.5274e-01, 1.9243e-01, ..., -3.9075e-01,
-3.1279e+00, -1.6732e+00],
[-3.7651e+00, -2.2772e+00, -3.6461e-01, ..., -5.0528e-01,
-2.3211e+00, -1.1568e+00]],
[[ 1.6119e+00, 1.3491e+00, 1.7700e+00, ..., 8.4839e-01,
9.7416e-01, 6.2239e-01],
[ 3.1489e+00, 3.3010e+00, 3.8289e+00, ..., 2.5674e+00,
9.4480e-01, 2.3541e+00],
[ 1.7507e+00, 1.8793e+00, 2.8401e+00, ..., 1.9898e+00,
5.2863e-01, 5.6213e-01],
...,
[ 5.7522e+00, 4.0849e+00, 2.0801e+00, ..., 1.2745e+00,
5.4313e-01, 7.6145e-01],
[ 3.0248e+00, 3.1705e+00, 2.0640e+00, ..., 1.7695e+00,
1.2768e+00, 1.2785e+00],
[ 2.5786e+00, 1.9774e+00, 1.6656e+00, ..., 6.2119e-01,
-2.7834e-01, -1.4919e-01]]],
[[[ 2.4025e-01, 1.3738e+00, 6.3433e-01, ..., -5.0817e-01,
-5.8149e-02, 7.3890e-01],
[ 4.2599e-01, 3.8919e-01, 6.1009e-01, ..., -1.2223e+00,
-1.2081e+00, 1.0326e+00],
[ 1.6767e-02, 2.2018e-01, 7.5844e-01, ..., -7.4951e-01,
1.5005e-01, 5.9038e-01],
...,
[ 4.0560e-01, -7.8560e-01, -1.6405e+00, ..., -2.0259e+00,
-1.6029e+00, -1.1692e+00],
[-1.3257e+00, -1.3999e+00, -3.6717e+00, ..., -2.5559e+00,
-3.5392e+00, -1.3978e+00],
[ 5.3120e-03, -2.3938e+00, -3.6469e+00, ..., -1.6828e+00,
-2.4718e+00, -8.9451e-01]],
[[-7.8491e-01, -2.3086e+00, -4.2109e+00, ..., -3.4091e+00,
-4.9947e+00, -3.3139e-01],
[-4.6976e+00, -5.9585e+00, -5.3745e+00, ..., -5.8277e+00,
-7.4975e+00, -1.8985e+00],
[-3.5023e+00, -5.2302e+00, -6.1956e+00, ..., -1.8417e+00,
-5.3169e+00, -3.9707e+00],
...,
[-4.6102e+00, -5.3721e+00, -5.5140e+00, ..., -6.9999e+00,
-5.3073e+00, -2.3108e+00],
[-7.6771e+00, -8.3363e+00, -8.4121e+00, ..., -9.0902e+00,
-9.1897e+00, -4.0798e+00],
[-1.0514e+00, -2.5585e+00, -3.1021e+00, ..., -3.0364e+00,
-2.8139e+00, -3.0106e-01]],
[[-3.0909e-01, 2.2168e+00, 1.3181e+00, ..., 9.0452e-01,
2.1464e+00, 9.6026e-01],
[-8.9272e-02, 1.0432e+00, 8.9217e-01, ..., 9.8401e-01,
1.1810e+00, -5.9000e-01],
[ 6.6713e-02, 1.8657e+00, 1.2959e+00, ..., 1.1226e+00,
1.6861e+00, 1.0251e-02],
...,
[ 2.7886e-01, 7.8411e-01, 1.4856e+00, ..., 1.7336e+00,
1.0296e+00, 7.8153e-01],
[ 2.2536e-01, 6.1344e-01, 1.3131e+00, ..., 2.0979e+00,
2.0681e+00, 7.7848e-01],
[-6.2927e-01, 1.0009e-01, 6.2423e-01, ..., 1.3195e+00,
1.2069e+00, -6.9505e-01]],
...,
[[-8.4647e-01, -1.8820e+00, -1.9094e+00, ..., -2.6172e+00,
-2.8324e+00, -4.8279e-01],
[-3.0177e+00, -4.2055e+00, -2.7531e+00, ..., -3.9840e+00,
-4.0485e+00, -2.8222e+00],
[-1.4645e+00, -2.6334e+00, -1.2665e+00, ..., -7.7229e-01,
-2.6867e+00, -9.1820e-01],
...,
[-9.4479e-01, -1.7663e+00, -1.5594e+00, ..., -2.7540e+00,
-4.1402e+00, -2.0080e+00],
[-2.6136e+00, -3.7250e+00, -3.4399e+00, ..., -2.9799e+00,
-5.3161e+00, -1.6011e+00],
[-5.9817e-01, -1.5648e+00, -8.7074e-01, ..., -3.6285e-01,
-1.3742e+00, 1.2026e+00]],
[[-1.6951e+00, -2.3060e+00, -2.9193e+00, ..., -1.5155e+00,
-1.4163e+00, -1.3207e+00],
[-4.2561e+00, -3.3600e+00, -3.1278e+00, ..., -3.3166e+00,
-2.0524e+00, -8.7088e-01],
[-2.6222e+00, -1.6478e+00, -2.1421e+00, ..., -1.2557e-01,
-7.4296e-01, -1.0497e-01],
...,
[-3.7501e+00, -4.0981e+00, -3.5203e+00, ..., -2.0153e+00,
-2.3738e+00, -1.0718e+00],
[-4.3801e+00, -2.7774e+00, -1.0579e+00, ..., -1.5171e-01,
-2.0034e+00, -4.7788e-01],
[-2.7172e+00, -2.0016e+00, -3.7015e-02, ..., 4.5738e-01,
-2.6383e-01, 9.3715e-02]],
[[ 1.0886e+00, 5.0371e-01, 7.4920e-01, ..., -5.3599e-01,
5.6420e-02, -3.8968e-01],
[ 3.5421e+00, 3.4280e+00, 2.2892e+00, ..., 1.8706e+00,
2.0379e+00, 2.4804e+00],
[ 1.6336e+00, 1.4330e+00, 1.3865e+00, ..., 4.2555e-01,
2.4277e+00, 2.6008e+00],
...,
[ 2.6610e+00, 1.7340e+00, 1.5154e+00, ..., 9.3078e-01,
1.1662e+00, 5.7794e-01],
[ 3.2999e+00, 4.2476e+00, 2.3074e+00, ..., 2.6655e+00,
1.3468e+00, 1.5918e+00],
[ 2.2977e+00, 1.5395e+00, 2.4905e-01, ..., 2.9054e-01,
2.1856e-01, 9.2509e-01]]]], device='cuda:0'),
tensor([[[[ 1.3212e+00, -3.0731e-01, -4.8935e+00, ..., -1.3311e+00,
3.4445e+00, 1.7956e+00],
[ 2.0190e+00, -2.5894e+00, -4.9106e+00, ..., 6.0449e-01,
4.8551e+00, 1.4725e+00],
[ 3.4308e-01, -7.3005e+00, -2.7289e+00, ..., 3.7143e+00,
3.8253e+00, -8.6868e-01],
...,
[ 8.0049e-01, -2.7545e+00, -9.7266e+00, ..., 8.3707e-02,
-4.2253e+00, -5.2753e+00],
[-1.2295e+00, -1.1793e+00, -1.0055e+01, ..., -1.6072e-01,
-1.1428e+00, -3.0130e+00],
[-1.4589e+00, 1.4649e+00, -5.8304e+00, ..., 5.6684e-01,
-2.3035e-01, -1.7646e+00]],
[[-9.6076e-01, -5.5655e+00, -6.4429e+00, ..., -5.3194e-01,
1.5711e+00, -1.9658e-01],
[-3.8365e+00, -1.0382e+01, -5.1423e+00, ..., 8.6409e-01,
-2.3197e+00, -5.2819e+00],
[-4.2586e+00, -1.1260e+01, -2.6436e+00, ..., 1.1847e+00,
-3.9215e+00, -7.1580e+00],
...,
[ 2.5777e+00, -2.0151e+00, -2.8138e+00, ..., -2.4999e+00,
-1.9993e+01, -4.4488e+00],
[-3.3277e-02, 4.3330e+00, -3.4268e+00, ..., -6.0234e+00,
-1.3809e+01, -4.5111e+00],
[-3.8861e-01, 1.7829e+00, 1.2889e+00, ..., -5.7251e-01,
-4.5861e+00, -1.4179e+00]],
[[ 3.4042e-01, 1.7127e+00, 5.7155e-01, ..., -1.7966e+00,
2.6463e+00, 2.1579e+00],
[-1.6672e+00, -2.2125e-01, -1.4847e+00, ..., -5.5155e+00,
-5.8978e-01, 1.0905e+00],
[-9.6333e-01, 2.7592e+00, -2.0233e+00, ..., -5.2767e+00,
3.8798e+00, 1.3793e+00],
...,
[-5.2485e+00, -5.9907e+00, -8.0014e+00, ..., 1.3553e+01,
5.2145e+00, -1.5074e+00],
[-6.2161e+00, 1.0483e+00, 1.0466e-01, ..., 7.5248e+00,
4.4029e+00, -2.6026e-01],
[-2.6599e+00, -2.0039e+00, 6.0804e-01, ..., 5.7894e-01,
5.6361e-01, -1.1653e+00]],
...,
[[ 5.4193e-01, -4.1874e-01, -8.4281e-04, ..., 2.4842e+00,
-6.7578e-02, -2.4104e+00],
[-8.7852e-01, -3.5378e+00, 9.7459e-01, ..., 4.3010e+00,
-4.2747e+00, -3.2462e+00],
[-2.1126e+00, -5.2724e+00, 2.4554e+00, ..., 4.8350e+00,
-7.2145e+00, -6.2348e-01],
...,
[ 8.8404e-01, -6.3360e+00, 2.0623e+00, ..., -1.3413e+01,
-6.2015e+00, -6.9941e+00],
[ 2.2699e+00, 4.1169e-01, -2.4610e+00, ..., -1.6554e+01,
-5.6848e+00, -4.4843e+00],
[ 3.3319e-02, -1.8715e+00, 2.3969e+00, ..., -4.5923e+00,
-1.9675e+00, -1.5001e+00]],
[[-2.0751e+00, -2.0741e+00, -1.2952e+00, ..., -2.2181e-01,
-1.9738e+00, 5.7501e-01],
[-4.0824e+00, -1.0809e+00, -9.9734e-01, ..., 2.4017e+00,
-2.4989e+00, -1.7011e+00],
[-5.2619e+00, -2.5381e+00, -5.6269e-01, ..., 4.2480e+00,
-5.0002e+00, -7.2183e-01],
...,
[-7.0702e+00, -1.7166e+01, -1.0959e+01, ..., -9.3095e+00,
4.2206e+00, -1.7196e-01],
[-1.6162e+00, -1.4421e+01, -9.5659e+00, ..., -5.0706e+00,
6.6475e+00, -6.4690e-02],
[ 2.1476e+00, -4.7986e+00, -5.7294e+00, ..., -7.9174e-01,
3.3878e+00, -7.4520e-02]],
[[ 1.0017e+00, 1.9070e+00, -6.0623e-01, ..., -2.4305e+00,
2.1007e+00, 2.1556e+00],
[ 2.7391e+00, 2.8771e+00, 1.2220e+00, ..., -9.1462e-01,
5.4517e+00, 4.0531e+00],
[ 1.5712e+00, 7.3280e-01, 2.2239e+00, ..., 2.6635e+00,
8.1254e+00, 3.3006e+00],
...,
[ 1.2170e-01, -2.7337e+00, -1.1944e+01, ..., 4.6121e+00,
2.2530e+00, 9.7209e-02],
[-1.7406e-01, -2.7632e+00, -8.4049e+00, ..., 7.7097e+00,
2.1775e+00, 1.6846e+00],
[ 2.9011e+00, -3.4174e-01, -3.5381e+00, ..., 5.0158e+00,
2.8506e+00, 1.4538e+00]]],
[[[ 5.5010e-01, 1.4116e+00, 9.0095e-01, ..., 1.5967e+00,
9.4179e-02, 5.4907e-01],
[ 1.7738e-01, -5.9057e-01, 1.1151e+00, ..., 2.3159e+00,
-3.3448e-01, -4.7782e-01],
[-2.7487e-02, -4.5530e-01, 7.9828e-01, ..., -3.0601e-01,
-1.9247e-01, -9.9586e-01],
...,
[-9.8552e-01, -1.2842e+00, -9.5729e-02, ..., -2.9507e+00,
-3.6039e+00, -7.4600e-01],
[-2.1166e+00, -1.0413e+00, -1.4074e-03, ..., -3.1844e+00,
-3.1860e+00, -2.4760e-02],
[-1.6279e+00, -3.1487e+00, -2.9922e+00, ..., -4.6140e+00,
-3.7687e+00, -1.6464e+00]],
[[-1.4018e+00, -1.9050e+00, -2.3396e+00, ..., 5.4410e-01,
-4.6125e+00, -1.1673e+00],
[-4.9589e+00, -6.5970e+00, -4.8483e+00, ..., -2.7893e+00,
-7.2784e+00, -4.0770e+00],
[-4.2941e+00, -4.8700e+00, -3.5099e+00, ..., -6.8972e+00,
-6.3110e+00, -4.1374e+00],
...,
[-6.2075e+00, -8.2011e+00, -7.3706e+00, ..., -8.1262e+00,
-9.4225e+00, -2.2768e+00],
[-8.9628e+00, -9.4569e+00, -8.5271e+00, ..., -1.1374e+01,
-1.1406e+01, -4.0558e+00],
[-1.9832e+00, -4.2463e+00, -4.1150e+00, ..., -4.8080e+00,
-4.3896e+00, -6.6371e-01]],
[[-3.5123e-02, 9.3323e-01, 6.7231e-01, ..., 2.7188e+00,
2.5090e+00, 3.7312e-01],
[ 2.3437e-01, 2.6397e-01, -1.2779e-01, ..., 6.9142e-01,
2.6469e+00, -5.6805e-01],
[ 8.7182e-01, 2.7619e+00, 1.4402e+00, ..., 1.1768e+00,
2.3530e+00, 8.9186e-01],
...,
[ 2.1285e-01, 1.0456e+00, 1.0602e-01, ..., 1.4605e+00,
5.7033e-01, -1.1753e+00],
[-8.5679e-01, 7.7780e-01, 1.3900e-01, ..., 5.2940e-01,
6.2869e-01, -7.1400e-01],
[ 5.4308e-02, 7.2824e-01, 7.6463e-01, ..., 1.4084e-01,
-4.2029e-01, -1.9371e+00]],
...,
[[-9.5054e-01, -1.0577e+00, -2.9936e-01, ..., -2.5652e+00,
-2.8375e+00, 3.4851e-02],
[-4.3592e+00, -5.1080e+00, -1.9918e+00, ..., -4.9039e+00,
-5.0555e+00, -3.2726e+00],
[-2.5881e+00, -1.7282e+00, -7.1462e-02, ..., -3.8642e+00,
-3.7850e+00, -2.2037e+00],
...,
[-2.1244e+00, -3.4526e+00, -1.7630e+00, ..., -4.3339e+00,
-5.9636e+00, -2.8933e+00],
[-2.8472e+00, -5.3691e+00, -2.5155e+00, ..., -4.5193e+00,
-5.4197e+00, -1.4622e+00],
[-2.4042e+00, -3.3355e+00, -2.8686e+00, ..., -2.9033e+00,
-2.6048e+00, 4.3520e-01]],
[[-1.1377e+00, 1.0226e-01, -1.1359e+00, ..., -1.3995e+00,
-1.7678e-01, 1.9010e-01],
[-4.0837e+00, -1.4982e+00, -1.5485e+00, ..., -2.7353e+00,
-2.2391e+00, 2.6411e-01],
[-3.6819e+00, -1.9744e+00, -1.4814e+00, ..., -2.6341e+00,
-2.4741e+00, -7.3068e-01],
...,
[-4.0998e+00, -2.1296e+00, -2.3609e+00, ..., -3.4498e+00,
-2.8097e+00, -1.0823e+00],
[-3.2047e+00, -1.0446e+00, -1.4914e+00, ..., -2.2404e+00,
-1.2745e+00, 1.9870e-01],
[-3.1616e+00, -2.9249e+00, -2.9935e+00, ..., -1.8977e+00,
-1.2282e+00, 2.9442e-01]],
[[ 1.6884e+00, 7.2006e-01, 5.4376e-01, ..., 2.2302e+00,
1.9729e+00, 6.7735e-02],
[ 3.1560e+00, 2.0582e+00, 1.8782e+00, ..., 2.2131e+00,
2.6732e+00, 3.5212e+00],
[ 2.1750e+00, 1.1642e-01, 4.6540e-01, ..., 2.5910e+00,
2.6941e+00, 2.2640e+00],
...,
[ 2.6313e+00, 2.9569e+00, 2.3407e+00, ..., 1.7888e+00,
1.3471e+00, 7.0427e-02],
[ 4.2187e+00, 4.6996e+00, 4.0345e+00, ..., 2.9588e+00,
1.7735e+00, 1.4638e+00],
[ 2.7231e+00, 2.6222e+00, 1.8489e+00, ..., 3.6846e-01,
6.1984e-01, 1.0087e+00]]],
[[[ 2.8360e-01, 1.3959e+00, 5.6684e-01, ..., -5.7102e-01,
-2.1705e-02, 6.3386e-01],
[ 4.6749e-01, 1.8954e-01, 4.4623e-01, ..., -9.5174e-01,
-9.8258e-01, 1.0424e+00],
[ 4.1621e-01, 5.7058e-01, 1.0103e+00, ..., -8.5282e-01,
-6.4948e-01, 5.1806e-01],
...,
[-1.4377e+00, -1.1084e+00, -1.1205e+00, ..., -1.2813e-01,
-7.2892e-01, -1.2407e+00],
[-1.7347e+00, -9.0882e-01, -1.8376e+00, ..., -1.7659e+00,
-2.5637e+00, -1.4454e+00],
[-5.7197e-01, -2.7633e+00, -3.9447e+00, ..., -2.1468e+00,
-2.1808e+00, -8.9622e-01]],
[[-8.3594e-01, -2.3767e+00, -4.2512e+00, ..., -3.3274e+00,
-4.9742e+00, -4.1089e-01],
[-4.5685e+00, -6.0443e+00, -6.2491e+00, ..., -5.2391e+00,
-7.1472e+00, -1.6888e+00],
[-3.9455e+00, -5.5318e+00, -6.2983e+00, ..., -6.7853e+00,
-7.7532e+00, -2.6652e+00],
...,
[-3.4873e+00, -1.2648e+00, -1.9918e+00, ..., -4.1342e+00,
-5.0185e+00, -2.4084e+00],
[-6.5397e-01, 7.2645e-01, 2.2337e+00, ..., -8.0941e+00,
-8.9978e+00, -4.1226e+00],
[ 2.1151e-01, -2.8767e+00, -3.5713e+00, ..., -2.6900e+00,
-2.6791e+00, -3.0266e-01]],
[[-3.1615e-01, 2.0786e+00, 1.2235e+00, ..., 8.9907e-01,
2.2088e+00, 9.8497e-01],
[ 3.2593e-02, 1.0617e+00, 9.6488e-01, ..., 4.8584e-01,
1.3104e+00, -3.1106e-01],
[-1.0966e-01, 1.7202e+00, 6.6162e-01, ..., -5.9166e-03,
1.4230e+00, 5.9101e-02],
...,
[ 1.4386e+00, 1.1520e+00, 1.7108e+00, ..., 1.4377e+00,
1.2914e+00, 5.5122e-01],
[-9.3391e-01, -5.8067e+00, -7.6937e+00, ..., 1.4790e+00,
2.2321e+00, 7.7093e-01],
[ 1.6673e-01, 5.9493e-01, 3.2407e-01, ..., 4.1611e-01,
1.0270e+00, -6.9937e-01]],
...,
[[-8.4327e-01, -1.8518e+00, -1.9123e+00, ..., -2.5178e+00,
-2.8732e+00, -4.9473e-01],
[-3.2312e+00, -4.7401e+00, -3.2727e+00, ..., -3.6807e+00,
-4.0573e+00, -2.6321e+00],
[-1.1201e+00, -2.6862e+00, -1.1080e+00, ..., -3.1079e+00,
-4.2578e+00, -1.5795e+00],
...,
[-8.0209e-01, 2.3259e+00, 2.2750e+00, ..., -1.3849e+00,
-3.9144e+00, -1.8613e+00],
[-1.2105e+00, 3.6026e-01, 1.7612e+00, ..., -2.9318e+00,
-4.9828e+00, -1.5303e+00],
[-1.2937e+00, -3.0875e+00, -2.7225e+00, ..., -4.1693e-01,
-1.3503e+00, 1.2406e+00]],
[[-1.7496e+00, -2.1643e+00, -3.0133e+00, ..., -1.6053e+00,
-1.5235e+00, -1.2940e+00],
[-4.4277e+00, -3.2518e+00, -2.9573e+00, ..., -2.7514e+00,
-1.6288e+00, -7.6381e-01],
[-3.0776e+00, -2.1786e+00, -2.4159e+00, ..., -2.5204e-01,
-3.1140e-01, -5.5750e-01],
...,
[-2.5391e+00, -1.9754e-02, -4.7112e-01, ..., -1.6620e+00,
-2.6909e+00, -1.2203e+00],
[-3.8736e-01, 2.8665e+00, 3.1854e+00, ..., -1.0933e+00,
-2.3216e+00, -6.8294e-01],
[-2.8045e+00, -1.1800e+00, -8.6282e-01, ..., -4.6014e-01,
-1.0197e+00, -1.1784e-02]],
[[ 1.1251e+00, 6.5814e-01, 7.2908e-01, ..., -6.1040e-01,
1.6922e-01, -4.0772e-01],
[ 3.4747e+00, 3.4081e+00, 2.6009e+00, ..., 1.6327e+00,
1.9618e+00, 2.6205e+00],
[ 1.7517e+00, 1.2356e+00, 1.4814e+00, ..., 2.0313e+00,
2.2552e+00, 1.7335e+00],
...,
[ 2.1066e+00, 9.7175e-01, 1.1232e+00, ..., 1.4019e+00,
1.2571e+00, 4.8986e-01],
[ 3.6432e-01, -4.6179e-01, -3.9404e+00, ..., 2.7945e+00,
1.5340e+00, 1.7518e+00],
[ 1.8470e-01, -1.4577e+00, -4.5489e+00, ..., 5.9499e-01,
1.8309e-01, 1.0793e+00]]],
[[[ 3.0766e-01, 1.2814e+00, 1.1834e-01, ..., -3.0900e+00,
-2.4482e+00, -1.0760e+00],
[ 4.3594e-01, 8.1663e-01, 1.1238e+00, ..., -1.8192e+00,
-2.4139e+00, -2.6029e-01],
[-3.4329e-01, -3.6484e-01, 3.8832e-01, ..., -1.8871e+00,
-3.0721e+00, 1.9869e-02],
...,
[-6.6969e-01, -2.1599e+00, -3.0366e-01, ..., -2.0249e+00,
-3.7461e+00, 4.1179e-01],
[-1.8998e+00, -1.6832e+00, -8.6821e-01, ..., -2.8818e+00,
-3.3906e+00, 5.2914e-01],
[-1.1412e+00, -3.3924e+00, -3.3127e+00, ..., -3.7738e+00,
-3.1158e+00, -1.4768e+00]],
[[-1.0425e+00, -1.5536e+00, -3.3471e+00, ..., -6.3795e+00,
-7.3544e+00, -1.3638e+00],
[-3.8240e+00, -4.1531e+00, -3.4595e+00, ..., -6.0057e+00,
-8.4080e+00, -2.3743e+00],
[-3.7937e+00, -4.1414e+00, -4.1631e+00, ..., -6.5507e+00,
-7.9205e+00, -2.2043e+00],
...,
[-5.8239e+00, -6.0188e+00, -4.6619e+00, ..., -7.6508e+00,
-8.8534e+00, -2.0709e+00],
[-7.5234e+00, -7.6803e+00, -7.0677e+00, ..., -8.4168e+00,
-1.0630e+01, -4.4169e+00],
[-1.7220e+00, -3.2866e+00, -3.8193e+00, ..., -3.1564e+00,
-3.7100e+00, -8.2926e-01]],
[[-5.3251e-02, 1.2346e+00, 1.7008e+00, ..., 1.5808e+00,
2.3800e+00, 9.3313e-01],
[ 1.4955e-02, -7.6240e-02, 8.2222e-01, ..., 1.5210e+00,
2.9721e+00, 2.7922e-02],
[ 6.7215e-01, 1.1802e+00, 1.4096e+00, ..., 5.6066e-01,
1.9776e+00, -2.5821e-01],
...,
[ 3.3205e-01, 9.8683e-01, 1.0836e+00, ..., 1.1382e+00,
1.8721e+00, -1.1967e+00],
[ 5.1439e-01, 6.5510e-01, 1.0385e+00, ..., -9.0345e-01,
1.3102e+00, -1.5095e+00],
[ 5.4118e-01, 1.1711e+00, 1.9678e+00, ..., -6.6825e-01,
-1.9505e-01, -1.8626e+00]],
...,
[[-5.7731e-01, -1.5446e+00, -2.2115e+00, ..., -5.1055e+00,
-5.1453e+00, -7.0061e-01],
[-2.7151e+00, -2.5965e+00, -1.5214e+00, ..., -3.9197e+00,
-4.7705e+00, -1.7788e+00],
[-2.0759e+00, -1.2365e+00, -1.0160e+00, ..., -5.2540e+00,
-5.3781e+00, -1.3720e+00],
...,
[-1.9106e+00, -1.7979e+00, -1.2704e+00, ..., -3.7007e+00,
-5.5249e+00, -1.5296e+00],
[-3.0906e+00, -3.1171e+00, -2.3071e+00, ..., -3.2778e+00,
-5.3622e+00, -9.2962e-01],
[-1.5017e+00, -2.1043e+00, -2.4530e+00, ..., -2.1523e+00,
-2.4480e+00, 3.5840e-01]],
[[-1.2539e+00, -2.3176e+00, -3.9886e+00, ..., -1.5680e+00,
-4.6537e-01, -6.0600e-01],
[-3.5457e+00, -3.5115e+00, -4.6698e+00, ..., -3.6090e+00,
-1.9739e+00, -4.0620e-01],
[-4.4878e+00, -3.8620e+00, -4.1641e+00, ..., -2.6422e+00,
-1.8722e+00, -3.1326e-01],
...,
[-4.9045e+00, -3.8523e+00, -3.2971e+00, ..., -2.8070e+00,
-1.5900e+00, -1.0605e+00],
[-4.2021e+00, -1.8887e+00, -1.7598e+00, ..., -9.1871e-01,
-2.6452e-01, 1.1593e+00],
[-3.2126e+00, -3.1719e+00, -2.1820e+00, ..., -2.3021e-01,
-8.2487e-01, 6.3944e-01]],
[[ 1.2046e+00, -7.7647e-02, 1.1195e+00, ..., 6.0901e-01,
7.8399e-01, -1.4411e+00],
[ 2.2580e+00, 1.1523e+00, 1.3370e+00, ..., 3.5831e+00,
3.0414e+00, 2.2536e+00],
[ 1.3782e+00, 7.3707e-01, 4.4503e-01, ..., 2.6036e+00,
1.5544e+00, 1.2104e+00],
...,
[ 2.4840e+00, 1.3537e+00, 1.3455e+00, ..., 3.2742e+00,
2.5131e+00, 6.1271e-01],
[ 2.9266e+00, 3.7887e+00, 2.2293e+00, ..., 2.8483e+00,
1.1895e+00, 1.8363e+00],
[ 1.9007e+00, 1.5391e+00, 7.6686e-01, ..., 8.6049e-01,
-1.4288e-01, 9.0022e-01]]],
[[[ 1.5528e-01, -2.0249e-01, -7.2952e+00, ..., -1.0170e+00,
-1.4341e+00, 1.0037e+00],
[ 1.8318e-01, 7.1083e-01, -7.5605e+00, ..., -2.9442e+00,
-4.5561e+00, -1.9840e+00],
[-6.0787e-01, 3.6172e+00, -8.8802e+00, ..., -1.7954e+00,
-1.2073e+01, -3.1519e+00],
...,
[-6.9701e-01, -1.5429e+01, 9.1519e-01, ..., 6.4278e+00,
1.6775e+00, -1.9774e+00],
[-5.3264e+00, -1.0502e+01, 3.4493e+00, ..., 5.1261e+00,
-9.2676e-01, -1.6285e+00],
[-3.0446e+00, -2.9145e+00, 3.6683e+00, ..., 4.2082e+00,
-1.9764e+00, -1.8521e+00]],
[[-1.8476e+00, -4.8082e+00, -1.6137e+01, ..., -1.2546e+00,
-1.8977e+00, 1.9048e+00],
[-6.2518e+00, -6.0164e+00, -1.6098e+01, ..., -4.1761e+00,
-9.4814e-01, 1.2687e+00],
[-5.7489e+00, -4.7913e+00, -2.6752e+01, ..., 5.9423e-01,
-1.3910e+01, -1.1766e+00],
...,
[-8.1004e+00, -1.7232e+01, 2.6881e+00, ..., 3.6921e+00,
-5.2657e+00, -7.5908e+00],
[-1.2363e+01, -1.2551e+01, 4.2247e+00, ..., -7.1867e-01,
-5.4274e+00, -6.3944e+00],
[-5.7414e-01, -1.2900e+00, 4.0284e+00, ..., 1.8004e+00,
-3.0024e+00, -1.6593e+00]],
[[-1.6629e-01, 1.6837e+00, 4.1822e+00, ..., -4.1438e-01,
-1.3035e-01, 2.5960e-01],
[-1.3473e+00, -1.3428e+00, 2.2475e+00, ..., -1.8261e-01,
-5.4456e+00, -6.7209e+00],
[-9.2195e-01, -2.4808e+00, -7.8572e-01, ..., -5.4808e+00,
-1.5445e+01, -1.1288e+01],
...,
[-3.0558e+00, 3.9358e+00, -1.1496e+01, ..., -2.5015e+00,
7.0063e+00, 2.0120e+00],
[-1.6775e+00, 2.3381e+00, -1.3595e+01, ..., -1.8109e+00,
5.6920e+00, -1.7858e-02],
[-1.8193e+00, -6.1077e+00, -1.7007e+01, ..., -2.3603e+00,
2.4109e+00, -1.2553e+00]],
...,
[[-6.9899e-01, -4.1976e+00, -4.4981e+00, ..., -7.6732e-01,
2.1085e+00, 1.5197e+00],
[-2.7681e+00, -5.3688e+00, -2.7399e+00, ..., -4.6158e+00,
1.1536e-01, -3.5933e-01],
[-1.3311e+00, -4.4775e+00, -6.9918e+00, ..., -9.4113e-01,
-2.8149e+00, -1.9187e-01],
...,
[-8.3771e-01, -4.2838e+00, 6.8371e+00, ..., 2.3444e+00,
-9.9106e+00, -1.5892e+00],
[-2.8876e+00, -2.7115e+00, 5.1794e+00, ..., -1.2691e+00,
-9.1276e+00, 7.0996e-01],
[-8.2890e-01, 5.9382e-01, 2.7257e+00, ..., -1.5385e+00,
-4.1232e+00, 8.6755e-01]],
[[-1.3222e+00, -6.4093e+00, -4.4073e+00, ..., -3.3235e+00,
-1.0643e-02, 1.5943e+00],
[-2.2806e+00, -4.3984e+00, -2.3713e+00, ..., -8.6817e+00,
-1.5796e+00, 1.7423e+00],
[-6.2827e-02, -5.5040e+00, -3.0823e+00, ..., -1.3117e+01,
-5.2419e+00, 1.0885e+00],
...,
[-8.1160e+00, -5.1505e+00, -3.0492e+00, ..., 3.2300e+00,
-7.3582e+00, -8.5094e-01],
[-7.9695e+00, -3.7034e+00, -9.1692e+00, ..., 1.2143e+00,
-6.6571e+00, 2.3985e-01],
[-2.4805e-01, 1.1884e+00, -7.9307e+00, ..., 7.2132e-01,
-3.4486e+00, 7.7369e-01]],
[[ 1.4080e+00, 1.7670e+00, 4.5265e+00, ..., -5.6060e+00,
-2.9234e+00, 1.0716e-01],
[ 3.4647e+00, 5.2781e+00, 5.6077e+00, ..., -1.0290e+01,
-8.9429e+00, -1.8248e+00],
[ 1.4701e+00, 6.1860e+00, 8.5530e+00, ..., -1.2629e+01,
-1.2133e+01, -2.9748e+00],
...,
[ 3.9285e+00, 1.0257e+00, 5.8984e+00, ..., 4.3733e+00,
4.1676e+00, 2.9808e-01],
[ 3.7665e+00, 6.6087e+00, 7.7678e+00, ..., 8.2687e+00,
2.3975e+00, 5.0437e-01],
[ 1.7330e+00, 4.8391e+00, 4.0087e+00, ..., 6.6854e+00,
6.2004e-01, 1.2659e+00]]]], device='cuda:0'),
tensor([[[[ 9.0091e-01, 1.5838e+00, 3.8596e-01, ..., -3.6236e+00,
-2.4979e+00, -1.0033e+00],
[ 1.7781e+00, 1.0971e+00, 7.5356e-01, ..., -2.7860e+00,
-2.5340e+00, -2.0457e-01],
[ 1.9351e-01, -4.5777e-01, -5.9120e-01, ..., -2.4083e+00,
-2.8635e+00, 3.2157e-02],
...,
[ 1.0520e+00, -3.0465e-01, -1.2084e-01, ..., -1.0040e+00,
-4.2486e+00, 3.9037e-01],
[ 2.2845e-02, -4.9707e-01, -7.3798e-01, ..., -3.0550e+00,
-3.9968e+00, 5.4965e-01],
[-3.7431e-01, -3.5233e+00, -3.3624e+00, ..., -3.7634e+00,
-3.4521e+00, -1.5460e+00]],
[[ 4.2209e-01, -1.1954e+00, -2.6904e+00, ..., -7.0916e+00,
-7.2928e+00, -1.3082e+00],
[-9.6800e-01, -2.7544e+00, -3.9008e+00, ..., -7.6686e+00,
-8.4689e+00, -2.4329e+00],
[-7.7224e-01, -1.4332e+00, -1.6112e+00, ..., -6.7579e+00,
-7.7303e+00, -2.4379e+00],
...,
[-2.0811e+00, -3.2826e+00, -3.9160e+00, ..., -6.5617e+00,
-9.2913e+00, -1.9326e+00],
[-4.6296e+00, -6.2268e+00, -5.5859e+00, ..., -1.0509e+01,
-1.1180e+01, -4.4740e+00],
[-4.0670e-01, -3.3631e+00, -3.6254e+00, ..., -4.7415e+00,
-3.5901e+00, -7.8692e-01]],
[[ 4.6112e-01, 2.1391e+00, 1.8303e+00, ..., 9.0210e-01,
2.2822e+00, 9.6705e-01],
[-3.5072e-01, 1.0634e+00, 7.7245e-01, ..., 1.0327e+00,
2.8241e+00, 4.9776e-02],
[ 8.8933e-01, 4.2306e+00, 2.7422e+00, ..., -8.9681e-01,
1.8968e+00, -2.5935e-01],
...,
[ 6.1630e-01, 3.0109e+00, 9.8717e-01, ..., 1.4103e+00,
1.6996e+00, -1.2631e+00],
[ 7.8002e-01, 1.7352e+00, 3.3046e-01, ..., 4.0154e-01,
7.2466e-01, -1.6168e+00],
[-1.9912e-01, 1.8929e+00, 1.1926e+00, ..., -1.1720e+00,
-7.1628e-01, -1.8204e+00]],
...,
[[-2.5831e-01, -9.3672e-01, -1.6648e+00, ..., -5.4536e+00,
-5.3296e+00, -7.2604e-01],
[-2.3809e+00, -3.9822e+00, -2.2192e+00, ..., -4.5663e+00,
-5.2248e+00, -1.8447e+00],
[-9.3817e-01, -2.5709e+00, -1.2390e+00, ..., -5.1641e+00,
-5.6386e+00, -1.2902e+00],
...,
[-6.8789e-01, -2.0309e+00, -5.6408e-01, ..., -5.6078e+00,
-4.6971e+00, -1.8944e+00],
[-2.1039e+00, -3.9446e+00, -2.0664e+00, ..., -4.5400e+00,
-4.9203e+00, -1.2269e+00],
[-1.2803e+00, -3.7573e+00, -2.6229e+00, ..., -1.9171e+00,
-2.2697e+00, 2.6609e-01]],
[[-9.8210e-01, -8.7362e-01, -1.1284e+00, ..., -2.1536e+00,
-4.0782e-01, -5.9975e-01],
[-2.8681e+00, -2.6418e+00, -1.4792e+00, ..., -3.8427e+00,
-1.8011e+00, -4.1830e-01],
[-2.4688e+00, -4.4423e+00, -9.1504e-01, ..., -2.2670e+00,
-1.6828e+00, -4.2927e-01],
...,
[-2.9578e+00, -4.8794e+00, -3.0748e+00, ..., -5.2921e+00,
-2.4941e+00, -1.0855e+00],
[-3.2351e+00, -2.2790e+00, -8.8145e-01, ..., -3.6705e+00,
-8.5248e-01, 1.2116e+00],
[-3.3639e+00, -3.4058e+00, -1.4287e+00, ..., -5.2844e-01,
-8.8119e-01, 6.5895e-01]],
[[ 8.6406e-01, 9.0058e-01, 9.5854e-01, ..., -2.3325e-01,
6.5039e-01, -1.3651e+00],
[ 1.3208e+00, 1.7323e+00, 1.6098e+00, ..., 3.1165e+00,
3.0708e+00, 2.3083e+00],
[ 5.4939e-01, -1.7720e-01, 2.4580e-01, ..., 2.3591e+00,
1.6770e+00, 1.2365e+00],
...,
[ 1.7150e+00, 4.3011e-01, 1.3398e+00, ..., 3.2039e+00,
1.9119e+00, 6.6070e-01],
[ 3.3841e+00, 3.7759e+00, 2.4622e+00, ..., 1.3616e+00,
1.1867e+00, 1.9192e+00],
[ 1.9881e+00, 6.7943e-01, 8.1111e-01, ..., 3.7430e-01,
2.8384e-01, 9.1349e-01]]],
[[[ 4.7762e-02, 1.1643e+00, 1.0640e+00, ..., -6.0982e+00,
-2.6991e+00, 2.8031e-01],
[-2.9031e-02, -2.2118e-01, 1.6737e-01, ..., -1.3922e+00,
-7.4370e-01, -2.0954e-01],
[ 1.0045e-02, 8.3980e-01, 1.9027e+00, ..., 2.5631e+00,
2.8179e+00, -5.5326e-01],
...,
[ 1.6129e+00, -1.5089e+00, -1.3990e+01, ..., 7.9064e-01,
-2.3306e+00, -1.1974e+00],
[-1.5222e+00, -4.1260e+00, -1.0893e+01, ..., -5.8724e-01,
-4.1473e+00, -2.1941e+00],
[ 3.2449e-01, -5.5283e+00, -5.4051e+00, ..., 2.7705e-01,
-2.5806e+00, -1.7011e+00]],
[[-1.0699e+00, -2.4687e+00, -3.5190e+00, ..., -5.5110e+00,
-5.8280e+00, -1.0028e+00],
[-4.6768e+00, -5.8527e+00, -6.3142e+00, ..., -2.4733e+00,
-6.8023e+00, -3.7993e+00],
[-3.6362e+00, -6.4408e+00, -1.0329e+01, ..., 1.5140e+00,
-3.3740e+00, -4.3607e+00],
...,
[-5.2988e+00, -1.0497e+01, -1.8760e+01, ..., -8.1906e+00,
-9.4652e+00, -2.2557e+00],
[-8.2095e+00, -1.0837e+01, -6.8768e+00, ..., -6.7478e+00,
-1.3132e+01, -4.2102e+00],
[-1.1841e-01, -3.4388e+00, -8.4888e-01, ..., -4.2173e-01,
-6.0486e+00, -4.4253e-01]],
[[-3.5155e-01, 1.9524e+00, 9.3091e-01, ..., 3.8580e+00,
3.3638e+00, 8.0934e-01],
[ 3.6321e-02, 8.6288e-01, 9.3137e-01, ..., 7.9538e+00,
5.8519e+00, -8.0167e-01],
[ 3.7020e-03, 2.6129e+00, 1.5367e+00, ..., -5.8393e+00,
5.2691e+00, 2.3017e+00],
...,
[ 1.4715e-01, -8.2766e-01, 3.9632e+00, ..., 4.6651e+00,
1.4809e+00, 2.4484e-01],
[-3.4542e-01, 6.3310e-01, 4.5762e+00, ..., 7.1056e+00,
2.9077e+00, 4.9660e-02],
[-1.4805e+00, 8.2373e-01, -3.2105e+00, ..., 3.2567e+00,
8.4797e-01, -7.1172e-01]],
...,
[[-1.1141e+00, -1.8407e+00, -5.1281e-01, ..., -6.9295e+00,
-4.9862e+00, -9.8710e-01],
[-3.4197e+00, -4.7315e+00, -1.9965e+00, ..., -3.8674e+00,
-6.7396e+00, -3.9428e+00],
[-1.5243e+00, -3.1099e+00, -3.7673e-01, ..., -1.3697e+00,
-5.0967e+00, -1.4552e+00],
...,
[ 7.9898e-02, -3.4704e+00, -9.5792e+00, ..., -4.9537e+00,
-6.5489e+00, -2.9894e+00],
[-1.8713e+00, -4.3518e+00, -1.8358e+00, ..., -7.5297e+00,
-4.6094e+00, -2.4068e+00],
[ 7.3333e-01, -1.1321e+00, 2.0899e+00, ..., -5.0143e+00,
-3.6819e-01, 3.8349e-01]],
[[-1.7359e+00, -1.8704e+00, -2.1899e+00, ..., 1.5037e+00,
-1.9583e-01, -1.2501e+00],
[-4.5172e+00, -3.0522e+00, -2.4536e+00, ..., -3.4434e+00,
-2.2515e+00, 3.4309e-01],
[-2.8031e+00, -2.1105e+00, -2.1530e+00, ..., 1.8418e+00,
-2.1081e+00, 4.0405e-01],
...,
[-4.9101e+00, -9.1027e+00, -5.9107e+00, ..., 3.3664e+00,
-1.2468e+00, -8.9152e-01],
[-5.4307e+00, -2.2520e+00, -2.2603e+00, ..., -1.1005e+00,
2.8487e+00, 3.1517e-02],
[-1.9618e+00, -1.5188e-01, 1.4033e+00, ..., -3.3694e+00,
2.5195e+00, 8.0530e-01]],
[[ 1.1464e+00, 4.8049e-01, 5.9547e-01, ..., 2.1118e-01,
1.3648e-01, -9.6556e-01],
[ 3.5212e+00, 3.2298e+00, 1.8307e+00, ..., 5.7597e+00,
6.7696e+00, 3.0540e+00],
[ 1.6575e+00, 1.6090e+00, 3.0070e+00, ..., 1.3377e+00,
4.7090e+00, 1.6709e+00],
...,
[ 2.6327e+00, 3.4097e+00, -1.1269e+00, ..., 6.4767e+00,
5.9857e-01, -5.5872e-02],
[ 3.5312e+00, 2.1982e+00, -2.5901e+00, ..., 7.1786e+00,
1.2647e-01, 1.0569e+00],
[ 1.9987e+00, -5.3757e-01, 5.7102e-01, ..., 8.0283e-01,
-3.6702e-01, 1.2278e+00]]],
[[[ 2.7186e-01, -3.8600e-01, -4.1300e+00, ..., -2.7156e+00,
-3.6105e+00, -1.6907e+00],
[ 7.1400e-01, -9.0955e-01, -4.7220e+00, ..., -1.6794e+00,
-1.4942e+00, -8.0460e-01],
[-1.8312e-01, -1.8135e+00, -3.7067e+00, ..., -2.0540e-01,
-9.2552e-01, -1.2781e+00],
...,
[-1.0248e+00, -2.0327e+00, -2.1305e+00, ..., -2.7517e+00,
-8.1504e-01, -5.2842e-01],
[-1.0933e+00, -8.6579e-01, -1.3079e+00, ..., -1.0884e+00,
-9.5019e-01, -1.7480e+00],
[-2.6707e+00, -5.0176e+00, -4.3184e+00, ..., -5.5767e+00,
-4.3682e+00, -3.7921e+00]],
[[-1.6111e+00, -2.3690e+00, -6.1686e+00, ..., -6.2817e+00,
-8.7006e+00, -2.2705e+00],
[-5.0373e+00, -5.2249e+00, -5.5303e+00, ..., -6.7626e+00,
-8.0157e+00, -3.0913e+00],
[-3.9862e+00, -5.4517e+00, -7.6133e+00, ..., -8.3431e+00,
-1.0616e+01, -5.5035e+00],
...,
[-5.6132e+00, -4.8757e+00, -4.6000e+00, ..., -8.5465e+00,
-5.7050e+00, -3.2793e+00],
[-7.1305e+00, -7.8302e+00, -6.8679e+00, ..., -8.0193e+00,
-8.2921e+00, -6.7316e+00],
[-2.5280e+00, -4.3605e+00, -3.9226e+00, ..., -4.1265e+00,
-4.2876e+00, -2.6399e+00]],
[[ 5.4274e-01, 1.9614e+00, 2.8714e+00, ..., 1.1426e+00,
2.8949e+00, 9.4666e-01],
[-6.2159e-01, -1.0883e-03, 2.1302e+00, ..., 1.1404e+00,
2.8718e+00, 2.9868e-01],
[ 8.3619e-01, 5.4170e-01, -4.6442e-01, ..., -4.3312e-02,
2.1258e-01, -6.1572e-01],
...,
[ 3.4743e-01, 8.5883e-01, 1.1712e-01, ..., 1.8843e+00,
-7.4256e-01, -5.6336e-01],
[ 7.8079e-01, 5.0354e-01, -2.7069e-01, ..., -4.9575e-01,
-1.4686e+00, -3.8359e-02],
[ 1.3103e+00, 1.6508e+00, 1.1239e+00, ..., 3.4879e-01,
1.1700e+00, 2.5355e-01]],
...,
[[-7.5810e-01, -2.9380e+00, -2.5066e+00, ..., -5.5250e+00,
-6.0294e+00, -1.8429e+00],
[-2.4264e+00, -4.5812e+00, -2.3077e+00, ..., -4.7753e+00,
-5.4450e+00, -2.4924e+00],
[-2.0401e+00, -3.8822e+00, -2.7027e+00, ..., -4.9710e+00,
-6.5548e+00, -2.9129e+00],
...,
[-1.8306e+00, -1.4672e+00, -1.2312e+00, ..., -2.7316e+00,
-2.1994e+00, -1.4928e+00],
[-2.9736e+00, -2.9213e+00, -2.3377e+00, ..., -1.1509e+00,
-4.4941e+00, -2.7203e+00],
[-3.3944e+00, -3.5832e+00, -3.1358e+00, ..., -3.4026e+00,
-4.2836e+00, -1.6217e+00]],
[[-1.7272e+00, -3.0606e+00, -9.5437e-01, ..., -2.5277e+00,
-1.1127e+00, 5.7994e-01],
[-3.7234e+00, -3.6248e+00, -5.0934e-01, ..., -2.8546e+00,
-2.4174e+00, -1.0752e-01],
[-2.6977e+00, -1.2349e+00, 4.5878e-01, ..., -5.3209e-01,
-1.0417e+00, 1.9994e-01],
...,
[-3.1470e+00, -1.7554e+00, -3.5091e+00, ..., -4.1103e+00,
-2.0146e+00, -3.8596e-01],
[-3.4864e+00, -8.2496e-01, -2.7413e+00, ..., -1.5912e+00,
-2.9058e+00, -6.0751e-01],
[-4.2546e+00, -3.0429e+00, -3.2235e+00, ..., -2.2426e+00,
-3.0617e+00, -1.0375e+00]],
[[ 1.4666e+00, 1.0349e+00, 6.8743e-01, ..., 1.7009e-01,
1.7792e+00, -1.2056e-01],
[ 3.1207e+00, 2.8753e+00, 2.7011e+00, ..., 2.8745e+00,
3.2403e+00, 3.8520e+00],
[ 2.0942e+00, 2.4364e+00, 1.2711e+00, ..., 3.4986e+00,
2.4701e+00, 1.5871e+00],
...,
[ 2.1469e+00, 3.5028e-01, 1.9063e+00, ..., 1.5725e+00,
2.2769e+00, 1.8129e+00],
[ 3.7283e+00, 3.5899e+00, 2.2609e+00, ..., 2.9891e+00,
2.8193e+00, 2.3445e+00],
[ 1.9572e+00, 6.6202e-01, -2.7375e-01, ..., 1.2092e+00,
-3.5905e-01, -2.5717e-01]]],
[[[ 3.6036e-01, 1.2554e+00, 9.0520e-01, ..., -4.1465e+00,
-2.4923e+00, -8.9307e-01],
[ 3.9428e-01, 1.0656e+00, 2.2704e+00, ..., -2.7754e+00,
-2.6712e+00, -1.5432e-01],
[-4.4126e-01, 1.0618e+00, 1.2107e+00, ..., -1.1877e+00,
-3.5169e+00, 7.1904e-02],
...,
[-5.9241e-01, -1.5841e+00, -7.6160e+00, ..., -5.6121e+00,
-4.6174e+00, 1.0683e+00],
[-1.9288e+00, -9.0446e-01, -4.8417e+00, ..., -5.1445e+00,
-3.4787e+00, 7.3609e-01],
[-1.3239e+00, -2.6741e+00, -3.7878e+00, ..., -3.8248e+00,
-2.8314e+00, -1.3319e+00]],
[[-1.0847e+00, -1.5047e+00, -3.5692e+00, ..., -7.7993e+00,
-7.1381e+00, -1.1887e+00],
[-4.0975e+00, -4.2977e+00, -3.0855e+00, ..., -9.0871e+00,
-8.2967e+00, -2.3344e+00],
[-3.9075e+00, -5.3583e+00, -5.8125e+00, ..., -1.0185e+01,
-8.8050e+00, -2.4332e+00],
...,
[-6.6817e+00, -5.5885e+00, -9.9880e+00, ..., -1.1273e+01,
-8.5743e+00, -1.6508e+00],
[-8.3584e+00, -7.2123e+00, -6.6533e+00, ..., -9.4868e+00,
-1.0189e+01, -4.6173e+00],
[-1.9227e+00, -3.2360e+00, -2.8253e+00, ..., -2.7969e+00,
-3.7235e+00, -7.0564e-01]],
[[-5.2495e-03, 1.1676e+00, 2.0035e+00, ..., 1.8524e+00,
2.1975e+00, 8.9115e-01],
[ 3.1548e-02, -5.5186e-02, -4.4490e-01, ..., 2.0097e+00,
2.4569e+00, -9.7384e-02],
[ 6.0033e-01, 1.7319e+00, -6.2534e-01, ..., 1.4379e+00,
1.3646e+00, -3.2470e-01],
...,
[ 3.1381e-01, -1.7477e+00, -4.1413e-01, ..., 9.3669e-01,
1.8798e-01, -1.1059e+00],
[ 4.6801e-01, -9.5408e-01, -1.6396e-01, ..., -1.0159e+00,
3.3025e-01, -1.4425e+00],
[ 6.6947e-01, 9.8333e-01, 1.0200e-01, ..., -3.6501e-01,
-2.2188e-01, -1.7881e+00]],
...,
[[-6.3926e-01, -9.5754e-01, -1.5397e+00, ..., -6.2715e+00,
-5.1723e+00, -6.3029e-01],
[-2.6969e+00, -1.9485e+00, -1.2589e+00, ..., -4.7715e+00,
-6.0450e+00, -1.6777e+00],
[-1.9038e+00, -1.0719e+00, -2.4897e+00, ..., -3.4386e+00,
-7.9509e+00, -1.0917e+00],
...,
[-1.8446e+00, -2.1560e+00, -6.4412e+00, ..., -3.6535e+00,
-4.8248e+00, -2.1230e+00],
[-3.3286e+00, -2.4911e+00, -3.9845e+00, ..., -2.8992e+00,
-5.1708e+00, -1.2704e+00],
[-1.5509e+00, -1.4776e+00, -2.8245e+00, ..., -2.0777e+00,
-2.6424e+00, 4.1991e-01]],
[[-1.3527e+00, -1.7314e+00, -4.4737e+00, ..., -1.7763e+00,
-6.2368e-01, -5.0105e-01],
[-3.7708e+00, -2.3476e+00, -4.2594e+00, ..., -3.3911e+00,
-2.5130e+00, -1.7637e-01],
[-4.6483e+00, -3.6872e+00, -5.8381e+00, ..., -4.3484e-01,
-2.4305e+00, -4.2847e-01],
...,
[-4.3014e+00, -5.7971e+00, -5.6483e+00, ..., -4.2740e+00,
-2.4161e+00, -6.3441e-01],
[-3.5105e+00, -6.5029e-01, -4.7987e+00, ..., -2.3447e+00,
-4.9889e-01, 1.2416e+00],
[-2.9274e+00, -1.7644e+00, -3.0531e+00, ..., 2.5612e-01,
-6.4936e-01, 6.2581e-01]],
[[ 1.1925e+00, -1.1731e-01, 9.1298e-01, ..., 4.5409e-01,
7.0066e-01, -1.4206e+00],
[ 2.3345e+00, 8.7861e-01, 1.2905e+00, ..., 3.6987e+00,
2.0974e+00, 2.1385e+00],
[ 1.3675e+00, 1.1976e+00, 2.3792e+00, ..., 2.9767e+00,
3.6472e-01, 1.3513e+00],
...,
[ 2.6765e+00, 2.3569e+00, 1.3512e+00, ..., 1.1102e+00,
2.3042e+00, 3.4988e-01],
[ 2.8356e+00, 3.3686e+00, 1.6010e+00, ..., 3.1610e+00,
1.2259e+00, 1.7109e+00],
[ 2.1509e+00, 2.4062e+00, 6.2191e-01, ..., 2.0486e+00,
-2.7812e-01, 9.1168e-01]]],
[[[ 8.9716e-01, 1.6265e+00, 5.7834e-01, ..., -3.6862e+00,
-2.6785e+00, -1.2974e+00],
[ 1.3448e+00, 7.4387e-01, 8.7849e-01, ..., -3.0766e+00,
-2.4564e+00, -2.7409e-01],
[-1.7616e-01, -1.4344e+00, -1.5679e+00, ..., -2.3953e+00,
-2.7964e+00, -2.2211e-01],
...,
[ 5.4172e-01, -8.7741e-01, -6.2660e-01, ..., -3.0815e+00,
-3.7672e+00, 5.1202e-01],
[ 2.6304e-01, -2.6968e-01, -4.7745e-01, ..., -4.0607e+00,
-3.8580e+00, 5.2712e-01],
[-2.7771e-01, -3.4909e+00, -3.2900e+00, ..., -3.7536e+00,
-3.2746e+00, -1.2261e+00]],
[[ 4.0883e-01, -1.2017e+00, -2.6014e+00, ..., -7.7277e+00,
-8.0251e+00, -1.6752e+00],
[-7.4360e-01, -2.2695e+00, -3.3258e+00, ..., -8.1728e+00,
-8.7384e+00, -2.5771e+00],
[-1.9055e+00, -2.4930e+00, -2.0657e+00, ..., -7.0898e+00,
-7.8779e+00, -2.8179e+00],
...,
[-2.3824e+00, -3.7437e+00, -3.3315e+00, ..., -7.6681e+00,
-8.6725e+00, -1.7918e+00],
[-4.1989e+00, -5.9851e+00, -5.1001e+00, ..., -9.8695e+00,
-1.0985e+01, -4.1261e+00],
[-3.4752e-01, -3.3782e+00, -3.5795e+00, ..., -3.8467e+00,
-3.6034e+00, -6.3553e-01]],
[[ 4.9181e-01, 2.1267e+00, 1.8381e+00, ..., 7.5947e-01,
2.3772e+00, 9.8389e-01],
[-1.7601e-01, 1.1391e+00, 8.2734e-01, ..., 9.1583e-01,
2.8817e+00, 4.0681e-02],
[ 1.0222e+00, 4.5556e+00, 3.3498e+00, ..., -7.4138e-01,
1.7650e+00, -4.3328e-01],
...,
[ 1.2605e+00, 2.2844e+00, 8.2185e-01, ..., -6.1883e-02,
1.5246e+00, -1.1198e+00],
[ 4.8151e-01, 2.1449e+00, 7.3374e-01, ..., -3.1852e-02,
1.2438e+00, -1.5480e+00],
[-3.3146e-01, 1.8774e+00, 9.7668e-01, ..., -5.4664e-02,
-3.2544e-01, -1.9422e+00]],
...,
[[-3.2915e-01, -8.2197e-01, -1.5312e+00, ..., -5.6151e+00,
-5.7781e+00, -8.5681e-01],
[-2.6274e+00, -4.1054e+00, -2.3177e+00, ..., -4.6331e+00,
-5.3601e+00, -1.9453e+00],
[-1.8839e+00, -3.7198e+00, -2.6718e+00, ..., -5.2160e+00,
-6.0954e+00, -1.7321e+00],
...,
[-1.7061e+00, -2.8855e+00, -1.3069e+00, ..., -3.5330e+00,
-5.2414e+00, -1.1897e+00],
[-1.9706e+00, -3.8178e+00, -1.7947e+00, ..., -3.7721e+00,
-5.2196e+00, -8.1360e-01],
[-1.1796e+00, -3.7086e+00, -2.4526e+00, ..., -1.6011e+00,
-1.9922e+00, 4.4235e-01]],
[[-1.0387e+00, -7.8532e-01, -1.1569e+00, ..., -2.3134e+00,
-8.6818e-01, -6.9674e-01],
[-3.0186e+00, -2.2488e+00, -9.8078e-01, ..., -3.8120e+00,
-2.0259e+00, -4.5480e-01],
[-3.7409e+00, -5.2855e+00, -1.7187e+00, ..., -2.2724e+00,
-1.7744e+00, -5.5486e-01],
...,
[-3.0730e+00, -4.1151e+00, -2.5523e+00, ..., -3.2554e+00,
-2.1181e+00, -8.6561e-01],
[-3.1824e+00, -2.2218e+00, -1.0406e+00, ..., -1.3972e+00,
-5.6105e-01, 1.2100e+00],
[-3.3558e+00, -3.3829e+00, -1.4520e+00, ..., -1.1573e-01,
-6.8675e-01, 7.3510e-01]],
[[ 8.9498e-01, 8.2751e-01, 9.5408e-01, ..., -2.1889e-01,
7.9669e-01, -1.4026e+00],
[ 1.1557e+00, 1.8543e+00, 1.5449e+00, ..., 3.3081e+00,
3.0230e+00, 2.5005e+00],
[ 2.0459e-01, -3.3277e-01, 8.2662e-01, ..., 2.6437e+00,
1.6429e+00, 1.4632e+00],
...,
[ 2.0811e+00, 1.6281e+00, 1.6258e+00, ..., 2.2831e+00,
1.6452e+00, 6.9883e-01],
[ 3.1571e+00, 3.4048e+00, 2.1220e+00, ..., 1.0817e+00,
9.5382e-01, 1.9070e+00],
[ 1.9644e+00, 5.9702e-01, 7.3735e-01, ..., 8.3840e-02,
-1.2866e-01, 8.3222e-01]]]], device='cuda:0'),
tensor([[[[-1.4352e-01, 6.0133e-01, 1.8624e+00, ..., -1.6699e+00,
-2.0882e+00, -1.7960e+00],
[ 1.5317e+00, 2.6972e+00, 4.6767e+00, ..., -3.2706e-01,
-1.8869e+00, -3.1526e+00],
[ 1.0153e+00, -6.9955e-02, -4.7268e+00, ..., -1.0791e+01,
-1.0327e+01, -4.0658e+00],
...,
[-3.0906e-01, -4.9347e+00, -2.6708e+00, ..., 6.1405e+00,
-1.3575e+00, -1.2843e+00],
[-1.5434e+00, -4.7624e+00, -1.9017e+00, ..., 3.6116e+00,
-2.3876e+00, -1.2726e+00],
[-2.5077e+00, -6.3992e+00, -3.7051e+00, ..., 8.9131e-01,
-2.2531e+00, -2.7961e+00]],
[[-6.1146e-01, -1.3658e+00, -2.1875e+00, ..., -8.5668e+00,
-7.6697e+00, -3.9396e+00],
[-3.3529e+00, -5.4947e+00, -5.2968e+00, ..., -6.3912e+00,
-1.1502e+01, -8.8307e+00],
[-4.1090e+00, -1.0791e+01, -8.6067e+00, ..., -1.6753e+01,
-8.7172e+00, -2.6902e+00],
...,
[-6.7390e+00, -9.3497e+00, -4.8238e+00, ..., -2.4566e+00,
-6.6332e+00, -2.8983e+00],
[-9.2286e+00, -1.2096e+01, -8.4505e+00, ..., -5.3219e+00,
-1.1229e+01, -3.0372e+00],
[-2.4412e+00, -4.9665e+00, -4.8256e+00, ..., 3.7228e-01,
-4.7782e+00, -1.2457e+00]],
[[ 9.0941e-01, 1.0196e+00, 1.7444e+00, ..., 1.9097e+00,
1.4326e+00, 7.1343e-01],
[ 1.2663e+00, 8.7944e-01, -1.1694e+00, ..., 2.8780e+00,
1.7180e+00, 1.3487e+00],
[-7.2060e-01, 4.7353e-01, 2.9951e+00, ..., 9.6647e+00,
8.9692e+00, 4.0293e+00],
...,
[ 3.2036e-01, 1.5533e+00, -3.9834e-01, ..., -1.4950e+00,
2.5739e+00, -4.8618e-01],
[-5.2606e-01, -2.1667e-01, -2.1445e+00, ..., 3.4115e+00,
6.9028e-01, -1.0993e+00],
[ 8.3559e-01, 1.0068e-01, -1.5090e+00, ..., 3.3138e-01,
-4.2358e-01, -8.6291e-01]],
...,
[[-1.0372e+00, 2.8585e-01, -1.4172e+00, ..., -4.2158e+00,
-6.3918e+00, -2.0795e+00],
[-1.7920e+00, -1.3270e+00, -2.3195e+00, ..., -2.9242e+00,
-7.3923e+00, -4.6171e+00],
[-2.4543e+00, -3.6784e+00, -4.2026e+00, ..., -1.2467e+01,
-1.1278e+01, -5.1436e+00],
...,
[-1.3689e+00, -3.0569e+00, -4.8746e-01, ..., -5.3249e+00,
-7.1094e+00, -1.8113e+00],
[-1.8403e+00, -4.6371e+00, -8.9745e-01, ..., -6.4111e+00,
-3.2338e+00, -1.2768e+00],
[-2.6811e+00, -4.4148e+00, -2.2901e+00, ..., -4.3669e+00,
-2.5585e-01, -1.4697e+00]],
[[-1.6733e+00, -4.4375e-01, -3.1794e+00, ..., -2.8562e+00,
-3.7243e+00, -7.8640e-01],
[-4.7154e+00, -2.5110e+00, -4.7333e+00, ..., -8.7255e-01,
-3.7789e+00, -1.6179e+00],
[-2.8628e+00, -3.6551e+00, -7.4039e+00, ..., -1.0316e+01,
-3.8067e+00, 1.7624e+00],
...,
[-3.8316e+00, -2.6730e+00, -8.2166e-01, ..., 2.5363e+00,
-3.8456e+00, -1.1758e+00],
[-3.4568e+00, 1.2205e-01, 1.7636e+00, ..., -2.5279e+00,
-1.3479e+00, -6.2977e-02],
[-3.8016e+00, -1.3497e+00, 1.9611e+00, ..., -3.9999e+00,
1.1471e+00, 9.6817e-01]],
[[ 8.3362e-01, -1.6049e-01, 7.4726e-01, ..., 1.1506e+00,
9.7833e-01, 4.1984e-01],
[ 2.0196e+00, 2.4725e+00, 3.0631e+00, ..., 3.5897e+00,
1.7081e+00, 2.7083e+00],
[ 2.0770e+00, 2.0181e+00, -1.0779e+00, ..., -2.4730e-02,
3.6860e+00, 2.1567e+00],
...,
[ 2.8248e+00, 1.6007e+00, 1.0898e+00, ..., 1.3201e+01,
2.5216e+00, -5.9886e-01],
[ 4.7260e+00, 3.7397e+00, 3.4565e+00, ..., 9.5700e+00,
2.7525e+00, 1.0127e+00],
[ 2.1353e+00, 8.7713e-01, 2.8607e+00, ..., 3.6144e-01,
-3.9770e-01, 1.0760e+00]]],
[[[-5.8913e-01, -1.3195e+00, -9.8442e-01, ..., 9.8013e-01,
2.3928e-01, -5.9810e-01],
[ 7.0864e-02, -1.6204e+00, 8.5153e-01, ..., 2.1180e-01,
4.4067e-01, -1.0944e+00],
[ 8.6576e-01, 2.3578e-01, 2.2191e+00, ..., 2.2494e-01,
1.4787e-01, -1.9526e+00],
...,
[-1.3259e+00, -1.4345e+00, -9.7402e-01, ..., -2.9619e+00,
-3.0307e+00, 1.1078e+00],
[-2.0153e+00, -1.0193e+00, -6.5275e-01, ..., -1.2195e+00,
-1.4501e+00, 1.0383e+00],
[-1.9238e+00, -4.3192e+00, -2.9971e+00, ..., -2.1641e+00,
-2.2068e+00, -1.0042e+00]],
[[-9.3294e-01, -1.9919e+00, -8.8169e-01, ..., -1.8277e-01,
-3.6691e+00, -2.7044e+00],
[-2.0015e+00, -3.2999e+00, -3.1665e-01, ..., -3.9518e+00,
-5.4209e+00, -4.8158e+00],
[-1.8489e+00, -3.5070e+00, -5.1462e+00, ..., -3.0713e+00,
-4.2013e+00, -4.3601e+00],
...,
[-7.1859e+00, -7.4800e+00, -5.6043e+00, ..., -5.1690e+00,
-6.5120e+00, -5.2940e-01],
[-8.6893e+00, -9.0911e+00, -6.8135e+00, ..., -5.3474e+00,
-7.6758e+00, -3.0445e+00],
[-2.1393e+00, -4.3955e+00, -3.5959e+00, ..., -2.3256e+00,
-3.5147e+00, -5.5599e-01]],
[[ 9.1752e-01, 1.5075e+00, 1.5515e+00, ..., 1.3149e+00,
1.8577e+00, 9.7199e-01],
[ 1.3527e+00, 1.4818e+00, -8.0964e-01, ..., 3.0075e-01,
1.9694e+00, 9.7961e-01],
[ 5.0229e-01, 3.1479e+00, 1.3822e+00, ..., 1.1487e+00,
2.4671e+00, 1.9548e+00],
...,
[ 4.6621e-01, 5.7864e-01, 7.5476e-01, ..., -4.3101e-01,
-4.1026e-01, -1.9698e+00],
[ 3.2696e-01, -1.7609e-01, -2.8008e-01, ..., -1.9666e+00,
-1.5817e+00, -1.4597e+00],
[ 5.3574e-01, 6.1473e-01, -8.6477e-02, ..., -1.5568e+00,
-1.3362e+00, -2.0721e+00]],
...,
[[-1.2285e+00, -1.8041e+00, -1.1583e+00, ..., -1.0791e+00,
-3.0390e+00, -1.0535e+00],
[-2.3822e+00, -2.5707e+00, -1.6366e-01, ..., -3.8311e+00,
-5.0745e+00, -3.5354e+00],
[-9.6472e-01, -1.7730e+00, -2.2057e+00, ..., -1.4851e+00,
-3.5650e+00, -2.9843e+00],
...,
[-1.9874e+00, -1.3318e+00, -5.9636e-01, ..., -1.0156e+00,
-2.3962e+00, -8.5674e-01],
[-2.7438e+00, -3.0966e+00, -9.4631e-01, ..., 6.8751e-02,
-2.3235e+00, -4.8837e-01],
[-2.3730e+00, -2.8474e+00, -1.5953e+00, ..., -7.4472e-01,
-1.7344e+00, 6.8266e-01]],
[[-2.2823e+00, -2.3015e+00, -3.9341e+00, ..., -2.1186e+00,
-3.0035e+00, -7.6013e-01],
[-4.9666e+00, -3.3568e+00, -2.1224e+00, ..., -3.3844e+00,
-5.4500e+00, -1.7508e+00],
[-4.0799e+00, -4.1351e+00, -3.8055e+00, ..., -1.9671e+00,
-4.4659e+00, -1.8724e+00],
...,
[-3.4819e+00, -2.2801e+00, -2.5929e+00, ..., -5.2322e-01,
-6.7866e-02, 3.6454e-01],
[-3.3188e+00, 6.7825e-01, -6.0254e-01, ..., 1.0279e+00,
7.3406e-01, 1.7088e+00],
[-3.9852e+00, -1.8133e+00, -9.2699e-01, ..., -2.7127e-02,
-1.9224e-01, 7.3235e-01]],
[[ 1.1383e+00, -5.3133e-01, -1.1385e+00, ..., 3.8776e-01,
1.4812e+00, 6.7708e-01],
[ 1.6652e+00, -4.4214e-01, -9.9978e-01, ..., -1.1818e-01,
1.3865e+00, 3.0932e+00],
[ 1.4102e+00, -1.9483e-01, 2.2915e+00, ..., 6.6913e-01,
1.6766e+00, 1.4392e+00],
...,
[ 3.2276e+00, 2.3056e+00, 4.4045e-01, ..., 9.6474e-01,
1.7051e+00, 1.3909e+00],
[ 4.6585e+00, 4.9482e+00, 2.9713e+00, ..., 2.0895e+00,
1.6649e+00, 2.1202e+00],
[ 2.4052e+00, 1.8288e+00, 1.0086e+00, ..., -2.5267e-01,
-2.4057e-02, 1.1162e+00]]],
[[[-1.7541e-01, 3.7432e-01, -1.0560e+00, ..., -2.1657e-01,
-2.6869e+00, -1.3537e+00],
[ 2.0568e-01, -3.6226e+00, -2.5236e+00, ..., -1.4028e+00,
-1.1847e+00, 2.8175e-01],
[-4.2950e+00, -6.7819e+00, 3.5399e-01, ..., -3.9004e+00,
-5.1534e+00, -3.3280e+00],
...,
[-2.3647e-01, -2.4053e-01, 1.3488e+00, ..., 1.1829e+00,
5.2410e-01, 6.6219e-01],
[-1.5627e+00, -2.6534e-01, 9.5292e-01, ..., -1.9888e+00,
-4.6591e+00, -1.0684e+00],
[-1.0436e+00, -2.2604e+00, -1.0605e+00, ..., -4.0682e+00,
-4.0848e+00, -1.9161e+00]],
[[-2.2681e+00, -2.7882e+00, -4.1147e+00, ..., -7.5714e+00,
-1.0852e+01, -3.5772e+00],
[-6.5968e+00, -9.8206e+00, -4.1651e+00, ..., -4.5861e+00,
-5.1090e+00, -2.7332e+00],
[-7.9065e+00, -5.7356e+00, 1.6198e+00, ..., -8.7966e+00,
-9.6715e+00, -3.5388e+00],
...,
[-4.7224e+00, -5.4140e+00, -5.4903e+00, ..., -3.0976e+00,
-5.7365e+00, -3.3790e+00],
[-7.9062e+00, -8.2887e+00, -7.5320e+00, ..., -1.1818e+01,
-1.2684e+01, -5.2949e+00],
[-1.3608e+00, -3.0171e+00, -2.1982e+00, ..., -3.9343e+00,
-4.1292e+00, -8.0917e-01]],
[[-1.0558e-01, 1.5768e+00, 8.8711e-01, ..., 2.5188e+00,
2.7315e+00, 5.1051e-01],
[-8.3085e-01, 1.0543e+00, 3.4054e+00, ..., 2.6957e+00,
2.9320e+00, -7.6566e-02],
[ 1.5516e+00, 2.5874e+00, -1.7387e+00, ..., 4.4068e+00,
6.6311e+00, 4.4896e+00],
...,
[-7.6002e-01, 7.5448e-01, 1.7717e+00, ..., -3.4252e-02,
7.7945e-01, 7.2780e-01],
[-1.3047e+00, 5.6108e-02, 1.1838e+00, ..., -1.0570e+00,
-1.1089e+00, -4.8963e-01],
[-5.2559e-01, -5.0211e-01, 2.7385e-04, ..., 8.3073e-01,
5.0354e-01, -1.2554e+00]],
...,
[[-1.6841e+00, -2.3413e+00, -1.8896e+00, ..., -3.3987e+00,
-5.8807e+00, -1.5686e+00],
[-4.6142e+00, -6.5098e+00, -3.1960e+00, ..., -2.4843e+00,
-4.3243e+00, -2.1882e+00],
[-5.4991e+00, -3.2338e+00, 7.7541e-01, ..., -3.4385e+00,
-5.1466e+00, -3.1234e+00],
...,
[-9.8345e-01, -2.3235e+00, -5.9402e-01, ..., -1.3073e+00,
-2.7552e+00, -2.1371e+00],
[-2.2362e+00, -3.9194e+00, -8.0692e-01, ..., -4.3456e+00,
-5.6386e+00, -1.2747e+00],
[-1.1223e+00, -1.1275e+00, 4.3889e-01, ..., -2.3664e-01,
-1.3440e+00, 6.2402e-01]],
[[-1.2533e+00, -1.0535e+00, -1.5830e+00, ..., -1.7830e+00,
-1.0384e+00, 1.5586e-01],
[-5.2454e+00, -4.1420e+00, -3.9630e+00, ..., -9.5855e-01,
-1.6877e+00, 6.4429e-01],
[-4.9948e+00, -1.3997e+00, -3.2837e+00, ..., -6.1918e+00,
-9.5485e+00, -2.6573e+00],
...,
[-2.8621e+00, -2.8250e+00, -3.1053e+00, ..., -6.9643e-01,
-2.5558e+00, -5.3750e-01],
[-2.2232e+00, -2.6597e-01, -1.4743e+00, ..., -2.9065e-01,
-1.9090e+00, 1.2560e+00],
[-2.1148e+00, -1.2451e+00, -3.6581e-01, ..., 5.3532e-01,
-7.8594e-01, 2.6297e-01]],
[[ 1.3367e+00, -4.0570e-01, -5.5486e-01, ..., 3.1133e+00,
2.1699e+00, 3.3137e-01],
[ 3.9802e+00, 2.6313e+00, 3.3519e+00, ..., 1.5485e+00,
1.6588e+00, 3.4210e+00],
[ 2.3256e+00, -1.0043e+00, 1.1273e+00, ..., 2.0315e+00,
3.7532e+00, 2.9038e+00],
...,
[ 3.1867e+00, 2.7556e+00, 2.7607e+00, ..., 1.3029e+00,
9.9281e-01, 2.0647e+00],
[ 3.8401e+00, 4.2148e+00, 3.0514e+00, ..., 4.1451e+00,
8.6633e-01, 2.5942e+00],
[ 2.3106e+00, 2.2272e+00, 1.6621e+00, ..., -3.0763e-01,
-6.7901e-01, 1.7706e+00]]],
[[[-5.9267e+00, -7.6160e+00, -7.8415e+00, ..., -1.7058e+00,
-1.5919e+00, 9.0336e-02],
[ 3.7385e+00, 4.5784e+00, 3.7222e+00, ..., -6.9325e-01,
1.2293e+00, 1.7276e+00],
[-4.2636e+00, -8.5795e+00, -8.4796e+00, ..., -1.8222e+00,
-1.2297e+00, 8.1311e-01],
...,
[-1.8066e+00, -2.1227e+00, -3.5267e+00, ..., -5.9161e+00,
-6.3144e+00, -2.6658e+00],
[-1.0863e+00, -2.6445e+00, -3.2284e+00, ..., -9.4135e+00,
-9.7181e+00, -8.5944e-01],
[-1.8068e+00, -5.1304e+00, -2.6453e+00, ..., -2.7429e+00,
-3.0229e+00, 1.4881e+00]],
[[-1.0308e+01, -1.6094e+01, -1.2899e+01, ..., -3.1466e+00,
-4.9793e+00, -4.1129e-01],
[ 3.6954e+00, 3.2016e+00, 4.7377e-01, ..., 4.8252e-01,
-4.9926e-01, -1.1485e-01],
[-2.1346e+00, -5.0761e+00, -1.7992e+00, ..., -3.3720e+00,
-4.1585e+00, -2.3042e-01],
...,
[-5.4008e+00, -5.1632e+00, -4.3153e+00, ..., -6.6044e+00,
-4.5682e+00, -2.3289e+00],
[-8.5324e+00, -9.3412e+00, -4.4284e+00, ..., -1.6894e-02,
-2.4446e+00, -2.2277e+00],
[-1.7120e+00, -4.0521e+00, -1.7442e+00, ..., 6.2412e-01,
-2.3427e+00, -8.0874e-01]],
[[ 1.1022e+00, 5.8246e+00, 7.5101e+00, ..., 2.3130e+00,
3.2914e+00, 1.6840e+00],
[-5.2371e+00, -4.8188e+00, -9.8446e-01, ..., -2.9925e+00,
-3.4046e-01, 1.0217e+00],
[-2.9096e+00, -1.0334e+01, -1.5293e+01, ..., 6.5494e-01,
-1.8793e-02, -1.7428e-02],
...,
[ 7.7414e-01, -4.6473e-01, 4.8545e-01, ..., -1.7393e+01,
-7.0382e+00, 1.8137e-01],
[-1.6034e+00, -1.7281e+00, 3.3914e-01, ..., -5.5169e+00,
-1.4368e+00, 1.1172e+00],
[-2.3596e-01, -1.0004e-01, -3.6016e-02, ..., 5.9386e-01,
7.4458e-01, 6.9623e-01]],
...,
[[-3.5207e+00, -3.9851e+00, -2.3864e+00, ..., -1.6622e+00,
-1.8079e+00, -1.3278e+00],
[ 4.4985e+00, 6.0494e+00, 8.5304e+00, ..., 1.4638e+00,
-5.9676e-01, -1.5983e+00],
[-2.1619e+00, -4.0366e+00, -7.0444e-01, ..., -3.0195e+00,
-3.1638e+00, -1.5586e+00],
...,
[-6.5655e-01, -9.7278e-01, -7.5123e-01, ..., -4.4437e+00,
-2.3313e+00, -1.4677e+00],
[-2.1938e+00, -4.6376e+00, -6.2961e-01, ..., 9.3469e-01,
-1.1127e+00, -2.1192e+00],
[-1.9920e+00, -2.4359e+00, -1.4612e-01, ..., 2.8233e+00,
1.0996e+00, 5.4716e-01]],
[[-5.7047e+00, -8.8644e+00, -4.6732e+00, ..., -1.4314e+00,
-2.1026e+00, -2.7919e-01],
[ 4.7640e+00, 4.0494e+00, 2.8267e+00, ..., 3.4678e+00,
2.7358e-01, -5.8891e-01],
[-2.1185e+00, -2.7089e+00, -1.6653e-01, ..., 2.5673e+00,
1.0341e+00, 4.4749e-01],
...,
[-3.3940e+00, -6.4144e-01, -1.0808e+00, ..., -3.2608e-03,
-1.0539e+00, 1.8087e-02],
[-2.6611e+00, 1.2987e+00, 1.5482e+00, ..., -4.6581e+00,
-3.2345e+00, -7.1911e-01],
[-3.1154e+00, -3.8566e-01, 8.4148e-01, ..., -3.2281e+00,
-3.9799e+00, -4.5218e-01]],
[[ 4.1633e-01, 2.5310e+00, 2.4875e+00, ..., 2.1256e-01,
1.3934e+00, 1.0393e+00],
[ 1.8325e+00, 1.8187e+00, 3.3184e+00, ..., 4.3480e-01,
1.4357e+00, 2.8625e+00],
[-6.7349e+00, -1.3732e+01, -1.2267e+01, ..., 1.4383e+00,
1.9890e+00, 2.1536e+00],
...,
[ 2.2470e+00, 1.6117e+00, -7.6293e-01, ..., -1.1695e+01,
-1.0090e+01, -2.5302e+00],
[ 4.4369e+00, 4.1981e+00, 1.7942e+00, ..., -1.0183e+01,
-1.2744e+01, -5.1136e+00],
[ 2.3219e+00, 1.0694e+00, 1.0744e+00, ..., -4.2949e+00,
-7.7758e+00, -1.3254e+00]]],
[[[ 2.9161e-02, 1.1847e+00, 3.5868e-01, ..., -8.2149e-01,
-2.0410e+00, -3.7584e-01],
[ 3.4224e-01, 2.7998e-01, 1.2601e+00, ..., -7.6181e-01,
-7.8819e-01, 2.2438e-01],
[-1.0674e+00, -2.0556e+00, 5.9999e-01, ..., -1.5399e+00,
-2.9572e+00, -1.0649e+00],
...,
[-4.9245e-01, -1.5627e+00, -1.2225e+00, ..., -4.9482e+00,
-2.5245e+00, -3.9670e-02],
[-1.8602e+00, -1.8480e+00, -1.0967e+00, ..., -3.8097e+00,
-3.1240e+00, -4.5053e-01],
[-1.0552e+00, -3.2930e+00, -2.6900e+00, ..., -4.1491e+00,
-3.4621e+00, -1.2832e+00]],
[[-1.6251e+00, -2.2760e+00, -3.0429e+00, ..., -4.5358e+00,
-6.1744e+00, -1.4042e+00],
[-3.6487e+00, -4.2288e+00, -3.5351e+00, ..., -8.0501e+00,
-6.2574e+00, -2.2927e+00],
[-3.3073e+00, -3.6447e+00, -2.1565e+00, ..., -8.5577e+00,
-7.9615e+00, -1.9511e+00],
...,
[-5.1504e+00, -5.5213e+00, -3.9074e+00, ..., -8.7251e+00,
-7.6489e+00, -2.8051e+00],
[-7.2720e+00, -7.0996e+00, -5.2164e+00, ..., -9.2739e+00,
-8.5165e+00, -3.4302e+00],
[-1.6127e+00, -3.0140e+00, -2.8180e+00, ..., -4.7327e+00,
-3.7049e+00, -3.9480e-01]],
[[ 1.9373e-01, 1.2955e+00, 1.5282e+00, ..., 6.4593e-01,
2.4166e+00, 1.1182e+00],
[ 7.4543e-02, 2.6569e-01, 2.6673e-01, ..., -1.6890e-01,
1.8538e+00, 1.7858e-01],
[ 1.4606e-01, 2.3283e+00, 2.0461e+00, ..., -7.2669e-01,
1.0065e+00, 6.0810e-01],
...,
[-1.2493e-02, 7.7512e-01, 2.7852e-01, ..., -4.4833e-01,
3.7694e-01, -4.1612e-01],
[ 2.8761e-01, 1.0245e-01, -7.4021e-01, ..., -4.4964e-01,
7.2074e-01, 7.0522e-02],
[ 5.3416e-01, 1.0703e+00, 5.5836e-01, ..., -5.8141e-01,
-5.7187e-01, -2.0115e+00]],
...,
[[-1.0152e+00, -1.7126e+00, -2.0871e+00, ..., -3.7019e+00,
-4.4289e+00, -1.0372e+00],
[-2.9235e+00, -2.5712e+00, -1.0567e+00, ..., -4.9825e+00,
-4.8064e+00, -1.5257e+00],
[-2.4671e+00, -7.8336e-01, 3.1671e-01, ..., -3.3100e+00,
-6.1689e+00, -1.0621e+00],
...,
[-5.1800e-01, -4.0373e-01, -1.0175e+00, ..., -4.4900e+00,
-6.4652e+00, -3.0178e+00],
[-2.7082e+00, -2.3231e+00, -2.5478e-01, ..., -2.6628e+00,
-4.2514e+00, -1.4672e-01],
[-1.2952e+00, -1.6669e+00, -9.2301e-01, ..., -1.7729e+00,
-2.1518e+00, 9.9043e-01]],
[[-1.7255e+00, -1.9924e+00, -4.0319e+00, ..., -2.6964e+00,
-1.6991e+00, -2.2412e-01],
[-3.9668e+00, -3.4679e+00, -3.8780e+00, ..., -2.9567e+00,
-3.0874e+00, -3.1645e-01],
[-4.2314e+00, -3.1735e+00, -2.6005e+00, ..., -2.7534e+00,
-3.7358e+00, -1.0951e+00],
...,
[-3.7991e+00, -3.9653e+00, -3.3380e+00, ..., -5.4246e+00,
-3.5748e+00, -3.4536e-01],
[-3.8958e+00, -1.2775e+00, -1.0994e+00, ..., -1.2176e+00,
-1.8471e+00, 4.2966e-01],
[-3.1534e+00, -2.7465e+00, -7.6345e-01, ..., 2.3079e-01,
-6.5165e-01, 6.9375e-01]],
[[ 1.3403e+00, -2.9355e-02, 1.1785e+00, ..., -9.8612e-01,
1.1823e-01, -2.8094e-01],
[ 2.6380e+00, 1.0490e+00, 1.1393e+00, ..., 1.5265e+00,
1.9914e+00, 3.0117e+00],
[ 6.5644e-01, -8.8048e-01, 4.6180e-01, ..., 1.3398e+00,
7.2054e-01, 8.7618e-01],
...,
[ 1.4262e+00, 4.6717e-01, 1.0772e+00, ..., 7.0705e-01,
1.3753e+00, 1.2725e+00],
[ 2.8816e+00, 3.8786e+00, 1.8242e+00, ..., 1.0273e-01,
1.7283e-01, 1.8394e+00],
[ 1.8262e+00, 1.2977e+00, 4.4630e-01, ..., -5.1162e-01,
-3.6135e-01, 8.9081e-01]]]], device='cuda:0'),
tensor([[[[ 9.7076e-02, 6.5047e-01, -1.8099e+00, ..., -4.5591e-01,
-3.1508e+00, -1.7407e+00],
[ 4.9303e-01, 7.0483e-02, 5.1986e-01, ..., 1.4111e+00,
-9.8340e-01, -7.9211e-01],
[ 2.6289e-01, 1.2714e+00, 2.3223e+00, ..., -6.0763e-01,
-1.1527e+00, -1.2760e+00],
...,
[-5.0755e-01, -2.2702e+00, -1.1850e+00, ..., -1.5187e+00,
-4.4611e-01, -6.7799e-01],
[-9.3658e-01, -1.9732e+00, -2.7157e+00, ..., -1.7486e+00,
-1.1638e+00, -1.6778e+00],
[-2.7715e+00, -5.8340e+00, -4.5464e+00, ..., -5.1411e+00,
-4.2644e+00, -3.7092e+00]],
[[-1.3927e+00, -2.5490e+00, -4.5860e+00, ..., -2.5038e+00,
-8.5728e+00, -2.3160e+00],
[-4.5077e+00, -5.4011e+00, -4.7057e+00, ..., -4.1861e+00,
-7.6088e+00, -2.9451e+00],
[-3.6996e+00, -6.8859e+00, -7.9481e+00, ..., -1.1747e+01,
-1.0870e+01, -5.1705e+00],
...,
[-5.9915e+00, -5.8841e+00, -5.2276e+00, ..., -4.9804e+00,
-4.9924e+00, -3.5193e+00],
[-7.8794e+00, -8.5606e+00, -6.0522e+00, ..., -8.2991e+00,
-8.1382e+00, -6.3595e+00],
[-2.6474e+00, -4.5652e+00, -3.1635e+00, ..., -4.6540e+00,
-4.0238e+00, -2.4560e+00]],
[[ 5.0344e-01, 2.5123e+00, 1.9368e+00, ..., 3.0845e+00,
3.5894e+00, 9.2463e-01],
[-5.4200e-01, 3.3814e-01, 1.1297e+00, ..., 1.4035e+00,
3.2316e+00, 2.8519e-01],
[ 5.2287e-01, 1.8349e+00, 6.4659e-02, ..., -5.5640e-01,
-2.8342e-01, -5.8554e-01],
...,
[ 4.9815e-01, 7.8941e-01, 6.9583e-01, ..., -4.3069e-01,
-2.1989e-01, -5.6394e-01],
[ 8.8773e-01, 6.4156e-01, 6.3305e-01, ..., 2.3047e-02,
-4.4463e-01, -2.4669e-01],
[ 1.4048e+00, 2.0294e+00, 1.0450e+00, ..., 1.8453e+00,
1.4489e+00, 4.9161e-02]],
...,
[[-7.9023e-01, -2.4531e+00, -3.6727e+00, ..., -4.6720e+00,
-5.4511e+00, -1.8442e+00],
[-2.3604e+00, -3.4463e+00, -2.1446e+00, ..., -4.7363e+00,
-4.3747e+00, -2.4550e+00],
[-1.7864e+00, -2.4659e+00, -1.5846e+00, ..., -6.8878e+00,
-6.1954e+00, -2.9357e+00],
...,
[-2.3955e+00, -2.0579e+00, -3.3825e-01, ..., -2.2080e+00,
-3.1219e+00, -1.1905e+00],
[-3.5965e+00, -3.8935e+00, -1.9379e+00, ..., -2.4120e+00,
-4.5080e+00, -2.2022e+00],
[-3.5841e+00, -3.8738e+00, -2.5351e+00, ..., -3.2000e+00,
-3.8951e+00, -1.5304e+00]],
[[-1.6275e+00, -2.1539e+00, -2.5202e+00, ..., -3.8369e+00,
-9.8275e-01, 7.6515e-01],
[-3.6464e+00, -2.0649e+00, -3.3081e+00, ..., -3.8928e+00,
-2.1697e+00, 1.5565e-01],
[-2.7646e+00, -1.1164e+00, -2.2976e+00, ..., -2.4150e+00,
-1.2270e+00, 3.9279e-01],
...,
[-2.9962e+00, -2.2187e+00, -2.3503e+00, ..., -1.1502e+00,
-2.0653e+00, -3.0171e-01],
[-3.7355e+00, -8.9306e-01, -3.9276e-01, ..., -2.3424e-01,
-2.7847e+00, -7.7532e-01],
[-4.5241e+00, -3.1134e+00, -1.3373e+00, ..., -1.9749e+00,
-3.2906e+00, -1.0890e+00]],
[[ 1.3848e+00, 9.5497e-01, 9.2910e-01, ..., 1.9339e+00,
1.7667e+00, -4.5278e-02],
[ 2.8437e+00, 2.4497e+00, 2.1793e+00, ..., 2.4682e+00,
3.3019e+00, 4.0968e+00],
[ 2.1259e+00, 2.2334e+00, 2.8104e+00, ..., 3.4626e+00,
2.7592e+00, 1.6942e+00],
...,
[ 2.5178e+00, 7.7166e-01, 7.3129e-01, ..., 1.4405e+00,
2.0393e+00, 1.7746e+00],
[ 4.2673e+00, 4.1834e+00, 1.4849e+00, ..., 3.1265e+00,
2.1085e+00, 2.1079e+00],
[ 1.9914e+00, 2.4467e-01, -8.1386e-01, ..., -6.7781e-01,
-4.9863e-01, -2.4367e-01]]],
[[[ 2.8877e-01, 1.7547e+00, 4.3638e-01, ..., -3.4496e+00,
7.4514e-01, 2.5186e-01],
[ 1.6396e+00, 5.9494e-02, -4.7161e+00, ..., -2.9366e+00,
-1.8363e+00, -2.1124e+00],
[ 1.4959e+00, -5.5835e+00, -1.0183e+00, ..., -2.8200e+00,
-6.3599e+00, -4.7745e+00],
...,
[-2.5532e-01, 1.2904e+00, -1.5521e+00, ..., -7.2642e-01,
3.0572e-01, -9.4516e-01],
[-1.9128e+00, 1.7312e+00, 9.6110e-01, ..., 2.0997e-03,
-1.9122e-01, -1.8141e+00],
[-1.6793e+00, -2.3211e+00, -9.1761e-01, ..., -3.1490e+00,
-3.0937e+00, -3.0224e+00]],
[[-1.4248e+00, -3.4650e+00, -4.2999e+00, ..., -1.7098e+00,
1.7053e-02, 3.7200e-01],
[-2.8411e+00, -1.2326e+01, -8.4693e+00, ..., 6.4008e-01,
-2.0416e+00, -1.2627e+00],
[-4.3588e+00, -2.0418e+01, -1.7453e+00, ..., 3.1682e+00,
-3.5972e-01, -2.3475e+00],
...,
[-4.1767e+00, -3.8426e+00, -2.0899e+00, ..., -4.1368e+00,
-4.0167e+00, -3.4147e+00],
[-7.5972e+00, -7.9781e+00, -4.3315e+00, ..., -5.5558e+00,
-6.3325e+00, -5.7222e+00],
[-1.3360e+00, -2.8397e+00, -1.7195e+00, ..., -3.4020e+00,
-3.1426e+00, -1.7688e+00]],
[[ 1.5125e-01, 1.2083e+00, 2.9529e-01, ..., 3.7905e-01,
-1.9245e-01, 3.0265e-01],
[-1.4231e+00, -3.7846e-01, 2.7320e+00, ..., 2.3722e-01,
-5.1893e-01, -1.2751e+00],
[-2.4021e+00, 2.3443e+00, 9.8986e-01, ..., -4.3861e+00,
-2.0275e+00, -6.5877e-01],
...,
[-9.9649e-01, -2.2597e+00, 1.9520e+00, ..., 2.0237e+00,
1.5178e+00, 9.2289e-01],
[-1.6997e+00, -1.5887e+00, -9.4326e-01, ..., 1.6755e+00,
1.3306e+00, 6.6244e-01],
[-7.1051e-01, -5.2339e-01, -6.4988e-01, ..., 1.9323e+00,
1.6867e+00, 5.4333e-01]],
...,
[[-6.7273e-01, -1.4477e+00, -3.0811e+00, ..., 1.3683e+00,
6.8933e-01, 6.9162e-01],
[-3.2395e+00, -5.5204e+00, -6.0880e+00, ..., 8.0419e-01,
-2.6100e+00, -1.8675e-01],
[-2.9788e+00, -4.7703e+00, 1.8614e+00, ..., 1.4753e+00,
-8.5297e-01, -1.1828e+00],
...,
[-9.0704e-01, -7.2804e-01, -2.3960e+00, ..., -3.0038e+00,
-3.0372e+00, -1.4923e+00],
[-2.2501e+00, -3.1943e+00, 3.4062e-03, ..., -2.4402e+00,
-3.7312e+00, -1.9778e+00],
[-1.3879e+00, -1.7558e+00, 7.4792e-02, ..., -2.8035e+00,
-2.6240e+00, -6.8881e-01]],
[[-2.6381e+00, -1.0618e+00, -1.4114e+00, ..., -1.9936e+00,
-1.2617e+00, -2.8138e-01],
[-4.1062e+00, -4.4168e+00, -3.4935e+00, ..., -3.7536e+00,
-4.6158e+00, -2.5497e+00],
[-4.8863e+00, -2.1982e+00, -2.8390e+00, ..., -1.7022e+00,
-7.0330e+00, -3.6802e+00],
...,
[-1.4640e+00, -3.5140e+00, -8.1268e+00, ..., -7.4071e-01,
-3.5070e+00, -2.2279e+00],
[-8.6478e-01, 3.3299e-01, -3.7287e+00, ..., -5.7801e-01,
-3.6741e+00, -1.9607e+00],
[-1.4427e+00, -5.4035e-01, -4.0275e-01, ..., -1.1424e+00,
-2.9097e+00, -1.3331e+00]],
[[ 5.3465e-01, 1.0479e+00, 1.4603e+00, ..., -1.8446e-01,
3.9665e-01, -1.0466e-01],
[ 2.5818e+00, 4.9544e+00, 1.3056e+00, ..., 8.8883e-01,
-2.9739e+00, -2.2564e+00],
[ 2.5906e+00, 5.7199e+00, -1.0978e+00, ..., -4.4412e+00,
-7.2688e+00, -3.8914e+00],
...,
[ 2.6322e+00, 1.7844e+00, 1.2574e+00, ..., 1.8324e+00,
7.4180e-01, 6.8103e-01],
[ 3.0956e+00, 3.5473e+00, 2.9738e+00, ..., 2.3109e+00,
1.3156e+00, 1.3995e+00],
[ 1.7507e+00, 2.1083e+00, 1.0354e+00, ..., 8.9417e-02,
-3.0715e-01, -1.3252e-01]]],
[[[-1.5000e-01, 1.2546e+00, 1.8769e+00, ..., 4.4964e-02,
-1.5844e+00, -2.0916e+00],
[ 2.3306e-01, 5.5030e-01, 2.9666e+00, ..., -2.9907e-01,
2.2817e+00, 9.2431e-01],
[-6.3515e-02, 1.2626e+00, 9.2550e-01, ..., -1.5115e+00,
-7.9867e-01, -2.3916e+00],
...,
[ 1.7872e+00, 3.6578e+00, 2.6460e-01, ..., -1.0847e+00,
5.0222e-02, -3.4344e-01],
[-3.6031e-01, 5.7776e-01, -1.2386e+00, ..., 1.6536e-01,
3.2987e-01, -1.3615e+00],
[-1.4486e+00, -2.8919e+00, -3.1081e+00, ..., -3.2445e+00,
-2.8814e+00, -2.7825e+00]],
[[-9.1309e-01, -1.2261e+00, -2.8677e+00, ..., -1.7525e+00,
-3.8455e+00, -2.8377e+00],
[-4.1237e+00, -4.7868e+00, -3.6381e+00, ..., -2.0776e+00,
1.3253e+00, -7.0276e-01],
[-3.3527e+00, -5.8861e+00, -9.2299e+00, ..., -7.2500e+00,
-5.1411e+00, -4.3213e+00],
...,
[-2.5078e+00, -4.7290e+00, -7.9492e+00, ..., -3.6661e+00,
-3.5370e+00, -2.9660e+00],
[-7.7717e+00, -7.8346e+00, -7.1647e+00, ..., -5.0965e+00,
-5.2537e+00, -5.1786e+00],
[-1.9365e+00, -4.0716e+00, -2.7582e+00, ..., -3.0746e+00,
-2.8256e+00, -1.6437e+00]],
[[-2.3299e-01, 2.0592e+00, 1.8455e+00, ..., 2.0148e+00,
4.3812e+00, 2.1217e+00],
[-2.6899e-01, 2.1246e-01, -1.7544e-01, ..., -1.1446e+01,
1.9762e-01, 2.9900e+00],
[-6.2646e-01, 9.5095e-01, -1.9140e+00, ..., -9.4144e+00,
-3.6296e+00, 1.6411e+00],
...,
[-1.0159e+00, -1.7829e+00, 2.8491e+00, ..., 1.1417e+00,
1.0515e+00, 9.8287e-01],
[-3.9881e+00, -2.9755e-01, 1.9721e+00, ..., 1.4918e+00,
1.0463e+00, 6.0918e-01],
[-1.1079e+00, 1.8439e-01, 7.1734e-01, ..., 1.6792e+00,
1.5766e+00, 5.9937e-01]],
...,
[[-1.0231e+00, -1.1817e+00, -1.3886e+00, ..., -6.5018e-01,
-4.4854e+00, -3.0752e+00],
[-2.9354e+00, -3.2301e+00, -1.8883e+00, ..., 5.3267e-01,
-9.1028e-01, -1.8098e+00],
[-1.8121e+00, -1.9766e+00, -2.1440e+00, ..., -3.0077e-01,
-5.5251e+00, -4.9387e+00],
...,
[ 4.9424e-01, -3.6072e+00, -2.6249e+00, ..., -2.3915e+00,
-2.7474e+00, -1.4369e+00],
[-8.7566e-01, -4.5619e+00, -2.5886e+00, ..., -1.7175e+00,
-2.9747e+00, -1.7402e+00],
[-1.4951e+00, -2.3512e+00, -1.6557e+00, ..., -1.9864e+00,
-2.2767e+00, -6.3660e-01]],
[[-1.4980e+00, -1.3514e+00, -2.7352e+00, ..., 1.3367e-01,
-2.4205e+00, 1.4264e+00],
[-3.7860e+00, -2.4124e+00, -2.3833e+00, ..., 3.6264e+00,
-6.4169e-01, 1.6608e+00],
[-2.3371e+00, -2.3879e+00, -4.6815e+00, ..., 2.4795e+00,
-2.3672e+00, 8.7833e-01],
...,
[-3.7921e-01, -4.0750e+00, -5.7274e+00, ..., -1.2557e-01,
-2.3703e+00, -1.2876e+00],
[ 4.0514e-01, -6.5785e-01, -3.4992e+00, ..., -8.5027e-01,
-3.4152e+00, -1.9903e+00],
[-1.6195e+00, -1.8102e+00, -2.2629e+00, ..., -1.4947e+00,
-2.6921e+00, -1.3159e+00]],
[[ 8.3202e-01, 2.3183e-02, 5.1057e-01, ..., 2.1712e-01,
2.6973e+00, 1.8254e+00],
[ 3.6000e+00, 2.3927e+00, 2.0499e+00, ..., -2.2359e+00,
9.8694e-01, 3.0439e+00],
[ 1.7461e+00, 2.2600e+00, 3.2848e+00, ..., 2.7815e+00,
4.8625e+00, 3.2240e+00],
...,
[ 2.7669e+00, 5.2187e+00, 3.5807e+00, ..., 1.1464e+00,
4.4352e-04, 2.5934e-01],
[ 3.3620e+00, 4.0233e+00, 2.8968e+00, ..., 1.1944e+00,
9.2292e-01, 1.2437e+00],
[ 1.8392e+00, 2.4196e+00, 2.1508e+00, ..., -1.4352e-01,
-3.0049e-01, -1.7852e-01]]],
[[[ 1.7443e-01, 1.2740e+00, 3.5523e-01, ..., -2.9965e+00,
-2.2613e+00, -1.0449e+00],
[ 1.8969e+00, 1.1056e+00, 1.4382e+00, ..., -1.6220e+00,
-2.2431e+00, -2.3602e-01],
[-5.5264e-01, -3.9553e+00, -5.4521e+00, ..., -1.1980e+00,
-2.9273e+00, -3.6394e-02],
...,
[-8.2470e-01, -4.3043e-01, -3.9283e-01, ..., -3.0927e+00,
2.4867e+00, 2.7673e+00],
[-1.5543e+00, -8.7043e-01, -1.2824e+00, ..., -4.3095e-01,
-1.5497e+00, -7.4024e-01],
[-9.8439e-01, -3.3226e+00, -3.7729e+00, ..., -2.0544e+00,
-2.3143e+00, -9.5553e-01]],
[[-1.1287e+00, -2.4435e+00, -4.5994e+00, ..., -6.0656e+00,
-7.1419e+00, -1.3439e+00],
[-4.2123e+00, -5.9102e+00, -6.1122e+00, ..., -5.8965e+00,
-8.3744e+00, -2.3647e+00],
[-6.1517e+00, -9.6773e+00, -9.2461e+00, ..., -5.0931e+00,
-7.6395e+00, -2.3511e+00],
...,
[-5.7467e+00, -8.6018e+00, -1.0796e+01, ..., -7.8752e-03,
-8.5699e-02, 5.2423e-01],
[-7.7134e+00, -7.2269e+00, -7.0837e+00, ..., -9.0754e+00,
-1.2128e+01, -7.7421e+00],
[-1.6988e+00, -3.5031e+00, -4.4445e+00, ..., -1.7115e+00,
-1.5731e+00, -3.3874e-01]],
[[ 2.9229e-01, 1.3291e+00, 1.9577e+00, ..., 1.5215e+00,
2.4405e+00, 9.8048e-01],
[-1.9389e-01, -1.9012e-02, 9.5179e-01, ..., 1.5205e+00,
3.0116e+00, 1.0458e-01],
[-1.5923e-01, 3.0254e+00, 4.9788e+00, ..., 1.9590e-01,
1.9860e+00, -2.3640e-01],
...,
[ 2.6501e-01, 1.7921e-01, 1.0911e+00, ..., 7.8080e-01,
6.9287e-01, -7.6988e-01],
[-3.4140e-01, 5.2828e-01, 1.0040e+00, ..., -1.7930e+00,
-8.6166e-01, -1.8203e-01],
[ 6.8362e-01, 1.4302e+00, 1.2290e+00, ..., -2.4543e+00,
-7.1654e-01, -1.4900e+00]],
...,
[[-6.8954e-01, -1.6038e+00, -2.6397e+00, ..., -4.9692e+00,
-5.1011e+00, -7.3216e-01],
[-2.4782e+00, -3.3648e+00, -2.5043e+00, ..., -3.7329e+00,
-4.9172e+00, -1.8422e+00],
[-2.6069e+00, -2.3572e+00, -2.9662e+00, ..., -4.5425e+00,
-5.2097e+00, -1.2010e+00],
...,
[-3.3155e+00, -3.8602e+00, -6.0441e+00, ..., -1.4161e+00,
-2.2728e+00, -3.1317e+00],
[-2.4802e+00, -1.3309e+00, -1.0050e+00, ..., -2.9642e+00,
-4.5813e+00, -2.5384e+00],
[-1.5889e+00, -2.3662e+00, -2.3310e+00, ..., -6.8508e-01,
-1.3302e+00, 1.2297e+00]],
[[-1.6478e+00, -1.9979e+00, -3.6800e+00, ..., -1.4647e+00,
-5.2490e-01, -6.2319e-01],
[-3.7137e+00, -3.3203e+00, -3.3247e+00, ..., -3.3647e+00,
-1.8144e+00, -3.7179e-01],
[-4.4277e+00, -4.6596e+00, -7.6152e+00, ..., -2.7615e+00,
-1.7667e+00, -4.4477e-01],
...,
[-3.4518e+00, -3.2459e+00, -5.5258e+00, ..., -2.4695e+00,
3.8182e-01, 9.2630e-01],
[-3.8275e+00, -1.4834e+00, -3.0152e+00, ..., -2.8577e+00,
-5.6962e-01, 2.2826e+00],
[-3.2190e+00, -3.3267e+00, -2.5573e+00, ..., 1.9935e+00,
7.3767e-01, 1.6182e+00]],
[[ 1.1705e+00, 2.6979e-01, 1.4238e+00, ..., 4.3716e-01,
7.8489e-01, -1.4560e+00],
[ 2.8462e+00, 1.8769e+00, 2.3805e+00, ..., 3.6006e+00,
3.0288e+00, 2.1695e+00],
[ 1.9337e+00, 2.2836e-01, -6.4474e-01, ..., 2.8085e+00,
1.7289e+00, 1.2402e+00],
...,
[ 1.7996e+00, 4.3306e+00, 4.7552e+00, ..., -8.6923e+00,
-1.3367e+00, 4.0786e-01],
[ 2.6677e+00, 3.6162e+00, 8.7028e-01, ..., 2.9164e+00,
2.7894e+00, 3.0127e+00],
[ 2.2994e+00, 2.1574e+00, 1.3787e+00, ..., 1.2552e+00,
-1.5461e+00, 9.6703e-01]]],
[[[ 2.9107e-01, 1.2386e+00, 9.4541e-02, ..., -3.2195e+00,
-2.5579e+00, -1.0632e+00],
[ 3.8588e-01, 8.5999e-01, 9.1846e-01, ..., -2.1515e+00,
-2.2423e+00, -4.4777e-01],
[-5.2715e-01, -6.4600e-01, -7.0499e-01, ..., -3.0209e+00,
-2.8657e+00, -1.8690e-01],
...,
[-3.5154e-01, -1.9918e+00, -6.4882e+00, ..., -3.6985e+00,
-3.8585e+00, 2.1378e-01],
[-1.8061e+00, -1.6936e+00, -5.7603e+00, ..., -3.6553e+00,
-3.3910e+00, 6.0124e-01],
[-1.1887e+00, -3.3747e+00, -4.5857e+00, ..., -3.6346e+00,
-3.1462e+00, -1.4491e+00]],
[[-1.0788e+00, -1.6297e+00, -3.3802e+00, ..., -7.0474e+00,
-7.4367e+00, -1.4249e+00],
[-3.9656e+00, -4.1967e+00, -3.6087e+00, ..., -4.9840e+00,
-8.8152e+00, -2.9870e+00],
[-4.0826e+00, -4.3586e+00, -3.7165e+00, ..., -3.7786e+00,
-6.3232e+00, -2.6041e+00],
...,
[-6.4030e+00, -5.8982e+00, -8.5739e+00, ..., -7.6243e+00,
-8.3192e+00, -1.9525e+00],
[-7.9448e+00, -7.1461e+00, -8.5880e+00, ..., -8.8928e+00,
-1.0719e+01, -4.6776e+00],
[-1.9812e+00, -3.0582e+00, -3.5572e+00, ..., -3.5268e+00,
-3.9935e+00, -7.7695e-01]],
[[-4.2796e-05, 1.3128e+00, 1.6557e+00, ..., 1.7436e+00,
2.3548e+00, 9.1083e-01],
[ 6.8780e-02, 2.9332e-02, 7.1344e-01, ..., 1.7527e+00,
3.1881e+00, 9.8468e-02],
[ 6.4823e-01, 1.1786e+00, 1.3451e+00, ..., 3.5761e+00,
3.1011e+00, 1.6656e-01],
...,
[ 3.3539e-01, 1.1846e-01, 2.0296e+00, ..., 9.0149e-01,
4.6456e-01, -1.0972e+00],
[ 4.6858e-01, -5.0804e-01, 4.0253e-01, ..., -1.1069e+00,
1.1702e+00, -1.3108e+00],
[ 6.2614e-01, 6.8066e-01, 2.4208e-01, ..., 3.5819e-01,
-1.7960e-01, -1.9333e+00]],
...,
[[-6.3645e-01, -1.5774e+00, -2.2411e+00, ..., -4.9578e+00,
-5.1035e+00, -6.5847e-01],
[-2.7652e+00, -2.6517e+00, -1.7040e+00, ..., -3.4299e+00,
-4.1460e+00, -1.9722e+00],
[-2.1956e+00, -1.4050e+00, -1.2931e+00, ..., -5.0007e+00,
-4.8837e+00, -1.6829e+00],
...,
[-1.7696e+00, -2.7463e+00, -3.4990e+00, ..., -2.8088e+00,
-4.9258e+00, -1.9264e+00],
[-3.1338e+00, -2.9225e+00, -3.1630e+00, ..., -3.3404e+00,
-5.7645e+00, -1.0455e+00],
[-1.5182e+00, -1.6806e+00, -2.3684e+00, ..., -2.5074e+00,
-2.2691e+00, 3.4334e-01]],
[[-1.3293e+00, -2.3181e+00, -3.9300e+00, ..., -1.7948e+00,
-5.9665e-01, -6.2186e-01],
[-3.6749e+00, -3.4921e+00, -4.7918e+00, ..., -3.2012e+00,
-1.7747e+00, -7.3319e-02],
[-4.6111e+00, -3.7837e+00, -4.2102e+00, ..., -3.8614e+00,
-1.3160e+00, 5.7202e-01],
...,
[-4.7247e+00, -5.4261e+00, -2.7142e+00, ..., -3.9083e+00,
-2.5456e+00, -9.2842e-01],
[-4.0110e+00, -2.4058e+00, -1.3517e+00, ..., -1.2995e+00,
-4.0851e-01, 1.2400e+00],
[-3.0753e+00, -2.7976e+00, -1.4542e+00, ..., -1.6208e-01,
-7.6720e-01, 5.9004e-01]],
[[ 1.1887e+00, -1.0918e-01, 1.0868e+00, ..., 9.5783e-01,
9.7217e-01, -1.3267e+00],
[ 2.3410e+00, 1.1700e+00, 1.3588e+00, ..., 2.9690e+00,
3.0205e+00, 2.2135e+00],
[ 1.3931e+00, 7.2346e-01, 1.6130e-01, ..., 2.1690e+00,
2.4554e+00, 1.4630e+00],
...,
[ 2.9278e+00, 2.4004e+00, 3.9100e-01, ..., 2.5661e+00,
3.0550e+00, 6.3221e-01],
[ 3.0580e+00, 3.9371e+00, 7.7178e-01, ..., 3.4758e+00,
8.4475e-01, 1.6669e+00],
[ 2.1436e+00, 1.6614e+00, -3.3265e-01, ..., 3.3293e-02,
-2.7464e-01, 9.2029e-01]]]], device='cuda:0'),
tensor([[[[ 6.1035e-01, 6.1887e-01, -9.2130e-02, ..., -4.0069e-02,
4.7361e-02, -2.4174e-01],
[ 1.5597e+00, 1.0794e+00, 2.0698e+00, ..., 4.6662e-01,
1.7672e-01, -7.4016e-01],
[ 8.6744e-01, -1.5958e-01, -2.1884e+00, ..., 2.8199e-01,
-8.3311e-01, -1.1711e+00],
...,
[ 1.4778e+00, 6.6570e-01, 2.1223e+00, ..., -9.2598e-01,
-7.3812e-01, -3.2234e-01],
[-1.8439e+00, -2.0321e+00, 2.0529e-01, ..., 2.6853e-01,
-2.9214e-01, -1.2996e+00],
[-8.5681e-01, -3.6133e+00, -3.3169e+00, ..., -2.3787e+00,
-2.9837e+00, -2.9527e+00]],
[[-1.4446e+00, -4.0525e+00, -7.1997e+00, ..., -2.7688e+00,
-3.8479e+00, -2.2231e+00],
[-4.5485e+00, -6.6426e+00, -7.1256e+00, ..., -3.7304e+00,
-4.3945e+00, -3.6805e+00],
[-4.0703e+00, -7.9815e+00, -1.1084e+01, ..., -1.9853e+00,
-3.1107e+00, -2.3844e+00],
...,
[-4.7162e+00, -5.7593e+00, -7.5956e+00, ..., -3.4937e+00,
-4.2876e+00, -3.9147e+00],
[-6.5567e+00, -9.1978e+00, -9.4750e+00, ..., -4.3556e+00,
-5.4996e+00, -4.9985e+00],
[-1.6286e+00, -4.4382e+00, -4.7459e+00, ..., -3.0402e+00,
-3.1218e+00, -1.8014e+00]],
[[ 7.6115e-01, 3.1693e+00, 2.0231e+00, ..., 1.7456e+00,
1.4950e+00, 9.0443e-01],
[-6.9198e-02, 1.4157e+00, 1.0558e+00, ..., 1.3544e+00,
1.2524e+00, 8.9031e-01],
[ 4.2950e-01, 1.6486e+00, 2.4232e+00, ..., 1.2737e+00,
1.7446e+00, 1.5411e+00],
...,
[-1.1941e+00, -3.2370e-01, -4.9778e-01, ..., 1.9291e-01,
1.6797e-01, 4.7331e-01],
[ 2.1294e-01, -2.1930e-01, -4.9750e-01, ..., 4.2149e-01,
2.6364e-01, 5.0545e-01],
[ 4.1597e-01, 1.0717e+00, 4.5918e-01, ..., 1.2952e+00,
1.5839e+00, 2.5505e-01]],
...,
[[-8.7937e-01, -3.2758e+00, -2.8035e+00, ..., -1.6015e+00,
-2.3297e+00, -5.1516e-01],
[-2.3365e+00, -4.2680e+00, -1.7603e+00, ..., -1.6234e+00,
-4.1953e+00, -3.1013e+00],
[-1.5413e+00, -4.5975e+00, -5.0186e+00, ..., 1.8898e-01,
-3.3452e+00, -1.4831e+00],
...,
[-1.3024e+00, -2.6895e+00, -1.4974e+00, ..., -6.8154e-01,
-3.4502e+00, -2.4018e+00],
[-2.4229e+00, -4.3031e+00, -1.7633e+00, ..., -9.5262e-01,
-3.2238e+00, -1.6317e+00],
[-1.8142e+00, -2.9149e+00, -1.2946e+00, ..., -2.0025e+00,
-2.8146e+00, -7.0250e-01]],
[[-2.4203e+00, -3.2888e+00, -1.5142e+00, ..., 4.1550e-01,
-1.3162e+00, -5.9972e-01],
[-4.1019e+00, -2.9963e+00, -1.7031e+00, ..., -3.6774e-01,
-3.7616e+00, -1.7961e+00],
[-3.7575e+00, -3.7579e+00, -1.8616e+00, ..., -6.8840e-01,
-3.8444e+00, -2.2526e+00],
...,
[-2.3194e+00, -4.4925e+00, -2.5688e+00, ..., -1.3652e+00,
-3.6298e+00, -1.7581e+00],
[-2.6122e+00, 1.9006e-01, -1.7683e-01, ..., 1.1846e-01,
-3.3026e+00, -1.8048e+00],
[-3.4610e+00, -2.1866e+00, -1.0414e-01, ..., -6.7836e-01,
-2.6541e+00, -1.2388e+00]],
[[ 1.4867e+00, 1.3066e+00, 2.0974e+00, ..., 5.4076e-01,
9.6127e-01, 5.7209e-01],
[ 3.2211e+00, 3.4519e+00, 4.6607e+00, ..., 1.7640e+00,
1.1612e+00, 2.4852e+00],
[ 1.6564e+00, 1.9283e+00, 2.4665e+00, ..., 9.9631e-01,
-1.3113e-01, 3.8474e-01],
...,
[ 5.0078e+00, 2.6652e+00, 2.9265e+00, ..., 1.7086e+00,
2.2805e+00, 2.1065e+00],
[ 3.1922e+00, 3.1992e+00, 1.7004e+00, ..., 1.6597e+00,
6.1838e-01, 1.3452e+00],
[ 2.5766e+00, 1.9915e+00, 2.0821e+00, ..., 6.8942e-01,
-1.8381e-01, 2.3410e-01]]],
[[[-1.1260e-01, 7.8331e-01, 1.4671e-01, ..., -1.3522e+00,
-9.2930e-01, -8.4065e-01],
[ 9.6529e-01, 5.9571e-01, 2.3410e-01, ..., -3.8530e-01,
3.4071e-01, -1.0925e+00],
[ 1.4476e+00, 7.7415e-02, -2.3440e+00, ..., -6.5744e-01,
-7.2903e-01, -2.1606e+00],
...,
[-6.3988e-01, -7.8469e-01, 3.5018e-01, ..., -6.5328e-01,
-3.9521e+00, -5.5039e-01],
[-2.1899e+00, -1.0399e+00, 2.1365e-01, ..., 1.6157e-01,
-3.7497e+00, -5.6543e-01],
[-1.6460e+00, -3.0312e+00, -2.3986e+00, ..., -1.1174e+00,
-3.6412e+00, -1.9341e+00]],
[[-7.6012e-01, -1.6769e+00, -2.7763e+00, ..., -2.9807e+00,
-4.5173e+00, -2.4684e+00],
[-2.0662e+00, -4.3732e+00, -5.1938e+00, ..., -3.1532e+00,
-4.8467e+00, -4.5626e+00],
[-1.8768e+00, -4.1316e+00, -7.8173e+00, ..., -2.5742e+00,
-4.3541e+00, -3.7329e+00],
...,
[-5.8642e+00, -7.9877e+00, -5.9021e+00, ..., -5.9443e+00,
-1.0160e+01, -2.1601e+00],
[-8.9259e+00, -1.0175e+01, -8.4381e+00, ..., -8.3301e+00,
-1.2566e+01, -4.3559e+00],
[-1.9735e+00, -4.2358e+00, -3.3265e+00, ..., -2.6090e+00,
-4.8125e+00, -6.7172e-01]],
[[ 7.0638e-01, 1.3404e+00, 1.1602e+00, ..., 1.5606e+00,
1.8530e+00, 8.6925e-01],
[ 9.7627e-01, 1.6038e+00, 1.0126e+00, ..., 1.2695e+00,
2.0952e+00, 9.5853e-01],
[ 6.1246e-01, 3.0192e+00, 2.0627e+00, ..., 1.1395e+00,
2.3114e+00, 1.9690e+00],
...,
[-5.5474e-02, 2.7376e-01, 6.6941e-01, ..., 4.2095e+00,
1.4986e+00, -8.4693e-01],
[-7.7682e-01, 6.0289e-01, -9.4225e-01, ..., 2.6037e+00,
1.5481e+00, -7.1401e-01],
[ 1.2532e-01, 6.0903e-01, -3.2001e-01, ..., 1.2776e+00,
2.1254e-01, -1.7008e+00]],
...,
[[-9.7309e-01, -1.5112e+00, -2.1470e+00, ..., -2.5359e+00,
-3.2116e+00, -1.2577e+00],
[-2.1461e+00, -3.8546e+00, -3.2546e+00, ..., -2.6193e+00,
-4.1397e+00, -3.5196e+00],
[-6.3311e-01, -3.2749e+00, -4.0089e+00, ..., -2.4294e+00,
-4.3229e+00, -3.1667e+00],
...,
[-2.0548e+00, -4.1421e+00, -3.0833e+00, ..., -5.2043e+00,
-6.2567e+00, -2.7391e+00],
[-2.9337e+00, -5.2637e+00, -1.6120e+00, ..., -5.5381e+00,
-5.1025e+00, -1.6483e+00],
[-2.3158e+00, -3.0764e+00, -2.0418e+00, ..., -3.0981e+00,
-1.6573e+00, 1.7177e-01]],
[[-1.5303e+00, -2.1633e+00, -3.3881e+00, ..., -2.9008e+00,
-2.8543e+00, -5.9349e-01],
[-4.3087e+00, -5.0869e+00, -4.5263e+00, ..., -4.0461e+00,
-5.7868e+00, -2.0853e+00],
[-4.3041e+00, -7.0924e+00, -4.6801e+00, ..., -2.9068e+00,
-4.7446e+00, -2.0559e+00],
...,
[-4.2651e+00, -2.4540e+00, -4.0584e+00, ..., -4.6616e+00,
-1.1591e+00, -1.7555e+00],
[-3.3395e+00, -7.0615e-01, -1.6951e+00, ..., -3.7306e+00,
3.3226e-02, 4.7101e-02],
[-3.0582e+00, -2.3470e+00, -1.6286e+00, ..., -2.9032e+00,
-6.8202e-01, 2.4349e-01]],
[[ 9.0342e-01, -9.6714e-02, 8.5729e-01, ..., 5.3124e-01,
1.3797e+00, 6.3058e-01],
[ 1.7733e+00, 1.2326e+00, 1.9382e+00, ..., 7.3681e-01,
1.9332e+00, 3.1153e+00],
[ 1.3509e+00, 1.1156e+00, 2.2222e+00, ..., 4.6501e-01,
9.9652e-01, 7.3830e-01],
...,
[ 3.4960e+00, 3.1823e+00, 3.4169e+00, ..., 5.2922e+00,
1.9769e+00, 2.3524e-02],
[ 3.9302e+00, 4.2094e+00, 3.5374e+00, ..., 3.9873e+00,
1.4280e+00, 1.4758e+00],
[ 2.6400e+00, 2.6184e+00, 1.9335e+00, ..., 5.0293e-01,
6.5740e-01, 1.1701e+00]]],
[[[ 2.8578e-01, 1.3232e+00, 4.9108e-01, ..., -7.3541e-01,
-2.1197e-03, 6.6218e-01],
[ 3.3229e-01, 4.1039e-01, 1.4220e+00, ..., -2.1595e+00,
-1.1536e+00, 1.0652e+00],
[-5.9262e-01, 2.0547e-01, 7.0836e-01, ..., -2.2173e+00,
-1.5471e+00, 3.5542e-01],
...,
[-9.6113e-01, -2.1042e+00, -2.0400e-01, ..., -5.5659e-01,
-1.8400e-01, -7.8628e-01],
[-1.1346e+00, -1.1680e+00, -2.0073e-01, ..., -4.5748e-01,
-8.9098e-01, -2.1880e+00],
[-2.6314e+00, -5.1405e+00, -4.4154e+00, ..., -4.5893e+00,
-4.1264e+00, -3.8102e+00]],
[[-8.1392e-01, -2.1403e+00, -4.1055e+00, ..., -3.1827e+00,
-4.7431e+00, -3.5254e-01],
[-4.8368e+00, -5.7758e+00, -6.1873e+00, ..., -6.8221e+00,
-7.3099e+00, -1.6382e+00],
[-3.6309e+00, -4.5110e+00, -4.5558e+00, ..., -5.3394e+00,
-7.2691e+00, -1.8748e+00],
...,
[-5.7670e+00, -5.8926e+00, -3.7531e+00, ..., -3.9344e+00,
-4.7636e+00, -4.3547e+00],
[-7.1299e+00, -7.9532e+00, -6.2138e+00, ..., -7.7509e+00,
-8.6525e+00, -6.7879e+00],
[-2.4691e+00, -4.3534e+00, -4.1245e+00, ..., -3.9439e+00,
-4.1224e+00, -2.4827e+00]],
[[-3.2908e-01, 2.0332e+00, 1.3433e+00, ..., 7.8668e-01,
2.1525e+00, 9.6916e-01],
[-1.4711e-01, 1.0891e+00, 1.4131e+00, ..., 5.6075e-01,
1.0381e+00, -5.3468e-01],
[-5.1174e-03, 2.2083e+00, 1.2550e+00, ..., 1.1623e-01,
1.4308e+00, 1.7551e-01],
...,
[ 9.3214e-01, 1.1220e+00, 1.0455e+00, ..., -2.5312e-01,
8.0763e-01, -3.5019e-01],
[ 8.6572e-01, 7.7966e-01, 6.3388e-01, ..., 2.5436e-01,
-5.8271e-01, -4.1387e-03],
[ 1.3063e+00, 1.6335e+00, 1.2128e+00, ..., 1.4030e+00,
1.1391e+00, 9.9887e-02]],
...,
[[-8.7172e-01, -1.6606e+00, -1.8956e+00, ..., -2.5012e+00,
-2.6015e+00, -4.5193e-01],
[-3.3392e+00, -3.8706e+00, -2.7476e+00, ..., -4.8794e+00,
-4.3116e+00, -2.5621e+00],
[-1.5291e+00, -1.8029e+00, -2.1673e+00, ..., -1.3659e+00,
-4.5501e+00, -1.6926e+00],
...,
[-2.8231e+00, -2.6684e+00, -1.0449e+00, ..., -1.4870e+00,
-3.4591e+00, -2.6484e+00],
[-3.0903e+00, -3.0843e+00, -1.7584e+00, ..., -2.1487e+00,
-4.4915e+00, -2.7164e+00],
[-3.3738e+00, -3.7808e+00, -3.3260e+00, ..., -2.9748e+00,
-3.6362e+00, -1.4748e+00]],
[[-1.7308e+00, -2.1427e+00, -3.0155e+00, ..., -1.6587e+00,
-1.5143e+00, -1.3787e+00],
[-4.4370e+00, -3.2561e+00, -3.5906e+00, ..., -3.3448e+00,
-1.9140e+00, -8.8415e-01],
[-3.4210e+00, -2.8520e+00, -3.8042e+00, ..., 3.4685e-01,
-1.1460e+00, -6.8377e-01],
...,
[-4.1745e+00, -2.4649e+00, -3.0175e+00, ..., -1.0836e-01,
-2.6240e+00, -8.3689e-01],
[-3.7968e+00, -9.2294e-01, -2.3816e+00, ..., -7.6197e-01,
-2.4466e+00, -6.8408e-01],
[-4.3934e+00, -3.1030e+00, -2.8250e+00, ..., -2.3477e+00,
-3.3841e+00, -1.1991e+00]],
[[ 1.1062e+00, 5.1344e-01, 6.8336e-01, ..., -8.0648e-01,
1.0834e-01, -3.8927e-01],
[ 3.5085e+00, 3.1470e+00, 2.5404e+00, ..., 1.2603e+00,
2.1386e+00, 2.5765e+00],
[ 1.4254e+00, 9.0347e-01, 8.7698e-01, ..., 9.8205e-01,
1.5251e+00, 1.4106e+00],
...,
[ 2.8507e+00, 1.0485e+00, 1.1791e+00, ..., 1.7185e+00,
1.7105e+00, 1.7603e+00],
[ 3.6583e+00, 3.5769e+00, 2.2829e+00, ..., 2.5012e+00,
2.8366e+00, 2.4074e+00],
[ 1.8940e+00, 7.3599e-01, 1.0394e-01, ..., -1.1418e-01,
-2.0829e-01, -1.0986e-01]]],
[[[ 3.0389e-01, 1.4227e+00, 5.0650e-01, ..., -6.6097e-01,
2.8728e-01, 7.0118e-01],
[ 3.7488e-01, 2.1882e-01, 3.5012e-01, ..., -1.3195e+00,
-8.0583e-01, 1.1906e+00],
[-5.3812e-01, -1.4137e-03, 7.1636e-01, ..., -1.4653e+00,
-1.8430e+00, 8.5179e-01],
...,
[-1.0617e+00, -1.9354e+00, 3.0129e-01, ..., -1.5135e+00,
-3.2892e-01, -7.2336e-01],
[-1.1590e+00, -1.0405e+00, 1.2973e-01, ..., -4.1816e-01,
-1.1551e-01, -1.7780e+00],
[-2.6784e+00, -5.1238e+00, -4.4278e+00, ..., -4.6326e+00,
-4.1461e+00, -3.8989e+00]],
[[-8.2294e-01, -2.3032e+00, -4.2729e+00, ..., -3.1435e+00,
-4.4229e+00, -2.2700e-01],
[-4.8238e+00, -6.1299e+00, -5.0449e+00, ..., -5.2589e+00,
-6.3315e+00, -1.5202e+00],
[-3.5959e+00, -4.2347e+00, -3.7175e+00, ..., -6.2495e+00,
-6.1899e+00, -1.3626e+00],
...,
[-5.8041e+00, -5.6098e+00, -3.5624e+00, ..., -5.2022e+00,
-5.0533e+00, -3.9094e+00],
[-7.0087e+00, -7.8580e+00, -6.1765e+00, ..., -7.4494e+00,
-8.2336e+00, -6.7669e+00],
[-2.4566e+00, -4.3555e+00, -4.0808e+00, ..., -3.8779e+00,
-4.0897e+00, -2.5624e+00]],
[[-3.1782e-01, 2.1338e+00, 1.3159e+00, ..., 9.2039e-01,
2.1285e+00, 1.0143e+00],
[-1.4136e-01, 1.0085e+00, 8.3533e-01, ..., 1.0986e-01,
7.3790e-01, -3.7262e-01],
[ 4.0793e-04, 1.9257e+00, 1.2470e+00, ..., -2.0246e-01,
1.6306e+00, 1.6752e-01],
...,
[ 8.6632e-01, 1.1031e+00, 9.4183e-01, ..., 5.6930e-01,
1.0138e+00, 4.2361e-01],
[ 8.1964e-01, 6.4942e-01, 8.4638e-01, ..., 1.0830e+00,
5.9964e-01, 3.0720e-01],
[ 1.3256e+00, 1.5055e+00, 1.3167e+00, ..., 1.5400e+00,
1.1623e+00, 1.5062e-01]],
...,
[[-8.4019e-01, -1.8426e+00, -1.9833e+00, ..., -2.3081e+00,
-2.6120e+00, -4.9135e-01],
[-3.2687e+00, -4.2210e+00, -2.6015e+00, ..., -3.3749e+00,
-3.8228e+00, -2.5173e+00],
[-1.5100e+00, -2.0021e+00, -9.6269e-01, ..., -1.5888e+00,
-4.7150e+00, -1.3700e+00],
...,
[-2.7366e+00, -2.4741e+00, -4.3470e-01, ..., -2.8532e+00,
-3.7524e+00, -1.8517e+00],
[-3.0026e+00, -3.0838e+00, -1.8034e+00, ..., -2.4006e+00,
-4.2910e+00, -2.4532e+00],
[-3.4310e+00, -3.7640e+00, -3.3774e+00, ..., -2.9232e+00,
-3.6039e+00, -1.5988e+00]],
[[-1.7279e+00, -2.2384e+00, -3.0919e+00, ..., -1.3495e+00,
-1.5111e+00, -1.3082e+00],
[-4.3737e+00, -3.3858e+00, -3.1629e+00, ..., -2.1346e+00,
-1.3146e+00, -8.8632e-01],
[-3.4554e+00, -2.9833e+00, -2.9326e+00, ..., -9.8516e-01,
-1.2115e+00, -4.6476e-01],
...,
[-4.2052e+00, -2.3608e+00, -3.6214e+00, ..., -1.8003e+00,
-3.0215e+00, -1.6245e+00],
[-3.7054e+00, -7.0942e-01, -2.2901e+00, ..., -9.7632e-01,
-3.1461e+00, -1.3200e+00],
[-4.3577e+00, -2.9305e+00, -2.8477e+00, ..., -2.3793e+00,
-3.4301e+00, -1.2744e+00]],
[[ 1.1017e+00, 5.7575e-01, 6.8627e-01, ..., -7.4762e-01,
1.4972e-01, -3.5066e-01],
[ 3.5346e+00, 3.2671e+00, 2.1956e+00, ..., 1.5127e+00,
1.8107e+00, 2.5049e+00],
[ 1.4291e+00, 5.4590e-01, 4.4457e-01, ..., 9.9587e-01,
1.1835e+00, 1.7091e+00],
...,
[ 2.8450e+00, 1.0012e+00, 9.0806e-01, ..., 2.5494e+00,
2.3236e+00, 1.2513e+00],
[ 3.5627e+00, 3.5751e+00, 2.4541e+00, ..., 2.5327e+00,
2.2731e+00, 2.0680e+00],
[ 1.8771e+00, 7.6581e-01, 1.9484e-01, ..., -1.6984e-01,
-2.2568e-01, -1.3013e-01]]],
[[[ 2.4950e-01, 1.0614e+00, 2.5589e-01, ..., -8.3572e-01,
-1.1464e+00, -2.7145e-01],
[ 2.1363e-01, 5.2611e-01, 1.8655e+00, ..., 1.6152e-01,
6.4248e-01, 6.1205e-01],
[-4.3265e-01, -1.1664e-01, 7.9986e-01, ..., -1.5489e+00,
-7.1565e+00, -3.1814e+00],
...,
[-5.3611e-01, -1.9691e+00, -3.1440e+00, ..., -5.4909e+00,
-1.7923e+00, 5.3543e-02],
[-1.8102e+00, -7.8973e-01, -1.8753e+00, ..., -3.9294e+00,
-2.5229e+00, 1.3926e-01],
[-1.2477e+00, -3.1804e+00, -2.8021e+00, ..., -3.9546e+00,
-3.3059e+00, -1.1744e+00]],
[[-1.0417e+00, -1.6076e+00, -2.5870e+00, ..., -4.3761e+00,
-5.7784e+00, -2.0065e+00],
[-3.9358e+00, -3.8805e+00, -2.3232e+00, ..., -7.5804e+00,
-5.8043e+00, -2.3288e+00],
[-3.7748e+00, -4.2943e+00, -3.4356e+00, ..., -1.1237e+01,
-1.1930e+01, -3.8022e+00],
...,
[-6.2404e+00, -5.6418e+00, -5.8929e+00, ..., -7.9635e+00,
-6.5923e+00, -2.7496e+00],
[-7.8348e+00, -7.5045e+00, -6.3781e+00, ..., -8.9316e+00,
-7.8373e+00, -3.3318e+00],
[-1.8134e+00, -3.5703e+00, -3.1383e+00, ..., -4.5423e+00,
-3.6048e+00, -4.0108e-01]],
[[-7.5157e-02, 1.3133e+00, 1.7001e+00, ..., 6.9444e-01,
2.6457e+00, 1.1007e+00],
[-8.9481e-02, 4.7538e-04, 5.6061e-01, ..., -1.0450e+00,
1.4487e+00, 2.0724e-01],
[ 6.7399e-01, 1.0873e+00, 7.8190e-01, ..., -2.2465e+00,
4.0123e+00, 3.2251e+00],
...,
[ 2.9002e-01, 2.4279e-01, 1.8717e+00, ..., 2.6619e-01,
-7.6540e-01, -2.5576e-02],
[ 5.1960e-01, -4.6066e-01, 2.4049e-01, ..., -1.2577e+00,
1.3043e-01, 1.2556e-01],
[ 6.2076e-01, 9.5582e-01, 2.4043e-01, ..., -6.4842e-01,
-5.5437e-01, -1.9724e+00]],
...,
[[-6.5575e-01, -1.6847e+00, -2.0245e+00, ..., -2.6896e+00,
-4.8622e+00, -1.0190e+00],
[-2.7973e+00, -2.4780e+00, -1.0595e+00, ..., -3.3610e+00,
-5.5571e+00, -1.7084e+00],
[-1.8257e+00, -7.9799e-01, -8.4589e-01, ..., -5.3092e+00,
-8.0011e+00, -2.0689e+00],
...,
[-1.8493e+00, -1.5654e+00, -1.9713e+00, ..., -3.1935e+00,
-5.3543e+00, -3.7287e+00],
[-3.2252e+00, -2.4120e+00, -2.1102e+00, ..., -2.2053e+00,
-4.2400e+00, -4.4398e-01],
[-1.6739e+00, -1.8615e+00, -1.1354e+00, ..., -1.7486e+00,
-2.2770e+00, 8.6919e-01]],
[[-1.3276e+00, -2.3554e+00, -3.8643e+00, ..., -2.7005e+00,
-1.9332e+00, -3.3981e-01],
[-3.6515e+00, -3.0934e+00, -3.9544e+00, ..., -2.3056e+00,
-1.7070e+00, 6.1619e-01],
[-4.4981e+00, -3.4174e+00, -3.2107e+00, ..., -3.3757e+00,
-6.4790e+00, -2.4171e+00],
...,
[-4.7644e+00, -5.1328e+00, -4.0910e+00, ..., -6.2204e+00,
-4.1707e+00, -9.8787e-01],
[-3.9666e+00, -1.6160e+00, -3.5828e+00, ..., -1.1525e+00,
-1.2232e+00, 5.6795e-01],
[-3.1606e+00, -2.7685e+00, -1.2713e+00, ..., 5.0474e-01,
-4.4220e-01, 8.1310e-01]],
[[ 1.2158e+00, -1.2996e-01, 1.1207e+00, ..., -1.0700e+00,
1.2442e-01, -7.0998e-02],
[ 2.3192e+00, 8.7222e-01, 1.1112e+00, ..., 2.0483e+00,
2.6176e+00, 2.4945e+00],
[ 1.3873e+00, 5.9726e-01, 8.1867e-01, ..., 8.4196e-01,
-7.4482e-03, 1.1519e+00],
...,
[ 2.7831e+00, 1.3910e+00, 1.0479e+00, ..., -6.1975e-02,
1.8891e+00, 1.2013e+00],
[ 3.0170e+00, 3.6341e+00, 1.4217e+00, ..., 4.8257e-01,
4.2213e-01, 1.9822e+00],
[ 2.0889e+00, 1.9677e+00, 4.9982e-01, ..., -5.9199e-01,
-5.5421e-01, 1.0461e+00]]]], device='cuda:0'),
tensor([[[[ 3.6725e-01, 1.0920e+00, -2.3630e-01, ..., -3.0864e+00,
-2.9245e+00, -1.6397e+00],
[ 1.7894e-01, 5.1232e-01, 1.0973e+00, ..., -5.1249e+00,
-2.2354e+00, -1.1798e+00],
[-8.4691e-01, -7.4789e-01, 4.6923e-01, ..., -5.4473e-01,
-3.1620e-01, -2.4544e+00],
...,
[-1.1146e+00, -5.5245e+00, -4.4636e+00, ..., -9.5770e+00,
-9.5803e+00, -3.6935e+00],
[-1.6900e+00, -2.1858e+00, -6.5003e+00, ..., -6.2584e+00,
-5.7055e+00, -1.7037e+00],
[-1.0636e+00, -2.2286e+00, -4.9959e+00, ..., -6.4440e+00,
-4.7795e+00, -2.0082e+00]],
[[-1.1086e+00, -1.8696e+00, -3.6452e+00, ..., -7.2595e+00,
-8.9415e+00, -2.1359e+00],
[-4.1312e+00, -4.3259e+00, -2.9555e+00, ..., -5.9140e+00,
-7.9987e+00, -3.3915e+00],
[-3.8717e+00, -4.1713e+00, -3.8661e+00, ..., -2.1274e+00,
-4.4808e+00, -5.3255e+00],
...,
[-5.4562e+00, -8.1455e+00, -1.4004e+01, ..., -2.2025e+01,
-1.5684e+01, -4.5866e+00],
[-7.1534e+00, -5.4641e+00, -1.0346e+01, ..., -9.2312e+00,
-9.9964e+00, -4.8735e+00],
[-1.8626e+00, -3.0396e+00, -2.1361e+00, ..., -3.9538e+00,
-3.5193e+00, -1.2889e+00]],
[[-1.2863e-01, 1.3082e+00, 1.7765e+00, ..., 1.6097e+00,
2.5358e+00, 9.2330e-01],
[-3.4659e-03, -1.7683e-02, 8.0224e-01, ..., 4.7464e+00,
3.7381e+00, -5.0549e-03],
[ 7.2155e-01, 1.3767e+00, 1.4929e+00, ..., 4.2590e+00,
4.2612e+00, 7.7763e-01],
...,
[-5.1183e-01, -8.8353e-03, -2.4003e+00, ..., -6.5537e-01,
-1.3306e+00, -2.3807e+00],
[ 1.1187e-01, -5.4166e-01, 1.2352e+00, ..., 1.0687e+00,
-5.8954e-02, -1.6380e+00],
[ 7.7766e-01, -4.6666e-01, 7.5529e-01, ..., -2.7262e+00,
-1.0385e+00, -1.5173e+00]],
...,
[[-6.9756e-01, -1.7489e+00, -2.3911e+00, ..., -5.2089e+00,
-5.4033e+00, -1.7425e+00],
[-2.9155e+00, -2.6035e+00, -1.2303e+00, ..., -5.8686e+00,
-4.5926e+00, -3.5561e+00],
[-2.2536e+00, -1.3300e+00, -8.5929e-01, ..., -2.5973e+00,
-5.8487e+00, -2.7682e+00],
...,
[-2.0614e+00, -3.9303e+00, -5.3293e+00, ..., -1.1419e+01,
-9.8445e+00, -4.1342e+00],
[-2.2823e+00, -3.0843e+00, -5.2648e+00, ..., -1.6052e+00,
-6.4243e+00, -3.4182e+00],
[-1.5179e+00, -4.7923e-01, -2.2929e+00, ..., -2.5983e+00,
-3.3908e+00, -2.0832e-01]],
[[-1.3189e+00, -2.3901e+00, -3.9719e+00, ..., -1.0592e+00,
-8.7415e-01, -7.2862e-01],
[-3.7923e+00, -3.1351e+00, -4.4754e+00, ..., -3.5747e+00,
-1.7970e+00, -5.3232e-01],
[-4.5928e+00, -3.3480e+00, -3.8385e+00, ..., -4.5956e+00,
-2.8216e+00, 1.8233e-01],
...,
[-5.0483e+00, -4.7822e+00, 3.8668e+00, ..., -1.1849e+01,
-4.5469e+00, -1.9610e+00],
[-3.0656e+00, -3.3959e+00, -3.7905e+00, ..., -3.8155e+00,
-7.6684e-01, 1.7552e+00],
[-2.2713e+00, -1.7653e+00, -4.3100e+00, ..., 4.7890e-01,
3.9459e-01, 1.0155e+00]],
[[ 1.2648e+00, -6.9288e-02, 1.1317e+00, ..., 7.9519e-01,
6.0939e-01, -1.7411e+00],
[ 2.3677e+00, 1.1238e+00, 1.2326e+00, ..., 2.6167e+00,
2.8529e+00, 1.5580e+00],
[ 1.2664e+00, 6.5541e-01, 2.4784e-01, ..., 1.7238e+00,
5.2422e+00, 2.0294e+00],
...,
[ 1.4359e+00, 3.7736e-01, 8.6677e+00, ..., 7.4681e+00,
3.7447e+00, 2.0365e+00],
[ 2.7594e+00, 3.2575e+00, 2.6717e+00, ..., -1.3177e+00,
1.0589e+00, 2.8867e+00],
[ 2.7426e+00, 1.7669e+00, -1.5808e+00, ..., 2.5055e+00,
3.3655e-01, 4.4112e-01]]],
[[[-3.5981e-01, 1.0684e+00, 1.3755e+00, ..., -1.0348e+00,
-3.4124e+00, -1.0223e+00],
[ 4.6478e-01, 2.8591e-01, 2.9145e+00, ..., 3.9916e-01,
-2.2481e+00, -6.7780e-01],
[-4.9350e-01, -4.0698e-01, -3.3857e-01, ..., 2.1179e-01,
-3.0396e+00, -1.0084e+00],
...,
[-3.3406e-01, 3.5164e-02, 1.4587e+00, ..., -3.3604e+00,
-3.6868e+00, 1.2096e-01],
[-1.4835e+00, -8.1240e-02, 9.0165e-01, ..., -3.1947e+00,
-3.9600e+00, -7.9033e-01],
[-1.2993e+00, -2.4139e+00, -1.2985e+00, ..., -3.6328e+00,
-3.7128e+00, -1.6420e+00]],
[[-1.5441e+00, -2.4963e+00, -3.2737e+00, ..., -5.3763e+00,
-9.1363e+00, -2.0210e+00],
[-4.8464e+00, -6.2752e+00, -5.4767e+00, ..., -5.6605e+00,
-8.8702e+00, -3.3484e+00],
[-4.1968e+00, -5.6879e+00, -9.0018e+00, ..., -6.4885e+00,
-8.7956e+00, -2.7534e+00],
...,
[-4.4228e+00, -5.1388e+00, -6.1170e+00, ..., -6.5487e+00,
-7.2311e+00, -1.9694e+00],
[-7.7897e+00, -8.0015e+00, -7.5175e+00, ..., -9.1350e+00,
-1.0861e+01, -4.3916e+00],
[-1.4928e+00, -3.1864e+00, -2.2928e+00, ..., -3.8490e+00,
-4.1334e+00, -5.8922e-01]],
[[-3.2673e-01, 2.3022e+00, 1.4306e+00, ..., 2.1056e+00,
2.4924e+00, 7.1250e-01],
[-4.9386e-01, 5.0541e-01, -4.3030e-01, ..., 1.1055e+00,
2.1312e+00, -3.2228e-01],
[ 7.0942e-02, 1.0849e+00, 5.3580e-01, ..., -2.8870e-01,
1.1049e+00, -3.6187e-01],
...,
[-6.4392e-01, 7.5849e-01, 1.2083e+00, ..., 1.2915e+00,
7.6787e-01, 1.7959e-01],
[-1.2441e+00, 2.1502e-01, 1.0566e+00, ..., -4.8510e-01,
-2.6518e-01, -6.8348e-01],
[-2.8724e-01, -2.8055e-01, 2.3228e-01, ..., 4.5760e-01,
3.8710e-01, -1.6831e+00]],
...,
[[-1.2073e+00, -2.0076e+00, -4.2187e-01, ..., -3.5231e+00,
-5.6234e+00, -1.3134e+00],
[-3.4485e+00, -3.9659e+00, -1.2265e+00, ..., -2.6968e+00,
-6.0220e+00, -2.4373e+00],
[-2.3165e+00, -3.2943e+00, -2.3161e+00, ..., -1.7603e+00,
-7.1935e+00, -1.8403e+00],
...,
[-1.1836e+00, -2.4921e+00, -7.0560e-01, ..., -1.5669e+00,
-4.2230e+00, -2.1913e+00],
[-2.2124e+00, -3.8270e+00, -6.1215e-01, ..., -2.2342e+00,
-4.6085e+00, -7.8170e-01],
[-1.5043e+00, -1.4019e+00, 2.5488e-01, ..., -1.4531e+00,
-2.3459e+00, 9.1825e-01]],
[[-1.7377e+00, -1.5994e+00, -1.4031e+00, ..., -2.1989e+00,
-1.0646e+00, 2.1691e-01],
[-3.8973e+00, -2.5053e+00, -1.2105e+00, ..., -2.5740e+00,
-2.8596e+00, -1.3414e-01],
[-3.7502e+00, -3.5598e+00, -2.6107e+00, ..., -1.4503e+00,
-2.6008e+00, 4.0222e-01],
...,
[-2.6277e+00, -2.7507e+00, -3.3107e+00, ..., -2.1200e+00,
-1.9975e+00, 1.2827e-01],
[-2.0845e+00, -3.1083e-01, -1.4934e+00, ..., -1.1461e-01,
-8.5405e-01, 1.1775e+00],
[-2.2099e+00, -1.3548e+00, -4.9109e-01, ..., -6.2819e-01,
-1.2001e+00, 5.9183e-01]],
[[ 1.1394e+00, 3.5229e-01, 9.1254e-01, ..., 1.7413e+00,
1.7314e+00, 8.5135e-02],
[ 4.0592e+00, 3.0528e+00, 3.6637e+00, ..., 3.4775e+00,
3.0400e+00, 3.7694e+00],
[ 1.5693e+00, 6.5124e-01, 1.4041e+00, ..., 2.9722e+00,
1.6550e+00, 1.7371e+00],
...,
[ 2.9236e+00, 2.7795e+00, 2.6803e+00, ..., 1.9766e+00,
1.6730e+00, 1.2470e+00],
[ 3.8709e+00, 3.9828e+00, 2.9328e+00, ..., 3.2969e+00,
1.4363e+00, 2.7780e+00],
[ 2.4330e+00, 2.3162e+00, 1.5674e+00, ..., 4.3047e-01,
-2.4222e-01, 1.2491e+00]]],
[[[-1.3589e-01, 1.0572e+00, 2.2529e+00, ..., -2.8393e+00,
-7.6877e-01, 3.7950e-01],
[ 1.1285e+00, 1.6880e+00, 3.1467e+00, ..., -8.4871e+00,
-1.5209e+00, 4.5424e-01],
[ 9.9247e-01, 8.6331e-01, -4.5252e+00, ..., 4.2498e-01,
2.3263e+00, 1.3520e+00],
...,
[-1.8256e-01, -2.7264e+00, -2.7009e+00, ..., 4.2513e+00,
1.4002e+00, -6.8665e-01],
[-5.5781e-01, -9.5869e-01, -3.4386e+00, ..., -3.3665e+00,
-1.2254e+00, 2.0816e+00],
[-2.8601e+00, -5.4211e+00, -5.6375e+00, ..., -3.6231e+00,
4.5052e-01, 2.9405e-01]],
[[-3.9558e-01, -1.6997e+00, -4.7169e+00, ..., -3.9419e+00,
-5.3316e+00, -5.4534e-01],
[-4.3356e+00, -7.4668e+00, -7.2284e+00, ..., -9.3122e+00,
-6.2933e+00, -2.0456e+00],
[-3.7655e+00, -8.5079e+00, -1.7542e+01, ..., 2.0286e+00,
-2.7733e+00, -1.9066e+00],
...,
[-6.3748e+00, -7.1727e+00, -8.2037e+00, ..., 2.9920e+00,
3.4541e-01, -1.9083e+00],
[-8.2431e+00, -8.8255e+00, -8.7981e+00, ..., -1.3502e+00,
-3.2035e-02, 8.7564e-01],
[-2.9437e+00, -5.0420e+00, -4.3795e+00, ..., -2.6872e+00,
-6.0494e-02, 6.5671e-01]],
[[-4.1558e-01, 1.8853e+00, 1.2557e+00, ..., 5.5199e-01,
2.1028e+00, 1.0256e+00],
[ 1.9620e-01, 1.0983e+00, -1.6267e+00, ..., 7.2278e+00,
4.1284e+00, -5.8054e-01],
[ 2.1052e-02, 1.2591e+00, 1.9111e+00, ..., 4.5771e-01,
4.9396e+00, 2.8089e+00],
...,
[ 9.8023e-01, 1.4431e+00, -1.0473e+00, ..., -9.7925e+00,
1.9893e+00, 4.1144e+00],
[ 7.9671e-01, 7.7990e-01, -1.5824e-01, ..., -9.5762e+00,
-9.0131e+00, -4.9985e-01],
[ 1.6376e+00, 1.9101e+00, 1.0668e+00, ..., -3.5985e-01,
-1.8989e+00, 1.4676e-01]],
...,
[[-6.7355e-01, -8.9915e-01, -7.7226e-01, ..., -8.2516e-01,
-1.6852e+00, -1.0179e+00],
[-2.9943e+00, -3.8913e+00, -3.5407e+00, ..., -5.4967e+00,
-4.8034e+00, -4.0488e+00],
[-4.6123e-01, -4.3988e+00, -1.0760e+01, ..., 3.3931e+00,
-2.0604e+00, -1.9369e+00],
...,
[-2.8967e+00, -4.6070e+00, -2.8838e+00, ..., 1.4192e-01,
-4.4610e+00, -5.3926e-01],
[-3.3234e+00, -4.3512e+00, -2.9420e+00, ..., -2.0805e+00,
-3.6263e+00, -8.7911e-01],
[-4.1130e+00, -4.6267e+00, -3.7895e+00, ..., -3.3982e+00,
-1.4373e+00, -1.8402e+00]],
[[-1.5255e+00, -1.7866e+00, -2.1838e+00, ..., 3.6855e-01,
-1.2036e+00, -1.1053e+00],
[-3.6879e+00, -2.3100e+00, -2.0202e+00, ..., -2.1095e+00,
-2.1234e-01, -3.6870e-01],
[-3.1548e+00, -5.7517e+00, -9.7268e+00, ..., 5.1446e+00,
-2.0302e+00, 3.1163e-01],
...,
[-4.1486e+00, -3.8420e+00, -1.3966e+00, ..., -1.9500e+00,
-3.3971e+00, 1.1933e+00],
[-3.7031e+00, -2.1918e+00, -1.5301e+00, ..., 1.3147e+00,
1.4224e-01, 1.2195e+00],
[-4.8624e+00, -3.8142e+00, -2.3083e+00, ..., -2.9999e-01,
7.3389e-01, -4.3819e-01]],
[[ 6.8010e-01, 1.2819e-01, 1.2731e+00, ..., -1.3923e+00,
-3.6164e-02, -6.8388e-01],
[ 3.1085e+00, 3.4757e+00, 4.4505e+00, ..., 4.2956e+00,
4.8816e+00, 2.5678e+00],
[ 1.3113e+00, 2.4159e+00, 4.2311e+00, ..., 2.9237e+00,
4.6114e+00, 2.5011e+00],
...,
[ 3.2228e+00, 1.2543e+00, 2.9718e+00, ..., 5.1868e-01,
5.2780e-01, 2.9201e+00],
[ 4.7917e+00, 4.4945e+00, 1.8526e+00, ..., -4.1315e+00,
-3.1707e+00, 1.0378e+00],
[ 2.5119e+00, 1.1232e+00, -5.3782e-01, ..., -4.4541e+00,
-8.0351e-02, 7.6286e-01]]],
[[[ 8.7959e-01, 1.3591e+00, -2.6446e+00, ..., -2.7232e-01,
-5.6844e-01, 2.9317e-01],
[ 1.3060e+00, 6.5701e-01, -5.4651e+00, ..., 2.6264e+00,
1.1436e+00, -8.9448e-01],
[ 7.4596e-01, -1.6658e+00, -6.9311e+00, ..., 4.4830e+00,
1.8070e+00, -2.0389e+00],
...,
[-3.2560e+00, -8.7264e+00, -1.4040e+00, ..., -3.1987e+00,
-3.9673e+00, -7.9126e-01],
[-2.5482e+00, -5.3230e+00, 9.6019e-01, ..., -2.8962e+00,
-3.0460e+00, 2.2135e-01],
[-1.5334e+00, -3.9391e+00, -3.1482e+00, ..., -4.2822e+00,
-3.6614e+00, -1.4686e+00]],
[[-1.5301e+00, -2.0823e+00, -9.0207e+00, ..., -1.7501e+00,
-5.0588e+00, -1.3160e+00],
[-5.1931e+00, -6.3917e+00, -9.3182e+00, ..., 3.8879e-02,
-6.3402e+00, -4.8153e+00],
[-4.5738e+00, -5.6889e+00, -6.4910e+00, ..., -3.4924e-01,
-6.1934e+00, -6.2287e+00],
...,
[-5.8732e+00, -1.0428e+01, -1.2139e+00, ..., -8.4122e+00,
-9.6955e+00, -2.2208e+00],
[-9.0037e+00, -1.0478e+01, -7.8115e+00, ..., -1.0768e+01,
-1.1028e+01, -3.7055e+00],
[-2.0980e+00, -3.5267e+00, -5.9208e+00, ..., -4.3096e+00,
-4.3669e+00, -6.3175e-01]],
[[ 3.1429e-01, 1.1185e+00, 1.6716e+00, ..., 3.2353e+00,
2.4811e+00, -1.1538e-01],
[-2.8749e-02, -2.0199e-01, 2.8707e+00, ..., 2.7331e+00,
3.1921e+00, -1.0875e+00],
[ 4.5996e-01, 1.8798e+00, 3.6361e+00, ..., 1.0435e+00,
3.4898e+00, 5.6095e-01],
...,
[ 2.3044e-01, -5.7224e-01, -6.5675e+00, ..., 1.4610e+00,
7.7804e-01, -1.0728e+00],
[-2.4314e-01, 7.0908e-01, -5.5029e+00, ..., 4.0975e-01,
6.0654e-01, -7.2750e-01],
[-2.1252e-01, 7.8665e-01, 3.5223e-01, ..., -9.1838e-02,
-5.3320e-01, -2.0219e+00]],
...,
[[-1.5454e-01, -1.8296e+00, -2.0408e+00, ..., -2.7366e+00,
-3.1277e+00, -1.0488e-01],
[-2.4175e+00, -4.1358e+00, -3.8706e+00, ..., -1.8354e+00,
-4.8090e+00, -3.2618e+00],
[-1.4039e+00, -2.1748e+00, -1.3924e+00, ..., -1.3253e+00,
-4.4279e+00, -2.2675e+00],
...,
[-4.7179e+00, -7.6306e+00, -2.5737e+00, ..., -4.7107e+00,
-6.1405e+00, -2.8815e+00],
[-3.4515e+00, -5.4671e+00, -3.4728e+00, ..., -4.1711e+00,
-5.1852e+00, -1.2541e+00],
[-2.5698e+00, -4.2304e+00, -6.3717e+00, ..., -2.9415e+00,
-2.6440e+00, 3.7906e-01]],
[[-9.6330e-01, -1.8987e+00, -3.8853e+00, ..., -1.4334e+00,
2.4317e-01, -3.4517e-01],
[-2.8739e+00, -4.7072e+00, -3.8588e+00, ..., -3.0241e+00,
-3.1196e+00, -7.9495e-01],
[-3.5963e+00, -4.4972e+00, -3.4420e+00, ..., -2.8947e+00,
-4.9711e+00, -1.3584e+00],
...,
[-5.0631e+00, -5.1099e-01, -1.7902e+00, ..., -3.4452e+00,
-2.6751e+00, -8.8278e-01],
[-3.4695e+00, 2.3260e-01, 5.2917e+00, ..., -1.9764e+00,
-9.7939e-01, 4.1056e-01],
[-2.1220e+00, -1.8007e+00, -4.9776e-01, ..., -1.8902e+00,
-1.1221e+00, 4.1260e-01]],
[[ 1.6677e+00, 2.9438e-01, 2.4321e+00, ..., 1.7218e+00,
1.8804e+00, -2.7970e-01],
[ 3.2609e+00, 2.7579e+00, 2.3102e+00, ..., 2.1297e+00,
4.0650e+00, 3.1901e+00],
[ 2.3927e+00, 9.6910e-01, 1.1609e+00, ..., 3.1713e+00,
3.3077e+00, 9.2463e-01],
...,
[ 2.1517e+00, 9.9830e-01, -1.7110e+00, ..., 1.7373e+00,
1.3243e+00, 2.1184e-01],
[ 4.5026e+00, 2.9464e+00, 3.5576e+00, ..., 3.0025e+00,
1.8883e+00, 1.5737e+00],
[ 2.8844e+00, 1.5920e+00, 3.4479e+00, ..., 4.9013e-01,
4.9890e-01, 9.7229e-01]]],
[[[ 1.8324e-01, 1.4494e+00, 1.1959e+00, ..., -7.1736e-01,
-1.2578e-02, 6.5655e-01],
[ 3.0762e-01, -1.7073e-01, 1.0919e+00, ..., -1.9180e+00,
-1.2010e+00, 1.0550e+00],
[-5.7360e-01, 2.6302e-01, 6.9912e-01, ..., -2.6533e+00,
-1.9662e+00, 8.3666e-01],
...,
[-1.0291e+00, -2.0081e+00, 5.6025e-01, ..., -8.1735e-01,
-5.8113e-01, -1.4545e+00],
[-1.1344e+00, -1.1077e+00, 2.2402e-01, ..., -9.2353e-01,
-3.9926e-01, -1.6788e+00],
[-2.7157e+00, -5.1616e+00, -4.5106e+00, ..., -4.7152e+00,
-3.9677e+00, -3.8271e+00]],
[[-9.3329e-01, -2.1858e+00, -4.2358e+00, ..., -3.3171e+00,
-4.9255e+00, -3.8608e-01],
[-5.0225e+00, -5.5310e+00, -4.6165e+00, ..., -5.8900e+00,
-7.2243e+00, -1.6595e+00],
[-3.6683e+00, -4.7248e+00, -5.6231e+00, ..., -5.0846e+00,
-5.6999e+00, -1.4969e+00],
...,
[-5.6368e+00, -5.3560e+00, -3.5443e+00, ..., -5.2971e+00,
-5.3809e+00, -3.8156e+00],
[-6.9995e+00, -7.5779e+00, -5.8388e+00, ..., -7.4337e+00,
-8.0305e+00, -6.5506e+00],
[-2.4904e+00, -4.4525e+00, -4.0190e+00, ..., -3.9576e+00,
-4.1742e+00, -2.5768e+00]],
[[-3.2179e-01, 1.9370e+00, 1.6279e+00, ..., 9.3836e-01,
2.1327e+00, 9.3911e-01],
[-4.8357e-02, 9.9559e-01, -1.2630e-02, ..., 6.0934e-01,
1.2441e+00, -4.2652e-01],
[-4.1337e-02, 2.2604e+00, 5.5847e-01, ..., 3.9242e-01,
1.8235e+00, 2.1310e-01],
...,
[ 9.0244e-01, 8.4538e-01, 8.0064e-01, ..., 1.8579e+00,
5.6706e-01, 4.5404e-01],
[ 7.1357e-01, 5.7883e-01, 6.2726e-01, ..., -2.3146e-01,
2.5710e-01, 5.2902e-01],
[ 1.3262e+00, 1.5234e+00, 1.1303e+00, ..., 1.5826e+00,
1.2527e+00, 2.3151e-01]],
...,
[[-9.4279e-01, -1.3460e+00, -1.6519e+00, ..., -2.8893e+00,
-2.7441e+00, -4.6372e-01],
[-3.4444e+00, -3.3636e+00, -2.8059e+00, ..., -5.0964e+00,
-4.3859e+00, -2.5921e+00],
[-1.5595e+00, -1.6974e+00, -3.3052e+00, ..., -2.6895e+00,
-5.1432e+00, -1.2950e+00],
...,
[-2.5554e+00, -2.5801e+00, -2.9716e-01, ..., -2.0751e+00,
-4.0805e+00, -2.0036e+00],
[-2.8888e+00, -3.0857e+00, -1.6758e+00, ..., -2.1376e+00,
-4.3608e+00, -2.5098e+00],
[-3.4584e+00, -3.8231e+00, -3.3422e+00, ..., -3.3088e+00,
-3.8017e+00, -1.7012e+00]],
[[-1.7871e+00, -1.8441e+00, -3.6699e+00, ..., -1.2058e+00,
-1.3883e+00, -1.3827e+00],
[-4.5508e+00, -2.7995e+00, -3.2235e+00, ..., -2.5166e+00,
-1.4506e+00, -9.3350e-01],
[-3.4567e+00, -2.9551e+00, -4.9034e+00, ..., -1.0999e+00,
-1.0277e+00, -4.4951e-01],
...,
[-3.9929e+00, -2.0314e+00, -2.9046e+00, ..., -1.5505e+00,
-2.7276e+00, -1.2973e+00],
[-3.5872e+00, -5.3265e-01, -1.9290e+00, ..., -4.7200e-01,
-2.5964e+00, -1.1866e+00],
[-4.3960e+00, -2.9448e+00, -2.7254e+00, ..., -1.8974e+00,
-3.2232e+00, -1.3036e+00]],
[[ 1.1182e+00, 6.6396e-01, 6.5122e-01, ..., -6.0908e-01,
1.0978e-01, -4.7142e-01],
[ 3.6039e+00, 2.9263e+00, 2.1335e+00, ..., 1.4437e+00,
2.1978e+00, 2.5354e+00],
[ 1.4970e+00, 6.1980e-01, 1.5603e+00, ..., 7.6400e-01,
1.1027e+00, 2.0177e+00],
...,
[ 2.7883e+00, 1.1392e+00, 1.1373e+00, ..., 2.4113e+00,
2.4244e+00, 1.7307e+00],
[ 3.5806e+00, 3.3334e+00, 2.4838e+00, ..., 3.7752e+00,
2.7602e+00, 2.1341e+00],
[ 1.8718e+00, 7.1930e-01, 1.1891e-01, ..., -1.4877e-01,
-2.7861e-01, -1.8416e-01]]]], device='cuda:0'),
tensor([[[[ 2.7346e-01, 1.1982e+00, 5.6399e-01, ..., -9.2838e-01,
-5.2690e-02, 6.8736e-01],
[ 5.8262e-01, -5.2239e-02, 9.5920e-01, ..., -1.1796e+00,
-1.5831e+00, 1.1353e+00],
[-3.7715e-01, -6.4130e-02, 1.4557e+00, ..., -3.5644e-01,
-1.4355e+00, 8.2441e-01],
...,
[-9.6072e-01, -2.0774e+00, 3.4409e-01, ..., 1.7402e+00,
-1.5713e+00, -2.2415e+00],
[-1.0184e+00, -1.1137e+00, 2.1146e-01, ..., 1.5214e+00,
-9.6685e-01, -2.7430e+00],
[-2.7066e+00, -5.2608e+00, -4.1994e+00, ..., -3.1416e+00,
-4.3270e+00, -4.3146e+00]],
[[-7.4805e-01, -2.4695e+00, -3.9366e+00, ..., -4.3215e+00,
-4.9487e+00, -3.9135e-01],
[-4.6958e+00, -6.0969e+00, -5.2167e+00, ..., -8.3249e+00,
-7.2720e+00, -1.6527e+00],
[-3.5768e+00, -4.2484e+00, -3.9537e+00, ..., -8.5942e+00,
-6.3413e+00, -1.4083e+00],
...,
[-5.7150e+00, -5.7642e+00, -4.2551e+00, ..., -3.9660e+00,
-9.2097e+00, -4.1201e+00],
[-7.1160e+00, -7.8509e+00, -6.3641e+00, ..., -4.3163e+00,
-9.5463e+00, -6.2036e+00],
[-2.5253e+00, -4.4606e+00, -4.1163e+00, ..., -3.3291e+00,
-4.8011e+00, -2.5822e+00]],
[[-2.6316e-01, 2.0913e+00, 1.1057e+00, ..., 7.8218e-01,
2.0376e+00, 9.4137e-01],
[-1.0731e-01, 1.2354e+00, 8.2972e-01, ..., 7.6095e-01,
9.4685e-01, -3.4193e-01],
[-1.8391e-01, 2.0559e+00, 1.3538e+00, ..., 3.8948e-01,
8.7926e-01, 3.7660e-01],
...,
[ 9.6296e-01, 1.0792e+00, 1.2083e+00, ..., 4.4389e+00,
1.4212e+00, 1.3201e-01],
[ 7.7243e-01, 7.7986e-01, 6.7512e-01, ..., 3.1053e+00,
8.9006e-01, -5.1997e-03],
[ 1.3866e+00, 1.6330e+00, 1.0554e+00, ..., 1.9292e+00,
1.2633e+00, 3.1630e-01]],
...,
[[-7.8400e-01, -1.9006e+00, -1.0900e+00, ..., -3.1006e+00,
-3.0845e+00, -4.4743e-01],
[-3.2456e+00, -4.3312e+00, -2.0203e+00, ..., -3.3119e+00,
-4.7036e+00, -2.7220e+00],
[-1.5238e+00, -2.1948e+00, -6.0490e-01, ..., -1.1846e+00,
-5.0867e+00, -1.4370e+00],
...,
[-2.6613e+00, -2.4176e+00, -8.8883e-01, ..., -6.0884e+00,
-3.0523e+00, -2.9178e+00],
[-2.9597e+00, -2.9611e+00, -1.5194e+00, ..., -5.0985e+00,
-3.3489e+00, -2.9529e+00],
[-3.4988e+00, -3.8494e+00, -3.0749e+00, ..., -4.6859e+00,
-3.4455e+00, -2.1631e+00]],
[[-1.7647e+00, -2.0399e+00, -2.0703e+00, ..., -1.4984e+00,
-1.7615e+00, -1.3687e+00],
[-4.4883e+00, -3.3349e+00, -1.9042e+00, ..., -2.9551e+00,
-2.7148e+00, -7.2177e-01],
[-3.4741e+00, -2.8716e+00, -2.5180e+00, ..., -4.3832e-01,
-2.5417e+00, -1.6848e-01],
...,
[-4.3210e+00, -2.5845e+00, -3.0265e+00, ..., -6.3045e+00,
-3.0201e+00, -1.8917e+00],
[-3.7765e+00, -9.7117e-01, -2.0443e+00, ..., -5.8249e+00,
-2.5882e+00, -1.2212e+00],
[-4.4524e+00, -3.0756e+00, -2.1299e+00, ..., -4.8162e+00,
-3.1955e+00, -1.0164e+00]],
[[ 1.0530e+00, 5.6654e-01, 6.6384e-01, ..., -8.8909e-01,
-4.2607e-01, -4.4475e-01],
[ 3.4995e+00, 3.2830e+00, 2.2333e+00, ..., 2.4655e+00,
1.2585e+00, 2.6113e+00],
[ 1.5149e+00, 4.6975e-01, 5.8833e-01, ..., 9.1329e-01,
1.0679e+00, 1.9854e+00],
...,
[ 2.7431e+00, 9.1704e-01, 1.5123e+00, ..., 3.1041e+00,
1.0321e+00, 1.0568e+00],
[ 3.7514e+00, 3.4454e+00, 2.4611e+00, ..., 2.4479e+00,
1.3435e+00, 1.9350e+00],
[ 1.9483e+00, 7.1685e-01, 6.1002e-01, ..., 7.2408e-01,
3.3177e-01, 6.8197e-02]]],
[[[ 5.5539e-01, 1.6561e+00, 7.1756e-01, ..., 1.9306e+00,
7.8321e-02, 5.8377e-01],
[ 1.6002e-01, -4.2756e-01, 1.0714e+00, ..., 6.0639e-01,
-9.4669e-01, -5.2282e-01],
[-3.6313e-02, -4.2129e-01, 4.6734e-01, ..., -1.7514e-01,
-6.0990e-01, -9.4611e-01],
...,
[-8.6750e-01, -5.1532e+00, -4.1643e+00, ..., -3.7257e+00,
-3.7986e+00, -8.6423e-01],
[-2.4448e+00, -2.1683e+00, -1.2822e+00, ..., -2.6712e+00,
-3.1242e+00, 3.5246e-02],
[-2.1039e+00, -3.4714e+00, -2.9343e+00, ..., -4.1383e+00,
-3.4854e+00, -1.4586e+00]],
[[-1.3962e+00, -2.0254e+00, -2.4287e+00, ..., -5.4847e-01,
-5.4983e+00, -1.1564e+00],
[-4.9097e+00, -6.6467e+00, -4.2924e+00, ..., -6.1468e+00,
-7.5798e+00, -3.9601e+00],
[-4.2849e+00, -4.8074e+00, -3.3940e+00, ..., -7.8095e+00,
-6.2041e+00, -4.3751e+00],
...,
[-6.9224e+00, -9.3313e+00, -8.4736e+00, ..., -8.2237e+00,
-9.5064e+00, -2.2978e+00],
[-9.9362e+00, -1.0043e+01, -8.4000e+00, ..., -1.0641e+01,
-1.1010e+01, -3.8094e+00],
[-2.4445e+00, -4.6566e+00, -3.6246e+00, ..., -4.3654e+00,
-4.0839e+00, -5.9282e-01]],
[[-5.1474e-02, 1.1114e+00, 1.0166e+00, ..., 2.3767e+00,
2.2442e+00, 2.2262e-01],
[ 1.7428e-01, 3.4034e-02, -7.5351e-01, ..., 6.8850e-01,
1.9949e+00, -6.8698e-01],
[ 8.4514e-01, 2.8383e+00, 1.3100e+00, ..., 2.3087e+00,
1.3026e+00, 6.5509e-01],
...,
[ 3.2689e-01, 6.7155e-01, -3.1570e+00, ..., 1.2603e+00,
8.1070e-01, -1.0302e+00],
[-8.7020e-01, -4.9014e-01, -1.0342e+00, ..., 2.0467e-01,
6.8280e-01, -5.2702e-01],
[ 3.7242e-01, 8.7656e-01, -6.9234e-03, ..., 2.3916e-02,
-6.0523e-01, -1.9001e+00]],
...,
[[-9.2071e-01, -8.5128e-01, -1.5531e+00, ..., -2.4621e+00,
-2.2571e+00, 1.8556e-02],
[-4.2851e+00, -4.6061e+00, -2.9991e+00, ..., -4.6096e+00,
-4.9871e+00, -3.2155e+00],
[-2.5360e+00, -1.5170e+00, -1.5567e+00, ..., -9.5035e-01,
-4.2142e+00, -2.4108e+00],
...,
[-2.4215e+00, -4.8409e+00, -2.4905e+00, ..., -4.8757e+00,
-6.0646e+00, -2.9466e+00],
[-3.4976e+00, -5.7497e+00, -3.3988e+00, ..., -4.6672e+00,
-5.2813e+00, -1.3997e+00],
[-3.0321e+00, -3.2098e+00, -2.7820e+00, ..., -2.9930e+00,
-2.3979e+00, 4.4303e-01]],
[[-1.1437e+00, 1.5247e-01, -2.9903e+00, ..., -1.6623e+00,
-3.5745e-01, 3.0928e-01],
[-4.1705e+00, -1.0302e+00, -3.3295e+00, ..., -2.6513e+00,
-2.4749e+00, 3.9577e-01],
[-3.9045e+00, -1.7395e+00, -3.3734e+00, ..., -1.6222e+00,
-2.7823e+00, -5.9043e-01],
...,
[-4.7272e+00, -3.8987e+00, -6.3274e-01, ..., -3.3299e+00,
-2.5612e+00, -8.4706e-01],
[-3.3601e+00, -3.9414e-01, -7.0247e-01, ..., -1.8350e+00,
-1.1268e+00, 3.8562e-01],
[-3.4593e+00, -2.4020e+00, -2.3946e+00, ..., -1.3802e+00,
-1.1099e+00, 3.3253e-01]],
[[ 1.6969e+00, 7.6634e-01, 8.6746e-01, ..., 2.0105e+00,
1.3583e+00, 1.1642e-01],
[ 3.1058e+00, 2.1126e+00, 2.0375e+00, ..., 1.9242e+00,
2.0633e+00, 3.5855e+00],
[ 2.0979e+00, 1.5325e-01, 1.0998e+00, ..., 2.6945e+00,
2.8694e+00, 2.2539e+00],
...,
[ 3.0464e+00, 1.9203e+00, 4.3466e+00, ..., 1.7840e+00,
1.3305e+00, 2.3454e-01],
[ 4.5355e+00, 4.0420e+00, 2.3427e+00, ..., 2.6811e+00,
1.8154e+00, 1.6069e+00],
[ 2.7956e+00, 3.0042e+00, 1.6637e+00, ..., 3.3459e-01,
5.8283e-01, 1.0396e+00]]],
[[[ 3.2699e-01, 1.2466e+00, 8.8156e-02, ..., -2.9696e+00,
-2.2777e+00, -1.0108e+00],
[ 3.2679e-01, 7.2397e-01, 9.2611e-01, ..., -1.4958e+00,
-2.2413e+00, -2.5601e-01],
[-3.3871e-01, 1.2850e-01, 1.0946e+00, ..., -1.1412e+00,
-2.6402e+00, 3.1375e-02],
...,
[-9.7025e-01, -6.2792e-01, -3.3470e-02, ..., -4.7127e+00,
-3.8824e+00, 2.6802e-01],
[-2.1294e+00, -1.2453e+00, -3.4577e-01, ..., -3.7252e+00,
-4.0407e+00, 3.3225e-01],
[-1.3168e+00, -3.4968e+00, -3.2886e+00, ..., -3.9206e+00,
-3.1842e+00, -1.5019e+00]],
[[-1.0437e+00, -1.6281e+00, -3.3925e+00, ..., -5.9062e+00,
-7.0886e+00, -1.3897e+00],
[-3.8878e+00, -4.1445e+00, -3.9589e+00, ..., -5.5902e+00,
-8.2646e+00, -2.4440e+00],
[-3.5871e+00, -4.5592e+00, -4.7143e+00, ..., -4.8518e+00,
-7.4580e+00, -2.3532e+00],
...,
[-6.0701e+00, -5.9731e+00, -3.5727e+00, ..., -9.8509e+00,
-8.8534e+00, -1.9310e+00],
[-7.7427e+00, -7.7870e+00, -7.5648e+00, ..., -9.8459e+00,
-1.0679e+01, -4.5483e+00],
[-1.8372e+00, -3.4307e+00, -3.6353e+00, ..., -3.7106e+00,
-3.7774e+00, -8.8333e-01]],
[[-7.9660e-02, 1.2666e+00, 1.6663e+00, ..., 1.5505e+00,
2.4886e+00, 9.7562e-01],
[-7.4109e-02, 1.7021e-01, 1.1047e+00, ..., 1.5072e+00,
3.1001e+00, 1.1786e-01],
[ 6.6745e-01, 1.4753e+00, 1.5276e+00, ..., 3.1387e-01,
2.1152e+00, -2.2547e-01],
...,
[ 3.8091e-01, 5.2671e-01, 5.0657e-01, ..., 8.7022e-01,
6.7091e-02, -1.3290e+00],
[ 4.7742e-01, 6.2640e-01, -9.7317e-03, ..., -3.7704e-01,
7.5231e-01, -1.2648e+00],
[ 5.7878e-01, 1.3042e+00, 1.7815e+00, ..., 2.9131e-01,
-3.0004e-02, -1.6626e+00]],
...,
[[-5.8743e-01, -1.5907e+00, -2.2739e+00, ..., -4.8659e+00,
-5.1545e+00, -7.5377e-01],
[-2.6924e+00, -2.6134e+00, -1.7177e+00, ..., -3.5012e+00,
-4.8419e+00, -1.8030e+00],
[-1.8726e+00, -9.8206e-01, -1.1801e+00, ..., -4.4864e+00,
-5.0139e+00, -1.0556e+00],
...,
[-2.2004e+00, -1.3617e+00, -1.5273e+00, ..., -5.7665e+00,
-5.1337e+00, -1.9146e+00],
[-3.3022e+00, -3.1317e+00, -2.5569e+00, ..., -4.1917e+00,
-5.6126e+00, -1.4071e+00],
[-1.7134e+00, -2.2743e+00, -2.6352e+00, ..., -2.3377e+00,
-2.2335e+00, 2.6008e-01]],
[[-1.3138e+00, -2.3544e+00, -3.9748e+00, ..., -1.4082e+00,
-5.1177e-01, -5.4675e-01],
[-3.6843e+00, -3.6022e+00, -4.3902e+00, ..., -3.4168e+00,
-1.9430e+00, -3.3254e-01],
[-4.4946e+00, -3.6588e+00, -3.3435e+00, ..., -3.1119e+00,
-1.8742e+00, -5.1782e-01],
...,
[-4.3644e+00, -3.9202e+00, -5.2968e+00, ..., -2.9988e+00,
-9.9170e-01, -5.1854e-01],
[-4.0349e+00, -1.4019e+00, -2.5646e+00, ..., -1.4403e+00,
-5.1661e-01, 1.0938e+00],
[-3.2635e+00, -3.0227e+00, -2.2071e+00, ..., 2.9430e-02,
-6.8946e-01, 6.3406e-01]],
[[ 1.1956e+00, -8.1522e-02, 1.1520e+00, ..., 6.0661e-01,
8.4313e-01, -1.3931e+00],
[ 2.2763e+00, 1.0597e+00, 1.4057e+00, ..., 3.6716e+00,
3.1759e+00, 2.2948e+00],
[ 1.3708e+00, 7.8819e-01, 8.5561e-01, ..., 2.7909e+00,
1.9630e+00, 1.3550e+00],
...,
[ 2.5681e+00, 1.8030e+00, 1.2283e+00, ..., 3.2719e+00,
3.0339e+00, 7.9995e-01],
[ 2.8107e+00, 4.5287e+00, 3.0117e+00, ..., 2.8653e+00,
1.4766e+00, 1.8050e+00],
[ 1.9069e+00, 1.8215e+00, 9.8879e-01, ..., -1.0274e-01,
-4.3184e-01, 7.3084e-01]]],
[[[-1.8181e-01, 8.2885e-01, 6.7791e-01, ..., -2.0482e+00,
-6.9856e-01, -9.1325e-01],
[-1.3806e-01, 6.1047e-01, 4.4873e+00, ..., -3.7695e+00,
-8.7771e-01, -1.4593e+00],
[-6.3807e-01, 1.4818e-01, 9.7313e-01, ..., -1.4928e+00,
3.5293e-01, -1.1380e-01],
...,
[-4.1006e-01, -8.5498e-01, -4.0613e-01, ..., -8.6539e-02,
1.8377e-01, -5.3717e-01],
[-1.8559e+00, -5.0182e-01, 2.3285e-01, ..., 1.8526e-01,
4.0759e-01, -1.1999e+00],
[-1.9505e+00, -3.1501e+00, -2.1499e+00, ..., -3.0789e+00,
-2.8760e+00, -2.7497e+00]],
[[-1.0396e+00, -1.6482e+00, -2.6944e+00, ..., -2.5272e+00,
-3.3094e+00, -1.8933e+00],
[-4.2466e+00, -5.3163e+00, -5.6738e+00, ..., -2.9196e+00,
-3.4972e+00, -3.1881e+00],
[-4.1160e+00, -4.8225e+00, -6.2626e+00, ..., 4.0284e-01,
-1.7181e-01, -2.4128e+00],
...,
[-5.8450e+00, -6.9910e+00, -6.5080e+00, ..., -2.7898e+00,
-3.0446e+00, -2.8081e+00],
[-8.7908e+00, -8.6300e+00, -6.5436e+00, ..., -5.2416e+00,
-5.5508e+00, -5.0669e+00],
[-2.0688e+00, -3.8522e+00, -2.5603e+00, ..., -3.0737e+00,
-2.7926e+00, -1.6091e+00]],
[[-2.6140e-01, 2.1909e+00, 1.6776e+00, ..., 2.2894e+00,
2.0497e+00, 6.9530e-01],
[-4.6720e-01, 4.5180e-01, 2.8993e-01, ..., 4.1836e+00,
3.4442e+00, 1.0750e+00],
[ 3.7857e-01, 7.7928e-01, 4.2799e-01, ..., -4.1710e+00,
2.1873e+00, 2.2350e+00],
...,
[-6.6385e-01, -5.1353e-01, 7.0144e-01, ..., 6.0765e-01,
1.1062e+00, 8.7513e-01],
[-8.3536e-01, 4.0733e-01, 4.6352e-01, ..., 1.1845e+00,
1.0757e+00, 5.2005e-01],
[ 1.6217e-01, -4.3373e-02, -6.3847e-02, ..., 1.7420e+00,
1.6425e+00, 5.3680e-01]],
...,
[[-1.0157e+00, -1.6308e+00, -1.3345e+00, ..., -2.4548e+00,
-2.3027e+00, -7.6569e-01],
[-3.2490e+00, -3.1517e+00, -2.0210e+00, ..., -3.6535e+00,
-4.3340e+00, -3.3215e+00],
[-1.9361e+00, -1.9150e+00, -3.8059e+00, ..., 8.4567e-01,
-1.4362e+00, -1.6602e+00],
...,
[-2.4355e+00, -4.3471e+00, -3.3760e+00, ..., -1.4809e+00,
-2.7142e+00, -1.1906e+00],
[-3.0748e+00, -4.1158e+00, -1.0782e+00, ..., -2.1337e+00,
-3.2739e+00, -1.6834e+00],
[-2.3969e+00, -2.2761e+00, -1.5309e+00, ..., -1.9632e+00,
-2.1876e+00, -5.7533e-01]],
[[-1.5637e+00, -1.7190e+00, -2.2223e+00, ..., -9.5625e-01,
-1.1229e+00, -8.2665e-01],
[-3.5661e+00, -1.6045e+00, -1.7244e+00, ..., -2.7855e+00,
-2.8912e+00, -6.7272e-01],
[-3.6711e+00, -2.2235e+00, -4.7503e+00, ..., 1.9234e+00,
-8.9794e-01, 2.5721e-01],
...,
[-2.5357e+00, -2.7666e+00, -3.5283e+00, ..., -1.0100e+00,
-3.0326e+00, -1.9014e+00],
[-2.3255e+00, 5.1877e-01, -1.1802e+00, ..., 2.2375e-01,
-2.8971e+00, -1.8225e+00],
[-2.7078e+00, -1.4279e+00, -1.5361e+00, ..., -1.3900e+00,
-2.7391e+00, -1.2673e+00]],
[[ 9.6940e-01, -2.0916e-02, 4.6726e-01, ..., -6.6886e-01,
1.3170e-02, -5.4176e-01],
[ 3.5688e+00, 2.7334e+00, 3.7048e+00, ..., 1.9837e+00,
3.4431e+00, 2.3518e+00],
[ 1.4895e+00, 1.2632e+00, 8.3756e-01, ..., -2.0914e+00,
4.9369e-01, 8.4764e-01],
...,
[ 3.4131e+00, 2.5943e+00, 1.4410e+00, ..., 2.4915e+00,
1.1230e+00, 5.7655e-01],
[ 4.1884e+00, 4.2402e+00, 2.5411e+00, ..., 1.0931e+00,
5.2726e-01, 1.1661e+00],
[ 2.4832e+00, 2.7804e+00, 1.6767e+00, ..., -1.2255e-01,
-2.9545e-01, -1.2211e-01]]],
[[[-9.8486e-02, 8.0451e-01, -2.2424e-01, ..., -1.3139e+00,
-1.0351e+00, -8.1840e-01],
[ 8.0910e-01, 3.6947e-01, -1.4043e-01, ..., -1.6510e-01,
2.4660e-01, -9.2437e-01],
[ 1.5457e+00, 4.4797e-02, 4.9418e-01, ..., -1.7708e-01,
2.2902e-02, -1.5357e+00],
...,
[-1.2407e+00, -8.6022e-01, -4.1410e+00, ..., 5.2835e-01,
-2.1100e+00, 5.6329e-01],
[-2.1108e+00, -3.9756e-01, -1.6034e+00, ..., 1.2960e+00,
-5.2097e-01, 7.8374e-01],
[-2.1264e+00, -4.1473e+00, -3.3028e+00, ..., 7.2156e-01,
-1.6627e+00, -1.4837e+00]],
[[-7.0074e-01, -1.4577e+00, -2.6156e+00, ..., -2.8808e+00,
-4.6842e+00, -2.3226e+00],
[-1.9379e+00, -3.5667e+00, -3.9998e+00, ..., -3.1274e+00,
-4.8444e+00, -4.3852e+00],
[-1.7909e+00, -3.5266e+00, -3.8460e+00, ..., -3.7004e+00,
-4.1417e+00, -3.6043e+00],
...,
[-7.2405e+00, -7.0107e+00, -8.6954e+00, ..., -1.1530e+00,
-6.1623e+00, -8.9315e-01],
[-8.9731e+00, -8.8530e+00, -7.2249e+00, ..., -2.2369e+00,
-5.9933e+00, -3.2132e+00],
[-2.2918e+00, -4.6233e+00, -2.4956e+00, ..., -6.5512e-01,
-3.3071e+00, -8.7679e-01]],
[[ 6.7407e-01, 1.1754e+00, 9.7944e-01, ..., 1.4836e+00,
1.9082e+00, 9.4322e-01],
[ 8.5683e-01, 1.5561e+00, 1.7280e+00, ..., 1.0075e+00,
2.3064e+00, 9.3887e-01],
[ 2.9160e-01, 3.1184e+00, 2.6809e+00, ..., 1.1130e+00,
1.9897e+00, 1.7827e+00],
...,
[ 7.5427e-01, 6.4759e-01, -1.0966e+00, ..., 4.5173e-01,
1.0309e+00, -1.6525e+00],
[ 2.2374e-01, -5.2559e-01, -2.2251e+00, ..., -3.0058e-01,
-2.9572e-03, -1.0687e+00],
[ 6.8724e-01, 6.9698e-01, -1.6107e+00, ..., -6.5135e-02,
-4.0071e-01, -1.5562e+00]],
...,
[[-9.7108e-01, -1.2843e+00, -2.0662e+00, ..., -2.5857e+00,
-3.3718e+00, -1.1868e+00],
[-2.3511e+00, -3.2935e+00, -2.7945e+00, ..., -2.7989e+00,
-4.2570e+00, -3.2993e+00],
[-8.9985e-01, -1.6059e+00, -7.1407e-01, ..., -2.5292e+00,
-3.5535e+00, -2.5608e+00],
...,
[-2.0538e+00, -1.5963e+00, -2.7200e+00, ..., -1.8681e+00,
-2.1932e+00, -8.9258e-01],
[-2.6888e+00, -2.5389e+00, -2.0287e+00, ..., -3.1328e+00,
-2.7530e+00, -3.6230e-02],
[-2.4925e+00, -2.4994e+00, -7.0392e-01, ..., -1.8117e+00,
-2.1938e+00, 5.2381e-01]],
[[-1.5376e+00, -2.3465e+00, -3.5098e+00, ..., -2.9612e+00,
-2.8334e+00, -5.4210e-01],
[-4.4153e+00, -4.9471e+00, -4.7808e+00, ..., -3.9735e+00,
-5.6890e+00, -1.6941e+00],
[-4.0039e+00, -4.8787e+00, -3.8137e+00, ..., -2.1317e+00,
-4.1295e+00, -1.5798e+00],
...,
[-3.6211e+00, -3.0826e+00, -2.7731e+00, ..., -4.8444e-01,
2.4849e-01, 3.6443e-02],
[-3.1883e+00, 5.5814e-01, -1.1804e+00, ..., -7.3104e-01,
-5.6887e-02, 1.4022e+00],
[-3.7782e+00, -1.5756e+00, -6.7149e-01, ..., -2.5971e-01,
-5.8875e-01, 8.0464e-01]],
[[ 9.0011e-01, -9.9268e-02, 7.7946e-01, ..., 4.4573e-01,
1.4767e+00, 5.7837e-01],
[ 1.6805e+00, 7.8574e-01, 3.7766e-01, ..., 8.0442e-01,
1.8540e+00, 3.2605e+00],
[ 1.6631e+00, 1.9154e-01, 1.8866e-01, ..., 7.1652e-01,
9.5293e-01, 1.0633e+00],
...,
[ 3.1055e+00, 2.7580e+00, 7.4124e-01, ..., 9.5842e-01,
1.6653e+00, 1.0802e+00],
[ 4.4773e+00, 4.6574e+00, 2.4827e+00, ..., 2.1407e+00,
1.5353e+00, 1.7538e+00],
[ 2.4197e+00, 2.1143e+00, 5.3886e-01, ..., 1.8987e+00,
6.1503e-01, 1.0109e+00]]]], device='cuda:0'),
tensor([[[[-1.3535e-01, 1.3096e+00, 1.5431e+00, ..., -1.0516e+00,
-3.4090e+00, -1.0529e+00],
[ 3.7720e-01, 1.1024e+00, 3.1935e+00, ..., 9.3819e-01,
-1.6319e+00, -8.3201e-01],
[-6.8212e-01, 5.9481e-01, 1.1140e+00, ..., 1.5529e+00,
2.3091e-01, -4.5823e-01],
...,
[-4.7592e-01, -3.5184e-01, 1.7861e+00, ..., -3.0324e+00,
-2.7708e+00, 2.8189e-01],
[-2.1122e+00, -6.4567e-01, 1.2687e+00, ..., -2.7036e+00,
-3.4606e+00, -5.9049e-01],
[-1.7732e+00, -3.0138e+00, -1.4756e+00, ..., -3.6263e+00,
-3.6853e+00, -1.6258e+00]],
[[-8.2729e-01, -1.7499e+00, -2.9635e+00, ..., -5.1537e+00,
-8.9744e+00, -2.0281e+00],
[-3.9803e+00, -4.8813e+00, -6.3559e+00, ..., -5.6027e+00,
-8.9095e+00, -4.2494e+00],
[-3.2002e+00, -4.4285e+00, -9.2814e+00, ..., -6.6494e+00,
-8.7274e+00, -5.7093e+00],
...,
[-5.6454e+00, -6.6481e+00, -6.3873e+00, ..., -6.2082e+00,
-7.9044e+00, -2.5918e+00],
[-8.8898e+00, -8.5331e+00, -7.4340e+00, ..., -9.3386e+00,
-1.0586e+01, -4.2850e+00],
[-2.1275e+00, -3.7969e+00, -2.7502e+00, ..., -3.9466e+00,
-4.1367e+00, -6.1367e-01]],
[[-2.8041e-01, 2.3320e+00, 1.3630e+00, ..., 2.1185e+00,
2.5336e+00, 7.5893e-01],
[-6.8091e-01, 5.5625e-01, -2.7714e-01, ..., 1.5159e+00,
2.2568e+00, -2.9933e-01],
[-1.7439e-01, 1.7342e+00, 5.4328e-01, ..., 1.3300e-01,
1.4422e+00, -4.0232e-01],
...,
[-8.2429e-01, 2.3144e-01, 8.0687e-01, ..., 2.0625e+00,
4.4280e-01, -3.1636e-01],
[-8.5220e-01, 9.0236e-02, 1.1029e+00, ..., -1.2591e+00,
-2.7185e-01, -2.8057e-01],
[ 2.1392e-01, -3.1099e-01, 2.8678e-01, ..., 1.9816e-01,
-4.3741e-02, -1.8936e+00]],
...,
[[-9.2762e-01, -1.5420e+00, -1.2621e-01, ..., -3.4620e+00,
-5.6297e+00, -1.3475e+00],
[-2.8250e+00, -3.3725e+00, -1.2441e+00, ..., -2.6649e+00,
-5.7993e+00, -2.3738e+00],
[-1.8084e+00, -2.7045e+00, -1.6559e+00, ..., -3.2277e+00,
-6.1978e+00, -2.0925e+00],
...,
[-2.0593e+00, -2.9031e+00, -2.5082e-01, ..., -1.3452e+00,
-4.6758e+00, -2.7863e+00],
[-3.0779e+00, -3.8753e+00, -8.9563e-01, ..., -1.5575e+00,
-4.0945e+00, -5.2538e-01],
[-2.2707e+00, -1.8361e+00, -3.4951e-01, ..., -1.2674e+00,
-2.2519e+00, 9.4511e-01]],
[[-1.5312e+00, -1.7632e+00, -1.2940e+00, ..., -2.2189e+00,
-1.0909e+00, 2.3808e-01],
[-3.4009e+00, -2.4668e+00, -2.0789e+00, ..., -2.0213e+00,
-2.9169e+00, -2.7164e-01],
[-3.1493e+00, -3.5904e+00, -2.3081e+00, ..., -3.1087e-01,
-1.4351e+00, -4.8915e-02],
...,
[-3.0189e+00, -2.4549e+00, -3.7614e+00, ..., -2.8206e+00,
-2.1545e+00, 4.8262e-01],
[-2.7179e+00, 3.5899e-01, -1.2625e+00, ..., -1.0474e-01,
-6.7190e-01, 1.4070e+00],
[-2.8805e+00, -1.4148e+00, -4.0725e-01, ..., -4.4849e-01,
-1.0717e+00, 5.8238e-01]],
[[ 7.7379e-01, 1.9013e-01, 7.6531e-01, ..., 1.6651e+00,
1.6554e+00, 9.0223e-02],
[ 3.5935e+00, 2.7586e+00, 3.4367e+00, ..., 3.5445e+00,
2.9068e+00, 3.7069e+00],
[ 1.4975e+00, 8.5185e-01, 2.7134e+00, ..., 4.1306e+00,
3.2390e+00, 2.6562e+00],
...,
[ 3.7656e+00, 2.7933e+00, 2.7412e+00, ..., 1.2439e+00,
2.9126e+00, 1.7476e+00],
[ 4.3608e+00, 4.3468e+00, 3.4332e+00, ..., 2.8737e+00,
1.3753e+00, 2.6934e+00],
[ 2.5887e+00, 2.6679e+00, 1.8846e+00, ..., 4.1552e-01,
-1.5940e-01, 1.2394e+00]]],
[[[-2.2717e-02, 1.4862e+00, 1.1546e-01, ..., 1.4885e-02,
5.1069e-01, -2.2052e-01],
[ 6.5133e-01, -9.0719e-01, 1.1408e-01, ..., 3.4374e-01,
3.9919e-01, -7.0190e-01],
[-5.3625e-01, -3.1766e+00, -2.7425e+00, ..., -2.8392e-01,
-5.3264e-01, -1.2153e+00],
...,
[-3.5413e-01, 5.5581e-03, 4.9376e-01, ..., 6.5620e-01,
5.5493e-01, -3.5403e-01],
[-1.4599e+00, -7.4398e-02, 3.1056e-01, ..., -3.4106e-02,
4.6217e-01, -1.0412e+00],
[-1.2863e+00, -2.4873e+00, -1.7522e+00, ..., -3.0955e+00,
-2.7835e+00, -2.6683e+00]],
[[-1.2741e+00, -2.7595e+00, -5.2049e+00, ..., -1.7598e+00,
-3.5329e+00, -2.3184e+00],
[-4.7922e+00, -7.9025e+00, -4.7226e+00, ..., -2.9149e+00,
-4.1777e+00, -3.5491e+00],
[-3.8433e+00, -7.7459e+00, -4.3319e+00, ..., -3.9755e+00,
-4.2339e+00, -3.0552e+00],
...,
[-4.5379e+00, -5.2197e+00, -4.9202e+00, ..., -2.6490e+00,
-2.8669e+00, -2.2322e+00],
[-7.7263e+00, -7.9836e+00, -6.1696e+00, ..., -5.0322e+00,
-5.1671e+00, -4.9985e+00],
[-1.4881e+00, -3.2939e+00, -1.9961e+00, ..., -3.0041e+00,
-2.7602e+00, -1.5831e+00]],
[[-2.2086e-01, 2.1544e+00, 1.3434e+00, ..., 1.8254e+00,
1.6813e+00, 8.7958e-01],
[-4.7016e-01, -1.0379e-01, 6.0884e-01, ..., 1.1409e+00,
1.3739e+00, 7.5008e-01],
[-2.2901e-01, 1.6872e+00, 1.7062e+00, ..., 1.3222e+00,
1.9789e+00, 1.5233e+00],
...,
[-6.8759e-01, 5.4756e-01, 9.5945e-01, ..., 8.8779e-01,
2.0835e-01, 5.2237e-01],
[-1.2859e+00, -1.3009e-01, 7.0506e-01, ..., 7.7117e-01,
8.3957e-01, 6.2188e-01],
[-3.3520e-01, -3.5731e-01, -1.5616e-01, ..., 1.5392e+00,
1.4672e+00, 5.0490e-01]],
...,
[[-1.1148e+00, -2.4706e+00, -2.4351e+00, ..., -1.4417e+00,
-1.9499e+00, -4.5086e-01],
[-3.5636e+00, -4.5278e+00, -2.7554e+00, ..., -2.7864e+00,
-3.8933e+00, -3.0771e+00],
[-3.0369e+00, -3.5928e+00, -2.6075e+00, ..., -7.9899e-01,
-3.4004e+00, -2.0601e+00],
...,
[-1.1654e+00, -2.5501e+00, -1.1429e+00, ..., -5.8695e-01,
-1.9601e+00, -1.1388e+00],
[-2.1435e+00, -3.8968e+00, -1.2064e+00, ..., -1.3223e+00,
-2.7396e+00, -1.6212e+00],
[-1.4810e+00, -1.5573e+00, -6.5555e-01, ..., -1.8434e+00,
-2.1431e+00, -5.1565e-01]],
[[-1.8079e+00, -1.3958e+00, -2.1784e+00, ..., -3.9494e-01,
-1.3982e+00, -7.9159e-01],
[-4.5895e+00, -1.8429e+00, -2.2530e-01, ..., -2.2676e+00,
-3.9006e+00, -1.8824e+00],
[-4.7468e+00, -2.4062e+00, -1.4150e+00, ..., -7.0627e-01,
-4.0322e+00, -2.7900e+00],
...,
[-2.6647e+00, -2.8461e+00, -3.7182e+00, ..., -1.5264e+00,
-2.5291e+00, -1.7198e+00],
[-2.0824e+00, -2.5668e-01, -1.6463e+00, ..., -6.2177e-01,
-2.8326e+00, -1.7858e+00],
[-2.2312e+00, -1.5596e+00, -1.6127e+00, ..., -1.5175e+00,
-2.6929e+00, -1.1787e+00]],
[[ 1.1386e+00, 6.9660e-01, 1.0514e+00, ..., 1.4263e-01,
1.1849e+00, 5.8384e-01],
[ 4.1389e+00, 4.0950e+00, 2.5046e+00, ..., 8.8292e-01,
1.1243e+00, 2.3658e+00],
[ 1.6859e+00, 1.2235e+00, -2.1214e+00, ..., 9.3809e-01,
2.1005e-01, 6.4663e-01],
...,
[ 2.9633e+00, 2.8874e+00, 2.4963e+00, ..., 4.9023e-01,
7.7034e-01, 5.2702e-01],
[ 3.8885e+00, 4.0929e+00, 2.7530e+00, ..., 1.3902e+00,
9.7980e-01, 1.0917e+00],
[ 2.4244e+00, 2.2788e+00, 1.5411e+00, ..., -1.0528e-01,
-3.1577e-01, -6.1362e-02]]],
[[[ 2.9554e-01, 1.4761e+00, 6.2901e-01, ..., -1.0101e+00,
-3.5109e-01, 6.0220e-01],
[ 4.2503e-01, 2.9112e-01, 5.9376e-01, ..., -8.7901e-01,
-1.3372e+00, 6.4691e-01],
[ 1.8198e-01, 3.5751e-01, 1.2882e+00, ..., 1.3711e+00,
5.2995e-01, 3.5118e-01],
...,
[ 4.3526e-01, 3.4533e-01, 8.7927e-01, ..., -3.4384e-01,
-8.1784e-01, -7.2223e-01],
[-1.0255e+00, -8.6016e-01, -3.1155e+00, ..., -9.9541e-01,
-2.9544e+00, -1.0488e+00],
[-7.2391e-02, -2.1581e+00, -3.7918e+00, ..., -1.4854e+00,
-2.4190e+00, -8.5016e-01]],
[[-7.3829e-01, -2.1925e+00, -4.0013e+00, ..., -3.3131e+00,
-5.4797e+00, -3.7128e-01],
[-4.7307e+00, -6.1955e+00, -5.2012e+00, ..., -4.2106e+00,
-8.5019e+00, -2.1348e+00],
[-3.5983e+00, -5.3166e+00, -6.3811e+00, ..., -1.3209e+00,
-8.0187e+00, -4.3089e+00],
...,
[-4.4428e+00, -5.7167e+00, -5.1230e+00, ..., -3.7611e+00,
-4.8182e+00, -1.9450e+00],
[-7.6358e+00, -8.8910e+00, -8.6056e+00, ..., -6.5440e+00,
-9.0675e+00, -3.9089e+00],
[-1.2413e+00, -2.7089e+00, -3.0167e+00, ..., -3.0806e+00,
-2.9544e+00, -2.2930e-01]],
[[-3.1029e-01, 2.1272e+00, 1.0273e+00, ..., 1.1883e+00,
2.1895e+00, 9.5214e-01],
[-3.4231e-02, 1.0256e+00, 8.1330e-01, ..., 2.5785e+00,
1.8839e+00, -6.0130e-01],
[-3.8821e-02, 1.8889e+00, 1.6323e+00, ..., 3.0366e+00,
3.3079e+00, -2.4216e-01],
...,
[ 4.8848e-02, 1.2946e+00, -2.2474e-01, ..., 2.0459e+00,
1.1383e+00, 4.8737e-01],
[ 3.0733e-01, 6.7765e-01, -2.5746e-01, ..., 1.9948e+00,
1.9918e+00, 4.1212e-01],
[-4.4743e-01, 1.8833e-01, 2.4758e-01, ..., 1.0344e+00,
1.0230e+00, -9.0570e-01]],
...,
[[-8.1652e-01, -1.7986e+00, -1.4451e+00, ..., -2.2483e+00,
-3.4002e+00, -6.5552e-01],
[-3.1408e+00, -4.4947e+00, -2.4387e+00, ..., -4.0853e+00,
-3.8797e+00, -3.2657e+00],
[-1.4082e+00, -2.5907e+00, -1.0123e+00, ..., -2.2691e+00,
-3.1998e+00, -2.4253e+00],
...,
[-8.0252e-01, -6.5468e-01, -5.3562e-01, ..., -9.1312e-01,
-4.0138e+00, -1.6913e+00],
[-2.5998e+00, -2.3939e+00, -1.5208e+00, ..., -3.2248e+00,
-4.9352e+00, -1.3404e+00],
[-7.6035e-01, -4.3394e-01, 1.9655e-01, ..., -1.3633e+00,
-1.1311e+00, 1.3417e+00]],
[[-1.6662e+00, -2.2116e+00, -2.5568e+00, ..., -3.1437e-01,
-1.3146e+00, -1.3130e+00],
[-4.3468e+00, -3.4517e+00, -2.4615e+00, ..., -1.4402e+00,
-1.1087e+00, -1.1941e+00],
[-2.5990e+00, -1.6612e+00, -1.8563e+00, ..., 1.3609e-02,
-4.4381e-02, -6.9647e-01],
...,
[-3.4854e+00, -3.7808e+00, -6.5003e+00, ..., -6.8610e-01,
-2.4043e+00, -1.2773e+00],
[-4.2367e+00, -2.2686e+00, -4.4379e+00, ..., -1.9432e-01,
-1.5358e+00, -4.7898e-01],
[-2.8302e+00, -8.7479e-01, -6.7864e-01, ..., -4.5383e-01,
-2.4352e-01, 3.8492e-02]],
[[ 1.0492e+00, 5.6312e-01, 7.3737e-01, ..., -3.5192e-01,
2.4272e-02, -4.4661e-01],
[ 3.5327e+00, 3.3916e+00, 2.0457e+00, ..., 4.2378e+00,
2.7850e+00, 2.0582e+00],
[ 1.7656e+00, 1.4779e+00, 1.3158e+00, ..., 4.0221e+00,
4.0069e+00, 1.4159e+00],
...,
[ 2.7233e+00, 1.5257e+00, 1.8557e+00, ..., 2.5240e+00,
1.0190e+00, 4.1057e-01],
[ 3.4744e+00, 4.4693e+00, 2.3513e+00, ..., 3.0981e+00,
1.0767e+00, 1.5555e+00],
[ 2.4090e+00, 1.6269e+00, -6.4163e-01, ..., 7.8246e-01,
1.7767e-01, 9.2002e-01]]],
[[[-3.2489e-01, 1.1059e+00, 1.2559e+00, ..., -9.3829e-01,
-3.4857e+00, -1.0597e+00],
[-1.0643e-02, 1.8000e-01, 3.3799e+00, ..., 1.0543e+00,
-1.6498e+00, -7.6386e-01],
[-1.3041e+00, 2.6879e-01, 2.7837e+00, ..., 7.4575e-01,
-2.9916e+00, -1.1868e+00],
...,
[-3.1457e-01, 2.8647e-01, -7.3566e-01, ..., -1.3808e-01,
-4.8377e+00, -2.9324e+00],
[-1.6527e+00, 7.5971e-01, 8.4844e-01, ..., -1.1692e+00,
-5.2466e+00, 2.7965e-01],
[-1.3635e+00, -2.2272e+00, -9.4787e-01, ..., -2.6751e-01,
-8.0374e-01, 2.5702e+00]],
[[-1.5738e+00, -2.1415e+00, -3.3573e+00, ..., -5.2388e+00,
-8.9907e+00, -1.9943e+00],
[-4.7920e+00, -4.4739e+00, -4.4865e+00, ..., -5.7948e+00,
-8.4918e+00, -3.9680e+00],
[-3.6392e+00, -4.0508e+00, -5.9117e+00, ..., -6.1932e+00,
-7.8650e+00, -2.6236e+00],
...,
[-4.5316e+00, -5.0686e+00, -5.5860e+00, ..., -6.7626e-02,
-4.4996e+00, 1.2944e-01],
[-7.6276e+00, -8.3042e+00, -7.3205e+00, ..., -1.5446e+00,
-6.4795e+00, 1.1159e+00],
[-1.3336e+00, -2.9941e+00, -2.4473e+00, ..., 2.2451e+00,
7.4416e-01, 6.4957e-01]],
[[-2.6448e-01, 2.0832e+00, 1.4274e+00, ..., 2.0936e+00,
2.4760e+00, 6.9110e-01],
[-4.6311e-01, -3.5634e-02, 8.0528e-01, ..., 1.5663e+00,
2.3305e+00, -1.6965e-01],
[-1.8320e-01, 1.9537e+00, 2.1862e+00, ..., 9.7539e-03,
1.0233e+00, 6.3196e-02],
...,
[-6.8846e-01, -6.0739e-01, -9.4917e-02, ..., -6.9192e+00,
-7.4652e+00, -4.2477e+00],
[-1.2609e+00, -5.3820e-01, -8.1746e-01, ..., -3.4703e+00,
-4.0758e+00, -2.9264e+00],
[-4.3560e-01, -2.0462e-01, -1.7249e-01, ..., -3.6637e+00,
-1.8585e+00, -2.7829e+00]],
...,
[[-1.4287e+00, -1.8021e+00, -4.7366e-01, ..., -3.4752e+00,
-5.6821e+00, -1.3463e+00],
[-3.6883e+00, -3.3263e+00, -1.1579e+00, ..., -2.4511e+00,
-5.7391e+00, -1.9849e+00],
[-2.6865e+00, -2.1949e+00, -8.6670e-01, ..., -2.5944e+00,
-6.5957e+00, -1.2873e+00],
...,
[-1.2465e+00, -2.3129e+00, -3.1222e-01, ..., -3.0346e+00,
-4.2698e+00, 7.1158e-01],
[-2.2557e+00, -3.8399e+00, -6.2290e-01, ..., -2.8622e+00,
-5.4212e+00, -3.2123e-01],
[-1.4175e+00, -1.5695e+00, 4.3027e-01, ..., 1.0243e+00,
1.2971e+00, 7.3991e-01]],
[[-1.7635e+00, -1.5756e+00, -2.1499e+00, ..., -2.1960e+00,
-1.0502e+00, 3.0516e-01],
[-3.9852e+00, -1.5901e+00, -2.6244e+00, ..., -2.3708e+00,
-2.7625e+00, -9.0247e-02],
[-3.6206e+00, -2.4646e+00, -4.7650e+00, ..., -1.9523e+00,
-2.4695e+00, 1.7336e-01],
...,
[-2.3649e+00, -3.1300e+00, -2.7719e+00, ..., -6.4566e+00,
-3.7609e+00, -1.6227e+00],
[-1.4869e+00, 3.1228e-01, -1.5417e+00, ..., -6.9631e+00,
-4.9718e+00, -2.3405e-02],
[-1.7805e+00, -8.6114e-01, -3.6348e-01, ..., -4.1737e-01,
-2.2037e-01, 1.6025e+00]],
[[ 1.1891e+00, 2.5792e-01, 7.1652e-01, ..., 1.5784e+00,
1.4803e+00, -4.2717e-02],
[ 4.0078e+00, 2.9567e+00, 2.6021e+00, ..., 3.7613e+00,
2.9685e+00, 3.7583e+00],
[ 1.4923e+00, 5.7820e-01, 9.8434e-01, ..., 3.7227e+00,
1.5802e+00, 1.2854e+00],
...,
[ 2.7579e+00, 2.2404e+00, 1.1892e+00, ..., -2.1785e+00,
-5.1333e+00, -4.2320e+00],
[ 3.4347e+00, 4.0465e+00, 2.3686e+00, ..., -2.6613e+00,
-6.2789e+00, -4.3493e-01],
[ 2.2272e+00, 2.5968e+00, 1.4157e+00, ..., -1.6352e+00,
-2.7169e+00, 2.0671e+00]]],
[[[ 2.3024e-01, 1.2957e+00, 1.0757e-01, ..., -4.5842e-01,
2.7042e-01, 6.5520e-01],
[ 4.1720e-01, 5.8848e-02, 1.3140e+00, ..., -2.9983e-01,
-1.9052e-01, 1.1208e+00],
[-5.7970e-01, -1.1991e-01, 1.8526e+00, ..., -2.8387e+00,
-3.6768e+00, -2.1434e+00],
...,
[-6.0783e-01, -1.1885e+00, -1.0518e+00, ..., -5.0213e+00,
-3.4127e+00, -2.6730e+00],
[-1.3848e+00, -1.6761e+00, -2.0192e+00, ..., -2.6843e+00,
-1.5908e+00, -1.5024e+00],
[-4.6211e-01, -1.4057e+00, -2.0457e+00, ..., -4.7621e+00,
-4.5904e+00, -4.1527e+00]],
[[-8.0166e-01, -2.1787e+00, -3.9669e+00, ..., -3.1710e+00,
-4.3875e+00, -3.8192e-01],
[-4.8347e+00, -5.3331e+00, -5.4771e+00, ..., -7.3974e+00,
-8.0247e+00, -2.5226e+00],
[-3.8884e+00, -4.5031e+00, -4.9548e+00, ..., -7.5069e+00,
-8.9564e+00, -2.2125e+00],
...,
[-5.6000e-01, -9.7238e-01, -3.5466e+00, ..., -1.4676e+01,
-1.2050e+01, -6.9392e+00],
[-1.2479e+00, -2.2166e+00, -1.1171e+00, ..., -6.6584e+00,
-5.8933e+00, -5.8186e+00],
[ 6.3707e-01, 1.5153e-01, -1.0522e+00, ..., -4.9761e+00,
-5.0491e+00, -2.9571e+00]],
[[-3.2998e-01, 2.1080e+00, 1.1058e+00, ..., 8.7288e-01,
1.9066e+00, 7.2751e-01],
[-2.0237e-01, 7.7164e-01, 1.4024e+00, ..., 1.7769e-01,
1.5924e-01, -1.1138e+00],
[-8.4348e-02, 1.8825e+00, 2.8093e+00, ..., 1.9107e+00,
3.3220e+00, 1.3817e+00],
...,
[-2.3116e-01, 3.1599e-01, 2.3395e+00, ..., -7.5050e-01,
-1.1663e+00, -1.4872e-01],
[ 1.9222e-01, -2.9090e+00, -3.0152e+00, ..., 7.8262e-02,
-5.4041e-01, 5.7695e-01],
[ 1.1345e+00, -5.3328e-02, -9.6031e-01, ..., -2.2317e-01,
-1.1004e+00, -2.1839e-01]],
...,
[[-8.9411e-01, -1.8095e+00, -1.7112e+00, ..., -2.6137e+00,
-2.7721e+00, -3.0357e-01],
[-3.3679e+00, -3.8588e+00, -2.5099e+00, ..., -4.2806e+00,
-4.3693e+00, -2.5679e+00],
[-1.6516e+00, -1.9416e+00, -1.2427e+00, ..., -4.3610e+00,
-4.7190e+00, -3.5325e+00],
...,
[-2.8275e-01, 5.3994e-01, 4.3514e-01, ..., -1.1220e+01,
-9.6229e+00, -5.2586e+00],
[-1.4912e+00, -6.1171e-01, 2.8654e-01, ..., -1.6548e-03,
-3.3443e+00, -2.5597e+00],
[ 3.3356e-01, 6.9275e-01, -1.4296e-01, ..., -2.7298e+00,
-3.5026e+00, -1.8790e+00]],
[[-1.7137e+00, -2.2702e+00, -2.9344e+00, ..., -1.5934e+00,
-1.6324e+00, -1.5052e+00],
[-4.3711e+00, -3.1917e+00, -3.0008e+00, ..., -2.4199e+00,
-1.9597e+00, -6.2053e-01],
[-3.5476e+00, -2.7613e+00, -4.1025e+00, ..., -1.4222e+00,
-3.9056e+00, -4.3778e-01],
...,
[ 4.7511e-02, -1.2507e+00, -3.6033e+00, ..., -4.0973e+00,
-4.9564e+00, -1.6092e+00],
[-3.1610e-01, 8.7772e-01, 8.1725e-02, ..., -8.1616e-01,
-2.1167e+00, -9.9215e-02],
[-6.8197e-01, -2.5381e-01, -9.1755e-01, ..., -1.1837e+00,
-2.0005e+00, 1.8704e-01]],
[[ 1.0808e+00, 4.8651e-01, 5.4478e-01, ..., -8.2201e-01,
-2.8655e-01, -5.9621e-01],
[ 3.5979e+00, 2.9878e+00, 1.8731e+00, ..., 2.1287e+00,
2.6734e+00, 2.6309e+00],
[ 1.6368e+00, 2.1947e-01, 1.0203e-01, ..., -6.4057e-01,
7.7465e-01, 1.5441e+00],
...,
[ 7.5813e-01, -1.2640e-01, 1.1940e+00, ..., 6.9696e+00,
5.3071e+00, 2.8075e+00],
[-9.1634e-01, -1.2630e+00, -3.4378e+00, ..., 1.0369e+00,
9.5241e-01, 1.2526e+00],
[-2.5922e-01, -1.5440e+00, -2.4668e+00, ..., 5.8630e-01,
3.3161e-01, 7.9102e-01]]]], device='cuda:0'),
tensor([[[[ 1.1050e+00, 1.5605e+00, 1.6187e+00, ..., -1.4452e+00,
-1.1209e+00, -9.2470e-01],
[ 1.5145e+00, 4.2729e-01, -5.5406e+00, ..., -7.5317e-01,
1.7385e-01, -1.2266e+00],
[ 8.4943e-01, -5.1759e+00, -7.0139e+00, ..., -4.4927e-01,
-1.6434e-01, -1.8659e+00],
...,
[-8.8711e-01, -6.7391e-01, -2.5632e-01, ..., -3.8251e-01,
-9.1154e-01, 2.1471e-01],
[-2.0831e+00, -9.7724e-01, -1.1015e+00, ..., 2.4283e+00,
1.6345e+00, 3.5788e-01],
[-2.1897e+00, -3.8748e+00, -2.3322e+00, ..., 2.2517e+00,
-9.5237e-01, -2.1277e+00]],
[[-2.6784e-01, -2.4803e+00, -4.8300e+00, ..., -3.3147e+00,
-4.8868e+00, -2.6507e+00],
[-2.2140e+00, -6.6474e+00, -1.1852e+01, ..., -3.7964e+00,
-5.2946e+00, -5.0379e+00],
[-4.8755e+00, -1.1959e+01, -7.3290e+00, ..., -3.4769e+00,
-4.1485e+00, -3.6430e+00],
...,
[-6.6468e+00, -7.5141e+00, -6.6319e+00, ..., -1.5284e+00,
-6.2674e+00, -1.7360e+00],
[-9.4654e+00, -1.0799e+01, -7.0898e+00, ..., 2.1982e+00,
-3.2661e+00, -4.1549e+00],
[-2.2927e+00, -4.9093e+00, -2.6083e+00, ..., 4.3777e-01,
-2.0945e+00, -1.4932e+00]],
[[ 5.9647e-01, 1.8094e+00, 9.0008e-01, ..., 1.4814e+00,
1.9324e+00, 8.5643e-01],
[-5.0195e-01, 1.7651e-01, 1.7944e+00, ..., 1.2768e+00,
2.5136e+00, 9.8936e-01],
[-1.7496e-01, 3.2703e+00, 3.7093e+00, ..., 1.1415e+00,
2.5336e+00, 2.1522e+00],
...,
[ 6.1899e-01, 1.1851e+00, -1.3181e+00, ..., 4.4309e+00,
4.0778e+00, -9.8789e-01],
[ 2.2113e-01, -3.0763e-01, -2.4198e+00, ..., -3.0506e+00,
2.0501e+00, 2.0772e-01],
[ 8.3038e-01, 3.7890e-01, -2.5878e+00, ..., -2.4661e+00,
1.0251e+00, 3.1427e-02]],
...,
[[ 3.7868e-02, -2.4376e+00, -3.9697e+00, ..., -3.0640e+00,
-3.5673e+00, -1.2678e+00],
[-2.3754e+00, -6.2070e+00, -9.2229e+00, ..., -3.6402e+00,
-4.7415e+00, -3.5597e+00],
[-3.1780e+00, -6.2081e+00, -3.4765e+00, ..., -3.0955e+00,
-3.8373e+00, -2.7600e+00],
...,
[-1.6291e+00, -1.3844e+00, -1.7033e+00, ..., -1.4856e+00,
-2.7496e+00, -2.3261e+00],
[-2.9238e+00, -2.6293e+00, -6.9288e-01, ..., -4.1110e+00,
-4.8030e+00, -9.0811e-01],
[-2.4985e+00, -2.1541e+00, -1.1629e+00, ..., -2.9450e+00,
-5.2345e+00, 2.7292e-01]],
[[-1.5750e+00, -2.5679e+00, -2.5631e+00, ..., -3.2966e+00,
-2.8421e+00, -4.5640e-01],
[-4.4347e+00, -6.6562e+00, -1.1189e+01, ..., -4.0475e+00,
-5.7247e+00, -1.7409e+00],
[-7.7667e+00, -5.2771e+00, -2.6604e+00, ..., -1.5581e+00,
-4.0037e+00, -1.6223e+00],
...,
[-3.2528e+00, -2.3172e+00, -3.9495e+00, ..., -1.8293e+00,
1.4475e+00, 1.8268e+00],
[-3.3117e+00, 6.6935e-01, -4.4880e+00, ..., -1.5565e+00,
1.5837e-01, 2.9629e+00],
[-3.8513e+00, -3.9678e-01, -8.7767e-01, ..., 5.6896e-02,
-3.6920e+00, 1.0898e+00]],
[[ 8.1283e-01, 2.3609e-01, 2.6275e+00, ..., 2.1131e-01,
1.4977e+00, 6.4725e-01],
[ 1.8782e+00, 2.7339e+00, 2.3900e+00, ..., 5.4327e-01,
2.1931e+00, 3.5485e+00],
[ 2.2735e+00, 1.5892e+00, 1.0946e+00, ..., 4.0287e-01,
1.1750e+00, 9.8969e-01],
...,
[ 3.3748e+00, 2.5315e+00, 1.4584e+00, ..., 2.6621e+00,
4.7462e+00, 1.5340e+00],
[ 4.8254e+00, 4.9325e+00, 2.6473e+00, ..., -1.0366e+00,
2.4194e+00, 1.3793e+00],
[ 2.3883e+00, 2.7410e+00, 2.2026e+00, ..., 5.5606e-01,
5.1419e-01, 7.3753e-01]]],
[[[-2.5601e+00, -1.8220e+00, -2.5914e-01, ..., 6.6962e-01,
-3.2214e+00, -2.2141e+00],
[-3.1320e+00, -4.3634e+00, -3.8558e-01, ..., -2.3290e+00,
-2.4722e+00, -1.5287e+00],
[-3.1131e+00, -4.1964e+00, -9.3260e-01, ..., 2.3073e+00,
1.5707e+00, 1.4751e-02],
...,
[-7.1164e-01, -1.0057e+00, -1.1851e+00, ..., -3.1600e-01,
-3.8275e-01, -8.8934e-01],
[-8.9921e-01, -7.3879e-01, -3.0188e+00, ..., 2.3161e+00,
-8.1791e-01, -2.3123e+00],
[-2.6646e+00, -5.4773e+00, -5.8611e+00, ..., -5.5330e-01,
-3.3197e+00, -4.1183e+00]],
[[-4.1711e+00, -2.6079e+00, 8.2938e-01, ..., -4.7088e+00,
-1.0197e+01, -3.7919e+00],
[-6.6518e+00, -4.9631e+00, -3.4665e-01, ..., -5.6510e+00,
-8.2627e+00, -4.3941e+00],
[-4.4905e+00, -6.9825e+00, -3.0293e-01, ..., 4.3960e-01,
-2.9533e+00, -3.2759e+00],
...,
[-6.1819e+00, -6.0292e+00, -6.8472e+00, ..., -3.0863e+00,
-5.8499e+00, -3.7821e+00],
[-7.9929e+00, -8.1179e+00, -9.0978e+00, ..., -3.8997e+00,
-9.2961e+00, -6.7892e+00],
[-3.0642e+00, -4.8886e+00, -5.3112e+00, ..., -2.3193e+00,
-4.1309e+00, -2.3607e+00]],
[[ 9.9396e-01, 2.0549e+00, -2.0987e-01, ..., 2.8882e+00,
2.9184e+00, 4.2734e-01],
[ 1.0672e+00, -6.1520e-01, -1.9904e+00, ..., 5.3953e+00,
5.9040e+00, 9.1236e-01],
[ 9.0540e-01, -1.0607e+00, -3.7782e+00, ..., 6.0120e-01,
5.0584e+00, 1.7642e+00],
...,
[ 4.8768e-01, 3.7012e-01, -6.9082e-01, ..., 1.6950e+00,
-2.6034e-01, -2.7809e-01],
[ 7.2564e-01, -2.7051e-01, 7.4544e-01, ..., 3.5935e+00,
1.3010e+00, 2.3621e-01],
[ 1.4933e+00, 1.5722e+00, 1.9817e+00, ..., 1.6721e+00,
1.7641e+00, 3.2201e-01]],
...,
[[-3.0629e+00, -1.6179e+00, 8.8301e-01, ..., -3.8942e+00,
-5.9215e+00, -2.0712e+00],
[-4.2853e+00, -3.3420e+00, -2.9810e+00, ..., -4.6106e+00,
-6.6008e+00, -2.4590e+00],
[-3.4057e+00, -2.3169e+00, -2.6469e+00, ..., -1.6171e-01,
-4.3480e+00, -1.8240e+00],
...,
[-2.2446e+00, -1.3365e+00, -7.9755e-01, ..., -9.2058e-01,
-3.0289e+00, -1.4154e+00],
[-3.1383e+00, -3.2602e+00, -2.5668e+00, ..., -2.7671e+00,
-4.1977e+00, -2.6438e+00],
[-3.6529e+00, -4.2193e+00, -3.3631e+00, ..., -4.3233e+00,
-2.9187e+00, -1.5463e+00]],
[[-2.0306e+00, -6.5450e-01, -3.1598e+00, ..., -2.7229e+00,
-1.1525e+00, 2.0933e-01],
[-4.7182e+00, 5.9621e-01, -5.3222e+00, ..., -6.5518e+00,
-3.1451e+00, -2.7316e-01],
[-2.4493e+00, 2.7923e+00, -2.0742e+00, ..., -2.9827e+00,
-4.6035e+00, 1.6513e-01],
...,
[-2.8577e+00, -2.6974e+00, -5.0376e+00, ..., -2.4842e-02,
-5.4871e-01, -1.0129e+00],
[-3.8631e+00, -1.6958e+00, -2.5138e+00, ..., -3.0175e+00,
-2.9880e+00, -1.5877e+00],
[-4.8804e+00, -3.8657e+00, -9.4943e-01, ..., -4.0253e+00,
-3.2004e+00, -1.3962e+00]],
[[ 2.1220e+00, -5.8770e-02, -7.7208e-01, ..., 1.8638e+00,
2.2307e+00, -4.5890e-01],
[ 4.3565e+00, 1.7725e+00, -1.7448e+00, ..., 1.9626e+00,
4.6979e+00, 3.9780e+00],
[ 1.6954e+00, 1.9390e+00, -2.8001e+00, ..., 1.3754e+00,
3.8410e+00, 2.9845e+00],
...,
[ 2.3871e+00, 1.2321e+00, 1.8754e+00, ..., 3.6632e+00,
2.0480e+00, 1.2140e+00],
[ 4.1345e+00, 4.5366e+00, 2.0089e+00, ..., 3.7385e+00,
1.9333e+00, 2.1136e+00],
[ 2.1640e+00, 7.4915e-01, -1.1323e+00, ..., -1.3692e-01,
-6.1024e-01, -4.8637e-03]]],
[[[ 9.9997e-02, 1.1023e+00, 6.5457e-01, ..., -4.5292e-01,
-4.4667e-02, 8.1171e-01],
[ 1.2592e-01, -7.9041e-01, 2.6609e-01, ..., -8.3203e-01,
-1.1422e+00, 1.1461e+00],
[-1.6483e-01, -6.7786e-01, 6.5059e-01, ..., -9.9390e-01,
-6.9474e-01, 4.2081e-01],
...,
[ 4.8881e-01, -3.7227e-01, -2.9029e+00, ..., -7.8982e-01,
-7.1169e-01, -1.1236e+00],
[-1.4023e+00, -1.7451e+00, -5.5178e+00, ..., -2.6192e+00,
-2.7681e+00, -1.3343e+00],
[-4.0090e-02, -2.0985e+00, -3.9165e+00, ..., -2.7108e+00,
-2.3648e+00, -9.2562e-01]],
[[-1.7460e+00, -3.0449e+00, -4.3616e+00, ..., -3.3773e+00,
-4.8264e+00, -2.7420e-01],
[-6.2416e+00, -7.4067e+00, -5.5983e+00, ..., -5.2466e+00,
-7.2689e+00, -1.7036e+00],
[-4.6598e+00, -6.3158e+00, -6.6674e+00, ..., -7.7462e+00,
-8.3565e+00, -3.3337e+00],
...,
[-4.6740e+00, -4.2121e+00, -7.7219e+00, ..., -4.6986e+00,
-5.0633e+00, -2.2923e+00],
[-8.2248e+00, -7.7108e+00, -7.0763e+00, ..., -8.6362e+00,
-9.3110e+00, -4.0901e+00],
[-1.0593e+00, -1.7839e+00, -2.4083e+00, ..., -2.7819e+00,
-2.8295e+00, -2.9620e-01]],
[[-3.5718e-01, 1.8833e+00, 1.1452e+00, ..., 9.8931e-01,
2.1123e+00, 9.5146e-01],
[ 2.7681e-01, 6.9752e-01, 7.1970e-01, ..., 5.9457e-01,
1.1610e+00, -5.4806e-01],
[ 4.9251e-01, 1.8119e+00, 1.3988e+00, ..., 2.3812e-02,
1.5359e+00, -1.0050e-01],
...,
[ 1.4433e-01, 4.0285e-02, 1.4178e+00, ..., 1.3912e+00,
1.1153e+00, 5.8792e-01],
[ 2.4733e-01, -8.9786e-02, 1.9521e+00, ..., 1.3440e+00,
1.9320e+00, 6.3282e-01],
[-6.7861e-01, -4.1229e-01, 1.8839e+00, ..., 6.7200e-01,
7.8605e-01, -8.1354e-01]],
...,
[[-1.3250e+00, -2.1165e+00, -2.1968e+00, ..., -2.4357e+00,
-2.7412e+00, -4.3808e-01],
[-4.3356e+00, -4.8701e+00, -3.2100e+00, ..., -3.8219e+00,
-4.0935e+00, -2.5403e+00],
[-2.5834e+00, -2.7234e+00, -1.6948e+00, ..., -3.8946e+00,
-4.7967e+00, -1.5584e+00],
...,
[-5.9818e-01, -1.6903e+00, -2.2020e+00, ..., -1.7697e+00,
-3.7923e+00, -1.8866e+00],
[-2.7534e+00, -3.6282e+00, -3.0530e+00, ..., -3.6498e+00,
-4.7580e+00, -1.4757e+00],
[-4.9143e-01, -6.4307e-01, 8.7394e-03, ..., -6.7440e-01,
-1.0311e+00, 1.2515e+00]],
[[-1.9212e+00, -1.7733e+00, -3.1656e+00, ..., -1.5660e+00,
-1.3099e+00, -1.3247e+00],
[-5.3710e+00, -3.0011e+00, -3.3379e+00, ..., -2.8911e+00,
-1.4963e+00, -6.4544e-01],
[-3.8168e+00, -1.5091e+00, -2.5842e+00, ..., -5.3923e-01,
-5.1705e-01, -3.6127e-01],
...,
[-3.7290e+00, -4.3930e+00, -3.6220e+00, ..., -2.0330e+00,
-2.5411e+00, -1.1921e+00],
[-4.6257e+00, -1.8370e+00, -3.2419e-01, ..., -1.5077e+00,
-1.9930e+00, -5.4768e-01],
[-2.7016e+00, -1.5370e+00, -1.1874e-01, ..., -5.1460e-01,
-7.6350e-01, 7.4692e-02]],
[[ 1.5730e+00, 6.6451e-01, 8.0087e-01, ..., -5.3653e-01,
-1.3213e-03, -4.2092e-01],
[ 4.4022e+00, 3.5467e+00, 2.3141e+00, ..., 1.6969e+00,
1.9749e+00, 2.3626e+00],
[ 2.0657e+00, 1.2759e+00, 1.4601e+00, ..., 2.0377e+00,
2.3435e+00, 1.8378e+00],
...,
[ 2.6992e+00, 1.6900e+00, 1.5960e+00, ..., 1.3602e+00,
1.5180e+00, 5.6892e-01],
[ 3.3975e+00, 3.9047e+00, -8.0102e-02, ..., 2.7650e+00,
1.4812e+00, 1.6323e+00],
[ 2.4001e+00, 1.3512e+00, -3.0803e-01, ..., 3.6704e-01,
5.3751e-02, 9.5386e-01]]],
[[[-7.4859e-02, 7.2711e-01, -2.4899e-01, ..., -1.2552e+00,
-9.5747e-01, -8.0067e-01],
[ 6.8456e-01, 4.2898e-01, -2.4774e-01, ..., -5.6710e-01,
1.6092e-01, -1.0658e+00],
[ 1.1680e+00, -2.5172e-01, 2.5667e-01, ..., -1.2088e+00,
-4.5560e-01, -2.1140e+00],
...,
[-2.4251e-01, -9.1970e-01, 7.8892e-01, ..., -1.7330e+00,
-2.2077e+00, 6.2075e-02],
[-2.1574e+00, -1.5033e+00, 6.1815e-01, ..., -2.3967e+00,
-2.5650e+00, 1.9004e-01],
[-2.3582e+00, -4.0180e+00, -3.1393e+00, ..., -4.1490e+00,
-3.5378e+00, -1.5892e+00]],
[[-7.4182e-01, -1.4732e+00, -2.5269e+00, ..., -3.0026e+00,
-4.5256e+00, -2.4008e+00],
[-1.9397e+00, -3.6741e+00, -4.0271e+00, ..., -3.0865e+00,
-5.0016e+00, -4.6118e+00],
[-1.7572e+00, -3.4249e+00, -3.6562e+00, ..., -2.7423e+00,
-4.1120e+00, -3.8084e+00],
...,
[-6.6092e+00, -8.8237e+00, -7.5248e+00, ..., -7.1455e+00,
-8.3241e+00, -1.7800e+00],
[-9.7617e+00, -1.0253e+01, -8.5261e+00, ..., -1.0526e+01,
-1.0751e+01, -3.7805e+00],
[-2.5623e+00, -5.0523e+00, -4.3274e+00, ..., -4.1485e+00,
-4.2567e+00, -6.8215e-01]],
[[ 6.7162e-01, 1.1995e+00, 1.0257e+00, ..., 1.4975e+00,
1.7893e+00, 8.4720e-01],
[ 8.4372e-01, 1.4350e+00, 1.4366e+00, ..., 1.1916e+00,
2.1467e+00, 8.8804e-01],
[ 5.0382e-01, 3.0975e+00, 2.1882e+00, ..., 1.6998e+00,
2.4872e+00, 1.9707e+00],
...,
[-3.8121e-02, 4.9488e-01, -5.8612e-02, ..., 1.1905e+00,
1.6383e+00, -2.9057e-01],
[-6.0260e-01, 7.5631e-01, 3.6192e-02, ..., 6.3736e-01,
1.2675e+00, -3.8547e-01],
[ 6.0209e-01, 1.1966e+00, 9.4700e-01, ..., -2.7677e-01,
-4.8244e-01, -1.8524e+00]],
...,
[[-1.0119e+00, -1.3150e+00, -1.9877e+00, ..., -2.5235e+00,
-3.2192e+00, -1.2163e+00],
[-2.3737e+00, -3.2400e+00, -2.4976e+00, ..., -2.9409e+00,
-4.1275e+00, -3.5405e+00],
[-9.1093e-01, -1.7184e+00, -5.1853e-01, ..., -3.2223e+00,
-4.0358e+00, -3.1452e+00],
...,
[-2.6639e+00, -4.2996e+00, -1.7434e+00, ..., -5.0965e+00,
-6.3055e+00, -2.4160e+00],
[-3.6943e+00, -5.9393e+00, -3.1316e+00, ..., -4.2264e+00,
-5.1935e+00, -1.2154e+00],
[-3.3141e+00, -3.7692e+00, -3.1633e+00, ..., -2.6759e+00,
-2.6614e+00, 4.1286e-01]],
[[-1.5413e+00, -2.3212e+00, -3.4307e+00, ..., -2.7948e+00,
-2.7109e+00, -5.3686e-01],
[-4.3675e+00, -4.8524e+00, -4.8568e+00, ..., -4.2394e+00,
-5.7866e+00, -1.9677e+00],
[-4.0644e+00, -5.4096e+00, -3.9073e+00, ..., -3.5050e+00,
-5.2706e+00, -2.1364e+00],
...,
[-4.2905e+00, -3.0673e+00, -3.8569e+00, ..., -3.4457e+00,
-2.7381e+00, -1.3291e+00],
[-3.6086e+00, -9.7571e-01, -2.0915e+00, ..., -1.9994e+00,
-1.1189e+00, 3.5065e-01],
[-3.8375e+00, -3.1873e+00, -3.1541e+00, ..., -1.7588e+00,
-1.2562e+00, 3.3913e-01]],
[[ 9.2771e-01, -1.3920e-01, 6.7838e-01, ..., 5.1708e-01,
1.3318e+00, 5.9967e-01],
[ 1.7234e+00, 8.6784e-01, 4.8733e-01, ..., 6.2543e-01,
1.8946e+00, 3.1613e+00],
[ 1.3967e+00, 1.3129e-01, 3.7329e-01, ..., -8.4442e-02,
1.0975e+00, 8.3083e-01],
...,
[ 3.9787e+00, 3.6845e+00, 3.7411e+00, ..., 3.3303e+00,
2.4095e+00, 5.0659e-01],
[ 4.6748e+00, 4.3943e+00, 3.8000e+00, ..., 2.6071e+00,
1.8306e+00, 1.4363e+00],
[ 2.8460e+00, 2.8379e+00, 2.0221e+00, ..., 5.6099e-01,
5.7051e-01, 1.0274e+00]]],
[[[-2.1775e-01, 1.6174e+00, 1.7370e+00, ..., -1.0789e+00,
-3.3392e+00, -9.8299e-01],
[ 1.3588e-01, 2.1525e-01, 1.8891e+00, ..., 7.9979e-01,
-1.4150e+00, -8.0454e-01],
[-1.2580e+00, -3.3010e-01, -1.6103e-01, ..., 2.0695e+00,
-1.2061e+00, -1.3783e+00],
...,
[-2.7475e-01, 8.3020e-03, 9.9625e-01, ..., -1.5688e+00,
-2.0555e+00, 5.3229e-01],
[-1.4917e+00, -1.0499e-02, 7.0941e-01, ..., -1.2178e+00,
-2.7497e+00, -5.0823e-01],
[-1.1652e+00, -2.3358e+00, -1.3002e+00, ..., -3.3003e+00,
-3.5278e+00, -1.6847e+00]],
[[-1.4771e+00, -2.4475e+00, -4.2436e+00, ..., -5.2412e+00,
-8.9640e+00, -1.9825e+00],
[-4.7808e+00, -5.1993e+00, -4.2728e+00, ..., -5.5182e+00,
-9.1761e+00, -4.2928e+00],
[-3.8859e+00, -4.9208e+00, -4.4019e+00, ..., -2.8027e+00,
-7.7121e+00, -3.3461e+00],
...,
[-4.6531e+00, -5.3467e+00, -5.9193e+00, ..., -6.2420e+00,
-8.1756e+00, -2.5843e+00],
[-7.7273e+00, -8.0148e+00, -7.8054e+00, ..., -7.1765e+00,
-1.0034e+01, -4.2438e+00],
[-1.4340e+00, -3.0906e+00, -2.3724e+00, ..., -3.7802e+00,
-4.2245e+00, -5.7165e-01]],
[[-2.0924e-01, 2.4202e+00, 1.1882e+00, ..., 1.9106e+00,
2.3933e+00, 6.7054e-01],
[-5.7238e-01, 6.5073e-01, -3.6774e-01, ..., 1.3241e+00,
2.3360e+00, -4.8679e-01],
[-1.1069e-01, 1.7628e+00, 3.5285e-01, ..., 9.4116e-01,
9.0003e-01, -5.5963e-01],
...,
[-7.4084e-01, 4.1970e-01, 1.4341e+00, ..., 1.9236e+00,
1.2385e+00, -3.6922e-01],
[-1.2152e+00, 7.1872e-02, 1.1216e+00, ..., 6.9267e-02,
9.5534e-01, -3.3209e-01],
[-3.7444e-01, -2.7873e-01, 2.3568e-01, ..., -1.3336e-01,
1.2789e-01, -1.6292e+00]],
...,
[[-1.2797e+00, -1.7026e+00, -5.5884e-01, ..., -3.4708e+00,
-5.6754e+00, -1.1940e+00],
[-3.4671e+00, -3.6322e+00, -1.1544e+00, ..., -2.1419e+00,
-6.1754e+00, -2.5966e+00],
[-2.6312e+00, -2.4894e+00, -1.4461e+00, ..., -2.1030e+00,
-6.3255e+00, -2.1404e+00],
...,
[-9.4692e-01, -2.1224e+00, -6.5553e-01, ..., -3.1630e+00,
-5.9154e+00, -2.7965e+00],
[-2.1387e+00, -3.8878e+00, -8.4363e-01, ..., -1.8710e+00,
-4.3959e+00, -4.7245e-01],
[-1.2915e+00, -1.3706e+00, 1.9352e-01, ..., -1.8114e+00,
-2.4620e+00, 8.1279e-01]],
[[-1.7753e+00, -1.3053e+00, -2.0222e+00, ..., -2.0521e+00,
-9.2476e-01, 1.8472e-01],
[-3.8686e+00, -2.3529e+00, -2.2931e+00, ..., -2.6457e+00,
-3.5670e+00, -5.6628e-01],
[-3.6226e+00, -3.2125e+00, -2.9546e+00, ..., -1.3841e+00,
-2.6960e+00, 1.5968e-01],
...,
[-2.8960e+00, -3.0599e+00, -3.2263e+00, ..., -1.9251e+00,
-1.4367e+00, 3.6040e-01],
[-2.0870e+00, -3.4948e-01, -1.3453e+00, ..., -2.8074e-01,
-1.1391e+00, 8.9632e-01],
[-2.1824e+00, -1.4434e+00, -4.1711e-01, ..., -5.4939e-01,
-1.0433e+00, 5.8747e-01]],
[[ 1.1963e+00, 7.2063e-01, 6.5071e-01, ..., 1.6139e+00,
1.8394e+00, 1.4870e-01],
[ 4.0695e+00, 3.3064e+00, 3.1895e+00, ..., 3.2797e+00,
3.2494e+00, 3.8400e+00],
[ 1.6598e+00, 1.0326e+00, 6.1128e-01, ..., 3.9244e+00,
2.2542e+00, 9.7168e-01],
...,
[ 3.0964e+00, 2.7724e+00, 2.2861e+00, ..., 3.8350e+00,
2.4819e+00, 1.5012e+00],
[ 3.8070e+00, 4.2414e+00, 2.9290e+00, ..., 2.8986e+00,
1.4688e+00, 2.7553e+00],
[ 2.4442e+00, 2.3043e+00, 1.5846e+00, ..., 3.9451e-01,
-2.9914e-01, 1.3055e+00]]]], device='cuda:0'),
tensor([[[[-2.8308e-01, 9.1905e-01, 5.5832e-01, ..., -3.6711e-01,
3.8587e-01, -3.0529e-01],
[ 1.4840e-01, 5.2512e-01, 2.7750e+00, ..., -7.3894e-01,
2.2972e-01, -5.5255e-01],
[-1.0003e+00, 1.4360e-01, 1.0071e+00, ..., -1.3971e+00,
-8.2391e-01, -1.1403e+00],
...,
[-3.0979e-01, -4.3156e-01, 1.2066e-02, ..., -1.5289e+00,
-3.3246e-01, -3.6014e-01],
[-1.5817e+00, -1.9790e-01, -1.7687e-01, ..., 3.2930e-01,
4.3202e-01, -9.6355e-01],
[-1.2870e+00, -2.3955e+00, -1.7956e+00, ..., -3.0373e+00,
-2.8323e+00, -2.6194e+00]],
[[-1.5060e+00, -2.3002e+00, -3.7168e+00, ..., -2.4033e+00,
-3.5959e+00, -2.3811e+00],
[-4.7656e+00, -5.3003e+00, -4.8627e+00, ..., -5.7386e+00,
-4.4184e+00, -3.4225e+00],
[-3.8552e+00, -5.0706e+00, -8.3928e+00, ..., -5.7119e+00,
-5.7209e+00, -2.8871e+00],
...,
[-4.7693e+00, -5.3748e+00, -4.9526e+00, ..., -3.0710e+00,
-2.9421e+00, -2.3901e+00],
[-7.8303e+00, -7.9663e+00, -6.3610e+00, ..., -5.0998e+00,
-5.4176e+00, -4.9209e+00],
[-1.4935e+00, -3.1849e+00, -1.9584e+00, ..., -2.9279e+00,
-2.7784e+00, -1.5064e+00]],
[[-2.8948e-01, 2.2529e+00, 1.5499e+00, ..., 1.5672e+00,
1.7672e+00, 9.1532e-01],
[-5.4336e-01, 3.9068e-01, 5.1229e-01, ..., 7.7413e-01,
1.6678e+00, 8.3536e-01],
[ 6.8891e-02, 1.6043e+00, 8.9022e-01, ..., 1.3964e+00,
1.7396e+00, 1.2997e+00],
...,
[-7.6928e-01, 7.0885e-01, 1.5182e+00, ..., 5.6830e-01,
1.9842e-01, 6.7190e-01],
[-1.1918e+00, -9.2570e-02, 1.2110e+00, ..., 1.0687e+00,
1.0100e+00, 5.1058e-01],
[-2.7908e-01, -4.7049e-01, 9.3872e-02, ..., 1.5540e+00,
1.6453e+00, 4.2530e-01]],
...,
[[-1.2343e+00, -2.1479e+00, -2.2065e+00, ..., -1.9594e+00,
-2.1381e+00, -5.3087e-01],
[-3.3025e+00, -3.7231e+00, -2.2821e+00, ..., -3.6436e+00,
-4.5147e+00, -2.8312e+00],
[-2.3361e+00, -3.2239e+00, -2.9483e+00, ..., -1.6251e+00,
-4.3406e+00, -2.3010e+00],
...,
[-1.1436e+00, -2.7509e+00, -1.2433e+00, ..., -1.9272e+00,
-2.7567e+00, -1.3691e+00],
[-2.2185e+00, -3.9129e+00, -1.7393e+00, ..., -1.8098e+00,
-3.1847e+00, -1.5473e+00],
[-1.4558e+00, -1.4607e+00, -7.6316e-01, ..., -1.8483e+00,
-2.0580e+00, -4.3875e-01]],
[[-1.7175e+00, -1.8859e+00, -2.6145e+00, ..., -6.8384e-01,
-1.3549e+00, -5.8865e-01],
[-3.8822e+00, -2.5384e+00, -2.8164e+00, ..., -1.9949e+00,
-4.3220e+00, -1.8013e+00],
[-3.8992e+00, -3.4897e+00, -3.9161e+00, ..., 1.6926e-01,
-3.4258e+00, -2.6117e+00],
...,
[-2.8727e+00, -2.9602e+00, -3.1542e+00, ..., -6.8847e-01,
-2.2386e+00, -1.5606e+00],
[-2.1061e+00, -3.2017e-01, -1.6108e+00, ..., -8.1008e-02,
-2.7519e+00, -1.6507e+00],
[-2.2197e+00, -1.5659e+00, -1.7228e+00, ..., -1.5264e+00,
-2.7019e+00, -1.1425e+00]],
[[ 1.1318e+00, 2.2852e-01, 8.2498e-01, ..., 9.1753e-02,
1.2946e+00, 6.3567e-01],
[ 3.9475e+00, 2.7656e+00, 3.1529e+00, ..., 6.9492e-01,
9.0632e-01, 2.4484e+00],
[ 1.5958e+00, 8.6585e-01, 2.3567e+00, ..., 1.6255e+00,
3.3394e-01, 6.7659e-01],
...,
[ 3.1487e+00, 2.6808e+00, 2.4327e+00, ..., 1.7788e+00,
1.0817e+00, 4.5898e-01],
[ 3.8064e+00, 4.0881e+00, 2.3853e+00, ..., 9.0371e-01,
4.2485e-01, 1.1151e+00],
[ 2.3952e+00, 2.3342e+00, 1.5541e+00, ..., -1.4795e-01,
-3.4921e-01, -1.0538e-01]]],
[[[ 2.4757e-01, 1.4217e+00, 4.5124e-01, ..., -2.3564e+00,
-4.6897e+00, -3.0134e+00],
[ 8.4976e-01, 1.0340e+00, 9.9792e-01, ..., 6.9323e-01,
-1.0057e+00, 3.0002e+00],
[ 4.3664e-01, -1.4129e+00, -4.1240e+00, ..., -2.6820e+00,
-4.1345e+00, 6.4107e-01],
...,
[-7.9365e-01, -1.9514e+00, 4.6933e-01, ..., -2.9150e+00,
-1.6357e+00, -6.1771e-02],
[-1.8281e+00, -1.7500e+00, -1.2899e+00, ..., -2.8439e+00,
-2.1731e+00, -3.1314e-02],
[-1.1334e+00, -3.2841e+00, -3.4089e+00, ..., -3.7008e+00,
-3.1289e+00, -1.2268e+00]],
[[-8.2114e-01, -1.4846e+00, -3.3224e+00, ..., -4.6015e+00,
-7.8869e+00, -2.3834e+00],
[-3.3035e+00, -5.0603e+00, -4.4938e+00, ..., -3.1604e-01,
-2.1217e+00, 9.5552e-01],
[-3.4466e+00, -7.2996e+00, -6.9091e+00, ..., -4.6654e+00,
-5.3176e+00, 9.6337e-01],
...,
[-5.7349e+00, -5.6594e+00, -3.8699e+00, ..., -7.5026e+00,
-6.1177e+00, -1.6822e+00],
[-7.3785e+00, -7.6161e+00, -6.8170e+00, ..., -8.7053e+00,
-8.1713e+00, -3.4777e+00],
[-1.7228e+00, -3.2538e+00, -3.0583e+00, ..., -4.2324e+00,
-3.4351e+00, -5.2097e-01]],
[[ 3.7290e-02, 1.4916e+00, 1.5533e+00, ..., 1.4396e+00,
3.4934e+00, 3.4845e+00],
[ 2.2764e-02, -2.1317e-01, 1.8246e-01, ..., -3.6809e+00,
-3.4885e+00, 1.2000e+00],
[ 2.2626e-01, 8.7160e-01, 2.5965e+00, ..., -4.1454e-01,
-4.2727e+00, -5.5526e+00],
...,
[ 3.0671e-01, 1.1293e+00, 1.2174e+00, ..., -3.0952e-01,
4.3512e-01, -5.1190e-01],
[ 4.6193e-01, 8.1358e-01, 9.4325e-01, ..., -5.3798e-01,
5.3862e-01, -4.6301e-01],
[ 5.0419e-01, 1.1165e+00, 1.3482e+00, ..., -5.6496e-01,
-1.0434e-02, -1.6067e+00]],
...,
[[-4.7075e-01, -1.6307e+00, -2.1306e+00, ..., -2.3019e+00,
-3.8944e+00, -1.9552e+00],
[-2.2074e+00, -3.2175e+00, -2.0915e+00, ..., 1.2076e-01,
4.6102e-01, 1.0372e+00],
[-1.9143e+00, -3.5639e+00, -4.4438e+00, ..., -4.4934e+00,
-1.7532e+00, -1.6392e-01],
...,
[-1.9343e+00, -1.7572e+00, -8.1552e-01, ..., -3.6881e+00,
-4.8669e+00, -1.8038e+00],
[-3.0915e+00, -3.2474e+00, -1.9761e+00, ..., -2.9770e+00,
-4.4056e+00, -2.5236e-01],
[-1.5212e+00, -1.9288e+00, -1.5984e+00, ..., -1.5842e+00,
-2.0325e+00, 9.7870e-01]],
[[-1.1733e+00, -2.1964e+00, -3.3367e+00, ..., -4.6054e+00,
-6.3587e+00, -1.4173e+00],
[-3.2701e+00, -3.0125e+00, -2.9451e+00, ..., 2.9496e-01,
-4.5647e-01, -2.2771e+00],
[-4.0523e+00, -4.7327e+00, -3.1755e+00, ..., -5.8188e+00,
-2.7597e+00, -1.3172e+00],
...,
[-4.6438e+00, -3.6318e+00, -2.7252e+00, ..., -3.8510e+00,
-3.8397e+00, -9.6568e-01],
[-4.1857e+00, -2.3954e+00, -2.1745e+00, ..., -4.5184e-01,
-1.6334e+00, 3.7248e-01],
[-3.2175e+00, -3.2735e+00, -1.5944e+00, ..., 2.7511e-01,
-5.4498e-01, 5.3096e-01]],
[[ 1.0762e+00, -4.7880e-04, 1.6177e+00, ..., 7.7742e-01,
4.7258e-02, 4.7992e-01],
[ 2.0191e+00, 1.6971e+00, 2.0262e+00, ..., 2.4109e-01,
-2.9985e+00, 2.5554e+00],
[ 1.6979e+00, 2.4419e+00, 3.7004e-01, ..., -6.4123e+00,
-4.8464e+00, -1.1154e+00],
...,
[ 2.3325e+00, 1.5716e+00, 1.7307e+00, ..., -8.5289e-02,
7.4867e-01, 7.2699e-01],
[ 2.9693e+00, 3.8462e+00, 2.4052e+00, ..., 5.2718e-01,
5.9318e-01, 1.3560e+00],
[ 1.9466e+00, 1.5319e+00, 1.6698e-01, ..., -3.5984e-01,
-3.6778e-01, 6.6370e-01]]],
[[[ 3.2766e-01, 1.1795e+00, 4.2358e-01, ..., -1.6536e-01,
-1.0411e+00, -4.0395e-01],
[ 7.3318e-01, 1.1772e+00, 2.3606e+00, ..., -2.7010e+00,
-2.1720e+00, -2.4774e-01],
[-5.2444e-01, -1.3251e+00, -1.2748e+00, ..., -3.2468e+00,
-3.9624e+00, -2.2938e+00],
...,
[-2.1015e-01, -1.6906e+00, -1.5729e+00, ..., -3.8724e+00,
-2.6925e+00, -6.8921e-01],
[-1.9354e+00, -8.0634e-01, -9.5052e-01, ..., -3.4736e+00,
-2.5677e+00, -1.5319e-01],
[-1.3542e+00, -3.2620e+00, -3.2834e+00, ..., -4.0667e+00,
-3.3025e+00, -1.2550e+00]],
[[-9.4514e-01, -1.5292e+00, -2.6124e+00, ..., -5.2726e+00,
-6.9207e+00, -2.6064e+00],
[-4.3567e+00, -4.8916e+00, -3.6007e+00, ..., -5.2923e+00,
-4.9078e+00, -1.4166e+00],
[-5.4979e+00, -6.8082e+00, -4.4894e+00, ..., -6.5134e+00,
-7.8787e+00, -3.0217e+00],
...,
[-6.0770e+00, -4.9284e+00, -4.1080e+00, ..., -9.5839e+00,
-7.1417e+00, -1.9688e+00],
[-7.8390e+00, -8.3293e+00, -6.5982e+00, ..., -9.0175e+00,
-8.5364e+00, -3.6572e+00],
[-1.7627e+00, -3.2565e+00, -3.3240e+00, ..., -4.5449e+00,
-3.5150e+00, -4.5365e-01]],
[[-3.6689e-02, 1.3599e+00, 1.6786e+00, ..., 7.0259e-01,
2.8129e+00, 1.0562e+00],
[-8.2985e-02, 1.8852e-02, 5.5892e-01, ..., 7.9855e-02,
2.5826e+00, 4.1146e-01],
[ 3.7687e-01, 8.3381e-01, 1.3678e+00, ..., 1.3678e+00,
4.0338e+00, 3.3373e+00],
...,
[-2.2341e-01, -4.2095e-01, 2.3010e+00, ..., -1.2986e+00,
-6.8583e-01, -7.4198e-01],
[ 5.1676e-01, -1.3049e-01, -5.6431e-01, ..., -1.2343e+00,
1.0556e-01, -4.9248e-01],
[ 6.3471e-01, 1.3692e+00, 7.9689e-01, ..., -8.8066e-01,
-5.5824e-02, -1.7243e+00]],
...,
[[-5.8428e-01, -1.5314e+00, -1.8717e+00, ..., -3.1363e+00,
-4.8213e+00, -1.1155e+00],
[-2.7506e+00, -2.6910e+00, -1.4036e+00, ..., -3.4922e+00,
-4.2649e+00, -1.6809e+00],
[-2.7214e+00, -1.4584e+00, -1.1460e+00, ..., -3.0691e+00,
-4.1048e+00, -1.4626e+00],
...,
[-1.5994e+00, -7.8699e-01, -1.3063e+00, ..., -3.4847e+00,
-4.8641e+00, -2.1401e+00],
[-3.6012e+00, -2.8952e+00, -7.0871e-01, ..., -1.8096e+00,
-4.5044e+00, -5.0440e-01],
[-1.6895e+00, -2.1157e+00, -1.9565e+00, ..., -1.5394e+00,
-2.2065e+00, 1.1005e+00]],
[[-1.3196e+00, -2.3404e+00, -3.7889e+00, ..., -2.3682e+00,
-1.3827e+00, -1.7713e-01],
[-3.6139e+00, -3.0739e+00, -3.6899e+00, ..., -3.3978e+00,
-2.7099e+00, 2.7706e-01],
[-4.6681e+00, -2.9016e+00, -2.9343e+00, ..., -4.5537e+00,
-6.2788e+00, -4.0389e+00],
...,
[-4.6633e+00, -5.2946e+00, -5.9292e+00, ..., -4.9066e+00,
-4.5823e+00, -9.2010e-01],
[-3.7391e+00, -2.1476e+00, -2.6914e+00, ..., -1.3941e+00,
-1.8861e+00, 4.4398e-01],
[-3.1048e+00, -3.0613e+00, -1.5030e+00, ..., 3.0207e-01,
-4.4280e-01, 6.6584e-01]],
[[ 1.1633e+00, -1.0408e-01, 1.1298e+00, ..., -2.6292e-01,
9.5406e-01, 3.1223e-01],
[ 2.6435e+00, 1.5049e+00, 1.8230e+00, ..., 1.6780e-02,
9.1309e-01, 2.2991e+00],
[ 2.1502e+00, 1.3290e+00, 4.3954e-02, ..., 1.3565e+00,
1.9254e+00, 2.4742e+00],
...,
[ 3.1810e+00, 1.2543e+00, 8.1951e-01, ..., 4.4307e-01,
1.7052e+00, 9.0597e-01],
[ 3.2185e+00, 4.3055e+00, 2.7345e+00, ..., 1.0234e+00,
1.0467e+00, 1.5268e+00],
[ 2.0766e+00, 2.1236e+00, 7.4104e-01, ..., -2.1304e-01,
-4.3823e-01, 7.7989e-01]]],
[[[ 4.1208e-01, -1.8922e-01, -3.6287e+00, ..., 4.1694e-02,
3.7742e-01, -1.6210e-01],
[-4.3432e-01, -4.5160e+00, -5.7086e+00, ..., 4.6509e-01,
7.2250e-01, -5.1856e-01],
[-2.0430e+00, -3.8490e+00, -3.9067e+00, ..., 1.8822e-01,
1.2128e-01, -9.9614e-01],
...,
[-1.7644e-01, -9.3710e-01, 8.4974e-01, ..., -3.7719e+00,
3.4633e+00, 2.5726e+00],
[-1.2066e+00, 4.1674e-01, 9.7668e-02, ..., 1.8925e+00,
3.1135e+00, -1.8851e+00],
[-1.0564e+00, -1.9893e+00, -1.5351e+00, ..., -5.0589e+00,
-5.6453e+00, -4.7182e+00]],
[[ 1.7737e+00, 6.3232e-01, -1.4682e+00, ..., -2.2973e+00,
-3.7546e+00, -2.1882e+00],
[ 1.5733e+00, -1.2066e+00, -5.2814e+00, ..., -2.4437e+00,
-4.1578e+00, -3.5531e+00],
[-4.5466e-01, 1.8884e+00, -2.3065e+00, ..., -1.5893e+00,
-2.9307e+00, -2.9138e+00],
...,
[-4.0317e+00, -5.5072e+00, -5.6772e+00, ..., 1.1769e-01,
1.5393e+00, 5.4586e-01],
[-7.6909e+00, -7.9053e+00, -6.2152e+00, ..., -2.7317e+00,
-8.4421e+00, -9.0241e+00],
[-1.3287e+00, -3.1554e+00, -2.1559e+00, ..., -6.2604e+00,
-5.0518e+00, -3.5373e+00]],
[[-3.5567e-01, -4.4758e-01, -1.0693e+00, ..., 1.6438e+00,
1.7917e+00, 8.9604e-01],
[-1.8652e+00, 6.6682e-01, 4.8261e-02, ..., 9.6173e-01,
1.4825e+00, 7.7535e-01],
[-3.5246e+00, 1.5698e+00, 2.8518e-01, ..., 1.8248e+00,
1.9147e+00, 1.2877e+00],
...,
[-1.5201e+00, -3.3016e-01, 1.2504e-01, ..., -1.4010e+00,
-1.2897e+00, -1.7385e+00],
[-1.7363e+00, -1.1827e+00, 6.9378e-01, ..., -1.9717e+00,
-2.6669e+00, -6.8390e-01],
[-5.7972e-01, -9.0880e-01, -6.4961e-01, ..., -2.0793e+00,
3.6957e-01, 3.0311e-01]],
...,
[[ 1.4025e+00, -4.1656e-01, 1.6868e-01, ..., -1.5701e+00,
-2.0583e+00, -4.4767e-01],
[ 1.2728e-01, -5.4389e+00, -2.9610e+00, ..., -2.4081e+00,
-3.5654e+00, -3.0466e+00],
[-4.6749e-01, -4.0002e+00, -3.0116e-01, ..., -1.5902e+00,
-2.6415e+00, -1.9289e+00],
...,
[-5.4225e-01, -2.6040e+00, -2.3337e+00, ..., -1.9065e+00,
-2.1428e+00, -3.3233e+00],
[-1.8122e+00, -3.7855e+00, -1.9781e+00, ..., -1.3283e+00,
-4.3977e+00, -5.4379e+00],
[-1.2403e+00, -1.2984e+00, -2.2930e-01, ..., -7.2874e+00,
-5.5643e+00, -2.2481e+00]],
[[-2.7214e+00, -4.7064e+00, -2.6403e+00, ..., -5.3419e-01,
-1.3908e+00, -6.3887e-01],
[-3.9952e+00, -1.0781e+01, -7.1656e+00, ..., -1.0979e+00,
-3.5923e+00, -1.8527e+00],
[-1.0246e+00, -8.9632e+00, -8.1782e+00, ..., -1.0919e+00,
-4.0074e+00, -2.4839e+00],
...,
[-2.3811e+00, -2.5062e+00, -7.7049e-01, ..., -2.4523e+00,
2.6756e+00, 1.2206e+00],
[-1.6716e+00, -3.4587e-01, -1.0863e+00, ..., -2.0816e+00,
-3.7754e-01, -1.2673e+00],
[-1.9516e+00, -1.2270e+00, -1.2250e+00, ..., -3.2194e+00,
-2.3162e+00, 1.5507e-01]],
[[-1.0380e-01, 4.8310e-01, -2.7233e+00, ..., 3.5713e-01,
1.2786e+00, 6.7857e-01],
[ 9.8411e-01, -2.7640e+00, -5.3800e+00, ..., 9.3124e-01,
1.2027e+00, 2.5129e+00],
[ 1.5247e+00, -3.5905e+00, -4.0282e+00, ..., 5.5453e-01,
2.9295e-01, 7.4327e-01],
...,
[ 2.7351e+00, 1.0944e+00, 3.8030e+00, ..., -8.3241e+00,
9.3196e-01, 2.8093e+00],
[ 3.7496e+00, 3.7495e+00, 2.8926e+00, ..., 1.0670e+00,
4.4113e+00, 2.2083e+00],
[ 2.4531e+00, 2.4449e+00, 1.5249e+00, ..., 2.6326e+00,
1.2327e+00, 7.7403e-01]]],
[[[ 5.3921e-02, 1.1178e+00, 6.2007e-01, ..., -5.5974e-01,
-1.4653e-01, 8.4641e-01],
[ 1.2376e-01, -6.7121e-01, 3.1036e-01, ..., -1.0773e+00,
-1.2291e+00, 1.2641e+00],
[-8.1452e-01, -9.9621e-01, 5.2207e-01, ..., -1.4509e+00,
-1.6050e+00, 9.4998e-01],
...,
[-9.7316e-01, -2.0444e+00, -2.5712e-01, ..., -1.5870e+00,
-5.4537e-01, -7.1935e-01],
[-1.1291e+00, -1.0429e+00, -4.2793e-01, ..., -9.4520e-01,
-3.0030e-01, -1.6026e+00],
[-2.6648e+00, -5.1102e+00, -4.4659e+00, ..., -4.8153e+00,
-4.1327e+00, -3.8026e+00]],
[[-1.7375e+00, -3.0284e+00, -4.4964e+00, ..., -3.4225e+00,
-4.8208e+00, -1.8724e-01],
[-6.2295e+00, -7.3644e+00, -5.3799e+00, ..., -5.3637e+00,
-6.8856e+00, -1.3754e+00],
[-4.7421e+00, -5.3180e+00, -3.9192e+00, ..., -4.8740e+00,
-6.3346e+00, -1.3594e+00],
...,
[-5.7789e+00, -5.5898e+00, -3.5967e+00, ..., -6.6736e+00,
-5.0883e+00, -3.6499e+00],
[-7.0113e+00, -7.7556e+00, -6.1490e+00, ..., -7.5688e+00,
-7.8596e+00, -6.5111e+00],
[-2.4747e+00, -4.3043e+00, -4.0109e+00, ..., -4.0225e+00,
-4.1654e+00, -2.5468e+00]],
[[-2.9803e-01, 1.9692e+00, 1.1856e+00, ..., 9.2268e-01,
2.1381e+00, 9.6226e-01],
[ 2.1913e-01, 8.7125e-01, 9.6818e-01, ..., 6.6276e-01,
1.0866e+00, -5.3833e-01],
[ 4.5984e-01, 1.9381e+00, 1.3403e+00, ..., 3.6611e-01,
1.5420e+00, 1.4632e-01],
...,
[ 8.6783e-01, 1.2097e+00, 6.6790e-01, ..., 3.2244e-01,
7.3446e-01, 3.5332e-01],
[ 8.6196e-01, 7.7295e-01, 6.3712e-01, ..., 1.2507e-01,
5.7472e-03, 2.8684e-01],
[ 1.3280e+00, 1.5463e+00, 1.2936e+00, ..., 1.3502e+00,
1.1028e+00, 2.3519e-01]],
...,
[[-1.3304e+00, -2.1433e+00, -2.3257e+00, ..., -2.5239e+00,
-2.6448e+00, -4.4409e-01],
[-4.3222e+00, -4.7408e+00, -3.0521e+00, ..., -3.9103e+00,
-3.8135e+00, -2.5288e+00],
[-2.6924e+00, -2.4806e+00, -1.2827e+00, ..., -2.2628e+00,
-3.8269e+00, -1.2245e+00],
...,
[-2.7245e+00, -2.4423e+00, -8.1822e-01, ..., -2.8878e+00,
-3.6091e+00, -1.7264e+00],
[-3.0420e+00, -3.0035e+00, -1.8931e+00, ..., -1.6760e+00,
-4.1811e+00, -2.4026e+00],
[-3.3948e+00, -3.6828e+00, -3.2915e+00, ..., -3.1463e+00,
-3.7723e+00, -1.6553e+00]],
[[-1.9562e+00, -1.8005e+00, -3.1126e+00, ..., -1.5926e+00,
-1.2918e+00, -1.2465e+00],
[-5.2690e+00, -3.1436e+00, -3.5279e+00, ..., -3.0077e+00,
-1.4989e+00, -6.5181e-01],
[-4.5352e+00, -2.6466e+00, -3.2726e+00, ..., -1.6318e+00,
-1.3372e+00, -2.9330e-01],
...,
[-4.2574e+00, -2.6572e+00, -3.0057e+00, ..., -2.2409e+00,
-3.0992e+00, -1.4851e+00],
[-3.6961e+00, -9.8952e-01, -2.3233e+00, ..., -8.7906e-01,
-2.8372e+00, -1.2505e+00],
[-4.3188e+00, -3.0028e+00, -2.9685e+00, ..., -2.0066e+00,
-3.1456e+00, -1.2231e+00]],
[[ 1.5552e+00, 6.3188e-01, 8.8241e-01, ..., -5.3742e-01,
-2.6468e-02, -3.4716e-01],
[ 4.3686e+00, 3.5184e+00, 2.1884e+00, ..., 1.8318e+00,
1.9594e+00, 2.4317e+00],
[ 1.8184e+00, 5.9486e-01, 4.1131e-01, ..., 1.1925e+00,
1.7961e+00, 1.8713e+00],
...,
[ 3.0315e+00, 1.0956e+00, 1.1655e+00, ..., 2.1118e+00,
2.2087e+00, 1.3473e+00],
[ 3.6193e+00, 3.4666e+00, 2.2470e+00, ..., 3.0410e+00,
2.4555e+00, 2.0534e+00],
[ 1.8901e+00, 7.2015e-01, 2.3529e-02, ..., -1.8305e-01,
-3.3313e-01, -2.3578e-01]]]], device='cuda:0'),
tensor([[[[ 5.4246e-02, 1.0500e+00, -4.1766e-01, ..., -8.5316e-01,
-3.5565e+00, -1.4376e+00],
[ 3.0806e-01, 2.4353e-01, 1.2714e+00, ..., 2.8569e-01,
-1.4557e+00, -4.8766e-01],
[ 2.3907e-01, 1.0398e+00, 1.5903e+00, ..., 4.4950e-01,
-7.8862e-01, -1.0196e+00],
...,
[-1.2738e+00, -2.5502e+00, -6.8697e-01, ..., 4.8177e+00,
-1.3178e-01, -2.1066e+00],
[-1.1792e+00, -9.3688e-01, -4.9769e-01, ..., 5.0350e-01,
-1.6850e+00, -2.4743e+00],
[-2.5315e+00, -4.9876e+00, -4.5064e+00, ..., -4.4804e+00,
-4.1600e+00, -4.0456e+00]],
[[-1.4535e+00, -2.5803e+00, -3.7687e+00, ..., -5.6104e+00,
-8.4777e+00, -1.9554e+00],
[-4.6474e+00, -5.7398e+00, -3.8850e+00, ..., -5.3420e+00,
-7.4718e+00, -2.7684e+00],
[-3.5231e+00, -6.2717e+00, -7.4900e+00, ..., -8.7025e+00,
-9.8655e+00, -5.1548e+00],
...,
[-5.6503e+00, -5.6045e+00, -3.2544e+00, ..., -7.8337e-01,
-5.1084e+00, -5.7614e+00],
[-7.0896e+00, -7.7603e+00, -6.4714e+00, ..., -1.0546e+01,
-9.6470e+00, -6.3347e+00],
[-2.4532e+00, -4.2704e+00, -4.1203e+00, ..., -4.1773e+00,
-4.4648e+00, -2.1788e+00]],
[[ 4.4417e-01, 2.1801e+00, 1.8302e+00, ..., 2.6282e+00,
2.8821e+00, 1.0243e+00],
[-2.5623e-01, 6.7639e-01, -1.0193e-01, ..., 2.0158e+00,
2.8045e+00, 2.6885e-01],
[ 5.5993e-01, 1.8745e+00, -6.1021e-01, ..., -1.3657e+00,
1.5621e-01, -4.9713e-01],
...,
[ 3.7902e-01, 8.2845e-01, 5.8275e-01, ..., -1.1608e+00,
3.1233e+00, 1.4778e+00],
[ 9.3925e-01, 7.9051e-01, 6.0099e-01, ..., -1.2663e+00,
1.1085e+00, 1.7688e-01],
[ 1.2622e+00, 1.4963e+00, 1.2839e+00, ..., 1.4552e+00,
-9.2854e-02, -3.5704e-01]],
...,
[[-8.9669e-01, -1.6308e+00, -3.3144e+00, ..., -3.7575e+00,
-5.9129e+00, -1.6951e+00],
[-2.6020e+00, -2.9470e+00, -2.0830e+00, ..., -2.8488e+00,
-5.0699e+00, -2.4097e+00],
[-1.7166e+00, -1.8816e+00, -3.1431e+00, ..., -4.6571e+00,
-6.5108e+00, -2.6945e+00],
...,
[-2.1778e+00, -1.7512e+00, 5.9614e-02, ..., -6.9303e-01,
-5.1073e+00, -1.6542e+00],
[-2.9204e+00, -2.8705e+00, -1.9959e+00, ..., -2.6552e+00,
-5.2496e+00, -2.0263e+00],
[-3.2916e+00, -3.7614e+00, -3.3519e+00, ..., -1.5054e+00,
-3.5440e+00, -2.0886e+00]],
[[-1.5629e+00, -1.4379e+00, -4.4777e+00, ..., -2.0362e+00,
-6.9317e-01, 5.8602e-01],
[-3.9108e+00, -1.5284e+00, -4.9369e+00, ..., -2.4033e+00,
-1.7344e+00, 4.8082e-02],
[-2.7831e+00, -6.4652e-01, -4.1325e+00, ..., -1.0764e+00,
-6.2928e-01, 2.8312e-01],
...,
[-3.4170e+00, -1.9321e+00, -2.7044e+00, ..., -2.4352e+00,
-7.0151e+00, -8.9927e-01],
[-3.6669e+00, -7.0028e-01, -2.3545e+00, ..., -3.5261e+00,
-5.6150e+00, -5.6959e-01],
[-4.3118e+00, -3.0673e+00, -2.9546e+00, ..., -1.4420e+00,
-2.6919e+00, -8.4623e-01]],
[[ 1.4287e+00, 8.0221e-01, 1.3671e+00, ..., 1.8115e+00,
1.7866e+00, -5.8447e-02],
[ 2.9613e+00, 2.7216e+00, 2.4087e+00, ..., 3.0080e+00,
3.3874e+00, 3.8866e+00],
[ 2.0620e+00, 2.3366e+00, 3.2129e+00, ..., 2.7423e+00,
2.5313e+00, 1.6261e+00],
...,
[ 1.9518e+00, 3.7384e-01, 1.1333e-01, ..., 2.4989e+00,
1.1561e+00, 1.4190e+00],
[ 3.6362e+00, 3.8902e+00, 2.6911e+00, ..., 5.4103e+00,
2.4383e+00, 2.9886e+00],
[ 1.9562e+00, 8.1989e-01, 5.9473e-03, ..., 8.6116e-02,
1.0539e+00, 1.2784e+00]]],
[[[-2.1547e-01, 1.1571e+00, 8.3530e-01, ..., 1.1692e-01,
4.5051e-01, -1.5372e-01],
[ 1.0310e+00, -4.7901e-02, 9.4832e-01, ..., 3.3279e-01,
6.4748e-01, -5.4847e-01],
[-1.5922e+00, -3.1290e+00, -3.3974e+00, ..., 4.9309e-02,
-1.6770e-01, -1.0484e+00],
...,
[-2.1955e-01, -4.4169e-01, 3.7519e-02, ..., 8.0087e-01,
2.6368e-01, -6.9023e-01],
[-1.5821e+00, -1.9526e-01, 1.2774e-01, ..., -4.4545e-03,
2.6044e-01, -1.0165e+00],
[-1.2907e+00, -2.4719e+00, -1.9555e+00, ..., -3.0501e+00,
-2.8624e+00, -2.7909e+00]],
[[-1.3631e+00, -2.1747e+00, -3.9555e+00, ..., -2.0317e+00,
-3.6083e+00, -2.2147e+00],
[-5.0488e+00, -5.8507e+00, -4.1227e+00, ..., -1.9951e+00,
-3.9523e+00, -3.4795e+00],
[-5.5342e+00, -7.6546e+00, -7.3080e+00, ..., -5.0433e+00,
-3.6528e+00, -2.7982e+00],
...,
[-4.7491e+00, -5.4256e+00, -4.8532e+00, ..., -2.4300e+00,
-3.3606e+00, -2.7467e+00],
[-7.7561e+00, -7.8999e+00, -6.2873e+00, ..., -5.2527e+00,
-5.5606e+00, -5.1254e+00],
[-1.5111e+00, -3.2111e+00, -2.2482e+00, ..., -2.9375e+00,
-2.8259e+00, -1.6615e+00]],
[[-1.0641e-01, 2.3288e+00, 1.7250e+00, ..., 1.6137e+00,
1.7332e+00, 8.6818e-01],
[-6.6724e-01, -1.0570e-01, 2.3712e-01, ..., 8.7549e-01,
1.4774e+00, 8.3329e-01],
[-1.2921e-01, 1.9776e+00, 3.0969e+00, ..., 1.5911e+00,
2.0767e+00, 1.3401e+00],
...,
[-8.5216e-01, 6.1872e-01, 1.7778e+00, ..., 1.1830e+00,
4.2774e-01, 5.0657e-01],
[-1.2247e+00, -2.7905e-01, 8.3068e-01, ..., 4.3799e-01,
5.7827e-01, 7.9233e-01],
[-2.9331e-01, -4.5603e-01, -3.8106e-02, ..., 1.5759e+00,
1.5975e+00, 5.7025e-01]],
...,
[[-1.1764e+00, -2.1447e+00, -2.3401e+00, ..., -1.4150e+00,
-1.9875e+00, -4.3927e-01],
[-3.0814e+00, -3.7179e+00, -2.5332e+00, ..., -2.4340e+00,
-3.5516e+00, -2.9952e+00],
[-3.3902e+00, -3.6982e+00, -3.2633e+00, ..., -2.5508e+00,
-3.1186e+00, -1.9682e+00],
...,
[-1.0988e+00, -2.5588e+00, -1.3048e+00, ..., -2.8266e-01,
-2.0166e+00, -1.4391e+00],
[-2.1137e+00, -3.7696e+00, -1.4527e+00, ..., -1.9911e+00,
-3.3243e+00, -1.9499e+00],
[-1.4213e+00, -1.5843e+00, -1.0358e+00, ..., -1.9338e+00,
-2.2320e+00, -5.9665e-01]],
[[-1.7216e+00, -1.7981e+00, -2.5478e+00, ..., -5.7024e-01,
-1.4381e+00, -6.4409e-01],
[-3.8831e+00, -2.6938e+00, -2.5727e+00, ..., -1.4016e+00,
-3.6274e+00, -1.8144e+00],
[-4.3994e+00, -3.5771e+00, -5.8219e+00, ..., -1.7428e+00,
-4.4925e+00, -2.6209e+00],
...,
[-2.8714e+00, -3.0764e+00, -3.5592e+00, ..., -1.4921e+00,
-2.8422e+00, -1.4195e+00],
[-2.1714e+00, -2.3969e-01, -1.4348e+00, ..., -5.6059e-01,
-2.4924e+00, -1.1648e+00],
[-2.2496e+00, -1.7297e+00, -1.6967e+00, ..., -1.6261e+00,
-2.7567e+00, -1.2381e+00]],
[[ 1.1171e+00, 3.8083e-01, 9.1726e-01, ..., 2.5589e-01,
1.2811e+00, 6.8121e-01],
[ 4.2557e+00, 3.0015e+00, 2.4727e+00, ..., 5.9921e-01,
1.0497e+00, 2.4322e+00],
[ 2.0694e+00, 8.2870e-03, 4.4500e-01, ..., 1.1243e+00,
4.2005e-01, 7.8103e-01],
...,
[ 3.1681e+00, 2.5807e+00, 2.5091e+00, ..., 3.0486e-01,
1.2002e+00, 1.5916e+00],
[ 3.7665e+00, 4.1047e+00, 2.6531e+00, ..., 1.2357e+00,
7.2380e-01, 9.0944e-01],
[ 2.2915e+00, 2.2180e+00, 1.4727e+00, ..., -1.0841e-01,
-3.0433e-01, -1.4028e-01]]],
[[[-5.0197e-01, -1.7927e+00, -3.4144e+00, ..., -1.4552e-01,
-5.9573e-03, -2.2787e-01],
[-1.6900e+00, -1.7050e+00, -6.1719e-02, ..., -1.9009e+00,
5.4665e-01, 2.0169e+00],
[-2.4344e+00, -4.7246e-01, -7.7778e-01, ..., -2.9856e+00,
-1.9187e+00, 1.2752e+00],
...,
[-1.5552e+00, -4.9851e+00, -2.0882e+00, ..., 1.6041e+00,
-7.7385e-01, -1.8110e+00],
[-2.6064e+00, -2.8869e+00, -1.0196e+00, ..., 3.3161e+00,
-1.7469e+00, -9.8900e-01],
[-1.6344e+00, -3.1653e+00, -2.6466e+00, ..., -1.3244e+00,
-2.4610e+00, -1.9689e+00]],
[[-2.5927e+00, -3.4492e+00, -4.9650e+00, ..., 4.7912e-01,
-1.1359e+00, -9.5445e-01],
[-6.3164e+00, -4.1306e+00, -1.3214e+00, ..., -4.1333e-01,
8.0480e-02, 1.2400e+00],
[-1.9968e+00, -3.9388e-01, -6.4225e-01, ..., -3.2542e+00,
-2.5247e+00, 1.5795e+00],
...,
[-6.7352e+00, -8.8054e+00, -5.8619e+00, ..., -1.1794e+00,
-6.7512e+00, -2.3534e+00],
[-9.3458e+00, -1.0796e+01, -1.0206e+01, ..., -5.0053e+00,
-1.0341e+01, -4.2512e+00],
[-2.0663e+00, -3.7651e+00, -3.7756e+00, ..., -4.0654e+00,
-5.0074e+00, -7.6748e-01]],
[[ 3.8949e-01, 2.4383e+00, 3.0355e+00, ..., -2.5661e-01,
1.3010e+00, 2.2096e+00],
[ 9.6828e-01, -7.0163e-01, 9.5811e-01, ..., -1.2606e+00,
-3.8969e+00, 9.5097e-01],
[ 1.5855e+00, -4.6319e-01, -1.4488e+00, ..., -1.8097e-01,
-5.9574e-01, -1.7725e+00],
...,
[ 8.3432e-01, 1.2809e+00, 1.4487e-01, ..., 1.4871e+00,
6.8677e-01, -8.4372e-01],
[-7.3005e-01, -3.3386e-01, -1.6539e+00, ..., 8.1620e-01,
1.9087e+00, -6.5819e-01],
[ 1.4828e-01, -1.5461e-01, 4.0465e-01, ..., -2.1112e-01,
-9.4893e-02, -1.5856e+00]],
...,
[[-1.7646e+00, -2.4534e+00, -3.4409e+00, ..., 9.1011e-01,
-2.5882e-01, -1.1180e+00],
[-4.3866e+00, -1.6104e+00, 4.2555e-01, ..., -7.0247e-01,
-2.9626e-01, -7.5812e-01],
[-3.3392e+00, -9.2230e-01, 3.5448e-01, ..., -1.9796e+00,
-8.7733e-01, -1.6301e+00],
...,
[-2.5894e+00, -5.2666e+00, -2.5648e+00, ..., -6.3403e+00,
-7.0209e+00, -2.8342e+00],
[-2.8698e+00, -5.7064e+00, -3.5043e+00, ..., -5.6608e+00,
-6.5395e+00, -1.5628e+00],
[-2.5043e+00, -3.3602e+00, -3.7499e+00, ..., -3.6720e+00,
-2.8521e+00, 3.8749e-01]],
[[-1.5751e+00, -2.3048e+00, -3.5357e+00, ..., -5.9652e-02,
-2.5424e+00, 3.6783e-01],
[-3.0657e+00, 1.6531e-02, -2.3379e+00, ..., -2.5027e-01,
-6.6469e-01, -6.2242e-02],
[-1.8253e+00, 9.0853e-01, 3.1703e-01, ..., -1.8889e+00,
-1.6332e+00, -7.1991e-01],
...,
[-4.9749e+00, -3.0046e+00, -1.2127e+00, ..., -3.3895e+00,
-1.5912e+00, 1.8867e-02],
[-3.2975e+00, -1.0164e+00, 1.3806e+00, ..., -2.6921e+00,
-1.6502e+00, 3.6728e-01],
[-3.0842e+00, -2.4376e+00, -1.4436e+00, ..., -1.9941e+00,
-6.3636e-01, 5.5873e-01]],
[[ 1.8113e+00, -9.8184e-01, -1.0185e-01, ..., -6.9476e-01,
8.6684e-01, 2.0323e+00],
[ 3.8947e+00, 1.7088e+00, 1.6112e-01, ..., -4.1242e+00,
-1.2440e+00, 2.2717e+00],
[ 8.6987e-01, -4.9484e-01, -2.9082e+00, ..., -4.5954e+00,
-2.9912e+00, 7.1581e-03],
...,
[ 2.4968e+00, 2.7368e+00, 1.8084e+00, ..., 2.5992e+00,
2.4618e+00, -3.8200e-01],
[ 3.9314e+00, 3.6439e+00, 4.0301e+00, ..., 3.4149e+00,
1.7528e+00, 1.4843e+00],
[ 2.8184e+00, 2.2017e+00, 1.0760e+00, ..., 1.7533e+00,
6.9744e-01, 1.2387e+00]]],
[[[-2.6863e-01, 1.1278e+00, 1.6218e+00, ..., 2.9893e-01,
1.5601e-01, -9.4480e-02],
[ 2.2552e-01, 2.7249e-02, 3.2960e+00, ..., 5.8214e-01,
4.8013e-02, -3.1716e-01],
[-1.1912e+00, -6.1144e-01, 2.2644e+00, ..., -4.9056e-01,
-6.4839e-01, -7.6517e-01],
...,
[-1.7793e-01, 3.7076e-01, -2.1783e+00, ..., -8.9191e-01,
5.8434e-01, -2.0715e-01],
[-1.4236e+00, 3.0970e-01, -2.4951e+00, ..., -4.3191e-01,
3.5523e-01, -9.8547e-01],
[-1.1644e+00, -2.1893e+00, -2.1397e+00, ..., -2.9972e+00,
-2.7679e+00, -2.6150e+00]],
[[-1.6014e+00, -2.2814e+00, -3.3887e+00, ..., -2.8096e+00,
-3.3006e+00, -1.9485e+00],
[-4.7355e+00, -4.8425e+00, -3.4826e+00, ..., -5.5732e+00,
-3.9687e+00, -3.1664e+00],
[-4.0652e+00, -4.5942e+00, -3.6163e+00, ..., -5.6119e+00,
-2.6921e+00, -2.4960e+00],
...,
[-4.5767e+00, -6.0175e+00, -6.3168e+00, ..., -2.8899e+00,
-2.8494e+00, -2.3143e+00],
[-7.7434e+00, -8.4002e+00, -7.7383e+00, ..., -4.4657e+00,
-5.0518e+00, -4.7350e+00],
[-1.4127e+00, -3.2611e+00, -2.0406e+00, ..., -2.8014e+00,
-2.7501e+00, -1.5265e+00]],
[[-1.7454e-01, 2.2557e+00, 1.7098e+00, ..., 1.3901e+00,
1.3002e+00, 9.1537e-01],
[-5.3869e-01, -3.2217e-02, 7.3027e-01, ..., 9.2018e-01,
9.1265e-01, 8.0140e-01],
[-1.1748e-01, 1.7367e+00, 1.3526e+00, ..., 1.3413e+00,
1.3388e+00, 1.4032e+00],
...,
[-7.0892e-01, 9.6514e-01, 5.5179e-01, ..., 1.2083e+00,
9.2854e-01, 6.0173e-01],
[-1.2527e+00, 3.0420e-01, 1.4828e-01, ..., 1.3661e+00,
9.0989e-01, 3.3247e-01],
[-3.8874e-01, -4.1860e-01, -9.8051e-01, ..., 1.9710e+00,
1.6305e+00, 4.7863e-01]],
...,
[[-1.3767e+00, -1.9867e+00, -1.0288e+00, ..., -1.0662e+00,
-2.2830e+00, -4.3818e-01],
[-3.5457e+00, -3.5997e+00, -1.1535e+00, ..., -1.2660e+00,
-4.1381e+00, -2.9011e+00],
[-2.8537e+00, -2.5379e+00, -1.7216e-01, ..., -9.6137e-02,
-3.4351e+00, -1.8105e+00],
...,
[-8.8012e-01, -2.0480e+00, -2.6811e+00, ..., -1.9908e+00,
-1.8520e+00, -7.3257e-01],
[-2.0709e+00, -3.6953e+00, -2.4930e+00, ..., -1.7798e+00,
-2.6617e+00, -1.4110e+00],
[-1.2798e+00, -1.1776e+00, -1.1062e+00, ..., -2.1212e+00,
-2.0725e+00, -4.7979e-01]],
[[-1.8727e+00, -1.6247e+00, -2.7443e+00, ..., -2.8262e-01,
-1.5737e+00, -6.4876e-01],
[-4.1342e+00, -1.6235e+00, -2.5161e+00, ..., -1.1697e+00,
-3.8930e+00, -1.7981e+00],
[-3.9195e+00, -2.6076e+00, -3.6762e+00, ..., -2.6159e+00,
-4.7386e+00, -2.5072e+00],
...,
[-2.8565e+00, -3.3450e+00, -5.4106e+00, ..., -1.0785e+00,
-3.3746e+00, -2.1747e+00],
[-2.0390e+00, -5.0458e-01, -2.3224e+00, ..., -7.0160e-01,
-3.2825e+00, -1.8968e+00],
[-2.1428e+00, -1.3755e+00, -1.9878e+00, ..., -1.1501e+00,
-2.7051e+00, -1.1851e+00]],
[[ 1.2329e+00, 3.7641e-01, 7.8417e-01, ..., 1.5461e-01,
7.6378e-01, 5.8748e-01],
[ 4.0923e+00, 3.0550e+00, 2.5107e+00, ..., 6.9806e-01,
8.6138e-01, 2.3974e+00],
[ 1.7345e+00, 2.7948e-01, 7.5386e-01, ..., -9.3144e-01,
-4.0516e-02, 6.1410e-01],
...,
[ 3.0727e+00, 2.7076e+00, 2.6970e+00, ..., 5.5266e-01,
2.5523e-01, 5.2329e-01],
[ 3.8145e+00, 3.9406e+00, 2.3021e+00, ..., 1.2167e+00,
7.9113e-01, 1.1602e+00],
[ 2.4059e+00, 2.2478e+00, 1.0180e+00, ..., 8.1731e-02,
-3.1679e-01, -7.9319e-02]]],
[[[ 2.8638e-01, 1.3826e+00, 3.4868e-01, ..., -8.1387e-01,
-2.0405e-02, 6.4983e-01],
[ 3.0937e-01, -3.8967e-03, 4.2271e-01, ..., -2.4615e+00,
-1.3022e+00, 1.0704e+00],
[-5.3405e-01, 5.6982e-02, 8.3952e-01, ..., -2.1220e+00,
-1.0291e+00, 6.7800e-01],
...,
[-6.5184e-01, -1.6589e+00, -1.3052e-01, ..., -2.0579e+00,
-5.2399e-01, -5.1525e-01],
[-9.7231e-01, -8.5116e-01, -1.4270e+00, ..., -1.7343e+00,
-5.2242e-01, -1.5962e+00],
[-2.7640e+00, -5.5428e+00, -4.3882e+00, ..., -5.2161e+00,
-4.3801e+00, -3.8463e+00]],
[[-8.4931e-01, -2.3312e+00, -4.2238e+00, ..., -3.1869e+00,
-4.6992e+00, -3.6141e-01],
[-4.8533e+00, -6.0973e+00, -5.0078e+00, ..., -6.8537e+00,
-6.9777e+00, -1.6601e+00],
[-3.5076e+00, -4.1901e+00, -4.0229e+00, ..., -4.7645e+00,
-5.7800e+00, -1.1783e+00],
...,
[-5.8517e+00, -6.5070e+00, -4.4449e+00, ..., -6.0838e+00,
-5.1653e+00, -3.7826e+00],
[-7.4771e+00, -9.0997e+00, -7.1600e+00, ..., -8.3248e+00,
-8.2129e+00, -6.5946e+00],
[-2.5995e+00, -5.0312e+00, -3.0587e+00, ..., -4.0715e+00,
-4.0817e+00, -2.6093e+00]],
[[-2.8507e-01, 2.0845e+00, 1.2179e+00, ..., 8.6483e-01,
2.1577e+00, 9.3172e-01],
[-5.3354e-02, 1.0732e+00, 9.1705e-01, ..., 5.0655e-01,
1.2405e+00, -3.7275e-01],
[ 6.2363e-02, 2.0271e+00, 1.5791e+00, ..., 1.4635e-01,
1.5548e+00, 1.4463e-01],
...,
[ 9.8456e-01, 1.6783e+00, 2.6801e-01, ..., 2.1981e-01,
1.1283e+00, 3.0354e-01],
[ 7.6985e-01, 1.2103e+00, 2.7622e-01, ..., 1.1775e+00,
6.5682e-01, 1.5553e-01],
[ 1.4003e+00, 2.1928e+00, 7.8694e-01, ..., 2.0506e+00,
1.3704e+00, 1.2919e-01]],
...,
[[-8.9486e-01, -1.8540e+00, -1.8938e+00, ..., -2.6309e+00,
-2.6454e+00, -4.0846e-01],
[-3.3547e+00, -4.2623e+00, -2.5585e+00, ..., -4.5219e+00,
-4.4451e+00, -2.5418e+00],
[-1.4413e+00, -2.0262e+00, -9.0883e-01, ..., -1.6561e+00,
-4.8805e+00, -1.5605e+00],
...,
[-3.0538e+00, -2.6891e+00, -1.0264e+00, ..., -2.8273e+00,
-3.9874e+00, -1.7014e+00],
[-3.4269e+00, -3.6988e+00, -2.7555e+00, ..., -2.3612e+00,
-4.4366e+00, -2.3034e+00],
[-3.7149e+00, -3.9493e+00, -2.9539e+00, ..., -2.9968e+00,
-3.8286e+00, -1.6237e+00]],
[[-1.7275e+00, -2.1385e+00, -2.9733e+00, ..., -1.6602e+00,
-1.4322e+00, -1.3226e+00],
[-4.4914e+00, -3.4040e+00, -2.7077e+00, ..., -2.8742e+00,
-1.6827e+00, -6.4373e-01],
[-3.4701e+00, -2.8321e+00, -2.8255e+00, ..., 8.7476e-01,
-4.3774e-01, -1.9880e-01],
...,
[-3.7882e+00, -2.6281e+00, -4.1496e+00, ..., -7.6290e-01,
-2.3812e+00, -1.5955e+00],
[-3.6757e+00, -1.2387e+00, -2.6478e+00, ..., -1.4127e+00,
-3.1235e+00, -1.2751e+00],
[-4.4959e+00, -3.1721e+00, -2.5016e+00, ..., -2.1553e+00,
-3.2610e+00, -1.2297e+00]],
[[ 1.1201e+00, 6.0161e-01, 5.8624e-01, ..., -8.1970e-01,
9.7444e-02, -4.3123e-01],
[ 3.5054e+00, 3.2330e+00, 2.0926e+00, ..., 1.0881e+00,
1.7739e+00, 2.4889e+00],
[ 1.3899e+00, 6.9367e-01, 2.2133e-01, ..., 6.4947e-01,
1.4697e+00, 1.5990e+00],
...,
[ 2.9778e+00, 1.6733e+00, 2.3095e+00, ..., 2.3536e+00,
1.7895e+00, 1.1806e+00],
[ 4.0977e+00, 3.9496e+00, 1.8922e+00, ..., 3.0877e+00,
2.2258e+00, 2.0871e+00],
[ 2.1188e+00, 7.1564e-01, -5.8941e-02, ..., -4.0519e-01,
-3.8811e-01, -2.1230e-01]]]], device='cuda:0'),
tensor([[[[-2.1939e+00, -4.2431e+00, -7.4999e-01, ..., 1.0554e+00,
-9.9348e-01, -1.9057e+00],
[-3.0547e+00, -5.1420e+00, -2.7826e-01, ..., 1.7469e+00,
-6.8295e-03, -1.3595e+00],
[-4.1132e+00, -4.7830e+00, -1.7815e+00, ..., 3.3878e-01,
-1.9800e+00, -2.0929e+00],
...,
[-9.7250e-01, -2.2423e+00, -7.2479e-01, ..., 2.9005e+00,
1.8544e+00, -6.1106e-01],
[-9.5655e-01, -8.8002e-01, -1.8355e-01, ..., -3.3487e+00,
-3.1152e+00, -2.4328e+00],
[-2.8214e+00, -5.1678e+00, -4.2148e+00, ..., -4.7163e+00,
-4.8504e+00, -4.6805e+00]],
[[-6.8080e+00, -6.1145e+00, 1.6567e+00, ..., -2.1157e+00,
-7.8836e+00, -2.8156e+00],
[-6.7951e+00, -4.3100e+00, 1.8357e+00, ..., -2.8881e+00,
-6.0141e+00, -3.3202e+00],
[-6.3403e+00, -7.8331e+00, -2.2068e+00, ..., -7.6174e+00,
-1.0950e+01, -5.6144e+00],
...,
[-5.7994e+00, -5.9261e+00, -3.2198e+00, ..., -3.9812e+00,
-6.3095e+00, -4.9612e+00],
[-7.5006e+00, -8.4331e+00, -5.6804e+00, ..., -1.0983e+01,
-9.4262e+00, -6.2628e+00],
[-2.6170e+00, -4.7713e+00, -4.1686e+00, ..., -4.3066e+00,
-4.4419e+00, -2.8798e+00]],
[[-1.9473e-01, 1.8017e+00, -8.0713e-01, ..., 2.9501e+00,
2.9998e+00, 5.3807e-01],
[-5.5742e-01, -3.0400e+00, -5.3396e+00, ..., -6.4901e-01,
2.8451e+00, 6.8404e-01],
[ 1.0887e-01, -3.4783e+00, -5.5345e+00, ..., -1.3872e+00,
-1.2343e-01, -9.2280e-01],
...,
[ 2.8481e-01, 1.3203e+00, -5.5585e-01, ..., -2.2750e+00,
-1.3622e+00, 8.4255e-01],
[ 7.4859e-01, 7.1489e-01, 2.7043e-02, ..., -2.7542e+00,
-1.8917e+00, -4.9725e-01],
[ 1.4069e+00, 1.4625e+00, 1.1690e+00, ..., 3.3328e-01,
7.1016e-01, -5.3772e-02]],
...,
[[-3.5927e+00, -1.7300e+00, 1.5956e+00, ..., -1.4797e+00,
-4.3890e+00, -2.0874e+00],
[-5.1882e+00, -8.0231e-01, 8.0483e-01, ..., -5.4011e+00,
-3.1430e+00, -2.9555e+00],
[-5.2757e+00, -4.1019e+00, -4.8891e+00, ..., -7.1583e+00,
-7.0447e+00, -4.0266e+00],
...,
[-1.8956e+00, -1.8263e+00, -6.5000e-01, ..., -5.0627e-01,
-3.8939e+00, -3.6917e+00],
[-2.8878e+00, -2.7885e+00, -1.7035e+00, ..., -4.6788e+00,
-6.1766e+00, -3.3807e+00],
[-3.7275e+00, -4.0227e+00, -2.7530e+00, ..., -2.7885e+00,
-5.0337e+00, -3.0692e+00]],
[[-1.5113e+00, -1.4689e-01, -4.8786e+00, ..., -1.8085e-01,
1.0623e+00, 5.7406e-01],
[-2.7182e+00, 1.2384e+00, -7.6895e+00, ..., -1.6197e+00,
1.2026e+00, 1.6920e-01],
[-1.9176e+00, 2.0191e+00, -8.5583e+00, ..., -2.1301e+00,
2.2736e-01, -3.8439e-02],
...,
[-3.1011e+00, -1.2713e+00, -1.9711e+00, ..., -2.2111e+00,
-3.0191e+00, -1.4293e+00],
[-3.7298e+00, -8.7222e-02, -1.9068e+00, ..., -3.1655e+00,
-2.9100e+00, 5.2081e-01],
[-4.6829e+00, -2.7693e+00, -2.6318e+00, ..., -5.0605e-01,
-1.8735e+00, 3.9448e-02]],
[[ 3.2622e+00, -5.0084e-01, -9.0999e-01, ..., 1.4117e+00,
3.1429e+00, 8.1424e-02],
[ 4.3073e+00, 1.4181e+00, -3.0556e-01, ..., 4.1367e+00,
4.0039e+00, 4.0695e+00],
[ 2.4932e+00, 1.0917e+00, -2.2918e+00, ..., 5.0223e+00,
3.5262e+00, 2.6274e+00],
...,
[ 2.2522e+00, 3.4174e-01, 1.7709e+00, ..., 1.3720e+00,
3.1741e+00, 2.6177e+00],
[ 4.1410e+00, 3.7455e+00, 2.0813e+00, ..., 4.0096e+00,
3.1508e+00, 2.9643e+00],
[ 2.1270e+00, 8.3981e-01, 2.4488e-01, ..., -3.5245e-01,
2.1381e-01, 1.3188e+00]]],
[[[-2.1979e+00, -1.5303e+00, 1.1465e-01, ..., 7.7902e-01,
-1.3630e+00, -2.0454e+00],
[-8.2977e+00, -1.2951e+01, -1.0989e+01, ..., 2.5903e+00,
7.0838e-01, -1.3031e-01],
[-7.2785e+00, -1.5187e+01, -1.3613e+01, ..., 1.4977e+00,
1.6237e+00, 3.2042e-01],
...,
[-1.7527e-01, -3.5351e+00, -5.6734e-01, ..., -1.2136e+01,
-7.7220e+00, -5.9179e-01],
[-1.4858e-01, -3.5759e+00, 1.3786e+00, ..., 1.0725e+00,
3.3211e+00, 3.1172e+00],
[-9.4788e-01, -3.8118e+00, -2.3354e-01, ..., -3.0107e+00,
-3.1905e+00, -3.4327e+00]],
[[-5.3089e-01, -3.5775e-01, 2.8755e+00, ..., -7.2216e-01,
-5.6014e+00, -2.4878e+00],
[ 8.9439e-01, -3.9029e+00, -7.2492e-01, ..., -8.6906e-01,
-5.1291e+00, -2.7453e+00],
[ 1.2738e+00, -2.7751e+00, -8.0080e+00, ..., -1.2013e+00,
-4.8608e+00, -4.2555e+00],
...,
[-2.5919e+00, -5.1937e+00, -3.0771e+00, ..., 1.2123e-01,
-2.0103e+00, -1.0693e+00],
[-5.7558e+00, -7.7528e+00, -1.5108e+00, ..., -2.4207e+00,
1.5539e+00, 6.4149e-01],
[-6.4917e-01, -3.3716e+00, -1.7000e+00, ..., -7.5571e+00,
-6.7790e+00, -3.1325e+00]],
[[ 6.3123e-01, -2.4292e+00, -2.8411e+00, ..., 1.6590e+00,
4.3162e+00, 2.1438e+00],
[ 1.8025e+00, 3.9799e-01, -3.5699e+00, ..., -8.7698e-01,
3.2668e+00, 1.8187e+00],
[ 1.9821e+00, 3.3064e-01, -1.7970e+00, ..., -2.8732e+00,
-1.3454e+00, -1.5434e+00],
...,
[ 1.3682e-01, 7.2288e-01, -1.1900e+00, ..., 5.1354e+00,
5.0635e+00, 4.1675e+00],
[ 2.0944e-01, 1.9811e+00, -4.6989e-01, ..., -3.2105e+00,
-1.3670e+00, 2.9780e+00],
[ 1.1349e-01, 1.4778e+00, -1.2172e+00, ..., -4.7437e+00,
-5.6677e+00, -2.2030e+00]],
...,
[[ 1.4096e+00, 2.8203e+00, 2.4710e+00, ..., -1.9996e+00,
-6.1217e+00, -1.9067e+00],
[-5.5057e-01, -3.1987e+00, -2.8541e+00, ..., -1.9890e+00,
-5.2176e+00, -1.1337e+00],
[ 3.7070e-01, -3.5667e+00, -5.0676e+00, ..., -4.4276e-01,
-2.9673e+00, -2.3151e+00],
...,
[-1.9891e+00, -3.6473e+00, -2.5517e+00, ..., -1.4547e+00,
-3.8644e+00, -6.7908e+00],
[-2.7196e+00, -3.9839e+00, 1.4534e+00, ..., 1.2294e+00,
1.7475e+00, 4.4805e-01],
[-1.3715e+00, -2.3743e+00, -5.9392e-01, ..., -8.2700e+00,
-1.0600e+01, -6.3355e+00]],
[[-2.3201e+00, -1.2680e+00, -2.5445e+00, ..., -3.1539e+00,
-1.9502e+00, 6.4013e-01],
[-1.4858e+01, -1.8412e+01, -1.9226e+01, ..., -9.5412e-01,
-2.4907e+00, 7.7876e-02],
[-1.3194e+01, -2.2812e+01, -2.4710e+01, ..., 1.8944e+00,
-1.8165e+00, -2.3697e-01],
...,
[-2.0533e+00, -2.2810e+00, -2.1102e-01, ..., -2.0779e+01,
-1.3000e+01, -1.5023e+00],
[-3.1527e+00, -6.2165e-01, 7.8382e-01, ..., -1.4508e+00,
3.2971e-02, 1.7374e+00],
[-2.5436e+00, -1.5993e+00, -1.5707e-02, ..., -1.7350e+00,
-4.2620e+00, -8.4649e-01]],
[[-1.7924e+00, -5.6752e+00, -4.3729e+00, ..., 1.8054e+00,
2.9645e+00, -6.1140e-01],
[-6.1093e+00, -1.5986e+01, -1.6811e+01, ..., 2.5317e+00,
3.7991e+00, 4.6770e+00],
[-4.8836e+00, -1.4080e+01, -1.3804e+01, ..., 1.7521e+00,
2.9880e+00, 2.0365e+00],
...,
[ 2.6612e+00, 1.0808e+00, 1.8612e+00, ..., -1.4941e+01,
-1.0664e+01, -2.2626e+00],
[ 3.9474e+00, 1.7271e+00, 1.5402e+00, ..., 6.0876e-01,
-4.7092e-01, -1.2818e+00],
[ 2.0789e+00, 3.6888e-02, 2.7102e+00, ..., 5.0814e+00,
5.4365e+00, 1.9280e+00]]],
[[[ 5.6340e-01, 1.4258e+00, 8.4828e-01, ..., 1.2295e+00,
7.2511e-02, 6.3749e-01],
[ 3.1127e-01, -6.0593e-01, 9.8460e-01, ..., 5.8613e-01,
-8.7945e-01, -2.0814e-01],
[ 2.4564e-01, -4.5526e-01, 7.7303e-01, ..., -5.5004e-01,
3.6775e-01, -7.0162e-01],
...,
[-7.3401e-01, -1.5596e+00, 4.3762e-01, ..., -3.0478e+00,
-3.7229e+00, -9.6484e-01],
[-2.2597e+00, -1.7359e+00, 3.7458e-01, ..., -2.7728e+00,
-2.9678e+00, 2.2126e-02],
[-2.3167e+00, -4.0066e+00, -3.1364e+00, ..., -3.8840e+00,
-3.4264e+00, -1.4856e+00]],
[[-1.3286e+00, -1.9067e+00, -2.2986e+00, ..., -4.0798e-02,
-4.9332e+00, -1.1180e+00],
[-5.0254e+00, -6.7992e+00, -5.1022e+00, ..., -3.8385e+00,
-7.2148e+00, -3.8366e+00],
[-4.2705e+00, -5.0188e+00, -3.4153e+00, ..., -5.5115e+00,
-5.2540e+00, -4.1848e+00],
...,
[-7.0300e+00, -8.7920e+00, -7.1171e+00, ..., -8.0707e+00,
-9.5605e+00, -2.3708e+00],
[-9.8245e+00, -1.0306e+01, -8.7908e+00, ..., -1.0118e+01,
-1.1081e+01, -3.8201e+00],
[-2.5023e+00, -4.9084e+00, -4.2821e+00, ..., -4.3165e+00,
-4.1840e+00, -6.1605e-01]],
[[-3.1875e-02, 9.4994e-01, 4.9084e-01, ..., 1.2939e+00,
2.3961e+00, 3.6791e-01],
[ 2.5695e-01, 2.9282e-01, -8.2910e-02, ..., -3.0912e-01,
2.2948e+00, -5.9487e-01],
[ 9.6903e-01, 2.7227e+00, 1.4435e+00, ..., 4.5350e-01,
1.7444e+00, 7.9404e-01],
...,
[ 3.9988e-01, 8.1126e-01, -2.7924e-01, ..., 1.3171e-01,
7.6511e-01, -9.4722e-01],
[-4.8288e-01, 7.9294e-01, -2.0041e-03, ..., 2.4069e-01,
1.0866e+00, -5.1520e-01],
[ 5.6654e-01, 1.2203e+00, 1.1220e+00, ..., 9.0435e-02,
-4.9116e-01, -1.9476e+00]],
...,
[[-9.3329e-01, -9.6166e-01, -7.2420e-01, ..., -1.3798e+00,
-2.2971e+00, 1.8040e-01],
[-4.3812e+00, -5.1045e+00, -2.5914e+00, ..., -4.4999e+00,
-4.6137e+00, -2.9368e+00],
[-2.5453e+00, -1.5486e+00, -1.1454e-01, ..., -2.5510e+00,
-3.2238e+00, -2.1570e+00],
...,
[-2.8242e+00, -3.8104e+00, -1.1308e+00, ..., -4.1474e+00,
-6.4661e+00, -2.9564e+00],
[-3.6533e+00, -5.7513e+00, -2.6886e+00, ..., -4.9449e+00,
-5.6099e+00, -1.3269e+00],
[-3.2834e+00, -3.7550e+00, -3.2830e+00, ..., -2.9478e+00,
-2.5702e+00, 4.2820e-01]],
[[-1.1330e+00, 6.3808e-02, -1.7673e+00, ..., -1.2615e+00,
-7.0968e-02, 3.6127e-01],
[-4.1475e+00, -1.6287e+00, -2.1351e+00, ..., -2.6671e+00,
-2.1510e+00, 3.8777e-01],
[-3.7583e+00, -1.8347e+00, -1.5576e+00, ..., -1.0987e+00,
-2.6207e+00, -6.9308e-01],
...,
[-4.2660e+00, -1.7110e+00, -2.5062e+00, ..., -3.7034e+00,
-2.1824e+00, -8.6097e-01],
[-3.5447e+00, -8.1758e-01, -1.6551e+00, ..., -1.9001e+00,
-8.5941e-01, 3.4906e-01],
[-3.7228e+00, -2.9888e+00, -2.9511e+00, ..., -1.3482e+00,
-1.1429e+00, 3.6847e-01]],
[[ 1.6395e+00, 6.9793e-01, 7.2220e-01, ..., 5.5194e-01,
1.5538e+00, 2.5948e-01],
[ 3.2298e+00, 2.1356e+00, 1.6990e+00, ..., 1.2011e+00,
2.7857e+00, 3.7412e+00],
[ 2.3050e+00, 3.4881e-01, 3.5559e-02, ..., 2.1701e+00,
2.8597e+00, 2.2621e+00],
...,
[ 3.0129e+00, 2.6560e+00, 2.2414e+00, ..., 3.0643e+00,
1.7487e+00, 2.2469e-01],
[ 4.7886e+00, 4.5887e+00, 3.9864e+00, ..., 2.6715e+00,
1.5154e+00, 1.6126e+00],
[ 2.8656e+00, 2.7722e+00, 2.0143e+00, ..., 2.2607e-01,
5.1600e-01, 1.0333e+00]]],
[[[-1.2940e-01, 9.9045e-01, 7.2681e-02, ..., -1.2466e+00,
-2.9409e-01, 7.8030e-01],
[ 8.2227e-03, 6.9370e-02, -4.4248e-01, ..., -2.1721e+00,
-2.0072e+00, 1.1891e+00],
[-8.5088e-01, 2.2072e-01, -9.7390e-01, ..., 5.5905e-02,
-2.1064e+00, 7.7680e-01],
...,
[-1.0517e+00, -2.0609e+00, 1.1028e+00, ..., 2.8954e+00,
2.8352e+00, -3.2538e-01],
[-1.0341e+00, -1.2301e+00, 1.9565e-01, ..., 1.7717e+00,
-1.0724e+00, -1.5998e+00],
[-2.6784e+00, -5.2492e+00, -4.4561e+00, ..., -5.3164e+00,
-4.7319e+00, -3.6664e+00]],
[[-1.8585e+00, -2.8980e+00, -4.4537e+00, ..., -3.7323e+00,
-4.5728e+00, -4.7201e-02],
[-5.9729e+00, -7.2115e+00, -7.8117e+00, ..., -6.7915e+00,
-6.7221e+00, -9.8143e-01],
[-4.5735e+00, -5.2092e+00, -9.0379e+00, ..., -4.6443e+00,
-7.7613e+00, -1.1083e+00],
...,
[-5.8999e+00, -5.6417e+00, -4.0442e+00, ..., 4.6554e-01,
-2.4950e+00, -3.7770e+00],
[-7.0371e+00, -7.7823e+00, -6.2981e+00, ..., -9.6055e+00,
-1.1148e+01, -6.2940e+00],
[-2.5378e+00, -4.4353e+00, -4.1342e+00, ..., -3.8677e+00,
-4.7712e+00, -2.5733e+00]],
[[-3.1179e-01, 2.0571e+00, 6.8322e-01, ..., 1.0921e+00,
2.0728e+00, 9.0055e-01],
[ 6.4636e-02, 9.7791e-01, 1.7133e-01, ..., 1.2413e+00,
7.5712e-01, -6.9087e-01],
[ 4.5817e-01, 1.7765e+00, 1.1919e+00, ..., 2.1090e+00,
1.3200e+00, -4.1346e-01],
...,
[ 9.3345e-01, 1.0190e+00, 6.1038e-01, ..., -6.7755e-01,
1.0032e+00, 1.4059e+00],
[ 8.6604e-01, 6.7603e-01, 8.4868e-01, ..., -1.6259e+00,
-1.8993e-02, 8.4207e-01],
[ 1.3219e+00, 1.5711e+00, 1.3608e+00, ..., 1.0203e+00,
3.1215e-01, -1.0311e+00]],
...,
[[-1.2365e+00, -1.7400e+00, -1.5542e+00, ..., -2.8105e+00,
-2.9647e+00, -4.2976e-01],
[-4.0243e+00, -3.8300e+00, -2.2790e+00, ..., -2.8274e+00,
-5.2414e+00, -2.6768e+00],
[-2.3910e+00, -1.9678e+00, -1.7408e+00, ..., -2.5789e+00,
-5.5664e+00, -2.6213e+00],
...,
[-2.8037e+00, -2.7527e+00, -4.4283e-01, ..., -2.0457e+00,
-5.9775e+00, -3.4207e+00],
[-2.9257e+00, -3.0721e+00, -1.4606e+00, ..., -2.4481e+00,
-4.9856e+00, -2.6480e+00],
[-3.4595e+00, -3.8853e+00, -3.4292e+00, ..., -3.4228e+00,
-3.9357e+00, -2.4807e+00]],
[[-1.9361e+00, -1.5731e+00, -2.1898e+00, ..., -8.1227e-01,
-7.0367e-01, -1.1744e+00],
[-4.7255e+00, -3.4662e+00, -3.0600e+00, ..., -2.2428e+00,
-7.5129e-01, -8.1206e-01],
[-4.0520e+00, -3.0640e+00, -3.3403e+00, ..., 2.0929e-01,
-6.6474e-01, -2.4878e-01],
...,
[-4.1053e+00, -2.0196e+00, -2.9438e+00, ..., -1.1478e+00,
-4.9602e+00, -1.8761e+00],
[-3.7529e+00, -6.0508e-01, -1.7719e+00, ..., -3.2923e+00,
-4.8729e+00, -1.0729e+00],
[-4.4252e+00, -2.9491e+00, -2.6672e+00, ..., -1.1920e+00,
-1.7986e+00, -3.3930e-01]],
[[ 1.5431e+00, 6.1394e-01, 6.3814e-01, ..., -8.0468e-01,
-1.4405e-01, -4.2129e-01],
[ 4.2407e+00, 3.2206e+00, 2.8840e+00, ..., 2.5019e+00,
1.3895e+00, 2.0845e+00],
[ 1.7259e+00, 1.0373e+00, 2.3717e+00, ..., 2.3410e+00,
1.5910e+00, 9.4790e-01],
...,
[ 2.9804e+00, 1.3104e+00, 1.3466e+00, ..., -1.6523e+00,
2.1564e+00, 2.0951e+00],
[ 3.6492e+00, 3.3334e+00, 2.2065e+00, ..., 4.6079e+00,
3.3273e+00, 2.9187e+00],
[ 1.9320e+00, 7.3633e-01, 2.6840e-01, ..., -5.8250e-01,
-4.2065e-02, 1.7112e+00]]],
[[[ 4.5792e-01, 1.4473e+00, 8.6139e-01, ..., 1.0624e+00,
1.6049e+00, 9.9136e-01],
[ 1.8935e-01, -2.4329e+00, -1.3663e+00, ..., 3.3442e-01,
1.8753e+00, 4.3524e-01],
[-1.7236e+00, -5.4052e+00, -1.5906e+00, ..., 1.6045e+00,
1.8720e+00, -3.9592e-01],
...,
[-1.4528e+00, -1.9973e+00, -4.8370e+00, ..., -2.5192e+00,
-4.4955e+00, -8.6689e-01],
[-2.4586e+00, -1.4211e+00, -3.1564e+00, ..., -3.1204e+00,
-2.7301e+00, 5.1192e-01],
[-1.7023e+00, -3.3756e+00, -3.6398e+00, ..., -4.4542e+00,
-3.6990e+00, -1.4584e+00]],
[[-1.7113e+00, -2.1208e+00, -1.4786e+00, ..., 1.8761e+00,
-2.6826e+00, -1.6861e+00],
[-5.7903e+00, -8.8089e+00, -4.2914e+00, ..., 5.8733e-01,
-3.1222e+00, -4.5227e+00],
[-5.3806e+00, -7.7844e+00, -8.6568e-01, ..., -3.9939e-01,
-2.7242e+00, -4.1177e+00],
...,
[-6.9851e+00, -8.5590e+00, -8.2003e+00, ..., -8.0187e+00,
-1.0188e+01, -2.1102e+00],
[-9.5247e+00, -9.6091e+00, -9.1757e+00, ..., -1.1426e+01,
-1.0451e+01, -3.5065e+00],
[-2.0879e+00, -3.9335e+00, -3.5482e+00, ..., -5.1293e+00,
-4.3157e+00, -6.8494e-01]],
[[-3.4814e-01, 6.2629e-01, 5.4675e-01, ..., 5.2103e-01,
3.0282e+00, 5.1676e-01],
[-5.0253e-01, -6.0578e-01, -1.5246e-01, ..., -3.6329e+00,
2.4128e+00, 5.1048e-01],
[ 7.4860e-01, 3.1363e+00, 7.1095e-01, ..., -1.1754e+00,
2.3506e+00, 2.3979e+00],
...,
[ 5.0126e-01, 7.3102e-01, 1.0355e+00, ..., 2.2188e+00,
1.1044e+00, -1.2022e+00],
[-7.6516e-01, -1.1729e-02, -1.3562e+00, ..., 2.9971e-02,
-7.1831e-02, -8.1377e-01],
[ 1.3355e-01, 8.0051e-01, -1.2762e-01, ..., -5.3111e-01,
-9.1520e-01, -1.9815e+00]],
...,
[[-5.9248e-01, -1.1131e+00, -2.0476e-01, ..., 1.8331e-01,
-2.1092e+00, -1.2950e-01],
[-4.2187e+00, -6.0281e+00, -3.4447e+00, ..., -1.1013e+00,
-5.6378e+00, -2.2818e+00],
[-3.9411e+00, -3.5710e+00, -1.7841e-01, ..., -1.8508e-01,
-5.2793e+00, -2.1078e+00],
...,
[-2.4435e+00, -3.8107e+00, -4.4683e+00, ..., -6.3728e+00,
-5.3899e+00, -3.1556e+00],
[-3.0969e+00, -5.0193e+00, -3.5171e+00, ..., -4.8782e+00,
-4.4922e+00, -1.4408e+00],
[-2.5598e+00, -3.5290e+00, -4.0720e+00, ..., -2.8982e+00,
-2.7353e+00, 1.0279e-01]],
[[-7.9990e-01, 8.2198e-01, -9.9998e-01, ..., -2.0166e+00,
-1.3945e+00, 1.3877e+00],
[-4.4779e+00, -1.1580e+00, -2.4973e+00, ..., -1.7813e+00,
-4.2987e+00, 1.0928e+00],
[-5.3639e+00, -2.0338e+00, -1.6635e+00, ..., -7.7438e-01,
-6.4385e+00, -7.2243e-01],
...,
[-4.3820e+00, -2.7835e+00, -1.9811e+00, ..., -5.2321e+00,
-2.7830e+00, -9.3851e-01],
[-3.1454e+00, -8.6694e-01, -1.2156e+00, ..., -2.3677e+00,
-8.5402e-01, 2.9476e-01],
[-2.9666e+00, -2.2239e+00, -1.7554e+00, ..., -1.5002e+00,
-8.9709e-01, 4.0790e-01]],
[[ 1.7223e+00, 7.6262e-01, 2.1816e-01, ..., 3.7469e-02,
1.8378e+00, 8.1102e-01],
[ 3.9019e+00, 2.9570e+00, 1.1474e+00, ..., -2.2151e+00,
2.4489e+00, 4.6409e+00],
[ 2.0183e+00, 1.8800e-01, -7.2322e-01, ..., 1.3776e-01,
2.7783e+00, 2.5493e+00],
...,
[ 2.8235e+00, 2.7550e+00, 2.0366e-01, ..., 2.1881e+00,
1.6386e+00, 2.0207e-01],
[ 4.2135e+00, 5.3496e+00, 2.8705e+00, ..., 2.6948e+00,
1.8002e+00, 1.4573e+00],
[ 2.7834e+00, 3.0487e+00, 1.0088e+00, ..., 4.4660e-01,
8.8364e-01, 9.6257e-01]]]], device='cuda:0'),
tensor([[[[ 1.2301e+00, 1.4798e+00, 1.2852e+00, ..., -2.1573e+00,
-1.3487e+00, -1.1986e+00],
[ 3.3834e-01, -4.0722e+00, -4.2583e+00, ..., -5.4219e-01,
-2.3093e-01, -1.6620e+00],
[-2.3718e+00, -4.7272e+00, -5.3412e-01, ..., 3.6849e-03,
-8.9900e-01, -2.0514e+00],
...,
[-9.8021e-01, -3.3522e-01, -7.9438e+00, ..., -4.8309e+00,
2.7594e+00, 3.3431e+00],
[-1.5689e+00, -8.4507e-01, -9.1010e+00, ..., -5.6770e+00,
4.2500e+00, 1.3852e+00],
[-1.6367e+00, -4.5710e+00, -8.1593e+00, ..., 1.0134e+00,
3.0440e+00, -1.1583e+00]],
[[-9.4570e-01, -2.9450e+00, -4.8786e+00, ..., -4.0820e+00,
-5.2190e+00, -3.2056e+00],
[-6.7303e+00, -9.3313e+00, -8.6362e+00, ..., -4.1204e+00,
-5.7763e+00, -5.4033e+00],
[-6.5520e+00, -5.7890e+00, -2.3036e+00, ..., -3.8912e+00,
-4.9889e+00, -3.6228e+00],
...,
[-6.6808e+00, -7.8392e+00, -1.1765e+01, ..., -4.3091e+00,
6.1136e-01, 8.4823e-01],
[-8.7420e+00, -8.2899e+00, -1.3859e+01, ..., -3.1669e+00,
3.2589e-01, -3.6443e+00],
[-2.6763e+00, -4.1943e+00, -5.5868e+00, ..., 5.5870e-01,
4.2651e-01, -1.6545e+00]],
[[ 3.5310e-01, 9.3722e-01, 1.1048e+00, ..., 1.9113e+00,
2.1136e+00, 7.9057e-01],
[-3.8834e-01, 1.3942e+00, 2.8743e+00, ..., 2.1098e+00,
2.6298e+00, 1.1128e+00],
[ 9.1300e-01, 2.1477e+00, 1.0695e+00, ..., 2.1508e+00,
2.5896e+00, 2.2705e+00],
...,
[ 1.7022e-02, 7.3251e-02, 2.0049e+00, ..., -9.1233e+00,
-5.1221e+00, 1.2518e-01],
[-2.2541e-01, -2.6646e+00, 3.2616e-01, ..., -2.6164e+00,
1.9256e+00, 2.5513e+00],
[ 7.8818e-01, -1.8470e+00, -2.3898e+00, ..., -1.0718e+00,
-3.2972e-02, 2.1003e+00]],
...,
[[-6.9002e-01, -2.0112e+00, -2.2636e+00, ..., -3.7701e+00,
-4.2143e+00, -1.3678e+00],
[-4.4722e+00, -5.8166e+00, -5.3873e+00, ..., -3.2266e+00,
-5.6915e+00, -3.7099e+00],
[-3.5511e+00, -8.7274e-01, 1.5359e+00, ..., -2.4879e+00,
-4.9579e+00, -2.7521e+00],
...,
[-4.5896e-01, -2.4877e+00, -4.1941e+00, ..., -3.0685e+00,
-2.2746e+00, -1.6329e+00],
[-1.1237e+00, -3.7812e+00, -5.8765e+00, ..., -2.6763e-01,
-4.3454e+00, -5.5520e+00],
[-1.8499e+00, -5.5639e+00, -3.6458e+00, ..., 8.6907e-01,
-3.7558e+00, -3.1965e+00]],
[[-2.3428e+00, -2.7487e+00, -3.5323e+00, ..., -2.7940e+00,
-2.7825e+00, -2.6535e-01],
[-5.4292e+00, -5.5606e+00, -8.0515e+00, ..., -2.9497e+00,
-5.6290e+00, -1.6315e+00],
[-4.1575e+00, -3.4362e-01, -2.2627e+00, ..., 5.2084e-02,
-3.7446e+00, -1.4317e+00],
...,
[-3.1987e+00, -3.3560e+00, -3.1264e-03, ..., -1.3362e+00,
-1.8326e+00, 8.9632e-01],
[-2.5706e+00, -1.5551e+00, -9.1822e-01, ..., -2.0767e+00,
-3.7789e+00, -4.8069e-01],
[-2.7298e+00, -2.1131e+00, 6.0317e-01, ..., 7.8070e-02,
-4.1047e+00, -2.1934e+00]],
[[ 1.0406e+00, 2.4221e-01, 1.8346e+00, ..., 4.8338e-01,
1.4643e+00, 6.9631e-01],
[ 3.6484e+00, -1.6590e-01, 3.8451e-01, ..., 1.7342e+00,
2.2146e+00, 3.4838e+00],
[ 2.3297e+00, -5.5197e-01, 7.6202e-01, ..., 1.0864e+00,
8.2722e-01, 8.2294e-01],
...,
[ 2.9525e+00, 3.1312e+00, 4.6670e-01, ..., -5.3317e+00,
5.6127e-01, 3.1593e+00],
[ 4.1559e+00, 4.5136e+00, 8.1821e-01, ..., -5.1512e+00,
3.3694e+00, 4.3964e+00],
[ 2.5347e+00, 1.9174e+00, 2.2055e-01, ..., -1.1297e+00,
3.7122e+00, 2.0453e+00]]],
[[[-6.9925e-01, -9.7614e-01, -1.5171e+00, ..., -1.4165e+00,
-2.0253e+00, -6.4227e-01],
[ 6.1533e-01, -2.3195e+00, -4.9366e-01, ..., 6.8944e-01,
2.4531e+00, 7.4538e-01],
[ 2.8607e-01, -6.3699e-01, 1.9852e+00, ..., -2.4810e+00,
1.4538e+00, 1.2314e+00],
...,
[-1.8952e+00, 1.0620e+00, -4.5640e+00, ..., -2.5173e+00,
-3.7147e+00, -1.1285e+00],
[-2.5684e+00, 9.8358e-01, -1.2181e+00, ..., -3.6471e+00,
-3.4644e+00, -7.1601e-02],
[-2.7592e+00, -2.1344e+00, -2.0717e+00, ..., -4.0769e+00,
-3.4285e+00, -1.5175e+00]],
[[ 4.5939e-01, -1.5845e+00, -4.0040e-01, ..., -3.3182e+00,
-2.6524e+00, -1.3848e+00],
[ 1.0462e+00, -2.0904e+00, -6.6012e-01, ..., -9.0287e-01,
7.6505e-01, -7.0484e-01],
[ 7.2947e-02, 1.1923e+00, 4.5566e-01, ..., -4.1638e+00,
-2.7625e+00, -1.1765e+00],
...,
[-2.2919e+00, 3.8050e+00, -3.9010e+00, ..., -8.0985e+00,
-9.6407e+00, -2.1637e+00],
[-1.0204e+01, -7.2582e+00, -2.9664e+00, ..., -1.0674e+01,
-1.0593e+01, -3.9242e+00],
[-2.1636e+00, -5.8065e+00, -4.8871e+00, ..., -4.3057e+00,
-4.2026e+00, -6.5924e-01]],
[[ 3.7708e-01, -7.1484e-01, -6.7498e-01, ..., 2.8586e+00,
4.1080e+00, 2.3172e+00],
[-4.5302e-01, -2.1995e-01, -1.0774e+00, ..., -2.3037e+00,
-2.0794e-01, 2.9906e+00],
[-2.0972e+00, 8.5618e-01, -1.0304e+00, ..., -2.6024e+00,
-3.7000e+00, 1.7454e+00],
...,
[-1.0166e+01, -9.5716e+00, -6.8860e+00, ..., 8.2463e-01,
-5.0759e-01, -1.6439e+00],
[-7.2939e+00, -1.1302e+01, -4.8718e+00, ..., -1.2241e-01,
1.0736e+00, 3.0673e-01],
[ 4.7442e-02, -2.5763e+00, -3.4156e+00, ..., -5.2044e-01,
-5.0684e-01, -1.8170e+00]],
...,
[[ 1.0651e-01, -4.2681e-01, 7.0623e-01, ..., -2.4054e+00,
-2.6676e+00, -1.4707e+00],
[ 1.6641e-01, -3.1592e+00, 5.8031e-02, ..., -1.7365e-01,
5.6121e-02, -2.8685e+00],
[ 2.3280e-01, -6.0389e-01, 8.6706e-01, ..., -5.1882e-01,
3.0347e-02, -2.3763e+00],
...,
[-2.8439e+00, -3.2174e+00, -2.3170e+00, ..., -6.3758e+00,
-7.3515e+00, -3.7442e+00],
[-5.1692e+00, -7.0873e+00, -3.5735e+00, ..., -4.7291e+00,
-5.8202e+00, -1.6460e+00],
[-4.2369e+00, -5.6218e+00, -5.4326e+00, ..., -2.8696e+00,
-2.5680e+00, 4.3022e-01]],
[[-3.2768e+00, -2.6226e+00, -1.7431e+00, ..., -2.3642e+00,
-4.1014e+00, 3.0309e-01],
[-3.1626e+00, -5.8558e+00, -3.8577e+00, ..., -8.4093e-01,
-3.1206e+00, -1.8171e+00],
[-1.0625e+00, -3.4950e+00, -1.7732e+00, ..., -1.0239e+00,
-1.3010e+00, -2.0805e+00],
...,
[ 2.9213e-01, -4.7592e+00, -5.2964e+00, ..., -2.8691e+00,
-2.8196e+00, -4.7769e-01],
[ 5.3678e-01, 2.2699e+00, 4.9937e-01, ..., -1.2525e+00,
-8.6649e-01, 6.0660e-01],
[-1.0353e+00, 1.9424e+00, -2.8848e-01, ..., -1.3388e+00,
-1.1163e+00, 3.2823e-01]],
[[-2.5962e-01, -1.6360e+00, -2.2873e+00, ..., 3.7252e-01,
1.0637e+00, 2.2697e+00],
[ 4.1938e-01, -4.0383e+00, -2.6218e+00, ..., -6.7572e-01,
-1.9857e-01, 3.8746e+00],
[ 9.3654e-01, -2.7942e+00, -5.1502e-01, ..., -3.3068e+00,
-2.4625e-01, 1.4954e+00],
...,
[ 1.7171e-01, -3.9919e+00, -7.2133e+00, ..., 2.8258e+00,
2.7362e+00, 1.9430e+00],
[ 6.4611e+00, 4.0317e+00, -1.4002e+00, ..., 2.8184e+00,
1.5987e+00, 8.8205e-01],
[ 1.8270e+00, 4.4060e+00, 2.0258e+00, ..., 7.0865e-01,
6.8830e-01, 1.0876e+00]]],
[[[ 5.0915e-01, 2.0893e+00, 1.7123e+00, ..., -9.8661e-01,
-3.4300e+00, -1.1821e+00],
[-3.2702e-01, -1.2889e+00, -3.0910e-01, ..., 5.6900e-01,
-1.9836e+00, -7.5801e-01],
[-2.4822e+00, -1.7283e+00, 2.2115e-01, ..., -4.0430e-01,
-1.4419e+00, -5.1176e-01],
...,
[-7.6689e-01, -2.3446e+00, 1.7236e+00, ..., -1.3970e+00,
-2.6299e+00, -3.2909e-01],
[-2.3155e+00, -1.9405e+00, 7.4487e-01, ..., -2.6103e+00,
-4.0717e+00, -1.1459e+00],
[-1.9789e+00, -3.6793e+00, -2.1053e+00, ..., -3.6747e+00,
-3.6155e+00, -1.6973e+00]],
[[-3.7908e-01, -1.3386e+00, -3.5055e+00, ..., -5.2391e+00,
-9.2371e+00, -2.3126e+00],
[-5.2366e+00, -6.0720e+00, -4.6490e+00, ..., -5.8856e+00,
-8.8102e+00, -3.7323e+00],
[-3.1096e+00, -3.8846e+00, -3.8837e+00, ..., -3.6220e+00,
-5.8836e+00, -3.0354e+00],
...,
[-7.2685e+00, -1.1846e+01, -7.2713e+00, ..., -5.7860e+00,
-7.1887e+00, -1.9789e+00],
[-8.5240e+00, -8.3671e+00, -7.9575e+00, ..., -9.3116e+00,
-1.0872e+01, -4.5830e+00],
[-2.4646e+00, -4.5944e+00, -2.7950e+00, ..., -3.9910e+00,
-3.9923e+00, -6.3250e-01]],
[[-3.4590e-01, 2.2178e+00, 1.2641e+00, ..., 2.0699e+00,
2.5688e+00, 7.9159e-01],
[-8.5876e-01, 6.5953e-01, 1.1010e+00, ..., 1.1613e+00,
2.3339e+00, -1.3321e-01],
[ 1.1684e-01, 8.6132e-01, 1.2552e+00, ..., -2.0695e-02,
1.5067e+00, -2.7565e-01],
...,
[-1.0042e+00, -1.7110e+00, -2.0591e+00, ..., -2.9384e-02,
4.7545e-01, 3.1686e-01],
[-1.5157e+00, 2.4105e-01, 8.7263e-01, ..., 1.9261e-01,
1.0059e-01, -8.5256e-01],
[-9.3819e-02, -1.1982e+00, -5.0297e-01, ..., 1.5397e-01,
2.3024e-01, -1.4885e+00]],
...,
[[-6.2987e-01, -1.4804e+00, -3.5321e-01, ..., -3.4954e+00,
-5.7941e+00, -1.3931e+00],
[-3.0154e+00, -3.2741e+00, -1.7797e+00, ..., -2.6970e+00,
-5.8213e+00, -2.4136e+00],
[-2.0873e+00, -7.7709e-01, 1.5591e-01, ..., -2.1241e+00,
-5.3019e+00, -1.5731e+00],
...,
[-3.8330e+00, -5.7087e+00, -1.4953e+00, ..., -1.0231e+00,
-4.2934e+00, -1.2229e+00],
[-2.3952e+00, -3.2302e+00, -1.3217e+00, ..., -2.0134e+00,
-4.7244e+00, -7.8569e-01],
[-2.2809e+00, -1.9494e+00, -8.2071e-03, ..., -1.4669e+00,
-2.3728e+00, 8.2894e-01]],
[[-1.3669e+00, -1.9097e+00, -1.0608e+00, ..., -2.2840e+00,
-1.2613e+00, 2.2719e-01],
[-3.6881e+00, -2.8681e+00, -1.3970e+00, ..., -1.8307e+00,
-2.6493e+00, -1.4190e-01],
[-1.4799e+00, 3.6975e-01, -9.0314e-01, ..., 7.4587e-01,
-1.6370e+00, 5.2886e-01],
...,
[-3.9757e+00, -2.3525e+00, -9.7159e-01, ..., -1.2710e+00,
-3.2516e+00, -2.3797e-01],
[-2.7659e+00, 4.1743e-01, -5.0390e-01, ..., 1.8322e-01,
-1.5340e+00, 1.0108e+00],
[-3.1653e+00, -1.9310e+00, -3.1459e-01, ..., -3.5531e-01,
-1.2034e+00, 5.4933e-01]],
[[ 8.1654e-01, 3.1321e-02, 1.2497e+00, ..., 1.6749e+00,
1.7843e+00, 1.6581e-01],
[ 3.7231e+00, 2.2704e+00, 1.6249e+00, ..., 3.7615e+00,
3.0153e+00, 3.9117e+00],
[ 9.0942e-01, 8.9026e-01, -2.0976e-01, ..., 2.1525e+00,
1.8658e+00, 2.2004e+00],
...,
[ 4.7449e+00, 4.7894e+00, 3.6114e+00, ..., 1.9329e+00,
1.0245e+00, 1.0758e+00],
[ 3.0374e+00, 2.5221e+00, 2.2436e+00, ..., 3.4209e+00,
1.7735e+00, 3.0499e+00],
[ 3.0273e+00, 2.9314e+00, 1.4972e+00, ..., 4.2777e-01,
-1.3390e-01, 1.3225e+00]]],
[[[-3.0059e-01, 1.0684e+00, 5.5022e-01, ..., -4.1112e-01,
1.4132e-01, -2.4041e-01],
[ 1.2079e-01, 6.3562e-01, 1.7195e+00, ..., 1.3950e-01,
6.2228e-01, -5.9356e-01],
[-1.2290e+00, 3.3174e-01, -2.1297e-01, ..., -4.2208e-01,
-1.2088e-01, -8.9505e-01],
...,
[-1.7172e-01, -7.8112e-01, -1.8446e+00, ..., -9.1549e-01,
4.7530e-01, -3.7922e-01],
[-1.5357e+00, -6.0798e-02, -3.9769e-01, ..., 1.4784e-01,
4.5214e-01, -1.2267e+00],
[-1.4088e+00, -2.5647e+00, -2.0796e+00, ..., -3.1084e+00,
-2.8054e+00, -2.9295e+00]],
[[-1.5471e+00, -2.2081e+00, -3.9081e+00, ..., -2.2546e+00,
-3.5073e+00, -2.2873e+00],
[-4.5433e+00, -4.7280e+00, -5.0205e+00, ..., -3.6411e+00,
-3.9082e+00, -3.6827e+00],
[-3.5187e+00, -4.0515e+00, -6.1449e+00, ..., -3.6449e+00,
-3.0276e+00, -2.9189e+00],
...,
[-4.6670e+00, -5.9249e+00, -7.9557e+00, ..., -3.9494e+00,
-4.9055e+00, -4.5630e+00],
[-7.9705e+00, -8.3105e+00, -6.2105e+00, ..., -5.0046e+00,
-4.7801e+00, -4.9885e+00],
[-1.4254e+00, -3.2311e+00, -2.4089e+00, ..., -3.2523e+00,
-3.0605e+00, -1.8693e+00]],
[[-1.6337e-01, 2.3282e+00, 1.5451e+00, ..., 1.6598e+00,
1.7622e+00, 8.5877e-01],
[-5.7896e-01, 3.0527e-01, 8.8320e-01, ..., 1.0983e+00,
1.4929e+00, 8.8964e-01],
[-1.3078e-01, 1.8447e+00, 1.6522e+00, ..., 1.7992e+00,
1.5967e+00, 1.4869e+00],
...,
[-6.0647e-01, 3.6815e-01, 8.8321e-01, ..., 1.7926e+00,
-3.4282e-01, 7.4014e-02],
[-9.8495e-01, -2.8125e-01, 8.5184e-01, ..., 8.1304e-01,
6.6815e-01, 1.2990e+00],
[-2.2471e-01, -1.7571e-01, -3.0452e-01, ..., 1.6257e+00,
1.3658e+00, 4.1476e-01]],
...,
[[-1.3633e+00, -2.1002e+00, -2.2575e+00, ..., -1.9330e+00,
-2.0064e+00, -4.7461e-01],
[-3.3499e+00, -3.6060e+00, -2.3357e+00, ..., -2.3039e+00,
-3.7694e+00, -2.8903e+00],
[-2.4852e+00, -2.9420e+00, -2.6971e+00, ..., -1.5592e+00,
-3.1172e+00, -1.7871e+00],
...,
[-1.2329e+00, -3.6823e+00, -3.3388e+00, ..., -2.6210e+00,
-3.5919e+00, -3.2896e+00],
[-2.4201e+00, -3.6194e+00, -6.0803e-01, ..., -1.3881e+00,
-2.2427e+00, -1.7674e+00],
[-1.6538e+00, -1.8977e+00, -1.1541e+00, ..., -2.2044e+00,
-2.6176e+00, -7.9301e-01]],
[[-1.8011e+00, -1.8887e+00, -2.6637e+00, ..., -9.4292e-01,
-1.5441e+00, -5.1680e-01],
[-3.8962e+00, -2.3811e+00, -3.2570e+00, ..., -1.3695e+00,
-4.0583e+00, -1.8194e+00],
[-3.6061e+00, -3.1648e+00, -3.6368e+00, ..., -2.0153e-01,
-3.9215e+00, -2.4576e+00],
...,
[-2.8063e+00, -4.1184e+00, -4.0237e+00, ..., -1.6497e+00,
-2.5635e+00, -1.4858e+00],
[-1.8236e+00, -3.9747e-02, -1.3357e+00, ..., -1.1489e+00,
-3.5013e+00, -1.8579e+00],
[-2.0237e+00, -1.3810e+00, -1.6138e+00, ..., -1.5918e+00,
-2.9788e+00, -1.2835e+00]],
[[ 1.1712e+00, 2.8037e-01, 9.2934e-01, ..., -4.3271e-03,
1.0240e+00, 5.8021e-01],
[ 3.9800e+00, 3.0306e+00, 2.9687e+00, ..., 9.0010e-01,
9.6448e-01, 2.4964e+00],
[ 1.3817e+00, 9.5241e-01, 1.4157e+00, ..., 9.7511e-01,
3.3068e-02, 6.7243e-01],
...,
[ 2.8755e+00, 2.3302e+00, 2.7677e+00, ..., -8.3061e-02,
1.8913e+00, 1.4783e+00],
[ 3.7830e+00, 3.9999e+00, 2.0599e+00, ..., 1.2519e+00,
1.2998e+00, 1.2511e+00],
[ 2.3944e+00, 2.5124e+00, 1.7560e+00, ..., 6.5476e-02,
-1.1781e-02, 2.3871e-01]]],
[[[ 3.6105e-01, 1.3621e+00, 3.0692e-01, ..., -3.0791e+00,
-2.2719e+00, -1.1990e+00],
[ 5.3046e-01, 1.0213e+00, 9.7358e-01, ..., -1.5662e+00,
-1.9309e+00, -2.5534e-01],
[-3.5052e-01, -4.3963e-01, -4.5705e-01, ..., -7.2978e-01,
-2.4491e+00, -7.0099e-02],
...,
[-6.3539e-01, -1.3169e+00, -1.2185e-01, ..., -1.8602e+00,
-3.6509e+00, 5.1949e-01],
[-1.9326e+00, -1.2331e+00, -1.0077e+00, ..., -2.8506e+00,
-3.4799e+00, 6.4194e-01],
[-1.1362e+00, -3.1760e+00, -3.3195e+00, ..., -3.4976e+00,
-3.1115e+00, -1.2229e+00]],
[[-9.3534e-01, -1.4742e+00, -3.4322e+00, ..., -6.2478e+00,
-7.4771e+00, -1.5793e+00],
[-3.7891e+00, -4.1762e+00, -3.8755e+00, ..., -5.8659e+00,
-8.2760e+00, -2.5817e+00],
[-3.8903e+00, -4.1099e+00, -3.8783e+00, ..., -4.8428e+00,
-7.6390e+00, -2.7121e+00],
...,
[-5.9583e+00, -5.9410e+00, -4.6944e+00, ..., -6.2134e+00,
-8.4178e+00, -1.8002e+00],
[-7.7253e+00, -7.6329e+00, -7.8789e+00, ..., -9.1259e+00,
-1.0880e+01, -4.2222e+00],
[-1.7992e+00, -3.2940e+00, -3.8895e+00, ..., -3.3657e+00,
-3.5819e+00, -6.4597e-01]],
[[-4.6754e-02, 1.2050e+00, 1.5839e+00, ..., 1.5584e+00,
2.6156e+00, 1.0232e+00],
[ 2.3990e-02, -1.0444e-02, 6.8583e-01, ..., 1.4710e+00,
3.0970e+00, 1.5412e-01],
[ 6.8518e-01, 1.1795e+00, 1.5891e+00, ..., -1.0529e-01,
1.8410e+00, -2.0151e-01],
...,
[ 2.2311e-01, 6.5117e-01, 9.4752e-01, ..., 3.0726e-01,
9.2394e-01, -1.4721e+00],
[ 3.8324e-01, 4.8922e-01, 1.0932e+00, ..., -6.4417e-02,
1.2698e+00, -1.3872e+00],
[ 5.1973e-01, 1.0124e+00, 1.7100e+00, ..., -2.5662e-01,
-5.2103e-01, -1.9053e+00]],
...,
[[-5.1213e-01, -1.5056e+00, -2.0729e+00, ..., -5.0506e+00,
-5.3949e+00, -8.5978e-01],
[-2.6114e+00, -2.7459e+00, -1.7095e+00, ..., -3.5492e+00,
-4.8380e+00, -1.8733e+00],
[-1.9762e+00, -1.2970e+00, -1.2021e+00, ..., -4.0272e+00,
-5.0527e+00, -1.3095e+00],
...,
[-1.8220e+00, -1.2899e+00, -3.5967e-01, ..., -2.9788e+00,
-5.0827e+00, -1.6789e+00],
[-3.2077e+00, -3.0952e+00, -2.7704e+00, ..., -3.5622e+00,
-5.1061e+00, -8.0776e-01],
[-1.5459e+00, -1.9616e+00, -2.3019e+00, ..., -1.6288e+00,
-2.0012e+00, 5.3738e-01]],
[[-1.2049e+00, -2.4399e+00, -3.8157e+00, ..., -1.6308e+00,
-7.7668e-01, -6.4654e-01],
[-3.5582e+00, -3.6377e+00, -4.7228e+00, ..., -3.3017e+00,
-2.0141e+00, -5.8428e-01],
[-4.5539e+00, -3.9038e+00, -4.2769e+00, ..., -2.0278e+00,
-1.7487e+00, -7.8662e-01],
...,
[-4.5633e+00, -4.1549e+00, -4.1381e+00, ..., -3.0612e+00,
-1.6272e+00, -6.2104e-01],
[-4.0374e+00, -2.2160e+00, -2.7747e+00, ..., -1.1594e+00,
-3.3217e-01, 1.3768e+00],
[-3.1935e+00, -3.3249e+00, -2.5828e+00, ..., -4.9806e-01,
-8.5849e-01, 7.5811e-01]],
[[ 1.1778e+00, 7.6257e-03, 1.1240e+00, ..., 6.0156e-01,
1.0071e+00, -1.4072e+00],
[ 2.1926e+00, 1.1925e+00, 1.4494e+00, ..., 3.6380e+00,
3.1169e+00, 2.5208e+00],
[ 1.4176e+00, 8.0346e-01, 3.8422e-01, ..., 3.1245e+00,
1.7787e+00, 1.4707e+00],
...,
[ 2.4370e+00, 1.6976e+00, 9.8542e-01, ..., 3.0456e+00,
2.1640e+00, 6.8637e-01],
[ 2.8809e+00, 4.1565e+00, 2.2995e+00, ..., 2.5548e+00,
1.0524e+00, 1.9310e+00],
[ 1.9910e+00, 1.6315e+00, 4.2854e-01, ..., 4.0161e-01,
-1.3335e-01, 8.9470e-01]]]], device='cuda:0'),
tensor([[[[ 2.1347e+00, 1.5209e+00, -3.8952e-01, ..., -1.8174e+00,
-1.5159e+00, -2.8395e-01],
[-1.2602e+00, -4.3392e+00, -3.9687e+00, ..., -1.8271e+00,
-6.2305e-01, -1.2951e+00],
[-1.8262e+00, 5.6208e-01, -8.1071e-01, ..., 2.8052e+00,
8.3331e-02, -2.5144e+00],
...,
[-2.5231e+00, -8.2875e+00, -3.7756e+00, ..., 2.2823e+00,
-3.8004e+00, -3.3511e+00],
[-3.9386e+00, -5.4263e+00, -2.1493e+00, ..., -4.1998e+00,
-2.5759e+00, -2.9562e+00],
[-1.7261e+00, -3.4072e+00, -4.7977e+00, ..., -4.1482e+00,
-2.8626e+00, -3.2144e+00]],
[[-1.0970e+00, -4.9313e+00, -8.2852e+00, ..., -7.2735e+00,
-6.7326e+00, -2.6131e+00],
[-6.8405e+00, -8.5674e+00, -7.1538e+00, ..., -7.7689e+00,
-9.9634e+00, -4.7107e+00],
[-3.1042e+00, -2.3609e+00, -4.3329e+00, ..., -2.2092e+00,
-7.8274e+00, -5.3173e+00],
...,
[-9.1389e+00, -2.0132e+01, -1.5257e+01, ..., -2.5534e+00,
-1.0331e+01, -4.3660e+00],
[-7.4949e+00, -8.7008e+00, -1.1170e+01, ..., -9.5092e+00,
-7.5971e+00, -5.2633e+00],
[-1.4372e+00, -4.4524e+00, -4.2817e+00, ..., -4.4901e+00,
-2.6595e+00, -1.8502e+00]],
[[ 3.0735e-01, 2.8142e+00, 1.3995e+00, ..., 1.4888e+00,
2.0373e+00, 9.0088e-01],
[ 9.6283e-01, 4.5926e+00, 3.7513e+00, ..., 5.3375e+00,
2.2102e+00, 6.7163e-01],
[ 2.7837e+00, 3.2668e+00, 3.3694e+00, ..., 7.8600e+00,
3.1531e+00, 8.8197e-01],
...,
[-1.5376e+00, -2.0914e-01, -4.6992e+00, ..., 4.9158e+00,
2.8358e+00, -7.4234e-01],
[ 1.2764e+00, 5.9928e+00, -1.5261e+00, ..., 1.9706e+00,
4.6425e-03, -9.1980e-03],
[ 2.4024e-02, -1.0748e+00, -8.7755e-01, ..., -4.9459e-01,
2.3414e-01, 7.0488e-01]],
...,
[[-8.1767e-01, -2.8468e+00, -1.7627e+00, ..., -2.0805e+00,
-4.8668e+00, -9.6237e-01],
[-5.4327e+00, -5.8124e+00, -3.5571e+00, ..., -6.3874e+00,
-5.3896e+00, -5.4994e+00],
[-1.3907e+00, 2.1115e-01, -1.1970e+00, ..., -4.7579e+00,
-2.8119e+00, -4.8045e+00],
...,
[-6.9977e+00, -1.3635e+01, -6.7136e+00, ..., -6.3597e+00,
-2.3002e+00, -3.4057e+00],
[-1.4586e+00, -2.8741e+00, -3.6861e+00, ..., -8.3825e+00,
-4.6840e+00, -3.0113e+00],
[-1.0487e+00, -1.2394e+00, 1.0338e+00, ..., -4.1843e+00,
-3.2800e+00, -1.5073e+00]],
[[-2.4444e+00, -2.7134e+00, -5.1451e-01, ..., -1.2393e-01,
-1.1273e+00, -5.1539e-01],
[-7.0069e+00, -7.3161e+00, -2.2803e+00, ..., 5.9743e-01,
-2.4285e+00, -2.2858e+00],
[-3.8948e+00, -2.8512e+00, -1.2847e+00, ..., -3.2004e+00,
-1.2612e+00, -1.2165e+00],
...,
[-2.3149e+00, -1.3766e+00, 4.9780e-01, ..., -5.2831e+00,
-3.2543e+00, -2.6845e-01],
[-3.3574e+00, -2.1445e+00, -6.4501e-02, ..., -5.7042e+00,
-1.3377e+00, 4.2260e-01],
[-3.0611e+00, -1.7038e+00, 2.3046e-01, ..., -4.4800e+00,
-1.3826e+00, 1.9406e-01]],
[[ 2.2770e+00, 2.0793e+00, 2.1363e+00, ..., 1.7677e+00,
1.1883e+00, 5.9474e-01],
[ 2.5136e+00, 1.4217e+00, 1.7379e+00, ..., 5.1777e+00,
2.9937e+00, 1.8819e+00],
[ 6.1592e-01, 2.0645e+00, 8.6215e-01, ..., 5.8682e+00,
3.0108e+00, -2.4634e-01],
...,
[ 6.8334e+00, 7.9268e+00, 7.3922e+00, ..., 4.2162e+00,
5.1770e-02, 3.4516e+00],
[ 1.5665e+00, 4.3591e-01, 3.4256e+00, ..., 2.8588e+00,
1.0554e+00, 1.5298e+00],
[ 2.8249e+00, 2.0168e+00, 3.4662e-01, ..., 1.3812e-01,
1.4305e+00, 1.1022e+00]]],
[[[ 4.3517e-01, 1.2398e+00, -1.6521e-01, ..., -3.9911e+00,
-2.0124e+00, -6.3248e-01],
[ 4.3367e-01, 6.7062e-01, 9.9999e-01, ..., -4.0149e+00,
-3.8817e+00, -4.6825e-01],
[-4.2415e-01, -9.4400e-01, 1.1394e-01, ..., -1.0977e+00,
-6.0412e+00, -1.8236e+00],
...,
[-4.2290e+00, -8.6523e+00, -5.4228e+00, ..., 6.5510e+00,
-1.3357e+00, 3.8043e-01],
[-3.6971e+00, -8.6066e+00, 1.2278e-01, ..., -2.6180e+00,
-3.9935e+00, 1.2888e+00],
[-9.7704e-01, -2.1015e+00, 2.8089e+00, ..., -4.4201e+00,
-3.3393e+00, -1.2669e+00]],
[[-1.4160e+00, -2.3286e+00, -3.7237e+00, ..., -7.8066e+00,
-6.1648e+00, -9.3146e-01],
[-4.7050e+00, -4.9502e+00, -3.6805e+00, ..., -1.1088e+01,
-1.0174e+01, -2.6361e+00],
[-4.7209e+00, -5.0473e+00, -4.4549e+00, ..., -6.6943e+00,
-1.3489e+01, -3.8679e+00],
...,
[ 2.9532e-01, -4.8800e+00, -2.2023e-02, ..., -2.6407e+00,
-6.6197e+00, -3.7624e+00],
[ 1.0200e+00, -5.1709e+00, -8.6553e-01, ..., -1.2841e+01,
-1.0890e+01, -2.5589e+00],
[ 1.5659e+00, 8.7259e-01, 2.3943e-01, ..., -1.9585e+00,
-3.2270e+00, -8.8566e-02]],
[[-1.2953e-02, 1.4226e+00, 1.6603e+00, ..., 1.5330e+00,
2.0998e+00, 7.9990e-01],
[ 3.6513e-02, 3.1277e-03, 8.0993e-01, ..., 1.8320e+00,
2.1750e+00, -5.5741e-01],
[ 6.9430e-01, 1.3215e+00, 1.6309e+00, ..., 4.4131e+00,
2.1489e+00, -5.9142e-01],
...,
[-3.7472e-01, -1.5149e+00, -5.1900e+00, ..., -1.3051e+01,
1.9114e+00, 1.2709e-01],
[ 3.1765e-01, 1.7399e+00, -8.2456e-02, ..., -9.6340e+00,
-1.6884e+00, -2.3185e+00],
[-1.0620e+00, 1.6312e+00, -3.6948e-01, ..., -1.5620e+00,
-3.3737e+00, -2.9876e+00]],
...,
[[-7.5597e-01, -1.8518e+00, -2.1461e+00, ..., -5.7626e+00,
-4.7686e+00, -2.4626e-01],
[-3.0380e+00, -2.8521e+00, -1.2480e+00, ..., -3.2348e+00,
-7.5040e+00, -2.4324e+00],
[-2.4359e+00, -1.7227e+00, -9.6102e-01, ..., -5.8285e+00,
-5.5379e+00, -4.8023e+00],
...,
[-1.0019e+00, -2.4690e+00, -5.3436e-01, ..., 1.3146e+00,
-9.9665e+00, -1.3864e+00],
[-1.8118e-01, -3.3687e+00, -1.7176e+00, ..., -8.7923e-01,
-8.4005e+00, -8.2141e-03],
[ 1.3146e+00, 1.9739e+00, 1.0295e+00, ..., -6.4188e-01,
-3.9920e+00, -6.6003e-01]],
[[-1.6260e+00, -2.5799e+00, -3.8984e+00, ..., -1.6895e+00,
-4.6133e-01, -7.1268e-01],
[-4.2970e+00, -3.4825e+00, -4.6843e+00, ..., -2.9042e+00,
-1.8779e+00, -6.1421e-01],
[-4.9839e+00, -3.9068e+00, -4.6672e+00, ..., 8.7096e-02,
-1.2199e+00, 1.6954e-01],
...,
[-7.9078e+00, -5.7791e+00, -2.4477e+00, ..., 2.0569e+00,
-4.6961e+00, 2.7114e+00],
[-7.5711e+00, -6.6591e+00, -1.9917e+00, ..., -2.3135e-01,
-1.7465e+00, 2.6808e+00],
[-1.4050e+00, -2.0858e+00, 4.4600e-01, ..., 2.2760e+00,
9.8278e-01, 1.4328e+00]],
[[ 1.3442e+00, 7.4665e-02, 1.1511e+00, ..., 3.1041e-01,
5.8826e-01, -1.5205e+00],
[ 2.6181e+00, 1.5006e+00, 1.1797e+00, ..., 4.0463e+00,
2.1024e+00, 1.5644e+00],
[ 1.5788e+00, 8.3178e-01, 2.6273e-01, ..., 5.8308e+00,
2.3099e+00, -4.2726e-01],
...,
[-5.2754e+00, -9.0322e+00, -6.1809e+00, ..., 6.6057e+00,
2.7695e+00, 2.4999e+00],
[-4.3404e+00, -9.6304e+00, -3.5005e+00, ..., 7.6682e+00,
1.9835e+00, 2.5927e+00],
[-9.3150e-01, -4.1475e+00, -1.6750e-01, ..., -6.3174e-01,
1.0554e+00, 1.2051e+00]]],
[[[-3.9884e-01, 7.3694e-01, 1.1671e+00, ..., -6.0028e-01,
-2.2754e+00, -1.2555e+00],
[-2.8974e-02, -1.3488e-01, 3.6877e+00, ..., 1.7440e+00,
-4.4711e-01, -8.1847e-01],
[-1.2231e+00, 2.3384e-01, 2.6196e+00, ..., -2.0279e+00,
-5.6618e+00, -5.0336e+00],
...,
[-5.4203e-01, -9.2377e-02, 2.5739e+00, ..., -1.5324e+00,
-4.1828e-01, 2.3629e+00],
[-1.5762e+00, 2.7699e-02, 1.2815e+00, ..., -1.8653e+00,
-3.0459e+00, -6.9870e-01],
[-1.1539e+00, -2.1378e+00, -1.1954e+00, ..., -3.7137e+00,
-3.5749e+00, -1.7064e+00]],
[[-1.7203e+00, -2.4225e+00, -2.9730e+00, ..., -5.9277e+00,
-9.0034e+00, -3.2888e+00],
[-4.8337e+00, -5.1448e+00, -4.9410e+00, ..., -4.9719e+00,
-7.7899e+00, -4.3191e+00],
[-3.9249e+00, -5.4656e+00, -7.2569e+00, ..., -6.1350e+00,
-1.0110e+01, -3.5429e+00],
...,
[-4.5092e+00, -5.1366e+00, -5.4814e+00, ..., -6.0168e+00,
-4.8507e+00, -1.7349e-01],
[-7.5520e+00, -7.6440e+00, -7.0386e+00, ..., -8.5572e+00,
-1.2325e+01, -7.8407e+00],
[-1.3830e+00, -3.0457e+00, -2.1940e+00, ..., -3.3177e+00,
-3.1168e+00, -3.8779e-01]],
[[-2.9977e-01, 2.0997e+00, 1.4696e+00, ..., 2.5035e+00,
2.7723e+00, 3.2254e-01],
[-5.7800e-01, 1.5415e-01, 3.7212e-01, ..., 1.2673e+00,
2.3844e+00, -2.4303e-01],
[-3.1273e-02, 2.2950e+00, 5.9035e-01, ..., 1.8825e+00,
5.0018e+00, 3.9425e+00],
...,
[-7.7692e-01, 4.8489e-01, 5.7936e-01, ..., -5.3797e+00,
-4.5046e+00, -4.1786e+00],
[-1.4914e+00, -6.6757e-02, 1.2668e+00, ..., -1.4953e-01,
2.2204e-01, -2.1439e+00],
[-6.3869e-01, -4.1447e-01, 3.5293e-01, ..., -9.2373e-01,
8.0336e-02, -1.1671e+00]],
...,
[[-1.4560e+00, -2.0676e+00, -5.8637e-01, ..., -3.3315e+00,
-5.7751e+00, -1.2988e+00],
[-3.6199e+00, -3.3677e+00, -9.3030e-01, ..., -1.6928e+00,
-5.8056e+00, -3.2007e+00],
[-2.7049e+00, -2.3425e+00, -5.2550e-01, ..., -4.7808e+00,
-7.0994e+00, -3.6617e+00],
...,
[-9.2224e-01, -2.4342e+00, -2.0213e-01, ..., -1.7442e+00,
-6.4875e+00, -1.1430e+00],
[-1.9133e+00, -3.8552e+00, -9.2016e-01, ..., -2.7497e+00,
-7.5996e+00, -3.3876e+00],
[-1.2019e+00, -1.2762e+00, 3.3600e-01, ..., -3.4883e-01,
-1.0061e+00, 1.0187e+00]],
[[-1.8206e+00, -1.6584e+00, -1.9334e+00, ..., -2.7304e+00,
-1.5615e+00, 1.0655e-02],
[-3.8918e+00, -1.7312e+00, -2.6637e+00, ..., -2.0953e+00,
-2.1876e+00, -1.7022e-01],
[-3.6037e+00, -3.1303e+00, -4.9884e+00, ..., -3.0648e+00,
-5.4143e+00, -2.4749e+00],
...,
[-2.6214e+00, -2.2503e+00, -3.5029e+00, ..., 3.0318e+00,
5.0050e-01, 8.7201e-01],
[-1.7805e+00, 1.2501e-01, -1.3205e+00, ..., 3.2684e-01,
-1.9725e+00, 1.0993e+00],
[-1.9630e+00, -1.1690e+00, -6.1470e-01, ..., 5.5344e-01,
-1.2030e-01, 3.3315e-01]],
[[ 1.2510e+00, 1.7873e-01, 6.5274e-01, ..., 1.9375e+00,
2.0221e+00, 6.6379e-01],
[ 4.0842e+00, 2.8710e+00, 2.8158e+00, ..., 2.8523e+00,
3.1489e+00, 3.5309e+00],
[ 1.6519e+00, 5.3709e-01, 1.9624e+00, ..., 2.0921e+00,
1.7730e+00, 1.2445e+00],
...,
[ 2.8558e+00, 2.7487e+00, 3.1310e+00, ..., 7.0478e+00,
2.9225e+00, 1.1937e+00],
[ 3.7098e+00, 3.7707e+00, 3.0299e+00, ..., 4.1259e+00,
2.1004e+00, 2.9591e+00],
[ 2.3459e+00, 2.2793e+00, 1.5662e+00, ..., 4.2131e-01,
-7.5791e-01, 9.3017e-01]]],
[[[-3.3211e+00, -6.8824e-01, -6.7085e+00, ..., -1.6832e+00,
-1.4740e+00, -1.2684e+00],
[-1.2384e+00, -5.2380e+00, -7.5574e+00, ..., -9.0888e-01,
-1.2756e-01, -1.7000e+00],
[-2.4917e+00, -3.1918e+00, -2.3852e+00, ..., -1.3017e+00,
-1.2204e+00, -2.5399e+00],
...,
[-1.2877e+00, -7.2723e+00, -4.0874e+00, ..., -7.6670e+00,
-4.4121e+00, -1.0159e+00],
[-2.0293e+00, -2.1340e+00, -4.9729e-01, ..., -4.1452e+00,
-3.2346e+00, 2.9886e-01],
[-2.0486e+00, -3.5681e+00, -4.4650e+00, ..., -5.5950e+00,
-3.3499e+00, -1.4010e+00]],
[[ 1.0648e+00, 9.9066e-01, -1.5630e+00, ..., -3.8016e+00,
-5.5341e+00, -3.2518e+00],
[ 1.1450e+00, -5.0400e+00, -4.7304e+00, ..., -3.8673e+00,
-5.6392e+00, -5.0203e+00],
[-3.0289e+00, 5.2276e-01, 4.6733e+00, ..., -4.7704e+00,
-4.6322e+00, -3.2259e+00],
...,
[-5.0319e+00, -8.5336e+00, -4.0116e+00, ..., -1.2731e+01,
-8.7752e+00, -1.9054e+00],
[-8.5187e+00, -8.9054e+00, -1.3160e+01, ..., -1.0024e+01,
-1.0019e+01, -3.7357e+00],
[-2.3246e+00, -4.4105e+00, -2.3212e+00, ..., -3.6346e+00,
-4.1648e+00, -9.0042e-01]],
[[-1.9834e-01, -2.4593e+00, -2.7820e+00, ..., 1.6070e+00,
2.2031e+00, 8.5983e-01],
[-4.4245e+00, -4.0579e+00, -3.6787e+00, ..., 1.4867e+00,
2.8150e+00, 1.1823e+00],
[-1.0121e+01, -6.2068e+00, -2.6468e+00, ..., 1.1859e+00,
2.4362e+00, 2.0695e+00],
...,
[ 6.8506e-02, -1.2811e-02, -6.3423e+00, ..., 2.0997e+00,
2.0132e-01, -6.2202e-01],
[-7.8597e-01, 1.0148e+00, -3.7071e-01, ..., -1.7879e+00,
-1.1622e-02, 7.5456e-02],
[ 9.2988e-02, 4.2411e-01, 8.0142e-01, ..., -9.3531e-01,
-1.2112e-01, -1.5635e+00]],
...,
[[ 1.8028e-01, -3.7356e-01, -9.9731e-01, ..., -3.2805e+00,
-4.2021e+00, -1.4331e+00],
[-1.6867e+00, -9.2082e+00, -6.4648e+00, ..., -3.1366e+00,
-5.0934e+00, -3.6443e+00],
[-1.8644e+00, -3.6104e+00, 1.7571e+00, ..., -2.7759e+00,
-4.7390e+00, -3.4185e+00],
...,
[-2.9365e+00, -6.2869e+00, -3.2096e-01, ..., -8.5580e+00,
-6.7638e+00, -4.2421e+00],
[-2.6852e+00, -5.4663e+00, -5.2930e+00, ..., -3.5360e+00,
-5.7878e+00, -2.0179e+00],
[-2.7953e+00, -2.0048e+00, -1.1772e-01, ..., -3.7849e+00,
-3.2674e+00, 9.5685e-02]],
[[-4.2155e+00, -7.4284e+00, -7.7109e+00, ..., -3.2318e+00,
-2.8361e+00, -2.3618e-01],
[-6.2943e+00, -1.0198e+01, -8.5572e+00, ..., -3.9888e+00,
-5.8201e+00, -1.8136e+00],
[-1.1735e+00, -3.8581e+00, -5.3710e+00, ..., -2.9990e+00,
-4.7265e+00, -1.7333e+00],
...,
[-4.2171e+00, -1.8984e+00, 7.5933e-01, ..., -7.7391e+00,
-2.6249e+00, -1.3944e-01],
[-2.8884e+00, -9.1141e-01, -1.4856e+00, ..., -1.8776e+00,
-8.3606e-01, 9.0084e-01],
[-3.5199e+00, -1.4607e+00, 8.9306e-01, ..., -6.0637e-01,
-6.6998e-01, 3.1958e-01]],
[[ 1.6202e+00, -7.2168e-01, -5.5568e+00, ..., 5.4304e-01,
1.6048e+00, 6.8515e-01],
[ 2.9267e+00, -2.4176e+00, -7.2294e+00, ..., 1.2000e+00,
2.4965e+00, 3.3425e+00],
[ 1.7540e+00, -3.4836e+00, -6.1324e+00, ..., 1.6184e+00,
1.2718e+00, 1.6179e-01],
...,
[ 2.7123e+00, -9.8829e-01, 6.8799e-01, ..., 1.8381e+00,
1.8466e+00, 8.0991e-01],
[ 4.3274e+00, 3.2703e+00, 6.6041e+00, ..., 2.2297e+00,
2.5371e+00, 1.3564e+00],
[ 3.1525e+00, 2.6548e+00, 1.1460e+00, ..., 7.5114e-01,
-2.7979e-01, 7.7520e-01]]],
[[[-4.3344e-01, 8.1465e-01, 1.1633e+00, ..., -9.7971e-01,
-4.9264e-02, -1.2569e-01],
[-3.4264e-02, 2.3538e-01, 2.7996e+00, ..., -3.9708e-02,
1.0832e-01, -5.7959e-01],
[-1.0425e+00, 5.5511e-01, 1.8265e+00, ..., 1.1045e+00,
-8.9887e-01, -1.4447e+00],
...,
[-3.0753e-01, 4.7222e-01, -2.3268e+00, ..., -5.9190e-01,
4.4387e-01, -3.8628e-01],
[-1.6554e+00, 4.9485e-01, -4.3060e-01, ..., 3.0074e-01,
2.7337e-01, -1.2409e+00],
[-1.3122e+00, -2.4052e+00, -2.2668e+00, ..., -2.9550e+00,
-2.9131e+00, -2.8058e+00]],
[[-1.7716e+00, -2.3452e+00, -3.1202e+00, ..., -4.9282e+00,
-4.1480e+00, -2.1134e+00],
[-4.5300e+00, -4.8670e+00, -4.1113e+00, ..., -4.7380e+00,
-5.5878e+00, -3.2361e+00],
[-3.5415e+00, -4.9757e+00, -5.8752e+00, ..., -3.4207e+00,
-5.9451e+00, -3.0329e+00],
...,
[-4.7811e+00, -5.9798e+00, -7.2176e+00, ..., -2.4583e+00,
-3.3950e+00, -2.6847e+00],
[-7.6688e+00, -8.1650e+00, -6.2634e+00, ..., -4.9537e+00,
-5.3673e+00, -5.0999e+00],
[-1.3308e+00, -3.0621e+00, -2.2519e+00, ..., -3.0695e+00,
-2.8203e+00, -1.6221e+00]],
[[-2.2067e-01, 2.1832e+00, 1.4102e+00, ..., 1.4378e+00,
1.8722e+00, 8.4080e-01],
[-5.2820e-01, 2.1331e-01, 4.3610e-01, ..., 2.1417e+00,
1.3076e+00, 7.2684e-01],
[ 4.6398e-02, 2.3080e+00, 1.6991e+00, ..., 3.7888e+00,
1.6660e+00, 1.1146e+00],
...,
[-7.5881e-01, -2.5527e-01, 1.7687e+00, ..., 7.6274e-01,
1.4282e+00, 5.9914e-01],
[-1.2381e+00, -4.2471e-01, 8.2119e-02, ..., 1.6720e+00,
1.1561e+00, 3.9674e-01],
[-4.6280e-01, -1.5836e-01, -3.6765e-01, ..., 1.8915e+00,
1.6410e+00, 5.1796e-01]],
...,
[[-1.5008e+00, -1.9820e+00, -1.2779e+00, ..., -1.4631e+00,
-3.0877e+00, -4.1444e-01],
[-3.4640e+00, -3.4248e+00, -9.0741e-01, ..., -2.5737e+00,
-4.8985e+00, -3.4680e+00],
[-2.5308e+00, -2.1264e+00, -1.6767e+00, ..., -3.5244e+00,
-3.9495e+00, -2.9994e+00],
...,
[-1.2258e+00, -3.5822e+00, -4.2816e+00, ..., -2.4161e+00,
-2.5540e+00, -9.4406e-01],
[-2.2470e+00, -4.0253e+00, -1.5111e+00, ..., -2.0477e+00,
-2.9335e+00, -1.5880e+00],
[-1.3144e+00, -2.0400e+00, -1.4354e+00, ..., -2.0557e+00,
-2.1655e+00, -5.9146e-01]],
[[-1.8798e+00, -1.6673e+00, -2.2468e+00, ..., -5.2856e-01,
-1.4881e+00, -6.5265e-01],
[-3.9389e+00, -1.8656e+00, -2.3823e+00, ..., -1.4546e-01,
-3.5386e+00, -1.9392e+00],
[-3.6071e+00, -3.0405e+00, -5.9628e+00, ..., -1.4112e+00,
-2.9542e+00, -2.7887e+00],
...,
[-2.3286e+00, -4.4060e+00, -6.8574e+00, ..., -6.1145e-01,
-2.9964e+00, -2.1222e+00],
[-1.5587e+00, 1.9738e-02, -2.4680e+00, ..., -4.0512e-01,
-3.3310e+00, -1.9586e+00],
[-1.7940e+00, -1.0880e+00, -1.1743e+00, ..., -1.5329e+00,
-2.7519e+00, -1.2833e+00]],
[[ 1.2609e+00, 2.8089e-01, 7.8587e-01, ..., 5.9980e-01,
9.2010e-01, 7.0794e-01],
[ 3.8489e+00, 2.9313e+00, 2.3123e+00, ..., 2.9649e+00,
1.1584e+00, 2.2450e+00],
[ 1.5504e+00, 7.3151e-01, 1.4268e+00, ..., 3.2628e+00,
1.8541e-01, 5.8464e-01],
...,
[ 2.6774e+00, 3.1045e+00, 2.7398e+00, ..., 1.5502e+00,
2.8547e-01, 5.8938e-01],
[ 3.4323e+00, 4.7179e+00, 2.9846e+00, ..., 1.2434e+00,
7.3869e-01, 1.2643e+00],
[ 2.1434e+00, 2.2848e+00, 1.0477e+00, ..., -2.4842e-01,
-3.1059e-01, -1.6425e-01]]]], device='cuda:0'),
tensor([[[[ 4.1265e-01, -1.0141e+01, -1.1347e+01, ..., -3.4539e+00,
-2.0359e+00, -8.4420e-01],
[-2.7651e+00, -7.7872e+00, 1.0031e-01, ..., -1.5960e+00,
-2.1658e+00, -1.1627e-01],
[-4.5369e+00, -4.8288e+00, -1.4798e+00, ..., -1.9293e+00,
-2.7638e+00, 1.3428e-03],
...,
[-8.3135e-02, -1.8952e+00, -2.7628e+00, ..., 5.9733e-01,
2.2424e+00, 6.8431e-01],
[-6.0210e-01, -6.7724e-01, 7.8594e-01, ..., 3.1704e-01,
1.0698e+00, -3.2930e-01],
[ 1.7698e-01, -1.8631e+00, -1.7375e+00, ..., -2.5969e-02,
5.5194e-01, -7.4946e-01]],
[[-5.3318e+00, -1.5886e+01, -1.2494e+01, ..., -6.3339e+00,
-6.5329e+00, -1.0985e+00],
[-8.9341e+00, -5.5055e+00, 4.8722e+00, ..., -6.7991e+00,
-8.5676e+00, -2.6535e+00],
[-4.9464e+00, -9.9368e-01, -8.8432e-01, ..., -4.5406e+00,
-7.0927e+00, -2.5130e+00],
...,
[-3.2189e+00, -4.5987e+00, -5.0564e+00, ..., 3.3608e-01,
-3.4952e+00, -2.0970e+00],
[-4.8999e+00, -5.3544e+00, -4.2829e+00, ..., 3.6531e-01,
-5.1505e+00, -5.0457e+00],
[-4.5932e-01, -2.0193e+00, -3.2365e+00, ..., 1.0139e+00,
-4.9278e-02, -6.5623e-01]],
[[ 2.3688e-01, 2.9738e+00, 3.8055e+00, ..., 1.5443e+00,
2.3679e+00, 1.0737e+00],
[ 1.6645e+00, 1.5491e+00, -5.6602e+00, ..., 1.5355e+00,
2.7713e+00, -2.0839e-03],
[ 2.3644e+00, -2.2484e+00, -9.4020e+00, ..., 2.3674e+00,
2.8048e+00, -1.5318e-01],
...,
[ 4.3203e-01, 1.0577e+00, 1.2704e+00, ..., -5.6833e+00,
1.8946e+00, 7.7705e-02],
[-6.3355e-01, 3.1977e-01, -1.2403e+00, ..., -6.9574e+00,
2.1576e+00, 2.4143e-02],
[-5.7943e-01, 1.1146e+00, -4.2204e-01, ..., -5.8266e+00,
-6.9872e-01, -1.1555e+00]],
...,
[[-3.9693e+00, -8.8018e+00, -6.1392e+00, ..., -4.9457e+00,
-4.9351e+00, -6.7076e-01],
[-5.9714e+00, -1.7231e+00, 4.1113e+00, ..., -3.2064e+00,
-4.9590e+00, -1.9832e+00],
[-4.1578e+00, -1.4502e+00, -3.0416e+00, ..., -3.8925e+00,
-4.3421e+00, -1.4897e+00],
...,
[-5.3105e-01, -1.6424e+00, -1.1338e+00, ..., 4.9630e+00,
-4.6946e+00, -1.3308e+00],
[-1.1420e+00, -1.8335e+00, 3.5833e-01, ..., 4.0681e+00,
-5.2112e+00, -7.1099e-01],
[ 3.0596e-01, -1.1383e+00, -1.6625e+00, ..., 1.6480e+00,
-3.7435e+00, 1.2470e+00]],
[[-6.0071e+00, -7.1097e+00, -3.9204e+00, ..., -1.4097e+00,
-4.5063e-01, -5.9935e-01],
[-4.5766e+00, -7.4264e-01, -2.1217e+00, ..., -2.7415e+00,
-1.9043e+00, -4.5521e-01],
[-1.3694e+00, 1.6560e+00, -4.5043e+00, ..., 9.1042e-02,
-3.0037e-01, -2.2087e-01],
...,
[-3.6053e+00, -3.8804e+00, -2.7756e+00, ..., 6.6489e-01,
-1.3974e+00, 9.8586e-01],
[-3.1341e+00, -1.2304e+00, 3.9618e-01, ..., -4.0386e-01,
-1.6760e+00, 2.4928e+00],
[-2.1243e+00, -3.1174e+00, -1.6700e+00, ..., 1.3515e+00,
-2.6188e+00, 1.8232e+00]],
[[ 4.5546e+00, 2.3244e-01, -4.2352e-01, ..., 4.3890e-01,
9.4006e-01, -1.2739e+00],
[ 3.5856e+00, -8.9885e-01, 1.7012e+00, ..., 3.8720e+00,
3.1323e+00, 2.3225e+00],
[ 9.0710e-01, 1.8560e+00, -1.4801e+00, ..., 2.9007e+00,
1.5737e+00, 1.2589e+00],
...,
[ 1.5657e+00, 1.0805e+00, -4.0064e-01, ..., 5.9861e-01,
4.5553e+00, 1.3475e+00],
[ 2.0310e+00, 2.3093e+00, 1.2891e+00, ..., -4.7051e-01,
3.1512e+00, 2.8565e+00],
[ 1.5202e+00, 7.0318e-01, 7.6380e-01, ..., -1.0601e+00,
1.0069e+00, 1.8621e+00]]],
[[[ 1.2434e+00, 2.1231e+00, 8.2956e-01, ..., -3.7115e+00,
-2.4095e+00, -8.3927e-01],
[-3.6936e-01, -3.0262e-01, -5.5540e-01, ..., -3.6269e+00,
-2.7289e+00, -4.4437e-02],
[ 1.1252e+00, 1.5733e+00, -9.8260e-01, ..., -3.4263e+00,
-3.0904e+00, 2.7719e-01],
...,
[ 7.1421e-01, -1.8733e+00, -4.6068e+00, ..., -3.2531e+00,
-3.6510e+00, 7.1672e-01],
[-2.9230e-01, 9.6359e-02, -9.2198e-01, ..., -4.9755e+00,
-3.7130e+00, 5.4180e-01],
[-5.8844e-01, -3.8280e+00, -4.2051e+00, ..., -3.9744e+00,
-3.3467e+00, -1.4886e+00]],
[[ 1.0474e+00, -1.2035e+00, -4.2807e+00, ..., -7.3452e+00,
-6.9776e+00, -1.1188e+00],
[-3.8106e+00, -2.8827e+00, -2.7560e+00, ..., -8.1701e+00,
-7.9141e+00, -2.2703e+00],
[-1.5890e+00, -3.6976e+00, -2.7730e+00, ..., -9.5385e+00,
-7.4911e+00, -2.0020e+00],
...,
[-5.1201e+00, -9.5028e+00, -9.5392e+00, ..., -8.7725e+00,
-8.4710e+00, -2.1913e+00],
[-3.7559e+00, -4.6583e+00, -4.7206e+00, ..., -9.9863e+00,
-1.0851e+01, -4.7822e+00],
[-7.7726e-01, -3.9470e+00, -4.3716e+00, ..., -4.0102e+00,
-3.8783e+00, -8.0343e-01]],
[[ 4.9491e-01, 2.4564e+00, 2.1677e+00, ..., 6.6465e-01,
2.1745e+00, 9.6373e-01],
[ 4.8160e-02, 3.1045e+00, 3.1553e+00, ..., 8.9568e-01,
2.7085e+00, 2.7419e-02],
[ 3.8182e-01, 3.2441e+00, 4.2477e+00, ..., -4.7920e-01,
1.5849e+00, -1.7741e-01],
...,
[ 9.1181e-01, 9.6906e-01, 1.3599e+00, ..., -1.0450e+00,
1.0567e+00, -8.4403e-01],
[-1.4130e-01, 1.1995e+00, 9.6582e-01, ..., -2.3893e+00,
1.1230e+00, -1.2916e+00],
[-2.1220e-03, 1.4775e+00, -8.0708e-01, ..., 2.7750e-01,
-1.8614e-01, -1.8954e+00]],
...,
[[-2.7082e-01, -9.8608e-01, -2.0927e+00, ..., -5.3283e+00,
-5.2199e+00, -5.6292e-01],
[-3.2047e+00, -4.5247e+00, -3.5854e+00, ..., -5.1551e+00,
-5.0003e+00, -1.7685e+00],
[ 8.9175e-01, -1.5275e+00, -2.5502e+00, ..., -6.6128e+00,
-6.0283e+00, -8.0992e-01],
...,
[-5.1468e+00, -7.3752e+00, -7.2464e+00, ..., -2.1751e+00,
-5.4561e+00, -2.2790e+00],
[-9.3240e-01, -1.4361e+00, -1.7502e+00, ..., -3.3202e+00,
-5.9719e+00, -1.2559e+00],
[-1.0895e+00, -3.5841e+00, -2.7970e+00, ..., -2.0308e+00,
-2.3583e+00, 3.0573e-01]],
[[-9.5615e-01, -8.6854e-01, -8.8943e-01, ..., -2.0508e+00,
-4.5518e-01, -5.8689e-01],
[-4.2050e+00, -4.3325e+00, -8.4519e-01, ..., -3.8447e+00,
-2.0063e+00, -2.9342e-01],
[-2.6909e+00, -5.0604e+00, -4.3841e+00, ..., -2.6692e+00,
-2.1574e+00, -5.1493e-01],
...,
[-2.1010e+00, -8.9449e+00, -7.4401e+00, ..., -3.6186e+00,
-2.0119e+00, -6.7407e-01],
[-1.5386e+00, -1.4127e+00, -1.9573e+00, ..., -1.7556e+00,
-3.7073e-01, 1.2305e+00],
[-2.6725e+00, -2.6540e+00, -6.3074e-01, ..., -4.1873e-03,
-8.1924e-01, 5.6897e-01]],
[[ 6.6594e-01, 1.3297e+00, 1.9648e+00, ..., -4.8083e-01,
4.5753e-01, -1.4064e+00],
[ 7.5716e-01, 2.6423e+00, 2.9605e+00, ..., 2.7085e+00,
2.5685e+00, 2.1878e+00],
[-5.8907e-01, -6.0795e-01, 2.3323e+00, ..., 2.0874e+00,
1.4932e+00, 1.3798e+00],
...,
[ 4.6917e+00, 5.3796e+00, 4.2095e+00, ..., 2.3571e+00,
2.7920e+00, 4.1656e-01],
[ 2.3904e+00, 2.5051e+00, 1.3120e+00, ..., 2.6906e+00,
8.0796e-01, 1.6456e+00],
[ 1.8521e+00, 1.2854e+00, 1.3775e+00, ..., -6.9597e-02,
-1.2863e-01, 9.0599e-01]]],
[[[ 1.0197e+00, 2.8907e+00, 1.6697e+00, ..., -2.6449e+00,
-4.5988e-01, 7.0140e-01],
[ 1.2079e+00, -5.9705e-01, -2.0481e+00, ..., -1.6433e-01,
-1.9684e+00, -1.2017e-01],
[ 4.6714e-01, -2.3801e+00, -1.6445e+00, ..., -1.5724e-02,
-3.7268e+00, -1.9504e+00],
...,
[-1.0829e+00, -1.7480e+00, 5.6513e-01, ..., 4.9991e+00,
-2.3231e+00, -3.3274e+00],
[-9.6941e-01, -9.8767e-01, -3.4219e-01, ..., -1.8963e+00,
-1.5534e+00, -2.7783e+00],
[-2.8192e+00, -5.3480e+00, -4.6387e+00, ..., -5.0040e+00,
-5.5784e+00, -4.6875e+00]],
[[-1.9308e-01, -3.0105e+00, -5.9450e+00, ..., -4.8879e+00,
-5.4800e+00, -6.6480e-02],
[-4.2233e+00, -9.4565e+00, -9.2190e+00, ..., -5.6395e+00,
-1.2637e+01, -2.6757e+00],
[-4.2756e+00, -1.0743e+01, -4.9465e+00, ..., -6.0464e+00,
-1.3075e+01, -3.4067e+00],
...,
[-6.0222e+00, -5.8886e+00, -4.0377e+00, ..., -5.7224e+00,
-1.0676e+01, -6.8145e+00],
[-7.1428e+00, -8.0553e+00, -6.1884e+00, ..., -1.0650e+01,
-7.9965e+00, -6.8424e+00],
[-2.5921e+00, -4.8101e+00, -4.1279e+00, ..., -4.1659e+00,
-4.9246e+00, -2.5488e+00]],
[[-1.5259e-01, 2.0980e+00, -1.6400e-01, ..., 1.8173e+00,
2.2954e+00, 7.9973e-01],
[-7.3883e-01, 4.2184e-01, 1.1502e+00, ..., 3.9515e+00,
7.1025e-01, -1.3517e+00],
[-7.3275e-01, 5.5250e-01, 1.5767e+00, ..., 6.5557e+00,
3.0606e+00, -2.1082e-01],
...,
[ 8.9966e-01, 1.1967e+00, 8.3209e-01, ..., -2.4170e+00,
4.1356e+00, 9.7487e-01],
[ 8.0366e-01, 7.7278e-01, 7.8667e-01, ..., 7.2361e-01,
8.8267e-01, -1.1095e+00],
[ 1.3945e+00, 1.6483e+00, 1.1702e+00, ..., 1.7375e+00,
-4.8479e-01, 6.2007e-02]],
...,
[[-2.8359e-02, -1.2585e+00, -2.4480e+00, ..., -3.1176e+00,
-4.5412e+00, -7.1543e-01],
[-2.5355e+00, -4.7454e+00, -6.7423e+00, ..., -6.2148e+00,
-5.3901e+00, -5.3936e+00],
[-1.9298e+00, -2.2171e+00, -2.3943e+00, ..., -1.0682e+01,
-4.4349e+00, -6.2130e+00],
...,
[-2.9053e+00, -2.7401e+00, -1.0535e+00, ..., -5.8812e+00,
-7.6457e+00, -1.5794e+00],
[-2.9156e+00, -3.2109e+00, -2.0488e+00, ..., -3.9099e+00,
-3.6918e+00, -2.1958e+00],
[-3.7139e+00, -4.1158e+00, -3.4290e+00, ..., -2.0092e+00,
-4.6564e+00, -2.2330e+00]],
[[-1.8231e+00, -1.3899e+00, -2.5388e+00, ..., 2.7734e-02,
-1.2878e+00, -1.2750e+00],
[-4.3687e+00, -2.5874e+00, -6.6014e+00, ..., -6.9375e-01,
-1.1206e+00, -4.0540e-01],
[-4.3021e+00, -1.5538e+00, -1.1210e+00, ..., -5.3155e+00,
-2.2382e-01, 3.5490e-01],
...,
[-3.9194e+00, -2.0238e+00, -2.0947e+00, ..., -4.1878e+00,
-9.1517e+00, -3.4700e+00],
[-3.6702e+00, -5.7724e-01, -1.8266e+00, ..., -2.6483e+00,
-6.3387e+00, -1.5804e+00],
[-4.6151e+00, -3.0394e+00, -2.7414e+00, ..., -2.8552e+00,
-3.9094e+00, -8.4098e-01]],
[[ 7.9400e-01, 9.3811e-01, 1.4735e+00, ..., 2.7003e-01,
2.4884e-01, -9.3693e-01],
[ 3.1453e+00, 3.5833e+00, 1.4340e+00, ..., 5.7442e+00,
2.9596e+00, 7.7803e-01],
[ 1.7353e+00, 3.5315e+00, 1.4864e+00, ..., 6.4561e+00,
1.6996e+00, -6.3662e-01],
...,
[ 2.8904e+00, 1.3051e+00, 1.5447e+00, ..., 6.2815e+00,
3.3470e+00, 9.4812e-01],
[ 3.8628e+00, 3.6616e+00, 2.3827e+00, ..., 2.9118e+00,
1.7694e+00, 2.5972e+00],
[ 2.0080e+00, 8.0621e-01, 1.7233e-01, ..., 8.3208e-01,
1.1932e+00, 3.6538e-01]]],
[[[-8.4352e-02, -2.9339e+00, -3.3298e+00, ..., -3.0420e+00,
-1.8386e+00, -1.1703e+00],
[-3.1618e+00, -3.6592e+00, 2.7454e+00, ..., -3.6171e+00,
-2.6960e+00, -5.9409e-01],
[-7.5814e+00, -3.3628e+00, -1.6063e+00, ..., 1.3857e+00,
2.0532e+00, 4.2403e-01],
...,
[-8.9490e-02, 1.3834e-02, -1.4630e+01, ..., -2.1121e+00,
-4.3940e+00, -1.2879e+00],
[-9.0459e-02, -3.0802e-01, -1.0165e+01, ..., -5.6256e-01,
-1.8136e+00, -9.2648e-01],
[-2.2511e+00, -4.6745e+00, -7.7000e+00, ..., -3.6552e+00,
-4.1589e+00, -1.2347e+00]],
[[-5.5373e+00, -6.0640e+00, -7.3710e+00, ..., -2.2767e+00,
-4.0385e+00, -5.7077e-03],
[-9.0759e+00, -5.0587e+00, -1.3712e+00, ..., -7.3025e+00,
-8.4436e+00, -2.1312e+00],
[-7.3430e+00, 2.8527e-01, -1.1694e+00, ..., -1.3069e+00,
6.2646e-01, 8.2120e-01],
...,
[-6.9884e+00, -3.8584e+00, -2.0263e+01, ..., 2.9275e-02,
2.2377e-01, -2.6422e+00],
[-8.4071e+00, -5.3111e+00, -1.3466e+01, ..., -6.5889e+00,
1.3055e-01, -2.7220e-01],
[-2.7359e+00, -3.4966e+00, -3.6320e+00, ..., -5.3691e+00,
-7.8677e+00, -1.2331e+00]],
[[-7.4614e-01, 2.2390e+00, 1.7362e+00, ..., 3.8642e+00,
2.1188e+00, 1.0926e+00],
[ 2.2989e+00, 1.5974e-01, -2.0901e-01, ..., 6.2049e+00,
2.4342e+00, -1.3506e-01],
[ 4.3359e+00, -7.0471e-01, -2.9413e+00, ..., -1.6496e+00,
-3.2543e+00, -4.7445e+00],
...,
[-1.4785e-01, -1.2247e+00, 2.4428e+00, ..., 7.8256e-01,
2.1125e+00, 8.7091e-01],
[-2.4844e-01, -1.1792e+00, -8.3930e-01, ..., -4.0336e+00,
3.5602e-01, 1.6284e+00],
[ 6.0407e-01, 3.0290e-01, 1.5646e+00, ..., -2.1146e+00,
-7.3443e-01, 1.1637e+00]],
...,
[[-2.7193e+00, -4.0562e+00, -3.1379e+00, ..., -2.9424e+00,
-2.2401e+00, 2.1606e-01],
[-5.9133e+00, -1.8231e+00, 6.0932e-01, ..., -3.8965e+00,
-1.5265e+00, -1.7425e+00],
[-5.5415e+00, 5.1250e-01, -8.4203e-01, ..., 1.5203e+00,
3.2956e+00, 1.7737e+00],
...,
[-9.8688e-01, -3.3766e+00, -1.0539e+01, ..., -3.8588e+00,
-3.5471e+00, -2.2579e+00],
[-1.3994e+00, -2.6406e+00, -3.8510e+00, ..., -3.4965e+00,
-2.9719e+00, -1.5275e+00],
[-2.8040e+00, -3.8394e+00, -4.7547e+00, ..., -7.0248e+00,
-7.1451e+00, -1.0721e+00]],
[[-1.7119e+00, -2.7519e+00, -7.0501e+00, ..., 1.9446e+00,
5.7461e-02, -8.7063e-01],
[-4.7541e+00, -4.3654e-01, -4.6608e+00, ..., -4.2929e+00,
-1.8008e+00, -1.3903e+00],
[-5.2491e+00, -2.3038e+00, -2.6281e+00, ..., -2.9991e+00,
6.9738e-01, 1.9928e+00],
...,
[-2.6991e+00, -4.0014e+00, -2.3366e+00, ..., -6.4929e+00,
-1.0762e+01, -6.9444e+00],
[-2.2241e+00, -1.6956e+00, -2.6942e+00, ..., 2.7376e+00,
-3.1704e+00, -3.7679e+00],
[-3.0904e+00, -3.5230e+00, -5.3305e+00, ..., -5.2281e-01,
-8.2873e-01, 2.1723e-01]],
[[ 3.7278e+00, -4.3870e-01, 1.0282e+00, ..., 9.0459e-01,
-7.0466e-01, -1.1327e+00],
[ 5.7001e+00, 2.9233e+00, 4.3815e+00, ..., 6.8332e+00,
4.1710e+00, 2.8650e+00],
[ 1.1508e+00, 1.6862e-01, -3.0973e+00, ..., 1.4236e+00,
5.8822e+00, -1.3609e+00],
...,
[ 2.9882e+00, 3.3750e+00, 1.7072e+00, ..., -1.4402e+00,
-7.3176e+00, -3.3909e+00],
[ 4.6784e+00, 4.5684e+00, 6.8983e-01, ..., 4.7858e+00,
-3.9290e+00, -6.3249e+00],
[ 3.0674e+00, 1.1236e+00, -1.0195e+00, ..., 3.8918e+00,
2.0017e+00, -3.0605e-01]]],
[[[-5.3729e-01, -8.7789e-02, -4.8039e-01, ..., -2.9118e+00,
-2.9060e+00, -1.7467e+00],
[ 1.4921e+00, 4.8957e-01, 1.2507e+00, ..., -1.8192e+00,
-3.6300e-01, -1.3912e+00],
[ 1.7886e+00, 9.8128e-02, -2.2748e+00, ..., -2.3293e+00,
-1.2982e+00, -1.5671e+00],
...,
[ 2.5418e+00, 1.4838e+00, -2.6338e+00, ..., -2.5293e+00,
-4.7929e+00, -3.7779e-01],
[ 1.6503e+00, 5.4090e+00, -5.2121e-01, ..., -2.7201e+00,
-3.4733e+00, 2.2925e-01],
[ 2.4688e+00, 3.1230e+00, -4.5501e+00, ..., -3.3743e+00,
-2.4150e+00, -9.9308e-01]],
[[-1.3994e+00, -3.5011e+00, -5.4189e+00, ..., -4.6612e+00,
-6.8921e+00, -3.3508e+00],
[-3.0390e+00, -5.9094e+00, -6.4816e+00, ..., -5.1072e+00,
-5.8389e+00, -4.1156e+00],
[-3.5562e+00, -7.5780e+00, -1.2785e+01, ..., -5.2256e+00,
-5.6162e+00, -2.6809e+00],
...,
[-6.3936e-01, -6.3680e-02, 9.7674e-02, ..., -1.0630e+01,
-9.2144e+00, -2.3073e+00],
[-6.0040e+00, -9.0764e+00, -2.0713e+01, ..., -1.0015e+01,
-9.4953e+00, -3.4383e+00],
[ 1.1757e+00, -1.3439e+00, -7.7749e+00, ..., -3.5086e+00,
-3.8357e+00, -3.8430e-01]],
[[ 1.1018e+00, 2.3672e+00, 2.1136e+00, ..., 1.7032e+00,
2.6440e+00, 1.0025e+00],
[ 7.3991e-01, 2.2551e+00, 1.5859e+00, ..., 1.5217e+00,
2.7126e+00, 1.1134e+00],
[ 6.9685e-01, 1.6456e+00, 1.1620e+00, ..., 1.6940e+00,
1.4275e+00, 1.5258e+00],
...,
[-4.3300e+00, -6.3736e+00, -1.3507e+01, ..., 1.0401e+00,
2.2965e+00, -3.8374e-01],
[-3.2288e+00, -1.0894e+01, -1.6509e+01, ..., 1.1702e-01,
1.5694e+00, -1.0628e-01],
[-2.6485e+00, -1.0585e+01, -1.2972e+01, ..., -1.7905e+00,
-6.0989e-01, -1.8438e+00]],
...,
[[-1.4559e+00, -2.7823e+00, -3.7779e+00, ..., -4.2254e+00,
-5.0352e+00, -1.8429e+00],
[-2.1237e+00, -3.1917e+00, -3.1717e+00, ..., -4.2400e+00,
-4.3171e+00, -3.0058e+00],
[-8.4888e-01, -3.5316e+00, -6.5176e+00, ..., -3.3510e+00,
-5.2495e+00, -2.6836e+00],
...,
[ 2.7084e+00, -2.0629e+00, 5.3373e+00, ..., -6.4412e+00,
-9.1605e+00, -2.9539e+00],
[ 4.7367e-01, -4.8160e+00, -6.0225e-01, ..., -4.4979e+00,
-6.0058e+00, -1.4716e+00],
[ 3.6269e+00, -1.8770e+00, -2.1271e+00, ..., -2.6422e+00,
-2.4057e+00, 6.9120e-01]],
[[-2.6359e+00, -2.3950e+00, -3.4719e+00, ..., -3.8256e+00,
-3.0964e+00, 8.4718e-02],
[-4.5890e+00, -4.3623e+00, -3.5425e+00, ..., -4.2482e+00,
-5.7956e+00, -1.4123e+00],
[-3.5922e+00, -3.8230e+00, -2.9542e+00, ..., -1.8685e+00,
-4.4709e+00, -1.1119e+00],
...,
[-8.9885e-01, -1.0606e+01, -6.5581e+00, ..., -5.8960e-01,
-7.5630e-01, -1.4334e+00],
[ 2.6952e+00, -1.9912e+00, 5.2843e+00, ..., 1.3817e-01,
4.7400e-01, 1.2300e-01],
[ 4.7352e-01, -1.8775e+00, 3.0383e+00, ..., 1.2358e+00,
3.8287e-01, 2.0070e-01]],
[[ 8.6194e-01, 5.1364e-01, 1.5916e+00, ..., 2.5543e-01,
1.4476e+00, 5.8473e-01],
[ 2.2667e+00, 2.6307e+00, 3.8328e+00, ..., 1.2845e+00,
2.5267e+00, 3.6384e+00],
[ 2.0692e+00, 2.8900e+00, 4.0018e+00, ..., 1.7468e-01,
6.7664e-01, 8.3867e-01],
...,
[ 2.9318e+00, 3.3471e+00, -5.4530e+00, ..., 2.8971e+00,
2.7381e+00, 4.1564e-01],
[ 4.8452e+00, 9.7527e+00, 1.1425e+01, ..., 1.4951e+00,
2.1769e+00, 1.4572e+00],
[ 1.3840e+00, 6.1761e+00, 5.4900e+00, ..., 2.6146e+00,
1.7733e+00, 1.2795e+00]]]], device='cuda:0'),
tensor([[[[ 1.9467e-01, 1.3742e+00, 3.8219e-01, ..., -5.8154e-01,
9.6485e-02, 7.9661e-01],
[ 2.4939e-01, 4.1827e-02, 3.4046e-01, ..., -1.7578e+00,
-1.1829e+00, 8.8311e-01],
[ 1.0236e-01, 3.6093e-01, 1.7405e+00, ..., -9.2495e-01,
-5.0581e-01, 2.1906e-01],
...,
[ 7.0185e-01, -5.2489e+00, -5.5967e+00, ..., 1.0660e+00,
-2.0593e+00, -2.5510e+00],
[-1.7472e+00, -5.9703e+00, -6.0652e+00, ..., 1.6264e+00,
-2.1952e+00, -2.4738e+00],
[-5.7086e-01, -5.2944e+00, -3.3542e+00, ..., 2.7964e-01,
-1.3671e+00, -1.0610e+00]],
[[-9.3232e-01, -2.3211e+00, -4.0608e+00, ..., -3.1305e+00,
-4.6897e+00, -3.1652e-01],
[-4.7883e+00, -5.7100e+00, -5.0747e+00, ..., -5.9242e+00,
-7.6613e+00, -1.8542e+00],
[-3.6046e+00, -5.3784e+00, -7.5484e+00, ..., -9.2476e+00,
-1.0062e+01, -3.8124e+00],
...,
[-5.6684e+00, -1.0115e+01, -4.8393e+00, ..., -1.8790e+00,
-7.7288e+00, -2.5919e+00],
[-7.8158e+00, -9.4424e+00, -4.4983e+00, ..., -2.4057e+00,
-8.6453e+00, -3.8539e+00],
[-9.6444e-01, -3.2239e+00, -6.2309e-01, ..., -1.4779e+00,
-3.1235e+00, -1.1022e-01]],
[[-3.4550e-01, 2.0068e+00, 1.1762e+00, ..., 8.5613e-01,
2.0869e+00, 8.4688e-01],
[ 4.4318e-02, 8.3824e-01, 8.1963e-01, ..., 8.3830e-01,
1.0715e+00, -6.5089e-01],
[-3.3016e-03, 1.9948e+00, 1.9022e+00, ..., -2.5991e-01,
1.4027e+00, -8.2804e-02],
...,
[ 1.7735e-01, -4.2331e-01, 4.0439e-01, ..., 4.2636e+00,
2.0528e+00, 4.0262e-01],
[ 3.4313e-01, 1.2309e+00, -1.0631e+00, ..., 2.8746e+00,
2.1022e+00, 4.7270e-01],
[-3.8268e-01, 6.4625e-01, -1.2811e+00, ..., 1.4079e+00,
1.7182e+00, -3.8795e-01]],
...,
[[-9.5225e-01, -1.8477e+00, -1.8271e+00, ..., -2.4099e+00,
-2.4630e+00, -3.2524e-01],
[-3.3208e+00, -4.3421e+00, -2.8226e+00, ..., -4.8866e+00,
-4.6102e+00, -2.4663e+00],
[-1.4971e+00, -2.4915e+00, -1.3719e+00, ..., -1.6951e+00,
-6.2675e+00, -1.8422e+00],
...,
[-1.4342e+00, -3.2017e+00, -3.0925e+00, ..., -3.2109e+00,
-4.7228e+00, -3.0329e+00],
[-2.7101e+00, -4.4958e+00, -1.7985e+00, ..., -5.9515e+00,
-4.2685e+00, -2.2729e+00],
[-9.8683e-01, -2.3243e+00, 1.1569e+00, ..., -4.2992e+00,
-1.2628e+00, 2.0906e-01]],
[[-1.7306e+00, -2.0921e+00, -3.0489e+00, ..., -1.6828e+00,
-1.5792e+00, -1.4120e+00],
[-4.5468e+00, -3.2076e+00, -2.7168e+00, ..., -4.6381e+00,
-2.4644e+00, -1.2029e+00],
[-2.8051e+00, -1.7426e+00, -2.8301e+00, ..., -5.8848e-01,
-6.7048e-01, -9.5462e-01],
...,
[-5.5188e+00, -3.7730e+00, -2.2195e+00, ..., -1.4643e+00,
-1.2540e+00, -4.6729e-01],
[-5.1592e+00, -1.5022e+00, -5.9450e-01, ..., -1.6845e+00,
1.2279e-01, -2.6564e-01],
[-2.4460e+00, -1.4038e-01, 8.4123e-01, ..., -2.0186e+00,
5.1528e-01, 7.9063e-01]],
[[ 1.1267e+00, 5.3358e-01, 5.1106e-01, ..., -7.4779e-01,
9.4972e-02, -3.8246e-01],
[ 3.5325e+00, 3.3404e+00, 1.9251e+00, ..., 1.2211e+00,
2.1825e+00, 2.3970e+00],
[ 1.7584e+00, 1.2949e+00, 1.0473e+00, ..., 2.9151e+00,
1.9529e+00, 1.4383e+00],
...,
[ 3.3414e+00, 1.4998e+00, 4.2194e-01, ..., 3.7858e+00,
1.4010e+00, 5.6089e-01],
[ 3.5528e+00, 1.6355e+00, 2.7888e-01, ..., 5.0478e+00,
1.9882e+00, 2.1001e+00],
[ 2.2786e+00, 5.2322e-01, 7.4024e-01, ..., 2.3420e+00,
2.2048e-01, 1.2634e+00]]],
[[[ 1.3006e+00, 2.6124e+00, -2.9070e+00, ..., 1.0798e+00,
-1.2056e+00, -2.3956e+00],
[ 3.4094e+00, 2.0667e+00, -9.3505e+00, ..., 4.1654e+00,
2.5146e+00, -1.1968e+00],
[ 4.0311e+00, -1.7641e+00, -1.0529e+01, ..., 4.7935e+00,
1.6451e+00, -1.8666e+00],
...,
[-1.9915e+00, -4.8644e+00, 1.6215e+00, ..., -6.5717e+00,
-8.0887e+00, -2.3718e+00],
[-2.2088e-01, -2.8205e+00, -2.9776e+00, ..., -5.8872e+00,
-4.6677e+00, -2.9776e+00],
[-2.3432e+00, -4.3322e+00, -5.0333e+00, ..., -5.2758e+00,
-5.7166e+00, -3.8767e+00]],
[[-1.6319e+00, -2.4529e+00, -1.6467e+01, ..., -1.3575e+00,
-7.4519e+00, -3.4496e+00],
[-4.7736e+00, -8.6295e+00, -1.7304e+01, ..., 9.6060e-01,
-6.3412e+00, -4.2869e+00],
[-4.4020e+00, -1.4710e+01, -1.0631e+01, ..., -1.9563e+00,
-1.0065e+01, -6.9341e+00],
...,
[-3.3460e+00, -1.1548e+01, -1.7534e+01, ..., -2.3254e+01,
-1.4880e+01, -4.2703e+00],
[-7.8832e+00, -8.0634e+00, -4.1515e+00, ..., -1.3829e+01,
-8.5824e+00, -6.9612e+00],
[-1.9334e+00, -4.8777e+00, -5.6278e+00, ..., -5.2274e+00,
-4.6428e+00, -2.6464e+00]],
[[ 9.0542e-01, 1.6621e+00, -8.8425e-02, ..., 2.3671e+00,
3.8945e+00, 9.7431e-01],
[ 1.6612e-02, -1.9300e+00, 8.2223e-01, ..., 9.9680e-01,
5.3189e+00, 1.1480e+00],
[ 4.4296e-02, -1.6992e+00, 3.2040e+00, ..., -1.2399e-01,
2.2642e+00, -4.4358e-01],
...,
[ 1.2329e-01, 1.0258e+00, -3.8776e+00, ..., -4.7792e+00,
-1.8114e+00, -3.6329e-01],
[-3.1022e-01, 2.2091e+00, 4.8840e-01, ..., 3.5659e+00,
-2.2190e+00, -1.4353e+00],
[ 1.7200e+00, 1.0409e+00, 5.0384e-01, ..., 6.3908e-01,
1.2368e+00, 7.5520e-01]],
...,
[[ 1.8012e-01, -1.5051e+00, -2.1056e+00, ..., -2.5092e+00,
-6.5020e+00, -2.1346e+00],
[-5.5700e-01, -2.0173e+00, -4.0584e+00, ..., -1.5307e+00,
-6.8318e+00, -2.1492e+00],
[ 4.2956e-01, 5.0586e-01, -1.1626e+00, ..., -3.3944e+00,
-7.5958e+00, -2.8673e+00],
...,
[-3.3439e+00, -5.3789e+00, -9.1847e+00, ..., -6.2618e+00,
-9.1264e+00, -2.0756e+00],
[-1.7180e+00, -2.0472e+00, 1.8983e-02, ..., -1.4018e+00,
-5.7522e+00, -3.9539e+00],
[-3.1999e+00, -3.7600e+00, -3.3683e+00, ..., -3.8970e+00,
-6.2309e+00, -2.2141e+00]],
[[-1.4111e+00, -4.5171e+00, -2.5607e+00, ..., -3.4673e+00,
-1.6955e+00, 1.1480e-01],
[-3.3213e+00, -6.8917e+00, -4.8290e+00, ..., -3.0034e+00,
-4.1222e+00, -9.1058e-01],
[-3.9302e+00, -8.4736e+00, -5.1817e+00, ..., -1.8862e+00,
-4.2914e+00, -1.1408e-02],
...,
[-4.7194e+00, -2.4050e+00, -6.6471e+00, ..., -4.5226e+00,
-6.9654e+00, -1.7262e-01],
[-3.8244e+00, -7.6884e-01, -2.7695e+00, ..., -2.1772e+00,
-3.2077e+00, 3.2785e-01],
[-4.1532e+00, -1.2815e+00, -1.9628e+00, ..., -1.2874e+00,
-1.9843e+00, -3.4611e-01]],
[[ 1.4430e+00, 1.0583e+00, 4.5463e+00, ..., 1.6589e+00,
3.3178e+00, -7.7421e-01],
[ 2.8904e+00, 3.4082e+00, 6.6787e-01, ..., 2.3970e+00,
5.3777e+00, 3.6310e+00],
[ 3.8414e+00, 5.0769e+00, -4.4955e-01, ..., 5.2176e+00,
4.5073e+00, 1.1127e+00],
...,
[ 1.4680e+00, 5.9378e+00, 1.1801e+01, ..., 5.0392e+00,
1.1340e+00, 2.9515e+00],
[ 3.6081e+00, 1.5158e+00, 1.6097e+00, ..., 9.4585e-01,
3.3734e+00, 3.7806e+00],
[ 2.6130e+00, 1.2929e+00, 9.6932e-01, ..., 1.1748e+00,
3.9403e-01, -5.3465e-01]]],
[[[-1.2123e-01, 7.5667e-01, -1.2700e-01, ..., -1.5052e+00,
-1.1545e+00, -8.8003e-01],
[ 1.1746e+00, 1.1980e+00, 9.4891e-01, ..., -1.3556e+00,
1.6540e-01, -9.9938e-01],
[ 1.8340e+00, 1.3845e-01, -2.9280e+00, ..., -7.1903e-01,
3.9571e-01, -1.3728e+00],
...,
[-1.3649e+00, -4.7892e-01, -1.0022e+00, ..., -3.3485e+00,
-3.1525e+00, 8.9019e-01],
[-1.9789e+00, -1.2376e+00, -3.0560e+00, ..., -7.7190e-01,
-1.5379e+00, 9.9686e-01],
[-2.0887e+00, -4.0390e+00, -4.2547e+00, ..., -1.7920e-01,
-2.3430e+00, -1.4731e+00]],
[[-7.3044e-01, -1.5961e+00, -2.8127e+00, ..., -2.9382e+00,
-4.7497e+00, -2.4995e+00],
[-2.1395e+00, -4.1985e+00, -5.6271e+00, ..., -4.6037e+00,
-4.8502e+00, -4.5066e+00],
[-2.4352e+00, -4.8532e+00, -7.3655e+00, ..., -3.4635e+00,
-3.3519e+00, -3.3703e+00],
...,
[-7.0332e+00, -7.6393e+00, -8.8658e+00, ..., -5.2286e+00,
-6.6552e+00, -4.4650e-01],
[-9.4375e+00, -9.2505e+00, -7.3570e+00, ..., -5.1692e+00,
-8.9042e+00, -3.1942e+00],
[-2.5553e+00, -3.7302e+00, -3.8696e+00, ..., -6.8482e-01,
-4.1318e+00, -8.7444e-01]],
[[ 6.9077e-01, 1.2895e+00, 1.2011e+00, ..., 1.3974e+00,
1.9436e+00, 9.5045e-01],
[ 8.2964e-01, 1.4244e+00, 8.8026e-01, ..., 1.4947e+00,
2.4254e+00, 1.0546e+00],
[ 6.2863e-02, 2.3368e+00, 2.6912e+00, ..., 1.0555e+00,
1.6565e+00, 2.0820e+00],
...,
[ 5.6578e-01, 9.4802e-01, -1.0743e+00, ..., 9.6582e-01,
-3.7781e-01, -2.0249e+00],
[ 4.1121e-01, -6.6446e-01, -4.7641e-01, ..., 2.3517e+00,
-1.3483e-01, -1.3485e+00],
[ 8.6117e-01, -1.6738e-01, -1.9815e-01, ..., 5.4103e-01,
-2.1660e-01, -1.6963e+00]],
...,
[[-9.0181e-01, -1.1896e+00, -1.9662e+00, ..., -2.8866e+00,
-3.2309e+00, -1.2533e+00],
[-1.9868e+00, -3.1243e+00, -2.7685e+00, ..., -4.0504e+00,
-4.3421e+00, -3.4474e+00],
[-8.2603e-01, -2.6661e+00, -3.7476e+00, ..., -1.4417e+00,
-3.5296e+00, -2.6405e+00],
...,
[-1.9843e+00, -1.3283e+00, -1.8414e+00, ..., -1.9946e+00,
-3.0943e+00, -1.1893e+00],
[-2.6717e+00, -3.2231e+00, -1.1891e+00, ..., -2.2122e+00,
-3.3017e+00, -7.7248e-01],
[-2.3130e+00, -2.5402e+00, -1.7356e+00, ..., -2.1441e+00,
-1.5734e+00, 1.2194e-01]],
[[-1.5773e+00, -2.3254e+00, -3.2328e+00, ..., -2.9695e+00,
-2.9373e+00, -4.7276e-01],
[-4.3588e+00, -5.0857e+00, -4.5540e+00, ..., -4.0926e+00,
-5.9008e+00, -1.6847e+00],
[-4.1888e+00, -5.9671e+00, -5.1409e+00, ..., -9.2940e-01,
-3.7936e+00, -1.8452e+00],
...,
[-3.3488e+00, -2.7351e+00, -4.0828e+00, ..., 1.9483e+00,
7.6155e-03, -4.2314e-02],
[-3.4276e+00, 7.0729e-02, -7.9207e-02, ..., 2.2082e-01,
1.1151e+00, 1.5496e+00],
[-3.8473e+00, -1.6241e+00, -4.5856e-01, ..., -1.5884e+00,
5.1200e-01, 8.2141e-01]],
[[ 8.7956e-01, -7.0328e-02, 8.7465e-01, ..., 4.4997e-01,
1.3919e+00, 6.6306e-01],
[ 1.7418e+00, 1.1708e+00, 2.0811e+00, ..., 4.3598e-01,
1.9331e+00, 3.5356e+00],
[ 1.7909e+00, 7.9302e-01, 3.3552e-01, ..., 3.3497e-01,
4.5001e-01, 1.0876e+00],
...,
[ 3.2029e+00, 2.6801e+00, 1.7448e+00, ..., 1.7124e+00,
9.2519e-01, 7.2877e-01],
[ 4.7038e+00, 5.0550e+00, 1.5640e+00, ..., 4.7740e+00,
1.7903e+00, 1.8240e+00],
[ 2.4815e+00, 2.0499e+00, -2.1193e-01, ..., 6.0821e-01,
-2.4717e-01, 1.1755e+00]]],
[[[ 7.1731e-01, 5.6000e-01, -6.2872e-01, ..., 3.7946e-01,
2.0263e-01, -4.3540e-01],
[ 1.2864e+00, 2.2285e-01, 1.1038e+00, ..., -2.0151e+00,
2.5582e-01, -6.1006e-01],
[-1.1743e+00, -1.2270e+00, 1.7720e+00, ..., 8.3995e-01,
1.1297e+00, -9.2961e-01],
...,
[ 1.5634e-02, -9.0194e-01, 2.9325e-01, ..., 3.1405e-01,
7.1215e-02, -3.5098e-01],
[-2.0475e+00, -1.8011e+00, -2.0209e-01, ..., 4.1475e-01,
-1.0546e-01, -1.1319e+00],
[-9.7360e-01, -3.4835e+00, -3.0091e+00, ..., -2.6529e+00,
-2.9775e+00, -2.7117e+00]],
[[-1.5192e+00, -3.9559e+00, -6.8180e+00, ..., -2.3859e+00,
-4.0621e+00, -2.3500e+00],
[-5.1976e+00, -6.7180e+00, -6.0957e+00, ..., -2.9851e+00,
-2.8439e+00, -3.1170e+00],
[-2.8313e+00, -3.7682e+00, -3.9273e+00, ..., 7.5709e-01,
-1.3503e+00, -2.8265e+00],
...,
[-4.7940e+00, -6.0196e+00, -6.6108e+00, ..., -2.5643e+00,
-3.2702e+00, -2.5478e+00],
[-7.4952e+00, -9.1876e+00, -1.0260e+01, ..., -4.0317e+00,
-5.1156e+00, -4.9687e+00],
[-1.4093e+00, -4.1303e+00, -4.2661e+00, ..., -3.0734e+00,
-2.8189e+00, -1.5852e+00]],
[[ 6.4103e-01, 2.9634e+00, 1.9165e+00, ..., 1.8497e+00,
1.6868e+00, 8.6498e-01],
[-7.8555e-02, 1.3784e+00, 1.5424e+00, ..., 2.9569e+00,
2.2657e+00, 1.0506e+00],
[ 4.9457e-01, 2.2185e+00, 1.4033e+00, ..., 1.1913e+00,
2.5761e+00, 1.7153e+00],
...,
[ 1.7929e-01, 1.2918e+00, -2.3094e-01, ..., 8.0001e-01,
1.0084e+00, 6.3628e-01],
[ 1.0944e+00, 1.7599e+00, 1.5509e-01, ..., 8.2832e-01,
1.0037e+00, 4.3770e-01],
[ 4.1358e-01, 1.3825e+00, 8.4844e-01, ..., 1.1913e+00,
1.6800e+00, 5.2804e-01]],
...,
[[-9.6671e-01, -3.3002e+00, -2.7077e+00, ..., -1.1602e+00,
-2.2519e+00, -5.6606e-01],
[-2.8895e+00, -4.3489e+00, -1.7105e+00, ..., -3.4110e+00,
-3.9848e+00, -3.3435e+00],
[-2.3132e+00, -1.8829e+00, -1.3015e+00, ..., 1.2289e+00,
-2.1222e+00, -1.7220e+00],
...,
[-1.3377e+00, -2.0070e+00, -2.6036e-01, ..., -8.1991e-01,
-2.6108e+00, -8.6515e-01],
[-2.9157e+00, -4.6594e+00, -2.3928e+00, ..., -9.0514e-01,
-3.0298e+00, -1.5627e+00],
[-1.9617e+00, -3.2620e+00, -1.3118e+00, ..., -1.9584e+00,
-2.4564e+00, -5.7093e-01]],
[[-2.4673e+00, -3.2198e+00, -1.6945e+00, ..., 1.9575e-01,
-1.8139e+00, -7.1436e-01],
[-4.2689e+00, -3.2804e+00, -2.2314e+00, ..., -1.6055e+00,
-4.1243e+00, -1.7439e+00],
[-3.2597e+00, -1.5144e+00, -3.2400e-01, ..., 4.2312e-01,
-4.1174e+00, -1.6709e+00],
...,
[-3.3149e+00, -2.3872e+00, -6.4296e-01, ..., -3.7406e-01,
-3.5268e+00, -2.1543e+00],
[-4.4620e+00, -1.3167e+00, -7.2657e-02, ..., -3.3759e-02,
-3.2282e+00, -1.9172e+00],
[-3.7963e+00, -2.6120e+00, -1.9328e-01, ..., -5.6398e-01,
-2.5032e+00, -1.2343e+00]],
[[ 1.6189e+00, 1.1740e+00, 1.7305e+00, ..., 6.3581e-01,
1.1571e+00, 5.6722e-01],
[ 3.4673e+00, 3.2107e+00, 3.5748e+00, ..., 1.2160e+00,
1.5468e+00, 2.3485e+00],
[ 5.8757e-01, 2.6437e-02, 1.9974e+00, ..., 5.0422e-01,
1.1522e+00, 3.1556e-01],
...,
[ 2.6911e+00, 2.1815e+00, 2.0329e+00, ..., 1.5248e+00,
2.7695e-01, 5.6281e-01],
[ 3.6175e+00, 4.3267e+00, 3.4723e+00, ..., 1.8873e+00,
7.3359e-01, 1.2328e+00],
[ 2.3024e+00, 1.4248e+00, 1.3961e+00, ..., 6.1410e-01,
-2.5988e-01, -1.2427e-01]]],
[[[-3.2609e+00, -3.1426e+00, -1.3298e+00, ..., -1.1674e+00,
-1.3775e+00, -9.8766e-01],
[ 1.1361e+00, 2.5505e+00, -3.8926e+00, ..., -9.4591e+00,
-6.9151e+00, -3.8939e+00],
[-8.2175e-01, 2.5728e+00, 3.5996e+00, ..., -1.8673e+01,
-1.3963e+01, -7.7505e+00],
...,
[ 1.8140e+00, 1.2991e+00, 2.6316e+00, ..., -1.6405e+01,
-1.1827e+01, -1.5537e+00],
[-1.1245e+00, -1.1819e+00, 1.1647e-01, ..., -1.0353e+01,
-8.5714e+00, 5.3603e-01],
[-3.5095e-01, -2.0949e+00, -1.5746e+00, ..., -3.0627e+00,
3.7742e+00, 3.6283e+00]],
[[-4.7282e+00, -3.3293e+00, -2.8044e+00, ..., -1.7318e+01,
-1.4277e+01, -5.2150e+00],
[-3.6370e+00, -9.2031e+00, -1.6530e+01, ..., -1.2176e+01,
-9.3518e+00, -4.6007e+00],
[ 2.5373e+00, 4.0199e+00, 3.6364e+00, ..., -2.3298e+01,
-2.0128e+01, -6.4989e+00],
...,
[-3.4351e+00, -2.1417e+00, -6.3172e-01, ..., -4.4641e+00,
-5.5873e+00, -1.5190e+00],
[-6.7295e+00, -1.1823e+01, -1.1674e+01, ..., -1.3913e+01,
-7.4787e+00, 5.4304e-01],
[-6.9784e-01, -2.4501e+00, -8.4038e-01, ..., 7.6922e-02,
-3.5281e-01, 1.6332e+00]],
[[ 2.0079e+00, 5.6804e+00, 3.3559e+00, ..., 1.6881e-01,
1.6988e+00, 8.5618e-01],
[ 1.5931e-01, 4.7219e+00, 1.0532e+01, ..., 2.0868e+00,
4.1416e+00, 1.6733e+00],
[-7.3933e+00, -1.0458e+01, -4.4604e+00, ..., 1.2813e+01,
1.2606e+01, 7.8327e+00],
...,
[-3.2148e+00, -1.3943e+00, -1.4851e+00, ..., -2.2610e+00,
-5.8709e+00, -3.1433e+00],
[-1.0115e+00, 8.4616e-01, -1.0232e+00, ..., 1.8027e+00,
3.4532e+00, 1.4356e+00],
[-3.0186e-01, 8.5524e-01, 8.1171e-01, ..., 3.8546e+00,
-1.1926e+00, -4.8283e+00]],
...,
[[-2.8270e+00, -3.8650e+00, -3.4535e+00, ..., -6.0798e+00,
-5.8412e+00, -1.9087e+00],
[-8.1129e-01, -3.8023e+00, -4.4456e+00, ..., -9.8762e+00,
-5.2447e+00, -3.4037e+00],
[ 1.4049e+00, 1.2413e+00, 4.1969e+00, ..., -1.5302e+01,
-1.2596e+01, -6.0105e+00],
...,
[-9.0125e-01, -4.2425e+00, -4.8914e-01, ..., -7.8125e+00,
-1.5908e+01, -3.8498e+00],
[-3.9665e+00, -8.2107e+00, -6.5878e+00, ..., -8.9543e+00,
-2.9375e+00, -1.8604e+00],
[-5.7364e-01, 6.8083e-01, 1.5911e+00, ..., 2.1962e+00,
1.8677e+00, 1.3192e+00]],
[[-4.1011e+00, -4.3925e+00, -1.8449e-01, ..., 4.3047e-01,
1.1332e+00, -4.6228e-01],
[-4.5380e+00, -1.0271e+01, -8.1308e+00, ..., -4.7323e+00,
-2.9637e+00, -2.7903e-01],
[ 2.9454e+00, 5.7121e+00, 6.1280e+00, ..., -1.6896e+01,
-7.8332e+00, -3.0028e+00],
...,
[ 1.0121e+00, -1.9050e+00, 6.6014e-01, ..., -1.5502e+01,
-1.3503e+01, -3.3968e+00],
[-7.1595e-01, -1.5196e+00, -2.8815e+00, ..., -2.4528e+01,
-1.5015e+01, -1.8127e+00],
[-1.4389e+00, -1.2680e-01, -1.3135e+00, ..., -1.1348e+01,
-4.3530e+00, 1.1442e+00]],
[[ 4.3304e-01, 2.2951e-01, 1.0941e+00, ..., 3.9622e+00,
4.0252e+00, 7.7113e-01],
[ 1.3812e+00, 6.5426e+00, 1.0227e+01, ..., 4.3780e-01,
5.1187e-02, 3.8611e-01],
[-2.3282e+00, -9.8925e-01, 3.0325e+00, ..., 3.6558e+00,
6.6248e+00, 6.3997e+00],
...,
[ 3.8511e+00, 8.8168e-01, -5.5965e-01, ..., -1.6908e+01,
-9.0702e+00, -4.4518e+00],
[ 5.0773e+00, 5.3455e+00, 5.8337e+00, ..., -1.4176e+01,
-8.0631e+00, -2.0574e+00],
[ 1.7115e+00, 7.0804e-01, -4.3081e-01, ..., -1.2796e+01,
-1.7573e+00, 7.9354e-01]]]], device='cuda:0'),
tensor([[[[-2.0278e+00, 6.0086e-01, 1.2043e-01, ..., 1.4444e+00,
-1.0694e+00, -1.5703e+00],
[-4.0435e+00, -5.3947e+00, -1.4479e+01, ..., -3.9502e+00,
-5.0572e+00, -3.1938e+00],
[-2.8681e+00, -9.3769e+00, -1.3951e+01, ..., 4.0047e+00,
-4.3683e-01, -5.0703e-01],
...,
[-3.5297e+00, -2.8481e+00, -3.6846e+00, ..., -3.9245e+00,
-4.2761e+00, 3.7880e-01],
[-6.0455e+00, -4.8137e+00, -3.9186e+00, ..., -2.4426e+00,
-4.8051e-01, 1.8217e+00],
[-2.7981e+00, -4.8980e+00, -5.5704e+00, ..., -3.4748e+00,
-4.1228e+00, -2.3502e+00]],
[[-2.3014e-01, 1.9492e+00, 8.3901e-01, ..., -3.9674e+00,
-1.0968e+01, -6.4312e+00],
[ 6.1837e-01, 2.1284e-01, -5.4972e+00, ..., -9.5455e+00,
-1.1592e+01, -3.0683e+00],
[-9.3417e-01, -8.7628e+00, -2.4533e+00, ..., 5.0084e-01,
-7.1388e+00, -1.9847e+00],
...,
[-1.6076e+01, -2.2865e+01, -2.5942e+01, ..., -3.4208e-01,
8.9817e-03, -4.8200e-01],
[-5.3488e+00, -5.5120e+00, -7.3515e+00, ..., -1.4950e+01,
-9.6744e+00, -2.1117e+00],
[-2.4064e+00, -5.0893e+00, -3.8991e+00, ..., -2.3649e+00,
-4.0177e+00, -3.3818e+00]],
[[-2.3105e-01, -2.2174e-02, -3.4726e+00, ..., 2.4525e+00,
3.5002e+00, 5.3392e-01],
[-3.0568e+00, -5.1395e+00, -2.2702e+00, ..., 7.1879e+00,
8.1801e+00, 2.1974e+00],
[ 1.8807e+00, 1.2999e-01, 1.3356e+00, ..., -4.4513e-01,
3.0202e+00, 4.3535e+00],
...,
[-9.5960e-01, -2.7786e+00, -6.3271e+00, ..., 6.4860e-01,
3.0867e+00, 2.3456e+00],
[-9.9044e-01, 3.6438e-01, -1.9355e-01, ..., -4.1264e-01,
-2.8470e+00, -7.7738e-01],
[-3.3304e+00, -4.9601e+00, -2.5523e+00, ..., 1.4015e+00,
-1.1875e-01, -1.6686e+00]],
...,
[[ 1.3022e+00, 3.0323e+00, 2.9727e+00, ..., -2.5878e+00,
-4.8111e+00, -1.1842e+00],
[ 3.4583e-01, -4.6808e-01, -7.9526e+00, ..., -4.6492e+00,
-5.9490e+00, -3.1298e+00],
[-5.7074e-01, -3.9571e+00, -5.1101e+00, ..., 3.0190e+00,
4.7323e-01, 4.3448e-01],
...,
[-9.5798e+00, -1.0535e+01, -9.8989e+00, ..., -4.0327e-01,
-1.5670e+00, -3.7885e+00],
[ 1.5627e-01, -3.0892e+00, -9.7487e-01, ..., -8.0703e+00,
-6.0025e+00, -4.1488e+00],
[-2.1967e+00, -4.3680e+00, -3.3967e+00, ..., -1.8153e+00,
-6.8865e+00, -4.9651e+00]],
[[-2.4082e+00, -6.8563e+00, -6.8854e+00, ..., -1.1725e+00,
-1.0698e+00, 5.8211e-01],
[-1.9240e+00, -1.0437e+01, -1.8726e+01, ..., -7.1954e+00,
-5.9578e+00, -1.1377e+00],
[-6.0343e+00, -2.0262e+01, -2.5140e+01, ..., 4.5376e-01,
-6.1180e+00, -5.3832e+00],
...,
[-7.0262e+00, -1.2369e+01, -1.3701e+01, ..., -3.4975e+00,
-7.2413e+00, -3.2570e+00],
[ 2.3628e+00, 4.7451e+00, 3.3062e+00, ..., 8.5411e-02,
7.9282e-01, -6.1597e-01],
[ 3.4466e-02, 9.1767e-01, 2.2441e+00, ..., -2.1180e-01,
9.9485e-01, 1.4405e+00]],
[[-2.0735e+00, -1.8386e+00, -2.5182e+00, ..., 2.0649e+00,
3.4585e+00, 1.8748e+00],
[-3.9005e+00, -9.8064e+00, -2.2521e+01, ..., 3.4755e+00,
4.0057e+00, 2.8936e+00],
[-4.4506e+00, -7.8229e+00, -1.6207e+01, ..., 3.7132e+00,
3.9423e+00, 4.0859e+00],
...,
[ 1.1127e+01, 1.4288e+01, 1.2707e+01, ..., -3.8955e+00,
-8.1934e+00, -5.5184e+00],
[ 2.5347e+00, 3.4285e+00, 6.1321e-01, ..., 7.1879e+00,
3.5699e+00, 3.3713e-01],
[ 2.3076e+00, 2.0257e+00, 8.5631e-02, ..., -4.4316e-01,
2.6105e+00, 2.5395e+00]]],
[[[ 5.1910e-01, 1.3705e+00, 7.2762e-01, ..., 2.2946e-01,
-5.1724e-01, 7.0878e-01],
[ 3.2821e-01, -6.0437e-01, 7.7671e-01, ..., 1.6092e-01,
-1.1830e+00, -2.8417e-01],
[ 1.2300e-01, -4.9085e-01, 6.2358e-01, ..., -1.0935e+00,
-2.6841e-01, -6.8882e-01],
...,
[-8.6043e-01, -1.6344e+00, -4.6792e-01, ..., -3.1156e+00,
-3.5625e+00, -9.0008e-01],
[-2.3459e+00, -1.5925e+00, 1.9770e-01, ..., -2.7778e+00,
-3.1473e+00, 4.4027e-02],
[-2.2743e+00, -4.0245e+00, -3.0583e+00, ..., -4.1171e+00,
-3.6055e+00, -1.5191e+00]],
[[-1.3486e+00, -1.9294e+00, -2.3347e+00, ..., -8.9842e-01,
-4.8635e+00, -9.4435e-01],
[-4.8770e+00, -6.4996e+00, -4.7699e+00, ..., -5.3361e+00,
-7.3300e+00, -3.8943e+00],
[-4.2478e+00, -4.9065e+00, -3.4474e+00, ..., -6.5341e+00,
-5.8473e+00, -4.1080e+00],
...,
[-7.0946e+00, -8.5611e+00, -6.7345e+00, ..., -9.3494e+00,
-9.5807e+00, -2.2270e+00],
[-9.9930e+00, -1.0387e+01, -8.7615e+00, ..., -1.0578e+01,
-1.0819e+01, -3.7938e+00],
[-2.4778e+00, -4.9036e+00, -4.2236e+00, ..., -4.3215e+00,
-4.3150e+00, -6.8481e-01]],
[[ 1.7165e-02, 9.5124e-01, 5.1577e-01, ..., 1.6617e+00,
2.1652e+00, 3.0977e-01],
[ 2.4184e-01, 3.9122e-01, -3.2613e-01, ..., 3.0880e-02,
2.0421e+00, -7.5301e-01],
[ 7.8290e-01, 2.7476e+00, 1.4656e+00, ..., 4.7576e-01,
1.9295e+00, 9.0538e-01],
...,
[ 5.1268e-01, 7.2459e-01, 5.7960e-01, ..., 1.2147e+00,
1.7065e-01, -9.9615e-01],
[-5.3483e-01, 4.2584e-01, -4.8594e-01, ..., -3.0380e-01,
3.3198e-01, -6.3379e-01],
[ 5.6145e-01, 1.3090e+00, 1.1265e+00, ..., -2.4273e-01,
-3.6620e-01, -1.8916e+00]],
...,
[[-9.4898e-01, -1.0668e+00, -6.3740e-01, ..., -2.4142e+00,
-2.6636e+00, 1.4536e-01],
[-4.2105e+00, -5.1212e+00, -2.5166e+00, ..., -4.3069e+00,
-5.4510e+00, -3.1423e+00],
[-2.5168e+00, -1.8818e+00, -2.6512e-01, ..., -2.7204e+00,
-3.9798e+00, -1.9280e+00],
...,
[-2.6054e+00, -3.0540e+00, -1.7936e+00, ..., -5.6909e+00,
-5.7207e+00, -2.9692e+00],
[-3.5760e+00, -5.4183e+00, -2.3813e+00, ..., -4.1142e+00,
-5.2930e+00, -1.3373e+00],
[-3.1590e+00, -3.7766e+00, -3.1285e+00, ..., -2.8863e+00,
-2.7598e+00, 4.1150e-01]],
[[-1.1030e+00, 1.4752e-01, -1.3448e+00, ..., -9.2267e-02,
6.4207e-01, 3.1995e-01],
[-4.0606e+00, -1.6143e+00, -1.4167e+00, ..., -1.9006e+00,
-1.6114e+00, 5.0909e-01],
[-3.7137e+00, -2.1655e+00, -1.5632e+00, ..., -6.5465e-01,
-2.3104e+00, -5.9520e-01],
...,
[-4.2688e+00, -2.5248e+00, -3.4259e+00, ..., -3.6300e+00,
-2.0720e+00, -5.0955e-01],
[-3.5510e+00, -5.6322e-01, -1.4555e+00, ..., -1.9626e+00,
-1.0422e+00, 5.0198e-01],
[-3.7770e+00, -2.9956e+00, -2.6424e+00, ..., -1.4250e+00,
-1.1153e+00, 4.2226e-01]],
[[ 1.6182e+00, 7.6581e-01, 6.0597e-01, ..., 7.7706e-01,
1.1935e+00, 8.2322e-02],
[ 3.1288e+00, 2.0459e+00, 1.7059e+00, ..., 2.0198e+00,
2.8837e+00, 3.5184e+00],
[ 2.2062e+00, 1.1317e-01, 3.4366e-01, ..., 2.0519e+00,
2.5908e+00, 2.1089e+00],
...,
[ 3.0387e+00, 2.7036e+00, 1.4703e+00, ..., 1.9568e+00,
1.7044e+00, 1.8055e-01],
[ 4.8283e+00, 4.8912e+00, 3.7886e+00, ..., 3.1222e+00,
1.9487e+00, 1.5430e+00],
[ 2.7822e+00, 2.8355e+00, 2.3518e+00, ..., 5.7340e-01,
4.3672e-01, 9.6110e-01]]],
[[[-3.0915e-01, 1.3661e+00, -1.0991e+00, ..., 6.9586e-02,
-3.1048e+00, -1.5277e+00],
[-2.6516e-01, 6.4245e-01, -1.4775e+00, ..., 1.3899e+00,
-2.9049e+00, -1.5257e+00],
[-1.2485e+00, 2.8179e-01, -3.3088e+00, ..., -4.5925e+00,
-2.8995e+00, 5.3402e-01],
...,
[-1.6975e-01, 3.3248e-02, 1.0769e+00, ..., -1.8872e+00,
-3.1336e+00, 2.0571e-01],
[-1.5322e+00, -2.0529e-03, 1.1340e+00, ..., -2.4578e+00,
-3.5836e+00, -4.9125e-01],
[-1.2634e+00, -2.4129e+00, -1.2367e+00, ..., -3.8123e+00,
-3.7642e+00, -1.6283e+00]],
[[-1.8669e+00, -2.4926e+00, -6.6820e+00, ..., -5.0299e+00,
-8.2229e+00, -2.4209e+00],
[-5.1294e+00, -4.8711e+00, -6.3857e+00, ..., -7.1041e+00,
-6.4000e+00, -2.5711e+00],
[-4.4232e+00, -3.6985e+00, -8.1780e+00, ..., -8.2840e+00,
-4.4044e+00, -5.7457e-01],
...,
[-4.4930e+00, -5.2788e+00, -5.7474e+00, ..., -6.2080e+00,
-7.0708e+00, -1.0331e+00],
[-7.6788e+00, -8.0150e+00, -7.4855e+00, ..., -8.5783e+00,
-1.0754e+01, -4.4549e+00],
[-1.4383e+00, -3.1957e+00, -2.3951e+00, ..., -4.1113e+00,
-4.2540e+00, -5.1726e-01]],
[[-2.0293e-01, 2.5428e+00, 2.0127e+00, ..., 2.0364e+00,
2.6953e+00, 9.8157e-01],
[-4.2859e-01, 5.1586e-01, 1.0127e+00, ..., 9.2930e-01,
2.9061e+00, 1.2975e+00],
[ 1.8845e-01, 1.3748e+00, 8.1444e-01, ..., 1.6064e+00,
1.4060e+00, 6.4272e-01],
...,
[-7.9361e-01, 5.4692e-01, 1.5072e+00, ..., -7.4729e-01,
-8.9250e-02, -1.5348e+00],
[-1.3046e+00, 8.5284e-02, 1.0944e+00, ..., -2.8174e-01,
-3.7477e-01, -1.3779e+00],
[-3.4230e-01, -3.1051e-01, 1.5613e-01, ..., 7.4130e-01,
2.5398e-01, -1.6769e+00]],
...,
[[-1.3197e+00, -2.5587e+00, -3.0811e+00, ..., -2.8319e+00,
-4.9823e+00, -1.1974e+00],
[-3.4048e+00, -3.5548e+00, -3.0566e+00, ..., -2.7571e+00,
-4.4629e+00, -1.3584e+00],
[-2.4174e+00, -1.7006e+00, -3.9230e+00, ..., -4.8431e+00,
-2.5903e+00, 9.3197e-01],
...,
[-7.8815e-01, -2.0925e+00, -5.1726e-01, ..., -1.9666e+00,
-4.0017e+00, -4.5931e-02],
[-2.0426e+00, -3.8418e+00, -6.1516e-01, ..., -1.8263e+00,
-4.2946e+00, -4.9372e-01],
[-1.4299e+00, -1.5145e+00, 1.7512e-01, ..., -1.7190e+00,
-2.1776e+00, 1.0299e+00]],
[[-1.6523e+00, -1.9040e+00, -4.5046e+00, ..., -2.5091e+00,
-9.4338e-01, 1.2181e+00],
[-3.5484e+00, -2.7252e+00, -2.3517e+00, ..., -1.9961e+00,
-3.1918e+00, -4.5506e-01],
[-3.6272e+00, -3.5968e+00, -3.1936e+00, ..., -2.8170e+00,
-4.7530e+00, -1.5106e+00],
...,
[-2.7746e+00, -2.9967e+00, -3.4044e+00, ..., -1.8913e+00,
-2.6589e+00, 6.9655e-01],
[-2.0082e+00, -1.2364e-01, -1.4289e+00, ..., 1.2987e-01,
-8.9474e-01, 1.4191e+00],
[-2.1809e+00, -1.3522e+00, -4.9714e-01, ..., -5.7737e-01,
-8.2555e-01, 5.4458e-01]],
[[ 1.4829e+00, 4.0650e-01, 2.1258e+00, ..., 1.9394e+00,
1.6227e+00, 2.9785e-01],
[ 4.1307e+00, 3.1013e+00, 2.0776e+00, ..., 3.7550e+00,
1.8894e+00, 3.1902e+00],
[ 1.5877e+00, 4.3391e-01, -2.9945e-01, ..., 1.7715e+00,
2.3793e+00, 1.8009e+00],
...,
[ 3.0109e+00, 2.6495e+00, 2.3883e+00, ..., 2.8640e+00,
1.2825e+00, 1.2980e+00],
[ 3.8145e+00, 4.0833e+00, 3.0491e+00, ..., 2.4762e+00,
6.7118e-01, 2.1817e+00],
[ 2.3976e+00, 2.3662e+00, 1.6996e+00, ..., 4.2254e-01,
-2.0994e-01, 1.3400e+00]]],
[[[ 7.9017e-01, 5.7242e-01, -1.9357e-01, ..., -5.9276e-01,
1.0844e-01, -2.2756e-01],
[ 1.4269e+00, 7.5417e-01, 1.1976e+00, ..., -1.4776e+00,
-7.3275e-01, -6.6701e-01],
[-4.0887e-01, 3.0700e-01, 2.5353e+00, ..., -2.1131e+00,
-2.1473e+00, -1.1189e+00],
...,
[-1.9774e+00, -3.8698e+00, -6.2323e+00, ..., -6.3207e-01,
-7.0717e-01, -6.3698e-01],
[ 1.2208e-01, 2.2716e+00, 9.7793e-01, ..., -2.2046e-01,
-5.6989e-01, -1.3836e+00],
[ 1.1100e+00, 4.8931e-01, -1.4264e+00, ..., -2.7792e+00,
-3.1468e+00, -2.8783e+00]],
[[-1.4856e+00, -4.1177e+00, -6.4483e+00, ..., -4.3184e+00,
-3.6800e+00, -2.3897e+00],
[-4.3190e+00, -5.6853e+00, -6.4648e+00, ..., -8.9884e+00,
-4.7424e+00, -3.5121e+00],
[-1.8963e+00, -2.6799e+00, -5.7424e+00, ..., -1.0224e+01,
-4.2139e+00, -2.5623e+00],
...,
[ 7.4959e-01, 2.6410e+00, -3.5773e+00, ..., -3.6380e+00,
-3.8425e+00, -3.0124e+00],
[-1.2904e+00, 5.9325e-01, -2.2261e+00, ..., -4.7001e+00,
-5.6543e+00, -5.2910e+00],
[-5.0602e-01, -2.2436e+00, -2.8738e+00, ..., -3.1956e+00,
-2.9816e+00, -1.6955e+00]],
[[ 7.4574e-01, 3.0276e+00, 1.8647e+00, ..., 1.0896e+00,
1.3864e+00, 8.9394e-01],
[-1.2264e-01, 1.6824e+00, 1.2680e+00, ..., 1.1287e+00,
1.0757e+00, 8.9366e-01],
[ 7.5982e-01, 2.6611e+00, 1.5405e+00, ..., 2.1202e+00,
1.6337e+00, 1.3642e+00],
...,
[-2.8747e+00, -8.9167e-01, 5.1059e-01, ..., 6.1310e-01,
7.3549e-01, 8.3376e-01],
[-2.3915e-01, 2.7043e-01, 4.4498e-01, ..., 9.4436e-01,
1.2253e+00, 4.7963e-01],
[-5.6730e-01, 7.7740e-02, 5.0388e-02, ..., 1.2091e+00,
1.8168e+00, 4.8188e-01]],
...,
[[-8.7553e-01, -3.2784e+00, -2.0709e+00, ..., -2.0347e+00,
-3.1543e+00, -4.5957e-01],
[-2.6940e+00, -4.2619e+00, -1.2757e+00, ..., -2.4473e+00,
-6.5228e+00, -3.0345e+00],
[-2.4498e+00, -2.8664e+00, -1.1118e+00, ..., -1.8692e+00,
-6.6897e+00, -2.0808e+00],
...,
[ 5.9722e-01, -2.1297e-01, -3.3999e+00, ..., -1.0877e+00,
-3.0437e+00, -1.1848e+00],
[-4.1116e-01, -2.6417e+00, -2.8146e+00, ..., -1.2317e+00,
-3.5040e+00, -1.7090e+00],
[-5.4768e-01, -3.4327e+00, -2.1124e+00, ..., -2.0852e+00,
-2.6051e+00, -6.0436e-01]],
[[-2.5989e+00, -3.2084e+00, -1.4693e+00, ..., -6.0069e-01,
-1.7025e+00, -5.4642e-01],
[-4.0930e+00, -3.3992e+00, -2.3387e+00, ..., 4.3247e-01,
-4.0246e+00, -1.8727e+00],
[-2.9839e+00, -1.9650e+00, -1.4362e+00, ..., 2.8276e+00,
-3.7333e+00, -2.7155e+00],
...,
[-4.3678e+00, -8.2050e+00, -6.8521e+00, ..., -4.6013e-01,
-3.2845e+00, -1.9491e+00],
[ 3.6270e-01, -2.7248e+00, -3.4895e+00, ..., 3.4612e-02,
-3.0779e+00, -1.8018e+00],
[ 3.9636e-01, -2.1192e+00, -2.6240e+00, ..., -6.8592e-01,
-2.5449e+00, -1.2281e+00]],
[[ 1.6198e+00, 1.3521e+00, 1.5616e+00, ..., 3.0845e-01,
6.9787e-01, 7.0280e-01],
[ 3.0123e+00, 3.1659e+00, 3.4067e+00, ..., 2.1298e+00,
1.0453e-01, 2.5940e+00],
[ 6.3113e-01, 2.5791e-01, 2.5003e+00, ..., 1.0959e+00,
-5.5086e-01, 8.7576e-01],
...,
[-5.6640e-01, -4.7821e+00, -8.8217e+00, ..., 1.8950e+00,
7.0634e-01, 6.1350e-01],
[ 1.7678e+00, 4.1799e-01, -3.6484e+00, ..., 2.1839e+00,
7.7164e-01, 1.2981e+00],
[ 1.8116e+00, 1.4820e+00, 2.9226e-01, ..., 7.4160e-01,
-2.1353e-01, -1.1895e-01]]],
[[[ 4.5569e-01, 7.7469e-01, -2.4127e+00, ..., 6.7322e-01,
4.6657e-01, -5.8708e-01],
[ 1.8293e+00, -7.4229e-02, -4.9317e+00, ..., 2.2166e+00,
8.6388e-01, -1.3320e+00],
[ 1.6496e+00, -2.1937e+00, -6.8862e+00, ..., 2.8199e+00,
-4.9655e-01, -2.6251e+00],
...,
[-5.4637e-01, -5.6792e+00, -3.0019e+00, ..., -3.0377e+00,
-2.7072e+00, -1.7054e-01],
[-2.9256e+00, -4.8562e+00, -2.8255e+00, ..., -2.5599e+00,
-2.7776e+00, 1.3883e-01],
[-2.3906e+00, -4.1131e+00, -3.7000e+00, ..., -3.5239e+00,
-3.4217e+00, -1.5982e+00]],
[[-8.6499e-01, -1.4089e+00, -4.3175e+00, ..., 3.4574e-01,
-3.3641e+00, -2.5952e+00],
[-2.8841e+00, -4.6321e+00, -6.6917e+00, ..., -2.9848e-01,
-4.6133e+00, -4.7672e+00],
[-3.5559e+00, -7.3145e+00, -8.2140e+00, ..., -6.2717e-01,
-5.4469e+00, -3.8626e+00],
...,
[-6.7402e+00, -1.0972e+01, -7.1536e+00, ..., -7.4127e+00,
-8.9992e+00, -2.0296e+00],
[-9.9827e+00, -1.0225e+01, -8.8709e+00, ..., -9.9058e+00,
-1.0512e+01, -3.8110e+00],
[-2.5089e+00, -3.8236e+00, -5.6706e+00, ..., -4.2633e+00,
-4.1453e+00, -6.7623e-01]],
[[ 8.0968e-01, 1.2716e+00, 2.4229e+00, ..., 1.5236e+00,
2.6840e+00, 1.0723e+00],
[ 6.5839e-01, 1.8175e-01, 2.8885e+00, ..., 1.0323e+00,
2.9824e+00, 1.2181e+00],
[ 6.8364e-01, 2.2677e+00, 3.3585e+00, ..., 2.5358e+00,
2.5223e+00, 1.7305e+00],
...,
[-2.4557e-01, -7.7142e-01, -4.9016e+00, ..., 1.6515e+00,
1.7289e+00, -2.7623e-01],
[-1.0762e+00, -3.6592e+00, -8.5927e+00, ..., 7.2485e-01,
1.2736e+00, -3.4272e-01],
[ 4.8130e-01, -1.1947e+00, -2.2596e+00, ..., 1.0250e-01,
-5.6287e-01, -1.8695e+00]],
...,
[[-3.8194e-01, -2.0387e+00, -2.7838e+00, ..., -6.7918e-01,
-3.3921e+00, -1.0543e+00],
[-1.7284e+00, -3.8245e+00, -4.3725e+00, ..., -3.0799e+00,
-5.4905e+00, -3.5827e+00],
[-1.3903e+00, -3.5508e+00, -3.8402e+00, ..., -4.5307e+00,
-5.3650e+00, -3.5603e+00],
...,
[-3.0897e+00, -6.1818e+00, -4.2493e+00, ..., -6.2670e+00,
-6.5337e+00, -2.5870e+00],
[-3.3584e+00, -5.7892e+00, -4.5547e+00, ..., -4.6150e+00,
-5.2231e+00, -1.2581e+00],
[-3.0314e+00, -5.0970e+00, -5.5540e+00, ..., -3.2387e+00,
-2.5977e+00, 3.9314e-01]],
[[-1.5894e+00, -3.8751e+00, -3.8715e+00, ..., -1.3430e+00,
-2.2997e+00, -3.1051e-01],
[-5.0937e+00, -6.3439e+00, -3.7985e+00, ..., -2.6714e+00,
-5.0959e+00, -1.5831e+00],
[-5.5176e+00, -7.2403e+00, -2.5869e+00, ..., -3.4855e+00,
-4.7820e+00, -1.7339e+00],
...,
[-3.7471e+00, -2.5297e+00, -3.5541e+00, ..., -3.6781e+00,
-3.1484e+00, -1.3582e+00],
[-3.1712e+00, -9.5366e-01, 3.4952e-01, ..., -1.6983e+00,
-1.4547e+00, 2.8584e-01],
[-2.9129e+00, -1.6837e+00, 9.2784e-01, ..., -7.9698e-01,
-1.2817e+00, 2.8987e-01]],
[[ 8.9928e-01, 3.3693e-01, 9.1257e-01, ..., 1.5305e+00,
2.3838e+00, 8.7471e-01],
[ 1.8162e+00, 1.7504e+00, 1.7654e-01, ..., 1.9971e+00,
3.0032e+00, 3.0763e+00],
[ 2.2637e+00, 1.9707e+00, -2.6835e-01, ..., 4.1577e+00,
2.4384e+00, 4.6895e-01],
...,
[ 4.2693e+00, 1.9948e+00, 3.7632e+00, ..., 2.8177e+00,
2.6435e+00, 6.0594e-01],
[ 4.3724e+00, 3.6164e+00, 3.1162e+00, ..., 3.1532e+00,
1.7773e+00, 1.4650e+00],
[ 2.4466e+00, 1.7392e+00, 4.3164e+00, ..., 3.9286e-01,
5.9117e-01, 1.0348e+00]]]], device='cuda:0'),
tensor([[[[ 4.7772e-01, 1.2445e+00, 1.9651e-01, ..., -3.1796e+00,
-2.8235e+00, -3.7346e-01],
[ 4.6930e-01, 8.2548e-01, 1.5779e+00, ..., -2.9745e+00,
-3.4503e+00, 8.9159e-01],
[-6.2543e-01, -9.6264e-01, -1.8643e-01, ..., -4.7392e+00,
-7.5154e+00, -3.0758e+00],
...,
[-5.4806e-01, -2.4361e+00, -6.9406e+00, ..., -1.2173e+00,
-2.7586e+00, 4.4531e+00],
[-2.2017e+00, -2.7345e+00, -7.5272e+00, ..., 3.3067e+00,
-3.4906e+00, -1.0800e+00],
[-1.5363e+00, -4.5780e+00, -5.9041e+00, ..., -2.0115e+00,
-4.0754e+00, -1.9352e+00]],
[[-1.1049e+00, -1.8380e+00, -3.1049e+00, ..., -8.6367e-01,
-4.2498e+00, -9.6676e-01],
[-4.3053e+00, -4.5125e+00, -3.0314e+00, ..., -1.4491e+00,
-2.5678e-01, 2.3816e+00],
[-4.3283e+00, -4.7787e+00, -3.4313e+00, ..., -1.6201e+00,
-6.3825e+00, -1.7313e-01],
...,
[-8.2154e+00, -6.8519e+00, -9.7766e+00, ..., 3.1051e+00,
1.0070e+00, 1.1405e+00],
[-9.4336e+00, -7.7396e+00, -9.8269e+00, ..., -6.1790e+00,
-1.1622e+01, -9.3312e+00],
[-2.4412e+00, -2.9665e+00, -5.0455e+00, ..., -2.6691e+00,
-3.6064e+00, -2.3174e-02]],
[[-7.4049e-02, 1.3407e+00, 1.7176e+00, ..., -6.6071e-02,
6.8841e-03, 1.6225e+00],
[-6.8530e-02, -2.7609e-02, 5.0780e-01, ..., -4.6323e+00,
-7.4341e+00, -4.9445e+00],
[ 6.1750e-01, 1.3191e+00, 1.3548e+00, ..., 9.7060e-01,
6.1958e-01, -2.3516e+00],
...,
[ 8.5669e-01, -1.1783e+00, 2.5053e+00, ..., -7.3512e+00,
-2.3936e+00, -1.4344e+00],
[ 5.8855e-01, -1.1533e+00, -5.7056e-01, ..., -4.8314e-01,
-1.2210e+00, -1.1703e+00],
[ 8.0593e-01, 4.2099e-01, 7.8916e-01, ..., -2.6415e-01,
6.6150e-01, -1.5621e+00]],
...,
[[-6.5833e-01, -1.6609e+00, -2.1058e+00, ..., 4.1277e-01,
-8.8511e-01, -7.1824e-01],
[-2.8325e+00, -2.6778e+00, -1.1735e+00, ..., -1.3435e+00,
-1.1802e+00, -7.3997e-01],
[-2.2752e+00, -1.6539e+00, -1.3124e+00, ..., -2.4258e+00,
-2.7516e+00, -1.1253e+00],
...,
[-2.5487e+00, -2.1926e+00, -4.4542e+00, ..., -3.4125e+00,
-1.8055e+00, -2.8058e-01],
[-3.4904e+00, -3.3972e+00, -4.8518e+00, ..., -5.0700e+00,
-6.3829e+00, -2.8146e+00],
[-1.8850e+00, -3.0964e+00, -5.2838e+00, ..., -3.5056e-01,
-2.4304e+00, 7.5190e-01]],
[[-1.4004e+00, -2.5046e+00, -3.9678e+00, ..., -4.7771e-01,
-3.4132e+00, -3.8249e+00],
[-3.9313e+00, -3.2515e+00, -4.2351e+00, ..., -2.4064e+00,
-8.1862e-01, -5.5043e-01],
[-4.8356e+00, -3.6497e+00, -3.1930e+00, ..., -7.5666e+00,
-6.4108e+00, -2.6966e+00],
...,
[-4.9118e+00, -5.3356e+00, -4.7500e+00, ..., -1.6895e+00,
-3.6540e+00, 1.9076e+00],
[-4.2163e+00, -1.5840e+00, -8.6425e-01, ..., 3.8817e-01,
-6.6159e+00, 7.5039e-01],
[-3.4465e+00, -2.6620e+00, -1.6040e+00, ..., 5.1329e+00,
-6.0450e-02, 1.7819e+00]],
[[ 1.2409e+00, 2.5325e-02, 1.1513e+00, ..., 8.5761e-01,
-3.2229e-01, 2.2072e+00],
[ 2.4231e+00, 1.2360e+00, 1.0945e+00, ..., -6.0455e+00,
-5.3895e+00, -1.1245e+00],
[ 1.4431e+00, 6.1579e-01, 3.2594e-01, ..., -6.5805e+00,
-9.1734e+00, -4.2886e+00],
...,
[ 3.6197e+00, 3.2454e+00, 9.8538e-01, ..., -4.4271e+00,
-7.4614e+00, 1.8771e-01],
[ 3.0314e+00, 5.4420e+00, 1.2196e+00, ..., 4.5985e+00,
2.9874e+00, 5.7616e+00],
[ 2.2531e+00, 1.9636e+00, 2.8066e-01, ..., 1.8456e+00,
-1.5739e+00, 1.6381e+00]]],
[[[-2.7945e-01, -3.7107e-01, -1.4236e+00, ..., -1.0651e+00,
1.5960e+00, -7.8687e-02],
[ 7.7414e-01, -9.9227e-01, -1.1931e+00, ..., -3.9203e-01,
3.5575e+00, -7.4395e-01],
[ 9.0518e-01, -1.6931e+00, -1.5468e+00, ..., 2.1292e+00,
2.3370e+00, -2.4805e+00],
...,
[-1.2564e+00, -1.3633e+00, 8.6091e-01, ..., -8.6762e+00,
-4.5471e+00, 1.6752e+00],
[-1.6884e+00, -1.0887e+00, 1.5621e+00, ..., -2.2252e+00,
1.3970e+00, 3.0634e+00],
[-2.1516e+00, -3.4035e+00, -2.5449e+00, ..., 5.6757e-01,
1.7439e+00, 5.1902e-01]],
[[-9.5275e-01, -1.8555e+00, -2.8093e+00, ..., -1.3506e+00,
-2.2057e+00, -1.8089e+00],
[-2.9315e+00, -3.4245e+00, -1.1495e+00, ..., -7.7824e-01,
-3.5547e+00, -3.9781e+00],
[-3.1798e+00, -6.6428e+00, -6.5955e+00, ..., -3.7130e+00,
-9.4314e+00, -7.5649e+00],
...,
[-6.3097e+00, -9.6901e+00, -8.3007e+00, ..., -6.2257e+00,
-6.2105e+00, 2.0840e+00],
[-9.2451e+00, -7.1508e+00, -4.4060e+00, ..., -1.3552e+00,
-5.5297e-01, -5.2253e-01],
[-2.6087e+00, -5.5521e+00, -3.1473e+00, ..., 1.3961e-02,
-4.7535e-01, -1.4856e+00]],
[[ 1.1144e+00, 2.1156e+00, 2.0372e+00, ..., 1.3464e+00,
3.0101e+00, 1.5236e+00],
[ 1.0850e+00, 7.6183e-01, -6.2363e-01, ..., 3.4024e-02,
1.0726e+00, 1.4441e+00],
[ 3.9603e-01, 2.6421e+00, 2.2841e+00, ..., 5.3567e-01,
1.1045e+00, 1.4498e+00],
...,
[-2.4570e-01, -1.0947e+00, -3.3742e+00, ..., -2.8487e-01,
-1.8991e+00, 1.3391e+00],
[-1.8951e+00, 3.0080e-01, -1.5913e+00, ..., 1.7218e+00,
1.4063e+00, -4.2941e-01],
[ 8.6003e-02, 3.7627e-01, -4.6588e-01, ..., -4.9568e-01,
-2.2894e+00, -2.4265e+00]],
...,
[[-1.4006e+00, -2.1685e+00, -3.5429e+00, ..., -2.0739e+00,
-2.3010e+00, -1.3140e+00],
[-1.9283e+00, -2.0282e+00, -6.6627e-01, ..., -4.5376e+00,
-5.2196e+00, -3.9783e+00],
[-1.7141e+00, -3.2205e+00, -2.6596e+00, ..., -3.8460e-01,
-4.2707e+00, -4.2777e+00],
...,
[-2.4261e+00, -4.2042e+00, -3.8949e+00, ..., -6.3441e+00,
-5.2767e+00, -3.9254e+00],
[-2.7409e+00, -4.0368e+00, -2.3305e+00, ..., -2.6560e+00,
-2.4066e+00, -2.5157e+00],
[-2.9859e+00, -2.9404e+00, -8.2086e-01, ..., 1.3359e-02,
6.5672e-01, 2.8377e-01]],
[[-2.0839e+00, -3.4093e+00, -5.7457e+00, ..., -5.0001e+00,
-4.6533e+00, -9.8937e-01],
[-4.7540e+00, -3.5659e+00, -3.4844e+00, ..., -5.3536e+00,
-7.3823e+00, -2.3869e+00],
[-4.5544e+00, -4.3707e+00, -4.1704e+00, ..., -1.0672e+00,
-4.2000e+00, -2.4574e+00],
...,
[-3.8921e+00, 1.1916e+00, -6.2078e-01, ..., -5.0857e+00,
-4.0508e+00, -2.0832e+00],
[-3.2918e+00, 1.6190e+00, -1.7962e+00, ..., -4.5512e+00,
-1.4912e+00, 1.1560e+00],
[-3.3504e+00, -1.8060e+00, -2.3392e+00, ..., -1.9299e+00,
-8.2914e-01, 7.3163e-01]],
[[ 1.0995e+00, 3.6339e-01, 2.9258e-01, ..., -4.5718e-01,
3.0492e+00, 1.2381e+00],
[ 2.1086e+00, 1.0015e+00, -5.0194e-01, ..., -4.9106e+00,
1.8790e+00, 3.2958e+00],
[ 1.9607e+00, 7.0561e-01, 1.2413e+00, ..., 9.1801e-01,
3.4473e+00, 1.9659e+00],
...,
[ 2.8624e+00, 4.2948e+00, 4.7076e+00, ..., -7.6891e+00,
-5.5414e+00, -2.1788e-01],
[ 4.2459e+00, 2.4270e+00, 2.8957e+00, ..., -8.1226e+00,
-3.9431e+00, 2.3314e+00],
[ 3.1243e+00, 2.9372e+00, 2.1500e+00, ..., -1.0391e+00,
8.2896e-01, 1.0912e+00]]],
[[[ 8.1833e-01, 1.6383e+00, 6.8520e-01, ..., -3.8118e+00,
-2.6480e+00, -1.1028e+00],
[ 1.6586e+00, 1.0562e+00, 7.6722e-01, ..., -2.9882e+00,
-2.4890e+00, -2.5438e-01],
[ 6.2483e-01, 6.4484e-01, 9.6730e-01, ..., -2.5235e+00,
-2.9795e+00, -9.7339e-02],
...,
[ 1.1391e+00, 1.0581e-01, -1.0382e-01, ..., -3.7716e+00,
-4.2263e+00, 7.8669e-02],
[ 1.0685e-01, -3.5068e-01, -5.9636e-01, ..., -3.7870e+00,
-4.0895e+00, 3.3372e-01],
[-3.6051e-01, -3.5280e+00, -3.2642e+00, ..., -4.1900e+00,
-3.3160e+00, -1.4516e+00]],
[[ 3.3215e-01, -1.1523e+00, -2.6104e+00, ..., -7.5770e+00,
-7.5349e+00, -1.4129e+00],
[-1.7510e+00, -3.5170e+00, -4.1969e+00, ..., -7.8178e+00,
-8.5424e+00, -2.4579e+00],
[ 1.5224e-02, -1.2802e+00, -2.4732e+00, ..., -7.6463e+00,
-7.8393e+00, -2.5484e+00],
...,
[-1.8939e+00, -3.6179e+00, -3.2742e+00, ..., -1.1513e+01,
-1.0129e+01, -2.0211e+00],
[-4.2079e+00, -5.8371e+00, -5.3670e+00, ..., -9.7675e+00,
-1.0826e+01, -4.2738e+00],
[-3.1464e-01, -3.2432e+00, -3.5292e+00, ..., -4.1843e+00,
-3.6194e+00, -8.6872e-01]],
[[ 3.8855e-01, 2.1115e+00, 1.8581e+00, ..., 8.8604e-01,
2.2382e+00, 9.8236e-01],
[-1.3818e-01, 1.0635e+00, 7.9299e-01, ..., 1.0887e+00,
2.8700e+00, 8.2741e-02],
[ 6.9685e-01, 3.2188e+00, 1.6601e+00, ..., -7.0241e-01,
1.6978e+00, -3.5573e-01],
...,
[ 4.2794e-01, 2.6903e+00, 6.7636e-01, ..., -5.3093e-01,
7.9369e-01, -1.5488e+00],
[ 6.9547e-01, 2.2090e+00, 9.4567e-01, ..., -4.5002e-01,
5.3719e-01, -1.4810e+00],
[-3.3609e-01, 1.7561e+00, 9.3232e-01, ..., -4.6908e-01,
-3.9465e-01, -1.7818e+00]],
...,
[[-2.9653e-01, -8.9810e-01, -1.5565e+00, ..., -5.5921e+00,
-5.5222e+00, -7.8473e-01],
[-2.5799e+00, -4.2728e+00, -2.4981e+00, ..., -4.5607e+00,
-5.2831e+00, -1.8703e+00],
[-8.8487e-01, -1.5284e+00, -5.8837e-01, ..., -5.9143e+00,
-5.8840e+00, -1.3329e+00],
...,
[-9.0997e-01, -2.3536e+00, -4.2550e-01, ..., -5.5201e+00,
-5.8884e+00, -1.5894e+00],
[-1.9365e+00, -3.8920e+00, -2.2424e+00, ..., -2.1162e+00,
-4.7511e+00, -1.1491e+00],
[-1.1480e+00, -3.6994e+00, -2.4871e+00, ..., -1.7052e+00,
-2.3294e+00, 2.9129e-01]],
[[-1.0083e+00, -8.1360e-01, -9.9702e-01, ..., -2.1655e+00,
-5.0902e-01, -6.0145e-01],
[-3.0280e+00, -2.8998e+00, -1.8761e+00, ..., -3.7803e+00,
-1.7668e+00, -4.2504e-01],
[-2.0546e+00, -2.9386e+00, -9.0573e-01, ..., -2.5634e+00,
-1.9757e+00, -4.1730e-01],
...,
[-2.3820e+00, -4.5345e+00, -3.0232e+00, ..., -3.5432e+00,
-2.0489e+00, -4.6838e-01],
[-3.0562e+00, -2.3224e+00, -7.8193e-01, ..., -1.7932e+00,
-9.1928e-01, 1.2094e+00],
[-3.2638e+00, -3.2754e+00, -1.1305e+00, ..., 3.1402e-01,
-6.2203e-01, 8.0120e-01]],
[[ 8.6973e-01, 8.1325e-01, 1.0162e+00, ..., -2.5291e-01,
6.3101e-01, -1.4162e+00],
[ 1.5577e+00, 1.8501e+00, 1.7338e+00, ..., 3.3089e+00,
3.0735e+00, 2.2975e+00],
[ 5.0881e-01, 4.2706e-02, 1.1099e+00, ..., 2.8092e+00,
1.6265e+00, 1.3088e+00],
...,
[ 1.9332e+00, 1.4545e+00, 1.4975e+00, ..., 2.4753e+00,
2.1258e+00, 6.6528e-01],
[ 3.2126e+00, 3.4196e+00, 2.5696e+00, ..., 1.8802e+00,
1.7528e+00, 2.0774e+00],
[ 1.9302e+00, 4.0733e-01, 5.5859e-01, ..., 4.8866e-01,
-7.3354e-02, 8.0480e-01]]],
[[[ 7.0087e-01, 7.4564e-01, -1.5712e-01, ..., 4.0356e-02,
2.5134e-01, -1.5530e-01],
[ 1.5867e+00, 1.2472e+00, 1.7992e+00, ..., 2.3161e-01,
3.9581e-01, -4.6083e-01],
[ 1.9079e-01, -8.7043e-01, -3.8269e+00, ..., 1.5186e-01,
-1.4092e-01, -8.3831e-01],
...,
[ 1.7640e+00, 1.2107e+00, 2.3112e+00, ..., 1.6599e-01,
2.7341e-01, -4.3291e-01],
[-1.8902e+00, -1.6884e+00, 6.6673e-02, ..., 6.0427e-01,
1.6315e-01, -1.2076e+00],
[-1.0243e+00, -3.5504e+00, -3.1312e+00, ..., -2.4474e+00,
-2.8274e+00, -2.6862e+00]],
[[-1.4000e+00, -3.8310e+00, -6.9954e+00, ..., -2.8352e+00,
-3.7922e+00, -2.1186e+00],
[-4.8232e+00, -6.4951e+00, -6.5792e+00, ..., -3.2774e+00,
-3.9579e+00, -3.3989e+00],
[-2.8047e+00, -7.1619e+00, -1.0638e+01, ..., -2.7093e+00,
-2.7658e+00, -2.5985e+00],
...,
[-4.9179e+00, -5.7253e+00, -6.5954e+00, ..., -2.3507e+00,
-3.1781e+00, -2.6339e+00],
[-6.9984e+00, -9.5479e+00, -8.8143e+00, ..., -3.7090e+00,
-4.9028e+00, -5.0313e+00],
[-1.6585e+00, -4.3104e+00, -4.5428e+00, ..., -2.9690e+00,
-2.7251e+00, -1.5934e+00]],
[[ 7.2574e-01, 3.1051e+00, 2.0328e+00, ..., 1.7915e+00,
1.5202e+00, 9.3285e-01],
[-5.9815e-02, 8.6025e-01, 7.2576e-01, ..., 1.1799e+00,
1.1010e+00, 8.9303e-01],
[ 9.6315e-02, 2.2965e+00, 2.1645e+00, ..., 1.7575e+00,
1.2970e+00, 1.3254e+00],
...,
[-2.5752e+00, -2.1089e+00, -2.0444e+00, ..., 1.0932e+00,
1.3516e+00, 6.4964e-01],
[-2.3394e-01, -4.6458e-01, -1.3472e+00, ..., 7.1296e-01,
1.1977e+00, 5.4112e-01],
[ 5.4287e-01, 1.1594e+00, 5.2343e-01, ..., 8.8397e-01,
1.6280e+00, 6.0759e-01]],
...,
[[-8.4012e-01, -3.1109e+00, -2.9112e+00, ..., -1.5731e+00,
-2.1941e+00, -4.5462e-01],
[-2.2678e+00, -3.9455e+00, -1.4908e+00, ..., -2.0726e+00,
-3.6008e+00, -2.9657e+00],
[-2.1504e+00, -4.1981e+00, -4.9282e+00, ..., -1.0145e+00,
-3.1832e+00, -1.8460e+00],
...,
[-1.0305e+00, -3.6038e+00, -1.2724e+00, ..., -1.0949e+00,
-2.6586e+00, -8.4343e-01],
[-2.3266e+00, -4.4047e+00, -1.4920e+00, ..., -6.8488e-01,
-3.0519e+00, -1.5597e+00],
[-2.1507e+00, -2.9943e+00, -1.4746e+00, ..., -1.8163e+00,
-2.4934e+00, -5.6311e-01]],
[[-2.3787e+00, -3.3263e+00, -1.7742e+00, ..., 5.7450e-01,
-1.3243e+00, -6.5541e-01],
[-4.3930e+00, -3.0326e+00, -1.3255e+00, ..., -2.7734e-01,
-3.9158e+00, -1.9176e+00],
[-4.0647e+00, -4.5197e+00, -4.0750e+00, ..., -1.0168e+00,
-4.6289e+00, -2.4881e+00],
...,
[-1.2068e+00, -3.5293e+00, -6.9144e-01, ..., -3.8325e-01,
-3.3190e+00, -2.1668e+00],
[-2.5749e+00, 1.3749e-01, 3.1169e-01, ..., 9.9999e-02,
-3.3541e+00, -1.9682e+00],
[-3.5519e+00, -1.9850e+00, 2.1821e-02, ..., -4.1348e-01,
-2.5922e+00, -1.2699e+00]],
[[ 1.5105e+00, 1.2445e+00, 1.8622e+00, ..., 6.8594e-01,
1.1474e+00, 6.3294e-01],
[ 3.2559e+00, 3.4742e+00, 4.0425e+00, ..., 1.4290e+00,
1.0713e+00, 2.5203e+00],
[ 1.2034e+00, 1.7870e+00, 1.2220e+00, ..., 8.2751e-01,
8.0805e-02, 7.1123e-01],
...,
[ 5.3453e+00, 3.7129e+00, 3.9000e+00, ..., 1.4328e+00,
3.8300e-01, 5.6011e-01],
[ 3.7540e+00, 3.3632e+00, 1.2356e+00, ..., 1.8813e+00,
7.4962e-01, 1.2503e+00],
[ 2.4306e+00, 1.5811e+00, 1.5314e+00, ..., 6.6688e-01,
-2.0776e-01, -1.4501e-01]]],
[[[-6.2922e-01, 6.2260e-01, 1.3297e+00, ..., -9.3087e-01,
9.0532e-02, -2.0809e-01],
[-2.1572e-01, 4.5393e-01, 2.7601e+00, ..., -9.1098e-01,
-1.8453e-01, -1.0679e+00],
[-1.2642e+00, 1.2111e+00, 8.4989e-01, ..., -7.3081e-01,
-7.8027e-01, -1.6098e+00],
...,
[-1.9056e-01, 4.0853e-01, -6.7534e-01, ..., -6.3719e-01,
-8.6535e-02, -6.5661e-01],
[-1.8925e+00, 5.0169e-01, 9.4200e-01, ..., 3.5791e-02,
-9.9223e-03, -1.3958e+00],
[-1.4556e+00, -2.6476e+00, -1.9131e+00, ..., -2.9609e+00,
-2.9129e+00, -2.6037e+00]],
[[-2.0493e+00, -2.3257e+00, -3.3779e+00, ..., -4.1425e+00,
-4.1936e+00, -2.3167e+00],
[-4.7310e+00, -5.2540e+00, -5.0812e+00, ..., -2.3496e+00,
-5.2988e+00, -3.9150e+00],
[-3.3882e+00, -4.7332e+00, -6.6865e+00, ..., -2.3048e+00,
-3.6810e+00, -3.2539e+00],
...,
[-4.6925e+00, -4.4014e+00, -3.7202e+00, ..., -2.8759e+00,
-3.4260e+00, -2.9520e+00],
[-7.7069e+00, -7.7756e+00, -6.9744e+00, ..., -5.2326e+00,
-5.5423e+00, -4.9700e+00],
[-1.4536e+00, -3.0733e+00, -1.7944e+00, ..., -3.1893e+00,
-2.9079e+00, -1.4553e+00]],
[[-1.9256e-01, 1.8560e+00, 1.8374e+00, ..., 1.9776e+00,
1.7706e+00, 8.2767e-01],
[-4.2807e-01, 2.5034e-01, 5.8717e-01, ..., 2.3677e+00,
1.5730e+00, 8.0490e-01],
[-5.1676e-02, 2.7159e+00, 3.6597e-01, ..., 4.7591e+00,
2.7425e+00, 1.1644e+00],
...,
[-7.9514e-01, -6.1929e-01, 1.4429e+00, ..., 5.9069e-01,
1.2119e+00, 1.0256e+00],
[-1.3448e+00, -6.7073e-01, -3.3061e-01, ..., 1.4371e+00,
1.2680e+00, 5.3968e-01],
[-2.4581e-01, -1.6096e-01, -1.5771e-01, ..., 1.9213e+00,
1.6338e+00, 4.6001e-01]],
...,
[[-1.6356e+00, -1.6549e+00, -1.6669e+00, ..., -2.0208e+00,
-2.6317e+00, -5.5793e-01],
[-3.6946e+00, -3.3078e+00, -2.3601e+00, ..., -2.4542e+00,
-4.1928e+00, -3.6669e+00],
[-2.6709e+00, -1.8559e+00, -2.7146e+00, ..., -3.2744e+00,
-4.1173e+00, -2.5844e+00],
...,
[-1.1608e+00, -1.4306e+00, -1.4450e+00, ..., -1.5892e+00,
-2.8828e+00, -1.2862e+00],
[-2.4689e+00, -3.7125e+00, -7.4561e-01, ..., -2.4635e+00,
-3.2967e+00, -1.6819e+00],
[-1.5756e+00, -1.6872e+00, -7.5355e-01, ..., -2.2463e+00,
-2.0970e+00, -4.2179e-01]],
[[-1.9576e+00, -9.8986e-01, -2.9751e+00, ..., -3.7976e-01,
-1.3136e+00, -6.4725e-01],
[-4.1159e+00, -1.3611e+00, -3.7728e+00, ..., -5.1152e-01,
-3.2669e+00, -1.7663e+00],
[-3.5251e+00, -3.2016e+00, -7.8079e+00, ..., -1.0100e-01,
-2.0103e+00, -1.8272e+00],
...,
[-2.5548e+00, -2.1884e+00, -6.0660e+00, ..., -7.2861e-01,
-2.8275e+00, -2.0127e+00],
[-1.6778e+00, 5.4650e-01, -1.9445e+00, ..., -1.4091e-02,
-3.1337e+00, -1.9837e+00],
[-1.9943e+00, -1.1726e+00, -1.5913e+00, ..., -1.3536e+00,
-2.5218e+00, -1.2662e+00]],
[[ 1.2800e+00, 2.0506e-01, 7.2209e-01, ..., 8.2549e-01,
1.2033e+00, 6.5125e-01],
[ 4.0477e+00, 2.9022e+00, 2.9489e+00, ..., 2.3134e+00,
1.3683e+00, 2.3182e+00],
[ 1.4840e+00, 6.3097e-01, 1.3656e+00, ..., 2.5018e+00,
1.2436e+00, 8.6447e-01],
...,
[ 2.9979e+00, 2.1969e+00, 1.2510e+00, ..., 2.2440e+00,
5.7260e-01, 5.0253e-01],
[ 3.5827e+00, 4.0068e+00, 2.1563e+00, ..., 1.6652e+00,
7.8305e-01, 1.2033e+00],
[ 2.3106e+00, 2.5072e+00, 1.6736e+00, ..., -3.4049e-01,
-3.4976e-01, -8.9215e-02]]]], device='cuda:0'),
tensor([[[[ 3.5592e-01, 1.1702e+00, 7.7375e-01, ..., 5.3260e-01,
-1.9531e-01, 6.9042e-01],
[-1.3509e-01, -8.4999e-01, 9.4327e-01, ..., 4.4353e-01,
-8.9764e-01, -2.9786e-01],
[-2.2796e-02, -2.3241e-01, 1.0770e+00, ..., 8.3379e-02,
-3.1592e-02, -8.1169e-01],
...,
[-6.9011e-01, -1.2067e+00, -9.4947e-01, ..., -2.9763e+00,
-3.6165e+00, -9.2328e-01],
[-2.2506e+00, -1.6105e+00, 3.4513e-01, ..., -2.9488e+00,
-3.0342e+00, 3.0979e-02],
[-2.3232e+00, -4.0368e+00, -2.9878e+00, ..., -4.0925e+00,
-3.5472e+00, -1.4979e+00]],
[[-1.7428e+00, -2.1246e+00, -2.3191e+00, ..., -1.2295e+00,
-5.2577e+00, -1.0259e+00],
[-5.3040e+00, -6.7612e+00, -4.5780e+00, ..., -4.1681e+00,
-7.2278e+00, -3.9397e+00],
[-3.9725e+00, -4.6058e+00, -3.2068e+00, ..., -4.6629e+00,
-5.7035e+00, -4.3070e+00],
...,
[-6.8896e+00, -8.5454e+00, -7.2301e+00, ..., -8.0887e+00,
-9.2924e+00, -2.2899e+00],
[-9.8112e+00, -1.0340e+01, -8.8859e+00, ..., -1.0840e+01,
-1.0969e+01, -3.7737e+00],
[-2.5029e+00, -4.9064e+00, -4.2556e+00, ..., -4.1800e+00,
-4.2313e+00, -6.3055e-01]],
[[-1.0889e-01, 9.5839e-01, 5.3225e-01, ..., 1.2776e+00,
2.0419e+00, 2.7969e-01],
[ 3.6618e-01, 2.4878e-01, 1.0824e-01, ..., 3.5536e-01,
2.1016e+00, -7.5190e-01],
[ 8.6914e-01, 2.2815e+00, 1.5696e+00, ..., 1.1952e+00,
2.2038e+00, 8.3460e-01],
...,
[ 2.8303e-01, 9.6431e-01, 5.2680e-01, ..., 1.2959e+00,
7.0169e-01, -8.9704e-01],
[-6.8907e-01, 5.2543e-01, -2.9458e-01, ..., 2.4972e-01,
6.3552e-01, -5.4279e-01],
[ 4.9896e-01, 1.3824e+00, 1.1782e+00, ..., -2.0138e-01,
-4.3177e-01, -1.9388e+00]],
...,
[[-1.0626e+00, -1.1490e+00, -7.4940e-01, ..., -1.6557e+00,
-2.3057e+00, 2.1120e-01],
[-4.3637e+00, -4.9510e+00, -2.7047e+00, ..., -4.1562e+00,
-5.0686e+00, -3.1349e+00],
[-2.2124e+00, -1.3494e+00, -4.7997e-01, ..., -2.2327e+00,
-3.8145e+00, -2.0385e+00],
...,
[-2.3810e+00, -3.2162e+00, -2.2045e+00, ..., -4.3497e+00,
-5.9749e+00, -3.0073e+00],
[-3.6169e+00, -5.7566e+00, -2.4877e+00, ..., -4.3485e+00,
-5.3509e+00, -1.3462e+00],
[-3.1992e+00, -3.7410e+00, -3.2354e+00, ..., -2.7443e+00,
-2.6355e+00, 4.4968e-01]],
[[-1.1464e+00, 1.6454e-01, -1.2408e+00, ..., -7.7693e-01,
2.0007e-01, 1.8985e-01],
[-4.2085e+00, -1.1564e+00, -1.8627e+00, ..., -1.9740e+00,
-2.0721e+00, 2.9253e-01],
[-3.5449e+00, -1.3741e+00, -1.4446e+00, ..., -1.0733e+00,
-2.6681e+00, -6.5360e-01],
...,
[-3.7622e+00, -2.0922e+00, -3.7575e+00, ..., -3.0940e+00,
-2.5688e+00, -9.2580e-01],
[-3.3409e+00, -7.0189e-01, -1.5479e+00, ..., -1.9427e+00,
-1.0102e+00, 3.3247e-01],
[-3.7393e+00, -3.0621e+00, -2.8967e+00, ..., -1.6707e+00,
-1.1765e+00, 3.4938e-01]],
[[ 1.7955e+00, 5.4759e-01, 6.0808e-01, ..., 6.9202e-01,
1.6064e+00, 1.6508e-01],
[ 3.3263e+00, 2.0658e+00, 1.9098e+00, ..., 1.8794e+00,
2.8894e+00, 3.5339e+00],
[ 2.0475e+00, 3.8672e-01, 6.8410e-01, ..., 1.9860e+00,
2.3744e+00, 2.0744e+00],
...,
[ 2.9246e+00, 2.8484e+00, 2.1094e+00, ..., 1.7776e+00,
1.3830e+00, 1.3529e-01],
[ 4.8479e+00, 4.6394e+00, 3.9775e+00, ..., 2.9467e+00,
1.8326e+00, 1.4815e+00],
[ 2.8075e+00, 2.6436e+00, 2.1579e+00, ..., 5.0249e-01,
6.3249e-01, 1.0736e+00]]],
[[[ 3.2635e-01, 1.3295e+00, 6.7380e-03, ..., -3.2334e+00,
-2.5180e+00, -1.1098e+00],
[ 3.4778e-01, 8.3190e-01, 1.2223e+00, ..., -1.5105e+00,
-2.0862e+00, -2.0316e-01],
[ 2.7098e-01, 4.0699e-01, 1.5089e-01, ..., -5.7965e-01,
-2.3040e+00, 1.2823e-01],
...,
[-5.2843e-01, -1.3945e+00, 3.6053e-01, ..., -1.3451e+00,
-3.7952e+00, -2.2809e-01],
[-2.0815e+00, -1.7349e+00, -1.0213e+00, ..., -1.4635e+00,
-2.6012e+00, 2.9069e-01],
[-1.1087e+00, -3.1970e+00, -3.2675e+00, ..., -3.9738e+00,
-3.4002e+00, -1.7593e+00]],
[[-1.0709e+00, -1.6035e+00, -3.4779e+00, ..., -6.2227e+00,
-7.3025e+00, -1.3899e+00],
[-3.9324e+00, -4.2564e+00, -4.1443e+00, ..., -5.4285e+00,
-8.0285e+00, -2.3285e+00],
[-4.5143e+00, -5.4993e+00, -4.3752e+00, ..., -3.9144e+00,
-7.2255e+00, -2.3198e+00],
...,
[-5.6561e+00, -6.1487e+00, -5.3436e+00, ..., -9.1391e+00,
-1.0941e+01, -3.0289e+00],
[-7.5984e+00, -7.5905e+00, -7.3524e+00, ..., -8.0756e+00,
-9.8548e+00, -4.2993e+00],
[-1.7265e+00, -3.2427e+00, -3.7008e+00, ..., -3.6455e+00,
-3.8407e+00, -9.9414e-01]],
[[-2.2567e-02, 1.2677e+00, 1.7086e+00, ..., 1.6581e+00,
2.5131e+00, 9.7782e-01],
[ 3.3294e-02, 2.0460e-01, 1.2874e+00, ..., 1.6232e+00,
3.2100e+00, 1.2439e-01],
[ 7.1980e-01, 6.8740e-01, 6.3633e-01, ..., -3.4543e-01,
2.0280e+00, -3.1112e-01],
...,
[ 3.5862e-01, 7.2324e-01, 4.8277e-01, ..., -4.1779e-01,
6.3605e-01, -1.3193e+00],
[ 5.0056e-01, 8.8434e-01, 1.4683e+00, ..., -1.8946e-01,
1.0473e+00, -1.6221e+00],
[ 5.3125e-01, 1.1369e+00, 1.5849e+00, ..., -5.3846e-01,
-8.2605e-01, -1.9540e+00]],
...,
[[-5.9164e-01, -1.5756e+00, -2.3158e+00, ..., -5.1395e+00,
-5.3201e+00, -7.9814e-01],
[-2.7191e+00, -2.7675e+00, -2.0205e+00, ..., -3.4073e+00,
-4.7371e+00, -1.6119e+00],
[-1.9982e+00, -1.5939e+00, -2.1742e+00, ..., -3.7292e+00,
-4.9972e+00, -1.0866e+00],
...,
[-2.0614e+00, -1.7373e+00, -8.9868e-01, ..., -4.5181e+00,
-6.4820e+00, -1.7404e+00],
[-3.2809e+00, -3.3888e+00, -3.1299e+00, ..., -2.3195e+00,
-4.6095e+00, -9.4027e-01],
[-1.4593e+00, -1.9566e+00, -2.4066e+00, ..., -1.8825e+00,
-2.4375e+00, 2.3529e-01]],
[[-1.3208e+00, -2.4665e+00, -4.0595e+00, ..., -1.5658e+00,
-4.1604e-01, -5.1099e-01],
[-3.6230e+00, -3.3480e+00, -3.9507e+00, ..., -3.3214e+00,
-1.8019e+00, -2.2617e-01],
[-4.6275e+00, -2.9595e+00, -2.7942e+00, ..., -1.9582e+00,
-1.5572e+00, -2.4582e-01],
...,
[-4.4362e+00, -3.8115e+00, -4.2202e+00, ..., -3.5991e+00,
-2.3898e+00, -4.8568e-01],
[-4.1014e+00, -2.0893e+00, -2.9789e+00, ..., -8.8366e-01,
-1.3132e+00, 9.5659e-01],
[-3.1685e+00, -3.2842e+00, -2.7716e+00, ..., -3.7738e-02,
-9.2936e-01, 7.0896e-01]],
[[ 1.2023e+00, -6.0278e-02, 1.1455e+00, ..., 6.0574e-01,
8.6789e-01, -1.4595e+00],
[ 2.2406e+00, 1.1339e+00, 1.5836e+00, ..., 3.6219e+00,
3.3054e+00, 2.3292e+00],
[ 1.8198e+00, 1.2667e+00, 3.7337e-01, ..., 2.8900e+00,
1.9019e+00, 1.2934e+00],
...,
[ 2.9498e+00, 2.2118e+00, 2.2396e+00, ..., 3.1252e+00,
2.5307e+00, 1.2929e+00],
[ 2.8819e+00, 4.0482e+00, 3.0168e+00, ..., 1.4939e+00,
1.0430e+00, 2.3217e+00],
[ 1.9443e+00, 1.6328e+00, 4.2027e-01, ..., 2.7209e-01,
-3.0103e-01, 7.9485e-01]]],
[[[-2.3082e-01, 1.0604e+00, 1.0563e+00, ..., -1.2024e+00,
-1.2387e-01, -1.8603e-01],
[ 2.4118e-02, 5.2776e-01, 2.7558e+00, ..., -8.2609e-01,
6.6702e-02, -5.3555e-01],
[-1.2315e+00, 5.3955e-01, 1.7907e+00, ..., 3.4461e-01,
-4.7974e-01, -9.4524e-01],
...,
[-1.3727e-01, 3.3079e-01, -1.6644e-01, ..., 6.4768e-01,
7.0212e-01, -1.3636e-01],
[-1.6400e+00, -5.1847e-02, 2.0785e-01, ..., 4.6355e-01,
7.5136e-01, -9.5036e-01],
[-1.2920e+00, -2.4210e+00, -1.8126e+00, ..., -3.0992e+00,
-2.8489e+00, -2.7520e+00]],
[[-1.5161e+00, -2.0687e+00, -3.3131e+00, ..., -4.0587e+00,
-3.8971e+00, -2.1244e+00],
[-4.5907e+00, -4.7112e+00, -4.1457e+00, ..., -4.7106e+00,
-4.3918e+00, -3.3088e+00],
[-3.7025e+00, -5.1989e+00, -5.4652e+00, ..., -1.9752e+00,
-2.9272e+00, -2.7173e+00],
...,
[-4.5883e+00, -5.7468e+00, -4.6562e+00, ..., -2.5422e+00,
-2.7455e+00, -2.2728e+00],
[-7.7456e+00, -8.0049e+00, -6.5879e+00, ..., -4.4738e+00,
-5.0026e+00, -4.8189e+00],
[-1.4593e+00, -3.1286e+00, -1.9851e+00, ..., -2.9570e+00,
-2.8821e+00, -1.6869e+00]],
[[-2.5027e-01, 2.2934e+00, 1.6581e+00, ..., 1.4050e+00,
1.8407e+00, 9.1439e-01],
[-6.5918e-01, 1.4720e-01, 6.8350e-01, ..., 2.1265e+00,
1.1986e+00, 8.1215e-01],
[-1.2686e-02, 2.3499e+00, 8.3750e-01, ..., 2.7680e+00,
1.8597e+00, 1.2189e+00],
...,
[-7.1878e-01, 8.3596e-02, 2.4556e-01, ..., 1.5986e+00,
9.0565e-01, 7.0740e-01],
[-1.1801e+00, -6.3500e-03, 4.2557e-01, ..., 1.3456e+00,
6.9705e-01, 4.2524e-01],
[-3.4657e-01, -3.5934e-01, 1.3276e-01, ..., 1.6502e+00,
1.5124e+00, 4.5824e-01]],
...,
[[-1.3564e+00, -1.9094e+00, -1.1381e+00, ..., -2.3900e+00,
-2.7084e+00, -4.5268e-01],
[-3.4492e+00, -3.3129e+00, -8.4301e-01, ..., -2.4070e+00,
-4.1702e+00, -3.0390e+00],
[-2.5664e+00, -2.0944e+00, -8.2238e-01, ..., -1.0173e+00,
-3.4242e+00, -1.8438e+00],
...,
[-1.0292e+00, -2.2747e+00, -1.0905e+00, ..., -1.4011e+00,
-1.7037e+00, -8.5852e-01],
[-2.2330e+00, -4.1664e+00, -1.6014e+00, ..., -1.2882e+00,
-2.4708e+00, -1.5433e+00],
[-1.4426e+00, -1.5469e+00, -6.9936e-01, ..., -1.9315e+00,
-2.2757e+00, -5.9369e-01]],
[[-1.8296e+00, -1.9361e+00, -2.4666e+00, ..., -7.4022e-01,
-1.3429e+00, -6.4802e-01],
[-3.9513e+00, -1.7357e+00, -2.8051e+00, ..., 2.4261e-01,
-3.3634e+00, -1.8218e+00],
[-3.6839e+00, -3.0844e+00, -5.4653e+00, ..., 1.8178e-01,
-3.5283e+00, -2.4716e+00],
...,
[-2.6122e+00, -2.7916e+00, -3.4115e+00, ..., -1.3320e+00,
-3.2933e+00, -2.1405e+00],
[-1.9297e+00, -4.7572e-02, -1.5417e+00, ..., -9.0549e-01,
-3.0795e+00, -1.8783e+00],
[-2.1461e+00, -1.4224e+00, -1.4911e+00, ..., -1.7734e+00,
-2.7546e+00, -1.1536e+00]],
[[ 1.1943e+00, 2.7499e-01, 7.6778e-01, ..., 1.4249e-01,
9.2630e-01, 6.7068e-01],
[ 3.9541e+00, 2.7817e+00, 2.2555e+00, ..., 1.9715e+00,
7.9856e-01, 2.3831e+00],
[ 1.5073e+00, 5.4429e-01, 1.1945e+00, ..., 7.1616e-01,
-1.2386e-01, 6.0977e-01],
...,
[ 2.8453e+00, 2.4704e+00, 1.7828e+00, ..., 2.4968e-01,
2.7494e-01, 5.2983e-01],
[ 3.7162e+00, 4.0172e+00, 2.6384e+00, ..., 9.6929e-01,
8.9887e-01, 1.0783e+00],
[ 2.4346e+00, 2.4143e+00, 1.6448e+00, ..., -1.8708e-01,
-3.6214e-01, -1.6566e-01]]],
[[[ 2.8689e-01, 1.3228e+00, -4.4446e-02, ..., -3.3833e+00,
-2.4449e+00, -1.0510e+00],
[ 7.5754e-01, 4.5554e-01, 8.4187e-01, ..., -1.6573e+00,
-2.8244e+00, -5.9582e-01],
[-1.7967e+00, -1.8745e+00, -4.1839e-01, ..., -4.9957e+00,
-3.9553e+00, -9.7562e-01],
...,
[-7.5560e-01, -1.8687e+00, -9.6018e-01, ..., -2.9563e+00,
-3.4863e+00, 7.9301e-01],
[-2.0037e+00, -1.4815e+00, -5.9722e-01, ..., -2.9097e+00,
-3.5170e+00, 6.8925e-01],
[-1.0786e+00, -3.1545e+00, -2.9977e+00, ..., -3.9745e+00,
-3.2321e+00, -1.4169e+00]],
[[-1.0268e+00, -2.1535e+00, -3.7061e+00, ..., -7.1026e+00,
-7.2374e+00, -1.3450e+00],
[-2.0943e+00, -3.7007e+00, -3.7688e+00, ..., -9.5248e+00,
-1.0591e+01, -3.2196e+00],
[-3.9472e+00, -3.8376e+00, -2.4954e+00, ..., -6.5735e+00,
-9.6115e+00, -3.4556e+00],
...,
[-5.8688e+00, -5.7664e+00, -4.6404e+00, ..., -8.4870e+00,
-9.0349e+00, -1.9783e+00],
[-7.3574e+00, -7.4713e+00, -7.4950e+00, ..., -9.5305e+00,
-1.0562e+01, -4.1690e+00],
[-1.6488e+00, -2.9276e+00, -3.6007e+00, ..., -3.5948e+00,
-3.7875e+00, -7.7554e-01]],
[[ 3.4560e-01, 1.3482e+00, 1.7371e+00, ..., 1.2573e+00,
2.1467e+00, 9.5296e-01],
[ 2.9204e-02, 1.6258e-01, 3.8098e-01, ..., 4.3437e-01,
2.5856e+00, -1.2773e-01],
[ 7.3765e-01, 3.5724e+00, 2.2479e+00, ..., 3.6089e+00,
2.6568e+00, -3.2182e-01],
...,
[ 1.9252e-01, 6.7743e-01, 6.3003e-01, ..., 3.1060e-01,
3.1333e-01, -1.3333e+00],
[ 4.8879e-01, 5.5178e-01, 6.2161e-01, ..., -1.3507e+00,
4.0043e-01, -1.6624e+00],
[ 4.6825e-01, 9.3254e-01, 1.5814e+00, ..., -2.6237e-01,
-3.3274e-01, -1.8489e+00]],
...,
[[-8.0999e-01, -1.5846e+00, -2.4042e+00, ..., -5.1469e+00,
-5.2773e+00, -5.9004e-01],
[-2.6199e+00, -1.9498e+00, -6.0287e-01, ..., -4.1623e+00,
-5.7027e+00, -2.4814e+00],
[-2.5613e+00, -1.7774e+00, -1.6145e-01, ..., -6.9677e+00,
-5.4390e+00, -2.7386e+00],
...,
[-1.8884e+00, -1.4908e+00, -1.3266e+00, ..., -4.1138e+00,
-5.2896e+00, -1.9529e+00],
[-3.0190e+00, -2.8941e+00, -2.1674e+00, ..., -3.7288e+00,
-5.0864e+00, -1.1179e+00],
[-1.3130e+00, -1.6857e+00, -2.0206e+00, ..., -1.9784e+00,
-2.1240e+00, 2.9678e-01]],
[[-1.6198e+00, -2.1413e+00, -3.9676e+00, ..., -1.8465e+00,
-7.9578e-01, -6.7850e-01],
[-3.5756e+00, -3.0103e+00, -3.7232e+00, ..., -3.4432e+00,
-2.9671e+00, -8.7126e-01],
[-4.4644e+00, -4.6663e+00, -2.5063e+00, ..., -1.6924e+00,
-8.4091e-01, 4.8916e-03],
...,
[-4.7680e+00, -3.9146e+00, -4.1158e+00, ..., -2.9553e+00,
-1.0473e+00, -6.2951e-01],
[-4.1476e+00, -1.9964e+00, -2.6711e+00, ..., -1.0534e+00,
-3.7946e-01, 1.1380e+00],
[-3.0425e+00, -2.9668e+00, -2.3270e+00, ..., -9.7789e-02,
-6.4993e-01, 6.9274e-01]],
[[ 1.2420e+00, 1.6844e-01, 1.0686e+00, ..., 2.6491e-01,
6.6942e-01, -1.3938e+00],
[ 1.8519e+00, 1.0453e+00, 1.0651e+00, ..., 4.0994e+00,
2.9042e+00, 2.2388e+00],
[ 2.3362e-02, -2.3875e-01, 7.9789e-01, ..., 2.6133e+00,
1.9327e+00, 5.0973e-01],
...,
[ 2.3621e+00, 1.5821e+00, 1.2357e+00, ..., 3.4934e+00,
2.3203e+00, 3.0655e-01],
[ 2.6913e+00, 3.7333e+00, 2.1616e+00, ..., 2.7183e+00,
1.2736e+00, 1.8386e+00],
[ 1.8864e+00, 1.5034e+00, 3.6685e-01, ..., 5.7075e-01,
-1.4548e-01, 8.5757e-01]]],
[[[-1.6641e-01, 1.2164e+00, 1.2014e+00, ..., -8.6791e-01,
-3.4602e+00, -1.0540e+00],
[ 5.1695e-02, 7.1900e-03, 2.5869e+00, ..., 7.1611e-01,
-2.2556e+00, -8.3676e-01],
[-1.3697e+00, -6.1304e-01, 1.6505e+00, ..., -6.2017e-01,
-4.5027e+00, -1.5619e+00],
...,
[-2.9965e-01, 2.1173e-01, 8.0216e-01, ..., -3.8677e+00,
5.1416e+00, 1.4461e+00],
[-1.5533e+00, 9.7109e-02, 5.9035e-01, ..., -2.5640e+00,
3.4773e+00, -2.2182e-01],
[-1.2339e+00, -2.2553e+00, -1.3230e+00, ..., -2.2305e+00,
1.2564e+00, 5.8441e-01]],
[[-1.5361e+00, -2.4677e+00, -3.3713e+00, ..., -5.2013e+00,
-8.8795e+00, -1.8530e+00],
[-4.9508e+00, -5.0852e+00, -3.7257e+00, ..., -5.6675e+00,
-8.2533e+00, -2.8552e+00],
[-4.1507e+00, -4.4634e+00, -4.5089e+00, ..., -6.7758e+00,
-8.4738e+00, -1.9281e+00],
...,
[-4.6144e+00, -5.6884e+00, -4.6517e+00, ..., -3.2705e+00,
4.3601e+00, -2.3955e+00],
[-7.7701e+00, -7.9850e+00, -6.7601e+00, ..., -6.5756e+00,
1.6401e+00, -2.0557e+00],
[-1.4338e+00, -3.2181e+00, -2.0540e+00, ..., -2.2882e+00,
8.9850e-01, 1.2076e+00]],
[[-2.2555e-01, 2.3253e+00, 1.4793e+00, ..., 2.1252e+00,
2.4645e+00, 6.9924e-01],
[-5.4867e-01, 1.8204e-01, 1.5040e-01, ..., 1.3088e+00,
2.0165e+00, -3.0407e-01],
[ 3.2100e-02, 1.7563e+00, 1.3510e+00, ..., 3.9345e-01,
1.2078e+00, -1.8324e-01],
...,
[-6.5847e-01, 6.7600e-01, 1.3170e-01, ..., -2.9061e+00,
-7.4542e+00, -3.6508e+00],
[-1.1598e+00, 2.8628e-01, 2.6588e-01, ..., -5.5896e+00,
-3.7144e+00, -1.5758e+00],
[-2.8524e-01, -2.5219e-01, -3.0883e-01, ..., -3.5166e+00,
-1.4608e+00, -1.3670e-01]],
...,
[[-1.2935e+00, -2.1000e+00, -5.8941e-01, ..., -3.3829e+00,
-5.5732e+00, -1.2921e+00],
[-3.6839e+00, -3.7687e+00, -1.5464e+00, ..., -2.4788e+00,
-5.1980e+00, -2.0359e+00],
[-2.8515e+00, -2.6796e+00, -1.2376e+00, ..., -3.1170e+00,
-6.0248e+00, -1.5916e+00],
...,
[-1.0015e+00, -2.1627e+00, -4.9950e-01, ..., 2.5166e+00,
1.3524e+00, -1.9111e+00],
[-2.1793e+00, -3.5642e+00, -3.0934e-01, ..., 1.8683e+00,
-2.0487e+00, -1.5961e+00],
[-1.3927e+00, -1.2614e+00, 1.7165e-01, ..., 2.1077e+00,
-5.3151e-01, 1.2659e+00]],
[[-1.8096e+00, -1.7793e+00, -1.9436e+00, ..., -2.3603e+00,
-1.2652e+00, 2.9611e-01],
[-4.1381e+00, -2.0478e+00, -1.8918e+00, ..., -2.8322e+00,
-2.7459e+00, 2.3025e-01],
[-3.8976e+00, -2.8264e+00, -2.6391e+00, ..., -2.6227e+00,
-2.9343e+00, -1.5275e-01],
...,
[-2.9148e+00, -2.9919e+00, -3.5270e+00, ..., 5.4572e-01,
-6.8154e+00, -3.2085e+00],
[-2.0728e+00, 7.5301e-02, -1.5664e+00, ..., 3.1694e+00,
-4.5695e+00, -3.7074e+00],
[-2.1489e+00, -1.1214e+00, -6.6228e-01, ..., 2.5887e+00,
-1.0176e-01, -8.6828e-01]],
[[ 1.2439e+00, 3.9284e-01, 9.0054e-01, ..., 1.6892e+00,
1.3885e+00, -1.4661e-01],
[ 4.0587e+00, 2.9787e+00, 2.6704e+00, ..., 3.7443e+00,
2.6033e+00, 3.3865e+00],
[ 1.6463e+00, 4.0586e-01, 5.6026e-01, ..., 3.1721e+00,
1.0446e+00, 1.7020e+00],
...,
[ 3.0208e+00, 2.6450e+00, 2.5769e+00, ..., 2.8506e+00,
2.1814e+00, -3.9506e+00],
[ 3.8195e+00, 3.9836e+00, 2.7258e+00, ..., 4.6205e+00,
-1.1613e-01, -4.0474e+00],
[ 2.4434e+00, 2.2530e+00, 9.8734e-01, ..., 3.5781e+00,
-3.1356e-01, -9.9240e-01]]]], device='cuda:0'),
tensor([[[[-3.3399e-01, 1.3132e+00, 2.4265e+00, ..., -9.5983e-01,
-3.4195e+00, -9.1343e-01],
[ 1.9173e-01, 2.7612e-01, 2.9505e+00, ..., 2.8931e-01,
-1.6216e+00, -5.8652e-01],
[-1.1694e+00, -1.8891e-01, -7.0112e-01, ..., 8.1546e-01,
-7.2786e-01, -3.4433e-01],
...,
[-4.6453e-01, -2.9159e-01, 1.9183e+00, ..., -2.2734e+00,
-3.7341e+00, -1.8479e-01],
[-1.5818e+00, -3.4285e-01, 1.2704e+00, ..., -2.8763e+00,
-4.1062e+00, -6.8004e-01],
[-1.0927e+00, -2.3335e+00, -1.1141e+00, ..., -3.7488e+00,
-3.8133e+00, -1.5425e+00]],
[[-1.5098e+00, -2.2899e+00, -4.0566e+00, ..., -5.0732e+00,
-8.7239e+00, -1.7920e+00],
[-4.8540e+00, -5.8931e+00, -5.6254e+00, ..., -5.2806e+00,
-7.8154e+00, -3.6687e+00],
[-3.9936e+00, -5.3690e+00, -1.1661e+01, ..., -5.1723e+00,
-6.2253e+00, -2.5676e+00],
...,
[-4.6460e+00, -5.4175e+00, -5.4349e+00, ..., -6.2516e+00,
-6.7879e+00, -9.8622e-01],
[-7.6644e+00, -7.9381e+00, -7.0049e+00, ..., -9.4015e+00,
-1.1083e+01, -4.3214e+00],
[-1.3616e+00, -3.0546e+00, -2.2071e+00, ..., -3.9442e+00,
-4.1447e+00, -4.8463e-01]],
[[-3.1467e-01, 2.3340e+00, 1.6272e+00, ..., 2.0447e+00,
2.5226e+00, 7.0550e-01],
[-6.0098e-01, 4.8779e-01, -8.6679e-01, ..., 1.5090e+00,
2.3695e+00, -1.6471e-01],
[-1.4354e-01, 1.3727e+00, 7.5867e-01, ..., 2.0172e-01,
8.0282e-01, -3.5992e-01],
...,
[-7.3500e-01, 5.5265e-01, 1.1024e+00, ..., -7.9248e-01,
1.0192e+00, -3.9975e-01],
[-1.1157e+00, 3.3545e-01, 1.2762e+00, ..., -2.8168e-01,
-3.8197e-01, -1.8489e+00],
[-4.4134e-01, -4.0154e-01, 1.4941e-01, ..., 4.9367e-01,
4.5698e-01, -1.7759e+00]],
...,
[[-1.2377e+00, -1.5990e+00, -7.8457e-02, ..., -3.3804e+00,
-5.5513e+00, -1.1681e+00],
[-3.2765e+00, -3.3088e+00, -1.0360e+00, ..., -2.6186e+00,
-5.8733e+00, -2.2131e+00],
[-2.3972e+00, -3.4643e+00, -2.8756e+00, ..., -2.5880e+00,
-6.1197e+00, -1.5286e+00],
...,
[-1.1154e+00, -2.5370e+00, -2.0534e-01, ..., -1.7196e+00,
-3.2623e+00, -7.9478e-01],
[-2.1344e+00, -3.8909e+00, -7.3001e-01, ..., -2.4400e+00,
-4.7722e+00, -7.5332e-01],
[-1.1759e+00, -1.2493e+00, 3.4590e-01, ..., -1.5688e+00,
-2.2476e+00, 1.1021e+00]],
[[-1.6951e+00, -1.4227e+00, -2.1125e+00, ..., -2.1995e+00,
-1.0405e+00, 2.6267e-01],
[-3.6083e+00, -2.3831e+00, -2.9588e+00, ..., -2.1497e+00,
-3.1422e+00, -2.7656e-01],
[-3.5351e+00, -3.3019e+00, -5.4742e+00, ..., -1.2650e+00,
-2.1164e+00, 4.2617e-01],
...,
[-2.8032e+00, -2.3206e+00, -3.2098e+00, ..., -1.9338e+00,
-2.1019e+00, -2.8640e-01],
[-2.0762e+00, 9.9057e-03, -1.0352e+00, ..., -1.7749e-01,
-6.6830e-01, 1.4396e+00],
[-2.1180e+00, -1.2819e+00, -4.8389e-01, ..., -6.5392e-01,
-9.2429e-01, 6.8193e-01]],
[[ 1.1838e+00, 4.9757e-01, 9.3401e-01, ..., 1.6673e+00,
1.4199e+00, 1.5375e-02],
[ 4.1741e+00, 2.8872e+00, 3.2809e+00, ..., 3.2109e+00,
2.5891e+00, 3.6276e+00],
[ 1.6847e+00, 1.2159e+00, 3.5905e+00, ..., 3.1781e+00,
2.1660e+00, 1.4242e+00],
...,
[ 3.0705e+00, 2.5496e+00, 2.7968e+00, ..., 3.4313e+00,
1.5825e+00, 1.2124e+00],
[ 3.7441e+00, 3.8551e+00, 3.1621e+00, ..., 2.3557e+00,
8.4211e-01, 2.6643e+00],
[ 2.3689e+00, 2.2201e+00, 1.5411e+00, ..., 4.2863e-01,
-1.7031e-01, 1.2685e+00]]],
[[[ 1.4636e-01, 8.7256e-01, -3.7658e-01, ..., 7.1138e-01,
-3.1537e-01, -7.7424e-01],
[ 7.9116e-01, 6.3985e-01, 1.8395e-01, ..., 1.1451e+00,
2.7615e-01, -1.2043e+00],
[ 1.4331e+00, 8.6257e-02, 2.9271e-01, ..., -2.2469e+00,
-9.8630e-01, -2.2858e+00],
...,
[-2.2765e-01, -5.7674e-01, 1.7272e-01, ..., -3.6680e+00,
-2.6601e+00, -1.8115e-02],
[-2.2260e+00, -1.5285e+00, 4.7068e-01, ..., -3.1993e+00,
-2.6132e+00, 2.4326e-01],
[-2.3981e+00, -4.0289e+00, -3.0855e+00, ..., -4.5457e+00,
-3.4938e+00, -1.4053e+00]],
[[-5.5038e-01, -1.4119e+00, -2.7312e+00, ..., -5.2351e-01,
-4.0252e+00, -2.3795e+00],
[-1.8625e+00, -3.8144e+00, -4.5841e+00, ..., -3.5630e+00,
-6.2792e+00, -4.5695e+00],
[-1.8360e+00, -4.2878e+00, -4.6563e+00, ..., -4.6951e+00,
-4.6071e+00, -3.5955e+00],
...,
[-6.6091e+00, -8.5182e+00, -7.2119e+00, ..., -1.1130e+01,
-9.1349e+00, -1.5551e+00],
[-9.7998e+00, -1.0372e+01, -8.9605e+00, ..., -1.0788e+01,
-1.0286e+01, -3.6681e+00],
[-2.5490e+00, -4.9481e+00, -4.2709e+00, ..., -4.1209e+00,
-3.8352e+00, -7.0649e-01]],
[[ 6.5369e-01, 1.2358e+00, 1.0983e+00, ..., 1.8687e+00,
2.1107e+00, 1.0628e+00],
[ 7.2491e-01, 1.6676e+00, 1.8974e+00, ..., 1.0156e+00,
2.1859e+00, 8.4042e-01],
[ 6.3494e-01, 3.1435e+00, 1.4658e+00, ..., 1.0713e+00,
2.0742e+00, 1.7770e+00],
...,
[-8.0804e-02, -2.9607e-01, -6.8435e-01, ..., 2.6686e+00,
-1.0323e-01, -5.0687e-01],
[-6.1340e-01, 6.3460e-01, -1.3254e+00, ..., -1.3944e+00,
-7.8512e-01, -3.7892e-01],
[ 6.4002e-01, 1.3553e+00, 1.0724e+00, ..., -1.6776e+00,
-7.4819e-01, -1.6555e+00]],
...,
[[-7.5327e-01, -1.3362e+00, -2.0938e+00, ..., -2.2735e+00,
-3.1391e+00, -1.0435e+00],
[-2.0619e+00, -3.1427e+00, -2.6220e+00, ..., -3.9459e+00,
-4.5139e+00, -3.7288e+00],
[-6.5175e-01, -2.0457e+00, -1.1535e+00, ..., -3.3888e+00,
-3.9123e+00, -3.4885e+00],
...,
[-2.5319e+00, -3.8911e+00, -2.2575e+00, ..., -7.8517e+00,
-4.9886e+00, -3.0902e+00],
[-3.6806e+00, -5.9220e+00, -2.7120e+00, ..., -4.6142e+00,
-4.8739e+00, -1.6798e+00],
[-3.2876e+00, -3.7720e+00, -3.3605e+00, ..., -2.9241e+00,
-2.8822e+00, 4.0898e-01]],
[[-1.4661e+00, -2.6426e+00, -3.7066e+00, ..., -3.1911e+00,
-3.3554e+00, -5.1307e-01],
[-4.2140e+00, -4.8285e+00, -4.7536e+00, ..., -5.1766e+00,
-5.8540e+00, -1.7365e+00],
[-4.0638e+00, -5.6717e+00, -2.8331e+00, ..., -4.6420e+00,
-4.7037e+00, -1.9335e+00],
...,
[-4.0878e+00, -3.1896e+00, -3.1115e+00, ..., -6.7046e+00,
-1.5884e+00, -5.7772e-01],
[-3.5465e+00, -9.9795e-01, -1.1813e+00, ..., -2.4337e+00,
-4.5683e-01, 7.1480e-01],
[-3.7737e+00, -3.0679e+00, -2.9163e+00, ..., -1.2425e+00,
-8.7506e-01, 5.5809e-01]],
[[ 8.3106e-01, -2.7783e-02, 7.4946e-01, ..., 1.1589e+00,
1.7660e+00, 5.4752e-01],
[ 1.5267e+00, 7.2021e-01, 7.4117e-01, ..., 3.2854e-01,
1.6504e+00, 3.1481e+00],
[ 1.3758e+00, 4.4476e-01, 1.4831e+00, ..., 1.2253e-01,
1.3332e+00, 1.0786e+00],
...,
[ 3.9546e+00, 3.6126e+00, 2.8957e+00, ..., 3.0933e+00,
2.9243e+00, 3.7879e-01],
[ 4.5607e+00, 4.3745e+00, 3.1199e+00, ..., 2.1347e+00,
2.4254e+00, 1.4266e+00],
[ 2.8373e+00, 2.8013e+00, 2.1706e+00, ..., 1.2973e+00,
5.6134e-01, 8.7488e-01]]],
[[[-3.0699e+00, -5.7735e+00, -5.0396e+00, ..., -3.0200e+00,
-1.6454e+00, -4.8488e-01],
[ 6.3654e-01, 4.1866e+00, 2.6713e+00, ..., -3.9315e+00,
-3.4585e+00, -1.3738e-01],
[ 2.8347e-01, 8.8503e-01, -9.3499e-01, ..., -3.3797e+00,
-4.6103e+00, -2.2013e-01],
...,
[-3.2016e+00, -2.2419e-01, 2.9855e+00, ..., -1.4770e+00,
2.6162e+00, 3.0338e+00],
[ 9.2747e-01, 4.5265e+00, 9.5517e-01, ..., 3.0567e+00,
2.6500e+00, -3.4383e-01],
[ 2.6848e-01, -4.1187e+00, -5.5777e+00, ..., -5.0326e+00,
-5.3793e+00, -2.8813e+00]],
[[-9.1846e+00, -8.9257e+00, -3.9480e+00, ..., -6.9982e+00,
-6.8285e+00, -1.0058e+00],
[-1.4348e+00, 1.7118e+00, 4.3342e-01, ..., -1.1223e+01,
-9.4566e+00, -2.4512e+00],
[ 1.2143e-01, -8.1142e-01, -2.4675e+00, ..., -1.0589e+01,
-9.7235e+00, -2.7585e+00],
...,
[-2.4024e+00, -2.6728e-01, -4.3391e-02, ..., 1.9679e-02,
-1.8618e-01, -7.7889e-01],
[ 2.0035e-01, -1.2310e+00, -7.4569e+00, ..., -1.8935e+00,
-8.0236e+00, -7.4765e+00],
[-1.9655e+00, -4.7502e+00, -4.8815e+00, ..., -6.9657e+00,
-6.5411e+00, -2.2076e+00]],
[[ 1.9565e+00, 4.4174e+00, 5.9262e+00, ..., 4.5857e-01,
1.9647e+00, 8.2886e-01],
[-1.8316e+00, 1.1656e-01, 4.6900e+00, ..., 1.1056e+00,
2.7729e+00, -1.8124e-01],
[-5.7828e+00, -9.4199e+00, -5.6865e+00, ..., 7.8730e-01,
2.7468e+00, -2.1683e-01],
...,
[-4.8116e+00, -4.1407e+00, -1.3485e+00, ..., 2.2299e+00,
1.7271e+00, 8.5341e-01],
[-3.1898e+00, -4.0848e+00, -2.3291e+00, ..., -1.8577e+00,
-9.3015e-01, 4.6701e-01],
[-1.2871e+00, -1.2373e+00, -1.7221e+00, ..., -1.9324e+00,
-1.4656e+00, -1.3206e+00]],
...,
[[-3.8847e+00, -6.5459e+00, -7.1411e+00, ..., -4.8789e+00,
-4.8513e+00, -3.9317e-01],
[ 1.7069e+00, 2.0557e+00, -3.9699e-01, ..., -6.2295e+00,
-6.6819e+00, -1.9528e+00],
[ 2.1466e+00, -2.5568e-01, -1.6591e+00, ..., -3.3678e+00,
-7.7210e+00, -1.5555e+00],
...,
[-1.9402e+00, -1.3187e+00, -2.8339e+00, ..., -2.3678e+00,
-5.3363e+00, -3.9773e+00],
[ 4.0978e-02, -2.9453e+00, -4.1166e+00, ..., -1.1734e+00,
-3.8530e+00, -2.8809e+00],
[-3.2082e+00, -6.8992e+00, -2.5510e+00, ..., -7.1069e+00,
-4.9934e+00, -8.4234e-01]],
[[-4.6353e+00, -3.9855e+00, -2.1151e+00, ..., -2.1161e+00,
-3.5535e-01, -9.2492e-01],
[-1.0428e+00, -1.4132e+00, -1.0802e+00, ..., -4.1246e+00,
-2.1858e+00, -7.8412e-01],
[-7.3835e-01, -1.3988e+00, 1.6918e+00, ..., -1.8762e+00,
-2.3032e+00, -1.1321e+00],
...,
[-3.4772e+00, -2.7930e+00, -1.1698e+00, ..., -7.1959e+00,
-4.1510e+00, 1.8125e-02],
[-4.5941e-01, 2.8105e-01, -1.9066e+00, ..., -2.8618e+00,
-3.9246e+00, -3.1466e-01],
[-1.7466e+00, -1.9694e+00, 9.1987e-02, ..., -3.2492e+00,
-1.7840e+00, 1.0084e+00]],
[[ 9.2857e-01, 2.9699e+00, 2.3822e+00, ..., -6.1192e-01,
5.2601e-01, -1.5697e+00],
[-1.2623e+00, 1.9354e+00, 3.5290e+00, ..., 2.9681e+00,
2.3889e+00, 1.7312e+00],
[-1.3104e+00, -9.7290e-01, -2.3018e+00, ..., 2.2764e+00,
1.2052e+00, 1.0993e+00],
...,
[-3.1198e+00, -7.1047e-01, 1.4339e+00, ..., -7.6612e+00,
-2.3209e+00, 2.1034e+00],
[ 5.2166e-01, 5.6191e+00, 5.2644e+00, ..., -4.2779e-01,
1.9553e+00, 2.7744e+00],
[ 3.7811e+00, 2.6780e+00, -1.4308e-01, ..., 1.2153e+00,
-3.2911e-01, 1.2691e+00]]],
[[[ 1.4586e-01, 8.3845e-01, -1.0913e+00, ..., 3.6118e-01,
-7.7414e-02, -7.3430e-01],
[ 8.4380e-01, 9.8518e-01, 5.2183e-01, ..., 1.8913e+00,
5.3933e-01, -1.2449e+00],
[ 1.7694e+00, 3.9886e-02, 3.7788e-01, ..., -7.3902e-01,
-8.2471e-01, -2.2163e+00],
...,
[-1.4971e+00, -8.9007e-01, 2.9593e-02, ..., -3.8998e+00,
-3.5250e+00, 9.8099e-01],
[-2.1370e+00, -1.4526e+00, -3.1852e+00, ..., -2.1285e+00,
-1.4661e+00, 1.0748e+00],
[-2.0152e+00, -3.9794e+00, -3.6502e+00, ..., -1.5890e+00,
-2.8171e+00, -1.3966e+00]],
[[-6.2895e-01, -1.5311e+00, -2.3045e+00, ..., -6.8777e-01,
-3.9300e+00, -2.5365e+00],
[-1.8606e+00, -3.7842e+00, -4.7181e+00, ..., -2.5011e+00,
-5.1230e+00, -4.9906e+00],
[-1.8882e+00, -4.1834e+00, -4.6467e+00, ..., -5.7812e+00,
-5.0859e+00, -4.0453e+00],
...,
[-7.2360e+00, -7.1610e+00, -7.1109e+00, ..., -5.3752e+00,
-6.3752e+00, -5.4641e-01],
[-9.2789e+00, -9.1600e+00, -6.9735e+00, ..., -6.6352e+00,
-8.3877e+00, -3.4469e+00],
[-2.4263e+00, -3.5793e+00, -3.4288e+00, ..., -1.7749e+00,
-3.8430e+00, -9.4939e-01]],
[[ 7.1949e-01, 1.4635e+00, 1.5509e+00, ..., 2.1093e+00,
2.4368e+00, 8.7724e-01],
[ 8.1726e-01, 1.7589e+00, 1.9432e+00, ..., 3.6753e-01,
2.5585e+00, 1.0076e+00],
[ 4.7904e-01, 2.6451e+00, 1.0128e+00, ..., 1.0965e+00,
2.1841e+00, 1.9226e+00],
...,
[ 5.7508e-01, 4.8785e-01, -5.4742e-01, ..., -1.1298e+00,
-1.1900e+00, -1.9084e+00],
[ 5.0013e-01, -2.8658e-01, 3.2353e-01, ..., -1.6925e+00,
-1.4605e+00, -1.4951e+00],
[ 7.5813e-01, -1.5309e-01, 3.2298e-01, ..., -3.9462e-01,
-5.7149e-01, -1.7843e+00]],
...,
[[-7.8892e-01, -1.3812e+00, -2.1899e+00, ..., -1.8836e+00,
-3.1685e+00, -1.0091e+00],
[-2.0753e+00, -2.8166e+00, -2.7945e+00, ..., -3.0181e+00,
-4.6614e+00, -3.4233e+00],
[-5.0106e-01, -1.9468e+00, -1.1226e+00, ..., -2.9863e+00,
-3.9746e+00, -2.8872e+00],
...,
[-2.1851e+00, -1.3455e+00, -3.8641e-01, ..., -1.1968e+00,
-2.8388e+00, -1.3336e+00],
[-2.8225e+00, -3.4768e+00, -1.3121e+00, ..., -1.2928e+00,
-3.1786e+00, -7.6206e-01],
[-2.2865e+00, -2.3231e+00, -1.1051e+00, ..., -6.7196e-01,
-2.0249e+00, 4.1802e-01]],
[[-1.5188e+00, -2.8943e+00, -4.1606e+00, ..., -2.9843e+00,
-3.2936e+00, -6.2482e-01],
[-4.1280e+00, -4.9906e+00, -5.9382e+00, ..., -3.8077e+00,
-6.0350e+00, -1.8507e+00],
[-4.0295e+00, -4.9222e+00, -3.3515e+00, ..., -3.0932e+00,
-4.6697e+00, -1.8741e+00],
...,
[-3.4545e+00, -1.9086e+00, -2.4475e+00, ..., -4.3451e-01,
-5.7168e-01, -2.7745e-02],
[-3.4312e+00, 5.3042e-01, 8.0157e-01, ..., 3.6772e-01,
6.8416e-01, 1.6455e+00],
[-3.7026e+00, -1.3011e+00, 6.7321e-02, ..., -9.1506e-01,
-5.1229e-01, 6.0724e-01]],
[[ 8.2668e-01, -1.0100e-01, 8.2741e-01, ..., 9.9613e-01,
1.6539e+00, 6.1083e-01],
[ 1.5179e+00, 8.4275e-01, 8.2161e-01, ..., 1.0970e+00,
1.5066e+00, 3.3505e+00],
[ 1.5416e+00, 2.8353e-01, 5.2123e-01, ..., 1.4763e+00,
1.4824e+00, 1.2042e+00],
...,
[ 3.3133e+00, 2.4539e+00, 1.4806e+00, ..., 1.9193e+00,
2.1647e+00, 1.1835e+00],
[ 4.6296e+00, 4.9906e+00, 1.6052e+00, ..., 2.7207e+00,
1.6701e+00, 1.9320e+00],
[ 2.4190e+00, 1.8578e+00, -1.1365e-01, ..., 2.2511e-01,
-1.7540e-01, 1.0040e+00]]],
[[[ 3.3222e-01, 1.1997e+00, 2.3768e-01, ..., -8.6516e-01,
-1.9703e+00, -2.8082e-01],
[ 4.7152e-01, 8.9258e-01, 1.5492e+00, ..., -5.4009e-01,
-1.9625e-01, 3.0394e-01],
[-4.5036e-01, -6.7200e-01, -3.3440e-01, ..., -5.7539e-01,
-1.0087e+00, -8.7090e-01],
...,
[-1.8144e+00, -5.1523e+00, -3.7253e-01, ..., -4.0343e+00,
-2.1009e+00, 1.0276e-01],
[-1.6072e+00, -1.8828e+00, -4.6481e-01, ..., -3.6758e+00,
-2.9547e+00, -4.5703e-01],
[-7.8507e-01, -2.6875e+00, -2.8892e+00, ..., -3.8740e+00,
-3.2894e+00, -1.1730e+00]],
[[-1.0594e+00, -1.5697e+00, -2.5635e+00, ..., -4.4174e+00,
-6.1017e+00, -1.2434e+00],
[-3.8947e+00, -4.1683e+00, -2.9849e+00, ..., -8.5087e+00,
-6.7509e+00, -3.2186e+00],
[-4.0121e+00, -4.3537e+00, -3.1863e+00, ..., -7.8479e+00,
-7.1588e+00, -2.9130e+00],
...,
[-6.8441e+00, -8.6240e+00, -3.5255e+00, ..., -7.9723e+00,
-6.0005e+00, -1.5547e+00],
[-7.0840e+00, -7.4330e+00, -9.4376e+00, ..., -9.6504e+00,
-9.1168e+00, -3.6728e+00],
[-2.1048e+00, -3.1361e+00, -2.3398e+00, ..., -4.4495e+00,
-3.4942e+00, -3.9130e-01]],
[[-5.5401e-03, 1.2615e+00, 1.5742e+00, ..., 5.5631e-01,
2.4500e+00, 1.2448e+00],
[ 4.7410e-02, 5.3926e-02, 3.1532e-01, ..., -4.9546e-01,
1.7069e+00, -6.3150e-02],
[ 6.3918e-01, 1.2549e+00, 1.0620e+00, ..., -1.5499e+00,
7.4821e-01, 1.3386e-01],
...,
[ 4.9130e-01, 2.2654e-01, -2.6420e+00, ..., -3.3485e-01,
2.8623e-01, -1.4481e-01],
[ 2.5508e-01, 4.8042e-01, 5.0036e-02, ..., -1.0282e+00,
-4.9177e-01, -9.6688e-01],
[ 4.8543e-01, 2.6177e-01, 5.9199e-01, ..., -5.5871e-01,
-7.4224e-02, -1.7797e+00]],
...,
[[-6.0589e-01, -1.5065e+00, -1.8065e+00, ..., -3.0643e+00,
-4.5829e+00, -1.2479e+00],
[-2.7009e+00, -2.5817e+00, -1.1352e+00, ..., -4.3794e+00,
-5.6381e+00, -2.0657e+00],
[-2.1724e+00, -1.4424e+00, -9.7305e-01, ..., -3.0947e+00,
-6.0174e+00, -2.0299e+00],
...,
[-3.4440e+00, -2.8744e+00, -7.5361e-01, ..., -3.1003e+00,
-5.1284e+00, -1.4091e+00],
[-2.9569e+00, -3.8685e+00, -4.5265e+00, ..., -2.8034e+00,
-4.6299e+00, -4.8918e-01],
[-1.4328e+00, -1.4021e+00, -6.6022e-01, ..., -1.3959e+00,
-2.0161e+00, 1.1193e+00]],
[[-1.3330e+00, -2.2777e+00, -3.7727e+00, ..., -2.7543e+00,
-1.5871e+00, -2.0890e-01],
[-3.6699e+00, -3.3570e+00, -4.0150e+00, ..., -1.9107e+00,
-2.4956e+00, -5.2730e-01],
[-4.5866e+00, -3.7280e+00, -3.1827e+00, ..., -9.6918e-01,
-2.7866e+00, -2.6323e-02],
...,
[-5.5374e+00, -1.9154e+00, 1.2849e+00, ..., -3.0112e+00,
-3.6686e+00, -3.1678e-01],
[-4.0542e+00, -2.0463e+00, -1.2088e+00, ..., -4.9024e-01,
-1.4032e+00, 4.8780e-01],
[-3.2029e+00, -2.4978e+00, -8.6243e-01, ..., 2.9154e-01,
-5.0769e-01, 5.4926e-01]],
[[ 1.2016e+00, -9.1951e-02, 9.5078e-01, ..., -1.0347e+00,
2.0819e-01, -1.2278e-01],
[ 2.2902e+00, 1.1408e+00, 1.1167e+00, ..., 1.7904e+00,
2.2400e+00, 3.4175e+00],
[ 1.4196e+00, 6.3638e-01, 1.4828e-01, ..., 1.7465e+00,
1.4616e+00, 7.4127e-01],
...,
[ 2.5504e+00, 2.0283e+00, 1.3554e+00, ..., 3.9311e-01,
-1.6801e-01, 8.4113e-01],
[ 2.9365e+00, 3.7898e+00, 4.5123e+00, ..., 4.1193e-01,
4.6509e-01, 1.6275e+00],
[ 2.3410e+00, 1.7511e+00, 2.7130e-01, ..., -3.8538e-01,
-4.0583e-01, 7.5475e-01]]]], device='cuda:0'),
tensor([[[[ 2.8514e-01, 1.4016e+00, 3.7091e-01, ..., -6.1549e-01,
8.2486e-02, 6.7973e-01],
[ 5.0991e-01, 3.6130e-01, 9.4798e-01, ..., -1.4205e+00,
-1.1327e+00, 9.2730e-01],
[ 2.3988e-01, 7.1861e-01, 1.7394e+00, ..., -1.5742e+00,
-1.2437e+00, 1.1492e-01],
...,
[ 3.9275e-01, -5.5922e-01, -3.6457e-01, ..., -1.0966e+00,
-1.1499e+00, -1.1116e+00],
[-1.3049e+00, -1.6995e+00, -3.9130e+00, ..., -2.8208e+00,
-2.8410e+00, -1.2935e+00],
[-3.0387e-02, -2.4588e+00, -3.7633e+00, ..., -2.4752e+00,
-2.3007e+00, -9.7836e-01]],
[[-7.9252e-01, -2.2501e+00, -4.0332e+00, ..., -3.2794e+00,
-4.9474e+00, -4.1885e-01],
[-4.7051e+00, -5.4802e+00, -5.8605e+00, ..., -5.9582e+00,
-7.3969e+00, -1.8321e+00],
[-3.5843e+00, -5.4363e+00, -7.9311e+00, ..., -9.0196e+00,
-9.5745e+00, -3.5183e+00],
...,
[-4.2817e+00, -4.6616e+00, -5.5221e+00, ..., -5.1292e+00,
-5.3296e+00, -2.2747e+00],
[-7.8307e+00, -8.3219e+00, -9.5195e+00, ..., -8.2023e+00,
-9.0867e+00, -4.2599e+00],
[-1.0673e+00, -2.5418e+00, -3.6060e+00, ..., -3.3238e+00,
-2.7613e+00, -3.0822e-01]],
[[-2.7999e-01, 2.1488e+00, 1.1534e+00, ..., 7.7002e-01,
2.1443e+00, 9.7095e-01],
[-1.8219e-02, 8.7532e-01, 9.6788e-01, ..., 4.2306e-01,
1.0500e+00, -5.1760e-01],
[-1.7649e-02, 1.9913e+00, 1.9651e+00, ..., -2.3377e-01,
1.4586e+00, -1.1492e-01],
...,
[ 1.7264e-01, 7.6765e-01, 1.4264e+00, ..., 1.0738e+00,
6.7645e-01, 7.9311e-01],
[ 2.0018e-01, 4.5458e-01, 1.4282e+00, ..., 1.3650e+00,
2.1761e+00, 8.7646e-01],
[-6.4874e-01, 2.7862e-01, 1.3624e+00, ..., 8.5517e-01,
8.6238e-01, -7.8978e-01]],
...,
[[-8.1710e-01, -1.8293e+00, -1.7654e+00, ..., -2.4203e+00,
-2.8699e+00, -4.8581e-01],
[-3.1086e+00, -4.1291e+00, -2.4606e+00, ..., -4.5302e+00,
-4.2461e+00, -2.4773e+00],
[-1.4709e+00, -2.5022e+00, -1.7926e+00, ..., -3.2781e+00,
-5.4352e+00, -1.6874e+00],
...,
[-6.7229e-01, -1.5541e+00, -1.5543e+00, ..., -2.3102e+00,
-4.2001e+00, -2.1767e+00],
[-2.7456e+00, -4.0531e+00, -3.2247e+00, ..., -3.3293e+00,
-4.9028e+00, -1.6610e+00],
[-7.4530e-01, -1.8260e+00, -1.2655e+00, ..., -6.3228e-01,
-1.2098e+00, 1.2256e+00]],
[[-1.7133e+00, -2.2003e+00, -2.9740e+00, ..., -1.5891e+00,
-1.5264e+00, -1.4346e+00],
[-4.3330e+00, -3.1614e+00, -2.7458e+00, ..., -3.8886e+00,
-2.2925e+00, -1.0777e+00],
[-2.7054e+00, -1.6935e+00, -3.3121e+00, ..., -2.2028e-02,
-1.2549e+00, -9.7133e-01],
...,
[-3.5449e+00, -3.8712e+00, -4.0472e+00, ..., -2.1036e+00,
-2.2568e+00, -9.0514e-01],
[-4.3442e+00, -2.7308e+00, -1.4629e+00, ..., -7.7757e-01,
-1.9005e+00, -3.7978e-01],
[-2.9711e+00, -2.4259e+00, -7.7017e-01, ..., -6.3683e-01,
-7.6796e-01, 8.0515e-02]],
[[ 1.0671e+00, 5.7342e-01, 5.6795e-01, ..., -6.4859e-01,
1.4586e-01, -4.1845e-01],
[ 3.5837e+00, 3.4347e+00, 2.2641e+00, ..., 1.6424e+00,
2.0252e+00, 2.4815e+00],
[ 1.7328e+00, 1.3834e+00, 1.5622e+00, ..., 1.7312e+00,
1.9447e+00, 1.6896e+00],
...,
[ 2.5686e+00, 1.3686e+00, 1.4784e+00, ..., 1.4388e+00,
1.6889e+00, 6.4389e-01],
[ 3.3735e+00, 4.3234e+00, 2.3286e+00, ..., 1.9687e+00,
7.6723e-01, 1.5181e+00],
[ 2.3336e+00, 1.5846e+00, 5.7644e-02, ..., 1.8258e-01,
1.7607e-01, 9.9463e-01]]],
[[[ 3.5408e-02, 3.4340e-01, -2.7214e+00, ..., -6.3795e-01,
-1.6130e-01, 9.2473e-01],
[-1.2632e-01, -1.7371e+00, -3.1642e+00, ..., -1.2395e+00,
-1.3459e+00, 1.2519e+00],
[-1.1029e+00, -7.5294e-01, 3.1674e-01, ..., -1.7899e+00,
-1.3176e+00, 7.8398e-01],
...,
[-1.0387e+00, -2.0054e+00, -7.2087e-02, ..., -1.7500e+00,
-4.1797e-01, -9.7475e-01],
[-1.0030e+00, -8.6819e-01, -2.2720e-01, ..., 3.5168e-02,
-4.2602e-01, -1.8071e+00],
[-2.6628e+00, -5.1203e+00, -4.5086e+00, ..., -4.8287e+00,
-4.1388e+00, -3.8362e+00]],
[[-1.8027e+00, -2.8084e+00, -5.8532e+00, ..., -3.6212e+00,
-4.2002e+00, -1.0708e-01],
[-6.1578e+00, -6.0882e+00, -5.7188e+00, ..., -6.2159e+00,
-6.6685e+00, -1.3568e+00],
[-4.7643e+00, -4.6917e+00, -3.9993e+00, ..., -3.4394e+00,
-5.4089e+00, -1.2844e+00],
...,
[-5.7335e+00, -5.6552e+00, -3.5643e+00, ..., -7.8568e+00,
-5.7715e+00, -3.7565e+00],
[-6.9777e+00, -7.7428e+00, -6.0967e+00, ..., -7.3765e+00,
-7.9915e+00, -6.5883e+00],
[-2.4762e+00, -4.4220e+00, -4.0471e+00, ..., -4.1937e+00,
-4.2068e+00, -2.6236e+00]],
[[-2.5760e-01, 1.3842e+00, 8.0309e-01, ..., 1.1216e+00,
1.9202e+00, 9.0825e-01],
[ 1.9915e-01, 8.8099e-01, -7.6883e-01, ..., 5.4603e-01,
7.2346e-01, -6.0745e-01],
[ 5.6355e-01, 1.8217e+00, 1.8335e+00, ..., 1.0014e+00,
1.6525e+00, 2.2301e-01],
...,
[ 1.0421e+00, 1.1629e+00, 8.2218e-01, ..., 1.9706e+00,
8.3082e-01, 2.3106e-01],
[ 7.6478e-01, 7.3709e-01, 6.2134e-01, ..., 1.7046e-01,
-2.1957e-01, 3.3658e-01],
[ 1.3142e+00, 1.5722e+00, 1.2584e+00, ..., 1.0357e+00,
1.2596e+00, 3.3103e-01]],
...,
[[-1.3678e+00, -2.7457e+00, -2.5942e+00, ..., -2.1198e+00,
-2.5094e+00, -3.5695e-01],
[-4.1943e+00, -5.0630e+00, -3.6355e+00, ..., -2.9523e+00,
-4.3242e+00, -2.6167e+00],
[-2.4692e+00, -2.1392e+00, -2.4773e+00, ..., -2.6315e+00,
-4.3922e+00, -1.6502e+00],
...,
[-2.7160e+00, -2.4199e+00, -6.7250e-01, ..., -4.7562e+00,
-3.7102e+00, -2.1727e+00],
[-2.8969e+00, -2.9721e+00, -1.8651e+00, ..., -1.8144e+00,
-4.3688e+00, -2.5743e+00],
[-3.4675e+00, -3.8495e+00, -3.3587e+00, ..., -3.3266e+00,
-4.0273e+00, -1.6396e+00]],
[[-1.8811e+00, -1.8172e+00, -1.0385e+00, ..., -1.4872e+00,
-1.1286e+00, -1.1468e+00],
[-5.1670e+00, -3.0658e+00, -1.1058e+00, ..., -1.9354e+00,
-9.0348e-01, -5.2425e-01],
[-4.3064e+00, -2.9291e+00, -3.6088e+00, ..., 4.2601e-02,
-8.4876e-01, -1.9519e-01],
...,
[-4.3191e+00, -2.6075e+00, -3.0041e+00, ..., -3.6977e+00,
-2.7122e+00, -1.3632e+00],
[-3.6996e+00, -8.9681e-01, -2.2607e+00, ..., -1.5317e+00,
-3.1006e+00, -1.1213e+00],
[-4.3974e+00, -3.0683e+00, -2.8522e+00, ..., -1.8047e+00,
-3.1096e+00, -1.1231e+00]],
[[ 1.6705e+00, 6.3384e-01, 7.7341e-02, ..., -5.6666e-01,
-1.8966e-01, -3.1728e-01],
[ 4.2775e+00, 2.6498e+00, 2.2437e+00, ..., 1.3114e+00,
1.7165e+00, 2.2078e+00],
[ 1.8137e+00, 3.2538e-01, 8.4612e-02, ..., 7.2659e-01,
1.2742e+00, 1.3068e+00],
...,
[ 2.8716e+00, 1.1484e+00, 1.1982e+00, ..., 2.2860e+00,
2.5449e+00, 1.2701e+00],
[ 3.6696e+00, 3.5588e+00, 2.3435e+00, ..., 2.8937e+00,
2.8590e+00, 2.1102e+00],
[ 1.9132e+00, 7.5819e-01, 1.8524e-01, ..., 8.3096e-02,
-3.1044e-01, -2.3798e-01]]],
[[[ 9.0278e-01, 1.7236e+00, 8.4264e-01, ..., -3.8828e+00,
-2.7262e+00, -1.0885e+00],
[ 1.5217e+00, 3.7983e-01, 4.2783e-01, ..., -2.9787e+00,
-2.5954e+00, -2.4283e-01],
[-1.0163e-01, 3.2218e-01, 2.3299e-01, ..., -2.3963e+00,
-2.9932e+00, -5.9413e-02],
...,
[ 1.2518e+00, 1.3226e+00, 4.0265e-01, ..., -1.4982e+00,
-5.6700e+00, -1.9690e-01],
[ 5.6524e-01, -5.7228e-01, -9.9621e-01, ..., -1.8412e+00,
-3.9576e+00, 6.7287e-01],
[ 1.2977e-01, -3.6031e+00, -3.8968e+00, ..., -2.9258e+00,
-2.9047e+00, -1.2017e+00]],
[[ 4.2416e-01, -1.1497e+00, -2.6584e+00, ..., -7.6602e+00,
-7.5538e+00, -1.3878e+00],
[-1.3656e+00, -3.3307e+00, -3.8436e+00, ..., -8.1106e+00,
-8.5198e+00, -2.4658e+00],
[-3.6627e-01, -2.3165e+00, -3.7073e+00, ..., -7.0109e+00,
-7.9443e+00, -2.5455e+00],
...,
[-5.7995e-01, -4.4183e+00, -6.2889e+00, ..., -4.0105e+00,
-1.0570e+01, -1.6085e+00],
[-4.1405e+00, -6.9324e+00, -4.5539e+00, ..., -7.4156e+00,
-1.0451e+01, -3.7628e+00],
[-2.4134e-01, -3.2734e+00, -3.8535e+00, ..., -4.3615e+00,
-3.5501e+00, -7.1388e-01]],
[[ 4.9970e-01, 2.1396e+00, 1.8227e+00, ..., 8.1592e-01,
2.2069e+00, 9.9269e-01],
[-7.3502e-03, 9.3713e-01, 5.4708e-01, ..., 1.1501e+00,
2.7636e+00, 7.3179e-02],
[ 1.2620e+00, 3.5933e+00, 2.4868e+00, ..., -8.8205e-01,
1.9748e+00, -2.8737e-01],
...,
[-1.1203e+00, -1.1171e+00, 8.0055e-01, ..., 3.8100e+00,
2.4742e+00, -1.3799e+00],
[-1.2370e+00, 5.2803e-01, 7.8201e-01, ..., 1.2600e+00,
1.6652e+00, -1.4711e+00],
[-5.2287e-01, 2.3276e+00, 2.6163e-01, ..., 5.8682e-01,
-2.4523e-01, -1.5698e+00]],
...,
[[-4.0214e-01, -8.6983e-01, -1.5335e+00, ..., -5.5507e+00,
-5.5905e+00, -8.2263e-01],
[-3.1493e+00, -4.4119e+00, -2.5663e+00, ..., -4.4591e+00,
-5.4469e+00, -1.8705e+00],
[-1.8715e+00, -2.5408e+00, -1.2189e+00, ..., -5.0452e+00,
-6.0661e+00, -1.3994e+00],
...,
[ 1.1573e-01, -2.1584e+00, -2.5908e+00, ..., -7.7268e+00,
-4.8160e+00, -1.9130e+00],
[-1.4931e+00, -3.6746e+00, -8.8545e-01, ..., -5.2980e+00,
-3.6844e+00, -1.3110e+00],
[-4.0558e-01, -3.0149e+00, -2.0521e+00, ..., -2.6045e+00,
-1.9486e+00, -4.2647e-02]],
[[-1.0268e+00, -4.9116e-01, -1.0253e+00, ..., -2.2788e+00,
-5.4840e-01, -6.0961e-01],
[-3.4887e+00, -2.4772e+00, -1.6397e+00, ..., -3.8961e+00,
-1.9309e+00, -4.9194e-01],
[-2.2924e+00, -2.4114e+00, -2.1122e+00, ..., -2.2064e+00,
-1.5831e+00, -4.7367e-01],
...,
[-1.4777e+00, -1.7760e+00, -4.0364e+00, ..., -6.6309e+00,
-2.0842e+00, -1.8406e+00],
[-2.6413e+00, -1.5605e+00, -6.4953e-01, ..., -4.1273e+00,
-1.4059e+00, 6.8032e-01],
[-3.1120e+00, -3.4261e+00, -1.2958e+00, ..., -6.7759e-01,
-7.5183e-01, 7.6486e-01]],
[[ 1.0425e+00, 8.5199e-01, 1.1773e+00, ..., -3.5577e-01,
5.8628e-01, -1.3459e+00],
[ 1.7904e+00, 1.4823e+00, 1.5907e+00, ..., 3.2168e+00,
3.0868e+00, 2.3828e+00],
[ 3.7697e-01, 3.1257e-01, 1.3685e+00, ..., 2.4258e+00,
1.6069e+00, 1.2477e+00],
...,
[ 9.7284e-01, 2.9477e+00, 2.7771e+00, ..., 2.3614e+00,
4.5584e-01, 6.2463e-01],
[ 2.4594e+00, 2.8112e+00, 1.7900e+00, ..., 1.2379e+00,
1.0428e+00, 2.1398e+00],
[ 2.2325e+00, 6.4481e-01, 1.1603e+00, ..., 3.7619e-01,
-6.0287e-01, 5.5589e-01]]],
[[[ 1.2268e+00, 5.2600e-01, -1.4284e+00, ..., -5.7271e+00,
1.6627e+00, 2.7330e+00],
[-7.6304e-01, -3.0565e+00, -9.1976e-01, ..., -1.0270e+01,
6.9872e+00, 4.3356e+00],
[-1.5392e+00, -2.6218e+00, 5.3770e-01, ..., -9.0212e+00,
7.2991e+00, 2.4938e+00],
...,
[-3.3959e+00, -5.4973e+00, -7.7909e+00, ..., -6.2813e-01,
2.3237e-01, 1.5087e+00],
[-5.1959e+00, -1.3663e+01, -1.2033e+01, ..., 1.3812e-01,
1.9667e+00, 2.4225e+00],
[-3.4268e+00, -1.0063e+01, -8.6671e+00, ..., -3.6179e-01,
-1.2459e+00, -2.1517e+00]],
[[-2.1844e+00, -4.9227e+00, -6.4597e+00, ..., -5.8447e+00,
4.0821e+00, 4.9745e-01],
[-6.3136e+00, -6.8643e+00, -3.2523e+00, ..., -7.8888e+00,
3.6590e+00, -3.0181e-02],
[-4.2558e+00, -4.0835e+00, -2.5873e+00, ..., -1.1153e+01,
2.4585e+00, 2.7144e+00],
...,
[ 9.9928e-01, 1.2130e+00, -5.4232e+00, ..., -4.6201e-01,
-1.4744e+00, 7.3967e-01],
[-4.1364e-01, -5.0031e+00, -9.9260e+00, ..., -2.3749e+00,
-2.8426e+00, -3.2779e+00],
[ 8.7351e-01, -1.8238e+00, -8.4468e-01, ..., -1.2786e+00,
-2.1444e+00, -2.4004e+00]],
[[-1.0270e-01, 2.3273e+00, 1.6301e+00, ..., -2.1712e+00,
4.2161e-01, 1.9047e+00],
[-1.9647e-01, 1.4257e+00, 1.5511e+00, ..., -6.7855e+00,
-8.8088e+00, 2.8176e+00],
[ 1.6371e+00, 2.3379e+00, 1.8147e+00, ..., -9.6066e+00,
-1.1541e+01, 1.5384e+00],
...,
[ 1.2486e+00, -5.8507e+00, -7.4703e+00, ..., 8.5167e-01,
-4.2657e-01, 4.8119e-01],
[ 2.1511e+00, -7.3648e+00, -8.9291e+00, ..., 5.9854e-01,
-1.1951e+00, -1.4359e+00],
[-1.9170e+00, -5.5054e+00, -5.0897e+00, ..., 4.7582e-01,
7.8369e-01, 5.4793e-01]],
...,
[[-1.1418e+00, -3.6024e+00, -3.3434e+00, ..., 9.3590e-01,
3.4677e+00, -1.5818e+00],
[-4.5058e+00, -4.8065e+00, -6.9253e-01, ..., 1.1400e+00,
5.6799e+00, -4.8844e+00],
[-3.2651e+00, -2.2335e+00, -8.6853e-01, ..., 1.9174e+00,
6.2564e+00, -5.0503e+00],
...,
[ 1.1418e+00, 1.7333e+00, -9.5286e-01, ..., -3.4024e+00,
1.1786e+00, -4.5075e-01],
[-1.4433e+00, -4.8347e+00, -3.3542e+00, ..., -2.2685e+00,
-7.1358e-01, -1.0875e+00],
[-2.3779e+00, -4.6079e+00, -3.1293e+00, ..., -7.2567e-01,
-1.8592e+00, -1.9397e+00]],
[[-2.2674e+00, -2.9968e+00, -2.3217e+00, ..., -3.3236e+00,
-4.1551e+00, 4.6336e-01],
[-5.5158e+00, -2.3231e+00, -1.2427e+00, ..., -1.1638e+00,
-7.2502e-01, -2.3800e+00],
[-5.2108e+00, -2.3297e+00, -2.1009e+00, ..., 1.4016e+00,
4.7943e+00, -2.1786e+00],
...,
[-7.0755e+00, -2.1677e+01, -9.6615e+00, ..., -1.4442e+00,
-8.5177e-01, -1.0468e+00],
[-1.4920e+01, -2.3523e+01, -1.3861e+01, ..., -4.6453e-01,
-7.4695e-01, -3.7612e-01],
[-8.0459e+00, -1.3599e+01, -8.7820e+00, ..., -8.3136e-01,
-2.5660e+00, -9.6692e-01]],
[[ 2.2452e+00, 9.9172e-01, 1.2582e+00, ..., -4.7427e+00,
-1.0930e+00, 4.1209e+00],
[ 3.5246e+00, 1.5565e+00, 1.9189e+00, ..., -1.0112e+01,
9.2400e-01, 5.6507e+00],
[ 1.5234e+00, 2.1852e-01, 1.0536e+00, ..., -9.3562e+00,
4.5649e+00, 4.6739e+00],
...,
[-1.2381e+00, -1.4155e+01, -1.9577e+01, ..., 8.5878e-02,
-1.0312e+00, -3.4573e-01],
[-1.0892e+00, -1.3117e+01, -1.7615e+01, ..., 5.6423e-01,
7.5253e-02, 1.9241e+00],
[ 1.1469e-01, -4.7573e+00, -1.0576e+01, ..., 5.8188e-01,
1.8280e-01, -2.3653e-01]]],
[[[ 3.1695e-01, 1.3089e+00, 3.0202e-01, ..., -9.6640e-01,
-2.1262e+00, -3.9673e-01],
[ 4.3996e-01, 1.0592e+00, 1.5177e+00, ..., -1.0071e+00,
-9.4529e-01, 3.7681e-01],
[-4.6253e-01, -3.3516e-01, -3.2404e-01, ..., -1.2340e+00,
-1.1392e+00, -3.8232e-01],
...,
[-2.1794e-01, -6.4620e-01, -5.2158e+00, ..., -4.9435e+00,
3.3892e-01, 7.2686e-01],
[-1.4620e+00, -9.3280e-01, -6.0324e+00, ..., -3.9175e+00,
-1.9104e+00, 1.3363e+00],
[-9.8252e-01, -3.1307e+00, -5.3860e+00, ..., -4.2474e+00,
-3.1160e+00, -9.6064e-01]],
[[-1.0241e+00, -1.5133e+00, -2.7080e+00, ..., -4.9358e+00,
-6.7334e+00, -1.4259e+00],
[-3.8013e+00, -4.0669e+00, -3.1675e+00, ..., -6.5945e+00,
-6.2718e+00, -2.7096e+00],
[-3.7910e+00, -3.9932e+00, -3.1502e+00, ..., -6.5118e+00,
-5.6859e+00, -2.1780e+00],
...,
[-6.8957e+00, -5.9335e+00, -8.3985e+00, ..., -9.5733e+00,
-2.7718e+00, 1.6518e+00],
[-8.2688e+00, -7.4785e+00, -1.0455e+01, ..., -9.6362e+00,
-8.7047e+00, -4.3194e+00],
[-2.0944e+00, -2.8316e+00, -4.3650e+00, ..., -4.4362e+00,
-2.5896e+00, -1.2707e-01]],
[[-3.7753e-02, 1.2026e+00, 1.5187e+00, ..., 5.9049e-01,
2.5618e+00, 1.1856e+00],
[ 3.3853e-02, 3.4292e-02, 3.9163e-01, ..., -4.4615e-01,
2.1424e+00, 2.1289e-01],
[ 6.9646e-01, 1.2569e+00, 1.1863e+00, ..., -1.2067e+00,
8.5031e-01, 2.3505e-01],
...,
[ 3.1268e-01, -3.8686e-01, 1.3465e+00, ..., -2.1972e+00,
-4.8947e+00, -1.5750e-01],
[ 3.8842e-01, -9.8753e-01, -2.1559e-01, ..., -1.3373e+00,
-7.4598e-01, -3.7705e-01],
[ 5.6571e-01, 4.1336e-01, 6.5380e-01, ..., -9.4447e-01,
-9.7498e-02, -1.1168e+00]],
...,
[[-5.7607e-01, -1.4608e+00, -1.8884e+00, ..., -3.4532e+00,
-4.8880e+00, -1.1899e+00],
[-2.7124e+00, -2.6190e+00, -1.3341e+00, ..., -3.9362e+00,
-5.0844e+00, -1.9859e+00],
[-2.0944e+00, -1.2560e+00, -1.1706e+00, ..., -3.8509e+00,
-5.2989e+00, -1.4888e+00],
...,
[-1.7354e+00, -4.4649e-01, -2.1302e+00, ..., -2.7978e+00,
-2.4172e+00, -8.9887e-01],
[-3.0689e+00, -1.9322e+00, -3.8952e+00, ..., -3.0006e+00,
-5.1006e+00, -1.9569e+00],
[-1.4824e+00, -1.6673e+00, -3.1919e+00, ..., -1.4627e+00,
-1.7031e+00, 8.6945e-01]],
[[-1.2798e+00, -2.2521e+00, -3.7854e+00, ..., -2.7744e+00,
-1.5791e+00, -1.7025e-01],
[-3.6227e+00, -3.4962e+00, -4.1468e+00, ..., -3.1807e+00,
-3.5228e+00, -4.5670e-01],
[-4.4419e+00, -3.7547e+00, -3.4814e+00, ..., -2.6971e+00,
-3.6640e+00, -3.7884e-01],
...,
[-4.9053e+00, -5.0410e+00, -3.4551e+00, ..., -3.1675e+00,
-1.4953e+00, -1.8604e+00],
[-3.9673e+00, -3.0648e+00, -8.2612e-01, ..., 4.3663e-01,
2.8967e-01, 1.1436e+00],
[-3.1062e+00, -3.4796e+00, -2.0592e+00, ..., 7.7974e-01,
2.3352e-01, 8.4742e-01]],
[[ 1.1922e+00, -5.1456e-02, 1.0487e+00, ..., -8.4079e-01,
4.3130e-01, -1.6388e-01],
[ 2.1912e+00, 1.1367e+00, 1.1858e+00, ..., 1.0458e+00,
1.8217e+00, 3.4590e+00],
[ 1.3479e+00, 6.8553e-01, 2.6821e-01, ..., 3.2413e-01,
1.0858e+00, 1.3075e+00],
...,
[ 2.9948e+00, 2.3743e+00, -1.4194e-02, ..., 1.3132e+00,
9.4010e-01, -1.2302e+00],
[ 3.2228e+00, 4.5792e+00, 1.5795e+00, ..., -4.6743e-01,
1.1130e+00, 2.1525e+00],
[ 2.2469e+00, 1.6043e+00, -5.2483e-01, ..., -6.0444e-01,
-6.4597e-01, 7.3419e-01]]]], device='cuda:0'),
tensor([[[[ 2.6365e-01, 1.7074e+00, 9.8156e-01, ..., -1.5213e+00,
1.9695e+00, 5.5955e-01],
[ 4.1836e-01, 2.4386e-01, -2.9430e-01, ..., -2.5543e+00,
-2.8041e+00, -3.0703e+00],
[-3.3307e-01, -2.4167e-01, -1.7125e+00, ..., -3.1265e+00,
-3.7352e+00, -2.6047e+00],
...,
[-7.9078e+00, -8.5069e+00, -3.6942e+00, ..., 8.5689e-02,
3.9781e-01, -6.6545e-01],
[-4.3968e+00, -6.4566e+00, -4.0144e+00, ..., -1.2040e-01,
1.0243e-01, -1.5215e+00],
[-2.7455e+00, -7.4961e+00, -6.1686e-01, ..., -4.5731e+00,
-4.0674e+00, -3.7965e+00]],
[[-8.1811e-01, -2.3801e+00, -4.8701e+00, ..., -1.3517e-01,
6.4643e-01, 2.2684e-01],
[-4.7000e+00, -6.4754e+00, -7.3052e+00, ..., 2.2657e+00,
-3.2864e-01, -3.0658e+00],
[-3.5823e+00, -5.3379e+00, -5.9183e+00, ..., -5.2853e+00,
-5.5420e+00, -2.1923e+00],
...,
[-9.1685e+00, -5.6806e+00, 1.7314e+00, ..., -4.1476e+00,
-4.7751e+00, -3.6676e+00],
[-6.1935e+00, -9.6912e+00, 1.6366e+00, ..., -7.0318e+00,
-7.9373e+00, -6.4198e+00],
[-1.0208e+00, -7.1326e+00, -4.3583e+00, ..., -3.8036e+00,
-4.1415e+00, -2.5546e+00]],
[[-3.1252e-01, 2.2941e+00, 1.0563e+00, ..., 2.6190e+00,
3.6254e-01, 1.2467e+00],
[-1.8001e-01, 1.1978e+00, 1.0596e+00, ..., -9.9983e+00,
-7.8746e+00, -4.6325e+00],
[-1.9109e-02, 1.7850e+00, 1.5728e+00, ..., -7.6540e+00,
-5.1518e+00, -2.0321e+00],
...,
[ 3.3943e-02, -1.6440e+00, 3.6675e-01, ..., 1.2791e+00,
1.0322e+00, 1.4403e-01],
[-1.0979e+00, -3.2319e+00, -1.4646e+00, ..., 8.3376e-01,
2.9279e-01, 1.5320e-01],
[-2.0624e-01, 1.4688e+00, -4.3355e+00, ..., 1.4654e+00,
1.0625e+00, 7.5174e-02]],
...,
[[-8.2620e-01, -1.7788e+00, -1.4998e+00, ..., 1.5291e+00,
1.5949e+00, 1.6674e+00],
[-3.1090e+00, -4.3774e+00, -3.4955e+00, ..., 1.0552e+00,
-1.4133e+00, -1.5802e+00],
[-1.2656e+00, -2.8234e+00, -2.2045e+00, ..., -2.8781e+00,
-3.6809e+00, -2.9418e-01],
...,
[-4.2247e+00, -8.7594e-02, -4.6874e+00, ..., -1.6468e+00,
-3.2904e+00, -1.5904e+00],
[-2.7313e+00, -2.6620e+00, -1.8576e+00, ..., -1.7993e+00,
-4.0232e+00, -2.2061e+00],
[-2.8203e+00, -5.7060e+00, -1.8053e+00, ..., -2.9030e+00,
-3.6214e+00, -1.6403e+00]],
[[-1.7402e+00, -2.1420e+00, -2.4259e+00, ..., 4.8698e-01,
6.2398e-01, 1.1399e+00],
[-4.3396e+00, -3.7914e+00, -4.0958e+00, ..., 3.3778e+00,
1.8040e+00, 2.0542e+00],
[-3.5605e+00, -3.1669e+00, -2.4919e+00, ..., -2.2375e+00,
-1.0073e+00, 9.2747e-01],
...,
[-3.9967e+00, -1.1459e+00, -1.3620e+01, ..., -2.0044e+00,
-3.4042e+00, -1.7528e+00],
[-3.5822e+00, 3.8126e+00, -4.7938e+00, ..., -1.1770e+00,
-3.0200e+00, -1.3380e+00],
[-4.2373e+00, 6.3589e-01, 2.8903e+00, ..., -2.3503e+00,
-3.2735e+00, -1.2143e+00]],
[[ 1.0881e+00, 7.3106e-01, 1.1057e+00, ..., 5.2900e-01,
1.2656e+00, -5.9517e-01],
[ 3.5529e+00, 3.5226e+00, 2.8711e+00, ..., -4.7672e+00,
-4.5363e+00, -2.4707e+00],
[ 1.4281e+00, 1.1943e+00, 5.7605e-01, ..., -4.0423e+00,
-3.4829e+00, -2.5129e+00],
...,
[ 2.1086e+00, 2.9345e+00, -5.9380e+00, ..., 1.7796e+00,
1.8456e+00, 1.3107e+00],
[ 1.4911e+00, 4.8932e+00, -5.3707e+00, ..., 2.6691e+00,
2.2751e+00, 2.0814e+00],
[ 7.8470e-01, 8.8829e-01, 2.1042e+00, ..., -2.0900e-01,
-2.6663e-01, -1.8140e-01]]],
[[[-1.1679e+00, -1.8051e+00, -1.3633e+00, ..., 8.6889e-01,
-1.5316e+00, -2.3269e+00],
[-4.0915e-01, -1.6105e+00, 5.7935e-01, ..., -2.5064e+00,
-9.8305e+00, -2.4482e+00],
[-1.8818e+00, -1.0486e+00, -1.4939e-01, ..., -8.2324e+00,
-1.2518e+01, -1.8343e+00],
...,
[-9.9535e+00, -1.7392e+00, 6.3156e+00, ..., 1.1396e+00,
4.5941e-01, 2.0111e+00],
[-1.4765e+01, 1.0407e+00, 4.5086e+00, ..., -5.8510e-01,
-2.3382e+00, -2.0457e-02],
[-4.6571e+00, 3.2268e+00, 1.3964e+00, ..., -2.9710e+00,
-2.6605e+00, -2.0684e+00]],
[[-3.4936e+00, -3.8936e+00, -2.7283e+00, ..., 2.4458e-01,
-8.6295e-01, -1.3016e+00],
[-4.7127e+00, -4.3342e+00, -2.9554e+00, ..., -4.4369e+00,
-8.4649e+00, -7.1341e-02],
[-4.7145e+00, -2.7411e+00, -2.4679e+00, ..., -7.2010e+00,
-2.9423e+00, 1.7297e+00],
...,
[-1.7866e+01, -3.1228e+00, 5.8827e+00, ..., -4.1407e+00,
-3.6767e+00, -1.0172e+00],
[-1.7126e+01, 6.9511e-01, 5.9380e+00, ..., -7.8680e+00,
-9.8340e+00, -5.5842e+00],
[-3.2870e+00, 2.5489e+00, 1.3796e+00, ..., -3.7409e+00,
-2.9903e+00, -6.4687e-01]],
[[ 1.3366e+00, 1.7943e+00, 1.4669e+00, ..., -1.4796e+00,
-4.8441e+00, -3.1219e+00],
[ 1.5295e+00, 5.4170e-01, 9.9216e-01, ..., -4.4577e+00,
-8.9768e+00, -2.8861e+00],
[ 7.3929e-01, 7.5567e-01, 1.1979e+00, ..., -4.6810e+00,
-5.6457e+00, -1.8494e+00],
...,
[ 7.0231e-01, -7.4808e+00, -1.4130e+01, ..., -4.1950e+00,
-2.7363e+00, -8.8202e-01],
[ 1.9419e+00, -1.4043e+01, -1.5841e+01, ..., -1.7857e+00,
1.5016e-01, -8.5326e-01],
[ 6.4767e-01, -1.3628e+01, -7.8062e+00, ..., -1.8154e+00,
-4.4716e-01, -1.0791e+00]],
...,
[[-2.8286e+00, -2.5935e+00, -2.4635e+00, ..., 2.2622e+00,
1.1008e+00, 1.7584e-01],
[-3.3016e+00, -2.6101e+00, -1.5637e+00, ..., -1.7752e+00,
-7.0850e-01, 3.7716e-02],
[-2.8577e+00, -3.8653e-01, -3.7397e-01, ..., -4.3903e+00,
3.4175e-01, 8.1958e-01],
...,
[-2.9672e+00, 6.5407e+00, 1.0523e+00, ..., -2.9830e+00,
-4.6940e+00, -1.0891e+00],
[-5.4597e+00, 5.6820e+00, -3.4736e+00, ..., -4.8371e+00,
-6.7063e+00, -1.7449e+00],
[ 3.3155e-01, 6.1624e+00, -5.0090e+00, ..., -1.0992e+00,
-1.8222e+00, 8.4374e-01]],
[[-2.8970e+00, -9.6226e-01, -2.2572e+00, ..., -4.8121e+00,
-1.0290e+00, 9.0352e-01],
[-5.1350e+00, -1.2350e+00, -3.2347e+00, ..., -8.1510e+00,
-4.6457e+00, -1.4729e+00],
[-3.9043e+00, -1.0291e+00, -1.5265e+00, ..., -1.2110e+01,
-7.3976e+00, -1.6651e+00],
...,
[-3.1392e+00, -1.3571e+00, -1.4483e+01, ..., 1.3348e-01,
-1.3063e+00, 7.8289e-01],
[-4.9547e+00, 9.1605e-01, -1.5284e+01, ..., 7.1614e-01,
-1.3169e+00, 2.2525e+00],
[-1.1377e+00, -1.1141e-01, -8.0705e+00, ..., 4.7473e-01,
-1.4686e+00, 2.5972e-01]],
[[ 2.2091e+00, -5.4053e-01, 1.2101e-01, ..., -1.8910e+00,
-2.7509e+00, -2.2264e+00],
[ 3.5645e+00, 1.1486e+00, 8.7467e-01, ..., -4.8203e+00,
-3.9904e+00, -2.4374e+00],
[ 2.3623e+00, 8.7506e-01, -4.0519e-01, ..., -8.7859e+00,
-4.9526e+00, -1.0406e+00],
...,
[ 6.6454e+00, 5.6338e+00, 4.7139e+00, ..., 3.3816e+00,
2.0339e+00, 1.1284e+00],
[ 5.0493e+00, 8.4029e+00, -1.0488e+00, ..., 1.1656e+00,
3.8865e-01, 1.9100e+00],
[ 4.0455e+00, 8.5152e+00, 5.3044e-01, ..., 8.5161e-01,
1.9521e-01, 8.0799e-01]]],
[[[ 3.0653e-01, 1.2535e+00, 4.7309e-01, ..., 8.2806e-01,
-7.9061e-01, -9.0477e-01],
[ 6.1580e-01, 7.0512e-01, 2.7214e+00, ..., -4.6292e+00,
-5.3527e+00, -3.5694e+00],
[-1.4947e+00, -2.6572e+00, -2.7199e+00, ..., 3.4536e-01,
-3.1121e-01, 1.2476e+00],
...,
[-5.3511e-01, -7.2297e-01, -2.0943e+00, ..., 4.5633e+00,
-8.3938e-01, -1.0025e+00],
[-2.3991e+00, -1.2634e+00, -8.6852e-01, ..., -3.6197e+00,
-4.2287e+00, -1.0335e+00],
[-1.3859e+00, -3.8131e+00, -4.3223e+00, ..., -4.2581e+00,
-3.8091e+00, -1.7385e+00]],
[[-1.3404e+00, -3.0395e+00, -4.5228e+00, ..., -4.9382e+00,
-6.3661e+00, -2.0563e+00],
[-2.3059e+00, -3.6008e+00, -5.0308e+00, ..., -8.6868e+00,
-9.5954e+00, -2.8272e+00],
[-4.8806e+00, -6.6847e+00, -5.2037e+00, ..., -3.4994e+00,
-3.1668e+00, 7.9307e-01],
...,
[-8.4530e+00, -1.2237e+01, -8.7771e+00, ..., -4.3689e+00,
-6.1485e+00, -3.0550e+00],
[-7.1474e+00, -5.8781e+00, -5.6340e+00, ..., -1.1736e+01,
-9.8991e+00, -3.5276e+00],
[-2.2691e+00, -3.8810e+00, -3.6493e+00, ..., -4.0626e+00,
-3.3881e+00, -5.4814e-01]],
[[ 4.8670e-01, 1.4196e+00, 1.9291e+00, ..., 1.0984e-01,
2.6349e+00, 3.0406e-01],
[ 8.9319e-01, 7.6301e-01, 4.7909e-02, ..., 9.2946e-01,
4.9096e+00, 3.0626e+00],
[ 5.4806e-01, 4.3613e+00, 4.5333e+00, ..., 4.1151e-01,
1.5194e+00, 2.1441e+00],
...,
[-2.2307e-01, -1.0026e+00, 2.4501e-01, ..., -2.0890e+00,
3.5896e+00, 9.4573e-01],
[-1.8359e-01, 5.1124e-01, -6.1350e-01, ..., -1.0578e+00,
2.7978e-01, -1.5672e+00],
[ 7.3777e-01, 1.1901e+00, 9.8246e-01, ..., 6.5559e-01,
-5.4219e-01, -2.0755e+00]],
...,
[[-8.0673e-01, -1.4517e+00, -2.3062e+00, ..., -2.6961e+00,
-5.0178e+00, -1.4595e+00],
[-2.9010e+00, -2.0128e+00, -8.1086e-01, ..., -5.2403e+00,
-6.3172e+00, -2.9324e+00],
[-2.3786e+00, -2.2365e+00, -2.8106e+00, ..., -5.4002e-01,
-7.2184e-01, 9.7388e-01],
...,
[-5.1494e+00, -5.2067e+00, -3.4255e+00, ..., -1.4699e+00,
-6.7371e+00, -1.5218e+00],
[-2.6010e+00, 1.4032e-01, 3.0376e-01, ..., -3.2084e+00,
-4.4192e+00, -1.7857e-01],
[-1.8493e+00, -2.5450e+00, -2.2321e+00, ..., -3.8656e-01,
-1.8921e+00, 6.0602e-01]],
[[-1.8870e+00, -2.0087e+00, -3.8861e+00, ..., -2.3053e+00,
-1.6189e+00, 1.7564e-01],
[-3.8407e+00, -3.6631e+00, -2.9855e+00, ..., -6.4974e+00,
-6.9629e+00, -3.4124e+00],
[-4.0256e+00, -7.1513e+00, -3.5502e+00, ..., -4.5717e+00,
-4.3695e+00, -3.0746e+00],
...,
[-3.1621e+00, -3.4819e+00, -4.9778e+00, ..., -1.1394e+00,
-5.3432e+00, -1.6793e+00],
[-2.5405e+00, 9.2889e-01, -1.8147e+00, ..., -2.8810e-01,
-1.9866e+00, 2.2605e-01],
[-3.2573e+00, -2.8366e+00, -1.2233e+00, ..., 2.1820e-01,
-5.4001e-01, 3.5274e-01]],
[[ 1.3154e+00, 4.3378e-01, 1.5606e+00, ..., 1.2765e-01,
7.5840e-01, -6.5796e-01],
[ 1.3716e+00, 1.3134e+00, 2.3150e+00, ..., 3.8917e-01,
5.4835e-01, 2.5808e+00],
[ 1.4497e-01, 2.4694e-01, 1.7766e+00, ..., 2.4879e+00,
1.7638e+00, 2.3868e+00],
...,
[ 4.6963e+00, 4.9591e+00, 2.8747e+00, ..., 3.6812e+00,
4.8408e-01, -1.4566e-01],
[ 1.6617e+00, 4.2964e+00, 8.7694e-01, ..., 3.0847e+00,
9.2901e-01, 1.8419e+00],
[ 2.4007e+00, 2.5000e+00, 1.1675e+00, ..., 3.1462e-01,
8.3644e-01, 1.0973e+00]]],
[[[ 1.7342e-01, 7.0073e-01, 1.7747e+00, ..., 1.1157e-01,
-2.5419e+00, -1.5711e+00],
[-1.0426e+00, -1.8085e+00, 3.2482e+00, ..., 9.2514e+00,
-2.2118e+00, 2.1659e+00],
[-1.0391e+00, -3.9131e-01, 3.8933e+00, ..., 2.5215e+00,
-3.9968e+00, 2.0154e+00],
...,
[-3.2854e-01, -1.0338e-01, -3.4908e+00, ..., 2.8673e-01,
-3.3774e+00, 2.3538e-01],
[-1.8878e+00, -6.8274e-01, -1.6148e+00, ..., -1.8466e-01,
-3.7374e+00, -6.3679e-01],
[-8.6688e-01, -2.3861e+00, -2.8478e+00, ..., -2.5060e+00,
-2.5012e+00, -1.5754e+00]],
[[-1.9299e+00, -2.7883e+00, -3.8744e+00, ..., -4.4962e+00,
-5.2059e+00, -8.1792e-01],
[-4.7775e+00, -5.3004e+00, -2.7189e+00, ..., -2.8460e-01,
-2.6095e+00, 1.8136e+00],
[-2.6488e+00, -4.9248e+00, -2.2449e+00, ..., -1.0392e+01,
-9.8942e+00, 1.1450e+00],
...,
[-5.6970e+00, -5.4181e+00, -6.7874e+00, ..., -4.7260e-01,
-5.1579e+00, -2.1395e+00],
[-7.2966e+00, -7.1003e+00, -7.9484e+00, ..., -7.5492e+00,
-1.0298e+01, -4.4088e+00],
[-1.5148e+00, -2.9782e+00, -2.8133e+00, ..., -3.5568e+00,
-3.5469e+00, -7.0336e-01]],
[[-6.8501e-02, 1.1981e+00, 2.4857e+00, ..., 2.6742e+00,
2.9103e+00, 2.3803e+00],
[ 5.6990e-01, 9.4316e-01, 7.9554e-01, ..., -5.8818e+00,
-6.9032e+00, 5.6703e-02],
[ 9.0906e-01, 1.8445e+00, -5.7051e-01, ..., -5.3506e+00,
-8.3026e+00, -1.8172e+00],
...,
[-5.7999e-01, -1.9778e+00, 5.0552e-01, ..., -2.1938e+00,
1.5726e+00, -2.1671e+00],
[ 2.1160e-01, -1.4108e+00, 2.1818e-01, ..., 3.3156e-01,
1.0158e+00, -2.0533e+00],
[ 2.9555e-01, 4.5357e-02, 2.0834e-01, ..., -7.8248e-01,
-1.2624e+00, -1.7444e+00]],
...,
[[-1.3347e+00, -1.1481e+00, -1.3905e+00, ..., -6.7682e-01,
-2.9678e+00, -1.5322e+00],
[-4.4297e+00, -2.0663e+00, 4.3197e-01, ..., -2.2547e-01,
1.4421e+00, 5.0842e-01],
[-1.9364e+00, 8.1865e-01, 6.3762e-01, ..., -2.3915e+00,
2.2783e+00, -1.2158e-01],
...,
[-1.9117e+00, -1.9325e+00, -3.0206e+00, ..., -4.3187e+00,
-5.7686e+00, -7.5022e-01],
[-2.9845e+00, -2.0330e+00, -2.7155e+00, ..., -5.5771e+00,
-5.7711e+00, -1.2672e+00],
[-1.0996e+00, -9.6376e-01, -1.0650e+00, ..., -2.1305e+00,
-1.7496e+00, 4.3757e-01]],
[[-1.8849e+00, -3.5171e-01, -4.8039e+00, ..., -2.0095e+00,
-1.4991e+00, 9.8004e-01],
[-4.8928e+00, -1.4863e+00, -4.2475e+00, ..., 3.2223e+00,
9.4967e-01, 5.4072e-01],
[-3.2035e+00, -5.8330e-01, -3.9183e+00, ..., -1.9124e+00,
-2.2713e-01, 2.4958e-02],
...,
[-2.7828e+00, -3.7469e+00, -3.8678e+00, ..., 3.8643e-01,
-2.8101e+00, -1.2483e+00],
[-3.4318e+00, -1.3591e+00, -2.6343e+00, ..., -1.4393e+00,
-1.5455e+00, 9.3086e-01],
[-2.7605e+00, -2.5255e+00, -2.1945e+00, ..., 1.1212e-01,
-7.0827e-01, 7.3068e-01]],
[[ 1.8076e+00, -1.6897e-01, 8.8284e-01, ..., 4.0070e+00,
1.4823e-01, 1.4070e+00],
[ 3.1009e+00, 1.2888e-01, 1.0252e+00, ..., 3.5750e+00,
-2.1412e+00, 2.7438e+00],
[ 1.1981e+00, 7.2607e-01, 1.3895e+00, ..., -1.7496e+00,
-2.8026e+00, 1.5783e+00],
...,
[ 2.2948e+00, 1.8780e+00, 1.9850e+00, ..., 3.2050e+00,
1.2130e+00, 6.7777e-01],
[ 1.9375e+00, 3.4915e+00, 1.2864e+00, ..., 3.7866e+00,
1.3837e+00, 1.7415e+00],
[ 1.9359e+00, 1.6940e+00, -1.2264e-01, ..., 3.5300e-01,
2.3400e-01, 1.0214e+00]]],
[[[ 8.3310e-01, 5.3719e-01, -1.2714e-01, ..., 9.1395e-01,
-9.0140e-01, -9.5521e-01],
[ 2.6166e+00, 1.8374e+00, 1.2913e+00, ..., -1.9834e+00,
-5.6579e-02, -2.0851e+00],
[-9.8898e-01, -1.2855e+00, -5.3748e-01, ..., 1.7003e+00,
6.3135e-01, -2.4475e+00],
...,
[ 1.1916e+00, 2.6471e+00, 4.0652e+00, ..., -7.9543e-01,
-7.0213e-01, -5.3642e-01],
[-1.2117e+00, -2.9219e+00, -6.9306e-01, ..., 4.0862e-02,
-4.8287e-01, -1.4223e+00],
[-1.2639e+00, -4.1251e+00, -3.0885e+00, ..., -2.6227e+00,
-3.0542e+00, -2.8589e+00]],
[[-1.4400e+00, -4.0263e+00, -6.9298e+00, ..., -3.3301e+00,
-6.0259e+00, -2.5031e+00],
[-4.5034e+00, -8.0879e+00, -1.0355e+01, ..., -3.0590e+00,
-4.6765e+00, -5.1512e+00],
[-2.4522e+00, -2.1061e+00, -7.1442e+00, ..., 6.2611e-01,
-3.0855e+00, -5.5169e+00],
...,
[-3.2611e+00, -3.8097e+00, -6.6563e+00, ..., -3.8019e+00,
-4.1587e+00, -3.0236e+00],
[-8.6925e+00, -1.3255e+01, -9.9731e+00, ..., -4.5655e+00,
-5.5776e+00, -5.2156e+00],
[-1.1268e+00, -3.6288e+00, -3.9711e+00, ..., -3.0600e+00,
-2.9110e+00, -1.6698e+00]],
[[ 7.1409e-01, 3.0644e+00, 1.8623e+00, ..., 2.1091e+00,
1.3587e+00, 6.6740e-01],
[ 1.1607e-02, 1.9709e+00, 1.3892e+00, ..., 4.1333e+00,
1.9838e+00, 1.0344e+00],
[ 5.5251e-01, 2.5273e+00, 1.6325e+00, ..., 3.4713e+00,
3.6125e+00, 1.5566e+00],
...,
[-3.1765e+00, -3.9823e+00, -5.1298e+00, ..., 4.0379e-01,
6.8779e-01, 1.0187e+00],
[-7.3928e-01, -2.1114e+00, -1.5969e+00, ..., 1.0580e+00,
1.1856e+00, 4.9347e-01],
[-5.5839e-01, 1.1056e+00, 6.1934e-01, ..., 1.2281e+00,
1.8076e+00, 5.7363e-01]],
...,
[[-8.1458e-01, -3.2353e+00, -2.5036e+00, ..., -2.0361e+00,
-2.8565e+00, -1.3680e+00],
[-2.6120e+00, -4.1204e+00, -1.2616e+00, ..., -4.1974e+00,
-3.8602e+00, -3.6623e+00],
[-3.3838e+00, -3.1951e+00, -1.6267e+00, ..., -5.0514e-01,
-5.1696e+00, -1.9069e+00],
...,
[-3.6375e-01, -1.8640e+00, -4.3305e-01, ..., -1.6686e+00,
-3.3190e+00, -1.4374e+00],
[-3.6367e+00, -6.8923e+00, -2.1274e+00, ..., -1.2156e+00,
-3.4093e+00, -1.6823e+00],
[-1.3191e+00, -1.5483e+00, -8.7233e-01, ..., -2.0336e+00,
-2.5251e+00, -6.2949e-01]],
[[-2.5179e+00, -3.1861e+00, -1.3700e+00, ..., 2.6540e-01,
-9.2356e-01, -4.8945e-01],
[-3.4594e+00, -3.8801e+00, -2.4087e+00, ..., -2.9394e+00,
-2.9722e+00, -2.1346e+00],
[-1.8088e+00, -3.2123e+00, -9.2695e-01, ..., -6.4367e+00,
-5.3640e+00, -1.0911e+00],
...,
[ 4.7390e-01, 9.6181e-02, -2.0789e+00, ..., 8.1918e-01,
-2.4983e+00, -1.2631e+00],
[-2.2500e+00, 2.2862e-01, -9.2921e-01, ..., -2.5890e-01,
-3.2766e+00, -1.9758e+00],
[-2.0439e+00, -9.5792e-01, 5.6240e-01, ..., -7.0226e-01,
-2.5874e+00, -1.3101e+00]],
[[ 1.5854e+00, 1.2809e+00, 1.7713e+00, ..., 1.1588e+00,
1.3261e+00, 2.0951e-01],
[ 3.7876e+00, 4.8555e+00, 5.4849e+00, ..., 1.8410e+00,
2.3884e+00, 2.0867e+00],
[ 3.9159e-01, -6.6202e-01, 1.6203e+00, ..., 2.3644e+00,
4.5012e+00, 3.9922e-01],
...,
[ 3.8292e+00, 3.2995e+00, 6.3599e+00, ..., 1.6197e+00,
1.9080e-01, 2.3424e-01],
[ 4.8657e+00, 4.5189e+00, 3.6940e+00, ..., 2.0561e+00,
8.1259e-01, 1.2394e+00],
[ 1.2445e+00, -5.3568e-01, 1.2376e+00, ..., 6.9798e-01,
-2.1196e-01, -1.4588e-01]]]], device='cuda:0'),
tensor([[[[-4.7605e+00, -9.2380e-01, 1.5430e+00, ..., 1.1021e+00,
1.1165e+00, -2.6762e-01],
[-2.4913e+00, 5.1193e-01, -1.2464e+00, ..., -1.5956e+00,
1.3650e+00, -2.4405e-01],
[-9.5753e-01, -4.9795e-02, -4.3647e+00, ..., 3.3786e+00,
6.3395e-01, 1.1237e+00],
...,
[-9.4327e-01, -2.0397e+00, -6.5521e-01, ..., -2.1851e+00,
-2.8572e+00, -2.6513e+00],
[-1.0036e+00, -8.1586e-01, -4.9639e-01, ..., -1.7643e+00,
-1.1702e+00, -1.4903e+00],
[-2.6734e+00, -5.1858e+00, -4.5618e+00, ..., -5.7165e+00,
-5.4351e+00, -4.3534e+00]],
[[-2.5000e+00, 1.2618e+00, 1.3399e+00, ..., 1.4977e+00,
-2.0686e+00, -1.2310e+00],
[-3.1488e-02, 1.1422e+00, -4.0701e+00, ..., -2.5074e+00,
-2.1223e+00, -6.9889e-01],
[-1.6251e+00, -1.3381e+00, -8.9534e+00, ..., 2.3684e+00,
-7.6259e-02, 1.9750e-01],
...,
[-5.7329e+00, -5.4327e+00, -3.7489e+00, ..., -1.2026e+01,
-1.2184e+01, -7.4469e+00],
[-7.3328e+00, -8.1114e+00, -6.6246e+00, ..., -6.7247e+00,
-7.6120e+00, -6.3530e+00],
[-2.5262e+00, -4.6594e+00, -4.2028e+00, ..., -5.1189e+00,
-5.2229e+00, -2.9914e+00]],
[[ 2.0528e+00, -1.0249e+00, -1.8599e+00, ..., -1.1156e+00,
3.2851e+00, 1.2537e+00],
[ 4.7317e-01, -5.9788e+00, -8.0224e+00, ..., -4.5911e+00,
1.2361e+00, 1.6657e+00],
[-3.5195e+00, -6.4948e+00, -9.6956e+00, ..., -7.1163e-01,
-1.5245e+00, 1.1964e+00],
...,
[ 2.9956e-01, 9.9260e-01, 7.1684e-01, ..., -7.5374e-02,
-5.3905e-01, 4.6918e-01],
[ 8.1587e-01, 8.7656e-01, 5.8476e-01, ..., -6.0055e-01,
1.8715e-01, -1.3725e-01],
[ 1.2942e+00, 1.6234e+00, 1.1342e+00, ..., 7.2680e-01,
5.8107e-03, -7.1692e-01]],
...,
[[-1.2495e+00, 2.3697e+00, 1.2142e+00, ..., -1.0861e-01,
-1.9133e+00, -5.7983e-01],
[-8.9049e-01, -1.3980e+00, -4.1163e+00, ..., -2.3631e+00,
-5.4545e+00, -3.2735e+00],
[-5.0739e-01, -5.7071e+00, -6.5491e+00, ..., -2.2263e+00,
-4.0281e+00, -1.7709e+00],
...,
[-1.9348e+00, -1.7099e+00, 5.0277e-03, ..., -7.6097e+00,
-7.5694e+00, -4.8829e+00],
[-2.9180e+00, -3.0767e+00, -1.8700e+00, ..., -2.9311e+00,
-3.6030e+00, -2.5401e+00],
[-3.5377e+00, -4.0143e+00, -3.3719e+00, ..., -3.7953e+00,
-4.4493e+00, -2.1959e+00]],
[[-2.5833e+00, -1.7977e+00, -4.7074e-01, ..., -3.2431e-01,
-2.2106e+00, 1.7777e+00],
[-2.0657e+00, -5.9030e+00, -3.1648e+00, ..., -1.5208e-01,
-3.2238e+00, 1.0971e+00],
[-1.9389e+00, -8.7590e+00, -5.3859e+00, ..., 2.7907e+00,
-1.1266e+00, 4.4140e-01],
...,
[-3.1109e+00, -2.1041e+00, -2.6833e+00, ..., -6.1208e+00,
-6.1133e+00, -1.2908e+00],
[-3.6266e+00, -7.3561e-01, -2.0708e+00, ..., 8.3747e-01,
-1.5201e+00, -3.7315e-02],
[-4.5312e+00, -3.0580e+00, -2.6906e+00, ..., -1.1953e+00,
-1.6704e+00, 2.9425e-01]],
[[ 6.1614e-01, 2.1969e+00, -4.6590e-01, ..., -1.0585e+00,
2.1080e+00, 1.4349e+00],
[ 2.3836e+00, 1.4106e+00, -2.2318e+00, ..., -2.5056e+00,
1.1079e+00, 4.5905e+00],
[ 7.6929e-01, -1.7422e+00, -4.1639e+00, ..., 1.9268e+00,
8.2317e-01, 1.9123e+00],
...,
[ 2.2164e+00, 6.7835e-01, 3.2208e-01, ..., 6.1722e+00,
4.6179e+00, 3.5469e+00],
[ 3.9805e+00, 3.9933e+00, 2.4955e+00, ..., 1.6236e+00,
1.6569e+00, 2.9942e+00],
[ 2.0152e+00, 7.5161e-01, -4.2931e-02, ..., 2.3419e-01,
-1.0420e-01, 4.4997e-01]]],
[[[ 4.6407e-01, 1.3343e+00, -2.0352e+00, ..., -3.7005e-01,
6.3903e-01, 9.5121e-01],
[ 1.0543e+00, 2.5802e-01, -4.4759e+00, ..., 6.9828e-01,
6.3701e-01, 1.1839e+00],
[ 3.9604e-01, -1.1776e+00, -4.0827e+00, ..., 2.2570e+00,
1.3626e+00, 7.7200e-01],
...,
[ 5.0618e-01, -1.0890e+00, -1.2067e+00, ..., 2.9880e-01,
-1.0604e+00, -1.2216e+00],
[-1.1847e+00, -1.3172e+00, -2.4273e+00, ..., -1.2637e+00,
-2.9792e+00, -1.5284e+00],
[-2.0306e-01, -2.5337e+00, -2.4670e+00, ..., -1.6156e+00,
-2.5467e+00, -8.6376e-01]],
[[-1.5350e+00, -2.3414e+00, -6.8777e+00, ..., -1.1388e+00,
-3.5928e+00, -3.1252e-01],
[-6.4801e+00, -6.8939e+00, -9.2752e+00, ..., -7.2446e-01,
-5.4074e+00, -2.0388e+00],
[-4.9240e+00, -6.3254e+00, -7.2367e+00, ..., -1.9804e-01,
-5.7563e+00, -3.5464e+00],
...,
[-4.3311e+00, -5.5284e+00, -4.6298e+00, ..., -4.4527e+00,
-5.4463e+00, -2.4307e+00],
[-7.2486e+00, -8.5258e+00, -7.3654e+00, ..., -8.4115e+00,
-9.4083e+00, -4.1538e+00],
[-1.0722e+00, -3.1552e+00, -2.2242e+00, ..., -2.9942e+00,
-2.7667e+00, -2.1381e-01]],
[[-3.2588e-01, 1.7845e+00, 1.6392e+00, ..., 1.2245e+00,
2.6998e+00, 9.4021e-01],
[-9.5420e-02, 1.0697e-01, 3.2360e+00, ..., -5.4965e-01,
2.0873e+00, -4.4001e-01],
[ 3.5675e-01, 1.1338e+00, 3.7089e+00, ..., -7.4052e-01,
2.6684e+00, 1.0579e-01],
...,
[ 5.1164e-02, 1.7212e+00, 1.9300e+00, ..., 1.5219e+00,
1.2939e+00, 6.0009e-01],
[ 1.9920e-01, 1.2480e+00, 1.0443e+00, ..., 1.4696e+00,
2.2775e+00, 7.4203e-01],
[-5.2997e-01, 7.2815e-01, 2.4075e-01, ..., -9.5086e-02,
9.3640e-01, -7.8637e-01]],
...,
[[-8.0976e-01, -2.3422e+00, -2.3397e+00, ..., -1.1269e+00,
-2.3931e+00, -3.7519e-01],
[-3.6365e+00, -4.0076e+00, -3.6844e+00, ..., -1.4083e+00,
-4.7015e+00, -2.2131e+00],
[-2.0619e+00, -2.7203e+00, -1.0696e+00, ..., -1.4737e+00,
-5.0209e+00, -7.3702e-01],
...,
[-8.8528e-01, -2.0280e+00, -1.5687e+00, ..., -1.1653e+00,
-4.0564e+00, -1.8833e+00],
[-2.6553e+00, -3.4234e+00, -2.7710e+00, ..., -2.1890e+00,
-5.0651e+00, -1.4899e+00],
[-9.8448e-01, -1.6162e+00, -1.0142e+00, ..., -6.3015e-02,
-1.2795e+00, 1.4108e+00]],
[[-1.6899e+00, -2.7186e+00, -2.7508e+00, ..., -1.2398e+00,
-6.5622e-01, -1.0441e+00],
[-4.4042e+00, -3.6806e+00, -2.4613e+00, ..., -8.4921e-01,
-1.2058e+00, -3.9630e-01],
[-3.2204e+00, -1.6311e+00, -9.9360e-01, ..., 1.4829e+00,
-7.9707e-01, -2.3197e-01],
...,
[-3.7836e+00, -3.8988e+00, -4.0104e+00, ..., -2.4362e+00,
-2.9821e+00, -1.3578e+00],
[-4.2610e+00, -1.9940e+00, -2.1193e+00, ..., -1.6697e+00,
-2.2457e+00, -7.1171e-01],
[-3.0134e+00, -1.8614e+00, -9.6555e-01, ..., -4.4161e-01,
-7.3603e-01, 4.1386e-02]],
[[ 1.5353e+00, 6.1866e-01, 7.3860e-01, ..., 7.0481e-02,
1.1384e+00, -3.0477e-01],
[ 4.4566e+00, 3.3051e+00, 1.6157e+00, ..., 1.7638e+00,
3.6257e+00, 2.5105e+00],
[ 2.3887e+00, 1.8721e+00, -8.0906e-01, ..., 3.4413e+00,
3.3949e+00, 1.7107e+00],
...,
[ 2.6534e+00, 1.2908e+00, 1.3222e+00, ..., 1.6923e+00,
1.1772e+00, 4.1627e-01],
[ 3.3465e+00, 3.7943e+00, 2.7073e+00, ..., 1.5699e+00,
1.3809e+00, 1.6697e+00],
[ 2.2558e+00, 1.4362e+00, 8.3201e-01, ..., 1.3272e-01,
1.5377e-01, 1.0652e+00]]],
[[[ 2.2972e-01, 1.2809e+00, 3.2330e-02, ..., -2.9369e-01,
-7.6593e-01, -1.4511e+00],
[ 3.8331e-01, 8.8439e-01, 7.6932e-01, ..., 1.8990e+00,
2.9710e+00, -5.6384e-01],
[-3.2195e-01, -5.4762e-01, -4.9979e-01, ..., 2.9104e+00,
2.7846e+00, -5.3742e-02],
...,
[-3.0290e-01, -5.7765e-01, -8.6159e+00, ..., -5.0425e+00,
-3.7036e+00, 3.9840e-01],
[-1.5302e+00, -8.0464e-01, -8.8577e+00, ..., -4.7534e+00,
-3.5263e+00, 5.5347e-01],
[-8.4197e-01, -2.1983e+00, -5.7233e+00, ..., -3.5912e+00,
-3.1660e+00, -1.4164e+00]],
[[-1.0980e+00, -1.5910e+00, -3.3816e+00, ..., -1.1317e+00,
-5.5214e+00, -2.5443e+00],
[-3.8304e+00, -4.1197e+00, -3.3297e+00, ..., 9.4454e-01,
-3.7763e+00, -4.8248e+00],
[-3.8165e+00, -4.2168e+00, -3.3151e+00, ..., 2.2547e+00,
-2.7526e+00, -4.7461e+00],
...,
[-7.1362e+00, -5.4391e+00, -1.3396e+01, ..., -7.3432e+00,
-8.5535e+00, -2.1348e+00],
[-8.3532e+00, -6.5259e+00, -1.4435e+01, ..., -9.3402e+00,
-1.0960e+01, -4.3089e+00],
[-2.2670e+00, -2.0870e+00, -3.9549e+00, ..., -3.5791e+00,
-3.7980e+00, -7.8288e-01]],
[[ 1.2517e-02, 1.3004e+00, 1.6913e+00, ..., 2.0618e+00,
4.0695e+00, 6.3147e-01],
[ 6.4141e-02, 5.1596e-02, 7.5894e-01, ..., -3.2606e+00,
4.8316e+00, 4.8658e-01],
[ 5.6698e-01, 1.2365e+00, 1.1747e+00, ..., -5.4608e+00,
4.6065e+00, 1.7685e+00],
...,
[ 3.7399e-01, -2.2623e+00, 1.9651e+00, ..., -2.1350e+00,
1.5727e+00, -1.1128e+00],
[-1.6052e-01, -2.0732e+00, 9.2308e-01, ..., -7.4274e-01,
1.6980e+00, -1.4825e+00],
[ 3.9593e-01, -7.5278e-01, 9.2237e-01, ..., 1.3513e-01,
-2.3963e-01, -1.9280e+00]],
...,
[[-6.4104e-01, -1.5865e+00, -2.2941e+00, ..., -1.6469e+00,
-4.3719e+00, -6.2652e-01],
[-2.7595e+00, -2.7592e+00, -1.5475e+00, ..., 4.5058e-01,
-5.3734e+00, -1.8753e+00],
[-2.0746e+00, -1.5557e+00, -1.0018e+00, ..., 3.3185e+00,
-7.9857e+00, -1.3616e+00],
...,
[-1.2185e+00, -1.1022e+00, -4.4759e+00, ..., -5.2405e+00,
-6.0799e+00, -1.5706e+00],
[-2.5270e+00, -2.7398e+00, -6.3242e+00, ..., -4.3964e+00,
-5.5403e+00, -8.1843e-01],
[-9.2959e-01, -1.7496e+00, -3.8717e+00, ..., -2.6403e+00,
-2.2589e+00, 3.4631e-01]],
[[-1.3351e+00, -2.3824e+00, -4.0010e+00, ..., -3.3484e+00,
-2.2531e+00, 2.2101e-01],
[-3.6178e+00, -3.5518e+00, -4.7967e+00, ..., -1.1623e+00,
-3.8328e+00, 2.9286e-01],
[-4.4740e+00, -3.8001e+00, -3.9916e+00, ..., -9.6756e-01,
-5.6050e+00, -1.5352e+00],
...,
[-4.5455e+00, -6.5309e+00, -2.5056e+00, ..., -2.7770e+00,
-1.8716e+00, -9.6895e-01],
[-3.5066e+00, -5.0574e+00, -2.5044e+00, ..., -5.5526e-01,
-6.4600e-01, 1.1743e+00],
[-2.6694e+00, -4.0295e+00, -3.4837e+00, ..., 2.4754e-01,
-8.5417e-01, 6.5830e-01]],
[[ 1.2016e+00, -8.8902e-02, 1.0773e+00, ..., 1.0754e+00,
3.0720e+00, -5.5973e-01],
[ 2.2785e+00, 1.1958e+00, 1.1556e+00, ..., -7.6350e-02,
4.5387e+00, 3.8969e+00],
[ 1.4250e+00, 7.1488e-01, 9.0579e-02, ..., 9.4894e-01,
3.9376e+00, 1.5977e+00],
...,
[ 3.3413e+00, 2.7162e+00, 1.8549e+00, ..., 4.2093e+00,
2.2913e+00, 5.3972e-01],
[ 2.6606e+00, 4.5397e+00, 4.0783e+00, ..., 3.2108e+00,
7.2098e-01, 1.9168e+00],
[ 2.3149e+00, 1.5861e+00, 3.8642e-01, ..., 1.0867e+00,
-1.1196e-01, 9.6225e-01]]],
[[[ 4.4944e-01, 1.9014e+00, 1.6472e+00, ..., -4.8738e+00,
-6.7543e+00, -2.9692e+00],
[ 4.4302e-01, -3.4331e+00, -3.7687e+00, ..., 3.0873e+00,
1.0090e+00, 1.4520e+00],
[-3.6046e+00, -9.7680e+00, -7.4081e+00, ..., -4.9937e+00,
-3.7737e+00, 3.4190e+00],
...,
[ 7.9361e-01, -7.6388e+00, -1.7842e-01, ..., -4.8253e+00,
-1.7869e+00, -1.1275e+00],
[ 3.7731e-01, -2.4959e+00, -2.5174e+00, ..., -1.6848e+00,
-1.2875e+00, -1.3708e+00],
[-1.4322e+00, -4.3202e+00, -6.1966e+00, ..., -5.7577e+00,
-3.9941e+00, -3.4901e+00]],
[[-1.0960e+00, -2.4550e+00, -5.9639e+00, ..., -1.0075e+01,
-7.0196e+00, -1.3767e+00],
[-8.2951e+00, -7.9469e+00, -5.8884e+00, ..., -2.1423e+00,
-2.3271e-01, -1.0104e+00],
[-1.0677e+01, -1.2200e+01, -1.1960e+01, ..., -4.9588e+00,
-4.3500e+00, 1.0696e+00],
...,
[-3.7813e+00, -1.1365e+01, -9.1933e+00, ..., -1.0169e+01,
-5.8819e+00, -3.2915e+00],
[-5.8142e+00, -7.0888e+00, -1.3861e+01, ..., -8.3263e+00,
-7.9706e+00, -6.6067e+00],
[-2.8270e+00, -3.3594e+00, -2.7223e+00, ..., -4.1449e+00,
-4.2634e+00, -2.4981e+00]],
[[-5.9487e-03, 2.3171e+00, 1.6209e+00, ..., 5.1959e+00,
7.6026e+00, 2.4865e+00],
[-6.1730e-01, 1.0257e+00, 2.7316e+00, ..., -1.6111e+00,
6.8598e-01, 4.2585e+00],
[-9.0939e-02, 3.5224e+00, 4.7823e+00, ..., -1.6472e+01,
-2.2618e+01, -6.9729e-01],
...,
[-1.5095e+00, -4.6787e-01, -1.0939e+01, ..., 7.3544e-01,
4.0681e-01, 2.4702e-01],
[-5.9947e-01, -1.0976e-01, -2.4712e+00, ..., -1.7825e+00,
-8.0209e-02, 4.8693e-01],
[ 1.1573e+00, 1.9147e-01, 1.2556e+00, ..., 9.8702e-01,
1.3553e+00, 1.3145e-01]],
...,
[[-6.2970e-01, -1.9531e+00, -1.8307e+00, ..., -5.9016e+00,
-7.8357e+00, -3.0958e+00],
[-4.2862e+00, -6.6047e+00, -4.1301e+00, ..., 1.4761e+00,
7.4719e-02, -1.8102e+00],
[-4.6291e+00, -8.0910e+00, -6.4288e+00, ..., -3.4407e+00,
-8.4298e-01, -2.3831e+00],
...,
[-3.3642e+00, -7.4769e+00, -1.6646e+00, ..., -6.7621e+00,
-4.8627e+00, -2.4471e+00],
[-3.6380e+00, -5.5166e+00, -6.0131e+00, ..., -4.4327e+00,
-5.7775e+00, -2.8180e+00],
[-3.2365e+00, -2.9871e+00, -9.4265e-01, ..., -3.7827e+00,
-4.2248e+00, -1.4913e+00]],
[[-1.4666e+00, -2.2090e+00, -2.0067e+00, ..., -5.6771e+00,
-4.9967e+00, 2.4266e+00],
[-4.2976e+00, -4.5707e+00, -5.2123e+00, ..., 1.3447e+00,
-3.9136e+00, -1.6481e+00],
[-3.4608e+00, -3.5356e+00, -7.9926e+00, ..., 7.1493e+00,
5.2748e+00, 2.7107e-01],
...,
[-3.3976e+00, -3.3524e+00, 5.0787e+00, ..., -6.8438e+00,
-2.4818e+00, -5.4903e-01],
[-2.0018e+00, -3.9809e+00, -6.4037e-01, ..., -1.3630e-01,
-2.0787e+00, -5.8264e-01],
[-3.2669e+00, -2.4248e+00, -1.2037e+00, ..., -1.4748e+00,
-2.6408e+00, -9.6590e-01]],
[[ 1.3588e+00, 1.0040e+00, 1.6415e+00, ..., 5.6324e-01,
9.1894e-01, 1.8052e-01],
[ 5.0913e+00, 3.1713e+00, 2.5883e-01, ..., 2.3007e+00,
-2.7008e-01, 6.9814e+00],
[ 5.7346e+00, 8.0302e-01, 3.3385e+00, ..., -5.8437e+00,
-4.6616e+00, 2.2442e+00],
...,
[ 2.6999e+00, 3.0688e+00, 8.4517e+00, ..., 3.6127e+00,
3.4839e+00, 1.9069e+00],
[ 3.3569e+00, 4.7964e+00, 7.3366e+00, ..., 3.7495e+00,
3.1341e+00, 2.0607e+00],
[ 2.4951e+00, 1.5269e+00, -2.3009e+00, ..., 2.4685e-01,
-6.3393e-01, -3.3878e-01]]],
[[[-2.4982e-02, -1.8197e+00, -2.2247e+00, ..., 1.9591e-01,
-1.3305e+00, 4.1006e-01],
[ 3.6436e-01, -4.8697e+00, -4.8566e+00, ..., -3.1460e-01,
-1.2322e+00, -4.9171e-01],
[-5.7657e-01, -5.8129e+00, -4.2319e+00, ..., 2.7465e+00,
-2.2631e-01, -1.2281e+00],
...,
[-1.3743e+00, -9.6952e-01, -2.0660e-01, ..., -3.3333e+00,
-3.7890e+00, -9.7756e-01],
[-4.1673e+00, -2.1039e+00, -6.0044e-01, ..., -2.0816e+00,
-2.9352e+00, -7.4091e-03],
[-3.2198e+00, -2.5812e+00, -6.9545e-01, ..., -4.2302e+00,
-3.5342e+00, -1.5372e+00]],
[[-1.8490e+00, -4.6105e+00, -3.0658e+00, ..., -3.4316e+00,
-6.7166e+00, -1.2956e+00],
[-5.5068e+00, -9.4486e+00, -6.3653e+00, ..., -4.9869e+00,
-7.5906e+00, -4.1291e+00],
[-4.6285e+00, -8.0667e+00, -6.2743e+00, ..., -9.9635e-01,
-6.0242e+00, -4.8992e+00],
...,
[-4.1130e+00, -5.5143e-01, 9.6015e-01, ..., -8.1677e+00,
-9.5658e+00, -2.3657e+00],
[-1.0259e+01, -6.1569e+00, 1.3975e+00, ..., -1.0244e+01,
-1.0971e+01, -3.8767e+00],
[-1.8405e+00, -3.9452e+00, 1.3641e+00, ..., -4.4829e+00,
-4.2664e+00, -6.3485e-01]],
[[ 3.8139e-01, 1.2697e+00, -4.6247e-01, ..., 1.9591e+00,
1.6762e+00, 1.7054e-01],
[ 5.1481e-02, 3.6704e-01, -1.1442e+00, ..., 3.3070e+00,
2.3480e+00, -9.6691e-01],
[ 1.0652e+00, 2.4403e+00, -2.2135e+00, ..., 3.3849e+00,
3.6526e+00, 8.5102e-01],
...,
[-1.9747e+00, -1.7737e+00, -4.1659e+00, ..., 1.5976e+00,
9.0765e-01, -9.7956e-01],
[-2.0960e+00, -3.7694e+00, -2.7512e+00, ..., 5.9065e-01,
9.2092e-01, -5.7051e-01],
[ 2.1140e-01, -4.2610e+00, -4.8088e+00, ..., -1.8343e-01,
-5.3119e-01, -1.8712e+00]],
...,
[[-1.2984e+00, -2.0667e+00, -4.7694e-01, ..., -1.7169e+00,
-3.4822e+00, -2.1394e-01],
[-4.3140e+00, -6.6255e+00, -5.5699e+00, ..., -3.2333e+00,
-5.5208e+00, -3.4822e+00],
[-2.6866e+00, -3.8129e+00, -2.1660e+00, ..., -1.4945e+00,
-4.3989e+00, -2.1442e+00],
...,
[-1.7935e+00, -2.9012e+00, -9.9152e-01, ..., -5.4408e+00,
-6.2646e+00, -2.9718e+00],
[-5.4903e+00, -4.1024e+00, -2.5417e+00, ..., -4.4559e+00,
-5.2439e+00, -1.4216e+00],
[-3.5136e+00, -2.8497e+00, -1.9711e+00, ..., -3.3180e+00,
-2.5746e+00, 4.4646e-01]],
[[-2.1761e+00, -5.6227e-01, 1.8507e-01, ..., -4.8329e-01,
-3.3746e-01, 1.5300e-02],
[-5.7683e+00, -3.2321e+00, -3.6179e+00, ..., -2.2179e+00,
-1.6397e+00, 9.1250e-02],
[-4.5782e+00, -4.1380e+00, -3.2180e+00, ..., -1.7725e+00,
-2.8148e+00, -8.1147e-01],
...,
[-9.4292e-01, -2.7492e+00, -4.6148e+00, ..., -3.0007e+00,
-2.7668e+00, -9.4677e-01],
[-8.5203e-01, 1.9847e+00, -6.6165e+00, ..., -1.5157e+00,
-1.2842e+00, 1.9884e-01],
[-1.1872e+00, 2.5860e+00, -2.1573e+00, ..., -1.6668e+00,
-1.2008e+00, 3.3489e-01]],
[[ 1.6541e+00, 1.3201e+00, 1.1592e+00, ..., 6.7114e-01,
1.4507e+00, -4.7643e-02],
[ 3.5987e+00, 2.4584e+00, 1.9053e+00, ..., 3.1425e+00,
3.1689e+00, 3.0506e+00],
[ 1.9249e+00, 4.6068e-01, 1.1295e+00, ..., 3.1684e+00,
2.2560e+00, 1.6698e+00],
...,
[ 1.7148e+00, -1.2207e+00, -1.8196e+00, ..., 2.4974e+00,
1.3385e+00, 1.5816e-01],
[ 4.5978e+00, 2.0711e+00, -1.8806e+00, ..., 2.7773e+00,
1.7648e+00, 1.5706e+00],
[ 2.1950e+00, 4.4100e+00, -4.5465e-01, ..., 6.1425e-01,
5.6270e-01, 1.0379e+00]]]], device='cuda:0'),
tensor([[[[-1.0196e+00, 8.6430e-01, 2.3523e+00, ..., -1.2258e+00,
-9.3570e-01, -5.7436e-01],
[-1.0198e+00, 6.6176e-01, 3.7807e+00, ..., -2.0145e+00,
-2.1297e+00, -2.7425e+00],
[ 5.1860e-01, 2.1530e+00, -8.8951e-01, ..., -7.8684e+00,
-3.8736e+00, -4.3563e+00],
...,
[-8.4062e+00, -6.0090e-03, 3.0175e+00, ..., -2.4822e+00,
-3.5733e+00, -4.0896e+00],
[-3.6146e+00, -2.6090e+00, -1.3968e+01, ..., -7.3047e+00,
-1.7365e+00, -2.3844e+00],
[-4.3318e-01, -5.1066e+00, -9.5975e+00, ..., -2.9702e+00,
-3.6042e+00, -3.1907e+00]],
[[-1.6476e+00, -2.4654e+00, -3.4964e+00, ..., -9.4348e+00,
-7.2702e+00, -2.4229e+00],
[-4.6556e+00, -5.7293e+00, -5.6105e+00, ..., -9.6742e+00,
-1.0964e+01, -6.2490e+00],
[-4.2654e+00, -6.2840e+00, -1.4656e+01, ..., -8.3827e+00,
-8.1501e+00, -7.1084e+00],
...,
[-1.0313e+01, -6.0488e+00, -8.5400e+00, ..., -7.5856e+00,
-1.0167e+01, -4.4500e+00],
[-1.1936e+01, -1.5887e+01, -2.9054e+01, ..., -1.5166e+01,
-7.5055e+00, -6.2167e+00],
[-3.3520e+00, -1.5666e+00, -2.7220e+00, ..., -4.6967e+00,
-3.1499e+00, -1.9609e+00]],
[[-5.1923e-01, 2.0587e+00, 1.0258e+00, ..., 1.3814e+00,
2.6739e+00, 5.3490e-01],
[-2.6202e-01, 4.7172e-01, -1.5577e+00, ..., 1.0908e+00,
2.1445e+00, 4.3166e-01],
[ 4.7698e-01, 6.2840e-02, -6.1720e-01, ..., 8.6838e+00,
4.1811e+00, 2.4011e+00],
...,
[-8.3004e-02, -1.1769e+01, -1.2169e+01, ..., 4.1744e+00,
1.3501e-01, 6.5287e-01],
[-4.1690e+00, -5.5691e+00, 1.6828e+00, ..., -8.8038e-01,
5.6251e-01, 7.6470e-01],
[-4.0360e+00, -5.4127e+00, 2.9426e+00, ..., -1.3695e+00,
-1.3716e+00, 4.3762e-01]],
...,
[[-1.5385e+00, -1.5438e+00, -4.0652e-01, ..., -5.0189e+00,
-4.1879e+00, -8.1705e-01],
[-3.6927e+00, -3.1910e+00, -3.3946e+00, ..., -3.5868e+00,
-7.0443e+00, -5.7633e+00],
[-1.7571e+00, -4.0217e+00, -6.0356e+00, ..., -1.0875e+01,
-8.6181e+00, -5.3136e+00],
...,
[-2.8438e+00, 2.4377e-01, -1.0859e+01, ..., -6.8550e+00,
-7.1711e+00, -4.6792e+00],
[-1.1582e+00, -6.0217e+00, -1.9719e+01, ..., -6.4565e+00,
-5.3236e+00, -3.1497e+00],
[ 8.6613e-01, -2.0947e+00, -3.6131e+00, ..., -9.7198e-01,
-4.2768e+00, -1.8390e+00]],
[[-1.6559e+00, -1.4582e+00, -2.8334e+00, ..., -3.0291e+00,
-2.0893e+00, 1.1716e-01],
[-2.9605e+00, -1.7702e+00, -3.4432e+00, ..., -1.1656e+00,
-3.5868e+00, -1.3993e+00],
[-1.9874e+00, -1.6324e+00, -5.7785e+00, ..., -1.8025e+00,
-2.0994e+00, -5.8353e-01],
...,
[-2.8214e+00, 1.6941e+00, 1.6101e+00, ..., -4.2377e+00,
-1.0911e+00, 1.8985e+00],
[-3.2075e+00, -9.0320e+00, 3.5280e+00, ..., -3.6164e+00,
-1.1643e+00, 1.1481e+00],
[ 2.1117e-01, 2.0157e+00, 1.1736e-01, ..., -3.3451e-01,
-3.1159e-01, 3.0666e-01]],
[[ 1.0404e+00, 1.7044e-02, 8.2111e-01, ..., 1.7442e+00,
2.1752e+00, 5.0765e-01],
[ 3.7014e+00, 3.3416e+00, 3.8524e+00, ..., 1.7039e+00,
1.6487e+00, 2.6850e+00],
[ 1.7851e+00, 5.2242e+00, 5.1207e+00, ..., 4.9387e+00,
3.0289e+00, 1.2299e-01],
...,
[ 2.1194e+00, 8.1792e+00, 3.5887e+00, ..., 6.4371e+00,
3.6528e+00, 3.4473e+00],
[ 4.2824e+00, 1.0565e+01, 8.6667e+00, ..., 3.5823e+00,
2.5448e+00, 2.4435e+00],
[ 2.8923e+00, 2.7093e+00, -2.2335e+00, ..., 2.3262e+00,
2.8928e+00, 6.3027e-01]]],
[[[ 3.7266e-01, 8.6136e-01, -4.9154e-01, ..., -1.4578e+00,
-2.0261e+00, -1.1022e+00],
[ 4.3712e-01, -8.0616e-01, 6.2795e-01, ..., 4.5899e+00,
1.1921e+00, -1.7829e-01],
[ 1.8893e-01, -5.2400e-01, 9.1434e-01, ..., 6.4490e-01,
-1.3469e+00, -9.4709e-01],
...,
[-1.0477e+00, -2.2130e+00, -6.2636e-01, ..., -2.2753e+00,
-3.1889e+00, -2.7402e+00],
[-9.9793e-01, -8.0419e-01, -5.5653e-01, ..., -1.2549e+00,
-4.0835e-01, -2.0231e+00],
[-2.7122e+00, -5.1649e+00, -4.5046e+00, ..., -5.2719e+00,
-4.8084e+00, -4.0010e+00]],
[[-1.4173e+00, -2.8627e+00, -4.7411e+00, ..., -1.5514e+00,
-4.8293e+00, -1.6321e+00],
[-5.1346e+00, -6.6662e+00, -4.2591e+00, ..., -1.1882e-01,
-5.5874e+00, -2.9998e+00],
[-4.0821e+00, -6.4975e+00, -6.3615e+00, ..., -9.3742e+00,
-1.1412e+01, -5.2477e+00],
...,
[-5.5512e+00, -5.1739e+00, -3.4029e+00, ..., -1.2291e+01,
-1.0829e+01, -6.2572e+00],
[-7.1432e+00, -7.8182e+00, -6.5941e+00, ..., -6.9363e+00,
-7.7537e+00, -6.9128e+00],
[-2.5394e+00, -4.4614e+00, -4.1128e+00, ..., -4.5827e+00,
-4.7424e+00, -2.7836e+00]],
[[ 4.7702e-01, 2.1599e+00, 1.9430e+00, ..., 2.4154e+00,
4.2395e+00, 1.6688e+00],
[-3.1710e-01, 4.9683e-01, 4.0027e-01, ..., -8.9664e-02,
2.6773e+00, 8.9146e-01],
[ 5.6634e-01, 1.9777e+00, 2.2099e-01, ..., -1.4263e+00,
7.5174e-02, -2.4018e-01],
...,
[ 2.6554e-01, 8.9313e-01, 6.6676e-01, ..., -1.3959e+00,
-1.1618e+00, 5.3909e-01],
[ 8.4971e-01, 7.3992e-01, 8.1371e-01, ..., 4.5083e-01,
2.8745e-01, 3.3706e-01],
[ 1.3184e+00, 1.5704e+00, 1.3382e+00, ..., 1.6293e+00,
9.8668e-01, 1.0076e-01]],
...,
[[-7.6607e-01, -2.2816e+00, -2.9550e+00, ..., -3.3215e+00,
-5.8312e+00, -1.6724e+00],
[-3.0776e+00, -4.4246e+00, -1.9361e+00, ..., -1.7686e+00,
-5.5397e+00, -2.1484e+00],
[-2.1346e+00, -2.5376e+00, -2.0310e+00, ..., -5.2899e+00,
-7.4968e+00, -2.1223e+00],
...,
[-2.0063e+00, -1.7895e+00, 5.0082e-02, ..., -7.6105e+00,
-7.0018e+00, -3.4025e+00],
[-2.9795e+00, -2.9814e+00, -1.9889e+00, ..., -6.1336e-01,
-3.3162e+00, -2.6625e+00],
[-3.5205e+00, -3.8463e+00, -3.3105e+00, ..., -3.4752e+00,
-4.1761e+00, -2.0712e+00]],
[[-1.7788e+00, -1.3699e+00, -2.7718e+00, ..., -3.7715e+00,
-1.9211e+00, 8.8577e-01],
[-4.3955e+00, -1.9716e+00, -2.3333e+00, ..., -1.9680e+00,
-2.5104e+00, 7.5495e-02],
[-3.3961e+00, -1.2048e+00, -1.1705e+00, ..., -4.5976e-01,
-1.1972e+00, 7.4244e-01],
...,
[-3.2176e+00, -2.1383e+00, -2.9454e+00, ..., -2.6042e+00,
-5.0772e+00, -1.0938e+00],
[-3.5922e+00, -8.9214e-01, -2.7082e+00, ..., -8.2364e-01,
-2.7730e+00, -1.3711e+00],
[-4.3719e+00, -3.0385e+00, -3.0153e+00, ..., -2.3038e+00,
-2.8730e+00, -7.2207e-01]],
[[ 1.5849e+00, 8.6135e-01, 1.2627e+00, ..., 7.9034e-01,
2.3110e+00, 1.2504e-01],
[ 3.2081e+00, 2.4280e+00, 2.0411e+00, ..., 1.6170e+00,
2.8531e+00, 3.8601e+00],
[ 2.2749e+00, 1.2185e+00, 1.5000e+00, ..., 3.4007e+00,
2.3851e+00, 1.8399e+00],
...,
[ 2.0432e+00, 2.7405e-01, 2.0042e-01, ..., 5.4929e+00,
3.6050e+00, 2.5348e+00],
[ 3.7758e+00, 3.8145e+00, 2.5647e+00, ..., 1.3682e+00,
1.6231e+00, 2.7166e+00],
[ 1.9349e+00, 6.9951e-01, -1.1803e-02, ..., 3.3079e-01,
-5.3542e-02, -8.4877e-02]]],
[[[ 8.2524e-01, 2.4285e+00, 2.4562e+00, ..., -3.3489e+00,
-2.7913e+00, -1.6749e+00],
[ 1.1890e+00, -4.3416e-01, -3.5746e+00, ..., -2.3334e+00,
-2.2814e+00, -2.7060e+00],
[-1.1998e+00, -9.9120e+00, -4.9593e+00, ..., -5.1679e+00,
-4.1983e+00, -3.7454e+00],
...,
[-2.5840e+00, -6.8344e+00, -6.3161e+00, ..., -3.5465e+00,
-3.5038e+00, -2.9771e-01],
[-6.5959e+00, -6.2886e+00, -4.4587e+00, ..., -3.1064e+00,
-3.1050e+00, 4.1614e-01],
[-4.2363e+00, -3.8169e+00, -8.1631e-01, ..., -3.8858e+00,
-2.9404e+00, -1.2592e+00]],
[[-7.1862e-01, -2.6560e+00, -5.4826e+00, ..., -7.8434e+00,
-7.1881e+00, -2.9052e+00],
[-6.6443e+00, -9.6509e+00, -1.0962e+01, ..., -1.1140e+01,
-1.1296e+01, -5.5212e+00],
[-1.2084e+01, -1.8729e+01, -7.8956e+00, ..., -5.1004e+00,
-8.1046e+00, -5.1805e+00],
...,
[-8.7162e+00, -9.7488e+00, -5.1545e+00, ..., -7.3144e+00,
-9.1822e+00, -2.2599e+00],
[-1.0600e+01, -3.9606e+00, -3.0099e-01, ..., -1.0034e+01,
-1.0642e+01, -3.7956e+00],
[-2.1908e+00, -1.5434e+00, 9.8051e-01, ..., -4.1797e+00,
-3.8951e+00, -5.6007e-01]],
[[ 8.7898e-01, 1.6687e+00, 1.0972e+00, ..., 1.7679e+00,
2.4496e+00, 8.7411e-01],
[-1.4739e+00, -1.5884e+00, 1.8166e+00, ..., 1.6570e+00,
2.0501e+00, 9.2565e-01],
[ 6.0514e-02, 3.3781e+00, 4.6570e+00, ..., 7.5965e+00,
3.7110e+00, 1.6839e+00],
...,
[ 7.1951e-01, -1.0204e+00, -2.3536e+00, ..., 2.2637e-01,
2.2991e-01, -4.8524e-01],
[ 1.5837e+00, -3.5145e-01, -5.0983e+00, ..., -4.1095e-01,
1.8076e+00, -4.8735e-01],
[ 1.1724e+00, -1.7814e+00, -2.5733e+00, ..., -4.9786e-01,
-8.7823e-01, -1.9711e+00]],
...,
[[-2.5374e-01, -1.0405e+00, -2.8642e+00, ..., -5.5870e+00,
-5.5551e+00, -1.6460e+00],
[-3.2675e+00, -4.7333e+00, -4.2843e+00, ..., -1.6981e+00,
-7.1292e+00, -4.8049e+00],
[-3.8291e+00, -7.8204e+00, -6.5145e-01, ..., -6.5862e+00,
-6.5164e+00, -3.8307e+00],
...,
[-3.6178e+00, -2.8296e+00, 1.3178e+00, ..., -5.2287e+00,
-7.7190e+00, -3.1258e+00],
[-4.6016e+00, -9.6196e-01, -1.3656e+00, ..., -4.4227e+00,
-5.2985e+00, -1.0804e+00],
[-2.4021e+00, 5.0529e-01, -2.4361e+00, ..., -2.6632e+00,
-2.3151e+00, 4.9205e-01]],
[[-1.8329e+00, -2.0304e+00, -2.9659e+00, ..., -4.2977e+00,
-3.1640e+00, 3.0077e-02],
[-2.0546e+00, -2.9258e+00, -7.0929e+00, ..., -2.8028e+00,
-5.2312e+00, -2.0899e+00],
[-6.7667e+00, -3.7414e+00, -5.2766e+00, ..., 2.9370e-01,
-1.0949e+00, -1.5205e+00],
...,
[-4.3207e+00, -1.6111e+00, -4.8517e+00, ..., -2.6193e+00,
-2.8789e+00, -8.8918e-01],
[-3.3195e+00, 2.1889e+00, -7.3297e+00, ..., -1.0933e+00,
-1.2658e+00, 6.1562e-01],
[-1.3351e+00, 1.8001e+00, -6.4458e+00, ..., -7.0886e-01,
-1.2039e+00, 2.8666e-01]],
[[ 8.0223e-01, 1.1792e+00, 2.8293e+00, ..., 4.5547e-01,
1.4757e+00, 2.9745e-01],
[ 2.6241e+00, 1.0502e+00, -1.5646e-01, ..., 2.5837e+00,
2.4305e+00, 2.6407e+00],
[ 6.0959e+00, 2.0485e+00, 4.3725e+00, ..., 2.9007e+00,
1.9491e-01, -4.5363e-01],
...,
[ 4.6740e+00, -3.4427e-01, 3.0508e-01, ..., 2.5724e+00,
1.8133e+00, -4.2335e-01],
[ 2.8426e+00, -2.0636e-01, -2.1621e+00, ..., 1.9374e+00,
1.3723e+00, 1.7139e+00],
[ 2.2313e+00, 2.3445e+00, 3.0021e-02, ..., 7.4946e-01,
6.4283e-01, 1.1602e+00]]],
[[[-4.0628e-01, -2.1485e+00, -2.2578e+00, ..., -1.8475e+00,
-1.4095e+00, -6.0442e-01],
[-7.0476e-01, -1.6806e+00, 5.3703e-01, ..., -3.5718e+00,
-5.5219e+00, -2.0829e-01],
[-2.2089e+00, -4.2898e-01, -3.5063e+00, ..., -2.3685e+00,
-1.2478e+00, 1.2934e+00],
...,
[-2.3388e-03, 1.6765e+00, -1.8492e+01, ..., -1.1506e+01,
-9.9939e+00, 3.4774e+00],
[-1.5822e+00, -7.0534e-01, -1.5306e+01, ..., -1.1135e+01,
-1.2277e+01, 2.4370e+00],
[-2.7910e+00, -5.2894e+00, -1.0671e+01, ..., -3.8527e+00,
-4.1958e+00, 3.0188e+00]],
[[-1.7971e+00, -6.0229e+00, -8.4042e+00, ..., -5.7793e-01,
-8.0949e-01, 8.1928e-01],
[-2.1895e+00, -2.1212e+00, -1.3664e+00, ..., -7.0106e+00,
-8.9686e+00, -1.4211e+00],
[-6.9390e+00, -1.1016e+01, -1.6454e+01, ..., 3.6444e+00,
2.4007e+00, 1.0565e+00],
...,
[-8.0839e+00, -9.3716e+00, -2.7223e+01, ..., -3.7337e+00,
-5.2278e+00, 5.3980e-01],
[-1.0269e+01, -1.1129e+01, -2.2427e+01, ..., -1.1997e+01,
-1.6842e+01, -4.0477e+00],
[-3.2268e+00, -4.1762e+00, -6.1999e+00, ..., -1.6101e+00,
-7.7878e+00, -2.3608e+00]],
[[ 1.4518e+00, 2.1892e+00, 2.0710e+00, ..., 1.2169e+00,
1.1245e+00, 5.1344e-01],
[-1.5371e-01, 1.7329e+00, 2.6630e+00, ..., 2.9238e+00,
3.8200e+00, 4.6062e-01],
[ 2.3296e-01, 1.9718e+00, 4.8506e+00, ..., -9.2295e-01,
1.5379e+00, -5.0592e-01],
...,
[-1.3708e-01, -4.5623e+00, -4.8844e+00, ..., -1.9855e+00,
-4.9456e+00, -1.5164e+00],
[-6.8142e-01, -2.1586e+00, -3.0953e+00, ..., 5.2028e+00,
2.2801e+00, -2.1600e+00],
[ 1.1611e+00, -8.2514e-01, -2.4584e+00, ..., 3.0075e+00,
6.0645e-01, -7.0458e+00]],
...,
[[-2.8759e+00, -4.4897e+00, -4.2517e+00, ..., -1.5748e+00,
-1.1522e+00, 6.3796e-01],
[-7.8652e-01, 5.8359e-01, 7.2526e-02, ..., -8.4816e+00,
-8.3055e+00, -2.1518e+00],
[-2.7249e+00, -2.7913e+00, -5.1573e+00, ..., 5.8547e+00,
3.8079e+00, 1.3210e+00],
...,
[-2.4242e+00, -2.9429e+00, -7.1194e+00, ..., -5.9111e+00,
-5.0517e+00, -6.7809e+00],
[-3.2169e+00, -5.2308e+00, -8.4614e+00, ..., -7.4458e+00,
-2.7727e+00, -4.4537e+00],
[-3.2606e+00, -3.5350e+00, -4.8224e+00, ..., 8.3926e-01,
-1.0933e+00, -3.1411e-01]],
[[-3.1100e+00, -2.5403e+00, -2.7742e+00, ..., -2.8229e+00,
5.1607e-01, 3.1844e-01],
[-3.3699e+00, -1.4865e+00, -2.7522e+00, ..., -1.3213e+01,
-6.5616e+00, 7.5740e-01],
[-4.7866e+00, -5.4978e+00, -7.8778e+00, ..., -1.8335e+00,
-3.6147e-01, 2.6508e+00],
...,
[-1.8070e+00, -7.1512e+00, -5.4708e+00, ..., -1.0417e+01,
-5.5926e+00, -1.2563e+00],
[-2.3344e+00, -4.9013e+00, -2.5773e+00, ..., -2.1697e+01,
-1.6687e+01, -4.9130e+00],
[-2.8797e+00, -3.0184e+00, -2.7933e+00, ..., -1.1432e+01,
-9.8583e+00, -3.1564e+00]],
[[ 1.8952e+00, 5.5122e-01, 1.3421e+00, ..., -1.8436e+00,
3.2190e-01, -8.5947e-01],
[ 7.4574e-01, -3.9678e-01, 3.9517e-01, ..., -6.3458e+00,
-2.4140e+00, 2.0175e+00],
[ 1.3665e+00, 6.1419e+00, 5.5678e+00, ..., -7.0538e+00,
-4.2489e+00, 1.4414e+00],
...,
[ 4.6782e+00, 4.5156e+00, 3.0760e+00, ..., -3.8221e+00,
-6.2584e+00, 3.2159e-01],
[ 5.2146e+00, 5.1386e+00, 3.1617e+00, ..., -5.4361e+00,
-5.3349e+00, 1.0002e+00],
[ 3.3677e+00, 1.7285e+00, -1.2669e-01, ..., -6.2649e+00,
-5.1408e+00, 5.6276e-01]]],
[[[ 8.0751e-01, 1.1890e+00, 1.1925e+00, ..., 1.9637e+00,
-1.0157e+00, -1.3733e+00],
[-6.1296e-01, -2.8479e+00, -3.0638e+00, ..., -3.3618e+00,
-6.8657e+00, -1.3521e+00],
[-1.0598e-01, 2.6258e-01, 1.1349e-01, ..., -1.7519e+01,
-3.9615e+00, 2.0143e+00],
...,
[-3.3920e-01, 3.1237e+00, -8.2669e+00, ..., -1.8658e+00,
-1.4426e+00, -6.9198e-01],
[-2.7555e+00, -5.7984e-02, 7.5766e-01, ..., -2.6971e-01,
-1.0869e+00, -1.9057e+00],
[-2.4574e+00, -2.6698e+00, -2.6011e+00, ..., -2.8906e+00,
-3.3356e+00, -3.0666e+00]],
[[-1.4834e+00, -2.7594e+00, -5.9997e+00, ..., -8.0240e+00,
-6.9451e+00, -2.4492e+00],
[-4.4136e+00, -1.0574e+01, -7.7486e+00, ..., -1.4973e+01,
-1.1210e+01, -2.8246e+00],
[-4.1857e-01, -2.2121e+00, -6.0828e+00, ..., -1.4783e+01,
-7.0300e-01, 1.1457e-01],
...,
[-3.0862e+00, 1.1985e+00, -8.2737e+00, ..., -4.9600e+00,
-4.5962e+00, -3.1587e+00],
[-8.2151e+00, -5.7749e+00, -3.9509e+00, ..., -5.9206e+00,
-7.0697e+00, -5.9818e+00],
[-2.3230e+00, -6.1084e+00, -7.2308e+00, ..., -3.3315e+00,
-3.2871e+00, -1.8273e+00]],
[[ 2.9846e-01, 3.7559e+00, 2.2135e+00, ..., 8.5935e-01,
3.9178e-01, 8.9759e-01],
[ 5.8653e-01, 4.1691e+00, 6.6926e+00, ..., 2.5348e+00,
3.0113e+00, 2.2199e+00],
[ 1.0628e+00, 7.2568e-01, 3.2612e+00, ..., 7.4483e+00,
3.5441e+00, -3.5325e-01],
...,
[-1.0152e+01, -6.7505e+00, -6.2781e+00, ..., -1.3632e+00,
-8.1380e-01, 8.7224e-01],
[-6.0884e+00, -8.7826e+00, -7.0486e+00, ..., 1.2019e+00,
8.0127e-01, 1.5019e-02],
[ 2.3453e-01, -1.4639e+00, -3.6871e+00, ..., 1.2468e+00,
1.8573e+00, 4.4231e-01]],
...,
[[-1.2816e+00, -3.3576e+00, -3.1641e+00, ..., -8.3989e-01,
-3.2777e+00, -4.0808e-01],
[-4.4627e+00, -6.5693e+00, -6.8680e+00, ..., -5.9205e+00,
-6.3496e+00, -1.7140e+00],
[-1.2963e+00, -2.4758e+00, -1.9376e+00, ..., -5.4368e+00,
2.2903e+00, -3.3630e-01],
...,
[-1.3000e+00, -5.8983e+00, -1.5685e+00, ..., -2.6011e+00,
-3.7029e+00, -2.3861e+00],
[-3.3590e+00, -4.7059e+00, -4.8351e-01, ..., -2.0320e+00,
-4.5038e+00, -2.1504e+00],
[-4.3179e+00, -6.4057e+00, -6.8604e+00, ..., -2.3846e+00,
-2.9099e+00, -7.7658e-01]],
[[-1.7087e+00, -2.7449e+00, -9.8836e-01, ..., 4.7029e-01,
-6.9889e-01, -4.6019e-01],
[-3.7578e+00, -7.8031e+00, -8.4214e+00, ..., -1.1356e+00,
-4.4371e+00, -3.2205e+00],
[-2.3996e+00, -5.3906e+00, -9.8548e+00, ..., -7.4487e+00,
-8.1466e+00, -2.3715e+00],
...,
[-2.4950e-01, -8.2559e+00, -3.0432e+00, ..., 9.4601e-01,
-2.0122e+00, -1.1256e+00],
[ 1.0973e+00, 1.8323e+00, 2.4915e+00, ..., -5.1096e-01,
-3.3229e+00, -1.8485e+00],
[-5.8507e-01, -1.0152e+00, 7.2019e-01, ..., -1.0469e+00,
-2.7268e+00, -1.2812e+00]],
[[ 1.8827e+00, 9.8696e-01, 2.4557e+00, ..., 4.6994e+00,
1.4894e+00, -4.5676e-01],
[ 2.7331e+00, 2.9210e+00, 3.7807e+00, ..., 3.6054e+00,
1.7370e-01, 3.0111e+00],
[ 8.2333e-01, 2.5807e+00, -4.3912e-01, ..., -1.8179e+00,
2.5627e+00, 2.9379e+00],
...,
[ 3.3222e+00, -2.2141e+00, -5.6658e+00, ..., 1.8691e+00,
8.2858e-01, 4.6455e-01],
[ 5.2367e+00, 3.3763e+00, 9.5018e-01, ..., 2.4248e+00,
1.4276e+00, 1.7507e+00],
[ 1.6200e+00, 5.4074e+00, 3.4136e+00, ..., 6.1016e-01,
-2.6844e-01, -1.3964e-01]]]], device='cuda:0'),
tensor([[[[ 3.3983e-01, 1.2108e+00, 7.5037e-02, ..., -1.7481e+00,
-1.6972e+00, -1.6770e+00],
[ 3.6936e-01, 7.9022e-01, 9.1857e-01, ..., -1.0073e-01,
-4.7070e+00, -2.6890e+00],
[-5.1818e-01, -7.0463e-01, 3.4970e-01, ..., -5.2999e+00,
-7.2662e+00, 6.2321e-01],
...,
[-6.7914e+00, -6.8718e+00, 4.0410e-01, ..., -2.3316e+00,
-3.6044e+00, -1.8390e-01],
[-4.9622e+00, -3.3186e+00, 5.0713e-01, ..., -3.4259e+00,
-3.9943e+00, 1.4900e-01],
[-2.0930e+00, -5.8526e+00, -8.0830e-01, ..., -3.4962e+00,
-3.0431e+00, -1.1233e+00]],
[[-1.1582e+00, -1.8280e+00, -3.6773e+00, ..., -6.1036e+00,
-6.8719e+00, -1.7240e+00],
[-4.2210e+00, -4.4481e+00, -3.6512e+00, ..., -8.1001e+00,
-1.0644e+01, -2.5670e+00],
[-3.7955e+00, -4.2623e+00, -3.9239e+00, ..., -1.1928e+01,
-8.1830e+00, 3.4252e-01],
...,
[-9.0712e+00, -6.3290e+00, 4.5503e-01, ..., -8.5365e+00,
-8.4354e+00, -1.3883e+00],
[-6.0942e+00, -9.4852e+00, -2.2742e+00, ..., -9.2899e+00,
-1.0967e+01, -4.3077e+00],
[-1.1393e+00, -4.7345e+00, -3.8651e+00, ..., -2.7298e+00,
-3.7113e+00, -5.8529e-01]],
[[ 2.9131e-03, 1.3193e+00, 1.7520e+00, ..., 1.5248e+00,
1.8143e+00, 6.2986e-01],
[ 7.9079e-02, -2.4559e-03, 7.8012e-01, ..., 7.4424e-01,
3.6067e+00, 8.0837e-01],
[ 7.3866e-01, 1.4067e+00, 1.7351e+00, ..., 8.5177e-01,
3.2308e+00, -1.5004e-01],
...,
[ 2.9490e-01, -2.4573e+00, -1.0339e+01, ..., -3.0336e-01,
-3.7653e-01, -7.6334e-01],
[-2.3357e+00, -6.4632e+00, -9.0670e+00, ..., -6.7113e-01,
-4.2581e-01, -2.0017e+00],
[-1.2531e+00, -2.9533e+00, -7.9253e+00, ..., -7.1901e-01,
-3.8029e-01, -1.9006e+00]],
...,
[[-6.7240e-01, -1.6693e+00, -2.3309e+00, ..., -4.2499e+00,
-4.9888e+00, -5.2571e-01],
[-2.8432e+00, -2.7458e+00, -1.6067e+00, ..., -4.4855e+00,
-5.8708e+00, -1.2134e+00],
[-2.1697e+00, -1.5706e+00, -1.6227e+00, ..., -6.9999e+00,
-8.2624e-01, 2.2750e+00],
...,
[-4.0674e+00, -1.9481e+00, -2.7265e+00, ..., -1.7045e+00,
-4.6744e+00, -1.0372e+00],
[-2.1383e+00, -2.4674e+00, -2.5510e+00, ..., -3.6197e+00,
-5.3766e+00, -1.4406e+00],
[-2.1864e+00, -4.3920e+00, -1.4396e+00, ..., -2.4656e+00,
-2.2632e+00, 4.8115e-01]],
[[-1.4169e+00, -2.4590e+00, -3.9172e+00, ..., -1.9140e+00,
-5.9136e-01, -2.8910e-01],
[-3.9254e+00, -3.4606e+00, -4.8428e+00, ..., -5.2566e+00,
-4.3247e+00, -1.5320e+00],
[-4.7876e+00, -3.3466e+00, -3.2564e+00, ..., -2.8695e+00,
-1.8364e+00, -2.4980e+00],
...,
[-3.7807e+00, 2.2694e-01, -6.5722e+00, ..., -1.0223e+00,
-3.8729e-01, -2.7015e-01],
[-2.9228e+00, -1.7081e-02, -5.5185e+00, ..., -3.7167e-01,
5.2143e-01, 1.0121e+00],
[-2.7285e+00, 5.8460e-01, 1.4829e+00, ..., 4.8462e-01,
2.0484e-01, 7.3301e-01]],
[[ 1.2377e+00, -3.7037e-02, 1.1447e+00, ..., 1.2421e+00,
8.5456e-01, -1.4353e+00],
[ 2.3827e+00, 1.2457e+00, 1.3601e+00, ..., 4.1504e+00,
1.7953e+00, 2.3507e+00],
[ 1.1842e+00, 9.1725e-01, 7.9862e-01, ..., 1.8713e+00,
-4.6332e-01, 2.0145e+00],
...,
[ 4.2254e-01, 3.5804e+00, 7.5326e-01, ..., 3.4531e+00,
2.6556e+00, 9.9558e-01],
[ 7.6461e-01, 6.6223e+00, -3.8875e-01, ..., 3.1197e+00,
1.8687e+00, 2.0435e+00],
[ 5.1112e-03, 3.4689e+00, 3.0833e-01, ..., 1.5312e+00,
1.7677e-01, 7.6612e-01]]],
[[[-2.1859e-02, 1.5780e+00, 7.8824e-01, ..., 1.9743e-01,
4.4992e-01, -1.7541e-01],
[ 3.0865e-01, 7.2207e-01, 1.4128e+00, ..., 2.6869e-01,
5.9505e-01, -5.3926e-01],
[-9.8652e-01, 1.8426e-01, 5.1162e-01, ..., -9.9688e-02,
-3.1030e-02, -8.0110e-01],
...,
[-2.9409e-01, -3.9798e-01, 3.2883e-01, ..., 1.4787e-01,
8.7550e-01, -2.4273e-01],
[-1.8785e+00, -6.2322e-01, -7.5318e-02, ..., 2.9281e-01,
4.5559e-01, -1.1507e+00],
[-1.8440e+00, -3.1666e+00, -2.0631e+00, ..., -3.1759e+00,
-2.8906e+00, -2.7839e+00]],
[[-7.4731e-01, -1.5851e+00, -3.5984e+00, ..., -1.9440e+00,
-3.6077e+00, -2.2928e+00],
[-4.0575e+00, -4.3829e+00, -3.9578e+00, ..., -2.6071e+00,
-3.7749e+00, -3.4596e+00],
[-3.2519e+00, -3.9253e+00, -4.3194e+00, ..., -2.7333e+00,
-2.7456e+00, -2.6133e+00],
...,
[-5.4514e+00, -6.1257e+00, -5.3624e+00, ..., -2.9514e+00,
-3.0602e+00, -2.5057e+00],
[-8.8050e+00, -8.5984e+00, -6.8472e+00, ..., -4.7022e+00,
-5.1523e+00, -5.0250e+00],
[-2.1370e+00, -3.9984e+00, -2.5046e+00, ..., -3.0114e+00,
-2.8076e+00, -1.6531e+00]],
[[-1.9175e-01, 2.4525e+00, 1.6928e+00, ..., 1.6630e+00,
1.7818e+00, 8.8131e-01],
[-6.8714e-01, 1.9933e-01, 8.2690e-01, ..., 8.0388e-01,
1.4966e+00, 8.9803e-01],
[-4.9419e-02, 1.7556e+00, 1.7129e+00, ..., 1.7008e+00,
2.1987e+00, 1.5355e+00],
...,
[-6.8423e-01, 7.9592e-01, 1.4205e+00, ..., 1.6515e+00,
1.0014e+00, 7.6074e-01],
[-1.0757e+00, -3.6231e-01, 7.9878e-01, ..., 1.4650e+00,
8.4941e-01, 4.7003e-01],
[ 1.4917e-01, -1.0767e-01, 3.1983e-01, ..., 1.6804e+00,
1.5887e+00, 6.1170e-01]],
...,
[[-9.5905e-01, -1.8389e+00, -2.0993e+00, ..., -1.4419e+00,
-2.0301e+00, -4.5385e-01],
[-2.9675e+00, -3.7308e+00, -2.6333e+00, ..., -2.4848e+00,
-3.6922e+00, -2.9187e+00],
[-2.0217e+00, -2.5986e+00, -1.7074e+00, ..., -1.2035e+00,
-2.7875e+00, -1.8479e+00],
...,
[-1.6790e+00, -2.2643e+00, -8.6953e-01, ..., -1.8871e+00,
-2.0587e+00, -9.9678e-01],
[-2.9226e+00, -4.2511e+00, -2.0280e+00, ..., -1.4631e+00,
-2.6519e+00, -1.6317e+00],
[-2.3963e+00, -2.0598e+00, -1.2488e+00, ..., -2.0194e+00,
-2.2337e+00, -6.1907e-01]],
[[-1.4384e+00, -2.1304e+00, -2.7651e+00, ..., -5.8976e-01,
-1.3817e+00, -6.0855e-01],
[-3.3223e+00, -2.6551e+00, -2.7907e+00, ..., -1.5774e+00,
-3.5821e+00, -1.6977e+00],
[-3.0025e+00, -2.9245e+00, -3.8061e+00, ..., -7.2172e-01,
-3.9501e+00, -2.5735e+00],
...,
[-2.9923e+00, -2.9555e+00, -3.8748e+00, ..., -6.8466e-01,
-2.8319e+00, -2.0800e+00],
[-2.5755e+00, 1.2058e-02, -1.6216e+00, ..., -1.0544e+00,
-3.2226e+00, -1.9061e+00],
[-2.9266e+00, -1.8854e+00, -1.8317e+00, ..., -1.6136e+00,
-2.7312e+00, -1.2330e+00]],
[[ 8.4696e-01, 2.1991e-01, 8.1583e-01, ..., 3.0884e-01,
1.2921e+00, 6.6667e-01],
[ 3.6633e+00, 2.8208e+00, 2.3901e+00, ..., 6.6324e-01,
1.0328e+00, 2.4943e+00],
[ 1.3343e+00, 6.7129e-01, 5.9157e-01, ..., 1.0115e-01,
1.0779e-01, 8.1051e-01],
...,
[ 3.4432e+00, 2.6240e+00, 2.6494e+00, ..., 3.9662e-01,
1.2024e-01, 5.0228e-01],
[ 4.4677e+00, 4.6186e+00, 3.0589e+00, ..., 1.1092e+00,
9.2556e-01, 1.1695e+00],
[ 2.7066e+00, 2.5649e+00, 1.6976e+00, ..., -1.9079e-01,
-3.0837e-01, -1.3498e-01]]],
[[[-1.7642e+00, -6.1048e+00, -2.8558e+00, ..., -3.5746e-01,
-3.3895e-01, 5.5061e-01],
[-7.0046e+00, -2.0925e+00, 5.6614e+00, ..., -2.8354e+00,
-7.7884e-01, 4.5324e-01],
[-4.5722e+00, 3.0702e-01, -3.4917e-01, ..., -4.8754e+00,
-1.0385e+00, -5.8965e-01],
...,
[ 1.4326e+00, -3.6007e+00, -1.5291e+01, ..., -2.0836e+00,
-9.9595e-01, -8.1771e-01],
[-1.2010e+00, -6.8291e+00, -7.6379e+00, ..., -3.0404e+00,
-2.4871e+00, -8.3652e-01],
[ 8.1848e-01, -7.2674e+00, -3.3389e+00, ..., -3.0418e+00,
-2.2978e+00, -8.2095e-01]],
[[-6.4865e+00, -9.8340e+00, -6.4516e+00, ..., -4.4141e+00,
-5.5874e+00, -2.3826e-01],
[-8.5582e+00, -1.1256e+00, 3.2061e+00, ..., -3.7418e+00,
-6.6314e+00, -1.9753e+00],
[-1.6099e-01, 3.8164e+00, -3.1524e+00, ..., -3.5369e+00,
-4.8627e+00, -4.7167e+00],
...,
[-4.4211e+00, -1.5919e+01, -1.8252e+01, ..., -5.9307e+00,
-5.7115e+00, -2.0118e+00],
[-5.7619e+00, -1.1319e+01, -2.9541e+00, ..., -8.7373e+00,
-9.0600e+00, -3.6550e+00],
[ 4.5342e-01, -4.0102e+00, 3.0072e-01, ..., -3.7014e+00,
-2.9525e+00, -2.4210e-01]],
[[-1.5016e-02, 4.1329e+00, 3.2818e+00, ..., 1.1796e+00,
2.3420e+00, 7.2362e-01],
[ 3.1803e+00, -5.4668e-01, -4.1952e+00, ..., 2.2483e+00,
1.4942e+00, -8.3971e-01],
[ 2.0180e+00, -4.9694e+00, -5.6248e+00, ..., 3.8879e+00,
3.2520e+00, 7.5221e-01],
...,
[-4.8153e-01, 3.8318e-01, 5.3685e+00, ..., 1.3447e+00,
1.1039e+00, 5.0668e-01],
[-9.8958e-01, 2.7409e+00, -1.5378e-01, ..., 1.4983e+00,
1.6235e+00, 2.1246e-01],
[-2.3865e+00, 1.5461e+00, -8.6376e+00, ..., 9.0289e-01,
5.1234e-01, -1.1724e+00]],
...,
[[-3.9774e+00, -5.2414e+00, -3.2625e+00, ..., -2.3680e+00,
-2.6386e+00, -4.3298e-01],
[-5.9290e+00, 2.4445e+00, 2.1854e+00, ..., -4.0023e+00,
-2.4774e+00, -3.0934e+00],
[-7.2507e-01, -6.2476e-02, -6.1960e+00, ..., -3.0105e+00,
-2.8112e+00, -1.3661e+00],
...,
[-8.5766e-02, -3.1872e+00, -8.2340e+00, ..., -5.0224e+00,
-4.3603e+00, -1.9152e+00],
[-1.0848e+00, -2.5753e+00, 1.7227e+00, ..., -5.0832e+00,
-4.4163e+00, -1.1938e+00],
[ 3.2987e-01, -1.8943e+00, 2.6701e+00, ..., -2.8175e+00,
-1.1848e+00, 1.2059e+00]],
[[-2.0267e+00, -1.6229e+00, -3.8386e+00, ..., -1.8026e+00,
-1.7814e+00, -1.4487e+00],
[-4.9464e+00, -1.6419e+00, 7.6106e-01, ..., -1.7839e+00,
-3.0505e-01, -4.6616e-01],
[-5.3625e+00, -8.8585e+00, -5.9233e+00, ..., -2.0865e+00,
-1.0648e+00, -4.4089e-01],
...,
[-5.1314e+00, -1.0174e+01, -6.4374e+00, ..., -2.0621e+00,
-1.8938e+00, -4.5958e-01],
[-4.4376e+00, -2.2733e+00, -1.9374e+00, ..., -4.3413e-01,
-1.1901e+00, 3.5749e-02],
[-1.2367e+00, 4.4127e-02, 1.1269e+00, ..., 6.5772e-01,
-4.9767e-01, 3.1823e-01]],
[[ 3.4599e+00, 2.3423e-02, 1.2804e+00, ..., 4.6934e-01,
4.7189e-01, -6.6252e-01],
[ 2.3098e+00, 4.4802e+00, 5.9852e+00, ..., 1.9668e-01,
1.1126e+00, 1.1935e+00],
[-1.4617e+00, 2.8036e+00, -2.3283e+00, ..., -8.0023e-01,
4.4029e+00, 2.3632e+00],
...,
[ 1.9484e+00, 4.6212e+00, -2.1169e+00, ..., 1.1485e+00,
1.1751e+00, 6.5561e-01],
[ 2.3344e+00, 8.8869e-01, -7.6171e-01, ..., 2.9477e+00,
1.7492e+00, 1.8529e+00],
[ 1.6498e+00, -1.1937e+00, 2.4925e+00, ..., 1.2269e+00,
-4.9751e-02, 8.0244e-01]]],
[[[-2.8768e+00, -4.1835e+00, -5.0176e+00, ..., -2.7222e+00,
-2.7394e-01, 9.5403e-01],
[-4.0362e+00, -4.4532e+00, -5.1780e+00, ..., -4.6800e+00,
-2.2244e+00, 1.1545e+00],
[-5.2185e+00, -3.4427e+00, -6.1876e+00, ..., -1.6661e+00,
-1.9450e+00, 6.7763e-01],
...,
[-8.6269e-01, -4.6433e+00, -3.7836e+00, ..., -1.8236e+00,
-1.2215e+00, -1.4664e-01],
[-2.6928e+00, -3.8420e+00, -1.5984e+00, ..., 1.3822e+00,
2.6050e-01, -1.7281e+00],
[-3.1359e+00, -5.1153e+00, -9.9495e-01, ..., -3.0468e+00,
-3.0956e+00, -3.5066e+00]],
[[-3.4507e+00, -2.2346e+00, -2.4748e+00, ..., -7.4635e+00,
-5.7910e+00, -3.7556e-01],
[-3.5320e+00, 8.6778e-01, 2.1186e+00, ..., -7.6657e+00,
-7.8655e+00, -1.8256e+00],
[-3.4544e+00, -3.3353e+00, -7.4588e+00, ..., -6.0549e+00,
-6.7916e+00, -1.2572e+00],
...,
[-7.5904e+00, -6.3073e+00, -9.7826e-01, ..., -6.2631e+00,
-6.6036e+00, -3.3375e+00],
[-7.1520e+00, -6.5276e+00, -1.8045e+00, ..., -3.7275e+00,
-7.1872e+00, -6.7142e+00],
[-2.1825e+00, -3.8804e+00, -1.0982e+00, ..., -3.0786e+00,
-3.3110e+00, -2.2390e+00]],
[[ 1.2503e+00, 3.4097e+00, 8.9287e-01, ..., 8.1176e-01,
1.8603e+00, 7.0110e-01],
[ 9.5293e-01, -3.3795e+00, -8.3924e+00, ..., 3.8811e+00,
1.6476e+00, -8.9925e-01],
[ 3.4795e-01, 1.6559e+00, 3.1232e+00, ..., 6.1566e+00,
5.3359e+00, 1.6014e+00],
...,
[-4.2000e-01, 1.7712e-01, 3.5631e+00, ..., 8.7339e-01,
1.1603e+00, -2.7363e-01],
[ 7.7146e-01, -6.0573e-01, -2.4342e+00, ..., 6.3896e-01,
3.7948e-01, -6.5631e-02],
[ 1.4903e+00, 1.4765e+00, -1.4521e-02, ..., 5.8786e-01,
5.3621e-01, 6.8554e-02]],
...,
[[-2.2223e+00, -2.7401e-01, 7.0786e-01, ..., -2.9029e+00,
-4.2947e+00, -4.6430e-01],
[-4.2698e+00, -7.8583e-02, 1.2807e+00, ..., -5.9173e+00,
-5.6922e+00, -3.3041e+00],
[-3.2519e+00, -4.8188e+00, -4.0903e+00, ..., -1.7175e+00,
-4.1352e+00, -2.2607e+00],
...,
[-2.1254e+00, -5.4479e-01, 3.1244e+00, ..., -3.1189e+00,
-4.5614e+00, -2.1680e+00],
[-3.8720e+00, -3.5704e+00, -9.0764e-01, ..., -8.6414e-01,
-3.5815e+00, -2.5361e+00],
[-3.7610e+00, -2.8644e+00, -1.4892e-01, ..., -3.1356e+00,
-3.6966e+00, -1.6118e+00]],
[[ 6.9160e-01, 1.0203e+00, 4.1595e-01, ..., -8.9501e-01,
-1.0037e+00, -1.3841e+00],
[-3.1921e+00, -1.4346e+00, 2.6136e+00, ..., 2.0158e+00,
1.7576e-01, -2.1992e-01],
[-2.0525e+00, -6.7937e+00, -7.3641e+00, ..., 7.1727e-01,
4.9147e-01, 5.4867e-01],
...,
[-2.5178e+00, -2.0581e+00, -4.4634e-01, ..., -4.7766e-01,
-1.1491e+00, -1.0325e-01],
[-2.9201e+00, 1.5134e+00, 8.1285e-01, ..., -9.5431e-01,
-3.3821e+00, -9.1973e-01],
[-4.7321e+00, -9.1778e-01, 3.0864e-01, ..., -1.9437e+00,
-3.4743e+00, -1.1053e+00]],
[[ 2.5653e+00, 4.3440e-01, 1.2713e+00, ..., 2.2869e-01,
-3.5463e-01, -5.1077e-01],
[ 1.4117e+00, -8.7116e-01, -3.2807e+00, ..., 4.7353e+00,
1.6891e+00, 2.2667e+00],
[ 1.3589e-01, -2.0008e+00, -3.6523e+00, ..., 2.9923e+00,
3.7399e+00, 2.8242e+00],
...,
[ 3.8488e+00, 1.5805e+00, -5.9157e-01, ..., 2.7493e+00,
2.5989e+00, 1.9550e+00],
[ 4.1415e+00, 2.3872e+00, -1.2114e+00, ..., 2.4360e+00,
3.0350e+00, 3.0951e+00],
[ 1.9103e+00, 9.5991e-01, -1.6026e-01, ..., -3.1946e-01,
-4.7667e-01, -8.6069e-02]]],
[[[ 7.3178e-01, 6.6595e-01, -2.1621e-01, ..., 5.6702e-01,
-2.1238e-01, -5.0083e-01],
[ 1.0664e+00, 1.5145e-01, 1.0840e+00, ..., 6.0031e-01,
-2.0201e+00, -1.6502e-01],
[-5.8728e-01, -3.5790e-01, 1.4214e+00, ..., -9.5976e-01,
-1.6065e+00, 1.7616e+00],
...,
[-1.5095e+00, -8.4098e-01, 7.7461e-01, ..., 9.1519e-02,
-1.2811e-02, -3.7706e-02],
[-1.7535e+00, -6.8033e-01, 2.2590e-01, ..., 5.6050e-01,
2.8615e-01, -1.1212e+00],
[-5.2465e-01, -2.9097e+00, -2.8132e+00, ..., -2.3911e+00,
-2.7955e+00, -2.9528e+00]],
[[-1.3930e+00, -3.8604e+00, -6.7725e+00, ..., -1.8493e+00,
-2.8035e+00, -1.3840e+00],
[-4.9673e+00, -5.8487e+00, -5.4675e+00, ..., -4.5256e+00,
-4.2460e+00, -1.6218e+00],
[-3.8476e+00, -4.8319e+00, -4.1955e+00, ..., -4.4296e+00,
-1.2984e+00, 4.3527e-02],
...,
[-5.8400e+00, -6.1416e+00, -6.4995e+00, ..., -2.7037e+00,
-3.4479e+00, -2.6484e+00],
[-6.8072e+00, -7.9658e+00, -8.9451e+00, ..., -3.7212e+00,
-5.1531e+00, -5.1348e+00],
[-1.2854e+00, -3.8374e+00, -4.2575e+00, ..., -2.9956e+00,
-2.8830e+00, -1.7796e+00]],
[[ 7.5845e-01, 2.8799e+00, 1.9867e+00, ..., 1.5732e+00,
1.1904e+00, 9.9647e-01],
[ 1.2242e-01, 1.6999e+00, 1.4894e+00, ..., 6.2626e-01,
1.2706e+00, 1.0373e+00],
[ 7.1732e-01, 2.1452e+00, 9.2147e-01, ..., 1.1823e+00,
9.7909e-01, 6.2943e-01],
...,
[ 1.5212e+00, 1.2539e+00, 1.5518e+00, ..., 1.3798e+00,
9.2320e-01, 4.5780e-01],
[ 5.0838e-01, 9.8883e-01, 8.6727e-01, ..., 8.1993e-01,
1.0790e+00, 4.1954e-01],
[-2.5683e-01, 8.4533e-01, 8.5640e-01, ..., 1.0932e+00,
1.6619e+00, 4.7626e-01]],
...,
[[-9.8413e-01, -3.1306e+00, -2.6209e+00, ..., -9.3800e-01,
-1.6884e+00, -1.0117e-01],
[-2.9848e+00, -3.9825e+00, -1.2251e+00, ..., -2.2350e+00,
-3.4492e+00, -1.4496e+00],
[-2.1989e+00, -2.5650e+00, -1.3230e+00, ..., -1.7998e+00,
-1.1098e-01, 1.9430e-01],
...,
[-2.5400e+00, -1.9723e+00, -1.7202e+00, ..., -1.3424e+00,
-2.7839e+00, -8.0552e-01],
[-2.3595e+00, -3.4429e+00, -2.3667e+00, ..., -1.0289e+00,
-3.0844e+00, -1.4926e+00],
[-1.5495e+00, -3.0211e+00, -1.4619e+00, ..., -1.9639e+00,
-2.7977e+00, -7.4059e-01]],
[[-2.4203e+00, -2.8499e+00, -1.9332e+00, ..., -1.5689e-01,
-1.5736e+00, -3.5651e-01],
[-4.4074e+00, -3.2405e+00, -2.3214e+00, ..., -1.8134e+00,
-4.1019e+00, -1.7661e+00],
[-3.7920e+00, -2.3160e+00, -1.5627e+00, ..., -1.2266e+00,
-3.3690e+00, -1.9583e+00],
...,
[-4.9014e+00, -3.0660e+00, -1.7813e+00, ..., -9.2054e-01,
-3.2013e+00, -1.9754e+00],
[-4.2123e+00, -1.3903e+00, -4.1625e-01, ..., -1.3856e-01,
-3.4380e+00, -1.9062e+00],
[-3.3225e+00, -2.4886e+00, -1.3154e-01, ..., -5.4752e-01,
-2.8656e+00, -1.3129e+00]],
[[ 1.6175e+00, 1.2771e+00, 1.6752e+00, ..., 4.0550e-01,
4.2744e-01, 7.9259e-02],
[ 3.2264e+00, 3.1550e+00, 3.4266e+00, ..., 1.6909e+00,
-5.4992e-01, 2.0941e+00],
[ 1.3548e+00, 1.0508e+00, 2.1903e+00, ..., 5.0015e-01,
-1.4052e+00, 8.4599e-01],
...,
[ 2.7439e+00, 3.0507e+00, 3.1256e+00, ..., 1.4083e+00,
9.5035e-01, 7.0879e-01],
[ 3.0764e+00, 4.4905e+00, 3.6305e+00, ..., 1.7506e+00,
1.0347e+00, 1.3009e+00],
[ 2.1860e+00, 1.6136e+00, 1.3392e+00, ..., 7.1041e-01,
-4.3523e-02, 6.1320e-03]]]], device='cuda:0'),
tensor([[[[-2.0862e-01, 1.7262e+00, -3.0368e-01, ..., -4.6408e-01,
3.3362e-01, -2.2519e-01],
[ 7.6386e-02, 1.2410e+00, 6.9172e-01, ..., -7.0385e-01,
8.4117e-01, -4.5876e-01],
[-1.4031e+00, 7.8081e-01, 1.7358e-01, ..., -1.6862e+00,
3.9675e-01, -7.2167e-01],
...,
[-3.5461e-01, -1.8739e-01, -2.0243e-01, ..., -3.5458e-01,
1.7781e-01, -3.2660e-01],
[-1.3604e+00, -1.8974e-01, 3.7756e-01, ..., 6.2026e-02,
3.8200e-01, -1.0936e+00],
[-1.1641e+00, -2.5863e+00, -2.1997e+00, ..., -3.2132e+00,
-2.7677e+00, -2.6703e+00]],
[[-1.6063e+00, -2.3091e+00, -4.6515e+00, ..., -1.6654e+00,
-3.6245e+00, -2.2272e+00],
[-4.9657e+00, -5.4323e+00, -3.9846e+00, ..., -2.9484e+00,
-3.7266e+00, -3.3740e+00],
[-4.1832e+00, -6.1798e+00, -5.0387e+00, ..., -2.6115e+00,
-2.3379e+00, -2.6056e+00],
...,
[-4.3915e+00, -5.7746e+00, -4.5039e+00, ..., -3.1415e+00,
-3.1238e+00, -2.4471e+00],
[-7.4539e+00, -8.0412e+00, -6.4257e+00, ..., -4.5936e+00,
-4.9882e+00, -4.8946e+00],
[-1.3229e+00, -3.1894e+00, -2.0529e+00, ..., -3.0189e+00,
-2.7337e+00, -1.5787e+00]],
[[-2.1617e-01, 2.7695e+00, 1.1817e+00, ..., 1.0813e+00,
1.9022e+00, 9.4814e-01],
[-5.4411e-01, 3.5715e-01, 3.3860e-02, ..., 2.0394e-01,
1.6167e+00, 9.6389e-01],
[-2.8156e-02, 1.9559e+00, 1.8655e-01, ..., 1.1867e+00,
1.8340e+00, 1.4838e+00],
...,
[-5.9908e-01, 6.5178e-01, 7.3399e-01, ..., 1.5882e+00,
1.2119e+00, 6.3711e-01],
[-1.2430e+00, 3.3042e-01, 1.0607e+00, ..., 1.3409e+00,
1.0898e+00, 4.2125e-01],
[-4.3632e-01, -2.3208e-01, 4.7691e-01, ..., 1.7349e+00,
1.6640e+00, 5.6639e-01]],
...,
[[-1.3346e+00, -1.5797e+00, -2.8252e+00, ..., -2.5370e+00,
-2.1121e+00, -4.8292e-01],
[-3.5920e+00, -3.0686e+00, -2.9114e+00, ..., -4.3579e+00,
-3.2821e+00, -2.8781e+00],
[-2.8895e+00, -2.4269e+00, -3.2140e+00, ..., -4.5479e+00,
-2.2829e+00, -1.8575e+00],
...,
[-8.9213e-01, -2.4758e+00, -7.9549e-01, ..., -2.4628e+00,
-2.1424e+00, -8.1187e-01],
[-1.9867e+00, -3.9690e+00, -1.1514e+00, ..., -1.6715e+00,
-2.8244e+00, -1.5249e+00],
[-1.3070e+00, -1.8390e+00, -1.2069e+00, ..., -2.1712e+00,
-2.1805e+00, -5.7347e-01]],
[[-1.7306e+00, -1.7767e+00, -5.4670e+00, ..., -1.0510e+00,
-1.5341e+00, -6.4982e-01],
[-3.8958e+00, -1.9325e+00, -5.1423e+00, ..., -1.3901e+00,
-3.6471e+00, -1.8665e+00],
[-3.8437e+00, -2.4986e+00, -6.0977e+00, ..., -2.5895e+00,
-3.9504e+00, -2.4917e+00],
...,
[-2.8455e+00, -2.8817e+00, -3.4256e+00, ..., -9.4281e-01,
-3.3039e+00, -2.2158e+00],
[-1.9356e+00, -2.3516e-01, -1.4965e+00, ..., -2.6361e-01,
-3.1789e+00, -1.9126e+00],
[-2.0405e+00, -1.4110e+00, -1.4033e+00, ..., -1.1488e+00,
-2.6769e+00, -1.2119e+00]],
[[ 1.3041e+00, 4.4317e-01, 1.3768e+00, ..., -5.0263e-01,
1.3474e+00, 6.1641e-01],
[ 4.1259e+00, 2.9648e+00, 2.3424e+00, ..., 1.3365e-01,
1.2194e+00, 2.5750e+00],
[ 1.7096e+00, 6.4935e-01, 2.2479e+00, ..., -3.3022e-01,
2.9995e-01, 8.4009e-01],
...,
[ 2.8836e+00, 2.1188e+00, 1.5800e+00, ..., 7.7419e-01,
3.9196e-01, 5.1114e-01],
[ 3.7869e+00, 3.6839e+00, 2.7180e+00, ..., 1.4817e+00,
8.3341e-01, 1.2188e+00],
[ 2.3787e+00, 2.0790e+00, 1.5885e+00, ..., 5.0229e-02,
-2.8000e-01, -1.0481e-01]]],
[[[ 4.2838e-01, 6.1394e-01, -4.5346e-01, ..., -2.0465e+00,
-1.3706e+00, -1.1537e-01],
[ 3.7175e-01, -3.0009e-01, 5.4432e-01, ..., -1.9620e+00,
-3.2706e+00, -3.4680e-01],
[-3.9345e-01, -5.7836e-01, 9.4447e-01, ..., -3.0240e+00,
-1.9352e+00, -1.5877e+00],
...,
[-1.1406e+00, -2.0575e+00, -4.3700e+00, ..., -5.5203e+00,
-4.0570e+00, -5.5241e-01],
[-2.4092e+00, -8.9739e-01, -1.9293e+00, ..., -5.4412e+00,
-6.1085e+00, 3.7869e+00],
[-2.0190e+00, -3.8725e+00, -2.8736e+00, ..., 9.4044e-01,
7.0149e-01, 4.5050e+00]],
[[-1.7810e+00, -3.1588e+00, -4.2999e+00, ..., -5.3402e+00,
-6.6257e+00, -6.7225e-01],
[-5.0381e+00, -5.1074e+00, -3.6959e+00, ..., -1.1823e+01,
-1.2427e+01, -3.9799e+00],
[-4.6082e+00, -5.4702e+00, -5.6699e+00, ..., -4.1098e+00,
-7.4731e+00, -5.9420e+00],
...,
[-8.2122e+00, -7.2178e+00, -8.4263e+00, ..., -4.5219e+00,
-9.5084e+00, -5.0439e+00],
[-9.3461e+00, -9.3804e+00, -8.1524e+00, ..., -1.8909e+00,
-4.0819e+00, 2.2724e-01],
[-2.3701e+00, -4.1968e+00, -5.0813e+00, ..., -5.2285e-01,
-1.7314e+00, -1.1341e+00]],
[[ 7.2432e-02, 1.5475e+00, 1.7795e+00, ..., 1.0321e+00,
1.9538e+00, 7.2578e-01],
[ 1.0047e-02, 3.8639e-01, 1.2066e+00, ..., 1.9861e+00,
2.7941e+00, -7.6017e-01],
[ 5.6219e-01, 1.8378e+00, 1.6678e+00, ..., 2.3210e+00,
1.7575e+00, -6.7521e-01],
...,
[ 1.9567e-01, -5.0268e-01, 1.5010e+00, ..., -2.7782e+00,
2.0995e+00, -2.0065e+00],
[ 5.7859e-01, -6.7490e-01, -1.4970e+00, ..., -2.5023e+00,
5.2965e-01, 2.4807e+00],
[ 8.3471e-01, 9.4293e-01, -2.2467e-01, ..., -2.5035e+00,
-5.6051e+00, -6.2368e+00]],
...,
[[-1.0435e+00, -2.2771e+00, -2.8809e+00, ..., -4.2124e+00,
-4.4115e+00, -2.0025e-01],
[-3.1255e+00, -3.0001e+00, -1.8819e+00, ..., -5.9157e+00,
-5.6558e+00, -2.4836e+00],
[-2.0403e+00, -1.5245e+00, -1.4141e+00, ..., -3.0140e+00,
-3.6447e+00, -2.3434e+00],
...,
[-2.8546e+00, -3.4245e+00, -5.0512e+00, ..., -1.7699e+00,
-8.6246e+00, -4.5355e+00],
[-3.7884e+00, -2.5345e+00, -2.5249e+00, ..., -1.7663e+00,
-1.9399e+00, -4.4048e+00],
[-2.3584e+00, -2.4299e+00, -2.2726e+00, ..., -2.4151e+00,
-1.8889e+00, -4.1663e-01]],
[[-1.9880e+00, -2.2558e+00, -3.6895e+00, ..., -1.3021e+00,
5.6966e-02, -9.2579e-01],
[-4.8993e+00, -3.1686e+00, -3.4872e+00, ..., -5.0745e+00,
-3.5081e+00, -6.6751e-01],
[-5.3412e+00, -3.2207e+00, -2.7126e+00, ..., -1.6339e+00,
-1.6659e+00, -1.0599e+00],
...,
[-5.0239e+00, -5.6067e+00, -6.0896e+00, ..., -1.1204e+01,
-6.4872e+00, -2.8929e+00],
[-4.2776e+00, -5.1123e-02, -2.7614e+00, ..., -1.2695e+01,
-1.3131e+01, -4.3319e+00],
[-3.7886e+00, -1.2606e+00, -1.4514e-01, ..., -2.6625e+00,
-3.2517e+00, -2.1610e+00]],
[[ 1.5107e+00, 1.7817e-01, 1.2201e+00, ..., 2.5690e-01,
6.4900e-01, -1.5233e+00],
[ 2.8749e+00, 1.3678e+00, 1.3662e+00, ..., 5.0728e+00,
3.2409e+00, 1.9420e+00],
[ 1.7267e+00, 4.3034e-01, 1.4463e+00, ..., 9.7541e-01,
1.0534e+00, 1.2512e+00],
...,
[ 3.4427e+00, 2.4381e+00, 2.5364e+00, ..., -1.5965e+01,
-8.9764e+00, -1.1192e+00],
[ 3.4192e+00, 4.8940e+00, 2.8584e+00, ..., -9.9371e+00,
-1.2999e+01, -1.8275e+00],
[ 1.8974e+00, 2.4134e+00, 7.0918e-01, ..., -2.3060e+00,
-1.1723e+00, 1.8582e+00]]],
[[[ 1.5397e+00, -1.4236e+00, -8.8919e+00, ..., -1.8890e+00,
1.3229e+00, 1.4173e+00],
[-1.0596e+00, -5.3177e+00, -3.7805e+00, ..., -3.4768e+00,
-1.5018e+00, 4.0313e-01],
[-1.1629e+00, -6.8748e+00, -5.4478e+00, ..., -7.2165e+00,
-3.3571e+00, -5.5144e-01],
...,
[-7.8145e-01, -5.0033e+00, 4.7915e-01, ..., -4.8664e+00,
-4.1942e+00, -1.6866e+00],
[ 2.4114e-02, -8.6226e-03, -9.4367e-01, ..., -5.1466e+00,
-2.7632e+00, -1.1085e+00],
[-2.2804e+00, -4.8872e+00, -4.1184e+00, ..., -5.6084e+00,
-4.7730e+00, -3.0972e+00]],
[[ 1.4392e+00, -8.1050e+00, -1.0349e+01, ..., -6.2239e+00,
-4.5708e+00, 1.3276e-01],
[-6.7961e+00, -1.2907e+01, -4.4778e+00, ..., -1.0953e+01,
-1.1817e+01, -3.8363e+00],
[-2.4749e+00, -9.2862e+00, -2.8381e+00, ..., -6.5861e+00,
-9.0945e+00, -3.2632e+00],
...,
[-4.9880e+00, -8.1392e+00, -1.0489e+01, ..., -1.0059e+01,
-9.7547e+00, -3.1232e+00],
[-7.6680e+00, -7.7442e+00, -4.8062e+00, ..., -8.8893e+00,
-8.1159e+00, -5.6488e+00],
[-2.1352e+00, -5.6359e+00, -4.9086e+00, ..., -4.3774e+00,
-3.2951e+00, -2.2946e+00]],
[[-6.8509e-01, -9.0780e-02, 2.8951e+00, ..., 1.1583e+00,
1.6773e+00, 3.6275e-01],
[-2.6478e-01, 1.0905e+00, 9.7937e-01, ..., 5.5086e-01,
-6.1001e-01, -1.7018e+00],
[-7.6812e-01, 2.7467e+00, -4.1180e+00, ..., 6.7556e+00,
2.6789e+00, -5.8556e-02],
...,
[-1.1819e-01, 1.4544e+00, -2.2819e+00, ..., 5.5606e+00,
1.1751e+00, -6.4368e-01],
[-9.7674e-01, -3.5787e-01, 1.1639e-01, ..., -2.0726e+00,
-3.2399e+00, -6.2322e-01],
[ 1.2867e+00, 1.0303e+00, -7.2609e-01, ..., -1.7144e+00,
-2.9874e-01, -3.4726e-01]],
...,
[[ 9.1922e-02, -3.0061e+00, -8.2445e+00, ..., -4.9510e+00,
-2.0822e+00, 3.0542e-01],
[-5.8299e+00, -6.8634e+00, -9.9729e-02, ..., -4.2955e+00,
-6.4493e+00, -4.6747e+00],
[ 2.4105e-02, -2.2114e+00, 2.1809e+00, ..., -9.9702e+00,
-6.7198e+00, -3.3963e+00],
...,
[-2.9078e+00, -3.2863e+00, -4.0353e+00, ..., -6.3851e+00,
-4.0114e+00, -2.6325e+00],
[-2.2998e+00, -2.4424e+00, 6.6626e-01, ..., -5.9665e+00,
-5.1547e+00, -2.5864e+00],
[-3.8088e+00, -4.8274e+00, -2.9321e+00, ..., -5.5453e+00,
-4.9444e+00, -1.6414e+00]],
[[-6.2978e-01, -3.7648e+00, -5.6624e+00, ..., -1.1684e+00,
-1.3610e+00, -1.7656e+00],
[-4.1682e+00, -2.6159e+00, -7.8656e-01, ..., -6.0609e-01,
-1.6427e+00, -1.8014e+00],
[-2.7489e+00, -5.8293e+00, -7.1857e+00, ..., 3.0050e-01,
-5.1207e-01, -1.4341e+00],
...,
[-4.3334e+00, -2.9731e+00, -2.7766e+00, ..., -6.4102e+00,
-2.3602e+00, 7.9913e-01],
[-3.0247e+00, 1.2593e+00, 7.1171e-01, ..., -3.8351e+00,
-7.0100e-01, 1.5011e+00],
[-4.1611e+00, -1.8951e+00, -1.1775e-01, ..., -8.7195e-01,
-2.0696e+00, -2.5249e-01]],
[[ 9.0555e-01, 2.2743e+00, 2.4612e+00, ..., -8.1637e-01,
1.8323e-01, -6.2219e-01],
[ 2.8751e+00, 5.6325e+00, 3.2246e+00, ..., 2.0535e+00,
1.7261e+00, 1.7704e+00],
[ 3.6902e+00, 8.7962e-01, 5.6402e-01, ..., 2.6603e+00,
1.0835e+00, 7.3284e-02],
...,
[ 3.6142e+00, 3.1339e+00, 8.2479e+00, ..., 4.5858e+00,
3.6223e+00, 2.7846e+00],
[ 4.4961e+00, 4.4196e+00, 2.5209e+00, ..., 3.4524e+00,
3.4222e+00, 2.6690e+00],
[ 2.1680e+00, 6.0789e-02, 3.2024e-01, ..., 2.2813e+00,
1.0109e+00, -9.3992e-02]]],
[[[ 7.0152e-02, 1.0247e+00, 6.1096e-01, ..., -5.9201e-01,
1.1688e-01, 6.6593e-01],
[ 4.9048e-01, 1.0044e+00, 2.2643e+00, ..., -2.1774e+00,
-1.5186e+00, 9.9957e-01],
[-6.5334e-01, 2.0302e-02, -3.0334e-01, ..., -1.7352e+00,
-1.9075e+00, 1.2160e-01],
...,
[-1.0968e+00, -1.9145e+00, 7.3359e-01, ..., -3.1316e+00,
-2.8834e+00, -2.2166e+00],
[-1.0693e+00, -1.1415e+00, -4.6520e-02, ..., -1.6014e+00,
-2.4230e+00, -2.5891e+00],
[-2.6978e+00, -5.2028e+00, -4.4636e+00, ..., -5.1437e+00,
-4.2764e+00, -3.7898e+00]],
[[-8.1611e-01, -2.2967e+00, -3.9938e+00, ..., -3.3751e+00,
-4.7036e+00, -3.5163e-01],
[-4.8210e+00, -6.4939e+00, -7.3360e+00, ..., -7.7236e+00,
-7.8829e+00, -1.7258e+00],
[-3.9596e+00, -5.1159e+00, -8.7745e+00, ..., -5.7632e+00,
-8.0864e+00, -2.1409e+00],
...,
[-5.7605e+00, -5.5992e+00, -3.8798e+00, ..., -9.1453e+00,
-7.8983e+00, -4.4891e+00],
[-7.1449e+00, -7.8010e+00, -5.9906e+00, ..., -7.8179e+00,
-9.2799e+00, -6.7309e+00],
[-2.5311e+00, -4.5037e+00, -4.1280e+00, ..., -4.2765e+00,
-4.3786e+00, -2.6507e+00]],
[[-3.4660e-01, 2.1403e+00, 1.2379e+00, ..., 9.4206e-01,
2.0528e+00, 9.1311e-01],
[-1.6328e-01, 1.2177e+00, 6.6473e-01, ..., 5.8448e-01,
1.2610e+00, -5.9481e-01],
[ 1.4861e-02, 1.3230e+00, 1.0409e+00, ..., 2.7856e-01,
1.2827e+00, 9.0870e-02],
...,
[ 8.4102e-01, 1.2906e+00, 9.7730e-01, ..., 4.9870e-01,
-1.6881e-01, -6.8458e-01],
[ 8.2852e-01, 7.1512e-01, 8.3622e-01, ..., 1.1362e+00,
-5.6510e-01, 5.1037e-02],
[ 1.3317e+00, 1.5739e+00, 1.1990e+00, ..., 1.2046e+00,
1.2804e+00, 3.4459e-01]],
...,
[[-7.5988e-01, -1.6675e+00, -1.6755e+00, ..., -2.5325e+00,
-2.5782e+00, -3.1191e-01],
[-2.8915e+00, -3.9386e+00, -2.7391e+00, ..., -5.0261e+00,
-4.9794e+00, -2.4796e+00],
[-1.3471e+00, -2.6724e+00, -2.8851e+00, ..., -1.6358e+00,
-4.7717e+00, -2.1624e+00],
...,
[-2.7719e+00, -2.8354e+00, -8.2949e-01, ..., -4.3823e+00,
-4.4606e+00, -2.1921e+00],
[-2.9686e+00, -3.1144e+00, -1.7244e+00, ..., -2.3885e+00,
-5.4528e+00, -3.1957e+00],
[-3.4807e+00, -3.8922e+00, -3.3925e+00, ..., -3.8228e+00,
-4.2878e+00, -1.7861e+00]],
[[-1.6504e+00, -2.2519e+00, -2.5561e+00, ..., -1.3869e+00,
-1.3934e+00, -1.4321e+00],
[-4.2316e+00, -3.3392e+00, -3.3914e+00, ..., -3.5131e+00,
-2.4984e+00, -1.1060e+00],
[-3.4982e+00, -3.0004e+00, -3.0215e+00, ..., 1.0744e+00,
-1.3624e+00, -8.5336e-01],
...,
[-3.8938e+00, -1.9288e+00, -2.7539e+00, ..., -3.1400e+00,
-5.2950e+00, -9.6674e-01],
[-3.7174e+00, -6.9771e-01, -2.0253e+00, ..., -1.7430e+00,
-3.4841e+00, -4.6998e-01],
[-4.4676e+00, -3.0343e+00, -2.8371e+00, ..., -1.7733e+00,
-2.9563e+00, -8.2696e-01]],
[[ 9.4741e-01, 4.8772e-01, 6.8067e-01, ..., -7.0544e-01,
1.1731e-01, -4.3993e-01],
[ 3.4432e+00, 3.1436e+00, 3.5259e+00, ..., 1.4260e+00,
2.0184e+00, 2.4234e+00],
[ 1.5768e+00, 1.1297e+00, 1.8855e+00, ..., 1.2383e+00,
1.5558e+00, 1.2840e+00],
...,
[ 2.8249e+00, 1.4392e+00, 1.6368e+00, ..., 3.7397e+00,
2.3594e+00, 2.9553e+00],
[ 3.7425e+00, 3.5640e+00, 2.5514e+00, ..., 2.7788e+00,
3.0854e+00, 2.5198e+00],
[ 1.9324e+00, 7.4924e-01, 1.3925e-01, ..., -1.1487e-01,
-5.5789e-01, -2.7839e-01]]],
[[[ 1.2584e-01, -1.0315e+00, -7.8698e-01, ..., 7.1795e-01,
-2.0531e-01, 9.0407e-01],
[ 1.6295e-01, -3.3285e+00, -2.3927e+00, ..., -1.1762e+00,
-1.3528e+00, -3.1742e-01],
[ 9.9123e-02, -1.3963e+00, -5.5076e-01, ..., -4.1495e-01,
-5.1373e-02, -7.3812e-01],
...,
[-9.5471e-01, -2.9078e-01, -5.3980e-01, ..., -2.7921e+00,
-3.6472e+00, -8.7622e-01],
[-2.5537e+00, -1.3765e+00, -1.1458e-02, ..., -2.7336e+00,
-2.9999e+00, -1.4999e-02],
[-2.2949e+00, -4.0051e+00, -3.6133e+00, ..., -4.1470e+00,
-3.4084e+00, -1.4740e+00]],
[[-2.0013e+00, -4.3205e+00, -2.4958e+00, ..., -3.1787e+00,
-5.5954e+00, -1.0197e+00],
[-5.1539e+00, -7.5113e+00, -5.8929e+00, ..., -7.1931e+00,
-7.3788e+00, -4.0395e+00],
[-4.3821e+00, -5.4352e+00, -5.7228e+00, ..., -3.9243e+00,
-5.1828e+00, -4.0074e+00],
...,
[-6.7278e+00, -9.6399e+00, -1.1177e+01, ..., -8.0392e+00,
-9.3675e+00, -2.3480e+00],
[-1.0094e+01, -9.5474e+00, -8.0466e+00, ..., -1.0711e+01,
-1.0936e+01, -3.8750e+00],
[-2.4383e+00, -4.8937e+00, -4.5396e+00, ..., -4.1386e+00,
-4.0980e+00, -5.8612e-01]],
[[-2.9970e-01, 6.6954e-02, -1.3838e+00, ..., 1.4448e+00,
1.8671e+00, 1.1210e-01],
[ 2.7259e-01, -9.7337e-01, -6.1228e+00, ..., 1.1749e+00,
2.0433e+00, -7.7530e-01],
[ 6.8323e-01, 1.5412e+00, -2.1382e+00, ..., 9.9087e-01,
2.2938e+00, 1.0240e+00],
...,
[ 5.4967e-01, 4.9754e-02, -6.0097e-01, ..., 1.2752e+00,
7.8845e-01, -9.1575e-01],
[-9.7083e-01, -7.8833e-02, 1.7008e-01, ..., 2.1536e-01,
7.9692e-01, -4.9372e-01],
[ 5.9063e-01, 1.2396e+00, 6.9447e-01, ..., -3.3771e-01,
-5.9296e-01, -1.9592e+00]],
...,
[[-1.4285e+00, -2.4207e+00, -1.4173e+00, ..., -1.3179e+00,
-2.6628e+00, 3.3405e-01],
[-4.2178e+00, -5.7026e+00, -4.1257e+00, ..., -4.6224e+00,
-5.9256e+00, -3.1846e+00],
[-2.3066e+00, -2.7646e+00, -2.6900e+00, ..., -1.8708e+00,
-4.4017e+00, -1.8426e+00],
...,
[-4.1314e+00, -6.0008e+00, -7.5804e+00, ..., -4.3250e+00,
-6.2452e+00, -2.9447e+00],
[-3.3751e+00, -4.5530e+00, -2.1027e+00, ..., -3.9395e+00,
-5.4206e+00, -1.3535e+00],
[-3.3164e+00, -4.0990e+00, -3.6548e+00, ..., -2.6361e+00,
-2.5531e+00, 4.3220e-01]],
[[-1.3917e+00, -3.7615e-01, -2.6252e-01, ..., -2.6897e-01,
2.5331e-01, 2.2534e-01],
[-4.8384e+00, -2.0420e+00, 4.1442e-01, ..., -8.1885e-01,
-1.7711e+00, 2.8025e-01],
[-4.3384e+00, -2.7034e+00, 3.8825e-03, ..., 3.9663e-02,
-2.4308e+00, -8.3405e-01],
...,
[-3.9314e+00, -2.6480e+00, -5.9043e+00, ..., -3.1688e+00,
-2.5638e+00, -8.1202e-01],
[-3.6612e+00, 1.6731e-02, -1.7892e+00, ..., -1.8926e+00,
-1.1241e+00, 3.1766e-01],
[-3.6882e+00, -2.8352e+00, -2.6267e+00, ..., -1.6287e+00,
-1.1361e+00, 3.6046e-01]],
[[ 1.7798e+00, 7.1238e-01, 7.4558e-02, ..., 3.6319e-01,
1.5955e+00, 2.4236e-01],
[ 3.1436e+00, 2.6042e+00, 2.3969e+00, ..., 1.6666e+00,
2.7436e+00, 3.5658e+00],
[ 2.1335e+00, 8.9093e-02, 2.6809e+00, ..., 2.4884e+00,
2.8433e+00, 2.1260e+00],
...,
[ 3.4696e+00, 6.0940e+00, 5.7508e+00, ..., 1.5804e+00,
1.2941e+00, 3.2059e-01],
[ 4.0372e+00, 3.7273e+00, 1.1619e+00, ..., 3.0512e+00,
1.9709e+00, 1.7319e+00],
[ 2.7993e+00, 3.2793e+00, 2.6271e+00, ..., 5.1893e-01,
6.1774e-01, 1.0419e+00]]]], device='cuda:0'),
tensor([[[[ 1.4606e-01, 1.3793e+00, 5.0049e-01, ..., -1.8276e+00,
-2.7935e+00, -1.9336e+00],
[ 4.8330e-01, 5.1958e-01, 1.3026e+00, ..., 4.2957e+00,
-7.6570e-01, 4.8783e-01],
[ 1.3094e+00, 8.9250e-01, 4.9706e-01, ..., 7.3459e-01,
-2.4248e+00, 1.2121e+00],
...,
[-1.6373e+00, 4.2029e-01, 4.0858e-01, ..., -1.6773e+00,
7.7850e-01, 1.7286e+00],
[-3.3804e+00, 1.5115e-01, -1.1950e+00, ..., 5.1193e-01,
-1.6061e-01, 1.0114e+00],
[-2.7524e+00, -2.6274e+00, -4.4217e-01, ..., -2.5558e-01,
-1.4989e+00, -6.0733e-01]],
[[-9.0961e-01, -2.3274e+00, -4.0993e+00, ..., -4.7392e+00,
-5.2287e+00, -1.5096e-01],
[-3.7172e+00, -6.1404e+00, -5.9166e+00, ..., -2.1303e+00,
-5.0200e+00, 1.3380e-01],
[-3.3360e+00, -7.1943e+00, -8.4565e+00, ..., -2.4858e+00,
-4.6005e+00, 8.5537e-01],
...,
[-5.4780e+00, -1.1517e+00, -1.1167e+00, ..., -1.0839e+00,
-6.4080e-01, 9.2946e-01],
[-7.7204e+00, -4.4297e+00, 3.5262e-01, ..., 1.6460e+00,
-1.7224e+00, -1.4197e+00],
[-1.7661e+00, -4.4679e+00, -3.2321e+00, ..., -6.1492e-02,
-1.7228e+00, -5.0809e-01]],
[[-3.3847e-01, 2.0571e+00, 1.1980e+00, ..., 2.0861e+00,
3.3018e+00, 2.0403e+00],
[-3.6982e-02, 9.5204e-01, 6.7633e-01, ..., -2.4023e+00,
-2.8808e+00, 1.4399e+00],
[ 4.3214e-01, 1.9983e+00, 3.1645e-01, ..., -5.5861e+00,
-1.2391e+01, -5.1996e+00],
...,
[-1.7566e+00, -2.3519e+00, -2.9451e+00, ..., -1.0210e+00,
-9.3425e-01, 3.9371e-01],
[-5.0406e-01, -5.5604e+00, -3.3797e+00, ..., -6.7930e-01,
-1.3661e+00, -2.5474e+00],
[ 9.8609e-01, -3.1708e+00, -1.6208e+00, ..., -1.7691e-01,
2.6268e-01, -1.2440e+00]],
...,
[[-9.0930e-01, -1.8256e+00, -1.8838e+00, ..., -1.5587e+00,
-2.2888e+00, -1.2686e+00],
[-3.2555e+00, -4.1656e+00, -3.0619e+00, ..., 1.3072e+00,
1.1268e+00, -2.1818e+00],
[-8.6942e-01, -3.6021e+00, -2.9075e+00, ..., -1.7787e+00,
2.9710e+00, -1.3607e-01],
...,
[-2.9081e+00, -4.1763e+00, -3.2243e+00, ..., 7.4663e-01,
-3.8503e-01, -3.5080e-01],
[-3.3066e+00, -3.9724e+00, -3.1946e+00, ..., 1.7597e+00,
-1.4205e+00, -8.4915e-02],
[-2.8594e+00, -3.3332e+00, -3.2831e+00, ..., -4.9526e-01,
-2.2423e+00, 3.0498e-01]],
[[-1.7454e+00, -2.1137e+00, -3.0872e+00, ..., -3.4602e+00,
-4.8013e+00, -1.8900e-01],
[-4.2907e+00, -3.2313e+00, -2.7444e+00, ..., -2.1855e+00,
-2.1574e+00, -3.1206e+00],
[-3.7095e+00, -3.0328e+00, -3.9874e-01, ..., -3.6424e+00,
9.2671e-01, 1.6285e+00],
...,
[ 1.1373e-01, -3.0669e+00, -4.4197e+00, ..., 9.1938e-02,
-1.0180e+00, -1.7084e+00],
[-9.2413e-01, 1.2581e+00, -2.4206e+00, ..., -8.5741e-01,
-1.6595e+00, -5.6587e-01],
[-2.3327e+00, 1.1836e+00, -1.0715e+00, ..., -2.3360e+00,
-2.7295e+00, -5.4209e-01]],
[[ 1.0786e+00, 5.8090e-01, 6.0480e-01, ..., 2.1367e-01,
-1.8423e-01, 6.2618e-01],
[ 3.1705e+00, 3.6250e+00, 3.0580e+00, ..., 4.9487e+00,
-4.6819e-02, 3.5047e+00],
[ 1.1896e+00, 1.9571e+00, 1.2970e+00, ..., -3.8390e+00,
-4.5595e+00, -5.4700e-02],
...,
[ 3.0990e+00, -4.4115e-01, -1.8365e+00, ..., -2.5390e+00,
-3.0543e-01, 4.1086e-01],
[ 3.6950e+00, 2.9502e+00, -4.2989e+00, ..., -1.1718e+00,
-1.2005e+00, 6.7475e-01],
[ 1.1704e+00, 2.8037e+00, 5.4512e-01, ..., -1.0945e+00,
-1.7937e+00, -3.3513e-01]]],
[[[ 1.1734e+00, 1.0228e+00, -8.3311e-01, ..., 2.1020e-01,
2.4371e-01, -1.8989e-01],
[ 2.0976e+00, -4.1489e-01, -3.5584e+00, ..., 7.4372e-01,
1.6136e-01, -1.0290e+00],
[-2.1238e-01, -2.5308e+00, -2.2760e+00, ..., -3.9689e-01,
-3.6517e+00, -2.1183e+00],
...,
[-1.8081e+00, -3.4489e+00, 7.2494e-01, ..., -1.0316e+00,
-1.1950e+00, -1.1060e+00],
[-4.1822e-01, 1.6548e+00, 3.7509e+00, ..., -2.5901e-01,
-6.6304e-01, -1.5484e+00],
[-5.3934e-01, -2.9117e+00, -3.6309e+00, ..., -2.6391e+00,
-3.0267e+00, -3.0544e+00]],
[[-1.4212e+00, -4.4939e+00, -7.8546e+00, ..., -2.9769e+00,
-4.0864e+00, -2.5685e+00],
[-4.8082e+00, -8.3529e+00, -1.0285e+01, ..., -3.6046e+00,
-3.9346e+00, -3.2342e+00],
[-1.8347e+00, -5.1417e+00, -4.0757e+00, ..., -4.5713e+00,
-5.4437e+00, -2.7967e+00],
...,
[-5.6483e-01, -1.5215e+00, -7.5060e-01, ..., -4.0535e+00,
-5.1649e+00, -4.6591e+00],
[-4.9877e+00, -4.7009e+00, -6.7985e+00, ..., -4.7717e+00,
-5.4484e+00, -5.3441e+00],
[-6.8139e-01, -4.2726e+00, -4.2078e+00, ..., -3.1563e+00,
-3.2722e+00, -1.8398e+00]],
[[ 6.8981e-01, 2.4617e+00, 1.2465e+00, ..., 1.8555e+00,
1.5997e+00, 8.2901e-01],
[-4.4403e-01, 1.0436e+00, 2.6802e+00, ..., 1.3609e+00,
8.0338e-01, 6.9120e-01],
[ 4.9657e-01, 2.7636e+00, 1.8853e+00, ..., 1.3858e+00,
2.8099e+00, 2.1934e+00],
...,
[-1.1700e+00, 5.9988e-01, -1.6461e+00, ..., 3.8695e-02,
1.0469e+00, 1.5029e+00],
[-1.9078e+00, -1.1632e+00, -1.9884e+00, ..., 2.6540e-01,
1.8777e-01, 8.7483e-01],
[-7.3660e-01, -7.7634e-01, -9.8287e-01, ..., 1.3380e+00,
1.4474e+00, 2.7953e-01]],
...,
[[-2.4462e-01, -3.0907e+00, -3.3991e+00, ..., -1.5181e+00,
-2.3173e+00, -5.1323e-01],
[-2.6921e+00, -5.4924e+00, -3.4387e+00, ..., -1.8012e+00,
-3.8759e+00, -2.7833e+00],
[-2.5199e+00, -4.1754e+00, -3.4101e-01, ..., -2.6797e+00,
-4.6356e+00, -1.8455e+00],
...,
[-6.3488e-01, -2.5130e+00, -1.6065e+00, ..., -2.2442e+00,
-4.2639e+00, -3.5602e+00],
[-7.1795e-01, -2.9670e+00, -1.6801e+00, ..., -1.3082e+00,
-3.6244e+00, -2.0427e+00],
[-2.3352e+00, -5.0888e+00, -2.7961e+00, ..., -2.2478e+00,
-3.0713e+00, -8.3795e-01]],
[[-2.3496e+00, -2.7536e+00, -2.9210e+00, ..., 1.8484e-01,
-1.3648e+00, -6.8693e-01],
[-3.8165e+00, -4.5366e+00, -2.7964e+00, ..., -8.6261e-01,
-3.5453e+00, -1.7570e+00],
[-3.0034e+00, -1.3512e+00, 2.0853e+00, ..., -3.3825e+00,
-5.9294e+00, -3.5520e+00],
...,
[-3.6329e+00, -4.3841e+00, -2.3740e-01, ..., -2.1987e+00,
-3.9518e+00, -1.2554e+00],
[-1.3367e+00, -5.5842e-01, 3.7357e-01, ..., -2.5779e-01,
-3.5863e+00, -1.6365e+00],
[-1.3922e+00, -1.2995e+00, 1.8039e-01, ..., -6.9592e-01,
-2.3929e+00, -1.0237e+00]],
[[ 1.4609e+00, 1.5324e+00, 1.8039e+00, ..., 8.8236e-01,
1.2457e+00, 6.2540e-01],
[ 3.3789e+00, 3.8108e+00, 1.9195e+00, ..., 2.1302e+00,
1.1285e+00, 1.8621e+00],
[ 6.9375e-01, 3.5615e-02, -5.4983e-01, ..., 1.5505e+00,
-7.5429e-01, 4.1877e-01],
...,
[-1.2605e+00, -4.6610e+00, -1.4055e+00, ..., 1.5989e+00,
2.2366e+00, 2.6469e+00],
[ 3.5745e+00, 3.1499e+00, 3.3681e+00, ..., 2.0469e+00,
9.1996e-01, 1.4799e+00],
[ 2.4607e+00, 1.9452e+00, 8.9486e-01, ..., 6.1439e-01,
-2.3480e-01, 2.8064e-01]]],
[[[-1.0357e-01, 6.2854e-01, 9.1933e-02, ..., -1.4331e+00,
-8.3836e-01, -7.8168e-01],
[ 8.1438e-01, 6.2677e-01, 1.5284e+00, ..., -1.8444e-01,
5.8085e-01, -1.0413e+00],
[ 1.0653e+00, 3.5148e-01, 1.3673e+00, ..., -8.8011e-02,
-7.3055e-01, -2.1013e+00],
...,
[-1.2848e+00, -3.3998e+00, -1.7433e+00, ..., -1.7792e+00,
-2.7012e+00, -8.8803e-02],
[-2.2183e+00, -1.4035e+00, 5.5638e-01, ..., -2.5636e+00,
-2.8765e+00, 1.6801e-01],
[-2.1424e+00, -3.4052e+00, -2.6268e+00, ..., -4.1303e+00,
-3.5665e+00, -1.5054e+00]],
[[-6.8259e-01, -1.3488e+00, -2.1896e+00, ..., -2.9490e+00,
-4.3553e+00, -2.4237e+00],
[-1.9329e+00, -3.5334e+00, -3.9533e+00, ..., -3.0744e+00,
-4.5137e+00, -4.6337e+00],
[-1.8020e+00, -3.4799e+00, -3.9535e+00, ..., -3.0691e+00,
-4.3951e+00, -3.7225e+00],
...,
[-6.5614e+00, -9.0685e+00, -7.8902e+00, ..., -7.9608e+00,
-8.9229e+00, -1.9730e+00],
[-9.9236e+00, -9.9846e+00, -9.1245e+00, ..., -1.0531e+01,
-1.0771e+01, -3.7068e+00],
[-2.6591e+00, -4.9817e+00, -4.2854e+00, ..., -4.3365e+00,
-4.2449e+00, -6.4268e-01]],
[[ 6.8059e-01, 1.0752e+00, 1.1520e+00, ..., 1.8148e+00,
2.0078e+00, 8.3590e-01],
[ 7.7824e-01, 1.3387e+00, 1.4085e+00, ..., 1.4995e+00,
2.2470e+00, 1.0064e+00],
[ 5.0012e-01, 3.3366e+00, 1.6453e+00, ..., 1.6850e+00,
2.2564e+00, 2.1242e+00],
...,
[ 2.4843e-01, -1.3432e+00, -6.3471e+00, ..., 1.4977e+00,
1.4120e+00, -2.4940e-01],
[-1.0259e+00, -4.8284e-01, -8.3892e-01, ..., 4.5147e-01,
9.7081e-01, -3.8886e-01],
[ 4.6265e-01, 7.3029e-01, -5.7405e-02, ..., 1.0982e-01,
-5.8143e-01, -1.9816e+00]],
...,
[[-9.6076e-01, -1.0060e+00, -1.3695e+00, ..., -2.7331e+00,
-3.1378e+00, -1.2151e+00],
[-2.2894e+00, -2.7195e+00, -1.4610e+00, ..., -2.7271e+00,
-4.4167e+00, -3.5174e+00],
[-9.3849e-01, -1.2592e+00, -2.4636e-01, ..., -2.3225e+00,
-4.8148e+00, -3.1273e+00],
...,
[-3.3593e+00, -4.8474e+00, -3.9860e+00, ..., -5.4598e+00,
-6.1949e+00, -2.6322e+00],
[-3.7613e+00, -6.4038e+00, -5.2826e+00, ..., -4.4819e+00,
-5.0875e+00, -1.2160e+00],
[-3.3666e+00, -3.5254e+00, -2.8827e+00, ..., -2.8817e+00,
-2.5591e+00, 3.8055e-01]],
[[-1.5773e+00, -2.2409e+00, -3.7057e+00, ..., -2.7923e+00,
-2.7548e+00, -5.1100e-01],
[-4.3963e+00, -4.3394e+00, -5.4053e+00, ..., -3.5078e+00,
-5.7377e+00, -2.0036e+00],
[-4.0440e+00, -4.8367e+00, -6.0330e+00, ..., -2.0231e+00,
-4.9582e+00, -2.1252e+00],
...,
[-4.4274e+00, -1.7230e+00, 1.3336e+00, ..., -4.1060e+00,
-2.9255e+00, -1.2189e+00],
[-3.3117e+00, -1.2289e-01, -4.4879e-01, ..., -2.3060e+00,
-1.2057e+00, 3.4362e-01],
[-3.6698e+00, -2.7075e+00, -2.0245e+00, ..., -1.5315e+00,
-1.1252e+00, 3.5339e-01]],
[[ 8.6429e-01, -2.3655e-01, 4.8255e-01, ..., 4.7260e-01,
1.3191e+00, 6.4064e-01],
[ 1.7819e+00, 6.9743e-01, 7.1092e-01, ..., 7.3450e-01,
1.7688e+00, 3.1787e+00],
[ 1.4655e+00, 1.8959e-01, 1.4892e+00, ..., 9.0074e-01,
7.3080e-01, 6.6103e-01],
...,
[ 3.4967e+00, 2.4540e+00, 3.3894e+00, ..., 2.9591e+00,
2.7668e+00, 6.5992e-01],
[ 4.2540e+00, 3.7280e+00, 3.4162e+00, ..., 2.7979e+00,
1.9051e+00, 1.5177e+00],
[ 2.8799e+00, 3.1825e+00, 1.8735e+00, ..., 3.1182e-01,
5.5113e-01, 1.0273e+00]]],
[[[ 5.4525e-02, 1.1224e+00, 4.7413e-01, ..., -2.9162e+00,
-2.2683e+00, -9.5560e-01],
[-1.2702e-01, -8.3308e-02, 1.0684e+00, ..., -1.6933e+00,
-2.4481e+00, -2.0010e-01],
[-1.7833e+00, -7.0916e-01, -1.9246e+00, ..., -1.2551e+00,
-2.3372e+00, -7.8398e-02],
...,
[ 7.9891e-01, 3.6596e-02, -2.0360e+00, ..., -2.0154e+00,
1.8403e+00, 2.8539e-01],
[-1.7884e+00, -1.3919e+00, -2.3583e+00, ..., -2.1283e+00,
-2.8810e+00, -8.2218e-01],
[-1.0382e+00, -3.0195e+00, -3.7489e+00, ..., -2.1768e+00,
-2.5104e+00, -1.8433e+00]],
[[-1.6131e+00, -2.3256e+00, -4.0859e+00, ..., -6.1449e+00,
-7.1389e+00, -1.2591e+00],
[-3.1758e+00, -3.7318e+00, -3.3401e+00, ..., -5.9787e+00,
-8.4835e+00, -2.3886e+00],
[-3.1530e+00, -3.8363e+00, -6.0871e+00, ..., -8.8929e+00,
-9.3912e+00, -3.4161e+00],
...,
[-6.0434e+00, -9.2349e+00, -1.2542e+01, ..., -1.1174e+00,
-6.9937e+00, -3.8822e+00],
[-6.3702e+00, -5.7787e+00, -7.1209e+00, ..., -9.6277e+00,
-1.0737e+01, -6.4375e+00],
[-2.1052e+00, -3.9697e+00, -4.3060e+00, ..., -2.9090e+00,
-3.7413e+00, -1.0230e+00]],
[[ 1.6750e-01, 1.4676e+00, 2.0286e+00, ..., 1.5082e+00,
2.3627e+00, 9.1392e-01],
[-1.1446e-01, 3.5296e-01, 1.0576e+00, ..., 1.5194e+00,
3.0617e+00, -3.5153e-03],
[ 9.3277e-01, 2.9067e+00, 3.7424e+00, ..., 5.1520e-01,
2.0887e+00, -4.8292e-01],
...,
[-8.9748e-01, 2.1019e-01, -8.0835e-02, ..., 1.1443e+00,
-1.9006e-01, -1.1699e+00],
[ 8.8230e-02, -7.0296e-01, 5.3710e-01, ..., -1.4599e+00,
1.0540e-01, -8.3342e-01],
[ 8.1681e-01, 3.5943e-01, -5.9687e-01, ..., -1.1513e+00,
-1.5348e+00, -2.2875e+00]],
...,
[[-9.8597e-01, -1.8420e+00, -2.5986e+00, ..., -4.9057e+00,
-5.0358e+00, -6.4469e-01],
[-2.9717e+00, -2.2251e+00, -1.2289e+00, ..., -3.7336e+00,
-4.9894e+00, -1.8480e+00],
[-2.1710e+00, -8.2767e-01, -1.8965e+00, ..., -5.7649e+00,
-5.7265e+00, -1.8517e+00],
...,
[-3.1326e+00, -6.2003e+00, -6.4118e+00, ..., -3.1660e+00,
-4.7401e+00, -4.2441e+00],
[-2.3929e+00, -1.1401e+00, -2.9145e+00, ..., -5.4785e+00,
-5.1897e+00, -2.3739e+00],
[-1.7345e+00, -1.7077e+00, -1.9372e+00, ..., -1.2164e+00,
-1.6641e+00, 4.9618e-01]],
[[-1.7291e+00, -2.1432e+00, -3.7562e+00, ..., -1.4200e+00,
-3.7549e-01, -6.0934e-01],
[-3.5644e+00, -3.1857e+00, -4.3382e+00, ..., -3.4183e+00,
-1.7258e+00, -2.7519e-01],
[-4.4099e+00, -4.2378e+00, -5.7828e+00, ..., -2.7379e+00,
-2.8875e+00, -1.3980e+00],
...,
[-2.1560e+00, -5.0216e+00, -3.3312e+00, ..., -3.6531e+00,
-3.4924e+00, -1.8693e-01],
[-1.9679e+00, -2.9666e-01, -2.2750e+00, ..., -2.7998e+00,
-1.3140e+00, 2.5805e+00],
[-2.9303e+00, -2.3493e+00, -1.7479e+00, ..., 1.2982e+00,
-1.2329e-01, 1.2638e+00]],
[[ 1.3496e+00, 3.3866e-02, 1.4635e+00, ..., 5.1021e-01,
7.3127e-01, -1.4823e+00],
[ 2.0306e+00, 4.1249e-01, 1.3822e+00, ..., 3.5636e+00,
2.9338e+00, 2.0955e+00],
[-4.1614e-03, -1.2039e-01, -9.1502e-04, ..., 3.3877e+00,
2.0437e+00, 1.3574e+00],
...,
[ 4.6415e+00, 5.4828e+00, 5.5700e+00, ..., -2.6608e+00,
1.0290e+00, 1.8965e+00],
[ 2.5288e+00, 3.0414e+00, -4.0426e-01, ..., 3.4549e+00,
1.3765e+00, 3.1199e+00],
[ 2.2412e+00, 2.2110e+00, 8.9051e-01, ..., 7.8240e-02,
-7.9404e-01, 1.7104e+00]]],
[[[ 2.4513e-01, 1.2784e+00, 1.8089e-01, ..., -3.0560e+00,
-2.2925e+00, -1.2314e+00],
[ 2.9083e-01, 7.5639e-01, 1.0339e+00, ..., -1.9351e+00,
-2.1185e+00, -3.5822e-01],
[-5.3622e-01, -6.4366e-01, -3.4872e-01, ..., -1.5501e+00,
-2.8703e+00, -3.8146e-01],
...,
[-7.3205e-01, -1.2917e+00, -2.5797e-01, ..., -1.8990e+00,
-3.5278e+00, 5.2533e-01],
[-1.9405e+00, -1.3793e+00, -7.6475e-01, ..., -3.1875e+00,
-3.3841e+00, 5.6485e-01],
[-1.2219e+00, -3.4623e+00, -3.2045e+00, ..., -3.5389e+00,
-3.0593e+00, -1.2440e+00]],
[[-1.0994e+00, -1.5905e+00, -3.4830e+00, ..., -6.4943e+00,
-7.6018e+00, -1.5682e+00],
[-3.8959e+00, -4.1882e+00, -3.5414e+00, ..., -6.2551e+00,
-8.5997e+00, -2.5696e+00],
[-3.8046e+00, -4.2210e+00, -3.5276e+00, ..., -6.0678e+00,
-7.3660e+00, -2.4130e+00],
...,
[-6.1954e+00, -5.9460e+00, -4.0933e+00, ..., -6.3583e+00,
-8.2095e+00, -1.8096e+00],
[-7.7259e+00, -7.9815e+00, -7.9005e+00, ..., -8.9578e+00,
-1.0535e+01, -4.2809e+00],
[-1.8200e+00, -3.4316e+00, -3.8289e+00, ..., -3.4404e+00,
-3.7163e+00, -6.7107e-01]],
[[-6.5416e-02, 1.2398e+00, 1.7065e+00, ..., 1.4810e+00,
2.5923e+00, 1.0359e+00],
[-1.5414e-02, -8.6529e-02, 6.9287e-01, ..., 1.3815e+00,
3.0096e+00, 2.0447e-01],
[ 6.3763e-01, 1.1590e+00, 1.2940e+00, ..., 2.7436e-01,
1.6990e+00, -3.7591e-01],
...,
[ 3.3963e-01, 6.0741e-01, 1.4081e+00, ..., 7.4619e-02,
1.1072e+00, -1.0074e+00],
[ 4.1632e-01, 6.1222e-01, 4.9689e-01, ..., -6.4089e-01,
1.1028e+00, -1.3399e+00],
[ 4.5236e-01, 1.2310e+00, 1.7263e+00, ..., -9.5736e-02,
-5.4842e-01, -1.8841e+00]],
...,
[[-6.1101e-01, -1.5717e+00, -2.2071e+00, ..., -5.1263e+00,
-5.4029e+00, -8.6608e-01],
[-2.8061e+00, -2.7179e+00, -1.6117e+00, ..., -3.8521e+00,
-4.7494e+00, -2.0621e+00],
[-2.1502e+00, -1.3918e+00, -9.8423e-01, ..., -4.6338e+00,
-5.2277e+00, -1.4391e+00],
...,
[-1.9865e+00, -1.2285e+00, -1.0274e+00, ..., -2.4515e+00,
-4.6204e+00, -1.4986e+00],
[-3.2109e+00, -3.2644e+00, -2.6161e+00, ..., -3.7580e+00,
-5.2989e+00, -9.8369e-01],
[-1.6319e+00, -2.2198e+00, -2.4563e+00, ..., -1.8131e+00,
-2.0272e+00, 4.1127e-01]],
[[-1.2825e+00, -2.3667e+00, -3.9301e+00, ..., -1.6960e+00,
-8.8107e-01, -6.7202e-01],
[-3.6385e+00, -3.5188e+00, -4.8740e+00, ..., -3.3841e+00,
-1.9967e+00, -2.9556e-01],
[-4.5569e+00, -3.7457e+00, -4.2772e+00, ..., -2.3747e+00,
-1.3628e+00, 1.1667e-01],
...,
[-4.7009e+00, -4.1561e+00, -4.4385e+00, ..., -3.4040e+00,
-1.9438e+00, -5.3055e-01],
[-4.0474e+00, -1.9913e+00, -2.7683e+00, ..., -1.0361e+00,
-2.3464e-01, 1.2979e+00],
[-3.2300e+00, -3.1488e+00, -2.4088e+00, ..., -3.2908e-01,
-6.5398e-01, 7.2091e-01]],
[[ 1.2278e+00, -4.8083e-02, 1.1290e+00, ..., 5.3040e-01,
9.9551e-01, -1.4145e+00],
[ 2.2891e+00, 1.1756e+00, 1.3923e+00, ..., 3.5814e+00,
3.1499e+00, 2.3463e+00],
[ 1.3270e+00, 7.4154e-01, 3.1999e-01, ..., 2.8590e+00,
1.7483e+00, 1.0842e+00],
...,
[ 2.6041e+00, 1.6436e+00, 8.3681e-01, ..., 3.0077e+00,
2.5101e+00, 8.2250e-01],
[ 2.8710e+00, 4.2338e+00, 2.4674e+00, ..., 2.7256e+00,
9.5271e-01, 1.8439e+00],
[ 2.0042e+00, 1.6727e+00, 5.6013e-01, ..., 2.3533e-01,
-1.5197e-01, 8.0671e-01]]]], device='cuda:0'),
tensor([[[[ 5.3585e-01, 1.5482e+00, 7.0052e-01, ..., 1.2951e+00,
-2.7515e-02, 6.6258e-01],
[ 2.3234e-01, -3.8900e-01, 1.4520e+00, ..., 5.9384e-01,
-9.4050e-01, -4.4914e-01],
[ 1.2098e-01, -2.9528e-01, 4.7279e-01, ..., 2.6331e-01,
-6.0313e-01, -9.2892e-01],
...,
[-9.2511e-01, -3.1419e+00, -3.7781e-01, ..., -3.0325e+00,
-3.6144e+00, -7.8573e-01],
[-2.2390e+00, -1.6359e+00, 3.4360e-01, ..., -2.7466e+00,
-2.9765e+00, -3.1122e-02],
[-2.2537e+00, -3.7142e+00, -3.0404e+00, ..., -4.2285e+00,
-3.5473e+00, -1.5684e+00]],
[[-1.4501e+00, -2.0849e+00, -2.1881e+00, ..., -7.7540e-01,
-5.6410e+00, -1.1732e+00],
[-5.0975e+00, -6.7900e+00, -4.6460e+00, ..., -3.7105e+00,
-7.5995e+00, -4.0587e+00],
[-4.4604e+00, -5.5677e+00, -4.7654e+00, ..., -5.7747e+00,
-6.1043e+00, -4.1212e+00],
...,
[-6.8683e+00, -8.1475e+00, -7.2981e+00, ..., -8.1875e+00,
-9.3606e+00, -2.3133e+00],
[-1.0112e+01, -1.0551e+01, -8.1626e+00, ..., -1.0805e+01,
-1.1097e+01, -3.9926e+00],
[-2.4706e+00, -4.8411e+00, -4.4194e+00, ..., -4.2652e+00,
-4.2026e+00, -6.2847e-01]],
[[-2.4936e-02, 1.1885e+00, 5.4268e-01, ..., 1.9433e+00,
2.1655e+00, 3.0264e-01],
[ 2.4174e-01, 1.2296e-01, -1.4263e+00, ..., 7.1604e-01,
2.2226e+00, -6.5188e-01],
[ 1.0056e+00, 2.4277e+00, 3.9479e-01, ..., 1.1085e+00,
2.2211e+00, 8.7758e-01],
...,
[-6.3012e-01, 3.7091e-01, -1.1230e-01, ..., 1.5541e+00,
6.3089e-01, -1.0032e+00],
[-8.5585e-01, -2.0753e-01, -5.2048e-01, ..., 2.6108e-01,
6.8711e-01, -5.2357e-01],
[ 5.7910e-01, 1.2937e+00, 9.6728e-01, ..., -1.9830e-01,
-4.6381e-01, -1.8845e+00]],
...,
[[-9.4933e-01, -7.5300e-01, -1.5604e+00, ..., -1.7953e+00,
-2.2578e+00, 7.2278e-02],
[-4.3155e+00, -4.3460e+00, -2.8082e+00, ..., -4.9249e+00,
-4.8447e+00, -3.1884e+00],
[-2.5503e+00, -1.4563e+00, -1.0225e+00, ..., -2.2359e+00,
-3.7944e+00, -2.1010e+00],
...,
[-2.3594e+00, -4.1512e+00, -4.3193e+00, ..., -4.4823e+00,
-6.1345e+00, -3.0095e+00],
[-3.5802e+00, -5.4586e+00, -2.2183e+00, ..., -4.0932e+00,
-5.4466e+00, -1.4446e+00],
[-3.1737e+00, -3.7801e+00, -3.3428e+00, ..., -2.7754e+00,
-2.5867e+00, 4.7793e-01]],
[[-1.1502e+00, -2.5560e-02, -3.0945e+00, ..., -8.6840e-01,
2.9739e-01, 2.1824e-01],
[-4.0683e+00, -1.1421e+00, -3.6905e+00, ..., -2.8536e+00,
-1.8936e+00, 3.0646e-01],
[-3.7862e+00, -1.8010e+00, -2.8458e+00, ..., -2.8520e+00,
-2.9845e+00, -6.2156e-01],
...,
[-4.6401e+00, -1.9079e+00, -2.0367e+00, ..., -3.0359e+00,
-2.5156e+00, -8.1947e-01],
[-3.6823e+00, -9.4776e-01, -1.2250e+00, ..., -1.9654e+00,
-1.2042e+00, 2.9606e-01],
[-3.6589e+00, -3.0714e+00, -2.8956e+00, ..., -1.7374e+00,
-1.2457e+00, 3.1194e-01]],
[[ 1.6768e+00, 6.2013e-01, 9.3899e-01, ..., 1.3663e+00,
1.6182e+00, 2.3144e-01],
[ 3.2209e+00, 2.2175e+00, 2.2630e+00, ..., 1.7725e+00,
2.9189e+00, 3.5973e+00],
[ 2.2775e+00, 5.9834e-01, 6.1713e-01, ..., 2.7438e+00,
2.5934e+00, 2.1466e+00],
...,
[ 2.5444e+00, 1.6025e+00, 3.8232e+00, ..., 1.7620e+00,
1.4398e+00, 2.5246e-01],
[ 4.8026e+00, 4.0761e+00, 3.2379e+00, ..., 2.9992e+00,
1.8687e+00, 1.6265e+00],
[ 2.8470e+00, 3.0212e+00, 2.4008e+00, ..., 5.4289e-01,
5.3453e-01, 1.0296e+00]]],
[[[ 1.1182e-01, 1.3292e+00, -1.3663e+00, ..., 3.1630e-01,
-1.9231e+00, -2.5877e+00],
[ 4.8388e-01, 8.4552e-01, 9.0415e-01, ..., 2.6779e+00,
2.6400e+00, -8.2654e-01],
[ 1.1219e-01, 2.5834e-02, 1.6331e+00, ..., 2.4894e+00,
3.3399e+00, -1.7077e-01],
...,
[-1.0859e+00, -2.3954e+00, -6.0497e-01, ..., 5.7211e+00,
1.0290e+00, -1.3672e+00],
[-1.1376e+00, -1.0385e+00, -4.6110e-01, ..., -1.7375e+00,
-2.1529e+00, -2.3090e+00],
[-2.6960e+00, -5.2414e+00, -4.5589e+00, ..., -4.1775e+00,
-4.8973e+00, -4.7503e+00]],
[[-1.4217e+00, -1.9761e+00, -4.6849e+00, ..., -1.7549e+00,
-7.9075e+00, -3.6113e+00],
[-4.5821e+00, -5.1764e+00, -2.9039e+00, ..., 1.3265e+00,
-3.8195e+00, -5.4347e+00],
[-3.5935e+00, -5.7762e+00, -5.5828e+00, ..., 1.0051e+00,
-4.0687e+00, -6.3631e+00],
...,
[-5.7837e+00, -5.5583e+00, -3.2524e+00, ..., -2.4629e+00,
-7.9505e+00, -5.7147e+00],
[-7.3602e+00, -8.2367e+00, -6.4461e+00, ..., -1.1752e+01,
-1.0145e+01, -6.3445e+00],
[-2.5298e+00, -4.5937e+00, -4.0678e+00, ..., -4.1705e+00,
-5.0432e+00, -2.4840e+00]],
[[ 4.0526e-01, 2.5004e+00, 1.6781e+00, ..., 3.1872e+00,
5.1983e+00, 5.2013e-01],
[-6.9180e-01, 2.5794e-01, 2.5912e-01, ..., -2.8108e+00,
6.6038e+00, 2.1409e+00],
[ 3.8251e-01, 1.4157e+00, 2.0440e-01, ..., -5.4489e+00,
3.4911e+00, 1.6367e+00],
...,
[ 3.5531e-01, 1.0166e+00, 5.2770e-01, ..., -4.4106e-01,
3.2091e+00, 1.5250e+00],
[ 8.6006e-01, 8.1260e-01, 6.7665e-01, ..., -1.2138e+00,
-1.8365e-01, -6.3592e-01],
[ 1.3455e+00, 1.5976e+00, 1.2094e+00, ..., 2.0377e+00,
1.4480e-01, -4.9715e-01]],
...,
[[-6.9236e-01, -1.8261e+00, -3.4004e+00, ..., -3.2514e+00,
-7.3488e+00, -2.3531e+00],
[-2.2009e+00, -2.6897e+00, -2.1509e+00, ..., -1.2258e+00,
-6.4423e+00, -1.5624e+00],
[-1.6635e+00, -1.5968e+00, -2.8925e+00, ..., -2.7965e+00,
-1.0080e+01, -1.3042e+00],
...,
[-2.0504e+00, -1.8920e+00, -2.8351e-01, ..., -2.3188e+00,
-5.7985e+00, -2.9527e+00],
[-3.0634e+00, -3.1193e+00, -2.0853e+00, ..., -4.4903e+00,
-4.9422e+00, -1.9825e+00],
[-3.5267e+00, -3.9174e+00, -3.4476e+00, ..., -2.9216e+00,
-3.9480e+00, -2.1671e+00]],
[[-1.6987e+00, -2.2445e+00, -2.4717e+00, ..., -5.5990e+00,
-8.1260e-01, 1.3267e+00],
[-3.7757e+00, -2.4164e+00, -5.2982e+00, ..., -2.8686e+00,
-2.9393e+00, 1.9596e+00],
[-2.6301e+00, -7.6508e-01, -4.7337e+00, ..., -7.9315e-01,
-4.3327e+00, 2.0591e+00],
...,
[-3.2528e+00, -1.9074e+00, -2.0543e+00, ..., 1.2523e+00,
-6.1575e+00, -2.2671e+00],
[-3.7278e+00, -5.9866e-01, -1.9191e+00, ..., -1.8177e+00,
-3.6280e+00, -3.8569e-01],
[-4.5384e+00, -3.0167e+00, -2.7370e+00, ..., -2.2465e+00,
-3.0999e+00, -1.0099e+00]],
[[ 1.3149e+00, 9.2927e-01, 1.4872e+00, ..., 1.8624e+00,
4.0710e+00, -2.1394e-01],
[ 2.8689e+00, 2.1779e+00, 1.8145e+00, ..., -1.2249e+00,
5.3035e+00, 4.9384e+00],
[ 1.9976e+00, 1.5492e+00, 1.6698e+00, ..., -7.1315e-01,
3.5582e+00, 2.8657e+00],
...,
[ 2.1806e+00, 4.6298e-01, 2.9636e-01, ..., 3.4962e+00,
2.7106e+00, 1.7294e+00],
[ 3.8743e+00, 3.9444e+00, 2.6989e+00, ..., 5.3448e+00,
2.1958e+00, 2.9245e+00],
[ 1.9814e+00, 8.0555e-01, 1.5088e-01, ..., 3.5349e-01,
1.4371e+00, 1.4139e+00]]],
[[[ 1.9601e+00, 1.4384e+00, 1.0000e+00, ..., 6.8020e-01,
-5.4313e-01, -8.0870e-01],
[-1.1344e+00, -3.3266e+00, -2.6586e+00, ..., -1.6482e+00,
-1.5480e-01, -1.5374e+00],
[-4.9937e+00, 1.1803e+00, 1.6846e+00, ..., 2.2764e+00,
1.6688e+00, -1.4170e+00],
...,
[-6.4208e-02, -2.9429e+00, -1.2289e+00, ..., -5.7591e-01,
-3.7751e-02, -5.1244e-01],
[-1.9873e+00, -1.9834e+00, 9.8265e-02, ..., 5.4068e-02,
-2.9065e-01, -1.3410e+00],
[-1.4210e+00, -3.9971e+00, -3.1599e+00, ..., -2.5927e+00,
-2.9458e+00, -2.7410e+00]],
[[-1.9091e+00, -3.8316e+00, -5.5506e+00, ..., -2.1202e+00,
-5.1501e+00, -2.8267e+00],
[-6.6853e+00, -9.1353e+00, -8.6789e+00, ..., -3.1943e+00,
-4.8582e+00, -4.8009e+00],
[-7.1973e+00, -2.0366e+00, -1.1736e+00, ..., 2.0777e+00,
-2.5197e+00, -4.6269e+00],
...,
[-6.6737e+00, -8.4542e+00, -7.6223e+00, ..., -3.0807e+00,
-3.6895e+00, -2.9275e+00],
[-7.6998e+00, -9.0760e+00, -9.0713e+00, ..., -4.3921e+00,
-5.3085e+00, -5.1797e+00],
[-1.8368e+00, -4.3134e+00, -4.2237e+00, ..., -3.0338e+00,
-2.8156e+00, -1.6265e+00]],
[[-4.1706e-02, 2.5000e+00, 2.1800e+00, ..., 2.0529e+00,
1.5072e+00, 7.2953e-01],
[ 7.2047e-02, 2.6252e+00, 4.2347e+00, ..., 5.1887e+00,
3.3045e+00, 1.0889e+00],
[ 3.3541e+00, 2.4228e+00, 5.8268e-01, ..., 2.0938e+00,
4.1447e+00, 1.7527e+00],
...,
[-9.6640e-01, 1.3762e+00, 1.7177e+00, ..., 1.0069e+00,
1.1435e+00, 9.2530e-01],
[ 1.2501e+00, 2.0221e+00, -8.6429e-01, ..., 1.0373e+00,
1.3136e+00, 5.0395e-01],
[ 6.2545e-01, 8.8752e-01, 6.2807e-01, ..., 1.0370e+00,
1.8079e+00, 5.8882e-01]],
...,
[[-5.1932e-01, -3.3263e+00, -2.6953e+00, ..., -1.3318e+00,
-2.6400e+00, -1.1106e+00],
[-4.4416e+00, -5.2612e+00, -3.5461e+00, ..., -3.4813e+00,
-4.8423e+00, -3.7795e+00],
[-4.2401e+00, 6.8858e-01, 3.2780e-01, ..., 3.3732e-01,
-3.6701e+00, -2.4211e+00],
...,
[-3.1828e+00, -2.8867e+00, 6.7046e-02, ..., -1.4957e+00,
-3.1592e+00, -1.1816e+00],
[-2.8091e+00, -3.6032e+00, -2.2136e+00, ..., -1.1694e+00,
-3.3806e+00, -1.6942e+00],
[-2.2662e+00, -3.1808e+00, -1.4451e+00, ..., -1.9575e+00,
-2.5344e+00, -5.5063e-01]],
[[-1.5682e+00, -2.4384e+00, -1.4870e+00, ..., 8.2138e-01,
-1.2670e+00, -6.7654e-01],
[-4.9528e+00, -3.7699e+00, -3.7777e+00, ..., -3.2895e+00,
-3.1532e+00, -1.7620e+00],
[-4.6935e+00, -7.0939e-01, 2.0358e+00, ..., -2.4147e+00,
-4.0316e+00, -7.5965e-01],
...,
[-3.1060e+00, -3.8269e+00, -2.3906e+00, ..., 1.0830e-01,
-2.9690e+00, -1.9705e+00],
[-4.2198e+00, -9.3026e-01, 2.0516e-01, ..., -7.8883e-02,
-3.2156e+00, -1.8809e+00],
[-3.9903e+00, -2.4060e+00, 8.7345e-02, ..., -5.1791e-01,
-2.5274e+00, -1.2257e+00]],
[[ 2.5285e+00, 1.2551e+00, 1.6260e+00, ..., 5.9220e-01,
1.3318e+00, 4.6046e-01],
[ 2.6414e+00, 2.5119e+00, 2.1874e+00, ..., 3.7213e+00,
3.2441e+00, 2.5282e+00],
[-1.0256e+00, 4.7966e+00, 2.4107e+00, ..., 2.1079e+00,
3.6462e+00, 5.4784e-01],
...,
[ 4.6670e+00, 1.5713e+00, 1.2478e+00, ..., 1.6237e+00,
1.6621e-01, 5.0163e-01],
[ 3.0043e+00, 4.3436e+00, 4.0589e+00, ..., 2.0227e+00,
7.8317e-01, 1.2468e+00],
[ 2.6039e+00, 1.9421e+00, 1.3866e+00, ..., 8.3502e-01,
-1.7274e-01, -1.5962e-01]]],
[[[-1.9613e-01, 5.5412e-01, 2.8218e-01, ..., -1.4607e+00,
-1.0209e+00, -8.7757e-01],
[ 2.1574e+00, 2.0541e+00, 2.6104e+00, ..., -8.3193e-01,
-1.3395e-01, -1.3935e+00],
[ 7.7627e-01, -1.9053e+00, -5.2283e+00, ..., -1.6905e+00,
-1.0499e+00, -2.9179e+00],
...,
[-6.9802e-02, -2.2694e-01, -9.7987e-01, ..., -1.8604e+00,
-3.4235e+00, -5.8605e-01],
[-2.0730e+00, -1.4873e+00, 6.9993e-01, ..., -3.6725e+00,
-3.1279e+00, 2.9222e-01],
[-2.3228e+00, -4.2170e+00, -3.2398e+00, ..., -4.0409e+00,
-3.5449e+00, -1.4884e+00]],
[[-6.7019e-01, -1.7132e+00, -3.2575e+00, ..., -2.9101e+00,
-4.6305e+00, -2.5965e+00],
[-2.6845e+00, -5.5170e+00, -8.4003e+00, ..., -7.7720e+00,
-6.3327e+00, -4.7169e+00],
[-4.9635e+00, -6.5480e+00, -1.1673e+01, ..., -5.9177e+00,
-8.1246e+00, -5.6987e+00],
...,
[-6.6065e+00, -8.2320e+00, -6.4841e+00, ..., -7.0626e+00,
-8.1339e+00, -1.8993e+00],
[-9.3268e+00, -1.0562e+01, -8.7420e+00, ..., -1.0761e+01,
-1.0912e+01, -3.9005e+00],
[-2.5808e+00, -5.1338e+00, -4.3847e+00, ..., -4.2306e+00,
-4.2570e+00, -6.5986e-01]],
[[ 7.5797e-01, 1.3870e+00, 1.2819e+00, ..., 1.3896e+00,
1.9412e+00, 8.8402e-01],
[ 6.7857e-01, 1.0159e+00, 1.3839e-01, ..., 1.1563e+00,
2.1609e+00, 8.3816e-01],
[-9.9284e-01, 1.0203e+00, 3.0610e+00, ..., 1.1919e+00,
1.8524e+00, 1.9658e+00],
...,
[-1.1891e+00, -5.4861e-01, 1.2377e+00, ..., 2.2276e+00,
6.9893e-01, -9.2371e-01],
[-5.6146e-01, 5.3492e-01, -1.0981e+00, ..., -1.5019e-01,
6.5535e-01, -1.0609e-01],
[ 4.8643e-01, 1.2485e+00, 9.1172e-01, ..., 5.3470e-02,
-5.5727e-01, -1.8817e+00]],
...,
[[-9.4229e-01, -1.1294e+00, -2.1854e+00, ..., -2.7283e+00,
-3.3643e+00, -1.2635e+00],
[-1.5764e+00, -3.4864e+00, -3.2952e+00, ..., -5.0067e+00,
-5.3298e+00, -3.8548e+00],
[-2.0587e+00, -5.3669e+00, -5.9787e+00, ..., -1.8534e+00,
-4.0920e+00, -4.2964e+00],
...,
[-2.1230e+00, -3.4384e+00, -5.5431e-01, ..., -3.6662e+00,
-6.0436e+00, -3.6223e+00],
[-3.9461e+00, -6.1628e+00, -2.1461e+00, ..., -4.9106e+00,
-5.7105e+00, -1.4545e+00],
[-3.3130e+00, -3.9306e+00, -3.3995e+00, ..., -2.7974e+00,
-2.5763e+00, 3.8224e-01]],
[[-1.6044e+00, -1.9438e+00, -2.5479e+00, ..., -2.9205e+00,
-2.8536e+00, -4.9393e-01],
[-3.6460e+00, -3.7435e+00, -2.8652e+00, ..., -4.4289e+00,
-6.3621e+00, -2.0172e+00],
[-2.5345e+00, -4.8149e+00, -6.4965e+00, ..., -1.5076e+00,
-3.7865e+00, -2.0343e+00],
...,
[-3.8550e+00, -3.5120e+00, -3.4878e+00, ..., -3.4245e+00,
-3.2417e+00, -1.0553e+00],
[-3.1206e+00, -8.5250e-01, -1.3782e+00, ..., -1.5983e+00,
-8.7206e-01, 5.3123e-01],
[-3.8281e+00, -3.0535e+00, -2.5138e+00, ..., -1.5992e+00,
-1.1702e+00, 3.2494e-01]],
[[ 7.9702e-01, -6.6202e-02, 9.3541e-01, ..., 2.1062e-01,
1.3817e+00, 6.7071e-01],
[ 2.2870e+00, 1.8549e+00, 4.1741e+00, ..., 7.2745e-01,
1.9304e+00, 3.0219e+00],
[ 2.3800e+00, -5.8916e-01, 1.1258e+00, ..., 5.5926e-02,
1.2589e+00, 2.3235e-01],
...,
[ 4.6263e+00, 3.2216e+00, 1.9240e+00, ..., 1.7425e+00,
2.5783e+00, 1.3541e+00],
[ 4.8003e+00, 4.4421e+00, 4.2595e+00, ..., 2.6815e+00,
1.3601e+00, 9.5420e-01],
[ 2.8758e+00, 2.6878e+00, 2.1454e+00, ..., 3.4130e-01,
5.4246e-01, 1.0417e+00]]],
[[[ 2.1598e-01, 1.3668e+00, 9.1403e-01, ..., -4.8757e+00,
-6.2835e+00, -3.2265e+00],
[ 4.4305e-01, 4.7404e-01, 8.6816e-01, ..., 2.4083e+00,
1.7875e+00, 1.0023e+00],
[-5.4858e-01, 4.4396e-01, 2.6706e-02, ..., -4.8792e+00,
-3.6768e+00, 2.6985e+00],
...,
[-2.4390e+00, -7.6302e+00, -2.7796e-01, ..., 1.6584e+00,
8.5149e-01, -9.4225e-01],
[-2.9725e+00, -6.6818e+00, -6.2605e-01, ..., 1.3017e+00,
5.8721e-01, -1.8022e+00],
[-3.1699e+00, -6.1204e+00, -8.8934e-01, ..., -4.0501e+00,
-3.8497e+00, -3.9748e+00]],
[[-8.3404e-01, -2.2693e+00, -3.9130e+00, ..., -7.3420e+00,
-7.3131e+00, -1.3392e+00],
[-4.6216e+00, -6.3636e+00, -6.6554e+00, ..., 1.0510e+00,
-1.1716e+00, -1.2685e+00],
[-3.6439e+00, -4.3381e+00, -5.7161e+00, ..., -6.1193e+00,
-3.9138e+00, 1.4144e+00],
...,
[-5.8149e+00, -7.1576e+00, 1.4103e+00, ..., -2.3487e+00,
-4.1915e+00, -4.2102e+00],
[-6.9626e+00, -1.0278e+01, -1.9393e+00, ..., -5.7860e+00,
-7.3796e+00, -6.7339e+00],
[-1.8839e+00, -4.6040e+00, -2.4375e-01, ..., -3.4584e+00,
-3.8310e+00, -2.5653e+00]],
[[-3.8341e-01, 2.0931e+00, 1.1343e+00, ..., 4.5371e+00,
6.2664e+00, 2.8195e+00],
[-2.2046e-01, 1.2785e+00, 8.0234e-01, ..., -2.1875e+00,
-4.3635e-01, 3.3717e+00],
[ 1.4088e-02, 1.9523e+00, 1.3669e+00, ..., -5.7039e+00,
-9.3438e+00, -2.7282e+00],
...,
[ 5.4928e-03, 2.1027e+00, -6.3747e+00, ..., 1.2861e-01,
1.7739e+00, 5.4096e-01],
[-4.0645e-01, -2.6357e-01, -6.3591e+00, ..., -8.5753e-03,
7.0574e-01, 3.8758e-01],
[ 1.5497e-01, -1.2186e+00, -7.0482e+00, ..., 1.1059e+00,
1.0809e+00, 2.0311e-01]],
...,
[[-8.2706e-01, -1.7047e+00, -1.3886e+00, ..., -5.2872e+00,
-5.7321e+00, -2.8723e+00],
[-3.1351e+00, -4.0486e+00, -2.6478e+00, ..., 2.3311e+00,
4.2948e-01, -1.5050e+00],
[-1.4805e+00, -2.2184e+00, -1.9948e+00, ..., -4.0706e+00,
-1.5209e+00, -1.7449e+00],
...,
[-2.9737e+00, -2.7938e+00, 1.1945e+00, ..., -1.5714e+00,
-4.1200e+00, -1.3855e+00],
[-4.6111e+00, -4.6889e+00, -2.4775e+00, ..., -1.7953e+00,
-4.4161e+00, -2.1262e+00],
[-3.7613e+00, -2.8890e+00, 6.6063e-01, ..., -2.5421e+00,
-3.6432e+00, -1.4843e+00]],
[[-1.7088e+00, -2.1301e+00, -2.6464e+00, ..., -4.5833e+00,
-6.1327e+00, 5.5388e-01],
[-4.2610e+00, -3.7197e+00, -3.9024e+00, ..., 1.6107e+00,
-3.1865e+00, -4.1647e+00],
[-3.3946e+00, -3.0594e+00, -3.2659e+00, ..., 9.9836e-01,
9.3512e-01, -6.2582e-01],
...,
[-5.8979e+00, -1.8761e+00, -3.5135e+00, ..., -1.8055e+00,
-5.5787e+00, -2.1284e+00],
[-3.5280e+00, -5.9678e-01, -3.8816e+00, ..., -1.6038e+00,
-5.4514e+00, -1.8205e+00],
[-3.3356e+00, -2.4529e-01, 1.0263e+00, ..., -1.9925e+00,
-4.2428e+00, -1.4722e+00]],
[[ 1.0894e+00, 5.5992e-01, 6.6974e-01, ..., -6.9804e-01,
5.3104e-01, 5.6011e-01],
[ 3.4911e+00, 3.2713e+00, 3.0542e+00, ..., 2.3666e+00,
4.1242e-01, 5.1743e+00],
[ 1.4733e+00, 1.1414e+00, 1.4701e+00, ..., -5.7663e+00,
-4.3995e+00, 7.1068e-01],
...,
[ 2.9879e+00, 1.0637e+00, 1.4743e+00, ..., 1.4801e+00,
1.7782e+00, 1.1145e+00],
[ 4.1341e+00, 3.3977e+00, 1.0443e-01, ..., 2.0797e+00,
2.1096e+00, 2.1558e+00],
[ 1.9563e+00, 1.8684e+00, 4.0495e-01, ..., -1.7999e-01,
-4.5949e-01, -2.4133e-01]]]], device='cuda:0'),
tensor([[[[ 3.9032e-03, 5.3896e-01, 5.3018e-01, ..., -3.0681e+00,
-2.9832e+00, 1.0917e+00],
[ 2.9919e-01, 6.1762e-01, 1.4910e+00, ..., -1.8246e+01,
-5.3231e+00, 4.9117e-01],
[ 2.2323e-02, 1.7725e-01, -1.9587e+00, ..., -2.1877e+01,
-3.2410e+00, 3.4535e+00],
...,
[ 1.9326e+00, -1.3607e+01, -4.6892e+00, ..., -1.2790e+00,
6.7754e-01, 2.2191e+00],
[-7.6601e-01, -1.4294e+01, -2.1847e+00, ..., -1.0572e+00,
-1.6761e+00, 8.2828e-01],
[-1.4757e+00, -1.0863e+01, -8.9790e-01, ..., -2.4297e+00,
-1.0795e+00, -1.3366e+00]],
[[-2.1631e+00, -3.4427e+00, -4.1619e+00, ..., -7.8759e-02,
-1.3462e+00, 1.3266e+00],
[-5.0868e+00, -5.4537e+00, -6.5218e+00, ..., -1.3092e+01,
-6.9225e+00, -9.2003e-01],
[-4.7461e+00, -5.4207e+00, -1.0344e+01, ..., -8.1473e+00,
3.3140e+00, 5.1246e+00],
...,
[-5.8186e+00, -2.0980e+01, -7.1434e+00, ..., -3.3247e+00,
-1.7827e+00, -5.8952e-02],
[-8.5906e+00, -1.9025e+01, -4.7229e+00, ..., -6.8147e+00,
-8.5225e+00, -4.7298e+00],
[-5.9783e-01, -6.9719e+00, -4.8592e-01, ..., -2.7569e+00,
-1.7453e+00, -5.4145e-01]],
[[ 2.5318e-01, 1.8674e+00, 1.7337e+00, ..., -4.0604e+00,
-6.1677e+00, -8.2376e-01],
[ 1.5605e-01, 5.5277e-01, 7.3609e-01, ..., -1.0540e+01,
-4.6852e+00, 3.1755e+00],
[ 5.1705e-01, 1.6357e+00, 1.3841e+00, ..., -9.2287e+00,
-2.8812e+00, 8.4997e-01],
...,
[-1.4183e+00, -1.0976e+00, -1.1781e+01, ..., -3.9492e+00,
-2.3940e+00, -5.8347e-01],
[-1.8641e+00, 1.0709e-01, -1.1829e+01, ..., -2.0465e+00,
-2.4036e-01, -4.6498e-01],
[-1.4463e+00, 1.5973e+00, -9.7518e+00, ..., -1.0767e+00,
-5.2699e-01, -1.2287e+00]],
...,
[[-1.1098e+00, -1.9557e+00, -2.0843e+00, ..., 9.2467e-01,
2.1429e+00, 1.9635e+00],
[-2.7364e+00, -2.0822e+00, -1.1476e+00, ..., -6.0939e+00,
-1.7680e+00, -1.6685e+00],
[-1.8332e+00, -1.4981e+00, -5.8933e-01, ..., -7.9128e-01,
5.5223e+00, 5.1637e+00],
...,
[ 7.6273e-01, -4.1117e+00, 6.0132e+00, ..., -8.5001e-01,
-2.8294e+00, -1.2989e+00],
[-1.1387e+00, -8.3457e+00, 5.7887e+00, ..., -3.3920e+00,
-5.3481e+00, -1.7659e+00],
[-5.3438e-01, -3.9448e+00, 5.5252e+00, ..., 6.1465e-01,
-1.1191e+00, 8.0549e-01]],
[[-2.1267e+00, -2.1866e+00, -3.7988e+00, ..., -2.8038e+00,
2.1919e+00, 3.1774e+00],
[-4.4330e+00, -2.6502e+00, -6.0303e+00, ..., -1.5123e+01,
-4.2007e+00, -4.4275e-01],
[-4.4254e+00, -3.5851e+00, -6.4722e+00, ..., -1.3964e+01,
2.0497e+00, 6.0739e+00],
...,
[-5.0397e+00, -6.2978e+00, -1.1724e+00, ..., -1.3342e+00,
-1.5789e+00, 1.2448e-01],
[-5.6116e+00, -4.2338e+00, -3.7526e-01, ..., -4.2468e-01,
-1.5971e+00, 1.6212e+00],
[-1.9679e+00, -2.5602e+00, 3.8420e+00, ..., 1.2428e+00,
-1.4093e+00, 4.1416e-01]],
[[ 1.6229e+00, 1.5809e-01, 1.4980e+00, ..., -6.4312e+00,
-4.5715e+00, -5.3239e-01],
[ 2.9575e+00, 1.5537e+00, 2.3439e+00, ..., -2.0932e+01,
-5.9590e+00, 4.9558e-01],
[ 2.0026e+00, 8.6393e-01, 2.8017e+00, ..., -2.0548e+01,
-3.7870e+00, 1.2581e+00],
...,
[ 3.7019e+00, 5.4591e+00, 6.8072e+00, ..., -2.4639e-01,
-2.2846e-02, 6.0816e-01],
[ 4.4995e+00, 3.2451e+00, 6.3256e+00, ..., -1.2759e+00,
2.3359e-01, 2.4016e+00],
[ 1.2938e+00, 1.3560e+00, 6.2721e+00, ..., 1.4320e-02,
4.8843e-01, 9.3611e-01]]],
[[[-1.1007e-01, 6.7189e-01, -3.8774e-01, ..., -1.3807e+00,
-1.3081e+00, -7.6235e-01],
[ 8.1893e-01, 4.1501e-01, -1.7192e-01, ..., -6.3809e-01,
3.7747e-01, -5.5171e-01],
[ 1.5397e+00, -2.6785e-01, 7.3397e-01, ..., 1.3745e+00,
1.6860e+00, -1.2205e+00],
...,
[-1.3254e+00, -7.1070e-01, -4.3440e-01, ..., -5.1668e+00,
-2.3353e+00, 2.5135e+00],
[-1.9821e+00, -5.3446e-01, -4.7089e-02, ..., -5.8320e+00,
1.6434e+00, 3.0978e+00],
[-1.5914e+00, -3.5924e+00, -2.8678e+00, ..., 1.8597e+00,
3.5256e+00, 1.2066e+00]],
[[-6.8277e-01, -1.5162e+00, -2.3966e+00, ..., -3.0907e+00,
-4.6798e+00, -2.1676e+00],
[-1.8088e+00, -3.6210e+00, -3.7785e+00, ..., -2.1937e+00,
-4.3954e+00, -3.8439e+00],
[-1.7476e+00, -3.2929e+00, -3.3897e+00, ..., 1.5895e-01,
-3.1619e+00, -4.1070e+00],
...,
[-6.0653e+00, -6.4158e+00, -5.4114e+00, ..., -7.2812e+00,
-5.5392e+00, 7.3519e-01],
[-7.7880e+00, -8.2646e+00, -6.7204e+00, ..., -7.6691e+00,
-6.3544e-02, 6.2871e-01],
[-1.7797e+00, -3.8906e+00, -3.7750e+00, ..., 3.1182e+00,
1.7007e+00, -2.6411e-01]],
[[ 6.5358e-01, 1.1092e+00, 8.1940e-01, ..., 1.4828e+00,
2.0115e+00, 1.0452e+00],
[ 7.8771e-01, 1.5090e+00, 1.4774e+00, ..., 2.0696e+00,
2.2647e+00, 8.3195e-01],
[ 2.1003e-01, 2.9955e+00, 2.4713e+00, ..., 2.3294e+00,
2.3043e+00, 1.5400e+00],
...,
[-3.6832e-02, 4.9303e-01, -1.2621e-01, ..., -9.9671e+00,
-7.8124e+00, -1.6966e+00],
[-8.8897e-02, -2.1433e-01, -1.6204e-01, ..., -9.4959e+00,
-3.0132e+00, 3.4809e-01],
[ 4.0610e-01, 1.1655e-01, -1.7010e-01, ..., -6.2795e+00,
-4.0684e+00, 8.1763e-01]],
...,
[[-9.4082e-01, -1.2966e+00, -1.8339e+00, ..., -2.3556e+00,
-3.4025e+00, -1.2501e+00],
[-2.1930e+00, -3.2543e+00, -2.5407e+00, ..., -3.2444e+00,
-3.9469e+00, -3.4470e+00],
[-8.4357e-01, -1.7038e+00, -2.7388e-01, ..., -1.2536e+00,
-2.6551e+00, -2.9655e+00],
...,
[-1.3372e+00, -1.2544e+00, -3.5112e-01, ..., -3.0651e+00,
6.4331e-01, -2.7649e-02],
[-2.0039e+00, -2.5164e+00, -1.0170e-01, ..., -2.8292e+00,
1.0822e-01, -2.7259e+00],
[-1.8456e+00, -2.3333e+00, -1.3207e+00, ..., 2.1252e+00,
8.1250e-01, -1.2988e+00]],
[[-1.5203e+00, -2.2874e+00, -3.3416e+00, ..., -2.9847e+00,
-2.7601e+00, -4.9936e-01],
[-4.2977e+00, -4.8858e+00, -4.5577e+00, ..., -2.8672e+00,
-4.6732e+00, -1.7321e+00],
[-3.9713e+00, -4.8490e+00, -3.5179e+00, ..., -9.3576e-01,
-3.0279e+00, -1.7756e+00],
...,
[-2.8692e+00, -2.1888e+00, -2.5488e+00, ..., 8.1349e-01,
1.9545e+00, 5.1375e-01],
[-2.8834e+00, 5.6060e-01, -5.5042e-01, ..., -5.9694e-01,
2.4635e+00, -1.3232e-01],
[-3.2960e+00, -1.4762e+00, -1.1891e+00, ..., 1.4961e+00,
1.4511e+00, -1.0068e+00]],
[[ 8.5615e-01, -1.2237e-01, 6.9254e-01, ..., 4.1132e-01,
1.2122e+00, 5.3345e-01],
[ 1.6215e+00, 7.4058e-01, 2.5998e-01, ..., 8.9385e-01,
1.8730e+00, 3.1382e+00],
[ 1.6279e+00, -1.2100e-01, 1.6731e-01, ..., 1.1095e+00,
1.1480e+00, 8.2421e-01],
...,
[ 2.6080e+00, 2.0647e+00, 3.5759e-01, ..., -4.7061e+00,
-1.6895e+00, 2.3978e+00],
[ 3.9558e+00, 4.4409e+00, 2.7864e+00, ..., -2.9271e+00,
2.7243e+00, 3.1714e+00],
[ 2.2063e+00, 1.7483e+00, 9.3107e-01, ..., 1.5214e+00,
4.4918e+00, 8.6441e-01]]],
[[[ 3.0465e-01, 1.3378e+00, 1.4906e-01, ..., -3.0580e+00,
-2.4771e+00, -1.0624e+00],
[ 4.4278e-01, 8.7217e-01, 9.5782e-01, ..., -1.6609e+00,
-2.2672e+00, -1.9960e-01],
[-3.6849e-01, -4.0529e-01, 8.1781e-02, ..., -1.2999e+00,
-2.5887e+00, -4.8625e-02],
...,
[-5.5764e-01, -1.9930e+00, 4.1374e-01, ..., -1.9103e+00,
-3.5274e+00, 5.0568e-01],
[-1.9513e+00, -1.6294e+00, -1.0547e+00, ..., -3.0757e+00,
-3.5891e+00, 4.9852e-01],
[-1.1031e+00, -3.2815e+00, -3.3042e+00, ..., -3.6196e+00,
-3.2824e+00, -1.5685e+00]],
[[-1.0079e+00, -1.4869e+00, -3.3745e+00, ..., -6.2466e+00,
-7.3400e+00, -1.3804e+00],
[-3.7970e+00, -4.0695e+00, -3.6740e+00, ..., -5.7461e+00,
-8.1472e+00, -2.3640e+00],
[-3.8299e+00, -4.0853e+00, -3.8772e+00, ..., -5.5187e+00,
-7.7456e+00, -2.4797e+00],
...,
[-5.8325e+00, -5.9449e+00, -4.5697e+00, ..., -6.6786e+00,
-8.9048e+00, -2.0063e+00],
[-7.6280e+00, -7.8225e+00, -7.6574e+00, ..., -9.1499e+00,
-1.1098e+01, -4.4608e+00],
[-1.7522e+00, -3.3362e+00, -3.7332e+00, ..., -3.6562e+00,
-3.8903e+00, -8.6870e-01]],
[[-3.8455e-02, 1.2451e+00, 1.6446e+00, ..., 1.6023e+00,
2.4312e+00, 9.5508e-01],
[-9.8688e-03, -7.0483e-02, 8.7178e-01, ..., 1.4702e+00,
3.0159e+00, 2.3316e-02],
[ 6.7510e-01, 1.1054e+00, 1.5663e+00, ..., 4.1968e-01,
1.8973e+00, -2.9685e-01],
...,
[ 2.6676e-01, 8.8156e-01, 1.4670e+00, ..., 8.0357e-01,
1.6602e+00, -1.3547e+00],
[ 4.7091e-01, 6.4808e-01, 1.1846e+00, ..., 1.3723e-02,
1.2728e+00, -1.6487e+00],
[ 5.2610e-01, 1.1986e+00, 1.7083e+00, ..., 1.6726e-02,
-2.5043e-01, -1.8798e+00]],
...,
[[-5.4413e-01, -1.5138e+00, -2.2410e+00, ..., -5.0991e+00,
-5.2318e+00, -7.3607e-01],
[-2.6645e+00, -2.5905e+00, -1.7762e+00, ..., -3.6366e+00,
-4.6992e+00, -1.7649e+00],
[-2.0536e+00, -1.1070e+00, -1.0064e+00, ..., -4.8785e+00,
-5.1322e+00, -1.4143e+00],
...,
[-1.8831e+00, -1.6295e+00, -8.2333e-01, ..., -3.2604e+00,
-5.5806e+00, -1.5618e+00],
[-3.2803e+00, -3.3284e+00, -2.8251e+00, ..., -3.6512e+00,
-5.4244e+00, -9.8740e-01],
[-1.5269e+00, -2.0187e+00, -2.3568e+00, ..., -1.9459e+00,
-2.3061e+00, 2.7572e-01]],
[[-1.2274e+00, -2.3810e+00, -3.9450e+00, ..., -1.5014e+00,
-4.6979e-01, -5.5692e-01],
[-3.4882e+00, -3.6461e+00, -4.7399e+00, ..., -3.3695e+00,
-1.7998e+00, -2.7977e-01],
[-4.4684e+00, -3.8160e+00, -4.2232e+00, ..., -2.5732e+00,
-1.8270e+00, -3.9203e-01],
...,
[-4.8023e+00, -3.8501e+00, -3.7493e+00, ..., -2.6545e+00,
-1.7898e+00, -1.1326e+00],
[-4.1433e+00, -2.1331e+00, -2.7743e+00, ..., -1.4964e+00,
-5.5439e-01, 1.1057e+00],
[-3.1955e+00, -3.3135e+00, -2.5883e+00, ..., -5.6871e-01,
-9.9235e-01, 6.1484e-01]],
[[ 1.1897e+00, -4.7187e-02, 1.1398e+00, ..., 5.6365e-01,
7.4040e-01, -1.3935e+00],
[ 2.2154e+00, 1.1297e+00, 1.3489e+00, ..., 3.5983e+00,
3.0463e+00, 2.2517e+00],
[ 1.3846e+00, 8.1060e-01, 3.3463e-01, ..., 2.7411e+00,
1.8596e+00, 1.1972e+00],
...,
[ 2.5412e+00, 1.3997e+00, 1.3710e+00, ..., 3.6105e+00,
2.0967e+00, 6.6591e-01],
[ 2.9812e+00, 3.9148e+00, 2.5825e+00, ..., 2.7052e+00,
1.1020e+00, 1.9082e+00],
[ 2.0121e+00, 1.6441e+00, 4.6625e-01, ..., 4.2061e-01,
-1.6197e-02, 9.2228e-01]]],
[[[ 4.3768e-01, 1.1777e+00, 5.9332e-01, ..., -4.2111e+00,
-3.9401e+00, -1.3569e+00],
[ 1.5495e-01, -5.0886e-01, 1.1723e+00, ..., -2.3824e+00,
-5.2780e-01, 1.6187e+00],
[-2.7004e-01, -9.1084e-01, 7.3218e-01, ..., 1.9149e+00,
3.1273e+00, 2.3581e+00],
...,
[-1.3117e+00, -1.8941e+00, -2.0787e-01, ..., -1.7338e+00,
-3.0147e+00, -1.1550e+00],
[-2.5690e+00, -1.6716e+00, 2.0611e-01, ..., -3.1314e+00,
-3.0552e+00, -2.7714e-02],
[-2.2061e+00, -3.8332e+00, -2.8900e+00, ..., -4.0809e+00,
-3.5176e+00, -1.5876e+00]],
[[-1.4144e+00, -1.6956e+00, -1.8906e+00, ..., -3.5461e+00,
-3.8145e+00, -3.2000e-01],
[-4.8752e+00, -6.2981e+00, -4.3889e+00, ..., 2.4101e+00,
-6.2597e-01, 5.4418e-01],
[-4.5277e+00, -5.1375e+00, -3.0528e+00, ..., -4.2182e+00,
-2.2018e+00, -2.6372e+00],
...,
[-7.2795e+00, -8.7523e+00, -6.9224e+00, ..., -6.6337e+00,
-8.6801e+00, -2.2212e+00],
[-9.9392e+00, -1.0177e+01, -8.2545e+00, ..., -1.0741e+01,
-1.0876e+01, -3.9580e+00],
[-2.5003e+00, -4.8318e+00, -4.1412e+00, ..., -4.1015e+00,
-4.1871e+00, -6.7973e-01]],
[[-2.4167e-01, 7.2953e-01, 5.1085e-01, ..., 7.6574e-01,
1.3175e+00, 8.4188e-01],
[-9.5047e-02, -1.5160e-01, -6.2250e-01, ..., 1.9249e+00,
2.3274e+00, 1.0081e+00],
[ 6.2680e-01, 2.2887e+00, 1.3586e+00, ..., -3.5773e+00,
-1.9210e-01, 3.6598e-01],
...,
[ 6.1410e-01, 7.9854e-01, 7.3680e-01, ..., 1.6743e+00,
1.1577e+00, -6.4499e-01],
[-5.7234e-01, 6.6352e-02, 8.4246e-02, ..., -4.3915e-01,
5.6198e-01, -4.2689e-01],
[ 4.4216e-01, 1.1445e+00, 1.0299e+00, ..., -3.7770e-01,
-4.8839e-01, -1.9240e+00]],
...,
[[-8.4166e-01, -8.2598e-01, -2.5498e-01, ..., -5.1695e+00,
-3.8607e+00, -1.0346e+00],
[-3.9238e+00, -4.7660e+00, -2.3748e+00, ..., 1.6178e-02,
-2.9352e+00, -3.0808e+00],
[-2.3885e+00, -1.4661e+00, -7.2601e-01, ..., 1.6642e-01,
5.9975e-02, -8.0768e-01],
...,
[-2.7865e+00, -3.4165e+00, -2.2803e+00, ..., -2.6639e+00,
-4.9549e+00, -2.9433e+00],
[-3.5875e+00, -5.4010e+00, -3.0279e+00, ..., -4.2731e+00,
-5.5543e+00, -1.4775e+00],
[-3.0574e+00, -3.4749e+00, -2.9669e+00, ..., -2.5752e+00,
-2.7954e+00, 4.0818e-01]],
[[-1.0275e+00, 1.9517e-01, -8.3749e-01, ..., -8.2537e+00,
-4.6004e+00, 1.8764e-01],
[-3.7677e+00, -1.1527e+00, -1.9210e+00, ..., -3.8862e+00,
-2.9276e+00, 5.1085e-02],
[-3.6673e+00, -1.5280e+00, -2.0778e+00, ..., 3.2819e+00,
1.5486e+00, 1.0519e+00],
...,
[-4.3148e+00, -2.1216e+00, -2.9872e+00, ..., -4.2925e+00,
-3.5665e+00, -1.0503e+00],
[-3.4588e+00, -3.6345e-01, -1.6180e+00, ..., -1.7633e+00,
-1.2770e+00, 3.8610e-01],
[-3.6534e+00, -2.9123e+00, -2.9299e+00, ..., -1.8077e+00,
-1.3691e+00, 3.6812e-01]],
[[ 1.6275e+00, 6.2133e-01, 3.0661e-01, ..., -2.0792e+00,
-5.9827e-01, 5.8394e-01],
[ 3.1076e+00, 1.9201e+00, 1.6283e+00, ..., -5.4648e+00,
-4.0849e+00, 8.3072e-01],
[ 2.2285e+00, -1.1280e-02, 7.0303e-02, ..., 1.9975e+00,
1.9501e-01, 2.4080e+00],
...,
[ 3.0242e+00, 2.9873e+00, 2.1647e+00, ..., 9.7685e-01,
1.6421e+00, 1.0420e+00],
[ 4.6541e+00, 4.8133e+00, 3.8022e+00, ..., 2.2777e+00,
1.3540e+00, 1.2959e+00],
[ 2.7368e+00, 2.6471e+00, 2.0220e+00, ..., 6.3636e-01,
6.1153e-01, 1.0778e+00]]],
[[[-2.1949e-01, 1.1613e+00, 6.0022e-01, ..., 3.3313e-02,
4.6933e-01, -1.4388e-01],
[ 1.0831e-01, -8.3131e-03, 1.8551e+00, ..., 1.4894e-01,
6.5474e-01, -5.1258e-01],
[-1.1979e+00, -6.6779e-01, 1.3485e+00, ..., -4.9599e-01,
1.3556e-01, -9.3337e-01],
...,
[-2.5303e-01, -1.7514e-01, 4.5234e-01, ..., 3.9133e-01,
6.3820e-01, -3.4342e-01],
[-1.4546e+00, -1.3998e-01, -3.0973e-02, ..., 6.0356e-01,
4.3290e-01, -1.1947e+00],
[-1.2830e+00, -2.3991e+00, -1.7823e+00, ..., -3.0655e+00,
-2.8881e+00, -2.7790e+00]],
[[-1.5601e+00, -2.4157e+00, -3.8677e+00, ..., -1.9164e+00,
-3.4743e+00, -2.1441e+00],
[-4.8008e+00, -5.0266e+00, -3.5564e+00, ..., -2.6349e+00,
-3.9824e+00, -3.5260e+00],
[-3.8654e+00, -4.1179e+00, -3.6979e+00, ..., -2.7718e+00,
-2.5701e+00, -2.9040e+00],
...,
[-4.5290e+00, -5.1832e+00, -4.9965e+00, ..., -2.0702e+00,
-3.1707e+00, -2.5480e+00],
[-7.6807e+00, -7.8791e+00, -6.2188e+00, ..., -4.7947e+00,
-5.2936e+00, -5.0648e+00],
[-1.4421e+00, -3.1663e+00, -1.8993e+00, ..., -2.9513e+00,
-2.7995e+00, -1.6311e+00]],
[[-2.1219e-01, 2.2989e+00, 1.5809e+00, ..., 1.6187e+00,
1.7138e+00, 8.6716e-01],
[-5.3097e-01, 1.9997e-01, 8.4255e-01, ..., 9.3606e-01,
1.4435e+00, 7.9689e-01],
[-1.0086e-01, 1.7915e+00, 1.8352e+00, ..., 1.6352e+00,
1.8729e+00, 1.3732e+00],
...,
[-6.9056e-01, 4.5613e-01, 1.5403e+00, ..., 1.6429e+00,
1.3396e+00, 5.6010e-01],
[-1.1651e+00, 3.3977e-02, 1.1336e+00, ..., 1.3557e+00,
1.0719e+00, 3.8049e-01],
[-2.4657e-01, -3.7566e-01, -2.6072e-02, ..., 1.6158e+00,
1.6417e+00, 5.5294e-01]],
...,
[[-1.3399e+00, -2.2439e+00, -2.3812e+00, ..., -1.4991e+00,
-1.9339e+00, -4.5109e-01],
[-3.6056e+00, -3.9709e+00, -2.4013e+00, ..., -2.8211e+00,
-3.5658e+00, -2.9890e+00],
[-2.7681e+00, -2.6933e+00, -1.8826e+00, ..., -2.2186e+00,
-2.7243e+00, -1.8302e+00],
...,
[-9.5770e-01, -2.2927e+00, -1.0894e+00, ..., -1.6062e+00,
-2.5564e+00, -8.7235e-01],
[-2.1697e+00, -4.0023e+00, -1.6557e+00, ..., -1.7701e+00,
-2.9300e+00, -1.5869e+00],
[-1.4871e+00, -1.5020e+00, -7.2580e-01, ..., -1.8813e+00,
-2.2223e+00, -5.7304e-01]],
[[-1.8196e+00, -1.9031e+00, -3.0697e+00, ..., -4.9330e-01,
-1.3586e+00, -6.8518e-01],
[-4.0639e+00, -2.3556e+00, -2.8577e+00, ..., -1.6548e+00,
-3.7566e+00, -1.8413e+00],
[-3.7433e+00, -2.9444e+00, -3.8602e+00, ..., -1.4202e+00,
-4.4528e+00, -2.4725e+00],
...,
[-2.8809e+00, -2.8716e+00, -3.3270e+00, ..., -8.2234e-01,
-3.1064e+00, -2.2160e+00],
[-2.0178e+00, -2.5565e-01, -1.5569e+00, ..., -6.0680e-01,
-3.2865e+00, -1.9281e+00],
[-2.1672e+00, -1.4860e+00, -1.7212e+00, ..., -1.6132e+00,
-2.8467e+00, -1.2178e+00]],
[[ 1.2155e+00, 3.5055e-01, 8.6116e-01, ..., 1.7766e-01,
1.1986e+00, 6.2196e-01],
[ 3.9687e+00, 3.0059e+00, 2.3368e+00, ..., 5.7500e-01,
1.0739e+00, 2.4478e+00],
[ 1.6276e+00, 2.0835e-01, 7.0494e-01, ..., 2.1898e-01,
1.6916e-01, 7.7526e-01],
...,
[ 3.0294e+00, 2.3667e+00, 2.4822e+00, ..., 1.3153e+00,
3.7484e-01, 5.9065e-01],
[ 3.8579e+00, 3.9327e+00, 2.6984e+00, ..., 1.0428e+00,
8.0857e-01, 1.2359e+00],
[ 2.4363e+00, 2.3165e+00, 1.5090e+00, ..., -8.2335e-02,
-2.9146e-01, -1.1667e-01]]]], device='cuda:0'),
tensor([[[[ 2.4575e-01, 1.3415e+00, -1.4386e+00, ..., -2.5510e+00,
-3.8215e+00, -1.5795e+00],
[ 3.5210e-01, 2.1006e-01, 7.2704e-01, ..., -9.1370e-01,
-2.3309e+00, -6.6238e-01],
[ 1.0177e-01, 7.2124e-01, 1.6208e-01, ..., 4.0444e-01,
-1.5371e+00, -1.1089e+00],
...,
[-7.8331e-02, -5.4845e+00, -2.3234e+00, ..., 2.7476e+00,
7.4207e-01, -1.2137e+00],
[-3.4682e-01, -3.4107e+00, -2.7966e+00, ..., 2.5919e+00,
2.3358e-01, -2.2134e+00],
[-2.1724e+00, -5.4942e+00, -5.3005e+00, ..., -3.6057e+00,
-3.9359e+00, -4.1549e+00]],
[[-1.2765e+00, -2.3533e+00, -4.6202e+00, ..., -7.4140e+00,
-8.6386e+00, -2.1352e+00],
[-4.5978e+00, -5.3582e+00, -2.8629e+00, ..., -1.0064e+01,
-8.6247e+00, -2.8463e+00],
[-3.6441e+00, -6.4712e+00, -7.6378e+00, ..., -1.0131e+01,
-1.0749e+01, -5.0745e+00],
...,
[-6.0413e+00, -8.2272e+00, -5.3391e+00, ..., -2.7556e-01,
-3.5045e+00, -4.2953e+00],
[-7.3848e+00, -8.5031e+00, -8.1024e+00, ..., -6.6784e+00,
-7.9689e+00, -6.7186e+00],
[-2.5215e+00, -3.8867e+00, -5.4692e+00, ..., -4.0097e+00,
-4.3676e+00, -2.6228e+00]],
[[ 4.3269e-01, 2.1750e+00, 1.7069e+00, ..., 9.4059e-01,
2.6443e+00, 9.5771e-01],
[-5.2056e-01, 2.1841e-01, 1.7391e-01, ..., 1.4040e+00,
2.3112e+00, 2.7500e-01],
[ 5.3454e-01, 2.1710e+00, -6.8807e-01, ..., 1.6114e+00,
3.5557e-02, -5.2757e-01],
...,
[ 8.0324e-02, 1.5079e+00, 6.5146e-01, ..., -9.5904e-01,
1.4707e+00, 3.1060e-01],
[ 5.4863e-01, -2.7938e-02, -5.3682e-02, ..., -6.2037e-01,
1.9580e-01, 2.3555e-01],
[ 1.2878e+00, 1.3250e+00, 6.6647e-01, ..., 1.3946e+00,
9.7955e-01, 2.9434e-01]],
...,
[[-6.2316e-01, -1.7941e+00, -3.0719e+00, ..., -5.0316e+00,
-6.3182e+00, -1.7402e+00],
[-2.3396e+00, -3.0588e+00, -1.5155e+00, ..., -3.4768e+00,
-6.0570e+00, -2.5321e+00],
[-1.6676e+00, -2.1359e+00, -3.2265e+00, ..., -3.7454e+00,
-7.1348e+00, -2.9486e+00],
...,
[-1.9168e+00, -3.4932e+00, -7.2761e-01, ..., -6.0794e-01,
-3.4993e+00, -1.0725e+00],
[-2.7629e+00, -4.2352e+00, -2.7252e+00, ..., -2.0149e+00,
-5.1730e+00, -2.5059e+00],
[-3.2387e+00, -5.0944e+00, -3.9384e+00, ..., -2.4512e+00,
-3.8388e+00, -1.7261e+00]],
[[-1.5081e+00, -1.6574e+00, -4.2832e+00, ..., -2.4436e+00,
-1.4840e+00, 5.6952e-01],
[-3.5631e+00, -9.3674e-01, -2.5222e+00, ..., -1.9428e+00,
-2.6819e+00, -1.0835e-02],
[-2.7567e+00, -1.3757e-01, -2.8513e+00, ..., 2.4584e-01,
-7.2870e-01, 2.3282e-01],
...,
[-3.1008e+00, -2.0732e+00, -3.3824e-01, ..., -1.4501e+00,
-3.2655e+00, -7.5296e-01],
[-3.5521e+00, -1.1906e+00, 8.7413e-01, ..., -2.4522e+00,
-4.7914e+00, -1.1118e+00],
[-4.5278e+00, -3.7338e+00, -4.0013e-01, ..., -2.7100e+00,
-3.8646e+00, -1.1257e+00]],
[[ 1.3998e+00, 8.8040e-01, 1.4377e+00, ..., 9.8817e-01,
1.1335e+00, -1.7261e-01],
[ 2.8774e+00, 2.4016e+00, 1.6768e+00, ..., 4.0271e+00,
2.4804e+00, 3.7780e+00],
[ 2.1001e+00, 1.6870e+00, 3.2363e+00, ..., 3.3580e+00,
2.1441e+00, 1.5216e+00],
...,
[ 2.9995e+00, 8.9175e-01, 2.4540e+00, ..., 2.1467e+00,
2.1228e+00, 1.2365e+00],
[ 3.9393e+00, 3.9913e+00, 3.6653e+00, ..., 3.3159e+00,
2.3833e+00, 2.3731e+00],
[ 1.7747e+00, 4.2343e-01, 1.0091e+00, ..., 6.9999e-02,
8.1169e-02, 6.7927e-02]]],
[[[ 2.5847e-01, 8.1643e-01, -7.0818e-01, ..., 2.2040e-01,
9.1092e-01, 5.7092e-02],
[ 1.1916e+00, 3.2786e-01, -1.5749e+00, ..., -8.3165e-01,
1.9104e+00, 3.3797e-01],
[ 1.1669e+00, -1.3567e+00, -4.0917e+00, ..., -7.7010e-02,
2.6538e+00, -6.5364e-01],
...,
[-3.3588e+00, -4.1802e+00, -5.3219e+00, ..., -2.2623e+00,
-2.2223e+00, -1.9148e-01],
[-4.7088e+00, -1.3845e+00, -9.6736e-02, ..., -2.4543e+00,
-2.9242e+00, 6.0442e-02],
[-3.9816e+00, -4.2541e+00, -2.6611e+00, ..., -4.4120e+00,
-3.5690e+00, -1.5624e+00]],
[[-5.3438e-01, -1.3940e+00, -2.8118e+00, ..., 4.8200e-01,
-1.6747e+00, -1.6923e+00],
[-2.0478e+00, -3.4501e+00, -4.2169e+00, ..., -1.9353e+00,
-1.1410e+00, -2.4653e+00],
[-2.6624e+00, -4.1531e+00, -3.6907e+00, ..., 6.6068e-01,
-3.4593e-01, -3.8442e+00],
...,
[-2.2432e+00, 2.1219e+00, -2.4986e+00, ..., -1.0076e+01,
-9.0342e+00, -1.8963e+00],
[-1.3088e+01, -7.8456e+00, 2.1979e+00, ..., -1.0197e+01,
-1.0674e+01, -3.8377e+00],
[-2.4578e+00, -8.0963e+00, -6.1936e+00, ..., -4.3513e+00,
-4.2393e+00, -6.5399e-01]],
[[ 6.8310e-01, 1.4034e+00, 1.0571e+00, ..., 2.1209e-01,
1.6614e+00, 1.2924e+00],
[ 4.5043e-01, 1.3861e+00, 1.7357e+00, ..., -4.9746e+00,
5.8396e-01, 1.4183e+00],
[ 3.9163e-01, 2.9853e+00, 2.8503e+00, ..., -8.0556e-01,
2.2383e+00, 1.7398e+00],
...,
[-4.0009e+00, 2.4494e+00, 9.3601e-01, ..., 2.3664e+00,
4.8571e-01, -1.4227e-01],
[-2.4749e+00, -6.7770e+00, -2.3916e+00, ..., -6.1925e-01,
3.8728e-01, -2.8055e-01],
[ 1.1066e+00, -4.1820e+00, -4.0087e+00, ..., -4.0376e-01,
-5.7566e-01, -1.8090e+00]],
...,
[[-7.0558e-01, -1.4689e+00, -2.0421e+00, ..., 1.1639e+00,
-8.6749e-01, -6.0847e-01],
[-1.9123e+00, -3.1770e+00, -2.7065e+00, ..., 1.4230e-01,
-2.7718e+00, -2.8354e+00],
[-1.1313e+00, -2.4711e+00, -1.2306e+00, ..., 2.0492e+00,
-3.8983e+00, -3.5472e+00],
...,
[-1.7269e-01, -2.5235e+00, -4.4026e+00, ..., -6.2171e+00,
-5.9930e+00, -2.8261e+00],
[-4.1428e+00, -4.7627e+00, -2.4840e+00, ..., -3.7483e+00,
-5.2241e+00, -1.5056e+00],
[-4.4789e+00, -7.7136e+00, -8.1985e+00, ..., -3.0641e+00,
-2.6741e+00, 3.5743e-01]],
[[-1.5845e+00, -3.2993e+00, -4.4616e+00, ..., 1.8252e-01,
-1.8479e+00, 2.3949e-01],
[-4.2318e+00, -5.5660e+00, -6.0340e+00, ..., 7.8094e-01,
-3.0674e+00, -6.0953e-01],
[-4.5167e+00, -5.6346e+00, -4.6312e+00, ..., -3.4747e-01,
-3.8885e+00, -1.3094e+00],
...,
[-4.4827e-01, -8.1528e+00, -9.3660e+00, ..., -4.8492e+00,
-3.0767e+00, -1.0231e+00],
[ 5.0415e-01, 4.8849e+00, -9.6559e-02, ..., -2.0793e+00,
-1.2603e+00, 3.3880e-01],
[-1.5949e+00, 2.8710e+00, -3.4954e+00, ..., -1.3928e+00,
-1.1098e+00, 3.3761e-01]],
[[ 8.3214e-01, -2.7955e-02, 7.9894e-01, ..., -4.1700e-01,
1.7402e+00, 1.4415e+00],
[ 1.8226e+00, 8.1789e-01, 7.9473e-01, ..., -9.6975e-01,
1.1480e+00, 3.7391e+00],
[ 1.9613e+00, 1.5502e-01, -1.1972e+00, ..., -9.5172e-01,
2.9889e+00, 2.2665e+00],
...,
[ 8.1964e-01, -5.7258e+00, -8.1829e+00, ..., 3.1943e+00,
3.1326e+00, 7.2192e-01],
[ 6.5274e+00, 2.6326e+00, -4.7320e+00, ..., 2.3763e+00,
2.0553e+00, 1.5003e+00],
[ 1.5610e+00, 5.7549e+00, 3.1608e+00, ..., 9.1097e-01,
6.3640e-01, 9.7146e-01]]],
[[[-2.1996e-01, 1.1217e+00, 5.1522e-01, ..., -1.0557e-01,
1.9292e-01, -2.6952e-01],
[ 1.6248e-01, 2.3636e-01, 2.3386e+00, ..., 2.8253e-01,
6.3056e-01, -5.2950e-01],
[-1.1771e+00, 1.4057e-01, 1.4545e+00, ..., -6.0917e-01,
2.2545e-01, -6.5963e-01],
...,
[-4.6253e-01, -1.6563e-01, 6.0970e-01, ..., 6.5982e-01,
6.0943e-01, -3.8291e-01],
[-1.4972e+00, -6.1120e-02, 8.9346e-02, ..., 2.1675e-02,
-1.0099e-01, -1.5695e+00],
[-1.0575e+00, -2.1325e+00, -1.7552e+00, ..., -3.2191e+00,
-2.9887e+00, -2.9309e+00]],
[[-1.4946e+00, -2.2836e+00, -3.9761e+00, ..., -1.7840e+00,
-3.5049e+00, -2.1466e+00],
[-4.6655e+00, -4.7984e+00, -4.5357e+00, ..., -2.7184e+00,
-3.6782e+00, -3.4227e+00],
[-4.0661e+00, -4.9975e+00, -5.2280e+00, ..., -2.5403e+00,
-1.7803e+00, -1.8896e+00],
...,
[-4.1992e+00, -4.8967e+00, -4.7023e+00, ..., -2.2286e+00,
-2.3396e+00, -2.6144e+00],
[-7.3945e+00, -7.5300e+00, -6.3104e+00, ..., -5.3763e+00,
-5.9540e+00, -5.5101e+00],
[-1.3186e+00, -2.9552e+00, -1.9289e+00, ..., -3.0831e+00,
-2.9002e+00, -1.6801e+00]],
[[-2.0783e-01, 2.3732e+00, 1.7928e+00, ..., 1.5371e+00,
1.7047e+00, 8.9830e-01],
[-6.3925e-01, 5.0224e-01, 1.3153e+00, ..., 6.1034e-01,
1.4575e+00, 7.8519e-01],
[-1.1949e-01, 2.3400e+00, 1.3250e+00, ..., 1.4738e+00,
1.7531e+00, 1.4606e+00],
...,
[-7.8705e-01, 5.1813e-01, 1.2685e+00, ..., 6.3092e-01,
1.8818e-01, 5.9650e-01],
[-1.4699e+00, -7.4295e-02, 1.0375e+00, ..., 9.7696e-01,
2.5698e-01, 4.7058e-03],
[-6.3078e-01, -3.2112e-01, 2.2628e-01, ..., 1.7872e+00,
1.7007e+00, 5.2456e-01]],
...,
[[-1.3236e+00, -2.2869e+00, -2.5169e+00, ..., -1.5248e+00,
-1.9325e+00, -4.9935e-01],
[-3.5325e+00, -3.6728e+00, -2.1101e+00, ..., -2.6840e+00,
-3.3152e+00, -2.7994e+00],
[-2.6451e+00, -2.1971e+00, -2.0741e+00, ..., -2.2504e+00,
-1.8676e+00, -1.6246e+00],
...,
[-7.2164e-01, -2.4200e+00, -9.9337e-01, ..., -2.2133e-02,
-1.1458e+00, -6.9257e-01],
[-1.7531e+00, -4.0273e+00, -1.8177e+00, ..., -2.1260e+00,
-3.4878e+00, -2.1370e+00],
[-1.0804e+00, -1.4789e+00, -6.5587e-01, ..., -2.0342e+00,
-2.3506e+00, -6.3606e-01]],
[[-1.8881e+00, -2.1056e+00, -3.1535e+00, ..., -8.0467e-01,
-1.4133e+00, -5.6310e-01],
[-4.0954e+00, -2.6183e+00, -3.5060e+00, ..., -1.9822e+00,
-3.7467e+00, -1.6485e+00],
[-3.7485e+00, -3.3233e+00, -5.5610e+00, ..., -7.5294e-01,
-3.4960e+00, -2.1289e+00],
...,
[-2.4714e+00, -2.4266e+00, -2.9487e+00, ..., -4.9750e-01,
-3.2984e+00, -2.2564e+00],
[-1.7432e+00, -2.5047e-01, -1.5511e+00, ..., -7.2765e-01,
-3.3764e+00, -1.8962e+00],
[-1.9393e+00, -1.5611e+00, -1.7282e+00, ..., -1.6929e+00,
-2.8652e+00, -1.2582e+00]],
[[ 1.1895e+00, 3.1579e-01, 9.3457e-01, ..., -1.6713e-01,
9.7750e-01, 4.7096e-01],
[ 4.0864e+00, 2.8691e+00, 2.5686e+00, ..., 5.5969e-01,
9.1061e-01, 2.3539e+00],
[ 1.7173e+00, 2.8861e-01, 1.1544e+00, ..., -5.1463e-01,
-2.4395e-01, 2.5972e-01],
...,
[ 2.8162e+00, 2.5916e+00, 2.3668e+00, ..., 4.3891e-01,
6.6283e-01, 5.8499e-01],
[ 3.5566e+00, 3.8545e+00, 2.6487e+00, ..., 1.5794e+00,
1.1807e+00, 1.3909e+00],
[ 2.3276e+00, 2.2599e+00, 1.5420e+00, ..., -1.9823e-02,
-2.5302e-01, -2.8676e-02]]],
[[[ 4.5842e-01, 1.1746e+00, 6.1350e-03, ..., -2.5097e+00,
-6.0079e+00, -4.9824e+00],
[ 4.8746e-01, 8.1018e-01, 1.7392e+00, ..., 1.2399e+00,
-1.2572e+00, 1.7741e+00],
[ 2.2937e-01, -1.7630e-01, -3.1992e-01, ..., 8.8160e-01,
-2.1690e+00, 3.0069e+00],
...,
[-1.0947e+00, -8.5755e+00, -1.7847e+00, ..., -2.4601e+00,
-6.1586e+00, -2.1247e+00],
[-2.2119e+00, -5.6849e+00, -1.2581e+00, ..., -5.2572e+00,
-4.4337e+00, -3.4803e-01],
[-1.1776e+00, -4.4170e+00, -3.6255e+00, ..., -4.6569e+00,
-4.2781e+00, -2.1198e+00]],
[[-1.3328e+00, -2.1287e+00, -3.3428e+00, ..., -5.9910e+00,
-1.0651e+01, -2.4068e+00],
[-4.5809e+00, -4.9867e+00, -3.4213e+00, ..., -2.2009e+00,
-8.0103e+00, 7.4375e-02],
[-5.4558e+00, -6.7467e+00, -3.6535e+00, ..., 1.3344e-01,
-7.1696e-01, 2.8968e+00],
...,
[-4.9344e+00, -8.6841e+00, -2.8118e+00, ..., -1.3466e+01,
-1.3871e+01, -6.3075e+00],
[-7.0382e+00, -8.7991e+00, -6.4611e+00, ..., -8.0096e+00,
-7.6783e+00, -2.9682e+00],
[-1.6626e+00, -2.9532e+00, -4.0503e+00, ..., -4.8850e+00,
-4.5818e+00, -9.5577e-01]],
[[-4.2332e-02, 1.4146e+00, 1.7112e+00, ..., 2.5375e+00,
4.8705e+00, 2.7433e+00],
[ 5.1986e-02, 1.6888e-01, 7.3998e-01, ..., 4.5242e-01,
2.9065e+00, 2.7893e+00],
[ 7.6061e-01, 6.9446e-01, 6.2386e-01, ..., -7.1608e+00,
-7.0759e+00, -6.8205e+00],
...,
[-4.5117e-01, 1.3665e+00, -3.0751e+00, ..., -3.4934e+00,
-8.9669e-01, -2.6309e+00],
[ 8.6868e-01, 1.3481e+00, -2.8408e+00, ..., -2.2049e+00,
6.2821e-02, -6.3674e-01],
[ 2.3195e-01, 5.5995e-01, 5.5254e-01, ..., -1.8125e+00,
-2.6708e+00, -2.8671e+00]],
...,
[[-7.1492e-01, -1.7148e+00, -2.2767e+00, ..., -3.9947e+00,
-7.6264e+00, -3.4195e+00],
[-3.0018e+00, -2.6146e+00, -1.5437e+00, ..., 2.8476e+00,
4.9633e-01, 1.1024e+00],
[-2.3654e+00, -1.8385e+00, -1.9829e+00, ..., 2.1636e-01,
2.1746e+00, 1.3796e+00],
...,
[-2.2154e+00, -3.6753e+00, -1.9619e-01, ..., -7.6937e+00,
-1.0078e+01, -3.4773e+00],
[-2.6558e+00, -3.1884e+00, -4.9725e-01, ..., 3.9084e-02,
-3.3048e+00, 1.4623e+00],
[-1.0188e+00, -1.6430e+00, -2.9028e+00, ..., -1.6830e+00,
-2.4145e+00, 2.0635e-01]],
[[-1.5870e+00, -2.6206e+00, -4.0327e+00, ..., -4.0180e+00,
-6.5016e+00, 9.0696e-03],
[-4.0674e+00, -2.8514e+00, -3.6540e+00, ..., -3.4358e+00,
-5.4143e+00, -4.2409e+00],
[-5.1100e+00, -2.6760e+00, -2.3866e+00, ..., -4.9463e+00,
-6.1082e-01, 1.9568e+00],
...,
[-3.8123e+00, -2.5033e+00, 7.5598e-02, ..., -7.2572e+00,
-7.4511e+00, -2.0240e+00],
[-3.6738e+00, -1.2194e+00, 2.7233e+00, ..., 7.1975e-02,
-1.8490e+00, 1.7537e+00],
[-3.0319e+00, -2.6087e+00, -3.0403e-01, ..., 7.7221e-01,
-4.2840e-01, 1.4376e+00]],
[[ 1.3069e+00, 8.2522e-02, 1.1635e+00, ..., -1.0823e+00,
5.9399e-01, -8.0051e-02],
[ 2.5279e+00, 1.5614e+00, 1.2947e+00, ..., 5.4757e+00,
3.8685e-01, 4.8554e+00],
[ 2.0612e+00, 1.6809e+00, 2.1646e-01, ..., -1.9863e+00,
-5.6374e+00, -2.2171e-02],
...,
[ 2.7062e+00, -4.0552e-01, 1.6193e+00, ..., 4.9156e+00,
4.4649e+00, 4.3376e+00],
[ 3.3688e+00, 2.5717e+00, 4.2417e+00, ..., 1.0137e+00,
-1.1719e+00, 1.5606e+00],
[ 1.5920e+00, 7.3414e-01, 1.4225e+00, ..., 3.2143e-01,
8.4583e-03, 1.2666e+00]]],
[[[-1.6329e+00, 1.5160e-01, 6.0531e-01, ..., -1.8212e+00,
-3.4203e+00, -1.7507e+00],
[ 5.3795e-01, -7.1742e-01, -2.3799e+00, ..., -3.3636e+00,
-3.4104e+00, -1.1885e+00],
[ 8.3842e-01, -1.5034e+00, -3.5069e+00, ..., 1.9116e+00,
5.9386e-01, 1.0461e+00],
...,
[-2.7184e+00, -3.8051e+00, -3.9714e-01, ..., -9.4917e-01,
4.9519e-01, -3.3020e-01],
[-1.6290e+00, -3.0909e-01, 2.0970e+00, ..., -7.3132e-01,
1.1068e-01, -1.4663e+00],
[-2.3663e+00, -4.3328e+00, -3.1603e+00, ..., -4.5869e+00,
-4.0285e+00, -3.7204e+00]],
[[-7.9122e-01, -1.7397e-01, 1.3694e+00, ..., -5.1244e+00,
-8.8964e+00, -2.7334e+00],
[ 8.7933e-02, -1.1427e+00, -5.4633e+00, ..., -6.0233e+00,
-6.5732e+00, -1.5378e+00],
[-5.6826e-01, -3.3944e+00, -4.4061e+00, ..., -7.7062e-01,
-2.1695e+00, -1.7858e-01],
...,
[-6.3101e+00, -5.3389e+00, -8.7864e-01, ..., -3.7973e+00,
-4.2575e+00, -2.9889e+00],
[-7.5143e+00, -7.2261e+00, -5.4162e+00, ..., -7.2404e+00,
-7.9182e+00, -6.2915e+00],
[-2.1995e+00, -3.7917e+00, -3.9592e+00, ..., -3.8860e+00,
-4.0483e+00, -2.4894e+00]],
[[ 1.3559e+00, -9.5715e-02, -2.5865e+00, ..., 1.6231e+00,
3.1938e+00, 6.5615e-01],
[-5.5457e-01, -3.2347e+00, -9.0379e+00, ..., 2.7625e+00,
5.0585e+00, 1.3157e+00],
[-1.0512e+00, -6.2170e-01, -6.9462e+00, ..., -4.2022e-01,
7.6205e-01, 1.6584e+00],
...,
[ 7.3156e-01, -2.1991e+00, -6.0836e+00, ..., -4.1826e-01,
-3.4175e-01, -2.6559e-01],
[ 3.2953e-01, -2.1704e+00, -4.8406e+00, ..., 7.6977e-01,
-4.4057e-01, -3.9416e-01],
[ 1.0391e+00, 1.0520e+00, 9.5284e-01, ..., 1.6727e+00,
1.1029e+00, 1.3568e-01]],
...,
[[-1.2366e+00, 1.2124e+00, 1.5494e+00, ..., -4.1672e+00,
-5.9857e+00, -1.6014e+00],
[-1.9512e+00, -3.5037e+00, -4.1247e+00, ..., -4.3053e+00,
-5.6264e+00, -2.9796e+00],
[-6.9071e-01, -4.8649e+00, -3.7281e+00, ..., -6.8916e-02,
-1.3789e+00, -1.3048e+00],
...,
[-3.5819e+00, -3.6498e+00, -2.6983e+00, ..., -4.5169e-01,
-2.0508e+00, -8.2413e-01],
[-3.5431e+00, -2.9146e+00, -2.5531e+00, ..., -2.4184e+00,
-3.9700e+00, -2.3112e+00],
[-3.3110e+00, -3.7347e+00, -3.9376e+00, ..., -3.0109e+00,
-3.5179e+00, -1.5629e+00]],
[[-1.2752e+00, 7.9407e-01, 7.5683e-01, ..., -2.2495e+00,
-7.5335e-01, 1.3429e+00],
[-2.4637e+00, -1.7560e+00, -1.3177e+00, ..., -4.1668e+00,
-3.2760e+00, 9.8612e-01],
[-1.2541e+00, -3.3441e+00, -6.6424e-01, ..., 1.4787e+00,
-1.1105e+00, -2.9296e-01],
...,
[-3.7861e+00, -2.5459e+00, -3.0330e+00, ..., -3.9954e-01,
-1.5819e+00, -9.9680e-01],
[-3.1142e+00, 1.8378e-01, -4.3775e-01, ..., -4.3593e-01,
-2.3151e+00, -1.1347e+00],
[-3.9511e+00, -2.3638e+00, -2.1046e+00, ..., -2.0877e+00,
-3.1358e+00, -1.1978e+00]],
[[ 1.1813e+00, 1.4947e+00, 4.2122e-01, ..., 8.3139e-01,
1.8232e+00, 6.5235e-02],
[ 1.5992e+00, 1.5930e+00, 2.3017e-01, ..., 2.8313e+00,
2.6680e+00, 4.4165e+00],
[ 1.3559e+00, -5.2155e-01, -1.5972e+00, ..., 2.8955e+00,
1.3027e+00, 3.2622e+00],
...,
[ 2.8692e+00, 1.8655e+00, -2.4388e-01, ..., 2.0330e+00,
2.1016e+00, 1.6028e+00],
[ 3.3447e+00, 4.2042e+00, 2.7205e+00, ..., 2.3465e+00,
1.9983e+00, 2.0722e+00],
[ 1.7840e+00, 1.0345e+00, 1.0646e+00, ..., -3.9540e-01,
-3.4563e-01, -2.7668e-01]]]], device='cuda:0'),
tensor([[[[ 7.7173e-01, 7.0538e-01, -5.0650e-01, ..., -1.7216e-01,
3.9447e-02, -2.8476e-01],
[ 1.4064e+00, 5.0503e-01, 1.1751e+00, ..., 1.5984e-01,
4.7546e-01, -6.6208e-01],
[-1.3079e+00, -1.0521e+00, 2.5186e+00, ..., 7.7483e-01,
4.2184e-03, -1.3905e+00],
...,
[ 4.6291e-01, -2.2927e-01, 1.0859e+00, ..., -1.1728e+00,
1.4707e-01, 8.4044e-01],
[-1.4637e+00, -1.2099e+00, 6.5380e-01, ..., 1.3349e-02,
2.5045e-01, -7.6057e-01],
[-1.0509e+00, -3.7297e+00, -3.0480e+00, ..., -2.4036e+00,
-2.7151e+00, -2.7515e+00]],
[[-1.4293e+00, -4.0492e+00, -6.7458e+00, ..., -2.5463e+00,
-3.5039e+00, -2.3683e+00],
[-3.5673e+00, -5.1671e+00, -5.8238e+00, ..., -3.7345e+00,
-4.1955e+00, -4.0214e+00],
[-3.1138e+00, -3.0037e+00, -4.0944e+00, ..., -2.3656e+00,
-2.8142e+00, -2.8344e+00],
...,
[-4.6437e+00, -6.9139e+00, -6.6863e+00, ..., -3.3087e+00,
-3.0498e+00, -1.8418e+00],
[-7.4231e+00, -9.2179e+00, -9.5499e+00, ..., -3.9543e+00,
-5.1036e+00, -5.0186e+00],
[-1.4997e+00, -4.2691e+00, -4.3070e+00, ..., -2.9423e+00,
-2.8409e+00, -1.6612e+00]],
[[ 8.8180e-01, 2.9625e+00, 1.7999e+00, ..., 1.9178e+00,
1.6615e+00, 9.0496e-01],
[-3.1180e-01, 1.6055e+00, 1.5297e+00, ..., 1.4665e+00,
1.4596e+00, 9.2438e-01],
[ 9.4761e-01, 3.4727e+00, 2.2400e+00, ..., 2.1989e+00,
1.3153e+00, 1.4193e+00],
...,
[ 4.1662e-01, 2.2172e+00, 5.1676e-02, ..., 3.5704e-01,
-9.8897e-01, -3.4644e-02],
[ 6.5934e-01, 1.7744e+00, 5.5870e-01, ..., 4.0887e-01,
3.8438e-01, 6.0390e-01],
[ 4.4702e-01, 1.6117e+00, 1.1165e+00, ..., 1.1272e+00,
1.4676e+00, 7.2238e-01]],
...,
[[-8.9560e-01, -3.3040e+00, -2.5730e+00, ..., -1.8175e+00,
-2.3665e+00, -4.8612e-01],
[-2.9038e+00, -4.2007e+00, -1.3871e+00, ..., -1.0472e+00,
-4.1156e+00, -2.8915e+00],
[-3.0403e+00, -3.0540e+00, -1.1636e+00, ..., -5.2539e-01,
-3.2317e+00, -1.8799e+00],
...,
[-1.2930e+00, -2.8059e+00, -4.9132e-01, ..., -1.3275e+00,
-1.8153e+00, -1.3196e+00],
[-2.7948e+00, -4.5697e+00, -1.4971e+00, ..., -9.2484e-01,
-2.5790e+00, -1.8372e+00],
[-2.0771e+00, -3.4123e+00, -1.4251e+00, ..., -1.9808e+00,
-2.5822e+00, -5.4125e-01]],
[[-2.6048e+00, -3.1545e+00, -1.6695e+00, ..., -2.3869e-01,
-1.3405e+00, -3.6597e-01],
[-4.0470e+00, -3.3020e+00, -2.1746e+00, ..., -2.4511e-02,
-3.7111e+00, -1.5888e+00],
[-3.8799e+00, -2.7666e+00, -1.5693e+00, ..., 8.1038e-01,
-3.3810e+00, -2.1959e+00],
...,
[-3.5659e+00, -2.7421e+00, -1.2735e+00, ..., -8.5555e-01,
-1.3176e+00, -1.5368e+00],
[-4.3603e+00, -1.0239e+00, -7.2057e-02, ..., -1.3025e-01,
-2.5386e+00, -1.5459e+00],
[-3.8586e+00, -2.6883e+00, -2.8495e-02, ..., -6.9656e-01,
-2.5838e+00, -1.1946e+00]],
[[ 1.5868e+00, 1.3608e+00, 1.5839e+00, ..., 4.6993e-01,
1.0935e+00, 5.3194e-01],
[ 2.7970e+00, 2.9579e+00, 3.4925e+00, ..., 1.4802e+00,
1.1901e+00, 2.6057e+00],
[ 6.2354e-01, -1.6910e-01, 2.1822e+00, ..., 1.6981e+00,
1.7162e-01, 2.8409e-01],
...,
[ 3.0335e+00, 2.2674e+00, 2.8010e+00, ..., 1.1928e+00,
9.9261e-01, 8.9943e-01],
[ 3.8730e+00, 4.2444e+00, 4.4281e+00, ..., 1.5646e+00,
1.0945e+00, 1.3663e+00],
[ 2.3837e+00, 1.4979e+00, 1.6377e+00, ..., 7.1920e-01,
-8.3074e-02, 7.2259e-02]]],
[[[ 3.1499e-01, 1.5150e+00, 6.0255e-01, ..., -5.1360e-01,
2.0692e-01, 7.4842e-01],
[ 5.1633e-01, 2.7062e-01, 6.5429e-01, ..., -1.3720e+00,
-1.2553e+00, 8.9734e-01],
[ 5.5524e-01, 1.1590e+00, 1.2363e+00, ..., -1.0779e+00,
-1.2864e+00, -5.8770e-01],
...,
[-3.6127e-01, -1.6388e+00, -1.2037e+00, ..., -8.0945e+00,
-5.0692e+00, -2.2648e+00],
[-1.2402e+00, -7.9206e-01, 2.3048e-02, ..., -2.7019e+00,
-3.2158e+00, -2.9192e+00],
[-2.7086e+00, -5.2179e+00, -4.4648e+00, ..., -5.3692e+00,
-4.3523e+00, -3.6700e+00]],
[[-6.2527e-01, -2.1362e+00, -4.1419e+00, ..., -3.0196e+00,
-4.5882e+00, -3.2273e-01],
[-3.6334e+00, -5.7079e+00, -5.8658e+00, ..., -7.7154e+00,
-8.2353e+00, -1.7828e+00],
[-3.6227e+00, -6.2431e+00, -6.9038e+00, ..., -6.9036e+00,
-1.0931e+01, -4.6345e+00],
...,
[-6.2006e+00, -5.9572e+00, -2.3198e+00, ..., -1.4988e+01,
-9.4595e+00, -4.5989e+00],
[-7.2724e+00, -8.4552e+00, -6.7861e+00, ..., -8.4897e+00,
-8.7818e+00, -7.1207e+00],
[-2.5416e+00, -4.4567e+00, -4.0667e+00, ..., -4.5944e+00,
-4.6057e+00, -2.7744e+00]],
[[-2.5199e-01, 2.1738e+00, 1.0923e+00, ..., 8.9863e-01,
2.0861e+00, 9.0868e-01],
[-3.0026e-01, 1.1500e+00, 1.0233e+00, ..., 5.0838e-01,
8.8909e-01, -8.1653e-01],
[ 9.4289e-02, 1.9906e+00, 3.8030e-01, ..., -2.3638e-01,
1.2108e+00, -5.8404e-02],
...,
[ 4.8519e-01, 4.4834e-01, 1.5916e+00, ..., -2.5991e+00,
6.1073e-01, -8.9554e-01],
[ 9.3308e-01, 4.0158e-01, 2.1574e-01, ..., 1.0540e+00,
-1.2319e-03, -8.4065e-01],
[ 1.3775e+00, 1.5199e+00, 1.4057e+00, ..., 8.7349e-01,
1.1558e+00, 7.0491e-01]],
...,
[[-6.9966e-01, -1.7494e+00, -1.6980e+00, ..., -2.2254e+00,
-2.4929e+00, -3.3242e-01],
[-3.0017e+00, -4.5280e+00, -3.4638e+00, ..., -5.0848e+00,
-4.8052e+00, -2.8615e+00],
[-9.7785e-01, -2.8978e+00, -3.0010e+00, ..., -1.3187e+00,
-4.1922e+00, -3.1071e+00],
...,
[-2.6271e+00, -2.1593e+00, 9.0438e-01, ..., -9.2166e+00,
-5.8222e+00, -2.2821e+00],
[-3.5463e+00, -3.6659e+00, -2.0650e+00, ..., -2.7483e+00,
-5.3949e+00, -3.8809e+00],
[-3.4895e+00, -3.9469e+00, -3.5398e+00, ..., -4.7124e+00,
-4.8511e+00, -2.2386e+00]],
[[-1.6195e+00, -2.1650e+00, -2.7553e+00, ..., -1.7020e+00,
-1.5869e+00, -1.4531e+00],
[-4.1130e+00, -3.2601e+00, -1.8162e+00, ..., -3.8027e+00,
-2.3376e+00, -1.0004e+00],
[-3.6468e+00, -2.6000e+00, -1.3433e+00, ..., 1.2265e+00,
-2.0788e+00, -1.3861e+00],
...,
[-4.2460e+00, -3.1615e+00, -2.8071e+00, ..., -2.6296e+00,
-3.3128e+00, -1.9966e+00],
[-3.3878e+00, -2.7665e-01, -1.2370e+00, ..., -2.1905e+00,
-2.5720e+00, -1.9471e-01],
[-4.4851e+00, -2.8617e+00, -2.6485e+00, ..., -1.3981e+00,
-2.7161e+00, -5.7695e-01]],
[[ 1.0216e+00, 6.4146e-01, 6.3611e-01, ..., -7.8767e-01,
1.4899e-01, -4.1001e-01],
[ 3.0190e+00, 3.4477e+00, 2.5601e+00, ..., 1.6742e+00,
2.2399e+00, 2.2437e+00],
[ 1.2723e+00, 1.8175e+00, 2.8548e+00, ..., 1.6475e+00,
2.5458e+00, 1.2403e+00],
...,
[ 3.5076e+00, 1.2951e+00, 1.3237e+00, ..., 4.2784e+00,
2.7430e+00, 3.1651e+00],
[ 3.5486e+00, 3.5423e+00, 2.5398e+00, ..., 2.2114e+00,
3.6307e+00, 3.4101e+00],
[ 1.9681e+00, 9.7465e-01, 3.6964e-01, ..., -3.6906e-01,
-7.7192e-01, -5.5552e-01]]],
[[[-2.2378e-01, 1.8493e+00, -6.6113e-02, ..., -1.0645e+00,
-8.8438e-01, -4.9958e-01],
[ 1.0313e+00, 3.3930e-01, -2.0136e+00, ..., -3.0971e-01,
-2.0465e-01, -7.2457e-01],
[ 1.2197e+00, -2.0489e+00, -6.9791e+00, ..., 1.0528e+00,
-1.6739e+00, -8.1128e-01],
...,
[-2.3759e+00, 5.0010e+00, -3.1865e+00, ..., -9.3761e-01,
7.5078e-01, -1.1308e-01],
[-3.4499e+00, 1.3567e-01, -4.1302e+00, ..., 1.1093e+00,
9.7371e-01, -1.6332e-01],
[-2.5666e+00, -1.0706e+00, -2.1845e+00, ..., -2.8847e+00,
-2.6164e+00, -2.2469e+00]],
[[-1.0704e+00, -2.0650e+00, -5.9207e+00, ..., -3.9501e+00,
-4.8944e+00, -1.9254e+00],
[-3.8657e+00, -5.5936e+00, -7.0076e+00, ..., -2.1302e+00,
-3.7138e+00, -2.2657e+00],
[-5.6540e+00, -9.1050e+00, -8.4249e+00, ..., -9.5735e-01,
-2.7948e+00, -1.3310e+00],
...,
[-3.9374e+00, 3.3619e+00, -8.0127e+00, ..., -5.1404e+00,
-7.1847e-01, -6.5236e-01],
[-7.2704e+00, -2.0735e-02, -2.6620e+00, ..., -5.2368e+00,
-5.7632e+00, -4.3306e+00],
[-2.8677e+00, -4.1295e+00, -3.1037e+00, ..., -2.7324e+00,
-3.2526e+00, -1.5949e+00]],
[[-2.9471e-01, 2.1956e+00, 1.1522e+00, ..., 1.6737e+00,
1.9528e+00, 9.6206e-01],
[ 6.6183e-02, 8.2468e-01, 1.0608e+00, ..., 2.4379e+00,
1.6172e+00, 7.4004e-01],
[-3.4927e-01, -1.4603e-01, 3.4930e+00, ..., 3.3430e+00,
2.9975e+00, 1.2525e+00],
...,
[-7.5220e+00, -9.7492e+00, -4.0896e+00, ..., -9.7326e-01,
-1.1140e+00, -5.1440e-01],
[-6.7090e+00, -7.6865e+00, -1.4769e+00, ..., -9.1605e-01,
-6.5266e-01, 5.0776e-01],
[-2.5327e+00, -5.6058e+00, -3.6996e+00, ..., 7.8533e-02,
9.4755e-01, 3.1496e-01]],
...,
[[-1.1006e+00, -2.1043e+00, -2.5298e+00, ..., -2.3901e+00,
-2.9483e+00, -6.7647e-01],
[-2.8850e+00, -4.6350e+00, -3.8173e+00, ..., -1.2822e+00,
-2.9758e+00, -2.5102e+00],
[-2.2813e+00, -3.8439e+00, -1.8175e+00, ..., -1.0617e+00,
-2.5055e+00, -1.3837e+00],
...,
[ 1.7634e+00, 1.4439e-01, -7.0634e+00, ..., 6.4935e-01,
-2.5704e+00, -5.7301e-01],
[-5.2151e-01, -8.5840e-01, -4.8600e+00, ..., -2.1675e-01,
-3.4302e+00, -9.9931e-01],
[-1.8158e+00, -2.5573e+00, -3.5543e+00, ..., -2.3609e+00,
-2.6888e+00, -6.6956e-01]],
[[-1.5744e+00, -1.6438e+00, -3.1289e+00, ..., -4.8956e-01,
-1.5733e+00, -3.0720e-01],
[-3.2928e+00, -3.7917e+00, -1.7221e+00, ..., -1.6105e+00,
-3.1439e+00, -1.4950e+00],
[-3.1034e+00, -2.8469e+00, -2.1310e+00, ..., -8.8799e-01,
-3.4239e+00, -2.6778e+00],
...,
[-1.5014e+00, -6.4241e+00, -5.1851e+00, ..., 4.4408e-01,
-2.5993e+00, -1.6995e+00],
[-1.2358e+00, -2.3229e+00, -2.5540e+00, ..., 1.1768e+00,
-1.9963e+00, -1.3379e+00],
[-3.2121e-01, 7.6954e-01, -1.3887e+00, ..., -3.6945e-01,
-1.0901e+00, -7.1851e-01]],
[[ 1.0878e+00, 3.6464e-01, 1.4536e+00, ..., 2.8935e-01,
1.0253e+00, 3.7321e-01],
[ 3.8809e+00, 3.6152e+00, 1.9626e+00, ..., 1.9042e+00,
1.0349e+00, 2.0411e+00],
[ 2.8958e+00, 7.0303e-01, -1.5651e+00, ..., 2.1163e+00,
-2.4824e-01, -6.3806e-02],
...,
[-4.7227e-01, -4.1942e-01, -4.1094e+00, ..., 1.7972e+00,
-8.9232e-01, -9.5622e-01],
[ 1.6729e+00, -8.6364e-01, -4.5564e+00, ..., 2.5159e+00,
1.6308e+00, 1.9744e+00],
[ 1.5581e+00, 3.7707e+00, 3.2762e+00, ..., 1.3192e+00,
3.8572e-02, 4.8168e-01]]],
[[[ 6.9293e-01, 5.5505e-01, -2.1559e-01, ..., 1.5674e-01,
8.0472e-02, -5.8429e-01],
[ 1.0575e+00, 5.1572e-01, 8.6010e-01, ..., 1.4222e+00,
5.6871e-02, -1.5517e+00],
[-2.6361e-01, 1.9577e-01, 2.4592e+00, ..., 1.7518e+00,
-2.0615e+00, -2.2753e+00],
...,
[ 2.2137e+00, 2.6037e+00, -1.3084e+00, ..., -1.0501e+00,
-6.5715e-01, -2.9045e-01],
[-8.7186e-01, -2.9485e+00, -1.3698e+00, ..., -3.4988e-02,
-2.0253e-01, -1.3920e+00],
[-1.0250e+00, -3.9775e+00, -3.2044e+00, ..., -2.3816e+00,
-3.0125e+00, -2.8994e+00]],
[[-1.3091e+00, -3.8774e+00, -6.8318e+00, ..., -2.1263e+00,
-4.3957e+00, -2.4447e+00],
[-4.7516e+00, -5.8516e+00, -5.8742e+00, ..., 3.1254e-01,
-3.8218e+00, -2.8899e+00],
[-3.2395e+00, -4.3958e+00, -5.7725e+00, ..., -3.2855e+00,
-3.1024e+00, -1.4436e+00],
...,
[-2.2382e+00, -5.3730e+00, -9.2064e+00, ..., -3.4104e+00,
-3.5654e+00, -3.1590e+00],
[-9.6229e+00, -1.2264e+01, -1.0071e+01, ..., -4.4459e+00,
-5.5762e+00, -5.3084e+00],
[-1.2263e+00, -4.0005e+00, -4.3932e+00, ..., -2.8028e+00,
-2.9998e+00, -1.6816e+00]],
[[ 7.7000e-01, 3.0775e+00, 1.9250e+00, ..., 2.8524e+00,
2.2079e+00, 6.3754e-01],
[ 2.6453e-02, 1.6440e+00, 1.6781e+00, ..., 2.5234e+00,
3.0562e+00, 1.0218e+00],
[ 6.4836e-01, 1.8430e+00, 8.8703e-01, ..., 2.8410e+00,
4.7824e+00, 2.7062e+00],
...,
[ 3.9534e-01, -1.2960e+00, 1.5918e+00, ..., -1.7741e-01,
7.3238e-01, 9.2370e-01],
[-9.7287e-01, 2.1258e+00, 4.5022e-01, ..., 1.4452e+00,
1.2454e+00, 5.8417e-01],
[ 6.3580e-01, 2.7134e+00, 1.1914e-01, ..., 1.4012e+00,
1.7897e+00, 5.8577e-01]],
...,
[[-8.9820e-01, -3.2875e+00, -2.1647e+00, ..., -1.7372e+00,
-2.6251e+00, -1.2924e+00],
[-2.7108e+00, -4.0311e+00, -1.2034e+00, ..., -2.7623e+00,
-4.6077e+00, -3.6828e+00],
[-2.0987e+00, -2.6639e+00, -1.3382e+00, ..., -4.5715e+00,
-4.9170e+00, -3.9099e+00],
...,
[-9.0272e-01, -3.9360e+00, -3.7492e+00, ..., -1.5607e+00,
-3.3449e+00, -1.5634e+00],
[-4.3907e+00, -6.1582e+00, -2.5842e+00, ..., -1.4696e+00,
-3.4806e+00, -1.8204e+00],
[-1.6007e+00, -2.8385e+00, -1.6347e+00, ..., -1.9636e+00,
-2.5362e+00, -6.3997e-01]],
[[-2.5223e+00, -3.2803e+00, -1.4877e+00, ..., 6.5983e-01,
-6.1422e-01, -4.8668e-01],
[-4.3655e+00, -3.6448e+00, -2.2708e+00, ..., -1.4256e+00,
-9.2466e-01, -4.9591e-01],
[-3.3484e+00, -1.9823e+00, -2.8506e+00, ..., -3.1948e+00,
-3.1753e+00, -5.5762e-01],
...,
[ 5.4139e-01, -4.7001e+00, -5.6872e+00, ..., -1.4374e-01,
-2.6413e+00, -1.5007e+00],
[-2.7916e+00, -2.3930e+00, -2.0046e+00, ..., 3.3053e-01,
-2.9317e+00, -1.9945e+00],
[-2.7583e+00, -2.6539e+00, -1.0018e-01, ..., -7.9417e-01,
-2.5699e+00, -1.3610e+00]],
[[ 1.4913e+00, 1.1523e+00, 1.5861e+00, ..., 2.7898e+00,
2.2320e+00, 5.4101e-01],
[ 2.9963e+00, 3.1493e+00, 2.9939e+00, ..., 3.5747e+00,
2.0936e+00, 1.4187e+00],
[ 1.2665e+00, 1.1638e+00, 2.3716e+00, ..., 3.2177e+00,
2.1575e+00, 1.8228e+00],
...,
[ 2.7983e+00, 5.5779e+00, 2.0208e+00, ..., 1.9811e+00,
3.6168e-01, 2.5887e-01],
[ 5.0917e+00, 4.3537e+00, 3.1087e+00, ..., 1.8045e+00,
5.9842e-01, 1.2154e+00],
[ 1.3081e+00, 8.7916e-01, 2.1724e+00, ..., 9.2793e-01,
-2.3227e-01, -1.1896e-01]]],
[[[ 1.6510e-01, 7.8789e-01, -2.3814e+00, ..., 9.8058e-01,
-2.9822e-01, -1.1438e+00],
[ 1.2543e+00, 3.7333e-01, -3.4845e+00, ..., 9.4392e-01,
-2.4957e-02, -2.0200e+00],
[ 1.7671e+00, 4.2663e-01, -2.7920e+00, ..., 1.1742e+00,
-3.8857e-01, -2.6646e+00],
...,
[-1.0770e+00, -5.2198e+00, -1.2925e+00, ..., 1.6689e+00,
7.5327e-01, 3.5959e-01],
[-1.7202e+00, -1.1849e+00, -4.6213e-01, ..., 2.4512e+00,
2.3971e-02, 1.9317e-01],
[-2.0288e+00, -3.9250e+00, -2.6606e+00, ..., 3.6654e+00,
-4.6098e-01, -1.6195e+00]],
[[-1.2132e+00, -1.7503e+00, -5.0717e+00, ..., -5.3893e-02,
-4.6130e+00, -2.8682e+00],
[-2.6652e+00, -3.6954e+00, -5.9842e+00, ..., -2.3409e+00,
-7.5113e+00, -4.7966e+00],
[-2.8028e+00, -3.8417e+00, -7.0793e+00, ..., -2.4699e+00,
-6.3802e+00, -4.2172e+00],
...,
[-6.3118e+00, -7.8984e+00, -5.8458e+00, ..., 7.0313e-01,
-4.5327e+00, -2.7789e+00],
[-8.3198e+00, -8.9393e+00, -8.3512e+00, ..., 1.0482e+00,
-4.9910e+00, -4.3071e+00],
[-2.3073e+00, -4.2931e+00, -1.7242e+00, ..., 2.0880e+00,
-1.2586e+00, -2.0193e+00]],
[[ 9.6925e-01, 1.2022e+00, 1.5776e+00, ..., 1.4805e+00,
1.8135e+00, 7.4829e-01],
[ 8.7156e-01, 1.2121e+00, 2.2304e+00, ..., 2.7242e+00,
2.7406e+00, 8.6141e-01],
[ 2.8783e-01, 2.5636e+00, 1.8827e+00, ..., 3.9141e+00,
2.5865e+00, 1.4115e+00],
...,
[ 1.1117e+00, 1.8963e+00, -2.9270e+00, ..., -6.0829e+00,
1.6275e+00, -5.7519e-01],
[ 2.9312e-01, 8.6075e-01, -1.2739e+00, ..., -3.3317e+00,
1.9352e+00, -3.0648e-01],
[ 5.2740e-01, -1.0433e-01, -1.0478e+00, ..., -3.2590e+00,
7.8784e-01, -1.2767e-01]],
...,
[[-6.9973e-01, -1.7785e+00, -2.9924e+00, ..., -1.5697e+00,
-2.8925e+00, -1.3721e+00],
[-2.1512e+00, -3.8399e+00, -3.9460e+00, ..., -3.3697e+00,
-4.0563e+00, -4.1900e+00],
[-6.5099e-01, -1.5392e+00, -2.2937e+00, ..., -2.0972e+00,
-2.7592e+00, -3.8190e+00],
...,
[-1.8045e+00, -3.6881e+00, -5.9318e-01, ..., -1.9986e+00,
-4.0140e+00, -6.9448e-01],
[-2.0727e+00, -3.4301e+00, -3.0256e+00, ..., -2.1665e+00,
-5.9541e+00, 1.9293e-01],
[-2.3725e+00, -1.8512e+00, -3.5301e-01, ..., -4.1484e-01,
-3.2370e+00, 6.8661e-01]],
[[-1.7501e+00, -4.2505e+00, -3.0794e+00, ..., -2.8212e+00,
-3.0559e+00, -4.7017e-01],
[-4.5064e+00, -7.1221e+00, -5.1589e+00, ..., -3.9026e+00,
-5.4913e+00, -1.7303e+00],
[-4.2546e+00, -6.5332e+00, -3.8437e+00, ..., -3.6921e+00,
-2.8241e+00, -1.6190e+00],
...,
[-3.3601e+00, -1.4410e+00, 1.5160e+00, ..., 5.4805e-03,
-5.2441e-01, 1.0112e+00],
[-3.1145e+00, 1.3831e-01, -3.3976e-01, ..., 6.8639e-01,
-2.4059e+00, 2.1558e+00],
[-3.6372e+00, -7.5906e-01, -1.8162e-01, ..., 2.0402e+00,
-3.3654e+00, 3.5844e-01]],
[[ 9.7071e-01, 4.0578e-01, 8.5716e-01, ..., 5.6787e-01,
1.6487e+00, 5.0693e-01],
[ 1.8532e+00, 1.4300e+00, 2.7758e-01, ..., 1.2329e+00,
1.9178e+00, 2.9250e+00],
[ 1.9680e+00, 9.9088e-01, 1.1148e+00, ..., 3.1767e+00,
2.0305e+00, 9.1957e-01],
...,
[ 2.7592e+00, 2.5444e-01, 2.4943e+00, ..., 7.6782e-01,
2.4689e+00, 3.2758e-01],
[ 4.2175e+00, 3.7709e+00, 2.7221e+00, ..., 1.3169e+00,
1.9306e+00, 1.6403e+00],
[ 2.5832e+00, 1.9909e+00, -1.0621e-03, ..., 1.9714e+00,
4.9792e-01, 2.0350e-01]]]], device='cuda:0'),
tensor([[[[-2.7115e-01, 1.3329e+00, 1.9771e+00, ..., -1.0191e+00,
-3.2267e+00, -9.0630e-01],
[ 5.3265e-01, -6.9251e-01, 1.0122e+00, ..., 8.3749e-01,
-1.3820e+00, -8.2316e-01],
[-1.0400e+00, -3.0986e+00, -2.7067e+00, ..., 4.6891e-01,
-1.9935e+00, -9.1218e-01],
...,
[-5.4262e-01, -3.5111e-02, 1.4622e+00, ..., -2.2032e+00,
-2.8274e+00, -7.5138e-01],
[-1.5073e+00, 1.2232e-01, 1.1048e+00, ..., -3.1493e+00,
-4.4863e+00, -8.2452e-01],
[-1.0762e+00, -2.0895e+00, -1.2124e+00, ..., -3.7219e+00,
-3.8860e+00, -1.8395e+00]],
[[-1.6795e+00, -2.7177e+00, -4.0878e+00, ..., -4.8878e+00,
-8.9763e+00, -1.9622e+00],
[-5.1446e+00, -7.0742e+00, -5.3676e+00, ..., -5.5485e+00,
-9.0132e+00, -4.3468e+00],
[-4.2188e+00, -6.1951e+00, -6.5174e+00, ..., -4.3419e+00,
-5.7239e+00, -2.2942e+00],
...,
[-4.2850e+00, -5.0589e+00, -5.6117e+00, ..., -5.6164e+00,
-9.6477e+00, -4.2981e+00],
[-7.4411e+00, -7.7090e+00, -7.3622e+00, ..., -9.0303e+00,
-1.0532e+01, -4.0683e+00],
[-1.2668e+00, -2.9950e+00, -2.3216e+00, ..., -3.8739e+00,
-4.2926e+00, -6.9933e-01]],
[[-2.2634e-01, 2.1844e+00, 1.5469e+00, ..., 2.2408e+00,
2.5368e+00, 6.2818e-01],
[-3.8708e-01, -1.4386e-01, -2.7529e-01, ..., 1.7674e+00,
2.4369e+00, -4.2730e-01],
[-6.9948e-01, 1.5247e+00, 1.3850e+00, ..., 3.9883e-01,
2.1161e+00, -4.9225e-02],
...,
[-5.6511e-01, 5.1264e-01, 1.2208e+00, ..., 2.1393e+00,
-2.7921e-01, -9.7390e-01],
[-1.2149e+00, 8.6583e-02, 1.0071e+00, ..., -1.1378e+00,
-8.6384e-02, -4.5614e-01],
[-5.1185e-01, -3.2549e-01, 2.4656e-01, ..., 2.6356e-01,
-3.7986e-01, -2.0329e+00]],
...,
[[-1.3182e+00, -2.0514e+00, -5.4311e-01, ..., -3.4952e+00,
-5.3743e+00, -1.1884e+00],
[-3.4569e+00, -3.9887e+00, -1.8389e+00, ..., -2.9281e+00,
-5.8062e+00, -2.6160e+00],
[-3.0850e+00, -2.9154e+00, -8.5864e-01, ..., -2.1456e+00,
-5.5367e+00, -2.1929e+00],
...,
[-1.0777e+00, -2.4890e+00, -5.3000e-01, ..., -1.9357e+00,
-6.9101e+00, -4.0066e+00],
[-1.9926e+00, -3.7889e+00, -6.5117e-01, ..., -1.5329e+00,
-3.7298e+00, -2.3834e-01],
[-1.1650e+00, -1.3509e+00, 2.2744e-01, ..., -1.3578e+00,
-2.0317e+00, 9.4273e-01]],
[[-1.9781e+00, -1.3457e+00, -1.8084e+00, ..., -2.0970e+00,
-9.8653e-01, 1.5490e-01],
[-4.3156e+00, -1.9830e+00, -6.6060e-01, ..., -2.0853e+00,
-2.8045e+00, -5.5864e-02],
[-3.8330e+00, -1.8868e+00, -1.5189e+00, ..., 8.9220e-02,
-2.2570e+00, 3.5246e-01],
...,
[-2.5987e+00, -2.9192e+00, -3.3227e+00, ..., -2.0831e+00,
-8.6896e-01, 1.3287e+00],
[-1.8809e+00, -1.7549e-01, -1.4865e+00, ..., -5.9454e-01,
-9.4144e-01, 1.6601e+00],
[-1.9783e+00, -1.3294e+00, -6.0554e-01, ..., -6.0364e-01,
-1.0434e+00, 6.3557e-01]],
[[ 1.2781e+00, 5.8633e-01, 1.3602e+00, ..., 1.7072e+00,
1.7726e+00, 1.1561e-01],
[ 4.3374e+00, 3.7204e+00, 2.2717e+00, ..., 3.7199e+00,
3.5108e+00, 4.0711e+00],
[ 2.0504e+00, 3.0755e-01, -7.0813e-01, ..., 2.3888e+00,
1.2142e+00, 1.6397e+00],
...,
[ 3.0332e+00, 2.9796e+00, 3.2569e+00, ..., 1.8488e+00,
3.0800e+00, 2.2772e+00],
[ 3.6760e+00, 3.9900e+00, 2.9943e+00, ..., 2.9210e+00,
8.4989e-01, 2.4574e+00],
[ 2.3844e+00, 2.2802e+00, 1.6041e+00, ..., 4.7357e-01,
-1.2102e-01, 1.3547e+00]]],
[[[ 6.9636e-01, 1.5540e+00, 3.2267e-02, ..., -1.8525e+00,
-2.1817e+00, 3.2708e-02],
[ 7.7666e-01, 2.3553e-01, 4.6931e-01, ..., 7.9875e-02,
-1.4760e+00, 3.5653e-01],
[-2.7469e+00, -5.3959e+00, -7.2915e+00, ..., 8.9324e-01,
-7.8121e-01, 8.6328e-01],
...,
[-6.6237e-01, 3.8964e-01, -6.0310e-01, ..., -2.6078e+00,
-2.0847e+00, 9.8343e-02],
[-1.9263e+00, -1.5152e+00, 2.3117e-01, ..., -2.8767e+00,
-2.2929e+00, 3.2540e-01],
[-2.1855e+00, -3.9846e+00, -3.3455e+00, ..., -4.0111e+00,
-3.3442e+00, -1.3497e+00]],
[[-2.7213e-01, -1.9513e+00, -4.1685e+00, ..., 4.0432e-01,
-1.2311e+00, 3.7078e-01],
[-1.5630e+00, -2.3522e+00, -4.8649e+00, ..., 2.6898e+00,
-2.0304e-01, -4.9195e-01],
[-8.6496e+00, -1.1485e+01, -1.2784e+01, ..., -2.3046e+00,
3.8973e-01, 3.5025e-01],
...,
[-5.4601e+00, -8.1496e+00, -9.6830e+00, ..., -8.4679e+00,
-8.0835e+00, -1.7423e+00],
[-9.8195e+00, -9.8942e+00, -9.1066e+00, ..., -1.0122e+01,
-1.0742e+01, -3.5820e+00],
[-2.5657e+00, -5.0421e+00, -4.5594e+00, ..., -4.3741e+00,
-4.1597e+00, -5.7246e-01]],
[[ 8.7507e-01, 1.6854e+00, 1.9162e+00, ..., 1.8810e+00,
1.3636e+00, 4.4334e-01],
[ 9.9779e-01, 2.5562e+00, 2.9191e+00, ..., 1.4516e+00,
2.3976e+00, -2.7020e-01],
[ 1.8386e+00, 5.1888e+00, 8.4300e+00, ..., -1.7277e+00,
1.2333e+00, 5.4492e-01],
...,
[ 9.9836e-02, -1.3417e+00, -3.0281e+00, ..., -2.9549e-01,
1.2258e+00, -1.2388e-01],
[-1.5551e+00, -3.2405e-01, -1.0915e+00, ..., -9.3901e-02,
1.0425e+00, -3.2014e-01],
[ 5.6972e-01, 1.3343e+00, 1.0163e+00, ..., 7.8250e-03,
-6.1548e-01, -1.9525e+00]],
...,
[[-7.4712e-01, -8.2627e-01, -2.4542e+00, ..., -1.3680e+00,
-1.4149e+00, 2.9716e-01],
[-2.8342e+00, -1.6227e+00, -7.8803e-01, ..., -9.9999e-01,
-1.9503e+00, -1.5790e+00],
[-3.2308e+00, -3.6564e+00, -5.7177e+00, ..., -2.9513e+00,
-1.2859e+00, -7.8187e-01],
...,
[-2.8199e+00, -5.6964e+00, -5.2636e+00, ..., -4.7914e+00,
-6.5651e+00, -2.5815e+00],
[-3.4348e+00, -5.7282e+00, -2.5913e+00, ..., -4.4072e+00,
-5.4327e+00, -1.2059e+00],
[-3.4043e+00, -3.7082e+00, -3.1661e+00, ..., -2.8218e+00,
-2.4669e+00, 4.1746e-01]],
[[-1.3787e+00, -2.6463e+00, -4.5263e+00, ..., -4.9266e+00,
-4.1529e+00, -1.3420e+00],
[-3.9689e+00, -4.6823e+00, -3.9777e+00, ..., -2.9965e+00,
-4.9797e+00, -3.5238e+00],
[-7.6150e+00, -1.1532e+01, -7.3842e+00, ..., 2.1569e-01,
-3.7143e-01, -5.3604e-01],
...,
[-2.9017e+00, -1.6737e+00, -3.7269e+00, ..., -3.8346e+00,
-2.0939e+00, -1.2430e+00],
[-2.5925e+00, 6.8065e-01, 1.2395e-01, ..., -1.7033e+00,
-7.0537e-01, 3.4584e-01],
[-3.6986e+00, -2.9900e+00, -2.9670e+00, ..., -1.1367e+00,
-9.0108e-01, 3.4666e-01]],
[[ 1.0611e+00, 6.2143e-01, 1.4625e+00, ..., -2.1228e+00,
-2.4077e+00, -9.5945e-01],
[ 1.5724e+00, 4.9693e-01, 2.2484e+00, ..., -3.5206e+00,
-4.1457e+00, -2.5048e+00],
[ 1.6796e+00, 1.5327e+00, 3.8907e+00, ..., -6.8867e-02,
-3.9395e+00, -2.8700e+00],
...,
[ 3.4728e+00, 5.4196e+00, 5.1553e+00, ..., 3.5214e+00,
3.0207e+00, 6.0646e-01],
[ 3.8990e+00, 3.2012e+00, 1.3111e+00, ..., 2.7368e+00,
1.5181e+00, 1.4161e+00],
[ 3.1125e+00, 3.0463e+00, 2.7004e+00, ..., 2.0032e-02,
4.9860e-01, 1.0089e+00]]],
[[[ 5.1778e-02, 8.1745e-01, -2.3737e-01, ..., -1.2410e+00,
-9.5512e-01, -8.6433e-01],
[ 7.1003e-01, 4.8374e-01, 2.8162e-02, ..., 4.7640e-01,
7.7885e-01, -1.0719e+00],
[ 1.7813e+00, 4.4212e-01, 1.1235e+00, ..., -1.9653e+00,
-4.8161e-01, -1.6514e+00],
...,
[-1.7909e+00, -1.8201e+00, -3.7721e-01, ..., -3.5793e+00,
-3.5669e+00, 8.0996e-01],
[-2.5878e+00, -1.6772e+00, -1.0168e+00, ..., -4.6115e-02,
4.6762e-01, 1.8364e+00],
[-2.0811e+00, -4.2213e+00, -2.9827e+00, ..., -1.3975e+00,
-1.6985e+00, -1.3904e+00]],
[[-6.3477e-01, -1.3927e+00, -2.4299e+00, ..., -2.7923e+00,
-4.6479e+00, -2.3976e+00],
[-1.8714e+00, -3.6983e+00, -4.0274e+00, ..., -3.0310e+00,
-5.3436e+00, -5.1503e+00],
[-1.6263e+00, -3.4202e+00, -3.8418e+00, ..., -2.5778e+00,
-3.3138e+00, -4.0381e+00],
...,
[-7.3247e+00, -6.9730e+00, -4.9334e+00, ..., -5.6269e+00,
-6.2347e+00, -5.7107e-01],
[-8.9560e+00, -9.1095e+00, -6.8253e+00, ..., -8.9192e-01,
-2.2326e+00, -1.3222e+00],
[-2.2336e+00, -4.3074e+00, -3.5520e+00, ..., -2.2279e+00,
-3.5770e+00, -1.2414e+00]],
[[ 6.1443e-01, 1.1963e+00, 1.0621e+00, ..., 1.6641e+00,
1.8582e+00, 8.6049e-01],
[ 8.6463e-01, 1.5197e+00, 1.6004e+00, ..., 1.1178e+00,
2.1050e+00, 1.0058e+00],
[ 3.7348e-01, 3.2353e+00, 1.7343e+00, ..., 2.8999e+00,
2.7834e+00, 1.9297e+00],
...,
[ 2.1307e-01, 7.3110e-02, 1.2436e+00, ..., 7.6647e-01,
4.4145e-01, -1.6447e+00],
[ 2.9206e-01, -4.6489e-01, -4.9104e-01, ..., -3.0121e+00,
-1.9507e+00, -1.7408e+00],
[ 6.7030e-01, 4.7928e-01, -2.5907e-02, ..., -3.7532e+00,
-3.0101e+00, -2.2068e+00]],
...,
[[-8.3053e-01, -1.2887e+00, -2.0405e+00, ..., -2.7599e+00,
-3.2640e+00, -1.2918e+00],
[-2.2784e+00, -3.2767e+00, -2.7406e+00, ..., -2.1858e+00,
-3.7685e+00, -3.6160e+00],
[-4.6311e-01, -1.5317e+00, -6.9472e-01, ..., -2.5543e+00,
-3.6677e+00, -2.8134e+00],
...,
[-2.4880e+00, -5.9264e-01, -9.1345e-02, ..., -1.3811e+00,
-2.1024e+00, -1.1576e+00],
[-3.2848e+00, -3.4268e+00, -1.6327e+00, ..., 2.0046e+00,
-2.8683e-01, 5.4635e-01],
[-2.4658e+00, -2.6702e+00, -1.5872e+00, ..., -1.3577e+00,
-2.7177e+00, -7.7777e-01]],
[[-1.4410e+00, -2.3888e+00, -3.5932e+00, ..., -3.0296e+00,
-2.9952e+00, -5.8954e-01],
[-4.2996e+00, -4.7391e+00, -4.6880e+00, ..., -3.3693e+00,
-5.4690e+00, -2.0371e+00],
[-3.9228e+00, -5.2135e+00, -3.4282e+00, ..., -2.1322e+00,
-3.6079e+00, -1.8318e+00],
...,
[-3.3380e+00, -1.4170e+00, -3.3915e+00, ..., -1.0845e+00,
-7.9997e-02, 7.6420e-01],
[-3.3146e+00, 8.1585e-01, -8.8590e-01, ..., 2.5721e+00,
1.6792e+00, 2.4511e+00],
[-3.8854e+00, -1.7118e+00, -1.0365e+00, ..., 1.5135e-01,
-6.4678e-01, 6.9261e-01]],
[[ 9.0047e-01, -1.1869e-01, 7.3884e-01, ..., 3.9969e-01,
1.4200e+00, 5.8608e-01],
[ 1.5592e+00, 8.0437e-01, 5.2413e-01, ..., 7.0195e-01,
1.8358e+00, 3.2073e+00],
[ 1.5958e+00, 3.8860e-01, 1.1154e+00, ..., -3.8183e-02,
9.3650e-01, 1.0759e+00],
...,
[ 2.7539e+00, 1.2877e+00, -1.4069e+00, ..., 6.5249e-01,
1.7761e+00, 1.6304e+00],
[ 4.5257e+00, 4.7048e+00, 2.6405e+00, ..., 1.3137e+00,
1.0568e+00, 1.8671e+00],
[ 2.3620e+00, 1.9819e+00, 1.0703e+00, ..., -1.0565e+00,
-7.8365e-01, 3.9144e-01]]],
[[[-9.9290e-01, -3.8050e+00, -2.0078e+00, ..., -1.9750e+00,
-3.7823e+00, -1.7959e+00],
[ 4.3627e-01, -3.1662e+00, -3.0106e-01, ..., -3.9607e-01,
-1.4535e+00, -5.9702e-01],
[ 8.0905e-01, -7.1075e-01, -6.0362e-01, ..., -1.4786e-01,
-1.4388e+00, -1.5510e+00],
...,
[-1.0531e+00, -2.3855e+00, 4.1516e-01, ..., 2.9833e+00,
-1.3673e+00, -2.6243e+00],
[-1.0248e+00, -1.0963e+00, -1.5642e-02, ..., 1.5598e+00,
-1.0826e+00, -1.7123e+00],
[-2.6919e+00, -5.2293e+00, -4.5129e+00, ..., -3.5947e+00,
-4.4252e+00, -4.7242e+00]],
[[-1.2153e+00, -4.1834e+00, -1.0001e+00, ..., -5.5514e+00,
-8.7806e+00, -2.3297e+00],
[-3.7904e+00, -5.9478e+00, -5.4326e+00, ..., -5.1960e+00,
-7.6565e+00, -2.7729e+00],
[-3.3490e+00, -5.2210e+00, -9.0203e+00, ..., -1.2253e+01,
-1.2559e+01, -6.1072e+00],
...,
[-5.6983e+00, -5.2797e+00, -2.6179e+00, ..., -4.5607e+00,
-9.7939e+00, -6.6001e+00],
[-7.1252e+00, -7.9425e+00, -6.2877e+00, ..., -3.8378e+00,
-7.7361e+00, -6.0699e+00],
[-2.5218e+00, -4.4987e+00, -3.9465e+00, ..., -4.8424e+00,
-5.1126e+00, -3.0234e+00]],
[[ 9.8635e-01, 2.1196e+00, -1.5049e-01, ..., 2.4446e+00,
3.0681e+00, 1.0894e+00],
[ 1.2929e-01, 1.2559e+00, -4.2722e+00, ..., 1.5019e+00,
2.8542e+00, 4.6609e-01],
[-1.5356e-01, 2.1895e+00, 1.9958e+00, ..., -1.8233e-01,
1.2828e-01, -3.8259e-01],
...,
[ 2.6651e-01, 9.5114e-01, -2.8854e-01, ..., -3.5652e-01,
-5.6116e-01, 4.0614e-01],
[ 7.9937e-01, 9.8341e-01, 2.9861e-01, ..., 2.9394e-01,
8.3576e-01, 3.8161e-01],
[ 1.2964e+00, 1.5688e+00, 1.3026e+00, ..., 9.0296e-01,
5.9819e-01, -2.4910e-01]],
...,
[[-1.7275e+00, -2.9906e+00, -4.7884e-02, ..., -4.6355e+00,
-6.1377e+00, -1.9424e+00],
[-3.0702e+00, -4.3742e+00, -3.0254e+00, ..., -3.7066e+00,
-4.9485e+00, -2.4678e+00],
[-1.3046e+00, -2.4873e+00, -4.0377e+00, ..., -5.7592e+00,
-6.5196e+00, -3.5492e+00],
...,
[-2.0410e+00, -2.0703e+00, 1.0600e+00, ..., -6.5237e+00,
-7.6523e+00, -4.1526e+00],
[-2.8997e+00, -3.2887e+00, -1.4851e+00, ..., -2.9394e+00,
-4.4378e+00, -2.9760e+00],
[-3.5053e+00, -3.9097e+00, -3.1996e+00, ..., -3.8814e+00,
-4.4499e+00, -2.0372e+00]],
[[-2.3165e+00, 4.0494e-01, 8.0104e-01, ..., -2.5715e+00,
-9.7148e-01, 6.3308e-01],
[-5.0614e+00, -1.9253e+00, 3.4059e+00, ..., -3.2188e+00,
-2.1894e+00, -6.7061e-02],
[-3.3841e+00, -3.3952e+00, -2.5173e+00, ..., -2.2286e+00,
-1.4253e+00, 3.0088e-01],
...,
[-3.2024e+00, -1.4376e+00, -1.4663e+00, ..., -8.2288e+00,
-9.2999e+00, -1.6017e+00],
[-3.6196e+00, -7.1766e-01, -1.5632e+00, ..., -1.5805e+00,
-4.0684e+00, -1.0492e-02],
[-4.4112e+00, -2.9959e+00, -2.4190e+00, ..., -2.3969e+00,
-2.5119e+00, 3.3432e-01]],
[[ 1.5242e+00, 8.5431e-01, -6.7261e-01, ..., 1.1168e+00,
1.6164e+00, -1.9106e-01],
[ 2.7084e+00, 5.7276e-01, 1.6672e+00, ..., 2.8019e+00,
3.3618e+00, 3.9598e+00],
[ 2.3295e+00, 6.0311e-01, 3.6437e+00, ..., 4.8660e+00,
3.8248e+00, 1.8771e+00],
...,
[ 2.1180e+00, 5.0406e-01, 2.1615e-01, ..., 3.2716e+00,
3.7463e+00, 2.7825e+00],
[ 3.7671e+00, 3.8591e+00, 2.4868e+00, ..., 1.7741e+00,
1.1577e+00, 2.9632e+00],
[ 1.9777e+00, 7.3817e-01, 8.8967e-02, ..., 6.3958e-01,
8.5355e-01, 1.2848e+00]]],
[[[-4.8379e-02, 7.3601e-01, 1.5204e-01, ..., -1.3068e+00,
-1.0027e+00, -8.2195e-01],
[ 1.4272e+00, 3.4207e-01, 7.8025e-01, ..., -4.0036e-01,
1.1496e-01, -1.0688e+00],
[ 5.9261e-02, -2.5613e+00, -3.8808e+00, ..., -7.1141e-01,
-4.3028e-01, -1.9236e+00],
...,
[-3.8349e-01, -1.9931e-01, 3.9850e-01, ..., 7.1246e-01,
-1.2252e+00, -2.3396e-01],
[-2.3949e+00, -1.6593e+00, 1.5207e+00, ..., -2.3093e+00,
-1.9211e+00, 1.6896e-01],
[-2.3144e+00, -4.2764e+00, -2.9859e+00, ..., -3.4807e+00,
-3.3311e+00, -1.6396e+00]],
[[-6.5871e-01, -1.5772e+00, -2.5633e+00, ..., -2.9602e+00,
-4.5772e+00, -2.3936e+00],
[-2.6382e+00, -4.2888e+00, -4.3523e+00, ..., -3.2213e+00,
-4.9995e+00, -4.5196e+00],
[-3.3547e+00, -6.0455e+00, -6.7656e+00, ..., -2.3776e+00,
-3.9076e+00, -3.6232e+00],
...,
[-6.8641e+00, -8.8781e+00, -6.3385e+00, ..., -5.9680e+00,
-8.1629e+00, -1.8206e+00],
[-9.6757e+00, -1.0132e+01, -8.5595e+00, ..., -1.0770e+01,
-1.0101e+01, -3.4885e+00],
[-2.5311e+00, -4.9600e+00, -4.1770e+00, ..., -4.0821e+00,
-3.8842e+00, -6.4091e-01]],
[[ 6.9446e-01, 1.1855e+00, 1.1504e+00, ..., 1.5325e+00,
1.8013e+00, 8.7258e-01],
[ 4.4533e-01, 1.0157e+00, 8.4956e-01, ..., 1.3642e+00,
2.1763e+00, 8.4652e-01],
[-1.2268e-01, 3.4117e+00, 4.2646e+00, ..., 1.2292e+00,
2.4830e+00, 1.9605e+00],
...,
[ 4.7649e-02, -2.8289e-01, -1.1364e+00, ..., 1.3897e+00,
8.9576e-01, -2.1791e-01],
[-5.9362e-01, 5.3216e-01, -1.1587e+00, ..., 5.4402e-01,
-1.5820e-01, -6.2938e-01],
[ 5.2809e-01, 1.2648e+00, 1.2466e+00, ..., -1.0012e+00,
-1.5910e+00, -1.8863e+00]],
...,
[[-8.5839e-01, -1.1320e+00, -1.9941e+00, ..., -2.5325e+00,
-3.2427e+00, -1.2105e+00],
[-2.1056e+00, -3.9335e+00, -2.8412e+00, ..., -2.7914e+00,
-4.3003e+00, -3.5217e+00],
[-1.7296e+00, -3.6214e+00, -3.4637e+00, ..., -2.3048e+00,
-3.8954e+00, -3.0706e+00],
...,
[-2.7525e+00, -3.5017e+00, -1.4037e+00, ..., -6.0026e+00,
-5.8707e+00, -2.7926e+00],
[-3.7951e+00, -5.7517e+00, -2.0720e+00, ..., -5.1886e+00,
-4.5096e+00, -1.5708e+00],
[-3.2318e+00, -3.9413e+00, -3.3044e+00, ..., -2.7719e+00,
-2.9071e+00, 1.9254e-01]],
[[-1.5813e+00, -2.1062e+00, -3.0394e+00, ..., -2.8002e+00,
-2.7493e+00, -5.6022e-01],
[-4.3156e+00, -4.2179e+00, -3.7938e+00, ..., -4.0608e+00,
-5.7271e+00, -1.9463e+00],
[-3.8454e+00, -5.6327e+00, -5.1097e+00, ..., -2.8663e+00,
-4.6516e+00, -2.0900e+00],
...,
[-3.9705e+00, -2.3705e+00, -4.8035e+00, ..., -5.1901e+00,
-2.9281e+00, -5.6135e-01],
[-3.5227e+00, 1.7023e-01, -1.4095e+00, ..., -2.7207e+00,
-1.8266e-01, 9.5117e-01],
[-3.7752e+00, -2.9393e+00, -2.6175e+00, ..., -1.0579e+00,
-9.6859e-01, 4.8714e-01]],
[[ 8.7944e-01, -1.0436e-01, 7.8509e-01, ..., 5.1513e-01,
1.3899e+00, 5.9111e-01],
[ 2.1376e+00, 9.4848e-01, 1.1797e+00, ..., 8.2831e-01,
1.8685e+00, 3.1949e+00],
[ 1.5830e+00, -6.8515e-02, -5.3807e-01, ..., 4.9354e-01,
1.0436e+00, 8.9411e-01],
...,
[ 4.2507e+00, 4.0886e+00, 3.1207e+00, ..., 2.7964e+00,
2.6950e+00, 4.9502e-01],
[ 4.5435e+00, 4.4503e+00, 3.6293e+00, ..., 2.8322e+00,
2.2374e+00, 1.3889e+00],
[ 2.8024e+00, 2.7355e+00, 2.3517e+00, ..., 4.2465e-01,
1.0800e+00, 1.0793e+00]]]], device='cuda:0'),
tensor([[[[ 2.6795e-01, 1.4066e+00, 5.0277e-01, ..., -1.1810e+00,
5.4490e-02, 4.3847e-01],
[ 4.1034e-01, 3.4235e-01, 4.0247e-01, ..., -2.2437e+00,
2.5417e+00, 1.2622e+00],
[ 7.2369e-02, 1.2912e-01, 1.3194e+00, ..., 1.7693e-01,
1.2121e+00, 6.1290e-01],
...,
[ 3.3998e-01, -1.8407e+00, -1.5444e+00, ..., 1.1066e+00,
1.3725e+00, -6.8531e-01],
[-1.2372e+00, -1.1391e+00, -1.1404e+00, ..., 6.0724e-01,
2.9179e-01, -9.2855e-01],
[ 1.8417e-02, -1.9703e+00, -1.6136e+00, ..., 2.8982e+00,
5.2451e-01, 1.7280e-01]],
[[-8.2315e-01, -2.3190e+00, -4.1197e+00, ..., -1.4443e+00,
-2.1348e+00, 5.5531e-01],
[-4.7389e+00, -6.0554e+00, -4.9655e+00, ..., -3.6464e+00,
-1.0526e+00, -5.5048e-01],
[-3.5512e+00, -5.1362e+00, -6.6275e+00, ..., 1.4557e+00,
1.1693e+00, -2.1571e-01],
...,
[-4.7775e+00, -5.2139e+00, -4.9788e+00, ..., 1.6345e+00,
-1.3145e+00, -2.5402e+00],
[-7.4700e+00, -8.1156e+00, -7.6051e+00, ..., -2.3893e+00,
-5.2583e+00, -3.4985e+00],
[-1.1405e+00, -3.3763e+00, -1.4509e+00, ..., 1.3482e+00,
-1.4390e+00, -3.6491e-01]],
[[-3.1050e-01, 2.1770e+00, 1.1797e+00, ..., -1.5848e+00,
1.6913e+00, 2.1220e+00],
[-5.5258e-02, 9.8847e-01, 7.6539e-01, ..., -5.1475e+00,
-2.6116e+00, 2.1600e+00],
[-7.2499e-02, 1.9026e+00, 1.6717e+00, ..., -2.7163e+00,
-5.4889e+00, -2.5922e+00],
...,
[ 2.9431e-01, 7.1739e-01, 1.2856e+00, ..., -1.5187e+00,
3.2984e+00, 1.6207e+00],
[ 1.1926e-01, -1.8835e-01, 1.4178e-01, ..., -4.1046e-01,
3.1415e+00, 1.4872e+00],
[-6.0172e-01, -3.4398e-01, -9.4394e-01, ..., -6.2066e-01,
5.8993e-01, -2.0242e-01]],
...,
[[-8.3749e-01, -1.9127e+00, -1.6854e+00, ..., 6.9394e-01,
-3.8394e-01, -1.8117e+00],
[-3.0908e+00, -4.3083e+00, -2.1704e+00, ..., -3.7785e-01,
-1.6826e+00, -3.9318e+00],
[-1.4137e+00, -2.5364e+00, -1.0792e+00, ..., 7.5254e-01,
-9.6945e-01, -4.9045e-01],
...,
[-1.0826e+00, -1.8764e+00, -1.4898e+00, ..., -1.2468e-01,
-2.8196e+00, -1.4779e+00],
[-2.6668e+00, -3.2858e+00, -2.7817e+00, ..., -3.6741e+00,
-5.8274e+00, -1.3840e+00],
[-8.3416e-01, -1.0196e+00, 9.4210e-01, ..., -8.1971e-01,
-2.1368e+00, 8.8273e-01]],
[[-1.7323e+00, -2.2506e+00, -2.7683e+00, ..., -1.3093e-02,
-2.5900e+00, 1.0480e+00],
[-4.3769e+00, -3.3416e+00, -2.5998e+00, ..., 3.4084e-02,
-2.3918e+00, -6.5469e-01],
[-2.6957e+00, -1.6398e+00, -2.2432e+00, ..., 9.4652e-01,
1.7749e+00, 7.3743e-01],
...,
[-4.1203e+00, -3.4793e+00, -6.7991e-01, ..., 5.9816e-01,
-1.0831e+00, -1.9661e-01],
[-4.3903e+00, -2.0046e+00, -2.4147e+00, ..., -1.4688e+00,
-2.5745e+00, 6.3571e-02],
[-2.9293e+00, -1.4033e+00, -5.0864e-01, ..., 1.7829e+00,
-1.5386e+00, 4.1847e-01]],
[[ 1.0879e+00, 5.3854e-01, 7.3590e-01, ..., -1.6995e+00,
1.0877e+00, 1.6567e+00],
[ 3.5795e+00, 3.4267e+00, 2.0612e+00, ..., -2.1088e+00,
2.0106e+00, 4.9198e+00],
[ 1.6804e+00, 1.3428e+00, 1.2883e+00, ..., -2.1799e+00,
1.3906e+00, 2.1863e+00],
...,
[ 2.7838e+00, 1.2003e+00, 1.7698e+00, ..., 8.8761e-01,
2.9043e+00, 5.4545e-01],
[ 3.3115e+00, 3.5690e+00, 2.4540e+00, ..., 2.8376e+00,
2.8216e+00, 1.5812e+00],
[ 2.3831e+00, 2.1420e+00, 7.9488e-01, ..., 2.3265e+00,
1.0503e+00, 9.8698e-01]]],
[[[-9.0239e-01, -2.2385e+00, -3.5394e+00, ..., -3.8525e+00,
-9.6148e+00, -3.1433e+00],
[-2.0613e+00, -2.9595e+00, 4.8142e-01, ..., -6.1256e+00,
-2.0672e+00, 2.9106e+00],
[-5.9882e+00, -3.6816e+00, -7.3036e-01, ..., -2.8126e-01,
1.5136e+00, 1.5398e+00],
...,
[-6.7974e-01, 2.5161e+00, 1.3033e+00, ..., -2.0823e+00,
-7.3880e-01, 2.0913e+00],
[-1.8774e+00, 6.7254e-01, 1.6127e+00, ..., -1.1036e+00,
2.7459e+00, 1.9240e+00],
[-1.2013e+00, -2.1275e+00, -1.3300e+00, ..., -4.0517e+00,
-3.1059e+00, -2.2564e+00]],
[[-5.0543e+00, -5.3412e+00, -7.9067e+00, ..., -1.0554e+01,
-1.1775e+01, -2.7962e+00],
[-8.3482e+00, -5.0830e+00, -2.4777e+00, ..., -9.1657e+00,
-1.3647e+00, 1.7922e+00],
[-4.6654e+00, 5.5969e-02, -7.7089e-01, ..., 8.7405e-02,
2.6432e-01, -1.2995e+00],
...,
[-4.4005e+00, -5.6979e+00, -5.5063e+00, ..., -5.0824e+00,
-1.3402e-01, 2.2573e+00],
[-7.2698e+00, -7.3136e+00, -7.5268e+00, ..., -5.9267e+00,
-4.1660e+00, 1.8108e-01],
[-1.3290e+00, -3.2847e+00, -2.3311e+00, ..., -4.1116e+00,
-6.7337e+00, -5.6388e+00]],
[[-5.4143e-01, 2.9347e+00, 3.9251e+00, ..., 7.6686e-01,
2.0464e+00, 2.4231e+00],
[ 2.4744e-01, 1.6444e+00, 1.9996e+00, ..., 4.6689e+00,
2.0104e-01, -9.8456e-01],
[ 2.8537e+00, -2.2038e+00, -4.5284e+00, ..., 1.1803e+00,
-5.3263e+00, -3.4104e+00],
...,
[-1.1836e+00, -1.0192e+00, -1.5662e+00, ..., -5.9571e+00,
-5.9219e+00, -3.2913e+00],
[-1.9657e+00, 1.1301e-01, -4.1430e-01, ..., -3.1573e+00,
-5.2777e+00, -1.9074e+00],
[-9.5771e-01, -2.5426e-01, 7.4453e-01, ..., -9.8665e-01,
-4.0753e-01, -2.3217e-01]],
...,
[[-2.6833e+00, -4.0761e+00, -5.2234e+00, ..., -6.7086e+00,
-6.5108e+00, -1.0517e+00],
[-4.5734e+00, -2.3736e+00, -9.3963e-01, ..., -4.4441e+00,
2.4060e+00, 1.6675e+00],
[-6.1544e+00, 2.9378e-01, -3.7685e-01, ..., 1.7187e+00,
5.7220e-01, 4.6222e-01],
...,
[-1.0984e+00, -1.3663e+00, -4.6326e-01, ..., -5.7874e+00,
-1.8861e+00, -1.1241e+00],
[-1.9477e+00, -4.2678e+00, -6.0822e-01, ..., -2.7309e+00,
-1.7408e+00, -2.9327e+00],
[-1.3895e+00, -1.8316e+00, 2.7237e-01, ..., -4.3520e+00,
-7.8147e+00, -5.5607e+00]],
[[-1.7178e+00, -3.1574e+00, -5.4951e+00, ..., -3.6227e+00,
-9.7760e-01, 6.8598e-01],
[-2.9176e+00, -1.6428e-01, -4.1495e+00, ..., -2.5360e+00,
7.6119e-02, 1.1713e+00],
[-4.4347e+00, -2.2358e+00, -2.3158e+00, ..., -1.3058e+00,
-3.1969e+00, -5.6484e-01],
...,
[-1.6829e+00, -1.9354e+00, -4.7729e+00, ..., -3.0405e+00,
-2.8978e+00, -1.3629e+00],
[-1.1577e+00, 4.9911e-01, -1.0187e+00, ..., 2.1121e-01,
3.3222e+00, 1.1691e+00],
[-1.7167e+00, -1.2320e+00, -6.0357e-01, ..., -2.4485e+00,
-1.5070e+00, 1.1498e-01]],
[[ 2.3152e+00, -4.8573e-01, 4.5981e-01, ..., 2.2526e+00,
1.1389e+00, 1.4904e+00],
[ 5.0966e+00, 2.7648e+00, 4.3903e+00, ..., 4.4201e+00,
3.9325e+00, 3.7228e+00],
[-1.4760e-01, -3.9671e-01, -3.6490e+00, ..., 3.0883e+00,
1.5025e+00, -4.9502e-01],
...,
[ 2.1905e+00, 3.0821e+00, 2.9667e+00, ..., -5.1742e-01,
-2.4882e+00, -1.2444e+00],
[ 3.1027e+00, 3.9675e+00, 1.6366e+00, ..., 2.0011e+00,
4.5462e+00, -1.3959e-02],
[ 2.3980e+00, 2.5112e+00, 1.6931e+00, ..., 2.2597e+00,
4.2996e+00, 2.1041e+00]]],
[[[ 5.0830e-02, 1.5029e+00, 8.1432e-01, ..., -3.3239e-02,
4.8354e-01, -1.1962e-01],
[ 3.5350e-01, 7.6654e-01, 1.8758e+00, ..., -3.0971e-01,
3.8323e-01, -4.2226e-01],
[-6.6629e-01, 4.3229e-01, 9.8100e-01, ..., -5.0253e-01,
-4.9559e-02, -7.9267e-01],
...,
[-1.1596e-01, 1.2947e-01, 4.5701e-01, ..., 1.7156e-01,
7.1012e-01, -1.9804e-01],
[-1.6757e+00, -3.4423e-01, 1.0754e-01, ..., 2.1953e-01,
3.8062e-01, -1.0292e+00],
[-1.7451e+00, -3.0322e+00, -1.9880e+00, ..., -3.1951e+00,
-2.8553e+00, -2.6390e+00]],
[[-6.7066e-01, -1.5543e+00, -3.4448e+00, ..., -2.1753e+00,
-3.3808e+00, -2.0852e+00],
[-3.8839e+00, -4.1494e+00, -3.8578e+00, ..., -3.5288e+00,
-3.9291e+00, -3.3663e+00],
[-3.0855e+00, -4.0253e+00, -4.9747e+00, ..., -3.1428e+00,
-3.5815e+00, -2.6630e+00],
...,
[-4.7958e+00, -5.6546e+00, -5.7651e+00, ..., -3.0681e+00,
-3.0329e+00, -2.3634e+00],
[-8.3991e+00, -8.4107e+00, -6.5065e+00, ..., -4.7978e+00,
-5.1301e+00, -4.8036e+00],
[-2.0086e+00, -3.8886e+00, -2.4854e+00, ..., -2.9910e+00,
-2.8118e+00, -1.5060e+00]],
[[-1.9281e-01, 2.4310e+00, 1.6407e+00, ..., 1.5416e+00,
1.6738e+00, 9.0560e-01],
[-6.3187e-01, 2.0802e-01, 8.0521e-01, ..., 6.6417e-01,
1.5107e+00, 8.2203e-01],
[-7.0440e-02, 1.7738e+00, 1.3669e+00, ..., 1.6358e+00,
2.0428e+00, 1.4677e+00],
...,
[-1.0542e+00, 9.5936e-01, 1.2814e+00, ..., 1.7269e+00,
9.5086e-01, 6.9610e-01],
[-1.2789e+00, -2.5304e-01, 8.3063e-01, ..., 1.4731e+00,
8.2570e-01, 4.1611e-01],
[ 6.2161e-02, -1.0633e-01, 2.3467e-01, ..., 1.7144e+00,
1.5148e+00, 4.7025e-01]],
...,
[[-8.6585e-01, -1.7494e+00, -1.9052e+00, ..., -1.5089e+00,
-1.9533e+00, -4.4675e-01],
[-2.8295e+00, -3.5760e+00, -2.1249e+00, ..., -2.6963e+00,
-3.9494e+00, -2.8769e+00],
[-1.9083e+00, -2.3443e+00, -1.5975e+00, ..., -1.3204e+00,
-3.1423e+00, -1.8112e+00],
...,
[-9.9727e-01, -2.1612e+00, -6.0011e-01, ..., -1.8062e+00,
-1.9745e+00, -9.4745e-01],
[-2.5695e+00, -4.2477e+00, -1.7290e+00, ..., -1.4578e+00,
-2.6471e+00, -1.5233e+00],
[-2.2870e+00, -2.0265e+00, -1.1273e+00, ..., -1.9352e+00,
-2.1263e+00, -5.0379e-01]],
[[-1.3917e+00, -2.0291e+00, -2.5895e+00, ..., -5.0994e-01,
-1.3019e+00, -6.2027e-01],
[-3.3390e+00, -2.4154e+00, -2.6512e+00, ..., -1.5180e+00,
-3.6157e+00, -1.7556e+00],
[-2.8820e+00, -2.8699e+00, -4.1679e+00, ..., -3.5977e-02,
-3.8475e+00, -2.4222e+00],
...,
[-2.5103e+00, -2.7554e+00, -3.5527e+00, ..., -8.3514e-01,
-3.0473e+00, -2.0556e+00],
[-2.2565e+00, 2.0795e-01, -1.4242e+00, ..., -1.0814e+00,
-3.2388e+00, -1.9070e+00],
[-2.7815e+00, -1.7281e+00, -1.7787e+00, ..., -1.6540e+00,
-2.6243e+00, -1.1754e+00]],
[[ 8.3913e-01, 2.0772e-01, 7.2319e-01, ..., 2.0874e-01,
1.1746e+00, 6.1159e-01],
[ 3.4996e+00, 2.7612e+00, 2.2473e+00, ..., 4.3154e-01,
7.9802e-01, 2.4537e+00],
[ 1.3992e+00, 7.1988e-01, 6.1248e-01, ..., 9.4234e-01,
4.2716e-01, 8.3817e-01],
...,
[ 3.2542e+00, 2.6693e+00, 2.6011e+00, ..., 2.3700e-01,
1.1571e-01, 4.9131e-01],
[ 4.3486e+00, 4.4424e+00, 3.1243e+00, ..., 1.0938e+00,
8.0062e-01, 1.1990e+00],
[ 2.6674e+00, 2.5565e+00, 1.7935e+00, ..., -1.5685e-01,
-3.2069e-01, -7.7041e-02]]],
[[[ 4.3235e-01, 1.3794e+00, 3.4036e-01, ..., 5.6572e-01,
-5.4609e-01, 8.6605e-01],
[ 9.9515e-02, -6.8822e-01, 8.9593e-01, ..., -5.3653e-01,
-1.5813e+00, -3.3005e-01],
[-3.5918e-01, -9.7793e-01, 9.1644e-01, ..., 3.6703e-01,
-5.5870e-01, -7.7173e-01],
...,
[ 5.6092e-02, -2.3916e+00, -3.0055e+00, ..., -3.8394e+00,
-3.7734e+00, -8.7896e-01],
[-2.1001e+00, -1.2302e+00, -2.2663e-01, ..., -2.7742e+00,
-2.8066e+00, -1.2597e-03],
[-2.4018e+00, -3.9319e+00, -3.0994e+00, ..., -4.0266e+00,
-3.5357e+00, -1.6445e+00]],
[[-1.5503e+00, -1.8560e+00, -1.9697e+00, ..., -2.5851e+00,
-5.3934e+00, -9.3446e-01],
[-5.1459e+00, -6.5506e+00, -4.0070e+00, ..., -6.4612e+00,
-7.5805e+00, -3.9420e+00],
[-4.6071e+00, -5.0790e+00, -2.7401e+00, ..., -3.8717e+00,
-5.5498e+00, -4.1028e+00],
...,
[-6.6601e+00, -8.5105e+00, -8.9395e+00, ..., -8.0350e+00,
-9.4543e+00, -2.3961e+00],
[-9.7073e+00, -1.0577e+01, -8.4271e+00, ..., -1.0137e+01,
-1.1013e+01, -3.9242e+00],
[-2.5005e+00, -4.9181e+00, -4.5441e+00, ..., -4.3139e+00,
-4.3156e+00, -6.8959e-01]],
[[-2.2020e-01, 7.7887e-01, 5.9508e-01, ..., 1.4029e+00,
1.6692e+00, 1.3943e-01],
[-6.8671e-03, -2.6274e-01, -6.0101e-01, ..., 1.5037e+00,
1.9098e+00, -8.6006e-01],
[ 7.6982e-01, 2.3519e+00, 1.2761e+00, ..., 1.5965e+00,
1.9053e+00, 8.0313e-01],
...,
[-3.5071e-01, 8.0874e-01, 1.0321e-01, ..., 2.0688e-01,
5.3742e-01, -1.0835e+00],
[-6.4998e-01, -2.0686e-01, -4.2240e-01, ..., 1.3514e-01,
9.7849e-01, -5.1827e-01],
[ 5.6087e-01, 1.2666e+00, 9.2423e-01, ..., -2.0658e-02,
-3.9751e-01, -1.8674e+00]],
...,
[[-8.7555e-01, -8.9432e-01, -1.0145e+00, ..., -1.5518e+00,
-2.6589e+00, 3.1047e-01],
[-4.0997e+00, -4.6831e+00, -2.5496e+00, ..., -3.4295e+00,
-5.6873e+00, -3.1903e+00],
[-2.5150e+00, -1.5927e+00, -6.5868e-01, ..., -8.0663e-01,
-4.2748e+00, -1.8674e+00],
...,
[-2.0083e+00, -4.5290e+00, -2.4679e+00, ..., -4.7252e+00,
-6.7003e+00, -3.0406e+00],
[-3.7101e+00, -5.5460e+00, -2.2156e+00, ..., -4.6812e+00,
-5.5657e+00, -1.3728e+00],
[-3.3562e+00, -4.0192e+00, -3.3814e+00, ..., -2.9570e+00,
-2.8096e+00, 4.5087e-01]],
[[-1.0373e+00, 8.7409e-02, -1.8135e+00, ..., -1.5650e-01,
1.1714e-01, 1.8254e-01],
[-3.9556e+00, -8.2935e-01, -2.0997e+00, ..., -1.2804e+00,
-1.8865e+00, 2.7885e-01],
[-3.7786e+00, -1.2810e+00, -1.5039e+00, ..., -4.7144e-01,
-2.6138e+00, -6.8006e-01],
...,
[-3.8114e+00, -3.7942e+00, -2.8278e+00, ..., -3.5591e+00,
-2.5819e+00, -8.4478e-01],
[-2.9117e+00, -8.3344e-01, -9.8629e-01, ..., -1.6448e+00,
-1.0218e+00, 2.8119e-01],
[-3.7388e+00, -2.9333e+00, -2.8027e+00, ..., -1.5536e+00,
-1.3036e+00, 2.9064e-01]],
[[ 1.6410e+00, 6.5848e-01, 5.8296e-01, ..., 7.8326e-01,
1.4027e+00, 2.0871e-01],
[ 3.2236e+00, 2.1074e+00, 1.3771e+00, ..., 1.5801e+00,
2.7803e+00, 3.4926e+00],
[ 2.1954e+00, -4.1163e-02, 9.8376e-03, ..., 2.6434e+00,
2.7058e+00, 2.0724e+00],
...,
[ 3.4309e+00, 1.8971e+00, 1.6386e+00, ..., 2.2682e+00,
1.5738e+00, 2.0516e-01],
[ 4.9079e+00, 4.9108e+00, 3.6087e+00, ..., 2.5146e+00,
1.5128e+00, 1.5602e+00],
[ 2.8352e+00, 2.9659e+00, 2.0500e+00, ..., 5.9042e-01,
5.6295e-01, 1.0298e+00]]],
[[[ 1.5368e+00, 2.2199e+00, 1.6503e+00, ..., -8.6638e-01,
-2.3034e+00, -4.6883e-01],
[ 2.8315e-02, -1.8418e+00, -2.2108e+00, ..., -2.8967e-01,
3.0937e-02, -1.3663e-01],
[-5.6617e+00, -3.6899e+00, -1.4417e-01, ..., -3.2106e+00,
-2.4082e+00, -1.5686e+00],
...,
[-1.6572e+00, -2.0300e+00, 1.8852e+00, ..., -3.5717e+00,
-2.4588e+00, -1.5288e+00],
[-1.8278e+00, -1.3907e+00, -4.1313e-01, ..., -4.1296e+00,
-2.9857e+00, -5.7273e-01],
[-8.6158e-01, -2.8795e+00, -2.9419e+00, ..., -4.2332e+00,
-3.3469e+00, -1.2247e+00]],
[[-1.3954e+00, -2.8463e+00, -4.4730e+00, ..., -4.7178e+00,
-6.5121e+00, -1.4224e+00],
[-4.5824e+00, -6.0382e+00, -5.8636e+00, ..., -8.9366e+00,
-7.5099e+00, -4.3687e+00],
[-8.8146e+00, -5.5776e+00, -2.0196e+00, ..., -5.8735e+00,
-4.4812e+00, -2.0118e+00],
...,
[-6.1955e+00, -5.9423e+00, -2.8075e+00, ..., -7.4108e+00,
-5.9107e+00, -1.7994e+00],
[-7.2690e+00, -6.8425e+00, -7.6853e+00, ..., -9.4104e+00,
-8.8907e+00, -3.9259e+00],
[-1.9050e+00, -2.9226e+00, -2.4598e+00, ..., -4.6115e+00,
-3.5058e+00, -3.8295e-01]],
[[ 3.0478e-01, 7.0920e-01, 1.4496e+00, ..., 7.1622e-01,
2.5843e+00, 1.1299e+00],
[-3.2368e-01, 6.2760e-01, 2.1947e+00, ..., -1.3877e-01,
2.1640e+00, -9.9820e-02],
[ 2.2177e+00, 3.8149e+00, 3.7538e+00, ..., -9.5760e-01,
1.4155e+00, 7.9359e-01],
...,
[ 8.6295e-02, -1.2495e+00, -1.5398e+00, ..., -2.4875e-02,
1.5550e-01, 1.4795e-01],
[-7.1695e-02, 2.4177e-01, 2.7382e-01, ..., -8.0803e-01,
8.6606e-02, -3.7606e-01],
[ 3.9188e-01, 7.5662e-01, 1.3040e+00, ..., -5.9287e-01,
3.9937e-02, -1.7122e+00]],
...,
[[-3.2881e-01, -1.1289e+00, -2.0644e+00, ..., -3.3251e+00,
-4.7877e+00, -1.1548e+00],
[-4.3090e+00, -2.9319e+00, -2.9086e+00, ..., -4.6809e+00,
-5.4397e+00, -2.0534e+00],
[-5.7808e+00, -1.2933e-01, -1.8682e+00, ..., -3.5903e+00,
-3.8317e+00, -1.0962e+00],
...,
[-2.0240e+00, -9.3080e-01, -8.9960e-01, ..., -1.7124e+00,
-3.9106e+00, -2.1536e+00],
[-2.9561e+00, -3.0189e+00, -3.3662e+00, ..., -2.6944e+00,
-4.6682e+00, -9.0552e-01],
[-1.3562e+00, -1.4642e+00, -9.7297e-01, ..., -1.7609e+00,
-2.1357e+00, 1.2045e+00]],
[[-1.6607e+00, -1.7875e+00, -3.3169e+00, ..., -2.9393e+00,
-1.4877e+00, -7.2964e-02],
[-3.1824e+00, -3.5711e+00, -4.7643e+00, ..., -1.8049e+00,
-2.1512e+00, -6.6926e-01],
[-4.8673e+00, -3.3033e+00, -2.6574e+00, ..., -1.0568e+00,
-2.4567e+00, 9.5215e-01],
...,
[-3.8979e+00, -5.1026e-01, -5.2923e-01, ..., -5.2708e+00,
-5.8039e+00, -2.1646e+00],
[-4.0038e+00, -1.4032e+00, -1.8407e+00, ..., -1.3961e+00,
-2.1674e+00, 1.8666e-01],
[-3.1264e+00, -2.9914e+00, -1.3313e+00, ..., 3.7856e-01,
-4.7626e-01, 6.5044e-01]],
[[ 1.4692e+00, 6.6046e-01, 2.1945e+00, ..., -8.3262e-01,
7.9135e-02, -2.9131e-01],
[ 2.1637e+00, -1.1404e+00, 5.6768e-01, ..., 2.2966e+00,
2.7182e+00, 3.8292e+00],
[ 1.7353e+00, -7.9960e-02, 1.4424e+00, ..., -3.1177e-01,
-1.0655e-01, 2.6550e-01],
...,
[ 2.0716e+00, 2.1587e+00, 1.2482e+00, ..., -8.9694e-01,
5.0129e-01, 1.0946e+00],
[ 2.4351e+00, 4.2712e+00, 2.9468e+00, ..., 8.1419e-01,
6.9991e-01, 1.2019e+00],
[ 2.0686e+00, 1.4761e+00, -1.4732e-01, ..., -3.5210e-01,
-3.8900e-01, 7.4671e-01]]]], device='cuda:0'),
tensor([[[[ 5.4927e-02, 1.0993e+00, 2.9533e-01, ..., -1.2586e+00,
-3.1185e+00, -9.7535e-01],
[ 3.3297e-01, 6.8172e-01, 1.3438e+00, ..., -1.5408e+00,
-1.1155e+00, 1.4005e+00],
[-1.7190e-01, 2.9650e-02, 9.5203e-02, ..., -3.3387e+00,
-1.8880e+00, 1.2118e+00],
...,
[-6.7205e-02, 2.9926e+00, 1.0090e+00, ..., -3.1779e+00,
-1.8401e+00, -3.4640e-02],
[ 2.4588e-01, 5.6757e-01, -1.1183e+00, ..., -2.9601e+00,
-2.5015e+00, -8.8458e-02],
[-4.2514e-01, -2.3718e+00, -3.0200e+00, ..., -3.8701e+00,
-3.1067e+00, -9.7214e-01]],
[[-1.2672e+00, -1.8594e+00, -2.6182e+00, ..., -5.4561e+00,
-6.8747e+00, -9.3898e-01],
[-3.5349e+00, -3.7303e+00, -2.7692e+00, ..., -7.2511e+00,
-4.1418e+00, -3.9742e-01],
[-2.5454e+00, -2.8001e+00, -2.4729e+00, ..., -8.1336e+00,
-4.6219e+00, -5.3295e-01],
...,
[-1.8128e+00, -8.4939e-01, -2.6045e+00, ..., -9.0072e+00,
-6.0638e+00, -1.4944e+00],
[-5.5399e+00, -7.5318e+00, -6.4687e+00, ..., -9.5128e+00,
-8.4195e+00, -3.6240e+00],
[-6.7019e-01, -2.7331e+00, -2.8572e+00, ..., -4.5179e+00,
-3.2616e+00, -2.3156e-01]],
[[ 2.2666e-01, 1.2568e+00, 1.4974e+00, ..., 5.8047e-01,
2.8155e+00, 1.4544e+00],
[ 3.1618e-01, 3.1207e-01, 4.9411e-02, ..., -7.0309e-01,
1.2716e+00, -1.3304e-01],
[ 5.5484e-01, 1.6943e+00, 9.5148e-01, ..., -1.1636e+00,
-1.9825e-01, -9.9921e-01],
...,
[-4.9529e-01, -1.3677e+00, 7.2272e-01, ..., -3.1201e-01,
3.8659e-01, 1.2964e-01],
[-6.4423e-01, -5.7975e-01, 3.0397e-01, ..., -7.5904e-01,
1.5049e-01, -3.0426e-01],
[-3.1294e-01, 1.5601e+00, 8.8739e-01, ..., -8.2256e-01,
-1.9847e-01, -1.6643e+00]],
...,
[[-9.0228e-01, -1.4580e+00, -1.9275e+00, ..., -3.1102e+00,
-4.6207e+00, -1.1089e+00],
[-2.9581e+00, -2.2679e+00, -1.0501e+00, ..., -2.7957e+00,
-2.4862e+00, -3.6679e-01],
[-1.6578e+00, -4.6970e-01, -1.0138e-01, ..., -3.8181e+00,
-3.3397e+00, -1.2220e+00],
...,
[ 5.6387e-01, 7.1870e-01, -1.3558e+00, ..., -2.3610e+00,
-5.0495e+00, -1.5381e+00],
[-1.4410e+00, -3.1469e+00, -2.0034e+00, ..., -2.2749e+00,
-4.7613e+00, -5.5609e-01],
[-3.9037e-01, -1.1755e+00, -7.0240e-01, ..., -1.2110e+00,
-2.1185e+00, 1.0999e+00]],
[[-1.6496e+00, -1.9661e+00, -4.0398e+00, ..., -3.5787e+00,
-2.2728e+00, 1.2786e-01],
[-3.8682e+00, -3.2181e+00, -3.7570e+00, ..., -3.0146e+00,
-2.2928e+00, -9.1555e-02],
[-3.7222e+00, -3.3815e+00, -2.5623e+00, ..., -3.2589e+00,
-2.8244e+00, -7.1931e-01],
...,
[-9.8045e-01, -2.5437e+00, -4.7925e+00, ..., -5.3717e+00,
-4.4599e+00, -1.1697e+00],
[-2.1622e+00, -2.4445e+00, -3.2556e+00, ..., -1.4071e+00,
-1.9306e+00, 1.5252e-01],
[-1.6660e+00, -2.5653e+00, -1.3987e+00, ..., 2.5036e-01,
-3.3758e-01, 5.3750e-01]],
[[ 1.1610e+00, -1.0042e-01, 9.5318e-01, ..., -9.6149e-01,
-1.8375e-01, -4.6082e-01],
[ 2.3391e+00, 1.1466e+00, 8.4298e-01, ..., 1.4053e+00,
7.5046e-01, 3.2765e+00],
[ 1.1748e+00, 3.1514e-01, 2.3344e-01, ..., -1.6917e-01,
2.3902e-01, 1.0996e+00],
...,
[-9.6650e-02, 1.5397e+00, 7.4987e-01, ..., -9.9090e-02,
8.9234e-01, 8.3868e-01],
[ 2.5374e+00, 4.1741e+00, 1.4082e+00, ..., 3.9346e-01,
6.8669e-01, 1.6107e+00],
[ 1.5237e+00, 6.5693e-01, 3.6002e-01, ..., -4.9886e-02,
-3.5833e-01, 7.6821e-01]]],
[[[-8.1698e-01, -6.1609e+00, -5.4414e+00, ..., -5.6967e-01,
-5.1287e-02, -2.3355e-01],
[-2.7307e+00, -5.1316e+00, 9.3698e-01, ..., -9.2557e-01,
-6.6932e-01, -6.3231e-01],
[-7.1594e+00, -5.7765e+00, -1.9961e+00, ..., -1.7103e+00,
-2.5323e+00, -1.5634e+00],
...,
[-4.6787e-01, -1.5254e+00, -8.1556e+00, ..., 6.0525e+00,
-3.0384e-01, -1.9547e+00],
[-1.5965e+00, -7.2592e-01, -3.5777e+00, ..., -4.3193e+00,
-3.8848e+00, -1.8559e+00],
[-1.1811e+00, -2.0183e+00, -2.3190e+00, ..., -3.2916e+00,
-3.0619e+00, -3.0034e+00]],
[[-5.2159e+00, -1.0141e+01, -6.0808e+00, ..., -3.4851e+00,
-3.9413e+00, -2.1006e+00],
[-9.4851e+00, -2.5125e+00, 3.0493e+00, ..., -5.7321e+00,
-5.8452e+00, -3.0768e+00],
[-1.2029e+01, -5.7128e+00, 2.8835e-02, ..., -2.6321e+00,
-7.0098e+00, -3.5488e+00],
...,
[-6.2193e+00, -5.4181e+00, -1.0374e+01, ..., -1.4038e+00,
-6.9869e+00, -4.8820e+00],
[-8.2231e+00, -7.7563e+00, -8.7682e+00, ..., -1.0983e+01,
-7.4796e+00, -3.4335e+00],
[-1.7563e+00, -3.3120e+00, -3.1874e+00, ..., -2.5234e+00,
-2.9334e+00, -8.0532e-01]],
[[-1.0083e-01, 3.4376e+00, 2.6792e+00, ..., 1.4533e+00,
1.7928e+00, 8.6920e-01],
[ 1.6070e+00, -2.0858e-01, -4.5417e+00, ..., 1.0388e+00,
1.4024e+00, 5.0823e-01],
[ 1.1258e+00, -1.0345e+00, -2.5252e+00, ..., 2.6700e+00,
2.1665e+00, 8.0820e-01],
...,
[-1.6205e+00, -2.3928e+00, -8.2537e-01, ..., -2.7847e-01,
2.7780e+00, 1.6658e+00],
[-1.8786e+00, -2.5244e+00, -3.0342e+00, ..., -2.3598e+00,
-3.2382e-02, 8.4091e-01],
[-4.2144e-01, -1.2616e+00, 3.3456e-01, ..., 1.9746e+00,
-2.9225e-01, -1.9011e-01]],
...,
[[-3.4456e+00, -6.1609e+00, -3.1524e+00, ..., -2.0698e+00,
-2.5900e+00, -4.7473e-01],
[-6.5712e+00, -1.2700e+00, 1.7418e+00, ..., -3.0536e+00,
-5.0391e+00, -3.0718e+00],
[-8.1701e+00, -3.4701e+00, -3.1104e+00, ..., -3.4145e+00,
-3.0939e+00, -3.4521e+00],
...,
[-7.9766e-01, -4.8619e+00, -7.7299e+00, ..., -3.9594e+00,
-7.0280e+00, -3.7449e+00],
[-1.7637e+00, -5.1584e+00, -3.8817e+00, ..., -6.0975e+00,
-4.7151e+00, -1.4516e+00],
[-9.1453e-01, -2.4398e+00, -3.3043e+00, ..., -2.9349e+00,
-3.0100e+00, -1.5104e+00]],
[[-2.1702e+00, -2.4397e+00, -2.6547e+00, ..., -1.0290e+00,
-1.1894e+00, -4.1550e-01],
[-3.0440e+00, -2.2404e-03, -3.9406e+00, ..., -1.4638e+00,
-3.7009e+00, -1.6829e+00],
[-1.4141e+00, 2.8787e+00, -8.6723e+00, ..., 5.2826e-01,
-2.7876e+00, -1.5826e+00],
...,
[-1.3421e+00, -5.1835e+00, -1.9563e+00, ..., -2.1329e+00,
-5.1342e+00, 2.5254e-01],
[-1.2270e+00, -9.7945e-01, -7.6947e-01, ..., -2.5927e+00,
-2.7324e+00, -3.2975e-02],
[-1.8550e+00, -1.1046e+00, -1.6700e+00, ..., 4.5831e-01,
-4.9304e-01, -1.2067e-01]],
[[ 4.3006e+00, -1.4148e+00, 1.3162e+00, ..., 4.5496e-01,
1.1894e+00, 5.5940e-01],
[ 6.2269e+00, -1.3353e+00, 1.0563e+00, ..., 1.2872e+00,
1.0221e+00, 2.2794e+00],
[ 2.9165e+00, 1.9753e-01, -9.2054e-01, ..., 1.8965e+00,
7.3479e-01, 7.2837e-01],
...,
[ 2.2745e+00, 3.2041e+00, -4.5928e-01, ..., 1.4085e+00,
1.9455e+00, 1.6091e+00],
[ 3.4158e+00, 4.1910e+00, 2.8088e+00, ..., 2.9101e+00,
7.6340e-01, 1.7285e+00],
[ 2.2670e+00, 2.5453e+00, 1.6512e+00, ..., 8.5262e-01,
1.2899e+00, 1.9013e+00]]],
[[[ 3.4936e+00, 3.3315e+00, -5.9006e-01, ..., -3.0460e+00,
-3.4271e+00, -1.8184e+00],
[-5.3824e+00, -1.1447e+01, -1.5502e+01, ..., -1.1887e+00,
-1.2493e+00, -1.6819e+00],
[ 2.9834e+00, 2.5919e+00, -1.7221e+00, ..., -4.4070e+00,
-2.1429e+00, -1.4893e+00],
...,
[-2.7021e+00, -4.8543e+00, -1.1877e+00, ..., -1.1309e+01,
-6.0865e+00, -1.5224e+00],
[-2.2619e+00, -4.1975e+00, -7.7875e-01, ..., -3.9613e+00,
-3.8248e+00, -9.1807e-01],
[-1.6899e+00, -4.2214e+00, -8.3422e-01, ..., 2.4508e+00,
1.2393e+00, 1.3896e+00]],
[[-1.6437e+00, -9.7999e+00, -1.6206e+01, ..., -5.9227e+00,
-7.8128e+00, -3.2335e+00],
[-1.8078e+01, -2.3262e+01, -2.1768e+01, ..., -8.4265e+00,
-8.5323e+00, -4.7880e+00],
[ 6.3248e-03, -1.5840e+00, -2.4967e+00, ..., -5.1927e+00,
-5.6742e+00, -3.1383e+00],
...,
[-6.2797e+00, -6.6851e+00, -2.8309e+00, ..., 4.0509e-01,
-4.6189e+00, -2.4556e+00],
[-9.3862e+00, -7.9930e+00, -2.3783e+00, ..., 2.6963e+00,
-2.8786e+00, -8.6543e-01],
[-1.8249e+00, -4.2526e+00, -1.6169e+00, ..., 1.3923e+00,
1.9299e+00, 1.2553e+00]],
[[ 2.3507e+00, 1.7163e+00, 1.6340e+00, ..., 1.6621e+00,
2.6684e+00, 8.7132e-01],
[ 4.0082e+00, 1.1345e+01, 1.1135e+01, ..., 1.3429e+00,
2.9297e+00, 9.9916e-01],
[-2.7038e-01, 6.3577e+00, 6.7865e+00, ..., 3.5266e+00,
2.7344e+00, 1.3670e+00],
...,
[ 1.6210e+00, -1.8127e-01, -1.6756e+00, ..., -3.7863e+00,
-6.5107e-01, 1.3475e+00],
[-1.2848e+00, -1.3604e+00, -1.1559e+00, ..., 4.1251e-01,
9.2293e-01, -2.6813e-01],
[ 1.2804e-01, -5.9312e-01, -9.2804e-01, ..., -4.1750e+00,
-3.0234e+00, -1.6720e+00]],
...,
[[-2.9156e+00, -3.1541e+00, -3.9080e+00, ..., -4.9837e+00,
-5.7664e+00, -1.8578e+00],
[-7.2678e+00, -1.0643e+01, -1.0843e+01, ..., -3.9743e+00,
-5.9845e+00, -3.2588e+00],
[ 4.7240e+00, 5.6322e+00, 5.9227e+00, ..., -5.8633e+00,
-5.6081e+00, -2.6097e+00],
...,
[-1.8611e+00, -1.9798e+00, -7.7155e-01, ..., 1.4037e+00,
-4.3749e+00, -8.6213e-01],
[-2.4207e+00, -3.6002e+00, 6.6577e-02, ..., 4.4533e+00,
-3.3120e+00, 2.0161e+00],
[-2.5199e+00, -2.5196e+00, -3.7200e-01, ..., 2.7504e+00,
-1.0458e+00, 4.3269e+00]],
[[-2.9226e+00, -3.3437e+00, -2.7027e+00, ..., -3.7395e+00,
-2.7359e+00, 2.0984e-01],
[-1.5795e+01, -1.6037e+01, -1.1308e+01, ..., -2.2948e+00,
-5.1677e+00, -1.3430e+00],
[-4.5235e+00, -5.1503e+00, 1.1071e-01, ..., 4.6414e-01,
-1.9993e+00, -2.6300e-01],
...,
[-4.4775e+00, -8.4074e-01, -5.3606e-01, ..., -1.1524e+01,
-7.6466e+00, -1.6691e+00],
[-2.6441e+00, 1.3657e+00, 1.5035e+00, ..., -1.1644e+01,
-1.0603e+01, -1.5957e+00],
[-2.4176e+00, 4.5195e-01, 1.4571e+00, ..., -1.9283e+00,
-3.4421e+00, 2.6593e+00]],
[[ 2.5351e+00, 4.1795e+00, 4.8772e+00, ..., 5.9970e-01,
1.5011e+00, 4.7186e-01],
[ 8.3330e-01, 6.3473e+00, 2.9665e+00, ..., 2.2639e+00,
2.5076e+00, 3.7131e+00],
[ 7.7202e-01, 3.0558e+00, 3.9194e+00, ..., 5.9050e-01,
7.1207e-01, 1.0018e+00],
...,
[ 1.7943e+00, 7.4579e-01, -8.9733e-01, ..., -1.0605e+01,
-1.1932e+01, -3.5824e+00],
[ 4.5080e+00, 3.3574e+00, 1.2958e+00, ..., -6.6877e+00,
-5.6936e+00, -3.3609e+00],
[ 2.9092e+00, 1.9627e+00, 2.1695e+00, ..., -6.3080e-01,
-1.2307e+00, -4.5696e-01]]],
[[[ 1.5182e-01, 1.3070e+00, 5.8079e-01, ..., -5.3836e-01,
-1.9812e-01, 5.7801e-01],
[ 3.7687e-01, 7.9855e-02, 4.4774e-01, ..., -7.1551e-01,
-1.0320e+00, 1.1886e+00],
[ 4.1392e-02, 1.0322e-01, 6.6693e-01, ..., -3.8482e+00,
-2.5027e+00, 4.2272e-01],
...,
[ 4.9685e-01, -8.6154e-01, 5.0421e-01, ..., 3.9180e-01,
-4.0520e-01, -2.1385e+00],
[-1.3968e+00, -1.6623e+00, -8.6285e-01, ..., 1.8655e+00,
-6.9960e-01, -2.4145e+00],
[-2.7110e-01, -2.8620e+00, -2.9417e+00, ..., 8.9741e-01,
3.3940e-01, -7.4910e-01]],
[[-9.8700e-01, -2.4253e+00, -4.1776e+00, ..., -3.8236e+00,
-5.2317e+00, -3.2037e-01],
[-4.9711e+00, -6.3478e+00, -5.5811e+00, ..., -7.0238e+00,
-6.8667e+00, -1.3204e+00],
[-3.7389e+00, -5.4180e+00, -6.1894e+00, ..., -7.1162e+00,
-6.9263e+00, -9.4501e-01],
...,
[-4.4998e+00, -4.9711e+00, -4.4364e+00, ..., -6.3299e-01,
-3.7609e+00, -3.1521e+00],
[-7.6647e+00, -8.0557e+00, -6.4984e+00, ..., -2.0733e+00,
-6.7988e+00, -4.7059e+00],
[-1.0443e+00, -3.6743e+00, -3.7551e+00, ..., -2.8188e-01,
-9.3065e-01, -3.4826e-01]],
[[-2.8514e-01, 2.0903e+00, 1.2452e+00, ..., 1.0726e+00,
2.1840e+00, 8.3497e-01],
[ 7.3664e-02, 1.0251e+00, 8.8331e-01, ..., -9.0676e-02,
7.3671e-01, -8.5163e-01],
[ 1.3876e-01, 1.9187e+00, 1.2651e+00, ..., 8.7764e-01,
2.1238e+00, 1.0222e+00],
...,
[ 9.8584e-02, 1.0440e+00, 1.4628e+00, ..., 4.2004e+00,
3.7153e+00, 9.1562e-01],
[ 1.6602e-01, 6.7253e-01, 6.1422e-02, ..., -4.1748e-01,
4.6846e+00, 1.9952e+00],
[-6.8203e-01, 1.0740e+00, 7.3739e-01, ..., -3.0461e+00,
8.9841e-01, -2.7471e-01]],
...,
[[-9.8950e-01, -1.8812e+00, -1.9645e+00, ..., -2.6728e+00,
-3.1364e+00, -4.9362e-01],
[-3.3437e+00, -4.3733e+00, -2.8892e+00, ..., -3.4166e+00,
-3.6413e+00, -2.2938e+00],
[-1.6736e+00, -2.5676e+00, -1.3672e+00, ..., -2.9746e+00,
-3.0532e+00, -1.8127e+00],
...,
[-1.2310e+00, -1.3410e+00, -6.9860e-01, ..., -2.6644e+00,
-4.8997e+00, -2.8156e+00],
[-3.1041e+00, -3.2980e+00, -2.0852e+00, ..., -1.7923e+00,
-5.0779e+00, -1.8863e+00],
[-9.5679e-01, -2.1047e+00, -1.9645e+00, ..., -3.3790e-01,
-1.4087e+00, 1.0980e+00]],
[[-1.7726e+00, -2.0910e+00, -3.0598e+00, ..., -1.3315e+00,
-1.2711e+00, -1.2845e+00],
[-4.5380e+00, -3.3075e+00, -3.2232e+00, ..., -1.5834e+00,
-1.1682e+00, -7.5551e-01],
[-2.8885e+00, -1.6209e+00, -2.1190e+00, ..., -6.3865e-01,
-8.4262e-01, -3.2069e-01],
...,
[-3.5028e+00, -3.2698e+00, -2.7357e+00, ..., -1.9110e+00,
-1.3598e+00, -6.3185e-01],
[-4.1947e+00, -1.5112e+00, -1.8094e+00, ..., 1.3681e-02,
-2.6893e+00, -4.7053e-01],
[-3.1062e+00, -2.8731e+00, -1.0960e+00, ..., 3.6876e-01,
-1.4470e+00, 3.7429e-01]],
[[ 1.1298e+00, 5.5782e-01, 6.6027e-01, ..., -8.3244e-01,
-1.4695e-01, -5.3295e-01],
[ 3.7204e+00, 3.4454e+00, 2.2388e+00, ..., 2.1577e+00,
1.4381e+00, 2.5010e+00],
[ 1.7546e+00, 1.4020e+00, 1.3753e+00, ..., -3.6889e-01,
6.5706e-01, 2.3236e+00],
...,
[ 2.8784e+00, 1.6055e+00, 1.2816e+00, ..., 3.6621e+00,
2.7930e+00, 9.4389e-02],
[ 3.6204e+00, 3.9772e+00, 2.6325e+00, ..., 8.4182e-01,
2.7211e+00, 1.2474e+00],
[ 2.2702e+00, 1.6147e+00, 6.1571e-01, ..., 6.1958e-01,
2.3161e-01, 1.1198e+00]]],
[[[ 6.8909e-01, -1.3202e+00, 1.6470e+00, ..., -2.9512e+00,
-2.3025e+00, -9.9667e-01],
[ 1.0212e+00, -1.3394e+00, 4.2350e+00, ..., -1.6874e+00,
-2.4009e+00, -2.3285e-01],
[-1.3080e-01, -8.8700e-01, 3.6682e+00, ..., -2.4633e+00,
-3.7474e+00, 9.5544e-04],
...,
[ 1.8835e-01, -1.0817e+00, -8.1937e+00, ..., 1.5754e+00,
-2.5667e+00, 3.9835e-03],
[-1.4784e+00, 8.5057e-02, -9.8020e+00, ..., -1.3231e-01,
-2.8615e+00, 1.9334e-01],
[-1.1218e+00, -2.8222e+00, -6.0959e+00, ..., -1.5798e+00,
-1.9922e+00, -1.5255e+00]],
[[ 1.1497e+00, -2.3596e+00, -1.7560e-01, ..., -6.1555e+00,
-7.2263e+00, -1.3699e+00],
[-1.2300e-01, -2.1649e+00, 1.7317e+00, ..., -6.0401e+00,
-8.1121e+00, -2.2906e+00],
[-1.1162e+00, -2.5034e+00, 5.1021e-01, ..., -9.6956e+00,
-8.4359e+00, -2.4663e+00],
...,
[-6.7836e+00, -5.2306e+00, -1.3469e+01, ..., -1.7901e+00,
-8.3568e+00, -2.2128e+00],
[-8.1277e+00, -5.6427e+00, -9.2966e+00, ..., -7.2156e+00,
-1.0714e+01, -4.6524e+00],
[-2.2099e+00, -1.5883e+00, -2.6076e+00, ..., -3.6082e+00,
-4.1652e+00, -8.8645e-01]],
[[-1.1349e+00, -1.3171e+00, 6.9363e-01, ..., 1.6239e+00,
2.3876e+00, 9.3695e-01],
[-3.2543e+00, -3.8562e+00, -9.7107e-01, ..., 1.6104e+00,
2.8755e+00, -2.6860e-03],
[-2.2794e+00, -3.7282e+00, -8.5463e-01, ..., -3.5235e-02,
2.4557e+00, -4.4051e-01],
...,
[ 5.2251e-01, -1.1476e+00, 9.4208e-01, ..., 3.4214e+00,
1.0054e+00, -8.7163e-01],
[-9.3286e-02, -2.2294e+00, 3.0804e+00, ..., 5.0678e-01,
1.3313e+00, -1.3144e+00],
[ 3.1071e-01, -1.3554e+00, 1.5925e+00, ..., 7.5906e-01,
-2.4397e-01, -1.6417e+00]],
...,
[[ 1.7644e+00, 7.9342e-01, 1.5535e+00, ..., -4.9850e+00,
-5.1547e+00, -7.0145e-01],
[ 1.6554e+00, 1.5254e+00, 2.5335e+00, ..., -4.0652e+00,
-4.8679e+00, -1.6170e+00],
[ 5.9212e-01, 1.9839e+00, 1.0739e+00, ..., -5.6011e+00,
-6.9250e+00, -1.1090e+00],
...,
[-9.9609e-01, -1.6512e+00, -4.2331e+00, ..., -4.2153e+00,
-4.4388e+00, -2.4400e+00],
[-2.1529e+00, -1.5085e+00, -3.5045e+00, ..., -7.4059e+00,
-4.9472e+00, -1.4401e+00],
[-1.0666e+00, -3.6612e-01, -1.5443e+00, ..., -4.1439e+00,
-1.7652e+00, 3.4706e-01]],
[[-2.4537e-01, -1.5400e-01, -1.2112e+00, ..., -1.4017e+00,
-3.7866e-01, -6.3228e-01],
[-9.9301e-01, -7.0198e-01, -1.4909e-01, ..., -3.7755e+00,
-1.8683e+00, -5.7722e-01],
[-1.6161e+00, -1.3524e-01, -4.2731e-01, ..., -3.1867e+00,
-3.0955e+00, -1.1263e+00],
...,
[-4.6761e+00, -6.5880e+00, -4.6107e+00, ..., -2.6726e+00,
1.9896e-01, -8.5343e-02],
[-3.5418e+00, -3.8597e+00, -4.0906e+00, ..., -2.9696e+00,
3.6658e-01, 1.5423e+00],
[-2.3770e+00, -1.1618e+00, -2.3775e+00, ..., -2.8673e+00,
-6.4978e-02, 7.3562e-01]],
[[-1.3234e+00, -2.3841e+00, 5.6445e-01, ..., 5.8268e-01,
7.5922e-01, -1.4511e+00],
[-1.1937e+00, -3.9058e+00, 1.9523e+00, ..., 3.5211e+00,
3.0084e+00, 2.2432e+00],
[-1.7148e+00, -1.7119e+00, 2.8769e+00, ..., 2.4283e+00,
1.0541e+00, 1.3536e+00],
...,
[ 2.7925e+00, 2.0809e+00, 1.9169e+00, ..., 4.1496e+00,
3.1438e+00, 6.3074e-01],
[ 3.2595e+00, 3.8042e+00, 2.6338e-02, ..., 3.9080e+00,
2.6119e-01, 1.6188e+00],
[ 2.4268e+00, 1.9179e+00, -1.1818e+00, ..., 1.6767e+00,
1.2300e-01, 9.9284e-01]]]], device='cuda:0'),
tensor([[[[ 1.2769e-01, 6.7905e-01, -6.9097e-01, ..., -2.7785e-01,
-6.7212e-01, -8.1523e-01],
[ 7.9914e-01, 7.6163e-01, 3.5695e-03, ..., 2.2384e-01,
-2.4985e-01, -1.2050e+00],
[ 1.0286e+00, -3.2374e-01, 3.8916e-01, ..., -5.7008e-01,
-1.0475e+00, -2.2109e+00],
...,
[-5.1811e-01, -4.5682e-01, 8.0941e-01, ..., -2.0099e+00,
-2.6030e+00, -2.2413e-02],
[-2.4609e+00, -1.6369e+00, 1.1816e+00, ..., -2.3933e+00,
-2.8085e+00, 1.4906e-01],
[-2.4169e+00, -4.2468e+00, -3.0282e+00, ..., -4.0184e+00,
-3.4186e+00, -1.4896e+00]],
[[-4.8609e-01, -1.3956e+00, -2.3190e+00, ..., -2.1657e+00,
-4.5763e+00, -2.5572e+00],
[-1.7853e+00, -3.3836e+00, -3.4434e+00, ..., -3.8382e+00,
-5.4310e+00, -4.7630e+00],
[-1.6565e+00, -3.3265e+00, -3.5372e+00, ..., -3.2376e+00,
-4.4231e+00, -3.8467e+00],
...,
[-6.9147e+00, -8.7146e+00, -6.4182e+00, ..., -7.7294e+00,
-8.6599e+00, -1.8911e+00],
[-9.8225e+00, -1.0275e+01, -8.6946e+00, ..., -1.0596e+01,
-1.0878e+01, -3.7889e+00],
[-2.5997e+00, -5.0256e+00, -4.1954e+00, ..., -4.1694e+00,
-4.1183e+00, -6.1420e-01]],
[[ 6.1082e-01, 1.2734e+00, 1.1882e+00, ..., 1.8612e+00,
1.7771e+00, 7.9176e-01],
[ 6.5338e-01, 1.3856e+00, 1.4557e+00, ..., 1.4435e+00,
2.1357e+00, 8.4536e-01],
[ 5.5513e-01, 3.0561e+00, 1.8814e+00, ..., 1.2808e+00,
2.6032e+00, 2.0553e+00],
...,
[-8.0325e-02, -7.5435e-01, -1.1800e+00, ..., 1.5434e+00,
1.2993e+00, -2.1794e-01],
[-5.9218e-01, 2.4564e-01, -1.4665e+00, ..., 8.3758e-01,
1.2945e+00, -4.2907e-01],
[ 6.2185e-01, 1.2792e+00, 9.0798e-01, ..., -1.5960e-01,
-5.2486e-01, -1.9171e+00]],
...,
[[-7.2373e-01, -1.1967e+00, -2.2396e+00, ..., -1.8508e+00,
-3.3695e+00, -1.1239e+00],
[-1.9967e+00, -2.5380e+00, -2.2135e+00, ..., -2.9800e+00,
-4.8440e+00, -3.7459e+00],
[-8.1746e-01, -1.2382e+00, -1.2831e+00, ..., -2.4624e+00,
-4.8103e+00, -3.2792e+00],
...,
[-2.7676e+00, -3.2909e+00, -1.6049e+00, ..., -5.4789e+00,
-6.1657e+00, -2.5706e+00],
[-3.8367e+00, -5.6904e+00, -2.1724e+00, ..., -4.5996e+00,
-5.2419e+00, -1.3045e+00],
[-3.3172e+00, -3.8264e+00, -3.1577e+00, ..., -2.7366e+00,
-2.5183e+00, 4.5682e-01]],
[[-1.3953e+00, -2.5853e+00, -4.1004e+00, ..., -2.2327e+00,
-2.4405e+00, -5.7288e-01],
[-4.0121e+00, -4.7576e+00, -6.0233e+00, ..., -3.4045e+00,
-5.3289e+00, -2.0428e+00],
[-4.0075e+00, -4.9147e+00, -5.6530e+00, ..., -3.4234e+00,
-4.9037e+00, -2.2419e+00],
...,
[-3.8943e+00, -2.4989e+00, -4.9421e+00, ..., -3.9030e+00,
-2.8550e+00, -1.2354e+00],
[-3.5778e+00, -8.1103e-02, -1.3707e+00, ..., -2.2666e+00,
-1.2196e+00, 3.3653e-01],
[-3.8773e+00, -2.9760e+00, -2.6109e+00, ..., -1.6104e+00,
-1.2617e+00, 3.7736e-01]],
[[ 7.9319e-01, -1.5699e-01, 7.7221e-01, ..., 8.0976e-01,
1.2022e+00, 6.0267e-01],
[ 1.4820e+00, 5.9726e-01, 5.7973e-01, ..., 1.1168e+00,
1.7800e+00, 3.2171e+00],
[ 1.3607e+00, -3.8129e-02, 7.5611e-01, ..., 4.2169e-01,
1.1300e+00, 9.2785e-01],
...,
[ 4.0637e+00, 3.7534e+00, 2.9910e+00, ..., 3.6335e+00,
2.8511e+00, 6.5629e-01],
[ 4.5535e+00, 4.5464e+00, 3.4331e+00, ..., 2.3727e+00,
1.8790e+00, 1.5054e+00],
[ 2.7862e+00, 2.7794e+00, 2.1961e+00, ..., 4.2865e-01,
5.7396e-01, 1.0292e+00]]],
[[[ 3.2103e-01, 1.1688e+00, 2.1462e-01, ..., -8.7427e-01,
-2.0252e+00, -3.4870e-01],
[ 4.6268e-01, 8.9279e-01, 1.6539e+00, ..., -7.6363e-01,
-5.0756e-01, 2.8520e-01],
[-4.2080e-01, -5.1053e-01, 3.3289e-02, ..., -3.1885e+00,
-3.8629e+00, -1.7694e+00],
...,
[-7.5474e-01, -1.5051e+00, 5.8230e-01, ..., -3.3817e+00,
-1.8306e+00, -1.2966e-01],
[-2.0407e+00, -1.6591e+00, -8.6200e-01, ..., -2.9480e+00,
-2.3047e+00, -1.8698e-01],
[-1.0878e+00, -3.2901e+00, -3.3205e+00, ..., -3.5982e+00,
-2.9828e+00, -1.1525e+00]],
[[-9.4661e-01, -1.4858e+00, -2.5524e+00, ..., -4.8784e+00,
-6.5815e+00, -1.4958e+00],
[-3.5882e+00, -3.8615e+00, -2.7360e+00, ..., -7.1053e+00,
-6.3134e+00, -2.8518e+00],
[-3.8007e+00, -4.0862e+00, -3.1941e+00, ..., -8.0017e+00,
-6.4706e+00, -2.1682e+00],
...,
[-5.8715e+00, -5.8937e+00, -3.1894e+00, ..., -7.9538e+00,
-6.2670e+00, -1.5964e+00],
[-7.5786e+00, -7.4577e+00, -6.5834e+00, ..., -8.9303e+00,
-8.3577e+00, -3.7185e+00],
[-1.7209e+00, -3.2284e+00, -3.0787e+00, ..., -4.3004e+00,
-3.3115e+00, -5.0051e-01]],
[[ 2.9273e-02, 1.2865e+00, 1.5646e+00, ..., 6.2765e-01,
2.5845e+00, 1.0970e+00],
[-4.2868e-02, 2.8805e-02, 3.5306e-01, ..., -8.2542e-01,
1.8756e+00, 7.5853e-02],
[ 5.7512e-01, 1.3269e+00, 1.1739e+00, ..., -1.1169e+00,
1.6042e+00, 9.9735e-01],
...,
[ 1.6540e-01, 5.5301e-01, 1.5312e-01, ..., -2.2072e-01,
3.7583e-01, -4.2581e-01],
[ 4.1600e-01, 6.1491e-01, 4.3616e-01, ..., -5.4601e-01,
6.0883e-01, -3.3366e-01],
[ 5.1002e-01, 1.1205e+00, 1.3222e+00, ..., -8.2534e-01,
1.1102e-02, -1.4953e+00]],
...,
[[-5.4257e-01, -1.5183e+00, -1.8169e+00, ..., -3.3606e+00,
-4.7892e+00, -1.0930e+00],
[-2.5896e+00, -2.5600e+00, -1.0181e+00, ..., -3.9918e+00,
-4.8947e+00, -2.0791e+00],
[-2.0169e+00, -1.2989e+00, -8.3475e-01, ..., -4.9663e+00,
-5.7343e+00, -1.4825e+00],
...,
[-1.8868e+00, -1.3546e+00, 4.1465e-01, ..., -3.9333e+00,
-4.8041e+00, -1.5926e+00],
[-3.1838e+00, -3.0312e+00, -1.6668e+00, ..., -2.8357e+00,
-4.5798e+00, -4.5205e-01],
[-1.4236e+00, -1.8844e+00, -1.6527e+00, ..., -1.2850e+00,
-2.0872e+00, 1.0663e+00]],
[[-1.2745e+00, -2.2467e+00, -3.7617e+00, ..., -2.8902e+00,
-1.7246e+00, -2.1379e-01],
[-3.4606e+00, -3.3729e+00, -4.0729e+00, ..., -2.7956e+00,
-2.9872e+00, -2.9461e-01],
[-4.4717e+00, -3.7316e+00, -3.1040e+00, ..., -3.5466e+00,
-5.3448e+00, -9.9880e-01],
...,
[-4.4910e+00, -3.7739e+00, -3.8319e+00, ..., -4.4104e+00,
-3.9386e+00, -1.0057e+00],
[-4.0585e+00, -1.8048e+00, -1.8754e+00, ..., -5.5598e-01,
-1.7459e+00, 3.8718e-01],
[-3.1649e+00, -3.1358e+00, -1.3933e+00, ..., 4.9519e-01,
-5.9777e-01, 4.8458e-01]],
[[ 1.1508e+00, -1.2848e-01, 9.7070e-01, ..., -7.9992e-01,
3.4195e-01, -1.5701e-01],
[ 2.1777e+00, 9.8509e-01, 1.1003e+00, ..., 1.4781e+00,
1.8222e+00, 3.4116e+00],
[ 1.3522e+00, 5.2648e-01, 3.5056e-01, ..., 3.4791e-02,
4.5966e-01, 1.0060e+00],
...,
[ 2.3643e+00, 1.8022e+00, 9.4867e-01, ..., 7.5418e-03,
1.0086e+00, 8.6087e-01],
[ 2.8152e+00, 3.9890e+00, 2.2631e+00, ..., 4.3808e-01,
7.8110e-01, 1.4806e+00],
[ 1.9452e+00, 1.5587e+00, 3.3444e-01, ..., -4.6814e-01,
-3.5252e-01, 6.7558e-01]]],
[[[-1.9401e-01, 5.9948e-01, -2.3622e-01, ..., -1.3801e+00,
-1.0399e+00, -7.4943e-01],
[ 7.4683e-01, 4.2587e-01, 4.9073e-01, ..., -6.0193e-01,
3.1618e-01, -8.7905e-01],
[ 1.8709e+00, -2.7180e-02, 9.9652e-01, ..., -8.5341e-01,
-5.2099e-01, -2.1672e+00],
...,
[-4.1785e-01, -1.6986e+00, -2.4889e-01, ..., -1.3743e+00,
-2.3470e+00, 1.0827e-02],
[-2.2799e+00, -1.8963e+00, -1.4317e-01, ..., -2.6748e+00,
-3.1576e+00, -1.8741e-01],
[-2.3725e+00, -4.0788e+00, -2.9270e+00, ..., -4.3734e+00,
-3.6449e+00, -1.5710e+00]],
[[-7.2224e-01, -1.5043e+00, -2.3037e+00, ..., -3.0036e+00,
-4.4853e+00, -2.2359e+00],
[-1.9416e+00, -4.0795e+00, -4.7895e+00, ..., -3.3031e+00,
-4.7278e+00, -4.3526e+00],
[-1.5794e+00, -3.8288e+00, -4.3219e+00, ..., -3.6819e+00,
-4.5831e+00, -3.8550e+00],
...,
[-7.1480e+00, -8.7919e+00, -6.8456e+00, ..., -8.2700e+00,
-8.9073e+00, -1.8008e+00],
[-9.6345e+00, -1.0042e+01, -8.1465e+00, ..., -1.0658e+01,
-1.0906e+01, -3.9124e+00],
[-2.5574e+00, -5.0295e+00, -4.2351e+00, ..., -4.3235e+00,
-4.2598e+00, -6.4874e-01]],
[[ 6.8677e-01, 1.2213e+00, 1.0480e+00, ..., 1.4920e+00,
1.8240e+00, 9.0123e-01],
[ 9.3598e-01, 1.7150e+00, 1.1684e+00, ..., 1.2184e+00,
2.1450e+00, 8.6795e-01],
[ 1.4628e-01, 2.9502e+00, 2.1356e+00, ..., 1.2247e+00,
1.9541e+00, 1.7662e+00],
...,
[-3.0442e-01, 3.1654e-01, 8.2062e-01, ..., -7.4624e-01,
-2.7979e-01, -2.6527e-01],
[-5.5164e-01, 7.1021e-01, 6.2734e-02, ..., 7.1402e-01,
1.1622e+00, -4.6123e-01],
[ 5.9333e-01, 1.1284e+00, 6.5571e-01, ..., -2.8593e-01,
-6.1663e-01, -1.9170e+00]],
...,
[[-1.0389e+00, -1.3385e+00, -1.8778e+00, ..., -2.5834e+00,
-3.1931e+00, -1.1881e+00],
[-2.3423e+00, -3.2921e+00, -2.5263e+00, ..., -2.7712e+00,
-4.0551e+00, -3.3721e+00],
[-6.2720e-01, -2.2119e+00, -6.7090e-01, ..., -2.4683e+00,
-4.3936e+00, -3.2523e+00],
...,
[-2.3751e+00, -2.8698e+00, -5.2270e-01, ..., -6.1875e+00,
-6.8843e+00, -2.1421e+00],
[-3.7699e+00, -5.7373e+00, -2.8910e+00, ..., -3.9382e+00,
-5.1837e+00, -1.3151e+00],
[-3.2730e+00, -3.6272e+00, -3.0043e+00, ..., -2.7331e+00,
-2.6041e+00, 4.0869e-01]],
[[-1.5711e+00, -2.2577e+00, -3.4198e+00, ..., -2.8768e+00,
-2.7520e+00, -5.8106e-01],
[-4.1039e+00, -4.0126e+00, -4.2934e+00, ..., -4.0727e+00,
-5.6645e+00, -1.9814e+00],
[-3.0843e+00, -4.1398e+00, -2.6516e+00, ..., -2.9317e+00,
-4.7337e+00, -2.2500e+00],
...,
[-4.2917e+00, -2.3745e+00, -3.7335e+00, ..., -4.1014e+00,
-3.2951e+00, -9.6092e-01],
[-3.5342e+00, -9.2925e-01, -2.1859e+00, ..., -2.1443e+00,
-1.5323e+00, 3.0906e-01],
[-3.8292e+00, -3.0280e+00, -2.9173e+00, ..., -1.7202e+00,
-1.1347e+00, 3.1263e-01]],
[[ 9.1308e-01, -2.2490e-01, 6.5900e-01, ..., 5.2915e-01,
1.2908e+00, 5.6829e-01],
[ 1.8845e+00, 1.1034e+00, 8.9903e-01, ..., 7.4977e-01,
1.8325e+00, 3.1486e+00],
[ 1.7636e+00, 1.1279e-01, 8.5749e-01, ..., 6.1215e-01,
9.7250e-01, 5.6767e-01],
...,
[ 3.7741e+00, 3.0019e+00, 1.5734e+00, ..., 1.8984e+00,
1.2588e+00, 6.0844e-01],
[ 4.8332e+00, 4.5347e+00, 4.1996e+00, ..., 2.3239e+00,
1.5307e+00, 1.8956e+00],
[ 2.8122e+00, 2.6353e+00, 1.9497e+00, ..., 6.1370e-01,
5.3204e-01, 1.0292e+00]]],
[[[ 5.2782e-02, 1.6935e+00, 1.5385e+00, ..., -1.7733e+00,
-3.1928e+00, -1.3649e+00],
[ 7.2217e-01, 6.5672e-01, 2.7442e+00, ..., -1.6589e+00,
-6.8081e-01, 1.5395e-01],
[-2.0447e+00, -2.6127e+00, -5.0834e-01, ..., 1.9444e+00,
2.0995e+00, 5.5189e-01],
...,
[-6.0249e-01, -1.7313e+00, 7.1717e-01, ..., -1.3429e+00,
4.8074e-01, 3.5054e+00],
[-1.7564e+00, -4.9195e-01, 1.1641e+00, ..., 1.4786e+00,
-8.9906e-01, -9.2730e-01],
[-1.8542e+00, -3.1712e+00, -1.5063e+00, ..., -2.9383e+00,
-2.6331e+00, -1.6201e+00]],
[[-6.1752e-01, -1.7180e+00, -3.3378e+00, ..., -3.6406e+00,
-7.9410e+00, -1.9127e+00],
[-3.7108e+00, -4.5240e+00, -4.9848e+00, ..., -2.5635e+00,
-5.2097e+00, -2.4697e+00],
[-3.1528e+00, -4.2118e+00, -5.6758e+00, ..., 1.2168e+00,
-2.3113e+00, -3.4280e+00],
...,
[-5.9275e+00, -6.7426e+00, -6.6072e+00, ..., 1.9158e+00,
1.3091e+00, 1.4136e+00],
[-8.9256e+00, -8.7905e+00, -7.0944e+00, ..., -8.7342e+00,
-1.2416e+01, -8.5775e+00],
[-2.0049e+00, -3.9640e+00, -2.9596e+00, ..., -2.8108e+00,
-2.7387e+00, -4.5040e-02]],
[[-1.4730e-01, 2.4728e+00, 1.4216e+00, ..., 2.4039e+00,
3.1032e+00, 7.0198e-01],
[-9.5789e-01, 2.7712e-01, 1.7463e-01, ..., 3.9733e+00,
4.6009e+00, 2.2701e-01],
[ 1.4932e-01, 2.4132e+00, 2.3985e+00, ..., -6.3048e-02,
3.5017e+00, 1.0207e+00],
...,
[-1.6282e+00, -6.3010e-01, 3.0859e-01, ..., -5.5578e+00,
-5.6017e+00, -3.2568e+00],
[-1.2622e+00, -8.7114e-01, 2.3531e-01, ..., -1.7174e+00,
-2.0788e+00, -1.4360e+00],
[ 2.5254e-01, -1.2013e-01, 7.2810e-02, ..., -5.1026e-01,
2.6116e-01, -7.6443e-01]],
...,
[[-8.7239e-01, -1.6441e+00, -3.2261e-01, ..., -2.8964e+00,
-4.7517e+00, -1.3520e+00],
[-3.0064e+00, -3.4300e+00, -1.3864e+00, ..., -2.9708e+00,
-4.7233e+00, -2.3333e+00],
[-2.8900e+00, -3.2079e+00, -2.0342e+00, ..., 6.1444e-01,
-3.8592e+00, -1.5590e+00],
...,
[-1.8690e+00, -3.2405e+00, -2.0088e+00, ..., -2.6563e-01,
-1.0552e-01, -9.8448e-01],
[-2.8070e+00, -3.7074e+00, -2.9473e-01, ..., -3.6169e+00,
-5.4475e+00, -3.7061e+00],
[-2.3830e+00, -2.1496e+00, -2.5350e-01, ..., -2.2037e-01,
-1.0969e+00, 1.2958e+00]],
[[-1.5520e+00, -1.8061e+00, -1.3957e+00, ..., -1.9570e+00,
-8.3058e-01, 1.0193e-01],
[-3.2033e+00, -1.7643e+00, -8.9689e-01, ..., -3.2459e+00,
-2.8573e+00, 9.6247e-02],
[-4.0080e+00, -4.6762e+00, -2.7439e+00, ..., 3.2936e-01,
-3.8462e+00, -2.7320e-01],
...,
[-3.2557e+00, -2.2752e+00, -1.3381e+00, ..., -2.5665e-01,
7.2408e-01, 2.3044e+00],
[-2.5236e+00, 5.2965e-01, -8.8953e-01, ..., 1.1143e-01,
-2.5100e+00, 1.3197e+00],
[-2.7828e+00, -1.5619e+00, -6.1649e-01, ..., 1.7619e+00,
-4.3202e-01, 2.9489e-01]],
[[ 7.6637e-01, 4.6250e-01, 7.3762e-01, ..., 8.0045e-01,
1.4520e+00, -2.9791e-01],
[ 3.7089e+00, 3.2232e+00, 3.2130e+00, ..., 2.2677e+00,
4.0867e+00, 3.9642e+00],
[ 1.0363e+00, 3.5860e-02, 3.1212e-01, ..., 6.1483e-01,
2.7233e+00, 2.0820e+00],
...,
[ 3.2233e+00, 1.8929e+00, 3.2695e+00, ..., -1.8282e+00,
-1.3827e+00, 2.9497e+00],
[ 4.4068e+00, 3.9786e+00, 2.5047e+00, ..., 5.2561e+00,
3.9013e+00, 3.9310e+00],
[ 2.6232e+00, 2.6584e+00, 2.1086e+00, ..., -1.5617e-01,
-1.9142e+00, 4.3322e-01]]],
[[[-8.6306e-02, 6.7413e-01, -2.8092e-01, ..., -1.3204e+00,
-1.0521e+00, -8.0413e-01],
[ 7.9120e-01, 6.2159e-01, 3.5915e-01, ..., -3.0328e-01,
3.0637e-01, -9.8553e-01],
[ 1.0695e+00, -4.7110e-01, -9.6986e-02, ..., -9.1642e-01,
-6.0033e-01, -2.0527e+00],
...,
[-6.4929e-01, -6.8704e-01, 8.0778e-01, ..., -1.8021e+00,
-2.5628e+00, 2.7571e-01],
[-2.1187e+00, -1.0347e+00, 6.9985e-01, ..., -2.3873e+00,
-2.7811e+00, 2.8107e-01],
[-1.6436e+00, -3.1611e+00, -2.8347e+00, ..., -4.3340e+00,
-3.8296e+00, -1.7112e+00]],
[[-7.8310e-01, -1.5066e+00, -2.7011e+00, ..., -3.0168e+00,
-4.5428e+00, -2.4011e+00],
[-1.9663e+00, -3.6068e+00, -4.2728e+00, ..., -3.1814e+00,
-4.8297e+00, -4.5219e+00],
[-1.8807e+00, -3.5516e+00, -3.6863e+00, ..., -2.6838e+00,
-4.2378e+00, -3.7009e+00],
...,
[-5.8052e+00, -7.3658e+00, -6.5764e+00, ..., -7.5846e+00,
-8.7108e+00, -1.7938e+00],
[-8.7828e+00, -9.5208e+00, -8.6901e+00, ..., -1.0647e+01,
-1.1230e+01, -4.1638e+00],
[-1.9827e+00, -4.1575e+00, -3.8725e+00, ..., -4.7830e+00,
-4.5512e+00, -7.3724e-01]],
[[ 7.1084e-01, 1.2285e+00, 1.0980e+00, ..., 1.5722e+00,
1.8494e+00, 8.8473e-01],
[ 7.9739e-01, 1.3035e+00, 1.5579e+00, ..., 1.2042e+00,
2.1825e+00, 9.3455e-01],
[ 5.5078e-01, 2.9520e+00, 1.8573e+00, ..., 1.2776e+00,
2.3249e+00, 1.9689e+00],
...,
[ 2.5965e-02, 5.2757e-01, 3.2466e-01, ..., 1.6441e+00,
1.0173e+00, -4.4149e-01],
[-7.2856e-01, 4.5030e-01, -2.3537e-01, ..., 6.7290e-01,
1.0620e+00, -6.6524e-01],
[ 4.8888e-02, 6.1007e-01, 7.5291e-01, ..., 1.3178e-01,
-2.3656e-01, -1.9206e+00]],
...,
[[-1.0306e+00, -1.3306e+00, -2.0896e+00, ..., -2.5497e+00,
-3.2568e+00, -1.2024e+00],
[-2.3330e+00, -2.9683e+00, -2.4554e+00, ..., -2.7208e+00,
-4.0697e+00, -3.4109e+00],
[-1.1675e+00, -1.7055e+00, -1.4037e+00, ..., -2.8742e+00,
-4.1548e+00, -3.1580e+00],
...,
[-2.1922e+00, -3.9103e+00, -1.0686e+00, ..., -4.9548e+00,
-5.9067e+00, -2.4440e+00],
[-2.9165e+00, -5.4617e+00, -2.3438e+00, ..., -4.2841e+00,
-5.4552e+00, -1.2296e+00],
[-2.3333e+00, -3.2106e+00, -2.5141e+00, ..., -3.0222e+00,
-2.8719e+00, 3.8975e-01]],
[[-1.6007e+00, -2.4470e+00, -3.5771e+00, ..., -2.7947e+00,
-2.7790e+00, -5.2984e-01],
[-4.3676e+00, -4.7929e+00, -5.4052e+00, ..., -3.9624e+00,
-5.7467e+00, -1.9995e+00],
[-4.1825e+00, -5.2294e+00, -4.9601e+00, ..., -2.9238e+00,
-4.8729e+00, -2.1063e+00],
...,
[-4.5680e+00, -2.7284e+00, -3.0596e+00, ..., -4.2778e+00,
-3.1807e+00, -1.4089e+00],
[-3.4181e+00, -9.4433e-01, -1.4487e+00, ..., -2.3965e+00,
-1.5353e+00, 1.7877e-01],
[-3.1214e+00, -2.6857e+00, -2.6826e+00, ..., -1.9390e+00,
-1.3904e+00, 3.5795e-01]],
[[ 9.0081e-01, -1.4359e-01, 7.5871e-01, ..., 4.8748e-01,
1.2875e+00, 6.1204e-01],
[ 1.7648e+00, 8.8134e-01, 6.0413e-01, ..., 7.4886e-01,
1.8701e+00, 3.2025e+00],
[ 1.4245e+00, 2.0302e-01, 5.7660e-01, ..., 1.6449e-01,
1.0492e+00, 8.0962e-01],
...,
[ 3.6474e+00, 3.3332e+00, 2.7530e+00, ..., 2.9392e+00,
2.6417e+00, 5.9071e-01],
[ 3.9532e+00, 4.2610e+00, 3.6864e+00, ..., 2.7609e+00,
1.9247e+00, 1.5299e+00],
[ 2.6628e+00, 2.5405e+00, 1.9506e+00, ..., 5.8236e-01,
4.2698e-01, 8.8187e-01]]]], device='cuda:0'),
tensor([[[[ 2.9101e-01, 1.3600e+00, 1.7595e-01, ..., 2.2270e-01,
-1.8886e-01, -6.0222e-01],
[ 3.9284e-01, 8.5335e-01, 1.6115e+00, ..., -2.1943e-02,
-4.5668e+00, -2.7120e+00],
[-4.5533e-01, 2.8235e-02, 1.4724e+00, ..., -4.8982e+00,
-9.8193e+00, 1.9282e+00],
...,
[-2.2304e+00, -2.7461e+00, -1.3621e+00, ..., -2.7383e+00,
-2.5697e+00, 2.1164e-02],
[-2.1206e+00, -2.7447e+00, -3.2864e+00, ..., -2.8692e+00,
-2.6062e+00, -9.2035e-03],
[-7.7430e-01, -3.3485e+00, -4.1533e+00, ..., -3.8574e+00,
-3.0529e+00, -1.2267e+00]],
[[-1.1272e+00, -1.5981e+00, -2.9530e+00, ..., -4.5077e+00,
-5.3197e+00, -2.1848e+00],
[-4.0750e+00, -4.2202e+00, -2.9396e+00, ..., -1.0546e+01,
-1.1349e+01, -3.3940e+00],
[-3.7685e+00, -4.5693e+00, -4.4987e+00, ..., -1.5359e+01,
-1.3106e+01, -8.1189e-01],
...,
[-6.6094e+00, -9.2660e+00, -6.8696e+00, ..., -7.7013e+00,
-5.8174e+00, -1.3591e+00],
[-6.5686e+00, -5.5281e+00, -9.6691e+00, ..., -8.5635e+00,
-8.1156e+00, -3.5010e+00],
[-2.1141e+00, -2.1696e+00, -4.0688e+00, ..., -3.9886e+00,
-3.3487e+00, -5.1745e-01]],
[[-4.5560e-02, 1.2946e+00, 1.5787e+00, ..., 6.6212e-01,
2.2596e+00, 7.7041e-01],
[-3.9358e-02, -1.3188e-01, 7.3054e-01, ..., -1.3693e+00,
3.1073e+00, 2.1146e+00],
[ 6.8723e-01, 1.4485e+00, 1.5620e+00, ..., -1.9163e+00,
3.9685e+00, 1.4950e+00],
...,
[ 2.9027e-02, -1.9597e+00, -2.1128e+00, ..., -5.2267e-02,
5.6428e-01, 9.3920e-02],
[-3.0029e-01, 3.5055e-01, -1.0536e+00, ..., -3.0933e-01,
2.7779e-01, -6.6394e-01],
[ 7.4753e-02, -1.7319e-01, 4.3206e-01, ..., -2.9394e-01,
-2.1083e-01, -1.7142e+00]],
...,
[[-6.3346e-01, -1.5082e+00, -2.0485e+00, ..., -1.6725e+00,
-4.2341e+00, -9.4464e-01],
[-2.7511e+00, -2.5935e+00, -1.1910e+00, ..., -2.9875e+00,
-7.4712e+00, -1.8133e+00],
[-2.0210e+00, -1.0275e+00, -7.3501e-01, ..., -5.7569e+00,
-3.8851e+00, 2.9005e+00],
...,
[-3.3274e+00, -4.7841e+00, -4.1375e+00, ..., -2.7023e+00,
-4.4595e+00, -1.2783e+00],
[-2.1260e+00, -2.9002e+00, -5.8170e+00, ..., -2.2120e+00,
-4.0674e+00, -4.7852e-02],
[-1.2365e+00, -1.5729e+00, -2.6809e+00, ..., -1.5336e+00,
-2.0264e+00, 1.0051e+00]],
[[-1.3505e+00, -2.4467e+00, -3.8466e+00, ..., -2.9640e+00,
-1.3753e+00, 9.0500e-01],
[-3.7861e+00, -3.4141e+00, -4.1743e+00, ..., -5.2620e+00,
-7.0581e+00, -1.7400e+00],
[-4.6695e+00, -3.7459e+00, -3.8028e+00, ..., -5.2302e+00,
-5.2458e+00, -4.2649e+00],
...,
[-3.3628e+00, 1.7787e+00, 1.7025e+00, ..., -3.3547e+00,
-3.2429e+00, -5.0419e-01],
[-2.9809e+00, -7.6734e-01, 1.5179e+00, ..., -8.1616e-01,
-1.6621e+00, 5.8076e-01],
[-2.7356e+00, -2.7875e+00, -9.0166e-01, ..., 2.3262e-01,
-6.4086e-01, 5.6331e-01]],
[[ 1.2259e+00, -2.2406e-02, 1.0995e+00, ..., -2.5115e-01,
1.1438e+00, -1.0477e-02],
[ 2.2945e+00, 1.1481e+00, 1.0268e+00, ..., 2.6451e+00,
1.5074e+00, 2.4531e+00],
[ 1.2827e+00, 7.8650e-01, 1.1472e+00, ..., 2.5459e+00,
-1.7019e+00, 4.4691e+00],
...,
[ 2.4897e+00, 5.8670e+00, 2.1699e+00, ..., 1.9054e-01,
9.7205e-01, 9.6691e-01],
[ 2.2004e+00, 3.9608e+00, 3.2674e+00, ..., 5.6666e-01,
6.2935e-01, 1.5121e+00],
[ 2.1756e+00, 1.2072e+00, 3.9162e-01, ..., -8.0150e-02,
-2.8542e-01, 7.1804e-01]]],
[[[ 2.4305e-01, 1.3390e+00, 5.5524e-01, ..., -6.2077e-01,
-5.5838e-02, 6.7701e-01],
[ 4.4479e-01, 1.2882e-01, 3.6584e-01, ..., -1.6839e+00,
-1.0008e+00, 9.8707e-01],
[ 2.3275e-02, 2.9804e-02, 8.7625e-01, ..., 3.8229e-01,
3.3055e-01, 3.3931e-01],
...,
[ 4.7815e-01, -7.4466e-01, 6.3402e-01, ..., -2.6305e-01,
-5.0313e-01, -9.7815e-01],
[-1.2533e+00, -1.2846e+00, -1.9438e+00, ..., -1.8242e+00,
-2.3669e+00, -1.3137e+00],
[-1.2216e-01, -2.4671e+00, -2.2887e+00, ..., -2.5054e+00,
-2.4333e+00, -1.0226e+00]],
[[-9.1484e-01, -2.3707e+00, -4.2417e+00, ..., -3.5711e+00,
-5.1811e+00, -3.6264e-01],
[-4.9708e+00, -6.3229e+00, -5.1995e+00, ..., -4.6547e+00,
-7.7345e+00, -1.9476e+00],
[-3.8214e+00, -5.4650e+00, -6.1858e+00, ..., -2.5504e+00,
-7.1652e+00, -3.3266e+00],
...,
[-4.4256e+00, -5.1537e+00, -3.6588e+00, ..., -4.3245e+00,
-4.9068e+00, -2.2400e+00],
[-7.3430e+00, -8.7422e+00, -7.5912e+00, ..., -8.2737e+00,
-9.0405e+00, -4.0384e+00],
[-1.0362e+00, -3.1640e+00, -2.2723e+00, ..., -3.0175e+00,
-2.9340e+00, -3.3925e-01]],
[[-3.5132e-01, 2.0725e+00, 1.2011e+00, ..., 8.9864e-01,
2.1486e+00, 9.4863e-01],
[-1.1180e-02, 9.4236e-01, 7.2988e-01, ..., 1.2131e+00,
1.5594e+00, -4.8359e-01],
[ 1.6761e-01, 1.8873e+00, 1.3337e+00, ..., 1.9006e+00,
2.3570e+00, -1.3034e-01],
...,
[ 9.2352e-02, 1.1519e+00, 9.7897e-01, ..., 9.6284e-01,
1.3141e+00, 6.1063e-01],
[ 3.2022e-01, 9.0518e-01, 5.6352e-01, ..., 1.7042e+00,
1.9355e+00, 5.5133e-01],
[-5.8849e-01, 6.1336e-01, 2.4705e-01, ..., 6.7526e-01,
7.2718e-01, -7.7816e-01]],
...,
[[-9.0924e-01, -1.8657e+00, -1.9116e+00, ..., -2.4850e+00,
-3.0866e+00, -5.2866e-01],
[-3.3333e+00, -4.4674e+00, -2.6607e+00, ..., -3.2793e+00,
-3.9571e+00, -2.9834e+00],
[-1.7190e+00, -2.5557e+00, -1.2633e+00, ..., -2.1468e+00,
-3.7720e+00, -1.8912e+00],
...,
[-9.3680e-01, -1.2395e+00, -4.2870e-01, ..., -1.6785e+00,
-3.7662e+00, -1.7777e+00],
[-2.7144e+00, -3.4236e+00, -2.6437e+00, ..., -2.8262e+00,
-4.5266e+00, -1.4537e+00],
[-8.6374e-01, -1.5354e+00, -2.6757e-01, ..., -6.0730e-01,
-1.1718e+00, 1.1971e+00]],
[[-1.7398e+00, -2.1371e+00, -2.9596e+00, ..., -1.4835e+00,
-1.3986e+00, -1.3103e+00],
[-4.4234e+00, -3.3414e+00, -3.0278e+00, ..., -1.7422e+00,
-1.2585e+00, -8.5104e-01],
[-2.8981e+00, -1.6667e+00, -2.0883e+00, ..., 1.6676e-01,
-8.8392e-02, -3.7641e-01],
...,
[-3.7799e+00, -3.5176e+00, -4.4371e+00, ..., -2.0328e+00,
-2.9582e+00, -1.4362e+00],
[-4.3380e+00, -1.9863e+00, -2.7122e+00, ..., -1.0950e+00,
-2.1524e+00, -6.4933e-01],
[-3.0185e+00, -1.8225e+00, -8.5934e-01, ..., -6.2102e-01,
-8.5336e-01, 3.5944e-02]],
[[ 1.1276e+00, 5.7181e-01, 7.1199e-01, ..., -5.5620e-01,
1.7824e-01, -3.6366e-01],
[ 3.7350e+00, 3.3607e+00, 2.1725e+00, ..., 1.8957e+00,
2.2210e+00, 2.2979e+00],
[ 1.7942e+00, 1.3403e+00, 1.2536e+00, ..., 3.6198e+00,
3.3596e+00, 1.4391e+00],
...,
[ 2.7191e+00, 1.2896e+00, 1.4488e+00, ..., 1.5131e+00,
1.1765e+00, 5.0533e-01],
[ 3.3342e+00, 4.1830e+00, 2.7062e+00, ..., 2.0592e+00,
1.2655e+00, 1.5560e+00],
[ 2.2274e+00, 1.6633e+00, 5.3192e-01, ..., 7.9814e-02,
9.2073e-02, 9.9463e-01]]],
[[[-7.8663e-02, 5.3588e-01, 2.1822e-01, ..., -1.5768e+00,
-1.2822e+00, -1.0466e+00],
[ 1.5778e+00, 3.2786e-01, 1.6548e+00, ..., -6.5243e-01,
1.8839e-01, -1.4435e+00],
[-1.7248e+00, -3.4655e+00, -2.9428e+00, ..., -4.6538e-01,
-2.9809e-01, -1.9485e+00],
...,
[-6.6669e-01, -5.3511e+00, -2.3060e+00, ..., -9.1906e-01,
-7.9059e+00, 1.3138e+00],
[-2.9324e+00, -8.9993e+00, -3.6715e+00, ..., -9.0369e+00,
-4.6130e+00, 2.3552e+00],
[-2.8474e+00, -7.8404e+00, -3.7375e+00, ..., -2.8903e+00,
5.7533e+00, 2.9897e+00]],
[[-1.0042e+00, -2.0892e+00, -4.1139e+00, ..., -3.3480e+00,
-5.1085e+00, -2.8988e+00],
[-3.2805e+00, -4.9354e+00, -5.4830e+00, ..., -4.0756e+00,
-5.5314e+00, -5.0918e+00],
[-4.7113e+00, -7.5900e+00, -7.4129e+00, ..., -3.7465e+00,
-4.6114e+00, -3.6214e+00],
...,
[-6.6292e+00, -6.3631e+00, 2.3639e+00, ..., -8.9045e+00,
-1.3552e+01, 8.2213e-01],
[-8.5962e+00, -1.2093e+01, 4.6354e-01, ..., -1.8988e+01,
-9.9461e+00, 1.1118e+00],
[-8.9106e-01, -4.2983e+00, 1.6482e+00, ..., -2.7013e-01,
2.2545e+00, 1.9451e+00]],
[[ 7.4983e-01, 1.3645e+00, 1.5254e+00, ..., 1.5355e+00,
1.9958e+00, 8.3534e-01],
[-6.1596e-02, 1.4691e+00, 2.6895e+00, ..., 1.3562e+00,
2.4822e+00, 1.0441e+00],
[ 2.8743e-01, 4.0390e+00, 5.3190e+00, ..., 9.9215e-01,
2.5012e+00, 2.1079e+00],
...,
[-1.9101e+00, -6.4250e-01, -9.0541e-01, ..., -1.1401e+01,
-2.1272e+01, -8.0617e+00],
[-3.0835e-01, 5.7204e-01, -1.4905e+00, ..., -1.8888e+01,
-1.4518e+01, -7.2384e+00],
[-5.1549e-01, 8.8175e-02, -5.1750e+00, ..., -1.1699e+01,
-1.0576e+01, -6.5227e+00]],
...,
[[-1.0056e+00, -1.4947e+00, -2.5077e+00, ..., -2.8767e+00,
-3.7538e+00, -1.3471e+00],
[-2.3998e+00, -4.1538e+00, -3.6699e+00, ..., -3.3702e+00,
-4.8745e+00, -3.6694e+00],
[-2.9399e+00, -3.1123e+00, -5.8796e+00, ..., -3.1023e+00,
-4.0045e+00, -2.8107e+00],
...,
[-2.2293e+00, -7.2876e-01, 1.3107e+00, ..., -4.8929e+00,
-9.2673e-01, 4.7407e-01],
[-4.1010e+00, -5.7166e+00, -1.9882e+00, ..., -5.3916e+00,
-2.2690e-01, -1.7310e+00],
[-3.1008e+00, -2.7745e+00, 1.4631e+00, ..., -2.4952e+00,
3.8417e-01, -1.2653e+00]],
[[-1.8473e+00, -1.9908e+00, -3.2026e+00, ..., -3.1869e+00,
-2.9265e+00, -3.6346e-01],
[-3.7131e+00, -3.1001e+00, -4.4641e+00, ..., -4.3278e+00,
-5.7308e+00, -1.7506e+00],
[-2.6286e+00, -2.2794e+00, -7.8997e+00, ..., -1.9480e+00,
-3.8927e+00, -1.4582e+00],
...,
[-2.2035e+00, 7.1415e-01, -6.4053e+00, ..., 1.3387e+00,
4.2425e+00, 3.4804e+00],
[-3.9180e+00, 2.3192e+00, -6.6880e+00, ..., -1.9385e+00,
5.4506e+00, 6.2225e+00],
[-3.0224e+00, 8.4726e-01, 3.0089e-01, ..., 6.2800e-01,
4.7433e+00, 1.5425e+00]],
[[ 9.5599e-01, 1.5662e-01, 9.0822e-01, ..., 4.1256e-01,
1.4831e+00, 6.8246e-01],
[ 2.7663e+00, 6.8093e-01, 1.3584e+00, ..., 1.0127e+00,
2.3165e+00, 3.4022e+00],
[ 1.8912e+00, 2.0242e-01, 6.5116e-01, ..., 2.2160e-01,
1.2548e+00, 8.6801e-01],
...,
[ 4.3088e+00, 2.2946e+00, 8.2670e-01, ..., -4.2792e+00,
-4.6506e+00, 1.8547e+00],
[ 3.7486e+00, 3.7253e+00, -3.5385e-01, ..., -3.9792e+00,
2.3959e+00, 6.2083e+00],
[ 1.0536e+00, 6.2439e-02, -3.2762e-01, ..., -3.9759e+00,
6.6621e+00, 6.0373e+00]]],
[[[ 9.1469e-02, -2.7748e-01, -5.4670e-01, ..., -2.9026e+00,
-1.5589e+00, -1.1670e+00],
[-1.6135e+00, -1.8236e+00, -3.1894e+00, ..., 2.0827e+00,
1.6784e+00, -1.7644e+00],
[-5.0552e-01, 1.5518e+00, 1.7793e+00, ..., 1.0620e+00,
1.2484e+00, -1.3623e+00],
...,
[ 9.9205e-01, 1.1524e+00, 3.3536e+00, ..., 6.5993e-01,
6.2495e-01, -2.9021e-01],
[ 7.7945e-01, -5.1318e-02, 1.0451e+00, ..., 9.1398e-01,
1.8105e-01, -1.3623e+00],
[-1.7382e-01, -2.7676e+00, -2.9006e+00, ..., -2.3527e+00,
-3.0685e+00, -3.0669e+00]],
[[-1.1756e+00, -2.6827e+00, -4.3040e+00, ..., -2.8080e+00,
-3.3942e+00, -2.9059e+00],
[-5.2889e+00, -6.7813e+00, -1.0001e+01, ..., 1.8994e+00,
-2.2972e+00, -5.1475e+00],
[ 2.4598e-01, 1.3328e+00, 1.1213e+00, ..., 1.5396e-01,
-1.3189e+00, -3.6408e+00],
...,
[ 8.9450e-02, -1.8147e+00, -3.3429e+00, ..., -1.9005e+00,
-3.0364e+00, -2.4295e+00],
[-6.5369e+00, -9.6295e+00, -9.5637e+00, ..., -3.4744e+00,
-5.4552e+00, -5.5073e+00],
[-6.9753e-01, -3.6219e+00, -3.6712e+00, ..., -2.8721e+00,
-2.7719e+00, -1.7354e+00]],
[[ 6.8900e-03, 2.5082e+00, 2.0380e+00, ..., 3.8672e+00,
2.7098e+00, 5.9672e-01],
[ 3.9400e-01, 2.6141e+00, 5.5811e+00, ..., 1.1826e+00,
3.0966e+00, 1.0549e+00],
[-5.2515e-02, -2.5310e+00, -1.9986e+00, ..., -4.2267e+00,
2.7058e+00, 1.7791e+00],
...,
[-2.0411e+00, -1.7809e+00, -2.1446e+00, ..., 2.1429e-01,
8.8096e-01, 5.7093e-01],
[-2.3840e+00, 1.8688e-01, -6.7654e-01, ..., 7.8833e-01,
9.8196e-01, 3.2330e-01],
[-1.1607e+00, 9.8161e-01, 1.0014e-01, ..., 1.1243e+00,
1.6415e+00, 5.2447e-01]],
...,
[[-1.5176e+00, -3.2681e+00, -2.8975e+00, ..., -3.2882e+00,
-3.1242e+00, -1.1762e+00],
[-4.4672e+00, -5.0254e+00, -5.6333e+00, ..., 7.8439e-01,
-3.7268e+00, -2.6830e+00],
[-5.6361e-01, 8.3188e-01, 1.3216e+00, ..., 6.5524e-01,
-4.8910e+00, -1.6071e+00],
...,
[ 1.5597e+00, -8.9217e-01, -8.3831e-01, ..., -5.1932e-01,
-2.6341e+00, -1.0468e+00],
[-1.8911e+00, -5.7759e+00, -1.8523e+00, ..., -3.2264e-01,
-3.0504e+00, -1.7330e+00],
[-1.0412e+00, -2.7991e+00, -2.0752e-01, ..., -1.5009e+00,
-2.5539e+00, -5.8285e-01]],
[[-1.8197e+00, -2.8666e+00, -2.1772e+00, ..., -1.6633e+00,
-3.6014e-01, 1.4050e-01],
[-4.9860e+00, -6.9783e+00, -8.2933e+00, ..., -1.2628e+00,
-3.5728e+00, -1.4769e-01],
[-8.6044e-01, 6.5051e-01, 6.7791e-01, ..., -5.4881e-01,
-3.5605e+00, -7.5726e-02],
...,
[-2.3698e+00, -1.5367e+00, 5.8390e-01, ..., -6.6472e-01,
-4.6948e+00, -2.4265e+00],
[-3.6088e+00, -1.2501e+00, 1.2486e+00, ..., -5.1565e-01,
-4.7004e+00, -2.2474e+00],
[-2.8200e+00, -2.5483e+00, 1.5458e-01, ..., -6.3518e-01,
-3.1897e+00, -1.4064e+00]],
[[ 1.4473e+00, 3.3504e-02, 2.4533e-01, ..., 7.1473e-01,
1.7260e+00, 5.7847e-01],
[ 2.1743e+00, 2.8029e+00, 3.2251e+00, ..., 1.2691e+00,
3.4807e+00, 2.2016e+00],
[ 1.8665e-02, 1.0402e+00, -1.5839e-01, ..., -2.6625e-01,
1.2784e+00, 2.1690e-01],
...,
[ 1.2700e+00, 1.6617e+00, 3.4845e+00, ..., 7.7811e-01,
2.8383e-01, 7.3904e-01],
[ 4.8149e+00, 4.5898e+00, 3.8686e+00, ..., 1.5151e+00,
8.4633e-01, 1.4614e+00],
[ 2.1743e+00, 4.7694e-01, 8.0604e-01, ..., 7.0900e-01,
-4.1467e-01, -1.1336e-01]]],
[[[ 6.3497e-01, 3.9513e+00, -2.1807e-01, ..., -3.9411e+00,
-2.6500e+00, -9.3654e-01],
[ 1.5306e+00, -1.7889e-01, -3.8990e+00, ..., -3.3028e+00,
-2.9503e+00, -2.8174e-01],
[ 1.4787e+00, -4.7122e+00, -4.9751e+00, ..., -1.6693e+00,
-3.5905e+00, -4.4776e-01],
...,
[ 8.6933e-01, -7.0258e-01, -1.5220e+00, ..., -2.9490e+00,
-3.5749e+00, 4.4338e-01],
[ 4.0289e-01, -1.9776e-01, -9.2993e-01, ..., -4.6751e+00,
-3.6000e+00, 5.1013e-01],
[-4.5822e-01, -3.4939e+00, -3.2890e+00, ..., -4.0409e+00,
-3.2205e+00, -1.5074e+00]],
[[ 5.0340e-01, 8.2627e-01, 1.4693e-01, ..., -8.8002e+00,
-7.3473e+00, -1.2647e+00],
[ 2.4081e+00, -3.0640e+00, -5.9654e+00, ..., -1.0143e+01,
-9.5030e+00, -2.6194e+00],
[ 1.7835e+00, -6.6996e+00, -4.4898e+00, ..., -7.3712e+00,
-1.0122e+01, -2.7964e+00],
...,
[-1.8558e+00, -3.2588e+00, -3.8439e+00, ..., -8.7411e+00,
-8.8955e+00, -2.3093e+00],
[-4.1594e+00, -5.8747e+00, -5.0427e+00, ..., -9.5899e+00,
-1.1150e+01, -4.4468e+00],
[-2.8663e-01, -3.0759e+00, -3.2549e+00, ..., -3.8995e+00,
-3.8946e+00, -8.3178e-01]],
[[ 7.8828e-01, -7.2747e-02, -2.0827e-01, ..., 6.6447e-01,
2.0961e+00, 9.6597e-01],
[-2.7733e+00, -5.5356e+00, -8.3525e+00, ..., 1.6584e+00,
2.7604e+00, -9.9856e-02],
[-4.0898e+00, -3.8697e+00, -4.3549e+00, ..., 3.1492e+00,
2.0251e+00, -4.8851e-01],
...,
[ 1.6748e-01, 4.0398e+00, 1.2119e+00, ..., -1.4582e+00,
1.0543e+00, -1.0123e+00],
[ 6.4128e-01, 1.8909e+00, 4.8246e-01, ..., -1.0120e+00,
1.4689e+00, -1.4611e+00],
[-2.5997e-01, 1.5850e+00, 8.7521e-01, ..., 2.5656e-01,
-1.7087e-01, -1.9068e+00]],
...,
[[ 1.4717e+00, 1.3477e+00, 1.4178e+00, ..., -5.6896e+00,
-5.6060e+00, -6.3319e-01],
[ 6.1267e-01, -4.2106e+00, -1.1456e+00, ..., -4.2967e+00,
-6.3928e+00, -2.1551e+00],
[-1.0701e+00, -5.5127e+00, -5.8679e-01, ..., -5.0225e+00,
-6.5905e+00, -2.1381e+00],
...,
[-9.6242e-01, -2.5324e+00, -9.4954e-01, ..., -2.9820e+00,
-6.1176e+00, -1.8397e+00],
[-1.4534e+00, -3.4712e+00, -1.7445e+00, ..., -3.4083e+00,
-5.6922e+00, -9.6510e-01],
[-1.1238e+00, -3.4873e+00, -2.3794e+00, ..., -1.9728e+00,
-2.3920e+00, 3.1477e-01]],
[[-2.5101e+00, 9.0112e-01, 1.8776e+00, ..., -1.9630e+00,
-6.5392e-01, -6.5598e-01],
[-6.6223e+00, -2.0132e+00, 2.5322e+00, ..., -2.6637e+00,
-1.8945e+00, -4.7838e-01],
[-6.7707e+00, -5.5400e+00, 6.8522e-01, ..., -1.8293e+00,
-1.6788e+00, -9.9601e-01],
...,
[-2.6073e+00, -4.7687e+00, -2.6862e+00, ..., -3.7083e+00,
-1.9387e+00, -9.0043e-01],
[-2.9149e+00, -2.1569e+00, -7.1418e-01, ..., -2.2658e+00,
-6.9219e-01, 1.0366e+00],
[-3.1463e+00, -2.9677e+00, -1.0210e+00, ..., -1.1294e-01,
-8.4639e-01, 5.7026e-01]],
[[ 1.1130e+00, 3.3393e+00, -6.3159e-01, ..., -1.5013e-01,
5.5887e-01, -1.3507e+00],
[ 2.3672e+00, -1.6175e+00, -3.7074e+00, ..., 3.8606e+00,
2.7076e+00, 2.0970e+00],
[ 3.8555e-01, -5.9362e+00, -5.0317e+00, ..., 5.0958e+00,
1.7462e+00, 6.1823e-01],
...,
[ 1.9093e+00, 7.6845e-01, 1.7770e+00, ..., 2.4219e+00,
1.7203e+00, 3.9188e-01],
[ 3.1960e+00, 3.3549e+00, 2.5934e+00, ..., 2.3705e+00,
7.9351e-01, 1.8919e+00],
[ 1.9674e+00, 3.7395e-01, 4.4079e-01, ..., 4.9565e-02,
-2.3464e-01, 9.2963e-01]]]], device='cuda:0'),
tensor([[[[-3.0682e-01, 1.0124e+00, 7.9801e-01, ..., 3.6490e-01,
9.2637e-01, -3.8685e-01],
[ 1.0873e-01, -3.3353e-01, 1.4613e+00, ..., 1.0449e+00,
1.3508e+00, -8.8634e-01],
[-3.9560e-01, -1.1704e+00, -1.3159e-01, ..., -5.1240e+00,
-4.6034e+00, -3.6738e+00],
...,
[-4.2045e-01, -2.0619e-01, 1.4891e-01, ..., -4.1265e-02,
-4.7597e-01, -7.6709e-01],
[-1.5561e+00, -6.3917e-02, 1.3784e-01, ..., 5.6961e-01,
2.2900e-01, -1.1956e+00],
[-1.1653e+00, -2.2875e+00, -1.8255e+00, ..., -2.9662e+00,
-2.8081e+00, -2.7307e+00]],
[[-1.7308e+00, -2.5570e+00, -3.7300e+00, ..., -3.6586e+00,
-4.7356e+00, -3.1521e+00],
[-4.7907e+00, -5.4784e+00, -4.0857e+00, ..., -4.5028e+00,
-4.7883e+00, -4.3485e+00],
[-4.4227e+00, -5.7041e+00, -4.7143e+00, ..., -6.8411e+00,
-6.1327e+00, -2.6213e+00],
...,
[-4.7711e+00, -5.0912e+00, -4.5162e+00, ..., -3.3959e+00,
-4.1653e+00, -3.0418e+00],
[-7.5204e+00, -7.7242e+00, -6.2980e+00, ..., -4.4531e+00,
-5.2012e+00, -4.9844e+00],
[-1.4088e+00, -3.0877e+00, -2.0656e+00, ..., -2.8958e+00,
-2.7085e+00, -1.5990e+00]],
[[-2.0467e-01, 2.3371e+00, 1.7008e+00, ..., 1.7963e+00,
1.8091e+00, 6.8903e-01],
[-2.9021e-01, 2.3992e-01, 9.0441e-01, ..., 5.4710e-01,
1.1966e+00, 4.6567e-01],
[-2.2616e-01, 1.0783e+00, 5.3478e-01, ..., 3.9512e+00,
4.5864e+00, 2.9052e+00],
...,
[-1.1817e+00, 6.7250e-02, 1.0647e+00, ..., 9.9682e-01,
4.6663e-01, 4.4431e-01],
[-1.2459e+00, -1.7566e-01, 7.7715e-01, ..., 1.3357e+00,
1.0421e+00, 3.5938e-01],
[-4.7314e-01, -3.8965e-01, 5.2753e-02, ..., 1.6478e+00,
1.6668e+00, 5.1794e-01]],
...,
[[-1.4430e+00, -2.3779e+00, -2.4565e+00, ..., -1.4338e+00,
-2.0522e+00, -6.9615e-01],
[-3.7157e+00, -4.5625e+00, -3.2899e+00, ..., -2.6360e+00,
-3.0550e+00, -2.9184e+00],
[-2.6595e+00, -3.1215e+00, -2.2386e+00, ..., -3.5332e+00,
-3.4736e+00, -2.8592e+00],
...,
[-1.1143e+00, -2.4559e+00, -1.0563e+00, ..., -3.1588e+00,
-3.7844e+00, -1.8719e+00],
[-1.9822e+00, -3.9470e+00, -1.6354e+00, ..., -1.4725e+00,
-2.7314e+00, -1.5268e+00],
[-1.2592e+00, -1.5269e+00, -8.4614e-01, ..., -1.9106e+00,
-2.1110e+00, -5.5775e-01]],
[[-1.8597e+00, -1.9131e+00, -3.1538e+00, ..., -7.6362e-01,
-1.6558e+00, -9.3865e-01],
[-4.2032e+00, -2.2158e+00, -2.1532e+00, ..., -3.1364e-01,
-3.0712e+00, -1.6086e+00],
[-4.1015e+00, -2.6364e+00, -2.6630e+00, ..., -6.1036e+00,
-7.7313e+00, -3.8736e+00],
...,
[-2.6330e+00, -2.6949e+00, -3.0664e+00, ..., -2.7834e-01,
-2.6002e+00, -1.3227e+00],
[-1.8747e+00, -1.8854e-01, -1.5202e+00, ..., -1.2307e+00,
-3.4197e+00, -1.9133e+00],
[-2.0952e+00, -1.6177e+00, -1.7143e+00, ..., -1.6344e+00,
-2.7902e+00, -1.2211e+00]],
[[ 1.3126e+00, 2.4502e-01, 9.3190e-01, ..., 9.9822e-01,
1.8306e+00, 9.3609e-01],
[ 4.1646e+00, 3.1122e+00, 2.6516e+00, ..., 1.3917e+00,
1.4687e+00, 1.9893e+00],
[ 2.0748e+00, 4.2940e-01, 8.4490e-01, ..., -7.5362e-01,
-3.0385e-01, -4.1145e-01],
...,
[ 3.0032e+00, 2.6944e+00, 1.8693e+00, ..., 3.0579e-01,
-5.0119e-02, 2.6552e-01],
[ 3.6999e+00, 4.0755e+00, 2.7932e+00, ..., 1.2774e+00,
8.3346e-01, 1.2171e+00],
[ 2.3517e+00, 2.2510e+00, 1.5002e+00, ..., -1.0283e-01,
-2.9835e-01, -1.2354e-01]]],
[[[-8.1643e-02, 1.1951e+00, 6.6522e-02, ..., -3.2495e+00,
-2.3034e+00, -9.2502e-01],
[ 7.4094e-01, 7.2907e-01, 6.0248e-01, ..., -2.0122e+00,
-2.1661e+00, -1.3317e-01],
[ 1.5810e-01, -6.5984e-01, 4.5505e-01, ..., -1.7388e+00,
-2.8756e+00, 1.3808e-01],
...,
[-1.4393e-01, -9.4498e-01, -7.2707e-01, ..., -3.3017e+00,
-2.8885e+00, 8.0785e-01],
[-1.6337e+00, -1.3637e+00, -1.1058e+00, ..., -3.4741e+00,
-3.0791e+00, 5.9979e-01],
[-7.8703e-01, -2.9454e+00, -3.0960e+00, ..., -3.6510e+00,
-3.1369e+00, -1.4676e+00]],
[[-1.4047e+00, -2.0888e+00, -3.6181e+00, ..., -6.3700e+00,
-6.9302e+00, -1.1947e+00],
[-4.1180e+00, -4.7293e+00, -4.1659e+00, ..., -5.8338e+00,
-7.8087e+00, -2.2248e+00],
[-2.6189e+00, -3.7977e+00, -4.2138e+00, ..., -5.9208e+00,
-7.4287e+00, -2.0581e+00],
...,
[-4.8274e+00, -5.1783e+00, -4.7335e+00, ..., -7.6041e+00,
-7.4496e+00, -1.6500e+00],
[-6.8036e+00, -7.4585e+00, -7.9338e+00, ..., -9.0479e+00,
-1.0478e+01, -4.7165e+00],
[-1.4241e+00, -2.8704e+00, -3.5750e+00, ..., -3.5699e+00,
-3.8598e+00, -7.5948e-01]],
[[ 2.2057e-01, 1.3157e+00, 1.5757e+00, ..., 1.4719e+00,
2.3815e+00, 1.0039e+00],
[ 3.5141e-01, 3.2622e-01, 6.5730e-01, ..., 1.4835e+00,
2.8384e+00, 7.7629e-02],
[-3.4959e-01, 1.0833e+00, 1.6645e+00, ..., 3.9838e-01,
1.8342e+00, -1.1361e-01],
...,
[-6.8154e-01, 6.1900e-01, 9.1022e-01, ..., -8.0734e-02,
5.4599e-01, -9.7706e-01],
[ 5.9775e-01, 1.2026e+00, 6.6675e-01, ..., -1.4069e+00,
3.8419e-01, -1.4586e+00],
[ 3.3425e-01, 1.0168e+00, 1.2944e+00, ..., 1.9266e-01,
-2.3399e-01, -1.8676e+00]],
...,
[[-9.6159e-01, -1.7431e+00, -2.3774e+00, ..., -5.0052e+00,
-4.9753e+00, -6.5861e-01],
[-2.9588e+00, -3.1976e+00, -2.2026e+00, ..., -3.6113e+00,
-4.5417e+00, -1.7976e+00],
[-1.9268e+00, -4.1212e-01, -1.2881e-01, ..., -5.0285e+00,
-5.0666e+00, -1.0377e+00],
...,
[-1.2846e+00, -5.7897e-01, 2.8165e-01, ..., -3.4320e+00,
-4.5420e+00, -1.7783e+00],
[-2.8552e+00, -2.9857e+00, -2.2671e+00, ..., -3.3971e+00,
-5.6346e+00, -1.0611e+00],
[-1.0505e+00, -1.6921e+00, -1.8358e+00, ..., -2.1058e+00,
-2.3023e+00, 4.3468e-01]],
[[-1.6901e+00, -2.0560e+00, -4.0020e+00, ..., -1.2765e+00,
-5.4997e-01, -6.1096e-01],
[-3.9297e+00, -3.9194e+00, -4.6299e+00, ..., -3.2227e+00,
-1.7345e+00, -3.8692e-01],
[-3.4723e+00, -2.6974e+00, -2.9568e+00, ..., -2.6172e+00,
-1.6751e+00, -7.2148e-01],
...,
[-3.6557e+00, -3.7027e+00, -3.8791e+00, ..., -3.1422e+00,
-1.7208e+00, -3.0419e-01],
[-3.9688e+00, -2.4935e+00, -2.9067e+00, ..., -9.2798e-01,
-2.9569e-01, 1.4155e+00],
[-2.9744e+00, -3.1485e+00, -2.3999e+00, ..., 1.5910e-01,
-7.6395e-01, 6.8789e-01]],
[[ 1.1978e+00, -2.7838e-02, 1.1291e+00, ..., 4.5442e-01,
7.2716e-01, -1.3457e+00],
[ 2.6601e+00, 1.5177e+00, 1.1439e+00, ..., 3.5790e+00,
2.9665e+00, 2.3138e+00],
[ 1.3622e+00, -1.9530e-01, 3.9371e-02, ..., 2.8130e+00,
1.6752e+00, 1.1496e+00],
...,
[ 1.8861e+00, 7.4523e-01, -2.0727e-01, ..., 2.3196e+00,
2.6317e+00, 6.2161e-01],
[ 2.6179e+00, 4.1165e+00, 2.7802e+00, ..., 3.3987e+00,
1.0579e+00, 1.7462e+00],
[ 1.8449e+00, 1.4137e+00, 2.5833e-01, ..., -2.5531e-02,
-4.9163e-01, 8.5074e-01]]],
[[[ 1.2846e+00, 9.9050e-01, 5.5454e-02, ..., -2.9094e+00,
-1.5518e+00, -1.0697e+00],
[ 5.8718e-01, 1.6154e+00, 1.0738e+00, ..., 1.3158e+00,
8.4146e-01, -2.0716e+00],
[-2.4492e+00, -8.6016e-01, -4.2183e+00, ..., 3.2515e+00,
2.1757e+00, -2.2341e+00],
...,
[-4.1714e+00, -1.8547e+00, -4.9708e+00, ..., -6.5228e-01,
-3.5303e-01, -1.0028e+00],
[ 7.5138e-01, 1.9676e+00, 2.1879e+00, ..., -7.4231e-01,
-1.2029e+00, -1.7149e+00],
[-2.0080e+00, -4.1160e+00, -5.6694e+00, ..., -2.6575e+00,
-3.1277e+00, -2.9808e+00]],
[[-1.6290e+00, -4.4616e+00, -8.3991e+00, ..., -3.4043e+00,
-3.5967e+00, -2.3550e+00],
[-3.1536e+00, -3.4393e+00, -4.8159e+00, ..., -8.7389e-01,
-3.1162e+00, -4.0650e+00],
[-4.0859e+00, -9.2809e+00, -8.3894e+00, ..., 2.0373e+00,
-1.8505e+00, -4.4238e+00],
...,
[ 6.2361e-01, 1.2410e+00, -5.0549e+00, ..., -3.1954e+00,
-4.3976e+00, -3.5819e+00],
[-6.8961e+00, -6.2897e+00, -6.9609e+00, ..., -5.3790e+00,
-6.2793e+00, -5.6364e+00],
[-2.2687e+00, -6.6376e+00, -8.0627e+00, ..., -3.1223e+00,
-3.0572e+00, -1.7653e+00]],
[[ 6.4232e-01, 3.1332e+00, 1.7201e+00, ..., 2.9406e+00,
2.6152e+00, 8.0147e-01],
[-3.5137e-01, 3.0900e+00, 2.9439e+00, ..., 3.1498e+00,
4.0380e+00, 1.4897e+00],
[ 1.7772e+00, 6.3694e+00, 6.5004e+00, ..., -5.5683e-02,
3.3253e+00, 2.1667e+00],
...,
[-9.8702e-01, 7.4682e-01, 3.3780e-01, ..., 1.6924e+00,
-7.4285e-01, 1.5098e-02],
[-2.4816e+00, -4.8830e+00, -3.6227e+00, ..., 4.5004e-01,
9.3190e-01, 9.2810e-01],
[-3.7854e-01, -1.1421e+00, -2.5886e+00, ..., 1.0687e+00,
1.8799e+00, 4.8556e-01]],
...,
[[-6.0707e-01, -3.1569e+00, -1.9904e+00, ..., -4.0406e+00,
-3.8558e+00, -1.4051e+00],
[-2.6395e+00, -4.2692e+00, -2.6846e+00, ..., -1.1695e+00,
-3.8333e+00, -3.8350e+00],
[-3.3040e+00, -5.4593e+00, -4.5876e+00, ..., -9.8656e-01,
-3.3864e+00, -2.3371e+00],
...,
[ 1.8552e+00, 4.7333e-01, -2.8710e+00, ..., -1.3029e+00,
-4.0209e+00, -2.2696e+00],
[-8.1330e-01, -3.9714e+00, -2.0928e+00, ..., -1.6481e+00,
-4.0295e+00, -2.2147e+00],
[-4.3173e+00, -8.4758e+00, -7.2978e+00, ..., -2.1401e+00,
-2.7868e+00, -6.6433e-01]],
[[-2.2639e+00, -2.9007e+00, -1.0236e+00, ..., 1.4505e+00,
-1.1336e+00, -2.8081e-01],
[-3.4088e+00, -3.4810e+00, -1.6247e+00, ..., -9.4586e-01,
-3.7887e+00, -1.7233e+00],
[-5.2670e+00, -9.4366e+00, -6.1600e+00, ..., -6.5898e-01,
-3.1948e+00, -1.1265e+00],
...,
[-4.1401e+00, -7.8825e+00, -4.7100e+00, ..., -1.4464e-01,
-2.4946e+00, -1.3743e+00],
[ 5.4291e-02, 4.5416e-01, -8.2505e-01, ..., 4.4261e-01,
-2.6066e+00, -1.4481e+00],
[-4.7058e-01, -6.7538e-01, -8.2513e-01, ..., -7.6461e-01,
-2.7502e+00, -1.2940e+00]],
[[ 2.1495e+00, 1.8044e+00, 2.2617e+00, ..., 1.2803e+00,
1.1015e+00, 1.3816e-01],
[ 1.7482e+00, 2.4675e+00, 2.7656e+00, ..., 3.8267e+00,
3.8823e+00, 2.7582e+00],
[ 2.8452e-02, 3.6968e+00, 7.7196e-01, ..., 1.9087e+00,
3.1127e+00, 2.4466e-01],
...,
[-4.6068e+00, -8.9414e+00, -8.8690e+00, ..., 1.2519e+00,
1.2610e+00, 1.1193e+00],
[ 2.9379e+00, 2.8076e+00, 9.4135e-01, ..., 2.2084e+00,
9.2482e-01, 9.6082e-01],
[ 2.1180e+00, 3.8044e+00, 3.4401e+00, ..., 7.6170e-01,
-2.0466e-01, -1.2194e-01]]],
[[[ 9.0183e-01, 1.6092e+00, 6.4537e-01, ..., -3.8334e+00,
-2.6822e+00, -1.0807e+00],
[ 1.4100e+00, 9.0431e-01, 9.0176e-01, ..., -3.1399e+00,
-2.6758e+00, -2.7923e-01],
[ 9.9722e-02, -9.2962e-01, -6.0466e-01, ..., -3.4493e+00,
-3.5830e+00, -5.1506e-02],
...,
[ 5.5237e-01, -1.0428e+00, -2.7617e-01, ..., -3.5652e+00,
-4.1213e+00, 3.4791e-01],
[ 2.4633e-01, -1.7212e-01, -4.8890e-01, ..., -4.3875e+00,
-3.6721e+00, 4.9036e-01],
[-3.2083e-01, -3.5353e+00, -3.2937e+00, ..., -4.1176e+00,
-3.3404e+00, -1.4642e+00]],
[[ 3.7431e-01, -1.0782e+00, -2.4109e+00, ..., -7.5825e+00,
-7.5723e+00, -1.4452e+00],
[-8.0842e-01, -2.4087e+00, -3.1711e+00, ..., -8.4432e+00,
-8.4369e+00, -2.4448e+00],
[-5.0280e-01, -2.2154e+00, -1.8713e+00, ..., -9.6209e+00,
-8.1513e+00, -2.4493e+00],
...,
[-2.2013e+00, -3.5907e+00, -3.1805e+00, ..., -8.9844e+00,
-8.8190e+00, -2.1531e+00],
[-4.1858e+00, -5.8967e+00, -5.1956e+00, ..., -9.6785e+00,
-1.1099e+01, -4.5208e+00],
[-3.3965e-01, -3.3994e+00, -3.7660e+00, ..., -3.9498e+00,
-3.8264e+00, -7.6648e-01]],
[[ 4.5495e-01, 2.2418e+00, 1.7742e+00, ..., 8.2933e-01,
2.1825e+00, 9.6721e-01],
[-3.6200e-01, 1.0293e+00, 6.3833e-01, ..., 1.1331e+00,
2.7538e+00, 7.1290e-02],
[ 9.6970e-01, 3.9827e+00, 2.5397e+00, ..., -4.2460e-01,
2.1048e+00, -4.3770e-01],
...,
[ 1.1346e+00, 2.7833e+00, 9.8186e-01, ..., -6.8819e-01,
1.1524e+00, -9.1706e-01],
[ 6.1334e-01, 1.9422e+00, 4.9218e-01, ..., -6.2380e-01,
1.6450e+00, -1.2701e+00],
[-3.1616e-01, 1.9087e+00, 9.7620e-01, ..., 1.9616e-01,
-1.5230e-01, -1.9068e+00]],
...,
[[-2.9935e-01, -7.5741e-01, -1.3940e+00, ..., -5.6365e+00,
-5.5472e+00, -8.2806e-01],
[-2.4035e+00, -3.8077e+00, -1.9976e+00, ..., -5.3093e+00,
-5.2834e+00, -1.7326e+00],
[-1.5526e+00, -2.9098e+00, -1.6852e+00, ..., -6.2627e+00,
-6.8445e+00, -1.2202e+00],
...,
[-1.3408e+00, -2.7462e+00, -1.2873e+00, ..., -3.7725e+00,
-6.3346e+00, -1.9297e+00],
[-1.8446e+00, -3.9046e+00, -2.1549e+00, ..., -3.1031e+00,
-5.3874e+00, -9.2205e-01],
[-1.1994e+00, -3.9077e+00, -2.8539e+00, ..., -1.7801e+00,
-2.3084e+00, 3.6868e-01]],
[[-9.6805e-01, -8.5948e-01, -1.0456e+00, ..., -2.1975e+00,
-4.2598e-01, -6.1264e-01],
[-2.7046e+00, -2.3788e+00, -1.2806e+00, ..., -3.6579e+00,
-1.9726e+00, -5.2493e-01],
[-2.9990e+00, -5.0547e+00, -1.7434e+00, ..., -1.6613e+00,
-1.7091e+00, -5.6521e-01],
...,
[-2.8459e+00, -4.2595e+00, -2.0556e+00, ..., -2.2622e+00,
-8.6021e-01, -7.6661e-01],
[-3.0038e+00, -2.1928e+00, -6.8191e-01, ..., -1.3521e+00,
-1.4011e-01, 1.0942e+00],
[-3.3098e+00, -3.4338e+00, -1.2867e+00, ..., -1.1205e-01,
-7.1807e-01, 6.3250e-01]],
[[ 8.9319e-01, 7.8485e-01, 8.5934e-01, ..., -2.8614e-01,
6.3440e-01, -1.3386e+00],
[ 1.3711e+00, 1.5498e+00, 1.4648e+00, ..., 3.3727e+00,
3.0960e+00, 2.3309e+00],
[ 3.4969e-01, -3.6284e-01, 6.2771e-01, ..., 2.3288e+00,
1.1505e+00, 1.2422e+00],
...,
[ 1.6674e+00, 1.2819e+00, 1.7845e+00, ..., 2.6094e+00,
2.0974e+00, 4.2491e-01],
[ 3.1937e+00, 3.5754e+00, 2.4563e+00, ..., 2.2112e+00,
9.6484e-01, 1.8431e+00],
[ 1.9320e+00, 5.0462e-01, 6.7191e-01, ..., 3.5451e-01,
-1.2136e-01, 8.3787e-01]]],
[[[ 7.0556e-01, 4.7897e-01, -7.4799e-01, ..., 5.0409e-02,
4.1739e-01, -1.9683e-01],
[ 1.0855e+00, 3.4020e-01, 9.1605e-01, ..., 9.4491e-02,
4.6561e-01, -5.3289e-01],
[-4.9331e-01, -4.0340e-01, 1.5958e+00, ..., -4.0048e-01,
-5.0505e-01, -9.2411e-01],
...,
[-3.4845e+00, -2.7058e+00, 5.3617e+00, ..., -2.7737e+00,
-4.3978e-01, -8.6834e-01],
[-3.3812e+00, 1.2224e+00, 4.7356e+00, ..., -3.9347e-01,
-3.7920e-01, -1.3940e+00],
[ 1.5740e-01, 2.6141e+00, -1.8886e-01, ..., -3.1537e+00,
-2.9258e+00, -2.6232e+00]],
[[-1.3549e+00, -3.9088e+00, -6.6158e+00, ..., -2.5667e+00,
-3.7105e+00, -2.3282e+00],
[-4.7996e+00, -5.9138e+00, -5.6018e+00, ..., -4.2336e+00,
-3.9238e+00, -3.5125e+00],
[-3.9682e+00, -4.9364e+00, -4.7323e+00, ..., -5.6788e+00,
-3.1750e+00, -2.7957e+00],
...,
[-3.5063e+00, -1.4408e+00, 1.7344e+00, ..., -5.5755e+00,
-3.6813e+00, -2.5378e+00],
[-3.3208e+00, -2.1841e+00, -3.2898e+00, ..., -4.4636e+00,
-5.1516e+00, -5.0156e+00],
[ 1.2263e-01, -1.1024e+00, -3.5661e+00, ..., -3.2372e+00,
-2.7765e+00, -1.7582e+00]],
[[ 7.7644e-01, 3.0372e+00, 1.8844e+00, ..., 1.6988e+00,
1.5045e+00, 9.3523e-01],
[-1.0894e-01, 1.6528e+00, 1.5651e+00, ..., 1.3314e+00,
1.1049e+00, 8.8991e-01],
[ 7.5055e-01, 2.0272e+00, 9.1787e-01, ..., 1.3739e+00,
1.5932e+00, 1.3951e+00],
...,
[ 7.3180e-02, -1.1862e+00, 7.2294e-03, ..., 1.1666e+00,
6.6495e-01, 7.5185e-01],
[-2.7271e-01, -2.2659e+00, -2.2365e+00, ..., 1.3843e-01,
-1.7915e-01, 6.0385e-01],
[-1.2971e+00, -1.8710e+00, -4.7450e-01, ..., 6.3199e-01,
1.7041e+00, 8.4258e-01]],
...,
[[-9.1563e-01, -3.3556e+00, -2.5093e+00, ..., -1.5785e+00,
-2.1756e+00, -4.8185e-01],
[-2.6879e+00, -3.9892e+00, -1.2810e+00, ..., -2.0512e+00,
-4.1728e+00, -2.9818e+00],
[-2.1647e+00, -2.7308e+00, -1.4611e+00, ..., -3.1238e-01,
-4.8552e+00, -1.9428e+00],
...,
[ 5.8315e-03, 2.6521e+00, -3.3580e-01, ..., -3.9963e+00,
-1.8295e+00, -1.3777e+00],
[-5.3458e-01, -1.4699e-01, -2.7344e+00, ..., -1.0976e+00,
-3.5507e+00, -2.1892e+00],
[ 1.1766e+00, -3.3330e-01, -1.9043e+00, ..., -2.0606e+00,
-2.7239e+00, -8.6298e-01]],
[[-2.5307e+00, -3.2731e+00, -1.5496e+00, ..., 6.6527e-01,
-1.3740e+00, -7.4424e-01],
[-4.1931e+00, -3.5521e+00, -2.1095e+00, ..., -5.9353e-01,
-3.8754e+00, -1.9296e+00],
[-3.6560e+00, -2.2281e+00, -1.4006e+00, ..., -1.6643e-01,
-3.8575e+00, -2.5576e+00],
...,
[-2.0408e+00, 6.1725e-02, -5.3268e-01, ..., -3.4720e+00,
-2.1116e+00, -9.9279e-01],
[-1.5733e+00, 2.4851e+00, -6.9617e-01, ..., -1.3353e+00,
-2.4709e+00, -1.1728e+00],
[-8.0656e-02, 7.4600e-01, -9.8681e-01, ..., 3.1822e-02,
-1.8259e+00, -8.4464e-01]],
[[ 1.5088e+00, 1.1793e+00, 1.4818e+00, ..., 5.7796e-01,
1.1828e+00, 6.9645e-01],
[ 3.0546e+00, 3.0866e+00, 3.3170e+00, ..., 1.5128e+00,
9.8110e-01, 2.5076e+00],
[ 1.4904e+00, 9.8969e-01, 2.2107e+00, ..., 1.0110e+00,
-3.1649e-01, 8.6230e-01],
...,
[-4.3400e+00, -3.1783e+00, 2.3892e+00, ..., 1.0315e+00,
9.5667e-01, 1.0738e+00],
[-3.2997e+00, 1.3795e+00, 4.0731e+00, ..., 1.5042e+00,
1.9195e+00, 1.5286e+00],
[ 4.1693e-01, 3.0005e+00, 3.3359e+00, ..., 3.5891e-01,
-4.6042e-01, -3.1995e-01]]]], device='cuda:0'),
tensor([[[[ 3.3506e-01, 1.3128e+00, 5.8624e-01, ..., 2.2921e+00,
7.0041e-01, 6.8019e-01],
[-2.7938e-02, -4.9902e-01, 1.1265e+00, ..., 2.5316e+00,
-4.9339e-01, -1.0975e+00],
[-3.0432e-01, -5.7774e-01, 8.3069e-01, ..., -2.1229e-02,
-1.4081e+00, -2.0183e+00],
...,
[-9.7258e-01, -1.9487e+00, -8.3501e-01, ..., -4.5966e+00,
-4.2736e+00, -1.0817e+00],
[-2.4815e+00, -1.7846e+00, 1.4341e-01, ..., -2.9535e+00,
-3.1937e+00, 1.5571e-01],
[-2.3697e+00, -3.9982e+00, -3.1328e+00, ..., -4.7488e+00,
-3.4705e+00, -1.4185e+00]],
[[-1.5900e+00, -1.9710e+00, -2.8187e+00, ..., 1.2871e+00,
-4.7902e+00, -1.1970e+00],
[-5.1121e+00, -6.5457e+00, -4.4897e+00, ..., -3.3050e+00,
-8.7580e+00, -4.3261e+00],
[-4.4007e+00, -5.1059e+00, -3.5671e+00, ..., -9.0015e+00,
-8.4283e+00, -4.2539e+00],
...,
[-7.4549e+00, -9.4026e+00, -7.7640e+00, ..., -1.0899e+01,
-9.6821e+00, -1.8914e+00],
[-1.0026e+01, -1.0177e+01, -8.5822e+00, ..., -1.0862e+01,
-1.0521e+01, -3.6603e+00],
[-2.5744e+00, -4.9050e+00, -4.2887e+00, ..., -4.1800e+00,
-4.1307e+00, -7.3844e-01]],
[[-2.6335e-01, 8.9176e-01, 5.8979e-01, ..., 3.2659e+00,
2.6608e+00, 2.2225e-01],
[ 8.6006e-02, 3.0011e-02, -6.5638e-01, ..., 1.4454e+00,
2.2638e+00, -8.6198e-01],
[ 8.2659e-01, 2.3314e+00, 1.3522e+00, ..., 2.9566e+00,
2.6698e+00, 3.4028e-01],
...,
[ 6.9401e-01, 8.6780e-01, 1.4665e-01, ..., 2.0017e+00,
-7.5850e-01, -1.3953e+00],
[-3.4091e-01, 5.8721e-01, -5.4257e-02, ..., -1.3480e+00,
-6.5686e-01, -5.8251e-01],
[ 6.3856e-01, 1.2081e+00, 9.6844e-01, ..., -7.2429e-01,
-2.3539e-01, -1.6464e+00]],
...,
[[-9.6219e-01, -1.0184e+00, -1.5984e+00, ..., -2.9234e+00,
-2.7481e+00, -1.8098e-02],
[-4.0659e+00, -4.6438e+00, -2.7940e+00, ..., -6.3813e+00,
-5.5272e+00, -3.6904e+00],
[-2.4086e+00, -1.5034e+00, -9.1281e-01, ..., -5.4258e+00,
-4.7316e+00, -2.6819e+00],
...,
[-2.9616e+00, -3.5847e+00, -2.2500e+00, ..., -6.5698e+00,
-5.5622e+00, -3.2817e+00],
[-3.7601e+00, -5.4431e+00, -3.0508e+00, ..., -4.1283e+00,
-5.3263e+00, -1.7005e+00],
[-3.2031e+00, -3.5534e+00, -3.1401e+00, ..., -2.8306e+00,
-2.7586e+00, 3.9283e-01]],
[[-1.0343e+00, 3.1888e-02, -2.3582e+00, ..., -1.7754e+00,
-5.0402e-01, 3.6571e-01],
[-3.9534e+00, -1.1114e+00, -2.3193e+00, ..., -3.8661e+00,
-2.0490e+00, 7.1168e-01],
[-3.7008e+00, -1.5546e+00, -1.9299e+00, ..., -6.2242e+00,
-3.1126e+00, -3.2616e-01],
...,
[-4.4509e+00, -2.6207e+00, -2.9511e+00, ..., -5.3130e+00,
-1.2181e+00, -1.4210e-01],
[-3.7544e+00, -8.4970e-01, -1.5214e+00, ..., -1.7222e+00,
-7.8300e-01, 6.7049e-01],
[-3.8502e+00, -3.0755e+00, -2.8407e+00, ..., -8.6320e-01,
-7.2115e-01, 5.6333e-01]],
[[ 1.6510e+00, 6.3576e-01, 8.6554e-01, ..., 2.5850e+00,
2.6059e+00, 1.3586e-01],
[ 3.1938e+00, 2.1091e+00, 1.9440e+00, ..., 3.0056e+00,
2.6257e+00, 3.1780e+00],
[ 2.1225e+00, 2.5369e-01, 5.1667e-01, ..., 2.6851e+00,
2.5476e+00, 2.5028e+00],
...,
[ 3.0245e+00, 2.9545e+00, 2.1147e+00, ..., 1.9322e+00,
1.8670e+00, 1.8901e-01],
[ 4.8529e+00, 4.8221e+00, 3.6694e+00, ..., 2.8514e+00,
2.3162e+00, 1.4890e+00],
[ 2.7722e+00, 2.7149e+00, 1.9497e+00, ..., 7.0804e-01,
2.8133e-01, 8.7427e-01]]],
[[[ 5.4033e-01, 8.6424e-01, -4.5542e-01, ..., -1.3839e+00,
-1.0678e+00, -9.2967e-01],
[ 1.5234e+00, -1.9270e+00, -3.9457e+00, ..., -1.5391e+00,
2.4867e-01, -1.1476e+00],
[-1.1507e+00, -5.2139e+00, -5.7717e+00, ..., -1.3149e+00,
-2.7119e-01, -1.8907e+00],
...,
[-9.2326e-01, 3.8344e-01, -1.7448e-01, ..., -4.6269e+00,
-3.7461e+00, 6.8955e-01],
[-1.8117e+00, -2.1287e-01, -4.4526e-01, ..., 4.0599e-01,
-9.0702e-01, 7.8774e-01],
[-1.4926e+00, -3.3494e+00, -2.7036e+00, ..., 1.1607e+00,
5.4252e-01, -1.3093e+00]],
[[-8.1291e-01, -1.6765e+00, -5.2719e+00, ..., -2.8073e+00,
-4.8263e+00, -2.6402e+00],
[-3.4073e+00, -6.3907e+00, -6.6380e+00, ..., -4.4821e+00,
-5.1886e+00, -4.9305e+00],
[-5.2390e+00, -6.7943e+00, -7.2425e+00, ..., -5.1160e+00,
-4.2495e+00, -3.8363e+00],
...,
[-5.1868e+00, -6.1110e+00, -5.2207e+00, ..., -5.1700e+00,
-7.4848e+00, -9.2055e-01],
[-7.4202e+00, -8.2214e+00, -6.2470e+00, ..., -1.6014e+00,
-7.6268e+00, -3.5316e+00],
[-1.6311e+00, -3.7997e+00, -2.6633e+00, ..., 4.5218e-01,
-1.4842e+00, -1.0172e+00]],
[[ 6.6299e-01, 5.5619e-01, 5.7510e-01, ..., 1.4273e+00,
1.9148e+00, 8.6460e-01],
[-3.9767e-01, 2.2908e-01, 2.6963e+00, ..., 1.4282e+00,
2.3680e+00, 9.0874e-01],
[ 1.2390e-01, 4.3695e+00, 3.7832e+00, ..., 1.4658e+00,
2.3539e+00, 2.1730e+00],
...,
[-1.1040e-01, 4.4542e-01, -5.2977e-01, ..., 3.1530e+00,
4.5104e-01, -2.2465e+00],
[-4.1396e-01, -1.2970e-01, -7.8434e-01, ..., 3.2956e+00,
2.7097e+00, -7.9890e-01],
[ 2.8735e-01, -5.1983e-02, -7.4753e-01, ..., -3.5990e+00,
-1.9097e-01, -1.4975e+00]],
...,
[[-3.6429e-01, -2.2391e+00, -2.8045e+00, ..., -2.7768e+00,
-3.3297e+00, -1.2931e+00],
[-2.6975e+00, -5.7544e+00, -3.5909e+00, ..., -4.8918e+00,
-4.6099e+00, -3.5710e+00],
[-3.6906e+00, -4.5012e+00, -5.0798e-01, ..., -3.8403e+00,
-4.4150e+00, -2.9343e+00],
...,
[-8.6567e-01, -8.7149e-01, 2.2151e-01, ..., -3.2437e+00,
-2.8287e+00, -1.6247e+00],
[-1.8143e+00, -2.7242e+00, -5.5076e-01, ..., -1.3184e-01,
-2.6733e+00, -1.0040e+00],
[-1.6764e+00, -2.0869e+00, -8.2916e-01, ..., -1.4545e+00,
-1.4132e+00, 9.3596e-01]],
[[-1.6649e+00, -2.3605e+00, -2.5438e+00, ..., -2.9289e+00,
-2.8130e+00, -5.4525e-01],
[-3.1888e+00, -5.1138e+00, -3.2156e+00, ..., -3.9290e+00,
-5.5724e+00, -1.7231e+00],
[-4.4071e+00, -6.8642e+00, -5.9382e+00, ..., -1.5154e+00,
-3.9575e+00, -1.6961e+00],
...,
[-2.0508e+00, -1.3314e+00, -1.6787e+00, ..., -1.7961e-01,
5.4239e-01, -8.7851e-02],
[-2.3398e+00, 6.0334e-01, 1.1255e-01, ..., -1.7640e+00,
-8.9858e-01, 1.2391e+00],
[-3.0130e+00, -1.3351e+00, -5.1186e-01, ..., 2.7373e-01,
-1.4875e+00, 9.0391e-01]],
[[ 1.0607e+00, 4.4934e-01, 1.7139e+00, ..., 2.5779e-01,
1.4444e+00, 6.3077e-01],
[ 3.0430e+00, 4.3209e-01, -3.9803e-01, ..., 3.3527e-01,
2.1624e+00, 3.4677e+00],
[ 2.8298e+00, -2.3447e-01, -1.5237e-01, ..., 2.7693e-01,
8.5814e-01, 1.0665e+00],
...,
[ 2.4660e+00, 1.9571e+00, 7.4252e-01, ..., 2.8151e+00,
1.0638e+00, 5.1836e-01],
[ 3.8410e+00, 4.4829e+00, 3.0688e+00, ..., 2.3096e+00,
2.8473e+00, 2.4504e+00],
[ 2.1437e+00, 1.7308e+00, 5.7687e-01, ..., -1.1348e+00,
7.4384e-01, 9.7931e-01]]],
[[[-1.1058e-01, 7.1836e-01, -2.7793e-01, ..., -1.4235e+00,
-8.7525e-01, -7.7955e-01],
[ 7.1347e-01, 4.1474e-01, -7.5109e-03, ..., -8.9561e-01,
4.7327e-01, -9.7478e-01],
[ 1.0174e+00, -4.8673e-01, 6.9567e-01, ..., -8.0719e-01,
-2.2921e-01, -2.0976e+00],
...,
[-3.6413e-01, -1.1115e+00, 9.5827e-01, ..., -1.5009e+00,
-2.5695e+00, -1.0883e-01],
[-2.3769e+00, -1.8884e+00, 7.1557e-01, ..., -2.4652e+00,
-2.7487e+00, 1.5714e-01],
[-2.3745e+00, -3.9214e+00, -3.0291e+00, ..., -4.1891e+00,
-3.4260e+00, -1.4721e+00]],
[[-7.4796e-01, -1.4696e+00, -2.5812e+00, ..., -2.9173e+00,
-4.3874e+00, -2.3835e+00],
[-1.9124e+00, -3.6383e+00, -3.9781e+00, ..., -3.5187e+00,
-4.6417e+00, -4.4555e+00],
[-1.7618e+00, -3.3476e+00, -3.7285e+00, ..., -2.9191e+00,
-3.9657e+00, -3.6312e+00],
...,
[-6.5489e+00, -8.4762e+00, -7.3274e+00, ..., -7.9854e+00,
-8.6548e+00, -1.9198e+00],
[-9.8006e+00, -1.0088e+01, -8.1967e+00, ..., -1.0699e+01,
-1.0639e+01, -3.7042e+00],
[-2.5166e+00, -4.8889e+00, -4.2327e+00, ..., -4.1291e+00,
-4.1137e+00, -6.0895e-01]],
[[ 6.8703e-01, 1.2066e+00, 1.0292e+00, ..., 1.5182e+00,
1.9134e+00, 8.7861e-01],
[ 8.7681e-01, 1.3756e+00, 1.5601e+00, ..., 1.4366e+00,
2.1198e+00, 9.6488e-01],
[ 5.5059e-01, 3.0602e+00, 2.5269e+00, ..., 1.2263e+00,
2.1419e+00, 2.1000e+00],
...,
[-9.6375e-02, 2.4631e-01, -6.1485e-01, ..., 1.6631e+00,
1.4260e+00, -1.1056e-01],
[-6.0695e-01, 7.9151e-01, -1.1707e-01, ..., 5.0769e-01,
1.0404e+00, -3.2200e-01],
[ 6.1151e-01, 1.1236e+00, 9.9450e-01, ..., -2.3745e-01,
-5.9594e-01, -1.9191e+00]],
...,
[[-1.0136e+00, -1.3073e+00, -2.0624e+00, ..., -2.5212e+00,
-3.1027e+00, -1.1843e+00],
[-2.3706e+00, -3.1779e+00, -2.3930e+00, ..., -3.1821e+00,
-3.9702e+00, -3.5674e+00],
[-1.0815e+00, -1.7242e+00, -5.6382e-01, ..., -2.4659e+00,
-4.1575e+00, -3.1514e+00],
...,
[-2.7164e+00, -4.2206e+00, -9.2441e-01, ..., -5.2008e+00,
-6.1248e+00, -2.6204e+00],
[-3.7812e+00, -6.0176e+00, -2.7990e+00, ..., -4.0967e+00,
-5.1399e+00, -1.3313e+00],
[-3.2380e+00, -3.6681e+00, -3.2556e+00, ..., -2.6330e+00,
-2.5056e+00, 3.7005e-01]],
[[-1.5328e+00, -2.3262e+00, -3.5092e+00, ..., -2.8926e+00,
-2.9025e+00, -5.7064e-01],
[-4.3580e+00, -4.8120e+00, -4.8715e+00, ..., -4.1218e+00,
-5.9341e+00, -2.0405e+00],
[-4.1599e+00, -5.0070e+00, -4.5941e+00, ..., -2.3819e+00,
-4.8699e+00, -2.1999e+00],
...,
[-4.3147e+00, -2.5594e+00, -3.6675e+00, ..., -4.3147e+00,
-2.9839e+00, -1.2673e+00],
[-3.5788e+00, -6.8734e-01, -1.6738e+00, ..., -2.2652e+00,
-1.2243e+00, 2.7885e-01],
[-3.7372e+00, -2.9797e+00, -2.9785e+00, ..., -1.6916e+00,
-1.1121e+00, 3.4702e-01]],
[[ 9.2270e-01, -1.4044e-01, 7.2613e-01, ..., 4.5812e-01,
1.3655e+00, 6.3457e-01],
[ 1.7061e+00, 8.2381e-01, 3.1317e-01, ..., 5.3999e-01,
1.8926e+00, 3.2164e+00],
[ 1.3968e+00, 5.0622e-02, 4.1210e-01, ..., 2.2145e-01,
9.8593e-01, 7.6031e-01],
...,
[ 4.0584e+00, 3.6569e+00, 2.7228e+00, ..., 3.0184e+00,
2.7990e+00, 7.3349e-01],
[ 4.6015e+00, 4.2682e+00, 3.7673e+00, ..., 2.4910e+00,
1.9093e+00, 1.4763e+00],
[ 2.7720e+00, 2.8291e+00, 2.0886e+00, ..., 4.6661e-01,
6.1405e-01, 9.7773e-01]]],
[[[-3.5969e-01, 8.8389e-01, 1.1065e+00, ..., -1.1226e+00,
-3.4523e+00, -9.1584e-01],
[ 8.4915e-02, -1.2365e-01, 2.9195e+00, ..., 5.2553e-01,
-2.2451e+00, -6.9039e-01],
[-1.2127e+00, 1.0643e-01, 2.9590e+00, ..., 7.8316e-01,
-1.9186e+00, -9.2236e-01],
...,
[-3.9576e-01, 3.3864e-01, 2.0115e+00, ..., -2.6387e+00,
-2.1493e+00, 6.8498e-01],
[-1.5745e+00, -2.8725e-01, 1.3001e+00, ..., -2.4940e+00,
-3.2293e+00, -8.7149e-02],
[-1.1964e+00, -2.2997e+00, -1.2335e+00, ..., -3.6355e+00,
-3.6877e+00, -1.5922e+00]],
[[-1.7062e+00, -2.3105e+00, -3.0070e+00, ..., -5.5631e+00,
-8.9050e+00, -1.8727e+00],
[-4.7567e+00, -4.7674e+00, -4.4118e+00, ..., -5.9183e+00,
-8.9930e+00, -3.5046e+00],
[-3.8094e+00, -4.6868e+00, -6.8502e+00, ..., -4.4989e+00,
-6.9121e+00, -2.5828e+00],
...,
[-4.3622e+00, -5.1596e+00, -5.8576e+00, ..., -6.6412e+00,
-8.7481e+00, -2.4949e+00],
[-7.6351e+00, -7.9634e+00, -7.3355e+00, ..., -8.5374e+00,
-1.0063e+01, -3.7975e+00],
[-1.3648e+00, -3.0594e+00, -2.2712e+00, ..., -3.9306e+00,
-4.0976e+00, -5.3980e-01]],
[[-1.6704e-01, 2.1513e+00, 1.3870e+00, ..., 2.0122e+00,
2.3634e+00, 6.9097e-01],
[-4.9017e-01, 9.8014e-02, 5.9831e-02, ..., 1.1794e+00,
2.0265e+00, -1.7526e-01],
[-1.0915e-01, 2.0520e+00, 1.5216e+00, ..., 3.2753e-01,
6.4365e-01, -4.8639e-01],
...,
[-7.2543e-01, 6.7033e-01, 2.9970e-01, ..., 1.4303e+00,
2.2494e-01, -5.8093e-01],
[-1.2454e+00, 5.0001e-01, 1.3056e+00, ..., -1.0538e-01,
6.6408e-01, -3.6404e-01],
[-3.8995e-01, -1.9677e-01, 2.5630e-01, ..., 2.7488e-01,
-1.0822e-01, -1.8944e+00]],
...,
[[-1.4095e+00, -1.9129e+00, -5.1218e-01, ..., -3.8947e+00,
-5.4763e+00, -1.2449e+00],
[-3.4781e+00, -3.3465e+00, -1.3747e+00, ..., -2.5654e+00,
-6.0667e+00, -2.4708e+00],
[-2.6144e+00, -2.2905e+00, -1.6330e+00, ..., -2.4474e+00,
-6.5631e+00, -1.7047e+00],
...,
[-1.2007e+00, -2.5843e+00, -1.8415e-01, ..., -2.7352e+00,
-5.5661e+00, -2.7120e+00],
[-2.2634e+00, -4.2826e+00, -6.3710e-01, ..., -1.8647e+00,
-4.0887e+00, -3.1861e-01],
[-1.3174e+00, -1.4260e+00, 2.2811e-01, ..., -1.5065e+00,
-2.2073e+00, 8.5915e-01]],
[[-1.8267e+00, -1.4880e+00, -1.8764e+00, ..., -2.1393e+00,
-1.0908e+00, 1.6143e-01],
[-3.9975e+00, -1.6361e+00, -2.0219e+00, ..., -2.4642e+00,
-3.2497e+00, -2.2754e-01],
[-3.8472e+00, -2.9891e+00, -4.3650e+00, ..., -6.4425e-01,
-2.3664e+00, 2.4133e-01],
...,
[-2.5388e+00, -2.2205e+00, -3.2768e+00, ..., -1.8850e+00,
-1.0777e+00, 1.5359e-01],
[-1.9217e+00, -7.2683e-02, -1.0660e+00, ..., -4.5027e-01,
-1.1369e+00, 1.0616e+00],
[-2.0955e+00, -1.3967e+00, -4.4751e-01, ..., -5.1129e-01,
-8.6266e-01, 7.5751e-01]],
[[ 1.2076e+00, 3.1546e-01, 5.7243e-01, ..., 1.8532e+00,
1.5387e+00, 1.1679e-01],
[ 4.0008e+00, 2.8296e+00, 2.5286e+00, ..., 3.3644e+00,
2.6716e+00, 3.7490e+00],
[ 1.5397e+00, 4.4484e-01, 1.6418e+00, ..., 2.6667e+00,
1.8616e+00, 1.7788e+00],
...,
[ 2.9214e+00, 2.9008e+00, 2.2791e+00, ..., 2.7157e+00,
2.9210e+00, 1.3634e+00],
[ 3.7665e+00, 3.9448e+00, 3.3187e+00, ..., 2.7161e+00,
1.1780e+00, 2.6217e+00],
[ 2.4252e+00, 2.3243e+00, 1.6093e+00, ..., 3.1969e-01,
-3.6888e-01, 1.3564e+00]]],
[[[-3.7941e-01, 9.4703e-01, 1.4345e+00, ..., -2.9662e+00,
-3.9554e+00, -2.3280e+00],
[ 9.2171e-01, 5.8058e-01, 1.2883e+00, ..., -3.3663e+00,
-1.2421e+00, -2.5069e+00],
[ 1.8638e+00, 1.3319e+00, -3.1890e+00, ..., -6.6417e+00,
-2.4264e+00, -2.1251e+00],
...,
[ 8.8268e-01, -2.7881e+00, -1.0575e+00, ..., 8.2515e+00,
-2.7435e+00, -6.6549e+00],
[-1.7707e+00, -1.0927e+00, -3.5044e-02, ..., 5.1417e+00,
-7.7974e+00, -3.1102e+00],
[-2.3386e+00, -3.9218e+00, -1.6826e+00, ..., -9.3813e+00,
-6.2595e+00, -1.7549e+00]],
[[-8.5565e-01, -2.8547e+00, -5.9878e+00, ..., -1.1073e+01,
-9.6102e+00, -3.6110e+00],
[-2.8940e+00, -4.3733e+00, -9.4754e+00, ..., -4.5886e+00,
-7.4451e+00, -5.0849e+00],
[-3.8497e+00, -6.9621e+00, -1.8675e+01, ..., -6.5878e+00,
-4.4005e+00, -4.9452e+00],
...,
[-6.7201e+00, -7.6937e+00, -8.1779e+00, ..., -5.3260e+00,
-2.0267e+01, -1.4648e+01],
[-9.3465e+00, -1.1075e+01, -5.1327e+00, ..., -6.7516e+00,
-1.1559e+01, -3.1122e+00],
[-2.3086e+00, -4.9284e+00, -4.5674e+00, ..., -8.5200e+00,
-8.1905e+00, -7.6939e-01]],
[[ 5.9417e-01, 2.5131e+00, 2.6903e+00, ..., 2.0505e+00,
2.7170e+00, 8.1680e-01],
[-1.6892e-02, 9.5254e-01, -1.7337e+00, ..., 3.3672e+00,
3.1375e+00, 1.3240e+00],
[ 6.5688e-01, 9.9780e-01, -1.3088e+00, ..., 8.7346e+00,
5.7153e+00, 1.5794e+00],
...,
[-2.3284e+00, -2.3155e+00, 5.1716e-02, ..., -5.9153e-01,
2.2251e+00, -4.0004e+00],
[-1.3657e+00, -8.9990e-01, -8.5457e-01, ..., -5.3871e+00,
6.5495e+00, 4.6874e-01],
[-1.2634e-01, 6.1789e-01, -1.8749e-01, ..., -6.7549e+00,
-6.4615e+00, -6.0416e+00]],
...,
[[-6.5055e-01, -2.1080e+00, -2.5034e+00, ..., -6.4113e+00,
-6.0228e+00, -2.2147e+00],
[-2.0748e+00, -2.4062e+00, -2.9825e+00, ..., -3.4253e+00,
-4.8090e+00, -3.4950e+00],
[-7.1743e-01, -3.7242e+00, -8.4337e+00, ..., -8.0515e+00,
-5.8313e+00, -2.1727e+00],
...,
[-2.4097e+00, -5.6977e+00, -3.1112e+00, ..., -4.7895e+00,
-1.3835e+01, -6.4449e+00],
[-3.2599e+00, -6.4944e+00, -3.7796e-01, ..., -2.2629e+00,
-8.1344e+00, -5.7731e-01],
[-3.1742e+00, -3.7751e+00, -1.2899e+00, ..., -8.0202e+00,
-6.0030e+00, -6.8358e-01]],
[[-2.0057e+00, -1.9471e+00, -3.6816e+00, ..., -4.0245e+00,
-3.0035e+00, 1.3735e-02],
[-3.5644e+00, -2.4919e+00, -3.5374e+00, ..., -2.2167e+00,
-4.1615e+00, -1.3300e+00],
[-4.4840e+00, -5.6497e+00, -2.4620e+00, ..., -7.2682e+00,
-3.3625e+00, 6.1203e-01],
...,
[-3.3258e+00, -4.6688e+00, -3.6466e+00, ..., -4.4374e+00,
-1.0855e+01, -9.3457e-01],
[-1.8088e+00, 1.3030e-02, -2.3179e+00, ..., 3.6743e+00,
-2.7935e+00, 2.5642e+00],
[-2.7055e+00, -1.4338e+00, -1.7490e+00, ..., -2.3482e-01,
-1.1196e+00, 2.5412e+00]],
[[ 7.6315e-01, 4.5177e-02, 2.8662e+00, ..., 2.5303e+00,
1.9977e+00, 2.5636e-01],
[ 2.1697e+00, 1.5366e+00, 3.2913e+00, ..., 8.3995e-01,
1.5196e+00, 2.5364e+00],
[ 1.3586e+00, 2.0122e+00, 5.7007e+00, ..., 2.7913e+00,
4.8376e+00, 3.0107e+00],
...,
[ 3.6348e+00, 2.7171e+00, 6.1091e+00, ..., 6.0625e+00,
1.0696e+01, 6.8027e+00],
[ 4.5472e+00, 3.1195e+00, 2.1562e+00, ..., 1.0452e+01,
2.7773e+00, 4.3158e+00],
[ 2.7178e+00, 2.9213e+00, 2.9391e+00, ..., 2.6497e+00,
1.1995e+00, 3.3052e+00]]]], device='cuda:0'),
tensor([[[[-2.9303e-01, 7.8451e-01, 1.1988e+00, ..., -9.0833e-02,
8.5568e-01, 4.8532e-02],
[ 2.7067e+00, 3.4798e+00, 6.8931e+00, ..., 3.2106e-01,
1.7888e+00, 3.6465e-01],
[-2.1346e-01, -3.0775e+00, -1.1069e+01, ..., 9.4704e-01,
2.9311e+00, -3.3990e-01],
...,
[ 3.5623e+00, 7.8376e-01, -5.8555e+00, ..., 1.6630e+00,
3.5758e+00, 1.6000e-01],
[ 2.6061e+00, -7.6939e+00, -4.4519e+00, ..., 1.8914e+00,
1.7754e+00, -7.4951e-02],
[ 1.0315e+00, -2.3105e+00, 2.6620e+00, ..., 1.6254e+00,
9.0106e-02, 3.8198e-01]],
[[-1.5728e+00, -5.7691e+00, -1.0796e+01, ..., -6.8437e-01,
-2.6173e+00, -1.0764e-01],
[-3.2072e+00, -7.6921e+00, -1.5809e+01, ..., -1.7894e+00,
-4.9772e+00, -6.6925e-01],
[-1.0061e+01, -2.1956e+01, -2.6012e+01, ..., -3.7280e+00,
-1.0293e+01, -3.3798e+00],
...,
[ 2.3069e+00, -6.1133e+00, -3.3663e+00, ..., 2.1637e+00,
4.4713e-01, -5.0555e-01],
[ 3.0309e-01, -1.3408e+01, -1.4089e+00, ..., 2.3058e+00,
-3.1360e+00, -1.2682e+00],
[ 1.8555e+00, -9.6020e-03, 1.7718e+00, ..., 2.5348e+00,
-1.4257e-02, 4.4615e-01]],
[[ 3.0754e-02, 2.0966e+00, 5.5981e-02, ..., 2.0224e+00,
1.6078e+00, 6.3649e-02],
[ 3.2433e-01, 1.6642e+00, 1.3102e+00, ..., 1.9385e+00,
-2.8430e-01, -5.7967e-01],
[-1.9112e+00, 8.2403e+00, 1.4020e+01, ..., 5.8352e+00,
3.5912e-01, -1.2803e+00],
...,
[-5.0398e+00, -1.1500e+01, -7.8832e+00, ..., -7.0446e+00,
-7.9385e+00, -3.5031e+00],
[-5.1252e+00, -9.6137e+00, -8.4845e+00, ..., -4.4233e+00,
-5.3916e+00, -2.3447e+00],
[-6.2316e+00, -6.5274e+00, -7.2269e+00, ..., -5.8221e+00,
-2.5920e+00, -1.4246e+00]],
...,
[[-8.6395e-01, -1.2572e+00, -2.3518e+00, ..., -7.7942e-01,
-5.0766e-01, -1.2136e+00],
[-3.6272e+00, -6.6450e+00, -5.8565e+00, ..., -5.0895e+00,
-5.0273e-01, -4.3162e+00],
[-2.1453e+00, -7.7245e+00, -9.5604e+00, ..., -9.5358e+00,
-7.0705e-01, -6.0100e+00],
...,
[ 2.9786e+00, -3.7212e+00, -1.1124e+00, ..., 4.3438e+00,
-1.8038e+00, 6.2315e-01],
[-8.8686e-01, -6.7990e+00, -9.6869e-01, ..., 2.4001e+00,
-3.1992e+00, 2.7876e-01],
[ 1.6668e+00, -2.7831e-01, 5.2386e-01, ..., 2.5606e+00,
-1.0476e+00, 1.3237e+00]],
[[-2.1658e-01, -3.8197e-01, 1.8485e+00, ..., -6.4953e-01,
6.2254e-01, -1.1441e+00],
[-1.3766e+00, -1.5018e-01, 5.3486e+00, ..., -1.6525e+00,
1.7352e+00, -9.9477e-01],
[-6.2408e+00, -1.2865e+01, -1.4228e+01, ..., -3.9093e+00,
1.3012e+00, -1.4130e+00],
...,
[-2.7881e+00, -8.4735e+00, -5.8843e+00, ..., 1.7190e+00,
-1.6569e+00, -2.0324e+00],
[-7.5592e+00, -9.9368e+00, -6.0259e+00, ..., 9.5636e-01,
-1.7284e+00, -1.5431e+00],
[-1.2313e+00, -2.3984e+00, 1.1585e+00, ..., 1.6544e+00,
-2.3923e-02, -4.5702e-02]],
[[ 1.8848e+00, 1.3677e+00, 1.9484e+00, ..., 2.1699e+00,
1.5756e+00, -1.6109e+00],
[ 2.8402e+00, 5.6375e+00, 7.4604e+00, ..., 3.0994e+00,
1.8414e+00, -1.9456e+00],
[-1.6951e+00, 2.2595e+00, 1.2970e+00, ..., 5.9986e+00,
3.2893e+00, -1.4669e+00],
...,
[ 2.3938e+00, -1.3609e+00, -2.8543e+00, ..., 4.7900e+00,
-2.6282e+00, -2.8368e+00],
[ 1.1591e+00, -1.6025e+00, -2.5913e+00, ..., 4.2114e+00,
-2.3706e+00, -1.4644e+00],
[ 4.7799e-01, -2.3189e+00, 2.7769e+00, ..., 2.3048e+00,
-2.1609e+00, -6.8835e-01]]],
[[[-4.0728e-01, -3.2431e+00, -5.1313e+00, ..., -1.9683e+00,
-4.4368e+00, -2.5271e+00],
[ 1.1348e+00, -2.0856e-02, 3.2085e+00, ..., 1.4756e+00,
-2.3064e+00, -2.5000e+00],
[-2.0234e+00, -3.9249e+00, -3.8534e-01, ..., 1.5813e+00,
-1.0453e+00, -3.0209e+00],
...,
[-2.3946e-01, -8.8175e-01, 2.1036e+00, ..., -2.4456e+00,
2.8330e+00, 2.9934e+00],
[-9.7612e-01, -2.6393e+00, -2.2279e+00, ..., 1.2611e+00,
2.7662e+00, 1.2363e+00],
[-1.8346e+00, -3.7620e+00, -3.5269e+00, ..., -4.0324e+00,
-2.6667e+00, -1.7855e+00]],
[[-1.6459e+00, -8.7050e+00, -5.6060e+00, ..., -4.1414e+00,
-9.0456e+00, -2.9270e+00],
[-2.8050e-02, -4.4329e-01, -2.5172e+00, ..., -3.0434e+00,
-1.1428e+01, -4.1794e+00],
[-2.0889e+00, -3.6830e+00, -2.8412e+00, ..., -1.7613e+00,
-7.1789e+00, -6.4339e+00],
...,
[-6.9686e-01, -1.6421e+00, -5.2519e+00, ..., -3.0840e+00,
-2.5335e-01, 1.6895e+00],
[-8.2013e+00, -1.0912e+01, -1.1420e+01, ..., -1.8041e+00,
-5.9804e+00, -5.9005e+00],
[-1.2752e+00, -3.3256e+00, -2.1488e+00, ..., -4.2618e+00,
-3.1883e+00, -9.8996e-01]],
[[ 1.0740e+00, 4.2501e+00, 5.8917e+00, ..., 3.6064e+00,
3.3575e+00, 7.0335e-01],
[-3.2154e+00, -7.9210e+00, 4.3292e-01, ..., 4.9763e+00,
4.3672e+00, 5.2784e-01],
[ 8.7173e-01, -2.2924e+00, -9.5203e+00, ..., 5.9965e+00,
3.2116e+00, 6.5394e-01],
...,
[-1.2423e+01, -1.2494e+01, -1.5979e+01, ..., -1.1610e+00,
-8.7937e-01, 3.0215e-01],
[-3.2837e+00, -4.9822e+00, -8.7444e+00, ..., -7.8420e-01,
-1.7285e+00, 5.6251e-01],
[-6.9839e-01, 4.7556e-01, -3.1536e-01, ..., -1.7401e+00,
-3.7889e-01, -1.1455e-01]],
...,
[[-2.2340e+00, -4.2366e+00, -6.5211e+00, ..., -4.7836e+00,
-5.8067e+00, -3.0646e+00],
[-1.1920e+00, 1.0290e+00, -1.9350e+00, ..., -3.7778e+00,
-2.9459e+00, -5.6267e+00],
[-1.6978e+00, -8.5745e-01, 5.5498e-01, ..., -6.8764e+00,
-2.8480e+00, -4.5045e+00],
...,
[-1.4174e+00, -4.0930e+00, -2.3326e+00, ..., -7.0248e+00,
-1.9190e+00, -4.4549e+00],
[-3.5491e+00, -9.5079e+00, -6.8629e+00, ..., -1.7609e+00,
-2.8332e+00, -3.0906e+00],
[-1.6304e+00, -3.0264e+00, -1.8473e+00, ..., -3.6172e+00,
-2.4807e+00, 5.0982e-01]],
[[-4.0680e+00, -8.9351e+00, -8.3348e+00, ..., -2.5131e+00,
-2.4165e+00, 6.7645e-01],
[-5.4929e-01, -5.9168e-01, -9.4576e+00, ..., -3.7079e+00,
-1.4647e+00, -3.7702e-01],
[-3.7440e+00, -1.4854e+00, -1.6379e+00, ..., -3.6494e+00,
1.9781e+00, 1.2846e+00],
...,
[ 1.3268e+00, -1.3741e+00, 3.3134e+00, ..., 3.7019e+00,
5.4541e+00, 6.0950e-01],
[-3.4348e-01, 3.4692e-01, 4.3119e+00, ..., 2.5708e+00,
3.1893e+00, 4.3235e-01],
[-2.9877e-01, 6.2673e-01, 2.0353e+00, ..., 8.4640e-01,
5.7823e-01, 6.1939e-01]],
[[ 3.5605e-01, 3.8726e-01, 3.1268e+00, ..., 3.5233e+00,
2.3224e+00, 7.7409e-02],
[-4.5075e-01, -2.5496e+00, 1.1362e+00, ..., 5.1834e+00,
3.3829e+00, 3.2441e+00],
[-6.5575e+00, -6.9168e+00, -4.3726e+00, ..., 3.5039e+00,
3.1137e+00, 1.8775e+00],
...,
[ 9.4403e-01, 5.5704e+00, 1.2464e+01, ..., -5.0778e+00,
2.1580e+00, 1.3924e+00],
[ 6.2868e+00, 4.6832e+00, 3.8150e+00, ..., -1.7659e+00,
2.8712e+00, 2.7889e+00],
[ 1.5425e+00, -1.4390e+00, -1.4086e+00, ..., 6.8311e-01,
1.4543e+00, 1.1806e+00]]],
[[[ 1.5117e-01, 1.2185e+00, 4.1583e-01, ..., -3.3511e+00,
-2.5739e+00, -1.0993e+00],
[ 9.6233e-01, 1.9670e+00, 2.7463e+00, ..., -1.8224e+00,
-2.2221e+00, -2.7096e-01],
[ 1.1263e+00, 5.0871e-01, -3.6680e+00, ..., -4.1103e-01,
-2.2990e+00, -3.3558e-01],
...,
[-5.5456e-01, -1.9491e+00, -2.2542e-01, ..., -2.1850e+00,
1.8134e+00, 2.1312e+00],
[-1.9840e+00, -1.6732e+00, -1.1170e+00, ..., 1.8171e+00,
9.8672e-01, 8.9640e-01],
[-1.1793e+00, -3.4152e+00, -3.3316e+00, ..., -3.3280e+00,
-3.0612e+00, -1.7628e+00]],
[[-8.7293e-01, -1.5032e+00, -3.4452e+00, ..., -6.5644e+00,
-7.3684e+00, -1.4176e+00],
[-3.5200e+00, -4.9668e+00, -5.2552e+00, ..., -6.3915e+00,
-8.5756e+00, -2.5323e+00],
[-4.5134e+00, -7.8803e+00, -1.0260e+01, ..., -4.2300e+00,
-8.4514e+00, -2.9970e+00],
...,
[-6.0921e+00, -6.2821e+00, -4.8908e+00, ..., -1.2251e+00,
-1.7070e+00, 5.0672e-01],
[-7.7931e+00, -7.9232e+00, -7.4608e+00, ..., -4.0705e+00,
-9.1806e+00, -5.9523e+00],
[-1.8120e+00, -3.4789e+00, -3.8366e+00, ..., -5.3649e+00,
-4.0499e+00, -8.2997e-01]],
[[-1.0042e-01, 1.3310e+00, 1.8939e+00, ..., 1.6275e+00,
2.4202e+00, 9.4765e-01],
[ 4.7979e-01, 2.6548e-01, 3.6057e-01, ..., 1.6381e+00,
2.9865e+00, -2.7957e-02],
[ 8.9653e-01, 1.5142e-01, 1.7244e+00, ..., 4.6000e-01,
2.1284e+00, -3.4348e-01],
...,
[ 4.8907e-01, 1.0204e+00, 1.7712e+00, ..., -1.6367e-01,
6.3220e-01, -1.0455e-01],
[ 4.5469e-01, 5.4744e-01, 1.0759e+00, ..., -2.1474e-01,
-1.3981e-01, -3.4129e-01],
[ 5.5458e-01, 1.2300e+00, 1.5846e+00, ..., -8.2036e-01,
-7.6059e-01, -1.1268e+00]],
...,
[[-4.7167e-01, -1.3807e+00, -2.2483e+00, ..., -5.2262e+00,
-5.2453e+00, -7.2748e-01],
[-2.1840e+00, -2.3744e+00, -2.5310e+00, ..., -3.8939e+00,
-5.0013e+00, -1.9543e+00],
[-1.5350e+00, -2.7173e+00, -6.3448e+00, ..., -3.9477e+00,
-5.2614e+00, -1.7850e+00],
...,
[-2.0943e+00, -1.8491e+00, -1.5405e+00, ..., -3.6151e+00,
-3.5551e+00, -2.7597e+00],
[-3.3289e+00, -3.2576e+00, -2.8173e+00, ..., -1.8369e+00,
-3.9505e+00, -2.8725e+00],
[-1.6264e+00, -2.1797e+00, -2.4563e+00, ..., -4.4896e+00,
-2.9009e+00, 5.3967e-03]],
[[-1.1129e+00, -2.1797e+00, -3.6197e+00, ..., -1.6295e+00,
-4.6687e-01, -6.1876e-01],
[-3.6312e+00, -3.5703e+00, -4.2556e+00, ..., -3.2956e+00,
-1.7959e+00, -4.1921e-01],
[-4.7355e+00, -5.3363e+00, -6.6181e+00, ..., -1.1670e+00,
-1.3933e+00, -4.6563e-01],
...,
[-4.9505e+00, -4.1519e+00, -3.6557e+00, ..., -2.6253e+00,
-7.1099e-01, 1.4093e-01],
[-4.2148e+00, -2.1105e+00, -2.8852e+00, ..., -1.4755e+00,
-1.4117e+00, 8.3895e-01],
[-3.3144e+00, -3.3450e+00, -2.6100e+00, ..., -6.6778e-01,
-1.0568e+00, 8.3693e-01]],
[[ 1.0799e+00, -6.0142e-02, 1.1383e+00, ..., 6.2983e-01,
8.2552e-01, -1.3965e+00],
[ 2.0016e+00, 1.7658e+00, 2.4660e+00, ..., 3.6201e+00,
3.3518e+00, 2.3115e+00],
[ 1.5721e+00, 1.9809e+00, -5.9287e-01, ..., 3.0969e+00,
1.9741e+00, 1.0944e+00],
...,
[ 2.5825e+00, 1.4752e+00, 1.7050e+00, ..., -4.8431e+00,
1.1210e+00, 1.3604e+00],
[ 3.0451e+00, 4.0603e+00, 2.5071e+00, ..., 6.4261e-01,
2.0477e+00, 2.7995e+00],
[ 1.9957e+00, 1.6745e+00, 5.3572e-01, ..., 4.0784e-01,
-1.0641e+00, 7.1706e-01]]],
[[[ 2.5233e-01, 5.7324e-01, -1.7690e+00, ..., 6.8774e-01,
-1.6080e-01, -1.5761e+00],
[ 1.1828e+00, -6.2796e-01, -2.4155e+00, ..., 2.0410e+00,
3.1580e+00, -2.3563e+00],
[ 1.2174e+00, -2.8661e+00, -1.9661e+00, ..., 2.6482e+00,
1.4685e-01, -3.6240e+00],
...,
[-1.3837e+00, -9.5791e-01, -1.1446e+00, ..., -1.0118e+00,
-4.9721e+00, 9.0264e-01],
[-2.0766e+00, -6.5521e-01, -7.3411e-02, ..., -1.0003e+00,
-3.2103e+00, 1.0383e+00],
[-2.0264e+00, -4.0589e+00, -2.9287e+00, ..., -2.6552e+00,
-3.3221e+00, -1.3855e+00]],
[[-5.2918e-01, -1.5925e+00, -3.7663e+00, ..., -1.3720e+00,
-5.4366e+00, -3.3480e+00],
[-2.0836e+00, -4.5532e+00, -3.2433e+00, ..., 1.4861e+00,
-4.1361e+00, -7.1739e+00],
[-2.0812e+00, -4.8707e+00, -1.4635e+00, ..., 5.9350e-01,
-4.0014e+00, -6.6243e+00],
...,
[-7.0361e+00, -7.5019e+00, -6.4161e+00, ..., -5.4953e+00,
-7.7218e+00, -8.9933e-01],
[-8.8899e+00, -9.1554e+00, -7.1923e+00, ..., -5.9444e+00,
-8.6933e+00, -3.1878e+00],
[-2.2491e+00, -4.4563e+00, -3.5016e+00, ..., -2.7605e+00,
-4.0321e+00, -7.9846e-01]],
[[ 5.5830e-01, 1.1858e+00, 2.1860e+00, ..., 3.5262e+00,
1.9516e+00, 1.6314e-01],
[ 4.3185e-01, 1.0448e+00, 2.8160e+00, ..., 1.7246e+00,
3.1323e+00, 1.0388e+00],
[ 2.5394e-01, 3.1737e+00, 2.2646e+00, ..., 2.2573e-01,
4.0969e+00, 2.6230e+00],
...,
[ 4.6693e-01, 8.9751e-01, 1.3074e-02, ..., 1.7405e+00,
4.4904e-01, -1.9942e+00],
[ 2.6896e-01, 1.6498e-01, -6.1831e-01, ..., -5.1021e-01,
-5.0048e-01, -1.5199e+00],
[ 6.0870e-01, 5.5971e-01, -3.2475e-01, ..., 7.7433e-01,
-1.5997e-01, -1.8543e+00]],
...,
[[-6.5164e-01, -2.1037e+00, -3.1613e+00, ..., -2.3429e+00,
-3.4827e+00, -1.7467e+00],
[-2.0325e+00, -4.1941e+00, -2.4325e+00, ..., -1.3422e+00,
-4.0171e+00, -4.4545e+00],
[-1.0228e+00, -3.3323e+00, -1.4722e-01, ..., -3.0975e+00,
-6.8858e+00, -3.6236e+00],
...,
[-2.1399e+00, -1.5423e+00, -2.2745e+00, ..., -1.1833e+00,
-4.0116e+00, -8.5063e-01],
[-2.7601e+00, -2.7627e+00, -8.4042e-01, ..., -2.1445e+00,
-2.9785e+00, -3.9683e-01],
[-2.4419e+00, -2.5435e+00, -1.4801e+00, ..., -2.0688e+00,
-1.9546e+00, 4.3632e-01]],
[[-1.4786e+00, -3.1956e+00, -4.9730e+00, ..., -4.1229e+00,
-2.5861e+00, -5.4291e-01],
[-4.4719e+00, -5.4282e+00, -5.0911e+00, ..., -3.6308e+00,
-5.3392e+00, -1.1702e+00],
[-4.5912e+00, -5.3244e+00, -2.9555e+00, ..., -3.9286e+00,
-6.5177e+00, -8.8767e-01],
...,
[-3.3962e+00, -2.1901e+00, -3.6564e+00, ..., -2.5862e-02,
-1.1182e+00, -2.1767e-01],
[-3.3447e+00, 9.6752e-01, -1.1991e+00, ..., 3.5548e-01,
6.9165e-01, 1.5882e+00],
[-3.9714e+00, -1.3863e+00, -1.2107e+00, ..., -3.0537e-01,
1.6383e-01, 5.8632e-01]],
[[ 8.0039e-01, -4.0578e-02, 5.5590e-01, ..., 1.5462e+00,
2.3463e+00, 9.4199e-01],
[ 1.6901e+00, 1.0150e+00, -7.4874e-01, ..., -3.2502e-01,
3.8793e+00, 3.1781e+00],
[ 1.6669e+00, -1.2812e-02, -8.1964e-01, ..., 9.6892e-01,
2.4035e+00, 8.4488e-02],
...,
[ 3.1189e+00, 2.3496e+00, 2.2369e+00, ..., 2.0919e+00,
1.1601e+00, 9.3033e-01],
[ 4.6779e+00, 4.8298e+00, 4.0410e+00, ..., 2.4249e+00,
5.4692e-01, 1.9440e+00],
[ 2.4685e+00, 2.1253e+00, 7.3477e-01, ..., -1.4248e-02,
-1.6403e-01, 1.0194e+00]]],
[[[ 1.2485e+00, 2.4583e+00, 1.1395e+00, ..., -1.1216e-01,
-2.3745e+00, -1.3538e+00],
[-1.6357e+00, -2.2910e+00, -1.2127e+00, ..., -1.8858e+00,
3.3682e+00, 2.8469e+00],
[ 2.7935e+00, 1.6434e+00, -8.6974e-01, ..., -5.8525e+00,
-3.9623e-01, 2.9082e+00],
...,
[-4.4531e-01, -1.4014e+00, -1.6656e+00, ..., 2.4590e+00,
-2.6476e+00, -6.1859e-01],
[-9.6642e-02, 6.0078e-03, -1.8003e-01, ..., -5.3889e+00,
-3.6598e+00, -2.9099e-02],
[-5.2328e-01, -3.2878e+00, -2.9881e+00, ..., -4.2447e+00,
-2.6311e+00, -1.5733e+00]],
[[ 1.1922e+00, -1.3513e+00, -6.8145e+00, ..., -2.5009e+00,
-5.0422e+00, -1.3957e+00],
[-7.8927e+00, -5.1064e+00, -3.2187e+00, ..., 1.0705e-01,
2.3507e+00, 2.5810e-01],
[ 3.6445e-01, -3.8579e+00, -3.3678e+00, ..., -3.7662e+00,
-7.2326e-01, 7.0291e-01],
...,
[-3.1767e+00, -4.9506e+00, -4.2696e+00, ..., -3.8669e+00,
-9.4983e+00, -2.9436e+00],
[-4.8347e+00, -6.1631e+00, -4.7381e+00, ..., -1.4380e+01,
-1.0541e+01, -4.3187e+00],
[-5.3983e-01, -3.2205e+00, -3.4012e+00, ..., -4.1226e+00,
-3.2493e+00, -6.7043e-01]],
[[ 1.1736e+00, 2.0892e+00, 1.7929e+00, ..., 6.7444e-01,
4.5257e+00, 2.9896e+00],
[ 1.9252e+00, 7.2130e+00, 5.0966e+00, ..., -8.4011e+00,
-1.8759e+00, 3.3455e+00],
[-2.0562e+00, 3.2983e+00, 8.0099e+00, ..., -4.5569e-01,
-3.8151e+00, 1.3119e+00],
...,
[ 1.6788e+00, 2.3029e+00, 8.9980e-01, ..., 2.0966e+00,
3.1171e+00, -8.1695e-01],
[ 7.3751e-01, 2.0755e+00, 6.1011e-01, ..., -2.2914e-01,
1.9075e+00, -2.2142e+00],
[-8.3635e-02, 1.6894e+00, 6.5593e-01, ..., -1.6217e+00,
-2.2621e+00, -2.0870e+00]],
...,
[[-6.6441e-01, -5.0692e-01, -2.1478e+00, ..., -1.3351e+00,
-3.0383e+00, -2.0607e+00],
[-5.2194e+00, -5.8593e+00, -3.3571e+00, ..., -1.1023e+00,
-1.3376e+00, -1.5169e+00],
[ 5.3735e-01, -2.4390e+00, -2.6959e+00, ..., -3.3530e+00,
-3.5556e+00, -5.2004e+00],
...,
[-2.8277e+00, -4.0031e+00, -2.2756e+00, ..., -5.8402e+00,
-7.1133e+00, -2.0811e+00],
[-2.1825e+00, -3.6253e+00, -1.5809e+00, ..., -5.9773e+00,
-5.4564e+00, -8.2454e-01],
[-1.3239e+00, -3.4672e+00, -2.5123e+00, ..., -1.0953e+00,
-3.3218e+00, 3.2438e-02]],
[[-1.3906e+00, -9.5443e-01, -1.3100e+00, ..., -2.2858e+00,
-5.5442e+00, 1.2357e+00],
[-7.3565e+00, -8.2436e+00, -1.9690e+00, ..., 2.4626e+00,
-3.0674e+00, -1.0176e+00],
[-1.4922e+00, -9.5376e+00, -1.2243e+01, ..., 4.7437e-01,
-2.5761e+00, -1.5063e+00],
...,
[-2.5614e+00, -2.8971e+00, -1.7210e+00, ..., -4.8811e+00,
-2.9903e+00, 7.4381e-02],
[-3.1369e+00, -2.1575e+00, -4.4893e-01, ..., -5.0417e+00,
-9.6696e-01, 1.7466e+00],
[-3.3488e+00, -3.4019e+00, -1.1534e+00, ..., 4.0943e-01,
-5.0539e-01, 1.0438e+00]],
[[ 1.0578e+00, 2.2838e+00, 2.5747e+00, ..., -2.2716e-01,
2.6546e+00, 2.0351e+00],
[ 9.5073e-01, 5.2592e+00, 3.8977e+00, ..., -3.8489e+00,
-8.9081e-01, 5.6440e+00],
[-2.2558e-01, -8.1898e-01, 6.2543e+00, ..., -8.3328e+00,
-3.3784e+00, 1.8771e+00],
...,
[ 1.5190e+00, 1.4669e+00, 2.3789e+00, ..., 3.2488e+00,
2.9630e+00, 7.1349e-01],
[ 3.2486e+00, 3.7655e+00, 2.5489e+00, ..., 3.6768e+00,
2.3983e+00, 2.4852e+00],
[ 1.9948e+00, 8.7720e-01, 7.4096e-01, ..., -1.8069e-01,
1.3013e+00, 1.4899e+00]]]], device='cuda:0'),
tensor([[[[ 7.7477e-03, 1.4262e+00, 7.5075e-01, ..., -8.5966e-03,
5.3365e-01, -1.3223e-01],
[ 3.3458e-01, 5.6295e-01, 1.8164e+00, ..., 5.3057e-02,
4.9686e-01, -4.9365e-01],
[-7.9115e-01, 1.7743e-01, 1.2074e+00, ..., -6.1181e-01,
-1.3179e-01, -9.4729e-01],
...,
[-1.2627e-01, -3.9796e-01, 3.1781e-01, ..., 5.5916e-01,
5.6734e-01, -3.7393e-01],
[-1.6554e+00, -5.8228e-01, 1.6353e-01, ..., 2.7633e-01,
2.2816e-01, -1.1555e+00],
[-1.8807e+00, -3.2112e+00, -2.0412e+00, ..., -3.0934e+00,
-2.8862e+00, -2.7298e+00]],
[[-6.8309e-01, -1.7049e+00, -3.4433e+00, ..., -2.1340e+00,
-3.4537e+00, -2.1681e+00],
[-3.9517e+00, -4.2858e+00, -3.4084e+00, ..., -3.4430e+00,
-4.2752e+00, -3.5677e+00],
[-2.9090e+00, -3.7784e+00, -3.8578e+00, ..., -2.2240e+00,
-3.0371e+00, -2.8486e+00],
...,
[-5.2452e+00, -5.9987e+00, -5.2754e+00, ..., -2.7158e+00,
-3.1573e+00, -2.6089e+00],
[-8.5931e+00, -8.6161e+00, -6.5510e+00, ..., -4.7112e+00,
-5.2691e+00, -5.0077e+00],
[-2.1390e+00, -4.0657e+00, -2.4425e+00, ..., -2.9208e+00,
-2.8260e+00, -1.6060e+00]],
[[-2.0060e-01, 2.4763e+00, 1.6363e+00, ..., 1.5208e+00,
1.6481e+00, 8.8624e-01],
[-5.9084e-01, 2.8727e-01, 6.7953e-01, ..., 8.5084e-01,
1.4764e+00, 7.9745e-01],
[-1.5692e-01, 1.6087e+00, 1.6861e+00, ..., 1.7568e+00,
2.0216e+00, 1.4376e+00],
...,
[-6.8035e-01, 6.1800e-01, 1.5438e+00, ..., 1.6745e+00,
1.0552e+00, 6.1268e-01],
[-1.2346e+00, -3.2907e-01, 8.9159e-01, ..., 1.3148e+00,
9.2310e-01, 3.6285e-01],
[ 1.6175e-01, -1.6501e-02, 4.0571e-01, ..., 1.7071e+00,
1.6589e+00, 5.1474e-01]],
...,
[[-8.9189e-01, -1.8206e+00, -1.8807e+00, ..., -1.4590e+00,
-1.9874e+00, -4.3925e-01],
[-2.8944e+00, -3.5010e+00, -1.9395e+00, ..., -2.5521e+00,
-3.9105e+00, -3.0275e+00],
[-1.9033e+00, -2.2757e+00, -1.2118e+00, ..., -1.5457e+00,
-2.8199e+00, -1.9157e+00],
...,
[-1.3157e+00, -2.3167e+00, -1.0664e+00, ..., -1.3566e+00,
-2.0274e+00, -8.6739e-01],
[-2.6926e+00, -4.3149e+00, -1.7450e+00, ..., -1.3910e+00,
-2.7911e+00, -1.5227e+00],
[-2.5074e+00, -2.1281e+00, -1.0660e+00, ..., -1.8954e+00,
-2.1979e+00, -5.7452e-01]],
[[-1.4736e+00, -2.0715e+00, -2.6035e+00, ..., -6.6700e-01,
-1.4185e+00, -7.0105e-01],
[-3.4237e+00, -2.4842e+00, -2.3314e+00, ..., -1.5235e+00,
-3.6713e+00, -1.8239e+00],
[-3.0285e+00, -2.8642e+00, -3.7437e+00, ..., -7.0043e-01,
-3.8870e+00, -2.4830e+00],
...,
[-3.0574e+00, -3.2182e+00, -3.9973e+00, ..., -9.8674e-01,
-3.3341e+00, -2.2065e+00],
[-2.3999e+00, 1.9435e-02, -1.7460e+00, ..., -1.0802e+00,
-3.3150e+00, -1.9027e+00],
[-2.8862e+00, -1.7416e+00, -1.8025e+00, ..., -1.5729e+00,
-2.7372e+00, -1.2080e+00]],
[[ 8.2279e-01, 2.3167e-01, 6.8334e-01, ..., 1.0813e-01,
1.1936e+00, 6.3740e-01],
[ 3.5119e+00, 2.8501e+00, 2.1756e+00, ..., 5.6322e-01,
9.2802e-01, 2.4888e+00],
[ 1.3365e+00, 5.9957e-01, 3.8199e-01, ..., 4.4383e-01,
3.0585e-01, 8.1513e-01],
...,
[ 3.3543e+00, 2.5825e+00, 2.3826e+00, ..., 4.6557e-01,
2.3965e-01, 5.7331e-01],
[ 4.4475e+00, 4.4182e+00, 2.8733e+00, ..., 1.2803e+00,
8.6847e-01, 1.2688e+00],
[ 2.7051e+00, 2.5743e+00, 1.8189e+00, ..., -6.8941e-02,
-2.9889e-01, -8.5715e-02]]],
[[[ 1.5647e-01, 1.4876e+00, 1.4495e+00, ..., -6.0723e-01,
3.4074e-02, 7.0642e-01],
[ 6.4757e-01, 9.1489e-01, 9.4586e-01, ..., -1.2251e+00,
-1.0377e+00, 1.0377e+00],
[-6.0691e-01, 6.4911e-01, -1.3694e+00, ..., -5.8902e-01,
-9.3108e-01, 7.0757e-01],
...,
[-5.3180e-01, -2.1504e+00, -7.8164e-01, ..., 3.7721e-02,
-2.4107e-01, -1.1091e+00],
[-7.8494e-01, -2.1975e+00, -3.7581e+00, ..., -6.2080e-01,
-2.1401e-01, -1.6003e+00],
[-2.7453e+00, -5.4421e+00, -5.0040e+00, ..., -4.5747e+00,
-4.0952e+00, -3.8489e+00]],
[[-7.9711e-01, -2.2431e+00, -4.3278e+00, ..., -3.2803e+00,
-4.8463e+00, -3.4741e-01],
[-4.6316e+00, -6.1195e+00, -7.6030e+00, ..., -5.3616e+00,
-7.1988e+00, -1.6785e+00],
[-3.8552e+00, -4.7339e+00, -1.0362e+01, ..., -2.6898e+00,
-6.0123e+00, -1.5640e+00],
...,
[-5.7844e+00, -6.2449e+00, -5.9361e+00, ..., -4.7107e+00,
-5.3517e+00, -3.9530e+00],
[-7.4629e+00, -8.1010e+00, -6.4907e+00, ..., -7.5691e+00,
-8.1874e+00, -6.6927e+00],
[-2.7350e+00, -4.1148e+00, -3.7387e+00, ..., -3.8086e+00,
-4.0428e+00, -2.5732e+00]],
[[-4.0244e-01, 2.2461e+00, 1.4491e+00, ..., 8.4939e-01,
2.0704e+00, 9.3066e-01],
[-2.7151e-01, 1.3760e+00, 2.6715e-01, ..., 6.7604e-01,
1.2093e+00, -5.5872e-01],
[ 4.9331e-02, 1.8461e+00, 9.1000e-01, ..., 4.3386e-01,
1.6628e+00, -7.1588e-02],
...,
[ 1.0936e+00, 9.4204e-01, 8.7265e-01, ..., 8.5331e-01,
4.8536e-01, 4.7113e-01],
[ 1.0155e+00, 8.1604e-01, 1.1751e+00, ..., 3.9101e-01,
4.2078e-01, 3.2663e-01],
[ 1.4697e+00, 1.8334e+00, 1.6860e+00, ..., 1.5629e+00,
1.1950e+00, 1.2339e-01]],
...,
[[-7.5985e-01, -1.4405e+00, -1.1150e+00, ..., -2.4336e+00,
-2.7837e+00, -3.9855e-01],
[-3.0091e+00, -3.8279e+00, -2.9722e+00, ..., -3.7782e+00,
-4.3656e+00, -2.4835e+00],
[-1.4649e+00, -2.7166e+00, -3.2745e+00, ..., -1.3399e+00,
-4.0278e+00, -1.2469e+00],
...,
[-2.9170e+00, -2.7570e+00, -1.7548e+00, ..., -1.2580e+00,
-3.5687e+00, -1.8590e+00],
[-3.2476e+00, -3.7994e+00, -2.0560e+00, ..., -2.1681e+00,
-4.4437e+00, -2.3079e+00],
[-3.4874e+00, -3.9618e+00, -2.5211e+00, ..., -2.9013e+00,
-3.6312e+00, -1.5998e+00]],
[[-1.6790e+00, -2.1705e+00, -2.9298e+00, ..., -1.4666e+00,
-1.4529e+00, -1.4179e+00],
[-3.8920e+00, -3.8936e+00, -4.7384e+00, ..., -3.3541e+00,
-2.1930e+00, -1.1393e+00],
[-3.2313e+00, -3.2794e+00, -5.0105e+00, ..., -3.1728e-01,
-1.7933e+00, -9.7520e-01],
...,
[-3.8178e+00, -2.8425e+00, -4.2154e+00, ..., -2.0517e+00,
-3.3131e+00, -1.3345e+00],
[-3.9678e+00, -1.6740e+00, -1.0947e+00, ..., -1.0956e+00,
-3.0650e+00, -1.0994e+00],
[-4.4668e+00, -3.3524e+00, -1.1082e+00, ..., -2.3588e+00,
-3.5168e+00, -1.2234e+00]],
[[ 1.0254e+00, 6.3942e-01, 8.4208e-01, ..., -5.7188e-01,
1.5055e-01, -4.3152e-01],
[ 3.5516e+00, 2.7810e+00, 3.1964e+00, ..., 1.5153e+00,
2.0015e+00, 2.3428e+00],
[ 1.3538e+00, 1.0064e+00, 3.3466e+00, ..., 4.8542e-01,
1.4481e+00, 1.4618e+00],
...,
[ 2.9064e+00, 1.3697e+00, 1.7226e+00, ..., 1.9011e+00,
2.7208e+00, 1.6076e+00],
[ 3.9582e+00, 3.5848e+00, 4.8823e-01, ..., 2.8076e+00,
2.3474e+00, 2.1161e+00],
[ 1.9209e+00, 3.3196e-01, -7.6814e-01, ..., -3.2022e-01,
-3.3170e-01, -1.6617e-01]]],
[[[ 9.0422e-01, 1.1983e+00, 5.2748e-01, ..., -1.2660e+00,
-1.8895e+00, -1.2424e+00],
[ 8.7897e-01, 1.2982e+00, -1.4464e+00, ..., -6.7957e+00,
3.8800e-01, -7.0291e-01],
[ 3.6536e+00, 3.7799e+00, -7.6479e+00, ..., -2.7546e+00,
-7.9413e+00, -5.5476e+00],
...,
[ 5.1960e-02, -1.6104e+00, -1.2276e+01, ..., 2.9738e-01,
1.7670e+00, 1.4836e+00],
[-2.5328e+00, -2.3230e+00, -6.2308e+00, ..., -2.8003e+00,
-4.6800e+00, -1.3660e+00],
[-1.4875e+00, -2.8868e+00, -2.7525e+00, ..., -4.4104e+00,
-4.1270e+00, -2.2452e+00]],
[[-9.0675e-01, -2.4228e+00, -6.8229e+00, ..., -1.3703e+01,
-1.1526e+01, -3.9788e+00],
[-4.4421e+00, -5.8961e+00, -1.6962e+01, ..., -7.6895e+00,
-5.9418e+00, -6.7859e+00],
[-6.8662e+00, -7.9634e+00, -2.0530e+01, ..., -1.1066e+01,
-1.5489e+01, -4.2304e+00],
...,
[-6.9827e+00, -5.2284e+00, -1.1083e+01, ..., -5.6032e+00,
-3.5551e+00, -1.8822e+00],
[-8.2591e+00, -6.7930e+00, -6.9351e+00, ..., -1.2001e+01,
-1.3475e+01, -6.1900e+00],
[-2.1497e+00, -3.7727e+00, -1.7206e+00, ..., -4.3593e+00,
-3.6488e+00, -9.4097e-01]],
[[ 4.7118e-02, 1.5734e+00, 3.4471e-01, ..., -1.6396e-02,
2.1719e+00, 1.0712e+00],
[ 8.6147e-01, -7.3607e-01, 2.3845e-01, ..., 6.1429e+00,
4.2198e+00, 1.3948e-01],
[ 1.4907e+00, -8.7451e-02, 2.6278e+00, ..., 6.6923e+00,
1.1907e+01, 6.3181e+00],
...,
[-1.3092e-01, -5.1444e-01, 3.1993e+00, ..., -4.6700e+00,
6.8688e-01, 6.5921e-02],
[-3.6658e-01, -2.5847e+00, -3.5913e+00, ..., -9.9717e-01,
1.2680e-01, -2.7090e-01],
[ 5.7680e-01, 2.3389e-01, -1.0411e+00, ..., -3.1834e-01,
6.4041e-01, -1.2522e+00]],
...,
[[ 1.7853e-02, -1.9950e+00, -2.8220e+00, ..., -3.7278e+00,
-6.5845e+00, -2.4808e+00],
[-2.1084e+00, -4.1959e+00, -5.6743e+00, ..., -6.0318e+00,
-4.8948e+00, -1.6239e+00],
[-7.5564e-01, -3.8997e+00, -7.9343e+00, ..., -5.8992e+00,
-9.9693e+00, -6.7121e+00],
...,
[-1.0035e+00, -1.7146e-01, -3.0516e+00, ..., -3.3628e+00,
-3.3453e+00, -1.2464e+00],
[-3.6383e+00, -3.3508e+00, -4.8507e+00, ..., -6.8066e+00,
-8.1513e+00, -1.8820e+00],
[-1.8075e+00, -1.3138e+00, -3.5375e-02, ..., -5.4450e-01,
-1.6485e+00, 7.2021e-01]],
[[-1.3047e+00, -1.5776e+00, -4.9787e+00, ..., -6.2234e-01,
-1.4700e+00, -5.4633e-01],
[-2.4999e+00, -2.9523e+00, -5.2897e+00, ..., -2.3275e+00,
5.0038e-01, 2.6452e-01],
[-2.8359e+00, -5.9429e+00, -3.4534e-01, ..., -1.1688e+01,
-1.0828e+01, -1.0551e+00],
...,
[-2.6656e+00, -3.7807e+00, -5.7032e+00, ..., 2.9916e-04,
-1.5242e+00, 3.4577e-01],
[-2.3113e+00, 2.2193e-01, -4.9695e+00, ..., -3.0910e-01,
-2.1541e+00, 1.1014e+00],
[-2.6661e+00, -1.5283e+00, -1.9795e+00, ..., 8.1377e-01,
-9.6679e-01, 1.3536e-01]],
[[ 1.3270e+00, -2.2542e-01, 2.4083e+00, ..., 2.3017e+00,
1.5980e+00, -3.7572e-03],
[ 2.3946e+00, 2.4344e+00, 5.4295e+00, ..., 2.2687e+00,
4.1410e+00, 3.9656e+00],
[ 2.9420e+00, 5.1670e+00, 2.9324e+00, ..., 7.2404e+00,
8.4142e+00, 2.9831e+00],
...,
[ 3.0058e+00, 2.7207e+00, -3.4980e+00, ..., 1.6470e+00,
2.0367e-01, 1.6244e+00],
[ 2.5782e+00, 1.9881e+00, 5.2319e-01, ..., 3.0881e+00,
1.2514e+00, 2.4477e+00],
[ 2.3524e+00, 2.2693e+00, 1.0837e+00, ..., -3.3597e-01,
-1.0844e+00, 9.3898e-01]]],
[[[ 2.7015e-01, 1.2953e+00, 7.9307e-02, ..., -3.6817e+00,
-2.4661e+00, -1.0984e+00],
[ 4.7723e-01, 9.6311e-01, 1.6923e+00, ..., -2.7094e+00,
-2.4112e+00, -2.5822e-01],
[-3.7320e-01, -3.8420e-01, 7.5827e-01, ..., -2.2844e+00,
-2.9644e+00, 6.3960e-03],
...,
[-7.2232e-01, -2.2396e+00, 3.8959e-01, ..., -1.1148e+00,
-5.6850e+00, 5.8729e-02],
[-2.1854e+00, -1.6600e+00, -7.6877e-02, ..., -2.0026e+00,
-5.0296e+00, 3.3766e-01],
[-1.0997e+00, -3.2951e+00, -2.8531e+00, ..., -2.8293e+00,
-3.6995e+00, -1.5630e+00]],
[[-1.0294e+00, -1.5032e+00, -3.1575e+00, ..., -6.6893e+00,
-7.1683e+00, -1.4296e+00],
[-3.7926e+00, -4.1489e+00, -3.3714e+00, ..., -7.3045e+00,
-8.0883e+00, -2.4501e+00],
[-3.8800e+00, -4.3007e+00, -3.9928e+00, ..., -9.2482e+00,
-7.7417e+00, -2.5049e+00],
...,
[-5.8647e+00, -5.8179e+00, -5.1934e+00, ..., -9.2405e+00,
-1.1102e+01, -2.2178e+00],
[-7.4999e+00, -7.3393e+00, -7.2592e+00, ..., -8.0666e+00,
-1.1611e+01, -4.2341e+00],
[-1.6381e+00, -3.1630e+00, -3.7038e+00, ..., -3.9630e+00,
-3.9664e+00, -7.3215e-01]],
[[-3.5196e-02, 1.2299e+00, 1.7093e+00, ..., 1.4404e+00,
2.3147e+00, 1.0087e+00],
[-2.4478e-02, -1.1226e-01, 8.1229e-01, ..., 9.1552e-01,
2.7752e+00, 1.4742e-01],
[ 6.2461e-01, 1.1373e+00, 1.3532e+00, ..., 5.4750e-02,
1.5053e+00, -2.6244e-01],
...,
[ 1.4851e-01, 5.6373e-01, 1.6102e+00, ..., 3.4505e+00,
1.9708e+00, -1.2256e+00],
[ 4.1830e-01, 2.6567e-01, 6.9201e-01, ..., 1.5025e+00,
9.8559e-01, -1.5964e+00],
[ 5.6152e-01, 1.0187e+00, 1.3822e+00, ..., 7.8490e-01,
-1.1240e-02, -1.8136e+00]],
...,
[[-5.7645e-01, -1.5022e+00, -1.9738e+00, ..., -5.7473e+00,
-5.2433e+00, -7.8783e-01],
[-2.6582e+00, -2.5586e+00, -1.2093e+00, ..., -5.4109e+00,
-5.2936e+00, -1.7847e+00],
[-2.0765e+00, -1.2828e+00, -8.4258e-01, ..., -5.7432e+00,
-6.6977e+00, -1.4191e+00],
...,
[-1.7469e+00, -1.2227e+00, -9.8954e-01, ..., -6.9302e+00,
-7.1069e+00, -1.9121e+00],
[-3.2017e+00, -2.9128e+00, -2.0399e+00, ..., -5.6133e+00,
-5.3145e+00, -1.0584e+00],
[-1.3736e+00, -1.7810e+00, -1.8717e+00, ..., -3.1813e+00,
-2.1332e+00, 2.0051e-01]],
[[-1.2538e+00, -2.3239e+00, -3.9236e+00, ..., -1.7678e+00,
-5.3186e-01, -5.6611e-01],
[-3.4985e+00, -3.3668e+00, -4.4566e+00, ..., -3.2602e+00,
-2.1299e+00, -3.6543e-01],
[-4.4984e+00, -3.7703e+00, -4.2793e+00, ..., -1.9663e+00,
-2.4206e+00, -2.6548e-01],
...,
[-4.7264e+00, -3.7599e+00, -3.5790e+00, ..., -3.8262e+00,
-2.2678e+00, -1.3050e+00],
[-4.1414e+00, -1.3111e+00, -1.8408e+00, ..., -3.1498e+00,
-6.9543e-01, 1.0028e+00],
[-3.2493e+00, -2.8899e+00, -1.7449e+00, ..., -1.5201e+00,
-5.2184e-01, 5.6499e-01]],
[[ 1.1906e+00, -7.8920e-02, 1.0680e+00, ..., 1.3707e-01,
7.4543e-01, -1.4233e+00],
[ 2.2338e+00, 1.1665e+00, 1.3039e+00, ..., 2.8231e+00,
2.7381e+00, 2.2680e+00],
[ 1.3872e+00, 7.9933e-01, 6.8065e-01, ..., 2.6578e+00,
6.1799e-01, 1.3169e+00],
...,
[ 2.5378e+00, 1.1924e+00, 1.4786e+00, ..., 5.4616e+00,
1.5879e+00, 4.6630e-01],
[ 2.7247e+00, 3.7238e+00, 2.2380e+00, ..., 2.8069e+00,
2.5639e-01, 1.8681e+00],
[ 1.8099e+00, 1.4545e+00, 4.3605e-01, ..., 1.0195e+00,
-5.1506e-02, 9.0484e-01]]],
[[[-2.3484e-01, 8.1506e-01, 7.2127e-01, ..., -1.5920e+00,
-1.0994e+00, -8.8246e-01],
[ 9.6473e-01, 1.7682e+00, 1.9262e+00, ..., -1.3034e-01,
3.7581e-01, -1.0222e+00],
[ 1.5546e+00, 1.0750e+00, -1.0682e+00, ..., 1.4901e+00,
2.1595e-01, -1.8120e+00],
...,
[-1.3615e+00, -1.4012e+00, -8.0216e-01, ..., 5.1367e-03,
-1.8698e+00, 8.4452e-01],
[-2.0632e+00, -9.8827e-01, 3.8109e-01, ..., 7.3997e-01,
-9.2195e-01, 8.6294e-01],
[-2.0843e+00, -4.1915e+00, -2.7753e+00, ..., 2.6588e-01,
-1.6664e+00, -1.5408e+00]],
[[-6.7570e-01, -1.2215e+00, -2.3091e+00, ..., -3.2849e+00,
-4.7804e+00, -2.4529e+00],
[-1.9891e+00, -4.1761e+00, -4.7408e+00, ..., -2.9413e+00,
-5.0932e+00, -4.7534e+00],
[-2.0915e+00, -3.9831e+00, -8.5171e+00, ..., -2.1253e+00,
-4.5829e+00, -3.9760e+00],
...,
[-7.1268e+00, -7.3147e+00, -6.4870e+00, ..., -1.6377e+00,
-5.2153e+00, -7.8932e-01],
[-8.9221e+00, -8.9490e+00, -7.7643e+00, ..., -2.7345e+00,
-6.6785e+00, -3.1698e+00],
[-2.2829e+00, -4.3193e+00, -3.6309e+00, ..., -4.3999e-01,
-3.6133e+00, -9.3774e-01]],
[[ 6.4851e-01, 1.0964e+00, 1.3048e+00, ..., 1.6886e+00,
1.9857e+00, 8.7625e-01],
[ 7.2643e-01, 1.4080e+00, -3.3567e-01, ..., 1.9039e+00,
2.4401e+00, 9.2920e-01],
[ 2.7177e-01, 2.3756e+00, 5.1725e-01, ..., 2.3189e+00,
2.5138e+00, 2.1324e+00],
...,
[ 6.0624e-01, 9.2273e-01, 6.5355e-01, ..., -1.7089e+00,
2.5696e-01, -1.9216e+00],
[ 3.6309e-01, 7.0666e-02, 3.1941e-01, ..., -5.4188e-01,
-5.5090e-01, -1.4327e+00],
[ 7.3029e-01, 4.6888e-01, 9.5046e-02, ..., 2.9920e-01,
-4.5002e-01, -1.7222e+00]],
...,
[[-8.6542e-01, -4.5414e-01, -1.0203e+00, ..., -2.6876e+00,
-3.5923e+00, -1.2439e+00],
[-2.0115e+00, -1.6666e+00, -2.1597e+00, ..., -2.8976e+00,
-4.8116e+00, -3.6248e+00],
[-6.0730e-01, -1.0822e+00, -3.3091e+00, ..., -1.2310e+00,
-3.9150e+00, -2.9138e+00],
...,
[-2.2754e+00, -1.6527e+00, -6.1815e-01, ..., -1.2156e+00,
-2.4749e+00, -7.7169e-01],
[-2.7668e+00, -2.9127e+00, -2.9020e-01, ..., -2.9155e+00,
-3.4614e+00, -3.0010e-01],
[-2.5407e+00, -2.7111e+00, -8.5993e-01, ..., -2.1257e+00,
-2.1831e+00, 3.3800e-01]],
[[-1.4995e+00, -1.8521e+00, -3.7956e+00, ..., -2.6197e+00,
-2.8387e+00, -5.4169e-01],
[-4.0229e+00, -3.9866e+00, -5.0614e+00, ..., -3.2904e+00,
-5.6884e+00, -1.6844e+00],
[-3.7693e+00, -5.1331e+00, -7.3469e+00, ..., -8.0345e-01,
-4.4298e+00, -1.6568e+00],
...,
[-3.5995e+00, -2.4363e+00, -2.9631e+00, ..., 7.7680e-01,
1.0300e+00, 2.0850e-01],
[-3.3848e+00, 5.5603e-01, -1.0106e+00, ..., 2.8983e-01,
6.0342e-01, 1.5288e+00],
[-3.8863e+00, -1.6147e+00, -4.3673e-01, ..., -4.4885e-01,
-1.6503e-01, 9.9213e-01]],
[[ 8.4184e-01, -3.2693e-01, 6.1514e-01, ..., 5.9259e-01,
1.4253e+00, 5.8410e-01],
[ 1.7291e+00, 6.8037e-01, 1.6018e+00, ..., 1.1788e+00,
1.9500e+00, 3.3125e+00],
[ 1.5142e+00, 5.2933e-01, 2.1383e+00, ..., 2.5620e+00,
1.0792e+00, 1.0114e+00],
...,
[ 3.1129e+00, 2.2636e+00, 8.3804e-01, ..., 2.1966e+00,
2.4850e+00, 9.2224e-01],
[ 4.5926e+00, 4.7209e+00, 3.4501e+00, ..., 3.6530e+00,
1.7825e+00, 1.5987e+00],
[ 2.4123e+00, 2.0678e+00, 1.4569e+00, ..., 2.3398e+00,
6.4502e-01, 1.0980e+00]]]], device='cuda:0'),
tensor([[[[ 2.6579e-01, -1.9460e+00, -3.4728e+00, ..., -3.1944e+00,
-2.2204e+00, -9.9019e-01],
[-5.1554e+00, -4.4848e+00, 1.1522e+00, ..., -2.0055e+00,
-2.1117e+00, -2.3608e-01],
[-3.4987e+00, 9.3691e-01, 2.2952e+00, ..., -1.2400e+00,
-1.7084e+00, -7.1872e-02],
...,
[-7.3361e-01, -1.8726e+00, -2.9258e-01, ..., -2.3516e+00,
-4.8930e+00, 1.8821e+00],
[-1.9921e+00, -1.5281e+00, -1.2119e+00, ..., -4.9707e+00,
-4.6240e+00, 5.8913e-01],
[-1.1952e+00, -3.3156e+00, -3.4447e+00, ..., -3.2700e+00,
-1.3032e+00, 7.5419e-01]],
[[-4.4298e+00, -5.1655e+00, -8.5247e+00, ..., -6.4266e+00,
-7.0504e+00, -1.3361e+00],
[-1.0128e+01, -7.5812e+00, -2.3373e+00, ..., -4.9160e+00,
-8.4635e+00, -2.7245e+00],
[-2.6648e+00, 3.2567e+00, 6.9913e-01, ..., -2.6735e+00,
-6.6472e+00, -3.0715e+00],
...,
[-5.9991e+00, -6.1715e+00, -4.8162e+00, ..., -1.6269e+00,
-3.2973e+00, 2.6996e+00],
[-7.6843e+00, -7.9159e+00, -7.7258e+00, ..., -2.7641e+00,
-3.8848e+00, 1.3059e+00],
[-1.7902e+00, -3.4538e+00, -3.8500e+00, ..., 8.1255e-01,
-1.5223e+00, 8.3490e-01]],
[[-1.0943e+00, 1.4005e+00, 4.2828e+00, ..., 1.4976e+00,
2.4247e+00, 9.9979e-01],
[ 2.0929e+00, 3.5774e+00, 3.6811e+00, ..., 1.8891e+00,
3.1288e+00, 1.0739e-01],
[ 1.5085e+00, -3.4891e+00, -6.7444e+00, ..., 2.2990e+00,
2.9577e+00, -1.4080e-01],
...,
[ 2.9437e-01, 7.5819e-01, 1.1943e+00, ..., -7.7104e-01,
-4.9117e+00, -4.1592e+00],
[ 4.9829e-01, 6.3323e-01, 1.0388e+00, ..., -6.7622e-01,
-9.6311e-01, -4.9443e+00],
[ 5.1807e-01, 1.1439e+00, 1.6065e+00, ..., 1.8749e-01,
5.4692e-01, -2.0466e+00]],
...,
[[-1.4767e+00, -3.4665e+00, -5.4918e+00, ..., -5.0739e+00,
-5.0811e+00, -7.9075e-01],
[-5.3991e+00, -3.2636e+00, -1.4941e+00, ..., -3.7082e+00,
-4.4671e+00, -2.1306e+00],
[-1.3443e+00, 1.6436e+00, -2.5323e+00, ..., -3.3696e+00,
-4.4568e+00, -1.4666e+00],
...,
[-1.9593e+00, -1.6166e+00, -9.8515e-01, ..., 3.6857e-01,
1.5256e+00, -1.1257e+00],
[-3.1402e+00, -3.1765e+00, -2.9208e+00, ..., -1.8895e+00,
-3.3569e-01, -6.2314e-01],
[-1.5655e+00, -2.1150e+00, -2.4905e+00, ..., 3.2042e-01,
-6.4566e-01, -2.2344e-01]],
[[-5.2458e-01, -2.5520e+00, -5.2928e+00, ..., -1.5521e+00,
-5.2920e-01, -6.3495e-01],
[-5.6985e+00, -3.4370e+00, -4.7517e+00, ..., -2.7728e+00,
-1.3456e+00, -1.8354e-01],
[-3.9040e+00, -2.6142e+00, 2.2688e-01, ..., -2.2246e+00,
-7.9342e-01, 5.1276e-01],
...,
[-4.8637e+00, -4.0425e+00, -4.1249e+00, ..., -3.8939e+00,
-1.6706e+00, -1.0438e+00],
[-4.2025e+00, -2.1575e+00, -3.0328e+00, ..., -4.3897e+00,
-2.4446e+00, 5.7482e-02],
[-3.2917e+00, -3.3134e+00, -2.6669e+00, ..., -1.8504e+00,
-5.7752e-01, 7.9236e-01]],
[[ 1.8765e+00, -8.4649e-02, 9.8083e-01, ..., 5.0641e-01,
8.6176e-01, -1.3275e+00],
[ 1.2229e+00, 3.6691e+00, 6.1984e+00, ..., 3.4078e+00,
3.2969e+00, 2.2533e+00],
[-1.3740e+00, 3.9212e+00, -1.0873e+00, ..., 2.8000e+00,
2.6558e+00, 1.6008e+00],
...,
[ 2.4527e+00, 1.5890e+00, 1.1782e+00, ..., -6.5732e+00,
-5.2501e+00, 4.5177e-01],
[ 2.9821e+00, 4.1289e+00, 2.5834e+00, ..., -7.6878e+00,
-5.0236e+00, 8.6715e-02],
[ 1.9677e+00, 1.7597e+00, 5.0131e-01, ..., -5.0401e+00,
-3.4246e+00, 6.3962e-01]]],
[[[ 2.7839e-01, 1.3763e+00, 5.5987e-01, ..., -6.3826e-01,
2.1948e-02, 6.5792e-01],
[ 4.5886e-01, 4.2381e-01, 8.1400e-01, ..., -6.8959e-01,
-1.7684e+00, 4.3384e-01],
[ 1.4440e-01, 6.0830e-01, 1.8051e+00, ..., -2.4092e+00,
-1.2039e+00, -3.7238e-01],
...,
[ 4.4142e-01, -1.0919e+00, 8.4600e-01, ..., -2.9029e-01,
-4.5407e-01, -1.0104e+00],
[-1.3279e+00, -1.3452e+00, -9.8496e-01, ..., -1.7195e+00,
-2.5223e+00, -1.2966e+00],
[-2.4155e-01, -2.6633e+00, -1.3138e+00, ..., -2.2785e+00,
-2.3133e+00, -8.8662e-01]],
[[-6.8947e-01, -2.1795e+00, -4.0336e+00, ..., -3.6571e+00,
-5.0713e+00, -3.7575e-01],
[-4.4351e+00, -5.2436e+00, -5.0243e+00, ..., -6.3116e+00,
-9.2424e+00, -2.6215e+00],
[-3.3828e+00, -5.2219e+00, -7.3649e+00, ..., -5.2361e+00,
-1.0770e+01, -5.3318e+00],
...,
[-4.5869e+00, -5.0314e+00, -3.4806e+00, ..., -4.8136e+00,
-4.9640e+00, -2.2100e+00],
[-7.6441e+00, -9.4365e+00, -7.3649e+00, ..., -8.3862e+00,
-8.9490e+00, -4.0181e+00],
[-1.0755e+00, -3.6475e+00, -1.8287e+00, ..., -3.3836e+00,
-2.8549e+00, -2.7817e-01]],
[[-2.7345e-01, 2.1707e+00, 1.2656e+00, ..., 9.9007e-01,
2.0074e+00, 8.9855e-01],
[-1.1685e-01, 8.8461e-01, 8.6292e-01, ..., -1.2104e-01,
7.4577e-01, -6.2333e-01],
[-5.7317e-02, 1.9675e+00, 2.0897e+00, ..., 2.3007e+00,
2.2245e+00, -4.1844e-01],
...,
[ 1.1599e-01, 1.0206e+00, 4.7115e-01, ..., 1.2573e+00,
1.2377e+00, 7.3699e-01],
[ 1.3754e-01, 1.0183e+00, 1.4029e-01, ..., 2.2350e+00,
2.0057e+00, 7.3989e-01],
[-8.0132e-01, 1.0080e+00, 4.2103e-01, ..., 1.0757e+00,
9.9280e-01, -6.5685e-01]],
...,
[[-7.2393e-01, -1.7559e+00, -1.7157e+00, ..., -2.4313e+00,
-3.0412e+00, -3.5474e-01],
[-2.8143e+00, -4.0615e+00, -2.5691e+00, ..., -3.4176e+00,
-4.6202e+00, -3.2151e+00],
[-1.2625e+00, -2.6019e+00, -1.4208e+00, ..., -4.9111e+00,
-4.2029e+00, -3.1611e+00],
...,
[-1.1690e+00, -1.1820e+00, -8.7124e-01, ..., -2.1107e+00,
-3.9592e+00, -1.8730e+00],
[-3.0643e+00, -3.6970e+00, -3.2620e+00, ..., -3.6337e+00,
-4.7372e+00, -1.5322e+00],
[-1.0062e+00, -1.6111e+00, -7.2303e-01, ..., -7.8599e-01,
-1.0436e+00, 1.2235e+00]],
[[-1.6393e+00, -2.2512e+00, -2.9263e+00, ..., -1.6171e+00,
-1.3213e+00, -1.3066e+00],
[-4.0725e+00, -3.1223e+00, -2.6681e+00, ..., -2.1500e+00,
-2.0722e+00, -1.1649e+00],
[-2.4879e+00, -1.6677e+00, -2.8588e+00, ..., -5.2691e-01,
-3.2480e-01, -5.5307e-01],
...,
[-3.5824e+00, -3.0431e+00, -3.1876e+00, ..., -1.2202e+00,
-2.2681e+00, -1.1418e+00],
[-4.1035e+00, -1.7736e+00, -3.8253e+00, ..., -2.4802e-01,
-2.0388e+00, -6.8486e-01],
[-3.0582e+00, -2.2014e+00, -1.9988e+00, ..., -2.6050e-01,
-5.6167e-01, -4.3974e-03]],
[[ 1.0570e+00, 5.6952e-01, 6.4448e-01, ..., -7.6673e-01,
7.7987e-02, -4.1355e-01],
[ 3.3395e+00, 3.3176e+00, 2.2154e+00, ..., 2.0902e+00,
2.0353e+00, 2.0947e+00],
[ 1.6377e+00, 1.3828e+00, 1.2069e+00, ..., 1.8676e+00,
3.1328e+00, 1.5136e+00],
...,
[ 2.8554e+00, 1.7950e+00, 1.4543e+00, ..., 1.3371e+00,
1.0548e+00, 4.0773e-01],
[ 3.6284e+00, 4.5940e+00, 3.2766e+00, ..., 2.5346e+00,
1.1073e+00, 1.5263e+00],
[ 2.2961e+00, 2.1565e+00, 1.4282e+00, ..., 3.1899e-01,
1.1599e-01, 9.1691e-01]]],
[[[ 8.5303e-01, 2.0701e+00, 7.3957e-01, ..., -3.7162e+00,
-2.5534e+00, -9.1211e-01],
[ 2.0611e+00, 8.8174e-01, -1.4523e+00, ..., -3.2006e+00,
-2.6604e+00, -1.5055e-01],
[ 6.7996e-01, -2.2445e+00, -4.6121e+00, ..., -2.6874e+00,
-2.9430e+00, -6.7153e-02],
...,
[-3.4076e-01, -7.7874e-01, 2.0762e+00, ..., -2.2816e+00,
-2.5929e+00, 2.2914e+00],
[ 5.3723e-01, 1.0234e+00, -3.8689e-01, ..., -7.1688e-01,
2.3815e+00, 3.5048e+00],
[ 1.7621e-01, -2.2240e+00, -2.8141e+00, ..., 1.6871e+00,
-6.6873e-01, -2.2161e+00]],
[[ 4.1075e-01, -1.2546e+00, -3.1165e+00, ..., -7.4336e+00,
-7.3101e+00, -1.2301e+00],
[-2.3854e+00, -4.4864e+00, -5.7791e+00, ..., -7.7345e+00,
-8.3510e+00, -2.3630e+00],
[-1.1452e+00, -5.2868e+00, -4.4205e+00, ..., -8.3195e+00,
-8.0991e+00, -2.5849e+00],
...,
[-1.0680e+00, -8.0130e-01, -1.4543e+00, ..., -4.1955e-01,
-4.5657e-01, 1.2793e+00],
[-3.2525e+00, -7.0118e+00, -6.5382e+00, ..., -1.4044e+00,
-2.0045e-01, -1.7201e+00],
[ 5.5075e-01, -2.0132e+00, -2.6329e+00, ..., -2.7564e+00,
-3.3813e+00, -2.7498e+00]],
[[ 4.5641e-01, 2.2525e+00, 1.6668e+00, ..., 7.2852e-01,
2.1819e+00, 9.4227e-01],
[-3.2671e-01, 3.9900e-01, 3.4190e-01, ..., 9.7940e-01,
2.7423e+00, -2.7491e-02],
[-3.6396e-02, 3.5803e+00, 3.1030e+00, ..., -3.5035e-01,
1.7875e+00, -3.3315e-01],
...,
[-1.4138e+00, 2.3985e-01, 4.6579e-01, ..., -8.0846e-02,
1.0603e+00, 8.3043e-01],
[-1.5226e-01, 7.2677e-01, -3.2175e-01, ..., -1.6016e-01,
1.7221e-01, -3.9688e-01],
[-1.2841e+00, 4.7735e-01, 2.4698e-01, ..., -3.8981e-01,
-5.5965e-01, 1.6010e-01]],
...,
[[-2.0969e-01, -9.5028e-01, -1.2849e+00, ..., -5.3738e+00,
-5.3496e+00, -6.5549e-01],
[-2.4394e+00, -4.1381e+00, -3.2280e+00, ..., -4.5554e+00,
-5.2306e+00, -1.8477e+00],
[-1.8233e+00, -3.4408e+00, -2.2584e+00, ..., -6.1397e+00,
-5.8082e+00, -1.3817e+00],
...,
[-1.4748e+00, -3.2508e+00, -1.2600e+00, ..., -1.7633e-01,
-1.8357e+00, -1.7704e+00],
[-1.7791e+00, -4.8227e+00, -2.9726e+00, ..., -2.1799e+00,
-2.4665e+00, -2.0674e+00],
[ 3.9358e-01, -1.7890e+00, -4.1309e-01, ..., -1.9906e+00,
-5.8238e+00, -4.4334e+00]],
[[-1.0794e+00, -8.8503e-01, -1.0142e+00, ..., -2.0770e+00,
-4.3977e-01, -6.0250e-01],
[-3.0701e+00, -3.3232e+00, -2.7609e+00, ..., -3.7380e+00,
-1.6839e+00, -2.7742e-01],
[-2.9856e+00, -4.6306e+00, -2.2321e+00, ..., -2.8404e+00,
-2.4675e+00, -4.7085e-01],
...,
[-3.1647e-01, -4.3205e+00, -3.1868e+00, ..., -4.8885e+00,
-4.0505e+00, -8.9510e-01],
[-1.1855e+00, -2.3984e+00, -2.4624e+00, ..., -1.6298e+00,
-1.8435e+00, 1.2089e+00],
[-1.1718e+00, -1.3895e+00, -4.5415e-01, ..., 7.1391e-02,
-4.0046e+00, -1.1431e+00]],
[[ 8.1741e-01, 1.1881e+00, 1.4653e+00, ..., -3.7407e-01,
5.0896e-01, -1.3715e+00],
[ 2.0392e+00, 3.1451e+00, 2.0593e+00, ..., 3.0561e+00,
2.8846e+00, 2.1786e+00],
[ 9.4362e-01, 2.6334e-01, -7.5042e-01, ..., 2.6700e+00,
1.9094e+00, 1.3924e+00],
...,
[ 2.0605e+00, -6.9452e-01, -4.7567e-01, ..., -4.7698e+00,
-6.4583e+00, -1.0706e+00],
[ 3.1942e+00, 4.5278e+00, 2.7990e+00, ..., -3.7162e+00,
-1.7323e+00, 2.3329e+00],
[ 1.3897e+00, -2.2124e-01, -3.4595e-01, ..., 1.4936e+00,
2.7241e+00, 1.7507e+00]]],
[[[ 1.3485e+00, 2.1235e+00, 1.7747e+00, ..., 2.3401e-01,
6.9580e-01, -3.4235e-01],
[ 1.5535e+00, -2.4963e-01, -3.0997e+00, ..., 4.0925e-01,
4.2112e-01, -9.1734e-01],
[-1.5566e-01, -3.4958e+00, -4.1519e+00, ..., -3.1590e+00,
-3.5730e+00, -2.9446e+00],
...,
[ 2.8194e+00, 1.4066e+00, 4.6707e+00, ..., -1.6968e+00,
-1.1876e+00, -1.3865e-01],
[ 2.4093e-02, -2.7288e+00, -4.7435e-01, ..., -7.1760e-01,
-6.3465e-02, -1.0550e+00],
[-9.1914e-01, -3.3152e+00, -2.5307e+00, ..., -2.6921e+00,
-3.2507e+00, -2.9713e+00]],
[[-1.0238e+00, -3.9934e+00, -7.7609e+00, ..., -4.1494e+00,
-4.7320e+00, -3.3623e+00],
[-4.7434e+00, -8.3072e+00, -1.0161e+01, ..., -4.1699e+00,
-3.3275e+00, -3.5473e+00],
[-2.9432e+00, -8.3459e+00, -5.7746e+00, ..., -4.3299e+00,
-4.6253e+00, -2.6860e+00],
...,
[-4.9031e-01, -3.5258e-02, -4.1936e+00, ..., -6.5041e+00,
-7.4695e+00, -3.9571e+00],
[-8.3061e+00, -1.4148e+01, -1.1526e+01, ..., -4.3399e+00,
-4.9942e+00, -5.7506e+00],
[-8.0018e-01, -3.6743e+00, -3.5467e+00, ..., -3.4968e+00,
-3.3485e+00, -1.7838e+00]],
[[ 5.9930e-01, 2.9014e+00, 1.7351e+00, ..., 1.8334e+00,
1.6070e+00, 6.3960e-01],
[-1.0629e+00, 3.1204e-01, 1.6873e+00, ..., 1.9425e+00,
1.4998e+00, 6.0121e-01],
[-4.9948e-01, 1.9682e+00, 2.3761e+00, ..., 3.7546e+00,
3.7709e+00, 3.4747e+00],
...,
[-4.5132e+00, -1.1233e+00, -8.5172e-01, ..., 7.0988e-01,
-1.1565e+00, -5.0213e-01],
[-2.6538e+00, -1.3074e+00, -1.8704e+00, ..., 5.7729e-01,
1.4669e-02, 6.7292e-01],
[-1.6356e-01, 1.0580e+00, 6.2025e-01, ..., 6.8000e-01,
1.0786e+00, 8.0676e-01]],
...,
[[-1.7712e-01, -2.3291e+00, -2.2138e+00, ..., -1.5456e+00,
-2.4735e+00, -5.6504e-01],
[-2.6237e+00, -5.2680e+00, -3.9463e+00, ..., -1.0756e+00,
-3.4177e+00, -3.1330e+00],
[-2.9147e+00, -3.8154e+00, -9.2117e-01, ..., -2.3572e+00,
-3.9874e+00, -2.8445e+00],
...,
[ 1.4692e+00, -1.3088e+00, 1.3044e-01, ..., -3.9212e+00,
-4.7781e+00, -3.3031e+00],
[-4.4248e+00, -9.4939e+00, -3.4047e+00, ..., -8.2127e-01,
-3.2291e+00, -2.4270e+00],
[-1.2029e+00, -1.5456e+00, 1.5439e-01, ..., -2.1023e+00,
-3.1982e+00, -7.4515e-01]],
[[-2.3201e+00, -2.9769e+00, -1.2860e+00, ..., 5.6462e-01,
-1.5492e+00, -9.3563e-01],
[-3.8044e+00, -4.9332e+00, -3.9304e+00, ..., 4.1687e-01,
-3.3663e+00, -9.7525e-01],
[-3.3514e+00, -1.9155e+00, -7.5141e-01, ..., -2.0971e+00,
-7.3552e+00, -4.1293e+00],
...,
[-1.3910e-01, -2.3026e+00, 2.4693e-01, ..., -1.9763e+00,
-1.2389e+00, -8.2406e-01],
[-2.7606e+00, -2.4604e+00, -9.1205e-01, ..., -2.0419e-01,
-2.3496e+00, -6.8021e-01],
[-2.2965e+00, -1.3442e+00, -9.1422e-02, ..., -2.9343e-01,
-2.3470e+00, -5.2748e-01]],
[[ 1.4696e+00, 1.6952e+00, 2.9028e+00, ..., 1.1689e+00,
1.6048e+00, 1.0039e+00],
[ 3.0267e+00, 2.8764e+00, 1.4960e+00, ..., 2.3323e+00,
1.4962e+00, 1.6320e+00],
[ 1.0302e+00, 2.0283e-01, 1.3685e-01, ..., 7.2079e-01,
-1.2378e-01, 3.0741e-01],
...,
[ 2.5612e+00, -1.2992e+00, -2.7036e-01, ..., 2.4732e+00,
2.7425e+00, 2.0449e+00],
[ 5.4230e+00, 4.5580e+00, 3.3127e+00, ..., 8.7369e-01,
1.4376e+00, 2.0087e+00],
[ 1.3767e+00, 2.6973e-01, 3.1596e-01, ..., 6.1122e-01,
-9.1380e-02, 2.9240e-01]]],
[[[ 2.1794e-01, 1.2421e+00, 8.7640e-02, ..., -3.2808e+00,
-2.4658e+00, -9.5802e-01],
[ 3.8515e-01, 7.8591e-01, 5.9025e-01, ..., -1.8301e+00,
-2.3043e+00, -7.1636e-02],
[-6.2775e-03, -3.7506e-01, 4.1815e-01, ..., -1.0084e+00,
-2.5522e+00, 1.6412e-01],
...,
[-4.8324e-01, -1.8581e+00, -1.5355e+00, ..., -1.8920e+00,
-3.7345e+00, -8.0032e-01],
[-2.3418e+00, -1.7183e+00, -1.9761e+00, ..., -2.3135e+00,
-2.1766e+00, 5.2023e-01],
[-1.1794e+00, -3.2212e+00, -3.3277e+00, ..., -4.4428e+00,
-3.6059e+00, -1.7246e+00]],
[[-1.0340e+00, -1.5415e+00, -3.4080e+00, ..., -6.3029e+00,
-7.1144e+00, -1.1231e+00],
[-3.8033e+00, -3.9400e+00, -3.4248e+00, ..., -5.8317e+00,
-8.0477e+00, -2.2325e+00],
[-4.3615e+00, -5.0367e+00, -3.7694e+00, ..., -4.6369e+00,
-7.4511e+00, -2.2352e+00],
...,
[-6.3659e+00, -5.2848e+00, -3.8736e+00, ..., -1.1322e+01,
-1.1662e+01, -4.2261e+00],
[-7.7063e+00, -7.9350e+00, -8.1173e+00, ..., -7.8808e+00,
-9.4776e+00, -3.8386e+00],
[-1.7498e+00, -3.2692e+00, -3.7246e+00, ..., -3.8595e+00,
-3.9034e+00, -1.0400e+00]],
[[ 1.8987e-02, 1.3088e+00, 1.5826e+00, ..., 1.5498e+00,
2.3805e+00, 9.8572e-01],
[-2.4938e-02, 6.9865e-02, 9.6678e-01, ..., 1.5504e+00,
2.8572e+00, -5.2834e-02],
[ 6.4365e-01, 1.1394e+00, 1.7688e+00, ..., 1.8088e-01,
1.9926e+00, -3.4269e-01],
...,
[-9.3561e-01, 1.2639e+00, 2.9636e+00, ..., -1.6683e+00,
-4.4521e-01, -7.4244e-01],
[ 4.1641e-01, 4.2589e-01, 1.1971e+00, ..., -5.3257e-01,
5.9406e-01, -1.1638e+00],
[ 6.3764e-01, 1.1193e+00, 1.5316e+00, ..., -1.2088e-01,
-7.7060e-01, -2.3145e+00]],
...,
[[-6.3694e-01, -1.5565e+00, -2.3011e+00, ..., -4.9970e+00,
-5.0217e+00, -7.1295e-01],
[-2.7315e+00, -2.8224e+00, -2.0499e+00, ..., -3.5390e+00,
-4.6193e+00, -1.7694e+00],
[-2.0949e+00, -1.3122e+00, -1.1879e+00, ..., -4.0373e+00,
-4.9662e+00, -1.2088e+00],
...,
[-1.3354e+00, 2.6835e-01, 1.1385e+00, ..., -6.3180e+00,
-7.0607e+00, -2.4955e+00],
[-3.4880e+00, -3.9132e+00, -3.7279e+00, ..., -2.5222e+00,
-4.4372e+00, -4.4168e-01],
[-1.5056e+00, -2.0072e+00, -2.5781e+00, ..., -2.3022e+00,
-2.3896e+00, 1.0296e-01]],
[[-1.3685e+00, -2.4807e+00, -3.9306e+00, ..., -1.4057e+00,
-3.8408e-01, -5.2833e-01],
[-3.5841e+00, -3.6651e+00, -4.8020e+00, ..., -3.2727e+00,
-1.6508e+00, -1.9720e-01],
[-4.2107e+00, -2.9431e+00, -3.2353e+00, ..., -1.8499e+00,
-1.3036e+00, -2.4374e-01],
...,
[-4.5235e+00, -4.6038e+00, -6.0685e+00, ..., -5.1603e+00,
-2.7442e+00, -3.4221e-01],
[-3.9461e+00, -2.2894e+00, -3.1152e+00, ..., -1.0563e+00,
-1.2910e+00, 3.8186e-01],
[-3.2422e+00, -3.1981e+00, -2.6180e+00, ..., 1.9272e-01,
-7.2619e-01, 5.6900e-01]],
[[ 1.1729e+00, -9.1427e-02, 1.1848e+00, ..., 4.7358e-01,
7.7456e-01, -1.3440e+00],
[ 2.3451e+00, 1.1122e+00, 1.2339e+00, ..., 3.4954e+00,
2.9810e+00, 2.2798e+00],
[ 1.8910e+00, 1.1282e+00, 6.6831e-01, ..., 2.7257e+00,
1.7465e+00, 1.2253e+00],
...,
[ 2.0692e+00, 4.0947e-01, -1.4795e+00, ..., 3.8618e+00,
2.2155e+00, 2.0919e+00],
[ 3.2233e+00, 4.5567e+00, 3.0313e+00, ..., 8.2588e-02,
2.9767e-01, 2.2651e+00],
[ 1.8858e+00, 1.8389e+00, 4.8203e-01, ..., -1.0937e-01,
-5.9739e-01, 7.1607e-01]]]], device='cuda:0'),
tensor([[[[-1.9689e-01, 5.6624e-01, -1.4706e-01, ..., -1.2838e+00,
-8.3050e-01, -7.4546e-01],
[ 6.6709e-01, 3.2428e-01, 4.8899e-01, ..., -3.4203e-01,
-5.0317e-02, -9.5618e-01],
[ 1.0635e+00, -6.3212e-01, 1.3333e+00, ..., -9.6237e-01,
-1.3330e+00, -1.9688e+00],
...,
[-4.2710e-01, -2.6238e+00, -4.4247e+00, ..., -2.0074e+00,
-2.3446e+00, -1.4869e-02],
[-2.4832e+00, -2.5887e+00, -1.2110e+00, ..., -2.3200e+00,
-1.8026e+00, 3.3660e-01],
[-2.4512e+00, -3.6573e+00, -2.7438e+00, ..., -4.0434e+00,
-3.1747e+00, -1.5117e+00]],
[[-7.7698e-01, -1.5004e+00, -2.1207e+00, ..., -2.8946e+00,
-4.2043e+00, -2.3003e+00],
[-1.8994e+00, -3.4366e+00, -3.5915e+00, ..., -4.6981e+00,
-5.0787e+00, -4.4378e+00],
[-1.7251e+00, -3.1378e+00, -3.4675e+00, ..., -5.8689e+00,
-4.8173e+00, -3.6429e+00],
...,
[-7.2848e+00, -8.4616e+00, -9.1534e+00, ..., -8.2728e+00,
-8.2455e+00, -1.6399e+00],
[-1.0136e+01, -1.0017e+01, -9.1267e+00, ..., -9.9403e+00,
-9.9771e+00, -3.7264e+00],
[-2.6531e+00, -4.5301e+00, -4.1914e+00, ..., -4.2357e+00,
-4.1554e+00, -7.3863e-01]],
[[ 7.1405e-01, 1.2202e+00, 1.0781e+00, ..., 1.3585e+00,
1.7900e+00, 8.7129e-01],
[ 9.0040e-01, 1.3196e+00, 1.5497e+00, ..., 1.0441e+00,
1.9384e+00, 8.5455e-01],
[ 5.0931e-01, 2.8593e+00, 2.6277e+00, ..., 1.5277e+00,
2.0634e+00, 1.8567e+00],
...,
[ 2.6899e-01, -2.8389e-01, -6.2773e-01, ..., 1.8280e+00,
1.4314e+00, 1.1884e-01],
[-4.1640e-01, 1.1386e-01, -1.6244e+00, ..., 2.4145e-01,
8.5485e-01, -1.3243e-01],
[ 8.1518e-01, 6.0619e-01, -3.6583e-01, ..., -7.4207e-01,
-7.8128e-01, -1.7369e+00]],
...,
[[-1.0673e+00, -1.3196e+00, -1.4275e+00, ..., -2.7789e+00,
-3.3107e+00, -1.1255e+00],
[-2.3363e+00, -2.9658e+00, -1.6063e+00, ..., -2.9830e+00,
-4.8948e+00, -3.3676e+00],
[-1.0146e+00, -1.7636e+00, -3.1375e-01, ..., -1.9500e+00,
-5.3939e+00, -3.0388e+00],
...,
[-2.9372e+00, -5.1008e+00, -3.4335e+00, ..., -5.3537e+00,
-6.0964e+00, -2.7630e+00],
[-3.8004e+00, -5.6083e+00, -3.2503e+00, ..., -3.5370e+00,
-4.9130e+00, -1.3951e+00],
[-3.2722e+00, -3.5093e+00, -3.7009e+00, ..., -2.6792e+00,
-2.9516e+00, 1.5645e-01]],
[[-1.5609e+00, -2.3039e+00, -3.1727e+00, ..., -2.8894e+00,
-3.0054e+00, -5.5841e-01],
[-4.3365e+00, -4.5378e+00, -4.3658e+00, ..., -3.9255e+00,
-6.0533e+00, -1.9549e+00],
[-4.0361e+00, -4.9360e+00, -4.3514e+00, ..., -3.4124e+00,
-5.4871e+00, -2.1017e+00],
...,
[-4.3886e+00, -4.2745e+00, -3.2606e+00, ..., -4.0752e+00,
-2.8925e+00, -1.1656e+00],
[-3.5840e+00, -1.1698e+00, -1.2161e+00, ..., -1.4604e+00,
-9.8715e-01, 2.4951e-01],
[-3.6386e+00, -2.5064e+00, -1.9725e+00, ..., -1.2407e+00,
-1.0752e+00, 3.9464e-01]],
[[ 8.9190e-01, -2.1320e-01, 5.1516e-01, ..., 2.7879e-01,
1.2099e+00, 5.5003e-01],
[ 1.7210e+00, 7.2762e-01, 1.4639e-01, ..., 1.1057e+00,
1.2552e+00, 3.1388e+00],
[ 1.3422e+00, -1.1373e-01, 3.4865e-01, ..., 1.4551e-01,
3.2956e-01, 8.5613e-01],
...,
[ 4.2529e+00, 3.8102e+00, 2.8414e+00, ..., 2.7475e+00,
2.8595e+00, 8.6663e-01],
[ 4.5055e+00, 4.2639e+00, 3.1474e+00, ..., 2.4285e+00,
2.2859e+00, 1.5773e+00],
[ 2.7492e+00, 3.1265e+00, 1.7182e+00, ..., 5.3223e-01,
6.5622e-01, 9.6010e-01]]],
[[[-3.7972e+00, -6.8068e+00, -2.6044e+00, ..., 7.3457e-01,
1.4711e+00, -8.5423e-01],
[-4.5273e+00, -7.9453e+00, -6.3495e+00, ..., 5.1800e-01,
2.4268e+00, -6.8880e-02],
[-5.8055e+00, -5.5810e+00, -6.6970e+00, ..., 5.4850e+00,
2.3950e+00, -2.2235e+00],
...,
[-8.7840e-01, -1.9191e+00, 6.7283e-01, ..., 3.9483e-01,
-2.4848e+00, -3.5435e+00],
[-1.2223e+00, -7.3126e-01, 2.0569e-02, ..., 7.1240e-01,
2.5932e-01, -2.0534e+00],
[-2.9270e+00, -5.4497e+00, -4.3978e+00, ..., 6.1849e-02,
-1.0702e+00, -2.8747e+00]],
[[-8.1916e+00, -6.2892e+00, -2.7496e-01, ..., 6.1488e-01,
-2.9558e+00, -2.8184e+00],
[-6.9204e+00, -2.4511e+00, -9.4210e-01, ..., -2.0412e+00,
-2.0240e+00, -3.3230e+00],
[-6.2275e+00, -1.3387e+00, -8.7330e-01, ..., 2.5309e+00,
-5.4932e+00, -6.4708e+00],
...,
[-6.2107e+00, -6.1796e+00, -5.6920e+00, ..., -7.4601e-01,
-9.3895e+00, -4.0652e+00],
[-7.8244e+00, -9.2574e+00, -7.3108e+00, ..., 1.7848e+00,
-3.8007e+00, -6.0270e+00],
[-2.6964e+00, -5.0873e+00, -3.0030e+00, ..., -6.0322e-01,
-1.6948e+00, -1.5125e+00]],
[[ 3.5248e-01, 2.8701e+00, -9.0484e-01, ..., -1.7122e+00,
3.2327e+00, 9.9351e-01],
[ 1.3570e+00, -7.5655e-02, -3.7050e+00, ..., -1.2972e+01,
-1.1316e+00, 1.2464e+00],
[ 1.0002e+00, 3.8428e-01, 3.7764e+00, ..., -4.2454e+00,
5.2062e-02, 1.1180e+00],
...,
[ 1.8923e-01, 1.1027e+00, 8.2534e-01, ..., 5.8632e+00,
3.3996e+00, -3.5642e-02],
[ 8.3867e-01, 1.1900e+00, 2.2514e-01, ..., -4.1524e+00,
2.1959e+00, 5.1888e-01],
[ 1.4764e+00, 1.7370e+00, 6.9399e-01, ..., -3.3909e+00,
4.9408e-01, 8.7313e-01]],
...,
[[-4.1502e+00, -2.9154e+00, 1.1120e+00, ..., -6.7984e-02,
-2.3098e+00, -6.9196e-01],
[-5.5703e+00, -2.3391e+00, -3.0125e+00, ..., -2.3993e+00,
-4.7708e+00, -1.8956e+00],
[-3.6961e+00, -1.2290e+00, -5.2023e+00, ..., -4.8505e+00,
-7.3268e+00, -2.5365e+00],
...,
[-1.9905e+00, -1.8333e+00, -5.6958e-01, ..., -6.5170e+00,
-3.4846e+00, -2.8081e+00],
[-3.0749e+00, -2.9490e+00, -1.5184e+00, ..., -3.9001e+00,
-3.7518e+00, -1.8727e+00],
[-3.8220e+00, -3.8209e+00, -2.2971e+00, ..., -3.6344e+00,
-5.0048e+00, -2.0243e+00]],
[[-1.0341e+00, 2.2993e+00, -1.5910e+00, ..., 2.3311e+00,
-2.3687e-01, 2.9960e-01],
[-3.3187e+00, -2.1703e+00, -8.6157e+00, ..., 4.7819e+00,
7.5011e-01, 8.6438e-01],
[-7.4105e-01, -1.7998e-01, -1.5911e+01, ..., 4.8860e+00,
-3.2161e+00, -1.1582e+00],
...,
[-3.0538e+00, -1.9514e+00, -3.4029e+00, ..., -6.5172e+00,
-1.0307e-01, 1.0753e+00],
[-3.6735e+00, -1.8996e-01, -1.6293e+00, ..., -9.0801e-01,
-1.2654e+00, 1.4784e+00],
[-4.7749e+00, -2.6174e+00, -1.7706e+00, ..., -7.0370e-01,
-3.8313e+00, -2.7378e-01]],
[[ 4.3296e+00, 2.2387e+00, 1.5984e+00, ..., -1.1260e+00,
1.9971e+00, 7.3736e-01],
[ 2.3475e+00, -2.9160e-01, -3.2863e+00, ..., 8.4512e-01,
3.5084e+00, 3.9819e+00],
[ 2.0972e+00, -1.4983e-01, -4.4323e+00, ..., 5.7802e+00,
4.0285e+00, 1.2870e+00],
...,
[ 2.5549e+00, 9.1330e-01, 1.5140e+00, ..., 1.9173e+00,
2.4753e+00, 2.0716e+00],
[ 4.2021e+00, 4.4260e+00, 3.4301e+00, ..., -1.6851e+00,
1.9484e+00, 2.8612e+00],
[ 2.1048e+00, 1.0583e+00, -3.5363e-01, ..., -6.2932e-01,
-7.4162e-02, 2.7132e-01]]],
[[[ 6.2801e-01, -2.1207e-02, -2.5568e+00, ..., -7.8726e-01,
-9.3752e-01, 8.1024e-01],
[ 5.4244e-01, -3.4319e+00, -4.6384e+00, ..., 7.3811e-01,
-1.5942e+00, 9.4377e-01],
[ 2.0471e-01, -5.1299e+00, -5.7507e+00, ..., 6.0810e-01,
-2.2331e-01, 2.8043e-01],
...,
[-5.9310e-01, -2.4726e+00, -5.3479e+00, ..., -6.8261e-01,
3.8005e-01, 1.7886e+00],
[-1.9771e+00, -3.9257e+00, -3.9441e+00, ..., 1.5531e+00,
1.8831e+00, 7.3718e-01],
[-7.0584e-01, -1.2637e+00, -2.4101e-01, ..., -8.2306e-01,
-1.2520e+00, -4.8392e-01]],
[[-1.6985e+00, -4.2157e+00, -7.3458e+00, ..., -4.8418e+00,
-4.9268e+00, -2.2970e-01],
[-6.9648e+00, -1.0808e+01, -7.0739e+00, ..., -3.7656e+00,
-8.1053e+00, -1.6487e+00],
[-5.3471e+00, -9.0593e+00, -7.9606e+00, ..., -1.9562e+00,
-7.7664e+00, -3.0741e+00],
...,
[ 6.4072e-01, -2.6125e-01, -2.1801e+00, ..., -4.7767e-01,
-9.1994e-01, 1.0099e+00],
[-3.4320e+00, -1.9526e+00, 7.5948e-01, ..., -8.4240e-01,
-4.4689e+00, -3.6060e+00],
[-7.3386e-01, -1.6572e+00, 1.6970e+00, ..., -2.3846e+00,
-2.5385e+00, -6.9176e-02]],
[[-3.8425e-01, 1.0417e+00, 7.0178e-01, ..., 1.7950e+00,
1.7962e+00, 9.8382e-01],
[ 3.8082e-01, -5.1469e-01, 8.1827e-01, ..., 1.9106e+00,
1.2623e+00, -8.1159e-01],
[ 7.6561e-01, 1.7096e+00, 2.5338e+00, ..., 8.5379e-01,
2.2806e+00, -7.6537e-02],
...,
[ 1.5642e-01, 8.8354e-01, -2.7106e-01, ..., 1.0825e-01,
1.4282e+00, 9.4966e-02],
[-2.2111e+00, -1.7797e-01, 2.9599e-01, ..., 4.0448e-01,
3.6949e-01, 5.1640e-01],
[-1.7544e+00, -1.9836e+00, -8.1327e-01, ..., 5.2788e-01,
-6.5880e-03, -2.4743e-01]],
...,
[[-9.6822e-01, -3.9237e+00, -5.2803e+00, ..., -2.4225e+00,
-3.1360e+00, -3.5647e-01],
[-4.5045e+00, -7.5475e+00, -4.6534e+00, ..., -2.1585e+00,
-3.6226e+00, -2.5951e+00],
[-2.6398e+00, -5.8703e+00, -3.6057e+00, ..., -3.4627e+00,
-3.6998e+00, -1.9630e+00],
...,
[-3.9153e-01, -2.6405e+00, -1.7050e+00, ..., -1.8389e+00,
-1.7916e+00, -9.5305e-01],
[-2.2702e+00, -2.8933e+00, -7.2025e-01, ..., -1.1898e+00,
-2.1297e+00, -1.7711e+00],
[-1.2076e+00, -1.0354e+00, 1.5089e+00, ..., -2.2234e+00,
-1.1567e+00, 1.1752e+00]],
[[-1.7340e+00, -2.3282e+00, -3.3773e+00, ..., -3.6383e-01,
-1.2194e+00, -1.8510e+00],
[-6.4709e+00, -4.5927e+00, -2.6959e+00, ..., -2.6042e+00,
-1.5347e+00, -9.8322e-01],
[-4.5953e+00, -3.2184e+00, -1.1616e+00, ..., -4.0280e-01,
-1.1788e-01, -4.1484e-01],
...,
[-1.2686e+00, -6.1304e+00, -8.1016e+00, ..., -1.8380e+00,
-8.3616e-01, -1.5756e-02],
[-1.5336e+00, -3.7610e+00, -7.3813e+00, ..., -4.7696e-01,
-8.3205e-01, -2.0302e-01],
[-4.0129e-01, 6.3801e-01, -1.4117e+00, ..., -9.3551e-01,
-2.5233e-01, 5.3276e-01]],
[[ 1.9227e+00, 7.1879e-01, 1.2781e+00, ..., 1.3980e+00,
-2.2813e-01, -8.1113e-01],
[ 4.9015e+00, 3.7720e+00, 1.8045e+00, ..., 3.9764e+00,
3.7670e+00, 2.4622e+00],
[ 2.7429e+00, 2.1470e+00, 1.4383e+00, ..., 7.3302e-01,
2.5630e+00, 1.2602e+00],
...,
[ 2.3841e-01, -2.7031e+00, -5.2968e+00, ..., -2.3697e+00,
-1.7386e-01, 1.3842e+00],
[ 1.3991e+00, -2.5768e+00, -5.1696e+00, ..., -9.7319e-01,
7.2498e-01, 1.5785e+00],
[ 1.9288e+00, 3.0871e-01, -1.5427e+00, ..., 1.1054e+00,
7.0025e-01, 7.1990e-01]]],
[[[-1.2796e-01, 7.5617e-01, 7.6303e-01, ..., -1.3719e+00,
-1.0825e+00, -7.9090e-01],
[ 6.9679e-01, 1.1574e+00, 2.3116e+00, ..., -8.7757e-01,
1.8881e-01, -8.8611e-01],
[ 1.0355e+00, 4.2382e-01, 9.8854e-01, ..., -1.3624e+00,
-4.1480e-01, -1.7145e+00],
...,
[-1.2349e-02, -1.5086e+00, -7.0963e+00, ..., -1.5389e+00,
-2.4979e+00, -1.2810e-01],
[-2.0037e+00, -1.9270e+00, -4.5083e+00, ..., -2.4778e+00,
-2.7522e+00, 1.2493e-01],
[-2.5416e+00, -4.1603e+00, -4.0286e+00, ..., -4.1749e+00,
-3.4413e+00, -1.5361e+00]],
[[-7.2029e-01, -1.0574e+00, -1.5302e+00, ..., -2.8049e+00,
-4.5450e+00, -2.3338e+00],
[-1.9236e+00, -3.3374e+00, -3.0482e+00, ..., -3.4992e+00,
-4.5725e+00, -4.3663e+00],
[-1.6542e+00, -3.6445e+00, -2.8537e+00, ..., -3.2187e+00,
-3.6369e+00, -3.5605e+00],
...,
[-7.1037e+00, -7.3708e+00, -9.9320e+00, ..., -7.8514e+00,
-8.5382e+00, -1.9066e+00],
[-1.0180e+01, -9.8659e+00, -1.0966e+01, ..., -1.0469e+01,
-1.0716e+01, -3.8343e+00],
[-2.6900e+00, -4.3336e+00, -4.6203e+00, ..., -4.2385e+00,
-4.1279e+00, -6.3499e-01]],
[[ 6.8908e-01, 9.5461e-01, 9.3433e-01, ..., 1.5510e+00,
2.0299e+00, 8.8968e-01],
[ 8.3563e-01, 1.2572e+00, 2.0322e-01, ..., 9.7754e-01,
2.2311e+00, 1.0506e+00],
[ 4.6442e-01, 3.0640e+00, 7.7668e-01, ..., 5.5832e-01,
2.3931e+00, 2.1299e+00],
...,
[ 3.2796e-01, -1.2029e+00, 6.0907e-01, ..., 1.7056e+00,
1.3554e+00, -7.1900e-02],
[-5.5175e-01, -4.6243e-01, -1.4533e+00, ..., 4.0527e-01,
1.1504e+00, -2.5443e-01],
[ 7.9822e-01, 1.1422e-01, -9.3005e-01, ..., -3.1252e-01,
-5.7326e-01, -1.8485e+00]],
...,
[[-1.0205e+00, -3.5923e-01, -6.9136e-01, ..., -2.5619e+00,
-3.3608e+00, -1.2472e+00],
[-2.3250e+00, -2.0191e+00, -8.5138e-01, ..., -3.2520e+00,
-4.2809e+00, -3.3799e+00],
[-9.9895e-01, -1.3408e+00, -4.0805e-01, ..., -2.9090e+00,
-4.1425e+00, -2.9728e+00],
...,
[-2.3080e+00, -3.9095e+00, -3.2301e+00, ..., -5.0343e+00,
-6.1018e+00, -2.5712e+00],
[-3.4732e+00, -5.5864e+00, -3.4468e+00, ..., -4.0776e+00,
-5.2567e+00, -1.3355e+00],
[-3.1921e+00, -3.6573e+00, -3.3899e+00, ..., -2.7718e+00,
-2.5270e+00, 4.0304e-01]],
[[-1.5531e+00, -1.4324e+00, -3.7891e+00, ..., -3.3858e+00,
-2.8180e+00, -4.8716e-01],
[-4.3487e+00, -3.4871e+00, -5.7772e+00, ..., -4.5913e+00,
-5.7799e+00, -1.9493e+00],
[-4.0304e+00, -4.6528e+00, -6.7129e+00, ..., -3.8738e+00,
-4.9720e+00, -2.1217e+00],
...,
[-3.7409e+00, -4.1037e+00, -2.2039e+00, ..., -4.3361e+00,
-3.0008e+00, -1.2486e+00],
[-3.1088e+00, -1.1387e+00, -1.1780e+00, ..., -2.4617e+00,
-1.3066e+00, 3.1583e-01],
[-3.5904e+00, -2.5054e+00, -1.4091e+00, ..., -1.6618e+00,
-1.1656e+00, 3.0107e-01]],
[[ 8.8308e-01, -4.1905e-01, 1.9873e-01, ..., 1.2105e-01,
1.2777e+00, 6.1459e-01],
[ 1.7675e+00, 9.0719e-01, 7.4502e-01, ..., 3.8984e-01,
1.8249e+00, 3.2934e+00],
[ 1.3184e+00, 5.8132e-01, 1.1726e+00, ..., -4.1405e-01,
9.8500e-01, 9.9180e-01],
...,
[ 4.2798e+00, 3.7407e+00, 2.0207e+00, ..., 2.9389e+00,
2.7210e+00, 5.8530e-01],
[ 4.5377e+00, 4.3687e+00, 3.3142e+00, ..., 2.7867e+00,
1.8565e+00, 1.4871e+00],
[ 2.6860e+00, 2.7974e+00, 1.4035e+00, ..., 5.6210e-01,
6.2510e-01, 1.0540e+00]]],
[[[-9.6181e-03, 1.1662e+00, 8.6530e-01, ..., -3.7784e-01,
2.0003e-01, 7.9774e-01],
[ 7.3134e-01, 9.8288e-01, 1.5265e+00, ..., -1.1503e+00,
-1.1135e+00, 9.8146e-01],
[ 7.3008e-01, 9.2803e-01, -4.9579e-02, ..., -1.1967e+00,
-1.3250e+00, 1.9422e-01],
...,
[-1.8602e+00, -3.3094e+00, -7.4714e+00, ..., -8.0082e-01,
-1.1697e+00, -1.1120e+00],
[-2.6451e+00, -6.5732e+00, -8.6202e+00, ..., -1.3407e+00,
-3.0168e+00, -1.1597e+00],
[-1.8548e+00, -3.5291e+00, -4.0694e+00, ..., -7.4315e-01,
-2.4334e+00, -9.1012e-01]],
[[-8.0396e-01, -2.2733e+00, -3.9965e+00, ..., -3.0546e+00,
-4.9044e+00, -2.9927e-01],
[-4.5253e+00, -6.7109e+00, -7.3325e+00, ..., -5.4920e+00,
-7.4364e+00, -1.8024e+00],
[-3.6008e+00, -6.0091e+00, -1.1621e+01, ..., -9.8759e+00,
-9.7665e+00, -3.6890e+00],
...,
[ 1.7862e+00, -9.7953e-01, -3.7638e+00, ..., -4.8226e+00,
-5.4961e+00, -2.1842e+00],
[ 1.5675e-01, -3.4590e+00, -5.8930e+00, ..., -8.1102e+00,
-9.5872e+00, -3.9591e+00],
[ 1.1744e+00, 3.2216e+00, 8.1518e-01, ..., -1.8098e+00,
-3.1001e+00, -2.2978e-01]],
[[-3.8078e-01, 2.1787e+00, 1.1567e+00, ..., 8.7612e-01,
2.0814e+00, 8.9535e-01],
[-1.3765e-01, 1.4812e+00, 3.1594e-01, ..., 4.4501e-01,
1.0156e+00, -6.1140e-01],
[-1.6054e-01, 1.4789e+00, 8.3690e-01, ..., -2.5597e-01,
1.4249e+00, -7.8622e-02],
...,
[-1.4333e-01, -1.7107e+00, -3.1711e-01, ..., 1.0527e+00,
1.0978e+00, 6.2759e-01],
[-3.9318e-01, 2.2180e+00, 2.0866e+00, ..., 3.2889e+00,
2.1388e+00, 5.0255e-01],
[-2.5584e+00, 1.1068e+00, 1.7051e+00, ..., 1.3771e+00,
1.0396e+00, -9.6744e-01]],
...,
[[-7.2148e-01, -1.6672e+00, -1.4208e+00, ..., -2.2702e+00,
-2.7106e+00, -4.0541e-01],
[-2.8378e+00, -4.4456e+00, -2.4897e+00, ..., -4.4824e+00,
-4.1147e+00, -2.5716e+00],
[-1.0228e+00, -3.7801e+00, -2.6539e+00, ..., -3.1541e+00,
-5.7555e+00, -1.8642e+00],
...,
[-4.7530e-02, -1.9131e+00, -2.5388e+00, ..., -2.4677e+00,
-4.2416e+00, -1.8468e+00],
[-8.9937e-01, -5.6917e+00, -4.8589e+00, ..., -2.9776e+00,
-4.9801e+00, -1.4101e+00],
[ 1.2131e+00, 1.3126e+00, 1.6891e+00, ..., -5.4785e-01,
-8.3782e-01, 1.1911e+00]],
[[-1.6640e+00, -2.0879e+00, -2.3816e+00, ..., -1.7018e+00,
-1.3864e+00, -1.3916e+00],
[-4.0094e+00, -3.7458e+00, -2.8712e+00, ..., -3.5996e+00,
-1.8090e+00, -6.9163e-01],
[-2.1586e+00, -2.8398e+00, -4.3050e+00, ..., 2.0828e-01,
-7.2397e-01, -7.0103e-01],
...,
[-4.5475e+00, -7.8253e+00, -4.6412e+00, ..., -3.3320e-02,
-1.9732e+00, -1.0268e+00],
[-6.1792e+00, -1.4107e+01, -1.1619e+01, ..., -3.7230e-01,
-1.7127e+00, -3.9400e-01],
[ 2.5075e-01, -4.2234e+00, -5.1906e+00, ..., -3.4449e-01,
-2.7542e-01, 2.0899e-01]],
[[ 9.1802e-01, 4.9321e-01, 7.7021e-01, ..., -5.7240e-01,
1.1818e-01, -4.7340e-01],
[ 3.4524e+00, 3.3190e+00, 3.0850e+00, ..., 1.6157e+00,
1.9629e+00, 2.3858e+00],
[ 1.7063e+00, 2.0500e+00, 2.5961e+00, ..., 2.7810e+00,
2.0309e+00, 1.7084e+00],
...,
[-7.6664e-01, -4.8773e+00, -9.5807e+00, ..., 1.4159e+00,
9.5607e-01, 5.8201e-01],
[-1.0634e+00, -6.9301e+00, -7.7515e+00, ..., 3.3276e+00,
1.3100e+00, 1.8136e+00],
[ 1.5739e-01, -3.1270e+00, -5.4925e+00, ..., 9.8864e-01,
-4.0171e-02, 9.9349e-01]]]], device='cuda:0'),
tensor([[[[ 5.0467e-01, 1.4180e+00, 5.5404e-01, ..., 5.0839e-01,
-4.4849e-01, 6.4101e-01],
[ 3.4034e-01, -4.6852e-01, 1.5568e+00, ..., -3.4495e-01,
-1.1645e+00, -2.7574e-01],
[ 1.2635e-01, -1.5180e-01, 1.8398e+00, ..., 3.0655e-01,
-3.7298e-01, -6.4105e-01],
...,
[-1.2609e+00, -2.3684e+00, -3.3462e+00, ..., -3.3127e+00,
-3.3891e+00, -7.6279e-01],
[-2.4523e+00, -7.2148e-01, 1.6272e-03, ..., -2.5843e+00,
-2.8554e+00, 5.6493e-02],
[-2.3918e+00, -3.6368e+00, -2.7723e+00, ..., -4.0869e+00,
-3.5426e+00, -1.5532e+00]],
[[-1.3705e+00, -1.9124e+00, -2.2764e+00, ..., -2.1317e+00,
-5.1393e+00, -1.0949e+00],
[-4.9414e+00, -6.4434e+00, -4.2140e+00, ..., -6.0194e+00,
-7.0200e+00, -3.8670e+00],
[-4.1876e+00, -5.2355e+00, -3.8066e+00, ..., -6.2102e+00,
-5.8839e+00, -4.1191e+00],
...,
[-7.5428e+00, -8.8265e+00, -8.9991e+00, ..., -7.8375e+00,
-9.1608e+00, -2.2260e+00],
[-1.0003e+01, -1.0039e+01, -8.2269e+00, ..., -1.0846e+01,
-1.0964e+01, -3.8861e+00],
[-2.5232e+00, -4.9259e+00, -4.7225e+00, ..., -4.4148e+00,
-4.1926e+00, -6.4640e-01]],
[[-4.5164e-02, 9.0920e-01, 5.3568e-01, ..., 1.4095e+00,
1.8323e+00, 3.2229e-01],
[ 2.0016e-01, 1.9274e-01, -5.6453e-01, ..., 7.0758e-01,
2.0078e+00, -5.9220e-01],
[ 7.4782e-01, 2.7939e+00, 6.9735e-01, ..., 1.3010e+00,
2.1521e+00, 9.0766e-01],
...,
[ 3.2310e-01, 8.8943e-02, -1.2387e-01, ..., 1.0031e+00,
6.4252e-01, -9.3721e-01],
[-5.5393e-01, -3.9005e-01, -1.0282e+00, ..., 2.0741e-01,
5.8522e-01, -5.5267e-01],
[ 5.6505e-01, 9.6696e-01, 1.5379e-01, ..., 1.1121e-01,
-5.2290e-01, -1.8844e+00]],
...,
[[-9.3968e-01, -7.6306e-01, -1.5902e+00, ..., -1.5724e+00,
-2.7139e+00, 1.3059e-01],
[-4.1800e+00, -4.4381e+00, -2.7530e+00, ..., -3.7398e+00,
-5.7994e+00, -3.0387e+00],
[-2.4252e+00, -1.2155e+00, -5.0684e-01, ..., -1.1048e+00,
-4.7291e+00, -1.8565e+00],
...,
[-2.8307e+00, -4.3321e+00, -4.1157e+00, ..., -4.7301e+00,
-6.0126e+00, -2.9759e+00],
[-3.5048e+00, -5.1164e+00, -3.5446e+00, ..., -4.8154e+00,
-5.2883e+00, -1.4041e+00],
[-3.2729e+00, -3.8896e+00, -3.0896e+00, ..., -2.8948e+00,
-2.5765e+00, 4.0905e-01]],
[[-1.0798e+00, 1.7477e-01, -2.4318e+00, ..., -6.9193e-01,
1.9649e-02, 1.4446e-01],
[-4.0812e+00, -1.2089e+00, -2.7254e+00, ..., -1.6887e+00,
-2.2457e+00, 1.7719e-01],
[-3.7506e+00, -1.6977e+00, -2.8326e+00, ..., -7.2467e-01,
-2.7989e+00, -7.0115e-01],
...,
[-3.8610e+00, -3.6906e+00, -4.3592e+00, ..., -2.4846e+00,
-2.4522e+00, -8.3550e-01],
[-3.1473e+00, -2.9967e-01, -2.3832e+00, ..., -1.4813e+00,
-9.9012e-01, 3.4191e-01],
[-3.6367e+00, -2.7516e+00, -2.7115e+00, ..., -1.4810e+00,
-1.1875e+00, 3.0944e-01]],
[[ 1.6468e+00, 7.1185e-01, 8.2880e-01, ..., 1.0822e+00,
1.5205e+00, 1.5719e-01],
[ 3.1937e+00, 2.0501e+00, 1.9521e+00, ..., 1.8696e+00,
2.6175e+00, 3.4997e+00],
[ 2.2164e+00, 2.2306e-01, 8.4243e-01, ..., 2.0880e+00,
2.0772e+00, 1.9954e+00],
...,
[ 2.7904e+00, 2.5478e+00, 2.3366e+00, ..., 2.1235e+00,
1.3482e+00, 1.4208e-01],
[ 4.6558e+00, 5.1770e+00, 4.0370e+00, ..., 2.8478e+00,
1.8952e+00, 1.5352e+00],
[ 2.6301e+00, 3.0818e+00, 1.9501e+00, ..., 2.9426e-01,
5.7494e-01, 9.9790e-01]]],
[[[ 8.9389e-02, 9.3770e-01, 3.1282e-02, ..., -1.1690e+00,
-9.4132e-01, -7.9489e-01],
[ 1.8800e+00, 7.1411e-01, 8.3256e-01, ..., -1.9615e-01,
3.4417e-01, -9.4557e-01],
[-5.4883e-01, -4.3296e+00, -6.4384e+00, ..., -5.8255e-02,
-5.3280e-02, -1.7833e+00],
...,
[-1.1448e+00, -6.3120e-01, -3.6318e+00, ..., -2.9250e+00,
-2.8940e+00, 1.1994e+00],
[-2.3547e+00, -3.9693e+00, -6.3247e-01, ..., -1.8427e+00,
-1.5949e+00, 1.1169e+00],
[-1.3495e+00, -4.8264e+00, -1.0078e+00, ..., -3.5030e+00,
-2.9563e+00, -1.2937e+00]],
[[-7.2391e-01, -1.6058e+00, -3.0953e+00, ..., -2.7959e+00,
-4.5054e+00, -2.3510e+00],
[-2.3875e+00, -4.2006e+00, -5.1815e+00, ..., -3.5249e+00,
-5.0142e+00, -4.7729e+00],
[-3.9675e+00, -7.9291e+00, -7.4472e+00, ..., -3.0669e+00,
-4.0626e+00, -3.8776e+00],
...,
[-5.7660e+00, -7.2237e+00, -7.2494e+00, ..., -5.6938e+00,
-6.6549e+00, -5.7768e-01],
[-7.3197e+00, -7.7425e+00, 1.0936e+00, ..., -7.2289e+00,
-8.5716e+00, -3.4145e+00],
[-9.2543e-01, -3.9351e+00, -5.8822e-01, ..., -3.1892e+00,
-3.7761e+00, -7.5258e-01]],
[[ 6.9622e-01, 1.2439e+00, 1.1450e+00, ..., 1.4770e+00,
1.8133e+00, 8.7148e-01],
[ 2.5307e-01, 1.1023e+00, 1.6073e+00, ..., 1.2895e+00,
2.2669e+00, 8.3623e-01],
[ 1.5695e-01, 3.9994e+00, 5.6426e+00, ..., 1.4340e+00,
2.4572e+00, 2.0207e+00],
...,
[-5.3296e-01, 2.3830e-01, 5.2587e-01, ..., -3.1780e-01,
-7.1779e-01, -2.1579e+00],
[-4.0301e-01, 4.1839e-01, -1.0962e+00, ..., -1.6828e+00,
-1.5385e+00, -1.5646e+00],
[-7.1280e-02, 1.2720e+00, -1.7536e+00, ..., -8.2595e-01,
-7.3678e-01, -1.8768e+00]],
...,
[[-8.2248e-01, -1.0268e+00, -2.0278e+00, ..., -2.4459e+00,
-3.2497e+00, -1.1640e+00],
[-2.2402e+00, -3.3307e+00, -2.5136e+00, ..., -2.9728e+00,
-4.3635e+00, -3.4669e+00],
[-2.2744e+00, -2.8158e+00, -3.6093e+00, ..., -1.8123e+00,
-3.5834e+00, -2.8104e+00],
...,
[-1.3479e+00, -1.9973e+00, -2.6018e+00, ..., -1.4809e+00,
-2.5314e+00, -8.7506e-01],
[-2.2480e+00, -2.7921e+00, 2.7028e+00, ..., -7.5256e-01,
-2.8055e+00, -6.3860e-01],
[-1.5179e+00, -2.1016e+00, -5.3377e-01, ..., -1.1528e+00,
-1.9817e+00, 6.0785e-01]],
[[-1.5910e+00, -2.1915e+00, -3.1019e+00, ..., -2.8402e+00,
-2.7991e+00, -5.6530e-01],
[-4.3669e+00, -4.6807e+00, -4.1231e+00, ..., -4.0895e+00,
-5.6527e+00, -1.7708e+00],
[-4.2666e+00, -7.3413e+00, -7.8728e+00, ..., -2.3681e+00,
-4.1729e+00, -1.7585e+00],
...,
[-2.6908e+00, -2.3386e+00, -3.5025e+00, ..., -1.1388e+00,
-4.9137e-01, 9.7387e-02],
[-2.4483e+00, 1.7653e+00, 1.2823e+00, ..., 6.0390e-01,
4.4616e-01, 1.5743e+00],
[-2.8483e+00, -9.0735e-01, -2.8178e+00, ..., -4.7998e-01,
-3.7258e-01, 7.0199e-01]],
[[ 9.1537e-01, 7.2995e-02, 8.5716e-01, ..., 5.1032e-01,
1.3457e+00, 5.6950e-01],
[ 2.3629e+00, 9.2008e-01, 1.3483e+00, ..., 9.5439e-01,
1.9163e+00, 3.3747e+00],
[ 1.5042e+00, -6.6240e-01, -1.5299e+00, ..., 7.9577e-01,
1.2223e+00, 1.0758e+00],
...,
[ 2.5299e+00, 2.3236e+00, -1.1623e-01, ..., 1.1308e+00,
1.7626e+00, 1.1998e+00],
[ 3.6298e+00, 2.2356e+00, 5.7057e-01, ..., 2.1744e+00,
1.7138e+00, 2.0910e+00],
[ 2.3002e+00, 4.3108e-01, 7.3604e-01, ..., -2.4873e-01,
-1.5019e-01, 9.9810e-01]]],
[[[ 8.2926e-01, 1.5852e+00, 6.4450e-01, ..., -4.0297e+00,
-2.6943e+00, -1.3124e+00],
[ 2.1207e+00, 1.2163e+00, 8.6408e-01, ..., -3.3645e+00,
-2.1797e+00, -1.5389e-02],
[ 7.0969e-01, 3.0115e-01, 9.8449e-01, ..., -3.4311e+00,
-3.1794e+00, 2.0633e-01],
...,
[ 8.5339e-01, -1.7939e-01, -3.3007e-01, ..., -4.7882e+00,
-3.7164e+00, 1.9952e-01],
[ 9.4728e-02, -1.3279e-01, -9.3275e-01, ..., -4.3563e+00,
-4.2280e+00, 3.7729e-01],
[-3.6874e-01, -3.5587e+00, -3.3571e+00, ..., -4.1828e+00,
-3.3323e+00, -1.3230e+00]],
[[ 3.1714e-01, -1.1973e+00, -2.6740e+00, ..., -7.9194e+00,
-7.7610e+00, -1.4811e+00],
[-1.1426e+00, -3.8541e+00, -4.6557e+00, ..., -7.6395e+00,
-7.5265e+00, -2.0193e+00],
[ 3.2471e-02, -5.8011e-01, -2.1603e+00, ..., -8.8877e+00,
-7.7599e+00, -2.2678e+00],
...,
[-1.9054e+00, -2.8910e+00, -2.7949e+00, ..., -1.0993e+01,
-8.7861e+00, -1.9287e+00],
[-4.3521e+00, -5.8607e+00, -5.3389e+00, ..., -1.0280e+01,
-1.0759e+01, -4.2160e+00],
[-3.1428e-01, -3.3245e+00, -3.6111e+00, ..., -3.8889e+00,
-3.6127e+00, -7.6767e-01]],
[[ 4.7472e-01, 2.1528e+00, 1.9332e+00, ..., 6.6498e-01,
2.4310e+00, 1.1587e+00],
[-1.4506e-01, 9.8575e-01, 5.4509e-01, ..., 6.0904e-01,
2.6539e+00, 3.4970e-01],
[ 5.1580e-01, 3.6723e+00, 2.0955e+00, ..., -6.4460e-01,
1.7804e+00, -2.8265e-01],
...,
[ 6.3849e-01, 2.8809e+00, 1.1084e+00, ..., 2.7766e-01,
4.1190e-01, -1.1495e+00],
[ 6.9425e-01, 1.7569e+00, 4.5992e-01, ..., -8.1431e-01,
6.8979e-01, -1.3450e+00],
[-3.2036e-01, 1.8211e+00, 9.5854e-01, ..., 1.0471e-01,
1.4759e-01, -1.6703e+00]],
...,
[[-3.1788e-01, -8.8329e-01, -1.5692e+00, ..., -5.6772e+00,
-5.7515e+00, -9.9063e-01],
[-2.4395e+00, -4.0816e+00, -2.1102e+00, ..., -4.3165e+00,
-4.6849e+00, -1.6663e+00],
[-7.3503e-01, -1.3443e+00, -8.3975e-01, ..., -6.0180e+00,
-6.0393e+00, -1.1326e+00],
...,
[-7.9962e-01, -1.0081e+00, 2.2798e-01, ..., -5.4281e+00,
-4.8137e+00, -2.0510e+00],
[-1.8994e+00, -3.5660e+00, -1.9233e+00, ..., -3.7590e+00,
-5.6060e+00, -1.2845e+00],
[-1.1551e+00, -3.7198e+00, -2.5484e+00, ..., -1.6594e+00,
-2.1522e+00, 3.3249e-01]],
[[-1.0494e+00, -7.7843e-01, -9.4919e-01, ..., -2.4298e+00,
-1.1039e+00, -6.4335e-01],
[-2.8075e+00, -2.8833e+00, -1.6661e+00, ..., -3.5441e+00,
-1.8244e+00, -4.7962e-01],
[-2.1553e+00, -3.5620e+00, -2.3968e-01, ..., -1.7951e+00,
-1.8871e+00, -7.7349e-01],
...,
[-2.3054e+00, -3.6175e+00, -2.2440e+00, ..., -3.2795e+00,
-7.9064e-01, -3.0737e-01],
[-3.0307e+00, -1.9748e+00, -7.4850e-01, ..., -1.3220e+00,
-3.8780e-01, 1.3383e+00],
[-3.3156e+00, -3.2845e+00, -1.2590e+00, ..., 2.2028e-01,
-5.3134e-01, 7.6904e-01]],
[[ 9.1196e-01, 8.5622e-01, 1.0761e+00, ..., -4.8790e-01,
7.8854e-01, -1.2623e+00],
[ 1.6400e+00, 2.1333e+00, 2.0572e+00, ..., 3.0328e+00,
3.0658e+00, 2.7089e+00],
[ 1.3512e-01, -5.4931e-01, 8.7063e-01, ..., 2.1295e+00,
1.5841e+00, 1.5294e+00],
...,
[ 1.5507e+00, 2.8955e-02, 1.2468e-01, ..., 2.0880e+00,
2.8096e+00, 7.6044e-01],
[ 3.0864e+00, 3.5720e+00, 2.7693e+00, ..., 1.4118e+00,
1.1934e+00, 1.8198e+00],
[ 1.9219e+00, 4.6377e-01, 5.3953e-01, ..., 1.4134e-01,
-3.6846e-01, 7.5693e-01]]],
[[[ 2.6454e-01, -3.2083e-01, 8.4083e-01, ..., 5.6022e+00,
3.1215e+00, 1.1471e+00],
[-8.1212e-01, -3.2293e+00, 2.2065e-03, ..., 8.0568e+00,
1.2687e+00, -1.2214e+00],
[-2.0725e+00, -1.4739e+00, 2.0724e+00, ..., 7.8296e+00,
1.4522e+00, -1.7622e+00],
...,
[-5.7982e-01, 7.3899e-01, -7.2987e+00, ..., -1.0368e+00,
-9.7900e-01, -1.4522e+00],
[-2.7800e+00, -4.3691e+00, -1.3355e+01, ..., -1.8737e+00,
-9.0299e-01, -1.5981e+00],
[-1.7796e+00, -4.4293e+00, -6.2352e+00, ..., -3.6000e+00,
-3.5957e+00, -2.9848e+00]],
[[-2.5109e+00, -4.7641e+00, -4.3069e+00, ..., 5.9543e+00,
-5.0411e+00, -1.2795e+01],
[-6.5053e+00, -7.3703e+00, -4.3793e+00, ..., 4.7260e+00,
-4.0612e+00, -1.7264e+01],
[-3.6506e+00, -1.7807e+00, -9.5051e-01, ..., 2.7374e+00,
6.1863e-01, -1.4368e+01],
...,
[ 2.7376e+00, 3.4602e+00, -5.1930e-01, ..., -6.0864e+00,
-7.0251e+00, -3.5348e+00],
[-5.8965e-01, -1.8564e+00, -6.4997e+00, ..., -7.6292e+00,
-6.7500e+00, -5.5902e+00],
[ 2.9112e-01, -7.3958e-02, -4.7248e-01, ..., -4.1986e+00,
-3.3600e+00, -2.0316e+00]],
[[ 1.8109e-01, 1.9713e+00, 8.1472e-01, ..., 9.8988e-01,
5.4589e+00, -1.1688e+00],
[ 5.7670e-01, 2.3694e+00, 1.1687e+00, ..., -9.8429e+00,
5.4368e+00, -3.6648e-01],
[ 1.8480e+00, 1.0374e+00, 1.3737e+00, ..., -6.7080e+00,
6.8599e+00, 1.2749e+00],
...,
[-1.6629e+00, 2.8307e-01, -1.5450e+00, ..., 3.2690e+00,
3.2202e-01, 9.9795e-01],
[ 1.4485e+00, -2.1499e+00, -6.0326e+00, ..., 1.8383e+00,
-3.3515e-01, 5.4817e-01],
[ 3.0111e-02, -2.0748e+00, -4.1445e+00, ..., 2.0683e-01,
8.3763e-01, 4.5453e-01]],
...,
[[-2.1918e+00, -3.3152e+00, -1.3868e+00, ..., 5.0243e+00,
-7.1642e+00, 2.1989e+00],
[-4.1355e+00, -4.4016e+00, -1.4340e+00, ..., 5.0297e+00,
-1.2574e+01, 4.4452e-01],
[-2.9538e+00, -6.5790e-01, -4.9791e-02, ..., 7.6676e+00,
-1.0427e+01, 5.3140e-03],
...,
[ 2.6193e+00, 3.5158e+00, 1.6665e+00, ..., -6.6925e+00,
-3.9493e+00, -3.2614e+00],
[-7.2647e-01, -4.5566e+00, -4.0894e+00, ..., -3.0868e+00,
-4.1935e+00, -2.8454e+00],
[-8.2386e-01, -3.6467e+00, -1.7147e+00, ..., -2.2830e+00,
-3.3947e+00, -1.2125e+00]],
[[-2.2337e+00, -1.3876e+00, -8.1791e-01, ..., -2.6739e+00,
-3.7729e+00, 3.5518e+00],
[-5.5842e+00, -4.0319e+00, -2.1832e+00, ..., 2.6313e+00,
-8.1017e+00, 2.3601e+00],
[-3.4281e+00, -1.4329e+00, -1.3293e+00, ..., 6.0638e+00,
-8.9409e+00, -1.2341e+00],
...,
[-2.2122e+00, -1.0338e+01, -1.8956e+01, ..., -7.4204e+00,
-5.4056e+00, -1.6782e+00],
[-5.2965e+00, -1.5035e+01, -1.9146e+01, ..., -2.4273e+00,
-3.5899e+00, -9.9059e-01],
[-4.4371e+00, -9.4439e+00, -6.3530e+00, ..., -9.0289e-01,
-2.2591e+00, -7.3729e-01]],
[[ 2.1190e+00, 2.1228e-01, 9.0142e-01, ..., 2.0819e+00,
5.7346e+00, 1.3952e+00],
[ 3.5613e+00, 1.0816e+00, 2.9398e+00, ..., 3.9020e+00,
4.4643e+00, 2.3544e+00],
[ 1.4937e+00, 5.2879e-01, 2.0980e+00, ..., 8.0375e+00,
4.7382e+00, -4.4243e-01],
...,
[-2.5140e+00, -5.3090e+00, -1.3059e+01, ..., 1.8536e+00,
1.0933e+00, 1.3213e+00],
[-1.9763e+00, -6.7890e+00, -1.1381e+01, ..., 2.3192e+00,
1.5730e+00, 1.3134e+00],
[-1.3498e+00, -3.0885e+00, -4.3656e+00, ..., 6.8980e-01,
-1.8666e-01, -4.6106e-01]]],
[[[ 3.2647e-01, 1.3592e+00, 1.2687e-01, ..., -1.5300e+00,
-1.9091e+00, -1.8866e+00],
[ 5.5958e-01, 9.5788e-01, 1.6203e+00, ..., -6.3283e+00,
-5.5735e+00, -2.9147e+00],
[-3.1188e-02, -1.2981e+00, -1.4762e+00, ..., -1.2151e+00,
-7.5957e-01, 7.8907e-01],
...,
[-4.1028e-01, -9.0520e-01, -1.4511e-01, ..., 2.9310e+00,
-2.0760e+00, 1.4346e-01],
[-2.0991e+00, -1.5776e+00, -8.9061e-01, ..., -2.4840e+00,
-3.8284e+00, 3.6092e-01],
[-1.1602e+00, -3.3793e+00, -3.4598e+00, ..., -3.4361e+00,
-3.3318e+00, -1.3897e+00]],
[[-9.6046e-01, -1.4931e+00, -3.5458e+00, ..., -8.0328e+00,
-8.5783e+00, -3.5642e+00],
[-4.3421e+00, -5.1463e+00, -4.9677e+00, ..., -9.0440e+00,
-9.9283e+00, -3.0854e+00],
[-4.5150e+00, -5.5618e+00, -3.2722e+00, ..., -8.2043e-01,
-1.4951e+00, -7.4025e-01],
...,
[-5.7907e+00, -6.7525e+00, -5.3186e+00, ..., -4.6017e+00,
-9.1203e+00, -2.5804e+00],
[-7.7662e+00, -7.7839e+00, -7.9389e+00, ..., -1.0805e+01,
-1.1651e+01, -4.0534e+00],
[-1.8082e+00, -3.3699e+00, -3.8466e+00, ..., -2.9706e+00,
-3.5626e+00, -4.7671e-01]],
[[-4.5308e-02, 1.2352e+00, 1.6948e+00, ..., 1.2341e+00,
2.5964e+00, 5.2242e-01],
[ 1.0172e-01, 1.8828e-01, 1.2973e+00, ..., 4.6367e+00,
5.9841e+00, 1.5982e+00],
[ 3.9374e-01, 8.9400e-01, 1.0838e+00, ..., 1.6193e+00,
3.5682e+00, 2.4797e+00],
...,
[ 4.1130e-01, -1.0601e-01, -2.2786e-01, ..., -1.3948e+00,
1.8846e+00, -1.0487e+00],
[ 6.1598e-01, 9.5048e-01, 8.3065e-01, ..., -1.1352e+00,
8.9029e-01, -1.7840e+00],
[ 5.3200e-01, 1.2028e+00, 1.8003e+00, ..., -1.5617e-01,
-1.0633e+00, -2.3018e+00]],
...,
[[-5.1837e-01, -1.5469e+00, -2.1468e+00, ..., -4.3653e+00,
-5.4142e+00, -1.0099e+00],
[-2.7661e+00, -3.0315e+00, -2.4024e+00, ..., -5.6685e+00,
-5.8883e+00, -2.6600e+00],
[-1.9633e+00, -1.8070e+00, -1.6182e+00, ..., 1.1539e+00,
-7.0302e-01, -6.6891e-01],
...,
[-2.1638e+00, -1.6925e+00, 3.3084e-02, ..., -2.7535e+00,
-6.7845e+00, -1.6860e+00],
[-3.4128e+00, -3.3673e+00, -2.4284e+00, ..., -4.1763e+00,
-5.9412e+00, -4.2206e-01],
[-1.5550e+00, -2.0929e+00, -2.3718e+00, ..., -1.1192e+00,
-1.8627e+00, 3.3291e-01]],
[[-1.2369e+00, -2.4501e+00, -3.7771e+00, ..., -1.3500e+00,
-1.0505e+00, -5.0133e-01],
[-3.4526e+00, -3.0215e+00, -3.5766e+00, ..., -8.0819e+00,
-6.6318e+00, -7.5427e-01],
[-4.6900e+00, -3.5357e+00, -4.6872e+00, ..., -3.7938e+00,
-4.2817e+00, -1.7331e+00],
...,
[-4.1997e+00, -4.0080e+00, -4.2755e+00, ..., -2.2420e+00,
-2.9939e+00, -3.7572e-01],
[-4.1624e+00, -2.1556e+00, -2.3402e+00, ..., -1.3027e+00,
-7.0928e-01, 1.7676e+00],
[-3.2354e+00, -3.3745e+00, -2.3469e+00, ..., -4.5059e-01,
-6.0261e-01, 8.4204e-01]],
[[ 1.1680e+00, 9.4347e-03, 1.1115e+00, ..., 1.7350e+00,
1.7089e+00, -6.5047e-01],
[ 2.4814e+00, 1.5737e+00, 2.0160e+00, ..., 2.1453e+00,
3.0745e+00, 3.1594e+00],
[ 2.1943e+00, 1.3756e+00, -2.0876e-01, ..., 3.4345e+00,
1.2832e+00, 2.0019e+00],
...,
[ 3.3497e+00, 2.2105e+00, 1.5433e+00, ..., 3.6208e+00,
1.9358e+00, 6.9795e-01],
[ 2.9243e+00, 4.4559e+00, 3.0235e+00, ..., 2.8752e+00,
8.0812e-01, 2.0950e+00],
[ 2.0207e+00, 1.6682e+00, 5.6918e-01, ..., 2.1561e-01,
5.2315e-01, 1.0736e+00]]]], device='cuda:0'),
tensor([[[[-7.8789e-02, 7.3064e-01, -3.6807e-01, ..., -1.3256e+00,
-8.6015e-01, -8.0933e-01],
[ 6.9031e-01, 2.1778e-01, -2.7287e-01, ..., -5.6191e-01,
1.9525e-01, -1.1494e+00],
[ 7.2554e-01, -1.9087e-01, 7.0295e-01, ..., -1.0463e+00,
-7.6127e-01, -2.1267e+00],
...,
[-2.1301e-01, -1.3153e+00, -1.9499e+00, ..., -4.3314e+00,
-3.3441e+00, -1.0544e-01],
[-2.0289e+00, -1.2668e+00, -2.2292e-02, ..., -2.8898e+00,
-2.4202e+00, 2.7332e-01],
[-2.3676e+00, -4.1049e+00, -3.2215e+00, ..., -4.0714e+00,
-3.5522e+00, -1.5770e+00]],
[[-7.1989e-01, -1.4440e+00, -2.5083e+00, ..., -2.9813e+00,
-4.4744e+00, -2.4343e+00],
[-2.0174e+00, -3.6721e+00, -3.6323e+00, ..., -3.3986e+00,
-4.9723e+00, -4.4975e+00],
[-2.1947e+00, -4.3573e+00, -6.8373e+00, ..., -3.4455e+00,
-4.7136e+00, -3.7757e+00],
...,
[-5.6707e+00, -9.6079e+00, -9.4098e+00, ..., -8.0336e+00,
-8.2767e+00, -1.8172e+00],
[-1.0059e+01, -1.0439e+01, -8.4419e+00, ..., -1.0772e+01,
-1.1003e+01, -3.9625e+00],
[-2.5375e+00, -5.0444e+00, -4.3169e+00, ..., -4.0609e+00,
-4.2763e+00, -7.0003e-01]],
[[ 6.7790e-01, 1.2162e+00, 1.2084e+00, ..., 1.5305e+00,
1.7750e+00, 8.5039e-01],
[ 7.2923e-01, 1.3556e+00, 1.3528e+00, ..., 1.2506e+00,
2.0345e+00, 9.0062e-01],
[ 8.0984e-01, 3.2291e+00, 2.0042e+00, ..., 1.3321e+00,
2.0575e+00, 1.8221e+00],
...,
[ 3.3335e-02, 3.7831e-01, 2.1278e+00, ..., 1.7287e+00,
5.2980e-01, 1.3293e-01],
[-1.1833e+00, 4.3861e-01, 1.1264e-01, ..., 2.7549e-01,
1.4259e+00, -1.9359e-01],
[ 7.9013e-01, 1.2863e+00, 6.9764e-01, ..., -3.4042e-01,
-4.4406e-01, -1.8916e+00]],
...,
[[-9.8318e-01, -1.4375e+00, -2.1634e+00, ..., -2.5298e+00,
-3.1239e+00, -1.2297e+00],
[-2.4300e+00, -3.4717e+00, -2.5464e+00, ..., -2.9142e+00,
-4.2652e+00, -3.6339e+00],
[-1.1609e+00, -2.5375e+00, -1.5653e+00, ..., -2.8733e+00,
-4.4133e+00, -3.2409e+00],
...,
[-3.6401e+00, -6.9634e+00, -5.8789e+00, ..., -6.0347e+00,
-6.7410e+00, -2.9562e+00],
[-4.0436e+00, -4.8903e+00, -3.2081e+00, ..., -5.0150e+00,
-5.5315e+00, -1.2506e+00],
[-3.4581e+00, -3.8961e+00, -3.3221e+00, ..., -2.5748e+00,
-2.7538e+00, 4.0867e-01]],
[[-1.5930e+00, -2.4717e+00, -3.6755e+00, ..., -2.7932e+00,
-2.7647e+00, -5.7152e-01],
[-4.4092e+00, -4.8257e+00, -4.7878e+00, ..., -4.0901e+00,
-5.7387e+00, -2.0191e+00],
[-3.1777e+00, -3.2577e+00, -1.7086e+00, ..., -3.0023e+00,
-4.8853e+00, -2.1950e+00],
...,
[-3.5752e+00, -3.1170e+00, -4.6271e+00, ..., -5.3326e+00,
-3.0642e+00, -9.8706e-01],
[-3.4305e+00, 2.1506e-02, -2.5643e+00, ..., -1.5796e+00,
-7.9672e-01, 4.3922e-01],
[-3.9338e+00, -3.0123e+00, -2.8763e+00, ..., -1.7569e+00,
-1.2866e+00, 3.7383e-01]],
[[ 8.9990e-01, -1.5443e-01, 7.1943e-01, ..., 5.0920e-01,
1.3792e+00, 6.2898e-01],
[ 1.7879e+00, 7.9350e-01, 3.5663e-01, ..., 7.5293e-01,
1.8453e+00, 3.0729e+00],
[ 1.5324e+00, 5.9900e-01, 1.1457e+00, ..., 2.7472e-01,
8.7319e-01, 6.4208e-01],
...,
[ 4.1426e+00, 4.2623e+00, 3.1874e+00, ..., 2.8871e+00,
3.4579e+00, 7.9445e-01],
[ 4.1625e+00, 3.4118e+00, 2.5158e+00, ..., 2.1748e+00,
1.3923e+00, 1.1998e+00],
[ 2.9738e+00, 2.9336e+00, 2.2025e+00, ..., 5.9051e-01,
5.2512e-01, 9.9896e-01]]],
[[[-1.2861e+00, -4.7827e+00, -2.5374e+00, ..., -4.2099e+00,
-1.9788e+00, -8.9812e-01],
[-2.1269e+00, -3.0135e+00, -6.3776e-01, ..., -3.5785e+00,
-2.5291e+00, -1.8295e-01],
[-3.7752e+00, -2.7374e+00, -2.0783e+00, ..., -2.5302e+00,
-3.3946e+00, -1.0510e-01],
...,
[-2.2875e-01, -1.3511e-02, -8.9770e-02, ..., -6.2087e+00,
-1.1379e+00, 1.5431e+00],
[-1.2792e+00, -5.5348e-01, 2.9047e-01, ..., -1.7488e+00,
3.0427e+00, 1.9252e+00],
[ 1.1528e+00, 1.0429e+00, 3.9329e-01, ..., 3.2680e+00,
1.5510e+00, -1.2096e+00]],
[[-3.1644e+00, -4.8225e+00, -2.2744e+00, ..., -6.7881e+00,
-6.4722e+00, -1.2480e+00],
[-3.4670e+00, -1.2544e-01, 1.0555e+00, ..., -8.6691e+00,
-8.2449e+00, -2.5213e+00],
[-2.3148e+00, -1.0949e+00, -2.5090e+00, ..., -7.1763e+00,
-7.7804e+00, -2.3668e+00],
...,
[-1.7279e-01, 1.3344e+00, 8.1542e-01, ..., -4.8328e+00,
-4.8772e+00, -6.3993e-02],
[-3.4898e+00, -3.1950e+00, -4.2009e-01, ..., -5.3568e-01,
-2.4893e+00, -3.9024e+00],
[ 1.3281e+00, 4.5939e-01, -1.5239e+00, ..., 1.3697e+00,
-4.3510e-01, -2.1802e+00]],
[[ 7.1007e-01, 3.7481e+00, 2.8190e+00, ..., 1.9460e+00,
2.3909e+00, 9.2219e-01],
[ 1.7803e+00, -7.3227e-01, -2.4622e+00, ..., 2.1111e+00,
2.7815e+00, -4.2078e-02],
[ 2.3376e+00, 5.7626e-01, 2.4635e-01, ..., 2.1034e+00,
2.1755e+00, -1.6183e-01],
...,
[-1.6628e-01, -2.3593e-01, -1.3851e+00, ..., -1.2008e+00,
-7.7953e-01, 1.2756e+00],
[ 6.1794e-01, -2.5421e-01, -1.4768e+00, ..., 1.1944e+00,
3.8400e+00, 2.0816e+00],
[-7.6447e-01, 2.8363e-01, -9.7310e-02, ..., -1.5121e+00,
-1.5954e+00, -1.8349e+00]],
...,
[[-3.0186e+00, -3.6522e+00, 1.4043e-01, ..., -7.0042e+00,
-4.9153e+00, -6.0840e-01],
[-4.2687e+00, 6.9306e-01, 9.9996e-01, ..., -5.2084e+00,
-5.9304e+00, -1.9948e+00],
[-3.7039e+00, -2.2729e+00, -1.3259e+00, ..., -4.9654e+00,
-6.9266e+00, -1.3979e+00],
...,
[ 1.9831e-01, 9.3262e-01, -7.9734e-01, ..., -6.0547e+00,
-6.5389e+00, -4.3024e+00],
[-1.9508e+00, -1.8041e+00, -1.2369e+00, ..., -4.1116e-01,
-6.3385e+00, -3.0457e+00],
[ 1.8966e+00, 5.1350e-01, -7.8343e-01, ..., 1.8154e+00,
-3.1049e+00, -2.5987e-01]],
[[-1.9861e+00, -2.4131e+00, -2.4446e+00, ..., -1.2248e+00,
-4.8671e-01, -6.7295e-01],
[-3.0220e+00, -5.3899e-01, -1.2520e+00, ..., -2.0923e+00,
-1.8267e+00, -3.5845e-01],
[-2.2257e+00, -2.7289e+00, -4.4532e+00, ..., 8.6375e-01,
-1.2502e+00, -5.9058e-01],
...,
[-5.6715e-01, -6.6900e-01, -2.3908e+00, ..., -4.7610e+00,
-4.7991e+00, -1.9263e-01],
[-1.8370e+00, -3.7725e-01, 3.2297e-01, ..., -5.4214e+00,
-4.3012e+00, 1.7788e+00],
[-1.0897e-01, -1.5617e-01, -4.1051e-01, ..., -3.9222e-02,
-3.3298e+00, 8.3488e-01]],
[[ 2.4015e+00, -1.3330e+00, -2.5174e-01, ..., -4.2981e-01,
9.0654e-01, -1.4138e+00],
[ 2.5452e+00, 1.2886e-01, -2.2976e-01, ..., 3.7397e+00,
2.5322e+00, 2.1703e+00],
[ 5.2147e-01, -2.4165e+00, -3.6317e+00, ..., 3.8695e+00,
1.1135e+00, 1.1850e+00],
...,
[-2.1799e-01, -1.1842e+00, -1.9155e+00, ..., -6.8859e+00,
-7.0339e-01, 1.6500e+00],
[-7.4454e-01, 3.7227e-01, -7.0828e-01, ..., -6.1307e+00,
1.0465e+00, 3.7668e+00],
[ 1.3098e-01, 5.4578e-01, 1.2350e+00, ..., -1.1574e+00,
1.2025e+00, 1.9142e+00]]],
[[[-1.8130e-01, 6.1184e-01, -1.7169e-01, ..., -1.3052e+00,
-1.0132e+00, -8.5848e-01],
[ 1.4168e+00, 1.4631e+00, 2.5155e+00, ..., -3.2562e-01,
2.3781e-01, -1.0668e+00],
[ 9.9282e-01, -8.9859e-01, -4.1689e+00, ..., -1.5518e+00,
-7.4198e-01, -1.9672e+00],
...,
[-3.9183e-01, 1.1685e-02, 4.4693e-01, ..., -5.3973e+00,
-2.5317e+00, -2.4216e-01],
[-2.4723e+00, -1.5386e+00, 1.1454e+00, ..., -3.3952e+00,
-3.2191e+00, 2.1573e-01],
[-2.4166e+00, -4.0047e+00, -3.1033e+00, ..., -4.5064e+00,
-3.2492e+00, -1.3067e+00]],
[[-7.5313e-01, -1.6834e+00, -2.8070e+00, ..., -2.9057e+00,
-4.6284e+00, -2.4610e+00],
[-1.9147e+00, -4.2733e+00, -5.7762e+00, ..., -2.8471e+00,
-4.8149e+00, -4.5119e+00],
[-3.6364e+00, -7.1860e+00, -7.2798e+00, ..., -4.1011e+00,
-3.9152e+00, -3.3745e+00],
...,
[-6.3288e+00, -8.9663e+00, -6.6549e+00, ..., -1.2093e+01,
-8.7646e+00, -1.4654e+00],
[-9.6775e+00, -1.0040e+01, -8.3249e+00, ..., -1.0866e+01,
-1.0343e+01, -3.4751e+00],
[-2.5601e+00, -4.8891e+00, -4.2908e+00, ..., -4.1233e+00,
-4.0459e+00, -6.9307e-01]],
[[ 8.2729e-01, 1.4353e+00, 1.3748e+00, ..., 1.5436e+00,
1.9157e+00, 8.8498e-01],
[ 1.1267e+00, 1.5733e+00, 1.0958e+00, ..., 1.0988e+00,
2.2546e+00, 9.5821e-01],
[ 6.8909e-01, 1.9325e+00, 2.3969e+00, ..., 1.3157e+00,
2.1892e+00, 1.9662e+00],
...,
[-3.3871e-01, -6.3334e-01, -6.3265e-01, ..., 1.5245e+00,
-3.7151e-01, -6.8986e-01],
[-5.6230e-01, 6.6979e-01, -1.3509e+00, ..., -2.8186e-01,
-4.3372e-02, -3.3024e-01],
[ 5.9854e-01, 1.3637e+00, 1.0003e+00, ..., -4.3970e-01,
-3.0334e-01, -1.6587e+00]],
...,
[[-1.0839e+00, -1.1995e+00, -1.9246e+00, ..., -2.5521e+00,
-3.3699e+00, -1.2054e+00],
[-2.0334e+00, -2.4078e+00, -2.7182e+00, ..., -2.7425e+00,
-4.0987e+00, -3.3802e+00],
[-2.1831e+00, -3.6895e+00, -5.0366e+00, ..., -3.3013e+00,
-4.4163e+00, -2.7978e+00],
...,
[-2.9338e+00, -4.2208e+00, -2.1869e+00, ..., -7.3306e+00,
-5.9324e+00, -3.0852e+00],
[-3.8943e+00, -5.7865e+00, -2.3107e+00, ..., -4.2074e+00,
-5.3475e+00, -1.7142e+00],
[-3.2817e+00, -3.8040e+00, -3.6405e+00, ..., -3.0586e+00,
-2.6644e+00, 3.3351e-01]],
[[-1.6023e+00, -2.2378e+00, -3.3495e+00, ..., -2.9655e+00,
-2.8694e+00, -5.1637e-01],
[-4.1832e+00, -4.0971e+00, -3.2807e+00, ..., -3.9744e+00,
-5.9168e+00, -1.9540e+00],
[-4.2256e+00, -5.8346e+00, -5.6471e+00, ..., -3.0929e+00,
-5.1236e+00, -1.9868e+00],
...,
[-2.9308e+00, -2.8467e+00, -3.9644e+00, ..., -3.4113e+00,
-2.0009e-01, -3.4134e-01],
[-3.2803e+00, -3.8386e-01, -1.6314e+00, ..., -1.9475e+00,
-5.2172e-01, 6.2136e-01],
[-3.7060e+00, -2.9658e+00, -2.8210e+00, ..., -7.4144e-01,
-5.6405e-01, 5.9496e-01]],
[[ 8.4522e-01, -1.1584e-01, 6.6165e-01, ..., 4.6269e-01,
1.4737e+00, 6.1807e-01],
[ 1.8669e+00, 1.5017e+00, 2.2358e+00, ..., 7.6060e-01,
1.8827e+00, 3.2141e+00],
[ 1.7624e+00, 1.2385e+00, -2.4611e-02, ..., 1.1806e-01,
7.8376e-01, 7.6176e-01],
...,
[ 4.3218e+00, 3.4927e+00, 2.7570e+00, ..., 3.1404e+00,
3.5245e+00, 7.7701e-01],
[ 4.5649e+00, 4.5531e+00, 4.1199e+00, ..., 2.5716e+00,
2.4110e+00, 1.4369e+00],
[ 2.8452e+00, 2.8353e+00, 2.3196e+00, ..., 1.7073e-01,
1.8221e-01, 8.5969e-01]]],
[[[-3.6480e-01, -2.6863e+00, -2.3164e+00, ..., -6.0644e-01,
-3.9674e+00, -1.8967e+00],
[ 4.7716e-01, -2.9149e+00, -6.2303e+00, ..., 1.0573e+00,
-2.2782e+00, -1.0606e+00],
[ 3.7169e-01, 1.8786e-01, -3.1916e+00, ..., 2.6416e-02,
-2.5110e+00, -1.5118e+00],
...,
[-1.0197e+00, -2.2014e+00, -6.2776e-01, ..., 3.2019e+00,
-6.9184e-01, -1.3607e+00],
[-1.0975e+00, -9.5230e-01, -5.6325e-01, ..., -6.7007e+00,
-3.9700e+00, -2.9337e+00],
[-2.7525e+00, -5.2030e+00, -4.1943e+00, ..., -5.0522e+00,
-5.2250e+00, -4.5674e+00]],
[[-1.6099e+00, -5.1308e+00, -4.3976e+00, ..., -5.2955e+00,
-9.3727e+00, -2.3411e+00],
[-3.6863e+00, -5.1909e+00, -7.0947e+00, ..., -8.3880e+00,
-9.7527e+00, -3.3629e+00],
[-2.7816e+00, -5.8172e+00, -1.3128e+01, ..., -6.4023e+00,
-1.0096e+01, -5.5346e+00],
...,
[-5.7568e+00, -5.5355e+00, -3.8085e+00, ..., -4.6069e+00,
-1.0226e+01, -5.4666e+00],
[-7.3885e+00, -8.1417e+00, -6.5361e+00, ..., -1.3481e+01,
-8.6135e+00, -6.1715e+00],
[-2.5777e+00, -4.5725e+00, -4.1581e+00, ..., -4.4328e+00,
-4.2732e+00, -2.2279e+00]],
[[ 4.3209e-02, 1.7524e+00, 1.4012e+00, ..., 2.5608e+00,
2.5281e+00, 9.5622e-01],
[-9.2627e-01, -1.0461e+00, -3.5057e+00, ..., 1.7747e+00,
2.7880e+00, 1.8919e-01],
[-5.6054e-02, 9.7914e-01, -1.5069e+00, ..., 2.2662e+00,
3.0613e-01, -9.4870e-01],
...,
[ 2.5515e-01, 8.2655e-01, 7.2958e-01, ..., -7.9821e-01,
2.6923e+00, 7.2767e-01],
[ 8.2536e-01, 8.3733e-01, 5.5748e-01, ..., -3.2840e+00,
-8.5869e-01, -3.6291e-01],
[ 1.3766e+00, 1.6192e+00, 1.0667e+00, ..., 7.1362e-01,
-4.7324e-01, 8.7249e-02]],
...,
[[-1.6192e+00, -4.2556e+00, -4.3189e+00, ..., -3.3825e+00,
-6.9220e+00, -1.8523e+00],
[-2.5189e+00, -5.5180e+00, -5.0821e+00, ..., -2.2563e+00,
-6.7946e+00, -2.6860e+00],
[-9.3266e-01, -4.6613e+00, -5.1372e+00, ..., -4.1786e+00,
-6.5566e+00, -3.3205e+00],
...,
[-1.9463e+00, -1.8040e+00, -4.0110e-01, ..., -3.9825e+00,
-6.0519e+00, -3.5230e+00],
[-2.9909e+00, -3.1143e+00, -2.0459e+00, ..., -8.0786e+00,
-6.0271e+00, -2.9045e+00],
[-3.5819e+00, -3.9504e+00, -3.3118e+00, ..., -3.5614e+00,
-4.8008e+00, -2.4557e+00]],
[[-2.6672e+00, -2.2417e+00, -2.8199e+00, ..., -2.4226e+00,
-1.1882e+00, 4.6625e-01],
[-3.4544e+00, -1.5823e+00, 1.1266e+00, ..., -1.4666e+00,
-1.6624e+00, -4.0632e-01],
[-1.6787e+00, -2.1356e+00, 1.1132e+00, ..., -2.0340e-01,
-1.0384e-01, 2.4140e-01],
...,
[-3.1493e+00, -2.0056e+00, -1.8315e+00, ..., -3.1161e+00,
-3.3711e+00, 9.3265e-01],
[-3.6454e+00, -6.8152e-01, -1.7628e+00, ..., -4.6552e+00,
-3.1258e+00, 1.0870e+00],
[-4.5272e+00, -3.0300e+00, -2.4266e+00, ..., -2.5775e+00,
-2.6189e+00, -1.9900e-01]],
[[ 1.7718e+00, 5.1587e-01, 5.2082e-01, ..., 3.0435e+00,
1.3088e+00, -2.1503e-01],
[ 2.7790e+00, 2.9892e+00, 1.7835e+00, ..., 4.1641e+00,
3.5776e+00, 3.6860e+00],
[ 1.8860e+00, 1.2835e+00, 3.9909e+00, ..., 3.9936e+00,
4.0450e+00, 1.2400e+00],
...,
[ 2.1596e+00, 4.8644e-01, 1.5703e-01, ..., 5.0663e+00,
5.1397e+00, 2.9195e+00],
[ 3.9112e+00, 3.9456e+00, 2.6053e+00, ..., 3.7542e+00,
1.6576e+00, 2.8583e+00],
[ 2.0229e+00, 8.1438e-01, 1.7573e-01, ..., 8.8194e-01,
2.0409e+00, 2.2173e+00]]],
[[[ 4.0617e-01, 1.3525e+00, 4.2707e-01, ..., -2.8753e+00,
-1.2234e+00, -2.4364e-01],
[ 9.0960e-01, 1.1467e+00, 1.6449e+00, ..., -2.8179e+00,
-2.3026e+00, 1.5676e-01],
[ 7.2806e-01, 9.4098e-01, -3.2169e+00, ..., -1.6998e+00,
-2.1456e+00, -3.1273e-02],
...,
[-3.4010e-01, -1.2146e+01, -8.4201e+00, ..., -4.7007e+00,
-1.7143e+00, -1.0267e+00],
[-2.2057e+00, -6.3752e+00, -7.8149e+00, ..., -3.4157e+00,
7.2626e-01, 2.2334e+00],
[-1.0037e+00, -4.3499e+00, -5.6309e+00, ..., -4.8855e+00,
-3.3429e+00, -2.0943e+00]],
[[-1.1075e+00, -2.9371e+00, -6.2900e+00, ..., -5.3551e+00,
-6.0714e+00, -7.4506e-01],
[-4.4079e+00, -6.4687e+00, -7.2959e+00, ..., -5.8099e+00,
-8.4233e+00, -2.5558e+00],
[-4.8157e+00, -6.6892e+00, -2.1539e+01, ..., -6.9267e+00,
-7.5914e+00, -3.1675e+00],
...,
[-5.4510e+00, -1.4645e+01, -1.2356e+01, ..., -9.6755e+00,
-1.7752e+00, -3.5417e-03],
[-7.4353e+00, -9.6205e+00, -1.5005e+01, ..., -8.3148e+00,
-3.2685e+00, -3.7693e-01],
[-1.9028e+00, -3.1498e+00, -3.5838e+00, ..., -4.8442e+00,
-7.2665e+00, -5.0958e+00]],
[[-8.2515e-02, 1.9559e+00, 2.3616e+00, ..., 1.3784e+00,
2.1102e+00, 8.3289e-01],
[ 2.1042e-02, 7.7245e-01, -1.4768e+00, ..., 1.3909e+00,
2.6114e+00, -2.8054e-01],
[ 1.5288e+00, 1.2873e+00, -4.1064e-01, ..., 8.2560e-01,
2.4908e+00, -1.3410e-01],
...,
[-2.3736e+00, -2.3368e+00, -9.3994e+00, ..., -7.1115e+00,
-2.1774e+00, -6.5247e-02],
[-5.5564e-01, -1.0572e+00, -3.6255e+00, ..., -7.0411e+00,
-4.5525e+00, -2.0819e+00],
[ 2.7288e-01, -8.1877e-01, -8.0700e-01, ..., -2.3850e+00,
-4.2936e-01, -3.6727e+00]],
...,
[[-4.8388e-01, -1.6660e+00, -2.6445e+00, ..., -4.7393e+00,
-4.1847e+00, -3.0946e-01],
[-2.5860e+00, -1.8301e+00, -2.6666e+00, ..., -5.6778e+00,
-5.4741e+00, -2.2965e+00],
[-1.2720e+00, -1.9917e+00, -5.7487e+00, ..., -5.4099e+00,
-6.4094e+00, -1.4014e+00],
...,
[-1.4854e+00, -6.9208e+00, 1.6912e-01, ..., -5.7718e+00,
-7.5388e+00, -1.9949e+00],
[-2.5125e+00, -4.6361e+00, -1.6986e+00, ..., -5.4118e-01,
-5.2277e+00, -2.5782e-01],
[-9.2014e-01, -2.1801e+00, -3.6821e+00, ..., -4.6260e+00,
-8.0642e+00, -4.7266e+00]],
[[-1.6343e+00, -1.9980e+00, -4.0428e+00, ..., -9.8339e-01,
-1.1271e-01, -9.4272e-01],
[-3.8568e+00, -2.6071e+00, -4.9550e+00, ..., -2.8672e+00,
-1.3465e+00, -2.9844e-01],
[-4.3748e+00, -4.7842e+00, -1.0958e+01, ..., -2.2991e+00,
-2.1998e+00, -7.9036e-01],
...,
[-6.9498e+00, -6.2288e+00, 5.2088e+00, ..., 6.2901e-01,
-4.5439e+00, -1.6503e+00],
[-3.7844e+00, -4.2289e+00, 2.8603e+00, ..., 5.0514e+00,
2.6258e+00, 2.0062e+00],
[-2.1651e+00, -2.2790e+00, -9.4760e-01, ..., 2.1502e+00,
-1.2776e+00, 2.5865e-01]],
[[ 1.2834e+00, 1.4234e-01, 2.1894e+00, ..., -2.7683e-01,
6.9278e-01, -1.4541e+00],
[ 2.8376e+00, 1.2349e+00, 2.3120e+00, ..., 1.9243e+00,
2.7190e+00, 1.7262e+00],
[ 1.4151e+00, 1.5068e+00, 6.2829e+00, ..., 2.7497e+00,
2.0007e+00, 1.3031e+00],
...,
[ 2.6221e+00, 1.8285e+00, 8.8278e+00, ..., 2.9096e+00,
-2.7995e+00, -2.8012e+00],
[ 2.6340e+00, 1.9783e+00, 5.9978e+00, ..., 6.6821e+00,
7.2809e-01, 1.3996e+00],
[ 2.0480e+00, 1.4899e+00, 2.0778e-01, ..., 2.0088e+00,
2.6278e+00, 2.2551e+00]]]], device='cuda:0'),
tensor([[[[ 8.2003e-01, 1.4797e+00, 6.0009e-01, ..., -3.7873e+00,
-2.5905e+00, -1.0075e+00],
[ 1.3450e+00, 1.0651e+00, 6.6499e-01, ..., -2.8738e+00,
-2.6608e+00, -2.5141e-01],
[ 4.1712e-01, -2.1211e-01, -5.5534e-02, ..., -2.3451e+00,
-3.0087e+00, -1.3637e-01],
...,
[ 4.4452e-01, -1.0460e+00, -7.0188e-01, ..., -3.1193e+00,
-3.8745e+00, 6.0754e-01],
[ 2.3340e-01, -3.0676e-01, -6.8017e-01, ..., -4.6030e+00,
-3.7939e+00, 6.0960e-01],
[-4.3987e-01, -3.5725e+00, -3.3507e+00, ..., -4.0941e+00,
-3.3000e+00, -1.4765e+00]],
[[ 3.3375e-01, -1.1824e+00, -2.4827e+00, ..., -7.6347e+00,
-7.3723e+00, -1.3367e+00],
[-1.0976e+00, -2.8370e+00, -3.8666e+00, ..., -8.1409e+00,
-8.7580e+00, -2.5618e+00],
[-2.8451e-01, -1.8009e+00, -2.1760e+00, ..., -5.9549e+00,
-7.8278e+00, -2.5160e+00],
...,
[-2.4112e+00, -3.4184e+00, -3.2078e+00, ..., -9.2814e+00,
-8.8125e+00, -1.9223e+00],
[-4.1832e+00, -5.8026e+00, -5.0183e+00, ..., -1.0117e+01,
-1.0733e+01, -4.5029e+00],
[-3.9275e-01, -3.3839e+00, -3.5450e+00, ..., -4.0809e+00,
-3.6197e+00, -8.5374e-01]],
[[ 3.9217e-01, 2.1528e+00, 1.7811e+00, ..., 9.1993e-01,
2.2351e+00, 9.6460e-01],
[-1.8425e-01, 1.1499e+00, 6.9145e-01, ..., 1.3306e+00,
2.9384e+00, 3.0769e-03],
[ 1.0903e+00, 4.0997e+00, 2.3223e+00, ..., 1.5094e-01,
2.2047e+00, -2.9087e-01],
...,
[ 1.0939e+00, 2.2946e+00, 1.3367e+00, ..., -4.4531e-01,
1.2993e+00, -1.0219e+00],
[ 4.5881e-01, 2.0821e+00, 8.7224e-01, ..., -1.0960e+00,
8.6708e-01, -1.4475e+00],
[-2.6729e-01, 1.8036e+00, 9.9808e-01, ..., -1.2253e-01,
-3.5933e-01, -1.8050e+00]],
...,
[[-3.5878e-01, -8.5510e-01, -1.4867e+00, ..., -5.4911e+00,
-5.3932e+00, -7.1799e-01],
[-2.6684e+00, -4.1561e+00, -2.1955e+00, ..., -4.3694e+00,
-5.4043e+00, -1.8947e+00],
[-1.0971e+00, -2.9504e+00, -1.4894e+00, ..., -4.1788e+00,
-5.8752e+00, -1.5392e+00],
...,
[-1.5395e+00, -2.9079e+00, -1.2796e+00, ..., -4.6362e+00,
-4.9393e+00, -1.8771e+00],
[-1.9130e+00, -3.6818e+00, -1.9498e+00, ..., -3.0545e+00,
-4.9988e+00, -1.1925e+00],
[-1.3158e+00, -3.7359e+00, -2.5522e+00, ..., -1.5523e+00,
-2.1442e+00, 2.1921e-01]],
[[-1.0018e+00, -8.1288e-01, -9.1865e-01, ..., -2.1773e+00,
-4.3560e-01, -6.2292e-01],
[-2.9171e+00, -2.6002e+00, -1.5084e+00, ..., -3.8423e+00,
-2.0716e+00, -4.1486e-01],
[-2.7425e+00, -4.4445e+00, -1.6773e+00, ..., -1.4279e+00,
-1.3773e+00, -2.5944e-01],
...,
[-3.0273e+00, -4.4121e+00, -2.2037e+00, ..., -4.3211e+00,
-2.1558e+00, -6.2082e-01],
[-3.0776e+00, -2.2935e+00, -8.9174e-01, ..., -2.1954e+00,
-7.7145e-01, 1.2897e+00],
[-3.3340e+00, -3.2899e+00, -1.3757e+00, ..., -1.7218e-01,
-8.9586e-01, 6.3937e-01]],
[[ 8.9735e-01, 7.4251e-01, 8.7410e-01, ..., -1.4787e-01,
7.3766e-01, -1.3279e+00],
[ 1.3178e+00, 1.7661e+00, 1.6842e+00, ..., 3.2076e+00,
3.1833e+00, 2.3351e+00],
[ 6.6515e-01, -1.0669e-01, 4.1534e-01, ..., 2.6644e+00,
1.6790e+00, 1.0900e+00],
...,
[ 1.9363e+00, 1.3736e+00, 1.4326e+00, ..., 1.8021e+00,
1.9988e+00, 4.7915e-01],
[ 3.1351e+00, 3.1889e+00, 1.9845e+00, ..., 1.7398e+00,
1.0479e+00, 1.8536e+00],
[ 1.9902e+00, 5.4623e-01, 6.8551e-01, ..., 4.5182e-01,
7.2797e-02, 8.5753e-01]]],
[[[ 1.6366e-01, 1.3448e+00, 5.5433e-01, ..., -6.3870e-01,
2.7059e-02, 6.5512e-01],
[ 4.4293e-01, 2.3225e-01, 6.2229e-01, ..., -1.0572e+00,
-9.6981e-01, 1.0793e+00],
[-5.2209e-01, -7.1785e-02, 6.0776e-01, ..., -9.6131e-01,
-1.3207e+00, 8.0065e-01],
...,
[-6.8380e-01, -2.0937e+00, 3.1936e-01, ..., -1.0173e+00,
2.2009e-01, -8.3857e-01],
[-1.1089e+00, -1.1175e+00, -8.6319e-02, ..., -4.5744e-01,
1.1410e-01, -1.5861e+00],
[-2.8964e+00, -5.4026e+00, -4.2603e+00, ..., -4.4992e+00,
-3.9430e+00, -3.7925e+00]],
[[-8.7571e-01, -2.3429e+00, -4.1430e+00, ..., -3.5100e+00,
-4.9821e+00, -3.9534e-01],
[-4.7292e+00, -6.0728e+00, -5.1385e+00, ..., -5.6160e+00,
-7.5902e+00, -1.7636e+00],
[-3.6497e+00, -4.3491e+00, -3.5496e+00, ..., -4.0462e+00,
-6.2457e+00, -1.4301e+00],
...,
[-5.9148e+00, -5.8724e+00, -3.9179e+00, ..., -4.8772e+00,
-5.0258e+00, -3.8735e+00],
[-7.4788e+00, -8.2263e+00, -6.3690e+00, ..., -7.3530e+00,
-7.9021e+00, -6.3800e+00],
[-2.5937e+00, -4.8503e+00, -4.1352e+00, ..., -3.7561e+00,
-3.9950e+00, -2.6858e+00]],
[[-2.3450e-01, 2.1228e+00, 1.1609e+00, ..., 8.6329e-01,
2.1436e+00, 9.7434e-01],
[-1.1494e-01, 1.0771e+00, 7.6910e-01, ..., 6.2730e-01,
1.2154e+00, -3.8889e-01],
[-5.4958e-02, 1.9536e+00, 1.3781e+00, ..., 2.5015e-01,
1.6520e+00, 1.2066e-01],
...,
[ 9.1438e-01, 1.2968e+00, 1.1462e+00, ..., 1.0191e+00,
1.3099e+00, 3.4658e-01],
[ 8.1400e-01, 8.5546e-01, 9.1355e-01, ..., 1.2601e+00,
3.3270e-01, 1.2777e-01],
[ 1.4157e+00, 1.9334e+00, 1.3794e+00, ..., 1.5547e+00,
1.1361e+00, 1.5428e-01]],
...,
[[-8.8750e-01, -1.8421e+00, -1.8378e+00, ..., -2.6019e+00,
-2.9049e+00, -5.1056e-01],
[-3.1876e+00, -4.2478e+00, -2.5987e+00, ..., -3.6539e+00,
-4.3172e+00, -2.7116e+00],
[-1.6115e+00, -2.1782e+00, -9.6374e-01, ..., -2.0654e+00,
-3.9976e+00, -1.4168e+00],
...,
[-3.1966e+00, -2.7591e+00, -6.4113e-01, ..., -2.1710e+00,
-3.6483e+00, -1.6953e+00],
[-3.6163e+00, -3.4855e+00, -2.1612e+00, ..., -2.2289e+00,
-4.0713e+00, -2.3320e+00],
[-3.8203e+00, -4.0201e+00, -3.3989e+00, ..., -2.7194e+00,
-3.6139e+00, -1.7062e+00]],
[[-1.7435e+00, -2.1207e+00, -2.8744e+00, ..., -1.5634e+00,
-1.5111e+00, -1.4048e+00],
[-4.3371e+00, -3.3811e+00, -3.0005e+00, ..., -2.7071e+00,
-1.6253e+00, -9.4079e-01],
[-3.4908e+00, -3.0436e+00, -2.9297e+00, ..., -4.0716e-01,
-9.9341e-01, -4.5855e-01],
...,
[-3.7575e+00, -2.4505e+00, -3.3849e+00, ..., -1.5927e+00,
-3.0837e+00, -1.7552e+00],
[-3.6420e+00, -7.6094e-01, -2.6981e+00, ..., -1.1881e+00,
-2.8324e+00, -1.2622e+00],
[-4.4359e+00, -3.0951e+00, -2.8848e+00, ..., -2.3846e+00,
-3.3506e+00, -1.1619e+00]],
[[ 1.0703e+00, 6.2943e-01, 6.6567e-01, ..., -5.9275e-01,
1.4645e-01, -3.7002e-01],
[ 3.5343e+00, 3.3093e+00, 2.2991e+00, ..., 1.6311e+00,
2.1846e+00, 2.5792e+00],
[ 1.5169e+00, 5.9564e-01, 2.7515e-01, ..., 1.3012e+00,
1.6688e+00, 1.6924e+00],
...,
[ 3.1263e+00, 9.9013e-01, 1.2243e+00, ..., 2.6387e+00,
2.1187e+00, 1.3358e+00],
[ 4.0652e+00, 3.7393e+00, 2.6388e+00, ..., 2.4067e+00,
2.3073e+00, 2.0468e+00],
[ 2.0602e+00, 7.8045e-01, 3.0226e-01, ..., -3.5768e-01,
-1.4680e-01, -1.5669e-01]]],
[[[ 2.7811e-01, 1.4046e+00, 4.7902e-01, ..., -6.8947e-01,
5.7444e-02, 6.7007e-01],
[ 3.5273e-01, 1.4125e-01, 3.8294e-01, ..., -1.6420e+00,
-1.0719e+00, 1.0600e+00],
[-6.4573e-01, 5.0742e-02, 7.8827e-01, ..., -1.1412e+00,
-6.8965e-01, 6.3427e-01],
...,
[-1.0162e+00, -2.2810e+00, 5.6107e-01, ..., -2.1333e+00,
-1.5199e+00, -9.2032e-01],
[-1.0827e+00, -1.2250e+00, 4.5887e-01, ..., -1.9213e+00,
-6.9140e-01, -1.4527e+00],
[-2.6423e+00, -5.1886e+00, -4.2843e+00, ..., -5.0859e+00,
-3.9875e+00, -3.8226e+00]],
[[-8.3498e-01, -2.3194e+00, -4.2096e+00, ..., -3.3480e+00,
-4.7437e+00, -3.4970e-01],
[-4.7898e+00, -6.1664e+00, -5.0742e+00, ..., -5.9856e+00,
-7.1149e+00, -1.6936e+00],
[-3.4974e+00, -3.9826e+00, -3.4586e+00, ..., -4.3052e+00,
-6.3924e+00, -1.3534e+00],
...,
[-5.7663e+00, -5.7341e+00, -3.6091e+00, ..., -6.9621e+00,
-5.7101e+00, -3.8001e+00],
[-7.0474e+00, -7.6693e+00, -6.1088e+00, ..., -7.7770e+00,
-8.0268e+00, -6.5420e+00],
[-2.5046e+00, -4.3150e+00, -4.0469e+00, ..., -4.0174e+00,
-4.2335e+00, -2.6146e+00]],
[[-3.2692e-01, 2.0927e+00, 1.2053e+00, ..., 7.7964e-01,
1.9888e+00, 9.1311e-01],
[-1.7418e-01, 1.0909e+00, 9.0927e-01, ..., 3.3177e-01,
1.1060e+00, -6.1931e-01],
[ 2.7051e-02, 1.9184e+00, 1.3210e+00, ..., 6.0994e-01,
1.1471e+00, -1.7110e-01],
...,
[ 9.5382e-01, 1.1038e+00, 1.2140e+00, ..., 6.7651e-01,
4.8939e-02, 4.3843e-01],
[ 8.7005e-01, 6.7271e-01, 8.3902e-01, ..., -8.5626e-01,
3.0511e-01, 3.5132e-01],
[ 1.3268e+00, 1.5386e+00, 1.3917e+00, ..., 1.6623e+00,
1.3331e+00, 2.0386e-01]],
...,
[[-8.5514e-01, -1.8657e+00, -1.8958e+00, ..., -2.5124e+00,
-2.8200e+00, -3.5586e-01],
[-3.2478e+00, -4.2926e+00, -2.6256e+00, ..., -3.8495e+00,
-4.6172e+00, -2.4993e+00],
[-1.4476e+00, -1.9836e+00, -8.5634e-01, ..., -2.4613e+00,
-4.4743e+00, -1.5062e+00],
...,
[-2.7568e+00, -2.7861e+00, -7.4932e-01, ..., -2.5982e+00,
-4.2973e+00, -2.0995e+00],
[-2.9496e+00, -3.1165e+00, -1.6066e+00, ..., -2.7145e+00,
-4.5315e+00, -2.3653e+00],
[-3.3862e+00, -3.8115e+00, -3.4661e+00, ..., -3.5921e+00,
-3.8604e+00, -1.7375e+00]],
[[-1.7130e+00, -2.1768e+00, -2.9818e+00, ..., -1.6485e+00,
-1.4988e+00, -1.4528e+00],
[-4.3948e+00, -3.5513e+00, -3.0332e+00, ..., -2.8651e+00,
-2.2655e+00, -1.2195e+00],
[-3.2555e+00, -2.8514e+00, -2.7013e+00, ..., 2.5161e-01,
-1.4941e+00, -8.2816e-01],
...,
[-4.1426e+00, -2.0392e+00, -2.7711e+00, ..., -3.2272e+00,
-3.1269e+00, -1.3375e+00],
[-3.6955e+00, -4.9135e-01, -1.5522e+00, ..., -1.2553e+00,
-2.5307e+00, -1.1288e+00],
[-4.3743e+00, -2.8799e+00, -2.5115e+00, ..., -1.7475e+00,
-2.9811e+00, -1.1271e+00]],
[[ 1.0980e+00, 5.8569e-01, 6.6462e-01, ..., -8.3155e-01,
1.4967e-01, -4.1582e-01],
[ 3.4759e+00, 3.3011e+00, 2.2464e+00, ..., 1.2141e+00,
1.6702e+00, 2.2238e+00],
[ 1.3522e+00, 6.2527e-01, 2.0280e-01, ..., 1.2265e+00,
1.3107e+00, 9.2702e-01],
...,
[ 2.8009e+00, 1.1411e+00, 1.2863e+00, ..., 1.7066e+00,
2.3849e+00, 1.3906e+00],
[ 3.6209e+00, 3.4512e+00, 2.6873e+00, ..., 3.5310e+00,
2.5744e+00, 2.0082e+00],
[ 1.9020e+00, 7.2984e-01, 3.1994e-01, ..., -2.8871e-01,
-4.1305e-01, -1.9933e-01]]],
[[[-6.9423e-02, -2.9941e+00, -2.4866e+00, ..., -4.0734e-01,
-6.8488e-01, -8.1985e-01],
[-3.6626e+00, -2.8613e+00, -6.9665e-01, ..., 2.3090e+00,
6.6432e-02, -1.6929e+00],
[ 1.2783e+00, 4.9571e+00, -2.1759e+00, ..., -8.3251e-01,
-1.6145e+00, -2.3687e+00],
...,
[-5.7769e+00, -7.4078e+00, -8.3569e+00, ..., -3.9461e-01,
-2.1792e-01, -5.7384e-01],
[-3.1870e+00, -5.6083e+00, -4.4528e+00, ..., 8.2539e-02,
-6.8364e-01, -1.4918e+00],
[-3.6971e-01, -9.4193e-01, 1.0858e+00, ..., -2.7377e+00,
-3.1413e+00, -2.9056e+00]],
[[-5.4047e-01, -4.2982e+00, -3.1083e+00, ..., -2.7614e+00,
-5.2191e+00, -2.5862e+00],
[-8.5156e+00, -5.6046e+00, 6.8517e-01, ..., -2.3097e-01,
-7.1818e+00, -3.4563e+00],
[ 6.7811e-01, 1.4872e+00, 4.3328e-01, ..., -2.5022e+00,
-6.5177e+00, -3.4073e+00],
...,
[ 6.6645e-01, -4.2436e+00, -7.7043e+00, ..., -3.4209e+00,
-4.0811e+00, -3.1152e+00],
[ 5.2935e-01, 1.4399e+00, -4.2792e+00, ..., -4.7996e+00,
-5.8523e+00, -5.3697e+00],
[-9.0902e-01, -1.0898e+00, 1.9139e-01, ..., -3.1663e+00,
-2.9974e+00, -1.6905e+00]],
[[ 7.9375e-01, 4.3639e+00, 1.8162e+00, ..., 3.0827e+00,
1.5125e+00, 5.4422e-01],
[ 2.5250e+00, 4.9839e+00, 1.7748e+00, ..., 2.8040e+00,
2.9397e+00, 8.7419e-01],
[-1.4911e+00, -2.9858e+00, -5.0055e+00, ..., 2.8623e+00,
3.0726e+00, 1.0674e+00],
...,
[ 2.3419e+00, 2.4239e+00, 1.4153e+00, ..., 8.7615e-01,
1.1083e+00, 8.7354e-01],
[-2.1357e+00, 4.8755e+00, 2.1904e+00, ..., 9.2407e-01,
1.3173e+00, 4.0806e-01],
[-1.7488e+00, -2.9180e+00, 3.5628e-01, ..., 1.1259e+00,
1.8219e+00, 5.0292e-01]],
...,
[[-2.9511e+00, -2.9818e+00, -9.1448e-01, ..., -4.3818e+00,
-3.4088e+00, -1.3972e+00],
[-6.4809e+00, -2.1393e+00, -6.2528e-01, ..., -2.8511e+00,
-4.2514e+00, -4.2117e+00],
[ 1.7643e+00, 1.6661e+00, 1.7344e+00, ..., -6.3356e+00,
-3.5054e+00, -4.4518e+00],
...,
[-5.1070e-01, -4.5034e+00, -3.3063e+00, ..., -1.5256e+00,
-3.5009e+00, -1.2687e+00],
[-7.9397e-02, -1.1388e+00, -2.7500e+00, ..., -1.2193e+00,
-3.6316e+00, -1.6953e+00],
[-2.1422e+00, -1.1978e+00, 1.0226e+00, ..., -2.0244e+00,
-2.6223e+00, -5.9654e-01]],
[[-2.2806e+00, -9.0580e-01, 5.8933e-01, ..., -2.9384e-01,
3.0146e-01, -4.5675e-01],
[-9.1674e+00, -4.5451e+00, -1.6535e+00, ..., -1.6335e+00,
-1.5178e+00, -1.1650e+00],
[-4.3067e+00, 2.3512e+00, 1.4086e+00, ..., -2.8795e+00,
-1.6299e+00, -1.1142e+00],
...,
[-7.6774e+00, -1.1628e+01, -7.1345e+00, ..., 1.1583e-01,
-3.1148e+00, -1.8953e+00],
[-3.5655e+00, -9.4199e+00, -7.2254e+00, ..., -2.8443e-01,
-3.5154e+00, -1.8957e+00],
[ 6.8558e-01, -2.1798e+00, -9.4599e-01, ..., -6.7621e-01,
-2.6830e+00, -1.2607e+00]],
[[ 1.4120e+00, 6.6928e-02, -1.4675e-02, ..., 3.2302e+00,
1.0976e+00, 1.4502e-01],
[ 1.4999e+00, 2.8732e+00, 8.9868e-01, ..., 3.0667e+00,
3.0528e+00, 2.7327e+00],
[ 2.2902e+00, 2.7156e+00, -2.6295e+00, ..., 4.1484e+00,
2.0826e+00, 1.1800e+00],
...,
[-4.0917e+00, -9.6128e+00, -8.2245e+00, ..., 1.5613e+00,
5.0749e-02, 5.2948e-01],
[-9.7881e-01, -8.1836e+00, -9.3987e+00, ..., 2.1521e+00,
8.1030e-01, 1.3776e+00],
[ 2.9979e+00, -8.5860e-01, -2.5141e+00, ..., 7.6333e-01,
-2.0864e-01, -8.6788e-02]]],
[[[ 2.2539e-01, 1.4173e+00, 3.0346e-02, ..., -3.2817e+00,
-2.2686e+00, -9.7547e-01],
[ 4.5666e-01, 9.9735e-01, 6.7759e-01, ..., -1.9367e+00,
-2.2266e+00, -2.6212e-01],
[-4.7710e-01, -1.8503e-01, 3.1393e-01, ..., -1.6288e+00,
-2.4246e+00, -4.0829e-01],
...,
[-5.3787e-01, -1.6670e+00, -4.2178e-01, ..., -2.7593e+00,
-4.1000e+00, 9.9563e-02],
[-1.9894e+00, -1.4602e+00, -8.3914e-01, ..., -3.8446e+00,
-3.2607e+00, 5.2099e-01],
[-1.1616e+00, -3.3950e+00, -3.3269e+00, ..., -3.3541e+00,
-3.1881e+00, -1.5163e+00]],
[[-1.1397e+00, -1.5695e+00, -3.5892e+00, ..., -6.0417e+00,
-6.8198e+00, -1.3107e+00],
[-4.0030e+00, -4.2808e+00, -3.8390e+00, ..., -7.0442e+00,
-8.4012e+00, -2.5867e+00],
[-4.0092e+00, -4.3609e+00, -4.3882e+00, ..., -5.0950e+00,
-8.2030e+00, -3.1657e+00],
...,
[-6.1201e+00, -5.9963e+00, -3.5691e+00, ..., -6.0462e+00,
-8.1323e+00, -2.1390e+00],
[-7.7777e+00, -7.7265e+00, -7.1376e+00, ..., -9.5209e+00,
-1.1110e+01, -4.8019e+00],
[-1.8179e+00, -3.2724e+00, -3.8556e+00, ..., -3.7322e+00,
-3.8783e+00, -7.9768e-01]],
[[-9.8607e-04, 1.3110e+00, 1.5940e+00, ..., 1.4725e+00,
2.5257e+00, 1.0530e+00],
[-1.4698e-02, -5.8019e-02, 9.8823e-01, ..., 1.0862e+00,
2.7363e+00, 1.4227e-01],
[ 5.9240e-01, 1.1742e+00, 1.7765e+00, ..., -1.4893e-01,
1.6414e+00, -3.3519e-02],
...,
[ 3.2145e-01, 8.6597e-01, 6.8773e-01, ..., -4.6525e-01,
2.1377e-01, -7.6545e-01],
[ 4.7565e-01, 8.1185e-01, 3.2776e-01, ..., -1.2803e+00,
1.1496e+00, -1.2171e+00],
[ 5.5236e-01, 1.1287e+00, 1.6576e+00, ..., 3.5773e-01,
-3.4575e-01, -1.9563e+00]],
...,
[[-6.2393e-01, -1.5118e+00, -2.3722e+00, ..., -4.9701e+00,
-5.1272e+00, -7.9012e-01],
[-2.7518e+00, -2.6283e+00, -1.8703e+00, ..., -3.9739e+00,
-4.4577e+00, -1.8609e+00],
[-2.1295e+00, -1.1109e+00, -1.0872e+00, ..., -3.8384e+00,
-3.8628e+00, -1.6911e+00],
...,
[-1.9355e+00, -1.2001e+00, -3.0659e-01, ..., -4.0000e+00,
-5.8193e+00, -2.1881e+00],
[-3.2307e+00, -3.0477e+00, -2.2010e+00, ..., -4.5319e+00,
-5.8033e+00, -1.0514e+00],
[-1.5510e+00, -1.9889e+00, -2.2366e+00, ..., -2.0423e+00,
-2.2848e+00, 3.4753e-01]],
[[-1.3556e+00, -2.4869e+00, -3.9983e+00, ..., -1.3907e+00,
-5.7407e-01, -5.6761e-01],
[-3.7056e+00, -3.7510e+00, -4.8659e+00, ..., -3.6269e+00,
-2.4563e+00, -3.8133e-01],
[-4.6312e+00, -3.9429e+00, -4.5775e+00, ..., -4.0913e-01,
-9.6634e-01, 4.6276e-01],
...,
[-4.7689e+00, -3.8839e+00, -4.2204e+00, ..., -2.1547e+00,
-1.5909e+00, -5.0693e-01],
[-4.2140e+00, -1.8098e+00, -1.7309e+00, ..., -2.9831e-01,
8.4012e-02, 1.2749e+00],
[-3.2572e+00, -3.1779e+00, -2.1310e+00, ..., 9.7089e-02,
-7.9483e-01, 6.0324e-01]],
[[ 1.2014e+00, -4.2161e-02, 1.1783e+00, ..., 3.7818e-01,
8.0656e-01, -1.3124e+00],
[ 2.3356e+00, 1.2421e+00, 1.1365e+00, ..., 3.8065e+00,
3.2763e+00, 2.5329e+00],
[ 1.4214e+00, 8.4287e-01, 3.7642e-01, ..., 1.7571e+00,
1.5325e+00, 1.1449e+00],
...,
[ 2.5111e+00, 1.4883e+00, 1.2550e+00, ..., 3.4582e+00,
3.0968e+00, 6.1091e-01],
[ 2.8484e+00, 3.9893e+00, 2.2619e+00, ..., 2.9983e+00,
7.5229e-01, 1.6433e+00],
[ 1.9520e+00, 1.5606e+00, 4.9137e-01, ..., -3.4435e-02,
-1.7140e-01, 8.9601e-01]]]], device='cuda:0'),
tensor([[[[ 2.9083e-01, 1.3068e+00, 1.1934e-01, ..., -3.0495e+00,
-2.4576e+00, -1.0402e+00],
[ 3.3928e-01, 1.0430e+00, 1.2357e+00, ..., -1.5558e+00,
-2.4072e+00, -2.4364e-01],
[-6.1904e-02, -3.1314e-01, -6.2505e-01, ..., -1.0554e+00,
-2.8641e+00, -2.6021e-02],
...,
[-4.3449e-01, -1.4519e+00, -5.4909e-02, ..., 2.4953e+00,
-2.3317e+00, 1.2199e-01],
[-1.9438e+00, -1.4085e+00, -1.0971e+00, ..., -1.1026e+00,
-3.2801e+00, 2.2821e-01],
[-1.1226e+00, -3.3703e+00, -3.7944e+00, ..., -2.7209e+00,
-3.1272e+00, -1.6043e+00]],
[[-1.0562e+00, -1.5593e+00, -3.4808e+00, ..., -6.3662e+00,
-7.3569e+00, -1.3365e+00],
[-3.8578e+00, -4.5278e+00, -4.4569e+00, ..., -5.8910e+00,
-8.4676e+00, -2.3724e+00],
[-3.8284e+00, -5.4443e+00, -4.6638e+00, ..., -5.4624e+00,
-7.7749e+00, -2.4231e+00],
...,
[-5.9342e+00, -5.9846e+00, -3.4175e+00, ..., -4.0489e+00,
-8.2754e+00, -2.1612e+00],
[-7.5952e+00, -7.8158e+00, -8.0154e+00, ..., -9.2412e+00,
-1.1245e+01, -4.5618e+00],
[-1.7997e+00, -3.3766e+00, -3.8521e+00, ..., -3.2430e+00,
-3.5680e+00, -6.5947e-01]],
[[-4.9816e-02, 1.2641e+00, 1.6575e+00, ..., 1.6344e+00,
2.3570e+00, 9.4761e-01],
[ 4.5757e-02, 1.1694e-01, 1.1033e+00, ..., 1.7472e+00,
3.0640e+00, 5.1158e-02],
[ 8.0327e-01, 1.0999e+00, 9.5294e-01, ..., 1.8505e-01,
1.9505e+00, -2.8602e-01],
...,
[ 1.6820e-01, 3.2119e-01, 1.0936e+00, ..., -1.6945e-01,
1.5583e+00, -1.2123e+00],
[ 6.1038e-01, 8.2827e-01, 4.8872e-01, ..., -1.6705e-01,
1.1281e+00, -1.8816e+00],
[ 5.4363e-01, 1.2634e+00, 1.9083e+00, ..., -8.7183e-01,
-1.3417e+00, -2.0833e+00]],
...,
[[-5.9156e-01, -1.5179e+00, -2.3232e+00, ..., -5.0663e+00,
-5.2061e+00, -7.4485e-01],
[-2.7744e+00, -2.7582e+00, -2.1067e+00, ..., -3.5759e+00,
-4.9342e+00, -1.8639e+00],
[-1.8741e+00, -1.5194e+00, -1.5486e+00, ..., -4.4683e+00,
-5.3063e+00, -1.1855e+00],
...,
[-1.9072e+00, -6.4512e-01, 9.6226e-01, ..., -3.4843e+00,
-6.1349e+00, -1.0911e+00],
[-3.2736e+00, -3.2521e+00, -2.8103e+00, ..., -3.1020e+00,
-5.7033e+00, -9.3903e-01],
[-1.5425e+00, -2.1351e+00, -2.7323e+00, ..., -1.3411e+00,
-1.9400e+00, 5.0802e-01]],
[[-1.2840e+00, -2.4004e+00, -3.9713e+00, ..., -1.3703e+00,
-3.2472e-01, -5.8285e-01],
[-3.6660e+00, -3.4793e+00, -4.5974e+00, ..., -3.3695e+00,
-1.7115e+00, -2.8585e-01],
[-4.5349e+00, -3.5197e+00, -3.4122e+00, ..., -2.7952e+00,
-1.6922e+00, -4.8834e-01],
...,
[-4.6584e+00, -4.2673e+00, -4.2489e+00, ..., -3.4995e+00,
-2.7497e+00, -5.0645e-01],
[-4.1206e+00, -2.0401e+00, -2.1735e+00, ..., -1.3556e+00,
-5.9974e-01, 1.5273e+00],
[-3.2872e+00, -3.4551e+00, -2.3760e+00, ..., -2.1124e-01,
-9.6377e-01, 7.3853e-01]],
[[ 1.2188e+00, -5.9324e-02, 1.1731e+00, ..., 6.3634e-01,
7.6626e-01, -1.4561e+00],
[ 2.2558e+00, 1.3958e+00, 1.8427e+00, ..., 3.8293e+00,
3.1214e+00, 2.2232e+00],
[ 1.4154e+00, 1.4880e+00, 7.3290e-01, ..., 3.1943e+00,
1.8600e+00, 1.2948e+00],
...,
[ 2.6917e+00, 1.4109e+00, 4.8124e-01, ..., 3.2392e+00,
1.8519e+00, 3.3593e-01],
[ 2.9740e+00, 4.3158e+00, 3.2592e+00, ..., 3.5828e+00,
1.5044e+00, 1.9208e+00],
[ 2.0242e+00, 1.6870e+00, 3.7708e-01, ..., 4.2934e-01,
3.4937e-01, 9.9443e-01]]],
[[[-2.7576e-01, -3.4356e-01, -1.9390e+00, ..., -3.2762e+00,
-4.2048e+00, -1.5176e+00],
[ 1.1477e-01, -1.0138e+00, -1.3164e+00, ..., 4.7404e-01,
3.3331e+00, 7.0682e-02],
[-6.0793e-01, -1.5001e+00, -4.8583e-02, ..., -3.1244e+00,
1.1012e+00, 3.5447e+00],
...,
[-1.6062e+00, -2.6216e-01, -2.2735e-01, ..., -2.2896e+00,
-2.4457e+00, -2.8733e-01],
[ 5.0573e-01, 1.3151e+00, 1.8458e+00, ..., -1.8435e+00,
-2.5147e+00, 2.9932e-01],
[-2.5182e+00, -3.5952e+00, -2.2110e+00, ..., -4.1952e+00,
-3.6952e+00, -1.4592e+00]],
[[-5.2178e-01, -1.5273e+00, -2.8688e+00, ..., -5.6161e+00,
-3.7896e+00, -1.4198e+00],
[-2.4515e+00, -2.0542e+00, -2.6284e+00, ..., -1.2140e+00,
3.8205e-01, -1.9597e+00],
[-3.3063e+00, -2.0265e+00, -7.1818e-01, ..., -2.2650e+00,
-1.1458e+00, -1.2904e-01],
...,
[ 1.8908e+00, 2.5183e+00, 1.8962e+00, ..., -8.1000e+00,
-8.9174e+00, -1.8462e+00],
[-9.5316e+00, -1.0565e+01, -9.9275e+00, ..., -9.9317e+00,
-1.0531e+01, -3.7429e+00],
[-7.1658e-01, -2.3963e+00, -2.5980e+00, ..., -4.3783e+00,
-4.3857e+00, -7.0249e-01]],
[[ 1.3560e+00, 2.1918e+00, 1.4369e+00, ..., 4.4513e+00,
6.2502e+00, 2.2044e+00],
[ 9.5995e-01, 2.3305e+00, 1.9405e+00, ..., -2.8272e+00,
4.0926e+00, 5.1191e+00],
[ 1.6400e+00, 1.9519e+00, 1.9726e+00, ..., -3.4710e+00,
-5.7397e+00, 3.9116e+00],
...,
[-1.0817e+00, 1.5953e+00, 1.4259e+00, ..., 3.0020e+00,
-2.7546e-01, -3.8240e-01],
[-2.3059e+00, -2.1724e+00, -2.0168e+00, ..., -1.9106e-01,
-4.0963e-02, -3.3822e-01],
[ 1.5201e-01, 5.4187e-01, 1.0666e-01, ..., -1.2687e+00,
-4.6831e-01, -1.8224e+00]],
...,
[[-1.5953e+00, -2.2350e+00, -2.2399e+00, ..., -4.5695e+00,
-4.8905e+00, -1.7697e+00],
[-2.9326e+00, -2.8753e+00, -1.7921e+00, ..., 2.1561e-01,
-1.5885e+00, -5.1160e+00],
[-2.4930e+00, -1.5803e+00, 9.6809e-02, ..., -3.6117e-01,
-2.2565e-01, -3.2528e+00],
...,
[ 1.9466e+00, 1.1061e+00, 8.6778e-01, ..., -5.9529e+00,
-5.2833e+00, -3.0862e+00],
[-2.2706e+00, -5.8821e+00, -4.1728e+00, ..., -3.5574e+00,
-4.8808e+00, -1.4395e+00],
[-1.1524e+00, -2.4261e+00, -2.3233e+00, ..., -3.0079e+00,
-3.0549e+00, 4.6266e-01]],
[[-2.3757e+00, -4.0320e+00, -4.4491e+00, ..., -7.6758e+00,
-8.2251e+00, 2.2707e-01],
[-4.1924e+00, -5.3611e+00, -5.8071e+00, ..., -3.8734e+00,
-1.2099e+01, -4.3647e+00],
[-3.9938e+00, -3.4607e+00, -4.5458e+00, ..., -2.3134e+00,
-4.8426e+00, -4.0974e+00],
...,
[-1.6423e+00, -1.5881e+00, -3.0516e+00, ..., -5.7868e+00,
-2.7673e+00, -9.9308e-01],
[-1.8908e+00, -8.9566e-01, -1.7165e+00, ..., -2.8769e+00,
-1.0267e+00, 3.9368e-01],
[-4.5196e-01, 6.1656e-01, -2.3904e-01, ..., -1.6569e+00,
-1.1454e+00, 5.0720e-01]],
[[ 7.1480e-01, -3.8625e-01, 3.5992e-01, ..., 1.8981e-01,
2.7143e+00, 1.5221e+00],
[ 2.3922e+00, -6.8483e-01, -1.7587e-01, ..., -2.0703e+00,
9.9437e-01, 6.3950e+00],
[ 2.1719e+00, -1.1491e+00, -1.4608e+00, ..., -6.8514e+00,
-2.4077e+00, 3.2187e+00],
...,
[-2.3619e+00, -3.9664e+00, -3.5374e+00, ..., 2.3537e+00,
2.4033e+00, 5.1364e-01],
[ 5.2502e+00, 3.8502e+00, 3.4823e+00, ..., 2.6633e+00,
2.3143e+00, 1.4377e+00],
[ 7.9714e-01, -3.7435e-01, 3.2675e-01, ..., 7.2157e-01,
5.0609e-01, 9.2534e-01]]],
[[[ 4.0398e-01, 1.0830e+00, 1.2453e+00, ..., 4.1261e-01,
6.5669e-01, -4.7220e-01],
[-1.9235e-01, -2.8768e+00, 1.2607e-02, ..., -9.1949e-01,
-3.0408e-01, -1.6784e+00],
[-4.7696e+00, -3.3719e+00, 1.5618e+00, ..., -4.0408e+00,
-3.9254e+00, -3.5049e+00],
...,
[-4.7874e-01, 3.4731e-01, 3.9038e-01, ..., -3.1642e-01,
-2.6885e-01, 1.2492e+00],
[-1.4080e+00, 4.9032e-02, -6.6934e-02, ..., -4.7027e-03,
-3.1886e-01, -1.1903e+00],
[-1.1351e+00, -2.1334e+00, -1.8008e+00, ..., -3.7283e+00,
-3.4774e+00, -3.3627e+00]],
[[-1.5477e+00, -2.2844e+00, -3.9534e+00, ..., -5.0649e+00,
-5.9761e+00, -3.6224e+00],
[-5.8021e+00, -5.9117e+00, -5.4807e+00, ..., -2.4526e+00,
-4.5190e+00, -3.4564e+00],
[-6.5787e+00, -3.4742e+00, -7.4141e-01, ..., -5.5872e+00,
-5.0353e+00, -2.0282e+00],
...,
[-4.2030e+00, -4.6963e+00, -5.3254e+00, ..., -8.5097e+00,
-5.0454e+00, -3.5040e+00],
[-7.5259e+00, -7.7659e+00, -6.5065e+00, ..., -4.8858e+00,
-6.4159e+00, -5.5938e+00],
[-1.3420e+00, -3.0200e+00, -2.0073e+00, ..., -3.7956e+00,
-3.4265e+00, -2.0955e+00]],
[[-2.0065e-01, 1.6708e+00, 1.8388e+00, ..., 1.7635e+00,
1.6661e+00, 6.7653e-01],
[-1.3960e+00, -1.1870e-01, 2.6890e+00, ..., 1.8823e+00,
1.6793e+00, 9.7431e-01],
[ 1.4317e+00, 2.4996e+00, 3.3175e-01, ..., 3.9769e+00,
6.0148e+00, 3.9375e+00],
...,
[-1.1478e+00, 2.7403e-01, 2.0129e-01, ..., -3.8551e+00,
-2.1906e+00, -1.6392e+00],
[-1.5070e+00, 7.6614e-02, 1.1029e+00, ..., -6.0861e-01,
-1.0469e+00, -6.7475e-01],
[-5.7867e-01, -3.8100e-01, 1.5004e-01, ..., 5.4348e-01,
1.1930e+00, 3.7204e-01]],
...,
[[-1.2588e+00, -2.3549e+00, -1.4384e+00, ..., -2.0057e+00,
-2.5321e+00, -9.1973e-01],
[-4.0881e+00, -3.5917e+00, -3.7485e+00, ..., -2.5731e+00,
-3.7595e+00, -3.0812e+00],
[-4.3989e+00, -1.5713e-01, -1.3553e+00, ..., -4.9173e+00,
-5.8919e+00, -3.5038e+00],
...,
[-1.0881e+00, -2.8804e+00, -1.6372e+00, ..., -5.4380e+00,
-4.7130e+00, -1.9240e+00],
[-1.9024e+00, -4.1777e+00, -1.5739e+00, ..., -1.7000e+00,
-4.7976e+00, -2.2048e+00],
[-1.2319e+00, -1.5119e+00, -6.4606e-01, ..., -2.5446e+00,
-2.8168e+00, -8.5904e-01]],
[[-1.6453e+00, -1.2427e+00, -2.7337e+00, ..., -7.7022e-01,
-1.9768e+00, -1.3068e+00],
[-2.8099e+00, -8.1583e-01, -4.7864e+00, ..., -2.0444e+00,
-3.2812e+00, -1.2391e+00],
[-2.5629e+00, 9.9430e-01, -1.9988e+00, ..., -6.2522e+00,
-5.8129e+00, -2.7076e+00],
...,
[-2.1890e+00, -2.5916e+00, -3.3185e+00, ..., -9.7558e-01,
-3.5535e+00, -1.0802e-01],
[-1.7836e+00, -5.1640e-01, -1.7181e+00, ..., 1.0345e-01,
-2.3196e+00, -4.7560e-01],
[-1.9811e+00, -1.5611e+00, -1.6760e+00, ..., -5.9761e-01,
-2.1545e+00, -1.1649e+00]],
[[ 1.5967e+00, 5.3372e-02, 9.5599e-01, ..., 1.2484e+00,
2.1874e+00, 9.7890e-01],
[ 3.7745e+00, 2.1341e+00, 2.2752e+00, ..., 1.1803e+00,
1.2942e+00, 1.2104e+00],
[ 3.9212e-01, 4.5485e-01, 1.1159e+00, ..., 2.9915e+00,
2.5084e+00, 1.5871e+00],
...,
[ 2.5873e+00, 3.0800e+00, 2.3491e+00, ..., 4.6497e+00,
2.0051e+00, 2.4870e+00],
[ 3.7594e+00, 4.0219e+00, 2.6286e+00, ..., 1.7373e+00,
1.0714e+00, 1.2543e+00],
[ 2.3975e+00, 2.3837e+00, 1.5939e+00, ..., 2.8138e-01,
-4.2974e-01, -1.1248e-01]]],
[[[ 3.5251e-01, 1.4500e+00, 5.9151e-01, ..., -5.1565e-01,
-2.5160e-02, 6.6121e-01],
[ 4.0137e-01, 1.1576e-01, 3.2805e-01, ..., -1.1045e+00,
-1.3173e+00, 9.4515e-01],
[-5.6425e-01, 1.2194e-01, 8.7232e-01, ..., -2.7930e-01,
-1.4229e+00, 5.1230e-01],
...,
[-1.0328e+00, -2.0508e+00, 4.2199e-01, ..., -1.5684e+00,
-1.2254e+00, -1.4958e+00],
[-1.0703e+00, -1.1440e+00, -1.1750e-01, ..., -9.1769e-01,
-3.0201e-01, -1.5849e+00],
[-2.7538e+00, -5.2260e+00, -4.4544e+00, ..., -4.5946e+00,
-4.0863e+00, -3.8235e+00]],
[[-7.6730e-01, -2.3045e+00, -4.2266e+00, ..., -3.4588e+00,
-5.0826e+00, -3.8999e-01],
[-4.9148e+00, -5.9484e+00, -5.5785e+00, ..., -5.7199e+00,
-7.4481e+00, -1.6242e+00],
[-3.5955e+00, -4.1519e+00, -4.3667e+00, ..., -5.5368e+00,
-7.5708e+00, -2.2287e+00],
...,
[-5.7877e+00, -5.6887e+00, -4.0106e+00, ..., -5.1471e+00,
-5.5177e+00, -3.9099e+00],
[-7.0847e+00, -7.8022e+00, -6.1575e+00, ..., -7.5784e+00,
-7.9820e+00, -6.4145e+00],
[-2.5443e+00, -4.4492e+00, -4.0431e+00, ..., -3.8714e+00,
-4.0909e+00, -2.5443e+00]],
[[-3.3578e-01, 2.1130e+00, 1.2308e+00, ..., 8.9664e-01,
2.1562e+00, 9.4717e-01],
[-1.5006e-01, 1.0234e+00, 8.8643e-01, ..., 8.0031e-01,
1.1276e+00, -6.0384e-01],
[ 4.3799e-02, 2.0193e+00, 1.7212e+00, ..., -6.2854e-01,
1.4953e+00, 1.5030e-01],
...,
[ 8.6545e-01, 1.1055e+00, 1.2612e+00, ..., 6.1452e-01,
-2.5424e-01, 2.6942e-01],
[ 8.9271e-01, 7.8729e-01, 8.2676e-01, ..., 6.5737e-01,
4.3329e-01, 3.2798e-01],
[ 1.3758e+00, 1.5752e+00, 1.2805e+00, ..., 1.5188e+00,
1.0837e+00, 1.3779e-01]],
...,
[[-8.1635e-01, -1.8359e+00, -1.9010e+00, ..., -2.5995e+00,
-2.9251e+00, -4.6120e-01],
[-3.3295e+00, -4.2713e+00, -2.8862e+00, ..., -4.2967e+00,
-4.5149e+00, -2.5330e+00],
[-1.5382e+00, -2.0127e+00, -1.1413e+00, ..., -1.9178e+00,
-4.3538e+00, -1.8023e+00],
...,
[-2.8182e+00, -2.7278e+00, -6.0614e-01, ..., -1.9069e+00,
-4.9328e+00, -2.4967e+00],
[-3.0707e+00, -3.1844e+00, -2.0220e+00, ..., -2.4376e+00,
-4.2878e+00, -2.3540e+00],
[-3.5708e+00, -3.8103e+00, -3.2433e+00, ..., -2.8975e+00,
-3.6068e+00, -1.6688e+00]],
[[-1.7055e+00, -2.1875e+00, -2.9325e+00, ..., -1.5337e+00,
-1.4591e+00, -1.3396e+00],
[-4.4434e+00, -3.4073e+00, -2.8587e+00, ..., -3.3454e+00,
-2.1324e+00, -1.1213e+00],
[-3.4038e+00, -2.8549e+00, -3.1084e+00, ..., -9.1718e-01,
-2.2205e+00, -1.1203e+00],
...,
[-4.0800e+00, -2.3320e+00, -3.0445e+00, ..., -7.0709e-01,
-2.8890e+00, -1.2267e+00],
[-3.7226e+00, -8.0481e-01, -2.3238e+00, ..., -7.8484e-01,
-2.7366e+00, -1.2315e+00],
[-4.4644e+00, -3.0304e+00, -2.9513e+00, ..., -2.3166e+00,
-3.3157e+00, -1.2133e+00]],
[[ 1.0804e+00, 5.7835e-01, 6.8993e-01, ..., -5.3530e-01,
1.1761e-01, -4.6664e-01],
[ 3.5499e+00, 3.2704e+00, 2.3175e+00, ..., 1.7613e+00,
2.0017e+00, 2.4306e+00],
[ 1.3999e+00, 6.3969e-01, 5.5785e-01, ..., 8.5362e-01,
1.1298e+00, 1.4969e+00],
...,
[ 2.8486e+00, 1.0749e+00, 1.3230e+00, ..., 2.0645e+00,
2.4195e+00, 1.2401e+00],
[ 3.7239e+00, 3.5210e+00, 2.3738e+00, ..., 2.9313e+00,
2.2819e+00, 1.9890e+00],
[ 1.9526e+00, 7.0394e-01, -5.6218e-03, ..., -2.8976e-01,
-2.4333e-01, -1.4038e-01]]],
[[[-1.2634e-01, 1.1782e+00, 5.8456e-01, ..., 4.3561e-01,
-2.1283e-01, 8.2776e-01],
[-2.3606e-03, -3.9259e-01, -3.1306e-01, ..., 4.5071e-01,
-2.1814e+00, 9.2343e-01],
[-4.3047e-01, -3.7098e-01, -1.6123e+00, ..., -9.3533e-01,
-1.6545e+00, 2.8464e-01],
...,
[ 1.5231e-01, -5.2385e+00, -8.0825e-01, ..., 4.0199e+00,
7.3340e-01, -1.5365e+00],
[-1.7202e+00, -6.6801e+00, -1.2480e+00, ..., -1.3012e-01,
-3.0220e+00, -1.7147e+00],
[-1.9139e-01, -5.7087e+00, -1.4646e+00, ..., -2.5643e+00,
-2.5972e+00, -9.5948e-01]],
[[-1.9923e+00, -3.0861e+00, -5.1278e+00, ..., -3.4138e+00,
-5.0514e+00, -2.1040e-01],
[-6.3964e+00, -6.8933e+00, -7.4611e+00, ..., -5.1783e+00,
-8.7996e+00, -1.6664e+00],
[-5.0738e+00, -6.4372e+00, -8.5814e+00, ..., -7.0663e+00,
-9.3866e+00, -3.2172e+00],
...,
[-3.9858e+00, -8.5793e+00, 1.0752e+00, ..., -4.8317e-01,
-4.0046e+00, -3.0665e+00],
[-7.0846e+00, -1.0515e+01, 1.1053e+00, ..., -9.1689e+00,
-1.0956e+01, -4.0327e+00],
[ 4.6496e-01, -5.0059e+00, 3.4322e-01, ..., -2.2760e+00,
-2.9041e+00, -2.1434e-01]],
[[-3.1851e-01, 1.9913e+00, 1.0785e+00, ..., 1.3386e+00,
1.7687e+00, 9.8530e-01],
[ 2.7660e-01, 6.7490e-01, -7.4495e-03, ..., 1.3075e+00,
9.9328e-01, -6.3189e-01],
[ 7.5660e-01, 1.4186e+00, 1.2622e-01, ..., 1.5753e+00,
1.7089e+00, -1.8131e-01],
...,
[-7.0888e-02, 3.2168e+00, 3.4430e-02, ..., -5.2415e-02,
2.7983e+00, 1.6519e+00],
[-1.5339e-01, 2.1714e+00, -1.6635e+00, ..., -1.2241e-01,
2.0618e+00, 7.5578e-01],
[-1.5943e+00, 1.3182e+00, -4.4869e+00, ..., 7.0182e-01,
7.2730e-02, -1.2679e+00]],
...,
[[-1.4586e+00, -1.7330e+00, -1.7836e+00, ..., -1.5734e+00,
-2.8427e+00, -4.1850e-01],
[-4.2333e+00, -3.8545e+00, -3.2633e+00, ..., -2.9923e+00,
-4.2036e+00, -2.6504e+00],
[-2.7889e+00, -2.2042e+00, -2.4375e+00, ..., -3.2746e+00,
-4.5873e+00, -1.7855e+00],
...,
[-1.9515e+00, -3.3621e+00, 1.2468e+00, ..., -1.6134e+00,
-5.7504e+00, -2.5030e+00],
[-3.6202e+00, -5.5359e+00, 8.2765e-01, ..., -3.3887e+00,
-5.6513e+00, -1.5220e+00],
[-1.1309e+00, -2.8295e+00, 2.2405e+00, ..., -1.5371e-01,
-9.9097e-01, 5.6486e-01]],
[[-1.9071e+00, -1.4521e+00, -3.0579e+00, ..., -6.8520e-01,
-1.5479e+00, -1.3762e+00],
[-5.0139e+00, -2.4409e+00, -4.3196e+00, ..., -3.2780e+00,
-1.7227e+00, -8.5397e-01],
[-3.7712e+00, -1.2238e+00, -3.3823e+00, ..., -1.3286e+00,
-2.8446e-01, -5.1267e-01],
...,
[-4.2415e+00, -1.7000e+00, -3.9818e+00, ..., -2.1805e+00,
-5.6054e+00, -1.8375e+00],
[-5.0930e+00, 1.1828e+00, -1.5860e+00, ..., -2.7202e+00,
-3.8518e+00, -8.5343e-01],
[-3.3222e+00, 3.4196e-01, 2.2621e+00, ..., -3.8014e-01,
-6.6467e-01, 1.5507e-01]],
[[ 1.7126e+00, 6.3531e-01, 1.0384e+00, ..., 9.3393e-01,
-7.3406e-02, -5.2179e-01],
[ 4.5434e+00, 3.4534e+00, 2.9735e+00, ..., 3.2734e+00,
2.1177e+00, 2.1313e+00],
[ 2.1987e+00, 2.0685e+00, 2.0256e+00, ..., 2.2762e+00,
1.7466e+00, 1.5080e+00],
...,
[ 2.6981e+00, 2.3727e+00, -1.6081e+00, ..., 7.8909e-01,
1.2310e+00, 3.0378e-01],
[ 3.7798e+00, 3.7260e+00, -1.2679e+00, ..., 3.3436e+00,
1.1792e+00, 1.7585e+00],
[ 2.0491e+00, 9.8652e-01, 8.0846e-01, ..., -1.4536e-01,
1.9992e-01, 1.6318e+00]]]], device='cuda:0'),
tensor([[[[ 1.2856e+00, 2.3037e+00, 1.7244e-01, ..., -3.5357e+00,
-2.2904e+00, -7.8003e-01],
[-2.1018e+00, -1.8943e+00, 1.4764e+00, ..., -3.8806e+00,
-2.7211e+00, -9.1439e-02],
[-8.0293e-01, -1.2575e+00, -2.5549e+00, ..., -2.6661e+00,
-3.1160e+00, -2.9744e-01],
...,
[ 2.2396e-01, -4.1149e+00, -3.4416e+00, ..., 5.0063e+00,
3.4449e+00, -1.8571e+00],
[-2.7232e-01, -1.8513e+00, -2.0665e+00, ..., -2.8047e+00,
-5.2329e+00, -2.0382e+00],
[-9.5967e-01, -4.0769e+00, -3.7860e+00, ..., -3.5829e+00,
-3.0950e+00, -1.8006e+00]],
[[ 9.2762e-01, -3.1872e+00, -6.5225e+00, ..., -6.8148e+00,
-7.0102e+00, -1.1438e+00],
[-6.0634e+00, -3.5275e+00, -5.7356e+00, ..., -9.3793e+00,
-8.2225e+00, -2.4807e+00],
[-3.6839e+00, -6.1319e+00, -4.2131e+00, ..., -1.1950e+01,
-9.6710e+00, -3.3917e+00],
...,
[-5.7590e+00, -9.6045e+00, -6.9256e+00, ..., -3.3905e+00,
-7.5145e+00, -5.1171e+00],
[-5.0714e+00, -6.9772e+00, -5.3876e+00, ..., -1.3535e+01,
-1.5712e+01, -5.4550e+00],
[-5.5145e-01, -3.2726e+00, -3.2971e+00, ..., -2.8795e+00,
-3.2642e+00, -8.6509e-01]],
[[ 7.5713e-01, 2.1105e+00, 1.7063e+00, ..., 5.9628e-01,
2.1389e+00, 9.0024e-01],
[ 1.9632e+00, 4.8129e+00, 2.0544e+00, ..., 9.8294e-01,
2.6597e+00, -1.1762e-01],
[ 2.1805e-02, 6.9881e+00, 1.0613e+01, ..., 4.0278e-01,
1.3677e+00, -4.9024e-01],
...,
[-1.8152e+00, 2.3033e-01, 2.7529e+00, ..., -3.7015e+00,
-1.8655e+00, 5.7023e-01],
[ 1.0781e+00, 4.1566e+00, 6.4629e-01, ..., -4.7996e+00,
-6.3894e-01, -1.7832e+00],
[ 2.9709e-01, 1.3127e+00, 7.1904e-01, ..., -4.5641e-01,
-4.3953e-01, -3.1009e+00]],
...,
[[-6.0941e-01, -8.9533e-01, -2.4515e+00, ..., -5.4245e+00,
-4.8951e+00, -5.5056e-01],
[-5.2863e+00, -6.2139e+00, -3.3441e+00, ..., -6.2742e+00,
-5.4348e+00, -1.9471e+00],
[-1.1039e+00, -4.8435e+00, -5.9635e+00, ..., -4.6414e+00,
-7.5721e+00, -1.1304e+00],
...,
[-4.9138e+00, -6.8745e+00, -1.0348e+00, ..., -1.0208e+00,
-7.2183e+00, -4.9183e+00],
[-2.2766e+00, -3.7968e+00, -2.6356e+00, ..., -3.8775e+00,
-7.8384e+00, -3.1787e+00],
[-1.2624e+00, -3.1751e+00, -2.6320e+00, ..., 1.4219e+00,
-1.7575e+00, -4.4583e-01]],
[[-1.3823e+00, -2.6042e-01, -1.5097e+00, ..., -2.3601e+00,
-6.0643e-01, -6.7765e-01],
[-6.8738e+00, -5.2951e+00, 4.0343e-01, ..., -4.1940e+00,
-2.1509e+00, -2.5916e-01],
[-6.0463e+00, -1.4283e+01, -1.0589e+01, ..., -2.6077e+00,
-2.7803e+00, -1.4254e+00],
...,
[-2.8458e+00, -3.3268e+00, -3.5235e+00, ..., 2.1963e+00,
-4.8487e+00, -3.8383e+00],
[-3.1757e+00, -2.2001e+00, -1.2751e+00, ..., -1.2485e+00,
-2.0054e+00, -1.5017e-01],
[-3.1973e+00, -2.4627e+00, -8.8471e-01, ..., 2.4341e+00,
1.2437e-01, 4.3536e-01]],
[[ 1.1590e+00, 2.1757e+00, 2.3628e+00, ..., -4.7275e-01,
5.2126e-01, -1.4918e+00],
[ 1.3567e+00, 2.9667e+00, 3.9237e+00, ..., 2.7459e+00,
2.3779e+00, 1.9737e+00],
[-1.0037e+00, 8.9863e-01, 6.3165e+00, ..., 3.4383e+00,
1.3793e+00, 1.4137e+00],
...,
[ 5.8511e+00, 3.6308e+00, 1.0235e+00, ..., 8.8594e-01,
5.4181e+00, 2.1812e+00],
[ 2.1764e+00, 1.7532e+00, 2.3225e+00, ..., 4.0474e+00,
3.0400e+00, 2.5207e+00],
[ 1.9868e+00, 9.8477e-01, 6.3239e-01, ..., -1.4846e+00,
2.4836e-01, 1.8094e+00]]],
[[[ 2.3911e-01, 1.3711e+00, 7.5185e-01, ..., -2.8882e+00,
-1.9649e+00, -1.5832e-01],
[ 4.5640e-01, 3.0101e-02, 5.8263e-01, ..., -1.3241e+00,
6.8342e-01, 1.5685e+00],
[-7.7366e-01, -2.4829e-02, 4.4976e-01, ..., -2.6010e+00,
2.5730e+00, 1.5264e+00],
...,
[-1.0886e+00, -2.2190e+00, -4.4264e+00, ..., -5.2698e-01,
3.9225e-01, -7.8791e-01],
[-1.0048e+00, -3.4598e-01, -1.2326e+00, ..., -5.3841e-01,
7.5544e-02, -1.6069e+00],
[-2.5731e+00, -4.1935e+00, -3.8582e+00, ..., -4.9426e+00,
-4.1176e+00, -3.8293e+00]],
[[-8.2313e-01, -2.3015e+00, -4.0961e+00, ..., -3.2966e+00,
-4.1619e+00, -2.5499e-01],
[-5.0583e+00, -6.7280e+00, -6.4303e+00, ..., -3.1512e+00,
-2.4873e+00, -8.5666e-01],
[-4.1141e+00, -4.6849e+00, -4.4538e+00, ..., -3.6269e+00,
-5.1141e-01, -6.1365e-01],
...,
[-5.5590e+00, -4.0497e+00, -4.7518e+00, ..., -4.6050e+00,
-4.8170e+00, -3.7129e+00],
[-6.9357e+00, -6.8261e+00, -4.8738e+00, ..., -7.1512e+00,
-7.9456e+00, -6.4421e+00],
[-2.4066e+00, -4.2173e+00, -3.7294e+00, ..., -4.0113e+00,
-4.1787e+00, -2.5877e+00]],
[[-2.9955e-01, 2.0692e+00, 1.1929e+00, ..., 2.8392e+00,
3.0959e+00, 1.6373e+00],
[-2.0590e-01, 1.1651e+00, 1.1751e+00, ..., -2.7900e+00,
8.1541e-01, 1.2433e+00],
[-6.4003e-02, 1.9537e+00, 1.7790e+00, ..., -8.1488e+00,
-9.8748e-01, 1.5198e+00],
...,
[ 8.7376e-01, 7.4820e-01, 9.9604e-01, ..., 1.4998e+00,
1.0724e+00, 2.5899e-01],
[ 5.6559e-01, 3.4052e-01, -4.0609e-03, ..., 9.5478e-01,
3.0771e-01, 2.4211e-01],
[ 9.6382e-01, 1.2705e+00, 3.2214e-01, ..., 1.6863e+00,
1.0349e+00, 1.2968e-01]],
...,
[[-9.1554e-01, -1.8025e+00, -1.7293e+00, ..., -2.3618e+00,
-2.4784e+00, -1.1075e+00],
[-3.3099e+00, -4.2161e+00, -2.7850e+00, ..., 4.8271e-01,
-1.4717e+00, -3.5246e+00],
[-1.6617e+00, -2.0810e+00, -1.6687e+00, ..., 1.9100e+00,
-1.7437e+00, -2.6497e+00],
...,
[-2.3215e+00, -2.5917e+00, -1.5256e+00, ..., -2.4816e+00,
-3.3542e+00, -1.6352e+00],
[-2.8373e+00, -2.3833e+00, -1.4383e+00, ..., -2.2068e+00,
-4.0000e+00, -2.3335e+00],
[-3.2610e+00, -3.0321e+00, -1.3944e+00, ..., -3.3480e+00,
-3.6873e+00, -1.6949e+00]],
[[-1.7099e+00, -2.1240e+00, -2.8350e+00, ..., -1.9100e+00,
-1.1102e+00, 1.7390e-01],
[-4.3861e+00, -3.6749e+00, -3.8258e+00, ..., 1.1315e-01,
-1.4895e+00, 8.5163e-01],
[-3.5884e+00, -2.6757e+00, -3.3093e+00, ..., 1.9703e+00,
-1.5155e+00, -3.4946e-03],
...,
[-3.3939e+00, -2.4641e+00, -4.2463e+00, ..., -1.3061e+00,
-3.3849e+00, -1.8238e+00],
[-3.0166e+00, -4.9174e-01, -4.2294e+00, ..., -5.1970e-01,
-2.9023e+00, -1.3070e+00],
[-3.8482e+00, -2.4609e+00, -2.5574e+00, ..., -1.9858e+00,
-3.2515e+00, -1.2255e+00]],
[[ 1.0935e+00, 5.5223e-01, 7.8808e-01, ..., -6.3191e-01,
1.2372e+00, 2.3883e-01],
[ 3.7606e+00, 3.1091e+00, 2.7448e+00, ..., -1.0827e+00,
2.9079e+00, 5.2919e+00],
[ 1.6594e+00, 7.2152e-01, 1.2842e+00, ..., -2.4087e+00,
2.1644e+00, 3.9054e+00],
...,
[ 2.7712e+00, 9.7113e-01, -1.8706e-01, ..., 2.1716e+00,
1.9054e+00, 1.2275e+00],
[ 3.4629e+00, 2.9944e+00, 1.4750e+00, ..., 2.6995e+00,
2.3466e+00, 2.0677e+00],
[ 1.9597e+00, 1.1930e+00, -1.2635e-01, ..., 2.8062e-02,
-1.5825e-01, -1.3040e-01]]],
[[[ 2.9985e-01, 1.2713e+00, 2.2198e-01, ..., -3.2708e+00,
-2.5982e+00, -1.0775e+00],
[ 5.7197e-01, 1.0547e+00, 1.1965e+00, ..., -1.6421e+00,
-2.2859e+00, -2.4542e-01],
[ 2.6651e-01, 5.8866e-02, -1.3597e-01, ..., -9.8439e-01,
-2.5614e+00, 1.0266e-01],
...,
[-6.0131e-01, -1.8211e+00, -3.1229e+00, ..., -1.1271e+00,
-2.4275e+00, 2.2190e-01],
[-2.0190e+00, -2.0124e+00, -2.3139e+00, ..., -3.6808e+00,
-3.4682e+00, 4.9565e-01],
[-1.3209e+00, -3.9403e+00, -4.6318e+00, ..., -3.6442e+00,
-3.1859e+00, -1.4505e+00]],
[[-9.3239e-01, -1.5347e+00, -3.2408e+00, ..., -6.4527e+00,
-7.3064e+00, -1.3330e+00],
[-3.6137e+00, -4.1144e+00, -3.5393e+00, ..., -5.9196e+00,
-8.1323e+00, -2.3689e+00],
[-3.9115e+00, -5.3382e+00, -5.9262e+00, ..., -6.0534e+00,
-8.0207e+00, -2.4333e+00],
...,
[-7.5540e+00, -1.4131e+01, -1.5309e+01, ..., -4.7679e+00,
-7.0718e+00, -1.4906e+00],
[-8.1963e+00, -7.1203e+00, -7.1409e+00, ..., -9.7185e+00,
-1.0811e+01, -4.7040e+00],
[-1.9115e+00, -3.8590e+00, -4.4859e+00, ..., -3.4958e+00,
-3.6823e+00, -7.5510e-01]],
[[-3.7898e-02, 1.2476e+00, 1.5873e+00, ..., 1.6417e+00,
2.4570e+00, 9.6115e-01],
[ 2.2391e-03, 1.7477e-02, 5.2084e-01, ..., 1.6616e+00,
3.1102e+00, 1.2358e-01],
[ 1.1019e+00, 1.5372e+00, 1.2974e+00, ..., -3.2660e-02,
1.7164e+00, -3.8520e-01],
...,
[ 1.8678e-01, -1.4533e+00, -2.0133e+00, ..., 5.1911e-01,
1.3467e+00, -1.2475e+00],
[ 3.4424e-01, 1.3217e+00, 1.7373e+00, ..., -9.4612e-01,
2.7565e-01, -1.5678e+00],
[ 1.1656e+00, 1.6381e+00, 1.0385e+00, ..., -2.0725e-01,
-6.7130e-01, -1.9403e+00]],
...,
[[-5.3014e-01, -1.5478e+00, -2.1266e+00, ..., -5.1988e+00,
-5.2719e+00, -7.6264e-01],
[-2.5639e+00, -2.6416e+00, -1.5140e+00, ..., -3.5922e+00,
-4.7789e+00, -1.7002e+00],
[-1.8668e+00, -1.6924e+00, -1.7225e+00, ..., -4.4850e+00,
-5.2581e+00, -1.2019e+00],
...,
[-5.3979e+00, -7.5543e+00, -8.1146e+00, ..., -1.2658e+00,
-3.9106e+00, -1.4743e+00],
[-3.5391e+00, -9.1097e-01, -1.0180e+00, ..., -3.9094e+00,
-5.3023e+00, -1.2098e+00],
[-1.8875e+00, -2.5398e+00, -3.0778e+00, ..., -1.4844e+00,
-2.1137e+00, 3.2653e-01]],
[[-1.2243e+00, -2.2933e+00, -3.9700e+00, ..., -1.6227e+00,
-4.1952e-01, -5.1960e-01],
[-3.5288e+00, -3.4916e+00, -4.5796e+00, ..., -3.4218e+00,
-1.8777e+00, -3.2937e-01],
[-4.2396e+00, -3.7684e+00, -3.8722e+00, ..., -2.2241e+00,
-1.8198e+00, -6.2160e-01],
...,
[-3.9996e+00, -3.4664e+00, -6.2150e+00, ..., -3.0714e+00,
-3.2720e+00, -1.6749e+00],
[-3.9023e+00, -3.8524e-01, -3.3014e+00, ..., -4.2735e-01,
-1.3445e-01, 1.2705e+00],
[-3.4748e+00, -3.2820e+00, -2.4659e+00, ..., -6.3938e-01,
-9.1564e-01, 6.6049e-01]],
[[ 1.1642e+00, -4.4661e-02, 1.1031e+00, ..., 5.7755e-01,
7.7162e-01, -1.4068e+00],
[ 2.1561e+00, 1.1696e+00, 1.4431e+00, ..., 3.6414e+00,
3.1157e+00, 2.3988e+00],
[ 1.6686e+00, 1.4898e+00, 1.0373e+00, ..., 3.1202e+00,
1.6681e+00, 1.3597e+00],
...,
[ 4.6361e+00, 7.0203e+00, 6.8571e+00, ..., 1.5777e+00,
2.2307e+00, 1.3005e+00],
[ 2.2356e+00, 2.7914e+00, -8.0834e-01, ..., 1.5229e+00,
3.3507e-01, 1.5043e+00],
[ 2.4894e+00, 2.0846e+00, 9.0388e-01, ..., 3.9999e-01,
8.8712e-02, 8.7704e-01]]],
[[[ 2.1303e-01, 1.2829e+00, 6.5467e-01, ..., -5.6329e-01,
-1.6390e-02, 6.9784e-01],
[ 2.9381e-01, -8.8419e-03, 4.2364e-01, ..., -1.3347e+00,
-1.1863e+00, 1.1043e+00],
[-6.4302e-01, -3.0761e-02, 8.3441e-01, ..., -1.5131e+00,
-1.6299e+00, 8.6291e-01],
...,
[-8.8085e-01, -1.8991e+00, 2.1183e-01, ..., -1.4019e+00,
1.4892e-01, -5.7263e-01],
[-9.9833e-01, -1.0079e+00, -8.5480e-02, ..., -6.9107e-01,
7.5295e-02, -1.6119e+00],
[-2.7276e+00, -5.2246e+00, -4.4015e+00, ..., -4.6577e+00,
-4.1759e+00, -3.8687e+00]],
[[-9.1725e-01, -2.3834e+00, -4.1371e+00, ..., -3.4758e+00,
-5.0014e+00, -3.3233e-01],
[-4.9660e+00, -6.1967e+00, -4.9932e+00, ..., -5.2238e+00,
-7.0491e+00, -1.6017e+00],
[-3.6973e+00, -4.3089e+00, -3.6548e+00, ..., -5.7795e+00,
-6.6057e+00, -1.4785e+00],
...,
[-5.6138e+00, -5.4204e+00, -3.4528e+00, ..., -4.7393e+00,
-4.9041e+00, -3.5926e+00],
[-7.0532e+00, -7.8971e+00, -6.2951e+00, ..., -7.2792e+00,
-8.0229e+00, -6.5311e+00],
[-2.5131e+00, -4.4409e+00, -4.0610e+00, ..., -3.9382e+00,
-4.1672e+00, -2.5604e+00]],
[[-3.4729e-01, 2.0734e+00, 1.1839e+00, ..., 9.1247e-01,
2.1646e+00, 9.7127e-01],
[-1.0736e-01, 9.9785e-01, 7.6680e-01, ..., 6.2527e-01,
1.0863e+00, -4.0046e-01],
[ 7.5771e-02, 1.8769e+00, 1.3344e+00, ..., 3.0681e-01,
1.7788e+00, 1.3745e-01],
...,
[ 7.9621e-01, 1.1137e+00, 1.1016e+00, ..., 4.8069e-01,
9.9006e-01, 2.1585e-01],
[ 7.8657e-01, 6.3132e-01, 7.5946e-01, ..., 9.7387e-01,
3.6953e-01, 1.2175e-01],
[ 1.3465e+00, 1.6228e+00, 1.2880e+00, ..., 1.6840e+00,
1.1606e+00, 1.2744e-01]],
...,
[[-9.1881e-01, -1.8419e+00, -1.8599e+00, ..., -2.6050e+00,
-2.8756e+00, -5.2331e-01],
[-3.3775e+00, -4.2561e+00, -2.6720e+00, ..., -3.9650e+00,
-3.7526e+00, -2.5968e+00],
[-1.6398e+00, -2.0909e+00, -1.0052e+00, ..., -2.4554e+00,
-4.3706e+00, -1.3448e+00],
...,
[-2.5156e+00, -2.4494e+00, -2.5074e-01, ..., -1.9565e+00,
-3.4435e+00, -1.4568e+00],
[-2.9851e+00, -3.1894e+00, -1.7424e+00, ..., -2.2904e+00,
-4.0169e+00, -2.3161e+00],
[-3.5035e+00, -3.8546e+00, -3.3420e+00, ..., -3.0444e+00,
-3.6945e+00, -1.6320e+00]],
[[-1.7339e+00, -2.1071e+00, -2.9344e+00, ..., -1.5241e+00,
-1.4410e+00, -1.3601e+00],
[-4.4352e+00, -3.3330e+00, -3.1021e+00, ..., -3.1622e+00,
-1.9021e+00, -9.2260e-01],
[-3.5199e+00, -2.8533e+00, -3.0826e+00, ..., -1.4431e+00,
-1.7718e+00, -5.2134e-01],
...,
[-3.9521e+00, -2.5088e+00, -3.1423e+00, ..., -1.1879e+00,
-2.7423e+00, -1.7238e+00],
[-3.6123e+00, -7.8586e-01, -2.1084e+00, ..., -8.6830e-01,
-2.8868e+00, -1.3406e+00],
[-4.4177e+00, -2.9882e+00, -2.6775e+00, ..., -2.1966e+00,
-3.3167e+00, -1.2140e+00]],
[[ 1.1183e+00, 5.6061e-01, 7.3314e-01, ..., -5.5727e-01,
8.2603e-02, -4.0760e-01],
[ 3.6102e+00, 3.2162e+00, 2.3068e+00, ..., 1.6545e+00,
2.0477e+00, 2.5623e+00],
[ 1.4736e+00, 5.8111e-01, 4.9266e-01, ..., 1.0821e+00,
1.6726e+00, 1.8630e+00],
...,
[ 2.7188e+00, 7.9540e-01, 9.5264e-01, ..., 2.4794e+00,
1.9075e+00, 1.1507e+00],
[ 3.7630e+00, 3.6250e+00, 2.4315e+00, ..., 2.6784e+00,
2.2182e+00, 2.0466e+00],
[ 1.8910e+00, 6.6849e-01, 2.1067e-01, ..., -2.7370e-01,
-2.6799e-01, -1.6924e-01]]],
[[[ 2.8881e-01, 1.1501e+00, -3.7807e-02, ..., -3.0521e+00,
-2.4085e+00, -1.0622e+00],
[ 1.3649e-01, 5.3368e-01, 1.0479e+00, ..., -1.4468e+00,
-2.2472e+00, -2.7408e-01],
[-8.1379e-01, -2.2553e-01, 2.5909e-01, ..., -7.2175e-01,
-2.7922e+00, -1.1421e-01],
...,
[-7.1152e-01, -2.0447e+00, -9.1471e-05, ..., 4.1845e+00,
-1.6206e+00, -1.5119e-01],
[-1.9677e+00, -1.9112e+00, -1.2999e+00, ..., -2.9978e+00,
-4.4456e+00, 1.6220e-01],
[-1.1985e+00, -3.4569e+00, -3.6238e+00, ..., -3.0797e+00,
-3.4362e+00, -1.7362e+00]],
[[-9.7077e-01, -1.5873e+00, -3.4659e+00, ..., -6.1451e+00,
-7.3100e+00, -1.3654e+00],
[-4.0218e+00, -3.9033e+00, -3.2552e+00, ..., -5.4979e+00,
-8.3469e+00, -2.4349e+00],
[-3.7310e+00, -3.7094e+00, -3.4267e+00, ..., -5.1123e+00,
-7.8625e+00, -2.5498e+00],
...,
[-6.1664e+00, -5.9465e+00, -4.5084e+00, ..., -3.2667e+00,
-8.5302e+00, -2.7002e+00],
[-7.8173e+00, -7.6944e+00, -7.0967e+00, ..., -1.1329e+01,
-1.2357e+01, -4.0611e+00],
[-1.8535e+00, -3.3511e+00, -3.6577e+00, ..., -2.8848e+00,
-3.8769e+00, -6.3285e-01]],
[[-7.9162e-02, 1.2760e+00, 1.7085e+00, ..., 1.7403e+00,
2.3641e+00, 9.2349e-01],
[-7.2106e-02, 1.0478e-01, 9.1941e-01, ..., 1.7763e+00,
3.0555e+00, 2.5309e-02],
[ 6.2055e-01, 1.2193e+00, 1.4148e+00, ..., 1.5679e-01,
2.0189e+00, -2.6016e-01],
...,
[ 3.2996e-01, 8.3878e-01, 7.7774e-01, ..., -1.5725e+00,
1.7424e+00, -2.4938e-01],
[ 5.2958e-01, 5.7879e-01, 8.5787e-01, ..., -1.3268e+00,
4.6623e-01, -1.9825e+00],
[ 6.5501e-01, 1.1793e+00, 1.5422e+00, ..., 1.4936e-01,
-9.7928e-01, -2.3759e+00]],
...,
[[-6.5977e-01, -1.6427e+00, -2.3384e+00, ..., -5.1550e+00,
-5.1456e+00, -7.5193e-01],
[-2.9054e+00, -2.6620e+00, -1.6182e+00, ..., -3.5977e+00,
-4.7812e+00, -1.8947e+00],
[-2.1659e+00, -8.7293e-01, -1.1315e+00, ..., -4.5509e+00,
-5.3360e+00, -1.3238e+00],
...,
[-1.9930e+00, -1.5326e+00, -5.6996e-01, ..., -2.1475e+00,
-6.9116e+00, -1.8681e+00],
[-3.2635e+00, -3.1830e+00, -2.6896e+00, ..., -4.2857e+00,
-6.3696e+00, -5.8928e-01],
[-1.6715e+00, -2.0631e+00, -2.3267e+00, ..., -1.0730e+00,
-2.1036e+00, 1.5401e-01]],
[[-1.3379e+00, -2.4352e+00, -3.9667e+00, ..., -1.3871e+00,
-3.9006e-01, -6.2588e-01],
[-3.7278e+00, -3.3325e+00, -4.5784e+00, ..., -3.3954e+00,
-1.7791e+00, -4.0959e-01],
[-4.3667e+00, -3.3138e+00, -4.0085e+00, ..., -2.8155e+00,
-1.7087e+00, -4.3293e-01],
...,
[-4.8889e+00, -3.8248e+00, -3.2301e+00, ..., -3.2918e+00,
-3.6231e+00, -1.0142e+00],
[-4.3268e+00, -1.8649e+00, -1.5231e+00, ..., -2.2475e+00,
-2.5594e-01, 1.9010e+00],
[-3.4086e+00, -3.2500e+00, -2.0227e+00, ..., -2.0145e-01,
-3.5317e-01, 7.9643e-01]],
[[ 1.1786e+00, -1.2766e-01, 1.1102e+00, ..., 5.8792e-01,
7.2903e-01, -1.4438e+00],
[ 2.3698e+00, 9.6791e-01, 1.3493e+00, ..., 3.9142e+00,
3.2036e+00, 2.2214e+00],
[ 1.3572e+00, 5.2884e-01, 3.4832e-01, ..., 3.2071e+00,
1.8749e+00, 1.2978e+00],
...,
[ 2.6180e+00, 1.7101e+00, 1.0445e+00, ..., 2.7624e+00,
1.8339e+00, 5.4181e-01],
[ 3.0180e+00, 3.8829e+00, 2.2700e+00, ..., 3.3951e+00,
1.3510e+00, 2.0434e+00],
[ 1.9908e+00, 1.6139e+00, 1.8598e-01, ..., -1.4399e-01,
4.2426e-01, 1.1675e+00]]]], device='cuda:0'),
tensor([[[[ 1.3033e-01, 1.1304e+00, 7.4534e-01, ..., -4.3517e-01,
3.2298e-02, 8.4660e-01],
[-4.5916e-02, -7.9999e-01, 5.1752e-01, ..., -8.8457e-01,
-1.0227e+00, 1.2380e+00],
[-1.1459e+00, -8.6057e-01, 7.1069e-01, ..., -9.0956e-01,
-1.4883e+00, 8.8554e-01],
...,
[-9.2073e-01, -1.8902e+00, -1.1278e-01, ..., -9.7096e-01,
-3.2240e-01, -7.8955e-01],
[-1.0383e+00, -9.7012e-01, -4.3642e-01, ..., -1.1219e+00,
-2.0546e-01, -1.6056e+00],
[-2.6693e+00, -5.1376e+00, -4.5636e+00, ..., -4.7607e+00,
-4.1471e+00, -3.9227e+00]],
[[-1.6751e+00, -3.0725e+00, -4.2709e+00, ..., -3.2464e+00,
-4.7189e+00, -2.7139e-01],
[-6.2425e+00, -7.4998e+00, -5.1769e+00, ..., -5.3182e+00,
-7.0414e+00, -1.5115e+00],
[-4.6139e+00, -5.0515e+00, -3.4726e+00, ..., -4.8743e+00,
-6.3593e+00, -1.4602e+00],
...,
[-5.7547e+00, -5.6863e+00, -3.7770e+00, ..., -5.4554e+00,
-5.2157e+00, -3.7304e+00],
[-6.9954e+00, -7.7455e+00, -6.1104e+00, ..., -7.5260e+00,
-8.0447e+00, -6.5754e+00],
[-2.4950e+00, -4.3972e+00, -4.0177e+00, ..., -3.9840e+00,
-4.2296e+00, -2.6464e+00]],
[[-3.6267e-01, 1.9182e+00, 1.0912e+00, ..., 9.2724e-01,
2.1450e+00, 9.3855e-01],
[ 2.0957e-01, 1.0419e+00, 8.9250e-01, ..., 5.9322e-01,
1.2303e+00, -4.2301e-01],
[ 6.5419e-01, 2.0291e+00, 1.2044e+00, ..., 4.1165e-01,
1.7328e+00, 3.1525e-01],
...,
[ 8.7438e-01, 1.0715e+00, 8.5379e-01, ..., 7.5871e-01,
1.0418e+00, 3.1351e-01],
[ 8.2429e-01, 7.0822e-01, 7.1698e-01, ..., 5.9871e-01,
4.7166e-01, 2.3818e-01],
[ 1.3044e+00, 1.5628e+00, 1.2935e+00, ..., 1.7003e+00,
1.2172e+00, 1.5259e-01]],
...,
[[-1.2963e+00, -2.0945e+00, -2.1299e+00, ..., -2.4760e+00,
-2.6785e+00, -4.5751e-01],
[-4.3396e+00, -4.5763e+00, -2.7611e+00, ..., -4.0504e+00,
-4.1316e+00, -2.6420e+00],
[-2.5411e+00, -2.2033e+00, -9.2445e-01, ..., -2.6578e+00,
-4.0599e+00, -1.4468e+00],
...,
[-2.7656e+00, -2.6147e+00, -8.7315e-01, ..., -2.4799e+00,
-3.5460e+00, -1.7423e+00],
[-2.9979e+00, -3.0437e+00, -1.8745e+00, ..., -2.2946e+00,
-4.1044e+00, -2.3432e+00],
[-3.4515e+00, -3.8101e+00, -3.4142e+00, ..., -3.1976e+00,
-3.7966e+00, -1.7418e+00]],
[[-1.9212e+00, -1.6314e+00, -3.0481e+00, ..., -1.4713e+00,
-1.3155e+00, -1.2680e+00],
[-5.4066e+00, -3.0291e+00, -3.4609e+00, ..., -2.7899e+00,
-1.4753e+00, -6.6072e-01],
[-4.4572e+00, -2.4803e+00, -3.3492e+00, ..., -1.4807e+00,
-1.3091e+00, -3.3029e-01],
...,
[-4.1440e+00, -2.6588e+00, -3.1066e+00, ..., -1.7116e+00,
-2.8631e+00, -1.8711e+00],
[-3.6579e+00, -9.9985e-01, -2.2683e+00, ..., -1.1590e+00,
-2.7935e+00, -1.3188e+00],
[-4.3343e+00, -3.0473e+00, -2.8820e+00, ..., -2.1713e+00,
-3.2560e+00, -1.2419e+00]],
[[ 1.5509e+00, 6.2185e-01, 8.5199e-01, ..., -5.1131e-01,
-5.3449e-03, -3.8549e-01],
[ 4.3172e+00, 3.5564e+00, 2.3543e+00, ..., 1.7532e+00,
1.9371e+00, 2.3972e+00],
[ 1.6497e+00, 8.0313e-01, 3.9948e-01, ..., 1.4275e+00,
1.8172e+00, 1.9025e+00],
...,
[ 2.8840e+00, 1.0796e+00, 1.3841e+00, ..., 2.3564e+00,
2.0636e+00, 1.2110e+00],
[ 3.6311e+00, 3.5159e+00, 2.2962e+00, ..., 3.0349e+00,
2.4441e+00, 2.1248e+00],
[ 1.8940e+00, 6.8261e-01, 7.5760e-02, ..., -2.6682e-01,
-1.9453e-01, -1.4890e-01]]],
[[[ 3.3352e-01, 7.6358e-01, -2.5990e+00, ..., -2.4083e+00,
-3.7212e+00, -1.6845e+00],
[ 4.1717e-01, -4.8475e-01, -1.4750e+00, ..., -5.3548e-01,
-1.4402e+00, -6.8687e-01],
[ 2.7915e-01, -1.1422e-01, 3.7813e-01, ..., -2.1735e-01,
-8.3364e-01, -9.4972e-01],
...,
[-1.1712e+00, -2.5158e+00, -5.8914e-01, ..., -6.4208e-01,
1.2297e-01, -4.8068e-01],
[-1.1993e+00, -1.0810e+00, -4.0752e-01, ..., -1.2827e+00,
-4.2486e-01, -1.7644e+00],
[-2.5945e+00, -5.1006e+00, -4.5476e+00, ..., -4.6882e+00,
-4.1138e+00, -3.8633e+00]],
[[-1.5057e+00, -2.9153e+00, -5.9612e+00, ..., -5.7121e+00,
-8.5358e+00, -2.2594e+00],
[-5.1372e+00, -6.3713e+00, -4.5426e+00, ..., -5.4961e+00,
-7.6860e+00, -2.8665e+00],
[-4.0447e+00, -6.3373e+00, -6.3620e+00, ..., -8.6437e+00,
-1.0130e+01, -5.1083e+00],
...,
[-5.6638e+00, -5.4625e+00, -3.1324e+00, ..., -4.4714e+00,
-4.6550e+00, -3.4620e+00],
[-7.3188e+00, -8.0847e+00, -6.4967e+00, ..., -7.8336e+00,
-8.2917e+00, -6.5990e+00],
[-2.4528e+00, -4.3525e+00, -4.0658e+00, ..., -3.8822e+00,
-4.1338e+00, -2.5521e+00]],
[[ 5.0239e-01, 2.3911e+00, 1.6476e+00, ..., 1.9283e+00,
2.9750e+00, 1.0646e+00],
[-2.7974e-01, 4.3183e-01, 2.1711e-01, ..., 1.1868e+00,
2.9491e+00, 6.0847e-01],
[ 5.4131e-01, 1.5706e+00, -4.4291e-03, ..., -7.3134e-02,
1.2074e-01, -4.6949e-01],
...,
[ 3.1757e-01, 8.4664e-01, 5.3727e-01, ..., -3.8181e-01,
-2.6103e-01, -1.5505e-01],
[ 9.7458e-01, 7.7825e-01, 5.5296e-01, ..., 4.2772e-02,
-2.7027e-01, -6.2192e-02],
[ 1.3475e+00, 1.5518e+00, 1.2002e+00, ..., 1.6599e+00,
1.2192e+00, 1.4395e-01]],
...,
[[-7.8339e-01, -2.7088e+00, -4.0481e+00, ..., -4.8698e+00,
-6.0055e+00, -1.9190e+00],
[-2.9992e+00, -4.1930e+00, -2.7515e+00, ..., -3.6869e+00,
-4.9192e+00, -2.4056e+00],
[-2.0161e+00, -2.3434e+00, -3.0985e+00, ..., -5.4467e+00,
-6.1108e+00, -2.6783e+00],
...,
[-2.1229e+00, -1.8970e+00, 8.9207e-02, ..., -5.1413e-01,
-2.2675e+00, -1.1562e+00],
[-3.0566e+00, -3.0857e+00, -1.9575e+00, ..., -2.2170e+00,
-4.3442e+00, -2.4464e+00],
[-3.3291e+00, -3.7699e+00, -3.3979e+00, ..., -3.0915e+00,
-3.6770e+00, -1.6575e+00]],
[[-1.7651e+00, -2.0880e+00, -2.8668e+00, ..., -2.1360e+00,
-1.0775e+00, 7.0250e-01],
[-4.3581e+00, -2.5736e+00, -2.7058e+00, ..., -2.5474e+00,
-2.2370e+00, -4.9793e-02],
[-3.2690e+00, -1.3521e+00, -2.3445e+00, ..., -1.4890e+00,
-8.4491e-01, 3.0849e-01],
...,
[-3.2940e+00, -1.8572e+00, -2.5212e+00, ..., -2.0007e-01,
-1.5571e+00, -9.1486e-01],
[-3.7136e+00, -6.8925e-01, -2.0491e+00, ..., -1.0713e+00,
-2.7285e+00, -1.1714e+00],
[-4.3604e+00, -3.0471e+00, -2.8103e+00, ..., -2.1944e+00,
-3.3092e+00, -1.2594e+00]],
[[ 1.5912e+00, 1.1094e+00, 1.0915e+00, ..., 9.8769e-01,
1.5771e+00, -1.0203e-01],
[ 3.2107e+00, 2.4404e+00, 2.0935e+00, ..., 3.1588e+00,
3.2378e+00, 3.9679e+00],
[ 2.2722e+00, 1.4786e+00, 1.6637e+00, ..., 2.8630e+00,
2.4277e+00, 1.7270e+00],
...,
[ 2.0982e+00, 2.8290e-01, 1.6806e-01, ..., 1.7261e+00,
2.1993e+00, 1.6830e+00],
[ 3.7238e+00, 3.9609e+00, 2.7086e+00, ..., 3.1489e+00,
2.3290e+00, 2.1877e+00],
[ 1.9271e+00, 8.5929e-01, 1.4312e-01, ..., -2.9145e-01,
-2.2384e-01, -1.1011e-01]]],
[[[ 8.6034e-01, 1.6650e+00, 9.2256e-01, ..., -3.9783e+00,
-2.3762e+00, -8.1351e-01],
[ 2.0884e+00, 6.6530e-01, 4.7533e-01, ..., -2.9478e+00,
-2.2286e+00, 1.3194e-02],
[ 7.8920e-01, -1.5075e-02, 1.1950e-01, ..., -3.2595e+00,
-3.0591e+00, 7.1687e-02],
...,
[ 4.6016e-01, -2.5450e-01, -5.1669e-01, ..., -4.5218e+00,
-4.6810e+00, 2.5301e-02],
[-1.7625e-02, -2.4503e-01, -6.0351e-01, ..., -4.8199e+00,
-3.8161e+00, 4.8839e-01],
[-3.0894e-01, -3.3576e+00, -3.1881e+00, ..., -4.5959e+00,
-3.1170e+00, -1.4508e+00]],
[[ 3.9383e-01, -1.1832e+00, -2.6034e+00, ..., -7.1325e+00,
-6.7735e+00, -1.0972e+00],
[-1.8449e+00, -3.9932e+00, -3.9674e+00, ..., -7.1711e+00,
-7.5424e+00, -2.2765e+00],
[ 7.1006e-02, -2.3715e+00, -3.1595e+00, ..., -9.0010e+00,
-7.6505e+00, -2.3883e+00],
...,
[-1.8458e+00, -2.7143e+00, -3.2468e+00, ..., -1.2650e+01,
-1.0743e+01, -2.3941e+00],
[-4.1382e+00, -5.5309e+00, -4.4697e+00, ..., -9.9042e+00,
-1.0524e+01, -4.3230e+00],
[-2.3741e-01, -3.0270e+00, -3.3614e+00, ..., -3.8273e+00,
-3.5049e+00, -9.0311e-01]],
[[ 5.3178e-01, 2.1956e+00, 1.9518e+00, ..., 7.2681e-01,
2.2457e+00, 1.0886e+00],
[ 3.0557e-02, 9.3115e-01, 4.0865e-01, ..., 7.4946e-01,
2.4961e+00, 3.5917e-02],
[ 8.8216e-01, 3.2095e+00, 2.1977e+00, ..., -6.4191e-01,
2.0347e+00, -2.3014e-01],
...,
[ 1.0666e+00, 3.1428e+00, 1.3568e-01, ..., 1.7602e+00,
1.3762e+00, -9.8760e-01],
[ 9.8076e-01, 1.4507e+00, 4.8849e-01, ..., -6.6363e-01,
2.0576e-01, -1.4378e+00],
[-2.3360e-01, 1.7961e+00, 7.9771e-01, ..., -2.6594e-01,
-3.3530e-01, -1.6386e+00]],
...,
[[-4.0321e-01, -9.2531e-01, -1.5033e+00, ..., -5.3345e+00,
-5.1812e+00, -6.9671e-01],
[-2.9931e+00, -4.3849e+00, -2.2430e+00, ..., -4.5687e+00,
-4.8387e+00, -1.6585e+00],
[-1.7173e+00, -1.8035e+00, -8.0132e-01, ..., -6.6117e+00,
-6.0383e+00, -9.8888e-01],
...,
[-8.5418e-01, -2.0298e+00, -1.0200e+00, ..., -8.7544e+00,
-7.3317e+00, -2.1353e+00],
[-1.6764e+00, -3.2825e+00, -1.8134e+00, ..., -3.8380e+00,
-4.8623e+00, -1.1330e+00],
[-9.5783e-01, -3.2184e+00, -2.1757e+00, ..., -2.4915e+00,
-2.7027e+00, 1.9010e-01]],
[[-1.0672e+00, -6.0809e-01, -1.0826e+00, ..., -1.9862e+00,
-4.7203e-01, -5.6596e-01],
[-3.3392e+00, -3.0964e+00, -2.3031e+00, ..., -3.3278e+00,
-1.8335e+00, -2.5043e-01],
[-2.1322e+00, -3.4349e+00, -2.0521e+00, ..., -1.8937e+00,
-2.1503e+00, -4.7085e-01],
...,
[-2.3388e+00, -4.0500e+00, -2.2562e+00, ..., -4.9902e+00,
-2.0472e+00, -3.2241e-02],
[-2.9417e+00, -1.8333e+00, -8.6271e-01, ..., -2.4884e+00,
-3.6562e-01, 1.4357e+00],
[-3.2072e+00, -3.0718e+00, -1.1410e+00, ..., 1.8492e-01,
-2.8561e-01, 7.7640e-01]],
[[ 1.0251e+00, 8.6390e-01, 1.1520e+00, ..., -5.4764e-01,
6.1243e-01, -1.2916e+00],
[ 2.0037e+00, 1.9184e+00, 1.5694e+00, ..., 2.8958e+00,
2.9704e+00, 2.2395e+00],
[ 7.7498e-01, -2.9284e-01, 5.7470e-01, ..., 1.8894e+00,
1.4265e+00, 1.3100e+00],
...,
[ 1.3977e+00, 1.1304e+00, 1.8179e+00, ..., 2.6571e+00,
2.0847e+00, 5.2711e-01],
[ 3.0633e+00, 3.0868e+00, 1.5124e+00, ..., 8.2993e-01,
2.0211e+00, 2.0904e+00],
[ 1.9348e+00, 3.1588e-01, 3.1823e-01, ..., 1.5005e+00,
1.4342e-01, 7.0588e-01]]],
[[[-5.1725e-01, -3.4138e+00, -1.9887e+00, ..., -1.9389e+00,
-1.7873e+00, -1.2381e+00],
[ 8.3989e-01, -4.6115e+00, -3.6786e+00, ..., -1.7225e+00,
-1.8640e-01, -1.4814e+00],
[ 1.1327e+00, -9.0909e-01, -5.8856e+00, ..., -3.2650e+00,
-6.2418e-01, -1.8190e+00],
...,
[-1.7636e+00, -8.1935e+00, -5.9730e+00, ..., -5.5206e+00,
9.7945e-01, 2.6475e+00],
[-2.2520e+00, -7.9886e+00, -4.7837e+00, ..., -1.0144e+00,
2.9192e+00, 2.2718e+00],
[-3.1583e+00, -7.1523e+00, -2.6696e+00, ..., 2.9858e+00,
3.2164e+00, -3.0843e-01]],
[[-1.5638e+00, -6.4712e+00, -3.3317e+00, ..., -3.5668e+00,
-5.5401e+00, -3.0940e+00],
[-2.3768e+00, -6.1104e+00, -6.8056e+00, ..., -5.4375e+00,
-5.7014e+00, -4.9546e+00],
[-4.1072e+00, -7.2611e+00, -1.9268e+01, ..., -5.8142e+00,
-4.6548e+00, -3.3318e+00],
...,
[-7.5300e+00, -1.2512e+01, -5.6022e+00, ..., -5.2240e+00,
1.7657e-01, 1.6160e+00],
[-8.0022e+00, -1.2743e+01, -7.2379e+00, ..., -4.0635e+00,
-1.8650e+00, -2.5511e+00],
[-1.7887e+00, -5.0572e+00, -1.9140e+00, ..., 8.3575e-01,
-4.7983e-01, -2.3139e+00]],
[[-4.2694e-02, 2.5059e-01, 6.7400e-01, ..., 1.9094e+00,
2.2135e+00, 9.0539e-01],
[ 3.3920e-01, -2.4960e+00, -6.7057e+00, ..., 1.2382e+00,
2.6083e+00, 1.1596e+00],
[-3.1872e-01, 1.1646e+00, -3.3249e+00, ..., -1.4743e-01,
2.4048e+00, 2.1596e+00],
...,
[-2.1109e-01, 1.7041e+00, -1.6031e+00, ..., -5.4985e+00,
-5.6212e+00, 6.7894e-01],
[-6.9554e-01, 2.7542e-01, -5.5823e+00, ..., 4.0384e+00,
2.0384e+00, 5.0325e-01],
[ 5.0120e-01, -6.5324e-01, -7.2845e+00, ..., -7.6970e-01,
-2.2889e+00, 4.3032e-01]],
...,
[[-1.4964e+00, -3.1212e+00, -1.5515e+00, ..., -3.8197e+00,
-4.4124e+00, -1.4548e+00],
[-2.2880e+00, -3.6630e+00, 3.1813e-02, ..., -5.6641e+00,
-5.4111e+00, -3.5662e+00],
[-9.9589e-01, -2.6709e+00, -4.7787e+00, ..., -6.3803e+00,
-4.6672e+00, -2.8203e+00],
...,
[-2.2416e+00, -5.4726e+00, -8.1929e-01, ..., -7.0903e+00,
-4.4709e+00, -1.4110e+00],
[-2.4808e+00, -7.1334e+00, -3.9014e+00, ..., -3.1106e+00,
-5.0976e+00, -4.2132e+00],
[-3.3487e+00, -3.9961e+00, 3.5535e-01, ..., 9.9350e-01,
-6.2913e-01, -1.3508e+00]],
[[-2.0995e+00, -1.0353e+00, -4.2874e+00, ..., -3.5143e+00,
-2.7578e+00, -1.4381e-01],
[-4.3748e+00, -1.1957e+00, -3.1076e+00, ..., -3.1071e+00,
-5.1399e+00, -1.5691e+00],
[-4.9950e+00, -4.6806e+00, -1.0609e-01, ..., -1.3960e+00,
-3.2730e+00, -1.3051e+00],
...,
[-6.6526e+00, -2.2108e+00, -1.0706e+00, ..., -5.6931e-01,
-2.4557e+00, 7.3515e-01],
[-3.4811e+00, -3.4251e-01, -4.3599e+00, ..., -9.7211e-01,
-9.7008e-01, 1.0445e+00],
[-2.6983e+00, 2.0610e-01, 3.3294e-01, ..., 1.1497e+00,
-1.3067e+00, -6.1727e-01]],
[[ 1.5738e+00, -1.4876e-01, 1.1477e+00, ..., 2.7417e-01,
1.2961e+00, 6.9132e-01],
[ 2.1631e+00, 8.4029e-01, 1.7306e+00, ..., 6.0824e-01,
2.7246e+00, 3.5608e+00],
[ 1.4553e+00, 2.2351e+00, 8.6448e+00, ..., 8.1451e-01,
1.5537e+00, 8.8933e-01],
...,
[ 2.7271e+00, 9.6558e-01, 2.8839e+00, ..., -8.0448e+00,
-1.0114e+00, 2.2533e+00],
[ 3.9948e+00, 2.2998e+00, 4.0307e+00, ..., -1.2077e+00,
2.4007e+00, 3.1591e+00],
[ 2.6083e+00, 9.6159e-01, 3.4982e+00, ..., -8.5268e-01,
1.6937e+00, 9.2993e-01]]],
[[[ 3.5121e-01, 1.2935e+00, 4.5943e-02, ..., -3.2865e+00,
-2.4093e+00, -1.1340e+00],
[ 3.4585e-01, 6.9333e-01, 1.2932e+00, ..., -3.0181e+00,
-3.0185e+00, -3.3043e-01],
[-2.6196e-01, 3.3306e-01, 1.2649e+00, ..., -3.5874e+00,
-3.8318e+00, -5.1960e-01],
...,
[-4.0697e+00, -6.8989e+00, 8.1574e-01, ..., -7.2354e+00,
-4.9941e+00, 2.4669e-02],
[-2.6282e+00, -4.4934e+00, 1.8511e+00, ..., -5.0651e+00,
-3.9485e+00, 5.0595e-01],
[-8.7185e-01, -3.9111e+00, -2.1763e+00, ..., -4.2334e+00,
-2.6675e+00, -1.2283e+00]],
[[-9.6264e-01, -1.5736e+00, -3.4825e+00, ..., -7.1299e+00,
-7.5839e+00, -1.3879e+00],
[-3.8352e+00, -4.0887e+00, -3.1260e+00, ..., -7.9856e+00,
-8.9504e+00, -2.3502e+00],
[-3.5861e+00, -4.7195e+00, -5.5899e+00, ..., -4.8106e+00,
-8.1610e+00, -2.2743e+00],
...,
[-7.2018e+00, -9.1765e+00, 7.8515e-01, ..., -8.9740e+00,
-8.3184e+00, -1.4548e+00],
[-8.2210e+00, -1.0997e+01, -7.4701e+00, ..., -9.6550e+00,
-9.9947e+00, -4.2677e+00],
[-1.2096e+00, -2.4542e+00, -3.3893e+00, ..., -3.6260e+00,
-3.6917e+00, -8.4397e-01]],
[[-1.2488e-01, 1.2766e+00, 1.7564e+00, ..., 1.1359e+00,
2.2831e+00, 9.6022e-01],
[-9.2206e-02, -5.8802e-02, 8.7638e-01, ..., 1.1252e+00,
2.9648e+00, -1.7434e-01],
[ 6.7327e-01, 1.3369e+00, 1.2226e+00, ..., 2.6759e+00,
2.4146e+00, -6.2676e-01],
...,
[-5.2880e-01, -4.3215e+00, -1.0288e+01, ..., 2.7770e+00,
1.4361e-01, -2.2614e+00],
[-1.2373e-01, -8.4052e-01, -7.2557e+00, ..., -1.1253e+00,
-3.1449e-01, -1.1089e+00],
[-1.1169e-01, 4.0065e-01, -4.7990e-02, ..., -2.3352e-01,
-4.5806e-01, -1.6693e+00]],
...,
[[-5.6980e-01, -1.6014e+00, -2.2015e+00, ..., -4.8738e+00,
-5.0649e+00, -6.0280e-01],
[-2.6156e+00, -2.4468e+00, -1.2938e+00, ..., -4.0571e+00,
-5.1524e+00, -2.0202e+00],
[-1.6833e+00, -9.6345e-01, -1.1148e+00, ..., -5.4307e+00,
-4.5827e+00, -2.0580e+00],
...,
[-4.4019e+00, -4.3679e+00, -3.5730e+00, ..., -6.9769e+00,
-5.5215e+00, -2.7662e+00],
[-4.2853e+00, -4.4566e+00, -3.7130e+00, ..., -4.6203e+00,
-5.3418e+00, -1.7827e+00],
[-1.1059e+00, -2.4160e+00, -4.1401e+00, ..., -2.8164e+00,
-2.3646e+00, 1.7090e-01]],
[[-1.2089e+00, -2.5098e+00, -4.0705e+00, ..., -2.2098e+00,
-1.1629e+00, -7.8157e-01],
[-3.6150e+00, -3.2549e+00, -4.2899e+00, ..., -3.3857e+00,
-2.2707e+00, -5.2605e-01],
[-4.5210e+00, -3.8231e+00, -4.2716e+00, ..., -1.5553e+00,
-1.8674e+00, -8.5341e-01],
...,
[-5.7454e+00, -2.7466e+00, -4.7917e+00, ..., -4.4050e+00,
-1.7460e+00, -1.6840e-01],
[-3.7935e+00, 2.9946e-01, 2.9312e+00, ..., -2.4575e-01,
1.3221e-01, 1.3848e+00],
[-2.6373e+00, -1.7709e+00, -2.8847e-01, ..., 9.4123e-01,
-1.0255e-01, 7.4533e-01]],
[[ 1.2085e+00, -3.7018e-02, 1.1352e+00, ..., 3.9315e-01,
9.1062e-01, -1.3840e+00],
[ 2.1922e+00, 9.9002e-01, 1.2463e+00, ..., 2.8199e+00,
2.2295e+00, 2.1898e+00],
[ 1.2956e+00, 7.9965e-01, 1.2688e+00, ..., 2.5045e+00,
1.3321e+00, 8.5337e-01],
...,
[ 3.1097e+00, 2.9269e+00, -2.5601e-01, ..., 2.4278e+00,
2.9982e+00, 1.8528e+00],
[ 2.7723e+00, 4.7377e+00, 5.5607e+00, ..., 3.1165e+00,
2.0084e+00, 1.5900e+00],
[ 1.8719e+00, 6.8600e-01, 2.5814e+00, ..., -6.2523e-01,
-8.1176e-01, 5.0831e-01]]]], device='cuda:0'),
tensor([[[[-1.2368e-02, 1.0900e+00, 3.3067e-01, ..., -7.3373e-01,
-1.7097e+00, -2.2278e-01],
[ 7.6173e-01, 8.4405e-01, 1.4843e+00, ..., -2.2657e+00,
-2.0429e+00, -2.6386e-01],
[-4.6076e-01, -8.5358e-01, 1.3204e+00, ..., -4.7092e+00,
-1.6914e+00, -7.5904e-01],
...,
[-1.5468e-01, -4.2821e-01, 1.8673e-01, ..., -4.7676e+00,
-3.5567e+00, -1.9103e+00],
[-1.9246e+00, -1.2904e+00, -1.0103e+00, ..., -3.9727e+00,
-3.2959e+00, -8.8043e-01],
[-1.0097e+00, -3.0691e+00, -3.2067e+00, ..., -4.8671e+00,
-3.5267e+00, -1.2791e+00]],
[[-1.4754e+00, -2.1289e+00, -2.8162e+00, ..., -5.2511e+00,
-6.1525e+00, -1.4307e+00],
[-3.2411e+00, -4.4149e+00, -4.2492e+00, ..., -6.8534e+00,
-4.8001e+00, -1.4225e+00],
[-2.1939e+00, -3.4019e+00, -3.8481e+00, ..., -7.6489e+00,
-5.3769e+00, -1.3371e+00],
...,
[-4.8692e+00, -5.2125e+00, -3.0008e+00, ..., -1.3803e+01,
-8.6763e+00, -2.1699e+00],
[-7.2088e+00, -7.2229e+00, -6.6945e+00, ..., -9.6230e+00,
-8.8676e+00, -3.8020e+00],
[-1.5843e+00, -3.0026e+00, -2.8053e+00, ..., -4.9557e+00,
-3.3589e+00, -4.8185e-01]],
[[ 2.1484e-01, 1.2557e+00, 1.5331e+00, ..., 4.7867e-01,
2.5673e+00, 1.0637e+00],
[ 1.6457e-01, 2.8063e-01, 3.6938e-01, ..., -7.9199e-01,
2.0275e+00, 3.8303e-01],
[ 2.5669e-02, 2.0050e+00, 1.6335e+00, ..., 1.4771e-01,
2.3015e+00, 1.7599e+00],
...,
[-5.6537e-02, 6.9764e-01, 5.1419e-01, ..., -5.4665e-01,
8.5887e-01, -4.7385e-01],
[ 3.6932e-01, 7.7665e-01, 4.9711e-01, ..., -5.7128e-01,
-4.4134e-01, -4.3343e-02],
[ 4.6024e-01, 1.0486e+00, 1.1011e+00, ..., -1.8524e+00,
-3.3265e-01, -1.2439e+00]],
...,
[[-9.5863e-01, -1.6308e+00, -2.0081e+00, ..., -3.1905e+00,
-4.5518e+00, -1.1052e+00],
[-2.5498e+00, -2.5991e+00, -1.2782e+00, ..., -3.3589e+00,
-3.8871e+00, -1.4683e+00],
[-1.7453e+00, 1.8576e-01, 8.3119e-01, ..., -3.5869e+00,
-2.4464e+00, -2.8517e-01],
...,
[-1.2161e+00, -5.0564e-01, 1.3805e+00, ..., -6.0442e+00,
-6.1156e+00, -2.4584e+00],
[-2.8608e+00, -2.7900e+00, -1.5044e+00, ..., -3.5578e-01,
-4.9740e+00, -1.0953e+00],
[-1.2597e+00, -1.6827e+00, -1.2987e+00, ..., -9.9355e-01,
-2.3839e+00, 9.5916e-01]],
[[-1.6859e+00, -1.9087e+00, -3.8816e+00, ..., -2.6438e+00,
-1.4394e+00, -6.2967e-03],
[-3.7428e+00, -3.4034e+00, -3.8895e+00, ..., -2.1350e+00,
-2.6767e+00, 4.0749e-01],
[-3.8278e+00, -3.2408e+00, -2.7402e+00, ..., -4.5322e+00,
-4.8263e+00, -2.8078e+00],
...,
[-3.8407e+00, -3.9405e+00, -4.1775e+00, ..., -6.5376e+00,
-5.5261e+00, -7.8274e-01],
[-3.9989e+00, -1.8778e+00, -2.1439e+00, ..., -1.8367e+00,
-2.0741e+00, 6.6960e-01],
[-3.0084e+00, -3.0380e+00, -1.2379e+00, ..., 4.6121e-01,
-2.3899e-01, 7.8699e-01]],
[[ 1.2718e+00, -4.0652e-02, 1.0547e+00, ..., -7.7616e-01,
1.5429e-01, 4.3544e-02],
[ 2.2738e+00, 1.4451e+00, 1.2963e+00, ..., 4.7590e-01,
8.2134e-01, 2.9006e+00],
[ 5.0328e-01, -5.3438e-01, 5.2414e-01, ..., -2.4587e-01,
1.3606e+00, 1.6915e+00],
...,
[ 1.5571e+00, 1.2471e+00, 9.9950e-01, ..., 3.5558e+00,
2.2512e+00, 9.6808e-01],
[ 2.5101e+00, 4.0235e+00, 2.1274e+00, ..., 1.9062e-01,
1.6504e+00, 1.6783e+00],
[ 1.8856e+00, 1.4653e+00, 1.4188e-01, ..., 5.4486e-01,
-2.8040e-01, 6.3889e-01]]],
[[[ 1.6025e-01, -3.5942e-01, -4.7120e+00, ..., -2.6231e+00,
-4.5765e+00, -1.4440e+00],
[ 7.2263e-01, -4.1836e-01, -5.0338e+00, ..., -4.4991e-01,
-2.0207e+00, -6.9410e-01],
[ 2.1242e-01, -5.3039e-01, -4.4912e+00, ..., 2.8710e+00,
-7.2887e-01, -1.4257e+00],
...,
[-1.2626e+00, -2.6884e+00, -3.5259e-01, ..., -8.5624e-01,
2.8426e-01, -4.4562e-01],
[-1.2264e+00, -1.3032e+00, -4.3833e-01, ..., -6.9331e-01,
1.2720e-01, -1.6901e+00],
[-2.5179e+00, -5.0538e+00, -4.3308e+00, ..., -4.5564e+00,
-4.0071e+00, -3.7947e+00]],
[[-1.5105e+00, -2.2223e+00, -6.2452e+00, ..., -7.7000e+00,
-9.5552e+00, -1.8920e+00],
[-4.8055e+00, -4.1520e+00, -5.2844e+00, ..., -6.3745e+00,
-9.1925e+00, -2.8026e+00],
[-3.8776e+00, -5.3745e+00, -9.3693e+00, ..., -6.2469e+00,
-1.2322e+01, -5.9553e+00],
...,
[-5.7192e+00, -5.4686e+00, -3.0695e+00, ..., -4.3615e+00,
-4.4183e+00, -3.3304e+00],
[-7.2043e+00, -8.0832e+00, -6.2816e+00, ..., -7.3338e+00,
-7.9612e+00, -6.4268e+00],
[-2.3861e+00, -4.2269e+00, -4.0531e+00, ..., -3.7464e+00,
-4.0640e+00, -2.5265e+00]],
[[ 5.9301e-01, 2.3527e+00, 3.5217e+00, ..., 2.6444e+00,
2.7190e+00, 8.5800e-01],
[-7.7635e-01, 2.1880e-01, 2.5917e+00, ..., 2.7724e+00,
2.5372e+00, -8.3002e-03],
[ 5.9724e-01, 7.4618e-01, -4.4988e-02, ..., 4.2210e+00,
8.7449e-01, -6.7727e-01],
...,
[ 3.6439e-01, 7.4708e-01, 7.6286e-01, ..., -6.3217e-01,
-1.6368e-01, 1.3320e-02],
[ 9.5711e-01, 7.6398e-01, 5.9360e-01, ..., 8.0328e-01,
-3.2684e-01, -8.5531e-02],
[ 1.2304e+00, 1.5337e+00, 1.1343e+00, ..., 1.4849e+00,
1.0370e+00, 1.8400e-01]],
...,
[[-6.0060e-01, -3.1565e+00, -3.4908e+00, ..., -4.2570e+00,
-6.9804e+00, -1.6871e+00],
[-1.9359e+00, -3.8580e+00, -2.5564e+00, ..., -4.6815e+00,
-5.8077e+00, -3.0269e+00],
[-1.6426e+00, -3.0630e+00, -3.6939e+00, ..., -4.5061e+00,
-5.5529e+00, -3.8266e+00],
...,
[-2.1991e+00, -1.8340e+00, -1.7645e-02, ..., -9.5052e-01,
-2.4372e+00, -1.0988e+00],
[-3.0715e+00, -3.0993e+00, -1.9714e+00, ..., -2.3626e+00,
-4.1721e+00, -2.5351e+00],
[-3.1886e+00, -3.6626e+00, -3.2206e+00, ..., -2.8627e+00,
-3.6349e+00, -1.6733e+00]],
[[-1.7985e+00, -3.0774e+00, -2.0567e+00, ..., -2.3948e+00,
-9.2497e-01, 5.9540e-01],
[-3.2481e+00, -2.9486e+00, -8.2581e-01, ..., -2.0184e+00,
-2.1062e+00, 2.5547e-02],
[-2.4481e+00, -1.2875e+00, -7.8437e-01, ..., -2.8958e+00,
-6.4427e-01, 3.3631e-02],
...,
[-3.4177e+00, -1.6824e+00, -2.3141e+00, ..., 2.4252e-01,
-1.6327e+00, -1.0647e+00],
[-3.7888e+00, -6.5824e-01, -2.0653e+00, ..., -4.6665e-01,
-2.6384e+00, -1.3786e+00],
[-4.3038e+00, -3.0064e+00, -2.7431e+00, ..., -2.3145e+00,
-3.2817e+00, -1.2913e+00]],
[[ 1.3837e+00, 1.0095e+00, 1.1111e+00, ..., 1.7497e+00,
1.2616e+00, -1.5001e-01],
[ 2.9392e+00, 2.6563e+00, 1.9582e+00, ..., 4.1935e+00,
3.2678e+00, 3.6614e+00],
[ 2.1457e+00, 2.3749e+00, 1.1029e+00, ..., 6.1723e+00,
3.7105e+00, 1.3282e+00],
...,
[ 2.0848e+00, 3.3969e-01, 2.9129e-01, ..., 2.4558e+00,
2.3054e+00, 1.6364e+00],
[ 3.5944e+00, 3.8888e+00, 2.6588e+00, ..., 2.3786e+00,
2.1057e+00, 2.1092e+00],
[ 1.8612e+00, 7.3750e-01, 5.2492e-02, ..., -2.3581e-01,
-2.0922e-01, -1.4793e-01]]],
[[[-8.5834e-02, -1.6982e-01, -1.0103e+00, ..., -2.0999e-01,
-3.0943e-02, -3.8596e-01],
[-2.6612e+00, -5.9017e+00, -1.9751e+00, ..., 7.9665e-01,
7.3629e-01, -1.2938e+00],
[-5.5444e+00, -2.7581e+00, 1.8940e+00, ..., -3.5781e+00,
-1.9428e+00, -2.2809e+00],
...,
[-3.5802e-01, -1.7843e-01, 4.3866e-01, ..., -2.0048e+00,
-2.2712e+00, -1.6829e+00],
[-1.4458e+00, 7.6177e-02, 1.3858e-01, ..., -6.1969e-01,
-4.7135e-01, -2.0504e+00],
[-1.2085e+00, -2.2503e+00, -1.7009e+00, ..., -4.0591e+00,
-3.5875e+00, -3.0902e+00]],
[[-2.5233e+00, -1.9899e+00, -2.5584e+00, ..., -4.3543e+00,
-4.7492e+00, -2.5305e+00],
[-8.5278e+00, -9.3077e+00, -8.7178e+00, ..., -4.3791e+00,
-6.8135e+00, -4.9190e+00],
[-3.1125e+00, 6.6899e-01, 1.7620e+00, ..., -3.7817e+00,
-3.6008e+00, -2.7574e+00],
...,
[-4.3466e+00, -4.8703e+00, -4.4301e+00, ..., -1.0324e+01,
-7.1835e+00, -4.4735e+00],
[-7.4909e+00, -7.7154e+00, -6.2421e+00, ..., -5.2265e+00,
-5.7734e+00, -5.8268e+00],
[-1.3666e+00, -3.1265e+00, -1.8749e+00, ..., -3.7742e+00,
-3.2349e+00, -1.8801e+00]],
[[-1.4993e-03, 1.8913e+00, 1.8862e+00, ..., 1.5879e+00,
1.6033e+00, 7.8056e-01],
[ 6.8660e-01, 1.9767e+00, 4.0545e+00, ..., 1.0357e+00,
1.3142e+00, 8.5605e-01],
[ 2.8130e+00, 2.0149e-01, -3.7926e+00, ..., 4.5411e+00,
3.4250e+00, 1.2277e+00],
...,
[-7.0699e-01, 8.4135e-01, 1.4925e+00, ..., -1.5139e+00,
-4.5057e-01, 6.0918e-01],
[-1.3854e+00, -1.8198e-02, 1.1732e+00, ..., 1.1662e+00,
7.2315e-01, 3.9199e-01],
[-5.0884e-01, -4.3640e-01, 5.7186e-02, ..., 1.1873e+00,
1.6663e+00, 1.0569e+00]],
...,
[[-2.3040e+00, -3.2152e+00, -2.1447e+00, ..., -1.9840e+00,
-2.3177e+00, -7.9897e-01],
[-6.0482e+00, -6.7455e+00, -4.5879e+00, ..., -2.2513e+00,
-3.2229e+00, -3.5457e+00],
[-4.6692e+00, 6.6580e-01, 2.4007e+00, ..., -3.7396e+00,
-2.7924e+00, -2.4628e+00],
...,
[-9.3648e-01, -2.4937e+00, -1.0466e+00, ..., -6.9190e+00,
-5.1132e+00, -1.6284e+00],
[-1.9279e+00, -4.0133e+00, -1.5971e+00, ..., -8.5783e-01,
-3.3794e+00, -2.1428e+00],
[-1.3374e+00, -1.5345e+00, -6.0893e-01, ..., -2.4530e+00,
-3.0279e+00, -9.9068e-01]],
[[-1.4961e+00, -3.0159e+00, -4.0410e+00, ..., -7.6379e-01,
-1.4383e+00, -5.9210e-01],
[-6.2588e+00, -4.3004e+00, -6.9087e+00, ..., -1.0221e+00,
-3.5582e+00, -2.2922e+00],
[-2.5762e+00, 2.1112e-01, 1.8749e+00, ..., -4.9614e+00,
-4.2313e+00, -1.0330e+00],
...,
[-2.5879e+00, -2.9957e+00, -3.6177e+00, ..., -3.7922e+00,
-4.8539e+00, -9.7401e-01],
[-1.8144e+00, -3.0697e-01, -1.7788e+00, ..., -1.4997e+00,
-3.2068e+00, -1.3319e+00],
[-2.0413e+00, -1.4772e+00, -1.6816e+00, ..., -1.2000e+00,
-2.2647e+00, -8.1267e-01]],
[[ 1.6698e+00, -6.6997e-01, -1.2682e+00, ..., 8.0794e-01,
1.3771e+00, 5.3987e-01],
[ 4.8049e+00, 3.1377e+00, 3.6049e+00, ..., 1.9048e+00,
1.9508e+00, 2.0898e+00],
[ 2.0854e-01, 6.9981e-01, 1.3750e-01, ..., -1.0772e-01,
5.0126e-01, 5.6819e-01],
...,
[ 2.9101e+00, 2.7266e+00, 2.5515e+00, ..., 3.3359e+00,
1.5542e+00, 1.1385e+00],
[ 3.7756e+00, 4.0256e+00, 2.6323e+00, ..., 9.3745e-01,
1.4415e+00, 1.9829e+00],
[ 2.4101e+00, 2.3552e+00, 1.5866e+00, ..., 2.6157e-01,
-3.4793e-01, -2.9759e-01]]],
[[[-1.9070e-01, 1.8441e-01, -2.0417e+00, ..., -1.5298e+00,
-1.1624e+00, -8.9733e-01],
[ 8.0435e-02, -2.4102e+00, -3.6818e+00, ..., -1.2595e+00,
1.4668e-01, -1.1639e+00],
[-2.7109e-01, -3.8607e+00, -6.2495e-01, ..., -2.2521e+00,
-9.8922e-01, -2.1153e+00],
...,
[-1.3512e+00, 3.7806e-01, -1.1322e+00, ..., -3.8278e+00,
-2.0649e+00, 3.2034e-02],
[-2.8445e+00, 3.9219e-01, 1.5533e-01, ..., -1.8993e+00,
-2.7095e+00, 1.4620e-01],
[-1.6620e+00, -2.7234e+00, -2.9300e+00, ..., -3.9479e+00,
-3.6108e+00, -1.6826e+00]],
[[-7.0180e-01, -1.9296e+00, -2.3868e+00, ..., -3.1315e+00,
-4.7891e+00, -2.5140e+00],
[-3.1080e+00, -7.7980e+00, -4.2044e+00, ..., -4.3820e+00,
-5.0915e+00, -4.6958e+00],
[-3.9475e+00, -7.3069e+00, 6.7621e-01, ..., -4.6480e+00,
-4.1786e+00, -3.6391e+00],
...,
[-6.1941e+00, -7.9249e+00, -4.4451e+00, ..., -1.0189e+01,
-8.7114e+00, -1.7795e+00],
[-8.7826e+00, -9.0090e+00, -7.7704e+00, ..., -9.8394e+00,
-1.0773e+01, -3.9987e+00],
[-1.9658e+00, -4.0604e+00, -3.6586e+00, ..., -4.0108e+00,
-4.5865e+00, -8.2064e-01]],
[[ 3.3736e-01, 8.0269e-01, 1.1597e+00, ..., 1.5046e+00,
1.9048e+00, 8.6043e-01],
[-1.8561e-02, 1.5011e+00, 1.9530e+00, ..., 1.3029e+00,
2.2920e+00, 9.3272e-01],
[-4.6110e-01, 4.1993e+00, -5.8449e-01, ..., 1.2390e+00,
2.2951e+00, 2.0346e+00],
...,
[ 1.1445e-01, -1.3563e-01, 2.8678e-01, ..., 2.4672e+00,
-4.7424e-01, -5.7738e-01],
[-6.7217e-01, 4.7422e-01, -1.3556e+00, ..., -2.2883e-01,
2.1650e-01, -3.3218e-01],
[ 4.2895e-03, 2.4011e-01, -4.4385e-02, ..., 8.0306e-02,
1.2983e-01, -1.5942e+00]],
...,
[[-1.0136e+00, -1.8914e+00, -2.3417e+00, ..., -2.9424e+00,
-3.3874e+00, -1.2395e+00],
[-3.4028e+00, -5.4413e+00, -2.7518e+00, ..., -3.9768e+00,
-4.3593e+00, -3.4869e+00],
[-2.3688e+00, -3.9080e+00, 1.4456e+00, ..., -3.3965e+00,
-5.0985e+00, -3.0179e+00],
...,
[-2.4634e+00, -2.2373e+00, -2.3974e+00, ..., -7.2221e+00,
-5.7694e+00, -3.1427e+00],
[-3.1126e+00, -4.0261e+00, -1.5750e+00, ..., -4.3376e+00,
-5.4966e+00, -1.7491e+00],
[-2.3397e+00, -2.8854e+00, -1.4328e+00, ..., -3.0289e+00,
-2.6730e+00, 1.9109e-01]],
[[-1.5659e+00, -1.7501e+00, -3.8788e+00, ..., -3.0593e+00,
-2.8482e+00, -4.9209e-01],
[-5.5329e+00, -4.3589e+00, -4.9491e+00, ..., -4.1769e+00,
-5.7783e+00, -1.8970e+00],
[-5.3370e+00, -2.3855e+00, -2.1281e+00, ..., -2.3744e+00,
-4.1268e+00, -1.9691e+00],
...,
[-3.1329e+00, -1.5642e+00, -6.3584e+00, ..., -5.2004e+00,
-2.2679e+00, -1.0967e+00],
[-2.9430e+00, -1.7165e-04, -2.7201e+00, ..., -1.8264e+00,
-9.8326e-01, 2.9932e-01],
[-2.9322e+00, -2.4138e+00, -1.6733e+00, ..., -1.4532e+00,
-1.0448e+00, 4.2379e-01]],
[[ 6.2730e-01, -2.6679e-01, -2.0857e-03, ..., 4.2217e-01,
1.3777e+00, 5.9038e-01],
[ 2.0497e+00, 1.8196e+00, 4.4964e-02, ..., 3.7519e-01,
1.8401e+00, 3.2156e+00],
[ 1.9084e+00, 8.9278e-01, -7.1413e-01, ..., 1.4771e-01,
6.7305e-01, 7.1703e-01],
...,
[ 3.4808e+00, 2.6654e+00, 2.5033e+00, ..., 3.1349e+00,
2.6328e+00, 2.9765e-01],
[ 3.5944e+00, 4.4658e+00, 3.3649e+00, ..., 2.6809e+00,
1.9189e+00, 1.2617e+00],
[ 2.5750e+00, 2.9089e+00, 1.8262e+00, ..., 9.3093e-01,
3.7432e-01, 8.8681e-01]]],
[[[ 1.7066e+00, 1.7018e+00, 4.9281e-01, ..., -2.8749e+00,
-2.5024e+00, -1.1356e+00],
[ 9.1453e-01, -1.7562e-01, -6.6744e+00, ..., 6.5930e-01,
-1.4816e+00, -1.1878e+00],
[-1.3007e-01, -5.6042e+00, -4.9197e+00, ..., 2.4720e+00,
-4.2737e-01, -9.6574e-01],
...,
[-1.0716e+00, -1.8728e+00, -9.8331e-01, ..., -6.8236e+00,
2.5476e+00, 1.7257e+00],
[-2.5997e+00, -1.3596e+00, -8.1020e-01, ..., 3.8945e-01,
4.9146e+00, 2.8780e-01],
[-1.6683e+00, -3.6604e+00, -3.4950e+00, ..., 1.4367e+00,
-2.0211e+00, -2.7325e+00]],
[[-1.0815e+00, -4.1350e+00, -6.1725e+00, ..., -4.0064e+00,
-7.6368e+00, -1.5180e+00],
[-3.5114e+00, -1.0333e+01, -1.3774e+01, ..., -3.0581e+00,
-9.4190e+00, -2.6373e+00],
[-7.2884e+00, -1.4690e+01, -4.1611e+00, ..., -3.5129e-01,
-6.1816e+00, -2.6176e+00],
...,
[-6.6695e+00, -5.9610e+00, -4.3338e+00, ..., -9.3079e+00,
-8.0741e-02, -1.6515e-01],
[-8.5292e+00, -8.9338e+00, -7.5338e+00, ..., -1.4964e+00,
-8.9540e-01, -4.1866e+00],
[-1.9107e+00, -3.4507e+00, -4.0971e+00, ..., -3.0708e+00,
-4.3878e+00, -3.0820e+00]],
[[ 2.2429e-01, 1.4353e+00, 1.0697e+00, ..., 3.3172e+00,
2.5949e+00, 4.4770e-01],
[-1.9203e+00, -2.3247e+00, 4.2606e+00, ..., 6.1821e+00,
4.7890e+00, -1.0085e+00],
[-8.0956e-01, 1.8806e+00, 3.0682e+00, ..., 9.6102e-01,
4.0124e+00, -4.1614e-01],
...,
[ 3.5286e-01, 3.3921e-01, 1.0057e+00, ..., 4.8442e-01,
-4.0648e+00, 1.6910e+00],
[ 6.0984e-01, 7.8364e-01, -2.3235e-01, ..., -6.9666e-01,
-2.5012e+00, 1.7518e+00],
[ 5.8450e-01, 1.1161e+00, 7.1372e-01, ..., -2.1557e+00,
-3.9236e+00, -1.0388e-01]],
...,
[[ 1.6896e-01, -3.2840e+00, -4.9599e+00, ..., -5.5513e+00,
-4.8603e+00, -1.5652e+00],
[-2.6350e+00, -5.4749e+00, -8.9832e+00, ..., -4.9877e+00,
-5.3069e+00, -3.6080e+00],
[-4.4213e+00, -6.3582e+00, -1.5230e+00, ..., -3.7928e+00,
-4.5172e+00, -2.9462e+00],
...,
[-2.3306e+00, -1.8512e+00, -3.1286e+00, ..., -3.8730e+00,
-3.4234e+00, -7.9241e+00],
[-3.6641e+00, -2.8636e+00, -1.9778e+00, ..., -2.4999e+00,
-4.9128e+00, -5.2384e+00],
[-1.7964e+00, -2.1070e+00, -2.6526e+00, ..., -1.7035e+00,
-4.0015e+00, -1.2906e+00]],
[[-1.5578e+00, -1.7212e+00, -2.3198e+00, ..., 7.6317e-01,
-6.1729e-01, -9.0709e-01],
[-2.2909e+00, -4.3952e+00, -6.4927e+00, ..., -3.9971e+00,
-2.7084e-01, -7.0506e-01],
[-6.6082e+00, -8.0513e-01, 8.3625e-02, ..., -1.5730e+00,
-8.3788e-01, -2.5072e-01],
...,
[-4.6135e+00, -3.5567e+00, -5.3917e+00, ..., -3.4108e+00,
-1.3285e+00, -3.1888e+00],
[-4.3874e+00, -1.1134e+00, -2.0404e+00, ..., -9.5259e-01,
2.9524e-01, -1.1408e-01],
[-3.3538e+00, -2.4325e+00, -1.5328e+00, ..., 5.7730e-02,
-2.4077e+00, -5.2644e-01]],
[[ 9.6494e-01, 1.2668e+00, 2.9819e+00, ..., 1.7676e+00,
9.4404e-01, -2.3368e+00],
[ 1.7306e+00, 2.7829e+00, 2.0326e+00, ..., 5.6640e+00,
5.0643e+00, 8.7970e-01],
[ 2.9311e+00, 4.4181e+00, -7.2259e-02, ..., 2.7596e+00,
3.8064e+00, 4.5116e-01],
...,
[ 2.8989e+00, 1.7452e+00, 1.4025e+00, ..., -7.2684e+00,
2.2270e+00, 2.0927e+00],
[ 2.9102e+00, 4.7801e+00, 2.6166e+00, ..., -6.9962e+00,
2.4152e+00, 1.2829e+00],
[ 1.8587e+00, 1.9851e+00, 2.6583e-01, ..., 1.4022e+00,
3.0014e+00, 1.1204e+00]]]], device='cuda:0'),
tensor([[[[ 5.1358e-01, 1.7319e+00, 1.7536e+00, ..., 7.3109e-01,
-3.6659e-01, 6.2828e-01],
[ 1.7639e-01, -1.6074e-01, 2.1488e+00, ..., 2.3193e-01,
-1.2117e+00, -2.4670e-01],
[-4.8934e-02, -4.6786e-01, 1.2864e+00, ..., 1.3296e-01,
-5.4990e-01, -6.8857e-01],
...,
[-1.1109e+00, -7.5190e-01, -1.1289e+00, ..., -3.5444e+00,
-3.7676e+00, -8.7374e-01],
[-2.1981e+00, -5.9861e-01, 3.5717e-01, ..., -3.0637e+00,
-2.7982e+00, -7.3717e-03],
[-1.6362e+00, -3.1456e+00, -2.8015e+00, ..., -3.8595e+00,
-3.2950e+00, -1.5189e+00]],
[[-1.3923e+00, -1.9527e+00, -1.8197e+00, ..., -1.6513e+00,
-5.4227e+00, -1.1394e+00],
[-4.9884e+00, -6.6888e+00, -4.4666e+00, ..., -6.4016e+00,
-7.1064e+00, -3.8142e+00],
[-4.3776e+00, -5.2594e+00, -3.3664e+00, ..., -7.2086e+00,
-5.8864e+00, -4.1597e+00],
...,
[-6.2951e+00, -8.0845e+00, -7.2476e+00, ..., -8.2482e+00,
-9.4260e+00, -2.2813e+00],
[-9.0749e+00, -9.6880e+00, -8.7279e+00, ..., -1.0821e+01,
-1.1060e+01, -3.9264e+00],
[-1.9809e+00, -4.3709e+00, -4.1814e+00, ..., -4.2584e+00,
-4.2686e+00, -6.1154e-01]],
[[-2.0044e-01, 8.8009e-01, 7.3297e-01, ..., 1.7788e+00,
1.9770e+00, 3.0527e-01],
[ 1.8179e-01, 1.4311e-01, -1.0833e+00, ..., 5.6073e-01,
1.8197e+00, -6.4108e-01],
[ 9.0283e-01, 2.6241e+00, 7.9884e-01, ..., 1.4775e+00,
1.3572e+00, 7.5585e-01],
...,
[ 5.6843e-02, 1.5951e-01, 8.4737e-02, ..., 9.2026e-01,
6.6728e-01, -1.0230e+00],
[-1.0002e+00, 3.1132e-01, -1.3227e+00, ..., 4.2317e-01,
8.7095e-01, -4.5173e-01],
[-1.2377e-02, 6.7727e-01, 1.7430e-01, ..., 2.6541e-02,
-5.5589e-01, -1.8594e+00]],
...,
[[-9.2570e-01, -6.3057e-01, -6.3837e-01, ..., -1.8269e+00,
-2.6743e+00, 1.5069e-01],
[-4.2836e+00, -4.4760e+00, -2.1804e+00, ..., -3.2220e+00,
-5.3738e+00, -2.9761e+00],
[-2.5725e+00, -1.4336e+00, -1.5559e-01, ..., -9.5353e-01,
-4.7818e+00, -2.0276e+00],
...,
[-1.9728e+00, -2.9696e+00, -3.3960e+00, ..., -4.4965e+00,
-6.5540e+00, -3.1932e+00],
[-2.8063e+00, -5.0187e+00, -2.2323e+00, ..., -4.5818e+00,
-5.3427e+00, -1.5899e+00],
[-2.4707e+00, -3.5620e+00, -3.0413e+00, ..., -2.7213e+00,
-2.4985e+00, 3.5115e-01]],
[[-9.9050e-01, 1.3327e-01, -2.7273e+00, ..., -7.9920e-01,
1.4799e-01, 2.3035e-01],
[-4.0473e+00, -9.0419e-01, -3.8586e+00, ..., -1.7847e+00,
-2.0480e+00, 2.6840e-01],
[-3.8889e+00, -1.6468e+00, -3.0566e+00, ..., -1.1114e+00,
-2.5681e+00, -6.5631e-01],
...,
[-3.3473e+00, -2.0123e+00, -3.9080e+00, ..., -3.2751e+00,
-2.5713e+00, -1.0501e+00],
[-2.9890e+00, -4.0579e-01, -1.4561e+00, ..., -1.7398e+00,
-8.9083e-01, 3.5141e-02],
[-3.1280e+00, -2.7773e+00, -2.6728e+00, ..., -1.4764e+00,
-1.0101e+00, 2.5362e-01]],
[[ 1.6684e+00, 7.3040e-01, 5.7897e-01, ..., 1.1721e+00,
1.2383e+00, 1.4889e-01],
[ 3.1947e+00, 2.2477e+00, 1.9279e+00, ..., 2.1149e+00,
2.6429e+00, 3.5392e+00],
[ 2.1717e+00, 1.8426e-01, 5.0138e-01, ..., 2.1279e+00,
2.6024e+00, 2.0854e+00],
...,
[ 2.3750e+00, 3.0838e+00, 2.1518e+00, ..., 2.3836e+00,
1.5167e+00, 2.2653e-01],
[ 4.0528e+00, 4.9080e+00, 3.7191e+00, ..., 2.9468e+00,
1.8326e+00, 1.5387e+00],
[ 2.7144e+00, 2.7638e+00, 2.1322e+00, ..., 5.1801e-01,
7.5643e-01, 1.0125e+00]]],
[[[ 2.4918e-01, 1.4380e+00, 4.8205e-01, ..., -6.6968e-01,
-3.8461e-01, -1.1685e-01],
[ 3.9289e-01, 2.6967e-01, 1.5297e+00, ..., 1.3188e-01,
2.7142e-01, 1.1566e+00],
[-4.6911e-01, 7.8243e-02, 1.0967e+00, ..., -1.1070e+00,
-1.1792e-01, 9.7571e-01],
...,
[-7.5824e-01, -3.6330e+00, -6.1420e+00, ..., 1.7897e+00,
-2.2166e+00, -2.8756e+00],
[-9.9988e-01, -3.5990e+00, -5.2826e+00, ..., 2.0612e+00,
-2.5456e-01, -2.8603e+00],
[-2.7349e+00, -6.0333e+00, -5.0434e+00, ..., -1.4288e+00,
-3.1827e+00, -4.2393e+00]],
[[-7.7112e-01, -2.2503e+00, -3.3667e+00, ..., -2.7981e+00,
-3.9150e+00, -3.9326e-01],
[-4.5481e+00, -6.0775e+00, -5.1488e+00, ..., -1.0600e+00,
-5.2463e+00, -1.7902e+00],
[-3.4142e+00, -4.1446e+00, -4.7570e+00, ..., -2.9853e+00,
-3.3519e+00, -5.4334e-01],
...,
[-7.1951e+00, -6.5628e+00, -4.9467e+00, ..., -3.3024e+00,
-7.3082e+00, -4.6683e+00],
[-8.4746e+00, -8.4321e+00, -6.0787e+00, ..., -3.1371e+00,
-7.4238e+00, -6.5342e+00],
[-2.8202e+00, -3.8787e+00, -3.7902e+00, ..., -2.3983e+00,
-4.0840e+00, -2.9322e+00]],
[[-3.2357e-01, 2.0663e+00, 9.1165e-01, ..., 2.3594e+00,
2.8312e+00, 1.1074e+00],
[-1.8069e-01, 1.0343e+00, 3.4962e-01, ..., -3.6189e+00,
9.5137e-01, -4.8629e-01],
[-1.6838e-01, 1.9755e+00, 2.0582e+00, ..., -7.2064e+00,
1.3579e+00, 6.8003e-01],
...,
[ 1.5170e+00, 7.5198e-01, 2.7892e+00, ..., 1.5949e+00,
2.1090e+00, 3.5140e-01],
[ 1.1196e+00, 1.8266e-01, 1.7869e+00, ..., 2.3262e+00,
1.2971e+00, 1.6377e-01],
[ 1.5777e+00, 1.2783e+00, 1.2079e+00, ..., 1.5057e+00,
1.6386e+00, 6.0112e-01]],
...,
[[-7.8932e-01, -1.8018e+00, -7.6535e-01, ..., -3.2853e+00,
-3.4699e+00, -6.4558e-01],
[-3.1238e+00, -4.3753e+00, -1.9554e+00, ..., -8.4404e-01,
-2.1593e+00, -2.3610e+00],
[-1.3628e+00, -2.0895e+00, -4.0164e-01, ..., -1.4724e+00,
-3.9087e+00, -2.3372e+00],
...,
[-3.1364e+00, -3.6082e+00, -3.6625e+00, ..., -4.9037e+00,
-6.4199e+00, -2.8505e+00],
[-3.6070e+00, -4.4116e+00, -3.3007e+00, ..., -3.9405e+00,
-4.9606e+00, -2.7317e+00],
[-3.5880e+00, -4.0410e+00, -2.0234e+00, ..., -4.3891e+00,
-4.3675e+00, -2.1648e+00]],
[[-1.6906e+00, -2.1146e+00, -2.4974e+00, ..., -1.9124e-01,
3.3841e-01, -7.0623e-01],
[-4.3121e+00, -3.5592e+00, -2.5420e+00, ..., 1.5810e+00,
2.3349e+00, 1.0995e+00],
[-3.3016e+00, -3.1567e+00, -3.0371e+00, ..., 5.7051e-01,
1.5673e+00, 1.1990e+00],
...,
[-3.8638e+00, -3.3935e+00, -3.6833e+00, ..., -5.1829e+00,
-3.7640e+00, -3.4549e-01],
[-4.0960e+00, -1.0980e+00, -1.2551e+00, ..., -5.2662e+00,
-3.3945e+00, -3.0931e-02],
[-4.8446e+00, -2.7179e+00, -1.1020e-01, ..., -4.7214e+00,
-2.8641e+00, -2.1518e-02]],
[[ 1.0649e+00, 5.9199e-01, 4.0023e-01, ..., 2.9486e+00,
2.5793e+00, -1.1759e+00],
[ 3.3552e+00, 3.3479e+00, 2.0477e+00, ..., 5.7482e-01,
3.3624e+00, 3.0888e+00],
[ 1.4340e+00, 6.4968e-01, 1.7751e-01, ..., 1.6740e+00,
4.0879e+00, 1.7656e+00],
...,
[ 3.8025e+00, 1.5698e+00, -6.9890e-01, ..., 3.8659e+00,
2.6978e+00, 1.1226e+00],
[ 4.1999e+00, 3.9788e+00, 4.9508e-01, ..., 3.1735e+00,
3.4229e+00, 3.0558e+00],
[ 2.0858e+00, 5.5844e-01, 2.2716e-01, ..., 1.1437e+00,
1.1508e+00, 2.4381e-01]]],
[[[ 5.4243e-02, 1.1431e+00, -7.7945e-02, ..., -5.6261e-01,
-1.0201e-01, 8.0645e-01],
[-3.6973e-02, -7.2087e-01, -1.3717e-01, ..., -1.6433e+00,
-1.2687e+00, 1.2207e+00],
[-1.0095e+00, -1.0349e+00, 9.6048e-01, ..., -1.7093e+00,
-1.7622e+00, 8.5642e-01],
...,
[-8.0965e-01, -1.8201e+00, -3.1038e-01, ..., -1.8178e+00,
-6.2096e-01, -9.1829e-01],
[-9.9730e-01, -8.8911e-01, -3.6168e-01, ..., -1.2790e+00,
-3.3245e-01, -1.7069e+00],
[-2.7627e+00, -5.1989e+00, -4.5726e+00, ..., -4.8180e+00,
-4.0031e+00, -3.9028e+00]],
[[-1.8435e+00, -3.0311e+00, -4.2294e+00, ..., -3.2168e+00,
-4.6614e+00, -2.3786e-01],
[-6.3310e+00, -6.9297e+00, -4.4186e+00, ..., -6.2994e+00,
-6.9471e+00, -1.4296e+00],
[-4.6831e+00, -5.1721e+00, -4.2022e+00, ..., -5.2593e+00,
-6.1579e+00, -1.2852e+00],
...,
[-5.7896e+00, -5.5828e+00, -3.9519e+00, ..., -6.4773e+00,
-5.1678e+00, -3.9318e+00],
[-7.1389e+00, -7.7863e+00, -6.0684e+00, ..., -7.7116e+00,
-7.8209e+00, -6.5807e+00],
[-2.5702e+00, -4.4624e+00, -4.1319e+00, ..., -4.0242e+00,
-4.1268e+00, -2.6161e+00]],
[[-3.4123e-01, 1.9207e+00, 9.2278e-01, ..., 9.5488e-01,
2.1078e+00, 9.8697e-01],
[ 2.2641e-01, 8.1516e-01, 7.5563e-01, ..., 5.7888e-01,
9.8718e-01, -4.4713e-01],
[ 5.7818e-01, 2.0523e+00, 1.5360e+00, ..., 1.4744e-01,
1.4560e+00, 1.2939e-01],
...,
[ 8.7855e-01, 1.1137e+00, 9.1828e-01, ..., 8.9018e-01,
9.8831e-01, 4.7153e-01],
[ 8.1836e-01, 5.9769e-01, 6.5047e-01, ..., -6.4337e-02,
1.5985e-01, 4.8493e-01],
[ 1.3498e+00, 1.6139e+00, 1.2901e+00, ..., 1.5487e+00,
1.2510e+00, 3.0466e-01]],
...,
[[-1.3613e+00, -2.0285e+00, -2.2090e+00, ..., -2.5230e+00,
-2.6307e+00, -4.6844e-01],
[-4.3807e+00, -4.2826e+00, -2.4353e+00, ..., -4.4328e+00,
-4.1974e+00, -2.5294e+00],
[-2.7274e+00, -2.1895e+00, -1.0393e+00, ..., -2.2310e+00,
-4.6452e+00, -1.4693e+00],
...,
[-2.6655e+00, -2.4831e+00, -1.0395e+00, ..., -2.6876e+00,
-3.4128e+00, -1.9399e+00],
[-3.0265e+00, -2.9564e+00, -1.6754e+00, ..., -1.9624e+00,
-4.1732e+00, -2.4645e+00],
[-3.5354e+00, -3.8325e+00, -3.5088e+00, ..., -3.2156e+00,
-3.7782e+00, -1.7877e+00]],
[[-1.9207e+00, -1.7763e+00, -2.9886e+00, ..., -1.3705e+00,
-1.3163e+00, -1.3127e+00],
[-5.2492e+00, -2.8867e+00, -3.0074e+00, ..., -2.8693e+00,
-1.4203e+00, -6.3664e-01],
[-4.5376e+00, -2.7439e+00, -3.4762e+00, ..., -9.7458e-01,
-9.3162e-01, -2.7644e-01],
...,
[-3.9551e+00, -2.6735e+00, -3.1217e+00, ..., -2.8930e+00,
-3.0710e+00, -1.6421e+00],
[-3.5870e+00, -8.5979e-01, -2.2047e+00, ..., -1.3221e+00,
-2.9701e+00, -1.3984e+00],
[-4.3996e+00, -3.0315e+00, -2.9670e+00, ..., -1.8705e+00,
-3.2133e+00, -1.3294e+00]],
[[ 1.5798e+00, 6.8540e-01, 4.8177e-01, ..., -6.7544e-01,
-6.6302e-02, -4.3262e-01],
[ 4.3704e+00, 3.4204e+00, 1.8779e+00, ..., 1.5171e+00,
1.8565e+00, 2.3423e+00],
[ 1.7317e+00, 5.4357e-01, 2.8220e-01, ..., 8.7660e-01,
1.2333e+00, 1.6910e+00],
...,
[ 2.9220e+00, 9.4251e-01, 1.4115e+00, ..., 1.9169e+00,
2.2098e+00, 1.3688e+00],
[ 3.7428e+00, 3.5397e+00, 2.3231e+00, ..., 3.1662e+00,
2.6179e+00, 2.0718e+00],
[ 1.9038e+00, 6.7362e-01, -1.8280e-02, ..., -2.0177e-01,
-2.3976e-01, -1.5664e-01]]],
[[[ 3.2304e-02, 9.4616e-01, 1.0233e+00, ..., -6.3810e-01,
2.9203e-01, 7.9974e-01],
[ 3.4279e-01, 1.4403e+00, 1.9867e+00, ..., -2.3708e+00,
-1.3687e+00, 9.7863e-01],
[-2.3939e-01, 1.2897e+00, 1.0572e-01, ..., -2.5098e+00,
-1.1601e+00, 8.4231e-02],
...,
[-1.3497e+00, -1.6109e+00, 1.2181e+00, ..., -2.6258e-01,
-8.2287e-01, -1.2837e+00],
[-1.3497e+00, -1.1897e+00, 1.1559e-01, ..., -5.5282e-01,
-8.5948e-01, -2.2334e+00],
[-2.7399e+00, -5.2038e+00, -4.3472e+00, ..., -4.9710e+00,
-4.7104e+00, -4.0288e+00]],
[[-9.0316e-01, -1.8743e+00, -3.4909e+00, ..., -3.0413e+00,
-4.4564e+00, -3.2895e-01],
[-4.2225e+00, -5.8343e+00, -8.4625e+00, ..., -7.9965e+00,
-7.7576e+00, -1.9261e+00],
[-3.5267e+00, -5.6886e+00, -1.1043e+01, ..., -3.8466e+00,
-6.9281e+00, -2.4678e+00],
...,
[-5.9821e+00, -5.5890e+00, -3.3435e+00, ..., -6.6017e+00,
-6.1700e+00, -4.0929e+00],
[-7.2737e+00, -8.0057e+00, -6.2328e+00, ..., -8.3681e+00,
-8.7964e+00, -6.7599e+00],
[-2.5393e+00, -4.3939e+00, -3.8887e+00, ..., -4.7610e+00,
-4.3829e+00, -2.6072e+00]],
[[-3.9204e-01, 1.9751e+00, 1.5369e+00, ..., 6.7025e-01,
1.9645e+00, 8.7783e-01],
[-3.0144e-01, 1.3109e+00, 1.1305e+00, ..., 4.3835e-01,
1.0283e+00, -7.8468e-01],
[-1.7885e-01, 2.9153e+00, -3.7969e-01, ..., 4.0965e-01,
1.5708e+00, 1.5910e-02],
...,
[ 7.1579e-01, 1.1974e+00, 6.0697e-01, ..., 9.9859e-01,
3.6407e-01, -1.9571e-01],
[ 8.7961e-01, 7.2336e-01, 6.4600e-01, ..., -6.2264e-01,
-1.2370e+00, 2.1901e-01],
[ 1.3598e+00, 1.5037e+00, 1.2418e+00, ..., 6.2448e-01,
7.8784e-01, 1.7929e-01]],
...,
[[-8.4001e-01, -1.3085e+00, -9.8916e-01, ..., -2.4303e+00,
-2.5308e+00, -3.2006e-01],
[-2.9939e+00, -3.8463e+00, -3.0618e+00, ..., -5.6813e+00,
-5.0165e+00, -2.6245e+00],
[-1.4221e+00, -2.4474e+00, -3.4009e+00, ..., -2.1274e+00,
-3.4927e+00, -2.1821e+00],
...,
[-2.9769e+00, -2.5382e+00, -1.1782e-02, ..., -4.3371e+00,
-4.3596e+00, -2.3998e+00],
[-3.1392e+00, -3.0971e+00, -1.7092e+00, ..., -3.4212e+00,
-5.2942e+00, -3.0064e+00],
[-3.4411e+00, -3.7225e+00, -3.3365e+00, ..., -3.3242e+00,
-4.1634e+00, -1.7590e+00]],
[[-1.7344e+00, -1.9339e+00, -2.5914e+00, ..., -1.6841e+00,
-1.3035e+00, -1.4577e+00],
[-4.0010e+00, -2.8021e+00, -4.0012e+00, ..., -3.5164e+00,
-2.8066e+00, -1.1791e+00],
[-2.8183e+00, -3.4295e+00, -5.9050e+00, ..., 1.2309e+00,
-1.1520e+00, -4.8605e-01],
...,
[-3.7466e+00, -1.8513e+00, -2.6303e+00, ..., -2.7467e+00,
-4.6441e+00, -6.8154e-01],
[-3.7599e+00, -6.6251e-01, -1.8355e+00, ..., -6.9786e-01,
-2.5354e+00, -5.2309e-01],
[-4.3864e+00, -2.8633e+00, -2.6974e+00, ..., -1.8887e+00,
-3.1598e+00, -9.5992e-01]],
[[ 9.9474e-01, 3.1328e-01, 5.0452e-01, ..., -9.8308e-01,
1.0159e-01, -3.3520e-01],
[ 3.2715e+00, 2.8534e+00, 3.0929e+00, ..., 8.5820e-01,
1.9734e+00, 2.3613e+00],
[ 1.4423e+00, 1.0694e+00, 1.8595e+00, ..., 4.0083e-01,
1.4157e+00, 7.5656e-01],
...,
[ 2.8084e+00, 1.3980e+00, 1.1085e+00, ..., 3.1313e+00,
2.7915e+00, 2.2853e+00],
[ 3.7078e+00, 3.5559e+00, 2.4445e+00, ..., 3.0941e+00,
2.7524e+00, 2.1415e+00],
[ 1.9421e+00, 7.8636e-01, 2.4391e-01, ..., 3.4459e-01,
8.4140e-02, -8.0756e-02]]],
[[[ 3.2760e-01, 1.0578e+00, 1.4100e-02, ..., -6.5701e-01,
-1.9067e+00, -5.9671e-01],
[ 5.1453e-01, 5.7743e-01, 1.7514e+00, ..., -2.3793e+00,
-1.2199e+00, -9.7577e-02],
[-4.1255e-01, -9.0405e-01, 6.1007e-01, ..., -5.8996e+00,
-2.3407e+00, -1.7076e+00],
...,
[-5.7874e-01, -1.5834e+00, 6.8333e-01, ..., -3.9704e+00,
-3.0054e+00, -4.2023e-01],
[-2.0339e+00, -1.7005e+00, -8.8075e-01, ..., -3.0305e+00,
-3.0501e+00, -1.0196e+00],
[-1.1172e+00, -3.1998e+00, -3.2666e+00, ..., -4.4106e+00,
-3.4270e+00, -1.3142e+00]],
[[-1.0131e+00, -1.5948e+00, -2.4150e+00, ..., -6.0161e+00,
-7.0735e+00, -1.8795e+00],
[-3.7737e+00, -3.9892e+00, -1.9363e+00, ..., -5.9419e+00,
-4.2517e+00, -1.1846e+00],
[-3.7697e+00, -3.9379e+00, -2.8110e+00, ..., -8.3596e+00,
-6.0072e+00, -1.4956e+00],
...,
[-5.5614e+00, -5.5392e+00, -3.7835e+00, ..., -1.1224e+01,
-7.7886e+00, -1.6319e+00],
[-7.5882e+00, -7.5528e+00, -6.4047e+00, ..., -8.1520e+00,
-8.5314e+00, -3.9146e+00],
[-1.7181e+00, -3.2510e+00, -3.0805e+00, ..., -4.6196e+00,
-3.3787e+00, -5.3378e-01]],
[[-9.2133e-02, 1.2007e+00, 1.7211e+00, ..., 6.9315e-01,
2.6724e+00, 1.0850e+00],
[-6.8315e-02, 7.7823e-03, 1.2573e-01, ..., -3.7242e-01,
2.1389e+00, 3.0248e-01],
[ 6.6724e-01, 1.2892e+00, 7.0119e-01, ..., 2.0286e+00,
3.8564e+00, 2.4343e+00],
...,
[ 7.2203e-02, 6.3310e-01, 8.5774e-01, ..., -1.6005e+00,
1.6009e+00, 3.1927e-02],
[ 4.3050e-01, 4.8583e-01, 9.6418e-01, ..., 3.2658e-01,
-1.3919e-01, -4.3779e-01],
[ 5.2099e-01, 1.0531e+00, 1.2526e+00, ..., -1.0936e+00,
-1.4450e-01, -1.2888e+00]],
...,
[[-6.0240e-01, -1.7818e+00, -2.1753e+00, ..., -3.5434e+00,
-4.8871e+00, -1.2033e+00],
[-2.6829e+00, -2.5714e+00, -6.2392e-01, ..., -3.8413e+00,
-3.5833e+00, -1.8765e+00],
[-1.9992e+00, -1.2665e+00, -2.9581e-01, ..., -4.6689e+00,
-4.6061e+00, -2.0983e+00],
...,
[-1.5818e+00, -1.5478e+00, -2.8948e-01, ..., -6.5709e+00,
-5.7066e+00, -1.8071e+00],
[-3.1990e+00, -3.1306e+00, -1.8782e+00, ..., -6.9164e-01,
-3.9578e+00, -8.7322e-01],
[-1.4925e+00, -1.8902e+00, -1.7546e+00, ..., -1.4163e+00,
-2.3856e+00, 8.9114e-01]],
[[-1.2731e+00, -2.3612e+00, -4.1161e+00, ..., -2.4954e+00,
-1.4613e+00, -1.8145e-01],
[-3.5514e+00, -2.9438e+00, -3.4413e+00, ..., -1.6768e+00,
-2.0178e+00, 4.9856e-01],
[-4.6007e+00, -3.3724e+00, -2.1030e+00, ..., -5.3241e+00,
-4.4203e+00, -2.4166e+00],
...,
[-4.3617e+00, -3.6036e+00, -3.2712e+00, ..., -4.6407e+00,
-4.7320e+00, -1.0505e+00],
[-4.0959e+00, -1.9279e+00, -2.0962e+00, ..., -2.2581e+00,
-2.6557e+00, 3.0742e-01],
[-3.2311e+00, -3.1762e+00, -1.5711e+00, ..., 4.9456e-01,
-5.6897e-01, 7.3631e-01]],
[[ 1.2301e+00, -5.2298e-02, 1.0308e+00, ..., -3.7668e-01,
3.4869e-01, -4.9934e-02],
[ 2.2776e+00, 1.0666e+00, 7.6227e-01, ..., -1.4227e-01,
7.9205e-01, 2.5830e+00],
[ 1.4114e+00, 4.3694e-01, 3.0142e-01, ..., 1.4028e-01,
2.1251e+00, 1.2795e+00],
...,
[ 2.4720e+00, 1.3059e+00, 1.1922e+00, ..., 2.0736e+00,
1.0830e-01, 8.0198e-01],
[ 2.8634e+00, 3.6743e+00, 2.3013e+00, ..., 2.5416e-01,
1.0470e+00, 2.1675e+00],
[ 1.9442e+00, 1.5869e+00, 3.5981e-01, ..., -2.6864e-02,
-3.7693e-01, 6.7922e-01]]]], device='cuda:0'),
tensor([[[[-2.4352e+00, -4.0422e+00, -1.0492e-01, ..., -8.1768e-01,
-3.4515e+00, -1.7311e+00],
[-1.6672e+00, -2.7832e+00, -5.2750e-01, ..., 4.1512e-01,
-1.4209e+00, -7.6373e-01],
[-4.5737e+00, -4.0015e+00, -2.8641e+00, ..., -3.9281e-01,
-1.1606e+00, -1.7556e+00],
...,
[-9.1381e-01, -2.2945e+00, -3.8721e-01, ..., -3.5765e+00,
-1.6655e+00, -1.5713e+00],
[-1.0506e+00, -9.7257e-01, -1.0675e-01, ..., -2.8940e+00,
-1.6720e+00, -1.7035e+00],
[-2.8070e+00, -5.3397e+00, -4.4135e+00, ..., -4.5133e+00,
-3.9335e+00, -3.7110e+00]],
[[-3.6410e+00, -4.0861e+00, -2.0572e-02, ..., -3.5417e+00,
-8.5906e+00, -2.3382e+00],
[-4.3504e+00, -1.1740e+00, 1.0772e+00, ..., -5.0152e+00,
-7.9774e+00, -3.0929e+00],
[-8.3673e+00, -7.8260e+00, -6.7176e+00, ..., -9.5305e+00,
-1.1065e+01, -6.5674e+00],
...,
[-5.7756e+00, -5.4645e+00, -3.3283e+00, ..., -4.9052e+00,
-5.9491e+00, -3.8352e+00],
[-7.4060e+00, -8.2305e+00, -6.8689e+00, ..., -8.3074e+00,
-8.5435e+00, -6.8239e+00],
[-2.6065e+00, -4.6501e+00, -4.1272e+00, ..., -3.9050e+00,
-4.2155e+00, -2.5106e+00]],
[[-1.2650e-01, 1.3456e+00, -1.6160e-01, ..., 2.3757e+00,
3.2529e+00, 9.5016e-01],
[-7.3536e-01, -1.0090e+00, -3.9086e+00, ..., 1.5388e+00,
3.1637e+00, 2.5621e-01],
[ 1.7216e-01, -6.3444e+00, -1.3055e+01, ..., -1.7091e-01,
3.2409e-01, -2.7743e-01],
...,
[ 2.8686e-01, 1.0998e+00, 5.2116e-02, ..., 2.1552e+00,
4.1601e-02, -2.1918e-01],
[ 8.5763e-01, 8.7471e-01, 5.9685e-01, ..., -5.8886e-01,
1.4222e-01, 6.0261e-01],
[ 1.3719e+00, 1.6187e+00, 1.3067e+00, ..., 1.6921e+00,
1.1068e+00, 5.1220e-02]],
...,
[[-3.4298e+00, -2.8739e+00, 6.5991e-01, ..., -4.7177e+00,
-5.9967e+00, -1.9205e+00],
[-3.9603e+00, 1.4966e-01, 2.7584e+00, ..., -4.1230e+00,
-5.2978e+00, -2.6819e+00],
[-7.1240e+00, -6.8086e+00, -5.8054e+00, ..., -4.1452e+00,
-6.5855e+00, -3.5038e+00],
...,
[-2.0311e+00, -1.9202e+00, 5.7321e-01, ..., -2.1114e+00,
-4.8238e+00, -2.3236e+00],
[-3.0203e+00, -3.1621e+00, -1.7209e+00, ..., -3.4949e+00,
-5.4251e+00, -2.9010e+00],
[-3.6420e+00, -4.0379e+00, -3.4215e+00, ..., -3.3065e+00,
-3.7981e+00, -1.5545e+00]],
[[-1.5212e+00, -1.2281e+00, -1.4807e+00, ..., -2.4940e+00,
-9.1749e-01, 5.6746e-01],
[-1.0648e+00, 1.0608e+00, -2.1000e-01, ..., -3.8018e+00,
-2.5109e+00, -1.1453e-01],
[-2.4657e+00, -1.3397e-01, -3.3367e+00, ..., -1.5508e+00,
-1.5713e+00, 6.5125e-03],
...,
[-3.1836e+00, -1.9664e+00, -2.5896e+00, ..., -3.5580e+00,
-2.7358e+00, -2.6338e-02],
[-3.6798e+00, -5.6982e-01, -2.2053e+00, ..., -5.3106e-01,
-1.9210e+00, -6.0562e-02],
[-4.5641e+00, -2.9967e+00, -2.8118e+00, ..., -1.8756e+00,
-3.0341e+00, -1.1403e+00]],
[[ 1.5348e+00, -2.3018e+00, -1.5009e+00, ..., 1.8823e+00,
1.8488e+00, -1.1261e-01],
[ 4.0731e+00, 8.6809e-01, -3.2859e-01, ..., 2.6076e+00,
3.5984e+00, 4.0898e+00],
[ 3.8242e+00, 2.0114e+00, -1.1240e+00, ..., 3.5661e+00,
3.6445e+00, 1.9198e+00],
...,
[ 2.2225e+00, 3.9447e-01, 4.6816e-01, ..., -1.8798e-01,
1.4732e+00, 3.0620e+00],
[ 3.9659e+00, 3.9530e+00, 2.7497e+00, ..., 3.4956e+00,
2.3402e+00, 1.5930e+00],
[ 1.9822e+00, 7.9162e-01, 2.4102e-01, ..., -4.0337e-01,
-3.0683e-01, -1.6227e-01]]],
[[[ 1.7595e-01, 9.6975e-01, -6.2743e-01, ..., -2.5891e+00,
-3.3275e+00, -1.3528e+00],
[ 5.2682e-01, 6.6256e-02, 4.0529e-01, ..., -7.1722e-01,
-1.0923e+00, -6.0357e-01],
[ 3.5970e-01, 6.4323e-01, 1.1493e+00, ..., 3.7989e-01,
-2.9197e-01, -1.1395e+00],
...,
[-4.7878e-01, -2.1105e+00, -9.1936e-01, ..., -2.6677e+00,
-7.4547e-01, -5.8892e-01],
[-1.0889e+00, -1.3308e+00, -1.0871e+00, ..., 1.7197e+00,
1.1199e+00, -1.8075e+00],
[-2.8418e+00, -5.5904e+00, -4.3863e+00, ..., -1.9139e+00,
-2.2046e+00, -3.3557e+00]],
[[-1.3400e+00, -2.4785e+00, -4.8045e+00, ..., -5.5604e+00,
-8.0324e+00, -1.9024e+00],
[-4.2826e+00, -5.7220e+00, -4.2385e+00, ..., -5.3360e+00,
-7.4331e+00, -3.0928e+00],
[-3.4636e+00, -5.9879e+00, -6.1172e+00, ..., -8.3579e+00,
-9.8714e+00, -5.6647e+00],
...,
[-5.6343e+00, -5.4270e+00, -3.3296e+00, ..., -3.7767e+00,
-3.6987e+00, -2.8989e+00],
[-8.0039e+00, -8.7805e+00, -6.7339e+00, ..., -2.5041e-01,
-4.8015e+00, -6.9169e+00],
[-2.6322e+00, -4.7510e+00, -3.6752e+00, ..., -4.0726e+00,
-3.7750e+00, -1.9884e+00]],
[[ 4.0698e-01, 2.1827e+00, 1.7734e+00, ..., 1.9969e+00,
2.9061e+00, 9.6108e-01],
[-4.2718e-01, 4.8527e-01, 8.4432e-01, ..., 1.1408e+00,
2.8002e+00, 2.5757e-01],
[ 3.2223e-01, 1.6946e+00, 6.3891e-01, ..., -5.9642e-01,
4.8074e-01, -3.2251e-01],
...,
[ 2.5350e-01, 7.1589e-01, 2.4750e-01, ..., 2.2882e+00,
6.6127e-01, -2.6003e-01],
[ 8.6404e-01, 8.3483e-01, 8.5292e-01, ..., -4.1670e-01,
6.5110e-01, 7.4192e-01],
[ 1.4947e+00, 1.8512e+00, 1.0265e+00, ..., -2.2572e+00,
2.5715e-01, 8.9815e-01]],
...,
[[-8.4730e-01, -2.2815e+00, -3.1848e+00, ..., -4.6975e+00,
-5.7626e+00, -1.6835e+00],
[-2.3986e+00, -3.7806e+00, -2.3875e+00, ..., -3.1687e+00,
-5.1200e+00, -2.3687e+00],
[-1.5876e+00, -2.3245e+00, -2.3895e+00, ..., -4.1598e+00,
-6.2167e+00, -2.5276e+00],
...,
[-2.2267e+00, -1.7941e+00, 4.2779e-01, ..., -2.1661e+00,
-3.0232e+00, -1.4445e+00],
[-3.9077e+00, -3.7837e+00, -2.2707e+00, ..., 6.9209e-01,
-3.2829e+00, -2.7913e+00],
[-3.7915e+00, -3.9231e+00, -3.1924e+00, ..., -4.1070e+00,
-4.0732e+00, -1.3350e+00]],
[[-1.4669e+00, -1.6196e+00, -2.5244e+00, ..., -2.1753e+00,
-8.0432e-01, 5.3488e-01],
[-3.7264e+00, -2.2637e+00, -2.5871e+00, ..., -2.5156e+00,
-2.0329e+00, 7.8825e-02],
[-2.6077e+00, -1.2502e+00, -1.9364e+00, ..., -7.0134e-01,
-1.0159e+00, 3.4589e-01],
...,
[-2.7711e+00, -1.8742e+00, -2.8347e+00, ..., -7.0719e-01,
-7.7799e-01, -1.3032e-01],
[-3.6871e+00, -9.2588e-01, -1.9376e+00, ..., -1.0720e-01,
-3.4685e+00, -1.6457e+00],
[-4.4716e+00, -3.0765e+00, -2.3529e+00, ..., -1.5289e+00,
-3.6217e+00, -1.4357e+00]],
[[ 1.4480e+00, 8.7905e-01, 1.2095e+00, ..., 8.3883e-01,
1.7378e+00, -1.3247e-01],
[ 2.8062e+00, 2.7372e+00, 2.2282e+00, ..., 2.7318e+00,
3.3790e+00, 3.8826e+00],
[ 2.1249e+00, 1.7585e+00, 1.6580e+00, ..., 2.7702e+00,
2.5607e+00, 1.8116e+00],
...,
[ 2.4854e+00, 5.5831e-01, 3.2863e-01, ..., 1.5862e+00,
2.1294e+00, 1.5006e+00],
[ 4.4383e+00, 4.2645e+00, 2.3572e+00, ..., 9.0891e-01,
2.2316e+00, 2.4991e+00],
[ 2.1691e+00, 7.9262e-01, 1.6300e-01, ..., -1.3951e+00,
-6.3899e-01, -6.0996e-01]]],
[[[ 4.2982e-01, 1.6908e+00, 7.9633e-01, ..., -5.3819e-01,
1.1713e-01, 6.7356e-01],
[ 4.9436e-01, 1.2834e-01, -5.2092e-01, ..., -1.3941e+00,
-1.1482e+00, 1.0886e+00],
[-5.6824e-01, -9.8836e-02, -3.3468e+00, ..., -1.2922e+00,
-1.9946e+00, 6.6787e-01],
...,
[-1.1856e+00, -2.1867e+00, -8.3917e-02, ..., -2.0738e-01,
-2.6550e+00, -1.3480e+00],
[-1.2683e+00, -1.0478e+00, -2.0504e-01, ..., 4.2306e-01,
-2.2878e-01, -1.9311e+00],
[-2.7347e+00, -5.2314e+00, -4.4926e+00, ..., -4.9683e+00,
-4.6368e+00, -4.0553e+00]],
[[-5.5230e-01, -2.5034e+00, -6.0237e+00, ..., -3.1589e+00,
-4.8169e+00, -3.6443e-01],
[-4.7630e+00, -6.3609e+00, -7.9162e+00, ..., -6.1458e+00,
-7.1621e+00, -1.6256e+00],
[-3.9862e+00, -5.9234e+00, -7.3879e+00, ..., -6.9938e+00,
-7.2964e+00, -1.5501e+00],
...,
[-5.9706e+00, -5.9384e+00, -3.4352e+00, ..., -9.0067e+00,
-9.2955e+00, -5.0860e+00],
[-7.1558e+00, -7.8973e+00, -6.1666e+00, ..., -6.7657e+00,
-8.1685e+00, -6.8248e+00],
[-2.5509e+00, -4.5043e+00, -4.0977e+00, ..., -4.5114e+00,
-4.6951e+00, -2.7797e+00]],
[[-3.3189e-01, 2.3890e+00, 8.3207e-01, ..., 8.9755e-01,
2.1470e+00, 9.6137e-01],
[-2.7752e-01, 5.9300e-01, 6.6629e-01, ..., 3.6619e-01,
9.6008e-01, -4.0100e-01],
[ 1.6768e-02, 8.2474e-01, 1.8945e+00, ..., 3.8029e-01,
1.4815e+00, 6.3584e-03],
...,
[ 9.5280e-01, 1.0949e+00, 9.8002e-01, ..., 1.6592e-02,
8.7613e-01, 4.2527e-03],
[ 9.0415e-01, 6.5403e-01, 5.5795e-01, ..., -1.4241e-02,
2.0546e-02, -4.9709e-02],
[ 1.3832e+00, 1.5239e+00, 1.2377e+00, ..., 1.3864e+00,
7.0581e-01, 4.5342e-02]],
...,
[[-5.8045e-01, -1.7023e+00, -1.7970e+00, ..., -2.7738e+00,
-2.6836e+00, -4.5648e-01],
[-2.6134e+00, -4.5699e+00, -3.6425e+00, ..., -4.6108e+00,
-4.3462e+00, -2.5005e+00],
[-1.1871e+00, -3.4060e+00, -3.4663e+00, ..., -2.0550e+00,
-5.4127e+00, -1.3110e+00],
...,
[-2.8387e+00, -2.5620e+00, -1.0027e+00, ..., -4.8108e+00,
-5.5494e+00, -1.8271e+00],
[-2.9926e+00, -2.9771e+00, -2.0304e+00, ..., -1.1419e+00,
-3.4798e+00, -2.2807e+00],
[-3.5006e+00, -3.8428e+00, -3.3303e+00, ..., -3.5517e+00,
-4.2317e+00, -1.9434e+00]],
[[-1.4868e+00, -2.3257e+00, -2.6107e+00, ..., -1.5304e+00,
-1.5062e+00, -1.3472e+00],
[-3.7920e+00, -3.2493e+00, -3.5390e+00, ..., -2.6580e+00,
-1.7692e+00, -9.0241e-01],
[-3.3701e+00, -2.4679e+00, -1.2993e+00, ..., -9.8655e-01,
-1.6286e+00, -8.0593e-01],
...,
[-4.2637e+00, -2.3888e+00, -2.7592e+00, ..., -4.6814e+00,
-5.0990e+00, -1.5218e-01],
[-3.7846e+00, -6.0987e-01, -2.3146e+00, ..., -7.1909e-01,
-2.9205e+00, -6.8616e-01],
[-4.5076e+00, -2.9358e+00, -2.9579e+00, ..., -1.6144e+00,
-2.7564e+00, -6.3045e-01]],
[[ 1.1245e+00, 7.3429e-01, 1.3963e+00, ..., -6.5436e-01,
1.4460e-01, -4.5640e-01],
[ 3.4226e+00, 3.0763e+00, 3.2374e+00, ..., 1.6797e+00,
1.9111e+00, 2.4743e+00],
[ 1.7833e+00, 1.9745e+00, 1.3805e+00, ..., 1.5706e+00,
1.1793e+00, 1.6425e+00],
...,
[ 2.9094e+00, 1.0301e+00, 1.2096e+00, ..., 3.9739e+00,
2.8584e+00, 2.3139e+00],
[ 3.6390e+00, 3.5647e+00, 2.3144e+00, ..., 2.3369e+00,
2.2023e+00, 2.8889e+00],
[ 1.9377e+00, 8.6015e-01, 5.0836e-02, ..., 2.5307e-01,
3.5951e-02, 1.3254e-01]]],
[[[ 8.9918e-02, 1.4348e+00, -2.5286e+00, ..., -1.5300e+00,
-1.0568e+00, -8.8924e-01],
[ 1.3924e+00, 1.5637e+00, -4.1451e+00, ..., -1.3658e+00,
2.4737e-01, -1.1292e+00],
[ 1.4896e+00, 1.0562e+00, -5.8579e+00, ..., -1.7827e+00,
-3.8297e-01, -1.9615e+00],
...,
[-3.8001e-01, -1.4367e+00, 1.0686e-01, ..., -3.2496e+00,
-2.5350e+00, 7.8707e-02],
[-2.1994e+00, -1.8568e+00, 3.8522e-01, ..., -2.5218e+00,
-2.3790e+00, 1.9303e-01],
[-2.3489e+00, -4.2429e+00, -2.8247e+00, ..., -3.4239e+00,
-3.1943e+00, -1.6021e+00]],
[[-9.8214e-01, -1.7473e+00, -7.2310e+00, ..., -3.0196e+00,
-4.6673e+00, -2.4675e+00],
[-2.5858e+00, -3.4424e+00, -9.0946e+00, ..., -3.6371e+00,
-4.8920e+00, -4.5644e+00],
[-2.3938e+00, -3.1548e+00, -8.8804e+00, ..., -3.0051e+00,
-3.8120e+00, -3.6545e+00],
...,
[-6.8514e+00, -9.2379e+00, -6.8844e+00, ..., -8.1847e+00,
-8.3733e+00, -1.7626e+00],
[-9.7851e+00, -1.0441e+01, -7.6504e+00, ..., -9.6375e+00,
-1.0368e+01, -3.8458e+00],
[-2.5793e+00, -4.9869e+00, -4.1463e+00, ..., -3.8062e+00,
-4.0980e+00, -7.2560e-01]],
[[ 8.3358e-01, 1.2647e+00, 1.2650e+00, ..., 1.3107e+00,
1.8354e+00, 8.6023e-01],
[ 8.0506e-01, 5.0746e-01, 1.0175e+00, ..., 9.7879e-01,
2.2931e+00, 9.4076e-01],
[ 6.1770e-01, 1.7131e+00, 7.0477e-01, ..., 8.7540e-01,
2.5514e+00, 1.9970e+00],
...,
[ 1.8367e-02, 6.5890e-01, -1.1890e+00, ..., 1.3891e+00,
1.3479e+00, -1.7191e-01],
[-5.2272e-01, 4.6717e-01, -1.4466e+00, ..., 3.2644e-01,
1.1420e+00, -3.3534e-01],
[ 6.5607e-01, 1.2594e+00, -2.4915e-01, ..., -5.6884e-01,
-4.0246e-01, -1.7722e+00]],
...,
[[-4.8493e-01, -1.4170e+00, -3.1993e+00, ..., -3.3263e+00,
-3.3112e+00, -1.2151e+00],
[-1.6725e+00, -2.2508e+00, -4.2945e+00, ..., -4.3717e+00,
-4.2725e+00, -3.4598e+00],
[-4.7139e-01, -7.6949e-01, -3.3802e+00, ..., -4.3157e+00,
-4.0456e+00, -2.9870e+00],
...,
[-2.6656e+00, -4.1315e+00, -1.6088e+00, ..., -5.3693e+00,
-6.1565e+00, -2.6124e+00],
[-3.6398e+00, -5.2395e+00, -1.5016e+00, ..., -3.8624e+00,
-5.3975e+00, -1.3440e+00],
[-3.2086e+00, -3.5598e+00, -2.4792e+00, ..., -2.6707e+00,
-2.9618e+00, 3.9399e-01]],
[[-1.6022e+00, -4.1536e+00, -4.4172e+00, ..., -3.3396e+00,
-2.7449e+00, -5.2878e-01],
[-4.0427e+00, -7.1440e+00, -5.7061e+00, ..., -4.2476e+00,
-5.8063e+00, -1.9868e+00],
[-3.8370e+00, -6.8421e+00, -5.3957e+00, ..., -3.4851e+00,
-4.7937e+00, -1.9651e+00],
...,
[-4.7616e+00, -2.3538e+00, -4.4559e+00, ..., -4.1543e+00,
-2.8392e+00, -1.0322e+00],
[-3.7856e+00, 3.1280e-01, -1.5089e+00, ..., -2.3172e+00,
-1.2558e+00, 4.2462e-01],
[-3.8630e+00, -2.6157e+00, -2.2840e+00, ..., -1.1441e+00,
-1.3810e+00, 3.6702e-01]],
[[ 9.0547e-01, 2.6483e-01, 1.8780e+00, ..., -2.1233e-02,
1.4025e+00, 5.7881e-01],
[ 1.9367e+00, 1.4866e+00, 9.7170e-01, ..., 5.6683e-02,
1.9006e+00, 3.1196e+00],
[ 1.6149e+00, 1.2455e+00, 6.1470e-01, ..., -1.2765e-01,
1.1258e+00, 8.2532e-01],
...,
[ 4.1110e+00, 3.3867e+00, 3.3268e+00, ..., 2.4612e+00,
2.8309e+00, 6.7272e-01],
[ 4.6580e+00, 4.2577e+00, 3.4046e+00, ..., 2.5450e+00,
2.2338e+00, 1.5031e+00],
[ 2.7991e+00, 2.6040e+00, 1.9474e+00, ..., 5.1216e-01,
6.4457e-01, 1.0579e+00]]],
[[[ 4.9351e-01, 1.6017e+00, 1.2995e+00, ..., 1.6751e-01,
-9.0366e-01, 4.9870e-01],
[ 1.3769e-01, -3.0548e-01, 1.3522e+00, ..., 5.7601e-01,
-1.2454e+00, -2.9001e-01],
[-1.2316e-03, -1.8257e-01, 7.3657e-01, ..., 1.1190e+00,
-8.0871e-01, -7.9769e-01],
...,
[-9.2725e-01, -5.1670e+00, -2.2627e+00, ..., 1.4386e+00,
-2.3255e+00, -1.5511e+00],
[-2.3308e+00, -3.8923e+00, 5.8865e-01, ..., 1.1566e+00,
-2.4426e+00, -5.8375e-01],
[-1.6674e+00, -4.1535e+00, -2.1441e+00, ..., -1.8016e+00,
-3.1024e+00, -1.9575e+00]],
[[-1.4589e+00, -1.8812e+00, -2.0041e+00, ..., -3.4395e+00,
-5.6142e+00, -1.1478e+00],
[-4.9400e+00, -6.5436e+00, -4.3818e+00, ..., -5.8775e+00,
-7.2576e+00, -3.8440e+00],
[-4.1791e+00, -4.9717e+00, -2.3623e+00, ..., -5.1642e+00,
-6.1055e+00, -4.2471e+00],
...,
[-6.2698e+00, -8.8027e+00, -5.0089e+00, ..., -2.9319e+00,
-8.3259e+00, -2.6863e+00],
[-9.1218e+00, -1.0371e+01, -7.5767e+00, ..., -7.5663e+00,
-1.2011e+01, -4.1945e+00],
[-1.8372e+00, -4.3171e+00, -3.7219e+00, ..., -3.4209e+00,
-4.8457e+00, -7.9760e-01]],
[[-6.2949e-02, 1.0185e+00, 3.5271e-01, ..., 1.0714e+00,
1.5152e+00, 3.7700e-01],
[ 2.6048e-01, 2.3031e-01, -9.5713e-01, ..., 1.1004e+00,
1.6201e+00, -6.2476e-01],
[ 8.3073e-01, 2.6118e+00, 1.0001e+00, ..., 2.4064e+00,
2.0771e+00, 6.6224e-01],
...,
[ 6.6120e-01, 2.0793e+00, -1.4423e+00, ..., 1.3926e+00,
1.5634e+00, -1.0682e+00],
[-7.2937e-01, 5.8646e-01, -2.5026e+00, ..., 2.7346e+00,
1.5623e+00, -8.6526e-01],
[-3.9867e-02, 2.5897e-01, -1.5677e+00, ..., 2.4732e+00,
-2.3169e-02, -1.7200e+00]],
...,
[[-9.9026e-01, -6.4960e-01, -7.5880e-01, ..., -1.5075e+00,
-3.1383e+00, -5.4743e-03],
[-4.3391e+00, -4.4727e+00, -2.4619e+00, ..., -2.9677e+00,
-5.7514e+00, -2.9762e+00],
[-2.5563e+00, -1.2815e+00, -1.0768e+00, ..., -1.0172e+00,
-4.6759e+00, -1.9171e+00],
...,
[-2.3240e+00, -4.2818e+00, -1.3730e+00, ..., -4.2025e+00,
-6.0425e+00, -2.8475e+00],
[-2.9702e+00, -5.9061e+00, -1.7212e+00, ..., -6.1853e+00,
-4.9836e+00, -1.6670e+00],
[-2.5065e+00, -3.8268e+00, -2.2271e+00, ..., -3.1766e+00,
-1.4131e+00, 1.5728e-01]],
[[-1.0926e+00, 3.0779e-01, -2.0550e+00, ..., -1.9441e-01,
-3.5131e-01, 7.0361e-02],
[-4.0779e+00, -9.5567e-01, -2.3497e+00, ..., -1.1311e+00,
-2.4978e+00, -1.6062e-02],
[-3.7182e+00, -1.6676e+00, -2.7954e+00, ..., -1.5453e+00,
-3.1411e+00, -6.1710e-01],
...,
[-4.5233e+00, -2.4705e+00, -1.1820e+00, ..., -2.4190e+00,
-3.4978e+00, -1.2822e+00],
[-3.9056e+00, -1.8519e-01, 1.0157e+00, ..., -2.7716e+00,
-1.7106e+00, -2.7105e-01],
[-3.4412e+00, -2.1841e+00, -3.0244e-01, ..., -2.5811e+00,
-1.0394e+00, 3.0846e-01]],
[[ 1.7044e+00, 7.1752e-01, 5.1610e-01, ..., 9.2047e-01,
1.0700e+00, 5.7869e-03],
[ 3.1783e+00, 2.0573e+00, 1.9044e+00, ..., 2.7546e+00,
2.5256e+00, 3.3569e+00],
[ 2.1135e+00, 1.4007e-01, 6.8221e-01, ..., 2.2769e+00,
2.1001e+00, 1.8837e+00],
...,
[ 2.4515e+00, 1.9455e+00, 9.6646e-01, ..., 3.8224e+00,
1.0766e+00, -3.2857e-01],
[ 4.1051e+00, 3.6834e+00, 3.8463e+00, ..., 4.5663e+00,
1.4215e+00, 1.4238e+00],
[ 2.5270e+00, 2.4217e+00, 2.5019e+00, ..., 5.4203e-01,
4.7690e-01, 1.0880e+00]]]], device='cuda:0'),
tensor([[[[ 5.7721e-01, -3.5452e+00, -1.8521e+00, ..., -5.4203e-01,
-4.1434e+00, -2.2903e+00],
[ 1.7982e+00, -5.6019e+00, -3.0125e+00, ..., -2.4517e+00,
-2.1002e+00, -1.1233e+00],
[ 8.7149e-01, -5.7643e+00, -4.0218e+00, ..., 2.2022e+00,
1.2652e+00, -1.4747e+00],
...,
[-2.9327e-01, -1.4582e+00, -4.7288e+00, ..., 1.7377e+00,
-6.0190e-01, -1.5292e+00],
[-7.4784e-01, -1.9132e+00, -5.6262e+00, ..., 1.8062e+00,
-6.8507e-01, -2.5316e+00],
[-2.8801e+00, -5.2157e+00, -4.5129e+00, ..., -2.4872e+00,
-3.6554e+00, -4.4963e+00]],
[[-8.8066e-01, -6.4215e+00, -6.6037e-01, ..., -6.1825e+00,
-1.0924e+01, -3.0418e+00],
[-3.4100e+00, -9.1001e+00, -3.8808e+00, ..., -6.7261e+00,
-9.1395e+00, -4.3737e+00],
[-3.1821e+00, -8.5658e+00, -6.5748e+00, ..., -2.9085e-01,
-7.2371e+00, -7.7143e+00],
...,
[-6.5258e+00, -5.7903e+00, -9.4232e+00, ..., -1.8138e+00,
-7.0493e+00, -3.7509e+00],
[-8.6547e+00, -8.3147e+00, -7.1688e+00, ..., -4.3651e+00,
-9.8361e+00, -6.5913e+00],
[-2.8713e+00, -3.2103e+00, -2.8592e+00, ..., -2.7930e+00,
-4.6532e+00, -2.8215e+00]],
[[ 4.2372e-01, 1.3467e+00, 9.4213e-02, ..., 2.3658e+00,
2.6837e+00, 8.0569e-01],
[-5.3961e-01, 1.2721e+00, -4.2438e+00, ..., 5.6274e+00,
3.7634e+00, -9.8623e-02],
[-1.8028e-01, 9.6200e-01, -4.1975e+00, ..., 4.1339e+00,
4.3843e+00, 1.5594e-01],
...,
[ 7.0605e-01, -9.9020e-01, -8.4393e-02, ..., 2.9952e+00,
-1.2309e-01, -4.6060e-01],
[ 7.8596e-01, -8.6494e-01, 1.3373e+00, ..., 3.2953e+00,
2.1714e-01, -3.5682e-01],
[ 1.4889e+00, 7.3326e-01, 1.0834e+00, ..., 3.0364e+00,
1.1812e+00, 2.6008e-01]],
...,
[[-9.8385e-01, -2.6007e+00, 2.3114e+00, ..., -4.1542e+00,
-6.0717e+00, -2.5357e+00],
[-2.5829e+00, -5.3522e+00, -1.4557e+00, ..., -5.5339e+00,
-5.5840e+00, -3.6013e+00],
[-1.5247e+00, -4.0716e+00, -1.3207e+00, ..., -1.5829e+00,
-5.3685e+00, -3.0977e+00],
...,
[-2.0058e+00, -1.9462e+00, -2.4132e+00, ..., -3.2345e+00,
-2.1144e+00, -2.0493e+00],
[-3.4987e+00, -3.7384e+00, -3.2947e+00, ..., -4.8974e+00,
-3.0972e+00, -3.0360e+00],
[-3.4043e+00, -2.7723e+00, -1.2728e+00, ..., -5.3920e+00,
-2.9994e+00, -2.1633e+00]],
[[-2.7839e+00, -1.5096e-01, 1.3906e-01, ..., -3.2123e+00,
-1.7819e+00, 2.0797e-02],
[-5.6441e+00, -3.1274e+00, -2.3101e-01, ..., -5.2172e+00,
-1.8929e+00, -2.7158e-01],
[-3.4757e+00, -2.9806e+00, -2.8802e-01, ..., -3.8894e+00,
-3.6624e+00, 1.9309e-01],
...,
[-2.6248e+00, -2.7654e+00, -3.6285e+00, ..., -2.1375e+00,
-1.4465e+00, -1.0358e+00],
[-3.8763e+00, -1.0038e+00, -1.0954e+00, ..., -4.0431e+00,
-3.0735e+00, -1.2704e+00],
[-4.2980e+00, -2.0148e+00, -1.0198e+00, ..., -4.4753e+00,
-3.5618e+00, -1.1716e+00]],
[[ 1.5113e+00, 5.5391e-01, -9.6260e-01, ..., 1.8237e+00,
1.9242e+00, -4.3935e-01],
[ 2.4423e+00, 2.9140e+00, 2.4613e+00, ..., 3.0311e+00,
4.3046e+00, 3.4284e+00],
[ 2.3198e+00, 1.9414e+00, 2.9014e+00, ..., 2.8652e+00,
5.5130e+00, 2.2894e+00],
...,
[ 2.5647e+00, 1.3858e+00, 1.8561e+00, ..., 3.1356e+00,
1.2658e+00, 1.4381e+00],
[ 4.5871e+00, 4.2288e+00, -1.9878e-01, ..., 3.7083e+00,
2.0844e+00, 2.1607e+00],
[ 2.2636e+00, 4.2715e-01, -1.0737e+00, ..., 1.0409e+00,
-5.6617e-01, -3.2072e-01]]],
[[[ 1.9437e-01, 1.3281e+00, 5.3977e-01, ..., -6.1860e-01,
6.0611e-02, 6.6936e-01],
[ 5.4833e-01, 5.8689e-01, 9.1346e-01, ..., -1.4949e+00,
-1.0737e+00, 9.6262e-01],
[ 3.3844e-01, 8.5018e-01, 7.1367e-01, ..., -1.3114e+00,
-6.3075e-01, 2.4573e-01],
...,
[ 4.3286e-01, -6.8447e-01, 2.9522e-01, ..., -1.0844e-01,
-6.7987e-01, -1.0957e+00],
[-1.3398e+00, -1.5255e+00, -3.4099e+00, ..., -1.8795e+00,
-2.6120e+00, -1.3635e+00],
[-7.0314e-02, -2.6617e+00, -3.2789e+00, ..., -2.5009e+00,
-2.3167e+00, -1.0371e+00]],
[[-8.2618e-01, -2.3094e+00, -4.1334e+00, ..., -3.1608e+00,
-4.7742e+00, -3.9580e-01],
[-4.6113e+00, -6.2522e+00, -6.4572e+00, ..., -6.1021e+00,
-7.6004e+00, -1.8589e+00],
[-3.5984e+00, -5.7164e+00, -8.1208e+00, ..., -4.8934e+00,
-7.8080e+00, -3.0893e+00],
...,
[-4.5119e+00, -5.0592e+00, -5.3520e+00, ..., -4.2336e+00,
-5.3093e+00, -2.3563e+00],
[-7.6017e+00, -8.3903e+00, -8.9439e+00, ..., -7.8831e+00,
-9.0541e+00, -4.1171e+00],
[-1.0957e+00, -2.8163e+00, -2.8043e+00, ..., -2.8800e+00,
-2.7640e+00, -3.4141e-01]],
[[-2.8223e-01, 2.1384e+00, 1.1935e+00, ..., 7.8812e-01,
2.1241e+00, 9.6440e-01],
[-7.6076e-02, 1.1053e+00, 8.4189e-01, ..., 4.0059e-01,
1.1755e+00, -4.8133e-01],
[-9.5226e-03, 1.8966e+00, 1.3683e+00, ..., 6.0280e-01,
1.4234e+00, -1.0975e-01],
...,
[ 6.6003e-02, 7.7812e-01, 1.0195e+00, ..., 1.3830e+00,
1.0550e+00, 5.0403e-01],
[ 3.5404e-01, 6.1275e-01, 1.7763e+00, ..., 1.5944e+00,
2.0669e+00, 5.3769e-01],
[-5.4535e-01, 4.7076e-01, 1.0465e+00, ..., 5.1271e-01,
8.2292e-01, -7.0426e-01]],
...,
[[-8.0558e-01, -1.8106e+00, -1.7909e+00, ..., -2.4654e+00,
-2.8015e+00, -4.4054e-01],
[-2.9421e+00, -4.3093e+00, -2.5312e+00, ..., -4.0086e+00,
-4.4285e+00, -2.6754e+00],
[-1.3673e+00, -2.8572e+00, -1.5465e+00, ..., -1.7411e+00,
-4.6893e+00, -1.6136e+00],
...,
[-8.1642e-01, -1.6600e+00, -1.7415e+00, ..., -1.7246e+00,
-3.7409e+00, -1.7870e+00],
[-2.7166e+00, -3.8307e+00, -3.3958e+00, ..., -3.1696e+00,
-4.6665e+00, -1.5168e+00],
[-8.0104e-01, -1.6116e+00, -7.3849e-01, ..., -6.4905e-01,
-1.1257e+00, 1.2289e+00]],
[[-1.7207e+00, -2.1981e+00, -2.8076e+00, ..., -1.4508e+00,
-1.4666e+00, -1.3522e+00],
[-4.2096e+00, -3.4937e+00, -3.3160e+00, ..., -3.0177e+00,
-1.8166e+00, -7.0242e-01],
[-2.5953e+00, -2.1275e+00, -3.2556e+00, ..., 5.4658e-01,
-8.5890e-02, -2.9371e-01],
...,
[-3.7049e+00, -4.1058e+00, -5.0948e+00, ..., -2.3396e+00,
-2.8485e+00, -1.2971e+00],
[-4.5035e+00, -2.8103e+00, -2.9577e+00, ..., -1.5961e+00,
-2.2926e+00, -5.8039e-01],
[-3.1251e+00, -2.3992e+00, -1.0114e+00, ..., -5.3921e-01,
-9.7451e-01, -1.4341e-03]],
[[ 1.0564e+00, 5.6963e-01, 6.9028e-01, ..., -7.2397e-01,
1.7521e-01, -4.2324e-01],
[ 3.5588e+00, 3.3939e+00, 2.4854e+00, ..., 1.6585e+00,
2.1526e+00, 2.4607e+00],
[ 1.6986e+00, 1.8533e+00, 2.2358e+00, ..., 1.9687e+00,
2.4450e+00, 1.6512e+00],
...,
[ 2.6874e+00, 1.1961e+00, 1.8824e+00, ..., 1.6372e+00,
1.3042e+00, 4.5479e-01],
[ 3.4005e+00, 4.4503e+00, 2.4773e+00, ..., 2.2780e+00,
1.2095e+00, 1.6309e+00],
[ 2.2410e+00, 1.6487e+00, -1.3395e-02, ..., 3.6818e-01,
1.2121e-01, 1.0252e+00]]],
[[[ 1.1318e+00, 1.9229e-01, 8.9799e-01, ..., -3.7553e+00,
-2.7729e+00, -1.0872e+00],
[ 1.4144e-01, -3.5395e+00, 3.2532e+00, ..., -2.6372e+00,
-2.9165e+00, -2.2182e-01],
[-2.1640e+00, -2.9741e+00, 3.0471e+00, ..., -1.8228e+00,
-3.2334e+00, 8.8275e-02],
...,
[ 1.2018e+00, -5.8133e-01, -9.7343e-01, ..., -3.1294e+00,
-3.9161e+00, 4.3070e-01],
[ 5.0056e-01, 1.0257e-01, -3.8911e-01, ..., -4.0855e+00,
-3.8376e+00, 4.6376e-01],
[-4.6432e-01, -3.4312e+00, -3.2187e+00, ..., -3.8250e+00,
-3.3020e+00, -1.4890e+00]],
[[ 2.2325e-01, -2.4286e-01, -1.6856e-01, ..., -7.7862e+00,
-7.5268e+00, -1.3284e+00],
[-2.0690e+00, -3.5651e+00, 1.4750e+00, ..., -9.0017e+00,
-8.7152e+00, -2.3807e+00],
[-3.7093e+00, -3.2528e+00, 1.5771e+00, ..., -7.8752e+00,
-8.0277e+00, -2.4426e+00],
...,
[-1.7518e+00, -3.2673e+00, -3.2034e+00, ..., -8.0960e+00,
-9.1689e+00, -2.0503e+00],
[-3.9649e+00, -5.3259e+00, -4.4853e+00, ..., -9.7806e+00,
-1.1041e+01, -4.4036e+00],
[-2.8017e-01, -3.1543e+00, -3.3580e+00, ..., -4.0857e+00,
-3.7291e+00, -8.0636e-01]],
[[-4.0514e-01, 8.9709e-01, 1.7996e+00, ..., 8.0581e-01,
2.1316e+00, 9.6077e-01],
[-1.9488e+00, -4.4281e+00, -2.0540e-01, ..., 1.3807e+00,
2.6464e+00, 5.0661e-02],
[-2.4103e-01, -1.7674e+00, 9.3546e-01, ..., -4.1618e-01,
1.6677e+00, -3.3457e-01],
...,
[ 4.3269e-01, 3.0721e+00, 9.4639e-01, ..., 3.4976e-01,
1.5883e+00, -1.2550e+00],
[ 2.6922e-01, 2.4989e+00, 6.2723e-01, ..., 6.8942e-02,
1.4251e+00, -1.5321e+00],
[-2.4571e-01, 1.6669e+00, 7.2205e-01, ..., -5.5932e-02,
-3.1112e-01, -1.9062e+00]],
...,
[[ 8.3070e-01, 1.0525e+00, -3.3107e-01, ..., -5.4824e+00,
-5.5872e+00, -7.4142e-01],
[-1.4171e+00, -2.8209e-01, 1.6291e-02, ..., -3.8953e+00,
-5.6542e+00, -1.7057e+00],
[-2.3389e+00, 1.2528e+00, -3.8733e-01, ..., -3.5485e+00,
-6.2714e+00, -1.2878e+00],
...,
[-1.4997e+00, -3.0847e+00, -7.8810e-01, ..., -4.2934e+00,
-5.7968e+00, -1.5449e+00],
[-1.4783e+00, -3.1512e+00, -1.1527e+00, ..., -3.5685e+00,
-5.3455e+00, -9.3648e-01],
[-1.1762e+00, -3.4733e+00, -2.3422e+00, ..., -1.8641e+00,
-2.2765e+00, 3.7516e-01]],
[[-9.9353e-01, -1.2735e+00, -1.9862e+00, ..., -2.4442e+00,
-7.6880e-01, -5.3564e-01],
[-6.7634e-01, -1.8022e-01, -1.7601e+00, ..., -4.0732e+00,
-2.2586e+00, -2.5177e-01],
[-1.7391e+00, 8.7662e-01, -2.3570e+00, ..., -2.0308e+00,
-1.9524e+00, -2.7256e-01],
...,
[-2.7621e+00, -4.2010e+00, -2.2072e+00, ..., -3.5309e+00,
-2.6431e+00, -1.1150e+00],
[-2.8913e+00, -2.0052e+00, -7.1567e-01, ..., -1.8417e+00,
-9.5977e-01, 1.0903e+00],
[-3.2221e+00, -2.9690e+00, -1.0357e+00, ..., -2.3751e-01,
-9.4767e-01, 6.3473e-01]],
[[ 7.4274e-01, -1.3097e-01, 7.5104e-01, ..., 2.1050e-02,
5.0216e-01, -1.4551e+00],
[ 3.1110e-02, -2.8448e+00, 1.5787e+00, ..., 3.7980e+00,
2.7031e+00, 2.2811e+00],
[-3.5050e+00, -3.5918e+00, 2.4219e+00, ..., 2.3866e+00,
1.1363e+00, 1.2436e+00],
...,
[ 1.6874e+00, 1.1409e+00, 1.0758e+00, ..., 2.3058e+00,
2.0197e+00, 6.6543e-01],
[ 2.9118e+00, 3.1400e+00, 2.1929e+00, ..., 1.5226e+00,
1.0997e+00, 1.8839e+00],
[ 1.9830e+00, 5.2833e-01, 5.1959e-01, ..., 2.1546e-01,
3.0611e-02, 9.1498e-01]]],
[[[ 1.1813e+00, 9.5155e-01, 4.7725e-01, ..., -3.6933e+00,
-2.6512e+00, -1.0971e+00],
[-1.4101e+00, -2.0643e+00, -1.8150e+00, ..., -2.7259e+00,
-2.3461e+00, -2.9811e-01],
[ 6.0947e-02, 2.1749e+00, -7.9467e-01, ..., -4.7747e+00,
-2.9175e+00, -1.0687e-01],
...,
[-3.6725e-01, -3.5159e+00, 3.3228e-01, ..., -2.4435e+00,
-4.2624e+00, -3.3455e-01],
[ 7.2104e-01, 5.5201e-02, 1.2168e+00, ..., -4.4677e+00,
-3.4021e+00, -2.1869e-01],
[ 1.5723e-01, -2.8209e+00, -3.0963e+00, ..., -4.1258e+00,
-3.4094e+00, -1.5393e+00]],
[[ 1.0144e+00, -1.0287e+00, -1.9568e+00, ..., -7.9988e+00,
-7.8813e+00, -1.4185e+00],
[-6.3133e+00, -6.2544e+00, -3.1570e+00, ..., -8.9836e+00,
-9.2125e+00, -2.8487e+00],
[ 8.5002e-01, 2.1744e+00, 2.2129e+00, ..., -6.2556e+00,
-6.8875e+00, -3.7913e+00],
...,
[-1.8758e+00, -3.9396e+00, -9.8362e-01, ..., -1.1150e+01,
-1.0221e+01, -3.1603e+00],
[-3.6618e+00, -6.3115e+00, -8.3129e+00, ..., -8.9064e+00,
-1.0195e+01, -4.4160e+00],
[-3.4500e-01, -3.3482e+00, -3.5241e+00, ..., -4.1100e+00,
-3.6445e+00, -8.0454e-01]],
[[ 3.6831e-01, 2.1548e+00, 2.2649e+00, ..., 6.4220e-01,
2.2134e+00, 9.3199e-01],
[ 7.0729e-01, 3.3078e+00, 3.9512e+00, ..., 6.2233e-01,
2.5914e+00, -1.3433e-01],
[-7.1599e-01, -1.1023e-01, -7.7176e-01, ..., 1.8771e+00,
2.7499e+00, -2.0029e-01],
...,
[ 4.5274e-02, 1.3654e+00, -2.0401e+00, ..., -1.3441e+00,
-2.8533e-01, -5.1087e-01],
[-9.0543e-01, 6.1707e-01, -1.3482e+00, ..., -5.3318e-01,
1.0407e+00, -1.2447e+00],
[-1.0946e+00, 4.3760e-01, -3.9892e-01, ..., -5.4000e-01,
-1.0588e+00, -2.2214e+00]],
...,
[[-3.5821e-01, -9.1733e-01, -1.6464e+00, ..., -5.5801e+00,
-5.6359e+00, -6.7758e-01],
[-4.0079e+00, -4.8242e+00, -3.7595e+00, ..., -4.8551e+00,
-4.9726e+00, -2.0495e+00],
[ 1.7007e+00, 9.9559e-01, 1.7988e+00, ..., -5.2053e+00,
-5.5541e+00, -1.8468e+00],
...,
[-1.3815e+00, -2.8281e+00, 3.6038e-01, ..., -6.2956e+00,
-6.8530e+00, -1.5209e+00],
[-1.3264e+00, -4.1359e+00, -2.6438e+00, ..., -2.0761e+00,
-4.6671e+00, -8.8477e-01],
[-6.8857e-01, -3.4159e+00, -1.2848e+00, ..., -1.5937e+00,
-2.3053e+00, 2.9777e-01]],
[[-1.1387e+00, -5.9696e-01, -1.0291e+00, ..., -2.2667e+00,
-7.5458e-01, -6.7005e-01],
[-5.5912e+00, -4.4274e+00, -2.8567e+00, ..., -3.1740e+00,
-2.4127e+00, -6.5911e-01],
[-1.4126e+00, 4.4972e-01, 2.4647e+00, ..., -2.2658e+00,
-9.6229e-01, -1.3444e-01],
...,
[-3.6296e+00, -2.3633e+00, 1.0079e+00, ..., -4.4826e+00,
-4.5975e+00, -1.4728e+00],
[-3.2471e+00, -2.1124e+00, -3.8661e-02, ..., -5.2282e-01,
-2.0606e+00, 5.2380e-01],
[-3.0329e+00, -2.4244e+00, 9.7231e-02, ..., 2.2439e-01,
-8.7812e-01, 6.4808e-01]],
[[ 5.0903e-01, 2.4768e-02, 4.6004e-01, ..., -3.1672e-01,
6.7548e-01, -1.3504e+00],
[ 1.3423e+00, 2.2632e+00, 2.1231e+00, ..., 3.4328e+00,
2.9544e+00, 2.4884e+00],
[ 1.1200e+00, 1.8222e+00, -1.2873e-01, ..., 1.3993e+00,
1.4115e+00, 1.5352e+00],
...,
[ 1.8696e+00, -2.0845e-01, 1.4074e-01, ..., 3.4101e+00,
2.1124e+00, 1.1574e+00],
[ 3.2195e+00, 3.1616e+00, 4.0988e+00, ..., 6.1989e-01,
3.1259e-01, 2.3066e+00],
[ 2.1454e+00, 5.4294e-01, 6.8292e-01, ..., 3.0285e-01,
-1.2734e-01, 8.7900e-01]]],
[[[-6.4673e-02, 1.2601e+00, 6.2998e-01, ..., -3.4881e+00,
2.9455e+00, 1.9375e+00],
[-3.1940e-01, -2.1961e-01, 1.6590e-01, ..., -7.9986e-01,
5.8305e+00, 1.9316e+00],
[-1.4449e+00, -1.7742e-02, 2.4300e+00, ..., 5.7054e+00,
4.0591e+00, -2.4861e+00],
...,
[ 2.5360e-01, -7.0524e-01, -2.2576e+00, ..., 5.6540e+00,
-5.0099e-01, -3.8388e+00],
[-9.3950e-01, 1.4692e+00, 1.5312e-01, ..., -7.8193e+00,
-5.7836e+00, -2.7363e+00],
[-3.2174e+00, -4.9792e+00, -3.4662e+00, ..., -6.5488e+00,
-6.7473e+00, -4.8633e+00]],
[[-9.8711e-01, -2.1875e+00, -3.5171e+00, ..., -1.0659e+00,
3.0220e+00, 1.1677e+00],
[-4.0448e+00, -5.3010e+00, -4.4781e+00, ..., -1.3338e+00,
8.0527e-01, -2.4876e+00],
[-3.0497e+00, -4.2484e+00, -4.0622e+00, ..., -1.0254e+00,
-9.7973e+00, -5.7669e+00],
...,
[-4.1108e+00, -2.2151e+00, -5.5601e+00, ..., -7.5429e+00,
-1.4155e+01, -8.6181e+00],
[-7.5154e+00, -9.3453e+00, -3.7839e+00, ..., -1.6259e+01,
-1.0406e+01, -5.4244e+00],
[-2.5146e+00, -4.7579e+00, -3.9219e+00, ..., -4.8680e+00,
-4.4463e+00, -2.1725e+00]],
[[-3.9694e-01, 1.9945e+00, 6.8183e-01, ..., -6.7539e-01,
1.9460e+00, 2.1746e+00],
[-7.2453e-02, 9.8197e-01, 6.7283e-01, ..., 2.9801e+00,
-9.3231e-02, 2.0228e-01],
[-2.4099e-01, 2.7001e+00, 2.9978e+00, ..., 1.0076e+00,
2.1559e+00, 2.0812e+00],
...,
[-7.3053e-01, 6.6550e-01, 1.6488e+00, ..., -3.6294e+00,
-4.0565e+00, -1.1740e-02],
[-4.5781e-01, -2.5979e+00, -8.3295e-01, ..., -5.1733e+00,
5.6378e-01, -1.5474e+00],
[ 1.3762e+00, 1.3627e+00, -8.5431e-01, ..., 8.7778e-01,
7.1998e-01, -1.0490e+00]],
...,
[[-1.1423e+00, -1.6843e+00, -7.7964e-01, ..., 6.7502e-01,
-3.4340e-01, -1.9572e+00],
[-3.4036e+00, -4.4687e+00, -2.4956e+00, ..., -1.6241e+00,
-6.9087e+00, -6.8907e+00],
[-2.0084e+00, -2.7183e+00, -1.6691e+00, ..., -2.5284e+00,
-6.9022e+00, -5.3057e+00],
...,
[-9.4100e-01, -2.0605e+00, -2.7927e+00, ..., -5.0480e+00,
-9.5251e+00, -5.2343e+00],
[-3.0251e+00, -2.9420e+00, -4.8941e-01, ..., -8.3452e+00,
-6.3882e+00, -2.0715e+00],
[-3.9635e+00, -4.3062e+00, -2.2636e+00, ..., -4.4920e+00,
-5.1696e+00, -2.4714e+00]],
[[-1.5832e+00, -1.8034e+00, -2.0196e+00, ..., 9.6229e-01,
-1.7922e+00, 1.7764e-01],
[-4.1467e+00, -2.9556e+00, -2.3331e+00, ..., 2.3201e+00,
-3.7020e+00, -2.4918e+00],
[-2.9714e+00, -1.4835e+00, -8.6897e-01, ..., 2.6467e+00,
-5.6342e+00, -3.8915e+00],
...,
[-1.5693e+00, -3.9534e+00, -4.9559e+00, ..., 1.4472e+00,
-4.8664e+00, -4.0052e+00],
[-2.1024e+00, 1.0329e+00, -1.8421e+00, ..., -7.0833e-01,
-1.4189e+00, -9.0639e-02],
[-4.2659e+00, -1.3561e+00, -1.1878e-01, ..., -1.7592e+00,
-1.7956e+00, -1.3191e-01]],
[[ 1.1295e+00, 3.8211e-01, 4.3554e-01, ..., -4.2674e+00,
7.5540e-01, 1.7632e+00],
[ 3.1674e+00, 3.2304e+00, 1.9749e+00, ..., -2.6060e+00,
3.2590e+00, 2.4941e+00],
[ 7.4209e-01, 1.4179e+00, 7.7022e-01, ..., 4.7026e+00,
6.0593e+00, 1.3253e+00],
...,
[ 2.6230e+00, 1.2713e+00, 4.7902e+00, ..., 5.5582e+00,
6.3208e+00, 2.9629e+00],
[ 4.1545e+00, 2.8824e+00, 1.2592e+00, ..., 5.2065e+00,
6.7158e-01, 2.5451e+00],
[ 1.8499e+00, 1.2134e+00, -3.5184e-01, ..., -4.1524e-01,
1.3695e+00, 1.6839e+00]]]], device='cuda:0'),
tensor([[[[-1.4029e-01, 5.7087e-01, -1.0766e+00, ..., -1.3544e+00,
-1.0526e+00, -8.2764e-01],
[ 9.3958e-01, -1.3886e+00, -3.0155e+00, ..., -2.1111e-01,
3.3696e-01, -9.3147e-01],
[ 1.0901e+00, -4.0797e+00, -2.3818e+00, ..., 1.8996e-01,
-1.1003e-01, -1.7381e+00],
...,
[-1.3810e+00, -1.2936e+00, -7.5847e-01, ..., -2.4908e+00,
-3.5338e+00, 1.1615e+00],
[-2.0881e+00, -8.5369e-01, -1.9543e+00, ..., -1.9637e+00,
-2.0961e+00, 1.1202e+00],
[-2.0153e+00, -4.1081e+00, -3.5346e+00, ..., -3.3296e+00,
-2.9870e+00, -1.3419e+00]],
[[-3.8846e-01, -1.6112e+00, -2.7064e+00, ..., -2.9870e+00,
-4.6722e+00, -2.3135e+00],
[-2.0649e+00, -5.6598e+00, -4.3832e+00, ..., -3.3613e+00,
-5.0982e+00, -4.6195e+00],
[-2.5286e+00, -6.6969e+00, -1.7760e+00, ..., -2.9805e+00,
-4.3673e+00, -4.0041e+00],
...,
[-7.0600e+00, -7.9159e+00, -6.0387e+00, ..., -7.2725e+00,
-6.6214e+00, -5.4740e-01],
[-8.9026e+00, -9.8861e+00, -7.6609e+00, ..., -8.1322e+00,
-8.1947e+00, -3.2383e+00],
[-2.2200e+00, -4.5943e+00, -3.1313e+00, ..., -3.2221e+00,
-3.8118e+00, -9.0973e-01]],
[[ 5.5993e-01, 1.2023e+00, 1.1145e+00, ..., 1.4770e+00,
1.8144e+00, 9.1383e-01],
[ 4.8048e-01, 1.2745e+00, 1.4364e+00, ..., 1.2801e+00,
2.1301e+00, 8.3534e-01],
[ 6.4869e-02, 3.2059e+00, 1.5455e+00, ..., 1.2849e+00,
2.3472e+00, 1.8366e+00],
...,
[ 5.9929e-01, 1.0128e+00, -4.3920e-01, ..., -7.1174e-01,
-7.0521e-01, -2.0286e+00],
[ 2.5492e-01, 1.6901e-02, -9.6414e-01, ..., -2.1103e+00,
-1.4774e+00, -1.4944e+00],
[ 5.7337e-01, 4.4519e-01, -6.4800e-01, ..., -5.4713e-01,
-5.1277e-01, -1.8973e+00]],
...,
[[-9.0548e-01, -1.8823e+00, -1.9752e+00, ..., -2.5549e+00,
-3.3241e+00, -1.2254e+00],
[-2.7012e+00, -4.8194e+00, -3.8200e+00, ..., -2.8679e+00,
-4.3817e+00, -3.4565e+00],
[-1.8215e+00, -3.8771e+00, -1.0226e+00, ..., -1.8933e+00,
-3.8292e+00, -2.8007e+00],
...,
[-2.2413e+00, -1.3834e+00, -7.5323e-01, ..., -4.9358e-01,
-2.8728e+00, -9.7907e-01],
[-2.7276e+00, -3.0123e+00, -1.6360e+00, ..., -5.9119e-01,
-3.0785e+00, -6.4763e-01],
[-2.3802e+00, -2.5273e+00, -1.6717e+00, ..., -7.7442e-01,
-2.3100e+00, 4.6203e-01]],
[[-1.3335e+00, -2.1059e+00, -2.9050e+00, ..., -2.8695e+00,
-2.8053e+00, -5.7849e-01],
[-5.0026e+00, -5.7276e+00, -4.2701e+00, ..., -3.9969e+00,
-5.5408e+00, -1.7855e+00],
[-4.9421e+00, -4.8224e+00, -3.2819e+00, ..., -2.2924e+00,
-4.1338e+00, -1.6163e+00],
...,
[-3.5832e+00, -2.0642e+00, -2.4433e+00, ..., -3.8674e-01,
-4.2983e-01, 1.4402e-01],
[-3.4087e+00, 4.1284e-01, -5.1693e-01, ..., 1.1440e+00,
7.3314e-01, 1.7621e+00],
[-3.9768e+00, -1.4986e+00, -6.6717e-01, ..., 3.3536e-01,
-1.1813e-01, 7.1352e-01]],
[[ 6.0826e-01, -1.4524e-01, 8.0694e-01, ..., 5.2312e-01,
1.4277e+00, 5.6281e-01],
[ 1.5984e+00, 1.4244e+00, -4.8389e-01, ..., 9.8349e-01,
1.9748e+00, 3.3236e+00],
[ 1.3619e+00, 8.4156e-03, -1.0199e+00, ..., 8.1280e-01,
1.1488e+00, 1.1268e+00],
...,
[ 3.1865e+00, 2.5949e+00, 9.5741e-01, ..., -1.0927e-02,
1.5493e+00, 1.2849e+00],
[ 4.6543e+00, 5.0657e+00, 3.1320e+00, ..., 1.8843e+00,
1.6331e+00, 2.0214e+00],
[ 2.3702e+00, 2.0437e+00, 5.6055e-01, ..., -5.8468e-01,
-1.1689e-01, 1.0063e+00]]],
[[[-1.1128e-01, 7.4544e-01, -3.4544e-01, ..., -1.4230e+00,
-1.2773e+00, -9.4320e-01],
[ 7.4243e-01, 2.1221e-01, 1.3274e-01, ..., -7.7288e-02,
-2.6541e-01, -1.2962e+00],
[ 1.5153e+00, 1.6556e-01, 1.6174e+00, ..., -4.9086e-01,
-1.1888e+00, -2.6337e+00],
...,
[-1.3560e+00, -7.8089e-01, -4.1284e-01, ..., 3.1998e-03,
-1.8981e+00, -1.7337e+00],
[-2.0591e+00, -6.2742e-01, -1.8801e-01, ..., -2.8317e+00,
-1.0643e-01, 1.2469e+00],
[-1.6564e+00, -3.5944e+00, -2.8925e+00, ..., -3.5689e+00,
-2.0201e+00, -3.5236e-01]],
[[-7.1111e-01, -1.4870e+00, -2.5522e+00, ..., -3.7180e+00,
-5.1142e+00, -2.5046e+00],
[-1.9081e+00, -3.4834e+00, -3.9392e+00, ..., -4.4846e+00,
-6.5133e+00, -5.2631e+00],
[-1.8335e+00, -3.6447e+00, -4.4439e+00, ..., -3.5405e+00,
-6.5587e+00, -4.5335e+00],
...,
[-6.1394e+00, -6.4147e+00, -5.5255e+00, ..., -1.0799e+00,
-4.6469e+00, -1.0935e+00],
[-7.7977e+00, -8.2561e+00, -6.4796e+00, ..., -8.4650e-02,
-2.8281e-01, 7.4619e-01],
[-1.8240e+00, -3.8289e+00, -3.5764e+00, ..., -4.3074e+00,
-3.5044e+00, -1.1250e+00]],
[[ 6.9019e-01, 1.1806e+00, 9.9876e-01, ..., 1.4257e+00,
1.6935e+00, 7.5529e-01],
[ 8.5088e-01, 1.5242e+00, 1.7118e+00, ..., 1.2407e+00,
2.4658e+00, 6.6136e-01],
[ 2.1474e-01, 3.0547e+00, 2.5530e+00, ..., 3.6094e+00,
3.1779e+00, 1.6927e+00],
...,
[-2.4012e-01, 5.1893e-01, -2.9896e-01, ..., 6.2280e-01,
1.0542e+00, 1.8701e+00],
[-1.1206e-01, -3.0743e-01, -2.9957e-02, ..., -1.3393e+01,
-8.1672e+00, -2.3876e+00],
[ 4.2974e-01, -3.4130e-03, -1.7194e-02, ..., -5.2116e+00,
-4.3557e+00, -3.9455e+00]],
...,
[[-9.5679e-01, -1.2966e+00, -1.9643e+00, ..., -2.4196e+00,
-3.8968e+00, -1.1594e+00],
[-2.3255e+00, -3.2551e+00, -2.6015e+00, ..., -1.6475e+00,
-5.1748e+00, -4.0777e+00],
[-7.9630e-01, -1.5552e+00, -1.3451e+00, ..., -3.7005e+00,
-4.4525e+00, -4.2240e+00],
...,
[-1.3466e+00, -1.2981e+00, -5.6622e-01, ..., 3.1284e+00,
-1.1336e-01, -1.5487e+00],
[-2.0010e+00, -2.5227e+00, -5.5840e-01, ..., 1.1239e+00,
-6.4100e-01, 1.3971e-01],
[-1.8511e+00, -2.2059e+00, -1.4011e+00, ..., -3.0339e+00,
-2.8124e+00, -1.3233e+00]],
[[-1.5588e+00, -2.3162e+00, -3.5157e+00, ..., -3.2469e+00,
-2.6871e+00, -5.3589e-01],
[-4.4257e+00, -4.7571e+00, -4.2102e+00, ..., -2.8716e+00,
-5.2627e+00, -1.7041e+00],
[-3.9925e+00, -4.9069e+00, -3.4620e+00, ..., -1.5246e+00,
-3.5217e+00, -1.6542e+00],
...,
[-2.8473e+00, -2.0666e+00, -2.2130e+00, ..., 1.0203e+00,
-1.6276e+00, 6.2702e-01],
[-2.8660e+00, 5.9819e-01, -5.7085e-01, ..., 3.9411e+00,
3.0663e+00, 3.2265e+00],
[-3.2755e+00, -1.4166e+00, -1.2358e+00, ..., 1.7191e-01,
3.4974e-01, 1.1548e+00]],
[[ 8.8721e-01, -8.3184e-02, 7.2520e-01, ..., 5.1590e-01,
1.3097e+00, 4.6366e-01],
[ 1.6866e+00, 6.7956e-01, 3.2563e-01, ..., 5.8429e-01,
1.6026e+00, 2.9538e+00],
[ 1.5756e+00, 8.8629e-03, 1.0435e+00, ..., 3.8854e+00,
2.0878e+00, 3.2651e-01],
...,
[ 2.6325e+00, 1.9191e+00, 2.8809e-01, ..., 1.1638e+00,
2.6065e+00, 2.7770e+00],
[ 3.8840e+00, 4.3312e+00, 2.9855e+00, ..., -6.9108e+00,
-4.5491e+00, 6.6535e-01],
[ 2.1926e+00, 1.7630e+00, 8.9103e-01, ..., -5.0128e+00,
-2.2757e+00, 3.1980e-01]]],
[[[ 1.7811e-01, 2.2357e-02, -2.9412e+00, ..., 1.2358e+00,
-6.0968e-01, -9.6782e-01],
[ 1.1792e+00, 1.8931e-01, -3.9205e+00, ..., -6.4445e-01,
-6.9058e-01, -1.6072e+00],
[ 1.2735e+00, -3.0346e-01, -5.9530e-01, ..., -1.4232e+00,
-6.7234e-01, -2.1896e+00],
...,
[-2.1448e+00, -4.0698e+00, 1.8839e+00, ..., -2.7171e+00,
1.1136e+00, 1.5384e+00],
[-1.2356e+00, -8.0311e-01, 6.2565e-01, ..., -3.1027e+00,
1.1412e+00, 1.6767e+00],
[-1.4720e+00, -3.2789e+00, -1.6744e+00, ..., 2.0553e-01,
1.2723e+00, -5.9277e-01]],
[[-9.2785e-01, -1.1882e+00, -3.9803e+00, ..., -8.3498e-01,
-5.1087e+00, -2.8109e+00],
[-2.2813e+00, -3.3578e+00, -5.7996e+00, ..., -6.5609e+00,
-6.8588e+00, -4.9935e+00],
[-2.6785e+00, -3.6427e+00, -4.3467e+00, ..., -5.7470e+00,
-5.4406e+00, -4.0401e+00],
...,
[-6.5704e+00, -8.9120e+00, -4.2010e+00, ..., -1.8104e+00,
-1.1407e+00, -7.9957e-01],
[-7.8595e+00, -9.4263e+00, -7.5853e+00, ..., -4.3833e+00,
-2.5376e+00, -1.6511e+00],
[-2.1988e+00, -5.0668e+00, -1.7664e+00, ..., 2.6591e-01,
1.4449e-01, -3.9479e-01]],
[[ 9.6502e-01, 1.0648e+00, 1.4856e+00, ..., 1.8722e+00,
2.2485e+00, 8.1261e-01],
[ 8.2985e-01, 1.4308e+00, 1.3558e+00, ..., 1.5775e+00,
2.5806e+00, 8.3368e-01],
[ 2.1194e-01, 2.2900e+00, 2.3405e+00, ..., 1.3818e+00,
2.0181e+00, 1.7940e+00],
...,
[-4.9117e-02, -7.4418e-01, -5.0255e+00, ..., -8.6463e+00,
-5.1190e-01, 6.6875e-01],
[-2.0602e+00, -2.0290e+00, -1.6316e+00, ..., -7.4765e+00,
-2.1460e+00, 1.9979e-01],
[ 1.0578e-01, -1.3565e+00, -1.9427e+00, ..., -1.7889e+00,
-2.3943e-01, -2.5619e-01]],
...,
[[-7.9422e-01, -1.5047e+00, -3.0196e+00, ..., -2.4040e+00,
-3.0822e+00, -1.1808e+00],
[-1.9904e+00, -3.6409e+00, -4.1955e+00, ..., -5.0554e+00,
-5.1871e+00, -3.8389e+00],
[-1.0021e+00, -1.3014e+00, -1.5724e+00, ..., -1.9117e+00,
-3.9431e+00, -3.5217e+00],
...,
[-2.2259e+00, -2.0377e+00, -2.1324e+00, ..., 4.2887e-01,
-1.9934e+00, -1.4261e+00],
[-1.5710e+00, -3.1088e+00, -3.6845e+00, ..., -1.0201e+00,
-3.3414e+00, -1.8696e+00],
[-1.9461e+00, -1.7139e+00, -3.0298e-01, ..., -1.6218e-02,
-2.1426e+00, -8.8432e-01]],
[[-1.8690e+00, -3.2941e+00, -2.6718e+00, ..., -3.3586e+00,
-2.9547e+00, -3.0508e-01],
[-4.6717e+00, -6.7948e+00, -4.3905e+00, ..., -5.7148e+00,
-5.7273e+00, -1.6048e+00],
[-4.1335e+00, -5.5734e+00, -4.7441e+00, ..., -2.1013e+00,
-3.7418e+00, -1.5664e+00],
...,
[-3.8829e+00, 7.0900e-01, 1.8739e+00, ..., 3.0638e+00,
-1.1327e+00, 4.1260e-01],
[-2.7945e+00, 6.6055e-01, -9.9385e-01, ..., 1.9588e+00,
-1.0537e+00, 8.8780e-01],
[-3.5055e+00, -9.5029e-01, -7.7838e-01, ..., 1.7426e+00,
-8.0005e-01, -5.5559e-01]],
[[ 8.7495e-01, 2.8056e-01, 5.1254e-01, ..., 1.1537e+00,
1.5003e+00, 6.5423e-01],
[ 1.7032e+00, 6.1732e-01, -5.9071e-01, ..., 7.0148e-01,
1.6222e+00, 3.3834e+00],
[ 1.7090e+00, 1.1082e-01, 1.4734e-01, ..., 9.5450e-01,
1.4372e+00, 1.1275e+00],
...,
[ 2.1142e+00, 3.3564e+00, 3.7769e+00, ..., -2.3159e+00,
1.7201e+00, 2.5373e+00],
[ 4.1539e+00, 2.5077e+00, 4.4464e+00, ..., -3.2709e+00,
5.9950e-01, 2.0764e+00],
[ 2.7775e+00, 1.4981e+00, 3.2323e-01, ..., -2.2749e-01,
2.3557e+00, 1.2647e+00]]],
[[[ 5.8187e-01, 1.4416e+00, -3.7396e+00, ..., 3.0907e+00,
1.4441e+00, 4.4982e-01],
[ 1.0228e+00, 3.7540e-01, -4.0254e+00, ..., 2.7469e+00,
1.3075e+00, -8.0665e-01],
[ 5.0484e-01, -3.4440e-01, -3.7013e+00, ..., 2.3783e+00,
-7.0920e-01, -2.3883e+00],
...,
[-1.1657e+00, -1.0164e+00, 3.6761e-01, ..., -3.3181e+00,
-3.6522e+00, -9.0039e-01],
[-2.5505e+00, -1.7245e+00, 1.0125e+00, ..., -2.3438e+00,
-2.8180e+00, -5.3934e-03],
[-2.4185e+00, -4.2121e+00, -3.1412e+00, ..., -3.9358e+00,
-3.3761e+00, -1.5365e+00]],
[[-1.4526e+00, -2.1747e+00, -8.4048e+00, ..., 2.9010e+00,
-2.8174e+00, -1.5220e+00],
[-5.0608e+00, -5.9160e+00, -7.4635e+00, ..., 1.1738e+00,
-5.6878e+00, -4.5876e+00],
[-4.5844e+00, -4.8603e+00, -7.1934e+00, ..., -4.2666e+00,
-7.8798e+00, -5.8251e+00],
...,
[-7.5323e+00, -9.6312e+00, -6.7994e+00, ..., -7.7948e+00,
-9.4738e+00, -2.3634e+00],
[-9.8657e+00, -1.0233e+01, -8.4945e+00, ..., -1.0103e+01,
-1.1071e+01, -3.8874e+00],
[-2.6090e+00, -5.0629e+00, -4.2941e+00, ..., -4.3730e+00,
-4.1908e+00, -6.6301e-01]],
[[ 6.2273e-01, 1.2219e+00, 2.3613e+00, ..., 2.2784e+00,
2.7857e+00, 2.6577e-01],
[-1.6264e-01, -1.3940e+00, 5.6298e-01, ..., -1.9008e+00,
1.1009e+00, -5.6354e-01],
[ 7.6000e-01, 1.8961e+00, 5.0033e-01, ..., 8.5945e-01,
1.6006e+00, 5.3763e-01],
...,
[ 2.4774e-01, -6.7405e-02, 4.2898e-02, ..., 4.7345e-01,
8.2460e-01, -9.7462e-01],
[-4.4654e-01, 2.8444e-01, -2.0966e+00, ..., 4.1965e-01,
1.0629e+00, -5.7237e-01],
[ 5.2298e-01, 1.3127e+00, 1.0954e+00, ..., -2.2338e-01,
-5.7487e-01, -1.8729e+00]],
...,
[[-5.7438e-01, -1.9642e+00, -2.6831e+00, ..., -1.4618e-01,
-2.2222e+00, 9.4519e-02],
[-2.9251e+00, -3.9189e+00, -2.5625e+00, ..., -4.6962e+00,
-7.1144e+00, -2.5516e+00],
[-1.7444e+00, -2.4396e+00, -2.1955e+00, ..., -4.1982e+00,
-6.1415e+00, -2.7440e+00],
...,
[-3.0263e+00, -3.0041e+00, -2.9080e+00, ..., -4.3660e+00,
-6.4940e+00, -3.0135e+00],
[-3.6907e+00, -5.4837e+00, -2.0794e+00, ..., -4.7659e+00,
-5.4755e+00, -1.4364e+00],
[-3.3232e+00, -3.9070e+00, -3.4688e+00, ..., -2.8957e+00,
-2.6513e+00, 4.2957e-01]],
[[-1.2828e+00, -2.4938e+00, -1.6475e+00, ..., -1.1015e+00,
-1.7258e-01, 1.0059e+00],
[-2.9015e+00, -2.1380e+00, -1.0426e+00, ..., -1.4278e+00,
-2.2601e+00, 9.8194e-01],
[-3.0334e+00, -1.8399e+00, -1.6007e+00, ..., -3.7617e-01,
-5.3526e+00, -6.3188e-03],
...,
[-3.8013e+00, -2.0219e+00, -5.1516e+00, ..., -3.5619e+00,
-2.4796e+00, -9.5262e-01],
[-3.4780e+00, -1.7972e-01, -1.0874e+00, ..., -1.5662e+00,
-9.6866e-01, 2.3405e-01],
[-3.8181e+00, -2.9876e+00, -2.5415e+00, ..., -1.3749e+00,
-1.2542e+00, 3.1660e-01]],
[[ 1.5548e+00, 1.4826e+00, 1.5731e+00, ..., 3.1475e+00,
2.8983e+00, 6.4880e-01],
[ 3.0830e+00, 2.6835e+00, 2.3746e+00, ..., 2.8972e+00,
3.5623e+00, 3.7737e+00],
[ 2.4715e+00, 1.0463e+00, 1.3856e+00, ..., 5.7527e+00,
3.6813e+00, 2.3974e+00],
...,
[ 2.9195e+00, 3.1582e+00, 2.7883e+00, ..., 3.3841e+00,
1.7581e+00, 2.1945e-01],
[ 4.6002e+00, 4.6841e+00, 3.8102e+00, ..., 2.4360e+00,
1.6936e+00, 1.5967e+00],
[ 2.8346e+00, 2.6958e+00, 2.3815e+00, ..., 4.7604e-01,
6.0735e-01, 1.0579e+00]]],
[[[ 1.2582e+00, 2.0248e+00, 1.1493e+00, ..., -3.8678e+00,
-2.8969e+00, -1.1439e+00],
[ 1.6047e+00, 1.0934e+00, 1.4233e+00, ..., -3.6237e+00,
-3.4722e+00, -3.5300e-01],
[-3.3686e-01, -2.8595e+00, -3.5593e+00, ..., -4.3855e+00,
-3.6613e+00, -3.4525e-01],
...,
[ 7.4093e-01, -8.3832e-01, -9.0372e-01, ..., -3.2955e+00,
-3.9303e+00, 4.2439e-01],
[ 1.9027e-01, -3.1818e-01, -7.0428e-01, ..., -4.3143e+00,
-3.9271e+00, 4.9657e-01],
[-3.5780e-01, -3.4948e+00, -3.2534e+00, ..., -3.9988e+00,
-3.3996e+00, -1.4901e+00]],
[[ 4.1765e-01, -1.6907e+00, -3.0390e+00, ..., -8.8471e+00,
-7.5403e+00, -1.4071e+00],
[-5.1566e-01, -1.1731e+00, -3.8625e+00, ..., -1.2261e+01,
-9.7377e+00, -2.5033e+00],
[-3.2000e+00, -5.0666e+00, -4.1937e+00, ..., -7.6991e+00,
-9.4751e+00, -2.5271e+00],
...,
[-2.2512e+00, -3.4988e+00, -3.2267e+00, ..., -8.3184e+00,
-9.5376e+00, -2.1701e+00],
[-4.0088e+00, -5.5662e+00, -4.6873e+00, ..., -9.7438e+00,
-1.1173e+01, -4.4421e+00],
[-2.3820e-01, -3.1487e+00, -3.3422e+00, ..., -4.1122e+00,
-3.8774e+00, -7.7990e-01]],
[[ 6.9037e-01, 2.5088e+00, 1.7169e+00, ..., 5.7498e-01,
2.0496e+00, 9.4752e-01],
[-2.1618e-01, 1.0481e+00, 3.9679e-01, ..., 1.1200e+00,
2.8012e+00, -1.4706e-01],
[ 1.2787e+00, 5.1103e+00, 5.0448e+00, ..., 2.3691e+00,
2.0388e+00, -5.3412e-01],
...,
[ 8.9632e-01, 2.6427e+00, 1.2823e+00, ..., 2.4971e-01,
1.5353e+00, -1.1670e+00],
[ 6.8361e-01, 1.9835e+00, 7.7367e-01, ..., -4.0808e-01,
1.5486e+00, -1.4125e+00],
[-2.9563e-01, 1.6372e+00, 7.8887e-01, ..., 1.3900e-01,
-2.4584e-01, -1.9401e+00]],
...,
[[-4.4289e-01, -7.4781e-01, -1.1826e+00, ..., -5.8470e+00,
-5.6385e+00, -5.8642e-01],
[-2.9994e+00, -3.2796e+00, -2.2632e+00, ..., -5.3023e+00,
-6.7866e+00, -1.7617e+00],
[-3.1967e+00, -5.2343e+00, -5.4375e+00, ..., -5.8149e+00,
-5.9980e+00, -2.0407e+00],
...,
[-1.0021e+00, -2.6445e+00, -1.3518e+00, ..., -3.9730e+00,
-6.3834e+00, -1.6538e+00],
[-1.6296e+00, -3.3171e+00, -1.7153e+00, ..., -3.1943e+00,
-5.5099e+00, -8.8700e-01],
[-1.0896e+00, -3.4638e+00, -2.2753e+00, ..., -2.0343e+00,
-2.3222e+00, 3.5276e-01]],
[[-1.0326e+00, -8.3413e-01, -1.2744e+00, ..., -2.4536e+00,
-9.4945e-01, -7.1748e-01],
[-2.7340e+00, -2.0689e+00, -1.0708e-01, ..., -3.8380e+00,
-2.9541e+00, -9.5222e-01],
[-4.5536e+00, -8.8823e+00, -2.7724e+00, ..., -2.4792e-01,
-1.5422e+00, -7.1412e-01],
...,
[-2.9646e+00, -4.3604e+00, -2.4135e+00, ..., -3.0954e+00,
-1.7308e+00, -9.3792e-01],
[-3.0569e+00, -2.1116e+00, -7.7293e-01, ..., -1.9610e+00,
-7.3472e-01, 1.0532e+00],
[-3.2319e+00, -2.9737e+00, -9.7666e-01, ..., -1.1417e-01,
-8.0600e-01, 6.1412e-01]],
[[ 1.1462e+00, 1.2919e+00, 1.3146e+00, ..., -7.5823e-02,
5.4068e-01, -1.3515e+00],
[ 1.4667e+00, 2.0613e+00, 1.9554e+00, ..., 3.5397e+00,
2.4687e+00, 2.2738e+00],
[ 2.8228e-01, -5.6325e-01, 2.0219e+00, ..., 2.7904e+00,
1.4214e+00, 6.6438e-01],
...,
[ 1.7373e+00, 9.2437e-01, 1.4541e+00, ..., 2.7386e+00,
1.9548e+00, 5.5527e-01],
[ 2.9960e+00, 3.1264e+00, 1.9678e+00, ..., 2.3320e+00,
1.0706e+00, 1.9362e+00],
[ 1.9362e+00, 3.4212e-01, 4.5819e-01, ..., 2.0047e-01,
-7.6091e-02, 9.0884e-01]]]], device='cuda:0'),
tensor([[[[-2.5366e-01, 1.2192e+00, 1.2435e+00, ..., -1.3793e+00,
-3.7600e+00, -8.7164e-01],
[ 1.2083e-01, 3.1586e-01, 2.9864e+00, ..., -8.0510e-03,
-2.3494e+00, -2.2535e-01],
[-1.1409e+00, -4.2906e-03, 1.9957e+00, ..., -1.7820e-01,
-2.6554e+00, -4.6732e-01],
...,
[-4.3134e-01, -1.4323e-01, 1.5061e+00, ..., -1.7204e+00,
-2.2136e+00, 2.5149e-01],
[-1.5778e+00, 1.5828e-02, 1.1536e+00, ..., -2.2557e+00,
-3.2434e+00, -3.6890e-01],
[-1.1764e+00, -2.1743e+00, -1.1730e+00, ..., -3.7046e+00,
-3.7459e+00, -1.6057e+00]],
[[-1.5566e+00, -2.2139e+00, -3.3358e+00, ..., -5.6287e+00,
-8.9960e+00, -1.5182e+00],
[-4.7807e+00, -4.8530e+00, -4.5068e+00, ..., -5.4574e+00,
-8.3947e+00, -2.7929e+00],
[-4.0046e+00, -4.8637e+00, -6.6476e+00, ..., -4.9977e+00,
-7.9924e+00, -2.5187e+00],
...,
[-4.4644e+00, -5.0733e+00, -5.4725e+00, ..., -6.3539e+00,
-8.5100e+00, -2.6125e+00],
[-7.7400e+00, -7.8478e+00, -7.1917e+00, ..., -8.6469e+00,
-1.0104e+01, -3.8990e+00],
[-1.3430e+00, -3.0017e+00, -2.2485e+00, ..., -3.9651e+00,
-4.1239e+00, -5.3958e-01]],
[[-2.1233e-01, 2.2331e+00, 1.4363e+00, ..., 2.0590e+00,
2.4986e+00, 7.8847e-01],
[-5.4214e-01, 2.1271e-01, 4.6468e-01, ..., 1.1600e+00,
1.9930e+00, -1.6267e-01],
[-7.8513e-02, 2.0163e+00, 9.4918e-01, ..., -1.0968e-02,
8.0180e-01, -5.2684e-01],
...,
[-6.6352e-01, 7.4449e-01, 1.6740e+00, ..., 1.1118e+00,
2.6431e-01, -1.8380e-01],
[-1.1095e+00, 8.7020e-02, 1.2047e+00, ..., -3.6175e-01,
6.4226e-01, -3.8087e-01],
[-3.2309e-01, -4.4027e-01, 1.9014e-01, ..., 3.3068e-01,
5.4313e-02, -1.9838e+00]],
...,
[[-1.3439e+00, -1.9757e+00, -5.8950e-01, ..., -3.7750e+00,
-5.4466e+00, -1.2691e+00],
[-3.4936e+00, -3.5793e+00, -1.3833e+00, ..., -2.8371e+00,
-5.2476e+00, -2.3763e+00],
[-2.6225e+00, -2.4620e+00, -8.9995e-01, ..., -3.2151e+00,
-6.5113e+00, -1.8795e+00],
...,
[-1.0886e+00, -2.2982e+00, -5.0951e-01, ..., -2.3967e+00,
-5.2872e+00, -2.4107e+00],
[-2.1547e+00, -3.6559e+00, -6.4465e-01, ..., -1.8246e+00,
-4.2690e+00, -1.2001e-01],
[-1.2498e+00, -1.2444e+00, 1.9130e-01, ..., -1.4446e+00,
-2.2701e+00, 9.4119e-01]],
[[-1.7645e+00, -1.6641e+00, -1.9464e+00, ..., -2.1504e+00,
-9.1465e-01, 3.4156e-01],
[-3.9542e+00, -1.9061e+00, -2.2094e+00, ..., -2.2308e+00,
-2.3183e+00, 8.1551e-02],
[-3.7568e+00, -2.9000e+00, -3.6833e+00, ..., -1.2666e+00,
-2.3406e+00, 3.6599e-01],
...,
[-2.7617e+00, -2.7229e+00, -3.4678e+00, ..., -1.7933e+00,
-1.8762e+00, -5.9073e-02],
[-2.0474e+00, -8.6337e-02, -1.4849e+00, ..., -2.1130e-01,
-1.5249e+00, 8.7185e-01],
[-2.0760e+00, -1.2627e+00, -5.3192e-01, ..., -4.8235e-01,
-1.0279e+00, 6.4559e-01]],
[[ 1.2003e+00, 3.3152e-01, 7.6505e-01, ..., 1.7529e+00,
1.4749e+00, 5.8087e-02],
[ 4.0340e+00, 3.0444e+00, 2.8467e+00, ..., 3.4500e+00,
2.8931e+00, 3.8880e+00],
[ 1.6902e+00, 6.4104e-01, 1.2854e+00, ..., 2.6509e+00,
1.6663e+00, 2.0607e+00],
...,
[ 2.9920e+00, 2.5752e+00, 2.7009e+00, ..., 2.5815e+00,
2.9612e+00, 1.4810e+00],
[ 3.7707e+00, 4.0284e+00, 3.0596e+00, ..., 2.6479e+00,
9.5609e-01, 2.5807e+00],
[ 2.3434e+00, 2.3595e+00, 1.6231e+00, ..., 3.4427e-01,
-3.0235e-01, 1.3201e+00]]],
[[[-1.1723e-01, 7.4040e-01, -3.2883e-01, ..., -1.3341e+00,
-8.3060e-01, -8.2009e-01],
[ 6.7112e-01, 3.0175e-01, 9.7651e-02, ..., -1.1104e-01,
5.5776e-01, -1.0684e+00],
[ 9.2353e-01, -4.3480e-01, 4.6041e-01, ..., -6.7261e-01,
-4.7753e-01, -2.0476e+00],
...,
[-2.9099e-01, -5.3420e+00, -8.9187e-01, ..., -1.5846e+00,
-2.2541e+00, 1.3372e-01],
[-1.7696e+00, -2.3031e+00, 1.3489e-01, ..., -2.3587e+00,
-2.6827e+00, 2.0299e-01],
[-2.1278e+00, -3.9187e+00, -3.6017e+00, ..., -4.0810e+00,
-3.4828e+00, -1.4410e+00]],
[[-7.1107e-01, -1.4467e+00, -2.5833e+00, ..., -2.7513e+00,
-4.4085e+00, -2.4336e+00],
[-2.0456e+00, -3.5619e+00, -3.8235e+00, ..., -2.6814e+00,
-4.7248e+00, -4.6916e+00],
[-1.8106e+00, -3.3999e+00, -3.5763e+00, ..., -3.2186e+00,
-4.0050e+00, -3.6633e+00],
...,
[-6.1024e+00, -9.5696e+00, -6.5456e+00, ..., -7.7821e+00,
-8.4691e+00, -1.7768e+00],
[-9.1552e+00, -9.9550e+00, -1.0045e+01, ..., -1.0487e+01,
-1.0669e+01, -3.7278e+00],
[-2.6180e+00, -5.1941e+00, -3.9605e+00, ..., -4.1420e+00,
-4.1584e+00, -5.8629e-01]],
[[ 7.0913e-01, 1.2719e+00, 1.1469e+00, ..., 1.7197e+00,
1.9312e+00, 8.4998e-01],
[ 7.4069e-01, 1.3567e+00, 1.6568e+00, ..., 1.3509e+00,
2.3478e+00, 9.3997e-01],
[ 5.3599e-01, 2.9737e+00, 2.3688e+00, ..., 1.0114e+00,
2.3830e+00, 1.9602e+00],
...,
[-7.2572e-02, 1.4792e-01, -5.4207e+00, ..., 1.1638e+00,
1.2603e+00, -1.9999e-01],
[-1.0079e+00, 4.2822e-01, -1.5797e+00, ..., 5.4908e-01,
1.1075e+00, -3.4435e-01],
[ 3.9355e-01, 4.1654e-01, 5.8625e-01, ..., -1.8729e-01,
-5.1627e-01, -1.9322e+00]],
...,
[[-9.7567e-01, -1.3716e+00, -2.1381e+00, ..., -2.5307e+00,
-3.2130e+00, -1.1409e+00],
[-2.3758e+00, -3.1805e+00, -2.4999e+00, ..., -2.6357e+00,
-4.1450e+00, -3.3574e+00],
[-1.0274e+00, -1.6620e+00, -4.3491e-01, ..., -2.6809e+00,
-4.3673e+00, -2.9875e+00],
...,
[-2.7073e+00, -5.3869e+00, -1.3460e+00, ..., -5.2757e+00,
-6.1956e+00, -2.5284e+00],
[-3.3556e+00, -6.7071e+00, -4.5368e+00, ..., -4.2991e+00,
-5.1577e+00, -1.1886e+00],
[-3.2838e+00, -3.5798e+00, -2.5749e+00, ..., -2.6990e+00,
-2.5323e+00, 4.9614e-01]],
[[-1.6187e+00, -2.4759e+00, -3.6872e+00, ..., -2.7719e+00,
-2.7540e+00, -6.1042e-01],
[-4.4274e+00, -4.8169e+00, -4.8734e+00, ..., -3.8787e+00,
-5.9086e+00, -2.0347e+00],
[-4.0884e+00, -4.8822e+00, -4.1083e+00, ..., -2.3833e+00,
-4.9047e+00, -2.0724e+00],
...,
[-5.2912e+00, -2.8059e+00, 1.2052e+00, ..., -3.5736e+00,
-2.5641e+00, -1.1933e+00],
[-3.4974e+00, -1.4699e+00, 4.2580e-01, ..., -2.0777e+00,
-1.1928e+00, 3.1364e-01],
[-3.8904e+00, -3.0511e+00, -1.8727e+00, ..., -1.5930e+00,
-1.1426e+00, 3.6621e-01]],
[[ 8.7753e-01, -1.6434e-01, 7.6858e-01, ..., 3.8649e-01,
1.4688e+00, 6.2159e-01],
[ 1.7932e+00, 8.4098e-01, 5.0106e-01, ..., 6.4372e-01,
1.9716e+00, 3.2037e+00],
[ 1.3054e+00, 3.7466e-02, 3.8257e-01, ..., 3.2364e-01,
7.7439e-01, 7.8357e-01],
...,
[ 3.6900e+00, 1.9011e+00, 4.9309e+00, ..., 3.0361e+00,
2.5765e+00, 5.2297e-01],
[ 4.6971e+00, 3.1930e+00, 4.4238e+00, ..., 2.5356e+00,
1.9416e+00, 1.4184e+00],
[ 3.1816e+00, 2.6876e+00, 6.9189e-01, ..., 4.5937e-01,
5.4056e-01, 1.0048e+00]]],
[[[ 3.0282e-01, 1.3820e+00, 2.7112e-01, ..., -4.6982e+00,
-2.8156e+00, -6.0094e-01],
[ 6.8759e-01, -2.7439e-01, -1.1167e+00, ..., -2.7981e+00,
-3.5893e+00, -1.0638e+00],
[-2.3782e+00, -4.2704e+00, -4.0668e+00, ..., -1.8917e+00,
-5.2360e+00, -2.7993e+00],
...,
[-2.0846e+00, -5.0593e+00, 5.0940e+00, ..., 9.0299e+00,
-4.0698e-01, 1.9721e-01],
[-7.1360e-01, 3.9323e-01, 4.7237e+00, ..., -3.0261e+00,
-4.9041e+00, 1.2233e+00],
[ 1.3027e-01, 1.2672e-01, -1.4082e+00, ..., -4.3489e+00,
-3.4210e+00, -1.7495e+00]],
[[-1.7975e+00, -2.7239e+00, -5.2174e+00, ..., -1.1409e+01,
-7.5112e+00, -9.2865e-01],
[-5.2689e+00, -5.5870e+00, -4.9130e+00, ..., -1.0056e+01,
-1.2967e+01, -2.9104e+00],
[-5.8403e+00, -6.2314e+00, -4.8368e+00, ..., -8.0255e+00,
-1.4312e+01, -5.0711e+00],
...,
[-7.7391e-01, -1.4509e+00, 4.0866e+00, ..., 3.3872e-01,
-5.7481e+00, -6.1401e+00],
[ 4.6253e-01, -2.1700e+00, -4.6407e+00, ..., -1.4757e+01,
-1.1259e+01, -2.1877e+00],
[-3.9901e-01, -1.5719e+00, -3.1045e+00, ..., -2.0128e+00,
-2.9233e+00, -2.4474e-01]],
[[ 2.3951e-01, 1.5304e+00, 1.8024e+00, ..., 1.2030e+00,
2.3797e+00, 7.8874e-01],
[-2.5230e-01, -3.9560e-01, 8.5442e-01, ..., 3.6839e+00,
1.6586e+00, -5.9440e-01],
[ 5.2004e-01, 2.4818e+00, 3.0728e+00, ..., 8.9511e+00,
3.2041e+00, -3.8952e-01],
...,
[-4.7763e-01, -2.5943e+00, -6.6490e+00, ..., -1.4580e+01,
3.4789e+00, -7.0389e-01],
[ 4.9470e-01, 7.9182e-01, -4.6132e+00, ..., -9.3635e+00,
2.7237e-01, -2.6748e+00],
[-7.1634e-01, -5.4502e-01, -2.0480e+00, ..., 1.8605e-01,
-2.4539e+00, -3.2914e+00]],
...,
[[-8.8594e-01, -2.2099e+00, -2.5891e+00, ..., -5.8659e+00,
-5.9215e+00, -3.9874e-01],
[-3.4358e+00, -3.1573e+00, -1.3602e+00, ..., -4.9468e+00,
-6.0410e+00, -4.0473e+00],
[-4.1058e+00, -2.3716e+00, -9.3463e-01, ..., -1.0225e+01,
-6.0977e+00, -5.6136e+00],
...,
[-6.0568e-01, 1.2287e+00, 1.4118e+00, ..., 2.3833e+00,
-1.2568e+01, -6.9673e-01],
[ 4.5265e-01, -1.6323e+00, -4.7560e+00, ..., -1.9778e+00,
-9.3549e+00, 1.0959e+00],
[ 4.9089e-01, -7.0553e-01, -1.8850e+00, ..., -7.1612e-01,
-3.2899e+00, -1.0376e+00]],
[[-1.9009e+00, -2.1169e+00, -3.5667e+00, ..., -1.1531e+00,
-9.9436e-01, -5.1927e-01],
[-4.0216e+00, -3.2809e+00, -4.3629e+00, ..., -1.1053e+00,
-1.4352e+00, -2.0985e-02],
[-5.5343e+00, -4.1893e+00, -3.1486e+00, ..., -6.2487e+00,
-7.1668e-01, -7.5703e-02],
...,
[-2.4551e+00, -1.4386e+00, 2.9513e-01, ..., 2.0305e+00,
-8.0284e+00, 9.0875e-01],
[-1.4968e+00, 1.6335e-01, -3.1567e+00, ..., -1.8687e+00,
-4.0393e+00, 1.9821e+00],
[-2.6050e-01, 1.2773e-01, -1.8417e+00, ..., 2.0676e+00,
1.3738e-01, 8.7546e-01]],
[[ 1.5055e+00, 3.7892e-01, 1.6239e+00, ..., 1.5695e+00,
2.8049e-01, -1.4951e+00],
[ 3.2501e+00, 1.9282e+00, 9.6573e-01, ..., 4.9435e+00,
3.2966e+00, 8.3995e-01],
[ 1.6959e+00, -4.9863e-01, -4.3031e-01, ..., 7.6404e+00,
3.1002e+00, -2.6390e-01],
...,
[-3.3321e+00, -5.9613e+00, 5.2052e-01, ..., 8.3762e+00,
2.9683e+00, 1.7127e+00],
[-2.8429e+00, -2.0505e+00, 2.8069e+00, ..., 8.8794e+00,
5.6092e-01, 1.9200e+00],
[ 2.7354e-01, 4.7992e-01, 1.4692e+00, ..., -4.1527e-01,
1.1276e+00, 1.3714e+00]]],
[[[ 2.9144e-01, 1.5590e+00, -7.8407e-01, ..., -6.4785e-01,
-8.8358e-01, -9.2434e-01],
[ 1.1212e+00, 1.3467e+00, -1.4770e+00, ..., 2.0003e+00,
3.8926e-01, -1.7939e+00],
[ 1.4967e+00, 9.1156e-01, -4.0842e+00, ..., 2.2997e+00,
-5.3802e-01, -3.0488e+00],
...,
[-2.0190e-01, -1.1439e+01, -2.1878e+00, ..., -1.6175e+00,
-2.5200e+00, -3.0912e-02],
[-2.8122e+00, -1.0251e+01, -1.0952e+00, ..., -2.1860e+00,
-2.6552e+00, 2.0522e-01],
[-2.2738e+00, -7.2274e+00, -2.0265e+00, ..., -4.0287e+00,
-3.5405e+00, -1.6182e+00]],
[[-6.3534e-01, -1.5507e+00, -4.0149e+00, ..., -2.4898e+00,
-5.1534e+00, -2.5602e+00],
[-2.1049e+00, -3.9240e+00, -5.7864e+00, ..., -1.9961e+00,
-7.2679e+00, -4.9007e+00],
[-2.2847e+00, -4.3579e+00, -8.0225e+00, ..., -2.4374e+00,
-8.0467e+00, -4.2137e+00],
...,
[-4.4807e+00, -1.2171e+01, -8.9610e-01, ..., -7.8199e+00,
-8.6578e+00, -1.8898e+00],
[-9.4168e+00, -1.3844e+01, -3.2866e+00, ..., -1.0342e+01,
-1.0662e+01, -3.7884e+00],
[-1.7064e+00, -5.6871e+00, -4.0847e+00, ..., -4.2211e+00,
-4.2591e+00, -6.9539e-01]],
[[ 6.7787e-01, 1.5224e+00, 1.7575e+00, ..., 2.3458e+00,
1.8872e+00, 7.9953e-01],
[ 7.1517e-01, 8.3542e-01, 1.1665e+00, ..., 2.9078e+00,
2.1922e+00, 5.4704e-01],
[ 5.1499e-01, 2.5190e+00, 2.4309e+00, ..., 4.0550e+00,
2.9529e+00, 1.5731e+00],
...,
[-2.4999e+00, 1.9819e+00, -5.0202e+00, ..., 1.4113e+00,
1.2527e+00, -2.3434e-01],
[-2.6352e+00, 6.2476e-01, -6.5349e+00, ..., 5.8018e-01,
1.0291e+00, -3.8790e-01],
[-1.1549e+00, 1.8420e+00, -3.7852e+00, ..., -1.2524e-01,
-4.5115e-01, -1.8723e+00]],
...,
[[-5.0636e-01, -1.2922e+00, -3.1676e+00, ..., -2.3434e+00,
-3.5951e+00, -1.4280e+00],
[-1.7637e+00, -2.7102e+00, -3.7875e+00, ..., -3.7466e+00,
-4.3241e+00, -4.6936e+00],
[-5.9717e-01, -1.6623e+00, -3.4141e+00, ..., -5.4914e+00,
-3.2831e+00, -4.7274e+00],
...,
[-2.9692e+00, -8.0208e+00, 4.7852e-01, ..., -5.3859e+00,
-6.1422e+00, -2.5923e+00],
[-3.4751e+00, -7.6609e+00, 8.2613e-02, ..., -4.3283e+00,
-5.1596e+00, -1.2739e+00],
[-3.2419e+00, -5.2061e+00, -2.6025e+00, ..., -2.8813e+00,
-2.6770e+00, 3.7961e-01]],
[[-1.3191e+00, -3.5118e+00, -6.0203e+00, ..., -1.8995e+00,
-2.7368e+00, -6.2529e-01],
[-3.9724e+00, -5.6590e+00, -5.7486e+00, ..., -3.4085e+00,
-5.2401e+00, -1.9970e+00],
[-4.2463e+00, -6.5201e+00, -5.1478e+00, ..., -4.7860e+00,
-4.9778e+00, -2.2745e+00],
...,
[-6.1538e+00, -3.9477e+00, -2.6666e+00, ..., -3.8877e+00,
-2.9360e+00, -1.2332e+00],
[-3.9194e+00, -2.8693e-01, 3.4483e+00, ..., -2.1340e+00,
-1.2669e+00, 3.0365e-01],
[-3.0746e+00, -1.8845e+00, 3.0042e+00, ..., -1.6684e+00,
-1.2704e+00, 3.2733e-01]],
[[ 8.4197e-01, 1.6175e-01, 1.7381e+00, ..., 1.3088e+00,
1.6811e+00, 2.1857e-01],
[ 1.5319e+00, 1.1182e+00, 1.0610e+00, ..., 3.2238e+00,
2.0586e+00, 2.4159e+00],
[ 1.6700e+00, 1.5967e+00, 1.1538e+00, ..., 3.0418e+00,
3.0558e-01, 4.2537e-01],
...,
[ 4.0939e+00, 3.2746e-01, -9.0449e-01, ..., 3.1116e+00,
2.7814e+00, 6.5517e-01],
[ 5.3039e+00, 1.0070e+00, 1.1183e+00, ..., 2.4977e+00,
1.8893e+00, 1.4012e+00],
[ 2.9182e+00, 1.1732e+00, 4.0729e+00, ..., 5.1493e-01,
5.4850e-01, 9.8122e-01]]],
[[[-2.2472e-01, 9.4546e-01, 1.5316e-01, ..., -1.4048e+00,
-1.1576e+00, -8.8994e-01],
[ 1.3026e+00, 8.7274e-01, 1.7137e-01, ..., -9.5183e-01,
-1.0572e-01, -1.1920e+00],
[ 1.6945e+00, -2.8692e-01, -4.8971e+00, ..., -1.6689e+00,
-8.5295e-01, -1.9552e+00],
...,
[-5.3412e-01, -3.1845e-01, -2.9838e-01, ..., -4.9328e+00,
-3.5561e+00, 4.6700e-02],
[-2.1893e+00, -1.1607e+00, 1.5957e+00, ..., -3.6645e+00,
-2.8729e+00, 4.6744e-01],
[-2.2479e+00, -4.2763e+00, -3.4671e+00, ..., -4.6836e+00,
-3.1832e+00, -1.4491e+00]],
[[-8.3162e-01, -1.5742e+00, -3.1023e+00, ..., -3.0851e+00,
-4.8519e+00, -2.5576e+00],
[-2.2889e+00, -4.4843e+00, -6.2033e+00, ..., -3.9509e+00,
-5.1104e+00, -4.5123e+00],
[-3.5746e+00, -5.5433e+00, -7.7274e+00, ..., -4.4546e+00,
-4.0075e+00, -3.1946e+00],
...,
[-6.6686e+00, -6.9603e+00, -5.0160e+00, ..., -1.0553e+01,
-8.5969e+00, -1.6216e+00],
[-9.6013e+00, -1.0130e+01, -8.6140e+00, ..., -1.0678e+01,
-1.0267e+01, -4.0609e+00],
[-2.4464e+00, -4.9315e+00, -4.2018e+00, ..., -3.8010e+00,
-4.1199e+00, -7.3964e-01]],
[[ 7.9229e-01, 1.2768e+00, 1.0205e+00, ..., 1.4354e+00,
1.9131e+00, 9.0771e-01],
[ 8.4616e-01, 1.2747e+00, 1.4169e+00, ..., 1.3285e+00,
2.3795e+00, 9.6439e-01],
[ 4.2022e-01, 1.9206e+00, 2.7193e+00, ..., 1.3256e+00,
2.1312e+00, 1.9008e+00],
...,
[-7.4197e-01, -4.1413e-01, 7.6404e-01, ..., 5.3526e-01,
-8.2120e-01, -3.2544e-01],
[-9.2712e-01, -1.8880e-01, -2.5354e+00, ..., -1.4005e+00,
8.8447e-01, -2.0717e-01],
[ 4.3581e-01, 1.3978e+00, 1.2042e+00, ..., -4.8934e-01,
-3.3310e-01, -1.7255e+00]],
...,
[[-9.4396e-01, -1.2349e+00, -2.2706e+00, ..., -2.6962e+00,
-3.4799e+00, -1.2116e+00],
[-1.9541e+00, -3.6893e+00, -3.6099e+00, ..., -3.3886e+00,
-4.4586e+00, -3.3786e+00],
[-1.0316e+00, -3.5742e+00, -2.9864e+00, ..., -3.3174e+00,
-4.5629e+00, -2.6839e+00],
...,
[-1.9581e+00, -2.2919e+00, -1.4016e+00, ..., -6.8370e+00,
-6.5001e+00, -2.9888e+00],
[-3.4174e+00, -5.5497e+00, -1.6918e+00, ..., -5.2477e+00,
-5.7758e+00, -1.9268e+00],
[-3.2434e+00, -4.3116e+00, -3.7792e+00, ..., -3.2486e+00,
-2.6850e+00, 2.7910e-01]],
[[-1.7070e+00, -2.3581e+00, -4.0563e+00, ..., -2.9856e+00,
-2.8455e+00, -5.3879e-01],
[-4.3285e+00, -5.0757e+00, -5.4562e+00, ..., -4.2438e+00,
-5.9092e+00, -1.8237e+00],
[-4.4249e+00, -5.7580e+00, -4.4988e+00, ..., -1.9919e+00,
-4.5685e+00, -1.9721e+00],
...,
[-3.5677e+00, -2.0650e+00, -3.1834e+00, ..., -9.2235e+00,
-3.4095e+00, -7.4499e-01],
[-2.9924e+00, 3.7079e-01, -3.6651e-01, ..., -2.0419e+00,
-8.2072e-01, 5.6101e-01],
[-3.6328e+00, -2.8987e+00, -2.4881e+00, ..., -1.0231e+00,
-7.5838e-01, 3.9308e-01]],
[[ 8.7355e-01, -5.5854e-03, 1.1078e+00, ..., 4.5906e-01,
1.4535e+00, 6.0819e-01],
[ 1.8793e+00, 1.6574e+00, 2.0491e+00, ..., 7.7271e-01,
1.9479e+00, 3.2344e+00],
[ 2.2325e+00, 1.4269e+00, -1.7483e-01, ..., 1.4726e-01,
5.5385e-01, 5.5728e-01],
...,
[ 3.5298e+00, 2.7910e+00, 1.2997e+00, ..., 3.1610e+00,
3.5145e+00, 9.2455e-01],
[ 4.3653e+00, 4.4910e+00, 3.4153e+00, ..., 2.6569e+00,
1.6134e+00, 1.1263e+00],
[ 2.7829e+00, 2.8582e+00, 2.2929e+00, ..., 5.9620e-01,
1.2068e-01, 8.9091e-01]]]], device='cuda:0'),
tensor([[[[-2.9645e-01, 1.0076e+00, 1.1682e+00, ..., -9.8077e-02,
5.4905e-01, -2.2006e-01],
[ 1.2068e-01, 3.5557e-01, 2.5778e+00, ..., 1.2401e-01,
8.2215e-01, -5.5480e-01],
[-1.2613e+00, 1.5803e-01, 1.3482e+00, ..., -3.9350e-01,
6.5194e-02, -9.9392e-01],
...,
[-3.4079e-01, 1.4342e-01, 4.9431e-01, ..., 1.0007e-01,
5.6029e-01, -3.4288e-01],
[-1.4348e+00, 1.0061e-01, -6.9107e-02, ..., 4.1760e-01,
4.0325e-01, -1.0994e+00],
[-1.2513e+00, -2.2845e+00, -1.9459e+00, ..., -3.0738e+00,
-2.8457e+00, -2.7137e+00]],
[[-1.6355e+00, -1.9777e+00, -3.0880e+00, ..., -2.1060e+00,
-3.4477e+00, -2.3149e+00],
[-4.8697e+00, -4.7133e+00, -3.6680e+00, ..., -3.3172e+00,
-3.8374e+00, -3.5731e+00],
[-4.0166e+00, -4.8350e+00, -4.7092e+00, ..., -3.5879e+00,
-2.8700e+00, -2.9606e+00],
...,
[-4.3568e+00, -4.9884e+00, -5.4427e+00, ..., -2.2218e+00,
-3.2591e+00, -2.5564e+00],
[-7.4025e+00, -7.5144e+00, -6.2923e+00, ..., -4.7140e+00,
-5.1832e+00, -4.9350e+00],
[-1.4037e+00, -3.1601e+00, -2.1027e+00, ..., -2.9375e+00,
-2.7918e+00, -1.5952e+00]],
[[-1.9848e-01, 2.1832e+00, 1.7457e+00, ..., 1.7719e+00,
1.7181e+00, 9.1927e-01],
[-5.0333e-01, -1.1647e-01, 2.1190e-01, ..., 8.3792e-01,
1.3798e+00, 8.7726e-01],
[-9.6767e-02, 1.8882e+00, 4.9972e-01, ..., 1.3819e+00,
1.5448e+00, 1.2980e+00],
...,
[-7.9359e-01, 3.3867e-01, 1.0162e+00, ..., 1.2996e+00,
1.2877e+00, 6.0144e-01],
[-1.3458e+00, 6.3181e-02, 1.1894e+00, ..., 1.3302e+00,
1.0675e+00, 3.2892e-01],
[-4.0566e-01, -3.3529e-01, 2.0434e-01, ..., 1.6684e+00,
1.6017e+00, 4.9189e-01]],
...,
[[-1.3759e+00, -1.6799e+00, -1.5586e+00, ..., -1.8924e+00,
-1.9097e+00, -4.9244e-01],
[-3.5397e+00, -3.2092e+00, -1.8766e+00, ..., -3.3430e+00,
-3.6330e+00, -3.0315e+00],
[-2.7942e+00, -2.1565e+00, -1.8533e+00, ..., -2.4500e+00,
-3.0977e+00, -1.8966e+00],
...,
[-7.1864e-01, -2.3318e+00, -1.0337e+00, ..., -1.5748e+00,
-2.4134e+00, -8.4497e-01],
[-1.8494e+00, -4.0346e+00, -1.7129e+00, ..., -1.7856e+00,
-2.8619e+00, -1.5276e+00],
[-1.3939e+00, -1.6714e+00, -9.3487e-01, ..., -1.8591e+00,
-2.1738e+00, -5.7565e-01]],
[[-1.7838e+00, -1.4426e+00, -2.8889e+00, ..., -7.9065e-01,
-1.5557e+00, -7.0494e-01],
[-3.9935e+00, -1.4063e+00, -2.5440e+00, ..., -1.4532e+00,
-4.1243e+00, -1.9523e+00],
[-3.8275e+00, -2.6941e+00, -4.1134e+00, ..., -5.5444e-01,
-4.6193e+00, -2.5269e+00],
...,
[-2.6433e+00, -2.6329e+00, -3.0415e+00, ..., -1.0656e+00,
-3.1237e+00, -2.2074e+00],
[-1.7862e+00, -2.0246e-01, -1.4602e+00, ..., -5.1378e-01,
-3.2061e+00, -1.9048e+00],
[-2.0946e+00, -1.6422e+00, -1.8358e+00, ..., -1.5063e+00,
-2.7224e+00, -1.2224e+00]],
[[ 1.2578e+00, 3.1197e-01, 6.6236e-01, ..., 1.7456e-01,
1.3348e+00, 6.8691e-01],
[ 4.0281e+00, 2.7706e+00, 2.1476e+00, ..., 3.4168e-01,
1.0062e+00, 2.4775e+00],
[ 1.6656e+00, 6.8855e-01, 7.9106e-01, ..., 3.2575e-01,
1.2552e-03, 7.5052e-01],
...,
[ 2.8564e+00, 2.4744e+00, 2.5456e+00, ..., 1.2629e+00,
4.1318e-01, 6.2659e-01],
[ 3.7553e+00, 3.9729e+00, 2.6798e+00, ..., 9.8313e-01,
7.1965e-01, 1.2457e+00],
[ 2.4071e+00, 2.3391e+00, 1.4882e+00, ..., -1.4824e-01,
-3.4141e-01, -1.3139e-01]]],
[[[ 1.0436e+00, 1.6582e+00, 7.8662e-01, ..., -3.6828e+00,
-2.4198e+00, -8.9083e-01],
[ 2.1180e+00, 9.5043e-01, 6.5385e-01, ..., -2.7435e+00,
-2.3413e+00, -1.3843e-01],
[ 2.0730e-03, -8.8032e-01, -7.9628e-02, ..., -2.2857e+00,
-2.8070e+00, -6.8107e-03],
...,
[ 3.4802e-01, -1.1726e+00, -9.8859e-01, ..., -3.2491e+00,
-3.9045e+00, 4.2916e-01],
[ 4.7242e-01, -4.3551e-02, -4.8722e-01, ..., -4.1035e+00,
-3.8613e+00, 4.9735e-01],
[-3.4990e-01, -3.4068e+00, -3.2048e+00, ..., -3.8604e+00,
-3.4175e+00, -1.3958e+00]],
[[ 3.9750e-01, -1.1887e+00, -2.5727e+00, ..., -7.0353e+00,
-6.9597e+00, -1.2313e+00],
[-8.6395e-01, -3.6877e+00, -4.7015e+00, ..., -7.6312e+00,
-8.2139e+00, -2.4200e+00],
[-9.4695e-01, -1.7722e+00, -2.5106e+00, ..., -6.9841e+00,
-7.8827e+00, -2.4660e+00],
...,
[-2.4148e+00, -3.2472e+00, -3.1217e+00, ..., -7.8885e+00,
-9.0038e+00, -2.0479e+00],
[-3.7267e+00, -5.3667e+00, -4.8195e+00, ..., -9.8439e+00,
-1.1168e+01, -4.3525e+00],
[-1.7964e-01, -2.9833e+00, -3.3418e+00, ..., -3.9763e+00,
-3.7744e+00, -7.3216e-01]],
[[ 5.3206e-01, 2.1875e+00, 1.8893e+00, ..., 7.6301e-01,
2.2592e+00, 1.0323e+00],
[-8.1890e-02, 1.0417e+00, 5.0144e-01, ..., 9.3270e-01,
2.6822e+00, 4.1746e-02],
[ 1.1853e+00, 4.7211e+00, 2.5445e+00, ..., -8.8006e-01,
1.9264e+00, -2.1544e-01],
...,
[ 1.1072e+00, 2.2543e+00, 1.0042e+00, ..., -3.8605e-01,
1.4174e+00, -1.2833e+00],
[ 3.9744e-01, 2.2303e+00, 7.5339e-01, ..., -1.6484e-01,
1.4070e+00, -1.4569e+00],
[-4.3348e-01, 1.6956e+00, 8.6413e-01, ..., -5.0490e-02,
-2.9932e-01, -1.9303e+00]],
...,
[[-3.1442e-01, -8.1207e-01, -1.4980e+00, ..., -5.3429e+00,
-5.2663e+00, -7.0651e-01],
[-2.5445e+00, -4.3634e+00, -2.0621e+00, ..., -4.3792e+00,
-5.2046e+00, -1.8374e+00],
[-1.5535e+00, -2.9322e+00, -1.4038e+00, ..., -5.1479e+00,
-5.7250e+00, -1.3005e+00],
...,
[-1.5681e+00, -2.9930e+00, -1.3311e+00, ..., -3.3583e+00,
-5.6336e+00, -1.5763e+00],
[-1.4306e+00, -3.3004e+00, -1.5252e+00, ..., -3.5255e+00,
-5.4531e+00, -8.5389e-01],
[-9.4522e-01, -3.3794e+00, -2.1877e+00, ..., -1.7531e+00,
-2.2244e+00, 4.4370e-01]],
[[-1.0509e+00, -5.6912e-01, -1.1155e+00, ..., -2.0026e+00,
-4.9724e-01, -5.8523e-01],
[-2.9690e+00, -2.6023e+00, -1.7860e+00, ..., -3.5679e+00,
-1.7168e+00, -3.5644e-01],
[-3.2098e+00, -4.5387e+00, -8.2762e-01, ..., -2.0746e+00,
-1.5838e+00, -4.4034e-01],
...,
[-3.3322e+00, -4.3133e+00, -2.1581e+00, ..., -3.7105e+00,
-2.2582e+00, -8.7640e-01],
[-3.0249e+00, -2.2904e+00, -6.9954e-01, ..., -1.5611e+00,
-6.5301e-01, 1.1879e+00],
[-3.0863e+00, -2.9679e+00, -8.3540e-01, ..., -1.9279e-01,
-8.5861e-01, 6.7965e-01]],
[[ 1.0498e+00, 8.7897e-01, 1.0274e+00, ..., -4.0558e-01,
5.9826e-01, -1.3164e+00],
[ 1.7128e+00, 2.2308e+00, 1.8969e+00, ..., 3.1408e+00,
2.9877e+00, 2.2894e+00],
[ 5.0716e-01, 4.6724e-01, 8.2417e-01, ..., 2.3970e+00,
1.5855e+00, 1.2551e+00],
...,
[ 2.0430e+00, 1.4326e+00, 1.6780e+00, ..., 2.9329e+00,
2.1239e+00, 6.7805e-01],
[ 2.8385e+00, 2.8728e+00, 1.9579e+00, ..., 1.6539e+00,
9.7318e-01, 1.9018e+00],
[ 1.8897e+00, 1.9379e-01, 2.7122e-01, ..., 1.8981e-01,
-9.7733e-02, 8.7919e-01]]],
[[[ 9.8270e-02, 9.7695e-01, 4.1004e-01, ..., -2.3699e+00,
-1.8191e+00, -6.0203e-01],
[ 2.2382e-01, 4.8446e-01, 1.3310e+00, ..., -3.4662e+00,
2.9790e+00, 1.7816e+00],
[ 1.7039e-01, 1.2576e+00, 1.5643e+00, ..., -5.1287e+00,
4.2097e+00, -8.3353e-02],
...,
[-3.3298e+00, -4.2875e+00, -1.7000e+00, ..., 5.7072e-01,
1.3520e+00, 6.6240e-01],
[-5.1530e+00, -3.9307e+00, -4.3918e-01, ..., -1.1938e+00,
3.5546e+00, 1.2578e+00],
[-3.1764e+00, -1.9169e+00, -1.9234e-01, ..., -4.7201e+00,
1.3139e+00, 2.3486e+00]],
[[-8.9447e-01, -1.8693e+00, -3.6812e+00, ..., -4.7266e+00,
-3.1640e+00, -7.0531e-01],
[-4.3927e+00, -5.6544e+00, -7.6040e+00, ..., -4.3055e+00,
1.4343e+00, -7.0313e-01],
[-3.2844e+00, -5.2497e+00, -6.3166e+00, ..., -1.0881e+01,
7.8048e-01, -5.0577e-01],
...,
[-7.0269e+00, -3.3220e+00, -7.3674e-01, ..., -2.8837e+00,
-1.2104e+00, -6.6539e-01],
[-1.0437e+01, -3.1707e+00, 1.6496e+00, ..., -8.4782e-01,
1.4875e+00, -1.5184e+00],
[-1.1361e+00, -5.4259e-01, 1.0276e+00, ..., -2.0807e+00,
8.4578e-01, 5.8529e-01]],
[[-4.6591e-01, 1.7510e+00, 9.6908e-01, ..., 3.9114e+00,
3.1968e+00, 1.7430e+00],
[-6.6514e-03, 8.9457e-01, 1.3680e+00, ..., -7.6060e+00,
-1.0005e+00, 2.0105e+00],
[-6.7919e-02, 2.2369e+00, 1.2727e+00, ..., -1.8538e+01,
-2.1893e+00, 3.8616e+00],
...,
[ 7.7921e-01, 3.1854e-01, -1.4866e+00, ..., 7.0140e-01,
5.7741e+00, 3.9377e+00],
[ 1.2303e+00, -1.5271e+00, 1.9086e-01, ..., -5.8409e+00,
-9.1371e-01, 3.7629e+00],
[ 1.0885e+00, -2.9039e+00, -1.6748e+00, ..., -1.0493e+00,
-7.5468e+00, 1.6713e+00]],
...,
[[-9.5075e-01, -1.4610e+00, -1.4360e+00, ..., -2.8343e+00,
-2.0149e+00, -1.5387e+00],
[-3.3972e+00, -4.1399e+00, -3.3719e+00, ..., 1.2755e+00,
-8.6103e-01, -3.4133e+00],
[-1.5034e+00, -2.1760e+00, -3.4887e+00, ..., 4.3418e+00,
-1.2447e+00, -2.5082e+00],
...,
[-2.3811e+00, -1.0268e+00, -1.1771e+00, ..., -2.2370e+00,
-1.3525e+00, -2.1437e+00],
[-4.6945e+00, 3.4555e-01, -1.9823e-01, ..., -4.3639e-01,
-1.4289e+00, -3.5434e+00],
[-1.7434e+00, 2.3272e-01, -7.5826e-01, ..., -3.8618e+00,
-2.3079e+00, -1.7973e+00]],
[[-1.6570e+00, -1.7766e+00, -2.2765e+00, ..., -2.8052e-01,
-8.5091e-01, 2.3300e-01],
[-4.4693e+00, -3.2369e+00, -3.7010e+00, ..., 1.0064e+00,
-1.4818e+00, -3.4612e-01],
[-2.6425e+00, -1.5176e+00, -3.7035e+00, ..., 4.4549e+00,
4.4966e-01, -2.2303e+00],
...,
[-2.0984e+00, 1.1712e+00, -1.6087e+00, ..., -5.1788e+00,
-6.1633e+00, 8.7439e-01],
[-1.4374e+00, 3.7955e+00, -2.9921e+00, ..., -1.2907e+00,
-2.9576e+00, -2.3700e+00],
[-8.6731e-01, 3.0004e+00, -1.7879e+00, ..., 1.6020e-01,
-1.5324e+00, -2.2638e+00]],
[[ 1.0605e+00, 2.6284e-01, 3.1815e-01, ..., 6.0162e-02,
1.9868e+00, 8.3248e-01],
[ 3.4022e+00, 3.3100e+00, 2.5721e+00, ..., -3.1556e+00,
2.1791e+00, 4.7828e+00],
[ 1.6652e+00, 1.5272e+00, 1.8955e+00, ..., -1.7736e+00,
4.7332e+00, 2.9131e+00],
...,
[ 2.5405e+00, -2.4766e-01, -5.2740e-01, ..., -1.4215e+00,
2.6628e+00, 5.1325e+00],
[ 3.1035e+00, 2.2402e+00, -5.7590e-01, ..., -3.1376e+00,
1.1485e+00, 4.7689e+00],
[-5.0063e-01, 2.1919e+00, -8.7508e-01, ..., -4.6469e+00,
-2.6019e-01, 1.5548e+00]]],
[[[ 8.1163e-01, 1.6411e+00, 6.5253e-01, ..., -3.3558e+00,
-2.5960e+00, -1.4318e+00],
[ 1.8120e+00, 1.2057e+00, 9.8873e-01, ..., -3.2750e+00,
-4.8138e-01, -3.7956e-01],
[ 6.8316e-01, -1.0454e+00, -9.0239e-01, ..., -4.6565e+00,
-5.7947e+00, -2.8055e+00],
...,
[ 4.4972e-01, -8.6476e-01, -2.8156e-02, ..., 1.4741e+00,
-8.3543e-01, 1.5214e+00],
[ 8.0409e-02, -2.7338e-01, -7.9619e-01, ..., -2.7826e+00,
-3.5254e+00, 7.5853e-02],
[-2.8124e-01, -3.5095e+00, -3.3904e+00, ..., -3.7172e+00,
-2.9994e+00, -1.8353e+00]],
[[ 2.8046e-01, -1.1499e+00, -2.6655e+00, ..., -9.6696e+00,
-9.4125e+00, -2.7001e+00],
[-2.1835e+00, -4.1563e+00, -4.7261e+00, ..., -6.0877e+00,
-8.1628e+00, -4.0686e+00],
[-6.9687e-01, -1.9236e+00, -1.6580e+00, ..., -9.9983e+00,
-1.0838e+01, -3.7177e+00],
...,
[-2.2300e+00, -3.2024e+00, -2.7280e+00, ..., -4.8967e+00,
-6.6036e+00, -2.2887e+00],
[-4.5614e+00, -6.1727e+00, -5.4346e+00, ..., -1.0945e+01,
-1.2619e+01, -5.5491e+00],
[-3.7538e-01, -3.3667e+00, -3.6333e+00, ..., -3.6622e+00,
-3.3071e+00, -8.4784e-01]],
[[ 4.3251e-01, 2.0925e+00, 1.9075e+00, ..., 8.4863e-01,
2.3228e+00, 6.6386e-01],
[-2.2384e-02, 8.7470e-01, 6.9264e-01, ..., 1.2987e+00,
2.7988e+00, -9.1335e-02],
[ 6.2977e-01, 2.9819e+00, 1.8272e+00, ..., 3.5405e+00,
6.4284e+00, 2.9564e+00],
...,
[ 1.0258e+00, 2.4989e+00, 1.1056e+00, ..., -1.7766e+00,
-2.0368e-01, -1.4064e+00],
[ 8.4692e-01, 1.7961e+00, 5.6638e-01, ..., -1.2829e+00,
-1.4798e-01, -1.8823e+00],
[-2.5292e-01, 1.8474e+00, 9.8849e-01, ..., -3.7745e-01,
-3.4200e-01, -1.6247e+00]],
...,
[[-3.2799e-01, -8.7205e-01, -1.5709e+00, ..., -5.9434e+00,
-6.1675e+00, -1.1822e+00],
[-2.6855e+00, -4.3258e+00, -2.6528e+00, ..., -3.4779e+00,
-4.8457e+00, -2.4550e+00],
[-1.1311e+00, -2.2032e+00, -1.1299e+00, ..., -6.0285e+00,
-7.3160e+00, -3.1718e+00],
...,
[-1.2315e+00, -1.6457e+00, -3.6091e-01, ..., -1.9786e+00,
-3.4266e+00, -1.3469e+00],
[-2.0485e+00, -3.5963e+00, -2.1214e+00, ..., -5.3868e+00,
-6.9054e+00, -1.6667e+00],
[-1.1715e+00, -3.7188e+00, -2.6201e+00, ..., -7.0821e-01,
-1.6192e+00, 3.3758e-01]],
[[-9.8080e-01, -8.4236e-01, -9.8503e-01, ..., -2.1535e+00,
-1.3506e+00, -8.8479e-01],
[-3.0564e+00, -2.4247e+00, -1.2807e+00, ..., -2.1370e+00,
-6.5626e-01, 1.1582e-01],
[-2.4285e+00, -3.7428e+00, -1.3312e+00, ..., -6.4074e+00,
-4.4564e+00, -5.4142e-01],
...,
[-2.8138e+00, -3.4618e+00, -2.3378e+00, ..., -2.5537e+00,
-1.9084e+00, -2.1831e-01],
[-3.4899e+00, -2.4578e+00, -1.1549e+00, ..., -1.7826e+00,
-1.2331e+00, 1.5846e+00],
[-3.4149e+00, -3.4768e+00, -1.4943e+00, ..., 9.7437e-02,
-1.3475e+00, 1.7496e-01]],
[[ 8.8643e-01, 8.5557e-01, 1.0553e+00, ..., 3.5289e-01,
9.4654e-01, -1.1837e+00],
[ 1.7797e+00, 2.3439e+00, 2.1218e+00, ..., 3.3168e+00,
3.5265e+00, 2.4988e+00],
[ 7.2293e-01, -3.6756e-01, 2.4151e-01, ..., 4.0432e+00,
3.0769e+00, 1.0713e+00],
...,
[ 1.9577e+00, 3.2663e-01, 9.0164e-01, ..., 1.4711e+00,
7.2992e-01, 1.1292e+00],
[ 3.3055e+00, 3.6616e+00, 2.8467e+00, ..., 1.6992e+00,
6.6493e-01, 1.7560e+00],
[ 1.9497e+00, 6.1314e-01, 7.9178e-01, ..., 1.7923e-01,
5.2197e-02, 1.0816e+00]]],
[[[-2.0363e-01, 1.0406e+00, 7.9559e-01, ..., -1.0868e+00,
-3.4443e+00, -1.0246e+00],
[ 1.9962e-01, -6.3613e-01, -4.3930e-01, ..., 1.1892e-01,
-2.2879e+00, -3.9018e-01],
[-7.7420e-01, -1.2218e+00, -7.0753e+00, ..., 1.1037e+00,
-1.2141e+00, -1.3615e+00],
...,
[-5.9079e+00, -4.1825e+00, -7.3859e+00, ..., -3.2790e+00,
-1.7542e-01, 8.1921e-01],
[-6.1228e+00, -5.1887e+00, -5.5429e+00, ..., -5.0489e+00,
-3.0159e+00, -1.4794e+00],
[-3.4220e+00, -4.4431e+00, -1.5913e-01, ..., -4.2950e+00,
-4.4724e+00, -1.7289e+00]],
[[-1.3777e+00, -4.4480e+00, -7.4654e+00, ..., -5.2509e+00,
-9.0223e+00, -1.7270e+00],
[-4.8931e+00, -4.7708e+00, -1.3740e+01, ..., -4.8022e+00,
-8.4409e+00, -2.8327e+00],
[-5.2726e+00, -9.4103e+00, -1.2747e+01, ..., -4.4368e+00,
-9.9771e+00, -6.5663e+00],
...,
[-7.0945e+00, -1.5109e-01, -2.1460e+00, ..., -1.5549e+01,
-7.0434e+00, 2.0368e+00],
[-1.1716e+01, -5.3797e+00, 2.4829e+00, ..., -8.4064e+00,
-1.5424e+01, -8.6430e+00],
[-2.1636e+00, -7.0451e+00, -2.2009e+00, ..., -1.8290e+00,
-2.1793e+00, 3.0196e-01]],
[[-3.3468e-01, 2.4970e+00, 2.0230e-01, ..., 2.1892e+00,
2.6152e+00, 7.5285e-01],
[-6.5352e-01, -9.9717e-01, -9.6911e-01, ..., 1.3776e+00,
2.3688e+00, -3.2217e-01],
[ 4.0108e-01, 3.0193e-02, 2.8211e+00, ..., 1.1137e+00,
2.0926e+00, -5.7328e-01],
...,
[ 1.4904e+00, -1.5918e-02, 2.5588e+00, ..., -3.7872e+00,
-7.5447e+00, 7.6387e-01],
[-1.8756e+00, -5.1253e+00, 1.7506e+00, ..., 2.5753e+00,
-2.4126e+00, -3.6342e+00],
[ 4.9789e-01, -2.4369e+00, -4.5495e+00, ..., -5.3941e-01,
1.1514e+00, -1.1780e+00]],
...,
[[-1.1910e+00, -3.6700e+00, -1.3705e+00, ..., -3.4460e+00,
-5.6769e+00, -1.2360e+00],
[-3.1814e+00, -4.9490e+00, -5.1023e+00, ..., -2.3332e+00,
-5.5539e+00, -2.2552e+00],
[-2.3326e+00, -5.7813e+00, -6.8948e+00, ..., -2.1201e+00,
-6.9628e+00, -1.8828e+00],
...,
[-2.8823e+00, -3.0426e+00, -7.5270e+00, ..., -6.2614e+00,
-6.6227e+00, -2.9878e+00],
[-4.0329e+00, -3.7002e+00, -9.1938e-01, ..., -3.0536e+00,
-9.7270e+00, -6.5821e+00],
[-2.5842e+00, -5.0484e+00, -1.3444e+00, ..., -4.9647e-01,
1.4252e-01, -2.2149e-01]],
[[-1.3990e+00, -1.8638e+00, -6.3800e-02, ..., -1.9709e+00,
-8.6596e-01, 2.6731e-01],
[-2.9664e+00, -2.6608e+00, -4.1470e+00, ..., -2.4730e+00,
-2.2935e+00, 9.6159e-02],
[-5.3401e+00, -6.1886e+00, -3.0332e+00, ..., -8.1402e-01,
-2.5599e+00, -2.4009e-01],
...,
[-1.5162e+00, -5.9109e+00, -1.7053e+01, ..., 7.6869e+00,
4.4130e+00, -2.1475e-01],
[-1.4138e+00, 4.4277e+00, -8.7358e+00, ..., 6.1837e-01,
2.1080e+00, -1.0627e-01],
[-1.1510e+00, 4.8643e+00, 2.0784e+00, ..., -4.1379e-01,
1.0851e+00, 3.6479e-01]],
[[ 1.4914e+00, 4.2422e-01, 2.0272e+00, ..., 1.7911e+00,
1.7716e+00, -9.6148e-02],
[ 3.8051e+00, 2.7725e+00, 6.9781e+00, ..., 2.9741e+00,
2.8697e+00, 3.5427e+00],
[ 1.6607e+00, 2.0115e+00, 3.1504e+00, ..., 2.6662e+00,
3.2152e+00, 3.3579e+00],
...,
[ 1.8088e+00, -2.0731e+00, -8.2718e+00, ..., 1.3391e+01,
3.6697e+00, -4.6342e-01],
[ 3.4416e+00, 1.4195e+00, -7.6948e+00, ..., 4.0240e+00,
7.3647e+00, 4.7395e+00],
[ 1.5358e+00, 4.5740e+00, 1.1384e+00, ..., -1.2252e+00,
-1.1935e+00, 8.3758e-01]]]], device='cuda:0'),
tensor([[[[-1.2645e-01, 6.0538e-01, -2.8848e-01, ..., -1.3709e+00,
-9.4766e-01, -8.3741e-01],
[ 6.6944e-01, 7.9298e-01, 9.2660e-01, ..., -6.1870e-01,
1.1452e-01, -1.1521e+00],
[ 1.5409e+00, -2.3268e-02, 1.0595e+00, ..., -1.6840e+00,
-6.2630e-01, -2.0049e+00],
...,
[-3.3796e-01, 7.5423e-02, 9.0203e-01, ..., -1.0292e+00,
-2.7027e+00, 1.8452e-02],
[-2.3213e+00, -1.7338e+00, 1.4503e+00, ..., -1.6648e+00,
-2.4837e+00, 2.0086e-01],
[-2.3803e+00, -4.2794e+00, -3.0031e+00, ..., -3.5066e+00,
-3.2448e+00, -1.3968e+00]],
[[-7.3070e-01, -1.4029e+00, -2.6024e+00, ..., -3.0684e+00,
-4.5849e+00, -2.4713e+00],
[-1.9219e+00, -3.8682e+00, -4.7953e+00, ..., -3.2195e+00,
-4.9335e+00, -4.6344e+00],
[-1.8206e+00, -4.0520e+00, -3.9825e+00, ..., -2.9853e+00,
-4.1021e+00, -3.5594e+00],
...,
[-6.3087e+00, -8.5106e+00, -6.3738e+00, ..., -8.2736e+00,
-8.8094e+00, -1.8424e+00],
[-9.6680e+00, -1.0229e+01, -8.5073e+00, ..., -9.0376e+00,
-1.0366e+01, -3.7694e+00],
[-2.5367e+00, -4.9692e+00, -4.1746e+00, ..., -3.6819e+00,
-3.9874e+00, -5.6436e-01]],
[[ 6.8959e-01, 1.1969e+00, 1.1916e+00, ..., 1.6027e+00,
1.8606e+00, 8.4641e-01],
[ 9.2698e-01, 1.5168e+00, 1.5789e+00, ..., 1.2714e+00,
2.2339e+00, 9.5890e-01],
[ 6.3088e-01, 2.9840e+00, 4.1389e-01, ..., 1.4219e+00,
2.5175e+00, 2.1073e+00],
...,
[-1.2639e-01, -1.4065e-01, -6.0945e-01, ..., 1.8393e+00,
1.6337e+00, -1.2520e-01],
[-7.0423e-01, 6.0824e-01, -1.1525e+00, ..., -4.3977e-02,
9.8020e-01, -3.2563e-01],
[ 5.8000e-01, 1.4334e+00, 1.2164e+00, ..., -1.6052e+00,
-7.0770e-01, -1.9085e+00]],
...,
[[-9.9785e-01, -1.2375e+00, -2.0702e+00, ..., -2.5574e+00,
-3.2772e+00, -1.2271e+00],
[-2.2840e+00, -2.8131e+00, -2.6234e+00, ..., -2.9873e+00,
-4.2249e+00, -3.5553e+00],
[-5.1514e-01, -1.8209e+00, -1.7091e+00, ..., -3.4454e+00,
-4.2379e+00, -3.1527e+00],
...,
[-2.4170e+00, -3.0991e+00, 1.4287e-01, ..., -5.2263e+00,
-6.7393e+00, -2.5747e+00],
[-3.7637e+00, -5.7551e+00, -1.6608e+00, ..., -3.3748e+00,
-5.4390e+00, -1.3092e+00],
[-3.2261e+00, -3.7328e+00, -3.0098e+00, ..., -2.8849e+00,
-2.6906e+00, 4.5866e-01]],
[[-1.5488e+00, -2.2759e+00, -3.5342e+00, ..., -2.8753e+00,
-2.7915e+00, -5.2224e-01],
[-4.3607e+00, -4.3223e+00, -5.1847e+00, ..., -3.9866e+00,
-5.7732e+00, -1.9499e+00],
[-4.1559e+00, -4.9168e+00, -3.0790e+00, ..., -2.5134e+00,
-4.8229e+00, -2.0638e+00],
...,
[-3.8531e+00, -2.6288e+00, -4.4850e+00, ..., -4.5342e+00,
-1.9438e+00, -1.1210e+00],
[-3.4528e+00, -1.3152e-01, -1.1687e+00, ..., -1.6786e+00,
-4.2673e-01, 3.6874e-01],
[-3.7375e+00, -2.9768e+00, -2.3821e+00, ..., -1.1955e+00,
-8.3125e-01, 3.6400e-01]],
[[ 8.8387e-01, -1.5007e-01, 6.8156e-01, ..., 5.3503e-01,
1.3847e+00, 6.1568e-01],
[ 1.6785e+00, 1.0489e+00, 1.1023e+00, ..., 7.2739e-01,
1.9589e+00, 3.2605e+00],
[ 1.3224e+00, 4.0484e-01, 1.8722e+00, ..., 6.5182e-03,
1.0700e+00, 8.7918e-01],
...,
[ 3.9241e+00, 3.7533e+00, 2.5736e+00, ..., 3.1654e+00,
3.0697e+00, 7.0179e-01],
[ 4.5923e+00, 4.1548e+00, 3.3103e+00, ..., 2.2303e+00,
1.6805e+00, 1.5768e+00],
[ 2.8275e+00, 2.5047e+00, 2.1638e+00, ..., 1.2997e+00,
7.4698e-01, 1.0220e+00]]],
[[[-4.2702e+00, -3.0321e+00, -4.9617e+00, ..., -8.5158e-01,
2.9786e+00, 2.1691e+00],
[-5.0489e+00, -4.3608e+00, -3.8763e+00, ..., -1.9907e+00,
3.2827e+00, 6.9879e-01],
[-3.2571e+00, -4.7091e+00, -1.8477e+00, ..., -8.3619e-01,
4.1006e-01, -2.3297e+00],
...,
[-6.9942e+00, -6.4132e+00, 3.5796e+00, ..., -3.7305e+00,
-4.0505e+00, -8.8661e-01],
[-5.2460e+00, -3.5320e+00, 2.5515e+00, ..., -3.2791e+00,
-2.8930e+00, 1.2891e-01],
[-2.0066e+00, -4.9186e+00, -1.1818e+00, ..., -3.4061e+00,
-2.6404e+00, -1.2512e+00]],
[[-4.6433e+00, -4.4990e-01, -2.4629e+00, ..., -1.6722e-01,
2.8084e+00, 4.8609e-01],
[-7.7551e+00, -5.1023e+00, 8.1715e-01, ..., -8.5868e-01,
-1.4022e+00, -3.1920e+00],
[-3.8957e+00, -8.1400e+00, -4.0371e+00, ..., -3.7040e+00,
-4.8157e+00, -4.9155e+00],
...,
[-9.0173e+00, -8.5792e+00, 2.0922e+00, ..., -7.8692e+00,
-8.8056e+00, -2.3220e+00],
[-9.4372e+00, -1.1087e+01, -1.8341e+00, ..., -9.8888e+00,
-1.0576e+01, -3.6484e+00],
[-1.7986e+00, -5.6006e+00, -5.5255e+00, ..., -3.7034e+00,
-4.0498e+00, -5.8917e-01]],
[[-4.6313e-01, 3.7727e+00, 3.7884e+00, ..., -2.0822e-01,
1.5818e+00, 1.7704e+00],
[-9.9416e-01, -4.5093e+00, 3.1389e+00, ..., -3.0327e+00,
-8.0892e+00, 9.6110e-01],
[ 1.6761e+00, 7.6360e-02, -1.1906e+00, ..., -4.9851e-01,
-3.3378e+00, 3.4679e-01],
...,
[ 4.9314e-01, -7.1200e+00, -1.8117e+01, ..., -4.0247e-01,
-1.2867e-01, -1.3897e+00],
[-2.3430e+00, -1.0402e+01, -2.0753e+01, ..., -1.8547e+00,
9.1255e-01, -5.9097e-01],
[-1.0480e+00, -3.6236e+00, -8.8281e+00, ..., -2.6905e-01,
-5.9344e-01, -1.8609e+00]],
...,
[[-2.7112e+00, -1.4619e+00, -2.7894e+00, ..., 5.8605e-01,
6.9620e-01, -1.2850e+00],
[-5.9781e+00, -3.5018e+00, 8.6706e-01, ..., -1.8671e+00,
-3.6889e+00, -7.3517e+00],
[-4.3692e+00, -6.5799e+00, -3.0753e+00, ..., -3.3478e+00,
-5.3292e+00, -5.9760e+00],
...,
[-3.4954e+00, -3.0394e+00, -4.6313e+00, ..., -4.6290e+00,
-6.4686e+00, -3.3191e+00],
[-3.6332e+00, -5.7651e+00, -6.0942e+00, ..., -4.0737e+00,
-5.7506e+00, -1.5042e+00],
[-2.6439e+00, -6.3212e+00, -6.6833e+00, ..., -2.6111e+00,
-2.4808e+00, 5.3051e-01]],
[[-2.8201e-01, -4.3192e+00, -1.0059e+01, ..., -1.3747e+00,
-2.0679e+00, 1.7793e+00],
[-7.4908e-01, 4.5089e+00, -7.8418e-01, ..., -2.6817e-01,
-3.6745e+00, -1.1925e+00],
[-3.3728e+00, 2.7038e+00, 4.3989e-01, ..., 4.1632e-01,
-3.3258e+00, -3.0625e+00],
...,
[-4.4449e+00, 9.6105e-01, -7.5839e+00, ..., -1.7785e+00,
-1.4057e+00, -9.6836e-01],
[-3.3345e+00, 3.2153e+00, 9.6584e-02, ..., -1.0236e+00,
-3.7542e-01, 3.8427e-01],
[-2.9442e+00, -5.2709e-01, -1.1255e+00, ..., -3.2446e-02,
-8.4166e-01, 2.7239e-01]],
[[ 2.6396e+00, -1.7274e+00, -3.3197e+00, ..., 5.4027e-02,
9.0771e-01, 1.7749e+00],
[ 2.4575e+00, 1.8683e+00, -4.2587e+00, ..., -1.3579e+00,
2.9124e+00, 4.3212e+00],
[-3.7354e-02, 3.5086e+00, 4.7809e-01, ..., -3.7590e-01,
4.5056e+00, 3.8168e+00],
...,
[ 2.0393e+00, 4.3798e+00, -3.9424e-01, ..., 1.9175e+00,
2.2424e+00, 1.0117e-01],
[ 2.6724e+00, 6.3243e+00, 1.0265e+00, ..., 4.1823e+00,
1.9681e+00, 1.4585e+00],
[ 1.5149e+00, 3.6436e+00, 4.3365e+00, ..., 1.2389e-01,
3.7160e-01, 1.1848e+00]]],
[[[-1.7778e+00, -7.6525e+00, -2.3624e-02, ..., -1.8119e-01,
1.0006e-01, -3.7704e-01],
[-3.1641e+00, -5.4789e+00, 1.8026e+00, ..., -4.1774e-01,
-2.6795e-01, -8.5794e-01],
[-5.4636e+00, -2.7464e+00, -8.1619e-01, ..., 1.1888e+00,
5.3927e-01, -5.4775e-01],
...,
[-1.7092e-01, -1.9303e-01, -5.6413e-01, ..., -9.6216e-01,
-1.1521e+00, 2.7225e+00],
[-1.4265e+00, 2.7979e-01, 1.6763e-02, ..., 2.4405e-01,
4.8925e-01, 6.7125e-01],
[-1.2786e+00, -2.3066e+00, -1.7032e+00, ..., -2.5136e+00,
-3.2446e+00, -2.7117e+00]],
[[-5.5602e+00, -9.5919e+00, -6.6488e-01, ..., -2.7170e+00,
-4.1388e+00, -2.5577e+00],
[-6.4726e+00, -1.2957e+00, 2.9573e+00, ..., -2.2323e+00,
-4.2304e+00, -3.6845e+00],
[-7.4487e+00, -1.4078e+00, -2.8235e-01, ..., -1.8617e-01,
-1.9815e+00, -2.3905e+00],
...,
[-4.5273e+00, -5.1230e+00, -5.7674e+00, ..., 1.5475e+00,
-3.7670e-01, 7.5925e-01],
[-7.7423e+00, -8.0936e+00, -6.6057e+00, ..., -6.6466e+00,
-4.5619e+00, -4.9276e+00],
[-1.4378e+00, -3.1298e+00, -2.1603e+00, ..., -2.3335e+00,
-2.9852e+00, -1.6733e+00]],
[[ 3.2098e-01, 4.6379e+00, -7.1066e-01, ..., 1.6420e+00,
1.7485e+00, 8.2645e-01],
[ 1.1378e+00, -1.1110e+00, -7.1719e+00, ..., 1.8523e+00,
1.8129e+00, 9.0182e-01],
[ 2.6144e+00, -1.6738e+00, -5.0393e+00, ..., 1.7549e+00,
1.9758e+00, 1.6428e+00],
...,
[-4.5507e-01, -3.6041e-01, 7.0698e-01, ..., 1.3435e+00,
1.3814e+00, -5.2593e-01],
[-1.2866e+00, -2.5614e-01, 3.7900e-02, ..., -7.0789e-01,
5.0387e-01, -8.5280e-01],
[-3.5839e-01, -3.0393e-01, -1.6791e-01, ..., 2.8273e-01,
6.0525e-01, 2.6693e-01]],
...,
[[-3.8445e+00, -4.2133e+00, 2.1802e+00, ..., -1.7677e+00,
-2.2029e+00, -5.0740e-01],
[-5.0395e+00, 3.1838e-01, 1.0459e-01, ..., -2.1476e+00,
-3.3212e+00, -2.8894e+00],
[-5.6146e+00, -6.4134e-01, -3.3342e+00, ..., -4.0768e-01,
-1.7817e+00, -1.6598e+00],
...,
[-9.4234e-01, -2.4186e+00, -1.6245e+00, ..., 3.9083e-01,
-2.2345e+00, -1.5989e+00],
[-2.1365e+00, -3.9773e+00, -9.4481e-01, ..., -2.7594e+00,
-2.8733e+00, -1.3734e+00],
[-1.4576e+00, -1.5530e+00, -6.7578e-01, ..., -3.0406e+00,
-3.9909e+00, -1.5525e+00]],
[[-4.0372e+00, -2.5622e+00, -2.8388e+00, ..., -3.5211e-01,
-1.2626e+00, -4.5541e-01],
[-3.7474e+00, -1.1963e+00, -7.4504e+00, ..., -2.3090e+00,
-3.8372e+00, -1.3375e+00],
[-2.6058e+00, -8.8127e-01, -9.9043e+00, ..., -1.3809e+00,
-3.9931e+00, -2.4058e+00],
...,
[-2.7385e+00, -3.2331e+00, -3.5270e+00, ..., -4.7478e+00,
-3.9658e+00, -1.2398e+00],
[-1.8952e+00, -2.0722e-01, -1.4252e+00, ..., -1.7766e+00,
-4.5727e+00, -9.5478e-01],
[-2.1202e+00, -1.4151e+00, -1.4800e+00, ..., -4.5610e-01,
-2.1134e+00, 2.7302e-01]],
[[ 3.3450e+00, -2.2133e+00, 2.1542e+00, ..., 3.5495e-01,
1.2283e+00, 5.4262e-01],
[ 5.1928e+00, -6.1675e-01, 2.3129e+00, ..., 1.0089e+00,
1.2990e+00, 2.5336e+00],
[ 1.4980e+00, 1.5374e-01, -2.8665e+00, ..., 5.8202e-01,
3.2569e-01, 7.8379e-01],
...,
[ 3.0236e+00, 2.4843e+00, 1.4620e+00, ..., -5.2215e+00,
-5.7474e+00, -2.7427e-01],
[ 3.8446e+00, 4.1749e+00, 2.3978e+00, ..., 2.4741e+00,
5.9247e-01, 2.0499e+00],
[ 2.4703e+00, 2.5169e+00, 1.7723e+00, ..., 3.0585e-01,
8.6440e-02, 1.4839e-01]]],
[[[-2.0696e+00, -1.5750e+00, 1.3917e+00, ..., -3.5441e-01,
-2.3530e+00, -1.0544e+00],
[-1.3523e+00, -6.5845e-01, 1.0782e+00, ..., -1.4814e+00,
-3.1124e+00, -2.2775e+00],
[-1.3401e+00, -8.5571e-01, -1.9755e+00, ..., 7.4698e-01,
7.2466e-01, -1.7374e-01],
...,
[-4.4022e-01, -1.9386e-01, 1.5227e+00, ..., 1.9466e+00,
-2.7969e+00, -1.1446e+00],
[-1.6693e+00, -1.1298e-01, 9.0562e-01, ..., -2.2782e+00,
-4.2226e+00, -1.0809e+00],
[-1.1915e+00, -2.2424e+00, -1.1418e+00, ..., -3.4624e+00,
-4.2522e+00, -1.9000e+00]],
[[-1.9364e+00, -2.6573e+00, 5.5217e-01, ..., -5.2975e+00,
-7.6833e+00, -1.9817e+00],
[-2.1112e+00, -3.3601e-02, 1.8548e+00, ..., -5.0098e+00,
-7.7300e+00, -2.1306e+00],
[ 1.0050e+00, -4.3174e-01, -4.5541e-02, ..., -1.8367e+00,
-2.9600e+00, -1.0436e-01],
...,
[-4.5375e+00, -5.4324e+00, -5.7191e+00, ..., -4.5654e+00,
-8.8907e+00, -2.9026e+00],
[-7.7865e+00, -7.9431e+00, -7.2713e+00, ..., -9.7573e+00,
-1.1229e+01, -4.4533e+00],
[-1.3946e+00, -3.0836e+00, -2.1908e+00, ..., -4.0996e+00,
-4.2300e+00, -5.4245e-01]],
[[ 6.1767e-01, 3.4262e+00, 1.1584e+00, ..., 2.1680e+00,
2.8846e+00, 5.5433e-01],
[-1.9008e-01, -1.9994e+00, -2.6300e+00, ..., 2.4813e+00,
4.3042e+00, 1.6346e+00],
[-3.4185e-01, -1.3429e+00, -1.3739e+00, ..., 1.4160e+00,
1.4864e+00, 1.4604e+00],
...,
[-7.2080e-01, 2.0049e-01, 1.4105e+00, ..., -1.4621e+00,
1.7890e+00, 7.3477e-01],
[-1.2442e+00, 1.6050e-01, 1.2335e+00, ..., -2.2742e-01,
-3.6148e-01, -1.7804e+00],
[-3.7781e-01, -3.0768e-01, 2.0220e-01, ..., 4.9484e-01,
-4.6779e-01, -1.8902e+00]],
...,
[[-2.1439e+00, -3.1013e-01, 1.4092e+00, ..., -2.7600e+00,
-5.0802e+00, -1.3736e+00],
[-1.9278e+00, 3.1860e-01, 9.1418e-01, ..., -1.6889e+00,
-4.0625e+00, -2.7723e+00],
[-8.7450e-02, -6.7386e-01, -1.7481e-01, ..., 1.6257e+00,
-9.3744e-01, -4.9607e-01],
...,
[-1.1642e+00, -2.5606e+00, -7.9325e-01, ..., -2.1619e+00,
-5.6903e+00, -2.3352e+00],
[-2.1951e+00, -3.9023e+00, -7.7820e-01, ..., -2.6039e+00,
-4.1985e+00, -7.4861e-01],
[-1.3163e+00, -1.2894e+00, 4.0355e-01, ..., -6.8014e-01,
-2.0132e+00, 7.7923e-01]],
[[-1.2268e+00, 2.3066e-02, -1.6889e-01, ..., -1.8477e+00,
-1.0306e+00, 8.8786e-01],
[-1.8764e+00, -2.2444e-01, -2.2810e+00, ..., -2.6297e+00,
-4.5426e+00, -3.2728e-01],
[-3.1992e+00, -1.8110e+00, -2.5051e+00, ..., -8.2707e-01,
-3.6908e+00, -2.1831e+00],
...,
[-2.5575e+00, -2.6975e+00, -3.0411e+00, ..., -2.8535e+00,
-4.0039e+00, 7.4803e-02],
[-1.9771e+00, -1.3591e-01, -1.4928e+00, ..., -4.6368e-01,
-1.4687e+00, 1.4114e+00],
[-2.0912e+00, -1.2174e+00, -4.9935e-01, ..., -9.1181e-01,
-1.2473e+00, 7.2532e-01]],
[[ 9.4739e-01, -3.9801e-01, 8.7617e-01, ..., 2.1245e+00,
1.9515e+00, 2.3296e-01],
[ 1.4137e+00, 1.4628e+00, 2.2083e-01, ..., 2.1021e+00,
2.6653e+00, 3.5908e+00],
[-1.2230e+00, -1.9165e+00, -3.2996e+00, ..., 4.0330e+00,
2.0159e+00, 2.6496e+00],
...,
[ 3.0257e+00, 2.4884e+00, 2.8264e+00, ..., 3.1727e+00,
1.3116e+00, 1.3018e+00],
[ 3.7420e+00, 3.9670e+00, 2.8546e+00, ..., 2.7904e+00,
6.6376e-01, 2.8965e+00],
[ 2.4105e+00, 2.3824e+00, 1.5989e+00, ..., 6.8033e-01,
5.0737e-01, 1.6106e+00]]],
[[[-1.3296e-01, 7.0421e-01, -3.2702e-01, ..., -5.5952e-01,
-1.5517e+00, -8.6641e-01],
[ 8.1752e-01, 4.3072e-01, -2.5833e-01, ..., 6.8656e-01,
3.5313e+00, 1.8545e+00],
[ 1.0254e+00, -6.9501e-01, -7.7137e-01, ..., -4.9888e+00,
3.8534e+00, 2.8242e+00],
...,
[-8.8135e-01, -5.2974e+00, -5.1715e+00, ..., 1.5282e+00,
-1.6093e+00, -7.2923e-01],
[-2.8340e+00, -5.4451e+00, -4.6929e+00, ..., -2.0640e+00,
-2.8748e+00, -8.4083e-02],
[-1.0442e+00, -1.4442e-01, 1.5852e+00, ..., -4.0340e+00,
-3.8868e+00, -1.6799e+00]],
[[-7.6459e-01, -1.5320e+00, -2.7320e+00, ..., -2.3710e+00,
-4.7457e+00, -1.7660e+00],
[-2.0042e+00, -3.8109e+00, -4.0252e+00, ..., 7.4768e-01,
1.7428e+00, -1.3721e+00],
[-1.9016e+00, -3.5110e+00, -3.3770e+00, ..., -2.8076e+00,
3.7576e+00, -2.1956e-01],
...,
[ 6.1370e-01, -5.2615e+00, -6.9974e+00, ..., -6.5563e+00,
-8.6829e+00, -2.4300e+00],
[-3.9519e+00, -4.6177e+00, -1.9340e+00, ..., -1.0665e+01,
-1.0897e+01, -3.8857e+00],
[ 2.2780e-01, 1.6917e+00, 8.3015e-01, ..., -4.0831e+00,
-4.3226e+00, -6.4487e-01]],
[[ 7.3405e-01, 1.2383e+00, 1.0405e+00, ..., 2.0319e+00,
5.4737e+00, 2.3336e+00],
[ 8.1910e-01, 1.4646e+00, 1.4303e+00, ..., -4.4428e+00,
4.6077e+00, 4.9567e+00],
[ 5.0920e-01, 3.0872e+00, 2.0836e+00, ..., -2.1146e+00,
-2.0837e+00, 2.8915e+00],
...,
[-2.8577e+00, -5.9992e+00, -9.8773e+00, ..., 1.1504e+00,
1.3232e+00, -5.1010e-02],
[-4.8077e+00, -6.1600e+00, -7.6170e+00, ..., 2.7080e-01,
5.7460e-01, -4.0808e-01],
[-6.3159e+00, -5.9581e+00, -5.4912e+00, ..., -6.5839e-01,
-9.0305e-01, -1.9856e+00]],
...,
[[-9.8287e-01, -1.3385e+00, -2.1264e+00, ..., -1.4892e+00,
-5.5254e+00, -2.5835e+00],
[-2.2462e+00, -3.3254e+00, -2.5175e+00, ..., -5.6475e-01,
-1.8822e+00, -3.4450e+00],
[-1.0113e+00, -2.2205e+00, -1.0383e+00, ..., -1.1835e-01,
-2.7076e+00, -5.0242e+00],
...,
[-8.0078e-01, -3.0256e+00, -1.1989e+00, ..., -5.9326e+00,
-6.8231e+00, -2.7709e+00],
[-3.8719e+00, -4.2131e+00, -3.2219e+00, ..., -4.2690e+00,
-5.6476e+00, -1.2793e+00],
[-3.2798e-01, 1.0182e+00, 1.1214e+00, ..., -2.5765e+00,
-2.6582e+00, 3.6336e-01]],
[[-1.6147e+00, -2.3979e+00, -3.5461e+00, ..., -6.8069e+00,
-5.1455e+00, 1.8733e+00],
[-4.4259e+00, -4.9038e+00, -5.1176e+00, ..., -2.4360e+00,
-6.7713e+00, 7.4933e-01],
[-4.0554e+00, -5.1851e+00, -4.0043e+00, ..., -2.2599e-01,
-5.5522e+00, -5.2027e-01],
...,
[-4.4888e+00, -6.4265e+00, -2.6969e+00, ..., -4.0757e+00,
-4.3981e+00, -1.2643e+00],
[-4.1195e+00, -6.2489e+00, -1.4936e+00, ..., -2.1723e+00,
-1.8072e+00, 2.9735e-01],
[ 4.5161e-01, 7.6057e-01, 2.1395e+00, ..., -1.8285e+00,
-1.3611e+00, 3.3255e-01]],
[[ 8.6862e-01, -1.1013e-01, 7.4156e-01, ..., 5.6588e-01,
3.3416e+00, 1.8909e+00],
[ 1.7550e+00, 9.3044e-01, 5.0173e-01, ..., -3.0050e+00,
2.8535e+00, 5.4132e+00],
[ 1.4106e+00, 1.8884e-01, 9.8119e-02, ..., -6.4194e+00,
-9.5713e-02, 4.1849e+00],
...,
[-6.1746e-01, -4.5697e+00, -2.7095e+00, ..., 3.6668e+00,
2.8493e+00, 5.5737e-01],
[-5.4443e-01, -3.6052e+00, -3.8510e+00, ..., 3.0121e+00,
1.7999e+00, 1.5012e+00],
[ 1.9723e-01, -2.5924e-01, 2.4730e+00, ..., 7.9166e-01,
7.8395e-01, 1.0557e+00]]]], device='cuda:0'),
tensor([[[[ 3.3663e-01, 1.8844e+00, -3.0009e+00, ..., 2.3174e+00,
2.3005e+00, 7.3142e-01],
[ 3.7867e-01, 9.6900e-01, -1.7080e+00, ..., 6.0683e+00,
-3.7997e-01, -1.1472e+00],
[-3.0647e-01, -3.8444e-01, 5.2826e-01, ..., 5.6261e-01,
-3.2106e+00, -2.6791e+00],
...,
[-1.8775e+00, -8.8831e+00, -2.9160e+00, ..., -4.7771e+00,
-4.5193e+00, -9.3004e-01],
[-2.3890e+00, -4.7046e+00, -1.0089e+00, ..., -4.1721e+00,
-3.1165e+00, 2.2357e-01],
[-1.6397e+00, -4.5293e+00, -4.6460e+00, ..., -3.9434e+00,
-3.1869e+00, -1.3285e+00]],
[[-1.3821e+00, -1.5821e+00, -8.0039e+00, ..., 3.0657e+00,
-2.3461e+00, -1.5176e+00],
[-4.9554e+00, -5.9591e+00, -5.7531e+00, ..., -1.5237e+00,
-1.0209e+01, -4.7182e+00],
[-4.4424e+00, -4.9892e+00, -3.1878e+00, ..., -1.1527e+01,
-1.3306e+01, -4.5488e+00],
...,
[-5.9143e+00, -1.1545e+01, -5.8286e+00, ..., -9.4513e+00,
-9.3054e+00, -2.1533e+00],
[-9.5420e+00, -1.0991e+01, -9.9722e+00, ..., -1.0024e+01,
-1.0863e+01, -3.8069e+00],
[-2.2660e+00, -4.8295e+00, -5.1681e+00, ..., -4.2128e+00,
-4.0462e+00, -6.1298e-01]],
[[ 8.0790e-02, 1.2314e+00, -1.9476e-01, ..., 1.0224e+00,
3.3308e+00, 4.0368e-01],
[-6.7789e-01, -5.9281e-01, -5.7863e-01, ..., 6.3738e-01,
2.4943e+00, -1.0329e-01],
[ 1.5579e-01, 1.1138e+00, 8.4915e-01, ..., 5.4158e+00,
2.6314e+00, 3.7071e-01],
...,
[-3.6238e-01, 2.4250e-01, -5.6387e+00, ..., -8.0844e-01,
-6.5453e-01, -9.6109e-01],
[-1.3536e+00, -6.0990e-01, -4.1481e+00, ..., 3.5848e-02,
1.0910e+00, -3.2826e-01],
[-1.4196e-01, 4.0637e-01, 7.6663e-02, ..., -2.8909e-01,
-7.5872e-01, -1.9027e+00]],
...,
[[-4.7552e-01, -1.2992e+00, -2.6785e+00, ..., -1.3658e+00,
-3.5948e+00, 1.3000e-01],
[-2.9836e+00, -3.4157e+00, -3.6879e+00, ..., -7.2473e+00,
-5.3167e+00, -3.3813e+00],
[-1.9517e+00, -6.8643e-01, -2.3380e+00, ..., -8.7785e+00,
-5.5224e+00, -2.6651e+00],
...,
[-3.7537e+00, -7.3637e+00, -2.5900e+00, ..., -6.6401e+00,
-7.3431e+00, -3.5129e+00],
[-3.5570e+00, -6.2276e+00, -3.7290e+00, ..., -5.0209e+00,
-5.6714e+00, -1.5335e+00],
[-2.8220e+00, -4.3377e+00, -4.8549e+00, ..., -3.0604e+00,
-2.3917e+00, 4.1001e-01]],
[[-1.1651e+00, -1.8185e+00, -9.3818e-01, ..., -1.4991e-01,
-1.7097e+00, 1.0747e+00],
[-2.9162e+00, -2.3178e+00, -4.2386e+00, ..., -4.2776e+00,
-4.3938e+00, 3.1174e-01],
[-2.8130e+00, -5.8641e-01, -7.1848e+00, ..., -9.5389e+00,
-5.6954e+00, -3.0428e-01],
...,
[-5.5532e+00, -1.2067e+00, 1.3542e+00, ..., -2.8442e+00,
-2.8718e+00, -1.0202e+00],
[-3.2786e+00, -1.1883e-01, 5.1701e+00, ..., 6.7538e-02,
-9.9213e-01, 3.4213e-01],
[-2.6517e+00, -1.7997e+00, -8.5185e-01, ..., -7.1412e-01,
-1.1540e+00, 3.0203e-01]],
[[ 1.3123e+00, 8.9164e-01, 2.1028e+00, ..., 7.5338e-01,
2.2969e+00, 2.6905e-02],
[ 3.0451e+00, 1.9920e+00, 2.4823e+00, ..., 6.0235e+00,
2.0699e+00, 2.9523e+00],
[ 2.1973e+00, 1.5888e-01, 1.5879e+00, ..., 5.2645e+00,
3.0244e+00, 2.4066e+00],
...,
[ 2.7599e+00, 3.3791e-01, 8.3043e-01, ..., 2.2449e+00,
2.4431e+00, 4.7425e-01],
[ 4.8048e+00, 3.3594e+00, 4.5094e+00, ..., 3.0617e+00,
1.0716e+00, 1.4007e+00],
[ 2.8218e+00, 2.1193e+00, 2.6156e+00, ..., 1.0175e+00,
5.9568e-01, 1.0962e+00]]],
[[[ 1.6492e-01, 7.6704e-01, -1.0618e+00, ..., 1.4357e+00,
-9.6553e-01, 1.5582e-01],
[-4.7400e-01, -4.8505e-01, 7.1579e-01, ..., -5.3107e-01,
-1.5170e+00, 1.6012e+00],
[-3.0504e+00, -3.4290e+00, -8.7702e-01, ..., -3.8607e+00,
-2.4925e+00, -2.8619e-01],
...,
[-6.8998e-01, -1.9755e+00, 2.2351e-01, ..., 9.9036e-01,
1.5489e+00, -1.0213e+00],
[-2.0267e+00, -1.7944e+00, -1.2023e+00, ..., 3.7366e+00,
5.9893e-01, -1.1379e+00],
[-1.0919e+00, -3.3126e+00, -3.4015e+00, ..., -1.5431e-01,
-1.7137e+00, -1.3874e+00]],
[[-1.3583e+00, -2.0463e+00, -2.8050e+00, ..., 4.8049e-02,
-2.6695e+00, 3.6992e-01],
[-4.5680e+00, -3.7339e+00, -1.7121e+00, ..., -2.4575e+00,
-2.5820e+00, 1.0450e+00],
[-5.4594e+00, -3.1289e+00, 9.1021e-02, ..., -2.2084e+00,
-4.7351e+00, 2.7123e-01],
...,
[-5.9878e+00, -5.9682e+00, -3.8937e+00, ..., -2.3758e-01,
-2.4181e+00, -2.8180e+00],
[-7.7818e+00, -7.6605e+00, -6.7230e+00, ..., -1.9484e+00,
-5.0421e+00, -4.3206e+00],
[-1.7966e+00, -3.2694e+00, -3.1817e+00, ..., -2.2169e+00,
-3.1162e+00, -7.1896e-01]],
[[-1.0878e-01, 1.1066e+00, 1.4626e+00, ..., -1.6306e-01,
-1.6865e-02, 1.2278e+00],
[-1.7320e-01, -4.4172e-02, 3.4612e-01, ..., -2.4855e+00,
-3.8391e+00, -2.5592e+00],
[ 1.1007e+00, 2.0504e+00, -8.5619e-02, ..., 9.8388e-01,
-1.1582e-01, -2.1108e+00],
...,
[ 2.8573e-01, 1.0670e+00, 1.0103e+00, ..., -1.4289e+00,
3.5180e+00, 1.1255e+00],
[ 5.3497e-01, 4.6944e-01, 5.7180e-01, ..., -9.8360e-01,
2.7938e+00, 7.8860e-01],
[ 6.1413e-01, 1.1062e+00, 1.2707e+00, ..., -1.3308e+00,
1.9841e+00, -8.9431e-01]],
...,
[[-9.7990e-01, -1.8517e+00, -2.1604e+00, ..., 3.7922e-01,
-1.7516e-01, -5.4178e-01],
[-3.6193e+00, -2.8716e+00, -3.7582e-01, ..., -2.4354e+00,
-7.4550e-01, -7.1974e-01],
[-4.5189e+00, -2.3579e+00, 2.2095e-02, ..., -4.6691e+00,
-2.2399e+00, 4.9900e-01],
...,
[-2.0244e+00, -1.5275e+00, -5.4530e-01, ..., -1.2197e+00,
-6.0273e+00, -1.2881e+00],
[-3.3383e+00, -3.0403e+00, -1.8439e+00, ..., -6.2162e-01,
-5.6404e+00, -9.7468e-02],
[-1.4993e+00, -1.9127e+00, -1.6267e+00, ..., -1.0081e-01,
-2.8692e+00, 1.3049e+00]],
[[-1.1576e+00, -2.1166e+00, -3.4464e+00, ..., -1.5316e+00,
-9.7745e-01, -7.2827e-01],
[-3.1971e+00, -1.5019e+00, -3.6471e+00, ..., -4.7657e+00,
-1.5766e+00, -1.3005e+00],
[-4.7518e+00, -2.3589e+00, -3.1669e+00, ..., -1.2058e+01,
-5.9981e+00, -1.2089e+00],
...,
[-4.7614e+00, -3.9118e+00, -3.0712e+00, ..., -2.6747e-01,
-5.2860e+00, -1.7203e+00],
[-4.2601e+00, -2.0301e+00, -1.7192e+00, ..., 2.2972e+00,
-3.7251e+00, -6.9962e-01],
[-3.2802e+00, -3.3176e+00, -1.3744e+00, ..., 1.7108e+00,
-1.7064e+00, 2.7341e-01]],
[[ 1.2196e+00, -1.1648e-01, 5.4433e-01, ..., -4.0276e-01,
-4.5083e-01, 3.0345e-01],
[ 2.9037e+00, 1.1553e+00, 1.1450e+00, ..., -3.5141e+00,
-2.0297e+00, 1.0416e+00],
[ 1.5422e+00, -2.4909e-01, -1.0413e+00, ..., -8.6908e+00,
-5.4687e+00, -1.7785e+00],
...,
[ 2.5825e+00, 1.7528e+00, 1.3670e+00, ..., 5.6194e-01,
2.0113e+00, 3.0815e-01],
[ 2.9658e+00, 3.8909e+00, 2.2181e+00, ..., 2.3015e+00,
3.0035e-01, 6.3852e-01],
[ 1.9550e+00, 1.5558e+00, 1.5202e-01, ..., 2.0463e+00,
5.0588e-02, 6.0148e-01]]],
[[[ 9.5528e-01, 1.7955e+00, 5.0732e-01, ..., -3.6704e+00,
-2.6106e+00, -1.0358e+00],
[ 2.4416e+00, 1.8830e+00, 7.1376e-01, ..., -2.8812e+00,
-2.6345e+00, -1.9805e-01],
[-2.3941e-01, -7.7514e-01, 2.7422e-01, ..., -2.6114e+00,
-3.1296e+00, -1.1240e-01],
...,
[ 3.3369e-01, -1.9177e-01, 4.4133e-01, ..., 8.1633e-01,
-3.1773e+00, 1.3364e-01],
[-8.9240e-02, -4.9280e-02, -7.8001e-01, ..., -3.7550e+00,
-3.4716e+00, 1.6635e-01],
[-3.5387e-01, -3.4885e+00, -3.3272e+00, ..., -4.4652e+00,
-3.5854e+00, -1.7446e+00]],
[[ 3.6167e-01, -1.2144e+00, -2.9173e+00, ..., -7.3298e+00,
-7.4094e+00, -1.3572e+00],
[-8.4097e-01, -4.5851e+00, -6.6228e+00, ..., -7.9711e+00,
-8.4855e+00, -2.3947e+00],
[-1.5673e+00, -1.4988e+00, -1.4590e+00, ..., -8.1671e+00,
-8.3781e+00, -2.6884e+00],
...,
[-1.5865e+00, -2.7386e+00, -2.2100e+00, ..., -8.5634e+00,
-1.0212e+01, -2.2940e+00],
[-4.5956e+00, -5.9124e+00, -5.3101e+00, ..., -1.0614e+01,
-1.0770e+01, -4.6369e+00],
[-3.8781e-01, -3.2921e+00, -3.5596e+00, ..., -4.2417e+00,
-3.6191e+00, -7.9163e-01]],
[[ 4.5729e-01, 2.1308e+00, 1.7667e+00, ..., 8.6244e-01,
2.2665e+00, 9.8449e-01],
[-2.8072e-01, 1.1125e+00, 5.7340e-01, ..., 1.0258e+00,
2.9088e+00, 5.9698e-02],
[ 4.2126e-01, 4.7967e+00, 2.6816e+00, ..., -7.9167e-01,
1.8258e+00, -3.5548e-01],
...,
[ 1.3232e+00, 1.9717e+00, 2.8501e-01, ..., -5.0114e-01,
1.4018e+00, -1.0333e+00],
[ 9.6574e-01, 1.7305e+00, 4.2327e-01, ..., -1.1855e+00,
6.7798e-01, -1.8671e+00],
[-2.7616e-01, 1.9103e+00, 9.9937e-01, ..., -3.6415e-01,
-1.0816e+00, -2.0823e+00]],
...,
[[-1.9291e-01, -7.9936e-01, -1.5915e+00, ..., -5.4818e+00,
-5.4730e+00, -7.3340e-01],
[-2.2569e+00, -4.1603e+00, -2.5232e+00, ..., -4.5933e+00,
-5.3789e+00, -1.8060e+00],
[-1.2209e+00, -2.5677e+00, -8.4085e-01, ..., -5.9116e+00,
-6.2555e+00, -1.5329e+00],
...,
[-6.4018e-01, -1.2877e+00, 3.1486e-01, ..., -4.8655e+00,
-6.8800e+00, -8.8268e-01],
[-2.0673e+00, -3.5451e+00, -2.0309e+00, ..., -2.9778e+00,
-5.1321e+00, -9.5569e-01],
[-1.1812e+00, -3.6132e+00, -2.4715e+00, ..., -1.1784e+00,
-2.2599e+00, 1.9102e-01]],
[[-1.0300e+00, -7.6826e-01, -9.6640e-01, ..., -2.2367e+00,
-4.3317e-01, -5.5451e-01],
[-2.4771e+00, -2.9072e+00, -1.6792e+00, ..., -3.9419e+00,
-1.7706e+00, -2.1620e-01],
[-2.6411e+00, -4.2937e+00, 4.2915e-01, ..., -2.4341e+00,
-1.9444e+00, -4.1326e-01],
...,
[-1.7666e+00, -2.9550e+00, -2.2259e+00, ..., -4.3886e+00,
-2.9687e+00, -6.4892e-01],
[-3.1852e+00, -1.9914e+00, -9.0463e-01, ..., -1.0481e+00,
-8.1368e-01, 1.1524e+00],
[-3.3260e+00, -3.4653e+00, -1.4329e+00, ..., -1.5981e-01,
-9.7346e-01, 5.8966e-01]],
[[ 8.8506e-01, 1.0365e+00, 1.0924e+00, ..., -2.6321e-01,
5.7620e-01, -1.4438e+00],
[ 1.6715e+00, 2.9773e+00, 2.6968e+00, ..., 3.2127e+00,
2.9055e+00, 2.2187e+00],
[ 4.0759e-03, -4.5340e-01, 5.1714e-01, ..., 2.4870e+00,
1.5540e+00, 1.3191e+00],
...,
[ 1.9905e+00, 1.0850e+00, 9.5284e-01, ..., 3.3640e+00,
1.8769e+00, 8.2318e-01],
[ 3.3135e+00, 3.7398e+00, 2.4125e+00, ..., 2.6355e-01,
6.5314e-01, 2.1689e+00],
[ 1.9589e+00, 4.3985e-01, 4.4101e-01, ..., 1.3496e+00,
6.4247e-01, 8.4156e-01]]],
[[[ 3.0786e-01, 1.4981e+00, 5.9975e-01, ..., -4.0602e-01,
7.9239e-02, 6.9499e-01],
[ 3.6703e-01, 1.8883e-01, 3.6675e-01, ..., -9.2717e-01,
-1.1389e+00, 1.0491e+00],
[-5.1462e-01, 6.0359e-02, 7.5486e-01, ..., -1.3655e+00,
-1.4496e+00, 8.9121e-01],
...,
[-1.0094e+00, -2.1247e+00, 2.4600e-01, ..., -9.7498e-01,
-1.9240e-01, -9.0413e-01],
[-1.0630e+00, -1.0554e+00, -3.8195e-01, ..., -3.1053e-01,
1.7104e-03, -1.7101e+00],
[-2.7163e+00, -5.1534e+00, -4.4634e+00, ..., -4.5843e+00,
-4.1761e+00, -3.9307e+00]],
[[-7.4897e-01, -2.2219e+00, -4.2816e+00, ..., -3.2955e+00,
-4.9863e+00, -3.7570e-01],
[-4.6835e+00, -5.9313e+00, -5.5737e+00, ..., -5.1526e+00,
-7.3330e+00, -1.6843e+00],
[-3.5068e+00, -4.0372e+00, -3.9448e+00, ..., -5.9014e+00,
-6.5687e+00, -1.4432e+00],
...,
[-5.8650e+00, -5.8405e+00, -4.1370e+00, ..., -5.0854e+00,
-5.1775e+00, -3.9989e+00],
[-7.0544e+00, -7.7547e+00, -6.1830e+00, ..., -7.3592e+00,
-8.0622e+00, -6.6701e+00],
[-2.5248e+00, -4.4406e+00, -4.0346e+00, ..., -3.8655e+00,
-4.1954e+00, -2.6141e+00]],
[[-3.0451e-01, 2.1127e+00, 1.2818e+00, ..., 1.0153e+00,
2.1468e+00, 9.3810e-01],
[-1.4969e-01, 1.0258e+00, 9.2251e-01, ..., 5.4671e-01,
1.3764e+00, -4.0143e-01],
[-3.1838e-02, 1.9968e+00, 1.6804e+00, ..., 2.0189e-01,
1.8183e+00, 2.1511e-01],
...,
[ 9.3385e-01, 1.1927e+00, 1.1661e+00, ..., 3.6394e-01,
9.5636e-01, 5.5239e-01],
[ 8.7205e-01, 7.6787e-01, 9.1710e-01, ..., 9.0829e-01,
4.3680e-01, 3.7668e-01],
[ 1.3247e+00, 1.5825e+00, 1.3496e+00, ..., 1.5603e+00,
1.0996e+00, 1.3752e-01]],
...,
[[-8.1948e-01, -1.8023e+00, -1.9672e+00, ..., -2.4738e+00,
-2.9060e+00, -4.4314e-01],
[-3.1826e+00, -4.2740e+00, -2.9398e+00, ..., -3.5775e+00,
-3.9904e+00, -2.6182e+00],
[-1.4237e+00, -2.0086e+00, -1.2964e+00, ..., -2.7813e+00,
-3.8358e+00, -1.3291e+00],
...,
[-2.8981e+00, -2.8680e+00, -8.8507e-01, ..., -2.3605e+00,
-3.9107e+00, -1.9242e+00],
[-3.0497e+00, -3.0991e+00, -2.0697e+00, ..., -2.1293e+00,
-4.1375e+00, -2.3805e+00],
[-3.5216e+00, -3.8056e+00, -3.3082e+00, ..., -2.9830e+00,
-3.7488e+00, -1.7243e+00]],
[[-1.6803e+00, -2.2317e+00, -3.0157e+00, ..., -1.3804e+00,
-1.3670e+00, -1.3817e+00],
[-4.3424e+00, -3.5156e+00, -3.0521e+00, ..., -3.1781e+00,
-1.7823e+00, -8.3634e-01],
[-3.3082e+00, -3.0693e+00, -3.0078e+00, ..., -1.8831e+00,
-1.3147e+00, -2.5431e-01],
...,
[-4.2033e+00, -2.4898e+00, -3.0849e+00, ..., -1.4945e+00,
-3.1590e+00, -1.8081e+00],
[-3.7018e+00, -9.4414e-01, -2.3350e+00, ..., -8.2203e-01,
-3.0439e+00, -1.3774e+00],
[-4.3529e+00, -3.0654e+00, -3.0233e+00, ..., -2.4194e+00,
-3.4161e+00, -1.2817e+00]],
[[ 1.0927e+00, 5.8971e-01, 7.6566e-01, ..., -4.9272e-01,
1.0639e-01, -4.8783e-01],
[ 3.4167e+00, 3.1906e+00, 2.2771e+00, ..., 1.8126e+00,
2.1181e+00, 2.4847e+00],
[ 1.4126e+00, 6.3456e-01, 5.9644e-01, ..., 1.1312e+00,
1.9920e+00, 1.9683e+00],
...,
[ 2.8688e+00, 1.0243e+00, 1.7412e+00, ..., 2.4189e+00,
2.3123e+00, 1.3197e+00],
[ 3.6880e+00, 3.5165e+00, 2.4068e+00, ..., 2.6385e+00,
2.3058e+00, 2.0939e+00],
[ 1.9202e+00, 7.4414e-01, 2.5199e-02, ..., -1.4994e-01,
-1.5947e-01, -9.9503e-02]]],
[[[ 4.3590e-01, 1.7245e-01, -4.8663e-01, ..., 6.7873e-02,
3.8409e-01, -2.3927e-01],
[ 8.4202e-01, 2.0822e-01, 1.1733e+00, ..., 4.2982e-01,
4.5031e-01, -5.5540e-01],
[-4.8018e-01, -4.7290e-01, 1.5671e+00, ..., -1.2297e-01,
-4.6356e-01, -8.4507e-01],
...,
[ 4.7618e-01, -1.5103e+00, -1.8371e-02, ..., 7.1379e-02,
-8.2858e-02, -3.6111e-01],
[-1.7180e+00, -1.9098e+00, -2.8903e-01, ..., 4.5116e-01,
-2.1836e-01, -1.2533e+00],
[-1.4067e+00, -4.1162e+00, -3.2487e+00, ..., -2.4001e+00,
-2.8907e+00, -2.7006e+00]],
[[-1.6374e+00, -3.8621e+00, -6.2201e+00, ..., -2.5290e+00,
-3.6769e+00, -2.2921e+00],
[-4.8376e+00, -5.7279e+00, -5.2877e+00, ..., -3.2848e+00,
-3.9354e+00, -3.5374e+00],
[-3.9377e+00, -4.8742e+00, -4.8059e+00, ..., -2.6598e+00,
-3.2233e+00, -2.8527e+00],
...,
[-5.5977e+00, -7.9419e+00, -8.1205e+00, ..., -2.5078e+00,
-3.3987e+00, -2.6594e+00],
[-8.5657e+00, -9.9163e+00, -1.0043e+01, ..., -3.7777e+00,
-5.0361e+00, -5.0541e+00],
[-1.8699e+00, -4.6965e+00, -4.5274e+00, ..., -2.8595e+00,
-2.6857e+00, -1.5226e+00]],
[[ 7.8685e-01, 3.0031e+00, 1.9685e+00, ..., 1.8638e+00,
1.4928e+00, 9.1010e-01],
[-3.5085e-03, 1.4049e+00, 1.4069e+00, ..., 1.1712e+00,
1.1378e+00, 8.6825e-01],
[ 8.9840e-01, 1.9138e+00, 8.7245e-01, ..., 1.2792e+00,
1.7975e+00, 1.4056e+00],
...,
[-2.0921e-01, 1.2529e-01, 1.0979e+00, ..., 7.4910e-01,
1.0219e+00, 6.5697e-01],
[ 6.4362e-01, 1.8551e+00, 4.5193e-01, ..., 8.1548e-01,
1.1451e+00, 3.8876e-01],
[ 5.2047e-01, 1.7441e+00, 9.7442e-01, ..., 1.1062e+00,
1.7044e+00, 5.2930e-01]],
...,
[[-1.0878e+00, -3.3049e+00, -2.3638e+00, ..., -1.5128e+00,
-2.1323e+00, -5.0784e-01],
[-2.7380e+00, -3.7240e+00, -1.3526e+00, ..., -1.7398e+00,
-4.0596e+00, -2.9008e+00],
[-2.1880e+00, -2.5952e+00, -1.6070e+00, ..., -1.3725e-01,
-3.7395e+00, -1.7580e+00],
...,
[-1.9105e+00, -3.6364e+00, -1.5847e+00, ..., -7.7459e-01,
-2.7505e+00, -9.3986e-01],
[-3.3501e+00, -4.7189e+00, -2.2609e+00, ..., -7.5301e-01,
-3.1219e+00, -1.6138e+00],
[-2.3472e+00, -3.5189e+00, -1.6103e+00, ..., -1.6846e+00,
-2.3950e+00, -5.2279e-01]],
[[-2.5131e+00, -3.0558e+00, -1.4811e+00, ..., 5.5039e-01,
-1.3578e+00, -6.5812e-01],
[-4.1497e+00, -3.1539e+00, -2.1750e+00, ..., -1.2831e-02,
-3.5818e+00, -1.8019e+00],
[-3.8488e+00, -1.9821e+00, -1.6824e+00, ..., -2.0369e-01,
-3.9583e+00, -2.4115e+00],
...,
[-3.5049e+00, -3.5032e+00, -2.0261e+00, ..., -2.7013e-01,
-3.4451e+00, -1.9588e+00],
[-4.3910e+00, -1.1672e+00, -3.2725e-01, ..., -1.6829e-01,
-3.5519e+00, -1.9325e+00],
[-4.0810e+00, -2.8831e+00, -1.3567e-01, ..., -4.8520e-01,
-2.6171e+00, -1.2262e+00]],
[[ 1.5436e+00, 9.8498e-01, 1.5835e+00, ..., 5.5271e-01,
1.1966e+00, 6.3411e-01],
[ 3.1084e+00, 2.9069e+00, 3.1964e+00, ..., 1.4392e+00,
9.9114e-01, 2.4409e+00],
[ 1.4386e+00, 9.5116e-01, 2.0758e+00, ..., 8.3659e-01,
-4.3795e-02, 8.6673e-01],
...,
[ 3.8627e+00, 2.4452e+00, 2.2861e+00, ..., 1.3472e+00,
2.0993e-01, 6.1715e-01],
[ 4.1746e+00, 4.4345e+00, 4.1361e+00, ..., 1.6815e+00,
6.3461e-01, 1.3140e+00],
[ 2.5439e+00, 1.9532e+00, 1.8789e+00, ..., 6.9807e-01,
-3.0982e-01, -1.1336e-01]]]], device='cuda:0'),
tensor([[[[ 3.5181e-01, 1.5093e+00, 5.9738e-01, ..., -4.9522e-01,
1.4245e-03, 5.7281e-01],
[ 3.5370e-01, 4.0331e-02, -7.8230e-02, ..., -7.0407e-01,
-5.9301e-01, 3.6593e-01],
[-4.3573e-01, 9.3763e-01, 1.4926e+00, ..., -6.0883e+00,
-2.8306e+00, -1.4332e-01],
...,
[-8.2089e-01, -1.7450e+00, -4.0280e-02, ..., -1.2651e+00,
-4.8488e-01, -6.2993e-01],
[-1.0167e+00, -9.4699e-01, -3.0143e-01, ..., -2.3466e-01,
-7.0331e-02, -1.6137e+00],
[-2.6514e+00, -5.1254e+00, -4.4575e+00, ..., -4.6151e+00,
-4.1232e+00, -3.8428e+00]],
[[-6.8656e-01, -2.1833e+00, -4.2506e+00, ..., -3.4411e+00,
-5.1022e+00, -4.5630e-01],
[-4.2157e+00, -5.7759e+00, -6.7780e+00, ..., -1.0784e+01,
-1.0461e+01, -3.5609e+00],
[-3.6303e+00, -5.6814e+00, -7.2175e+00, ..., -6.0612e+00,
-6.5026e+00, -1.6097e+00],
...,
[-5.8164e+00, -5.7839e+00, -3.8763e+00, ..., -5.5944e+00,
-5.6023e+00, -4.0980e+00],
[-6.9810e+00, -7.7962e+00, -6.3171e+00, ..., -7.2262e+00,
-8.2732e+00, -6.5681e+00],
[-2.4636e+00, -4.3581e+00, -4.0281e+00, ..., -3.7003e+00,
-4.0900e+00, -2.5890e+00]],
[[-2.9413e-01, 2.1502e+00, 1.2321e+00, ..., 9.7747e-01,
2.2800e+00, 9.7672e-01],
[-3.4065e-01, 8.9335e-01, 1.0154e+00, ..., 6.5157e-01,
9.7179e-01, -5.9830e-01],
[ 1.5614e-01, 2.5130e+00, 1.6604e+00, ..., -3.2691e-02,
9.1741e-01, -1.6850e-02],
...,
[ 9.5082e-01, 1.3530e+00, 9.8269e-01, ..., 4.5709e-01,
4.1717e-01, -9.0452e-02],
[ 8.1517e-01, 8.2900e-01, 8.0540e-01, ..., 7.6117e-01,
1.0397e-01, 1.6215e-01],
[ 1.2955e+00, 1.6529e+00, 1.3398e+00, ..., 1.4641e+00,
1.1567e+00, 1.1146e-01]],
...,
[[-7.5860e-01, -1.7870e+00, -1.8717e+00, ..., -2.6176e+00,
-2.9727e+00, -5.4629e-01],
[-3.0835e+00, -4.7585e+00, -3.4315e+00, ..., -5.2745e+00,
-4.5831e+00, -3.1942e+00],
[-1.2486e+00, -2.6876e+00, -2.2936e+00, ..., -1.0953e+00,
-1.4835e+00, -1.2441e+00],
...,
[-2.8846e+00, -2.7491e+00, -5.6320e-01, ..., -3.8073e+00,
-4.9910e+00, -2.6291e+00],
[-3.1196e+00, -3.2147e+00, -1.9954e+00, ..., -2.0315e+00,
-4.2449e+00, -2.4155e+00],
[-3.4187e+00, -3.7487e+00, -3.2018e+00, ..., -2.7943e+00,
-3.5816e+00, -1.5961e+00]],
[[-1.6422e+00, -2.2334e+00, -2.8619e+00, ..., -1.6382e+00,
-1.5688e+00, -1.3408e+00],
[-4.1814e+00, -3.4042e+00, -2.2584e+00, ..., -1.5861e+00,
-1.4531e+00, -1.5114e+00],
[-3.6030e+00, -3.1883e+00, -2.8242e+00, ..., -3.1437e+00,
-1.2235e+00, 3.7341e-01],
...,
[-3.8968e+00, -2.5778e+00, -3.0421e+00, ..., -1.5395e+00,
-2.3182e+00, -7.0969e-01],
[-3.6123e+00, -9.8853e-01, -2.5087e+00, ..., -1.4064e+00,
-2.8498e+00, -1.2331e+00],
[-4.3029e+00, -3.0604e+00, -2.9099e+00, ..., -2.3991e+00,
-3.3209e+00, -1.1788e+00]],
[[ 1.0445e+00, 5.9398e-01, 7.0365e-01, ..., -5.7201e-01,
9.1561e-02, -4.9537e-01],
[ 3.2160e+00, 3.1465e+00, 2.7252e+00, ..., 3.8939e+00,
2.9782e+00, 2.4478e+00],
[ 1.1400e+00, 1.2318e+00, 2.4606e+00, ..., -2.0507e+00,
1.8764e-02, 4.7372e-01],
...,
[ 2.8747e+00, 1.2303e+00, 8.6359e-01, ..., 1.8589e+00,
1.3566e+00, 6.6955e-01],
[ 3.7257e+00, 3.6971e+00, 2.4847e+00, ..., 2.5984e+00,
2.4250e+00, 2.1103e+00],
[ 1.8819e+00, 6.5283e-01, 1.9123e-02, ..., -4.6586e-01,
-3.2569e-01, -1.2561e-01]]],
[[[ 1.1330e+00, 1.5829e+00, 9.6943e-01, ..., -3.8324e+00,
-2.3738e+00, -8.5040e-01],
[ 5.7147e-01, -1.4382e+00, -1.7740e+00, ..., -2.9523e+00,
-2.3572e+00, -1.0136e-01],
[-1.2869e+00, -1.3513e+00, -1.7987e+00, ..., -2.5347e+00,
-2.3221e+00, 2.6193e-01],
...,
[-3.8026e-01, -2.5591e+00, -7.4955e-01, ..., 2.4925e+00,
4.9177e+00, 4.0794e-01],
[-8.9174e-01, -1.5682e+00, -7.1600e-01, ..., 1.4356e+00,
-2.1692e+00, -1.3317e+00],
[-6.3770e-01, -3.8225e+00, -4.1097e+00, ..., -4.1830e+00,
-3.3252e+00, -1.3254e+00]],
[[ 2.9968e-01, -1.0834e+00, -2.3375e+00, ..., -7.0996e+00,
-6.8687e+00, -1.2048e+00],
[-3.6021e+00, -4.2940e+00, -4.1278e+00, ..., -7.6476e+00,
-7.9918e+00, -2.3827e+00],
[-1.5218e+00, -1.0335e+00, -1.8382e+00, ..., -6.6558e+00,
-7.0864e+00, -2.1760e+00],
...,
[-5.5185e+00, -1.1422e+01, -6.3505e+00, ..., 8.8913e-01,
-1.5137e+00, -2.6943e+00],
[-4.1509e+00, -5.2551e+00, -5.7645e+00, ..., -1.0301e+01,
-1.3899e+01, -6.0505e+00],
[-1.1117e+00, -3.9296e+00, -3.6599e+00, ..., -2.7983e+00,
-3.5379e+00, -6.7906e-01]],
[[ 7.0868e-02, 1.5770e+00, 1.9683e+00, ..., 7.3921e-01,
2.3234e+00, 1.0466e+00],
[-6.6987e-01, 1.0579e+00, 1.7381e+00, ..., 9.1364e-01,
2.7908e+00, 2.3709e-01],
[ 1.0893e+00, 2.6731e+00, 1.5884e+00, ..., -1.3443e+00,
1.7895e+00, -1.0789e-01],
...,
[ 6.7813e-01, -2.1085e-01, -1.6186e+00, ..., 7.3945e-01,
-4.2189e-01, 2.2715e+00],
[ 1.3689e+00, 2.5897e+00, -9.8100e-02, ..., -3.6043e+00,
-3.1767e-03, 9.3023e-01],
[ 4.7619e-01, 1.1623e+00, -6.7842e-04, ..., -1.2265e+00,
1.0734e+00, -1.9659e+00]],
...,
[[-8.2534e-02, -1.1125e+00, -1.5101e+00, ..., -5.4130e+00,
-5.2573e+00, -7.5909e-01],
[-2.9605e+00, -4.2070e+00, -2.9584e+00, ..., -4.3030e+00,
-5.0854e+00, -1.8436e+00],
[-1.0793e+00, -6.7087e-01, -2.9816e-01, ..., -4.8008e+00,
-5.4268e+00, -1.0398e+00],
...,
[-5.4437e+00, -7.3707e+00, -3.4916e+00, ..., -2.0846e+00,
-5.7179e+00, -4.0709e+00],
[-1.6997e+00, -1.6447e+00, -3.3014e+00, ..., -2.4759e+00,
-6.7424e+00, -1.6661e+00],
[-1.4040e+00, -3.1776e+00, -1.9386e+00, ..., -9.0050e-01,
-2.1492e+00, 3.6124e-01]],
[[-8.9908e-01, -5.7027e-01, -9.4169e-01, ..., -1.9910e+00,
-4.9157e-01, -5.8383e-01],
[-3.0464e+00, -3.3206e+00, -4.1584e+00, ..., -3.6568e+00,
-1.8074e+00, -3.1998e-01],
[-1.4101e+00, -1.4301e+00, -2.1748e+00, ..., -1.5990e+00,
-1.3451e+00, -2.7179e-01],
...,
[-1.6140e+00, -3.1228e+00, -3.2702e+00, ..., 6.5725e-01,
-2.8449e+00, -3.4050e+00],
[-3.0143e+00, -5.2048e-01, -5.5285e-01, ..., 3.1818e-01,
-2.8474e+00, -1.2841e+00],
[-3.8870e+00, -3.5660e+00, -1.5225e+00, ..., 1.8714e+00,
-1.9596e-01, 4.6514e-01]],
[[ 9.3473e-01, 8.9708e-01, 1.1516e+00, ..., -3.8619e-01,
6.2512e-01, -1.2483e+00],
[ 2.1975e+00, 1.6639e+00, 1.0089e+00, ..., 3.0541e+00,
3.1204e+00, 2.3969e+00],
[ 3.3852e-01, -1.8021e-01, -7.3440e-01, ..., 1.9954e+00,
1.7203e+00, 1.3233e+00],
...,
[ 5.6259e+00, 5.9987e+00, 2.9762e+00, ..., -2.1037e+00,
2.9494e+00, 2.0423e-01],
[ 2.7526e+00, 2.0535e+00, 1.0228e+00, ..., 4.3759e+00,
3.9709e+00, 2.2332e+00],
[ 2.2463e+00, 1.1330e+00, 9.1468e-01, ..., 6.5300e-02,
-7.1701e-01, 1.3961e+00]]],
[[[ 2.3641e-01, 1.4356e+00, 5.0029e-01, ..., -1.9398e+00,
2.3869e-01, 9.6406e-01],
[-3.1292e-02, -7.8298e-02, 1.6845e+00, ..., -5.2039e+00,
-5.5492e+00, -7.0313e-01],
[-3.0511e-01, -4.5754e-01, 1.0758e+00, ..., -4.1331e+00,
-4.0104e+00, 2.4411e+00],
...,
[-7.4013e-01, -9.7245e-01, 5.5341e-01, ..., -4.0034e+00,
-3.8147e+00, -1.0913e+00],
[-2.3884e+00, -1.7875e+00, 4.8204e-01, ..., -2.9906e+00,
-3.6691e+00, -1.5466e-01],
[-2.3318e+00, -4.0395e+00, -3.2565e+00, ..., -4.4406e+00,
-3.2740e+00, -1.4394e+00]],
[[-1.5618e+00, -1.7032e+00, -1.9569e+00, ..., 5.8321e-01,
1.4955e+00, 9.3670e-01],
[-5.0018e+00, -6.1314e+00, -4.3821e+00, ..., -3.3348e+00,
-5.5935e+00, 8.5997e-01],
[-4.3368e+00, -4.6516e+00, -2.4609e+00, ..., -1.4091e+00,
-4.3135e+00, 1.5797e+00],
...,
[-6.9896e+00, -9.0907e+00, -7.5718e+00, ..., -1.1809e+01,
-1.0271e+01, -2.2514e+00],
[-9.7719e+00, -1.0056e+01, -8.1041e+00, ..., -1.0740e+01,
-1.0795e+01, -3.7922e+00],
[-2.5653e+00, -5.0130e+00, -4.2157e+00, ..., -4.0644e+00,
-3.9675e+00, -7.2859e-01]],
[[-1.4153e-01, 1.0561e+00, 4.3904e-01, ..., 7.3640e-01,
-1.1773e-02, 3.3070e-01],
[ 1.7628e-02, 3.4854e-03, -3.3709e-01, ..., 2.5442e+00,
1.3327e+00, -1.2781e+00],
[ 6.4543e-01, 2.1158e+00, 1.2163e+00, ..., 2.1227e+00,
2.5067e+00, 1.2437e+00],
...,
[ 2.9444e-01, 7.2976e-01, 4.3379e-01, ..., 9.9226e-01,
-2.4599e-02, -1.0533e+00],
[-5.7736e-01, 8.9692e-01, -1.5072e-01, ..., -1.3669e-01,
-1.5638e-01, -5.7949e-01],
[ 4.2637e-01, 1.2260e+00, 1.0968e+00, ..., -6.6166e-01,
-3.4202e-01, -1.7193e+00]],
...,
[[-9.2672e-01, -7.8106e-01, -1.7739e+00, ..., -2.2068e-01,
1.0970e-01, 3.0607e-01],
[-3.8772e+00, -4.3381e+00, -3.2293e+00, ..., -4.4737e+00,
-4.5049e+00, -2.0997e+00],
[-2.2305e+00, -1.4056e+00, -8.7934e-01, ..., -2.2570e+00,
-3.0333e+00, -1.6011e+00],
...,
[-2.6466e+00, -3.3050e+00, -5.9547e-01, ..., -6.0901e+00,
-6.7927e+00, -2.9844e+00],
[-3.6754e+00, -5.7883e+00, -2.3105e+00, ..., -3.3063e+00,
-5.2281e+00, -1.7553e+00],
[-3.2818e+00, -3.7666e+00, -3.1173e+00, ..., -2.9568e+00,
-2.7563e+00, 3.0854e-01]],
[[-1.0200e+00, -2.1201e-01, -2.3804e+00, ..., -1.8345e+00,
-2.2939e-01, -8.2167e-01],
[-3.7856e+00, -8.6403e-01, -3.7027e+00, ..., -6.5424e+00,
-3.6869e+00, -2.4452e+00],
[-3.4546e+00, -8.9991e-01, -2.5007e+00, ..., -9.4302e+00,
-6.2817e+00, -1.2934e+00],
...,
[-3.7255e+00, -2.0966e+00, -4.3431e+00, ..., -3.8961e+00,
-2.3623e+00, -1.4565e-01],
[-3.3996e+00, -8.1119e-01, -1.8732e+00, ..., -2.1491e+00,
-1.5288e+00, 4.1435e-01],
[-3.7374e+00, -3.0825e+00, -2.9148e+00, ..., -7.8116e-01,
-9.1266e-01, 4.4369e-01]],
[[ 1.6473e+00, 5.6270e-01, 7.8641e-01, ..., -2.3755e+00,
-9.7771e-01, 1.2616e+00],
[ 3.1110e+00, 1.9702e+00, 1.6141e+00, ..., -9.5329e+00,
-7.6532e+00, -1.4745e+00],
[ 2.1686e+00, 3.1620e-01, -1.6685e-01, ..., -8.8185e+00,
-8.5545e+00, 1.2462e-01],
...,
[ 2.9719e+00, 2.9600e+00, 1.7296e+00, ..., 1.6821e+00,
1.4088e+00, 4.4342e-01],
[ 4.8841e+00, 4.9212e+00, 3.9526e+00, ..., 2.8933e+00,
2.7031e+00, 1.8167e+00],
[ 2.8561e+00, 2.7518e+00, 2.1525e+00, ..., 3.9437e-01,
2.9615e-01, 8.9906e-01]]],
[[[-1.2030e-01, 8.3247e-01, 8.5353e-02, ..., -7.8114e-01,
-2.1872e+00, -8.7578e-01],
[ 2.3262e-01, 1.1426e-01, 1.0482e+00, ..., -3.2729e+00,
-3.1248e+00, -1.3407e+00],
[-8.0225e-01, -1.7127e+00, -5.9105e-01, ..., -5.2208e-01,
2.2782e+00, 8.9569e-01],
...,
[-1.3827e-01, -1.1323e+01, -1.8067e+00, ..., -1.5859e+00,
-1.5961e+00, 3.5405e+00],
[-1.7827e+00, -8.3977e+00, -4.8873e-01, ..., -2.0845e+00,
-3.5845e+00, -4.1319e-01],
[-9.9768e-01, -5.6643e+00, -2.5131e+00, ..., -4.3534e+00,
-3.3190e+00, -1.9073e+00]],
[[-1.4297e+00, -1.8162e+00, -2.5231e+00, ..., -4.3272e+00,
-5.9402e+00, -9.9369e-01],
[-4.1558e+00, -4.1737e+00, -2.1396e+00, ..., -6.9347e+00,
-6.4472e+00, -1.4932e+00],
[-4.4434e+00, -4.7068e+00, -2.4158e+00, ..., -1.8330e+00,
-2.2575e-01, 8.1262e-01],
...,
[-4.5960e+00, -1.2921e+01, -4.6486e-01, ..., 2.4300e+00,
3.1461e-01, 4.6213e-01],
[-6.1828e+00, -1.0446e+01, -3.8801e+00, ..., -1.6313e+01,
-1.1576e+01, -6.8070e+00],
[-1.0909e+00, -3.1108e+00, -4.2654e+00, ..., -2.9913e+00,
-2.6086e+00, -4.4801e-01]],
[[ 1.2962e-01, 1.4122e+00, 1.6611e+00, ..., 2.8301e-01,
2.3784e+00, 8.3634e-01],
[-1.0225e-01, -5.5036e-02, 4.2725e-01, ..., 4.0483e-01,
3.0231e+00, 1.3059e+00],
[ 1.4753e-01, 1.2097e+00, 1.6766e+00, ..., -1.7652e+00,
-5.5222e-01, -9.0729e-01],
...,
[-1.1811e+00, 1.0098e+00, -5.0716e+00, ..., -9.8832e-01,
2.4807e+00, 2.7569e-01],
[-8.3465e-01, -1.0965e-01, -5.9598e+00, ..., -1.4040e+00,
3.7227e-01, 1.0078e-01],
[-6.3872e-01, 5.8282e-01, -1.1588e+00, ..., -6.2966e-01,
-4.3803e-01, -1.0195e+00]],
...,
[[-1.1194e+00, -1.7942e+00, -2.0268e+00, ..., -3.0078e+00,
-4.8049e+00, -1.4341e+00],
[-3.3431e+00, -3.1180e+00, -1.2862e+00, ..., -3.5443e+00,
-3.4106e+00, -1.9039e+00],
[-3.3678e+00, -2.3102e+00, -9.5227e-01, ..., 5.9635e-01,
6.9602e-01, 1.4801e+00],
...,
[-2.3487e+00, -4.7832e+00, 1.9653e+00, ..., -1.3287e+00,
-9.9283e-01, -1.0631e+00],
[-2.0088e+00, -3.7072e+00, 1.5226e+00, ..., -8.1934e+00,
-5.9273e+00, -1.8918e+00],
[-5.7248e-01, -2.5374e+00, -2.8588e+00, ..., 6.0183e-01,
-1.9272e+00, 2.8343e-01]],
[[-1.7448e+00, -2.2109e+00, -4.0545e+00, ..., -2.1081e+00,
-7.8275e-01, 5.3476e-01],
[-3.8736e+00, -2.9735e+00, -3.6226e+00, ..., -5.4677e+00,
-3.6211e+00, -6.5042e-01],
[-4.6034e+00, -2.5815e+00, -2.1173e+00, ..., -3.2269e+00,
-1.6412e+00, 1.2115e+00],
...,
[-6.5567e+00, -3.1556e+00, 1.7235e+00, ..., -4.2357e+00,
-4.6076e+00, -1.4387e+00],
[-4.4023e+00, -1.5153e+00, 4.9117e+00, ..., -2.6209e+00,
-6.2225e+00, 3.0798e-01],
[-2.9041e+00, -3.6732e+00, 4.7270e-01, ..., 2.2592e+00,
1.1384e-02, 1.1960e+00]],
[[ 1.2709e+00, -2.2626e-01, 9.7381e-01, ..., -6.0726e-01,
3.2529e-01, -4.9228e-01],
[ 2.8329e+00, 1.2369e+00, 5.8815e-01, ..., 1.1000e+00,
1.8884e+00, 2.8155e+00],
[ 1.7702e+00, 1.6458e-02, -6.6278e-01, ..., -7.6344e-01,
7.1314e-01, 6.8736e-01],
...,
[ 2.5267e+00, 7.8907e-01, 5.4470e-01, ..., -7.1810e+00,
-7.3214e+00, -7.6886e-01],
[ 2.4490e+00, 1.9718e+00, 5.4135e+00, ..., 3.5259e+00,
2.0472e+00, 2.7993e+00],
[ 1.1267e+00, 1.9242e-02, 3.7050e+00, ..., -1.0226e+00,
-1.3234e+00, 8.3810e-01]]],
[[[-9.4003e-02, 1.6677e+00, -2.0946e+00, ..., 1.8747e+00,
2.9731e-01, -6.6414e-01],
[ 1.1425e-01, 3.4872e-01, -3.9484e+00, ..., 1.3256e+00,
3.3896e-01, -1.4804e+00],
[-1.0049e+00, -1.1597e+00, -4.7832e+00, ..., -5.3477e-01,
8.7040e-02, -1.9570e+00],
...,
[-5.6490e-01, 5.1883e-01, 8.9955e-01, ..., -4.3297e+00,
3.0816e+00, 7.0679e-01],
[-1.6465e+00, -4.8761e-02, 3.4729e-02, ..., -2.3196e+00,
2.1061e+00, -1.4323e+00],
[-1.2549e+00, -2.3658e+00, -1.8754e+00, ..., -4.4314e-01,
3.1173e-01, -3.6601e-01]],
[[-1.6280e+00, -2.3333e+00, -6.2127e+00, ..., 3.4140e-02,
-3.7042e+00, -2.6888e+00],
[-4.9052e+00, -5.2876e+00, -6.1603e+00, ..., -3.6922e-01,
-6.5039e+00, -3.7532e+00],
[-4.2006e+00, -5.2534e+00, -7.5545e+00, ..., 3.5633e-01,
-5.8540e+00, -2.8274e+00],
...,
[-4.5974e+00, -5.3873e+00, -4.9722e+00, ..., -2.9627e+00,
2.1531e+00, -4.7031e-01],
[-7.6719e+00, -7.6837e+00, -6.2766e+00, ..., -4.3738e-01,
1.0959e-01, -2.9002e+00],
[-1.5018e+00, -3.2198e+00, -2.0544e+00, ..., 7.6390e-01,
-2.0371e-01, -5.1652e-01]],
[[-2.3132e-01, 2.3401e+00, 1.6212e+00, ..., 1.2795e+00,
1.0048e+00, 3.0731e-01],
[-5.3411e-01, 1.1177e-01, 2.2298e+00, ..., 2.5664e+00,
5.3490e-01, 4.7391e-01],
[ 1.9598e-01, 2.0420e+00, 1.7097e+00, ..., 3.4240e+00,
1.4418e+00, 1.0685e+00],
...,
[-7.4259e-01, 2.6763e-01, -5.3325e-01, ..., 6.4671e-02,
-3.9114e+00, -2.4640e+00],
[-1.2813e+00, 7.9285e-03, 5.7597e-01, ..., -3.4734e+00,
-5.1787e+00, -3.0621e+00],
[-3.5561e-01, -4.0239e-01, -3.3439e-02, ..., -5.9836e+00,
-3.2049e+00, -1.6376e+00]],
...,
[[-1.2402e+00, -2.4192e+00, -2.5029e+00, ..., -6.0277e-01,
-1.5383e+00, -1.3239e+00],
[-3.4030e+00, -4.5759e+00, -4.6749e+00, ..., -3.6685e+00,
-2.6635e+00, -4.4132e+00],
[-2.5531e+00, -3.2913e+00, -3.6840e+00, ..., -6.2698e+00,
-2.4071e+00, -3.5618e+00],
...,
[-1.3065e+00, -2.6304e+00, -7.8036e-01, ..., 1.2009e+00,
2.1653e+00, -1.9983e-01],
[-2.2869e+00, -4.2795e+00, -1.6909e+00, ..., 3.3665e+00,
-1.1685e+00, -1.2693e+00],
[-1.4406e+00, -1.5156e+00, -7.0787e-01, ..., 3.3579e+00,
-2.1141e+00, 4.8626e-01]],
[[-1.6977e+00, -1.7452e+00, -1.5634e+00, ..., 6.7804e-01,
-1.1774e+00, -2.0135e-01],
[-3.8815e+00, -3.3667e+00, -2.3841e+00, ..., -6.4498e-01,
-2.7232e+00, -1.1212e+00],
[-4.0657e+00, -3.9515e+00, -3.8180e+00, ..., -4.5155e-01,
-2.5107e+00, -1.6026e+00],
...,
[-2.4652e+00, -2.0290e+00, -3.5939e+00, ..., -2.8894e-01,
-3.0083e+00, -9.4826e-01],
[-1.9222e+00, 2.7506e-02, -1.3474e+00, ..., -1.6857e+00,
-4.1187e+00, -1.7157e+00],
[-2.2022e+00, -1.6368e+00, -1.6355e+00, ..., -7.9597e-01,
-2.1906e+00, -4.7438e-01]],
[[ 1.3678e+00, 6.2075e-01, 1.3530e+00, ..., 3.1185e+00,
1.7445e+00, 4.5739e-01],
[ 4.0344e+00, 3.2324e+00, 6.6332e-01, ..., 4.0299e+00,
2.1935e+00, 2.6799e+00],
[ 1.8452e+00, 9.0283e-01, 4.3011e-01, ..., 3.5202e+00,
1.3695e+00, 1.4057e+00],
...,
[ 2.7233e+00, 3.3693e+00, 2.9187e+00, ..., 8.6034e-01,
2.4921e+00, -2.2878e+00],
[ 3.7054e+00, 4.1701e+00, 3.1043e+00, ..., 1.9897e+00,
-9.3045e-01, -3.8652e+00],
[ 2.3925e+00, 2.2639e+00, 1.3658e+00, ..., 1.3370e+00,
-1.7524e+00, -2.3790e+00]]]], device='cuda:0'),
tensor([[[[-1.2564e-01, 7.0685e-01, -4.2450e-01, ..., -1.1246e+00,
-9.9246e-01, -7.8469e-01],
[ 7.7173e-01, 4.1234e-01, -1.2499e-01, ..., 5.3293e-01,
8.7554e-01, -2.1688e-01],
[ 1.2175e+00, -5.9586e-01, 9.7485e-01, ..., -1.9337e+00,
8.5061e-01, -2.3714e-01],
...,
[-2.3232e+00, -5.1951e-01, 7.5379e-01, ..., -2.7156e+00,
-5.2923e+00, 9.7782e-02],
[-1.7754e+00, 5.2304e-01, 1.7190e+00, ..., -3.7671e+00,
-2.5764e+00, 1.9565e+00],
[-1.7798e+00, -3.7274e+00, -2.5353e+00, ..., -4.4661e-01,
1.0648e+00, 1.4246e+00]],
[[-7.8452e-01, -1.4985e+00, -2.6821e+00, ..., -1.5807e+00,
-3.5835e+00, -2.0221e+00],
[-2.0158e+00, -3.6494e+00, -3.9167e+00, ..., -5.1273e-01,
-3.0258e+00, -3.7613e+00],
[-2.1519e+00, -3.3371e+00, -3.0721e+00, ..., -2.7232e+00,
-1.8746e+00, -2.6494e+00],
...,
[-6.0682e+00, -3.4173e+00, -1.0402e+00, ..., -3.8337e+00,
-7.8656e+00, -1.5030e-02],
[-8.6638e+00, -8.9491e+00, -6.9873e+00, ..., -7.0488e+00,
-3.7745e+00, 6.3111e-01],
[-1.8790e+00, -3.8418e+00, -3.0028e+00, ..., 3.7286e-01,
1.9901e-01, 5.7518e-01]],
[[ 7.4387e-01, 1.2824e+00, 1.1138e+00, ..., 2.5243e+00,
2.2515e+00, 9.6793e-01],
[ 8.0089e-01, 1.5779e+00, 1.8270e+00, ..., 1.0658e+00,
2.5840e+00, 1.1725e+00],
[ 3.2605e-01, 2.9678e+00, 2.4203e+00, ..., -4.4615e+00,
1.2473e+00, 2.3643e+00],
...,
[ 2.8622e-01, -2.8752e+00, -3.3508e+00, ..., -7.3137e+00,
-8.1840e+00, -4.7252e+00],
[-9.3333e-01, -2.7770e+00, -1.1198e+00, ..., -7.6239e+00,
-4.0526e+00, -1.2509e+00],
[-1.2737e-01, -3.8285e-01, -9.1332e-01, ..., -3.6708e+00,
-2.5681e+00, -1.3375e+00]],
...,
[[-1.0547e+00, -1.4522e+00, -2.1255e+00, ..., -2.0812e+00,
-3.0803e+00, -1.1775e+00],
[-2.4960e+00, -3.4484e+00, -2.6804e+00, ..., -1.2300e+00,
-2.8547e+00, -3.0224e+00],
[-1.2384e+00, -1.6625e+00, -3.2599e-01, ..., -1.2945e-02,
-1.9581e+00, -2.1886e+00],
...,
[-1.4525e+00, -3.7347e-01, -2.5195e-01, ..., -2.0780e+00,
-1.7768e+00, -9.1783e-02],
[-2.9804e+00, -3.6222e+00, -2.7911e+00, ..., -2.6467e+00,
-1.3353e+00, -1.3658e-01],
[-2.1068e+00, -1.9196e+00, -5.5037e-01, ..., 6.6140e-01,
2.2482e-01, 3.4088e-01]],
[[-1.6718e+00, -2.5590e+00, -3.5981e+00, ..., -2.5207e+00,
-1.9865e+00, -3.0689e-01],
[-4.4422e+00, -5.0134e+00, -4.9321e+00, ..., -1.8277e+00,
-4.7619e+00, -1.6572e+00],
[-4.1190e+00, -4.6168e+00, -3.0927e+00, ..., 1.5483e+00,
-2.1835e+00, -1.0035e+00],
...,
[-2.7220e+00, -1.1294e+00, -5.4848e-01, ..., 1.1852e+00,
1.4895e+00, 1.9814e+00],
[-2.8368e+00, 1.6821e+00, -5.4425e-01, ..., -2.2683e-01,
1.5610e+00, 2.3612e+00],
[-3.2095e+00, -7.0886e-01, -3.7673e-01, ..., 7.3157e-01,
1.4428e+00, 9.9867e-01]],
[[ 8.8936e-01, -1.4201e-01, 7.3858e-01, ..., 4.0675e-01,
1.2543e+00, 6.9491e-01],
[ 1.8362e+00, 7.8666e-01, 3.3513e-01, ..., -1.6466e-01,
1.6980e+00, 3.6375e+00],
[ 1.6298e+00, -1.0370e-01, 2.9208e-01, ..., -2.7204e+00,
1.0941e+00, 1.9421e+00],
...,
[ 2.9881e+00, 1.1410e+00, -5.2802e-01, ..., -2.6827e+00,
-3.3234e+00, -2.7192e-01],
[ 4.5632e+00, 5.2334e+00, 3.8166e+00, ..., -1.3926e+00,
-1.0634e+00, 2.2719e+00],
[ 2.3100e+00, 1.4070e+00, -2.2761e-01, ..., -5.6925e-01,
1.9947e+00, 2.6023e+00]]],
[[[ 2.2860e-01, 1.3580e+00, 6.5589e-01, ..., -5.5102e-01,
5.1952e-02, 6.9919e-01],
[ 4.7832e-01, 2.6109e-01, 7.0136e-01, ..., -1.0295e+00,
-1.1360e+00, 1.0108e+00],
[-1.1187e+00, -2.2137e+00, -2.1480e+00, ..., -5.1506e-01,
-8.4385e-01, 4.8178e-01],
...,
[-1.1355e+00, -2.0924e+00, -4.2777e-01, ..., -1.5743e+00,
-6.9593e-01, -1.3497e+00],
[-1.1432e+00, -1.0898e+00, -4.8153e-01, ..., -8.8002e-01,
-5.0619e-01, -1.8321e+00],
[-2.5459e+00, -5.0078e+00, -4.3853e+00, ..., -4.6703e+00,
-4.0454e+00, -3.7452e+00]],
[[-8.1510e-01, -2.3550e+00, -4.2532e+00, ..., -3.2912e+00,
-4.9168e+00, -4.0546e-01],
[-5.0763e+00, -6.5077e+00, -6.6847e+00, ..., -5.2561e+00,
-7.2022e+00, -1.7021e+00],
[-4.7052e+00, -4.9451e+00, -3.5406e+00, ..., -8.3284e+00,
-8.6101e+00, -2.9601e+00],
...,
[-5.6382e+00, -5.7237e+00, -4.3292e+00, ..., -4.1569e+00,
-4.4893e+00, -3.5214e+00],
[-6.8710e+00, -7.5806e+00, -6.1287e+00, ..., -7.8435e+00,
-8.5054e+00, -6.8534e+00],
[-2.4020e+00, -4.2219e+00, -3.8875e+00, ..., -3.8868e+00,
-4.0770e+00, -2.5005e+00]],
[[-2.9802e-01, 2.1776e+00, 1.2462e+00, ..., 1.0307e+00,
2.2075e+00, 9.3185e-01],
[-7.3246e-02, 1.2519e+00, 1.2302e+00, ..., 4.6068e-01,
1.3772e+00, -4.7902e-01],
[-4.9358e-01, 2.2007e+00, 1.5295e+00, ..., -4.9189e-02,
1.9679e+00, 2.8017e-01],
...,
[ 1.0798e+00, 1.5560e+00, 1.2225e+00, ..., 2.0951e+00,
1.4654e+00, 2.2649e-01],
[ 8.6966e-01, 7.5934e-01, 7.6063e-01, ..., 7.1185e-01,
1.3918e-01, 2.0261e-02],
[ 1.2379e+00, 1.4799e+00, 1.2921e+00, ..., 1.5701e+00,
1.0913e+00, 1.5187e-01]],
...,
[[-8.0700e-01, -1.8776e+00, -1.8654e+00, ..., -2.5200e+00,
-2.9197e+00, -4.2466e-01],
[-3.2334e+00, -4.4020e+00, -2.8901e+00, ..., -3.8619e+00,
-4.3986e+00, -2.6561e+00],
[-1.8197e+00, -1.8566e+00, -6.1534e-01, ..., -3.0317e+00,
-4.6417e+00, -1.8098e+00],
...,
[-3.1176e+00, -3.0928e+00, -1.5186e+00, ..., -2.1019e-01,
-2.3524e+00, -2.1489e+00],
[-3.0248e+00, -2.9426e+00, -1.8368e+00, ..., -2.5839e+00,
-4.5724e+00, -3.0218e+00],
[-3.2586e+00, -3.5789e+00, -3.1631e+00, ..., -2.9832e+00,
-3.5969e+00, -1.5604e+00]],
[[-1.7351e+00, -2.2212e+00, -2.8260e+00, ..., -1.4358e+00,
-1.3939e+00, -1.3576e+00],
[-4.1511e+00, -3.0154e+00, -2.4990e+00, ..., -2.6208e+00,
-1.5177e+00, -6.4475e-01],
[-2.9694e+00, -2.7135e+00, -3.2321e+00, ..., 6.6671e-03,
-1.2185e+00, -7.2836e-01],
...,
[-3.9994e+00, -3.0704e+00, -3.5750e+00, ..., -2.8972e+00,
-3.7409e+00, -1.8129e+00],
[-3.7161e+00, -1.1942e+00, -2.6476e+00, ..., -8.8520e-01,
-2.5234e+00, -9.6592e-01],
[-4.1968e+00, -3.0081e+00, -2.9551e+00, ..., -2.4218e+00,
-3.1310e+00, -1.2520e+00]],
[[ 1.0885e+00, 5.8271e-01, 7.6854e-01, ..., -5.4121e-01,
1.2610e-01, -4.8389e-01],
[ 3.5919e+00, 3.5676e+00, 3.0877e+00, ..., 1.7169e+00,
2.0366e+00, 2.4916e+00],
[ 1.9192e+00, 4.8230e-01, -3.8889e-01, ..., 2.3127e+00,
2.1491e+00, 1.8012e+00],
...,
[ 3.1778e+00, 1.5537e+00, 1.8007e+00, ..., -3.9074e-01,
6.6374e-01, 1.2944e+00],
[ 3.5336e+00, 3.3757e+00, 2.1401e+00, ..., 2.9681e+00,
2.3438e+00, 1.8178e+00],
[ 1.8820e+00, 6.3730e-01, -7.1934e-02, ..., -2.7966e-01,
-2.2883e-01, -2.6393e-01]]],
[[[ 2.5064e-02, 9.7091e-01, 1.3635e-01, ..., -5.8721e-01,
-7.7690e-02, 7.9758e-01],
[-2.8995e-02, -7.9656e-01, 1.7203e-01, ..., -1.5564e+00,
-1.3611e+00, 1.0420e+00],
[-1.2399e-01, -6.4971e-01, 1.4064e+00, ..., -1.1765e+00,
-5.3712e-01, 2.5384e-01],
...,
[ 2.8539e-01, -2.7456e-01, 7.6762e-01, ..., -1.2275e+00,
-1.3844e+00, -9.6998e-01],
[-1.3685e+00, -1.0542e+00, -7.4275e-01, ..., -2.1393e+00,
-3.1226e+00, -1.3614e+00],
[-1.5681e-01, -2.3323e+00, -2.3942e+00, ..., -1.4572e+00,
-2.5608e+00, -8.8480e-01]],
[[-1.9516e+00, -3.0953e+00, -4.3448e+00, ..., -3.7005e+00,
-4.7123e+00, -2.6572e-01],
[-6.2935e+00, -6.7257e+00, -5.1670e+00, ..., -5.9925e+00,
-7.6878e+00, -1.7157e+00],
[-4.6939e+00, -6.2674e+00, -6.9361e+00, ..., -3.8484e+00,
-7.6034e+00, -3.3132e+00],
...,
[-4.6341e+00, -5.4934e+00, -3.5739e+00, ..., -6.7143e+00,
-5.3243e+00, -2.2162e+00],
[-7.6415e+00, -9.0079e+00, -7.0756e+00, ..., -8.9753e+00,
-9.1673e+00, -4.0229e+00],
[-1.1071e+00, -3.2700e+00, -1.9308e+00, ..., -3.3920e+00,
-2.8105e+00, -2.8792e-01]],
[[-3.3300e-01, 1.9107e+00, 1.1113e+00, ..., 8.0364e-01,
2.0120e+00, 9.4172e-01],
[ 4.0165e-01, 5.0696e-01, 9.7050e-01, ..., 5.6388e-01,
1.1664e+00, -6.0716e-01],
[ 6.3269e-01, 1.7909e+00, 2.0697e+00, ..., 6.7817e-01,
1.6600e+00, -1.3693e-01],
...,
[ 8.1542e-02, 1.2302e+00, -1.3259e-01, ..., 1.6364e+00,
8.8363e-01, 5.4665e-01],
[ 3.1742e-01, 6.4446e-01, -5.5912e-01, ..., 1.6311e+00,
1.9391e+00, 7.5612e-01],
[-5.5867e-01, 4.7138e-01, -8.9484e-01, ..., 6.2045e-01,
1.1746e+00, -6.1670e-01]],
...,
[[-1.4002e+00, -2.0835e+00, -2.0788e+00, ..., -2.3101e+00,
-2.6791e+00, -3.8691e-01],
[-4.3894e+00, -4.3901e+00, -2.8240e+00, ..., -3.3548e+00,
-4.2515e+00, -2.5900e+00],
[-2.6066e+00, -2.3349e+00, -1.3081e+00, ..., -2.4704e+00,
-4.2027e+00, -1.8913e+00],
...,
[-9.5783e-01, -8.7615e-01, -3.3593e-01, ..., -9.1614e-01,
-4.4678e+00, -1.8002e+00],
[-2.7836e+00, -2.5551e+00, -1.4835e+00, ..., -1.8959e+00,
-5.1861e+00, -1.6144e+00],
[-8.5971e-01, -1.1614e+00, 4.2687e-01, ..., -2.5052e-01,
-1.4150e+00, 1.1907e+00]],
[[-1.9581e+00, -1.7738e+00, -2.8758e+00, ..., -1.9070e+00,
-1.3233e+00, -1.3160e+00],
[-5.3548e+00, -2.7696e+00, -2.9702e+00, ..., -2.9424e+00,
-1.4996e+00, -7.2102e-01],
[-3.8716e+00, -1.5664e+00, -3.0885e+00, ..., 1.0219e+00,
-1.1808e-01, -2.8606e-01],
...,
[-3.6673e+00, -3.5228e+00, -5.7713e+00, ..., -9.4242e-01,
-2.5899e+00, -1.2373e+00],
[-4.3092e+00, -1.4470e+00, -3.7097e+00, ..., -6.7210e-01,
-2.1201e+00, -6.7004e-01],
[-2.9903e+00, -1.1610e+00, -5.9698e-01, ..., 1.2122e-01,
-5.1426e-01, -3.3664e-02]],
[[ 1.7152e+00, 6.7381e-01, 7.5645e-01, ..., -5.2605e-01,
3.0785e-02, -3.8866e-01],
[ 4.4367e+00, 3.4016e+00, 2.1725e+00, ..., 1.3306e+00,
1.9797e+00, 2.2800e+00],
[ 2.0846e+00, 1.1855e+00, 1.0084e+00, ..., 1.2597e+00,
2.1898e+00, 1.3185e+00],
...,
[ 2.7196e+00, 1.3958e+00, 1.5581e+00, ..., 8.9190e-01,
8.7132e-01, 5.1054e-01],
[ 3.3594e+00, 4.3285e+00, 2.3509e+00, ..., 1.6019e+00,
1.2863e+00, 1.5744e+00],
[ 2.2432e+00, 1.9060e+00, 4.0092e-01, ..., 6.0160e-01,
4.1382e-01, 1.0336e+00]]],
[[[ 2.3025e-01, 1.4111e+00, 5.7430e-01, ..., -3.9653e-01,
5.8810e-02, 6.6542e-01],
[ 4.1835e-01, 4.0011e-01, 6.1613e-01, ..., -8.3681e-01,
-1.1237e+00, 1.0271e+00],
[ 1.3828e-01, 4.2086e-01, 9.2257e-01, ..., -7.9513e-01,
-9.7223e-01, 3.3012e-01],
...,
[ 4.6955e-01, -1.0623e-01, -2.3806e-01, ..., -2.7087e-01,
-5.6172e-01, -1.0766e+00],
[-1.2409e+00, -1.5445e+00, -4.6953e+00, ..., -1.9071e+00,
-2.5425e+00, -1.3851e+00],
[-1.0263e-01, -2.5295e+00, -4.3522e+00, ..., -2.1823e+00,
-2.3912e+00, -9.3679e-01]],
[[-7.6123e-01, -2.2368e+00, -4.2039e+00, ..., -3.3206e+00,
-5.0480e+00, -4.2982e-01],
[-4.5936e+00, -6.0127e+00, -5.5573e+00, ..., -5.5171e+00,
-7.5168e+00, -1.8458e+00],
[-3.3717e+00, -5.1663e+00, -6.3331e+00, ..., -7.8233e+00,
-8.5594e+00, -3.1759e+00],
...,
[-4.5205e+00, -4.3340e+00, -5.9441e+00, ..., -3.9172e+00,
-5.0561e+00, -2.3457e+00],
[-8.1356e+00, -8.4892e+00, -9.3849e+00, ..., -7.5261e+00,
-9.2265e+00, -4.0379e+00],
[-1.2686e+00, -2.6861e+00, -3.3908e+00, ..., -2.4836e+00,
-2.8836e+00, -3.2547e-01]],
[[-2.6266e-01, 2.1564e+00, 1.2926e+00, ..., 8.8857e-01,
2.1703e+00, 9.6948e-01],
[-1.5460e-02, 1.0999e+00, 8.9382e-01, ..., 4.5363e-01,
1.2638e+00, -4.4033e-01],
[-1.0394e-02, 1.9676e+00, 1.4063e+00, ..., 7.4140e-02,
1.4838e+00, -1.1450e-01],
...,
[ 2.5146e-02, 8.9381e-01, 6.8479e-01, ..., 1.1661e+00,
1.1896e+00, 5.3809e-01],
[ 1.9323e-01, 6.7128e-01, 1.3668e+00, ..., 1.8262e+00,
1.9848e+00, 6.6115e-01],
[-7.2869e-01, 5.8427e-01, 1.8101e+00, ..., 7.7361e-01,
9.0244e-01, -7.1144e-01]],
...,
[[-8.0860e-01, -1.8152e+00, -1.8885e+00, ..., -2.5095e+00,
-2.9539e+00, -4.8770e-01],
[-2.9354e+00, -4.2459e+00, -2.8451e+00, ..., -3.9430e+00,
-4.1463e+00, -2.5880e+00],
[-1.3023e+00, -2.5673e+00, -1.4525e+00, ..., -3.3668e+00,
-4.7767e+00, -1.6314e+00],
...,
[-9.0969e-01, -1.1053e+00, -1.7580e+00, ..., -1.6246e+00,
-3.6702e+00, -1.7703e+00],
[-2.9556e+00, -3.7902e+00, -3.1329e+00, ..., -3.2214e+00,
-4.6552e+00, -1.4329e+00],
[-6.8297e-01, -1.0812e+00, -4.2708e-01, ..., -1.1296e+00,
-1.0874e+00, 1.2092e+00]],
[[-1.6597e+00, -2.2323e+00, -2.9003e+00, ..., -1.4871e+00,
-1.4742e+00, -1.3734e+00],
[-4.2193e+00, -3.3073e+00, -3.0837e+00, ..., -2.9248e+00,
-1.7905e+00, -7.4020e-01],
[-2.5269e+00, -1.6948e+00, -2.0189e+00, ..., -6.7301e-01,
-6.3648e-01, -4.3977e-01],
...,
[-3.2047e+00, -3.5875e+00, -4.4080e+00, ..., -1.7273e+00,
-2.8059e+00, -1.2226e+00],
[-3.9798e+00, -3.0937e+00, -1.6159e+00, ..., -1.3994e+00,
-2.3732e+00, -6.8276e-01],
[-2.6842e+00, -2.2855e+00, -4.2338e-01, ..., -1.7959e-01,
-9.9890e-01, -1.9997e-02]],
[[ 1.0728e+00, 5.4904e-01, 7.1016e-01, ..., -5.5456e-01,
1.2784e-01, -4.5684e-01],
[ 3.4131e+00, 3.4442e+00, 2.3137e+00, ..., 1.9572e+00,
2.0027e+00, 2.4681e+00],
[ 1.6124e+00, 1.5028e+00, 1.5886e+00, ..., 2.5304e+00,
2.3642e+00, 1.8077e+00],
...,
[ 2.8089e+00, 1.7571e+00, 1.9051e+00, ..., 1.1837e+00,
1.1560e+00, 5.8810e-01],
[ 3.6676e+00, 4.3795e+00, 1.5827e+00, ..., 2.1828e+00,
1.2596e+00, 1.6906e+00],
[ 2.4576e+00, 1.3344e+00, -9.8867e-01, ..., 7.2279e-01,
1.4353e-02, 1.0447e+00]]],
[[[-3.7794e+00, -6.7101e+00, -4.1652e+00, ..., -6.4178e+00,
-2.5278e+00, 1.4894e+00],
[-3.0143e+00, -8.8890e+00, -7.0134e+00, ..., -9.6580e+00,
2.0132e+00, 2.2160e+00],
[-3.4854e+00, -7.7187e+00, -4.3401e+00, ..., -1.2034e+00,
2.3493e+00, -1.3291e+00],
...,
[-6.7585e+00, -3.2860e+00, 3.4109e+00, ..., -2.8023e+00,
-5.5437e+00, -4.9722e+00],
[-5.6614e+00, -2.4746e+00, -2.4008e-01, ..., -3.2457e+00,
-2.8271e+00, -1.6819e+00],
[-2.1634e+00, -4.7266e+00, -1.1513e+00, ..., -4.9929e+00,
-5.8637e+00, -3.5830e+00]],
[[-3.7449e+00, -6.1509e+00, -3.0861e-01, ..., -6.3593e+00,
-3.5152e+00, 1.0655e+00],
[-4.3622e+00, -7.7779e+00, -2.8375e+00, ..., -6.8458e+00,
2.5966e+00, 8.3626e-01],
[-7.4885e+00, -1.1337e+01, -4.7771e+00, ..., 5.8086e-01,
1.4877e+00, -5.6392e+00],
...,
[-9.3622e+00, -4.5297e+00, 1.7307e+00, ..., -1.5432e+01,
-1.5320e+01, -7.8218e+00],
[-1.1569e+01, -9.0079e+00, 1.0235e+00, ..., -8.1011e+00,
-7.7669e+00, -5.6813e+00],
[-1.9056e+00, -7.0572e+00, -6.5453e+00, ..., -4.8736e+00,
-4.1381e+00, -2.2164e+00]],
[[ 1.0733e+00, 2.9686e+00, 3.0746e+00, ..., -5.5176e+00,
-1.6091e+00, 1.3101e+00],
[ 5.0858e-01, 5.9333e-01, -5.6723e-01, ..., -7.8766e+00,
-1.7176e+00, 9.2274e-01],
[ 4.3425e-01, 1.5765e+00, -1.3397e+00, ..., -1.7206e+00,
-3.2734e+00, -1.3325e+00],
...,
[-1.6812e+00, -4.2566e+00, -5.1051e+00, ..., -2.4065e+00,
-3.4686e+00, -7.3174e-01],
[-4.9341e+00, -9.4853e+00, -8.3556e+00, ..., 1.8999e+00,
2.8812e-01, -1.5458e+00],
[-3.1323e+00, -4.6707e+00, -5.9540e+00, ..., -7.3192e-01,
3.6027e-01, -5.1257e-01]],
...,
[[-3.2945e+00, -3.9552e+00, -1.8144e+00, ..., -1.1887e+00,
4.7704e-01, 1.8410e-01],
[-5.2798e+00, -5.0027e+00, -2.6425e+00, ..., 2.2772e-04,
6.1073e-01, -2.3059e+00],
[-4.0960e+00, -4.9456e+00, -2.4335e+00, ..., 7.4346e-01,
-1.2729e+00, -1.5048e+00],
...,
[-2.1012e+00, -1.3925e+00, -4.7948e+00, ..., -8.9271e+00,
-8.8367e+00, -4.1549e+00],
[-1.6022e+00, -2.6145e+00, -1.4803e+00, ..., -1.2719e+00,
-3.4940e+00, -1.4408e+00],
[-1.7590e+00, -4.6751e+00, -4.7807e+00, ..., -2.3205e+00,
-4.3088e+00, -1.6984e+00]],
[[-3.9195e+00, -2.5524e+00, -9.7267e+00, ..., -1.2402e+00,
-3.4094e-01, -2.0852e-02],
[-5.1510e+00, -6.9344e-01, -2.2267e+00, ..., -3.4144e+00,
1.3589e+00, 3.2282e-01],
[-5.2480e+00, -2.9039e+00, -1.8031e+00, ..., 7.8033e-01,
2.7352e+00, -6.7290e-01],
...,
[-3.3228e+00, -2.5121e+00, -8.5851e+00, ..., -1.1619e+01,
-6.6527e+00, -2.7295e+00],
[-2.5080e-01, 5.2362e+00, 1.7896e+00, ..., -2.7789e+00,
-1.5371e+00, -1.2983e-01],
[-7.2150e-01, 2.2856e+00, 3.0158e+00, ..., -9.2138e-01,
-5.2717e-01, 8.1656e-01]],
[[ 2.5472e+00, 6.6625e-01, -2.7811e+00, ..., -6.8929e+00,
-2.1765e+00, 2.0981e+00],
[ 3.2026e+00, 1.9218e-01, -2.5111e+00, ..., -7.9984e+00,
1.2202e+00, 4.3635e+00],
[ 2.6444e+00, 1.5422e+00, 1.8451e+00, ..., 4.3326e+00,
7.4690e+00, 4.1070e+00],
...,
[ 2.8699e+00, 6.3467e+00, 1.3329e+00, ..., 6.3151e+00,
4.6633e+00, 1.6479e+00],
[ 2.7758e+00, 7.3972e+00, -1.3108e+00, ..., 1.3383e-01,
1.5943e+00, 3.4071e+00],
[ 1.3026e+00, 4.5727e+00, 4.4465e+00, ..., -8.0042e-01,
-7.0354e-01, -5.2484e-01]]]], device='cuda:0'),
tensor([[[[-3.1099e+00, -2.9489e+00, -2.4926e+00, ..., -4.0902e+00,
-2.2824e+00, -1.8670e+00],
[ 2.3715e+00, 7.6795e-01, 1.2561e+00, ..., 1.6604e+00,
1.6303e+00, -3.8188e+00],
[ 1.4327e+00, -3.3930e+00, 4.6483e-01, ..., 6.8825e+00,
1.1514e+00, -3.5696e+00],
...,
[-8.5668e-01, 2.5916e-01, 1.4421e+00, ..., -3.8925e+00,
-2.7547e+00, -1.6613e+00],
[-1.3625e+00, 4.5688e-01, 1.4271e-01, ..., -2.4133e+00,
-1.7235e+00, -1.6220e+00],
[-9.8946e-01, -1.9313e+00, -1.6694e+00, ..., -4.0378e+00,
-3.2882e+00, -2.9425e+00]],
[[-5.2136e+00, -5.1584e+00, -3.7486e+00, ..., -4.1435e+00,
-8.0741e+00, -4.5827e+00],
[-2.5203e+00, -3.6731e+00, -9.4002e-01, ..., -3.7541e-01,
-7.0533e+00, -8.5203e+00],
[ 1.4362e-01, -8.2190e+00, -1.0730e+01, ..., 5.0897e+00,
-6.8565e+00, -1.2278e+01],
...,
[-4.3145e+00, -4.8054e+00, -4.0027e+00, ..., -9.0652e+00,
-7.2669e+00, -2.8360e+00],
[-7.1193e+00, -7.2411e+00, -5.2373e+00, ..., -5.4306e+00,
-6.1097e+00, -5.7554e+00],
[-1.1517e+00, -3.2001e+00, -1.8360e+00, ..., -3.2120e+00,
-2.6629e+00, -2.2043e+00]],
[[ 6.3289e-01, 4.9375e+00, 4.0269e+00, ..., 4.4311e+00,
2.6462e+00, 1.6773e-01],
[-9.5046e-01, -8.7311e-01, 3.6209e+00, ..., 9.1708e+00,
7.2595e+00, 1.3136e+00],
[-5.0007e+00, -1.1359e+01, -5.5570e+00, ..., 1.5166e+00,
8.0633e+00, 1.1693e+00],
...,
[-6.9075e-01, 7.3180e-01, 3.5249e-01, ..., 4.7209e+00,
7.0929e-01, 4.4900e-01],
[-2.0433e+00, 1.1005e-01, 8.9472e-01, ..., 4.3305e-01,
-8.9421e-01, 8.5630e-01],
[-1.1981e+00, -4.9196e-01, 3.3858e-01, ..., -6.6296e-01,
1.1125e+00, 6.4395e-01]],
...,
[[-3.5121e+00, -4.1889e+00, -4.5233e+00, ..., -6.2292e+00,
-4.9965e+00, -2.2661e+00],
[-1.1526e+00, -8.5331e-01, -1.2583e+00, ..., -5.6289e+00,
-8.4291e+00, -3.7500e+00],
[-1.3151e+00, -1.6576e+00, -2.5875e+00, ..., -2.5484e+00,
-9.4420e+00, -1.4142e+00],
...,
[-9.0591e-01, -2.4170e+00, -4.6081e-01, ..., -8.3569e+00,
-3.4010e+00, -2.7864e+00],
[-1.5943e+00, -4.4159e+00, -1.5702e+00, ..., -2.7207e+00,
-4.1200e+00, -3.0601e+00],
[-1.0574e+00, -1.7667e+00, -4.1396e-01, ..., -3.7030e+00,
-3.7138e+00, -1.3480e+00]],
[[-3.2390e+00, -5.8996e+00, -1.2143e+00, ..., -6.2407e-01,
-1.3394e+00, -1.0837e+00],
[-2.0231e+00, -2.0227e+00, -3.9054e+00, ..., -8.6006e+00,
-2.2116e+00, -1.6025e+00],
[ 3.0120e-01, 1.8002e+00, -1.7129e+00, ..., -9.4200e+00,
-6.3256e+00, 7.6514e-01],
...,
[-2.4371e+00, -2.5051e+00, -2.2735e+00, ..., -1.1810e+01,
-5.4805e+00, -1.2168e+00],
[-1.2912e+00, -8.9634e-02, -7.9769e-01, ..., -5.8523e+00,
-4.1722e+00, -5.8486e-01],
[-1.5545e+00, -1.2957e+00, -7.2003e-01, ..., -1.3886e+00,
-2.4328e+00, -7.6471e-01]],
[[ 8.3112e-01, 2.9297e-01, 1.6681e+00, ..., 1.9339e+00,
1.0252e+00, -1.3058e-01],
[ 3.7482e+00, 1.6174e+00, 3.8765e+00, ..., 8.1461e+00,
6.3400e+00, 1.5752e+00],
[-3.3825e-01, -3.4674e+00, 1.0917e+00, ..., 3.6320e+00,
6.2787e+00, -5.3146e-01],
...,
[ 2.5591e+00, 2.9421e+00, 2.5067e+00, ..., 6.3369e-01,
8.4054e-01, 1.2689e+00],
[ 3.5430e+00, 3.8445e+00, 2.4787e+00, ..., 3.0375e+00,
2.9758e+00, 1.6290e+00],
[ 2.3832e+00, 2.4233e+00, 1.2844e+00, ..., 1.7809e+00,
-3.3956e-01, -6.3207e-01]]],
[[[ 2.3790e-01, 2.0428e+00, 1.5596e+00, ..., 1.1290e-01,
4.8595e-01, -1.0575e-01],
[ 6.8688e-01, 2.8136e-01, -7.6636e-01, ..., 5.9724e-02,
4.8441e-01, -5.6027e-01],
[-7.0260e-01, -2.3660e+00, -2.4953e+00, ..., -7.5249e-02,
-6.0738e-01, -1.1668e+00],
...,
[-6.7356e-01, 4.8691e-01, 1.5870e+00, ..., 6.8321e-01,
9.5488e-02, -7.7933e-01],
[-6.1121e-01, 5.7295e-01, 5.7245e-01, ..., -5.1860e-01,
-1.9067e-01, -1.1903e+00],
[-1.5479e+00, -2.5871e+00, -1.4933e+00, ..., -3.1770e+00,
-2.9108e+00, -2.7246e+00]],
[[-2.5063e-01, -1.1903e+00, -3.1892e+00, ..., -1.9334e+00,
-3.5651e+00, -2.1397e+00],
[-3.9458e+00, -6.2951e+00, -5.5476e+00, ..., -3.0645e+00,
-3.9594e+00, -3.5053e+00],
[-3.1394e+00, -6.9654e+00, -3.8708e+00, ..., -6.5527e+00,
-5.1238e+00, -3.1663e+00],
...,
[-2.7690e+00, -1.1950e+00, -1.3519e+00, ..., -2.6674e+00,
-3.3732e+00, -2.9497e+00],
[-8.0027e+00, -8.5898e+00, -8.9713e+00, ..., -5.6438e+00,
-5.8717e+00, -5.1822e+00],
[-1.4349e+00, -2.8586e+00, -1.3192e+00, ..., -2.9391e+00,
-2.8270e+00, -1.6001e+00]],
[[-1.0706e-01, 2.3634e+00, 1.2987e+00, ..., 1.6883e+00,
1.7054e+00, 8.5943e-01],
[-9.8056e-01, -2.5894e-01, 8.5432e-01, ..., 9.0498e-01,
1.3351e+00, 7.7222e-01],
[-6.2521e-01, 1.7987e+00, 1.7831e+00, ..., 1.5415e+00,
1.7281e+00, 1.3421e+00],
...,
[-8.8548e-01, -1.4032e+00, -7.1598e-01, ..., 1.0173e-01,
3.4554e-01, 2.5779e-01],
[-9.5410e-01, 4.4695e-01, -3.4804e-01, ..., 6.8427e-01,
6.5591e-01, 3.7773e-01],
[-6.0003e-01, -5.4251e-01, -8.8044e-02, ..., 1.7099e+00,
1.6617e+00, 5.2730e-01]],
...,
[[-5.3958e-01, -1.4871e+00, -1.4355e+00, ..., -1.4363e+00,
-1.9463e+00, -4.2264e-01],
[-2.6970e+00, -3.7411e+00, -3.4904e+00, ..., -2.6945e+00,
-3.8072e+00, -3.0294e+00],
[-2.3202e+00, -3.4434e+00, -1.4332e+00, ..., -1.7924e+00,
-3.5348e+00, -2.1489e+00],
...,
[-1.2489e+00, -2.3124e+00, -6.7844e-01, ..., -1.4886e+00,
-2.3394e+00, -1.1531e+00],
[-2.6262e+00, -5.1571e+00, -3.3080e+00, ..., -2.2330e+00,
-3.6022e+00, -2.0519e+00],
[-1.8937e+00, -1.2254e+00, 6.4829e-01, ..., -1.8545e+00,
-2.2211e+00, -5.6654e-01]],
[[-1.4399e+00, -1.7918e+00, -2.0551e+00, ..., -5.0344e-01,
-1.3242e+00, -6.6845e-01],
[-3.2599e+00, -2.7114e+00, -3.0388e+00, ..., -1.9597e+00,
-3.7849e+00, -1.7847e+00],
[-3.0428e+00, -2.1274e+00, -2.1704e+00, ..., -1.2658e+00,
-4.3110e+00, -3.0043e+00],
...,
[-1.5445e+00, -1.5292e+00, -6.6235e-01, ..., -8.5395e-01,
-3.2022e+00, -1.3204e+00],
[-1.4249e+00, -8.1436e-01, -2.4483e+00, ..., -2.5512e-01,
-2.6137e+00, -1.2553e+00],
[-1.1900e+00, 5.2483e-01, -7.7924e-01, ..., -1.5555e+00,
-2.7227e+00, -1.1848e+00]],
[[ 7.1208e-01, 1.9889e-01, 7.6207e-01, ..., 1.6918e-01,
1.2192e+00, 6.6370e-01],
[ 3.5314e+00, 3.2507e+00, 1.9106e+00, ..., 6.4049e-01,
9.4253e-01, 2.4005e+00],
[ 1.3814e+00, 8.4975e-01, -6.8504e-01, ..., 1.6653e+00,
5.5820e-01, 8.1018e-01],
...,
[ 2.2947e+00, -1.1233e-01, -6.7790e-01, ..., 2.6423e-01,
1.0261e+00, 1.4433e+00],
[ 4.5070e+00, 3.8502e+00, 3.2614e+00, ..., 1.2549e+00,
6.5233e-01, 1.0735e+00],
[ 2.0186e+00, 1.7351e+00, 4.7237e-01, ..., -1.0820e-01,
-3.5316e-01, -1.1037e-01]]],
[[[-2.5731e-01, 1.0157e+00, 7.6832e-01, ..., -1.8086e-01,
1.0935e-01, -2.6443e-01],
[ 1.2011e-01, 1.0223e-01, 1.8549e+00, ..., 2.6420e-01,
7.0044e-01, -5.1617e-01],
[-1.1814e+00, -4.0753e-01, 6.4675e-02, ..., -2.4443e-01,
7.0659e-01, -7.1446e-01],
...,
[-2.6250e-01, -1.7530e-01, 2.3867e-01, ..., 3.7505e-01,
5.3231e-01, -4.3832e-01],
[-1.4855e+00, -5.5420e-02, -5.5273e-02, ..., 1.1482e-01,
3.2373e-01, -1.2588e+00],
[-1.2755e+00, -2.3484e+00, -1.7285e+00, ..., -3.2866e+00,
-2.9428e+00, -2.8609e+00]],
[[-1.6625e+00, -2.5438e+00, -3.8283e+00, ..., -2.0215e+00,
-3.6174e+00, -2.2596e+00],
[-4.9195e+00, -5.5148e+00, -4.5459e+00, ..., -2.0023e+00,
-3.7033e+00, -3.4397e+00],
[-4.1689e+00, -5.0463e+00, -5.9251e+00, ..., -2.0492e+00,
-1.8257e+00, -2.4363e+00],
...,
[-4.4645e+00, -4.9725e+00, -4.6891e+00, ..., -2.7686e+00,
-3.1526e+00, -2.6220e+00],
[-7.6289e+00, -7.9182e+00, -6.4113e+00, ..., -5.0407e+00,
-5.3421e+00, -5.1139e+00],
[-1.4218e+00, -3.1339e+00, -1.8230e+00, ..., -3.0844e+00,
-2.8352e+00, -1.6436e+00]],
[[-2.4841e-01, 2.3117e+00, 1.5662e+00, ..., 1.6298e+00,
1.7848e+00, 9.0474e-01],
[-5.4635e-01, 3.7890e-01, 6.2008e-01, ..., 9.4228e-01,
1.6193e+00, 9.4098e-01],
[ 6.7170e-02, 1.7239e+00, 1.4630e+00, ..., 1.6220e+00,
1.7041e+00, 1.3860e+00],
...,
[-8.0481e-01, 6.9611e-01, 1.4860e+00, ..., 1.4093e+00,
1.0875e+00, 5.2732e-01],
[-1.2699e+00, 4.5372e-02, 1.2455e+00, ..., 1.4205e+00,
9.1737e-01, 3.1781e-01],
[-3.8006e-01, -4.7169e-01, 3.6225e-02, ..., 1.7099e+00,
1.6816e+00, 6.3043e-01]],
...,
[[-1.3203e+00, -2.2963e+00, -2.2243e+00, ..., -1.5968e+00,
-2.0607e+00, -5.2547e-01],
[-3.5786e+00, -4.0460e+00, -2.3932e+00, ..., -2.4279e+00,
-3.2310e+00, -2.8752e+00],
[-2.7592e+00, -3.0468e+00, -2.0338e+00, ..., -2.0969e+00,
-2.1732e+00, -1.6783e+00],
...,
[-9.8434e-01, -2.5556e+00, -1.1257e+00, ..., -1.4522e+00,
-2.1045e+00, -8.6823e-01],
[-2.0912e+00, -4.0091e+00, -1.7203e+00, ..., -1.6434e+00,
-2.8000e+00, -1.6434e+00],
[-1.4229e+00, -1.4590e+00, -6.5456e-01, ..., -1.9887e+00,
-2.2895e+00, -6.5043e-01]],
[[-1.8267e+00, -1.8582e+00, -2.7933e+00, ..., -6.6946e-01,
-1.4403e+00, -4.9249e-01],
[-4.0166e+00, -2.5116e+00, -2.7853e+00, ..., -1.6111e+00,
-3.8345e+00, -1.7564e+00],
[-3.8661e+00, -3.0954e+00, -3.7055e+00, ..., -1.0139e+00,
-4.1705e+00, -2.4600e+00],
...,
[-2.6848e+00, -2.9647e+00, -3.3581e+00, ..., -1.2447e+00,
-3.3074e+00, -2.3447e+00],
[-1.9650e+00, -4.3324e-01, -1.8355e+00, ..., -9.4111e-01,
-3.2094e+00, -1.9874e+00],
[-2.1038e+00, -1.4677e+00, -1.6983e+00, ..., -1.6129e+00,
-2.7675e+00, -1.2529e+00]],
[[ 1.2426e+00, 3.0064e-01, 9.1186e-01, ..., -2.6126e-02,
1.1125e+00, 5.8416e-01],
[ 4.0905e+00, 3.0950e+00, 2.7973e+00, ..., 5.1173e-01,
1.1735e+00, 2.4901e+00],
[ 1.7354e+00, 9.6522e-01, 9.4566e-01, ..., -5.5597e-02,
2.7111e-01, 5.5624e-01],
...,
[ 2.9766e+00, 2.5474e+00, 2.2454e+00, ..., 9.2342e-01,
5.1344e-01, 5.7449e-01],
[ 3.7886e+00, 4.0371e+00, 2.5694e+00, ..., 1.2615e+00,
9.8967e-01, 1.2582e+00],
[ 2.3868e+00, 2.3461e+00, 1.5327e+00, ..., -7.0270e-02,
-1.7890e-01, -4.8455e-02]]],
[[[ 2.1089e-01, 9.7188e-01, -1.6973e-02, ..., -3.1760e+00,
-2.4663e+00, -1.0713e+00],
[-3.6314e-02, 3.7213e-01, 1.0509e+00, ..., -2.0161e+00,
-2.4115e+00, -2.1716e-01],
[-6.5201e-01, -4.9070e-01, -1.5493e-02, ..., -7.1460e-01,
-2.8045e+00, -1.8652e-01],
...,
[-6.7047e-01, -1.7164e+00, -2.1039e+00, ..., -4.0105e+00,
-3.6810e+00, 5.5753e-01],
[-2.1275e+00, -1.4459e+00, -1.5317e+00, ..., -3.5295e+00,
-3.3708e+00, 4.4384e-01],
[-1.0725e+00, -2.8467e+00, -3.2100e+00, ..., -3.9128e+00,
-3.1085e+00, -1.5663e+00]],
[[-1.0411e+00, -1.8016e+00, -3.3591e+00, ..., -6.3194e+00,
-7.2228e+00, -1.3743e+00],
[-3.9558e+00, -3.8428e+00, -2.8792e+00, ..., -6.5572e+00,
-8.4997e+00, -2.3669e+00],
[-3.4655e+00, -3.4734e+00, -2.5750e+00, ..., -5.1843e+00,
-8.6518e+00, -2.7536e+00],
...,
[-5.7061e+00, -5.6196e+00, -4.0468e+00, ..., -9.6499e+00,
-8.3675e+00, -2.1351e+00],
[-7.6416e+00, -7.1807e+00, -5.8660e+00, ..., -9.2676e+00,
-1.0831e+01, -4.7145e+00],
[-1.6831e+00, -3.0950e+00, -2.7410e+00, ..., -3.2824e+00,
-3.9816e+00, -8.6939e-01]],
[[-8.0782e-02, 1.2791e+00, 1.6679e+00, ..., 1.5351e+00,
2.4460e+00, 9.7342e-01],
[ 1.2431e-02, 2.1720e-01, 8.5897e-01, ..., 1.5873e+00,
2.9732e+00, -4.0527e-02],
[ 7.4917e-01, 1.1702e+00, 1.2040e+00, ..., 7.4728e-01,
2.2219e+00, -5.1328e-01],
...,
[ 1.9125e-01, 1.3724e+00, 1.6211e+00, ..., 6.2270e-01,
7.1468e-01, -8.9816e-01],
[ 4.4651e-01, 2.8628e-01, 1.2135e+00, ..., -1.0488e+00,
1.0154e+00, -1.3020e+00],
[ 5.0565e-01, 8.5496e-01, 9.9639e-01, ..., -1.4896e-01,
-2.0265e-01, -1.8649e+00]],
...,
[[-7.4109e-01, -1.7274e+00, -2.2636e+00, ..., -5.0091e+00,
-5.1816e+00, -7.2036e-01],
[-3.0923e+00, -2.6901e+00, -1.6275e+00, ..., -3.6984e+00,
-5.1416e+00, -1.7750e+00],
[-2.0822e+00, -9.3459e-01, -9.7364e-01, ..., -3.4511e+00,
-5.4895e+00, -1.6890e+00],
...,
[-1.6529e+00, -1.3146e+00, -2.0081e+00, ..., -3.4445e+00,
-5.2018e+00, -2.0983e+00],
[-3.2205e+00, -2.2880e+00, -2.2725e+00, ..., -2.9836e+00,
-5.4938e+00, -1.1528e+00],
[-1.3488e+00, -1.1497e+00, -8.1701e-01, ..., -2.2998e+00,
-2.4087e+00, 2.4314e-01]],
[[-1.4231e+00, -2.4727e+00, -4.0095e+00, ..., -1.4389e+00,
-4.9068e-01, -5.8480e-01],
[-3.8278e+00, -3.1846e+00, -4.6937e+00, ..., -3.5468e+00,
-1.9515e+00, -4.6345e-01],
[-4.3408e+00, -3.0823e+00, -3.8621e+00, ..., -1.6240e+00,
-1.6617e+00, -7.3468e-01],
...,
[-4.5930e+00, -4.3047e+00, -4.4711e+00, ..., -3.2180e+00,
-1.9554e+00, -5.2388e-01],
[-4.0316e+00, -1.5851e+00, -2.9459e+00, ..., -1.5620e+00,
-2.4694e-01, 1.2714e+00],
[-3.0862e+00, -2.5547e+00, -2.0747e+00, ..., -2.5361e-01,
-7.2846e-01, 5.7608e-01]],
[[ 1.1939e+00, -1.2091e-01, 1.0892e+00, ..., 4.7072e-01,
7.8379e-01, -1.3915e+00],
[ 2.3522e+00, 9.6200e-01, 1.3157e+00, ..., 3.6950e+00,
3.1220e+00, 2.3096e+00],
[ 1.2240e+00, 5.8339e-01, 2.4734e-01, ..., 3.1347e+00,
1.5888e+00, 1.1070e+00],
...,
[ 2.5327e+00, 1.4784e+00, 7.3150e-01, ..., 1.9057e+00,
2.4608e+00, 3.8087e-01],
[ 2.8167e+00, 3.4179e+00, 1.2867e+00, ..., 3.2716e+00,
1.1417e+00, 1.6732e+00],
[ 1.9023e+00, 1.7071e+00, -5.7742e-01, ..., 5.8002e-01,
-2.3875e-01, 9.1600e-01]]],
[[[-2.7575e-01, 1.1567e+00, 1.3138e+00, ..., -9.1283e-01,
-3.3584e+00, -1.1551e+00],
[ 4.5835e-02, 2.9870e-01, 2.8927e+00, ..., 4.6723e-01,
-2.1288e+00, -7.2997e-01],
[-1.2923e+00, 1.5869e-01, 1.6797e+00, ..., 3.5237e-01,
-2.0318e+00, -1.0760e+00],
...,
[-2.8651e-01, -2.2858e-01, 5.7238e-01, ..., -2.2914e+00,
-3.2510e+00, -7.7947e-02],
[-1.5604e+00, 1.4412e-03, 1.2555e+00, ..., -2.4624e+00,
-3.5493e+00, -2.3490e-01],
[-1.3541e+00, -2.5505e+00, -1.1866e+00, ..., -3.7586e+00,
-3.8826e+00, -1.6420e+00]],
[[-1.5680e+00, -2.3562e+00, -3.3587e+00, ..., -5.3651e+00,
-9.2289e+00, -2.3485e+00],
[-4.8092e+00, -4.9665e+00, -4.5651e+00, ..., -5.3578e+00,
-8.8871e+00, -3.6018e+00],
[-3.8077e+00, -4.4579e+00, -5.5496e+00, ..., -4.7472e+00,
-7.5098e+00, -3.3261e+00],
...,
[-4.8300e+00, -5.4207e+00, -5.3768e+00, ..., -6.8107e+00,
-9.1656e+00, -2.7155e+00],
[-7.8261e+00, -8.1722e+00, -7.2402e+00, ..., -8.6451e+00,
-1.0271e+01, -3.9011e+00],
[-1.5168e+00, -3.1793e+00, -2.4851e+00, ..., -3.9346e+00,
-4.2029e+00, -5.5241e-01]],
[[-2.6179e-01, 2.2750e+00, 1.4117e+00, ..., 2.0497e+00,
2.5102e+00, 7.7200e-01],
[-6.0055e-01, 2.5279e-01, 2.6250e-01, ..., 1.2928e+00,
2.2373e+00, -6.4594e-02],
[-5.9980e-02, 1.9831e+00, 1.2760e+00, ..., 3.0981e-01,
1.2015e+00, -4.0903e-01],
...,
[-6.0284e-01, 6.0036e-01, 9.4834e-01, ..., -2.3841e-01,
-3.9810e-01, -5.5160e-01],
[-1.1271e+00, -3.2546e-02, 2.2233e-01, ..., -7.4729e-01,
8.1639e-01, -5.5893e-01],
[-2.5654e-01, -1.9308e-01, 2.3554e-01, ..., 3.5398e-01,
-3.7596e-02, -2.2055e+00]],
...,
[[-1.3345e+00, -1.9812e+00, -5.5918e-01, ..., -3.4789e+00,
-5.7184e+00, -1.4503e+00],
[-3.5573e+00, -3.5440e+00, -1.2622e+00, ..., -2.9153e+00,
-5.8915e+00, -2.5810e+00],
[-2.6533e+00, -2.4454e+00, -1.1004e+00, ..., -3.1471e+00,
-6.6570e+00, -2.0020e+00],
...,
[-1.2084e+00, -2.4566e+00, -8.3675e-01, ..., -1.8647e+00,
-4.2757e+00, -2.4191e+00],
[-2.2810e+00, -3.7691e+00, -4.3705e-01, ..., -1.1993e+00,
-3.8404e+00, -1.9738e-01],
[-1.5315e+00, -1.5301e+00, 1.6759e-01, ..., -1.4504e+00,
-2.3058e+00, 8.9639e-01]],
[[-1.8217e+00, -1.7599e+00, -1.8491e+00, ..., -2.2187e+00,
-1.2375e+00, 1.6121e-01],
[-3.9781e+00, -2.1250e+00, -2.4125e+00, ..., -2.3049e+00,
-3.0317e+00, -2.9975e-01],
[-3.6952e+00, -3.1053e+00, -4.1764e+00, ..., -1.4267e+00,
-2.9402e+00, -7.4009e-02],
...,
[-3.0410e+00, -2.9664e+00, -2.6285e+00, ..., -1.8619e+00,
-1.3280e+00, 3.9816e-01],
[-2.2012e+00, 9.2587e-02, -1.0313e+00, ..., -2.1630e-01,
-1.3884e+00, 1.3281e+00],
[-2.2420e+00, -1.2878e+00, -4.2613e-01, ..., -5.1716e-01,
-1.0837e+00, 6.9691e-01]],
[[ 1.1692e+00, 2.9895e-01, 9.0450e-01, ..., 1.7955e+00,
1.7684e+00, 1.8983e-01],
[ 4.0361e+00, 2.9298e+00, 2.8954e+00, ..., 3.4991e+00,
2.9877e+00, 3.8489e+00],
[ 1.4520e+00, 5.7175e-01, 1.0946e+00, ..., 3.1312e+00,
2.0250e+00, 2.0790e+00],
...,
[ 3.0468e+00, 2.3720e+00, 1.5305e+00, ..., 2.5553e+00,
3.3985e+00, 1.6545e+00],
[ 3.7674e+00, 4.0268e+00, 2.7646e+00, ..., 2.6786e+00,
9.8070e-01, 2.7449e+00],
[ 2.3641e+00, 2.3035e+00, 1.6475e+00, ..., 3.7293e-01,
-2.1427e-01, 1.4315e+00]]]], device='cuda:0'),
tensor([[[[ 1.7139e-01, 1.2659e+00, 2.0021e-01, ..., -3.2156e+00,
-2.3186e+00, -1.0192e+00],
[ 4.1578e-01, 8.8708e-01, 1.1224e+00, ..., -1.7035e+00,
-2.1211e+00, -2.0972e-01],
[-4.5404e-01, -3.1127e-01, 4.2462e-01, ..., -1.2169e+00,
-2.6100e+00, -1.7034e-02],
...,
[-5.1676e-01, -1.7536e+00, 3.7133e-01, ..., -2.2176e+00,
-3.4136e+00, 4.2921e-01],
[-1.8430e+00, -1.7065e+00, -1.2304e+00, ..., -3.2447e+00,
-3.4477e+00, 4.6843e-01],
[-1.1495e+00, -3.3604e+00, -3.5604e+00, ..., -3.5996e+00,
-3.3230e+00, -1.5093e+00]],
[[-1.0923e+00, -1.5658e+00, -3.3296e+00, ..., -6.3051e+00,
-7.1084e+00, -1.3463e+00],
[-3.8425e+00, -4.2043e+00, -3.9795e+00, ..., -5.7494e+00,
-8.2180e+00, -2.4374e+00],
[-3.9358e+00, -4.3598e+00, -3.8032e+00, ..., -4.5303e+00,
-7.3892e+00, -2.4506e+00],
...,
[-5.9312e+00, -6.1085e+00, -4.8607e+00, ..., -6.8986e+00,
-8.6202e+00, -2.0071e+00],
[-7.7499e+00, -7.8717e+00, -7.6744e+00, ..., -9.0032e+00,
-1.1037e+01, -4.4009e+00],
[-1.8390e+00, -3.4025e+00, -3.7314e+00, ..., -3.6127e+00,
-3.9545e+00, -7.8567e-01]],
[[-8.0246e-03, 1.2421e+00, 1.7065e+00, ..., 1.5182e+00,
2.4192e+00, 9.8023e-01],
[-4.0224e-02, 1.2810e-02, 9.4498e-01, ..., 1.4726e+00,
2.9219e+00, 7.2813e-02],
[ 5.1299e-01, 1.3008e+00, 1.4682e+00, ..., -4.9641e-02,
1.9308e+00, -2.4623e-01],
...,
[ 2.8128e-01, 9.6918e-01, 1.0193e+00, ..., 1.9801e-01,
1.6045e+00, -1.2653e+00],
[ 4.3351e-01, 7.0155e-01, 1.2230e+00, ..., -9.3992e-02,
1.3752e+00, -1.4693e+00],
[ 5.3419e-01, 1.2018e+00, 1.6812e+00, ..., 2.1968e-01,
-2.5671e-01, -1.9032e+00]],
...,
[[-6.4002e-01, -1.5548e+00, -2.2441e+00, ..., -5.0155e+00,
-5.0922e+00, -7.3653e-01],
[-2.6872e+00, -2.7008e+00, -1.8190e+00, ..., -3.4781e+00,
-4.7643e+00, -1.8524e+00],
[-2.0751e+00, -1.1967e+00, -8.5209e-01, ..., -4.1538e+00,
-4.9745e+00, -1.3200e+00],
...,
[-1.9288e+00, -1.7319e+00, -6.6911e-01, ..., -3.3331e+00,
-5.5654e+00, -1.6073e+00],
[-3.2628e+00, -3.4189e+00, -3.0203e+00, ..., -3.7826e+00,
-5.5901e+00, -9.3287e-01],
[-1.6145e+00, -2.0922e+00, -2.4303e+00, ..., -2.0265e+00,
-2.3160e+00, 3.7234e-01]],
[[-1.3104e+00, -2.2612e+00, -3.9850e+00, ..., -1.4392e+00,
-5.3791e-01, -6.5132e-01],
[-3.5697e+00, -3.6068e+00, -4.9717e+00, ..., -3.2347e+00,
-1.8191e+00, -4.0147e-01],
[-4.5091e+00, -3.8500e+00, -4.2371e+00, ..., -2.0976e+00,
-1.7396e+00, -3.3314e-01],
...,
[-4.7355e+00, -4.0248e+00, -3.6790e+00, ..., -2.4810e+00,
-1.7219e+00, -9.7981e-01],
[-4.1810e+00, -2.2123e+00, -2.3930e+00, ..., -1.1135e+00,
-4.2717e-01, 1.0936e+00],
[-3.2737e+00, -3.4037e+00, -2.5291e+00, ..., -3.0175e-01,
-7.9923e-01, 6.7769e-01]],
[[ 1.1909e+00, -8.0460e-02, 1.1001e+00, ..., 5.0157e-01,
8.1944e-01, -1.3682e+00],
[ 2.2812e+00, 1.1886e+00, 1.4528e+00, ..., 3.5136e+00,
3.1592e+00, 2.3297e+00],
[ 1.4683e+00, 6.9655e-01, 4.8337e-01, ..., 2.6883e+00,
1.6587e+00, 1.3241e+00],
...,
[ 2.5266e+00, 1.7208e+00, 1.3360e+00, ..., 3.6086e+00,
2.2775e+00, 6.8246e-01],
[ 3.0671e+00, 3.9541e+00, 2.5636e+00, ..., 3.0070e+00,
9.4285e-01, 1.8508e+00],
[ 2.0341e+00, 1.6125e+00, 3.3869e-01, ..., 3.0490e-01,
-1.8265e-01, 8.4849e-01]]],
[[[ 9.4702e-01, 7.3642e-01, 8.5187e-02, ..., -1.4686e+00,
-1.3227e+00, -2.4400e-01],
[-4.0434e-01, -3.1157e+00, 2.7641e-02, ..., -1.8243e+00,
1.2945e+00, 1.4810e+00],
[-4.1966e+00, -2.3746e+00, 2.9351e+00, ..., -1.5901e+00,
3.2222e+00, 9.0490e-01],
...,
[ 8.8678e-02, -1.9603e+00, -9.0167e-01, ..., 5.0747e+00,
-1.7720e+00, -1.4031e+00],
[-2.1793e+00, -1.7294e+00, 7.3651e-01, ..., -2.0420e-01,
-4.7112e+00, -8.5203e-01],
[-2.2338e+00, -4.0354e+00, -3.5481e+00, ..., -4.0886e+00,
-4.0294e+00, -1.8503e+00]],
[[-6.4364e-01, -1.4409e+00, -1.8528e+00, ..., -1.3502e+00,
-2.4598e+00, -6.5451e-01],
[-5.3510e+00, -6.1326e+00, -3.9300e+00, ..., -1.3424e+00,
-1.8616e+00, -1.1202e+00],
[-5.0892e+00, -3.8127e+00, -4.1533e-01, ..., -4.0695e-02,
-1.1322e+00, -2.5848e+00],
...,
[-6.7400e+00, -8.8497e+00, -5.2313e+00, ..., -1.3434e+00,
-8.8788e+00, -4.2625e+00],
[-9.5440e+00, -1.0790e+01, -8.3932e+00, ..., -1.1431e+01,
-1.1523e+01, -3.6149e+00],
[-2.4308e+00, -4.8517e+00, -4.1094e+00, ..., -4.7104e+00,
-4.3838e+00, -4.9101e-01]],
[[ 1.9045e-01, 1.1636e+00, 1.8125e+00, ..., -5.7917e-01,
1.3054e+00, 1.5839e+00],
[-2.1709e-01, 8.4909e-01, 3.5132e+00, ..., -6.6931e-01,
1.3446e+00, 1.6168e+00],
[ 2.3751e+00, 2.6007e+00, 4.8825e-01, ..., 2.3716e-02,
1.1583e+00, 1.5460e+00],
...,
[-9.0497e-01, -7.3461e-01, 2.4598e-01, ..., -4.2552e-01,
3.4659e+00, 1.2302e+00],
[-4.8113e-01, 2.3227e-01, -9.9415e-01, ..., -9.1860e-01,
2.2566e+00, -5.7898e-01],
[ 5.4800e-01, 1.2230e+00, 1.4808e+00, ..., 4.3524e-01,
-9.2045e-01, -2.4069e+00]],
...,
[[-8.1082e-01, -1.4463e+00, -1.7042e+00, ..., -1.6230e+00,
-2.2020e+00, -8.9696e-01],
[-3.8601e+00, -2.7161e+00, -2.1850e+00, ..., -2.7959e+00,
-2.7545e+00, -2.8361e+00],
[-3.4170e+00, 1.6285e+00, 1.2899e+00, ..., -2.0953e+00,
-2.5123e+00, -3.4067e+00],
...,
[-2.6657e+00, -4.3290e+00, -5.9876e-02, ..., -6.3132e+00,
-9.1790e+00, -2.7693e+00],
[-3.9894e+00, -6.1413e+00, -2.5001e+00, ..., -6.2110e+00,
-6.3781e+00, -6.5265e-01],
[-3.1458e+00, -3.8238e+00, -3.6003e+00, ..., -3.3976e+00,
-2.8943e+00, -3.5948e-01]],
[[-1.7351e+00, -3.2964e+00, -4.8209e+00, ..., -2.5629e+00,
-2.5761e+00, -1.9135e-01],
[-4.6746e+00, -3.3433e+00, -5.7965e+00, ..., -3.7868e+00,
-4.3741e+00, -1.1755e+00],
[-4.3046e+00, -1.6232e+00, -1.8780e+00, ..., -3.8834e+00,
-4.5563e+00, -1.7276e+00],
...,
[-4.8216e+00, -2.4983e+00, -1.3241e+00, ..., -4.5916e+00,
-6.9882e+00, -1.8255e+00],
[-3.3475e+00, -5.0103e-01, -9.1803e-02, ..., -4.5488e+00,
-3.9562e+00, 4.0552e-01],
[-3.6526e+00, -2.9340e+00, -2.4367e+00, ..., -2.3903e-01,
-6.0006e-01, 6.8108e-01]],
[[ 1.1369e+00, -7.4686e-01, 3.2013e-01, ..., -1.7577e+00,
6.3816e-01, 9.1814e-01],
[ 2.3901e+00, -6.3625e-01, 2.8218e-01, ..., -4.6004e+00,
-9.8740e-01, 3.6186e+00],
[-1.4439e-01, 1.5870e+00, 1.4428e+00, ..., -4.4462e+00,
8.4043e-01, 2.5411e+00],
...,
[ 4.4325e+00, 2.7604e+00, 2.6659e+00, ..., 1.7516e+00,
2.5365e+00, 6.0576e-01],
[ 4.3336e+00, 3.9049e+00, 3.8540e+00, ..., 4.9894e+00,
1.5412e+00, 1.9024e+00],
[ 2.8259e+00, 2.7625e+00, 2.1111e+00, ..., -2.0254e-02,
8.7807e-01, 1.7111e+00]]],
[[[-3.6549e-01, 9.4954e-01, 5.4493e-01, ..., 1.2209e-02,
4.9979e-01, -1.8082e-01],
[-1.0697e-01, -1.1013e-01, 2.4015e+00, ..., -1.4762e-01,
6.5829e-01, -4.9111e-01],
[-1.3233e+00, -3.7003e-01, 1.7105e+00, ..., -8.7346e-01,
1.7707e-01, -8.2572e-01],
...,
[-2.0104e-01, -4.3936e-01, 1.4423e-01, ..., 2.8860e-01,
2.3621e-01, -3.5112e-01],
[-1.4794e+00, -7.4209e-02, -2.0197e-02, ..., 1.4545e-01,
2.7459e-01, -1.0717e+00],
[-1.3318e+00, -2.3793e+00, -1.8517e+00, ..., -3.0537e+00,
-2.8140e+00, -2.6708e+00]],
[[-1.8180e+00, -2.4744e+00, -3.6296e+00, ..., -1.8321e+00,
-3.5208e+00, -2.2171e+00],
[-4.9935e+00, -4.6341e+00, -3.3492e+00, ..., -2.7386e+00,
-3.8404e+00, -3.4343e+00],
[-4.0373e+00, -4.4361e+00, -4.0530e+00, ..., -4.0091e+00,
-2.3863e+00, -2.7621e+00],
...,
[-4.5169e+00, -5.2159e+00, -4.8605e+00, ..., -2.5574e+00,
-3.1325e+00, -2.4812e+00],
[-7.7151e+00, -7.8655e+00, -6.3032e+00, ..., -4.8264e+00,
-5.2613e+00, -4.8563e+00],
[-1.4908e+00, -3.1888e+00, -2.0087e+00, ..., -2.8937e+00,
-2.7554e+00, -1.5738e+00]],
[[-1.6151e-01, 2.3089e+00, 1.7704e+00, ..., 1.5229e+00,
1.6991e+00, 9.0728e-01],
[-4.1209e-01, 1.2111e-01, 1.1891e+00, ..., 9.1124e-01,
1.4167e+00, 8.0436e-01],
[ 2.1444e-02, 1.8161e+00, 1.7990e+00, ..., 1.5435e+00,
1.8114e+00, 1.3485e+00],
...,
[-7.5219e-01, 5.8459e-01, 1.6209e+00, ..., 9.7213e-01,
1.1498e+00, 6.0672e-01],
[-1.2607e+00, 4.1663e-02, 1.1719e+00, ..., 1.2501e+00,
1.0889e+00, 3.8442e-01],
[-3.0500e-01, -3.2902e-01, 1.3131e-01, ..., 1.6183e+00,
1.6598e+00, 5.2162e-01]],
...,
[[-1.5316e+00, -2.2311e+00, -2.2815e+00, ..., -1.4994e+00,
-1.9576e+00, -4.7292e-01],
[-3.7258e+00, -3.4355e+00, -2.2266e+00, ..., -3.2180e+00,
-3.4889e+00, -2.9147e+00],
[-2.7988e+00, -2.2805e+00, -1.9446e+00, ..., -2.5973e+00,
-2.8797e+00, -1.7951e+00],
...,
[-9.2929e-01, -2.4867e+00, -1.3040e+00, ..., -1.1733e+00,
-2.2030e+00, -8.5630e-01],
[-2.0959e+00, -3.9786e+00, -1.6419e+00, ..., -1.7304e+00,
-2.9069e+00, -1.4738e+00],
[-1.5147e+00, -1.5733e+00, -8.6349e-01, ..., -1.8624e+00,
-2.1098e+00, -5.4013e-01]],
[[-1.9302e+00, -1.9348e+00, -3.2254e+00, ..., -4.4483e-01,
-1.3487e+00, -7.1292e-01],
[-4.1653e+00, -2.0242e+00, -2.8891e+00, ..., -1.7002e+00,
-3.7273e+00, -1.8908e+00],
[-3.7677e+00, -2.6435e+00, -4.4046e+00, ..., -1.2278e+00,
-4.2731e+00, -2.4887e+00],
...,
[-2.9008e+00, -2.9845e+00, -3.2560e+00, ..., -1.3243e+00,
-2.9269e+00, -2.1488e+00],
[-2.0120e+00, -3.6361e-01, -1.7014e+00, ..., -5.8543e-01,
-3.0268e+00, -1.7876e+00],
[-2.2012e+00, -1.6080e+00, -1.7644e+00, ..., -1.5551e+00,
-2.6841e+00, -1.1631e+00]],
[[ 1.2933e+00, 2.8250e-01, 8.2280e-01, ..., 6.0877e-02,
1.2363e+00, 6.3941e-01],
[ 4.1425e+00, 2.9106e+00, 2.2494e+00, ..., 3.6281e-01,
1.0974e+00, 2.4641e+00],
[ 1.6128e+00, 3.0424e-01, 3.9490e-01, ..., 3.5100e-01,
1.0434e-01, 8.6224e-01],
...,
[ 3.0176e+00, 2.3129e+00, 2.1086e+00, ..., 1.5159e+00,
6.4713e-01, 5.2582e-01],
[ 3.8279e+00, 3.9984e+00, 2.6198e+00, ..., 1.2575e+00,
6.9775e-01, 1.1932e+00],
[ 2.4305e+00, 2.3620e+00, 1.5433e+00, ..., -1.2749e-01,
-2.8160e-01, -9.0142e-02]]],
[[[ 3.0624e-01, 1.2742e+00, 2.5391e-01, ..., 4.3127e-01,
-1.1790e+00, -9.7013e-01],
[ 4.4650e-01, 9.6555e-01, 1.4926e+00, ..., -4.6284e+00,
-4.2572e+00, -1.9112e+00],
[-3.7054e-01, -8.8676e-01, -1.6531e-02, ..., -1.3114e+01,
-2.7047e-01, 2.6289e-01],
...,
[-6.3110e-01, -9.1476e+00, -5.7495e-01, ..., -5.2377e+00,
-3.1051e+00, -1.1680e+00],
[-4.0389e-01, -3.1562e+00, -3.5427e+00, ..., -4.2398e+00,
-3.8064e+00, -1.2757e+00],
[-3.1157e-01, -2.5317e+00, -4.1924e+00, ..., -4.4368e+00,
-3.3251e+00, -1.1681e+00]],
[[-1.2107e+00, -1.8934e+00, -3.0935e+00, ..., -5.8413e+00,
-6.9966e+00, -2.1900e+00],
[-4.1713e+00, -4.4594e+00, -3.3198e+00, ..., -1.3357e+01,
-9.1586e+00, -1.7258e+00],
[-3.9099e+00, -4.5478e+00, -3.2189e+00, ..., -1.5597e+01,
-5.9508e+00, 2.7810e-01],
...,
[-2.3683e+00, -7.8671e+00, -6.2717e+00, ..., -9.3964e+00,
-6.7455e+00, -2.0902e+00],
[-6.4965e+00, -7.5556e+00, -1.1597e+01, ..., -9.9891e+00,
-9.1486e+00, -3.8282e+00],
[-1.1922e+00, -2.1697e+00, -1.5851e+00, ..., -4.6890e+00,
-3.3547e+00, -4.6842e-01]],
[[ 9.0974e-03, 1.3291e+00, 1.6403e+00, ..., -1.0411e-01,
1.8692e+00, 8.8831e-01],
[ 3.4037e-02, 6.2824e-02, 5.2426e-01, ..., -8.1962e-01,
3.2867e+00, 1.7914e+00],
[ 7.3986e-01, 1.4058e+00, 1.4871e+00, ..., 3.3379e+00,
3.3693e+00, 1.9980e+00],
...,
[-1.7764e+00, 4.0038e-01, -5.3561e+00, ..., 3.1432e+00,
-4.2697e-01, -5.5190e-01],
[-2.6621e-01, -1.8593e-01, 2.3769e-01, ..., -2.0249e+00,
-2.9029e+00, -5.9615e-01],
[ 5.3349e-01, -1.4993e-01, 1.3662e+00, ..., -1.5411e+00,
-8.1087e-01, -1.9102e+00]],
...,
[[-7.1707e-01, -1.6172e+00, -2.1124e+00, ..., -3.1396e+00,
-4.5672e+00, -1.6617e+00],
[-2.8436e+00, -2.6651e+00, -1.3481e+00, ..., -7.4118e+00,
-4.6513e+00, -2.6541e+00],
[-2.2567e+00, -1.2573e+00, -1.5995e+00, ..., -8.1231e+00,
2.7886e-01, 2.8803e-01],
...,
[-1.8505e+00, -5.0377e+00, 1.1021e-01, ..., -3.9620e+00,
-2.2715e+00, -3.9743e+00],
[-2.2918e+00, -4.4706e+00, -3.0819e+00, ..., -8.5872e-01,
-3.2901e+00, -1.0235e+00],
[-8.0656e-01, -1.0850e+00, 6.9657e-01, ..., -7.2357e-01,
-1.9987e+00, 9.1820e-01]],
[[-1.4745e+00, -2.4302e+00, -3.9721e+00, ..., -2.5198e+00,
-2.3342e+00, 5.1954e-02],
[-3.9408e+00, -3.5203e+00, -4.3130e+00, ..., -2.7782e+00,
-3.9034e+00, -5.6934e-01],
[-4.6596e+00, -3.5668e+00, -2.7815e+00, ..., -4.6928e+00,
-5.1424e+00, -1.9337e-01],
...,
[-4.4290e+00, -4.5606e+00, 7.6228e+00, ..., -5.7211e+00,
-2.6616e+00, 4.3820e-02],
[-2.8613e+00, -4.6196e+00, -1.0959e+00, ..., -1.2238e+00,
-1.4190e+00, 1.7278e+00],
[-2.2801e+00, -2.6010e+00, -1.2056e-01, ..., 1.2507e-01,
-7.0955e-01, 8.1282e-01]],
[[ 1.2303e+00, 1.7025e-02, 1.1609e+00, ..., 5.7641e-01,
6.7170e-01, -8.2765e-02],
[ 2.3643e+00, 1.3174e+00, 1.2138e+00, ..., 2.6618e+00,
7.5481e-01, 1.5034e+00],
[ 1.3648e+00, 8.2245e-01, 2.6368e-01, ..., -1.2010e+00,
5.2371e+00, 2.3093e+00],
...,
[ 1.2971e+00, -9.2164e-01, 8.9102e+00, ..., -1.5253e+00,
8.3036e-01, 1.1023e+00],
[ 2.9099e+00, 2.8788e+00, 5.7873e+00, ..., 5.8628e-02,
1.4197e+00, 2.0674e+00],
[ 2.8346e+00, 1.3947e+00, -1.4061e+00, ..., 5.8620e-01,
-7.4259e-02, 7.7674e-01]]],
[[[ 8.7094e-01, 1.4295e+00, -8.9901e-01, ..., 1.4559e+00,
4.1187e+00, 1.5228e+00],
[ 1.1497e+00, -2.8400e+00, 2.3777e-01, ..., 5.1499e-01,
8.0919e-01, 1.8634e+00],
[ 3.8516e-01, -4.3159e+00, 2.8132e+00, ..., 5.3472e-01,
-3.0069e+00, 1.7986e+00],
...,
[-5.1163e-01, 8.8650e-01, -1.3492e+01, ..., -9.1792e+00,
-1.2058e+01, -5.0485e+00],
[-5.8381e-02, -3.7900e-01, -1.1928e+01, ..., -6.9892e+00,
-7.7970e+00, -2.3332e+00],
[-1.5906e+00, -3.7509e+00, -5.9525e+00, ..., -5.6499e+00,
-4.9054e+00, -2.5780e+00]],
[[ 1.3816e+00, 1.2312e-01, -6.2404e-01, ..., 6.9369e-01,
1.1748e-01, -3.8660e+00],
[ 7.4862e-01, -7.4547e+00, -6.6612e-01, ..., 6.2643e-02,
-4.0360e+00, -2.1796e+00],
[-3.2249e-01, -5.3112e+00, 2.7565e+00, ..., -1.1432e+00,
-1.4098e+01, -2.3387e+00],
...,
[-6.5039e+00, -3.5391e+00, -1.5274e+01, ..., -2.4272e+01,
-2.7116e+01, -1.1072e+01],
[-7.6042e+00, -6.8184e+00, -1.5217e+01, ..., -7.9323e+00,
-1.1398e+01, -4.8355e+00],
[-2.8492e+00, -2.5053e+00, -4.1147e+00, ..., -5.3614e+00,
-6.4280e+00, -1.6730e+00]],
[[-1.2469e+00, -1.6911e+00, -9.9858e-01, ..., -1.1332e+00,
-3.1992e+00, -2.6247e+00],
[-2.6581e+00, -5.5590e+00, -7.9258e+00, ..., -1.1397e+01,
-1.1285e+01, -6.0501e+00],
[-1.4802e+00, -4.2079e+00, -3.1626e+00, ..., -5.8997e+00,
-1.1217e+01, -4.6232e+00],
...,
[-8.3796e-01, -6.1269e+00, 1.3422e+00, ..., -6.5914e+00,
-2.5561e+00, -5.1105e+00],
[-1.0976e+00, -5.6844e+00, 3.4325e-01, ..., -4.3695e-01,
2.3192e+00, -3.0802e+00],
[ 2.1657e-01, -3.5679e+00, -3.0148e+00, ..., -1.8566e+00,
-3.1991e+00, -4.1091e+00]],
...,
[[ 1.6868e+00, 8.8763e-01, 6.7852e-01, ..., 2.4830e+00,
2.4967e+00, 1.4154e+00],
[ 4.2329e-01, -2.7974e+00, 2.1235e+00, ..., 4.1724e+00,
1.5066e+00, 2.3634e+00],
[-8.5328e-01, -1.4297e+00, 1.2233e+00, ..., 3.5026e+00,
2.7497e+00, 2.7179e+00],
...,
[-8.2499e-01, -2.0869e+00, -6.7387e+00, ..., -1.4858e+01,
-1.6733e+01, -4.3562e+00],
[-1.2774e+00, -3.3256e+00, -6.7055e+00, ..., -6.8576e-01,
-4.4366e+00, -4.3200e-01],
[-1.6355e+00, -2.6451e+00, -2.3251e+00, ..., -5.2697e+00,
-4.8670e+00, -5.2206e-01]],
[[-3.2907e-01, 5.2941e-01, -3.2704e+00, ..., 1.2647e+00,
-6.6964e-01, 2.8595e+00],
[-1.0617e+00, 8.9584e-02, 4.7827e-01, ..., 1.4486e+00,
7.4692e-01, 1.5733e+00],
[-3.6310e-01, 1.2105e-01, 2.3101e+00, ..., -4.8843e+00,
-1.9146e+00, -5.6181e-01],
...,
[-2.6463e+00, -8.9452e+00, -7.2862e+00, ..., -1.1159e+01,
-9.7164e+00, -9.0830e-01],
[-2.4779e+00, -6.4391e+00, -5.7169e+00, ..., -1.7849e+00,
-3.3011e+00, 6.8620e-01],
[-1.7894e+00, -1.7537e+00, -1.5417e+00, ..., 2.5591e+00,
9.8581e-01, 1.7500e+00]],
[[ 3.9955e-01, -6.1875e-01, -3.7291e-01, ..., 3.7563e-03,
1.5111e+00, 1.4263e+00],
[-1.3251e+00, -6.5210e+00, -2.8511e+00, ..., -7.6473e-01,
-3.0058e+00, 2.6404e+00],
[-2.6154e+00, -8.5216e+00, 1.2910e-01, ..., -2.1172e+00,
-3.6820e+00, 1.9953e+00],
...,
[ 2.5279e+00, 5.6595e+00, 5.1151e+00, ..., 1.3264e+01,
8.5969e+00, 4.8355e+00],
[ 3.3179e+00, 5.2154e+00, 3.3598e+00, ..., 3.8616e+00,
-1.4408e+00, 2.7237e+00],
[ 1.9340e+00, 9.1961e-01, 9.1880e-01, ..., 1.6520e-01,
-7.9958e-01, 7.0594e-01]]]], device='cuda:0'),
tensor([[[[-1.7191e+00, -1.4650e+00, -3.5563e+00, ..., 1.4144e-01,
-1.0421e+00, -1.3987e+00],
[-2.1626e+00, -2.8867e+00, -4.7857e+00, ..., 1.9503e+00,
-7.9700e-02, -2.1453e+00],
[-1.2214e+00, -3.1615e+00, -2.2590e+00, ..., 6.2472e-01,
-5.9125e-01, -2.6076e+00],
...,
[-9.3821e-01, -6.4148e+00, -5.2224e+00, ..., -6.2796e+00,
-3.8568e+00, -9.3045e-01],
[-2.0718e+00, -2.4819e+00, -1.2075e+00, ..., -4.5074e+00,
-2.9110e+00, -2.5096e-01],
[-1.8378e+00, -3.1723e+00, -4.2463e+00, ..., -5.1184e+00,
-3.5447e+00, -1.4485e+00]],
[[-2.9507e+00, -2.0710e-01, -1.0930e+00, ..., -3.2938e+00,
-6.5827e+00, -3.3768e+00],
[-4.5489e+00, -3.9285e+00, -2.2357e+00, ..., -2.3296e+00,
-7.6937e+00, -5.2493e+00],
[-5.3469e+00, -9.1600e+00, -1.2533e+00, ..., -1.5722e+00,
-5.1242e+00, -3.4562e+00],
...,
[-4.3303e+00, -6.4395e+00, -2.3761e+00, ..., -1.2748e+01,
-8.5248e+00, -1.7087e+00],
[-7.6000e+00, -8.3454e+00, -1.3608e+01, ..., -7.7059e+00,
-1.0053e+01, -3.8115e+00],
[-1.9074e+00, -4.0433e+00, -1.6348e+00, ..., -3.5988e+00,
-3.7388e+00, -9.0137e-01]],
[[-1.5893e-02, -3.0354e-01, -3.1845e+00, ..., 2.6822e+00,
2.0574e+00, 8.2309e-01],
[-1.8937e+00, -5.4549e+00, -4.8375e+00, ..., 2.4059e+00,
2.8785e+00, 9.2155e-01],
[-2.8633e+00, -7.3600e+00, -5.6523e+00, ..., 2.5741e+00,
3.2900e+00, 1.8395e+00],
...,
[ 4.1254e-01, 3.5313e-01, -4.8639e+00, ..., 1.4538e+00,
4.4171e-01, -1.7522e+00],
[-1.2721e+00, 1.1485e+00, 4.0905e-02, ..., -6.0933e-01,
-9.7726e-01, -2.6759e-01],
[-8.4482e-01, -3.9007e-02, 1.8179e+00, ..., -2.4182e+00,
-7.8269e-01, -1.4322e+00]],
...,
[[-1.8560e+00, -2.7486e+00, -1.5859e+00, ..., -2.5786e+00,
-4.0429e+00, -1.6481e+00],
[-4.5834e+00, -7.4926e+00, -4.4976e+00, ..., -2.9743e+00,
-4.5184e+00, -4.3039e+00],
[-2.4737e+00, -4.6360e+00, 3.5702e-01, ..., -3.6940e+00,
-4.6050e+00, -3.6747e+00],
...,
[-2.3942e+00, -4.8887e+00, 5.7795e-01, ..., -1.0019e+01,
-5.8240e+00, -3.6860e+00],
[-2.7102e+00, -6.0285e+00, -7.0376e+00, ..., -3.1271e+00,
-6.5421e+00, -2.6050e+00],
[-2.4287e+00, -1.4965e+00, 6.8647e-01, ..., -3.0350e+00,
-3.4000e+00, 3.1337e-01]],
[[-1.5999e+00, -5.6200e+00, -6.9213e+00, ..., -3.1361e+00,
-2.1146e+00, -4.4581e-01],
[-2.6865e+00, -4.9475e+00, -6.7437e+00, ..., -4.0738e+00,
-4.5424e+00, -1.9409e+00],
[-2.0372e+00, 1.1010e+00, 7.9377e-01, ..., -3.6417e+00,
-3.3140e+00, -2.1197e+00],
...,
[-3.2145e+00, -1.1190e+00, 3.5771e-01, ..., -6.8158e+00,
-4.3168e+00, -6.0993e-02],
[-2.6639e+00, -7.5723e-01, -2.2746e+00, ..., -1.2085e+00,
-5.9711e-01, 1.4802e+00],
[-3.0595e+00, -1.2434e+00, -7.6289e-02, ..., -5.2834e-02,
-8.4031e-01, 7.8058e-01]],
[[ 2.3438e+00, 1.3575e-01, -3.2496e+00, ..., 1.8590e+00,
2.0194e+00, 4.2508e-01],
[ 3.5445e+00, -1.5599e-01, -3.4183e+00, ..., 2.4920e+00,
3.1484e+00, 2.9892e+00],
[ 3.8251e+00, 3.4281e+00, -1.3108e+00, ..., 1.1262e+00,
1.8719e+00, 8.0404e-01],
...,
[ 2.1610e+00, -1.3637e+00, -5.9028e-01, ..., 3.5995e+00,
2.8046e+00, 2.0538e+00],
[ 3.8734e+00, 2.5384e+00, 6.8856e+00, ..., 1.0518e+00,
2.1601e+00, 1.5312e+00],
[ 3.2306e+00, 2.2873e+00, 3.1961e-01, ..., 1.8482e+00,
3.1842e-01, 8.3935e-01]]],
[[[-2.3859e-01, 8.2660e-01, 8.6282e-01, ..., 7.1552e-01,
7.2637e-01, 3.2014e-02],
[ 1.0421e-01, -3.0213e-01, 3.2477e+00, ..., 2.7407e+00,
7.2514e-01, -7.0478e-01],
[-1.2874e+00, -6.3719e-01, 3.3541e+00, ..., 2.5374e+00,
3.5575e-01, -1.3052e+00],
...,
[-3.1772e-01, -6.3114e+00, 1.5283e+00, ..., 7.9256e-01,
6.3809e-01, -3.2936e-01],
[-2.3008e+00, -5.7646e+00, 1.5834e+00, ..., 5.1747e-01,
5.5145e-01, -1.1659e+00],
[-2.0652e+00, -4.9371e+00, 7.2653e-01, ..., -3.1815e+00,
-2.8681e+00, -2.7684e+00]],
[[-1.5911e+00, -2.3194e+00, -3.1611e+00, ..., -2.8529e+00,
-4.8670e+00, -2.0119e+00],
[-4.9376e+00, -4.6335e+00, -4.7104e+00, ..., -6.0255e-01,
-7.4865e+00, -3.5209e+00],
[-3.8846e+00, -4.0221e+00, -5.7584e+00, ..., -1.2391e+00,
-7.1128e+00, -2.4710e+00],
...,
[-4.3697e+00, -8.7227e+00, 1.0928e+00, ..., -2.5454e+00,
-2.9033e+00, -2.5604e+00],
[-7.9425e+00, -1.1247e+01, -4.1565e-01, ..., -4.9100e+00,
-5.1540e+00, -5.0332e+00],
[-1.4451e+00, -5.2390e+00, -1.3242e+00, ..., -3.0310e+00,
-2.8415e+00, -1.6340e+00]],
[[-1.9675e-01, 2.2413e+00, 1.4311e+00, ..., 2.2822e+00,
1.2234e+00, 6.4077e-01],
[-4.8027e-01, -1.7199e-03, 1.1065e+00, ..., 3.6020e+00,
1.8993e+00, 3.9605e-01],
[-9.6204e-02, 1.5645e+00, 2.1997e+00, ..., 4.3768e+00,
2.3109e+00, 1.0172e+00],
...,
[-1.9815e+00, 6.1624e-01, -7.7206e+00, ..., 1.6172e+00,
1.0658e+00, 6.5609e-01],
[-2.1822e+00, -2.2722e+00, -9.0478e+00, ..., 1.4565e+00,
8.5746e-01, 5.1944e-01],
[-1.1799e+00, -1.2331e+00, -6.7128e+00, ..., 1.6807e+00,
1.5105e+00, 5.6824e-01]],
...,
[[-1.3045e+00, -2.1053e+00, -4.9572e-01, ..., -1.7599e+00,
-2.3890e+00, -1.0332e+00],
[-3.6365e+00, -3.5388e+00, -1.0892e+00, ..., -4.3113e+00,
-3.9269e+00, -4.1784e+00],
[-2.7463e+00, -2.3867e+00, -5.6877e-01, ..., -7.2020e+00,
-1.4076e+00, -3.7281e+00],
...,
[-1.1613e+00, -3.9189e+00, 2.2401e+00, ..., -1.7692e+00,
-1.8994e+00, -9.3052e-01],
[-2.0819e+00, -5.3853e+00, -4.5619e-01, ..., -1.6797e+00,
-2.6874e+00, -1.6246e+00],
[-2.1759e+00, -3.9241e+00, -4.2564e-01, ..., -1.9822e+00,
-2.2251e+00, -6.3205e-01]],
[[-1.8040e+00, -1.7347e+00, -1.6055e+00, ..., 4.5993e-01,
-2.2368e-01, -2.8463e-01],
[-4.1105e+00, -1.9930e+00, -2.0468e+00, ..., -3.4602e+00,
-2.7280e+00, -1.5760e+00],
[-3.9418e+00, -2.5487e+00, -4.8561e+00, ..., -5.9253e+00,
-3.0598e+00, -1.9835e+00],
...,
[-3.6185e+00, -2.2164e+00, -3.2013e+00, ..., -1.1772e+00,
-3.5457e+00, -2.2548e+00],
[-1.3759e+00, 1.0581e+00, 7.9064e-02, ..., -9.9142e-01,
-3.3868e+00, -2.0366e+00],
[-1.7805e+00, 4.5415e-01, 2.9189e+00, ..., -1.6684e+00,
-2.7606e+00, -1.3032e+00]],
[[ 1.2183e+00, 1.9418e-01, 6.9845e-01, ..., 3.3519e+00,
1.6368e+00, -3.0965e-02],
[ 4.0569e+00, 2.7792e+00, 2.4168e+00, ..., 4.2431e+00,
1.9441e+00, 2.2197e+00],
[ 1.5203e+00, -3.8938e-02, 9.3107e-01, ..., 3.6217e+00,
4.7014e-01, 1.1965e+00],
...,
[ 2.8253e+00, 2.0620e+00, 3.7176e+00, ..., 5.4785e-01,
4.2380e-01, 5.9932e-01],
[ 4.1402e+00, 2.8380e+00, 2.4171e+00, ..., 1.0304e+00,
9.8527e-01, 1.1929e+00],
[ 1.8864e+00, 1.5273e+00, 3.5315e+00, ..., -1.6073e-01,
-2.8521e-01, -1.1999e-01]]],
[[[ 2.9651e-01, 1.4231e+00, 5.1110e-01, ..., -1.4213e+00,
6.9735e-01, -3.1706e-01],
[ 4.1079e-01, 2.1068e-01, 5.6102e-01, ..., -1.8313e-02,
2.5661e+00, 4.6495e-01],
[-5.6393e-01, 9.0981e-02, 8.2329e-01, ..., -1.7168e+00,
3.1058e+00, 1.2539e+00],
...,
[-1.8749e-01, -5.6843e-01, -6.4001e+00, ..., 4.4773e-01,
-1.7721e+00, -2.2029e+00],
[-6.4336e-01, -2.7645e-01, -4.8110e+00, ..., -4.1830e-01,
-1.4006e+00, -2.2112e+00],
[-2.6877e+00, -4.8607e+00, -5.5334e+00, ..., -4.9071e+00,
-4.7179e+00, -4.0418e+00]],
[[-7.6420e-01, -2.2798e+00, -4.1979e+00, ..., -1.3244e+00,
-3.3389e+00, -8.2529e-01],
[-4.7294e+00, -6.1088e+00, -5.0159e+00, ..., -6.1579e-01,
-3.4757e+00, -3.3328e+00],
[-3.6158e+00, -4.3140e+00, -3.4728e+00, ..., -8.6215e-01,
-1.3982e+00, -2.4062e+00],
...,
[-6.0346e+00, -5.7544e+00, -9.1966e+00, ..., -5.0276e+00,
-8.2831e+00, -3.9948e+00],
[-7.5943e+00, -7.6556e+00, -9.2349e+00, ..., -7.9132e+00,
-8.5798e+00, -6.3513e+00],
[-2.7910e+00, -4.7182e+00, -4.2749e+00, ..., -4.9557e+00,
-4.2201e+00, -2.5750e+00]],
[[-3.3442e-01, 2.1583e+00, 1.1963e+00, ..., 2.3306e+00,
3.6175e+00, 1.4459e+00],
[-2.1679e-01, 1.1004e+00, 8.7315e-01, ..., -4.3094e+00,
2.6935e+00, 1.0506e+00],
[-8.8802e-02, 1.9530e+00, 1.4540e+00, ..., -2.8056e+00,
3.9528e+00, 2.0745e+00],
...,
[ 1.1449e+00, 6.8205e-01, 1.2239e+00, ..., 3.8401e+00,
7.3238e-01, 1.6465e-02],
[ 5.5523e-01, 3.1750e-01, -2.6343e-01, ..., 1.9081e+00,
-4.5800e-02, 2.5056e-02],
[ 1.1871e+00, 1.4551e+00, 4.7777e-01, ..., 1.7373e+00,
1.0794e+00, 4.8348e-01]],
...,
[[-8.1151e-01, -1.8646e+00, -1.8207e+00, ..., -1.2155e+00,
-1.7654e+00, -1.0146e+00],
[-3.2126e+00, -4.3437e+00, -2.4830e+00, ..., -1.4162e+00,
-3.8322e+00, -2.5771e+00],
[-1.5156e+00, -2.1944e+00, -8.1593e-01, ..., -1.1523e+00,
-6.2797e+00, -2.1866e+00],
...,
[-2.3797e+00, -1.9124e+00, -2.6844e+00, ..., -5.6444e+00,
-3.4959e+00, -2.6199e+00],
[-3.0251e+00, -3.2366e+00, -3.0670e+00, ..., -3.8663e+00,
-3.8880e+00, -2.7528e+00],
[-3.6283e+00, -3.8433e+00, -3.9714e+00, ..., -3.9327e+00,
-3.7941e+00, -2.2207e+00]],
[[-1.7010e+00, -2.2624e+00, -2.9297e+00, ..., -9.9037e-01,
-1.5860e+00, -1.4394e-01],
[-4.3241e+00, -3.5874e+00, -2.8205e+00, ..., 5.4995e-01,
-2.7886e+00, 1.4155e-01],
[-3.3187e+00, -2.9536e+00, -2.7157e+00, ..., 2.1733e-01,
-4.6202e+00, -8.7537e-01],
...,
[-3.4955e+00, -3.4514e+00, -4.1229e+00, ..., -5.8682e+00,
-3.4520e+00, -1.6578e+00],
[-3.0939e+00, -1.6354e+00, -2.7443e+00, ..., -4.6835e+00,
-3.1842e+00, -1.0672e+00],
[-3.9437e+00, -3.2066e+00, -3.6623e+00, ..., -3.2990e+00,
-3.1008e+00, -1.1018e+00]],
[[ 1.0745e+00, 5.9072e-01, 6.7256e-01, ..., -3.0158e-01,
2.1925e+00, 6.5941e-02],
[ 3.4828e+00, 3.2458e+00, 2.2027e+00, ..., -9.5485e-01,
3.5350e+00, 4.1577e+00],
[ 1.4789e+00, 6.3553e-01, 1.6823e-01, ..., -3.8596e+00,
2.1012e+00, 2.7622e+00],
...,
[ 3.2986e+00, 1.5114e+00, 2.2030e+00, ..., 2.7493e+00,
9.3645e-01, 1.0819e+00],
[ 4.1940e+00, 3.6135e+00, 3.0134e+00, ..., 2.8284e+00,
1.7772e+00, 2.1257e+00],
[ 2.3169e+00, 7.5210e-01, -1.1371e-01, ..., -4.4589e-01,
1.5639e-01, -8.3155e-02]]],
[[[ 6.9516e-02, -1.0974e+00, -3.9115e+00, ..., -3.2226e+00,
-3.8904e+00, -1.4826e+00],
[ 4.7583e-01, -1.5011e+00, -4.8986e+00, ..., -2.2309e+00,
-1.2683e+00, -6.4261e-01],
[ 6.0607e-02, -1.3236e+00, -3.9539e+00, ..., 1.9925e+00,
-6.3417e-02, -1.3912e+00],
...,
[-6.5198e-01, -1.9950e+00, -1.3669e-01, ..., -6.0675e-01,
-8.2577e-02, -4.3264e-01],
[-9.5251e-01, -1.2353e+00, -2.4935e+00, ..., -1.4328e+00,
-4.6601e-01, -1.7324e+00],
[-2.7405e+00, -5.6615e+00, -4.8818e+00, ..., -4.8389e+00,
-4.2037e+00, -3.8879e+00]],
[[-1.5374e+00, -2.6493e+00, -5.2904e+00, ..., -7.6888e+00,
-9.0413e+00, -2.0245e+00],
[-4.7955e+00, -4.7217e+00, -4.9227e+00, ..., -6.4639e+00,
-8.6318e+00, -3.0734e+00],
[-3.7223e+00, -5.3266e+00, -8.7862e+00, ..., -3.4837e+00,
-1.0672e+01, -5.7017e+00],
...,
[-5.7599e+00, -5.7760e+00, -4.5225e+00, ..., -4.4648e+00,
-4.6329e+00, -3.5776e+00],
[-7.7222e+00, -8.9391e+00, -7.6310e+00, ..., -7.9256e+00,
-8.2701e+00, -6.6454e+00],
[-2.6914e+00, -4.5996e+00, -3.2325e+00, ..., -4.1571e+00,
-4.0908e+00, -2.6079e+00]],
[[ 4.8633e-01, 2.4211e+00, 2.6697e+00, ..., 1.4100e+00,
3.1239e+00, 9.1303e-01],
[-5.9492e-01, -1.1948e-01, 9.5956e-01, ..., 2.1825e+00,
2.7661e+00, 5.0481e-03],
[ 7.7531e-01, 1.0923e+00, -5.9292e-01, ..., 3.5049e+00,
1.3266e+00, -6.6040e-01],
...,
[ 4.9215e-01, 1.1235e+00, -2.0258e-01, ..., -1.2295e-01,
2.7823e-01, -1.4499e-01],
[ 9.1799e-01, 9.0402e-01, 5.8739e-01, ..., 2.0028e-01,
-6.2172e-02, -1.0252e-01],
[ 1.4564e+00, 2.0170e+00, 1.3334e+00, ..., 1.7977e+00,
1.2436e+00, 1.7236e-01]],
...,
[[-9.0433e-01, -3.3439e+00, -2.6785e+00, ..., -5.2204e+00,
-6.7620e+00, -1.5643e+00],
[-2.5170e+00, -4.8422e+00, -2.1423e+00, ..., -4.4498e+00,
-5.8277e+00, -2.8493e+00],
[-1.7936e+00, -3.8301e+00, -3.4041e+00, ..., -4.1806e+00,
-5.5703e+00, -3.3216e+00],
...,
[-2.3267e+00, -1.9843e+00, 4.7610e-01, ..., -9.7692e-01,
-2.6111e+00, -1.1357e+00],
[-3.5547e+00, -3.9487e+00, -2.4092e+00, ..., -2.1949e+00,
-4.4496e+00, -2.4276e+00],
[-3.5911e+00, -3.7703e+00, -2.6625e+00, ..., -3.0965e+00,
-3.7441e+00, -1.7226e+00]],
[[-1.8640e+00, -3.3804e+00, -1.4398e+00, ..., -2.6916e+00,
-1.4086e+00, 5.7890e-01],
[-3.7797e+00, -2.9886e+00, -6.6106e-02, ..., -1.5598e+00,
-2.2934e+00, -2.2132e-01],
[-2.7555e+00, -1.1437e+00, 4.6693e-02, ..., -1.5865e+00,
-9.0703e-01, -1.6608e-02],
...,
[-2.9832e+00, -1.9708e+00, -2.6267e+00, ..., 8.7404e-02,
-1.5931e+00, -8.8647e-01],
[-3.7489e+00, -1.3912e+00, -1.5695e+00, ..., -1.3271e+00,
-2.7679e+00, -1.1390e+00],
[-4.4521e+00, -3.2003e+00, -1.9630e+00, ..., -2.2651e+00,
-3.2859e+00, -1.2521e+00]],
[[ 1.5271e+00, 6.9215e-01, 6.2822e-01, ..., 8.7797e-01,
1.6597e+00, -1.2823e-02],
[ 3.0246e+00, 3.0016e+00, 2.1668e+00, ..., 4.0261e+00,
3.3343e+00, 3.6413e+00],
[ 2.0826e+00, 1.9097e+00, 1.5323e+00, ..., 5.3832e+00,
3.7683e+00, 1.3638e+00],
...,
[ 2.4052e+00, 7.2685e-01, 1.4102e+00, ..., 1.9742e+00,
1.9551e+00, 1.6231e+00],
[ 4.1762e+00, 4.4834e+00, 1.8710e+00, ..., 3.0316e+00,
2.1616e+00, 2.2574e+00],
[ 2.0024e+00, 6.8431e-01, -9.5589e-01, ..., -7.8726e-01,
-4.1441e-01, -2.0383e-01]]],
[[[ 6.8475e-01, 5.7080e-01, -6.7602e-01, ..., 1.4729e-01,
2.2471e-01, -1.3391e-01],
[ 1.1574e+00, 3.8577e-01, 7.7974e-01, ..., 3.5367e-01,
1.9818e-01, -4.5316e-01],
[-1.1425e-01, -2.0532e-01, 1.0191e+00, ..., 7.3383e-02,
-2.9826e-01, -7.6723e-01],
...,
[-8.3811e-01, -4.1623e+00, -3.7907e+00, ..., -4.8378e-01,
-1.4564e+00, -7.0887e-01],
[-3.9094e-01, -9.7168e-01, 4.5698e-01, ..., -1.1866e+00,
-5.6414e-01, -1.4949e+00],
[-2.6499e+00, -2.0815e+00, -1.2961e+00, ..., -3.6733e+00,
-3.3383e+00, -2.7845e+00]],
[[-1.3001e+00, -3.7536e+00, -6.5760e+00, ..., -2.6882e+00,
-3.7213e+00, -2.0071e+00],
[-4.8281e+00, -5.8838e+00, -5.7268e+00, ..., -3.3329e+00,
-4.2661e+00, -3.2734e+00],
[-4.2913e+00, -5.3922e+00, -4.9668e+00, ..., -2.0223e+00,
-2.6703e+00, -2.4652e+00],
...,
[ 2.1606e+00, 7.0871e-01, -5.2618e+00, ..., -6.2134e+00,
-5.0381e+00, -2.5627e+00],
[-3.9443e+00, 1.1211e+00, -4.4584e-02, ..., -4.9791e+00,
-4.9360e+00, -4.8021e+00],
[-2.8459e+00, -6.0682e+00, -1.9320e+00, ..., -3.7390e+00,
-2.8390e+00, -1.5811e+00]],
[[ 8.0788e-01, 2.9532e+00, 1.8766e+00, ..., 1.7940e+00,
1.4035e+00, 8.9143e-01],
[-4.7259e-02, 1.5700e+00, 1.4944e+00, ..., 1.3006e+00,
1.1205e+00, 7.8723e-01],
[ 9.9346e-01, 2.2932e+00, 9.5304e-01, ..., 1.8045e+00,
1.5294e+00, 1.2580e+00],
...,
[-2.1292e+00, 2.5806e+00, 1.0131e+00, ..., 4.8886e-01,
1.3065e+00, 4.2983e-01],
[-5.2005e+00, -8.1247e-01, 1.5701e+00, ..., 8.0805e-01,
-1.9304e-01, 6.9215e-02],
[-1.6150e+00, -1.7509e+00, 1.2954e+00, ..., 8.5711e-01,
1.2532e+00, 8.4410e-01]],
...,
[[-8.8804e-01, -3.2256e+00, -2.4155e+00, ..., -1.5054e+00,
-2.1121e+00, -4.4474e-01],
[-2.6747e+00, -3.9557e+00, -1.3317e+00, ..., -2.2722e+00,
-3.8740e+00, -2.8970e+00],
[-2.3239e+00, -2.7913e+00, -1.7747e+00, ..., -1.3668e+00,
-3.0758e+00, -1.6745e+00],
...,
[ 1.1480e+00, -4.4205e+00, -4.0478e+00, ..., -2.7955e+00,
-3.7591e+00, -1.2622e+00],
[-9.0584e-01, -4.4808e+00, -1.4549e-01, ..., -1.2650e+00,
-3.1799e+00, -2.0542e+00],
[-4.1724e+00, -5.3417e+00, -2.0044e+00, ..., -1.7756e+00,
-2.6365e+00, -7.9398e-01]],
[[-2.4238e+00, -3.1496e+00, -1.5263e+00, ..., 5.7259e-01,
-1.2127e+00, -6.4595e-01],
[-4.1987e+00, -3.4874e+00, -2.1985e+00, ..., -2.2624e-01,
-3.5341e+00, -1.7369e+00],
[-3.8917e+00, -2.4577e+00, -1.8167e+00, ..., -9.4547e-01,
-4.4539e+00, -2.3370e+00],
...,
[-2.9427e+00, -8.6749e+00, -5.0250e+00, ..., -1.8123e+00,
-3.4517e+00, -1.0711e+00],
[ 2.5263e-01, -2.7801e+00, 3.4449e-02, ..., -7.8884e-01,
-2.3447e+00, -1.4107e+00],
[-9.3806e-01, -9.2828e-01, -1.1187e+00, ..., -5.4439e-01,
-1.8198e+00, -9.5866e-01]],
[[ 1.4716e+00, 1.1299e+00, 1.4914e+00, ..., 7.3725e-01,
9.9687e-01, 5.3737e-01],
[ 3.0857e+00, 3.1040e+00, 3.2572e+00, ..., 1.6666e+00,
9.8944e-01, 2.3452e+00],
[ 1.6862e+00, 1.1825e+00, 2.0981e+00, ..., 1.0296e+00,
2.8838e-02, 7.2665e-01],
...,
[-4.7759e-01, -4.5455e+00, -3.6606e+00, ..., 2.8344e+00,
5.0936e-01, 7.5058e-01],
[ 3.9587e+00, -1.7458e+00, -4.1006e+00, ..., 1.1647e+00,
1.4284e+00, 1.6188e+00],
[ 4.0685e+00, 2.5186e+00, 7.2187e-01, ..., 9.5014e-01,
2.6485e-01, -2.1789e-01]]]], device='cuda:0'),
tensor([[[[ 2.6980e-01, 1.2243e+00, 1.9821e-01, ..., -2.2853e+00,
-3.0200e+00, 3.0367e-01],
[ 2.8337e-01, 9.0618e-01, 1.5830e+00, ..., 4.6350e-01,
-1.6966e+00, 2.8733e+00],
[-5.5826e-01, -2.5652e-01, -1.3466e-01, ..., -2.2258e+00,
-2.0385e+00, 1.8614e+00],
...,
[-1.0873e+00, -1.2726e+00, 1.6794e-01, ..., -3.9021e+00,
-2.9040e+00, 1.5804e-01],
[-2.3493e+00, -1.5254e+00, -6.2373e-01, ..., -3.5963e+00,
-3.3569e+00, -5.0930e-01],
[-1.1934e+00, -3.4126e+00, -2.4069e+00, ..., -4.3829e+00,
-3.4478e+00, -1.2520e+00]],
[[-1.2085e+00, -1.6873e+00, -2.8547e+00, ..., -5.0580e+00,
-2.7279e+00, 5.1372e-01],
[-4.1744e+00, -4.2315e+00, -3.0203e+00, ..., -2.1665e+00,
-1.6359e-01, 9.3735e-01],
[-4.0119e+00, -4.4288e+00, -4.0866e+00, ..., -4.5456e+00,
-6.1474e-01, 1.6238e+00],
...,
[-6.3956e+00, -5.9643e+00, -2.5750e+00, ..., -8.1169e+00,
-6.2022e+00, -2.1428e+00],
[-8.0511e+00, -7.7315e+00, -5.8332e+00, ..., -9.3851e+00,
-8.6643e+00, -3.8026e+00],
[-1.8430e+00, -3.2286e+00, -3.0985e+00, ..., -4.3921e+00,
-3.5246e+00, -4.1398e-01]],
[[-4.7115e-02, 1.3001e+00, 1.6535e+00, ..., 1.1768e+00,
1.6184e+00, -3.2209e-01],
[-4.5033e-03, -1.2156e-01, 4.6657e-01, ..., -2.8110e+00,
-5.4214e+00, -2.3728e+00],
[ 6.7725e-01, 1.1398e+00, 8.1189e-01, ..., -2.2484e+00,
-4.3839e+00, -1.3523e+00],
...,
[ 2.8046e-01, 6.9611e-01, 9.8847e-02, ..., -5.8634e-01,
-8.9609e-02, 1.1291e-01],
[ 5.8973e-01, 2.4921e-01, -1.7302e-01, ..., -1.1276e+00,
-1.6728e-02, -1.0038e-01],
[ 5.6517e-01, 9.9000e-01, 9.9550e-01, ..., -4.3933e-01,
-5.6280e-01, -1.9098e+00]],
...,
[[-7.1546e-01, -1.5553e+00, -2.0880e+00, ..., -1.7269e+00,
-2.2949e-01, 1.6995e+00],
[-2.7898e+00, -2.4519e+00, -1.3650e+00, ..., 1.4818e-01,
1.8182e+00, -1.0754e+00],
[-2.0383e+00, -1.0844e+00, -1.3653e+00, ..., -1.5306e+00,
3.7171e+00, -1.7459e-01],
...,
[-2.3446e+00, -9.8660e-01, -1.3143e+00, ..., -2.3027e+00,
-4.7662e+00, -3.0266e+00],
[-3.4886e+00, -2.5388e+00, -1.0301e+00, ..., -1.9868e+00,
-4.2111e+00, -6.8794e-01],
[-1.6053e+00, -1.6623e+00, -6.5586e-01, ..., -1.2907e+00,
-1.9696e+00, 1.1315e+00]],
[[-1.3558e+00, -2.3700e+00, -3.9482e+00, ..., -1.4473e+00,
-9.0019e-01, 1.1986e-01],
[-3.7331e+00, -3.1756e+00, -4.2014e+00, ..., -1.8971e+00,
-5.1492e-01, -1.3254e+00],
[-4.6344e+00, -3.5696e+00, -3.5208e+00, ..., -5.2862e+00,
-2.3351e+00, -1.6969e+00],
...,
[-4.6444e+00, -2.9465e+00, -6.0375e+00, ..., -3.8292e+00,
-2.5175e+00, -2.1942e-01],
[-4.1769e+00, -8.0088e-01, -2.4642e+00, ..., -1.4017e+00,
-1.8965e+00, 6.5351e-01],
[-3.3363e+00, -2.2974e+00, -5.9719e-01, ..., -1.1174e-01,
-8.1926e-01, 6.7738e-01]],
[[ 1.2528e+00, -3.2533e-02, 1.1195e+00, ..., 3.5530e-02,
-1.0722e+00, 1.9937e+00],
[ 2.3498e+00, 1.2155e+00, 1.2096e+00, ..., 1.1984e+00,
-1.3186e+00, 1.5626e+00],
[ 1.2920e+00, 8.2210e-01, 7.4879e-01, ..., -7.7683e+00,
-4.6579e+00, -1.2331e+00],
...,
[ 2.5253e+00, 2.0995e+00, 2.2375e+00, ..., -1.4496e+00,
1.0176e+00, 9.9139e-01],
[ 2.9388e+00, 4.1805e+00, 1.1864e+00, ..., 7.2851e-02,
8.5449e-01, 1.7327e+00],
[ 2.0249e+00, 1.9856e+00, 8.9480e-01, ..., -1.1646e-01,
-1.1589e-01, 9.0290e-01]]],
[[[ 1.0695e-01, 8.2377e-01, -6.2680e-01, ..., -3.3703e-01,
-6.3493e-01, -8.2193e-01],
[ 9.0062e-01, 9.9925e-01, 5.9265e-01, ..., -4.0476e-01,
-1.3987e-02, -1.1852e+00],
[ 1.2173e+00, -6.4543e-02, 3.4744e-01, ..., -9.6733e-01,
-5.8603e-01, -1.9214e+00],
...,
[-1.2592e+00, -1.6226e+00, -3.8117e+00, ..., -2.6458e-01,
-1.9570e+00, 6.9690e-01],
[-2.0872e+00, -8.6495e-02, -8.9240e-01, ..., 9.3751e-01,
-3.2595e-01, 1.0078e+00],
[-2.1746e+00, -3.9522e+00, -2.6817e+00, ..., 6.5645e-01,
-1.6001e+00, -1.3806e+00]],
[[-6.0294e-01, -1.3527e+00, -2.3108e+00, ..., -1.8473e+00,
-4.4519e+00, -2.5585e+00],
[-1.9832e+00, -3.5181e+00, -4.0742e+00, ..., -5.7451e+00,
-5.6421e+00, -4.8668e+00],
[-2.0271e+00, -3.6147e+00, -3.3143e+00, ..., -5.2760e+00,
-4.6484e+00, -4.0100e+00],
...,
[-7.3473e+00, -6.6141e+00, -8.5416e+00, ..., -1.3673e+00,
-6.3534e+00, -8.3722e-01],
[-9.1557e+00, -8.7410e+00, -7.4246e+00, ..., -2.7961e+00,
-6.1035e+00, -3.1196e+00],
[-2.3237e+00, -4.5100e+00, -4.1010e+00, ..., -1.9707e-01,
-3.0005e+00, -8.8994e-01]],
[[ 6.5116e-01, 1.2707e+00, 1.1258e+00, ..., 1.6561e+00,
2.1467e+00, 8.3624e-01],
[ 6.8494e-01, 1.4088e+00, 1.7849e+00, ..., 1.2447e+00,
2.1175e+00, 7.8951e-01],
[ 2.9278e-01, 2.7659e+00, 1.0661e+00, ..., 8.5739e-01,
1.9688e+00, 1.9784e+00],
...,
[ 8.2257e-01, -1.6705e-01, -3.7500e-01, ..., 3.7807e-01,
4.6118e-01, -1.9039e+00],
[ 1.9168e-01, -1.2299e+00, -1.4107e+00, ..., -3.0475e-01,
-3.1705e-01, -1.3061e+00],
[ 7.1154e-01, 5.8770e-01, -9.0198e-01, ..., -9.1623e-02,
-4.6109e-01, -1.7067e+00]],
...,
[[-8.0037e-01, -1.1397e+00, -2.2294e+00, ..., -2.3306e+00,
-3.2368e+00, -1.1146e+00],
[-2.2039e+00, -2.5964e+00, -2.4426e+00, ..., -4.1565e+00,
-4.9421e+00, -3.7670e+00],
[-1.0033e+00, -1.0355e+00, -6.8345e-01, ..., -2.2608e+00,
-4.4432e+00, -3.0331e+00],
...,
[-1.9300e+00, -2.0919e+00, -1.8087e+00, ..., -1.3750e+00,
-2.0049e+00, -9.8187e-01],
[-2.8040e+00, -2.5779e+00, -1.6790e+00, ..., -3.1597e+00,
-2.7609e+00, -1.7731e-01],
[-2.5505e+00, -2.6195e+00, -8.0242e-01, ..., -1.7189e+00,
-2.1886e+00, 4.2387e-01]],
[[-1.4988e+00, -2.5435e+00, -4.0790e+00, ..., -3.0648e+00,
-2.8732e+00, -5.6042e-01],
[-4.2219e+00, -4.6565e+00, -6.3255e+00, ..., -4.3665e+00,
-5.2808e+00, -1.8461e+00],
[-3.9997e+00, -4.5663e+00, -4.9900e+00, ..., -2.4564e+00,
-3.7267e+00, -1.7825e+00],
...,
[-3.5485e+00, -3.5713e+00, -2.6282e+00, ..., 3.1476e-03,
4.1759e-01, -2.4432e-02],
[-3.1467e+00, 9.2390e-01, -1.6147e+00, ..., -2.9280e-01,
7.0640e-01, 1.6209e+00],
[-3.8844e+00, -1.5983e+00, -9.1565e-01, ..., -8.3850e-02,
-5.2673e-01, 8.1529e-01]],
[[ 8.3665e-01, -1.5951e-01, 8.4676e-01, ..., 8.9679e-01,
1.3090e+00, 6.3529e-01],
[ 1.6116e+00, 7.3677e-01, 6.1133e-01, ..., 9.2252e-01,
1.7076e+00, 3.3853e+00],
[ 1.6354e+00, 1.0584e-01, 5.0160e-01, ..., 7.5700e-01,
1.2557e+00, 1.2271e+00],
...,
[ 3.0587e+00, 2.3541e+00, 8.6689e-01, ..., 9.9512e-01,
1.7705e+00, 9.4868e-01],
[ 4.4792e+00, 4.6107e+00, 1.8604e+00, ..., 1.5874e+00,
1.6508e+00, 1.7470e+00],
[ 2.4631e+00, 2.3716e+00, 7.8336e-01, ..., 2.1942e+00,
7.0184e-01, 1.0047e+00]]],
[[[ 1.6807e+00, 1.6001e+00, 4.0308e-01, ..., -5.9885e-01,
1.3332e-01, 7.2178e-01],
[ 1.1985e+00, -5.1391e-01, -3.7111e+00, ..., -1.7509e+00,
-1.0745e+00, 1.0409e+00],
[-2.2381e-01, -4.7467e+00, -5.0772e+00, ..., -1.9840e+00,
-1.2609e+00, -7.8492e-02],
...,
[-2.5068e+00, -1.5077e+00, 4.1591e-01, ..., 8.5010e-01,
1.7429e+00, 1.7964e-01],
[-1.2668e+00, 4.6324e-02, -1.2664e+00, ..., 2.8121e-01,
-9.0688e-01, -1.3952e+00],
[-4.8811e-01, -1.5218e+00, -1.9750e+00, ..., -4.3734e+00,
-4.0997e+00, -4.1585e+00]],
[[-4.8287e-01, -4.1475e+00, -6.8600e+00, ..., -2.9912e+00,
-4.7229e+00, -3.4677e-01],
[-4.4280e+00, -9.5665e+00, -1.0237e+01, ..., -7.0211e+00,
-7.4843e+00, -1.7312e+00],
[-5.5705e+00, -1.0204e+01, -4.4806e+00, ..., -4.4506e+00,
-6.9361e+00, -2.7540e+00],
...,
[-3.3876e+00, -1.2264e+00, 2.3103e-01, ..., -2.6381e+00,
-4.9166e+00, -5.5441e+00],
[-1.7308e+00, 6.2592e-02, -1.1008e+00, ..., -9.2160e+00,
-8.7800e+00, -6.1710e+00],
[-1.0764e-02, -9.8226e-01, -2.6333e+00, ..., -3.8203e+00,
-4.3496e+00, -3.0545e+00]],
[[ 7.2668e-03, 1.4256e+00, 5.6277e-01, ..., 8.1128e-01,
2.1198e+00, 9.4942e-01],
[-1.1731e+00, -1.8796e-01, 2.4303e+00, ..., 8.4542e-01,
9.0829e-01, -6.0366e-01],
[-7.9033e-01, 2.8268e+00, 2.1918e+00, ..., -9.0117e-02,
1.3107e+00, 4.6780e-01],
...,
[ 2.7194e+00, 1.3979e+00, 2.4624e-01, ..., -1.2529e+00,
-1.0583e+00, -4.3450e-01],
[ 4.8471e-01, -1.4094e+00, -2.0741e+00, ..., -1.1427e+00,
-1.3687e+00, 2.1381e-01],
[ 2.4535e-01, 5.6859e-01, 3.7191e-02, ..., 8.5020e-01,
3.2911e-01, -3.4070e-01]],
...,
[[ 2.1289e-01, -2.5860e+00, -3.2733e+00, ..., -2.2642e+00,
-2.5749e+00, -4.6122e-01],
[-3.1788e+00, -5.8913e+00, -6.2398e+00, ..., -5.6393e+00,
-4.0379e+00, -2.7836e+00],
[-3.1176e+00, -6.2720e+00, -2.2470e+00, ..., -2.3442e+00,
-3.6276e+00, -2.3845e+00],
...,
[-2.3212e+00, 6.2061e-01, 1.3048e+00, ..., -1.5684e+00,
-2.5667e+00, -2.9194e+00],
[-1.7002e+00, -8.1869e-01, -6.1840e-01, ..., -4.7423e+00,
-4.2042e+00, -2.6057e+00],
[-8.6770e-01, -2.1133e+00, -1.3932e+00, ..., -2.3310e+00,
-3.1527e+00, -2.0960e+00]],
[[-1.6998e+00, -1.1953e+00, -1.9813e+00, ..., -1.4372e+00,
-1.5014e+00, -1.3955e+00],
[-3.8435e+00, -4.2125e+00, -3.5349e+00, ..., -4.4831e+00,
-2.7034e+00, -1.0655e+00],
[-5.0251e+00, -3.4960e+00, -1.9324e+00, ..., -1.2699e-01,
-1.0708e+00, -4.5437e-01],
...,
[-2.9716e+00, -1.2908e+00, -4.5530e-01, ..., -1.1922e+00,
-3.1817e+00, -1.3318e+00],
[-1.4276e+00, -6.0169e-01, -4.2042e-01, ..., -5.7277e-01,
-3.1096e+00, -3.1959e-01],
[-1.7279e+00, -1.9250e+00, -1.3270e+00, ..., -1.1747e+00,
-2.6162e+00, -6.8698e-01]],
[[ 1.1509e+00, 1.0910e+00, 8.7036e-01, ..., -6.8785e-01,
9.7644e-02, -4.4157e-01],
[ 2.8672e+00, 4.1878e+00, 8.0607e-01, ..., 1.4328e+00,
2.2567e+00, 2.4902e+00],
[ 2.3484e+00, 2.3286e+00, -2.6020e+00, ..., -5.9860e-02,
1.1425e+00, 1.2784e+00],
...,
[ 1.2715e+00, 7.1736e-01, 2.1364e-02, ..., 9.1793e-01,
1.4418e+00, 3.5434e+00],
[ 1.5114e+00, 6.9445e-01, -3.3857e+00, ..., 4.3523e+00,
2.8688e+00, 2.6490e+00],
[ 1.3897e+00, -4.1847e-01, -1.5578e+00, ..., -8.8076e-01,
-8.2685e-01, 5.1516e-01]]],
[[[-2.0973e-01, 1.3200e+00, 1.0463e+00, ..., 2.8484e-02,
4.7386e-01, -2.0275e-01],
[ 1.0944e+00, -6.0910e-01, 1.0655e+00, ..., 3.7304e-01,
3.9046e-01, -8.2287e-01],
[-3.1226e+00, -5.0931e+00, -4.8585e+00, ..., 7.8568e-01,
4.1896e-01, -1.4742e+00],
...,
[-3.8713e-01, -3.5287e-01, 2.4159e-01, ..., 5.8726e-02,
3.5668e-01, -4.2002e-01],
[-1.6393e+00, -3.0448e-01, 4.9608e-02, ..., 3.6648e-01,
3.2583e-01, -1.0852e+00],
[-1.1426e+00, -2.1962e+00, -1.5258e+00, ..., -3.0531e+00,
-2.8100e+00, -2.6787e+00]],
[[-1.3054e+00, -2.3400e+00, -4.6852e+00, ..., -2.1001e+00,
-3.6627e+00, -2.2721e+00],
[-5.2730e+00, -6.5731e+00, -5.1936e+00, ..., -4.4123e+00,
-5.0216e+00, -3.7748e+00],
[-6.5475e+00, -9.2946e+00, -1.0665e+01, ..., -3.9027e+00,
-4.9262e+00, -4.0442e+00],
...,
[-4.5789e+00, -5.0563e+00, -4.4238e+00, ..., -3.1397e+00,
-3.1616e+00, -2.7202e+00],
[-7.8557e+00, -7.9332e+00, -6.0709e+00, ..., -4.7631e+00,
-5.4196e+00, -4.8839e+00],
[-1.3222e+00, -2.9279e+00, -1.7020e+00, ..., -2.8176e+00,
-2.7955e+00, -1.5789e+00]],
[[-1.3310e-01, 2.3882e+00, 1.8505e+00, ..., 1.6624e+00,
1.7476e+00, 8.8219e-01],
[-9.0909e-01, 1.2651e-01, 3.8516e-01, ..., 8.2954e-01,
1.3568e+00, 7.0623e-01],
[ 3.6142e-01, 3.0746e+00, 4.6477e+00, ..., 1.5793e+00,
1.6266e+00, 1.3597e+00],
...,
[-6.8434e-01, 4.2094e-01, 1.6301e+00, ..., 1.2772e+00,
6.6151e-01, 6.7113e-01],
[-1.0402e+00, 8.8925e-03, 1.1610e+00, ..., 1.2609e+00,
6.2316e-01, 4.0564e-01],
[-3.6175e-01, -5.3015e-01, -5.6084e-02, ..., 1.7602e+00,
1.4903e+00, 3.8822e-01]],
...,
[[-1.0777e+00, -2.1640e+00, -2.5538e+00, ..., -1.5433e+00,
-2.0225e+00, -4.4465e-01],
[-3.2181e+00, -4.0336e+00, -2.6413e+00, ..., -2.7945e+00,
-3.9785e+00, -3.3013e+00],
[-4.0772e+00, -3.6684e+00, -4.2642e+00, ..., -1.1626e+00,
-2.2402e+00, -1.9190e+00],
...,
[-1.0439e+00, -2.3169e+00, -1.0619e+00, ..., -1.0461e+00,
-2.2576e+00, -1.4653e+00],
[-2.2329e+00, -3.9385e+00, -1.5235e+00, ..., -1.4413e+00,
-2.7626e+00, -1.7142e+00],
[-1.1665e+00, -1.1883e+00, -5.0995e-01, ..., -1.8294e+00,
-2.0436e+00, -5.7543e-01]],
[[-1.7868e+00, -1.7098e+00, -2.6293e+00, ..., -5.7639e-01,
-1.4401e+00, -6.5405e-01],
[-4.4485e+00, -3.4219e+00, -2.6721e+00, ..., -1.5526e+00,
-4.1554e+00, -2.1495e+00],
[-4.6861e+00, -5.0820e+00, -7.7831e+00, ..., 1.1308e-01,
-3.4163e+00, -2.9781e+00],
...,
[-2.8836e+00, -2.9133e+00, -3.8246e+00, ..., -7.0674e-01,
-3.1812e+00, -2.1614e+00],
[-2.1397e+00, -2.6001e-01, -1.8180e+00, ..., -1.3618e+00,
-3.3169e+00, -2.1182e+00],
[-2.0418e+00, -1.2938e+00, -1.6185e+00, ..., -1.8734e+00,
-2.6500e+00, -1.2356e+00]],
[[ 1.0113e+00, 5.1345e-01, 1.3193e+00, ..., 2.0222e-01,
1.3017e+00, 6.5202e-01],
[ 4.3562e+00, 3.1876e+00, 2.8832e+00, ..., 1.2049e+00,
1.1571e+00, 2.3938e+00],
[ 1.6581e+00, 1.9676e-01, 9.5680e-01, ..., 1.0378e+00,
1.0217e+00, 7.6322e-01],
...,
[ 3.0355e+00, 2.4576e+00, 2.2019e+00, ..., 2.5421e-01,
1.8383e-01, 5.5219e-01],
[ 3.7759e+00, 4.0183e+00, 2.5049e+00, ..., 1.1955e+00,
9.9005e-01, 1.2012e+00],
[ 2.3100e+00, 2.2982e+00, 1.5330e+00, ..., -2.8144e-01,
-3.2433e-01, -6.0959e-02]]],
[[[ 2.4278e-01, 1.4185e+00, 5.2014e-01, ..., -6.0858e-01,
5.2024e-02, 6.7260e-01],
[ 4.8808e-01, 3.3942e-01, 5.3371e-01, ..., -1.6583e+00,
-1.0935e+00, 1.0017e+00],
[ 1.8534e-01, 3.8646e-01, 1.3383e+00, ..., -1.2349e+00,
-9.7095e-01, 2.6102e-01],
...,
[ 4.4964e-01, -7.6143e-01, -2.6567e-01, ..., -5.0324e-01,
-7.6131e-01, -9.6091e-01],
[-1.2504e+00, -1.4001e+00, -2.4283e+00, ..., -2.0065e+00,
-2.5007e+00, -1.1377e+00],
[-1.5105e-01, -2.5441e+00, -2.8401e+00, ..., -2.7704e+00,
-2.4135e+00, -9.1778e-01]],
[[-7.2798e-01, -2.2468e+00, -4.1469e+00, ..., -3.2549e+00,
-4.8051e+00, -4.2145e-01],
[-4.5002e+00, -6.0227e+00, -5.1851e+00, ..., -6.1587e+00,
-7.4701e+00, -1.8044e+00],
[-3.3956e+00, -5.1240e+00, -6.2499e+00, ..., -7.0763e+00,
-8.3372e+00, -3.0972e+00],
...,
[-4.5102e+00, -4.8599e+00, -4.7862e+00, ..., -4.5968e+00,
-4.8444e+00, -2.1284e+00],
[-7.5485e+00, -8.1435e+00, -8.8896e+00, ..., -7.8128e+00,
-9.1407e+00, -4.0992e+00],
[-1.1404e+00, -2.8914e+00, -3.2245e+00, ..., -2.8979e+00,
-2.9038e+00, -2.8637e-01]],
[[-2.7109e-01, 2.1445e+00, 1.2406e+00, ..., 8.0826e-01,
2.1488e+00, 9.6387e-01],
[-8.9812e-02, 1.1040e+00, 8.6224e-01, ..., 5.0674e-01,
1.1138e+00, -4.3414e-01],
[-1.1257e-01, 1.9963e+00, 1.4913e+00, ..., 3.7640e-01,
1.4076e+00, -6.1689e-03],
...,
[ 1.0750e-01, 8.4096e-01, 1.6444e+00, ..., 9.9936e-01,
1.1345e+00, 8.7487e-01],
[ 2.4557e-01, 6.6941e-01, 1.1494e+00, ..., 1.1049e+00,
1.9128e+00, 5.7913e-01],
[-6.3028e-01, 5.0966e-01, 1.0237e+00, ..., 3.6055e-01,
7.7565e-01, -8.1597e-01]],
...,
[[-7.5561e-01, -1.8466e+00, -1.8529e+00, ..., -2.5386e+00,
-2.7436e+00, -4.3883e-01],
[-2.9549e+00, -4.4301e+00, -2.5005e+00, ..., -4.2483e+00,
-4.3945e+00, -2.5991e+00],
[-1.3177e+00, -2.6502e+00, -9.2035e-01, ..., -1.9520e+00,
-5.2380e+00, -1.6119e+00],
...,
[-8.6449e-01, -1.5237e+00, -1.0649e+00, ..., -1.0639e+00,
-3.6708e+00, -1.7934e+00],
[-2.7562e+00, -3.7969e+00, -2.8803e+00, ..., -3.0617e+00,
-4.8508e+00, -1.5926e+00],
[-9.5546e-01, -1.9706e+00, -1.1347e+00, ..., -8.9042e-01,
-1.1545e+00, 1.2116e+00]],
[[-1.7002e+00, -2.2362e+00, -2.9588e+00, ..., -1.5796e+00,
-1.5255e+00, -1.4225e+00],
[-4.2607e+00, -3.4916e+00, -2.7372e+00, ..., -3.0662e+00,
-1.7183e+00, -7.2301e-01],
[-2.5497e+00, -1.7983e+00, -1.8695e+00, ..., 9.1785e-02,
-1.6246e-01, -3.6628e-01],
...,
[-3.6096e+00, -3.8073e+00, -3.8546e+00, ..., -1.6639e+00,
-2.3634e+00, -1.0957e+00],
[-4.2490e+00, -2.3764e+00, -1.8008e+00, ..., -1.1971e+00,
-1.9718e+00, -5.1476e-01],
[-3.0379e+00, -2.4367e+00, -1.0844e+00, ..., -8.7125e-01,
-9.1888e-01, 5.0383e-02]],
[[ 1.0410e+00, 6.1600e-01, 6.8260e-01, ..., -6.5519e-01,
1.0254e-01, -4.2852e-01],
[ 3.3696e+00, 3.4264e+00, 2.1012e+00, ..., 1.5576e+00,
2.0751e+00, 2.4691e+00],
[ 1.6748e+00, 1.4207e+00, 1.2384e+00, ..., 2.5440e+00,
1.9928e+00, 1.5831e+00],
...,
[ 2.7237e+00, 1.2531e+00, 1.1177e+00, ..., 1.7997e+00,
1.7138e+00, 6.8157e-01],
[ 3.4381e+00, 4.2353e+00, 2.5727e+00, ..., 2.6256e+00,
1.3465e+00, 1.6664e+00],
[ 2.3236e+00, 1.5844e+00, 5.9050e-01, ..., 2.9523e-01,
-5.4725e-02, 9.3875e-01]]]], device='cuda:0'),
tensor([[[[-1.4739e-01, -1.8455e+00, -7.8336e-01, ..., 1.0630e-01,
-5.4466e-01, -1.0642e+00],
[ 8.5128e-01, -3.7800e+00, -9.9446e-01, ..., 6.8446e-01,
2.6868e+00, 5.6615e-01],
[ 7.0839e-01, -3.2031e+00, -1.4736e+00, ..., 2.7712e+00,
3.1050e+00, 1.7898e-01],
...,
[-1.0695e+00, -3.1870e+00, -5.1658e-01, ..., 9.8087e-01,
5.3130e+00, 2.0116e+00],
[-1.5604e+00, -1.0681e+00, 9.8037e-01, ..., 1.0459e-01,
-1.6486e+00, -2.4251e+00],
[-2.9882e+00, -5.0348e+00, -3.3177e+00, ..., -3.8093e+00,
-3.6770e+00, -4.0091e+00]],
[[-1.1446e+00, -3.0737e+00, -7.6907e-01, ..., -5.6799e-01,
-4.9801e+00, -2.7422e+00],
[-4.0554e+00, -6.2975e+00, -1.2718e+00, ..., -9.3614e-01,
-2.0002e+00, -3.7937e+00],
[-3.2190e+00, -6.7132e+00, -4.6663e+00, ..., 1.2551e+00,
-2.6625e+00, -4.9678e+00],
...,
[-5.9727e+00, -4.9447e+00, -1.7286e+00, ..., 1.1048e+00,
-3.2185e-01, -3.2290e+00],
[-7.8083e+00, -7.3109e+00, -4.1932e+00, ..., -1.1632e+01,
-1.1762e+01, -8.3320e+00],
[-2.5330e+00, -4.4319e+00, -3.5579e+00, ..., -2.9202e+00,
-3.6233e+00, -2.8659e+00]],
[[ 2.9058e-01, 2.5499e+00, 1.4264e+00, ..., -5.0000e-02,
4.2705e+00, 1.3248e+00],
[-3.4583e-01, 1.1741e+00, -9.3232e-01, ..., -8.5895e+00,
4.2224e+00, 2.0744e+00],
[ 1.9918e-01, 2.2563e+00, -9.8762e-01, ..., -5.2512e+00,
2.0829e+00, 1.8775e+00],
...,
[ 3.9518e-01, 1.1321e+00, 4.0429e-01, ..., -2.9804e-02,
-4.5109e-01, 1.4461e+00],
[ 7.6165e-01, 3.6006e-01, 6.2510e-01, ..., -1.6151e+00,
-1.3694e+00, 6.3014e-02],
[ 1.4329e+00, 1.5273e+00, 1.0454e+00, ..., 3.9893e-01,
1.4600e+00, -3.9310e-02]],
...,
[[-1.1901e+00, -2.8467e+00, -4.6720e-01, ..., -9.8578e-01,
-4.9468e+00, -1.3283e+00],
[-2.6632e+00, -4.7156e+00, -4.5078e-01, ..., -1.8849e-01,
-5.5255e+00, -1.4693e+00],
[-1.6611e+00, -2.9955e+00, -8.6590e-01, ..., -2.0460e+00,
-7.5848e+00, -1.3251e+00],
...,
[-2.5259e+00, -1.7041e+00, 3.7988e-01, ..., 6.4376e-01,
-1.2353e+00, -2.2286e+00],
[-3.6802e+00, -2.8885e+00, -1.3819e+00, ..., -3.6782e+00,
-5.2510e+00, -3.0741e+00],
[-3.8251e+00, -3.6574e+00, -2.9167e+00, ..., -9.8015e-01,
-2.6333e+00, -1.5677e+00]],
[[-2.4050e+00, -9.4521e-01, -2.0419e+00, ..., -2.1243e+00,
-1.7028e+00, 1.5185e+00],
[-4.8610e+00, -1.3315e+00, 4.6913e-01, ..., 1.6302e+00,
-2.2364e+00, 1.2637e+00],
[-3.5757e+00, -1.0903e+00, 2.0938e+00, ..., 3.6170e+00,
-3.3315e+00, 8.3336e-01],
...,
[-3.1823e+00, -1.6557e+00, -1.9968e+00, ..., -1.5456e-01,
-1.4535e+00, -1.4669e+00],
[-3.4870e+00, 3.6951e-01, -2.3157e+00, ..., -3.0848e+00,
-4.1544e+00, -2.2770e-01],
[-4.4813e+00, -2.4168e+00, -2.5456e+00, ..., -5.9539e-01,
-2.7711e+00, -1.0292e+00]],
[[ 1.4703e+00, 3.6847e-01, -5.3142e-01, ..., -9.1531e-01,
3.4760e+00, 1.1423e+00],
[ 2.9808e+00, 2.4157e+00, 1.1427e-01, ..., -2.2182e+00,
2.6023e+00, 5.1273e+00],
[ 2.3455e+00, 5.8162e-01, 2.1906e-01, ..., 8.3665e-01,
3.1272e+00, 2.7311e+00],
...,
[ 2.5430e+00, 1.6947e-02, -1.1539e-01, ..., -3.5171e+00,
1.1420e+00, 2.2910e+00],
[ 4.1209e+00, 3.6642e+00, 2.2893e+00, ..., 4.2502e+00,
3.5172e+00, 2.5275e+00],
[ 2.1312e+00, 9.9427e-01, 1.5007e-01, ..., -4.5607e-01,
-8.6369e-01, 7.8229e-01]]],
[[[-2.7118e-01, 1.0803e+00, 5.1844e-01, ..., -5.6713e-01,
-1.0724e-01, -1.7949e-01],
[-4.0125e-02, 1.4431e-01, 2.3731e+00, ..., 1.5900e+00,
3.4612e-01, -8.5684e-01],
[-9.0227e-01, 4.0717e-01, 2.3443e+00, ..., 1.9340e+00,
-7.2067e-01, -1.5017e+00],
...,
[-4.8112e-01, -1.2303e+00, 4.6767e-01, ..., 7.5465e-01,
6.5881e-01, -4.3233e-01],
[-1.3174e+00, -8.2340e-01, -1.0704e+00, ..., 3.9135e-01,
2.7491e-01, -1.2212e+00],
[-1.1986e+00, -2.4323e+00, -2.0696e+00, ..., -3.1639e+00,
-2.8766e+00, -2.7081e+00]],
[[-1.6182e+00, -2.3618e+00, -3.7130e+00, ..., -2.0700e+00,
-4.2755e+00, -2.1035e+00],
[-4.5867e+00, -4.5288e+00, -3.9147e+00, ..., -1.0473e+00,
-5.5220e+00, -3.6211e+00],
[-3.5671e+00, -4.8542e+00, -5.1117e+00, ..., -1.1750e+00,
-4.5039e+00, -3.1472e+00],
...,
[-4.3783e+00, -6.0947e+00, -7.4697e+00, ..., -2.8015e+00,
-3.2339e+00, -2.7023e+00],
[-8.0097e+00, -7.4688e+00, -5.7072e+00, ..., -4.8391e+00,
-5.2555e+00, -5.0224e+00],
[-1.3542e+00, -3.1715e+00, -2.1649e+00, ..., -3.0276e+00,
-2.8254e+00, -1.5644e+00]],
[[-2.3135e-01, 2.2705e+00, 1.4198e+00, ..., 2.3102e+00,
1.8127e+00, 8.8964e-01],
[-5.8660e-01, 3.8093e-01, 1.1122e+00, ..., 2.2778e+00,
1.8344e+00, 5.6436e-01],
[ 2.7801e-01, 2.4170e+00, 1.4894e+00, ..., 3.1075e+00,
2.6305e+00, 1.1862e+00],
...,
[-1.3328e+00, -8.6954e-02, -2.6554e+00, ..., 1.7264e+00,
1.1153e+00, 6.7773e-01],
[-1.2203e+00, -8.0619e-02, 1.3085e+00, ..., 1.3936e+00,
9.7411e-01, 4.6774e-01],
[-2.2735e-01, -4.7450e-01, -1.6075e-01, ..., 1.6347e+00,
1.5518e+00, 5.0735e-01]],
...,
[[-1.3885e+00, -2.2066e+00, -1.7279e+00, ..., -1.4004e+00,
-2.5382e+00, -5.4368e-01],
[-3.5827e+00, -3.6915e+00, -1.4669e+00, ..., -2.2597e+00,
-3.5631e+00, -3.5480e+00],
[-2.5557e+00, -2.5429e+00, -1.6022e+00, ..., -2.8706e+00,
-2.5783e+00, -2.5560e+00],
...,
[-1.3449e+00, -3.9315e+00, -4.1100e+00, ..., -1.7635e+00,
-2.1739e+00, -9.6214e-01],
[-2.0450e+00, -3.9148e+00, -1.8068e+00, ..., -1.4765e+00,
-2.8132e+00, -1.5961e+00],
[-1.3637e+00, -1.7035e+00, -1.2224e+00, ..., -1.9821e+00,
-2.1591e+00, -5.5827e-01]],
[[-1.8632e+00, -1.9277e+00, -2.4352e+00, ..., 3.9345e-01,
-1.3271e+00, -7.4772e-01],
[-4.0616e+00, -2.1866e+00, -2.2395e+00, ..., -1.4677e+00,
-3.3558e+00, -1.9424e+00],
[-3.6479e+00, -2.9186e+00, -4.0543e+00, ..., -2.9489e+00,
-3.0899e+00, -2.5693e+00],
...,
[-3.1683e+00, -2.9524e+00, -1.0056e+00, ..., -7.6174e-01,
-3.2272e+00, -2.1436e+00],
[-2.1759e+00, -4.4453e-01, -1.4079e-01, ..., -9.6292e-01,
-3.3538e+00, -1.9544e+00],
[-2.0357e+00, -1.4588e+00, -1.3631e+00, ..., -1.5699e+00,
-2.6578e+00, -1.2432e+00]],
[[ 1.1963e+00, 3.6151e-01, 8.4631e-01, ..., 1.0716e+00,
1.3815e+00, 5.2861e-01],
[ 3.9037e+00, 2.8938e+00, 2.2346e+00, ..., 3.0117e+00,
1.7629e+00, 2.2170e+00],
[ 1.5512e+00, 6.5052e-01, 1.1727e+00, ..., 2.8976e+00,
2.2238e-01, 5.6844e-01],
...,
[ 2.7415e+00, 1.9744e+00, 5.1886e+00, ..., 4.0714e-01,
2.4383e-01, 5.7916e-01],
[ 3.6965e+00, 3.1374e+00, 1.7716e+00, ..., 1.2801e+00,
8.9124e-01, 1.2432e+00],
[ 2.3080e+00, 2.0800e+00, 1.2901e+00, ..., -1.3526e-01,
-2.9939e-01, -1.2279e-01]]],
[[[ 1.6417e-02, 1.0316e+00, 3.4631e-01, ..., -4.6572e+00,
-4.9767e+00, -2.2291e+00],
[ 1.1709e+00, 1.0802e+00, 1.7799e+00, ..., -2.3093e+00,
2.3762e+00, 9.1498e-01],
[-1.0370e+00, -1.1579e+00, 8.1900e-01, ..., 3.3977e-01,
9.7011e-01, -1.0558e-01],
...,
[ 9.9553e-01, 4.3968e-02, -9.1455e-01, ..., 2.9431e+00,
-1.7179e+00, -7.0355e-01],
[-1.3201e+00, -7.1415e-01, -1.5201e+00, ..., 7.1745e-01,
-2.5261e+00, -6.4816e-01],
[-9.4782e-01, -3.4910e+00, -3.8116e+00, ..., -4.2169e+00,
-3.6299e+00, -1.4911e+00]],
[[-9.5693e-01, -1.7768e+00, -2.8417e+00, ..., -7.2091e+00,
-6.9968e+00, -8.5602e-01],
[-2.4275e+00, -4.2503e+00, -4.8007e+00, ..., -4.0213e+00,
-6.5163e-01, -1.1207e-01],
[-2.6453e+00, -3.3399e+00, -2.6626e+00, ..., 7.1849e-01,
-1.2143e+00, -8.7965e-01],
...,
[-5.0604e+00, -7.4725e+00, -6.2705e+00, ..., 5.0825e-01,
-4.5824e+00, -2.0112e+00],
[-6.2890e+00, -6.8603e+00, -6.7328e+00, ..., -8.8064e+00,
-8.9190e+00, -3.6104e+00],
[-1.7150e+00, -3.4362e+00, -3.0649e+00, ..., -5.8293e+00,
-4.1435e+00, -5.6197e-01]],
[[ 2.6475e-01, 1.4031e+00, 1.8601e+00, ..., 8.7930e-01,
3.2252e+00, 2.6074e+00],
[ 1.5298e-01, 2.2795e-01, 8.1924e-01, ..., 3.8702e-01,
9.0104e-01, 8.1521e-01],
[ 7.1048e-01, 2.4704e+00, 8.1939e-01, ..., -5.0898e+00,
-4.1838e+00, -3.0980e+00],
...,
[-7.8890e-01, -5.6390e-01, 8.7960e-01, ..., 1.4497e+00,
3.2506e+00, 1.0842e+00],
[-4.1058e-01, 9.6016e-01, 6.8886e-02, ..., -4.5817e-01,
1.2482e+00, -1.1013e+00],
[ 5.1583e-01, 9.9022e-01, 4.3159e-01, ..., -3.0750e-01,
-1.0640e+00, -1.9446e+00]],
...,
[[-9.0524e-01, -1.4875e+00, -1.8858e+00, ..., -4.3371e+00,
-4.0797e+00, -1.6784e+00],
[-2.5553e+00, -1.9504e+00, -4.3480e-01, ..., -5.3366e-01,
7.9307e-01, 4.5517e-01],
[-2.7584e+00, -3.6141e-01, 7.0968e-01, ..., 1.3020e+00,
-1.3263e+00, -7.8715e-02],
...,
[-1.3728e+00, -2.0344e+00, -1.1534e+00, ..., -3.0996e+00,
-4.3702e+00, -1.2371e+00],
[-1.8922e+00, -1.7371e+00, -9.7544e-01, ..., -4.3446e+00,
-3.6483e+00, -3.0473e-01],
[-1.3681e+00, -2.2080e+00, -1.4388e+00, ..., -2.8476e+00,
-1.8877e+00, 7.9927e-01]],
[[-1.5977e+00, -1.9208e+00, -3.7251e+00, ..., -4.2873e+00,
-2.4508e+00, 6.5933e-01],
[-3.3581e+00, -2.9579e+00, -4.7984e+00, ..., -3.9543e+00,
-2.5166e+00, -5.7726e-01],
[-2.9650e+00, -1.7612e+00, -3.6940e+00, ..., -4.0747e+00,
-2.6598e+00, -3.2635e-01],
...,
[-2.8357e+00, -4.7522e+00, -4.5802e+00, ..., -1.4852e+00,
-4.3487e+00, -1.5954e+00],
[-3.0090e+00, -1.7821e+00, -2.3975e+00, ..., -9.2249e-01,
-1.8185e+00, 5.2889e-01],
[-2.8825e+00, -3.0020e+00, -8.1834e-01, ..., 4.7476e-01,
-2.3026e-01, 7.8422e-01]],
[[ 1.0793e+00, -1.6101e-01, 1.0205e+00, ..., -1.9610e+00,
2.6346e-01, -2.1724e-01],
[ 2.3301e+00, 1.3808e+00, 1.4807e+00, ..., 1.0610e+00,
3.0603e+00, 3.6093e+00],
[ 3.8041e-01, -4.2338e-01, 4.0598e-01, ..., -2.1116e+00,
-3.3656e+00, -1.0223e+00],
...,
[ 2.8435e+00, 4.0319e+00, 7.8385e-01, ..., 3.0875e+00,
1.4145e+00, 4.0878e-01],
[ 2.6153e+00, 3.3632e+00, 1.6824e+00, ..., 3.0478e+00,
1.2483e+00, 1.9806e+00],
[ 2.1098e+00, 1.7601e+00, 3.7326e-01, ..., 4.1247e-01,
5.2150e-01, 1.1501e+00]]],
[[[ 8.9081e-01, 1.6860e+00, 6.1142e-01, ..., -3.9200e+00,
-2.6712e+00, -9.6562e-01],
[ 1.5753e+00, 9.3187e-01, 7.2775e-01, ..., -3.1628e+00,
-2.6053e+00, -7.2705e-02],
[ 3.8424e-01, -3.6272e-01, -3.8001e-01, ..., -2.6098e+00,
-3.0207e+00, 6.7277e-02],
...,
[ 5.5615e-01, -9.4494e-01, -5.1607e-01, ..., -2.8846e+00,
-3.5374e+00, 4.0117e-01],
[ 6.9700e-02, -3.3079e-01, -5.2113e-01, ..., -4.2121e+00,
-3.6499e+00, 4.7297e-01],
[-3.2488e-01, -3.1276e+00, -3.1084e+00, ..., -3.6406e+00,
-3.2741e+00, -1.4862e+00]],
[[ 3.9935e-01, -1.1595e+00, -2.7623e+00, ..., -7.5493e+00,
-7.3470e+00, -1.1440e+00],
[-1.2287e+00, -3.0004e+00, -3.6563e+00, ..., -7.9181e+00,
-8.2680e+00, -2.2609e+00],
[-8.4358e-01, -2.2181e+00, -2.1565e+00, ..., -7.6951e+00,
-7.7634e+00, -2.3164e+00],
...,
[-2.3012e+00, -3.3106e+00, -3.0711e+00, ..., -8.2571e+00,
-8.3570e+00, -1.8993e+00],
[-4.2063e+00, -5.5964e+00, -4.9489e+00, ..., -9.5346e+00,
-1.0869e+01, -4.5189e+00],
[-3.0458e-01, -2.9929e+00, -3.4112e+00, ..., -3.7691e+00,
-3.7317e+00, -7.9234e-01]],
[[ 4.4507e-01, 2.1978e+00, 1.8882e+00, ..., 7.9064e-01,
2.1931e+00, 1.0050e+00],
[-2.7453e-01, 1.0271e+00, 7.3059e-01, ..., 1.0656e+00,
2.6686e+00, -3.8960e-02],
[ 9.5379e-01, 3.7722e+00, 2.2848e+00, ..., -7.3928e-01,
1.7454e+00, -3.6716e-01],
...,
[ 9.2306e-01, 2.3315e+00, 1.6740e+00, ..., -7.8052e-01,
8.7678e-01, -1.2070e+00],
[ 5.6446e-01, 2.1591e+00, 9.2872e-01, ..., -7.3431e-01,
1.0681e+00, -1.3666e+00],
[-2.9362e-01, 1.6471e+00, 1.0274e+00, ..., 1.3043e-01,
-2.2523e-01, -1.8924e+00]],
...,
[[-2.8516e-01, -9.0087e-01, -1.6804e+00, ..., -5.5035e+00,
-5.3588e+00, -7.2705e-01],
[-2.5918e+00, -4.2198e+00, -2.3992e+00, ..., -4.4614e+00,
-5.1131e+00, -1.7835e+00],
[-1.4524e+00, -3.0507e+00, -1.6557e+00, ..., -5.4564e+00,
-5.9079e+00, -1.1378e+00],
...,
[-1.7571e+00, -2.9047e+00, -1.4270e+00, ..., -2.9084e+00,
-4.9908e+00, -1.7980e+00],
[-1.9654e+00, -3.7636e+00, -1.9206e+00, ..., -3.0642e+00,
-5.3872e+00, -1.0361e+00],
[-1.0029e+00, -3.3447e+00, -2.4749e+00, ..., -1.6753e+00,
-2.2094e+00, 3.9248e-01]],
[[-1.0079e+00, -8.5433e-01, -1.0870e+00, ..., -2.0950e+00,
-3.7423e-01, -5.3662e-01],
[-3.0380e+00, -2.7797e+00, -1.5966e+00, ..., -3.7312e+00,
-1.5546e+00, -1.5898e-01],
[-3.0218e+00, -4.5634e+00, -2.0197e+00, ..., -1.9997e+00,
-1.4508e+00, -3.5259e-01],
...,
[-3.3379e+00, -4.2050e+00, -2.2571e+00, ..., -3.7450e+00,
-2.1879e+00, -8.3249e-01],
[-3.2064e+00, -2.2832e+00, -9.0934e-01, ..., -1.6619e+00,
-5.5556e-01, 1.1736e+00],
[-3.1600e+00, -3.1980e+00, -1.3461e+00, ..., -1.1132e-01,
-8.5160e-01, 6.1899e-01]],
[[ 8.7745e-01, 8.6343e-01, 1.0586e+00, ..., -3.2146e-01,
5.9455e-01, -1.3570e+00],
[ 1.3875e+00, 1.6814e+00, 1.5514e+00, ..., 3.1102e+00,
2.9509e+00, 2.2829e+00],
[ 5.7136e-01, -3.6388e-01, 5.3769e-01, ..., 2.5579e+00,
1.6270e+00, 1.2380e+00],
...,
[ 1.8095e+00, 1.1778e+00, 1.2365e+00, ..., 1.8496e+00,
2.3490e+00, 6.8088e-01],
[ 2.9874e+00, 3.2848e+00, 2.2146e+00, ..., 1.9147e+00,
8.9185e-01, 1.7587e+00],
[ 1.8558e+00, 6.3117e-01, 8.0944e-01, ..., 2.5214e-01,
-8.1391e-02, 8.7279e-01]]],
[[[-2.3367e+00, -7.5918e+00, -3.3043e+00, ..., -1.6868e+00,
-3.5575e-01, 1.8228e-01],
[-2.5996e+00, -7.7296e+00, -4.3132e+00, ..., 2.5304e-02,
-4.7023e-01, -8.6651e-02],
[-5.1456e+00, -6.5162e+00, -4.1565e+00, ..., -6.1275e-01,
-1.2153e+00, 6.3918e-01],
...,
[-3.9972e-01, -6.9903e-01, 2.2177e-01, ..., -4.3894e+00,
-5.7101e-01, 1.9988e+00],
[-1.5935e+00, -1.1596e-01, 1.2090e-01, ..., -3.4029e-01,
3.6351e+00, 1.7795e+00],
[-1.2048e+00, -2.4312e+00, -1.0844e+00, ..., 1.6573e+00,
-7.4485e-01, -2.0609e+00]],
[[-4.9106e+00, -7.5631e+00, -6.5964e-01, ..., -9.1973e-01,
-4.3051e-01, 4.4525e-01],
[-4.2848e+00, -2.0633e+00, 1.0687e+00, ..., 1.7865e+00,
-6.6851e-01, -4.4809e-01],
[-5.8705e+00, -5.2617e+00, -2.3680e+00, ..., -7.9903e-01,
-2.7527e-01, 1.0843e+00],
...,
[-4.7165e+00, -5.9854e+00, -4.3109e+00, ..., -2.2460e+00,
-1.3759e+00, 2.0323e-01],
[-7.7454e+00, -8.0593e+00, -5.5275e+00, ..., 1.4987e+00,
-8.7978e-02, -3.2048e+00],
[-1.4454e+00, -3.2363e+00, -2.4765e+00, ..., -7.1721e-01,
-1.9187e+00, -2.0856e+00]],
[[ 4.9455e-01, 5.3872e+00, 1.3971e+00, ..., 1.9547e+00,
7.3812e-01, 8.2678e-02],
[ 3.4131e-01, 1.4193e+00, -3.3236e+00, ..., -1.7099e+00,
-3.2912e-01, -3.9336e-01],
[ 2.1459e+00, 1.7954e+00, 4.1822e-01, ..., -2.1475e+00,
-5.5338e-01, -6.7079e-01],
...,
[-7.3699e-01, 6.6920e-01, 3.4331e-01, ..., 1.4511e+00,
8.0869e-01, 1.0129e+00],
[-1.3434e+00, 1.1460e-01, 5.2871e-01, ..., 1.8493e+00,
1.0804e+00, 1.0610e+00],
[-4.6009e-01, -3.8078e-01, -9.2846e-03, ..., -2.8923e-01,
-2.6317e-01, 7.8401e-01]],
...,
[[-3.9876e+00, -5.0211e+00, 9.4434e-01, ..., 4.3654e-01,
6.2021e-01, 6.3407e-01],
[-4.8061e+00, -1.6607e+00, 8.8092e-01, ..., 1.4559e+00,
-2.8916e-01, -3.1404e-01],
[-5.7642e+00, -5.3741e+00, -3.8993e+00, ..., -2.2713e-01,
4.4137e-01, 1.2506e+00],
...,
[-1.0847e+00, -2.4639e+00, -1.8583e+00, ..., -2.2701e+00,
-2.7670e+00, -2.5946e+00],
[-2.1640e+00, -4.0090e+00, -1.1469e+00, ..., -2.4334e-01,
-1.8555e+00, -2.2072e+00],
[-1.3873e+00, -1.5905e+00, -1.5096e-01, ..., -2.1781e+00,
-2.9720e+00, -1.8081e+00]],
[[-3.2082e+00, -2.4373e+00, -4.7074e+00, ..., -1.8385e+00,
-1.4624e+00, -1.5535e-01],
[-4.0337e+00, -2.4709e+00, -5.2881e+00, ..., -5.8602e-01,
-2.7367e+00, -1.6579e+00],
[-3.8294e+00, -2.5960e+00, -9.5281e+00, ..., -2.1810e+00,
-2.7722e+00, -1.1638e+00],
...,
[-2.7784e+00, -2.3544e+00, -3.2385e+00, ..., -5.4740e+00,
-4.3980e+00, -2.1529e+00],
[-1.9552e+00, 3.3943e-01, -1.7573e+00, ..., -3.6266e+00,
-3.6575e+00, -1.5496e+00],
[-2.0859e+00, -8.9449e-01, -1.0605e+00, ..., -3.4313e+00,
-5.4437e+00, -1.4015e+00]],
[[ 3.5726e+00, 4.0025e-01, -5.7527e-01, ..., -9.3007e-01,
-5.0590e-01, -6.6300e-01],
[ 4.1918e+00, -2.4090e-01, -1.5678e+00, ..., -1.2701e+00,
-2.3241e+00, -1.3314e+00],
[ 2.0875e+00, -1.9033e+00, -2.8948e+00, ..., -3.8497e+00,
-4.1057e+00, -1.9099e+00],
...,
[ 2.9717e+00, 1.8399e+00, 1.7968e+00, ..., -6.0404e+00,
-3.7585e+00, 1.3404e+00],
[ 3.6948e+00, 3.5299e+00, 2.2268e+00, ..., -3.9000e+00,
-1.6184e+00, 2.0389e+00],
[ 2.4224e+00, 2.1792e+00, 1.5524e+00, ..., 8.1737e-01,
1.1751e+00, 7.4870e-01]]]], device='cuda:0'),
tensor([[[[-3.0826e-01, 9.3700e-01, 5.3948e-01, ..., -1.7192e-02,
5.5894e-01, -9.7149e-02],
[ 2.1406e-01, 7.2633e-02, 2.2241e+00, ..., 1.4308e-01,
7.7400e-01, -5.1412e-01],
[-1.2207e+00, -4.0250e-02, 9.9350e-01, ..., -7.4459e-01,
1.7362e-01, -9.0417e-01],
...,
[-3.0785e-01, -4.8616e-01, 3.8834e-01, ..., -3.4072e-01,
3.1133e-01, -2.3408e-01],
[-1.5691e+00, -2.3508e-01, 8.1030e-02, ..., 2.1976e-01,
4.4145e-01, -1.0857e+00],
[-1.2394e+00, -2.3943e+00, -1.7757e+00, ..., -3.1689e+00,
-2.9014e+00, -2.7450e+00]],
[[-1.5968e+00, -2.4245e+00, -3.8305e+00, ..., -1.8566e+00,
-3.4487e+00, -2.1309e+00],
[-4.7913e+00, -4.6606e+00, -3.6683e+00, ..., -2.1492e+00,
-3.8606e+00, -3.5951e+00],
[-4.0111e+00, -4.3972e+00, -4.5986e+00, ..., -3.5863e+00,
-2.4760e+00, -2.8752e+00],
...,
[-4.6674e+00, -5.3840e+00, -4.6700e+00, ..., -2.9948e+00,
-3.0952e+00, -2.4784e+00],
[-7.8043e+00, -7.9408e+00, -6.3325e+00, ..., -4.8604e+00,
-5.1505e+00, -4.9863e+00],
[-1.4588e+00, -3.1682e+00, -1.9746e+00, ..., -3.0153e+00,
-2.8210e+00, -1.6128e+00]],
[[-1.7171e-01, 2.3504e+00, 1.6616e+00, ..., 1.6090e+00,
1.7041e+00, 8.7503e-01],
[-5.3317e-01, 1.8723e-01, 1.1727e+00, ..., 8.9274e-01,
1.4665e+00, 7.8091e-01],
[-1.4161e-01, 1.7718e+00, 1.6859e+00, ..., 1.6038e+00,
2.1689e+00, 1.4239e+00],
...,
[-6.8846e-01, 4.5080e-01, 1.4449e+00, ..., 9.6960e-01,
1.0716e+00, 7.4694e-01],
[-1.2292e+00, 2.7943e-02, 9.6652e-01, ..., 1.3393e+00,
9.8885e-01, 4.4528e-01],
[-3.2865e-01, -3.4529e-01, 5.1224e-02, ..., 1.6249e+00,
1.5400e+00, 5.5195e-01]],
...,
[[-1.3463e+00, -2.3101e+00, -2.4015e+00, ..., -1.4228e+00,
-1.9040e+00, -4.3107e-01],
[-3.4856e+00, -3.5655e+00, -2.3140e+00, ..., -2.8200e+00,
-3.4545e+00, -3.0341e+00],
[-2.7942e+00, -2.5089e+00, -2.1614e+00, ..., -2.5339e+00,
-2.8538e+00, -1.7984e+00],
...,
[-9.9892e-01, -2.4543e+00, -1.2873e+00, ..., -1.5120e+00,
-2.3607e+00, -9.8146e-01],
[-2.1760e+00, -4.0305e+00, -1.5878e+00, ..., -1.4936e+00,
-2.7690e+00, -1.5846e+00],
[-1.4064e+00, -1.5137e+00, -7.1284e-01, ..., -1.8968e+00,
-2.2279e+00, -6.0869e-01]],
[[-1.8399e+00, -1.9062e+00, -2.8910e+00, ..., -5.1239e-01,
-1.3668e+00, -7.0861e-01],
[-4.0160e+00, -2.2080e+00, -2.7840e+00, ..., -1.6492e+00,
-3.6195e+00, -1.8342e+00],
[-3.7858e+00, -2.7981e+00, -4.1999e+00, ..., -9.7561e-01,
-4.3453e+00, -2.4847e+00],
...,
[-2.9031e+00, -2.7827e+00, -3.3789e+00, ..., -8.3316e-01,
-2.8690e+00, -1.9978e+00],
[-2.0797e+00, -2.4119e-01, -1.7083e+00, ..., -6.2931e-01,
-3.1986e+00, -1.8836e+00],
[-2.1784e+00, -1.5546e+00, -1.6981e+00, ..., -1.6201e+00,
-2.7548e+00, -1.2407e+00]],
[[ 1.1963e+00, 2.8036e-01, 8.5924e-01, ..., 1.0951e-01,
1.2698e+00, 6.3853e-01],
[ 3.9897e+00, 3.0141e+00, 2.5174e+00, ..., 4.5389e-01,
1.1016e+00, 2.4601e+00],
[ 1.6832e+00, 5.7668e-01, 5.1963e-01, ..., 1.7820e-01,
2.0027e-01, 7.9768e-01],
...,
[ 3.0366e+00, 2.3634e+00, 2.1737e+00, ..., 1.2351e+00,
2.9600e-01, 4.9991e-01],
[ 3.8107e+00, 4.0153e+00, 2.5915e+00, ..., 1.1873e+00,
7.9327e-01, 1.2230e+00],
[ 2.3916e+00, 2.2961e+00, 1.5517e+00, ..., -1.7889e-01,
-3.4648e-01, -1.4776e-01]]],
[[[ 2.4266e-01, 1.4276e+00, 5.6951e-01, ..., 3.2789e-01,
1.0433e-01, 5.0005e-01],
[ 4.9920e-01, 3.9560e-01, 6.2868e-01, ..., 2.0550e+00,
2.0005e-01, 6.8335e-01],
[-3.7179e-01, 3.2929e-01, 5.6394e-01, ..., 2.8487e+00,
-4.6132e-01, 2.4440e-01],
...,
[-9.5500e-01, -1.8999e+00, 7.5173e-01, ..., -7.3259e-01,
-1.2732e+00, -6.8643e-01],
[-1.0276e+00, -9.2304e-01, 9.7219e-02, ..., -4.2201e-01,
-8.6423e-01, -1.6111e+00],
[-2.7186e+00, -5.1759e+00, -4.0408e+00, ..., -4.8546e+00,
-4.4160e+00, -3.8693e+00]],
[[-6.6624e-01, -2.2187e+00, -4.2193e+00, ..., -2.1754e+00,
-5.9389e+00, -4.1738e-01],
[-4.3645e+00, -5.8778e+00, -5.2941e+00, ..., -2.3973e+00,
-9.1538e+00, -1.5819e+00],
[-3.2700e+00, -4.0533e+00, -3.4315e+00, ..., -1.5831e+00,
-8.4596e+00, -1.9510e+00],
...,
[-5.7881e+00, -5.9648e+00, -4.0384e+00, ..., -7.7042e+00,
-5.4248e+00, -3.7788e+00],
[-6.9745e+00, -7.9464e+00, -6.0770e+00, ..., -8.1737e+00,
-7.9363e+00, -6.5975e+00],
[-2.5323e+00, -4.5390e+00, -3.6948e+00, ..., -4.2705e+00,
-4.1667e+00, -2.5779e+00]],
[[-2.2814e-01, 2.1823e+00, 1.2485e+00, ..., 2.3078e+00,
2.2548e+00, 7.4181e-01],
[-2.0271e-01, 1.1824e+00, 8.7145e-01, ..., 2.8230e+00,
1.6079e+00, -9.8070e-01],
[-1.9622e-01, 2.0826e+00, 1.1454e+00, ..., 2.8897e+00,
2.7258e+00, -4.4110e-01],
...,
[ 9.1425e-01, 1.2448e+00, 9.9238e-01, ..., 8.1021e-01,
9.1865e-01, 1.9572e-01],
[ 8.2711e-01, 7.3301e-01, 3.3411e-01, ..., 4.0667e-01,
3.1833e-01, 1.9622e-01],
[ 1.3407e+00, 1.5815e+00, 1.0364e+00, ..., 9.8353e-01,
1.3469e+00, 1.3593e-01]],
...,
[[-7.1281e-01, -1.8212e+00, -1.8758e+00, ..., -2.5331e+00,
-2.6087e+00, -5.8447e-01],
[-2.9239e+00, -4.2527e+00, -2.7086e+00, ..., -4.6387e+00,
-2.8854e+00, -3.1412e+00],
[-1.1877e+00, -2.2133e+00, -7.9739e-01, ..., -4.5861e+00,
-2.2551e+00, -2.2359e+00],
...,
[-2.7613e+00, -2.2036e+00, -4.6137e-01, ..., -8.4996e-01,
-4.6802e+00, -1.6250e+00],
[-2.9925e+00, -2.7177e+00, -1.0864e+00, ..., -8.1789e-01,
-4.7647e+00, -2.2890e+00],
[-3.5105e+00, -3.5608e+00, -2.7870e+00, ..., -2.7404e+00,
-4.1637e+00, -1.6513e+00]],
[[-1.6866e+00, -2.2573e+00, -2.9361e+00, ..., -1.3795e+00,
-1.5830e+00, -1.6032e+00],
[-4.1870e+00, -3.5880e+00, -2.8645e+00, ..., -2.5977e+00,
-1.4190e+00, -1.5494e+00],
[-3.1307e+00, -2.9994e+00, -2.6110e+00, ..., -2.8420e+00,
-2.0113e+00, -1.6238e+00],
...,
[-4.1706e+00, -2.6638e+00, -3.1972e+00, ..., -2.5959e+00,
-3.3868e+00, -1.6651e+00],
[-3.7190e+00, -7.2119e-01, -2.2496e+00, ..., -1.6854e+00,
-3.2008e+00, -1.3066e+00],
[-4.4231e+00, -2.8979e+00, -2.6661e+00, ..., -2.4153e+00,
-3.1372e+00, -1.2373e+00]],
[[ 1.0141e+00, 6.5305e-01, 7.2499e-01, ..., 1.5551e+00,
4.6234e-01, -8.9432e-01],
[ 3.2262e+00, 3.3320e+00, 2.3083e+00, ..., 4.0009e+00,
2.9103e+00, 1.6843e+00],
[ 1.3977e+00, 7.5184e-01, 2.2581e-01, ..., 3.3921e+00,
1.4505e+00, 1.2301e+00],
...,
[ 2.8310e+00, 1.0031e+00, 1.9379e+00, ..., 1.6093e+00,
1.8541e+00, 1.3106e+00],
[ 3.6606e+00, 3.3735e+00, 2.2926e+00, ..., 2.5079e+00,
2.0846e+00, 2.0891e+00],
[ 1.8998e+00, 6.7598e-01, 3.4105e-02, ..., 2.5069e-01,
-1.5635e-01, -1.9256e-01]]],
[[[ 1.5853e+00, 2.0541e+00, 6.5801e-01, ..., -9.3748e-01,
-2.4755e+00, -1.0811e+00],
[ 5.2188e-01, -2.4775e+00, -4.1982e+00, ..., 3.1671e+00,
8.4873e-02, -5.6455e-01],
[-1.5041e+00, -7.5451e+00, -9.0169e-01, ..., -1.5775e+00,
-7.2676e+00, -4.9329e+00],
...,
[-3.7256e-01, 4.8770e-01, 9.2470e-01, ..., -5.2788e+00,
-8.3072e-02, 1.3310e+00],
[-1.5332e+00, 3.1406e-01, 1.8502e+00, ..., -4.9444e+00,
-5.7361e+00, -1.2511e+00],
[-1.0344e+00, -2.0666e+00, -1.2193e+00, ..., -4.1382e+00,
-4.8958e+00, -1.8060e+00]],
[[-1.4847e+00, -2.6839e+00, -3.6185e+00, ..., -5.4769e+00,
-8.5580e+00, -2.8821e+00],
[-7.0834e+00, -1.0304e+01, -7.9486e+00, ..., -6.1611e+00,
-8.1700e+00, -4.4296e+00],
[-8.3137e+00, -9.6747e+00, -8.4753e-01, ..., -9.8327e+00,
-1.2649e+01, -3.7089e+00],
...,
[-4.2154e+00, -5.6606e+00, -5.8113e+00, ..., -1.2894e+01,
-1.3678e+01, -2.1111e-01],
[-7.2581e+00, -7.7363e+00, -7.5501e+00, ..., -7.5547e+00,
-1.3537e+01, -8.9531e+00],
[-1.2109e+00, -3.0687e+00, -2.1731e+00, ..., -3.1715e+00,
-3.1692e+00, 1.5206e-01]],
[[-2.0405e-01, 1.4982e+00, 1.0812e-01, ..., 2.4622e+00,
2.7487e+00, 4.2496e-01],
[-1.5494e+00, -8.1429e-01, 1.9913e+00, ..., 1.2986e+00,
2.3296e+00, -3.3201e-01],
[ 3.9842e-01, 2.8859e+00, 1.5192e+00, ..., -1.0902e+00,
4.4175e+00, 3.0918e+00],
...,
[-1.1272e+00, -4.8301e-02, -7.9370e-01, ..., 2.0435e+00,
-4.1200e+00, -2.4554e+00],
[-1.6643e+00, 4.7978e-03, 6.3362e-01, ..., 1.3228e+00,
-1.6635e+00, -1.3275e+00],
[-8.0523e-01, -4.6952e-01, 4.8012e-01, ..., -5.6687e-01,
4.3537e-01, -1.5779e+00]],
...,
[[-5.8241e-01, -2.5465e+00, -1.5281e+00, ..., -3.1898e+00,
-5.7227e+00, -1.1176e+00],
[-4.7167e+00, -7.4354e+00, -3.6494e+00, ..., -2.3926e+00,
-5.1619e+00, -2.7306e+00],
[-4.2283e+00, -4.5130e+00, 3.6006e+00, ..., -6.3770e+00,
-6.4820e+00, -2.3894e+00],
...,
[-9.4862e-01, -2.6618e+00, -1.3597e+00, ..., -6.3761e+00,
-7.3377e+00, -4.4158e+00],
[-1.7367e+00, -3.9347e+00, -8.5638e-01, ..., -2.2931e+00,
-6.4907e+00, -5.4359e+00],
[-1.0355e+00, -1.4167e+00, 3.2063e-02, ..., 2.2243e-01,
-5.0896e-01, 1.7820e+00]],
[[-2.2300e+00, -2.3251e+00, -1.8450e+00, ..., -2.2341e+00,
-1.1079e+00, 2.2618e-01],
[-4.9272e+00, -4.0164e+00, -4.3124e+00, ..., -2.2975e+00,
-2.1546e+00, 4.9004e-01],
[-4.9864e+00, -2.2229e+00, -4.1725e+00, ..., -2.5266e+00,
-6.4909e+00, -2.0423e+00],
...,
[-2.3054e+00, -2.2679e+00, -2.4370e+00, ..., 3.0524e+00,
6.3867e+00, -1.1275e+00],
[-1.5863e+00, 1.2249e-01, -1.1948e+00, ..., -2.2002e+00,
7.4558e-01, 1.8409e+00],
[-1.7507e+00, -1.1568e+00, -2.1096e-01, ..., 8.8064e-02,
-5.4046e-01, 1.7246e+00]],
[[ 1.2422e+00, 2.5973e-01, 8.6578e-02, ..., 1.7466e+00,
1.5767e+00, 1.1691e-01],
[ 4.5784e+00, 1.7636e+00, 1.0575e+00, ..., 4.9960e+00,
3.8424e+00, 3.2911e+00],
[ 3.1503e+00, -2.1197e+00, 4.4833e-01, ..., 1.7588e+00,
1.0435e-01, 1.6545e-01],
...,
[ 2.7520e+00, 2.1873e+00, 3.0386e+00, ..., 9.5239e+00,
7.7616e+00, 8.4355e-01],
[ 3.4934e+00, 3.5967e+00, 3.4848e+00, ..., 5.3048e-01,
4.3365e+00, 3.9091e+00],
[ 2.3286e+00, 2.2119e+00, 1.3590e+00, ..., -7.3837e-01,
-1.3001e+00, 1.0278e+00]]],
[[[ 2.9393e-01, 1.2409e+00, -8.2556e-02, ..., -3.3076e+00,
-2.3796e+00, -1.0095e+00],
[ 1.8892e-01, 5.2594e-01, 1.7429e+00, ..., -3.4883e+00,
-3.3712e+00, -2.7450e-01],
[-4.5370e-01, 1.9076e-02, 1.2545e+00, ..., -3.9601e+00,
-4.5829e+00, -6.6135e-01],
...,
[-7.1721e-01, -3.4222e+00, -2.6521e-01, ..., -4.9884e+00,
-4.4956e+00, 7.6831e-01],
[-1.8664e+00, -1.8804e+00, -1.6136e+00, ..., -3.6634e+00,
-3.5028e+00, 8.9651e-01],
[-1.1729e+00, -3.1317e+00, -2.9626e+00, ..., -3.7176e+00,
-3.1593e+00, -1.3269e+00]],
[[-1.0295e+00, -1.6947e+00, -3.2318e+00, ..., -6.5083e+00,
-6.7560e+00, -1.3019e+00],
[-4.0829e+00, -4.1128e+00, -2.8949e+00, ..., -1.0389e+01,
-9.3204e+00, -2.6138e+00],
[-3.8481e+00, -4.1839e+00, -4.1585e+00, ..., -5.8772e+00,
-9.1939e+00, -2.9565e+00],
...,
[-5.7737e+00, -6.0912e+00, -5.3901e+00, ..., -9.8228e+00,
-8.7325e+00, -1.8696e+00],
[-7.9670e+00, -6.9224e+00, -5.6496e+00, ..., -8.8982e+00,
-9.9780e+00, -4.5731e+00],
[-1.9403e+00, -3.2954e+00, -3.3746e+00, ..., -2.8380e+00,
-3.6084e+00, -7.4479e-01]],
[[-1.1750e-01, 1.2641e+00, 1.7374e+00, ..., 1.2715e+00,
2.2513e+00, 9.6206e-01],
[-5.8978e-02, -1.5523e-01, 7.9374e-01, ..., 1.1586e+00,
3.0321e+00, -9.7564e-02],
[ 6.3774e-01, 1.1409e+00, 1.4830e+00, ..., 2.2304e+00,
2.5750e+00, -9.2619e-02],
...,
[ 4.0794e-01, 1.2038e+00, 1.5403e-01, ..., 1.1077e+00,
6.9280e-02, -1.0758e+00],
[ 4.0716e-01, -7.6494e-04, 5.9600e-01, ..., -6.9054e-01,
1.0129e+00, -1.2666e+00],
[ 4.7338e-01, 7.7382e-01, 8.6680e-01, ..., -1.0502e+00,
-3.8318e-01, -1.8176e+00]],
...,
[[-5.6297e-01, -1.6354e+00, -2.1201e+00, ..., -5.1358e+00,
-4.9923e+00, -5.4570e-01],
[-2.7597e+00, -2.5975e+00, -1.3193e+00, ..., -5.5235e+00,
-6.3122e+00, -1.8340e+00],
[-1.7257e+00, -8.7931e-01, -6.7768e-01, ..., -4.5650e+00,
-4.4372e+00, -2.3301e+00],
...,
[-1.6884e+00, -1.3187e+00, -2.7341e+00, ..., -4.3390e+00,
-5.2787e+00, -2.2320e+00],
[-3.0760e+00, -2.0216e+00, -1.4582e+00, ..., -2.6454e+00,
-5.1686e+00, -1.1893e+00],
[-1.6481e+00, -1.6028e+00, -1.1953e+00, ..., -1.9917e+00,
-2.5358e+00, 3.2680e-01]],
[[-1.2849e+00, -2.4235e+00, -3.8864e+00, ..., -2.1221e+00,
-5.7468e-01, -6.5115e-01],
[-3.6956e+00, -3.2147e+00, -4.1134e+00, ..., -4.1103e+00,
-2.8618e+00, -3.9009e-01],
[-4.4588e+00, -3.3827e+00, -3.9769e+00, ..., 8.9710e-01,
-1.3659e+00, -1.6565e-01],
...,
[-5.1470e+00, -2.5828e+00, -1.7822e+00, ..., -5.8468e+00,
-2.2502e+00, -5.6336e-01],
[-4.1915e+00, -8.5702e-01, -1.4991e+00, ..., -1.4067e+00,
-3.1965e-01, 1.3539e+00],
[-3.2782e+00, -2.5487e+00, -1.5739e+00, ..., 3.2103e-02,
-6.3190e-01, 7.2609e-01]],
[[ 1.2576e+00, -5.9605e-02, 1.1329e+00, ..., 3.1370e-01,
6.5302e-01, -1.3866e+00],
[ 2.2579e+00, 9.9895e-01, 1.3900e+00, ..., 3.4021e+00,
2.5327e+00, 2.2895e+00],
[ 1.2727e+00, 7.0584e-01, 9.0892e-01, ..., 2.7476e+00,
1.5142e+00, 6.0212e-01],
...,
[ 2.3387e+00, 7.9870e-01, 2.6867e+00, ..., 1.9503e+00,
2.8039e+00, 3.9738e-01],
[ 3.2941e+00, 3.4624e+00, 2.0549e+00, ..., 2.7947e+00,
1.0482e+00, 1.6077e+00],
[ 2.1193e+00, 1.8262e+00, 5.3304e-01, ..., 1.4852e+00,
-1.3912e-01, 8.0951e-01]]],
[[[-1.4326e-01, 1.4731e+00, 1.9715e+00, ..., 3.6072e+00,
-2.1363e+00, 2.9474e+00],
[-8.9728e-02, 1.2984e+00, 2.3939e+00, ..., 5.8896e+00,
-3.1076e+00, 3.6388e+00],
[-7.6997e-01, 1.1508e+00, 1.0635e+00, ..., -2.6231e+00,
-1.1120e+00, 3.9645e+00],
...,
[-3.3178e-01, -1.7879e+00, 1.3894e+00, ..., -5.0701e+00,
-4.5484e+00, -3.1003e+00],
[-9.8501e-01, -1.2575e+00, 3.6213e-01, ..., -1.2262e+00,
-1.6926e+00, -1.4134e+00],
[-2.6773e+00, -5.5334e+00, -4.5067e+00, ..., -4.7128e+00,
-3.8576e+00, -3.2690e+00]],
[[-9.8336e-01, -1.4575e+00, -3.5778e+00, ..., -8.2049e-01,
-9.4203e+00, 1.8481e+00],
[-4.3192e+00, -5.4271e+00, -4.9401e+00, ..., -3.8136e+00,
-9.7813e+00, -3.1204e-01],
[-2.9484e+00, -5.8332e+00, -1.0107e+01, ..., -5.9095e+00,
-3.0625e+00, 8.4408e-01],
...,
[-6.0850e+00, -6.0854e+00, -4.8938e+00, ..., -9.9768e+00,
-7.9468e+00, -3.8732e+00],
[-7.7508e+00, -8.6836e+00, -6.7123e+00, ..., -8.0101e+00,
-8.6015e+00, -6.0700e+00],
[-2.5569e+00, -5.0720e+00, -4.4133e+00, ..., -4.1799e+00,
-4.3758e+00, -2.3737e+00]],
[[-4.5967e-01, 1.5714e+00, 1.5785e+00, ..., -4.5386e+00,
-2.2862e+00, -1.5974e+00],
[-2.0811e-02, 2.6668e-01, -2.3649e+00, ..., -1.1793e+01,
-9.0548e+00, -6.4005e+00],
[-1.8264e-01, 2.5852e+00, -1.4038e+00, ..., -2.0646e+01,
-2.6114e+01, -1.5420e+01],
...,
[ 9.8654e-01, 1.3683e+00, -1.5788e-01, ..., 4.9024e-01,
-2.8612e+00, -3.9384e+00],
[ 7.8821e-01, 1.1464e+00, 1.0384e+00, ..., -4.7724e-01,
-1.8505e+00, -4.8620e-01],
[ 1.3889e+00, 1.5905e+00, 1.1641e+00, ..., 5.0273e-01,
9.7741e-01, -2.8363e-01]],
...,
[[-1.0963e+00, -5.4254e-01, -1.3372e+00, ..., 4.1250e+00,
4.6499e+00, 3.1625e+00],
[-3.3581e+00, -2.7115e+00, -2.2242e+00, ..., 9.4131e+00,
9.5736e+00, 6.2850e+00],
[-1.3285e+00, -1.7957e+00, -3.4994e+00, ..., 9.2815e+00,
8.2689e+00, 3.8471e+00],
...,
[-2.3943e+00, -3.0491e+00, -2.5070e-01, ..., -4.1840e+00,
-3.9497e+00, -2.5178e+00],
[-2.9540e+00, -3.6586e+00, -1.8647e+00, ..., -2.3789e+00,
-5.8452e+00, -2.6093e+00],
[-3.7710e+00, -4.5111e+00, -3.5647e+00, ..., -3.6183e+00,
-4.5114e+00, -1.3711e+00]],
[[-1.5658e+00, -1.4712e+00, -3.4873e+00, ..., 3.4683e+00,
4.1834e+00, 2.8029e+00],
[-4.1445e+00, -3.2285e+00, -3.0883e+00, ..., -1.0308e+00,
1.0770e+01, 4.4411e+00],
[-3.2616e+00, -4.8135e+00, -4.9526e+00, ..., 3.7815e+00,
1.0221e+01, 6.1285e+00],
...,
[-3.4051e+00, -1.9481e+00, -2.0137e+00, ..., -2.5504e-01,
3.7599e-01, 1.6697e+00],
[-3.6069e+00, 2.9189e-01, -4.3580e-01, ..., 3.1941e-01,
-2.3529e+00, 1.7090e-01],
[-4.7642e+00, -2.6472e+00, -1.9537e+00, ..., -1.0469e+00,
-2.4938e+00, -5.4325e-01]],
[[ 1.0240e+00, 7.6941e-02, 5.0302e-01, ..., 2.2504e+00,
-2.0265e+00, 3.4352e+00],
[ 3.3297e+00, 3.0856e+00, 2.9932e+00, ..., 4.2071e+00,
-1.4039e+00, 6.1888e+00],
[ 9.9119e-01, 1.5795e+00, 1.7234e+00, ..., -1.2492e+00,
1.3592e+00, 5.2856e+00],
...,
[ 3.4966e+00, 2.1003e+00, 2.2092e+00, ..., 1.8942e+00,
2.8156e+00, 3.1311e+00],
[ 4.4689e+00, 3.7303e+00, 2.6864e+00, ..., 3.6591e+00,
3.5197e+00, 2.7042e+00],
[ 2.3049e+00, 9.6137e-01, 3.3572e-02, ..., 1.9739e-01,
-7.9715e-01, -2.0895e-01]]]], device='cuda:0'),
tensor([[[[ 6.1925e-01, -5.9081e-01, -2.9972e+00, ..., 8.3632e-02,
3.6322e-01, -2.0111e-01],
[ 7.8374e-01, -2.6865e+00, 1.2481e-01, ..., 6.3275e-01,
4.9240e-01, -5.4137e-01],
[-6.5191e-01, -3.4075e+00, 9.6345e-01, ..., 5.8962e-01,
2.0190e-03, -9.4429e-01],
...,
[ 7.5560e-01, -1.4128e+00, -3.6403e+00, ..., 1.4287e+00,
1.6170e+00, -9.5217e-02],
[ 1.0818e+00, -3.4827e-01, -1.7480e+00, ..., 1.9510e+00,
4.9210e-01, -1.2853e+00],
[ 1.6823e-01, -4.9794e-02, 3.0246e-01, ..., -2.3896e+00,
-2.9674e+00, -2.9609e+00]],
[[-1.6282e+00, -4.4621e+00, -5.3528e+00, ..., -2.4977e+00,
-3.6995e+00, -2.2083e+00],
[-4.9700e+00, -6.0264e+00, -1.2061e+00, ..., -2.6709e+00,
-4.0771e+00, -3.5461e+00],
[-3.8745e+00, -5.2132e+00, -1.4470e-01, ..., -2.0209e+00,
-2.6565e+00, -2.9638e+00],
...,
[ 1.6135e+00, 7.7141e-02, -4.0101e+00, ..., -4.8899e-01,
-2.4646e+00, -2.3780e+00],
[-1.1492e+00, -1.0195e+00, -4.6641e+00, ..., -3.4981e+00,
-5.9425e+00, -5.5425e+00],
[ 1.8408e-01, -3.9573e-01, -3.9790e-01, ..., -3.1977e+00,
-2.9745e+00, -1.9216e+00]],
[[ 5.8890e-01, 2.8827e+00, 2.2145e+00, ..., 1.8212e+00,
1.5918e+00, 9.0867e-01],
[-1.4125e-01, 1.9684e+00, -1.7937e-01, ..., 1.3904e+00,
1.2802e+00, 8.2910e-01],
[ 9.3390e-01, 2.7950e+00, 7.3550e-01, ..., 1.2393e+00,
1.6159e+00, 1.3723e+00],
...,
[-6.4135e-01, 8.5166e-01, -1.3645e+00, ..., 1.1085e+00,
8.9378e-01, 5.3927e-01],
[-1.8675e+00, 1.9222e+00, 7.4216e-01, ..., 1.2609e+00,
8.4120e-01, 3.1809e-01],
[-1.7530e+00, 1.2610e+00, 1.6787e+00, ..., 1.6585e+00,
1.4219e+00, 4.2798e-01]],
...,
[[-1.5235e+00, -3.7416e+00, -2.8193e+00, ..., -1.6419e+00,
-2.0940e+00, -4.4372e-01],
[-3.3654e+00, -5.1527e+00, 7.6111e-01, ..., -1.9114e+00,
-3.7642e+00, -2.9894e+00],
[-2.3924e+00, -4.0240e+00, -5.4035e-01, ..., -7.5566e-01,
-3.2492e+00, -1.8688e+00],
...,
[ 1.6363e+00, -6.7850e-01, 4.4978e-02, ..., -9.8505e-01,
-2.6455e+00, -1.2198e+00],
[ 6.7337e-01, -2.3665e+00, -7.1164e-01, ..., -1.2176e+00,
-2.9382e+00, -2.0787e+00],
[ 2.9037e-01, -7.4658e-01, 1.6179e+00, ..., -2.1903e+00,
-2.2288e+00, -8.2237e-01]],
[[-2.5157e+00, -2.2744e+00, -1.8790e+00, ..., 5.4214e-01,
-1.3157e+00, -6.9397e-01],
[-4.9807e+00, -1.5037e+00, 5.2262e-01, ..., 5.8895e-03,
-3.4617e+00, -1.8148e+00],
[-4.2040e+00, -1.1322e+00, -4.5228e-01, ..., -3.7861e-01,
-4.1334e+00, -2.4096e+00],
...,
[-2.3691e+00, -5.4907e+00, -3.7765e+00, ..., -2.2286e+00,
-4.2236e+00, -1.7021e+00],
[-2.0828e+00, -5.5148e+00, -3.6048e+00, ..., -1.7635e+00,
-3.9175e+00, -1.4591e+00],
[-3.1519e-01, -1.1668e+00, -4.7312e-01, ..., -1.3754e+00,
-2.2425e+00, -9.7141e-01]],
[[ 1.8729e+00, 1.3822e+00, -7.9640e-01, ..., 6.8026e-01,
1.1210e+00, 5.7414e-01],
[ 3.4395e+00, 1.6237e+00, 1.0091e+00, ..., 1.5630e+00,
1.1398e+00, 2.4809e+00],
[ 1.7253e+00, -4.1827e-01, -6.2774e-01, ..., 1.1978e+00,
1.7905e-01, 7.8570e-01],
...,
[ 5.7791e-01, -2.7004e+00, -3.5435e+00, ..., -1.9376e-01,
9.5217e-01, 7.1735e-01],
[ 1.6802e+00, -2.8652e-01, -1.7989e+00, ..., 1.5690e+00,
1.5631e+00, 1.4343e+00],
[ 2.1723e+00, 2.6550e-01, 8.8027e-02, ..., 7.7238e-01,
-3.1470e-01, 1.4405e-01]]],
[[[-1.8314e+00, -2.3098e+00, -4.3385e+00, ..., -1.4038e+00,
-1.1444e+00, -9.5913e-01],
[-3.3930e+00, -2.9204e+00, -3.3346e+00, ..., -8.5160e-01,
3.4116e-03, -1.4068e+00],
[-8.7808e-01, -3.2331e-01, -5.9670e-01, ..., -2.3196e+00,
-8.5450e-01, -2.2465e+00],
...,
[ 2.5815e-01, -1.1269e-01, -1.2875e-01, ..., -3.7926e+00,
-4.9078e+00, -9.0442e-01],
[-2.0481e+00, -6.3717e-01, 1.1111e+00, ..., 2.1834e-01,
-2.9733e+00, -5.4283e-01],
[-1.6581e+00, -3.1557e+00, -1.9676e+00, ..., 1.3152e+00,
-1.4789e-01, -1.7509e+00]],
[[-2.7842e+00, -3.7691e+00, -5.7593e+00, ..., -2.9494e+00,
-4.9816e+00, -2.7790e+00],
[-5.0998e+00, -2.9718e+00, -5.8290e+00, ..., -3.9990e+00,
-5.3703e+00, -4.9650e+00],
[ 5.1419e-02, 4.4097e-01, -8.1620e-01, ..., -3.9128e+00,
-4.4525e+00, -3.6275e+00],
...,
[-5.6654e+00, -6.7943e+00, -5.1309e+00, ..., -5.4779e+00,
-1.0636e+01, -2.3120e+00],
[-8.7374e+00, -1.0876e+01, -9.2772e+00, ..., -3.8224e+00,
-1.0814e+01, -4.0588e+00],
[-1.9639e+00, -4.4000e+00, -3.6385e+00, ..., 8.8574e-01,
-2.5375e+00, -9.7774e-01]],
[[ 7.7237e-01, 3.1026e+00, 4.6771e+00, ..., 1.4682e+00,
1.9578e+00, 8.1975e-01],
[ 2.7681e+00, 1.0524e+00, 2.7569e+00, ..., 9.2197e-01,
2.4109e+00, 9.7051e-01],
[-5.6516e-01, -3.3435e+00, -5.7019e+00, ..., 1.1155e+00,
2.4248e+00, 2.0646e+00],
...,
[-1.2539e+00, -8.9398e-01, 7.8189e-01, ..., 5.0912e+00,
2.4347e+00, -1.0694e+00],
[-9.3937e-01, -2.2996e-01, -1.9886e+00, ..., 5.9804e+00,
4.7787e+00, -4.3353e-01],
[-3.7149e-02, 7.8355e-01, 2.6580e-01, ..., -3.8298e+00,
-9.1719e-01, -2.0980e+00]],
...,
[[-1.8739e+00, -3.2248e+00, -5.7169e+00, ..., -2.7817e+00,
-3.5605e+00, -1.2550e+00],
[-4.4997e+00, -1.3559e+00, -4.8499e+00, ..., -4.1113e+00,
-4.6056e+00, -3.6559e+00],
[-3.0739e-01, 8.9234e-01, -8.9620e-01, ..., -5.1853e+00,
-4.3326e+00, -3.1754e+00],
...,
[-1.3845e+00, -3.0045e+00, -4.2177e-01, ..., -8.1517e+00,
-6.8404e+00, -2.9890e+00],
[-3.1995e+00, -5.9973e+00, -1.9929e+00, ..., -3.5035e+00,
-4.0959e+00, -8.2114e-01],
[-2.5039e+00, -3.4447e+00, -1.9171e+00, ..., -5.3784e-01,
-1.2117e+00, 1.3296e+00]],
[[-6.9633e-01, -2.3924e+00, -6.5058e+00, ..., -3.4742e+00,
-2.8733e+00, -4.5139e-01],
[-3.7132e+00, -2.2052e+00, -8.0072e+00, ..., -4.2263e+00,
-5.9456e+00, -1.9697e+00],
[-2.5486e+00, -2.3000e+00, -1.1250e-01, ..., -2.1795e+00,
-4.6824e+00, -1.9569e+00],
...,
[-3.3125e+00, -2.6191e+00, -1.9144e+00, ..., -4.3020e+00,
-3.0577e-01, -5.4991e-01],
[-2.5920e+00, -7.3712e-02, -1.1505e+00, ..., -3.6387e+00,
-1.4355e+00, 8.3916e-01],
[-3.0443e+00, -2.5457e+00, -2.5449e+00, ..., 1.1654e+00,
-9.7322e-01, 9.6567e-01]],
[[ 1.3997e+00, 1.1600e+00, 1.2060e+00, ..., 1.5832e-01,
1.4953e+00, 6.7373e-01],
[ 2.3553e+00, 4.2715e+00, 9.2748e-01, ..., 4.0831e-01,
2.1098e+00, 3.2771e+00],
[ 5.6932e-01, -9.2616e-02, -5.2443e+00, ..., -2.6994e-01,
1.0784e+00, 6.7159e-01],
...,
[ 3.8331e+00, 3.0931e+00, 1.7300e+00, ..., 4.5857e+00,
1.4466e+00, -3.3724e-01],
[ 3.6700e+00, 3.9931e+00, 3.4568e+00, ..., 2.8740e+00,
2.6953e+00, 2.1053e+00],
[ 2.6711e+00, 2.7579e+00, 1.8996e+00, ..., -6.3276e-01,
7.0581e-01, 1.3682e+00]]],
[[[-3.6489e+00, -2.6764e+00, 5.2920e-01, ..., -2.4423e+00,
-3.6879e+00, -1.6233e+00],
[-2.0762e+00, -1.3462e+00, 3.2127e-01, ..., -8.0432e-01,
-1.4721e+00, -7.6479e-01],
[-1.2434e+00, -1.1782e+00, -2.8580e+00, ..., -2.3278e-01,
-7.9968e-01, -1.1285e+00],
...,
[-8.8834e-01, -2.4349e+00, -1.9038e+00, ..., -7.4408e-01,
3.8539e-01, -3.3965e-01],
[-9.9361e-01, -7.7435e-02, 3.1824e-01, ..., -4.0260e-01,
-7.2619e-02, -1.8115e+00],
[-2.4920e+00, -4.4655e+00, -3.5082e+00, ..., -4.3708e+00,
-4.1182e+00, -3.8486e+00]],
[[-4.0678e+00, -1.1509e+00, 1.4499e+00, ..., -5.9287e+00,
-8.5496e+00, -2.2265e+00],
[-2.4995e+00, 5.8551e-01, -1.1386e+00, ..., -5.6633e+00,
-7.6287e+00, -2.9772e+00],
[-4.3247e+00, -3.6665e+00, -7.0061e+00, ..., -8.7998e+00,
-9.9212e+00, -5.2803e+00],
...,
[-5.6055e+00, -4.7246e+00, -5.1926e+00, ..., -3.6689e+00,
-4.3357e+00, -3.3920e+00],
[-7.2126e+00, -7.3028e+00, -5.5062e+00, ..., -7.3945e+00,
-8.2021e+00, -6.6477e+00],
[-2.4832e+00, -4.1971e+00, -4.3461e+00, ..., -3.7737e+00,
-4.0580e+00, -2.4412e+00]],
[[ 2.2320e+00, 1.2691e+00, -1.9872e+00, ..., 2.0059e+00,
2.9288e+00, 9.6112e-01],
[ 1.1046e+00, -4.5900e+00, -9.3360e+00, ..., 1.4181e+00,
2.9848e+00, 4.8811e-01],
[ 3.2672e-01, -4.9306e+00, -1.0723e+01, ..., -1.1152e-01,
3.1973e-01, -2.5242e-01],
...,
[ 2.8467e-01, 3.6442e-01, -5.3199e-01, ..., -5.5036e-01,
-2.4061e-01, -1.7931e-01],
[ 8.4487e-01, -3.8557e-01, -7.1857e-02, ..., 8.9981e-01,
-2.6468e-01, -2.0347e-01],
[ 1.1684e+00, 1.3250e+00, 5.3146e-01, ..., 1.7991e+00,
1.1926e+00, 1.2074e-01]],
...,
[[-2.9401e+00, 1.2908e+00, 2.4546e+00, ..., -4.8964e+00,
-5.9311e+00, -1.8594e+00],
[-2.3351e+00, -4.7918e-01, -1.9516e+00, ..., -3.7958e+00,
-5.0702e+00, -2.5343e+00],
[-1.6675e+00, -4.3552e+00, -7.5315e+00, ..., -5.3004e+00,
-6.3299e+00, -2.7481e+00],
...,
[-2.0199e+00, -2.4461e+00, -2.0163e+00, ..., -5.2912e-01,
-2.1396e+00, -9.3052e-01],
[-2.9746e+00, -2.2295e+00, -2.2748e+00, ..., -2.6298e+00,
-4.3198e+00, -2.4739e+00],
[-3.2796e+00, -3.3030e+00, -2.8733e+00, ..., -2.8273e+00,
-3.4864e+00, -1.5286e+00]],
[[-1.2374e+00, 2.0171e-01, -4.6940e-01, ..., -2.1435e+00,
-9.4043e-01, 7.3263e-01],
[-6.1789e-01, 2.5626e-01, -2.1428e+00, ..., -2.7293e+00,
-2.2502e+00, 7.8680e-02],
[-5.5248e-01, -2.0116e+00, -4.0674e+00, ..., -1.2482e+00,
-1.0446e+00, 2.5840e-01],
...,
[-3.1909e+00, -2.9629e+00, -2.5142e+00, ..., 1.7138e-01,
-1.8450e+00, -1.0347e+00],
[-3.4283e+00, -2.3339e-01, -3.3650e+00, ..., -6.0419e-01,
-2.9213e+00, -1.3913e+00],
[-4.2510e+00, -2.3660e+00, -2.3431e+00, ..., -2.2278e+00,
-3.4198e+00, -1.3625e+00]],
[[ 2.0653e+00, 8.2413e-01, 1.6052e-01, ..., 1.0842e+00,
1.6209e+00, -8.0857e-02],
[ 1.1203e+00, 1.5430e+00, 1.5286e-01, ..., 3.0383e+00,
3.2712e+00, 3.9442e+00],
[ 2.5088e+00, 5.9292e-01, -6.8598e-01, ..., 2.9423e+00,
2.5207e+00, 1.7811e+00],
...,
[ 2.0895e+00, 5.1719e-01, 1.5263e+00, ..., 2.1823e+00,
2.0888e+00, 1.7395e+00],
[ 3.6565e+00, 3.5686e+00, 1.7315e+00, ..., 2.4788e+00,
2.1598e+00, 2.1653e+00],
[ 1.8940e+00, 1.0684e+00, -1.8241e-03, ..., -1.7566e-01,
-3.2706e-01, -2.6445e-01]]],
[[[-2.6629e-01, 1.1431e+00, 1.3945e+00, ..., -1.4208e+00,
-3.4494e+00, -7.9351e-01],
[ 1.3603e-01, 1.5268e-01, 2.5856e+00, ..., 3.0574e-01,
-1.9186e+00, -2.8885e-01],
[-1.2148e+00, -3.8761e-01, 1.5711e+00, ..., 3.0009e-01,
-1.3569e+00, -8.3704e-01],
...,
[-2.6607e-01, -6.4422e-03, 9.8665e-01, ..., -2.5482e+00,
-3.5858e+00, -4.3871e-01],
[-1.4280e+00, -5.2641e-02, 8.0025e-01, ..., -2.6462e+00,
-4.0765e+00, -9.3699e-01],
[-1.1001e+00, -2.4322e+00, -1.3273e+00, ..., -3.8559e+00,
-4.0417e+00, -1.7109e+00]],
[[-1.5958e+00, -2.3843e+00, -3.2768e+00, ..., -5.3361e+00,
-8.4433e+00, -1.5477e+00],
[-4.9457e+00, -5.1328e+00, -3.9315e+00, ..., -5.0881e+00,
-7.9406e+00, -2.9912e+00],
[-3.9872e+00, -4.5113e+00, -5.5103e+00, ..., -5.6136e+00,
-7.8848e+00, -3.9610e+00],
...,
[-4.5438e+00, -5.3067e+00, -5.9179e+00, ..., -6.5834e+00,
-9.9912e+00, -5.1261e+00],
[-7.5501e+00, -7.9767e+00, -7.8230e+00, ..., -9.2157e+00,
-1.0007e+01, -3.6546e+00],
[-1.3725e+00, -3.0963e+00, -2.3988e+00, ..., -3.9259e+00,
-4.2678e+00, -6.8654e-01]],
[[-1.9758e-01, 2.2926e+00, 1.4412e+00, ..., 2.0753e+00,
2.5383e+00, 7.4543e-01],
[-4.7669e-01, 1.3615e-01, 1.9146e-01, ..., 9.8111e-01,
2.1682e+00, -7.9777e-02],
[-2.9539e-02, 1.7853e+00, 1.2700e+00, ..., 2.0696e-01,
1.3937e+00, -3.8725e-01],
...,
[-7.2122e-01, 5.8942e-01, 1.5379e+00, ..., 2.3920e-01,
1.2122e-01, -1.0391e-01],
[-1.2125e+00, 1.2112e-01, 1.0770e+00, ..., -1.7018e+00,
5.5682e-01, -5.1208e-01],
[-4.1514e-01, -1.4113e-01, 2.8784e-01, ..., 3.2043e-01,
-8.0295e-02, -2.2417e+00]],
...,
[[-1.3063e+00, -2.0632e+00, -5.5587e-01, ..., -3.5061e+00,
-5.4615e+00, -1.1396e+00],
[-3.5374e+00, -3.7492e+00, -1.7715e+00, ..., -2.3265e+00,
-5.4043e+00, -2.2749e+00],
[-2.7633e+00, -2.8090e+00, -1.3825e+00, ..., -3.0648e+00,
-6.9084e+00, -1.7233e+00],
...,
[-8.8769e-01, -2.1644e+00, -6.2127e-01, ..., -2.6544e+00,
-6.5917e+00, -4.0770e+00],
[-2.0298e+00, -3.8659e+00, -7.9647e-01, ..., -1.5476e+00,
-4.0720e+00, 2.8462e-01],
[-1.1858e+00, -1.4641e+00, 1.8170e-01, ..., -1.6517e+00,
-2.4694e+00, 7.9437e-01]],
[[-1.7520e+00, -1.6328e+00, -1.8588e+00, ..., -2.1489e+00,
-8.8769e-01, 2.6656e-01],
[-4.0039e+00, -1.9248e+00, -1.7849e+00, ..., -1.9788e+00,
-2.7205e+00, -6.7319e-02],
[-3.8126e+00, -2.6834e+00, -2.8542e+00, ..., -6.5465e-01,
-2.3626e+00, -2.2355e-01],
...,
[-2.9096e+00, -3.0583e+00, -3.3854e+00, ..., -2.6553e+00,
-1.2488e+00, 1.0963e+00],
[-2.0809e+00, -3.1678e-01, -1.3697e+00, ..., -5.6131e-01,
-1.6909e+00, 1.5126e+00],
[-2.1569e+00, -1.5041e+00, -4.6299e-01, ..., -4.7945e-01,
-9.0427e-01, 6.4641e-01]],
[[ 1.2151e+00, 3.5781e-01, 8.7549e-01, ..., 1.4657e+00,
1.3650e+00, 2.6230e-02],
[ 4.0560e+00, 3.2009e+00, 2.8304e+00, ..., 3.3394e+00,
2.9612e+00, 3.7980e+00],
[ 1.6564e+00, 8.2976e-01, 1.0486e+00, ..., 3.1669e+00,
2.1187e+00, 1.9486e+00],
...,
[ 3.0660e+00, 2.7791e+00, 2.5282e+00, ..., 2.0516e+00,
4.9741e+00, 2.3922e+00],
[ 3.7282e+00, 4.1018e+00, 2.9318e+00, ..., 3.0178e+00,
7.4614e-01, 2.5238e+00],
[ 2.3906e+00, 2.1646e+00, 1.5472e+00, ..., 4.5677e-01,
-1.1630e-01, 1.2788e+00]]],
[[[ 3.2042e-01, 1.0984e+00, 1.0801e+00, ..., -1.4455e+00,
-1.1338e+00, -9.3650e-01],
[ 1.3636e+00, 5.1965e-01, -3.5868e+00, ..., -5.6861e-01,
1.5350e-01, -1.2512e+00],
[-6.8695e-02, -6.0495e+00, -7.4408e+00, ..., -1.3622e+00,
-5.3241e-01, -2.0334e+00],
...,
[-2.9011e-01, -1.5993e+00, -1.4467e+01, ..., 3.9423e+00,
-2.3287e+00, -1.3342e+00],
[-2.1558e+00, -6.0675e-01, -5.8753e+00, ..., -4.7656e+00,
-5.7636e+00, -9.2256e-01],
[-2.6447e+00, -3.0534e+00, -2.1639e+00, ..., -3.7327e+00,
-4.5319e+00, -2.1965e+00]],
[[-4.9789e-01, -1.9887e+00, -3.1155e+00, ..., -3.1639e+00,
-4.8326e+00, -2.6806e+00],
[-3.2782e+00, -5.0174e+00, -8.4098e+00, ..., -3.0096e+00,
-5.0031e+00, -4.7807e+00],
[-5.9360e+00, -1.1013e+01, -8.0200e+00, ..., -3.6925e+00,
-3.7735e+00, -3.4792e+00],
...,
[-7.4659e+00, -5.0416e+00, -1.2724e+01, ..., -5.1783e+00,
-1.0593e+01, -3.5516e+00],
[-9.7636e+00, -8.7619e+00, -7.0165e+00, ..., -1.4025e+01,
-1.2366e+01, -3.4192e+00],
[-2.3185e+00, -4.0310e+00, -2.4430e+00, ..., -4.8155e+00,
-4.6037e+00, -5.2349e-01]],
[[ 7.7329e-01, 1.5851e+00, 9.8768e-01, ..., 1.5906e+00,
1.9923e+00, 8.6867e-01],
[-3.0293e-02, -2.4097e-01, 1.9985e+00, ..., 1.2878e+00,
2.4520e+00, 1.0182e+00],
[-2.1123e-01, 3.0991e+00, 5.0948e+00, ..., 9.7329e-01,
2.5483e+00, 2.1683e+00],
...,
[-5.7786e-01, -7.0809e-01, 5.2195e+00, ..., 7.6791e-01,
3.2722e+00, 9.2476e-01],
[-6.1806e-01, -1.4329e+00, -3.1070e-01, ..., 1.8801e-01,
1.0508e+00, -8.4469e-01],
[ 4.2664e-01, 7.0065e-01, 2.4580e-02, ..., 8.6608e-01,
-9.6120e-01, -2.3595e+00]],
...,
[[-4.0959e-01, -1.4264e+00, -2.2871e+00, ..., -2.6216e+00,
-3.4914e+00, -1.2973e+00],
[-2.4993e+00, -4.1386e+00, -5.3259e+00, ..., -2.9532e+00,
-4.3972e+00, -3.5566e+00],
[-2.8168e+00, -6.6567e+00, -2.1220e+00, ..., -3.5124e+00,
-4.5821e+00, -2.9728e+00],
...,
[-1.7176e+00, -4.8744e+00, -5.9231e+00, ..., -4.5271e+00,
-8.2897e+00, -3.7680e+00],
[-3.3631e+00, -4.1193e+00, -2.4960e+00, ..., -6.6922e+00,
-7.0524e+00, -1.1774e+00],
[-2.8079e+00, -1.7645e+00, -2.1094e+00, ..., -2.0403e+00,
-3.6077e+00, -1.7124e-01]],
[[-1.6219e+00, -2.5913e+00, -3.0619e+00, ..., -3.0655e+00,
-2.8107e+00, -4.3927e-01],
[-4.1611e+00, -5.6554e+00, -8.9912e+00, ..., -4.1741e+00,
-5.8340e+00, -1.8767e+00],
[-6.2066e+00, -5.7905e+00, -4.2106e+00, ..., -2.2482e+00,
-4.5997e+00, -2.0074e+00],
...,
[-2.0977e+00, -7.2944e+00, -5.0403e+00, ..., -3.4284e+00,
-4.7851e+00, -9.6812e-01],
[-1.8411e+00, 1.7575e-01, -3.7564e+00, ..., -3.8091e+00,
-5.1951e-01, 1.6226e+00],
[-2.8335e+00, -1.0089e+00, -2.2165e+00, ..., 4.2910e-01,
3.1321e-01, 1.0002e+00]],
[[ 7.8328e-01, -4.6527e-02, 1.9785e+00, ..., 5.1487e-01,
1.4582e+00, 6.7714e-01],
[ 2.0035e+00, 1.0717e+00, -6.2192e-03, ..., 7.2170e-01,
2.2198e+00, 3.3722e+00],
[ 2.5410e+00, -5.1276e-01, -8.7036e-01, ..., 1.5431e-01,
9.1455e-01, 6.5606e-01],
...,
[ 5.1739e+00, 4.3147e+00, 1.8863e-01, ..., 3.9477e+00,
3.3973e+00, 1.1831e+00],
[ 4.5696e+00, 3.6714e+00, -1.4235e-01, ..., 4.6337e+00,
1.8687e+00, 2.0568e+00],
[ 2.8472e+00, 2.5986e+00, 1.0756e+00, ..., 8.9089e-02,
1.4371e+00, 1.9051e+00]]]], device='cuda:0'),
tensor([[[[-1.0850e+00, -3.7866e+00, -4.2338e+00, ..., -2.1649e+00,
-3.2987e+00, -1.5653e+00],
[ 8.9133e-01, 1.5314e-01, 1.5538e+00, ..., -6.1564e-01,
-1.1875e+00, -8.7969e-01],
[-6.0103e-03, 2.3784e-01, 7.4129e-01, ..., 4.8958e-01,
3.5424e-01, -9.7066e-01],
...,
[-9.6183e-01, -3.0578e+00, -1.1541e-01, ..., -1.1395e+00,
-9.8646e-01, -1.0135e+00],
[-8.0168e-01, -9.9112e-01, -4.7145e-01, ..., -7.6950e-01,
-3.0619e-01, -1.6126e+00],
[-2.7378e+00, -5.2597e+00, -4.6173e+00, ..., -4.4655e+00,
-4.1310e+00, -3.9909e+00]],
[[-2.3479e+00, -6.6171e+00, -7.1703e+00, ..., -5.5570e+00,
-8.3941e+00, -2.2002e+00],
[-3.7101e+00, -2.7716e+00, -1.3200e+00, ..., -6.8581e+00,
-8.6510e+00, -3.8188e+00],
[-3.7347e+00, -6.7405e+00, -7.6551e+00, ..., -9.4476e+00,
-9.1725e+00, -5.7160e+00],
...,
[-5.4086e+00, -5.9030e+00, -5.1870e+00, ..., -4.3879e+00,
-5.2871e+00, -3.7030e+00],
[-7.4894e+00, -8.5837e+00, -6.0038e+00, ..., -7.5368e+00,
-8.3752e+00, -6.9465e+00],
[-2.5580e+00, -4.6966e+00, -4.4458e+00, ..., -3.8277e+00,
-4.1206e+00, -2.6772e+00]],
[[ 1.3329e+00, 3.2049e+00, 1.8089e+00, ..., 1.8388e+00,
2.8788e+00, 9.7143e-01],
[-2.9820e-02, 7.8209e-01, -2.0286e-01, ..., 1.4675e+00,
2.7391e+00, 1.5775e-01],
[-3.2218e-01, -1.0055e+00, -1.8726e+00, ..., -6.8890e-01,
-2.5857e-01, -3.9235e-01],
...,
[-2.9410e-01, 9.2992e-01, -1.6000e-01, ..., 1.0247e+00,
1.6309e-01, -1.1905e-01],
[ 5.0809e-01, -1.7404e-01, 1.3976e-02, ..., 4.4918e-01,
-5.0632e-02, 1.2506e-01],
[ 1.4750e+00, 1.9396e+00, 1.8982e+00, ..., 1.5572e+00,
1.2281e+00, 1.9475e-01]],
...,
[[-2.3473e+00, -5.4903e+00, -5.2745e+00, ..., -4.6094e+00,
-5.7547e+00, -1.7548e+00],
[-1.1241e+00, -1.0261e+00, 5.6744e-01, ..., -3.7861e+00,
-5.1736e+00, -2.7504e+00],
[-2.3955e+00, -3.8355e+00, -4.2894e+00, ..., -3.8813e+00,
-5.0154e+00, -2.7307e+00],
...,
[-2.0957e+00, -3.0463e+00, -2.6841e+00, ..., -1.9479e+00,
-3.3725e+00, -1.6122e+00],
[-2.9300e+00, -2.7624e+00, -9.7081e-01, ..., -2.8078e+00,
-4.6608e+00, -2.4893e+00],
[-3.7532e+00, -4.2979e+00, -3.8489e+00, ..., -2.8868e+00,
-3.7421e+00, -1.7570e+00]],
[[-3.8091e+00, -4.2965e+00, -2.3971e+00, ..., -2.2885e+00,
-1.0182e+00, 5.8837e-01],
[-3.7552e+00, -2.0209e+00, -1.2574e+00, ..., -2.7060e+00,
-2.2623e+00, -1.6396e-01],
[-2.3271e+00, -5.6389e-01, -9.0891e-01, ..., -1.1969e+00,
-1.0207e+00, -1.1828e-01],
...,
[-3.6004e+00, -2.1311e+00, -2.3252e+00, ..., -2.0939e+00,
-2.8362e+00, -6.1396e-01],
[-3.6213e+00, -6.6769e-01, -2.2310e+00, ..., -5.9026e-01,
-2.6815e+00, -8.0755e-01],
[-4.5465e+00, -3.2522e+00, -2.9261e+00, ..., -2.3807e+00,
-3.5887e+00, -1.2873e+00]],
[[ 1.4721e+00, -3.4155e-01, 4.2908e-01, ..., 9.4083e-01,
1.8240e+00, -7.2558e-02],
[ 3.2672e+00, 1.5940e+00, 2.6584e+00, ..., 3.0743e+00,
3.2276e+00, 3.8039e+00],
[ 3.0878e+00, 2.6869e+00, 2.0504e+00, ..., 3.3155e+00,
2.7191e+00, 1.8551e+00],
...,
[ 2.1666e+00, 3.1477e-01, 3.5529e+00, ..., 1.0395e+00,
2.4481e+00, 2.1923e+00],
[ 4.1195e+00, 3.6510e+00, 2.1445e+00, ..., 2.5291e+00,
2.0385e+00, 2.1362e+00],
[ 2.1492e+00, 1.0960e+00, 6.4346e-01, ..., -1.9025e-01,
-2.3285e-01, -1.9683e-01]]],
[[[ 2.2599e-01, 1.2886e+00, -6.2111e-02, ..., -3.2943e+00,
-2.4300e+00, -9.9611e-01],
[ 4.4946e-01, 8.6079e-01, 1.1493e+00, ..., -2.1552e+00,
-2.9001e+00, -1.5220e-01],
[-5.5404e-01, -6.4885e-01, 8.3154e-01, ..., -1.5771e+00,
-2.9671e+00, -2.6042e-01],
...,
[-8.0688e-01, -2.0104e+00, -4.5654e+00, ..., -4.8825e+00,
-4.1561e+00, 4.4614e-01],
[-2.1295e+00, -9.0020e-01, -2.1660e+00, ..., -4.4135e+00,
-3.5513e+00, 5.7841e-01],
[-1.3586e+00, -3.0868e+00, -2.7292e+00, ..., -4.0000e+00,
-3.0324e+00, -1.5916e+00]],
[[-1.1237e+00, -1.6355e+00, -3.4891e+00, ..., -6.6654e+00,
-6.9857e+00, -1.2473e+00],
[-3.9000e+00, -4.2060e+00, -3.3933e+00, ..., -8.6548e+00,
-9.0327e+00, -2.5789e+00],
[-4.0939e+00, -4.3624e+00, -4.0081e+00, ..., -4.9997e+00,
-8.9157e+00, -2.3512e+00],
...,
[-6.5778e+00, -6.3521e+00, -6.1307e+00, ..., -1.0240e+01,
-8.5730e+00, -1.9751e+00],
[-7.8893e+00, -7.3375e+00, -6.4579e+00, ..., -9.4033e+00,
-1.0545e+01, -4.6348e+00],
[-1.8702e+00, -3.4643e+00, -3.3311e+00, ..., -3.4103e+00,
-4.0262e+00, -8.8561e-01]],
[[-2.2303e-02, 1.3447e+00, 1.7282e+00, ..., 1.5209e+00,
2.2134e+00, 9.0274e-01],
[-2.7879e-02, -6.2585e-02, 9.7920e-01, ..., 1.0876e+00,
2.9132e+00, -2.4852e-01],
[ 6.0812e-01, 1.1096e+00, 1.8013e+00, ..., 2.2925e+00,
1.7842e+00, -6.4719e-01],
...,
[ 2.9849e-01, 5.3587e-01, 1.2040e+00, ..., 4.6402e-01,
4.9380e-01, -9.0614e-01],
[ 4.1878e-01, 2.4451e-01, 6.5403e-01, ..., -2.0970e+00,
9.6819e-01, -1.3425e+00],
[ 6.7425e-01, 1.0792e+00, 2.8089e-01, ..., -2.9836e-02,
-2.1583e-01, -1.7631e+00]],
...,
[[-6.6891e-01, -1.6286e+00, -2.2003e+00, ..., -4.9099e+00,
-5.0770e+00, -4.6264e-01],
[-2.7277e+00, -2.6057e+00, -1.4025e+00, ..., -3.0002e+00,
-5.8511e+00, -1.7353e+00],
[-2.2904e+00, -1.2843e+00, -7.0537e-01, ..., -5.3647e+00,
-5.8659e+00, -1.6019e+00],
...,
[-2.0690e+00, -1.7899e+00, -3.5438e+00, ..., -4.0286e+00,
-5.4443e+00, -2.2399e+00],
[-3.2965e+00, -2.7576e+00, -3.9569e+00, ..., -3.0458e+00,
-5.7136e+00, -1.2430e+00],
[-1.7414e+00, -1.9661e+00, -1.6752e+00, ..., -2.6818e+00,
-2.5369e+00, 2.0093e-01]],
[[-1.3240e+00, -2.4387e+00, -3.9945e+00, ..., -1.8037e+00,
-5.2924e-01, -6.4177e-01],
[-3.5950e+00, -3.6229e+00, -4.6718e+00, ..., -3.1716e+00,
-2.2637e+00, -6.1144e-01],
[-4.5929e+00, -3.8330e+00, -4.1815e+00, ..., -1.5210e-01,
-1.3315e+00, -9.8133e-01],
...,
[-4.6154e+00, -3.7074e+00, -4.9978e+00, ..., -3.2402e+00,
-1.8983e+00, -7.2044e-01],
[-3.7955e+00, -9.3770e-01, -3.4347e+00, ..., -1.1099e+00,
-8.7280e-02, 1.3374e+00],
[-3.1400e+00, -2.4633e+00, -1.5934e+00, ..., 1.3056e-01,
-7.9817e-01, 6.2541e-01]],
[[ 1.2044e+00, -8.2062e-02, 1.1248e+00, ..., 5.4258e-01,
7.6572e-01, -1.3354e+00],
[ 2.3454e+00, 1.2107e+00, 1.2992e+00, ..., 3.5666e+00,
2.4881e+00, 2.1531e+00],
[ 1.4217e+00, 7.4207e-01, 3.2686e-01, ..., 3.3698e+00,
1.4952e+00, 4.1343e-01],
...,
[ 2.7213e+00, 1.6639e+00, 1.4436e+00, ..., 2.1470e+00,
2.5789e+00, 4.4334e-01],
[ 2.8543e+00, 3.8736e+00, 1.8210e+00, ..., 3.8479e+00,
1.3410e+00, 1.6991e+00],
[ 2.0416e+00, 2.3095e+00, 8.9526e-01, ..., 5.3248e-01,
-2.8019e-01, 9.0317e-01]]],
[[[-4.9923e-01, 1.5000e-02, -5.4295e-01, ..., -2.0472e+00,
-2.6915e+00, -2.4225e+00],
[ 8.4078e-01, 9.0407e-01, 1.3089e+00, ..., -4.3210e+00,
-8.3824e-01, -2.9924e+00],
[ 5.8902e-01, 3.1178e-01, -1.4267e+00, ..., 2.4554e+00,
9.8886e-01, -3.7306e+00],
...,
[-5.8701e+00, -1.3206e+00, -4.1914e+00, ..., -1.3133e+00,
-4.1314e+00, -3.2313e+00],
[-5.3369e+00, -6.6059e-01, -1.3537e-01, ..., -5.0459e+00,
-4.8337e+00, -4.3057e-01],
[-1.9082e+00, -2.9399e+00, -3.4894e+00, ..., -6.1361e+00,
-5.6072e+00, -2.4821e+00]],
[[-8.8086e-01, -2.1084e+00, -3.5200e+00, ..., -4.2109e+00,
-7.5374e+00, -4.3581e+00],
[-2.4462e+00, -4.7792e+00, -7.0965e+00, ..., -3.7697e+00,
-4.4927e+00, -7.5020e+00],
[-2.2649e+00, -3.5563e+00, -1.2018e+01, ..., 1.8212e+00,
-4.6433e+00, -7.8229e+00],
...,
[-8.9612e+00, -1.1856e+00, -1.1503e+00, ..., -1.1018e+01,
-1.2992e+01, -5.6550e+00],
[-1.1831e+01, -8.2970e+00, 2.1643e-01, ..., -1.1942e+01,
-1.0766e+01, -3.2280e+00],
[-1.8373e+00, -6.9467e+00, -8.0652e+00, ..., -4.6141e+00,
-4.9139e+00, -1.1867e+00]],
[[ 7.9870e-01, 1.6019e+00, 1.6128e+00, ..., 2.0160e+00,
1.9270e+00, 7.8197e-01],
[ 5.1539e-01, 2.2298e+00, 1.9841e+00, ..., 5.2328e+00,
4.0283e+00, 1.4866e+00],
[ 3.1438e-01, 2.6264e+00, 1.2192e+00, ..., 2.3246e+00,
6.2592e+00, 2.7447e+00],
...,
[-3.1032e+00, -1.2513e+01, -1.7046e+01, ..., -1.1968e+00,
-2.5705e+00, -3.4208e-01],
[-3.9741e+00, -1.2296e+01, -1.1757e+01, ..., -1.6662e+00,
4.9887e-01, -1.8526e+00],
[-6.2302e-01, -2.1182e+00, -2.3017e+00, ..., 4.1099e-01,
-1.9035e-01, -2.8120e+00]],
...,
[[-1.0610e+00, -1.6976e+00, -2.2084e+00, ..., -3.0214e+00,
-3.6750e+00, -2.5684e+00],
[-1.9993e+00, -2.6188e+00, -3.1526e+00, ..., -5.4535e+00,
-6.0309e+00, -3.8413e+00],
[-7.8099e-01, -2.1571e+00, -4.6473e+00, ..., -3.5294e-01,
-7.8817e+00, -2.6352e+00],
...,
[-5.2578e+00, -5.6008e+00, -6.1021e+00, ..., -6.8568e+00,
-7.7229e+00, -3.3774e+00],
[-5.5649e+00, -5.6084e+00, -4.2952e+00, ..., -6.5296e+00,
-6.1102e+00, -1.5689e+00],
[-3.7434e+00, -7.5253e+00, -9.5257e+00, ..., -2.8306e+00,
-3.3358e+00, -7.9190e-01]],
[[-1.8602e+00, -2.3857e+00, -3.8331e+00, ..., -2.6132e+00,
-2.1657e+00, -9.1700e-02],
[-4.2183e+00, -4.7922e+00, -5.6583e+00, ..., -4.3566e+00,
-4.0102e+00, -2.5444e+00],
[-3.5960e+00, -4.3392e+00, -7.9981e+00, ..., -3.4264e+00,
-5.5816e+00, -1.2299e+00],
...,
[-2.8143e+00, -5.6475e+00, -7.8646e+00, ..., -8.1527e+00,
-5.3422e+00, -1.6720e+00],
[-9.8550e-01, 7.2441e+00, 1.5375e+00, ..., 1.8961e+00,
2.8731e-01, 1.0801e+00],
[-1.7180e+00, 2.6157e+00, -2.0740e+00, ..., 1.8693e+00,
-2.7406e-02, 1.0338e+00]],
[[ 7.8349e-01, -2.2311e-01, 6.6525e-01, ..., 7.0081e-01,
1.4522e+00, 3.0704e-01],
[ 1.9843e+00, 7.4629e-01, 2.5891e+00, ..., 2.1813e+00,
3.4556e+00, 2.9648e+00],
[ 1.3843e+00, 6.6634e-01, 4.2035e+00, ..., 1.7499e+00,
6.3694e+00, 5.1427e-02],
...,
[ 4.0608e+00, -6.5970e-01, -5.5949e+00, ..., 3.7780e+00,
3.8365e+00, 1.7288e+00],
[ 4.8089e+00, 4.0474e+00, -3.6766e+00, ..., 1.7523e+00,
1.3465e+00, 2.9790e+00],
[ 2.1369e+00, 6.6385e+00, 4.1418e+00, ..., -6.2900e-01,
-4.9906e-01, 9.1693e-01]]],
[[[-4.9521e-01, 1.1905e+00, 2.4388e+00, ..., -1.1002e+00,
-1.1104e+00, -8.1675e-01],
[ 1.0758e+00, 5.1412e-01, 2.7072e+00, ..., -1.7818e+00,
-1.2991e+00, -1.8312e+00],
[-7.5727e-01, -2.3615e+00, -4.1850e+00, ..., 1.3066e+00,
-6.5962e-01, -2.9500e+00],
...,
[ 3.0195e-01, -3.8976e+00, 1.0338e+00, ..., -7.8179e-01,
3.3479e-01, -5.3968e-01],
[-8.0041e-01, -7.6954e-01, -3.1685e-01, ..., 3.0716e-01,
6.7192e-02, -1.3126e+00],
[-1.0193e+00, -2.2880e+00, -2.2607e+00, ..., -3.0383e+00,
-3.0424e+00, -2.8313e+00]],
[[-1.3592e+00, -2.2451e+00, -4.1097e+00, ..., -4.8856e+00,
-7.4415e+00, -2.8287e+00],
[-5.3687e+00, -6.2602e+00, -4.6157e+00, ..., -4.6089e+00,
-8.4535e+00, -5.1253e+00],
[-6.0267e+00, -9.1605e+00, -1.0513e+01, ..., 9.6705e-02,
-4.8081e+00, -4.9728e+00],
...,
[-3.0156e+00, -5.4833e+00, -5.2467e+00, ..., -2.4595e+00,
-3.7148e+00, -2.9498e+00],
[-7.3058e+00, -7.8160e+00, -8.3095e+00, ..., -5.2501e+00,
-5.5034e+00, -5.1040e+00],
[-1.3939e+00, -3.2991e+00, -1.4898e+00, ..., -3.1311e+00,
-2.9544e+00, -1.6307e+00]],
[[-4.1811e-01, 2.0700e+00, 1.7552e+00, ..., 1.2464e+00,
9.9193e-01, 8.0630e-01],
[-4.0013e-01, -2.3460e-03, -8.0595e-01, ..., 6.6785e+00,
2.2661e+00, 7.5858e-01],
[-2.7859e-01, 3.8786e-01, 3.8633e+00, ..., 5.8763e+00,
3.7668e+00, 1.2033e+00],
...,
[-1.7926e+00, 1.5491e+00, -4.7089e+00, ..., 1.4993e+00,
1.3999e+00, 7.6426e-01],
[-1.9510e+00, 1.7061e-01, 3.7072e-01, ..., 1.8101e+00,
1.2557e+00, 2.9969e-01],
[-6.4892e-01, -1.4522e+00, -3.2587e-01, ..., 1.7932e+00,
1.7431e+00, 4.3430e-01]],
...,
[[-1.3792e+00, -1.6685e+00, -7.1653e-01, ..., -3.1680e+00,
-3.3623e+00, -1.9497e+00],
[-3.4733e+00, -4.0302e+00, -4.0664e+00, ..., -5.9607e+00,
-4.5864e+00, -4.9647e+00],
[-3.3168e+00, -5.1239e+00, -7.1944e+00, ..., -4.0613e+00,
-5.3169e+00, -3.5611e+00],
...,
[ 1.7135e-01, -3.6051e+00, -1.1190e+00, ..., -2.3508e+00,
-3.0908e+00, -1.1105e+00],
[-1.0881e+00, -4.5348e+00, -3.5117e+00, ..., -2.1320e+00,
-3.3158e+00, -1.5357e+00],
[-1.2274e+00, -1.4866e+00, -3.3255e-01, ..., -2.0195e+00,
-2.2906e+00, -5.1464e-01]],
[[-1.8124e+00, -1.2557e+00, -2.0270e+00, ..., 2.3801e-02,
-1.9182e+00, -9.1880e-01],
[-4.2538e+00, -1.6412e+00, -2.0280e+00, ..., -6.6272e+00,
-3.0997e+00, -2.1650e+00],
[-4.4362e+00, -3.9669e+00, -1.0704e+01, ..., -6.6885e+00,
-4.0737e+00, -7.6924e-01],
...,
[-2.3063e+00, -3.4055e+00, 1.1822e+00, ..., 3.9368e-01,
-3.0633e+00, -1.9646e+00],
[-1.3173e+00, -1.6779e+00, -5.7131e-01, ..., 4.2814e-02,
-3.5065e+00, -1.9116e+00],
[-1.4276e+00, -9.2586e-01, 3.1499e-01, ..., -1.2359e+00,
-2.8421e+00, -1.2540e+00]],
[[ 1.0622e+00, 2.6812e-01, 1.1972e+00, ..., 1.0629e+00,
1.3335e+00, 2.2225e-01],
[ 4.5857e+00, 3.3560e+00, 3.6249e+00, ..., 5.0511e+00,
2.8266e+00, 2.1556e+00],
[ 1.9563e+00, -1.7092e-01, 1.5440e+00, ..., 2.4791e+00,
2.8746e+00, 1.0468e+00],
...,
[ 2.5282e+00, 4.9473e-02, 5.3769e+00, ..., 1.6997e+00,
8.1909e-02, 5.9507e-01],
[ 4.0532e+00, 2.9470e+00, 4.9255e+00, ..., 1.5840e+00,
6.7831e-01, 1.3209e+00],
[ 2.4506e+00, 1.7580e+00, 2.0729e-01, ..., -1.5434e-01,
-3.8742e-01, -1.4978e-01]]],
[[[-4.2116e-01, -5.0749e+00, -1.7678e+00, ..., -2.3909e+00,
-6.4706e-01, -2.7520e-01],
[ 5.3319e-01, -9.1264e+00, -1.7319e+00, ..., 1.2430e+00,
2.4169e+00, -1.6171e-01],
[-1.1701e+00, -9.1784e+00, 9.3475e-01, ..., 1.5845e+00,
1.1703e+00, -4.3591e-01],
...,
[-1.5224e-01, -8.4744e-02, 2.7436e-01, ..., -5.1476e+00,
2.3956e+00, 1.8950e+00],
[-1.5885e+00, 2.0087e-01, -3.2228e-02, ..., 4.7028e+00,
5.9328e+00, -5.5628e-01],
[-1.1808e+00, -2.1442e+00, -1.5263e+00, ..., 1.2959e-01,
-5.6790e+00, -4.3638e+00]],
[[-9.1547e-01, -7.0853e+00, 1.5426e+00, ..., -2.6061e+00,
-2.7210e+00, -9.5661e-01],
[-2.8832e+00, -9.5866e+00, 1.8276e+00, ..., 7.6593e-01,
-1.2153e+00, -2.3507e+00],
[-4.1346e+00, -1.0121e+01, 2.4559e+00, ..., 4.9729e-01,
-2.2159e+00, -2.4189e+00],
...,
[-4.5924e+00, -4.9547e+00, -5.2803e+00, ..., -4.6962e+00,
1.9609e+00, 1.3799e-01],
[-7.8397e+00, -8.0788e+00, -6.0280e+00, ..., 3.2753e+00,
-2.3978e-01, -5.9849e+00],
[-1.2641e+00, -2.9693e+00, -2.1832e+00, ..., -4.2952e+00,
-8.5828e+00, -3.5282e+00]],
[[-5.3643e-01, 3.6645e+00, -2.1176e-01, ..., 3.0623e+00,
2.6257e+00, 1.1577e+00],
[-1.5051e+00, 9.4843e-01, -6.0288e+00, ..., 3.5754e-01,
4.3769e-01, 7.6301e-01],
[-3.2360e-01, 1.4937e+00, -6.4274e+00, ..., -1.0805e+00,
-2.2229e-01, 6.0175e-01],
...,
[-1.2673e+00, -4.3624e-01, 8.7693e-01, ..., -3.0604e+00,
-2.3806e+00, -2.4075e-01],
[-1.3699e+00, -6.8216e-01, 1.8257e-01, ..., -2.4826e+00,
-4.5157e+00, 8.6629e-01],
[-3.5774e-01, -4.3550e-01, -1.8578e-01, ..., -3.3642e+00,
-1.4538e+00, -1.3367e-01]],
...,
[[-8.5717e-01, -2.1272e+00, 1.7252e+00, ..., -2.1905e+00,
-1.3719e+00, -5.2172e-01],
[-3.3564e+00, -4.5748e+00, 1.9227e+00, ..., 3.3249e-01,
-9.6617e-01, -2.1010e+00],
[-3.9378e+00, -3.8302e+00, 3.9520e+00, ..., -2.0991e-01,
-2.5053e+00, -1.0067e+00],
...,
[-9.9909e-01, -2.0196e+00, -2.3339e+00, ..., -2.7356e+00,
-1.7549e+00, -2.9833e+00],
[-2.2196e+00, -3.5218e+00, -6.4450e-01, ..., -1.2925e+00,
-3.3242e+00, -3.8355e+00],
[-1.2825e+00, -1.3105e+00, -4.7452e-01, ..., -6.4047e+00,
-7.8740e+00, -3.1984e+00]],
[[-2.2807e+00, -1.8687e+00, -5.3344e+00, ..., -2.0165e+00,
-1.7156e+00, -6.4439e-02],
[-4.9165e+00, -2.1412e+00, -2.4655e+00, ..., -6.2722e-01,
-2.7217e+00, -1.1881e+00],
[-4.7880e+00, -3.4016e+00, -2.5034e+00, ..., -5.4834e-01,
-2.8369e+00, -1.5638e+00],
...,
[-2.8326e+00, -2.6970e+00, -3.4492e+00, ..., 4.5392e-01,
1.9134e+00, 5.3483e-02],
[-1.9543e+00, -1.5295e-01, -1.7522e+00, ..., 4.2100e+00,
-5.9732e-01, -2.4480e+00],
[-1.9718e+00, -1.1660e+00, -1.4962e+00, ..., -2.8967e+00,
-3.9292e+00, -1.9156e+00]],
[[ 8.6820e-01, 1.0047e+00, -9.1138e-01, ..., -6.3470e-01,
6.3852e-01, 2.4906e-01],
[ 3.7764e+00, 1.8248e+00, 1.5264e-01, ..., 7.7661e-01,
1.6409e+00, 2.1639e+00],
[ 2.6359e+00, 1.7872e-01, 1.6984e+00, ..., 3.6631e-02,
-7.2436e-01, 2.0556e-01],
...,
[ 2.9156e+00, 2.2802e+00, 2.6485e+00, ..., -2.7088e+00,
2.8726e+00, 2.9580e+00],
[ 3.7358e+00, 4.0259e+00, 2.0825e+00, ..., 1.6749e+00,
4.4100e+00, 2.0628e+00],
[ 2.3822e+00, 2.5000e+00, 1.9071e+00, ..., 2.9538e+00,
1.4413e+00, -3.6756e-01]]]], device='cuda:0'),
tensor([[[[-2.4360e-02, 1.5290e+00, 1.4870e+00, ..., -1.0344e+00,
-3.2523e+00, -9.8086e-01],
[ 2.7098e-01, 5.3734e-01, 2.7243e+00, ..., 3.1620e-01,
-2.1840e+00, -5.4717e-01],
[-9.3753e-01, 2.7420e-02, 2.2090e+00, ..., 4.1231e-01,
-1.9853e+00, -1.0950e+00],
...,
[-4.2765e-01, -4.2292e-01, 1.1337e+00, ..., -2.0945e+00,
-1.2116e+00, 8.1902e-01],
[-1.7532e+00, -4.7986e-01, 1.1234e+00, ..., -2.4195e+00,
-2.8145e+00, 2.7740e-01],
[-1.8351e+00, -3.1564e+00, -1.5452e+00, ..., -3.6466e+00,
-3.7485e+00, -1.9003e+00]],
[[-8.4843e-01, -1.6935e+00, -3.0049e+00, ..., -5.2048e+00,
-8.9514e+00, -2.0601e+00],
[-4.1766e+00, -4.3592e+00, -3.6076e+00, ..., -5.3194e+00,
-8.6996e+00, -3.4147e+00],
[-3.2491e+00, -3.8338e+00, -5.0123e+00, ..., -5.4836e+00,
-8.2404e+00, -3.8133e+00],
...,
[-5.6178e+00, -6.3983e+00, -6.4363e+00, ..., -7.5617e+00,
-8.8693e+00, -2.4715e+00],
[-8.6589e+00, -8.6232e+00, -7.6003e+00, ..., -7.8835e+00,
-9.0423e+00, -3.9630e+00],
[-2.0862e+00, -4.0072e+00, -2.7318e+00, ..., -4.0994e+00,
-4.2800e+00, -6.8048e-01]],
[[-2.0440e-01, 2.3725e+00, 1.4236e+00, ..., 2.0907e+00,
2.4873e+00, 7.0358e-01],
[-5.7749e-01, 2.3480e-01, 2.9369e-01, ..., 1.1049e+00,
2.1989e+00, -2.2221e-01],
[-1.9220e-01, 1.4903e+00, 1.5307e+00, ..., 3.4285e-01,
1.4215e+00, -3.3439e-01],
...,
[-4.0927e-01, 5.0629e-01, 1.3166e+00, ..., 5.6276e-01,
-1.3306e+00, -1.2586e+00],
[-1.1122e+00, -8.0802e-02, 9.4455e-01, ..., -9.0359e-02,
2.0965e-01, -4.7948e-01],
[ 1.7421e-01, 6.8090e-02, 3.7899e-01, ..., -4.6654e-01,
-4.3213e-01, -1.7321e+00]],
...,
[[-9.6361e-01, -1.6631e+00, -2.4147e-01, ..., -3.3759e+00,
-5.4952e+00, -1.2454e+00],
[-3.0920e+00, -3.4349e+00, -1.4161e+00, ..., -2.6945e+00,
-5.8162e+00, -2.5239e+00],
[-2.2086e+00, -2.1154e+00, -7.4968e-01, ..., -3.1936e+00,
-7.0346e+00, -2.1208e+00],
...,
[-1.5565e+00, -2.2430e+00, -6.9168e-01, ..., -3.5444e+00,
-6.7507e+00, -2.7627e+00],
[-2.7766e+00, -4.0959e+00, -7.5237e-01, ..., -1.1802e+00,
-4.1299e+00, -1.0664e+00],
[-2.4236e+00, -2.0590e+00, -1.5315e-01, ..., -1.2722e+00,
-2.6151e+00, 7.6019e-01]],
[[-1.5027e+00, -1.7900e+00, -1.8181e+00, ..., -1.9761e+00,
-1.0686e+00, 1.5631e-01],
[-3.5732e+00, -2.1764e+00, -1.9823e+00, ..., -2.1409e+00,
-2.9209e+00, -1.2857e-01],
[-3.1981e+00, -2.8101e+00, -3.0333e+00, ..., -1.2285e+00,
-3.3065e+00, -5.0707e-01],
...,
[-3.2734e+00, -2.9857e+00, -3.7414e+00, ..., -1.0858e+00,
-1.8830e+00, -4.6610e-01],
[-2.4645e+00, 2.3338e-01, -1.3706e+00, ..., 2.6846e-01,
-7.0649e-01, 9.6798e-01],
[-2.8752e+00, -1.6390e+00, -5.3997e-01, ..., -2.3926e-01,
-7.9747e-01, 6.2405e-01]],
[[ 8.5692e-01, 2.6328e-01, 6.7862e-01, ..., 1.6360e+00,
1.7664e+00, 1.5143e-01],
[ 3.7207e+00, 2.8654e+00, 2.3762e+00, ..., 3.3424e+00,
3.0111e+00, 3.8760e+00],
[ 1.4393e+00, 4.4107e-01, 7.0212e-01, ..., 3.3194e+00,
2.0070e+00, 2.1721e+00],
...,
[ 3.3892e+00, 2.9850e+00, 2.8717e+00, ..., 4.8391e+00,
2.6569e+00, 1.5236e+00],
[ 4.4216e+00, 4.5376e+00, 3.3555e+00, ..., 2.0454e+00,
9.7318e-01, 2.1587e+00],
[ 2.7258e+00, 2.5943e+00, 1.9190e+00, ..., 2.3904e-01,
-1.9978e-01, 1.4853e+00]]],
[[[-1.7122e-01, 1.0676e+00, 8.0964e-01, ..., -5.8467e+00,
-3.4025e+00, -1.0985e+00],
[ 1.3631e+00, 7.9362e-01, -4.7930e+00, ..., -7.6638e+00,
-6.7254e-01, 3.4159e-01],
[ 7.4092e-01, -5.1425e+00, -1.2607e+00, ..., -4.7510e+00,
-2.6785e+00, -3.1292e+00],
...,
[ 7.7473e-01, 1.0763e+00, -5.2404e+00, ..., 1.6482e+00,
-1.2143e+00, -2.3805e+00],
[-1.6499e+00, 1.8483e+00, -1.2664e+00, ..., -1.2453e-01,
-8.5981e-01, -2.7020e+00],
[-1.7101e+00, -2.0396e+00, -1.0299e+00, ..., -3.7758e+00,
-3.7673e+00, -3.5637e+00]],
[[-1.3993e+00, -2.7098e+00, -3.2202e+00, ..., -7.4973e+00,
-5.3470e+00, -1.1634e+00],
[-2.7304e+00, -1.0041e+01, -8.6941e+00, ..., -4.2879e+00,
2.7261e-01, -2.9953e-01],
[-3.5587e+00, -1.8607e+01, -2.0815e+00, ..., 2.3512e+00,
4.6905e-01, -3.4746e+00],
...,
[-4.4348e+00, -2.8925e+00, -6.5662e+00, ..., -5.5276e+00,
-8.2375e+00, -5.9557e+00],
[-7.4694e+00, -6.6568e+00, -4.3579e+00, ..., -6.7954e+00,
-6.5482e+00, -6.0726e+00],
[-1.3298e+00, -2.9716e+00, -1.3810e+00, ..., -3.9398e+00,
-3.5472e+00, -2.0058e+00]],
[[ 6.9488e-02, 1.5471e+00, 8.7671e-01, ..., 2.3463e+00,
1.6839e+00, 1.8941e+00],
[-9.8810e-01, 4.7079e-04, 3.1967e+00, ..., 2.1282e+00,
-1.1676e+00, -9.1850e-01],
[-1.7312e+00, 2.4994e+00, 1.3850e+00, ..., -1.4382e+00,
-1.2871e+00, -2.0931e-02],
...,
[-1.1742e+00, -2.7628e+00, 3.1721e+00, ..., 2.3263e+00,
9.7432e-01, 6.5590e-01],
[-1.6060e+00, -2.2483e+00, 1.3842e+00, ..., 1.4561e+00,
1.3425e+00, 7.0451e-01],
[-6.4954e-01, -9.1466e-01, -1.0491e+00, ..., 1.2178e+00,
1.3069e+00, 4.6142e-01]],
...,
[[-1.1320e+00, -1.1206e+00, -2.8448e+00, ..., -4.3774e+00,
-1.7978e+00, -5.1408e-01],
[-3.1305e+00, -4.2731e+00, -6.0750e+00, ..., 2.4651e+00,
1.0721e+00, 3.0535e-01],
[-2.8811e+00, -3.8686e+00, 1.5015e+00, ..., 3.2453e+00,
-2.6966e+00, -2.0687e+00],
...,
[-7.2859e-02, -2.0830e+00, -4.4881e+00, ..., -5.1591e+00,
-5.3243e+00, -2.9623e+00],
[-2.0257e+00, -2.7017e+00, -2.2963e+00, ..., -2.6168e+00,
-3.1363e+00, -2.0405e+00],
[-1.4082e+00, -1.5727e+00, 6.8205e-01, ..., -2.7002e+00,
-3.0141e+00, -9.5655e-01]],
[[-1.9694e+00, -1.5567e+00, -1.9303e+00, ..., -2.1743e+00,
-3.9224e+00, -1.8200e-01],
[-3.2536e+00, -5.1088e+00, -4.7144e+00, ..., -2.9421e+00,
-5.8733e+00, -8.7439e-01],
[-4.2028e+00, -2.5013e+00, -3.4960e+00, ..., 1.2937e+00,
-7.2736e+00, -4.4070e+00],
...,
[-1.5320e+00, -4.3980e+00, -2.9057e+00, ..., -4.7319e+00,
-5.1328e+00, -1.9721e+00],
[-4.0652e-01, 1.0683e+00, -2.8851e+00, ..., -3.2929e+00,
-4.5905e+00, -2.3708e+00],
[-1.2971e+00, -2.8513e-02, -6.2459e-01, ..., -2.1127e+00,
-2.8254e+00, -1.1289e+00]],
[[ 7.6941e-01, 2.3601e-01, 1.0766e+00, ..., -1.7426e+00,
1.8455e+00, 6.2931e-01],
[ 2.6037e+00, 4.4477e+00, 1.2244e+00, ..., -1.1547e+00,
1.2291e+00, 5.3158e-01],
[ 1.8837e+00, 4.6432e+00, -1.6361e+00, ..., 7.5554e-01,
-4.5191e+00, -3.5674e+00],
...,
[ 2.2344e+00, 3.9442e+00, 2.4532e+00, ..., 2.4559e+00,
1.5724e+00, 1.1236e+00],
[ 2.9905e+00, 2.5769e+00, 1.1613e+00, ..., 1.5251e+00,
9.2527e-01, 1.5081e+00],
[ 1.9275e+00, 2.4328e+00, 9.9643e-01, ..., 7.4651e-01,
5.0077e-01, 2.6246e-01]]],
[[[ 2.0315e-01, 1.0689e+00, 9.9152e-01, ..., -1.3738e+00,
-1.0472e+00, -9.2705e-01],
[ 8.3763e-01, -2.0317e-01, -3.6485e+00, ..., -4.5556e-01,
3.0642e-01, -1.2392e+00],
[ 4.5545e-01, -5.4223e+00, -3.4553e+00, ..., -4.3913e-01,
-1.9191e-01, -2.0142e+00],
...,
[-6.8525e-01, -8.5929e+00, -2.8279e+00, ..., -1.4906e+00,
-2.5230e+00, 8.7221e-01],
[-2.3699e+00, -6.8410e+00, -3.2918e+00, ..., -1.1303e+00,
-1.9300e+00, 9.0851e-01],
[-2.3323e+00, -6.8423e+00, -4.3954e+00, ..., -1.3534e+00,
-3.3593e+00, -1.3855e+00]],
[[-1.0372e+00, -1.4681e+00, -2.1239e+00, ..., -3.1235e+00,
-4.7581e+00, -2.6880e+00],
[-2.9091e+00, -6.9569e+00, -7.6107e+00, ..., -3.7724e+00,
-5.2091e+00, -4.9471e+00],
[-3.6287e+00, -9.0119e+00, -4.0963e+00, ..., -4.1108e+00,
-4.5204e+00, -3.9394e+00],
...,
[-6.6707e+00, -1.1941e+01, -3.7401e+00, ..., -5.6332e+00,
-6.2881e+00, -8.5827e-01],
[-8.6540e+00, -1.2424e+01, -6.6746e+00, ..., -8.2303e+00,
-8.9404e+00, -3.2962e+00],
[-1.7998e+00, -4.9473e+00, -5.1896e+00, ..., -1.5383e+00,
-4.5982e+00, -8.9038e-01]],
[[ 6.4526e-01, 6.2086e-01, 6.1027e-01, ..., 1.5635e+00,
1.9270e+00, 8.4431e-01],
[ 1.6960e-01, -2.3460e-01, 2.5068e+00, ..., 1.3526e+00,
2.3766e+00, 9.8120e-01],
[-7.0639e-01, 2.8165e+00, 3.9731e+00, ..., 1.2252e+00,
2.4862e+00, 2.1524e+00],
...,
[ 1.7056e-01, 2.9463e+00, -2.5423e+00, ..., -4.0061e-01,
-2.0366e-01, -1.6750e+00],
[-2.8015e-01, -8.4899e-01, -3.4456e+00, ..., 9.9894e-01,
-8.0220e-01, -1.3096e+00],
[ 8.6400e-01, 3.6226e-01, -2.0445e+00, ..., 8.8290e-01,
-4.3163e-01, -1.7795e+00]],
...,
[[-5.7197e-01, -7.5980e-01, -2.1855e+00, ..., -2.6532e+00,
-3.4799e+00, -1.3183e+00],
[-2.7935e+00, -4.5139e+00, -5.1377e+00, ..., -3.0930e+00,
-4.6459e+00, -3.6600e+00],
[-2.1498e+00, -3.9611e+00, -4.7522e-01, ..., -2.6979e+00,
-4.0508e+00, -2.9576e+00],
...,
[-2.3529e+00, -5.2056e+00, -6.5648e-01, ..., -9.5563e-01,
-3.9698e+00, -9.9030e-01],
[-2.2987e+00, -5.3790e+00, -6.0314e-01, ..., -1.9020e+00,
-3.8385e+00, -4.3111e-01],
[-2.5051e+00, -4.9281e+00, -2.4176e+00, ..., -1.1370e+00,
-1.9946e+00, 2.8227e-01]],
[[-1.7681e+00, -2.0872e+00, -3.0019e+00, ..., -2.9528e+00,
-2.8248e+00, -4.7021e-01],
[-4.7068e+00, -5.4600e+00, -6.6632e+00, ..., -4.2148e+00,
-5.5983e+00, -1.7669e+00],
[-5.1499e+00, -4.5967e+00, -4.2527e+00, ..., -1.8976e+00,
-4.0560e+00, -1.6456e+00],
...,
[-4.7352e+00, -1.3684e+00, -1.4753e+00, ..., 2.0491e+00,
-8.7552e-01, -1.1517e-02],
[-3.8680e+00, 6.9829e-02, 1.7713e+00, ..., 1.2658e+00,
5.6950e-01, 1.4803e+00],
[-4.0420e+00, -1.9308e+00, 2.5637e+00, ..., -6.4302e-01,
2.1337e-01, 6.1133e-01]],
[[ 8.5501e-01, -2.0320e-01, 1.1043e+00, ..., 4.8074e-01,
1.4397e+00, 6.8103e-01],
[ 1.5125e+00, 9.7907e-01, -1.3122e-01, ..., 8.9752e-01,
2.1556e+00, 3.3469e+00],
[ 1.8720e+00, -1.3929e-01, 6.7361e-01, ..., 6.2965e-01,
1.3951e+00, 1.0289e+00],
...,
[ 2.6070e+00, 1.6777e+00, 3.3658e-01, ..., 1.4936e+00,
6.4737e-01, 8.9945e-01],
[ 4.6404e+00, 3.5371e+00, 2.0168e+00, ..., 3.1255e+00,
1.7469e+00, 1.7090e+00],
[ 2.0948e+00, 1.7757e-01, 2.3423e+00, ..., 9.8962e-02,
-1.1713e-01, 1.0585e+00]]],
[[[ 7.9990e-01, 6.1358e-01, -1.6130e-01, ..., -1.3457e+00,
-5.5306e-01, -5.6911e-01],
[-1.4678e-01, -3.7437e+00, -2.5194e+00, ..., 2.9138e-01,
1.8595e+00, -1.4217e-01],
[-2.8930e+00, -2.5252e+00, 1.4101e+00, ..., 1.8443e+00,
1.9063e+00, -1.5403e+00],
...,
[-1.6501e+00, -9.7416e+00, -2.6005e+00, ..., -3.3813e-01,
-8.2698e-01, 5.8280e-01],
[-3.2137e+00, -8.8930e+00, -8.8987e-01, ..., 6.8085e-01,
2.0727e-01, 4.2448e-01],
[-2.8282e+00, -6.9451e+00, -1.7450e+00, ..., 1.1673e+00,
-7.0815e-01, -1.4220e+00]],
[[-6.4147e-01, -1.6501e+00, -1.7170e+00, ..., -2.6819e+00,
-4.0776e+00, -2.1652e+00],
[-5.7256e+00, -7.5892e+00, -6.2358e+00, ..., -2.1678e+00,
-3.4713e+00, -3.9899e+00],
[-4.5758e+00, -1.8590e+00, 1.1953e+00, ..., -1.1895e+00,
-3.2430e+00, -4.2417e+00],
...,
[-6.8495e+00, -9.9949e+00, 9.8095e-01, ..., -1.3805e+00,
-4.9384e+00, -1.2648e+00],
[-7.9439e+00, -1.0874e+01, 6.1157e-01, ..., -7.4202e-01,
-4.5672e+00, -3.3105e+00],
[-1.5289e+00, -5.4290e+00, 3.1692e-01, ..., 5.1660e-02,
-2.0333e+00, -8.6447e-01]],
[[ 4.4462e-02, 5.9657e-01, 1.2180e+00, ..., 2.0547e+00,
2.3656e+00, 1.1150e+00],
[-5.4692e-01, 1.6314e+00, 3.1562e+00, ..., 1.5545e+00,
2.5278e+00, 1.3153e+00],
[ 2.1135e+00, 2.3254e+00, 6.7462e-01, ..., 7.8491e-01,
1.6422e+00, 2.0481e+00],
...,
[ 6.4352e-01, 4.2494e+00, -4.0006e+00, ..., -3.8176e+00,
-1.3596e-01, -1.3929e+00],
[-1.0421e+00, 8.0626e-01, -5.0725e+00, ..., -4.1576e+00,
-2.5964e-01, -6.1779e-01],
[-1.0017e-01, -4.8619e-01, -7.4019e+00, ..., -3.5582e+00,
-7.1037e-01, -1.0468e+00]],
...,
[[-9.0413e-01, -2.2405e+00, -1.7101e+00, ..., -2.1246e+00,
-2.6855e+00, -1.0783e+00],
[-4.1334e+00, -5.4231e+00, -3.8235e+00, ..., -1.7450e+00,
-3.1486e+00, -3.5227e+00],
[-4.2402e+00, 3.6054e-01, 2.6002e+00, ..., -5.3724e-01,
-3.0030e+00, -3.1486e+00],
...,
[-2.4439e+00, -3.8487e+00, 1.2641e+00, ..., -7.7427e-02,
-3.7424e+00, -9.7027e-01],
[-3.9700e+00, -4.2672e+00, -2.7765e-01, ..., 9.5398e-01,
-3.6362e+00, -2.1410e-01],
[-3.9150e+00, -3.1003e+00, 1.0037e+00, ..., 1.2139e+00,
-2.5291e+00, 7.6665e-01]],
[[-1.6667e+00, -2.6331e+00, -4.1191e+00, ..., -3.2735e+00,
-2.8928e+00, -7.0269e-01],
[-5.0867e+00, -5.0605e+00, -6.7144e+00, ..., -4.0219e+00,
-5.3935e+00, -2.2203e+00],
[-3.4779e+00, -7.3099e-01, -5.4188e-01, ..., -1.4191e+00,
-4.2858e+00, -2.4048e+00],
...,
[-5.8262e+00, -2.7052e+00, -5.5877e+00, ..., -5.9807e-01,
-1.9045e+00, 1.0008e+00],
[-4.0554e+00, 5.7862e-01, -4.3151e+00, ..., -3.9262e-01,
-2.1861e+00, 1.7665e+00],
[-3.4315e+00, 5.5637e-01, 7.4294e-01, ..., 1.0231e+00,
-1.5440e+00, 6.5488e-01]],
[[ 1.0525e+00, -6.7063e-01, 9.2528e-02, ..., 1.7900e-01,
1.2113e+00, 7.2178e-01],
[ 2.9950e+00, -3.4468e-01, 1.2451e+00, ..., 5.6681e-01,
2.2287e+00, 3.5909e+00],
[ 1.6112e+00, 1.4178e-01, 1.3665e+00, ..., 3.7773e-01,
1.8094e+00, 1.1462e+00],
...,
[ 3.0343e+00, 1.1733e+00, 7.3401e-01, ..., 1.6390e-01,
2.1732e+00, 1.2435e+00],
[ 4.7268e+00, 3.2545e+00, 6.2275e-01, ..., 4.7184e-01,
1.2977e+00, 1.9558e+00],
[ 2.9306e+00, 2.3562e+00, 6.5612e-01, ..., 1.1917e+00,
7.5882e-01, 1.2791e+00]]],
[[[ 5.0811e-01, 1.3551e+00, 9.2521e-01, ..., 5.4875e-01,
-5.0098e-01, 6.7564e-01],
[ 2.3777e-01, -6.5552e-01, 1.1159e+00, ..., 1.6373e-01,
-9.7061e-01, -1.4758e-01],
[ 5.0982e-02, -6.4220e-01, 9.8706e-01, ..., 4.9468e-02,
-4.0594e-02, -6.5426e-01],
...,
[-7.8718e-01, -3.5235e+00, -2.7092e+00, ..., -3.1620e+00,
-3.8364e+00, -9.4359e-01],
[-2.3365e+00, -1.7938e+00, -2.6283e-01, ..., -2.7376e+00,
-2.8628e+00, 5.2725e-02],
[-2.2848e+00, -3.5751e+00, -2.6863e+00, ..., -4.2553e+00,
-3.4768e+00, -1.5585e+00]],
[[-1.4903e+00, -1.9649e+00, -2.3553e+00, ..., -1.5570e+00,
-5.2855e+00, -1.0049e+00],
[-4.9253e+00, -6.5248e+00, -4.7983e+00, ..., -5.2878e+00,
-6.9266e+00, -3.7516e+00],
[-4.4129e+00, -4.9676e+00, -3.2588e+00, ..., -5.0836e+00,
-5.2466e+00, -3.9987e+00],
...,
[-6.8386e+00, -8.6582e+00, -8.4382e+00, ..., -8.5258e+00,
-9.5548e+00, -2.3420e+00],
[-1.0026e+01, -9.8860e+00, -7.9399e+00, ..., -1.0705e+01,
-1.0832e+01, -3.8577e+00],
[-2.5473e+00, -4.8623e+00, -4.0822e+00, ..., -4.2785e+00,
-4.2098e+00, -6.8669e-01]],
[[ 7.5722e-03, 9.9282e-01, 4.9397e-01, ..., 1.4616e+00,
1.9556e+00, 2.8437e-01],
[ 2.4032e-01, 1.9659e-01, -2.9244e-01, ..., 6.5428e-01,
2.0992e+00, -7.2404e-01],
[ 8.8741e-01, 2.6119e+00, 1.5019e+00, ..., 1.4640e+00,
2.0898e+00, 7.9333e-01],
...,
[ 3.4409e-01, 6.6845e-01, -2.5801e+00, ..., 1.5627e+00,
8.9419e-01, -8.9038e-01],
[-6.6409e-01, 5.9804e-02, -1.3894e-01, ..., 2.9627e-01,
6.5530e-01, -5.1238e-01],
[ 5.5138e-01, 9.3022e-01, 1.0701e-01, ..., -2.7808e-01,
-6.0002e-01, -1.9151e+00]],
...,
[[-9.5267e-01, -1.0355e+00, -4.7453e-01, ..., -1.5999e+00,
-2.6239e+00, 1.9756e-01],
[-4.2565e+00, -4.9642e+00, -2.2919e+00, ..., -3.6248e+00,
-5.3115e+00, -2.9005e+00],
[-2.6319e+00, -1.6376e+00, -1.2496e-01, ..., -1.2082e+00,
-4.0607e+00, -1.7712e+00],
...,
[-2.5541e+00, -4.4856e+00, -2.9267e+00, ..., -4.8401e+00,
-6.1885e+00, -3.0828e+00],
[-3.5500e+00, -5.4254e+00, -3.7609e+00, ..., -3.9749e+00,
-5.2207e+00, -1.4187e+00],
[-3.1582e+00, -3.2476e+00, -2.5541e+00, ..., -2.8524e+00,
-2.6751e+00, 2.5690e-01]],
[[-1.1432e+00, 1.6359e-01, -1.2372e+00, ..., -8.3849e-01,
1.1341e-02, 1.9724e-01],
[-4.0982e+00, -1.4396e+00, -1.6272e+00, ..., -1.6275e+00,
-2.0229e+00, 2.6466e-01],
[-3.9674e+00, -2.0511e+00, -1.5985e+00, ..., -7.7047e-01,
-2.6337e+00, -7.2471e-01],
...,
[-4.3033e+00, -2.9738e+00, -1.2326e+00, ..., -3.4408e+00,
-2.6924e+00, -9.2285e-01],
[-3.5117e+00, -7.3973e-01, -1.2370e+00, ..., -1.8131e+00,
-1.0713e+00, 2.7051e-01],
[-3.6115e+00, -2.7292e+00, -2.5504e+00, ..., -1.7535e+00,
-1.1771e+00, 3.5137e-01]],
[[ 1.7184e+00, 7.0415e-01, 6.6432e-01, ..., 7.6284e-01,
1.4609e+00, 1.5232e-01],
[ 3.1519e+00, 2.0385e+00, 1.7646e+00, ..., 1.7909e+00,
2.6562e+00, 3.5279e+00],
[ 2.1933e+00, 1.1158e-01, 3.0969e-01, ..., 2.0940e+00,
2.3195e+00, 1.9563e+00],
...,
[ 2.8184e+00, 1.9118e+00, 3.0627e+00, ..., 1.7284e+00,
1.4122e+00, 2.9566e-01],
[ 4.6979e+00, 4.1885e+00, 2.5168e+00, ..., 2.8652e+00,
1.9618e+00, 1.6539e+00],
[ 2.8045e+00, 3.0846e+00, 1.6685e+00, ..., 5.1003e-01,
6.0903e-01, 9.8626e-01]]]], device='cuda:0'),
tensor([[[[-1.1677e-01, 6.6739e-01, -2.3056e-01, ..., -1.3217e+00,
-9.9425e-01, -7.2540e-01],
[ 7.7481e-01, 6.3494e-01, 7.7813e-01, ..., -2.2899e-01,
4.9455e-01, -8.4350e-01],
[ 1.5256e+00, 6.6722e-02, 7.2476e-01, ..., -8.9472e-02,
-7.8221e-02, -1.7122e+00],
...,
[-1.3219e+00, -5.8651e-01, -1.6150e-01, ..., -2.3759e+00,
-2.8301e+00, 1.1952e+00],
[-2.1616e+00, -5.0082e-01, -1.1068e-03, ..., -5.6015e-01,
-1.7841e+00, 1.0652e+00],
[-1.4563e+00, -3.3738e+00, -2.7730e+00, ..., -2.0811e+00,
-3.2378e+00, -1.3416e+00]],
[[-7.1363e-01, -1.4390e+00, -2.4844e+00, ..., -2.8584e+00,
-4.5050e+00, -2.2112e+00],
[-1.8452e+00, -3.4817e+00, -3.8527e+00, ..., -3.5722e+00,
-4.9943e+00, -4.6697e+00],
[-1.7726e+00, -3.4431e+00, -3.4464e+00, ..., -3.7027e+00,
-4.3436e+00, -4.0109e+00],
...,
[-5.9906e+00, -6.2940e+00, -5.2896e+00, ..., -4.4983e+00,
-6.3498e+00, -5.2225e-01],
[-7.6339e+00, -7.9985e+00, -6.4786e+00, ..., -6.2853e+00,
-9.1501e+00, -3.5664e+00],
[-1.6360e+00, -3.7847e+00, -3.4424e+00, ..., -2.1616e+00,
-4.0059e+00, -7.7114e-01]],
[[ 6.8735e-01, 1.2356e+00, 1.1360e+00, ..., 1.4778e+00,
1.8031e+00, 8.9215e-01],
[ 8.5869e-01, 1.4928e+00, 1.6486e+00, ..., 1.2130e+00,
2.1383e+00, 8.6511e-01],
[ 2.5443e-01, 3.0755e+00, 1.6292e+00, ..., 1.2227e+00,
2.2295e+00, 1.8769e+00],
...,
[-2.0645e-01, 4.6441e-01, -2.6768e-02, ..., 5.0024e-02,
-5.2765e-01, -2.1217e+00],
[-2.4397e-01, -4.4104e-01, -2.9483e-03, ..., -9.3430e-01,
-1.2339e+00, -1.5588e+00],
[ 2.4446e-01, -1.4583e-01, -3.1742e-02, ..., -4.7022e-01,
-2.8791e-01, -1.7802e+00]],
...,
[[-9.9612e-01, -1.3083e+00, -1.9759e+00, ..., -2.5195e+00,
-3.1936e+00, -1.1847e+00],
[-2.3092e+00, -3.0180e+00, -2.2451e+00, ..., -3.0619e+00,
-4.2433e+00, -3.4876e+00],
[-9.2577e-01, -1.5169e+00, -8.2878e-01, ..., -2.3110e+00,
-3.8229e+00, -2.8074e+00],
...,
[-1.1756e+00, -1.2009e+00, -6.0127e-01, ..., -9.5275e-01,
-2.6122e+00, -9.1644e-01],
[-1.8640e+00, -2.3729e+00, -6.2532e-01, ..., -1.1668e+00,
-3.0763e+00, -6.7116e-01],
[-1.5419e+00, -2.0995e+00, -1.2909e+00, ..., -1.3096e+00,
-1.8946e+00, 5.8787e-01]],
[[-1.5271e+00, -2.3512e+00, -3.5793e+00, ..., -2.7999e+00,
-2.6698e+00, -5.3739e-01],
[-4.3083e+00, -4.5333e+00, -5.1135e+00, ..., -4.0929e+00,
-5.5142e+00, -1.8177e+00],
[-3.9733e+00, -4.7101e+00, -4.1486e+00, ..., -2.2202e+00,
-4.1239e+00, -1.6981e+00],
...,
[-2.6356e+00, -2.0618e+00, -2.4766e+00, ..., -3.8316e-01,
-5.2312e-01, -4.9738e-02],
[-2.7080e+00, 6.4352e-01, -5.5807e-01, ..., 3.3358e-02,
4.2167e-01, 1.5188e+00],
[-3.0337e+00, -1.3355e+00, -1.1722e+00, ..., -1.6249e+00,
-5.4880e-01, 6.3553e-01]],
[[ 8.8152e-01, -1.5236e-01, 6.9268e-01, ..., 4.6435e-01,
1.3321e+00, 5.8527e-01],
[ 1.6666e+00, 7.6277e-01, 6.3678e-01, ..., 8.2694e-01,
1.9407e+00, 3.3695e+00],
[ 1.6198e+00, 6.3371e-02, 7.7914e-01, ..., 6.1847e-01,
1.2405e+00, 1.1836e+00],
...,
[ 2.5871e+00, 2.0780e+00, 6.2554e-01, ..., 1.1812e+00,
1.4475e+00, 9.4309e-01],
[ 3.7110e+00, 4.3242e+00, 3.0321e+00, ..., 2.6821e+00,
1.6529e+00, 1.9314e+00],
[ 2.1289e+00, 1.7501e+00, 9.6592e-01, ..., 1.0893e-01,
-4.9542e-01, 9.9648e-01]]],
[[[-3.0493e-01, 5.6248e-01, 4.1546e-01, ..., -1.2580e+00,
-9.8996e-01, -8.1525e-01],
[ 1.4630e+00, 1.0831e+00, 2.4021e+00, ..., -3.9931e-01,
2.6363e-01, -9.8055e-01],
[ 3.3530e-01, -2.9646e+00, -8.4832e+00, ..., -9.0824e-01,
-6.3267e-01, -2.0503e+00],
...,
[-6.2187e-01, -1.5974e-01, 1.0421e+00, ..., 4.7159e-01,
2.0315e-01, 4.2850e+00],
[-2.4412e+00, -1.8458e+00, 4.2021e-01, ..., -3.2489e+00,
-5.1059e+00, -1.3628e+00],
[-2.2184e+00, -3.7381e+00, -3.2163e+00, ..., -4.3556e+00,
-2.4244e+00, -1.5478e+00]],
[[-3.7644e-01, -1.5659e+00, -4.2700e+00, ..., -2.8954e+00,
-4.5692e+00, -2.3846e+00],
[-1.4035e+00, -3.1600e+00, -4.3423e+00, ..., -3.0667e+00,
-4.7836e+00, -4.4208e+00],
[-2.9900e+00, -9.6022e+00, -1.3754e+01, ..., -2.3943e+00,
-4.0603e+00, -3.5931e+00],
...,
[-6.6094e+00, -8.3156e+00, -8.0587e+00, ..., 2.2973e+00,
-6.6582e-01, -5.4846e-01],
[-9.6957e+00, -9.9899e+00, -8.1858e+00, ..., -1.8316e+01,
-1.4549e+01, -5.8022e+00],
[-2.3914e+00, -4.7875e+00, -4.3013e+00, ..., -2.5847e+00,
-2.7153e+00, -5.6089e-01]],
[[ 5.0471e-01, 1.2929e+00, 1.1234e+00, ..., 1.4863e+00,
1.8950e+00, 8.8702e-01],
[ 6.6618e-02, 1.7561e+00, 2.8940e+00, ..., 1.1405e+00,
2.1871e+00, 9.5651e-01],
[-6.3672e-01, 4.1141e+00, 8.8803e+00, ..., 1.1584e+00,
2.3127e+00, 1.9619e+00],
...,
[-1.4271e-01, -2.7182e-01, -9.7555e-01, ..., -4.4486e-01,
2.3512e+00, -1.6667e-01],
[-5.4586e-01, 7.5896e-01, 1.0509e-02, ..., -1.4321e+00,
-5.5284e-02, 9.0801e-02],
[ 5.2202e-01, 1.1154e+00, 1.0873e+00, ..., 2.0052e+00,
-4.2706e-01, -7.1810e-01]],
...,
[[-7.3570e-01, -7.3317e-01, -2.1711e+00, ..., -2.5838e+00,
-3.3216e+00, -1.1963e+00],
[-2.1053e+00, -2.4571e+00, -3.9811e+00, ..., -2.7603e+00,
-4.1558e+00, -3.4602e+00],
[-1.7840e+00, -3.6789e+00, -7.1861e+00, ..., -2.9414e+00,
-4.0265e+00, -3.0864e+00],
...,
[-2.9421e+00, -3.9439e+00, -1.7077e+00, ..., -1.1303e-01,
-1.2225e+00, -4.9061e-01],
[-3.6384e+00, -6.0249e+00, -3.1645e+00, ..., -7.1851e+00,
-5.7644e+00, -2.5527e+00],
[-3.0490e+00, -3.6722e+00, -3.3623e+00, ..., 6.2752e-02,
-1.1087e+00, -4.6189e-01]],
[[-1.1025e+00, -1.7493e+00, -2.7578e+00, ..., -2.8232e+00,
-2.8126e+00, -5.1551e-01],
[-2.4080e+00, -3.8889e+00, -4.2893e+00, ..., -3.9445e+00,
-5.7374e+00, -1.9763e+00],
[-1.5260e+00, -8.0887e+00, -1.5550e+01, ..., -3.4080e+00,
-4.9818e+00, -2.1085e+00],
...,
[-3.9009e+00, -2.3999e+00, -3.1278e+00, ..., -2.8346e+00,
-3.2220e+00, -3.4883e-01],
[-3.5003e+00, -5.7378e-01, -9.8870e-01, ..., -4.2580e+00,
-4.7302e+00, 9.1525e-01],
[-3.6097e+00, -3.0206e+00, -2.9560e+00, ..., 1.3360e+00,
2.5198e-01, 4.2893e-01]],
[[ 8.3472e-01, -3.9535e-02, 7.8353e-01, ..., 4.9852e-01,
1.3883e+00, 6.0599e-01],
[ 1.5568e+00, 3.2894e-01, 1.0682e+00, ..., 6.7518e-01,
1.8154e+00, 3.1783e+00],
[ 1.4744e+00, 7.8658e-01, -6.9994e-01, ..., 3.3703e-01,
1.0308e+00, 8.2756e-01],
...,
[ 4.3138e+00, 4.3481e+00, 4.5793e+00, ..., -5.3252e+00,
-6.0043e+00, 1.7656e+00],
[ 4.4162e+00, 4.2581e+00, 3.7100e+00, ..., 5.7911e+00,
2.5695e+00, 3.5480e+00],
[ 2.7474e+00, 2.8949e+00, 2.1089e+00, ..., -1.5351e+00,
-6.1688e-01, 1.4702e+00]]],
[[[ 2.1490e-01, 1.4175e+00, 5.4578e-01, ..., -1.1564e+00,
-3.7397e-01, 5.4542e-01],
[ 4.3690e-01, 2.2416e-01, 4.2374e-01, ..., -3.1703e+00,
-1.0740e+00, 1.0164e+00],
[-4.9113e-01, -1.4333e-01, 9.4277e-01, ..., 1.0705e+00,
9.1263e-01, 8.1943e-01],
...,
[-9.9715e-01, -2.0224e+00, 4.2160e-01, ..., -1.3530e+00,
1.4511e-01, -4.1550e-01],
[-1.2224e+00, -1.2191e+00, 1.6882e-01, ..., -9.2290e-01,
2.7957e-01, -1.3437e+00],
[-2.7348e+00, -5.2526e+00, -4.4314e+00, ..., -4.6278e+00,
-3.9705e+00, -3.7036e+00]],
[[-8.0415e-01, -2.2802e+00, -4.1147e+00, ..., -3.5590e+00,
-5.6771e+00, -4.7532e-01],
[-4.6552e+00, -6.1545e+00, -5.1150e+00, ..., -4.2142e+00,
-6.9247e+00, -1.6839e+00],
[-3.5961e+00, -4.2343e+00, -3.4705e+00, ..., 1.8313e-01,
-4.8382e+00, -2.1445e+00],
...,
[-5.8826e+00, -5.8491e+00, -3.9025e+00, ..., -5.1788e+00,
-4.8756e+00, -3.4187e+00],
[-7.2890e+00, -8.0216e+00, -6.1737e+00, ..., -7.1266e+00,
-7.6285e+00, -6.1649e+00],
[-2.5401e+00, -4.4849e+00, -4.1261e+00, ..., -3.8087e+00,
-4.0280e+00, -2.4906e+00]],
[[-2.6654e-01, 2.1317e+00, 1.1090e+00, ..., 9.7257e-01,
2.4379e+00, 9.8336e-01],
[-1.5982e-01, 1.0762e+00, 8.7503e-01, ..., 3.2583e+00,
2.4665e+00, -5.7147e-01],
[-3.8905e-02, 1.8969e+00, 1.5009e+00, ..., 2.1533e+00,
3.2933e+00, 4.8420e-01],
...,
[ 8.9306e-01, 9.7899e-01, 9.6311e-01, ..., 2.0816e-01,
9.4311e-01, 3.8176e-01],
[ 8.5048e-01, 5.4718e-01, 8.7146e-01, ..., 8.9077e-01,
2.4655e-01, 1.0344e-01],
[ 1.3911e+00, 1.6300e+00, 1.3499e+00, ..., 1.6025e+00,
9.9275e-01, 7.4524e-02]],
...,
[[-8.5158e-01, -1.8499e+00, -1.7846e+00, ..., -2.1636e+00,
-3.0442e+00, -8.6039e-01],
[-3.2033e+00, -4.3324e+00, -2.6170e+00, ..., -4.4626e+00,
-4.3383e+00, -3.4613e+00],
[-1.5539e+00, -2.1339e+00, -7.9924e-01, ..., -8.7040e-01,
-3.9029e+00, -2.0968e+00],
...,
[-2.8247e+00, -2.6892e+00, -7.4361e-01, ..., -2.1670e+00,
-3.5677e+00, -1.6192e+00],
[-3.1300e+00, -3.1749e+00, -1.6912e+00, ..., -2.3770e+00,
-3.7865e+00, -2.2092e+00],
[-3.5041e+00, -3.8990e+00, -3.5318e+00, ..., -2.9420e+00,
-3.5342e+00, -1.6042e+00]],
[[-1.7213e+00, -2.1493e+00, -2.8411e+00, ..., -1.0054e+00,
-1.0797e+00, -1.2744e+00],
[-4.3287e+00, -3.5752e+00, -2.8513e+00, ..., -2.5724e+00,
-3.4582e-01, -5.6210e-01],
[-3.4755e+00, -2.8905e+00, -2.5780e+00, ..., -9.9195e-01,
-1.3068e+00, -1.5459e-01],
...,
[-4.0848e+00, -2.1690e+00, -3.4607e+00, ..., -1.7662e+00,
-2.6433e+00, -1.7219e+00],
[-3.7720e+00, -4.3149e-01, -2.0512e+00, ..., -7.5887e-01,
-2.5225e+00, -1.2881e+00],
[-4.4797e+00, -2.9541e+00, -2.8505e+00, ..., -2.1400e+00,
-3.0584e+00, -1.1643e+00]],
[[ 1.0499e+00, 5.9693e-01, 6.7181e-01, ..., -3.7702e-01,
1.5164e-01, -4.5693e-01],
[ 3.4995e+00, 3.3435e+00, 2.1483e+00, ..., 2.4498e+00,
3.1119e+00, 2.1913e+00],
[ 1.4938e+00, 5.5712e-01, 1.5474e-01, ..., 1.6044e+00,
4.3194e+00, 1.9462e+00],
...,
[ 2.9750e+00, 1.1844e+00, 1.1484e+00, ..., 2.6426e+00,
2.2129e+00, 1.1964e+00],
[ 3.6860e+00, 3.5788e+00, 2.3726e+00, ..., 2.4285e+00,
2.0577e+00, 1.9442e+00],
[ 1.9329e+00, 8.8836e-01, 3.4702e-01, ..., -4.1288e-01,
-3.3392e-01, -2.4599e-01]]],
[[[ 7.8205e-01, 7.0036e-01, -4.0422e-01, ..., 5.6463e-01,
7.6134e-02, -3.7507e-01],
[ 1.1794e+00, 4.2856e-01, 1.3618e+00, ..., -1.0671e+00,
-8.8791e-01, -7.9360e-01],
[ 9.3689e-01, -1.0238e-01, 2.0496e+00, ..., 5.4884e-01,
2.8175e-02, -9.1463e-01],
...,
[ 4.7944e-01, -6.6901e-01, 7.8234e-02, ..., 2.4431e+00,
1.4407e+00, -8.8820e-02],
[-2.0837e+00, -1.6997e+00, -1.6986e-01, ..., -1.9685e-01,
-3.1557e+00, -3.6052e+00],
[-9.6560e-01, -3.7083e+00, -3.2009e+00, ..., -2.4064e+00,
-3.6307e+00, -3.3266e+00]],
[[-1.4896e+00, -3.9812e+00, -6.8234e+00, ..., -3.0934e+00,
-4.4109e+00, -2.5707e+00],
[-4.8423e+00, -6.3557e+00, -6.6094e+00, ..., -2.8002e+00,
-3.7769e+00, -3.3392e+00],
[-4.6926e+00, -6.4797e+00, -6.5014e+00, ..., 2.2007e-01,
-2.0002e+00, -2.9918e+00],
...,
[-4.4067e+00, -7.5484e+00, -7.9615e+00, ..., -1.1597e+00,
-3.3019e+00, -4.0938e+00],
[-7.7134e+00, -9.0976e+00, -9.5577e+00, ..., -8.8504e+00,
-9.9913e+00, -6.1522e+00],
[-1.4474e+00, -4.2715e+00, -4.4628e+00, ..., -2.4389e+00,
-2.6494e+00, -1.6699e+00]],
[[ 6.4525e-01, 2.8348e+00, 1.8613e+00, ..., 2.0176e+00,
1.5548e+00, 7.9989e-01],
[-1.6636e-02, 1.4443e+00, 1.3078e+00, ..., 3.1714e+00,
1.9792e+00, 7.4930e-01],
[ 1.5420e-01, 1.9526e+00, 5.9352e-01, ..., 2.2603e+00,
2.8188e+00, 1.9370e+00],
...,
[-5.8765e-01, -2.0932e-01, -1.1594e-01, ..., -2.8717e+00,
-1.9070e+00, -7.4674e-01],
[ 1.0521e-01, 1.0876e+00, 4.5647e-01, ..., 1.5699e+00,
1.3744e+00, 8.0353e-01],
[ 6.0427e-01, 1.5186e+00, 3.9097e-01, ..., 1.5772e+00,
1.8836e+00, 4.3456e-01]],
...,
[[-8.5290e-01, -3.2221e+00, -2.7238e+00, ..., -1.4878e+00,
-2.4590e+00, -5.4249e-01],
[-2.7013e+00, -4.4322e+00, -1.7460e+00, ..., -2.6551e+00,
-3.8257e+00, -3.1671e+00],
[-1.8253e+00, -3.7143e+00, -2.5578e+00, ..., 4.6140e-01,
-2.6625e+00, -1.8248e+00],
...,
[-2.1736e+00, -4.4174e+00, -1.3307e+00, ..., -1.7141e+00,
-3.6651e+00, -3.1092e+00],
[-2.9779e+00, -3.5815e+00, -1.9363e+00, ..., -4.3826e+00,
-5.5233e+00, -2.1791e+00],
[-2.0064e+00, -3.3531e+00, -1.3733e+00, ..., 3.4737e-01,
-1.3840e+00, -5.6963e-01]],
[[-2.3828e+00, -3.0954e+00, -1.8685e+00, ..., 5.7973e-01,
-1.5042e+00, -8.1445e-01],
[-4.2600e+00, -3.0006e+00, -1.6694e+00, ..., -1.7041e+00,
-3.7952e+00, -1.4828e+00],
[-3.9818e+00, -2.6213e+00, -1.1800e+00, ..., 7.9750e-02,
-4.0481e+00, -1.7592e+00],
...,
[-2.5638e+00, -1.6812e+00, -1.5409e+00, ..., 2.2674e+00,
-2.0538e+00, -1.7329e+00],
[-4.1642e+00, -2.6971e-02, -3.1143e-01, ..., -1.0367e+00,
-5.2417e+00, -2.3702e+00],
[-3.8692e+00, -2.5250e+00, -1.4418e-01, ..., -4.0204e-01,
-2.9938e+00, -1.3584e+00]],
[[ 1.6131e+00, 1.2701e+00, 1.7593e+00, ..., 1.0112e+00,
1.1510e+00, 5.4317e-01],
[ 3.1836e+00, 3.0443e+00, 3.7190e+00, ..., 1.1651e+00,
1.0108e+00, 2.1479e+00],
[ 2.2317e+00, 9.0430e-01, 3.0467e+00, ..., 9.7256e-01,
1.0169e+00, 8.2718e-01],
...,
[ 3.7492e+00, 3.3927e+00, 1.8142e+00, ..., 1.4357e+00,
1.9524e+00, 1.2621e+00],
[ 3.0866e+00, 3.2207e+00, 2.9178e+00, ..., 3.8889e+00,
1.2704e+00, 5.6046e-01],
[ 2.5384e+00, 1.7632e+00, 1.7402e+00, ..., -9.2009e-02,
-8.8443e-01, 4.9316e-01]]],
[[[ 2.7610e-01, 1.2979e+00, 2.2308e-01, ..., -3.1121e+00,
-2.1969e+00, -1.1080e+00],
[ 4.2423e-01, 9.7536e-01, 1.2180e+00, ..., -1.7367e+00,
-1.7142e+00, -3.2752e-01],
[-4.2394e-01, -4.4581e-01, -1.0774e-01, ..., -3.7746e+00,
-3.7127e+00, -7.5241e-01],
...,
[-5.5586e-01, -2.3651e+00, -1.1757e+00, ..., -2.4310e+00,
-3.1537e+00, -2.5947e-02],
[-2.0124e+00, -1.3498e+00, -1.3854e+00, ..., -3.6681e+00,
-3.7328e+00, 4.1445e-01],
[-1.2028e+00, -3.2263e+00, -3.1499e+00, ..., -3.7931e+00,
-3.0823e+00, -1.3340e+00]],
[[-9.9358e-01, -1.5266e+00, -3.3931e+00, ..., -6.8144e+00,
-7.3783e+00, -1.7679e+00],
[-3.7637e+00, -4.1238e+00, -3.4851e+00, ..., -6.6634e+00,
-6.9157e+00, -2.0926e+00],
[-3.8386e+00, -4.2807e+00, -3.9461e+00, ..., -8.7799e+00,
-7.1999e+00, -2.0298e+00],
...,
[-6.4028e+00, -5.9608e+00, -5.4540e+00, ..., -7.7524e+00,
-7.7960e+00, -1.5261e+00],
[-8.0227e+00, -7.9048e+00, -7.3726e+00, ..., -9.5283e+00,
-1.0513e+01, -4.7171e+00],
[-1.8489e+00, -3.3873e+00, -4.0926e+00, ..., -3.4972e+00,
-3.4718e+00, -8.0742e-01]],
[[-5.6790e-02, 1.2757e+00, 1.7060e+00, ..., 1.4533e+00,
2.6899e+00, 1.0120e+00],
[-3.1866e-02, -4.5283e-02, 6.5414e-01, ..., 6.3057e-01,
2.8267e+00, 4.1620e-01],
[ 6.2612e-01, 1.1860e+00, 1.3373e+00, ..., -1.2467e-01,
2.1352e+00, 1.0459e+00],
...,
[ 2.7451e-01, 9.9973e-02, 1.1653e+00, ..., 3.7103e-02,
6.0005e-01, -6.9165e-01],
[ 4.8150e-01, 2.3627e-01, 2.3402e-01, ..., -1.3038e+00,
-3.6246e-01, -2.0102e+00],
[ 5.4690e-01, 1.2440e+00, 1.4483e+00, ..., -6.7694e-01,
-5.2129e-01, -1.7168e+00]],
...,
[[-5.6616e-01, -1.5565e+00, -2.1120e+00, ..., -5.1572e+00,
-5.3634e+00, -8.4507e-01],
[-2.6573e+00, -2.6414e+00, -1.3666e+00, ..., -3.6726e+00,
-4.0224e+00, -1.7792e+00],
[-2.0254e+00, -1.3676e+00, -9.7407e-01, ..., -5.6840e+00,
-4.4763e+00, -1.2637e+00],
...,
[-1.9698e+00, -1.9366e+00, -1.8572e+00, ..., -2.6785e+00,
-3.3924e+00, -1.5865e+00],
[-3.4058e+00, -3.1142e+00, -2.2420e+00, ..., -3.5598e+00,
-5.2144e+00, -1.6661e+00],
[-1.6298e+00, -2.1692e+00, -2.3697e+00, ..., -1.6716e+00,
-1.8961e+00, 2.5797e-01]],
[[-1.2196e+00, -2.3546e+00, -3.9505e+00, ..., -1.6182e+00,
-9.2219e-01, -5.8652e-01],
[-3.4774e+00, -3.4175e+00, -4.7297e+00, ..., -2.3468e+00,
-1.2660e+00, 1.4666e-01],
[-4.4910e+00, -3.8446e+00, -4.3780e+00, ..., -1.6730e+00,
-1.9596e+00, -9.9876e-01],
...,
[-4.8477e+00, -4.3932e+00, -3.3308e+00, ..., -4.4645e+00,
-1.9339e+00, -1.2290e+00],
[-4.1205e+00, -2.0991e+00, -1.9609e+00, ..., -1.0293e+00,
-1.7568e-01, 1.2893e+00],
[-3.1996e+00, -3.2563e+00, -2.3058e+00, ..., -3.4129e-01,
-6.8813e-01, 6.9834e-01]],
[[ 1.1933e+00, -5.6125e-02, 1.1205e+00, ..., 5.4765e-01,
8.6354e-01, -1.2420e+00],
[ 2.2143e+00, 1.1460e+00, 1.3952e+00, ..., 3.5484e+00,
2.6885e+00, 2.8207e+00],
[ 1.3713e+00, 7.6931e-01, 4.7937e-01, ..., 1.6209e+00,
1.1205e+00, 1.2608e+00],
...,
[ 2.5717e+00, 1.5595e+00, 9.0852e-01, ..., 2.5801e+00,
2.2523e+00, 5.4945e-01],
[ 2.9010e+00, 4.2186e+00, 2.3458e+00, ..., 2.3165e+00,
1.2182e+00, 1.9299e+00],
[ 1.9927e+00, 1.8252e+00, 5.1792e-01, ..., 5.9401e-01,
-1.1338e-02, 7.7653e-01]]]], device='cuda:0'),
tensor([[[[-9.1010e-02, 6.9379e-01, -2.9307e-01, ..., -1.5731e+00,
-1.1318e+00, -7.2096e-01],
[ 9.6894e-01, 6.4057e-01, 2.7826e-01, ..., -3.5430e-01,
5.6760e-01, -7.4848e-01],
[ 1.7940e+00, -5.0365e-01, -1.6219e+00, ..., -7.6114e-01,
-4.0844e-01, -1.7460e+00],
...,
[ 4.4989e-02, -3.6912e-02, 2.4076e-01, ..., -2.2594e+00,
-2.6481e+00, 2.7737e-02],
[-2.3367e+00, -1.7237e+00, 1.1374e+00, ..., -2.8535e+00,
-2.7758e+00, 1.5157e-01],
[-2.3419e+00, -4.1657e+00, -3.0403e+00, ..., -3.9887e+00,
-3.3466e+00, -1.5510e+00]],
[[-7.6099e-01, -1.5523e+00, -2.5067e+00, ..., -3.0815e+00,
-4.4707e+00, -2.0672e+00],
[-2.0135e+00, -4.1225e+00, -4.7271e+00, ..., -2.7394e+00,
-4.4208e+00, -4.0944e+00],
[-3.1380e+00, -4.6424e+00, -3.5213e+00, ..., -2.5003e+00,
-4.0384e+00, -3.5140e+00],
...,
[-6.2931e+00, -8.3614e+00, -5.9429e+00, ..., -8.9752e+00,
-8.4686e+00, -1.8086e+00],
[-9.7553e+00, -1.0443e+01, -8.6869e+00, ..., -1.0409e+01,
-1.0507e+01, -3.8043e+00],
[-2.5057e+00, -4.8521e+00, -4.0711e+00, ..., -4.1354e+00,
-4.2858e+00, -6.6771e-01]],
[[ 6.7993e-01, 1.1884e+00, 1.0933e+00, ..., 1.5951e+00,
1.9743e+00, 9.5252e-01],
[ 1.0873e+00, 1.5135e+00, 1.4129e+00, ..., 1.0596e+00,
2.0349e+00, 1.0446e+00],
[ 1.6304e-01, 2.2841e+00, 2.3297e+00, ..., 9.9538e-01,
2.1335e+00, 1.9103e+00],
...,
[-5.4233e-01, -2.9615e-03, 3.5607e-01, ..., 8.8732e-01,
8.5190e-01, -2.7506e-01],
[-5.5846e-01, 6.2117e-01, -1.3335e+00, ..., -4.2034e-01,
1.0045e+00, -4.0756e-01],
[ 5.4857e-01, 1.3415e+00, 1.0454e+00, ..., -1.4259e-01,
-3.0263e-01, -1.8162e+00]],
...,
[[-9.9244e-01, -1.3590e+00, -2.0922e+00, ..., -2.6751e+00,
-3.1677e+00, -1.1951e+00],
[-2.2601e+00, -3.4399e+00, -2.6520e+00, ..., -2.3995e+00,
-3.7051e+00, -3.3212e+00],
[-1.3583e+00, -2.6999e+00, -1.4479e+00, ..., -2.1465e+00,
-3.9499e+00, -2.9485e+00],
...,
[-2.2365e+00, -3.2808e+00, -1.5064e+00, ..., -5.2937e+00,
-6.2464e+00, -2.6247e+00],
[-3.9063e+00, -5.8850e+00, -2.1303e+00, ..., -3.6864e+00,
-5.4609e+00, -1.4181e+00],
[-3.2017e+00, -3.8095e+00, -3.4450e+00, ..., -2.8729e+00,
-2.7185e+00, 4.1668e-01]],
[[-1.5764e+00, -2.3366e+00, -3.6278e+00, ..., -3.1345e+00,
-2.8216e+00, -4.8674e-01],
[-4.2684e+00, -4.6418e+00, -4.8949e+00, ..., -3.8591e+00,
-5.5786e+00, -2.0043e+00],
[-3.9153e+00, -4.1056e+00, -3.6383e+00, ..., -2.9046e+00,
-4.6666e+00, -1.9847e+00],
...,
[-3.9136e+00, -3.0486e+00, -4.0212e+00, ..., -3.7031e+00,
-2.6111e+00, -1.2040e+00],
[-3.3816e+00, -3.2505e-01, -1.1523e+00, ..., -2.3799e+00,
-1.2090e+00, 3.7587e-01],
[-3.7090e+00, -2.8438e+00, -2.4552e+00, ..., -1.1361e+00,
-1.0728e+00, 3.1683e-01]],
[[ 9.1711e-01, -1.4398e-01, 7.0450e-01, ..., 4.3232e-01,
1.2493e+00, 6.0956e-01],
[ 1.7509e+00, 1.0723e+00, 7.0314e-01, ..., 6.2086e-01,
1.8416e+00, 3.1917e+00],
[ 2.3125e+00, -1.8660e-01, -7.9511e-01, ..., 3.0833e-01,
9.1687e-01, 8.2310e-01],
...,
[ 4.2238e+00, 2.9392e+00, 2.4191e+00, ..., 3.2568e+00,
2.7320e+00, 5.3789e-01],
[ 4.6448e+00, 4.2168e+00, 3.9455e+00, ..., 3.3561e+00,
2.0759e+00, 1.4350e+00],
[ 2.7984e+00, 2.6558e+00, 2.3539e+00, ..., 3.1528e-01,
4.1383e-01, 1.0428e+00]]],
[[[ 2.4610e-01, 1.4188e+00, 5.9788e-01, ..., -7.8547e+00,
-3.9318e+00, -4.6626e-01],
[ 2.7603e-01, 3.1496e-01, 5.8687e-01, ..., -2.7786e-01,
-4.7788e-01, -8.0301e-01],
[-1.1842e-01, 2.9286e-01, 1.0042e+00, ..., 8.6902e-01,
3.1789e+00, -2.1323e-01],
...,
[ 3.2807e-01, -4.4511e+00, -2.2338e+00, ..., 2.9612e+00,
-5.4893e-01, -1.8098e+00],
[-1.5894e+00, -4.7951e+00, -9.9335e-01, ..., 2.2384e+00,
-2.1393e+00, -1.9146e+00],
[-1.3911e-01, -3.4403e+00, -1.2701e+00, ..., 2.6339e-01,
-9.8878e-01, -1.2076e+00]],
[[-8.0752e-01, -2.1872e+00, -4.1265e+00, ..., -7.9880e+00,
-6.5131e+00, -1.5249e+00],
[-4.7353e+00, -5.9630e+00, -5.1861e+00, ..., -9.2451e-01,
-5.7750e+00, -3.0302e+00],
[-3.5709e+00, -5.0635e+00, -5.9175e+00, ..., 7.3775e-01,
-1.7334e+00, -2.7971e+00],
...,
[-4.6514e+00, -7.8237e+00, -4.5586e+00, ..., 1.0134e+00,
-5.1314e+00, -2.7162e+00],
[-7.6413e+00, -9.1688e+00, -5.3825e+00, ..., -2.9622e+00,
-9.4824e+00, -4.1789e+00],
[-7.3823e-01, -2.7033e+00, -2.8077e+00, ..., -2.1456e+00,
-2.9997e+00, -3.2998e-01]],
[[-3.0847e-01, 2.1726e+00, 1.2314e+00, ..., 4.2761e+00,
3.5254e+00, 8.0992e-01],
[-1.1283e-01, 9.7487e-01, 8.5612e-01, ..., 4.8031e+00,
5.5603e+00, 2.8632e-01],
[ 3.9339e-02, 1.8228e+00, 1.3931e+00, ..., -9.5748e+00,
3.3050e+00, 2.3776e+00],
...,
[ 3.7374e-01, 1.3103e+00, 1.1944e+00, ..., 2.1264e+00,
2.4719e+00, 3.9919e-01],
[ 5.1382e-01, 1.1007e+00, -6.9568e-01, ..., 3.1525e+00,
2.7746e+00, 1.9235e-01],
[-5.8194e-01, 1.0833e-01, -1.6635e+00, ..., 2.1701e+00,
5.9621e-01, -8.9344e-01]],
...,
[[-8.2767e-01, -1.7848e+00, -1.8685e+00, ..., -6.4977e+00,
-4.5755e+00, -1.0327e+00],
[-3.0068e+00, -4.2770e+00, -2.7585e+00, ..., -1.8339e+00,
-4.8798e+00, -3.3786e+00],
[-1.4192e+00, -2.5543e+00, -1.2387e+00, ..., -1.1980e+00,
-4.5756e+00, -1.4790e+00],
...,
[-1.1359e+00, -3.3141e+00, -1.6185e+00, ..., -2.8683e+00,
-4.2180e+00, -1.7811e+00],
[-2.3900e+00, -3.7029e+00, -9.1118e-01, ..., -5.4930e+00,
-3.2654e+00, -1.5283e+00],
[-4.0736e-01, -2.2506e+00, -1.2511e+00, ..., -2.5893e+00,
1.6765e-01, 9.5765e-01]],
[[-1.6558e+00, -2.2056e+00, -2.8207e+00, ..., -2.6323e+00,
1.5009e-01, -1.3480e+00],
[-4.2627e+00, -3.2653e+00, -2.9495e+00, ..., -6.0339e+00,
-4.4940e+00, 8.8909e-02],
[-2.5495e+00, -1.4872e+00, -1.9362e+00, ..., 1.3930e+00,
-3.6788e+00, 1.5882e-01],
...,
[-4.4343e+00, -3.5852e+00, -5.9194e-01, ..., -6.2837e-01,
-2.9082e+00, -1.8064e+00],
[-4.7762e+00, -1.1989e+00, 1.6467e+00, ..., -2.7676e+00,
-2.6181e+00, -6.1259e-01],
[-2.9413e+00, -1.5257e+00, 9.0018e-01, ..., -1.4433e+00,
-5.9343e-01, 2.9322e-01]],
[[ 1.0903e+00, 5.5576e-01, 7.6175e-01, ..., -3.4668e-01,
8.0364e-01, -7.4958e-01],
[ 3.4802e+00, 3.3444e+00, 2.3234e+00, ..., 4.3800e+00,
7.0903e+00, 4.1279e+00],
[ 1.6500e+00, 1.4869e+00, 1.2937e+00, ..., -7.6272e-01,
2.8353e+00, 2.8842e+00],
...,
[ 2.9666e+00, 7.7466e-01, 7.4910e-01, ..., 5.1615e+00,
1.8434e+00, 3.8084e-02],
[ 3.2175e+00, 2.8830e+00, 2.8327e+00, ..., 4.3105e+00,
5.1028e-01, 1.7081e+00],
[ 1.9440e+00, 1.2168e+00, 2.3270e+00, ..., 5.1224e-01,
-5.1084e-01, 1.3141e+00]]],
[[[-1.4912e-01, 6.6034e-01, 3.4397e-01, ..., -1.7121e+00,
-1.0148e+00, -8.3546e-01],
[ 7.0080e-01, 6.0954e-01, 1.4586e+00, ..., -4.3937e-02,
-4.9815e-03, -1.1733e+00],
[ 1.0335e+00, 1.2103e-01, 1.3866e+00, ..., 4.4525e-01,
-1.4469e+00, -2.2790e+00],
...,
[-2.4109e-01, -4.0273e+00, -2.8938e-01, ..., 2.5656e+00,
-2.6642e-01, -1.4359e+00],
[-1.7359e+00, -1.4372e+00, 5.7562e-01, ..., 1.7548e+00,
-1.5464e+00, -5.4970e-01],
[-1.3875e+00, -2.6327e+00, -3.0269e+00, ..., -1.3512e+00,
-3.6524e+00, -2.1447e+00]],
[[-7.3563e-01, -1.4091e+00, -2.0667e+00, ..., -3.4132e+00,
-4.3444e+00, -2.4611e+00],
[-1.9359e+00, -3.6224e+00, -3.9164e+00, ..., -4.0319e+00,
-5.2214e+00, -4.6372e+00],
[-1.7538e+00, -3.4083e+00, -4.0539e+00, ..., -5.3380e+00,
-5.3486e+00, -3.8573e+00],
...,
[-5.2639e+00, -8.4820e+00, -6.2530e+00, ..., -2.2966e+00,
-7.3259e+00, -3.2992e+00],
[-8.5253e+00, -9.2612e+00, -9.6016e+00, ..., -5.7815e+00,
-9.6252e+00, -4.2693e+00],
[-1.9199e+00, -4.2055e+00, -3.6363e+00, ..., -2.5589e+00,
-4.5290e+00, -1.1186e+00]],
[[ 6.8526e-01, 1.1593e+00, 1.0828e+00, ..., 1.5568e+00,
1.7671e+00, 8.7544e-01],
[ 8.7651e-01, 1.3651e+00, 1.4274e+00, ..., 1.4774e+00,
1.7313e+00, 9.4515e-01],
[ 5.2672e-01, 3.1036e+00, 2.0939e+00, ..., 2.0629e+00,
2.0109e+00, 1.9190e+00],
...,
[-1.4176e-01, 2.6980e-01, -3.9748e+00, ..., -2.7504e+00,
2.6325e+00, 7.3889e-01],
[-1.1893e+00, 2.0433e-01, -1.2511e+00, ..., -2.2217e+00,
1.6751e+00, -3.3960e-03],
[-1.9456e-01, -4.4457e-01, 2.5470e-01, ..., 8.4567e-01,
1.5442e+00, -1.0048e+00]],
...,
[[-1.0103e+00, -1.1497e+00, -1.1012e+00, ..., -2.6311e+00,
-3.5239e+00, -1.2622e+00],
[-2.3461e+00, -2.8998e+00, -1.2548e+00, ..., -1.9725e+00,
-4.8713e+00, -3.6374e+00],
[-9.5276e-01, -1.5319e+00, -6.2087e-02, ..., -1.6674e+00,
-5.6903e+00, -3.3010e+00],
...,
[-2.0405e+00, -4.7495e+00, -9.0598e-01, ..., -3.0522e+00,
-7.0020e+00, -1.8308e+00],
[-2.5941e+00, -6.1921e+00, -4.2315e+00, ..., -5.1705e+00,
-7.6486e+00, -7.8338e-01],
[-2.1695e+00, -3.0287e+00, -2.4993e+00, ..., -2.9502e+00,
-4.2256e+00, 2.2913e-01]],
[[-1.5193e+00, -2.1687e+00, -3.1584e+00, ..., -2.9833e+00,
-3.1613e+00, -5.8463e-01],
[-4.3675e+00, -4.3796e+00, -4.8771e+00, ..., -3.8698e+00,
-6.0584e+00, -2.1069e+00],
[-4.0465e+00, -4.9392e+00, -5.1296e+00, ..., -3.1438e+00,
-5.5016e+00, -2.0857e+00],
...,
[-4.9222e+00, -2.6404e+00, 1.2333e+00, ..., -1.2372e+00,
-2.9559e+00, -5.2996e-01],
[-3.4664e+00, -1.7774e+00, 4.9405e-01, ..., -2.2431e+00,
-3.5921e+00, 4.1128e-01],
[-2.9237e+00, -2.7354e+00, -1.9839e+00, ..., -2.6607e+00,
-3.5603e+00, 1.5971e-01]],
[[ 8.9639e-01, -2.2453e-01, 5.2767e-01, ..., 4.2293e-01,
1.0635e+00, 6.2445e-01],
[ 1.7290e+00, 8.3029e-01, 5.7293e-01, ..., 1.0430e+00,
1.3436e+00, 3.0764e+00],
[ 1.3759e+00, 2.5423e-01, 1.0720e+00, ..., 1.6494e+00,
4.7331e-01, 7.3000e-01],
...,
[ 3.2105e+00, 2.3904e+00, 4.8189e+00, ..., 1.5533e+00,
3.4113e+00, 1.3687e+00],
[ 3.9112e+00, 3.4373e+00, 4.1045e+00, ..., 2.7478e+00,
1.0306e+00, 1.4749e+00],
[ 2.8751e+00, 2.4656e+00, 1.0316e+00, ..., 2.2391e+00,
4.1483e-01, 1.2804e+00]]],
[[[-4.2928e-02, 7.8236e-01, -2.2727e-01, ..., -1.2900e+00,
-9.8259e-01, -8.1830e-01],
[ 6.1219e-01, 3.6689e-01, -1.5465e-01, ..., -6.1295e-01,
2.1591e-01, -1.0299e+00],
[ 1.4175e+00, 6.3753e-01, 1.7070e+00, ..., -1.3227e+00,
-4.6416e-01, -1.9777e+00],
...,
[-3.6539e-01, -6.9425e-01, 1.2179e-01, ..., -2.2893e+00,
-2.2947e+00, 5.6512e-02],
[-2.3761e+00, -1.7090e+00, 4.9646e-01, ..., -2.3007e+00,
-2.6889e+00, 2.2316e-01],
[-2.3952e+00, -3.9543e+00, -2.9217e+00, ..., -4.1443e+00,
-3.5633e+00, -1.4462e+00]],
[[-6.5159e-01, -1.3977e+00, -2.5173e+00, ..., -2.9599e+00,
-4.5769e+00, -2.4057e+00],
[-1.8239e+00, -3.6984e+00, -4.2033e+00, ..., -3.0820e+00,
-4.7680e+00, -4.5736e+00],
[-1.5027e+00, -3.8209e+00, -4.3286e+00, ..., -3.3923e+00,
-3.9234e+00, -3.6472e+00],
...,
[-6.9676e+00, -9.0450e+00, -7.1346e+00, ..., -7.4509e+00,
-8.4332e+00, -1.7886e+00],
[-9.8806e+00, -1.0375e+01, -8.4711e+00, ..., -1.0779e+01,
-1.0976e+01, -3.6860e+00],
[-2.5743e+00, -4.8662e+00, -4.1210e+00, ..., -4.1347e+00,
-4.2144e+00, -5.9901e-01]],
[[ 6.5647e-01, 1.1940e+00, 1.0367e+00, ..., 1.5424e+00,
1.8108e+00, 8.3824e-01],
[ 8.8688e-01, 1.6980e+00, 1.7052e+00, ..., 1.0743e+00,
2.2097e+00, 8.8820e-01],
[ 6.0757e-01, 3.3258e+00, 1.9284e+00, ..., 1.2838e+00,
2.5490e+00, 2.0239e+00],
...,
[-1.3718e-01, -4.9950e-01, 3.6879e-01, ..., 9.4477e-01,
1.3233e+00, -1.6567e-01],
[-4.6313e-01, 5.7185e-01, -6.0906e-01, ..., 5.8857e-01,
1.2777e+00, -3.7241e-01],
[ 6.3865e-01, 1.2027e+00, 9.4769e-01, ..., -2.5023e-01,
-4.5619e-01, -1.9276e+00]],
...,
[[-9.2861e-01, -1.2869e+00, -2.0111e+00, ..., -2.5916e+00,
-3.2442e+00, -1.1695e+00],
[-2.2541e+00, -3.3781e+00, -2.7399e+00, ..., -2.8773e+00,
-4.1523e+00, -3.4240e+00],
[-5.4477e-01, -1.4612e+00, -8.7932e-01, ..., -2.8923e+00,
-4.1805e+00, -3.0109e+00],
...,
[-2.8072e+00, -3.8164e+00, -1.6178e+00, ..., -4.9840e+00,
-6.2170e+00, -2.6052e+00],
[-3.9525e+00, -5.9114e+00, -2.9897e+00, ..., -4.5492e+00,
-5.2176e+00, -1.2249e+00],
[-3.2556e+00, -3.4859e+00, -3.0427e+00, ..., -2.6490e+00,
-2.5265e+00, 4.8852e-01]],
[[-1.4800e+00, -2.3327e+00, -3.4808e+00, ..., -2.8850e+00,
-2.7371e+00, -5.1959e-01],
[-4.3524e+00, -4.7397e+00, -4.4816e+00, ..., -4.1724e+00,
-5.7617e+00, -1.9653e+00],
[-3.8600e+00, -4.9654e+00, -2.9627e+00, ..., -3.1594e+00,
-5.0355e+00, -2.1064e+00],
...,
[-4.5365e+00, -3.8529e+00, -4.1461e+00, ..., -3.9387e+00,
-2.4243e+00, -1.2077e+00],
[-3.5402e+00, -9.2289e-01, -1.9979e+00, ..., -2.0098e+00,
-8.8139e-01, 4.2296e-01],
[-3.7710e+00, -2.9622e+00, -2.9704e+00, ..., -1.8543e+00,
-1.1317e+00, 4.0530e-01]],
[[ 8.8640e-01, -1.1134e-01, 7.0328e-01, ..., 5.0502e-01,
1.3708e+00, 6.0550e-01],
[ 1.6110e+00, 8.0783e-01, 6.0898e-01, ..., 6.1226e-01,
1.8252e+00, 3.1972e+00],
[ 1.3133e+00, 4.3984e-01, 1.7565e+00, ..., -3.6103e-02,
1.0056e+00, 8.3213e-01],
...,
[ 4.3384e+00, 3.7342e+00, 2.3687e+00, ..., 3.9836e+00,
2.7906e+00, 5.5195e-01],
[ 4.5405e+00, 4.2490e+00, 3.5903e+00, ..., 2.2229e+00,
1.7309e+00, 1.3908e+00],
[ 2.8338e+00, 2.7338e+00, 2.0694e+00, ..., 5.0888e-01,
5.4094e-01, 1.0291e+00]]],
[[[ 2.2497e-01, 1.2886e+00, 4.4325e-01, ..., -9.7275e-01,
-1.9049e-01, 7.4127e-01],
[ 3.6447e-01, -4.6896e-02, 5.1436e-01, ..., 8.8340e-02,
-5.3915e-01, 1.0968e+00],
[-7.5900e-01, -2.3674e-01, 6.9906e-01, ..., 2.3690e+00,
-3.1331e-01, 6.9990e-01],
...,
[-1.0671e+00, -1.8770e+00, -2.7938e-01, ..., -2.2331e+00,
5.8125e-02, -6.6430e-01],
[-1.0649e+00, -9.1285e-01, 5.0016e-02, ..., -1.2249e+00,
-1.8172e-02, -1.6403e+00],
[-2.6588e+00, -5.0666e+00, -3.8448e+00, ..., -4.9817e+00,
-4.1047e+00, -3.8462e+00]],
[[-8.8455e-01, -2.3291e+00, -4.1577e+00, ..., -2.9039e+00,
-4.9959e+00, -2.6254e-01],
[-5.2126e+00, -6.4492e+00, -5.0261e+00, ..., -3.4782e+00,
-7.2798e+00, -1.5358e+00],
[-3.9099e+00, -4.4996e+00, -3.2312e+00, ..., -1.0651e+00,
-6.5267e+00, -1.5540e+00],
...,
[-5.8348e+00, -6.0020e+00, -3.2811e+00, ..., -5.0792e+00,
-4.9409e+00, -3.7322e+00],
[-7.0862e+00, -7.7706e+00, -6.0489e+00, ..., -7.0319e+00,
-8.0144e+00, -6.5363e+00],
[-2.5327e+00, -4.4605e+00, -4.1309e+00, ..., -4.0274e+00,
-4.1456e+00, -2.5325e+00]],
[[-3.5552e-01, 2.1170e+00, 1.2555e+00, ..., 1.5109e+00,
2.2179e+00, 9.1108e-01],
[-7.7434e-02, 9.6548e-01, 9.8901e-01, ..., 1.5416e+00,
1.4263e+00, -6.9729e-01],
[ 1.2246e-01, 1.6958e+00, 1.0281e+00, ..., 1.4215e+00,
2.2208e+00, -6.8668e-02],
...,
[ 1.0240e+00, 1.4775e+00, 9.7341e-01, ..., 6.1860e-01,
1.0811e+00, 1.8049e-01],
[ 8.2834e-01, 6.3766e-01, 6.9490e-01, ..., 4.1979e-01,
3.3960e-01, 2.1154e-01],
[ 1.3195e+00, 1.4522e+00, 1.0059e+00, ..., 1.7048e+00,
1.1284e+00, 1.3717e-01]],
...,
[[-9.2638e-01, -1.8779e+00, -1.9596e+00, ..., -2.4873e+00,
-2.9838e+00, -4.8435e-01],
[-3.4445e+00, -4.2199e+00, -2.6155e+00, ..., -2.2567e+00,
-3.8987e+00, -2.6654e+00],
[-1.7463e+00, -1.9391e+00, -8.7777e-01, ..., -2.0252e+00,
-3.2794e+00, -1.4591e+00],
...,
[-2.7692e+00, -2.5155e+00, -1.6666e+00, ..., -2.3922e+00,
-3.4609e+00, -1.6438e+00],
[-2.9204e+00, -3.0106e+00, -2.0454e+00, ..., -2.1329e+00,
-4.1469e+00, -2.3455e+00],
[-3.4638e+00, -3.7774e+00, -2.8145e+00, ..., -3.4046e+00,
-3.6649e+00, -1.6585e+00]],
[[-1.7117e+00, -2.1888e+00, -2.9444e+00, ..., -5.6989e-01,
-1.3505e+00, -1.3867e+00],
[-4.4894e+00, -3.3409e+00, -2.9870e+00, ..., -2.0387e+00,
-1.1659e+00, -1.0588e+00],
[-3.6508e+00, -2.6467e+00, -2.7110e+00, ..., -2.2292e-01,
-1.3391e+00, -1.0155e+00],
...,
[-4.3124e+00, -2.1295e+00, -5.1120e+00, ..., -2.2291e+00,
-3.0117e+00, -1.7178e+00],
[-3.7101e+00, -2.7462e-01, -3.5763e+00, ..., -1.0344e+00,
-3.0012e+00, -1.3944e+00],
[-4.3914e+00, -2.6304e+00, -2.6762e+00, ..., -2.0547e+00,
-3.2766e+00, -1.2702e+00]],
[[ 1.1155e+00, 4.7781e-01, 6.9865e-01, ..., -2.7432e-01,
-7.7933e-02, -5.5562e-01],
[ 3.7554e+00, 3.2621e+00, 2.2778e+00, ..., 3.6006e+00,
2.6904e+00, 2.1633e+00],
[ 1.5277e+00, 6.3760e-01, 7.7330e-02, ..., 3.1977e+00,
2.3773e+00, 1.3202e+00],
...,
[ 2.7527e+00, 1.3393e+00, 1.6745e+00, ..., 2.1867e+00,
1.9168e+00, 1.3121e+00],
[ 3.7104e+00, 3.5471e+00, 1.9435e+00, ..., 2.7493e+00,
2.2322e+00, 2.1319e+00],
[ 1.9390e+00, 8.4835e-01, 2.7712e-01, ..., -1.8809e-01,
-2.7781e-01, -1.3651e-01]]]], device='cuda:0')]}
In [5]:
for layer_name, acts in activations.items():
if len(acts) > 1:
acts = torch.cat(acts, dim=0) # Shape: [Batch x Channels x Height x Width]
avg_activation = acts.mean(dim=(0, 2, 3)) # Shape: [Channels]
top_indices = torch.topk(avg_activation, 10).indices
print(f"Top 10 feature maps for {layer_name}: {top_indices.tolist()}")
Top 10 feature maps for rgb_features: [47, 5, 43, 61, 46, 62, 17, 37, 38, 10] Top 10 feature maps for d_features: [8, 53, 40, 62, 47, 5, 17, 57, 23, 32] Top 10 feature maps for features: [24, 18, 38, 36, 9, 30, 45, 60, 15, 54] Top 10 feature maps for cls: [1, 15, 8, 12, 5, 3, 9, 11, 14, 10]
In [6]:
avg_activation
Out[6]:
tensor([-1.5345, -0.8570, -1.4611, -1.2415, -2.6134], device='cuda:0')
In [6]:
# Debugging: Check stored activations
for layer_name, acts in activations.items():
print(f"Layer: {layer_name}, Activations Count: {len(acts)}")
if len(acts) > 0:
print(f"Activation Shape: {acts[0].shape}")
Layer: rgb_features, Activations Count: 400 Activation Shape: torch.Size([1, 64, 27, 27]) Layer: d_features, Activations Count: 400 Activation Shape: torch.Size([1, 64, 27, 27]) Layer: features, Activations Count: 400 Activation Shape: torch.Size([1, 64, 13, 13]) Layer: grasp, Activations Count: 0 Layer: cls, Activations Count: 400 Activation Shape: torch.Size([1, 5, 224, 224])
In [6]:
# Get test acc for Grasp model
accuracy, loss = get_grasp_acc(model, include_depth=True, seed=None, dataset=path, truncation=None)
Dataset size: 400 Training steps: 360 -- Val steps: 40
In [7]:
# Debugging: Check stored activations
for layer_name, acts in activations.items():
print(f"Layer: {layer_name}, Activations Count: {len(acts)}")
if len(acts) > 0:
print(f"Activation Shape: {acts[0].shape}")
Layer: rgb_features, Activations Count: 800 Activation Shape: torch.Size([1, 64, 27, 27]) Layer: d_features, Activations Count: 800 Activation Shape: torch.Size([1, 64, 27, 27]) Layer: features, Activations Count: 800 Activation Shape: torch.Size([1, 64, 13, 13]) Layer: grasp, Activations Count: 400 Activation Shape: torch.Size([1, 16, 54, 54]) Layer: cls, Activations Count: 400 Activation Shape: torch.Size([1, 16, 54, 54])
In [8]:
for layer_name, acts in activations.items():
if len(acts) > 1:
acts = torch.cat(acts, dim=0) # Shape: [Batch x Channels x Height x Width]
avg_activation = acts.mean(dim=(0, 2, 3)) # Shape: [Channels]
top_indices = torch.topk(avg_activation, 10).indices
print(f"Top 10 feature maps for {layer_name}: {top_indices.tolist()}")
Top 10 feature maps for rgb_features: [47, 5, 43, 61, 46, 62, 17, 37, 38, 10] Top 10 feature maps for d_features: [8, 53, 40, 62, 47, 5, 17, 57, 23, 32] Top 10 feature maps for features: [24, 18, 38, 36, 9, 30, 45, 60, 15, 54] Top 10 feature maps for grasp: [2, 3, 7, 8, 10, 5, 12, 4, 1, 6] Top 10 feature maps for cls: [1, 15, 8, 12, 5, 3, 9, 11, 14, 10]
In [9]:
print('Grasp: %s' % accuracy, loss)
print('CLS: %s' % c_accuracy, c_loss)
Grasp: 81.5 0.0 CLS: 85.0 0.0
For Shuffled Data¶
In [12]:
"""
This file tests the model's performance on the testing dataset.
For CLS, this script returns the testing accuracy.
For Grasp, this script returns the testing accuracy and visualizes the
grasp prediction.
Comment or uncomment certain lines of code for swapping between
training CLS model and Grasping model.
E.g. Uncomment the lines with NO SPACE between '#' and the codes:
# Get test acc for CLS model
#accuracy, loss = get_test_acc(model)
# Get test acc for Grasp model
accuracy, loss = get_grasp_acc(model)
----->
# Get test acc for CLS model
accuracy, loss = get_test_acc(model)
# Get test acc for Grasp model
#accuracy, loss = get_grasp_acc(model)
"""
import torch
import os
import time
import sys
from utils.parameters import Params
from multi_task_models.grcn_multi_alex import Multi_AlexnetMap_v3
from training.single_task.evaluation import get_cls_acc, get_grasp_acc, visualize_grasp, visualize_cls
params = Params()
model_name = params.MODEL_NAME
weights_dir = params.MODEL_PATH
weights_path = os.path.join(weights_dir, model_name, model_name + '_final.pth')
# AlexNet with 1st, 2nd layer pretrained on Imagenet
model = Multi_AlexnetMap_v3().to('cuda')
model.load_state_dict(torch.load(weights_path))
model.eval()
path = params.TEST_PATH_SHUFFLE
# if len(sys.argv) > 1:
# path = params.TEST_PATH_SHUFFLE
# Visualize CLS predictions one by one
#visualize_cls(model)
# Visualize grasp predictions one by one
#visualize_grasp(model)
/home/billy/anaconda3/envs/gr_cls_multitask/lib/python3.11/site-packages/torchvision/models/_utils.py:208: UserWarning: The parameter 'pretrained' is deprecated since 0.13 and may be removed in the future, please use 'weights' instead. warnings.warn( /home/billy/anaconda3/envs/gr_cls_multitask/lib/python3.11/site-packages/torchvision/models/_utils.py:223: UserWarning: Arguments other than a weight enum or `None` for 'weights' are deprecated since 0.13 and may be removed in the future. The current behavior is equivalent to passing `weights=AlexNet_Weights.IMAGENET1K_V1`. You can also use `weights=AlexNet_Weights.DEFAULT` to get the most up-to-date weights. warnings.warn(msg) /tmp/ipykernel_37298/609354409.py:41: FutureWarning: You are using `torch.load` with `weights_only=False` (the current default value), which uses the default pickle module implicitly. It is possible to construct malicious pickle data which will execute arbitrary code during unpickling (See https://github.com/pytorch/pytorch/blob/main/SECURITY.md#untrusted-models for more details). In a future release, the default value for `weights_only` will be flipped to `True`. This limits the functions that could be executed during unpickling. Arbitrary objects will no longer be allowed to be loaded via this mode unless they are explicitly allowlisted by the user via `torch.serialization.add_safe_globals`. We recommend you start setting `weights_only=True` for any use case where you don't have full control of the loaded file. Please open an issue on GitHub for any issues related to this experimental feature. model.load_state_dict(torch.load(weights_path))
In [13]:
# Register hooks
hooks, activations = register_custom_hooks(model)
In [14]:
# Get test acc for CLS model
c_accuracy, c_loss = get_cls_acc(model, include_depth=True, seed=None, dataset=path, truncation=None)
Dataset size: 400 Training steps: 72 -- Val steps: 8
In [15]:
for layer_name, acts in activations.items():
if len(acts) > 1:
acts = torch.cat(acts, dim=0) # Shape: [Batch x Channels x Height x Width]
avg_activation = acts.mean(dim=(0, 2, 3)) # Shape: [Channels]
top_indices = torch.topk(avg_activation, 10).indices
print(f"Top 10 feature maps for {layer_name}: {top_indices.tolist()}")
Top 10 feature maps for rgb_features: [47, 5, 43, 61, 46, 62, 17, 37, 38, 10] Top 10 feature maps for d_features: [8, 53, 40, 62, 47, 5, 17, 32, 23, 43] Top 10 feature maps for features: [24, 18, 38, 9, 36, 45, 30, 60, 15, 54] Top 10 feature maps for cls: [15, 1, 8, 12, 5, 3, 9, 11, 14, 0]
In [16]:
# Get test acc for Grasp model
accuracy, loss = get_grasp_acc(model, include_depth=True, seed=None, dataset=path, truncation=None)
Dataset size: 400 Training steps: 72 -- Val steps: 8
In [17]:
for layer_name, acts in activations.items():
if len(acts) > 1:
acts = torch.cat(acts, dim=0) # Shape: [Batch x Channels x Height x Width]
avg_activation = acts.mean(dim=(0, 2, 3)) # Shape: [Channels]
top_indices = torch.topk(avg_activation, 10).indices
print(f"Top 10 feature maps for {layer_name}: {top_indices.tolist()}")
Top 10 feature maps for rgb_features: [47, 5, 43, 61, 46, 62, 17, 37, 38, 10] Top 10 feature maps for d_features: [8, 53, 40, 62, 47, 5, 17, 32, 23, 43] Top 10 feature maps for features: [24, 18, 38, 9, 36, 45, 30, 60, 15, 54] Top 10 feature maps for grasp: [2, 3, 7, 8, 10, 5, 12, 4, 1, 11] Top 10 feature maps for cls: [15, 1, 8, 12, 5, 3, 9, 11, 14, 0]
In [18]:
print('Grasp: %s' % accuracy, loss)
print('CLS: %s' % c_accuracy, c_loss)
Grasp: 53.25 0.0 CLS: 66.25 0.0